Skip to content

Commit fe2d309

Browse files
committed
Add ChannelReady event
This adds a `ChannelReady` event that is emitted as soon as a new channel becomes usable, i.e., after both sides have sent `channel_ready`.
1 parent ad7ff0b commit fe2d309

File tree

11 files changed

+212
-11
lines changed

11 files changed

+212
-11
lines changed

fuzz/src/chanmon_consistency.rs

+1
Original file line numberDiff line numberDiff line change
@@ -872,6 +872,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out) {
872872
// looking like probes.
873873
},
874874
events::Event::PaymentForwarded { .. } if $node == 1 => {},
875+
events::Event::ChannelReady { .. } => {},
875876
events::Event::PendingHTLCsForwardable { .. } => {
876877
nodes[$node].process_pending_htlc_forwards();
877878
},

lightning-background-processor/src/lib.rs

+13-5
Original file line numberDiff line numberDiff line change
@@ -988,9 +988,12 @@ mod tests {
988988

989989
// Set up a background event handler for FundingGenerationReady events.
990990
let (sender, receiver) = std::sync::mpsc::sync_channel(1);
991-
let event_handler = move |event: &Event| {
992-
sender.send(handle_funding_generation_ready!(event, channel_value)).unwrap();
991+
let event_handler = move |event: &Event| match event {
992+
Event::FundingGenerationReady{ .. } => sender.send(handle_funding_generation_ready!(event, channel_value)).unwrap(),
993+
Event::ChannelReady { .. } => {},
994+
_ => panic!("Unexpected event: {:?}", event),
993995
};
996+
994997
let bg_processor = BackgroundProcessor::start(persister, event_handler, nodes[0].chain_monitor.clone(), nodes[0].node.clone(), nodes[0].no_gossip_sync(), nodes[0].peer_manager.clone(), nodes[0].logger.clone(), Some(nodes[0].scorer.clone()));
995998

996999
// Open a channel and check that the FundingGenerationReady event was handled.
@@ -1014,20 +1017,25 @@ mod tests {
10141017

10151018
// Set up a background event handler for SpendableOutputs events.
10161019
let (sender, receiver) = std::sync::mpsc::sync_channel(1);
1017-
let event_handler = move |event: &Event| sender.send(event.clone()).unwrap();
1020+
let event_handler = move |event: &Event| match event {
1021+
Event::SpendableOutputs { .. } => sender.send(event.clone()).unwrap(),
1022+
Event::ChannelReady { .. } => {},
1023+
Event::ChannelClosed { .. } => {},
1024+
_ => panic!("Unexpected event: {:?}", event),
1025+
};
10181026
let persister = Arc::new(Persister::new(data_dir));
10191027
let bg_processor = BackgroundProcessor::start(persister, event_handler, nodes[0].chain_monitor.clone(), nodes[0].node.clone(), nodes[0].no_gossip_sync(), nodes[0].peer_manager.clone(), nodes[0].logger.clone(), Some(nodes[0].scorer.clone()));
10201028

10211029
// Force close the channel and check that the SpendableOutputs event was handled.
10221030
nodes[0].node.force_close_broadcasting_latest_txn(&nodes[0].node.list_channels()[0].channel_id, &nodes[1].node.get_our_node_id()).unwrap();
10231031
let commitment_tx = nodes[0].tx_broadcaster.txn_broadcasted.lock().unwrap().pop().unwrap();
10241032
confirm_transaction_depth(&mut nodes[0], &commitment_tx, BREAKDOWN_TIMEOUT as u32);
1033+
10251034
let event = receiver
10261035
.recv_timeout(Duration::from_secs(EVENT_DEADLINE))
1027-
.expect("SpendableOutputs not handled within deadline");
1036+
.expect("Events not handled within deadline");
10281037
match event {
10291038
Event::SpendableOutputs { .. } => {},
1030-
Event::ChannelClosed { .. } => {},
10311039
_ => panic!("Unexpected event: {:?}", event),
10321040
}
10331041

lightning-invoice/src/utils.rs

+5
Original file line numberDiff line numberDiff line change
@@ -869,6 +869,9 @@ mod test {
869869
get_event_msg!(nodes[2], MessageSendEvent::SendChannelUpdate, nodes[0].node.get_our_node_id());
870870
nodes[0].node.handle_channel_ready(&nodes[2].node.get_our_node_id(), &as_channel_ready);
871871
get_event_msg!(nodes[0], MessageSendEvent::SendChannelUpdate, nodes[2].node.get_our_node_id());
872+
873+
expect_channel_ready_event!(nodes[0], nodes[2].node.get_our_node_id());
874+
expect_channel_ready_event!(nodes[2], nodes[0].node.get_our_node_id());
872875

873876
// As `msgs::ChannelUpdate` was never handled for the participating node(s) of the second
874877
// channel, the channel will never be assigned any `counterparty.forwarding_info`.
@@ -1257,6 +1260,8 @@ mod test {
12571260
get_event_msg!(nodes[1], MessageSendEvent::SendChannelUpdate, nodes[3].node.get_our_node_id());
12581261
nodes[3].node.handle_channel_ready(&nodes[1].node.get_our_node_id(), &as_channel_ready);
12591262
get_event_msg!(nodes[3], MessageSendEvent::SendChannelUpdate, nodes[1].node.get_our_node_id());
1263+
expect_channel_ready_event!(nodes[1], nodes[3].node.get_our_node_id());
1264+
expect_channel_ready_event!(nodes[3], nodes[1].node.get_our_node_id());
12601265

12611266
// As `msgs::ChannelUpdate` was never handled for the participating node(s) of the third
12621267
// channel, the channel will never be assigned any `counterparty.forwarding_info`.

lightning/src/ln/chanmon_update_fail_tests.rs

+2
Original file line numberDiff line numberDiff line change
@@ -1915,6 +1915,8 @@ fn do_during_funding_monitor_fail(confirm_a_first: bool, restore_b_before_conf:
19151915
node.gossip_sync.handle_channel_update(&as_update).unwrap();
19161916
node.gossip_sync.handle_channel_update(&bs_update).unwrap();
19171917
}
1918+
expect_channel_ready_event!(nodes[1], nodes[0].node.get_our_node_id());
1919+
expect_channel_ready_event!(nodes[0], nodes[1].node.get_our_node_id());
19181920

19191921
send_payment(&nodes[0], &[&nodes[1]], 8000000);
19201922
close_channel(&nodes[0], &nodes[1], &channel_id, funding_tx, true);

lightning/src/ln/channel.rs

+24
Original file line numberDiff line numberDiff line change
@@ -736,6 +736,9 @@ pub(super) struct Channel<Signer: Sign> {
736736
// don't currently support node id aliases and eventually privacy should be provided with
737737
// blinded paths instead of simple scid+node_id aliases.
738738
outbound_scid_alias: u64,
739+
740+
// We track whether we already emitted a `ChannelReady` event.
741+
channel_ready_event_emitted: bool,
739742
}
740743

741744
#[cfg(any(test, fuzzing))]
@@ -1063,6 +1066,8 @@ impl<Signer: Sign> Channel<Signer> {
10631066
latest_inbound_scid_alias: None,
10641067
outbound_scid_alias,
10651068

1069+
channel_ready_event_emitted: false,
1070+
10661071
#[cfg(any(test, fuzzing))]
10671072
historical_inbound_htlc_fulfills: HashSet::new(),
10681073

@@ -1397,6 +1402,8 @@ impl<Signer: Sign> Channel<Signer> {
13971402
latest_inbound_scid_alias: None,
13981403
outbound_scid_alias,
13991404

1405+
channel_ready_event_emitted: false,
1406+
14001407
#[cfg(any(test, fuzzing))]
14011408
historical_inbound_htlc_fulfills: HashSet::new(),
14021409

@@ -4598,6 +4605,16 @@ impl<Signer: Sign> Channel<Signer> {
45984605
self.prev_config.map(|prev_config| prev_config.0)
45994606
}
46004607

4608+
// Checks whether we should emit a `ChannelReady` event.
4609+
pub(crate) fn should_emit_channel_ready_event(&mut self) -> bool {
4610+
self.is_usable() && !self.channel_ready_event_emitted
4611+
}
4612+
4613+
// Remembers that we already emitted a `ChannelReady` event.
4614+
pub(crate) fn set_channel_ready_event_emitted(&mut self) {
4615+
self.channel_ready_event_emitted = true;
4616+
}
4617+
46014618
/// Tracks the number of ticks elapsed since the previous [`ChannelConfig`] was updated. Once
46024619
/// [`EXPIRE_PREV_CONFIG_TICKS`] is reached, the previous config is considered expired and will
46034620
/// no longer be considered when forwarding HTLCs.
@@ -6230,6 +6247,8 @@ impl<Signer: Sign> Writeable for Channel<Signer> {
62306247
if self.holder_max_htlc_value_in_flight_msat != Self::get_holder_max_htlc_value_in_flight_msat(self.channel_value_satoshis, &old_max_in_flight_percent_config)
62316248
{ Some(self.holder_max_htlc_value_in_flight_msat) } else { None };
62326249

6250+
let channel_ready_event_emitted = Some(self.channel_ready_event_emitted);
6251+
62336252
write_tlv_fields!(writer, {
62346253
(0, self.announcement_sigs, option),
62356254
// minimum_depth and counterparty_selected_channel_reserve_satoshis used to have a
@@ -6252,6 +6271,7 @@ impl<Signer: Sign> Writeable for Channel<Signer> {
62526271
(17, self.announcement_sigs_state, required),
62536272
(19, self.latest_inbound_scid_alias, option),
62546273
(21, self.outbound_scid_alias, required),
6274+
(23, channel_ready_event_emitted, option),
62556275
});
62566276

62576277
Ok(())
@@ -6509,6 +6529,7 @@ impl<'a, Signer: Sign, K: Deref> ReadableArgs<(&'a K, u32)> for Channel<Signer>
65096529
let mut announcement_sigs_state = Some(AnnouncementSigsState::NotSent);
65106530
let mut latest_inbound_scid_alias = None;
65116531
let mut outbound_scid_alias = None;
6532+
let mut channel_ready_event_emitted = None;
65126533

65136534
read_tlv_fields!(reader, {
65146535
(0, announcement_sigs, option),
@@ -6526,6 +6547,7 @@ impl<'a, Signer: Sign, K: Deref> ReadableArgs<(&'a K, u32)> for Channel<Signer>
65266547
(17, announcement_sigs_state, option),
65276548
(19, latest_inbound_scid_alias, option),
65286549
(21, outbound_scid_alias, option),
6550+
(23, channel_ready_event_emitted, option),
65296551
});
65306552

65316553
if let Some(preimages) = preimages_opt {
@@ -6666,6 +6688,8 @@ impl<'a, Signer: Sign, K: Deref> ReadableArgs<(&'a K, u32)> for Channel<Signer>
66666688
// Later in the ChannelManager deserialization phase we scan for channels and assign scid aliases if its missing
66676689
outbound_scid_alias: outbound_scid_alias.unwrap_or(0),
66686690

6691+
channel_ready_event_emitted: channel_ready_event_emitted.unwrap_or(false),
6692+
66696693
#[cfg(any(test, fuzzing))]
66706694
historical_inbound_htlc_fulfills,
66716695

lightning/src/ln/channelmanager.rs

+52
Original file line numberDiff line numberDiff line change
@@ -1466,6 +1466,20 @@ macro_rules! send_channel_ready {
14661466
}
14671467
}
14681468

1469+
macro_rules! emit_channel_ready_event {
1470+
($pending_events: expr, $channel: expr) => {
1471+
if $channel.should_emit_channel_ready_event() {
1472+
$pending_events.push(events::Event::ChannelReady {
1473+
channel_id: $channel.channel_id(),
1474+
user_channel_id: $channel.get_user_id(),
1475+
counterparty_node_id: $channel.get_counterparty_node_id(),
1476+
channel_type: $channel.get_channel_type().clone(),
1477+
});
1478+
$channel.set_channel_ready_event_emitted();
1479+
}
1480+
}
1481+
}
1482+
14691483
macro_rules! handle_chan_restoration_locked {
14701484
($self: ident, $channel_lock: expr, $channel_state: expr, $channel_entry: expr,
14711485
$raa: expr, $commitment_update: expr, $order: expr, $chanmon_update: expr,
@@ -1509,6 +1523,11 @@ macro_rules! handle_chan_restoration_locked {
15091523
});
15101524
}
15111525

1526+
{
1527+
let mut pending_events = $self.pending_events.lock().unwrap();
1528+
emit_channel_ready_event!(pending_events, $channel_entry.get_mut());
1529+
}
1530+
15121531
let funding_broadcastable: Option<Transaction> = $funding_broadcastable; // Force type-checking to resolve
15131532
if let Some(monitor_update) = chanmon_update {
15141533
// We only ever broadcast a funding transaction in response to a funding_signed
@@ -4672,6 +4691,12 @@ impl<M: Deref, T: Deref, K: Deref, F: Deref, L: Deref> ChannelManager<M, T, K, F
46724691
});
46734692
}
46744693
}
4694+
4695+
{
4696+
let mut pending_events = self.pending_events.lock().unwrap();
4697+
emit_channel_ready_event!(pending_events, chan.get_mut());
4698+
}
4699+
46754700
Ok(())
46764701
},
46774702
hash_map::Entry::Vacant(_) => Err(MsgHandleErrInternal::send_err_msg_no_close("Failed to find corresponding channel".to_owned(), msg.channel_id))
@@ -5829,6 +5854,12 @@ where
58295854
log_trace!(self.logger, "Sending channel_ready WITHOUT channel_update for {}", log_bytes!(channel.channel_id()));
58305855
}
58315856
}
5857+
5858+
{
5859+
let mut pending_events = self.pending_events.lock().unwrap();
5860+
emit_channel_ready_event!(pending_events, channel);
5861+
}
5862+
58325863
if let Some(announcement_sigs) = announcement_sigs {
58335864
log_trace!(self.logger, "Sending announcement_signatures for channel {}", log_bytes!(channel.channel_id()));
58345865
pending_msg_events.push(events::MessageSendEvent::SendAnnouncementSignatures {
@@ -7813,6 +7844,9 @@ mod tests {
78137844
let (channel_ready, _) = create_chan_between_nodes_with_value_confirm(&nodes[0], &nodes[1], &tx);
78147845
let (announcement, nodes_0_update, nodes_1_update) = create_chan_between_nodes_with_value_b(&nodes[0], &nodes[1], &channel_ready);
78157846
update_nodes_with_chan_announce(&nodes, 0, 1, &announcement, &nodes_0_update, &nodes_1_update);
7847+
expect_channel_ready_event!(nodes[0], nodes[1].node.get_our_node_id());
7848+
expect_channel_ready_event!(nodes[1], nodes[0].node.get_our_node_id());
7849+
78167850

78177851
nodes[0].node.close_channel(channel_id, &nodes[1].node.get_our_node_id()).unwrap();
78187852
nodes[1].node.handle_shutdown(&nodes[0].node.get_our_node_id(), &channelmanager::provided_init_features(), &get_event_msg!(nodes[0], MessageSendEvent::SendShutdown, nodes[1].node.get_our_node_id()));
@@ -7981,6 +8015,24 @@ pub mod bench {
79818015
_ => panic!(),
79828016
}
79838017

8018+
let events_a = node_a.get_and_clear_pending_events();
8019+
assert_eq!(events_a.len(), 1);
8020+
match events_a[0] {
8021+
Event::ChannelReady{ ref counterparty_node_id, .. } => {
8022+
assert_eq!(*counterparty_node_id, node_b.get_our_node_id());
8023+
},
8024+
_ => panic!("Unexpected event"),
8025+
}
8026+
8027+
let events_b = node_b.get_and_clear_pending_events();
8028+
assert_eq!(events_b.len(), 1);
8029+
match events_b[0] {
8030+
Event::ChannelReady{ ref counterparty_node_id, .. } => {
8031+
assert_eq!(*counterparty_node_id, node_a.get_our_node_id());
8032+
},
8033+
_ => panic!("Unexpected event"),
8034+
}
8035+
79848036
let dummy_graph = NetworkGraph::new(genesis_hash, &logger_a);
79858037

79868038
let mut payment_count: u64 = 0;

lightning/src/ln/functional_test_utils.rs

+22
Original file line numberDiff line numberDiff line change
@@ -407,9 +407,26 @@ pub fn create_chan_between_nodes<'a, 'b, 'c, 'd>(node_a: &'a Node<'b, 'c, 'd>, n
407407
create_chan_between_nodes_with_value(node_a, node_b, 100000, 10001, a_flags, b_flags)
408408
}
409409

410+
#[macro_export]
411+
#[cfg(any(test, feature = "_bench_unstable", feature = "_test_utils"))]
412+
macro_rules! expect_channel_ready_event {
413+
($node: expr, $expected_counterparty_node_id: expr) => {
414+
let events = $node.node.get_and_clear_pending_events();
415+
assert_eq!(events.len(), 1);
416+
match events[0] {
417+
$crate::util::events::Event::ChannelReady{ ref counterparty_node_id, .. } => {
418+
assert_eq!($expected_counterparty_node_id, *counterparty_node_id);
419+
},
420+
_ => panic!("Unexpected event"),
421+
}
422+
}
423+
}
424+
410425
pub fn create_chan_between_nodes_with_value<'a, 'b, 'c, 'd>(node_a: &'a Node<'b, 'c, 'd>, node_b: &'a Node<'b, 'c, 'd>, channel_value: u64, push_msat: u64, a_flags: InitFeatures, b_flags: InitFeatures) -> (msgs::ChannelAnnouncement, msgs::ChannelUpdate, msgs::ChannelUpdate, [u8; 32], Transaction) {
411426
let (channel_ready, channel_id, tx) = create_chan_between_nodes_with_value_a(node_a, node_b, channel_value, push_msat, a_flags, b_flags);
412427
let (announcement, as_update, bs_update) = create_chan_between_nodes_with_value_b(node_a, node_b, &channel_ready);
428+
expect_channel_ready_event!(node_a, node_b.node.get_our_node_id());
429+
expect_channel_ready_event!(node_b, node_a.node.get_our_node_id());
413430
(announcement, as_update, bs_update, channel_id, tx)
414431
}
415432

@@ -741,6 +758,9 @@ pub fn open_zero_conf_channel<'a, 'b, 'c, 'd>(initiator: &'a Node<'b, 'c, 'd>, r
741758
assert_eq!(initiator.node.list_usable_channels().len(), initiator_channels + 1);
742759
assert_eq!(receiver.node.list_usable_channels().len(), receiver_channels + 1);
743760

761+
expect_channel_ready_event!(initiator, receiver.node.get_our_node_id());
762+
expect_channel_ready_event!(receiver, initiator.node.get_our_node_id());
763+
744764
(tx, as_channel_ready.channel_id)
745765
}
746766

@@ -870,8 +890,10 @@ pub fn create_unannounced_chan_between_nodes_with_value<'a, 'b, 'c, 'd>(nodes: &
870890
connect_blocks(&nodes[b], CHAN_CONFIRM_DEPTH - 1);
871891
let as_channel_ready = get_event_msg!(nodes[a], MessageSendEvent::SendChannelReady, nodes[b].node.get_our_node_id());
872892
nodes[a].node.handle_channel_ready(&nodes[b].node.get_our_node_id(), &get_event_msg!(nodes[b], MessageSendEvent::SendChannelReady, nodes[a].node.get_our_node_id()));
893+
expect_channel_ready_event!(nodes[a], nodes[b].node.get_our_node_id());
873894
let as_update = get_event_msg!(nodes[a], MessageSendEvent::SendChannelUpdate, nodes[b].node.get_our_node_id());
874895
nodes[b].node.handle_channel_ready(&nodes[a].node.get_our_node_id(), &as_channel_ready);
896+
expect_channel_ready_event!(nodes[b], nodes[a].node.get_our_node_id());
875897
let bs_update = get_event_msg!(nodes[b], MessageSendEvent::SendChannelUpdate, nodes[a].node.get_our_node_id());
876898

877899
nodes[a].node.handle_channel_update(&nodes[b].node.get_our_node_id(), &bs_update);

0 commit comments

Comments
 (0)