You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.5 KiB
60 lines
1.5 KiB
// Copyright (c) 2013-2015 The btcsuite developers |
|
// Use of this source code is governed by an ISC |
|
// license that can be found in the LICENSE file. |
|
|
|
package wire_test |
|
|
|
import ( |
|
"io" |
|
|
|
"github.com/btcsuite/btcd/wire" |
|
) |
|
|
|
// fakeMessage implements the wire.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 wire.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 |
|
// wire.Message interface. |
|
func (msg *fakeMessage) BtcEncode(w io.Writer, pver uint32) error { |
|
if msg.forceEncodeErr { |
|
err := &wire.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 |
|
// wire.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 wire.Message interface. |
|
func (msg *fakeMessage) MaxPayloadLength(pver uint32) uint32 { |
|
lenp := uint32(len(msg.payload)) |
|
if msg.forceLenErr { |
|
return lenp - 1 |
|
} |
|
|
|
return lenp |
|
}
|
|
|