14a1da417f
This commit changes MsgBlock to enforce a 1MB max payload per the spec. Previously it was only limited to the max overall message size. While here, also enforce max payloads per message type (instead of only the max overall message payload) when writing messages.
59 lines
1.6 KiB
Go
59 lines
1.6 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_test
|
|
|
|
import (
|
|
"github.com/conformal/btcwire"
|
|
"io"
|
|
)
|
|
|
|
// fakeMessage implements the btcwire.Message interface and is used to force
|
|
// encode errors in messages.
|
|
type fakeMessage struct {
|
|
command string
|
|
payload []byte
|
|
forceEncodeErr bool
|
|
forceLenErr bool
|
|
}
|
|
|
|
// BtcDecode doesn't do anything. It just satisfies the btcwire.Message
|
|
// interface.
|
|
func (msg *fakeMessage) BtcDecode(r io.Reader, pver uint32) error {
|
|
return nil
|
|
}
|
|
|
|
// BtcEncode writes the payload field of the fake message or forces an error
|
|
// if the forceEncodeErr flag of the fake message is set. It also satisfies the
|
|
// btcwire.Message interface.
|
|
func (msg *fakeMessage) BtcEncode(w io.Writer, pver uint32) error {
|
|
if msg.forceEncodeErr {
|
|
err := &btcwire.MessageError{
|
|
Func: "fakeMessage.BtcEncode",
|
|
Description: "intentional error",
|
|
}
|
|
return err
|
|
}
|
|
|
|
_, err := w.Write(msg.payload)
|
|
return err
|
|
}
|
|
|
|
// Command returns the command field of the fake message and satisfies the
|
|
// btcwire.Message interface.
|
|
func (msg *fakeMessage) Command() string {
|
|
return msg.command
|
|
}
|
|
|
|
// MaxPayloadLength returns the length of the payload field of fake message
|
|
// or a smaller value if the forceLenErr flag of the fake message is set. It
|
|
// satisfies the btcwire.Message interface.
|
|
func (msg *fakeMessage) MaxPayloadLength(pver uint32) uint32 {
|
|
lenp := uint32(len(msg.payload))
|
|
if msg.forceLenErr {
|
|
return lenp - 1
|
|
}
|
|
|
|
return lenp
|
|
}
|