Add PendingOutboundPayment::InvoiceReceived
[rust-lightning] / lightning / src / ln / channelmanager.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 //! The top-level channel management and payment tracking stuff lives here.
11 //!
12 //! The [`ChannelManager`] is the main chunk of logic implementing the lightning protocol and is
13 //! responsible for tracking which channels are open, HTLCs are in flight and reestablishing those
14 //! upon reconnect to the relevant peer(s).
15 //!
16 //! It does not manage routing logic (see [`Router`] for that) nor does it manage constructing
17 //! on-chain transactions (it only monitors the chain to watch for any force-closes that might
18 //! imply it needs to fail HTLCs/payments/channels it manages).
19
20 use bitcoin::blockdata::block::BlockHeader;
21 use bitcoin::blockdata::transaction::Transaction;
22 use bitcoin::blockdata::constants::{genesis_block, ChainHash};
23 use bitcoin::network::constants::Network;
24
25 use bitcoin::hashes::Hash;
26 use bitcoin::hashes::sha256::Hash as Sha256;
27 use bitcoin::hash_types::{BlockHash, Txid};
28
29 use bitcoin::secp256k1::{SecretKey,PublicKey};
30 use bitcoin::secp256k1::Secp256k1;
31 use bitcoin::{LockTime, secp256k1, Sequence};
32
33 use crate::chain;
34 use crate::chain::{Confirm, ChannelMonitorUpdateStatus, Watch, BestBlock};
35 use crate::chain::chaininterface::{BroadcasterInterface, ConfirmationTarget, FeeEstimator, LowerBoundedFeeEstimator};
36 use crate::chain::channelmonitor::{ChannelMonitor, ChannelMonitorUpdate, ChannelMonitorUpdateStep, HTLC_FAIL_BACK_BUFFER, CLTV_CLAIM_BUFFER, LATENCY_GRACE_PERIOD_BLOCKS, ANTI_REORG_DELAY, MonitorEvent, CLOSED_CHANNEL_UPDATE_ID};
37 use crate::chain::transaction::{OutPoint, TransactionData};
38 use crate::events;
39 use crate::events::{Event, EventHandler, EventsProvider, MessageSendEvent, MessageSendEventsProvider, ClosureReason, HTLCDestination, PaymentFailureReason};
40 // Since this struct is returned in `list_channels` methods, expose it here in case users want to
41 // construct one themselves.
42 use crate::ln::{inbound_payment, ChannelId, PaymentHash, PaymentPreimage, PaymentSecret};
43 use crate::ln::channel::{Channel, ChannelContext, ChannelError, ChannelUpdateStatus, ShutdownResult, UnfundedChannelContext, UpdateFulfillCommitFetch, OutboundV1Channel, InboundV1Channel};
44 use crate::ln::features::{ChannelFeatures, ChannelTypeFeatures, InitFeatures, NodeFeatures};
45 #[cfg(any(feature = "_test_utils", test))]
46 use crate::ln::features::Bolt11InvoiceFeatures;
47 use crate::routing::gossip::NetworkGraph;
48 use crate::routing::router::{BlindedTail, DefaultRouter, InFlightHtlcs, Path, Payee, PaymentParameters, Route, RouteParameters, Router};
49 use crate::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringFeeParameters};
50 use crate::ln::msgs;
51 use crate::ln::onion_utils;
52 use crate::ln::onion_utils::HTLCFailReason;
53 use crate::ln::msgs::{ChannelMessageHandler, DecodeError, LightningError};
54 #[cfg(test)]
55 use crate::ln::outbound_payment;
56 use crate::ln::outbound_payment::{OutboundPayments, PaymentAttempts, PendingOutboundPayment, SendAlongPathArgs};
57 use crate::ln::wire::Encode;
58 use crate::sign::{EntropySource, KeysManager, NodeSigner, Recipient, SignerProvider, WriteableEcdsaChannelSigner};
59 use crate::util::config::{UserConfig, ChannelConfig, ChannelConfigUpdate};
60 use crate::util::wakers::{Future, Notifier};
61 use crate::util::scid_utils::fake_scid;
62 use crate::util::string::UntrustedString;
63 use crate::util::ser::{BigSize, FixedLengthReader, Readable, ReadableArgs, MaybeReadable, Writeable, Writer, VecWriter};
64 use crate::util::logger::{Level, Logger};
65 use crate::util::errors::APIError;
66
67 use alloc::collections::BTreeMap;
68
69 use crate::io;
70 use crate::prelude::*;
71 use core::{cmp, mem};
72 use core::cell::RefCell;
73 use crate::io::Read;
74 use crate::sync::{Arc, Mutex, RwLock, RwLockReadGuard, FairRwLock, LockTestExt, LockHeldState};
75 use core::sync::atomic::{AtomicUsize, AtomicBool, Ordering};
76 use core::time::Duration;
77 use core::ops::Deref;
78
79 // Re-export this for use in the public API.
80 pub use crate::ln::outbound_payment::{PaymentSendFailure, Retry, RetryableSendFailure, RecipientOnionFields};
81 use crate::ln::script::ShutdownScript;
82
83 // We hold various information about HTLC relay in the HTLC objects in Channel itself:
84 //
85 // Upon receipt of an HTLC from a peer, we'll give it a PendingHTLCStatus indicating if it should
86 // forward the HTLC with information it will give back to us when it does so, or if it should Fail
87 // the HTLC with the relevant message for the Channel to handle giving to the remote peer.
88 //
89 // Once said HTLC is committed in the Channel, if the PendingHTLCStatus indicated Forward, the
90 // Channel will return the PendingHTLCInfo back to us, and we will create an HTLCForwardInfo
91 // with it to track where it came from (in case of onwards-forward error), waiting a random delay
92 // before we forward it.
93 //
94 // We will then use HTLCForwardInfo's PendingHTLCInfo to construct an outbound HTLC, with a
95 // relevant HTLCSource::PreviousHopData filled in to indicate where it came from (which we can use
96 // to either fail-backwards or fulfill the HTLC backwards along the relevant path).
97 // Alternatively, we can fill an outbound HTLC with a HTLCSource::OutboundRoute indicating this is
98 // our payment, which we can use to decode errors or inform the user that the payment was sent.
99
100 #[derive(Clone)] // See Channel::revoke_and_ack for why, tl;dr: Rust bug
101 pub(super) enum PendingHTLCRouting {
102         Forward {
103                 onion_packet: msgs::OnionPacket,
104                 /// The SCID from the onion that we should forward to. This could be a real SCID or a fake one
105                 /// generated using `get_fake_scid` from the scid_utils::fake_scid module.
106                 short_channel_id: u64, // This should be NonZero<u64> eventually when we bump MSRV
107         },
108         Receive {
109                 payment_data: msgs::FinalOnionHopData,
110                 payment_metadata: Option<Vec<u8>>,
111                 incoming_cltv_expiry: u32, // Used to track when we should expire pending HTLCs that go unclaimed
112                 phantom_shared_secret: Option<[u8; 32]>,
113                 /// See [`RecipientOnionFields::custom_tlvs`] for more info.
114                 custom_tlvs: Vec<(u64, Vec<u8>)>,
115         },
116         ReceiveKeysend {
117                 /// This was added in 0.0.116 and will break deserialization on downgrades.
118                 payment_data: Option<msgs::FinalOnionHopData>,
119                 payment_preimage: PaymentPreimage,
120                 payment_metadata: Option<Vec<u8>>,
121                 incoming_cltv_expiry: u32, // Used to track when we should expire pending HTLCs that go unclaimed
122                 /// See [`RecipientOnionFields::custom_tlvs`] for more info.
123                 custom_tlvs: Vec<(u64, Vec<u8>)>,
124         },
125 }
126
127 #[derive(Clone)] // See Channel::revoke_and_ack for why, tl;dr: Rust bug
128 pub(super) struct PendingHTLCInfo {
129         pub(super) routing: PendingHTLCRouting,
130         pub(super) incoming_shared_secret: [u8; 32],
131         payment_hash: PaymentHash,
132         /// Amount received
133         pub(super) incoming_amt_msat: Option<u64>, // Added in 0.0.113
134         /// Sender intended amount to forward or receive (actual amount received
135         /// may overshoot this in either case)
136         pub(super) outgoing_amt_msat: u64,
137         pub(super) outgoing_cltv_value: u32,
138         /// The fee being skimmed off the top of this HTLC. If this is a forward, it'll be the fee we are
139         /// skimming. If we're receiving this HTLC, it's the fee that our counterparty skimmed.
140         pub(super) skimmed_fee_msat: Option<u64>,
141 }
142
143 #[derive(Clone)] // See Channel::revoke_and_ack for why, tl;dr: Rust bug
144 pub(super) enum HTLCFailureMsg {
145         Relay(msgs::UpdateFailHTLC),
146         Malformed(msgs::UpdateFailMalformedHTLC),
147 }
148
149 /// Stores whether we can't forward an HTLC or relevant forwarding info
150 #[derive(Clone)] // See Channel::revoke_and_ack for why, tl;dr: Rust bug
151 pub(super) enum PendingHTLCStatus {
152         Forward(PendingHTLCInfo),
153         Fail(HTLCFailureMsg),
154 }
155
156 pub(super) struct PendingAddHTLCInfo {
157         pub(super) forward_info: PendingHTLCInfo,
158
159         // These fields are produced in `forward_htlcs()` and consumed in
160         // `process_pending_htlc_forwards()` for constructing the
161         // `HTLCSource::PreviousHopData` for failed and forwarded
162         // HTLCs.
163         //
164         // Note that this may be an outbound SCID alias for the associated channel.
165         prev_short_channel_id: u64,
166         prev_htlc_id: u64,
167         prev_funding_outpoint: OutPoint,
168         prev_user_channel_id: u128,
169 }
170
171 pub(super) enum HTLCForwardInfo {
172         AddHTLC(PendingAddHTLCInfo),
173         FailHTLC {
174                 htlc_id: u64,
175                 err_packet: msgs::OnionErrorPacket,
176         },
177 }
178
179 /// Tracks the inbound corresponding to an outbound HTLC
180 #[derive(Clone, Hash, PartialEq, Eq)]
181 pub(crate) struct HTLCPreviousHopData {
182         // Note that this may be an outbound SCID alias for the associated channel.
183         short_channel_id: u64,
184         user_channel_id: Option<u128>,
185         htlc_id: u64,
186         incoming_packet_shared_secret: [u8; 32],
187         phantom_shared_secret: Option<[u8; 32]>,
188
189         // This field is consumed by `claim_funds_from_hop()` when updating a force-closed backwards
190         // channel with a preimage provided by the forward channel.
191         outpoint: OutPoint,
192 }
193
194 enum OnionPayload {
195         /// Indicates this incoming onion payload is for the purpose of paying an invoice.
196         Invoice {
197                 /// This is only here for backwards-compatibility in serialization, in the future it can be
198                 /// removed, breaking clients running 0.0.106 and earlier.
199                 _legacy_hop_data: Option<msgs::FinalOnionHopData>,
200         },
201         /// Contains the payer-provided preimage.
202         Spontaneous(PaymentPreimage),
203 }
204
205 /// HTLCs that are to us and can be failed/claimed by the user
206 struct ClaimableHTLC {
207         prev_hop: HTLCPreviousHopData,
208         cltv_expiry: u32,
209         /// The amount (in msats) of this MPP part
210         value: u64,
211         /// The amount (in msats) that the sender intended to be sent in this MPP
212         /// part (used for validating total MPP amount)
213         sender_intended_value: u64,
214         onion_payload: OnionPayload,
215         timer_ticks: u8,
216         /// The total value received for a payment (sum of all MPP parts if the payment is a MPP).
217         /// Gets set to the amount reported when pushing [`Event::PaymentClaimable`].
218         total_value_received: Option<u64>,
219         /// The sender intended sum total of all MPP parts specified in the onion
220         total_msat: u64,
221         /// The extra fee our counterparty skimmed off the top of this HTLC.
222         counterparty_skimmed_fee_msat: Option<u64>,
223 }
224
225 impl From<&ClaimableHTLC> for events::ClaimedHTLC {
226         fn from(val: &ClaimableHTLC) -> Self {
227                 events::ClaimedHTLC {
228                         channel_id: val.prev_hop.outpoint.to_channel_id(),
229                         user_channel_id: val.prev_hop.user_channel_id.unwrap_or(0),
230                         cltv_expiry: val.cltv_expiry,
231                         value_msat: val.value,
232                 }
233         }
234 }
235
236 /// A payment identifier used to uniquely identify a payment to LDK.
237 ///
238 /// This is not exported to bindings users as we just use [u8; 32] directly
239 #[derive(Hash, Copy, Clone, PartialEq, Eq, Debug)]
240 pub struct PaymentId(pub [u8; Self::LENGTH]);
241
242 impl PaymentId {
243         /// Number of bytes in the id.
244         pub const LENGTH: usize = 32;
245 }
246
247 impl Writeable for PaymentId {
248         fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
249                 self.0.write(w)
250         }
251 }
252
253 impl Readable for PaymentId {
254         fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
255                 let buf: [u8; 32] = Readable::read(r)?;
256                 Ok(PaymentId(buf))
257         }
258 }
259
260 impl core::fmt::Display for PaymentId {
261         fn fmt(&self, f: &mut core::fmt::Formatter) -> core::fmt::Result {
262                 crate::util::logger::DebugBytes(&self.0).fmt(f)
263         }
264 }
265
266 /// An identifier used to uniquely identify an intercepted HTLC to LDK.
267 ///
268 /// This is not exported to bindings users as we just use [u8; 32] directly
269 #[derive(Hash, Copy, Clone, PartialEq, Eq, Debug)]
270 pub struct InterceptId(pub [u8; 32]);
271
272 impl Writeable for InterceptId {
273         fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
274                 self.0.write(w)
275         }
276 }
277
278 impl Readable for InterceptId {
279         fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
280                 let buf: [u8; 32] = Readable::read(r)?;
281                 Ok(InterceptId(buf))
282         }
283 }
284
285 #[derive(Clone, Copy, PartialEq, Eq, Hash)]
286 /// Uniquely describes an HTLC by its source. Just the guaranteed-unique subset of [`HTLCSource`].
287 pub(crate) enum SentHTLCId {
288         PreviousHopData { short_channel_id: u64, htlc_id: u64 },
289         OutboundRoute { session_priv: SecretKey },
290 }
291 impl SentHTLCId {
292         pub(crate) fn from_source(source: &HTLCSource) -> Self {
293                 match source {
294                         HTLCSource::PreviousHopData(hop_data) => Self::PreviousHopData {
295                                 short_channel_id: hop_data.short_channel_id,
296                                 htlc_id: hop_data.htlc_id,
297                         },
298                         HTLCSource::OutboundRoute { session_priv, .. } =>
299                                 Self::OutboundRoute { session_priv: *session_priv },
300                 }
301         }
302 }
303 impl_writeable_tlv_based_enum!(SentHTLCId,
304         (0, PreviousHopData) => {
305                 (0, short_channel_id, required),
306                 (2, htlc_id, required),
307         },
308         (2, OutboundRoute) => {
309                 (0, session_priv, required),
310         };
311 );
312
313
314 /// Tracks the inbound corresponding to an outbound HTLC
315 #[allow(clippy::derive_hash_xor_eq)] // Our Hash is faithful to the data, we just don't have SecretKey::hash
316 #[derive(Clone, PartialEq, Eq)]
317 pub(crate) enum HTLCSource {
318         PreviousHopData(HTLCPreviousHopData),
319         OutboundRoute {
320                 path: Path,
321                 session_priv: SecretKey,
322                 /// Technically we can recalculate this from the route, but we cache it here to avoid
323                 /// doing a double-pass on route when we get a failure back
324                 first_hop_htlc_msat: u64,
325                 payment_id: PaymentId,
326         },
327 }
328 #[allow(clippy::derive_hash_xor_eq)] // Our Hash is faithful to the data, we just don't have SecretKey::hash
329 impl core::hash::Hash for HTLCSource {
330         fn hash<H: core::hash::Hasher>(&self, hasher: &mut H) {
331                 match self {
332                         HTLCSource::PreviousHopData(prev_hop_data) => {
333                                 0u8.hash(hasher);
334                                 prev_hop_data.hash(hasher);
335                         },
336                         HTLCSource::OutboundRoute { path, session_priv, payment_id, first_hop_htlc_msat } => {
337                                 1u8.hash(hasher);
338                                 path.hash(hasher);
339                                 session_priv[..].hash(hasher);
340                                 payment_id.hash(hasher);
341                                 first_hop_htlc_msat.hash(hasher);
342                         },
343                 }
344         }
345 }
346 impl HTLCSource {
347         #[cfg(all(feature = "_test_vectors", not(feature = "grind_signatures")))]
348         #[cfg(test)]
349         pub fn dummy() -> Self {
350                 HTLCSource::OutboundRoute {
351                         path: Path { hops: Vec::new(), blinded_tail: None },
352                         session_priv: SecretKey::from_slice(&[1; 32]).unwrap(),
353                         first_hop_htlc_msat: 0,
354                         payment_id: PaymentId([2; 32]),
355                 }
356         }
357
358         #[cfg(debug_assertions)]
359         /// Checks whether this HTLCSource could possibly match the given HTLC output in a commitment
360         /// transaction. Useful to ensure different datastructures match up.
361         pub(crate) fn possibly_matches_output(&self, htlc: &super::chan_utils::HTLCOutputInCommitment) -> bool {
362                 if let HTLCSource::OutboundRoute { first_hop_htlc_msat, .. } = self {
363                         *first_hop_htlc_msat == htlc.amount_msat
364                 } else {
365                         // There's nothing we can check for forwarded HTLCs
366                         true
367                 }
368         }
369 }
370
371 struct InboundOnionErr {
372         err_code: u16,
373         err_data: Vec<u8>,
374         msg: &'static str,
375 }
376
377 /// This enum is used to specify which error data to send to peers when failing back an HTLC
378 /// using [`ChannelManager::fail_htlc_backwards_with_reason`].
379 ///
380 /// For more info on failure codes, see <https://github.com/lightning/bolts/blob/master/04-onion-routing.md#failure-messages>.
381 #[derive(Clone, Copy)]
382 pub enum FailureCode {
383         /// We had a temporary error processing the payment. Useful if no other error codes fit
384         /// and you want to indicate that the payer may want to retry.
385         TemporaryNodeFailure,
386         /// We have a required feature which was not in this onion. For example, you may require
387         /// some additional metadata that was not provided with this payment.
388         RequiredNodeFeatureMissing,
389         /// You may wish to use this when a `payment_preimage` is unknown, or the CLTV expiry of
390         /// the HTLC is too close to the current block height for safe handling.
391         /// Using this failure code in [`ChannelManager::fail_htlc_backwards_with_reason`] is
392         /// equivalent to calling [`ChannelManager::fail_htlc_backwards`].
393         IncorrectOrUnknownPaymentDetails,
394         /// We failed to process the payload after the onion was decrypted. You may wish to
395         /// use this when receiving custom HTLC TLVs with even type numbers that you don't recognize.
396         ///
397         /// If available, the tuple data may include the type number and byte offset in the
398         /// decrypted byte stream where the failure occurred.
399         InvalidOnionPayload(Option<(u64, u16)>),
400 }
401
402 impl Into<u16> for FailureCode {
403     fn into(self) -> u16 {
404                 match self {
405                         FailureCode::TemporaryNodeFailure => 0x2000 | 2,
406                         FailureCode::RequiredNodeFeatureMissing => 0x4000 | 0x2000 | 3,
407                         FailureCode::IncorrectOrUnknownPaymentDetails => 0x4000 | 15,
408                         FailureCode::InvalidOnionPayload(_) => 0x4000 | 22,
409                 }
410         }
411 }
412
413 /// Error type returned across the peer_state mutex boundary. When an Err is generated for a
414 /// Channel, we generally end up with a ChannelError::Close for which we have to close the channel
415 /// immediately (ie with no further calls on it made). Thus, this step happens inside a
416 /// peer_state lock. We then return the set of things that need to be done outside the lock in
417 /// this struct and call handle_error!() on it.
418
419 struct MsgHandleErrInternal {
420         err: msgs::LightningError,
421         chan_id: Option<(ChannelId, u128)>, // If Some a channel of ours has been closed
422         shutdown_finish: Option<(ShutdownResult, Option<msgs::ChannelUpdate>)>,
423         channel_capacity: Option<u64>,
424 }
425 impl MsgHandleErrInternal {
426         #[inline]
427         fn send_err_msg_no_close(err: String, channel_id: ChannelId) -> Self {
428                 Self {
429                         err: LightningError {
430                                 err: err.clone(),
431                                 action: msgs::ErrorAction::SendErrorMessage {
432                                         msg: msgs::ErrorMessage {
433                                                 channel_id,
434                                                 data: err
435                                         },
436                                 },
437                         },
438                         chan_id: None,
439                         shutdown_finish: None,
440                         channel_capacity: None,
441                 }
442         }
443         #[inline]
444         fn from_no_close(err: msgs::LightningError) -> Self {
445                 Self { err, chan_id: None, shutdown_finish: None, channel_capacity: None }
446         }
447         #[inline]
448         fn from_finish_shutdown(err: String, channel_id: ChannelId, user_channel_id: u128, shutdown_res: ShutdownResult, channel_update: Option<msgs::ChannelUpdate>, channel_capacity: u64) -> Self {
449                 Self {
450                         err: LightningError {
451                                 err: err.clone(),
452                                 action: msgs::ErrorAction::SendErrorMessage {
453                                         msg: msgs::ErrorMessage {
454                                                 channel_id,
455                                                 data: err
456                                         },
457                                 },
458                         },
459                         chan_id: Some((channel_id, user_channel_id)),
460                         shutdown_finish: Some((shutdown_res, channel_update)),
461                         channel_capacity: Some(channel_capacity)
462                 }
463         }
464         #[inline]
465         fn from_chan_no_close(err: ChannelError, channel_id: ChannelId) -> Self {
466                 Self {
467                         err: match err {
468                                 ChannelError::Warn(msg) =>  LightningError {
469                                         err: msg.clone(),
470                                         action: msgs::ErrorAction::SendWarningMessage {
471                                                 msg: msgs::WarningMessage {
472                                                         channel_id,
473                                                         data: msg
474                                                 },
475                                                 log_level: Level::Warn,
476                                         },
477                                 },
478                                 ChannelError::Ignore(msg) => LightningError {
479                                         err: msg,
480                                         action: msgs::ErrorAction::IgnoreError,
481                                 },
482                                 ChannelError::Close(msg) => LightningError {
483                                         err: msg.clone(),
484                                         action: msgs::ErrorAction::SendErrorMessage {
485                                                 msg: msgs::ErrorMessage {
486                                                         channel_id,
487                                                         data: msg
488                                                 },
489                                         },
490                                 },
491                         },
492                         chan_id: None,
493                         shutdown_finish: None,
494                         channel_capacity: None,
495                 }
496         }
497 }
498
499 /// We hold back HTLCs we intend to relay for a random interval greater than this (see
500 /// Event::PendingHTLCsForwardable for the API guidelines indicating how long should be waited).
501 /// This provides some limited amount of privacy. Ideally this would range from somewhere like one
502 /// second to 30 seconds, but people expect lightning to be, you know, kinda fast, sadly.
503 pub(super) const MIN_HTLC_RELAY_HOLDING_CELL_MILLIS: u64 = 100;
504
505 /// For events which result in both a RevokeAndACK and a CommitmentUpdate, by default they should
506 /// be sent in the order they appear in the return value, however sometimes the order needs to be
507 /// variable at runtime (eg Channel::channel_reestablish needs to re-send messages in the order
508 /// they were originally sent). In those cases, this enum is also returned.
509 #[derive(Clone, PartialEq)]
510 pub(super) enum RAACommitmentOrder {
511         /// Send the CommitmentUpdate messages first
512         CommitmentFirst,
513         /// Send the RevokeAndACK message first
514         RevokeAndACKFirst,
515 }
516
517 /// Information about a payment which is currently being claimed.
518 struct ClaimingPayment {
519         amount_msat: u64,
520         payment_purpose: events::PaymentPurpose,
521         receiver_node_id: PublicKey,
522         htlcs: Vec<events::ClaimedHTLC>,
523         sender_intended_value: Option<u64>,
524 }
525 impl_writeable_tlv_based!(ClaimingPayment, {
526         (0, amount_msat, required),
527         (2, payment_purpose, required),
528         (4, receiver_node_id, required),
529         (5, htlcs, optional_vec),
530         (7, sender_intended_value, option),
531 });
532
533 struct ClaimablePayment {
534         purpose: events::PaymentPurpose,
535         onion_fields: Option<RecipientOnionFields>,
536         htlcs: Vec<ClaimableHTLC>,
537 }
538
539 /// Information about claimable or being-claimed payments
540 struct ClaimablePayments {
541         /// Map from payment hash to the payment data and any HTLCs which are to us and can be
542         /// failed/claimed by the user.
543         ///
544         /// Note that, no consistency guarantees are made about the channels given here actually
545         /// existing anymore by the time you go to read them!
546         ///
547         /// When adding to the map, [`Self::pending_claiming_payments`] must also be checked to ensure
548         /// we don't get a duplicate payment.
549         claimable_payments: HashMap<PaymentHash, ClaimablePayment>,
550
551         /// Map from payment hash to the payment data for HTLCs which we have begun claiming, but which
552         /// are waiting on a [`ChannelMonitorUpdate`] to complete in order to be surfaced to the user
553         /// as an [`events::Event::PaymentClaimed`].
554         pending_claiming_payments: HashMap<PaymentHash, ClaimingPayment>,
555 }
556
557 /// Events which we process internally but cannot be processed immediately at the generation site
558 /// usually because we're running pre-full-init. They are handled immediately once we detect we are
559 /// running normally, and specifically must be processed before any other non-background
560 /// [`ChannelMonitorUpdate`]s are applied.
561 enum BackgroundEvent {
562         /// Handle a ChannelMonitorUpdate which closes the channel or for an already-closed channel.
563         /// This is only separated from [`Self::MonitorUpdateRegeneratedOnStartup`] as the
564         /// maybe-non-closing variant needs a public key to handle channel resumption, whereas if the
565         /// channel has been force-closed we do not need the counterparty node_id.
566         ///
567         /// Note that any such events are lost on shutdown, so in general they must be updates which
568         /// are regenerated on startup.
569         ClosedMonitorUpdateRegeneratedOnStartup((OutPoint, ChannelMonitorUpdate)),
570         /// Handle a ChannelMonitorUpdate which may or may not close the channel and may unblock the
571         /// channel to continue normal operation.
572         ///
573         /// In general this should be used rather than
574         /// [`Self::ClosedMonitorUpdateRegeneratedOnStartup`], however in cases where the
575         /// `counterparty_node_id` is not available as the channel has closed from a [`ChannelMonitor`]
576         /// error the other variant is acceptable.
577         ///
578         /// Note that any such events are lost on shutdown, so in general they must be updates which
579         /// are regenerated on startup.
580         MonitorUpdateRegeneratedOnStartup {
581                 counterparty_node_id: PublicKey,
582                 funding_txo: OutPoint,
583                 update: ChannelMonitorUpdate
584         },
585         /// Some [`ChannelMonitorUpdate`] (s) completed before we were serialized but we still have
586         /// them marked pending, thus we need to run any [`MonitorUpdateCompletionAction`] (s) pending
587         /// on a channel.
588         MonitorUpdatesComplete {
589                 counterparty_node_id: PublicKey,
590                 channel_id: ChannelId,
591         },
592 }
593
594 #[derive(Debug)]
595 pub(crate) enum MonitorUpdateCompletionAction {
596         /// Indicates that a payment ultimately destined for us was claimed and we should emit an
597         /// [`events::Event::PaymentClaimed`] to the user if we haven't yet generated such an event for
598         /// this payment. Note that this is only best-effort. On restart it's possible such a duplicate
599         /// event can be generated.
600         PaymentClaimed { payment_hash: PaymentHash },
601         /// Indicates an [`events::Event`] should be surfaced to the user and possibly resume the
602         /// operation of another channel.
603         ///
604         /// This is usually generated when we've forwarded an HTLC and want to block the outbound edge
605         /// from completing a monitor update which removes the payment preimage until the inbound edge
606         /// completes a monitor update containing the payment preimage. In that case, after the inbound
607         /// edge completes, we will surface an [`Event::PaymentForwarded`] as well as unblock the
608         /// outbound edge.
609         EmitEventAndFreeOtherChannel {
610                 event: events::Event,
611                 downstream_counterparty_and_funding_outpoint: Option<(PublicKey, OutPoint, RAAMonitorUpdateBlockingAction)>,
612         },
613 }
614
615 impl_writeable_tlv_based_enum_upgradable!(MonitorUpdateCompletionAction,
616         (0, PaymentClaimed) => { (0, payment_hash, required) },
617         (2, EmitEventAndFreeOtherChannel) => {
618                 (0, event, upgradable_required),
619                 // LDK prior to 0.0.116 did not have this field as the monitor update application order was
620                 // required by clients. If we downgrade to something prior to 0.0.116 this may result in
621                 // monitor updates which aren't properly blocked or resumed, however that's fine - we don't
622                 // support async monitor updates even in LDK 0.0.116 and once we do we'll require no
623                 // downgrades to prior versions.
624                 (1, downstream_counterparty_and_funding_outpoint, option),
625         },
626 );
627
628 #[derive(Clone, Debug, PartialEq, Eq)]
629 pub(crate) enum EventCompletionAction {
630         ReleaseRAAChannelMonitorUpdate {
631                 counterparty_node_id: PublicKey,
632                 channel_funding_outpoint: OutPoint,
633         },
634 }
635 impl_writeable_tlv_based_enum!(EventCompletionAction,
636         (0, ReleaseRAAChannelMonitorUpdate) => {
637                 (0, channel_funding_outpoint, required),
638                 (2, counterparty_node_id, required),
639         };
640 );
641
642 #[derive(Clone, PartialEq, Eq, Debug)]
643 /// If something is blocked on the completion of an RAA-generated [`ChannelMonitorUpdate`] we track
644 /// the blocked action here. See enum variants for more info.
645 pub(crate) enum RAAMonitorUpdateBlockingAction {
646         /// A forwarded payment was claimed. We block the downstream channel completing its monitor
647         /// update which removes the HTLC preimage until the upstream channel has gotten the preimage
648         /// durably to disk.
649         ForwardedPaymentInboundClaim {
650                 /// The upstream channel ID (i.e. the inbound edge).
651                 channel_id: ChannelId,
652                 /// The HTLC ID on the inbound edge.
653                 htlc_id: u64,
654         },
655 }
656
657 impl RAAMonitorUpdateBlockingAction {
658         #[allow(unused)]
659         fn from_prev_hop_data(prev_hop: &HTLCPreviousHopData) -> Self {
660                 Self::ForwardedPaymentInboundClaim {
661                         channel_id: prev_hop.outpoint.to_channel_id(),
662                         htlc_id: prev_hop.htlc_id,
663                 }
664         }
665 }
666
667 impl_writeable_tlv_based_enum!(RAAMonitorUpdateBlockingAction,
668         (0, ForwardedPaymentInboundClaim) => { (0, channel_id, required), (2, htlc_id, required) }
669 ;);
670
671
672 /// State we hold per-peer.
673 pub(super) struct PeerState<SP: Deref> where SP::Target: SignerProvider {
674         /// `channel_id` -> `Channel`.
675         ///
676         /// Holds all funded channels where the peer is the counterparty.
677         pub(super) channel_by_id: HashMap<ChannelId, Channel<SP>>,
678         /// `temporary_channel_id` -> `OutboundV1Channel`.
679         ///
680         /// Holds all outbound V1 channels where the peer is the counterparty. Once an outbound channel has
681         /// been assigned a `channel_id`, the entry in this map is removed and one is created in
682         /// `channel_by_id`.
683         pub(super) outbound_v1_channel_by_id: HashMap<ChannelId, OutboundV1Channel<SP>>,
684         /// `temporary_channel_id` -> `InboundV1Channel`.
685         ///
686         /// Holds all inbound V1 channels where the peer is the counterparty. Once an inbound channel has
687         /// been assigned a `channel_id`, the entry in this map is removed and one is created in
688         /// `channel_by_id`.
689         pub(super) inbound_v1_channel_by_id: HashMap<ChannelId, InboundV1Channel<SP>>,
690         /// `temporary_channel_id` -> `InboundChannelRequest`.
691         ///
692         /// When manual channel acceptance is enabled, this holds all unaccepted inbound channels where
693         /// the peer is the counterparty. If the channel is accepted, then the entry in this table is
694         /// removed, and an InboundV1Channel is created and placed in the `inbound_v1_channel_by_id` table. If
695         /// the channel is rejected, then the entry is simply removed.
696         pub(super) inbound_channel_request_by_id: HashMap<ChannelId, InboundChannelRequest>,
697         /// The latest `InitFeatures` we heard from the peer.
698         latest_features: InitFeatures,
699         /// Messages to send to the peer - pushed to in the same lock that they are generated in (except
700         /// for broadcast messages, where ordering isn't as strict).
701         pub(super) pending_msg_events: Vec<MessageSendEvent>,
702         /// Map from Channel IDs to pending [`ChannelMonitorUpdate`]s which have been passed to the
703         /// user but which have not yet completed.
704         ///
705         /// Note that the channel may no longer exist. For example if the channel was closed but we
706         /// later needed to claim an HTLC which is pending on-chain, we may generate a monitor update
707         /// for a missing channel.
708         in_flight_monitor_updates: BTreeMap<OutPoint, Vec<ChannelMonitorUpdate>>,
709         /// Map from a specific channel to some action(s) that should be taken when all pending
710         /// [`ChannelMonitorUpdate`]s for the channel complete updating.
711         ///
712         /// Note that because we generally only have one entry here a HashMap is pretty overkill. A
713         /// BTreeMap currently stores more than ten elements per leaf node, so even up to a few
714         /// channels with a peer this will just be one allocation and will amount to a linear list of
715         /// channels to walk, avoiding the whole hashing rigmarole.
716         ///
717         /// Note that the channel may no longer exist. For example, if a channel was closed but we
718         /// later needed to claim an HTLC which is pending on-chain, we may generate a monitor update
719         /// for a missing channel. While a malicious peer could construct a second channel with the
720         /// same `temporary_channel_id` (or final `channel_id` in the case of 0conf channels or prior
721         /// to funding appearing on-chain), the downstream `ChannelMonitor` set is required to ensure
722         /// duplicates do not occur, so such channels should fail without a monitor update completing.
723         monitor_update_blocked_actions: BTreeMap<ChannelId, Vec<MonitorUpdateCompletionAction>>,
724         /// If another channel's [`ChannelMonitorUpdate`] needs to complete before a channel we have
725         /// with this peer can complete an RAA [`ChannelMonitorUpdate`] (e.g. because the RAA update
726         /// will remove a preimage that needs to be durably in an upstream channel first), we put an
727         /// entry here to note that the channel with the key's ID is blocked on a set of actions.
728         actions_blocking_raa_monitor_updates: BTreeMap<ChannelId, Vec<RAAMonitorUpdateBlockingAction>>,
729         /// The peer is currently connected (i.e. we've seen a
730         /// [`ChannelMessageHandler::peer_connected`] and no corresponding
731         /// [`ChannelMessageHandler::peer_disconnected`].
732         is_connected: bool,
733 }
734
735 impl <SP: Deref> PeerState<SP> where SP::Target: SignerProvider {
736         /// Indicates that a peer meets the criteria where we're ok to remove it from our storage.
737         /// If true is passed for `require_disconnected`, the function will return false if we haven't
738         /// disconnected from the node already, ie. `PeerState::is_connected` is set to `true`.
739         fn ok_to_remove(&self, require_disconnected: bool) -> bool {
740                 if require_disconnected && self.is_connected {
741                         return false
742                 }
743                 self.channel_by_id.is_empty() && self.monitor_update_blocked_actions.is_empty()
744                         && self.in_flight_monitor_updates.is_empty()
745         }
746
747         // Returns a count of all channels we have with this peer, including unfunded channels.
748         fn total_channel_count(&self) -> usize {
749                 self.channel_by_id.len() +
750                         self.outbound_v1_channel_by_id.len() +
751                         self.inbound_v1_channel_by_id.len() +
752                         self.inbound_channel_request_by_id.len()
753         }
754
755         // Returns a bool indicating if the given `channel_id` matches a channel we have with this peer.
756         fn has_channel(&self, channel_id: &ChannelId) -> bool {
757                 self.channel_by_id.contains_key(&channel_id) ||
758                         self.outbound_v1_channel_by_id.contains_key(&channel_id) ||
759                         self.inbound_v1_channel_by_id.contains_key(&channel_id) ||
760                         self.inbound_channel_request_by_id.contains_key(&channel_id)
761         }
762 }
763
764 /// A not-yet-accepted inbound (from counterparty) channel. Once
765 /// accepted, the parameters will be used to construct a channel.
766 pub(super) struct InboundChannelRequest {
767         /// The original OpenChannel message.
768         pub open_channel_msg: msgs::OpenChannel,
769         /// The number of ticks remaining before the request expires.
770         pub ticks_remaining: i32,
771 }
772
773 /// The number of ticks that may elapse while we're waiting for an unaccepted inbound channel to be
774 /// accepted. An unaccepted channel that exceeds this limit will be abandoned.
775 const UNACCEPTED_INBOUND_CHANNEL_AGE_LIMIT_TICKS: i32 = 2;
776
777 /// Stores a PaymentSecret and any other data we may need to validate an inbound payment is
778 /// actually ours and not some duplicate HTLC sent to us by a node along the route.
779 ///
780 /// For users who don't want to bother doing their own payment preimage storage, we also store that
781 /// here.
782 ///
783 /// Note that this struct will be removed entirely soon, in favor of storing no inbound payment data
784 /// and instead encoding it in the payment secret.
785 struct PendingInboundPayment {
786         /// The payment secret that the sender must use for us to accept this payment
787         payment_secret: PaymentSecret,
788         /// Time at which this HTLC expires - blocks with a header time above this value will result in
789         /// this payment being removed.
790         expiry_time: u64,
791         /// Arbitrary identifier the user specifies (or not)
792         user_payment_id: u64,
793         // Other required attributes of the payment, optionally enforced:
794         payment_preimage: Option<PaymentPreimage>,
795         min_value_msat: Option<u64>,
796 }
797
798 /// [`SimpleArcChannelManager`] is useful when you need a [`ChannelManager`] with a static lifetime, e.g.
799 /// when you're using `lightning-net-tokio` (since `tokio::spawn` requires parameters with static
800 /// lifetimes). Other times you can afford a reference, which is more efficient, in which case
801 /// [`SimpleRefChannelManager`] is the more appropriate type. Defining these type aliases prevents
802 /// issues such as overly long function definitions. Note that the `ChannelManager` can take any type
803 /// that implements [`NodeSigner`], [`EntropySource`], and [`SignerProvider`] for its keys manager,
804 /// or, respectively, [`Router`] for its router, but this type alias chooses the concrete types
805 /// of [`KeysManager`] and [`DefaultRouter`].
806 ///
807 /// This is not exported to bindings users as Arcs don't make sense in bindings
808 pub type SimpleArcChannelManager<M, T, F, L> = ChannelManager<
809         Arc<M>,
810         Arc<T>,
811         Arc<KeysManager>,
812         Arc<KeysManager>,
813         Arc<KeysManager>,
814         Arc<F>,
815         Arc<DefaultRouter<
816                 Arc<NetworkGraph<Arc<L>>>,
817                 Arc<L>,
818                 Arc<Mutex<ProbabilisticScorer<Arc<NetworkGraph<Arc<L>>>, Arc<L>>>>,
819                 ProbabilisticScoringFeeParameters,
820                 ProbabilisticScorer<Arc<NetworkGraph<Arc<L>>>, Arc<L>>,
821         >>,
822         Arc<L>
823 >;
824
825 /// [`SimpleRefChannelManager`] is a type alias for a ChannelManager reference, and is the reference
826 /// counterpart to the [`SimpleArcChannelManager`] type alias. Use this type by default when you don't
827 /// need a ChannelManager with a static lifetime. You'll need a static lifetime in cases such as
828 /// usage of lightning-net-tokio (since `tokio::spawn` requires parameters with static lifetimes).
829 /// But if this is not necessary, using a reference is more efficient. Defining these type aliases
830 /// issues such as overly long function definitions. Note that the ChannelManager can take any type
831 /// that implements [`NodeSigner`], [`EntropySource`], and [`SignerProvider`] for its keys manager,
832 /// or, respectively, [`Router`]  for its router, but this type alias chooses the concrete types
833 /// of [`KeysManager`] and [`DefaultRouter`].
834 ///
835 /// This is not exported to bindings users as Arcs don't make sense in bindings
836 pub type SimpleRefChannelManager<'a, 'b, 'c, 'd, 'e, 'f, 'g, 'h, M, T, F, L> =
837         ChannelManager<
838                 &'a M,
839                 &'b T,
840                 &'c KeysManager,
841                 &'c KeysManager,
842                 &'c KeysManager,
843                 &'d F,
844                 &'e DefaultRouter<
845                         &'f NetworkGraph<&'g L>,
846                         &'g L,
847                         &'h Mutex<ProbabilisticScorer<&'f NetworkGraph<&'g L>, &'g L>>,
848                         ProbabilisticScoringFeeParameters,
849                         ProbabilisticScorer<&'f NetworkGraph<&'g L>, &'g L>
850                 >,
851                 &'g L
852         >;
853
854 macro_rules! define_test_pub_trait { ($vis: vis) => {
855 /// A trivial trait which describes any [`ChannelManager`] used in testing.
856 $vis trait AChannelManager {
857         type Watch: chain::Watch<Self::Signer> + ?Sized;
858         type M: Deref<Target = Self::Watch>;
859         type Broadcaster: BroadcasterInterface + ?Sized;
860         type T: Deref<Target = Self::Broadcaster>;
861         type EntropySource: EntropySource + ?Sized;
862         type ES: Deref<Target = Self::EntropySource>;
863         type NodeSigner: NodeSigner + ?Sized;
864         type NS: Deref<Target = Self::NodeSigner>;
865         type Signer: WriteableEcdsaChannelSigner + Sized;
866         type SignerProvider: SignerProvider<Signer = Self::Signer> + ?Sized;
867         type SP: Deref<Target = Self::SignerProvider>;
868         type FeeEstimator: FeeEstimator + ?Sized;
869         type F: Deref<Target = Self::FeeEstimator>;
870         type Router: Router + ?Sized;
871         type R: Deref<Target = Self::Router>;
872         type Logger: Logger + ?Sized;
873         type L: Deref<Target = Self::Logger>;
874         fn get_cm(&self) -> &ChannelManager<Self::M, Self::T, Self::ES, Self::NS, Self::SP, Self::F, Self::R, Self::L>;
875 }
876 } }
877 #[cfg(any(test, feature = "_test_utils"))]
878 define_test_pub_trait!(pub);
879 #[cfg(not(any(test, feature = "_test_utils")))]
880 define_test_pub_trait!(pub(crate));
881 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref> AChannelManager
882 for ChannelManager<M, T, ES, NS, SP, F, R, L>
883 where
884         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
885         T::Target: BroadcasterInterface,
886         ES::Target: EntropySource,
887         NS::Target: NodeSigner,
888         SP::Target: SignerProvider,
889         F::Target: FeeEstimator,
890         R::Target: Router,
891         L::Target: Logger,
892 {
893         type Watch = M::Target;
894         type M = M;
895         type Broadcaster = T::Target;
896         type T = T;
897         type EntropySource = ES::Target;
898         type ES = ES;
899         type NodeSigner = NS::Target;
900         type NS = NS;
901         type Signer = <SP::Target as SignerProvider>::Signer;
902         type SignerProvider = SP::Target;
903         type SP = SP;
904         type FeeEstimator = F::Target;
905         type F = F;
906         type Router = R::Target;
907         type R = R;
908         type Logger = L::Target;
909         type L = L;
910         fn get_cm(&self) -> &ChannelManager<M, T, ES, NS, SP, F, R, L> { self }
911 }
912
913 /// Manager which keeps track of a number of channels and sends messages to the appropriate
914 /// channel, also tracking HTLC preimages and forwarding onion packets appropriately.
915 ///
916 /// Implements [`ChannelMessageHandler`], handling the multi-channel parts and passing things through
917 /// to individual Channels.
918 ///
919 /// Implements [`Writeable`] to write out all channel state to disk. Implies [`peer_disconnected`] for
920 /// all peers during write/read (though does not modify this instance, only the instance being
921 /// serialized). This will result in any channels which have not yet exchanged [`funding_created`] (i.e.,
922 /// called [`funding_transaction_generated`] for outbound channels) being closed.
923 ///
924 /// Note that you can be a bit lazier about writing out `ChannelManager` than you can be with
925 /// [`ChannelMonitor`]. With [`ChannelMonitor`] you MUST write each monitor update out to disk before
926 /// returning from [`chain::Watch::watch_channel`]/[`update_channel`], with ChannelManagers, writing updates
927 /// happens out-of-band (and will prevent any other `ChannelManager` operations from occurring during
928 /// the serialization process). If the deserialized version is out-of-date compared to the
929 /// [`ChannelMonitor`] passed by reference to [`read`], those channels will be force-closed based on the
930 /// `ChannelMonitor` state and no funds will be lost (mod on-chain transaction fees).
931 ///
932 /// Note that the deserializer is only implemented for `(`[`BlockHash`]`, `[`ChannelManager`]`)`, which
933 /// tells you the last block hash which was connected. You should get the best block tip before using the manager.
934 /// See [`chain::Listen`] and [`chain::Confirm`] for more details.
935 ///
936 /// Note that `ChannelManager` is responsible for tracking liveness of its channels and generating
937 /// [`ChannelUpdate`] messages informing peers that the channel is temporarily disabled. To avoid
938 /// spam due to quick disconnection/reconnection, updates are not sent until the channel has been
939 /// offline for a full minute. In order to track this, you must call
940 /// [`timer_tick_occurred`] roughly once per minute, though it doesn't have to be perfect.
941 ///
942 /// To avoid trivial DoS issues, `ChannelManager` limits the number of inbound connections and
943 /// inbound channels without confirmed funding transactions. This may result in nodes which we do
944 /// not have a channel with being unable to connect to us or open new channels with us if we have
945 /// many peers with unfunded channels.
946 ///
947 /// Because it is an indication of trust, inbound channels which we've accepted as 0conf are
948 /// exempted from the count of unfunded channels. Similarly, outbound channels and connections are
949 /// never limited. Please ensure you limit the count of such channels yourself.
950 ///
951 /// Rather than using a plain `ChannelManager`, it is preferable to use either a [`SimpleArcChannelManager`]
952 /// a [`SimpleRefChannelManager`], for conciseness. See their documentation for more details, but
953 /// essentially you should default to using a [`SimpleRefChannelManager`], and use a
954 /// [`SimpleArcChannelManager`] when you require a `ChannelManager` with a static lifetime, such as when
955 /// you're using lightning-net-tokio.
956 ///
957 /// [`peer_disconnected`]: msgs::ChannelMessageHandler::peer_disconnected
958 /// [`funding_created`]: msgs::FundingCreated
959 /// [`funding_transaction_generated`]: Self::funding_transaction_generated
960 /// [`BlockHash`]: bitcoin::hash_types::BlockHash
961 /// [`update_channel`]: chain::Watch::update_channel
962 /// [`ChannelUpdate`]: msgs::ChannelUpdate
963 /// [`timer_tick_occurred`]: Self::timer_tick_occurred
964 /// [`read`]: ReadableArgs::read
965 //
966 // Lock order:
967 // The tree structure below illustrates the lock order requirements for the different locks of the
968 // `ChannelManager`. Locks can be held at the same time if they are on the same branch in the tree,
969 // and should then be taken in the order of the lowest to the highest level in the tree.
970 // Note that locks on different branches shall not be taken at the same time, as doing so will
971 // create a new lock order for those specific locks in the order they were taken.
972 //
973 // Lock order tree:
974 //
975 // `total_consistency_lock`
976 //  |
977 //  |__`forward_htlcs`
978 //  |   |
979 //  |   |__`pending_intercepted_htlcs`
980 //  |
981 //  |__`per_peer_state`
982 //  |   |
983 //  |   |__`pending_inbound_payments`
984 //  |       |
985 //  |       |__`claimable_payments`
986 //  |       |
987 //  |       |__`pending_outbound_payments` // This field's struct contains a map of pending outbounds
988 //  |           |
989 //  |           |__`peer_state`
990 //  |               |
991 //  |               |__`id_to_peer`
992 //  |               |
993 //  |               |__`short_to_chan_info`
994 //  |               |
995 //  |               |__`outbound_scid_aliases`
996 //  |               |
997 //  |               |__`best_block`
998 //  |               |
999 //  |               |__`pending_events`
1000 //  |                   |
1001 //  |                   |__`pending_background_events`
1002 //
1003 pub struct ChannelManager<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref>
1004 where
1005         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
1006         T::Target: BroadcasterInterface,
1007         ES::Target: EntropySource,
1008         NS::Target: NodeSigner,
1009         SP::Target: SignerProvider,
1010         F::Target: FeeEstimator,
1011         R::Target: Router,
1012         L::Target: Logger,
1013 {
1014         default_configuration: UserConfig,
1015         genesis_hash: BlockHash,
1016         fee_estimator: LowerBoundedFeeEstimator<F>,
1017         chain_monitor: M,
1018         tx_broadcaster: T,
1019         #[allow(unused)]
1020         router: R,
1021
1022         /// See `ChannelManager` struct-level documentation for lock order requirements.
1023         #[cfg(test)]
1024         pub(super) best_block: RwLock<BestBlock>,
1025         #[cfg(not(test))]
1026         best_block: RwLock<BestBlock>,
1027         secp_ctx: Secp256k1<secp256k1::All>,
1028
1029         /// Storage for PaymentSecrets and any requirements on future inbound payments before we will
1030         /// expose them to users via a PaymentClaimable event. HTLCs which do not meet the requirements
1031         /// here are failed when we process them as pending-forwardable-HTLCs, and entries are removed
1032         /// after we generate a PaymentClaimable upon receipt of all MPP parts or when they time out.
1033         ///
1034         /// See `ChannelManager` struct-level documentation for lock order requirements.
1035         pending_inbound_payments: Mutex<HashMap<PaymentHash, PendingInboundPayment>>,
1036
1037         /// The session_priv bytes and retry metadata of outbound payments which are pending resolution.
1038         /// The authoritative state of these HTLCs resides either within Channels or ChannelMonitors
1039         /// (if the channel has been force-closed), however we track them here to prevent duplicative
1040         /// PaymentSent/PaymentPathFailed events. Specifically, in the case of a duplicative
1041         /// update_fulfill_htlc message after a reconnect, we may "claim" a payment twice.
1042         /// Additionally, because ChannelMonitors are often not re-serialized after connecting block(s)
1043         /// which may generate a claim event, we may receive similar duplicate claim/fail MonitorEvents
1044         /// after reloading from disk while replaying blocks against ChannelMonitors.
1045         ///
1046         /// See `PendingOutboundPayment` documentation for more info.
1047         ///
1048         /// See `ChannelManager` struct-level documentation for lock order requirements.
1049         pending_outbound_payments: OutboundPayments,
1050
1051         /// SCID/SCID Alias -> forward infos. Key of 0 means payments received.
1052         ///
1053         /// Note that because we may have an SCID Alias as the key we can have two entries per channel,
1054         /// though in practice we probably won't be receiving HTLCs for a channel both via the alias
1055         /// and via the classic SCID.
1056         ///
1057         /// Note that no consistency guarantees are made about the existence of a channel with the
1058         /// `short_channel_id` here, nor the `short_channel_id` in the `PendingHTLCInfo`!
1059         ///
1060         /// See `ChannelManager` struct-level documentation for lock order requirements.
1061         #[cfg(test)]
1062         pub(super) forward_htlcs: Mutex<HashMap<u64, Vec<HTLCForwardInfo>>>,
1063         #[cfg(not(test))]
1064         forward_htlcs: Mutex<HashMap<u64, Vec<HTLCForwardInfo>>>,
1065         /// Storage for HTLCs that have been intercepted and bubbled up to the user. We hold them here
1066         /// until the user tells us what we should do with them.
1067         ///
1068         /// See `ChannelManager` struct-level documentation for lock order requirements.
1069         pending_intercepted_htlcs: Mutex<HashMap<InterceptId, PendingAddHTLCInfo>>,
1070
1071         /// The sets of payments which are claimable or currently being claimed. See
1072         /// [`ClaimablePayments`]' individual field docs for more info.
1073         ///
1074         /// See `ChannelManager` struct-level documentation for lock order requirements.
1075         claimable_payments: Mutex<ClaimablePayments>,
1076
1077         /// The set of outbound SCID aliases across all our channels, including unconfirmed channels
1078         /// and some closed channels which reached a usable state prior to being closed. This is used
1079         /// only to avoid duplicates, and is not persisted explicitly to disk, but rebuilt from the
1080         /// active channel list on load.
1081         ///
1082         /// See `ChannelManager` struct-level documentation for lock order requirements.
1083         outbound_scid_aliases: Mutex<HashSet<u64>>,
1084
1085         /// `channel_id` -> `counterparty_node_id`.
1086         ///
1087         /// Only `channel_id`s are allowed as keys in this map, and not `temporary_channel_id`s. As
1088         /// multiple channels with the same `temporary_channel_id` to different peers can exist,
1089         /// allowing `temporary_channel_id`s in this map would cause collisions for such channels.
1090         ///
1091         /// Note that this map should only be used for `MonitorEvent` handling, to be able to access
1092         /// the corresponding channel for the event, as we only have access to the `channel_id` during
1093         /// the handling of the events.
1094         ///
1095         /// Note that no consistency guarantees are made about the existence of a peer with the
1096         /// `counterparty_node_id` in our other maps.
1097         ///
1098         /// TODO:
1099         /// The `counterparty_node_id` isn't passed with `MonitorEvent`s currently. To pass it, we need
1100         /// to make `counterparty_node_id`'s a required field in `ChannelMonitor`s, which unfortunately
1101         /// would break backwards compatability.
1102         /// We should add `counterparty_node_id`s to `MonitorEvent`s, and eventually rely on it in the
1103         /// future. That would make this map redundant, as only the `ChannelManager::per_peer_state` is
1104         /// required to access the channel with the `counterparty_node_id`.
1105         ///
1106         /// See `ChannelManager` struct-level documentation for lock order requirements.
1107         id_to_peer: Mutex<HashMap<ChannelId, PublicKey>>,
1108
1109         /// SCIDs (and outbound SCID aliases) -> `counterparty_node_id`s and `channel_id`s.
1110         ///
1111         /// Outbound SCID aliases are added here once the channel is available for normal use, with
1112         /// SCIDs being added once the funding transaction is confirmed at the channel's required
1113         /// confirmation depth.
1114         ///
1115         /// Note that while this holds `counterparty_node_id`s and `channel_id`s, no consistency
1116         /// guarantees are made about the existence of a peer with the `counterparty_node_id` nor a
1117         /// channel with the `channel_id` in our other maps.
1118         ///
1119         /// See `ChannelManager` struct-level documentation for lock order requirements.
1120         #[cfg(test)]
1121         pub(super) short_to_chan_info: FairRwLock<HashMap<u64, (PublicKey, ChannelId)>>,
1122         #[cfg(not(test))]
1123         short_to_chan_info: FairRwLock<HashMap<u64, (PublicKey, ChannelId)>>,
1124
1125         our_network_pubkey: PublicKey,
1126
1127         inbound_payment_key: inbound_payment::ExpandedKey,
1128
1129         /// LDK puts the [fake scids] that it generates into namespaces, to identify the type of an
1130         /// incoming payment. To make it harder for a third-party to identify the type of a payment,
1131         /// we encrypt the namespace identifier using these bytes.
1132         ///
1133         /// [fake scids]: crate::util::scid_utils::fake_scid
1134         fake_scid_rand_bytes: [u8; 32],
1135
1136         /// When we send payment probes, we generate the [`PaymentHash`] based on this cookie secret
1137         /// and a random [`PaymentId`]. This allows us to discern probes from real payments, without
1138         /// keeping additional state.
1139         probing_cookie_secret: [u8; 32],
1140
1141         /// The highest block timestamp we've seen, which is usually a good guess at the current time.
1142         /// Assuming most miners are generating blocks with reasonable timestamps, this shouldn't be
1143         /// very far in the past, and can only ever be up to two hours in the future.
1144         highest_seen_timestamp: AtomicUsize,
1145
1146         /// The bulk of our storage. Currently the `per_peer_state` stores our channels on a per-peer
1147         /// basis, as well as the peer's latest features.
1148         ///
1149         /// If we are connected to a peer we always at least have an entry here, even if no channels
1150         /// are currently open with that peer.
1151         ///
1152         /// Because adding or removing an entry is rare, we usually take an outer read lock and then
1153         /// operate on the inner value freely. This opens up for parallel per-peer operation for
1154         /// channels.
1155         ///
1156         /// Note that the same thread must never acquire two inner `PeerState` locks at the same time.
1157         ///
1158         /// See `ChannelManager` struct-level documentation for lock order requirements.
1159         #[cfg(not(any(test, feature = "_test_utils")))]
1160         per_peer_state: FairRwLock<HashMap<PublicKey, Mutex<PeerState<SP>>>>,
1161         #[cfg(any(test, feature = "_test_utils"))]
1162         pub(super) per_peer_state: FairRwLock<HashMap<PublicKey, Mutex<PeerState<SP>>>>,
1163
1164         /// The set of events which we need to give to the user to handle. In some cases an event may
1165         /// require some further action after the user handles it (currently only blocking a monitor
1166         /// update from being handed to the user to ensure the included changes to the channel state
1167         /// are handled by the user before they're persisted durably to disk). In that case, the second
1168         /// element in the tuple is set to `Some` with further details of the action.
1169         ///
1170         /// Note that events MUST NOT be removed from pending_events after deserialization, as they
1171         /// could be in the middle of being processed without the direct mutex held.
1172         ///
1173         /// See `ChannelManager` struct-level documentation for lock order requirements.
1174         #[cfg(not(any(test, feature = "_test_utils")))]
1175         pending_events: Mutex<VecDeque<(events::Event, Option<EventCompletionAction>)>>,
1176         #[cfg(any(test, feature = "_test_utils"))]
1177         pub(crate) pending_events: Mutex<VecDeque<(events::Event, Option<EventCompletionAction>)>>,
1178
1179         /// A simple atomic flag to ensure only one task at a time can be processing events asynchronously.
1180         pending_events_processor: AtomicBool,
1181
1182         /// If we are running during init (either directly during the deserialization method or in
1183         /// block connection methods which run after deserialization but before normal operation) we
1184         /// cannot provide the user with [`ChannelMonitorUpdate`]s through the normal update flow -
1185         /// prior to normal operation the user may not have loaded the [`ChannelMonitor`]s into their
1186         /// [`ChainMonitor`] and thus attempting to update it will fail or panic.
1187         ///
1188         /// Thus, we place them here to be handled as soon as possible once we are running normally.
1189         ///
1190         /// See `ChannelManager` struct-level documentation for lock order requirements.
1191         ///
1192         /// [`ChainMonitor`]: crate::chain::chainmonitor::ChainMonitor
1193         pending_background_events: Mutex<Vec<BackgroundEvent>>,
1194         /// Used when we have to take a BIG lock to make sure everything is self-consistent.
1195         /// Essentially just when we're serializing ourselves out.
1196         /// Taken first everywhere where we are making changes before any other locks.
1197         /// When acquiring this lock in read mode, rather than acquiring it directly, call
1198         /// `PersistenceNotifierGuard::notify_on_drop(..)` and pass the lock to it, to ensure the
1199         /// Notifier the lock contains sends out a notification when the lock is released.
1200         total_consistency_lock: RwLock<()>,
1201
1202         background_events_processed_since_startup: AtomicBool,
1203
1204         persistence_notifier: Notifier,
1205
1206         entropy_source: ES,
1207         node_signer: NS,
1208         signer_provider: SP,
1209
1210         logger: L,
1211 }
1212
1213 /// Chain-related parameters used to construct a new `ChannelManager`.
1214 ///
1215 /// Typically, the block-specific parameters are derived from the best block hash for the network,
1216 /// as a newly constructed `ChannelManager` will not have created any channels yet. These parameters
1217 /// are not needed when deserializing a previously constructed `ChannelManager`.
1218 #[derive(Clone, Copy, PartialEq)]
1219 pub struct ChainParameters {
1220         /// The network for determining the `chain_hash` in Lightning messages.
1221         pub network: Network,
1222
1223         /// The hash and height of the latest block successfully connected.
1224         ///
1225         /// Used to track on-chain channel funding outputs and send payments with reliable timelocks.
1226         pub best_block: BestBlock,
1227 }
1228
1229 #[derive(Copy, Clone, PartialEq)]
1230 #[must_use]
1231 enum NotifyOption {
1232         DoPersist,
1233         SkipPersist,
1234 }
1235
1236 /// Whenever we release the `ChannelManager`'s `total_consistency_lock`, from read mode, it is
1237 /// desirable to notify any listeners on `await_persistable_update_timeout`/
1238 /// `await_persistable_update` when new updates are available for persistence. Therefore, this
1239 /// struct is responsible for locking the total consistency lock and, upon going out of scope,
1240 /// sending the aforementioned notification (since the lock being released indicates that the
1241 /// updates are ready for persistence).
1242 ///
1243 /// We allow callers to either always notify by constructing with `notify_on_drop` or choose to
1244 /// notify or not based on whether relevant changes have been made, providing a closure to
1245 /// `optionally_notify` which returns a `NotifyOption`.
1246 struct PersistenceNotifierGuard<'a, F: Fn() -> NotifyOption> {
1247         persistence_notifier: &'a Notifier,
1248         should_persist: F,
1249         // We hold onto this result so the lock doesn't get released immediately.
1250         _read_guard: RwLockReadGuard<'a, ()>,
1251 }
1252
1253 impl<'a> PersistenceNotifierGuard<'a, fn() -> NotifyOption> { // We don't care what the concrete F is here, it's unused
1254         fn notify_on_drop<C: AChannelManager>(cm: &'a C) -> PersistenceNotifierGuard<'a, impl Fn() -> NotifyOption> {
1255                 let read_guard = cm.get_cm().total_consistency_lock.read().unwrap();
1256                 let _ = cm.get_cm().process_background_events(); // We always persist
1257
1258                 PersistenceNotifierGuard {
1259                         persistence_notifier: &cm.get_cm().persistence_notifier,
1260                         should_persist: || -> NotifyOption { NotifyOption::DoPersist },
1261                         _read_guard: read_guard,
1262                 }
1263
1264         }
1265
1266         /// Note that if any [`ChannelMonitorUpdate`]s are possibly generated,
1267         /// [`ChannelManager::process_background_events`] MUST be called first.
1268         fn optionally_notify<F: Fn() -> NotifyOption>(lock: &'a RwLock<()>, notifier: &'a Notifier, persist_check: F) -> PersistenceNotifierGuard<'a, F> {
1269                 let read_guard = lock.read().unwrap();
1270
1271                 PersistenceNotifierGuard {
1272                         persistence_notifier: notifier,
1273                         should_persist: persist_check,
1274                         _read_guard: read_guard,
1275                 }
1276         }
1277 }
1278
1279 impl<'a, F: Fn() -> NotifyOption> Drop for PersistenceNotifierGuard<'a, F> {
1280         fn drop(&mut self) {
1281                 if (self.should_persist)() == NotifyOption::DoPersist {
1282                         self.persistence_notifier.notify();
1283                 }
1284         }
1285 }
1286
1287 /// The amount of time in blocks we require our counterparty wait to claim their money (ie time
1288 /// between when we, or our watchtower, must check for them having broadcast a theft transaction).
1289 ///
1290 /// This can be increased (but not decreased) through [`ChannelHandshakeConfig::our_to_self_delay`]
1291 ///
1292 /// [`ChannelHandshakeConfig::our_to_self_delay`]: crate::util::config::ChannelHandshakeConfig::our_to_self_delay
1293 pub const BREAKDOWN_TIMEOUT: u16 = 6 * 24;
1294 /// The amount of time in blocks we're willing to wait to claim money back to us. This matches
1295 /// the maximum required amount in lnd as of March 2021.
1296 pub(crate) const MAX_LOCAL_BREAKDOWN_TIMEOUT: u16 = 2 * 6 * 24 * 7;
1297
1298 /// The minimum number of blocks between an inbound HTLC's CLTV and the corresponding outbound
1299 /// HTLC's CLTV. The current default represents roughly seven hours of blocks at six blocks/hour.
1300 ///
1301 /// This can be increased (but not decreased) through [`ChannelConfig::cltv_expiry_delta`]
1302 ///
1303 /// [`ChannelConfig::cltv_expiry_delta`]: crate::util::config::ChannelConfig::cltv_expiry_delta
1304 // This should always be a few blocks greater than channelmonitor::CLTV_CLAIM_BUFFER,
1305 // i.e. the node we forwarded the payment on to should always have enough room to reliably time out
1306 // the HTLC via a full update_fail_htlc/commitment_signed dance before we hit the
1307 // CLTV_CLAIM_BUFFER point (we static assert that it's at least 3 blocks more).
1308 pub const MIN_CLTV_EXPIRY_DELTA: u16 = 6*7;
1309 // This should be long enough to allow a payment path drawn across multiple routing hops with substantial
1310 // `cltv_expiry_delta`. Indeed, the length of those values is the reaction delay offered to a routing node
1311 // in case of HTLC on-chain settlement. While appearing less competitive, a node operator could decide to
1312 // scale them up to suit its security policy. At the network-level, we shouldn't constrain them too much,
1313 // while avoiding to introduce a DoS vector. Further, a low CTLV_FAR_FAR_AWAY could be a source of
1314 // routing failure for any HTLC sender picking up an LDK node among the first hops.
1315 pub(super) const CLTV_FAR_FAR_AWAY: u32 = 14 * 24 * 6;
1316
1317 /// Minimum CLTV difference between the current block height and received inbound payments.
1318 /// Invoices generated for payment to us must set their `min_final_cltv_expiry_delta` field to at least
1319 /// this value.
1320 // Note that we fail if exactly HTLC_FAIL_BACK_BUFFER + 1 was used, so we need to add one for
1321 // any payments to succeed. Further, we don't want payments to fail if a block was found while
1322 // a payment was being routed, so we add an extra block to be safe.
1323 pub const MIN_FINAL_CLTV_EXPIRY_DELTA: u16 = HTLC_FAIL_BACK_BUFFER as u16 + 3;
1324
1325 // Check that our CLTV_EXPIRY is at least CLTV_CLAIM_BUFFER + ANTI_REORG_DELAY + LATENCY_GRACE_PERIOD_BLOCKS,
1326 // ie that if the next-hop peer fails the HTLC within
1327 // LATENCY_GRACE_PERIOD_BLOCKS then we'll still have CLTV_CLAIM_BUFFER left to timeout it onchain,
1328 // then waiting ANTI_REORG_DELAY to be reorg-safe on the outbound HLTC and
1329 // failing the corresponding htlc backward, and us now seeing the last block of ANTI_REORG_DELAY before
1330 // LATENCY_GRACE_PERIOD_BLOCKS.
1331 #[deny(const_err)]
1332 #[allow(dead_code)]
1333 const CHECK_CLTV_EXPIRY_SANITY: u32 = MIN_CLTV_EXPIRY_DELTA as u32 - LATENCY_GRACE_PERIOD_BLOCKS - CLTV_CLAIM_BUFFER - ANTI_REORG_DELAY - LATENCY_GRACE_PERIOD_BLOCKS;
1334
1335 // Check for ability of an attacker to make us fail on-chain by delaying an HTLC claim. See
1336 // ChannelMonitor::should_broadcast_holder_commitment_txn for a description of why this is needed.
1337 #[deny(const_err)]
1338 #[allow(dead_code)]
1339 const CHECK_CLTV_EXPIRY_SANITY_2: u32 = MIN_CLTV_EXPIRY_DELTA as u32 - LATENCY_GRACE_PERIOD_BLOCKS - 2*CLTV_CLAIM_BUFFER;
1340
1341 /// The number of ticks of [`ChannelManager::timer_tick_occurred`] until expiry of incomplete MPPs
1342 pub(crate) const MPP_TIMEOUT_TICKS: u8 = 3;
1343
1344 /// The number of ticks of [`ChannelManager::timer_tick_occurred`] where a peer is disconnected
1345 /// until we mark the channel disabled and gossip the update.
1346 pub(crate) const DISABLE_GOSSIP_TICKS: u8 = 10;
1347
1348 /// The number of ticks of [`ChannelManager::timer_tick_occurred`] where a peer is connected until
1349 /// we mark the channel enabled and gossip the update.
1350 pub(crate) const ENABLE_GOSSIP_TICKS: u8 = 5;
1351
1352 /// The maximum number of unfunded channels we can have per-peer before we start rejecting new
1353 /// (inbound) ones. The number of peers with unfunded channels is limited separately in
1354 /// [`MAX_UNFUNDED_CHANNEL_PEERS`].
1355 const MAX_UNFUNDED_CHANS_PER_PEER: usize = 4;
1356
1357 /// The maximum number of peers from which we will allow pending unfunded channels. Once we reach
1358 /// this many peers we reject new (inbound) channels from peers with which we don't have a channel.
1359 const MAX_UNFUNDED_CHANNEL_PEERS: usize = 50;
1360
1361 /// The maximum number of peers which we do not have a (funded) channel with. Once we reach this
1362 /// many peers we reject new (inbound) connections.
1363 const MAX_NO_CHANNEL_PEERS: usize = 250;
1364
1365 /// Information needed for constructing an invoice route hint for this channel.
1366 #[derive(Clone, Debug, PartialEq)]
1367 pub struct CounterpartyForwardingInfo {
1368         /// Base routing fee in millisatoshis.
1369         pub fee_base_msat: u32,
1370         /// Amount in millionths of a satoshi the channel will charge per transferred satoshi.
1371         pub fee_proportional_millionths: u32,
1372         /// The minimum difference in cltv_expiry between an ingoing HTLC and its outgoing counterpart,
1373         /// such that the outgoing HTLC is forwardable to this counterparty. See `msgs::ChannelUpdate`'s
1374         /// `cltv_expiry_delta` for more details.
1375         pub cltv_expiry_delta: u16,
1376 }
1377
1378 /// Channel parameters which apply to our counterparty. These are split out from [`ChannelDetails`]
1379 /// to better separate parameters.
1380 #[derive(Clone, Debug, PartialEq)]
1381 pub struct ChannelCounterparty {
1382         /// The node_id of our counterparty
1383         pub node_id: PublicKey,
1384         /// The Features the channel counterparty provided upon last connection.
1385         /// Useful for routing as it is the most up-to-date copy of the counterparty's features and
1386         /// many routing-relevant features are present in the init context.
1387         pub features: InitFeatures,
1388         /// The value, in satoshis, that must always be held in the channel for our counterparty. This
1389         /// value ensures that if our counterparty broadcasts a revoked state, we can punish them by
1390         /// claiming at least this value on chain.
1391         ///
1392         /// This value is not included in [`inbound_capacity_msat`] as it can never be spent.
1393         ///
1394         /// [`inbound_capacity_msat`]: ChannelDetails::inbound_capacity_msat
1395         pub unspendable_punishment_reserve: u64,
1396         /// Information on the fees and requirements that the counterparty requires when forwarding
1397         /// payments to us through this channel.
1398         pub forwarding_info: Option<CounterpartyForwardingInfo>,
1399         /// The smallest value HTLC (in msat) the remote peer will accept, for this channel. This field
1400         /// is only `None` before we have received either the `OpenChannel` or `AcceptChannel` message
1401         /// from the remote peer, or for `ChannelCounterparty` objects serialized prior to LDK 0.0.107.
1402         pub outbound_htlc_minimum_msat: Option<u64>,
1403         /// The largest value HTLC (in msat) the remote peer currently will accept, for this channel.
1404         pub outbound_htlc_maximum_msat: Option<u64>,
1405 }
1406
1407 /// Details of a channel, as returned by [`ChannelManager::list_channels`] and [`ChannelManager::list_usable_channels`]
1408 ///
1409 /// Balances of a channel are available through [`ChainMonitor::get_claimable_balances`] and
1410 /// [`ChannelMonitor::get_claimable_balances`], calculated with respect to the corresponding on-chain
1411 /// transactions.
1412 ///
1413 /// [`ChainMonitor::get_claimable_balances`]: crate::chain::chainmonitor::ChainMonitor::get_claimable_balances
1414 #[derive(Clone, Debug, PartialEq)]
1415 pub struct ChannelDetails {
1416         /// The channel's ID (prior to funding transaction generation, this is a random 32 bytes,
1417         /// thereafter this is the txid of the funding transaction xor the funding transaction output).
1418         /// Note that this means this value is *not* persistent - it can change once during the
1419         /// lifetime of the channel.
1420         pub channel_id: ChannelId,
1421         /// Parameters which apply to our counterparty. See individual fields for more information.
1422         pub counterparty: ChannelCounterparty,
1423         /// The Channel's funding transaction output, if we've negotiated the funding transaction with
1424         /// our counterparty already.
1425         ///
1426         /// Note that, if this has been set, `channel_id` will be equivalent to
1427         /// `funding_txo.unwrap().to_channel_id()`.
1428         pub funding_txo: Option<OutPoint>,
1429         /// The features which this channel operates with. See individual features for more info.
1430         ///
1431         /// `None` until negotiation completes and the channel type is finalized.
1432         pub channel_type: Option<ChannelTypeFeatures>,
1433         /// The position of the funding transaction in the chain. None if the funding transaction has
1434         /// not yet been confirmed and the channel fully opened.
1435         ///
1436         /// Note that if [`inbound_scid_alias`] is set, it must be used for invoices and inbound
1437         /// payments instead of this. See [`get_inbound_payment_scid`].
1438         ///
1439         /// For channels with [`confirmations_required`] set to `Some(0)`, [`outbound_scid_alias`] may
1440         /// be used in place of this in outbound routes. See [`get_outbound_payment_scid`].
1441         ///
1442         /// [`inbound_scid_alias`]: Self::inbound_scid_alias
1443         /// [`outbound_scid_alias`]: Self::outbound_scid_alias
1444         /// [`get_inbound_payment_scid`]: Self::get_inbound_payment_scid
1445         /// [`get_outbound_payment_scid`]: Self::get_outbound_payment_scid
1446         /// [`confirmations_required`]: Self::confirmations_required
1447         pub short_channel_id: Option<u64>,
1448         /// An optional [`short_channel_id`] alias for this channel, randomly generated by us and
1449         /// usable in place of [`short_channel_id`] to reference the channel in outbound routes when
1450         /// the channel has not yet been confirmed (as long as [`confirmations_required`] is
1451         /// `Some(0)`).
1452         ///
1453         /// This will be `None` as long as the channel is not available for routing outbound payments.
1454         ///
1455         /// [`short_channel_id`]: Self::short_channel_id
1456         /// [`confirmations_required`]: Self::confirmations_required
1457         pub outbound_scid_alias: Option<u64>,
1458         /// An optional [`short_channel_id`] alias for this channel, randomly generated by our
1459         /// counterparty and usable in place of [`short_channel_id`] in invoice route hints. Our
1460         /// counterparty will recognize the alias provided here in place of the [`short_channel_id`]
1461         /// when they see a payment to be routed to us.
1462         ///
1463         /// Our counterparty may choose to rotate this value at any time, though will always recognize
1464         /// previous values for inbound payment forwarding.
1465         ///
1466         /// [`short_channel_id`]: Self::short_channel_id
1467         pub inbound_scid_alias: Option<u64>,
1468         /// The value, in satoshis, of this channel as appears in the funding output
1469         pub channel_value_satoshis: u64,
1470         /// The value, in satoshis, that must always be held in the channel for us. This value ensures
1471         /// that if we broadcast a revoked state, our counterparty can punish us by claiming at least
1472         /// this value on chain.
1473         ///
1474         /// This value is not included in [`outbound_capacity_msat`] as it can never be spent.
1475         ///
1476         /// This value will be `None` for outbound channels until the counterparty accepts the channel.
1477         ///
1478         /// [`outbound_capacity_msat`]: ChannelDetails::outbound_capacity_msat
1479         pub unspendable_punishment_reserve: Option<u64>,
1480         /// The `user_channel_id` value passed in to [`ChannelManager::create_channel`] for outbound
1481         /// channels, or to [`ChannelManager::accept_inbound_channel`] for inbound channels if
1482         /// [`UserConfig::manually_accept_inbound_channels`] config flag is set to true. Otherwise
1483         /// `user_channel_id` will be randomized for an inbound channel.  This may be zero for objects
1484         /// serialized with LDK versions prior to 0.0.113.
1485         ///
1486         /// [`ChannelManager::create_channel`]: crate::ln::channelmanager::ChannelManager::create_channel
1487         /// [`ChannelManager::accept_inbound_channel`]: crate::ln::channelmanager::ChannelManager::accept_inbound_channel
1488         /// [`UserConfig::manually_accept_inbound_channels`]: crate::util::config::UserConfig::manually_accept_inbound_channels
1489         pub user_channel_id: u128,
1490         /// The currently negotiated fee rate denominated in satoshi per 1000 weight units,
1491         /// which is applied to commitment and HTLC transactions.
1492         ///
1493         /// This value will be `None` for objects serialized with LDK versions prior to 0.0.115.
1494         pub feerate_sat_per_1000_weight: Option<u32>,
1495         /// The available outbound capacity for sending HTLCs to the remote peer. This does not include
1496         /// any pending HTLCs which are not yet fully resolved (and, thus, whose balance is not
1497         /// available for inclusion in new outbound HTLCs). This further does not include any pending
1498         /// outgoing HTLCs which are awaiting some other resolution to be sent.
1499         ///
1500         /// This value is not exact. Due to various in-flight changes, feerate changes, and our
1501         /// conflict-avoidance policy, exactly this amount is not likely to be spendable. However, we
1502         /// should be able to spend nearly this amount.
1503         pub outbound_capacity_msat: u64,
1504         /// The available outbound capacity for sending a single HTLC to the remote peer. This is
1505         /// similar to [`ChannelDetails::outbound_capacity_msat`] but it may be further restricted by
1506         /// the current state and per-HTLC limit(s). This is intended for use when routing, allowing us
1507         /// to use a limit as close as possible to the HTLC limit we can currently send.
1508         ///
1509         /// See also [`ChannelDetails::next_outbound_htlc_minimum_msat`] and
1510         /// [`ChannelDetails::outbound_capacity_msat`].
1511         pub next_outbound_htlc_limit_msat: u64,
1512         /// The minimum value for sending a single HTLC to the remote peer. This is the equivalent of
1513         /// [`ChannelDetails::next_outbound_htlc_limit_msat`] but represents a lower-bound, rather than
1514         /// an upper-bound. This is intended for use when routing, allowing us to ensure we pick a
1515         /// route which is valid.
1516         pub next_outbound_htlc_minimum_msat: u64,
1517         /// The available inbound capacity for the remote peer to send HTLCs to us. This does not
1518         /// include any pending HTLCs which are not yet fully resolved (and, thus, whose balance is not
1519         /// available for inclusion in new inbound HTLCs).
1520         /// Note that there are some corner cases not fully handled here, so the actual available
1521         /// inbound capacity may be slightly higher than this.
1522         ///
1523         /// This value is not exact. Due to various in-flight changes, feerate changes, and our
1524         /// counterparty's conflict-avoidance policy, exactly this amount is not likely to be spendable.
1525         /// However, our counterparty should be able to spend nearly this amount.
1526         pub inbound_capacity_msat: u64,
1527         /// The number of required confirmations on the funding transaction before the funding will be
1528         /// considered "locked". This number is selected by the channel fundee (i.e. us if
1529         /// [`is_outbound`] is *not* set), and can be selected for inbound channels with
1530         /// [`ChannelHandshakeConfig::minimum_depth`] or limited for outbound channels with
1531         /// [`ChannelHandshakeLimits::max_minimum_depth`].
1532         ///
1533         /// This value will be `None` for outbound channels until the counterparty accepts the channel.
1534         ///
1535         /// [`is_outbound`]: ChannelDetails::is_outbound
1536         /// [`ChannelHandshakeConfig::minimum_depth`]: crate::util::config::ChannelHandshakeConfig::minimum_depth
1537         /// [`ChannelHandshakeLimits::max_minimum_depth`]: crate::util::config::ChannelHandshakeLimits::max_minimum_depth
1538         pub confirmations_required: Option<u32>,
1539         /// The current number of confirmations on the funding transaction.
1540         ///
1541         /// This value will be `None` for objects serialized with LDK versions prior to 0.0.113.
1542         pub confirmations: Option<u32>,
1543         /// The number of blocks (after our commitment transaction confirms) that we will need to wait
1544         /// until we can claim our funds after we force-close the channel. During this time our
1545         /// counterparty is allowed to punish us if we broadcasted a stale state. If our counterparty
1546         /// force-closes the channel and broadcasts a commitment transaction we do not have to wait any
1547         /// time to claim our non-HTLC-encumbered funds.
1548         ///
1549         /// This value will be `None` for outbound channels until the counterparty accepts the channel.
1550         pub force_close_spend_delay: Option<u16>,
1551         /// True if the channel was initiated (and thus funded) by us.
1552         pub is_outbound: bool,
1553         /// True if the channel is confirmed, channel_ready messages have been exchanged, and the
1554         /// channel is not currently being shut down. `channel_ready` message exchange implies the
1555         /// required confirmation count has been reached (and we were connected to the peer at some
1556         /// point after the funding transaction received enough confirmations). The required
1557         /// confirmation count is provided in [`confirmations_required`].
1558         ///
1559         /// [`confirmations_required`]: ChannelDetails::confirmations_required
1560         pub is_channel_ready: bool,
1561         /// The stage of the channel's shutdown.
1562         /// `None` for `ChannelDetails` serialized on LDK versions prior to 0.0.116.
1563         pub channel_shutdown_state: Option<ChannelShutdownState>,
1564         /// True if the channel is (a) confirmed and channel_ready messages have been exchanged, (b)
1565         /// the peer is connected, and (c) the channel is not currently negotiating a shutdown.
1566         ///
1567         /// This is a strict superset of `is_channel_ready`.
1568         pub is_usable: bool,
1569         /// True if this channel is (or will be) publicly-announced.
1570         pub is_public: bool,
1571         /// The smallest value HTLC (in msat) we will accept, for this channel. This field
1572         /// is only `None` for `ChannelDetails` objects serialized prior to LDK 0.0.107
1573         pub inbound_htlc_minimum_msat: Option<u64>,
1574         /// The largest value HTLC (in msat) we currently will accept, for this channel.
1575         pub inbound_htlc_maximum_msat: Option<u64>,
1576         /// Set of configurable parameters that affect channel operation.
1577         ///
1578         /// This field is only `None` for `ChannelDetails` objects serialized prior to LDK 0.0.109.
1579         pub config: Option<ChannelConfig>,
1580 }
1581
1582 impl ChannelDetails {
1583         /// Gets the current SCID which should be used to identify this channel for inbound payments.
1584         /// This should be used for providing invoice hints or in any other context where our
1585         /// counterparty will forward a payment to us.
1586         ///
1587         /// This is either the [`ChannelDetails::inbound_scid_alias`], if set, or the
1588         /// [`ChannelDetails::short_channel_id`]. See those for more information.
1589         pub fn get_inbound_payment_scid(&self) -> Option<u64> {
1590                 self.inbound_scid_alias.or(self.short_channel_id)
1591         }
1592
1593         /// Gets the current SCID which should be used to identify this channel for outbound payments.
1594         /// This should be used in [`Route`]s to describe the first hop or in other contexts where
1595         /// we're sending or forwarding a payment outbound over this channel.
1596         ///
1597         /// This is either the [`ChannelDetails::short_channel_id`], if set, or the
1598         /// [`ChannelDetails::outbound_scid_alias`]. See those for more information.
1599         pub fn get_outbound_payment_scid(&self) -> Option<u64> {
1600                 self.short_channel_id.or(self.outbound_scid_alias)
1601         }
1602
1603         fn from_channel_context<SP: Deref, F: Deref>(
1604                 context: &ChannelContext<SP>, best_block_height: u32, latest_features: InitFeatures,
1605                 fee_estimator: &LowerBoundedFeeEstimator<F>
1606         ) -> Self
1607         where
1608                 SP::Target: SignerProvider,
1609                 F::Target: FeeEstimator
1610         {
1611                 let balance = context.get_available_balances(fee_estimator);
1612                 let (to_remote_reserve_satoshis, to_self_reserve_satoshis) =
1613                         context.get_holder_counterparty_selected_channel_reserve_satoshis();
1614                 ChannelDetails {
1615                         channel_id: context.channel_id(),
1616                         counterparty: ChannelCounterparty {
1617                                 node_id: context.get_counterparty_node_id(),
1618                                 features: latest_features,
1619                                 unspendable_punishment_reserve: to_remote_reserve_satoshis,
1620                                 forwarding_info: context.counterparty_forwarding_info(),
1621                                 // Ensures that we have actually received the `htlc_minimum_msat` value
1622                                 // from the counterparty through the `OpenChannel` or `AcceptChannel`
1623                                 // message (as they are always the first message from the counterparty).
1624                                 // Else `Channel::get_counterparty_htlc_minimum_msat` could return the
1625                                 // default `0` value set by `Channel::new_outbound`.
1626                                 outbound_htlc_minimum_msat: if context.have_received_message() {
1627                                         Some(context.get_counterparty_htlc_minimum_msat()) } else { None },
1628                                 outbound_htlc_maximum_msat: context.get_counterparty_htlc_maximum_msat(),
1629                         },
1630                         funding_txo: context.get_funding_txo(),
1631                         // Note that accept_channel (or open_channel) is always the first message, so
1632                         // `have_received_message` indicates that type negotiation has completed.
1633                         channel_type: if context.have_received_message() { Some(context.get_channel_type().clone()) } else { None },
1634                         short_channel_id: context.get_short_channel_id(),
1635                         outbound_scid_alias: if context.is_usable() { Some(context.outbound_scid_alias()) } else { None },
1636                         inbound_scid_alias: context.latest_inbound_scid_alias(),
1637                         channel_value_satoshis: context.get_value_satoshis(),
1638                         feerate_sat_per_1000_weight: Some(context.get_feerate_sat_per_1000_weight()),
1639                         unspendable_punishment_reserve: to_self_reserve_satoshis,
1640                         inbound_capacity_msat: balance.inbound_capacity_msat,
1641                         outbound_capacity_msat: balance.outbound_capacity_msat,
1642                         next_outbound_htlc_limit_msat: balance.next_outbound_htlc_limit_msat,
1643                         next_outbound_htlc_minimum_msat: balance.next_outbound_htlc_minimum_msat,
1644                         user_channel_id: context.get_user_id(),
1645                         confirmations_required: context.minimum_depth(),
1646                         confirmations: Some(context.get_funding_tx_confirmations(best_block_height)),
1647                         force_close_spend_delay: context.get_counterparty_selected_contest_delay(),
1648                         is_outbound: context.is_outbound(),
1649                         is_channel_ready: context.is_usable(),
1650                         is_usable: context.is_live(),
1651                         is_public: context.should_announce(),
1652                         inbound_htlc_minimum_msat: Some(context.get_holder_htlc_minimum_msat()),
1653                         inbound_htlc_maximum_msat: context.get_holder_htlc_maximum_msat(),
1654                         config: Some(context.config()),
1655                         channel_shutdown_state: Some(context.shutdown_state()),
1656                 }
1657         }
1658 }
1659
1660 #[derive(Clone, Copy, Debug, PartialEq, Eq)]
1661 /// Further information on the details of the channel shutdown.
1662 /// Upon channels being forced closed (i.e. commitment transaction confirmation detected
1663 /// by `ChainMonitor`), ChannelShutdownState will be set to `ShutdownComplete` or
1664 /// the channel will be removed shortly.
1665 /// Also note, that in normal operation, peers could disconnect at any of these states
1666 /// and require peer re-connection before making progress onto other states
1667 pub enum ChannelShutdownState {
1668         /// Channel has not sent or received a shutdown message.
1669         NotShuttingDown,
1670         /// Local node has sent a shutdown message for this channel.
1671         ShutdownInitiated,
1672         /// Shutdown message exchanges have concluded and the channels are in the midst of
1673         /// resolving all existing open HTLCs before closing can continue.
1674         ResolvingHTLCs,
1675         /// All HTLCs have been resolved, nodes are currently negotiating channel close onchain fee rates.
1676         NegotiatingClosingFee,
1677         /// We've successfully negotiated a closing_signed dance. At this point `ChannelManager` is about
1678         /// to drop the channel.
1679         ShutdownComplete,
1680 }
1681
1682 /// Used by [`ChannelManager::list_recent_payments`] to express the status of recent payments.
1683 /// These include payments that have yet to find a successful path, or have unresolved HTLCs.
1684 #[derive(Debug, PartialEq)]
1685 pub enum RecentPaymentDetails {
1686         /// When an invoice was requested and thus a payment has not yet been sent.
1687         AwaitingInvoice {
1688                 /// Identifier for the payment to ensure idempotency.
1689                 payment_id: PaymentId,
1690         },
1691         /// When a payment is still being sent and awaiting successful delivery.
1692         Pending {
1693                 /// Hash of the payment that is currently being sent but has yet to be fulfilled or
1694                 /// abandoned.
1695                 payment_hash: PaymentHash,
1696                 /// Total amount (in msat, excluding fees) across all paths for this payment,
1697                 /// not just the amount currently inflight.
1698                 total_msat: u64,
1699         },
1700         /// When a pending payment is fulfilled, we continue tracking it until all pending HTLCs have
1701         /// been resolved. Upon receiving [`Event::PaymentSent`], we delay for a few minutes before the
1702         /// payment is removed from tracking.
1703         Fulfilled {
1704                 /// Hash of the payment that was claimed. `None` for serializations of [`ChannelManager`]
1705                 /// made before LDK version 0.0.104.
1706                 payment_hash: Option<PaymentHash>,
1707         },
1708         /// After a payment's retries are exhausted per the provided [`Retry`], or it is explicitly
1709         /// abandoned via [`ChannelManager::abandon_payment`], it is marked as abandoned until all
1710         /// pending HTLCs for this payment resolve and an [`Event::PaymentFailed`] is generated.
1711         Abandoned {
1712                 /// Hash of the payment that we have given up trying to send.
1713                 payment_hash: PaymentHash,
1714         },
1715 }
1716
1717 /// Route hints used in constructing invoices for [phantom node payents].
1718 ///
1719 /// [phantom node payments]: crate::sign::PhantomKeysManager
1720 #[derive(Clone)]
1721 pub struct PhantomRouteHints {
1722         /// The list of channels to be included in the invoice route hints.
1723         pub channels: Vec<ChannelDetails>,
1724         /// A fake scid used for representing the phantom node's fake channel in generating the invoice
1725         /// route hints.
1726         pub phantom_scid: u64,
1727         /// The pubkey of the real backing node that would ultimately receive the payment.
1728         pub real_node_pubkey: PublicKey,
1729 }
1730
1731 macro_rules! handle_error {
1732         ($self: ident, $internal: expr, $counterparty_node_id: expr) => { {
1733                 // In testing, ensure there are no deadlocks where the lock is already held upon
1734                 // entering the macro.
1735                 debug_assert_ne!($self.pending_events.held_by_thread(), LockHeldState::HeldByThread);
1736                 debug_assert_ne!($self.per_peer_state.held_by_thread(), LockHeldState::HeldByThread);
1737
1738                 match $internal {
1739                         Ok(msg) => Ok(msg),
1740                         Err(MsgHandleErrInternal { err, chan_id, shutdown_finish, channel_capacity }) => {
1741                                 let mut msg_events = Vec::with_capacity(2);
1742
1743                                 if let Some((shutdown_res, update_option)) = shutdown_finish {
1744                                         $self.finish_force_close_channel(shutdown_res);
1745                                         if let Some(update) = update_option {
1746                                                 msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
1747                                                         msg: update
1748                                                 });
1749                                         }
1750                                         if let Some((channel_id, user_channel_id)) = chan_id {
1751                                                 $self.pending_events.lock().unwrap().push_back((events::Event::ChannelClosed {
1752                                                         channel_id, user_channel_id,
1753                                                         reason: ClosureReason::ProcessingError { err: err.err.clone() },
1754                                                         counterparty_node_id: Some($counterparty_node_id),
1755                                                         channel_capacity_sats: channel_capacity,
1756                                                 }, None));
1757                                         }
1758                                 }
1759
1760                                 log_error!($self.logger, "{}", err.err);
1761                                 if let msgs::ErrorAction::IgnoreError = err.action {
1762                                 } else {
1763                                         msg_events.push(events::MessageSendEvent::HandleError {
1764                                                 node_id: $counterparty_node_id,
1765                                                 action: err.action.clone()
1766                                         });
1767                                 }
1768
1769                                 if !msg_events.is_empty() {
1770                                         let per_peer_state = $self.per_peer_state.read().unwrap();
1771                                         if let Some(peer_state_mutex) = per_peer_state.get(&$counterparty_node_id) {
1772                                                 let mut peer_state = peer_state_mutex.lock().unwrap();
1773                                                 peer_state.pending_msg_events.append(&mut msg_events);
1774                                         }
1775                                 }
1776
1777                                 // Return error in case higher-API need one
1778                                 Err(err)
1779                         },
1780                 }
1781         } };
1782         ($self: ident, $internal: expr) => {
1783                 match $internal {
1784                         Ok(res) => Ok(res),
1785                         Err((chan, msg_handle_err)) => {
1786                                 let counterparty_node_id = chan.get_counterparty_node_id();
1787                                 handle_error!($self, Err(msg_handle_err), counterparty_node_id).map_err(|err| (chan, err))
1788                         },
1789                 }
1790         };
1791 }
1792
1793 macro_rules! update_maps_on_chan_removal {
1794         ($self: expr, $channel_context: expr) => {{
1795                 $self.id_to_peer.lock().unwrap().remove(&$channel_context.channel_id());
1796                 let mut short_to_chan_info = $self.short_to_chan_info.write().unwrap();
1797                 if let Some(short_id) = $channel_context.get_short_channel_id() {
1798                         short_to_chan_info.remove(&short_id);
1799                 } else {
1800                         // If the channel was never confirmed on-chain prior to its closure, remove the
1801                         // outbound SCID alias we used for it from the collision-prevention set. While we
1802                         // generally want to avoid ever re-using an outbound SCID alias across all channels, we
1803                         // also don't want a counterparty to be able to trivially cause a memory leak by simply
1804                         // opening a million channels with us which are closed before we ever reach the funding
1805                         // stage.
1806                         let alias_removed = $self.outbound_scid_aliases.lock().unwrap().remove(&$channel_context.outbound_scid_alias());
1807                         debug_assert!(alias_removed);
1808                 }
1809                 short_to_chan_info.remove(&$channel_context.outbound_scid_alias());
1810         }}
1811 }
1812
1813 /// Returns (boolean indicating if we should remove the Channel object from memory, a mapped error)
1814 macro_rules! convert_chan_err {
1815         ($self: ident, $err: expr, $channel: expr, $channel_id: expr) => {
1816                 match $err {
1817                         ChannelError::Warn(msg) => {
1818                                 (false, MsgHandleErrInternal::from_chan_no_close(ChannelError::Warn(msg), $channel_id.clone()))
1819                         },
1820                         ChannelError::Ignore(msg) => {
1821                                 (false, MsgHandleErrInternal::from_chan_no_close(ChannelError::Ignore(msg), $channel_id.clone()))
1822                         },
1823                         ChannelError::Close(msg) => {
1824                                 log_error!($self.logger, "Closing channel {} due to close-required error: {}", &$channel_id, msg);
1825                                 update_maps_on_chan_removal!($self, &$channel.context);
1826                                 let shutdown_res = $channel.context.force_shutdown(true);
1827                                 (true, MsgHandleErrInternal::from_finish_shutdown(msg, *$channel_id, $channel.context.get_user_id(),
1828                                         shutdown_res, $self.get_channel_update_for_broadcast(&$channel).ok(), $channel.context.get_value_satoshis()))
1829                         },
1830                 }
1831         };
1832         ($self: ident, $err: expr, $channel_context: expr, $channel_id: expr, UNFUNDED) => {
1833                 match $err {
1834                         // We should only ever have `ChannelError::Close` when unfunded channels error.
1835                         // In any case, just close the channel.
1836                         ChannelError::Warn(msg) | ChannelError::Ignore(msg) | ChannelError::Close(msg) => {
1837                                 log_error!($self.logger, "Closing unfunded channel {} due to an error: {}", &$channel_id, msg);
1838                                 update_maps_on_chan_removal!($self, &$channel_context);
1839                                 let shutdown_res = $channel_context.force_shutdown(false);
1840                                 (true, MsgHandleErrInternal::from_finish_shutdown(msg, *$channel_id, $channel_context.get_user_id(),
1841                                         shutdown_res, None, $channel_context.get_value_satoshis()))
1842                         },
1843                 }
1844         }
1845 }
1846
1847 macro_rules! break_chan_entry {
1848         ($self: ident, $res: expr, $entry: expr) => {
1849                 match $res {
1850                         Ok(res) => res,
1851                         Err(e) => {
1852                                 let (drop, res) = convert_chan_err!($self, e, $entry.get_mut(), $entry.key());
1853                                 if drop {
1854                                         $entry.remove_entry();
1855                                 }
1856                                 break Err(res);
1857                         }
1858                 }
1859         }
1860 }
1861
1862 macro_rules! try_v1_outbound_chan_entry {
1863         ($self: ident, $res: expr, $entry: expr) => {
1864                 match $res {
1865                         Ok(res) => res,
1866                         Err(e) => {
1867                                 let (drop, res) = convert_chan_err!($self, e, $entry.get_mut().context, $entry.key(), UNFUNDED);
1868                                 if drop {
1869                                         $entry.remove_entry();
1870                                 }
1871                                 return Err(res);
1872                         }
1873                 }
1874         }
1875 }
1876
1877 macro_rules! try_chan_entry {
1878         ($self: ident, $res: expr, $entry: expr) => {
1879                 match $res {
1880                         Ok(res) => res,
1881                         Err(e) => {
1882                                 let (drop, res) = convert_chan_err!($self, e, $entry.get_mut(), $entry.key());
1883                                 if drop {
1884                                         $entry.remove_entry();
1885                                 }
1886                                 return Err(res);
1887                         }
1888                 }
1889         }
1890 }
1891
1892 macro_rules! remove_channel {
1893         ($self: expr, $entry: expr) => {
1894                 {
1895                         let channel = $entry.remove_entry().1;
1896                         update_maps_on_chan_removal!($self, &channel.context);
1897                         channel
1898                 }
1899         }
1900 }
1901
1902 macro_rules! send_channel_ready {
1903         ($self: ident, $pending_msg_events: expr, $channel: expr, $channel_ready_msg: expr) => {{
1904                 $pending_msg_events.push(events::MessageSendEvent::SendChannelReady {
1905                         node_id: $channel.context.get_counterparty_node_id(),
1906                         msg: $channel_ready_msg,
1907                 });
1908                 // Note that we may send a `channel_ready` multiple times for a channel if we reconnect, so
1909                 // we allow collisions, but we shouldn't ever be updating the channel ID pointed to.
1910                 let mut short_to_chan_info = $self.short_to_chan_info.write().unwrap();
1911                 let outbound_alias_insert = short_to_chan_info.insert($channel.context.outbound_scid_alias(), ($channel.context.get_counterparty_node_id(), $channel.context.channel_id()));
1912                 assert!(outbound_alias_insert.is_none() || outbound_alias_insert.unwrap() == ($channel.context.get_counterparty_node_id(), $channel.context.channel_id()),
1913                         "SCIDs should never collide - ensure you weren't behind the chain tip by a full month when creating channels");
1914                 if let Some(real_scid) = $channel.context.get_short_channel_id() {
1915                         let scid_insert = short_to_chan_info.insert(real_scid, ($channel.context.get_counterparty_node_id(), $channel.context.channel_id()));
1916                         assert!(scid_insert.is_none() || scid_insert.unwrap() == ($channel.context.get_counterparty_node_id(), $channel.context.channel_id()),
1917                                 "SCIDs should never collide - ensure you weren't behind the chain tip by a full month when creating channels");
1918                 }
1919         }}
1920 }
1921
1922 macro_rules! emit_channel_pending_event {
1923         ($locked_events: expr, $channel: expr) => {
1924                 if $channel.context.should_emit_channel_pending_event() {
1925                         $locked_events.push_back((events::Event::ChannelPending {
1926                                 channel_id: $channel.context.channel_id(),
1927                                 former_temporary_channel_id: $channel.context.temporary_channel_id(),
1928                                 counterparty_node_id: $channel.context.get_counterparty_node_id(),
1929                                 user_channel_id: $channel.context.get_user_id(),
1930                                 funding_txo: $channel.context.get_funding_txo().unwrap().into_bitcoin_outpoint(),
1931                         }, None));
1932                         $channel.context.set_channel_pending_event_emitted();
1933                 }
1934         }
1935 }
1936
1937 macro_rules! emit_channel_ready_event {
1938         ($locked_events: expr, $channel: expr) => {
1939                 if $channel.context.should_emit_channel_ready_event() {
1940                         debug_assert!($channel.context.channel_pending_event_emitted());
1941                         $locked_events.push_back((events::Event::ChannelReady {
1942                                 channel_id: $channel.context.channel_id(),
1943                                 user_channel_id: $channel.context.get_user_id(),
1944                                 counterparty_node_id: $channel.context.get_counterparty_node_id(),
1945                                 channel_type: $channel.context.get_channel_type().clone(),
1946                         }, None));
1947                         $channel.context.set_channel_ready_event_emitted();
1948                 }
1949         }
1950 }
1951
1952 macro_rules! handle_monitor_update_completion {
1953         ($self: ident, $peer_state_lock: expr, $peer_state: expr, $per_peer_state_lock: expr, $chan: expr) => { {
1954                 let mut updates = $chan.monitor_updating_restored(&$self.logger,
1955                         &$self.node_signer, $self.genesis_hash, &$self.default_configuration,
1956                         $self.best_block.read().unwrap().height());
1957                 let counterparty_node_id = $chan.context.get_counterparty_node_id();
1958                 let channel_update = if updates.channel_ready.is_some() && $chan.context.is_usable() {
1959                         // We only send a channel_update in the case where we are just now sending a
1960                         // channel_ready and the channel is in a usable state. We may re-send a
1961                         // channel_update later through the announcement_signatures process for public
1962                         // channels, but there's no reason not to just inform our counterparty of our fees
1963                         // now.
1964                         if let Ok(msg) = $self.get_channel_update_for_unicast($chan) {
1965                                 Some(events::MessageSendEvent::SendChannelUpdate {
1966                                         node_id: counterparty_node_id,
1967                                         msg,
1968                                 })
1969                         } else { None }
1970                 } else { None };
1971
1972                 let update_actions = $peer_state.monitor_update_blocked_actions
1973                         .remove(&$chan.context.channel_id()).unwrap_or(Vec::new());
1974
1975                 let htlc_forwards = $self.handle_channel_resumption(
1976                         &mut $peer_state.pending_msg_events, $chan, updates.raa,
1977                         updates.commitment_update, updates.order, updates.accepted_htlcs,
1978                         updates.funding_broadcastable, updates.channel_ready,
1979                         updates.announcement_sigs);
1980                 if let Some(upd) = channel_update {
1981                         $peer_state.pending_msg_events.push(upd);
1982                 }
1983
1984                 let channel_id = $chan.context.channel_id();
1985                 core::mem::drop($peer_state_lock);
1986                 core::mem::drop($per_peer_state_lock);
1987
1988                 $self.handle_monitor_update_completion_actions(update_actions);
1989
1990                 if let Some(forwards) = htlc_forwards {
1991                         $self.forward_htlcs(&mut [forwards][..]);
1992                 }
1993                 $self.finalize_claims(updates.finalized_claimed_htlcs);
1994                 for failure in updates.failed_htlcs.drain(..) {
1995                         let receiver = HTLCDestination::NextHopChannel { node_id: Some(counterparty_node_id), channel_id };
1996                         $self.fail_htlc_backwards_internal(&failure.0, &failure.1, &failure.2, receiver);
1997                 }
1998         } }
1999 }
2000
2001 macro_rules! handle_new_monitor_update {
2002         ($self: ident, $update_res: expr, $peer_state_lock: expr, $peer_state: expr, $per_peer_state_lock: expr, $chan: expr, _internal, $remove: expr, $completed: expr) => { {
2003                 // update_maps_on_chan_removal needs to be able to take id_to_peer, so make sure we can in
2004                 // any case so that it won't deadlock.
2005                 debug_assert_ne!($self.id_to_peer.held_by_thread(), LockHeldState::HeldByThread);
2006                 debug_assert!($self.background_events_processed_since_startup.load(Ordering::Acquire));
2007                 match $update_res {
2008                         ChannelMonitorUpdateStatus::InProgress => {
2009                                 log_debug!($self.logger, "ChannelMonitor update for {} in flight, holding messages until the update completes.",
2010                                         &$chan.context.channel_id());
2011                                 Ok(false)
2012                         },
2013                         ChannelMonitorUpdateStatus::PermanentFailure => {
2014                                 log_error!($self.logger, "Closing channel {} due to monitor update ChannelMonitorUpdateStatus::PermanentFailure",
2015                                         &$chan.context.channel_id());
2016                                 update_maps_on_chan_removal!($self, &$chan.context);
2017                                 let res = Err(MsgHandleErrInternal::from_finish_shutdown(
2018                                         "ChannelMonitor storage failure".to_owned(), $chan.context.channel_id(),
2019                                         $chan.context.get_user_id(), $chan.context.force_shutdown(false),
2020                                         $self.get_channel_update_for_broadcast(&$chan).ok(), $chan.context.get_value_satoshis()));
2021                                 $remove;
2022                                 res
2023                         },
2024                         ChannelMonitorUpdateStatus::Completed => {
2025                                 $completed;
2026                                 Ok(true)
2027                         },
2028                 }
2029         } };
2030         ($self: ident, $update_res: expr, $peer_state_lock: expr, $peer_state: expr, $per_peer_state_lock: expr, $chan: expr, MANUALLY_REMOVING_INITIAL_MONITOR, $remove: expr) => {
2031                 handle_new_monitor_update!($self, $update_res, $peer_state_lock, $peer_state,
2032                         $per_peer_state_lock, $chan, _internal, $remove,
2033                         handle_monitor_update_completion!($self, $peer_state_lock, $peer_state, $per_peer_state_lock, $chan))
2034         };
2035         ($self: ident, $update_res: expr, $peer_state_lock: expr, $peer_state: expr, $per_peer_state_lock: expr, $chan_entry: expr, INITIAL_MONITOR) => {
2036                 handle_new_monitor_update!($self, $update_res, $peer_state_lock, $peer_state, $per_peer_state_lock, $chan_entry.get_mut(), MANUALLY_REMOVING_INITIAL_MONITOR, $chan_entry.remove_entry())
2037         };
2038         ($self: ident, $funding_txo: expr, $update: expr, $peer_state_lock: expr, $peer_state: expr, $per_peer_state_lock: expr, $chan: expr, MANUALLY_REMOVING, $remove: expr) => { {
2039                 let in_flight_updates = $peer_state.in_flight_monitor_updates.entry($funding_txo)
2040                         .or_insert_with(Vec::new);
2041                 // During startup, we push monitor updates as background events through to here in
2042                 // order to replay updates that were in-flight when we shut down. Thus, we have to
2043                 // filter for uniqueness here.
2044                 let idx = in_flight_updates.iter().position(|upd| upd == &$update)
2045                         .unwrap_or_else(|| {
2046                                 in_flight_updates.push($update);
2047                                 in_flight_updates.len() - 1
2048                         });
2049                 let update_res = $self.chain_monitor.update_channel($funding_txo, &in_flight_updates[idx]);
2050                 handle_new_monitor_update!($self, update_res, $peer_state_lock, $peer_state,
2051                         $per_peer_state_lock, $chan, _internal, $remove,
2052                         {
2053                                 let _ = in_flight_updates.remove(idx);
2054                                 if in_flight_updates.is_empty() && $chan.blocked_monitor_updates_pending() == 0 {
2055                                         handle_monitor_update_completion!($self, $peer_state_lock, $peer_state, $per_peer_state_lock, $chan);
2056                                 }
2057                         })
2058         } };
2059         ($self: ident, $funding_txo: expr, $update: expr, $peer_state_lock: expr, $peer_state: expr, $per_peer_state_lock: expr, $chan_entry: expr) => {
2060                 handle_new_monitor_update!($self, $funding_txo, $update, $peer_state_lock, $peer_state, $per_peer_state_lock, $chan_entry.get_mut(), MANUALLY_REMOVING, $chan_entry.remove_entry())
2061         }
2062 }
2063
2064 macro_rules! process_events_body {
2065         ($self: expr, $event_to_handle: expr, $handle_event: expr) => {
2066                 let mut processed_all_events = false;
2067                 while !processed_all_events {
2068                         if $self.pending_events_processor.compare_exchange(false, true, Ordering::Acquire, Ordering::Relaxed).is_err() {
2069                                 return;
2070                         }
2071
2072                         let mut result = NotifyOption::SkipPersist;
2073
2074                         {
2075                                 // We'll acquire our total consistency lock so that we can be sure no other
2076                                 // persists happen while processing monitor events.
2077                                 let _read_guard = $self.total_consistency_lock.read().unwrap();
2078
2079                                 // Because `handle_post_event_actions` may send `ChannelMonitorUpdate`s to the user we must
2080                                 // ensure any startup-generated background events are handled first.
2081                                 if $self.process_background_events() == NotifyOption::DoPersist { result = NotifyOption::DoPersist; }
2082
2083                                 // TODO: This behavior should be documented. It's unintuitive that we query
2084                                 // ChannelMonitors when clearing other events.
2085                                 if $self.process_pending_monitor_events() {
2086                                         result = NotifyOption::DoPersist;
2087                                 }
2088                         }
2089
2090                         let pending_events = $self.pending_events.lock().unwrap().clone();
2091                         let num_events = pending_events.len();
2092                         if !pending_events.is_empty() {
2093                                 result = NotifyOption::DoPersist;
2094                         }
2095
2096                         let mut post_event_actions = Vec::new();
2097
2098                         for (event, action_opt) in pending_events {
2099                                 $event_to_handle = event;
2100                                 $handle_event;
2101                                 if let Some(action) = action_opt {
2102                                         post_event_actions.push(action);
2103                                 }
2104                         }
2105
2106                         {
2107                                 let mut pending_events = $self.pending_events.lock().unwrap();
2108                                 pending_events.drain(..num_events);
2109                                 processed_all_events = pending_events.is_empty();
2110                                 // Note that `push_pending_forwards_ev` relies on `pending_events_processor` being
2111                                 // updated here with the `pending_events` lock acquired.
2112                                 $self.pending_events_processor.store(false, Ordering::Release);
2113                         }
2114
2115                         if !post_event_actions.is_empty() {
2116                                 $self.handle_post_event_actions(post_event_actions);
2117                                 // If we had some actions, go around again as we may have more events now
2118                                 processed_all_events = false;
2119                         }
2120
2121                         if result == NotifyOption::DoPersist {
2122                                 $self.persistence_notifier.notify();
2123                         }
2124                 }
2125         }
2126 }
2127
2128 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref> ChannelManager<M, T, ES, NS, SP, F, R, L>
2129 where
2130         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
2131         T::Target: BroadcasterInterface,
2132         ES::Target: EntropySource,
2133         NS::Target: NodeSigner,
2134         SP::Target: SignerProvider,
2135         F::Target: FeeEstimator,
2136         R::Target: Router,
2137         L::Target: Logger,
2138 {
2139         /// Constructs a new `ChannelManager` to hold several channels and route between them.
2140         ///
2141         /// The current time or latest block header time can be provided as the `current_timestamp`.
2142         ///
2143         /// This is the main "logic hub" for all channel-related actions, and implements
2144         /// [`ChannelMessageHandler`].
2145         ///
2146         /// Non-proportional fees are fixed according to our risk using the provided fee estimator.
2147         ///
2148         /// Users need to notify the new `ChannelManager` when a new block is connected or
2149         /// disconnected using its [`block_connected`] and [`block_disconnected`] methods, starting
2150         /// from after [`params.best_block.block_hash`]. See [`chain::Listen`] and [`chain::Confirm`] for
2151         /// more details.
2152         ///
2153         /// [`block_connected`]: chain::Listen::block_connected
2154         /// [`block_disconnected`]: chain::Listen::block_disconnected
2155         /// [`params.best_block.block_hash`]: chain::BestBlock::block_hash
2156         pub fn new(
2157                 fee_est: F, chain_monitor: M, tx_broadcaster: T, router: R, logger: L, entropy_source: ES,
2158                 node_signer: NS, signer_provider: SP, config: UserConfig, params: ChainParameters,
2159                 current_timestamp: u32,
2160         ) -> Self {
2161                 let mut secp_ctx = Secp256k1::new();
2162                 secp_ctx.seeded_randomize(&entropy_source.get_secure_random_bytes());
2163                 let inbound_pmt_key_material = node_signer.get_inbound_payment_key_material();
2164                 let expanded_inbound_key = inbound_payment::ExpandedKey::new(&inbound_pmt_key_material);
2165                 ChannelManager {
2166                         default_configuration: config.clone(),
2167                         genesis_hash: genesis_block(params.network).header.block_hash(),
2168                         fee_estimator: LowerBoundedFeeEstimator::new(fee_est),
2169                         chain_monitor,
2170                         tx_broadcaster,
2171                         router,
2172
2173                         best_block: RwLock::new(params.best_block),
2174
2175                         outbound_scid_aliases: Mutex::new(HashSet::new()),
2176                         pending_inbound_payments: Mutex::new(HashMap::new()),
2177                         pending_outbound_payments: OutboundPayments::new(),
2178                         forward_htlcs: Mutex::new(HashMap::new()),
2179                         claimable_payments: Mutex::new(ClaimablePayments { claimable_payments: HashMap::new(), pending_claiming_payments: HashMap::new() }),
2180                         pending_intercepted_htlcs: Mutex::new(HashMap::new()),
2181                         id_to_peer: Mutex::new(HashMap::new()),
2182                         short_to_chan_info: FairRwLock::new(HashMap::new()),
2183
2184                         our_network_pubkey: node_signer.get_node_id(Recipient::Node).unwrap(),
2185                         secp_ctx,
2186
2187                         inbound_payment_key: expanded_inbound_key,
2188                         fake_scid_rand_bytes: entropy_source.get_secure_random_bytes(),
2189
2190                         probing_cookie_secret: entropy_source.get_secure_random_bytes(),
2191
2192                         highest_seen_timestamp: AtomicUsize::new(current_timestamp as usize),
2193
2194                         per_peer_state: FairRwLock::new(HashMap::new()),
2195
2196                         pending_events: Mutex::new(VecDeque::new()),
2197                         pending_events_processor: AtomicBool::new(false),
2198                         pending_background_events: Mutex::new(Vec::new()),
2199                         total_consistency_lock: RwLock::new(()),
2200                         background_events_processed_since_startup: AtomicBool::new(false),
2201                         persistence_notifier: Notifier::new(),
2202
2203                         entropy_source,
2204                         node_signer,
2205                         signer_provider,
2206
2207                         logger,
2208                 }
2209         }
2210
2211         /// Gets the current configuration applied to all new channels.
2212         pub fn get_current_default_configuration(&self) -> &UserConfig {
2213                 &self.default_configuration
2214         }
2215
2216         fn create_and_insert_outbound_scid_alias(&self) -> u64 {
2217                 let height = self.best_block.read().unwrap().height();
2218                 let mut outbound_scid_alias = 0;
2219                 let mut i = 0;
2220                 loop {
2221                         if cfg!(fuzzing) { // fuzzing chacha20 doesn't use the key at all so we always get the same alias
2222                                 outbound_scid_alias += 1;
2223                         } else {
2224                                 outbound_scid_alias = fake_scid::Namespace::OutboundAlias.get_fake_scid(height, &self.genesis_hash, &self.fake_scid_rand_bytes, &self.entropy_source);
2225                         }
2226                         if outbound_scid_alias != 0 && self.outbound_scid_aliases.lock().unwrap().insert(outbound_scid_alias) {
2227                                 break;
2228                         }
2229                         i += 1;
2230                         if i > 1_000_000 { panic!("Your RNG is busted or we ran out of possible outbound SCID aliases (which should never happen before we run out of memory to store channels"); }
2231                 }
2232                 outbound_scid_alias
2233         }
2234
2235         /// Creates a new outbound channel to the given remote node and with the given value.
2236         ///
2237         /// `user_channel_id` will be provided back as in
2238         /// [`Event::FundingGenerationReady::user_channel_id`] to allow tracking of which events
2239         /// correspond with which `create_channel` call. Note that the `user_channel_id` defaults to a
2240         /// randomized value for inbound channels. `user_channel_id` has no meaning inside of LDK, it
2241         /// is simply copied to events and otherwise ignored.
2242         ///
2243         /// Raises [`APIError::APIMisuseError`] when `channel_value_satoshis` > 2**24 or `push_msat` is
2244         /// greater than `channel_value_satoshis * 1k` or `channel_value_satoshis < 1000`.
2245         ///
2246         /// Raises [`APIError::ChannelUnavailable`] if the channel cannot be opened due to failing to
2247         /// generate a shutdown scriptpubkey or destination script set by
2248         /// [`SignerProvider::get_shutdown_scriptpubkey`] or [`SignerProvider::get_destination_script`].
2249         ///
2250         /// Note that we do not check if you are currently connected to the given peer. If no
2251         /// connection is available, the outbound `open_channel` message may fail to send, resulting in
2252         /// the channel eventually being silently forgotten (dropped on reload).
2253         ///
2254         /// Returns the new Channel's temporary `channel_id`. This ID will appear as
2255         /// [`Event::FundingGenerationReady::temporary_channel_id`] and in
2256         /// [`ChannelDetails::channel_id`] until after
2257         /// [`ChannelManager::funding_transaction_generated`] is called, swapping the Channel's ID for
2258         /// one derived from the funding transaction's TXID. If the counterparty rejects the channel
2259         /// immediately, this temporary ID will appear in [`Event::ChannelClosed::channel_id`].
2260         ///
2261         /// [`Event::FundingGenerationReady::user_channel_id`]: events::Event::FundingGenerationReady::user_channel_id
2262         /// [`Event::FundingGenerationReady::temporary_channel_id`]: events::Event::FundingGenerationReady::temporary_channel_id
2263         /// [`Event::ChannelClosed::channel_id`]: events::Event::ChannelClosed::channel_id
2264         pub fn create_channel(&self, their_network_key: PublicKey, channel_value_satoshis: u64, push_msat: u64, user_channel_id: u128, override_config: Option<UserConfig>) -> Result<ChannelId, APIError> {
2265                 if channel_value_satoshis < 1000 {
2266                         return Err(APIError::APIMisuseError { err: format!("Channel value must be at least 1000 satoshis. It was {}", channel_value_satoshis) });
2267                 }
2268
2269                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
2270                 // We want to make sure the lock is actually acquired by PersistenceNotifierGuard.
2271                 debug_assert!(&self.total_consistency_lock.try_write().is_err());
2272
2273                 let per_peer_state = self.per_peer_state.read().unwrap();
2274
2275                 let peer_state_mutex = per_peer_state.get(&their_network_key)
2276                         .ok_or_else(|| APIError::APIMisuseError{ err: format!("Not connected to node: {}", their_network_key) })?;
2277
2278                 let mut peer_state = peer_state_mutex.lock().unwrap();
2279                 let channel = {
2280                         let outbound_scid_alias = self.create_and_insert_outbound_scid_alias();
2281                         let their_features = &peer_state.latest_features;
2282                         let config = if override_config.is_some() { override_config.as_ref().unwrap() } else { &self.default_configuration };
2283                         match OutboundV1Channel::new(&self.fee_estimator, &self.entropy_source, &self.signer_provider, their_network_key,
2284                                 their_features, channel_value_satoshis, push_msat, user_channel_id, config,
2285                                 self.best_block.read().unwrap().height(), outbound_scid_alias)
2286                         {
2287                                 Ok(res) => res,
2288                                 Err(e) => {
2289                                         self.outbound_scid_aliases.lock().unwrap().remove(&outbound_scid_alias);
2290                                         return Err(e);
2291                                 },
2292                         }
2293                 };
2294                 let res = channel.get_open_channel(self.genesis_hash.clone());
2295
2296                 let temporary_channel_id = channel.context.channel_id();
2297                 match peer_state.outbound_v1_channel_by_id.entry(temporary_channel_id) {
2298                         hash_map::Entry::Occupied(_) => {
2299                                 if cfg!(fuzzing) {
2300                                         return Err(APIError::APIMisuseError { err: "Fuzzy bad RNG".to_owned() });
2301                                 } else {
2302                                         panic!("RNG is bad???");
2303                                 }
2304                         },
2305                         hash_map::Entry::Vacant(entry) => { entry.insert(channel); }
2306                 }
2307
2308                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendOpenChannel {
2309                         node_id: their_network_key,
2310                         msg: res,
2311                 });
2312                 Ok(temporary_channel_id)
2313         }
2314
2315         fn list_funded_channels_with_filter<Fn: FnMut(&(&ChannelId, &Channel<SP>)) -> bool + Copy>(&self, f: Fn) -> Vec<ChannelDetails> {
2316                 // Allocate our best estimate of the number of channels we have in the `res`
2317                 // Vec. Sadly the `short_to_chan_info` map doesn't cover channels without
2318                 // a scid or a scid alias, and the `id_to_peer` shouldn't be used outside
2319                 // of the ChannelMonitor handling. Therefore reallocations may still occur, but is
2320                 // unlikely as the `short_to_chan_info` map often contains 2 entries for
2321                 // the same channel.
2322                 let mut res = Vec::with_capacity(self.short_to_chan_info.read().unwrap().len());
2323                 {
2324                         let best_block_height = self.best_block.read().unwrap().height();
2325                         let per_peer_state = self.per_peer_state.read().unwrap();
2326                         for (_cp_id, peer_state_mutex) in per_peer_state.iter() {
2327                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
2328                                 let peer_state = &mut *peer_state_lock;
2329                                 // Only `Channels` in the channel_by_id map can be considered funded.
2330                                 for (_channel_id, channel) in peer_state.channel_by_id.iter().filter(f) {
2331                                         let details = ChannelDetails::from_channel_context(&channel.context, best_block_height,
2332                                                 peer_state.latest_features.clone(), &self.fee_estimator);
2333                                         res.push(details);
2334                                 }
2335                         }
2336                 }
2337                 res
2338         }
2339
2340         /// Gets the list of open channels, in random order. See [`ChannelDetails`] field documentation for
2341         /// more information.
2342         pub fn list_channels(&self) -> Vec<ChannelDetails> {
2343                 // Allocate our best estimate of the number of channels we have in the `res`
2344                 // Vec. Sadly the `short_to_chan_info` map doesn't cover channels without
2345                 // a scid or a scid alias, and the `id_to_peer` shouldn't be used outside
2346                 // of the ChannelMonitor handling. Therefore reallocations may still occur, but is
2347                 // unlikely as the `short_to_chan_info` map often contains 2 entries for
2348                 // the same channel.
2349                 let mut res = Vec::with_capacity(self.short_to_chan_info.read().unwrap().len());
2350                 {
2351                         let best_block_height = self.best_block.read().unwrap().height();
2352                         let per_peer_state = self.per_peer_state.read().unwrap();
2353                         for (_cp_id, peer_state_mutex) in per_peer_state.iter() {
2354                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
2355                                 let peer_state = &mut *peer_state_lock;
2356                                 for (_channel_id, channel) in peer_state.channel_by_id.iter() {
2357                                         let details = ChannelDetails::from_channel_context(&channel.context, best_block_height,
2358                                                 peer_state.latest_features.clone(), &self.fee_estimator);
2359                                         res.push(details);
2360                                 }
2361                                 for (_channel_id, channel) in peer_state.inbound_v1_channel_by_id.iter() {
2362                                         let details = ChannelDetails::from_channel_context(&channel.context, best_block_height,
2363                                                 peer_state.latest_features.clone(), &self.fee_estimator);
2364                                         res.push(details);
2365                                 }
2366                                 for (_channel_id, channel) in peer_state.outbound_v1_channel_by_id.iter() {
2367                                         let details = ChannelDetails::from_channel_context(&channel.context, best_block_height,
2368                                                 peer_state.latest_features.clone(), &self.fee_estimator);
2369                                         res.push(details);
2370                                 }
2371                         }
2372                 }
2373                 res
2374         }
2375
2376         /// Gets the list of usable channels, in random order. Useful as an argument to
2377         /// [`Router::find_route`] to ensure non-announced channels are used.
2378         ///
2379         /// These are guaranteed to have their [`ChannelDetails::is_usable`] value set to true, see the
2380         /// documentation for [`ChannelDetails::is_usable`] for more info on exactly what the criteria
2381         /// are.
2382         pub fn list_usable_channels(&self) -> Vec<ChannelDetails> {
2383                 // Note we use is_live here instead of usable which leads to somewhat confused
2384                 // internal/external nomenclature, but that's ok cause that's probably what the user
2385                 // really wanted anyway.
2386                 self.list_funded_channels_with_filter(|&(_, ref channel)| channel.context.is_live())
2387         }
2388
2389         /// Gets the list of channels we have with a given counterparty, in random order.
2390         pub fn list_channels_with_counterparty(&self, counterparty_node_id: &PublicKey) -> Vec<ChannelDetails> {
2391                 let best_block_height = self.best_block.read().unwrap().height();
2392                 let per_peer_state = self.per_peer_state.read().unwrap();
2393
2394                 if let Some(peer_state_mutex) = per_peer_state.get(counterparty_node_id) {
2395                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
2396                         let peer_state = &mut *peer_state_lock;
2397                         let features = &peer_state.latest_features;
2398                         let chan_context_to_details = |context| {
2399                                 ChannelDetails::from_channel_context(context, best_block_height, features.clone(), &self.fee_estimator)
2400                         };
2401                         return peer_state.channel_by_id
2402                                 .iter()
2403                                 .map(|(_, channel)| &channel.context)
2404                                 .chain(peer_state.outbound_v1_channel_by_id.iter().map(|(_, channel)| &channel.context))
2405                                 .chain(peer_state.inbound_v1_channel_by_id.iter().map(|(_, channel)| &channel.context))
2406                                 .map(chan_context_to_details)
2407                                 .collect();
2408                 }
2409                 vec![]
2410         }
2411
2412         /// Returns in an undefined order recent payments that -- if not fulfilled -- have yet to find a
2413         /// successful path, or have unresolved HTLCs.
2414         ///
2415         /// This can be useful for payments that may have been prepared, but ultimately not sent, as a
2416         /// result of a crash. If such a payment exists, is not listed here, and an
2417         /// [`Event::PaymentSent`] has not been received, you may consider resending the payment.
2418         ///
2419         /// [`Event::PaymentSent`]: events::Event::PaymentSent
2420         pub fn list_recent_payments(&self) -> Vec<RecentPaymentDetails> {
2421                 self.pending_outbound_payments.pending_outbound_payments.lock().unwrap().iter()
2422                         .filter_map(|(payment_id, pending_outbound_payment)| match pending_outbound_payment {
2423                                 PendingOutboundPayment::AwaitingInvoice { .. } => {
2424                                         Some(RecentPaymentDetails::AwaitingInvoice { payment_id: *payment_id })
2425                                 },
2426                                 // InvoiceReceived is an intermediate state and doesn't need to be exposed
2427                                 PendingOutboundPayment::InvoiceReceived { .. } => {
2428                                         Some(RecentPaymentDetails::AwaitingInvoice { payment_id: *payment_id })
2429                                 },
2430                                 PendingOutboundPayment::Retryable { payment_hash, total_msat, .. } => {
2431                                         Some(RecentPaymentDetails::Pending {
2432                                                 payment_hash: *payment_hash,
2433                                                 total_msat: *total_msat,
2434                                         })
2435                                 },
2436                                 PendingOutboundPayment::Abandoned { payment_hash, .. } => {
2437                                         Some(RecentPaymentDetails::Abandoned { payment_hash: *payment_hash })
2438                                 },
2439                                 PendingOutboundPayment::Fulfilled { payment_hash, .. } => {
2440                                         Some(RecentPaymentDetails::Fulfilled { payment_hash: *payment_hash })
2441                                 },
2442                                 PendingOutboundPayment::Legacy { .. } => None
2443                         })
2444                         .collect()
2445         }
2446
2447         /// Helper function that issues the channel close events
2448         fn issue_channel_close_events(&self, context: &ChannelContext<SP>, closure_reason: ClosureReason) {
2449                 let mut pending_events_lock = self.pending_events.lock().unwrap();
2450                 match context.unbroadcasted_funding() {
2451                         Some(transaction) => {
2452                                 pending_events_lock.push_back((events::Event::DiscardFunding {
2453                                         channel_id: context.channel_id(), transaction
2454                                 }, None));
2455                         },
2456                         None => {},
2457                 }
2458                 pending_events_lock.push_back((events::Event::ChannelClosed {
2459                         channel_id: context.channel_id(),
2460                         user_channel_id: context.get_user_id(),
2461                         reason: closure_reason,
2462                         counterparty_node_id: Some(context.get_counterparty_node_id()),
2463                         channel_capacity_sats: Some(context.get_value_satoshis()),
2464                 }, None));
2465         }
2466
2467         fn close_channel_internal(&self, channel_id: &ChannelId, counterparty_node_id: &PublicKey, target_feerate_sats_per_1000_weight: Option<u32>, override_shutdown_script: Option<ShutdownScript>) -> Result<(), APIError> {
2468                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
2469
2470                 let mut failed_htlcs: Vec<(HTLCSource, PaymentHash)>;
2471                 let result: Result<(), _> = loop {
2472                         {
2473                                 let per_peer_state = self.per_peer_state.read().unwrap();
2474
2475                                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
2476                                         .ok_or_else(|| APIError::ChannelUnavailable { err: format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id) })?;
2477
2478                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
2479                                 let peer_state = &mut *peer_state_lock;
2480
2481                                 match peer_state.channel_by_id.entry(channel_id.clone()) {
2482                                         hash_map::Entry::Occupied(mut chan_entry) => {
2483                                                 let funding_txo_opt = chan_entry.get().context.get_funding_txo();
2484                                                 let their_features = &peer_state.latest_features;
2485                                                 let (shutdown_msg, mut monitor_update_opt, htlcs) = chan_entry.get_mut()
2486                                                         .get_shutdown(&self.signer_provider, their_features, target_feerate_sats_per_1000_weight, override_shutdown_script)?;
2487                                                 failed_htlcs = htlcs;
2488
2489                                                 // We can send the `shutdown` message before updating the `ChannelMonitor`
2490                                                 // here as we don't need the monitor update to complete until we send a
2491                                                 // `shutdown_signed`, which we'll delay if we're pending a monitor update.
2492                                                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendShutdown {
2493                                                         node_id: *counterparty_node_id,
2494                                                         msg: shutdown_msg,
2495                                                 });
2496
2497                                                 // Update the monitor with the shutdown script if necessary.
2498                                                 if let Some(monitor_update) = monitor_update_opt.take() {
2499                                                         break handle_new_monitor_update!(self, funding_txo_opt.unwrap(), monitor_update,
2500                                                                 peer_state_lock, peer_state, per_peer_state, chan_entry).map(|_| ());
2501                                                 }
2502
2503                                                 if chan_entry.get().is_shutdown() {
2504                                                         let channel = remove_channel!(self, chan_entry);
2505                                                         if let Ok(channel_update) = self.get_channel_update_for_broadcast(&channel) {
2506                                                                 peer_state.pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
2507                                                                         msg: channel_update
2508                                                                 });
2509                                                         }
2510                                                         self.issue_channel_close_events(&channel.context, ClosureReason::HolderForceClosed);
2511                                                 }
2512                                                 break Ok(());
2513                                         },
2514                                         hash_map::Entry::Vacant(_) => (),
2515                                 }
2516                         }
2517                         // If we reach this point, it means that the channel_id either refers to an unfunded channel or
2518                         // it does not exist for this peer. Either way, we can attempt to force-close it.
2519                         //
2520                         // An appropriate error will be returned for non-existence of the channel if that's the case.
2521                         return self.force_close_channel_with_peer(&channel_id, counterparty_node_id, None, false).map(|_| ())
2522                         // TODO(dunxen): This is still not ideal as we're doing some extra lookups.
2523                         // Fix this with https://github.com/lightningdevkit/rust-lightning/issues/2422
2524                 };
2525
2526                 for htlc_source in failed_htlcs.drain(..) {
2527                         let reason = HTLCFailReason::from_failure_code(0x4000 | 8);
2528                         let receiver = HTLCDestination::NextHopChannel { node_id: Some(*counterparty_node_id), channel_id: *channel_id };
2529                         self.fail_htlc_backwards_internal(&htlc_source.0, &htlc_source.1, &reason, receiver);
2530                 }
2531
2532                 let _ = handle_error!(self, result, *counterparty_node_id);
2533                 Ok(())
2534         }
2535
2536         /// Begins the process of closing a channel. After this call (plus some timeout), no new HTLCs
2537         /// will be accepted on the given channel, and after additional timeout/the closing of all
2538         /// pending HTLCs, the channel will be closed on chain.
2539         ///
2540         ///  * If we are the channel initiator, we will pay between our [`Background`] and
2541         ///    [`ChannelConfig::force_close_avoidance_max_fee_satoshis`] plus our [`Normal`] fee
2542         ///    estimate.
2543         ///  * If our counterparty is the channel initiator, we will require a channel closing
2544         ///    transaction feerate of at least our [`Background`] feerate or the feerate which
2545         ///    would appear on a force-closure transaction, whichever is lower. We will allow our
2546         ///    counterparty to pay as much fee as they'd like, however.
2547         ///
2548         /// May generate a [`SendShutdown`] message event on success, which should be relayed.
2549         ///
2550         /// Raises [`APIError::ChannelUnavailable`] if the channel cannot be closed due to failing to
2551         /// generate a shutdown scriptpubkey or destination script set by
2552         /// [`SignerProvider::get_shutdown_scriptpubkey`]. A force-closure may be needed to close the
2553         /// channel.
2554         ///
2555         /// [`ChannelConfig::force_close_avoidance_max_fee_satoshis`]: crate::util::config::ChannelConfig::force_close_avoidance_max_fee_satoshis
2556         /// [`Background`]: crate::chain::chaininterface::ConfirmationTarget::Background
2557         /// [`Normal`]: crate::chain::chaininterface::ConfirmationTarget::Normal
2558         /// [`SendShutdown`]: crate::events::MessageSendEvent::SendShutdown
2559         pub fn close_channel(&self, channel_id: &ChannelId, counterparty_node_id: &PublicKey) -> Result<(), APIError> {
2560                 self.close_channel_internal(channel_id, counterparty_node_id, None, None)
2561         }
2562
2563         /// Begins the process of closing a channel. After this call (plus some timeout), no new HTLCs
2564         /// will be accepted on the given channel, and after additional timeout/the closing of all
2565         /// pending HTLCs, the channel will be closed on chain.
2566         ///
2567         /// `target_feerate_sat_per_1000_weight` has different meanings depending on if we initiated
2568         /// the channel being closed or not:
2569         ///  * If we are the channel initiator, we will pay at least this feerate on the closing
2570         ///    transaction. The upper-bound is set by
2571         ///    [`ChannelConfig::force_close_avoidance_max_fee_satoshis`] plus our [`Normal`] fee
2572         ///    estimate (or `target_feerate_sat_per_1000_weight`, if it is greater).
2573         ///  * If our counterparty is the channel initiator, we will refuse to accept a channel closure
2574         ///    transaction feerate below `target_feerate_sat_per_1000_weight` (or the feerate which
2575         ///    will appear on a force-closure transaction, whichever is lower).
2576         ///
2577         /// The `shutdown_script` provided  will be used as the `scriptPubKey` for the closing transaction.
2578         /// Will fail if a shutdown script has already been set for this channel by
2579         /// ['ChannelHandshakeConfig::commit_upfront_shutdown_pubkey`]. The given shutdown script must
2580         /// also be compatible with our and the counterparty's features.
2581         ///
2582         /// May generate a [`SendShutdown`] message event on success, which should be relayed.
2583         ///
2584         /// Raises [`APIError::ChannelUnavailable`] if the channel cannot be closed due to failing to
2585         /// generate a shutdown scriptpubkey or destination script set by
2586         /// [`SignerProvider::get_shutdown_scriptpubkey`]. A force-closure may be needed to close the
2587         /// channel.
2588         ///
2589         /// [`ChannelConfig::force_close_avoidance_max_fee_satoshis`]: crate::util::config::ChannelConfig::force_close_avoidance_max_fee_satoshis
2590         /// [`Background`]: crate::chain::chaininterface::ConfirmationTarget::Background
2591         /// [`Normal`]: crate::chain::chaininterface::ConfirmationTarget::Normal
2592         /// [`SendShutdown`]: crate::events::MessageSendEvent::SendShutdown
2593         pub fn close_channel_with_feerate_and_script(&self, channel_id: &ChannelId, counterparty_node_id: &PublicKey, target_feerate_sats_per_1000_weight: Option<u32>, shutdown_script: Option<ShutdownScript>) -> Result<(), APIError> {
2594                 self.close_channel_internal(channel_id, counterparty_node_id, target_feerate_sats_per_1000_weight, shutdown_script)
2595         }
2596
2597         #[inline]
2598         fn finish_force_close_channel(&self, shutdown_res: ShutdownResult) {
2599                 let (monitor_update_option, mut failed_htlcs) = shutdown_res;
2600                 log_debug!(self.logger, "Finishing force-closure of channel with {} HTLCs to fail", failed_htlcs.len());
2601                 for htlc_source in failed_htlcs.drain(..) {
2602                         let (source, payment_hash, counterparty_node_id, channel_id) = htlc_source;
2603                         let reason = HTLCFailReason::from_failure_code(0x4000 | 8);
2604                         let receiver = HTLCDestination::NextHopChannel { node_id: Some(counterparty_node_id), channel_id };
2605                         self.fail_htlc_backwards_internal(&source, &payment_hash, &reason, receiver);
2606                 }
2607                 if let Some((_, funding_txo, monitor_update)) = monitor_update_option {
2608                         // There isn't anything we can do if we get an update failure - we're already
2609                         // force-closing. The monitor update on the required in-memory copy should broadcast
2610                         // the latest local state, which is the best we can do anyway. Thus, it is safe to
2611                         // ignore the result here.
2612                         let _ = self.chain_monitor.update_channel(funding_txo, &monitor_update);
2613                 }
2614         }
2615
2616         /// `peer_msg` should be set when we receive a message from a peer, but not set when the
2617         /// user closes, which will be re-exposed as the `ChannelClosed` reason.
2618         fn force_close_channel_with_peer(&self, channel_id: &ChannelId, peer_node_id: &PublicKey, peer_msg: Option<&String>, broadcast: bool)
2619         -> Result<PublicKey, APIError> {
2620                 let per_peer_state = self.per_peer_state.read().unwrap();
2621                 let peer_state_mutex = per_peer_state.get(peer_node_id)
2622                         .ok_or_else(|| APIError::ChannelUnavailable { err: format!("Can't find a peer matching the passed counterparty node_id {}", peer_node_id) })?;
2623                 let (update_opt, counterparty_node_id) = {
2624                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
2625                         let peer_state = &mut *peer_state_lock;
2626                         let closure_reason = if let Some(peer_msg) = peer_msg {
2627                                 ClosureReason::CounterpartyForceClosed { peer_msg: UntrustedString(peer_msg.to_string()) }
2628                         } else {
2629                                 ClosureReason::HolderForceClosed
2630                         };
2631                         if let hash_map::Entry::Occupied(chan) = peer_state.channel_by_id.entry(channel_id.clone()) {
2632                                 log_error!(self.logger, "Force-closing channel {}", &channel_id);
2633                                 self.issue_channel_close_events(&chan.get().context, closure_reason);
2634                                 let mut chan = remove_channel!(self, chan);
2635                                 self.finish_force_close_channel(chan.context.force_shutdown(broadcast));
2636                                 (self.get_channel_update_for_broadcast(&chan).ok(), chan.context.get_counterparty_node_id())
2637                         } else if let hash_map::Entry::Occupied(chan) = peer_state.outbound_v1_channel_by_id.entry(channel_id.clone()) {
2638                                 log_error!(self.logger, "Force-closing channel {}", &channel_id);
2639                                 self.issue_channel_close_events(&chan.get().context, closure_reason);
2640                                 let mut chan = remove_channel!(self, chan);
2641                                 self.finish_force_close_channel(chan.context.force_shutdown(false));
2642                                 // Unfunded channel has no update
2643                                 (None, chan.context.get_counterparty_node_id())
2644                         } else if let hash_map::Entry::Occupied(chan) = peer_state.inbound_v1_channel_by_id.entry(channel_id.clone()) {
2645                                 log_error!(self.logger, "Force-closing channel {}", &channel_id);
2646                                 self.issue_channel_close_events(&chan.get().context, closure_reason);
2647                                 let mut chan = remove_channel!(self, chan);
2648                                 self.finish_force_close_channel(chan.context.force_shutdown(false));
2649                                 // Unfunded channel has no update
2650                                 (None, chan.context.get_counterparty_node_id())
2651                         } else if peer_state.inbound_channel_request_by_id.remove(channel_id).is_some() {
2652                                 log_error!(self.logger, "Force-closing channel {}", &channel_id);
2653                                 // N.B. that we don't send any channel close event here: we
2654                                 // don't have a user_channel_id, and we never sent any opening
2655                                 // events anyway.
2656                                 (None, *peer_node_id)
2657                         } else {
2658                                 return Err(APIError::ChannelUnavailable{ err: format!("Channel with id {} not found for the passed counterparty node_id {}", channel_id, peer_node_id) });
2659                         }
2660                 };
2661                 if let Some(update) = update_opt {
2662                         let mut peer_state = peer_state_mutex.lock().unwrap();
2663                         peer_state.pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
2664                                 msg: update
2665                         });
2666                 }
2667
2668                 Ok(counterparty_node_id)
2669         }
2670
2671         fn force_close_sending_error(&self, channel_id: &ChannelId, counterparty_node_id: &PublicKey, broadcast: bool) -> Result<(), APIError> {
2672                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
2673                 match self.force_close_channel_with_peer(channel_id, counterparty_node_id, None, broadcast) {
2674                         Ok(counterparty_node_id) => {
2675                                 let per_peer_state = self.per_peer_state.read().unwrap();
2676                                 if let Some(peer_state_mutex) = per_peer_state.get(&counterparty_node_id) {
2677                                         let mut peer_state = peer_state_mutex.lock().unwrap();
2678                                         peer_state.pending_msg_events.push(
2679                                                 events::MessageSendEvent::HandleError {
2680                                                         node_id: counterparty_node_id,
2681                                                         action: msgs::ErrorAction::SendErrorMessage {
2682                                                                 msg: msgs::ErrorMessage { channel_id: *channel_id, data: "Channel force-closed".to_owned() }
2683                                                         },
2684                                                 }
2685                                         );
2686                                 }
2687                                 Ok(())
2688                         },
2689                         Err(e) => Err(e)
2690                 }
2691         }
2692
2693         /// Force closes a channel, immediately broadcasting the latest local transaction(s) and
2694         /// rejecting new HTLCs on the given channel. Fails if `channel_id` is unknown to
2695         /// the manager, or if the `counterparty_node_id` isn't the counterparty of the corresponding
2696         /// channel.
2697         pub fn force_close_broadcasting_latest_txn(&self, channel_id: &ChannelId, counterparty_node_id: &PublicKey)
2698         -> Result<(), APIError> {
2699                 self.force_close_sending_error(channel_id, counterparty_node_id, true)
2700         }
2701
2702         /// Force closes a channel, rejecting new HTLCs on the given channel but skips broadcasting
2703         /// the latest local transaction(s). Fails if `channel_id` is unknown to the manager, or if the
2704         /// `counterparty_node_id` isn't the counterparty of the corresponding channel.
2705         ///
2706         /// You can always get the latest local transaction(s) to broadcast from
2707         /// [`ChannelMonitor::get_latest_holder_commitment_txn`].
2708         pub fn force_close_without_broadcasting_txn(&self, channel_id: &ChannelId, counterparty_node_id: &PublicKey)
2709         -> Result<(), APIError> {
2710                 self.force_close_sending_error(channel_id, counterparty_node_id, false)
2711         }
2712
2713         /// Force close all channels, immediately broadcasting the latest local commitment transaction
2714         /// for each to the chain and rejecting new HTLCs on each.
2715         pub fn force_close_all_channels_broadcasting_latest_txn(&self) {
2716                 for chan in self.list_channels() {
2717                         let _ = self.force_close_broadcasting_latest_txn(&chan.channel_id, &chan.counterparty.node_id);
2718                 }
2719         }
2720
2721         /// Force close all channels rejecting new HTLCs on each but without broadcasting the latest
2722         /// local transaction(s).
2723         pub fn force_close_all_channels_without_broadcasting_txn(&self) {
2724                 for chan in self.list_channels() {
2725                         let _ = self.force_close_without_broadcasting_txn(&chan.channel_id, &chan.counterparty.node_id);
2726                 }
2727         }
2728
2729         fn construct_fwd_pending_htlc_info(
2730                 &self, msg: &msgs::UpdateAddHTLC, hop_data: msgs::InboundOnionPayload, hop_hmac: [u8; 32],
2731                 new_packet_bytes: [u8; onion_utils::ONION_DATA_LEN], shared_secret: [u8; 32],
2732                 next_packet_pubkey_opt: Option<Result<PublicKey, secp256k1::Error>>
2733         ) -> Result<PendingHTLCInfo, InboundOnionErr> {
2734                 debug_assert!(next_packet_pubkey_opt.is_some());
2735                 let outgoing_packet = msgs::OnionPacket {
2736                         version: 0,
2737                         public_key: next_packet_pubkey_opt.unwrap_or(Err(secp256k1::Error::InvalidPublicKey)),
2738                         hop_data: new_packet_bytes,
2739                         hmac: hop_hmac,
2740                 };
2741
2742                 let (short_channel_id, amt_to_forward, outgoing_cltv_value) = match hop_data {
2743                         msgs::InboundOnionPayload::Forward { short_channel_id, amt_to_forward, outgoing_cltv_value } =>
2744                                 (short_channel_id, amt_to_forward, outgoing_cltv_value),
2745                         msgs::InboundOnionPayload::Receive { .. } =>
2746                                 return Err(InboundOnionErr {
2747                                         msg: "Final Node OnionHopData provided for us as an intermediary node",
2748                                         err_code: 0x4000 | 22,
2749                                         err_data: Vec::new(),
2750                                 }),
2751                 };
2752
2753                 Ok(PendingHTLCInfo {
2754                         routing: PendingHTLCRouting::Forward {
2755                                 onion_packet: outgoing_packet,
2756                                 short_channel_id,
2757                         },
2758                         payment_hash: msg.payment_hash,
2759                         incoming_shared_secret: shared_secret,
2760                         incoming_amt_msat: Some(msg.amount_msat),
2761                         outgoing_amt_msat: amt_to_forward,
2762                         outgoing_cltv_value,
2763                         skimmed_fee_msat: None,
2764                 })
2765         }
2766
2767         fn construct_recv_pending_htlc_info(
2768                 &self, hop_data: msgs::InboundOnionPayload, shared_secret: [u8; 32], payment_hash: PaymentHash,
2769                 amt_msat: u64, cltv_expiry: u32, phantom_shared_secret: Option<[u8; 32]>, allow_underpay: bool,
2770                 counterparty_skimmed_fee_msat: Option<u64>,
2771         ) -> Result<PendingHTLCInfo, InboundOnionErr> {
2772                 let (payment_data, keysend_preimage, custom_tlvs, onion_amt_msat, outgoing_cltv_value, payment_metadata) = match hop_data {
2773                         msgs::InboundOnionPayload::Receive {
2774                                 payment_data, keysend_preimage, custom_tlvs, amt_msat, outgoing_cltv_value, payment_metadata, ..
2775                         } =>
2776                                 (payment_data, keysend_preimage, custom_tlvs, amt_msat, outgoing_cltv_value, payment_metadata),
2777                         _ =>
2778                                 return Err(InboundOnionErr {
2779                                         err_code: 0x4000|22,
2780                                         err_data: Vec::new(),
2781                                         msg: "Got non final data with an HMAC of 0",
2782                                 }),
2783                 };
2784                 // final_incorrect_cltv_expiry
2785                 if outgoing_cltv_value > cltv_expiry {
2786                         return Err(InboundOnionErr {
2787                                 msg: "Upstream node set CLTV to less than the CLTV set by the sender",
2788                                 err_code: 18,
2789                                 err_data: cltv_expiry.to_be_bytes().to_vec()
2790                         })
2791                 }
2792                 // final_expiry_too_soon
2793                 // We have to have some headroom to broadcast on chain if we have the preimage, so make sure
2794                 // we have at least HTLC_FAIL_BACK_BUFFER blocks to go.
2795                 //
2796                 // Also, ensure that, in the case of an unknown preimage for the received payment hash, our
2797                 // payment logic has enough time to fail the HTLC backward before our onchain logic triggers a
2798                 // channel closure (see HTLC_FAIL_BACK_BUFFER rationale).
2799                 let current_height: u32 = self.best_block.read().unwrap().height();
2800                 if (outgoing_cltv_value as u64) <= current_height as u64 + HTLC_FAIL_BACK_BUFFER as u64 + 1 {
2801                         let mut err_data = Vec::with_capacity(12);
2802                         err_data.extend_from_slice(&amt_msat.to_be_bytes());
2803                         err_data.extend_from_slice(&current_height.to_be_bytes());
2804                         return Err(InboundOnionErr {
2805                                 err_code: 0x4000 | 15, err_data,
2806                                 msg: "The final CLTV expiry is too soon to handle",
2807                         });
2808                 }
2809                 if (!allow_underpay && onion_amt_msat > amt_msat) ||
2810                         (allow_underpay && onion_amt_msat >
2811                          amt_msat.saturating_add(counterparty_skimmed_fee_msat.unwrap_or(0)))
2812                 {
2813                         return Err(InboundOnionErr {
2814                                 err_code: 19,
2815                                 err_data: amt_msat.to_be_bytes().to_vec(),
2816                                 msg: "Upstream node sent less than we were supposed to receive in payment",
2817                         });
2818                 }
2819
2820                 let routing = if let Some(payment_preimage) = keysend_preimage {
2821                         // We need to check that the sender knows the keysend preimage before processing this
2822                         // payment further. Otherwise, an intermediary routing hop forwarding non-keysend-HTLC X
2823                         // could discover the final destination of X, by probing the adjacent nodes on the route
2824                         // with a keysend payment of identical payment hash to X and observing the processing
2825                         // time discrepancies due to a hash collision with X.
2826                         let hashed_preimage = PaymentHash(Sha256::hash(&payment_preimage.0).into_inner());
2827                         if hashed_preimage != payment_hash {
2828                                 return Err(InboundOnionErr {
2829                                         err_code: 0x4000|22,
2830                                         err_data: Vec::new(),
2831                                         msg: "Payment preimage didn't match payment hash",
2832                                 });
2833                         }
2834                         if !self.default_configuration.accept_mpp_keysend && payment_data.is_some() {
2835                                 return Err(InboundOnionErr {
2836                                         err_code: 0x4000|22,
2837                                         err_data: Vec::new(),
2838                                         msg: "We don't support MPP keysend payments",
2839                                 });
2840                         }
2841                         PendingHTLCRouting::ReceiveKeysend {
2842                                 payment_data,
2843                                 payment_preimage,
2844                                 payment_metadata,
2845                                 incoming_cltv_expiry: outgoing_cltv_value,
2846                                 custom_tlvs,
2847                         }
2848                 } else if let Some(data) = payment_data {
2849                         PendingHTLCRouting::Receive {
2850                                 payment_data: data,
2851                                 payment_metadata,
2852                                 incoming_cltv_expiry: outgoing_cltv_value,
2853                                 phantom_shared_secret,
2854                                 custom_tlvs,
2855                         }
2856                 } else {
2857                         return Err(InboundOnionErr {
2858                                 err_code: 0x4000|0x2000|3,
2859                                 err_data: Vec::new(),
2860                                 msg: "We require payment_secrets",
2861                         });
2862                 };
2863                 Ok(PendingHTLCInfo {
2864                         routing,
2865                         payment_hash,
2866                         incoming_shared_secret: shared_secret,
2867                         incoming_amt_msat: Some(amt_msat),
2868                         outgoing_amt_msat: onion_amt_msat,
2869                         outgoing_cltv_value,
2870                         skimmed_fee_msat: counterparty_skimmed_fee_msat,
2871                 })
2872         }
2873
2874         fn decode_update_add_htlc_onion(
2875                 &self, msg: &msgs::UpdateAddHTLC
2876         ) -> Result<(onion_utils::Hop, [u8; 32], Option<Result<PublicKey, secp256k1::Error>>), HTLCFailureMsg> {
2877                 macro_rules! return_malformed_err {
2878                         ($msg: expr, $err_code: expr) => {
2879                                 {
2880                                         log_info!(self.logger, "Failed to accept/forward incoming HTLC: {}", $msg);
2881                                         return Err(HTLCFailureMsg::Malformed(msgs::UpdateFailMalformedHTLC {
2882                                                 channel_id: msg.channel_id,
2883                                                 htlc_id: msg.htlc_id,
2884                                                 sha256_of_onion: Sha256::hash(&msg.onion_routing_packet.hop_data).into_inner(),
2885                                                 failure_code: $err_code,
2886                                         }));
2887                                 }
2888                         }
2889                 }
2890
2891                 if let Err(_) = msg.onion_routing_packet.public_key {
2892                         return_malformed_err!("invalid ephemeral pubkey", 0x8000 | 0x4000 | 6);
2893                 }
2894
2895                 let shared_secret = self.node_signer.ecdh(
2896                         Recipient::Node, &msg.onion_routing_packet.public_key.unwrap(), None
2897                 ).unwrap().secret_bytes();
2898
2899                 if msg.onion_routing_packet.version != 0 {
2900                         //TODO: Spec doesn't indicate if we should only hash hop_data here (and in other
2901                         //sha256_of_onion error data packets), or the entire onion_routing_packet. Either way,
2902                         //the hash doesn't really serve any purpose - in the case of hashing all data, the
2903                         //receiving node would have to brute force to figure out which version was put in the
2904                         //packet by the node that send us the message, in the case of hashing the hop_data, the
2905                         //node knows the HMAC matched, so they already know what is there...
2906                         return_malformed_err!("Unknown onion packet version", 0x8000 | 0x4000 | 4);
2907                 }
2908                 macro_rules! return_err {
2909                         ($msg: expr, $err_code: expr, $data: expr) => {
2910                                 {
2911                                         log_info!(self.logger, "Failed to accept/forward incoming HTLC: {}", $msg);
2912                                         return Err(HTLCFailureMsg::Relay(msgs::UpdateFailHTLC {
2913                                                 channel_id: msg.channel_id,
2914                                                 htlc_id: msg.htlc_id,
2915                                                 reason: HTLCFailReason::reason($err_code, $data.to_vec())
2916                                                         .get_encrypted_failure_packet(&shared_secret, &None),
2917                                         }));
2918                                 }
2919                         }
2920                 }
2921
2922                 let next_hop = match onion_utils::decode_next_payment_hop(shared_secret, &msg.onion_routing_packet.hop_data[..], msg.onion_routing_packet.hmac, msg.payment_hash) {
2923                         Ok(res) => res,
2924                         Err(onion_utils::OnionDecodeErr::Malformed { err_msg, err_code }) => {
2925                                 return_malformed_err!(err_msg, err_code);
2926                         },
2927                         Err(onion_utils::OnionDecodeErr::Relay { err_msg, err_code }) => {
2928                                 return_err!(err_msg, err_code, &[0; 0]);
2929                         },
2930                 };
2931                 let (outgoing_scid, outgoing_amt_msat, outgoing_cltv_value, next_packet_pk_opt) = match next_hop {
2932                         onion_utils::Hop::Forward {
2933                                 next_hop_data: msgs::InboundOnionPayload::Forward {
2934                                         short_channel_id, amt_to_forward, outgoing_cltv_value
2935                                 }, ..
2936                         } => {
2937                                 let next_packet_pk = onion_utils::next_hop_pubkey(&self.secp_ctx,
2938                                         msg.onion_routing_packet.public_key.unwrap(), &shared_secret);
2939                                 (short_channel_id, amt_to_forward, outgoing_cltv_value, Some(next_packet_pk))
2940                         },
2941                         // We'll do receive checks in [`Self::construct_pending_htlc_info`] so we have access to the
2942                         // inbound channel's state.
2943                         onion_utils::Hop::Receive { .. } => return Ok((next_hop, shared_secret, None)),
2944                         onion_utils::Hop::Forward { next_hop_data: msgs::InboundOnionPayload::Receive { .. }, .. } => {
2945                                 return_err!("Final Node OnionHopData provided for us as an intermediary node", 0x4000 | 22, &[0; 0]);
2946                         }
2947                 };
2948
2949                 // Perform outbound checks here instead of in [`Self::construct_pending_htlc_info`] because we
2950                 // can't hold the outbound peer state lock at the same time as the inbound peer state lock.
2951                 if let Some((err, mut code, chan_update)) = loop {
2952                         let id_option = self.short_to_chan_info.read().unwrap().get(&outgoing_scid).cloned();
2953                         let forwarding_chan_info_opt = match id_option {
2954                                 None => { // unknown_next_peer
2955                                         // Note that this is likely a timing oracle for detecting whether an scid is a
2956                                         // phantom or an intercept.
2957                                         if (self.default_configuration.accept_intercept_htlcs &&
2958                                                 fake_scid::is_valid_intercept(&self.fake_scid_rand_bytes, outgoing_scid, &self.genesis_hash)) ||
2959                                                 fake_scid::is_valid_phantom(&self.fake_scid_rand_bytes, outgoing_scid, &self.genesis_hash)
2960                                         {
2961                                                 None
2962                                         } else {
2963                                                 break Some(("Don't have available channel for forwarding as requested.", 0x4000 | 10, None));
2964                                         }
2965                                 },
2966                                 Some((cp_id, id)) => Some((cp_id.clone(), id.clone())),
2967                         };
2968                         let chan_update_opt = if let Some((counterparty_node_id, forwarding_id)) = forwarding_chan_info_opt {
2969                                 let per_peer_state = self.per_peer_state.read().unwrap();
2970                                 let peer_state_mutex_opt = per_peer_state.get(&counterparty_node_id);
2971                                 if peer_state_mutex_opt.is_none() {
2972                                         break Some(("Don't have available channel for forwarding as requested.", 0x4000 | 10, None));
2973                                 }
2974                                 let mut peer_state_lock = peer_state_mutex_opt.unwrap().lock().unwrap();
2975                                 let peer_state = &mut *peer_state_lock;
2976                                 let chan = match peer_state.channel_by_id.get_mut(&forwarding_id) {
2977                                         None => {
2978                                                 // Channel was removed. The short_to_chan_info and channel_by_id maps
2979                                                 // have no consistency guarantees.
2980                                                 break Some(("Don't have available channel for forwarding as requested.", 0x4000 | 10, None));
2981                                         },
2982                                         Some(chan) => chan
2983                                 };
2984                                 if !chan.context.should_announce() && !self.default_configuration.accept_forwards_to_priv_channels {
2985                                         // Note that the behavior here should be identical to the above block - we
2986                                         // should NOT reveal the existence or non-existence of a private channel if
2987                                         // we don't allow forwards outbound over them.
2988                                         break Some(("Refusing to forward to a private channel based on our config.", 0x4000 | 10, None));
2989                                 }
2990                                 if chan.context.get_channel_type().supports_scid_privacy() && outgoing_scid != chan.context.outbound_scid_alias() {
2991                                         // `option_scid_alias` (referred to in LDK as `scid_privacy`) means
2992                                         // "refuse to forward unless the SCID alias was used", so we pretend
2993                                         // we don't have the channel here.
2994                                         break Some(("Refusing to forward over real channel SCID as our counterparty requested.", 0x4000 | 10, None));
2995                                 }
2996                                 let chan_update_opt = self.get_channel_update_for_onion(outgoing_scid, chan).ok();
2997
2998                                 // Note that we could technically not return an error yet here and just hope
2999                                 // that the connection is reestablished or monitor updated by the time we get
3000                                 // around to doing the actual forward, but better to fail early if we can and
3001                                 // hopefully an attacker trying to path-trace payments cannot make this occur
3002                                 // on a small/per-node/per-channel scale.
3003                                 if !chan.context.is_live() { // channel_disabled
3004                                         // If the channel_update we're going to return is disabled (i.e. the
3005                                         // peer has been disabled for some time), return `channel_disabled`,
3006                                         // otherwise return `temporary_channel_failure`.
3007                                         if chan_update_opt.as_ref().map(|u| u.contents.flags & 2 == 2).unwrap_or(false) {
3008                                                 break Some(("Forwarding channel has been disconnected for some time.", 0x1000 | 20, chan_update_opt));
3009                                         } else {
3010                                                 break Some(("Forwarding channel is not in a ready state.", 0x1000 | 7, chan_update_opt));
3011                                         }
3012                                 }
3013                                 if outgoing_amt_msat < chan.context.get_counterparty_htlc_minimum_msat() { // amount_below_minimum
3014                                         break Some(("HTLC amount was below the htlc_minimum_msat", 0x1000 | 11, chan_update_opt));
3015                                 }
3016                                 if let Err((err, code)) = chan.htlc_satisfies_config(&msg, outgoing_amt_msat, outgoing_cltv_value) {
3017                                         break Some((err, code, chan_update_opt));
3018                                 }
3019                                 chan_update_opt
3020                         } else {
3021                                 if (msg.cltv_expiry as u64) < (outgoing_cltv_value) as u64 + MIN_CLTV_EXPIRY_DELTA as u64 {
3022                                         // We really should set `incorrect_cltv_expiry` here but as we're not
3023                                         // forwarding over a real channel we can't generate a channel_update
3024                                         // for it. Instead we just return a generic temporary_node_failure.
3025                                         break Some((
3026                                                         "Forwarding node has tampered with the intended HTLC values or origin node has an obsolete cltv_expiry_delta",
3027                                                         0x2000 | 2, None,
3028                                         ));
3029                                 }
3030                                 None
3031                         };
3032
3033                         let cur_height = self.best_block.read().unwrap().height() + 1;
3034                         // Theoretically, channel counterparty shouldn't send us a HTLC expiring now,
3035                         // but we want to be robust wrt to counterparty packet sanitization (see
3036                         // HTLC_FAIL_BACK_BUFFER rationale).
3037                         if msg.cltv_expiry <= cur_height + HTLC_FAIL_BACK_BUFFER as u32 { // expiry_too_soon
3038                                 break Some(("CLTV expiry is too close", 0x1000 | 14, chan_update_opt));
3039                         }
3040                         if msg.cltv_expiry > cur_height + CLTV_FAR_FAR_AWAY as u32 { // expiry_too_far
3041                                 break Some(("CLTV expiry is too far in the future", 21, None));
3042                         }
3043                         // If the HTLC expires ~now, don't bother trying to forward it to our
3044                         // counterparty. They should fail it anyway, but we don't want to bother with
3045                         // the round-trips or risk them deciding they definitely want the HTLC and
3046                         // force-closing to ensure they get it if we're offline.
3047                         // We previously had a much more aggressive check here which tried to ensure
3048                         // our counterparty receives an HTLC which has *our* risk threshold met on it,
3049                         // but there is no need to do that, and since we're a bit conservative with our
3050                         // risk threshold it just results in failing to forward payments.
3051                         if (outgoing_cltv_value) as u64 <= (cur_height + LATENCY_GRACE_PERIOD_BLOCKS) as u64 {
3052                                 break Some(("Outgoing CLTV value is too soon", 0x1000 | 14, chan_update_opt));
3053                         }
3054
3055                         break None;
3056                 }
3057                 {
3058                         let mut res = VecWriter(Vec::with_capacity(chan_update.serialized_length() + 2 + 8 + 2));
3059                         if let Some(chan_update) = chan_update {
3060                                 if code == 0x1000 | 11 || code == 0x1000 | 12 {
3061                                         msg.amount_msat.write(&mut res).expect("Writes cannot fail");
3062                                 }
3063                                 else if code == 0x1000 | 13 {
3064                                         msg.cltv_expiry.write(&mut res).expect("Writes cannot fail");
3065                                 }
3066                                 else if code == 0x1000 | 20 {
3067                                         // TODO: underspecified, follow https://github.com/lightning/bolts/issues/791
3068                                         0u16.write(&mut res).expect("Writes cannot fail");
3069                                 }
3070                                 (chan_update.serialized_length() as u16 + 2).write(&mut res).expect("Writes cannot fail");
3071                                 msgs::ChannelUpdate::TYPE.write(&mut res).expect("Writes cannot fail");
3072                                 chan_update.write(&mut res).expect("Writes cannot fail");
3073                         } else if code & 0x1000 == 0x1000 {
3074                                 // If we're trying to return an error that requires a `channel_update` but
3075                                 // we're forwarding to a phantom or intercept "channel" (i.e. cannot
3076                                 // generate an update), just use the generic "temporary_node_failure"
3077                                 // instead.
3078                                 code = 0x2000 | 2;
3079                         }
3080                         return_err!(err, code, &res.0[..]);
3081                 }
3082                 Ok((next_hop, shared_secret, next_packet_pk_opt))
3083         }
3084
3085         fn construct_pending_htlc_status<'a>(
3086                 &self, msg: &msgs::UpdateAddHTLC, shared_secret: [u8; 32], decoded_hop: onion_utils::Hop,
3087                 allow_underpay: bool, next_packet_pubkey_opt: Option<Result<PublicKey, secp256k1::Error>>
3088         ) -> PendingHTLCStatus {
3089                 macro_rules! return_err {
3090                         ($msg: expr, $err_code: expr, $data: expr) => {
3091                                 {
3092                                         log_info!(self.logger, "Failed to accept/forward incoming HTLC: {}", $msg);
3093                                         return PendingHTLCStatus::Fail(HTLCFailureMsg::Relay(msgs::UpdateFailHTLC {
3094                                                 channel_id: msg.channel_id,
3095                                                 htlc_id: msg.htlc_id,
3096                                                 reason: HTLCFailReason::reason($err_code, $data.to_vec())
3097                                                         .get_encrypted_failure_packet(&shared_secret, &None),
3098                                         }));
3099                                 }
3100                         }
3101                 }
3102                 match decoded_hop {
3103                         onion_utils::Hop::Receive(next_hop_data) => {
3104                                 // OUR PAYMENT!
3105                                 match self.construct_recv_pending_htlc_info(next_hop_data, shared_secret, msg.payment_hash,
3106                                         msg.amount_msat, msg.cltv_expiry, None, allow_underpay, msg.skimmed_fee_msat)
3107                                 {
3108                                         Ok(info) => {
3109                                                 // Note that we could obviously respond immediately with an update_fulfill_htlc
3110                                                 // message, however that would leak that we are the recipient of this payment, so
3111                                                 // instead we stay symmetric with the forwarding case, only responding (after a
3112                                                 // delay) once they've send us a commitment_signed!
3113                                                 PendingHTLCStatus::Forward(info)
3114                                         },
3115                                         Err(InboundOnionErr { err_code, err_data, msg }) => return_err!(msg, err_code, &err_data)
3116                                 }
3117                         },
3118                         onion_utils::Hop::Forward { next_hop_data, next_hop_hmac, new_packet_bytes } => {
3119                                 match self.construct_fwd_pending_htlc_info(msg, next_hop_data, next_hop_hmac,
3120                                         new_packet_bytes, shared_secret, next_packet_pubkey_opt) {
3121                                         Ok(info) => PendingHTLCStatus::Forward(info),
3122                                         Err(InboundOnionErr { err_code, err_data, msg }) => return_err!(msg, err_code, &err_data)
3123                                 }
3124                         }
3125                 }
3126         }
3127
3128         /// Gets the current [`channel_update`] for the given channel. This first checks if the channel is
3129         /// public, and thus should be called whenever the result is going to be passed out in a
3130         /// [`MessageSendEvent::BroadcastChannelUpdate`] event.
3131         ///
3132         /// Note that in [`internal_closing_signed`], this function is called without the `peer_state`
3133         /// corresponding to the channel's counterparty locked, as the channel been removed from the
3134         /// storage and the `peer_state` lock has been dropped.
3135         ///
3136         /// [`channel_update`]: msgs::ChannelUpdate
3137         /// [`internal_closing_signed`]: Self::internal_closing_signed
3138         fn get_channel_update_for_broadcast(&self, chan: &Channel<SP>) -> Result<msgs::ChannelUpdate, LightningError> {
3139                 if !chan.context.should_announce() {
3140                         return Err(LightningError {
3141                                 err: "Cannot broadcast a channel_update for a private channel".to_owned(),
3142                                 action: msgs::ErrorAction::IgnoreError
3143                         });
3144                 }
3145                 if chan.context.get_short_channel_id().is_none() {
3146                         return Err(LightningError{err: "Channel not yet established".to_owned(), action: msgs::ErrorAction::IgnoreError});
3147                 }
3148                 log_trace!(self.logger, "Attempting to generate broadcast channel update for channel {}", &chan.context.channel_id());
3149                 self.get_channel_update_for_unicast(chan)
3150         }
3151
3152         /// Gets the current [`channel_update`] for the given channel. This does not check if the channel
3153         /// is public (only returning an `Err` if the channel does not yet have an assigned SCID),
3154         /// and thus MUST NOT be called unless the recipient of the resulting message has already
3155         /// provided evidence that they know about the existence of the channel.
3156         ///
3157         /// Note that through [`internal_closing_signed`], this function is called without the
3158         /// `peer_state`  corresponding to the channel's counterparty locked, as the channel been
3159         /// removed from the storage and the `peer_state` lock has been dropped.
3160         ///
3161         /// [`channel_update`]: msgs::ChannelUpdate
3162         /// [`internal_closing_signed`]: Self::internal_closing_signed
3163         fn get_channel_update_for_unicast(&self, chan: &Channel<SP>) -> Result<msgs::ChannelUpdate, LightningError> {
3164                 log_trace!(self.logger, "Attempting to generate channel update for channel {}", &chan.context.channel_id());
3165                 let short_channel_id = match chan.context.get_short_channel_id().or(chan.context.latest_inbound_scid_alias()) {
3166                         None => return Err(LightningError{err: "Channel not yet established".to_owned(), action: msgs::ErrorAction::IgnoreError}),
3167                         Some(id) => id,
3168                 };
3169
3170                 self.get_channel_update_for_onion(short_channel_id, chan)
3171         }
3172
3173         fn get_channel_update_for_onion(&self, short_channel_id: u64, chan: &Channel<SP>) -> Result<msgs::ChannelUpdate, LightningError> {
3174                 log_trace!(self.logger, "Generating channel update for channel {}", &chan.context.channel_id());
3175                 let were_node_one = self.our_network_pubkey.serialize()[..] < chan.context.get_counterparty_node_id().serialize()[..];
3176
3177                 let enabled = chan.context.is_usable() && match chan.channel_update_status() {
3178                         ChannelUpdateStatus::Enabled => true,
3179                         ChannelUpdateStatus::DisabledStaged(_) => true,
3180                         ChannelUpdateStatus::Disabled => false,
3181                         ChannelUpdateStatus::EnabledStaged(_) => false,
3182                 };
3183
3184                 let unsigned = msgs::UnsignedChannelUpdate {
3185                         chain_hash: self.genesis_hash,
3186                         short_channel_id,
3187                         timestamp: chan.context.get_update_time_counter(),
3188                         flags: (!were_node_one) as u8 | ((!enabled as u8) << 1),
3189                         cltv_expiry_delta: chan.context.get_cltv_expiry_delta(),
3190                         htlc_minimum_msat: chan.context.get_counterparty_htlc_minimum_msat(),
3191                         htlc_maximum_msat: chan.context.get_announced_htlc_max_msat(),
3192                         fee_base_msat: chan.context.get_outbound_forwarding_fee_base_msat(),
3193                         fee_proportional_millionths: chan.context.get_fee_proportional_millionths(),
3194                         excess_data: Vec::new(),
3195                 };
3196                 // Panic on failure to signal LDK should be restarted to retry signing the `ChannelUpdate`.
3197                 // If we returned an error and the `node_signer` cannot provide a signature for whatever
3198                 // reason`, we wouldn't be able to receive inbound payments through the corresponding
3199                 // channel.
3200                 let sig = self.node_signer.sign_gossip_message(msgs::UnsignedGossipMessage::ChannelUpdate(&unsigned)).unwrap();
3201
3202                 Ok(msgs::ChannelUpdate {
3203                         signature: sig,
3204                         contents: unsigned
3205                 })
3206         }
3207
3208         #[cfg(test)]
3209         pub(crate) fn test_send_payment_along_path(&self, path: &Path, payment_hash: &PaymentHash, recipient_onion: RecipientOnionFields, total_value: u64, cur_height: u32, payment_id: PaymentId, keysend_preimage: &Option<PaymentPreimage>, session_priv_bytes: [u8; 32]) -> Result<(), APIError> {
3210                 let _lck = self.total_consistency_lock.read().unwrap();
3211                 self.send_payment_along_path(SendAlongPathArgs {
3212                         path, payment_hash, recipient_onion, total_value, cur_height, payment_id, keysend_preimage,
3213                         session_priv_bytes
3214                 })
3215         }
3216
3217         fn send_payment_along_path(&self, args: SendAlongPathArgs) -> Result<(), APIError> {
3218                 let SendAlongPathArgs {
3219                         path, payment_hash, recipient_onion, total_value, cur_height, payment_id, keysend_preimage,
3220                         session_priv_bytes
3221                 } = args;
3222                 // The top-level caller should hold the total_consistency_lock read lock.
3223                 debug_assert!(self.total_consistency_lock.try_write().is_err());
3224
3225                 log_trace!(self.logger,
3226                         "Attempting to send payment with payment hash {} along path with next hop {}",
3227                         payment_hash, path.hops.first().unwrap().short_channel_id);
3228                 let prng_seed = self.entropy_source.get_secure_random_bytes();
3229                 let session_priv = SecretKey::from_slice(&session_priv_bytes[..]).expect("RNG is busted");
3230
3231                 let onion_keys = onion_utils::construct_onion_keys(&self.secp_ctx, &path, &session_priv)
3232                         .map_err(|_| APIError::InvalidRoute{err: "Pubkey along hop was maliciously selected".to_owned()})?;
3233                 let (onion_payloads, htlc_msat, htlc_cltv) = onion_utils::build_onion_payloads(path, total_value, recipient_onion, cur_height, keysend_preimage)?;
3234
3235                 let onion_packet = onion_utils::construct_onion_packet(onion_payloads, onion_keys, prng_seed, payment_hash)
3236                         .map_err(|_| APIError::InvalidRoute { err: "Route size too large considering onion data".to_owned()})?;
3237
3238                 let err: Result<(), _> = loop {
3239                         let (counterparty_node_id, id) = match self.short_to_chan_info.read().unwrap().get(&path.hops.first().unwrap().short_channel_id) {
3240                                 None => return Err(APIError::ChannelUnavailable{err: "No channel available with first hop!".to_owned()}),
3241                                 Some((cp_id, chan_id)) => (cp_id.clone(), chan_id.clone()),
3242                         };
3243
3244                         let per_peer_state = self.per_peer_state.read().unwrap();
3245                         let peer_state_mutex = per_peer_state.get(&counterparty_node_id)
3246                                 .ok_or_else(|| APIError::ChannelUnavailable{err: "No peer matching the path's first hop found!".to_owned() })?;
3247                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
3248                         let peer_state = &mut *peer_state_lock;
3249                         if let hash_map::Entry::Occupied(mut chan) = peer_state.channel_by_id.entry(id) {
3250                                 if !chan.get().context.is_live() {
3251                                         return Err(APIError::ChannelUnavailable{err: "Peer for first hop currently disconnected".to_owned()});
3252                                 }
3253                                 let funding_txo = chan.get().context.get_funding_txo().unwrap();
3254                                 let send_res = chan.get_mut().send_htlc_and_commit(htlc_msat, payment_hash.clone(),
3255                                         htlc_cltv, HTLCSource::OutboundRoute {
3256                                                 path: path.clone(),
3257                                                 session_priv: session_priv.clone(),
3258                                                 first_hop_htlc_msat: htlc_msat,
3259                                                 payment_id,
3260                                         }, onion_packet, None, &self.fee_estimator, &self.logger);
3261                                 match break_chan_entry!(self, send_res, chan) {
3262                                         Some(monitor_update) => {
3263                                                 match handle_new_monitor_update!(self, funding_txo, monitor_update, peer_state_lock, peer_state, per_peer_state, chan) {
3264                                                         Err(e) => break Err(e),
3265                                                         Ok(false) => {
3266                                                                 // Note that MonitorUpdateInProgress here indicates (per function
3267                                                                 // docs) that we will resend the commitment update once monitor
3268                                                                 // updating completes. Therefore, we must return an error
3269                                                                 // indicating that it is unsafe to retry the payment wholesale,
3270                                                                 // which we do in the send_payment check for
3271                                                                 // MonitorUpdateInProgress, below.
3272                                                                 return Err(APIError::MonitorUpdateInProgress);
3273                                                         },
3274                                                         Ok(true) => {},
3275                                                 }
3276                                         },
3277                                         None => { },
3278                                 }
3279                         } else {
3280                                 // The channel was likely removed after we fetched the id from the
3281                                 // `short_to_chan_info` map, but before we successfully locked the
3282                                 // `channel_by_id` map.
3283                                 // This can occur as no consistency guarantees exists between the two maps.
3284                                 return Err(APIError::ChannelUnavailable{err: "No channel available with first hop!".to_owned()});
3285                         }
3286                         return Ok(());
3287                 };
3288
3289                 match handle_error!(self, err, path.hops.first().unwrap().pubkey) {
3290                         Ok(_) => unreachable!(),
3291                         Err(e) => {
3292                                 Err(APIError::ChannelUnavailable { err: e.err })
3293                         },
3294                 }
3295         }
3296
3297         /// Sends a payment along a given route.
3298         ///
3299         /// Value parameters are provided via the last hop in route, see documentation for [`RouteHop`]
3300         /// fields for more info.
3301         ///
3302         /// May generate [`UpdateHTLCs`] message(s) event on success, which should be relayed (e.g. via
3303         /// [`PeerManager::process_events`]).
3304         ///
3305         /// # Avoiding Duplicate Payments
3306         ///
3307         /// If a pending payment is currently in-flight with the same [`PaymentId`] provided, this
3308         /// method will error with an [`APIError::InvalidRoute`]. Note, however, that once a payment
3309         /// is no longer pending (either via [`ChannelManager::abandon_payment`], or handling of an
3310         /// [`Event::PaymentSent`] or [`Event::PaymentFailed`]) LDK will not stop you from sending a
3311         /// second payment with the same [`PaymentId`].
3312         ///
3313         /// Thus, in order to ensure duplicate payments are not sent, you should implement your own
3314         /// tracking of payments, including state to indicate once a payment has completed. Because you
3315         /// should also ensure that [`PaymentHash`]es are not re-used, for simplicity, you should
3316         /// consider using the [`PaymentHash`] as the key for tracking payments. In that case, the
3317         /// [`PaymentId`] should be a copy of the [`PaymentHash`] bytes.
3318         ///
3319         /// Additionally, in the scenario where we begin the process of sending a payment, but crash
3320         /// before `send_payment` returns (or prior to [`ChannelMonitorUpdate`] persistence if you're
3321         /// using [`ChannelMonitorUpdateStatus::InProgress`]), the payment may be lost on restart. See
3322         /// [`ChannelManager::list_recent_payments`] for more information.
3323         ///
3324         /// # Possible Error States on [`PaymentSendFailure`]
3325         ///
3326         /// Each path may have a different return value, and [`PaymentSendFailure`] may return a `Vec` with
3327         /// each entry matching the corresponding-index entry in the route paths, see
3328         /// [`PaymentSendFailure`] for more info.
3329         ///
3330         /// In general, a path may raise:
3331         ///  * [`APIError::InvalidRoute`] when an invalid route or forwarding parameter (cltv_delta, fee,
3332         ///    node public key) is specified.
3333         ///  * [`APIError::ChannelUnavailable`] if the next-hop channel is not available for updates
3334         ///    (including due to previous monitor update failure or new permanent monitor update
3335         ///    failure).
3336         ///  * [`APIError::MonitorUpdateInProgress`] if a new monitor update failure prevented sending the
3337         ///    relevant updates.
3338         ///
3339         /// Note that depending on the type of the [`PaymentSendFailure`] the HTLC may have been
3340         /// irrevocably committed to on our end. In such a case, do NOT retry the payment with a
3341         /// different route unless you intend to pay twice!
3342         ///
3343         /// [`RouteHop`]: crate::routing::router::RouteHop
3344         /// [`Event::PaymentSent`]: events::Event::PaymentSent
3345         /// [`Event::PaymentFailed`]: events::Event::PaymentFailed
3346         /// [`UpdateHTLCs`]: events::MessageSendEvent::UpdateHTLCs
3347         /// [`PeerManager::process_events`]: crate::ln::peer_handler::PeerManager::process_events
3348         /// [`ChannelMonitorUpdateStatus::InProgress`]: crate::chain::ChannelMonitorUpdateStatus::InProgress
3349         pub fn send_payment_with_route(&self, route: &Route, payment_hash: PaymentHash, recipient_onion: RecipientOnionFields, payment_id: PaymentId) -> Result<(), PaymentSendFailure> {
3350                 let best_block_height = self.best_block.read().unwrap().height();
3351                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3352                 self.pending_outbound_payments
3353                         .send_payment_with_route(route, payment_hash, recipient_onion, payment_id,
3354                                 &self.entropy_source, &self.node_signer, best_block_height,
3355                                 |args| self.send_payment_along_path(args))
3356         }
3357
3358         /// Similar to [`ChannelManager::send_payment_with_route`], but will automatically find a route based on
3359         /// `route_params` and retry failed payment paths based on `retry_strategy`.
3360         pub fn send_payment(&self, payment_hash: PaymentHash, recipient_onion: RecipientOnionFields, payment_id: PaymentId, route_params: RouteParameters, retry_strategy: Retry) -> Result<(), RetryableSendFailure> {
3361                 let best_block_height = self.best_block.read().unwrap().height();
3362                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3363                 self.pending_outbound_payments
3364                         .send_payment(payment_hash, recipient_onion, payment_id, retry_strategy, route_params,
3365                                 &self.router, self.list_usable_channels(), || self.compute_inflight_htlcs(),
3366                                 &self.entropy_source, &self.node_signer, best_block_height, &self.logger,
3367                                 &self.pending_events, |args| self.send_payment_along_path(args))
3368         }
3369
3370         #[cfg(test)]
3371         pub(super) fn test_send_payment_internal(&self, route: &Route, payment_hash: PaymentHash, recipient_onion: RecipientOnionFields, keysend_preimage: Option<PaymentPreimage>, payment_id: PaymentId, recv_value_msat: Option<u64>, onion_session_privs: Vec<[u8; 32]>) -> Result<(), PaymentSendFailure> {
3372                 let best_block_height = self.best_block.read().unwrap().height();
3373                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3374                 self.pending_outbound_payments.test_send_payment_internal(route, payment_hash, recipient_onion,
3375                         keysend_preimage, payment_id, recv_value_msat, onion_session_privs, &self.node_signer,
3376                         best_block_height, |args| self.send_payment_along_path(args))
3377         }
3378
3379         #[cfg(test)]
3380         pub(crate) fn test_add_new_pending_payment(&self, payment_hash: PaymentHash, recipient_onion: RecipientOnionFields, payment_id: PaymentId, route: &Route) -> Result<Vec<[u8; 32]>, PaymentSendFailure> {
3381                 let best_block_height = self.best_block.read().unwrap().height();
3382                 self.pending_outbound_payments.test_add_new_pending_payment(payment_hash, recipient_onion, payment_id, route, None, &self.entropy_source, best_block_height)
3383         }
3384
3385         #[cfg(test)]
3386         pub(crate) fn test_set_payment_metadata(&self, payment_id: PaymentId, new_payment_metadata: Option<Vec<u8>>) {
3387                 self.pending_outbound_payments.test_set_payment_metadata(payment_id, new_payment_metadata);
3388         }
3389
3390
3391         /// Signals that no further attempts for the given payment should occur. Useful if you have a
3392         /// pending outbound payment with retries remaining, but wish to stop retrying the payment before
3393         /// retries are exhausted.
3394         ///
3395         /// # Event Generation
3396         ///
3397         /// If no [`Event::PaymentFailed`] event had been generated before, one will be generated as soon
3398         /// as there are no remaining pending HTLCs for this payment.
3399         ///
3400         /// Note that calling this method does *not* prevent a payment from succeeding. You must still
3401         /// wait until you receive either a [`Event::PaymentFailed`] or [`Event::PaymentSent`] event to
3402         /// determine the ultimate status of a payment.
3403         ///
3404         /// # Requested Invoices
3405         ///
3406         /// In the case of paying a [`Bolt12Invoice`], abandoning the payment prior to receiving the
3407         /// invoice will result in an [`Event::InvoiceRequestFailed`] and prevent any attempts at paying
3408         /// it once received. The other events may only be generated once the invoice has been received.
3409         ///
3410         /// # Restart Behavior
3411         ///
3412         /// If an [`Event::PaymentFailed`] is generated and we restart without first persisting the
3413         /// [`ChannelManager`], another [`Event::PaymentFailed`] may be generated; likewise for
3414         /// [`Event::InvoiceRequestFailed`].
3415         ///
3416         /// [`Bolt12Invoice`]: crate::offers::invoice::Bolt12Invoice
3417         pub fn abandon_payment(&self, payment_id: PaymentId) {
3418                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3419                 self.pending_outbound_payments.abandon_payment(payment_id, PaymentFailureReason::UserAbandoned, &self.pending_events);
3420         }
3421
3422         /// Send a spontaneous payment, which is a payment that does not require the recipient to have
3423         /// generated an invoice. Optionally, you may specify the preimage. If you do choose to specify
3424         /// the preimage, it must be a cryptographically secure random value that no intermediate node
3425         /// would be able to guess -- otherwise, an intermediate node may claim the payment and it will
3426         /// never reach the recipient.
3427         ///
3428         /// See [`send_payment`] documentation for more details on the return value of this function
3429         /// and idempotency guarantees provided by the [`PaymentId`] key.
3430         ///
3431         /// Similar to regular payments, you MUST NOT reuse a `payment_preimage` value. See
3432         /// [`send_payment`] for more information about the risks of duplicate preimage usage.
3433         ///
3434         /// [`send_payment`]: Self::send_payment
3435         pub fn send_spontaneous_payment(&self, route: &Route, payment_preimage: Option<PaymentPreimage>, recipient_onion: RecipientOnionFields, payment_id: PaymentId) -> Result<PaymentHash, PaymentSendFailure> {
3436                 let best_block_height = self.best_block.read().unwrap().height();
3437                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3438                 self.pending_outbound_payments.send_spontaneous_payment_with_route(
3439                         route, payment_preimage, recipient_onion, payment_id, &self.entropy_source,
3440                         &self.node_signer, best_block_height, |args| self.send_payment_along_path(args))
3441         }
3442
3443         /// Similar to [`ChannelManager::send_spontaneous_payment`], but will automatically find a route
3444         /// based on `route_params` and retry failed payment paths based on `retry_strategy`.
3445         ///
3446         /// See [`PaymentParameters::for_keysend`] for help in constructing `route_params` for spontaneous
3447         /// payments.
3448         ///
3449         /// [`PaymentParameters::for_keysend`]: crate::routing::router::PaymentParameters::for_keysend
3450         pub fn send_spontaneous_payment_with_retry(&self, payment_preimage: Option<PaymentPreimage>, recipient_onion: RecipientOnionFields, payment_id: PaymentId, route_params: RouteParameters, retry_strategy: Retry) -> Result<PaymentHash, RetryableSendFailure> {
3451                 let best_block_height = self.best_block.read().unwrap().height();
3452                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3453                 self.pending_outbound_payments.send_spontaneous_payment(payment_preimage, recipient_onion,
3454                         payment_id, retry_strategy, route_params, &self.router, self.list_usable_channels(),
3455                         || self.compute_inflight_htlcs(),  &self.entropy_source, &self.node_signer, best_block_height,
3456                         &self.logger, &self.pending_events, |args| self.send_payment_along_path(args))
3457         }
3458
3459         /// Send a payment that is probing the given route for liquidity. We calculate the
3460         /// [`PaymentHash`] of probes based on a static secret and a random [`PaymentId`], which allows
3461         /// us to easily discern them from real payments.
3462         pub fn send_probe(&self, path: Path) -> Result<(PaymentHash, PaymentId), PaymentSendFailure> {
3463                 let best_block_height = self.best_block.read().unwrap().height();
3464                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3465                 self.pending_outbound_payments.send_probe(path, self.probing_cookie_secret,
3466                         &self.entropy_source, &self.node_signer, best_block_height,
3467                         |args| self.send_payment_along_path(args))
3468         }
3469
3470         /// Returns whether a payment with the given [`PaymentHash`] and [`PaymentId`] is, in fact, a
3471         /// payment probe.
3472         #[cfg(test)]
3473         pub(crate) fn payment_is_probe(&self, payment_hash: &PaymentHash, payment_id: &PaymentId) -> bool {
3474                 outbound_payment::payment_is_probe(payment_hash, payment_id, self.probing_cookie_secret)
3475         }
3476
3477         /// Handles the generation of a funding transaction, optionally (for tests) with a function
3478         /// which checks the correctness of the funding transaction given the associated channel.
3479         fn funding_transaction_generated_intern<FundingOutput: Fn(&OutboundV1Channel<SP>, &Transaction) -> Result<OutPoint, APIError>>(
3480                 &self, temporary_channel_id: &ChannelId, counterparty_node_id: &PublicKey, funding_transaction: Transaction, find_funding_output: FundingOutput
3481         ) -> Result<(), APIError> {
3482                 let per_peer_state = self.per_peer_state.read().unwrap();
3483                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
3484                         .ok_or_else(|| APIError::ChannelUnavailable { err: format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id) })?;
3485
3486                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
3487                 let peer_state = &mut *peer_state_lock;
3488                 let (chan, msg) = match peer_state.outbound_v1_channel_by_id.remove(&temporary_channel_id) {
3489                         Some(chan) => {
3490                                 let funding_txo = find_funding_output(&chan, &funding_transaction)?;
3491
3492                                 let funding_res = chan.get_funding_created(funding_transaction, funding_txo, &self.logger)
3493                                         .map_err(|(mut chan, e)| if let ChannelError::Close(msg) = e {
3494                                                 let channel_id = chan.context.channel_id();
3495                                                 let user_id = chan.context.get_user_id();
3496                                                 let shutdown_res = chan.context.force_shutdown(false);
3497                                                 let channel_capacity = chan.context.get_value_satoshis();
3498                                                 (chan, MsgHandleErrInternal::from_finish_shutdown(msg, channel_id, user_id, shutdown_res, None, channel_capacity))
3499                                         } else { unreachable!(); });
3500                                 match funding_res {
3501                                         Ok((chan, funding_msg)) => (chan, funding_msg),
3502                                         Err((chan, err)) => {
3503                                                 mem::drop(peer_state_lock);
3504                                                 mem::drop(per_peer_state);
3505
3506                                                 let _: Result<(), _> = handle_error!(self, Err(err), chan.context.get_counterparty_node_id());
3507                                                 return Err(APIError::ChannelUnavailable {
3508                                                         err: "Signer refused to sign the initial commitment transaction".to_owned()
3509                                                 });
3510                                         },
3511                                 }
3512                         },
3513                         None => {
3514                                 return Err(APIError::ChannelUnavailable {
3515                                         err: format!(
3516                                                 "Channel with id {} not found for the passed counterparty node_id {}",
3517                                                 temporary_channel_id, counterparty_node_id),
3518                                 })
3519                         },
3520                 };
3521
3522                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendFundingCreated {
3523                         node_id: chan.context.get_counterparty_node_id(),
3524                         msg,
3525                 });
3526                 match peer_state.channel_by_id.entry(chan.context.channel_id()) {
3527                         hash_map::Entry::Occupied(_) => {
3528                                 panic!("Generated duplicate funding txid?");
3529                         },
3530                         hash_map::Entry::Vacant(e) => {
3531                                 let mut id_to_peer = self.id_to_peer.lock().unwrap();
3532                                 if id_to_peer.insert(chan.context.channel_id(), chan.context.get_counterparty_node_id()).is_some() {
3533                                         panic!("id_to_peer map already contained funding txid, which shouldn't be possible");
3534                                 }
3535                                 e.insert(chan);
3536                         }
3537                 }
3538                 Ok(())
3539         }
3540
3541         #[cfg(test)]
3542         pub(crate) fn funding_transaction_generated_unchecked(&self, temporary_channel_id: &ChannelId, counterparty_node_id: &PublicKey, funding_transaction: Transaction, output_index: u16) -> Result<(), APIError> {
3543                 self.funding_transaction_generated_intern(temporary_channel_id, counterparty_node_id, funding_transaction, |_, tx| {
3544                         Ok(OutPoint { txid: tx.txid(), index: output_index })
3545                 })
3546         }
3547
3548         /// Call this upon creation of a funding transaction for the given channel.
3549         ///
3550         /// Returns an [`APIError::APIMisuseError`] if the funding_transaction spent non-SegWit outputs
3551         /// or if no output was found which matches the parameters in [`Event::FundingGenerationReady`].
3552         ///
3553         /// Returns [`APIError::APIMisuseError`] if the funding transaction is not final for propagation
3554         /// across the p2p network.
3555         ///
3556         /// Returns [`APIError::ChannelUnavailable`] if a funding transaction has already been provided
3557         /// for the channel or if the channel has been closed as indicated by [`Event::ChannelClosed`].
3558         ///
3559         /// May panic if the output found in the funding transaction is duplicative with some other
3560         /// channel (note that this should be trivially prevented by using unique funding transaction
3561         /// keys per-channel).
3562         ///
3563         /// Do NOT broadcast the funding transaction yourself. When we have safely received our
3564         /// counterparty's signature the funding transaction will automatically be broadcast via the
3565         /// [`BroadcasterInterface`] provided when this `ChannelManager` was constructed.
3566         ///
3567         /// Note that this includes RBF or similar transaction replacement strategies - lightning does
3568         /// not currently support replacing a funding transaction on an existing channel. Instead,
3569         /// create a new channel with a conflicting funding transaction.
3570         ///
3571         /// Note to keep the miner incentives aligned in moving the blockchain forward, we recommend
3572         /// the wallet software generating the funding transaction to apply anti-fee sniping as
3573         /// implemented by Bitcoin Core wallet. See <https://bitcoinops.org/en/topics/fee-sniping/>
3574         /// for more details.
3575         ///
3576         /// [`Event::FundingGenerationReady`]: crate::events::Event::FundingGenerationReady
3577         /// [`Event::ChannelClosed`]: crate::events::Event::ChannelClosed
3578         pub fn funding_transaction_generated(&self, temporary_channel_id: &ChannelId, counterparty_node_id: &PublicKey, funding_transaction: Transaction) -> Result<(), APIError> {
3579                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3580
3581                 if !funding_transaction.is_coin_base() {
3582                         for inp in funding_transaction.input.iter() {
3583                                 if inp.witness.is_empty() {
3584                                         return Err(APIError::APIMisuseError {
3585                                                 err: "Funding transaction must be fully signed and spend Segwit outputs".to_owned()
3586                                         });
3587                                 }
3588                         }
3589                 }
3590                 {
3591                         let height = self.best_block.read().unwrap().height();
3592                         // Transactions are evaluated as final by network mempools if their locktime is strictly
3593                         // lower than the next block height. However, the modules constituting our Lightning
3594                         // node might not have perfect sync about their blockchain views. Thus, if the wallet
3595                         // module is ahead of LDK, only allow one more block of headroom.
3596                         if !funding_transaction.input.iter().all(|input| input.sequence == Sequence::MAX) && LockTime::from(funding_transaction.lock_time).is_block_height() && funding_transaction.lock_time.0 > height + 1 {
3597                                 return Err(APIError::APIMisuseError {
3598                                         err: "Funding transaction absolute timelock is non-final".to_owned()
3599                                 });
3600                         }
3601                 }
3602                 self.funding_transaction_generated_intern(temporary_channel_id, counterparty_node_id, funding_transaction, |chan, tx| {
3603                         if tx.output.len() > u16::max_value() as usize {
3604                                 return Err(APIError::APIMisuseError {
3605                                         err: "Transaction had more than 2^16 outputs, which is not supported".to_owned()
3606                                 });
3607                         }
3608
3609                         let mut output_index = None;
3610                         let expected_spk = chan.context.get_funding_redeemscript().to_v0_p2wsh();
3611                         for (idx, outp) in tx.output.iter().enumerate() {
3612                                 if outp.script_pubkey == expected_spk && outp.value == chan.context.get_value_satoshis() {
3613                                         if output_index.is_some() {
3614                                                 return Err(APIError::APIMisuseError {
3615                                                         err: "Multiple outputs matched the expected script and value".to_owned()
3616                                                 });
3617                                         }
3618                                         output_index = Some(idx as u16);
3619                                 }
3620                         }
3621                         if output_index.is_none() {
3622                                 return Err(APIError::APIMisuseError {
3623                                         err: "No output matched the script_pubkey and value in the FundingGenerationReady event".to_owned()
3624                                 });
3625                         }
3626                         Ok(OutPoint { txid: tx.txid(), index: output_index.unwrap() })
3627                 })
3628         }
3629
3630         /// Atomically applies partial updates to the [`ChannelConfig`] of the given channels.
3631         ///
3632         /// Once the updates are applied, each eligible channel (advertised with a known short channel
3633         /// ID and a change in [`forwarding_fee_proportional_millionths`], [`forwarding_fee_base_msat`],
3634         /// or [`cltv_expiry_delta`]) has a [`BroadcastChannelUpdate`] event message generated
3635         /// containing the new [`ChannelUpdate`] message which should be broadcast to the network.
3636         ///
3637         /// Returns [`ChannelUnavailable`] when a channel is not found or an incorrect
3638         /// `counterparty_node_id` is provided.
3639         ///
3640         /// Returns [`APIMisuseError`] when a [`cltv_expiry_delta`] update is to be applied with a value
3641         /// below [`MIN_CLTV_EXPIRY_DELTA`].
3642         ///
3643         /// If an error is returned, none of the updates should be considered applied.
3644         ///
3645         /// [`forwarding_fee_proportional_millionths`]: ChannelConfig::forwarding_fee_proportional_millionths
3646         /// [`forwarding_fee_base_msat`]: ChannelConfig::forwarding_fee_base_msat
3647         /// [`cltv_expiry_delta`]: ChannelConfig::cltv_expiry_delta
3648         /// [`BroadcastChannelUpdate`]: events::MessageSendEvent::BroadcastChannelUpdate
3649         /// [`ChannelUpdate`]: msgs::ChannelUpdate
3650         /// [`ChannelUnavailable`]: APIError::ChannelUnavailable
3651         /// [`APIMisuseError`]: APIError::APIMisuseError
3652         pub fn update_partial_channel_config(
3653                 &self, counterparty_node_id: &PublicKey, channel_ids: &[ChannelId], config_update: &ChannelConfigUpdate,
3654         ) -> Result<(), APIError> {
3655                 if config_update.cltv_expiry_delta.map(|delta| delta < MIN_CLTV_EXPIRY_DELTA).unwrap_or(false) {
3656                         return Err(APIError::APIMisuseError {
3657                                 err: format!("The chosen CLTV expiry delta is below the minimum of {}", MIN_CLTV_EXPIRY_DELTA),
3658                         });
3659                 }
3660
3661                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3662                 let per_peer_state = self.per_peer_state.read().unwrap();
3663                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
3664                         .ok_or_else(|| APIError::ChannelUnavailable { err: format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id) })?;
3665                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
3666                 let peer_state = &mut *peer_state_lock;
3667                 for channel_id in channel_ids {
3668                         if !peer_state.has_channel(channel_id) {
3669                                 return Err(APIError::ChannelUnavailable {
3670                                         err: format!("Channel with ID {} was not found for the passed counterparty_node_id {}", channel_id, counterparty_node_id),
3671                                 });
3672                         };
3673                 }
3674                 for channel_id in channel_ids {
3675                         if let Some(channel) = peer_state.channel_by_id.get_mut(channel_id) {
3676                                 let mut config = channel.context.config();
3677                                 config.apply(config_update);
3678                                 if !channel.context.update_config(&config) {
3679                                         continue;
3680                                 }
3681                                 if let Ok(msg) = self.get_channel_update_for_broadcast(channel) {
3682                                         peer_state.pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate { msg });
3683                                 } else if let Ok(msg) = self.get_channel_update_for_unicast(channel) {
3684                                         peer_state.pending_msg_events.push(events::MessageSendEvent::SendChannelUpdate {
3685                                                 node_id: channel.context.get_counterparty_node_id(),
3686                                                 msg,
3687                                         });
3688                                 }
3689                                 continue;
3690                         }
3691
3692                         let context = if let Some(channel) = peer_state.inbound_v1_channel_by_id.get_mut(channel_id) {
3693                                 &mut channel.context
3694                         } else if let Some(channel) = peer_state.outbound_v1_channel_by_id.get_mut(channel_id) {
3695                                 &mut channel.context
3696                         } else {
3697                                 // This should not be reachable as we've already checked for non-existence in the previous channel_id loop.
3698                                 debug_assert!(false);
3699                                 return Err(APIError::ChannelUnavailable {
3700                                         err: format!(
3701                                                 "Channel with ID {} for passed counterparty_node_id {} disappeared after we confirmed its existence - this should not be reachable!",
3702                                                 channel_id, counterparty_node_id),
3703                                 });
3704                         };
3705                         let mut config = context.config();
3706                         config.apply(config_update);
3707                         // We update the config, but we MUST NOT broadcast a `channel_update` before `channel_ready`
3708                         // which would be the case for pending inbound/outbound channels.
3709                         context.update_config(&config);
3710                 }
3711                 Ok(())
3712         }
3713
3714         /// Atomically updates the [`ChannelConfig`] for the given channels.
3715         ///
3716         /// Once the updates are applied, each eligible channel (advertised with a known short channel
3717         /// ID and a change in [`forwarding_fee_proportional_millionths`], [`forwarding_fee_base_msat`],
3718         /// or [`cltv_expiry_delta`]) has a [`BroadcastChannelUpdate`] event message generated
3719         /// containing the new [`ChannelUpdate`] message which should be broadcast to the network.
3720         ///
3721         /// Returns [`ChannelUnavailable`] when a channel is not found or an incorrect
3722         /// `counterparty_node_id` is provided.
3723         ///
3724         /// Returns [`APIMisuseError`] when a [`cltv_expiry_delta`] update is to be applied with a value
3725         /// below [`MIN_CLTV_EXPIRY_DELTA`].
3726         ///
3727         /// If an error is returned, none of the updates should be considered applied.
3728         ///
3729         /// [`forwarding_fee_proportional_millionths`]: ChannelConfig::forwarding_fee_proportional_millionths
3730         /// [`forwarding_fee_base_msat`]: ChannelConfig::forwarding_fee_base_msat
3731         /// [`cltv_expiry_delta`]: ChannelConfig::cltv_expiry_delta
3732         /// [`BroadcastChannelUpdate`]: events::MessageSendEvent::BroadcastChannelUpdate
3733         /// [`ChannelUpdate`]: msgs::ChannelUpdate
3734         /// [`ChannelUnavailable`]: APIError::ChannelUnavailable
3735         /// [`APIMisuseError`]: APIError::APIMisuseError
3736         pub fn update_channel_config(
3737                 &self, counterparty_node_id: &PublicKey, channel_ids: &[ChannelId], config: &ChannelConfig,
3738         ) -> Result<(), APIError> {
3739                 return self.update_partial_channel_config(counterparty_node_id, channel_ids, &(*config).into());
3740         }
3741
3742         /// Attempts to forward an intercepted HTLC over the provided channel id and with the provided
3743         /// amount to forward. Should only be called in response to an [`HTLCIntercepted`] event.
3744         ///
3745         /// Intercepted HTLCs can be useful for Lightning Service Providers (LSPs) to open a just-in-time
3746         /// channel to a receiving node if the node lacks sufficient inbound liquidity.
3747         ///
3748         /// To make use of intercepted HTLCs, set [`UserConfig::accept_intercept_htlcs`] and use
3749         /// [`ChannelManager::get_intercept_scid`] to generate short channel id(s) to put in the
3750         /// receiver's invoice route hints. These route hints will signal to LDK to generate an
3751         /// [`HTLCIntercepted`] event when it receives the forwarded HTLC, and this method or
3752         /// [`ChannelManager::fail_intercepted_htlc`] MUST be called in response to the event.
3753         ///
3754         /// Note that LDK does not enforce fee requirements in `amt_to_forward_msat`, and will not stop
3755         /// you from forwarding more than you received. See
3756         /// [`HTLCIntercepted::expected_outbound_amount_msat`] for more on forwarding a different amount
3757         /// than expected.
3758         ///
3759         /// Errors if the event was not handled in time, in which case the HTLC was automatically failed
3760         /// backwards.
3761         ///
3762         /// [`UserConfig::accept_intercept_htlcs`]: crate::util::config::UserConfig::accept_intercept_htlcs
3763         /// [`HTLCIntercepted`]: events::Event::HTLCIntercepted
3764         /// [`HTLCIntercepted::expected_outbound_amount_msat`]: events::Event::HTLCIntercepted::expected_outbound_amount_msat
3765         // TODO: when we move to deciding the best outbound channel at forward time, only take
3766         // `next_node_id` and not `next_hop_channel_id`
3767         pub fn forward_intercepted_htlc(&self, intercept_id: InterceptId, next_hop_channel_id: &ChannelId, next_node_id: PublicKey, amt_to_forward_msat: u64) -> Result<(), APIError> {
3768                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3769
3770                 let next_hop_scid = {
3771                         let peer_state_lock = self.per_peer_state.read().unwrap();
3772                         let peer_state_mutex = peer_state_lock.get(&next_node_id)
3773                                 .ok_or_else(|| APIError::ChannelUnavailable { err: format!("Can't find a peer matching the passed counterparty node_id {}", next_node_id) })?;
3774                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
3775                         let peer_state = &mut *peer_state_lock;
3776                         match peer_state.channel_by_id.get(&next_hop_channel_id) {
3777                                 Some(chan) => {
3778                                         if !chan.context.is_usable() {
3779                                                 return Err(APIError::ChannelUnavailable {
3780                                                         err: format!("Channel with id {} not fully established", next_hop_channel_id)
3781                                                 })
3782                                         }
3783                                         chan.context.get_short_channel_id().unwrap_or(chan.context.outbound_scid_alias())
3784                                 },
3785                                 None => return Err(APIError::ChannelUnavailable {
3786                                         err: format!("Funded channel with id {} not found for the passed counterparty node_id {}. Channel may still be opening.",
3787                                                 next_hop_channel_id, next_node_id)
3788                                 })
3789                         }
3790                 };
3791
3792                 let payment = self.pending_intercepted_htlcs.lock().unwrap().remove(&intercept_id)
3793                         .ok_or_else(|| APIError::APIMisuseError {
3794                                 err: format!("Payment with intercept id {} not found", log_bytes!(intercept_id.0))
3795                         })?;
3796
3797                 let routing = match payment.forward_info.routing {
3798                         PendingHTLCRouting::Forward { onion_packet, .. } => {
3799                                 PendingHTLCRouting::Forward { onion_packet, short_channel_id: next_hop_scid }
3800                         },
3801                         _ => unreachable!() // Only `PendingHTLCRouting::Forward`s are intercepted
3802                 };
3803                 let skimmed_fee_msat =
3804                         payment.forward_info.outgoing_amt_msat.saturating_sub(amt_to_forward_msat);
3805                 let pending_htlc_info = PendingHTLCInfo {
3806                         skimmed_fee_msat: if skimmed_fee_msat == 0 { None } else { Some(skimmed_fee_msat) },
3807                         outgoing_amt_msat: amt_to_forward_msat, routing, ..payment.forward_info
3808                 };
3809
3810                 let mut per_source_pending_forward = [(
3811                         payment.prev_short_channel_id,
3812                         payment.prev_funding_outpoint,
3813                         payment.prev_user_channel_id,
3814                         vec![(pending_htlc_info, payment.prev_htlc_id)]
3815                 )];
3816                 self.forward_htlcs(&mut per_source_pending_forward);
3817                 Ok(())
3818         }
3819
3820         /// Fails the intercepted HTLC indicated by intercept_id. Should only be called in response to
3821         /// an [`HTLCIntercepted`] event. See [`ChannelManager::forward_intercepted_htlc`].
3822         ///
3823         /// Errors if the event was not handled in time, in which case the HTLC was automatically failed
3824         /// backwards.
3825         ///
3826         /// [`HTLCIntercepted`]: events::Event::HTLCIntercepted
3827         pub fn fail_intercepted_htlc(&self, intercept_id: InterceptId) -> Result<(), APIError> {
3828                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3829
3830                 let payment = self.pending_intercepted_htlcs.lock().unwrap().remove(&intercept_id)
3831                         .ok_or_else(|| APIError::APIMisuseError {
3832                                 err: format!("Payment with intercept id {} not found", log_bytes!(intercept_id.0))
3833                         })?;
3834
3835                 if let PendingHTLCRouting::Forward { short_channel_id, .. } = payment.forward_info.routing {
3836                         let htlc_source = HTLCSource::PreviousHopData(HTLCPreviousHopData {
3837                                 short_channel_id: payment.prev_short_channel_id,
3838                                 user_channel_id: Some(payment.prev_user_channel_id),
3839                                 outpoint: payment.prev_funding_outpoint,
3840                                 htlc_id: payment.prev_htlc_id,
3841                                 incoming_packet_shared_secret: payment.forward_info.incoming_shared_secret,
3842                                 phantom_shared_secret: None,
3843                         });
3844
3845                         let failure_reason = HTLCFailReason::from_failure_code(0x4000 | 10);
3846                         let destination = HTLCDestination::UnknownNextHop { requested_forward_scid: short_channel_id };
3847                         self.fail_htlc_backwards_internal(&htlc_source, &payment.forward_info.payment_hash, &failure_reason, destination);
3848                 } else { unreachable!() } // Only `PendingHTLCRouting::Forward`s are intercepted
3849
3850                 Ok(())
3851         }
3852
3853         /// Processes HTLCs which are pending waiting on random forward delay.
3854         ///
3855         /// Should only really ever be called in response to a PendingHTLCsForwardable event.
3856         /// Will likely generate further events.
3857         pub fn process_pending_htlc_forwards(&self) {
3858                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
3859
3860                 let mut new_events = VecDeque::new();
3861                 let mut failed_forwards = Vec::new();
3862                 let mut phantom_receives: Vec<(u64, OutPoint, u128, Vec<(PendingHTLCInfo, u64)>)> = Vec::new();
3863                 {
3864                         let mut forward_htlcs = HashMap::new();
3865                         mem::swap(&mut forward_htlcs, &mut self.forward_htlcs.lock().unwrap());
3866
3867                         for (short_chan_id, mut pending_forwards) in forward_htlcs {
3868                                 if short_chan_id != 0 {
3869                                         macro_rules! forwarding_channel_not_found {
3870                                                 () => {
3871                                                         for forward_info in pending_forwards.drain(..) {
3872                                                                 match forward_info {
3873                                                                         HTLCForwardInfo::AddHTLC(PendingAddHTLCInfo {
3874                                                                                 prev_short_channel_id, prev_htlc_id, prev_funding_outpoint, prev_user_channel_id,
3875                                                                                 forward_info: PendingHTLCInfo {
3876                                                                                         routing, incoming_shared_secret, payment_hash, outgoing_amt_msat,
3877                                                                                         outgoing_cltv_value, ..
3878                                                                                 }
3879                                                                         }) => {
3880                                                                                 macro_rules! failure_handler {
3881                                                                                         ($msg: expr, $err_code: expr, $err_data: expr, $phantom_ss: expr, $next_hop_unknown: expr) => {
3882                                                                                                 log_info!(self.logger, "Failed to accept/forward incoming HTLC: {}", $msg);
3883
3884                                                                                                 let htlc_source = HTLCSource::PreviousHopData(HTLCPreviousHopData {
3885                                                                                                         short_channel_id: prev_short_channel_id,
3886                                                                                                         user_channel_id: Some(prev_user_channel_id),
3887                                                                                                         outpoint: prev_funding_outpoint,
3888                                                                                                         htlc_id: prev_htlc_id,
3889                                                                                                         incoming_packet_shared_secret: incoming_shared_secret,
3890                                                                                                         phantom_shared_secret: $phantom_ss,
3891                                                                                                 });
3892
3893                                                                                                 let reason = if $next_hop_unknown {
3894                                                                                                         HTLCDestination::UnknownNextHop { requested_forward_scid: short_chan_id }
3895                                                                                                 } else {
3896                                                                                                         HTLCDestination::FailedPayment{ payment_hash }
3897                                                                                                 };
3898
3899                                                                                                 failed_forwards.push((htlc_source, payment_hash,
3900                                                                                                         HTLCFailReason::reason($err_code, $err_data),
3901                                                                                                         reason
3902                                                                                                 ));
3903                                                                                                 continue;
3904                                                                                         }
3905                                                                                 }
3906                                                                                 macro_rules! fail_forward {
3907                                                                                         ($msg: expr, $err_code: expr, $err_data: expr, $phantom_ss: expr) => {
3908                                                                                                 {
3909                                                                                                         failure_handler!($msg, $err_code, $err_data, $phantom_ss, true);
3910                                                                                                 }
3911                                                                                         }
3912                                                                                 }
3913                                                                                 macro_rules! failed_payment {
3914                                                                                         ($msg: expr, $err_code: expr, $err_data: expr, $phantom_ss: expr) => {
3915                                                                                                 {
3916                                                                                                         failure_handler!($msg, $err_code, $err_data, $phantom_ss, false);
3917                                                                                                 }
3918                                                                                         }
3919                                                                                 }
3920                                                                                 if let PendingHTLCRouting::Forward { onion_packet, .. } = routing {
3921                                                                                         let phantom_pubkey_res = self.node_signer.get_node_id(Recipient::PhantomNode);
3922                                                                                         if phantom_pubkey_res.is_ok() && fake_scid::is_valid_phantom(&self.fake_scid_rand_bytes, short_chan_id, &self.genesis_hash) {
3923                                                                                                 let phantom_shared_secret = self.node_signer.ecdh(Recipient::PhantomNode, &onion_packet.public_key.unwrap(), None).unwrap().secret_bytes();
3924                                                                                                 let next_hop = match onion_utils::decode_next_payment_hop(phantom_shared_secret, &onion_packet.hop_data, onion_packet.hmac, payment_hash) {
3925                                                                                                         Ok(res) => res,
3926                                                                                                         Err(onion_utils::OnionDecodeErr::Malformed { err_msg, err_code }) => {
3927                                                                                                                 let sha256_of_onion = Sha256::hash(&onion_packet.hop_data).into_inner();
3928                                                                                                                 // In this scenario, the phantom would have sent us an
3929                                                                                                                 // `update_fail_malformed_htlc`, meaning here we encrypt the error as
3930                                                                                                                 // if it came from us (the second-to-last hop) but contains the sha256
3931                                                                                                                 // of the onion.
3932                                                                                                                 failed_payment!(err_msg, err_code, sha256_of_onion.to_vec(), None);
3933                                                                                                         },
3934                                                                                                         Err(onion_utils::OnionDecodeErr::Relay { err_msg, err_code }) => {
3935                                                                                                                 failed_payment!(err_msg, err_code, Vec::new(), Some(phantom_shared_secret));
3936                                                                                                         },
3937                                                                                                 };
3938                                                                                                 match next_hop {
3939                                                                                                         onion_utils::Hop::Receive(hop_data) => {
3940                                                                                                                 match self.construct_recv_pending_htlc_info(hop_data,
3941                                                                                                                         incoming_shared_secret, payment_hash, outgoing_amt_msat,
3942                                                                                                                         outgoing_cltv_value, Some(phantom_shared_secret), false, None)
3943                                                                                                                 {
3944                                                                                                                         Ok(info) => phantom_receives.push((prev_short_channel_id, prev_funding_outpoint, prev_user_channel_id, vec![(info, prev_htlc_id)])),
3945                                                                                                                         Err(InboundOnionErr { err_code, err_data, msg }) => failed_payment!(msg, err_code, err_data, Some(phantom_shared_secret))
3946                                                                                                                 }
3947                                                                                                         },
3948                                                                                                         _ => panic!(),
3949                                                                                                 }
3950                                                                                         } else {
3951                                                                                                 fail_forward!(format!("Unknown short channel id {} for forward HTLC", short_chan_id), 0x4000 | 10, Vec::new(), None);
3952                                                                                         }
3953                                                                                 } else {
3954                                                                                         fail_forward!(format!("Unknown short channel id {} for forward HTLC", short_chan_id), 0x4000 | 10, Vec::new(), None);
3955                                                                                 }
3956                                                                         },
3957                                                                         HTLCForwardInfo::FailHTLC { .. } => {
3958                                                                                 // Channel went away before we could fail it. This implies
3959                                                                                 // the channel is now on chain and our counterparty is
3960                                                                                 // trying to broadcast the HTLC-Timeout, but that's their
3961                                                                                 // problem, not ours.
3962                                                                         }
3963                                                                 }
3964                                                         }
3965                                                 }
3966                                         }
3967                                         let (counterparty_node_id, forward_chan_id) = match self.short_to_chan_info.read().unwrap().get(&short_chan_id) {
3968                                                 Some((cp_id, chan_id)) => (cp_id.clone(), chan_id.clone()),
3969                                                 None => {
3970                                                         forwarding_channel_not_found!();
3971                                                         continue;
3972                                                 }
3973                                         };
3974                                         let per_peer_state = self.per_peer_state.read().unwrap();
3975                                         let peer_state_mutex_opt = per_peer_state.get(&counterparty_node_id);
3976                                         if peer_state_mutex_opt.is_none() {
3977                                                 forwarding_channel_not_found!();
3978                                                 continue;
3979                                         }
3980                                         let mut peer_state_lock = peer_state_mutex_opt.unwrap().lock().unwrap();
3981                                         let peer_state = &mut *peer_state_lock;
3982                                         match peer_state.channel_by_id.entry(forward_chan_id) {
3983                                                 hash_map::Entry::Vacant(_) => {
3984                                                         forwarding_channel_not_found!();
3985                                                         continue;
3986                                                 },
3987                                                 hash_map::Entry::Occupied(mut chan) => {
3988                                                         for forward_info in pending_forwards.drain(..) {
3989                                                                 match forward_info {
3990                                                                         HTLCForwardInfo::AddHTLC(PendingAddHTLCInfo {
3991                                                                                 prev_short_channel_id, prev_htlc_id, prev_funding_outpoint, prev_user_channel_id,
3992                                                                                 forward_info: PendingHTLCInfo {
3993                                                                                         incoming_shared_secret, payment_hash, outgoing_amt_msat, outgoing_cltv_value,
3994                                                                                         routing: PendingHTLCRouting::Forward { onion_packet, .. }, skimmed_fee_msat, ..
3995                                                                                 },
3996                                                                         }) => {
3997                                                                                 log_trace!(self.logger, "Adding HTLC from short id {} with payment_hash {} to channel with short id {} after delay", prev_short_channel_id, &payment_hash, short_chan_id);
3998                                                                                 let htlc_source = HTLCSource::PreviousHopData(HTLCPreviousHopData {
3999                                                                                         short_channel_id: prev_short_channel_id,
4000                                                                                         user_channel_id: Some(prev_user_channel_id),
4001                                                                                         outpoint: prev_funding_outpoint,
4002                                                                                         htlc_id: prev_htlc_id,
4003                                                                                         incoming_packet_shared_secret: incoming_shared_secret,
4004                                                                                         // Phantom payments are only PendingHTLCRouting::Receive.
4005                                                                                         phantom_shared_secret: None,
4006                                                                                 });
4007                                                                                 if let Err(e) = chan.get_mut().queue_add_htlc(outgoing_amt_msat,
4008                                                                                         payment_hash, outgoing_cltv_value, htlc_source.clone(),
4009                                                                                         onion_packet, skimmed_fee_msat, &self.fee_estimator,
4010                                                                                         &self.logger)
4011                                                                                 {
4012                                                                                         if let ChannelError::Ignore(msg) = e {
4013                                                                                                 log_trace!(self.logger, "Failed to forward HTLC with payment_hash {}: {}", &payment_hash, msg);
4014                                                                                         } else {
4015                                                                                                 panic!("Stated return value requirements in send_htlc() were not met");
4016                                                                                         }
4017                                                                                         let (failure_code, data) = self.get_htlc_temp_fail_err_and_data(0x1000|7, short_chan_id, chan.get());
4018                                                                                         failed_forwards.push((htlc_source, payment_hash,
4019                                                                                                 HTLCFailReason::reason(failure_code, data),
4020                                                                                                 HTLCDestination::NextHopChannel { node_id: Some(chan.get().context.get_counterparty_node_id()), channel_id: forward_chan_id }
4021                                                                                         ));
4022                                                                                         continue;
4023                                                                                 }
4024                                                                         },
4025                                                                         HTLCForwardInfo::AddHTLC { .. } => {
4026                                                                                 panic!("short_channel_id != 0 should imply any pending_forward entries are of type Forward");
4027                                                                         },
4028                                                                         HTLCForwardInfo::FailHTLC { htlc_id, err_packet } => {
4029                                                                                 log_trace!(self.logger, "Failing HTLC back to channel with short id {} (backward HTLC ID {}) after delay", short_chan_id, htlc_id);
4030                                                                                 if let Err(e) = chan.get_mut().queue_fail_htlc(
4031                                                                                         htlc_id, err_packet, &self.logger
4032                                                                                 ) {
4033                                                                                         if let ChannelError::Ignore(msg) = e {
4034                                                                                                 log_trace!(self.logger, "Failed to fail HTLC with ID {} backwards to short_id {}: {}", htlc_id, short_chan_id, msg);
4035                                                                                         } else {
4036                                                                                                 panic!("Stated return value requirements in queue_fail_htlc() were not met");
4037                                                                                         }
4038                                                                                         // fail-backs are best-effort, we probably already have one
4039                                                                                         // pending, and if not that's OK, if not, the channel is on
4040                                                                                         // the chain and sending the HTLC-Timeout is their problem.
4041                                                                                         continue;
4042                                                                                 }
4043                                                                         },
4044                                                                 }
4045                                                         }
4046                                                 }
4047                                         }
4048                                 } else {
4049                                         'next_forwardable_htlc: for forward_info in pending_forwards.drain(..) {
4050                                                 match forward_info {
4051                                                         HTLCForwardInfo::AddHTLC(PendingAddHTLCInfo {
4052                                                                 prev_short_channel_id, prev_htlc_id, prev_funding_outpoint, prev_user_channel_id,
4053                                                                 forward_info: PendingHTLCInfo {
4054                                                                         routing, incoming_shared_secret, payment_hash, incoming_amt_msat, outgoing_amt_msat,
4055                                                                         skimmed_fee_msat, ..
4056                                                                 }
4057                                                         }) => {
4058                                                                 let (cltv_expiry, onion_payload, payment_data, phantom_shared_secret, mut onion_fields) = match routing {
4059                                                                         PendingHTLCRouting::Receive { payment_data, payment_metadata, incoming_cltv_expiry, phantom_shared_secret, custom_tlvs } => {
4060                                                                                 let _legacy_hop_data = Some(payment_data.clone());
4061                                                                                 let onion_fields = RecipientOnionFields { payment_secret: Some(payment_data.payment_secret),
4062                                                                                                 payment_metadata, custom_tlvs };
4063                                                                                 (incoming_cltv_expiry, OnionPayload::Invoice { _legacy_hop_data },
4064                                                                                         Some(payment_data), phantom_shared_secret, onion_fields)
4065                                                                         },
4066                                                                         PendingHTLCRouting::ReceiveKeysend { payment_data, payment_preimage, payment_metadata, incoming_cltv_expiry, custom_tlvs } => {
4067                                                                                 let onion_fields = RecipientOnionFields {
4068                                                                                         payment_secret: payment_data.as_ref().map(|data| data.payment_secret),
4069                                                                                         payment_metadata,
4070                                                                                         custom_tlvs,
4071                                                                                 };
4072                                                                                 (incoming_cltv_expiry, OnionPayload::Spontaneous(payment_preimage),
4073                                                                                         payment_data, None, onion_fields)
4074                                                                         },
4075                                                                         _ => {
4076                                                                                 panic!("short_channel_id == 0 should imply any pending_forward entries are of type Receive");
4077                                                                         }
4078                                                                 };
4079                                                                 let claimable_htlc = ClaimableHTLC {
4080                                                                         prev_hop: HTLCPreviousHopData {
4081                                                                                 short_channel_id: prev_short_channel_id,
4082                                                                                 user_channel_id: Some(prev_user_channel_id),
4083                                                                                 outpoint: prev_funding_outpoint,
4084                                                                                 htlc_id: prev_htlc_id,
4085                                                                                 incoming_packet_shared_secret: incoming_shared_secret,
4086                                                                                 phantom_shared_secret,
4087                                                                         },
4088                                                                         // We differentiate the received value from the sender intended value
4089                                                                         // if possible so that we don't prematurely mark MPP payments complete
4090                                                                         // if routing nodes overpay
4091                                                                         value: incoming_amt_msat.unwrap_or(outgoing_amt_msat),
4092                                                                         sender_intended_value: outgoing_amt_msat,
4093                                                                         timer_ticks: 0,
4094                                                                         total_value_received: None,
4095                                                                         total_msat: if let Some(data) = &payment_data { data.total_msat } else { outgoing_amt_msat },
4096                                                                         cltv_expiry,
4097                                                                         onion_payload,
4098                                                                         counterparty_skimmed_fee_msat: skimmed_fee_msat,
4099                                                                 };
4100
4101                                                                 let mut committed_to_claimable = false;
4102
4103                                                                 macro_rules! fail_htlc {
4104                                                                         ($htlc: expr, $payment_hash: expr) => {
4105                                                                                 debug_assert!(!committed_to_claimable);
4106                                                                                 let mut htlc_msat_height_data = $htlc.value.to_be_bytes().to_vec();
4107                                                                                 htlc_msat_height_data.extend_from_slice(
4108                                                                                         &self.best_block.read().unwrap().height().to_be_bytes(),
4109                                                                                 );
4110                                                                                 failed_forwards.push((HTLCSource::PreviousHopData(HTLCPreviousHopData {
4111                                                                                                 short_channel_id: $htlc.prev_hop.short_channel_id,
4112                                                                                                 user_channel_id: $htlc.prev_hop.user_channel_id,
4113                                                                                                 outpoint: prev_funding_outpoint,
4114                                                                                                 htlc_id: $htlc.prev_hop.htlc_id,
4115                                                                                                 incoming_packet_shared_secret: $htlc.prev_hop.incoming_packet_shared_secret,
4116                                                                                                 phantom_shared_secret,
4117                                                                                         }), payment_hash,
4118                                                                                         HTLCFailReason::reason(0x4000 | 15, htlc_msat_height_data),
4119                                                                                         HTLCDestination::FailedPayment { payment_hash: $payment_hash },
4120                                                                                 ));
4121                                                                                 continue 'next_forwardable_htlc;
4122                                                                         }
4123                                                                 }
4124                                                                 let phantom_shared_secret = claimable_htlc.prev_hop.phantom_shared_secret;
4125                                                                 let mut receiver_node_id = self.our_network_pubkey;
4126                                                                 if phantom_shared_secret.is_some() {
4127                                                                         receiver_node_id = self.node_signer.get_node_id(Recipient::PhantomNode)
4128                                                                                 .expect("Failed to get node_id for phantom node recipient");
4129                                                                 }
4130
4131                                                                 macro_rules! check_total_value {
4132                                                                         ($purpose: expr) => {{
4133                                                                                 let mut payment_claimable_generated = false;
4134                                                                                 let is_keysend = match $purpose {
4135                                                                                         events::PaymentPurpose::SpontaneousPayment(_) => true,
4136                                                                                         events::PaymentPurpose::InvoicePayment { .. } => false,
4137                                                                                 };
4138                                                                                 let mut claimable_payments = self.claimable_payments.lock().unwrap();
4139                                                                                 if claimable_payments.pending_claiming_payments.contains_key(&payment_hash) {
4140                                                                                         fail_htlc!(claimable_htlc, payment_hash);
4141                                                                                 }
4142                                                                                 let ref mut claimable_payment = claimable_payments.claimable_payments
4143                                                                                         .entry(payment_hash)
4144                                                                                         // Note that if we insert here we MUST NOT fail_htlc!()
4145                                                                                         .or_insert_with(|| {
4146                                                                                                 committed_to_claimable = true;
4147                                                                                                 ClaimablePayment {
4148                                                                                                         purpose: $purpose.clone(), htlcs: Vec::new(), onion_fields: None,
4149                                                                                                 }
4150                                                                                         });
4151                                                                                 if $purpose != claimable_payment.purpose {
4152                                                                                         let log_keysend = |keysend| if keysend { "keysend" } else { "non-keysend" };
4153                                                                                         log_trace!(self.logger, "Failing new {} HTLC with payment_hash {} as we already had an existing {} HTLC with the same payment hash", log_keysend(is_keysend), &payment_hash, log_keysend(!is_keysend));
4154                                                                                         fail_htlc!(claimable_htlc, payment_hash);
4155                                                                                 }
4156                                                                                 if !self.default_configuration.accept_mpp_keysend && is_keysend && !claimable_payment.htlcs.is_empty() {
4157                                                                                         log_trace!(self.logger, "Failing new keysend HTLC with payment_hash {} as we already had an existing keysend HTLC with the same payment hash and our config states we don't accept MPP keysend", &payment_hash);
4158                                                                                         fail_htlc!(claimable_htlc, payment_hash);
4159                                                                                 }
4160                                                                                 if let Some(earlier_fields) = &mut claimable_payment.onion_fields {
4161                                                                                         if earlier_fields.check_merge(&mut onion_fields).is_err() {
4162                                                                                                 fail_htlc!(claimable_htlc, payment_hash);
4163                                                                                         }
4164                                                                                 } else {
4165                                                                                         claimable_payment.onion_fields = Some(onion_fields);
4166                                                                                 }
4167                                                                                 let ref mut htlcs = &mut claimable_payment.htlcs;
4168                                                                                 let mut total_value = claimable_htlc.sender_intended_value;
4169                                                                                 let mut earliest_expiry = claimable_htlc.cltv_expiry;
4170                                                                                 for htlc in htlcs.iter() {
4171                                                                                         total_value += htlc.sender_intended_value;
4172                                                                                         earliest_expiry = cmp::min(earliest_expiry, htlc.cltv_expiry);
4173                                                                                         if htlc.total_msat != claimable_htlc.total_msat {
4174                                                                                                 log_trace!(self.logger, "Failing HTLCs with payment_hash {} as the HTLCs had inconsistent total values (eg {} and {})",
4175                                                                                                         &payment_hash, claimable_htlc.total_msat, htlc.total_msat);
4176                                                                                                 total_value = msgs::MAX_VALUE_MSAT;
4177                                                                                         }
4178                                                                                         if total_value >= msgs::MAX_VALUE_MSAT { break; }
4179                                                                                 }
4180                                                                                 // The condition determining whether an MPP is complete must
4181                                                                                 // match exactly the condition used in `timer_tick_occurred`
4182                                                                                 if total_value >= msgs::MAX_VALUE_MSAT {
4183                                                                                         fail_htlc!(claimable_htlc, payment_hash);
4184                                                                                 } else if total_value - claimable_htlc.sender_intended_value >= claimable_htlc.total_msat {
4185                                                                                         log_trace!(self.logger, "Failing HTLC with payment_hash {} as payment is already claimable",
4186                                                                                                 &payment_hash);
4187                                                                                         fail_htlc!(claimable_htlc, payment_hash);
4188                                                                                 } else if total_value >= claimable_htlc.total_msat {
4189                                                                                         #[allow(unused_assignments)] {
4190                                                                                                 committed_to_claimable = true;
4191                                                                                         }
4192                                                                                         let prev_channel_id = prev_funding_outpoint.to_channel_id();
4193                                                                                         htlcs.push(claimable_htlc);
4194                                                                                         let amount_msat = htlcs.iter().map(|htlc| htlc.value).sum();
4195                                                                                         htlcs.iter_mut().for_each(|htlc| htlc.total_value_received = Some(amount_msat));
4196                                                                                         let counterparty_skimmed_fee_msat = htlcs.iter()
4197                                                                                                 .map(|htlc| htlc.counterparty_skimmed_fee_msat.unwrap_or(0)).sum();
4198                                                                                         debug_assert!(total_value.saturating_sub(amount_msat) <=
4199                                                                                                 counterparty_skimmed_fee_msat);
4200                                                                                         new_events.push_back((events::Event::PaymentClaimable {
4201                                                                                                 receiver_node_id: Some(receiver_node_id),
4202                                                                                                 payment_hash,
4203                                                                                                 purpose: $purpose,
4204                                                                                                 amount_msat,
4205                                                                                                 counterparty_skimmed_fee_msat,
4206                                                                                                 via_channel_id: Some(prev_channel_id),
4207                                                                                                 via_user_channel_id: Some(prev_user_channel_id),
4208                                                                                                 claim_deadline: Some(earliest_expiry - HTLC_FAIL_BACK_BUFFER),
4209                                                                                                 onion_fields: claimable_payment.onion_fields.clone(),
4210                                                                                         }, None));
4211                                                                                         payment_claimable_generated = true;
4212                                                                                 } else {
4213                                                                                         // Nothing to do - we haven't reached the total
4214                                                                                         // payment value yet, wait until we receive more
4215                                                                                         // MPP parts.
4216                                                                                         htlcs.push(claimable_htlc);
4217                                                                                         #[allow(unused_assignments)] {
4218                                                                                                 committed_to_claimable = true;
4219                                                                                         }
4220                                                                                 }
4221                                                                                 payment_claimable_generated
4222                                                                         }}
4223                                                                 }
4224
4225                                                                 // Check that the payment hash and secret are known. Note that we
4226                                                                 // MUST take care to handle the "unknown payment hash" and
4227                                                                 // "incorrect payment secret" cases here identically or we'd expose
4228                                                                 // that we are the ultimate recipient of the given payment hash.
4229                                                                 // Further, we must not expose whether we have any other HTLCs
4230                                                                 // associated with the same payment_hash pending or not.
4231                                                                 let mut payment_secrets = self.pending_inbound_payments.lock().unwrap();
4232                                                                 match payment_secrets.entry(payment_hash) {
4233                                                                         hash_map::Entry::Vacant(_) => {
4234                                                                                 match claimable_htlc.onion_payload {
4235                                                                                         OnionPayload::Invoice { .. } => {
4236                                                                                                 let payment_data = payment_data.unwrap();
4237                                                                                                 let (payment_preimage, min_final_cltv_expiry_delta) = match inbound_payment::verify(payment_hash, &payment_data, self.highest_seen_timestamp.load(Ordering::Acquire) as u64, &self.inbound_payment_key, &self.logger) {
4238                                                                                                         Ok(result) => result,
4239                                                                                                         Err(()) => {
4240                                                                                                                 log_trace!(self.logger, "Failing new HTLC with payment_hash {} as payment verification failed", &payment_hash);
4241                                                                                                                 fail_htlc!(claimable_htlc, payment_hash);
4242                                                                                                         }
4243                                                                                                 };
4244                                                                                                 if let Some(min_final_cltv_expiry_delta) = min_final_cltv_expiry_delta {
4245                                                                                                         let expected_min_expiry_height = (self.current_best_block().height() + min_final_cltv_expiry_delta as u32) as u64;
4246                                                                                                         if (cltv_expiry as u64) < expected_min_expiry_height {
4247                                                                                                                 log_trace!(self.logger, "Failing new HTLC with payment_hash {} as its CLTV expiry was too soon (had {}, earliest expected {})",
4248                                                                                                                         &payment_hash, cltv_expiry, expected_min_expiry_height);
4249                                                                                                                 fail_htlc!(claimable_htlc, payment_hash);
4250                                                                                                         }
4251                                                                                                 }
4252                                                                                                 let purpose = events::PaymentPurpose::InvoicePayment {
4253                                                                                                         payment_preimage: payment_preimage.clone(),
4254                                                                                                         payment_secret: payment_data.payment_secret,
4255                                                                                                 };
4256                                                                                                 check_total_value!(purpose);
4257                                                                                         },
4258                                                                                         OnionPayload::Spontaneous(preimage) => {
4259                                                                                                 let purpose = events::PaymentPurpose::SpontaneousPayment(preimage);
4260                                                                                                 check_total_value!(purpose);
4261                                                                                         }
4262                                                                                 }
4263                                                                         },
4264                                                                         hash_map::Entry::Occupied(inbound_payment) => {
4265                                                                                 if let OnionPayload::Spontaneous(_) = claimable_htlc.onion_payload {
4266                                                                                         log_trace!(self.logger, "Failing new keysend HTLC with payment_hash {} because we already have an inbound payment with the same payment hash", &payment_hash);
4267                                                                                         fail_htlc!(claimable_htlc, payment_hash);
4268                                                                                 }
4269                                                                                 let payment_data = payment_data.unwrap();
4270                                                                                 if inbound_payment.get().payment_secret != payment_data.payment_secret {
4271                                                                                         log_trace!(self.logger, "Failing new HTLC with payment_hash {} as it didn't match our expected payment secret.", &payment_hash);
4272                                                                                         fail_htlc!(claimable_htlc, payment_hash);
4273                                                                                 } else if inbound_payment.get().min_value_msat.is_some() && payment_data.total_msat < inbound_payment.get().min_value_msat.unwrap() {
4274                                                                                         log_trace!(self.logger, "Failing new HTLC with payment_hash {} as it didn't match our minimum value (had {}, needed {}).",
4275                                                                                                 &payment_hash, payment_data.total_msat, inbound_payment.get().min_value_msat.unwrap());
4276                                                                                         fail_htlc!(claimable_htlc, payment_hash);
4277                                                                                 } else {
4278                                                                                         let purpose = events::PaymentPurpose::InvoicePayment {
4279                                                                                                 payment_preimage: inbound_payment.get().payment_preimage,
4280                                                                                                 payment_secret: payment_data.payment_secret,
4281                                                                                         };
4282                                                                                         let payment_claimable_generated = check_total_value!(purpose);
4283                                                                                         if payment_claimable_generated {
4284                                                                                                 inbound_payment.remove_entry();
4285                                                                                         }
4286                                                                                 }
4287                                                                         },
4288                                                                 };
4289                                                         },
4290                                                         HTLCForwardInfo::FailHTLC { .. } => {
4291                                                                 panic!("Got pending fail of our own HTLC");
4292                                                         }
4293                                                 }
4294                                         }
4295                                 }
4296                         }
4297                 }
4298
4299                 let best_block_height = self.best_block.read().unwrap().height();
4300                 self.pending_outbound_payments.check_retry_payments(&self.router, || self.list_usable_channels(),
4301                         || self.compute_inflight_htlcs(), &self.entropy_source, &self.node_signer, best_block_height,
4302                         &self.pending_events, &self.logger, |args| self.send_payment_along_path(args));
4303
4304                 for (htlc_source, payment_hash, failure_reason, destination) in failed_forwards.drain(..) {
4305                         self.fail_htlc_backwards_internal(&htlc_source, &payment_hash, &failure_reason, destination);
4306                 }
4307                 self.forward_htlcs(&mut phantom_receives);
4308
4309                 // Freeing the holding cell here is relatively redundant - in practice we'll do it when we
4310                 // next get a `get_and_clear_pending_msg_events` call, but some tests rely on it, and it's
4311                 // nice to do the work now if we can rather than while we're trying to get messages in the
4312                 // network stack.
4313                 self.check_free_holding_cells();
4314
4315                 if new_events.is_empty() { return }
4316                 let mut events = self.pending_events.lock().unwrap();
4317                 events.append(&mut new_events);
4318         }
4319
4320         /// Free the background events, generally called from [`PersistenceNotifierGuard`] constructors.
4321         ///
4322         /// Expects the caller to have a total_consistency_lock read lock.
4323         fn process_background_events(&self) -> NotifyOption {
4324                 debug_assert_ne!(self.total_consistency_lock.held_by_thread(), LockHeldState::NotHeldByThread);
4325
4326                 self.background_events_processed_since_startup.store(true, Ordering::Release);
4327
4328                 let mut background_events = Vec::new();
4329                 mem::swap(&mut *self.pending_background_events.lock().unwrap(), &mut background_events);
4330                 if background_events.is_empty() {
4331                         return NotifyOption::SkipPersist;
4332                 }
4333
4334                 for event in background_events.drain(..) {
4335                         match event {
4336                                 BackgroundEvent::ClosedMonitorUpdateRegeneratedOnStartup((funding_txo, update)) => {
4337                                         // The channel has already been closed, so no use bothering to care about the
4338                                         // monitor updating completing.
4339                                         let _ = self.chain_monitor.update_channel(funding_txo, &update);
4340                                 },
4341                                 BackgroundEvent::MonitorUpdateRegeneratedOnStartup { counterparty_node_id, funding_txo, update } => {
4342                                         let mut updated_chan = false;
4343                                         let res = {
4344                                                 let per_peer_state = self.per_peer_state.read().unwrap();
4345                                                 if let Some(peer_state_mutex) = per_peer_state.get(&counterparty_node_id) {
4346                                                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
4347                                                         let peer_state = &mut *peer_state_lock;
4348                                                         match peer_state.channel_by_id.entry(funding_txo.to_channel_id()) {
4349                                                                 hash_map::Entry::Occupied(mut chan) => {
4350                                                                         updated_chan = true;
4351                                                                         handle_new_monitor_update!(self, funding_txo, update.clone(),
4352                                                                                 peer_state_lock, peer_state, per_peer_state, chan).map(|_| ())
4353                                                                 },
4354                                                                 hash_map::Entry::Vacant(_) => Ok(()),
4355                                                         }
4356                                                 } else { Ok(()) }
4357                                         };
4358                                         if !updated_chan {
4359                                                 // TODO: Track this as in-flight even though the channel is closed.
4360                                                 let _ = self.chain_monitor.update_channel(funding_txo, &update);
4361                                         }
4362                                         // TODO: If this channel has since closed, we're likely providing a payment
4363                                         // preimage update, which we must ensure is durable! We currently don't,
4364                                         // however, ensure that.
4365                                         if res.is_err() {
4366                                                 log_error!(self.logger,
4367                                                         "Failed to provide ChannelMonitorUpdate to closed channel! This likely lost us a payment preimage!");
4368                                         }
4369                                         let _ = handle_error!(self, res, counterparty_node_id);
4370                                 },
4371                                 BackgroundEvent::MonitorUpdatesComplete { counterparty_node_id, channel_id } => {
4372                                         let per_peer_state = self.per_peer_state.read().unwrap();
4373                                         if let Some(peer_state_mutex) = per_peer_state.get(&counterparty_node_id) {
4374                                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
4375                                                 let peer_state = &mut *peer_state_lock;
4376                                                 if let Some(chan) = peer_state.channel_by_id.get_mut(&channel_id) {
4377                                                         handle_monitor_update_completion!(self, peer_state_lock, peer_state, per_peer_state, chan);
4378                                                 } else {
4379                                                         let update_actions = peer_state.monitor_update_blocked_actions
4380                                                                 .remove(&channel_id).unwrap_or(Vec::new());
4381                                                         mem::drop(peer_state_lock);
4382                                                         mem::drop(per_peer_state);
4383                                                         self.handle_monitor_update_completion_actions(update_actions);
4384                                                 }
4385                                         }
4386                                 },
4387                         }
4388                 }
4389                 NotifyOption::DoPersist
4390         }
4391
4392         #[cfg(any(test, feature = "_test_utils"))]
4393         /// Process background events, for functional testing
4394         pub fn test_process_background_events(&self) {
4395                 let _lck = self.total_consistency_lock.read().unwrap();
4396                 let _ = self.process_background_events();
4397         }
4398
4399         fn update_channel_fee(&self, chan_id: &ChannelId, chan: &mut Channel<SP>, new_feerate: u32) -> NotifyOption {
4400                 if !chan.context.is_outbound() { return NotifyOption::SkipPersist; }
4401                 // If the feerate has decreased by less than half, don't bother
4402                 if new_feerate <= chan.context.get_feerate_sat_per_1000_weight() && new_feerate * 2 > chan.context.get_feerate_sat_per_1000_weight() {
4403                         log_trace!(self.logger, "Channel {} does not qualify for a feerate change from {} to {}.",
4404                                 &chan_id, chan.context.get_feerate_sat_per_1000_weight(), new_feerate);
4405                         return NotifyOption::SkipPersist;
4406                 }
4407                 if !chan.context.is_live() {
4408                         log_trace!(self.logger, "Channel {} does not qualify for a feerate change from {} to {} as it cannot currently be updated (probably the peer is disconnected).",
4409                                 &chan_id, chan.context.get_feerate_sat_per_1000_weight(), new_feerate);
4410                         return NotifyOption::SkipPersist;
4411                 }
4412                 log_trace!(self.logger, "Channel {} qualifies for a feerate change from {} to {}.",
4413                         &chan_id, chan.context.get_feerate_sat_per_1000_weight(), new_feerate);
4414
4415                 chan.queue_update_fee(new_feerate, &self.fee_estimator, &self.logger);
4416                 NotifyOption::DoPersist
4417         }
4418
4419         #[cfg(fuzzing)]
4420         /// In chanmon_consistency we want to sometimes do the channel fee updates done in
4421         /// timer_tick_occurred, but we can't generate the disabled channel updates as it considers
4422         /// these a fuzz failure (as they usually indicate a channel force-close, which is exactly what
4423         /// it wants to detect). Thus, we have a variant exposed here for its benefit.
4424         pub fn maybe_update_chan_fees(&self) {
4425                 PersistenceNotifierGuard::optionally_notify(&self.total_consistency_lock, &self.persistence_notifier, || {
4426                         let mut should_persist = self.process_background_events();
4427
4428                         let normal_feerate = self.fee_estimator.bounded_sat_per_1000_weight(ConfirmationTarget::Normal);
4429                         let min_mempool_feerate = self.fee_estimator.bounded_sat_per_1000_weight(ConfirmationTarget::MempoolMinimum);
4430
4431                         let per_peer_state = self.per_peer_state.read().unwrap();
4432                         for (_cp_id, peer_state_mutex) in per_peer_state.iter() {
4433                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
4434                                 let peer_state = &mut *peer_state_lock;
4435                                 for (chan_id, chan) in peer_state.channel_by_id.iter_mut() {
4436                                         let new_feerate = if chan.context.get_channel_type().supports_anchors_zero_fee_htlc_tx() {
4437                                                 min_mempool_feerate
4438                                         } else {
4439                                                 normal_feerate
4440                                         };
4441                                         let chan_needs_persist = self.update_channel_fee(chan_id, chan, new_feerate);
4442                                         if chan_needs_persist == NotifyOption::DoPersist { should_persist = NotifyOption::DoPersist; }
4443                                 }
4444                         }
4445
4446                         should_persist
4447                 });
4448         }
4449
4450         /// Performs actions which should happen on startup and roughly once per minute thereafter.
4451         ///
4452         /// This currently includes:
4453         ///  * Increasing or decreasing the on-chain feerate estimates for our outbound channels,
4454         ///  * Broadcasting [`ChannelUpdate`] messages if we've been disconnected from our peer for more
4455         ///    than a minute, informing the network that they should no longer attempt to route over
4456         ///    the channel.
4457         ///  * Expiring a channel's previous [`ChannelConfig`] if necessary to only allow forwarding HTLCs
4458         ///    with the current [`ChannelConfig`].
4459         ///  * Removing peers which have disconnected but and no longer have any channels.
4460         ///  * Force-closing and removing channels which have not completed establishment in a timely manner.
4461         ///
4462         /// Note that this may cause reentrancy through [`chain::Watch::update_channel`] calls or feerate
4463         /// estimate fetches.
4464         ///
4465         /// [`ChannelUpdate`]: msgs::ChannelUpdate
4466         /// [`ChannelConfig`]: crate::util::config::ChannelConfig
4467         pub fn timer_tick_occurred(&self) {
4468                 PersistenceNotifierGuard::optionally_notify(&self.total_consistency_lock, &self.persistence_notifier, || {
4469                         let mut should_persist = self.process_background_events();
4470
4471                         let normal_feerate = self.fee_estimator.bounded_sat_per_1000_weight(ConfirmationTarget::Normal);
4472                         let min_mempool_feerate = self.fee_estimator.bounded_sat_per_1000_weight(ConfirmationTarget::MempoolMinimum);
4473
4474                         let mut handle_errors: Vec<(Result<(), _>, _)> = Vec::new();
4475                         let mut timed_out_mpp_htlcs = Vec::new();
4476                         let mut pending_peers_awaiting_removal = Vec::new();
4477                         {
4478                                 let per_peer_state = self.per_peer_state.read().unwrap();
4479                                 for (counterparty_node_id, peer_state_mutex) in per_peer_state.iter() {
4480                                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
4481                                         let peer_state = &mut *peer_state_lock;
4482                                         let pending_msg_events = &mut peer_state.pending_msg_events;
4483                                         let counterparty_node_id = *counterparty_node_id;
4484                                         peer_state.channel_by_id.retain(|chan_id, chan| {
4485                                                 let new_feerate = if chan.context.get_channel_type().supports_anchors_zero_fee_htlc_tx() {
4486                                                         min_mempool_feerate
4487                                                 } else {
4488                                                         normal_feerate
4489                                                 };
4490                                                 let chan_needs_persist = self.update_channel_fee(chan_id, chan, new_feerate);
4491                                                 if chan_needs_persist == NotifyOption::DoPersist { should_persist = NotifyOption::DoPersist; }
4492
4493                                                 if let Err(e) = chan.timer_check_closing_negotiation_progress() {
4494                                                         let (needs_close, err) = convert_chan_err!(self, e, chan, chan_id);
4495                                                         handle_errors.push((Err(err), counterparty_node_id));
4496                                                         if needs_close { return false; }
4497                                                 }
4498
4499                                                 match chan.channel_update_status() {
4500                                                         ChannelUpdateStatus::Enabled if !chan.context.is_live() => chan.set_channel_update_status(ChannelUpdateStatus::DisabledStaged(0)),
4501                                                         ChannelUpdateStatus::Disabled if chan.context.is_live() => chan.set_channel_update_status(ChannelUpdateStatus::EnabledStaged(0)),
4502                                                         ChannelUpdateStatus::DisabledStaged(_) if chan.context.is_live()
4503                                                                 => chan.set_channel_update_status(ChannelUpdateStatus::Enabled),
4504                                                         ChannelUpdateStatus::EnabledStaged(_) if !chan.context.is_live()
4505                                                                 => chan.set_channel_update_status(ChannelUpdateStatus::Disabled),
4506                                                         ChannelUpdateStatus::DisabledStaged(mut n) if !chan.context.is_live() => {
4507                                                                 n += 1;
4508                                                                 if n >= DISABLE_GOSSIP_TICKS {
4509                                                                         chan.set_channel_update_status(ChannelUpdateStatus::Disabled);
4510                                                                         if let Ok(update) = self.get_channel_update_for_broadcast(&chan) {
4511                                                                                 pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
4512                                                                                         msg: update
4513                                                                                 });
4514                                                                         }
4515                                                                         should_persist = NotifyOption::DoPersist;
4516                                                                 } else {
4517                                                                         chan.set_channel_update_status(ChannelUpdateStatus::DisabledStaged(n));
4518                                                                 }
4519                                                         },
4520                                                         ChannelUpdateStatus::EnabledStaged(mut n) if chan.context.is_live() => {
4521                                                                 n += 1;
4522                                                                 if n >= ENABLE_GOSSIP_TICKS {
4523                                                                         chan.set_channel_update_status(ChannelUpdateStatus::Enabled);
4524                                                                         if let Ok(update) = self.get_channel_update_for_broadcast(&chan) {
4525                                                                                 pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
4526                                                                                         msg: update
4527                                                                                 });
4528                                                                         }
4529                                                                         should_persist = NotifyOption::DoPersist;
4530                                                                 } else {
4531                                                                         chan.set_channel_update_status(ChannelUpdateStatus::EnabledStaged(n));
4532                                                                 }
4533                                                         },
4534                                                         _ => {},
4535                                                 }
4536
4537                                                 chan.context.maybe_expire_prev_config();
4538
4539                                                 if chan.should_disconnect_peer_awaiting_response() {
4540                                                         log_debug!(self.logger, "Disconnecting peer {} due to not making any progress on channel {}",
4541                                                                         counterparty_node_id, chan_id);
4542                                                         pending_msg_events.push(MessageSendEvent::HandleError {
4543                                                                 node_id: counterparty_node_id,
4544                                                                 action: msgs::ErrorAction::DisconnectPeerWithWarning {
4545                                                                         msg: msgs::WarningMessage {
4546                                                                                 channel_id: *chan_id,
4547                                                                                 data: "Disconnecting due to timeout awaiting response".to_owned(),
4548                                                                         },
4549                                                                 },
4550                                                         });
4551                                                 }
4552
4553                                                 true
4554                                         });
4555
4556                                         let process_unfunded_channel_tick = |
4557                                                 chan_id: &ChannelId,
4558                                                 chan_context: &mut ChannelContext<SP>,
4559                                                 unfunded_chan_context: &mut UnfundedChannelContext,
4560                                                 pending_msg_events: &mut Vec<MessageSendEvent>,
4561                                         | {
4562                                                 chan_context.maybe_expire_prev_config();
4563                                                 if unfunded_chan_context.should_expire_unfunded_channel() {
4564                                                         log_error!(self.logger,
4565                                                                 "Force-closing pending channel with ID {} for not establishing in a timely manner",
4566                                                                 &chan_id);
4567                                                         update_maps_on_chan_removal!(self, &chan_context);
4568                                                         self.issue_channel_close_events(&chan_context, ClosureReason::HolderForceClosed);
4569                                                         self.finish_force_close_channel(chan_context.force_shutdown(false));
4570                                                         pending_msg_events.push(MessageSendEvent::HandleError {
4571                                                                 node_id: counterparty_node_id,
4572                                                                 action: msgs::ErrorAction::SendErrorMessage {
4573                                                                         msg: msgs::ErrorMessage {
4574                                                                                 channel_id: *chan_id,
4575                                                                                 data: "Force-closing pending channel due to timeout awaiting establishment handshake".to_owned(),
4576                                                                         },
4577                                                                 },
4578                                                         });
4579                                                         false
4580                                                 } else {
4581                                                         true
4582                                                 }
4583                                         };
4584                                         peer_state.outbound_v1_channel_by_id.retain(|chan_id, chan| process_unfunded_channel_tick(
4585                                                 chan_id, &mut chan.context, &mut chan.unfunded_context, pending_msg_events));
4586                                         peer_state.inbound_v1_channel_by_id.retain(|chan_id, chan| process_unfunded_channel_tick(
4587                                                 chan_id, &mut chan.context, &mut chan.unfunded_context, pending_msg_events));
4588
4589                                         for (chan_id, req) in peer_state.inbound_channel_request_by_id.iter_mut() {
4590                                                 if { req.ticks_remaining -= 1 ; req.ticks_remaining } <= 0 {
4591                                                         log_error!(self.logger, "Force-closing unaccepted inbound channel {} for not accepting in a timely manner", &chan_id);
4592                                                         peer_state.pending_msg_events.push(
4593                                                                 events::MessageSendEvent::HandleError {
4594                                                                         node_id: counterparty_node_id,
4595                                                                         action: msgs::ErrorAction::SendErrorMessage {
4596                                                                                 msg: msgs::ErrorMessage { channel_id: chan_id.clone(), data: "Channel force-closed".to_owned() }
4597                                                                         },
4598                                                                 }
4599                                                         );
4600                                                 }
4601                                         }
4602                                         peer_state.inbound_channel_request_by_id.retain(|_, req| req.ticks_remaining > 0);
4603
4604                                         if peer_state.ok_to_remove(true) {
4605                                                 pending_peers_awaiting_removal.push(counterparty_node_id);
4606                                         }
4607                                 }
4608                         }
4609
4610                         // When a peer disconnects but still has channels, the peer's `peer_state` entry in the
4611                         // `per_peer_state` is not removed by the `peer_disconnected` function. If the channels
4612                         // of to that peer is later closed while still being disconnected (i.e. force closed),
4613                         // we therefore need to remove the peer from `peer_state` separately.
4614                         // To avoid having to take the `per_peer_state` `write` lock once the channels are
4615                         // closed, we instead remove such peers awaiting removal here on a timer, to limit the
4616                         // negative effects on parallelism as much as possible.
4617                         if pending_peers_awaiting_removal.len() > 0 {
4618                                 let mut per_peer_state = self.per_peer_state.write().unwrap();
4619                                 for counterparty_node_id in pending_peers_awaiting_removal {
4620                                         match per_peer_state.entry(counterparty_node_id) {
4621                                                 hash_map::Entry::Occupied(entry) => {
4622                                                         // Remove the entry if the peer is still disconnected and we still
4623                                                         // have no channels to the peer.
4624                                                         let remove_entry = {
4625                                                                 let peer_state = entry.get().lock().unwrap();
4626                                                                 peer_state.ok_to_remove(true)
4627                                                         };
4628                                                         if remove_entry {
4629                                                                 entry.remove_entry();
4630                                                         }
4631                                                 },
4632                                                 hash_map::Entry::Vacant(_) => { /* The PeerState has already been removed */ }
4633                                         }
4634                                 }
4635                         }
4636
4637                         self.claimable_payments.lock().unwrap().claimable_payments.retain(|payment_hash, payment| {
4638                                 if payment.htlcs.is_empty() {
4639                                         // This should be unreachable
4640                                         debug_assert!(false);
4641                                         return false;
4642                                 }
4643                                 if let OnionPayload::Invoice { .. } = payment.htlcs[0].onion_payload {
4644                                         // Check if we've received all the parts we need for an MPP (the value of the parts adds to total_msat).
4645                                         // In this case we're not going to handle any timeouts of the parts here.
4646                                         // This condition determining whether the MPP is complete here must match
4647                                         // exactly the condition used in `process_pending_htlc_forwards`.
4648                                         if payment.htlcs[0].total_msat <= payment.htlcs.iter()
4649                                                 .fold(0, |total, htlc| total + htlc.sender_intended_value)
4650                                         {
4651                                                 return true;
4652                                         } else if payment.htlcs.iter_mut().any(|htlc| {
4653                                                 htlc.timer_ticks += 1;
4654                                                 return htlc.timer_ticks >= MPP_TIMEOUT_TICKS
4655                                         }) {
4656                                                 timed_out_mpp_htlcs.extend(payment.htlcs.drain(..)
4657                                                         .map(|htlc: ClaimableHTLC| (htlc.prev_hop, *payment_hash)));
4658                                                 return false;
4659                                         }
4660                                 }
4661                                 true
4662                         });
4663
4664                         for htlc_source in timed_out_mpp_htlcs.drain(..) {
4665                                 let source = HTLCSource::PreviousHopData(htlc_source.0.clone());
4666                                 let reason = HTLCFailReason::from_failure_code(23);
4667                                 let receiver = HTLCDestination::FailedPayment { payment_hash: htlc_source.1 };
4668                                 self.fail_htlc_backwards_internal(&source, &htlc_source.1, &reason, receiver);
4669                         }
4670
4671                         for (err, counterparty_node_id) in handle_errors.drain(..) {
4672                                 let _ = handle_error!(self, err, counterparty_node_id);
4673                         }
4674
4675                         self.pending_outbound_payments.remove_stale_payments(&self.pending_events);
4676
4677                         // Technically we don't need to do this here, but if we have holding cell entries in a
4678                         // channel that need freeing, it's better to do that here and block a background task
4679                         // than block the message queueing pipeline.
4680                         if self.check_free_holding_cells() {
4681                                 should_persist = NotifyOption::DoPersist;
4682                         }
4683
4684                         should_persist
4685                 });
4686         }
4687
4688         /// Indicates that the preimage for payment_hash is unknown or the received amount is incorrect
4689         /// after a PaymentClaimable event, failing the HTLC back to its origin and freeing resources
4690         /// along the path (including in our own channel on which we received it).
4691         ///
4692         /// Note that in some cases around unclean shutdown, it is possible the payment may have
4693         /// already been claimed by you via [`ChannelManager::claim_funds`] prior to you seeing (a
4694         /// second copy of) the [`events::Event::PaymentClaimable`] event. Alternatively, the payment
4695         /// may have already been failed automatically by LDK if it was nearing its expiration time.
4696         ///
4697         /// While LDK will never claim a payment automatically on your behalf (i.e. without you calling
4698         /// [`ChannelManager::claim_funds`]), you should still monitor for
4699         /// [`events::Event::PaymentClaimed`] events even for payments you intend to fail, especially on
4700         /// startup during which time claims that were in-progress at shutdown may be replayed.
4701         pub fn fail_htlc_backwards(&self, payment_hash: &PaymentHash) {
4702                 self.fail_htlc_backwards_with_reason(payment_hash, FailureCode::IncorrectOrUnknownPaymentDetails);
4703         }
4704
4705         /// This is a variant of [`ChannelManager::fail_htlc_backwards`] that allows you to specify the
4706         /// reason for the failure.
4707         ///
4708         /// See [`FailureCode`] for valid failure codes.
4709         pub fn fail_htlc_backwards_with_reason(&self, payment_hash: &PaymentHash, failure_code: FailureCode) {
4710                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
4711
4712                 let removed_source = self.claimable_payments.lock().unwrap().claimable_payments.remove(payment_hash);
4713                 if let Some(payment) = removed_source {
4714                         for htlc in payment.htlcs {
4715                                 let reason = self.get_htlc_fail_reason_from_failure_code(failure_code, &htlc);
4716                                 let source = HTLCSource::PreviousHopData(htlc.prev_hop);
4717                                 let receiver = HTLCDestination::FailedPayment { payment_hash: *payment_hash };
4718                                 self.fail_htlc_backwards_internal(&source, &payment_hash, &reason, receiver);
4719                         }
4720                 }
4721         }
4722
4723         /// Gets error data to form an [`HTLCFailReason`] given a [`FailureCode`] and [`ClaimableHTLC`].
4724         fn get_htlc_fail_reason_from_failure_code(&self, failure_code: FailureCode, htlc: &ClaimableHTLC) -> HTLCFailReason {
4725                 match failure_code {
4726                         FailureCode::TemporaryNodeFailure => HTLCFailReason::from_failure_code(failure_code.into()),
4727                         FailureCode::RequiredNodeFeatureMissing => HTLCFailReason::from_failure_code(failure_code.into()),
4728                         FailureCode::IncorrectOrUnknownPaymentDetails => {
4729                                 let mut htlc_msat_height_data = htlc.value.to_be_bytes().to_vec();
4730                                 htlc_msat_height_data.extend_from_slice(&self.best_block.read().unwrap().height().to_be_bytes());
4731                                 HTLCFailReason::reason(failure_code.into(), htlc_msat_height_data)
4732                         },
4733                         FailureCode::InvalidOnionPayload(data) => {
4734                                 let fail_data = match data {
4735                                         Some((typ, offset)) => [BigSize(typ).encode(), offset.encode()].concat(),
4736                                         None => Vec::new(),
4737                                 };
4738                                 HTLCFailReason::reason(failure_code.into(), fail_data)
4739                         }
4740                 }
4741         }
4742
4743         /// Gets an HTLC onion failure code and error data for an `UPDATE` error, given the error code
4744         /// that we want to return and a channel.
4745         ///
4746         /// This is for failures on the channel on which the HTLC was *received*, not failures
4747         /// forwarding
4748         fn get_htlc_inbound_temp_fail_err_and_data(&self, desired_err_code: u16, chan: &Channel<SP>) -> (u16, Vec<u8>) {
4749                 // We can't be sure what SCID was used when relaying inbound towards us, so we have to
4750                 // guess somewhat. If its a public channel, we figure best to just use the real SCID (as
4751                 // we're not leaking that we have a channel with the counterparty), otherwise we try to use
4752                 // an inbound SCID alias before the real SCID.
4753                 let scid_pref = if chan.context.should_announce() {
4754                         chan.context.get_short_channel_id().or(chan.context.latest_inbound_scid_alias())
4755                 } else {
4756                         chan.context.latest_inbound_scid_alias().or(chan.context.get_short_channel_id())
4757                 };
4758                 if let Some(scid) = scid_pref {
4759                         self.get_htlc_temp_fail_err_and_data(desired_err_code, scid, chan)
4760                 } else {
4761                         (0x4000|10, Vec::new())
4762                 }
4763         }
4764
4765
4766         /// Gets an HTLC onion failure code and error data for an `UPDATE` error, given the error code
4767         /// that we want to return and a channel.
4768         fn get_htlc_temp_fail_err_and_data(&self, desired_err_code: u16, scid: u64, chan: &Channel<SP>) -> (u16, Vec<u8>) {
4769                 debug_assert_eq!(desired_err_code & 0x1000, 0x1000);
4770                 if let Ok(upd) = self.get_channel_update_for_onion(scid, chan) {
4771                         let mut enc = VecWriter(Vec::with_capacity(upd.serialized_length() + 6));
4772                         if desired_err_code == 0x1000 | 20 {
4773                                 // No flags for `disabled_flags` are currently defined so they're always two zero bytes.
4774                                 // See https://github.com/lightning/bolts/blob/341ec84/04-onion-routing.md?plain=1#L1008
4775                                 0u16.write(&mut enc).expect("Writes cannot fail");
4776                         }
4777                         (upd.serialized_length() as u16 + 2).write(&mut enc).expect("Writes cannot fail");
4778                         msgs::ChannelUpdate::TYPE.write(&mut enc).expect("Writes cannot fail");
4779                         upd.write(&mut enc).expect("Writes cannot fail");
4780                         (desired_err_code, enc.0)
4781                 } else {
4782                         // If we fail to get a unicast channel_update, it implies we don't yet have an SCID,
4783                         // which means we really shouldn't have gotten a payment to be forwarded over this
4784                         // channel yet, or if we did it's from a route hint. Either way, returning an error of
4785                         // PERM|no_such_channel should be fine.
4786                         (0x4000|10, Vec::new())
4787                 }
4788         }
4789
4790         // Fail a list of HTLCs that were just freed from the holding cell. The HTLCs need to be
4791         // failed backwards or, if they were one of our outgoing HTLCs, then their failure needs to
4792         // be surfaced to the user.
4793         fn fail_holding_cell_htlcs(
4794                 &self, mut htlcs_to_fail: Vec<(HTLCSource, PaymentHash)>, channel_id: ChannelId,
4795                 counterparty_node_id: &PublicKey
4796         ) {
4797                 let (failure_code, onion_failure_data) = {
4798                         let per_peer_state = self.per_peer_state.read().unwrap();
4799                         if let Some(peer_state_mutex) = per_peer_state.get(counterparty_node_id) {
4800                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
4801                                 let peer_state = &mut *peer_state_lock;
4802                                 match peer_state.channel_by_id.entry(channel_id) {
4803                                         hash_map::Entry::Occupied(chan_entry) => {
4804                                                 self.get_htlc_inbound_temp_fail_err_and_data(0x1000|7, &chan_entry.get())
4805                                         },
4806                                         hash_map::Entry::Vacant(_) => (0x4000|10, Vec::new())
4807                                 }
4808                         } else { (0x4000|10, Vec::new()) }
4809                 };
4810
4811                 for (htlc_src, payment_hash) in htlcs_to_fail.drain(..) {
4812                         let reason = HTLCFailReason::reason(failure_code, onion_failure_data.clone());
4813                         let receiver = HTLCDestination::NextHopChannel { node_id: Some(counterparty_node_id.clone()), channel_id };
4814                         self.fail_htlc_backwards_internal(&htlc_src, &payment_hash, &reason, receiver);
4815                 }
4816         }
4817
4818         /// Fails an HTLC backwards to the sender of it to us.
4819         /// Note that we do not assume that channels corresponding to failed HTLCs are still available.
4820         fn fail_htlc_backwards_internal(&self, source: &HTLCSource, payment_hash: &PaymentHash, onion_error: &HTLCFailReason, destination: HTLCDestination) {
4821                 // Ensure that no peer state channel storage lock is held when calling this function.
4822                 // This ensures that future code doesn't introduce a lock-order requirement for
4823                 // `forward_htlcs` to be locked after the `per_peer_state` peer locks, which calling
4824                 // this function with any `per_peer_state` peer lock acquired would.
4825                 for (_, peer) in self.per_peer_state.read().unwrap().iter() {
4826                         debug_assert_ne!(peer.held_by_thread(), LockHeldState::HeldByThread);
4827                 }
4828
4829                 //TODO: There is a timing attack here where if a node fails an HTLC back to us they can
4830                 //identify whether we sent it or not based on the (I presume) very different runtime
4831                 //between the branches here. We should make this async and move it into the forward HTLCs
4832                 //timer handling.
4833
4834                 // Note that we MUST NOT end up calling methods on self.chain_monitor here - we're called
4835                 // from block_connected which may run during initialization prior to the chain_monitor
4836                 // being fully configured. See the docs for `ChannelManagerReadArgs` for more.
4837                 match source {
4838                         HTLCSource::OutboundRoute { ref path, ref session_priv, ref payment_id, .. } => {
4839                                 if self.pending_outbound_payments.fail_htlc(source, payment_hash, onion_error, path,
4840                                         session_priv, payment_id, self.probing_cookie_secret, &self.secp_ctx,
4841                                         &self.pending_events, &self.logger)
4842                                 { self.push_pending_forwards_ev(); }
4843                         },
4844                         HTLCSource::PreviousHopData(HTLCPreviousHopData { ref short_channel_id, ref htlc_id, ref incoming_packet_shared_secret, ref phantom_shared_secret, ref outpoint, .. }) => {
4845                                 log_trace!(self.logger, "Failing HTLC with payment_hash {} backwards from us with {:?}", &payment_hash, onion_error);
4846                                 let err_packet = onion_error.get_encrypted_failure_packet(incoming_packet_shared_secret, phantom_shared_secret);
4847
4848                                 let mut push_forward_ev = false;
4849                                 let mut forward_htlcs = self.forward_htlcs.lock().unwrap();
4850                                 if forward_htlcs.is_empty() {
4851                                         push_forward_ev = true;
4852                                 }
4853                                 match forward_htlcs.entry(*short_channel_id) {
4854                                         hash_map::Entry::Occupied(mut entry) => {
4855                                                 entry.get_mut().push(HTLCForwardInfo::FailHTLC { htlc_id: *htlc_id, err_packet });
4856                                         },
4857                                         hash_map::Entry::Vacant(entry) => {
4858                                                 entry.insert(vec!(HTLCForwardInfo::FailHTLC { htlc_id: *htlc_id, err_packet }));
4859                                         }
4860                                 }
4861                                 mem::drop(forward_htlcs);
4862                                 if push_forward_ev { self.push_pending_forwards_ev(); }
4863                                 let mut pending_events = self.pending_events.lock().unwrap();
4864                                 pending_events.push_back((events::Event::HTLCHandlingFailed {
4865                                         prev_channel_id: outpoint.to_channel_id(),
4866                                         failed_next_destination: destination,
4867                                 }, None));
4868                         },
4869                 }
4870         }
4871
4872         /// Provides a payment preimage in response to [`Event::PaymentClaimable`], generating any
4873         /// [`MessageSendEvent`]s needed to claim the payment.
4874         ///
4875         /// This method is guaranteed to ensure the payment has been claimed but only if the current
4876         /// height is strictly below [`Event::PaymentClaimable::claim_deadline`]. To avoid race
4877         /// conditions, you should wait for an [`Event::PaymentClaimed`] before considering the payment
4878         /// successful. It will generally be available in the next [`process_pending_events`] call.
4879         ///
4880         /// Note that if you did not set an `amount_msat` when calling [`create_inbound_payment`] or
4881         /// [`create_inbound_payment_for_hash`] you must check that the amount in the `PaymentClaimable`
4882         /// event matches your expectation. If you fail to do so and call this method, you may provide
4883         /// the sender "proof-of-payment" when they did not fulfill the full expected payment.
4884         ///
4885         /// This function will fail the payment if it has custom TLVs with even type numbers, as we
4886         /// will assume they are unknown. If you intend to accept even custom TLVs, you should use
4887         /// [`claim_funds_with_known_custom_tlvs`].
4888         ///
4889         /// [`Event::PaymentClaimable`]: crate::events::Event::PaymentClaimable
4890         /// [`Event::PaymentClaimable::claim_deadline`]: crate::events::Event::PaymentClaimable::claim_deadline
4891         /// [`Event::PaymentClaimed`]: crate::events::Event::PaymentClaimed
4892         /// [`process_pending_events`]: EventsProvider::process_pending_events
4893         /// [`create_inbound_payment`]: Self::create_inbound_payment
4894         /// [`create_inbound_payment_for_hash`]: Self::create_inbound_payment_for_hash
4895         /// [`claim_funds_with_known_custom_tlvs`]: Self::claim_funds_with_known_custom_tlvs
4896         pub fn claim_funds(&self, payment_preimage: PaymentPreimage) {
4897                 self.claim_payment_internal(payment_preimage, false);
4898         }
4899
4900         /// This is a variant of [`claim_funds`] that allows accepting a payment with custom TLVs with
4901         /// even type numbers.
4902         ///
4903         /// # Note
4904         ///
4905         /// You MUST check you've understood all even TLVs before using this to
4906         /// claim, otherwise you may unintentionally agree to some protocol you do not understand.
4907         ///
4908         /// [`claim_funds`]: Self::claim_funds
4909         pub fn claim_funds_with_known_custom_tlvs(&self, payment_preimage: PaymentPreimage) {
4910                 self.claim_payment_internal(payment_preimage, true);
4911         }
4912
4913         fn claim_payment_internal(&self, payment_preimage: PaymentPreimage, custom_tlvs_known: bool) {
4914                 let payment_hash = PaymentHash(Sha256::hash(&payment_preimage.0).into_inner());
4915
4916                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
4917
4918                 let mut sources = {
4919                         let mut claimable_payments = self.claimable_payments.lock().unwrap();
4920                         if let Some(payment) = claimable_payments.claimable_payments.remove(&payment_hash) {
4921                                 let mut receiver_node_id = self.our_network_pubkey;
4922                                 for htlc in payment.htlcs.iter() {
4923                                         if htlc.prev_hop.phantom_shared_secret.is_some() {
4924                                                 let phantom_pubkey = self.node_signer.get_node_id(Recipient::PhantomNode)
4925                                                         .expect("Failed to get node_id for phantom node recipient");
4926                                                 receiver_node_id = phantom_pubkey;
4927                                                 break;
4928                                         }
4929                                 }
4930
4931                                 let htlcs = payment.htlcs.iter().map(events::ClaimedHTLC::from).collect();
4932                                 let sender_intended_value = payment.htlcs.first().map(|htlc| htlc.total_msat);
4933                                 let dup_purpose = claimable_payments.pending_claiming_payments.insert(payment_hash,
4934                                         ClaimingPayment { amount_msat: payment.htlcs.iter().map(|source| source.value).sum(),
4935                                         payment_purpose: payment.purpose, receiver_node_id, htlcs, sender_intended_value
4936                                 });
4937                                 if dup_purpose.is_some() {
4938                                         debug_assert!(false, "Shouldn't get a duplicate pending claim event ever");
4939                                         log_error!(self.logger, "Got a duplicate pending claimable event on payment hash {}! Please report this bug",
4940                                                 &payment_hash);
4941                                 }
4942
4943                                 if let Some(RecipientOnionFields { ref custom_tlvs, .. }) = payment.onion_fields {
4944                                         if !custom_tlvs_known && custom_tlvs.iter().any(|(typ, _)| typ % 2 == 0) {
4945                                                 log_info!(self.logger, "Rejecting payment with payment hash {} as we cannot accept payment with unknown even TLVs: {}",
4946                                                         &payment_hash, log_iter!(custom_tlvs.iter().map(|(typ, _)| typ).filter(|typ| *typ % 2 == 0)));
4947                                                 claimable_payments.pending_claiming_payments.remove(&payment_hash);
4948                                                 mem::drop(claimable_payments);
4949                                                 for htlc in payment.htlcs {
4950                                                         let reason = self.get_htlc_fail_reason_from_failure_code(FailureCode::InvalidOnionPayload(None), &htlc);
4951                                                         let source = HTLCSource::PreviousHopData(htlc.prev_hop);
4952                                                         let receiver = HTLCDestination::FailedPayment { payment_hash };
4953                                                         self.fail_htlc_backwards_internal(&source, &payment_hash, &reason, receiver);
4954                                                 }
4955                                                 return;
4956                                         }
4957                                 }
4958
4959                                 payment.htlcs
4960                         } else { return; }
4961                 };
4962                 debug_assert!(!sources.is_empty());
4963
4964                 // Just in case one HTLC has been failed between when we generated the `PaymentClaimable`
4965                 // and when we got here we need to check that the amount we're about to claim matches the
4966                 // amount we told the user in the last `PaymentClaimable`. We also do a sanity-check that
4967                 // the MPP parts all have the same `total_msat`.
4968                 let mut claimable_amt_msat = 0;
4969                 let mut prev_total_msat = None;
4970                 let mut expected_amt_msat = None;
4971                 let mut valid_mpp = true;
4972                 let mut errs = Vec::new();
4973                 let per_peer_state = self.per_peer_state.read().unwrap();
4974                 for htlc in sources.iter() {
4975                         if prev_total_msat.is_some() && prev_total_msat != Some(htlc.total_msat) {
4976                                 log_error!(self.logger, "Somehow ended up with an MPP payment with different expected total amounts - this should not be reachable!");
4977                                 debug_assert!(false);
4978                                 valid_mpp = false;
4979                                 break;
4980                         }
4981                         prev_total_msat = Some(htlc.total_msat);
4982
4983                         if expected_amt_msat.is_some() && expected_amt_msat != htlc.total_value_received {
4984                                 log_error!(self.logger, "Somehow ended up with an MPP payment with different received total amounts - this should not be reachable!");
4985                                 debug_assert!(false);
4986                                 valid_mpp = false;
4987                                 break;
4988                         }
4989                         expected_amt_msat = htlc.total_value_received;
4990                         claimable_amt_msat += htlc.value;
4991                 }
4992                 mem::drop(per_peer_state);
4993                 if sources.is_empty() || expected_amt_msat.is_none() {
4994                         self.claimable_payments.lock().unwrap().pending_claiming_payments.remove(&payment_hash);
4995                         log_info!(self.logger, "Attempted to claim an incomplete payment which no longer had any available HTLCs!");
4996                         return;
4997                 }
4998                 if claimable_amt_msat != expected_amt_msat.unwrap() {
4999                         self.claimable_payments.lock().unwrap().pending_claiming_payments.remove(&payment_hash);
5000                         log_info!(self.logger, "Attempted to claim an incomplete payment, expected {} msat, had {} available to claim.",
5001                                 expected_amt_msat.unwrap(), claimable_amt_msat);
5002                         return;
5003                 }
5004                 if valid_mpp {
5005                         for htlc in sources.drain(..) {
5006                                 if let Err((pk, err)) = self.claim_funds_from_hop(
5007                                         htlc.prev_hop, payment_preimage,
5008                                         |_| Some(MonitorUpdateCompletionAction::PaymentClaimed { payment_hash }))
5009                                 {
5010                                         if let msgs::ErrorAction::IgnoreError = err.err.action {
5011                                                 // We got a temporary failure updating monitor, but will claim the
5012                                                 // HTLC when the monitor updating is restored (or on chain).
5013                                                 log_error!(self.logger, "Temporary failure claiming HTLC, treating as success: {}", err.err.err);
5014                                         } else { errs.push((pk, err)); }
5015                                 }
5016                         }
5017                 }
5018                 if !valid_mpp {
5019                         for htlc in sources.drain(..) {
5020                                 let mut htlc_msat_height_data = htlc.value.to_be_bytes().to_vec();
5021                                 htlc_msat_height_data.extend_from_slice(&self.best_block.read().unwrap().height().to_be_bytes());
5022                                 let source = HTLCSource::PreviousHopData(htlc.prev_hop);
5023                                 let reason = HTLCFailReason::reason(0x4000 | 15, htlc_msat_height_data);
5024                                 let receiver = HTLCDestination::FailedPayment { payment_hash };
5025                                 self.fail_htlc_backwards_internal(&source, &payment_hash, &reason, receiver);
5026                         }
5027                         self.claimable_payments.lock().unwrap().pending_claiming_payments.remove(&payment_hash);
5028                 }
5029
5030                 // Now we can handle any errors which were generated.
5031                 for (counterparty_node_id, err) in errs.drain(..) {
5032                         let res: Result<(), _> = Err(err);
5033                         let _ = handle_error!(self, res, counterparty_node_id);
5034                 }
5035         }
5036
5037         fn claim_funds_from_hop<ComplFunc: FnOnce(Option<u64>) -> Option<MonitorUpdateCompletionAction>>(&self,
5038                 prev_hop: HTLCPreviousHopData, payment_preimage: PaymentPreimage, completion_action: ComplFunc)
5039         -> Result<(), (PublicKey, MsgHandleErrInternal)> {
5040                 //TODO: Delay the claimed_funds relaying just like we do outbound relay!
5041
5042                 // If we haven't yet run background events assume we're still deserializing and shouldn't
5043                 // actually pass `ChannelMonitorUpdate`s to users yet. Instead, queue them up as
5044                 // `BackgroundEvent`s.
5045                 let during_init = !self.background_events_processed_since_startup.load(Ordering::Acquire);
5046
5047                 {
5048                         let per_peer_state = self.per_peer_state.read().unwrap();
5049                         let chan_id = prev_hop.outpoint.to_channel_id();
5050                         let counterparty_node_id_opt = match self.short_to_chan_info.read().unwrap().get(&prev_hop.short_channel_id) {
5051                                 Some((cp_id, _dup_chan_id)) => Some(cp_id.clone()),
5052                                 None => None
5053                         };
5054
5055                         let peer_state_opt = counterparty_node_id_opt.as_ref().map(
5056                                 |counterparty_node_id| per_peer_state.get(counterparty_node_id)
5057                                         .map(|peer_mutex| peer_mutex.lock().unwrap())
5058                         ).unwrap_or(None);
5059
5060                         if peer_state_opt.is_some() {
5061                                 let mut peer_state_lock = peer_state_opt.unwrap();
5062                                 let peer_state = &mut *peer_state_lock;
5063                                 if let hash_map::Entry::Occupied(mut chan) = peer_state.channel_by_id.entry(chan_id) {
5064                                         let counterparty_node_id = chan.get().context.get_counterparty_node_id();
5065                                         let fulfill_res = chan.get_mut().get_update_fulfill_htlc_and_commit(prev_hop.htlc_id, payment_preimage, &self.logger);
5066
5067                                         if let UpdateFulfillCommitFetch::NewClaim { htlc_value_msat, monitor_update } = fulfill_res {
5068                                                 if let Some(action) = completion_action(Some(htlc_value_msat)) {
5069                                                         log_trace!(self.logger, "Tracking monitor update completion action for channel {}: {:?}",
5070                                                                 &chan_id, action);
5071                                                         peer_state.monitor_update_blocked_actions.entry(chan_id).or_insert(Vec::new()).push(action);
5072                                                 }
5073                                                 if !during_init {
5074                                                         let res = handle_new_monitor_update!(self, prev_hop.outpoint, monitor_update, peer_state_lock,
5075                                                                 peer_state, per_peer_state, chan);
5076                                                         if let Err(e) = res {
5077                                                                 // TODO: This is a *critical* error - we probably updated the outbound edge
5078                                                                 // of the HTLC's monitor with a preimage. We should retry this monitor
5079                                                                 // update over and over again until morale improves.
5080                                                                 log_error!(self.logger, "Failed to update channel monitor with preimage {:?}", payment_preimage);
5081                                                                 return Err((counterparty_node_id, e));
5082                                                         }
5083                                                 } else {
5084                                                         // If we're running during init we cannot update a monitor directly -
5085                                                         // they probably haven't actually been loaded yet. Instead, push the
5086                                                         // monitor update as a background event.
5087                                                         self.pending_background_events.lock().unwrap().push(
5088                                                                 BackgroundEvent::MonitorUpdateRegeneratedOnStartup {
5089                                                                         counterparty_node_id,
5090                                                                         funding_txo: prev_hop.outpoint,
5091                                                                         update: monitor_update.clone(),
5092                                                                 });
5093                                                 }
5094                                         }
5095                                         return Ok(());
5096                                 }
5097                         }
5098                 }
5099                 let preimage_update = ChannelMonitorUpdate {
5100                         update_id: CLOSED_CHANNEL_UPDATE_ID,
5101                         updates: vec![ChannelMonitorUpdateStep::PaymentPreimage {
5102                                 payment_preimage,
5103                         }],
5104                 };
5105
5106                 if !during_init {
5107                         // We update the ChannelMonitor on the backward link, after
5108                         // receiving an `update_fulfill_htlc` from the forward link.
5109                         let update_res = self.chain_monitor.update_channel(prev_hop.outpoint, &preimage_update);
5110                         if update_res != ChannelMonitorUpdateStatus::Completed {
5111                                 // TODO: This needs to be handled somehow - if we receive a monitor update
5112                                 // with a preimage we *must* somehow manage to propagate it to the upstream
5113                                 // channel, or we must have an ability to receive the same event and try
5114                                 // again on restart.
5115                                 log_error!(self.logger, "Critical error: failed to update channel monitor with preimage {:?}: {:?}",
5116                                         payment_preimage, update_res);
5117                         }
5118                 } else {
5119                         // If we're running during init we cannot update a monitor directly - they probably
5120                         // haven't actually been loaded yet. Instead, push the monitor update as a background
5121                         // event.
5122                         // Note that while it's safe to use `ClosedMonitorUpdateRegeneratedOnStartup` here (the
5123                         // channel is already closed) we need to ultimately handle the monitor update
5124                         // completion action only after we've completed the monitor update. This is the only
5125                         // way to guarantee this update *will* be regenerated on startup (otherwise if this was
5126                         // from a forwarded HTLC the downstream preimage may be deleted before we claim
5127                         // upstream). Thus, we need to transition to some new `BackgroundEvent` type which will
5128                         // complete the monitor update completion action from `completion_action`.
5129                         self.pending_background_events.lock().unwrap().push(
5130                                 BackgroundEvent::ClosedMonitorUpdateRegeneratedOnStartup((
5131                                         prev_hop.outpoint, preimage_update,
5132                                 )));
5133                 }
5134                 // Note that we do process the completion action here. This totally could be a
5135                 // duplicate claim, but we have no way of knowing without interrogating the
5136                 // `ChannelMonitor` we've provided the above update to. Instead, note that `Event`s are
5137                 // generally always allowed to be duplicative (and it's specifically noted in
5138                 // `PaymentForwarded`).
5139                 self.handle_monitor_update_completion_actions(completion_action(None));
5140                 Ok(())
5141         }
5142
5143         fn finalize_claims(&self, sources: Vec<HTLCSource>) {
5144                 self.pending_outbound_payments.finalize_claims(sources, &self.pending_events);
5145         }
5146
5147         fn claim_funds_internal(&self, source: HTLCSource, payment_preimage: PaymentPreimage, forwarded_htlc_value_msat: Option<u64>, from_onchain: bool, next_channel_outpoint: OutPoint) {
5148                 match source {
5149                         HTLCSource::OutboundRoute { session_priv, payment_id, path, .. } => {
5150                                 debug_assert!(self.background_events_processed_since_startup.load(Ordering::Acquire),
5151                                         "We don't support claim_htlc claims during startup - monitors may not be available yet");
5152                                 let ev_completion_action = EventCompletionAction::ReleaseRAAChannelMonitorUpdate {
5153                                         channel_funding_outpoint: next_channel_outpoint,
5154                                         counterparty_node_id: path.hops[0].pubkey,
5155                                 };
5156                                 self.pending_outbound_payments.claim_htlc(payment_id, payment_preimage,
5157                                         session_priv, path, from_onchain, ev_completion_action, &self.pending_events,
5158                                         &self.logger);
5159                         },
5160                         HTLCSource::PreviousHopData(hop_data) => {
5161                                 let prev_outpoint = hop_data.outpoint;
5162                                 let res = self.claim_funds_from_hop(hop_data, payment_preimage,
5163                                         |htlc_claim_value_msat| {
5164                                                 if let Some(forwarded_htlc_value) = forwarded_htlc_value_msat {
5165                                                         let fee_earned_msat = if let Some(claimed_htlc_value) = htlc_claim_value_msat {
5166                                                                 Some(claimed_htlc_value - forwarded_htlc_value)
5167                                                         } else { None };
5168
5169                                                         Some(MonitorUpdateCompletionAction::EmitEventAndFreeOtherChannel {
5170                                                                 event: events::Event::PaymentForwarded {
5171                                                                         fee_earned_msat,
5172                                                                         claim_from_onchain_tx: from_onchain,
5173                                                                         prev_channel_id: Some(prev_outpoint.to_channel_id()),
5174                                                                         next_channel_id: Some(next_channel_outpoint.to_channel_id()),
5175                                                                         outbound_amount_forwarded_msat: forwarded_htlc_value_msat,
5176                                                                 },
5177                                                                 downstream_counterparty_and_funding_outpoint: None,
5178                                                         })
5179                                                 } else { None }
5180                                         });
5181                                 if let Err((pk, err)) = res {
5182                                         let result: Result<(), _> = Err(err);
5183                                         let _ = handle_error!(self, result, pk);
5184                                 }
5185                         },
5186                 }
5187         }
5188
5189         /// Gets the node_id held by this ChannelManager
5190         pub fn get_our_node_id(&self) -> PublicKey {
5191                 self.our_network_pubkey.clone()
5192         }
5193
5194         fn handle_monitor_update_completion_actions<I: IntoIterator<Item=MonitorUpdateCompletionAction>>(&self, actions: I) {
5195                 for action in actions.into_iter() {
5196                         match action {
5197                                 MonitorUpdateCompletionAction::PaymentClaimed { payment_hash } => {
5198                                         let payment = self.claimable_payments.lock().unwrap().pending_claiming_payments.remove(&payment_hash);
5199                                         if let Some(ClaimingPayment {
5200                                                 amount_msat,
5201                                                 payment_purpose: purpose,
5202                                                 receiver_node_id,
5203                                                 htlcs,
5204                                                 sender_intended_value: sender_intended_total_msat,
5205                                         }) = payment {
5206                                                 self.pending_events.lock().unwrap().push_back((events::Event::PaymentClaimed {
5207                                                         payment_hash,
5208                                                         purpose,
5209                                                         amount_msat,
5210                                                         receiver_node_id: Some(receiver_node_id),
5211                                                         htlcs,
5212                                                         sender_intended_total_msat,
5213                                                 }, None));
5214                                         }
5215                                 },
5216                                 MonitorUpdateCompletionAction::EmitEventAndFreeOtherChannel {
5217                                         event, downstream_counterparty_and_funding_outpoint
5218                                 } => {
5219                                         self.pending_events.lock().unwrap().push_back((event, None));
5220                                         if let Some((node_id, funding_outpoint, blocker)) = downstream_counterparty_and_funding_outpoint {
5221                                                 self.handle_monitor_update_release(node_id, funding_outpoint, Some(blocker));
5222                                         }
5223                                 },
5224                         }
5225                 }
5226         }
5227
5228         /// Handles a channel reentering a functional state, either due to reconnect or a monitor
5229         /// update completion.
5230         fn handle_channel_resumption(&self, pending_msg_events: &mut Vec<MessageSendEvent>,
5231                 channel: &mut Channel<SP>, raa: Option<msgs::RevokeAndACK>,
5232                 commitment_update: Option<msgs::CommitmentUpdate>, order: RAACommitmentOrder,
5233                 pending_forwards: Vec<(PendingHTLCInfo, u64)>, funding_broadcastable: Option<Transaction>,
5234                 channel_ready: Option<msgs::ChannelReady>, announcement_sigs: Option<msgs::AnnouncementSignatures>)
5235         -> Option<(u64, OutPoint, u128, Vec<(PendingHTLCInfo, u64)>)> {
5236                 log_trace!(self.logger, "Handling channel resumption for channel {} with {} RAA, {} commitment update, {} pending forwards, {}broadcasting funding, {} channel ready, {} announcement",
5237                         &channel.context.channel_id(),
5238                         if raa.is_some() { "an" } else { "no" },
5239                         if commitment_update.is_some() { "a" } else { "no" }, pending_forwards.len(),
5240                         if funding_broadcastable.is_some() { "" } else { "not " },
5241                         if channel_ready.is_some() { "sending" } else { "without" },
5242                         if announcement_sigs.is_some() { "sending" } else { "without" });
5243
5244                 let mut htlc_forwards = None;
5245
5246                 let counterparty_node_id = channel.context.get_counterparty_node_id();
5247                 if !pending_forwards.is_empty() {
5248                         htlc_forwards = Some((channel.context.get_short_channel_id().unwrap_or(channel.context.outbound_scid_alias()),
5249                                 channel.context.get_funding_txo().unwrap(), channel.context.get_user_id(), pending_forwards));
5250                 }
5251
5252                 if let Some(msg) = channel_ready {
5253                         send_channel_ready!(self, pending_msg_events, channel, msg);
5254                 }
5255                 if let Some(msg) = announcement_sigs {
5256                         pending_msg_events.push(events::MessageSendEvent::SendAnnouncementSignatures {
5257                                 node_id: counterparty_node_id,
5258                                 msg,
5259                         });
5260                 }
5261
5262                 macro_rules! handle_cs { () => {
5263                         if let Some(update) = commitment_update {
5264                                 pending_msg_events.push(events::MessageSendEvent::UpdateHTLCs {
5265                                         node_id: counterparty_node_id,
5266                                         updates: update,
5267                                 });
5268                         }
5269                 } }
5270                 macro_rules! handle_raa { () => {
5271                         if let Some(revoke_and_ack) = raa {
5272                                 pending_msg_events.push(events::MessageSendEvent::SendRevokeAndACK {
5273                                         node_id: counterparty_node_id,
5274                                         msg: revoke_and_ack,
5275                                 });
5276                         }
5277                 } }
5278                 match order {
5279                         RAACommitmentOrder::CommitmentFirst => {
5280                                 handle_cs!();
5281                                 handle_raa!();
5282                         },
5283                         RAACommitmentOrder::RevokeAndACKFirst => {
5284                                 handle_raa!();
5285                                 handle_cs!();
5286                         },
5287                 }
5288
5289                 if let Some(tx) = funding_broadcastable {
5290                         log_info!(self.logger, "Broadcasting funding transaction with txid {}", tx.txid());
5291                         self.tx_broadcaster.broadcast_transactions(&[&tx]);
5292                 }
5293
5294                 {
5295                         let mut pending_events = self.pending_events.lock().unwrap();
5296                         emit_channel_pending_event!(pending_events, channel);
5297                         emit_channel_ready_event!(pending_events, channel);
5298                 }
5299
5300                 htlc_forwards
5301         }
5302
5303         fn channel_monitor_updated(&self, funding_txo: &OutPoint, highest_applied_update_id: u64, counterparty_node_id: Option<&PublicKey>) {
5304                 debug_assert!(self.total_consistency_lock.try_write().is_err()); // Caller holds read lock
5305
5306                 let counterparty_node_id = match counterparty_node_id {
5307                         Some(cp_id) => cp_id.clone(),
5308                         None => {
5309                                 // TODO: Once we can rely on the counterparty_node_id from the
5310                                 // monitor event, this and the id_to_peer map should be removed.
5311                                 let id_to_peer = self.id_to_peer.lock().unwrap();
5312                                 match id_to_peer.get(&funding_txo.to_channel_id()) {
5313                                         Some(cp_id) => cp_id.clone(),
5314                                         None => return,
5315                                 }
5316                         }
5317                 };
5318                 let per_peer_state = self.per_peer_state.read().unwrap();
5319                 let mut peer_state_lock;
5320                 let peer_state_mutex_opt = per_peer_state.get(&counterparty_node_id);
5321                 if peer_state_mutex_opt.is_none() { return }
5322                 peer_state_lock = peer_state_mutex_opt.unwrap().lock().unwrap();
5323                 let peer_state = &mut *peer_state_lock;
5324                 let channel =
5325                         if let Some(chan) = peer_state.channel_by_id.get_mut(&funding_txo.to_channel_id()) {
5326                                 chan
5327                         } else {
5328                                 let update_actions = peer_state.monitor_update_blocked_actions
5329                                         .remove(&funding_txo.to_channel_id()).unwrap_or(Vec::new());
5330                                 mem::drop(peer_state_lock);
5331                                 mem::drop(per_peer_state);
5332                                 self.handle_monitor_update_completion_actions(update_actions);
5333                                 return;
5334                         };
5335                 let remaining_in_flight =
5336                         if let Some(pending) = peer_state.in_flight_monitor_updates.get_mut(funding_txo) {
5337                                 pending.retain(|upd| upd.update_id > highest_applied_update_id);
5338                                 pending.len()
5339                         } else { 0 };
5340                 log_trace!(self.logger, "ChannelMonitor updated to {}. Current highest is {}. {} pending in-flight updates.",
5341                         highest_applied_update_id, channel.context.get_latest_monitor_update_id(),
5342                         remaining_in_flight);
5343                 if !channel.is_awaiting_monitor_update() || channel.context.get_latest_monitor_update_id() != highest_applied_update_id {
5344                         return;
5345                 }
5346                 handle_monitor_update_completion!(self, peer_state_lock, peer_state, per_peer_state, channel);
5347         }
5348
5349         /// Accepts a request to open a channel after a [`Event::OpenChannelRequest`].
5350         ///
5351         /// The `temporary_channel_id` parameter indicates which inbound channel should be accepted,
5352         /// and the `counterparty_node_id` parameter is the id of the peer which has requested to open
5353         /// the channel.
5354         ///
5355         /// The `user_channel_id` parameter will be provided back in
5356         /// [`Event::ChannelClosed::user_channel_id`] to allow tracking of which events correspond
5357         /// with which `accept_inbound_channel`/`accept_inbound_channel_from_trusted_peer_0conf` call.
5358         ///
5359         /// Note that this method will return an error and reject the channel, if it requires support
5360         /// for zero confirmations. Instead, `accept_inbound_channel_from_trusted_peer_0conf` must be
5361         /// used to accept such channels.
5362         ///
5363         /// [`Event::OpenChannelRequest`]: events::Event::OpenChannelRequest
5364         /// [`Event::ChannelClosed::user_channel_id`]: events::Event::ChannelClosed::user_channel_id
5365         pub fn accept_inbound_channel(&self, temporary_channel_id: &ChannelId, counterparty_node_id: &PublicKey, user_channel_id: u128) -> Result<(), APIError> {
5366                 self.do_accept_inbound_channel(temporary_channel_id, counterparty_node_id, false, user_channel_id)
5367         }
5368
5369         /// Accepts a request to open a channel after a [`events::Event::OpenChannelRequest`], treating
5370         /// it as confirmed immediately.
5371         ///
5372         /// The `user_channel_id` parameter will be provided back in
5373         /// [`Event::ChannelClosed::user_channel_id`] to allow tracking of which events correspond
5374         /// with which `accept_inbound_channel`/`accept_inbound_channel_from_trusted_peer_0conf` call.
5375         ///
5376         /// Unlike [`ChannelManager::accept_inbound_channel`], this method accepts the incoming channel
5377         /// and (if the counterparty agrees), enables forwarding of payments immediately.
5378         ///
5379         /// This fully trusts that the counterparty has honestly and correctly constructed the funding
5380         /// transaction and blindly assumes that it will eventually confirm.
5381         ///
5382         /// If it does not confirm before we decide to close the channel, or if the funding transaction
5383         /// does not pay to the correct script the correct amount, *you will lose funds*.
5384         ///
5385         /// [`Event::OpenChannelRequest`]: events::Event::OpenChannelRequest
5386         /// [`Event::ChannelClosed::user_channel_id`]: events::Event::ChannelClosed::user_channel_id
5387         pub fn accept_inbound_channel_from_trusted_peer_0conf(&self, temporary_channel_id: &ChannelId, counterparty_node_id: &PublicKey, user_channel_id: u128) -> Result<(), APIError> {
5388                 self.do_accept_inbound_channel(temporary_channel_id, counterparty_node_id, true, user_channel_id)
5389         }
5390
5391         fn do_accept_inbound_channel(&self, temporary_channel_id: &ChannelId, counterparty_node_id: &PublicKey, accept_0conf: bool, user_channel_id: u128) -> Result<(), APIError> {
5392                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
5393
5394                 let peers_without_funded_channels =
5395                         self.peers_without_funded_channels(|peer| { peer.total_channel_count() > 0 });
5396                 let per_peer_state = self.per_peer_state.read().unwrap();
5397                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5398                         .ok_or_else(|| APIError::ChannelUnavailable { err: format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id) })?;
5399                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5400                 let peer_state = &mut *peer_state_lock;
5401                 let is_only_peer_channel = peer_state.total_channel_count() == 1;
5402
5403                 // Find (and remove) the channel in the unaccepted table. If it's not there, something weird is
5404                 // happening and return an error. N.B. that we create channel with an outbound SCID of zero so
5405                 // that we can delay allocating the SCID until after we're sure that the checks below will
5406                 // succeed.
5407                 let mut channel = match peer_state.inbound_channel_request_by_id.remove(temporary_channel_id) {
5408                         Some(unaccepted_channel) => {
5409                                 let best_block_height = self.best_block.read().unwrap().height();
5410                                 InboundV1Channel::new(&self.fee_estimator, &self.entropy_source, &self.signer_provider,
5411                                         counterparty_node_id.clone(), &self.channel_type_features(), &peer_state.latest_features,
5412                                         &unaccepted_channel.open_channel_msg, user_channel_id, &self.default_configuration, best_block_height,
5413                                         &self.logger, accept_0conf).map_err(|e| APIError::ChannelUnavailable { err: e.to_string() })
5414                         }
5415                         _ => Err(APIError::APIMisuseError { err: "No such channel awaiting to be accepted.".to_owned() })
5416                 }?;
5417
5418                 if accept_0conf {
5419                         // This should have been correctly configured by the call to InboundV1Channel::new.
5420                         debug_assert!(channel.context.minimum_depth().unwrap() == 0);
5421                 } else if channel.context.get_channel_type().requires_zero_conf() {
5422                         let send_msg_err_event = events::MessageSendEvent::HandleError {
5423                                 node_id: channel.context.get_counterparty_node_id(),
5424                                 action: msgs::ErrorAction::SendErrorMessage{
5425                                         msg: msgs::ErrorMessage { channel_id: temporary_channel_id.clone(), data: "No zero confirmation channels accepted".to_owned(), }
5426                                 }
5427                         };
5428                         peer_state.pending_msg_events.push(send_msg_err_event);
5429                         return Err(APIError::APIMisuseError { err: "Please use accept_inbound_channel_from_trusted_peer_0conf to accept channels with zero confirmations.".to_owned() });
5430                 } else {
5431                         // If this peer already has some channels, a new channel won't increase our number of peers
5432                         // with unfunded channels, so as long as we aren't over the maximum number of unfunded
5433                         // channels per-peer we can accept channels from a peer with existing ones.
5434                         if is_only_peer_channel && peers_without_funded_channels >= MAX_UNFUNDED_CHANNEL_PEERS {
5435                                 let send_msg_err_event = events::MessageSendEvent::HandleError {
5436                                         node_id: channel.context.get_counterparty_node_id(),
5437                                         action: msgs::ErrorAction::SendErrorMessage{
5438                                                 msg: msgs::ErrorMessage { channel_id: temporary_channel_id.clone(), data: "Have too many peers with unfunded channels, not accepting new ones".to_owned(), }
5439                                         }
5440                                 };
5441                                 peer_state.pending_msg_events.push(send_msg_err_event);
5442                                 return Err(APIError::APIMisuseError { err: "Too many peers with unfunded channels, refusing to accept new ones".to_owned() });
5443                         }
5444                 }
5445
5446                 // Now that we know we have a channel, assign an outbound SCID alias.
5447                 let outbound_scid_alias = self.create_and_insert_outbound_scid_alias();
5448                 channel.context.set_outbound_scid_alias(outbound_scid_alias);
5449
5450                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendAcceptChannel {
5451                         node_id: channel.context.get_counterparty_node_id(),
5452                         msg: channel.accept_inbound_channel(),
5453                 });
5454
5455                 peer_state.inbound_v1_channel_by_id.insert(temporary_channel_id.clone(), channel);
5456
5457                 Ok(())
5458         }
5459
5460         /// Gets the number of peers which match the given filter and do not have any funded, outbound,
5461         /// or 0-conf channels.
5462         ///
5463         /// The filter is called for each peer and provided with the number of unfunded, inbound, and
5464         /// non-0-conf channels we have with the peer.
5465         fn peers_without_funded_channels<Filter>(&self, maybe_count_peer: Filter) -> usize
5466         where Filter: Fn(&PeerState<SP>) -> bool {
5467                 let mut peers_without_funded_channels = 0;
5468                 let best_block_height = self.best_block.read().unwrap().height();
5469                 {
5470                         let peer_state_lock = self.per_peer_state.read().unwrap();
5471                         for (_, peer_mtx) in peer_state_lock.iter() {
5472                                 let peer = peer_mtx.lock().unwrap();
5473                                 if !maybe_count_peer(&*peer) { continue; }
5474                                 let num_unfunded_channels = Self::unfunded_channel_count(&peer, best_block_height);
5475                                 if num_unfunded_channels == peer.total_channel_count() {
5476                                         peers_without_funded_channels += 1;
5477                                 }
5478                         }
5479                 }
5480                 return peers_without_funded_channels;
5481         }
5482
5483         fn unfunded_channel_count(
5484                 peer: &PeerState<SP>, best_block_height: u32
5485         ) -> usize {
5486                 let mut num_unfunded_channels = 0;
5487                 for (_, chan) in peer.channel_by_id.iter() {
5488                         // This covers non-zero-conf inbound `Channel`s that we are currently monitoring, but those
5489                         // which have not yet had any confirmations on-chain.
5490                         if !chan.context.is_outbound() && chan.context.minimum_depth().unwrap_or(1) != 0 &&
5491                                 chan.context.get_funding_tx_confirmations(best_block_height) == 0
5492                         {
5493                                 num_unfunded_channels += 1;
5494                         }
5495                 }
5496                 for (_, chan) in peer.inbound_v1_channel_by_id.iter() {
5497                         if chan.context.minimum_depth().unwrap_or(1) != 0 {
5498                                 num_unfunded_channels += 1;
5499                         }
5500                 }
5501                 num_unfunded_channels + peer.inbound_channel_request_by_id.len()
5502         }
5503
5504         fn internal_open_channel(&self, counterparty_node_id: &PublicKey, msg: &msgs::OpenChannel) -> Result<(), MsgHandleErrInternal> {
5505                 if msg.chain_hash != self.genesis_hash {
5506                         return Err(MsgHandleErrInternal::send_err_msg_no_close("Unknown genesis block hash".to_owned(), msg.temporary_channel_id.clone()));
5507                 }
5508
5509                 if !self.default_configuration.accept_inbound_channels {
5510                         return Err(MsgHandleErrInternal::send_err_msg_no_close("No inbound channels accepted".to_owned(), msg.temporary_channel_id.clone()));
5511                 }
5512
5513                 // Get the number of peers with channels, but without funded ones. We don't care too much
5514                 // about peers that never open a channel, so we filter by peers that have at least one
5515                 // channel, and then limit the number of those with unfunded channels.
5516                 let channeled_peers_without_funding =
5517                         self.peers_without_funded_channels(|node| node.total_channel_count() > 0);
5518
5519                 let per_peer_state = self.per_peer_state.read().unwrap();
5520                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5521                     .ok_or_else(|| {
5522                                 debug_assert!(false);
5523                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.temporary_channel_id.clone())
5524                         })?;
5525                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5526                 let peer_state = &mut *peer_state_lock;
5527
5528                 // If this peer already has some channels, a new channel won't increase our number of peers
5529                 // with unfunded channels, so as long as we aren't over the maximum number of unfunded
5530                 // channels per-peer we can accept channels from a peer with existing ones.
5531                 if peer_state.total_channel_count() == 0 &&
5532                         channeled_peers_without_funding >= MAX_UNFUNDED_CHANNEL_PEERS &&
5533                         !self.default_configuration.manually_accept_inbound_channels
5534                 {
5535                         return Err(MsgHandleErrInternal::send_err_msg_no_close(
5536                                 "Have too many peers with unfunded channels, not accepting new ones".to_owned(),
5537                                 msg.temporary_channel_id.clone()));
5538                 }
5539
5540                 let best_block_height = self.best_block.read().unwrap().height();
5541                 if Self::unfunded_channel_count(peer_state, best_block_height) >= MAX_UNFUNDED_CHANS_PER_PEER {
5542                         return Err(MsgHandleErrInternal::send_err_msg_no_close(
5543                                 format!("Refusing more than {} unfunded channels.", MAX_UNFUNDED_CHANS_PER_PEER),
5544                                 msg.temporary_channel_id.clone()));
5545                 }
5546
5547                 let channel_id = msg.temporary_channel_id;
5548                 let channel_exists = peer_state.has_channel(&channel_id);
5549                 if channel_exists {
5550                         return Err(MsgHandleErrInternal::send_err_msg_no_close("temporary_channel_id collision for the same peer!".to_owned(), msg.temporary_channel_id.clone()));
5551                 }
5552
5553                 // If we're doing manual acceptance checks on the channel, then defer creation until we're sure we want to accept.
5554                 if self.default_configuration.manually_accept_inbound_channels {
5555                         let mut pending_events = self.pending_events.lock().unwrap();
5556                         pending_events.push_back((events::Event::OpenChannelRequest {
5557                                 temporary_channel_id: msg.temporary_channel_id.clone(),
5558                                 counterparty_node_id: counterparty_node_id.clone(),
5559                                 funding_satoshis: msg.funding_satoshis,
5560                                 push_msat: msg.push_msat,
5561                                 channel_type: msg.channel_type.clone().unwrap(),
5562                         }, None));
5563                         peer_state.inbound_channel_request_by_id.insert(channel_id, InboundChannelRequest {
5564                                 open_channel_msg: msg.clone(),
5565                                 ticks_remaining: UNACCEPTED_INBOUND_CHANNEL_AGE_LIMIT_TICKS,
5566                         });
5567                         return Ok(());
5568                 }
5569
5570                 // Otherwise create the channel right now.
5571                 let mut random_bytes = [0u8; 16];
5572                 random_bytes.copy_from_slice(&self.entropy_source.get_secure_random_bytes()[..16]);
5573                 let user_channel_id = u128::from_be_bytes(random_bytes);
5574                 let mut channel = match InboundV1Channel::new(&self.fee_estimator, &self.entropy_source, &self.signer_provider,
5575                         counterparty_node_id.clone(), &self.channel_type_features(), &peer_state.latest_features, msg, user_channel_id,
5576                         &self.default_configuration, best_block_height, &self.logger, /*is_0conf=*/false)
5577                 {
5578                         Err(e) => {
5579                                 return Err(MsgHandleErrInternal::from_chan_no_close(e, msg.temporary_channel_id));
5580                         },
5581                         Ok(res) => res
5582                 };
5583
5584                 let channel_type = channel.context.get_channel_type();
5585                 if channel_type.requires_zero_conf() {
5586                         return Err(MsgHandleErrInternal::send_err_msg_no_close("No zero confirmation channels accepted".to_owned(), msg.temporary_channel_id.clone()));
5587                 }
5588                 if channel_type.requires_anchors_zero_fee_htlc_tx() {
5589                         return Err(MsgHandleErrInternal::send_err_msg_no_close("No channels with anchor outputs accepted".to_owned(), msg.temporary_channel_id.clone()));
5590                 }
5591
5592                 let outbound_scid_alias = self.create_and_insert_outbound_scid_alias();
5593                 channel.context.set_outbound_scid_alias(outbound_scid_alias);
5594
5595                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendAcceptChannel {
5596                         node_id: counterparty_node_id.clone(),
5597                         msg: channel.accept_inbound_channel(),
5598                 });
5599                 peer_state.inbound_v1_channel_by_id.insert(channel_id, channel);
5600                 Ok(())
5601         }
5602
5603         fn internal_accept_channel(&self, counterparty_node_id: &PublicKey, msg: &msgs::AcceptChannel) -> Result<(), MsgHandleErrInternal> {
5604                 let (value, output_script, user_id) = {
5605                         let per_peer_state = self.per_peer_state.read().unwrap();
5606                         let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5607                                 .ok_or_else(|| {
5608                                         debug_assert!(false);
5609                                         MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.temporary_channel_id)
5610                                 })?;
5611                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5612                         let peer_state = &mut *peer_state_lock;
5613                         match peer_state.outbound_v1_channel_by_id.entry(msg.temporary_channel_id) {
5614                                 hash_map::Entry::Occupied(mut chan) => {
5615                                         try_v1_outbound_chan_entry!(self, chan.get_mut().accept_channel(&msg, &self.default_configuration.channel_handshake_limits, &peer_state.latest_features), chan);
5616                                         (chan.get().context.get_value_satoshis(), chan.get().context.get_funding_redeemscript().to_v0_p2wsh(), chan.get().context.get_user_id())
5617                                 },
5618                                 hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.temporary_channel_id))
5619                         }
5620                 };
5621                 let mut pending_events = self.pending_events.lock().unwrap();
5622                 pending_events.push_back((events::Event::FundingGenerationReady {
5623                         temporary_channel_id: msg.temporary_channel_id,
5624                         counterparty_node_id: *counterparty_node_id,
5625                         channel_value_satoshis: value,
5626                         output_script,
5627                         user_channel_id: user_id,
5628                 }, None));
5629                 Ok(())
5630         }
5631
5632         fn internal_funding_created(&self, counterparty_node_id: &PublicKey, msg: &msgs::FundingCreated) -> Result<(), MsgHandleErrInternal> {
5633                 let best_block = *self.best_block.read().unwrap();
5634
5635                 let per_peer_state = self.per_peer_state.read().unwrap();
5636                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5637                         .ok_or_else(|| {
5638                                 debug_assert!(false);
5639                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.temporary_channel_id)
5640                         })?;
5641
5642                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5643                 let peer_state = &mut *peer_state_lock;
5644                 let (chan, funding_msg, monitor) =
5645                         match peer_state.inbound_v1_channel_by_id.remove(&msg.temporary_channel_id) {
5646                                 Some(inbound_chan) => {
5647                                         match inbound_chan.funding_created(msg, best_block, &self.signer_provider, &self.logger) {
5648                                                 Ok(res) => res,
5649                                                 Err((mut inbound_chan, err)) => {
5650                                                         // We've already removed this inbound channel from the map in `PeerState`
5651                                                         // above so at this point we just need to clean up any lingering entries
5652                                                         // concerning this channel as it is safe to do so.
5653                                                         update_maps_on_chan_removal!(self, &inbound_chan.context);
5654                                                         let user_id = inbound_chan.context.get_user_id();
5655                                                         let shutdown_res = inbound_chan.context.force_shutdown(false);
5656                                                         return Err(MsgHandleErrInternal::from_finish_shutdown(format!("{}", err),
5657                                                                 msg.temporary_channel_id, user_id, shutdown_res, None, inbound_chan.context.get_value_satoshis()));
5658                                                 },
5659                                         }
5660                                 },
5661                                 None => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.temporary_channel_id))
5662                         };
5663
5664                 match peer_state.channel_by_id.entry(funding_msg.channel_id) {
5665                         hash_map::Entry::Occupied(_) => {
5666                                 Err(MsgHandleErrInternal::send_err_msg_no_close("Already had channel with the new channel_id".to_owned(), funding_msg.channel_id))
5667                         },
5668                         hash_map::Entry::Vacant(e) => {
5669                                 match self.id_to_peer.lock().unwrap().entry(chan.context.channel_id()) {
5670                                         hash_map::Entry::Occupied(_) => {
5671                                                 return Err(MsgHandleErrInternal::send_err_msg_no_close(
5672                                                         "The funding_created message had the same funding_txid as an existing channel - funding is not possible".to_owned(),
5673                                                         funding_msg.channel_id))
5674                                         },
5675                                         hash_map::Entry::Vacant(i_e) => {
5676                                                 i_e.insert(chan.context.get_counterparty_node_id());
5677                                         }
5678                                 }
5679
5680                                 // There's no problem signing a counterparty's funding transaction if our monitor
5681                                 // hasn't persisted to disk yet - we can't lose money on a transaction that we haven't
5682                                 // accepted payment from yet. We do, however, need to wait to send our channel_ready
5683                                 // until we have persisted our monitor.
5684                                 let new_channel_id = funding_msg.channel_id;
5685                                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendFundingSigned {
5686                                         node_id: counterparty_node_id.clone(),
5687                                         msg: funding_msg,
5688                                 });
5689
5690                                 let monitor_res = self.chain_monitor.watch_channel(monitor.get_funding_txo().0, monitor);
5691
5692                                 let chan = e.insert(chan);
5693                                 let mut res = handle_new_monitor_update!(self, monitor_res, peer_state_lock, peer_state,
5694                                         per_peer_state, chan, MANUALLY_REMOVING_INITIAL_MONITOR,
5695                                         { peer_state.channel_by_id.remove(&new_channel_id) });
5696
5697                                 // Note that we reply with the new channel_id in error messages if we gave up on the
5698                                 // channel, not the temporary_channel_id. This is compatible with ourselves, but the
5699                                 // spec is somewhat ambiguous here. Not a huge deal since we'll send error messages for
5700                                 // any messages referencing a previously-closed channel anyway.
5701                                 // We do not propagate the monitor update to the user as it would be for a monitor
5702                                 // that we didn't manage to store (and that we don't care about - we don't respond
5703                                 // with the funding_signed so the channel can never go on chain).
5704                                 if let Err(MsgHandleErrInternal { shutdown_finish: Some((res, _)), .. }) = &mut res {
5705                                         res.0 = None;
5706                                 }
5707                                 res.map(|_| ())
5708                         }
5709                 }
5710         }
5711
5712         fn internal_funding_signed(&self, counterparty_node_id: &PublicKey, msg: &msgs::FundingSigned) -> Result<(), MsgHandleErrInternal> {
5713                 let best_block = *self.best_block.read().unwrap();
5714                 let per_peer_state = self.per_peer_state.read().unwrap();
5715                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5716                         .ok_or_else(|| {
5717                                 debug_assert!(false);
5718                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
5719                         })?;
5720
5721                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5722                 let peer_state = &mut *peer_state_lock;
5723                 match peer_state.channel_by_id.entry(msg.channel_id) {
5724                         hash_map::Entry::Occupied(mut chan) => {
5725                                 let monitor = try_chan_entry!(self,
5726                                         chan.get_mut().funding_signed(&msg, best_block, &self.signer_provider, &self.logger), chan);
5727                                 let update_res = self.chain_monitor.watch_channel(chan.get().context.get_funding_txo().unwrap(), monitor);
5728                                 let mut res = handle_new_monitor_update!(self, update_res, peer_state_lock, peer_state, per_peer_state, chan, INITIAL_MONITOR);
5729                                 if let Err(MsgHandleErrInternal { ref mut shutdown_finish, .. }) = res {
5730                                         // We weren't able to watch the channel to begin with, so no updates should be made on
5731                                         // it. Previously, full_stack_target found an (unreachable) panic when the
5732                                         // monitor update contained within `shutdown_finish` was applied.
5733                                         if let Some((ref mut shutdown_finish, _)) = shutdown_finish {
5734                                                 shutdown_finish.0.take();
5735                                         }
5736                                 }
5737                                 res.map(|_| ())
5738                         },
5739                         hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close("Failed to find corresponding channel".to_owned(), msg.channel_id))
5740                 }
5741         }
5742
5743         fn internal_channel_ready(&self, counterparty_node_id: &PublicKey, msg: &msgs::ChannelReady) -> Result<(), MsgHandleErrInternal> {
5744                 let per_peer_state = self.per_peer_state.read().unwrap();
5745                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5746                         .ok_or_else(|| {
5747                                 debug_assert!(false);
5748                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
5749                         })?;
5750                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5751                 let peer_state = &mut *peer_state_lock;
5752                 match peer_state.channel_by_id.entry(msg.channel_id) {
5753                         hash_map::Entry::Occupied(mut chan) => {
5754                                 let announcement_sigs_opt = try_chan_entry!(self, chan.get_mut().channel_ready(&msg, &self.node_signer,
5755                                         self.genesis_hash.clone(), &self.default_configuration, &self.best_block.read().unwrap(), &self.logger), chan);
5756                                 if let Some(announcement_sigs) = announcement_sigs_opt {
5757                                         log_trace!(self.logger, "Sending announcement_signatures for channel {}", &chan.get().context.channel_id());
5758                                         peer_state.pending_msg_events.push(events::MessageSendEvent::SendAnnouncementSignatures {
5759                                                 node_id: counterparty_node_id.clone(),
5760                                                 msg: announcement_sigs,
5761                                         });
5762                                 } else if chan.get().context.is_usable() {
5763                                         // If we're sending an announcement_signatures, we'll send the (public)
5764                                         // channel_update after sending a channel_announcement when we receive our
5765                                         // counterparty's announcement_signatures. Thus, we only bother to send a
5766                                         // channel_update here if the channel is not public, i.e. we're not sending an
5767                                         // announcement_signatures.
5768                                         log_trace!(self.logger, "Sending private initial channel_update for our counterparty on channel {}", &chan.get().context.channel_id());
5769                                         if let Ok(msg) = self.get_channel_update_for_unicast(chan.get()) {
5770                                                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendChannelUpdate {
5771                                                         node_id: counterparty_node_id.clone(),
5772                                                         msg,
5773                                                 });
5774                                         }
5775                                 }
5776
5777                                 {
5778                                         let mut pending_events = self.pending_events.lock().unwrap();
5779                                         emit_channel_ready_event!(pending_events, chan.get_mut());
5780                                 }
5781
5782                                 Ok(())
5783                         },
5784                         hash_map::Entry::Vacant(_) => Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
5785                 }
5786         }
5787
5788         fn internal_shutdown(&self, counterparty_node_id: &PublicKey, msg: &msgs::Shutdown) -> Result<(), MsgHandleErrInternal> {
5789                 let mut dropped_htlcs: Vec<(HTLCSource, PaymentHash)>;
5790                 let result: Result<(), _> = loop {
5791                         let per_peer_state = self.per_peer_state.read().unwrap();
5792                         let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5793                                 .ok_or_else(|| {
5794                                         debug_assert!(false);
5795                                         MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
5796                                 })?;
5797                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5798                         let peer_state = &mut *peer_state_lock;
5799                         // TODO(dunxen): Fix this duplication when we switch to a single map with enums as per
5800                         // https://github.com/lightningdevkit/rust-lightning/issues/2422
5801                         if let hash_map::Entry::Occupied(chan_entry) = peer_state.outbound_v1_channel_by_id.entry(msg.channel_id.clone()) {
5802                                 log_error!(self.logger, "Immediately closing unfunded channel {} as peer asked to cooperatively shut it down (which is unnecessary)", &msg.channel_id);
5803                                 self.issue_channel_close_events(&chan_entry.get().context, ClosureReason::CounterpartyCoopClosedUnfundedChannel);
5804                                 let mut chan = remove_channel!(self, chan_entry);
5805                                 self.finish_force_close_channel(chan.context.force_shutdown(false));
5806                                 return Ok(());
5807                         } else if let hash_map::Entry::Occupied(chan_entry) = peer_state.inbound_v1_channel_by_id.entry(msg.channel_id.clone()) {
5808                                 log_error!(self.logger, "Immediately closing unfunded channel {} as peer asked to cooperatively shut it down (which is unnecessary)", &msg.channel_id);
5809                                 self.issue_channel_close_events(&chan_entry.get().context, ClosureReason::CounterpartyCoopClosedUnfundedChannel);
5810                                 let mut chan = remove_channel!(self, chan_entry);
5811                                 self.finish_force_close_channel(chan.context.force_shutdown(false));
5812                                 return Ok(());
5813                         } else if let hash_map::Entry::Occupied(mut chan_entry) = peer_state.channel_by_id.entry(msg.channel_id.clone()) {
5814                                 if !chan_entry.get().received_shutdown() {
5815                                         log_info!(self.logger, "Received a shutdown message from our counterparty for channel {}{}.",
5816                                                 &msg.channel_id,
5817                                                 if chan_entry.get().sent_shutdown() { " after we initiated shutdown" } else { "" });
5818                                 }
5819
5820                                 let funding_txo_opt = chan_entry.get().context.get_funding_txo();
5821                                 let (shutdown, monitor_update_opt, htlcs) = try_chan_entry!(self,
5822                                         chan_entry.get_mut().shutdown(&self.signer_provider, &peer_state.latest_features, &msg), chan_entry);
5823                                 dropped_htlcs = htlcs;
5824
5825                                 if let Some(msg) = shutdown {
5826                                         // We can send the `shutdown` message before updating the `ChannelMonitor`
5827                                         // here as we don't need the monitor update to complete until we send a
5828                                         // `shutdown_signed`, which we'll delay if we're pending a monitor update.
5829                                         peer_state.pending_msg_events.push(events::MessageSendEvent::SendShutdown {
5830                                                 node_id: *counterparty_node_id,
5831                                                 msg,
5832                                         });
5833                                 }
5834
5835                                 // Update the monitor with the shutdown script if necessary.
5836                                 if let Some(monitor_update) = monitor_update_opt {
5837                                         break handle_new_monitor_update!(self, funding_txo_opt.unwrap(), monitor_update,
5838                                                 peer_state_lock, peer_state, per_peer_state, chan_entry).map(|_| ());
5839                                 }
5840                                 break Ok(());
5841                         } else {
5842                                 return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
5843                         }
5844                 };
5845                 for htlc_source in dropped_htlcs.drain(..) {
5846                         let receiver = HTLCDestination::NextHopChannel { node_id: Some(counterparty_node_id.clone()), channel_id: msg.channel_id };
5847                         let reason = HTLCFailReason::from_failure_code(0x4000 | 8);
5848                         self.fail_htlc_backwards_internal(&htlc_source.0, &htlc_source.1, &reason, receiver);
5849                 }
5850
5851                 result
5852         }
5853
5854         fn internal_closing_signed(&self, counterparty_node_id: &PublicKey, msg: &msgs::ClosingSigned) -> Result<(), MsgHandleErrInternal> {
5855                 let per_peer_state = self.per_peer_state.read().unwrap();
5856                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5857                         .ok_or_else(|| {
5858                                 debug_assert!(false);
5859                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
5860                         })?;
5861                 let (tx, chan_option) = {
5862                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5863                         let peer_state = &mut *peer_state_lock;
5864                         match peer_state.channel_by_id.entry(msg.channel_id.clone()) {
5865                                 hash_map::Entry::Occupied(mut chan_entry) => {
5866                                         let (closing_signed, tx) = try_chan_entry!(self, chan_entry.get_mut().closing_signed(&self.fee_estimator, &msg), chan_entry);
5867                                         if let Some(msg) = closing_signed {
5868                                                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendClosingSigned {
5869                                                         node_id: counterparty_node_id.clone(),
5870                                                         msg,
5871                                                 });
5872                                         }
5873                                         if tx.is_some() {
5874                                                 // We're done with this channel, we've got a signed closing transaction and
5875                                                 // will send the closing_signed back to the remote peer upon return. This
5876                                                 // also implies there are no pending HTLCs left on the channel, so we can
5877                                                 // fully delete it from tracking (the channel monitor is still around to
5878                                                 // watch for old state broadcasts)!
5879                                                 (tx, Some(remove_channel!(self, chan_entry)))
5880                                         } else { (tx, None) }
5881                                 },
5882                                 hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
5883                         }
5884                 };
5885                 if let Some(broadcast_tx) = tx {
5886                         log_info!(self.logger, "Broadcasting {}", log_tx!(broadcast_tx));
5887                         self.tx_broadcaster.broadcast_transactions(&[&broadcast_tx]);
5888                 }
5889                 if let Some(chan) = chan_option {
5890                         if let Ok(update) = self.get_channel_update_for_broadcast(&chan) {
5891                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5892                                 let peer_state = &mut *peer_state_lock;
5893                                 peer_state.pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
5894                                         msg: update
5895                                 });
5896                         }
5897                         self.issue_channel_close_events(&chan.context, ClosureReason::CooperativeClosure);
5898                 }
5899                 Ok(())
5900         }
5901
5902         fn internal_update_add_htlc(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateAddHTLC) -> Result<(), MsgHandleErrInternal> {
5903                 //TODO: BOLT 4 points out a specific attack where a peer may re-send an onion packet and
5904                 //determine the state of the payment based on our response/if we forward anything/the time
5905                 //we take to respond. We should take care to avoid allowing such an attack.
5906                 //
5907                 //TODO: There exists a further attack where a node may garble the onion data, forward it to
5908                 //us repeatedly garbled in different ways, and compare our error messages, which are
5909                 //encrypted with the same key. It's not immediately obvious how to usefully exploit that,
5910                 //but we should prevent it anyway.
5911
5912                 let decoded_hop_res = self.decode_update_add_htlc_onion(msg);
5913                 let per_peer_state = self.per_peer_state.read().unwrap();
5914                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5915                         .ok_or_else(|| {
5916                                 debug_assert!(false);
5917                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
5918                         })?;
5919                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5920                 let peer_state = &mut *peer_state_lock;
5921                 match peer_state.channel_by_id.entry(msg.channel_id) {
5922                         hash_map::Entry::Occupied(mut chan) => {
5923
5924                                 let pending_forward_info = match decoded_hop_res {
5925                                         Ok((next_hop, shared_secret, next_packet_pk_opt)) =>
5926                                                 self.construct_pending_htlc_status(msg, shared_secret, next_hop,
5927                                                         chan.get().context.config().accept_underpaying_htlcs, next_packet_pk_opt),
5928                                         Err(e) => PendingHTLCStatus::Fail(e)
5929                                 };
5930                                 let create_pending_htlc_status = |chan: &Channel<SP>, pending_forward_info: PendingHTLCStatus, error_code: u16| {
5931                                         // If the update_add is completely bogus, the call will Err and we will close,
5932                                         // but if we've sent a shutdown and they haven't acknowledged it yet, we just
5933                                         // want to reject the new HTLC and fail it backwards instead of forwarding.
5934                                         match pending_forward_info {
5935                                                 PendingHTLCStatus::Forward(PendingHTLCInfo { ref incoming_shared_secret, .. }) => {
5936                                                         let reason = if (error_code & 0x1000) != 0 {
5937                                                                 let (real_code, error_data) = self.get_htlc_inbound_temp_fail_err_and_data(error_code, chan);
5938                                                                 HTLCFailReason::reason(real_code, error_data)
5939                                                         } else {
5940                                                                 HTLCFailReason::from_failure_code(error_code)
5941                                                         }.get_encrypted_failure_packet(incoming_shared_secret, &None);
5942                                                         let msg = msgs::UpdateFailHTLC {
5943                                                                 channel_id: msg.channel_id,
5944                                                                 htlc_id: msg.htlc_id,
5945                                                                 reason
5946                                                         };
5947                                                         PendingHTLCStatus::Fail(HTLCFailureMsg::Relay(msg))
5948                                                 },
5949                                                 _ => pending_forward_info
5950                                         }
5951                                 };
5952                                 try_chan_entry!(self, chan.get_mut().update_add_htlc(&msg, pending_forward_info, create_pending_htlc_status, &self.fee_estimator, &self.logger), chan);
5953                         },
5954                         hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
5955                 }
5956                 Ok(())
5957         }
5958
5959         fn internal_update_fulfill_htlc(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateFulfillHTLC) -> Result<(), MsgHandleErrInternal> {
5960                 let funding_txo;
5961                 let (htlc_source, forwarded_htlc_value) = {
5962                         let per_peer_state = self.per_peer_state.read().unwrap();
5963                         let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5964                                 .ok_or_else(|| {
5965                                         debug_assert!(false);
5966                                         MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
5967                                 })?;
5968                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5969                         let peer_state = &mut *peer_state_lock;
5970                         match peer_state.channel_by_id.entry(msg.channel_id) {
5971                                 hash_map::Entry::Occupied(mut chan) => {
5972                                         let res = try_chan_entry!(self, chan.get_mut().update_fulfill_htlc(&msg), chan);
5973                                         funding_txo = chan.get().context.get_funding_txo().expect("We won't accept a fulfill until funded");
5974                                         res
5975                                 },
5976                                 hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
5977                         }
5978                 };
5979                 self.claim_funds_internal(htlc_source, msg.payment_preimage.clone(), Some(forwarded_htlc_value), false, funding_txo);
5980                 Ok(())
5981         }
5982
5983         fn internal_update_fail_htlc(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateFailHTLC) -> Result<(), MsgHandleErrInternal> {
5984                 let per_peer_state = self.per_peer_state.read().unwrap();
5985                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
5986                         .ok_or_else(|| {
5987                                 debug_assert!(false);
5988                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
5989                         })?;
5990                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
5991                 let peer_state = &mut *peer_state_lock;
5992                 match peer_state.channel_by_id.entry(msg.channel_id) {
5993                         hash_map::Entry::Occupied(mut chan) => {
5994                                 try_chan_entry!(self, chan.get_mut().update_fail_htlc(&msg, HTLCFailReason::from_msg(msg)), chan);
5995                         },
5996                         hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
5997                 }
5998                 Ok(())
5999         }
6000
6001         fn internal_update_fail_malformed_htlc(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateFailMalformedHTLC) -> Result<(), MsgHandleErrInternal> {
6002                 let per_peer_state = self.per_peer_state.read().unwrap();
6003                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
6004                         .ok_or_else(|| {
6005                                 debug_assert!(false);
6006                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
6007                         })?;
6008                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6009                 let peer_state = &mut *peer_state_lock;
6010                 match peer_state.channel_by_id.entry(msg.channel_id) {
6011                         hash_map::Entry::Occupied(mut chan) => {
6012                                 if (msg.failure_code & 0x8000) == 0 {
6013                                         let chan_err: ChannelError = ChannelError::Close("Got update_fail_malformed_htlc with BADONION not set".to_owned());
6014                                         try_chan_entry!(self, Err(chan_err), chan);
6015                                 }
6016                                 try_chan_entry!(self, chan.get_mut().update_fail_malformed_htlc(&msg, HTLCFailReason::reason(msg.failure_code, msg.sha256_of_onion.to_vec())), chan);
6017                                 Ok(())
6018                         },
6019                         hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
6020                 }
6021         }
6022
6023         fn internal_commitment_signed(&self, counterparty_node_id: &PublicKey, msg: &msgs::CommitmentSigned) -> Result<(), MsgHandleErrInternal> {
6024                 let per_peer_state = self.per_peer_state.read().unwrap();
6025                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
6026                         .ok_or_else(|| {
6027                                 debug_assert!(false);
6028                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
6029                         })?;
6030                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6031                 let peer_state = &mut *peer_state_lock;
6032                 match peer_state.channel_by_id.entry(msg.channel_id) {
6033                         hash_map::Entry::Occupied(mut chan) => {
6034                                 let funding_txo = chan.get().context.get_funding_txo();
6035                                 let monitor_update_opt = try_chan_entry!(self, chan.get_mut().commitment_signed(&msg, &self.logger), chan);
6036                                 if let Some(monitor_update) = monitor_update_opt {
6037                                         handle_new_monitor_update!(self, funding_txo.unwrap(), monitor_update, peer_state_lock,
6038                                                 peer_state, per_peer_state, chan).map(|_| ())
6039                                 } else { Ok(()) }
6040                         },
6041                         hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
6042                 }
6043         }
6044
6045         #[inline]
6046         fn forward_htlcs(&self, per_source_pending_forwards: &mut [(u64, OutPoint, u128, Vec<(PendingHTLCInfo, u64)>)]) {
6047                 for &mut (prev_short_channel_id, prev_funding_outpoint, prev_user_channel_id, ref mut pending_forwards) in per_source_pending_forwards {
6048                         let mut push_forward_event = false;
6049                         let mut new_intercept_events = VecDeque::new();
6050                         let mut failed_intercept_forwards = Vec::new();
6051                         if !pending_forwards.is_empty() {
6052                                 for (forward_info, prev_htlc_id) in pending_forwards.drain(..) {
6053                                         let scid = match forward_info.routing {
6054                                                 PendingHTLCRouting::Forward { short_channel_id, .. } => short_channel_id,
6055                                                 PendingHTLCRouting::Receive { .. } => 0,
6056                                                 PendingHTLCRouting::ReceiveKeysend { .. } => 0,
6057                                         };
6058                                         // Pull this now to avoid introducing a lock order with `forward_htlcs`.
6059                                         let is_our_scid = self.short_to_chan_info.read().unwrap().contains_key(&scid);
6060
6061                                         let mut forward_htlcs = self.forward_htlcs.lock().unwrap();
6062                                         let forward_htlcs_empty = forward_htlcs.is_empty();
6063                                         match forward_htlcs.entry(scid) {
6064                                                 hash_map::Entry::Occupied(mut entry) => {
6065                                                         entry.get_mut().push(HTLCForwardInfo::AddHTLC(PendingAddHTLCInfo {
6066                                                                 prev_short_channel_id, prev_funding_outpoint, prev_htlc_id, prev_user_channel_id, forward_info }));
6067                                                 },
6068                                                 hash_map::Entry::Vacant(entry) => {
6069                                                         if !is_our_scid && forward_info.incoming_amt_msat.is_some() &&
6070                                                            fake_scid::is_valid_intercept(&self.fake_scid_rand_bytes, scid, &self.genesis_hash)
6071                                                         {
6072                                                                 let intercept_id = InterceptId(Sha256::hash(&forward_info.incoming_shared_secret).into_inner());
6073                                                                 let mut pending_intercepts = self.pending_intercepted_htlcs.lock().unwrap();
6074                                                                 match pending_intercepts.entry(intercept_id) {
6075                                                                         hash_map::Entry::Vacant(entry) => {
6076                                                                                 new_intercept_events.push_back((events::Event::HTLCIntercepted {
6077                                                                                         requested_next_hop_scid: scid,
6078                                                                                         payment_hash: forward_info.payment_hash,
6079                                                                                         inbound_amount_msat: forward_info.incoming_amt_msat.unwrap(),
6080                                                                                         expected_outbound_amount_msat: forward_info.outgoing_amt_msat,
6081                                                                                         intercept_id
6082                                                                                 }, None));
6083                                                                                 entry.insert(PendingAddHTLCInfo {
6084                                                                                         prev_short_channel_id, prev_funding_outpoint, prev_htlc_id, prev_user_channel_id, forward_info });
6085                                                                         },
6086                                                                         hash_map::Entry::Occupied(_) => {
6087                                                                                 log_info!(self.logger, "Failed to forward incoming HTLC: detected duplicate intercepted payment over short channel id {}", scid);
6088                                                                                 let htlc_source = HTLCSource::PreviousHopData(HTLCPreviousHopData {
6089                                                                                         short_channel_id: prev_short_channel_id,
6090                                                                                         user_channel_id: Some(prev_user_channel_id),
6091                                                                                         outpoint: prev_funding_outpoint,
6092                                                                                         htlc_id: prev_htlc_id,
6093                                                                                         incoming_packet_shared_secret: forward_info.incoming_shared_secret,
6094                                                                                         phantom_shared_secret: None,
6095                                                                                 });
6096
6097                                                                                 failed_intercept_forwards.push((htlc_source, forward_info.payment_hash,
6098                                                                                                 HTLCFailReason::from_failure_code(0x4000 | 10),
6099                                                                                                 HTLCDestination::InvalidForward { requested_forward_scid: scid },
6100                                                                                 ));
6101                                                                         }
6102                                                                 }
6103                                                         } else {
6104                                                                 // We don't want to generate a PendingHTLCsForwardable event if only intercepted
6105                                                                 // payments are being processed.
6106                                                                 if forward_htlcs_empty {
6107                                                                         push_forward_event = true;
6108                                                                 }
6109                                                                 entry.insert(vec!(HTLCForwardInfo::AddHTLC(PendingAddHTLCInfo {
6110                                                                         prev_short_channel_id, prev_funding_outpoint, prev_htlc_id, prev_user_channel_id, forward_info })));
6111                                                         }
6112                                                 }
6113                                         }
6114                                 }
6115                         }
6116
6117                         for (htlc_source, payment_hash, failure_reason, destination) in failed_intercept_forwards.drain(..) {
6118                                 self.fail_htlc_backwards_internal(&htlc_source, &payment_hash, &failure_reason, destination);
6119                         }
6120
6121                         if !new_intercept_events.is_empty() {
6122                                 let mut events = self.pending_events.lock().unwrap();
6123                                 events.append(&mut new_intercept_events);
6124                         }
6125                         if push_forward_event { self.push_pending_forwards_ev() }
6126                 }
6127         }
6128
6129         fn push_pending_forwards_ev(&self) {
6130                 let mut pending_events = self.pending_events.lock().unwrap();
6131                 let is_processing_events = self.pending_events_processor.load(Ordering::Acquire);
6132                 let num_forward_events = pending_events.iter().filter(|(ev, _)|
6133                         if let events::Event::PendingHTLCsForwardable { .. } = ev { true } else { false }
6134                 ).count();
6135                 // We only want to push a PendingHTLCsForwardable event if no others are queued. Processing
6136                 // events is done in batches and they are not removed until we're done processing each
6137                 // batch. Since handling a `PendingHTLCsForwardable` event will call back into the
6138                 // `ChannelManager`, we'll still see the original forwarding event not removed. Phantom
6139                 // payments will need an additional forwarding event before being claimed to make them look
6140                 // real by taking more time.
6141                 if (is_processing_events && num_forward_events <= 1) || num_forward_events < 1 {
6142                         pending_events.push_back((Event::PendingHTLCsForwardable {
6143                                 time_forwardable: Duration::from_millis(MIN_HTLC_RELAY_HOLDING_CELL_MILLIS),
6144                         }, None));
6145                 }
6146         }
6147
6148         /// Checks whether [`ChannelMonitorUpdate`]s generated by the receipt of a remote
6149         /// [`msgs::RevokeAndACK`] should be held for the given channel until some other action
6150         /// completes. Note that this needs to happen in the same [`PeerState`] mutex as any release of
6151         /// the [`ChannelMonitorUpdate`] in question.
6152         fn raa_monitor_updates_held(&self,
6153                 actions_blocking_raa_monitor_updates: &BTreeMap<ChannelId, Vec<RAAMonitorUpdateBlockingAction>>,
6154                 channel_funding_outpoint: OutPoint, counterparty_node_id: PublicKey
6155         ) -> bool {
6156                 actions_blocking_raa_monitor_updates
6157                         .get(&channel_funding_outpoint.to_channel_id()).map(|v| !v.is_empty()).unwrap_or(false)
6158                 || self.pending_events.lock().unwrap().iter().any(|(_, action)| {
6159                         action == &Some(EventCompletionAction::ReleaseRAAChannelMonitorUpdate {
6160                                 channel_funding_outpoint,
6161                                 counterparty_node_id,
6162                         })
6163                 })
6164         }
6165
6166         fn internal_revoke_and_ack(&self, counterparty_node_id: &PublicKey, msg: &msgs::RevokeAndACK) -> Result<(), MsgHandleErrInternal> {
6167                 let (htlcs_to_fail, res) = {
6168                         let per_peer_state = self.per_peer_state.read().unwrap();
6169                         let mut peer_state_lock = per_peer_state.get(counterparty_node_id)
6170                                 .ok_or_else(|| {
6171                                         debug_assert!(false);
6172                                         MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
6173                                 }).map(|mtx| mtx.lock().unwrap())?;
6174                         let peer_state = &mut *peer_state_lock;
6175                         match peer_state.channel_by_id.entry(msg.channel_id) {
6176                                 hash_map::Entry::Occupied(mut chan) => {
6177                                         let funding_txo_opt = chan.get().context.get_funding_txo();
6178                                         let mon_update_blocked = if let Some(funding_txo) = funding_txo_opt {
6179                                                 self.raa_monitor_updates_held(
6180                                                         &peer_state.actions_blocking_raa_monitor_updates, funding_txo,
6181                                                         *counterparty_node_id)
6182                                         } else { false };
6183                                         let (htlcs_to_fail, monitor_update_opt) = try_chan_entry!(self,
6184                                                 chan.get_mut().revoke_and_ack(&msg, &self.fee_estimator, &self.logger, mon_update_blocked), chan);
6185                                         let res = if let Some(monitor_update) = monitor_update_opt {
6186                                                 let funding_txo = funding_txo_opt
6187                                                         .expect("Funding outpoint must have been set for RAA handling to succeed");
6188                                                 handle_new_monitor_update!(self, funding_txo, monitor_update,
6189                                                         peer_state_lock, peer_state, per_peer_state, chan).map(|_| ())
6190                                         } else { Ok(()) };
6191                                         (htlcs_to_fail, res)
6192                                 },
6193                                 hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
6194                         }
6195                 };
6196                 self.fail_holding_cell_htlcs(htlcs_to_fail, msg.channel_id, counterparty_node_id);
6197                 res
6198         }
6199
6200         fn internal_update_fee(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateFee) -> Result<(), MsgHandleErrInternal> {
6201                 let per_peer_state = self.per_peer_state.read().unwrap();
6202                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
6203                         .ok_or_else(|| {
6204                                 debug_assert!(false);
6205                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
6206                         })?;
6207                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6208                 let peer_state = &mut *peer_state_lock;
6209                 match peer_state.channel_by_id.entry(msg.channel_id) {
6210                         hash_map::Entry::Occupied(mut chan) => {
6211                                 try_chan_entry!(self, chan.get_mut().update_fee(&self.fee_estimator, &msg, &self.logger), chan);
6212                         },
6213                         hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
6214                 }
6215                 Ok(())
6216         }
6217
6218         fn internal_announcement_signatures(&self, counterparty_node_id: &PublicKey, msg: &msgs::AnnouncementSignatures) -> Result<(), MsgHandleErrInternal> {
6219                 let per_peer_state = self.per_peer_state.read().unwrap();
6220                 let peer_state_mutex = per_peer_state.get(counterparty_node_id)
6221                         .ok_or_else(|| {
6222                                 debug_assert!(false);
6223                                 MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
6224                         })?;
6225                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6226                 let peer_state = &mut *peer_state_lock;
6227                 match peer_state.channel_by_id.entry(msg.channel_id) {
6228                         hash_map::Entry::Occupied(mut chan) => {
6229                                 if !chan.get().context.is_usable() {
6230                                         return Err(MsgHandleErrInternal::from_no_close(LightningError{err: "Got an announcement_signatures before we were ready for it".to_owned(), action: msgs::ErrorAction::IgnoreError}));
6231                                 }
6232
6233                                 peer_state.pending_msg_events.push(events::MessageSendEvent::BroadcastChannelAnnouncement {
6234                                         msg: try_chan_entry!(self, chan.get_mut().announcement_signatures(
6235                                                 &self.node_signer, self.genesis_hash.clone(), self.best_block.read().unwrap().height(),
6236                                                 msg, &self.default_configuration
6237                                         ), chan),
6238                                         // Note that announcement_signatures fails if the channel cannot be announced,
6239                                         // so get_channel_update_for_broadcast will never fail by the time we get here.
6240                                         update_msg: Some(self.get_channel_update_for_broadcast(chan.get()).unwrap()),
6241                                 });
6242                         },
6243                         hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
6244                 }
6245                 Ok(())
6246         }
6247
6248         /// Returns ShouldPersist if anything changed, otherwise either SkipPersist or an Err.
6249         fn internal_channel_update(&self, counterparty_node_id: &PublicKey, msg: &msgs::ChannelUpdate) -> Result<NotifyOption, MsgHandleErrInternal> {
6250                 let (chan_counterparty_node_id, chan_id) = match self.short_to_chan_info.read().unwrap().get(&msg.contents.short_channel_id) {
6251                         Some((cp_id, chan_id)) => (cp_id.clone(), chan_id.clone()),
6252                         None => {
6253                                 // It's not a local channel
6254                                 return Ok(NotifyOption::SkipPersist)
6255                         }
6256                 };
6257                 let per_peer_state = self.per_peer_state.read().unwrap();
6258                 let peer_state_mutex_opt = per_peer_state.get(&chan_counterparty_node_id);
6259                 if peer_state_mutex_opt.is_none() {
6260                         return Ok(NotifyOption::SkipPersist)
6261                 }
6262                 let mut peer_state_lock = peer_state_mutex_opt.unwrap().lock().unwrap();
6263                 let peer_state = &mut *peer_state_lock;
6264                 match peer_state.channel_by_id.entry(chan_id) {
6265                         hash_map::Entry::Occupied(mut chan) => {
6266                                 if chan.get().context.get_counterparty_node_id() != *counterparty_node_id {
6267                                         if chan.get().context.should_announce() {
6268                                                 // If the announcement is about a channel of ours which is public, some
6269                                                 // other peer may simply be forwarding all its gossip to us. Don't provide
6270                                                 // a scary-looking error message and return Ok instead.
6271                                                 return Ok(NotifyOption::SkipPersist);
6272                                         }
6273                                         return Err(MsgHandleErrInternal::send_err_msg_no_close("Got a channel_update for a channel from the wrong node - it shouldn't know about our private channels!".to_owned(), chan_id));
6274                                 }
6275                                 let were_node_one = self.get_our_node_id().serialize()[..] < chan.get().context.get_counterparty_node_id().serialize()[..];
6276                                 let msg_from_node_one = msg.contents.flags & 1 == 0;
6277                                 if were_node_one == msg_from_node_one {
6278                                         return Ok(NotifyOption::SkipPersist);
6279                                 } else {
6280                                         log_debug!(self.logger, "Received channel_update for channel {}.", &chan_id);
6281                                         try_chan_entry!(self, chan.get_mut().channel_update(&msg), chan);
6282                                 }
6283                         },
6284                         hash_map::Entry::Vacant(_) => return Ok(NotifyOption::SkipPersist)
6285                 }
6286                 Ok(NotifyOption::DoPersist)
6287         }
6288
6289         fn internal_channel_reestablish(&self, counterparty_node_id: &PublicKey, msg: &msgs::ChannelReestablish) -> Result<(), MsgHandleErrInternal> {
6290                 let htlc_forwards;
6291                 let need_lnd_workaround = {
6292                         let per_peer_state = self.per_peer_state.read().unwrap();
6293
6294                         let peer_state_mutex = per_peer_state.get(counterparty_node_id)
6295                                 .ok_or_else(|| {
6296                                         debug_assert!(false);
6297                                         MsgHandleErrInternal::send_err_msg_no_close(format!("Can't find a peer matching the passed counterparty node_id {}", counterparty_node_id), msg.channel_id)
6298                                 })?;
6299                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6300                         let peer_state = &mut *peer_state_lock;
6301                         match peer_state.channel_by_id.entry(msg.channel_id) {
6302                                 hash_map::Entry::Occupied(mut chan) => {
6303                                         // Currently, we expect all holding cell update_adds to be dropped on peer
6304                                         // disconnect, so Channel's reestablish will never hand us any holding cell
6305                                         // freed HTLCs to fail backwards. If in the future we no longer drop pending
6306                                         // add-HTLCs on disconnect, we may be handed HTLCs to fail backwards here.
6307                                         let responses = try_chan_entry!(self, chan.get_mut().channel_reestablish(
6308                                                 msg, &self.logger, &self.node_signer, self.genesis_hash,
6309                                                 &self.default_configuration, &*self.best_block.read().unwrap()), chan);
6310                                         let mut channel_update = None;
6311                                         if let Some(msg) = responses.shutdown_msg {
6312                                                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendShutdown {
6313                                                         node_id: counterparty_node_id.clone(),
6314                                                         msg,
6315                                                 });
6316                                         } else if chan.get().context.is_usable() {
6317                                                 // If the channel is in a usable state (ie the channel is not being shut
6318                                                 // down), send a unicast channel_update to our counterparty to make sure
6319                                                 // they have the latest channel parameters.
6320                                                 if let Ok(msg) = self.get_channel_update_for_unicast(chan.get()) {
6321                                                         channel_update = Some(events::MessageSendEvent::SendChannelUpdate {
6322                                                                 node_id: chan.get().context.get_counterparty_node_id(),
6323                                                                 msg,
6324                                                         });
6325                                                 }
6326                                         }
6327                                         let need_lnd_workaround = chan.get_mut().context.workaround_lnd_bug_4006.take();
6328                                         htlc_forwards = self.handle_channel_resumption(
6329                                                 &mut peer_state.pending_msg_events, chan.get_mut(), responses.raa, responses.commitment_update, responses.order,
6330                                                 Vec::new(), None, responses.channel_ready, responses.announcement_sigs);
6331                                         if let Some(upd) = channel_update {
6332                                                 peer_state.pending_msg_events.push(upd);
6333                                         }
6334                                         need_lnd_workaround
6335                                 },
6336                                 hash_map::Entry::Vacant(_) => return Err(MsgHandleErrInternal::send_err_msg_no_close(format!("Got a message for a channel from the wrong node! No such channel for the passed counterparty_node_id {}", counterparty_node_id), msg.channel_id))
6337                         }
6338                 };
6339
6340                 if let Some(forwards) = htlc_forwards {
6341                         self.forward_htlcs(&mut [forwards][..]);
6342                 }
6343
6344                 if let Some(channel_ready_msg) = need_lnd_workaround {
6345                         self.internal_channel_ready(counterparty_node_id, &channel_ready_msg)?;
6346                 }
6347                 Ok(())
6348         }
6349
6350         /// Process pending events from the [`chain::Watch`], returning whether any events were processed.
6351         fn process_pending_monitor_events(&self) -> bool {
6352                 debug_assert!(self.total_consistency_lock.try_write().is_err()); // Caller holds read lock
6353
6354                 let mut failed_channels = Vec::new();
6355                 let mut pending_monitor_events = self.chain_monitor.release_pending_monitor_events();
6356                 let has_pending_monitor_events = !pending_monitor_events.is_empty();
6357                 for (funding_outpoint, mut monitor_events, counterparty_node_id) in pending_monitor_events.drain(..) {
6358                         for monitor_event in monitor_events.drain(..) {
6359                                 match monitor_event {
6360                                         MonitorEvent::HTLCEvent(htlc_update) => {
6361                                                 if let Some(preimage) = htlc_update.payment_preimage {
6362                                                         log_trace!(self.logger, "Claiming HTLC with preimage {} from our monitor", &preimage);
6363                                                         self.claim_funds_internal(htlc_update.source, preimage, htlc_update.htlc_value_satoshis.map(|v| v * 1000), true, funding_outpoint);
6364                                                 } else {
6365                                                         log_trace!(self.logger, "Failing HTLC with hash {} from our monitor", &htlc_update.payment_hash);
6366                                                         let receiver = HTLCDestination::NextHopChannel { node_id: counterparty_node_id, channel_id: funding_outpoint.to_channel_id() };
6367                                                         let reason = HTLCFailReason::from_failure_code(0x4000 | 8);
6368                                                         self.fail_htlc_backwards_internal(&htlc_update.source, &htlc_update.payment_hash, &reason, receiver);
6369                                                 }
6370                                         },
6371                                         MonitorEvent::CommitmentTxConfirmed(funding_outpoint) |
6372                                         MonitorEvent::UpdateFailed(funding_outpoint) => {
6373                                                 let counterparty_node_id_opt = match counterparty_node_id {
6374                                                         Some(cp_id) => Some(cp_id),
6375                                                         None => {
6376                                                                 // TODO: Once we can rely on the counterparty_node_id from the
6377                                                                 // monitor event, this and the id_to_peer map should be removed.
6378                                                                 let id_to_peer = self.id_to_peer.lock().unwrap();
6379                                                                 id_to_peer.get(&funding_outpoint.to_channel_id()).cloned()
6380                                                         }
6381                                                 };
6382                                                 if let Some(counterparty_node_id) = counterparty_node_id_opt {
6383                                                         let per_peer_state = self.per_peer_state.read().unwrap();
6384                                                         if let Some(peer_state_mutex) = per_peer_state.get(&counterparty_node_id) {
6385                                                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6386                                                                 let peer_state = &mut *peer_state_lock;
6387                                                                 let pending_msg_events = &mut peer_state.pending_msg_events;
6388                                                                 if let hash_map::Entry::Occupied(chan_entry) = peer_state.channel_by_id.entry(funding_outpoint.to_channel_id()) {
6389                                                                         let mut chan = remove_channel!(self, chan_entry);
6390                                                                         failed_channels.push(chan.context.force_shutdown(false));
6391                                                                         if let Ok(update) = self.get_channel_update_for_broadcast(&chan) {
6392                                                                                 pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
6393                                                                                         msg: update
6394                                                                                 });
6395                                                                         }
6396                                                                         let reason = if let MonitorEvent::UpdateFailed(_) = monitor_event {
6397                                                                                 ClosureReason::ProcessingError { err: "Failed to persist ChannelMonitor update during chain sync".to_string() }
6398                                                                         } else {
6399                                                                                 ClosureReason::CommitmentTxConfirmed
6400                                                                         };
6401                                                                         self.issue_channel_close_events(&chan.context, reason);
6402                                                                         pending_msg_events.push(events::MessageSendEvent::HandleError {
6403                                                                                 node_id: chan.context.get_counterparty_node_id(),
6404                                                                                 action: msgs::ErrorAction::SendErrorMessage {
6405                                                                                         msg: msgs::ErrorMessage { channel_id: chan.context.channel_id(), data: "Channel force-closed".to_owned() }
6406                                                                                 },
6407                                                                         });
6408                                                                 }
6409                                                         }
6410                                                 }
6411                                         },
6412                                         MonitorEvent::Completed { funding_txo, monitor_update_id } => {
6413                                                 self.channel_monitor_updated(&funding_txo, monitor_update_id, counterparty_node_id.as_ref());
6414                                         },
6415                                 }
6416                         }
6417                 }
6418
6419                 for failure in failed_channels.drain(..) {
6420                         self.finish_force_close_channel(failure);
6421                 }
6422
6423                 has_pending_monitor_events
6424         }
6425
6426         /// In chanmon_consistency_target, we'd like to be able to restore monitor updating without
6427         /// handling all pending events (i.e. not PendingHTLCsForwardable). Thus, we expose monitor
6428         /// update events as a separate process method here.
6429         #[cfg(fuzzing)]
6430         pub fn process_monitor_events(&self) {
6431                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
6432                 self.process_pending_monitor_events();
6433         }
6434
6435         /// Check the holding cell in each channel and free any pending HTLCs in them if possible.
6436         /// Returns whether there were any updates such as if pending HTLCs were freed or a monitor
6437         /// update was applied.
6438         fn check_free_holding_cells(&self) -> bool {
6439                 let mut has_monitor_update = false;
6440                 let mut failed_htlcs = Vec::new();
6441                 let mut handle_errors = Vec::new();
6442
6443                 // Walk our list of channels and find any that need to update. Note that when we do find an
6444                 // update, if it includes actions that must be taken afterwards, we have to drop the
6445                 // per-peer state lock as well as the top level per_peer_state lock. Thus, we loop until we
6446                 // manage to go through all our peers without finding a single channel to update.
6447                 'peer_loop: loop {
6448                         let per_peer_state = self.per_peer_state.read().unwrap();
6449                         for (_cp_id, peer_state_mutex) in per_peer_state.iter() {
6450                                 'chan_loop: loop {
6451                                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6452                                         let peer_state: &mut PeerState<_> = &mut *peer_state_lock;
6453                                         for (channel_id, chan) in peer_state.channel_by_id.iter_mut() {
6454                                                 let counterparty_node_id = chan.context.get_counterparty_node_id();
6455                                                 let funding_txo = chan.context.get_funding_txo();
6456                                                 let (monitor_opt, holding_cell_failed_htlcs) =
6457                                                         chan.maybe_free_holding_cell_htlcs(&self.fee_estimator, &self.logger);
6458                                                 if !holding_cell_failed_htlcs.is_empty() {
6459                                                         failed_htlcs.push((holding_cell_failed_htlcs, *channel_id, counterparty_node_id));
6460                                                 }
6461                                                 if let Some(monitor_update) = monitor_opt {
6462                                                         has_monitor_update = true;
6463
6464                                                         let channel_id: ChannelId = *channel_id;
6465                                                         let res = handle_new_monitor_update!(self, funding_txo.unwrap(), monitor_update,
6466                                                                 peer_state_lock, peer_state, per_peer_state, chan, MANUALLY_REMOVING,
6467                                                                 peer_state.channel_by_id.remove(&channel_id));
6468                                                         if res.is_err() {
6469                                                                 handle_errors.push((counterparty_node_id, res));
6470                                                         }
6471                                                         continue 'peer_loop;
6472                                                 }
6473                                         }
6474                                         break 'chan_loop;
6475                                 }
6476                         }
6477                         break 'peer_loop;
6478                 }
6479
6480                 let has_update = has_monitor_update || !failed_htlcs.is_empty() || !handle_errors.is_empty();
6481                 for (failures, channel_id, counterparty_node_id) in failed_htlcs.drain(..) {
6482                         self.fail_holding_cell_htlcs(failures, channel_id, &counterparty_node_id);
6483                 }
6484
6485                 for (counterparty_node_id, err) in handle_errors.drain(..) {
6486                         let _ = handle_error!(self, err, counterparty_node_id);
6487                 }
6488
6489                 has_update
6490         }
6491
6492         /// Check whether any channels have finished removing all pending updates after a shutdown
6493         /// exchange and can now send a closing_signed.
6494         /// Returns whether any closing_signed messages were generated.
6495         fn maybe_generate_initial_closing_signed(&self) -> bool {
6496                 let mut handle_errors: Vec<(PublicKey, Result<(), _>)> = Vec::new();
6497                 let mut has_update = false;
6498                 {
6499                         let per_peer_state = self.per_peer_state.read().unwrap();
6500
6501                         for (_cp_id, peer_state_mutex) in per_peer_state.iter() {
6502                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6503                                 let peer_state = &mut *peer_state_lock;
6504                                 let pending_msg_events = &mut peer_state.pending_msg_events;
6505                                 peer_state.channel_by_id.retain(|channel_id, chan| {
6506                                         match chan.maybe_propose_closing_signed(&self.fee_estimator, &self.logger) {
6507                                                 Ok((msg_opt, tx_opt)) => {
6508                                                         if let Some(msg) = msg_opt {
6509                                                                 has_update = true;
6510                                                                 pending_msg_events.push(events::MessageSendEvent::SendClosingSigned {
6511                                                                         node_id: chan.context.get_counterparty_node_id(), msg,
6512                                                                 });
6513                                                         }
6514                                                         if let Some(tx) = tx_opt {
6515                                                                 // We're done with this channel. We got a closing_signed and sent back
6516                                                                 // a closing_signed with a closing transaction to broadcast.
6517                                                                 if let Ok(update) = self.get_channel_update_for_broadcast(&chan) {
6518                                                                         pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
6519                                                                                 msg: update
6520                                                                         });
6521                                                                 }
6522
6523                                                                 self.issue_channel_close_events(&chan.context, ClosureReason::CooperativeClosure);
6524
6525                                                                 log_info!(self.logger, "Broadcasting {}", log_tx!(tx));
6526                                                                 self.tx_broadcaster.broadcast_transactions(&[&tx]);
6527                                                                 update_maps_on_chan_removal!(self, &chan.context);
6528                                                                 false
6529                                                         } else { true }
6530                                                 },
6531                                                 Err(e) => {
6532                                                         has_update = true;
6533                                                         let (close_channel, res) = convert_chan_err!(self, e, chan, channel_id);
6534                                                         handle_errors.push((chan.context.get_counterparty_node_id(), Err(res)));
6535                                                         !close_channel
6536                                                 }
6537                                         }
6538                                 });
6539                         }
6540                 }
6541
6542                 for (counterparty_node_id, err) in handle_errors.drain(..) {
6543                         let _ = handle_error!(self, err, counterparty_node_id);
6544                 }
6545
6546                 has_update
6547         }
6548
6549         /// Handle a list of channel failures during a block_connected or block_disconnected call,
6550         /// pushing the channel monitor update (if any) to the background events queue and removing the
6551         /// Channel object.
6552         fn handle_init_event_channel_failures(&self, mut failed_channels: Vec<ShutdownResult>) {
6553                 for mut failure in failed_channels.drain(..) {
6554                         // Either a commitment transactions has been confirmed on-chain or
6555                         // Channel::block_disconnected detected that the funding transaction has been
6556                         // reorganized out of the main chain.
6557                         // We cannot broadcast our latest local state via monitor update (as
6558                         // Channel::force_shutdown tries to make us do) as we may still be in initialization,
6559                         // so we track the update internally and handle it when the user next calls
6560                         // timer_tick_occurred, guaranteeing we're running normally.
6561                         if let Some((counterparty_node_id, funding_txo, update)) = failure.0.take() {
6562                                 assert_eq!(update.updates.len(), 1);
6563                                 if let ChannelMonitorUpdateStep::ChannelForceClosed { should_broadcast } = update.updates[0] {
6564                                         assert!(should_broadcast);
6565                                 } else { unreachable!(); }
6566                                 self.pending_background_events.lock().unwrap().push(
6567                                         BackgroundEvent::MonitorUpdateRegeneratedOnStartup {
6568                                                 counterparty_node_id, funding_txo, update
6569                                         });
6570                         }
6571                         self.finish_force_close_channel(failure);
6572                 }
6573         }
6574
6575         /// Gets a payment secret and payment hash for use in an invoice given to a third party wishing
6576         /// to pay us.
6577         ///
6578         /// This differs from [`create_inbound_payment_for_hash`] only in that it generates the
6579         /// [`PaymentHash`] and [`PaymentPreimage`] for you.
6580         ///
6581         /// The [`PaymentPreimage`] will ultimately be returned to you in the [`PaymentClaimable`], which
6582         /// will have the [`PaymentClaimable::purpose`] be [`PaymentPurpose::InvoicePayment`] with
6583         /// its [`PaymentPurpose::InvoicePayment::payment_preimage`] field filled in. That should then be
6584         /// passed directly to [`claim_funds`].
6585         ///
6586         /// See [`create_inbound_payment_for_hash`] for detailed documentation on behavior and requirements.
6587         ///
6588         /// Note that a malicious eavesdropper can intuit whether an inbound payment was created by
6589         /// `create_inbound_payment` or `create_inbound_payment_for_hash` based on runtime.
6590         ///
6591         /// # Note
6592         ///
6593         /// If you register an inbound payment with this method, then serialize the `ChannelManager`, then
6594         /// deserialize it with a node running 0.0.103 and earlier, the payment will fail to be received.
6595         ///
6596         /// Errors if `min_value_msat` is greater than total bitcoin supply.
6597         ///
6598         /// If `min_final_cltv_expiry_delta` is set to some value, then the payment will not be receivable
6599         /// on versions of LDK prior to 0.0.114.
6600         ///
6601         /// [`claim_funds`]: Self::claim_funds
6602         /// [`PaymentClaimable`]: events::Event::PaymentClaimable
6603         /// [`PaymentClaimable::purpose`]: events::Event::PaymentClaimable::purpose
6604         /// [`PaymentPurpose::InvoicePayment`]: events::PaymentPurpose::InvoicePayment
6605         /// [`PaymentPurpose::InvoicePayment::payment_preimage`]: events::PaymentPurpose::InvoicePayment::payment_preimage
6606         /// [`create_inbound_payment_for_hash`]: Self::create_inbound_payment_for_hash
6607         pub fn create_inbound_payment(&self, min_value_msat: Option<u64>, invoice_expiry_delta_secs: u32,
6608                 min_final_cltv_expiry_delta: Option<u16>) -> Result<(PaymentHash, PaymentSecret), ()> {
6609                 inbound_payment::create(&self.inbound_payment_key, min_value_msat, invoice_expiry_delta_secs,
6610                         &self.entropy_source, self.highest_seen_timestamp.load(Ordering::Acquire) as u64,
6611                         min_final_cltv_expiry_delta)
6612         }
6613
6614         /// Gets a [`PaymentSecret`] for a given [`PaymentHash`], for which the payment preimage is
6615         /// stored external to LDK.
6616         ///
6617         /// A [`PaymentClaimable`] event will only be generated if the [`PaymentSecret`] matches a
6618         /// payment secret fetched via this method or [`create_inbound_payment`], and which is at least
6619         /// the `min_value_msat` provided here, if one is provided.
6620         ///
6621         /// The [`PaymentHash`] (and corresponding [`PaymentPreimage`]) should be globally unique, though
6622         /// note that LDK will not stop you from registering duplicate payment hashes for inbound
6623         /// payments.
6624         ///
6625         /// `min_value_msat` should be set if the invoice being generated contains a value. Any payment
6626         /// received for the returned [`PaymentHash`] will be required to be at least `min_value_msat`
6627         /// before a [`PaymentClaimable`] event will be generated, ensuring that we do not provide the
6628         /// sender "proof-of-payment" unless they have paid the required amount.
6629         ///
6630         /// `invoice_expiry_delta_secs` describes the number of seconds that the invoice is valid for
6631         /// in excess of the current time. This should roughly match the expiry time set in the invoice.
6632         /// After this many seconds, we will remove the inbound payment, resulting in any attempts to
6633         /// pay the invoice failing. The BOLT spec suggests 3,600 secs as a default validity time for
6634         /// invoices when no timeout is set.
6635         ///
6636         /// Note that we use block header time to time-out pending inbound payments (with some margin
6637         /// to compensate for the inaccuracy of block header timestamps). Thus, in practice we will
6638         /// accept a payment and generate a [`PaymentClaimable`] event for some time after the expiry.
6639         /// If you need exact expiry semantics, you should enforce them upon receipt of
6640         /// [`PaymentClaimable`].
6641         ///
6642         /// Note that invoices generated for inbound payments should have their `min_final_cltv_expiry_delta`
6643         /// set to at least [`MIN_FINAL_CLTV_EXPIRY_DELTA`].
6644         ///
6645         /// Note that a malicious eavesdropper can intuit whether an inbound payment was created by
6646         /// `create_inbound_payment` or `create_inbound_payment_for_hash` based on runtime.
6647         ///
6648         /// # Note
6649         ///
6650         /// If you register an inbound payment with this method, then serialize the `ChannelManager`, then
6651         /// deserialize it with a node running 0.0.103 and earlier, the payment will fail to be received.
6652         ///
6653         /// Errors if `min_value_msat` is greater than total bitcoin supply.
6654         ///
6655         /// If `min_final_cltv_expiry_delta` is set to some value, then the payment will not be receivable
6656         /// on versions of LDK prior to 0.0.114.
6657         ///
6658         /// [`create_inbound_payment`]: Self::create_inbound_payment
6659         /// [`PaymentClaimable`]: events::Event::PaymentClaimable
6660         pub fn create_inbound_payment_for_hash(&self, payment_hash: PaymentHash, min_value_msat: Option<u64>,
6661                 invoice_expiry_delta_secs: u32, min_final_cltv_expiry: Option<u16>) -> Result<PaymentSecret, ()> {
6662                 inbound_payment::create_from_hash(&self.inbound_payment_key, min_value_msat, payment_hash,
6663                         invoice_expiry_delta_secs, self.highest_seen_timestamp.load(Ordering::Acquire) as u64,
6664                         min_final_cltv_expiry)
6665         }
6666
6667         /// Gets an LDK-generated payment preimage from a payment hash and payment secret that were
6668         /// previously returned from [`create_inbound_payment`].
6669         ///
6670         /// [`create_inbound_payment`]: Self::create_inbound_payment
6671         pub fn get_payment_preimage(&self, payment_hash: PaymentHash, payment_secret: PaymentSecret) -> Result<PaymentPreimage, APIError> {
6672                 inbound_payment::get_payment_preimage(payment_hash, payment_secret, &self.inbound_payment_key)
6673         }
6674
6675         /// Gets a fake short channel id for use in receiving [phantom node payments]. These fake scids
6676         /// are used when constructing the phantom invoice's route hints.
6677         ///
6678         /// [phantom node payments]: crate::sign::PhantomKeysManager
6679         pub fn get_phantom_scid(&self) -> u64 {
6680                 let best_block_height = self.best_block.read().unwrap().height();
6681                 let short_to_chan_info = self.short_to_chan_info.read().unwrap();
6682                 loop {
6683                         let scid_candidate = fake_scid::Namespace::Phantom.get_fake_scid(best_block_height, &self.genesis_hash, &self.fake_scid_rand_bytes, &self.entropy_source);
6684                         // Ensure the generated scid doesn't conflict with a real channel.
6685                         match short_to_chan_info.get(&scid_candidate) {
6686                                 Some(_) => continue,
6687                                 None => return scid_candidate
6688                         }
6689                 }
6690         }
6691
6692         /// Gets route hints for use in receiving [phantom node payments].
6693         ///
6694         /// [phantom node payments]: crate::sign::PhantomKeysManager
6695         pub fn get_phantom_route_hints(&self) -> PhantomRouteHints {
6696                 PhantomRouteHints {
6697                         channels: self.list_usable_channels(),
6698                         phantom_scid: self.get_phantom_scid(),
6699                         real_node_pubkey: self.get_our_node_id(),
6700                 }
6701         }
6702
6703         /// Gets a fake short channel id for use in receiving intercepted payments. These fake scids are
6704         /// used when constructing the route hints for HTLCs intended to be intercepted. See
6705         /// [`ChannelManager::forward_intercepted_htlc`].
6706         ///
6707         /// Note that this method is not guaranteed to return unique values, you may need to call it a few
6708         /// times to get a unique scid.
6709         pub fn get_intercept_scid(&self) -> u64 {
6710                 let best_block_height = self.best_block.read().unwrap().height();
6711                 let short_to_chan_info = self.short_to_chan_info.read().unwrap();
6712                 loop {
6713                         let scid_candidate = fake_scid::Namespace::Intercept.get_fake_scid(best_block_height, &self.genesis_hash, &self.fake_scid_rand_bytes, &self.entropy_source);
6714                         // Ensure the generated scid doesn't conflict with a real channel.
6715                         if short_to_chan_info.contains_key(&scid_candidate) { continue }
6716                         return scid_candidate
6717                 }
6718         }
6719
6720         /// Gets inflight HTLC information by processing pending outbound payments that are in
6721         /// our channels. May be used during pathfinding to account for in-use channel liquidity.
6722         pub fn compute_inflight_htlcs(&self) -> InFlightHtlcs {
6723                 let mut inflight_htlcs = InFlightHtlcs::new();
6724
6725                 let per_peer_state = self.per_peer_state.read().unwrap();
6726                 for (_cp_id, peer_state_mutex) in per_peer_state.iter() {
6727                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6728                         let peer_state = &mut *peer_state_lock;
6729                         for chan in peer_state.channel_by_id.values() {
6730                                 for (htlc_source, _) in chan.inflight_htlc_sources() {
6731                                         if let HTLCSource::OutboundRoute { path, .. } = htlc_source {
6732                                                 inflight_htlcs.process_path(path, self.get_our_node_id());
6733                                         }
6734                                 }
6735                         }
6736                 }
6737
6738                 inflight_htlcs
6739         }
6740
6741         #[cfg(any(test, feature = "_test_utils"))]
6742         pub fn get_and_clear_pending_events(&self) -> Vec<events::Event> {
6743                 let events = core::cell::RefCell::new(Vec::new());
6744                 let event_handler = |event: events::Event| events.borrow_mut().push(event);
6745                 self.process_pending_events(&event_handler);
6746                 events.into_inner()
6747         }
6748
6749         #[cfg(feature = "_test_utils")]
6750         pub fn push_pending_event(&self, event: events::Event) {
6751                 let mut events = self.pending_events.lock().unwrap();
6752                 events.push_back((event, None));
6753         }
6754
6755         #[cfg(test)]
6756         pub fn pop_pending_event(&self) -> Option<events::Event> {
6757                 let mut events = self.pending_events.lock().unwrap();
6758                 events.pop_front().map(|(e, _)| e)
6759         }
6760
6761         #[cfg(test)]
6762         pub fn has_pending_payments(&self) -> bool {
6763                 self.pending_outbound_payments.has_pending_payments()
6764         }
6765
6766         #[cfg(test)]
6767         pub fn clear_pending_payments(&self) {
6768                 self.pending_outbound_payments.clear_pending_payments()
6769         }
6770
6771         /// When something which was blocking a channel from updating its [`ChannelMonitor`] (e.g. an
6772         /// [`Event`] being handled) completes, this should be called to restore the channel to normal
6773         /// operation. It will double-check that nothing *else* is also blocking the same channel from
6774         /// making progress and then let any blocked [`ChannelMonitorUpdate`]s fly.
6775         fn handle_monitor_update_release(&self, counterparty_node_id: PublicKey, channel_funding_outpoint: OutPoint, mut completed_blocker: Option<RAAMonitorUpdateBlockingAction>) {
6776                 let mut errors = Vec::new();
6777                 loop {
6778                         let per_peer_state = self.per_peer_state.read().unwrap();
6779                         if let Some(peer_state_mtx) = per_peer_state.get(&counterparty_node_id) {
6780                                 let mut peer_state_lck = peer_state_mtx.lock().unwrap();
6781                                 let peer_state = &mut *peer_state_lck;
6782
6783                                 if let Some(blocker) = completed_blocker.take() {
6784                                         // Only do this on the first iteration of the loop.
6785                                         if let Some(blockers) = peer_state.actions_blocking_raa_monitor_updates
6786                                                 .get_mut(&channel_funding_outpoint.to_channel_id())
6787                                         {
6788                                                 blockers.retain(|iter| iter != &blocker);
6789                                         }
6790                                 }
6791
6792                                 if self.raa_monitor_updates_held(&peer_state.actions_blocking_raa_monitor_updates,
6793                                         channel_funding_outpoint, counterparty_node_id) {
6794                                         // Check that, while holding the peer lock, we don't have anything else
6795                                         // blocking monitor updates for this channel. If we do, release the monitor
6796                                         // update(s) when those blockers complete.
6797                                         log_trace!(self.logger, "Delaying monitor unlock for channel {} as another channel's mon update needs to complete first",
6798                                                 &channel_funding_outpoint.to_channel_id());
6799                                         break;
6800                                 }
6801
6802                                 if let hash_map::Entry::Occupied(mut chan) = peer_state.channel_by_id.entry(channel_funding_outpoint.to_channel_id()) {
6803                                         debug_assert_eq!(chan.get().context.get_funding_txo().unwrap(), channel_funding_outpoint);
6804                                         if let Some((monitor_update, further_update_exists)) = chan.get_mut().unblock_next_blocked_monitor_update() {
6805                                                 log_debug!(self.logger, "Unlocking monitor updating for channel {} and updating monitor",
6806                                                         &channel_funding_outpoint.to_channel_id());
6807                                                 if let Err(e) = handle_new_monitor_update!(self, channel_funding_outpoint, monitor_update,
6808                                                         peer_state_lck, peer_state, per_peer_state, chan)
6809                                                 {
6810                                                         errors.push((e, counterparty_node_id));
6811                                                 }
6812                                                 if further_update_exists {
6813                                                         // If there are more `ChannelMonitorUpdate`s to process, restart at the
6814                                                         // top of the loop.
6815                                                         continue;
6816                                                 }
6817                                         } else {
6818                                                 log_trace!(self.logger, "Unlocked monitor updating for channel {} without monitors to update",
6819                                                         &channel_funding_outpoint.to_channel_id());
6820                                         }
6821                                 }
6822                         } else {
6823                                 log_debug!(self.logger,
6824                                         "Got a release post-RAA monitor update for peer {} but the channel is gone",
6825                                         log_pubkey!(counterparty_node_id));
6826                         }
6827                         break;
6828                 }
6829                 for (err, counterparty_node_id) in errors {
6830                         let res = Err::<(), _>(err);
6831                         let _ = handle_error!(self, res, counterparty_node_id);
6832                 }
6833         }
6834
6835         fn handle_post_event_actions(&self, actions: Vec<EventCompletionAction>) {
6836                 for action in actions {
6837                         match action {
6838                                 EventCompletionAction::ReleaseRAAChannelMonitorUpdate {
6839                                         channel_funding_outpoint, counterparty_node_id
6840                                 } => {
6841                                         self.handle_monitor_update_release(counterparty_node_id, channel_funding_outpoint, None);
6842                                 }
6843                         }
6844                 }
6845         }
6846
6847         /// Processes any events asynchronously in the order they were generated since the last call
6848         /// using the given event handler.
6849         ///
6850         /// See the trait-level documentation of [`EventsProvider`] for requirements.
6851         pub async fn process_pending_events_async<Future: core::future::Future, H: Fn(Event) -> Future>(
6852                 &self, handler: H
6853         ) {
6854                 let mut ev;
6855                 process_events_body!(self, ev, { handler(ev).await });
6856         }
6857 }
6858
6859 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref> MessageSendEventsProvider for ChannelManager<M, T, ES, NS, SP, F, R, L>
6860 where
6861         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
6862         T::Target: BroadcasterInterface,
6863         ES::Target: EntropySource,
6864         NS::Target: NodeSigner,
6865         SP::Target: SignerProvider,
6866         F::Target: FeeEstimator,
6867         R::Target: Router,
6868         L::Target: Logger,
6869 {
6870         /// Returns `MessageSendEvent`s strictly ordered per-peer, in the order they were generated.
6871         /// The returned array will contain `MessageSendEvent`s for different peers if
6872         /// `MessageSendEvent`s to more than one peer exists, but `MessageSendEvent`s to the same peer
6873         /// is always placed next to each other.
6874         ///
6875         /// Note that that while `MessageSendEvent`s are strictly ordered per-peer, the peer order for
6876         /// the chunks of `MessageSendEvent`s for different peers is random. I.e. if the array contains
6877         /// `MessageSendEvent`s  for both `node_a` and `node_b`, the `MessageSendEvent`s for `node_a`
6878         /// will randomly be placed first or last in the returned array.
6879         ///
6880         /// Note that even though `BroadcastChannelAnnouncement` and `BroadcastChannelUpdate`
6881         /// `MessageSendEvent`s are intended to be broadcasted to all peers, they will be pleaced among
6882         /// the `MessageSendEvent`s to the specific peer they were generated under.
6883         fn get_and_clear_pending_msg_events(&self) -> Vec<MessageSendEvent> {
6884                 let events = RefCell::new(Vec::new());
6885                 PersistenceNotifierGuard::optionally_notify(&self.total_consistency_lock, &self.persistence_notifier, || {
6886                         let mut result = self.process_background_events();
6887
6888                         // TODO: This behavior should be documented. It's unintuitive that we query
6889                         // ChannelMonitors when clearing other events.
6890                         if self.process_pending_monitor_events() {
6891                                 result = NotifyOption::DoPersist;
6892                         }
6893
6894                         if self.check_free_holding_cells() {
6895                                 result = NotifyOption::DoPersist;
6896                         }
6897                         if self.maybe_generate_initial_closing_signed() {
6898                                 result = NotifyOption::DoPersist;
6899                         }
6900
6901                         let mut pending_events = Vec::new();
6902                         let per_peer_state = self.per_peer_state.read().unwrap();
6903                         for (_cp_id, peer_state_mutex) in per_peer_state.iter() {
6904                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
6905                                 let peer_state = &mut *peer_state_lock;
6906                                 if peer_state.pending_msg_events.len() > 0 {
6907                                         pending_events.append(&mut peer_state.pending_msg_events);
6908                                 }
6909                         }
6910
6911                         if !pending_events.is_empty() {
6912                                 events.replace(pending_events);
6913                         }
6914
6915                         result
6916                 });
6917                 events.into_inner()
6918         }
6919 }
6920
6921 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref> EventsProvider for ChannelManager<M, T, ES, NS, SP, F, R, L>
6922 where
6923         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
6924         T::Target: BroadcasterInterface,
6925         ES::Target: EntropySource,
6926         NS::Target: NodeSigner,
6927         SP::Target: SignerProvider,
6928         F::Target: FeeEstimator,
6929         R::Target: Router,
6930         L::Target: Logger,
6931 {
6932         /// Processes events that must be periodically handled.
6933         ///
6934         /// An [`EventHandler`] may safely call back to the provider in order to handle an event.
6935         /// However, it must not call [`Writeable::write`] as doing so would result in a deadlock.
6936         fn process_pending_events<H: Deref>(&self, handler: H) where H::Target: EventHandler {
6937                 let mut ev;
6938                 process_events_body!(self, ev, handler.handle_event(ev));
6939         }
6940 }
6941
6942 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref> chain::Listen for ChannelManager<M, T, ES, NS, SP, F, R, L>
6943 where
6944         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
6945         T::Target: BroadcasterInterface,
6946         ES::Target: EntropySource,
6947         NS::Target: NodeSigner,
6948         SP::Target: SignerProvider,
6949         F::Target: FeeEstimator,
6950         R::Target: Router,
6951         L::Target: Logger,
6952 {
6953         fn filtered_block_connected(&self, header: &BlockHeader, txdata: &TransactionData, height: u32) {
6954                 {
6955                         let best_block = self.best_block.read().unwrap();
6956                         assert_eq!(best_block.block_hash(), header.prev_blockhash,
6957                                 "Blocks must be connected in chain-order - the connected header must build on the last connected header");
6958                         assert_eq!(best_block.height(), height - 1,
6959                                 "Blocks must be connected in chain-order - the connected block height must be one greater than the previous height");
6960                 }
6961
6962                 self.transactions_confirmed(header, txdata, height);
6963                 self.best_block_updated(header, height);
6964         }
6965
6966         fn block_disconnected(&self, header: &BlockHeader, height: u32) {
6967                 let _persistence_guard = PersistenceNotifierGuard::optionally_notify(&self.total_consistency_lock,
6968                         &self.persistence_notifier, || -> NotifyOption { NotifyOption::DoPersist });
6969                 let new_height = height - 1;
6970                 {
6971                         let mut best_block = self.best_block.write().unwrap();
6972                         assert_eq!(best_block.block_hash(), header.block_hash(),
6973                                 "Blocks must be disconnected in chain-order - the disconnected header must be the last connected header");
6974                         assert_eq!(best_block.height(), height,
6975                                 "Blocks must be disconnected in chain-order - the disconnected block must have the correct height");
6976                         *best_block = BestBlock::new(header.prev_blockhash, new_height)
6977                 }
6978
6979                 self.do_chain_event(Some(new_height), |channel| channel.best_block_updated(new_height, header.time, self.genesis_hash.clone(), &self.node_signer, &self.default_configuration, &self.logger));
6980         }
6981 }
6982
6983 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref> chain::Confirm for ChannelManager<M, T, ES, NS, SP, F, R, L>
6984 where
6985         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
6986         T::Target: BroadcasterInterface,
6987         ES::Target: EntropySource,
6988         NS::Target: NodeSigner,
6989         SP::Target: SignerProvider,
6990         F::Target: FeeEstimator,
6991         R::Target: Router,
6992         L::Target: Logger,
6993 {
6994         fn transactions_confirmed(&self, header: &BlockHeader, txdata: &TransactionData, height: u32) {
6995                 // Note that we MUST NOT end up calling methods on self.chain_monitor here - we're called
6996                 // during initialization prior to the chain_monitor being fully configured in some cases.
6997                 // See the docs for `ChannelManagerReadArgs` for more.
6998
6999                 let block_hash = header.block_hash();
7000                 log_trace!(self.logger, "{} transactions included in block {} at height {} provided", txdata.len(), block_hash, height);
7001
7002                 let _persistence_guard = PersistenceNotifierGuard::optionally_notify(&self.total_consistency_lock,
7003                         &self.persistence_notifier, || -> NotifyOption { NotifyOption::DoPersist });
7004                 self.do_chain_event(Some(height), |channel| channel.transactions_confirmed(&block_hash, height, txdata, self.genesis_hash.clone(), &self.node_signer, &self.default_configuration, &self.logger)
7005                         .map(|(a, b)| (a, Vec::new(), b)));
7006
7007                 let last_best_block_height = self.best_block.read().unwrap().height();
7008                 if height < last_best_block_height {
7009                         let timestamp = self.highest_seen_timestamp.load(Ordering::Acquire);
7010                         self.do_chain_event(Some(last_best_block_height), |channel| channel.best_block_updated(last_best_block_height, timestamp as u32, self.genesis_hash.clone(), &self.node_signer, &self.default_configuration, &self.logger));
7011                 }
7012         }
7013
7014         fn best_block_updated(&self, header: &BlockHeader, height: u32) {
7015                 // Note that we MUST NOT end up calling methods on self.chain_monitor here - we're called
7016                 // during initialization prior to the chain_monitor being fully configured in some cases.
7017                 // See the docs for `ChannelManagerReadArgs` for more.
7018
7019                 let block_hash = header.block_hash();
7020                 log_trace!(self.logger, "New best block: {} at height {}", block_hash, height);
7021
7022                 let _persistence_guard = PersistenceNotifierGuard::optionally_notify(&self.total_consistency_lock,
7023                         &self.persistence_notifier, || -> NotifyOption { NotifyOption::DoPersist });
7024                 *self.best_block.write().unwrap() = BestBlock::new(block_hash, height);
7025
7026                 self.do_chain_event(Some(height), |channel| channel.best_block_updated(height, header.time, self.genesis_hash.clone(), &self.node_signer, &self.default_configuration, &self.logger));
7027
7028                 macro_rules! max_time {
7029                         ($timestamp: expr) => {
7030                                 loop {
7031                                         // Update $timestamp to be the max of its current value and the block
7032                                         // timestamp. This should keep us close to the current time without relying on
7033                                         // having an explicit local time source.
7034                                         // Just in case we end up in a race, we loop until we either successfully
7035                                         // update $timestamp or decide we don't need to.
7036                                         let old_serial = $timestamp.load(Ordering::Acquire);
7037                                         if old_serial >= header.time as usize { break; }
7038                                         if $timestamp.compare_exchange(old_serial, header.time as usize, Ordering::AcqRel, Ordering::Relaxed).is_ok() {
7039                                                 break;
7040                                         }
7041                                 }
7042                         }
7043                 }
7044                 max_time!(self.highest_seen_timestamp);
7045                 let mut payment_secrets = self.pending_inbound_payments.lock().unwrap();
7046                 payment_secrets.retain(|_, inbound_payment| {
7047                         inbound_payment.expiry_time > header.time as u64
7048                 });
7049         }
7050
7051         fn get_relevant_txids(&self) -> Vec<(Txid, Option<BlockHash>)> {
7052                 let mut res = Vec::with_capacity(self.short_to_chan_info.read().unwrap().len());
7053                 for (_cp_id, peer_state_mutex) in self.per_peer_state.read().unwrap().iter() {
7054                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
7055                         let peer_state = &mut *peer_state_lock;
7056                         for chan in peer_state.channel_by_id.values() {
7057                                 if let (Some(funding_txo), Some(block_hash)) = (chan.context.get_funding_txo(), chan.context.get_funding_tx_confirmed_in()) {
7058                                         res.push((funding_txo.txid, Some(block_hash)));
7059                                 }
7060                         }
7061                 }
7062                 res
7063         }
7064
7065         fn transaction_unconfirmed(&self, txid: &Txid) {
7066                 let _persistence_guard = PersistenceNotifierGuard::optionally_notify(&self.total_consistency_lock,
7067                         &self.persistence_notifier, || -> NotifyOption { NotifyOption::DoPersist });
7068                 self.do_chain_event(None, |channel| {
7069                         if let Some(funding_txo) = channel.context.get_funding_txo() {
7070                                 if funding_txo.txid == *txid {
7071                                         channel.funding_transaction_unconfirmed(&self.logger).map(|()| (None, Vec::new(), None))
7072                                 } else { Ok((None, Vec::new(), None)) }
7073                         } else { Ok((None, Vec::new(), None)) }
7074                 });
7075         }
7076 }
7077
7078 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref> ChannelManager<M, T, ES, NS, SP, F, R, L>
7079 where
7080         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
7081         T::Target: BroadcasterInterface,
7082         ES::Target: EntropySource,
7083         NS::Target: NodeSigner,
7084         SP::Target: SignerProvider,
7085         F::Target: FeeEstimator,
7086         R::Target: Router,
7087         L::Target: Logger,
7088 {
7089         /// Calls a function which handles an on-chain event (blocks dis/connected, transactions
7090         /// un/confirmed, etc) on each channel, handling any resulting errors or messages generated by
7091         /// the function.
7092         fn do_chain_event<FN: Fn(&mut Channel<SP>) -> Result<(Option<msgs::ChannelReady>, Vec<(HTLCSource, PaymentHash)>, Option<msgs::AnnouncementSignatures>), ClosureReason>>
7093                         (&self, height_opt: Option<u32>, f: FN) {
7094                 // Note that we MUST NOT end up calling methods on self.chain_monitor here - we're called
7095                 // during initialization prior to the chain_monitor being fully configured in some cases.
7096                 // See the docs for `ChannelManagerReadArgs` for more.
7097
7098                 let mut failed_channels = Vec::new();
7099                 let mut timed_out_htlcs = Vec::new();
7100                 {
7101                         let per_peer_state = self.per_peer_state.read().unwrap();
7102                         for (_cp_id, peer_state_mutex) in per_peer_state.iter() {
7103                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
7104                                 let peer_state = &mut *peer_state_lock;
7105                                 let pending_msg_events = &mut peer_state.pending_msg_events;
7106                                 peer_state.channel_by_id.retain(|_, channel| {
7107                                         let res = f(channel);
7108                                         if let Ok((channel_ready_opt, mut timed_out_pending_htlcs, announcement_sigs)) = res {
7109                                                 for (source, payment_hash) in timed_out_pending_htlcs.drain(..) {
7110                                                         let (failure_code, data) = self.get_htlc_inbound_temp_fail_err_and_data(0x1000|14 /* expiry_too_soon */, &channel);
7111                                                         timed_out_htlcs.push((source, payment_hash, HTLCFailReason::reason(failure_code, data),
7112                                                                 HTLCDestination::NextHopChannel { node_id: Some(channel.context.get_counterparty_node_id()), channel_id: channel.context.channel_id() }));
7113                                                 }
7114                                                 if let Some(channel_ready) = channel_ready_opt {
7115                                                         send_channel_ready!(self, pending_msg_events, channel, channel_ready);
7116                                                         if channel.context.is_usable() {
7117                                                                 log_trace!(self.logger, "Sending channel_ready with private initial channel_update for our counterparty on channel {}", &channel.context.channel_id());
7118                                                                 if let Ok(msg) = self.get_channel_update_for_unicast(channel) {
7119                                                                         pending_msg_events.push(events::MessageSendEvent::SendChannelUpdate {
7120                                                                                 node_id: channel.context.get_counterparty_node_id(),
7121                                                                                 msg,
7122                                                                         });
7123                                                                 }
7124                                                         } else {
7125                                                                 log_trace!(self.logger, "Sending channel_ready WITHOUT channel_update for {}", &channel.context.channel_id());
7126                                                         }
7127                                                 }
7128
7129                                                 {
7130                                                         let mut pending_events = self.pending_events.lock().unwrap();
7131                                                         emit_channel_ready_event!(pending_events, channel);
7132                                                 }
7133
7134                                                 if let Some(announcement_sigs) = announcement_sigs {
7135                                                         log_trace!(self.logger, "Sending announcement_signatures for channel {}", &channel.context.channel_id());
7136                                                         pending_msg_events.push(events::MessageSendEvent::SendAnnouncementSignatures {
7137                                                                 node_id: channel.context.get_counterparty_node_id(),
7138                                                                 msg: announcement_sigs,
7139                                                         });
7140                                                         if let Some(height) = height_opt {
7141                                                                 if let Some(announcement) = channel.get_signed_channel_announcement(&self.node_signer, self.genesis_hash, height, &self.default_configuration) {
7142                                                                         pending_msg_events.push(events::MessageSendEvent::BroadcastChannelAnnouncement {
7143                                                                                 msg: announcement,
7144                                                                                 // Note that announcement_signatures fails if the channel cannot be announced,
7145                                                                                 // so get_channel_update_for_broadcast will never fail by the time we get here.
7146                                                                                 update_msg: Some(self.get_channel_update_for_broadcast(channel).unwrap()),
7147                                                                         });
7148                                                                 }
7149                                                         }
7150                                                 }
7151                                                 if channel.is_our_channel_ready() {
7152                                                         if let Some(real_scid) = channel.context.get_short_channel_id() {
7153                                                                 // If we sent a 0conf channel_ready, and now have an SCID, we add it
7154                                                                 // to the short_to_chan_info map here. Note that we check whether we
7155                                                                 // can relay using the real SCID at relay-time (i.e.
7156                                                                 // enforce option_scid_alias then), and if the funding tx is ever
7157                                                                 // un-confirmed we force-close the channel, ensuring short_to_chan_info
7158                                                                 // is always consistent.
7159                                                                 let mut short_to_chan_info = self.short_to_chan_info.write().unwrap();
7160                                                                 let scid_insert = short_to_chan_info.insert(real_scid, (channel.context.get_counterparty_node_id(), channel.context.channel_id()));
7161                                                                 assert!(scid_insert.is_none() || scid_insert.unwrap() == (channel.context.get_counterparty_node_id(), channel.context.channel_id()),
7162                                                                         "SCIDs should never collide - ensure you weren't behind by a full {} blocks when creating channels",
7163                                                                         fake_scid::MAX_SCID_BLOCKS_FROM_NOW);
7164                                                         }
7165                                                 }
7166                                         } else if let Err(reason) = res {
7167                                                 update_maps_on_chan_removal!(self, &channel.context);
7168                                                 // It looks like our counterparty went on-chain or funding transaction was
7169                                                 // reorged out of the main chain. Close the channel.
7170                                                 failed_channels.push(channel.context.force_shutdown(true));
7171                                                 if let Ok(update) = self.get_channel_update_for_broadcast(&channel) {
7172                                                         pending_msg_events.push(events::MessageSendEvent::BroadcastChannelUpdate {
7173                                                                 msg: update
7174                                                         });
7175                                                 }
7176                                                 let reason_message = format!("{}", reason);
7177                                                 self.issue_channel_close_events(&channel.context, reason);
7178                                                 pending_msg_events.push(events::MessageSendEvent::HandleError {
7179                                                         node_id: channel.context.get_counterparty_node_id(),
7180                                                         action: msgs::ErrorAction::SendErrorMessage { msg: msgs::ErrorMessage {
7181                                                                 channel_id: channel.context.channel_id(),
7182                                                                 data: reason_message,
7183                                                         } },
7184                                                 });
7185                                                 return false;
7186                                         }
7187                                         true
7188                                 });
7189                         }
7190                 }
7191
7192                 if let Some(height) = height_opt {
7193                         self.claimable_payments.lock().unwrap().claimable_payments.retain(|payment_hash, payment| {
7194                                 payment.htlcs.retain(|htlc| {
7195                                         // If height is approaching the number of blocks we think it takes us to get
7196                                         // our commitment transaction confirmed before the HTLC expires, plus the
7197                                         // number of blocks we generally consider it to take to do a commitment update,
7198                                         // just give up on it and fail the HTLC.
7199                                         if height >= htlc.cltv_expiry - HTLC_FAIL_BACK_BUFFER {
7200                                                 let mut htlc_msat_height_data = htlc.value.to_be_bytes().to_vec();
7201                                                 htlc_msat_height_data.extend_from_slice(&height.to_be_bytes());
7202
7203                                                 timed_out_htlcs.push((HTLCSource::PreviousHopData(htlc.prev_hop.clone()), payment_hash.clone(),
7204                                                         HTLCFailReason::reason(0x4000 | 15, htlc_msat_height_data),
7205                                                         HTLCDestination::FailedPayment { payment_hash: payment_hash.clone() }));
7206                                                 false
7207                                         } else { true }
7208                                 });
7209                                 !payment.htlcs.is_empty() // Only retain this entry if htlcs has at least one entry.
7210                         });
7211
7212                         let mut intercepted_htlcs = self.pending_intercepted_htlcs.lock().unwrap();
7213                         intercepted_htlcs.retain(|_, htlc| {
7214                                 if height >= htlc.forward_info.outgoing_cltv_value - HTLC_FAIL_BACK_BUFFER {
7215                                         let prev_hop_data = HTLCSource::PreviousHopData(HTLCPreviousHopData {
7216                                                 short_channel_id: htlc.prev_short_channel_id,
7217                                                 user_channel_id: Some(htlc.prev_user_channel_id),
7218                                                 htlc_id: htlc.prev_htlc_id,
7219                                                 incoming_packet_shared_secret: htlc.forward_info.incoming_shared_secret,
7220                                                 phantom_shared_secret: None,
7221                                                 outpoint: htlc.prev_funding_outpoint,
7222                                         });
7223
7224                                         let requested_forward_scid /* intercept scid */ = match htlc.forward_info.routing {
7225                                                 PendingHTLCRouting::Forward { short_channel_id, .. } => short_channel_id,
7226                                                 _ => unreachable!(),
7227                                         };
7228                                         timed_out_htlcs.push((prev_hop_data, htlc.forward_info.payment_hash,
7229                                                         HTLCFailReason::from_failure_code(0x2000 | 2),
7230                                                         HTLCDestination::InvalidForward { requested_forward_scid }));
7231                                         log_trace!(self.logger, "Timing out intercepted HTLC with requested forward scid {}", requested_forward_scid);
7232                                         false
7233                                 } else { true }
7234                         });
7235                 }
7236
7237                 self.handle_init_event_channel_failures(failed_channels);
7238
7239                 for (source, payment_hash, reason, destination) in timed_out_htlcs.drain(..) {
7240                         self.fail_htlc_backwards_internal(&source, &payment_hash, &reason, destination);
7241                 }
7242         }
7243
7244         /// Gets a [`Future`] that completes when this [`ChannelManager`] needs to be persisted.
7245         ///
7246         /// Note that callbacks registered on the [`Future`] MUST NOT call back into this
7247         /// [`ChannelManager`] and should instead register actions to be taken later.
7248         ///
7249         pub fn get_persistable_update_future(&self) -> Future {
7250                 self.persistence_notifier.get_future()
7251         }
7252
7253         #[cfg(any(test, feature = "_test_utils"))]
7254         pub fn get_persistence_condvar_value(&self) -> bool {
7255                 self.persistence_notifier.notify_pending()
7256         }
7257
7258         /// Gets the latest best block which was connected either via the [`chain::Listen`] or
7259         /// [`chain::Confirm`] interfaces.
7260         pub fn current_best_block(&self) -> BestBlock {
7261                 self.best_block.read().unwrap().clone()
7262         }
7263
7264         /// Fetches the set of [`NodeFeatures`] flags which are provided by or required by
7265         /// [`ChannelManager`].
7266         pub fn node_features(&self) -> NodeFeatures {
7267                 provided_node_features(&self.default_configuration)
7268         }
7269
7270         /// Fetches the set of [`Bolt11InvoiceFeatures`] flags which are provided by or required by
7271         /// [`ChannelManager`].
7272         ///
7273         /// Note that the invoice feature flags can vary depending on if the invoice is a "phantom invoice"
7274         /// or not. Thus, this method is not public.
7275         #[cfg(any(feature = "_test_utils", test))]
7276         pub fn invoice_features(&self) -> Bolt11InvoiceFeatures {
7277                 provided_invoice_features(&self.default_configuration)
7278         }
7279
7280         /// Fetches the set of [`ChannelFeatures`] flags which are provided by or required by
7281         /// [`ChannelManager`].
7282         pub fn channel_features(&self) -> ChannelFeatures {
7283                 provided_channel_features(&self.default_configuration)
7284         }
7285
7286         /// Fetches the set of [`ChannelTypeFeatures`] flags which are provided by or required by
7287         /// [`ChannelManager`].
7288         pub fn channel_type_features(&self) -> ChannelTypeFeatures {
7289                 provided_channel_type_features(&self.default_configuration)
7290         }
7291
7292         /// Fetches the set of [`InitFeatures`] flags which are provided by or required by
7293         /// [`ChannelManager`].
7294         pub fn init_features(&self) -> InitFeatures {
7295                 provided_init_features(&self.default_configuration)
7296         }
7297 }
7298
7299 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref>
7300         ChannelMessageHandler for ChannelManager<M, T, ES, NS, SP, F, R, L>
7301 where
7302         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
7303         T::Target: BroadcasterInterface,
7304         ES::Target: EntropySource,
7305         NS::Target: NodeSigner,
7306         SP::Target: SignerProvider,
7307         F::Target: FeeEstimator,
7308         R::Target: Router,
7309         L::Target: Logger,
7310 {
7311         fn handle_open_channel(&self, counterparty_node_id: &PublicKey, msg: &msgs::OpenChannel) {
7312                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7313                 let _ = handle_error!(self, self.internal_open_channel(counterparty_node_id, msg), *counterparty_node_id);
7314         }
7315
7316         fn handle_open_channel_v2(&self, counterparty_node_id: &PublicKey, msg: &msgs::OpenChannelV2) {
7317                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7318                         "Dual-funded channels not supported".to_owned(),
7319                          msg.temporary_channel_id.clone())), *counterparty_node_id);
7320         }
7321
7322         fn handle_accept_channel(&self, counterparty_node_id: &PublicKey, msg: &msgs::AcceptChannel) {
7323                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7324                 let _ = handle_error!(self, self.internal_accept_channel(counterparty_node_id, msg), *counterparty_node_id);
7325         }
7326
7327         fn handle_accept_channel_v2(&self, counterparty_node_id: &PublicKey, msg: &msgs::AcceptChannelV2) {
7328                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7329                         "Dual-funded channels not supported".to_owned(),
7330                          msg.temporary_channel_id.clone())), *counterparty_node_id);
7331         }
7332
7333         fn handle_funding_created(&self, counterparty_node_id: &PublicKey, msg: &msgs::FundingCreated) {
7334                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7335                 let _ = handle_error!(self, self.internal_funding_created(counterparty_node_id, msg), *counterparty_node_id);
7336         }
7337
7338         fn handle_funding_signed(&self, counterparty_node_id: &PublicKey, msg: &msgs::FundingSigned) {
7339                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7340                 let _ = handle_error!(self, self.internal_funding_signed(counterparty_node_id, msg), *counterparty_node_id);
7341         }
7342
7343         fn handle_channel_ready(&self, counterparty_node_id: &PublicKey, msg: &msgs::ChannelReady) {
7344                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7345                 let _ = handle_error!(self, self.internal_channel_ready(counterparty_node_id, msg), *counterparty_node_id);
7346         }
7347
7348         fn handle_shutdown(&self, counterparty_node_id: &PublicKey, msg: &msgs::Shutdown) {
7349                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7350                 let _ = handle_error!(self, self.internal_shutdown(counterparty_node_id, msg), *counterparty_node_id);
7351         }
7352
7353         fn handle_closing_signed(&self, counterparty_node_id: &PublicKey, msg: &msgs::ClosingSigned) {
7354                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7355                 let _ = handle_error!(self, self.internal_closing_signed(counterparty_node_id, msg), *counterparty_node_id);
7356         }
7357
7358         fn handle_update_add_htlc(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateAddHTLC) {
7359                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7360                 let _ = handle_error!(self, self.internal_update_add_htlc(counterparty_node_id, msg), *counterparty_node_id);
7361         }
7362
7363         fn handle_update_fulfill_htlc(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateFulfillHTLC) {
7364                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7365                 let _ = handle_error!(self, self.internal_update_fulfill_htlc(counterparty_node_id, msg), *counterparty_node_id);
7366         }
7367
7368         fn handle_update_fail_htlc(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateFailHTLC) {
7369                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7370                 let _ = handle_error!(self, self.internal_update_fail_htlc(counterparty_node_id, msg), *counterparty_node_id);
7371         }
7372
7373         fn handle_update_fail_malformed_htlc(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateFailMalformedHTLC) {
7374                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7375                 let _ = handle_error!(self, self.internal_update_fail_malformed_htlc(counterparty_node_id, msg), *counterparty_node_id);
7376         }
7377
7378         fn handle_commitment_signed(&self, counterparty_node_id: &PublicKey, msg: &msgs::CommitmentSigned) {
7379                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7380                 let _ = handle_error!(self, self.internal_commitment_signed(counterparty_node_id, msg), *counterparty_node_id);
7381         }
7382
7383         fn handle_revoke_and_ack(&self, counterparty_node_id: &PublicKey, msg: &msgs::RevokeAndACK) {
7384                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7385                 let _ = handle_error!(self, self.internal_revoke_and_ack(counterparty_node_id, msg), *counterparty_node_id);
7386         }
7387
7388         fn handle_update_fee(&self, counterparty_node_id: &PublicKey, msg: &msgs::UpdateFee) {
7389                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7390                 let _ = handle_error!(self, self.internal_update_fee(counterparty_node_id, msg), *counterparty_node_id);
7391         }
7392
7393         fn handle_announcement_signatures(&self, counterparty_node_id: &PublicKey, msg: &msgs::AnnouncementSignatures) {
7394                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7395                 let _ = handle_error!(self, self.internal_announcement_signatures(counterparty_node_id, msg), *counterparty_node_id);
7396         }
7397
7398         fn handle_channel_update(&self, counterparty_node_id: &PublicKey, msg: &msgs::ChannelUpdate) {
7399                 PersistenceNotifierGuard::optionally_notify(&self.total_consistency_lock, &self.persistence_notifier, || {
7400                         let force_persist = self.process_background_events();
7401                         if let Ok(persist) = handle_error!(self, self.internal_channel_update(counterparty_node_id, msg), *counterparty_node_id) {
7402                                 if force_persist == NotifyOption::DoPersist { NotifyOption::DoPersist } else { persist }
7403                         } else {
7404                                 NotifyOption::SkipPersist
7405                         }
7406                 });
7407         }
7408
7409         fn handle_channel_reestablish(&self, counterparty_node_id: &PublicKey, msg: &msgs::ChannelReestablish) {
7410                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7411                 let _ = handle_error!(self, self.internal_channel_reestablish(counterparty_node_id, msg), *counterparty_node_id);
7412         }
7413
7414         fn peer_disconnected(&self, counterparty_node_id: &PublicKey) {
7415                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7416                 let mut failed_channels = Vec::new();
7417                 let mut per_peer_state = self.per_peer_state.write().unwrap();
7418                 let remove_peer = {
7419                         log_debug!(self.logger, "Marking channels with {} disconnected and generating channel_updates.",
7420                                 log_pubkey!(counterparty_node_id));
7421                         if let Some(peer_state_mutex) = per_peer_state.get(counterparty_node_id) {
7422                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
7423                                 let peer_state = &mut *peer_state_lock;
7424                                 let pending_msg_events = &mut peer_state.pending_msg_events;
7425                                 peer_state.channel_by_id.retain(|_, chan| {
7426                                         chan.remove_uncommitted_htlcs_and_mark_paused(&self.logger);
7427                                         if chan.is_shutdown() {
7428                                                 update_maps_on_chan_removal!(self, &chan.context);
7429                                                 self.issue_channel_close_events(&chan.context, ClosureReason::DisconnectedPeer);
7430                                                 return false;
7431                                         }
7432                                         true
7433                                 });
7434                                 peer_state.inbound_v1_channel_by_id.retain(|_, chan| {
7435                                         update_maps_on_chan_removal!(self, &chan.context);
7436                                         self.issue_channel_close_events(&chan.context, ClosureReason::DisconnectedPeer);
7437                                         false
7438                                 });
7439                                 peer_state.outbound_v1_channel_by_id.retain(|_, chan| {
7440                                         update_maps_on_chan_removal!(self, &chan.context);
7441                                         self.issue_channel_close_events(&chan.context, ClosureReason::DisconnectedPeer);
7442                                         false
7443                                 });
7444                                 // Note that we don't bother generating any events for pre-accept channels -
7445                                 // they're not considered "channels" yet from the PoV of our events interface.
7446                                 peer_state.inbound_channel_request_by_id.clear();
7447                                 pending_msg_events.retain(|msg| {
7448                                         match msg {
7449                                                 // V1 Channel Establishment
7450                                                 &events::MessageSendEvent::SendAcceptChannel { .. } => false,
7451                                                 &events::MessageSendEvent::SendOpenChannel { .. } => false,
7452                                                 &events::MessageSendEvent::SendFundingCreated { .. } => false,
7453                                                 &events::MessageSendEvent::SendFundingSigned { .. } => false,
7454                                                 // V2 Channel Establishment
7455                                                 &events::MessageSendEvent::SendAcceptChannelV2 { .. } => false,
7456                                                 &events::MessageSendEvent::SendOpenChannelV2 { .. } => false,
7457                                                 // Common Channel Establishment
7458                                                 &events::MessageSendEvent::SendChannelReady { .. } => false,
7459                                                 &events::MessageSendEvent::SendAnnouncementSignatures { .. } => false,
7460                                                 // Interactive Transaction Construction
7461                                                 &events::MessageSendEvent::SendTxAddInput { .. } => false,
7462                                                 &events::MessageSendEvent::SendTxAddOutput { .. } => false,
7463                                                 &events::MessageSendEvent::SendTxRemoveInput { .. } => false,
7464                                                 &events::MessageSendEvent::SendTxRemoveOutput { .. } => false,
7465                                                 &events::MessageSendEvent::SendTxComplete { .. } => false,
7466                                                 &events::MessageSendEvent::SendTxSignatures { .. } => false,
7467                                                 &events::MessageSendEvent::SendTxInitRbf { .. } => false,
7468                                                 &events::MessageSendEvent::SendTxAckRbf { .. } => false,
7469                                                 &events::MessageSendEvent::SendTxAbort { .. } => false,
7470                                                 // Channel Operations
7471                                                 &events::MessageSendEvent::UpdateHTLCs { .. } => false,
7472                                                 &events::MessageSendEvent::SendRevokeAndACK { .. } => false,
7473                                                 &events::MessageSendEvent::SendClosingSigned { .. } => false,
7474                                                 &events::MessageSendEvent::SendShutdown { .. } => false,
7475                                                 &events::MessageSendEvent::SendChannelReestablish { .. } => false,
7476                                                 &events::MessageSendEvent::HandleError { .. } => false,
7477                                                 // Gossip
7478                                                 &events::MessageSendEvent::SendChannelAnnouncement { .. } => false,
7479                                                 &events::MessageSendEvent::BroadcastChannelAnnouncement { .. } => true,
7480                                                 &events::MessageSendEvent::BroadcastChannelUpdate { .. } => true,
7481                                                 &events::MessageSendEvent::BroadcastNodeAnnouncement { .. } => true,
7482                                                 &events::MessageSendEvent::SendChannelUpdate { .. } => false,
7483                                                 &events::MessageSendEvent::SendChannelRangeQuery { .. } => false,
7484                                                 &events::MessageSendEvent::SendShortIdsQuery { .. } => false,
7485                                                 &events::MessageSendEvent::SendReplyChannelRange { .. } => false,
7486                                                 &events::MessageSendEvent::SendGossipTimestampFilter { .. } => false,
7487                                         }
7488                                 });
7489                                 debug_assert!(peer_state.is_connected, "A disconnected peer cannot disconnect");
7490                                 peer_state.is_connected = false;
7491                                 peer_state.ok_to_remove(true)
7492                         } else { debug_assert!(false, "Unconnected peer disconnected"); true }
7493                 };
7494                 if remove_peer {
7495                         per_peer_state.remove(counterparty_node_id);
7496                 }
7497                 mem::drop(per_peer_state);
7498
7499                 for failure in failed_channels.drain(..) {
7500                         self.finish_force_close_channel(failure);
7501                 }
7502         }
7503
7504         fn peer_connected(&self, counterparty_node_id: &PublicKey, init_msg: &msgs::Init, inbound: bool) -> Result<(), ()> {
7505                 if !init_msg.features.supports_static_remote_key() {
7506                         log_debug!(self.logger, "Peer {} does not support static remote key, disconnecting", log_pubkey!(counterparty_node_id));
7507                         return Err(());
7508                 }
7509
7510                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7511
7512                 // If we have too many peers connected which don't have funded channels, disconnect the
7513                 // peer immediately (as long as it doesn't have funded channels). If we have a bunch of
7514                 // unfunded channels taking up space in memory for disconnected peers, we still let new
7515                 // peers connect, but we'll reject new channels from them.
7516                 let connected_peers_without_funded_channels = self.peers_without_funded_channels(|node| node.is_connected);
7517                 let inbound_peer_limited = inbound && connected_peers_without_funded_channels >= MAX_NO_CHANNEL_PEERS;
7518
7519                 {
7520                         let mut peer_state_lock = self.per_peer_state.write().unwrap();
7521                         match peer_state_lock.entry(counterparty_node_id.clone()) {
7522                                 hash_map::Entry::Vacant(e) => {
7523                                         if inbound_peer_limited {
7524                                                 return Err(());
7525                                         }
7526                                         e.insert(Mutex::new(PeerState {
7527                                                 channel_by_id: HashMap::new(),
7528                                                 outbound_v1_channel_by_id: HashMap::new(),
7529                                                 inbound_v1_channel_by_id: HashMap::new(),
7530                                                 inbound_channel_request_by_id: HashMap::new(),
7531                                                 latest_features: init_msg.features.clone(),
7532                                                 pending_msg_events: Vec::new(),
7533                                                 in_flight_monitor_updates: BTreeMap::new(),
7534                                                 monitor_update_blocked_actions: BTreeMap::new(),
7535                                                 actions_blocking_raa_monitor_updates: BTreeMap::new(),
7536                                                 is_connected: true,
7537                                         }));
7538                                 },
7539                                 hash_map::Entry::Occupied(e) => {
7540                                         let mut peer_state = e.get().lock().unwrap();
7541                                         peer_state.latest_features = init_msg.features.clone();
7542
7543                                         let best_block_height = self.best_block.read().unwrap().height();
7544                                         if inbound_peer_limited &&
7545                                                 Self::unfunded_channel_count(&*peer_state, best_block_height) ==
7546                                                 peer_state.channel_by_id.len()
7547                                         {
7548                                                 return Err(());
7549                                         }
7550
7551                                         debug_assert!(!peer_state.is_connected, "A peer shouldn't be connected twice");
7552                                         peer_state.is_connected = true;
7553                                 },
7554                         }
7555                 }
7556
7557                 log_debug!(self.logger, "Generating channel_reestablish events for {}", log_pubkey!(counterparty_node_id));
7558
7559                 let per_peer_state = self.per_peer_state.read().unwrap();
7560                 if let Some(peer_state_mutex) = per_peer_state.get(counterparty_node_id) {
7561                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
7562                         let peer_state = &mut *peer_state_lock;
7563                         let pending_msg_events = &mut peer_state.pending_msg_events;
7564
7565                         // Since unfunded channel maps are cleared upon disconnecting a peer, and they're not persisted
7566                         // (so won't be recovered after a crash) we don't need to bother closing unfunded channels and
7567                         // clearing their maps here. Instead we can just send queue channel_reestablish messages for
7568                         // channels in the channel_by_id map.
7569                         peer_state.channel_by_id.iter_mut().for_each(|(_, chan)| {
7570                                 pending_msg_events.push(events::MessageSendEvent::SendChannelReestablish {
7571                                         node_id: chan.context.get_counterparty_node_id(),
7572                                         msg: chan.get_channel_reestablish(&self.logger),
7573                                 });
7574                         });
7575                 }
7576                 //TODO: Also re-broadcast announcement_signatures
7577                 Ok(())
7578         }
7579
7580         fn handle_error(&self, counterparty_node_id: &PublicKey, msg: &msgs::ErrorMessage) {
7581                 let _persistence_guard = PersistenceNotifierGuard::notify_on_drop(self);
7582
7583                 match &msg.data as &str {
7584                         "cannot co-op close channel w/ active htlcs"|
7585                         "link failed to shutdown" =>
7586                         {
7587                                 // LND hasn't properly handled shutdown messages ever, and force-closes any time we
7588                                 // send one while HTLCs are still present. The issue is tracked at
7589                                 // https://github.com/lightningnetwork/lnd/issues/6039 and has had multiple patches
7590                                 // to fix it but none so far have managed to land upstream. The issue appears to be
7591                                 // very low priority for the LND team despite being marked "P1".
7592                                 // We're not going to bother handling this in a sensible way, instead simply
7593                                 // repeating the Shutdown message on repeat until morale improves.
7594                                 if !msg.channel_id.is_zero() {
7595                                         let per_peer_state = self.per_peer_state.read().unwrap();
7596                                         let peer_state_mutex_opt = per_peer_state.get(counterparty_node_id);
7597                                         if peer_state_mutex_opt.is_none() { return; }
7598                                         let mut peer_state = peer_state_mutex_opt.unwrap().lock().unwrap();
7599                                         if let Some(chan) = peer_state.channel_by_id.get(&msg.channel_id) {
7600                                                 if let Some(msg) = chan.get_outbound_shutdown() {
7601                                                         peer_state.pending_msg_events.push(events::MessageSendEvent::SendShutdown {
7602                                                                 node_id: *counterparty_node_id,
7603                                                                 msg,
7604                                                         });
7605                                                 }
7606                                                 peer_state.pending_msg_events.push(events::MessageSendEvent::HandleError {
7607                                                         node_id: *counterparty_node_id,
7608                                                         action: msgs::ErrorAction::SendWarningMessage {
7609                                                                 msg: msgs::WarningMessage {
7610                                                                         channel_id: msg.channel_id,
7611                                                                         data: "You appear to be exhibiting LND bug 6039, we'll keep sending you shutdown messages until you handle them correctly".to_owned()
7612                                                                 },
7613                                                                 log_level: Level::Trace,
7614                                                         }
7615                                                 });
7616                                         }
7617                                 }
7618                                 return;
7619                         }
7620                         _ => {}
7621                 }
7622
7623                 if msg.channel_id.is_zero() {
7624                         let channel_ids: Vec<ChannelId> = {
7625                                 let per_peer_state = self.per_peer_state.read().unwrap();
7626                                 let peer_state_mutex_opt = per_peer_state.get(counterparty_node_id);
7627                                 if peer_state_mutex_opt.is_none() { return; }
7628                                 let mut peer_state_lock = peer_state_mutex_opt.unwrap().lock().unwrap();
7629                                 let peer_state = &mut *peer_state_lock;
7630                                 // Note that we don't bother generating any events for pre-accept channels -
7631                                 // they're not considered "channels" yet from the PoV of our events interface.
7632                                 peer_state.inbound_channel_request_by_id.clear();
7633                                 peer_state.channel_by_id.keys().cloned()
7634                                         .chain(peer_state.outbound_v1_channel_by_id.keys().cloned())
7635                                         .chain(peer_state.inbound_v1_channel_by_id.keys().cloned()).collect()
7636                         };
7637                         for channel_id in channel_ids {
7638                                 // Untrusted messages from peer, we throw away the error if id points to a non-existent channel
7639                                 let _ = self.force_close_channel_with_peer(&channel_id, counterparty_node_id, Some(&msg.data), true);
7640                         }
7641                 } else {
7642                         {
7643                                 // First check if we can advance the channel type and try again.
7644                                 let per_peer_state = self.per_peer_state.read().unwrap();
7645                                 let peer_state_mutex_opt = per_peer_state.get(counterparty_node_id);
7646                                 if peer_state_mutex_opt.is_none() { return; }
7647                                 let mut peer_state_lock = peer_state_mutex_opt.unwrap().lock().unwrap();
7648                                 let peer_state = &mut *peer_state_lock;
7649                                 if let Some(chan) = peer_state.outbound_v1_channel_by_id.get_mut(&msg.channel_id) {
7650                                         if let Ok(msg) = chan.maybe_handle_error_without_close(self.genesis_hash, &self.fee_estimator) {
7651                                                 peer_state.pending_msg_events.push(events::MessageSendEvent::SendOpenChannel {
7652                                                         node_id: *counterparty_node_id,
7653                                                         msg,
7654                                                 });
7655                                                 return;
7656                                         }
7657                                 }
7658                         }
7659
7660                         // Untrusted messages from peer, we throw away the error if id points to a non-existent channel
7661                         let _ = self.force_close_channel_with_peer(&msg.channel_id, counterparty_node_id, Some(&msg.data), true);
7662                 }
7663         }
7664
7665         fn provided_node_features(&self) -> NodeFeatures {
7666                 provided_node_features(&self.default_configuration)
7667         }
7668
7669         fn provided_init_features(&self, _their_init_features: &PublicKey) -> InitFeatures {
7670                 provided_init_features(&self.default_configuration)
7671         }
7672
7673         fn get_genesis_hashes(&self) -> Option<Vec<ChainHash>> {
7674                 Some(vec![ChainHash::from(&self.genesis_hash[..])])
7675         }
7676
7677         fn handle_tx_add_input(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxAddInput) {
7678                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7679                         "Dual-funded channels not supported".to_owned(),
7680                          msg.channel_id.clone())), *counterparty_node_id);
7681         }
7682
7683         fn handle_tx_add_output(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxAddOutput) {
7684                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7685                         "Dual-funded channels not supported".to_owned(),
7686                          msg.channel_id.clone())), *counterparty_node_id);
7687         }
7688
7689         fn handle_tx_remove_input(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxRemoveInput) {
7690                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7691                         "Dual-funded channels not supported".to_owned(),
7692                          msg.channel_id.clone())), *counterparty_node_id);
7693         }
7694
7695         fn handle_tx_remove_output(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxRemoveOutput) {
7696                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7697                         "Dual-funded channels not supported".to_owned(),
7698                          msg.channel_id.clone())), *counterparty_node_id);
7699         }
7700
7701         fn handle_tx_complete(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxComplete) {
7702                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7703                         "Dual-funded channels not supported".to_owned(),
7704                          msg.channel_id.clone())), *counterparty_node_id);
7705         }
7706
7707         fn handle_tx_signatures(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxSignatures) {
7708                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7709                         "Dual-funded channels not supported".to_owned(),
7710                          msg.channel_id.clone())), *counterparty_node_id);
7711         }
7712
7713         fn handle_tx_init_rbf(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxInitRbf) {
7714                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7715                         "Dual-funded channels not supported".to_owned(),
7716                          msg.channel_id.clone())), *counterparty_node_id);
7717         }
7718
7719         fn handle_tx_ack_rbf(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxAckRbf) {
7720                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7721                         "Dual-funded channels not supported".to_owned(),
7722                          msg.channel_id.clone())), *counterparty_node_id);
7723         }
7724
7725         fn handle_tx_abort(&self, counterparty_node_id: &PublicKey, msg: &msgs::TxAbort) {
7726                 let _: Result<(), _> = handle_error!(self, Err(MsgHandleErrInternal::send_err_msg_no_close(
7727                         "Dual-funded channels not supported".to_owned(),
7728                          msg.channel_id.clone())), *counterparty_node_id);
7729         }
7730 }
7731
7732 /// Fetches the set of [`NodeFeatures`] flags which are provided by or required by
7733 /// [`ChannelManager`].
7734 pub(crate) fn provided_node_features(config: &UserConfig) -> NodeFeatures {
7735         let mut node_features = provided_init_features(config).to_context();
7736         node_features.set_keysend_optional();
7737         node_features
7738 }
7739
7740 /// Fetches the set of [`Bolt11InvoiceFeatures`] flags which are provided by or required by
7741 /// [`ChannelManager`].
7742 ///
7743 /// Note that the invoice feature flags can vary depending on if the invoice is a "phantom invoice"
7744 /// or not. Thus, this method is not public.
7745 #[cfg(any(feature = "_test_utils", test))]
7746 pub(crate) fn provided_invoice_features(config: &UserConfig) -> Bolt11InvoiceFeatures {
7747         provided_init_features(config).to_context()
7748 }
7749
7750 /// Fetches the set of [`ChannelFeatures`] flags which are provided by or required by
7751 /// [`ChannelManager`].
7752 pub(crate) fn provided_channel_features(config: &UserConfig) -> ChannelFeatures {
7753         provided_init_features(config).to_context()
7754 }
7755
7756 /// Fetches the set of [`ChannelTypeFeatures`] flags which are provided by or required by
7757 /// [`ChannelManager`].
7758 pub(crate) fn provided_channel_type_features(config: &UserConfig) -> ChannelTypeFeatures {
7759         ChannelTypeFeatures::from_init(&provided_init_features(config))
7760 }
7761
7762 /// Fetches the set of [`InitFeatures`] flags which are provided by or required by
7763 /// [`ChannelManager`].
7764 pub fn provided_init_features(config: &UserConfig) -> InitFeatures {
7765         // Note that if new features are added here which other peers may (eventually) require, we
7766         // should also add the corresponding (optional) bit to the [`ChannelMessageHandler`] impl for
7767         // [`ErroringMessageHandler`].
7768         let mut features = InitFeatures::empty();
7769         features.set_data_loss_protect_required();
7770         features.set_upfront_shutdown_script_optional();
7771         features.set_variable_length_onion_required();
7772         features.set_static_remote_key_required();
7773         features.set_payment_secret_required();
7774         features.set_basic_mpp_optional();
7775         features.set_wumbo_optional();
7776         features.set_shutdown_any_segwit_optional();
7777         features.set_channel_type_optional();
7778         features.set_scid_privacy_optional();
7779         features.set_zero_conf_optional();
7780         if config.channel_handshake_config.negotiate_anchors_zero_fee_htlc_tx {
7781                 features.set_anchors_zero_fee_htlc_tx_optional();
7782         }
7783         features
7784 }
7785
7786 const SERIALIZATION_VERSION: u8 = 1;
7787 const MIN_SERIALIZATION_VERSION: u8 = 1;
7788
7789 impl_writeable_tlv_based!(CounterpartyForwardingInfo, {
7790         (2, fee_base_msat, required),
7791         (4, fee_proportional_millionths, required),
7792         (6, cltv_expiry_delta, required),
7793 });
7794
7795 impl_writeable_tlv_based!(ChannelCounterparty, {
7796         (2, node_id, required),
7797         (4, features, required),
7798         (6, unspendable_punishment_reserve, required),
7799         (8, forwarding_info, option),
7800         (9, outbound_htlc_minimum_msat, option),
7801         (11, outbound_htlc_maximum_msat, option),
7802 });
7803
7804 impl Writeable for ChannelDetails {
7805         fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
7806                 // `user_channel_id` used to be a single u64 value. In order to remain backwards compatible with
7807                 // versions prior to 0.0.113, the u128 is serialized as two separate u64 values.
7808                 let user_channel_id_low = self.user_channel_id as u64;
7809                 let user_channel_id_high_opt = Some((self.user_channel_id >> 64) as u64);
7810                 write_tlv_fields!(writer, {
7811                         (1, self.inbound_scid_alias, option),
7812                         (2, self.channel_id, required),
7813                         (3, self.channel_type, option),
7814                         (4, self.counterparty, required),
7815                         (5, self.outbound_scid_alias, option),
7816                         (6, self.funding_txo, option),
7817                         (7, self.config, option),
7818                         (8, self.short_channel_id, option),
7819                         (9, self.confirmations, option),
7820                         (10, self.channel_value_satoshis, required),
7821                         (12, self.unspendable_punishment_reserve, option),
7822                         (14, user_channel_id_low, required),
7823                         (16, self.next_outbound_htlc_limit_msat, required),  // Forwards compatibility for removed balance_msat field.
7824                         (18, self.outbound_capacity_msat, required),
7825                         (19, self.next_outbound_htlc_limit_msat, required),
7826                         (20, self.inbound_capacity_msat, required),
7827                         (21, self.next_outbound_htlc_minimum_msat, required),
7828                         (22, self.confirmations_required, option),
7829                         (24, self.force_close_spend_delay, option),
7830                         (26, self.is_outbound, required),
7831                         (28, self.is_channel_ready, required),
7832                         (30, self.is_usable, required),
7833                         (32, self.is_public, required),
7834                         (33, self.inbound_htlc_minimum_msat, option),
7835                         (35, self.inbound_htlc_maximum_msat, option),
7836                         (37, user_channel_id_high_opt, option),
7837                         (39, self.feerate_sat_per_1000_weight, option),
7838                         (41, self.channel_shutdown_state, option),
7839                 });
7840                 Ok(())
7841         }
7842 }
7843
7844 impl Readable for ChannelDetails {
7845         fn read<R: Read>(reader: &mut R) -> Result<Self, DecodeError> {
7846                 _init_and_read_len_prefixed_tlv_fields!(reader, {
7847                         (1, inbound_scid_alias, option),
7848                         (2, channel_id, required),
7849                         (3, channel_type, option),
7850                         (4, counterparty, required),
7851                         (5, outbound_scid_alias, option),
7852                         (6, funding_txo, option),
7853                         (7, config, option),
7854                         (8, short_channel_id, option),
7855                         (9, confirmations, option),
7856                         (10, channel_value_satoshis, required),
7857                         (12, unspendable_punishment_reserve, option),
7858                         (14, user_channel_id_low, required),
7859                         (16, _balance_msat, option),  // Backwards compatibility for removed balance_msat field.
7860                         (18, outbound_capacity_msat, required),
7861                         // Note that by the time we get past the required read above, outbound_capacity_msat will be
7862                         // filled in, so we can safely unwrap it here.
7863                         (19, next_outbound_htlc_limit_msat, (default_value, outbound_capacity_msat.0.unwrap() as u64)),
7864                         (20, inbound_capacity_msat, required),
7865                         (21, next_outbound_htlc_minimum_msat, (default_value, 0)),
7866                         (22, confirmations_required, option),
7867                         (24, force_close_spend_delay, option),
7868                         (26, is_outbound, required),
7869                         (28, is_channel_ready, required),
7870                         (30, is_usable, required),
7871                         (32, is_public, required),
7872                         (33, inbound_htlc_minimum_msat, option),
7873                         (35, inbound_htlc_maximum_msat, option),
7874                         (37, user_channel_id_high_opt, option),
7875                         (39, feerate_sat_per_1000_weight, option),
7876                         (41, channel_shutdown_state, option),
7877                 });
7878
7879                 // `user_channel_id` used to be a single u64 value. In order to remain backwards compatible with
7880                 // versions prior to 0.0.113, the u128 is serialized as two separate u64 values.
7881                 let user_channel_id_low: u64 = user_channel_id_low.0.unwrap();
7882                 let user_channel_id = user_channel_id_low as u128 +
7883                         ((user_channel_id_high_opt.unwrap_or(0 as u64) as u128) << 64);
7884
7885                 let _balance_msat: Option<u64> = _balance_msat;
7886
7887                 Ok(Self {
7888                         inbound_scid_alias,
7889                         channel_id: channel_id.0.unwrap(),
7890                         channel_type,
7891                         counterparty: counterparty.0.unwrap(),
7892                         outbound_scid_alias,
7893                         funding_txo,
7894                         config,
7895                         short_channel_id,
7896                         channel_value_satoshis: channel_value_satoshis.0.unwrap(),
7897                         unspendable_punishment_reserve,
7898                         user_channel_id,
7899                         outbound_capacity_msat: outbound_capacity_msat.0.unwrap(),
7900                         next_outbound_htlc_limit_msat: next_outbound_htlc_limit_msat.0.unwrap(),
7901                         next_outbound_htlc_minimum_msat: next_outbound_htlc_minimum_msat.0.unwrap(),
7902                         inbound_capacity_msat: inbound_capacity_msat.0.unwrap(),
7903                         confirmations_required,
7904                         confirmations,
7905                         force_close_spend_delay,
7906                         is_outbound: is_outbound.0.unwrap(),
7907                         is_channel_ready: is_channel_ready.0.unwrap(),
7908                         is_usable: is_usable.0.unwrap(),
7909                         is_public: is_public.0.unwrap(),
7910                         inbound_htlc_minimum_msat,
7911                         inbound_htlc_maximum_msat,
7912                         feerate_sat_per_1000_weight,
7913                         channel_shutdown_state,
7914                 })
7915         }
7916 }
7917
7918 impl_writeable_tlv_based!(PhantomRouteHints, {
7919         (2, channels, required_vec),
7920         (4, phantom_scid, required),
7921         (6, real_node_pubkey, required),
7922 });
7923
7924 impl_writeable_tlv_based_enum!(PendingHTLCRouting,
7925         (0, Forward) => {
7926                 (0, onion_packet, required),
7927                 (2, short_channel_id, required),
7928         },
7929         (1, Receive) => {
7930                 (0, payment_data, required),
7931                 (1, phantom_shared_secret, option),
7932                 (2, incoming_cltv_expiry, required),
7933                 (3, payment_metadata, option),
7934                 (5, custom_tlvs, optional_vec),
7935         },
7936         (2, ReceiveKeysend) => {
7937                 (0, payment_preimage, required),
7938                 (2, incoming_cltv_expiry, required),
7939                 (3, payment_metadata, option),
7940                 (4, payment_data, option), // Added in 0.0.116
7941                 (5, custom_tlvs, optional_vec),
7942         },
7943 ;);
7944
7945 impl_writeable_tlv_based!(PendingHTLCInfo, {
7946         (0, routing, required),
7947         (2, incoming_shared_secret, required),
7948         (4, payment_hash, required),
7949         (6, outgoing_amt_msat, required),
7950         (8, outgoing_cltv_value, required),
7951         (9, incoming_amt_msat, option),
7952         (10, skimmed_fee_msat, option),
7953 });
7954
7955
7956 impl Writeable for HTLCFailureMsg {
7957         fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
7958                 match self {
7959                         HTLCFailureMsg::Relay(msgs::UpdateFailHTLC { channel_id, htlc_id, reason }) => {
7960                                 0u8.write(writer)?;
7961                                 channel_id.write(writer)?;
7962                                 htlc_id.write(writer)?;
7963                                 reason.write(writer)?;
7964                         },
7965                         HTLCFailureMsg::Malformed(msgs::UpdateFailMalformedHTLC {
7966                                 channel_id, htlc_id, sha256_of_onion, failure_code
7967                         }) => {
7968                                 1u8.write(writer)?;
7969                                 channel_id.write(writer)?;
7970                                 htlc_id.write(writer)?;
7971                                 sha256_of_onion.write(writer)?;
7972                                 failure_code.write(writer)?;
7973                         },
7974                 }
7975                 Ok(())
7976         }
7977 }
7978
7979 impl Readable for HTLCFailureMsg {
7980         fn read<R: Read>(reader: &mut R) -> Result<Self, DecodeError> {
7981                 let id: u8 = Readable::read(reader)?;
7982                 match id {
7983                         0 => {
7984                                 Ok(HTLCFailureMsg::Relay(msgs::UpdateFailHTLC {
7985                                         channel_id: Readable::read(reader)?,
7986                                         htlc_id: Readable::read(reader)?,
7987                                         reason: Readable::read(reader)?,
7988                                 }))
7989                         },
7990                         1 => {
7991                                 Ok(HTLCFailureMsg::Malformed(msgs::UpdateFailMalformedHTLC {
7992                                         channel_id: Readable::read(reader)?,
7993                                         htlc_id: Readable::read(reader)?,
7994                                         sha256_of_onion: Readable::read(reader)?,
7995                                         failure_code: Readable::read(reader)?,
7996                                 }))
7997                         },
7998                         // In versions prior to 0.0.101, HTLCFailureMsg objects were written with type 0 or 1 but
7999                         // weren't length-prefixed and thus didn't support reading the TLV stream suffix of the network
8000                         // messages contained in the variants.
8001                         // In version 0.0.101, support for reading the variants with these types was added, and
8002                         // we should migrate to writing these variants when UpdateFailHTLC or
8003                         // UpdateFailMalformedHTLC get TLV fields.
8004                         2 => {
8005                                 let length: BigSize = Readable::read(reader)?;
8006                                 let mut s = FixedLengthReader::new(reader, length.0);
8007                                 let res = Readable::read(&mut s)?;
8008                                 s.eat_remaining()?; // Return ShortRead if there's actually not enough bytes
8009                                 Ok(HTLCFailureMsg::Relay(res))
8010                         },
8011                         3 => {
8012                                 let length: BigSize = Readable::read(reader)?;
8013                                 let mut s = FixedLengthReader::new(reader, length.0);
8014                                 let res = Readable::read(&mut s)?;
8015                                 s.eat_remaining()?; // Return ShortRead if there's actually not enough bytes
8016                                 Ok(HTLCFailureMsg::Malformed(res))
8017                         },
8018                         _ => Err(DecodeError::UnknownRequiredFeature),
8019                 }
8020         }
8021 }
8022
8023 impl_writeable_tlv_based_enum!(PendingHTLCStatus, ;
8024         (0, Forward),
8025         (1, Fail),
8026 );
8027
8028 impl_writeable_tlv_based!(HTLCPreviousHopData, {
8029         (0, short_channel_id, required),
8030         (1, phantom_shared_secret, option),
8031         (2, outpoint, required),
8032         (4, htlc_id, required),
8033         (6, incoming_packet_shared_secret, required),
8034         (7, user_channel_id, option),
8035 });
8036
8037 impl Writeable for ClaimableHTLC {
8038         fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
8039                 let (payment_data, keysend_preimage) = match &self.onion_payload {
8040                         OnionPayload::Invoice { _legacy_hop_data } => (_legacy_hop_data.as_ref(), None),
8041                         OnionPayload::Spontaneous(preimage) => (None, Some(preimage)),
8042                 };
8043                 write_tlv_fields!(writer, {
8044                         (0, self.prev_hop, required),
8045                         (1, self.total_msat, required),
8046                         (2, self.value, required),
8047                         (3, self.sender_intended_value, required),
8048                         (4, payment_data, option),
8049                         (5, self.total_value_received, option),
8050                         (6, self.cltv_expiry, required),
8051                         (8, keysend_preimage, option),
8052                         (10, self.counterparty_skimmed_fee_msat, option),
8053                 });
8054                 Ok(())
8055         }
8056 }
8057
8058 impl Readable for ClaimableHTLC {
8059         fn read<R: Read>(reader: &mut R) -> Result<Self, DecodeError> {
8060                 _init_and_read_len_prefixed_tlv_fields!(reader, {
8061                         (0, prev_hop, required),
8062                         (1, total_msat, option),
8063                         (2, value_ser, required),
8064                         (3, sender_intended_value, option),
8065                         (4, payment_data_opt, option),
8066                         (5, total_value_received, option),
8067                         (6, cltv_expiry, required),
8068                         (8, keysend_preimage, option),
8069                         (10, counterparty_skimmed_fee_msat, option),
8070                 });
8071                 let payment_data: Option<msgs::FinalOnionHopData> = payment_data_opt;
8072                 let value = value_ser.0.unwrap();
8073                 let onion_payload = match keysend_preimage {
8074                         Some(p) => {
8075                                 if payment_data.is_some() {
8076                                         return Err(DecodeError::InvalidValue)
8077                                 }
8078                                 if total_msat.is_none() {
8079                                         total_msat = Some(value);
8080                                 }
8081                                 OnionPayload::Spontaneous(p)
8082                         },
8083                         None => {
8084                                 if total_msat.is_none() {
8085                                         if payment_data.is_none() {
8086                                                 return Err(DecodeError::InvalidValue)
8087                                         }
8088                                         total_msat = Some(payment_data.as_ref().unwrap().total_msat);
8089                                 }
8090                                 OnionPayload::Invoice { _legacy_hop_data: payment_data }
8091                         },
8092                 };
8093                 Ok(Self {
8094                         prev_hop: prev_hop.0.unwrap(),
8095                         timer_ticks: 0,
8096                         value,
8097                         sender_intended_value: sender_intended_value.unwrap_or(value),
8098                         total_value_received,
8099                         total_msat: total_msat.unwrap(),
8100                         onion_payload,
8101                         cltv_expiry: cltv_expiry.0.unwrap(),
8102                         counterparty_skimmed_fee_msat,
8103                 })
8104         }
8105 }
8106
8107 impl Readable for HTLCSource {
8108         fn read<R: Read>(reader: &mut R) -> Result<Self, DecodeError> {
8109                 let id: u8 = Readable::read(reader)?;
8110                 match id {
8111                         0 => {
8112                                 let mut session_priv: crate::util::ser::RequiredWrapper<SecretKey> = crate::util::ser::RequiredWrapper(None);
8113                                 let mut first_hop_htlc_msat: u64 = 0;
8114                                 let mut path_hops = Vec::new();
8115                                 let mut payment_id = None;
8116                                 let mut payment_params: Option<PaymentParameters> = None;
8117                                 let mut blinded_tail: Option<BlindedTail> = None;
8118                                 read_tlv_fields!(reader, {
8119                                         (0, session_priv, required),
8120                                         (1, payment_id, option),
8121                                         (2, first_hop_htlc_msat, required),
8122                                         (4, path_hops, required_vec),
8123                                         (5, payment_params, (option: ReadableArgs, 0)),
8124                                         (6, blinded_tail, option),
8125                                 });
8126                                 if payment_id.is_none() {
8127                                         // For backwards compat, if there was no payment_id written, use the session_priv bytes
8128                                         // instead.
8129                                         payment_id = Some(PaymentId(*session_priv.0.unwrap().as_ref()));
8130                                 }
8131                                 let path = Path { hops: path_hops, blinded_tail };
8132                                 if path.hops.len() == 0 {
8133                                         return Err(DecodeError::InvalidValue);
8134                                 }
8135                                 if let Some(params) = payment_params.as_mut() {
8136                                         if let Payee::Clear { ref mut final_cltv_expiry_delta, .. } = params.payee {
8137                                                 if final_cltv_expiry_delta == &0 {
8138                                                         *final_cltv_expiry_delta = path.final_cltv_expiry_delta().ok_or(DecodeError::InvalidValue)?;
8139                                                 }
8140                                         }
8141                                 }
8142                                 Ok(HTLCSource::OutboundRoute {
8143                                         session_priv: session_priv.0.unwrap(),
8144                                         first_hop_htlc_msat,
8145                                         path,
8146                                         payment_id: payment_id.unwrap(),
8147                                 })
8148                         }
8149                         1 => Ok(HTLCSource::PreviousHopData(Readable::read(reader)?)),
8150                         _ => Err(DecodeError::UnknownRequiredFeature),
8151                 }
8152         }
8153 }
8154
8155 impl Writeable for HTLCSource {
8156         fn write<W: Writer>(&self, writer: &mut W) -> Result<(), crate::io::Error> {
8157                 match self {
8158                         HTLCSource::OutboundRoute { ref session_priv, ref first_hop_htlc_msat, ref path, payment_id } => {
8159                                 0u8.write(writer)?;
8160                                 let payment_id_opt = Some(payment_id);
8161                                 write_tlv_fields!(writer, {
8162                                         (0, session_priv, required),
8163                                         (1, payment_id_opt, option),
8164                                         (2, first_hop_htlc_msat, required),
8165                                         // 3 was previously used to write a PaymentSecret for the payment.
8166                                         (4, path.hops, required_vec),
8167                                         (5, None::<PaymentParameters>, option), // payment_params in LDK versions prior to 0.0.115
8168                                         (6, path.blinded_tail, option),
8169                                  });
8170                         }
8171                         HTLCSource::PreviousHopData(ref field) => {
8172                                 1u8.write(writer)?;
8173                                 field.write(writer)?;
8174                         }
8175                 }
8176                 Ok(())
8177         }
8178 }
8179
8180 impl_writeable_tlv_based!(PendingAddHTLCInfo, {
8181         (0, forward_info, required),
8182         (1, prev_user_channel_id, (default_value, 0)),
8183         (2, prev_short_channel_id, required),
8184         (4, prev_htlc_id, required),
8185         (6, prev_funding_outpoint, required),
8186 });
8187
8188 impl_writeable_tlv_based_enum!(HTLCForwardInfo,
8189         (1, FailHTLC) => {
8190                 (0, htlc_id, required),
8191                 (2, err_packet, required),
8192         };
8193         (0, AddHTLC)
8194 );
8195
8196 impl_writeable_tlv_based!(PendingInboundPayment, {
8197         (0, payment_secret, required),
8198         (2, expiry_time, required),
8199         (4, user_payment_id, required),
8200         (6, payment_preimage, required),
8201         (8, min_value_msat, required),
8202 });
8203
8204 impl<M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref> Writeable for ChannelManager<M, T, ES, NS, SP, F, R, L>
8205 where
8206         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
8207         T::Target: BroadcasterInterface,
8208         ES::Target: EntropySource,
8209         NS::Target: NodeSigner,
8210         SP::Target: SignerProvider,
8211         F::Target: FeeEstimator,
8212         R::Target: Router,
8213         L::Target: Logger,
8214 {
8215         fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
8216                 let _consistency_lock = self.total_consistency_lock.write().unwrap();
8217
8218                 write_ver_prefix!(writer, SERIALIZATION_VERSION, MIN_SERIALIZATION_VERSION);
8219
8220                 self.genesis_hash.write(writer)?;
8221                 {
8222                         let best_block = self.best_block.read().unwrap();
8223                         best_block.height().write(writer)?;
8224                         best_block.block_hash().write(writer)?;
8225                 }
8226
8227                 let mut serializable_peer_count: u64 = 0;
8228                 {
8229                         let per_peer_state = self.per_peer_state.read().unwrap();
8230                         let mut unfunded_channels = 0;
8231                         let mut number_of_channels = 0;
8232                         for (_, peer_state_mutex) in per_peer_state.iter() {
8233                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
8234                                 let peer_state = &mut *peer_state_lock;
8235                                 if !peer_state.ok_to_remove(false) {
8236                                         serializable_peer_count += 1;
8237                                 }
8238                                 number_of_channels += peer_state.channel_by_id.len();
8239                                 for (_, channel) in peer_state.channel_by_id.iter() {
8240                                         if !channel.context.is_funding_initiated() {
8241                                                 unfunded_channels += 1;
8242                                         }
8243                                 }
8244                         }
8245
8246                         ((number_of_channels - unfunded_channels) as u64).write(writer)?;
8247
8248                         for (_, peer_state_mutex) in per_peer_state.iter() {
8249                                 let mut peer_state_lock = peer_state_mutex.lock().unwrap();
8250                                 let peer_state = &mut *peer_state_lock;
8251                                 for (_, channel) in peer_state.channel_by_id.iter() {
8252                                         if channel.context.is_funding_initiated() {
8253                                                 channel.write(writer)?;
8254                                         }
8255                                 }
8256                         }
8257                 }
8258
8259                 {
8260                         let forward_htlcs = self.forward_htlcs.lock().unwrap();
8261                         (forward_htlcs.len() as u64).write(writer)?;
8262                         for (short_channel_id, pending_forwards) in forward_htlcs.iter() {
8263                                 short_channel_id.write(writer)?;
8264                                 (pending_forwards.len() as u64).write(writer)?;
8265                                 for forward in pending_forwards {
8266                                         forward.write(writer)?;
8267                                 }
8268                         }
8269                 }
8270
8271                 let per_peer_state = self.per_peer_state.write().unwrap();
8272
8273                 let pending_inbound_payments = self.pending_inbound_payments.lock().unwrap();
8274                 let claimable_payments = self.claimable_payments.lock().unwrap();
8275                 let pending_outbound_payments = self.pending_outbound_payments.pending_outbound_payments.lock().unwrap();
8276
8277                 let mut htlc_purposes: Vec<&events::PaymentPurpose> = Vec::new();
8278                 let mut htlc_onion_fields: Vec<&_> = Vec::new();
8279                 (claimable_payments.claimable_payments.len() as u64).write(writer)?;
8280                 for (payment_hash, payment) in claimable_payments.claimable_payments.iter() {
8281                         payment_hash.write(writer)?;
8282                         (payment.htlcs.len() as u64).write(writer)?;
8283                         for htlc in payment.htlcs.iter() {
8284                                 htlc.write(writer)?;
8285                         }
8286                         htlc_purposes.push(&payment.purpose);
8287                         htlc_onion_fields.push(&payment.onion_fields);
8288                 }
8289
8290                 let mut monitor_update_blocked_actions_per_peer = None;
8291                 let mut peer_states = Vec::new();
8292                 for (_, peer_state_mutex) in per_peer_state.iter() {
8293                         // Because we're holding the owning `per_peer_state` write lock here there's no chance
8294                         // of a lockorder violation deadlock - no other thread can be holding any
8295                         // per_peer_state lock at all.
8296                         peer_states.push(peer_state_mutex.unsafe_well_ordered_double_lock_self());
8297                 }
8298
8299                 (serializable_peer_count).write(writer)?;
8300                 for ((peer_pubkey, _), peer_state) in per_peer_state.iter().zip(peer_states.iter()) {
8301                         // Peers which we have no channels to should be dropped once disconnected. As we
8302                         // disconnect all peers when shutting down and serializing the ChannelManager, we
8303                         // consider all peers as disconnected here. There's therefore no need write peers with
8304                         // no channels.
8305                         if !peer_state.ok_to_remove(false) {
8306                                 peer_pubkey.write(writer)?;
8307                                 peer_state.latest_features.write(writer)?;
8308                                 if !peer_state.monitor_update_blocked_actions.is_empty() {
8309                                         monitor_update_blocked_actions_per_peer
8310                                                 .get_or_insert_with(Vec::new)
8311                                                 .push((*peer_pubkey, &peer_state.monitor_update_blocked_actions));
8312                                 }
8313                         }
8314                 }
8315
8316                 let events = self.pending_events.lock().unwrap();
8317                 // LDK versions prior to 0.0.115 don't support post-event actions, thus if there's no
8318                 // actions at all, skip writing the required TLV. Otherwise, pre-0.0.115 versions will
8319                 // refuse to read the new ChannelManager.
8320                 let events_not_backwards_compatible = events.iter().any(|(_, action)| action.is_some());
8321                 if events_not_backwards_compatible {
8322                         // If we're gonna write a even TLV that will overwrite our events anyway we might as
8323                         // well save the space and not write any events here.
8324                         0u64.write(writer)?;
8325                 } else {
8326                         (events.len() as u64).write(writer)?;
8327                         for (event, _) in events.iter() {
8328                                 event.write(writer)?;
8329                         }
8330                 }
8331
8332                 // LDK versions prior to 0.0.116 wrote the `pending_background_events`
8333                 // `MonitorUpdateRegeneratedOnStartup`s here, however there was never a reason to do so -
8334                 // the closing monitor updates were always effectively replayed on startup (either directly
8335                 // by calling `broadcast_latest_holder_commitment_txn` on a `ChannelMonitor` during
8336                 // deserialization or, in 0.0.115, by regenerating the monitor update itself).
8337                 0u64.write(writer)?;
8338
8339                 // Prior to 0.0.111 we tracked node_announcement serials here, however that now happens in
8340                 // `PeerManager`, and thus we simply write the `highest_seen_timestamp` twice, which is
8341                 // likely to be identical.
8342                 (self.highest_seen_timestamp.load(Ordering::Acquire) as u32).write(writer)?;
8343                 (self.highest_seen_timestamp.load(Ordering::Acquire) as u32).write(writer)?;
8344
8345                 (pending_inbound_payments.len() as u64).write(writer)?;
8346                 for (hash, pending_payment) in pending_inbound_payments.iter() {
8347                         hash.write(writer)?;
8348                         pending_payment.write(writer)?;
8349                 }
8350
8351                 // For backwards compat, write the session privs and their total length.
8352                 let mut num_pending_outbounds_compat: u64 = 0;
8353                 for (_, outbound) in pending_outbound_payments.iter() {
8354                         if !outbound.is_fulfilled() && !outbound.abandoned() {
8355                                 num_pending_outbounds_compat += outbound.remaining_parts() as u64;
8356                         }
8357                 }
8358                 num_pending_outbounds_compat.write(writer)?;
8359                 for (_, outbound) in pending_outbound_payments.iter() {
8360                         match outbound {
8361                                 PendingOutboundPayment::Legacy { session_privs } |
8362                                 PendingOutboundPayment::Retryable { session_privs, .. } => {
8363                                         for session_priv in session_privs.iter() {
8364                                                 session_priv.write(writer)?;
8365                                         }
8366                                 }
8367                                 PendingOutboundPayment::AwaitingInvoice { .. } => {},
8368                                 PendingOutboundPayment::InvoiceReceived { .. } => {},
8369                                 PendingOutboundPayment::Fulfilled { .. } => {},
8370                                 PendingOutboundPayment::Abandoned { .. } => {},
8371                         }
8372                 }
8373
8374                 // Encode without retry info for 0.0.101 compatibility.
8375                 let mut pending_outbound_payments_no_retry: HashMap<PaymentId, HashSet<[u8; 32]>> = HashMap::new();
8376                 for (id, outbound) in pending_outbound_payments.iter() {
8377                         match outbound {
8378                                 PendingOutboundPayment::Legacy { session_privs } |
8379                                 PendingOutboundPayment::Retryable { session_privs, .. } => {
8380                                         pending_outbound_payments_no_retry.insert(*id, session_privs.clone());
8381                                 },
8382                                 _ => {},
8383                         }
8384                 }
8385
8386                 let mut pending_intercepted_htlcs = None;
8387                 let our_pending_intercepts = self.pending_intercepted_htlcs.lock().unwrap();
8388                 if our_pending_intercepts.len() != 0 {
8389                         pending_intercepted_htlcs = Some(our_pending_intercepts);
8390                 }
8391
8392                 let mut pending_claiming_payments = Some(&claimable_payments.pending_claiming_payments);
8393                 if pending_claiming_payments.as_ref().unwrap().is_empty() {
8394                         // LDK versions prior to 0.0.113 do not know how to read the pending claimed payments
8395                         // map. Thus, if there are no entries we skip writing a TLV for it.
8396                         pending_claiming_payments = None;
8397                 }
8398
8399                 let mut in_flight_monitor_updates: Option<HashMap<(&PublicKey, &OutPoint), &Vec<ChannelMonitorUpdate>>> = None;
8400                 for ((counterparty_id, _), peer_state) in per_peer_state.iter().zip(peer_states.iter()) {
8401                         for (funding_outpoint, updates) in peer_state.in_flight_monitor_updates.iter() {
8402                                 if !updates.is_empty() {
8403                                         if in_flight_monitor_updates.is_none() { in_flight_monitor_updates = Some(HashMap::new()); }
8404                                         in_flight_monitor_updates.as_mut().unwrap().insert((counterparty_id, funding_outpoint), updates);
8405                                 }
8406                         }
8407                 }
8408
8409                 write_tlv_fields!(writer, {
8410                         (1, pending_outbound_payments_no_retry, required),
8411                         (2, pending_intercepted_htlcs, option),
8412                         (3, pending_outbound_payments, required),
8413                         (4, pending_claiming_payments, option),
8414                         (5, self.our_network_pubkey, required),
8415                         (6, monitor_update_blocked_actions_per_peer, option),
8416                         (7, self.fake_scid_rand_bytes, required),
8417                         (8, if events_not_backwards_compatible { Some(&*events) } else { None }, option),
8418                         (9, htlc_purposes, required_vec),
8419                         (10, in_flight_monitor_updates, option),
8420                         (11, self.probing_cookie_secret, required),
8421                         (13, htlc_onion_fields, optional_vec),
8422                 });
8423
8424                 Ok(())
8425         }
8426 }
8427
8428 impl Writeable for VecDeque<(Event, Option<EventCompletionAction>)> {
8429         fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
8430                 (self.len() as u64).write(w)?;
8431                 for (event, action) in self.iter() {
8432                         event.write(w)?;
8433                         action.write(w)?;
8434                         #[cfg(debug_assertions)] {
8435                                 // Events are MaybeReadable, in some cases indicating that they shouldn't actually
8436                                 // be persisted and are regenerated on restart. However, if such an event has a
8437                                 // post-event-handling action we'll write nothing for the event and would have to
8438                                 // either forget the action or fail on deserialization (which we do below). Thus,
8439                                 // check that the event is sane here.
8440                                 let event_encoded = event.encode();
8441                                 let event_read: Option<Event> =
8442                                         MaybeReadable::read(&mut &event_encoded[..]).unwrap();
8443                                 if action.is_some() { assert!(event_read.is_some()); }
8444                         }
8445                 }
8446                 Ok(())
8447         }
8448 }
8449 impl Readable for VecDeque<(Event, Option<EventCompletionAction>)> {
8450         fn read<R: Read>(reader: &mut R) -> Result<Self, DecodeError> {
8451                 let len: u64 = Readable::read(reader)?;
8452                 const MAX_ALLOC_SIZE: u64 = 1024 * 16;
8453                 let mut events: Self = VecDeque::with_capacity(cmp::min(
8454                         MAX_ALLOC_SIZE/mem::size_of::<(events::Event, Option<EventCompletionAction>)>() as u64,
8455                         len) as usize);
8456                 for _ in 0..len {
8457                         let ev_opt = MaybeReadable::read(reader)?;
8458                         let action = Readable::read(reader)?;
8459                         if let Some(ev) = ev_opt {
8460                                 events.push_back((ev, action));
8461                         } else if action.is_some() {
8462                                 return Err(DecodeError::InvalidValue);
8463                         }
8464                 }
8465                 Ok(events)
8466         }
8467 }
8468
8469 impl_writeable_tlv_based_enum!(ChannelShutdownState,
8470         (0, NotShuttingDown) => {},
8471         (2, ShutdownInitiated) => {},
8472         (4, ResolvingHTLCs) => {},
8473         (6, NegotiatingClosingFee) => {},
8474         (8, ShutdownComplete) => {}, ;
8475 );
8476
8477 /// Arguments for the creation of a ChannelManager that are not deserialized.
8478 ///
8479 /// At a high-level, the process for deserializing a ChannelManager and resuming normal operation
8480 /// is:
8481 /// 1) Deserialize all stored [`ChannelMonitor`]s.
8482 /// 2) Deserialize the [`ChannelManager`] by filling in this struct and calling:
8483 ///    `<(BlockHash, ChannelManager)>::read(reader, args)`
8484 ///    This may result in closing some channels if the [`ChannelMonitor`] is newer than the stored
8485 ///    [`ChannelManager`] state to ensure no loss of funds. Thus, transactions may be broadcasted.
8486 /// 3) If you are not fetching full blocks, register all relevant [`ChannelMonitor`] outpoints the
8487 ///    same way you would handle a [`chain::Filter`] call using
8488 ///    [`ChannelMonitor::get_outputs_to_watch`] and [`ChannelMonitor::get_funding_txo`].
8489 /// 4) Reconnect blocks on your [`ChannelMonitor`]s.
8490 /// 5) Disconnect/connect blocks on the [`ChannelManager`].
8491 /// 6) Re-persist the [`ChannelMonitor`]s to ensure the latest state is on disk.
8492 ///    Note that if you're using a [`ChainMonitor`] for your [`chain::Watch`] implementation, you
8493 ///    will likely accomplish this as a side-effect of calling [`chain::Watch::watch_channel`] in
8494 ///    the next step.
8495 /// 7) Move the [`ChannelMonitor`]s into your local [`chain::Watch`]. If you're using a
8496 ///    [`ChainMonitor`], this is done by calling [`chain::Watch::watch_channel`].
8497 ///
8498 /// Note that the ordering of #4-7 is not of importance, however all four must occur before you
8499 /// call any other methods on the newly-deserialized [`ChannelManager`].
8500 ///
8501 /// Note that because some channels may be closed during deserialization, it is critical that you
8502 /// always deserialize only the latest version of a ChannelManager and ChannelMonitors available to
8503 /// you. If you deserialize an old ChannelManager (during which force-closure transactions may be
8504 /// broadcast), and then later deserialize a newer version of the same ChannelManager (which will
8505 /// not force-close the same channels but consider them live), you may end up revoking a state for
8506 /// which you've already broadcasted the transaction.
8507 ///
8508 /// [`ChainMonitor`]: crate::chain::chainmonitor::ChainMonitor
8509 pub struct ChannelManagerReadArgs<'a, M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref>
8510 where
8511         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
8512         T::Target: BroadcasterInterface,
8513         ES::Target: EntropySource,
8514         NS::Target: NodeSigner,
8515         SP::Target: SignerProvider,
8516         F::Target: FeeEstimator,
8517         R::Target: Router,
8518         L::Target: Logger,
8519 {
8520         /// A cryptographically secure source of entropy.
8521         pub entropy_source: ES,
8522
8523         /// A signer that is able to perform node-scoped cryptographic operations.
8524         pub node_signer: NS,
8525
8526         /// The keys provider which will give us relevant keys. Some keys will be loaded during
8527         /// deserialization and KeysInterface::read_chan_signer will be used to read per-Channel
8528         /// signing data.
8529         pub signer_provider: SP,
8530
8531         /// The fee_estimator for use in the ChannelManager in the future.
8532         ///
8533         /// No calls to the FeeEstimator will be made during deserialization.
8534         pub fee_estimator: F,
8535         /// The chain::Watch for use in the ChannelManager in the future.
8536         ///
8537         /// No calls to the chain::Watch will be made during deserialization. It is assumed that
8538         /// you have deserialized ChannelMonitors separately and will add them to your
8539         /// chain::Watch after deserializing this ChannelManager.
8540         pub chain_monitor: M,
8541
8542         /// The BroadcasterInterface which will be used in the ChannelManager in the future and may be
8543         /// used to broadcast the latest local commitment transactions of channels which must be
8544         /// force-closed during deserialization.
8545         pub tx_broadcaster: T,
8546         /// The router which will be used in the ChannelManager in the future for finding routes
8547         /// on-the-fly for trampoline payments. Absent in private nodes that don't support forwarding.
8548         ///
8549         /// No calls to the router will be made during deserialization.
8550         pub router: R,
8551         /// The Logger for use in the ChannelManager and which may be used to log information during
8552         /// deserialization.
8553         pub logger: L,
8554         /// Default settings used for new channels. Any existing channels will continue to use the
8555         /// runtime settings which were stored when the ChannelManager was serialized.
8556         pub default_config: UserConfig,
8557
8558         /// A map from channel funding outpoints to ChannelMonitors for those channels (ie
8559         /// value.context.get_funding_txo() should be the key).
8560         ///
8561         /// If a monitor is inconsistent with the channel state during deserialization the channel will
8562         /// be force-closed using the data in the ChannelMonitor and the channel will be dropped. This
8563         /// is true for missing channels as well. If there is a monitor missing for which we find
8564         /// channel data Err(DecodeError::InvalidValue) will be returned.
8565         ///
8566         /// In such cases the latest local transactions will be sent to the tx_broadcaster included in
8567         /// this struct.
8568         ///
8569         /// This is not exported to bindings users because we have no HashMap bindings
8570         pub channel_monitors: HashMap<OutPoint, &'a mut ChannelMonitor<<SP::Target as SignerProvider>::Signer>>,
8571 }
8572
8573 impl<'a, M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref>
8574                 ChannelManagerReadArgs<'a, M, T, ES, NS, SP, F, R, L>
8575 where
8576         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
8577         T::Target: BroadcasterInterface,
8578         ES::Target: EntropySource,
8579         NS::Target: NodeSigner,
8580         SP::Target: SignerProvider,
8581         F::Target: FeeEstimator,
8582         R::Target: Router,
8583         L::Target: Logger,
8584 {
8585         /// Simple utility function to create a ChannelManagerReadArgs which creates the monitor
8586         /// HashMap for you. This is primarily useful for C bindings where it is not practical to
8587         /// populate a HashMap directly from C.
8588         pub fn new(entropy_source: ES, node_signer: NS, signer_provider: SP, fee_estimator: F, chain_monitor: M, tx_broadcaster: T, router: R, logger: L, default_config: UserConfig,
8589                         mut channel_monitors: Vec<&'a mut ChannelMonitor<<SP::Target as SignerProvider>::Signer>>) -> Self {
8590                 Self {
8591                         entropy_source, node_signer, signer_provider, fee_estimator, chain_monitor, tx_broadcaster, router, logger, default_config,
8592                         channel_monitors: channel_monitors.drain(..).map(|monitor| { (monitor.get_funding_txo().0, monitor) }).collect()
8593                 }
8594         }
8595 }
8596
8597 // Implement ReadableArgs for an Arc'd ChannelManager to make it a bit easier to work with the
8598 // SipmleArcChannelManager type:
8599 impl<'a, M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref>
8600         ReadableArgs<ChannelManagerReadArgs<'a, M, T, ES, NS, SP, F, R, L>> for (BlockHash, Arc<ChannelManager<M, T, ES, NS, SP, F, R, L>>)
8601 where
8602         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
8603         T::Target: BroadcasterInterface,
8604         ES::Target: EntropySource,
8605         NS::Target: NodeSigner,
8606         SP::Target: SignerProvider,
8607         F::Target: FeeEstimator,
8608         R::Target: Router,
8609         L::Target: Logger,
8610 {
8611         fn read<Reader: io::Read>(reader: &mut Reader, args: ChannelManagerReadArgs<'a, M, T, ES, NS, SP, F, R, L>) -> Result<Self, DecodeError> {
8612                 let (blockhash, chan_manager) = <(BlockHash, ChannelManager<M, T, ES, NS, SP, F, R, L>)>::read(reader, args)?;
8613                 Ok((blockhash, Arc::new(chan_manager)))
8614         }
8615 }
8616
8617 impl<'a, M: Deref, T: Deref, ES: Deref, NS: Deref, SP: Deref, F: Deref, R: Deref, L: Deref>
8618         ReadableArgs<ChannelManagerReadArgs<'a, M, T, ES, NS, SP, F, R, L>> for (BlockHash, ChannelManager<M, T, ES, NS, SP, F, R, L>)
8619 where
8620         M::Target: chain::Watch<<SP::Target as SignerProvider>::Signer>,
8621         T::Target: BroadcasterInterface,
8622         ES::Target: EntropySource,
8623         NS::Target: NodeSigner,
8624         SP::Target: SignerProvider,
8625         F::Target: FeeEstimator,
8626         R::Target: Router,
8627         L::Target: Logger,
8628 {
8629         fn read<Reader: io::Read>(reader: &mut Reader, mut args: ChannelManagerReadArgs<'a, M, T, ES, NS, SP, F, R, L>) -> Result<Self, DecodeError> {
8630                 let _ver = read_ver_prefix!(reader, SERIALIZATION_VERSION);
8631
8632                 let genesis_hash: BlockHash = Readable::read(reader)?;
8633                 let best_block_height: u32 = Readable::read(reader)?;
8634                 let best_block_hash: BlockHash = Readable::read(reader)?;
8635
8636                 let mut failed_htlcs = Vec::new();
8637
8638                 let channel_count: u64 = Readable::read(reader)?;
8639                 let mut funding_txo_set = HashSet::with_capacity(cmp::min(channel_count as usize, 128));
8640                 let mut peer_channels: HashMap<PublicKey, HashMap<ChannelId, Channel<SP>>> = HashMap::with_capacity(cmp::min(channel_count as usize, 128));
8641                 let mut id_to_peer = HashMap::with_capacity(cmp::min(channel_count as usize, 128));
8642                 let mut short_to_chan_info = HashMap::with_capacity(cmp::min(channel_count as usize, 128));
8643                 let mut channel_closures = VecDeque::new();
8644                 let mut close_background_events = Vec::new();
8645                 for _ in 0..channel_count {
8646                         let mut channel: Channel<SP> = Channel::read(reader, (
8647                                 &args.entropy_source, &args.signer_provider, best_block_height, &provided_channel_type_features(&args.default_config)
8648                         ))?;
8649                         let funding_txo = channel.context.get_funding_txo().ok_or(DecodeError::InvalidValue)?;
8650                         funding_txo_set.insert(funding_txo.clone());
8651                         if let Some(ref mut monitor) = args.channel_monitors.get_mut(&funding_txo) {
8652                                 if channel.get_cur_holder_commitment_transaction_number() > monitor.get_cur_holder_commitment_number() ||
8653                                                 channel.get_revoked_counterparty_commitment_transaction_number() > monitor.get_min_seen_secret() ||
8654                                                 channel.get_cur_counterparty_commitment_transaction_number() > monitor.get_cur_counterparty_commitment_number() ||
8655                                                 channel.context.get_latest_monitor_update_id() < monitor.get_latest_update_id() {
8656                                         // But if the channel is behind of the monitor, close the channel:
8657                                         log_error!(args.logger, "A ChannelManager is stale compared to the current ChannelMonitor!");
8658                                         log_error!(args.logger, " The channel will be force-closed and the latest commitment transaction from the ChannelMonitor broadcast.");
8659                                         log_error!(args.logger, " The ChannelMonitor for channel {} is at update_id {} but the ChannelManager is at update_id {}.",
8660                                                 &channel.context.channel_id(), monitor.get_latest_update_id(), channel.context.get_latest_monitor_update_id());
8661                                         let (monitor_update, mut new_failed_htlcs) = channel.context.force_shutdown(true);
8662                                         if let Some((counterparty_node_id, funding_txo, update)) = monitor_update {
8663                                                 close_background_events.push(BackgroundEvent::MonitorUpdateRegeneratedOnStartup {
8664                                                         counterparty_node_id, funding_txo, update
8665                                                 });
8666                                         }
8667                                         failed_htlcs.append(&mut new_failed_htlcs);
8668                                         channel_closures.push_back((events::Event::ChannelClosed {
8669                                                 channel_id: channel.context.channel_id(),
8670                                                 user_channel_id: channel.context.get_user_id(),
8671                                                 reason: ClosureReason::OutdatedChannelManager,
8672                                                 counterparty_node_id: Some(channel.context.get_counterparty_node_id()),
8673                                                 channel_capacity_sats: Some(channel.context.get_value_satoshis()),
8674                                         }, None));
8675                                         for (channel_htlc_source, payment_hash) in channel.inflight_htlc_sources() {
8676                                                 let mut found_htlc = false;
8677                                                 for (monitor_htlc_source, _) in monitor.get_all_current_outbound_htlcs() {
8678                                                         if *channel_htlc_source == monitor_htlc_source { found_htlc = true; break; }
8679                                                 }
8680                                                 if !found_htlc {
8681                                                         // If we have some HTLCs in the channel which are not present in the newer
8682                                                         // ChannelMonitor, they have been removed and should be failed back to
8683                                                         // ensure we don't forget them entirely. Note that if the missing HTLC(s)
8684                                                         // were actually claimed we'd have generated and ensured the previous-hop
8685                                                         // claim update ChannelMonitor updates were persisted prior to persising
8686                                                         // the ChannelMonitor update for the forward leg, so attempting to fail the
8687                                                         // backwards leg of the HTLC will simply be rejected.
8688                                                         log_info!(args.logger,
8689                                                                 "Failing HTLC with hash {} as it is missing in the ChannelMonitor for channel {} but was present in the (stale) ChannelManager",
8690                                                                 &channel.context.channel_id(), &payment_hash);
8691                                                         failed_htlcs.push((channel_htlc_source.clone(), *payment_hash, channel.context.get_counterparty_node_id(), channel.context.channel_id()));
8692                                                 }
8693                                         }
8694                                 } else {
8695                                         log_info!(args.logger, "Successfully loaded channel {} at update_id {} against monitor at update id {}",
8696                                                 &channel.context.channel_id(), channel.context.get_latest_monitor_update_id(),
8697                                                 monitor.get_latest_update_id());
8698                                         if let Some(short_channel_id) = channel.context.get_short_channel_id() {
8699                                                 short_to_chan_info.insert(short_channel_id, (channel.context.get_counterparty_node_id(), channel.context.channel_id()));
8700                                         }
8701                                         if channel.context.is_funding_initiated() {
8702                                                 id_to_peer.insert(channel.context.channel_id(), channel.context.get_counterparty_node_id());
8703                                         }
8704                                         match peer_channels.entry(channel.context.get_counterparty_node_id()) {
8705                                                 hash_map::Entry::Occupied(mut entry) => {
8706                                                         let by_id_map = entry.get_mut();
8707                                                         by_id_map.insert(channel.context.channel_id(), channel);
8708                                                 },
8709                                                 hash_map::Entry::Vacant(entry) => {
8710                                                         let mut by_id_map = HashMap::new();
8711                                                         by_id_map.insert(channel.context.channel_id(), channel);
8712                                                         entry.insert(by_id_map);
8713                                                 }
8714                                         }
8715                                 }
8716                         } else if channel.is_awaiting_initial_mon_persist() {
8717                                 // If we were persisted and shut down while the initial ChannelMonitor persistence
8718                                 // was in-progress, we never broadcasted the funding transaction and can still
8719                                 // safely discard the channel.
8720                                 let _ = channel.context.force_shutdown(false);
8721                                 channel_closures.push_back((events::Event::ChannelClosed {
8722                                         channel_id: channel.context.channel_id(),
8723                                         user_channel_id: channel.context.get_user_id(),
8724                                         reason: ClosureReason::DisconnectedPeer,
8725                                         counterparty_node_id: Some(channel.context.get_counterparty_node_id()),
8726                                         channel_capacity_sats: Some(channel.context.get_value_satoshis()),
8727                                 }, None));
8728                         } else {
8729                                 log_error!(args.logger, "Missing ChannelMonitor for channel {} needed by ChannelManager.", &channel.context.channel_id());
8730                                 log_error!(args.logger, " The chain::Watch API *requires* that monitors are persisted durably before returning,");
8731                                 log_error!(args.logger, " client applications must ensure that ChannelMonitor data is always available and the latest to avoid funds loss!");
8732                                 log_error!(args.logger, " Without the ChannelMonitor we cannot continue without risking funds.");
8733                                 log_error!(args.logger, " Please ensure the chain::Watch API requirements are met and file a bug report at https://github.com/lightningdevkit/rust-lightning");
8734                                 return Err(DecodeError::InvalidValue);
8735                         }
8736                 }
8737
8738                 for (funding_txo, _) in args.channel_monitors.iter() {
8739                         if !funding_txo_set.contains(funding_txo) {
8740                                 log_info!(args.logger, "Queueing monitor update to ensure missing channel {} is force closed",
8741                                         &funding_txo.to_channel_id());
8742                                 let monitor_update = ChannelMonitorUpdate {
8743                                         update_id: CLOSED_CHANNEL_UPDATE_ID,
8744                                         updates: vec![ChannelMonitorUpdateStep::ChannelForceClosed { should_broadcast: true }],
8745                                 };
8746                                 close_background_events.push(BackgroundEvent::ClosedMonitorUpdateRegeneratedOnStartup((*funding_txo, monitor_update)));
8747                         }
8748                 }
8749
8750                 const MAX_ALLOC_SIZE: usize = 1024 * 64;
8751                 let forward_htlcs_count: u64 = Readable::read(reader)?;
8752                 let mut forward_htlcs = HashMap::with_capacity(cmp::min(forward_htlcs_count as usize, 128));
8753                 for _ in 0..forward_htlcs_count {
8754                         let short_channel_id = Readable::read(reader)?;
8755                         let pending_forwards_count: u64 = Readable::read(reader)?;
8756                         let mut pending_forwards = Vec::with_capacity(cmp::min(pending_forwards_count as usize, MAX_ALLOC_SIZE/mem::size_of::<HTLCForwardInfo>()));
8757                         for _ in 0..pending_forwards_count {
8758                                 pending_forwards.push(Readable::read(reader)?);
8759                         }
8760                         forward_htlcs.insert(short_channel_id, pending_forwards);
8761                 }
8762
8763                 let claimable_htlcs_count: u64 = Readable::read(reader)?;
8764                 let mut claimable_htlcs_list = Vec::with_capacity(cmp::min(claimable_htlcs_count as usize, 128));
8765                 for _ in 0..claimable_htlcs_count {
8766                         let payment_hash = Readable::read(reader)?;
8767                         let previous_hops_len: u64 = Readable::read(reader)?;
8768                         let mut previous_hops = Vec::with_capacity(cmp::min(previous_hops_len as usize, MAX_ALLOC_SIZE/mem::size_of::<ClaimableHTLC>()));
8769                         for _ in 0..previous_hops_len {
8770                                 previous_hops.push(<ClaimableHTLC as Readable>::read(reader)?);
8771                         }
8772                         claimable_htlcs_list.push((payment_hash, previous_hops));
8773                 }
8774
8775                 let peer_state_from_chans = |channel_by_id| {
8776                         PeerState {
8777                                 channel_by_id,
8778                                 outbound_v1_channel_by_id: HashMap::new(),
8779                                 inbound_v1_channel_by_id: HashMap::new(),
8780                                 inbound_channel_request_by_id: HashMap::new(),
8781                                 latest_features: InitFeatures::empty(),
8782                                 pending_msg_events: Vec::new(),
8783                                 in_flight_monitor_updates: BTreeMap::new(),
8784                                 monitor_update_blocked_actions: BTreeMap::new(),
8785                                 actions_blocking_raa_monitor_updates: BTreeMap::new(),
8786                                 is_connected: false,
8787                         }
8788                 };
8789
8790                 let peer_count: u64 = Readable::read(reader)?;
8791                 let mut per_peer_state = HashMap::with_capacity(cmp::min(peer_count as usize, MAX_ALLOC_SIZE/mem::size_of::<(PublicKey, Mutex<PeerState<SP>>)>()));
8792                 for _ in 0..peer_count {
8793                         let peer_pubkey = Readable::read(reader)?;
8794                         let peer_chans = peer_channels.remove(&peer_pubkey).unwrap_or(HashMap::new());
8795                         let mut peer_state = peer_state_from_chans(peer_chans);
8796                         peer_state.latest_features = Readable::read(reader)?;
8797                         per_peer_state.insert(peer_pubkey, Mutex::new(peer_state));
8798                 }
8799
8800                 let event_count: u64 = Readable::read(reader)?;
8801                 let mut pending_events_read: VecDeque<(events::Event, Option<EventCompletionAction>)> =
8802                         VecDeque::with_capacity(cmp::min(event_count as usize, MAX_ALLOC_SIZE/mem::size_of::<(events::Event, Option<EventCompletionAction>)>()));
8803                 for _ in 0..event_count {
8804                         match MaybeReadable::read(reader)? {
8805                                 Some(event) => pending_events_read.push_back((event, None)),
8806                                 None => continue,
8807                         }
8808                 }
8809
8810                 let background_event_count: u64 = Readable::read(reader)?;
8811                 for _ in 0..background_event_count {
8812                         match <u8 as Readable>::read(reader)? {
8813                                 0 => {
8814                                         // LDK versions prior to 0.0.116 wrote pending `MonitorUpdateRegeneratedOnStartup`s here,
8815                                         // however we really don't (and never did) need them - we regenerate all
8816                                         // on-startup monitor updates.
8817                                         let _: OutPoint = Readable::read(reader)?;
8818                                         let _: ChannelMonitorUpdate = Readable::read(reader)?;
8819                                 }
8820                                 _ => return Err(DecodeError::InvalidValue),
8821                         }
8822                 }
8823
8824                 let _last_node_announcement_serial: u32 = Readable::read(reader)?; // Only used < 0.0.111
8825                 let highest_seen_timestamp: u32 = Readable::read(reader)?;
8826
8827                 let pending_inbound_payment_count: u64 = Readable::read(reader)?;
8828                 let mut pending_inbound_payments: HashMap<PaymentHash, PendingInboundPayment> = HashMap::with_capacity(cmp::min(pending_inbound_payment_count as usize, MAX_ALLOC_SIZE/(3*32)));
8829                 for _ in 0..pending_inbound_payment_count {
8830                         if pending_inbound_payments.insert(Readable::read(reader)?, Readable::read(reader)?).is_some() {
8831                                 return Err(DecodeError::InvalidValue);
8832                         }
8833                 }
8834
8835                 let pending_outbound_payments_count_compat: u64 = Readable::read(reader)?;
8836                 let mut pending_outbound_payments_compat: HashMap<PaymentId, PendingOutboundPayment> =
8837                         HashMap::with_capacity(cmp::min(pending_outbound_payments_count_compat as usize, MAX_ALLOC_SIZE/32));
8838                 for _ in 0..pending_outbound_payments_count_compat {
8839                         let session_priv = Readable::read(reader)?;
8840                         let payment = PendingOutboundPayment::Legacy {
8841                                 session_privs: [session_priv].iter().cloned().collect()
8842                         };
8843                         if pending_outbound_payments_compat.insert(PaymentId(session_priv), payment).is_some() {
8844                                 return Err(DecodeError::InvalidValue)
8845                         };
8846                 }
8847
8848                 // pending_outbound_payments_no_retry is for compatibility with 0.0.101 clients.
8849                 let mut pending_outbound_payments_no_retry: Option<HashMap<PaymentId, HashSet<[u8; 32]>>> = None;
8850                 let mut pending_outbound_payments = None;
8851                 let mut pending_intercepted_htlcs: Option<HashMap<InterceptId, PendingAddHTLCInfo>> = Some(HashMap::new());
8852                 let mut received_network_pubkey: Option<PublicKey> = None;
8853                 let mut fake_scid_rand_bytes: Option<[u8; 32]> = None;
8854                 let mut probing_cookie_secret: Option<[u8; 32]> = None;
8855                 let mut claimable_htlc_purposes = None;
8856                 let mut claimable_htlc_onion_fields = None;
8857                 let mut pending_claiming_payments = Some(HashMap::new());
8858                 let mut monitor_update_blocked_actions_per_peer: Option<Vec<(_, BTreeMap<_, Vec<_>>)>> = Some(Vec::new());
8859                 let mut events_override = None;
8860                 let mut in_flight_monitor_updates: Option<HashMap<(PublicKey, OutPoint), Vec<ChannelMonitorUpdate>>> = None;
8861                 read_tlv_fields!(reader, {
8862                         (1, pending_outbound_payments_no_retry, option),
8863                         (2, pending_intercepted_htlcs, option),
8864                         (3, pending_outbound_payments, option),
8865                         (4, pending_claiming_payments, option),
8866                         (5, received_network_pubkey, option),
8867                         (6, monitor_update_blocked_actions_per_peer, option),
8868                         (7, fake_scid_rand_bytes, option),
8869                         (8, events_override, option),
8870                         (9, claimable_htlc_purposes, optional_vec),
8871                         (10, in_flight_monitor_updates, option),
8872                         (11, probing_cookie_secret, option),
8873                         (13, claimable_htlc_onion_fields, optional_vec),
8874                 });
8875                 if fake_scid_rand_bytes.is_none() {
8876                         fake_scid_rand_bytes = Some(args.entropy_source.get_secure_random_bytes());
8877                 }
8878
8879                 if probing_cookie_secret.is_none() {
8880                         probing_cookie_secret = Some(args.entropy_source.get_secure_random_bytes());
8881                 }
8882
8883                 if let Some(events) = events_override {
8884                         pending_events_read = events;
8885                 }
8886
8887                 if !channel_closures.is_empty() {
8888                         pending_events_read.append(&mut channel_closures);
8889                 }
8890
8891                 if pending_outbound_payments.is_none() && pending_outbound_payments_no_retry.is_none() {
8892                         pending_outbound_payments = Some(pending_outbound_payments_compat);
8893                 } else if pending_outbound_payments.is_none() {
8894                         let mut outbounds = HashMap::new();
8895                         for (id, session_privs) in pending_outbound_payments_no_retry.unwrap().drain() {
8896                                 outbounds.insert(id, PendingOutboundPayment::Legacy { session_privs });
8897                         }
8898                         pending_outbound_payments = Some(outbounds);
8899                 }
8900                 let pending_outbounds = OutboundPayments {
8901                         pending_outbound_payments: Mutex::new(pending_outbound_payments.unwrap()),
8902                         retry_lock: Mutex::new(())
8903                 };
8904
8905                 // We have to replay (or skip, if they were completed after we wrote the `ChannelManager`)
8906                 // each `ChannelMonitorUpdate` in `in_flight_monitor_updates`. After doing so, we have to
8907                 // check that each channel we have isn't newer than the latest `ChannelMonitorUpdate`(s) we
8908                 // replayed, and for each monitor update we have to replay we have to ensure there's a
8909                 // `ChannelMonitor` for it.
8910                 //
8911                 // In order to do so we first walk all of our live channels (so that we can check their
8912                 // state immediately after doing the update replays, when we have the `update_id`s
8913                 // available) and then walk any remaining in-flight updates.
8914                 //
8915                 // Because the actual handling of the in-flight updates is the same, it's macro'ized here:
8916                 let mut pending_background_events = Vec::new();
8917                 macro_rules! handle_in_flight_updates {
8918                         ($counterparty_node_id: expr, $chan_in_flight_upds: expr, $funding_txo: expr,
8919                          $monitor: expr, $peer_state: expr, $channel_info_log: expr
8920                         ) => { {
8921                                 let mut max_in_flight_update_id = 0;
8922                                 $chan_in_flight_upds.retain(|upd| upd.update_id > $monitor.get_latest_update_id());
8923                                 for update in $chan_in_flight_upds.iter() {
8924                                         log_trace!(args.logger, "Replaying ChannelMonitorUpdate {} for {}channel {}",
8925                                                 update.update_id, $channel_info_log, &$funding_txo.to_channel_id());
8926                                         max_in_flight_update_id = cmp::max(max_in_flight_update_id, update.update_id);
8927                                         pending_background_events.push(
8928                                                 BackgroundEvent::MonitorUpdateRegeneratedOnStartup {
8929                                                         counterparty_node_id: $counterparty_node_id,
8930                                                         funding_txo: $funding_txo,
8931                                                         update: update.clone(),
8932                                                 });
8933                                 }
8934                                 if $chan_in_flight_upds.is_empty() {
8935                                         // We had some updates to apply, but it turns out they had completed before we
8936                                         // were serialized, we just weren't notified of that. Thus, we may have to run
8937                                         // the completion actions for any monitor updates, but otherwise are done.
8938                                         pending_background_events.push(
8939                                                 BackgroundEvent::MonitorUpdatesComplete {
8940                                                         counterparty_node_id: $counterparty_node_id,
8941                                                         channel_id: $funding_txo.to_channel_id(),
8942                                                 });
8943                                 }
8944                                 if $peer_state.in_flight_monitor_updates.insert($funding_txo, $chan_in_flight_upds).is_some() {
8945                                         log_error!(args.logger, "Duplicate in-flight monitor update set for the same channel!");
8946                                         return Err(DecodeError::InvalidValue);
8947                                 }
8948                                 max_in_flight_update_id
8949                         } }
8950                 }
8951
8952                 for (counterparty_id, peer_state_mtx) in per_peer_state.iter_mut() {
8953                         let mut peer_state_lock = peer_state_mtx.lock().unwrap();
8954                         let peer_state = &mut *peer_state_lock;
8955                         for (_, chan) in peer_state.channel_by_id.iter() {
8956                                 // Channels that were persisted have to be funded, otherwise they should have been
8957                                 // discarded.
8958                                 let funding_txo = chan.context.get_funding_txo().ok_or(DecodeError::InvalidValue)?;
8959                                 let monitor = args.channel_monitors.get(&funding_txo)
8960                                         .expect("We already checked for monitor presence when loading channels");
8961                                 let mut max_in_flight_update_id = monitor.get_latest_update_id();
8962                                 if let Some(in_flight_upds) = &mut in_flight_monitor_updates {
8963                                         if let Some(mut chan_in_flight_upds) = in_flight_upds.remove(&(*counterparty_id, funding_txo)) {
8964                                                 max_in_flight_update_id = cmp::max(max_in_flight_update_id,
8965                                                         handle_in_flight_updates!(*counterparty_id, chan_in_flight_upds,
8966                                                                 funding_txo, monitor, peer_state, ""));
8967                                         }
8968                                 }
8969                                 if chan.get_latest_unblocked_monitor_update_id() > max_in_flight_update_id {
8970                                         // If the channel is ahead of the monitor, return InvalidValue:
8971                                         log_error!(args.logger, "A ChannelMonitor is stale compared to the current ChannelManager! This indicates a potentially-critical violation of the chain::Watch API!");
8972                                         log_error!(args.logger, " The ChannelMonitor for channel {} is at update_id {} with update_id through {} in-flight",
8973                                                 &chan.context.channel_id(), monitor.get_latest_update_id(), max_in_flight_update_id);
8974                                         log_error!(args.logger, " but the ChannelManager is at update_id {}.", chan.get_latest_unblocked_monitor_update_id());
8975                                         log_error!(args.logger, " The chain::Watch API *requires* that monitors are persisted durably before returning,");
8976                                         log_error!(args.logger, " client applications must ensure that ChannelMonitor data is always available and the latest to avoid funds loss!");
8977                                         log_error!(args.logger, " Without the latest ChannelMonitor we cannot continue without risking funds.");
8978                                         log_error!(args.logger, " Please ensure the chain::Watch API requirements are met and file a bug report at https://github.com/lightningdevkit/rust-lightning");
8979                                         return Err(DecodeError::InvalidValue);
8980                                 }
8981                         }
8982                 }
8983
8984                 if let Some(in_flight_upds) = in_flight_monitor_updates {
8985                         for ((counterparty_id, funding_txo), mut chan_in_flight_updates) in in_flight_upds {
8986                                 if let Some(monitor) = args.channel_monitors.get(&funding_txo) {
8987                                         // Now that we've removed all the in-flight monitor updates for channels that are
8988                                         // still open, we need to replay any monitor updates that are for closed channels,
8989                                         // creating the neccessary peer_state entries as we go.
8990                                         let peer_state_mutex = per_peer_state.entry(counterparty_id).or_insert_with(|| {
8991                                                 Mutex::new(peer_state_from_chans(HashMap::new()))
8992                                         });
8993                                         let mut peer_state = peer_state_mutex.lock().unwrap();
8994                                         handle_in_flight_updates!(counterparty_id, chan_in_flight_updates,
8995                                                 funding_txo, monitor, peer_state, "closed ");
8996                                 } else {
8997                                         log_error!(args.logger, "A ChannelMonitor is missing even though we have in-flight updates for it! This indicates a potentially-critical violation of the chain::Watch API!");
8998                                         log_error!(args.logger, " The ChannelMonitor for channel {} is missing.",
8999                                                 &funding_txo.to_channel_id());
9000                                         log_error!(args.logger, " The chain::Watch API *requires* that monitors are persisted durably before returning,");
9001                                         log_error!(args.logger, " client applications must ensure that ChannelMonitor data is always available and the latest to avoid funds loss!");
9002                                         log_error!(args.logger, " Without the latest ChannelMonitor we cannot continue without risking funds.");
9003                                         log_error!(args.logger, " Please ensure the chain::Watch API requirements are met and file a bug report at https://github.com/lightningdevkit/rust-lightning");
9004                                         return Err(DecodeError::InvalidValue);
9005                                 }
9006                         }
9007                 }
9008
9009                 // Note that we have to do the above replays before we push new monitor updates.
9010                 pending_background_events.append(&mut close_background_events);
9011
9012                 // If there's any preimages for forwarded HTLCs hanging around in ChannelMonitors we
9013                 // should ensure we try them again on the inbound edge. We put them here and do so after we
9014                 // have a fully-constructed `ChannelManager` at the end.
9015                 let mut pending_claims_to_replay = Vec::new();
9016
9017                 {
9018                         // If we're tracking pending payments, ensure we haven't lost any by looking at the
9019                         // ChannelMonitor data for any channels for which we do not have authorative state
9020                         // (i.e. those for which we just force-closed above or we otherwise don't have a
9021                         // corresponding `Channel` at all).
9022                         // This avoids several edge-cases where we would otherwise "forget" about pending
9023                         // payments which are still in-flight via their on-chain state.
9024                         // We only rebuild the pending payments map if we were most recently serialized by
9025                         // 0.0.102+
9026                         for (_, monitor) in args.channel_monitors.iter() {
9027                                 let counterparty_opt = id_to_peer.get(&monitor.get_funding_txo().0.to_channel_id());
9028                                 if counterparty_opt.is_none() {
9029                                         for (htlc_source, (htlc, _)) in monitor.get_pending_or_resolved_outbound_htlcs() {
9030                                                 if let HTLCSource::OutboundRoute { payment_id, session_priv, path, .. } = htlc_source {
9031                                                         if path.hops.is_empty() {
9032                                                                 log_error!(args.logger, "Got an empty path for a pending payment");
9033                                                                 return Err(DecodeError::InvalidValue);
9034                                                         }
9035
9036                                                         let path_amt = path.final_value_msat();
9037                                                         let mut session_priv_bytes = [0; 32];
9038                                                         session_priv_bytes[..].copy_from_slice(&session_priv[..]);
9039                                                         match pending_outbounds.pending_outbound_payments.lock().unwrap().entry(payment_id) {
9040                                                                 hash_map::Entry::Occupied(mut entry) => {
9041                                                                         let newly_added = entry.get_mut().insert(session_priv_bytes, &path);
9042                                                                         log_info!(args.logger, "{} a pending payment path for {} msat for session priv {} on an existing pending payment with payment hash {}",
9043                                                                                 if newly_added { "Added" } else { "Had" }, path_amt, log_bytes!(session_priv_bytes), &htlc.payment_hash);
9044                                                                 },
9045                                                                 hash_map::Entry::Vacant(entry) => {
9046                                                                         let path_fee = path.fee_msat();
9047                                                                         entry.insert(PendingOutboundPayment::Retryable {
9048                                                                                 retry_strategy: None,
9049                                                                                 attempts: PaymentAttempts::new(),
9050                                                                                 payment_params: None,
9051                                                                                 session_privs: [session_priv_bytes].iter().map(|a| *a).collect(),
9052                                                                                 payment_hash: htlc.payment_hash,
9053                                                                                 payment_secret: None, // only used for retries, and we'll never retry on startup
9054                                                                                 payment_metadata: None, // only used for retries, and we'll never retry on startup
9055                                                                                 keysend_preimage: None, // only used for retries, and we'll never retry on startup
9056                                                                                 custom_tlvs: Vec::new(), // only used for retries, and we'll never retry on startup
9057                                                                                 pending_amt_msat: path_amt,
9058                                                                                 pending_fee_msat: Some(path_fee),
9059                                                                                 total_msat: path_amt,
9060                                                                                 starting_block_height: best_block_height,
9061                                                                         });
9062                                                                         log_info!(args.logger, "Added a pending payment for {} msat with payment hash {} for path with session priv {}",
9063                                                                                 path_amt, &htlc.payment_hash,  log_bytes!(session_priv_bytes));
9064                                                                 }
9065                                                         }
9066                                                 }
9067                                         }
9068                                         for (htlc_source, (htlc, preimage_opt)) in monitor.get_all_current_outbound_htlcs() {
9069                                                 match htlc_source {
9070                                                         HTLCSource::PreviousHopData(prev_hop_data) => {
9071                                                                 let pending_forward_matches_htlc = |info: &PendingAddHTLCInfo| {
9072                                                                         info.prev_funding_outpoint == prev_hop_data.outpoint &&
9073                                                                                 info.prev_htlc_id == prev_hop_data.htlc_id
9074                                                                 };
9075                                                                 // The ChannelMonitor is now responsible for this HTLC's
9076                                                                 // failure/success and will let us know what its outcome is. If we
9077                                                                 // still have an entry for this HTLC in `forward_htlcs` or
9078                                                                 // `pending_intercepted_htlcs`, we were apparently not persisted after
9079                                                                 // the monitor was when forwarding the payment.
9080                                                                 forward_htlcs.retain(|_, forwards| {
9081                                                                         forwards.retain(|forward| {
9082                                                                                 if let HTLCForwardInfo::AddHTLC(htlc_info) = forward {
9083                                                                                         if pending_forward_matches_htlc(&htlc_info) {
9084                                                                                                 log_info!(args.logger, "Removing pending to-forward HTLC with hash {} as it was forwarded to the closed channel {}",
9085                                                                                                         &htlc.payment_hash, &monitor.get_funding_txo().0.to_channel_id());
9086                                                                                                 false
9087                                                                                         } else { true }
9088                                                                                 } else { true }
9089                                                                         });
9090                                                                         !forwards.is_empty()
9091                                                                 });
9092                                                                 pending_intercepted_htlcs.as_mut().unwrap().retain(|intercepted_id, htlc_info| {
9093                                                                         if pending_forward_matches_htlc(&htlc_info) {
9094                                                                                 log_info!(args.logger, "Removing pending intercepted HTLC with hash {} as it was forwarded to the closed channel {}",
9095                                                                                         &htlc.payment_hash, &monitor.get_funding_txo().0.to_channel_id());
9096                                                                                 pending_events_read.retain(|(event, _)| {
9097                                                                                         if let Event::HTLCIntercepted { intercept_id: ev_id, .. } = event {
9098                                                                                                 intercepted_id != ev_id
9099                                                                                         } else { true }
9100                                                                                 });
9101                                                                                 false
9102                                                                         } else { true }
9103                                                                 });
9104                                                         },
9105                                                         HTLCSource::OutboundRoute { payment_id, session_priv, path, .. } => {
9106                                                                 if let Some(preimage) = preimage_opt {
9107                                                                         let pending_events = Mutex::new(pending_events_read);
9108                                                                         // Note that we set `from_onchain` to "false" here,
9109                                                                         // deliberately keeping the pending payment around forever.
9110                                                                         // Given it should only occur when we have a channel we're
9111                                                                         // force-closing for being stale that's okay.
9112                                                                         // The alternative would be to wipe the state when claiming,
9113                                                                         // generating a `PaymentPathSuccessful` event but regenerating
9114                                                                         // it and the `PaymentSent` on every restart until the
9115                                                                         // `ChannelMonitor` is removed.
9116                                                                         let compl_action =
9117                                                                                 EventCompletionAction::ReleaseRAAChannelMonitorUpdate {
9118                                                                                         channel_funding_outpoint: monitor.get_funding_txo().0,
9119                                                                                         counterparty_node_id: path.hops[0].pubkey,
9120                                                                                 };
9121                                                                         pending_outbounds.claim_htlc(payment_id, preimage, session_priv,
9122                                                                                 path, false, compl_action, &pending_events, &args.logger);
9123                                                                         pending_events_read = pending_events.into_inner().unwrap();
9124                                                                 }
9125                                                         },
9126                                                 }
9127                                         }
9128                                 }
9129
9130                                 // Whether the downstream channel was closed or not, try to re-apply any payment
9131                                 // preimages from it which may be needed in upstream channels for forwarded
9132                                 // payments.
9133                                 let outbound_claimed_htlcs_iter = monitor.get_all_current_outbound_htlcs()
9134                                         .into_iter()
9135                                         .filter_map(|(htlc_source, (htlc, preimage_opt))| {
9136                                                 if let HTLCSource::PreviousHopData(_) = htlc_source {
9137                                                         if let Some(payment_preimage) = preimage_opt {
9138                                                                 Some((htlc_source, payment_preimage, htlc.amount_msat,
9139                                                                         // Check if `counterparty_opt.is_none()` to see if the
9140                                                                         // downstream chan is closed (because we don't have a
9141                                                                         // channel_id -> peer map entry).
9142                                                                         counterparty_opt.is_none(),
9143                                                                         monitor.get_funding_txo().0))
9144                                                         } else { None }
9145                                                 } else {
9146                                                         // If it was an outbound payment, we've handled it above - if a preimage
9147                                                         // came in and we persisted the `ChannelManager` we either handled it and
9148                                                         // are good to go or the channel force-closed - we don't have to handle the
9149                                                         // channel still live case here.
9150                                                         None
9151                                                 }
9152                                         });
9153                                 for tuple in outbound_claimed_htlcs_iter {
9154                                         pending_claims_to_replay.push(tuple);
9155                                 }
9156                         }
9157                 }
9158
9159                 if !forward_htlcs.is_empty() || pending_outbounds.needs_abandon() {
9160                         // If we have pending HTLCs to forward, assume we either dropped a
9161                         // `PendingHTLCsForwardable` or the user received it but never processed it as they
9162                         // shut down before the timer hit. Either way, set the time_forwardable to a small
9163                         // constant as enough time has likely passed that we should simply handle the forwards
9164                         // now, or at least after the user gets a chance to reconnect to our peers.
9165                         pending_events_read.push_back((events::Event::PendingHTLCsForwardable {
9166                                 time_forwardable: Duration::from_secs(2),
9167                         }, None));
9168                 }
9169
9170                 let inbound_pmt_key_material = args.node_signer.get_inbound_payment_key_material();
9171                 let expanded_inbound_key = inbound_payment::ExpandedKey::new(&inbound_pmt_key_material);
9172
9173                 let mut claimable_payments = HashMap::with_capacity(claimable_htlcs_list.len());
9174                 if let Some(purposes) = claimable_htlc_purposes {
9175                         if purposes.len() != claimable_htlcs_list.len() {
9176                                 return Err(DecodeError::InvalidValue);
9177                         }
9178                         if let Some(onion_fields) = claimable_htlc_onion_fields {
9179                                 if onion_fields.len() != claimable_htlcs_list.len() {
9180                                         return Err(DecodeError::InvalidValue);
9181                                 }
9182                                 for (purpose, (onion, (payment_hash, htlcs))) in
9183                                         purposes.into_iter().zip(onion_fields.into_iter().zip(claimable_htlcs_list.into_iter()))
9184                                 {
9185                                         let existing_payment = claimable_payments.insert(payment_hash, ClaimablePayment {
9186                                                 purpose, htlcs, onion_fields: onion,
9187                                         });
9188                                         if existing_payment.is_some() { return Err(DecodeError::InvalidValue); }
9189                                 }
9190                         } else {
9191                                 for (purpose, (payment_hash, htlcs)) in purposes.into_iter().zip(claimable_htlcs_list.into_iter()) {
9192                                         let existing_payment = claimable_payments.insert(payment_hash, ClaimablePayment {
9193                                                 purpose, htlcs, onion_fields: None,
9194                                         });
9195                                         if existing_payment.is_some() { return Err(DecodeError::InvalidValue); }
9196                                 }
9197                         }
9198                 } else {
9199                         // LDK versions prior to 0.0.107 did not write a `pending_htlc_purposes`, but do
9200                         // include a `_legacy_hop_data` in the `OnionPayload`.
9201                         for (payment_hash, htlcs) in claimable_htlcs_list.drain(..) {
9202                                 if htlcs.is_empty() {
9203                                         return Err(DecodeError::InvalidValue);
9204                                 }
9205                                 let purpose = match &htlcs[0].onion_payload {
9206                                         OnionPayload::Invoice { _legacy_hop_data } => {
9207                                                 if let Some(hop_data) = _legacy_hop_data {
9208                                                         events::PaymentPurpose::InvoicePayment {
9209                                                                 payment_preimage: match pending_inbound_payments.get(&payment_hash) {
9210                                                                         Some(inbound_payment) => inbound_payment.payment_preimage,
9211                                                                         None => match inbound_payment::verify(payment_hash, &hop_data, 0, &expanded_inbound_key, &args.logger) {
9212                                                                                 Ok((payment_preimage, _)) => payment_preimage,
9213                                                                                 Err(()) => {
9214                                                                                         log_error!(args.logger, "Failed to read claimable payment data for HTLC with payment hash {} - was not a pending inbound payment and didn't match our payment key", &payment_hash);
9215                                                                                         return Err(DecodeError::InvalidValue);
9216                                                                                 }
9217                                                                         }
9218                                                                 },
9219                                                                 payment_secret: hop_data.payment_secret,
9220                                                         }
9221                                                 } else { return Err(DecodeError::InvalidValue); }
9222                                         },
9223                                         OnionPayload::Spontaneous(payment_preimage) =>
9224                                                 events::PaymentPurpose::SpontaneousPayment(*payment_preimage),
9225                                 };
9226                                 claimable_payments.insert(payment_hash, ClaimablePayment {
9227                                         purpose, htlcs, onion_fields: None,
9228                                 });
9229                         }
9230                 }
9231
9232                 let mut secp_ctx = Secp256k1::new();
9233                 secp_ctx.seeded_randomize(&args.entropy_source.get_secure_random_bytes());
9234
9235                 let our_network_pubkey = match args.node_signer.get_node_id(Recipient::Node) {
9236                         Ok(key) => key,
9237                         Err(()) => return Err(DecodeError::InvalidValue)
9238                 };
9239                 if let Some(network_pubkey) = received_network_pubkey {
9240                         if network_pubkey != our_network_pubkey {
9241                                 log_error!(args.logger, "Key that was generated does not match the existing key.");
9242                                 return Err(DecodeError::InvalidValue);
9243                         }
9244                 }
9245
9246                 let mut outbound_scid_aliases = HashSet::new();
9247                 for (_peer_node_id, peer_state_mutex) in per_peer_state.iter_mut() {
9248                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
9249                         let peer_state = &mut *peer_state_lock;
9250                         for (chan_id, chan) in peer_state.channel_by_id.iter_mut() {
9251                                 if chan.context.outbound_scid_alias() == 0 {
9252                                         let mut outbound_scid_alias;
9253                                         loop {
9254                                                 outbound_scid_alias = fake_scid::Namespace::OutboundAlias
9255                                                         .get_fake_scid(best_block_height, &genesis_hash, fake_scid_rand_bytes.as_ref().unwrap(), &args.entropy_source);
9256                                                 if outbound_scid_aliases.insert(outbound_scid_alias) { break; }
9257                                         }
9258                                         chan.context.set_outbound_scid_alias(outbound_scid_alias);
9259                                 } else if !outbound_scid_aliases.insert(chan.context.outbound_scid_alias()) {
9260                                         // Note that in rare cases its possible to hit this while reading an older
9261                                         // channel if we just happened to pick a colliding outbound alias above.
9262                                         log_error!(args.logger, "Got duplicate outbound SCID alias; {}", chan.context.outbound_scid_alias());
9263                                         return Err(DecodeError::InvalidValue);
9264                                 }
9265                                 if chan.context.is_usable() {
9266                                         if short_to_chan_info.insert(chan.context.outbound_scid_alias(), (chan.context.get_counterparty_node_id(), *chan_id)).is_some() {
9267                                                 // Note that in rare cases its possible to hit this while reading an older
9268                                                 // channel if we just happened to pick a colliding outbound alias above.
9269                                                 log_error!(args.logger, "Got duplicate outbound SCID alias; {}", chan.context.outbound_scid_alias());
9270                                                 return Err(DecodeError::InvalidValue);
9271                                         }
9272                                 }
9273                         }
9274                 }
9275
9276                 let bounded_fee_estimator = LowerBoundedFeeEstimator::new(args.fee_estimator);
9277
9278                 for (_, monitor) in args.channel_monitors.iter() {
9279                         for (payment_hash, payment_preimage) in monitor.get_stored_preimages() {
9280                                 if let Some(payment) = claimable_payments.remove(&payment_hash) {
9281                                         log_info!(args.logger, "Re-claiming HTLCs with payment hash {} as we've released the preimage to a ChannelMonitor!", &payment_hash);
9282                                         let mut claimable_amt_msat = 0;
9283                                         let mut receiver_node_id = Some(our_network_pubkey);
9284                                         let phantom_shared_secret = payment.htlcs[0].prev_hop.phantom_shared_secret;
9285                                         if phantom_shared_secret.is_some() {
9286                                                 let phantom_pubkey = args.node_signer.get_node_id(Recipient::PhantomNode)
9287                                                         .expect("Failed to get node_id for phantom node recipient");
9288                                                 receiver_node_id = Some(phantom_pubkey)
9289                                         }
9290                                         for claimable_htlc in &payment.htlcs {
9291                                                 claimable_amt_msat += claimable_htlc.value;
9292
9293                                                 // Add a holding-cell claim of the payment to the Channel, which should be
9294                                                 // applied ~immediately on peer reconnection. Because it won't generate a
9295                                                 // new commitment transaction we can just provide the payment preimage to
9296                                                 // the corresponding ChannelMonitor and nothing else.
9297                                                 //
9298                                                 // We do so directly instead of via the normal ChannelMonitor update
9299                                                 // procedure as the ChainMonitor hasn't yet been initialized, implying
9300                                                 // we're not allowed to call it directly yet. Further, we do the update
9301                                                 // without incrementing the ChannelMonitor update ID as there isn't any
9302                                                 // reason to.
9303                                                 // If we were to generate a new ChannelMonitor update ID here and then
9304                                                 // crash before the user finishes block connect we'd end up force-closing
9305                                                 // this channel as well. On the flip side, there's no harm in restarting
9306                                                 // without the new monitor persisted - we'll end up right back here on
9307                                                 // restart.
9308                                                 let previous_channel_id = claimable_htlc.prev_hop.outpoint.to_channel_id();
9309                                                 if let Some(peer_node_id) = id_to_peer.get(&previous_channel_id){
9310                                                         let peer_state_mutex = per_peer_state.get(peer_node_id).unwrap();
9311                                                         let mut peer_state_lock = peer_state_mutex.lock().unwrap();
9312                                                         let peer_state = &mut *peer_state_lock;
9313                                                         if let Some(channel) = peer_state.channel_by_id.get_mut(&previous_channel_id) {
9314                                                                 channel.claim_htlc_while_disconnected_dropping_mon_update(claimable_htlc.prev_hop.htlc_id, payment_preimage, &args.logger);
9315                                                         }
9316                                                 }
9317                                                 if let Some(previous_hop_monitor) = args.channel_monitors.get(&claimable_htlc.prev_hop.outpoint) {
9318                                                         previous_hop_monitor.provide_payment_preimage(&payment_hash, &payment_preimage, &args.tx_broadcaster, &bounded_fee_estimator, &args.logger);
9319                                                 }
9320                                         }
9321                                         pending_events_read.push_back((events::Event::PaymentClaimed {
9322                                                 receiver_node_id,
9323                                                 payment_hash,
9324                                                 purpose: payment.purpose,
9325                                                 amount_msat: claimable_amt_msat,
9326                                                 htlcs: payment.htlcs.iter().map(events::ClaimedHTLC::from).collect(),
9327                                                 sender_intended_total_msat: payment.htlcs.first().map(|htlc| htlc.total_msat),
9328                                         }, None));
9329                                 }
9330                         }
9331                 }
9332
9333                 for (node_id, monitor_update_blocked_actions) in monitor_update_blocked_actions_per_peer.unwrap() {
9334                         if let Some(peer_state) = per_peer_state.get(&node_id) {
9335                                 for (_, actions) in monitor_update_blocked_actions.iter() {
9336                                         for action in actions.iter() {
9337                                                 if let MonitorUpdateCompletionAction::EmitEventAndFreeOtherChannel {
9338                                                         downstream_counterparty_and_funding_outpoint:
9339                                                                 Some((blocked_node_id, blocked_channel_outpoint, blocking_action)), ..
9340                                                 } = action {
9341                                                         if let Some(blocked_peer_state) = per_peer_state.get(&blocked_node_id) {
9342                                                                 blocked_peer_state.lock().unwrap().actions_blocking_raa_monitor_updates
9343                                                                         .entry(blocked_channel_outpoint.to_channel_id())
9344                                                                         .or_insert_with(Vec::new).push(blocking_action.clone());
9345                                                         } else {
9346                                                                 // If the channel we were blocking has closed, we don't need to
9347                                                                 // worry about it - the blocked monitor update should never have
9348                                                                 // been released from the `Channel` object so it can't have
9349                                                                 // completed, and if the channel closed there's no reason to bother
9350                                                                 // anymore.
9351                                                         }
9352                                                 }
9353                                         }
9354                                 }
9355                                 peer_state.lock().unwrap().monitor_update_blocked_actions = monitor_update_blocked_actions;
9356                         } else {
9357                                 log_error!(args.logger, "Got blocked actions without a per-peer-state for {}", node_id);
9358                                 return Err(DecodeError::InvalidValue);
9359                         }
9360                 }
9361
9362                 let channel_manager = ChannelManager {
9363                         genesis_hash,
9364                         fee_estimator: bounded_fee_estimator,
9365                         chain_monitor: args.chain_monitor,
9366                         tx_broadcaster: args.tx_broadcaster,
9367                         router: args.router,
9368
9369                         best_block: RwLock::new(BestBlock::new(best_block_hash, best_block_height)),
9370
9371                         inbound_payment_key: expanded_inbound_key,
9372                         pending_inbound_payments: Mutex::new(pending_inbound_payments),
9373                         pending_outbound_payments: pending_outbounds,
9374                         pending_intercepted_htlcs: Mutex::new(pending_intercepted_htlcs.unwrap()),
9375
9376                         forward_htlcs: Mutex::new(forward_htlcs),
9377                         claimable_payments: Mutex::new(ClaimablePayments { claimable_payments, pending_claiming_payments: pending_claiming_payments.unwrap() }),
9378                         outbound_scid_aliases: Mutex::new(outbound_scid_aliases),
9379                         id_to_peer: Mutex::new(id_to_peer),
9380                         short_to_chan_info: FairRwLock::new(short_to_chan_info),
9381                         fake_scid_rand_bytes: fake_scid_rand_bytes.unwrap(),
9382
9383                         probing_cookie_secret: probing_cookie_secret.unwrap(),
9384
9385                         our_network_pubkey,
9386                         secp_ctx,
9387
9388                         highest_seen_timestamp: AtomicUsize::new(highest_seen_timestamp as usize),
9389
9390                         per_peer_state: FairRwLock::new(per_peer_state),
9391
9392                         pending_events: Mutex::new(pending_events_read),
9393                         pending_events_processor: AtomicBool::new(false),
9394                         pending_background_events: Mutex::new(pending_background_events),
9395                         total_consistency_lock: RwLock::new(()),
9396                         background_events_processed_since_startup: AtomicBool::new(false),
9397                         persistence_notifier: Notifier::new(),
9398
9399                         entropy_source: args.entropy_source,
9400                         node_signer: args.node_signer,
9401                         signer_provider: args.signer_provider,
9402
9403                         logger: args.logger,
9404                         default_configuration: args.default_config,
9405                 };
9406
9407                 for htlc_source in failed_htlcs.drain(..) {
9408                         let (source, payment_hash, counterparty_node_id, channel_id) = htlc_source;
9409                         let receiver = HTLCDestination::NextHopChannel { node_id: Some(counterparty_node_id), channel_id };
9410                         let reason = HTLCFailReason::from_failure_code(0x4000 | 8);
9411                         channel_manager.fail_htlc_backwards_internal(&source, &payment_hash, &reason, receiver);
9412                 }
9413
9414                 for (source, preimage, downstream_value, downstream_closed, downstream_funding) in pending_claims_to_replay {
9415                         // We use `downstream_closed` in place of `from_onchain` here just as a guess - we
9416                         // don't remember in the `ChannelMonitor` where we got a preimage from, but if the
9417                         // channel is closed we just assume that it probably came from an on-chain claim.
9418                         channel_manager.claim_funds_internal(source, preimage, Some(downstream_value),
9419                                 downstream_closed, downstream_funding);
9420                 }
9421
9422                 //TODO: Broadcast channel update for closed channels, but only after we've made a
9423                 //connection or two.
9424
9425                 Ok((best_block_hash.clone(), channel_manager))
9426         }
9427 }
9428
9429 #[cfg(test)]
9430 mod tests {
9431         use bitcoin::hashes::Hash;
9432         use bitcoin::hashes::sha256::Hash as Sha256;
9433         use bitcoin::secp256k1::{PublicKey, Secp256k1, SecretKey};
9434         use core::sync::atomic::Ordering;
9435         use crate::events::{Event, HTLCDestination, MessageSendEvent, MessageSendEventsProvider, ClosureReason};
9436         use crate::ln::{PaymentPreimage, PaymentHash, PaymentSecret};
9437         use crate::ln::ChannelId;
9438         use crate::ln::channelmanager::{inbound_payment, PaymentId, PaymentSendFailure, RecipientOnionFields, InterceptId};
9439         use crate::ln::functional_test_utils::*;
9440         use crate::ln::msgs::{self, ErrorAction};
9441         use crate::ln::msgs::ChannelMessageHandler;
9442         use crate::routing::router::{PaymentParameters, RouteParameters, find_route};
9443         use crate::util::errors::APIError;
9444         use crate::util::test_utils;
9445         use crate::util::config::{ChannelConfig, ChannelConfigUpdate};
9446         use crate::sign::EntropySource;
9447
9448         #[test]
9449         fn test_notify_limits() {
9450                 // Check that a few cases which don't require the persistence of a new ChannelManager,
9451                 // indeed, do not cause the persistence of a new ChannelManager.
9452                 let chanmon_cfgs = create_chanmon_cfgs(3);
9453                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
9454                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs, &[None, None, None]);
9455                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
9456
9457                 // All nodes start with a persistable update pending as `create_network` connects each node
9458                 // with all other nodes to make most tests simpler.
9459                 assert!(nodes[0].node.get_persistable_update_future().poll_is_complete());
9460                 assert!(nodes[1].node.get_persistable_update_future().poll_is_complete());
9461                 assert!(nodes[2].node.get_persistable_update_future().poll_is_complete());
9462
9463                 let mut chan = create_announced_chan_between_nodes(&nodes, 0, 1);
9464
9465                 // We check that the channel info nodes have doesn't change too early, even though we try
9466                 // to connect messages with new values
9467                 chan.0.contents.fee_base_msat *= 2;
9468                 chan.1.contents.fee_base_msat *= 2;
9469                 let node_a_chan_info = nodes[0].node.list_channels_with_counterparty(
9470                         &nodes[1].node.get_our_node_id()).pop().unwrap();
9471                 let node_b_chan_info = nodes[1].node.list_channels_with_counterparty(
9472                         &nodes[0].node.get_our_node_id()).pop().unwrap();
9473
9474                 // The first two nodes (which opened a channel) should now require fresh persistence
9475                 assert!(nodes[0].node.get_persistable_update_future().poll_is_complete());
9476                 assert!(nodes[1].node.get_persistable_update_future().poll_is_complete());
9477                 // ... but the last node should not.
9478                 assert!(!nodes[2].node.get_persistable_update_future().poll_is_complete());
9479                 // After persisting the first two nodes they should no longer need fresh persistence.
9480                 assert!(!nodes[0].node.get_persistable_update_future().poll_is_complete());
9481                 assert!(!nodes[1].node.get_persistable_update_future().poll_is_complete());
9482
9483                 // Node 3, unrelated to the only channel, shouldn't care if it receives a channel_update
9484                 // about the channel.
9485                 nodes[2].node.handle_channel_update(&nodes[1].node.get_our_node_id(), &chan.0);
9486                 nodes[2].node.handle_channel_update(&nodes[1].node.get_our_node_id(), &chan.1);
9487                 assert!(!nodes[2].node.get_persistable_update_future().poll_is_complete());
9488
9489                 // The nodes which are a party to the channel should also ignore messages from unrelated
9490                 // parties.
9491                 nodes[0].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan.0);
9492                 nodes[0].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan.1);
9493                 nodes[1].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan.0);
9494                 nodes[1].node.handle_channel_update(&nodes[2].node.get_our_node_id(), &chan.1);
9495                 assert!(!nodes[0].node.get_persistable_update_future().poll_is_complete());
9496                 assert!(!nodes[1].node.get_persistable_update_future().poll_is_complete());
9497
9498                 // At this point the channel info given by peers should still be the same.
9499                 assert_eq!(nodes[0].node.list_channels()[0], node_a_chan_info);
9500                 assert_eq!(nodes[1].node.list_channels()[0], node_b_chan_info);
9501
9502                 // An earlier version of handle_channel_update didn't check the directionality of the
9503                 // update message and would always update the local fee info, even if our peer was
9504                 // (spuriously) forwarding us our own channel_update.
9505                 let as_node_one = nodes[0].node.get_our_node_id().serialize()[..] < nodes[1].node.get_our_node_id().serialize()[..];
9506                 let as_update = if as_node_one == (chan.0.contents.flags & 1 == 0 /* chan.0 is from node one */) { &chan.0 } else { &chan.1 };
9507                 let bs_update = if as_node_one == (chan.0.contents.flags & 1 == 0 /* chan.0 is from node one */) { &chan.1 } else { &chan.0 };
9508
9509                 // First deliver each peers' own message, checking that the node doesn't need to be
9510                 // persisted and that its channel info remains the same.
9511                 nodes[0].node.handle_channel_update(&nodes[1].node.get_our_node_id(), &as_update);
9512                 nodes[1].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &bs_update);
9513                 assert!(!nodes[0].node.get_persistable_update_future().poll_is_complete());
9514                 assert!(!nodes[1].node.get_persistable_update_future().poll_is_complete());
9515                 assert_eq!(nodes[0].node.list_channels()[0], node_a_chan_info);
9516                 assert_eq!(nodes[1].node.list_channels()[0], node_b_chan_info);
9517
9518                 // Finally, deliver the other peers' message, ensuring each node needs to be persisted and
9519                 // the channel info has updated.
9520                 nodes[0].node.handle_channel_update(&nodes[1].node.get_our_node_id(), &bs_update);
9521                 nodes[1].node.handle_channel_update(&nodes[0].node.get_our_node_id(), &as_update);
9522                 assert!(nodes[0].node.get_persistable_update_future().poll_is_complete());
9523                 assert!(nodes[1].node.get_persistable_update_future().poll_is_complete());
9524                 assert_ne!(nodes[0].node.list_channels()[0], node_a_chan_info);
9525                 assert_ne!(nodes[1].node.list_channels()[0], node_b_chan_info);
9526         }
9527
9528         #[test]
9529         fn test_keysend_dup_hash_partial_mpp() {
9530                 // Test that a keysend payment with a duplicate hash to an existing partial MPP payment fails as
9531                 // expected.
9532                 let chanmon_cfgs = create_chanmon_cfgs(2);
9533                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
9534                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
9535                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
9536                 create_announced_chan_between_nodes(&nodes, 0, 1);
9537
9538                 // First, send a partial MPP payment.
9539                 let (route, our_payment_hash, payment_preimage, payment_secret) = get_route_and_payment_hash!(&nodes[0], nodes[1], 100_000);
9540                 let mut mpp_route = route.clone();
9541                 mpp_route.paths.push(mpp_route.paths[0].clone());
9542
9543                 let payment_id = PaymentId([42; 32]);
9544                 // Use the utility function send_payment_along_path to send the payment with MPP data which
9545                 // indicates there are more HTLCs coming.
9546                 let cur_height = CHAN_CONFIRM_DEPTH + 1; // route_payment calls send_payment, which adds 1 to the current height. So we do the same here to match.
9547                 let session_privs = nodes[0].node.test_add_new_pending_payment(our_payment_hash,
9548                         RecipientOnionFields::secret_only(payment_secret), payment_id, &mpp_route).unwrap();
9549                 nodes[0].node.test_send_payment_along_path(&mpp_route.paths[0], &our_payment_hash,
9550                         RecipientOnionFields::secret_only(payment_secret), 200_000, cur_height, payment_id, &None, session_privs[0]).unwrap();
9551                 check_added_monitors!(nodes[0], 1);
9552                 let mut events = nodes[0].node.get_and_clear_pending_msg_events();
9553                 assert_eq!(events.len(), 1);
9554                 pass_along_path(&nodes[0], &[&nodes[1]], 200_000, our_payment_hash, Some(payment_secret), events.drain(..).next().unwrap(), false, None);
9555
9556                 // Next, send a keysend payment with the same payment_hash and make sure it fails.
9557                 nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
9558                         RecipientOnionFields::spontaneous_empty(), PaymentId(payment_preimage.0)).unwrap();
9559                 check_added_monitors!(nodes[0], 1);
9560                 let mut events = nodes[0].node.get_and_clear_pending_msg_events();
9561                 assert_eq!(events.len(), 1);
9562                 let ev = events.drain(..).next().unwrap();
9563                 let payment_event = SendEvent::from_event(ev);
9564                 nodes[1].node.handle_update_add_htlc(&nodes[0].node.get_our_node_id(), &payment_event.msgs[0]);
9565                 check_added_monitors!(nodes[1], 0);
9566                 commitment_signed_dance!(nodes[1], nodes[0], payment_event.commitment_msg, false);
9567                 expect_pending_htlcs_forwardable!(nodes[1]);
9568                 expect_pending_htlcs_forwardable_and_htlc_handling_failed!(nodes[1], vec![HTLCDestination::FailedPayment { payment_hash: our_payment_hash }]);
9569                 check_added_monitors!(nodes[1], 1);
9570                 let updates = get_htlc_update_msgs!(nodes[1], nodes[0].node.get_our_node_id());
9571                 assert!(updates.update_add_htlcs.is_empty());
9572                 assert!(updates.update_fulfill_htlcs.is_empty());
9573                 assert_eq!(updates.update_fail_htlcs.len(), 1);
9574                 assert!(updates.update_fail_malformed_htlcs.is_empty());
9575                 assert!(updates.update_fee.is_none());
9576                 nodes[0].node.handle_update_fail_htlc(&nodes[1].node.get_our_node_id(), &updates.update_fail_htlcs[0]);
9577                 commitment_signed_dance!(nodes[0], nodes[1], updates.commitment_signed, true, true);
9578                 expect_payment_failed!(nodes[0], our_payment_hash, true);
9579
9580                 // Send the second half of the original MPP payment.
9581                 nodes[0].node.test_send_payment_along_path(&mpp_route.paths[1], &our_payment_hash,
9582                         RecipientOnionFields::secret_only(payment_secret), 200_000, cur_height, payment_id, &None, session_privs[1]).unwrap();
9583                 check_added_monitors!(nodes[0], 1);
9584                 let mut events = nodes[0].node.get_and_clear_pending_msg_events();
9585                 assert_eq!(events.len(), 1);
9586                 pass_along_path(&nodes[0], &[&nodes[1]], 200_000, our_payment_hash, Some(payment_secret), events.drain(..).next().unwrap(), true, None);
9587
9588                 // Claim the full MPP payment. Note that we can't use a test utility like
9589                 // claim_funds_along_route because the ordering of the messages causes the second half of the
9590                 // payment to be put in the holding cell, which confuses the test utilities. So we exchange the
9591                 // lightning messages manually.
9592                 nodes[1].node.claim_funds(payment_preimage);
9593                 expect_payment_claimed!(nodes[1], our_payment_hash, 200_000);
9594                 check_added_monitors!(nodes[1], 2);
9595
9596                 let bs_first_updates = get_htlc_update_msgs!(nodes[1], nodes[0].node.get_our_node_id());
9597                 nodes[0].node.handle_update_fulfill_htlc(&nodes[1].node.get_our_node_id(), &bs_first_updates.update_fulfill_htlcs[0]);
9598                 expect_payment_sent(&nodes[0], payment_preimage, None, false, false);
9599                 nodes[0].node.handle_commitment_signed(&nodes[1].node.get_our_node_id(), &bs_first_updates.commitment_signed);
9600                 check_added_monitors!(nodes[0], 1);
9601                 let (as_first_raa, as_first_cs) = get_revoke_commit_msgs!(nodes[0], nodes[1].node.get_our_node_id());
9602                 nodes[1].node.handle_revoke_and_ack(&nodes[0].node.get_our_node_id(), &as_first_raa);
9603                 check_added_monitors!(nodes[1], 1);
9604                 let bs_second_updates = get_htlc_update_msgs!(nodes[1], nodes[0].node.get_our_node_id());
9605                 nodes[1].node.handle_commitment_signed(&nodes[0].node.get_our_node_id(), &as_first_cs);
9606                 check_added_monitors!(nodes[1], 1);
9607                 let bs_first_raa = get_event_msg!(nodes[1], MessageSendEvent::SendRevokeAndACK, nodes[0].node.get_our_node_id());
9608                 nodes[0].node.handle_update_fulfill_htlc(&nodes[1].node.get_our_node_id(), &bs_second_updates.update_fulfill_htlcs[0]);
9609                 nodes[0].node.handle_commitment_signed(&nodes[1].node.get_our_node_id(), &bs_second_updates.commitment_signed);
9610                 check_added_monitors!(nodes[0], 1);
9611                 let as_second_raa = get_event_msg!(nodes[0], MessageSendEvent::SendRevokeAndACK, nodes[1].node.get_our_node_id());
9612                 nodes[0].node.handle_revoke_and_ack(&nodes[1].node.get_our_node_id(), &bs_first_raa);
9613                 let as_second_updates = get_htlc_update_msgs!(nodes[0], nodes[1].node.get_our_node_id());
9614                 check_added_monitors!(nodes[0], 1);
9615                 nodes[1].node.handle_revoke_and_ack(&nodes[0].node.get_our_node_id(), &as_second_raa);
9616                 check_added_monitors!(nodes[1], 1);
9617                 nodes[1].node.handle_commitment_signed(&nodes[0].node.get_our_node_id(), &as_second_updates.commitment_signed);
9618                 check_added_monitors!(nodes[1], 1);
9619                 let bs_third_raa = get_event_msg!(nodes[1], MessageSendEvent::SendRevokeAndACK, nodes[0].node.get_our_node_id());
9620                 nodes[0].node.handle_revoke_and_ack(&nodes[1].node.get_our_node_id(), &bs_third_raa);
9621                 check_added_monitors!(nodes[0], 1);
9622
9623                 // Note that successful MPP payments will generate a single PaymentSent event upon the first
9624                 // path's success and a PaymentPathSuccessful event for each path's success.
9625                 let events = nodes[0].node.get_and_clear_pending_events();
9626                 assert_eq!(events.len(), 2);
9627                 match events[0] {
9628                         Event::PaymentPathSuccessful { payment_id: ref actual_payment_id, ref payment_hash, ref path } => {
9629                                 assert_eq!(payment_id, *actual_payment_id);
9630                                 assert_eq!(our_payment_hash, *payment_hash.as_ref().unwrap());
9631                                 assert_eq!(route.paths[0], *path);
9632                         },
9633                         _ => panic!("Unexpected event"),
9634                 }
9635                 match events[1] {
9636                         Event::PaymentPathSuccessful { payment_id: ref actual_payment_id, ref payment_hash, ref path } => {
9637                                 assert_eq!(payment_id, *actual_payment_id);
9638                                 assert_eq!(our_payment_hash, *payment_hash.as_ref().unwrap());
9639                                 assert_eq!(route.paths[0], *path);
9640                         },
9641                         _ => panic!("Unexpected event"),
9642                 }
9643         }
9644
9645         #[test]
9646         fn test_keysend_dup_payment_hash() {
9647                 do_test_keysend_dup_payment_hash(false);
9648                 do_test_keysend_dup_payment_hash(true);
9649         }
9650
9651         fn do_test_keysend_dup_payment_hash(accept_mpp_keysend: bool) {
9652                 // (1): Test that a keysend payment with a duplicate payment hash to an existing pending
9653                 //      outbound regular payment fails as expected.
9654                 // (2): Test that a regular payment with a duplicate payment hash to an existing keysend payment
9655                 //      fails as expected.
9656                 // (3): Test that a keysend payment with a duplicate payment hash to an existing keysend
9657                 //      payment fails as expected. When `accept_mpp_keysend` is false, this tests that we
9658                 //      reject MPP keysend payments, since in this case where the payment has no payment
9659                 //      secret, a keysend payment with a duplicate hash is basically an MPP keysend. If
9660                 //      `accept_mpp_keysend` is true, this tests that we only accept MPP keysends with
9661                 //      payment secrets and reject otherwise.
9662                 let chanmon_cfgs = create_chanmon_cfgs(2);
9663                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
9664                 let mut mpp_keysend_cfg = test_default_channel_config();
9665                 mpp_keysend_cfg.accept_mpp_keysend = accept_mpp_keysend;
9666                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, Some(mpp_keysend_cfg)]);
9667                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
9668                 create_announced_chan_between_nodes(&nodes, 0, 1);
9669                 let scorer = test_utils::TestScorer::new();
9670                 let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
9671
9672                 // To start (1), send a regular payment but don't claim it.
9673                 let expected_route = [&nodes[1]];
9674                 let (payment_preimage, payment_hash, _) = route_payment(&nodes[0], &expected_route, 100_000);
9675
9676                 // Next, attempt a keysend payment and make sure it fails.
9677                 let route_params = RouteParameters::from_payment_params_and_value(
9678                         PaymentParameters::for_keysend(expected_route.last().unwrap().node.get_our_node_id(),
9679                         TEST_FINAL_CLTV, false), 100_000);
9680                 let route = find_route(
9681                         &nodes[0].node.get_our_node_id(), &route_params, &nodes[0].network_graph,
9682                         None, nodes[0].logger, &scorer, &(), &random_seed_bytes
9683                 ).unwrap();
9684                 nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
9685                         RecipientOnionFields::spontaneous_empty(), PaymentId(payment_preimage.0)).unwrap();
9686                 check_added_monitors!(nodes[0], 1);
9687                 let mut events = nodes[0].node.get_and_clear_pending_msg_events();
9688                 assert_eq!(events.len(), 1);
9689                 let ev = events.drain(..).next().unwrap();
9690                 let payment_event = SendEvent::from_event(ev);
9691                 nodes[1].node.handle_update_add_htlc(&nodes[0].node.get_our_node_id(), &payment_event.msgs[0]);
9692                 check_added_monitors!(nodes[1], 0);
9693                 commitment_signed_dance!(nodes[1], nodes[0], payment_event.commitment_msg, false);
9694                 // We have to forward pending HTLCs twice - once tries to forward the payment forward (and
9695                 // fails), the second will process the resulting failure and fail the HTLC backward
9696                 expect_pending_htlcs_forwardable!(nodes[1]);
9697                 expect_pending_htlcs_forwardable_and_htlc_handling_failed!(nodes[1], vec![HTLCDestination::FailedPayment { payment_hash }]);
9698                 check_added_monitors!(nodes[1], 1);
9699                 let updates = get_htlc_update_msgs!(nodes[1], nodes[0].node.get_our_node_id());
9700                 assert!(updates.update_add_htlcs.is_empty());
9701                 assert!(updates.update_fulfill_htlcs.is_empty());
9702                 assert_eq!(updates.update_fail_htlcs.len(), 1);
9703                 assert!(updates.update_fail_malformed_htlcs.is_empty());
9704                 assert!(updates.update_fee.is_none());
9705                 nodes[0].node.handle_update_fail_htlc(&nodes[1].node.get_our_node_id(), &updates.update_fail_htlcs[0]);
9706                 commitment_signed_dance!(nodes[0], nodes[1], updates.commitment_signed, true, true);
9707                 expect_payment_failed!(nodes[0], payment_hash, true);
9708
9709                 // Finally, claim the original payment.
9710                 claim_payment(&nodes[0], &expected_route, payment_preimage);
9711
9712                 // To start (2), send a keysend payment but don't claim it.
9713                 let payment_preimage = PaymentPreimage([42; 32]);
9714                 let route = find_route(
9715                         &nodes[0].node.get_our_node_id(), &route_params, &nodes[0].network_graph,
9716                         None, nodes[0].logger, &scorer, &(), &random_seed_bytes
9717                 ).unwrap();
9718                 let payment_hash = nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
9719                         RecipientOnionFields::spontaneous_empty(), PaymentId(payment_preimage.0)).unwrap();
9720                 check_added_monitors!(nodes[0], 1);
9721                 let mut events = nodes[0].node.get_and_clear_pending_msg_events();
9722                 assert_eq!(events.len(), 1);
9723                 let event = events.pop().unwrap();
9724                 let path = vec![&nodes[1]];
9725                 pass_along_path(&nodes[0], &path, 100_000, payment_hash, None, event, true, Some(payment_preimage));
9726
9727                 // Next, attempt a regular payment and make sure it fails.
9728                 let payment_secret = PaymentSecret([43; 32]);
9729                 nodes[0].node.send_payment_with_route(&route, payment_hash,
9730                         RecipientOnionFields::secret_only(payment_secret), PaymentId(payment_hash.0)).unwrap();
9731                 check_added_monitors!(nodes[0], 1);
9732                 let mut events = nodes[0].node.get_and_clear_pending_msg_events();
9733                 assert_eq!(events.len(), 1);
9734                 let ev = events.drain(..).next().unwrap();
9735                 let payment_event = SendEvent::from_event(ev);
9736                 nodes[1].node.handle_update_add_htlc(&nodes[0].node.get_our_node_id(), &payment_event.msgs[0]);
9737                 check_added_monitors!(nodes[1], 0);
9738                 commitment_signed_dance!(nodes[1], nodes[0], payment_event.commitment_msg, false);
9739                 expect_pending_htlcs_forwardable!(nodes[1]);
9740                 expect_pending_htlcs_forwardable_and_htlc_handling_failed!(nodes[1], vec![HTLCDestination::FailedPayment { payment_hash }]);
9741                 check_added_monitors!(nodes[1], 1);
9742                 let updates = get_htlc_update_msgs!(nodes[1], nodes[0].node.get_our_node_id());
9743                 assert!(updates.update_add_htlcs.is_empty());
9744                 assert!(updates.update_fulfill_htlcs.is_empty());
9745                 assert_eq!(updates.update_fail_htlcs.len(), 1);
9746                 assert!(updates.update_fail_malformed_htlcs.is_empty());
9747                 assert!(updates.update_fee.is_none());
9748                 nodes[0].node.handle_update_fail_htlc(&nodes[1].node.get_our_node_id(), &updates.update_fail_htlcs[0]);
9749                 commitment_signed_dance!(nodes[0], nodes[1], updates.commitment_signed, true, true);
9750                 expect_payment_failed!(nodes[0], payment_hash, true);
9751
9752                 // Finally, succeed the keysend payment.
9753                 claim_payment(&nodes[0], &expected_route, payment_preimage);
9754
9755                 // To start (3), send a keysend payment but don't claim it.
9756                 let payment_id_1 = PaymentId([44; 32]);
9757                 let payment_hash = nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
9758                         RecipientOnionFields::spontaneous_empty(), payment_id_1).unwrap();
9759                 check_added_monitors!(nodes[0], 1);
9760                 let mut events = nodes[0].node.get_and_clear_pending_msg_events();
9761                 assert_eq!(events.len(), 1);
9762                 let event = events.pop().unwrap();
9763                 let path = vec![&nodes[1]];
9764                 pass_along_path(&nodes[0], &path, 100_000, payment_hash, None, event, true, Some(payment_preimage));
9765
9766                 // Next, attempt a keysend payment and make sure it fails.
9767                 let route_params = RouteParameters::from_payment_params_and_value(
9768                         PaymentParameters::for_keysend(expected_route.last().unwrap().node.get_our_node_id(), TEST_FINAL_CLTV, false),
9769                         100_000
9770                 );
9771                 let route = find_route(
9772                         &nodes[0].node.get_our_node_id(), &route_params, &nodes[0].network_graph,
9773                         None, nodes[0].logger, &scorer, &(), &random_seed_bytes
9774                 ).unwrap();
9775                 let payment_id_2 = PaymentId([45; 32]);
9776                 nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
9777                         RecipientOnionFields::spontaneous_empty(), payment_id_2).unwrap();
9778                 check_added_monitors!(nodes[0], 1);
9779                 let mut events = nodes[0].node.get_and_clear_pending_msg_events();
9780                 assert_eq!(events.len(), 1);
9781                 let ev = events.drain(..).next().unwrap();
9782                 let payment_event = SendEvent::from_event(ev);
9783                 nodes[1].node.handle_update_add_htlc(&nodes[0].node.get_our_node_id(), &payment_event.msgs[0]);
9784                 check_added_monitors!(nodes[1], 0);
9785                 commitment_signed_dance!(nodes[1], nodes[0], payment_event.commitment_msg, false);
9786                 expect_pending_htlcs_forwardable!(nodes[1]);
9787                 expect_pending_htlcs_forwardable_and_htlc_handling_failed!(nodes[1], vec![HTLCDestination::FailedPayment { payment_hash }]);
9788                 check_added_monitors!(nodes[1], 1);
9789                 let updates = get_htlc_update_msgs!(nodes[1], nodes[0].node.get_our_node_id());
9790                 assert!(updates.update_add_htlcs.is_empty());
9791                 assert!(updates.update_fulfill_htlcs.is_empty());
9792                 assert_eq!(updates.update_fail_htlcs.len(), 1);
9793                 assert!(updates.update_fail_malformed_htlcs.is_empty());
9794                 assert!(updates.update_fee.is_none());
9795                 nodes[0].node.handle_update_fail_htlc(&nodes[1].node.get_our_node_id(), &updates.update_fail_htlcs[0]);
9796                 commitment_signed_dance!(nodes[0], nodes[1], updates.commitment_signed, true, true);
9797                 expect_payment_failed!(nodes[0], payment_hash, true);
9798
9799                 // Finally, claim the original payment.
9800                 claim_payment(&nodes[0], &expected_route, payment_preimage);
9801         }
9802
9803         #[test]
9804         fn test_keysend_hash_mismatch() {
9805                 // Test that if we receive a keysend `update_add_htlc` msg, we fail as expected if the keysend
9806                 // preimage doesn't match the msg's payment hash.
9807                 let chanmon_cfgs = create_chanmon_cfgs(2);
9808                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
9809                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
9810                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
9811
9812                 let payer_pubkey = nodes[0].node.get_our_node_id();
9813                 let payee_pubkey = nodes[1].node.get_our_node_id();
9814
9815                 let _chan = create_chan_between_nodes(&nodes[0], &nodes[1]);
9816                 let route_params = RouteParameters::from_payment_params_and_value(
9817                         PaymentParameters::for_keysend(payee_pubkey, 40, false), 10_000);
9818                 let network_graph = nodes[0].network_graph.clone();
9819                 let first_hops = nodes[0].node.list_usable_channels();
9820                 let scorer = test_utils::TestScorer::new();
9821                 let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
9822                 let route = find_route(
9823                         &payer_pubkey, &route_params, &network_graph, Some(&first_hops.iter().collect::<Vec<_>>()),
9824                         nodes[0].logger, &scorer, &(), &random_seed_bytes
9825                 ).unwrap();
9826
9827                 let test_preimage = PaymentPreimage([42; 32]);
9828                 let mismatch_payment_hash = PaymentHash([43; 32]);
9829                 let session_privs = nodes[0].node.test_add_new_pending_payment(mismatch_payment_hash,
9830                         RecipientOnionFields::spontaneous_empty(), PaymentId(mismatch_payment_hash.0), &route).unwrap();
9831                 nodes[0].node.test_send_payment_internal(&route, mismatch_payment_hash,
9832                         RecipientOnionFields::spontaneous_empty(), Some(test_preimage), PaymentId(mismatch_payment_hash.0), None, session_privs).unwrap();
9833                 check_added_monitors!(nodes[0], 1);
9834
9835                 let updates = get_htlc_update_msgs!(nodes[0], nodes[1].node.get_our_node_id());
9836                 assert_eq!(updates.update_add_htlcs.len(), 1);
9837                 assert!(updates.update_fulfill_htlcs.is_empty());
9838                 assert!(updates.update_fail_htlcs.is_empty());
9839                 assert!(updates.update_fail_malformed_htlcs.is_empty());
9840                 assert!(updates.update_fee.is_none());
9841                 nodes[1].node.handle_update_add_htlc(&nodes[0].node.get_our_node_id(), &updates.update_add_htlcs[0]);
9842
9843                 nodes[1].logger.assert_log_contains("lightning::ln::channelmanager", "Payment preimage didn't match payment hash", 1);
9844         }
9845
9846         #[test]
9847         fn test_keysend_msg_with_secret_err() {
9848                 // Test that we error as expected if we receive a keysend payment that includes a payment
9849                 // secret when we don't support MPP keysend.
9850                 let mut reject_mpp_keysend_cfg = test_default_channel_config();
9851                 reject_mpp_keysend_cfg.accept_mpp_keysend = false;
9852                 let chanmon_cfgs = create_chanmon_cfgs(2);
9853                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
9854                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, Some(reject_mpp_keysend_cfg)]);
9855                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
9856
9857                 let payer_pubkey = nodes[0].node.get_our_node_id();
9858                 let payee_pubkey = nodes[1].node.get_our_node_id();
9859
9860                 let _chan = create_chan_between_nodes(&nodes[0], &nodes[1]);
9861                 let route_params = RouteParameters::from_payment_params_and_value(
9862                         PaymentParameters::for_keysend(payee_pubkey, 40, false), 10_000);
9863                 let network_graph = nodes[0].network_graph.clone();
9864                 let first_hops = nodes[0].node.list_usable_channels();
9865                 let scorer = test_utils::TestScorer::new();
9866                 let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
9867                 let route = find_route(
9868                         &payer_pubkey, &route_params, &network_graph, Some(&first_hops.iter().collect::<Vec<_>>()),
9869                         nodes[0].logger, &scorer, &(), &random_seed_bytes
9870                 ).unwrap();
9871
9872                 let test_preimage = PaymentPreimage([42; 32]);
9873                 let test_secret = PaymentSecret([43; 32]);
9874                 let payment_hash = PaymentHash(Sha256::hash(&test_preimage.0).into_inner());
9875                 let session_privs = nodes[0].node.test_add_new_pending_payment(payment_hash,
9876                         RecipientOnionFields::secret_only(test_secret), PaymentId(payment_hash.0), &route).unwrap();
9877                 nodes[0].node.test_send_payment_internal(&route, payment_hash,
9878                         RecipientOnionFields::secret_only(test_secret), Some(test_preimage),
9879                         PaymentId(payment_hash.0), None, session_privs).unwrap();
9880                 check_added_monitors!(nodes[0], 1);
9881
9882                 let updates = get_htlc_update_msgs!(nodes[0], nodes[1].node.get_our_node_id());
9883                 assert_eq!(updates.update_add_htlcs.len(), 1);
9884                 assert!(updates.update_fulfill_htlcs.is_empty());
9885                 assert!(updates.update_fail_htlcs.is_empty());
9886                 assert!(updates.update_fail_malformed_htlcs.is_empty());
9887                 assert!(updates.update_fee.is_none());
9888                 nodes[1].node.handle_update_add_htlc(&nodes[0].node.get_our_node_id(), &updates.update_add_htlcs[0]);
9889
9890                 nodes[1].logger.assert_log_contains("lightning::ln::channelmanager", "We don't support MPP keysend payments", 1);
9891         }
9892
9893         #[test]
9894         fn test_multi_hop_missing_secret() {
9895                 let chanmon_cfgs = create_chanmon_cfgs(4);
9896                 let node_cfgs = create_node_cfgs(4, &chanmon_cfgs);
9897                 let node_chanmgrs = create_node_chanmgrs(4, &node_cfgs, &[None, None, None, None]);
9898                 let nodes = create_network(4, &node_cfgs, &node_chanmgrs);
9899
9900                 let chan_1_id = create_announced_chan_between_nodes(&nodes, 0, 1).0.contents.short_channel_id;
9901                 let chan_2_id = create_announced_chan_between_nodes(&nodes, 0, 2).0.contents.short_channel_id;
9902                 let chan_3_id = create_announced_chan_between_nodes(&nodes, 1, 3).0.contents.short_channel_id;
9903                 let chan_4_id = create_announced_chan_between_nodes(&nodes, 2, 3).0.contents.short_channel_id;
9904
9905                 // Marshall an MPP route.
9906                 let (mut route, payment_hash, _, _) = get_route_and_payment_hash!(&nodes[0], nodes[3], 100000);
9907                 let path = route.paths[0].clone();
9908                 route.paths.push(path);
9909                 route.paths[0].hops[0].pubkey = nodes[1].node.get_our_node_id();
9910                 route.paths[0].hops[0].short_channel_id = chan_1_id;
9911                 route.paths[0].hops[1].short_channel_id = chan_3_id;
9912                 route.paths[1].hops[0].pubkey = nodes[2].node.get_our_node_id();
9913                 route.paths[1].hops[0].short_channel_id = chan_2_id;
9914                 route.paths[1].hops[1].short_channel_id = chan_4_id;
9915
9916                 match nodes[0].node.send_payment_with_route(&route, payment_hash,
9917                         RecipientOnionFields::spontaneous_empty(), PaymentId(payment_hash.0))
9918                 .unwrap_err() {
9919                         PaymentSendFailure::ParameterError(APIError::APIMisuseError { ref err }) => {
9920                                 assert!(regex::Regex::new(r"Payment secret is required for multi-path payments").unwrap().is_match(err))
9921                         },
9922                         _ => panic!("unexpected error")
9923                 }
9924         }
9925
9926         #[test]
9927         fn test_drop_disconnected_peers_when_removing_channels() {
9928                 let chanmon_cfgs = create_chanmon_cfgs(2);
9929                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
9930                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
9931                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
9932
9933                 let chan = create_announced_chan_between_nodes(&nodes, 0, 1);
9934
9935                 nodes[0].node.peer_disconnected(&nodes[1].node.get_our_node_id());
9936                 nodes[1].node.peer_disconnected(&nodes[0].node.get_our_node_id());
9937
9938                 nodes[0].node.force_close_broadcasting_latest_txn(&chan.2, &nodes[1].node.get_our_node_id()).unwrap();
9939                 check_closed_broadcast!(nodes[0], true);
9940                 check_added_monitors!(nodes[0], 1);
9941                 check_closed_event!(nodes[0], 1, ClosureReason::HolderForceClosed, [nodes[1].node.get_our_node_id()], 100000);
9942
9943                 {
9944                         // Assert that nodes[1] is awaiting removal for nodes[0] once nodes[1] has been
9945                         // disconnected and the channel between has been force closed.
9946                         let nodes_0_per_peer_state = nodes[0].node.per_peer_state.read().unwrap();
9947                         // Assert that nodes[1] isn't removed before `timer_tick_occurred` has been executed.
9948                         assert_eq!(nodes_0_per_peer_state.len(), 1);
9949                         assert!(nodes_0_per_peer_state.get(&nodes[1].node.get_our_node_id()).is_some());
9950                 }
9951
9952                 nodes[0].node.timer_tick_occurred();
9953
9954                 {
9955                         // Assert that nodes[1] has now been removed.
9956                         assert_eq!(nodes[0].node.per_peer_state.read().unwrap().len(), 0);
9957                 }
9958         }
9959
9960         #[test]
9961         fn bad_inbound_payment_hash() {
9962                 // Add coverage for checking that a user-provided payment hash matches the payment secret.
9963                 let chanmon_cfgs = create_chanmon_cfgs(2);
9964                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
9965                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
9966                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
9967
9968                 let (_, payment_hash, payment_secret) = get_payment_preimage_hash!(&nodes[0]);
9969                 let payment_data = msgs::FinalOnionHopData {
9970                         payment_secret,
9971                         total_msat: 100_000,
9972                 };
9973
9974                 // Ensure that if the payment hash given to `inbound_payment::verify` differs from the original,
9975                 // payment verification fails as expected.
9976                 let mut bad_payment_hash = payment_hash.clone();
9977                 bad_payment_hash.0[0] += 1;
9978                 match inbound_payment::verify(bad_payment_hash, &payment_data, nodes[0].node.highest_seen_timestamp.load(Ordering::Acquire) as u64, &nodes[0].node.inbound_payment_key, &nodes[0].logger) {
9979                         Ok(_) => panic!("Unexpected ok"),
9980                         Err(()) => {
9981                                 nodes[0].logger.assert_log_contains("lightning::ln::inbound_payment", "Failing HTLC with user-generated payment_hash", 1);
9982                         }
9983                 }
9984
9985                 // Check that using the original payment hash succeeds.
9986                 assert!(inbound_payment::verify(payment_hash, &payment_data, nodes[0].node.highest_seen_timestamp.load(Ordering::Acquire) as u64, &nodes[0].node.inbound_payment_key, &nodes[0].logger).is_ok());
9987         }
9988
9989         #[test]
9990         fn test_id_to_peer_coverage() {
9991                 // Test that the `ChannelManager:id_to_peer` contains channels which have been assigned
9992                 // a `channel_id` (i.e. have had the funding tx created), and that they are removed once
9993                 // the channel is successfully closed.
9994                 let chanmon_cfgs = create_chanmon_cfgs(2);
9995                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
9996                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
9997                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
9998
9999                 nodes[0].node.create_channel(nodes[1].node.get_our_node_id(), 1_000_000, 500_000_000, 42, None).unwrap();
10000                 let open_channel = get_event_msg!(nodes[0], MessageSendEvent::SendOpenChannel, nodes[1].node.get_our_node_id());
10001                 nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel);
10002                 let accept_channel = get_event_msg!(nodes[1], MessageSendEvent::SendAcceptChannel, nodes[0].node.get_our_node_id());
10003                 nodes[0].node.handle_accept_channel(&nodes[1].node.get_our_node_id(), &accept_channel);
10004
10005                 let (temporary_channel_id, tx, _funding_output) = create_funding_transaction(&nodes[0], &nodes[1].node.get_our_node_id(), 1_000_000, 42);
10006                 let channel_id = ChannelId::from_bytes(tx.txid().into_inner());
10007                 {
10008                         // Ensure that the `id_to_peer` map is empty until either party has received the
10009                         // funding transaction, and have the real `channel_id`.
10010                         assert_eq!(nodes[0].node.id_to_peer.lock().unwrap().len(), 0);
10011                         assert_eq!(nodes[1].node.id_to_peer.lock().unwrap().len(), 0);
10012                 }
10013
10014                 nodes[0].node.funding_transaction_generated(&temporary_channel_id, &nodes[1].node.get_our_node_id(), tx.clone()).unwrap();
10015                 {
10016                         // Assert that `nodes[0]`'s `id_to_peer` map is populated with the channel as soon as
10017                         // as it has the funding transaction.
10018                         let nodes_0_lock = nodes[0].node.id_to_peer.lock().unwrap();
10019                         assert_eq!(nodes_0_lock.len(), 1);
10020                         assert!(nodes_0_lock.contains_key(&channel_id));
10021                 }
10022
10023                 assert_eq!(nodes[1].node.id_to_peer.lock().unwrap().len(), 0);
10024
10025                 let funding_created_msg = get_event_msg!(nodes[0], MessageSendEvent::SendFundingCreated, nodes[1].node.get_our_node_id());
10026
10027                 nodes[1].node.handle_funding_created(&nodes[0].node.get_our_node_id(), &funding_created_msg);
10028                 {
10029                         let nodes_0_lock = nodes[0].node.id_to_peer.lock().unwrap();
10030                         assert_eq!(nodes_0_lock.len(), 1);
10031                         assert!(nodes_0_lock.contains_key(&channel_id));
10032                 }
10033                 expect_channel_pending_event(&nodes[1], &nodes[0].node.get_our_node_id());
10034
10035                 {
10036                         // Assert that `nodes[1]`'s `id_to_peer` map is populated with the channel as soon as
10037                         // as it has the funding transaction.
10038                         let nodes_1_lock = nodes[1].node.id_to_peer.lock().unwrap();
10039                         assert_eq!(nodes_1_lock.len(), 1);
10040                         assert!(nodes_1_lock.contains_key(&channel_id));
10041                 }
10042                 check_added_monitors!(nodes[1], 1);
10043                 let funding_signed = get_event_msg!(nodes[1], MessageSendEvent::SendFundingSigned, nodes[0].node.get_our_node_id());
10044                 nodes[0].node.handle_funding_signed(&nodes[1].node.get_our_node_id(), &funding_signed);
10045                 check_added_monitors!(nodes[0], 1);
10046                 expect_channel_pending_event(&nodes[0], &nodes[1].node.get_our_node_id());
10047                 let (channel_ready, _) = create_chan_between_nodes_with_value_confirm(&nodes[0], &nodes[1], &tx);
10048                 let (announcement, nodes_0_update, nodes_1_update) = create_chan_between_nodes_with_value_b(&nodes[0], &nodes[1], &channel_ready);
10049                 update_nodes_with_chan_announce(&nodes, 0, 1, &announcement, &nodes_0_update, &nodes_1_update);
10050
10051                 nodes[0].node.close_channel(&channel_id, &nodes[1].node.get_our_node_id()).unwrap();
10052                 nodes[1].node.handle_shutdown(&nodes[0].node.get_our_node_id(), &get_event_msg!(nodes[0], MessageSendEvent::SendShutdown, nodes[1].node.get_our_node_id()));
10053                 let nodes_1_shutdown = get_event_msg!(nodes[1], MessageSendEvent::SendShutdown, nodes[0].node.get_our_node_id());
10054                 nodes[0].node.handle_shutdown(&nodes[1].node.get_our_node_id(), &nodes_1_shutdown);
10055
10056                 let closing_signed_node_0 = get_event_msg!(nodes[0], MessageSendEvent::SendClosingSigned, nodes[1].node.get_our_node_id());
10057                 nodes[1].node.handle_closing_signed(&nodes[0].node.get_our_node_id(), &closing_signed_node_0);
10058                 {
10059                         // Assert that the channel is kept in the `id_to_peer` map for both nodes until the
10060                         // channel can be fully closed by both parties (i.e. no outstanding htlcs exists, the
10061                         // fee for the closing transaction has been negotiated and the parties has the other
10062                         // party's signature for the fee negotiated closing transaction.)
10063                         let nodes_0_lock = nodes[0].node.id_to_peer.lock().unwrap();
10064                         assert_eq!(nodes_0_lock.len(), 1);
10065                         assert!(nodes_0_lock.contains_key(&channel_id));
10066                 }
10067
10068                 {
10069                         // At this stage, `nodes[1]` has proposed a fee for the closing transaction in the
10070                         // `handle_closing_signed` call above. As `nodes[1]` has not yet received the signature
10071                         // from `nodes[0]` for the closing transaction with the proposed fee, the channel is
10072                         // kept in the `nodes[1]`'s `id_to_peer` map.
10073                         let nodes_1_lock = nodes[1].node.id_to_peer.lock().unwrap();
10074                         assert_eq!(nodes_1_lock.len(), 1);
10075                         assert!(nodes_1_lock.contains_key(&channel_id));
10076                 }
10077
10078                 nodes[0].node.handle_closing_signed(&nodes[1].node.get_our_node_id(), &get_event_msg!(nodes[1], MessageSendEvent::SendClosingSigned, nodes[0].node.get_our_node_id()));
10079                 {
10080                         // `nodes[0]` accepts `nodes[1]`'s proposed fee for the closing transaction, and
10081                         // therefore has all it needs to fully close the channel (both signatures for the
10082                         // closing transaction).
10083                         // Assert that the channel is removed from `nodes[0]`'s `id_to_peer` map as it can be
10084                         // fully closed by `nodes[0]`.
10085                         assert_eq!(nodes[0].node.id_to_peer.lock().unwrap().len(), 0);
10086
10087                         // Assert that the channel is still in `nodes[1]`'s  `id_to_peer` map, as `nodes[1]`
10088                         // doesn't have `nodes[0]`'s signature for the closing transaction yet.
10089                         let nodes_1_lock = nodes[1].node.id_to_peer.lock().unwrap();
10090                         assert_eq!(nodes_1_lock.len(), 1);
10091                         assert!(nodes_1_lock.contains_key(&channel_id));
10092                 }
10093
10094                 let (_nodes_0_update, closing_signed_node_0) = get_closing_signed_broadcast!(nodes[0].node, nodes[1].node.get_our_node_id());
10095
10096                 nodes[1].node.handle_closing_signed(&nodes[0].node.get_our_node_id(), &closing_signed_node_0.unwrap());
10097                 {
10098                         // Assert that the channel has now been removed from both parties `id_to_peer` map once
10099                         // they both have everything required to fully close the channel.
10100                         assert_eq!(nodes[1].node.id_to_peer.lock().unwrap().len(), 0);
10101                 }
10102                 let (_nodes_1_update, _none) = get_closing_signed_broadcast!(nodes[1].node, nodes[0].node.get_our_node_id());
10103
10104                 check_closed_event!(nodes[0], 1, ClosureReason::CooperativeClosure, [nodes[1].node.get_our_node_id()], 1000000);
10105                 check_closed_event!(nodes[1], 1, ClosureReason::CooperativeClosure, [nodes[0].node.get_our_node_id()], 1000000);
10106         }
10107
10108         fn check_not_connected_to_peer_error<T>(res_err: Result<T, APIError>, expected_public_key: PublicKey) {
10109                 let expected_message = format!("Not connected to node: {}", expected_public_key);
10110                 check_api_error_message(expected_message, res_err)
10111         }
10112
10113         fn check_unkown_peer_error<T>(res_err: Result<T, APIError>, expected_public_key: PublicKey) {
10114                 let expected_message = format!("Can't find a peer matching the passed counterparty node_id {}", expected_public_key);
10115                 check_api_error_message(expected_message, res_err)
10116         }
10117
10118         fn check_api_error_message<T>(expected_err_message: String, res_err: Result<T, APIError>) {
10119                 match res_err {
10120                         Err(APIError::APIMisuseError { err }) => {
10121                                 assert_eq!(err, expected_err_message);
10122                         },
10123                         Err(APIError::ChannelUnavailable { err }) => {
10124                                 assert_eq!(err, expected_err_message);
10125                         },
10126                         Ok(_) => panic!("Unexpected Ok"),
10127                         Err(_) => panic!("Unexpected Error"),
10128                 }
10129         }
10130
10131         #[test]
10132         fn test_api_calls_with_unkown_counterparty_node() {
10133                 // Tests that our API functions that expects a `counterparty_node_id` as input, behaves as
10134                 // expected if the `counterparty_node_id` is an unkown peer in the
10135                 // `ChannelManager::per_peer_state` map.
10136                 let chanmon_cfg = create_chanmon_cfgs(2);
10137                 let node_cfg = create_node_cfgs(2, &chanmon_cfg);
10138                 let node_chanmgr = create_node_chanmgrs(2, &node_cfg, &[None, None]);
10139                 let nodes = create_network(2, &node_cfg, &node_chanmgr);
10140
10141                 // Dummy values
10142                 let channel_id = ChannelId::from_bytes([4; 32]);
10143                 let unkown_public_key = PublicKey::from_secret_key(&Secp256k1::signing_only(), &SecretKey::from_slice(&[42; 32]).unwrap());
10144                 let intercept_id = InterceptId([0; 32]);
10145
10146                 // Test the API functions.
10147                 check_not_connected_to_peer_error(nodes[0].node.create_channel(unkown_public_key, 1_000_000, 500_000_000, 42, None), unkown_public_key);
10148
10149                 check_unkown_peer_error(nodes[0].node.accept_inbound_channel(&channel_id, &unkown_public_key, 42), unkown_public_key);
10150
10151                 check_unkown_peer_error(nodes[0].node.close_channel(&channel_id, &unkown_public_key), unkown_public_key);
10152
10153                 check_unkown_peer_error(nodes[0].node.force_close_broadcasting_latest_txn(&channel_id, &unkown_public_key), unkown_public_key);
10154
10155                 check_unkown_peer_error(nodes[0].node.force_close_without_broadcasting_txn(&channel_id, &unkown_public_key), unkown_public_key);
10156
10157                 check_unkown_peer_error(nodes[0].node.forward_intercepted_htlc(intercept_id, &channel_id, unkown_public_key, 1_000_000), unkown_public_key);
10158
10159                 check_unkown_peer_error(nodes[0].node.update_channel_config(&unkown_public_key, &[channel_id], &ChannelConfig::default()), unkown_public_key);
10160         }
10161
10162         #[test]
10163         fn test_connection_limiting() {
10164                 // Test that we limit un-channel'd peers and un-funded channels properly.
10165                 let chanmon_cfgs = create_chanmon_cfgs(2);
10166                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
10167                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
10168                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
10169
10170                 // Note that create_network connects the nodes together for us
10171
10172                 nodes[0].node.create_channel(nodes[1].node.get_our_node_id(), 100_000, 0, 42, None).unwrap();
10173                 let mut open_channel_msg = get_event_msg!(nodes[0], MessageSendEvent::SendOpenChannel, nodes[1].node.get_our_node_id());
10174
10175                 let mut funding_tx = None;
10176                 for idx in 0..super::MAX_UNFUNDED_CHANS_PER_PEER {
10177                         nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
10178                         let accept_channel = get_event_msg!(nodes[1], MessageSendEvent::SendAcceptChannel, nodes[0].node.get_our_node_id());
10179
10180                         if idx == 0 {
10181                                 nodes[0].node.handle_accept_channel(&nodes[1].node.get_our_node_id(), &accept_channel);
10182                                 let (temporary_channel_id, tx, _) = create_funding_transaction(&nodes[0], &nodes[1].node.get_our_node_id(), 100_000, 42);
10183                                 funding_tx = Some(tx.clone());
10184                                 nodes[0].node.funding_transaction_generated(&temporary_channel_id, &nodes[1].node.get_our_node_id(), tx).unwrap();
10185                                 let funding_created_msg = get_event_msg!(nodes[0], MessageSendEvent::SendFundingCreated, nodes[1].node.get_our_node_id());
10186
10187                                 nodes[1].node.handle_funding_created(&nodes[0].node.get_our_node_id(), &funding_created_msg);
10188                                 check_added_monitors!(nodes[1], 1);
10189                                 expect_channel_pending_event(&nodes[1], &nodes[0].node.get_our_node_id());
10190
10191                                 let funding_signed = get_event_msg!(nodes[1], MessageSendEvent::SendFundingSigned, nodes[0].node.get_our_node_id());
10192
10193                                 nodes[0].node.handle_funding_signed(&nodes[1].node.get_our_node_id(), &funding_signed);
10194                                 check_added_monitors!(nodes[0], 1);
10195                                 expect_channel_pending_event(&nodes[0], &nodes[1].node.get_our_node_id());
10196                         }
10197                         open_channel_msg.temporary_channel_id = ChannelId::temporary_from_entropy_source(&nodes[0].keys_manager);
10198                 }
10199
10200                 // A MAX_UNFUNDED_CHANS_PER_PEER + 1 channel will be summarily rejected
10201                 open_channel_msg.temporary_channel_id = ChannelId::temporary_from_entropy_source(&nodes[0].keys_manager);
10202                 nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
10203                 assert_eq!(get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id()).channel_id,
10204                         open_channel_msg.temporary_channel_id);
10205
10206                 // Further, because all of our channels with nodes[0] are inbound, and none of them funded,
10207                 // it doesn't count as a "protected" peer, i.e. it counts towards the MAX_NO_CHANNEL_PEERS
10208                 // limit.
10209                 let mut peer_pks = Vec::with_capacity(super::MAX_NO_CHANNEL_PEERS);
10210                 for _ in 1..super::MAX_NO_CHANNEL_PEERS {
10211                         let random_pk = PublicKey::from_secret_key(&nodes[0].node.secp_ctx,
10212                                 &SecretKey::from_slice(&nodes[1].keys_manager.get_secure_random_bytes()).unwrap());
10213                         peer_pks.push(random_pk);
10214                         nodes[1].node.peer_connected(&random_pk, &msgs::Init {
10215                                 features: nodes[0].node.init_features(), networks: None, remote_network_address: None
10216                         }, true).unwrap();
10217                 }
10218                 let last_random_pk = PublicKey::from_secret_key(&nodes[0].node.secp_ctx,
10219                         &SecretKey::from_slice(&nodes[1].keys_manager.get_secure_random_bytes()).unwrap());
10220                 nodes[1].node.peer_connected(&last_random_pk, &msgs::Init {
10221                         features: nodes[0].node.init_features(), networks: None, remote_network_address: None
10222                 }, true).unwrap_err();
10223
10224                 // Also importantly, because nodes[0] isn't "protected", we will refuse a reconnection from
10225                 // them if we have too many un-channel'd peers.
10226                 nodes[1].node.peer_disconnected(&nodes[0].node.get_our_node_id());
10227                 let chan_closed_events = nodes[1].node.get_and_clear_pending_events();
10228                 assert_eq!(chan_closed_events.len(), super::MAX_UNFUNDED_CHANS_PER_PEER - 1);
10229                 for ev in chan_closed_events {
10230                         if let Event::ChannelClosed { .. } = ev { } else { panic!(); }
10231                 }
10232                 nodes[1].node.peer_connected(&last_random_pk, &msgs::Init {
10233                         features: nodes[0].node.init_features(), networks: None, remote_network_address: None
10234                 }, true).unwrap();
10235                 nodes[1].node.peer_connected(&nodes[0].node.get_our_node_id(), &msgs::Init {
10236                         features: nodes[0].node.init_features(), networks: None, remote_network_address: None
10237                 }, true).unwrap_err();
10238
10239                 // but of course if the connection is outbound its allowed...
10240                 nodes[1].node.peer_connected(&nodes[0].node.get_our_node_id(), &msgs::Init {
10241                         features: nodes[0].node.init_features(), networks: None, remote_network_address: None
10242                 }, false).unwrap();
10243                 nodes[1].node.peer_disconnected(&nodes[0].node.get_our_node_id());
10244
10245                 // Now nodes[0] is disconnected but still has a pending, un-funded channel lying around.
10246                 // Even though we accept one more connection from new peers, we won't actually let them
10247                 // open channels.
10248                 assert!(peer_pks.len() > super::MAX_UNFUNDED_CHANNEL_PEERS - 1);
10249                 for i in 0..super::MAX_UNFUNDED_CHANNEL_PEERS - 1 {
10250                         nodes[1].node.handle_open_channel(&peer_pks[i], &open_channel_msg);
10251                         get_event_msg!(nodes[1], MessageSendEvent::SendAcceptChannel, peer_pks[i]);
10252                         open_channel_msg.temporary_channel_id = ChannelId::temporary_from_entropy_source(&nodes[0].keys_manager);
10253                 }
10254                 nodes[1].node.handle_open_channel(&last_random_pk, &open_channel_msg);
10255                 assert_eq!(get_err_msg(&nodes[1], &last_random_pk).channel_id,
10256                         open_channel_msg.temporary_channel_id);
10257
10258                 // Of course, however, outbound channels are always allowed
10259                 nodes[1].node.create_channel(last_random_pk, 100_000, 0, 42, None).unwrap();
10260                 get_event_msg!(nodes[1], MessageSendEvent::SendOpenChannel, last_random_pk);
10261
10262                 // If we fund the first channel, nodes[0] has a live on-chain channel with us, it is now
10263                 // "protected" and can connect again.
10264                 mine_transaction(&nodes[1], funding_tx.as_ref().unwrap());
10265                 nodes[1].node.peer_connected(&nodes[0].node.get_our_node_id(), &msgs::Init {
10266                         features: nodes[0].node.init_features(), networks: None, remote_network_address: None
10267                 }, true).unwrap();
10268                 get_event_msg!(nodes[1], MessageSendEvent::SendChannelReestablish, nodes[0].node.get_our_node_id());
10269
10270                 // Further, because the first channel was funded, we can open another channel with
10271                 // last_random_pk.
10272                 nodes[1].node.handle_open_channel(&last_random_pk, &open_channel_msg);
10273                 get_event_msg!(nodes[1], MessageSendEvent::SendAcceptChannel, last_random_pk);
10274         }
10275
10276         #[test]
10277         fn test_outbound_chans_unlimited() {
10278                 // Test that we never refuse an outbound channel even if a peer is unfuned-channel-limited
10279                 let chanmon_cfgs = create_chanmon_cfgs(2);
10280                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
10281                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, None]);
10282                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
10283
10284                 // Note that create_network connects the nodes together for us
10285
10286                 nodes[0].node.create_channel(nodes[1].node.get_our_node_id(), 100_000, 0, 42, None).unwrap();
10287                 let mut open_channel_msg = get_event_msg!(nodes[0], MessageSendEvent::SendOpenChannel, nodes[1].node.get_our_node_id());
10288
10289                 for _ in 0..super::MAX_UNFUNDED_CHANS_PER_PEER {
10290                         nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
10291                         get_event_msg!(nodes[1], MessageSendEvent::SendAcceptChannel, nodes[0].node.get_our_node_id());
10292                         open_channel_msg.temporary_channel_id = ChannelId::temporary_from_entropy_source(&nodes[0].keys_manager);
10293                 }
10294
10295                 // Once we have MAX_UNFUNDED_CHANS_PER_PEER unfunded channels, new inbound channels will be
10296                 // rejected.
10297                 nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
10298                 assert_eq!(get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id()).channel_id,
10299                         open_channel_msg.temporary_channel_id);
10300
10301                 // but we can still open an outbound channel.
10302                 nodes[1].node.create_channel(nodes[0].node.get_our_node_id(), 100_000, 0, 42, None).unwrap();
10303                 get_event_msg!(nodes[1], MessageSendEvent::SendOpenChannel, nodes[0].node.get_our_node_id());
10304
10305                 // but even with such an outbound channel, additional inbound channels will still fail.
10306                 nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
10307                 assert_eq!(get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id()).channel_id,
10308                         open_channel_msg.temporary_channel_id);
10309         }
10310
10311         #[test]
10312         fn test_0conf_limiting() {
10313                 // Tests that we properly limit inbound channels when we have the manual-channel-acceptance
10314                 // flag set and (sometimes) accept channels as 0conf.
10315                 let chanmon_cfgs = create_chanmon_cfgs(2);
10316                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
10317                 let mut settings = test_default_channel_config();
10318                 settings.manually_accept_inbound_channels = true;
10319                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[None, Some(settings)]);
10320                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
10321
10322                 // Note that create_network connects the nodes together for us
10323
10324                 nodes[0].node.create_channel(nodes[1].node.get_our_node_id(), 100_000, 0, 42, None).unwrap();
10325                 let mut open_channel_msg = get_event_msg!(nodes[0], MessageSendEvent::SendOpenChannel, nodes[1].node.get_our_node_id());
10326
10327                 // First, get us up to MAX_UNFUNDED_CHANNEL_PEERS so we can test at the edge
10328                 for _ in 0..super::MAX_UNFUNDED_CHANNEL_PEERS - 1 {
10329                         let random_pk = PublicKey::from_secret_key(&nodes[0].node.secp_ctx,
10330                                 &SecretKey::from_slice(&nodes[1].keys_manager.get_secure_random_bytes()).unwrap());
10331                         nodes[1].node.peer_connected(&random_pk, &msgs::Init {
10332                                 features: nodes[0].node.init_features(), networks: None, remote_network_address: None
10333                         }, true).unwrap();
10334
10335                         nodes[1].node.handle_open_channel(&random_pk, &open_channel_msg);
10336                         let events = nodes[1].node.get_and_clear_pending_events();
10337                         match events[0] {
10338                                 Event::OpenChannelRequest { temporary_channel_id, .. } => {
10339                                         nodes[1].node.accept_inbound_channel(&temporary_channel_id, &random_pk, 23).unwrap();
10340                                 }
10341                                 _ => panic!("Unexpected event"),
10342                         }
10343                         get_event_msg!(nodes[1], MessageSendEvent::SendAcceptChannel, random_pk);
10344                         open_channel_msg.temporary_channel_id = ChannelId::temporary_from_entropy_source(&nodes[0].keys_manager);
10345                 }
10346
10347                 // If we try to accept a channel from another peer non-0conf it will fail.
10348                 let last_random_pk = PublicKey::from_secret_key(&nodes[0].node.secp_ctx,
10349                         &SecretKey::from_slice(&nodes[1].keys_manager.get_secure_random_bytes()).unwrap());
10350                 nodes[1].node.peer_connected(&last_random_pk, &msgs::Init {
10351                         features: nodes[0].node.init_features(), networks: None, remote_network_address: None
10352                 }, true).unwrap();
10353                 nodes[1].node.handle_open_channel(&last_random_pk, &open_channel_msg);
10354                 let events = nodes[1].node.get_and_clear_pending_events();
10355                 match events[0] {
10356                         Event::OpenChannelRequest { temporary_channel_id, .. } => {
10357                                 match nodes[1].node.accept_inbound_channel(&temporary_channel_id, &last_random_pk, 23) {
10358                                         Err(APIError::APIMisuseError { err }) =>
10359                                                 assert_eq!(err, "Too many peers with unfunded channels, refusing to accept new ones"),
10360                                         _ => panic!(),
10361                                 }
10362                         }
10363                         _ => panic!("Unexpected event"),
10364                 }
10365                 assert_eq!(get_err_msg(&nodes[1], &last_random_pk).channel_id,
10366                         open_channel_msg.temporary_channel_id);
10367
10368                 // ...however if we accept the same channel 0conf it should work just fine.
10369                 nodes[1].node.handle_open_channel(&last_random_pk, &open_channel_msg);
10370                 let events = nodes[1].node.get_and_clear_pending_events();
10371                 match events[0] {
10372                         Event::OpenChannelRequest { temporary_channel_id, .. } => {
10373                                 nodes[1].node.accept_inbound_channel_from_trusted_peer_0conf(&temporary_channel_id, &last_random_pk, 23).unwrap();
10374                         }
10375                         _ => panic!("Unexpected event"),
10376                 }
10377                 get_event_msg!(nodes[1], MessageSendEvent::SendAcceptChannel, last_random_pk);
10378         }
10379
10380         #[test]
10381         fn reject_excessively_underpaying_htlcs() {
10382                 let chanmon_cfg = create_chanmon_cfgs(1);
10383                 let node_cfg = create_node_cfgs(1, &chanmon_cfg);
10384                 let node_chanmgr = create_node_chanmgrs(1, &node_cfg, &[None]);
10385                 let node = create_network(1, &node_cfg, &node_chanmgr);
10386                 let sender_intended_amt_msat = 100;
10387                 let extra_fee_msat = 10;
10388                 let hop_data = msgs::InboundOnionPayload::Receive {
10389                         amt_msat: 100,
10390                         outgoing_cltv_value: 42,
10391                         payment_metadata: None,
10392                         keysend_preimage: None,
10393                         payment_data: Some(msgs::FinalOnionHopData {
10394                                 payment_secret: PaymentSecret([0; 32]), total_msat: sender_intended_amt_msat,
10395                         }),
10396                         custom_tlvs: Vec::new(),
10397                 };
10398                 // Check that if the amount we received + the penultimate hop extra fee is less than the sender
10399                 // intended amount, we fail the payment.
10400                 if let Err(crate::ln::channelmanager::InboundOnionErr { err_code, .. }) =
10401                         node[0].node.construct_recv_pending_htlc_info(hop_data, [0; 32], PaymentHash([0; 32]),
10402                                 sender_intended_amt_msat - extra_fee_msat - 1, 42, None, true, Some(extra_fee_msat))
10403                 {
10404                         assert_eq!(err_code, 19);
10405                 } else { panic!(); }
10406
10407                 // If amt_received + extra_fee is equal to the sender intended amount, we're fine.
10408                 let hop_data = msgs::InboundOnionPayload::Receive { // This is the same payload as above, InboundOnionPayload doesn't implement Clone
10409                         amt_msat: 100,
10410                         outgoing_cltv_value: 42,
10411                         payment_metadata: None,
10412                         keysend_preimage: None,
10413                         payment_data: Some(msgs::FinalOnionHopData {
10414                                 payment_secret: PaymentSecret([0; 32]), total_msat: sender_intended_amt_msat,
10415                         }),
10416                         custom_tlvs: Vec::new(),
10417                 };
10418                 assert!(node[0].node.construct_recv_pending_htlc_info(hop_data, [0; 32], PaymentHash([0; 32]),
10419                         sender_intended_amt_msat - extra_fee_msat, 42, None, true, Some(extra_fee_msat)).is_ok());
10420         }
10421
10422         #[test]
10423         fn test_inbound_anchors_manual_acceptance() {
10424                 // Tests that we properly limit inbound channels when we have the manual-channel-acceptance
10425                 // flag set and (sometimes) accept channels as 0conf.
10426                 let mut anchors_cfg = test_default_channel_config();
10427                 anchors_cfg.channel_handshake_config.negotiate_anchors_zero_fee_htlc_tx = true;
10428
10429                 let mut anchors_manual_accept_cfg = anchors_cfg.clone();
10430                 anchors_manual_accept_cfg.manually_accept_inbound_channels = true;
10431
10432                 let chanmon_cfgs = create_chanmon_cfgs(3);
10433                 let node_cfgs = create_node_cfgs(3, &chanmon_cfgs);
10434                 let node_chanmgrs = create_node_chanmgrs(3, &node_cfgs,
10435                         &[Some(anchors_cfg.clone()), Some(anchors_cfg.clone()), Some(anchors_manual_accept_cfg.clone())]);
10436                 let nodes = create_network(3, &node_cfgs, &node_chanmgrs);
10437
10438                 nodes[0].node.create_channel(nodes[1].node.get_our_node_id(), 100_000, 0, 42, None).unwrap();
10439                 let open_channel_msg = get_event_msg!(nodes[0], MessageSendEvent::SendOpenChannel, nodes[1].node.get_our_node_id());
10440
10441                 nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
10442                 assert!(nodes[1].node.get_and_clear_pending_events().is_empty());
10443                 let msg_events = nodes[1].node.get_and_clear_pending_msg_events();
10444                 match &msg_events[0] {
10445                         MessageSendEvent::HandleError { node_id, action } => {
10446                                 assert_eq!(*node_id, nodes[0].node.get_our_node_id());
10447                                 match action {
10448                                         ErrorAction::SendErrorMessage { msg } =>
10449                                                 assert_eq!(msg.data, "No channels with anchor outputs accepted".to_owned()),
10450                                         _ => panic!("Unexpected error action"),
10451                                 }
10452                         }
10453                         _ => panic!("Unexpected event"),
10454                 }
10455
10456                 nodes[2].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
10457                 let events = nodes[2].node.get_and_clear_pending_events();
10458                 match events[0] {
10459                         Event::OpenChannelRequest { temporary_channel_id, .. } =>
10460                                 nodes[2].node.accept_inbound_channel(&temporary_channel_id, &nodes[0].node.get_our_node_id(), 23).unwrap(),
10461                         _ => panic!("Unexpected event"),
10462                 }
10463                 get_event_msg!(nodes[2], MessageSendEvent::SendAcceptChannel, nodes[0].node.get_our_node_id());
10464         }
10465
10466         #[test]
10467         fn test_anchors_zero_fee_htlc_tx_fallback() {
10468                 // Tests that if both nodes support anchors, but the remote node does not want to accept
10469                 // anchor channels at the moment, an error it sent to the local node such that it can retry
10470                 // the channel without the anchors feature.
10471                 let chanmon_cfgs = create_chanmon_cfgs(2);
10472                 let node_cfgs = create_node_cfgs(2, &chanmon_cfgs);
10473                 let mut anchors_config = test_default_channel_config();
10474                 anchors_config.channel_handshake_config.negotiate_anchors_zero_fee_htlc_tx = true;
10475                 anchors_config.manually_accept_inbound_channels = true;
10476                 let node_chanmgrs = create_node_chanmgrs(2, &node_cfgs, &[Some(anchors_config.clone()), Some(anchors_config.clone())]);
10477                 let nodes = create_network(2, &node_cfgs, &node_chanmgrs);
10478
10479                 nodes[0].node.create_channel(nodes[1].node.get_our_node_id(), 100_000, 0, 0, None).unwrap();
10480                 let open_channel_msg = get_event_msg!(nodes[0], MessageSendEvent::SendOpenChannel, nodes[1].node.get_our_node_id());
10481                 assert!(open_channel_msg.channel_type.as_ref().unwrap().supports_anchors_zero_fee_htlc_tx());
10482
10483                 nodes[1].node.handle_open_channel(&nodes[0].node.get_our_node_id(), &open_channel_msg);
10484                 let events = nodes[1].node.get_and_clear_pending_events();
10485                 match events[0] {
10486                         Event::OpenChannelRequest { temporary_channel_id, .. } => {
10487                                 nodes[1].node.force_close_broadcasting_latest_txn(&temporary_channel_id, &nodes[0].node.get_our_node_id()).unwrap();
10488                         }
10489                         _ => panic!("Unexpected event"),
10490                 }
10491
10492                 let error_msg = get_err_msg(&nodes[1], &nodes[0].node.get_our_node_id());
10493                 nodes[0].node.handle_error(&nodes[1].node.get_our_node_id(), &error_msg);
10494
10495                 let open_channel_msg = get_event_msg!(nodes[0], MessageSendEvent::SendOpenChannel, nodes[1].node.get_our_node_id());
10496                 assert!(!open_channel_msg.channel_type.unwrap().supports_anchors_zero_fee_htlc_tx());
10497
10498                 // Since nodes[1] should not have accepted the channel, it should
10499                 // not have generated any events.
10500                 assert!(nodes[1].node.get_and_clear_pending_events().is_empty());
10501         }
10502
10503         #[test]
10504         fn test_update_channel_config() {
10505                 let chanmon_cfg = create_chanmon_cfgs(2);
10506                 let node_cfg = create_node_cfgs(2, &chanmon_cfg);
10507                 let mut user_config = test_default_channel_config();
10508                 let node_chanmgr = create_node_chanmgrs(2, &node_cfg, &[Some(user_config), Some(user_config)]);
10509                 let nodes = create_network(2, &node_cfg, &node_chanmgr);
10510                 let _ = create_announced_chan_between_nodes(&nodes, 0, 1);
10511                 let channel = &nodes[0].node.list_channels()[0];
10512
10513                 nodes[0].node.update_channel_config(&channel.counterparty.node_id, &[channel.channel_id], &user_config.channel_config).unwrap();
10514                 let events = nodes[0].node.get_and_clear_pending_msg_events();
10515                 assert_eq!(events.len(), 0);
10516
10517                 user_config.channel_config.forwarding_fee_base_msat += 10;
10518                 nodes[0].node.update_channel_config(&channel.counterparty.node_id, &[channel.channel_id], &user_config.channel_config).unwrap();
10519                 assert_eq!(nodes[0].node.list_channels()[0].config.unwrap().forwarding_fee_base_msat, user_config.channel_config.forwarding_fee_base_msat);
10520                 let events = nodes[0].node.get_and_clear_pending_msg_events();
10521                 assert_eq!(events.len(), 1);
10522                 match &events[0] {
10523                         MessageSendEvent::BroadcastChannelUpdate { .. } => {},
10524                         _ => panic!("expected BroadcastChannelUpdate event"),
10525                 }
10526
10527                 nodes[0].node.update_partial_channel_config(&channel.counterparty.node_id, &[channel.channel_id], &ChannelConfigUpdate::default()).unwrap();
10528                 let events = nodes[0].node.get_and_clear_pending_msg_events();
10529                 assert_eq!(events.len(), 0);
10530
10531                 let new_cltv_expiry_delta = user_config.channel_config.cltv_expiry_delta + 6;
10532                 nodes[0].node.update_partial_channel_config(&channel.counterparty.node_id, &[channel.channel_id], &ChannelConfigUpdate {
10533                         cltv_expiry_delta: Some(new_cltv_expiry_delta),
10534                         ..Default::default()
10535                 }).unwrap();
10536                 assert_eq!(nodes[0].node.list_channels()[0].config.unwrap().cltv_expiry_delta, new_cltv_expiry_delta);
10537                 let events = nodes[0].node.get_and_clear_pending_msg_events();
10538                 assert_eq!(events.len(), 1);
10539                 match &events[0] {
10540                         MessageSendEvent::BroadcastChannelUpdate { .. } => {},
10541                         _ => panic!("expected BroadcastChannelUpdate event"),
10542                 }
10543
10544                 let new_fee = user_config.channel_config.forwarding_fee_proportional_millionths + 100;
10545                 nodes[0].node.update_partial_channel_config(&channel.counterparty.node_id, &[channel.channel_id], &ChannelConfigUpdate {
10546                         forwarding_fee_proportional_millionths: Some(new_fee),
10547                         ..Default::default()
10548                 }).unwrap();
10549                 assert_eq!(nodes[0].node.list_channels()[0].config.unwrap().cltv_expiry_delta, new_cltv_expiry_delta);
10550                 assert_eq!(nodes[0].node.list_channels()[0].config.unwrap().forwarding_fee_proportional_millionths, new_fee);
10551                 let events = nodes[0].node.get_and_clear_pending_msg_events();
10552                 assert_eq!(events.len(), 1);
10553                 match &events[0] {
10554                         MessageSendEvent::BroadcastChannelUpdate { .. } => {},
10555                         _ => panic!("expected BroadcastChannelUpdate event"),
10556                 }
10557
10558                 // If we provide a channel_id not associated with the peer, we should get an error and no updates
10559                 // should be applied to ensure update atomicity as specified in the API docs.
10560                 let bad_channel_id = ChannelId::v1_from_funding_txid(&[10; 32], 10);
10561                 let current_fee = nodes[0].node.list_channels()[0].config.unwrap().forwarding_fee_proportional_millionths;
10562                 let new_fee = current_fee + 100;
10563                 assert!(
10564                         matches!(
10565                                 nodes[0].node.update_partial_channel_config(&channel.counterparty.node_id, &[channel.channel_id, bad_channel_id], &ChannelConfigUpdate {
10566                                         forwarding_fee_proportional_millionths: Some(new_fee),
10567                                         ..Default::default()
10568                                 }),
10569                                 Err(APIError::ChannelUnavailable { err: _ }),
10570                         )
10571                 );
10572                 // Check that the fee hasn't changed for the channel that exists.
10573                 assert_eq!(nodes[0].node.list_channels()[0].config.unwrap().forwarding_fee_proportional_millionths, current_fee);
10574                 let events = nodes[0].node.get_and_clear_pending_msg_events();
10575                 assert_eq!(events.len(), 0);
10576         }
10577
10578         #[test]
10579         fn test_payment_display() {
10580                 let payment_id = PaymentId([42; 32]);
10581                 assert_eq!(format!("{}", &payment_id), "2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a");
10582                 let payment_hash = PaymentHash([42; 32]);
10583                 assert_eq!(format!("{}", &payment_hash), "2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a");
10584                 let payment_preimage = PaymentPreimage([42; 32]);
10585                 assert_eq!(format!("{}", &payment_preimage), "2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a2a");
10586         }
10587 }
10588
10589 #[cfg(ldk_bench)]
10590 pub mod bench {
10591         use crate::chain::Listen;
10592         use crate::chain::chainmonitor::{ChainMonitor, Persist};
10593         use crate::sign::{KeysManager, InMemorySigner};
10594         use crate::events::{Event, MessageSendEvent, MessageSendEventsProvider};
10595         use crate::ln::channelmanager::{BestBlock, ChainParameters, ChannelManager, PaymentHash, PaymentPreimage, PaymentId, RecipientOnionFields, Retry};
10596         use crate::ln::functional_test_utils::*;
10597         use crate::ln::msgs::{ChannelMessageHandler, Init};
10598         use crate::routing::gossip::NetworkGraph;
10599         use crate::routing::router::{PaymentParameters, RouteParameters};
10600         use crate::util::test_utils;
10601         use crate::util::config::{UserConfig, MaxDustHTLCExposure};
10602
10603         use bitcoin::hashes::Hash;
10604         use bitcoin::hashes::sha256::Hash as Sha256;
10605         use bitcoin::{Block, BlockHeader, PackedLockTime, Transaction, TxMerkleNode, TxOut};
10606
10607         use crate::sync::{Arc, Mutex, RwLock};
10608
10609         use criterion::Criterion;
10610
10611         type Manager<'a, P> = ChannelManager<
10612                 &'a ChainMonitor<InMemorySigner, &'a test_utils::TestChainSource,
10613                         &'a test_utils::TestBroadcaster, &'a test_utils::TestFeeEstimator,
10614                         &'a test_utils::TestLogger, &'a P>,
10615                 &'a test_utils::TestBroadcaster, &'a KeysManager, &'a KeysManager, &'a KeysManager,
10616                 &'a test_utils::TestFeeEstimator, &'a test_utils::TestRouter<'a>,
10617                 &'a test_utils::TestLogger>;
10618
10619         struct ANodeHolder<'node_cfg, 'chan_mon_cfg: 'node_cfg, P: Persist<InMemorySigner>> {
10620                 node: &'node_cfg Manager<'chan_mon_cfg, P>,
10621         }
10622         impl<'node_cfg, 'chan_mon_cfg: 'node_cfg, P: Persist<InMemorySigner>> NodeHolder for ANodeHolder<'node_cfg, 'chan_mon_cfg, P> {
10623                 type CM = Manager<'chan_mon_cfg, P>;
10624                 #[inline]
10625                 fn node(&self) -> &Manager<'chan_mon_cfg, P> { self.node }
10626                 #[inline]
10627                 fn chain_monitor(&self) -> Option<&test_utils::TestChainMonitor> { None }
10628         }
10629
10630         pub fn bench_sends(bench: &mut Criterion) {
10631                 bench_two_sends(bench, "bench_sends", test_utils::TestPersister::new(), test_utils::TestPersister::new());
10632         }
10633
10634         pub fn bench_two_sends<P: Persist<InMemorySigner>>(bench: &mut Criterion, bench_name: &str, persister_a: P, persister_b: P) {
10635                 // Do a simple benchmark of sending a payment back and forth between two nodes.
10636                 // Note that this is unrealistic as each payment send will require at least two fsync
10637                 // calls per node.
10638                 let network = bitcoin::Network::Testnet;
10639                 let genesis_block = bitcoin::blockdata::constants::genesis_block(network);
10640
10641                 let tx_broadcaster = test_utils::TestBroadcaster::new(network);
10642                 let fee_estimator = test_utils::TestFeeEstimator { sat_per_kw: Mutex::new(253) };
10643                 let logger_a = test_utils::TestLogger::with_id("node a".to_owned());
10644                 let scorer = RwLock::new(test_utils::TestScorer::new());
10645                 let router = test_utils::TestRouter::new(Arc::new(NetworkGraph::new(network, &logger_a)), &scorer);
10646
10647                 let mut config: UserConfig = Default::default();
10648                 config.channel_config.max_dust_htlc_exposure = MaxDustHTLCExposure::FeeRateMultiplier(5_000_000 / 253);
10649                 config.channel_handshake_config.minimum_depth = 1;
10650
10651                 let chain_monitor_a = ChainMonitor::new(None, &tx_broadcaster, &logger_a, &fee_estimator, &persister_a);
10652                 let seed_a = [1u8; 32];
10653                 let keys_manager_a = KeysManager::new(&seed_a, 42, 42);
10654                 let node_a = ChannelManager::new(&fee_estimator, &chain_monitor_a, &tx_broadcaster, &router, &logger_a, &keys_manager_a, &keys_manager_a, &keys_manager_a, config.clone(), ChainParameters {
10655                         network,
10656                         best_block: BestBlock::from_network(network),
10657                 }, genesis_block.header.time);
10658                 let node_a_holder = ANodeHolder { node: &node_a };
10659
10660                 let logger_b = test_utils::TestLogger::with_id("node a".to_owned());
10661                 let chain_monitor_b = ChainMonitor::new(None, &tx_broadcaster, &logger_a, &fee_estimator, &persister_b);
10662                 let seed_b = [2u8; 32];
10663                 let keys_manager_b = KeysManager::new(&seed_b, 42, 42);
10664                 let node_b = ChannelManager::new(&fee_estimator, &chain_monitor_b, &tx_broadcaster, &router, &logger_b, &keys_manager_b, &keys_manager_b, &keys_manager_b, config.clone(), ChainParameters {
10665                         network,
10666                         best_block: BestBlock::from_network(network),
10667                 }, genesis_block.header.time);
10668                 let node_b_holder = ANodeHolder { node: &node_b };
10669
10670                 node_a.peer_connected(&node_b.get_our_node_id(), &Init {
10671                         features: node_b.init_features(), networks: None, remote_network_address: None
10672                 }, true).unwrap();
10673                 node_b.peer_connected(&node_a.get_our_node_id(), &Init {
10674                         features: node_a.init_features(), networks: None, remote_network_address: None
10675                 }, false).unwrap();
10676                 node_a.create_channel(node_b.get_our_node_id(), 8_000_000, 100_000_000, 42, None).unwrap();
10677                 node_b.handle_open_channel(&node_a.get_our_node_id(), &get_event_msg!(node_a_holder, MessageSendEvent::SendOpenChannel, node_b.get_our_node_id()));
10678                 node_a.handle_accept_channel(&node_b.get_our_node_id(), &get_event_msg!(node_b_holder, MessageSendEvent::SendAcceptChannel, node_a.get_our_node_id()));
10679
10680                 let tx;
10681                 if let Event::FundingGenerationReady { temporary_channel_id, output_script, .. } = get_event!(node_a_holder, Event::FundingGenerationReady) {
10682                         tx = Transaction { version: 2, lock_time: PackedLockTime::ZERO, input: Vec::new(), output: vec![TxOut {
10683                                 value: 8_000_000, script_pubkey: output_script,
10684                         }]};
10685                         node_a.funding_transaction_generated(&temporary_channel_id, &node_b.get_our_node_id(), tx.clone()).unwrap();
10686                 } else { panic!(); }
10687
10688                 node_b.handle_funding_created(&node_a.get_our_node_id(), &get_event_msg!(node_a_holder, MessageSendEvent::SendFundingCreated, node_b.get_our_node_id()));
10689                 let events_b = node_b.get_and_clear_pending_events();
10690                 assert_eq!(events_b.len(), 1);
10691                 match events_b[0] {
10692                         Event::ChannelPending{ ref counterparty_node_id, .. } => {
10693                                 assert_eq!(*counterparty_node_id, node_a.get_our_node_id());
10694                         },
10695                         _ => panic!("Unexpected event"),
10696                 }
10697
10698                 node_a.handle_funding_signed(&node_b.get_our_node_id(), &get_event_msg!(node_b_holder, MessageSendEvent::SendFundingSigned, node_a.get_our_node_id()));
10699                 let events_a = node_a.get_and_clear_pending_events();
10700                 assert_eq!(events_a.len(), 1);
10701                 match events_a[0] {
10702                         Event::ChannelPending{ ref counterparty_node_id, .. } => {
10703                                 assert_eq!(*counterparty_node_id, node_b.get_our_node_id());
10704                         },
10705                         _ => panic!("Unexpected event"),
10706                 }
10707
10708                 assert_eq!(&tx_broadcaster.txn_broadcasted.lock().unwrap()[..], &[tx.clone()]);
10709
10710                 let block = create_dummy_block(BestBlock::from_network(network).block_hash(), 42, vec![tx]);
10711                 Listen::block_connected(&node_a, &block, 1);
10712                 Listen::block_connected(&node_b, &block, 1);
10713
10714                 node_a.handle_channel_ready(&node_b.get_our_node_id(), &get_event_msg!(node_b_holder, MessageSendEvent::SendChannelReady, node_a.get_our_node_id()));
10715                 let msg_events = node_a.get_and_clear_pending_msg_events();
10716                 assert_eq!(msg_events.len(), 2);
10717                 match msg_events[0] {
10718                         MessageSendEvent::SendChannelReady { ref msg, .. } => {
10719                                 node_b.handle_channel_ready(&node_a.get_our_node_id(), msg);
10720                                 get_event_msg!(node_b_holder, MessageSendEvent::SendChannelUpdate, node_a.get_our_node_id());
10721                         },
10722                         _ => panic!(),
10723                 }
10724                 match msg_events[1] {
10725                         MessageSendEvent::SendChannelUpdate { .. } => {},
10726                         _ => panic!(),
10727                 }
10728
10729                 let events_a = node_a.get_and_clear_pending_events();
10730                 assert_eq!(events_a.len(), 1);
10731                 match events_a[0] {
10732                         Event::ChannelReady{ ref counterparty_node_id, .. } => {
10733                                 assert_eq!(*counterparty_node_id, node_b.get_our_node_id());
10734                         },
10735                         _ => panic!("Unexpected event"),
10736                 }
10737
10738                 let events_b = node_b.get_and_clear_pending_events();
10739                 assert_eq!(events_b.len(), 1);
10740                 match events_b[0] {
10741                         Event::ChannelReady{ ref counterparty_node_id, .. } => {
10742                                 assert_eq!(*counterparty_node_id, node_a.get_our_node_id());
10743                         },
10744                         _ => panic!("Unexpected event"),
10745                 }
10746
10747                 let mut payment_count: u64 = 0;
10748                 macro_rules! send_payment {
10749                         ($node_a: expr, $node_b: expr) => {
10750                                 let payment_params = PaymentParameters::from_node_id($node_b.get_our_node_id(), TEST_FINAL_CLTV)
10751                                         .with_bolt11_features($node_b.invoice_features()).unwrap();
10752                                 let mut payment_preimage = PaymentPreimage([0; 32]);
10753                                 payment_preimage.0[0..8].copy_from_slice(&payment_count.to_le_bytes());
10754                                 payment_count += 1;
10755                                 let payment_hash = PaymentHash(Sha256::hash(&payment_preimage.0[..]).into_inner());
10756                                 let payment_secret = $node_b.create_inbound_payment_for_hash(payment_hash, None, 7200, None).unwrap();
10757
10758                                 $node_a.send_payment(payment_hash, RecipientOnionFields::secret_only(payment_secret),
10759                                         PaymentId(payment_hash.0),
10760                                         RouteParameters::from_payment_params_and_value(payment_params, 10_000),
10761                                         Retry::Attempts(0)).unwrap();
10762                                 let payment_event = SendEvent::from_event($node_a.get_and_clear_pending_msg_events().pop().unwrap());
10763                                 $node_b.handle_update_add_htlc(&$node_a.get_our_node_id(), &payment_event.msgs[0]);
10764                                 $node_b.handle_commitment_signed(&$node_a.get_our_node_id(), &payment_event.commitment_msg);
10765                                 let (raa, cs) = get_revoke_commit_msgs(&ANodeHolder { node: &$node_b }, &$node_a.get_our_node_id());
10766                                 $node_a.handle_revoke_and_ack(&$node_b.get_our_node_id(), &raa);
10767                                 $node_a.handle_commitment_signed(&$node_b.get_our_node_id(), &cs);
10768                                 $node_b.handle_revoke_and_ack(&$node_a.get_our_node_id(), &get_event_msg!(ANodeHolder { node: &$node_a }, MessageSendEvent::SendRevokeAndACK, $node_b.get_our_node_id()));
10769
10770                                 expect_pending_htlcs_forwardable!(ANodeHolder { node: &$node_b });
10771                                 expect_payment_claimable!(ANodeHolder { node: &$node_b }, payment_hash, payment_secret, 10_000);
10772                                 $node_b.claim_funds(payment_preimage);
10773                                 expect_payment_claimed!(ANodeHolder { node: &$node_b }, payment_hash, 10_000);
10774
10775                                 match $node_b.get_and_clear_pending_msg_events().pop().unwrap() {
10776                                         MessageSendEvent::UpdateHTLCs { node_id, updates } => {
10777                                                 assert_eq!(node_id, $node_a.get_our_node_id());
10778                                                 $node_a.handle_update_fulfill_htlc(&$node_b.get_our_node_id(), &updates.update_fulfill_htlcs[0]);
10779                                                 $node_a.handle_commitment_signed(&$node_b.get_our_node_id(), &updates.commitment_signed);
10780                                         },
10781                                         _ => panic!("Failed to generate claim event"),
10782                                 }
10783
10784                                 let (raa, cs) = get_revoke_commit_msgs(&ANodeHolder { node: &$node_a }, &$node_b.get_our_node_id());
10785                                 $node_b.handle_revoke_and_ack(&$node_a.get_our_node_id(), &raa);
10786                                 $node_b.handle_commitment_signed(&$node_a.get_our_node_id(), &cs);
10787                                 $node_a.handle_revoke_and_ack(&$node_b.get_our_node_id(), &get_event_msg!(ANodeHolder { node: &$node_b }, MessageSendEvent::SendRevokeAndACK, $node_a.get_our_node_id()));
10788
10789                                 expect_payment_sent!(ANodeHolder { node: &$node_a }, payment_preimage);
10790                         }
10791                 }
10792
10793                 bench.bench_function(bench_name, |b| b.iter(|| {
10794                         send_payment!(node_a, node_b);
10795                         send_payment!(node_b, node_a);
10796                 }));
10797         }
10798 }