80 lines
1.9 KiB
Go
80 lines
1.9 KiB
Go
|
// Copyright (c) 2013 Conformal Systems LLC.
|
||
|
|
// Use of this source code is governed by an ISC
|
||
|
|
// license that can be found in the LICENSE file.
|
||
|
|
|
||
|
|
package btcwire
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
// MaxInvPerMsg is the maximum number of inventory vectors that can be in a
|
||
|
|
// single bitcoin inv message.
|
||
|
|
MaxInvPerMsg = 50000
|
||
|
|
|
||
|
|
// Maximum payload size for an inventory vector.
|
||
|
|
maxInvVectPayload = 4 + HashSize
|
||
|
|
)
|
||
|
|
|
||
|
|
// InvType represents the allowed types of inventory vectors. See InvVect.
|
||
|
|
type InvType uint32
|
||
|
|
|
||
|
|
const (
|
||
|
|
InvVect_Error InvType = 0
|
||
|
|
InvVect_Tx InvType = 1
|
||
|
|
InvVect_Block InvType = 2
|
||
|
|
)
|
||
|
|
|
||
|
|
// Map of service flags back to their constant names for pretty printing.
|
||
|
|
var ivStrings = map[InvType]string{
|
||
|
|
InvVect_Error: "ERROR",
|
||
|
|
InvVect_Tx: "MSG_TX",
|
||
|
|
InvVect_Block: "MSG_BLOCK",
|
||
|
|
}
|
||
|
|
|
||
|
|
// String returns the InvType in human-readable form.
|
||
|
|
func (invtype InvType) String() string {
|
||
|
|
if s, ok := ivStrings[invtype]; ok {
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
return fmt.Sprintf("Unknown InvType (%d)", uint32(invtype))
|
||
|
|
}
|
||
|
|
|
||
|
|
// InvVect defines a bitcoin inventory vector which is used to describe data,
|
||
|
|
// as specified by the Type field, that a peer wants, has, or does not have to
|
||
|
|
// another peer.
|
||
|
|
type InvVect struct {
|
||
|
|
Type InvType // Type of data
|
||
|
|
Hash ShaHash // Hash of the data
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewInvVect returns a new InvVect using the provided type and hash.
|
||
|
|
func NewInvVect(typ InvType, hash *ShaHash) *InvVect {
|
||
|
|
return &InvVect{
|
||
|
|
Type: typ,
|
||
|
|
Hash: *hash,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// readInvVect reads an encoded InvVect from r depending on the protocol
|
||
|
|
// version.
|
||
|
|
func readInvVect(r io.Reader, pver uint32, iv *InvVect) error {
|
||
|
|
err := readElements(r, &iv.Type, &iv.Hash)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// writeInvVect serializes an InvVect to w depending on the protocol version.
|
||
|
|
func writeInvVect(w io.Writer, pver uint32, iv *InvVect) error {
|
||
|
|
err := writeElements(w, iv.Type, iv.Hash)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|