Skip to content

Commit 3b926f0

Browse files
dunxendunxen
committed
Add new wire messaging and events but don't handle them
1 parent 27dc363 commit 3b926f0

File tree

9 files changed

+480
-4
lines changed

9 files changed

+480
-4
lines changed

‎lightning-net-tokio/src/lib.rs‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -529,6 +529,17 @@ mod tests {
529529
fn handle_update_fee(&self, _their_node_id: &PublicKey, _msg: &UpdateFee) {}
530530
fn handle_announcement_signatures(&self, _their_node_id: &PublicKey, _msg: &AnnouncementSignatures) {}
531531
fn handle_channel_update(&self, _their_node_id: &PublicKey, _msg: &ChannelUpdate) {}
532+
fn handle_open_channel_v2(&self, _their_node_id: &PublicKey, _msg: &OpenChannelV2) {}
533+
fn handle_accept_channel_v2(&self, _their_node_id: &PublicKey, _msg: &AcceptChannelV2) {}
534+
fn handle_tx_add_input(&self, _their_node_id: &PublicKey, _msg: &TxAddInput) {}
535+
fn handle_tx_add_output(&self, _their_node_id: &PublicKey, _msg: &TxAddOutput) {}
536+
fn handle_tx_remove_input(&self, _their_node_id: &PublicKey, _msg: &TxRemoveInput) {}
537+
fn handle_tx_remove_output(&self, _their_node_id: &PublicKey, _msg: &TxRemoveOutput) {}
538+
fn handle_tx_complete(&self, _their_node_id: &PublicKey, _msg: &TxComplete) {}
539+
fn handle_tx_signatures(&self, _their_node_id: &PublicKey, _msg: &TxSignatures) {}
540+
fn handle_tx_init_rbf(&self, _their_node_id: &PublicKey, _msg: &TxInitRbf) {}
541+
fn handle_tx_ack_rbf(&self, _their_node_id: &PublicKey, _msg: &TxAckRbf) {}
542+
fn handle_tx_abort(&self, _their_node_id: &PublicKey, _msg: &TxAbort) {}
532543
fn peer_disconnected(&self, their_node_id: &PublicKey) {
533544
if *their_node_id == self.expected_pubkey {
534545
self.disconnected_flag.store(true, Ordering::SeqCst);

‎lightning/src/events/mod.rs‎

Lines changed: 79 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1436,6 +1436,14 @@ pub enum MessageSendEvent {
14361436
/// The message which should be sent.
14371437
msg: msgs::AcceptChannel,
14381438
},
1439+
/// Used to indicate that we've accepted a V2 channel open and should send the accept_channel2
1440+
/// message provided to the given peer.
1441+
SendAcceptChannelV2 {
1442+
/// The node_id of the node which should receive this message
1443+
node_id: PublicKey,
1444+
/// The message which should be sent.
1445+
msg: msgs::AcceptChannelV2,
1446+
},
14391447
/// Used to indicate that we've initiated a channel open and should send the open_channel
14401448
/// message provided to the given peer.
14411449
SendOpenChannel {
@@ -1444,6 +1452,14 @@ pub enum MessageSendEvent {
14441452
/// The message which should be sent.
14451453
msg: msgs::OpenChannel,
14461454
},
1455+
/// Used to indicate that we've initiated a V2 channel open and should send the open_channel2
1456+
/// message provided to the given peer.
1457+
SendOpenChannelV2 {
1458+
/// The node_id of the node which should receive this message
1459+
node_id: PublicKey,
1460+
/// The message which should be sent.
1461+
msg: msgs::OpenChannelV2,
1462+
},
14471463
/// Used to indicate that a funding_created message should be sent to the peer with the given node_id.
14481464
SendFundingCreated {
14491465
/// The node_id of the node which should receive this message
@@ -1458,6 +1474,69 @@ pub enum MessageSendEvent {
14581474
/// The message which should be sent.
14591475
msg: msgs::FundingSigned,
14601476
},
1477+
/// Used to indicate that a tx_add_input message should be sent to the peer with the given node_id.
1478+
SendTxAddInput {
1479+
/// The node_id of the node which should receive this message
1480+
node_id: PublicKey,
1481+
/// The message which should be sent.
1482+
msg: msgs::TxAddInput,
1483+
},
1484+
/// Used to indicate that a tx_add_output message should be sent to the peer with the given node_id.
1485+
SendTxAddOutput {
1486+
/// The node_id of the node which should receive this message
1487+
node_id: PublicKey,
1488+
/// The message which should be sent.
1489+
msg: msgs::TxAddOutput,
1490+
},
1491+
/// Used to indicate that a tx_remove_input message should be sent to the peer with the given node_id.
1492+
SendTxRemoveInput {
1493+
/// The node_id of the node which should receive this message
1494+
node_id: PublicKey,
1495+
/// The message which should be sent.
1496+
msg: msgs::TxRemoveInput,
1497+
},
1498+
/// Used to indicate that a tx_remove_output message should be sent to the peer with the given node_id.
1499+
SendTxRemoveOutput {
1500+
/// The node_id of the node which should receive this message
1501+
node_id: PublicKey,
1502+
/// The message which should be sent.
1503+
msg: msgs::TxRemoveOutput,
1504+
},
1505+
/// Used to indicate that a tx_complete message should be sent to the peer with the given node_id.
1506+
SendTxComplete {
1507+
/// The node_id of the node which should receive this message
1508+
node_id: PublicKey,
1509+
/// The message which should be sent.
1510+
msg: msgs::TxComplete,
1511+
},
1512+
/// Used to indicate that a tx_signatures message should be sent to the peer with the given node_id.
1513+
SendTxSignatures {
1514+
/// The node_id of the node which should receive this message
1515+
node_id: PublicKey,
1516+
/// The message which should be sent.
1517+
msg: msgs::TxSignatures,
1518+
},
1519+
/// Used to indicate that a tx_init_rbf message should be sent to the peer with the given node_id.
1520+
SendTxInitRbf {
1521+
/// The node_id of the node which should receive this message
1522+
node_id: PublicKey,
1523+
/// The message which should be sent.
1524+
msg: msgs::TxInitRbf,
1525+
},
1526+
/// Used to indicate that a tx_ack_rbf message should be sent to the peer with the given node_id.
1527+
SendTxAckRbf {
1528+
/// The node_id of the node which should receive this message
1529+
node_id: PublicKey,
1530+
/// The message which should be sent.
1531+
msg: msgs::TxAckRbf,
1532+
},
1533+
/// Used to indicate that a tx_abort message should be sent to the peer with the given node_id.
1534+
SendTxAbort {
1535+
/// The node_id of the node which should receive this message
1536+
node_id: PublicKey,
1537+
/// The message which should be sent.
1538+
msg: msgs::TxAddInput,
1539+
},
14611540
/// Used to indicate that a channel_ready message should be sent to the peer with the given node_id.
14621541
SendChannelReady {
14631542
/// The node_id of the node which should receive these message(s)

‎lightning/src/ln/channel.rs‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5695,8 +5695,9 @@ impl<Signer: WriteableEcdsaChannelSigner> Channel<Signer> {
56955695
next_remote_commitment_number: INITIAL_COMMITMENT_NUMBER - self.cur_counterparty_commitment_transaction_number - 1,
56965696
your_last_per_commitment_secret: remote_last_secret,
56975697
my_current_per_commitment_point: dummy_pubkey,
5698-
// TODO(dual_funding): If we've sent `commtiment_signed` for an interactive transaction construction but have not received `tx_signatures`
5699-
// we MUST set `next_funding_txid` to the txid of that interactive transaction, else we MUST NOT set it.
5698+
// TODO(dual_funding): If we've sent `commtiment_signed` for an interactive transaction
5699+
// construction but have not received `tx_signatures` we MUST set `next_funding_txid` to the
5700+
// txid of that interactive transaction, else we MUST NOT set it.
57005701
next_funding_txid: None,
57015702
}
57025703
}

‎lightning/src/ln/channelmanager.rs‎

Lines changed: 84 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6381,11 +6381,23 @@ where
63816381
let _ = handle_error!(self, self.internal_open_channel(counterparty_node_id, msg), *counterparty_node_id);
63826382
}
63836383

6384+
fn handle_open_channel_v2(&self, counterparty_node_id: &PublicKey, msg: &msgs::OpenChannelV2) {
6385+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6386+
"Dual-funded channels not supported".to_owned(),
6387+
msg.temporary_channel_id.clone())), *counterparty_node_id);
6388+
}
6389+
63846390
fn handle_accept_channel(&self, counterparty_node_id: &PublicKey, msg: &msgs::AcceptChannel) {
63856391
let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(&self.total_consistency_lock, &self.persistence_notifier);
63866392
let _ = handle_error!(self, self.internal_accept_channel(counterparty_node_id, msg), *counterparty_node_id);
63876393
}
63886394

6395+
fn handle_accept_channel_v2(&self, counterparty_node_id: &PublicKey, msg: &msgs::AcceptChannelV2) {
6396+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6397+
"Dual-funded channels not supported".to_owned(),
6398+
msg.temporary_channel_id.clone())), *counterparty_node_id);
6399+
}
6400+
63896401
fn handle_funding_created(&self, counterparty_node_id: &PublicKey, msg: &msgs::FundingCreated) {
63906402
let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(&self.total_consistency_lock, &self.persistence_notifier);
63916403
let _ = handle_error!(self, self.internal_funding_created(counterparty_node_id, msg), *counterparty_node_id);
@@ -6488,23 +6500,40 @@ where
64886500
});
64896501
pending_msg_events.retain(|msg| {
64906502
match msg {
6503+
// V1 Channel Establishment
64916504
&events::MessageSendEvent::SendAcceptChannel { .. } => false,
64926505
&events::MessageSendEvent::SendOpenChannel { .. } => false,
64936506
&events::MessageSendEvent::SendFundingCreated { .. } => false,
64946507
&events::MessageSendEvent::SendFundingSigned { .. } => false,
6508+
// V2 Channel Establishment
6509+
&events::MessageSendEvent::SendAcceptChannelV2 { .. } => false,
6510+
&events::MessageSendEvent::SendOpenChannelV2 { .. } => false,
6511+
// Common Channel Establishment
64956512
&events::MessageSendEvent::SendChannelReady { .. } => false,
64966513
&events::MessageSendEvent::SendAnnouncementSignatures { .. } => false,
6514+
// Interactive Transaction Construction
6515+
&events::MessageSendEvent::SendTxAddInput { .. } => false,
6516+
&events::MessageSendEvent::SendTxAddOutput { .. } => false,
6517+
&events::MessageSendEvent::SendTxRemoveInput { .. } => false,
6518+
&events::MessageSendEvent::SendTxRemoveOutput { .. } => false,
6519+
&events::MessageSendEvent::SendTxComplete { .. } => false,
6520+
&events::MessageSendEvent::SendTxSignatures { .. } => false,
6521+
&events::MessageSendEvent::SendTxInitRbf { .. } => false,
6522+
&events::MessageSendEvent::SendTxAckRbf { .. } => false,
6523+
&events::MessageSendEvent::SendTxAbort { .. } => false,
6524+
// Channel Operations
64976525
&events::MessageSendEvent::UpdateHTLCs { .. } => false,
64986526
&events::MessageSendEvent::SendRevokeAndACK { .. } => false,
64996527
&events::MessageSendEvent::SendClosingSigned { .. } => false,
65006528
&events::MessageSendEvent::SendShutdown { .. } => false,
65016529
&events::MessageSendEvent::SendChannelReestablish { .. } => false,
6530+
&events::MessageSendEvent::HandleError { .. } => false,
6531+
// Gossip
65026532
&events::MessageSendEvent::SendChannelAnnouncement { .. } => false,
65036533
&events::MessageSendEvent::BroadcastChannelAnnouncement { .. } => true,
65046534
&events::MessageSendEvent::BroadcastChannelUpdate { .. } => true,
65056535
&events::MessageSendEvent::BroadcastNodeAnnouncement { .. } => true,
65066536
&events::MessageSendEvent::SendChannelUpdate { .. } => false,
6507-
&events::MessageSendEvent::HandleError { .. } => false,
65086537
&events::MessageSendEvent::SendChannelRangeQuery { .. } => false,
65096538
&events::MessageSendEvent::SendShortIdsQuery { .. } => false,
65106539
&events::MessageSendEvent::SendReplyChannelRange { .. } => false,
@@ -6661,6 +6690,60 @@ where
66616690
fn provided_init_features(&self, _their_init_features: &PublicKey) -> InitFeatures {
66626691
provided_init_features(&self.default_configuration)
66636692
}
6693+
6694+
fn handle_tx_add_input(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxAddInput) {
6695+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6696+
"Dual-funded channels not supported".to_owned(),
6697+
msg.channel_id.clone())), *counterparty_node_id);
6698+
}
6699+
6700+
fn handle_tx_add_output(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxAddOutput) {
6701+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6702+
"Dual-funded channels not supported".to_owned(),
6703+
msg.channel_id.clone())), *counterparty_node_id);
6704+
}
6705+
6706+
fn handle_tx_remove_input(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxRemoveInput) {
6707+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6708+
"Dual-funded channels not supported".to_owned(),
6709+
msg.channel_id.clone())), *counterparty_node_id);
6710+
}
6711+
6712+
fn handle_tx_remove_output(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxRemoveOutput) {
6713+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6714+
"Dual-funded channels not supported".to_owned(),
6715+
msg.channel_id.clone())), *counterparty_node_id);
6716+
}
6717+
6718+
fn handle_tx_complete(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxComplete) {
6719+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6720+
"Dual-funded channels not supported".to_owned(),
6721+
msg.channel_id.clone())), *counterparty_node_id);
6722+
}
6723+
6724+
fn handle_tx_signatures(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxSignatures) {
6725+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6726+
"Dual-funded channels not supported".to_owned(),
6727+
msg.channel_id.clone())), *counterparty_node_id);
6728+
}
6729+
6730+
fn handle_tx_init_rbf(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxInitRbf) {
6731+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6732+
"Dual-funded channels not supported".to_owned(),
6733+
msg.channel_id.clone())), *counterparty_node_id);
6734+
}
6735+
6736+
fn handle_tx_ack_rbf(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxAckRbf) {
6737+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6738+
"Dual-funded channels not supported".to_owned(),
6739+
msg.channel_id.clone())), *counterparty_node_id);
6740+
}
6741+
6742+
fn handle_tx_abort(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxAbort) {
6743+
let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
6744+
"Dual-funded channels not supported".to_owned(),
6745+
msg.channel_id.clone())), *counterparty_node_id);
6746+
}
66646747
}
66656748

66666749
/// Fetches the set of [`NodeFeatures`] flags which are provided by or required by

‎lightning/src/ln/functional_test_utils.rs‎

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -723,6 +723,39 @@ pub fn remove_first_msg_event_to_node(msg_node_id: &PublicKey, msg_events: &mut
723723
MessageSendEvent::SendGossipTimestampFilter { node_id, .. } => {
724724
node_id == msg_node_id
725725
},
726+
MessageSendEvent::SendAcceptChannelV2 { node_id, .. } => {
727+
node_id == msg_node_id
728+
},
729+
MessageSendEvent::SendOpenChannelV2 { node_id, .. } => {
730+
node_id == msg_node_id
731+
},
732+
MessageSendEvent::SendTxAddInput { node_id, .. } => {
733+
node_id == msg_node_id
734+
},
735+
MessageSendEvent::SendTxAddOutput { node_id, .. } => {
736+
node_id == msg_node_id
737+
},
738+
MessageSendEvent::SendTxRemoveInput { node_id, .. } => {
739+
node_id == msg_node_id
740+
},
741+
MessageSendEvent::SendTxRemoveOutput { node_id, .. } => {
742+
node_id == msg_node_id
743+
},
744+
MessageSendEvent::SendTxComplete { node_id, .. } => {
745+
node_id == msg_node_id
746+
},
747+
MessageSendEvent::SendTxSignatures { node_id, .. } => {
748+
node_id == msg_node_id
749+
},
750+
MessageSendEvent::SendTxInitRbf { node_id, .. } => {
751+
node_id == msg_node_id
752+
},
753+
MessageSendEvent::SendTxAckRbf { node_id, .. } => {
754+
node_id == msg_node_id
755+
},
756+
MessageSendEvent::SendTxAbort { node_id, .. } => {
757+
node_id == msg_node_id
758+
},
726759
}});
727760
if ev_index.is_some() {
728761
msg_events.remove(ev_index.unwrap())

‎lightning/src/ln/msgs.rs‎

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1198,8 +1198,12 @@ pub trait ChannelMessageHandler : MessageSendEventsProvider {
11981198
// Channel init:
11991199
/// Handle an incoming `open_channel` message from the given peer.
12001200
fn handle_open_channel(&self, their_node_id: &PublicKey, msg: &OpenChannel);
1201+
/// Handle an incoming `open_channel2` message from the given peer.
1202+
fn handle_open_channel_v2(&self, their_node_id: &PublicKey, msg: &OpenChannelV2);
12011203
/// Handle an incoming `accept_channel` message from the given peer.
12021204
fn handle_accept_channel(&self, their_node_id: &PublicKey, msg: &AcceptChannel);
1205+
/// Handle an incoming `accept_channel2` message from the given peer.
1206+
fn handle_accept_channel_v2(&self, their_node_id: &PublicKey, msg: &AcceptChannelV2);
12031207
/// Handle an incoming `funding_created` message from the given peer.
12041208
fn handle_funding_created(&self, their_node_id: &PublicKey, msg: &FundingCreated);
12051209
/// Handle an incoming `funding_signed` message from the given peer.
@@ -1213,6 +1217,26 @@ pub trait ChannelMessageHandler : MessageSendEventsProvider {
12131217
/// Handle an incoming `closing_signed` message from the given peer.
12141218
fn handle_closing_signed(&self, their_node_id: &PublicKey, msg: &ClosingSigned);
12151219

1220+
// Interactive channel construction
1221+
/// Handle an incoming `tx_add_input message` from the given peer.
1222+
fn handle_tx_add_input(&self, their_node_id: &PublicKey, msg: &TxAddInput);
1223+
/// Handle an incoming `tx_add_output` message from the given peer.
1224+
fn handle_tx_add_output(&self, their_node_id: &PublicKey, msg: &TxAddOutput);
1225+
/// Handle an incoming `tx_remove_input` message from the given peer.
1226+
fn handle_tx_remove_input(&self, their_node_id: &PublicKey, msg: &TxRemoveInput);
1227+
/// Handle an incoming `tx_remove_output` message from the given peer.
1228+
fn handle_tx_remove_output(&self, their_node_id: &PublicKey, msg: &TxRemoveOutput);
1229+
/// Handle an incoming `tx_complete message` from the given peer.
1230+
fn handle_tx_complete(&self, their_node_id: &PublicKey, msg: &TxComplete);
1231+
/// Handle an incoming `tx_signatures` message from the given peer.
1232+
fn handle_tx_signatures(&self, their_node_id: &PublicKey, msg: &TxSignatures);
1233+
/// Handle an incoming `tx_init_rbf` message from the given peer.
1234+
fn handle_tx_init_rbf(&self, their_node_id: &PublicKey, msg: &TxInitRbf);
1235+
/// Handle an incoming `tx_ack_rbf` message from the given peer.
1236+
fn handle_tx_ack_rbf(&self, their_node_id: &PublicKey, msg: &TxAckRbf);
1237+
/// Handle an incoming `tx_abort message` from the given peer.
1238+
fn handle_tx_abort(&self, their_node_id: &PublicKey, msg: &TxAbort);
1239+
12161240
// HTLC handling:
12171241
/// Handle an incoming `update_add_htlc` message from the given peer.
12181242
fn handle_update_add_htlc(&self, their_node_id: &PublicKey, msg: &UpdateAddHTLC);

0 commit comments

Comments
 (0)