@@ -64,6 +64,7 @@ use crate::sign::EntropySource;
6464use crate :: io;
6565use crate :: blinded_path:: BlindedPath ;
6666use crate :: ln:: PaymentHash ;
67+ use crate :: ln:: channelmanager:: PaymentId ;
6768use crate :: ln:: features:: InvoiceRequestFeatures ;
6869use crate :: ln:: inbound_payment:: { ExpandedKey , IV_LEN , Nonce } ;
6970use crate :: ln:: msgs:: DecodeError ;
@@ -128,10 +129,12 @@ impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, ExplicitPayerI
128129 }
129130
130131 pub ( super ) fn deriving_metadata < ES : Deref > (
131- offer : & ' a Offer , payer_id : PublicKey , expanded_key : & ExpandedKey , entropy_source : ES
132+ offer : & ' a Offer , payer_id : PublicKey , expanded_key : & ExpandedKey , entropy_source : ES ,
133+ payment_id : PaymentId ,
132134 ) -> Self where ES :: Target : EntropySource {
133135 let nonce = Nonce :: from_entropy_source ( entropy_source) ;
134- let derivation_material = MetadataMaterial :: new ( nonce, expanded_key, IV_BYTES ) ;
136+ let payment_id = Some ( payment_id) ;
137+ let derivation_material = MetadataMaterial :: new ( nonce, expanded_key, IV_BYTES , payment_id) ;
135138 let metadata = Metadata :: Derived ( derivation_material) ;
136139 Self {
137140 offer,
@@ -145,10 +148,12 @@ impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, ExplicitPayerI
145148
146149impl < ' a , ' b , T : secp256k1:: Signing > InvoiceRequestBuilder < ' a , ' b , DerivedPayerId , T > {
147150 pub ( super ) fn deriving_payer_id < ES : Deref > (
148- offer : & ' a Offer , expanded_key : & ExpandedKey , entropy_source : ES , secp_ctx : & ' b Secp256k1 < T >
151+ offer : & ' a Offer , expanded_key : & ExpandedKey , entropy_source : ES ,
152+ secp_ctx : & ' b Secp256k1 < T > , payment_id : PaymentId
149153 ) -> Self where ES :: Target : EntropySource {
150154 let nonce = Nonce :: from_entropy_source ( entropy_source) ;
151- let derivation_material = MetadataMaterial :: new ( nonce, expanded_key, IV_BYTES ) ;
155+ let payment_id = Some ( payment_id) ;
156+ let derivation_material = MetadataMaterial :: new ( nonce, expanded_key, IV_BYTES , payment_id) ;
152157 let metadata = Metadata :: DerivedSigningPubkey ( derivation_material) ;
153158 Self {
154159 offer,
@@ -259,7 +264,7 @@ impl<'a, 'b, P: PayerIdStrategy, T: secp256k1::Signing> InvoiceRequestBuilder<'a
259264 let mut tlv_stream = self . invoice_request . as_tlv_stream ( ) ;
260265 debug_assert ! ( tlv_stream. 2 . payer_id. is_none( ) ) ;
261266 tlv_stream. 0 . metadata = None ;
262- if !metadata. derives_keys ( ) {
267+ if !metadata. derives_payer_keys ( ) {
263268 tlv_stream. 2 . payer_id = self . payer_id . as_ref ( ) ;
264269 }
265270
@@ -680,7 +685,7 @@ impl InvoiceRequestContents {
680685 }
681686
682687 pub ( super ) fn derives_keys ( & self ) -> bool {
683- self . inner . payer . 0 . derives_keys ( )
688+ self . inner . payer . 0 . derives_payer_keys ( )
684689 }
685690
686691 pub ( super ) fn chain ( & self ) -> ChainHash {
@@ -913,6 +918,7 @@ mod tests {
913918 #[ cfg( feature = "std" ) ]
914919 use core:: time:: Duration ;
915920 use crate :: sign:: KeyMaterial ;
921+ use crate :: ln:: channelmanager:: PaymentId ;
916922 use crate :: ln:: features:: { InvoiceRequestFeatures , OfferFeatures } ;
917923 use crate :: ln:: inbound_payment:: ExpandedKey ;
918924 use crate :: ln:: msgs:: { DecodeError , MAX_VALUE_MSAT } ;
@@ -1058,12 +1064,13 @@ mod tests {
10581064 let expanded_key = ExpandedKey :: new ( & KeyMaterial ( [ 42 ; 32 ] ) ) ;
10591065 let entropy = FixedEntropy { } ;
10601066 let secp_ctx = Secp256k1 :: new ( ) ;
1067+ let payment_id = PaymentId ( [ 1 ; 32 ] ) ;
10611068
10621069 let offer = OfferBuilder :: new ( "foo" . into ( ) , recipient_pubkey ( ) )
10631070 . amount_msats ( 1000 )
10641071 . build ( ) . unwrap ( ) ;
10651072 let invoice_request = offer
1066- . request_invoice_deriving_metadata ( payer_id, & expanded_key, & entropy)
1073+ . request_invoice_deriving_metadata ( payer_id, & expanded_key, & entropy, payment_id )
10671074 . unwrap ( )
10681075 . build ( ) . unwrap ( )
10691076 . sign ( payer_sign) . unwrap ( ) ;
@@ -1073,7 +1080,10 @@ mod tests {
10731080 . unwrap ( )
10741081 . build ( ) . unwrap ( )
10751082 . sign ( recipient_sign) . unwrap ( ) ;
1076- assert ! ( invoice. verify( & expanded_key, & secp_ctx) ) ;
1083+ match invoice. verify ( & expanded_key, & secp_ctx) {
1084+ Ok ( payment_id) => assert_eq ! ( payment_id, PaymentId ( [ 1 ; 32 ] ) ) ,
1085+ Err ( ( ) ) => panic ! ( "verification failed" ) ,
1086+ }
10771087
10781088 // Fails verification with altered fields
10791089 let (
@@ -1096,7 +1106,7 @@ mod tests {
10961106 signature_tlv_stream. write ( & mut encoded_invoice) . unwrap ( ) ;
10971107
10981108 let invoice = Bolt12Invoice :: try_from ( encoded_invoice) . unwrap ( ) ;
1099- assert ! ( ! invoice. verify( & expanded_key, & secp_ctx) ) ;
1109+ assert ! ( invoice. verify( & expanded_key, & secp_ctx) . is_err ( ) ) ;
11001110
11011111 // Fails verification with altered metadata
11021112 let (
@@ -1119,20 +1129,21 @@ mod tests {
11191129 signature_tlv_stream. write ( & mut encoded_invoice) . unwrap ( ) ;
11201130
11211131 let invoice = Bolt12Invoice :: try_from ( encoded_invoice) . unwrap ( ) ;
1122- assert ! ( ! invoice. verify( & expanded_key, & secp_ctx) ) ;
1132+ assert ! ( invoice. verify( & expanded_key, & secp_ctx) . is_err ( ) ) ;
11231133 }
11241134
11251135 #[ test]
11261136 fn builds_invoice_request_with_derived_payer_id ( ) {
11271137 let expanded_key = ExpandedKey :: new ( & KeyMaterial ( [ 42 ; 32 ] ) ) ;
11281138 let entropy = FixedEntropy { } ;
11291139 let secp_ctx = Secp256k1 :: new ( ) ;
1140+ let payment_id = PaymentId ( [ 1 ; 32 ] ) ;
11301141
11311142 let offer = OfferBuilder :: new ( "foo" . into ( ) , recipient_pubkey ( ) )
11321143 . amount_msats ( 1000 )
11331144 . build ( ) . unwrap ( ) ;
11341145 let invoice_request = offer
1135- . request_invoice_deriving_payer_id ( & expanded_key, & entropy, & secp_ctx)
1146+ . request_invoice_deriving_payer_id ( & expanded_key, & entropy, & secp_ctx, payment_id )
11361147 . unwrap ( )
11371148 . build_and_sign ( )
11381149 . unwrap ( ) ;
@@ -1141,7 +1152,10 @@ mod tests {
11411152 . unwrap ( )
11421153 . build ( ) . unwrap ( )
11431154 . sign ( recipient_sign) . unwrap ( ) ;
1144- assert ! ( invoice. verify( & expanded_key, & secp_ctx) ) ;
1155+ match invoice. verify ( & expanded_key, & secp_ctx) {
1156+ Ok ( payment_id) => assert_eq ! ( payment_id, PaymentId ( [ 1 ; 32 ] ) ) ,
1157+ Err ( ( ) ) => panic ! ( "verification failed" ) ,
1158+ }
11451159
11461160 // Fails verification with altered fields
11471161 let (
@@ -1164,7 +1178,7 @@ mod tests {
11641178 signature_tlv_stream. write ( & mut encoded_invoice) . unwrap ( ) ;
11651179
11661180 let invoice = Bolt12Invoice :: try_from ( encoded_invoice) . unwrap ( ) ;
1167- assert ! ( ! invoice. verify( & expanded_key, & secp_ctx) ) ;
1181+ assert ! ( invoice. verify( & expanded_key, & secp_ctx) . is_err ( ) ) ;
11681182
11691183 // Fails verification with altered payer id
11701184 let (
@@ -1187,7 +1201,7 @@ mod tests {
11871201 signature_tlv_stream. write ( & mut encoded_invoice) . unwrap ( ) ;
11881202
11891203 let invoice = Bolt12Invoice :: try_from ( encoded_invoice) . unwrap ( ) ;
1190- assert ! ( ! invoice. verify( & expanded_key, & secp_ctx) ) ;
1204+ assert ! ( invoice. verify( & expanded_key, & secp_ctx) . is_err ( ) ) ;
11911205 }
11921206
11931207 #[ test]
0 commit comments