udp: Add optional broadcast capabilities

Signed-off-by: Konrad Gräfe <[email protected]>

Author: kgraefe

src/adapters/udp.rs

@@ -26,6 +26,11 @@ pub const MAX_LOCAL_PAYLOAD_LEN: usize = 65535 - 20 - 8;
 #[cfg(target_os = "macos")]
 pub const MAX_LOCAL_PAYLOAD_LEN: usize = 9216 - 20 - 8;
 
+#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug, Default)]
+pub struct UdpConfig {
+    set_broadcast: bool,
+}
+
 pub(crate) struct UdpAdapter;
 impl Adapter for UdpAdapter {
     type Remote = RemoteResource;
@@ -44,11 +49,21 @@ impl Resource for RemoteResource {
 
 impl Remote for RemoteResource {
     fn connect_with(
-        _: TransportConnect,
+        config: TransportConnect,
         remote_addr: RemoteAddr,
     ) -> io::Result<ConnectionInfo<Self>> {
+        let config = match config {
+            TransportConnect::Udp(config) => config,
+            _ => unreachable!(),
+        };
+
         let socket = UdpSocket::bind("0.0.0.0:0".parse().unwrap())?;
         let peer_addr = *remote_addr.socket_addr();
+
+        if config.set_broadcast {
+            socket.set_broadcast(true)?;
+        }
+
         socket.connect(peer_addr)?;
         let local_addr = socket.local_addr()?;
         Ok(ConnectionInfo { remote: RemoteResource { socket }, local_addr, peer_addr })
@@ -98,7 +113,12 @@ impl Resource for LocalResource {
 impl Local for LocalResource {
     type Remote = RemoteResource;
 
-    fn listen_with(_: TransportListen, addr: SocketAddr) -> io::Result<ListeningInfo<Self>> {
+    fn listen_with(config: TransportListen, addr: SocketAddr) -> io::Result<ListeningInfo<Self>> {
+        let config = match config {
+            TransportListen::Udp(config) => config,
+            _ => unreachable!(),
+        };
+
         let socket = match addr {
             SocketAddr::V4(addr) if addr.ip().is_multicast() => {
                 let listening_addr = SocketAddrV4::new(Ipv4Addr::UNSPECIFIED, addr.port());
@@ -115,6 +135,10 @@ impl Local for LocalResource {
             _ => UdpSocket::bind(addr)?,
         };
 
+        if config.set_broadcast {
+            socket.set_broadcast(true)?;
+        }
+
         let local_addr = socket.local_addr().unwrap();
         Ok(ListeningInfo { local: { LocalResource { socket } }, local_addr })
     }

src/network/transport.rs

@@ -5,7 +5,7 @@ use crate::adapters::tcp::{TcpAdapter};
 #[cfg(feature = "tcp")]
 use crate::adapters::framed_tcp::{FramedTcpAdapter};
 #[cfg(feature = "udp")]
-use crate::adapters::udp::{self, UdpAdapter};
+use crate::adapters::udp::{self, UdpAdapter, UdpConfig};
 #[cfg(feature = "websocket")]
 use crate::adapters::ws::{self, WsAdapter};
 
@@ -160,7 +160,7 @@ impl std::fmt::Display for Transport {
 pub enum TransportConnect {
     Tcp,
     FramedTcp,
-    Udp,
+    Udp(UdpConfig),
     Ws,
 }
 
@@ -169,7 +169,7 @@ impl TransportConnect {
         let transport = match self {
             Self::Tcp => Transport::Tcp,
             Self::FramedTcp => Transport::FramedTcp,
-            Self::Udp => Transport::Udp,
+            Self::Udp(_) => Transport::Udp,
             Self::Ws => Transport::Ws,
         };
 
@@ -182,7 +182,7 @@ impl From<Transport> for TransportConnect {
         match transport {
             Transport::Tcp => Self::Tcp,
             Transport::FramedTcp => Self::FramedTcp,
-            Transport::Udp => Self::Udp,
+            Transport::Udp => Self::Udp(UdpConfig::default()),
             Transport::Ws => Self::Ws,
         }
     }
@@ -191,7 +191,7 @@ impl From<Transport> for TransportConnect {
 pub enum TransportListen {
     Tcp,
     FramedTcp,
-    Udp,
+    Udp(UdpConfig),
     Ws,
 }
 
@@ -200,7 +200,7 @@ impl TransportListen {
         let transport = match self {
             Self::Tcp => Transport::Tcp,
             Self::FramedTcp => Transport::FramedTcp,
-            Self::Udp => Transport::Udp,
+            Self::Udp(_) => Transport::Udp,
             Self::Ws => Transport::Ws,
         };
 
@@ -213,7 +213,7 @@ impl From<Transport> for TransportListen {
         match transport {
             Transport::Tcp => Self::Tcp,
             Transport::FramedTcp => Self::FramedTcp,
-            Transport::Udp => Self::Udp,
+            Transport::Udp => Self::Udp(UdpConfig::default()),
             Transport::Ws => Self::Ws,
         }
     }