diff --git a/contractcourt/chain_watcher.go b/contractcourt/chain_watcher.go index ef4a4e200..ed34b5fa4 100644 --- a/contractcourt/chain_watcher.go +++ b/contractcourt/chain_watcher.go @@ -436,6 +436,7 @@ func (c *chainWatcher) handleUnknownLocalState( return s.FetchLeavesFromCommit( lnwallet.NewAuxChanState(c.cfg.chanState), c.cfg.chanState.LocalCommitment, *commitKeyRing, + lntypes.Local, ) }, ).Unpack() diff --git a/lnwallet/aux_leaf_store.go b/lnwallet/aux_leaf_store.go index 4558c2f81..c457a9250 100644 --- a/lnwallet/aux_leaf_store.go +++ b/lnwallet/aux_leaf_store.go @@ -178,7 +178,8 @@ type AuxLeafStore interface { // commitment. FetchLeavesFromCommit(chanState AuxChanState, commit channeldb.ChannelCommitment, - keyRing CommitmentKeyRing) fn.Result[CommitDiffAuxResult] + keyRing CommitmentKeyRing, whoseCommit lntypes.ChannelParty, + ) fn.Result[CommitDiffAuxResult] // FetchLeavesFromRevocation attempts to fetch the auxiliary leaves // from a channel revocation that stores balance + blob information. diff --git a/lnwallet/channel.go b/lnwallet/channel.go index 67cd7b947..ace6a8adc 100644 --- a/lnwallet/channel.go +++ b/lnwallet/channel.go @@ -666,6 +666,7 @@ func (lc *LightningChannel) diskCommitToMemCommit( return s.FetchLeavesFromCommit( NewAuxChanState(lc.channelState), *diskCommit, *commitKeys.GetForParty(whoseCommit), + whoseCommit, ) }, ).Unpack() @@ -1798,7 +1799,7 @@ func (lc *LightningChannel) restorePendingLocalUpdates( func(s AuxLeafStore) fn.Result[CommitDiffAuxResult] { return s.FetchLeavesFromCommit( NewAuxChanState(lc.channelState), pendingCommit, - *pendingRemoteKeys, + *pendingRemoteKeys, lntypes.Remote, ) }, ).Unpack() @@ -3219,7 +3220,7 @@ func genRemoteHtlcSigJobs(keyRing *CommitmentKeyRing, leafStore, func(s AuxLeafStore) fn.Result[CommitDiffAuxResult] { return s.FetchLeavesFromCommit( NewAuxChanState(chanState), *diskCommit, - *keyRing, + *keyRing, lntypes.Remote, ) }, ).Unpack() @@ -4764,7 +4765,7 @@ func genHtlcSigValidationJobs(chanState *channeldb.OpenChannel, leafStore, func(s AuxLeafStore) fn.Result[CommitDiffAuxResult] { return s.FetchLeavesFromCommit( NewAuxChanState(chanState), *diskCommit, - *keyRing, + *keyRing, lntypes.Local, ) }, ).Unpack() @@ -6668,7 +6669,7 @@ func NewUnilateralCloseSummary(chanState *channeldb.OpenChannel, //nolint:funlen leafStore, func(s AuxLeafStore) fn.Result[CommitDiffAuxResult] { return s.FetchLeavesFromCommit( NewAuxChanState(chanState), remoteCommit, - *keyRing, + *keyRing, lntypes.Remote, ) }, ).Unpack() @@ -7890,6 +7891,7 @@ func NewLocalForceCloseSummary(chanState *channeldb.OpenChannel, return s.FetchLeavesFromCommit( NewAuxChanState(chanState), chanState.LocalCommitment, *keyRing, + lntypes.Local, ) }, ).Unpack() diff --git a/lnwallet/commitment.go b/lnwallet/commitment.go index 170efece1..3f47819ea 100644 --- a/lnwallet/commitment.go +++ b/lnwallet/commitment.go @@ -1307,7 +1307,7 @@ func findOutputIndexesFromRemote(revocationPreimage *chainhash.Hash, leafStore, func(a AuxLeafStore) fn.Result[CommitDiffAuxResult] { return a.FetchLeavesFromCommit( NewAuxChanState(chanState), chanCommit, - *keyRing, + *keyRing, lntypes.Remote, ) }, ).Unpack() diff --git a/lnwallet/mock.go b/lnwallet/mock.go index e588c91c0..16f2d464b 100644 --- a/lnwallet/mock.go +++ b/lnwallet/mock.go @@ -19,6 +19,7 @@ import ( "github.com/lightningnetwork/lnd/chainntnfs" "github.com/lightningnetwork/lnd/channeldb" "github.com/lightningnetwork/lnd/fn" + "github.com/lightningnetwork/lnd/lntypes" "github.com/lightningnetwork/lnd/lnwallet/chainfee" "github.com/lightningnetwork/lnd/tlv" "github.com/stretchr/testify/mock" @@ -412,8 +413,8 @@ func (*MockAuxLeafStore) FetchLeavesFromView( // correspond to the passed aux blob, and an existing channel // commitment. func (*MockAuxLeafStore) FetchLeavesFromCommit(_ AuxChanState, - _ channeldb.ChannelCommitment, - _ CommitmentKeyRing) fn.Result[CommitDiffAuxResult] { + _ channeldb.ChannelCommitment, _ CommitmentKeyRing, + _ lntypes.ChannelParty) fn.Result[CommitDiffAuxResult] { return fn.Ok(CommitDiffAuxResult{}) }