Skip to content

Commit 65a65ea

Browse files
henghongleehenghonglee
committed
Added changes
1 parent 88c63e9 commit 65a65ea

File tree

10 files changed

+502
-491
lines changed

10 files changed

+502
-491
lines changed

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

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -241,21 +241,21 @@ fn update_scorer<'a, S: 'static + Deref<Target = SC> + Send + Sync, SC: 'a + Wri
241241
let mut score = scorer.lock();
242242
match event {
243243
Event::PaymentPathFailed { ref path, short_channel_id: Some(scid), .. } => {
244-
score.payment_path_failed(path, *scid);
244+
score.payment_path_failed(path, *scid, &0);
245245
},
246246
Event::PaymentPathFailed { ref path, payment_failed_permanently: true, .. } => {
247247
// Reached if the destination explicitly failed it back. We treat this as a successful probe
248248
// because the payment made it all the way to the destination with sufficient liquidity.
249-
score.probe_successful(path);
249+
score.probe_successful(path, &0);
250250
},
251251
Event::PaymentPathSuccessful { path, .. } => {
252-
score.payment_path_successful(path);
252+
score.payment_path_successful(path, &0);
253253
},
254254
Event::ProbeSuccessful { path, .. } => {
255-
score.probe_successful(path);
255+
score.probe_successful(path, &0);
256256
},
257257
Event::ProbeFailed { path, short_channel_id: Some(scid), .. } => {
258-
score.probe_failed(path, *scid);
258+
score.probe_failed(path, *scid, &0);
259259
},
260260
_ => {},
261261
}
@@ -1015,11 +1015,12 @@ mod tests {
10151015
}
10161016

10171017
impl Score for TestScorer {
1018+
type ScoreParams = i32;
10181019
fn channel_penalty_msat(
1019-
&self, _short_channel_id: u64, _source: &NodeId, _target: &NodeId, _usage: ChannelUsage
1020+
&self, _short_channel_id: u64, _source: &NodeId, _target: &NodeId, _usage: ChannelUsage, _score_params: &Self::ScoreParams
10201021
) -> u64 { unimplemented!(); }
10211022

1022-
fn payment_path_failed(&mut self, actual_path: &Path, actual_short_channel_id: u64) {
1023+
fn payment_path_failed(&mut self, actual_path: &Path, actual_short_channel_id: u64, _score_params: &Self::ScoreParams) {
10231024
if let Some(expectations) = &mut self.event_expectations {
10241025
match expectations.pop_front().unwrap() {
10251026
TestResult::PaymentFailure { path, short_channel_id } => {
@@ -1039,7 +1040,7 @@ mod tests {
10391040
}
10401041
}
10411042

1042-
fn payment_path_successful(&mut self, actual_path: &Path) {
1043+
fn payment_path_successful(&mut self, actual_path: &Path, _score_params: &Self::ScoreParams) {
10431044
if let Some(expectations) = &mut self.event_expectations {
10441045
match expectations.pop_front().unwrap() {
10451046
TestResult::PaymentFailure { path, .. } => {
@@ -1058,7 +1059,7 @@ mod tests {
10581059
}
10591060
}
10601061

1061-
fn probe_failed(&mut self, actual_path: &Path, _: u64) {
1062+
fn probe_failed(&mut self, actual_path: &Path, _: u64, _score_params: &Self::ScoreParams) {
10621063
if let Some(expectations) = &mut self.event_expectations {
10631064
match expectations.pop_front().unwrap() {
10641065
TestResult::PaymentFailure { path, .. } => {
@@ -1076,7 +1077,7 @@ mod tests {
10761077
}
10771078
}
10781079
}
1079-
fn probe_successful(&mut self, actual_path: &Path) {
1080+
fn probe_successful(&mut self, actual_path: &Path, _score_params: &Self::ScoreParams) {
10801081
if let Some(expectations) = &mut self.event_expectations {
10811082
match expectations.pop_front().unwrap() {
10821083
TestResult::PaymentFailure { path, .. } => {

‎lightning/src/ln/channelmanager.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8324,7 +8324,7 @@ mod tests {
83248324
};
83258325
let route = find_route(
83268326
&nodes[0].node.get_our_node_id(), &route_params, &nodes[0].network_graph,
8327-
None, nodes[0].logger, &scorer, &random_seed_bytes
8327+
None, nodes[0].logger, &scorer, &0, &random_seed_bytes
83288328
).unwrap();
83298329
nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
83308330
RecipientOnionFields::spontaneous_empty(), PaymentId(payment_preimage.0)).unwrap();
@@ -8358,7 +8358,7 @@ mod tests {
83588358
let payment_preimage = PaymentPreimage([42; 32]);
83598359
let route = find_route(
83608360
&nodes[0].node.get_our_node_id(), &route_params, &nodes[0].network_graph,
8361-
None, nodes[0].logger, &scorer, &random_seed_bytes
8361+
None, nodes[0].logger, &scorer,&0, &random_seed_bytes
83628362
).unwrap();
83638363
let payment_hash = nodes[0].node.send_spontaneous_payment(&route, Some(payment_preimage),
83648364
RecipientOnionFields::spontaneous_empty(), PaymentId(payment_preimage.0)).unwrap();
@@ -8421,7 +8421,7 @@ mod tests {
84218421
let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
84228422
let route = find_route(
84238423
&payer_pubkey, &route_params, &network_graph, Some(&first_hops.iter().collect::<Vec<_>>()),
8424-
nodes[0].logger, &scorer, &random_seed_bytes
8424+
nodes[0].logger, &scorer,&0, &random_seed_bytes
84258425
).unwrap();
84268426

84278427
let test_preimage = PaymentPreimage([42; 32]);
@@ -8465,7 +8465,7 @@ mod tests {
84658465
let random_seed_bytes = chanmon_cfgs[1].keys_manager.get_secure_random_bytes();
84668466
let route = find_route(
84678467
&payer_pubkey, &route_params, &network_graph, Some(&first_hops.iter().collect::<Vec<_>>()),
8468-
nodes[0].logger, &scorer, &random_seed_bytes
8468+
nodes[0].logger, &scorer, &0,&random_seed_bytes
84698469
).unwrap();
84708470

84718471
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
@@ -1704,7 +1704,7 @@ pub fn get_route(send_node: &Node, payment_params: &PaymentParameters, recv_valu
17041704
router::get_route(
17051705
&send_node.node.get_our_node_id(), payment_params, &send_node.network_graph.read_only(),
17061706
Some(&send_node.node.list_usable_channels().iter().collect::<Vec<_>>()),
1707-
recv_value, final_cltv_expiry_delta, send_node.logger, &scorer, &random_seed_bytes
1707+
recv_value, final_cltv_expiry_delta, send_node.logger, &scorer, &0, &random_seed_bytes
17081708
)
17091709
}
17101710

@@ -2294,7 +2294,7 @@ pub fn route_over_limit<'a, 'b, 'c>(origin_node: &Node<'a, 'b, 'c>, expected_rou
22942294
let random_seed_bytes = keys_manager.get_secure_random_bytes();
22952295
let route = router::get_route(
22962296
&origin_node.node.get_our_node_id(), &payment_params, &network_graph,
2297-
None, recv_value, TEST_FINAL_CLTV, origin_node.logger, &scorer, &random_seed_bytes).unwrap();
2297+
None, recv_value, TEST_FINAL_CLTV, origin_node.logger, &scorer, &0, &random_seed_bytes).unwrap();
22982298
assert_eq!(route.paths.len(), 1);
22992299
assert_eq!(route.paths[0].hops.len(), expected_route.len());
23002300
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, TEST_FINAL_CLTV, 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, TEST_FINAL_CLTV, 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, 50, nodes[0].logger, &scorer, &random_seed_bytes).unwrap();
7301+
3_000_000, 50, 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, 50, nodes[0].logger, &scorer, &random_seed_bytes).unwrap();
7305+
3_000_000, 50, 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, TEST_FINAL_CLTV, $nodes[0].logger, &scorer, &[0u8; 32]
996+
$amt, TEST_FINAL_CLTV, $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, TEST_FINAL_CLTV, nodes[0].logger, &scorer, &random_seed_bytes).unwrap();
867+
amt_msat, TEST_FINAL_CLTV, 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-
route_params.payment_params.final_cltv_expiry_delta, nodes[0].logger, &scorer,
1421+
route_params.payment_params.final_cltv_expiry_delta, nodes[0].logger, &scorer, &0,
14221422
&random_seed_bytes,
14231423
).unwrap();
14241424

‎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, TEST_FINAL_CLTV, &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, TEST_FINAL_CLTV, &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, TEST_FINAL_CLTV, &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, TEST_FINAL_CLTV, &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)