Skip to content

Commit 6562669

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 ba1017c commit 6562669

27 files changed

+474
-477
lines changed

fuzz/src/chanmon_consistency.rs

Lines changed: 32 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ use lightning::ln::functional_test_utils::*;
5353
use lightning::ln::inbound_payment::ExpandedKey;
5454
use lightning::ln::msgs::{
5555
self, BaseMessageHandler, ChannelMessageHandler, CommitmentUpdate, DecodeError, Init,
56-
UpdateAddHTLC,
56+
MessageSendEvent, UpdateAddHTLC,
5757
};
5858
use lightning::ln::script::ShutdownScript;
5959
use lightning::ln::types::ChannelId;
@@ -782,7 +782,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
782782
let open_channel = {
783783
let events = $source.get_and_clear_pending_msg_events();
784784
assert_eq!(events.len(), 1);
785-
if let events::MessageSendEvent::SendOpenChannel { ref msg, .. } = events[0] {
785+
if let MessageSendEvent::SendOpenChannel { ref msg, .. } = events[0] {
786786
msg.clone()
787787
} else {
788788
panic!("Wrong event type");
@@ -817,7 +817,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
817817
}
818818
let events = $dest.get_and_clear_pending_msg_events();
819819
assert_eq!(events.len(), 1);
820-
if let events::MessageSendEvent::SendAcceptChannel { ref msg, .. } = events[0] {
820+
if let MessageSendEvent::SendAcceptChannel { ref msg, .. } = events[0] {
821821
msg.clone()
822822
} else {
823823
panic!("Wrong event type");
@@ -860,7 +860,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
860860
let funding_created = {
861861
let events = $source.get_and_clear_pending_msg_events();
862862
assert_eq!(events.len(), 1);
863-
if let events::MessageSendEvent::SendFundingCreated { ref msg, .. } = events[0] {
863+
if let MessageSendEvent::SendFundingCreated { ref msg, .. } = events[0] {
864864
msg.clone()
865865
} else {
866866
panic!("Wrong event type");
@@ -871,7 +871,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
871871
let funding_signed = {
872872
let events = $dest.get_and_clear_pending_msg_events();
873873
assert_eq!(events.len(), 1);
874-
if let events::MessageSendEvent::SendFundingSigned { ref msg, .. } = events[0] {
874+
if let MessageSendEvent::SendFundingSigned { ref msg, .. } = events[0] {
875875
msg.clone()
876876
} else {
877877
panic!("Wrong event type");
@@ -926,9 +926,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
926926
}
927927
for (idx, node_event) in node_events.iter().enumerate() {
928928
for event in node_event {
929-
if let events::MessageSendEvent::SendChannelReady { ref node_id, ref msg } =
930-
event
931-
{
929+
if let MessageSendEvent::SendChannelReady { ref node_id, ref msg } = event {
932930
for node in $nodes.iter() {
933931
if node.get_our_node_id() == *node_id {
934932
node.handle_channel_ready($nodes[idx].get_our_node_id(), msg);
@@ -943,7 +941,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
943941
for node in $nodes.iter() {
944942
let events = node.get_and_clear_pending_msg_events();
945943
for event in events {
946-
if let events::MessageSendEvent::SendAnnouncementSignatures { .. } = event {
944+
if let MessageSendEvent::SendAnnouncementSignatures { .. } = event {
947945
} else {
948946
panic!("Wrong event type");
949947
}
@@ -1026,21 +1024,21 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
10261024
let expect_drop_id = if let Some(id) = expect_drop_node { Some(nodes[id].get_our_node_id()) } else { None };
10271025
for event in $excess_events {
10281026
let push_a = match event {
1029-
events::MessageSendEvent::UpdateHTLCs { ref node_id, .. } => {
1027+
MessageSendEvent::UpdateHTLCs { ref node_id, .. } => {
10301028
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10311029
*node_id == a_id
10321030
},
1033-
events::MessageSendEvent::SendRevokeAndACK { ref node_id, .. } => {
1031+
MessageSendEvent::SendRevokeAndACK { ref node_id, .. } => {
10341032
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10351033
*node_id == a_id
10361034
},
1037-
events::MessageSendEvent::SendChannelReestablish { ref node_id, .. } => {
1035+
MessageSendEvent::SendChannelReestablish { ref node_id, .. } => {
10381036
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10391037
*node_id == a_id
10401038
},
1041-
events::MessageSendEvent::SendChannelReady { .. } => continue,
1042-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => continue,
1043-
events::MessageSendEvent::SendChannelUpdate { ref node_id, ref msg } => {
1039+
MessageSendEvent::SendChannelReady { .. } => continue,
1040+
MessageSendEvent::SendAnnouncementSignatures { .. } => continue,
1041+
MessageSendEvent::SendChannelUpdate { ref node_id, ref msg } => {
10441042
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
10451043
if Some(*node_id) == expect_drop_id { panic!("peer_disconnected should drop msgs bound for the disconnected peer"); }
10461044
*node_id == a_id
@@ -1096,7 +1094,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
10961094
for event in &mut events_iter {
10971095
had_events = true;
10981096
match event {
1099-
events::MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate { update_add_htlcs, update_fail_htlcs, update_fulfill_htlcs, update_fail_malformed_htlcs, update_fee, commitment_signed } } => {
1097+
MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate { update_add_htlcs, update_fail_htlcs, update_fulfill_htlcs, update_fail_malformed_htlcs, update_fee, commitment_signed } } => {
11001098
for (idx, dest) in nodes.iter().enumerate() {
11011099
if dest.get_our_node_id() == node_id {
11021100
for update_add in update_add_htlcs.iter() {
@@ -1134,7 +1132,7 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
11341132
!update_fail_htlcs.is_empty() || !update_fail_malformed_htlcs.is_empty();
11351133
if $limit_events != ProcessMessages::AllMessages && processed_change {
11361134
// If we only want to process some messages, don't deliver the CS until later.
1137-
extra_ev = Some(events::MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate {
1135+
extra_ev = Some(MessageSendEvent::UpdateHTLCs { node_id, updates: CommitmentUpdate {
11381136
update_add_htlcs: Vec::new(),
11391137
update_fail_htlcs: Vec::new(),
11401138
update_fulfill_htlcs: Vec::new(),
@@ -1150,29 +1148,29 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
11501148
}
11511149
}
11521150
},
1153-
events::MessageSendEvent::SendRevokeAndACK { ref node_id, ref msg } => {
1151+
MessageSendEvent::SendRevokeAndACK { ref node_id, ref msg } => {
11541152
for (idx, dest) in nodes.iter().enumerate() {
11551153
if dest.get_our_node_id() == *node_id {
11561154
out.locked_write(format!("Delivering revoke_and_ack to node {}.\n", idx).as_bytes());
11571155
dest.handle_revoke_and_ack(nodes[$node].get_our_node_id(), msg);
11581156
}
11591157
}
11601158
},
1161-
events::MessageSendEvent::SendChannelReestablish { ref node_id, ref msg } => {
1159+
MessageSendEvent::SendChannelReestablish { ref node_id, ref msg } => {
11621160
for (idx, dest) in nodes.iter().enumerate() {
11631161
if dest.get_our_node_id() == *node_id {
11641162
out.locked_write(format!("Delivering channel_reestablish to node {}.\n", idx).as_bytes());
11651163
dest.handle_channel_reestablish(nodes[$node].get_our_node_id(), msg);
11661164
}
11671165
}
11681166
},
1169-
events::MessageSendEvent::SendChannelReady { .. } => {
1167+
MessageSendEvent::SendChannelReady { .. } => {
11701168
// Can be generated as a reestablish response
11711169
},
1172-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {
1170+
MessageSendEvent::SendAnnouncementSignatures { .. } => {
11731171
// Can be generated as a reestablish response
11741172
},
1175-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1173+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
11761174
// When we reconnect we will resend a channel_update to make sure our
11771175
// counterparty has the latest parameters for receiving payments
11781176
// through us. We do, however, check that the message does not include
@@ -1215,12 +1213,12 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
12151213
if $counterparty_id == 0 {
12161214
for event in nodes[0].get_and_clear_pending_msg_events() {
12171215
match event {
1218-
events::MessageSendEvent::UpdateHTLCs { .. } => {},
1219-
events::MessageSendEvent::SendRevokeAndACK { .. } => {},
1220-
events::MessageSendEvent::SendChannelReestablish { .. } => {},
1221-
events::MessageSendEvent::SendChannelReady { .. } => {},
1222-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1223-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1216+
MessageSendEvent::UpdateHTLCs { .. } => {},
1217+
MessageSendEvent::SendRevokeAndACK { .. } => {},
1218+
MessageSendEvent::SendChannelReestablish { .. } => {},
1219+
MessageSendEvent::SendChannelReady { .. } => {},
1220+
MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1221+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
12241222
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
12251223
},
12261224
_ => {
@@ -1241,12 +1239,12 @@ pub fn do_test<Out: Output>(data: &[u8], underlying_out: Out, anchors: bool) {
12411239
} else {
12421240
for event in nodes[2].get_and_clear_pending_msg_events() {
12431241
match event {
1244-
events::MessageSendEvent::UpdateHTLCs { .. } => {},
1245-
events::MessageSendEvent::SendRevokeAndACK { .. } => {},
1246-
events::MessageSendEvent::SendChannelReestablish { .. } => {},
1247-
events::MessageSendEvent::SendChannelReady { .. } => {},
1248-
events::MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1249-
events::MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
1242+
MessageSendEvent::UpdateHTLCs { .. } => {},
1243+
MessageSendEvent::SendRevokeAndACK { .. } => {},
1244+
MessageSendEvent::SendChannelReestablish { .. } => {},
1245+
MessageSendEvent::SendChannelReady { .. } => {},
1246+
MessageSendEvent::SendAnnouncementSignatures { .. } => {},
1247+
MessageSendEvent::SendChannelUpdate { ref msg, .. } => {
12501248
assert_eq!(msg.contents.channel_flags & 2, 0); // The disable bit must never be set!
12511249
},
12521250
_ => {

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)