fix named pipe

tokio-rs · Nov 18, 2022 · dbdad85 · dbdad85
1 parent 4fe51c9
commit dbdad85
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/tokio/src/net/windows/named_pipe.rs b/tokio/src/net/windows/named_pipe.rs
@@ -192,10 +192,21 @@ impl NamedPipeServer {
     /// # Ok(()) }
     /// ```
     pub async fn connect(&self) -> io::Result<()> {
-        self.io
-            .registration()
-            .async_io(Interest::WRITABLE, || self.io.connect())
-            .await
+        loop {
+            match self.io.connect() {
+                Ok(()) => break,
+                Err(e) if e.kind() == io::ErrorKind::WouldBlock => {
+                    return self
+                        .io
+                        .registration()
+                        .async_io(Interest::WRITABLE, || self.io.connect())
+                        .await
+                }
+                Err(e) => return Err(e),
+            }
+        }
+
+        Ok(())
     }
 
     /// Disconnects the server end of a named pipe instance from a client