msg/json: fix panic error when length is invalid

fatedier · May 21, 2018 · 674e937 · 674e937
1 parent 354693c
commit 674e937
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/msg/json/process.go b/msg/json/process.go
@@ -23,6 +23,7 @@ import (
 var (
 	ErrMsgType      = errors.New("message type error")
 	ErrMaxMsgLength = errors.New("message length exceed the limit")
+	ErrMsgLength    = errors.New("message length error")
 	ErrMsgFormat    = errors.New("message format error")
 )
 
@@ -46,6 +47,9 @@ func (msgCtl *MsgCtl) readMsg(c io.Reader) (typeByte byte, buffer []byte, err er
 	if length > msgCtl.maxMsgLength {
 		err = ErrMaxMsgLength
 		return
+	} else if length < 0 {
+		err = ErrMsgLength
+		return
 	}
 
 	buffer = make([]byte, length)