add async waker methods and async feature

tritiumcan-smoltcp/Cargo.toml

@@ -20,3 +20,4 @@ zerocopy = { version = "0.7.34", features = ["derive"] }
 
 [features]
 defmt-03 = ["dep:defmt", "smoltcp/defmt", "tritiumcan/defmt-03"]
+async = ["smoltcp/async"]

tritiumcan-smoltcp/src/tcp.rs

@@ -165,4 +165,34 @@ impl Server {
             Ok(Some(frame))
         }
     }
+
+    /// Register a waker for receive operations.
+    ///
+    /// See [smoltcp documentation](https://docs.rs/smoltcp/latest/smoltcp/socket/tcp/struct.Socket.html#method.register_recv_waker)
+    /// for the rules around receive wakers.
+    #[cfg(feature = "async")]
+    pub fn register_recv_waker(
+        &mut self,
+        sockets: &mut SocketSet,
+        waker: &core::task::Waker,
+    ) {
+        let socket = sockets.get_mut::<Socket>(self.handle);
+
+        socket.register_recv_waker(waker);
+    }
+
+    /// Register a waker for send operations.
+    ///
+    /// See [smoltcp documentation](https://docs.rs/smoltcp/latest/smoltcp/socket/tcp/struct.Socket.html#method.register_send_waker)
+    /// for the rules around receive wakers.
+    #[cfg(feature = "async")]
+    pub fn register_send_waker(
+        &mut self,
+        sockets: &mut SocketSet,
+        waker: &core::task::Waker,
+    ) {
+        let socket = sockets.get_mut::<Socket>(self.handle);
+
+        socket.register_send_waker(waker);
+    }
 }

tritiumcan-smoltcp/src/udp.rs

@@ -159,4 +159,34 @@ impl Server {
             Ok(Some(packet.frame.into()))
         }
     }
+
+    /// Register a waker for receive operations.
+    ///
+    /// See [smoltcp documentation](https://docs.rs/smoltcp/latest/smoltcp/socket/udp/struct.Socket.html#method.register_recv_waker)
+    /// for the rules around receive wakers.
+    #[cfg(feature = "async")]
+    pub fn register_recv_waker(
+        &mut self,
+        sockets: &mut SocketSet,
+        waker: &core::task::Waker,
+    ) {
+        let socket = sockets.get_mut::<Socket>(self.handle);
+
+        socket.register_recv_waker(waker);
+    }
+
+    /// Register a waker for send operations.
+    ///
+    /// See [smoltcp documentation](https://docs.rs/smoltcp/latest/smoltcp/socket/udp/struct.Socket.html#method.register_send_waker)
+    /// for the rules around receive wakers.
+    #[cfg(feature = "async")]
+    pub fn register_send_waker(
+        &mut self,
+        sockets: &mut SocketSet,
+        waker: &core::task::Waker,
+    ) {
+        let socket = sockets.get_mut::<Socket>(self.handle);
+
+        socket.register_send_waker(waker);
+    }
 }