avila-molecule 0.1.0

Network protocol molecules - TCP, UDP, TLS building blocks
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

//! UDP (User Datagram Protocol) - Comunicação sem conexão

use crate::NetworkAddress;
use avila_error::{Error, ErrorKind, Result};
use tokio::net::UdpSocket;

/// Socket UDP
pub struct UdpEndpoint {
    socket: UdpSocket,
    address: NetworkAddress,
}

impl UdpEndpoint {
    /// Cria novo endpoint UDP
    pub async fn bind(address: NetworkAddress) -> Result<Self> {
        let socket = UdpSocket::bind(address.to_string())
            .await
            .map_err(|e| Error::new(ErrorKind::Network, format!("Falha ao vincular UDP: {}", e)))?;

        Ok(Self { socket, address })
    }

    /// Envia datagrama
    pub async fn send_to(&self, data: &[u8], target: &NetworkAddress) -> Result<usize> {
        self.socket
            .send_to(data, target.to_string())
            .await
            .map_err(|e| Error::new(ErrorKind::Network, format!("Falha ao enviar: {}", e)))
    }

    /// Recebe datagrama
    pub async fn receive_from(&self, buffer: &mut [u8]) -> Result<(usize, std::net::SocketAddr)> {
        self.socket
            .recv_from(buffer)
            .await
            .map_err(|e| Error::new(ErrorKind::Network, format!("Falha ao receber: {}", e)))
    }

    /// Endereço local
    pub fn address(&self) -> &NetworkAddress {
        &self.address
    }
}