Skip to content

Commit cc320a5

Browse files
shaavanshaavan
committed
Update create_blinded_paths to accept Vec<MessageForwardNode>
To prepare for supporting both standard and compact blinded paths, this commit updates the `create_blinded_paths` function to take a `Vec<MessageForwardNode>` as input. This change ensures the function has all the information it needs to handle both types of blinded path creation. This refactor that sets the stage for upcoming enhancements.
1 parent 830ffa0 commit cc320a5

File tree

7 files changed

+24
-22
lines changed

7 files changed

+24
-22
lines changed

‎fuzz/src/chanmon_consistency.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ use bitcoin::hashes::sha256d::Hash as Sha256dHash;
3333
use bitcoin::hashes::Hash as TraitImport;
3434
use bitcoin::WPubkeyHash;
3535

36-
use lightning::blinded_path::message::{BlindedMessagePath, MessageContext};
36+
use lightning::blinded_path::message::{BlindedMessagePath, MessageContext, MessageForwardNode};
3737
use lightning::blinded_path::payment::{BlindedPaymentPath, ReceiveTlvs};
3838
use lightning::chain;
3939
use lightning::chain::chaininterface::{BroadcasterInterface, ConfirmationTarget, FeeEstimator};
@@ -139,7 +139,7 @@ impl MessageRouter for FuzzRouter {
139139
}
140140

141141
fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
142-
&self, _recipient: PublicKey, _context: MessageContext, _peers: Vec<PublicKey>,
142+
&self, _recipient: PublicKey, _context: MessageContext, _peers: Vec<MessageForwardNode>,
143143
_secp_ctx: &Secp256k1<T>,
144144
) -> Result<Vec<BlindedMessagePath>, ()> {
145145
unreachable!()

‎fuzz/src/full_stack.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ use bitcoin::hashes::Hash as _;
3030
use bitcoin::hex::FromHex;
3131
use bitcoin::WPubkeyHash;
3232

33-
use lightning::blinded_path::message::{BlindedMessagePath, MessageContext};
33+
use lightning::blinded_path::message::{BlindedMessagePath, MessageContext, MessageForwardNode};
3434
use lightning::blinded_path::payment::{BlindedPaymentPath, ReceiveTlvs};
3535
use lightning::chain;
3636
use lightning::chain::chaininterface::{BroadcasterInterface, ConfirmationTarget, FeeEstimator};
@@ -171,7 +171,7 @@ impl MessageRouter for FuzzRouter {
171171
}
172172

173173
fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
174-
&self, _recipient: PublicKey, _context: MessageContext, _peers: Vec<PublicKey>,
174+
&self, _recipient: PublicKey, _context: MessageContext, _peers: Vec<MessageForwardNode>,
175175
_secp_ctx: &Secp256k1<T>,
176176
) -> Result<Vec<BlindedMessagePath>, ()> {
177177
unreachable!()

‎fuzz/src/onion_message.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ use bitcoin::secp256k1::schnorr;
66
use bitcoin::secp256k1::{self, PublicKey, Scalar, Secp256k1, SecretKey};
77

88
use lightning::blinded_path::message::{
9-
AsyncPaymentsContext, BlindedMessagePath, MessageContext, OffersContext,
9+
AsyncPaymentsContext, BlindedMessagePath, MessageContext, MessageForwardNode, OffersContext,
1010
};
1111
use lightning::blinded_path::EmptyNodeIdLookUp;
1212
use lightning::ln::inbound_payment::ExpandedKey;
@@ -103,7 +103,7 @@ impl MessageRouter for TestMessageRouter {
103103
}
104104

105105
fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
106-
&self, _recipient: PublicKey, _context: MessageContext, _peers: Vec<PublicKey>,
106+
&self, _recipient: PublicKey, _context: MessageContext, _peers: Vec<MessageForwardNode>,
107107
_secp_ctx: &Secp256k1<T>,
108108
) -> Result<Vec<BlindedMessagePath>, ()> {
109109
unreachable!()

‎lightning-dns-resolver/src/lib.rs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,9 @@ mod test {
159159
use bitcoin::secp256k1::{self, PublicKey, Secp256k1};
160160
use bitcoin::Block;
161161

162-
use lightning::blinded_path::message::{BlindedMessagePath, MessageContext};
162+
use lightning::blinded_path::message::{
163+
BlindedMessagePath, MessageContext, MessageForwardNode,
164+
};
163165
use lightning::blinded_path::NodeIdLookUp;
164166
use lightning::events::{Event, PaymentPurpose};
165167
use lightning::ln::channelmanager::{PaymentId, Retry};
@@ -228,7 +230,7 @@ mod test {
228230
}
229231

230232
fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
231-
&self, recipient: PublicKey, context: MessageContext, _peers: Vec<PublicKey>,
233+
&self, recipient: PublicKey, context: MessageContext, _peers: Vec<MessageForwardNode>,
232234
secp_ctx: &Secp256k1<T>,
233235
) -> Result<Vec<BlindedMessagePath>, ()> {
234236
let keys = KeysManager::new(&[0; 32], 42, 43);

‎lightning/src/ln/channelmanager.rs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10883,7 +10883,10 @@ where
1088310883
.map(|(node_id, peer_state)| (node_id, peer_state.lock().unwrap()))
1088410884
.filter(|(_, peer)| peer.is_connected)
1088510885
.filter(|(_, peer)| peer.latest_features.supports_onion_messages())
10886-
.map(|(node_id, _)| *node_id)
10886+
.map(|(node_id, _)| MessageForwardNode {
10887+
node_id: *node_id,
10888+
short_channel_id: None
10889+
})
1088710890
.collect::<Vec<_>>();
1088810891

1088910892
self.message_router

‎lightning/src/onion_message/messenger.rs

Lines changed: 9 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ where
215215
/// # })
216216
/// # }
217217
/// # fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
218-
/// # &self, _recipient: PublicKey, _context: MessageContext, _peers: Vec<PublicKey>, _secp_ctx: &Secp256k1<T>
218+
/// # &self, _recipient: PublicKey, _context: MessageContext, _peers: Vec<MessageForwardNode>, _secp_ctx: &Secp256k1<T>
219219
/// # ) -> Result<Vec<BlindedMessagePath>, ()> {
220220
/// # unreachable!()
221221
/// # }
@@ -493,7 +493,7 @@ pub trait MessageRouter {
493493
/// Creates [`BlindedMessagePath`]s to the `recipient` node. The nodes in `peers` are assumed to
494494
/// be direct peers with the `recipient`.
495495
fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
496-
&self, recipient: PublicKey, context: MessageContext, peers: Vec<PublicKey>,
496+
&self, recipient: PublicKey, context: MessageContext, peers: Vec<MessageForwardNode>,
497497
secp_ctx: &Secp256k1<T>,
498498
) -> Result<Vec<BlindedMessagePath>, ()>;
499499

@@ -514,10 +514,6 @@ pub trait MessageRouter {
514514
&self, recipient: PublicKey, context: MessageContext, peers: Vec<MessageForwardNode>,
515515
secp_ctx: &Secp256k1<T>,
516516
) -> Result<Vec<BlindedMessagePath>, ()> {
517-
let peers = peers
518-
.into_iter()
519-
.map(|MessageForwardNode { node_id, short_channel_id: _ }| node_id)
520-
.collect();
521517
self.create_blinded_paths(recipient, context, peers, secp_ctx)
522518
}
523519
}
@@ -661,11 +657,9 @@ where
661657
}
662658

663659
pub(crate) fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
664-
network_graph: &G, recipient: PublicKey, context: MessageContext, peers: Vec<PublicKey>,
665-
entropy_source: &ES, secp_ctx: &Secp256k1<T>,
660+
network_graph: &G, recipient: PublicKey, context: MessageContext,
661+
peers: Vec<MessageForwardNode>, entropy_source: &ES, secp_ctx: &Secp256k1<T>,
666662
) -> Result<Vec<BlindedMessagePath>, ()> {
667-
let peers =
668-
peers.into_iter().map(|node_id| MessageForwardNode { node_id, short_channel_id: None });
669663
Self::create_blinded_paths_from_iter(
670664
network_graph,
671665
recipient,
@@ -706,7 +700,7 @@ where
706700
}
707701

708702
fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
709-
&self, recipient: PublicKey, context: MessageContext, peers: Vec<PublicKey>,
703+
&self, recipient: PublicKey, context: MessageContext, peers: Vec<MessageForwardNode>,
710704
secp_ctx: &Secp256k1<T>,
711705
) -> Result<Vec<BlindedMessagePath>, ()> {
712706
Self::create_blinded_paths(
@@ -1402,7 +1396,10 @@ where
14021396
message_recipients
14031397
.iter()
14041398
.filter(|(_, peer)| matches!(peer, OnionMessageRecipient::ConnectedPeer(_)))
1405-
.map(|(node_id, _)| *node_id)
1399+
.map(|(node_id, _)| MessageForwardNode {
1400+
node_id: *node_id,
1401+
short_channel_id: None,
1402+
})
14061403
.collect::<Vec<_>>()
14071404
};
14081405

‎lightning/src/util/test_utils.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -335,7 +335,7 @@ impl<'a> MessageRouter for TestMessageRouter<'a> {
335335
}
336336

337337
fn create_blinded_paths<T: secp256k1::Signing + secp256k1::Verification>(
338-
&self, recipient: PublicKey, context: MessageContext, peers: Vec<PublicKey>,
338+
&self, recipient: PublicKey, context: MessageContext, peers: Vec<MessageForwardNode>,
339339
secp_ctx: &Secp256k1<T>,
340340
) -> Result<Vec<BlindedMessagePath>, ()> {
341341
self.inner.create_blinded_paths(recipient, context, peers, secp_ctx)

0 commit comments

Comments
 (0)