_ 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 lightning;
   5
   6 use lightning::ln::channelmonitor;
   7 use lightning::util::reset_rng_state;
   8 use lightning::util::ser::{ReadableArgs, Writer};
   9
  10 mod utils;
  11 use utils::test_logger;
  12
  13 use std::io::Cursor;
  14 use std::sync::Arc;
  15
  16 struct VecWriter(Vec<u8>);
  17 impl Writer for VecWriter {
  18         fn write_all(&mut self, buf: &[u8]) -> Result<(), ::std::io::Error> {
  19                 self.0.extend_from_slice(buf);
  20                 Ok(())
  21         }
  22         fn size_hint(&mut self, size: usize) {
  23                 self.0.reserve_exact(size);
  24         }
  25 }
  26
  27 #[inline]
  28 pub fn do_test(data: &[u8]) {
  29         reset_rng_state();
  30         let logger = Arc::new(test_logger::TestLogger{});
  31         if let Ok(monitor) = channelmonitor::ChannelMonitor::read(&mut Cursor::new(data), logger.clone()) {
  32                 let mut w = VecWriter(Vec::new());
  33                 monitor.write_for_disk(&mut w).unwrap();
  34                 assert!(channelmonitor::ChannelMonitor::read(&mut Cursor::new(&w.0), logger.clone()).unwrap() == monitor);
  35                 w.0.clear();
  36                 monitor.write_for_watchtower(&mut w).unwrap();
  37         }
  38 }
  39
  40 #[cfg(feature = "afl")]
  41 #[macro_use] extern crate afl;
  42 #[cfg(feature = "afl")]
  43 fn main() {
  44         fuzz!(|data| {
  45                 do_test(data);
  46         });
  47 }
  48
  49 #[cfg(feature = "honggfuzz")]
  50 #[macro_use] extern crate honggfuzz;
  51 #[cfg(feature = "honggfuzz")]
  52 fn main() {
  53         loop {
  54                 fuzz!(|data| {
  55                         do_test(data);
  56                 });
  57         }
  58 }
  59
  60 extern crate hex;
  61 #[cfg(test)]
  62 mod tests {
  63
  64         #[test]
  65         fn duplicate_crash() {
  66                 super::do_test(&::hex::decode("00").unwrap());
  67         }
  68 }