mirror of
https://github.com/lightningnetwork/lnd.git
synced 2026-08-13 12:32:48 +02:00
b720fd5cd5
Copy all HTLC fields when cloning channel commitment state. The old copy method only copied a subset of scalar fields and copied into nil slices for Signature and ExtraData. Allocate those slices and deep-copy custom record values so snapshots and channel copies retain complete HTLC metadata.
67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package chanstate
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/btcsuite/btcd/btcec/v2"
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
|
"github.com/lightningnetwork/lnd/tlv"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// TestHTLCCopy asserts that copying an HTLC produces an independent deep copy.
|
|
func TestHTLCCopy(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
_, blindingPoint := btcec.PrivKeyFromBytes(bytes.Repeat([]byte{1}, 32))
|
|
|
|
var rHash [32]byte
|
|
copy(rHash[:], bytes.Repeat([]byte{2}, len(rHash)))
|
|
|
|
var onionBlob [lnwire.OnionPacketSize]byte
|
|
copy(onionBlob[:], bytes.Repeat([]byte{3}, len(onionBlob)))
|
|
|
|
htlc := HTLC{
|
|
Signature: []byte{4, 5, 6},
|
|
RHash: rHash,
|
|
Amt: 1000,
|
|
RefundTimeout: 144,
|
|
OutputIndex: 3,
|
|
Incoming: true,
|
|
OnionBlob: onionBlob,
|
|
HtlcIndex: 42,
|
|
LogIndex: 43,
|
|
ExtraData: lnwire.ExtraOpaqueData{7, 8, 9},
|
|
BlindingPoint: tlv.SomeRecordT(
|
|
tlv.NewPrimitiveRecord[lnwire.BlindingPointTlvType](
|
|
blindingPoint,
|
|
),
|
|
),
|
|
CustomRecords: lnwire.CustomRecords{
|
|
lnwire.MinCustomRecordsTlvType: []byte{10, 11, 12},
|
|
},
|
|
}
|
|
|
|
clone := htlc.Copy()
|
|
require.Equal(t, htlc, clone)
|
|
|
|
clone.Signature[0] = 0
|
|
require.Equal(t, byte(4), htlc.Signature[0])
|
|
|
|
clone.ExtraData[0] = 0
|
|
require.Equal(t, byte(7), htlc.ExtraData[0])
|
|
|
|
clone.CustomRecords[lnwire.MinCustomRecordsTlvType] = []byte{0}
|
|
require.Equal(
|
|
t, []byte{10, 11, 12},
|
|
htlc.CustomRecords[lnwire.MinCustomRecordsTlvType],
|
|
)
|
|
|
|
clone = htlc.Copy()
|
|
clone.CustomRecords[lnwire.MinCustomRecordsTlvType][0] = 0
|
|
require.Equal(
|
|
t, []byte{10, 11, 12},
|
|
htlc.CustomRecords[lnwire.MinCustomRecordsTlvType],
|
|
)
|
|
}
|