add handling of corner case (#1117)

getgauge · Jul 20, 2018 · 69b8239 · 69b8239
1 parent e62f840
commit 69b8239
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/conn/network.go b/conn/network.go
@@ -95,7 +95,7 @@ func readResponse(conn net.Conn) ([]byte, error) {
 
 		buffer.Write(data[0:n])
 		messageLength, bytesRead := proto.DecodeVarint(buffer.Bytes())
-		if messageLength > 0 && messageLength < uint64(buffer.Len()) {
+		if (messageLength > 0 && messageLength < uint64(buffer.Len())) && ((messageLength + uint64(bytesRead)) <= uint64(buffer.Len())) {
 			return buffer.Bytes()[bytesRead : messageLength+uint64(bytesRead)], nil
 		}
 	}