From 02417c313432c738f87662613c16e70d28e3577e Mon Sep 17 00:00:00 2001
From: Jacob Rothstein <hi@jbr.me>
Date: Tue, 19 Mar 2024 12:42:59 -0700
Subject: [PATCH] Take poll_close changes from
 https://github.com/rustls/tokio-rustls/pull/42

---
 src/common/mod.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/common/mod.rs b/src/common/mod.rs
index 6d82c38..5169f9d 100644
--- a/src/common/mod.rs
+++ b/src/common/mod.rs
@@ -297,7 +297,13 @@ where
         while self.session.wants_write() {
             ready!(self.write_io(cx))?;
         }
-        Pin::new(&mut self.io).poll_close(cx)
+
+        Poll::Ready(match ready!(Pin::new(&mut self.io).poll_close(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),
+        })
     }
 
     fn poll_write_vectored(