fuzzy exports

Author: joostjager

fuzz/.gitignore

@@ -1,3 +1,4 @@
 hfuzz_target
 target
 hfuzz_workspace
+corpus

fuzz/src/process_onion_failure.rs

@@ -92,7 +92,7 @@ fn do_test<Out: test_logger::Output>(data: &[u8], out: Out) {
 
 	let mut hops = Vec::<RouteHop>::new();
 	let hop_count = get_slice!(1)[0] as usize;
-	for i in 0..hop_count {
+	for _ in 0..hop_count {
 		hops.push(RouteHop {
 			pubkey: get_pubkey!(),
 			node_features: NodeFeatures::empty(),
@@ -108,7 +108,7 @@ fn do_test<Out: test_logger::Output>(data: &[u8], out: Out) {
 		true => {
 			let mut trampoline_hops = Vec::<TrampolineHop>::new();
 			let trampoline_hop_count = get_slice!(1)[0] as usize;
-			for i in 0..trampoline_hop_count {
+			for _ in 0..trampoline_hop_count {
 				trampoline_hops.push(TrampolineHop {
 					pubkey: get_pubkey!(),
 					node_features: NodeFeatures::empty(),
@@ -118,7 +118,7 @@ fn do_test<Out: test_logger::Output>(data: &[u8], out: Out) {
 			}
 			let mut blinded_hops = Vec::<BlindedHop>::new();
 			let blinded_hop_count = get_slice!(1)[0] as usize;
-			for i in 0..blinded_hop_count {
+			for _ in 0..blinded_hop_count {
 				blinded_hops.push(BlindedHop {
 					blinded_node_id: get_pubkey!(),
 					encrypted_payload: get_slice!(get_u8!()).to_vec(),
@@ -147,12 +147,7 @@ fn do_test<Out: test_logger::Output>(data: &[u8], out: Out) {
 	let failure_len = u16::from_be_bytes(get_slice!(2).try_into().unwrap());
 	let encrypted_packet = OnionErrorPacket { data: get_slice!(failure_len as u64).into() };
 
-	lightning::ln::onion_utils::process_onion_failure(
-		&secp_ctx,
-		&logger,
-		&htlc_source,
-		encrypted_packet,
-	);
+	lightning::ln::process_onion_failure(&secp_ctx, &logger, &htlc_source, encrypted_packet);
 }
 
 /// Method that needs to be added manually, {name}_test

lightning/src/ln/channelmanager.rs

@@ -674,21 +674,29 @@ impl_writeable_tlv_based_enum!(SentHTLCId,
 	},
 );
 
-
-/// Tracks the inbound corresponding to an outbound HTLC
-#[allow(clippy::derive_hash_xor_eq)] // Our Hash is faithful to the data, we just don't have SecretKey::hash
-#[derive(Clone, Debug, PartialEq, Eq)]
-pub enum HTLCSource {
-	PreviousHopData(HTLCPreviousHopData),
-	OutboundRoute {
-		path: Path,
-		session_priv: SecretKey,
-		/// Technically we can recalculate this from the route, but we cache it here to avoid
-		/// doing a double-pass on route when we get a failure back
-		first_hop_htlc_msat: u64,
-		payment_id: PaymentId,
-	},
+mod fuzzy_channelmanager {
+	use super::*;
+
+	/// Tracks the inbound corresponding to an outbound HTLC
+	#[allow(clippy::derive_hash_xor_eq)] // Our Hash is faithful to the data, we just don't have SecretKey::hash
+	#[derive(Clone, Debug, PartialEq, Eq)]
+	pub enum HTLCSource {
+		PreviousHopData(HTLCPreviousHopData),
+		OutboundRoute {
+			path: Path,
+			session_priv: SecretKey,
+			/// Technically we can recalculate this from the route, but we cache it here to avoid
+			/// doing a double-pass on route when we get a failure back
+			first_hop_htlc_msat: u64,
+			payment_id: PaymentId,
+		},
+	}
 }
+#[cfg(fuzzing)]
+pub use self::fuzzy_channelmanager::*;
+#[cfg(not(fuzzing))]
+pub(crate) use self::fuzzy_channelmanager::*;
+
 #[allow(clippy::derive_hash_xor_eq)] // Our Hash is faithful to the data, we just don't have SecretKey::hash
 impl core::hash::Hash for HTLCSource {
 	fn hash<H: core::hash::Hasher>(&self, hasher: &mut H) {

lightning/src/ln/mod.rs

@@ -39,7 +39,7 @@ pub mod channel;
 #[cfg(not(fuzzing))]
 pub(crate) mod channel;
 
-pub mod onion_utils;
+pub(crate) mod onion_utils;
 mod outbound_payment;
 pub mod wire;
 
@@ -51,6 +51,9 @@ pub use onion_utils::create_payment_onion;
 // without the node parameter being mut. This is incorrect, and thus newer rustcs will complain
 // about an unnecessary mut. Thus, we silence the unused_mut warning in two test modules below.
 
+#[cfg(fuzzing)]
+pub use onion_utils::process_onion_failure;
+
 #[cfg(test)]
 #[allow(unused_mut)]
 pub mod bolt11_payment_tests;

lightning/src/ln/msgs.rs

@@ -2241,6 +2241,13 @@ mod fuzzy_internal_msgs {
 		pub(crate) failuremsg: Vec<u8>,
 		pub(crate) pad: Vec<u8>,
 	}
+
+	#[derive(Clone, Debug, Hash, PartialEq, Eq)]
+	pub struct OnionErrorPacket {
+		// This really should be a constant size slice, but the spec lets these things be up to 128KB?
+		// (TODO) We limit it in decode to much lower...
+		pub data: Vec<u8>,
+	}
 }
 #[cfg(fuzzing)]
 pub use self::fuzzy_internal_msgs::*;
@@ -2349,13 +2356,6 @@ impl Debug for TrampolineOnionPacket {
 	}
 }
 
-#[derive(Clone, Debug, Hash, PartialEq, Eq)]
-pub struct OnionErrorPacket {
-	// This really should be a constant size slice, but the spec lets these things be up to 128KB?
-	// (TODO) We limit it in decode to much lower...
-	pub data: Vec<u8>,
-}
-
 impl From<UpdateFailHTLC> for OnionErrorPacket {
 	fn from(msg: UpdateFailHTLC) -> Self {
 		OnionErrorPacket {

lightning/src/ln/onion_utils.rs

@@ -926,16 +926,24 @@ pub(super) fn build_failure_packet(
 	onion_error_packet
 }
 
-pub struct DecodedOnionFailure {
-	pub(crate) network_update: Option<NetworkUpdate>,
-	pub(crate) short_channel_id: Option<u64>,
-	pub(crate) payment_failed_permanently: bool,
-	pub(crate) failed_within_blinded_path: bool,
-	#[cfg(any(test, feature = "_test_utils"))]
-	pub(crate) onion_error_code: Option<u16>,
-	#[cfg(any(test, feature = "_test_utils"))]
-	pub(crate) onion_error_data: Option<Vec<u8>>,
+mod fuzzy_onion_utils {
+	use super::*;
+
+	pub struct DecodedOnionFailure {
+		pub(crate) network_update: Option<NetworkUpdate>,
+		pub(crate) short_channel_id: Option<u64>,
+		pub(crate) payment_failed_permanently: bool,
+		pub(crate) failed_within_blinded_path: bool,
+		#[cfg(any(test, feature = "_test_utils"))]
+		pub(crate) onion_error_code: Option<u16>,
+		#[cfg(any(test, feature = "_test_utils"))]
+		pub(crate) onion_error_data: Option<Vec<u8>>,
+	}
 }
+#[cfg(fuzzing)]
+pub use self::fuzzy_onion_utils::*;
+#[cfg(not(fuzzing))]
+pub(crate) use self::fuzzy_onion_utils::*;
 
 pub fn process_onion_failure<T: secp256k1::Signing, L: Deref>(
 	secp_ctx: &Secp256k1<T>, logger: &L, htlc_source: &HTLCSource,