lnwallet: handle nonce init in ProcessChanSyncMsg

In this commit, we update the logic to handle nonce init in
ProcessChanSyncMsg. Once a channel is already open, this is where we'll
get the new nonce data from the remote party we'll use to gain the nonce
we need to sign for their next state.
This commit is contained in:
Olaoluwa Osuntokun
2023-07-11 19:04:31 -07:00
parent 911becb431
commit 3270a29e3e
2 changed files with 34 additions and 11 deletions

View File

@@ -660,7 +660,6 @@ func (l *channelLink) syncChanStates() error {
return fmt.Errorf("unable to generate chan sync message for "+ return fmt.Errorf("unable to generate chan sync message for "+
"ChannelPoint(%v)", l.channel.ChannelPoint()) "ChannelPoint(%v)", l.channel.ChannelPoint())
} }
if err := l.cfg.Peer.SendMessage(true, localChanSyncMsg); err != nil { if err := l.cfg.Peer.SendMessage(true, localChanSyncMsg); err != nil {
return fmt.Errorf("unable to send chan sync message for "+ return fmt.Errorf("unable to send chan sync message for "+
"ChannelPoint(%v): %v", l.channel.ChannelPoint(), err) "ChannelPoint(%v): %v", l.channel.ChannelPoint(), err)
@@ -703,6 +702,13 @@ func (l *channelLink) syncChanStates() error {
l.ChanID(), nextRevocation, l.ChanID(), nextRevocation,
) )
// If this is a taproot channel, then we'll send the
// very same nonce that we sent above, as they should
// take the latest verification nonce we send.
if chanState.ChanType.IsTaproot() {
fundingLockedMsg.NextLocalNonce = localChanSyncMsg.LocalNonce
}
// For channels that negotiated the option-scid-alias // For channels that negotiated the option-scid-alias
// feature bit, ensure that we send over the alias in // feature bit, ensure that we send over the alias in
// the channel_ready message. We'll send the first // the channel_ready message. We'll send the first
@@ -1900,8 +1906,8 @@ func (l *channelLink) handleUpstreamMsg(msg lnwire.Message) {
// chain, validate this new commitment, closing the link if // chain, validate this new commitment, closing the link if
// invalid. // invalid.
err = l.channel.ReceiveNewCommitment(&lnwallet.CommitSigs{ err = l.channel.ReceiveNewCommitment(&lnwallet.CommitSigs{
CommitSig: msg.CommitSig, CommitSig: msg.CommitSig,
HtlcSigs: msg.HtlcSigs, HtlcSigs: msg.HtlcSigs,
}) })
if err != nil { if err != nil {
// If we were unable to reconstruct their proposed // If we were unable to reconstruct their proposed
@@ -2265,9 +2271,9 @@ func (l *channelLink) updateCommitTx() error {
} }
commitSig := &lnwire.CommitSig{ commitSig := &lnwire.CommitSig{
ChanID: l.ChanID(), ChanID: l.ChanID(),
CommitSig: newCommit.CommitSig, CommitSig: newCommit.CommitSig,
HtlcSigs: newCommit.HtlcSigs, HtlcSigs: newCommit.HtlcSigs,
} }
l.cfg.Peer.SendMessage(false, commitSig) l.cfg.Peer.SendMessage(false, commitSig)

View File

@@ -4092,8 +4092,6 @@ func (lc *LightningChannel) ProcessChanSyncMsg(
msg *lnwire.ChannelReestablish) ([]lnwire.Message, []models.CircuitKey, msg *lnwire.ChannelReestablish) ([]lnwire.Message, []models.CircuitKey,
[]models.CircuitKey, error) { []models.CircuitKey, error) {
// TODO(roasbeef): need to replace w/ received nonces
// Now we'll examine the state we have, vs what was contained in the // Now we'll examine the state we have, vs what was contained in the
// chain sync message. If we're de-synchronized, then we'll send a // chain sync message. If we're de-synchronized, then we'll send a
// batch of messages which when applied will kick start the chain // batch of messages which when applied will kick start the chain
@@ -4133,6 +4131,25 @@ func (lc *LightningChannel) ProcessChanSyncMsg(
} }
} }
// If this is a taproot channel, then we expect that the remote party
// has sent the next verification nonce. If they haven't, then we'll
// bail out, otherwise we'll init our local session then continue as
// normal.
switch {
case lc.channelState.ChanType.IsTaproot() && msg.LocalNonce == nil:
return nil, nil, nil, fmt.Errorf("remote verification nonce " +
"not sent")
case lc.channelState.ChanType.IsTaproot() && msg.LocalNonce != nil:
err := lc.InitRemoteMusigNonces(&musig2.Nonces{
PubNonce: *msg.LocalNonce,
})
if err != nil {
return nil, nil, nil, fmt.Errorf("unable to init "+
"remote nonce: %w", err)
}
}
// If we detect that this is is a restored channel, then we can skip a // If we detect that this is is a restored channel, then we can skip a
// portion of the verification, as we already know that we're unable to // portion of the verification, as we already know that we're unable to
// proceed with any updates. // proceed with any updates.
@@ -4210,12 +4227,12 @@ func (lc *LightningChannel) ProcessChanSyncMsg(
"while we have %v, we owe them a revocation", "while we have %v, we owe them a revocation",
msg.RemoteCommitTailHeight, localTailHeight) msg.RemoteCommitTailHeight, localTailHeight)
revocationMsg, err := lc.generateRevocation( heightToRetransmit := localTailHeight - 1
localTailHeight - 1, revocationMsg, err := lc.generateRevocation(heightToRetransmit)
)
if err != nil { if err != nil {
return nil, nil, nil, err return nil, nil, nil, err
} }
updates = append(updates, revocationMsg) updates = append(updates, revocationMsg)
// Next, as a precaution, we'll check a special edge case. If // Next, as a precaution, we'll check a special edge case. If