2015-12-31 09:28:14 +03:00
|
|
|
package lnwire
|
|
|
|
|
|
|
|
import (
|
2017-04-20 02:16:55 +03:00
|
|
|
"bytes"
|
2015-12-31 09:28:14 +03:00
|
|
|
"encoding/hex"
|
2017-04-20 02:16:55 +03:00
|
|
|
"math"
|
|
|
|
"math/big"
|
|
|
|
"math/rand"
|
2016-12-22 23:17:24 +03:00
|
|
|
"net"
|
2017-04-20 02:16:55 +03:00
|
|
|
"reflect"
|
|
|
|
"testing"
|
|
|
|
"testing/quick"
|
2016-12-22 23:17:24 +03:00
|
|
|
|
2017-08-22 08:46:56 +03:00
|
|
|
"github.com/davecgh/go-spew/spew"
|
2016-05-23 23:47:23 +03:00
|
|
|
"github.com/roasbeef/btcd/btcec"
|
2017-01-06 00:56:27 +03:00
|
|
|
"github.com/roasbeef/btcd/chaincfg/chainhash"
|
2016-05-23 23:47:23 +03:00
|
|
|
"github.com/roasbeef/btcd/wire"
|
2017-04-20 02:16:55 +03:00
|
|
|
"github.com/roasbeef/btcutil"
|
2015-12-31 09:28:14 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
2016-06-30 21:53:11 +03:00
|
|
|
revHash = [32]byte{
|
|
|
|
0xb7, 0x94, 0x38, 0x5f, 0x2d, 0x1e, 0xf7, 0xab,
|
|
|
|
0x4d, 0x92, 0x73, 0xd1, 0x90, 0x63, 0x81, 0xb4,
|
|
|
|
0x4f, 0x2f, 0x6f, 0x25, 0x88, 0xa3, 0xef, 0xb9,
|
|
|
|
0x6a, 0x49, 0x18, 0x83, 0x31, 0x98, 0x47, 0x53,
|
|
|
|
}
|
|
|
|
|
2017-02-16 15:34:44 +03:00
|
|
|
shaHash1Bytes, _ = hex.DecodeString("e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855")
|
|
|
|
shaHash1, _ = chainhash.NewHash(shaHash1Bytes)
|
|
|
|
outpoint1 = wire.NewOutPoint(shaHash1, 0)
|
2017-04-20 02:16:55 +03:00
|
|
|
testSig = &btcec.Signature{
|
|
|
|
R: new(big.Int),
|
|
|
|
S: new(big.Int),
|
|
|
|
}
|
|
|
|
_, _ = testSig.R.SetString("63724406601629180062774974542967536251589935445068131219452686511677818569431", 10)
|
|
|
|
_, _ = testSig.S.SetString("18801056069249825825291287104931333862866033135609736119018462340006816851118", 10)
|
|
|
|
|
|
|
|
// TODO(roasbeef): randomly generate from three types of addrs
|
|
|
|
a1 = &net.TCPAddr{IP: (net.IP)([]byte{0x7f, 0x0, 0x0, 0x1}), Port: 8333}
|
|
|
|
a2, _ = net.ResolveTCPAddr("tcp", "[2001:db8:85a3:0:0:8a2e:370:7334]:80")
|
|
|
|
testAddrs = []net.Addr{a1, a2}
|
|
|
|
)
|
|
|
|
|
|
|
|
func randPubKey() (*btcec.PublicKey, error) {
|
|
|
|
priv, err := btcec.NewPrivateKey(btcec.S256())
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return priv.PubKey(), nil
|
|
|
|
}
|
2016-01-05 19:19:22 +03:00
|
|
|
|
2017-10-11 21:37:54 +03:00
|
|
|
func randRawFeatureVector(r *rand.Rand) *RawFeatureVector {
|
|
|
|
featureVec := NewRawFeatureVector()
|
|
|
|
for i := 0; i < 10000; i++ {
|
|
|
|
if r.Int31n(2) == 0 {
|
|
|
|
featureVec.Set(FeatureBit(i))
|
2017-04-20 02:16:55 +03:00
|
|
|
}
|
2016-12-07 18:46:22 +03:00
|
|
|
}
|
2017-10-11 21:37:54 +03:00
|
|
|
return featureVec
|
2017-04-20 02:16:55 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestMaxOutPointIndex(t *testing.T) {
|
2017-06-17 01:59:20 +03:00
|
|
|
t.Parallel()
|
|
|
|
|
2017-04-20 02:16:55 +03:00
|
|
|
op := wire.OutPoint{
|
|
|
|
Index: math.MaxUint32,
|
2016-12-07 18:46:22 +03:00
|
|
|
}
|
2017-03-20 12:24:55 +03:00
|
|
|
|
2017-04-20 02:16:55 +03:00
|
|
|
var b bytes.Buffer
|
|
|
|
if err := writeElement(&b, op); err == nil {
|
|
|
|
t.Fatalf("write of outPoint should fail, index exceeds 16-bits")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestEmptyMessageUnknownType(t *testing.T) {
|
2017-06-17 01:59:20 +03:00
|
|
|
t.Parallel()
|
|
|
|
|
2017-04-20 02:16:55 +03:00
|
|
|
fakeType := MessageType(math.MaxUint16)
|
|
|
|
if _, err := makeEmptyMessage(fakeType); err == nil {
|
|
|
|
t.Fatalf("should not be able to make an empty message of an " +
|
|
|
|
"unknown type")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TestLightningWireProtocol uses the testing/quick package to create a series
|
|
|
|
// of fuzz tests to attempt to break a primary scenario which is implemented as
|
|
|
|
// property based testing scenario.
|
|
|
|
func TestLightningWireProtocol(t *testing.T) {
|
2017-06-17 01:59:20 +03:00
|
|
|
t.Parallel()
|
|
|
|
|
2017-04-20 02:16:55 +03:00
|
|
|
// mainScenario is the primary test that will programmatically be
|
|
|
|
// executed for all registered wire messages. The quick-checker within
|
|
|
|
// testing/quick will attempt to find an input to this function, s.t
|
|
|
|
// the function returns false, if so then we've found an input that
|
|
|
|
// violates our model of the system.
|
|
|
|
mainScenario := func(msg Message) bool {
|
|
|
|
// Give a new message, we'll serialize the message into a new
|
|
|
|
// bytes buffer.
|
|
|
|
var b bytes.Buffer
|
|
|
|
if _, err := WriteMessage(&b, msg, 0); err != nil {
|
|
|
|
t.Fatalf("unable to write msg: %v", err)
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
// Next, we'll ensure that the serialized payload (subtracting
|
|
|
|
// the 2 bytes for the message type) is _below_ the specified
|
|
|
|
// max payload size for this message.
|
|
|
|
payloadLen := uint32(b.Len()) - 2
|
|
|
|
if payloadLen > msg.MaxPayloadLength(0) {
|
|
|
|
t.Fatalf("msg payload constraint violated: %v > %v",
|
|
|
|
payloadLen, msg.MaxPayloadLength(0))
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
// Finally, we'll deserialize the message from the written
|
|
|
|
// buffer, and finally assert that the messages are equal.
|
2017-04-21 01:41:43 +03:00
|
|
|
newMsg, err := ReadMessage(&b, 0)
|
2017-04-20 02:16:55 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to read msg: %v", err)
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
if !reflect.DeepEqual(msg, newMsg) {
|
|
|
|
t.Fatalf("messages don't match after re-encoding: %v "+
|
2017-08-22 08:33:20 +03:00
|
|
|
"vs %v", spew.Sdump(msg), spew.Sdump(newMsg))
|
2017-04-20 02:16:55 +03:00
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// customTypeGen is a map of functions that are able to randomly
|
|
|
|
// generate a given type. These functions are needed for types which
|
|
|
|
// are too complex for the testing/quick package to automatically
|
|
|
|
// generate.
|
|
|
|
customTypeGen := map[MessageType]func([]reflect.Value, *rand.Rand){
|
|
|
|
MsgInit: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := NewInitMessage(
|
2017-10-11 21:37:54 +03:00
|
|
|
randRawFeatureVector(r),
|
|
|
|
randRawFeatureVector(r),
|
2017-04-20 02:16:55 +03:00
|
|
|
)
|
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(*req)
|
|
|
|
},
|
2017-07-29 02:32:26 +03:00
|
|
|
MsgOpenChannel: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := OpenChannel{
|
|
|
|
FundingAmount: btcutil.Amount(r.Int63()),
|
2017-08-22 08:33:20 +03:00
|
|
|
PushAmount: MilliSatoshi(r.Int63()),
|
2017-07-29 02:32:26 +03:00
|
|
|
DustLimit: btcutil.Amount(r.Int63()),
|
2017-08-22 08:33:20 +03:00
|
|
|
MaxValueInFlight: MilliSatoshi(r.Int63()),
|
2017-07-29 02:32:26 +03:00
|
|
|
ChannelReserve: btcutil.Amount(r.Int63()),
|
2017-08-22 08:33:20 +03:00
|
|
|
HtlcMinimum: MilliSatoshi(r.Int31()),
|
2017-07-29 02:32:26 +03:00
|
|
|
FeePerKiloWeight: uint32(r.Int63()),
|
|
|
|
CsvDelay: uint16(r.Int31()),
|
|
|
|
MaxAcceptedHTLCs: uint16(r.Int31()),
|
2017-11-09 22:59:30 +03:00
|
|
|
ChannelFlags: FundingFlag(uint8(r.Int31())),
|
2017-07-29 02:32:26 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if _, err := r.Read(req.ChainHash[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chain hash: %v", err)
|
|
|
|
return
|
2017-04-20 02:16:55 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if _, err := r.Read(req.PendingChannelID[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate pending chan id: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
var err error
|
2017-07-29 02:32:26 +03:00
|
|
|
req.FundingKey, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
req.RevocationPoint, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
req.PaymentPoint, err = randPubKey()
|
2017-04-20 02:16:55 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-07-29 02:32:26 +03:00
|
|
|
req.DelayedPaymentPoint, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-11-15 07:26:51 +03:00
|
|
|
req.HtlcPoint, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-07-29 02:32:26 +03:00
|
|
|
req.FirstCommitmentPoint, err = randPubKey()
|
2017-04-20 02:16:55 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(req)
|
|
|
|
},
|
2017-07-29 02:32:26 +03:00
|
|
|
MsgAcceptChannel: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := AcceptChannel{
|
|
|
|
DustLimit: btcutil.Amount(r.Int63()),
|
2017-08-22 08:33:20 +03:00
|
|
|
MaxValueInFlight: MilliSatoshi(r.Int63()),
|
2017-07-29 02:32:26 +03:00
|
|
|
ChannelReserve: btcutil.Amount(r.Int63()),
|
|
|
|
MinAcceptDepth: uint32(r.Int31()),
|
2017-08-22 08:33:20 +03:00
|
|
|
HtlcMinimum: MilliSatoshi(r.Int31()),
|
2017-07-29 02:32:26 +03:00
|
|
|
CsvDelay: uint16(r.Int31()),
|
|
|
|
MaxAcceptedHTLCs: uint16(r.Int31()),
|
2017-04-20 02:16:55 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
if _, err := r.Read(req.PendingChannelID[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate pending chan id: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
var err error
|
2017-07-29 02:32:26 +03:00
|
|
|
req.FundingKey, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
req.RevocationPoint, err = randPubKey()
|
2017-04-20 02:16:55 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-07-29 02:32:26 +03:00
|
|
|
req.PaymentPoint, err = randPubKey()
|
2017-04-20 02:16:55 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-07-29 02:32:26 +03:00
|
|
|
req.DelayedPaymentPoint, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-11-15 07:26:51 +03:00
|
|
|
req.HtlcPoint, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-07-29 02:32:26 +03:00
|
|
|
req.FirstCommitmentPoint, err = randPubKey()
|
2017-04-20 02:16:55 +03:00
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(req)
|
|
|
|
},
|
2017-07-29 02:32:26 +03:00
|
|
|
MsgFundingCreated: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := FundingCreated{}
|
2017-04-20 02:16:55 +03:00
|
|
|
|
|
|
|
if _, err := r.Read(req.PendingChannelID[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate pending chan id: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2017-07-29 02:32:26 +03:00
|
|
|
if _, err := r.Read(req.FundingPoint.Hash[:]); err != nil {
|
2017-04-20 02:16:55 +03:00
|
|
|
t.Fatalf("unable to generate hash: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-07-29 02:32:26 +03:00
|
|
|
req.FundingPoint.Index = uint32(r.Int31()) % math.MaxUint16
|
2017-04-20 02:16:55 +03:00
|
|
|
|
2017-07-29 02:32:26 +03:00
|
|
|
req.CommitSig = testSig
|
2017-04-20 02:16:55 +03:00
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(req)
|
|
|
|
},
|
2017-07-29 02:32:26 +03:00
|
|
|
MsgFundingSigned: func(v []reflect.Value, r *rand.Rand) {
|
2017-04-20 02:16:55 +03:00
|
|
|
var c [32]byte
|
|
|
|
if _, err := r.Read(c[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chan id: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2017-07-29 02:32:26 +03:00
|
|
|
req := FundingSigned{
|
|
|
|
ChanID: ChannelID(c),
|
|
|
|
CommitSig: testSig,
|
|
|
|
}
|
2017-04-20 02:16:55 +03:00
|
|
|
|
2017-07-29 02:32:26 +03:00
|
|
|
v[0] = reflect.ValueOf(req)
|
2017-04-20 02:16:55 +03:00
|
|
|
},
|
|
|
|
MsgFundingLocked: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
|
|
|
|
var c [32]byte
|
|
|
|
if _, err := r.Read(c[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chan id: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
pubKey, err := randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
req := NewFundingLocked(ChannelID(c), pubKey)
|
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(*req)
|
|
|
|
},
|
2017-03-09 02:32:11 +03:00
|
|
|
MsgClosingSigned: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := ClosingSigned{
|
|
|
|
FeeSatoshis: uint64(r.Int63()),
|
|
|
|
Signature: testSig,
|
2017-04-20 02:16:55 +03:00
|
|
|
}
|
|
|
|
|
2017-03-09 02:32:11 +03:00
|
|
|
if _, err := r.Read(req.ChannelID[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chan id: %v", err)
|
2017-04-20 02:16:55 +03:00
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(req)
|
|
|
|
},
|
|
|
|
MsgCommitSig: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := NewCommitSig()
|
|
|
|
if _, err := r.Read(req.ChanID[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chan id: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
req.CommitSig = testSig
|
|
|
|
|
2017-08-29 05:16:03 +03:00
|
|
|
// Only create the slice if there will be any signatures
|
|
|
|
// in it to prevent false positive test failures due to
|
|
|
|
// an empty slice versus a nil slice.
|
2017-07-29 02:32:26 +03:00
|
|
|
numSigs := uint16(r.Int31n(1020))
|
2017-08-29 05:16:03 +03:00
|
|
|
if numSigs > 0 {
|
|
|
|
req.HtlcSigs = make([]*btcec.Signature, numSigs)
|
|
|
|
}
|
2017-07-29 02:32:26 +03:00
|
|
|
for i := 0; i < int(numSigs); i++ {
|
|
|
|
req.HtlcSigs[i] = testSig
|
|
|
|
}
|
|
|
|
|
2017-04-20 02:16:55 +03:00
|
|
|
v[0] = reflect.ValueOf(*req)
|
|
|
|
},
|
|
|
|
MsgRevokeAndAck: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := NewRevokeAndAck()
|
|
|
|
if _, err := r.Read(req.ChanID[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chan id: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if _, err := r.Read(req.Revocation[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate bytes: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
var err error
|
|
|
|
req.NextRevocationKey, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(*req)
|
|
|
|
},
|
|
|
|
MsgChannelAnnouncement: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := ChannelAnnouncement{
|
|
|
|
ShortChannelID: NewShortChanIDFromInt(uint64(r.Int63())),
|
2017-10-11 21:37:54 +03:00
|
|
|
Features: randRawFeatureVector(r),
|
2017-04-20 02:16:55 +03:00
|
|
|
}
|
|
|
|
req.NodeSig1 = testSig
|
|
|
|
req.NodeSig2 = testSig
|
|
|
|
req.BitcoinSig1 = testSig
|
|
|
|
req.BitcoinSig2 = testSig
|
|
|
|
|
|
|
|
var err error
|
|
|
|
req.NodeID1, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
req.NodeID2, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
req.BitcoinKey1, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
req.BitcoinKey2, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-08-22 08:46:56 +03:00
|
|
|
if _, err := r.Read(req.ChainHash[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chain hash: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-04-20 02:16:55 +03:00
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(req)
|
|
|
|
},
|
|
|
|
MsgNodeAnnouncement: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
var a [32]byte
|
|
|
|
if _, err := r.Read(a[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate alias: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
req := NodeAnnouncement{
|
|
|
|
Signature: testSig,
|
2017-10-11 21:37:54 +03:00
|
|
|
Features: randRawFeatureVector(r),
|
2017-04-20 02:16:55 +03:00
|
|
|
Timestamp: uint32(r.Int31()),
|
2017-08-22 08:42:50 +03:00
|
|
|
Alias: a,
|
2017-04-20 02:16:55 +03:00
|
|
|
RGBColor: RGB{
|
|
|
|
red: uint8(r.Int31()),
|
|
|
|
green: uint8(r.Int31()),
|
|
|
|
blue: uint8(r.Int31()),
|
|
|
|
},
|
2017-09-19 03:27:37 +03:00
|
|
|
// TODO(roasbeef): proper gen rand addrs
|
2017-04-20 02:16:55 +03:00
|
|
|
Addresses: testAddrs,
|
|
|
|
}
|
|
|
|
|
|
|
|
var err error
|
|
|
|
req.NodeID, err = randPubKey()
|
|
|
|
if err != nil {
|
|
|
|
t.Fatalf("unable to generate key: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(req)
|
|
|
|
},
|
|
|
|
MsgChannelUpdate: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := ChannelUpdate{
|
2017-06-16 23:46:31 +03:00
|
|
|
Signature: testSig,
|
|
|
|
ShortChannelID: NewShortChanIDFromInt(uint64(r.Int63())),
|
|
|
|
Timestamp: uint32(r.Int31()),
|
|
|
|
Flags: uint16(r.Int31()),
|
|
|
|
TimeLockDelta: uint16(r.Int31()),
|
2017-08-22 08:33:20 +03:00
|
|
|
HtlcMinimumMsat: MilliSatoshi(r.Int63()),
|
2017-06-16 23:46:31 +03:00
|
|
|
BaseFee: uint32(r.Int31()),
|
|
|
|
FeeRate: uint32(r.Int31()),
|
2017-04-20 02:16:55 +03:00
|
|
|
}
|
2017-08-22 08:46:56 +03:00
|
|
|
if _, err := r.Read(req.ChainHash[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chain hash: %v", err)
|
|
|
|
return
|
|
|
|
}
|
2017-04-20 02:16:55 +03:00
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(req)
|
|
|
|
},
|
|
|
|
MsgAnnounceSignatures: func(v []reflect.Value, r *rand.Rand) {
|
|
|
|
req := AnnounceSignatures{
|
|
|
|
ShortChannelID: NewShortChanIDFromInt(uint64(r.Int63())),
|
|
|
|
NodeSignature: testSig,
|
|
|
|
BitcoinSignature: testSig,
|
|
|
|
}
|
|
|
|
if _, err := r.Read(req.ChannelID[:]); err != nil {
|
|
|
|
t.Fatalf("unable to generate chan id: %v", err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
v[0] = reflect.ValueOf(req)
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
// With the above types defined, we'll now generate a slice of
|
|
|
|
// scenarios to feed into quick.Check. The function scans in input
|
|
|
|
// space of the target function under test, so we'll need to create a
|
|
|
|
// series of wrapper functions to force it to iterate over the target
|
|
|
|
// types, but re-use the mainScenario defined above.
|
|
|
|
tests := []struct {
|
|
|
|
msgType MessageType
|
|
|
|
scenario interface{}
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
msgType: MsgInit,
|
|
|
|
scenario: func(m Init) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgError,
|
|
|
|
scenario: func(m Error) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgPing,
|
|
|
|
scenario: func(m Ping) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgPong,
|
|
|
|
scenario: func(m Pong) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2017-07-29 02:32:26 +03:00
|
|
|
msgType: MsgOpenChannel,
|
|
|
|
scenario: func(m OpenChannel) bool {
|
2017-04-20 02:16:55 +03:00
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2017-07-29 02:32:26 +03:00
|
|
|
msgType: MsgAcceptChannel,
|
|
|
|
scenario: func(m AcceptChannel) bool {
|
2017-04-20 02:16:55 +03:00
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2017-07-29 02:32:26 +03:00
|
|
|
msgType: MsgFundingCreated,
|
|
|
|
scenario: func(m FundingCreated) bool {
|
2017-04-20 02:16:55 +03:00
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2017-07-29 02:32:26 +03:00
|
|
|
msgType: MsgFundingSigned,
|
|
|
|
scenario: func(m FundingSigned) bool {
|
2017-04-20 02:16:55 +03:00
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgFundingLocked,
|
|
|
|
scenario: func(m FundingLocked) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2017-03-09 02:32:11 +03:00
|
|
|
msgType: MsgShutdown,
|
|
|
|
scenario: func(m Shutdown) bool {
|
2017-04-20 02:16:55 +03:00
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2017-03-09 02:32:11 +03:00
|
|
|
msgType: MsgClosingSigned,
|
|
|
|
scenario: func(m ClosingSigned) bool {
|
2017-04-20 02:16:55 +03:00
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgUpdateAddHTLC,
|
|
|
|
scenario: func(m UpdateAddHTLC) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgUpdateFufillHTLC,
|
|
|
|
scenario: func(m UpdateFufillHTLC) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgUpdateFailHTLC,
|
|
|
|
scenario: func(m UpdateFailHTLC) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgCommitSig,
|
|
|
|
scenario: func(m CommitSig) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgRevokeAndAck,
|
|
|
|
scenario: func(m RevokeAndAck) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
2017-07-14 21:28:40 +03:00
|
|
|
{
|
|
|
|
msgType: MsgUpdateFee,
|
|
|
|
scenario: func(m UpdateFee) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
2017-06-28 18:22:23 +03:00
|
|
|
{
|
|
|
|
|
|
|
|
msgType: MsgUpdateFailMalformedHTLC,
|
|
|
|
scenario: func(m UpdateFailMalformedHTLC) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
2017-07-09 01:12:36 +03:00
|
|
|
{
|
|
|
|
msgType: MsgChannelReestablish,
|
|
|
|
scenario: func(m ChannelReestablish) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
2017-04-20 02:16:55 +03:00
|
|
|
{
|
|
|
|
msgType: MsgChannelAnnouncement,
|
|
|
|
scenario: func(m ChannelAnnouncement) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgNodeAnnouncement,
|
|
|
|
scenario: func(m NodeAnnouncement) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgChannelUpdate,
|
|
|
|
scenario: func(m ChannelUpdate) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
|
|
|
msgType: MsgAnnounceSignatures,
|
|
|
|
scenario: func(m AnnounceSignatures) bool {
|
|
|
|
return mainScenario(&m)
|
|
|
|
},
|
|
|
|
},
|
|
|
|
}
|
|
|
|
for _, test := range tests {
|
|
|
|
var config *quick.Config
|
|
|
|
|
|
|
|
// If the type defined is within the custom type gen map above,
|
|
|
|
// then we'll modify the default config to use this Value
|
|
|
|
// function that knows how to generate the proper types.
|
|
|
|
if valueGen, ok := customTypeGen[test.msgType]; ok {
|
|
|
|
config = &quick.Config{
|
|
|
|
Values: valueGen,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
t.Logf("Running fuzz tests for msgType=%v", test.msgType)
|
|
|
|
if err := quick.Check(test.scenario, config); err != nil {
|
|
|
|
t.Fatalf("fuzz checks for msg=%v failed: %v",
|
|
|
|
test.msgType, err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|