8
8
// licenses.
9
9
10
10
use bitcoin:: hashes:: hex:: FromHex ;
11
+ use bitcoin:: hashes:: sha256:: Hash as Sha256 ;
12
+ use bitcoin:: hashes:: Hash ;
11
13
use bitcoin:: hex:: DisplayHex ;
12
14
use bitcoin:: secp256k1:: { PublicKey , Scalar , Secp256k1 , SecretKey , schnorr} ;
13
15
use bitcoin:: secp256k1:: ecdh:: SharedSecret ;
@@ -2091,6 +2093,10 @@ fn do_test_trampoline_single_hop_receive(success: bool) {
2091
2093
route_params : None ,
2092
2094
} ;
2093
2095
2096
+ // We need the session priv to construct an invalid onion packet later.
2097
+ let override_random_bytes = [ 3 ; 32 ] ;
2098
+ * nodes[ 0 ] . keys_manager . override_random_bytes . lock ( ) . unwrap ( ) = Some ( override_random_bytes) ;
2099
+
2094
2100
nodes[ 0 ] . node . send_payment_with_route ( route. clone ( ) , payment_hash, RecipientOnionFields :: spontaneous_empty ( ) , PaymentId ( payment_hash. 0 ) ) . unwrap ( ) ;
2095
2101
2096
2102
check_added_monitors ! ( & nodes[ 0 ] , 1 ) ;
@@ -2101,8 +2107,7 @@ fn do_test_trampoline_single_hop_receive(success: bool) {
2101
2107
} else {
2102
2108
let replacement_onion = {
2103
2109
// create a substitute onion where the last Trampoline hop is a forward
2104
- let trampoline_secret_key = secret_from_hex ( "0134928f7b7ca6769080d70f16be84c812c741f545b49a34db47ce338a205799" ) ;
2105
- let prng_seed = secret_from_hex ( "fe02b4b9054302a3ddf4e1e9f7c411d644aebbd295218ab009dca94435f775a9" ) ;
2110
+ let trampoline_secret_key = SecretKey :: from_slice ( & override_random_bytes) . unwrap ( ) ;
2106
2111
let recipient_onion_fields = RecipientOnionFields :: spontaneous_empty ( ) ;
2107
2112
2108
2113
let mut blinded_tail = route. paths [ 0 ] . blinded_tail . clone ( ) . unwrap ( ) ;
@@ -2122,19 +2127,22 @@ fn do_test_trampoline_single_hop_receive(success: bool) {
2122
2127
let trampoline_packet = onion_utils:: construct_trampoline_onion_packet (
2123
2128
trampoline_payloads,
2124
2129
trampoline_onion_keys,
2125
- prng_seed . secret_bytes ( ) ,
2130
+ override_random_bytes ,
2126
2131
& payment_hash,
2127
2132
None ,
2128
2133
) . unwrap ( ) ;
2129
2134
2130
- let outer_session_priv = secret_from_hex ( "e52c20461ed7acd46c4e7b591a37610519179482887bd73bf3b94617f8f03677" ) ;
2135
+ let outer_session_priv = {
2136
+ let session_priv_hash = Sha256 :: hash ( & override_random_bytes) . to_byte_array ( ) ;
2137
+ SecretKey :: from_slice ( & session_priv_hash[ ..] ) . expect ( "You broke SHA-256!" )
2138
+ } ;
2131
2139
2132
2140
let ( outer_payloads, _, _) = onion_utils:: build_onion_payloads ( & route. paths [ 0 ] , outer_total_msat, & recipient_onion_fields, outer_starting_htlc_offset, & None , None , Some ( trampoline_packet) ) . unwrap ( ) ;
2133
2141
let outer_onion_keys = onion_utils:: construct_onion_keys ( & secp_ctx, & route. clone ( ) . paths [ 0 ] , & outer_session_priv) ;
2134
2142
let outer_packet = onion_utils:: construct_onion_packet (
2135
2143
outer_payloads,
2136
2144
outer_onion_keys,
2137
- prng_seed . secret_bytes ( ) ,
2145
+ override_random_bytes ,
2138
2146
& payment_hash,
2139
2147
) . unwrap ( ) ;
2140
2148
0 commit comments