diff --git a/src/main.rs b/src/main.rs index 12714ca..e7e68e0 100644 --- a/src/main.rs +++ b/src/main.rs @@ -68,7 +68,7 @@ async fn main() { tokio::task::spawn(async move { let mut buf = [0u8; 256]; while let Ok(n) = con_read.read(&mut buf).await { - if prox_write.write_all(&buf[0..n]).await.is_err() { + if n > 0 && prox_write.write_all(&buf[0..n]).await.is_err() { break; } } @@ -76,7 +76,7 @@ async fn main() { tokio::task::spawn(async move { let mut buf = [0u8; 256]; while let Ok(n) = prox_read.read(&mut buf).await { - if con_write.write_all(&buf[0..n]).await.is_err() { + if n > 0 && con_write.write_all(&buf[0..n]).await.is_err() { break; } }