fix resp package BulkString funtion

distributedio · Jan 20, 2019 · 367a45f · 367a45f
1 parent cc4c698
commit 367a45f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/encoding/resp/resp.go b/encoding/resp/resp.go
@@ -227,7 +227,7 @@ func (r *Decoder) BulkString() (string, error) {
 	}
 
 	remain, err := strconv.Atoi(string(hdr[1 : l-2]))
-	if err != nil {
+	if err != nil || remain < 0 {
 		return "", ErrInvalidProtocol
 	}
 
@@ -256,7 +256,7 @@ func (r *Decoder) Array() (int, error) {
 		return -1, ErrInvalidProtocol
 	}
 	remain, err := strconv.Atoi(string(hdr[1 : l-2]))
-	if err != nil {
+	if err != nil || remain < 0 {
 		return -1, ErrInvalidProtocol
 	}
 	return remain, nil