Skip to content

Commit f4c2d40

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 f4c2d40

File tree

12 files changed

+202
-12
lines changed

12 files changed

+202
-12
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

+4
Original file line numberDiff line numberDiff line change
@@ -869,6 +869,8 @@ 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+
expect_channel_ready_event(&nodes[0], &nodes[2].node.get_our_node_id());
873+
expect_channel_ready_event(&nodes[2], &nodes[0].node.get_our_node_id());
872874

873875
// As `msgs::ChannelUpdate` was never handled for the participating node(s) of the second
874876
// channel, the channel will never be assigned any `counterparty.forwarding_info`.
@@ -1257,6 +1259,8 @@ mod test {
12571259
get_event_msg!(nodes[1], MessageSendEvent::SendChannelUpdate, nodes[3].node.get_our_node_id());
12581260
nodes[3].node.handle_channel_ready(&nodes[1].node.get_our_node_id(), &as_channel_ready);
12591261
get_event_msg!(nodes[3], MessageSendEvent::SendChannelUpdate, nodes[1].node.get_our_node_id());
1262+
expect_channel_ready_event(&nodes[1], &nodes[3].node.get_our_node_id());
1263+
expect_channel_ready_event(&nodes[3], &nodes[1].node.get_our_node_id());
12601264

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

lightning/src/ln/chanmon_update_fail_tests.rs

+7
Original file line numberDiff line numberDiff line change
@@ -1916,6 +1916,13 @@ fn do_during_funding_monitor_fail(confirm_a_first: bool, restore_b_before_conf:
19161916
node.gossip_sync.handle_channel_update(&bs_update).unwrap();
19171917
}
19181918

1919+
if !restore_b_before_lock {
1920+
expect_channel_ready_event(&nodes[1], &nodes[0].node.get_our_node_id());
1921+
} else {
1922+
expect_channel_ready_event(&nodes[0], &nodes[1].node.get_our_node_id());
1923+
}
1924+
1925+
19191926
send_payment(&nodes[0], &[&nodes[1]], 8000000);
19201927
close_channel(&nodes[0], &nodes[1], &channel_id, funding_tx, true);
19211928
check_closed_event!(nodes[0], 1, ClosureReason::CooperativeClosure);

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(true),
6692+
66696693
#[cfg(any(test, fuzzing))]
66706694
historical_inbound_htlc_fulfills,
66716695

lightning/src/ln/channelmanager.rs

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

1469+
macro_rules! emit_channel_ready_event {
1470+
($self: expr, $channel: expr) => {
1471+
if $channel.should_emit_channel_ready_event() {
1472+
{
1473+
let mut pending_events = $self.pending_events.lock().unwrap();
1474+
pending_events.push(events::Event::ChannelReady {
1475+
channel_id: $channel.channel_id(),
1476+
user_channel_id: $channel.get_user_id(),
1477+
counterparty_node_id: $channel.get_counterparty_node_id(),
1478+
channel_type: $channel.get_channel_type().clone(),
1479+
});
1480+
}
1481+
$channel.set_channel_ready_event_emitted();
1482+
}
1483+
}
1484+
}
1485+
14691486
macro_rules! handle_chan_restoration_locked {
14701487
($self: ident, $channel_lock: expr, $channel_state: expr, $channel_entry: expr,
14711488
$raa: expr, $commitment_update: expr, $order: expr, $chanmon_update: expr,
@@ -1509,6 +1526,8 @@ macro_rules! handle_chan_restoration_locked {
15091526
});
15101527
}
15111528

1529+
emit_channel_ready_event!($self, $channel_entry.get_mut());
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,9 @@ impl<M: Deref, T: Deref, K: Deref, F: Deref, L: Deref> ChannelManager<M, T, K, F
46724691
});
46734692
}
46744693
}
4694+
4695+
emit_channel_ready_event!(self, chan.get_mut());
4696+
46754697
Ok(())
46764698
},
46774699
hash_map::Entry::Vacant(_) => Err(MsgHandleErrInternal::send_err_msg_no_close("Failed to find corresponding channel".to_owned(), msg.channel_id))
@@ -5829,6 +5851,9 @@ where
58295851
log_trace!(self.logger, "Sending channel_ready WITHOUT channel_update for {}", log_bytes!(channel.channel_id()));
58305852
}
58315853
}
5854+
5855+
emit_channel_ready_event!(self, channel);
5856+
58325857
if let Some(announcement_sigs) = announcement_sigs {
58335858
log_trace!(self.logger, "Sending announcement_signatures for channel {}", log_bytes!(channel.channel_id()));
58345859
pending_msg_events.push(events::MessageSendEvent::SendAnnouncementSignatures {
@@ -7981,6 +8006,24 @@ pub mod bench {
79818006
_ => panic!(),
79828007
}
79838008

8009+
let events_a = node_a.get_and_clear_pending_events();
8010+
assert_eq!(events_a.len(), 1);
8011+
match events_a[0] {
8012+
Event::ChannelReady{ ref counterparty_node_id, .. } => {
8013+
assert_eq!(*counterparty_node_id, node_b.get_our_node_id());
8014+
},
8015+
_ => panic!("Unexpected event"),
8016+
}
8017+
8018+
let events_b = node_b.get_and_clear_pending_events();
8019+
assert_eq!(events_b.len(), 1);
8020+
match events_b[0] {
8021+
Event::ChannelReady{ ref counterparty_node_id, .. } => {
8022+
assert_eq!(*counterparty_node_id, node_a.get_our_node_id());
8023+
},
8024+
_ => panic!("Unexpected event"),
8025+
}
8026+
79848027
let dummy_graph = NetworkGraph::new(genesis_hash, &logger_a);
79858028

79868029
let mut payment_count: u64 = 0;

lightning/src/ln/functional_test_utils.rs

+20
Original file line numberDiff line numberDiff line change
@@ -741,6 +741,9 @@ pub fn open_zero_conf_channel<'a, 'b, 'c, 'd>(initiator: &'a Node<'b, 'c, 'd>, r
741741
assert_eq!(initiator.node.list_usable_channels().len(), initiator_channels + 1);
742742
assert_eq!(receiver.node.list_usable_channels().len(), receiver_channels + 1);
743743

744+
expect_channel_ready_event(&initiator, &receiver.node.get_our_node_id());
745+
expect_channel_ready_event(&receiver, &initiator.node.get_our_node_id());
746+
744747
(tx, as_channel_ready.channel_id)
745748
}
746749

@@ -794,6 +797,7 @@ pub fn create_chan_between_nodes_with_value_confirm<'a, 'b, 'c, 'd>(node_a: &'a
794797
create_chan_between_nodes_with_value_confirm_first(node_a, node_b, tx, conf_height);
795798
confirm_transaction_at(node_a, tx, conf_height);
796799
connect_blocks(node_a, CHAN_CONFIRM_DEPTH - 1);
800+
expect_channel_ready_event(&node_a, &node_b.node.get_our_node_id());
797801
create_chan_between_nodes_with_value_confirm_second(node_b, node_a)
798802
}
799803

@@ -832,6 +836,7 @@ pub fn create_chan_between_nodes_with_value_b<'a, 'b, 'c>(node_a: &Node<'a, 'b,
832836

833837
*node_a.network_chan_count.borrow_mut() += 1;
834838

839+
expect_channel_ready_event(&node_b, &node_a.node.get_our_node_id());
835840
((*announcement).clone(), (*as_update).clone(), (*bs_update).clone())
836841
}
837842

@@ -870,8 +875,10 @@ pub fn create_unannounced_chan_between_nodes_with_value<'a, 'b, 'c, 'd>(nodes: &
870875
connect_blocks(&nodes[b], CHAN_CONFIRM_DEPTH - 1);
871876
let as_channel_ready = get_event_msg!(nodes[a], MessageSendEvent::SendChannelReady, nodes[b].node.get_our_node_id());
872877
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()));
878+
expect_channel_ready_event(&nodes[a], &nodes[b].node.get_our_node_id());
873879
let as_update = get_event_msg!(nodes[a], MessageSendEvent::SendChannelUpdate, nodes[b].node.get_our_node_id());
874880
nodes[b].node.handle_channel_ready(&nodes[a].node.get_our_node_id(), &as_channel_ready);
881+
expect_channel_ready_event(&nodes[b], &nodes[a].node.get_our_node_id());
875882
let bs_update = get_event_msg!(nodes[b], MessageSendEvent::SendChannelUpdate, nodes[a].node.get_our_node_id());
876883

877884
nodes[a].node.handle_channel_update(&nodes[b].node.get_our_node_id(), &bs_update);
@@ -1503,6 +1510,19 @@ macro_rules! expect_payment_forwarded {
15031510
}
15041511
}
15051512

1513+
#[cfg(any(test, feature = "_bench_unstable", feature = "_test_utils"))]
1514+
pub fn expect_channel_ready_event<'a, 'b, 'c, 'd>(node: &'a Node<'b, 'c, 'd>, expected_counterparty_node_id: &PublicKey) {
1515+
let events = node.node.get_and_clear_pending_events();
1516+
assert_eq!(events.len(), 1);
1517+
match events[0] {
1518+
crate::util::events::Event::ChannelReady{ ref counterparty_node_id, .. } => {
1519+
assert_eq!(*expected_counterparty_node_id, *counterparty_node_id);
1520+
},
1521+
_ => panic!("Unexpected event"),
1522+
}
1523+
}
1524+
1525+
15061526
pub struct PaymentFailedConditions<'a> {
15071527
pub(crate) expected_htlc_error_data: Option<(u16, &'a [u8])>,
15081528
pub(crate) expected_blamed_scid: Option<u64>,

lightning/src/ln/functional_tests.rs

+21-5
Original file line numberDiff line numberDiff line change
@@ -558,6 +558,7 @@ fn do_test_sanity_on_in_flight_opens(steps: u8) {
558558
confirm_transaction_at(&nodes[0], &tx, 2);
559559
connect_blocks(&nodes[0], CHAN_CONFIRM_DEPTH);
560560
create_chan_between_nodes_with_value_confirm_second(&nodes[1], &nodes[0]);
561+
expect_channel_ready_event(&nodes[0], &nodes[1].node.get_our_node_id());
561562
}
562563

563564
#[test]
@@ -3711,11 +3712,23 @@ fn do_test_drop_messages_peer_disconnect(messages_delivered: u8, simulate_broken
37113712
}
37123713

37133714
let events_1 = nodes[1].node.get_and_clear_pending_events();
3714-
assert_eq!(events_1.len(), 1);
3715-
match events_1[0] {
3716-
Event::PendingHTLCsForwardable { .. } => { },
3717-
_ => panic!("Unexpected event"),
3718-
};
3715+
if messages_delivered == 0 {
3716+
assert_eq!(events_1.len(), 2);
3717+
match events_1[0] {
3718+
Event::ChannelReady { .. } => { },
3719+
_ => panic!("Unexpected event"),
3720+
};
3721+
match events_1[1] {
3722+
Event::PendingHTLCsForwardable { .. } => { },
3723+
_ => panic!("Unexpected event"),
3724+
};
3725+
} else {
3726+
assert_eq!(events_1.len(), 1);
3727+
match events_1[0] {
3728+
Event::PendingHTLCsForwardable { .. } => { },
3729+
_ => panic!("Unexpected event"),
3730+
};
3731+
}
37193732

37203733
nodes[0].node.peer_disconnected(&nodes[1].node.get_our_node_id(), false);
37213734
nodes[1].node.peer_disconnected(&nodes[0].node.get_our_node_id(), false);
@@ -3952,6 +3965,8 @@ fn test_funding_peer_disconnect() {
39523965
},
39533966
_ => panic!("Unexpected event {:?}", events_6[0]),
39543967
};
3968+
expect_channel_ready_event(&nodes[0], &nodes[1].node.get_our_node_id());
3969+
expect_channel_ready_event(&nodes[1], &nodes[0].node.get_our_node_id());
39553970

39563971
// When we deliver nodes[1]'s announcement_signatures to nodes[0], nodes[0] should immediately
39573972
// broadcast the channel announcement globally, as well as re-send its (now-public)
@@ -9429,6 +9444,7 @@ fn test_duplicate_chan_id() {
94299444
let (channel_ready, _) = create_chan_between_nodes_with_value_confirm(&nodes[0], &nodes[1], &tx);
94309445
let (announcement, as_update, bs_update) = create_chan_between_nodes_with_value_b(&nodes[0], &nodes[1], &channel_ready);
94319446
update_nodes_with_chan_announce(&nodes, 0, 1, &announcement, &as_update, &bs_update);
9447+
94329448
send_payment(&nodes[0], &[&nodes[1]], 8000000);
94339449
}
94349450

lightning/src/ln/payment_tests.rs

+1-1
Original file line numberDiff line numberDiff line change
@@ -483,6 +483,7 @@ fn do_retry_with_no_persist(confirm_before_reload: bool) {
483483
nodes[1].node.handle_update_fulfill_htlc(&nodes[2].node.get_our_node_id(), &htlc_fulfill_updates.update_fulfill_htlcs[0]);
484484
check_added_monitors!(nodes[1], 1);
485485
commitment_signed_dance!(nodes[1], nodes[2], htlc_fulfill_updates.commitment_signed, false);
486+
expect_payment_forwarded!(nodes[1], nodes[0], nodes[2], None, false, false);
486487

487488
if confirm_before_reload {
488489
let best_block = nodes[0].blocks.lock().unwrap().last().unwrap().clone();
@@ -499,7 +500,6 @@ fn do_retry_with_no_persist(confirm_before_reload: bool) {
499500
let bs_htlc_claim_txn = nodes[1].tx_broadcaster.txn_broadcasted.lock().unwrap().split_off(0);
500501
assert_eq!(bs_htlc_claim_txn.len(), 1);
501502
check_spends!(bs_htlc_claim_txn[0], as_commitment_tx);
502-
expect_payment_forwarded!(nodes[1], nodes[0], nodes[2], None, false, false);
503503

504504
if !confirm_before_reload {
505505
mine_transaction(&nodes[0], &as_commitment_tx);

0 commit comments

Comments
 (0)