_ 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::{KeysInterface, Recipient};
  17 use crate::ln::{PaymentHash, PaymentPreimage, PaymentSecret};
  18 use crate::ln::channelmanager::{HTLCSource, IDEMPOTENCY_TIMEOUT_TICKS, PaymentId};
  19 use crate::ln::msgs::DecodeError;
  20 use crate::ln::onion_utils::HTLCFailReason;
  21 use crate::routing::router::{PaymentParameters, Route, RouteHop, RouteParameters, RoutePath};
  22 use crate::util::errors::APIError;
  23 use crate::util::events;
  24 use crate::util::logger::Logger;
  25
  26 use core::cmp;
  27 use core::ops::Deref;
  28 use crate::prelude::*;
  29 use crate::sync::Mutex;
  30
  31 /// Stores the session_priv for each part of a payment that is still pending. For versions 0.0.102
  32 /// and later, also stores information for retrying the payment.
  33 pub(crate) enum PendingOutboundPayment {
  34         Legacy {
  35                 session_privs: HashSet<[u8; 32]>,
  36         },
  37         Retryable {
  38                 session_privs: HashSet<[u8; 32]>,
  39                 payment_hash: PaymentHash,
  40                 payment_secret: Option<PaymentSecret>,
  41                 pending_amt_msat: u64,
  42                 /// Used to track the fee paid. Only present if the payment was serialized on 0.0.103+.
  43                 pending_fee_msat: Option<u64>,
  44                 /// The total payment amount across all paths, used to verify that a retry is not overpaying.
  45                 total_msat: u64,
  46                 /// Our best known block height at the time this payment was initiated.
  47                 starting_block_height: u32,
  48         },
  49         /// When a pending payment is fulfilled, we continue tracking it until all pending HTLCs have
  50         /// been resolved. This ensures we don't look up pending payments in ChannelMonitors on restart
  51         /// and add a pending payment that was already fulfilled.
  52         Fulfilled {
  53                 session_privs: HashSet<[u8; 32]>,
  54                 payment_hash: Option<PaymentHash>,
  55                 timer_ticks_without_htlcs: u8,
  56         },
  57         /// When a payer gives up trying to retry a payment, they inform us, letting us generate a
  58         /// `PaymentFailed` event when all HTLCs have irrevocably failed. This avoids a number of race
  59         /// conditions in MPP-aware payment retriers (1), where the possibility of multiple
  60         /// `PaymentPathFailed` events with `all_paths_failed` can be pending at once, confusing a
  61         /// downstream event handler as to when a payment has actually failed.
  62         ///
  63         /// (1) https://github.com/lightningdevkit/rust-lightning/issues/1164
  64         Abandoned {
  65                 session_privs: HashSet<[u8; 32]>,
  66                 payment_hash: PaymentHash,
  67         },
  68 }
  69
  70 impl PendingOutboundPayment {
  71         pub(super) fn is_fulfilled(&self) -> bool {
  72                 match self {
  73                         PendingOutboundPayment::Fulfilled { .. } => true,
  74                         _ => false,
  75                 }
  76         }
  77         pub(super) fn abandoned(&self) -> bool {
  78                 match self {
  79                         PendingOutboundPayment::Abandoned { .. } => true,
  80                         _ => false,
  81                 }
  82         }
  83         pub(super) fn get_pending_fee_msat(&self) -> Option<u64> {
  84                 match self {
  85                         PendingOutboundPayment::Retryable { pending_fee_msat, .. } => pending_fee_msat.clone(),
  86                         _ => None,
  87                 }
  88         }
  89
  90         pub(super) fn payment_hash(&self) -> Option<PaymentHash> {
  91                 match self {
  92                         PendingOutboundPayment::Legacy { .. } => None,
  93                         PendingOutboundPayment::Retryable { payment_hash, .. } => Some(*payment_hash),
  94                         PendingOutboundPayment::Fulfilled { payment_hash, .. } => *payment_hash,
  95                         PendingOutboundPayment::Abandoned { payment_hash, .. } => Some(*payment_hash),
  96                 }
  97         }
  98
  99         pub(super) fn mark_fulfilled(&mut self) {
 100                 let mut session_privs = HashSet::new();
 101                 core::mem::swap(&mut session_privs, match self {
 102                         PendingOutboundPayment::Legacy { session_privs } |
 103                                 PendingOutboundPayment::Retryable { session_privs, .. } |
 104                                 PendingOutboundPayment::Fulfilled { session_privs, .. } |
 105                                 PendingOutboundPayment::Abandoned { session_privs, .. }
 106                         => session_privs,
 107                 });
 108                 let payment_hash = self.payment_hash();
 109                 *self = PendingOutboundPayment::Fulfilled { session_privs, payment_hash, timer_ticks_without_htlcs: 0 };
 110         }
 111
 112         pub(super) fn mark_abandoned(&mut self) -> Result<(), ()> {
 113                 let mut session_privs = HashSet::new();
 114                 let our_payment_hash;
 115                 core::mem::swap(&mut session_privs, match self {
 116                         PendingOutboundPayment::Legacy { .. } |
 117                                 PendingOutboundPayment::Fulfilled { .. } =>
 118                                 return Err(()),
 119                                 PendingOutboundPayment::Retryable { session_privs, payment_hash, .. } |
 120                                         PendingOutboundPayment::Abandoned { session_privs, payment_hash, .. } => {
 121                                                 our_payment_hash = *payment_hash;
 122                                                 session_privs
 123                                         },
 124                 });
 125                 *self = PendingOutboundPayment::Abandoned { session_privs, payment_hash: our_payment_hash };
 126                 Ok(())
 127         }
 128
 129         /// panics if path is None and !self.is_fulfilled
 130         pub(super) fn remove(&mut self, session_priv: &[u8; 32], path: Option<&Vec<RouteHop>>) -> bool {
 131                 let remove_res = match self {
 132                         PendingOutboundPayment::Legacy { session_privs } |
 133                                 PendingOutboundPayment::Retryable { session_privs, .. } |
 134                                 PendingOutboundPayment::Fulfilled { session_privs, .. } |
 135                                 PendingOutboundPayment::Abandoned { session_privs, .. } => {
 136                                         session_privs.remove(session_priv)
 137                                 }
 138                 };
 139                 if remove_res {
 140                         if let PendingOutboundPayment::Retryable { ref mut pending_amt_msat, ref mut pending_fee_msat, .. } = self {
 141                                 let path = path.expect("Fulfilling a payment should always come with a path");
 142                                 let path_last_hop = path.last().expect("Outbound payments must have had a valid path");
 143                                 *pending_amt_msat -= path_last_hop.fee_msat;
 144                                 if let Some(fee_msat) = pending_fee_msat.as_mut() {
 145                                         *fee_msat -= path.get_path_fees();
 146                                 }
 147                         }
 148                 }
 149                 remove_res
 150         }
 151
 152         pub(super) fn insert(&mut self, session_priv: [u8; 32], path: &Vec<RouteHop>) -> bool {
 153                 let insert_res = match self {
 154                         PendingOutboundPayment::Legacy { session_privs } |
 155                                 PendingOutboundPayment::Retryable { session_privs, .. } => {
 156                                         session_privs.insert(session_priv)
 157                                 }
 158                         PendingOutboundPayment::Fulfilled { .. } => false,
 159                         PendingOutboundPayment::Abandoned { .. } => false,
 160                 };
 161                 if insert_res {
 162                         if let PendingOutboundPayment::Retryable { ref mut pending_amt_msat, ref mut pending_fee_msat, .. } = self {
 163                                 let path_last_hop = path.last().expect("Outbound payments must have had a valid path");
 164                                 *pending_amt_msat += path_last_hop.fee_msat;
 165                                 if let Some(fee_msat) = pending_fee_msat.as_mut() {
 166                                         *fee_msat += path.get_path_fees();
 167                                 }
 168                         }
 169                 }
 170                 insert_res
 171         }
 172
 173         pub(super) fn remaining_parts(&self) -> usize {
 174                 match self {
 175                         PendingOutboundPayment::Legacy { session_privs } |
 176                                 PendingOutboundPayment::Retryable { session_privs, .. } |
 177                                 PendingOutboundPayment::Fulfilled { session_privs, .. } |
 178                                 PendingOutboundPayment::Abandoned { session_privs, .. } => {
 179                                         session_privs.len()
 180                                 }
 181                 }
 182         }
 183 }
 184
 185 /// If a payment fails to send, it can be in one of several states. This enum is returned as the
 186 /// Err() type describing which state the payment is in, see the description of individual enum
 187 /// states for more.
 188 #[derive(Clone, Debug)]
 189 pub enum PaymentSendFailure {
 190         /// A parameter which was passed to send_payment was invalid, preventing us from attempting to
 191         /// send the payment at all.
 192         ///
 193         /// You can freely resend the payment in full (with the parameter error fixed).
 194         ///
 195         /// Because the payment failed outright, no payment tracking is done, you do not need to call
 196         /// [`ChannelManager::abandon_payment`] and [`ChannelManager::retry_payment`] will *not* work
 197         /// for this payment.
 198         ///
 199         /// [`ChannelManager::abandon_payment`]: crate::ln::channelmanager::ChannelManager::abandon_payment
 200         /// [`ChannelManager::retry_payment`]: crate::ln::channelmanager::ChannelManager::retry_payment
 201         ParameterError(APIError),
 202         /// A parameter in a single path which was passed to send_payment was invalid, preventing us
 203         /// from attempting to send the payment at all.
 204         ///
 205         /// You can freely resend the payment in full (with the parameter error fixed).
 206         ///
 207         /// The results here are ordered the same as the paths in the route object which was passed to
 208         /// send_payment.
 209         ///
 210         /// Because the payment failed outright, no payment tracking is done, you do not need to call
 211         /// [`ChannelManager::abandon_payment`] and [`ChannelManager::retry_payment`] will *not* work
 212         /// for this payment.
 213         ///
 214         /// [`ChannelManager::abandon_payment`]: crate::ln::channelmanager::ChannelManager::abandon_payment
 215         /// [`ChannelManager::retry_payment`]: crate::ln::channelmanager::ChannelManager::retry_payment
 216         PathParameterError(Vec<Result<(), APIError>>),
 217         /// All paths which were attempted failed to send, with no channel state change taking place.
 218         /// You can freely resend the payment in full (though you probably want to do so over different
 219         /// paths than the ones selected).
 220         ///
 221         /// Because the payment failed outright, no payment tracking is done, you do not need to call
 222         /// [`ChannelManager::abandon_payment`] and [`ChannelManager::retry_payment`] will *not* work
 223         /// for this payment.
 224         ///
 225         /// [`ChannelManager::abandon_payment`]: crate::ln::channelmanager::ChannelManager::abandon_payment
 226         /// [`ChannelManager::retry_payment`]: crate::ln::channelmanager::ChannelManager::retry_payment
 227         AllFailedResendSafe(Vec<APIError>),
 228         /// Indicates that a payment for the provided [`PaymentId`] is already in-flight and has not
 229         /// yet completed (i.e. generated an [`Event::PaymentSent`]) or been abandoned (via
 230         /// [`ChannelManager::abandon_payment`]).
 231         ///
 232         /// [`PaymentId`]: crate::ln::channelmanager::PaymentId
 233         /// [`Event::PaymentSent`]: crate::util::events::Event::PaymentSent
 234         /// [`ChannelManager::abandon_payment`]: crate::ln::channelmanager::ChannelManager::abandon_payment
 235         DuplicatePayment,
 236         /// Some paths which were attempted failed to send, though possibly not all. At least some
 237         /// paths have irrevocably committed to the HTLC and retrying the payment in full would result
 238         /// in over-/re-payment.
 239         ///
 240         /// The results here are ordered the same as the paths in the route object which was passed to
 241         /// send_payment, and any `Err`s which are not [`APIError::MonitorUpdateInProgress`] can be
 242         /// safely retried via [`ChannelManager::retry_payment`].
 243         ///
 244         /// Any entries which contain `Err(APIError::MonitorUpdateInprogress)` or `Ok(())` MUST NOT be
 245         /// retried as they will result in over-/re-payment. These HTLCs all either successfully sent
 246         /// (in the case of `Ok(())`) or will send once a [`MonitorEvent::Completed`] is provided for
 247         /// the next-hop channel with the latest update_id.
 248         ///
 249         /// [`ChannelManager::retry_payment`]: crate::ln::channelmanager::ChannelManager::retry_payment
 250         /// [`MonitorEvent::Completed`]: crate::chain::channelmonitor::MonitorEvent::Completed
 251         PartialFailure {
 252                 /// The errors themselves, in the same order as the route hops.
 253                 results: Vec<Result<(), APIError>>,
 254                 /// If some paths failed without irrevocably committing to the new HTLC(s), this will
 255                 /// contain a [`RouteParameters`] object which can be used to calculate a new route that
 256                 /// will pay all remaining unpaid balance.
 257                 failed_paths_retry: Option<RouteParameters>,
 258                 /// The payment id for the payment, which is now at least partially pending.
 259                 payment_id: PaymentId,
 260         },
 261 }
 262
 263 pub(super) struct OutboundPayments {
 264         pub(super) pending_outbound_payments: Mutex<HashMap<PaymentId, PendingOutboundPayment>>,
 265 }
 266
 267 impl OutboundPayments {
 268         pub(super) fn new() -> Self {
 269                 Self {
 270                         pending_outbound_payments: Mutex::new(HashMap::new())
 271                 }
 272         }
 273
 274         pub(super) fn send_payment<K: Deref, F>(
 275                 &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>,
 276                 payment_id: PaymentId, keys_manager: &K, best_block_height: u32, send_payment_along_path: F
 277         ) -> Result<(), PaymentSendFailure>
 278         where
 279                 K::Target: KeysInterface,
 280                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 281                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 282         {
 283                 let onion_session_privs = self.add_new_pending_payment(payment_hash, *payment_secret, payment_id, route, keys_manager, best_block_height)?;
 284                 self.send_payment_internal(route, payment_hash, payment_secret, None, payment_id, None, onion_session_privs, keys_manager, best_block_height, send_payment_along_path)
 285         }
 286
 287         pub(super) fn send_spontaneous_payment<K: Deref, F>(
 288                 &self, route: &Route, payment_preimage: Option<PaymentPreimage>, payment_id: PaymentId,
 289                 keys_manager: &K, best_block_height: u32, send_payment_along_path: F
 290         ) -> Result<PaymentHash, PaymentSendFailure>
 291         where
 292                 K::Target: KeysInterface,
 293                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 294                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 295         {
 296                 let preimage = match payment_preimage {
 297                         Some(p) => p,
 298                         None => PaymentPreimage(keys_manager.get_secure_random_bytes()),
 299                 };
 300                 let payment_hash = PaymentHash(Sha256::hash(&preimage.0).into_inner());
 301                 let onion_session_privs = self.add_new_pending_payment(payment_hash, None, payment_id, &route, keys_manager, best_block_height)?;
 302
 303                 match self.send_payment_internal(route, payment_hash, &None, Some(preimage), payment_id, None, onion_session_privs, keys_manager, best_block_height, send_payment_along_path) {
 304                         Ok(()) => Ok(payment_hash),
 305                         Err(e) => Err(e)
 306                 }
 307         }
 308
 309         pub(super) fn retry_payment<K: Deref, F>(
 310                 &self, route: &Route, payment_id: PaymentId, keys_manager: &K, best_block_height: u32,
 311                 send_payment_along_path: F
 312         ) -> Result<(), PaymentSendFailure>
 313         where
 314                 K::Target: KeysInterface,
 315                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 316                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 317         {
 318                 const RETRY_OVERFLOW_PERCENTAGE: u64 = 10;
 319                 for path in route.paths.iter() {
 320                         if path.len() == 0 {
 321                                 return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 322                                         err: "length-0 path in route".to_string()
 323                                 }))
 324                         }
 325                 }
 326
 327                 let mut onion_session_privs = Vec::with_capacity(route.paths.len());
 328                 for _ in 0..route.paths.len() {
 329                         onion_session_privs.push(keys_manager.get_secure_random_bytes());
 330                 }
 331
 332                 let (total_msat, payment_hash, payment_secret) = {
 333                         let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 334                         match outbounds.get_mut(&payment_id) {
 335                                 Some(payment) => {
 336                                         let res = match payment {
 337                                                 PendingOutboundPayment::Retryable {
 338                                                         total_msat, payment_hash, payment_secret, pending_amt_msat, ..
 339                                                 } => {
 340                                                         let retry_amt_msat: u64 = route.paths.iter().map(|path| path.last().unwrap().fee_msat).sum();
 341                                                         if retry_amt_msat + *pending_amt_msat > *total_msat * (100 + RETRY_OVERFLOW_PERCENTAGE) / 100 {
 342                                                                 return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 343                                                                         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()
 344                                                                 }))
 345                                                         }
 346                                                         (*total_msat, *payment_hash, *payment_secret)
 347                                                 },
 348                                                 PendingOutboundPayment::Legacy { .. } => {
 349                                                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 350                                                                 err: "Unable to retry payments that were initially sent on LDK versions prior to 0.0.102".to_string()
 351                                                         }))
 352                                                 },
 353                                                 PendingOutboundPayment::Fulfilled { .. } => {
 354                                                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 355                                                                 err: "Payment already completed".to_owned()
 356                                                         }));
 357                                                 },
 358                                                 PendingOutboundPayment::Abandoned { .. } => {
 359                                                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 360                                                                 err: "Payment already abandoned (with some HTLCs still pending)".to_owned()
 361                                                         }));
 362                                                 },
 363                                         };
 364                                         for (path, session_priv_bytes) in route.paths.iter().zip(onion_session_privs.iter()) {
 365                                                 assert!(payment.insert(*session_priv_bytes, path));
 366                                         }
 367                                         res
 368                                 },
 369                                 None =>
 370                                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 371                                                 err: format!("Payment with ID {} not found", log_bytes!(payment_id.0)),
 372                                         })),
 373                         }
 374                 };
 375                 self.send_payment_internal(route, payment_hash, &payment_secret, None, payment_id, Some(total_msat), onion_session_privs, keys_manager, best_block_height, send_payment_along_path)
 376         }
 377
 378         pub(super) fn send_probe<K: Deref, F>(
 379                 &self, hops: Vec<RouteHop>, probing_cookie_secret: [u8; 32], keys_manager: &K,
 380                 best_block_height: u32, send_payment_along_path: F
 381         ) -> Result<(PaymentHash, PaymentId), PaymentSendFailure>
 382         where
 383                 K::Target: KeysInterface,
 384                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 385                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 386         {
 387                 let payment_id = PaymentId(keys_manager.get_secure_random_bytes());
 388
 389                 let payment_hash = probing_cookie_from_id(&payment_id, probing_cookie_secret);
 390
 391                 if hops.len() < 2 {
 392                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError {
 393                                 err: "No need probing a path with less than two hops".to_string()
 394                         }))
 395                 }
 396
 397                 let route = Route { paths: vec![hops], payment_params: None };
 398                 let onion_session_privs = self.add_new_pending_payment(payment_hash, None, payment_id, &route, keys_manager, best_block_height)?;
 399
 400                 match self.send_payment_internal(&route, payment_hash, &None, None, payment_id, None, onion_session_privs, keys_manager, best_block_height, send_payment_along_path) {
 401                         Ok(()) => Ok((payment_hash, payment_id)),
 402                         Err(e) => Err(e)
 403                 }
 404         }
 405
 406         pub(super) fn add_new_pending_payment<K: Deref>(
 407                 &self, payment_hash: PaymentHash, payment_secret: Option<PaymentSecret>, payment_id: PaymentId,
 408                 route: &Route, keys_manager: &K, best_block_height: u32
 409         ) -> Result<Vec<[u8; 32]>, PaymentSendFailure> where K::Target: KeysInterface {
 410                 let mut onion_session_privs = Vec::with_capacity(route.paths.len());
 411                 for _ in 0..route.paths.len() {
 412                         onion_session_privs.push(keys_manager.get_secure_random_bytes());
 413                 }
 414
 415                 let mut pending_outbounds = self.pending_outbound_payments.lock().unwrap();
 416                 match pending_outbounds.entry(payment_id) {
 417                         hash_map::Entry::Occupied(_) => Err(PaymentSendFailure::DuplicatePayment),
 418                         hash_map::Entry::Vacant(entry) => {
 419                                 let payment = entry.insert(PendingOutboundPayment::Retryable {
 420                                         session_privs: HashSet::new(),
 421                                         pending_amt_msat: 0,
 422                                         pending_fee_msat: Some(0),
 423                                         payment_hash,
 424                                         payment_secret,
 425                                         starting_block_height: best_block_height,
 426                                         total_msat: route.get_total_amount(),
 427                                 });
 428
 429                                 for (path, session_priv_bytes) in route.paths.iter().zip(onion_session_privs.iter()) {
 430                                         assert!(payment.insert(*session_priv_bytes, path));
 431                                 }
 432
 433                                 Ok(onion_session_privs)
 434                         },
 435                 }
 436         }
 437
 438         fn send_payment_internal<K: Deref, F>(
 439                 &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>,
 440                 keysend_preimage: Option<PaymentPreimage>, payment_id: PaymentId, recv_value_msat: Option<u64>,
 441                 onion_session_privs: Vec<[u8; 32]>, keys_manager: &K, best_block_height: u32,
 442                 send_payment_along_path: F) -> Result<(), PaymentSendFailure>
 443         where
 444                 K::Target: KeysInterface,
 445                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 446                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 447         {
 448                 if route.paths.len() < 1 {
 449                         return Err(PaymentSendFailure::ParameterError(APIError::InvalidRoute{err: "There must be at least one path to send over"}));
 450                 }
 451                 if payment_secret.is_none() && route.paths.len() > 1 {
 452                         return Err(PaymentSendFailure::ParameterError(APIError::APIMisuseError{err: "Payment secret is required for multi-path payments".to_string()}));
 453                 }
 454                 let mut total_value = 0;
 455                 let our_node_id = keys_manager.get_node_id(Recipient::Node).unwrap(); // TODO no unwrap
 456                 let mut path_errs = Vec::with_capacity(route.paths.len());
 457                 'path_check: for path in route.paths.iter() {
 458                         if path.len() < 1 || path.len() > 20 {
 459                                 path_errs.push(Err(APIError::InvalidRoute{err: "Path didn't go anywhere/had bogus size"}));
 460                                 continue 'path_check;
 461                         }
 462                         for (idx, hop) in path.iter().enumerate() {
 463                                 if idx != path.len() - 1 && hop.pubkey == our_node_id {
 464                                         path_errs.push(Err(APIError::InvalidRoute{err: "Path went through us but wasn't a simple rebalance loop to us"}));
 465                                         continue 'path_check;
 466                                 }
 467                         }
 468                         total_value += path.last().unwrap().fee_msat;
 469                         path_errs.push(Ok(()));
 470                 }
 471                 if path_errs.iter().any(|e| e.is_err()) {
 472                         return Err(PaymentSendFailure::PathParameterError(path_errs));
 473                 }
 474                 if let Some(amt_msat) = recv_value_msat {
 475                         debug_assert!(amt_msat >= total_value);
 476                         total_value = amt_msat;
 477                 }
 478
 479                 let cur_height = best_block_height + 1;
 480                 let mut results = Vec::new();
 481                 debug_assert_eq!(route.paths.len(), onion_session_privs.len());
 482                 for (path, session_priv) in route.paths.iter().zip(onion_session_privs.into_iter()) {
 483                         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);
 484                         match path_res {
 485                                 Ok(_) => {},
 486                                 Err(APIError::MonitorUpdateInProgress) => {
 487                                         // While a MonitorUpdateInProgress is an Err(_), the payment is still
 488                                         // considered "in flight" and we shouldn't remove it from the
 489                                         // PendingOutboundPayment set.
 490                                 },
 491                                 Err(_) => {
 492                                         let mut pending_outbounds = self.pending_outbound_payments.lock().unwrap();
 493                                         if let Some(payment) = pending_outbounds.get_mut(&payment_id) {
 494                                                 let removed = payment.remove(&session_priv, Some(path));
 495                                                 debug_assert!(removed, "This can't happen as the payment has an entry for this path added by callers");
 496                                         } else {
 497                                                 debug_assert!(false, "This can't happen as the payment was added by callers");
 498                                                 path_res = Err(APIError::APIMisuseError { err: "Internal error: payment disappeared during processing. Please report this bug!".to_owned() });
 499                                         }
 500                                 }
 501                         }
 502                         results.push(path_res);
 503                 }
 504                 let mut has_ok = false;
 505                 let mut has_err = false;
 506                 let mut pending_amt_unsent = 0;
 507                 let mut max_unsent_cltv_delta = 0;
 508                 for (res, path) in results.iter().zip(route.paths.iter()) {
 509                         if res.is_ok() { has_ok = true; }
 510                         if res.is_err() { has_err = true; }
 511                         if let &Err(APIError::MonitorUpdateInProgress) = res {
 512                                 // MonitorUpdateInProgress is inherently unsafe to retry, so we call it a
 513                                 // PartialFailure.
 514                                 has_err = true;
 515                                 has_ok = true;
 516                         } else if res.is_err() {
 517                                 pending_amt_unsent += path.last().unwrap().fee_msat;
 518                                 max_unsent_cltv_delta = cmp::max(max_unsent_cltv_delta, path.last().unwrap().cltv_expiry_delta);
 519                         }
 520                 }
 521                 if has_err && has_ok {
 522                         Err(PaymentSendFailure::PartialFailure {
 523                                 results,
 524                                 payment_id,
 525                                 failed_paths_retry: if pending_amt_unsent != 0 {
 526                                         if let Some(payment_params) = &route.payment_params {
 527                                                 Some(RouteParameters {
 528                                                         payment_params: payment_params.clone(),
 529                                                         final_value_msat: pending_amt_unsent,
 530                                                         final_cltv_expiry_delta: max_unsent_cltv_delta,
 531                                                 })
 532                                         } else { None }
 533                                 } else { None },
 534                         })
 535                 } else if has_err {
 536                         // If we failed to send any paths, we should remove the new PaymentId from the
 537                         // `pending_outbound_payments` map, as the user isn't expected to `abandon_payment`.
 538                         let removed = self.pending_outbound_payments.lock().unwrap().remove(&payment_id).is_some();
 539                         debug_assert!(removed, "We should always have a pending payment to remove here");
 540                         Err(PaymentSendFailure::AllFailedResendSafe(results.drain(..).map(|r| r.unwrap_err()).collect()))
 541                 } else {
 542                         Ok(())
 543                 }
 544         }
 545
 546         #[cfg(test)]
 547         pub(super) fn test_send_payment_internal<K: Deref, F>(
 548                 &self, route: &Route, payment_hash: PaymentHash, payment_secret: &Option<PaymentSecret>,
 549                 keysend_preimage: Option<PaymentPreimage>, payment_id: PaymentId, recv_value_msat: Option<u64>,
 550                 onion_session_privs: Vec<[u8; 32]>, keys_manager: &K, best_block_height: u32,
 551                 send_payment_along_path: F) -> Result<(), PaymentSendFailure>
 552         where
 553                 K::Target: KeysInterface,
 554                 F: Fn(&Vec<RouteHop>, &Option<PaymentParameters>, &PaymentHash, &Option<PaymentSecret>, u64,
 555                    u32, PaymentId, &Option<PaymentPreimage>, [u8; 32]) -> Result<(), APIError>
 556         {
 557                 self.send_payment_internal(route, payment_hash, payment_secret, keysend_preimage, payment_id,
 558                         recv_value_msat, onion_session_privs, keys_manager, best_block_height,
 559                         send_payment_along_path)
 560         }
 561
 562         pub(super) fn claim_htlc<L: Deref>(
 563                 &self, payment_id: PaymentId, payment_preimage: PaymentPreimage, session_priv: SecretKey,
 564                 path: Vec<RouteHop>, from_onchain: bool, pending_events: &Mutex<Vec<events::Event>>, logger: &L)
 565          where L::Target: Logger {
 566                 let mut session_priv_bytes = [0; 32];
 567                 session_priv_bytes.copy_from_slice(&session_priv[..]);
 568                 let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 569                 let mut pending_events = pending_events.lock().unwrap();
 570                 if let hash_map::Entry::Occupied(mut payment) = outbounds.entry(payment_id) {
 571                         if !payment.get().is_fulfilled() {
 572                                 let payment_hash = PaymentHash(Sha256::hash(&payment_preimage.0).into_inner());
 573                                 let fee_paid_msat = payment.get().get_pending_fee_msat();
 574                                 pending_events.push(
 575                                         events::Event::PaymentSent {
 576                                                 payment_id: Some(payment_id),
 577                                                 payment_preimage,
 578                                                 payment_hash,
 579                                                 fee_paid_msat,
 580                                         }
 581                                 );
 582                                 payment.get_mut().mark_fulfilled();
 583                         }
 584
 585                         if from_onchain {
 586                                 // We currently immediately remove HTLCs which were fulfilled on-chain.
 587                                 // This could potentially lead to removing a pending payment too early,
 588                                 // with a reorg of one block causing us to re-add the fulfilled payment on
 589                                 // restart.
 590                                 // TODO: We should have a second monitor event that informs us of payments
 591                                 // irrevocably fulfilled.
 592                                 if payment.get_mut().remove(&session_priv_bytes, Some(&path)) {
 593                                         let payment_hash = Some(PaymentHash(Sha256::hash(&payment_preimage.0).into_inner()));
 594                                         pending_events.push(
 595                                                 events::Event::PaymentPathSuccessful {
 596                                                         payment_id,
 597                                                         payment_hash,
 598                                                         path,
 599                                                 }
 600                                         );
 601                                 }
 602                         }
 603                 } else {
 604                         log_trace!(logger, "Received duplicative fulfill for HTLC with payment_preimage {}", log_bytes!(payment_preimage.0));
 605                 }
 606         }
 607
 608         pub(super) fn finalize_claims(&self, sources: Vec<HTLCSource>, pending_events: &Mutex<Vec<events::Event>>) {
 609                 let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 610                 let mut pending_events = pending_events.lock().unwrap();
 611                 for source in sources {
 612                         if let HTLCSource::OutboundRoute { session_priv, payment_id, path, .. } = source {
 613                                 let mut session_priv_bytes = [0; 32];
 614                                 session_priv_bytes.copy_from_slice(&session_priv[..]);
 615                                 if let hash_map::Entry::Occupied(mut payment) = outbounds.entry(payment_id) {
 616                                         assert!(payment.get().is_fulfilled());
 617                                         if payment.get_mut().remove(&session_priv_bytes, None) {
 618                                                 pending_events.push(
 619                                                         events::Event::PaymentPathSuccessful {
 620                                                                 payment_id,
 621                                                                 payment_hash: payment.get().payment_hash(),
 622                                                                 path,
 623                                                         }
 624                                                 );
 625                                         }
 626                                 }
 627                         }
 628                 }
 629         }
 630
 631         pub(super) fn remove_stale_resolved_payments(&self, pending_events: &Mutex<Vec<events::Event>>) {
 632                 // If an outbound payment was completed, and no pending HTLCs remain, we should remove it
 633                 // from the map. However, if we did that immediately when the last payment HTLC is claimed,
 634                 // this could race the user making a duplicate send_payment call and our idempotency
 635                 // guarantees would be violated. Instead, we wait a few timer ticks to do the actual
 636                 // removal. This should be more than sufficient to ensure the idempotency of any
 637                 // `send_payment` calls that were made at the same time the `PaymentSent` event was being
 638                 // processed.
 639                 let mut pending_outbound_payments = self.pending_outbound_payments.lock().unwrap();
 640                 let pending_events = pending_events.lock().unwrap();
 641                 pending_outbound_payments.retain(|payment_id, payment| {
 642                         if let PendingOutboundPayment::Fulfilled { session_privs, timer_ticks_without_htlcs, .. } = payment {
 643                                 let mut no_remaining_entries = session_privs.is_empty();
 644                                 if no_remaining_entries {
 645                                         for ev in pending_events.iter() {
 646                                                 match ev {
 647                                                         events::Event::PaymentSent { payment_id: Some(ev_payment_id), .. } |
 648                                                                 events::Event::PaymentPathSuccessful { payment_id: ev_payment_id, .. } |
 649                                                                 events::Event::PaymentPathFailed { payment_id: Some(ev_payment_id), .. } => {
 650                                                                         if payment_id == ev_payment_id {
 651                                                                                 no_remaining_entries = false;
 652                                                                                 break;
 653                                                                         }
 654                                                                 },
 655                                                         _ => {},
 656                                                 }
 657                                         }
 658                                 }
 659                                 if no_remaining_entries {
 660                                         *timer_ticks_without_htlcs += 1;
 661                                         *timer_ticks_without_htlcs <= IDEMPOTENCY_TIMEOUT_TICKS
 662                                 } else {
 663                                         *timer_ticks_without_htlcs = 0;
 664                                         true
 665                                 }
 666                         } else { true }
 667                 });
 668         }
 669
 670         pub(super) fn fail_htlc<L: Deref>(
 671                 &self, source: &HTLCSource, payment_hash: &PaymentHash, onion_error: &HTLCFailReason,
 672                 path: &Vec<RouteHop>, session_priv: &SecretKey, payment_id: &PaymentId,
 673                 payment_params: &Option<PaymentParameters>, probing_cookie_secret: [u8; 32],
 674                 secp_ctx: &Secp256k1<secp256k1::All>, pending_events: &Mutex<Vec<events::Event>>, logger: &L)
 675         where L::Target: Logger {
 676                 let mut session_priv_bytes = [0; 32];
 677                 session_priv_bytes.copy_from_slice(&session_priv[..]);
 678                 let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 679                 let mut all_paths_failed = false;
 680                 let mut full_failure_ev = None;
 681                 if let hash_map::Entry::Occupied(mut payment) = outbounds.entry(*payment_id) {
 682                         if !payment.get_mut().remove(&session_priv_bytes, Some(&path)) {
 683                                 log_trace!(logger, "Received duplicative fail for HTLC with payment_hash {}", log_bytes!(payment_hash.0));
 684                                 return
 685                         }
 686                         if payment.get().is_fulfilled() {
 687                                 log_trace!(logger, "Received failure of HTLC with payment_hash {} after payment completion", log_bytes!(payment_hash.0));
 688                                 return
 689                         }
 690                         if payment.get().remaining_parts() == 0 {
 691                                 all_paths_failed = true;
 692                                 if payment.get().abandoned() {
 693                                         full_failure_ev = Some(events::Event::PaymentFailed {
 694                                                 payment_id: *payment_id,
 695                                                 payment_hash: payment.get().payment_hash().expect("PendingOutboundPayments::RetriesExceeded always has a payment hash set"),
 696                                         });
 697                                         payment.remove();
 698                                 }
 699                         }
 700                 } else {
 701                         log_trace!(logger, "Received duplicative fail for HTLC with payment_hash {}", log_bytes!(payment_hash.0));
 702                         return
 703                 }
 704                 let mut retry = if let Some(payment_params_data) = payment_params {
 705                         let path_last_hop = path.last().expect("Outbound payments must have had a valid path");
 706                         Some(RouteParameters {
 707                                 payment_params: payment_params_data.clone(),
 708                                 final_value_msat: path_last_hop.fee_msat,
 709                                 final_cltv_expiry_delta: path_last_hop.cltv_expiry_delta,
 710                         })
 711                 } else { None };
 712                 log_trace!(logger, "Failing outbound payment HTLC with payment_hash {}", log_bytes!(payment_hash.0));
 713
 714                 let path_failure = {
 715         #[cfg(test)]
 716                         let (network_update, short_channel_id, payment_retryable, onion_error_code, onion_error_data) = onion_error.decode_onion_failure(secp_ctx, logger, &source);
 717         #[cfg(not(test))]
 718                         let (network_update, short_channel_id, payment_retryable, _, _) = onion_error.decode_onion_failure(secp_ctx, logger, &source);
 719
 720                         if payment_is_probe(payment_hash, &payment_id, probing_cookie_secret) {
 721                                 if !payment_retryable {
 722                                         events::Event::ProbeSuccessful {
 723                                                 payment_id: *payment_id,
 724                                                 payment_hash: payment_hash.clone(),
 725                                                 path: path.clone(),
 726                                         }
 727                                 } else {
 728                                         events::Event::ProbeFailed {
 729                                                 payment_id: *payment_id,
 730                                                 payment_hash: payment_hash.clone(),
 731                                                 path: path.clone(),
 732                                                 short_channel_id,
 733                                         }
 734                                 }
 735                         } else {
 736                                 // TODO: If we decided to blame ourselves (or one of our channels) in
 737                                 // process_onion_failure we should close that channel as it implies our
 738                                 // next-hop is needlessly blaming us!
 739                                 if let Some(scid) = short_channel_id {
 740                                         retry.as_mut().map(|r| r.payment_params.previously_failed_channels.push(scid));
 741                                 }
 742                                 events::Event::PaymentPathFailed {
 743                                         payment_id: Some(*payment_id),
 744                                         payment_hash: payment_hash.clone(),
 745                                         payment_failed_permanently: !payment_retryable,
 746                                         network_update,
 747                                         all_paths_failed,
 748                                         path: path.clone(),
 749                                         short_channel_id,
 750                                         retry,
 751                                         #[cfg(test)]
 752                                         error_code: onion_error_code,
 753                                         #[cfg(test)]
 754                                         error_data: onion_error_data
 755                                 }
 756                         }
 757                 };
 758                 let mut pending_events = pending_events.lock().unwrap();
 759                 pending_events.push(path_failure);
 760                 if let Some(ev) = full_failure_ev { pending_events.push(ev); }
 761         }
 762
 763         pub(super) fn abandon_payment(&self, payment_id: PaymentId) -> Option<events::Event> {
 764                 let mut failed_ev = None;
 765                 let mut outbounds = self.pending_outbound_payments.lock().unwrap();
 766                 if let hash_map::Entry::Occupied(mut payment) = outbounds.entry(payment_id) {
 767                         if let Ok(()) = payment.get_mut().mark_abandoned() {
 768                                 if payment.get().remaining_parts() == 0 {
 769                                         failed_ev = Some(events::Event::PaymentFailed {
 770                                                 payment_id,
 771                                                 payment_hash: payment.get().payment_hash().expect("PendingOutboundPayments::RetriesExceeded always has a payment hash set"),
 772                                         });
 773                                         payment.remove();
 774                                 }
 775                         }
 776                 }
 777                 failed_ev
 778         }
 779
 780         #[cfg(test)]
 781         pub fn has_pending_payments(&self) -> bool {
 782                 !self.pending_outbound_payments.lock().unwrap().is_empty()
 783         }
 784
 785         #[cfg(test)]
 786         pub fn clear_pending_payments(&self) {
 787                 self.pending_outbound_payments.lock().unwrap().clear()
 788         }
 789 }
 790
 791 /// Returns whether a payment with the given [`PaymentHash`] and [`PaymentId`] is, in fact, a
 792 /// payment probe.
 793 pub(super) fn payment_is_probe(payment_hash: &PaymentHash, payment_id: &PaymentId,
 794         probing_cookie_secret: [u8; 32]) -> bool
 795 {
 796         let target_payment_hash = probing_cookie_from_id(payment_id, probing_cookie_secret);
 797         target_payment_hash == *payment_hash
 798 }
 799
 800 /// Returns the 'probing cookie' for the given [`PaymentId`].
 801 fn probing_cookie_from_id(payment_id: &PaymentId, probing_cookie_secret: [u8; 32]) -> PaymentHash {
 802         let mut preimage = [0u8; 64];
 803         preimage[..32].copy_from_slice(&probing_cookie_secret);
 804         preimage[32..].copy_from_slice(&payment_id.0);
 805         PaymentHash(Sha256::hash(&preimage).into_inner())
 806 }
 807
 808 impl_writeable_tlv_based_enum_upgradable!(PendingOutboundPayment,
 809         (0, Legacy) => {
 810                 (0, session_privs, required),
 811         },
 812         (1, Fulfilled) => {
 813                 (0, session_privs, required),
 814                 (1, payment_hash, option),
 815                 (3, timer_ticks_without_htlcs, (default_value, 0)),
 816         },
 817         (2, Retryable) => {
 818                 (0, session_privs, required),
 819                 (1, pending_fee_msat, option),
 820                 (2, payment_hash, required),
 821                 (4, payment_secret, option),
 822                 (6, total_msat, required),
 823                 (8, pending_amt_msat, required),
 824                 (10, starting_block_height, required),
 825         },
 826         (3, Abandoned) => {
 827                 (0, session_privs, required),
 828                 (2, payment_hash, required),
 829         },
 830 );