Fix EOF issue

cloudflare · Aug 28, 2018 · 9d057d9 · 9d057d9
1 parent 11b089f
commit 9d057d9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/server.go b/lib/server.go
@@ -632,11 +632,12 @@ func (c *Client) Start() {
 	buf := make([]byte, 8000)
 	for c.connected {
 		length, err := c.tcpconn.Read(buf)
-		if err != nil {
+		if err != nil || length == 0 {
 			log.Debugf("Error %v", err)
 			c.Disconnect()
 			return
 		}
+
 		pkt := buf[0:length]
 		dec, _ := DecodeBytes(pkt)
 		if !c.disableVersionCheck {