Merge #470

470: Add TCP listen. r=Dirbaio a=matoushybl



Co-authored-by: Matous Hybl <[email protected]>

1 files changed, 20 insertions, 0 deletions

diff --git a/embassy-net/src/tcp_socket.rs b/embassy-net/src/tcp_socket.rs
index 2433771c8..bb1b626e7 100644
--- a/embassy-net/src/tcp_socket.rs
+++ b/embassy-net/src/tcp_socket.rs
@@ -58,6 +58,26 @@ impl<'a> TcpSocket<'a> {
         .await
     }
 
+    pub async fn listen<T>(&mut self, local_endpoint: T) -> Result<()>
+    where
+        T: Into<IpEndpoint>,
+    {
+        self.with(|s| s.listen(local_endpoint))?;
+
+        futures::future::poll_fn(|cx| {
+            self.with(|s| match s.state() {
+                TcpState::Closed | TcpState::TimeWait => Poll::Ready(Err(Error::Unaddressable)),
+                TcpState::Listen => Poll::Ready(Ok(())),
+                TcpState::SynSent | TcpState::SynReceived => {
+                    s.register_send_waker(cx.waker());
+                    Poll::Pending
+                }
+                _ => Poll::Ready(Ok(())),
+            })
+        })
+        .await
+    }
+
     pub fn set_timeout(&mut self, duration: Option<Duration>) {
         self.with(|s| s.set_timeout(duration))
     }