Convert MsgNotFound errors to MessageError type.

This commit is contained in:
Dave Collins 2013-05-10 11:32:20 -05:00
parent e41a3bcaf3
commit 9b798b6306

View file

@ -23,8 +23,9 @@ type MsgNotFound struct {
// AddInvVect adds an inventory vector to the message. // AddInvVect adds an inventory vector to the message.
func (msg *MsgNotFound) AddInvVect(iv *InvVect) error { func (msg *MsgNotFound) AddInvVect(iv *InvVect) error {
if len(msg.InvList)+1 > MaxInvPerMsg { if len(msg.InvList)+1 > MaxInvPerMsg {
str := "MsgNotFound.AddAddress: too many invvect in message [max %v]" str := fmt.Sprintf("too many invvect in message [max %v]",
return fmt.Errorf(str, MaxInvPerMsg) MaxInvPerMsg)
return messageError("MsgNotFound.AddInvVect", str)
} }
msg.InvList = append(msg.InvList, iv) msg.InvList = append(msg.InvList, iv)
@ -41,8 +42,8 @@ func (msg *MsgNotFound) BtcDecode(r io.Reader, pver uint32) error {
// Limit to max inventory vectors per message. // Limit to max inventory vectors per message.
if count > MaxInvPerMsg { if count > MaxInvPerMsg {
str := "MsgNotFound.BtcDecode: too many invvect in message [%v]" str := fmt.Sprintf("too many invvect in message [%v]", count)
return fmt.Errorf(str, count) return messageError("MsgNotFound.BtcDecode", str)
} }
for i := uint64(0); i < count; i++ { for i := uint64(0); i < count; i++ {
@ -63,8 +64,8 @@ func (msg *MsgNotFound) BtcEncode(w io.Writer, pver uint32) error {
// Limit to max inventory vectors per message. // Limit to max inventory vectors per message.
count := len(msg.InvList) count := len(msg.InvList)
if count > MaxInvPerMsg { if count > MaxInvPerMsg {
str := "MsgNotFound.BtcEncode: too many invvect in message [%v]" str := fmt.Sprintf("too many invvect in message [%v]", count)
return fmt.Errorf(str, count) return messageError("MsgNotFound.BtcEncode", str)
} }
err := writeVarInt(w, pver, uint64(count)) err := writeVarInt(w, pver, uint64(count))