fix: check for ErrorKind::WouldBlock in MidHandshake::SendAlert poll (#…

…47)
rustls · Mar 6, 2024 · 4a4a448 · 4a4a448
1 parent 9fab4e2
commit 4a4a448
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/common/handshake.rs b/src/common/handshake.rs
@@ -52,9 +52,13 @@ where
                 mut alert,
                 error,
             } => {
-                let mut writer = SyncWriteAdapter { io: &mut io, cx };
-                let _ = alert.write(&mut writer); // best effort
-                return Poll::Ready(Err((error, io)));
+                return match alert.write(&mut SyncWriteAdapter { io: &mut io, cx }) {
+                    Err(ref e) if e.kind() == io::ErrorKind::WouldBlock => {
+                        *this = MidHandshake::SendAlert { io, error, alert };
+                        Poll::Pending
+                    }
+                    _ => Poll::Ready(Err((error, io))),
+                };
             }
             // Starting the handshake returned an error; fail the future immediately.
             MidHandshake::Error { io, error } => return Poll::Ready(Err((error, io))),