From e54045ef5a71a5520960def70dcdb0466d02281b Mon Sep 17 00:00:00 2001
From: Zack Scholl <zack.scholl@gmail.com>
Date: Tue, 30 Apr 2019 07:07:07 -0700
Subject: [PATCH] read until 4 bytes

---
 src/comm/comm.go | 25 ++++++++++++++-----------
 1 file changed, 14 insertions(+), 11 deletions(-)

diff --git a/src/comm/comm.go b/src/comm/comm.go
index be44855..7ed4b3a 100644
--- a/src/comm/comm.go
+++ b/src/comm/comm.go
@@ -65,18 +65,21 @@ func (c *Comm) Write(b []byte) (int, error) {
 }
 
 func (c *Comm) Read() (buf []byte, numBytes int, bs []byte, err error) {
-	// read until we get 5 bytes
-	header := make([]byte, 4)
-	n, err := c.connection.Read(header)
-	if err != nil {
-		return
+	// read until we get 4 bytes for the header
+	var header []byte
+	numBytes = 4
+	for {
+		tmp := make([]byte,numBytes-len(header))
+		n, errRead := c.connection.Read(tmp)
+		if errRead != nil {
+			err = errRead
+			return
+		}
+		header = append(header, tmp[:n]...)
+		if numBytes == len(header) {
+			break
+		}
 	}
-	if n < 4 {
-		err = fmt.Errorf("not enough bytes: %d", n)
-		return
-	}
-	// make it so it won't change
-	header = append([]byte(nil), header...)
 
 	var numBytesUint32 uint32
 	rbuf := bytes.NewReader(header)