Skip to content

Commit b893b6c

Browse files
dunxendunxen
committed
f wip: move flags to FundingNegotiated state and make FundingNegotiated a Funded state
1 parent 6cf6994 commit b893b6c

File tree

2 files changed

+50
-44
lines changed

2 files changed

+50
-44
lines changed

‎lightning/src/ln/channel.rs‎

Lines changed: 43 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -434,12 +434,6 @@ macro_rules! define_state_flags {
434434
is_remote_shutdown_sent, set_remote_shutdown_sent, clear_remote_shutdown_sent);
435435
define_state_flags!($flag_type, FundedStateFlags::LOCAL_SHUTDOWN_SENT,
436436
is_local_shutdown_sent, set_local_shutdown_sent, clear_local_shutdown_sent);
437-
define_state_flags!($flag_type, FundedStateFlags::INTERACTIVE_SIGNING,
438-
is_interactive_signing, set_interactive_signing, clear_interactive_signing);
439-
define_state_flags!($flag_type, FundedStateFlags::THEIR_TX_SIGNATURES_SENT,
440-
is_their_tx_signatures_sent, set_their_tx_signatures_sent, clear_their_tx_signatures_sent);
441-
define_state_flags!($flag_type, FundedStateFlags::OUR_TX_SIGNATURES_SENT,
442-
is_our_tx_signatures_sent, set_our_tx_signatures_sent, clear_our_tx_signatures_sent);
443437

444438
impl core::ops::BitOr<FundedStateFlags> for $flag_type {
445439
type Output = Self;
@@ -506,16 +500,7 @@ define_state_flags!(
506500
is_remote_shutdown_sent, set_remote_shutdown_sent, clear_remote_shutdown_sent),
507501
("Indicates we sent a `shutdown` message. At this point, we may not add any new HTLCs to \
508502
the channel.", LOCAL_SHUTDOWN_SENT, state_flags::LOCAL_SHUTDOWN_SENT,
509-
is_local_shutdown_sent, set_local_shutdown_sent, clear_local_shutdown_sent),
510-
("Indicates we have an active interactive signing session for an interactive transaction",
511-
INTERACTIVE_SIGNING, state_flags::INTERACTIVE_SIGNING,
512-
is_interactive_signing, set_interactive_signing, clear_interactive_singing),
513-
("Indicates they sent us a `tx_signatures` message.",
514-
THEIR_TX_SIGNATURES_SENT, state_flags::THEIR_TX_SIGNATURES_SENT,
515-
is_their_tx_signatures_sent, set_their_tx_signatures_sent, clear_their_tx_signatures_sent),
516-
("Indicates we sent them a `tx_signatures` message.",
517-
OUR_TX_SIGNATURES_SENT, state_flags::OUR_TX_SIGNATURES_SENT,
518-
is_our_tx_signatures_sent, set_our_tx_signatures_sent, clear_our_tx_signatures_sent)
503+
is_local_shutdown_sent, set_local_shutdown_sent, clear_local_shutdown_sent)
519504
]
520505
);
521506

@@ -529,6 +514,21 @@ define_state_flags!(
529514
]
530515
);
531516

517+
define_state_flags!(
518+
"Flags that only apply to [`ChannelState::FundingNegotiated`].",
519+
FUNDED_STATE, FundingNegotiatedFlags, [
520+
("Indicates we have an active interactive signing session for an interactive transaction",
521+
INTERACTIVE_SIGNING, state_flags::INTERACTIVE_SIGNING,
522+
is_interactive_signing, set_interactive_signing, clear_interactive_signing),
523+
("Indicates they sent us a `tx_signatures` message.",
524+
THEIR_TX_SIGNATURES_SENT, state_flags::THEIR_TX_SIGNATURES_SENT,
525+
is_their_tx_signatures_sent, set_their_tx_signatures_sent, clear_their_tx_signatures_sent),
526+
("Indicates we sent them a `tx_signatures` message.",
527+
OUR_TX_SIGNATURES_SENT, state_flags::OUR_TX_SIGNATURES_SENT,
528+
is_our_tx_signatures_sent, set_our_tx_signatures_sent, clear_our_tx_signatures_sent)
529+
]
530+
);
531+
532532
define_state_flags!(
533533
"Flags that only apply to [`ChannelState::AwaitingChannelReady`].",
534534
FUNDED_STATE, AwaitingChannelReadyFlags, [
@@ -588,7 +588,7 @@ enum ChannelState {
588588
/// `AwaitingChannelReady`. Note that this is nonsense for an inbound channel as we immediately generate
589589
/// `funding_signed` upon receipt of `funding_created`, so simply skip this state. For dual-funded (V2)
590590
/// channels, this state is also skipped.
591-
FundingNegotiated,
591+
FundingNegotiated(FundingNegotiatedFlags),
592592
/// We've received/sent `funding_created` and `funding_signed` and are thus now waiting on the
593593
/// funding transaction to confirm.
594594
AwaitingChannelReady(AwaitingChannelReadyFlags),
@@ -631,7 +631,7 @@ macro_rules! impl_state_flag {
631631
}
632632
};
633633
($get: ident, $set: ident, $clear: ident, FUNDED_STATES) => {
634-
impl_state_flag!($get, $set, $clear, [AwaitingChannelReady, ChannelReady]);
634+
impl_state_flag!($get, $set, $clear, [FundingNegotiated, AwaitingChannelReady, ChannelReady]);
635635
};
636636
($get: ident, $set: ident, $clear: ident, $state: ident) => {
637637
impl_state_flag!($get, $set, $clear, [$state]);
@@ -641,10 +641,12 @@ macro_rules! impl_state_flag {
641641
impl ChannelState {
642642
fn from_u32(state: u32) -> Result<Self, ()> {
643643
match state {
644-
state_flags::FUNDING_NEGOTIATED => Ok(ChannelState::FundingNegotiated),
645644
state_flags::SHUTDOWN_COMPLETE => Ok(ChannelState::ShutdownComplete),
646645
val => {
647-
if val & state_flags::AWAITING_CHANNEL_READY == state_flags::AWAITING_CHANNEL_READY {
646+
if val & state_flags::FUNDING_NEGOTIATED == state_flags::FUNDING_NEGOTIATED {
647+
FundingNegotiatedFlags::from_u32(val & !state_flags::FUNDING_NEGOTIATED)
648+
.map(|flags| ChannelState::FundingNegotiated(flags))
649+
} else if val & state_flags::AWAITING_CHANNEL_READY == state_flags::AWAITING_CHANNEL_READY {
648650
AwaitingChannelReadyFlags::from_u32(val & !state_flags::AWAITING_CHANNEL_READY)
649651
.map(|flags| ChannelState::AwaitingChannelReady(flags))
650652
} else if val & state_flags::CHANNEL_READY == state_flags::CHANNEL_READY {
@@ -662,15 +664,15 @@ impl ChannelState {
662664
fn to_u32(self) -> u32 {
663665
match self {
664666
ChannelState::NegotiatingFunding(flags) => flags.0,
665-
ChannelState::FundingNegotiated => state_flags::FUNDING_NEGOTIATED,
667+
ChannelState::FundingNegotiated(flags) => state_flags::FUNDING_NEGOTIATED | flags.0,
666668
ChannelState::AwaitingChannelReady(flags) => state_flags::AWAITING_CHANNEL_READY | flags.0,
667669
ChannelState::ChannelReady(flags) => state_flags::CHANNEL_READY | flags.0,
668670
ChannelState::ShutdownComplete => state_flags::SHUTDOWN_COMPLETE,
669671
}
670672
}
671673

672674
fn is_pre_funded_state(&self) -> bool {
673-
matches!(self, ChannelState::NegotiatingFunding(_)|ChannelState::FundingNegotiated)
675+
matches!(self, ChannelState::NegotiatingFunding(_)|ChannelState::FundingNegotiated(_))
674676
}
675677

676678
fn is_both_sides_shutdown(&self) -> bool {
@@ -704,9 +706,9 @@ impl ChannelState {
704706
impl_state_flag!(is_monitor_update_in_progress, set_monitor_update_in_progress, clear_monitor_update_in_progress, FUNDED_STATES);
705707
impl_state_flag!(is_local_shutdown_sent, set_local_shutdown_sent, clear_local_shutdown_sent, FUNDED_STATES);
706708
impl_state_flag!(is_remote_shutdown_sent, set_remote_shutdown_sent, clear_remote_shutdown_sent, FUNDED_STATES);
707-
impl_state_flag!(is_interactive_signing, set_interactive_signing, clear_interactive_signing, FUNDED_STATES);
708-
impl_state_flag!(is_our_tx_signatures_sent, set_our_tx_signatures_sent, clear_our_tx_signatures_sent, FUNDED_STATES);
709-
impl_state_flag!(is_their_tx_signatures_sent, set_their_tx_signatures_sent, clear_their_tx_signatures_sent, FUNDED_STATES);
709+
impl_state_flag!(is_interactive_signing, set_interactive_signing, clear_interactive_signing, FundingNegotiated);
710+
impl_state_flag!(is_our_tx_signatures_sent, set_our_tx_signatures_sent, clear_our_tx_signatures_sent, FundingNegotiated);
711+
impl_state_flag!(is_their_tx_signatures_sent, set_their_tx_signatures_sent, clear_their_tx_signatures_sent, FundingNegotiated);
710712
impl_state_flag!(is_our_channel_ready, set_our_channel_ready, clear_our_channel_ready, AwaitingChannelReady);
711713
impl_state_flag!(is_their_channel_ready, set_their_channel_ready, clear_their_channel_ready, AwaitingChannelReady);
712714
impl_state_flag!(is_waiting_for_batch, set_waiting_for_batch, clear_waiting_for_batch, AwaitingChannelReady);
@@ -2375,7 +2377,7 @@ impl<SP: Deref> PendingV2Channel<SP> where SP::Target: SignerProvider {
23752377
)));
23762378
};
23772379

2378-
let mut channel_state = ChannelState::AwaitingChannelReady(AwaitingChannelReadyFlags::new());
2380+
let mut channel_state = ChannelState::FundingNegotiated(FundingNegotiatedFlags::new());
23792381
channel_state.set_interactive_signing();
23802382
self.context.channel_state = channel_state;
23812383

@@ -3468,7 +3470,7 @@ impl<SP: Deref> ChannelContext<SP> where SP::Target: SignerProvider {
34683470

34693471
fn unset_funding_info(&mut self, funding: &mut FundingScope) {
34703472
debug_assert!(
3471-
matches!(self.channel_state, ChannelState::FundingNegotiated)
3473+
matches!(self.channel_state, ChannelState::FundingNegotiated(_))
34723474
|| matches!(self.channel_state, ChannelState::AwaitingChannelReady(_))
34733475
);
34743476
funding.channel_transaction_parameters.funding_outpoint = None;
@@ -4460,11 +4462,10 @@ impl<SP: Deref> ChannelContext<SP> where SP::Target: SignerProvider {
44604462

44614463
fn if_unbroadcasted_funding<F, O>(&self, f: F) -> Option<O> where F: Fn() -> Option<O> {
44624464
match self.channel_state {
4463-
ChannelState::FundingNegotiated => f(),
4465+
ChannelState::FundingNegotiated(_) => f(),
44644466
ChannelState::AwaitingChannelReady(flags) =>
44654467
if flags.is_set(AwaitingChannelReadyFlags::WAITING_FOR_BATCH) ||
4466-
flags.is_set(FundedStateFlags::MONITOR_UPDATE_IN_PROGRESS.into()) ||
4467-
(flags.is_set(FundedStateFlags::THEIR_TX_SIGNATURES_SENT.into()) && flags.is_set(FundedStateFlags::OUR_TX_SIGNATURES_SENT.into()))
4468+
flags.is_set(FundedStateFlags::MONITOR_UPDATE_IN_PROGRESS.into())
44684469
{
44694470
f()
44704471
} else {
@@ -5671,7 +5672,7 @@ impl<SP: Deref> FundedChannel<SP> where
56715672
) -> Result<ChannelMonitor<<SP::Target as SignerProvider>::EcdsaSigner>, ChannelError>
56725673
where L::Target: Logger
56735674
{
5674-
if !matches!(self.context.channel_state, ChannelState::AwaitingChannelReady(_)) {
5675+
if !matches!(self.context.channel_state, ChannelState::FundingNegotiated(_)) {
56755676
return Err(ChannelError::Close(
56765677
(
56775678
"Received initial commitment_signed before funding transaction constructed!".to_owned(),
@@ -6277,7 +6278,7 @@ impl<SP: Deref> FundedChannel<SP> where
62776278
pub fn tx_signatures<L: Deref>(&mut self, msg: &msgs::TxSignatures, logger: &L) -> Result<Option<msgs::TxSignatures>, ChannelError>
62786279
where L::Target: Logger
62796280
{
6280-
if !matches!(self.context.channel_state, ChannelState::AwaitingChannelReady(flags) if flags.is_interactive_signing()) {
6281+
if !matches!(self.context.channel_state, ChannelState::FundingNegotiated(flags) if flags.is_interactive_signing()) {
62816282
return Err(ChannelError::close("Received tx_signatures in strange state!".to_owned()));
62826283
}
62836284

@@ -6334,6 +6335,10 @@ impl<SP: Deref> FundedChannel<SP> where
63346335
self.context.channel_state.set_our_tx_signatures_sent();
63356336
}
63366337

6338+
self.context.channel_state = ChannelState::AwaitingChannelReady(AwaitingChannelReadyFlags::new());
6339+
let need_channel_ready = self.check_get_channel_ready(0, logger).is_some()
6340+
|| self.context.signer_pending_channel_ready;
6341+
self.monitor_updating_paused(false, false, need_channel_ready, Vec::new(), Vec::new(), Vec::new());
63376342
Ok(holder_tx_signatures_opt)
63386343
} else {
63396344
Err(ChannelError::Close((
@@ -6556,7 +6561,7 @@ impl<SP: Deref> FundedChannel<SP> where
65566561
// first received the funding_signed.
65576562
let mut funding_broadcastable = None;
65586563
if let Some(funding_transaction) = &self.funding.funding_transaction {
6559-
if (self.funding.is_outbound() || (self.is_v2_established() && self.context.channel_state.is_their_tx_signatures_sent())) &&
6564+
if (self.funding.is_outbound() || self.is_v2_established()) &&
65606565
(matches!(self.context.channel_state, ChannelState::AwaitingChannelReady(flags) if !flags.is_set(AwaitingChannelReadyFlags::WAITING_FOR_BATCH)) ||
65616566
matches!(self.context.channel_state, ChannelState::ChannelReady(_)))
65626567
{
@@ -9441,7 +9446,7 @@ impl<SP: Deref> OutboundV1Channel<SP> where SP::Target: SignerProvider {
94419446

94429447
// Now that we're past error-generating stuff, update our local state:
94439448

9444-
self.context.channel_state = ChannelState::FundingNegotiated;
9449+
self.context.channel_state = ChannelState::FundingNegotiated(FundingNegotiatedFlags::new());
94459450
self.context.channel_id = ChannelId::v1_from_funding_outpoint(funding_txo);
94469451

94479452
// If the funding transaction is a coinbase transaction, we need to set the minimum depth to 100.
@@ -9558,7 +9563,7 @@ impl<SP: Deref> OutboundV1Channel<SP> where SP::Target: SignerProvider {
95589563
if !self.funding.is_outbound() {
95599564
return Err((self, ChannelError::close("Received funding_signed for an inbound channel?".to_owned())));
95609565
}
9561-
if !matches!(self.context.channel_state, ChannelState::FundingNegotiated) {
9566+
if !matches!(self.context.channel_state, ChannelState::FundingNegotiated(_)) {
95629567
return Err((self, ChannelError::close("Received funding_signed in strange state!".to_owned())));
95639568
}
95649569
let mut holder_commitment_point = match self.unfunded_context.holder_commitment_point {
@@ -11299,11 +11304,12 @@ mod tests {
1129911304
#[test]
1130011305
fn test_channel_state_order() {
1130111306
use crate::ln::channel::NegotiatingFundingFlags;
11307+
use crate::ln::channel::FundingNegotiatedFlags;
1130211308
use crate::ln::channel::AwaitingChannelReadyFlags;
1130311309
use crate::ln::channel::ChannelReadyFlags;
1130411310

11305-
assert!(ChannelState::NegotiatingFunding(NegotiatingFundingFlags::new()) < ChannelState::FundingNegotiated);
11306-
assert!(ChannelState::FundingNegotiated < ChannelState::AwaitingChannelReady(AwaitingChannelReadyFlags::new()));
11311+
assert!(ChannelState::NegotiatingFunding(NegotiatingFundingFlags::new()) < ChannelState::FundingNegotiated(FundingNegotiatedFlags::new()));
11312+
assert!(ChannelState::FundingNegotiated(FundingNegotiatedFlags::new()) < ChannelState::AwaitingChannelReady(AwaitingChannelReadyFlags::new()));
1130711313
assert!(ChannelState::AwaitingChannelReady(AwaitingChannelReadyFlags::new()) < ChannelState::ChannelReady(ChannelReadyFlags::new()));
1130811314
assert!(ChannelState::ChannelReady(ChannelReadyFlags::new()) < ChannelState::ShutdownComplete);
1130911315
}

‎lightning/src/ln/dual_funding_tests.rs‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -208,13 +208,6 @@ fn do_test_v2_channel_establishment(session: V2ChannelEstablishmentTestSession)
208208
assert!(events.is_empty());
209209
nodes[1].chain_monitor.complete_sole_pending_chan_update(&channel_id);
210210

211-
let events = nodes[1].node.get_and_clear_pending_events();
212-
assert_eq!(events.len(), 1);
213-
match events[0] {
214-
Event::ChannelPending { channel_id: chan_id, .. } => assert_eq!(chan_id, channel_id),
215-
_ => panic!("Unexpected event"),
216-
};
217-
218211
let tx_signatures_msg = get_event_msg!(
219212
nodes[1],
220213
MessageSendEvent::SendTxSignatures,
@@ -236,6 +229,13 @@ fn do_test_v2_channel_establishment(session: V2ChannelEstablishmentTestSession)
236229
},
237230
);
238231

232+
let events = nodes[1].node.get_and_clear_pending_events();
233+
assert_eq!(events.len(), 1);
234+
match events[0] {
235+
Event::ChannelPending { channel_id: chan_id, .. } => assert_eq!(chan_id, channel_id),
236+
_ => panic!("Unexpected event"),
237+
};
238+
239239
// For an inbound channel V2 channel the transaction should be broadcast once receiving a
240240
// tx_signature and applying local tx_signatures:
241241
let broadcasted_txs = nodes[1].tx_broadcaster.txn_broadcast();

0 commit comments

Comments
 (0)