g3-openssl: return unexpected eof error when BIO read 0

bytedance · Dec 12, 2024 · d082dfe · d082dfe
1 parent c9a5c12
commit d082dfe
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/lib/g3-openssl/src/ssl/wrapper.rs b/lib/g3-openssl/src/ssl/wrapper.rs
@@ -65,7 +65,17 @@ impl<S: AsyncRead + Unpin> Read for SslIoWrapper<S> {
         self.with_context(|stream, cx| {
             let mut buf = ReadBuf::new(buf);
             match stream.poll_read(cx, &mut buf) {
-                Poll::Ready(Ok(_)) => Ok(buf.filled().len()),
+                Poll::Ready(Ok(_)) => {
+                    let nr = buf.filled().len();
+                    if nr == 0 {
+                        Err(io::Error::new(
+                            io::ErrorKind::UnexpectedEof,
+                            "connection closed when do SSL BIO read",
+                        ))
+                    } else {
+                        Ok(nr)
+                    }
+                }
                 Poll::Ready(Err(e)) => Err(e),
                 Poll::Pending => Err(io::Error::from(io::ErrorKind::WouldBlock)),
             }