@@ -1278,48 +1278,17 @@ fn successful_probe_yields_event() {
1278
1278
create_announced_chan_between_nodes ( & nodes, 0 , 1 ) ;
1279
1279
create_announced_chan_between_nodes ( & nodes, 1 , 2 ) ;
1280
1280
1281
- let ( route, _, _, _) = get_route_and_payment_hash ! ( & nodes[ 0 ] , nodes[ 2 ] , 100_000 ) ;
1281
+ let recv_value = 100_000 ;
1282
+ let ( route, payment_hash, _, _) = get_route_and_payment_hash ! ( & nodes[ 0 ] , nodes[ 2 ] , recv_value) ;
1282
1283
1283
- let ( payment_hash , payment_id ) = nodes[ 0 ] . node . send_probe ( route. paths [ 0 ] . clone ( ) ) . unwrap ( ) ;
1284
+ let res = nodes[ 0 ] . node . send_probe ( route. paths [ 0 ] . clone ( ) ) . unwrap ( ) ;
1284
1285
1285
- // node[0] -- update_add_htlcs -> node[1]
1286
- check_added_monitors ! ( nodes[ 0 ] , 1 ) ;
1287
- let updates = get_htlc_update_msgs ! ( nodes[ 0 ] , nodes[ 1 ] . node. get_our_node_id( ) ) ;
1288
- let probe_event = SendEvent :: from_commitment_update ( nodes[ 1 ] . node . get_our_node_id ( ) , updates) ;
1289
- nodes[ 1 ] . node . handle_update_add_htlc ( & nodes[ 0 ] . node . get_our_node_id ( ) , & probe_event. msgs [ 0 ] ) ;
1290
- check_added_monitors ! ( nodes[ 1 ] , 0 ) ;
1291
- commitment_signed_dance ! ( nodes[ 1 ] , nodes[ 0 ] , probe_event. commitment_msg, false ) ;
1292
- expect_pending_htlcs_forwardable ! ( nodes[ 1 ] ) ;
1286
+ let expected_route: & [ & [ & Node ] ] = & [ & [ & nodes[ 1 ] , & nodes[ 2 ] ] ] ;
1293
1287
1294
- // node[1] -- update_add_htlcs -> node[2]
1295
- check_added_monitors ! ( nodes[ 1 ] , 1 ) ;
1296
- let updates = get_htlc_update_msgs ! ( nodes[ 1 ] , nodes[ 2 ] . node. get_our_node_id( ) ) ;
1297
- let probe_event = SendEvent :: from_commitment_update ( nodes[ 1 ] . node . get_our_node_id ( ) , updates) ;
1298
- nodes[ 2 ] . node . handle_update_add_htlc ( & nodes[ 1 ] . node . get_our_node_id ( ) , & probe_event. msgs [ 0 ] ) ;
1299
- check_added_monitors ! ( nodes[ 2 ] , 0 ) ;
1300
- commitment_signed_dance ! ( nodes[ 2 ] , nodes[ 1 ] , probe_event. commitment_msg, true , true ) ;
1288
+ send_probe_along_route ( & nodes[ 0 ] , expected_route) ;
1301
1289
1302
- // node[1] <- update_fail_htlcs -- node[2]
1303
- let updates = get_htlc_update_msgs ! ( nodes[ 2 ] , nodes[ 1 ] . node. get_our_node_id( ) ) ;
1304
- nodes[ 1 ] . node . handle_update_fail_htlc ( & nodes[ 2 ] . node . get_our_node_id ( ) , & updates. update_fail_htlcs [ 0 ] ) ;
1305
- check_added_monitors ! ( nodes[ 1 ] , 0 ) ;
1306
- commitment_signed_dance ! ( nodes[ 1 ] , nodes[ 2 ] , updates. commitment_signed, true ) ;
1307
-
1308
- // node[0] <- update_fail_htlcs -- node[1]
1309
- let updates = get_htlc_update_msgs ! ( nodes[ 1 ] , nodes[ 0 ] . node. get_our_node_id( ) ) ;
1310
- nodes[ 0 ] . node . handle_update_fail_htlc ( & nodes[ 1 ] . node . get_our_node_id ( ) , & updates. update_fail_htlcs [ 0 ] ) ;
1311
- check_added_monitors ! ( nodes[ 0 ] , 0 ) ;
1312
- commitment_signed_dance ! ( nodes[ 0 ] , nodes[ 1 ] , updates. commitment_signed, false ) ;
1290
+ expect_probe_successful_events ( & nodes[ 0 ] , vec ! [ res] ) ;
1313
1291
1314
- let mut events = nodes[ 0 ] . node . get_and_clear_pending_events ( ) ;
1315
- assert_eq ! ( events. len( ) , 1 ) ;
1316
- match events. drain ( ..) . next ( ) . unwrap ( ) {
1317
- crate :: events:: Event :: ProbeSuccessful { payment_id : ev_pid, payment_hash : ev_ph, .. } => {
1318
- assert_eq ! ( payment_id, ev_pid) ;
1319
- assert_eq ! ( payment_hash, ev_ph) ;
1320
- } ,
1321
- _ => panic ! ( ) ,
1322
- } ;
1323
1292
assert ! ( !nodes[ 0 ] . node. has_pending_payments( ) ) ;
1324
1293
}
1325
1294
@@ -1424,6 +1393,94 @@ fn onchain_failed_probe_yields_event() {
1424
1393
assert ! ( !nodes[ 0 ] . node. has_pending_payments( ) ) ;
1425
1394
}
1426
1395
1396
+ #[ test]
1397
+ fn preflight_probes_yield_event_skip_private_hop ( ) {
1398
+ let chanmon_cfgs = create_chanmon_cfgs ( 5 ) ;
1399
+ let node_cfgs = create_node_cfgs ( 5 , & chanmon_cfgs) ;
1400
+
1401
+ // We alleviate the HTLC max-in-flight limit, as otherwise we'd always be limited through that.
1402
+ let mut no_htlc_limit_config = test_default_channel_config ( ) ;
1403
+ no_htlc_limit_config. channel_handshake_config . max_inbound_htlc_value_in_flight_percent_of_channel = 100 ;
1404
+
1405
+ let user_configs = std:: iter:: repeat ( no_htlc_limit_config) . take ( 5 ) . map ( |c| Some ( c) ) . collect :: < Vec < Option < UserConfig > > > ( ) ;
1406
+ let node_chanmgrs = create_node_chanmgrs ( 5 , & node_cfgs, & user_configs) ;
1407
+ let nodes = create_network ( 5 , & node_cfgs, & node_chanmgrs) ;
1408
+
1409
+ // Setup channel topology:
1410
+ // N0 -(1M:0)- N1 -(1M:0)- N2 -(70k:0)- N3 -(50k:0)- N4
1411
+
1412
+ create_announced_chan_between_nodes_with_value ( & nodes, 0 , 1 , 1_000_000 , 0 ) ;
1413
+ create_announced_chan_between_nodes_with_value ( & nodes, 1 , 2 , 1_000_000 , 0 ) ;
1414
+ create_announced_chan_between_nodes_with_value ( & nodes, 2 , 3 , 70_000 , 0 ) ;
1415
+ create_unannounced_chan_between_nodes_with_value ( & nodes, 3 , 4 , 50_000 , 0 ) ;
1416
+
1417
+ let mut invoice_features = Bolt11InvoiceFeatures :: empty ( ) ;
1418
+ invoice_features. set_basic_mpp_optional ( ) ;
1419
+
1420
+ let payment_params = PaymentParameters :: from_node_id ( nodes[ 3 ] . node . get_our_node_id ( ) , TEST_FINAL_CLTV )
1421
+ . with_bolt11_features ( invoice_features) . unwrap ( ) ;
1422
+
1423
+ let recv_value = 50_000_000 ;
1424
+ let route_params = RouteParameters :: from_payment_params_and_value ( payment_params, recv_value) ;
1425
+ let res = nodes[ 0 ] . node . send_preflight_probes ( route_params, None ) . unwrap ( ) ;
1426
+
1427
+ let expected_route: & [ & [ & Node ] ] = & [ & [ & nodes[ 1 ] , & nodes[ 2 ] , & nodes[ 3 ] ] ] ;
1428
+
1429
+ assert_eq ! ( res. len( ) , expected_route. len( ) ) ;
1430
+
1431
+ send_probe_along_route ( & nodes[ 0 ] , expected_route) ;
1432
+
1433
+ expect_probe_successful_events ( & nodes[ 0 ] , res. clone ( ) ) ;
1434
+
1435
+ assert ! ( !nodes[ 0 ] . node. has_pending_payments( ) ) ;
1436
+ }
1437
+
1438
+ #[ test]
1439
+ fn preflight_probes_yield_event ( ) {
1440
+ let chanmon_cfgs = create_chanmon_cfgs ( 4 ) ;
1441
+ let node_cfgs = create_node_cfgs ( 4 , & chanmon_cfgs) ;
1442
+
1443
+ // We alleviate the HTLC max-in-flight limit, as otherwise we'd always be limited through that.
1444
+ let mut no_htlc_limit_config = test_default_channel_config ( ) ;
1445
+ no_htlc_limit_config. channel_handshake_config . max_inbound_htlc_value_in_flight_percent_of_channel = 100 ;
1446
+
1447
+ let user_configs = std:: iter:: repeat ( no_htlc_limit_config) . take ( 4 ) . map ( |c| Some ( c) ) . collect :: < Vec < Option < UserConfig > > > ( ) ;
1448
+ let node_chanmgrs = create_node_chanmgrs ( 4 , & node_cfgs, & user_configs) ;
1449
+ let nodes = create_network ( 4 , & node_cfgs, & node_chanmgrs) ;
1450
+
1451
+ // Setup channel topology:
1452
+ // (1M:0)- N1 -(30k:0)
1453
+ // / \
1454
+ // N0 N4
1455
+ // \ /
1456
+ // (1M:0)- N2 -(70k:0)
1457
+ //
1458
+ create_announced_chan_between_nodes_with_value ( & nodes, 0 , 1 , 1_000_000 , 0 ) ;
1459
+ create_announced_chan_between_nodes_with_value ( & nodes, 0 , 2 , 1_000_000 , 0 ) ;
1460
+ create_announced_chan_between_nodes_with_value ( & nodes, 1 , 3 , 30_000 , 0 ) ;
1461
+ create_announced_chan_between_nodes_with_value ( & nodes, 2 , 3 , 70_000 , 0 ) ;
1462
+
1463
+ let mut invoice_features = Bolt11InvoiceFeatures :: empty ( ) ;
1464
+ invoice_features. set_basic_mpp_optional ( ) ;
1465
+
1466
+ let payment_params = PaymentParameters :: from_node_id ( nodes[ 3 ] . node . get_our_node_id ( ) , TEST_FINAL_CLTV )
1467
+ . with_bolt11_features ( invoice_features) . unwrap ( ) ;
1468
+
1469
+ let recv_value = 50_000_000 ;
1470
+ let route_params = RouteParameters :: from_payment_params_and_value ( payment_params, recv_value) ;
1471
+ let res = nodes[ 0 ] . node . send_preflight_probes ( route_params, None ) . unwrap ( ) ;
1472
+
1473
+ let expected_route: & [ & [ & Node ] ] = & [ & [ & nodes[ 1 ] , & nodes[ 3 ] ] , & [ & nodes[ 2 ] , & nodes[ 3 ] ] ] ;
1474
+
1475
+ assert_eq ! ( res. len( ) , expected_route. len( ) ) ;
1476
+
1477
+ send_probe_along_route ( & nodes[ 0 ] , expected_route) ;
1478
+
1479
+ expect_probe_successful_events ( & nodes[ 0 ] , res. clone ( ) ) ;
1480
+
1481
+ assert ! ( !nodes[ 0 ] . node. has_pending_payments( ) ) ;
1482
+ }
1483
+
1427
1484
#[ test]
1428
1485
fn preflight_probes_yield_event_and_skip ( ) {
1429
1486
let chanmon_cfgs = create_chanmon_cfgs ( 5 ) ;
@@ -1444,7 +1501,7 @@ fn preflight_probes_yield_event_and_skip() {
1444
1501
// \ /
1445
1502
// (70k:0)- N3 -(1M:0)
1446
1503
//
1447
- let first_chan_update = create_announced_chan_between_nodes_with_value ( & nodes, 0 , 1 , 100_000 , 0 ) . 0 ;
1504
+ create_announced_chan_between_nodes_with_value ( & nodes, 0 , 1 , 100_000 , 0 ) ;
1448
1505
create_announced_chan_between_nodes_with_value ( & nodes, 1 , 2 , 30_000 , 0 ) ;
1449
1506
create_announced_chan_between_nodes_with_value ( & nodes, 1 , 3 , 70_000 , 0 ) ;
1450
1507
create_announced_chan_between_nodes_with_value ( & nodes, 2 , 4 , 1_000_000 , 0 ) ;
@@ -1453,70 +1510,22 @@ fn preflight_probes_yield_event_and_skip() {
1453
1510
let mut invoice_features = Bolt11InvoiceFeatures :: empty ( ) ;
1454
1511
invoice_features. set_basic_mpp_optional ( ) ;
1455
1512
1456
- let mut payment_params = PaymentParameters :: from_node_id ( nodes[ 4 ] . node . get_our_node_id ( ) , TEST_FINAL_CLTV )
1513
+ let payment_params = PaymentParameters :: from_node_id ( nodes[ 4 ] . node . get_our_node_id ( ) , TEST_FINAL_CLTV )
1457
1514
. with_bolt11_features ( invoice_features) . unwrap ( ) ;
1458
1515
1459
- let route_params = RouteParameters :: from_payment_params_and_value ( payment_params, 80_000_000 ) ;
1516
+ let recv_value = 80_000_000 ;
1517
+ let route_params = RouteParameters :: from_payment_params_and_value ( payment_params, recv_value) ;
1460
1518
let res = nodes[ 0 ] . node . send_preflight_probes ( route_params, None ) . unwrap ( ) ;
1461
1519
1520
+ let expected_route : & [ & [ & Node ] ] = & [ & [ & nodes[ 1 ] , & nodes[ 2 ] , & nodes[ 4 ] ] ] ;
1521
+
1462
1522
// We check that only one probe was sent, the other one was skipped due to limited liquidity.
1463
1523
assert_eq ! ( res. len( ) , 1 ) ;
1464
- let log_msg = format ! ( "Skipped sending payment probe to avoid putting channel {} under the liquidity limit." ,
1465
- first_chan_update. contents. short_channel_id) ;
1466
- node_cfgs[ 0 ] . logger . assert_log_contains ( "lightning::ln::channelmanager" , & log_msg, 1 ) ;
1467
1524
1468
- let ( payment_hash , payment_id ) = res . first ( ) . unwrap ( ) ;
1525
+ send_probe_along_route ( & nodes [ 0 ] , expected_route ) ;
1469
1526
1470
- // node[0] -- update_add_htlcs -> node[1]
1471
- check_added_monitors ! ( nodes[ 0 ] , 1 ) ;
1472
- let probe_event = SendEvent :: from_node ( & nodes[ 0 ] ) ;
1473
- nodes[ 1 ] . node . handle_update_add_htlc ( & nodes[ 0 ] . node . get_our_node_id ( ) , & probe_event. msgs [ 0 ] ) ;
1474
- check_added_monitors ! ( nodes[ 1 ] , 0 ) ;
1475
- commitment_signed_dance ! ( nodes[ 1 ] , nodes[ 0 ] , probe_event. commitment_msg, false ) ;
1476
- expect_pending_htlcs_forwardable ! ( nodes[ 1 ] ) ;
1527
+ expect_probe_successful_events ( & nodes[ 0 ] , res. clone ( ) ) ;
1477
1528
1478
- // node[1] -- update_add_htlcs -> node[2]
1479
- check_added_monitors ! ( nodes[ 1 ] , 1 ) ;
1480
- let probe_event = SendEvent :: from_node ( & nodes[ 1 ] ) ;
1481
- nodes[ 2 ] . node . handle_update_add_htlc ( & nodes[ 1 ] . node . get_our_node_id ( ) , & probe_event. msgs [ 0 ] ) ;
1482
- check_added_monitors ! ( nodes[ 2 ] , 0 ) ;
1483
- commitment_signed_dance ! ( nodes[ 2 ] , nodes[ 1 ] , probe_event. commitment_msg, false ) ;
1484
- expect_pending_htlcs_forwardable ! ( nodes[ 2 ] ) ;
1485
-
1486
- // node[2] -- update_add_htlcs -> node[4]
1487
- check_added_monitors ! ( nodes[ 2 ] , 1 ) ;
1488
- let probe_event = SendEvent :: from_node ( & nodes[ 2 ] ) ;
1489
- nodes[ 4 ] . node . handle_update_add_htlc ( & nodes[ 2 ] . node . get_our_node_id ( ) , & probe_event. msgs [ 0 ] ) ;
1490
- check_added_monitors ! ( nodes[ 4 ] , 0 ) ;
1491
- commitment_signed_dance ! ( nodes[ 4 ] , nodes[ 2 ] , probe_event. commitment_msg, true , true ) ;
1492
-
1493
- // node[2] <- update_fail_htlcs -- node[4]
1494
- let updates = get_htlc_update_msgs ! ( nodes[ 4 ] , nodes[ 2 ] . node. get_our_node_id( ) ) ;
1495
- nodes[ 2 ] . node . handle_update_fail_htlc ( & nodes[ 4 ] . node . get_our_node_id ( ) , & updates. update_fail_htlcs [ 0 ] ) ;
1496
- check_added_monitors ! ( nodes[ 2 ] , 0 ) ;
1497
- commitment_signed_dance ! ( nodes[ 2 ] , nodes[ 4 ] , updates. commitment_signed, true ) ;
1498
-
1499
- // node[1] <- update_fail_htlcs -- node[2]
1500
- let updates = get_htlc_update_msgs ! ( nodes[ 2 ] , nodes[ 1 ] . node. get_our_node_id( ) ) ;
1501
- nodes[ 1 ] . node . handle_update_fail_htlc ( & nodes[ 2 ] . node . get_our_node_id ( ) , & updates. update_fail_htlcs [ 0 ] ) ;
1502
- check_added_monitors ! ( nodes[ 1 ] , 0 ) ;
1503
- commitment_signed_dance ! ( nodes[ 1 ] , nodes[ 2 ] , updates. commitment_signed, true ) ;
1504
-
1505
- // node[0] <- update_fail_htlcs -- node[1]
1506
- let updates = get_htlc_update_msgs ! ( nodes[ 1 ] , nodes[ 0 ] . node. get_our_node_id( ) ) ;
1507
- nodes[ 0 ] . node . handle_update_fail_htlc ( & nodes[ 1 ] . node . get_our_node_id ( ) , & updates. update_fail_htlcs [ 0 ] ) ;
1508
- check_added_monitors ! ( nodes[ 0 ] , 0 ) ;
1509
- commitment_signed_dance ! ( nodes[ 0 ] , nodes[ 1 ] , updates. commitment_signed, false ) ;
1510
-
1511
- let mut events = nodes[ 0 ] . node . get_and_clear_pending_events ( ) ;
1512
- assert_eq ! ( events. len( ) , 1 ) ;
1513
- match events. drain ( ..) . next ( ) . unwrap ( ) {
1514
- crate :: events:: Event :: ProbeSuccessful { payment_id : ev_pid, payment_hash : ev_ph, .. } => {
1515
- assert_eq ! ( * payment_id, ev_pid) ;
1516
- assert_eq ! ( * payment_hash, ev_ph) ;
1517
- } ,
1518
- _ => panic ! ( ) ,
1519
- } ;
1520
1529
assert ! ( !nodes[ 0 ] . node. has_pending_payments( ) ) ;
1521
1530
}
1522
1531
0 commit comments