Fix pipe leaking when splite_n returns 0.

yuguorui · Apr 30, 2022 · c482865 · c482865
1 parent 5f19739
commit c482865
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/utils.rs b/src/utils.rs
@@ -178,7 +178,7 @@ async fn _copy_bidirectional(inbound: &mut TcpStream, outbound: &mut TcpStream)
         Ok::<(), std::io::Error>(())
     };
 
-    let (_, _) = tokio::join!(client_to_server, server_to_client);
+    let _ = tokio::try_join!(client_to_server, server_to_client);
 
     Ok(())
 }
@@ -244,7 +244,7 @@ where
                 if ret > 0 {
                     n = ret;
                 } else {
-                    break;
+                    return Err(io::Error::new(io::ErrorKind::UnexpectedEof, "zero_copy"));
                 }
             }
             Err(err) if err.kind() == io::ErrorKind::WouldBlock => {
@@ -278,7 +278,5 @@ where
             }
         }
     }
-
-    Ok(())
 }