Skip to content

Commit 422d73d

Browse files
henghongleehenghonglee
committed
Added Changes
1 parent e94647c commit 422d73d

File tree

11 files changed

+634
-588
lines changed

11 files changed

+634
-588
lines changed

‎fuzz/src/router.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,7 @@ pub fn do_test<Out: test_logger::Output>(data: &[u8], out: Out) {
293293
}]));
294294
}
295295
}
296-
let scorer = ProbabilisticScorer::new(Default::default(), &net_graph, &logger);
296+
let scorer = ProbabilisticScorer::new(&net_graph, &logger);
297297
let random_seed_bytes: [u8; 32] = [get_slice!(1)[0]; 32];
298298
for target in node_pks.iter() {
299299
let final_value_msat = slice_to_be64(get_slice!(8));

‎lightning-background-processor/src/lib.rs

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -866,7 +866,7 @@ mod tests {
866866
fn disconnect_socket(&mut self) {}
867867
}
868868

869-
type ChannelManager = channelmanager::ChannelManager<Arc<ChainMonitor>, Arc<test_utils::TestBroadcaster>, Arc<KeysManager>, Arc<KeysManager>, Arc<KeysManager>, Arc<test_utils::TestFeeEstimator>, Arc<DefaultRouter< Arc<NetworkGraph<Arc<test_utils::TestLogger>>>, Arc<test_utils::TestLogger>, Arc<Mutex<TestScorer>>>>, Arc<test_utils::TestLogger>>;
869+
type ChannelManager = channelmanager::ChannelManager<Arc<ChainMonitor>, Arc<test_utils::TestBroadcaster>, Arc<KeysManager>, Arc<KeysManager>, Arc<KeysManager>, Arc<test_utils::TestFeeEstimator>, Arc<DefaultRouter< Arc<NetworkGraph<Arc<test_utils::TestLogger>>>, Arc<test_utils::TestLogger>, Arc<Mutex<TestScorer>>, i32, TestScorer>>, Arc<test_utils::TestLogger>>;
870870

871871
type ChainMonitor = chainmonitor::ChainMonitor<InMemorySigner, Arc<test_utils::TestChainSource>, Arc<test_utils::TestBroadcaster>, Arc<test_utils::TestFeeEstimator>, Arc<test_utils::TestLogger>, Arc<FilesystemPersister>>;
872872

@@ -1000,8 +1000,9 @@ mod tests {
10001000
}
10011001

10021002
impl Score for TestScorer {
1003+
type ScoreParams = i32;
10031004
fn channel_penalty_msat(
1004-
&self, _short_channel_id: u64, _source: &NodeId, _target: &NodeId, _usage: ChannelUsage
1005+
&self, _short_channel_id: u64, _source: &NodeId, _target: &NodeId, _usage: ChannelUsage, _score_params: &Self::ScoreParams
10051006
) -> u64 { unimplemented!(); }
10061007

10071008
fn payment_path_failed(&mut self, actual_path: &Path, actual_short_channel_id: u64) {
@@ -1114,7 +1115,7 @@ mod tests {
11141115
let network_graph = Arc::new(NetworkGraph::new(network, logger.clone()));
11151116
let scorer = Arc::new(Mutex::new(TestScorer::new()));
11161117
let seed = [i as u8; 32];
1117-
let router = Arc::new(DefaultRouter::new(network_graph.clone(), logger.clone(), seed, scorer.clone()));
1118+
let router = Arc::new(DefaultRouter::new(network_graph.clone(), logger.clone(), seed, scorer.clone(), &0));
11181119
let chain_source = Arc::new(test_utils::TestChainSource::new(Network::Testnet));
11191120
let persister = Arc::new(FilesystemPersister::new(format!("{}_persister_{}", &persist_dir, i)));
11201121
let now = Duration::from_secs(genesis_block.header.time as u64);

‎lightning/src/ln/channelmanager.rs

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ use crate::ln::features::{ChannelFeatures, ChannelTypeFeatures, InitFeatures, No
4646
use crate::ln::features::InvoiceFeatures;
4747
use crate::routing::gossip::NetworkGraph;
4848
use crate::routing::router::{BlindedTail, DefaultRouter, InFlightHtlcs, Path, PaymentParameters, Route, RouteHop, RouteParameters, Router};
49-
use crate::routing::scoring::ProbabilisticScorer;
49+
use crate::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringFeeParameters};
5050
use crate::ln::msgs;
5151
use crate::ln::onion_utils;
5252
use crate::ln::onion_utils::HTLCFailReason;
@@ -623,7 +623,9 @@ pub type SimpleArcChannelManager<M, T, F, L> = ChannelManager<
623623
Arc<DefaultRouter<
624624
Arc<NetworkGraph<Arc<L>>>,
625625
Arc<L>,
626-
Arc<Mutex<ProbabilisticScorer<Arc<NetworkGraph<Arc<L>>>, Arc<L>>>>
626+
Arc<Mutex<ProbabilisticScorer<Arc<NetworkGraph<Arc<L>>>, Arc<L>>>>,
627+
ProbabilisticScoringFeeParameters,
628+
ProbabilisticScorer<Arc<NetworkGraph<Arc<L>>>, Arc<L>>,
627629
>>,
628630
Arc<L>
629631
>;
@@ -639,7 +641,7 @@ pub type SimpleArcChannelManager<M, T, F, L> = ChannelManager<
639641
/// of [`KeysManager`] and [`DefaultRouter`].
640642
///
641643
/// This is not exported to bindings users as Arcs don't make sense in bindings
642-
pub type SimpleRefChannelManager<'a, 'b, 'c, 'd, 'e, 'f, 'g, 'h, M, T, F, L> = ChannelManager<&'a M, &'b T, &'c KeysManager, &'c KeysManager, &'c KeysManager, &'d F, &'e DefaultRouter<&'f NetworkGraph<&'g L>, &'g L, &'h Mutex<ProbabilisticScorer<&'f NetworkGraph<&'g L>, &'g L>>>, &'g L>;
644+
pub type SimpleRefChannelManager<'a, 'b, 'c, 'd, 'e, 'f, 'g, 'h, M, T, F, L> = ChannelManager<&'a M, &'b T, &'c KeysManager, &'c KeysManager, &'c KeysManager, &'d F, &'e DefaultRouter<&'f NetworkGraph<&'g L>, &'g L, &'h Mutex<ProbabilisticScorer<&'f NetworkGraph<&'g L>, &'g L>>, ProbabilisticScoringFeeParameters, ProbabilisticScorer<&'f NetworkGraph<&'g L>, &'g L>>, &'g L>;
643645

644646
/// A trivial trait which describes any [`ChannelManager`] used in testing.
645647
#[cfg(any(test, feature = "_test_utils"))]
@@ -8501,7 +8503,7 @@ mod tests {
85018503
};
85028504
let route = find_route(
85038505
&nodes[0].node.get_our_node_id(), &route_params, &nodes[0].network_graph,
8504-
None, nodes[0].logger, &scorer, &random_seed_bytes
8506+
None, nodes[0].logger, &scorer, &0, &random_seed_bytes
85058507
).unwrap();
85068508
nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
85078509
RecipientOnionFields::spontaneous_empty(), PaymentId(payment_preimage.0)).unwrap();
@@ -8535,7 +8537,7 @@ mod tests {
85358537
let payment_preimage = PaymentPreimage([42; 32]);
85368538
let route = find_route(
85378539
&nodes[0].node.get_our_node_id(), &route_params, &nodes[0].network_graph,
8538-
None, nodes[0].logger, &scorer, &random_seed_bytes
8540+
None, nodes[0].logger, &scorer,&0, &random_seed_bytes
85398541
).unwrap();
85408542
let payment_hash = nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
85418543
RecipientOnionFields::spontaneous_empty(), PaymentId(payment_preimage.0)).unwrap();
@@ -8598,7 +8600,7 @@ mod tests {
85988600
let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
85998601
let route = find_route(
86008602
&payer_pubkey, &route_params, &network_graph, Some(&first_hops.iter().collect::<Vec<_>>()),
8601-
nodes[0].logger, &scorer, &random_seed_bytes
8603+
nodes[0].logger, &scorer,&0, &random_seed_bytes
86028604
).unwrap();
86038605

86048606
let test_preimage = PaymentPreimage([42; 32]);
@@ -8642,7 +8644,7 @@ mod tests {
86428644
let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
86438645
let route = find_route(
86448646
&payer_pubkey, &route_params, &network_graph, Some(&first_hops.iter().collect::<Vec<_>>()),
8645-
nodes[0].logger, &scorer, &random_seed_bytes
8647+
nodes[0].logger, &scorer, &0, &random_seed_bytes
86468648
).unwrap();
86478649

86488650
let test_preimage = PaymentPreimage([42; 32]);

‎lightning/src/ln/functional_test_utils.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1705,7 +1705,7 @@ pub fn get_route(send_node: &Node, payment_params: &PaymentParameters, recv_valu
17051705
router::get_route(
17061706
&send_node.node.get_our_node_id(), payment_params, &send_node.network_graph.read_only(),
17071707
Some(&send_node.node.list_usable_channels().iter().collect::<Vec<_>>()),
1708-
recv_value, send_node.logger, &scorer, &random_seed_bytes
1708+
recv_value, send_node.logger, &scorer, &0, &random_seed_bytes
17091709
)
17101710
}
17111711

@@ -2295,7 +2295,7 @@ pub fn route_over_limit<'a, 'b, 'c>(origin_node: &Node<'a, 'b, 'c>, expected_rou
22952295
let random_seed_bytes = keys_manager.get_secure_random_bytes();
22962296
let route = router::get_route(
22972297
&origin_node.node.get_our_node_id(), &payment_params, &network_graph,
2298-
None, recv_value, origin_node.logger, &scorer, &random_seed_bytes).unwrap();
2298+
None, recv_value, origin_node.logger, &scorer, &0, &random_seed_bytes).unwrap();
22992299
assert_eq!(route.paths.len(), 1);
23002300
assert_eq!(route.paths[0].hops.len(), expected_route.len());
23012301
for (node, hop) in expected_route.iter().zip(route.paths[0].hops.iter()) {

‎lightning/src/ln/functional_tests.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -7044,7 +7044,7 @@ fn test_check_htlc_underpaying() {
70447044
let scorer = test_utils::TestScorer::new();
70457045
let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
70467046
let payment_params = PaymentParameters::from_node_id(nodes[1].node.get_our_node_id(), TEST_FINAL_CLTV).with_features(nodes[1].node.invoice_features());
7047-
let route = get_route(&nodes[0].node.get_our_node_id(), &payment_params, &nodes[0].network_graph.read_only(), None, 10_000, nodes[0].logger, &scorer, &random_seed_bytes).unwrap();
7047+
let route = get_route(&nodes[0].node.get_our_node_id(), &payment_params, &nodes[0].network_graph.read_only(), None, 10_000, nodes[0].logger, &scorer, &0, &random_seed_bytes).unwrap();
70487048
let (_, our_payment_hash, _) = get_payment_preimage_hash!(nodes[0]);
70497049
let our_payment_secret = nodes[1].node.create_inbound_payment_for_hash(our_payment_hash, Some(100_000), 7200, None).unwrap();
70507050
nodes[0].node.send_payment_with_route(&route, our_payment_hash,
@@ -7298,11 +7298,11 @@ fn test_bump_penalty_txn_on_revoked_htlcs() {
72987298
let scorer = test_utils::TestScorer::new();
72997299
let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
73007300
let route = get_route(&nodes[0].node.get_our_node_id(), &payment_params, &nodes[0].network_graph.read_only(), None,
7301-
3_000_000, nodes[0].logger, &scorer, &random_seed_bytes).unwrap();
7301+
3_000_000, nodes[0].logger, &scorer, &0, &random_seed_bytes).unwrap();
73027302
let payment_preimage = send_along_route(&nodes[0], route, &[&nodes[1]], 3_000_000).0;
73037303
let payment_params = PaymentParameters::from_node_id(nodes[0].node.get_our_node_id(), 50).with_features(nodes[0].node.invoice_features());
73047304
let route = get_route(&nodes[1].node.get_our_node_id(), &payment_params, &nodes[1].network_graph.read_only(), None,
7305-
3_000_000, nodes[0].logger, &scorer, &random_seed_bytes).unwrap();
7305+
3_000_000, nodes[0].logger, &scorer, &0, &random_seed_bytes).unwrap();
73067306
send_along_route(&nodes[1], route, &[&nodes[0]], 3_000_000);
73077307

73087308
let revoked_local_txn = get_local_commitment_txn!(nodes[1], chan.2);
@@ -9520,7 +9520,7 @@ fn test_keysend_payments_to_public_node() {
95209520
};
95219521
let scorer = test_utils::TestScorer::new();
95229522
let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
9523-
let route = find_route(&payer_pubkey, &route_params, &network_graph, None, nodes[0].logger, &scorer, &random_seed_bytes).unwrap();
9523+
let route = find_route(&payer_pubkey, &route_params, &network_graph, None, nodes[0].logger, &scorer, &0, &random_seed_bytes).unwrap();
95249524

95259525
let test_preimage = PaymentPreimage([42; 32]);
95269526
let payment_hash = nodes[0].node.send_spontaneous_payment(&route, Some(test_preimage),
@@ -9555,7 +9555,7 @@ fn test_keysend_payments_to_private_node() {
95559555
let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
95569556
let route = find_route(
95579557
&payer_pubkey, &route_params, &network_graph, Some(&first_hops.iter().collect::<Vec<_>>()),
9558-
nodes[0].logger, &scorer, &random_seed_bytes
9558+
nodes[0].logger, &scorer, &0, &random_seed_bytes
95599559
).unwrap();
95609560

95619561
let test_preimage = PaymentPreimage([42; 32]);

‎lightning/src/ln/onion_route_tests.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -993,7 +993,7 @@ macro_rules! get_phantom_route {
993993
(get_route(
994994
&$nodes[0].node.get_our_node_id(), &payment_params, &network_graph,
995995
Some(&$nodes[0].node.list_usable_channels().iter().collect::<Vec<_>>()),
996-
$amt, $nodes[0].logger, &scorer, &[0u8; 32]
996+
$amt, $nodes[0].logger, &scorer, &0, &[0u8; 32]
997997
).unwrap(), phantom_route_hint.phantom_scid)
998998
}
999999
}}

‎lightning/src/ln/payment_tests.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -864,7 +864,7 @@ fn get_ldk_payment_preimage() {
864864
let route = get_route(
865865
&nodes[0].node.get_our_node_id(), &payment_params, &nodes[0].network_graph.read_only(),
866866
Some(&nodes[0].node.list_usable_channels().iter().collect::<Vec<_>>()),
867-
amt_msat, nodes[0].logger, &scorer, &random_seed_bytes).unwrap();
867+
amt_msat, nodes[0].logger, &scorer, &0, &random_seed_bytes).unwrap();
868868
nodes[0].node.send_payment_with_route(&route, payment_hash,
869869
RecipientOnionFields::secret_only(payment_secret), PaymentId(payment_hash.0)).unwrap();
870870
check_added_monitors!(nodes[0], 1);
@@ -1418,7 +1418,7 @@ fn do_test_intercepted_payment(test: InterceptTest) {
14181418
let route = get_route(
14191419
&nodes[0].node.get_our_node_id(), &route_params.payment_params,
14201420
&nodes[0].network_graph.read_only(), None, route_params.final_value_msat,
1421-
nodes[0].logger, &scorer, &random_seed_bytes,
1421+
nodes[0].logger, &scorer, &0, &random_seed_bytes,
14221422
).unwrap();
14231423

14241424
let (payment_hash, payment_secret) = nodes[2].node.create_inbound_payment(Some(amt_msat), 60 * 60, None).unwrap();

‎lightning/src/ln/shutdown_tests.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,9 @@ fn updates_shutdown_wait() {
9595
let (_, payment_hash, payment_secret) = get_payment_preimage_hash!(nodes[0]);
9696

9797
let payment_params_1 = PaymentParameters::from_node_id(nodes[1].node.get_our_node_id(), TEST_FINAL_CLTV).with_features(nodes[1].node.invoice_features());
98-
let route_1 = get_route(&nodes[0].node.get_our_node_id(), &payment_params_1, &nodes[0].network_graph.read_only(), None, 100000, &logger, &scorer, &random_seed_bytes).unwrap();
98+
let route_1 = get_route(&nodes[0].node.get_our_node_id(), &payment_params_1, &nodes[0].network_graph.read_only(), None, 100000, &logger, &scorer, &0, &random_seed_bytes).unwrap();
9999
let payment_params_2 = PaymentParameters::from_node_id(nodes[0].node.get_our_node_id(), TEST_FINAL_CLTV).with_features(nodes[0].node.invoice_features());
100-
let route_2 = get_route(&nodes[1].node.get_our_node_id(), &payment_params_2, &nodes[1].network_graph.read_only(), None, 100000, &logger, &scorer, &random_seed_bytes).unwrap();
100+
let route_2 = get_route(&nodes[1].node.get_our_node_id(), &payment_params_2, &nodes[1].network_graph.read_only(), None, 100000, &logger, &scorer, &0, &random_seed_bytes).unwrap();
101101
unwrap_send_err!(nodes[0].node.send_payment_with_route(&route_1, payment_hash,
102102
RecipientOnionFields::secret_only(payment_secret), PaymentId(payment_hash.0)
103103
), true, APIError::ChannelUnavailable {..}, {});

0 commit comments

Comments
 (0)