udp: Add optional broadcast capabilities

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

Author: kgraefe

src/adapters/udp.rs

@@ -26,6 +26,16 @@ 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 UdpConnectConfig {
+    pub set_broadcast: bool,
+}
+
+#[derive(Clone, Copy, PartialEq, Eq, Hash, Debug, Default)]
+pub struct UdpListenConfig {
+    pub set_broadcast: bool,
+}
+
 pub(crate) struct UdpAdapter;
 impl Adapter for UdpAdapter {
     type Remote = RemoteResource;
@@ -44,11 +54,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 +118,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 +140,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, UdpConnectConfig, UdpListenConfig};
 #[cfg(feature = "websocket")]
 use crate::adapters::ws::{self, WsAdapter};
 
@@ -163,7 +163,7 @@ pub enum TransportConnect {
     #[cfg(feature = "tcp")]
     FramedTcp,
     #[cfg(feature = "udp")]
-    Udp,
+    Udp(UdpConnectConfig),
     #[cfg(feature = "websocket")]
     Ws,
 }
@@ -176,7 +176,7 @@ impl TransportConnect {
             #[cfg(feature = "tcp")]
             Self::FramedTcp => Transport::FramedTcp,
             #[cfg(feature = "udp")]
-            Self::Udp => Transport::Udp,
+            Self::Udp(_) => Transport::Udp,
             #[cfg(feature = "websocket")]
             Self::Ws => Transport::Ws,
         };
@@ -193,7 +193,7 @@ impl From<Transport> for TransportConnect {
             #[cfg(feature = "tcp")]
             Transport::FramedTcp => Self::FramedTcp,
             #[cfg(feature = "udp")]
-            Transport::Udp => Self::Udp,
+            Transport::Udp => Self::Udp(UdpConnectConfig::default()),
             #[cfg(feature = "websocket")]
             Transport::Ws => Self::Ws,
         }
@@ -206,7 +206,7 @@ pub enum TransportListen {
     #[cfg(feature = "tcp")]
     FramedTcp,
     #[cfg(feature = "udp")]
-    Udp,
+    Udp(UdpListenConfig),
     #[cfg(feature = "websocket")]
     Ws,
 }
@@ -219,7 +219,7 @@ impl TransportListen {
             #[cfg(feature = "tcp")]
             Self::FramedTcp => Transport::FramedTcp,
             #[cfg(feature = "udp")]
-            Self::Udp => Transport::Udp,
+            Self::Udp(_) => Transport::Udp,
             #[cfg(feature = "websocket")]
             Self::Ws => Transport::Ws,
         };
@@ -236,7 +236,7 @@ impl From<Transport> for TransportListen {
             #[cfg(feature = "tcp")]
             Transport::FramedTcp => Self::FramedTcp,
             #[cfg(feature = "udp")]
-            Transport::Udp => Self::Udp,
+            Transport::Udp => Self::Udp(UdpListenConfig::default()),
             #[cfg(feature = "websocket")]
             Transport::Ws => Self::Ws,
         }