Merge pull request from GHSA-x462-89pf-6r5h

This commit is contained in:
Jakob Borg 2021-04-06 08:00:00 +02:00
parent 0d7a77ba85
commit fb4fdaf4c0
1 changed files with 4 additions and 0 deletions

View File

@ -4,6 +4,7 @@ package protocol
import (
"errors"
"fmt"
"io"
)
@ -86,6 +87,9 @@ func ReadMessage(r io.Reader) (interface{}, error) {
if header.magic != magic {
return nil, errors.New("magic mismatch")
}
if header.messageLength < 0 || header.messageLength > 1024 {
return nil, fmt.Errorf("bad length (%d)", header.messageLength)
}
buf = make([]byte, int(header.messageLength))
if _, err := io.ReadFull(r, buf); err != nil {