@@ -294,6 +294,9 @@ fn do_test_async_commitment_signature_for_commitment_signed_revoke_and_ack(
294
294
// Send a payment.
295
295
let src = & nodes[ 0 ] ;
296
296
let dst = & nodes[ 1 ] ;
297
+ let src_node_id = src. node . get_our_node_id ( ) ;
298
+ let dst_node_id = dst. node . get_our_node_id ( ) ;
299
+
297
300
let ( route, our_payment_hash, _our_payment_preimage, our_payment_secret) =
298
301
get_route_and_payment_hash ! ( src, dst, 8000000 ) ;
299
302
let recipient_fields = RecipientOnionFields :: secret_only ( our_payment_secret) ;
@@ -309,52 +312,37 @@ fn do_test_async_commitment_signature_for_commitment_signed_revoke_and_ack(
309
312
assert_eq ! ( events. len( ) , 1 ) ;
310
313
SendEvent :: from_event ( events. remove ( 0 ) )
311
314
} ;
312
- assert_eq ! ( payment_event. node_id, dst . node . get_our_node_id ( ) ) ;
315
+ assert_eq ! ( payment_event. node_id, dst_node_id ) ;
313
316
assert_eq ! ( payment_event. msgs. len( ) , 1 ) ;
314
317
315
- dst. node . handle_update_add_htlc ( src . node . get_our_node_id ( ) , & payment_event. msgs [ 0 ] ) ;
318
+ dst. node . handle_update_add_htlc ( src_node_id , & payment_event. msgs [ 0 ] ) ;
316
319
317
320
// Mark dst's signer as unavailable and handle src's commitment_signed: while dst won't yet have a
318
321
// `commitment_signed` of its own to offer, it should publish a `revoke_and_ack`.
319
- dst. disable_channel_signer_op (
320
- & src. node . get_our_node_id ( ) ,
321
- & chan_id,
322
- SignerOp :: GetPerCommitmentPoint ,
323
- ) ;
324
- dst. disable_channel_signer_op (
325
- & src. node . get_our_node_id ( ) ,
326
- & chan_id,
327
- SignerOp :: ReleaseCommitmentSecret ,
328
- ) ;
329
- dst. disable_channel_signer_op (
330
- & src. node . get_our_node_id ( ) ,
331
- & chan_id,
332
- SignerOp :: SignCounterpartyCommitment ,
333
- ) ;
334
- dst. node . handle_commitment_signed_batch_test (
335
- src. node . get_our_node_id ( ) ,
336
- & payment_event. commitment_msg ,
337
- ) ;
322
+ dst. disable_channel_signer_op ( & src_node_id, & chan_id, SignerOp :: GetPerCommitmentPoint ) ;
323
+ dst. disable_channel_signer_op ( & src_node_id, & chan_id, SignerOp :: ReleaseCommitmentSecret ) ;
324
+ dst. disable_channel_signer_op ( & src_node_id, & chan_id, SignerOp :: SignCounterpartyCommitment ) ;
325
+ dst. node . handle_commitment_signed_batch_test ( src_node_id, & payment_event. commitment_msg ) ;
338
326
check_added_monitors ( dst, 1 ) ;
339
327
340
328
let mut enabled_signer_ops = new_hash_set ( ) ;
341
329
log_trace ! ( dst. logger, "enable_signer_op_order={:?}" , enable_signer_op_order) ;
342
330
for op in enable_signer_op_order {
343
331
enabled_signer_ops. insert ( op) ;
344
- dst. enable_channel_signer_op ( & src . node . get_our_node_id ( ) , & chan_id, op) ;
345
- dst. node . signer_unblocked ( Some ( ( src . node . get_our_node_id ( ) , chan_id) ) ) ;
332
+ dst. enable_channel_signer_op ( & src_node_id , & chan_id, op) ;
333
+ dst. node . signer_unblocked ( Some ( ( src_node_id , chan_id) ) ) ;
346
334
347
335
if enabled_signer_ops. contains ( & SignerOp :: GetPerCommitmentPoint )
348
336
&& enabled_signer_ops. contains ( & SignerOp :: ReleaseCommitmentSecret )
349
337
{
350
338
// We are just able to send revoke_and_ack
351
339
if op == SignerOp :: GetPerCommitmentPoint || op == SignerOp :: ReleaseCommitmentSecret {
352
- get_event_msg ! ( dst, MessageSendEvent :: SendRevokeAndACK , src . node . get_our_node_id ( ) ) ;
340
+ get_event_msg ! ( dst, MessageSendEvent :: SendRevokeAndACK , src_node_id ) ;
353
341
}
354
342
// We either just sent or previously sent revoke_and_ack
355
343
// and now we are able to send commitment_signed
356
344
if op == SignerOp :: SignCounterpartyCommitment {
357
- get_htlc_update_msgs ( dst, & src . node . get_our_node_id ( ) ) ;
345
+ get_htlc_update_msgs ( dst, & src_node_id ) ;
358
346
}
359
347
} else {
360
348
// We can't send either message until RAA is unblocked
@@ -533,6 +521,9 @@ fn do_test_async_raa_peer_disconnect(
533
521
// Send a payment.
534
522
let src = & nodes[ 0 ] ;
535
523
let dst = & nodes[ 1 ] ;
524
+ let src_node_id = src. node . get_our_node_id ( ) ;
525
+ let dst_node_id = dst. node . get_our_node_id ( ) ;
526
+
536
527
let ( route, our_payment_hash, _our_payment_preimage, our_payment_secret) =
537
528
get_route_and_payment_hash ! ( src, dst, 8000000 ) ;
538
529
let recipient_fields = RecipientOnionFields :: secret_only ( our_payment_secret) ;
@@ -548,10 +539,10 @@ fn do_test_async_raa_peer_disconnect(
548
539
assert_eq ! ( events. len( ) , 1 ) ;
549
540
SendEvent :: from_event ( events. remove ( 0 ) )
550
541
} ;
551
- assert_eq ! ( payment_event. node_id, dst . node . get_our_node_id ( ) ) ;
542
+ assert_eq ! ( payment_event. node_id, dst_node_id ) ;
552
543
assert_eq ! ( payment_event. msgs. len( ) , 1 ) ;
553
544
554
- dst. node . handle_update_add_htlc ( src . node . get_our_node_id ( ) , & payment_event. msgs [ 0 ] ) ;
545
+ dst. node . handle_update_add_htlc ( src_node_id , & payment_event. msgs [ 0 ] ) ;
555
546
556
547
if test_case == UnblockSignerAcrossDisconnectCase :: BeforeMonitorRestored {
557
548
// Fail to persist the monitor update when handling the commitment_signed.
@@ -560,47 +551,44 @@ fn do_test_async_raa_peer_disconnect(
560
551
561
552
// Mark dst's signer as unavailable and handle src's commitment_signed: while dst won't yet have a
562
553
// `commitment_signed` of its own to offer, it should publish a `revoke_and_ack`.
563
- dst. disable_channel_signer_op ( & src. node . get_our_node_id ( ) , & chan_id, block_raa_signer_op) ;
564
- dst. node . handle_commitment_signed_batch_test (
565
- src. node . get_our_node_id ( ) ,
566
- & payment_event. commitment_msg ,
567
- ) ;
554
+ dst. disable_channel_signer_op ( & src_node_id, & chan_id, block_raa_signer_op) ;
555
+ dst. node . handle_commitment_signed_batch_test ( src_node_id, & payment_event. commitment_msg ) ;
568
556
check_added_monitors ( dst, 1 ) ;
569
557
570
558
let events = dst. node . get_and_clear_pending_msg_events ( ) ;
571
559
assert ! ( events. is_empty( ) , "expected no message, got {}" , events. len( ) ) ;
572
560
573
561
// Now disconnect and reconnect the peers.
574
- src. node . peer_disconnected ( dst . node . get_our_node_id ( ) ) ;
575
- dst. node . peer_disconnected ( src . node . get_our_node_id ( ) ) ;
562
+ src. node . peer_disconnected ( dst_node_id ) ;
563
+ dst. node . peer_disconnected ( src_node_id ) ;
576
564
577
565
// do reestablish stuff
578
566
let init_msg = & msgs:: Init {
579
567
features : dst. node . init_features ( ) ,
580
568
networks : None ,
581
569
remote_network_address : None ,
582
570
} ;
583
- src. node . peer_connected ( dst . node . get_our_node_id ( ) , init_msg, true ) . unwrap ( ) ;
571
+ src. node . peer_connected ( dst_node_id , init_msg, true ) . unwrap ( ) ;
584
572
let reestablish_1 = get_chan_reestablish_msgs ! ( src, dst) ;
585
573
assert_eq ! ( reestablish_1. len( ) , 1 ) ;
586
574
let init_msg = & msgs:: Init {
587
575
features : src. node . init_features ( ) ,
588
576
networks : None ,
589
577
remote_network_address : None ,
590
578
} ;
591
- dst. node . peer_connected ( src . node . get_our_node_id ( ) , init_msg, false ) . unwrap ( ) ;
579
+ dst. node . peer_connected ( src_node_id , init_msg, false ) . unwrap ( ) ;
592
580
let reestablish_2 = get_chan_reestablish_msgs ! ( dst, src) ;
593
581
assert_eq ! ( reestablish_2. len( ) , 1 ) ;
594
582
595
583
if test_case == UnblockSignerAcrossDisconnectCase :: BeforeReestablish {
596
584
// Reenable the signer before the reestablish.
597
- dst. enable_channel_signer_op ( & src . node . get_our_node_id ( ) , & chan_id, block_raa_signer_op) ;
585
+ dst. enable_channel_signer_op ( & src_node_id , & chan_id, block_raa_signer_op) ;
598
586
}
599
587
600
- dst. node . handle_channel_reestablish ( src . node . get_our_node_id ( ) , & reestablish_1[ 0 ] ) ;
588
+ dst. node . handle_channel_reestablish ( src_node_id , & reestablish_1[ 0 ] ) ;
601
589
602
590
if test_case == UnblockSignerAcrossDisconnectCase :: BeforeMonitorRestored {
603
- dst. enable_channel_signer_op ( & src . node . get_our_node_id ( ) , & chan_id, block_raa_signer_op) ;
591
+ dst. enable_channel_signer_op ( & src_node_id , & chan_id, block_raa_signer_op) ;
604
592
chanmon_cfgs[ 1 ] . persister . set_update_ret ( ChannelMonitorUpdateStatus :: Completed ) ;
605
593
let latest_update;
606
594
{
@@ -624,8 +612,8 @@ fn do_test_async_raa_peer_disconnect(
624
612
}
625
613
626
614
// Mark dst's signer as available and retry: we now expect to see dst's RAA + CS.
627
- dst. enable_channel_signer_op ( & src . node . get_our_node_id ( ) , & chan_id, block_raa_signer_op) ;
628
- dst. node . signer_unblocked ( Some ( ( src . node . get_our_node_id ( ) , chan_id) ) ) ;
615
+ dst. enable_channel_signer_op ( & src_node_id , & chan_id, block_raa_signer_op) ;
616
+ dst. node . signer_unblocked ( Some ( ( src_node_id , chan_id) ) ) ;
629
617
630
618
if test_case == UnblockSignerAcrossDisconnectCase :: AtEnd {
631
619
let ( _, revoke_and_ack, commitment_signed, resend_order) =
@@ -681,6 +669,9 @@ fn do_test_async_commitment_signature_peer_disconnect(
681
669
// Send a payment.
682
670
let src = & nodes[ 0 ] ;
683
671
let dst = & nodes[ 1 ] ;
672
+ let src_node_id = src. node . get_our_node_id ( ) ;
673
+ let dst_node_id = dst. node . get_our_node_id ( ) ;
674
+
684
675
let ( route, our_payment_hash, _our_payment_preimage, our_payment_secret) =
685
676
get_route_and_payment_hash ! ( src, dst, 8000000 ) ;
686
677
let recipient_fields = RecipientOnionFields :: secret_only ( our_payment_secret) ;
@@ -696,10 +687,10 @@ fn do_test_async_commitment_signature_peer_disconnect(
696
687
assert_eq ! ( events. len( ) , 1 ) ;
697
688
SendEvent :: from_event ( events. remove ( 0 ) )
698
689
} ;
699
- assert_eq ! ( payment_event. node_id, dst . node . get_our_node_id ( ) ) ;
690
+ assert_eq ! ( payment_event. node_id, dst_node_id ) ;
700
691
assert_eq ! ( payment_event. msgs. len( ) , 1 ) ;
701
692
702
- dst. node . handle_update_add_htlc ( src . node . get_our_node_id ( ) , & payment_event. msgs [ 0 ] ) ;
693
+ dst. node . handle_update_add_htlc ( src_node_id , & payment_event. msgs [ 0 ] ) ;
703
694
704
695
if test_case == UnblockSignerAcrossDisconnectCase :: BeforeMonitorRestored {
705
696
// Fail to persist the monitor update when handling the commitment_signed.
@@ -708,60 +699,45 @@ fn do_test_async_commitment_signature_peer_disconnect(
708
699
709
700
// Mark dst's signer as unavailable and handle src's commitment_signed: while dst won't yet have a
710
701
// `commitment_signed` of its own to offer, it should publish a `revoke_and_ack`.
711
- dst. disable_channel_signer_op (
712
- & src. node . get_our_node_id ( ) ,
713
- & chan_id,
714
- SignerOp :: SignCounterpartyCommitment ,
715
- ) ;
716
- dst. node . handle_commitment_signed_batch_test (
717
- src. node . get_our_node_id ( ) ,
718
- & payment_event. commitment_msg ,
719
- ) ;
702
+ dst. disable_channel_signer_op ( & src_node_id, & chan_id, SignerOp :: SignCounterpartyCommitment ) ;
703
+ dst. node . handle_commitment_signed_batch_test ( src_node_id, & payment_event. commitment_msg ) ;
720
704
check_added_monitors ( dst, 1 ) ;
721
705
722
706
if test_case != UnblockSignerAcrossDisconnectCase :: BeforeMonitorRestored {
723
- get_event_msg ! ( dst, MessageSendEvent :: SendRevokeAndACK , src . node . get_our_node_id ( ) ) ;
707
+ get_event_msg ! ( dst, MessageSendEvent :: SendRevokeAndACK , src_node_id ) ;
724
708
}
725
709
726
710
// Now disconnect and reconnect the peers.
727
- src. node . peer_disconnected ( dst . node . get_our_node_id ( ) ) ;
728
- dst. node . peer_disconnected ( src . node . get_our_node_id ( ) ) ;
711
+ src. node . peer_disconnected ( dst_node_id ) ;
712
+ dst. node . peer_disconnected ( src_node_id ) ;
729
713
730
714
// do reestablish stuff
731
715
let init_msg = & msgs:: Init {
732
716
features : dst. node . init_features ( ) ,
733
717
networks : None ,
734
718
remote_network_address : None ,
735
719
} ;
736
- src. node . peer_connected ( dst . node . get_our_node_id ( ) , init_msg, true ) . unwrap ( ) ;
720
+ src. node . peer_connected ( dst_node_id , init_msg, true ) . unwrap ( ) ;
737
721
let reestablish_1 = get_chan_reestablish_msgs ! ( src, dst) ;
738
722
assert_eq ! ( reestablish_1. len( ) , 1 ) ;
739
723
let init_msg = & msgs:: Init {
740
724
features : src. node . init_features ( ) ,
741
725
networks : None ,
742
726
remote_network_address : None ,
743
727
} ;
744
- dst. node . peer_connected ( src . node . get_our_node_id ( ) , init_msg, false ) . unwrap ( ) ;
728
+ dst. node . peer_connected ( src_node_id , init_msg, false ) . unwrap ( ) ;
745
729
let reestablish_2 = get_chan_reestablish_msgs ! ( dst, src) ;
746
730
assert_eq ! ( reestablish_2. len( ) , 1 ) ;
747
731
748
732
if test_case == UnblockSignerAcrossDisconnectCase :: BeforeReestablish {
749
733
// Reenable the signer before the reestablish.
750
- dst. enable_channel_signer_op (
751
- & src. node . get_our_node_id ( ) ,
752
- & chan_id,
753
- SignerOp :: SignCounterpartyCommitment ,
754
- ) ;
734
+ dst. enable_channel_signer_op ( & src_node_id, & chan_id, SignerOp :: SignCounterpartyCommitment ) ;
755
735
}
756
736
757
- dst. node . handle_channel_reestablish ( src . node . get_our_node_id ( ) , & reestablish_1[ 0 ] ) ;
737
+ dst. node . handle_channel_reestablish ( src_node_id , & reestablish_1[ 0 ] ) ;
758
738
759
739
if test_case == UnblockSignerAcrossDisconnectCase :: BeforeMonitorRestored {
760
- dst. enable_channel_signer_op (
761
- & src. node . get_our_node_id ( ) ,
762
- & chan_id,
763
- SignerOp :: SignCounterpartyCommitment ,
764
- ) ;
740
+ dst. enable_channel_signer_op ( & src_node_id, & chan_id, SignerOp :: SignCounterpartyCommitment ) ;
765
741
chanmon_cfgs[ 1 ] . persister . set_update_ret ( ChannelMonitorUpdateStatus :: Completed ) ;
766
742
let latest_update;
767
743
{
@@ -782,12 +758,8 @@ fn do_test_async_commitment_signature_peer_disconnect(
782
758
}
783
759
784
760
// Mark dst's signer as available and retry: we now expect to see dst's `commitment_signed`.
785
- dst. enable_channel_signer_op (
786
- & src. node . get_our_node_id ( ) ,
787
- & chan_id,
788
- SignerOp :: SignCounterpartyCommitment ,
789
- ) ;
790
- dst. node . signer_unblocked ( Some ( ( src. node . get_our_node_id ( ) , chan_id) ) ) ;
761
+ dst. enable_channel_signer_op ( & src_node_id, & chan_id, SignerOp :: SignCounterpartyCommitment ) ;
762
+ dst. node . signer_unblocked ( Some ( ( src_node_id, chan_id) ) ) ;
791
763
792
764
if test_case == UnblockSignerAcrossDisconnectCase :: AtEnd {
793
765
let ( _, _, commitment_signed, _) = handle_chan_reestablish_msgs ! ( dst, src) ;
0 commit comments