From 417517e54d211fcf320966d3ccfbc163ffddae64 Mon Sep 17 00:00:00 2001
From: Vesa Vilhonen <vesa@vilhonen.com>
Date: Wed, 27 May 2020 07:09:43 +0300
Subject: [PATCH] Don't keep reading on EOF and get killed by OOM

---
 src/main.rs | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/main.rs b/src/main.rs
index 1c96d84..20ec0c3 100644
--- 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()