Don't keep reading on EOF and get killed by OOM

vvilhonen · May 27, 2020 · 417517e · 417517e
1 parent e08d8cb
commit 417517e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/main.rs b/src/main.rs
@@ -289,6 +289,9 @@ impl Shell {
         while buf.len() < 1024 {
             let mut chunk = [0u8; 256];
             let n = timeout(Duration::from_secs(500), self.socket.read(&mut chunk)).await??;
+            if n == 0 {
+                break;
+            }
             buf.extend(&chunk[..n]);
             let empty = buf
                 .iter()