TrueCloudLab/neo-go
8
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-11-22 19:29:39 +00:00
Code Issues Projects Releases Packages Wiki Activity

network: allow NullPayload only for specific commands

Browse source 
CMDFilterClear, CMDGetAddr, CMDMempool all have empty payloads.
This commit is contained in:
Evgenii Stratonikov 2020-06-19 11:18:43 +03:00
parent 025bf9c235
commit 9f42108ef2
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
pkg/network/message.go
View file

@ -106,7 +106,12 @@ func (m *Message) Decode(br *io.BinReader) error {
// check the length first in order not to allocate memory
// for an empty compressed payload
if l == 0 {
m.Payload = payload.NewNullPayload()
switch m.Command {
case CMDFilterClear, CMDGetAddr, CMDMempool:
m.Payload = payload.NewNullPayload()
default:
return errors.New("unexpected empty payload")
}
return nil
}
if l > PayloadMaxSize {