Drop now-unused `T: Time` bound on `ProbabilisticScorer`
[rust-lightning] / lightning / src / routing / scoring.rs
1 // This file is Copyright its original authors, visible in version control
2 // history.
3 //
4 // This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
5 // or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
6 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
7 // You may not use this file except in accordance with one or both of these
8 // licenses.
9
10 //! Utilities for scoring payment channels.
11 //!
12 //! [`ProbabilisticScorer`] may be given to [`find_route`] to score payment channels during path
13 //! finding when a custom [`ScoreLookUp`] implementation is not needed.
14 //!
15 //! # Example
16 //!
17 //! ```
18 //! # extern crate bitcoin;
19 //! #
20 //! # use lightning::routing::gossip::NetworkGraph;
21 //! # use lightning::routing::router::{RouteParameters, find_route};
22 //! # use lightning::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringFeeParameters, ProbabilisticScoringDecayParameters};
23 //! # use lightning::sign::KeysManager;
24 //! # use lightning::util::logger::{Logger, Record};
25 //! # use bitcoin::secp256k1::PublicKey;
26 //! #
27 //! # struct FakeLogger {};
28 //! # impl Logger for FakeLogger {
29 //! #     fn log(&self, record: Record) { unimplemented!() }
30 //! # }
31 //! # fn find_scored_route(payer: PublicKey, route_params: RouteParameters, network_graph: NetworkGraph<&FakeLogger>) {
32 //! # let logger = FakeLogger {};
33 //! #
34 //! // Use the default channel penalties.
35 //! let params = ProbabilisticScoringFeeParameters::default();
36 //! let decay_params = ProbabilisticScoringDecayParameters::default();
37 //! let scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
38 //!
39 //! // Or use custom channel penalties.
40 //! let params = ProbabilisticScoringFeeParameters {
41 //!     liquidity_penalty_multiplier_msat: 2 * 1000,
42 //!     ..ProbabilisticScoringFeeParameters::default()
43 //! };
44 //! let decay_params = ProbabilisticScoringDecayParameters::default();
45 //! let scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
46 //! # let random_seed_bytes = [42u8; 32];
47 //!
48 //! let route = find_route(&payer, &route_params, &network_graph, None, &logger, &scorer, &params, &random_seed_bytes);
49 //! # }
50 //! ```
51 //!
52 //! # Note
53 //!
54 //! Persisting when built with feature `no-std` and restoring without it, or vice versa, uses
55 //! different types and thus is undefined.
56 //!
57 //! [`find_route`]: crate::routing::router::find_route
58
59 use crate::ln::msgs::DecodeError;
60 use crate::routing::gossip::{EffectiveCapacity, NetworkGraph, NodeId};
61 use crate::routing::router::{Path, CandidateRouteHop};
62 use crate::util::ser::{Readable, ReadableArgs, Writeable, Writer};
63 use crate::util::logger::Logger;
64
65 use crate::prelude::*;
66 use core::{cmp, fmt};
67 use core::cell::{RefCell, RefMut, Ref};
68 use core::convert::TryInto;
69 use core::ops::{Deref, DerefMut};
70 use core::time::Duration;
71 use crate::io::{self, Read};
72 use crate::sync::{Mutex, MutexGuard, RwLock, RwLockReadGuard, RwLockWriteGuard};
73
74 /// We define Score ever-so-slightly differently based on whether we are being built for C bindings
75 /// or not. For users, `LockableScore` must somehow be writeable to disk. For Rust users, this is
76 /// no problem - you move a `Score` that implements `Writeable` into a `Mutex`, lock it, and now
77 /// you have the original, concrete, `Score` type, which presumably implements `Writeable`.
78 ///
79 /// For C users, once you've moved the `Score` into a `LockableScore` all you have after locking it
80 /// is an opaque trait object with an opaque pointer with no type info. Users could take the unsafe
81 /// approach of blindly casting that opaque pointer to a concrete type and calling `Writeable` from
82 /// there, but other languages downstream of the C bindings (e.g. Java) can't even do that.
83 /// Instead, we really want `Score` and `LockableScore` to implement `Writeable` directly, which we
84 /// do here by defining `Score` differently for `cfg(c_bindings)`.
85 macro_rules! define_score { ($($supertrait: path)*) => {
86 /// An interface used to score payment channels for path finding.
87 ///
88 /// `ScoreLookUp` is used to determine the penalty for a given channel.
89 ///
90 /// Scoring is in terms of fees willing to be paid in order to avoid routing through a channel.
91 pub trait ScoreLookUp {
92         /// A configurable type which should contain various passed-in parameters for configuring the scorer,
93         /// on a per-routefinding-call basis through to the scorer methods,
94         /// which are used to determine the parameters for the suitability of channels for use.
95         type ScoreParams;
96         /// Returns the fee in msats willing to be paid to avoid routing `send_amt_msat` through the
97         /// given channel in the direction from `source` to `target`.
98         ///
99         /// The channel's capacity (less any other MPP parts that are also being considered for use in
100         /// the same payment) is given by `capacity_msat`. It may be determined from various sources
101         /// such as a chain data, network gossip, or invoice hints. For invoice hints, a capacity near
102         /// [`u64::max_value`] is given to indicate sufficient capacity for the invoice's full amount.
103         /// Thus, implementations should be overflow-safe.
104         fn channel_penalty_msat(
105                 &self, candidate: &CandidateRouteHop, usage: ChannelUsage, score_params: &Self::ScoreParams
106         ) -> u64;
107 }
108
109 /// `ScoreUpdate` is used to update the scorer's internal state after a payment attempt.
110 pub trait ScoreUpdate {
111         /// Handles updating channel penalties after failing to route through a channel.
112         fn payment_path_failed(&mut self, path: &Path, short_channel_id: u64, duration_since_epoch: Duration);
113
114         /// Handles updating channel penalties after successfully routing along a path.
115         fn payment_path_successful(&mut self, path: &Path, duration_since_epoch: Duration);
116
117         /// Handles updating channel penalties after a probe over the given path failed.
118         fn probe_failed(&mut self, path: &Path, short_channel_id: u64, duration_since_epoch: Duration);
119
120         /// Handles updating channel penalties after a probe over the given path succeeded.
121         fn probe_successful(&mut self, path: &Path, duration_since_epoch: Duration);
122
123         /// Scorers may wish to reduce their certainty of channel liquidity information over time.
124         /// Thus, this method is provided to allow scorers to observe the passage of time - the holder
125         /// of this object should call this method regularly (generally via the
126         /// `lightning-background-processor` crate).
127         fn time_passed(&mut self, duration_since_epoch: Duration);
128 }
129
130 /// A trait which can both lookup and update routing channel penalty scores.
131 ///
132 /// This is used in places where both bounds are required and implemented for all types which
133 /// implement [`ScoreLookUp`] and [`ScoreUpdate`].
134 ///
135 /// Bindings users may need to manually implement this for their custom scoring implementations.
136 pub trait Score : ScoreLookUp + ScoreUpdate $(+ $supertrait)* {}
137
138 #[cfg(not(c_bindings))]
139 impl<T: ScoreLookUp + ScoreUpdate $(+ $supertrait)*> Score for T {}
140
141 #[cfg(not(c_bindings))]
142 impl<S: ScoreLookUp, T: Deref<Target=S>> ScoreLookUp for T {
143         type ScoreParams = S::ScoreParams;
144         fn channel_penalty_msat(
145                 &self, candidate: &CandidateRouteHop, usage: ChannelUsage, score_params: &Self::ScoreParams
146         ) -> u64 {
147                 self.deref().channel_penalty_msat(candidate, usage, score_params)
148         }
149 }
150
151 #[cfg(not(c_bindings))]
152 impl<S: ScoreUpdate, T: DerefMut<Target=S>> ScoreUpdate for T {
153         fn payment_path_failed(&mut self, path: &Path, short_channel_id: u64, duration_since_epoch: Duration) {
154                 self.deref_mut().payment_path_failed(path, short_channel_id, duration_since_epoch)
155         }
156
157         fn payment_path_successful(&mut self, path: &Path, duration_since_epoch: Duration) {
158                 self.deref_mut().payment_path_successful(path, duration_since_epoch)
159         }
160
161         fn probe_failed(&mut self, path: &Path, short_channel_id: u64, duration_since_epoch: Duration) {
162                 self.deref_mut().probe_failed(path, short_channel_id, duration_since_epoch)
163         }
164
165         fn probe_successful(&mut self, path: &Path, duration_since_epoch: Duration) {
166                 self.deref_mut().probe_successful(path, duration_since_epoch)
167         }
168
169         fn time_passed(&mut self, duration_since_epoch: Duration) {
170                 self.deref_mut().time_passed(duration_since_epoch)
171         }
172 }
173 } }
174
175 #[cfg(c_bindings)]
176 define_score!(Writeable);
177
178 #[cfg(not(c_bindings))]
179 define_score!();
180
181 /// A scorer that is accessed under a lock.
182 ///
183 /// Needed so that calls to [`ScoreLookUp::channel_penalty_msat`] in [`find_route`] can be made while
184 /// having shared ownership of a scorer but without requiring internal locking in [`ScoreUpdate`]
185 /// implementations. Internal locking would be detrimental to route finding performance and could
186 /// result in [`ScoreLookUp::channel_penalty_msat`] returning a different value for the same channel.
187 ///
188 /// [`find_route`]: crate::routing::router::find_route
189 pub trait LockableScore<'a> {
190         /// The [`ScoreUpdate`] type.
191         type ScoreUpdate: 'a + ScoreUpdate;
192         /// The [`ScoreLookUp`] type.
193         type ScoreLookUp: 'a + ScoreLookUp;
194
195         /// The write locked [`ScoreUpdate`] type.
196         type WriteLocked: DerefMut<Target = Self::ScoreUpdate> + Sized;
197
198         /// The read locked [`ScoreLookUp`] type.
199         type ReadLocked: Deref<Target = Self::ScoreLookUp> + Sized;
200
201         /// Returns read locked scorer.
202         fn read_lock(&'a self) -> Self::ReadLocked;
203
204         /// Returns write locked scorer.
205         fn write_lock(&'a self) -> Self::WriteLocked;
206 }
207
208 /// Refers to a scorer that is accessible under lock and also writeable to disk
209 ///
210 /// We need this trait to be able to pass in a scorer to `lightning-background-processor` that will enable us to
211 /// use the Persister to persist it.
212 pub trait WriteableScore<'a>: LockableScore<'a> + Writeable {}
213
214 #[cfg(not(c_bindings))]
215 impl<'a, T> WriteableScore<'a> for T where T: LockableScore<'a> + Writeable {}
216 #[cfg(not(c_bindings))]
217 impl<'a, T: Score + 'a> LockableScore<'a> for Mutex<T> {
218         type ScoreUpdate = T;
219         type ScoreLookUp = T;
220
221         type WriteLocked = MutexGuard<'a, Self::ScoreUpdate>;
222         type ReadLocked = MutexGuard<'a, Self::ScoreLookUp>;
223
224         fn read_lock(&'a self) -> Self::ReadLocked {
225                 Mutex::lock(self).unwrap()
226         }
227
228         fn write_lock(&'a self) -> Self::WriteLocked {
229                 Mutex::lock(self).unwrap()
230         }
231 }
232
233 #[cfg(not(c_bindings))]
234 impl<'a, T: Score + 'a> LockableScore<'a> for RefCell<T> {
235         type ScoreUpdate = T;
236         type ScoreLookUp = T;
237
238         type WriteLocked = RefMut<'a, Self::ScoreUpdate>;
239         type ReadLocked = Ref<'a, Self::ScoreLookUp>;
240
241         fn write_lock(&'a self) -> Self::WriteLocked {
242                 self.borrow_mut()
243         }
244
245         fn read_lock(&'a self) -> Self::ReadLocked {
246                 self.borrow()
247         }
248 }
249
250 #[cfg(not(c_bindings))]
251 impl<'a, T: Score + 'a> LockableScore<'a> for RwLock<T> {
252         type ScoreUpdate = T;
253         type ScoreLookUp = T;
254
255         type WriteLocked = RwLockWriteGuard<'a, Self::ScoreLookUp>;
256         type ReadLocked = RwLockReadGuard<'a, Self::ScoreUpdate>;
257
258         fn read_lock(&'a self) -> Self::ReadLocked {
259                 RwLock::read(self).unwrap()
260         }
261
262         fn write_lock(&'a self) -> Self::WriteLocked {
263                 RwLock::write(self).unwrap()
264         }
265 }
266
267 #[cfg(c_bindings)]
268 /// A concrete implementation of [`LockableScore`] which supports multi-threading.
269 pub struct MultiThreadedLockableScore<T: Score> {
270         score: RwLock<T>,
271 }
272
273 #[cfg(c_bindings)]
274 impl<'a, T: Score + 'a> LockableScore<'a> for MultiThreadedLockableScore<T> {
275         type ScoreUpdate = T;
276         type ScoreLookUp = T;
277         type WriteLocked = MultiThreadedScoreLockWrite<'a, Self::ScoreUpdate>;
278         type ReadLocked = MultiThreadedScoreLockRead<'a, Self::ScoreLookUp>;
279
280         fn read_lock(&'a self) -> Self::ReadLocked {
281                 MultiThreadedScoreLockRead(self.score.read().unwrap())
282         }
283
284         fn write_lock(&'a self) -> Self::WriteLocked {
285                 MultiThreadedScoreLockWrite(self.score.write().unwrap())
286         }
287 }
288
289 #[cfg(c_bindings)]
290 impl<T: Score> Writeable for MultiThreadedLockableScore<T> {
291         fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
292                 self.score.read().unwrap().write(writer)
293         }
294 }
295
296 #[cfg(c_bindings)]
297 impl<'a, T: Score + 'a> WriteableScore<'a> for MultiThreadedLockableScore<T> {}
298
299 #[cfg(c_bindings)]
300 impl<T: Score> MultiThreadedLockableScore<T> {
301         /// Creates a new [`MultiThreadedLockableScore`] given an underlying [`Score`].
302         pub fn new(score: T) -> Self {
303                 MultiThreadedLockableScore { score: RwLock::new(score) }
304         }
305 }
306
307 #[cfg(c_bindings)]
308 /// A locked `MultiThreadedLockableScore`.
309 pub struct MultiThreadedScoreLockRead<'a, T: Score>(RwLockReadGuard<'a, T>);
310
311 #[cfg(c_bindings)]
312 /// A locked `MultiThreadedLockableScore`.
313 pub struct MultiThreadedScoreLockWrite<'a, T: Score>(RwLockWriteGuard<'a, T>);
314
315 #[cfg(c_bindings)]
316 impl<'a, T: 'a + Score> Deref for MultiThreadedScoreLockRead<'a, T> {
317         type Target = T;
318
319         fn deref(&self) -> &Self::Target {
320                 self.0.deref()
321         }
322 }
323
324 #[cfg(c_bindings)]
325 impl<'a, T: Score> ScoreLookUp for MultiThreadedScoreLockRead<'a, T> {
326         type ScoreParams = T::ScoreParams;
327         fn channel_penalty_msat(&self, candidate:&CandidateRouteHop, usage: ChannelUsage, score_params: &Self::ScoreParams
328         ) -> u64 {
329                 self.0.channel_penalty_msat(candidate, usage, score_params)
330         }
331 }
332
333 #[cfg(c_bindings)]
334 impl<'a, T: Score> Writeable for MultiThreadedScoreLockWrite<'a, T> {
335         fn write<W: Writer>(&self, writer: &mut W) -> Result<(), io::Error> {
336                 self.0.write(writer)
337         }
338 }
339
340 #[cfg(c_bindings)]
341 impl<'a, T: 'a + Score> Deref for MultiThreadedScoreLockWrite<'a, T> {
342         type Target = T;
343
344         fn deref(&self) -> &Self::Target {
345                 self.0.deref()
346         }
347 }
348
349 #[cfg(c_bindings)]
350 impl<'a, T: 'a + Score> DerefMut for MultiThreadedScoreLockWrite<'a, T> {
351         fn deref_mut(&mut self) -> &mut Self::Target {
352                 self.0.deref_mut()
353         }
354 }
355
356 #[cfg(c_bindings)]
357 impl<'a, T: Score> ScoreUpdate for MultiThreadedScoreLockWrite<'a, T> {
358         fn payment_path_failed(&mut self, path: &Path, short_channel_id: u64, duration_since_epoch: Duration) {
359                 self.0.payment_path_failed(path, short_channel_id, duration_since_epoch)
360         }
361
362         fn payment_path_successful(&mut self, path: &Path, duration_since_epoch: Duration) {
363                 self.0.payment_path_successful(path, duration_since_epoch)
364         }
365
366         fn probe_failed(&mut self, path: &Path, short_channel_id: u64, duration_since_epoch: Duration) {
367                 self.0.probe_failed(path, short_channel_id, duration_since_epoch)
368         }
369
370         fn probe_successful(&mut self, path: &Path, duration_since_epoch: Duration) {
371                 self.0.probe_successful(path, duration_since_epoch)
372         }
373
374         fn time_passed(&mut self, duration_since_epoch: Duration) {
375                 self.0.time_passed(duration_since_epoch)
376         }
377 }
378
379
380 /// Proposed use of a channel passed as a parameter to [`ScoreLookUp::channel_penalty_msat`].
381 #[derive(Clone, Copy, Debug, PartialEq)]
382 pub struct ChannelUsage {
383         /// The amount to send through the channel, denominated in millisatoshis.
384         pub amount_msat: u64,
385
386         /// Total amount, denominated in millisatoshis, already allocated to send through the channel
387         /// as part of a multi-path payment.
388         pub inflight_htlc_msat: u64,
389
390         /// The effective capacity of the channel.
391         pub effective_capacity: EffectiveCapacity,
392 }
393
394 #[derive(Clone)]
395 /// [`ScoreLookUp`] implementation that uses a fixed penalty.
396 pub struct FixedPenaltyScorer {
397         penalty_msat: u64,
398 }
399
400 impl FixedPenaltyScorer {
401         /// Creates a new scorer using `penalty_msat`.
402         pub fn with_penalty(penalty_msat: u64) -> Self {
403                 Self { penalty_msat }
404         }
405 }
406
407 impl ScoreLookUp for FixedPenaltyScorer {
408         type ScoreParams = ();
409         fn channel_penalty_msat(&self, _: &CandidateRouteHop, _: ChannelUsage, _score_params: &Self::ScoreParams) -> u64 {
410                 self.penalty_msat
411         }
412 }
413
414 impl ScoreUpdate for FixedPenaltyScorer {
415         fn payment_path_failed(&mut self, _path: &Path, _short_channel_id: u64, _duration_since_epoch: Duration) {}
416
417         fn payment_path_successful(&mut self, _path: &Path, _duration_since_epoch: Duration) {}
418
419         fn probe_failed(&mut self, _path: &Path, _short_channel_id: u64, _duration_since_epoch: Duration) {}
420
421         fn probe_successful(&mut self, _path: &Path, _duration_since_epoch: Duration) {}
422
423         fn time_passed(&mut self, _duration_since_epoch: Duration) {}
424 }
425
426 impl Writeable for FixedPenaltyScorer {
427         #[inline]
428         fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
429                 write_tlv_fields!(w, {});
430                 Ok(())
431         }
432 }
433
434 impl ReadableArgs<u64> for FixedPenaltyScorer {
435         #[inline]
436         fn read<R: Read>(r: &mut R, penalty_msat: u64) -> Result<Self, DecodeError> {
437                 read_tlv_fields!(r, {});
438                 Ok(Self { penalty_msat })
439         }
440 }
441
442 /// [`ScoreLookUp`] implementation using channel success probability distributions.
443 ///
444 /// Channels are tracked with upper and lower liquidity bounds - when an HTLC fails at a channel,
445 /// we learn that the upper-bound on the available liquidity is lower than the amount of the HTLC.
446 /// When a payment is forwarded through a channel (but fails later in the route), we learn the
447 /// lower-bound on the channel's available liquidity must be at least the value of the HTLC.
448 ///
449 /// These bounds are then used to determine a success probability using the formula from
450 /// *Optimally Reliable & Cheap Payment Flows on the Lightning Network* by Rene Pickhardt
451 /// and Stefan Richter [[1]] (i.e. `(upper_bound - payment_amount) / (upper_bound - lower_bound)`).
452 ///
453 /// This probability is combined with the [`liquidity_penalty_multiplier_msat`] and
454 /// [`liquidity_penalty_amount_multiplier_msat`] parameters to calculate a concrete penalty in
455 /// milli-satoshis. The penalties, when added across all hops, have the property of being linear in
456 /// terms of the entire path's success probability. This allows the router to directly compare
457 /// penalties for different paths. See the documentation of those parameters for the exact formulas.
458 ///
459 /// The liquidity bounds are decayed by halving them every [`liquidity_offset_half_life`].
460 ///
461 /// Further, we track the history of our upper and lower liquidity bounds for each channel,
462 /// allowing us to assign a second penalty (using [`historical_liquidity_penalty_multiplier_msat`]
463 /// and [`historical_liquidity_penalty_amount_multiplier_msat`]) based on the same probability
464 /// formula, but using the history of a channel rather than our latest estimates for the liquidity
465 /// bounds.
466 ///
467 /// # Note
468 ///
469 /// Mixing the `no-std` feature between serialization and deserialization results in undefined
470 /// behavior.
471 ///
472 /// [1]: https://arxiv.org/abs/2107.05322
473 /// [`liquidity_penalty_multiplier_msat`]: ProbabilisticScoringFeeParameters::liquidity_penalty_multiplier_msat
474 /// [`liquidity_penalty_amount_multiplier_msat`]: ProbabilisticScoringFeeParameters::liquidity_penalty_amount_multiplier_msat
475 /// [`liquidity_offset_half_life`]: ProbabilisticScoringDecayParameters::liquidity_offset_half_life
476 /// [`historical_liquidity_penalty_multiplier_msat`]: ProbabilisticScoringFeeParameters::historical_liquidity_penalty_multiplier_msat
477 /// [`historical_liquidity_penalty_amount_multiplier_msat`]: ProbabilisticScoringFeeParameters::historical_liquidity_penalty_amount_multiplier_msat
478 pub struct ProbabilisticScorer<G: Deref<Target = NetworkGraph<L>>, L: Deref>
479 where L::Target: Logger {
480         decay_params: ProbabilisticScoringDecayParameters,
481         network_graph: G,
482         logger: L,
483         channel_liquidities: HashMap<u64, ChannelLiquidity>,
484 }
485
486 /// Parameters for configuring [`ProbabilisticScorer`].
487 ///
488 /// Used to configure base, liquidity, and amount penalties, the sum of which comprises the channel
489 /// penalty (i.e., the amount in msats willing to be paid to avoid routing through the channel).
490 ///
491 /// The penalty applied to any channel by the [`ProbabilisticScorer`] is the sum of each of the
492 /// parameters here.
493 #[derive(Clone)]
494 pub struct ProbabilisticScoringFeeParameters {
495         /// A fixed penalty in msats to apply to each channel.
496         ///
497         /// Default value: 500 msat
498         pub base_penalty_msat: u64,
499
500         /// A multiplier used with the total amount flowing over a channel to calculate a fixed penalty
501         /// applied to each channel, in excess of the [`base_penalty_msat`].
502         ///
503         /// The purpose of the amount penalty is to avoid having fees dominate the channel cost (i.e.,
504         /// fees plus penalty) for large payments. The penalty is computed as the product of this
505         /// multiplier and `2^30`ths of the total amount flowing over a channel (i.e. the payment
506         /// amount plus the amount of any other HTLCs flowing we sent over the same channel).
507         ///
508         /// ie `base_penalty_amount_multiplier_msat * amount_msat / 2^30`
509         ///
510         /// Default value: 8,192 msat
511         ///
512         /// [`base_penalty_msat`]: Self::base_penalty_msat
513         pub base_penalty_amount_multiplier_msat: u64,
514
515         /// A multiplier used in conjunction with the negative `log10` of the channel's success
516         /// probability for a payment, as determined by our latest estimates of the channel's
517         /// liquidity, to determine the liquidity penalty.
518         ///
519         /// The penalty is based in part on the knowledge learned from prior successful and unsuccessful
520         /// payments. This knowledge is decayed over time based on [`liquidity_offset_half_life`]. The
521         /// penalty is effectively limited to `2 * liquidity_penalty_multiplier_msat` (corresponding to
522         /// lower bounding the success probability to `0.01`) when the amount falls within the
523         /// uncertainty bounds of the channel liquidity balance. Amounts above the upper bound will
524         /// result in a `u64::max_value` penalty, however.
525         ///
526         /// `-log10(success_probability) * liquidity_penalty_multiplier_msat`
527         ///
528         /// Default value: 30,000 msat
529         ///
530         /// [`liquidity_offset_half_life`]: ProbabilisticScoringDecayParameters::liquidity_offset_half_life
531         pub liquidity_penalty_multiplier_msat: u64,
532
533         /// A multiplier used in conjunction with the total amount flowing over a channel and the
534         /// negative `log10` of the channel's success probability for the payment, as determined by our
535         /// latest estimates of the channel's liquidity, to determine the amount penalty.
536         ///
537         /// The purpose of the amount penalty is to avoid having fees dominate the channel cost (i.e.,
538         /// fees plus penalty) for large payments. The penalty is computed as the product of this
539         /// multiplier and `2^20`ths of the amount flowing over this channel, weighted by the negative
540         /// `log10` of the success probability.
541         ///
542         /// `-log10(success_probability) * liquidity_penalty_amount_multiplier_msat * amount_msat / 2^20`
543         ///
544         /// In practice, this means for 0.1 success probability (`-log10(0.1) == 1`) each `2^20`th of
545         /// the amount will result in a penalty of the multiplier. And, as the success probability
546         /// decreases, the negative `log10` weighting will increase dramatically. For higher success
547         /// probabilities, the multiplier will have a decreasing effect as the negative `log10` will
548         /// fall below `1`.
549         ///
550         /// Default value: 192 msat
551         pub liquidity_penalty_amount_multiplier_msat: u64,
552
553         /// A multiplier used in conjunction with the negative `log10` of the channel's success
554         /// probability for the payment, as determined based on the history of our estimates of the
555         /// channel's available liquidity, to determine a penalty.
556         ///
557         /// This penalty is similar to [`liquidity_penalty_multiplier_msat`], however, instead of using
558         /// only our latest estimate for the current liquidity available in the channel, it estimates
559         /// success probability based on the estimated liquidity available in the channel through
560         /// history. Specifically, every time we update our liquidity bounds on a given channel, we
561         /// track which of several buckets those bounds fall into, exponentially decaying the
562         /// probability of each bucket as new samples are added.
563         ///
564         /// Default value: 10,000 msat
565         ///
566         /// [`liquidity_penalty_multiplier_msat`]: Self::liquidity_penalty_multiplier_msat
567         pub historical_liquidity_penalty_multiplier_msat: u64,
568
569         /// A multiplier used in conjunction with the total amount flowing over a channel and the
570         /// negative `log10` of the channel's success probability for the payment, as determined based
571         /// on the history of our estimates of the channel's available liquidity, to determine a
572         /// penalty.
573         ///
574         /// The purpose of the amount penalty is to avoid having fees dominate the channel cost for
575         /// large payments. The penalty is computed as the product of this multiplier and `2^20`ths
576         /// of the amount flowing over this channel, weighted by the negative `log10` of the success
577         /// probability.
578         ///
579         /// This penalty is similar to [`liquidity_penalty_amount_multiplier_msat`], however, instead
580         /// of using only our latest estimate for the current liquidity available in the channel, it
581         /// estimates success probability based on the estimated liquidity available in the channel
582         /// through history. Specifically, every time we update our liquidity bounds on a given
583         /// channel, we track which of several buckets those bounds fall into, exponentially decaying
584         /// the probability of each bucket as new samples are added.
585         ///
586         /// Default value: 64 msat
587         ///
588         /// [`liquidity_penalty_amount_multiplier_msat`]: Self::liquidity_penalty_amount_multiplier_msat
589         pub historical_liquidity_penalty_amount_multiplier_msat: u64,
590
591         /// Manual penalties used for the given nodes. Allows to set a particular penalty for a given
592         /// node. Note that a manual penalty of `u64::max_value()` means the node would not ever be
593         /// considered during path finding.
594         ///
595         /// This is not exported to bindings users
596         pub manual_node_penalties: HashMap<NodeId, u64>,
597
598         /// This penalty is applied when `htlc_maximum_msat` is equal to or larger than half of the
599         /// channel's capacity, (ie. htlc_maximum_msat >= 0.5 * channel_capacity) which makes us
600         /// prefer nodes with a smaller `htlc_maximum_msat`. We treat such nodes preferentially
601         /// as this makes balance discovery attacks harder to execute, thereby creating an incentive
602         /// to restrict `htlc_maximum_msat` and improve privacy.
603         ///
604         /// Default value: 250 msat
605         pub anti_probing_penalty_msat: u64,
606
607         /// This penalty is applied when the total amount flowing over a channel exceeds our current
608         /// estimate of the channel's available liquidity. The total amount is the amount of the
609         /// current HTLC plus any HTLCs which we've sent over the same channel.
610         ///
611         /// Note that in this case all other penalties, including the
612         /// [`liquidity_penalty_multiplier_msat`] and [`liquidity_penalty_amount_multiplier_msat`]-based
613         /// penalties, as well as the [`base_penalty_msat`] and the [`anti_probing_penalty_msat`], if
614         /// applicable, are still included in the overall penalty.
615         ///
616         /// If you wish to avoid creating paths with such channels entirely, setting this to a value of
617         /// `u64::max_value()` will guarantee that.
618         ///
619         /// Default value: 1_0000_0000_000 msat (1 Bitcoin)
620         ///
621         /// [`liquidity_penalty_multiplier_msat`]: Self::liquidity_penalty_multiplier_msat
622         /// [`liquidity_penalty_amount_multiplier_msat`]: Self::liquidity_penalty_amount_multiplier_msat
623         /// [`base_penalty_msat`]: Self::base_penalty_msat
624         /// [`anti_probing_penalty_msat`]: Self::anti_probing_penalty_msat
625         pub considered_impossible_penalty_msat: u64,
626
627         /// In order to calculate most of the scores above, we must first convert a lower and upper
628         /// bound on the available liquidity in a channel into the probability that we think a payment
629         /// will succeed. That probability is derived from a Probability Density Function for where we
630         /// think the liquidity in a channel likely lies, given such bounds.
631         ///
632         /// If this flag is set, that PDF is simply a constant - we assume that the actual available
633         /// liquidity in a channel is just as likely to be at any point between our lower and upper
634         /// bounds.
635         ///
636         /// If this flag is *not* set, that PDF is `(x - 0.5*capacity) ^ 2`. That is, we use an
637         /// exponential curve which expects the liquidity of a channel to lie "at the edges". This
638         /// matches experimental results - most routing nodes do not aggressively rebalance their
639         /// channels and flows in the network are often unbalanced, leaving liquidity usually
640         /// unavailable.
641         ///
642         /// Thus, for the "best" routes, leave this flag `false`. However, the flag does imply a number
643         /// of floating-point multiplications in the hottest routing code, which may lead to routing
644         /// performance degradation on some machines.
645         ///
646         /// Default value: false
647         pub linear_success_probability: bool,
648 }
649
650 impl Default for ProbabilisticScoringFeeParameters {
651         fn default() -> Self {
652                 Self {
653                         base_penalty_msat: 500,
654                         base_penalty_amount_multiplier_msat: 8192,
655                         liquidity_penalty_multiplier_msat: 30_000,
656                         liquidity_penalty_amount_multiplier_msat: 192,
657                         manual_node_penalties: HashMap::new(),
658                         anti_probing_penalty_msat: 250,
659                         considered_impossible_penalty_msat: 1_0000_0000_000,
660                         historical_liquidity_penalty_multiplier_msat: 10_000,
661                         historical_liquidity_penalty_amount_multiplier_msat: 64,
662                         linear_success_probability: false,
663                 }
664         }
665 }
666
667 impl ProbabilisticScoringFeeParameters {
668         /// Marks the node with the given `node_id` as banned,
669         /// i.e it will be avoided during path finding.
670         pub fn add_banned(&mut self, node_id: &NodeId) {
671                 self.manual_node_penalties.insert(*node_id, u64::max_value());
672         }
673
674         /// Marks all nodes in the given list as banned, i.e.,
675         /// they will be avoided during path finding.
676         pub fn add_banned_from_list(&mut self, node_ids: Vec<NodeId>) {
677                 for id in node_ids {
678                         self.manual_node_penalties.insert(id, u64::max_value());
679                 }
680         }
681
682         /// Removes the node with the given `node_id` from the list of nodes to avoid.
683         pub fn remove_banned(&mut self, node_id: &NodeId) {
684                 self.manual_node_penalties.remove(node_id);
685         }
686
687         /// Sets a manual penalty for the given node.
688         pub fn set_manual_penalty(&mut self, node_id: &NodeId, penalty: u64) {
689                 self.manual_node_penalties.insert(*node_id, penalty);
690         }
691
692         /// Removes the node with the given `node_id` from the list of manual penalties.
693         pub fn remove_manual_penalty(&mut self, node_id: &NodeId) {
694                 self.manual_node_penalties.remove(node_id);
695         }
696
697         /// Clears the list of manual penalties that are applied during path finding.
698         pub fn clear_manual_penalties(&mut self) {
699                 self.manual_node_penalties = HashMap::new();
700         }
701 }
702
703 #[cfg(test)]
704 impl ProbabilisticScoringFeeParameters {
705         fn zero_penalty() -> Self {
706                 Self {
707                         base_penalty_msat: 0,
708                         base_penalty_amount_multiplier_msat: 0,
709                         liquidity_penalty_multiplier_msat: 0,
710                         liquidity_penalty_amount_multiplier_msat: 0,
711                         historical_liquidity_penalty_multiplier_msat: 0,
712                         historical_liquidity_penalty_amount_multiplier_msat: 0,
713                         manual_node_penalties: HashMap::new(),
714                         anti_probing_penalty_msat: 0,
715                         considered_impossible_penalty_msat: 0,
716                         linear_success_probability: true,
717                 }
718         }
719 }
720
721 /// Parameters for configuring [`ProbabilisticScorer`].
722 ///
723 /// Used to configure decay parameters that are static throughout the lifetime of the scorer.
724 /// these decay parameters affect the score of the channel penalty and are not changed on a
725 /// per-route penalty cost call.
726 #[derive(Copy, Clone)]
727 pub struct ProbabilisticScoringDecayParameters {
728         /// If we aren't learning any new datapoints for a channel, the historical liquidity bounds
729         /// tracking can simply live on with increasingly stale data. Instead, when a channel has not
730         /// seen a liquidity estimate update for this amount of time, the historical datapoints are
731         /// decayed by half.
732         /// For an example of historical_no_updates_half_life being used see [`historical_estimated_channel_liquidity_probabilities`]
733         ///
734         /// Note that after 16 or more half lives all historical data will be completely gone.
735         ///
736         /// Default value: 14 days
737         ///
738         /// [`historical_estimated_channel_liquidity_probabilities`]: ProbabilisticScorer::historical_estimated_channel_liquidity_probabilities
739         pub historical_no_updates_half_life: Duration,
740
741         /// Whenever this amount of time elapses since the last update to a channel's liquidity bounds,
742         /// the distance from the bounds to "zero" is cut in half. In other words, the lower-bound on
743         /// the available liquidity is halved and the upper-bound moves half-way to the channel's total
744         /// capacity.
745         ///
746         /// Because halving the liquidity bounds grows the uncertainty on the channel's liquidity,
747         /// the penalty for an amount within the new bounds may change. See the [`ProbabilisticScorer`]
748         /// struct documentation for more info on the way the liquidity bounds are used.
749         ///
750         /// For example, if the channel's capacity is 1 million sats, and the current upper and lower
751         /// liquidity bounds are 200,000 sats and 600,000 sats, after this amount of time the upper
752         /// and lower liquidity bounds will be decayed to 100,000 and 800,000 sats.
753         ///
754         /// Default value: 6 hours
755         ///
756         /// # Note
757         ///
758         /// When built with the `no-std` feature, time will never elapse. Therefore, the channel
759         /// liquidity knowledge will never decay except when the bounds cross.
760         pub liquidity_offset_half_life: Duration,
761 }
762
763 impl Default for ProbabilisticScoringDecayParameters {
764         fn default() -> Self {
765                 Self {
766                         liquidity_offset_half_life: Duration::from_secs(6 * 60 * 60),
767                         historical_no_updates_half_life: Duration::from_secs(60 * 60 * 24 * 14),
768                 }
769         }
770 }
771
772 #[cfg(test)]
773 impl ProbabilisticScoringDecayParameters {
774         fn zero_penalty() -> Self {
775                 Self {
776                         liquidity_offset_half_life: Duration::from_secs(6 * 60 * 60),
777                         historical_no_updates_half_life: Duration::from_secs(60 * 60 * 24 * 14),
778                 }
779         }
780 }
781
782 /// Accounting for channel liquidity balance uncertainty.
783 ///
784 /// Direction is defined in terms of [`NodeId`] partial ordering, where the source node is the
785 /// first node in the ordering of the channel's counterparties. Thus, swapping the two liquidity
786 /// offset fields gives the opposite direction.
787 struct ChannelLiquidity {
788         /// Lower channel liquidity bound in terms of an offset from zero.
789         min_liquidity_offset_msat: u64,
790
791         /// Upper channel liquidity bound in terms of an offset from the effective capacity.
792         max_liquidity_offset_msat: u64,
793
794         min_liquidity_offset_history: HistoricalBucketRangeTracker,
795         max_liquidity_offset_history: HistoricalBucketRangeTracker,
796
797         /// Time when the liquidity bounds were last modified as an offset since the unix epoch.
798         last_updated: Duration,
799
800         /// Time when the historical liquidity bounds were last modified as an offset against the unix
801         /// epoch.
802         offset_history_last_updated: Duration,
803 }
804
805 /// A snapshot of [`ChannelLiquidity`] in one direction assuming a certain channel capacity.
806 struct DirectedChannelLiquidity<L: Deref<Target = u64>, BRT: Deref<Target = HistoricalBucketRangeTracker>, T: Deref<Target = Duration>> {
807         min_liquidity_offset_msat: L,
808         max_liquidity_offset_msat: L,
809         liquidity_history: HistoricalMinMaxBuckets<BRT>,
810         capacity_msat: u64,
811         last_updated: T,
812         offset_history_last_updated: T,
813         decay_params: ProbabilisticScoringDecayParameters,
814 }
815
816 impl<G: Deref<Target = NetworkGraph<L>>, L: Deref> ProbabilisticScorer<G, L> where L::Target: Logger {
817         /// Creates a new scorer using the given scoring parameters for sending payments from a node
818         /// through a network graph.
819         pub fn new(decay_params: ProbabilisticScoringDecayParameters, network_graph: G, logger: L) -> Self {
820                 Self {
821                         decay_params,
822                         network_graph,
823                         logger,
824                         channel_liquidities: HashMap::new(),
825                 }
826         }
827
828         #[cfg(test)]
829         fn with_channel(mut self, short_channel_id: u64, liquidity: ChannelLiquidity) -> Self {
830                 assert!(self.channel_liquidities.insert(short_channel_id, liquidity).is_none());
831                 self
832         }
833
834         /// Dump the contents of this scorer into the configured logger.
835         ///
836         /// Note that this writes roughly one line per channel for which we have a liquidity estimate,
837         /// which may be a substantial amount of log output.
838         pub fn debug_log_liquidity_stats(&self) {
839                 let graph = self.network_graph.read_only();
840                 for (scid, liq) in self.channel_liquidities.iter() {
841                         if let Some(chan_debug) = graph.channels().get(scid) {
842                                 let log_direction = |source, target| {
843                                         if let Some((directed_info, _)) = chan_debug.as_directed_to(target) {
844                                                 let amt = directed_info.effective_capacity().as_msat();
845                                                 let dir_liq = liq.as_directed(source, target, amt, self.decay_params);
846
847                                                 let min_buckets = &dir_liq.liquidity_history.min_liquidity_offset_history.buckets;
848                                                 let max_buckets = &dir_liq.liquidity_history.max_liquidity_offset_history.buckets;
849
850                                                 log_debug!(self.logger, core::concat!(
851                                                         "Liquidity from {} to {} via {} is in the range ({}, {}).\n",
852                                                         "\tHistorical min liquidity bucket relative probabilities:\n",
853                                                         "\t\t{} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {}\n",
854                                                         "\tHistorical max liquidity bucket relative probabilities:\n",
855                                                         "\t\t{} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {} {}"),
856                                                         source, target, scid, dir_liq.min_liquidity_msat(), dir_liq.max_liquidity_msat(),
857                                                         min_buckets[ 0], min_buckets[ 1], min_buckets[ 2], min_buckets[ 3],
858                                                         min_buckets[ 4], min_buckets[ 5], min_buckets[ 6], min_buckets[ 7],
859                                                         min_buckets[ 8], min_buckets[ 9], min_buckets[10], min_buckets[11],
860                                                         min_buckets[12], min_buckets[13], min_buckets[14], min_buckets[15],
861                                                         min_buckets[16], min_buckets[17], min_buckets[18], min_buckets[19],
862                                                         min_buckets[20], min_buckets[21], min_buckets[22], min_buckets[23],
863                                                         min_buckets[24], min_buckets[25], min_buckets[26], min_buckets[27],
864                                                         min_buckets[28], min_buckets[29], min_buckets[30], min_buckets[31],
865                                                         // Note that the liquidity buckets are an offset from the edge, so we
866                                                         // inverse the max order to get the probabilities from zero.
867                                                         max_buckets[31], max_buckets[30], max_buckets[29], max_buckets[28],
868                                                         max_buckets[27], max_buckets[26], max_buckets[25], max_buckets[24],
869                                                         max_buckets[23], max_buckets[22], max_buckets[21], max_buckets[20],
870                                                         max_buckets[19], max_buckets[18], max_buckets[17], max_buckets[16],
871                                                         max_buckets[15], max_buckets[14], max_buckets[13], max_buckets[12],
872                                                         max_buckets[11], max_buckets[10], max_buckets[ 9], max_buckets[ 8],
873                                                         max_buckets[ 7], max_buckets[ 6], max_buckets[ 5], max_buckets[ 4],
874                                                         max_buckets[ 3], max_buckets[ 2], max_buckets[ 1], max_buckets[ 0]);
875                                         } else {
876                                                 log_debug!(self.logger, "No amount known for SCID {} from {:?} to {:?}", scid, source, target);
877                                         }
878                                 };
879
880                                 log_direction(&chan_debug.node_one, &chan_debug.node_two);
881                                 log_direction(&chan_debug.node_two, &chan_debug.node_one);
882                         } else {
883                                 log_debug!(self.logger, "No network graph entry for SCID {}", scid);
884                         }
885                 }
886         }
887
888         /// Query the estimated minimum and maximum liquidity available for sending a payment over the
889         /// channel with `scid` towards the given `target` node.
890         pub fn estimated_channel_liquidity_range(&self, scid: u64, target: &NodeId) -> Option<(u64, u64)> {
891                 let graph = self.network_graph.read_only();
892
893                 if let Some(chan) = graph.channels().get(&scid) {
894                         if let Some(liq) = self.channel_liquidities.get(&scid) {
895                                 if let Some((directed_info, source)) = chan.as_directed_to(target) {
896                                         let amt = directed_info.effective_capacity().as_msat();
897                                         let dir_liq = liq.as_directed(source, target, amt, self.decay_params);
898                                         return Some((dir_liq.min_liquidity_msat(), dir_liq.max_liquidity_msat()));
899                                 }
900                         }
901                 }
902                 None
903         }
904
905         /// Query the historical estimated minimum and maximum liquidity available for sending a
906         /// payment over the channel with `scid` towards the given `target` node.
907         ///
908         /// Returns two sets of 32 buckets. The first set describes the lower-bound liquidity history,
909         /// the second set describes the upper-bound liquidity history. Each bucket describes the
910         /// relative frequency at which we've seen a liquidity bound in the bucket's range relative to
911         /// the channel's total capacity, on an arbitrary scale. Because the values are slowly decayed,
912         /// more recent data points are weighted more heavily than older datapoints.
913         ///
914         /// Note that the range of each bucket varies by its location to provide more granular results
915         /// at the edges of a channel's capacity, where it is more likely to sit.
916         ///
917         /// When scoring, the estimated probability that an upper-/lower-bound lies in a given bucket
918         /// is calculated by dividing that bucket's value with the total value of all buckets.
919         ///
920         /// For example, using a lower bucket count for illustrative purposes, a value of
921         /// `[0, 0, 0, ..., 0, 32]` indicates that we believe the probability of a bound being very
922         /// close to the channel's capacity to be 100%, and have never (recently) seen it in any other
923         /// bucket. A value of `[31, 0, 0, ..., 0, 0, 32]` indicates we've seen the bound being both
924         /// in the top and bottom bucket, and roughly with similar (recent) frequency.
925         ///
926         /// Because the datapoints are decayed slowly over time, values will eventually return to
927         /// `Some(([0; 32], [0; 32]))` or `None` if no data remains for a channel.
928         ///
929         /// In order to fetch a single success probability from the buckets provided here, as used in
930         /// the scoring model, see [`Self::historical_estimated_payment_success_probability`].
931         pub fn historical_estimated_channel_liquidity_probabilities(&self, scid: u64, target: &NodeId)
932         -> Option<([u16; 32], [u16; 32])> {
933                 let graph = self.network_graph.read_only();
934
935                 if let Some(chan) = graph.channels().get(&scid) {
936                         if let Some(liq) = self.channel_liquidities.get(&scid) {
937                                 if let Some((directed_info, source)) = chan.as_directed_to(target) {
938                                         let amt = directed_info.effective_capacity().as_msat();
939                                         let dir_liq = liq.as_directed(source, target, amt, self.decay_params);
940
941                                         let min_buckets = dir_liq.liquidity_history.min_liquidity_offset_history.buckets;
942                                         let mut max_buckets = dir_liq.liquidity_history.max_liquidity_offset_history.buckets;
943
944                                         // Note that the liquidity buckets are an offset from the edge, so we inverse
945                                         // the max order to get the probabilities from zero.
946                                         max_buckets.reverse();
947                                         return Some((min_buckets, max_buckets));
948                                 }
949                         }
950                 }
951                 None
952         }
953
954         /// Query the probability of payment success sending the given `amount_msat` over the channel
955         /// with `scid` towards the given `target` node, based on the historical estimated liquidity
956         /// bounds.
957         ///
958         /// These are the same bounds as returned by
959         /// [`Self::historical_estimated_channel_liquidity_probabilities`] (but not those returned by
960         /// [`Self::estimated_channel_liquidity_range`]).
961         pub fn historical_estimated_payment_success_probability(
962                 &self, scid: u64, target: &NodeId, amount_msat: u64, params: &ProbabilisticScoringFeeParameters)
963         -> Option<f64> {
964                 let graph = self.network_graph.read_only();
965
966                 if let Some(chan) = graph.channels().get(&scid) {
967                         if let Some(liq) = self.channel_liquidities.get(&scid) {
968                                 if let Some((directed_info, source)) = chan.as_directed_to(target) {
969                                         let capacity_msat = directed_info.effective_capacity().as_msat();
970                                         let dir_liq = liq.as_directed(source, target, capacity_msat, self.decay_params);
971
972                                         return dir_liq.liquidity_history.calculate_success_probability_times_billion(
973                                                 &params, amount_msat, capacity_msat
974                                         ).map(|p| p as f64 / (1024 * 1024 * 1024) as f64);
975                                 }
976                         }
977                 }
978                 None
979         }
980 }
981
982 impl ChannelLiquidity {
983         fn new(last_updated: Duration) -> Self {
984                 Self {
985                         min_liquidity_offset_msat: 0,
986                         max_liquidity_offset_msat: 0,
987                         min_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
988                         max_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
989                         last_updated,
990                         offset_history_last_updated: last_updated,
991                 }
992         }
993
994         /// Returns a view of the channel liquidity directed from `source` to `target` assuming
995         /// `capacity_msat`.
996         fn as_directed(
997                 &self, source: &NodeId, target: &NodeId, capacity_msat: u64, decay_params: ProbabilisticScoringDecayParameters
998         ) -> DirectedChannelLiquidity<&u64, &HistoricalBucketRangeTracker, &Duration> {
999                 let (min_liquidity_offset_msat, max_liquidity_offset_msat, min_liquidity_offset_history, max_liquidity_offset_history) =
1000                         if source < target {
1001                                 (&self.min_liquidity_offset_msat, &self.max_liquidity_offset_msat,
1002                                         &self.min_liquidity_offset_history, &self.max_liquidity_offset_history)
1003                         } else {
1004                                 (&self.max_liquidity_offset_msat, &self.min_liquidity_offset_msat,
1005                                         &self.max_liquidity_offset_history, &self.min_liquidity_offset_history)
1006                         };
1007
1008                 DirectedChannelLiquidity {
1009                         min_liquidity_offset_msat,
1010                         max_liquidity_offset_msat,
1011                         liquidity_history: HistoricalMinMaxBuckets {
1012                                 min_liquidity_offset_history,
1013                                 max_liquidity_offset_history,
1014                         },
1015                         capacity_msat,
1016                         last_updated: &self.last_updated,
1017                         offset_history_last_updated: &self.offset_history_last_updated,
1018                         decay_params: decay_params,
1019                 }
1020         }
1021
1022         /// Returns a mutable view of the channel liquidity directed from `source` to `target` assuming
1023         /// `capacity_msat`.
1024         fn as_directed_mut(
1025                 &mut self, source: &NodeId, target: &NodeId, capacity_msat: u64, decay_params: ProbabilisticScoringDecayParameters
1026         ) -> DirectedChannelLiquidity<&mut u64, &mut HistoricalBucketRangeTracker, &mut Duration> {
1027                 let (min_liquidity_offset_msat, max_liquidity_offset_msat, min_liquidity_offset_history, max_liquidity_offset_history) =
1028                         if source < target {
1029                                 (&mut self.min_liquidity_offset_msat, &mut self.max_liquidity_offset_msat,
1030                                         &mut self.min_liquidity_offset_history, &mut self.max_liquidity_offset_history)
1031                         } else {
1032                                 (&mut self.max_liquidity_offset_msat, &mut self.min_liquidity_offset_msat,
1033                                         &mut self.max_liquidity_offset_history, &mut self.min_liquidity_offset_history)
1034                         };
1035
1036                 DirectedChannelLiquidity {
1037                         min_liquidity_offset_msat,
1038                         max_liquidity_offset_msat,
1039                         liquidity_history: HistoricalMinMaxBuckets {
1040                                 min_liquidity_offset_history,
1041                                 max_liquidity_offset_history,
1042                         },
1043                         capacity_msat,
1044                         last_updated: &mut self.last_updated,
1045                         offset_history_last_updated: &mut self.offset_history_last_updated,
1046                         decay_params: decay_params,
1047                 }
1048         }
1049
1050         fn decayed_offset(
1051                 &self, offset: u64, duration_since_epoch: Duration,
1052                 decay_params: ProbabilisticScoringDecayParameters,
1053         ) -> u64 {
1054                 let half_life = decay_params.liquidity_offset_half_life.as_secs_f64();
1055                 if half_life != 0.0 {
1056                         let elapsed_time = duration_since_epoch.saturating_sub(self.last_updated).as_secs_f64();
1057                         ((offset as f64) * powf64(0.5, elapsed_time / half_life)) as u64
1058                 } else {
1059                         0
1060                 }
1061         }
1062 }
1063
1064 /// Bounds `-log10` to avoid excessive liquidity penalties for payments with low success
1065 /// probabilities.
1066 const NEGATIVE_LOG10_UPPER_BOUND: u64 = 2;
1067
1068 /// The rough cutoff at which our precision falls off and we should stop bothering to try to log a
1069 /// ratio, as X in 1/X.
1070 const PRECISION_LOWER_BOUND_DENOMINATOR: u64 = approx::LOWER_BITS_BOUND;
1071
1072 /// The divisor used when computing the amount penalty.
1073 const AMOUNT_PENALTY_DIVISOR: u64 = 1 << 20;
1074 const BASE_AMOUNT_PENALTY_DIVISOR: u64 = 1 << 30;
1075
1076 /// Raises three `f64`s to the 3rd power, without `powi` because it requires `std` (dunno why).
1077 #[inline(always)]
1078 fn three_f64_pow_3(a: f64, b: f64, c: f64) -> (f64, f64, f64) {
1079         (a * a * a, b * b * b, c * c * c)
1080 }
1081
1082 /// Given liquidity bounds, calculates the success probability (in the form of a numerator and
1083 /// denominator) of an HTLC. This is a key assumption in our scoring models.
1084 ///
1085 /// Must not return a numerator or denominator greater than 2^31 for arguments less than 2^31.
1086 ///
1087 /// min_zero_implies_no_successes signals that a `min_liquidity_msat` of 0 means we've not
1088 /// (recently) seen an HTLC successfully complete over this channel.
1089 #[inline(always)]
1090 fn success_probability(
1091         amount_msat: u64, min_liquidity_msat: u64, max_liquidity_msat: u64, capacity_msat: u64,
1092         params: &ProbabilisticScoringFeeParameters, min_zero_implies_no_successes: bool,
1093 ) -> (u64, u64) {
1094         debug_assert!(min_liquidity_msat <= amount_msat);
1095         debug_assert!(amount_msat < max_liquidity_msat);
1096         debug_assert!(max_liquidity_msat <= capacity_msat);
1097
1098         let (numerator, mut denominator) =
1099                 if params.linear_success_probability {
1100                         (max_liquidity_msat - amount_msat,
1101                                 (max_liquidity_msat - min_liquidity_msat).saturating_add(1))
1102                 } else {
1103                         let capacity = capacity_msat as f64;
1104                         let min = (min_liquidity_msat as f64) / capacity;
1105                         let max = (max_liquidity_msat as f64) / capacity;
1106                         let amount = (amount_msat as f64) / capacity;
1107
1108                         // Assume the channel has a probability density function of (x - 0.5)^2 for values from
1109                         // 0 to 1 (where 1 is the channel's full capacity). The success probability given some
1110                         // liquidity bounds is thus the integral under the curve from the amount to maximum
1111                         // estimated liquidity, divided by the same integral from the minimum to the maximum
1112                         // estimated liquidity bounds.
1113                         //
1114                         // Because the integral from x to y is simply (y - 0.5)^3 - (x - 0.5)^3, we can
1115                         // calculate the cumulative density function between the min/max bounds trivially. Note
1116                         // that we don't bother to normalize the CDF to total to 1, as it will come out in the
1117                         // division of num / den.
1118                         let (max_pow, amt_pow, min_pow) = three_f64_pow_3(max - 0.5, amount - 0.5, min - 0.5);
1119                         let num = max_pow - amt_pow;
1120                         let den = max_pow - min_pow;
1121
1122                         // Because our numerator and denominator max out at 0.5^3 we need to multiply them by
1123                         // quite a large factor to get something useful (ideally in the 2^30 range).
1124                         const BILLIONISH: f64 = 1024.0 * 1024.0 * 1024.0;
1125                         let numerator = (num * BILLIONISH) as u64 + 1;
1126                         let denominator = (den * BILLIONISH) as u64 + 1;
1127                         debug_assert!(numerator <= 1 << 30, "Got large numerator ({}) from float {}.", numerator, num);
1128                         debug_assert!(denominator <= 1 << 30, "Got large denominator ({}) from float {}.", denominator, den);
1129                         (numerator, denominator)
1130                 };
1131
1132         if min_zero_implies_no_successes && min_liquidity_msat == 0 &&
1133                 denominator < u64::max_value() / 21
1134         {
1135                 // If we have no knowledge of the channel, scale probability down by ~75%
1136                 // Note that we prefer to increase the denominator rather than decrease the numerator as
1137                 // the denominator is more likely to be larger and thus provide greater precision. This is
1138                 // mostly an overoptimization but makes a large difference in tests.
1139                 denominator = denominator * 21 / 16
1140         }
1141
1142         (numerator, denominator)
1143 }
1144
1145 impl<L: Deref<Target = u64>, BRT: Deref<Target = HistoricalBucketRangeTracker>, T: Deref<Target = Duration>>
1146 DirectedChannelLiquidity< L, BRT, T> {
1147         /// Returns a liquidity penalty for routing the given HTLC `amount_msat` through the channel in
1148         /// this direction.
1149         fn penalty_msat(&self, amount_msat: u64, score_params: &ProbabilisticScoringFeeParameters) -> u64 {
1150                 let available_capacity = self.capacity_msat;
1151                 let max_liquidity_msat = self.max_liquidity_msat();
1152                 let min_liquidity_msat = core::cmp::min(self.min_liquidity_msat(), max_liquidity_msat);
1153
1154                 let mut res = if amount_msat <= min_liquidity_msat {
1155                         0
1156                 } else if amount_msat >= max_liquidity_msat {
1157                         // Equivalent to hitting the else clause below with the amount equal to the effective
1158                         // capacity and without any certainty on the liquidity upper bound, plus the
1159                         // impossibility penalty.
1160                         let negative_log10_times_2048 = NEGATIVE_LOG10_UPPER_BOUND * 2048;
1161                         Self::combined_penalty_msat(amount_msat, negative_log10_times_2048,
1162                                         score_params.liquidity_penalty_multiplier_msat,
1163                                         score_params.liquidity_penalty_amount_multiplier_msat)
1164                                 .saturating_add(score_params.considered_impossible_penalty_msat)
1165                 } else {
1166                         let (numerator, denominator) = success_probability(amount_msat,
1167                                 min_liquidity_msat, max_liquidity_msat, available_capacity, score_params, false);
1168                         if denominator - numerator < denominator / PRECISION_LOWER_BOUND_DENOMINATOR {
1169                                 // If the failure probability is < 1.5625% (as 1 - numerator/denominator < 1/64),
1170                                 // don't bother trying to use the log approximation as it gets too noisy to be
1171                                 // particularly helpful, instead just round down to 0.
1172                                 0
1173                         } else {
1174                                 let negative_log10_times_2048 =
1175                                         approx::negative_log10_times_2048(numerator, denominator);
1176                                 Self::combined_penalty_msat(amount_msat, negative_log10_times_2048,
1177                                         score_params.liquidity_penalty_multiplier_msat,
1178                                         score_params.liquidity_penalty_amount_multiplier_msat)
1179                         }
1180                 };
1181
1182                 if amount_msat >= available_capacity {
1183                         // We're trying to send more than the capacity, use a max penalty.
1184                         res = res.saturating_add(Self::combined_penalty_msat(amount_msat,
1185                                 NEGATIVE_LOG10_UPPER_BOUND * 2048,
1186                                 score_params.historical_liquidity_penalty_multiplier_msat,
1187                                 score_params.historical_liquidity_penalty_amount_multiplier_msat));
1188                         return res;
1189                 }
1190
1191                 if score_params.historical_liquidity_penalty_multiplier_msat != 0 ||
1192                    score_params.historical_liquidity_penalty_amount_multiplier_msat != 0 {
1193                         if let Some(cumulative_success_prob_times_billion) = self.liquidity_history
1194                                 .calculate_success_probability_times_billion(
1195                                         score_params, amount_msat, self.capacity_msat)
1196                         {
1197                                 let historical_negative_log10_times_2048 = approx::negative_log10_times_2048(cumulative_success_prob_times_billion + 1, 1024 * 1024 * 1024);
1198                                 res = res.saturating_add(Self::combined_penalty_msat(amount_msat,
1199                                         historical_negative_log10_times_2048, score_params.historical_liquidity_penalty_multiplier_msat,
1200                                         score_params.historical_liquidity_penalty_amount_multiplier_msat));
1201                         } else {
1202                                 // If we don't have any valid points (or, once decayed, we have less than a full
1203                                 // point), redo the non-historical calculation with no liquidity bounds tracked and
1204                                 // the historical penalty multipliers.
1205                                 let (numerator, denominator) = success_probability(amount_msat, 0,
1206                                         available_capacity, available_capacity, score_params, true);
1207                                 let negative_log10_times_2048 =
1208                                         approx::negative_log10_times_2048(numerator, denominator);
1209                                 res = res.saturating_add(Self::combined_penalty_msat(amount_msat, negative_log10_times_2048,
1210                                         score_params.historical_liquidity_penalty_multiplier_msat,
1211                                         score_params.historical_liquidity_penalty_amount_multiplier_msat));
1212                         }
1213                 }
1214
1215                 res
1216         }
1217
1218         /// Computes the liquidity penalty from the penalty multipliers.
1219         #[inline(always)]
1220         fn combined_penalty_msat(amount_msat: u64, mut negative_log10_times_2048: u64,
1221                 liquidity_penalty_multiplier_msat: u64, liquidity_penalty_amount_multiplier_msat: u64,
1222         ) -> u64 {
1223                 negative_log10_times_2048 =
1224                         negative_log10_times_2048.min(NEGATIVE_LOG10_UPPER_BOUND * 2048);
1225
1226                 // Upper bound the liquidity penalty to ensure some channel is selected.
1227                 let liquidity_penalty_msat = negative_log10_times_2048
1228                         .saturating_mul(liquidity_penalty_multiplier_msat) / 2048;
1229                 let amount_penalty_msat = negative_log10_times_2048
1230                         .saturating_mul(liquidity_penalty_amount_multiplier_msat)
1231                         .saturating_mul(amount_msat) / 2048 / AMOUNT_PENALTY_DIVISOR;
1232
1233                 liquidity_penalty_msat.saturating_add(amount_penalty_msat)
1234         }
1235
1236         /// Returns the lower bound of the channel liquidity balance in this direction.
1237         #[inline(always)]
1238         fn min_liquidity_msat(&self) -> u64 {
1239                 self.decayed_offset_msat(*self.min_liquidity_offset_msat)
1240         }
1241
1242         /// Returns the upper bound of the channel liquidity balance in this direction.
1243         #[inline(always)]
1244         fn max_liquidity_msat(&self) -> u64 {
1245                 self.capacity_msat
1246                         .saturating_sub(self.decayed_offset_msat(*self.max_liquidity_offset_msat))
1247         }
1248
1249         fn decayed_offset_msat(&self, offset_msat: u64) -> u64 {
1250                 offset_msat
1251         }
1252 }
1253
1254 impl<L: DerefMut<Target = u64>, BRT: DerefMut<Target = HistoricalBucketRangeTracker>, T: DerefMut<Target = Duration>>
1255 DirectedChannelLiquidity<L, BRT, T> {
1256         /// Adjusts the channel liquidity balance bounds when failing to route `amount_msat`.
1257         fn failed_at_channel<Log: Deref>(
1258                 &mut self, amount_msat: u64, duration_since_epoch: Duration, chan_descr: fmt::Arguments, logger: &Log
1259         ) where Log::Target: Logger {
1260                 let existing_max_msat = self.max_liquidity_msat();
1261                 if amount_msat < existing_max_msat {
1262                         log_debug!(logger, "Setting max liquidity of {} from {} to {}", chan_descr, existing_max_msat, amount_msat);
1263                         self.set_max_liquidity_msat(amount_msat, duration_since_epoch);
1264                 } else {
1265                         log_trace!(logger, "Max liquidity of {} is {} (already less than or equal to {})",
1266                                 chan_descr, existing_max_msat, amount_msat);
1267                 }
1268                 self.update_history_buckets(0, duration_since_epoch);
1269         }
1270
1271         /// Adjusts the channel liquidity balance bounds when failing to route `amount_msat` downstream.
1272         fn failed_downstream<Log: Deref>(
1273                 &mut self, amount_msat: u64, duration_since_epoch: Duration, chan_descr: fmt::Arguments, logger: &Log
1274         ) where Log::Target: Logger {
1275                 let existing_min_msat = self.min_liquidity_msat();
1276                 if amount_msat > existing_min_msat {
1277                         log_debug!(logger, "Setting min liquidity of {} from {} to {}", existing_min_msat, chan_descr, amount_msat);
1278                         self.set_min_liquidity_msat(amount_msat, duration_since_epoch);
1279                 } else {
1280                         log_trace!(logger, "Min liquidity of {} is {} (already greater than or equal to {})",
1281                                 chan_descr, existing_min_msat, amount_msat);
1282                 }
1283                 self.update_history_buckets(0, duration_since_epoch);
1284         }
1285
1286         /// Adjusts the channel liquidity balance bounds when successfully routing `amount_msat`.
1287         fn successful<Log: Deref>(&mut self,
1288                 amount_msat: u64, duration_since_epoch: Duration, chan_descr: fmt::Arguments, logger: &Log
1289         ) where Log::Target: Logger {
1290                 let max_liquidity_msat = self.max_liquidity_msat().checked_sub(amount_msat).unwrap_or(0);
1291                 log_debug!(logger, "Subtracting {} from max liquidity of {} (setting it to {})", amount_msat, chan_descr, max_liquidity_msat);
1292                 self.set_max_liquidity_msat(max_liquidity_msat, duration_since_epoch);
1293                 self.update_history_buckets(amount_msat, duration_since_epoch);
1294         }
1295
1296         /// Updates the history buckets for this channel. Because the history buckets track what we now
1297         /// know about the channel's state *prior to our payment* (i.e. what we assume is "steady
1298         /// state"), we allow the caller to set an offset applied to our liquidity bounds which
1299         /// represents the amount of the successful payment we just made.
1300         fn update_history_buckets(&mut self, bucket_offset_msat: u64, duration_since_epoch: Duration) {
1301                 let half_lives =
1302                         duration_since_epoch.checked_sub(*self.offset_history_last_updated)
1303                         .unwrap_or(Duration::ZERO).as_secs()
1304                         .checked_div(self.decay_params.historical_no_updates_half_life.as_secs())
1305                         .map(|v| v.try_into().unwrap_or(u32::max_value())).unwrap_or(u32::max_value());
1306                 self.liquidity_history.min_liquidity_offset_history.time_decay_data(half_lives);
1307                 self.liquidity_history.max_liquidity_offset_history.time_decay_data(half_lives);
1308
1309                 let min_liquidity_offset_msat = self.decayed_offset_msat(*self.min_liquidity_offset_msat);
1310                 self.liquidity_history.min_liquidity_offset_history.track_datapoint(
1311                         min_liquidity_offset_msat + bucket_offset_msat, self.capacity_msat
1312                 );
1313                 let max_liquidity_offset_msat = self.decayed_offset_msat(*self.max_liquidity_offset_msat);
1314                 self.liquidity_history.max_liquidity_offset_history.track_datapoint(
1315                         max_liquidity_offset_msat.saturating_sub(bucket_offset_msat), self.capacity_msat
1316                 );
1317                 *self.offset_history_last_updated = duration_since_epoch;
1318         }
1319
1320         /// Adjusts the lower bound of the channel liquidity balance in this direction.
1321         fn set_min_liquidity_msat(&mut self, amount_msat: u64, duration_since_epoch: Duration) {
1322                 *self.min_liquidity_offset_msat = amount_msat;
1323                 if amount_msat > self.max_liquidity_msat() {
1324                         *self.max_liquidity_offset_msat = 0;
1325                 }
1326                 *self.last_updated = duration_since_epoch;
1327         }
1328
1329         /// Adjusts the upper bound of the channel liquidity balance in this direction.
1330         fn set_max_liquidity_msat(&mut self, amount_msat: u64, duration_since_epoch: Duration) {
1331                 *self.max_liquidity_offset_msat = self.capacity_msat.checked_sub(amount_msat).unwrap_or(0);
1332                 if amount_msat < *self.min_liquidity_offset_msat {
1333                         *self.min_liquidity_offset_msat = 0;
1334                 }
1335                 *self.last_updated = duration_since_epoch;
1336         }
1337 }
1338
1339 impl<G: Deref<Target = NetworkGraph<L>>, L: Deref> ScoreLookUp for ProbabilisticScorer<G, L> where L::Target: Logger {
1340         type ScoreParams = ProbabilisticScoringFeeParameters;
1341         fn channel_penalty_msat(
1342                 &self, candidate: &CandidateRouteHop, usage: ChannelUsage, score_params: &ProbabilisticScoringFeeParameters
1343         ) -> u64 {
1344                 let (scid, target) = match candidate {
1345                         CandidateRouteHop::PublicHop { info, short_channel_id } => {
1346                                 (short_channel_id, info.target())
1347                         },
1348                         _ => return 0,
1349                 };
1350                 let source = candidate.source();
1351                 if let Some(penalty) = score_params.manual_node_penalties.get(&target) {
1352                         return *penalty;
1353                 }
1354
1355                 let base_penalty_msat = score_params.base_penalty_msat.saturating_add(
1356                         score_params.base_penalty_amount_multiplier_msat
1357                                 .saturating_mul(usage.amount_msat) / BASE_AMOUNT_PENALTY_DIVISOR);
1358
1359                 let mut anti_probing_penalty_msat = 0;
1360                 match usage.effective_capacity {
1361                         EffectiveCapacity::ExactLiquidity { liquidity_msat: amount_msat } |
1362                                 EffectiveCapacity::HintMaxHTLC { amount_msat } =>
1363                         {
1364                                 if usage.amount_msat > amount_msat {
1365                                         return u64::max_value();
1366                                 } else {
1367                                         return base_penalty_msat;
1368                                 }
1369                         },
1370                         EffectiveCapacity::Total { capacity_msat, htlc_maximum_msat } => {
1371                                 if htlc_maximum_msat >= capacity_msat/2 {
1372                                         anti_probing_penalty_msat = score_params.anti_probing_penalty_msat;
1373                                 }
1374                         },
1375                         _ => {},
1376                 }
1377
1378                 let amount_msat = usage.amount_msat.saturating_add(usage.inflight_htlc_msat);
1379                 let capacity_msat = usage.effective_capacity.as_msat();
1380                 self.channel_liquidities
1381                         .get(&scid)
1382                         .unwrap_or(&ChannelLiquidity::new(Duration::ZERO))
1383                         .as_directed(&source, &target, capacity_msat, self.decay_params)
1384                         .penalty_msat(amount_msat, score_params)
1385                         .saturating_add(anti_probing_penalty_msat)
1386                         .saturating_add(base_penalty_msat)
1387         }
1388 }
1389
1390 impl<G: Deref<Target = NetworkGraph<L>>, L: Deref> ScoreUpdate for ProbabilisticScorer<G, L> where L::Target: Logger {
1391         fn payment_path_failed(&mut self, path: &Path, short_channel_id: u64, duration_since_epoch: Duration) {
1392                 let amount_msat = path.final_value_msat();
1393                 log_trace!(self.logger, "Scoring path through to SCID {} as having failed at {} msat", short_channel_id, amount_msat);
1394                 let network_graph = self.network_graph.read_only();
1395                 for (hop_idx, hop) in path.hops.iter().enumerate() {
1396                         let target = NodeId::from_pubkey(&hop.pubkey);
1397                         let channel_directed_from_source = network_graph.channels()
1398                                 .get(&hop.short_channel_id)
1399                                 .and_then(|channel| channel.as_directed_to(&target));
1400
1401                         let at_failed_channel = hop.short_channel_id == short_channel_id;
1402                         if at_failed_channel && hop_idx == 0 {
1403                                 log_warn!(self.logger, "Payment failed at the first hop - we do not attempt to learn channel info in such cases as we can directly observe local state.\n\tBecause we know the local state, we should generally not see failures here - this may be an indication that your channel peer on channel {} is broken and you may wish to close the channel.", hop.short_channel_id);
1404                         }
1405
1406                         // Only score announced channels.
1407                         if let Some((channel, source)) = channel_directed_from_source {
1408                                 let capacity_msat = channel.effective_capacity().as_msat();
1409                                 if at_failed_channel {
1410                                         self.channel_liquidities
1411                                                 .entry(hop.short_channel_id)
1412                                                 .or_insert_with(|| ChannelLiquidity::new(duration_since_epoch))
1413                                                 .as_directed_mut(source, &target, capacity_msat, self.decay_params)
1414                                                 .failed_at_channel(amount_msat, duration_since_epoch,
1415                                                         format_args!("SCID {}, towards {:?}", hop.short_channel_id, target), &self.logger);
1416                                 } else {
1417                                         self.channel_liquidities
1418                                                 .entry(hop.short_channel_id)
1419                                                 .or_insert_with(|| ChannelLiquidity::new(duration_since_epoch))
1420                                                 .as_directed_mut(source, &target, capacity_msat, self.decay_params)
1421                                                 .failed_downstream(amount_msat, duration_since_epoch,
1422                                                         format_args!("SCID {}, towards {:?}", hop.short_channel_id, target), &self.logger);
1423                                 }
1424                         } else {
1425                                 log_debug!(self.logger, "Not able to penalize channel with SCID {} as we do not have graph info for it (likely a route-hint last-hop).",
1426                                         hop.short_channel_id);
1427                         }
1428                         if at_failed_channel { break; }
1429                 }
1430         }
1431
1432         fn payment_path_successful(&mut self, path: &Path, duration_since_epoch: Duration) {
1433                 let amount_msat = path.final_value_msat();
1434                 log_trace!(self.logger, "Scoring path through SCID {} as having succeeded at {} msat.",
1435                         path.hops.split_last().map(|(hop, _)| hop.short_channel_id).unwrap_or(0), amount_msat);
1436                 let network_graph = self.network_graph.read_only();
1437                 for hop in &path.hops {
1438                         let target = NodeId::from_pubkey(&hop.pubkey);
1439                         let channel_directed_from_source = network_graph.channels()
1440                                 .get(&hop.short_channel_id)
1441                                 .and_then(|channel| channel.as_directed_to(&target));
1442
1443                         // Only score announced channels.
1444                         if let Some((channel, source)) = channel_directed_from_source {
1445                                 let capacity_msat = channel.effective_capacity().as_msat();
1446                                 self.channel_liquidities
1447                                         .entry(hop.short_channel_id)
1448                                         .or_insert_with(|| ChannelLiquidity::new(duration_since_epoch))
1449                                         .as_directed_mut(source, &target, capacity_msat, self.decay_params)
1450                                         .successful(amount_msat, duration_since_epoch,
1451                                                 format_args!("SCID {}, towards {:?}", hop.short_channel_id, target), &self.logger);
1452                         } else {
1453                                 log_debug!(self.logger, "Not able to learn for channel with SCID {} as we do not have graph info for it (likely a route-hint last-hop).",
1454                                         hop.short_channel_id);
1455                         }
1456                 }
1457         }
1458
1459         fn probe_failed(&mut self, path: &Path, short_channel_id: u64, duration_since_epoch: Duration) {
1460                 self.payment_path_failed(path, short_channel_id, duration_since_epoch)
1461         }
1462
1463         fn probe_successful(&mut self, path: &Path, duration_since_epoch: Duration) {
1464                 self.payment_path_failed(path, u64::max_value(), duration_since_epoch)
1465         }
1466
1467         fn time_passed(&mut self, duration_since_epoch: Duration) {
1468                 let decay_params = self.decay_params;
1469                 self.channel_liquidities.retain(|_scid, liquidity| {
1470                         liquidity.min_liquidity_offset_msat =
1471                                 liquidity.decayed_offset(liquidity.min_liquidity_offset_msat, duration_since_epoch, decay_params);
1472                         liquidity.max_liquidity_offset_msat =
1473                                 liquidity.decayed_offset(liquidity.max_liquidity_offset_msat, duration_since_epoch, decay_params);
1474                         liquidity.last_updated = duration_since_epoch;
1475
1476                         let elapsed_time =
1477                                 duration_since_epoch.saturating_sub(liquidity.offset_history_last_updated);
1478                         if elapsed_time > decay_params.historical_no_updates_half_life {
1479                                 let half_life = decay_params.historical_no_updates_half_life.as_secs_f64();
1480                                 if half_life != 0.0 {
1481                                         let divisor = powf64(2048.0, elapsed_time.as_secs_f64() / half_life) as u64;
1482                                         for bucket in liquidity.min_liquidity_offset_history.buckets.iter_mut() {
1483                                                 *bucket = ((*bucket as u64) * 1024 / divisor) as u16;
1484                                         }
1485                                         for bucket in liquidity.max_liquidity_offset_history.buckets.iter_mut() {
1486                                                 *bucket = ((*bucket as u64) * 1024 / divisor) as u16;
1487                                         }
1488                                         liquidity.offset_history_last_updated = duration_since_epoch;
1489                                 }
1490                         }
1491                         liquidity.min_liquidity_offset_msat != 0 || liquidity.max_liquidity_offset_msat != 0 ||
1492                                 liquidity.min_liquidity_offset_history.buckets != [0; 32] ||
1493                                 liquidity.max_liquidity_offset_history.buckets != [0; 32]
1494                 });
1495         }
1496 }
1497
1498 #[cfg(c_bindings)]
1499 impl<G: Deref<Target = NetworkGraph<L>>, L: Deref> Score for ProbabilisticScorer<G, L>
1500 where L::Target: Logger {}
1501
1502 #[cfg(feature = "std")]
1503 #[inline]
1504 fn powf64(n: f64, exp: f64) -> f64 {
1505         n.powf(exp)
1506 }
1507 #[cfg(not(feature = "std"))]
1508 fn powf64(n: f64, exp: f64) -> f64 {
1509         libm::powf(n as f32, exp as f32) as f64
1510 }
1511
1512 mod approx {
1513         const BITS: u32 = 64;
1514         const HIGHEST_BIT: u32 = BITS - 1;
1515         const LOWER_BITS: u32 = 6;
1516         pub(super) const LOWER_BITS_BOUND: u64 = 1 << LOWER_BITS;
1517         const LOWER_BITMASK: u64 = (1 << LOWER_BITS) - 1;
1518
1519         /// Look-up table for `log10(x) * 2048` where row `i` is used for each `x` having `i` as the
1520         /// most significant bit. The next 4 bits of `x`, if applicable, are used for the second index.
1521         const LOG10_TIMES_2048: [[u16; (LOWER_BITS_BOUND) as usize]; BITS as usize] = [
1522                 [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1523                         0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1524                         0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1525                         0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
1526                 [617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617,
1527                         617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617, 617,
1528                         977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977,
1529                         977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977, 977],
1530                 [1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233, 1233,
1531                         1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431, 1431,
1532                         1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594, 1594,
1533                         1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731, 1731],
1534                 [1850, 1850, 1850, 1850, 1850, 1850, 1850, 1850, 1954, 1954, 1954, 1954, 1954, 1954, 1954, 1954,
1535                         2048, 2048, 2048, 2048, 2048, 2048, 2048, 2048, 2133, 2133, 2133, 2133, 2133, 2133, 2133, 2133,
1536                         2210, 2210, 2210, 2210, 2210, 2210, 2210, 2210, 2281, 2281, 2281, 2281, 2281, 2281, 2281, 2281,
1537                         2347, 2347, 2347, 2347, 2347, 2347, 2347, 2347, 2409, 2409, 2409, 2409, 2409, 2409, 2409, 2409],
1538                 [2466, 2466, 2466, 2466, 2520, 2520, 2520, 2520, 2571, 2571, 2571, 2571, 2619, 2619, 2619, 2619,
1539                         2665, 2665, 2665, 2665, 2708, 2708, 2708, 2708, 2749, 2749, 2749, 2749, 2789, 2789, 2789, 2789,
1540                         2827, 2827, 2827, 2827, 2863, 2863, 2863, 2863, 2898, 2898, 2898, 2898, 2931, 2931, 2931, 2931,
1541                         2964, 2964, 2964, 2964, 2995, 2995, 2995, 2995, 3025, 3025, 3025, 3025, 3054, 3054, 3054, 3054],
1542                 [3083, 3083, 3110, 3110, 3136, 3136, 3162, 3162, 3187, 3187, 3212, 3212, 3235, 3235, 3259, 3259,
1543                         3281, 3281, 3303, 3303, 3324, 3324, 3345, 3345, 3366, 3366, 3386, 3386, 3405, 3405, 3424, 3424,
1544                         3443, 3443, 3462, 3462, 3479, 3479, 3497, 3497, 3514, 3514, 3531, 3531, 3548, 3548, 3564, 3564,
1545                         3580, 3580, 3596, 3596, 3612, 3612, 3627, 3627, 3642, 3642, 3656, 3656, 3671, 3671, 3685, 3685],
1546                 [3699, 3713, 3726, 3740, 3753, 3766, 3779, 3791, 3804, 3816, 3828, 3840, 3852, 3864, 3875, 3886,
1547                         3898, 3909, 3919, 3930, 3941, 3951, 3962, 3972, 3982, 3992, 4002, 4012, 4022, 4031, 4041, 4050,
1548                         4060, 4069, 4078, 4087, 4096, 4105, 4114, 4122, 4131, 4139, 4148, 4156, 4164, 4173, 4181, 4189,
1549                         4197, 4205, 4213, 4220, 4228, 4236, 4243, 4251, 4258, 4266, 4273, 4280, 4287, 4294, 4302, 4309],
1550                 [4316, 4329, 4343, 4356, 4369, 4382, 4395, 4408, 4420, 4433, 4445, 4457, 4468, 4480, 4492, 4503,
1551                         4514, 4525, 4536, 4547, 4557, 4568, 4578, 4589, 4599, 4609, 4619, 4629, 4638, 4648, 4657, 4667,
1552                         4676, 4685, 4695, 4704, 4713, 4721, 4730, 4739, 4747, 4756, 4764, 4773, 4781, 4789, 4797, 4805,
1553                         4813, 4821, 4829, 4837, 4845, 4852, 4860, 4867, 4875, 4882, 4889, 4897, 4904, 4911, 4918, 4925],
1554                 [4932, 4946, 4959, 4973, 4986, 4999, 5012, 5024, 5037, 5049, 5061, 5073, 5085, 5097, 5108, 5119,
1555                         5131, 5142, 5153, 5163, 5174, 5184, 5195, 5205, 5215, 5225, 5235, 5245, 5255, 5264, 5274, 5283,
1556                         5293, 5302, 5311, 5320, 5329, 5338, 5347, 5355, 5364, 5372, 5381, 5389, 5397, 5406, 5414, 5422,
1557                         5430, 5438, 5446, 5453, 5461, 5469, 5476, 5484, 5491, 5499, 5506, 5513, 5520, 5527, 5535, 5542],
1558                 [5549, 5562, 5576, 5589, 5603, 5615, 5628, 5641, 5653, 5666, 5678, 5690, 5701, 5713, 5725, 5736,
1559                         5747, 5758, 5769, 5780, 5790, 5801, 5811, 5822, 5832, 5842, 5852, 5862, 5871, 5881, 5890, 5900,
1560                         5909, 5918, 5928, 5937, 5946, 5954, 5963, 5972, 5980, 5989, 5997, 6006, 6014, 6022, 6030, 6038,
1561                         6046, 6054, 6062, 6070, 6078, 6085, 6093, 6100, 6108, 6115, 6122, 6130, 6137, 6144, 6151, 6158],
1562                 [6165, 6179, 6192, 6206, 6219, 6232, 6245, 6257, 6270, 6282, 6294, 6306, 6318, 6330, 6341, 6352,
1563                         6364, 6375, 6386, 6396, 6407, 6417, 6428, 6438, 6448, 6458, 6468, 6478, 6488, 6497, 6507, 6516,
1564                         6526, 6535, 6544, 6553, 6562, 6571, 6580, 6588, 6597, 6605, 6614, 6622, 6630, 6639, 6647, 6655,
1565                         6663, 6671, 6679, 6686, 6694, 6702, 6709, 6717, 6724, 6732, 6739, 6746, 6753, 6761, 6768, 6775],
1566                 [6782, 6795, 6809, 6822, 6836, 6849, 6861, 6874, 6886, 6899, 6911, 6923, 6934, 6946, 6958, 6969,
1567                         6980, 6991, 7002, 7013, 7023, 7034, 7044, 7055, 7065, 7075, 7085, 7095, 7104, 7114, 7124, 7133,
1568                         7142, 7151, 7161, 7170, 7179, 7187, 7196, 7205, 7213, 7222, 7230, 7239, 7247, 7255, 7263, 7271,
1569                         7279, 7287, 7295, 7303, 7311, 7318, 7326, 7333, 7341, 7348, 7355, 7363, 7370, 7377, 7384, 7391],
1570                 [7398, 7412, 7425, 7439, 7452, 7465, 7478, 7490, 7503, 7515, 7527, 7539, 7551, 7563, 7574, 7585,
1571                         7597, 7608, 7619, 7629, 7640, 7651, 7661, 7671, 7681, 7691, 7701, 7711, 7721, 7731, 7740, 7749,
1572                         7759, 7768, 7777, 7786, 7795, 7804, 7813, 7821, 7830, 7838, 7847, 7855, 7864, 7872, 7880, 7888,
1573                         7896, 7904, 7912, 7919, 7927, 7935, 7942, 7950, 7957, 7965, 7972, 7979, 7986, 7994, 8001, 8008],
1574                 [8015, 8028, 8042, 8055, 8069, 8082, 8094, 8107, 8119, 8132, 8144, 8156, 8167, 8179, 8191, 8202,
1575                         8213, 8224, 8235, 8246, 8256, 8267, 8277, 8288, 8298, 8308, 8318, 8328, 8337, 8347, 8357, 8366,
1576                         8375, 8384, 8394, 8403, 8412, 8420, 8429, 8438, 8446, 8455, 8463, 8472, 8480, 8488, 8496, 8504,
1577                         8512, 8520, 8528, 8536, 8544, 8551, 8559, 8566, 8574, 8581, 8588, 8596, 8603, 8610, 8617, 8624],
1578                 [8631, 8645, 8659, 8672, 8685, 8698, 8711, 8723, 8736, 8748, 8760, 8772, 8784, 8796, 8807, 8818,
1579                         8830, 8841, 8852, 8862, 8873, 8884, 8894, 8904, 8914, 8924, 8934, 8944, 8954, 8964, 8973, 8982,
1580                         8992, 9001, 9010, 9019, 9028, 9037, 9046, 9054, 9063, 9071, 9080, 9088, 9097, 9105, 9113, 9121,
1581                         9129, 9137, 9145, 9152, 9160, 9168, 9175, 9183, 9190, 9198, 9205, 9212, 9219, 9227, 9234, 9241],
1582                 [9248, 9261, 9275, 9288, 9302, 9315, 9327, 9340, 9352, 9365, 9377, 9389, 9400, 9412, 9424, 9435,
1583                         9446, 9457, 9468, 9479, 9490, 9500, 9510, 9521, 9531, 9541, 9551, 9561, 9570, 9580, 9590, 9599,
1584                         9608, 9617, 9627, 9636, 9645, 9653, 9662, 9671, 9679, 9688, 9696, 9705, 9713, 9721, 9729, 9737,
1585                         9745, 9753, 9761, 9769, 9777, 9784, 9792, 9799, 9807, 9814, 9821, 9829, 9836, 9843, 9850, 9857],
1586                 [9864, 9878, 9892, 9905, 9918, 9931, 9944, 9956, 9969, 9981, 9993, 10005, 10017, 10029, 10040, 10051,
1587                         10063, 10074, 10085, 10095, 10106, 10117, 10127, 10137, 10147, 10157, 10167, 10177, 10187, 10197, 10206, 10215,
1588                         10225, 10234, 10243, 10252, 10261, 10270, 10279, 10287, 10296, 10304, 10313, 10321, 10330, 10338, 10346, 10354,
1589                         10362, 10370, 10378, 10385, 10393, 10401, 10408, 10416, 10423, 10431, 10438, 10445, 10452, 10460, 10467, 10474],
1590                 [10481, 10494, 10508, 10521, 10535, 10548, 10560, 10573, 10585, 10598, 10610, 10622, 10634, 10645, 10657, 10668,
1591                         10679, 10690, 10701, 10712, 10723, 10733, 10743, 10754, 10764, 10774, 10784, 10794, 10803, 10813, 10823, 10832,
1592                         10841, 10851, 10860, 10869, 10878, 10886, 10895, 10904, 10912, 10921, 10929, 10938, 10946, 10954, 10962, 10970,
1593                         10978, 10986, 10994, 11002, 11010, 11017, 11025, 11032, 11040, 11047, 11054, 11062, 11069, 11076, 11083, 11090],
1594                 [11097, 11111, 11125, 11138, 11151, 11164, 11177, 11189, 11202, 11214, 11226, 11238, 11250, 11262, 11273, 11284,
1595                         11296, 11307, 11318, 11328, 11339, 11350, 11360, 11370, 11380, 11390, 11400, 11410, 11420, 11430, 11439, 11448,
1596                         11458, 11467, 11476, 11485, 11494, 11503, 11512, 11520, 11529, 11538, 11546, 11554, 11563, 11571, 11579, 11587,
1597                         11595, 11603, 11611, 11618, 11626, 11634, 11641, 11649, 11656, 11664, 11671, 11678, 11685, 11693, 11700, 11707],
1598                 [11714, 11727, 11741, 11754, 11768, 11781, 11793, 11806, 11818, 11831, 11843, 11855, 11867, 11878, 11890, 11901,
1599                         11912, 11923, 11934, 11945, 11956, 11966, 11976, 11987, 11997, 12007, 12017, 12027, 12036, 12046, 12056, 12065,
1600                         12074, 12084, 12093, 12102, 12111, 12119, 12128, 12137, 12146, 12154, 12162, 12171, 12179, 12187, 12195, 12203,
1601                         12211, 12219, 12227, 12235, 12243, 12250, 12258, 12265, 12273, 12280, 12287, 12295, 12302, 12309, 12316, 12323],
1602                 [12330, 12344, 12358, 12371, 12384, 12397, 12410, 12423, 12435, 12447, 12459, 12471, 12483, 12495, 12506, 12517,
1603                         12529, 12540, 12551, 12561, 12572, 12583, 12593, 12603, 12613, 12623, 12633, 12643, 12653, 12663, 12672, 12682,
1604                         12691, 12700, 12709, 12718, 12727, 12736, 12745, 12753, 12762, 12771, 12779, 12787, 12796, 12804, 12812, 12820,
1605                         12828, 12836, 12844, 12851, 12859, 12867, 12874, 12882, 12889, 12897, 12904, 12911, 12918, 12926, 12933, 12940],
1606                 [12947, 12960, 12974, 12987, 13001, 13014, 13026, 13039, 13051, 13064, 13076, 13088, 13100, 13111, 13123, 13134,
1607                         13145, 13156, 13167, 13178, 13189, 13199, 13209, 13220, 13230, 13240, 13250, 13260, 13269, 13279, 13289, 13298,
1608                         13307, 13317, 13326, 13335, 13344, 13352, 13361, 13370, 13379, 13387, 13395, 13404, 13412, 13420, 13428, 13436,
1609                         13444, 13452, 13460, 13468, 13476, 13483, 13491, 13498, 13506, 13513, 13521, 13528, 13535, 13542, 13549, 13556],
1610                 [13563, 13577, 13591, 13604, 13617, 13630, 13643, 13656, 13668, 13680, 13692, 13704, 13716, 13728, 13739, 13750,
1611                         13762, 13773, 13784, 13794, 13805, 13816, 13826, 13836, 13846, 13857, 13866, 13876, 13886, 13896, 13905, 13915,
1612                         13924, 13933, 13942, 13951, 13960, 13969, 13978, 13986, 13995, 14004, 14012, 14020, 14029, 14037, 14045, 14053,
1613                         14061, 14069, 14077, 14084, 14092, 14100, 14107, 14115, 14122, 14130, 14137, 14144, 14151, 14159, 14166, 14173],
1614                 [14180, 14194, 14207, 14220, 14234, 14247, 14259, 14272, 14284, 14297, 14309, 14321, 14333, 14344, 14356, 14367,
1615                         14378, 14389, 14400, 14411, 14422, 14432, 14443, 14453, 14463, 14473, 14483, 14493, 14502, 14512, 14522, 14531,
1616                         14540, 14550, 14559, 14568, 14577, 14586, 14594, 14603, 14612, 14620, 14628, 14637, 14645, 14653, 14661, 14669,
1617                         14677, 14685, 14693, 14701, 14709, 14716, 14724, 14731, 14739, 14746, 14754, 14761, 14768, 14775, 14782, 14789],
1618                 [14796, 14810, 14824, 14837, 14850, 14863, 14876, 14889, 14901, 14913, 14925, 14937, 14949, 14961, 14972, 14984,
1619                         14995, 15006, 15017, 15027, 15038, 15049, 15059, 15069, 15079, 15090, 15099, 15109, 15119, 15129, 15138, 15148,
1620                         15157, 15166, 15175, 15184, 15193, 15202, 15211, 15219, 15228, 15237, 15245, 15253, 15262, 15270, 15278, 15286,
1621                         15294, 15302, 15310, 15317, 15325, 15333, 15340, 15348, 15355, 15363, 15370, 15377, 15384, 15392, 15399, 15406],
1622                 [15413, 15427, 15440, 15453, 15467, 15480, 15492, 15505, 15517, 15530, 15542, 15554, 15566, 15577, 15589, 15600,
1623                         15611, 15622, 15633, 15644, 15655, 15665, 15676, 15686, 15696, 15706, 15716, 15726, 15736, 15745, 15755, 15764,
1624                         15773, 15783, 15792, 15801, 15810, 15819, 15827, 15836, 15845, 15853, 15862, 15870, 15878, 15886, 15894, 15903,
1625                         15910, 15918, 15926, 15934, 15942, 15949, 15957, 15964, 15972, 15979, 15987, 15994, 16001, 16008, 16015, 16022],
1626                 [16029, 16043, 16057, 16070, 16083, 16096, 16109, 16122, 16134, 16146, 16158, 16170, 16182, 16194, 16205, 16217,
1627                         16228, 16239, 16250, 16260, 16271, 16282, 16292, 16302, 16312, 16323, 16332, 16342, 16352, 16362, 16371, 16381,
1628                         16390, 16399, 16408, 16417, 16426, 16435, 16444, 16452, 16461, 16470, 16478, 16486, 16495, 16503, 16511, 16519,
1629                         16527, 16535, 16543, 16550, 16558, 16566, 16573, 16581, 16588, 16596, 16603, 16610, 16618, 16625, 16632, 16639],
1630                 [16646, 16660, 16673, 16686, 16700, 16713, 16725, 16738, 16751, 16763, 16775, 16787, 16799, 16810, 16822, 16833,
1631                         16844, 16855, 16866, 16877, 16888, 16898, 16909, 16919, 16929, 16939, 16949, 16959, 16969, 16978, 16988, 16997,
1632                         17006, 17016, 17025, 17034, 17043, 17052, 17060, 17069, 17078, 17086, 17095, 17103, 17111, 17119, 17127, 17136,
1633                         17143, 17151, 17159, 17167, 17175, 17182, 17190, 17197, 17205, 17212, 17220, 17227, 17234, 17241, 17248, 17255],
1634                 [17262, 17276, 17290, 17303, 17316, 17329, 17342, 17355, 17367, 17379, 17391, 17403, 17415, 17427, 17438, 17450,
1635                         17461, 17472, 17483, 17493, 17504, 17515, 17525, 17535, 17546, 17556, 17565, 17575, 17585, 17595, 17604, 17614,
1636                         17623, 17632, 17641, 17650, 17659, 17668, 17677, 17685, 17694, 17703, 17711, 17719, 17728, 17736, 17744, 17752,
1637                         17760, 17768, 17776, 17784, 17791, 17799, 17806, 17814, 17821, 17829, 17836, 17843, 17851, 17858, 17865, 17872],
1638                 [17879, 17893, 17906, 17920, 17933, 17946, 17958, 17971, 17984, 17996, 18008, 18020, 18032, 18043, 18055, 18066,
1639                         18077, 18088, 18099, 18110, 18121, 18131, 18142, 18152, 18162, 18172, 18182, 18192, 18202, 18211, 18221, 18230,
1640                         18239, 18249, 18258, 18267, 18276, 18285, 18293, 18302, 18311, 18319, 18328, 18336, 18344, 18352, 18360, 18369,
1641                         18377, 18384, 18392, 18400, 18408, 18415, 18423, 18430, 18438, 18445, 18453, 18460, 18467, 18474, 18481, 18488],
1642                 [18495, 18509, 18523, 18536, 18549, 18562, 18575, 18588, 18600, 18612, 18624, 18636, 18648, 18660, 18671, 18683,
1643                         18694, 18705, 18716, 18726, 18737, 18748, 18758, 18768, 18779, 18789, 18799, 18808, 18818, 18828, 18837, 18847,
1644                         18856, 18865, 18874, 18883, 18892, 18901, 18910, 18919, 18927, 18936, 18944, 18952, 18961, 18969, 18977, 18985,
1645                         18993, 19001, 19009, 19017, 19024, 19032, 19039, 19047, 19054, 19062, 19069, 19076, 19084, 19091, 19098, 19105],
1646                 [19112, 19126, 19139, 19153, 19166, 19179, 19191, 19204, 19217, 19229, 19241, 19253, 19265, 19276, 19288, 19299,
1647                         19310, 19321, 19332, 19343, 19354, 19364, 19375, 19385, 19395, 19405, 19415, 19425, 19435, 19444, 19454, 19463,
1648                         19472, 19482, 19491, 19500, 19509, 19518, 19526, 19535, 19544, 19552, 19561, 19569, 19577, 19585, 19594, 19602,
1649                         19610, 19617, 19625, 19633, 19641, 19648, 19656, 19663, 19671, 19678, 19686, 19693, 19700, 19707, 19714, 19721],
1650                 [19728, 19742, 19756, 19769, 19782, 19795, 19808, 19821, 19833, 19845, 19857, 19869, 19881, 19893, 19904, 19916,
1651                         19927, 19938, 19949, 19960, 19970, 19981, 19991, 20001, 20012, 20022, 20032, 20041, 20051, 20061, 20070, 20080,
1652                         20089, 20098, 20107, 20116, 20125, 20134, 20143, 20152, 20160, 20169, 20177, 20185, 20194, 20202, 20210, 20218,
1653                         20226, 20234, 20242, 20250, 20257, 20265, 20272, 20280, 20287, 20295, 20302, 20309, 20317, 20324, 20331, 20338],
1654                 [20345, 20359, 20372, 20386, 20399, 20412, 20425, 20437, 20450, 20462, 20474, 20486, 20498, 20509, 20521, 20532,
1655                         20543, 20554, 20565, 20576, 20587, 20597, 20608, 20618, 20628, 20638, 20648, 20658, 20668, 20677, 20687, 20696,
1656                         20705, 20715, 20724, 20733, 20742, 20751, 20759, 20768, 20777, 20785, 20794, 20802, 20810, 20818, 20827, 20835,
1657                         20843, 20850, 20858, 20866, 20874, 20881, 20889, 20896, 20904, 20911, 20919, 20926, 20933, 20940, 20947, 20954],
1658                 [20961, 20975, 20989, 21002, 21015, 21028, 21041, 21054, 21066, 21078, 21090, 21102, 21114, 21126, 21137, 21149,
1659                         21160, 21171, 21182, 21193, 21203, 21214, 21224, 21234, 21245, 21255, 21265, 21274, 21284, 21294, 21303, 21313,
1660                         21322, 21331, 21340, 21349, 21358, 21367, 21376, 21385, 21393, 21402, 21410, 21418, 21427, 21435, 21443, 21451,
1661                         21459, 21467, 21475, 21483, 21490, 21498, 21505, 21513, 21520, 21528, 21535, 21542, 21550, 21557, 21564, 21571],
1662                 [21578, 21592, 21605, 21619, 21632, 21645, 21658, 21670, 21683, 21695, 21707, 21719, 21731, 21742, 21754, 21765,
1663                         21776, 21787, 21798, 21809, 21820, 21830, 21841, 21851, 21861, 21871, 21881, 21891, 21901, 21910, 21920, 21929,
1664                         21938, 21948, 21957, 21966, 21975, 21984, 21992, 22001, 22010, 22018, 22027, 22035, 22043, 22051, 22060, 22068,
1665                         22076, 22083, 22091, 22099, 22107, 22114, 22122, 22129, 22137, 22144, 22152, 22159, 22166, 22173, 22180, 22187],
1666                 [22194, 22208, 22222, 22235, 22248, 22261, 22274, 22287, 22299, 22311, 22323, 22335, 22347, 22359, 22370, 22382,
1667                         22393, 22404, 22415, 22426, 22436, 22447, 22457, 22467, 22478, 22488, 22498, 22507, 22517, 22527, 22536, 22546,
1668                         22555, 22564, 22573, 22582, 22591, 22600, 22609, 22618, 22626, 22635, 22643, 22651, 22660, 22668, 22676, 22684,
1669                         22692, 22700, 22708, 22716, 22723, 22731, 22738, 22746, 22753, 22761, 22768, 22775, 22783, 22790, 22797, 22804],
1670                 [22811, 22825, 22838, 22852, 22865, 22878, 22891, 22903, 22916, 22928, 22940, 22952, 22964, 22975, 22987, 22998,
1671                         23009, 23020, 23031, 23042, 23053, 23063, 23074, 23084, 23094, 23104, 23114, 23124, 23134, 23143, 23153, 23162,
1672                         23171, 23181, 23190, 23199, 23208, 23217, 23225, 23234, 23243, 23251, 23260, 23268, 23276, 23284, 23293, 23301,
1673                         23309, 23316, 23324, 23332, 23340, 23347, 23355, 23363, 23370, 23377, 23385, 23392, 23399, 23406, 23413, 23420],
1674                 [23427, 23441, 23455, 23468, 23481, 23494, 23507, 23520, 23532, 23544, 23556, 23568, 23580, 23592, 23603, 23615,
1675                         23626, 23637, 23648, 23659, 23669, 23680, 23690, 23700, 23711, 23721, 23731, 23740, 23750, 23760, 23769, 23779,
1676                         23788, 23797, 23806, 23815, 23824, 23833, 23842, 23851, 23859, 23868, 23876, 23884, 23893, 23901, 23909, 23917,
1677                         23925, 23933, 23941, 23949, 23956, 23964, 23972, 23979, 23986, 23994, 24001, 24008, 24016, 24023, 24030, 24037],
1678                 [24044, 24058, 24071, 24085, 24098, 24111, 24124, 24136, 24149, 24161, 24173, 24185, 24197, 24208, 24220, 24231,
1679                         24242, 24253, 24264, 24275, 24286, 24296, 24307, 24317, 24327, 24337, 24347, 24357, 24367, 24376, 24386, 24395,
1680                         24405, 24414, 24423, 24432, 24441, 24450, 24458, 24467, 24476, 24484, 24493, 24501, 24509, 24517, 24526, 24534,
1681                         24542, 24550, 24557, 24565, 24573, 24580, 24588, 24596, 24603, 24610, 24618, 24625, 24632, 24639, 24646, 24653],
1682                 [24660, 24674, 24688, 24701, 24714, 24727, 24740, 24753, 24765, 24777, 24790, 24801, 24813, 24825, 24836, 24848,
1683                         24859, 24870, 24881, 24892, 24902, 24913, 24923, 24933, 24944, 24954, 24964, 24973, 24983, 24993, 25002, 25012,
1684                         25021, 25030, 25039, 25048, 25057, 25066, 25075, 25084, 25092, 25101, 25109, 25117, 25126, 25134, 25142, 25150,
1685                         25158, 25166, 25174, 25182, 25189, 25197, 25205, 25212, 25219, 25227, 25234, 25241, 25249, 25256, 25263, 25270],
1686                 [25277, 25291, 25304, 25318, 25331, 25344, 25357, 25369, 25382, 25394, 25406, 25418, 25430, 25441, 25453, 25464,
1687                         25475, 25486, 25497, 25508, 25519, 25529, 25540, 25550, 25560, 25570, 25580, 25590, 25600, 25609, 25619, 25628,
1688                         25638, 25647, 25656, 25665, 25674, 25683, 25691, 25700, 25709, 25717, 25726, 25734, 25742, 25750, 25759, 25767,
1689                         25775, 25783, 25790, 25798, 25806, 25813, 25821, 25829, 25836, 25843, 25851, 25858, 25865, 25872, 25879, 25886],
1690                 [25893, 25907, 25921, 25934, 25947, 25960, 25973, 25986, 25998, 26010, 26023, 26034, 26046, 26058, 26069, 26081,
1691                         26092, 26103, 26114, 26125, 26135, 26146, 26156, 26166, 26177, 26187, 26197, 26206, 26216, 26226, 26235, 26245,
1692                         26254, 26263, 26272, 26281, 26290, 26299, 26308, 26317, 26325, 26334, 26342, 26351, 26359, 26367, 26375, 26383,
1693                         26391, 26399, 26407, 26415, 26422, 26430, 26438, 26445, 26453, 26460, 26467, 26474, 26482, 26489, 26496, 26503],
1694                 [26510, 26524, 26537, 26551, 26564, 26577, 26590, 26602, 26615, 26627, 26639, 26651, 26663, 26674, 26686, 26697,
1695                         26708, 26719, 26730, 26741, 26752, 26762, 26773, 26783, 26793, 26803, 26813, 26823, 26833, 26842, 26852, 26861,
1696                         26871, 26880, 26889, 26898, 26907, 26916, 26924, 26933, 26942, 26950, 26959, 26967, 26975, 26983, 26992, 27000,
1697                         27008, 27016, 27023, 27031, 27039, 27046, 27054, 27062, 27069, 27076, 27084, 27091, 27098, 27105, 27112, 27119],
1698                 [27126, 27140, 27154, 27167, 27180, 27193, 27206, 27219, 27231, 27243, 27256, 27267, 27279, 27291, 27302, 27314,
1699                         27325, 27336, 27347, 27358, 27368, 27379, 27389, 27399, 27410, 27420, 27430, 27439, 27449, 27459, 27468, 27478,
1700                         27487, 27496, 27505, 27514, 27523, 27532, 27541, 27550, 27558, 27567, 27575, 27584, 27592, 27600, 27608, 27616,
1701                         27624, 27632, 27640, 27648, 27655, 27663, 27671, 27678, 27686, 27693, 27700, 27707, 27715, 27722, 27729, 27736],
1702                 [27743, 27757, 27770, 27784, 27797, 27810, 27823, 27835, 27848, 27860, 27872, 27884, 27896, 27907, 27919, 27930,
1703                         27941, 27952, 27963, 27974, 27985, 27995, 28006, 28016, 28026, 28036, 28046, 28056, 28066, 28075, 28085, 28094,
1704                         28104, 28113, 28122, 28131, 28140, 28149, 28157, 28166, 28175, 28183, 28192, 28200, 28208, 28217, 28225, 28233,
1705                         28241, 28249, 28256, 28264, 28272, 28280, 28287, 28295, 28302, 28309, 28317, 28324, 28331, 28338, 28345, 28352],
1706                 [28359, 28373, 28387, 28400, 28413, 28426, 28439, 28452, 28464, 28476, 28489, 28501, 28512, 28524, 28535, 28547,
1707                         28558, 28569, 28580, 28591, 28601, 28612, 28622, 28633, 28643, 28653, 28663, 28672, 28682, 28692, 28701, 28711,
1708                         28720, 28729, 28738, 28747, 28756, 28765, 28774, 28783, 28791, 28800, 28808, 28817, 28825, 28833, 28841, 28849,
1709                         28857, 28865, 28873, 28881, 28888, 28896, 28904, 28911, 28919, 28926, 28933, 28941, 28948, 28955, 28962, 28969],
1710                 [28976, 28990, 29003, 29017, 29030, 29043, 29056, 29068, 29081, 29093, 29105, 29117, 29129, 29140, 29152, 29163,
1711                         29174, 29185, 29196, 29207, 29218, 29228, 29239, 29249, 29259, 29269, 29279, 29289, 29299, 29308, 29318, 29327,
1712                         29337, 29346, 29355, 29364, 29373, 29382, 29390, 29399, 29408, 29416, 29425, 29433, 29441, 29450, 29458, 29466,
1713                         29474, 29482, 29489, 29497, 29505, 29513, 29520, 29528, 29535, 29542, 29550, 29557, 29564, 29571, 29578, 29585],
1714                 [29592, 29606, 29620, 29633, 29646, 29659, 29672, 29685, 29697, 29709, 29722, 29734, 29745, 29757, 29768, 29780,
1715                         29791, 29802, 29813, 29824, 29834, 29845, 29855, 29866, 29876, 29886, 29896, 29906, 29915, 29925, 29934, 29944,
1716                         29953, 29962, 29971, 29980, 29989, 29998, 30007, 30016, 30024, 30033, 30041, 30050, 30058, 30066, 30074, 30082,
1717                         30090, 30098, 30106, 30114, 30121, 30129, 30137, 30144, 30152, 30159, 30166, 30174, 30181, 30188, 30195, 30202],
1718                 [30209, 30223, 30236, 30250, 30263, 30276, 30289, 30301, 30314, 30326, 30338, 30350, 30362, 30373, 30385, 30396,
1719                         30407, 30418, 30429, 30440, 30451, 30461, 30472, 30482, 30492, 30502, 30512, 30522, 30532, 30541, 30551, 30560,
1720                         30570, 30579, 30588, 30597, 30606, 30615, 30624, 30632, 30641, 30649, 30658, 30666, 30674, 30683, 30691, 30699,
1721                         30707, 30715, 30722, 30730, 30738, 30746, 30753, 30761, 30768, 30775, 30783, 30790, 30797, 30804, 30811, 30818],
1722                 [30825, 30839, 30853, 30866, 30879, 30892, 30905, 30918, 30930, 30943, 30955, 30967, 30978, 30990, 31001, 31013,
1723                         31024, 31035, 31046, 31057, 31067, 31078, 31088, 31099, 31109, 31119, 31129, 31139, 31148, 31158, 31167, 31177,
1724                         31186, 31195, 31204, 31213, 31222, 31231, 31240, 31249, 31257, 31266, 31274, 31283, 31291, 31299, 31307, 31315,
1725                         31323, 31331, 31339, 31347, 31354, 31362, 31370, 31377, 31385, 31392, 31399, 31407, 31414, 31421, 31428, 31435],
1726                 [31442, 31456, 31469, 31483, 31496, 31509, 31522, 31534, 31547, 31559, 31571, 31583, 31595, 31606, 31618, 31629,
1727                         31640, 31652, 31662, 31673, 31684, 31694, 31705, 31715, 31725, 31735, 31745, 31755, 31765, 31774, 31784, 31793,
1728                         31803, 31812, 31821, 31830, 31839, 31848, 31857, 31865, 31874, 31882, 31891, 31899, 31907, 31916, 31924, 31932,
1729                         31940, 31948, 31955, 31963, 31971, 31979, 31986, 31994, 32001, 32008, 32016, 32023, 32030, 32037, 32044, 32052],
1730                 [32058, 32072, 32086, 32099, 32112, 32125, 32138, 32151, 32163, 32176, 32188, 32200, 32211, 32223, 32234, 32246,
1731                         32257, 32268, 32279, 32290, 32300, 32311, 32321, 32332, 32342, 32352, 32362, 32372, 32381, 32391, 32400, 32410,
1732                         32419, 32428, 32437, 32446, 32455, 32464, 32473, 32482, 32490, 32499, 32507, 32516, 32524, 32532, 32540, 32548,
1733                         32556, 32564, 32572, 32580, 32587, 32595, 32603, 32610, 32618, 32625, 32632, 32640, 32647, 32654, 32661, 32668],
1734                 [32675, 32689, 32702, 32716, 32729, 32742, 32755, 32767, 32780, 32792, 32804, 32816, 32828, 32839, 32851, 32862,
1735                         32873, 32885, 32895, 32906, 32917, 32927, 32938, 32948, 32958, 32968, 32978, 32988, 32998, 33007, 33017, 33026,
1736                         33036, 33045, 33054, 33063, 33072, 33081, 33090, 33098, 33107, 33115, 33124, 33132, 33140, 33149, 33157, 33165,
1737                         33173, 33181, 33188, 33196, 33204, 33212, 33219, 33227, 33234, 33241, 33249, 33256, 33263, 33270, 33278, 33285],
1738                 [33292, 33305, 33319, 33332, 33345, 33358, 33371, 33384, 33396, 33409, 33421, 33433, 33444, 33456, 33467, 33479,
1739                         33490, 33501, 33512, 33523, 33533, 33544, 33554, 33565, 33575, 33585, 33595, 33605, 33614, 33624, 33633, 33643,
1740                         33652, 33661, 33670, 33680, 33688, 33697, 33706, 33715, 33723, 33732, 33740, 33749, 33757, 33765, 33773, 33781,
1741                         33789, 33797, 33805, 33813, 33820, 33828, 33836, 33843, 33851, 33858, 33865, 33873, 33880, 33887, 33894, 33901],
1742                 [33908, 33922, 33935, 33949, 33962, 33975, 33988, 34000, 34013, 34025, 34037, 34049, 34061, 34072, 34084, 34095,
1743                         34106, 34118, 34128, 34139, 34150, 34160, 34171, 34181, 34191, 34201, 34211, 34221, 34231, 34240, 34250, 34259,
1744                         34269, 34278, 34287, 34296, 34305, 34314, 34323, 34331, 34340, 34348, 34357, 34365, 34373, 34382, 34390, 34398,
1745                         34406, 34414, 34422, 34429, 34437, 34445, 34452, 34460, 34467, 34475, 34482, 34489, 34496, 34503, 34511, 34518],
1746                 [34525, 34538, 34552, 34565, 34578, 34591, 34604, 34617, 34629, 34642, 34654, 34666, 34677, 34689, 34700, 34712,
1747                         34723, 34734, 34745, 34756, 34766, 34777, 34787, 34798, 34808, 34818, 34828, 34838, 34847, 34857, 34866, 34876,
1748                         34885, 34894, 34904, 34913, 34921, 34930, 34939, 34948, 34956, 34965, 34973, 34982, 34990, 34998, 35006, 35014,
1749                         35022, 35030, 35038, 35046, 35053, 35061, 35069, 35076, 35084, 35091, 35098, 35106, 35113, 35120, 35127, 35134],
1750                 [35141, 35155, 35168, 35182, 35195, 35208, 35221, 35233, 35246, 35258, 35270, 35282, 35294, 35306, 35317, 35328,
1751                         35340, 35351, 35361, 35372, 35383, 35393, 35404, 35414, 35424, 35434, 35444, 35454, 35464, 35473, 35483, 35492,
1752                         35502, 35511, 35520, 35529, 35538, 35547, 35556, 35564, 35573, 35581, 35590, 35598, 35606, 35615, 35623, 35631,
1753                         35639, 35647, 35655, 35662, 35670, 35678, 35685, 35693, 35700, 35708, 35715, 35722, 35729, 35736, 35744, 35751],
1754                 [35758, 35771, 35785, 35798, 35811, 35824, 35837, 35850, 35862, 35875, 35887, 35899, 35910, 35922, 35934, 35945,
1755                         35956, 35967, 35978, 35989, 35999, 36010, 36020, 36031, 36041, 36051, 36061, 36071, 36080, 36090, 36099, 36109,
1756                         36118, 36127, 36137, 36146, 36154, 36163, 36172, 36181, 36189, 36198, 36206, 36215, 36223, 36231, 36239, 36247,
1757                         36255, 36263, 36271, 36279, 36287, 36294, 36302, 36309, 36317, 36324, 36331, 36339, 36346, 36353, 36360, 36367],
1758                 [36374, 36388, 36401, 36415, 36428, 36441, 36454, 36466, 36479, 36491, 36503, 36515, 36527, 36539, 36550, 36561,
1759                         36573, 36584, 36594, 36605, 36616, 36626, 36637, 36647, 36657, 36667, 36677, 36687, 36697, 36706, 36716, 36725,
1760                         36735, 36744, 36753, 36762, 36771, 36780, 36789, 36797, 36806, 36814, 36823, 36831, 36839, 36848, 36856, 36864,
1761                         36872, 36880, 36888, 36895, 36903, 36911, 36918, 36926, 36933, 36941, 36948, 36955, 36962, 36969, 36977, 36984],
1762                 [36991, 37004, 37018, 37031, 37044, 37057, 37070, 37083, 37095, 37108, 37120, 37132, 37143, 37155, 37167, 37178,
1763                         37189, 37200, 37211, 37222, 37232, 37243, 37253, 37264, 37274, 37284, 37294, 37304, 37313, 37323, 37332, 37342,
1764                         37351, 37360, 37370, 37379, 37388, 37396, 37405, 37414, 37422, 37431, 37439, 37448, 37456, 37464, 37472, 37480,
1765                         37488, 37496, 37504, 37512, 37520, 37527, 37535, 37542, 37550, 37557, 37564, 37572, 37579, 37586, 37593, 37600],
1766                 [37607, 37621, 37634, 37648, 37661, 37674, 37687, 37699, 37712, 37724, 37736, 37748, 37760, 37772, 37783, 37794,
1767                         37806, 37817, 37828, 37838, 37849, 37859, 37870, 37880, 37890, 37900, 37910, 37920, 37930, 37939, 37949, 37958,
1768                         37968, 37977, 37986, 37995, 38004, 38013, 38022, 38030, 38039, 38047, 38056, 38064, 38072, 38081, 38089, 38097,
1769                         38105, 38113, 38121, 38128, 38136, 38144, 38151, 38159, 38166, 38174, 38181, 38188, 38195, 38202, 38210, 38217],
1770                 [38224, 38237, 38251, 38264, 38278, 38290, 38303, 38316, 38328, 38341, 38353, 38365, 38376, 38388, 38400, 38411,
1771                         38422, 38433, 38444, 38455, 38465, 38476, 38486, 38497, 38507, 38517, 38527, 38537, 38546, 38556, 38565, 38575,
1772                         38584, 38593, 38603, 38612, 38621, 38629, 38638, 38647, 38655, 38664, 38672, 38681, 38689, 38697, 38705, 38713,
1773                         38721, 38729, 38737, 38745, 38753, 38760, 38768, 38775, 38783, 38790, 38797, 38805, 38812, 38819, 38826, 38833],
1774                 [38840, 38854, 38867, 38881, 38894, 38907, 38920, 38932, 38945, 38957, 38969, 38981, 38993, 39005, 39016, 39027,
1775                         39039, 39050, 39061, 39071, 39082, 39092, 39103, 39113, 39123, 39133, 39143, 39153, 39163, 39172, 39182, 39191,
1776                         39201, 39210, 39219, 39228, 39237, 39246, 39255, 39263, 39272, 39280, 39289, 39297, 39305, 39314, 39322, 39330,
1777                         39338, 39346, 39354, 39361, 39369, 39377, 39384, 39392, 39399, 39407, 39414, 39421, 39428, 39436, 39443, 39450],
1778         ];
1779
1780         /// Approximate `log10(numerator / denominator) * 2048` using a look-up table.
1781         #[inline]
1782         pub fn negative_log10_times_2048(numerator: u64, denominator: u64) -> u64 {
1783                 // Multiply the -1 through to avoid needing to use signed numbers.
1784                 (log10_times_2048(denominator) - log10_times_2048(numerator)) as u64
1785         }
1786
1787         #[inline]
1788         fn log10_times_2048(x: u64) -> u16 {
1789                 debug_assert_ne!(x, 0);
1790                 let most_significant_bit = HIGHEST_BIT - x.leading_zeros();
1791                 let lower_bits = (x >> most_significant_bit.saturating_sub(LOWER_BITS)) & LOWER_BITMASK;
1792                 LOG10_TIMES_2048[most_significant_bit as usize][lower_bits as usize]
1793         }
1794
1795         #[cfg(test)]
1796         mod tests {
1797                 use super::*;
1798
1799                 #[test]
1800                 fn prints_negative_log10_times_2048_lookup_table() {
1801                         for msb in 0..BITS {
1802                                 for i in 0..LOWER_BITS_BOUND {
1803                                         let x = ((LOWER_BITS_BOUND + i) << (HIGHEST_BIT - LOWER_BITS)) >> (HIGHEST_BIT - msb);
1804                                         let log10_times_2048 = ((x as f64).log10() * 2048.0).round() as u16;
1805                                         assert_eq!(log10_times_2048, LOG10_TIMES_2048[msb as usize][i as usize]);
1806
1807                                         if i % LOWER_BITS_BOUND == 0 {
1808                                                 print!("\t\t[{}, ", log10_times_2048);
1809                                         } else if i % LOWER_BITS_BOUND == LOWER_BITS_BOUND - 1 {
1810                                                 println!("{}],", log10_times_2048);
1811                                         } else if i % (LOWER_BITS_BOUND/4) == LOWER_BITS_BOUND/4 - 1 {
1812                                                 print!("{},\n\t\t\t", log10_times_2048);
1813                                         } else {
1814                                                 print!("{}, ", log10_times_2048);
1815                                         }
1816                                 }
1817                         }
1818                 }
1819         }
1820 }
1821
1822 mod bucketed_history {
1823         use super::*;
1824
1825         // Because liquidity is often skewed heavily in one direction, we store historical state
1826         // distribution in buckets of different size. For backwards compatibility, buckets of size 1/8th
1827         // must fit evenly into the buckets here.
1828         //
1829         // The smallest bucket is 2^-14th of the channel, for each of our 32 buckets here we define the
1830         // width of the bucket in 2^14'ths of the channel. This increases exponentially until we reach
1831         // a full 16th of the channel's capacity, which is reapeated a few times for backwards
1832         // compatibility. The four middle buckets represent full octiles of the channel's capacity.
1833         //
1834         // For a 1 BTC channel, this let's us differentiate between failures in the bottom 6k sats, or
1835         // between the 12,000th sat and 24,000th sat, while only needing to store and operate on 32
1836         // buckets in total.
1837
1838         const BUCKET_START_POS: [u16; 33] = [
1839                 0, 1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 3072, 4096, 6144, 8192, 10240, 12288,
1840                 13312, 14336, 15360, 15872, 16128, 16256, 16320, 16352, 16368, 16376, 16380, 16382, 16383, 16384,
1841         ];
1842
1843         const LEGACY_TO_BUCKET_RANGE: [(u8, u8); 8] = [
1844                 (0, 12), (12, 14), (14, 15), (15, 16), (16, 17), (17, 18), (18, 20), (20, 32)
1845         ];
1846
1847         const POSITION_TICKS: u16 = 1 << 14;
1848
1849         fn pos_to_bucket(pos: u16) -> usize {
1850                 for bucket in 0..32 {
1851                         if pos < BUCKET_START_POS[bucket + 1] {
1852                                 return bucket;
1853                         }
1854                 }
1855                 debug_assert!(false);
1856                 return 32;
1857         }
1858
1859         #[cfg(test)]
1860         #[test]
1861         fn check_bucket_maps() {
1862                 const BUCKET_WIDTH_IN_16384S: [u16; 32] = [
1863                         1, 1, 2, 4, 8, 16, 32, 64, 128, 256, 512, 1024, 1024, 1024, 2048, 2048,
1864                         2048, 2048, 1024, 1024, 1024, 512, 256, 128, 64, 32, 16, 8, 4, 2, 1, 1];
1865
1866                 let mut min_size_iter = 0;
1867                 let mut legacy_bucket_iter = 0;
1868                 for (bucket, width) in BUCKET_WIDTH_IN_16384S.iter().enumerate() {
1869                         assert_eq!(BUCKET_START_POS[bucket], min_size_iter);
1870                         for i in 0..*width {
1871                                 assert_eq!(pos_to_bucket(min_size_iter + i) as usize, bucket);
1872                         }
1873                         min_size_iter += *width;
1874                         if min_size_iter % (POSITION_TICKS / 8) == 0 {
1875                                 assert_eq!(LEGACY_TO_BUCKET_RANGE[legacy_bucket_iter].1 as usize, bucket + 1);
1876                                 if legacy_bucket_iter + 1 < 8 {
1877                                         assert_eq!(LEGACY_TO_BUCKET_RANGE[legacy_bucket_iter + 1].0 as usize, bucket + 1);
1878                                 }
1879                                 legacy_bucket_iter += 1;
1880                         }
1881                 }
1882                 assert_eq!(BUCKET_START_POS[32], POSITION_TICKS);
1883                 assert_eq!(min_size_iter, POSITION_TICKS);
1884         }
1885
1886         #[inline]
1887         fn amount_to_pos(amount_msat: u64, capacity_msat: u64) -> u16 {
1888                 let pos = if amount_msat < u64::max_value() / (POSITION_TICKS as u64) {
1889                         (amount_msat * (POSITION_TICKS as u64) / capacity_msat.saturating_add(1))
1890                                 .try_into().unwrap_or(POSITION_TICKS)
1891                 } else {
1892                         // Only use 128-bit arithmetic when multiplication will overflow to avoid 128-bit
1893                         // division. This branch should only be hit in fuzz testing since the amount would
1894                         // need to be over 2.88 million BTC in practice.
1895                         ((amount_msat as u128) * (POSITION_TICKS as u128)
1896                                         / (capacity_msat as u128).saturating_add(1))
1897                                 .try_into().unwrap_or(POSITION_TICKS)
1898                 };
1899                 // If we are running in a client that doesn't validate gossip, its possible for a channel's
1900                 // capacity to change due to a `channel_update` message which, if received while a payment
1901                 // is in-flight, could cause this to fail. Thus, we only assert in test.
1902                 #[cfg(test)]
1903                 debug_assert!(pos < POSITION_TICKS);
1904                 pos
1905         }
1906
1907         /// Prior to LDK 0.0.117 we used eight buckets which were split evenly across the either
1908         /// octiles. This was changed to use 32 buckets for accuracy reasons in 0.0.117, however we
1909         /// support reading the legacy values here for backwards compatibility.
1910         pub(super) struct LegacyHistoricalBucketRangeTracker {
1911                 buckets: [u16; 8],
1912         }
1913
1914         impl LegacyHistoricalBucketRangeTracker {
1915                 pub(crate) fn into_current(&self) -> HistoricalBucketRangeTracker {
1916                         let mut buckets = [0; 32];
1917                         for (idx, legacy_bucket) in self.buckets.iter().enumerate() {
1918                                 let mut new_val = *legacy_bucket;
1919                                 let (start, end) = LEGACY_TO_BUCKET_RANGE[idx];
1920                                 new_val /= (end - start) as u16;
1921                                 for i in start..end {
1922                                         buckets[i as usize] = new_val;
1923                                 }
1924                         }
1925                         HistoricalBucketRangeTracker { buckets }
1926                 }
1927         }
1928
1929         /// Tracks the historical state of a distribution as a weighted average of how much time was spent
1930         /// in each of 32 buckets.
1931         #[derive(Clone, Copy)]
1932         pub(super) struct HistoricalBucketRangeTracker {
1933                 pub(super) buckets: [u16; 32],
1934         }
1935
1936         /// Buckets are stored in fixed point numbers with a 5 bit fractional part. Thus, the value
1937         /// "one" is 32, or this constant.
1938         pub const BUCKET_FIXED_POINT_ONE: u16 = 32;
1939
1940         impl HistoricalBucketRangeTracker {
1941                 pub(super) fn new() -> Self { Self { buckets: [0; 32] } }
1942                 pub(super) fn track_datapoint(&mut self, liquidity_offset_msat: u64, capacity_msat: u64) {
1943                         // We have 32 leaky buckets for min and max liquidity. Each bucket tracks the amount of time
1944                         // we spend in each bucket as a 16-bit fixed-point number with a 5 bit fractional part.
1945                         //
1946                         // Each time we update our liquidity estimate, we add 32 (1.0 in our fixed-point system) to
1947                         // the buckets for the current min and max liquidity offset positions.
1948                         //
1949                         // We then decay each bucket by multiplying by 2047/2048 (avoiding dividing by a
1950                         // non-power-of-two). This ensures we can't actually overflow the u16 - when we get to
1951                         // 63,457 adding 32 and decaying by 2047/2048 leaves us back at 63,457.
1952                         //
1953                         // In total, this allows us to track data for the last 8,000 or so payments across a given
1954                         // channel.
1955                         //
1956                         // These constants are a balance - we try to fit in 2 bytes per bucket to reduce overhead,
1957                         // and need to balance having more bits in the decimal part (to ensure decay isn't too
1958                         // non-linear) with having too few bits in the mantissa, causing us to not store very many
1959                         // datapoints.
1960                         //
1961                         // The constants were picked experimentally, selecting a decay amount that restricts us
1962                         // from overflowing buckets without having to cap them manually.
1963
1964                         let pos: u16 = amount_to_pos(liquidity_offset_msat, capacity_msat);
1965                         if pos < POSITION_TICKS {
1966                                 for e in self.buckets.iter_mut() {
1967                                         *e = ((*e as u32) * 2047 / 2048) as u16;
1968                                 }
1969                                 let bucket = pos_to_bucket(pos);
1970                                 self.buckets[bucket] = self.buckets[bucket].saturating_add(BUCKET_FIXED_POINT_ONE);
1971                         }
1972                 }
1973                 /// Decay all buckets by the given number of half-lives. Used to more aggressively remove old
1974                 /// datapoints as we receive newer information.
1975                 #[inline]
1976                 pub(super) fn time_decay_data(&mut self, half_lives: u32) {
1977                         for e in self.buckets.iter_mut() {
1978                                 *e = e.checked_shr(half_lives).unwrap_or(0);
1979                         }
1980                 }
1981         }
1982
1983         impl_writeable_tlv_based!(HistoricalBucketRangeTracker, { (0, buckets, required) });
1984         impl_writeable_tlv_based!(LegacyHistoricalBucketRangeTracker, { (0, buckets, required) });
1985
1986         /// A set of buckets representing the history of where we've seen the minimum- and maximum-
1987         /// liquidity bounds for a given channel.
1988         pub(super) struct HistoricalMinMaxBuckets<D: Deref<Target = HistoricalBucketRangeTracker>> {
1989                 /// Buckets tracking where and how often we've seen the minimum liquidity bound for a
1990                 /// channel.
1991                 pub(super) min_liquidity_offset_history: D,
1992                 /// Buckets tracking where and how often we've seen the maximum liquidity bound for a
1993                 /// channel.
1994                 pub(super) max_liquidity_offset_history: D,
1995         }
1996
1997         impl<D: Deref<Target = HistoricalBucketRangeTracker>> HistoricalMinMaxBuckets<D> {
1998                 #[inline]
1999                 pub(super) fn calculate_success_probability_times_billion(
2000                         &self, params: &ProbabilisticScoringFeeParameters, amount_msat: u64,
2001                         capacity_msat: u64
2002                 ) -> Option<u64> {
2003                         // If historical penalties are enabled, we try to calculate a probability of success
2004                         // given our historical distribution of min- and max-liquidity bounds in a channel.
2005                         // To do so, we walk the set of historical liquidity bucket (min, max) combinations
2006                         // (where min_idx < max_idx, as having a minimum above our maximum is an invalid
2007                         // state). For each pair, we calculate the probability as if the bucket's corresponding
2008                         // min- and max- liquidity bounds were our current liquidity bounds and then multiply
2009                         // that probability by the weight of the selected buckets.
2010                         let payment_pos = amount_to_pos(amount_msat, capacity_msat);
2011                         if payment_pos >= POSITION_TICKS { return None; }
2012
2013                         let mut total_valid_points_tracked = 0;
2014                         for (min_idx, min_bucket) in self.min_liquidity_offset_history.buckets.iter().enumerate() {
2015                                 for max_bucket in self.max_liquidity_offset_history.buckets.iter().take(32 - min_idx) {
2016                                         total_valid_points_tracked += (*min_bucket as u64) * (*max_bucket as u64);
2017                                 }
2018                         }
2019
2020                         // If the total valid points is smaller than 1.0 (i.e. 32 in our fixed-point scheme),
2021                         // treat it as if we were fully decayed.
2022                         const FULLY_DECAYED: u16 = BUCKET_FIXED_POINT_ONE * BUCKET_FIXED_POINT_ONE;
2023                         if total_valid_points_tracked < FULLY_DECAYED.into() {
2024                                 return None;
2025                         }
2026
2027                         let mut cumulative_success_prob_times_billion = 0;
2028                         // Special-case the 0th min bucket - it generally means we failed a payment, so only
2029                         // consider the highest (i.e. largest-offset-from-max-capacity) max bucket for all
2030                         // points against the 0th min bucket. This avoids the case where we fail to route
2031                         // increasingly lower values over a channel, but treat each failure as a separate
2032                         // datapoint, many of which may have relatively high maximum-available-liquidity
2033                         // values, which will result in us thinking we have some nontrivial probability of
2034                         // routing up to that amount.
2035                         if self.min_liquidity_offset_history.buckets[0] != 0 {
2036                                 let mut highest_max_bucket_with_points = 0; // The highest max-bucket with any data
2037                                 let mut total_max_points = 0; // Total points in max-buckets to consider
2038                                 for (max_idx, max_bucket) in self.max_liquidity_offset_history.buckets.iter().enumerate() {
2039                                         if *max_bucket >= BUCKET_FIXED_POINT_ONE {
2040                                                 highest_max_bucket_with_points = cmp::max(highest_max_bucket_with_points, max_idx);
2041                                         }
2042                                         total_max_points += *max_bucket as u64;
2043                                 }
2044                                 let max_bucket_end_pos = BUCKET_START_POS[32 - highest_max_bucket_with_points] - 1;
2045                                 if payment_pos < max_bucket_end_pos {
2046                                         let (numerator, denominator) = success_probability(payment_pos as u64, 0,
2047                                                 max_bucket_end_pos as u64, POSITION_TICKS as u64 - 1, params, true);
2048                                         let bucket_prob_times_billion =
2049                                                 (self.min_liquidity_offset_history.buckets[0] as u64) * total_max_points
2050                                                         * 1024 * 1024 * 1024 / total_valid_points_tracked;
2051                                         cumulative_success_prob_times_billion += bucket_prob_times_billion *
2052                                                 numerator / denominator;
2053                                 }
2054                         }
2055
2056                         for (min_idx, min_bucket) in self.min_liquidity_offset_history.buckets.iter().enumerate().skip(1) {
2057                                 let min_bucket_start_pos = BUCKET_START_POS[min_idx];
2058                                 for (max_idx, max_bucket) in self.max_liquidity_offset_history.buckets.iter().enumerate().take(32 - min_idx) {
2059                                         let max_bucket_end_pos = BUCKET_START_POS[32 - max_idx] - 1;
2060                                         // Note that this multiply can only barely not overflow - two 16 bit ints plus
2061                                         // 30 bits is 62 bits.
2062                                         let bucket_prob_times_billion = (*min_bucket as u64) * (*max_bucket as u64)
2063                                                 * 1024 * 1024 * 1024 / total_valid_points_tracked;
2064                                         if payment_pos >= max_bucket_end_pos {
2065                                                 // Success probability 0, the payment amount may be above the max liquidity
2066                                                 break;
2067                                         } else if payment_pos < min_bucket_start_pos {
2068                                                 cumulative_success_prob_times_billion += bucket_prob_times_billion;
2069                                         } else {
2070                                                 let (numerator, denominator) = success_probability(payment_pos as u64,
2071                                                         min_bucket_start_pos as u64, max_bucket_end_pos as u64,
2072                                                         POSITION_TICKS as u64 - 1, params, true);
2073                                                 cumulative_success_prob_times_billion += bucket_prob_times_billion *
2074                                                         numerator / denominator;
2075                                         }
2076                                 }
2077                         }
2078
2079                         Some(cumulative_success_prob_times_billion)
2080                 }
2081         }
2082 }
2083 use bucketed_history::{LegacyHistoricalBucketRangeTracker, HistoricalBucketRangeTracker, HistoricalMinMaxBuckets};
2084
2085 impl<G: Deref<Target = NetworkGraph<L>>, L: Deref> Writeable for ProbabilisticScorer<G, L> where L::Target: Logger {
2086         #[inline]
2087         fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
2088                 write_tlv_fields!(w, {
2089                         (0, self.channel_liquidities, required),
2090                 });
2091                 Ok(())
2092         }
2093 }
2094
2095 impl<G: Deref<Target = NetworkGraph<L>>, L: Deref>
2096 ReadableArgs<(ProbabilisticScoringDecayParameters, G, L)> for ProbabilisticScorer<G, L> where L::Target: Logger {
2097         #[inline]
2098         fn read<R: Read>(
2099                 r: &mut R, args: (ProbabilisticScoringDecayParameters, G, L)
2100         ) -> Result<Self, DecodeError> {
2101                 let (decay_params, network_graph, logger) = args;
2102                 let mut channel_liquidities = HashMap::new();
2103                 read_tlv_fields!(r, {
2104                         (0, channel_liquidities, required),
2105                 });
2106                 Ok(Self {
2107                         decay_params,
2108                         network_graph,
2109                         logger,
2110                         channel_liquidities,
2111                 })
2112         }
2113 }
2114
2115 impl Writeable for ChannelLiquidity {
2116         #[inline]
2117         fn write<W: Writer>(&self, w: &mut W) -> Result<(), io::Error> {
2118                 write_tlv_fields!(w, {
2119                         (0, self.min_liquidity_offset_msat, required),
2120                         // 1 was the min_liquidity_offset_history in octile form
2121                         (2, self.max_liquidity_offset_msat, required),
2122                         // 3 was the max_liquidity_offset_history in octile form
2123                         (4, self.last_updated, required),
2124                         (5, Some(self.min_liquidity_offset_history), option),
2125                         (7, Some(self.max_liquidity_offset_history), option),
2126                         (9, self.offset_history_last_updated, required),
2127                 });
2128                 Ok(())
2129         }
2130 }
2131
2132 impl Readable for ChannelLiquidity {
2133         #[inline]
2134         fn read<R: Read>(r: &mut R) -> Result<Self, DecodeError> {
2135                 let mut min_liquidity_offset_msat = 0;
2136                 let mut max_liquidity_offset_msat = 0;
2137                 let mut legacy_min_liq_offset_history: Option<LegacyHistoricalBucketRangeTracker> = None;
2138                 let mut legacy_max_liq_offset_history: Option<LegacyHistoricalBucketRangeTracker> = None;
2139                 let mut min_liquidity_offset_history: Option<HistoricalBucketRangeTracker> = None;
2140                 let mut max_liquidity_offset_history: Option<HistoricalBucketRangeTracker> = None;
2141                 let mut last_updated = Duration::from_secs(0);
2142                 let mut offset_history_last_updated = None;
2143                 read_tlv_fields!(r, {
2144                         (0, min_liquidity_offset_msat, required),
2145                         (1, legacy_min_liq_offset_history, option),
2146                         (2, max_liquidity_offset_msat, required),
2147                         (3, legacy_max_liq_offset_history, option),
2148                         (4, last_updated, required),
2149                         (5, min_liquidity_offset_history, option),
2150                         (7, max_liquidity_offset_history, option),
2151                         (9, offset_history_last_updated, option),
2152                 });
2153
2154                 if min_liquidity_offset_history.is_none() {
2155                         if let Some(legacy_buckets) = legacy_min_liq_offset_history {
2156                                 min_liquidity_offset_history = Some(legacy_buckets.into_current());
2157                         } else {
2158                                 min_liquidity_offset_history = Some(HistoricalBucketRangeTracker::new());
2159                         }
2160                 }
2161                 if max_liquidity_offset_history.is_none() {
2162                         if let Some(legacy_buckets) = legacy_max_liq_offset_history {
2163                                 max_liquidity_offset_history = Some(legacy_buckets.into_current());
2164                         } else {
2165                                 max_liquidity_offset_history = Some(HistoricalBucketRangeTracker::new());
2166                         }
2167                 }
2168                 Ok(Self {
2169                         min_liquidity_offset_msat,
2170                         max_liquidity_offset_msat,
2171                         min_liquidity_offset_history: min_liquidity_offset_history.unwrap(),
2172                         max_liquidity_offset_history: max_liquidity_offset_history.unwrap(),
2173                         last_updated,
2174                         offset_history_last_updated: offset_history_last_updated.unwrap_or(last_updated),
2175                 })
2176         }
2177 }
2178
2179 #[cfg(test)]
2180 mod tests {
2181         use super::{ChannelLiquidity, HistoricalBucketRangeTracker, ProbabilisticScoringFeeParameters, ProbabilisticScoringDecayParameters, ProbabilisticScorer};
2182         use crate::blinded_path::{BlindedHop, BlindedPath};
2183         use crate::util::config::UserConfig;
2184         use crate::util::time::tests::SinceEpoch;
2185
2186         use crate::ln::channelmanager;
2187         use crate::ln::msgs::{ChannelAnnouncement, ChannelUpdate, UnsignedChannelAnnouncement, UnsignedChannelUpdate};
2188         use crate::routing::gossip::{EffectiveCapacity, NetworkGraph, NodeId};
2189         use crate::routing::router::{BlindedTail, Path, RouteHop, CandidateRouteHop};
2190         use crate::routing::scoring::{ChannelUsage, ScoreLookUp, ScoreUpdate};
2191         use crate::util::ser::{ReadableArgs, Writeable};
2192         use crate::util::test_utils::{self, TestLogger};
2193
2194         use bitcoin::blockdata::constants::ChainHash;
2195         use bitcoin::hashes::Hash;
2196         use bitcoin::hashes::sha256d::Hash as Sha256dHash;
2197         use bitcoin::network::constants::Network;
2198         use bitcoin::secp256k1::{PublicKey, Secp256k1, SecretKey};
2199         use core::time::Duration;
2200         use crate::io;
2201
2202         fn source_privkey() -> SecretKey {
2203                 SecretKey::from_slice(&[42; 32]).unwrap()
2204         }
2205
2206         fn target_privkey() -> SecretKey {
2207                 SecretKey::from_slice(&[43; 32]).unwrap()
2208         }
2209
2210         fn source_pubkey() -> PublicKey {
2211                 let secp_ctx = Secp256k1::new();
2212                 PublicKey::from_secret_key(&secp_ctx, &source_privkey())
2213         }
2214
2215         fn target_pubkey() -> PublicKey {
2216                 let secp_ctx = Secp256k1::new();
2217                 PublicKey::from_secret_key(&secp_ctx, &target_privkey())
2218         }
2219
2220         fn source_node_id() -> NodeId {
2221                 NodeId::from_pubkey(&source_pubkey())
2222         }
2223
2224         fn target_node_id() -> NodeId {
2225                 NodeId::from_pubkey(&target_pubkey())
2226         }
2227
2228         // `ProbabilisticScorer` tests
2229
2230         fn sender_privkey() -> SecretKey {
2231                 SecretKey::from_slice(&[41; 32]).unwrap()
2232         }
2233
2234         fn recipient_privkey() -> SecretKey {
2235                 SecretKey::from_slice(&[45; 32]).unwrap()
2236         }
2237
2238         fn sender_pubkey() -> PublicKey {
2239                 let secp_ctx = Secp256k1::new();
2240                 PublicKey::from_secret_key(&secp_ctx, &sender_privkey())
2241         }
2242
2243         fn recipient_pubkey() -> PublicKey {
2244                 let secp_ctx = Secp256k1::new();
2245                 PublicKey::from_secret_key(&secp_ctx, &recipient_privkey())
2246         }
2247
2248         fn recipient_node_id() -> NodeId {
2249                 NodeId::from_pubkey(&recipient_pubkey())
2250         }
2251
2252         fn network_graph(logger: &TestLogger) -> NetworkGraph<&TestLogger> {
2253                 let mut network_graph = NetworkGraph::new(Network::Testnet, logger);
2254                 add_channel(&mut network_graph, 42, source_privkey(), target_privkey());
2255                 add_channel(&mut network_graph, 43, target_privkey(), recipient_privkey());
2256
2257                 network_graph
2258         }
2259
2260         fn add_channel(
2261                 network_graph: &mut NetworkGraph<&TestLogger>, short_channel_id: u64, node_1_key: SecretKey,
2262                 node_2_key: SecretKey
2263         ) {
2264                 let genesis_hash = ChainHash::using_genesis_block(Network::Testnet);
2265                 let node_1_secret = &SecretKey::from_slice(&[39; 32]).unwrap();
2266                 let node_2_secret = &SecretKey::from_slice(&[40; 32]).unwrap();
2267                 let secp_ctx = Secp256k1::new();
2268                 let unsigned_announcement = UnsignedChannelAnnouncement {
2269                         features: channelmanager::provided_channel_features(&UserConfig::default()),
2270                         chain_hash: genesis_hash,
2271                         short_channel_id,
2272                         node_id_1: NodeId::from_pubkey(&PublicKey::from_secret_key(&secp_ctx, &node_1_key)),
2273                         node_id_2: NodeId::from_pubkey(&PublicKey::from_secret_key(&secp_ctx, &node_2_key)),
2274                         bitcoin_key_1: NodeId::from_pubkey(&PublicKey::from_secret_key(&secp_ctx, &node_1_secret)),
2275                         bitcoin_key_2: NodeId::from_pubkey(&PublicKey::from_secret_key(&secp_ctx, &node_2_secret)),
2276                         excess_data: Vec::new(),
2277                 };
2278                 let msghash = hash_to_message!(&Sha256dHash::hash(&unsigned_announcement.encode()[..])[..]);
2279                 let signed_announcement = ChannelAnnouncement {
2280                         node_signature_1: secp_ctx.sign_ecdsa(&msghash, &node_1_key),
2281                         node_signature_2: secp_ctx.sign_ecdsa(&msghash, &node_2_key),
2282                         bitcoin_signature_1: secp_ctx.sign_ecdsa(&msghash, &node_1_secret),
2283                         bitcoin_signature_2: secp_ctx.sign_ecdsa(&msghash, &node_2_secret),
2284                         contents: unsigned_announcement,
2285                 };
2286                 let chain_source: Option<&crate::util::test_utils::TestChainSource> = None;
2287                 network_graph.update_channel_from_announcement(
2288                         &signed_announcement, &chain_source).unwrap();
2289                 update_channel(network_graph, short_channel_id, node_1_key, 0, 1_000, 100);
2290                 update_channel(network_graph, short_channel_id, node_2_key, 1, 0, 100);
2291         }
2292
2293         fn update_channel(
2294                 network_graph: &mut NetworkGraph<&TestLogger>, short_channel_id: u64, node_key: SecretKey,
2295                 flags: u8, htlc_maximum_msat: u64, timestamp: u32,
2296         ) {
2297                 let genesis_hash = ChainHash::using_genesis_block(Network::Testnet);
2298                 let secp_ctx = Secp256k1::new();
2299                 let unsigned_update = UnsignedChannelUpdate {
2300                         chain_hash: genesis_hash,
2301                         short_channel_id,
2302                         timestamp,
2303                         flags,
2304                         cltv_expiry_delta: 18,
2305                         htlc_minimum_msat: 0,
2306                         htlc_maximum_msat,
2307                         fee_base_msat: 1,
2308                         fee_proportional_millionths: 0,
2309                         excess_data: Vec::new(),
2310                 };
2311                 let msghash = hash_to_message!(&Sha256dHash::hash(&unsigned_update.encode()[..])[..]);
2312                 let signed_update = ChannelUpdate {
2313                         signature: secp_ctx.sign_ecdsa(&msghash, &node_key),
2314                         contents: unsigned_update,
2315                 };
2316                 network_graph.update_channel(&signed_update).unwrap();
2317         }
2318
2319         fn path_hop(pubkey: PublicKey, short_channel_id: u64, fee_msat: u64) -> RouteHop {
2320                 let config = UserConfig::default();
2321                 RouteHop {
2322                         pubkey,
2323                         node_features: channelmanager::provided_node_features(&config),
2324                         short_channel_id,
2325                         channel_features: channelmanager::provided_channel_features(&config),
2326                         fee_msat,
2327                         cltv_expiry_delta: 18,
2328                         maybe_announced_channel: true,
2329                 }
2330         }
2331
2332         fn payment_path_for_amount(amount_msat: u64) -> Path {
2333                 Path {
2334                         hops: vec![
2335                                 path_hop(source_pubkey(), 41, 1),
2336                                 path_hop(target_pubkey(), 42, 2),
2337                                 path_hop(recipient_pubkey(), 43, amount_msat),
2338                         ], blinded_tail: None,
2339                 }
2340         }
2341
2342         #[test]
2343         fn liquidity_bounds_directed_from_lowest_node_id() {
2344                 let logger = TestLogger::new();
2345                 let last_updated = Duration::ZERO;
2346                 let offset_history_last_updated = Duration::ZERO;
2347                 let network_graph = network_graph(&logger);
2348                 let decay_params = ProbabilisticScoringDecayParameters::default();
2349                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger)
2350                         .with_channel(42,
2351                                 ChannelLiquidity {
2352                                         min_liquidity_offset_msat: 700, max_liquidity_offset_msat: 100,
2353                                         last_updated, offset_history_last_updated,
2354                                         min_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2355                                         max_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2356                                 })
2357                         .with_channel(43,
2358                                 ChannelLiquidity {
2359                                         min_liquidity_offset_msat: 700, max_liquidity_offset_msat: 100,
2360                                         last_updated, offset_history_last_updated,
2361                                         min_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2362                                         max_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2363                                 });
2364                 let source = source_node_id();
2365                 let target = target_node_id();
2366                 let recipient = recipient_node_id();
2367                 assert!(source > target);
2368                 assert!(target < recipient);
2369
2370                 // Update minimum liquidity.
2371
2372                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2373                         .as_directed(&source, &target, 1_000, decay_params);
2374                 assert_eq!(liquidity.min_liquidity_msat(), 100);
2375                 assert_eq!(liquidity.max_liquidity_msat(), 300);
2376
2377                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2378                         .as_directed(&target, &source, 1_000, decay_params);
2379                 assert_eq!(liquidity.min_liquidity_msat(), 700);
2380                 assert_eq!(liquidity.max_liquidity_msat(), 900);
2381
2382                 scorer.channel_liquidities.get_mut(&42).unwrap()
2383                         .as_directed_mut(&source, &target, 1_000, decay_params)
2384                         .set_min_liquidity_msat(200, Duration::ZERO);
2385
2386                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2387                         .as_directed(&source, &target, 1_000, decay_params);
2388                 assert_eq!(liquidity.min_liquidity_msat(), 200);
2389                 assert_eq!(liquidity.max_liquidity_msat(), 300);
2390
2391                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2392                         .as_directed(&target, &source, 1_000, decay_params);
2393                 assert_eq!(liquidity.min_liquidity_msat(), 700);
2394                 assert_eq!(liquidity.max_liquidity_msat(), 800);
2395
2396                 // Update maximum liquidity.
2397
2398                 let liquidity = scorer.channel_liquidities.get(&43).unwrap()
2399                         .as_directed(&target, &recipient, 1_000, decay_params);
2400                 assert_eq!(liquidity.min_liquidity_msat(), 700);
2401                 assert_eq!(liquidity.max_liquidity_msat(), 900);
2402
2403                 let liquidity = scorer.channel_liquidities.get(&43).unwrap()
2404                         .as_directed(&recipient, &target, 1_000, decay_params);
2405                 assert_eq!(liquidity.min_liquidity_msat(), 100);
2406                 assert_eq!(liquidity.max_liquidity_msat(), 300);
2407
2408                 scorer.channel_liquidities.get_mut(&43).unwrap()
2409                         .as_directed_mut(&target, &recipient, 1_000, decay_params)
2410                         .set_max_liquidity_msat(200, Duration::ZERO);
2411
2412                 let liquidity = scorer.channel_liquidities.get(&43).unwrap()
2413                         .as_directed(&target, &recipient, 1_000, decay_params);
2414                 assert_eq!(liquidity.min_liquidity_msat(), 0);
2415                 assert_eq!(liquidity.max_liquidity_msat(), 200);
2416
2417                 let liquidity = scorer.channel_liquidities.get(&43).unwrap()
2418                         .as_directed(&recipient, &target, 1_000, decay_params);
2419                 assert_eq!(liquidity.min_liquidity_msat(), 800);
2420                 assert_eq!(liquidity.max_liquidity_msat(), 1000);
2421         }
2422
2423         #[test]
2424         fn resets_liquidity_upper_bound_when_crossed_by_lower_bound() {
2425                 let logger = TestLogger::new();
2426                 let last_updated = Duration::ZERO;
2427                 let offset_history_last_updated = Duration::ZERO;
2428                 let network_graph = network_graph(&logger);
2429                 let decay_params = ProbabilisticScoringDecayParameters::default();
2430                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger)
2431                         .with_channel(42,
2432                                 ChannelLiquidity {
2433                                         min_liquidity_offset_msat: 200, max_liquidity_offset_msat: 400,
2434                                         last_updated, offset_history_last_updated,
2435                                         min_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2436                                         max_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2437                                 });
2438                 let source = source_node_id();
2439                 let target = target_node_id();
2440                 assert!(source > target);
2441
2442                 // Check initial bounds.
2443                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2444                         .as_directed(&source, &target, 1_000, decay_params);
2445                 assert_eq!(liquidity.min_liquidity_msat(), 400);
2446                 assert_eq!(liquidity.max_liquidity_msat(), 800);
2447
2448                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2449                         .as_directed(&target, &source, 1_000, decay_params);
2450                 assert_eq!(liquidity.min_liquidity_msat(), 200);
2451                 assert_eq!(liquidity.max_liquidity_msat(), 600);
2452
2453                 // Reset from source to target.
2454                 scorer.channel_liquidities.get_mut(&42).unwrap()
2455                         .as_directed_mut(&source, &target, 1_000, decay_params)
2456                         .set_min_liquidity_msat(900, Duration::ZERO);
2457
2458                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2459                         .as_directed(&source, &target, 1_000, decay_params);
2460                 assert_eq!(liquidity.min_liquidity_msat(), 900);
2461                 assert_eq!(liquidity.max_liquidity_msat(), 1_000);
2462
2463                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2464                         .as_directed(&target, &source, 1_000, decay_params);
2465                 assert_eq!(liquidity.min_liquidity_msat(), 0);
2466                 assert_eq!(liquidity.max_liquidity_msat(), 100);
2467
2468                 // Reset from target to source.
2469                 scorer.channel_liquidities.get_mut(&42).unwrap()
2470                         .as_directed_mut(&target, &source, 1_000, decay_params)
2471                         .set_min_liquidity_msat(400, Duration::ZERO);
2472
2473                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2474                         .as_directed(&source, &target, 1_000, decay_params);
2475                 assert_eq!(liquidity.min_liquidity_msat(), 0);
2476                 assert_eq!(liquidity.max_liquidity_msat(), 600);
2477
2478                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2479                         .as_directed(&target, &source, 1_000, decay_params);
2480                 assert_eq!(liquidity.min_liquidity_msat(), 400);
2481                 assert_eq!(liquidity.max_liquidity_msat(), 1_000);
2482         }
2483
2484         #[test]
2485         fn resets_liquidity_lower_bound_when_crossed_by_upper_bound() {
2486                 let logger = TestLogger::new();
2487                 let last_updated = Duration::ZERO;
2488                 let offset_history_last_updated = Duration::ZERO;
2489                 let network_graph = network_graph(&logger);
2490                 let decay_params = ProbabilisticScoringDecayParameters::default();
2491                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger)
2492                         .with_channel(42,
2493                                 ChannelLiquidity {
2494                                         min_liquidity_offset_msat: 200, max_liquidity_offset_msat: 400,
2495                                         last_updated, offset_history_last_updated,
2496                                         min_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2497                                         max_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2498                                 });
2499                 let source = source_node_id();
2500                 let target = target_node_id();
2501                 assert!(source > target);
2502
2503                 // Check initial bounds.
2504                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2505                         .as_directed(&source, &target, 1_000, decay_params);
2506                 assert_eq!(liquidity.min_liquidity_msat(), 400);
2507                 assert_eq!(liquidity.max_liquidity_msat(), 800);
2508
2509                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2510                         .as_directed(&target, &source, 1_000, decay_params);
2511                 assert_eq!(liquidity.min_liquidity_msat(), 200);
2512                 assert_eq!(liquidity.max_liquidity_msat(), 600);
2513
2514                 // Reset from source to target.
2515                 scorer.channel_liquidities.get_mut(&42).unwrap()
2516                         .as_directed_mut(&source, &target, 1_000, decay_params)
2517                         .set_max_liquidity_msat(300, Duration::ZERO);
2518
2519                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2520                         .as_directed(&source, &target, 1_000, decay_params);
2521                 assert_eq!(liquidity.min_liquidity_msat(), 0);
2522                 assert_eq!(liquidity.max_liquidity_msat(), 300);
2523
2524                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2525                         .as_directed(&target, &source, 1_000, decay_params);
2526                 assert_eq!(liquidity.min_liquidity_msat(), 700);
2527                 assert_eq!(liquidity.max_liquidity_msat(), 1_000);
2528
2529                 // Reset from target to source.
2530                 scorer.channel_liquidities.get_mut(&42).unwrap()
2531                         .as_directed_mut(&target, &source, 1_000, decay_params)
2532                         .set_max_liquidity_msat(600, Duration::ZERO);
2533
2534                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2535                         .as_directed(&source, &target, 1_000, decay_params);
2536                 assert_eq!(liquidity.min_liquidity_msat(), 400);
2537                 assert_eq!(liquidity.max_liquidity_msat(), 1_000);
2538
2539                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
2540                         .as_directed(&target, &source, 1_000, decay_params);
2541                 assert_eq!(liquidity.min_liquidity_msat(), 0);
2542                 assert_eq!(liquidity.max_liquidity_msat(), 600);
2543         }
2544
2545         #[test]
2546         fn increased_penalty_nearing_liquidity_upper_bound() {
2547                 let logger = TestLogger::new();
2548                 let network_graph = network_graph(&logger);
2549                 let params = ProbabilisticScoringFeeParameters {
2550                         liquidity_penalty_multiplier_msat: 1_000,
2551                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2552                 };
2553                 let decay_params = ProbabilisticScoringDecayParameters::default();
2554                 let scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
2555                 let source = source_node_id();
2556
2557                 let usage = ChannelUsage {
2558                         amount_msat: 1_024,
2559                         inflight_htlc_msat: 0,
2560                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: 1_000 },
2561                 };
2562                 let network_graph = network_graph.read_only();
2563                 let channel = network_graph.channel(42).unwrap();
2564                 let (info, _) = channel.as_directed_from(&source).unwrap();
2565                 let candidate = CandidateRouteHop::PublicHop {
2566                         info,
2567                         short_channel_id: 42,
2568                 };
2569                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2570                 let usage = ChannelUsage { amount_msat: 10_240, ..usage };
2571                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2572                 let usage = ChannelUsage { amount_msat: 102_400, ..usage };
2573                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 47);
2574                 let usage = ChannelUsage { amount_msat: 1_023_999, ..usage };
2575                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 2_000);
2576
2577                 let usage = ChannelUsage {
2578                         amount_msat: 128,
2579                         inflight_htlc_msat: 0,
2580                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_000 },
2581                 };
2582                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 58);
2583                 let usage = ChannelUsage { amount_msat: 256, ..usage };
2584                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 125);
2585                 let usage = ChannelUsage { amount_msat: 374, ..usage };
2586                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 198);
2587                 let usage = ChannelUsage { amount_msat: 512, ..usage };
2588                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 300);
2589                 let usage = ChannelUsage { amount_msat: 640, ..usage };
2590                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 425);
2591                 let usage = ChannelUsage { amount_msat: 768, ..usage };
2592                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 602);
2593                 let usage = ChannelUsage { amount_msat: 896, ..usage };
2594                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 902);
2595         }
2596
2597         #[test]
2598         fn constant_penalty_outside_liquidity_bounds() {
2599                 let logger = TestLogger::new();
2600                 let last_updated = Duration::ZERO;
2601                 let offset_history_last_updated = Duration::ZERO;
2602                 let network_graph = network_graph(&logger);
2603                 let params = ProbabilisticScoringFeeParameters {
2604                         liquidity_penalty_multiplier_msat: 1_000,
2605                         considered_impossible_penalty_msat: u64::max_value(),
2606                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2607                 };
2608                 let decay_params = ProbabilisticScoringDecayParameters {
2609                         ..ProbabilisticScoringDecayParameters::zero_penalty()
2610                 };
2611                 let scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger)
2612                         .with_channel(42,
2613                                 ChannelLiquidity {
2614                                         min_liquidity_offset_msat: 40, max_liquidity_offset_msat: 40,
2615                                         last_updated, offset_history_last_updated,
2616                                         min_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2617                                         max_liquidity_offset_history: HistoricalBucketRangeTracker::new(),
2618                                 });
2619                 let source = source_node_id();
2620
2621                 let usage = ChannelUsage {
2622                         amount_msat: 39,
2623                         inflight_htlc_msat: 0,
2624                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 100, htlc_maximum_msat: 1_000 },
2625                 };
2626                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
2627                 let (info, _) = channel.as_directed_from(&source).unwrap();
2628                 let candidate = CandidateRouteHop::PublicHop {
2629                         info,
2630                         short_channel_id: 42,
2631                 };
2632                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2633                 let usage = ChannelUsage { amount_msat: 50, ..usage };
2634                 assert_ne!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2635                 assert_ne!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2636                 let usage = ChannelUsage { amount_msat: 61, ..usage };
2637                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2638         }
2639
2640         #[test]
2641         fn does_not_further_penalize_own_channel() {
2642                 let logger = TestLogger::new();
2643                 let network_graph = network_graph(&logger);
2644                 let params = ProbabilisticScoringFeeParameters {
2645                         liquidity_penalty_multiplier_msat: 1_000,
2646                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2647                 };
2648                 let mut scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
2649                 let source = source_node_id();
2650                 let usage = ChannelUsage {
2651                         amount_msat: 500,
2652                         inflight_htlc_msat: 0,
2653                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_000, htlc_maximum_msat: 1_000 },
2654                 };
2655                 let failed_path = payment_path_for_amount(500);
2656                 let successful_path = payment_path_for_amount(200);
2657                 let channel = &network_graph.read_only().channel(42).unwrap().to_owned();
2658                 let (info, _) = channel.as_directed_from(&source).unwrap();
2659                 let candidate = CandidateRouteHop::PublicHop {
2660                         info,
2661                         short_channel_id: 41,
2662                 };
2663
2664                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 301);
2665
2666                 scorer.payment_path_failed(&failed_path, 41, Duration::ZERO);
2667                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 301);
2668
2669                 scorer.payment_path_successful(&successful_path, Duration::ZERO);
2670                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 301);
2671         }
2672
2673         #[test]
2674         fn sets_liquidity_lower_bound_on_downstream_failure() {
2675                 let logger = TestLogger::new();
2676                 let network_graph = network_graph(&logger);
2677                 let params = ProbabilisticScoringFeeParameters {
2678                         liquidity_penalty_multiplier_msat: 1_000,
2679                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2680                 };
2681                 let mut scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
2682                 let source = source_node_id();
2683                 let path = payment_path_for_amount(500);
2684
2685                 let usage = ChannelUsage {
2686                         amount_msat: 250,
2687                         inflight_htlc_msat: 0,
2688                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_000, htlc_maximum_msat: 1_000 },
2689                 };
2690                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
2691                 let (info, _) = channel.as_directed_from(&source).unwrap();
2692                 let candidate = CandidateRouteHop::PublicHop {
2693                         info,
2694                         short_channel_id: 42,
2695                 };
2696                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 128);
2697                 let usage = ChannelUsage { amount_msat: 500, ..usage };
2698                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 301);
2699                 let usage = ChannelUsage { amount_msat: 750, ..usage };
2700                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 602);
2701
2702                 scorer.payment_path_failed(&path, 43, Duration::ZERO);
2703
2704                 let usage = ChannelUsage { amount_msat: 250, ..usage };
2705                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2706                 let usage = ChannelUsage { amount_msat: 500, ..usage };
2707                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2708                 let usage = ChannelUsage { amount_msat: 750, ..usage };
2709                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 300);
2710         }
2711
2712         #[test]
2713         fn sets_liquidity_upper_bound_on_failure() {
2714                 let logger = TestLogger::new();
2715                 let network_graph = network_graph(&logger);
2716                 let params = ProbabilisticScoringFeeParameters {
2717                         liquidity_penalty_multiplier_msat: 1_000,
2718                         considered_impossible_penalty_msat: u64::max_value(),
2719                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2720                 };
2721                 let mut scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
2722                 let source = source_node_id();
2723                 let path = payment_path_for_amount(500);
2724
2725                 let usage = ChannelUsage {
2726                         amount_msat: 250,
2727                         inflight_htlc_msat: 0,
2728                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_000, htlc_maximum_msat: 1_000 },
2729                 };
2730                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
2731                 let (info, _) = channel.as_directed_from(&source).unwrap();
2732                 let candidate = CandidateRouteHop::PublicHop {
2733                         info,
2734                         short_channel_id: 42,
2735                 };
2736                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 128);
2737                 let usage = ChannelUsage { amount_msat: 500, ..usage };
2738                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 301);
2739                 let usage = ChannelUsage { amount_msat: 750, ..usage };
2740                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 602);
2741
2742                 scorer.payment_path_failed(&path, 42, Duration::ZERO);
2743
2744                 let usage = ChannelUsage { amount_msat: 250, ..usage };
2745                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 300);
2746                 let usage = ChannelUsage { amount_msat: 500, ..usage };
2747                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2748                 let usage = ChannelUsage { amount_msat: 750, ..usage };
2749                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2750         }
2751
2752         #[test]
2753         fn ignores_channels_after_removed_failed_channel() {
2754                 // Previously, if we'd tried to send over a channel which was removed from the network
2755                 // graph before we call `payment_path_failed` (which is the default if the we get a "no
2756                 // such channel" error in the `InvoicePayer`), we would call `failed_downstream` on all
2757                 // channels in the route, even ones which they payment never reached. This tests to ensure
2758                 // we do not score such channels.
2759                 let secp_ctx = Secp256k1::new();
2760                 let logger = TestLogger::new();
2761                 let mut network_graph = NetworkGraph::new(Network::Testnet, &logger);
2762                 let secret_a = SecretKey::from_slice(&[42; 32]).unwrap();
2763                 let secret_b = SecretKey::from_slice(&[43; 32]).unwrap();
2764                 let secret_c = SecretKey::from_slice(&[44; 32]).unwrap();
2765                 let secret_d = SecretKey::from_slice(&[45; 32]).unwrap();
2766                 add_channel(&mut network_graph, 42, secret_a, secret_b);
2767                 // Don't add the channel from B -> C.
2768                 add_channel(&mut network_graph, 44, secret_c, secret_d);
2769
2770                 let pub_a = PublicKey::from_secret_key(&secp_ctx, &secret_a);
2771                 let pub_b = PublicKey::from_secret_key(&secp_ctx, &secret_b);
2772                 let pub_c = PublicKey::from_secret_key(&secp_ctx, &secret_c);
2773                 let pub_d = PublicKey::from_secret_key(&secp_ctx, &secret_d);
2774
2775                 let path = vec![
2776                         path_hop(pub_b, 42, 1),
2777                         path_hop(pub_c, 43, 2),
2778                         path_hop(pub_d, 44, 100),
2779                 ];
2780
2781                 let node_a = NodeId::from_pubkey(&pub_a);
2782                 let node_b = NodeId::from_pubkey(&pub_b);
2783                 let node_c = NodeId::from_pubkey(&pub_c);
2784
2785                 let params = ProbabilisticScoringFeeParameters {
2786                         liquidity_penalty_multiplier_msat: 1_000,
2787                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2788                 };
2789                 let mut scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
2790
2791                 let usage = ChannelUsage {
2792                         amount_msat: 250,
2793                         inflight_htlc_msat: 0,
2794                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_000, htlc_maximum_msat: 1_000 },
2795                 };
2796                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
2797                 let (info, _) = channel.as_directed_from(&node_a).unwrap();
2798                 let candidate = CandidateRouteHop::PublicHop {
2799                         info,
2800                         short_channel_id: 42,
2801                 };
2802                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 128);
2803                 // Note that a default liquidity bound is used for B -> C as no channel exists
2804                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
2805                 let (info, _) = channel.as_directed_from(&node_b).unwrap();
2806                 let candidate = CandidateRouteHop::PublicHop {
2807                         info,
2808                         short_channel_id: 43,
2809                 };
2810                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 128);
2811                 let channel = network_graph.read_only().channel(44).unwrap().to_owned();
2812                 let (info, _) = channel.as_directed_from(&node_c).unwrap();
2813                 let candidate = CandidateRouteHop::PublicHop {
2814                         info,
2815                         short_channel_id: 44,
2816                 };
2817                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 128);
2818
2819                 scorer.payment_path_failed(&Path { hops: path, blinded_tail: None }, 43, Duration::ZERO);
2820
2821                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
2822                 let (info, _) = channel.as_directed_from(&node_a).unwrap();
2823                 let candidate = CandidateRouteHop::PublicHop {
2824                         info,
2825                         short_channel_id: 42,
2826                 };
2827                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 80);
2828                 // Note that a default liquidity bound is used for B -> C as no channel exists
2829                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
2830                 let (info, _) = channel.as_directed_from(&node_b).unwrap();
2831                 let candidate = CandidateRouteHop::PublicHop {
2832                         info,
2833                         short_channel_id: 43,
2834                 };
2835                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 128);
2836                 let channel = network_graph.read_only().channel(44).unwrap().to_owned();
2837                 let (info, _) = channel.as_directed_from(&node_c).unwrap();
2838                 let candidate = CandidateRouteHop::PublicHop {
2839                         info,
2840                         short_channel_id: 44,
2841                 };
2842                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 128);
2843         }
2844
2845         #[test]
2846         fn reduces_liquidity_upper_bound_along_path_on_success() {
2847                 let logger = TestLogger::new();
2848                 let network_graph = network_graph(&logger);
2849                 let params = ProbabilisticScoringFeeParameters {
2850                         liquidity_penalty_multiplier_msat: 1_000,
2851                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2852                 };
2853                 let mut scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
2854                 let source = source_node_id();
2855                 let usage = ChannelUsage {
2856                         amount_msat: 250,
2857                         inflight_htlc_msat: 0,
2858                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_000, htlc_maximum_msat: 1_000 },
2859                 };
2860                 let network_graph = network_graph.read_only().channels().clone();
2861                 let channel_42 = network_graph.get(&42).unwrap();
2862                 let channel_43 = network_graph.get(&43).unwrap();
2863                 let (info, _) = channel_42.as_directed_from(&source).unwrap();
2864                 let candidate_41 = CandidateRouteHop::PublicHop {
2865                         info,
2866                         short_channel_id: 41,
2867                 };
2868                 let (info, target) = channel_42.as_directed_from(&source).unwrap();
2869                 let candidate_42 = CandidateRouteHop::PublicHop {
2870                         info,
2871                         short_channel_id: 42,
2872                 };
2873                 let (info, _) = channel_43.as_directed_from(&target).unwrap();
2874                 let candidate_43 = CandidateRouteHop::PublicHop {
2875                         info,
2876                         short_channel_id: 43,
2877                 };
2878                 assert_eq!(scorer.channel_penalty_msat(&candidate_41, usage, &params), 128);
2879                 assert_eq!(scorer.channel_penalty_msat(&candidate_42, usage, &params), 128);
2880                 assert_eq!(scorer.channel_penalty_msat(&candidate_43, usage, &params), 128);
2881
2882                 scorer.payment_path_successful(&payment_path_for_amount(500), Duration::ZERO);
2883
2884                 assert_eq!(scorer.channel_penalty_msat(&candidate_41, usage, &params), 128);
2885                 assert_eq!(scorer.channel_penalty_msat(&candidate_42, usage, &params), 300);
2886                 assert_eq!(scorer.channel_penalty_msat(&candidate_43, usage, &params), 300);
2887         }
2888
2889         #[test]
2890         fn decays_liquidity_bounds_over_time() {
2891                 let logger = TestLogger::new();
2892                 let network_graph = network_graph(&logger);
2893                 let params = ProbabilisticScoringFeeParameters {
2894                         liquidity_penalty_multiplier_msat: 1_000,
2895                         considered_impossible_penalty_msat: u64::max_value(),
2896                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2897                 };
2898                 let decay_params = ProbabilisticScoringDecayParameters {
2899                         liquidity_offset_half_life: Duration::from_secs(10),
2900                         ..ProbabilisticScoringDecayParameters::zero_penalty()
2901                 };
2902                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
2903                 let source = source_node_id();
2904
2905                 let usage = ChannelUsage {
2906                         amount_msat: 0,
2907                         inflight_htlc_msat: 0,
2908                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_024 },
2909                 };
2910                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
2911                 let (info, _) = channel.as_directed_from(&source).unwrap();
2912                 let candidate = CandidateRouteHop::PublicHop {
2913                         info,
2914                         short_channel_id: 42,
2915                 };
2916                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2917                 let usage = ChannelUsage { amount_msat: 1_023, ..usage };
2918                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 2_000);
2919
2920                 scorer.payment_path_failed(&payment_path_for_amount(768), 42, Duration::ZERO);
2921                 scorer.payment_path_failed(&payment_path_for_amount(128), 43, Duration::ZERO);
2922
2923                 // Initial penalties
2924                 let usage = ChannelUsage { amount_msat: 128, ..usage };
2925                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2926                 let usage = ChannelUsage { amount_msat: 256, ..usage };
2927                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 93);
2928                 let usage = ChannelUsage { amount_msat: 768, ..usage };
2929                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 1_479);
2930                 let usage = ChannelUsage { amount_msat: 896, ..usage };
2931                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2932
2933                 // Half decay (i.e., three-quarter life)
2934                 SinceEpoch::advance(Duration::from_secs(5));
2935                 scorer.time_passed(Duration::from_secs(5));
2936                 let usage = ChannelUsage { amount_msat: 128, ..usage };
2937                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 22);
2938                 let usage = ChannelUsage { amount_msat: 256, ..usage };
2939                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 106);
2940                 let usage = ChannelUsage { amount_msat: 768, ..usage };
2941                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 921);
2942                 let usage = ChannelUsage { amount_msat: 896, ..usage };
2943                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2944
2945                 // One decay (i.e., half life)
2946                 SinceEpoch::advance(Duration::from_secs(5));
2947                 scorer.time_passed(Duration::from_secs(10));
2948                 let usage = ChannelUsage { amount_msat: 64, ..usage };
2949                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2950                 let usage = ChannelUsage { amount_msat: 128, ..usage };
2951                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 34);
2952                 let usage = ChannelUsage { amount_msat: 896, ..usage };
2953                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 1_970);
2954                 let usage = ChannelUsage { amount_msat: 960, ..usage };
2955                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2956
2957                 // Fully decay liquidity lower bound.
2958                 SinceEpoch::advance(Duration::from_secs(10 * 7));
2959                 scorer.time_passed(Duration::from_secs(10 * 8));
2960                 let usage = ChannelUsage { amount_msat: 0, ..usage };
2961                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2962                 let usage = ChannelUsage { amount_msat: 1, ..usage };
2963                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2964                 let usage = ChannelUsage { amount_msat: 1_023, ..usage };
2965                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 2_000);
2966                 let usage = ChannelUsage { amount_msat: 1_024, ..usage };
2967                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2968
2969                 // Fully decay liquidity upper bound.
2970                 SinceEpoch::advance(Duration::from_secs(10));
2971                 scorer.time_passed(Duration::from_secs(10 * 9));
2972                 let usage = ChannelUsage { amount_msat: 0, ..usage };
2973                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2974                 let usage = ChannelUsage { amount_msat: 1_024, ..usage };
2975                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2976
2977                 SinceEpoch::advance(Duration::from_secs(10));
2978                 scorer.time_passed(Duration::from_secs(10 * 10));
2979                 let usage = ChannelUsage { amount_msat: 0, ..usage };
2980                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
2981                 let usage = ChannelUsage { amount_msat: 1_024, ..usage };
2982                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
2983         }
2984
2985         #[test]
2986         fn decays_liquidity_bounds_without_shift_overflow() {
2987                 let logger = TestLogger::new();
2988                 let network_graph = network_graph(&logger);
2989                 let params = ProbabilisticScoringFeeParameters {
2990                         liquidity_penalty_multiplier_msat: 1_000,
2991                         ..ProbabilisticScoringFeeParameters::zero_penalty()
2992                 };
2993                 let decay_params = ProbabilisticScoringDecayParameters {
2994                         liquidity_offset_half_life: Duration::from_secs(10),
2995                         ..ProbabilisticScoringDecayParameters::default()
2996                 };
2997                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
2998                 let source = source_node_id();
2999                 let usage = ChannelUsage {
3000                         amount_msat: 256,
3001                         inflight_htlc_msat: 0,
3002                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_000 },
3003                 };
3004                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3005                 let (info, _) = channel.as_directed_from(&source).unwrap();
3006                 let candidate = CandidateRouteHop::PublicHop {
3007                         info,
3008                         short_channel_id: 42,
3009                 };
3010                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 125);
3011
3012                 scorer.payment_path_failed(&payment_path_for_amount(512), 42, Duration::ZERO);
3013                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 281);
3014
3015                 // An unchecked right shift 64 bits or more in DirectedChannelLiquidity::decayed_offset_msat
3016                 // would cause an overflow.
3017                 SinceEpoch::advance(Duration::from_secs(10 * 64));
3018                 scorer.time_passed(Duration::from_secs(10 * 64));
3019                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 125);
3020
3021                 SinceEpoch::advance(Duration::from_secs(10));
3022                 scorer.time_passed(Duration::from_secs(10 * 65));
3023                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 125);
3024         }
3025
3026         #[test]
3027         fn restricts_liquidity_bounds_after_decay() {
3028                 let logger = TestLogger::new();
3029                 let network_graph = network_graph(&logger);
3030                 let params = ProbabilisticScoringFeeParameters {
3031                         liquidity_penalty_multiplier_msat: 1_000,
3032                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3033                 };
3034                 let decay_params = ProbabilisticScoringDecayParameters {
3035                         liquidity_offset_half_life: Duration::from_secs(10),
3036                         ..ProbabilisticScoringDecayParameters::default()
3037                 };
3038                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
3039                 let source = source_node_id();
3040                 let usage = ChannelUsage {
3041                         amount_msat: 512,
3042                         inflight_htlc_msat: 0,
3043                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_000 },
3044                 };
3045                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3046                 let (info, _) = channel.as_directed_from(&source).unwrap();
3047                 let candidate = CandidateRouteHop::PublicHop {
3048                         info,
3049                         short_channel_id: 42,
3050                 };
3051
3052                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 300);
3053
3054                 // More knowledge gives higher confidence (256, 768), meaning a lower penalty.
3055                 scorer.payment_path_failed(&payment_path_for_amount(768), 42, Duration::ZERO);
3056                 scorer.payment_path_failed(&payment_path_for_amount(256), 43, Duration::ZERO);
3057                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 281);
3058
3059                 // Decaying knowledge gives less confidence (128, 896), meaning a higher penalty.
3060                 SinceEpoch::advance(Duration::from_secs(10));
3061                 scorer.time_passed(Duration::from_secs(10));
3062                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 291);
3063
3064                 // Reducing the upper bound gives more confidence (128, 832) that the payment amount (512)
3065                 // is closer to the upper bound, meaning a higher penalty.
3066                 scorer.payment_path_successful(&payment_path_for_amount(64), Duration::from_secs(10));
3067                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 331);
3068
3069                 // Increasing the lower bound gives more confidence (256, 832) that the payment amount (512)
3070                 // is closer to the lower bound, meaning a lower penalty.
3071                 scorer.payment_path_failed(&payment_path_for_amount(256), 43, Duration::from_secs(10));
3072                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 245);
3073
3074                 // Further decaying affects the lower bound more than the upper bound (128, 928).
3075                 SinceEpoch::advance(Duration::from_secs(10));
3076                 scorer.time_passed(Duration::from_secs(20));
3077                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 280);
3078         }
3079
3080         #[test]
3081         fn restores_persisted_liquidity_bounds() {
3082                 let logger = TestLogger::new();
3083                 let network_graph = network_graph(&logger);
3084                 let params = ProbabilisticScoringFeeParameters {
3085                         liquidity_penalty_multiplier_msat: 1_000,
3086                         considered_impossible_penalty_msat: u64::max_value(),
3087                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3088                 };
3089                 let decay_params = ProbabilisticScoringDecayParameters {
3090                         liquidity_offset_half_life: Duration::from_secs(10),
3091                         ..ProbabilisticScoringDecayParameters::default()
3092                 };
3093                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
3094                 let source = source_node_id();
3095                 let usage = ChannelUsage {
3096                         amount_msat: 500,
3097                         inflight_htlc_msat: 0,
3098                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_000, htlc_maximum_msat: 1_000 },
3099                 };
3100
3101                 scorer.payment_path_failed(&payment_path_for_amount(500), 42, Duration::ZERO);
3102                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3103                 let (info, _) = channel.as_directed_from(&source).unwrap();
3104                 let candidate = CandidateRouteHop::PublicHop {
3105                         info,
3106                         short_channel_id: 42,
3107                 };
3108                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
3109
3110                 SinceEpoch::advance(Duration::from_secs(10));
3111                 scorer.time_passed(Duration::from_secs(10));
3112                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 473);
3113
3114                 scorer.payment_path_failed(&payment_path_for_amount(250), 43, Duration::from_secs(10));
3115                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 300);
3116
3117                 let mut serialized_scorer = Vec::new();
3118                 scorer.write(&mut serialized_scorer).unwrap();
3119
3120                 let mut serialized_scorer = io::Cursor::new(&serialized_scorer);
3121                 let deserialized_scorer =
3122                         <ProbabilisticScorer<_, _>>::read(&mut serialized_scorer, (decay_params, &network_graph, &logger)).unwrap();
3123                 assert_eq!(deserialized_scorer.channel_penalty_msat(&candidate, usage, &params), 300);
3124         }
3125
3126         fn do_decays_persisted_liquidity_bounds(decay_before_reload: bool) {
3127                 let logger = TestLogger::new();
3128                 let network_graph = network_graph(&logger);
3129                 let params = ProbabilisticScoringFeeParameters {
3130                         liquidity_penalty_multiplier_msat: 1_000,
3131                         considered_impossible_penalty_msat: u64::max_value(),
3132                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3133                 };
3134                 let decay_params = ProbabilisticScoringDecayParameters {
3135                         liquidity_offset_half_life: Duration::from_secs(10),
3136                         ..ProbabilisticScoringDecayParameters::zero_penalty()
3137                 };
3138                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
3139                 let source = source_node_id();
3140                 let usage = ChannelUsage {
3141                         amount_msat: 500,
3142                         inflight_htlc_msat: 0,
3143                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_000, htlc_maximum_msat: 1_000 },
3144                 };
3145
3146                 scorer.payment_path_failed(&payment_path_for_amount(500), 42, Duration::ZERO);
3147                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3148                 let (info, _) = channel.as_directed_from(&source).unwrap();
3149                 let candidate = CandidateRouteHop::PublicHop {
3150                         info,
3151                         short_channel_id: 42,
3152                 };
3153                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
3154
3155                 if decay_before_reload {
3156                         SinceEpoch::advance(Duration::from_secs(10));
3157                         scorer.time_passed(Duration::from_secs(10));
3158                 }
3159
3160                 let mut serialized_scorer = Vec::new();
3161                 scorer.write(&mut serialized_scorer).unwrap();
3162
3163                 let mut serialized_scorer = io::Cursor::new(&serialized_scorer);
3164                 let mut deserialized_scorer =
3165                         <ProbabilisticScorer<_, _>>::read(&mut serialized_scorer, (decay_params, &network_graph, &logger)).unwrap();
3166                 if !decay_before_reload {
3167                         SinceEpoch::advance(Duration::from_secs(10));
3168                         scorer.time_passed(Duration::from_secs(10));
3169                         deserialized_scorer.time_passed(Duration::from_secs(10));
3170                 }
3171                 assert_eq!(deserialized_scorer.channel_penalty_msat(&candidate, usage, &params), 473);
3172
3173                 scorer.payment_path_failed(&payment_path_for_amount(250), 43, Duration::from_secs(10));
3174                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 300);
3175
3176                 SinceEpoch::advance(Duration::from_secs(10));
3177                 deserialized_scorer.time_passed(Duration::from_secs(20));
3178                 assert_eq!(deserialized_scorer.channel_penalty_msat(&candidate, usage, &params), 370);
3179         }
3180
3181         #[test]
3182         fn decays_persisted_liquidity_bounds() {
3183                 do_decays_persisted_liquidity_bounds(false);
3184                 do_decays_persisted_liquidity_bounds(true);
3185         }
3186
3187         #[test]
3188         fn scores_realistic_payments() {
3189                 // Shows the scores of "realistic" sends of 100k sats over channels of 1-10m sats (with a
3190                 // 50k sat reserve).
3191                 let logger = TestLogger::new();
3192                 let network_graph = network_graph(&logger);
3193                 let params = ProbabilisticScoringFeeParameters::default();
3194                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3195                 let source = source_node_id();
3196
3197                 let usage = ChannelUsage {
3198                         amount_msat: 100_000_000,
3199                         inflight_htlc_msat: 0,
3200                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 950_000_000, htlc_maximum_msat: 1_000 },
3201                 };
3202                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3203                 let (info, _) = channel.as_directed_from(&source).unwrap();
3204                 let candidate = CandidateRouteHop::PublicHop {
3205                         info,
3206                         short_channel_id: 42,
3207                 };
3208                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 11497);
3209                 let usage = ChannelUsage {
3210                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3211                 };
3212                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 7408);
3213                 let usage = ChannelUsage {
3214                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 2_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3215                 };
3216                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 6151);
3217                 let usage = ChannelUsage {
3218                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 3_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3219                 };
3220                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 5427);
3221                 let usage = ChannelUsage {
3222                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 4_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3223                 };
3224                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 4955);
3225                 let usage = ChannelUsage {
3226                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 5_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3227                 };
3228                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 4736);
3229                 let usage = ChannelUsage {
3230                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 6_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3231                 };
3232                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 4484);
3233                 let usage = ChannelUsage {
3234                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 7_450_000_000, htlc_maximum_msat: 1_000 }, ..usage
3235                 };
3236                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 4484);
3237                 let usage = ChannelUsage {
3238                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 7_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3239                 };
3240                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 4263);
3241                 let usage = ChannelUsage {
3242                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 8_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3243                 };
3244                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 4263);
3245                 let usage = ChannelUsage {
3246                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 9_950_000_000, htlc_maximum_msat: 1_000 }, ..usage
3247                 };
3248                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 4044);
3249         }
3250
3251         #[test]
3252         fn adds_base_penalty_to_liquidity_penalty() {
3253                 let logger = TestLogger::new();
3254                 let network_graph = network_graph(&logger);
3255                 let source = source_node_id();
3256                 let usage = ChannelUsage {
3257                         amount_msat: 128,
3258                         inflight_htlc_msat: 0,
3259                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_000 },
3260                 };
3261
3262                 let params = ProbabilisticScoringFeeParameters {
3263                         liquidity_penalty_multiplier_msat: 1_000,
3264                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3265                 };
3266                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3267                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3268                 let (info, _) = channel.as_directed_from(&source).unwrap();
3269                 let candidate = CandidateRouteHop::PublicHop {
3270                         info,
3271                         short_channel_id: 42,
3272                 };
3273                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 58);
3274
3275                 let params = ProbabilisticScoringFeeParameters {
3276                         base_penalty_msat: 500, liquidity_penalty_multiplier_msat: 1_000,
3277                         anti_probing_penalty_msat: 0, ..ProbabilisticScoringFeeParameters::zero_penalty()
3278                 };
3279                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3280                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 558);
3281
3282                 let params = ProbabilisticScoringFeeParameters {
3283                         base_penalty_msat: 500, liquidity_penalty_multiplier_msat: 1_000,
3284                         base_penalty_amount_multiplier_msat: (1 << 30),
3285                         anti_probing_penalty_msat: 0, ..ProbabilisticScoringFeeParameters::zero_penalty()
3286                 };
3287
3288                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3289                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 558 + 128);
3290         }
3291
3292         #[test]
3293         fn adds_amount_penalty_to_liquidity_penalty() {
3294                 let logger = TestLogger::new();
3295                 let network_graph = network_graph(&logger);
3296                 let source = source_node_id();
3297                 let usage = ChannelUsage {
3298                         amount_msat: 512_000,
3299                         inflight_htlc_msat: 0,
3300                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: 1_000 },
3301                 };
3302
3303                 let params = ProbabilisticScoringFeeParameters {
3304                         liquidity_penalty_multiplier_msat: 1_000,
3305                         liquidity_penalty_amount_multiplier_msat: 0,
3306                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3307                 };
3308                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3309                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3310                 let (info, _) = channel.as_directed_from(&source).unwrap();
3311                 let candidate = CandidateRouteHop::PublicHop {
3312                         info,
3313                         short_channel_id: 42,
3314                 };
3315                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 300);
3316
3317                 let params = ProbabilisticScoringFeeParameters {
3318                         liquidity_penalty_multiplier_msat: 1_000,
3319                         liquidity_penalty_amount_multiplier_msat: 256,
3320                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3321                 };
3322                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3323                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 337);
3324         }
3325
3326         #[test]
3327         fn calculates_log10_without_overflowing_u64_max_value() {
3328                 let logger = TestLogger::new();
3329                 let network_graph = network_graph(&logger);
3330                 let source = source_node_id();
3331                 let usage = ChannelUsage {
3332                         amount_msat: u64::max_value(),
3333                         inflight_htlc_msat: 0,
3334                         effective_capacity: EffectiveCapacity::Infinite,
3335                 };
3336                 let params = ProbabilisticScoringFeeParameters {
3337                         liquidity_penalty_multiplier_msat: 40_000,
3338                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3339                 };
3340                 let decay_params = ProbabilisticScoringDecayParameters::zero_penalty();
3341                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3342                 let (info, _) = channel.as_directed_from(&source).unwrap();
3343                 let candidate = CandidateRouteHop::PublicHop {
3344                         info,
3345                         short_channel_id: 42,
3346                 };
3347                 let scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
3348                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 80_000);
3349         }
3350
3351         #[test]
3352         fn accounts_for_inflight_htlc_usage() {
3353                 let logger = TestLogger::new();
3354                 let network_graph = network_graph(&logger);
3355                 let params = ProbabilisticScoringFeeParameters {
3356                         considered_impossible_penalty_msat: u64::max_value(),
3357                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3358                 };
3359                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3360                 let source = source_node_id();
3361
3362                 let usage = ChannelUsage {
3363                         amount_msat: 750,
3364                         inflight_htlc_msat: 0,
3365                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_000, htlc_maximum_msat: 1_000 },
3366                 };
3367                 let network_graph = network_graph.read_only();
3368                 let channel = network_graph.channel(42).unwrap();
3369                 let (info, _) = channel.as_directed_from(&source).unwrap();
3370                 let candidate = CandidateRouteHop::PublicHop {
3371                         info,
3372                         short_channel_id: 42,
3373                 };
3374                 assert_ne!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
3375
3376                 let usage = ChannelUsage { inflight_htlc_msat: 251, ..usage };
3377                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
3378         }
3379
3380         #[test]
3381         fn removes_uncertainity_when_exact_liquidity_known() {
3382                 let logger = TestLogger::new();
3383                 let network_graph = network_graph(&logger);
3384                 let params = ProbabilisticScoringFeeParameters::default();
3385                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3386                 let source = source_node_id();
3387
3388                 let base_penalty_msat = params.base_penalty_msat;
3389                 let usage = ChannelUsage {
3390                         amount_msat: 750,
3391                         inflight_htlc_msat: 0,
3392                         effective_capacity: EffectiveCapacity::ExactLiquidity { liquidity_msat: 1_000 },
3393                 };
3394                 let network_graph = network_graph.read_only();
3395                 let channel = network_graph.channel(42).unwrap();
3396                 let (info, _) = channel.as_directed_from(&source).unwrap();
3397                 let candidate = CandidateRouteHop::PublicHop {
3398                         info,
3399                         short_channel_id: 42,
3400                 };
3401                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), base_penalty_msat);
3402
3403                 let usage = ChannelUsage { amount_msat: 1_000, ..usage };
3404                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), base_penalty_msat);
3405
3406                 let usage = ChannelUsage { amount_msat: 1_001, ..usage };
3407                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), u64::max_value());
3408         }
3409
3410         #[test]
3411         fn remembers_historical_failures() {
3412                 let logger = TestLogger::new();
3413                 let network_graph = network_graph(&logger);
3414                 let params = ProbabilisticScoringFeeParameters {
3415                         historical_liquidity_penalty_multiplier_msat: 1024,
3416                         historical_liquidity_penalty_amount_multiplier_msat: 1024,
3417                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3418                 };
3419                 let decay_params = ProbabilisticScoringDecayParameters {
3420                         liquidity_offset_half_life: Duration::from_secs(60 * 60),
3421                         historical_no_updates_half_life: Duration::from_secs(10),
3422                 };
3423                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
3424                 let source = source_node_id();
3425                 let target = target_node_id();
3426
3427                 let usage = ChannelUsage {
3428                         amount_msat: 100,
3429                         inflight_htlc_msat: 0,
3430                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_024 },
3431                 };
3432                 let usage_1 = ChannelUsage {
3433                         amount_msat: 1,
3434                         inflight_htlc_msat: 0,
3435                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_024 },
3436                 };
3437
3438                 {
3439                         let network_graph = network_graph.read_only();
3440                         let channel = network_graph.channel(42).unwrap();
3441                         let (info, _) = channel.as_directed_from(&source).unwrap();
3442                         let candidate = CandidateRouteHop::PublicHop {
3443                                 info,
3444                                 short_channel_id: 42,
3445                         };
3446
3447                         // With no historical data the normal liquidity penalty calculation is used.
3448                         assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 168);
3449                 }
3450                 assert_eq!(scorer.historical_estimated_channel_liquidity_probabilities(42, &target),
3451                 None);
3452                 assert_eq!(scorer.historical_estimated_payment_success_probability(42, &target, 42, &params),
3453                 None);
3454
3455                 scorer.payment_path_failed(&payment_path_for_amount(1), 42, Duration::ZERO);
3456                 {
3457                         let network_graph = network_graph.read_only();
3458                         let channel = network_graph.channel(42).unwrap();
3459                         let (info, _) = channel.as_directed_from(&source).unwrap();
3460                         let candidate = CandidateRouteHop::PublicHop {
3461                                 info,
3462                                 short_channel_id: 42,
3463                         };
3464
3465                         assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 2048);
3466                         assert_eq!(scorer.channel_penalty_msat(&candidate, usage_1, &params), 249);
3467                 }
3468                 // The "it failed" increment is 32, where the probability should lie several buckets into
3469                 // the first octile.
3470                 assert_eq!(scorer.historical_estimated_channel_liquidity_probabilities(42, &target),
3471                         Some(([32, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
3472                                 [0, 0, 0, 0, 0, 0, 32, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0])));
3473                 assert!(scorer.historical_estimated_payment_success_probability(42, &target, 1, &params)
3474                         .unwrap() > 0.35);
3475                 assert_eq!(scorer.historical_estimated_payment_success_probability(42, &target, 500, &params),
3476                         Some(0.0));
3477
3478                 // Even after we tell the scorer we definitely have enough available liquidity, it will
3479                 // still remember that there was some failure in the past, and assign a non-0 penalty.
3480                 scorer.payment_path_failed(&payment_path_for_amount(1000), 43, Duration::ZERO);
3481                 {
3482                         let network_graph = network_graph.read_only();
3483                         let channel = network_graph.channel(42).unwrap();
3484                         let (info, _) = channel.as_directed_from(&source).unwrap();
3485                         let candidate = CandidateRouteHop::PublicHop {
3486                                 info,
3487                                 short_channel_id: 42,
3488                         };
3489
3490                         assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 105);
3491                 }
3492                 // The first points should be decayed just slightly and the last bucket has a new point.
3493                 assert_eq!(scorer.historical_estimated_channel_liquidity_probabilities(42, &target),
3494                         Some(([31, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 32, 0, 0, 0, 0, 0],
3495                                 [0, 0, 0, 0, 0, 0, 31, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 32])));
3496
3497                 // The exact success probability is a bit complicated and involves integer rounding, so we
3498                 // simply check bounds here.
3499                 let five_hundred_prob =
3500                         scorer.historical_estimated_payment_success_probability(42, &target, 500, &params).unwrap();
3501                 assert!(five_hundred_prob > 0.59);
3502                 assert!(five_hundred_prob < 0.60);
3503                 let one_prob =
3504                         scorer.historical_estimated_payment_success_probability(42, &target, 1, &params).unwrap();
3505                 assert!(one_prob < 0.85);
3506                 assert!(one_prob > 0.84);
3507
3508                 // Advance the time forward 16 half-lives (which the docs claim will ensure all data is
3509                 // gone), and check that we're back to where we started.
3510                 SinceEpoch::advance(Duration::from_secs(10 * 16));
3511                 scorer.time_passed(Duration::from_secs(10 * 16));
3512                 {
3513                         let network_graph = network_graph.read_only();
3514                         let channel = network_graph.channel(42).unwrap();
3515                         let (info, _) = channel.as_directed_from(&source).unwrap();
3516                         let candidate = CandidateRouteHop::PublicHop {
3517                                 info,
3518                                 short_channel_id: 42,
3519                         };
3520
3521                         assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 168);
3522                 }
3523                 // Once fully decayed we still have data, but its all-0s. In the future we may remove the
3524                 // data entirely instead.
3525                 assert_eq!(scorer.historical_estimated_channel_liquidity_probabilities(42, &target),
3526                         Some(([0; 32], [0; 32])));
3527                 assert_eq!(scorer.historical_estimated_payment_success_probability(42, &target, 1, &params), None);
3528
3529                 let mut usage = ChannelUsage {
3530                         amount_msat: 100,
3531                         inflight_htlc_msat: 1024,
3532                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_024 },
3533                 };
3534                 scorer.payment_path_failed(&payment_path_for_amount(1), 42, Duration::from_secs(10 * 16));
3535                 {
3536                         let network_graph = network_graph.read_only();
3537                         let channel = network_graph.channel(42).unwrap();
3538                         let (info, _) = channel.as_directed_from(&source).unwrap();
3539                         let candidate = CandidateRouteHop::PublicHop {
3540                                 info,
3541                                 short_channel_id: 42,
3542                         };
3543
3544                         assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 2050);
3545
3546                         let usage = ChannelUsage {
3547                                 amount_msat: 1,
3548                                 inflight_htlc_msat: 0,
3549                                 effective_capacity: EffectiveCapacity::AdvertisedMaxHTLC { amount_msat: 0 },
3550                         };
3551                         assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 2048);
3552                 }
3553
3554                 // Advance to decay all liquidity offsets to zero.
3555                 SinceEpoch::advance(Duration::from_secs(60 * 60 * 10));
3556                 scorer.time_passed(Duration::from_secs(10 * (16 + 60 * 60)));
3557
3558                 // Once even the bounds have decayed information about the channel should be removed
3559                 // entirely.
3560                 assert_eq!(scorer.historical_estimated_channel_liquidity_probabilities(42, &target),
3561                         None);
3562
3563                 // Use a path in the opposite direction, which have zero for htlc_maximum_msat. This will
3564                 // ensure that the effective capacity is zero to test division-by-zero edge cases.
3565                 let path = vec![
3566                         path_hop(target_pubkey(), 43, 2),
3567                         path_hop(source_pubkey(), 42, 1),
3568                         path_hop(sender_pubkey(), 41, 0),
3569                 ];
3570                 scorer.payment_path_failed(&Path { hops: path, blinded_tail: None }, 42, Duration::from_secs(10 * (16 + 60 * 60)));
3571         }
3572
3573         #[test]
3574         fn adds_anti_probing_penalty() {
3575                 let logger = TestLogger::new();
3576                 let network_graph = network_graph(&logger);
3577                 let source = source_node_id();
3578                 let params = ProbabilisticScoringFeeParameters {
3579                         anti_probing_penalty_msat: 500,
3580                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3581                 };
3582                 let scorer = ProbabilisticScorer::new(ProbabilisticScoringDecayParameters::default(), &network_graph, &logger);
3583
3584                 // Check we receive no penalty for a low htlc_maximum_msat.
3585                 let usage = ChannelUsage {
3586                         amount_msat: 512_000,
3587                         inflight_htlc_msat: 0,
3588                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: 1_000 },
3589                 };
3590                 let network_graph = network_graph.read_only();
3591                 let channel = network_graph.channel(42).unwrap();
3592                 let (info, _) = channel.as_directed_from(&source).unwrap();
3593                 let candidate = CandidateRouteHop::PublicHop {
3594                         info,
3595                         short_channel_id: 42,
3596                 };
3597                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
3598
3599                 // Check we receive anti-probing penalty for htlc_maximum_msat == channel_capacity.
3600                 let usage = ChannelUsage {
3601                         amount_msat: 512_000,
3602                         inflight_htlc_msat: 0,
3603                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: 1_024_000 },
3604                 };
3605                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 500);
3606
3607                 // Check we receive anti-probing penalty for htlc_maximum_msat == channel_capacity/2.
3608                 let usage = ChannelUsage {
3609                         amount_msat: 512_000,
3610                         inflight_htlc_msat: 0,
3611                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: 512_000 },
3612                 };
3613                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 500);
3614
3615                 // Check we receive no anti-probing penalty for htlc_maximum_msat == channel_capacity/2 - 1.
3616                 let usage = ChannelUsage {
3617                         amount_msat: 512_000,
3618                         inflight_htlc_msat: 0,
3619                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024_000, htlc_maximum_msat: 511_999 },
3620                 };
3621                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 0);
3622         }
3623
3624         #[test]
3625         fn scores_with_blinded_path() {
3626                 // Make sure we'll account for a blinded path's final_value_msat in scoring
3627                 let logger = TestLogger::new();
3628                 let network_graph = network_graph(&logger);
3629                 let params = ProbabilisticScoringFeeParameters {
3630                         liquidity_penalty_multiplier_msat: 1_000,
3631                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3632                 };
3633                 let decay_params = ProbabilisticScoringDecayParameters::default();
3634                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
3635                 let source = source_node_id();
3636                 let usage = ChannelUsage {
3637                         amount_msat: 512,
3638                         inflight_htlc_msat: 0,
3639                         effective_capacity: EffectiveCapacity::Total { capacity_msat: 1_024, htlc_maximum_msat: 1_000 },
3640                 };
3641                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3642                 let (info, target) = channel.as_directed_from(&source).unwrap();
3643                 let candidate = CandidateRouteHop::PublicHop {
3644                         info,
3645                         short_channel_id: 42,
3646                 };
3647                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 300);
3648
3649                 let mut path = payment_path_for_amount(768);
3650                 let recipient_hop = path.hops.pop().unwrap();
3651                 let blinded_path = BlindedPath {
3652                         introduction_node_id: path.hops.last().as_ref().unwrap().pubkey,
3653                         blinding_point: test_utils::pubkey(42),
3654                         blinded_hops: vec![
3655                                 BlindedHop { blinded_node_id: test_utils::pubkey(44), encrypted_payload: Vec::new() }
3656                         ],
3657                 };
3658                 path.blinded_tail = Some(BlindedTail {
3659                         hops: blinded_path.blinded_hops,
3660                         blinding_point: blinded_path.blinding_point,
3661                         excess_final_cltv_expiry_delta: recipient_hop.cltv_expiry_delta,
3662                         final_value_msat: recipient_hop.fee_msat,
3663                 });
3664
3665                 // Check the liquidity before and after scoring payment failures to ensure the blinded path's
3666                 // final value is taken into account.
3667                 assert!(scorer.channel_liquidities.get(&42).is_none());
3668
3669                 scorer.payment_path_failed(&path, 42, Duration::ZERO);
3670                 path.blinded_tail.as_mut().unwrap().final_value_msat = 256;
3671                 scorer.payment_path_failed(&path, 43, Duration::ZERO);
3672
3673                 let liquidity = scorer.channel_liquidities.get(&42).unwrap()
3674                         .as_directed(&source, &target, 1_000, decay_params);
3675                 assert_eq!(liquidity.min_liquidity_msat(), 256);
3676                 assert_eq!(liquidity.max_liquidity_msat(), 768);
3677         }
3678
3679         #[test]
3680         fn realistic_historical_failures() {
3681                 // The motivation for the unequal sized buckets came largely from attempting to pay 10k
3682                 // sats over a one bitcoin channel. This tests that case explicitly, ensuring that we score
3683                 // properly.
3684                 let logger = TestLogger::new();
3685                 let mut network_graph = network_graph(&logger);
3686                 let params = ProbabilisticScoringFeeParameters {
3687                         historical_liquidity_penalty_multiplier_msat: 1024,
3688                         historical_liquidity_penalty_amount_multiplier_msat: 1024,
3689                         ..ProbabilisticScoringFeeParameters::zero_penalty()
3690                 };
3691                 let decay_params = ProbabilisticScoringDecayParameters {
3692                         liquidity_offset_half_life: Duration::from_secs(60 * 60),
3693                         historical_no_updates_half_life: Duration::from_secs(10),
3694                         ..ProbabilisticScoringDecayParameters::default()
3695                 };
3696
3697                 let capacity_msat = 100_000_000_000;
3698                 update_channel(&mut network_graph, 42, source_privkey(), 0, capacity_msat, 200);
3699                 update_channel(&mut network_graph, 42, target_privkey(), 1, capacity_msat, 200);
3700
3701                 let mut scorer = ProbabilisticScorer::new(decay_params, &network_graph, &logger);
3702                 let source = source_node_id();
3703
3704                 let mut amount_msat = 10_000_000;
3705                 let usage = ChannelUsage {
3706                         amount_msat,
3707                         inflight_htlc_msat: 0,
3708                         effective_capacity: EffectiveCapacity::Total { capacity_msat, htlc_maximum_msat: capacity_msat },
3709                 };
3710                 let channel = network_graph.read_only().channel(42).unwrap().to_owned();
3711                 let (info, target) = channel.as_directed_from(&source).unwrap();
3712                 let candidate = CandidateRouteHop::PublicHop {
3713                         info,
3714                         short_channel_id: 42,
3715                 };
3716                 // With no historical data the normal liquidity penalty calculation is used, which results
3717                 // in a success probability of ~75%.
3718                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params), 1269);
3719                 assert_eq!(scorer.historical_estimated_channel_liquidity_probabilities(42, &target),
3720                         None);
3721                 assert_eq!(scorer.historical_estimated_payment_success_probability(42, &target, 42, &params),
3722                         None);
3723
3724                 // Fail to pay once, and then check the buckets and penalty.
3725                 scorer.payment_path_failed(&payment_path_for_amount(amount_msat), 42, Duration::ZERO);
3726                 // The penalty should be the maximum penalty, as the payment we're scoring is now in the
3727                 // same bucket which is the only maximum datapoint.
3728                 assert_eq!(scorer.channel_penalty_msat(&candidate, usage, &params),
3729                         2048 + 2048 * amount_msat / super::AMOUNT_PENALTY_DIVISOR);
3730                 // The "it failed" increment is 32, which we should apply to the first upper-bound (between
3731                 // 6k sats and 12k sats).
3732                 assert_eq!(scorer.historical_estimated_channel_liquidity_probabilities(42, &target),
3733                         Some(([32, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
3734                                 [0, 32, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0])));
3735                 // The success probability estimate itself should be zero.
3736                 assert_eq!(scorer.historical_estimated_payment_success_probability(42, &target, amount_msat, &params),
3737                         Some(0.0));
3738
3739                 // Now test again with the amount in the bottom bucket.
3740                 amount_msat /= 2;
3741                 // The new amount is entirely within the only minimum bucket with score, so the probability
3742                 // we assign is 1/2.
3743                 assert_eq!(scorer.historical_estimated_payment_success_probability(42, &target, amount_msat, &params),
3744                         Some(0.5));
3745
3746                 // ...but once we see a failure, we consider the payment to be substantially less likely,
3747                 // even though not a probability of zero as we still look at the second max bucket which
3748                 // now shows 31.
3749                 scorer.payment_path_failed(&payment_path_for_amount(amount_msat), 42, Duration::ZERO);
3750                 assert_eq!(scorer.historical_estimated_channel_liquidity_probabilities(42, &target),
3751                         Some(([63, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
3752                                 [32, 31, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0])));
3753                 assert_eq!(scorer.historical_estimated_payment_success_probability(42, &target, amount_msat, &params),
3754                         Some(0.0));
3755         }
3756 }