X-Git-Url: http://git.bitcoin.ninja/index.cgi?a=blobdiff_plain;f=fuzz%2Ffuzz_targets%2Ffull_stack_target.rs;h=75456127eb2c7e0d195f50425f29bdf601410120;hb=ada2154b40237d22edc72ab553ca4a4d42d4a2c1;hp=54cb838f857c654337a26e5511df66425d68954c;hpb=609054eae0c7c8cf08645cc6399fd11e840a6d96;p=rust-lightning

diff --git a/fuzz/fuzz_targets/full_stack_target.rs b/fuzz/fuzz_targets/full_stack_target.rs
index 54cb838f..75456127 100644
--- a/fuzz/fuzz_targets/full_stack_target.rs
+++ b/fuzz/fuzz_targets/full_stack_target.rs
@@ -10,7 +10,6 @@ use bitcoin::network::constants::Network;
 use bitcoin::network::serialize::{serialize, BitcoinHash};
 use bitcoin::util::hash::Sha256dHash;
 
-use crypto::sha2::Sha256;
 use crypto::digest::Digest;
 
 use lightning::chain::chaininterface::{BroadcasterInterface,ConfirmationTarget,ChainListener,FeeEstimator,ChainWatchInterfaceUtil};
@@ -22,6 +21,7 @@ use lightning::ln::router::Router;
 use lightning::util::events::{EventsProvider,Event};
 use lightning::util::reset_rng_state;
 use lightning::util::logger::Logger;
+use lightning::util::sha2::Sha256;
 
 mod utils;
 
@@ -174,7 +174,7 @@ pub fn do_test(data: &[u8]) {
 		Err(_) => return,
 	};
 
-	let logger: Arc<Logger> = Arc::new(test_logger::TestLogger::new());
+	let logger: Arc<Logger> = Arc::new(test_logger::TestLogger{});
 	let monitor = Arc::new(TestChannelMonitor{});
 	let watch = Arc::new(ChainWatchInterfaceUtil::new(Arc::clone(&logger)));
 	let broadcast = Arc::new(TestBroadcaster{});
@@ -189,7 +189,7 @@ pub fn do_test(data: &[u8]) {
 	}, our_network_key, Arc::clone(&logger));
 
 	let mut should_forward = false;
-	let mut payments_received = Vec::new();
+	let mut payments_received: Vec<[u8; 32]> = Vec::new();
 	let mut payments_sent = 0;
 	let mut pending_funding_generation: Vec<([u8; 32], u64, Script)> = Vec::new();
 	let mut pending_funding_signatures = HashMap::new();
@@ -246,7 +246,6 @@ pub fn do_test(data: &[u8]) {
 				let mut sha = Sha256::new();
 				sha.input(&payment_hash);
 				sha.result(&mut payment_hash);
-				for i in 1..32 { payment_hash[i] = 0; }
 				payments_sent += 1;
 				match channelmanager.send_payment(route, payment_hash) {
 					Ok(_) => {},
@@ -258,7 +257,8 @@ pub fn do_test(data: &[u8]) {
 				if !peers.borrow()[peer_id as usize] { return; }
 				let their_key = get_pubkey!();
 				let chan_value = slice_to_be24(get_slice!(3)) as u64;
-				if channelmanager.create_channel(their_key, chan_value, 0).is_err() { return; }
+				let push_msat_value = slice_to_be24(get_slice!(3)) as u64;
+				if channelmanager.create_channel(their_key, chan_value, push_msat_value, 0).is_err() { return; }
 			},
 			6 => {
 				let mut channels = channelmanager.list_channels();
@@ -276,22 +276,14 @@ pub fn do_test(data: &[u8]) {
 			},
 			8 => {
 				for payment in payments_received.drain(..) {
-					let mut payment_preimage = None;
-					for i in 0..payments_sent {
-						let mut payment_hash = [0; 32];
-						payment_hash[0..8].copy_from_slice(&be64_to_array(i));
-						let mut sha = Sha256::new();
-						sha.input(&payment_hash);
-						sha.result(&mut payment_hash);
-						for i in 1..32 { payment_hash[i] = 0; }
-						if payment_hash == payment {
-							payment_hash = [0; 32];
-							payment_hash[0..8].copy_from_slice(&be64_to_array(i));
-							payment_preimage = Some(payment_hash);
-							break;
-						}
-					}
-					channelmanager.claim_funds(payment_preimage.unwrap());
+					let mut payment_preimage = [0; 32];
+					payment_preimage[0] = payment[0];
+					let mut sha = Sha256::new();
+					sha.input(&payment_preimage);
+					let mut payment_hash_check = [0; 32];
+					sha.result(&mut payment_hash_check);
+					assert!(payment_hash_check == payment);
+					channelmanager.claim_funds(payment_preimage);
 				}
 			},
 			9 => {