_ Git - rust-lightning/blob - lightning/src/ln/outbound_payment.rs

   1 // This file is Copyright its original authors, visible in version control
   2 // history.
   3 //
   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
   8 // licenses.
   9
  10 //! Utilities to send payments and manage outbound payment information.
  11
  12 use bitcoin::hashes::Hash;
  13 use bitcoin::hashes::sha256::Hash as Sha256;
  14 use bitcoin::secp256k1::{self, Secp256k1, SecretKey};
  15
  16 use crate::chain::keysinterface::{EntropySource, NodeSigner, Recipient};
  17 use crate::ln::{PaymentHash, PaymentPreimage, PaymentSecret};
  18 use crate::ln::channelmanager::{ChannelDetails, HTLCSource, IDEMPOTENCY_TIMEOUT_TICKS, MIN_HTLC_RELAY_HOLDING_CELL_MILLIS, PaymentId};
  19 use crate::ln::channelmanager::MIN_FINAL_CLTV_EXPIRY_DELTA as LDK_DEFAULT_MIN_FINAL_CLTV_EXPIRY_DELTA;
  20 use crate::ln::msgs::DecodeError;
  21 use crate::ln::onion_utils::HTLCFailReason;
  22 use crate::routing::router::{InFlightHtlcs, PaymentParameters, Route, RouteHop, RouteParameters, RoutePath, Router};
  23 use crate::util::errors::APIError;
  24 use crate::util::events;
  25 use crate::util::logger::Logger;
  26 use crate::util::time::Time;
  27 #[cfg(all(not(feature = "no-std"), test))]
  28 use crate::util::time::tests::SinceEpoch;
  29
  30 use core::cmp;
  31 use core::fmt::{self, Display, Formatter};
  32 use core::ops::Deref;
  33 use core::time::Duration;
  34
  35 use crate::prelude::*;
  36 use crate::sync::Mutex;
  37
  38 /// Stores the session_priv for each part of a payment that is still pending. For versions 0.0.102
  39 /// and later, also stores information for retrying the payment.
  40 pub(crate) enum PendingOutboundPayment {
  41         Legacy {
  42                 session_privs: HashSet<[u8; 32]>,
  43         },
  44         Retryable {
  45                 retry_strategy: Option<Retry>,
  46                 attempts: PaymentAttempts,
  47                 payment_params: Option<PaymentParameters>,
  48                 session_privs: HashSet<[u8; 32]>,
  49                 payment_hash: PaymentHash,
  50                 payment_secret: Option<PaymentSecret>,
  51                 keysend_preimage: Option<PaymentPreimage>,
  52                 pending_amt_msat: u64,
  53                 /// Used to track the fee paid. Only present if the payment was serialized on 0.0.103+.
  54                 pending_fee_msat: Option<u64>,
  55                 /// The total payment amount across all paths, used to verify that a retry is not overpaying.
  56                 total_msat: u64,
  57                 /// Our best known block height at the time this payment was initiated.
  58                 starting_block_height: u32,
  59         },
  60         /// When a pending payment is fulfilled, we continue tracking it until all pending HTLCs have
  61         /// been resolved. This ensures we don't look up pending payments in ChannelMonitors on restart
  62         /// and add a pending payment that was already fulfilled.
  63         Fulfilled {
  64                 session_privs: HashSet<[u8; 32]>,
  65                 payment_hash: Option<PaymentHash>,
  66                 timer_ticks_without_htlcs: u8,
  67         },
  68         /// When a payer gives up trying to retry a payment, they inform us, letting us generate a
  69         /// `PaymentFailed` event when all HTLCs have irrevocably failed. This avoids a number of race
  70         /// conditions in MPP-aware payment retriers (1), where the possibility of multiple
  71         /// `PaymentPathFailed` events with `all_paths_failed` can be pending at once, confusing a
  72         /// downstream event handler as to when a payment has actually failed.
  73         ///
  74         /// (1) <https://github.com/lightningdevkit/rust-lightning/issues/1164>
  75         Abandoned {
  76                 session_privs: HashSet<[u8; 32]>,
  77                 payment_hash: PaymentHash,
  78         },
  79 }
  80
  81 impl PendingOutboundPayment {
  82         fn increment_attempts(&mut self) {
  83                 if let PendingOutboundPayment::Retryable { attempts, .. } = self {
  84                         attempts.count += 1;
  85                 }
  86         }
  87         fn is_auto_retryable_now(&self) -> bool {
  88                 match self {
  89                         PendingOutboundPayment::Retryable { retry_strategy: Some(strategy), attempts, .. } => {
  90                                 strategy.is_retryable_now(&attempts)
  91                         },
  92                         _ => false,
  93                 }
  94         }
  95         fn is_retryable_now(&self) -> bool {
  96                 match self {
  97                         PendingOutboundPayment::Retryable { retry_strategy: None, .. } => {
  98                                 // We're handling retries manually, we can always retry.
  99                                 true
 100                         },
 101                         PendingOutboundPayment::Retryable { retry_strategy: Some(strategy), attempts, .. } => {
 102                                 strategy.is_retryable_now(&attempts)
 103                         },
 104                         _ => false,
 105                 }
 106         }
 107         fn payment_parameters(&mut self) -> Option<&mut PaymentParameters> {
 108                 match self {
 109                         PendingOutboundPayment::Retryable { payment_params: Some(ref mut params), .. } => {
 110                                 Some(params)
 111                         },
 112                         _ => None,
 113                 }
 114         }
 115         pub fn insert_previously_failed_scid(&mut self, scid: u64) {
 116                 if let PendingOutboundPayment::Retryable { payment_params: Some(params), .. } = self {
 117                         params.previously_failed_channels.push(scid);
 118                 }
 119         }
 120         pub(super) fn is_fulfilled(&self) -> bool {
 121                 match self {
 122                         PendingOutboundPayment::Fulfilled { .. } => true,
 123                         _ => false,
 124                 }
 125         }
 126         pub(super) fn abandoned(&self) -> bool {
 127                 match self {
 128                         PendingOutboundPayment::Abandoned { .. } => true,
 129                         _ => false,
 130                 }
 131         }
 132         fn get_pending_fee_msat(&self) -> Option<u64> {
 133                 match self {
 134                         PendingOutboundPayment::Retryable { pending_fee_msat, .. } => pending_fee_msat.clone(),
 135                         _ => None,
 136                 }
 137         }
 138
 139         fn payment_hash(&self) -> Option<PaymentHash> {
 140                 match self {
 141                         PendingOutboundPayment::Legacy { .. } => None,
 142                         PendingOutboundPayment::Retryable { payment_hash, .. } => Some(*payment_hash),
 143                         PendingOutboundPayment::Fulfilled { payment_hash, .. } => *payment_hash,
 144                         PendingOutboundPayment::Abandoned { payment_hash, .. } => Some(*payment_hash),
 145                 }
 146         }
 147
 148         fn mark_fulfilled(&mut self) {
 149                 let mut session_privs = HashSet::new();
 150                 core::mem::swap(&mut session_privs, match self {
 151                         PendingOutboundPayment::Legacy { session_privs } |
 152                                 PendingOutboundPayment::Retryable { session_privs, .. } |
 153                                 PendingOutboundPayment::Fulfilled { session_privs, .. } |
 154                                 PendingOutboundPayment::Abandoned { session_privs, .. }
 155                         => session_privs,
 156                 });
 157                 let payment_hash = self.payment_hash();
 158                 *self = PendingOutboundPayment::Fulfilled { session_privs, payment_hash, timer_ticks_without_htlcs: 0 };
 159         }
 160
 161         fn mark_abandoned(&mut self) -> Result<(), ()> {
 162                 let mut session_privs = HashSet::new();
 163                 let our_payment_hash;
 164                 core::mem::swap(&mut session_privs, match self {
 165                         PendingOutboundPayment::Legacy { .. } |
 166                                 PendingOutboundPayment::Fulfilled { .. } =>
 167                                 return Err(()),
 168                                 PendingOutboundPayment::Retryable { session_privs, payment_hash, .. } |
 169                                         PendingOutboundPayment::Abandoned { session_privs, payment_hash, .. } => {
 170                                                 our_payment_hash = *payment_hash;
 171                                                 session_privs
 172                                         },
 173                 });
 174                 *self = PendingOutboundPayment::Abandoned { session_privs, payment_hash: our_payment_hash };
 175                 Ok(())
 176         }
 177
 178         /// panics if path is None and !self.is_fulfilled
 179         fn remove(&mut self, session_priv: &[u8; 32], path: Option<&Vec<RouteHop>>) -> bool {
 180                 let remove_res = match self {
 181                         PendingOutboundPayment::Legacy { session_privs } |
 182                                 PendingOutboundPayment::Retryable { session_privs, .. } |
 183                                 PendingOutboundPayment::Fulfilled { session_privs, .. } |
 184                                 PendingOutboundPayment::Abandoned { session_privs, .. } => {
 185                                         session_privs.remove(session_priv)
 186                                 }
 187                 };
 188                 if remove_res {
 189                         if let PendingOutboundPayment::Retryable { ref mut pending_amt_msat, ref mut pending_fee_msat, .. } = self {
 190                                 let path = path.expect("Fulfilling a payment should always come with a path");
 191                                 let path_last_hop = path.last().expect("Outbound payments must have had a valid path");
 192                                 *pending_amt_msat -= path_last_hop.fee_msat;
 193                                 if let Some(fee_msat) = pending_fee_msat.as_mut() {
 194                                         *fee_msat -= path.get_path_fees();
 195                                 }
 196                         }
 197                 }
 198                 remove_res
 199         }
 200
 201         pub(super) fn insert(&mut self, session_priv: [u8; 32], path: &Vec<RouteHop>) -> bool {
 202                 let insert_res = match self {
 203                         PendingOutboundPayment::Legacy { session_privs } |
 204                                 PendingOutboundPayment::Retryable { session_privs, .. } => {
 205                                         session_privs.insert(session_priv)
 206                                 }
 207                         PendingOutboundPayment::Fulfilled { .. } => false,
 208                         PendingOutboundPayment::Abandoned { .. } => false,
 209                 };
 210                 if insert_res {
 211                         if let PendingOutboundPayment::Retryable { ref mut pending_amt_msat, ref mut pending_fee_msat, .. } = self {
 212                                 let path_last_hop = path.last().expect("Outbound payments must have had a valid path");
 213                                 *pending_amt_msat += path_last_hop.fee_msat;
 214                                 if let Some(fee_msat) = pending_fee_msat.as_mut() {
 215                                         *fee_msat += path.get_path_fees();
 216                                 }
 217                         }
 218                 }
 219                 insert_res
 220         }
 221
 222         pub(super) fn remaining_parts(&self) -> usize {
 223                 match self {
 224                         PendingOutboundPayment::Legacy { session_privs } |
 225                                 PendingOutboundPayment::Retryable { session_privs, .. } |
 226                                 PendingOutboundPayment::Fulfilled { session_privs, .. } |
 227                                 PendingOutboundPayment::Abandoned { session_privs, .. } => {
 228                                         session_privs.len()
 229                                 }
 230                 }
 231         }
 232 }
 233
 234 /// Strategies available to retry payment path failures.
 235 #[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
 236 pub enum Retry {
 237         /// Max number of attempts to retry payment.
 238         ///
 239         /// Each attempt may be multiple HTLCs along multiple paths if the router decides to split up a
 240         /// retry, and may retry multiple failed HTLCs at once if they failed around the same time and
 241         /// were retried along a route from a single call to [`Router::find_route`].
 242         Attempts(usize),
 243         #[cfg(not(feature = "no-std"))]
 244         /// Time elapsed before abandoning retries for a payment.
 245         Timeout(core::time::Duration),
 246 }
 247
 248 impl Retry {
 249         pub(crate) fn is_retryable_now(&self, attempts: &PaymentAttempts) -> bool {
 250                 match (self, attempts) {
 251                         (Retry::Attempts(max_retry_count), PaymentAttempts { count, .. }) => {
 252                                 max_retry_count > count
 253                         },
 254                         #[cfg(all(not(feature = "no-std"), not(test)))]
 255                         (Retry::Timeout(max_duration), PaymentAttempts { first_attempted_at, .. }) =>
 256                                 *max_duration >= std::time::Instant::now().duration_since(*first_attempted_at),
 257                         #[cfg(all(not(feature = "no-std"), test))]
 258                         (Retry::Timeout(max_duration), PaymentAttempts { first_attempted_at, .. }) =>
 259                                 *max_duration >= SinceEpoch::now().duration_since(*first_attempted_at),
 260                 }
 261         }
 262 }
 263
 264 #[cfg(feature = "std")]
 265 pub(super) fn has_expired(route_params: &RouteParameters) -> bool {
 266         if let Some(expiry_time) = route_params.payment_params.expiry_time {
 267                 if let Ok(elapsed) = std::time::SystemTime::UNIX_EPOCH.elapsed() {
 268                         return elapsed > core::time::Duration::from_secs(expiry_time)
 269                 }
 270         }
 271         false
 272 }
 273
 274 pub(crate) type PaymentAttempts = PaymentAttemptsUsingTime<ConfiguredTime>;
 275
 276 /// Storing minimal payment attempts information required for determining if a outbound payment can
 277 /// be retried.
 278 pub(crate) struct PaymentAttemptsUsingTime<T: Time> {
 279         /// This count will be incremented only after the result of the attempt is known. When it's 0,
 280         /// it means the result of the first attempt is not known yet.
 281         pub(crate) count: usize,
 282         /// This field is only used when retry is `Retry::Timeout` which is only build with feature std
 283         first_attempted_at: T
 284 }
 285
 286 #[cfg(not(any(feature = "no-std", test)))]
 287 type ConfiguredTime = std::time::Instant;
 288 #[cfg(feature = "no-std")]
 289 type ConfiguredTime = crate::util::time::Eternity;
 290 #[cfg(all(not(feature = "no-std"), test))]
 291 type ConfiguredTime = SinceEpoch;
 292
 293 impl<T: Time> PaymentAttemptsUsingTime<T> {
 294         pub(crate) fn new() -> Self {
 295                 PaymentAttemptsUsingTime {
 296                         count: 0,
 297                         first_attempted_at: T::now()
 298                 }
 299         }
 300 }
 301
 302 impl<T: Time> Display for PaymentAttemptsUsingTime<T> {
 303         fn fmt(&self, f: &mut Formatter) -> Result<(), fmt::Error> {
 304                 #[cfg(feature = "no-std")]
 305                 return write!(f, "attempts: {}", self.count);
 306                 #[cfg(not(feature = "no-std"))]
 307                 return write!(
 308                         f,
 309                         "attempts: {}, duration: {}s",
 310                         self.count,
 311                         T::now().duration_since(self.first_attempted_at).as_secs()
 312                 );
 313         }
 314 }
 315
 316 /// If a payment fails to send, it can be in one of several states. This enum is returned as the
 317 /// Err() type describing which state the payment is in, see the description of individual enum
 318 /// states for more.
 319 #[derive(Clone, Debug)]
 320 pub enum PaymentSendFailure {
 321         /// A parameter which was passed to send_payment was invalid, preventing us from attempting to
 322         /// send the payment at all.
 323         ///
 324         /// You can freely resend the payment in full (with the parameter error fixed).
 325         ///
 326         /// Because the payment failed outright, no payment tracking is done, you do not need to call
 327         /// [`ChannelManager::abandon_payment`] and [`ChannelManager::retry_payment`] will *not* work
 328         /// for this payment.
 329         ///
 330         /// [`ChannelManager::abandon_payment`]: crate::ln::channelmanager::ChannelManager::abandon_payment
 331         /// [`ChannelManager::retry_payment`]: crate::ln::channelmanager::ChannelManager::retry_payment
 332         ParameterError(APIError),
 333         /// A parameter in a single path which was passed to send_payment was invalid, preventing us
 334         /// from attempting to send the payment at all.
 335         ///
 336         /// You can freely resend the payment in full (with the parameter error fixed).
 337         ///
 338         /// The results here are ordered the same as the paths in the route object which was passed to
 339         /// send_payment.
 340         ///
 341         /// Because the payment failed outright, no payment tracking is done, you do not need to call
 342         /// [`ChannelManager::abandon_payment`] and [`ChannelManager::retry_payment`] will *not* work
 343         /// for this payment.
 344         ///
 345         /// [`ChannelManager::abandon_payment`]: crate::ln::channelmanager::ChannelManager::abandon_payment
 346         /// [`ChannelManager::retry_payment`]: crate::ln::channelmanager::ChannelManager::retry_payment
 347         PathParameterError(Vec<Result<(), APIError>>),
 348         /// All paths which were attempted failed to send, with no channel state change taking place.
 349         /// You can freely resend the payment in full (though you probably want to do so over different
 350         /// paths than the ones selected).
 351         ///
 352         /// Because the payment failed outright, no payment tracking is done, you do not need to call
 353         /// [`ChannelManager::abandon_payment`] and [`ChannelManager::retry_payment`] will *not* work
 354         /// for this payment.
 355         ///
 356         /// [`ChannelManager::abandon_payment`]: crate::ln::channelmanager::ChannelManager::abandon_payment
 357         /// [`ChannelManager::retry_payment`]: crate::ln::channelmanager::ChannelManager::retry_payment
 358         AllFailedResendSafe(Vec<APIError>),
 359         /// Indicates that a payment for the provided [`PaymentId`] is already in-flight and has not
 360         /// yet completed (i.e. generated an [`Event::PaymentSent`]) or been abandoned (via
 361         /// [`ChannelManager::abandon_payment`]).
 362         ///
 363         /// [`PaymentId`]: crate::ln::channelmanager::PaymentId
 364         /// [`Event::PaymentSent`]: crate::util::events::Event::PaymentSent
 365         /// [`ChannelManager::abandon_payment`]: crate::ln::channelmanager::ChannelManager::abandon_payment
 366         DuplicatePayment,
 367         /// Some paths which were attempted failed to send, though possibly not all. At least some
 368         /// paths have irrevocably committed to the HTLC and retrying the payment in full would result
 369         /// in over-/re-payment.
 370         ///
 371         /// The results here are ordered the same as the paths in the route object which was passed to
 372         /// send_payment, and any `Err`s which are not [`APIError::MonitorUpdateInProgress`] can be
 373         /// safely retried via [`ChannelManager::retry_payment`].
 374         ///
 375         /// Any entries which contain `Err(APIError::MonitorUpdateInprogress)` or `Ok(())` MUST NOT be
 376         /// retried as they will result in over-/re-payment. These HTLCs all either successfully sent
 377         /// (in the case of `Ok(())`) or will send once a [`MonitorEvent::Completed`] is provided for
 378         /// the next-hop channel with the latest update_id.
 379         ///
 380         /// [`ChannelManager::retry_payment`]: crate::ln::channelmanager::ChannelManager::retry_payment
 381         /// [`MonitorEvent::Completed`]: crate::chain::channelmonitor::MonitorEvent::Completed
 382         PartialFailure {
 383                 /// The errors themselves, in the same order as the route hops.
 384                 results: Vec<Result<(), APIError>>,
 385                 /// If some paths failed without irrevocably committing to the new HTLC(s), this will
 386                 /// contain a [`RouteParameters`] object which can be used to calculate a new route that
 387                 /// will pay all remaining unpaid balance.
 388                 failed_paths_retry: Option<RouteParameters>,
 389                 /// The payment id for the payment, which is now at least partially pending.
 390                 payment_id: PaymentId,
 391         },
 392 }
 393
 394 pub(super) struct OutboundPayments {
 395         pub(super) pending_outbound_payments: Mutex<HashMap<PaymentId, PendingOutboundPayment>>,
 396 }
 397
 398 impl OutboundPayments {
 399         pub(super) fn new() -> Self {
 400                 Self {
 401                         pending_outbound_payments: Mutex::new(HashMap::new())
 402                 }
 403         }
 404
 405         pub(super) fn send_payment<R: Deref, ES: Deref, NS: Deref, F, L: Deref>(
 406                 &self, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>, payment_id: PaymentId,
 407                 retry_strategy: Retry, route_params: RouteParameters, router: &R,
 408                 first_hops: Vec<ChannelDetails>, inflight_htlcs: InFlightHtlcs, entropy_source: &ES,
 409                 node_signer: &NS, best_block_height: u32, logger: &L, send_payment_along_path: F,
 410         ) -> Result<(), PaymentSendFailure>
 411         where
 412                 R::Target: Router,
 413                 ES::Target: EntropySource,
 414                 NS::Target: NodeSigner,
 415                 L::Target: Logger,
 416                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 417                          u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>,
 418         {
 419                 self.pay_internal(payment_id, Some((payment_hash, payment_secret, None, retry_strategy)),
 420                         route_params, router, first_hops, inflight_htlcs, entropy_source, node_signer,
 421                         best_block_height, logger, &send_payment_along_path)
 422                         .map_err(|e| { self.remove_outbound_if_all_failed(payment_id, &e); e })
 423         }
 424
 425         pub(super) fn send_payment_with_route<ES: Deref, NS: Deref, F>(
 426                 &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>,
 427                 payment_id: PaymentId, entropy_source: &ES, node_signer: &NS, best_block_height: u32,
 428                 send_payment_along_path: F
 429         ) -> Result<(), PaymentSendFailure>
 430         where
 431                 ES::Target: EntropySource,
 432                 NS::Target: NodeSigner,
 433                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 434                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 435         {
 436                 let onion_session_privs = self.add_new_pending_payment(payment_hash, *payment_secret, payment_id, None, route, None, None, entropy_source, best_block_height)?;
 437                 self.pay_route_internal(route, payment_hash, payment_secret, None, payment_id, None,
 438                         onion_session_privs, node_signer, best_block_height, &send_payment_along_path)
 439                         .map_err(|e| { self.remove_outbound_if_all_failed(payment_id, &e); e })
 440         }
 441
 442         pub(super) fn send_spontaneous_payment<R: Deref, ES: Deref, NS: Deref, F, L: Deref>(
 443                 &self, payment_preimage: Option<PaymentPreimage>, payment_id: PaymentId,
 444                 retry_strategy: Retry, route_params: RouteParameters, router: &R,
 445                 first_hops: Vec<ChannelDetails>, inflight_htlcs: InFlightHtlcs, entropy_source: &ES,
 446                 node_signer: &NS, best_block_height: u32, logger: &L, send_payment_along_path: F
 447         ) -> Result<PaymentHash, PaymentSendFailure>
 448         where
 449                 R::Target: Router,
 450                 ES::Target: EntropySource,
 451                 NS::Target: NodeSigner,
 452                 L::Target: Logger,
 453                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 454                          u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>,
 455         {
 456                 let preimage = payment_preimage
 457                         .unwrap_or_else(|| PaymentPreimage(entropy_source.get_secure_random_bytes()));
 458                 let payment_hash = PaymentHash(Sha256::hash(&preimage.0).into_inner());
 459                 self.pay_internal(payment_id, Some((payment_hash, &None, Some(preimage), retry_strategy)),
 460                         route_params, router, first_hops, inflight_htlcs, entropy_source, node_signer,
 461                         best_block_height, logger, &send_payment_along_path)
 462                         .map(|()| payment_hash)
 463                         .map_err(|e| { self.remove_outbound_if_all_failed(payment_id, &e); e })
 464         }
 465
 466         pub(super) fn send_spontaneous_payment_with_route<ES: Deref, NS: Deref, F>(
 467                 &self, route: &Route, payment_preimage: Option<PaymentPreimage>, payment_id: PaymentId,
 468                 entropy_source: &ES, node_signer: &NS, best_block_height: u32, send_payment_along_path: F
 469         ) -> Result<PaymentHash, PaymentSendFailure>
 470         where
 471                 ES::Target: EntropySource,
 472                 NS::Target: NodeSigner,
 473                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 474                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 475         {
 476                 let preimage = payment_preimage
 477                         .unwrap_or_else(|| PaymentPreimage(entropy_source.get_secure_random_bytes()));
 478                 let payment_hash = PaymentHash(Sha256::hash(&preimage.0).into_inner());
 479                 let onion_session_privs = self.add_new_pending_payment(payment_hash, None, payment_id, Some(preimage), &route, None, None, entropy_source, best_block_height)?;
 480
 481                 match self.pay_route_internal(route, payment_hash, &None, Some(preimage), payment_id, None, onion_session_privs, node_signer, best_block_height, &send_payment_along_path) {
 482                         Ok(()) => Ok(payment_hash),
 483                         Err(e) => {
 484                                 self.remove_outbound_if_all_failed(payment_id, &e);
 485                                 Err(e)
 486                         }
 487                 }
 488         }
 489
 490         pub(super) fn check_retry_payments<R: Deref, ES: Deref, NS: Deref, SP, IH, FH, L: Deref>(
 491                 &self, router: &R, first_hops: FH, inflight_htlcs: IH, entropy_source: &ES, node_signer: &NS,
 492                 best_block_height: u32, logger: &L, send_payment_along_path: SP,
 493         )
 494         where
 495                 R::Target: Router,
 496                 ES::Target: EntropySource,
 497                 NS::Target: NodeSigner,
 498                 SP: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 499                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>,
 500                 IH: Fn() -> InFlightHtlcs,
 501                 FH: Fn() -> Vec<ChannelDetails>,
 502                 L::Target: Logger,
 503         {
 504                 loop {
 505                         let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 506                         let mut retry_id_route_params = None;
 507                         for (pmt_id, pmt) in outbounds.iter_mut() {
 508                                 if pmt.is_auto_retryable_now() {
 509                                         if let PendingOutboundPayment::Retryable { pending_amt_msat, total_msat, payment_params: Some(params), .. } = pmt {
 510                                                 if pending_amt_msat < total_msat {
 511                                                         retry_id_route_params = Some((*pmt_id, RouteParameters {
 512                                                                 final_value_msat: *total_msat - *pending_amt_msat,
 513                                                                 final_cltv_expiry_delta:
 514                                                                         if let Some(delta) = params.final_cltv_expiry_delta { delta }
 515                                                                         else {
 516                                                                                 debug_assert!(false, "We always set the final_cltv_expiry_delta when a path fails");
 517                                                                                 LDK_DEFAULT_MIN_FINAL_CLTV_EXPIRY_DELTA.into()
 518                                                                         },
 519                                                                 payment_params: params.clone(),
 520                                                         }));
 521                                                         break
 522                                                 }
 523                                         }
 524                                 }
 525                         }
 526                         if let Some((payment_id, route_params)) = retry_id_route_params {
 527                                 core::mem::drop(outbounds);
 528                                 if let Err(e) = self.pay_internal(payment_id, None, route_params, router, first_hops(), inflight_htlcs(), entropy_source, node_signer, best_block_height, logger, &send_payment_along_path) {
 529                                         log_info!(logger, "Errored retrying payment: {:?}", e);
 530                                 }
 531                         } else { break }
 532                 }
 533         }
 534
 535         fn pay_internal<R: Deref, NS: Deref, ES: Deref, F, L: Deref>(
 536                 &self, payment_id: PaymentId,
 537                 initial_send_info: Option<(PaymentHash, &Option<PaymentSecret>, Option<PaymentPreimage>, Retry)>,
 538                 route_params: RouteParameters, router: &R, first_hops: Vec<ChannelDetails>,
 539                 inflight_htlcs: InFlightHtlcs, entropy_source: &ES, node_signer: &NS, best_block_height: u32,
 540                 logger: &L, send_payment_along_path: &F,
 541         ) -> Result<(), PaymentSendFailure>
 542         where
 543                 R::Target: Router,
 544                 ES::Target: EntropySource,
 545                 NS::Target: NodeSigner,
 546                 L::Target: Logger,
 547                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 548                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 549         {
 550                 #[cfg(feature = "std")] {
 551                         if has_expired(&route_params) {
 552                                 return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 553                                         err: format!("Invoice expired for payment id {}", log_bytes!(payment_id.0)),
 554                                 }))
 555                         }
 556                 }
 557
 558                 let route = router.find_route(
 559                         &node_signer.get_node_id(Recipient::Node).unwrap(), &route_params,
 560                         Some(&first_hops.iter().collect::<Vec<_>>()), &inflight_htlcs
 561                 ).map_err(|e| PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 562                         err: format!("Failed to find a route for payment {}: {:?}", log_bytes!(payment_id.0), e), // TODO: add APIError::RouteNotFound
 563                 }))?;
 564
 565                 let res = if let Some((payment_hash, payment_secret, keysend_preimage, retry_strategy)) = initial_send_info {
 566                         let onion_session_privs = self.add_new_pending_payment(payment_hash, *payment_secret, payment_id, keysend_preimage, &route, Some(retry_strategy), Some(route_params.payment_params.clone()), entropy_source, best_block_height)?;
 567                         self.pay_route_internal(&route, payment_hash, payment_secret, None, payment_id, None, onion_session_privs, node_signer, best_block_height, send_payment_along_path)
 568                 } else {
 569                         self.retry_payment_with_route(&route, payment_id, entropy_source, node_signer, best_block_height, send_payment_along_path)
 570                 };
 571                 match res {
 572                         Err(PaymentSendFailure::AllFailedResendSafe(_)) => {
 573                                 let retry_res = self.pay_internal(payment_id, None, route_params, router, first_hops, inflight_htlcs, entropy_source, node_signer, best_block_height, logger, send_payment_along_path);
 574                                 log_info!(logger, "Result retrying payment id {}: {:?}", log_bytes!(payment_id.0), retry_res);
 575                                 if let Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError { err })) = &retry_res {
 576                                         if err.starts_with("Retries exhausted ") { return res; }
 577                                 }
 578                                 retry_res
 579                         },
 580                         Err(PaymentSendFailure::PartialFailure { failed_paths_retry: Some(retry), .. }) => {
 581                                 // Some paths were sent, even if we failed to send the full MPP value our recipient may
 582                                 // misbehave and claim the funds, at which point we have to consider the payment sent, so
 583                                 // return `Ok()` here, ignoring any retry errors.
 584                                 let retry_res = self.pay_internal(payment_id, None, retry, router, first_hops, inflight_htlcs, entropy_source, node_signer, best_block_height, logger, send_payment_along_path);
 585                                 log_info!(logger, "Result retrying payment id {}: {:?}", log_bytes!(payment_id.0), retry_res);
 586                                 Ok(())
 587                         },
 588                         Err(PaymentSendFailure::PartialFailure { failed_paths_retry: None, .. }) => {
 589                                 // This may happen if we send a payment and some paths fail, but only due to a temporary
 590                                 // monitor failure or the like, implying they're really in-flight, but we haven't sent the
 591                                 // initial HTLC-Add messages yet.
 592                                 Ok(())
 593                         },
 594                         res => res,
 595                 }
 596         }
 597
 598         pub(super) fn retry_payment_with_route<ES: Deref, NS: Deref, F>(
 599                 &self, route: &Route, payment_id: PaymentId, entropy_source: &ES, node_signer: &NS, best_block_height: u32,
 600                 send_payment_along_path: F
 601         ) -> Result<(), PaymentSendFailure>
 602         where
 603                 ES::Target: EntropySource,
 604                 NS::Target: NodeSigner,
 605                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 606                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 607         {
 608                 const RETRY_OVERFLOW_PERCENTAGE: u64 = 10;
 609                 for path in route.paths.iter() {
 610                         if path.len() == 0 {
 611                                 return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 612                                         err: "length-0 path in route".to_string()
 613                                 }))
 614                         }
 615                 }
 616
 617                 let mut onion_session_privs = Vec::with_capacity(route.paths.len());
 618                 for _ in 0..route.paths.len() {
 619                         onion_session_privs.push(entropy_source.get_secure_random_bytes());
 620                 }
 621
 622                 let (total_msat, payment_hash, payment_secret, keysend_preimage) = {
 623                         let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 624                         match outbounds.get_mut(&payment_id) {
 625                                 Some(payment) => {
 626                                         let res = match payment {
 627                                                 PendingOutboundPayment::Retryable {
 628                                                         total_msat, payment_hash, keysend_preimage, payment_secret, pending_amt_msat, ..
 629                                                 } => {
 630                                                         let retry_amt_msat: u64 = route.paths.iter().map(|path| path.last().unwrap().fee_msat).sum();
 631                                                         if retry_amt_msat + *pending_amt_msat > *total_msat * (100 + RETRY_OVERFLOW_PERCENTAGE) / 100 {
 632                                                                 return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 633                                                                         err: format!("retry_amt_msat of {} will put pending_amt_msat (currently: {}) more than 10% over total_payment_amt_msat of {}", retry_amt_msat, pending_amt_msat, total_msat).to_string()
 634                                                                 }))
 635                                                         }
 636                                                         (*total_msat, *payment_hash, *payment_secret, *keysend_preimage)
 637                                                 },
 638                                                 PendingOutboundPayment::Legacy { .. } => {
 639                                                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 640                                                                 err: "Unable to retry payments that were initially sent on LDK versions prior to 0.0.102".to_string()
 641                                                         }))
 642                                                 },
 643                                                 PendingOutboundPayment::Fulfilled { .. } => {
 644                                                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 645                                                                 err: "Payment already completed".to_owned()
 646                                                         }));
 647                                                 },
 648                                                 PendingOutboundPayment::Abandoned { .. } => {
 649                                                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 650                                                                 err: "Payment already abandoned (with some HTLCs still pending)".to_owned()
 651                                                         }));
 652                                                 },
 653                                         };
 654                                         if !payment.is_retryable_now() {
 655                                                 return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 656                                                         err: format!("Retries exhausted for payment id {}", log_bytes!(payment_id.0)),
 657                                                 }))
 658                                         }
 659                                         payment.increment_attempts();
 660                                         for (path, session_priv_bytes) in route.paths.iter().zip(onion_session_privs.iter()) {
 661                                                 assert!(payment.insert(*session_priv_bytes, path));
 662                                         }
 663                                         res
 664                                 },
 665                                 None =>
 666                                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 667                                                 err: format!("Payment with ID {} not found", log_bytes!(payment_id.0)),
 668                                         })),
 669                         }
 670                 };
 671                 self.pay_route_internal(route, payment_hash, &payment_secret, keysend_preimage, payment_id, Some(total_msat), onion_session_privs, node_signer, best_block_height, &send_payment_along_path)
 672         }
 673
 674         pub(super) fn send_probe<ES: Deref, NS: Deref, F>(
 675                 &self, hops: Vec<RouteHop>, probing_cookie_secret: [u8; 32], entropy_source: &ES,
 676                 node_signer: &NS, best_block_height: u32, send_payment_along_path: F
 677         ) -> Result<(PaymentHash, PaymentId), PaymentSendFailure>
 678         where
 679                 ES::Target: EntropySource,
 680                 NS::Target: NodeSigner,
 681                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 682                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 683         {
 684                 let payment_id = PaymentId(entropy_source.get_secure_random_bytes());
 685
 686                 let payment_hash = probing_cookie_from_id(&payment_id, probing_cookie_secret);
 687
 688                 if hops.len() < 2 {
 689                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 690                                 err: "No need probing a path with less than two hops".to_string()
 691                         }))
 692                 }
 693
 694                 let route = Route { paths: vec![hops], payment_params: None };
 695                 let onion_session_privs = self.add_new_pending_payment(payment_hash, None, payment_id, None, &route, None, None, entropy_source, best_block_height)?;
 696
 697                 match self.pay_route_internal(&route, payment_hash, &None, None, payment_id, None, onion_session_privs, node_signer, best_block_height, &send_payment_along_path) {
 698                         Ok(()) => Ok((payment_hash, payment_id)),
 699                         Err(e) => {
 700                                 self.remove_outbound_if_all_failed(payment_id, &e);
 701                                 Err(e)
 702                         }
 703                 }
 704         }
 705
 706         #[cfg(test)]
 707         pub(super) fn test_add_new_pending_payment<ES: Deref>(
 708                 &self, payment_hash: PaymentHash, payment_secret: Option<PaymentSecret>, payment_id: PaymentId,
 709                 route: &Route, retry_strategy: Option<Retry>, entropy_source: &ES, best_block_height: u32
 710         ) -> Result<Vec<[u8; 32]>, PaymentSendFailure> where ES::Target: EntropySource {
 711                 self.add_new_pending_payment(payment_hash, payment_secret, payment_id, None, route, retry_strategy, None, entropy_source, best_block_height)
 712         }
 713
 714         pub(super) fn add_new_pending_payment<ES: Deref>(
 715                 &self, payment_hash: PaymentHash, payment_secret: Option<PaymentSecret>, payment_id: PaymentId,
 716                 keysend_preimage: Option<PaymentPreimage>, route: &Route, retry_strategy: Option<Retry>,
 717                 payment_params: Option<PaymentParameters>, entropy_source: &ES, best_block_height: u32
 718         ) -> Result<Vec<[u8; 32]>, PaymentSendFailure> where ES::Target: EntropySource {
 719                 let mut onion_session_privs = Vec::with_capacity(route.paths.len());
 720                 for _ in 0..route.paths.len() {
 721                         onion_session_privs.push(entropy_source.get_secure_random_bytes());
 722                 }
 723
 724                 let mut pending_outbounds = self.pending_outbound_payments.lock().unwrap();
 725                 match pending_outbounds.entry(payment_id) {
 726                         hash_map::Entry::Occupied(_) => Err(PaymentSendFailure::DuplicatePayment),
 727                         hash_map::Entry::Vacant(entry) => {
 728                                 let payment = entry.insert(PendingOutboundPayment::Retryable {
 729                                         retry_strategy,
 730                                         attempts: PaymentAttempts::new(),
 731                                         payment_params,
 732                                         session_privs: HashSet::new(),
 733                                         pending_amt_msat: 0,
 734                                         pending_fee_msat: Some(0),
 735                                         payment_hash,
 736                                         payment_secret,
 737                                         keysend_preimage,
 738                                         starting_block_height: best_block_height,
 739                                         total_msat: route.get_total_amount(),
 740                                 });
 741
 742                                 for (path, session_priv_bytes) in route.paths.iter().zip(onion_session_privs.iter()) {
 743                                         assert!(payment.insert(*session_priv_bytes, path));
 744                                 }
 745
 746                                 Ok(onion_session_privs)
 747                         },
 748                 }
 749         }
 750
 751         fn pay_route_internal<NS: Deref, F>(
 752                 &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>,
 753                 keysend_preimage: Option<PaymentPreimage>, payment_id: PaymentId, recv_value_msat: Option<u64>,
 754                 onion_session_privs: Vec<[u8; 32]>, node_signer: &NS, best_block_height: u32,
 755                 send_payment_along_path: &F
 756         ) -> Result<(), PaymentSendFailure>
 757         where
 758                 NS::Target: NodeSigner,
 759                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 760                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 761         {
 762                 if route.paths.len() < 1 {
 763                         return Err(PaymentSendFailure::ParameterError(APIError::InvalidRoute{err: "There must be at least one path to send over"}));
 764                 }
 765                 if payment_secret.is_none() && route.paths.len() > 1 {
 766                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError{err: "Payment secret is required for multi-path payments".to_string()}));
 767                 }
 768                 let mut total_value = 0;
 769                 let our_node_id = node_signer.get_node_id(Recipient::Node).unwrap(); // TODO no unwrap
 770                 let mut path_errs = Vec::with_capacity(route.paths.len());
 771                 'path_check: for path in route.paths.iter() {
 772                         if path.len() < 1 || path.len() > 20 {
 773                                 path_errs.push(Err(APIError::InvalidRoute{err: "Path didn't go anywhere/had bogus size"}));
 774                                 continue 'path_check;
 775                         }
 776                         for (idx, hop) in path.iter().enumerate() {
 777                                 if idx != path.len() - 1 && hop.pubkey == our_node_id {
 778                                         path_errs.push(Err(APIError::InvalidRoute{err: "Path went through us but wasn't a simple rebalance loop to us"}));
 779                                         continue 'path_check;
 780                                 }
 781                         }
 782                         total_value += path.last().unwrap().fee_msat;
 783                         path_errs.push(Ok(()));
 784                 }
 785                 if path_errs.iter().any(|e| e.is_err()) {
 786                         return Err(PaymentSendFailure::PathParameterError(path_errs));
 787                 }
 788                 if let Some(amt_msat) = recv_value_msat {
 789                         debug_assert!(amt_msat >= total_value);
 790                         total_value = amt_msat;
 791                 }
 792
 793                 let cur_height = best_block_height + 1;
 794                 let mut results = Vec::new();
 795                 debug_assert_eq!(route.paths.len(), onion_session_privs.len());
 796                 for (path, session_priv) in route.paths.iter().zip(onion_session_privs.into_iter()) {
 797                         let mut path_res = send_payment_along_path(&path, &route.payment_params, &payment_hash, payment_secret, total_value, cur_height, payment_id, &keysend_preimage, session_priv);
 798                         match path_res {
 799                                 Ok(_) => {},
 800                                 Err(APIError::MonitorUpdateInProgress) => {
 801                                         // While a MonitorUpdateInProgress is an Err(_), the payment is still
 802                                         // considered "in flight" and we shouldn't remove it from the
 803                                         // PendingOutboundPayment set.
 804                                 },
 805                                 Err(_) => {
 806                                         let mut pending_outbounds = self.pending_outbound_payments.lock().unwrap();
 807                                         if let Some(payment) = pending_outbounds.get_mut(&payment_id) {
 808                                                 let removed = payment.remove(&session_priv, Some(path));
 809                                                 debug_assert!(removed, "This can't happen as the payment has an entry for this path added by callers");
 810                                         } else {
 811                                                 debug_assert!(false, "This can't happen as the payment was added by callers");
 812                                                 path_res = Err(APIError::APIMisuseError { err: "Internal error: payment disappeared during processing. Please report this bug!".to_owned() });
 813                                         }
 814                                 }
 815                         }
 816                         results.push(path_res);
 817                 }
 818                 let mut has_ok = false;
 819                 let mut has_err = false;
 820                 let mut pending_amt_unsent = 0;
 821                 let mut max_unsent_cltv_delta = 0;
 822                 for (res, path) in results.iter().zip(route.paths.iter()) {
 823                         if res.is_ok() { has_ok = true; }
 824                         if res.is_err() { has_err = true; }
 825                         if let &Err(APIError::MonitorUpdateInProgress) = res {
 826                                 // MonitorUpdateInProgress is inherently unsafe to retry, so we call it a
 827                                 // PartialFailure.
 828                                 has_err = true;
 829                                 has_ok = true;
 830                         } else if res.is_err() {
 831                                 pending_amt_unsent += path.last().unwrap().fee_msat;
 832                                 max_unsent_cltv_delta = cmp::max(max_unsent_cltv_delta, path.last().unwrap().cltv_expiry_delta);
 833                         }
 834                 }
 835                 if has_err && has_ok {
 836                         Err(PaymentSendFailure::PartialFailure {
 837                                 results,
 838                                 payment_id,
 839                                 failed_paths_retry: if pending_amt_unsent != 0 {
 840                                         if let Some(payment_params) = &route.payment_params {
 841                                                 Some(RouteParameters {
 842                                                         payment_params: payment_params.clone(),
 843                                                         final_value_msat: pending_amt_unsent,
 844                                                         final_cltv_expiry_delta:
 845                                                                 if let Some(delta) = payment_params.final_cltv_expiry_delta { delta }
 846                                                                 else { max_unsent_cltv_delta },
 847                                                 })
 848                                         } else { None }
 849                                 } else { None },
 850                         })
 851                 } else if has_err {
 852                         Err(PaymentSendFailure::AllFailedResendSafe(results.drain(..).map(|r| r.unwrap_err()).collect()))
 853                 } else {
 854                         Ok(())
 855                 }
 856         }
 857
 858         #[cfg(test)]
 859         pub(super) fn test_send_payment_internal<NS: Deref, F>(
 860                 &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>,
 861                 keysend_preimage: Option<PaymentPreimage>, payment_id: PaymentId, recv_value_msat: Option<u64>,
 862                 onion_session_privs: Vec<[u8; 32]>, node_signer: &NS, best_block_height: u32,
 863                 send_payment_along_path: F
 864         ) -> Result<(), PaymentSendFailure>
 865         where
 866                 NS::Target: NodeSigner,
 867                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 868                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 869         {
 870                 self.pay_route_internal(route, payment_hash, payment_secret, keysend_preimage, payment_id,
 871                         recv_value_msat, onion_session_privs, node_signer, best_block_height,
 872                         &send_payment_along_path)
 873                         .map_err(|e| { self.remove_outbound_if_all_failed(payment_id, &e); e })
 874         }
 875
 876         // If we failed to send any paths, we should remove the new PaymentId from the
 877         // `pending_outbound_payments` map, as the user isn't expected to `abandon_payment`.
 878         fn remove_outbound_if_all_failed(&self, payment_id: PaymentId, err: &PaymentSendFailure) {
 879                 if let &PaymentSendFailure::AllFailedResendSafe(_) = err {
 880                         let removed = self.pending_outbound_payments.lock().unwrap().remove(&payment_id).is_some();
 881                         debug_assert!(removed, "We should always have a pending payment to remove here");
 882                 }
 883         }
 884
 885         pub(super) fn claim_htlc<L: Deref>(
 886                 &self, payment_id: PaymentId, payment_preimage: PaymentPreimage, session_priv: SecretKey,
 887                 path: Vec<RouteHop>, from_onchain: bool, pending_events: &Mutex<Vec<events::Event>>, logger: &L
 888         ) where L::Target: Logger {
 889                 let mut session_priv_bytes = [0; 32];
 890                 session_priv_bytes.copy_from_slice(&session_priv[..]);
 891                 let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 892                 let mut pending_events = pending_events.lock().unwrap();
 893                 if let hash_map::Entry::Occupied(mut payment) = outbounds.entry(payment_id) {
 894                         if !payment.get().is_fulfilled() {
 895                                 let payment_hash = PaymentHash(Sha256::hash(&payment_preimage.0).into_inner());
 896                                 let fee_paid_msat = payment.get().get_pending_fee_msat();
 897                                 pending_events.push(
 898                                         events::Event::PaymentSent {
 899                                                 payment_id: Some(payment_id),
 900                                                 payment_preimage,
 901                                                 payment_hash,
 902                                                 fee_paid_msat,
 903                                         }
 904                                 );
 905                                 payment.get_mut().mark_fulfilled();
 906                         }
 907
 908                         if from_onchain {
 909                                 // We currently immediately remove HTLCs which were fulfilled on-chain.
 910                                 // This could potentially lead to removing a pending payment too early,
 911                                 // with a reorg of one block causing us to re-add the fulfilled payment on
 912                                 // restart.
 913                                 // TODO: We should have a second monitor event that informs us of payments
 914                                 // irrevocably fulfilled.
 915                                 if payment.get_mut().remove(&session_priv_bytes, Some(&path)) {
 916                                         let payment_hash = Some(PaymentHash(Sha256::hash(&payment_preimage.0).into_inner()));
 917                                         pending_events.push(
 918                                                 events::Event::PaymentPathSuccessful {
 919                                                         payment_id,
 920                                                         payment_hash,
 921                                                         path,
 922                                                 }
 923                                         );
 924                                 }
 925                         }
 926                 } else {
 927                         log_trace!(logger, "Received duplicative fulfill for HTLC with payment_preimage {}", log_bytes!(payment_preimage.0));
 928                 }
 929         }
 930
 931         pub(super) fn finalize_claims(&self, sources: Vec<HTLCSource>, pending_events: &Mutex<Vec<events::Event>>) {
 932                 let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 933                 let mut pending_events = pending_events.lock().unwrap();
 934                 for source in sources {
 935                         if let HTLCSource::OutboundRoute { session_priv, payment_id, path, .. } = source {
 936                                 let mut session_priv_bytes = [0; 32];
 937                                 session_priv_bytes.copy_from_slice(&session_priv[..]);
 938                                 if let hash_map::Entry::Occupied(mut payment) = outbounds.entry(payment_id) {
 939                                         assert!(payment.get().is_fulfilled());
 940                                         if payment.get_mut().remove(&session_priv_bytes, None) {
 941                                                 pending_events.push(
 942                                                         events::Event::PaymentPathSuccessful {
 943                                                                 payment_id,
 944                                                                 payment_hash: payment.get().payment_hash(),
 945                                                                 path,
 946                                                         }
 947                                                 );
 948                                         }
 949                                 }
 950                         }
 951                 }
 952         }
 953
 954         pub(super) fn remove_stale_resolved_payments(&self, pending_events: &Mutex<Vec<events::Event>>) {
 955                 // If an outbound payment was completed, and no pending HTLCs remain, we should remove it
 956                 // from the map. However, if we did that immediately when the last payment HTLC is claimed,
 957                 // this could race the user making a duplicate send_payment call and our idempotency
 958                 // guarantees would be violated. Instead, we wait a few timer ticks to do the actual
 959                 // removal. This should be more than sufficient to ensure the idempotency of any
 960                 // `send_payment` calls that were made at the same time the `PaymentSent` event was being
 961                 // processed.
 962                 let mut pending_outbound_payments = self.pending_outbound_payments.lock().unwrap();
 963                 let pending_events = pending_events.lock().unwrap();
 964                 pending_outbound_payments.retain(|payment_id, payment| {
 965                         if let PendingOutboundPayment::Fulfilled { session_privs, timer_ticks_without_htlcs, .. } = payment {
 966                                 let mut no_remaining_entries = session_privs.is_empty();
 967                                 if no_remaining_entries {
 968                                         for ev in pending_events.iter() {
 969                                                 match ev {
 970                                                         events::Event::PaymentSent { payment_id: Some(ev_payment_id), .. } |
 971                                                                 events::Event::PaymentPathSuccessful { payment_id: ev_payment_id, .. } |
 972                                                                 events::Event::PaymentPathFailed { payment_id: Some(ev_payment_id), .. } => {
 973                                                                         if payment_id == ev_payment_id {
 974                                                                                 no_remaining_entries = false;
 975                                                                                 break;
 976                                                                         }
 977                                                                 },
 978                                                         _ => {},
 979                                                 }
 980                                         }
 981                                 }
 982                                 if no_remaining_entries {
 983                                         *timer_ticks_without_htlcs += 1;
 984                                         *timer_ticks_without_htlcs <= IDEMPOTENCY_TIMEOUT_TICKS
 985                                 } else {
 986                                         *timer_ticks_without_htlcs = 0;
 987                                         true
 988                                 }
 989                         } else { true }
 990                 });
 991         }
 992
 993         pub(super) fn fail_htlc<L: Deref>(
 994                 &self, source: &HTLCSource, payment_hash: &PaymentHash, onion_error: &HTLCFailReason,
 995                 path: &Vec<RouteHop>, session_priv: &SecretKey, payment_id: &PaymentId,
 996                 payment_params: &Option<PaymentParameters>, probing_cookie_secret: [u8; 32],
 997                 secp_ctx: &Secp256k1<secp256k1::All>, pending_events: &Mutex<Vec<events::Event>>, logger: &L
 998         ) where L::Target: Logger {
 999                 #[cfg(test)]
1000                 let (network_update, short_channel_id, payment_retryable, onion_error_code, onion_error_data) = onion_error.decode_onion_failure(secp_ctx, logger, &source);
1001                 #[cfg(not(test))]
1002                 let (network_update, short_channel_id, payment_retryable, _, _) = onion_error.decode_onion_failure(secp_ctx, logger, &source);
1003
1004                 let mut session_priv_bytes = [0; 32];
1005                 session_priv_bytes.copy_from_slice(&session_priv[..]);
1006                 let mut outbounds = self.pending_outbound_payments.lock().unwrap();
1007                 let mut all_paths_failed = false;
1008                 let mut full_failure_ev = None;
1009                 let mut pending_retry_ev = None;
1010                 let mut retry = None;
1011                 let attempts_remaining = if let hash_map::Entry::Occupied(mut payment) = outbounds.entry(*payment_id) {
1012                         if !payment.get_mut().remove(&session_priv_bytes, Some(&path)) {
1013                                 log_trace!(logger, "Received duplicative fail for HTLC with payment_hash {}", log_bytes!(payment_hash.0));
1014                                 return
1015                         }
1016                         if payment.get().is_fulfilled() {
1017                                 log_trace!(logger, "Received failure of HTLC with payment_hash {} after payment completion", log_bytes!(payment_hash.0));
1018                                 return
1019                         }
1020                         let is_retryable_now = payment.get().is_auto_retryable_now();
1021                         if let Some(scid) = short_channel_id {
1022                                 payment.get_mut().insert_previously_failed_scid(scid);
1023                         }
1024
1025                         // We want to move towards only using the `PaymentParameters` in the outbound payments
1026                         // map. However, for backwards-compatibility, we still need to support passing the
1027                         // `PaymentParameters` data that was shoved in the HTLC (and given to us via
1028                         // `payment_params`) back to the user.
1029                         let path_last_hop = path.last().expect("Outbound payments must have had a valid path");
1030                         if let Some(params) = payment.get_mut().payment_parameters() {
1031                                 if params.final_cltv_expiry_delta.is_none() {
1032                                         // This should be rare, but a user could provide None for the payment data, and
1033                                         // we need it when we go to retry the payment, so fill it in.
1034                                         params.final_cltv_expiry_delta = Some(path_last_hop.cltv_expiry_delta);
1035                                 }
1036                                 retry = Some(RouteParameters {
1037                                         payment_params: params.clone(),
1038                                         final_value_msat: path_last_hop.fee_msat,
1039                                         final_cltv_expiry_delta: params.final_cltv_expiry_delta.unwrap(),
1040                                 });
1041                         } else if let Some(params) = payment_params {
1042                                 retry = Some(RouteParameters {
1043                                         payment_params: params.clone(),
1044                                         final_value_msat: path_last_hop.fee_msat,
1045                                         final_cltv_expiry_delta:
1046                                                 if let Some(delta) = params.final_cltv_expiry_delta { delta }
1047                                                 else { path_last_hop.cltv_expiry_delta },
1048                                 });
1049                         }
1050
1051                         if payment.get().remaining_parts() == 0 {
1052                                 all_paths_failed = true;
1053                                 if payment.get().abandoned() {
1054                                         full_failure_ev = Some(events::Event::PaymentFailed {
1055                                                 payment_id: *payment_id,
1056                                                 payment_hash: payment.get().payment_hash().expect("PendingOutboundPayments::RetriesExceeded always has a payment hash set"),
1057                                         });
1058                                         payment.remove();
1059                                 }
1060                         }
1061                         is_retryable_now
1062                 } else {
1063                         log_trace!(logger, "Received duplicative fail for HTLC with payment_hash {}", log_bytes!(payment_hash.0));
1064                         return
1065                 };
1066                 core::mem::drop(outbounds);
1067                 log_trace!(logger, "Failing outbound payment HTLC with payment_hash {}", log_bytes!(payment_hash.0));
1068
1069                 let path_failure = {
1070                         if payment_is_probe(payment_hash, &payment_id, probing_cookie_secret) {
1071                                 if !payment_retryable {
1072                                         events::Event::ProbeSuccessful {
1073                                                 payment_id: *payment_id,
1074                                                 payment_hash: payment_hash.clone(),
1075                                                 path: path.clone(),
1076                                         }
1077                                 } else {
1078                                         events::Event::ProbeFailed {
1079                                                 payment_id: *payment_id,
1080                                                 payment_hash: payment_hash.clone(),
1081                                                 path: path.clone(),
1082                                                 short_channel_id,
1083                                         }
1084                                 }
1085                         } else {
1086                                 // TODO: If we decided to blame ourselves (or one of our channels) in
1087                                 // process_onion_failure we should close that channel as it implies our
1088                                 // next-hop is needlessly blaming us!
1089                                 if let Some(scid) = short_channel_id {
1090                                         retry.as_mut().map(|r| r.payment_params.previously_failed_channels.push(scid));
1091                                 }
1092                                 if payment_retryable && attempts_remaining && retry.is_some() {
1093                                         debug_assert!(full_failure_ev.is_none());
1094                                         pending_retry_ev = Some(events::Event::PendingHTLCsForwardable {
1095                                                 time_forwardable: Duration::from_millis(MIN_HTLC_RELAY_HOLDING_CELL_MILLIS),
1096                                         });
1097                                 }
1098                                 events::Event::PaymentPathFailed {
1099                                         payment_id: Some(*payment_id),
1100                                         payment_hash: payment_hash.clone(),
1101                                         payment_failed_permanently: !payment_retryable,
1102                                         network_update,
1103                                         all_paths_failed,
1104                                         path: path.clone(),
1105                                         short_channel_id,
1106                                         retry,
1107                                         #[cfg(test)]
1108                                         error_code: onion_error_code,
1109                                         #[cfg(test)]
1110                                         error_data: onion_error_data
1111                                 }
1112                         }
1113                 };
1114                 let mut pending_events = pending_events.lock().unwrap();
1115                 pending_events.push(path_failure);
1116                 if let Some(ev) = full_failure_ev { pending_events.push(ev); }
1117                 if let Some(ev) = pending_retry_ev { pending_events.push(ev); }
1118         }
1119
1120         pub(super) fn abandon_payment(&self, payment_id: PaymentId) -> Option<events::Event> {
1121                 let mut failed_ev = None;
1122                 let mut outbounds = self.pending_outbound_payments.lock().unwrap();
1123                 if let hash_map::Entry::Occupied(mut payment) = outbounds.entry(payment_id) {
1124                         if let Ok(()) = payment.get_mut().mark_abandoned() {
1125                                 if payment.get().remaining_parts() == 0 {
1126                                         failed_ev = Some(events::Event::PaymentFailed {
1127                                                 payment_id,
1128                                                 payment_hash: payment.get().payment_hash().expect("PendingOutboundPayments::RetriesExceeded always has a payment hash set"),
1129                                         });
1130                                         payment.remove();
1131                                 }
1132                         }
1133                 }
1134                 failed_ev
1135         }
1136
1137         #[cfg(test)]
1138         pub fn has_pending_payments(&self) -> bool {
1139                 !self.pending_outbound_payments.lock().unwrap().is_empty()
1140         }
1141
1142         #[cfg(test)]
1143         pub fn clear_pending_payments(&self) {
1144                 self.pending_outbound_payments.lock().unwrap().clear()
1145         }
1146 }
1147
1148 /// Returns whether a payment with the given [`PaymentHash`] and [`PaymentId`] is, in fact, a
1149 /// payment probe.
1150 pub(super) fn payment_is_probe(payment_hash: &PaymentHash, payment_id: &PaymentId,
1151         probing_cookie_secret: [u8; 32]) -> bool
1152 {
1153         let target_payment_hash = probing_cookie_from_id(payment_id, probing_cookie_secret);
1154         target_payment_hash == *payment_hash
1155 }
1156
1157 /// Returns the 'probing cookie' for the given [`PaymentId`].
1158 fn probing_cookie_from_id(payment_id: &PaymentId, probing_cookie_secret: [u8; 32]) -> PaymentHash {
1159         let mut preimage = [0u8; 64];
1160         preimage[..32].copy_from_slice(&probing_cookie_secret);
1161         preimage[32..].copy_from_slice(&payment_id.0);
1162         PaymentHash(Sha256::hash(&preimage).into_inner())
1163 }
1164
1165 impl_writeable_tlv_based_enum_upgradable!(PendingOutboundPayment,
1166         (0, Legacy) => {
1167                 (0, session_privs, required),
1168         },
1169         (1, Fulfilled) => {
1170                 (0, session_privs, required),
1171                 (1, payment_hash, option),
1172                 (3, timer_ticks_without_htlcs, (default_value, 0)),
1173         },
1174         (2, Retryable) => {
1175                 (0, session_privs, required),
1176                 (1, pending_fee_msat, option),
1177                 (2, payment_hash, required),
1178                 (3, payment_params, option),
1179                 (4, payment_secret, option),
1180                 (5, keysend_preimage, option),
1181                 (6, total_msat, required),
1182                 (8, pending_amt_msat, required),
1183                 (10, starting_block_height, required),
1184                 (not_written, retry_strategy, (static_value, None)),
1185                 (not_written, attempts, (static_value, PaymentAttempts::new())),
1186         },
1187         (3, Abandoned) => {
1188                 (0, session_privs, required),
1189                 (2, payment_hash, required),
1190         },
1191 );
1192
1193 #[cfg(test)]
1194 mod tests {
1195         use bitcoin::blockdata::constants::genesis_block;
1196         use bitcoin::network::constants::Network;
1197         use bitcoin::secp256k1::{PublicKey, Secp256k1, SecretKey};
1198
1199         use crate::ln::PaymentHash;
1200         use crate::ln::channelmanager::{PaymentId, PaymentSendFailure};
1201         use crate::ln::msgs::{ErrorAction, LightningError};
1202         use crate::ln::outbound_payment::{OutboundPayments, Retry};
1203         use crate::routing::gossip::NetworkGraph;
1204         use crate::routing::router::{InFlightHtlcs, PaymentParameters, Route, RouteParameters};
1205         use crate::sync::Arc;
1206         use crate::util::errors::APIError;
1207         use crate::util::test_utils;
1208
1209         #[test]
1210         #[cfg(feature = "std")]
1211         fn fails_paying_after_expiration() {
1212                 do_fails_paying_after_expiration(false);
1213                 do_fails_paying_after_expiration(true);
1214         }
1215         #[cfg(feature = "std")]
1216         fn do_fails_paying_after_expiration(on_retry: bool) {
1217                 let outbound_payments = OutboundPayments::new();
1218                 let logger = test_utils::TestLogger::new();
1219                 let genesis_hash = genesis_block(Network::Testnet).header.block_hash();
1220                 let network_graph = Arc::new(NetworkGraph::new(genesis_hash, &logger));
1221                 let router = test_utils::TestRouter::new(network_graph);
1222                 let secp_ctx = Secp256k1::new();
1223                 let keys_manager = test_utils::TestKeysInterface::new(&[0; 32], Network::Testnet);
1224
1225                 let past_expiry_time = std::time::SystemTime::UNIX_EPOCH.elapsed().unwrap().as_secs() - 2;
1226                 let payment_params = PaymentParameters::from_node_id(
1227                                 PublicKey::from_secret_key(&secp_ctx, &SecretKey::from_slice(&[42; 32]).unwrap()),
1228                                 0
1229                         ).with_expiry_time(past_expiry_time);
1230                 let expired_route_params = RouteParameters {
1231                         payment_params,
1232                         final_value_msat: 0,
1233                         final_cltv_expiry_delta: 0,
1234                 };
1235                 let err = if on_retry {
1236                         outbound_payments.pay_internal(
1237                                 PaymentId([0; 32]), None, expired_route_params, &&router, vec![], InFlightHtlcs::new(),
1238                                 &&keys_manager, &&keys_manager, 0, &&logger, &|_, _, _, _, _, _, _, _, _| Ok(())).unwrap_err()
1239                 } else {
1240                         outbound_payments.send_payment(
1241                                 PaymentHash([0; 32]), &None, PaymentId([0; 32]), Retry::Attempts(0), expired_route_params,
1242                                 &&router, vec![], InFlightHtlcs::new(), &&keys_manager, &&keys_manager, 0, &&logger,
1243                                 |_, _, _, _, _, _, _, _, _| Ok(())).unwrap_err()
1244                 };
1245                 if let PaymentSendFailure::ParameterError(APIError::APIMisuseError { err }) = err {
1246                         assert!(err.contains("Invoice expired"));
1247                 } else { panic!("Unexpected error"); }
1248         }
1249
1250         #[test]
1251         fn find_route_error() {
1252                 do_find_route_error(false);
1253                 do_find_route_error(true);
1254         }
1255         fn do_find_route_error(on_retry: bool) {
1256                 let outbound_payments = OutboundPayments::new();
1257                 let logger = test_utils::TestLogger::new();
1258                 let genesis_hash = genesis_block(Network::Testnet).header.block_hash();
1259                 let network_graph = Arc::new(NetworkGraph::new(genesis_hash, &logger));
1260                 let router = test_utils::TestRouter::new(network_graph);
1261                 let secp_ctx = Secp256k1::new();
1262                 let keys_manager = test_utils::TestKeysInterface::new(&[0; 32], Network::Testnet);
1263
1264                 let payment_params = PaymentParameters::from_node_id(
1265                         PublicKey::from_secret_key(&secp_ctx, &SecretKey::from_slice(&[42; 32]).unwrap()), 0);
1266                 let route_params = RouteParameters {
1267                         payment_params,
1268                         final_value_msat: 0,
1269                         final_cltv_expiry_delta: 0,
1270                 };
1271                 router.expect_find_route(route_params.clone(),
1272                         Err(LightningError { err: String::new(), action: ErrorAction::IgnoreError }));
1273
1274                 let err = if on_retry {
1275                         outbound_payments.add_new_pending_payment(PaymentHash([0; 32]), None, PaymentId([0; 32]), None,
1276                                 &Route { paths: vec![], payment_params: None }, Some(Retry::Attempts(1)),
1277                                 Some(route_params.payment_params.clone()), &&keys_manager, 0).unwrap();
1278                         outbound_payments.pay_internal(
1279                                 PaymentId([0; 32]), None, route_params, &&router, vec![], InFlightHtlcs::new(),
1280                                 &&keys_manager, &&keys_manager, 0, &&logger, &|_, _, _, _, _, _, _, _, _| Ok(())).unwrap_err()
1281                 } else {
1282                         outbound_payments.send_payment(
1283                                 PaymentHash([0; 32]), &None, PaymentId([0; 32]), Retry::Attempts(0), route_params,
1284                                 &&router, vec![], InFlightHtlcs::new(), &&keys_manager, &&keys_manager, 0, &&logger,
1285                                 |_, _, _, _, _, _, _, _, _| Ok(())).unwrap_err()
1286                 };
1287                 if let PaymentSendFailure::ParameterError(APIError::APIMisuseError { err }) = err {
1288                         assert!(err.contains("Failed to find a route"));
1289                 } else { panic!("Unexpected error"); }
1290         }
1291 }