Macro-ize InvoiceRequestBuilder

InvoiceRequestBuilder is not exported to bindings because it has methods
that take `self` by value and are only implemented for certain type
parameterizations. Define these methods using macros such that different
builders and related methods can be defined for c_bindings.
This commit is contained in:
Jeffrey Czyz 2024-02-22 16:47:38 -06:00
parent 702a003270
commit c7b877efdd
No known key found for this signature in database
GPG key ID: 912EF12EA67705F5
2 changed files with 124 additions and 92 deletions

View file

@ -117,7 +117,7 @@ pub struct DerivedPayerId {}
impl PayerIdStrategy for ExplicitPayerId {} impl PayerIdStrategy for ExplicitPayerId {}
impl PayerIdStrategy for DerivedPayerId {} impl PayerIdStrategy for DerivedPayerId {}
impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, ExplicitPayerId, T> { macro_rules! invoice_request_explicit_payer_id_builder_methods { ($self: ident, $self_type: ty) => {
pub(super) fn new(offer: &'a Offer, metadata: Vec<u8>, payer_id: PublicKey) -> Self { pub(super) fn new(offer: &'a Offer, metadata: Vec<u8>, payer_id: PublicKey) -> Self {
Self { Self {
offer, offer,
@ -147,14 +147,14 @@ impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, ExplicitPayerI
/// Builds an unsigned [`InvoiceRequest`] after checking for valid semantics. It can be signed /// Builds an unsigned [`InvoiceRequest`] after checking for valid semantics. It can be signed
/// by [`UnsignedInvoiceRequest::sign`]. /// by [`UnsignedInvoiceRequest::sign`].
pub fn build(self) -> Result<UnsignedInvoiceRequest, Bolt12SemanticError> { pub fn build($self: $self_type) -> Result<UnsignedInvoiceRequest, Bolt12SemanticError> {
let (unsigned_invoice_request, keys, _) = self.build_with_checks()?; let (unsigned_invoice_request, keys, _) = $self.build_with_checks()?;
debug_assert!(keys.is_none()); debug_assert!(keys.is_none());
Ok(unsigned_invoice_request) Ok(unsigned_invoice_request)
} }
} } }
impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, DerivedPayerId, T> { macro_rules! invoice_request_derived_payer_id_builder_methods { ($self: ident, $self_type: ty) => {
pub(super) fn deriving_payer_id<ES: Deref>( pub(super) fn deriving_payer_id<ES: Deref>(
offer: &'a Offer, expanded_key: &ExpandedKey, entropy_source: ES, offer: &'a Offer, expanded_key: &ExpandedKey, entropy_source: ES,
secp_ctx: &'b Secp256k1<T>, payment_id: PaymentId secp_ctx: &'b Secp256k1<T>, payment_id: PaymentId
@ -173,8 +173,8 @@ impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, DerivedPayerId
} }
/// Builds a signed [`InvoiceRequest`] after checking for valid semantics. /// Builds a signed [`InvoiceRequest`] after checking for valid semantics.
pub fn build_and_sign(self) -> Result<InvoiceRequest, Bolt12SemanticError> { pub fn build_and_sign($self: $self_type) -> Result<InvoiceRequest, Bolt12SemanticError> {
let (unsigned_invoice_request, keys, secp_ctx) = self.build_with_checks()?; let (unsigned_invoice_request, keys, secp_ctx) = $self.build_with_checks()?;
debug_assert!(keys.is_some()); debug_assert!(keys.is_some());
let secp_ctx = secp_ctx.unwrap(); let secp_ctx = secp_ctx.unwrap();
@ -186,9 +186,11 @@ impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, DerivedPayerId
.unwrap(); .unwrap();
Ok(invoice_request) Ok(invoice_request)
} }
} } }
impl<'a, 'b, P: PayerIdStrategy, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, P, T> { macro_rules! invoice_request_builder_methods { (
$self: ident, $self_type: ty, $return_type: ty, $return_value: expr
) => {
fn create_contents(offer: &Offer, metadata: Metadata) -> InvoiceRequestContentsWithoutPayerId { fn create_contents(offer: &Offer, metadata: Metadata) -> InvoiceRequestContentsWithoutPayerId {
let offer = offer.contents.clone(); let offer = offer.contents.clone();
InvoiceRequestContentsWithoutPayerId { InvoiceRequestContentsWithoutPayerId {
@ -202,8 +204,8 @@ impl<'a, 'b, P: PayerIdStrategy, T: secp256k1::Signing> InvoiceRequestBuilder<'a
/// by the offer. /// by the offer.
/// ///
/// Successive calls to this method will override the previous setting. /// Successive calls to this method will override the previous setting.
pub fn chain(self, network: Network) -> Result<Self, Bolt12SemanticError> { pub fn chain($self: $self_type, network: Network) -> Result<$return_type, Bolt12SemanticError> {
self.chain_hash(ChainHash::using_genesis_block(network)) $self.chain_hash(ChainHash::using_genesis_block(network))
} }
/// Sets the [`InvoiceRequest::chain`] for paying an invoice. If not called, the chain hash of /// Sets the [`InvoiceRequest::chain`] for paying an invoice. If not called, the chain hash of
@ -211,13 +213,13 @@ impl<'a, 'b, P: PayerIdStrategy, T: secp256k1::Signing> InvoiceRequestBuilder<'a
/// offer. /// offer.
/// ///
/// Successive calls to this method will override the previous setting. /// Successive calls to this method will override the previous setting.
pub(crate) fn chain_hash(mut self, chain: ChainHash) -> Result<Self, Bolt12SemanticError> { pub(crate) fn chain_hash(mut $self: $self_type, chain: ChainHash) -> Result<$return_type, Bolt12SemanticError> {
if !self.offer.supports_chain(chain) { if !$self.offer.supports_chain(chain) {
return Err(Bolt12SemanticError::UnsupportedChain); return Err(Bolt12SemanticError::UnsupportedChain);
} }
self.invoice_request.chain = Some(chain); $self.invoice_request.chain = Some(chain);
Ok(self) Ok($return_value)
} }
/// Sets the [`InvoiceRequest::amount_msats`] for paying an invoice. Errors if `amount_msats` is /// Sets the [`InvoiceRequest::amount_msats`] for paying an invoice. Errors if `amount_msats` is
@ -226,130 +228,147 @@ impl<'a, 'b, P: PayerIdStrategy, T: secp256k1::Signing> InvoiceRequestBuilder<'a
/// Successive calls to this method will override the previous setting. /// Successive calls to this method will override the previous setting.
/// ///
/// [`quantity`]: Self::quantity /// [`quantity`]: Self::quantity
pub fn amount_msats(mut self, amount_msats: u64) -> Result<Self, Bolt12SemanticError> { pub fn amount_msats(mut $self: $self_type, amount_msats: u64) -> Result<$return_type, Bolt12SemanticError> {
self.invoice_request.offer.check_amount_msats_for_quantity( $self.invoice_request.offer.check_amount_msats_for_quantity(
Some(amount_msats), self.invoice_request.quantity Some(amount_msats), $self.invoice_request.quantity
)?; )?;
self.invoice_request.amount_msats = Some(amount_msats); $self.invoice_request.amount_msats = Some(amount_msats);
Ok(self) Ok($return_value)
} }
/// Sets [`InvoiceRequest::quantity`] of items. If not set, `1` is assumed. Errors if `quantity` /// Sets [`InvoiceRequest::quantity`] of items. If not set, `1` is assumed. Errors if `quantity`
/// does not conform to [`Offer::is_valid_quantity`]. /// does not conform to [`Offer::is_valid_quantity`].
/// ///
/// Successive calls to this method will override the previous setting. /// Successive calls to this method will override the previous setting.
pub fn quantity(mut self, quantity: u64) -> Result<Self, Bolt12SemanticError> { pub fn quantity(mut $self: $self_type, quantity: u64) -> Result<$return_type, Bolt12SemanticError> {
self.invoice_request.offer.check_quantity(Some(quantity))?; $self.invoice_request.offer.check_quantity(Some(quantity))?;
self.invoice_request.quantity = Some(quantity); $self.invoice_request.quantity = Some(quantity);
Ok(self) Ok($return_value)
} }
/// Sets the [`InvoiceRequest::payer_note`]. /// Sets the [`InvoiceRequest::payer_note`].
/// ///
/// Successive calls to this method will override the previous setting. /// Successive calls to this method will override the previous setting.
pub fn payer_note(mut self, payer_note: String) -> Self { pub fn payer_note(mut $self: $self_type, payer_note: String) -> $return_type {
self.invoice_request.payer_note = Some(payer_note); $self.invoice_request.payer_note = Some(payer_note);
self $return_value
} }
fn build_with_checks(mut self) -> Result< fn build_with_checks(mut $self: $self_type) -> Result<
(UnsignedInvoiceRequest, Option<KeyPair>, Option<&'b Secp256k1<T>>), (UnsignedInvoiceRequest, Option<KeyPair>, Option<&'b Secp256k1<T>>),
Bolt12SemanticError Bolt12SemanticError
> { > {
#[cfg(feature = "std")] { #[cfg(feature = "std")] {
if self.offer.is_expired() { if $self.offer.is_expired() {
return Err(Bolt12SemanticError::AlreadyExpired); return Err(Bolt12SemanticError::AlreadyExpired);
} }
} }
let chain = self.invoice_request.chain(); let chain = $self.invoice_request.chain();
if !self.offer.supports_chain(chain) { if !$self.offer.supports_chain(chain) {
return Err(Bolt12SemanticError::UnsupportedChain); return Err(Bolt12SemanticError::UnsupportedChain);
} }
if chain == self.offer.implied_chain() { if chain == $self.offer.implied_chain() {
self.invoice_request.chain = None; $self.invoice_request.chain = None;
} }
if self.offer.amount().is_none() && self.invoice_request.amount_msats.is_none() { if $self.offer.amount().is_none() && $self.invoice_request.amount_msats.is_none() {
return Err(Bolt12SemanticError::MissingAmount); return Err(Bolt12SemanticError::MissingAmount);
} }
self.invoice_request.offer.check_quantity(self.invoice_request.quantity)?; $self.invoice_request.offer.check_quantity($self.invoice_request.quantity)?;
self.invoice_request.offer.check_amount_msats_for_quantity( $self.invoice_request.offer.check_amount_msats_for_quantity(
self.invoice_request.amount_msats, self.invoice_request.quantity $self.invoice_request.amount_msats, $self.invoice_request.quantity
)?; )?;
Ok(self.build_without_checks()) Ok($self.build_without_checks())
} }
fn build_without_checks(mut self) -> fn build_without_checks(mut $self: $self_type) ->
(UnsignedInvoiceRequest, Option<KeyPair>, Option<&'b Secp256k1<T>>) (UnsignedInvoiceRequest, Option<KeyPair>, Option<&'b Secp256k1<T>>)
{ {
// Create the metadata for stateless verification of a Bolt12Invoice. // Create the metadata for stateless verification of a Bolt12Invoice.
let mut keys = None; let mut keys = None;
let secp_ctx = self.secp_ctx.clone(); let secp_ctx = $self.secp_ctx.clone();
if self.invoice_request.payer.0.has_derivation_material() { if $self.invoice_request.payer.0.has_derivation_material() {
let mut metadata = core::mem::take(&mut self.invoice_request.payer.0); let mut metadata = core::mem::take(&mut $self.invoice_request.payer.0);
let mut tlv_stream = self.invoice_request.as_tlv_stream(); let mut tlv_stream = $self.invoice_request.as_tlv_stream();
debug_assert!(tlv_stream.2.payer_id.is_none()); debug_assert!(tlv_stream.2.payer_id.is_none());
tlv_stream.0.metadata = None; tlv_stream.0.metadata = None;
if !metadata.derives_payer_keys() { if !metadata.derives_payer_keys() {
tlv_stream.2.payer_id = self.payer_id.as_ref(); tlv_stream.2.payer_id = $self.payer_id.as_ref();
} }
let (derived_metadata, derived_keys) = metadata.derive_from(tlv_stream, self.secp_ctx); let (derived_metadata, derived_keys) = metadata.derive_from(tlv_stream, $self.secp_ctx);
metadata = derived_metadata; metadata = derived_metadata;
keys = derived_keys; keys = derived_keys;
if let Some(keys) = keys { if let Some(keys) = keys {
debug_assert!(self.payer_id.is_none()); debug_assert!($self.payer_id.is_none());
self.payer_id = Some(keys.public_key()); $self.payer_id = Some(keys.public_key());
} }
self.invoice_request.payer.0 = metadata; $self.invoice_request.payer.0 = metadata;
} }
debug_assert!(self.invoice_request.payer.0.as_bytes().is_some()); debug_assert!($self.invoice_request.payer.0.as_bytes().is_some());
debug_assert!(self.payer_id.is_some()); debug_assert!($self.payer_id.is_some());
let payer_id = self.payer_id.unwrap(); let payer_id = $self.payer_id.unwrap();
let invoice_request = InvoiceRequestContents { let invoice_request = InvoiceRequestContents {
inner: self.invoice_request, inner: $self.invoice_request,
payer_id, payer_id,
}; };
let unsigned_invoice_request = UnsignedInvoiceRequest::new(self.offer, invoice_request); let unsigned_invoice_request = UnsignedInvoiceRequest::new($self.offer, invoice_request);
(unsigned_invoice_request, keys, secp_ctx) (unsigned_invoice_request, keys, secp_ctx)
} }
} } }
#[cfg(test)] #[cfg(test)]
impl<'a, 'b, P: PayerIdStrategy, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, P, T> { macro_rules! invoice_request_builder_test_methods { (
fn chain_unchecked(mut self, network: Network) -> Self { $self: ident, $self_type: ty, $return_type: ty, $return_value: expr
) => {
fn chain_unchecked(mut $self: $self_type, network: Network) -> $return_type {
let chain = ChainHash::using_genesis_block(network); let chain = ChainHash::using_genesis_block(network);
self.invoice_request.chain = Some(chain); $self.invoice_request.chain = Some(chain);
self $return_value
} }
fn amount_msats_unchecked(mut self, amount_msats: u64) -> Self { fn amount_msats_unchecked(mut $self: $self_type, amount_msats: u64) -> $return_type {
self.invoice_request.amount_msats = Some(amount_msats); $self.invoice_request.amount_msats = Some(amount_msats);
self $return_value
} }
fn features_unchecked(mut self, features: InvoiceRequestFeatures) -> Self { fn features_unchecked(mut $self: $self_type, features: InvoiceRequestFeatures) -> $return_type {
self.invoice_request.features = features; $self.invoice_request.features = features;
self $return_value
} }
fn quantity_unchecked(mut self, quantity: u64) -> Self { fn quantity_unchecked(mut $self: $self_type, quantity: u64) -> $return_type {
self.invoice_request.quantity = Some(quantity); $self.invoice_request.quantity = Some(quantity);
self $return_value
} }
pub(super) fn build_unchecked(self) -> UnsignedInvoiceRequest { pub(super) fn build_unchecked($self: $self_type) -> UnsignedInvoiceRequest {
self.build_without_checks().0 $self.build_without_checks().0
} }
} }
impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, ExplicitPayerId, T> {
invoice_request_explicit_payer_id_builder_methods!(self, Self);
}
impl<'a, 'b, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, DerivedPayerId, T> {
invoice_request_derived_payer_id_builder_methods!(self, Self);
}
impl<'a, 'b, P: PayerIdStrategy, T: secp256k1::Signing> InvoiceRequestBuilder<'a, 'b, P, T> {
invoice_request_builder_methods!(self, Self, Self, self);
#[cfg(test)]
invoice_request_builder_test_methods!(self, Self, Self, self);
} }
/// A semantically valid [`InvoiceRequest`] that hasn't been signed. /// A semantically valid [`InvoiceRequest`] that hasn't been signed.
@ -385,31 +404,37 @@ impl UnsignedInvoiceRequest {
pub fn tagged_hash(&self) -> &TaggedHash { pub fn tagged_hash(&self) -> &TaggedHash {
&self.tagged_hash &self.tagged_hash
} }
}
macro_rules! unsigned_invoice_request_sign_method { ($self: ident, $self_type: ty) => {
/// Signs the [`TaggedHash`] of the invoice request using the given function. /// Signs the [`TaggedHash`] of the invoice request using the given function.
/// ///
/// Note: The hash computation may have included unknown, odd TLV records. /// Note: The hash computation may have included unknown, odd TLV records.
/// ///
/// This is not exported to bindings users as functions are not yet mapped. /// This is not exported to bindings users as functions are not yet mapped.
pub fn sign<F, E>(mut self, sign: F) -> Result<InvoiceRequest, SignError<E>> pub fn sign<F, E>(mut $self: $self_type, sign: F) -> Result<InvoiceRequest, SignError<E>>
where where
F: FnOnce(&Self) -> Result<Signature, E> F: FnOnce(&Self) -> Result<Signature, E>
{ {
let pubkey = self.contents.payer_id; let pubkey = $self.contents.payer_id;
let signature = merkle::sign_message(sign, &self, pubkey)?; let signature = merkle::sign_message(sign, &$self, pubkey)?;
// Append the signature TLV record to the bytes. // Append the signature TLV record to the bytes.
let signature_tlv_stream = SignatureTlvStreamRef { let signature_tlv_stream = SignatureTlvStreamRef {
signature: Some(&signature), signature: Some(&signature),
}; };
signature_tlv_stream.write(&mut self.bytes).unwrap(); signature_tlv_stream.write(&mut $self.bytes).unwrap();
Ok(InvoiceRequest { Ok(InvoiceRequest {
bytes: self.bytes, bytes: $self.bytes,
contents: self.contents, contents: $self.contents,
signature, signature,
}) })
} }
} }
impl UnsignedInvoiceRequest {
unsigned_invoice_request_sign_method!(self, Self);
} }
impl AsRef<TaggedHash> for UnsignedInvoiceRequest { impl AsRef<TaggedHash> for UnsignedInvoiceRequest {

View file

@ -584,7 +584,9 @@ impl Offer {
pub fn expects_quantity(&self) -> bool { pub fn expects_quantity(&self) -> bool {
self.contents.expects_quantity() self.contents.expects_quantity()
} }
}
macro_rules! request_invoice_derived_payer_id { ($self: ident, $builder: ty) => {
/// Similar to [`Offer::request_invoice`] except it: /// Similar to [`Offer::request_invoice`] except it:
/// - derives the [`InvoiceRequest::payer_id`] such that a different key can be used for each /// - derives the [`InvoiceRequest::payer_id`] such that a different key can be used for each
/// request, /// request,
@ -603,21 +605,21 @@ impl Offer {
/// [`Bolt12Invoice::verify`]: crate::offers::invoice::Bolt12Invoice::verify /// [`Bolt12Invoice::verify`]: crate::offers::invoice::Bolt12Invoice::verify
/// [`ExpandedKey`]: crate::ln::inbound_payment::ExpandedKey /// [`ExpandedKey`]: crate::ln::inbound_payment::ExpandedKey
pub fn request_invoice_deriving_payer_id<'a, 'b, ES: Deref, T: secp256k1::Signing>( pub fn request_invoice_deriving_payer_id<'a, 'b, ES: Deref, T: secp256k1::Signing>(
&'a self, expanded_key: &ExpandedKey, entropy_source: ES, secp_ctx: &'b Secp256k1<T>, &'a $self, expanded_key: &ExpandedKey, entropy_source: ES, secp_ctx: &'b Secp256k1<T>,
payment_id: PaymentId payment_id: PaymentId
) -> Result<InvoiceRequestBuilder<'a, 'b, DerivedPayerId, T>, Bolt12SemanticError> ) -> Result<$builder, Bolt12SemanticError>
where where
ES::Target: EntropySource, ES::Target: EntropySource,
{ {
if self.offer_features().requires_unknown_bits() { if $self.offer_features().requires_unknown_bits() {
return Err(Bolt12SemanticError::UnknownRequiredFeatures); return Err(Bolt12SemanticError::UnknownRequiredFeatures);
} }
Ok(InvoiceRequestBuilder::deriving_payer_id( Ok(<$builder>::deriving_payer_id($self, expanded_key, entropy_source, secp_ctx, payment_id))
self, expanded_key, entropy_source, secp_ctx, payment_id
))
} }
} }
macro_rules! request_invoice_explicit_payer_id { ($self: ident, $builder: ty) => {
/// Similar to [`Offer::request_invoice_deriving_payer_id`] except uses `payer_id` for the /// Similar to [`Offer::request_invoice_deriving_payer_id`] except uses `payer_id` for the
/// [`InvoiceRequest::payer_id`] instead of deriving a different key for each request. /// [`InvoiceRequest::payer_id`] instead of deriving a different key for each request.
/// ///
@ -627,19 +629,17 @@ impl Offer {
/// ///
/// [`InvoiceRequest::payer_id`]: crate::offers::invoice_request::InvoiceRequest::payer_id /// [`InvoiceRequest::payer_id`]: crate::offers::invoice_request::InvoiceRequest::payer_id
pub fn request_invoice_deriving_metadata<ES: Deref>( pub fn request_invoice_deriving_metadata<ES: Deref>(
&self, payer_id: PublicKey, expanded_key: &ExpandedKey, entropy_source: ES, &$self, payer_id: PublicKey, expanded_key: &ExpandedKey, entropy_source: ES,
payment_id: PaymentId payment_id: PaymentId
) -> Result<InvoiceRequestBuilder<ExplicitPayerId, secp256k1::SignOnly>, Bolt12SemanticError> ) -> Result<$builder, Bolt12SemanticError>
where where
ES::Target: EntropySource, ES::Target: EntropySource,
{ {
if self.offer_features().requires_unknown_bits() { if $self.offer_features().requires_unknown_bits() {
return Err(Bolt12SemanticError::UnknownRequiredFeatures); return Err(Bolt12SemanticError::UnknownRequiredFeatures);
} }
Ok(InvoiceRequestBuilder::deriving_metadata( Ok(<$builder>::deriving_metadata($self, payer_id, expanded_key, entropy_source, payment_id))
self, payer_id, expanded_key, entropy_source, payment_id
))
} }
/// Creates an [`InvoiceRequestBuilder`] for the offer with the given `metadata` and `payer_id`, /// Creates an [`InvoiceRequestBuilder`] for the offer with the given `metadata` and `payer_id`,
@ -658,16 +658,23 @@ impl Offer {
/// ///
/// [`InvoiceRequest`]: crate::offers::invoice_request::InvoiceRequest /// [`InvoiceRequest`]: crate::offers::invoice_request::InvoiceRequest
pub fn request_invoice( pub fn request_invoice(
&self, metadata: Vec<u8>, payer_id: PublicKey &$self, metadata: Vec<u8>, payer_id: PublicKey
) -> Result<InvoiceRequestBuilder<ExplicitPayerId, secp256k1::SignOnly>, Bolt12SemanticError> { ) -> Result<$builder, Bolt12SemanticError> {
if self.offer_features().requires_unknown_bits() { if $self.offer_features().requires_unknown_bits() {
return Err(Bolt12SemanticError::UnknownRequiredFeatures); return Err(Bolt12SemanticError::UnknownRequiredFeatures);
} }
Ok(InvoiceRequestBuilder::new(self, metadata, payer_id)) Ok(<$builder>::new($self, metadata, payer_id))
} }
} }
#[cfg(test)] impl Offer {
request_invoice_derived_payer_id!(self, InvoiceRequestBuilder<'a, 'b, DerivedPayerId, T>);
request_invoice_explicit_payer_id!(self, InvoiceRequestBuilder<ExplicitPayerId, secp256k1::SignOnly>);
}
#[cfg(test)]
impl Offer {
pub(super) fn as_tlv_stream(&self) -> OfferTlvStreamRef { pub(super) fn as_tlv_stream(&self) -> OfferTlvStreamRef {
self.contents.as_tlv_stream() self.contents.as_tlv_stream()
} }