Skip to content

Commit f1a1288

Browse files
committed
Move MessageSendEvent to ln::msgs from events
`MessageSendEvent` got placed in the `events` module next to `Event` but it really doesn't make a whole lot of sense to be there (despite the superficially similar name). `MessageSendEvent`s aren't really "events" so much as simply queued messages, handled by the `PeerManager` rather than the user. Further, they're used entirely by structs implementing the traits in `ln::msgs` and basically define an enum over `ln::msgs` structs. Thus, it only really makes sense to have them in `ln::msgs`, which we do here.
1 parent f3c33eb commit f1a1288

28 files changed

+505
-509
lines changed

fuzz/src/chanmon_consistency.rs

Lines changed: 37 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,8 @@ use lightning::ln::channelmanager::{
5252
use lightning::ln::functional_test_utils::*;
5353
use lightning::ln::inbound_payment::ExpandedKey;
5454
use lightning::ln::msgs::{
55-
BaseMessageHandler, ChannelMessageHandler, CommitmentUpdate, DecodeError, Init, UpdateAddHTLC,
55+
BaseMessageHandler, ChannelMessageHandler, CommitmentUpdate, DecodeError, Init,
56+
MessageSendEvent, UpdateAddHTLC,
5657
};
5758
use lightning::ln::script::ShutdownScript;
5859
use lightning::ln::types::ChannelId;
@@ -781,7 +782,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
781782
let open_channel = {
782783
let events = $source.get_and_clear_pending_msg_events();
783784
assert_eq!(events.len(), 1);
784-
if let events::MessageSendEvent::SendOpenChannel { ref msg, .. } = events[0] {
785+
if let MessageSendEvent::SendOpenChannel { ref msg, .. } = events[0] {
785786
msg.clone()
786787
} else {
787788
panic!("Wrong event type");
@@ -817,7 +818,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
817818
}
818819
let events = $dest.get_and_clear_pending_msg_events();
819820
assert_eq!(events.len(), 1);
820-
if let events::MessageSendEvent::SendAcceptChannel { ref msg, .. } = events[0] {
821+
if let MessageSendEvent::SendAcceptChannel { ref msg, .. } = events[0] {
821822
msg.clone()
822823
} else {
823824
panic!("Wrong event type");
@@ -860,7 +861,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
860861
let funding_created = {
861862
let events = $source.get_and_clear_pending_msg_events();
862863
assert_eq!(events.len(), 1);
863-
if let events::MessageSendEvent::SendFundingCreated { ref msg, .. } = events[0] {
864+
if let MessageSendEvent::SendFundingCreated { ref msg, .. } = events[0] {
864865
msg.clone()
865866
} else {
866867
panic!("Wrong event type");
@@ -871,7 +872,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
871872
let funding_signed = {
872873
let events = $dest.get_and_clear_pending_msg_events();
873874
assert_eq!(events.len(), 1);
874-
if let events::MessageSendEvent::SendFundingSigned { ref msg, .. } = events[0] {
875+
if let MessageSendEvent::SendFundingSigned { ref msg, .. } = events[0] {
875876
msg.clone()
876877
} else {
877878
panic!("Wrong event type");
@@ -926,9 +927,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
926927
}
927928
for (idx, node_event) in node_events.iter().enumerate() {
928929
for event in node_event {
929-
if let events::MessageSendEvent::SendChannelReady { ref node_id, ref msg } =
930-
event
931-
{
930+
if let MessageSendEvent::SendChannelReady { ref node_id, ref msg } = event {
932931
for node in $nodes.iter() {
933932
if node.get_our_node_id() == *node_id {
934933
node.handle_channel_ready($nodes[idx].get_our_node_id(), msg);
@@ -943,7 +942,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
943942
for node in $nodes.iter() {
944943
let events = node.get_and_clear_pending_msg_events();
945944
for event in events {
946-
if let events::MessageSendEvent::SendAnnouncementSignatures { .. } = event {
945+
if let MessageSendEvent::SendAnnouncementSignatures { .. } = event {
947946
} else {
948947
panic!("Wrong event type");
949948
}
@@ -1028,25 +1027,25 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
10281027
let expect_drop_id = if let Some(id) = expect_drop_node { Some(nodes[id].get_our_node_id()) } else { None };
10291028
for event in $excess_events {
10301029
let push_a = match event {
1031-
events::MessageSendEvent::UpdateHTLCs { ref node_id, .. } => {
1030+
MessageSendEvent::UpdateHTLCs { ref node_id, .. } => {
10321031
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10331032
*node_id == a_id
10341033
},
1035-
events::MessageSendEvent::SendRevokeAndACK { ref node_id, .. } => {
1034+
MessageSendEvent::SendRevokeAndACK { ref node_id, .. } => {
10361035
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10371036
*node_id == a_id
10381037
},
1039-
events::MessageSendEvent::SendChannelReestablish { ref node_id, .. } => {
1038+
MessageSendEvent::SendChannelReestablish { ref node_id, .. } => {
10401039
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10411040
*node_id == a_id
10421041
},
1043-
events::MessageSendEvent::SendStfu { ref node_id, .. } => {
1042+
MessageSendEvent::SendStfu { ref node_id, .. } => {
10441043
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10451044
*node_id == a_id
10461045
},
1047-
events::MessageSendEvent::SendChannelReady { .. } => continue,
1048-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => continue,
1049-
events::MessageSendEvent::SendChannelUpdate { ref node_id, ref msg } => {
1046+
MessageSendEvent::SendChannelReady { .. } => continue,
1047+
MessageSendEvent::SendAnnouncementSignatures { .. } => continue,
1048+
MessageSendEvent::SendChannelUpdate { ref node_id, ref msg } => {
10501049
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
10511050
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10521051
*node_id == a_id
@@ -1102,7 +1101,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
11021101
for event in &mut events_iter {
11031102
had_events = true;
11041103
match event {
1105-
events::MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate { update_add_htlcs, update_fail_htlcs, update_fulfill_htlcs, update_fail_malformed_htlcs, update_fee, commitment_signed } } => {
1104+
MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate { update_add_htlcs, update_fail_htlcs, update_fulfill_htlcs, update_fail_malformed_htlcs, update_fee, commitment_signed } } => {
11061105
for (idx, dest) in nodes.iter().enumerate() {
11071106
if dest.get_our_node_id() == node_id {
11081107
for update_add in update_add_htlcs.iter() {
@@ -1140,7 +1139,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
11401139
!update_fail_htlcs.is_empty() || !update_fail_malformed_htlcs.is_empty();
11411140
if $limit_events != ProcessMessages::AllMessages && processed_change {
11421141
// If we only want to process some messages, don't deliver the CS until later.
1143-
extra_ev = Some(events::MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate {
1142+
extra_ev = Some(MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate {
11441143
update_add_htlcs: Vec::new(),
11451144
update_fail_htlcs: Vec::new(),
11461145
update_fulfill_htlcs: Vec::new(),
@@ -1156,37 +1155,37 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
11561155
}
11571156
}
11581157
},
1159-
events::MessageSendEvent::SendRevokeAndACK { ref node_id, ref msg } => {
1158+
MessageSendEvent::SendRevokeAndACK { ref node_id, ref msg } => {
11601159
for (idx, dest) in nodes.iter().enumerate() {
11611160
if dest.get_our_node_id() == *node_id {
11621161
out.locked_write(format!("Delivering revoke_and_ack from node {} to node {}.\n", $node, idx).as_bytes());
11631162
dest.handle_revoke_and_ack(nodes[$node].get_our_node_id(), msg);
11641163
}
11651164
}
11661165
},
1167-
events::MessageSendEvent::SendChannelReestablish { ref node_id, ref msg } => {
1166+
MessageSendEvent::SendChannelReestablish { ref node_id, ref msg } => {
11681167
for (idx, dest) in nodes.iter().enumerate() {
11691168
if dest.get_our_node_id() == *node_id {
11701169
out.locked_write(format!("Delivering channel_reestablish from node {} to node {}.\n", $node, idx).as_bytes());
11711170
dest.handle_channel_reestablish(nodes[$node].get_our_node_id(), msg);
11721171
}
11731172
}
11741173
},
1175-
events::MessageSendEvent::SendStfu { ref node_id, ref msg } => {
1174+
MessageSendEvent::SendStfu { ref node_id, ref msg } => {
11761175
for (idx, dest) in nodes.iter().enumerate() {
11771176
if dest.get_our_node_id() == *node_id {
11781177
out.locked_write(format!("Delivering stfu from node {} to node {}.\n", $node, idx).as_bytes());
11791178
dest.handle_stfu(nodes[$node].get_our_node_id(), msg);
11801179
}
11811180
}
11821181
}
1183-
events::MessageSendEvent::SendChannelReady { .. } => {
1182+
MessageSendEvent::SendChannelReady { .. } => {
11841183
// Can be generated as a reestablish response
11851184
},
1186-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {
1185+
MessageSendEvent::SendAnnouncementSignatures { .. } => {
11871186
// Can be generated as a reestablish response
11881187
},
1189-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1188+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
11901189
// When we reconnect we will resend a channel_update to make sure our
11911190
// counterparty has the latest parameters for receiving payments
11921191
// through us. We do, however, check that the message does not include
@@ -1229,13 +1228,13 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
12291228
if $counterparty_id == 0 {
12301229
for event in nodes[0].get_and_clear_pending_msg_events() {
12311230
match event {
1232-
events::MessageSendEvent::UpdateHTLCs { .. } => {},
1233-
events::MessageSendEvent::SendRevokeAndACK { .. } => {},
1234-
events::MessageSendEvent::SendChannelReestablish { .. } => {},
1235-
events::MessageSendEvent::SendStfu { .. } => {},
1236-
events::MessageSendEvent::SendChannelReady { .. } => {},
1237-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1238-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1231+
MessageSendEvent::UpdateHTLCs { .. } => {},
1232+
MessageSendEvent::SendRevokeAndACK { .. } => {},
1233+
MessageSendEvent::SendChannelReestablish { .. } => {},
1234+
MessageSendEvent::SendStfu { .. } => {},
1235+
MessageSendEvent::SendChannelReady { .. } => {},
1236+
MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1237+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
12391238
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
12401239
},
12411240
_ => {
@@ -1256,13 +1255,13 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
12561255
} else {
12571256
for event in nodes[2].get_and_clear_pending_msg_events() {
12581257
match event {
1259-
events::MessageSendEvent::UpdateHTLCs { .. } => {},
1260-
events::MessageSendEvent::SendRevokeAndACK { .. } => {},
1261-
events::MessageSendEvent::SendChannelReestablish { .. } => {},
1262-
events::MessageSendEvent::SendStfu { .. } => {},
1263-
events::MessageSendEvent::SendChannelReady { .. } => {},
1264-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1265-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1258+
MessageSendEvent::UpdateHTLCs { .. } => {},
1259+
MessageSendEvent::SendRevokeAndACK { .. } => {},
1260+
MessageSendEvent::SendChannelReestablish { .. } => {},
1261+
MessageSendEvent::SendStfu { .. } => {},
1262+
MessageSendEvent::SendChannelReady { .. } => {},
1263+
MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1264+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
12661265
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
12671266
},
12681267
_ => {

lightning-background-processor/src/lib.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1070,13 +1070,13 @@ mod tests {
10701070
use lightning::chain::channelmonitor::ANTI_REORG_DELAY;
10711071
use lightning::chain::transaction::OutPoint;
10721072
use lightning::chain::{chainmonitor, BestBlock, Confirm, Filter};
1073-
use lightning::events::{Event, MessageSendEvent, PathFailure, ReplayEvent};
1073+
use lightning::events::{Event, PathFailure, ReplayEvent};
10741074
use lightning::ln::channelmanager;
10751075
use lightning::ln::channelmanager::{
10761076
ChainParameters, PaymentId, BREAKDOWN_TIMEOUT, MIN_CLTV_EXPIRY_DELTA,
10771077
};
10781078
use lightning::ln::functional_test_utils::*;
1079-
use lightning::ln::msgs::{BaseMessageHandler, ChannelMessageHandler, Init};
1079+
use lightning::ln::msgs::{BaseMessageHandler, ChannelMessageHandler, Init, MessageSendEvent};
10801080
use lightning::ln::peer_handler::{
10811081
IgnoringMessageHandler, MessageHandler, PeerManager, SocketDescriptor,
10821082
};

lightning-net-tokio/src/lib.rs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -623,7 +623,6 @@ mod tests {
623623
use bitcoin::constants::ChainHash;
624624
use bitcoin::secp256k1::{PublicKey, Secp256k1, SecretKey};
625625
use bitcoin::Network;
626-
use lightning::events::*;
627626
use lightning::ln::msgs::*;
628627
use lightning::ln::peer_handler::{IgnoringMessageHandler, MessageHandler, PeerManager};
629628
use lightning::routing::gossip::NodeId;

lightning/src/chain/chainmonitor.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -928,9 +928,9 @@ mod tests {
928928
use crate::{get_htlc_update_msgs, get_revoke_commit_msgs};
929929
use crate::chain::{ChannelMonitorUpdateStatus, Watch};
930930
use crate::chain::channelmonitor::ANTI_REORG_DELAY;
931-
use crate::events::{ClosureReason, Event, MessageSendEvent};
931+
use crate::events::{ClosureReason, Event};
932932
use crate::ln::functional_test_utils::*;
933-
use crate::ln::msgs::{BaseMessageHandler, ChannelMessageHandler};
933+
use crate::ln::msgs::{BaseMessageHandler, ChannelMessageHandler, MessageSendEvent};
934934

935935
const CHAINSYNC_MONITOR_PARTITION_FACTOR: u32 = 5;
936936

0 commit comments

Comments
 (0)