tribufu/rust-gamedig
1
0
Fork 0
mirror of https://github.com/tribufu/rust-gamedig synced 2026-06-01 09:42:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

socket: Add method to get local address

Browse source
This commit is contained in:
Douile 2023-11-22 01:11:50 +00:00
parent 7510fe3de0
commit 3b1edd8e3d
No known key found for this signature in database
GPG key ID: 1EC3120F9A1028F2
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
crates/lib/src/socket.rs
View file

@ -24,7 +24,10 @@ pub trait Socket {
fn send(&mut self, data: &[u8]) -> GDResult<()>; fn send(&mut self, data: &[u8]) -> GDResult<()>;
fn receive(&mut self, size: Option<usize>) -> GDResult<Vec<u8>>; fn receive(&mut self, size: Option<usize>) -> GDResult<Vec<u8>>;
/// Get the remote port
fn port(&self) -> u16; fn port(&self) -> u16;
/// Get the local SocketAddr
fn local_addr(&self) -> std::io::Result<SocketAddr>;
} }
pub struct TcpSocket { pub struct TcpSocket {
@ -73,6 +76,7 @@ impl Socket for TcpSocket {
} }
fn port(&self) -> u16 { self.address.port() } fn port(&self) -> u16 { self.address.port() }
fn local_addr(&self) -> std::io::Result<SocketAddr> { self.socket.local_addr() }
} }
pub struct UdpSocket { pub struct UdpSocket {
@ -121,6 +125,7 @@ impl Socket for UdpSocket {
} }
fn port(&self) -> u16 { self.address.port() } fn port(&self) -> u16 { self.address.port() }
fn local_addr(&self) -> std::io::Result<SocketAddr> { self.socket.local_addr() }
} }
#[cfg(test)] #[cfg(test)]