Flushing a closed stream is okay

libp2p · May 19, 2023 · 6c03282 · 6c03282
1 parent 76a7e6f
commit 6c03282
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/yamux/src/connection/stream.rs b/yamux/src/connection/stream.rs
@@ -363,6 +363,10 @@ impl AsyncWrite for Stream {
     }
 
     fn poll_flush(mut self: Pin<&mut Self>, cx: &mut Context) -> Poll<io::Result<()>> {
+        if self.is_closed() {
+            return Poll::Ready(Ok(()));
+        }
+
         let num_frames = {
             let shared = self.shared();