_ Git - rust-lightning/blob - fuzz/fuzz_targets/chanmon_deser_target.rs

   1 // This file is auto-generated by gen_target.sh based on msg_target_template.txt
   2 // To modify it, modify msg_target_template.txt and run gen_target.sh instead.
   3
   4 extern crate bitcoin;
   5 extern crate bitcoin_hashes;
   6 extern crate lightning;
   7
   8 use bitcoin_hashes::sha256d::Hash as Sha256dHash;
   9
  10 use lightning::ln::channelmonitor;
  11 use lightning::util::reset_rng_state;
  12 use lightning::util::ser::{ReadableArgs, Writer};
  13
  14 mod utils;
  15 use utils::test_logger;
  16
  17 use std::io::Cursor;
  18 use std::sync::Arc;
  19
  20 struct VecWriter(Vec<u8>);
  21 impl Writer for VecWriter {
  22         fn write_all(&mut self, buf: &[u8]) -> Result<(), ::std::io::Error> {
  23                 self.0.extend_from_slice(buf);
  24                 Ok(())
  25         }
  26         fn size_hint(&mut self, size: usize) {
  27                 self.0.reserve_exact(size);
  28         }
  29 }
  30
  31 #[inline]
  32 pub fn do_test(data: &[u8]) {
  33         reset_rng_state();
  34         let logger = Arc::new(test_logger::TestLogger::new("".to_owned()));
  35         if let Ok((latest_block_hash, monitor)) = <(Sha256dHash, channelmonitor::ChannelMonitor)>::read(&mut Cursor::new(data), logger.clone()) {
  36                 let mut w = VecWriter(Vec::new());
  37                 monitor.write_for_disk(&mut w).unwrap();
  38                 let deserialized_copy = <(Sha256dHash, channelmonitor::ChannelMonitor)>::read(&mut Cursor::new(&w.0), logger.clone()).unwrap();
  39                 assert!(latest_block_hash == deserialized_copy.0);
  40                 assert!(monitor == deserialized_copy.1);
  41                 w.0.clear();
  42                 monitor.write_for_watchtower(&mut w).unwrap();
  43         }
  44 }
  45
  46 #[cfg(feature = "afl")]
  47 #[macro_use] extern crate afl;
  48 #[cfg(feature = "afl")]
  49 fn main() {
  50         fuzz!(|data| {
  51                 do_test(data);
  52         });
  53 }
  54
  55 #[cfg(feature = "honggfuzz")]
  56 #[macro_use] extern crate honggfuzz;
  57 #[cfg(feature = "honggfuzz")]
  58 fn main() {
  59         loop {
  60                 fuzz!(|data| {
  61                         do_test(data);
  62                 });
  63         }
  64 }
  65
  66 extern crate hex;
  67 #[cfg(test)]
  68 mod tests {
  69
  70         #[test]
  71         fn duplicate_crash() {
  72                 super::do_test(&::hex::decode("00").unwrap());
  73         }
  74 }