Ignore NotConnected error in poll_shutdown()

rustls · Feb 29, 2024 · cf32e6c · cf32e6c
1 parent ff32e1e
commit cf32e6c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/common/mod.rs b/src/common/mod.rs
@@ -335,7 +335,13 @@ where
         while self.session.wants_write() {
             ready!(self.write_io(cx))?;
         }
-        Pin::new(&mut self.io).poll_shutdown(cx)
+
+        Poll::Ready(match ready!(Pin::new(&mut self.io).poll_shutdown(cx)) {
+            Ok(()) => Ok(()),
+            // When trying to shutdown, not being connected seems fine
+            Err(err) if err.kind() == io::ErrorKind::NotConnected => Ok(()),
+            Err(err) => Err(err),
+        })
     }
 }