Skip to content

Commit 2d4e628

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 a32d795 commit 2d4e628

29 files changed

+507
-511
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, Init, UpdateAddHTLC,
55+
BaseMessageHandler, ChannelMessageHandler, CommitmentUpdate, Init, MessageSendEvent,
56+
UpdateAddHTLC,
5657
};
5758
use lightning::ln::script::ShutdownScript;
5859
use lightning::ln::types::ChannelId;
@@ -767,7 +768,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
767768
let open_channel = {
768769
let events = $source.get_and_clear_pending_msg_events();
769770
assert_eq!(events.len(), 1);
770-
if let events::MessageSendEvent::SendOpenChannel { ref msg, .. } = events[0] {
771+
if let MessageSendEvent::SendOpenChannel { ref msg, .. } = events[0] {
771772
msg.clone()
772773
} else {
773774
panic!("Wrong event type");
@@ -803,7 +804,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
803804
}
804805
let events = $dest.get_and_clear_pending_msg_events();
805806
assert_eq!(events.len(), 1);
806-
if let events::MessageSendEvent::SendAcceptChannel { ref msg, .. } = events[0] {
807+
if let MessageSendEvent::SendAcceptChannel { ref msg, .. } = events[0] {
807808
msg.clone()
808809
} else {
809810
panic!("Wrong event type");
@@ -846,7 +847,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
846847
let funding_created = {
847848
let events = $source.get_and_clear_pending_msg_events();
848849
assert_eq!(events.len(), 1);
849-
if let events::MessageSendEvent::SendFundingCreated { ref msg, .. } = events[0] {
850+
if let MessageSendEvent::SendFundingCreated { ref msg, .. } = events[0] {
850851
msg.clone()
851852
} else {
852853
panic!("Wrong event type");
@@ -857,7 +858,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
857858
let funding_signed = {
858859
let events = $dest.get_and_clear_pending_msg_events();
859860
assert_eq!(events.len(), 1);
860-
if let events::MessageSendEvent::SendFundingSigned { ref msg, .. } = events[0] {
861+
if let MessageSendEvent::SendFundingSigned { ref msg, .. } = events[0] {
861862
msg.clone()
862863
} else {
863864
panic!("Wrong event type");
@@ -912,9 +913,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
912913
}
913914
for (idx, node_event) in node_events.iter().enumerate() {
914915
for event in node_event {
915-
if let events::MessageSendEvent::SendChannelReady { ref node_id, ref msg } =
916-
event
917-
{
916+
if let MessageSendEvent::SendChannelReady { ref node_id, ref msg } = event {
918917
for node in $nodes.iter() {
919918
if node.get_our_node_id() == *node_id {
920919
node.handle_channel_ready($nodes[idx].get_our_node_id(), msg);
@@ -929,7 +928,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
929928
for node in $nodes.iter() {
930929
let events = node.get_and_clear_pending_msg_events();
931930
for event in events {
932-
if let events::MessageSendEvent::SendAnnouncementSignatures { .. } = event {
931+
if let MessageSendEvent::SendAnnouncementSignatures { .. } = event {
933932
} else {
934933
panic!("Wrong event type");
935934
}
@@ -1014,25 +1013,25 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
10141013
let expect_drop_id = if let Some(id) = expect_drop_node { Some(nodes[id].get_our_node_id()) } else { None };
10151014
for event in $excess_events {
10161015
let push_a = match event {
1017-
events::MessageSendEvent::UpdateHTLCs { ref node_id, .. } => {
1016+
MessageSendEvent::UpdateHTLCs { ref node_id, .. } => {
10181017
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10191018
*node_id == a_id
10201019
},
1021-
events::MessageSendEvent::SendRevokeAndACK { ref node_id, .. } => {
1020+
MessageSendEvent::SendRevokeAndACK { ref node_id, .. } => {
10221021
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10231022
*node_id == a_id
10241023
},
1025-
events::MessageSendEvent::SendChannelReestablish { ref node_id, .. } => {
1024+
MessageSendEvent::SendChannelReestablish { ref node_id, .. } => {
10261025
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10271026
*node_id == a_id
10281027
},
1029-
events::MessageSendEvent::SendStfu { ref node_id, .. } => {
1028+
MessageSendEvent::SendStfu { ref node_id, .. } => {
10301029
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10311030
*node_id == a_id
10321031
},
1033-
events::MessageSendEvent::SendChannelReady { .. } => continue,
1034-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => continue,
1035-
events::MessageSendEvent::SendChannelUpdate { ref node_id, ref msg } => {
1032+
MessageSendEvent::SendChannelReady { .. } => continue,
1033+
MessageSendEvent::SendAnnouncementSignatures { .. } => continue,
1034+
MessageSendEvent::SendChannelUpdate { ref node_id, ref msg } => {
10361035
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
10371036
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10381037
*node_id == a_id
@@ -1088,7 +1087,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
10881087
for event in &mut events_iter {
10891088
had_events = true;
10901089
match event {
1091-
events::MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate { update_add_htlcs, update_fail_htlcs, update_fulfill_htlcs, update_fail_malformed_htlcs, update_fee, commitment_signed } } => {
1090+
MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate { update_add_htlcs, update_fail_htlcs, update_fulfill_htlcs, update_fail_malformed_htlcs, update_fee, commitment_signed } } => {
10921091
for (idx, dest) in nodes.iter().enumerate() {
10931092
if dest.get_our_node_id() == node_id {
10941093
for update_add in update_add_htlcs.iter() {
@@ -1126,7 +1125,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
11261125
!update_fail_htlcs.is_empty() || !update_fail_malformed_htlcs.is_empty();
11271126
if $limit_events != ProcessMessages::AllMessages && processed_change {
11281127
// If we only want to process some messages, don't deliver the CS until later.
1129-
extra_ev = Some(events::MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate {
1128+
extra_ev = Some(MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate {
11301129
update_add_htlcs: Vec::new(),
11311130
update_fail_htlcs: Vec::new(),
11321131
update_fulfill_htlcs: Vec::new(),
@@ -1142,37 +1141,37 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
11421141
}
11431142
}
11441143
},
1145-
events::MessageSendEvent::SendRevokeAndACK { ref node_id, ref msg } => {
1144+
MessageSendEvent::SendRevokeAndACK { ref node_id, ref msg } => {
11461145
for (idx, dest) in nodes.iter().enumerate() {
11471146
if dest.get_our_node_id() == *node_id {
11481147
out.locked_write(format!("Delivering revoke_and_ack from node {} to node {}.\n", $node, idx).as_bytes());
11491148
dest.handle_revoke_and_ack(nodes[$node].get_our_node_id(), msg);
11501149
}
11511150
}
11521151
},
1153-
events::MessageSendEvent::SendChannelReestablish { ref node_id, ref msg } => {
1152+
MessageSendEvent::SendChannelReestablish { ref node_id, ref msg } => {
11541153
for (idx, dest) in nodes.iter().enumerate() {
11551154
if dest.get_our_node_id() == *node_id {
11561155
out.locked_write(format!("Delivering channel_reestablish from node {} to node {}.\n", $node, idx).as_bytes());
11571156
dest.handle_channel_reestablish(nodes[$node].get_our_node_id(), msg);
11581157
}
11591158
}
11601159
},
1161-
events::MessageSendEvent::SendStfu { ref node_id, ref msg } => {
1160+
MessageSendEvent::SendStfu { ref node_id, ref msg } => {
11621161
for (idx, dest) in nodes.iter().enumerate() {
11631162
if dest.get_our_node_id() == *node_id {
11641163
out.locked_write(format!("Delivering stfu from node {} to node {}.\n", $node, idx).as_bytes());
11651164
dest.handle_stfu(nodes[$node].get_our_node_id(), msg);
11661165
}
11671166
}
11681167
}
1169-
events::MessageSendEvent::SendChannelReady { .. } => {
1168+
MessageSendEvent::SendChannelReady { .. } => {
11701169
// Can be generated as a reestablish response
11711170
},
1172-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {
1171+
MessageSendEvent::SendAnnouncementSignatures { .. } => {
11731172
// Can be generated as a reestablish response
11741173
},
1175-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1174+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
11761175
// When we reconnect we will resend a channel_update to make sure our
11771176
// counterparty has the latest parameters for receiving payments
11781177
// through us. We do, however, check that the message does not include
@@ -1215,13 +1214,13 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
12151214
if $counterparty_id == 0 {
12161215
for event in nodes[0].get_and_clear_pending_msg_events() {
12171216
match event {
1218-
events::MessageSendEvent::UpdateHTLCs { .. } => {},
1219-
events::MessageSendEvent::SendRevokeAndACK { .. } => {},
1220-
events::MessageSendEvent::SendChannelReestablish { .. } => {},
1221-
events::MessageSendEvent::SendStfu { .. } => {},
1222-
events::MessageSendEvent::SendChannelReady { .. } => {},
1223-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1224-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1217+
MessageSendEvent::UpdateHTLCs { .. } => {},
1218+
MessageSendEvent::SendRevokeAndACK { .. } => {},
1219+
MessageSendEvent::SendChannelReestablish { .. } => {},
1220+
MessageSendEvent::SendStfu { .. } => {},
1221+
MessageSendEvent::SendChannelReady { .. } => {},
1222+
MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1223+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
12251224
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
12261225
},
12271226
_ => {
@@ -1242,13 +1241,13 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
12421241
} else {
12431242
for event in nodes[2].get_and_clear_pending_msg_events() {
12441243
match event {
1245-
events::MessageSendEvent::UpdateHTLCs { .. } => {},
1246-
events::MessageSendEvent::SendRevokeAndACK { .. } => {},
1247-
events::MessageSendEvent::SendChannelReestablish { .. } => {},
1248-
events::MessageSendEvent::SendStfu { .. } => {},
1249-
events::MessageSendEvent::SendChannelReady { .. } => {},
1250-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1251-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1244+
MessageSendEvent::UpdateHTLCs { .. } => {},
1245+
MessageSendEvent::SendRevokeAndACK { .. } => {},
1246+
MessageSendEvent::SendChannelReestablish { .. } => {},
1247+
MessageSendEvent::SendStfu { .. } => {},
1248+
MessageSendEvent::SendChannelReady { .. } => {},
1249+
MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1250+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
12521251
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
12531252
},
12541253
_ => {

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)