1 use std::sync::{Arc, RwLock};
3 use bitcoin::secp256k1::PublicKey;
4 use lightning::ln::msgs::{ChannelAnnouncement, ChannelUpdate, Init, LightningError, NodeAnnouncement, QueryChannelRange, QueryShortChannelIds, ReplyChannelRange, ReplyShortChannelIdsEnd, RoutingMessageHandler};
5 use lightning::routing::gossip::{NetworkGraph, P2PGossipSync};
6 use lightning::util::events::{MessageSendEvent, MessageSendEventsProvider};
8 use tokio::sync::mpsc::error::TrySendError;
10 use crate::TestLogger;
11 use crate::types::{GossipMessage, GossipChainAccess};
12 use crate::verifier::ChainVerifier;
14 pub(crate) struct GossipCounter {
15 pub(crate) channel_announcements: u64,
16 pub(crate) channel_updates: u64,
17 pub(crate) channel_updates_without_htlc_max_msats: u64,
18 pub(crate) channel_announcements_with_mismatched_scripts: u64
22 pub(crate) fn new() -> Self {
24 channel_announcements: 0,
26 channel_updates_without_htlc_max_msats: 0,
27 channel_announcements_with_mismatched_scripts: 0,
32 pub(crate) struct GossipRouter {
33 native_router: P2PGossipSync<Arc<NetworkGraph<Arc<TestLogger>>>, GossipChainAccess, Arc<TestLogger>>,
34 pub(crate) counter: RwLock<GossipCounter>,
35 sender: mpsc::Sender<GossipMessage>,
39 pub(crate) fn new(network_graph: Arc<NetworkGraph<Arc<TestLogger>>>, sender: mpsc::Sender<GossipMessage>) -> Self {
41 native_router: P2PGossipSync::new(network_graph, Some(Arc::new(ChainVerifier::new())),
42 Arc::new(TestLogger::new())),
43 counter: RwLock::new(GossipCounter::new()),
49 impl MessageSendEventsProvider for GossipRouter {
50 fn get_and_clear_pending_msg_events(&self) -> Vec<MessageSendEvent> {
51 self.native_router.get_and_clear_pending_msg_events()
55 impl RoutingMessageHandler for GossipRouter {
56 fn handle_node_announcement(&self, msg: &NodeAnnouncement) -> Result<bool, LightningError> {
57 self.native_router.handle_node_announcement(msg)
60 fn handle_channel_announcement(&self, msg: &ChannelAnnouncement) -> Result<bool, LightningError> {
61 let mut counter = self.counter.write().unwrap();
63 let output_value = self.native_router.handle_channel_announcement(msg).map_err(|error| {
64 let error_string = format!("{:?}", error);
65 if error_string.contains("announced on an unknown chain"){
68 counter.channel_announcements_with_mismatched_scripts += 1;
72 counter.channel_announcements += 1;
73 let gossip_message = GossipMessage::ChannelAnnouncement(msg.clone());
74 if let Err(err) = self.sender.try_send(gossip_message) {
75 let gossip_message = match err { TrySendError::Full(msg)|TrySendError::Closed(msg) => msg };
76 tokio::task::block_in_place(move || { tokio::runtime::Handle::current().block_on(async move {
77 self.sender.send(gossip_message).await.unwrap();
84 fn handle_channel_update(&self, msg: &ChannelUpdate) -> Result<bool, LightningError> {
85 let output_value = self.native_router.handle_channel_update(msg)?;
87 let mut counter = self.counter.write().unwrap();
88 counter.channel_updates += 1;
89 let gossip_message = GossipMessage::ChannelUpdate(msg.clone());
91 if let Err(err) = self.sender.try_send(gossip_message) {
92 let gossip_message = match err { TrySendError::Full(msg)|TrySendError::Closed(msg) => msg };
93 tokio::task::block_in_place(move || { tokio::runtime::Handle::current().block_on(async move {
94 self.sender.send(gossip_message).await.unwrap();
101 fn get_next_channel_announcements(&self, starting_point: u64, batch_amount: u8) -> Vec<(ChannelAnnouncement, Option<ChannelUpdate>, Option<ChannelUpdate>)> {
102 self.native_router.get_next_channel_announcements(starting_point, batch_amount)
105 fn get_next_node_announcements(&self, starting_point: Option<&PublicKey>, batch_amount: u8) -> Vec<NodeAnnouncement> {
106 self.native_router.get_next_node_announcements(starting_point, batch_amount)
109 fn peer_connected(&self, their_node_id: &PublicKey, init: &Init) {
110 self.native_router.peer_connected(their_node_id, init)
113 fn handle_reply_channel_range(&self, their_node_id: &PublicKey, msg: ReplyChannelRange) -> Result<(), LightningError> {
114 self.native_router.handle_reply_channel_range(their_node_id, msg)
117 fn handle_reply_short_channel_ids_end(&self, their_node_id: &PublicKey, msg: ReplyShortChannelIdsEnd) -> Result<(), LightningError> {
118 self.native_router.handle_reply_short_channel_ids_end(their_node_id, msg)
121 fn handle_query_channel_range(&self, their_node_id: &PublicKey, msg: QueryChannelRange) -> Result<(), LightningError> {
122 self.native_router.handle_query_channel_range(their_node_id, msg)
125 fn handle_query_short_channel_ids(&self, their_node_id: &PublicKey, msg: QueryShortChannelIds) -> Result<(), LightningError> {
126 self.native_router.handle_query_short_channel_ids(their_node_id, msg)