Refactor OutboundPayments::retry_payment_internal

Consolidate the creation and insertion of onion_session_privs to the
PendingOutboundPayment::Retryable arm. In an upcoming commit, this
method will be reused for an initial BOLT 12 invoice payment. However,
onion_session_privs are created using a different helper.
This commit is contained in:
Jeffrey Czyz 2023-09-01 16:23:27 -05:00
parent b5169301d8
commit 283d9b4e03
No known key found for this signature in database
GPG key ID: 3A4E08275D5E96D2

View file

@ -839,12 +839,6 @@ impl OutboundPayments {
} }
} }
const RETRY_OVERFLOW_PERCENTAGE: u64 = 10;
let mut onion_session_privs = Vec::with_capacity(route.paths.len());
for _ in 0..route.paths.len() {
onion_session_privs.push(entropy_source.get_secure_random_bytes());
}
macro_rules! abandon_with_entry { macro_rules! abandon_with_entry {
($payment: expr, $reason: expr) => { ($payment: expr, $reason: expr) => {
$payment.get_mut().mark_abandoned($reason); $payment.get_mut().mark_abandoned($reason);
@ -860,26 +854,49 @@ impl OutboundPayments {
} }
} }
} }
let (total_msat, recipient_onion, keysend_preimage) = { let (total_msat, recipient_onion, keysend_preimage, onion_session_privs) = {
let mut outbounds = self.pending_outbound_payments.lock().unwrap(); let mut outbounds = self.pending_outbound_payments.lock().unwrap();
match outbounds.entry(payment_id) { match outbounds.entry(payment_id) {
hash_map::Entry::Occupied(mut payment) => { hash_map::Entry::Occupied(mut payment) => {
let res = match payment.get() { match payment.get() {
PendingOutboundPayment::Retryable { PendingOutboundPayment::Retryable {
total_msat, keysend_preimage, payment_secret, payment_metadata, total_msat, keysend_preimage, payment_secret, payment_metadata,
custom_tlvs, pending_amt_msat, .. custom_tlvs, pending_amt_msat, ..
} => { } => {
const RETRY_OVERFLOW_PERCENTAGE: u64 = 10;
let retry_amt_msat = route.get_total_amount(); let retry_amt_msat = route.get_total_amount();
if retry_amt_msat + *pending_amt_msat > *total_msat * (100 + RETRY_OVERFLOW_PERCENTAGE) / 100 { if retry_amt_msat + *pending_amt_msat > *total_msat * (100 + RETRY_OVERFLOW_PERCENTAGE) / 100 {
log_error!(logger, "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); log_error!(logger, "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);
abandon_with_entry!(payment, PaymentFailureReason::UnexpectedError); abandon_with_entry!(payment, PaymentFailureReason::UnexpectedError);
return return
} }
(*total_msat, RecipientOnionFields {
if !payment.get().is_retryable_now() {
log_error!(logger, "Retries exhausted for payment id {}", &payment_id);
abandon_with_entry!(payment, PaymentFailureReason::RetriesExhausted);
return
}
let total_msat = *total_msat;
let recipient_onion = RecipientOnionFields {
payment_secret: *payment_secret, payment_secret: *payment_secret,
payment_metadata: payment_metadata.clone(), payment_metadata: payment_metadata.clone(),
custom_tlvs: custom_tlvs.clone(), custom_tlvs: custom_tlvs.clone(),
}, *keysend_preimage) };
let keysend_preimage = *keysend_preimage;
let mut onion_session_privs = Vec::with_capacity(route.paths.len());
for _ in 0..route.paths.len() {
onion_session_privs.push(entropy_source.get_secure_random_bytes());
}
for (path, session_priv_bytes) in route.paths.iter().zip(onion_session_privs.iter()) {
assert!(payment.get_mut().insert(*session_priv_bytes, path));
}
payment.get_mut().increment_attempts();
(total_msat, recipient_onion, keysend_preimage, onion_session_privs)
}, },
PendingOutboundPayment::Legacy { .. } => { PendingOutboundPayment::Legacy { .. } => {
log_error!(logger, "Unable to retry payments that were initially sent on LDK versions prior to 0.0.102"); log_error!(logger, "Unable to retry payments that were initially sent on LDK versions prior to 0.0.102");
@ -899,17 +916,7 @@ impl OutboundPayments {
log_error!(logger, "Payment already abandoned (with some HTLCs still pending)"); log_error!(logger, "Payment already abandoned (with some HTLCs still pending)");
return return
}, },
};
if !payment.get().is_retryable_now() {
log_error!(logger, "Retries exhausted for payment id {}", &payment_id);
abandon_with_entry!(payment, PaymentFailureReason::RetriesExhausted);
return
} }
payment.get_mut().increment_attempts();
for (path, session_priv_bytes) in route.paths.iter().zip(onion_session_privs.iter()) {
assert!(payment.get_mut().insert(*session_priv_bytes, path));
}
res
}, },
hash_map::Entry::Vacant(_) => { hash_map::Entry::Vacant(_) => {
log_error!(logger, "Payment with ID {} not found", &payment_id); log_error!(logger, "Payment with ID {} not found", &payment_id);