mirror of
https://github.com/btcsuite/btcd.git
synced 2024-11-19 09:50:08 +01:00
86 lines
2.5 KiB
Go
86 lines
2.5 KiB
Go
// Copyright (c) 2017 The btcsuite developers
|
|
// Use of this source code is governed by an ISC
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package wire
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/btcsuite/btcd/chaincfg/chainhash"
|
|
)
|
|
|
|
// MsgGetCFHeaders is a message similar to MsgGetHeaders, but for committed
|
|
// filter headers. It allows to set the FilterType field to get headers in the
|
|
// chain of basic (0x00) or extended (0x01) headers.
|
|
type MsgGetCFHeaders struct {
|
|
FilterType FilterType
|
|
StartHeight uint32
|
|
StopHash chainhash.Hash
|
|
}
|
|
|
|
// BtcDecode decodes r using the bitcoin protocol encoding into the receiver.
|
|
// This is part of the Message interface implementation.
|
|
func (msg *MsgGetCFHeaders) BtcDecode(r io.Reader, pver uint32, _ MessageEncoding) error {
|
|
buf := binarySerializer.Borrow()
|
|
defer binarySerializer.Return(buf)
|
|
|
|
if _, err := io.ReadFull(r, buf[:1]); err != nil {
|
|
return err
|
|
}
|
|
msg.FilterType = FilterType(buf[0])
|
|
|
|
if _, err := io.ReadFull(r, buf[:4]); err != nil {
|
|
return err
|
|
}
|
|
msg.StartHeight = littleEndian.Uint32(buf[:4])
|
|
|
|
_, err := io.ReadFull(r, msg.StopHash[:])
|
|
return err
|
|
}
|
|
|
|
// BtcEncode encodes the receiver to w using the bitcoin protocol encoding.
|
|
// This is part of the Message interface implementation.
|
|
func (msg *MsgGetCFHeaders) BtcEncode(w io.Writer, pver uint32, _ MessageEncoding) error {
|
|
buf := binarySerializer.Borrow()
|
|
defer binarySerializer.Return(buf)
|
|
|
|
buf[0] = byte(msg.FilterType)
|
|
if _, err := w.Write(buf[:1]); err != nil {
|
|
return err
|
|
}
|
|
|
|
littleEndian.PutUint32(buf[:4], msg.StartHeight)
|
|
if _, err := w.Write(buf[:4]); err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err := w.Write(msg.StopHash[:])
|
|
return err
|
|
}
|
|
|
|
// Command returns the protocol command string for the message. This is part
|
|
// of the Message interface implementation.
|
|
func (msg *MsgGetCFHeaders) Command() string {
|
|
return CmdGetCFHeaders
|
|
}
|
|
|
|
// MaxPayloadLength returns the maximum length the payload can be for the
|
|
// receiver. This is part of the Message interface implementation.
|
|
func (msg *MsgGetCFHeaders) MaxPayloadLength(pver uint32) uint32 {
|
|
// Filter type + uint32 + block hash
|
|
return 1 + 4 + chainhash.HashSize
|
|
}
|
|
|
|
// NewMsgGetCFHeaders returns a new bitcoin getcfheader message that conforms to
|
|
// the Message interface using the passed parameters and defaults for the
|
|
// remaining fields.
|
|
func NewMsgGetCFHeaders(filterType FilterType, startHeight uint32,
|
|
stopHash *chainhash.Hash) *MsgGetCFHeaders {
|
|
return &MsgGetCFHeaders{
|
|
FilterType: filterType,
|
|
StartHeight: startHeight,
|
|
StopHash: *stopHash,
|
|
}
|
|
}
|