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))),