d7185c999ee4356cd8f755cc2f319c986e3950bc
[rust-lightning] / lightning-invoice / src / utils.rs
1 //! Convenient utilities to create an invoice.
2
3 use {CreationError, Currency, Invoice, InvoiceBuilder, SignOrCreationError};
4 use payment::{InFlightHtlcs, Payer, Router};
5
6 use crate::{prelude::*, Description, InvoiceDescription, Sha256};
7 use bech32::ToBase32;
8 use bitcoin_hashes::{Hash, sha256};
9 use lightning::chain;
10 use lightning::chain::chaininterface::{BroadcasterInterface, FeeEstimator};
11 use lightning::chain::keysinterface::{Recipient, KeysInterface, Sign};
12 use lightning::ln::{PaymentHash, PaymentPreimage, PaymentSecret};
13 use lightning::ln::channelmanager::{ChannelDetails, ChannelManager, PaymentId, PaymentSendFailure, MIN_FINAL_CLTV_EXPIRY};
14 #[cfg(feature = "std")]
15 use lightning::ln::channelmanager::{PhantomRouteHints, MIN_CLTV_EXPIRY_DELTA};
16 use lightning::ln::inbound_payment::{create, create_from_hash, ExpandedKey};
17 use lightning::ln::msgs::LightningError;
18 use lightning::routing::gossip::{NetworkGraph, NodeId, RoutingFees};
19 use lightning::routing::router::{Route, RouteHint, RouteHintHop, RouteParameters, find_route, RouteHop};
20 use lightning::routing::scoring::{ChannelUsage, LockableScore, Score};
21 use lightning::util::logger::Logger;
22 use secp256k1::PublicKey;
23 use core::ops::Deref;
24 use core::time::Duration;
25 use sync::Mutex;
26
27 #[cfg(feature = "std")]
28 /// Utility to create an invoice that can be paid to one of multiple nodes, or a "phantom invoice."
29 /// See [`PhantomKeysManager`] for more information on phantom node payments.
30 ///
31 /// `phantom_route_hints` parameter:
32 /// * Contains channel info for all nodes participating in the phantom invoice
33 /// * Entries are retrieved from a call to [`ChannelManager::get_phantom_route_hints`] on each
34 ///   participating node
35 /// * It is fine to cache `phantom_route_hints` and reuse it across invoices, as long as the data is
36 ///   updated when a channel becomes disabled or closes
37 /// * Note that if too many channels are included in [`PhantomRouteHints::channels`], the invoice
38 ///   may be too long for QR code scanning. To fix this, `PhantomRouteHints::channels` may be pared
39 ///   down
40 ///
41 /// `payment_hash` can be specified if you have a specific need for a custom payment hash (see the difference
42 /// between [`ChannelManager::create_inbound_payment`] and [`ChannelManager::create_inbound_payment_for_hash`]).
43 /// If `None` is provided for `payment_hash`, then one will be created.
44 ///
45 /// `invoice_expiry_delta_secs` describes the number of seconds that the invoice is valid for
46 /// in excess of the current time.
47 ///
48 /// Note that the provided `keys_manager`'s `KeysInterface` implementation must support phantom
49 /// invoices in its `sign_invoice` implementation ([`PhantomKeysManager`] satisfies this
50 /// requirement).
51 ///
52 /// [`PhantomKeysManager`]: lightning::chain::keysinterface::PhantomKeysManager
53 /// [`ChannelManager::get_phantom_route_hints`]: lightning::ln::channelmanager::ChannelManager::get_phantom_route_hints
54 /// [`ChannelManager::create_inbound_payment`]: lightning::ln::channelmanager::ChannelManager::create_inbound_payment
55 /// [`ChannelManager::create_inbound_payment_for_hash`]: lightning::ln::channelmanager::ChannelManager::create_inbound_payment_for_hash
56 /// [`PhantomRouteHints::channels`]: lightning::ln::channelmanager::PhantomRouteHints::channels
57 pub fn create_phantom_invoice<Signer: Sign, K: Deref>(
58         amt_msat: Option<u64>, payment_hash: Option<PaymentHash>, description: String, invoice_expiry_delta_secs: u32,
59         phantom_route_hints: Vec<PhantomRouteHints>, keys_manager: K, network: Currency,
60 ) -> Result<Invoice, SignOrCreationError<()>> where K::Target: KeysInterface {
61         let description = Description::new(description).map_err(SignOrCreationError::CreationError)?;
62         let description = InvoiceDescription::Direct(&description,);
63         _create_phantom_invoice::<Signer, K>(
64                 amt_msat, payment_hash, description, invoice_expiry_delta_secs, phantom_route_hints, keys_manager, network,
65         )
66 }
67
68 #[cfg(feature = "std")]
69 /// Utility to create an invoice that can be paid to one of multiple nodes, or a "phantom invoice."
70 /// See [`PhantomKeysManager`] for more information on phantom node payments.
71 ///
72 /// `phantom_route_hints` parameter:
73 /// * Contains channel info for all nodes participating in the phantom invoice
74 /// * Entries are retrieved from a call to [`ChannelManager::get_phantom_route_hints`] on each
75 ///   participating node
76 /// * It is fine to cache `phantom_route_hints` and reuse it across invoices, as long as the data is
77 ///   updated when a channel becomes disabled or closes
78 /// * Note that if too many channels are included in [`PhantomRouteHints::channels`], the invoice
79 ///   may be too long for QR code scanning. To fix this, `PhantomRouteHints::channels` may be pared
80 ///   down
81 ///
82 /// `description_hash` is a SHA-256 hash of the description text
83 ///
84 /// `payment_hash` can be specified if you have a specific need for a custom payment hash (see the difference
85 /// between [`ChannelManager::create_inbound_payment`] and [`ChannelManager::create_inbound_payment_for_hash`]).
86 /// If `None` is provided for `payment_hash`, then one will be created.
87 ///
88 /// `invoice_expiry_delta_secs` describes the number of seconds that the invoice is valid for
89 /// in excess of the current time.
90 ///
91 /// Note that the provided `keys_manager`'s `KeysInterface` implementation must support phantom
92 /// invoices in its `sign_invoice` implementation ([`PhantomKeysManager`] satisfies this
93 /// requirement).
94 ///
95 /// [`PhantomKeysManager`]: lightning::chain::keysinterface::PhantomKeysManager
96 /// [`ChannelManager::get_phantom_route_hints`]: lightning::ln::channelmanager::ChannelManager::get_phantom_route_hints
97 /// [`ChannelManager::create_inbound_payment`]: lightning::ln::channelmanager::ChannelManager::create_inbound_payment
98 /// [`ChannelManager::create_inbound_payment_for_hash`]: lightning::ln::channelmanager::ChannelManager::create_inbound_payment_for_hash
99 /// [`PhantomRouteHints::channels`]: lightning::ln::channelmanager::PhantomRouteHints::channels
100 pub fn create_phantom_invoice_with_description_hash<Signer: Sign, K: Deref>(
101         amt_msat: Option<u64>, payment_hash: Option<PaymentHash>, invoice_expiry_delta_secs: u32,
102         description_hash: Sha256, phantom_route_hints: Vec<PhantomRouteHints>, keys_manager: K, network: Currency,
103 ) -> Result<Invoice, SignOrCreationError<()>> where K::Target: KeysInterface
104 {
105         _create_phantom_invoice::<Signer, K>(
106                 amt_msat, payment_hash, InvoiceDescription::Hash(&description_hash),
107                 invoice_expiry_delta_secs, phantom_route_hints, keys_manager, network,
108         )
109 }
110
111 #[cfg(feature = "std")]
112 fn _create_phantom_invoice<Signer: Sign, K: Deref>(
113         amt_msat: Option<u64>, payment_hash: Option<PaymentHash>, description: InvoiceDescription,
114         invoice_expiry_delta_secs: u32, phantom_route_hints: Vec<PhantomRouteHints>, keys_manager: K, network: Currency,
115 ) -> Result<Invoice, SignOrCreationError<()>> where K::Target: KeysInterface
116 {
117         use std::time::{SystemTime, UNIX_EPOCH};
118
119         if phantom_route_hints.len() == 0 {
120                 return Err(SignOrCreationError::CreationError(
121                         CreationError::MissingRouteHints,
122                 ));
123         }
124         let invoice = match description {
125                 InvoiceDescription::Direct(description) => {
126                         InvoiceBuilder::new(network).description(description.0.clone())
127                 }
128                 InvoiceDescription::Hash(hash) => InvoiceBuilder::new(network).description_hash(hash.0),
129         };
130
131         // If we ever see performance here being too slow then we should probably take this ExpandedKey as a parameter instead.
132         let keys = ExpandedKey::new(&keys_manager.get_inbound_payment_key_material());
133         let (payment_hash, payment_secret) = if let Some(payment_hash) = payment_hash {
134                 let payment_secret = create_from_hash(
135                         &keys,
136                         amt_msat,
137                         payment_hash,
138                         invoice_expiry_delta_secs,
139                         SystemTime::now()
140                                 .duration_since(UNIX_EPOCH)
141                                 .expect("Time must be > 1970")
142                                 .as_secs(),
143                 )
144                 .map_err(|_| SignOrCreationError::CreationError(CreationError::InvalidAmount))?;
145                 (payment_hash, payment_secret)
146         } else {
147                 create(
148                         &keys,
149                         amt_msat,
150                         invoice_expiry_delta_secs,
151                         &keys_manager,
152                         SystemTime::now()
153                                 .duration_since(UNIX_EPOCH)
154                                 .expect("Time must be > 1970")
155                                 .as_secs(),
156                 )
157                 .map_err(|_| SignOrCreationError::CreationError(CreationError::InvalidAmount))?
158         };
159
160         let mut invoice = invoice
161                 .current_timestamp()
162                 .payment_hash(Hash::from_slice(&payment_hash.0).unwrap())
163                 .payment_secret(payment_secret)
164                 .min_final_cltv_expiry(MIN_FINAL_CLTV_EXPIRY.into())
165                 .expiry_time(Duration::from_secs(invoice_expiry_delta_secs.into()));
166         if let Some(amt) = amt_msat {
167                 invoice = invoice.amount_milli_satoshis(amt);
168         }
169
170         for PhantomRouteHints { channels, phantom_scid, real_node_pubkey } in phantom_route_hints {
171                 let mut route_hints = filter_channels(channels, amt_msat);
172
173                 // If we have any public channel, the route hints from `filter_channels` will be empty.
174                 // In that case we create a RouteHint on which we will push a single hop with the phantom
175                 // route into the invoice, and let the sender find the path to the `real_node_pubkey`
176                 // node by looking at our public channels.
177                 if route_hints.is_empty() {
178                         route_hints.push(RouteHint(vec![]))
179                 }
180                 for mut route_hint in route_hints {
181                         route_hint.0.push(RouteHintHop {
182                                 src_node_id: real_node_pubkey,
183                                 short_channel_id: phantom_scid,
184                                 fees: RoutingFees {
185                                         base_msat: 0,
186                                         proportional_millionths: 0,
187                                 },
188                                 cltv_expiry_delta: MIN_CLTV_EXPIRY_DELTA,
189                                 htlc_minimum_msat: None,
190                                 htlc_maximum_msat: None,});
191                         invoice = invoice.private_route(route_hint.clone());
192                 }
193         }
194
195         let raw_invoice = match invoice.build_raw() {
196                 Ok(inv) => inv,
197                 Err(e) => return Err(SignOrCreationError::CreationError(e))
198         };
199         let hrp_str = raw_invoice.hrp.to_string();
200         let hrp_bytes = hrp_str.as_bytes();
201         let data_without_signature = raw_invoice.data.to_base32();
202         let signed_raw_invoice = raw_invoice.sign(|_| keys_manager.sign_invoice(hrp_bytes, &data_without_signature, Recipient::PhantomNode));
203         match signed_raw_invoice {
204                 Ok(inv) => Ok(Invoice::from_signed(inv).unwrap()),
205                 Err(e) => Err(SignOrCreationError::SignError(e))
206         }
207 }
208
209 #[cfg(feature = "std")]
210 /// Utility to construct an invoice. Generally, unless you want to do something like a custom
211 /// cltv_expiry, this is what you should be using to create an invoice. The reason being, this
212 /// method stores the invoice's payment secret and preimage in `ChannelManager`, so (a) the user
213 /// doesn't have to store preimage/payment secret information and (b) `ChannelManager` can verify
214 /// that the payment secret is valid when the invoice is paid.
215 ///
216 /// `invoice_expiry_delta_secs` describes the number of seconds that the invoice is valid for
217 /// in excess of the current time.
218 pub fn create_invoice_from_channelmanager<Signer: Sign, M: Deref, T: Deref, K: Deref, F: Deref, L: Deref>(
219         channelmanager: &ChannelManager<Signer, M, T, K, F, L>, keys_manager: K, network: Currency,
220         amt_msat: Option<u64>, description: String, invoice_expiry_delta_secs: u32
221 ) -> Result<Invoice, SignOrCreationError<()>>
222 where
223         M::Target: chain::Watch<Signer>,
224         T::Target: BroadcasterInterface,
225         K::Target: KeysInterface<Signer = Signer>,
226         F::Target: FeeEstimator,
227         L::Target: Logger,
228 {
229         use std::time::SystemTime;
230         let duration = SystemTime::now().duration_since(SystemTime::UNIX_EPOCH)
231                 .expect("for the foreseeable future this shouldn't happen");
232         create_invoice_from_channelmanager_and_duration_since_epoch(
233                 channelmanager, keys_manager, network, amt_msat, description, duration,
234                 invoice_expiry_delta_secs
235         )
236 }
237
238 #[cfg(feature = "std")]
239 /// Utility to construct an invoice. Generally, unless you want to do something like a custom
240 /// cltv_expiry, this is what you should be using to create an invoice. The reason being, this
241 /// method stores the invoice's payment secret and preimage in `ChannelManager`, so (a) the user
242 /// doesn't have to store preimage/payment secret information and (b) `ChannelManager` can verify
243 /// that the payment secret is valid when the invoice is paid.
244 /// Use this variant if you want to pass the `description_hash` to the invoice.
245 ///
246 /// `invoice_expiry_delta_secs` describes the number of seconds that the invoice is valid for
247 /// in excess of the current time.
248 pub fn create_invoice_from_channelmanager_with_description_hash<Signer: Sign, M: Deref, T: Deref, K: Deref, F: Deref, L: Deref>(
249         channelmanager: &ChannelManager<Signer, M, T, K, F, L>, keys_manager: K, network: Currency,
250         amt_msat: Option<u64>, description_hash: Sha256, invoice_expiry_delta_secs: u32
251 ) -> Result<Invoice, SignOrCreationError<()>>
252 where
253         M::Target: chain::Watch<Signer>,
254         T::Target: BroadcasterInterface,
255         K::Target: KeysInterface<Signer = Signer>,
256         F::Target: FeeEstimator,
257         L::Target: Logger,
258 {
259         use std::time::SystemTime;
260
261         let duration = SystemTime::now()
262                 .duration_since(SystemTime::UNIX_EPOCH)
263                 .expect("for the foreseeable future this shouldn't happen");
264
265         create_invoice_from_channelmanager_with_description_hash_and_duration_since_epoch(
266                 channelmanager, keys_manager, network, amt_msat,
267                 description_hash, duration, invoice_expiry_delta_secs
268         )
269 }
270
271 /// See [`create_invoice_from_channelmanager_with_description_hash`]
272 /// This version can be used in a `no_std` environment, where [`std::time::SystemTime`] is not
273 /// available and the current time is supplied by the caller.
274 pub fn create_invoice_from_channelmanager_with_description_hash_and_duration_since_epoch<Signer: Sign, M: Deref, T: Deref, K: Deref, F: Deref, L: Deref>(
275         channelmanager: &ChannelManager<Signer, M, T, K, F, L>, keys_manager: K, network: Currency,
276         amt_msat: Option<u64>, description_hash: Sha256, duration_since_epoch: Duration,
277         invoice_expiry_delta_secs: u32
278 ) -> Result<Invoice, SignOrCreationError<()>>
279 where
280         M::Target: chain::Watch<Signer>,
281         T::Target: BroadcasterInterface,
282         K::Target: KeysInterface<Signer = Signer>,
283         F::Target: FeeEstimator,
284         L::Target: Logger,
285 {
286         _create_invoice_from_channelmanager_and_duration_since_epoch(
287                 channelmanager, keys_manager, network, amt_msat,
288                 InvoiceDescription::Hash(&description_hash),
289                 duration_since_epoch, invoice_expiry_delta_secs
290         )
291 }
292
293 /// See [`create_invoice_from_channelmanager`]
294 /// This version can be used in a `no_std` environment, where [`std::time::SystemTime`] is not
295 /// available and the current time is supplied by the caller.
296 pub fn create_invoice_from_channelmanager_and_duration_since_epoch<Signer: Sign, M: Deref, T: Deref, K: Deref, F: Deref, L: Deref>(
297         channelmanager: &ChannelManager<Signer, M, T, K, F, L>, keys_manager: K, network: Currency,
298         amt_msat: Option<u64>, description: String, duration_since_epoch: Duration,
299         invoice_expiry_delta_secs: u32
300 ) -> Result<Invoice, SignOrCreationError<()>>
301 where
302         M::Target: chain::Watch<Signer>,
303         T::Target: BroadcasterInterface,
304         K::Target: KeysInterface<Signer = Signer>,
305         F::Target: FeeEstimator,
306         L::Target: Logger,
307 {
308         _create_invoice_from_channelmanager_and_duration_since_epoch(
309                 channelmanager, keys_manager, network, amt_msat,
310                 InvoiceDescription::Direct(
311                         &Description::new(description).map_err(SignOrCreationError::CreationError)?,
312                 ),
313                 duration_since_epoch, invoice_expiry_delta_secs
314         )
315 }
316
317 fn _create_invoice_from_channelmanager_and_duration_since_epoch<Signer: Sign, M: Deref, T: Deref, K: Deref, F: Deref, L: Deref>(
318         channelmanager: &ChannelManager<Signer, M, T, K, F, L>, keys_manager: K, network: Currency,
319         amt_msat: Option<u64>, description: InvoiceDescription, duration_since_epoch: Duration,
320         invoice_expiry_delta_secs: u32
321 ) -> Result<Invoice, SignOrCreationError<()>>
322 where
323         M::Target: chain::Watch<Signer>,
324         T::Target: BroadcasterInterface,
325         K::Target: KeysInterface<Signer = Signer>,
326         F::Target: FeeEstimator,
327         L::Target: Logger,
328 {
329         let route_hints = filter_channels(channelmanager.list_usable_channels(), amt_msat);
330
331         // `create_inbound_payment` only returns an error if the amount is greater than the total bitcoin
332         // supply.
333         let (payment_hash, payment_secret) = channelmanager
334                 .create_inbound_payment(amt_msat, invoice_expiry_delta_secs)
335                 .map_err(|()| SignOrCreationError::CreationError(CreationError::InvalidAmount))?;
336         let our_node_pubkey = channelmanager.get_our_node_id();
337
338         let invoice = match description {
339                 InvoiceDescription::Direct(description) => {
340                         InvoiceBuilder::new(network).description(description.0.clone())
341                 }
342                 InvoiceDescription::Hash(hash) => InvoiceBuilder::new(network).description_hash(hash.0),
343         };
344
345         let mut invoice = invoice
346                 .duration_since_epoch(duration_since_epoch)
347                 .payee_pub_key(our_node_pubkey)
348                 .payment_hash(Hash::from_slice(&payment_hash.0).unwrap())
349                 .payment_secret(payment_secret)
350                 .basic_mpp()
351                 .min_final_cltv_expiry(MIN_FINAL_CLTV_EXPIRY.into())
352                 .expiry_time(Duration::from_secs(invoice_expiry_delta_secs.into()));
353         if let Some(amt) = amt_msat {
354                 invoice = invoice.amount_milli_satoshis(amt);
355         }
356         for hint in route_hints {
357                 invoice = invoice.private_route(hint);
358         }
359
360         let raw_invoice = match invoice.build_raw() {
361                 Ok(inv) => inv,
362                 Err(e) => return Err(SignOrCreationError::CreationError(e))
363         };
364         let hrp_str = raw_invoice.hrp.to_string();
365         let hrp_bytes = hrp_str.as_bytes();
366         let data_without_signature = raw_invoice.data.to_base32();
367         let signed_raw_invoice = raw_invoice.sign(|_| keys_manager.sign_invoice(hrp_bytes, &data_without_signature, Recipient::Node));
368         match signed_raw_invoice {
369                 Ok(inv) => Ok(Invoice::from_signed(inv).unwrap()),
370                 Err(e) => Err(SignOrCreationError::SignError(e))
371         }
372 }
373
374 /// Filters the `channels` for an invoice, and returns the corresponding `RouteHint`s to include
375 /// in the invoice.
376 ///
377 /// The filtering is based on the following criteria:
378 /// * Only one channel per counterparty node
379 /// * Always select the channel with the highest inbound capacity per counterparty node
380 /// * Filter out channels with a lower inbound capacity than `min_inbound_capacity_msat`, if any
381 /// channel with a higher or equal inbound capacity than `min_inbound_capacity_msat` exists
382 /// * If any public channel exists, the returned `RouteHint`s will be empty, and the sender will
383 /// need to find the path by looking at the public channels instead
384 fn filter_channels(channels: Vec<ChannelDetails>, min_inbound_capacity_msat: Option<u64>) -> Vec<RouteHint>{
385         let mut filtered_channels: HashMap<PublicKey, &ChannelDetails> = HashMap::new();
386         let min_inbound_capacity = min_inbound_capacity_msat.unwrap_or(0);
387         let mut min_capacity_channel_exists = false;
388
389         for channel in channels.iter() {
390                 if channel.get_inbound_payment_scid().is_none() || channel.counterparty.forwarding_info.is_none() {
391                         continue;
392                 }
393
394                 if channel.is_public {
395                         // If any public channel exists, return no hints and let the sender
396                         // look at the public channels instead.
397                         return vec![]
398                 }
399
400                 if channel.inbound_capacity_msat >= min_inbound_capacity {
401                         min_capacity_channel_exists = true;
402                 };
403                 match filtered_channels.entry(channel.counterparty.node_id) {
404                         hash_map::Entry::Occupied(mut entry) => {
405                                 let current_max_capacity = entry.get().inbound_capacity_msat;
406                                 if channel.inbound_capacity_msat < current_max_capacity {
407                                         continue;
408                                 }
409                                 entry.insert(channel);
410                         }
411                         hash_map::Entry::Vacant(entry) => {
412                                 entry.insert(channel);
413                         }
414                 }
415         }
416
417         let route_hint_from_channel = |channel: &ChannelDetails| {
418                 let forwarding_info = channel.counterparty.forwarding_info.as_ref().unwrap();
419                 RouteHint(vec![RouteHintHop {
420                         src_node_id: channel.counterparty.node_id,
421                         short_channel_id: channel.get_inbound_payment_scid().unwrap(),
422                         fees: RoutingFees {
423                                 base_msat: forwarding_info.fee_base_msat,
424                                 proportional_millionths: forwarding_info.fee_proportional_millionths,
425                         },
426                         cltv_expiry_delta: forwarding_info.cltv_expiry_delta,
427                         htlc_minimum_msat: channel.inbound_htlc_minimum_msat,
428                         htlc_maximum_msat: channel.inbound_htlc_maximum_msat,}])
429         };
430         // If all channels are private, return the route hint for the highest inbound capacity channel
431         // per counterparty node. If channels with an higher inbound capacity than the
432         // min_inbound_capacity exists, filter out the channels with a lower capacity than that.
433         filtered_channels.into_iter()
434                 .filter(|(_counterparty_id, channel)| {
435                         min_capacity_channel_exists && channel.inbound_capacity_msat >= min_inbound_capacity ||
436                         !min_capacity_channel_exists
437                 })
438                 .map(|(_counterparty_id, channel)| route_hint_from_channel(&channel))
439                 .collect::<Vec<RouteHint>>()
440 }
441
442 /// A [`Router`] implemented using [`find_route`].
443 pub struct DefaultRouter<G: Deref<Target = NetworkGraph<L>>, L: Deref, S: Deref> where
444         L::Target: Logger,
445         S::Target: for <'a> LockableScore<'a>,
446 {
447         network_graph: G,
448         logger: L,
449         random_seed_bytes: Mutex<[u8; 32]>,
450         scorer: S
451 }
452
453 impl<G: Deref<Target = NetworkGraph<L>>, L: Deref, S: Deref> DefaultRouter<G, L, S> where
454         L::Target: Logger,
455         S::Target: for <'a> LockableScore<'a>,
456 {
457         /// Creates a new router using the given [`NetworkGraph`], a [`Logger`], and a randomness source
458         /// `random_seed_bytes`.
459         pub fn new(network_graph: G, logger: L, random_seed_bytes: [u8; 32], scorer: S) -> Self {
460                 let random_seed_bytes = Mutex::new(random_seed_bytes);
461                 Self { network_graph, logger, random_seed_bytes, scorer }
462         }
463 }
464
465 impl<G: Deref<Target = NetworkGraph<L>>, L: Deref, S: Deref> Router for DefaultRouter<G, L, S> where
466         L::Target: Logger,
467         S::Target: for <'a> LockableScore<'a>,
468 {
469         fn find_route(
470                 &self, payer: &PublicKey, params: &RouteParameters, _payment_hash: &PaymentHash,
471                 first_hops: Option<&[&ChannelDetails]>, inflight_htlcs: InFlightHtlcs
472         ) -> Result<Route, LightningError> {
473                 let random_seed_bytes = {
474                         let mut locked_random_seed_bytes = self.random_seed_bytes.lock().unwrap();
475                         *locked_random_seed_bytes = sha256::Hash::hash(&*locked_random_seed_bytes).into_inner();
476                         *locked_random_seed_bytes
477                 };
478
479                 find_route(
480                         payer, params, &self.network_graph, first_hops, &*self.logger,
481                         &ScorerAccountingForInFlightHtlcs::new(&mut self.scorer.lock(), inflight_htlcs),
482                         &random_seed_bytes
483                 )
484         }
485
486         fn notify_payment_path_failed(&self, path: &[&RouteHop], short_channel_id: u64) {
487                 self.scorer.lock().payment_path_failed(path, short_channel_id);
488         }
489
490         fn notify_payment_path_successful(&self, path: &[&RouteHop]) {
491                 self.scorer.lock().payment_path_successful(path);
492         }
493
494         fn notify_payment_probe_successful(&self, path: &[&RouteHop]) {
495                 self.scorer.lock().probe_successful(path);
496         }
497
498         fn notify_payment_probe_failed(&self, path: &[&RouteHop], short_channel_id: u64) {
499                 self.scorer.lock().probe_failed(path, short_channel_id);
500         }
501 }
502
503 impl<Signer: Sign, M: Deref, T: Deref, K: Deref, F: Deref, L: Deref> Payer for ChannelManager<Signer, M, T, K, F, L>
504 where
505         M::Target: chain::Watch<Signer>,
506         T::Target: BroadcasterInterface,
507         K::Target: KeysInterface<Signer = Signer>,
508         F::Target: FeeEstimator,
509         L::Target: Logger,
510 {
511         fn node_id(&self) -> PublicKey {
512                 self.get_our_node_id()
513         }
514
515         fn first_hops(&self) -> Vec<ChannelDetails> {
516                 self.list_usable_channels()
517         }
518
519         fn send_payment(
520                 &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>
521         ) -> Result<PaymentId, PaymentSendFailure> {
522                 self.send_payment(route, payment_hash, payment_secret)
523         }
524
525         fn send_spontaneous_payment(
526                 &self, route: &Route, payment_preimage: PaymentPreimage,
527         ) -> Result<PaymentId, PaymentSendFailure> {
528                 self.send_spontaneous_payment(route, Some(payment_preimage))
529                         .map(|(_, payment_id)| payment_id)
530         }
531
532         fn retry_payment(
533                 &self, route: &Route, payment_id: PaymentId
534         ) -> Result<(), PaymentSendFailure> {
535                 self.retry_payment(route, payment_id)
536         }
537
538         fn abandon_payment(&self, payment_id: PaymentId) {
539                 self.abandon_payment(payment_id)
540         }
541 }
542
543
544 /// Used to store information about all the HTLCs that are inflight across all payment attempts.
545 pub(crate) struct ScorerAccountingForInFlightHtlcs<'a, S: Score> {
546         scorer: &'a mut S,
547         /// Maps a channel's short channel id and its direction to the liquidity used up.
548         inflight_htlcs: InFlightHtlcs,
549 }
550
551 impl<'a, S: Score> ScorerAccountingForInFlightHtlcs<'a, S> {
552         pub(crate) fn new(scorer: &'a mut S, inflight_htlcs: InFlightHtlcs) -> Self {
553                 ScorerAccountingForInFlightHtlcs {
554                         scorer,
555                         inflight_htlcs
556                 }
557         }
558 }
559
560 #[cfg(c_bindings)]
561 impl<'a, S:Score> lightning::util::ser::Writeable for ScorerAccountingForInFlightHtlcs<'a, S> {
562         fn write<W: lightning::util::ser::Writer>(&self, writer: &mut W) -> Result<(), lightning::io::Error> { self.scorer.write(writer) }
563 }
564
565 impl<'a, S: Score> Score for ScorerAccountingForInFlightHtlcs<'a, S> {
566         fn channel_penalty_msat(&self, short_channel_id: u64, source: &NodeId, target: &NodeId, usage: ChannelUsage) -> u64 {
567                 if let Some(used_liqudity) = self.inflight_htlcs.used_liquidity_msat(
568                         source, target, short_channel_id
569                 ) {
570                         let usage = ChannelUsage {
571                                 inflight_htlc_msat: usage.inflight_htlc_msat + used_liqudity,
572                                 ..usage
573                         };
574
575                         self.scorer.channel_penalty_msat(short_channel_id, source, target, usage)
576                 } else {
577                         self.scorer.channel_penalty_msat(short_channel_id, source, target, usage)
578                 }
579         }
580
581         fn payment_path_failed(&mut self, _path: &[&RouteHop], _short_channel_id: u64) { unreachable!() }
582
583         fn payment_path_successful(&mut self, _path: &[&RouteHop]) { unreachable!() }
584
585         fn probe_failed(&mut self, _path: &[&RouteHop], _short_channel_id: u64) { unreachable!() }
586
587         fn probe_successful(&mut self, _path: &[&RouteHop]) { unreachable!() }
588 }
589
590
591 #[cfg(test)]
592 mod test {
593         use core::time::Duration;
594         use {Currency, Description, InvoiceDescription};
595         use bitcoin_hashes::Hash;
596         use bitcoin_hashes::sha256::Hash as Sha256;
597         use lightning::chain::keysinterface::PhantomKeysManager;
598         use lightning::ln::{PaymentPreimage, PaymentHash};
599         use lightning::ln::channelmanager::{self, PhantomRouteHints, MIN_FINAL_CLTV_EXPIRY};
600         use lightning::ln::functional_test_utils::*;
601         use lightning::ln::msgs::ChannelMessageHandler;
602         use lightning::routing::router::{PaymentParameters, RouteParameters, find_route};
603         use lightning::util::enforcing_trait_impls::EnforcingSigner;
604         use lightning::util::events::{MessageSendEvent, MessageSendEventsProvider, Event};
605         use lightning::util::test_utils;
606         use lightning::util::config::UserConfig;
607         use lightning::chain::keysinterface::KeysInterface;
608         use utils::create_invoice_from_channelmanager_and_duration_since_epoch;
609         use std::collections::HashSet;
610
611         #[test]
612         fn test_from_channelmanager() {
613                 let chanmon_cfgs = create_chanmon_cfgs(2);
614                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
615                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
616                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
617                 create_unannounced_chan_between_nodes_with_value(&nodes, 0, 1, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
618                 let non_default_invoice_expiry_secs = 4200;
619                 let invoice = create_invoice_from_channelmanager_and_duration_since_epoch(
620                         &nodes[1].node, nodes[1].keys_manager, Currency::BitcoinTestnet, Some(10_000), "test".to_string(),
621                         Duration::from_secs(1234567), non_default_invoice_expiry_secs).unwrap();
622                 assert_eq!(invoice.amount_pico_btc(), Some(100_000));
623                 assert_eq!(invoice.min_final_cltv_expiry(), MIN_FINAL_CLTV_EXPIRY as u64);
624                 assert_eq!(invoice.description(), InvoiceDescription::Direct(&Description("test".to_string())));
625                 assert_eq!(invoice.expiry_time(), Duration::from_secs(non_default_invoice_expiry_secs.into()));
626
627                 // Invoice SCIDs should always use inbound SCID aliases over the real channel ID, if one is
628                 // available.
629                 let chan = &nodes[1].node.list_usable_channels()[0];
630                 assert_eq!(invoice.route_hints().len(), 1);
631                 assert_eq!(invoice.route_hints()[0].0.len(), 1);
632                 assert_eq!(invoice.route_hints()[0].0[0].short_channel_id, chan.inbound_scid_alias.unwrap());
633
634                 assert_eq!(invoice.route_hints()[0].0[0].htlc_minimum_msat, chan.inbound_htlc_minimum_msat);
635                 assert_eq!(invoice.route_hints()[0].0[0].htlc_maximum_msat, chan.inbound_htlc_maximum_msat);
636
637                 let payment_params = PaymentParameters::from_node_id(invoice.recover_payee_pub_key())
638                         .with_features(invoice.features().unwrap().clone())
639                         .with_route_hints(invoice.route_hints());
640                 let route_params = RouteParameters {
641                         payment_params,
642                         final_value_msat: invoice.amount_milli_satoshis().unwrap(),
643                         final_cltv_expiry_delta: invoice.min_final_cltv_expiry() as u32,
644                 };
645                 let first_hops = nodes[0].node.list_usable_channels();
646                 let network_graph = &node_cfgs[0].network_graph;
647                 let logger = test_utils::TestLogger::new();
648                 let scorer = test_utils::TestScorer::with_penalty(0);
649                 let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
650                 let route = find_route(
651                         &nodes[0].node.get_our_node_id(), &route_params, &network_graph,
652                         Some(&first_hops.iter().collect::<Vec<_>>()), &logger, &scorer, &random_seed_bytes
653                 ).unwrap();
654
655                 let payment_event = {
656                         let mut payment_hash = PaymentHash([0; 32]);
657                         payment_hash.0.copy_from_slice(&invoice.payment_hash().as_ref()[0..32]);
658                         nodes[0].node.send_payment(&route, payment_hash, &Some(invoice.payment_secret().clone())).unwrap();
659                         let mut added_monitors = nodes[0].chain_monitor.added_monitors.lock().unwrap();
660                         assert_eq!(added_monitors.len(), 1);
661                         added_monitors.clear();
662
663                         let mut events = nodes[0].node.get_and_clear_pending_msg_events();
664                         assert_eq!(events.len(), 1);
665                         SendEvent::from_event(events.remove(0))
666
667                 };
668                 nodes[1].node.handle_update_add_htlc(&nodes[0].node.get_our_node_id(), &payment_event.msgs[0]);
669                 nodes[1].node.handle_commitment_signed(&nodes[0].node.get_our_node_id(), &payment_event.commitment_msg);
670                 let mut added_monitors = nodes[1].chain_monitor.added_monitors.lock().unwrap();
671                 assert_eq!(added_monitors.len(), 1);
672                 added_monitors.clear();
673                 let events = nodes[1].node.get_and_clear_pending_msg_events();
674                 assert_eq!(events.len(), 2);
675         }
676
677         #[test]
678         fn test_create_invoice_with_description_hash() {
679                 let chanmon_cfgs = create_chanmon_cfgs(2);
680                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
681                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
682                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
683                 let description_hash = crate::Sha256(Hash::hash("Testing description_hash".as_bytes()));
684                 let invoice = ::utils::create_invoice_from_channelmanager_with_description_hash_and_duration_since_epoch(
685                         &nodes[1].node, nodes[1].keys_manager, Currency::BitcoinTestnet, Some(10_000),
686                         description_hash, Duration::from_secs(1234567), 3600
687                 ).unwrap();
688                 assert_eq!(invoice.amount_pico_btc(), Some(100_000));
689                 assert_eq!(invoice.min_final_cltv_expiry(), MIN_FINAL_CLTV_EXPIRY as u64);
690                 assert_eq!(invoice.description(), InvoiceDescription::Hash(&crate::Sha256(Sha256::hash("Testing description_hash".as_bytes()))));
691         }
692
693         #[test]
694         fn test_hints_includes_single_channels_to_nodes() {
695                 let chanmon_cfgs = create_chanmon_cfgs(3);
696                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
697                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
698                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
699
700                 let chan_1_0 = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 0, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
701                 let chan_2_0 = create_unannounced_chan_between_nodes_with_value(&nodes, 2, 0, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
702
703                 let mut scid_aliases = HashSet::new();
704                 scid_aliases.insert(chan_1_0.0.short_channel_id_alias.unwrap());
705                 scid_aliases.insert(chan_2_0.0.short_channel_id_alias.unwrap());
706
707                 match_invoice_routes(Some(5000), &nodes[0], scid_aliases);
708         }
709
710         #[test]
711         fn test_hints_has_only_highest_inbound_capacity_channel() {
712                 let chanmon_cfgs = create_chanmon_cfgs(2);
713                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
714                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
715                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
716                 let _chan_1_0_low_inbound_capacity = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 0, 100_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
717                 let chan_1_0_high_inbound_capacity = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 0, 10_000_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
718                 let _chan_1_0_medium_inbound_capacity = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 0, 1_000_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
719
720                 let mut scid_aliases = HashSet::new();
721                 scid_aliases.insert(chan_1_0_high_inbound_capacity.0.short_channel_id_alias.unwrap());
722
723                 match_invoice_routes(Some(5000), &nodes[0], scid_aliases);
724         }
725
726         #[test]
727         fn test_forwarding_info_not_assigned_channel_excluded_from_hints() {
728                 let chanmon_cfgs = create_chanmon_cfgs(3);
729                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
730                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
731                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
732                 let chan_1_0 = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 0, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
733
734                 // Create an unannonced channel between `nodes[2]` and `nodes[0]`, for which the
735                 // `msgs::ChannelUpdate` is never handled for the node(s). As the `msgs::ChannelUpdate`
736                 // is never handled, the `channel.counterparty.forwarding_info` is never assigned.
737                 let mut private_chan_cfg = UserConfig::default();
738                 private_chan_cfg.channel_handshake_config.announced_channel = false;
739                 let temporary_channel_id = nodes[2].node.create_channel(nodes[0].node.get_our_node_id(), 1_000_000, 500_000_000, 42, Some(private_chan_cfg)).unwrap();
740                 let open_channel = get_event_msg!(nodes[2], MessageSendEvent::SendOpenChannel, nodes[0].node.get_our_node_id());
741                 nodes[0].node.handle_open_channel(&nodes[2].node.get_our_node_id(), channelmanager::provided_init_features(), &open_channel);
742                 let accept_channel = get_event_msg!(nodes[0], MessageSendEvent::SendAcceptChannel, nodes[2].node.get_our_node_id());
743                 nodes[2].node.handle_accept_channel(&nodes[0].node.get_our_node_id(), channelmanager::provided_init_features(), &accept_channel);
744
745                 let tx = sign_funding_transaction(&nodes[2], &nodes[0], 1_000_000, temporary_channel_id);
746
747                 let conf_height = core::cmp::max(nodes[2].best_block_info().1 + 1, nodes[0].best_block_info().1 + 1);
748                 confirm_transaction_at(&nodes[2], &tx, conf_height);
749                 connect_blocks(&nodes[2], CHAN_CONFIRM_DEPTH - 1);
750                 confirm_transaction_at(&nodes[0], &tx, conf_height);
751                 connect_blocks(&nodes[0], CHAN_CONFIRM_DEPTH - 1);
752                 let as_channel_ready = get_event_msg!(nodes[2], MessageSendEvent::SendChannelReady, nodes[0].node.get_our_node_id());
753                 nodes[2].node.handle_channel_ready(&nodes[0].node.get_our_node_id(), &get_event_msg!(nodes[0], MessageSendEvent::SendChannelReady, nodes[2].node.get_our_node_id()));
754                 get_event_msg!(nodes[2], MessageSendEvent::SendChannelUpdate, nodes[0].node.get_our_node_id());
755                 nodes[0].node.handle_channel_ready(&nodes[2].node.get_our_node_id(), &as_channel_ready);
756                 get_event_msg!(nodes[0], MessageSendEvent::SendChannelUpdate, nodes[2].node.get_our_node_id());
757
758                 // As `msgs::ChannelUpdate` was never handled for the participating node(s) of the second
759                 // channel, the channel will never be assigned any `counterparty.forwarding_info`.
760                 // Therefore only `chan_1_0` should be included in the hints.
761                 let mut scid_aliases = HashSet::new();
762                 scid_aliases.insert(chan_1_0.0.short_channel_id_alias.unwrap());
763                 match_invoice_routes(Some(5000), &nodes[0], scid_aliases);
764         }
765
766         #[test]
767         fn test_no_hints_if_a_mix_between_public_and_private_channel_exists() {
768                 let chanmon_cfgs = create_chanmon_cfgs(3);
769                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
770                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
771                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
772                 let _chan_1_0 = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 0, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
773
774                 let chan_2_0 = create_announced_chan_between_nodes_with_value(&nodes, 2, 0, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
775                 nodes[2].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &chan_2_0.1);
776                 nodes[0].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan_2_0.0);
777
778                 // Ensure that the invoice doesn't include any route hints for any of `nodes[0]` channels,
779                 // even though all channels between `nodes[1]` and `nodes[0]` are private, as there is a
780                 // public channel between `nodes[2]` and `nodes[0]`
781                 match_invoice_routes(Some(5000), &nodes[0], HashSet::new());
782         }
783
784         #[test]
785         fn test_only_public_channels_includes_no_channels_in_hints() {
786                 let chanmon_cfgs = create_chanmon_cfgs(3);
787                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
788                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
789                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
790                 let chan_1_0 = create_announced_chan_between_nodes_with_value(&nodes, 1, 0, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
791                 nodes[0].node.handle_channel_update(&nodes[1].node.get_our_node_id(), &chan_1_0.0);
792                 nodes[1].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &chan_1_0.1);
793
794                 let chan_2_0 = create_announced_chan_between_nodes_with_value(&nodes, 2, 0, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
795                 nodes[2].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &chan_2_0.1);
796                 nodes[0].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan_2_0.0);
797
798                 // As all of `nodes[0]` channels are public, no channels should be included in the hints
799                 match_invoice_routes(Some(5000), &nodes[0], HashSet::new());
800         }
801
802         #[test]
803         fn test_channels_with_lower_inbound_capacity_than_invoice_amt_hints_filtering() {
804                 let chanmon_cfgs = create_chanmon_cfgs(3);
805                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
806                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
807                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
808                 let chan_1_0 = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 0, 100_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
809                 let chan_2_0 = create_unannounced_chan_between_nodes_with_value(&nodes, 2, 0, 1_000_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
810
811                 // As the invoice amt is 1 msat above chan_1_0's inbound capacity, it shouldn't be included
812                 let mut scid_aliases_99_000_001_msat = HashSet::new();
813                 scid_aliases_99_000_001_msat.insert(chan_2_0.0.short_channel_id_alias.unwrap());
814
815                 match_invoice_routes(Some(99_000_001), &nodes[0], scid_aliases_99_000_001_msat);
816
817                 // As the invoice amt is exactly at chan_1_0's inbound capacity, it should be included
818                 let mut scid_aliases_99_000_000_msat = HashSet::new();
819                 scid_aliases_99_000_000_msat.insert(chan_1_0.0.short_channel_id_alias.unwrap());
820                 scid_aliases_99_000_000_msat.insert(chan_2_0.0.short_channel_id_alias.unwrap());
821
822                 match_invoice_routes(Some(99_000_000), &nodes[0], scid_aliases_99_000_000_msat);
823
824                 // As the invoice amt is above all channels' inbound capacity, they will still be included
825                 let mut scid_aliases_2_000_000_000_msat = HashSet::new();
826                 scid_aliases_2_000_000_000_msat.insert(chan_1_0.0.short_channel_id_alias.unwrap());
827                 scid_aliases_2_000_000_000_msat.insert(chan_2_0.0.short_channel_id_alias.unwrap());
828
829                 match_invoice_routes(Some(2_000_000_000), &nodes[0], scid_aliases_2_000_000_000_msat);
830
831                 // An invoice with no specified amount should include all channels in the route hints.
832                 let mut scid_aliases_no_specified_amount = HashSet::new();
833                 scid_aliases_no_specified_amount.insert(chan_1_0.0.short_channel_id_alias.unwrap());
834                 scid_aliases_no_specified_amount.insert(chan_2_0.0.short_channel_id_alias.unwrap());
835
836                 match_invoice_routes(None, &nodes[0], scid_aliases_no_specified_amount);
837         }
838
839         fn match_invoice_routes<'a, 'b: 'a, 'c: 'b>(
840                 invoice_amt: Option<u64>,
841                 invoice_node: &Node<'a, 'b, 'c>,
842                 mut chan_ids_to_match: HashSet<u64>
843         ) {
844                 let invoice = create_invoice_from_channelmanager_and_duration_since_epoch(
845                         &invoice_node.node, invoice_node.keys_manager, Currency::BitcoinTestnet, invoice_amt, "test".to_string(),
846                         Duration::from_secs(1234567), 3600).unwrap();
847                 let hints = invoice.private_routes();
848
849                 for hint in hints {
850                         let hint_short_chan_id = (hint.0).0[0].short_channel_id;
851                         assert!(chan_ids_to_match.remove(&hint_short_chan_id));
852                 }
853                 assert!(chan_ids_to_match.is_empty(), "Unmatched short channel ids: {:?}", chan_ids_to_match);
854         }
855
856         #[test]
857         #[cfg(feature = "std")]
858         fn test_multi_node_receive() {
859                 do_test_multi_node_receive(true);
860                 do_test_multi_node_receive(false);
861         }
862
863         #[cfg(feature = "std")]
864         fn do_test_multi_node_receive(user_generated_pmt_hash: bool) {
865                 let mut chanmon_cfgs = create_chanmon_cfgs(3);
866                 let seed_1 = [42 as u8; 32];
867                 let seed_2 = [43 as u8; 32];
868                 let cross_node_seed = [44 as u8; 32];
869                 chanmon_cfgs[1].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
870                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
871                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
872                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
873                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
874                 let chan_0_1 = create_announced_chan_between_nodes_with_value(&nodes, 0, 1, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
875                 nodes[0].node.handle_channel_update(&nodes[1].node.get_our_node_id(), &chan_0_1.1);
876                 nodes[1].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &chan_0_1.0);
877                 let chan_0_2 = create_announced_chan_between_nodes_with_value(&nodes, 0, 2, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
878                 nodes[0].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan_0_2.1);
879                 nodes[2].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &chan_0_2.0);
880
881                 let payment_amt = 10_000;
882                 let route_hints = vec![
883                         nodes[1].node.get_phantom_route_hints(),
884                         nodes[2].node.get_phantom_route_hints(),
885                 ];
886
887                 let user_payment_preimage = PaymentPreimage([1; 32]);
888                 let payment_hash = if user_generated_pmt_hash {
889                         Some(PaymentHash(Sha256::hash(&user_payment_preimage.0[..]).into_inner()))
890                 } else {
891                         None
892                 };
893                 let non_default_invoice_expiry_secs = 4200;
894
895                 let invoice =
896                         ::utils::create_phantom_invoice::<EnforcingSigner, &test_utils::TestKeysInterface>(
897                                 Some(payment_amt), payment_hash, "test".to_string(), non_default_invoice_expiry_secs,
898                                 route_hints, &nodes[1].keys_manager, Currency::BitcoinTestnet
899                         ).unwrap();
900                 let (payment_hash, payment_secret) = (PaymentHash(invoice.payment_hash().into_inner()), *invoice.payment_secret());
901                 let payment_preimage = if user_generated_pmt_hash {
902                         user_payment_preimage
903                 } else {
904                         nodes[1].node.get_payment_preimage(payment_hash, payment_secret).unwrap()
905                 };
906
907                 assert_eq!(invoice.min_final_cltv_expiry(), MIN_FINAL_CLTV_EXPIRY as u64);
908                 assert_eq!(invoice.description(), InvoiceDescription::Direct(&Description("test".to_string())));
909                 assert_eq!(invoice.route_hints().len(), 2);
910                 assert_eq!(invoice.expiry_time(), Duration::from_secs(non_default_invoice_expiry_secs.into()));
911                 assert!(!invoice.features().unwrap().supports_basic_mpp());
912
913                 let payment_params = PaymentParameters::from_node_id(invoice.recover_payee_pub_key())
914                         .with_features(invoice.features().unwrap().clone())
915                         .with_route_hints(invoice.route_hints());
916                 let params = RouteParameters {
917                         payment_params,
918                         final_value_msat: invoice.amount_milli_satoshis().unwrap(),
919                         final_cltv_expiry_delta: invoice.min_final_cltv_expiry() as u32,
920                 };
921                 let first_hops = nodes[0].node.list_usable_channels();
922                 let network_graph = &node_cfgs[0].network_graph;
923                 let logger = test_utils::TestLogger::new();
924                 let scorer = test_utils::TestScorer::with_penalty(0);
925                 let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
926                 let route = find_route(
927                         &nodes[0].node.get_our_node_id(), &params, &network_graph,
928                         Some(&first_hops.iter().collect::<Vec<_>>()), &logger, &scorer, &random_seed_bytes
929                 ).unwrap();
930                 let (payment_event, fwd_idx) = {
931                         let mut payment_hash = PaymentHash([0; 32]);
932                         payment_hash.0.copy_from_slice(&invoice.payment_hash().as_ref()[0..32]);
933                         nodes[0].node.send_payment(&route, payment_hash, &Some(invoice.payment_secret().clone())).unwrap();
934                         let mut added_monitors = nodes[0].chain_monitor.added_monitors.lock().unwrap();
935                         assert_eq!(added_monitors.len(), 1);
936                         added_monitors.clear();
937
938                         let mut events = nodes[0].node.get_and_clear_pending_msg_events();
939                         assert_eq!(events.len(), 1);
940                         let fwd_idx = match events[0] {
941                                 MessageSendEvent::UpdateHTLCs { node_id, .. } => {
942                                         if node_id == nodes[1].node.get_our_node_id() {
943                                                 1
944                                         } else { 2 }
945                                 },
946                                 _ => panic!("Unexpected event")
947                         };
948                         (SendEvent::from_event(events.remove(0)), fwd_idx)
949                 };
950                 nodes[fwd_idx].node.handle_update_add_htlc(&nodes[0].node.get_our_node_id(), &payment_event.msgs[0]);
951                 commitment_signed_dance!(nodes[fwd_idx], nodes[0], &payment_event.commitment_msg, false, true);
952
953                 // Note that we have to "forward pending HTLCs" twice before we see the PaymentReceived as
954                 // this "emulates" the payment taking two hops, providing some privacy to make phantom node
955                 // payments "look real" by taking more time.
956                 expect_pending_htlcs_forwardable_ignore!(nodes[fwd_idx]);
957                 nodes[fwd_idx].node.process_pending_htlc_forwards();
958                 expect_pending_htlcs_forwardable_ignore!(nodes[fwd_idx]);
959                 nodes[fwd_idx].node.process_pending_htlc_forwards();
960
961                 let payment_preimage_opt = if user_generated_pmt_hash { None } else { Some(payment_preimage) };
962                 expect_payment_received!(&nodes[fwd_idx], payment_hash, payment_secret, payment_amt, payment_preimage_opt);
963                 do_claim_payment_along_route(&nodes[0], &vec!(&vec!(&nodes[fwd_idx])[..]), false, payment_preimage);
964                 let events = nodes[0].node.get_and_clear_pending_events();
965                 assert_eq!(events.len(), 2);
966                 match events[0] {
967                         Event::PaymentSent { payment_preimage: ref ev_preimage, payment_hash: ref ev_hash, ref fee_paid_msat, .. } => {
968                                 assert_eq!(payment_preimage, *ev_preimage);
969                                 assert_eq!(payment_hash, *ev_hash);
970                                 assert_eq!(fee_paid_msat, &Some(0));
971                         },
972                         _ => panic!("Unexpected event")
973                 }
974                 match events[1] {
975                         Event::PaymentPathSuccessful { payment_hash: hash, .. } => {
976                                 assert_eq!(hash, Some(payment_hash));
977                         },
978                         _ => panic!("Unexpected event")
979                 }
980         }
981
982         #[test]
983         #[cfg(feature = "std")]
984         fn test_multi_node_hints_has_htlc_min_max_values() {
985                 let mut chanmon_cfgs = create_chanmon_cfgs(3);
986                 let seed_1 = [42 as u8; 32];
987                 let seed_2 = [43 as u8; 32];
988                 let cross_node_seed = [44 as u8; 32];
989                 chanmon_cfgs[1].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
990                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
991                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
992                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
993                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
994
995                 create_unannounced_chan_between_nodes_with_value(&nodes, 0, 1, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
996                 create_unannounced_chan_between_nodes_with_value(&nodes, 0, 2, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
997
998                 let payment_amt = 20_000;
999                 let (payment_hash, _payment_secret) = nodes[1].node.create_inbound_payment(Some(payment_amt), 3600).unwrap();
1000                 let route_hints = vec![
1001                         nodes[1].node.get_phantom_route_hints(),
1002                         nodes[2].node.get_phantom_route_hints(),
1003                 ];
1004
1005                 let invoice = ::utils::create_phantom_invoice::<EnforcingSigner, &test_utils::TestKeysInterface>(Some(payment_amt), Some(payment_hash), "test".to_string(), 3600, route_hints, &nodes[1].keys_manager, Currency::BitcoinTestnet).unwrap();
1006
1007                 let chan_0_1 = &nodes[1].node.list_usable_channels()[0];
1008                 assert_eq!(invoice.route_hints()[0].0[0].htlc_minimum_msat, chan_0_1.inbound_htlc_minimum_msat);
1009                 assert_eq!(invoice.route_hints()[0].0[0].htlc_maximum_msat, chan_0_1.inbound_htlc_maximum_msat);
1010
1011                 let chan_0_2 = &nodes[2].node.list_usable_channels()[0];
1012                 assert_eq!(invoice.route_hints()[1].0[0].htlc_minimum_msat, chan_0_2.inbound_htlc_minimum_msat);
1013                 assert_eq!(invoice.route_hints()[1].0[0].htlc_maximum_msat, chan_0_2.inbound_htlc_maximum_msat);
1014         }
1015
1016         #[test]
1017         #[cfg(feature = "std")]
1018         fn create_phantom_invoice_with_description_hash() {
1019                 let chanmon_cfgs = create_chanmon_cfgs(3);
1020                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
1021                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
1022                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
1023
1024                 let payment_amt = 20_000;
1025                 let route_hints = vec![
1026                         nodes[1].node.get_phantom_route_hints(),
1027                         nodes[2].node.get_phantom_route_hints(),
1028                 ];
1029
1030                 let description_hash = crate::Sha256(Hash::hash("Description hash phantom invoice".as_bytes()));
1031                 let non_default_invoice_expiry_secs = 4200;
1032                 let invoice = ::utils::create_phantom_invoice_with_description_hash::<
1033                         EnforcingSigner, &test_utils::TestKeysInterface,
1034                 >(
1035                         Some(payment_amt), None, non_default_invoice_expiry_secs, description_hash,
1036                         route_hints, &nodes[1].keys_manager, Currency::BitcoinTestnet
1037                 )
1038                 .unwrap();
1039                 assert_eq!(invoice.amount_pico_btc(), Some(200_000));
1040                 assert_eq!(invoice.min_final_cltv_expiry(), MIN_FINAL_CLTV_EXPIRY as u64);
1041                 assert_eq!(invoice.expiry_time(), Duration::from_secs(non_default_invoice_expiry_secs.into()));
1042                 assert_eq!(invoice.description(), InvoiceDescription::Hash(&crate::Sha256(Sha256::hash("Description hash phantom invoice".as_bytes()))));
1043         }
1044
1045         #[test]
1046         #[cfg(feature = "std")]
1047         fn test_multi_node_hints_includes_single_channels_to_participating_nodes() {
1048                 let mut chanmon_cfgs = create_chanmon_cfgs(3);
1049                 let seed_1 = [42 as u8; 32];
1050                 let seed_2 = [43 as u8; 32];
1051                 let cross_node_seed = [44 as u8; 32];
1052                 chanmon_cfgs[1].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
1053                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
1054                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
1055                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
1056                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
1057
1058                 let chan_0_1 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 1, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1059                 let chan_0_2 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 2, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1060
1061                 let mut scid_aliases = HashSet::new();
1062                 scid_aliases.insert(chan_0_1.0.short_channel_id_alias.unwrap());
1063                 scid_aliases.insert(chan_0_2.0.short_channel_id_alias.unwrap());
1064
1065                 match_multi_node_invoice_routes(
1066                         Some(10_000),
1067                         &nodes[1],
1068                         vec![&nodes[1], &nodes[2],],
1069                         scid_aliases,
1070                         false
1071                 );
1072         }
1073
1074         #[test]
1075         #[cfg(feature = "std")]
1076         fn test_multi_node_hints_includes_one_channel_of_each_counterparty_nodes_per_participating_node() {
1077                 let mut chanmon_cfgs = create_chanmon_cfgs(4);
1078                 let seed_1 = [42 as u8; 32];
1079                 let seed_2 = [43 as u8; 32];
1080                 let cross_node_seed = [44 as u8; 32];
1081                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
1082                 chanmon_cfgs[3].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
1083                 let node_cfgs = create_node_cfgs(4, &chanmon_cfgs);
1084                 let node_chanmgrs = create_node_chanmgrs(4, &node_cfgs, &[None, None, None, None]);
1085                 let nodes = create_network(4, &node_cfgs, &node_chanmgrs);
1086
1087                 let chan_0_2 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 2, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1088                 let chan_0_3 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 3, 1000000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1089                 let chan_1_3 = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 3, 3_000_000, 10005, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1090
1091                 let mut scid_aliases = HashSet::new();
1092                 scid_aliases.insert(chan_0_2.0.short_channel_id_alias.unwrap());
1093                 scid_aliases.insert(chan_0_3.0.short_channel_id_alias.unwrap());
1094                 scid_aliases.insert(chan_1_3.0.short_channel_id_alias.unwrap());
1095
1096                 match_multi_node_invoice_routes(
1097                         Some(10_000),
1098                         &nodes[2],
1099                         vec![&nodes[2], &nodes[3],],
1100                         scid_aliases,
1101                         false
1102                 );
1103         }
1104
1105         #[test]
1106         #[cfg(feature = "std")]
1107         fn test_multi_node_forwarding_info_not_assigned_channel_excluded_from_hints() {
1108                 let mut chanmon_cfgs = create_chanmon_cfgs(4);
1109                 let seed_1 = [42 as u8; 32];
1110                 let seed_2 = [43 as u8; 32];
1111                 let cross_node_seed = [44 as u8; 32];
1112                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
1113                 chanmon_cfgs[3].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
1114                 let node_cfgs = create_node_cfgs(4, &chanmon_cfgs);
1115                 let node_chanmgrs = create_node_chanmgrs(4, &node_cfgs, &[None, None, None, None]);
1116                 let nodes = create_network(4, &node_cfgs, &node_chanmgrs);
1117
1118                 let chan_0_2 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 2, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1119                 let chan_0_3 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 3, 1000000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1120
1121                 // Create an unannonced channel between `nodes[1]` and `nodes[3]`, for which the
1122                 // `msgs::ChannelUpdate` is never handled for the node(s). As the `msgs::ChannelUpdate`
1123                 // is never handled, the `channel.counterparty.forwarding_info` is never assigned.
1124                 let mut private_chan_cfg = UserConfig::default();
1125                 private_chan_cfg.channel_handshake_config.announced_channel = false;
1126                 let temporary_channel_id = nodes[1].node.create_channel(nodes[3].node.get_our_node_id(), 1_000_000, 500_000_000, 42, Some(private_chan_cfg)).unwrap();
1127                 let open_channel = get_event_msg!(nodes[1], MessageSendEvent::SendOpenChannel, nodes[3].node.get_our_node_id());
1128                 nodes[3].node.handle_open_channel(&nodes[1].node.get_our_node_id(), channelmanager::provided_init_features(), &open_channel);
1129                 let accept_channel = get_event_msg!(nodes[3], MessageSendEvent::SendAcceptChannel, nodes[1].node.get_our_node_id());
1130                 nodes[1].node.handle_accept_channel(&nodes[3].node.get_our_node_id(), channelmanager::provided_init_features(), &accept_channel);
1131
1132                 let tx = sign_funding_transaction(&nodes[1], &nodes[3], 1_000_000, temporary_channel_id);
1133
1134                 let conf_height = core::cmp::max(nodes[1].best_block_info().1 + 1, nodes[3].best_block_info().1 + 1);
1135                 confirm_transaction_at(&nodes[1], &tx, conf_height);
1136                 connect_blocks(&nodes[1], CHAN_CONFIRM_DEPTH - 1);
1137                 confirm_transaction_at(&nodes[3], &tx, conf_height);
1138                 connect_blocks(&nodes[3], CHAN_CONFIRM_DEPTH - 1);
1139                 let as_channel_ready = get_event_msg!(nodes[1], MessageSendEvent::SendChannelReady, nodes[3].node.get_our_node_id());
1140                 nodes[1].node.handle_channel_ready(&nodes[3].node.get_our_node_id(), &get_event_msg!(nodes[3], MessageSendEvent::SendChannelReady, nodes[1].node.get_our_node_id()));
1141                 get_event_msg!(nodes[1], MessageSendEvent::SendChannelUpdate, nodes[3].node.get_our_node_id());
1142                 nodes[3].node.handle_channel_ready(&nodes[1].node.get_our_node_id(), &as_channel_ready);
1143                 get_event_msg!(nodes[3], MessageSendEvent::SendChannelUpdate, nodes[1].node.get_our_node_id());
1144
1145                 // As `msgs::ChannelUpdate` was never handled for the participating node(s) of the third
1146                 // channel, the channel will never be assigned any `counterparty.forwarding_info`.
1147                 // Therefore only `chan_0_3` should be included in the hints for `nodes[3]`.
1148                 let mut scid_aliases = HashSet::new();
1149                 scid_aliases.insert(chan_0_2.0.short_channel_id_alias.unwrap());
1150                 scid_aliases.insert(chan_0_3.0.short_channel_id_alias.unwrap());
1151
1152                 match_multi_node_invoice_routes(
1153                         Some(10_000),
1154                         &nodes[2],
1155                         vec![&nodes[2], &nodes[3],],
1156                         scid_aliases,
1157                         false
1158                 );
1159         }
1160
1161         #[test]
1162         #[cfg(feature = "std")]
1163         fn test_multi_node_with_only_public_channels_hints_includes_only_phantom_route() {
1164                 let mut chanmon_cfgs = create_chanmon_cfgs(3);
1165                 let seed_1 = [42 as u8; 32];
1166                 let seed_2 = [43 as u8; 32];
1167                 let cross_node_seed = [44 as u8; 32];
1168                 chanmon_cfgs[1].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
1169                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
1170                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
1171                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
1172                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
1173
1174                 let chan_0_1 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 1, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1175
1176                 let chan_2_0 = create_announced_chan_between_nodes_with_value(&nodes, 2, 0, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1177                 nodes[2].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &chan_2_0.1);
1178                 nodes[0].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan_2_0.0);
1179
1180                 // Hints should include `chan_0_1` from as `nodes[1]` only have private channels, but not
1181                 // `chan_0_2` as `nodes[2]` only has public channels.
1182                 let mut scid_aliases = HashSet::new();
1183                 scid_aliases.insert(chan_0_1.0.short_channel_id_alias.unwrap());
1184
1185                 match_multi_node_invoice_routes(
1186                         Some(10_000),
1187                         &nodes[1],
1188                         vec![&nodes[1], &nodes[2],],
1189                         scid_aliases,
1190                         true
1191                 );
1192         }
1193
1194         #[test]
1195         #[cfg(feature = "std")]
1196         fn test_multi_node_with_mixed_public_and_private_channel_hints_includes_only_phantom_route() {
1197                 let mut chanmon_cfgs = create_chanmon_cfgs(4);
1198                 let seed_1 = [42 as u8; 32];
1199                 let seed_2 = [43 as u8; 32];
1200                 let cross_node_seed = [44 as u8; 32];
1201                 chanmon_cfgs[1].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
1202                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
1203                 let node_cfgs = create_node_cfgs(4, &chanmon_cfgs);
1204                 let node_chanmgrs = create_node_chanmgrs(4, &node_cfgs, &[None, None, None, None]);
1205                 let nodes = create_network(4, &node_cfgs, &node_chanmgrs);
1206
1207                 let chan_0_2 = create_announced_chan_between_nodes_with_value(&nodes, 0, 2, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1208                 nodes[0].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan_0_2.1);
1209                 nodes[2].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &chan_0_2.0);
1210                 let _chan_1_2 = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 2, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1211
1212                 let chan_0_3 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 3, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1213
1214                 // Hints should include `chan_0_3` from as `nodes[3]` only have private channels, and no
1215                 // channels for `nodes[2]` as it contains a mix of public and private channels.
1216                 let mut scid_aliases = HashSet::new();
1217                 scid_aliases.insert(chan_0_3.0.short_channel_id_alias.unwrap());
1218
1219                 match_multi_node_invoice_routes(
1220                         Some(10_000),
1221                         &nodes[2],
1222                         vec![&nodes[2], &nodes[3],],
1223                         scid_aliases,
1224                         true
1225                 );
1226         }
1227
1228         #[test]
1229         #[cfg(feature = "std")]
1230         fn test_multi_node_hints_has_only_highest_inbound_capacity_channel() {
1231                 let mut chanmon_cfgs = create_chanmon_cfgs(3);
1232                 let seed_1 = [42 as u8; 32];
1233                 let seed_2 = [43 as u8; 32];
1234                 let cross_node_seed = [44 as u8; 32];
1235                 chanmon_cfgs[1].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
1236                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
1237                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
1238                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
1239                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
1240
1241                 let _chan_0_1_low_inbound_capacity = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 1, 100_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1242                 let chan_0_1_high_inbound_capacity = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 1, 10_000_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1243                 let _chan_0_1_medium_inbound_capacity = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 1, 1_000_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1244                 let chan_0_2 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 2, 100000, 10001, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1245
1246                 let mut scid_aliases = HashSet::new();
1247                 scid_aliases.insert(chan_0_1_high_inbound_capacity.0.short_channel_id_alias.unwrap());
1248                 scid_aliases.insert(chan_0_2.0.short_channel_id_alias.unwrap());
1249
1250                 match_multi_node_invoice_routes(
1251                         Some(10_000),
1252                         &nodes[1],
1253                         vec![&nodes[1], &nodes[2],],
1254                         scid_aliases,
1255                         false
1256                 );
1257         }
1258
1259         #[test]
1260         #[cfg(feature = "std")]
1261         fn test_multi_node_channels_inbound_capacity_lower_than_invoice_amt_filtering() {
1262                 let mut chanmon_cfgs = create_chanmon_cfgs(4);
1263                 let seed_1 = [42 as u8; 32];
1264                 let seed_2 = [43 as u8; 32];
1265                 let cross_node_seed = [44 as u8; 32];
1266                 chanmon_cfgs[1].keys_manager.backing = PhantomKeysManager::new(&seed_1, 43, 44, &cross_node_seed);
1267                 chanmon_cfgs[2].keys_manager.backing = PhantomKeysManager::new(&seed_2, 43, 44, &cross_node_seed);
1268                 let node_cfgs = create_node_cfgs(4, &chanmon_cfgs);
1269                 let node_chanmgrs = create_node_chanmgrs(4, &node_cfgs, &[None, None, None, None]);
1270                 let nodes = create_network(4, &node_cfgs, &node_chanmgrs);
1271
1272                 let chan_0_2 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 2, 1_000_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1273                 let chan_0_3 = create_unannounced_chan_between_nodes_with_value(&nodes, 0, 3, 100_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1274                 let chan_1_3 = create_unannounced_chan_between_nodes_with_value(&nodes, 1, 3, 200_000, 0, channelmanager::provided_init_features(), channelmanager::provided_init_features());
1275
1276                 // Since the invoice 1 msat above chan_0_3's inbound capacity, it should be filtered out.
1277                 let mut scid_aliases_99_000_001_msat = HashSet::new();
1278                 scid_aliases_99_000_001_msat.insert(chan_0_2.0.short_channel_id_alias.unwrap());
1279                 scid_aliases_99_000_001_msat.insert(chan_1_3.0.short_channel_id_alias.unwrap());
1280
1281                 match_multi_node_invoice_routes(
1282                         Some(99_000_001),
1283                         &nodes[2],
1284                         vec![&nodes[2], &nodes[3],],
1285                         scid_aliases_99_000_001_msat,
1286                         false
1287                 );
1288
1289                 // Since the invoice is exactly at chan_0_3's inbound capacity, it should be included.
1290                 let mut scid_aliases_99_000_000_msat = HashSet::new();
1291                 scid_aliases_99_000_000_msat.insert(chan_0_2.0.short_channel_id_alias.unwrap());
1292                 scid_aliases_99_000_000_msat.insert(chan_0_3.0.short_channel_id_alias.unwrap());
1293                 scid_aliases_99_000_000_msat.insert(chan_1_3.0.short_channel_id_alias.unwrap());
1294
1295                 match_multi_node_invoice_routes(
1296                         Some(99_000_000),
1297                         &nodes[2],
1298                         vec![&nodes[2], &nodes[3],],
1299                         scid_aliases_99_000_000_msat,
1300                         false
1301                 );
1302
1303                 // Since the invoice is above all of `nodes[2]` channels' inbound capacity, all of
1304                 // `nodes[2]` them should be included.
1305                 let mut scid_aliases_300_000_000_msat = HashSet::new();
1306                 scid_aliases_300_000_000_msat.insert(chan_0_2.0.short_channel_id_alias.unwrap());
1307                 scid_aliases_300_000_000_msat.insert(chan_0_3.0.short_channel_id_alias.unwrap());
1308                 scid_aliases_300_000_000_msat.insert(chan_1_3.0.short_channel_id_alias.unwrap());
1309
1310                 match_multi_node_invoice_routes(
1311                         Some(300_000_000),
1312                         &nodes[2],
1313                         vec![&nodes[2], &nodes[3],],
1314                         scid_aliases_300_000_000_msat,
1315                         false
1316                 );
1317
1318                 // Since the no specified amount, all channels should included.
1319                 let mut scid_aliases_no_specified_amount = HashSet::new();
1320                 scid_aliases_no_specified_amount.insert(chan_0_2.0.short_channel_id_alias.unwrap());
1321                 scid_aliases_no_specified_amount.insert(chan_0_3.0.short_channel_id_alias.unwrap());
1322                 scid_aliases_no_specified_amount.insert(chan_1_3.0.short_channel_id_alias.unwrap());
1323
1324                 match_multi_node_invoice_routes(
1325                         None,
1326                         &nodes[2],
1327                         vec![&nodes[2], &nodes[3],],
1328                         scid_aliases_no_specified_amount,
1329                         false
1330                 );
1331         }
1332
1333         #[cfg(feature = "std")]
1334         fn match_multi_node_invoice_routes<'a, 'b: 'a, 'c: 'b>(
1335                 invoice_amt: Option<u64>,
1336                 invoice_node: &Node<'a, 'b, 'c>,
1337                 network_multi_nodes: Vec<&Node<'a, 'b, 'c>>,
1338                 mut chan_ids_to_match: HashSet<u64>,
1339                 nodes_contains_public_channels: bool
1340         ){
1341                 let phantom_route_hints = network_multi_nodes.iter()
1342                         .map(|node| node.node.get_phantom_route_hints())
1343                         .collect::<Vec<PhantomRouteHints>>();
1344                 let phantom_scids = phantom_route_hints.iter()
1345                         .map(|route_hint| route_hint.phantom_scid)
1346                         .collect::<HashSet<u64>>();
1347
1348                 let invoice = ::utils::create_phantom_invoice::<EnforcingSigner, &test_utils::TestKeysInterface>(invoice_amt, None, "test".to_string(), 3600, phantom_route_hints, &invoice_node.keys_manager, Currency::BitcoinTestnet).unwrap();
1349
1350                 let invoice_hints = invoice.private_routes();
1351
1352                 for hint in invoice_hints {
1353                         let hints = &(hint.0).0;
1354                         match hints.len() {
1355                                 1 => {
1356                                         assert!(nodes_contains_public_channels);
1357                                         let phantom_scid = hints[0].short_channel_id;
1358                                         assert!(phantom_scids.contains(&phantom_scid));
1359                                 },
1360                                 2 => {
1361                                         let hint_short_chan_id = hints[0].short_channel_id;
1362                                         assert!(chan_ids_to_match.remove(&hint_short_chan_id));
1363                                         let phantom_scid = hints[1].short_channel_id;
1364                                         assert!(phantom_scids.contains(&phantom_scid));
1365                                 },
1366                                 _ => panic!("Incorrect hint length generated")
1367                         }
1368                 }
1369                 assert!(chan_ids_to_match.is_empty(), "Unmatched short channel ids: {:?}", chan_ids_to_match);
1370         }
1371 }