forked from TrueCloudLab/neoneo-go
61 lines
1.5 KiB
Go
61 lines
1.5 KiB
Go
package payload
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/wire/command"
|
|
"github.com/CityOfZion/neo-go/pkg/wire/util"
|
|
)
|
|
|
|
//GetHeadersMessage represents a GetHeaders message on the neo-network
|
|
type GetHeadersMessage struct {
|
|
cmd command.Type
|
|
hashStart []util.Uint256
|
|
hashStop util.Uint256
|
|
}
|
|
|
|
// NewGetHeadersMessage returns a NewGetHeaders object
|
|
// Start contains the list of all headers you want to fetch
|
|
// End contains the list of the highest header hash you would like to fetch
|
|
func NewGetHeadersMessage(start []util.Uint256, stop util.Uint256) (*GetHeadersMessage, error) {
|
|
getHeaders := &GetHeadersMessage{command.GetHeaders, start, stop}
|
|
|
|
return getHeaders, nil
|
|
|
|
}
|
|
|
|
func newAbstractGetHeaders(start []util.Uint256, stop util.Uint256, cmd command.Type) (*GetHeadersMessage, error) {
|
|
getHeaders, err := NewGetHeadersMessage(start, stop)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
getHeaders.cmd = cmd
|
|
return getHeaders, nil
|
|
}
|
|
|
|
// DecodePayload Implements Messager interface
|
|
func (v *GetHeadersMessage) DecodePayload(r io.Reader) error {
|
|
|
|
br := util.BinReader{R: r}
|
|
lenStart := br.VarUint()
|
|
v.hashStart = make([]util.Uint256, lenStart)
|
|
br.Read(&v.hashStart)
|
|
br.Read(&v.hashStop)
|
|
|
|
return br.Err
|
|
}
|
|
|
|
// EncodePayload Implements messager interface
|
|
func (v *GetHeadersMessage) EncodePayload(w io.Writer) error {
|
|
bw := &util.BinWriter{W: w}
|
|
bw.VarUint(uint64(len(v.hashStart)))
|
|
bw.Write(v.hashStart)
|
|
bw.Write(v.hashStop)
|
|
return bw.Err
|
|
}
|
|
|
|
// Command Implements messager interface
|
|
func (v *GetHeadersMessage) Command() command.Type {
|
|
return v.cmd
|
|
}
|