_ Git - rust-lightning/blob - fuzz/src/offer_deser.rs

   1 // This file is Copyright its original authors, visible in version control
   2 // history.
   3 //
   4 // This file is licensed under the Apache License, Version 2.0 <LICENSE-APACHE
   5 // or http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   6 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your option.
   7 // You may not use this file except in accordance with one or both of these
   8 // licenses.
   9
  10 use bitcoin::secp256k1::{KeyPair, PublicKey, Secp256k1, SecretKey};
  11 use crate::utils::test_logger;
  12 use core::convert::{Infallible, TryFrom};
  13 use lightning::offers::invoice_request::UnsignedInvoiceRequest;
  14 use lightning::offers::offer::{Amount, Offer, Quantity};
  15 use lightning::offers::parse::Bolt12SemanticError;
  16 use lightning::util::ser::Writeable;
  17
  18 #[inline]
  19 pub fn do_test<Out: test_logger::Output>(data: &[u8], _out: Out) {
  20         if let Ok(offer) = Offer::try_from(data.to_vec()) {
  21                 let mut bytes = Vec::with_capacity(data.len());
  22                 offer.write(&mut bytes).unwrap();
  23                 assert_eq!(data, bytes);
  24
  25                 let secp_ctx = Secp256k1::new();
  26                 let keys = KeyPair::from_secret_key(&secp_ctx, &SecretKey::from_slice(&[42; 32]).unwrap());
  27                 let pubkey = PublicKey::from(keys);
  28                 let mut buffer = Vec::new();
  29
  30                 if let Ok(invoice_request) = build_response(&offer, pubkey) {
  31                         invoice_request
  32                                 .sign::<_, Infallible>(
  33                                         |digest| Ok(secp_ctx.sign_schnorr_no_aux_rand(digest, &keys))
  34                                 )
  35                                 .unwrap()
  36                                 .write(&mut buffer)
  37                                 .unwrap();
  38                 }
  39         }
  40 }
  41
  42 fn build_response<'a>(
  43         offer: &'a Offer, pubkey: PublicKey
  44 ) -> Result<UnsignedInvoiceRequest<'a>, Bolt12SemanticError> {
  45         let mut builder = offer.request_invoice(vec![42; 64], pubkey)?;
  46
  47         builder = match offer.amount() {
  48                 None => builder.amount_msats(1000).unwrap(),
  49                 Some(Amount::Bitcoin { amount_msats }) => builder.amount_msats(amount_msats + 1)?,
  50                 Some(Amount::Currency { .. }) => return Err(Bolt12SemanticError::UnsupportedCurrency),
  51         };
  52
  53         builder = match offer.supported_quantity() {
  54                 Quantity::Bounded(n) => builder.quantity(n.get()).unwrap(),
  55                 Quantity::Unbounded => builder.quantity(10).unwrap(),
  56                 Quantity::One => builder,
  57         };
  58
  59         builder.build()
  60 }
  61
  62 pub fn offer_deser_test<Out: test_logger::Output>(data: &[u8], out: Out) {
  63         do_test(data, out);
  64 }
  65
  66 #[no_mangle]
  67 pub extern "C" fn offer_deser_run(data: *const u8, datalen: usize) {
  68         do_test(unsafe { std::slice::from_raw_parts(data, datalen) }, test_logger::DevNull {});
  69 }