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

Merge pull request #1072 from nspcc-dev/fix/payload

Browse source 
Allow NullPayload only for CMDGetAddr
This commit is contained in:
Roman Khimov 2020-06-19 15:58:40 +03:00 committed by GitHub
commit 75cc109837
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 5 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 // check the length first in order not to allocate memory
// for an empty compressed payload // for an empty compressed payload
if l == 0 { 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 return nil
} }
if l > PayloadMaxSize { if l > PayloadMaxSize {

8
pkg/network/payload/payload.go
View file

@ -12,12 +12,12 @@ type NullPayload struct {
} }
// NewNullPayload returns zero-sized stub payload. // NewNullPayload returns zero-sized stub payload.
func NewNullPayload() *NullPayload { func NewNullPayload() NullPayload {
return &NullPayload{} return NullPayload{}
} }
// DecodeBinary implements Serializable interface. // DecodeBinary implements Serializable interface.
func (p *NullPayload) DecodeBinary(r *io.BinReader) {} func (p NullPayload) DecodeBinary(r *io.BinReader) {}
// EncodeBinary implements Serializable interface. // EncodeBinary implements Serializable interface.
func (p *NullPayload) EncodeBinary(w *io.BinWriter) {} func (p NullPayload) EncodeBinary(w *io.BinWriter) {}