13980885714

Committed 20 Mar 2025 10:53PM UTC coverage: 58.613% (-10.2%) from 68.789%

Build # 13980885714

Build Type

Pull #9623

github

Committed by

web-flow

Commit Message

Merge 9eaec1f7a into 09b674508

Pull Request Pull Request #9623: Size msg test msg

Run Details

0 of 1572 new or added lines in 42 files covered. (0.0%)

27755 existing lines in 442 files now uncovered.

96886 of 165299 relevant lines covered (58.61%)

1.82 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

30.66

/lnwire/channel_update.go

package lnwire

import (
        "bytes"
        "fmt"
        "io"

        "github.com/btcsuite/btcd/chaincfg/chainhash"
        "pgregory.net/rapid"
)

// ChanUpdateMsgFlags is a bitfield that signals whether optional fields are
// present in the ChannelUpdate.
type ChanUpdateMsgFlags uint8

const (
        // ChanUpdateRequiredMaxHtlc is a bit that indicates whether the
        // required htlc_maximum_msat field is present in this ChannelUpdate.
        ChanUpdateRequiredMaxHtlc ChanUpdateMsgFlags = 1 << iota
)

// String returns the bitfield flags as a string.
func (c ChanUpdateMsgFlags) String() string {
        return fmt.Sprintf("%08b", c)
}

// HasMaxHtlc returns true if the htlc_maximum_msat option bit is set in the
// message flags.
func (c ChanUpdateMsgFlags) HasMaxHtlc() bool {
        return c&ChanUpdateRequiredMaxHtlc != 0
}

// ChanUpdateChanFlags is a bitfield that signals various options concerning a
// particular channel edge. Each bit is to be examined in order to determine
// how the ChannelUpdate message is to be interpreted.
type ChanUpdateChanFlags uint8

const (
        // ChanUpdateDirection indicates the direction of a channel update. If
        // this bit is set to 0 if Node1 (the node with the "smaller" Node ID)
        // is updating the channel, and to 1 otherwise.
        ChanUpdateDirection ChanUpdateChanFlags = 1 << iota

        // ChanUpdateDisabled is a bit that indicates if the channel edge
        // selected by the ChanUpdateDirection bit is to be treated as being
        // disabled.
        ChanUpdateDisabled
)

// IsDisabled determines whether the channel flags has the disabled bit set.
func (c ChanUpdateChanFlags) IsDisabled() bool {
        return c&ChanUpdateDisabled == ChanUpdateDisabled
}

// String returns the bitfield flags as a string.
func (c ChanUpdateChanFlags) String() string {
        return fmt.Sprintf("%08b", c)
}

// ChannelUpdate1 message is used after channel has been initially announced.
// Each side independently announces its fees and minimum expiry for HTLCs and
// other parameters. Also this message is used to redeclare initially set
// channel parameters.
type ChannelUpdate1 struct {
        // Signature is used to validate the announced data and prove the
        // ownership of node id.
        Signature Sig

        // ChainHash denotes the target chain that this channel was opened
        // within. This value should be the genesis hash of the target chain.
        // Along with the short channel ID, this uniquely identifies the
        // channel globally in a blockchain.
        ChainHash chainhash.Hash

        // ShortChannelID is the unique description of the funding transaction.
        ShortChannelID ShortChannelID

        // Timestamp allows ordering in the case of multiple announcements. We
        // should ignore the message if timestamp is not greater than
        // the last-received.
        Timestamp uint32

        // MessageFlags is a bitfield that describes whether optional fields
        // are present in this update. Currently, the least-significant bit
        // must be set to 1 if the optional field MaxHtlc is present.
        MessageFlags ChanUpdateMsgFlags

        // ChannelFlags is a bitfield that describes additional meta-data
        // concerning how the update is to be interpreted. Currently, the
        // least-significant bit must be set to 0 if the creating node
        // corresponds to the first node in the previously sent channel
        // announcement and 1 otherwise. If the second bit is set, then the
        // channel is set to be disabled.
        ChannelFlags ChanUpdateChanFlags

        // TimeLockDelta is the minimum number of blocks this node requires to
        // be added to the expiry of HTLCs. This is a security parameter
        // determined by the node operator. This value represents the required
        // gap between the time locks of the incoming and outgoing HTLC's set
        // to this node.
        TimeLockDelta uint16

        // HtlcMinimumMsat is the minimum HTLC value which will be accepted.
        HtlcMinimumMsat MilliSatoshi

        // BaseFee is the base fee that must be used for incoming HTLC's to
        // this particular channel. This value will be tacked onto the required
        // for a payment independent of the size of the payment.
        BaseFee uint32

        // FeeRate is the fee rate that will be charged per millionth of a
        // satoshi.
        FeeRate uint32

        // HtlcMaximumMsat is the maximum HTLC value which will be accepted.
        HtlcMaximumMsat MilliSatoshi

        // ExtraData is the set of data that was appended to this message to
        // fill out the full maximum transport message size. These fields can
        // be used to specify optional data such as custom TLV fields.
        ExtraOpaqueData ExtraOpaqueData
}

// A compile time check to ensure ChannelUpdate implements the lnwire.Message
// interface.
var _ Message = (*ChannelUpdate1)(nil)

// A compile time check to ensure ChannelUpdate1 implements the
// lnwire.SizeableMessage interface.
var _ SizeableMessage = (*ChannelUpdate1)(nil)

// Decode deserializes a serialized ChannelUpdate stored in the passed
// io.Reader observing the specified protocol version.
//
// This is part of the lnwire.Message interface.
func (a *ChannelUpdate1) Decode(r io.Reader, pver uint32) error {
        err := ReadElements(r,
                &a.Signature,
                a.ChainHash[:],
                &a.ShortChannelID,
                &a.Timestamp,
                &a.MessageFlags,
                &a.ChannelFlags,
                &a.TimeLockDelta,
                &a.HtlcMinimumMsat,
                &a.BaseFee,
                &a.FeeRate,
        )
        if err != nil {
                return err
        }

        // Now check whether the max HTLC field is present and read it if so.
        if a.MessageFlags.HasMaxHtlc() {
                if err := ReadElements(r, &a.HtlcMaximumMsat); err != nil {
                        return err
                }
        }

        return a.ExtraOpaqueData.Decode(r)
}

// Encode serializes the target ChannelUpdate into the passed io.Writer
// observing the protocol version specified.
//
// This is part of the lnwire.Message interface.
func (a *ChannelUpdate1) Encode(w *bytes.Buffer, pver uint32) error {
        if err := WriteSig(w, a.Signature); err != nil {
                return err
        }

        if err := WriteBytes(w, a.ChainHash[:]); err != nil {
                return err
        }

        if err := WriteShortChannelID(w, a.ShortChannelID); err != nil {
                return err
        }

        if err := WriteUint32(w, a.Timestamp); err != nil {
                return err
        }

        if err := WriteChanUpdateMsgFlags(w, a.MessageFlags); err != nil {
                return err
        }

        if err := WriteChanUpdateChanFlags(w, a.ChannelFlags); err != nil {
                return err
        }

        if err := WriteUint16(w, a.TimeLockDelta); err != nil {
                return err
        }

        if err := WriteMilliSatoshi(w, a.HtlcMinimumMsat); err != nil {
                return err
        }

        if err := WriteUint32(w, a.BaseFee); err != nil {
                return err
        }

        if err := WriteUint32(w, a.FeeRate); err != nil {
                return err
        }

        // Now append optional fields if they are set. Currently, the only
        // optional field is max HTLC.
        if a.MessageFlags.HasMaxHtlc() {
                err := WriteMilliSatoshi(w, a.HtlcMaximumMsat)
                if err != nil {
                        return err
                }
        }

        // Finally, append any extra opaque data.
        return WriteBytes(w, a.ExtraOpaqueData)
}

// MsgType returns the integer uniquely identifying this message type on the
// wire.
//
// This is part of the lnwire.Message interface.
func (a *ChannelUpdate1) MsgType() MessageType {
        return MsgChannelUpdate
}

// DataToSign is used to retrieve part of the announcement message which should
// be signed.
func (a *ChannelUpdate1) DataToSign() ([]byte, error) {
        // We should not include the signatures itself.
        b := make([]byte, 0, MaxMsgBody)
        buf := bytes.NewBuffer(b)
        if err := WriteBytes(buf, a.ChainHash[:]); err != nil {
                return nil, err
        }

        if err := WriteShortChannelID(buf, a.ShortChannelID); err != nil {
                return nil, err
        }

        if err := WriteUint32(buf, a.Timestamp); err != nil {
                return nil, err
        }

        if err := WriteChanUpdateMsgFlags(buf, a.MessageFlags); err != nil {
                return nil, err
        }

        if err := WriteChanUpdateChanFlags(buf, a.ChannelFlags); err != nil {
                return nil, err
        }

        if err := WriteUint16(buf, a.TimeLockDelta); err != nil {
                return nil, err
        }

        if err := WriteMilliSatoshi(buf, a.HtlcMinimumMsat); err != nil {
                return nil, err
        }

        if err := WriteUint32(buf, a.BaseFee); err != nil {
                return nil, err
        }

        if err := WriteUint32(buf, a.FeeRate); err != nil {
                return nil, err
        }

        // Now append optional fields if they are set. Currently, the only
        // optional field is max HTLC.
        if a.MessageFlags.HasMaxHtlc() {
                err := WriteMilliSatoshi(buf, a.HtlcMaximumMsat)
                if err != nil {
                        return nil, err
                }
        }

        // Finally, append any extra opaque data.
        if err := WriteBytes(buf, a.ExtraOpaqueData); err != nil {
                return nil, err
        }

        return buf.Bytes(), nil
}

// SCID returns the ShortChannelID of the channel that the update applies to.
//
// NOTE: this is part of the ChannelUpdate interface.
func (a *ChannelUpdate1) SCID() ShortChannelID {
        return a.ShortChannelID
}

// IsNode1 is true if the update was produced by node 1 of the channel peers.
// Node 1 is the node with the lexicographically smaller public key.
//
// NOTE: this is part of the ChannelUpdate interface.
func (a *ChannelUpdate1) IsNode1() bool {
        return a.ChannelFlags&ChanUpdateDirection == 0
}

// IsDisabled is true if the update is announcing that the channel should be
// considered disabled.
//
// NOTE: this is part of the ChannelUpdate interface.
func (a *ChannelUpdate1) IsDisabled() bool {
        return a.ChannelFlags&ChanUpdateDisabled == ChanUpdateDisabled
}

// GetChainHash returns the hash of the chain that the message is referring to.
//
// NOTE: this is part of the ChannelUpdate interface.
func (a *ChannelUpdate1) GetChainHash() chainhash.Hash {
        return a.ChainHash
}

// ForwardingPolicy returns the set of forwarding constraints of the update.
//
// NOTE: this is part of the ChannelUpdate interface.
func (a *ChannelUpdate1) ForwardingPolicy() *ForwardingPolicy {
        return &ForwardingPolicy{
                TimeLockDelta: a.TimeLockDelta,
                BaseFee:       MilliSatoshi(a.BaseFee),
                FeeRate:       MilliSatoshi(a.FeeRate),
                MinHTLC:       a.HtlcMinimumMsat,
                HasMaxHTLC:    a.MessageFlags.HasMaxHtlc(),
                MaxHTLC:       a.HtlcMaximumMsat,
        }
}

// CmpAge can be used to determine if the update is older or newer than the
// passed update. It returns 1 if this update is newer, -1 if it is older, and
// 0 if they are the same age.
//
// NOTE: this is part of the ChannelUpdate interface.
func (a *ChannelUpdate1) CmpAge(update ChannelUpdate) (CompareResult, error) {
        other, ok := update.(*ChannelUpdate1)
        if !ok {
                return 0, fmt.Errorf("expected *ChannelUpdate1, got: %T",
                        update)
        }

        switch {
        case a.Timestamp > other.Timestamp:
                return GreaterThan, nil
        case a.Timestamp < other.Timestamp:
                return LessThan, nil
        default:
                return EqualTo, nil
        }
}

// SetDisabledFlag can be used to adjust the disabled flag of an update.
//
// NOTE: this is part of the ChannelUpdate interface.
func (a *ChannelUpdate1) SetDisabledFlag(disabled bool) {
        if disabled {
                a.ChannelFlags |= ChanUpdateDisabled
        } else {
                a.ChannelFlags &= ^ChanUpdateDisabled
        }
}

// SetSCID can be used to overwrite the SCID of the update.
//
// NOTE: this is part of the ChannelUpdate interface.
func (a *ChannelUpdate1) SetSCID(scid ShortChannelID) {
        a.ShortChannelID = scid
}

// A compile time assertion to ensure ChannelUpdate1 implements the
// ChannelUpdate interface.
var _ ChannelUpdate = (*ChannelUpdate1)(nil)

// SerializedSize returns the serialized size of the message in bytes.
//
// This is part of the lnwire.SizeableMessage interface.
func (a *ChannelUpdate1) SerializedSize() (uint32, error) {
        return MessageSerializedSize(a)
}

// A compile time check to ensure ChannelUpdate1 implements the TestMessage
// interface.
var _ TestMessage = (*ChannelUpdate1)(nil)

// RandTestMessage populates the message with random data suitable for testing.
// It uses the rapid testing framework to generate random values.
//
// This is part of the TestMessage interface.
func (a *ChannelUpdate1) RandTestMessage(t *rapid.T) Message {
        // Generate random message flags
        // Randomly decide whether to include max HTLC field
        includeMaxHtlc := rapid.Bool().Draw(t, "includeMaxHtlc")
        var msgFlags ChanUpdateMsgFlags
        if includeMaxHtlc {
                msgFlags |= ChanUpdateRequiredMaxHtlc
        }

        // Generate random channel flags
        // Randomly decide direction (node1 or node2)
        isNode2 := rapid.Bool().Draw(t, "isNode2")
        var chanFlags ChanUpdateChanFlags
        if isNode2 {
                chanFlags |= ChanUpdateDirection
        }

        // Randomly decide if channel is disabled
        isDisabled := rapid.Bool().Draw(t, "isDisabled")
        if isDisabled {
                chanFlags |= ChanUpdateDisabled
        }

        // Generate chain hash
        chainHash := RandChainHash(t)
        var hash chainhash.Hash
        copy(hash[:], chainHash[:])

        // Generate other random fields
        maxHtlc := MilliSatoshi(rapid.Uint64().Draw(t, "maxHtlc"))

        // If max HTLC flag is not set, we need to zero the value
        if !includeMaxHtlc {
                maxHtlc = 0
        }

        return &ChannelUpdate1{
                Signature:      RandSignature(t),
                ChainHash:      hash,
                ShortChannelID: RandShortChannelID(t),
                Timestamp: uint32(rapid.IntRange(0, 0x7FFFFFFF).Draw(
                        t, "timestamp"),
                ),
                MessageFlags: msgFlags,
                ChannelFlags: chanFlags,
                TimeLockDelta: uint16(rapid.IntRange(0, 65535).Draw(
                        t, "timelockDelta"),
                ),
                HtlcMinimumMsat: MilliSatoshi(rapid.Uint64().Draw(
                        t, "htlcMinimum"),
                ),
                BaseFee: uint32(rapid.IntRange(0, 0x7FFFFFFF).Draw(
                        t, "baseFee"),
                ),
                FeeRate: uint32(rapid.IntRange(0, 0x7FFFFFFF).Draw(
                        t, "feeRate"),
                ),
                HtlcMaximumMsat: maxHtlc,
                ExtraOpaqueData: RandExtraOpaqueData(t, nil),
        }
}

1	package lnwire
2
3	import (
4	"bytes"
5	"fmt"
6	"io"
7
8	"github.com/btcsuite/btcd/chaincfg/chainhash"
9	"pgregory.net/rapid"
10	)
11
12	// ChanUpdateMsgFlags is a bitfield that signals whether optional fields are
13	// present in the ChannelUpdate.
14	type ChanUpdateMsgFlags uint8
15
16	const (
17	// ChanUpdateRequiredMaxHtlc is a bit that indicates whether the
18	// required htlc_maximum_msat field is present in this ChannelUpdate.
19	ChanUpdateRequiredMaxHtlc ChanUpdateMsgFlags = 1 << iota
20	)
21
22	// String returns the bitfield flags as a string.
23	func (c ChanUpdateMsgFlags) String() string {	3✔
24	return fmt.Sprintf("%08b", c)	3✔
25	}	3✔
26
27	// HasMaxHtlc returns true if the htlc_maximum_msat option bit is set in the
28	// message flags.
29	func (c ChanUpdateMsgFlags) HasMaxHtlc() bool {	3✔
30	return c&ChanUpdateRequiredMaxHtlc != 0	3✔
31	}	3✔
32
33	// ChanUpdateChanFlags is a bitfield that signals various options concerning a
34	// particular channel edge. Each bit is to be examined in order to determine
35	// how the ChannelUpdate message is to be interpreted.
36	type ChanUpdateChanFlags uint8
37
38	const (
39	// ChanUpdateDirection indicates the direction of a channel update. If
40	// this bit is set to 0 if Node1 (the node with the "smaller" Node ID)
41	// is updating the channel, and to 1 otherwise.
42	ChanUpdateDirection ChanUpdateChanFlags = 1 << iota
43
44	// ChanUpdateDisabled is a bit that indicates if the channel edge
45	// selected by the ChanUpdateDirection bit is to be treated as being
46	// disabled.
47	ChanUpdateDisabled
48	)
49
50	// IsDisabled determines whether the channel flags has the disabled bit set.
51	func (c ChanUpdateChanFlags) IsDisabled() bool {	3✔
52	return c&ChanUpdateDisabled == ChanUpdateDisabled	3✔
53	}	3✔
54
55	// String returns the bitfield flags as a string.
56	func (c ChanUpdateChanFlags) String() string {	3✔
57	return fmt.Sprintf("%08b", c)	3✔
58	}	3✔
59
60	// ChannelUpdate1 message is used after channel has been initially announced.
61	// Each side independently announces its fees and minimum expiry for HTLCs and
62	// other parameters. Also this message is used to redeclare initially set
63	// channel parameters.
64	type ChannelUpdate1 struct {
65	// Signature is used to validate the announced data and prove the
66	// ownership of node id.
67	Signature Sig
68
69	// ChainHash denotes the target chain that this channel was opened
70	// within. This value should be the genesis hash of the target chain.
71	// Along with the short channel ID, this uniquely identifies the
72	// channel globally in a blockchain.
73	ChainHash chainhash.Hash
74
75	// ShortChannelID is the unique description of the funding transaction.
76	ShortChannelID ShortChannelID
77
78	// Timestamp allows ordering in the case of multiple announcements. We
79	// should ignore the message if timestamp is not greater than
80	// the last-received.
81	Timestamp uint32
82
83	// MessageFlags is a bitfield that describes whether optional fields
84	// are present in this update. Currently, the least-significant bit
85	// must be set to 1 if the optional field MaxHtlc is present.
86	MessageFlags ChanUpdateMsgFlags
87
88	// ChannelFlags is a bitfield that describes additional meta-data
89	// concerning how the update is to be interpreted. Currently, the
90	// least-significant bit must be set to 0 if the creating node
91	// corresponds to the first node in the previously sent channel
92	// announcement and 1 otherwise. If the second bit is set, then the
93	// channel is set to be disabled.
94	ChannelFlags ChanUpdateChanFlags
95
96	// TimeLockDelta is the minimum number of blocks this node requires to
97	// be added to the expiry of HTLCs. This is a security parameter
98	// determined by the node operator. This value represents the required
99	// gap between the time locks of the incoming and outgoing HTLC's set
100	// to this node.
101	TimeLockDelta uint16
102
103	// HtlcMinimumMsat is the minimum HTLC value which will be accepted.
104	HtlcMinimumMsat MilliSatoshi
105
106	// BaseFee is the base fee that must be used for incoming HTLC's to
107	// this particular channel. This value will be tacked onto the required
108	// for a payment independent of the size of the payment.
109	BaseFee uint32
110
111	// FeeRate is the fee rate that will be charged per millionth of a
112	// satoshi.
113	FeeRate uint32
114
115	// HtlcMaximumMsat is the maximum HTLC value which will be accepted.
116	HtlcMaximumMsat MilliSatoshi
117
118	// ExtraData is the set of data that was appended to this message to
119	// fill out the full maximum transport message size. These fields can
120	// be used to specify optional data such as custom TLV fields.
121	ExtraOpaqueData ExtraOpaqueData
122	}
123
124	// A compile time check to ensure ChannelUpdate implements the lnwire.Message
125	// interface.
126	var _ Message = (*ChannelUpdate1)(nil)
127
128	// A compile time check to ensure ChannelUpdate1 implements the
129	// lnwire.SizeableMessage interface.
130	var _ SizeableMessage = (*ChannelUpdate1)(nil)
131
132	// Decode deserializes a serialized ChannelUpdate stored in the passed
133	// io.Reader observing the specified protocol version.
134	//
135	// This is part of the lnwire.Message interface.
136	func (a *ChannelUpdate1) Decode(r io.Reader, pver uint32) error {	3✔
137	err := ReadElements(r,	3✔
138	&a.Signature,	3✔
139	a.ChainHash[:],	3✔
140	&a.ShortChannelID,	3✔
141	&a.Timestamp,	3✔
142	&a.MessageFlags,	3✔
143	&a.ChannelFlags,	3✔
144	&a.TimeLockDelta,	3✔
145	&a.HtlcMinimumMsat,	3✔
146	&a.BaseFee,	3✔
147	&a.FeeRate,	3✔
148	)	3✔
149	if err != nil {	3✔
UNCOV 150	return err	×
UNCOV 151	}	×
152
153	// Now check whether the max HTLC field is present and read it if so.
154	if a.MessageFlags.HasMaxHtlc() {	6✔
155	if err := ReadElements(r, &a.HtlcMaximumMsat); err != nil {	3✔
UNCOV 156	return err	×
UNCOV 157	}	×
158	}
159
160	return a.ExtraOpaqueData.Decode(r)	3✔
161	}
162
163	// Encode serializes the target ChannelUpdate into the passed io.Writer
164	// observing the protocol version specified.
165	//
166	// This is part of the lnwire.Message interface.
167	func (a ChannelUpdate1) Encode(w bytes.Buffer, pver uint32) error {	3✔
168	if err := WriteSig(w, a.Signature); err != nil {	3✔
169	return err	×
170	}	×
171
172	if err := WriteBytes(w, a.ChainHash[:]); err != nil {	3✔
173	return err	×
174	}	×
175
176	if err := WriteShortChannelID(w, a.ShortChannelID); err != nil {	3✔
177	return err	×
178	}	×
179
180	if err := WriteUint32(w, a.Timestamp); err != nil {	3✔
181	return err	×
182	}	×
183
184	if err := WriteChanUpdateMsgFlags(w, a.MessageFlags); err != nil {	3✔
185	return err	×
186	}	×
187
188	if err := WriteChanUpdateChanFlags(w, a.ChannelFlags); err != nil {	3✔
189	return err	×
190	}	×
191
192	if err := WriteUint16(w, a.TimeLockDelta); err != nil {	3✔
193	return err	×
194	}	×
195
196	if err := WriteMilliSatoshi(w, a.HtlcMinimumMsat); err != nil {	3✔
197	return err	×
198	}	×
199
200	if err := WriteUint32(w, a.BaseFee); err != nil {	3✔
201	return err	×
202	}	×
203
204	if err := WriteUint32(w, a.FeeRate); err != nil {	3✔
205	return err	×
206	}	×
207
208	// Now append optional fields if they are set. Currently, the only
209	// optional field is max HTLC.
210	if a.MessageFlags.HasMaxHtlc() {	6✔
211	err := WriteMilliSatoshi(w, a.HtlcMaximumMsat)	3✔
212	if err != nil {	3✔
213	return err	×
214	}	×
215	}
216
217	// Finally, append any extra opaque data.
218	return WriteBytes(w, a.ExtraOpaqueData)	3✔
219	}
220
221	// MsgType returns the integer uniquely identifying this message type on the
222	// wire.
223	//
224	// This is part of the lnwire.Message interface.
225	func (a *ChannelUpdate1) MsgType() MessageType {	3✔
226	return MsgChannelUpdate	3✔
227	}	3✔
228
229	// DataToSign is used to retrieve part of the announcement message which should
230	// be signed.
231	func (a *ChannelUpdate1) DataToSign() ([]byte, error) {	3✔
232	// We should not include the signatures itself.	3✔
233	b := make([]byte, 0, MaxMsgBody)	3✔
234	buf := bytes.NewBuffer(b)	3✔
235	if err := WriteBytes(buf, a.ChainHash[:]); err != nil {	3✔
236	return nil, err	×
237	}	×
238
239	if err := WriteShortChannelID(buf, a.ShortChannelID); err != nil {	3✔
240	return nil, err	×
241	}	×
242
243	if err := WriteUint32(buf, a.Timestamp); err != nil {	3✔
244	return nil, err	×
245	}	×
246
247	if err := WriteChanUpdateMsgFlags(buf, a.MessageFlags); err != nil {	3✔
248	return nil, err	×
249	}	×
250
251	if err := WriteChanUpdateChanFlags(buf, a.ChannelFlags); err != nil {	3✔
252	return nil, err	×
253	}	×
254
255	if err := WriteUint16(buf, a.TimeLockDelta); err != nil {	3✔
256	return nil, err	×
257	}	×
258
259	if err := WriteMilliSatoshi(buf, a.HtlcMinimumMsat); err != nil {	3✔
260	return nil, err	×
261	}	×
262
263	if err := WriteUint32(buf, a.BaseFee); err != nil {	3✔
264	return nil, err	×
265	}	×
266
267	if err := WriteUint32(buf, a.FeeRate); err != nil {	3✔
268	return nil, err	×
269	}	×
270
271	// Now append optional fields if they are set. Currently, the only
272	// optional field is max HTLC.
273	if a.MessageFlags.HasMaxHtlc() {	6✔
274	err := WriteMilliSatoshi(buf, a.HtlcMaximumMsat)	3✔
275	if err != nil {	3✔
276	return nil, err	×
277	}	×
278	}
279
280	// Finally, append any extra opaque data.
281	if err := WriteBytes(buf, a.ExtraOpaqueData); err != nil {	3✔
282	return nil, err	×
283	}	×
284
285	return buf.Bytes(), nil	3✔
286	}
287
288	// SCID returns the ShortChannelID of the channel that the update applies to.
289	//
290	// NOTE: this is part of the ChannelUpdate interface.
291	func (a *ChannelUpdate1) SCID() ShortChannelID {	×
292	return a.ShortChannelID	×
293	}	×
294
295	// IsNode1 is true if the update was produced by node 1 of the channel peers.
296	// Node 1 is the node with the lexicographically smaller public key.
297	//
298	// NOTE: this is part of the ChannelUpdate interface.
299	func (a *ChannelUpdate1) IsNode1() bool {	×
300	return a.ChannelFlags&ChanUpdateDirection == 0	×
301	}	×
302
303	// IsDisabled is true if the update is announcing that the channel should be
304	// considered disabled.
305	//
306	// NOTE: this is part of the ChannelUpdate interface.
307	func (a *ChannelUpdate1) IsDisabled() bool {	×
308	return a.ChannelFlags&ChanUpdateDisabled == ChanUpdateDisabled	×
309	}	×
310
311	// GetChainHash returns the hash of the chain that the message is referring to.
312	//
313	// NOTE: this is part of the ChannelUpdate interface.
314	func (a *ChannelUpdate1) GetChainHash() chainhash.Hash {	×
315	return a.ChainHash	×
316	}	×
317
318	// ForwardingPolicy returns the set of forwarding constraints of the update.
319	//
320	// NOTE: this is part of the ChannelUpdate interface.
321	func (a ChannelUpdate1) ForwardingPolicy() ForwardingPolicy {	×
322	return &ForwardingPolicy{	×
323	TimeLockDelta: a.TimeLockDelta,	×
324	BaseFee: MilliSatoshi(a.BaseFee),	×
325	FeeRate: MilliSatoshi(a.FeeRate),	×
326	MinHTLC: a.HtlcMinimumMsat,	×
327	HasMaxHTLC: a.MessageFlags.HasMaxHtlc(),	×
328	MaxHTLC: a.HtlcMaximumMsat,	×
329	}	×
330	}	×
331
332	// CmpAge can be used to determine if the update is older or newer than the
333	// passed update. It returns 1 if this update is newer, -1 if it is older, and
334	// 0 if they are the same age.
335	//
336	// NOTE: this is part of the ChannelUpdate interface.
337	func (a *ChannelUpdate1) CmpAge(update ChannelUpdate) (CompareResult, error) {	×
338	other, ok := update.(*ChannelUpdate1)	×
339	if !ok {	×
340	return 0, fmt.Errorf("expected *ChannelUpdate1, got: %T",	×
341	update)	×
342	}	×
343
344	switch {	×
345	case a.Timestamp > other.Timestamp:	×
346	return GreaterThan, nil	×
347	case a.Timestamp < other.Timestamp:	×
348	return LessThan, nil	×
349	default:	×
350	return EqualTo, nil	×
351	}
352	}
353
354	// SetDisabledFlag can be used to adjust the disabled flag of an update.
355	//
356	// NOTE: this is part of the ChannelUpdate interface.
357	func (a *ChannelUpdate1) SetDisabledFlag(disabled bool) {	×
358	if disabled {	×
359	a.ChannelFlags \|= ChanUpdateDisabled	×
360	} else {	×
361	a.ChannelFlags &= ^ChanUpdateDisabled	×
362	}	×
363	}
364
365	// SetSCID can be used to overwrite the SCID of the update.
366	//
367	// NOTE: this is part of the ChannelUpdate interface.
368	func (a *ChannelUpdate1) SetSCID(scid ShortChannelID) {	×
369	a.ShortChannelID = scid	×
370	}	×
371
372	// A compile time assertion to ensure ChannelUpdate1 implements the
373	// ChannelUpdate interface.
374	var _ ChannelUpdate = (*ChannelUpdate1)(nil)
375
376	// SerializedSize returns the serialized size of the message in bytes.
377	//
378	// This is part of the lnwire.SizeableMessage interface.
NEW 379	func (a *ChannelUpdate1) SerializedSize() (uint32, error) {	×
NEW 380	return MessageSerializedSize(a)	×
NEW 381	}	×
382
383	// A compile time check to ensure ChannelUpdate1 implements the TestMessage
384	// interface.
385	var _ TestMessage = (*ChannelUpdate1)(nil)
386
387	// RandTestMessage populates the message with random data suitable for testing.
388	// It uses the rapid testing framework to generate random values.
389	//
390	// This is part of the TestMessage interface.
NEW 391	func (a ChannelUpdate1) RandTestMessage(t rapid.T) Message {	×
NEW 392	// Generate random message flags	×
NEW 393	// Randomly decide whether to include max HTLC field	×
NEW 394	includeMaxHtlc := rapid.Bool().Draw(t, "includeMaxHtlc")	×
NEW 395	var msgFlags ChanUpdateMsgFlags	×
NEW 396	if includeMaxHtlc {	×
NEW 397	msgFlags \|= ChanUpdateRequiredMaxHtlc	×
NEW 398	}	×
399
400	// Generate random channel flags
401	// Randomly decide direction (node1 or node2)
NEW 402	isNode2 := rapid.Bool().Draw(t, "isNode2")	×
NEW 403	var chanFlags ChanUpdateChanFlags	×
NEW 404	if isNode2 {	×
NEW 405	chanFlags \|= ChanUpdateDirection	×
NEW 406	}	×
407
408	// Randomly decide if channel is disabled
NEW 409	isDisabled := rapid.Bool().Draw(t, "isDisabled")	×
NEW 410	if isDisabled {	×
NEW 411	chanFlags \|= ChanUpdateDisabled	×
NEW 412	}	×
413
414	// Generate chain hash
NEW 415	chainHash := RandChainHash(t)	×
NEW 416	var hash chainhash.Hash	×
NEW 417	copy(hash[:], chainHash[:])	×
NEW 418		×
NEW 419	// Generate other random fields	×
NEW 420	maxHtlc := MilliSatoshi(rapid.Uint64().Draw(t, "maxHtlc"))	×
NEW 421		×
NEW 422	// If max HTLC flag is not set, we need to zero the value	×
NEW 423	if !includeMaxHtlc {	×
NEW 424	maxHtlc = 0	×
NEW 425	}	×
426
NEW 427	return &ChannelUpdate1{	×
NEW 428	Signature: RandSignature(t),	×
NEW 429	ChainHash: hash,	×
NEW 430	ShortChannelID: RandShortChannelID(t),	×
NEW 431	Timestamp: uint32(rapid.IntRange(0, 0x7FFFFFFF).Draw(	×
NEW 432	t, "timestamp"),	×
NEW 433	),	×
NEW 434	MessageFlags: msgFlags,	×
NEW 435	ChannelFlags: chanFlags,	×
NEW 436	TimeLockDelta: uint16(rapid.IntRange(0, 65535).Draw(	×
NEW 437	t, "timelockDelta"),	×
NEW 438	),	×
NEW 439	HtlcMinimumMsat: MilliSatoshi(rapid.Uint64().Draw(	×
NEW 440	t, "htlcMinimum"),	×
NEW 441	),	×
NEW 442	BaseFee: uint32(rapid.IntRange(0, 0x7FFFFFFF).Draw(	×
NEW 443	t, "baseFee"),	×
NEW 444	),	×
NEW 445	FeeRate: uint32(rapid.IntRange(0, 0x7FFFFFFF).Draw(	×
NEW 446	t, "feeRate"),	×
NEW 447	),	×
NEW 448	HtlcMaximumMsat: maxHtlc,	×
NEW 449	ExtraOpaqueData: RandExtraOpaqueData(t, nil),	×
NEW 450	}	×
451	}

lightningnetwork / lnd / 13980885714

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous