LBRYCommunity/lbcd
1
0
Fork 0
Code Issues 17 Pull requests 4 Packages Projects Releases 28 Wiki Activity

Convert MsgHeaders errors to MessageError type.

Browse source
This commit is contained in:
Dave Collins 2013-05-10 23:10:59 -05:00
parent 88e1d7634f
commit 40a9a90e4b
1 changed files with 15 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
msgheaders.go
View file

@ -25,9 +25,9 @@ type MsgHeaders struct {
// AddBlockHeader adds a new block header to the message. // AddBlockHeader adds a new block header to the message.
func (msg *MsgHeaders) AddBlockHeader(bh *BlockHeader) error { func (msg *MsgHeaders) AddBlockHeader(bh *BlockHeader) error {
if len(msg.Headers)+1 > MaxBlockHeadersPerMsg { if len(msg.Headers)+1 > MaxBlockHeadersPerMsg {
str := "MsgHeaders.AddBlockHeader: too many block headers " + str := fmt.Sprintf("too many block headers in message [max %v]",
"for message [max %v]" MaxBlockHeadersPerMsg)
return fmt.Errorf(str, MaxBlockHeadersPerMsg) return messageError("MsgHeaders.AddBlockHeader", str)
} }
msg.Headers = append(msg.Headers, bh) msg.Headers = append(msg.Headers, bh)
@ -44,8 +44,9 @@ func (msg *MsgHeaders) BtcDecode(r io.Reader, pver uint32) error {
// Limit to max block headers per message. // Limit to max block headers per message.
if count > MaxBlockHeadersPerMsg { if count > MaxBlockHeadersPerMsg {
str := "MsgHeaders.BtcDecode: too many block headers in message [%v]" str := fmt.Sprintf("too many block headers for message "+
return fmt.Errorf(str, count) "[count %v, max %v]", count, MaxBlockHeadersPerMsg)
return messageError("MsgHeaders.BtcDecode", str)
} }
for i := uint64(0); i < count; i++ { for i := uint64(0); i < count; i++ {
@ -57,9 +58,9 @@ func (msg *MsgHeaders) BtcDecode(r io.Reader, pver uint32) error {
// Ensure the transaction count is zero for headers. // Ensure the transaction count is zero for headers.
if bh.TxnCount > 0 { if bh.TxnCount > 0 {
str := "MsgHeaders.BtcDecode: block headers may not " + str := fmt.Sprintf("block headers may not contain "+
"contain transactions [%v]" "transactions [count %v]", bh.TxnCount)
return fmt.Errorf(str, count) return messageError("MsgHeaders.BtcDecode", str)
} }
msg.AddBlockHeader(&bh) msg.AddBlockHeader(&bh)
} }
@ -73,8 +74,9 @@ func (msg *MsgHeaders) BtcEncode(w io.Writer, pver uint32) error {
// Limit to max block headers per message. // Limit to max block headers per message.
count := len(msg.Headers) count := len(msg.Headers)
if count > MaxBlockHeadersPerMsg { if count > MaxBlockHeadersPerMsg {
str := "MsgHeaders.BtcEncode: too many block headers in message [%v]" str := fmt.Sprintf("too many block headers for message "+
return fmt.Errorf(str, count) "[count %v, max %v]", count, MaxBlockHeadersPerMsg)
return messageError("MsgHeaders.BtcEncode", str)
} }
err := writeVarInt(w, pver, uint64(count)) err := writeVarInt(w, pver, uint64(count))
@ -85,9 +87,9 @@ func (msg *MsgHeaders) BtcEncode(w io.Writer, pver uint32) error {
for _, bh := range msg.Headers { for _, bh := range msg.Headers {
// Ensure block headers do not contain a transaction count. // Ensure block headers do not contain a transaction count.
if bh.TxnCount > 0 { if bh.TxnCount > 0 {
str := "MsgHeaders.BtcEncode: block headers " + str := fmt.Sprintf("block headers may not contain "+
"may not contain transactions [%v]" "transactions [count %v]", bh.TxnCount)
return fmt.Errorf(str, count) return messageError("MsgHeaders.BtcEncode", str)
} }
err := writeBlockHeader(w, pver, bh) err := writeBlockHeader(w, pver, bh)