Builder for creating static invoices from offers
[rust-lightning] / lightning / src / offers / invoice.rs
index 7fdbab3379c3e59f6bc0f671f168a4b2b1ec76b1..6695121b34631b32b0b9f5e9fa3cd7b138134461 100644 (file)
@@ -200,7 +200,7 @@ pub struct ExplicitSigningPubkey {}
 /// [`Bolt12Invoice::signing_pubkey`] was derived.
 ///
 /// This is not exported to bindings users as builder patterns don't map outside of move semantics.
-pub struct DerivedSigningPubkey(Keypair);
+pub struct DerivedSigningPubkey(pub(super) Keypair);
 
 impl SigningPubkeyStrategy for ExplicitSigningPubkey {}
 impl SigningPubkeyStrategy for DerivedSigningPubkey {}
@@ -383,35 +383,35 @@ impl<'a> InvoiceBuilder<'a, DerivedSigningPubkey> {
 
 impl<'a, S: SigningPubkeyStrategy> InvoiceBuilder<'a, S> {
        invoice_builder_methods!(self, Self, Self, self, S, mut);
-       invoice_builder_methods_common!(self, Self, self.invoice.fields_mut(), Self, self, S, mut);
+       invoice_builder_methods_common!(self, Self, self.invoice.fields_mut(), Self, self, S, Bolt12Invoice, mut);
 }
 
 #[cfg(all(c_bindings, not(test)))]
 impl<'a> InvoiceWithExplicitSigningPubkeyBuilder<'a> {
        invoice_explicit_signing_pubkey_builder_methods!(self, &mut Self);
        invoice_builder_methods!(self, &mut Self, (), (), ExplicitSigningPubkey);
-       invoice_builder_methods_common!(self, &mut Self, self.invoice.fields_mut(), (), (), ExplicitSigningPubkey);
+       invoice_builder_methods_common!(self, &mut Self, self.invoice.fields_mut(), (), (), ExplicitSigningPubkey, Bolt12Invoice);
 }
 
 #[cfg(all(c_bindings, test))]
 impl<'a> InvoiceWithExplicitSigningPubkeyBuilder<'a> {
        invoice_explicit_signing_pubkey_builder_methods!(self, &mut Self);
        invoice_builder_methods!(self, &mut Self, &mut Self, self, ExplicitSigningPubkey);
-       invoice_builder_methods_common!(self, &mut Self, self.invoice.fields_mut(), &mut Self, self, ExplicitSigningPubkey);
+       invoice_builder_methods_common!(self, &mut Self, self.invoice.fields_mut(), &mut Self, self, ExplicitSigningPubkey, Bolt12Invoice);
 }
 
 #[cfg(all(c_bindings, not(test)))]
 impl<'a> InvoiceWithDerivedSigningPubkeyBuilder<'a> {
        invoice_derived_signing_pubkey_builder_methods!(self, &mut Self);
        invoice_builder_methods!(self, &mut Self, (), (), DerivedSigningPubkey);
-       invoice_builder_methods_common!(self, &mut Self, self.invoice.fields_mut(), (), (), DerivedSigningPubkey);
+       invoice_builder_methods_common!(self, &mut Self, self.invoice.fields_mut(), (), (), DerivedSigningPubkey, Bolt12Invoice);
 }
 
 #[cfg(all(c_bindings, test))]
 impl<'a> InvoiceWithDerivedSigningPubkeyBuilder<'a> {
        invoice_derived_signing_pubkey_builder_methods!(self, &mut Self);
        invoice_builder_methods!(self, &mut Self, &mut Self, self, DerivedSigningPubkey);
-       invoice_builder_methods_common!(self, &mut Self, self.invoice.fields_mut(), &mut Self, self, DerivedSigningPubkey);
+       invoice_builder_methods_common!(self, &mut Self, self.invoice.fields_mut(), &mut Self, self, DerivedSigningPubkey, Bolt12Invoice);
 }
 
 #[cfg(c_bindings)]
@@ -752,12 +752,12 @@ macro_rules! invoice_accessors { ($self: ident, $contents: expr) => {
 } }
 
 impl UnsignedBolt12Invoice {
-       invoice_accessors_common!(self, self.contents);
+       invoice_accessors_common!(self, self.contents, Bolt12Invoice);
        invoice_accessors!(self, self.contents);
 }
 
 impl Bolt12Invoice {
-       invoice_accessors_common!(self, self.contents);
+       invoice_accessors_common!(self, self.contents, Bolt12Invoice);
        invoice_accessors!(self, self.contents);
 
        /// Signature of the invoice verified using [`Bolt12Invoice::signing_pubkey`].
@@ -958,14 +958,7 @@ impl InvoiceContents {
 
        #[cfg(feature = "std")]
        fn is_expired(&self) -> bool {
-               let absolute_expiry = self.created_at().checked_add(self.relative_expiry());
-               match absolute_expiry {
-                       Some(seconds_from_epoch) => match SystemTime::UNIX_EPOCH.elapsed() {
-                               Ok(elapsed) => elapsed > seconds_from_epoch,
-                               Err(_) => false,
-                       },
-                       None => false,
-               }
+               is_expired(self.created_at(), self.relative_expiry())
        }
 
        fn payment_hash(&self) -> PaymentHash {
@@ -977,36 +970,9 @@ impl InvoiceContents {
        }
 
        fn fallbacks(&self) -> Vec<Address> {
-               let chain = self.chain();
-               let network = if chain == ChainHash::using_genesis_block(Network::Bitcoin) {
-                       Network::Bitcoin
-               } else if chain == ChainHash::using_genesis_block(Network::Testnet) {
-                       Network::Testnet
-               } else if chain == ChainHash::using_genesis_block(Network::Signet) {
-                       Network::Signet
-               } else if chain == ChainHash::using_genesis_block(Network::Regtest) {
-                       Network::Regtest
-               } else {
-                       return Vec::new()
-               };
-
-               let to_valid_address = |address: &FallbackAddress| {
-                       let version = match WitnessVersion::try_from(address.version) {
-                               Ok(version) => version,
-                               Err(_) => return None,
-                       };
-
-                       let program = address.program.clone();
-                       let witness_program = match WitnessProgram::new(version, program) {
-                               Ok(witness_program) => witness_program,
-                               Err(_) => return None,
-                       };
-                       Some(Address::new(network, Payload::WitnessProgram(witness_program)))
-               };
-
                self.fields().fallbacks
                        .as_ref()
-                       .map(|fallbacks| fallbacks.iter().filter_map(to_valid_address).collect())
+                       .map(|fallbacks| filter_fallbacks(self.chain(), fallbacks))
                        .unwrap_or_else(Vec::new)
        }
 
@@ -1075,6 +1041,50 @@ impl InvoiceContents {
        }
 }
 
+#[cfg(feature = "std")]
+pub(super) fn is_expired(created_at: Duration, relative_expiry: Duration) -> bool {
+       let absolute_expiry = created_at.checked_add(relative_expiry);
+       match absolute_expiry {
+               Some(seconds_from_epoch) => match SystemTime::UNIX_EPOCH.elapsed() {
+                       Ok(elapsed) => elapsed > seconds_from_epoch,
+                       Err(_) => false,
+               },
+               None => false,
+       }
+}
+
+pub(super) fn filter_fallbacks(
+       chain: ChainHash, fallbacks: &Vec<FallbackAddress>
+) -> Vec<Address> {
+       let network = if chain == ChainHash::using_genesis_block(Network::Bitcoin) {
+               Network::Bitcoin
+       } else if chain == ChainHash::using_genesis_block(Network::Testnet) {
+               Network::Testnet
+       } else if chain == ChainHash::using_genesis_block(Network::Signet) {
+               Network::Signet
+       } else if chain == ChainHash::using_genesis_block(Network::Regtest) {
+               Network::Regtest
+       } else {
+               return Vec::new()
+       };
+
+       let to_valid_address = |address: &FallbackAddress| {
+               let version = match WitnessVersion::try_from(address.version) {
+                       Ok(version) => version,
+                       Err(_) => return None,
+               };
+
+               let program = address.program.clone();
+               let witness_program = match WitnessProgram::new(version, program) {
+                       Ok(witness_program) => witness_program,
+                       Err(_) => return None,
+               };
+               Some(Address::new(network, Payload::WitnessProgram(witness_program)))
+       };
+
+       fallbacks.iter().filter_map(to_valid_address).collect()
+}
+
 impl InvoiceFields {
        fn as_tlv_stream(&self) -> InvoiceTlvStreamRef {
                let features = {
@@ -1092,6 +1102,7 @@ impl InvoiceFields {
                        fallbacks: self.fallbacks.as_ref(),
                        features,
                        node_id: Some(&self.signing_pubkey),
+                       message_paths: None,
                }
        }
 }
@@ -1152,14 +1163,16 @@ tlv_stream!(InvoiceTlvStream, InvoiceTlvStreamRef, 160..240, {
        (172, fallbacks: (Vec<FallbackAddress>, WithoutLength)),
        (174, features: (Bolt12InvoiceFeatures, WithoutLength)),
        (176, node_id: PublicKey),
+       // Only present in `StaticInvoice`s.
+       (238, message_paths: (Vec<BlindedPath>, WithoutLength)),
 });
 
-type BlindedPathIter<'a> = core::iter::Map<
+pub(super) type BlindedPathIter<'a> = core::iter::Map<
        core::slice::Iter<'a, (BlindedPayInfo, BlindedPath)>,
        for<'r> fn(&'r (BlindedPayInfo, BlindedPath)) -> &'r BlindedPath,
 >;
 
-type BlindedPayInfoIter<'a> = core::iter::Map<
+pub(super) type BlindedPayInfoIter<'a> = core::iter::Map<
        core::slice::Iter<'a, (BlindedPayInfo, BlindedPath)>,
        for<'r> fn(&'r (BlindedPayInfo, BlindedPath)) -> &'r BlindedPayInfo,
 >;
@@ -1205,8 +1218,8 @@ impl_writeable!(BlindedPayInfo, {
 /// Wire representation for an on-chain fallback address.
 #[derive(Clone, Debug, PartialEq)]
 pub(super) struct FallbackAddress {
-       version: u8,
-       program: Vec<u8>,
+       pub(super) version: u8,
+       pub(super) program: Vec<u8>,
 }
 
 impl_writeable!(FallbackAddress, { version, program });
@@ -1268,10 +1281,9 @@ impl TryFrom<ParsedMessage<FullInvoiceTlvStream>> for Bolt12Invoice {
                        (payer_tlv_stream, offer_tlv_stream, invoice_request_tlv_stream, invoice_tlv_stream)
                )?;
 
-               let signature = match signature {
-                       None => return Err(Bolt12ParseError::InvalidSemantics(Bolt12SemanticError::MissingSignature)),
-                       Some(signature) => signature,
-               };
+               let signature = signature.ok_or(
+                       Bolt12ParseError::InvalidSemantics(Bolt12SemanticError::MissingSignature)
+               )?;
                let tagged_hash = TaggedHash::from_valid_tlv_stream_bytes(SIGNATURE_TAG, &bytes);
                let pubkey = contents.fields().signing_pubkey;
                merkle::verify_signature(&signature, &tagged_hash, pubkey)?;
@@ -1290,21 +1302,13 @@ impl TryFrom<PartialInvoiceTlvStream> for InvoiceContents {
                        invoice_request_tlv_stream,
                        InvoiceTlvStream {
                                paths, blindedpay, created_at, relative_expiry, payment_hash, amount, fallbacks,
-                               features, node_id,
+                               features, node_id, message_paths,
                        },
                ) = tlv_stream;
 
-               let payment_paths = match (blindedpay, paths) {
-                       (_, None) => return Err(Bolt12SemanticError::MissingPaths),
-                       (None, _) => return Err(Bolt12SemanticError::InvalidPayInfo),
-                       (_, Some(paths)) if paths.is_empty() => return Err(Bolt12SemanticError::MissingPaths),
-                       (Some(blindedpay), Some(paths)) if paths.len() != blindedpay.len() => {
-                               return Err(Bolt12SemanticError::InvalidPayInfo);
-                       },
-                       (Some(blindedpay), Some(paths)) => {
-                               blindedpay.into_iter().zip(paths.into_iter()).collect::<Vec<_>>()
-                       },
-               };
+               if message_paths.is_some() { return Err(Bolt12SemanticError::UnexpectedPaths) }
+
+               let payment_paths = construct_payment_paths(blindedpay, paths)?;
 
                let created_at = match created_at {
                        None => return Err(Bolt12SemanticError::MissingCreationTime),
@@ -1315,63 +1319,74 @@ impl TryFrom<PartialInvoiceTlvStream> for InvoiceContents {
                        .map(Into::<u64>::into)
                        .map(Duration::from_secs);
 
-               let payment_hash = match payment_hash {
-                       None => return Err(Bolt12SemanticError::MissingPaymentHash),
-                       Some(payment_hash) => payment_hash,
-               };
+               let payment_hash = payment_hash.ok_or(Bolt12SemanticError::MissingPaymentHash)?;
 
-               let amount_msats = match amount {
-                       None => return Err(Bolt12SemanticError::MissingAmount),
-                       Some(amount) => amount,
-               };
+               let amount_msats = amount.ok_or(Bolt12SemanticError::MissingAmount)?;
 
                let features = features.unwrap_or_else(Bolt12InvoiceFeatures::empty);
 
-               let signing_pubkey = match node_id {
-                       None => return Err(Bolt12SemanticError::MissingSigningPubkey),
-                       Some(node_id) => node_id,
-               };
+               let signing_pubkey = node_id.ok_or(Bolt12SemanticError::MissingSigningPubkey)?;
 
                let fields = InvoiceFields {
                        payment_paths, created_at, relative_expiry, payment_hash, amount_msats, fallbacks,
                        features, signing_pubkey,
                };
 
-               match (offer_tlv_stream.node_id, &offer_tlv_stream.paths) {
-                       (Some(expected_signing_pubkey), _) => {
-                               if fields.signing_pubkey != expected_signing_pubkey {
-                                       return Err(Bolt12SemanticError::InvalidSigningPubkey);
-                               }
+               check_invoice_signing_pubkey(&fields.signing_pubkey, &offer_tlv_stream)?;
 
-                               let invoice_request = InvoiceRequestContents::try_from(
-                                       (payer_tlv_stream, offer_tlv_stream, invoice_request_tlv_stream)
-                               )?;
-                               Ok(InvoiceContents::ForOffer { invoice_request, fields })
-                       },
-                       (None, Some(paths)) => {
-                               if !paths
-                                       .iter()
-                                       .filter_map(|path| path.blinded_hops.last())
-                                       .any(|last_hop| fields.signing_pubkey == last_hop.blinded_node_id)
-                               {
-                                       return Err(Bolt12SemanticError::InvalidSigningPubkey);
-                               }
-
-                               let invoice_request = InvoiceRequestContents::try_from(
-                                       (payer_tlv_stream, offer_tlv_stream, invoice_request_tlv_stream)
-                               )?;
-                               Ok(InvoiceContents::ForOffer { invoice_request, fields })
-                       },
-                       (None, None) => {
-                               let refund = RefundContents::try_from(
-                                       (payer_tlv_stream, offer_tlv_stream, invoice_request_tlv_stream)
-                               )?;
-                               Ok(InvoiceContents::ForRefund { refund, fields })
-                       },
+               if offer_tlv_stream.node_id.is_none() && offer_tlv_stream.paths.is_none() {
+                       let refund = RefundContents::try_from(
+                               (payer_tlv_stream, offer_tlv_stream, invoice_request_tlv_stream)
+                       )?;
+                       Ok(InvoiceContents::ForRefund { refund, fields })
+               } else {
+                       let invoice_request = InvoiceRequestContents::try_from(
+                               (payer_tlv_stream, offer_tlv_stream, invoice_request_tlv_stream)
+                       )?;
+                       Ok(InvoiceContents::ForOffer { invoice_request, fields })
                }
        }
 }
 
+pub(super) fn construct_payment_paths(
+       blinded_payinfos: Option<Vec<BlindedPayInfo>>, blinded_paths: Option<Vec<BlindedPath>>
+) -> Result<Vec<(BlindedPayInfo, BlindedPath)>, Bolt12SemanticError> {
+       match (blinded_payinfos, blinded_paths) {
+               (_, None) => Err(Bolt12SemanticError::MissingPaths),
+               (None, _) => Err(Bolt12SemanticError::InvalidPayInfo),
+               (_, Some(paths)) if paths.is_empty() => Err(Bolt12SemanticError::MissingPaths),
+               (Some(blindedpay), Some(paths)) if paths.len() != blindedpay.len() => {
+                       Err(Bolt12SemanticError::InvalidPayInfo)
+               },
+               (Some(blindedpay), Some(paths)) => {
+                       Ok(blindedpay.into_iter().zip(paths.into_iter()).collect::<Vec<_>>())
+               },
+       }
+}
+
+pub(super) fn check_invoice_signing_pubkey(
+       invoice_signing_pubkey: &PublicKey, offer_tlv_stream: &OfferTlvStream
+) -> Result<(), Bolt12SemanticError> {
+       match (&offer_tlv_stream.node_id, &offer_tlv_stream.paths) {
+               (Some(expected_signing_pubkey), _) => {
+                       if invoice_signing_pubkey != expected_signing_pubkey {
+                               return Err(Bolt12SemanticError::InvalidSigningPubkey);
+                       }
+               },
+               (None, Some(paths)) => {
+                       if !paths
+                               .iter()
+                               .filter_map(|path| path.blinded_hops.last())
+                               .any(|last_hop| invoice_signing_pubkey == &last_hop.blinded_node_id)
+                       {
+                               return Err(Bolt12SemanticError::InvalidSigningPubkey);
+                       }
+               },
+               _ => {},
+       }
+       Ok(())
+}
+
 #[cfg(test)]
 mod tests {
        use super::{Bolt12Invoice, DEFAULT_RELATIVE_EXPIRY, FallbackAddress, FullInvoiceTlvStreamRef, InvoiceTlvStreamRef, SIGNATURE_TAG, UnsignedBolt12Invoice};
@@ -1558,6 +1573,7 @@ mod tests {
                                        fallbacks: None,
                                        features: None,
                                        node_id: Some(&recipient_pubkey()),
+                                       message_paths: None,
                                },
                                SignatureTlvStreamRef { signature: Some(&invoice.signature()) },
                        ),
@@ -1649,6 +1665,7 @@ mod tests {
                                        fallbacks: None,
                                        features: None,
                                        node_id: Some(&recipient_pubkey()),
+                                       message_paths: None,
                                },
                                SignatureTlvStreamRef { signature: Some(&invoice.signature()) },
                        ),
@@ -2419,4 +2436,35 @@ mod tests {
                        Err(e) => assert_eq!(e, Bolt12ParseError::Decode(DecodeError::InvalidValue)),
                }
        }
+
+       #[test]
+       fn fails_parsing_invoice_with_message_paths() {
+               let invoice = OfferBuilder::new(recipient_pubkey())
+                       .amount_msats(1000)
+                       .build().unwrap()
+                       .request_invoice(vec![1; 32], payer_pubkey()).unwrap()
+                       .build().unwrap()
+                       .sign(payer_sign).unwrap()
+                       .respond_with_no_std(payment_paths(), payment_hash(), now()).unwrap()
+                       .build().unwrap()
+                       .sign(recipient_sign).unwrap();
+
+               let blinded_path = BlindedPath {
+                       introduction_node: IntroductionNode::NodeId(pubkey(40)),
+                       blinding_point: pubkey(41),
+                       blinded_hops: vec![
+                               BlindedHop { blinded_node_id: pubkey(42), encrypted_payload: vec![0; 43] },
+                               BlindedHop { blinded_node_id: pubkey(43), encrypted_payload: vec![0; 44] },
+                       ],
+               };
+
+               let mut tlv_stream = invoice.as_tlv_stream();
+               let message_paths = vec![blinded_path];
+               tlv_stream.3.message_paths = Some(&message_paths);
+
+               match Bolt12Invoice::try_from(tlv_stream.to_bytes()) {
+                       Ok(_) => panic!("expected error"),
+                       Err(e) => assert_eq!(e, Bolt12ParseError::InvalidSemantics(Bolt12SemanticError::UnexpectedPaths)),
+               }
+       }
 }