5bf00db2c9
The logic here is that we'll have all binary encoding/decoding done via our io package, which simplifies error handling. This functionality doesn't belong to util, so it's moved. This also expands BufBinWriter with Reset() method to fit the needs of core package.
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package payload
|
|
|
|
import (
|
|
"github.com/CityOfZion/neo-go/pkg/core"
|
|
"github.com/CityOfZion/neo-go/pkg/io"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// Headers payload
|
|
type Headers struct {
|
|
Hdrs []*core.Header
|
|
}
|
|
|
|
// Users can at most request 2k header
|
|
const (
|
|
maxHeadersAllowed = 2000
|
|
)
|
|
|
|
// DecodeBinary implements the Payload interface.
|
|
func (p *Headers) DecodeBinary(br *io.BinReader) error {
|
|
lenHeaders := br.ReadVarUint()
|
|
if br.Err != nil {
|
|
return br.Err
|
|
}
|
|
// C# node does it silently
|
|
if lenHeaders > maxHeadersAllowed {
|
|
log.Warnf("received %d headers, capping to %d", lenHeaders, maxHeadersAllowed)
|
|
lenHeaders = maxHeadersAllowed
|
|
}
|
|
|
|
p.Hdrs = make([]*core.Header, lenHeaders)
|
|
|
|
for i := 0; i < int(lenHeaders); i++ {
|
|
header := &core.Header{}
|
|
if err := header.DecodeBinary(br); err != nil {
|
|
return err
|
|
}
|
|
p.Hdrs[i] = header
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// EncodeBinary implements the Payload interface.
|
|
func (p *Headers) EncodeBinary(bw *io.BinWriter) error {
|
|
bw.WriteVarUint(uint64(len(p.Hdrs)))
|
|
if bw.Err != nil {
|
|
return bw.Err
|
|
}
|
|
|
|
for _, header := range p.Hdrs {
|
|
if err := header.EncodeBinary(bw); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|