1 // This file is Copyright its original authors, visible in version control
4 // This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
5 // or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
6 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
7 // You may not use this file except in accordance with one or both of these
10 //! A module for paying Lightning invoices.
12 //! Defines an [`InvoicePayer`] utility for paying invoices, parameterized by [`Payer`] and
13 //! [`Router`] traits. Implementations of [`Payer`] provide the payer's node id, channels, and means
14 //! to send a payment over a [`Route`]. Implementations of [`Router`] find a [`Route`] between payer
15 //! and payee using information provided by the payer and from the payee's [`Invoice`].
17 //! [`InvoicePayer`] is capable of retrying failed payments. It accomplishes this by implementing
18 //! [`EventHandler`] which decorates a user-provided handler. It will intercept any
19 //! [`Event::PaymentPathFailed`] events and retry the failed paths for a fixed number of total
20 //! attempts or until retry is no longer possible. In such a situation, [`InvoicePayer`] will pass
21 //! along the events to the user-provided handler.
26 //! # extern crate lightning;
27 //! # extern crate lightning_invoice;
28 //! # extern crate secp256k1;
30 //! # use lightning::ln::{PaymentHash, PaymentSecret};
31 //! # use lightning::ln::channelmanager::{ChannelDetails, PaymentId, PaymentSendFailure};
32 //! # use lightning::ln::msgs::LightningError;
33 //! # use lightning::routing;
34 //! # use lightning::routing::network_graph::NodeId;
35 //! # use lightning::routing::router::{Route, RouteHop, RouteParameters};
36 //! # use lightning::util::events::{Event, EventHandler, EventsProvider};
37 //! # use lightning::util::logger::{Logger, Record};
38 //! # use lightning_invoice::Invoice;
39 //! # use lightning_invoice::payment::{InvoicePayer, Payer, RetryAttempts, Router};
40 //! # use secp256k1::key::PublicKey;
41 //! # use std::ops::Deref;
43 //! # struct FakeEventProvider {}
44 //! # impl EventsProvider for FakeEventProvider {
45 //! # fn process_pending_events<H: Deref>(&self, handler: H) where H::Target: EventHandler {}
48 //! # struct FakePayer {}
49 //! # impl Payer for FakePayer {
50 //! # fn node_id(&self) -> PublicKey { unimplemented!() }
51 //! # fn first_hops(&self) -> Vec<ChannelDetails> { unimplemented!() }
52 //! # fn send_payment(
53 //! # &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>
54 //! # ) -> Result<PaymentId, PaymentSendFailure> { unimplemented!() }
55 //! # fn retry_payment(
56 //! # &self, route: &Route, payment_id: PaymentId
57 //! # ) -> Result<(), PaymentSendFailure> { unimplemented!() }
60 //! # struct FakeRouter {};
61 //! # impl Router for FakeRouter {
62 //! # fn find_route<S: routing::Score>(
63 //! # &self, payer: &PublicKey, params: &RouteParameters,
64 //! # first_hops: Option<&[&ChannelDetails]>, scorer: &S
65 //! # ) -> Result<Route, LightningError> { unimplemented!() }
68 //! # struct FakeScorer {};
69 //! # impl routing::Score for FakeScorer {
70 //! # fn channel_penalty_msat(
71 //! # &self, _short_channel_id: u64, _source: &NodeId, _target: &NodeId
73 //! # fn payment_path_failed(&mut self, _path: &Vec<RouteHop>, _short_channel_id: u64) {}
76 //! # struct FakeLogger {};
77 //! # impl Logger for FakeLogger {
78 //! # fn log(&self, record: &Record) { unimplemented!() }
82 //! let event_handler = |event: &Event| {
84 //! Event::PaymentPathFailed { .. } => println!("payment failed after retries"),
85 //! Event::PaymentSent { .. } => println!("payment successful"),
89 //! # let payer = FakePayer {};
90 //! # let router = FakeRouter {};
91 //! # let scorer = FakeScorer {};
92 //! # let logger = FakeLogger {};
93 //! let invoice_payer = InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
95 //! let invoice = "...";
96 //! let invoice = invoice.parse::<Invoice>().unwrap();
97 //! invoice_payer.pay_invoice(&invoice).unwrap();
99 //! # let event_provider = FakeEventProvider {};
101 //! event_provider.process_pending_events(&invoice_payer);
108 //! The [`Route`] is computed before each payment attempt. Any updates affecting path finding such
109 //! as updates to the network graph or changes to channel scores should be applied prior to
110 //! retries, typically by way of composing [`EventHandler`]s accordingly.
114 use bitcoin_hashes::Hash;
116 use lightning::ln::{PaymentHash, PaymentSecret};
117 use lightning::ln::channelmanager::{ChannelDetails, PaymentId, PaymentSendFailure};
118 use lightning::ln::msgs::LightningError;
119 use lightning::routing;
120 use lightning::routing::router::{Payee, Route, RouteParameters};
121 use lightning::util::events::{Event, EventHandler};
122 use lightning::util::logger::Logger;
124 use secp256k1::key::PublicKey;
126 use std::collections::hash_map::{self, HashMap};
128 use std::sync::{Mutex, RwLock};
129 use std::time::{Duration, SystemTime};
131 /// A utility for paying [`Invoice]`s.
132 pub struct InvoicePayer<P: Deref, R, S, L: Deref, E>
145 payment_cache: Mutex<HashMap<PaymentHash, usize>>,
146 retry_attempts: RetryAttempts,
149 /// A trait defining behavior of an [`Invoice`] payer.
151 /// Returns the payer's node id.
152 fn node_id(&self) -> PublicKey;
154 /// Returns the payer's channels.
155 fn first_hops(&self) -> Vec<ChannelDetails>;
157 /// Sends a payment over the Lightning Network using the given [`Route`].
159 &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>
160 ) -> Result<PaymentId, PaymentSendFailure>;
162 /// Retries a failed payment path for the [`PaymentId`] using the given [`Route`].
163 fn retry_payment(&self, route: &Route, payment_id: PaymentId) -> Result<(), PaymentSendFailure>;
166 /// A trait defining behavior for routing an [`Invoice`] payment.
168 /// Finds a [`Route`] between `payer` and `payee` for a payment with the given values.
169 fn find_route<S: routing::Score>(
170 &self, payer: &PublicKey, params: &RouteParameters, first_hops: Option<&[&ChannelDetails]>,
172 ) -> Result<Route, LightningError>;
175 /// Number of attempts to retry payment path failures for an [`Invoice`].
176 #[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
177 pub struct RetryAttempts(pub usize);
179 /// An error that may occur when making a payment.
180 #[derive(Clone, Debug)]
181 pub enum PaymentError {
182 /// An error resulting from the provided [`Invoice`] or payment hash.
183 Invoice(&'static str),
184 /// An error occurring when finding a route.
185 Routing(LightningError),
186 /// An error occurring when sending a payment.
187 Sending(PaymentSendFailure),
190 impl<P: Deref, R, S, L: Deref, E> InvoicePayer<P, R, S, L, E>
198 /// Creates an invoice payer that retries failed payment paths.
200 /// Will forward any [`Event::PaymentPathFailed`] events to the decorated `event_handler` once
201 /// `retry_attempts` has been exceeded for a given [`Invoice`].
203 payer: P, router: R, scorer: S, logger: L, event_handler: E, retry_attempts: RetryAttempts
208 scorer: RwLock::new(scorer),
211 payment_cache: Mutex::new(HashMap::new()),
216 /// Returns a read-only reference to the parameterized [`routing::Score`].
218 /// Useful if the scorer needs to be persisted. Be sure to drop the returned guard immediately
219 /// after use since retrying failed payment paths require write access.
220 pub fn scorer(&self) -> std::sync::RwLockReadGuard<'_, S> {
221 self.scorer.read().unwrap()
224 /// Pays the given [`Invoice`], caching it for later use in case a retry is needed.
225 pub fn pay_invoice(&self, invoice: &Invoice) -> Result<PaymentId, PaymentError> {
226 if invoice.amount_milli_satoshis().is_none() {
227 Err(PaymentError::Invoice("amount missing"))
229 self.pay_invoice_internal(invoice, None)
233 /// Pays the given zero-value [`Invoice`] using the given amount, caching it for later use in
234 /// case a retry is needed.
235 pub fn pay_zero_value_invoice(
236 &self, invoice: &Invoice, amount_msats: u64
237 ) -> Result<PaymentId, PaymentError> {
238 if invoice.amount_milli_satoshis().is_some() {
239 Err(PaymentError::Invoice("amount unexpected"))
241 self.pay_invoice_internal(invoice, Some(amount_msats))
245 fn pay_invoice_internal(
246 &self, invoice: &Invoice, amount_msats: Option<u64>
247 ) -> Result<PaymentId, PaymentError> {
248 debug_assert!(invoice.amount_milli_satoshis().is_some() ^ amount_msats.is_some());
249 let payment_hash = PaymentHash(invoice.payment_hash().clone().into_inner());
250 let mut payment_cache = self.payment_cache.lock().unwrap();
251 match payment_cache.entry(payment_hash) {
252 hash_map::Entry::Vacant(entry) => {
253 let payer = self.payer.node_id();
254 let mut payee = Payee::new(invoice.recover_payee_pub_key())
255 .with_expiry_time(expiry_time_from_unix_epoch(&invoice).as_secs())
256 .with_route_hints(invoice.route_hints());
257 if let Some(features) = invoice.features() {
258 payee = payee.with_features(features.clone());
260 let params = RouteParameters {
262 final_value_msat: invoice.amount_milli_satoshis().or(amount_msats).unwrap(),
263 final_cltv_expiry_delta: invoice.min_final_cltv_expiry() as u32,
265 let first_hops = self.payer.first_hops();
266 let route = self.router.find_route(
269 Some(&first_hops.iter().collect::<Vec<_>>()),
270 &*self.scorer.read().unwrap(),
271 ).map_err(|e| PaymentError::Routing(e))?;
273 let payment_hash = PaymentHash(invoice.payment_hash().clone().into_inner());
274 let payment_secret = Some(invoice.payment_secret().clone());
275 let payment_id = self.payer.send_payment(&route, payment_hash, &payment_secret)
276 .map_err(|e| PaymentError::Sending(e))?;
280 hash_map::Entry::Occupied(_) => Err(PaymentError::Invoice("payment pending")),
285 &self, payment_id: PaymentId, params: &RouteParameters
286 ) -> Result<(), PaymentError> {
287 let payer = self.payer.node_id();
288 let first_hops = self.payer.first_hops();
289 let route = self.router.find_route(
290 &payer, ¶ms, Some(&first_hops.iter().collect::<Vec<_>>()),
291 &*self.scorer.read().unwrap()
292 ).map_err(|e| PaymentError::Routing(e))?;
293 self.payer.retry_payment(&route, payment_id).map_err(|e| PaymentError::Sending(e))
296 /// Removes the payment cached by the given payment hash.
298 /// Should be called once a payment has failed or succeeded if not using [`InvoicePayer`] as an
299 /// [`EventHandler`]. Otherwise, calling this method is unnecessary.
300 pub fn remove_cached_payment(&self, payment_hash: &PaymentHash) {
301 self.payment_cache.lock().unwrap().remove(payment_hash);
305 fn expiry_time_from_unix_epoch(invoice: &Invoice) -> Duration {
306 invoice.timestamp().duration_since(SystemTime::UNIX_EPOCH).unwrap() + invoice.expiry_time()
309 fn has_expired(params: &RouteParameters) -> bool {
310 let expiry_time = Duration::from_secs(params.payee.expiry_time.unwrap());
311 Invoice::is_expired_from_epoch(&SystemTime::UNIX_EPOCH, expiry_time)
314 impl<P: Deref, R, S, L: Deref, E> EventHandler for InvoicePayer<P, R, S, L, E>
322 fn handle_event(&self, event: &Event) {
324 Event::PaymentPathFailed {
325 payment_id, payment_hash, rejected_by_dest, path, short_channel_id, retry, ..
327 if let Some(short_channel_id) = short_channel_id {
328 self.scorer.write().unwrap().payment_path_failed(path, *short_channel_id);
331 let mut payment_cache = self.payment_cache.lock().unwrap();
333 let entry = payment_cache.entry(*payment_hash);
335 hash_map::Entry::Occupied(_) => break entry,
336 hash_map::Entry::Vacant(entry) => entry.insert(0),
339 if let hash_map::Entry::Occupied(mut entry) = entry {
340 let max_payment_attempts = self.retry_attempts.0 + 1;
341 let attempts = entry.get_mut();
344 if *rejected_by_dest {
345 log_trace!(self.logger, "Payment {} rejected by destination; not retrying (attempts: {})", log_bytes!(payment_hash.0), attempts);
346 } else if payment_id.is_none() {
347 log_trace!(self.logger, "Payment {} has no id; not retrying (attempts: {})", log_bytes!(payment_hash.0), attempts);
348 } else if *attempts >= max_payment_attempts {
349 log_trace!(self.logger, "Payment {} exceeded maximum attempts; not retrying (attempts: {})", log_bytes!(payment_hash.0), attempts);
350 } else if retry.is_none() {
351 log_trace!(self.logger, "Payment {} missing retry params; not retrying (attempts: {})", log_bytes!(payment_hash.0), attempts);
352 } else if has_expired(retry.as_ref().unwrap()) {
353 log_trace!(self.logger, "Invoice expired for payment {}; not retrying (attempts: {})", log_bytes!(payment_hash.0), attempts);
354 } else if self.retry_payment(*payment_id.as_ref().unwrap(), retry.as_ref().unwrap()).is_err() {
355 log_trace!(self.logger, "Error retrying payment {}; not retrying (attempts: {})", log_bytes!(payment_hash.0), attempts);
357 log_trace!(self.logger, "Payment {} failed; retrying (attempts: {})", log_bytes!(payment_hash.0), attempts);
361 // Either the payment was rejected, the maximum attempts were exceeded, or an
362 // error occurred when attempting to retry.
368 Event::PaymentSent { payment_hash, .. } => {
369 let mut payment_cache = self.payment_cache.lock().unwrap();
370 let attempts = payment_cache
371 .remove(payment_hash)
372 .map_or(1, |attempts| attempts + 1);
373 log_trace!(self.logger, "Payment {} succeeded (attempts: {})", log_bytes!(payment_hash.0), attempts);
378 // Delegate to the decorated event handler unless the payment is retried.
379 self.event_handler.handle_event(event)
386 use crate::{DEFAULT_EXPIRY_TIME, InvoiceBuilder, Currency};
387 use bitcoin_hashes::sha256::Hash as Sha256;
388 use lightning::ln::PaymentPreimage;
389 use lightning::ln::features::{ChannelFeatures, NodeFeatures};
390 use lightning::ln::msgs::{ErrorAction, LightningError};
391 use lightning::routing::network_graph::NodeId;
392 use lightning::routing::router::{Payee, Route, RouteHop};
393 use lightning::util::test_utils::TestLogger;
394 use lightning::util::errors::APIError;
395 use lightning::util::events::Event;
396 use secp256k1::{SecretKey, PublicKey, Secp256k1};
397 use std::time::{SystemTime, Duration};
399 fn invoice(payment_preimage: PaymentPreimage) -> Invoice {
400 let payment_hash = Sha256::hash(&payment_preimage.0);
401 let private_key = SecretKey::from_slice(&[42; 32]).unwrap();
402 InvoiceBuilder::new(Currency::Bitcoin)
403 .description("test".into())
404 .payment_hash(payment_hash)
405 .payment_secret(PaymentSecret([0; 32]))
407 .min_final_cltv_expiry(144)
408 .amount_milli_satoshis(128)
409 .build_signed(|hash| {
410 Secp256k1::new().sign_recoverable(hash, &private_key)
415 fn zero_value_invoice(payment_preimage: PaymentPreimage) -> Invoice {
416 let payment_hash = Sha256::hash(&payment_preimage.0);
417 let private_key = SecretKey::from_slice(&[42; 32]).unwrap();
418 InvoiceBuilder::new(Currency::Bitcoin)
419 .description("test".into())
420 .payment_hash(payment_hash)
421 .payment_secret(PaymentSecret([0; 32]))
423 .min_final_cltv_expiry(144)
424 .build_signed(|hash| {
425 Secp256k1::new().sign_recoverable(hash, &private_key)
430 fn expired_invoice(payment_preimage: PaymentPreimage) -> Invoice {
431 let payment_hash = Sha256::hash(&payment_preimage.0);
432 let private_key = SecretKey::from_slice(&[42; 32]).unwrap();
433 let timestamp = SystemTime::now()
434 .checked_sub(Duration::from_secs(DEFAULT_EXPIRY_TIME * 2))
436 InvoiceBuilder::new(Currency::Bitcoin)
437 .description("test".into())
438 .payment_hash(payment_hash)
439 .payment_secret(PaymentSecret([0; 32]))
440 .timestamp(timestamp)
441 .min_final_cltv_expiry(144)
442 .amount_milli_satoshis(128)
443 .build_signed(|hash| {
444 Secp256k1::new().sign_recoverable(hash, &private_key)
450 fn pays_invoice_on_first_attempt() {
451 let event_handled = core::cell::RefCell::new(false);
452 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
454 let payment_preimage = PaymentPreimage([1; 32]);
455 let invoice = invoice(payment_preimage);
456 let payment_hash = PaymentHash(invoice.payment_hash().clone().into_inner());
458 let payer = TestPayer::new();
459 let router = TestRouter {};
460 let scorer = TestScorer::new();
461 let logger = TestLogger::new();
463 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(0));
465 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
466 assert_eq!(*payer.attempts.borrow(), 1);
468 invoice_payer.handle_event(&Event::PaymentSent {
469 payment_id, payment_preimage, payment_hash
471 assert_eq!(*event_handled.borrow(), true);
472 assert_eq!(*payer.attempts.borrow(), 1);
476 fn pays_invoice_on_retry() {
477 let event_handled = core::cell::RefCell::new(false);
478 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
480 let payment_preimage = PaymentPreimage([1; 32]);
481 let invoice = invoice(payment_preimage);
482 let payment_hash = PaymentHash(invoice.payment_hash().clone().into_inner());
483 let final_value_msat = invoice.amount_milli_satoshis().unwrap();
485 let payer = TestPayer::new()
486 .expect_value_msat(final_value_msat)
487 .expect_value_msat(final_value_msat / 2);
488 let router = TestRouter {};
489 let scorer = TestScorer::new();
490 let logger = TestLogger::new();
492 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
494 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
495 assert_eq!(*payer.attempts.borrow(), 1);
497 let event = Event::PaymentPathFailed {
500 network_update: None,
501 rejected_by_dest: false,
502 all_paths_failed: false,
503 path: TestRouter::path_for_value(final_value_msat),
504 short_channel_id: None,
505 retry: Some(TestRouter::retry_for_invoice(&invoice)),
507 invoice_payer.handle_event(&event);
508 assert_eq!(*event_handled.borrow(), false);
509 assert_eq!(*payer.attempts.borrow(), 2);
511 invoice_payer.handle_event(&Event::PaymentSent {
512 payment_id, payment_preimage, payment_hash
514 assert_eq!(*event_handled.borrow(), true);
515 assert_eq!(*payer.attempts.borrow(), 2);
519 fn retries_payment_path_for_unknown_payment() {
520 let event_handled = core::cell::RefCell::new(false);
521 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
523 let payment_preimage = PaymentPreimage([1; 32]);
524 let invoice = invoice(payment_preimage);
525 let payment_hash = PaymentHash(invoice.payment_hash().clone().into_inner());
526 let final_value_msat = invoice.amount_milli_satoshis().unwrap();
528 let payer = TestPayer::new();
529 let router = TestRouter {};
530 let scorer = TestScorer::new();
531 let logger = TestLogger::new();
533 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
535 let payment_id = Some(PaymentId([1; 32]));
536 let event = Event::PaymentPathFailed {
539 network_update: None,
540 rejected_by_dest: false,
541 all_paths_failed: false,
542 path: TestRouter::path_for_value(final_value_msat),
543 short_channel_id: None,
544 retry: Some(TestRouter::retry_for_invoice(&invoice)),
546 invoice_payer.handle_event(&event);
547 assert_eq!(*event_handled.borrow(), false);
548 assert_eq!(*payer.attempts.borrow(), 1);
550 invoice_payer.handle_event(&event);
551 assert_eq!(*event_handled.borrow(), false);
552 assert_eq!(*payer.attempts.borrow(), 2);
554 invoice_payer.handle_event(&Event::PaymentSent {
555 payment_id, payment_preimage, payment_hash
557 assert_eq!(*event_handled.borrow(), true);
558 assert_eq!(*payer.attempts.borrow(), 2);
562 fn fails_paying_invoice_after_max_retries() {
563 let event_handled = core::cell::RefCell::new(false);
564 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
566 let payment_preimage = PaymentPreimage([1; 32]);
567 let invoice = invoice(payment_preimage);
568 let final_value_msat = invoice.amount_milli_satoshis().unwrap();
570 let payer = TestPayer::new()
571 .expect_value_msat(final_value_msat)
572 .expect_value_msat(final_value_msat / 2)
573 .expect_value_msat(final_value_msat / 2);
574 let router = TestRouter {};
575 let scorer = TestScorer::new();
576 let logger = TestLogger::new();
578 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
580 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
581 assert_eq!(*payer.attempts.borrow(), 1);
583 let event = Event::PaymentPathFailed {
585 payment_hash: PaymentHash(invoice.payment_hash().clone().into_inner()),
586 network_update: None,
587 rejected_by_dest: false,
588 all_paths_failed: true,
589 path: TestRouter::path_for_value(final_value_msat),
590 short_channel_id: None,
591 retry: Some(TestRouter::retry_for_invoice(&invoice)),
593 invoice_payer.handle_event(&event);
594 assert_eq!(*event_handled.borrow(), false);
595 assert_eq!(*payer.attempts.borrow(), 2);
597 let event = Event::PaymentPathFailed {
599 payment_hash: PaymentHash(invoice.payment_hash().clone().into_inner()),
600 network_update: None,
601 rejected_by_dest: false,
602 all_paths_failed: false,
603 path: TestRouter::path_for_value(final_value_msat / 2),
604 short_channel_id: None,
605 retry: Some(RouteParameters {
606 final_value_msat: final_value_msat / 2, ..TestRouter::retry_for_invoice(&invoice)
609 invoice_payer.handle_event(&event);
610 assert_eq!(*event_handled.borrow(), false);
611 assert_eq!(*payer.attempts.borrow(), 3);
613 invoice_payer.handle_event(&event);
614 assert_eq!(*event_handled.borrow(), true);
615 assert_eq!(*payer.attempts.borrow(), 3);
619 fn fails_paying_invoice_with_missing_retry_params() {
620 let event_handled = core::cell::RefCell::new(false);
621 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
623 let payer = TestPayer::new();
624 let router = TestRouter {};
625 let scorer = TestScorer::new();
626 let logger = TestLogger::new();
628 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
630 let payment_preimage = PaymentPreimage([1; 32]);
631 let invoice = invoice(payment_preimage);
632 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
633 assert_eq!(*payer.attempts.borrow(), 1);
635 let event = Event::PaymentPathFailed {
637 payment_hash: PaymentHash(invoice.payment_hash().clone().into_inner()),
638 network_update: None,
639 rejected_by_dest: false,
640 all_paths_failed: false,
642 short_channel_id: None,
645 invoice_payer.handle_event(&event);
646 assert_eq!(*event_handled.borrow(), true);
647 assert_eq!(*payer.attempts.borrow(), 1);
651 fn fails_paying_invoice_after_expiration() {
652 let event_handled = core::cell::RefCell::new(false);
653 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
655 let payer = TestPayer::new();
656 let router = TestRouter {};
657 let scorer = TestScorer::new();
658 let logger = TestLogger::new();
660 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
662 let payment_preimage = PaymentPreimage([1; 32]);
663 let invoice = expired_invoice(payment_preimage);
664 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
665 assert_eq!(*payer.attempts.borrow(), 1);
667 let event = Event::PaymentPathFailed {
669 payment_hash: PaymentHash(invoice.payment_hash().clone().into_inner()),
670 network_update: None,
671 rejected_by_dest: false,
672 all_paths_failed: false,
674 short_channel_id: None,
675 retry: Some(TestRouter::retry_for_invoice(&invoice)),
677 invoice_payer.handle_event(&event);
678 assert_eq!(*event_handled.borrow(), true);
679 assert_eq!(*payer.attempts.borrow(), 1);
683 fn fails_paying_invoice_after_retry_error() {
684 let event_handled = core::cell::RefCell::new(false);
685 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
687 let payment_preimage = PaymentPreimage([1; 32]);
688 let invoice = invoice(payment_preimage);
689 let final_value_msat = invoice.amount_milli_satoshis().unwrap();
691 let payer = TestPayer::new()
693 .expect_value_msat(final_value_msat);
694 let router = TestRouter {};
695 let scorer = TestScorer::new();
696 let logger = TestLogger::new();
698 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
700 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
701 assert_eq!(*payer.attempts.borrow(), 1);
703 let event = Event::PaymentPathFailed {
705 payment_hash: PaymentHash(invoice.payment_hash().clone().into_inner()),
706 network_update: None,
707 rejected_by_dest: false,
708 all_paths_failed: false,
709 path: TestRouter::path_for_value(final_value_msat / 2),
710 short_channel_id: None,
711 retry: Some(TestRouter::retry_for_invoice(&invoice)),
713 invoice_payer.handle_event(&event);
714 assert_eq!(*event_handled.borrow(), true);
715 assert_eq!(*payer.attempts.borrow(), 2);
719 fn fails_paying_invoice_after_rejected_by_payee() {
720 let event_handled = core::cell::RefCell::new(false);
721 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
723 let payer = TestPayer::new();
724 let router = TestRouter {};
725 let scorer = TestScorer::new();
726 let logger = TestLogger::new();
728 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
730 let payment_preimage = PaymentPreimage([1; 32]);
731 let invoice = invoice(payment_preimage);
732 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
733 assert_eq!(*payer.attempts.borrow(), 1);
735 let event = Event::PaymentPathFailed {
737 payment_hash: PaymentHash(invoice.payment_hash().clone().into_inner()),
738 network_update: None,
739 rejected_by_dest: true,
740 all_paths_failed: false,
742 short_channel_id: None,
743 retry: Some(TestRouter::retry_for_invoice(&invoice)),
745 invoice_payer.handle_event(&event);
746 assert_eq!(*event_handled.borrow(), true);
747 assert_eq!(*payer.attempts.borrow(), 1);
751 fn fails_repaying_invoice_with_pending_payment() {
752 let event_handled = core::cell::RefCell::new(false);
753 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
755 let payer = TestPayer::new();
756 let router = TestRouter {};
757 let scorer = TestScorer::new();
758 let logger = TestLogger::new();
760 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(0));
762 let payment_preimage = PaymentPreimage([1; 32]);
763 let invoice = invoice(payment_preimage);
764 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
766 // Cannot repay an invoice pending payment.
767 match invoice_payer.pay_invoice(&invoice) {
768 Err(PaymentError::Invoice("payment pending")) => {},
769 Err(_) => panic!("unexpected error"),
770 Ok(_) => panic!("expected invoice error"),
773 // Can repay an invoice once cleared from cache.
774 let payment_hash = PaymentHash(invoice.payment_hash().clone().into_inner());
775 invoice_payer.remove_cached_payment(&payment_hash);
776 assert!(invoice_payer.pay_invoice(&invoice).is_ok());
778 // Cannot retry paying an invoice if cleared from cache.
779 invoice_payer.remove_cached_payment(&payment_hash);
780 let event = Event::PaymentPathFailed {
783 network_update: None,
784 rejected_by_dest: false,
785 all_paths_failed: false,
787 short_channel_id: None,
788 retry: Some(TestRouter::retry_for_invoice(&invoice)),
790 invoice_payer.handle_event(&event);
791 assert_eq!(*event_handled.borrow(), true);
795 fn fails_paying_invoice_with_routing_errors() {
796 let payer = TestPayer::new();
797 let router = FailingRouter {};
798 let scorer = TestScorer::new();
799 let logger = TestLogger::new();
801 InvoicePayer::new(&payer, router, scorer, &logger, |_: &_| {}, RetryAttempts(0));
803 let payment_preimage = PaymentPreimage([1; 32]);
804 let invoice = invoice(payment_preimage);
805 match invoice_payer.pay_invoice(&invoice) {
806 Err(PaymentError::Routing(_)) => {},
807 Err(_) => panic!("unexpected error"),
808 Ok(_) => panic!("expected routing error"),
813 fn fails_paying_invoice_with_sending_errors() {
814 let payer = TestPayer::new().fails_on_attempt(1);
815 let router = TestRouter {};
816 let scorer = TestScorer::new();
817 let logger = TestLogger::new();
819 InvoicePayer::new(&payer, router, scorer, &logger, |_: &_| {}, RetryAttempts(0));
821 let payment_preimage = PaymentPreimage([1; 32]);
822 let invoice = invoice(payment_preimage);
823 match invoice_payer.pay_invoice(&invoice) {
824 Err(PaymentError::Sending(_)) => {},
825 Err(_) => panic!("unexpected error"),
826 Ok(_) => panic!("expected sending error"),
831 fn pays_zero_value_invoice_using_amount() {
832 let event_handled = core::cell::RefCell::new(false);
833 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
835 let payment_preimage = PaymentPreimage([1; 32]);
836 let invoice = zero_value_invoice(payment_preimage);
837 let payment_hash = PaymentHash(invoice.payment_hash().clone().into_inner());
838 let final_value_msat = 100;
840 let payer = TestPayer::new().expect_value_msat(final_value_msat);
841 let router = TestRouter {};
842 let scorer = TestScorer::new();
843 let logger = TestLogger::new();
845 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(0));
848 Some(invoice_payer.pay_zero_value_invoice(&invoice, final_value_msat).unwrap());
849 assert_eq!(*payer.attempts.borrow(), 1);
851 invoice_payer.handle_event(&Event::PaymentSent {
852 payment_id, payment_preimage, payment_hash
854 assert_eq!(*event_handled.borrow(), true);
855 assert_eq!(*payer.attempts.borrow(), 1);
859 fn fails_paying_zero_value_invoice_with_amount() {
860 let event_handled = core::cell::RefCell::new(false);
861 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
863 let payer = TestPayer::new();
864 let router = TestRouter {};
865 let scorer = TestScorer::new();
866 let logger = TestLogger::new();
868 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(0));
870 let payment_preimage = PaymentPreimage([1; 32]);
871 let invoice = invoice(payment_preimage);
873 // Cannot repay an invoice pending payment.
874 match invoice_payer.pay_zero_value_invoice(&invoice, 100) {
875 Err(PaymentError::Invoice("amount unexpected")) => {},
876 Err(_) => panic!("unexpected error"),
877 Ok(_) => panic!("expected invoice error"),
882 fn scores_failed_channel() {
883 let event_handled = core::cell::RefCell::new(false);
884 let event_handler = |_: &_| { *event_handled.borrow_mut() = true; };
886 let payment_preimage = PaymentPreimage([1; 32]);
887 let invoice = invoice(payment_preimage);
888 let payment_hash = PaymentHash(invoice.payment_hash().clone().into_inner());
889 let final_value_msat = invoice.amount_milli_satoshis().unwrap();
890 let path = TestRouter::path_for_value(final_value_msat);
891 let short_channel_id = Some(path[0].short_channel_id);
893 let payer = TestPayer::new();
894 let router = TestRouter {};
895 let scorer = TestScorer::new().expect_channel_failure(short_channel_id.unwrap());
896 let logger = TestLogger::new();
898 InvoicePayer::new(&payer, router, scorer, &logger, event_handler, RetryAttempts(2));
900 let payment_id = Some(invoice_payer.pay_invoice(&invoice).unwrap());
901 let event = Event::PaymentPathFailed {
904 network_update: None,
905 rejected_by_dest: false,
906 all_paths_failed: false,
909 retry: Some(TestRouter::retry_for_invoice(&invoice)),
911 invoice_payer.handle_event(&event);
917 fn route_for_value(final_value_msat: u64) -> Route {
921 pubkey: PublicKey::from_slice(&hex::decode("02eec7245d6b7d2ccb30380bfbe2a3648cd7a942653f5aa340edcea1f283686619").unwrap()[..]).unwrap(),
922 channel_features: ChannelFeatures::empty(),
923 node_features: NodeFeatures::empty(),
924 short_channel_id: 0, fee_msat: final_value_msat / 2, cltv_expiry_delta: 144
927 pubkey: PublicKey::from_slice(&hex::decode("0324653eac434488002cc06bbfb7f10fe18991e35f9fe4302dbea6d2353dc0ab1c").unwrap()[..]).unwrap(),
928 channel_features: ChannelFeatures::empty(),
929 node_features: NodeFeatures::empty(),
930 short_channel_id: 1, fee_msat: final_value_msat / 2, cltv_expiry_delta: 144
937 fn path_for_value(final_value_msat: u64) -> Vec<RouteHop> {
938 TestRouter::route_for_value(final_value_msat).paths[0].clone()
941 fn retry_for_invoice(invoice: &Invoice) -> RouteParameters {
942 let mut payee = Payee::new(invoice.recover_payee_pub_key())
943 .with_expiry_time(expiry_time_from_unix_epoch(invoice).as_secs())
944 .with_route_hints(invoice.route_hints());
945 if let Some(features) = invoice.features() {
946 payee = payee.with_features(features.clone());
948 let final_value_msat = invoice.amount_milli_satoshis().unwrap() / 2;
952 final_cltv_expiry_delta: invoice.min_final_cltv_expiry() as u32,
957 impl Router for TestRouter {
958 fn find_route<S: routing::Score>(
961 params: &RouteParameters,
962 _first_hops: Option<&[&ChannelDetails]>,
964 ) -> Result<Route, LightningError> {
966 payee: Some(params.payee.clone()), ..Self::route_for_value(params.final_value_msat)
971 struct FailingRouter;
973 impl Router for FailingRouter {
974 fn find_route<S: routing::Score>(
977 _params: &RouteParameters,
978 _first_hops: Option<&[&ChannelDetails]>,
980 ) -> Result<Route, LightningError> {
981 Err(LightningError { err: String::new(), action: ErrorAction::IgnoreError })
986 expectations: std::collections::VecDeque<u64>,
992 expectations: std::collections::VecDeque::new(),
996 fn expect_channel_failure(mut self, short_channel_id: u64) -> Self {
997 self.expectations.push_back(short_channel_id);
1002 impl routing::Score for TestScorer {
1003 fn channel_penalty_msat(
1004 &self, _short_channel_id: u64, _source: &NodeId, _target: &NodeId
1007 fn payment_path_failed(&mut self, _path: &Vec<RouteHop>, short_channel_id: u64) {
1008 if let Some(expected_short_channel_id) = self.expectations.pop_front() {
1009 assert_eq!(short_channel_id, expected_short_channel_id);
1014 impl Drop for TestScorer {
1015 fn drop(&mut self) {
1016 if std::thread::panicking() {
1020 if !self.expectations.is_empty() {
1021 panic!("Unsatisfied channel failure expectations: {:?}", self.expectations);
1027 expectations: core::cell::RefCell<std::collections::VecDeque<u64>>,
1028 attempts: core::cell::RefCell<usize>,
1029 failing_on_attempt: Option<usize>,
1035 expectations: core::cell::RefCell::new(std::collections::VecDeque::new()),
1036 attempts: core::cell::RefCell::new(0),
1037 failing_on_attempt: None,
1041 fn expect_value_msat(self, value_msat: u64) -> Self {
1042 self.expectations.borrow_mut().push_back(value_msat);
1046 fn fails_on_attempt(self, attempt: usize) -> Self {
1048 expectations: core::cell::RefCell::new(self.expectations.borrow().clone()),
1049 attempts: core::cell::RefCell::new(0),
1050 failing_on_attempt: Some(attempt),
1054 fn check_attempts(&self) -> bool {
1055 let mut attempts = self.attempts.borrow_mut();
1057 match self.failing_on_attempt {
1059 Some(attempt) if attempt != *attempts => true,
1064 fn check_value_msats(&self, route: &Route) {
1065 let expected_value_msats = self.expectations.borrow_mut().pop_front();
1066 if let Some(expected_value_msats) = expected_value_msats {
1067 let actual_value_msats = route.get_total_amount();
1068 assert_eq!(actual_value_msats, expected_value_msats);
1073 impl Drop for TestPayer {
1074 fn drop(&mut self) {
1075 if std::thread::panicking() {
1079 if !self.expectations.borrow().is_empty() {
1080 panic!("Unsatisfied payment expectations: {:?}", self.expectations.borrow());
1085 impl Payer for TestPayer {
1086 fn node_id(&self) -> PublicKey {
1087 let secp_ctx = Secp256k1::new();
1088 PublicKey::from_secret_key(&secp_ctx, &SecretKey::from_slice(&[42; 32]).unwrap())
1091 fn first_hops(&self) -> Vec<ChannelDetails> {
1098 _payment_hash: PaymentHash,
1099 _payment_secret: &Option<PaymentSecret>
1100 ) -> Result<PaymentId, PaymentSendFailure> {
1101 if self.check_attempts() {
1102 self.check_value_msats(route);
1103 Ok(PaymentId([1; 32]))
1105 Err(PaymentSendFailure::ParameterError(APIError::MonitorUpdateFailed))
1110 &self, route: &Route, _payment_id: PaymentId
1111 ) -> Result<(), PaymentSendFailure> {
1112 if self.check_attempts() {
1113 self.check_value_msats(route);
1116 Err(PaymentSendFailure::ParameterError(APIError::MonitorUpdateFailed))