_ Git - rust-lightning/blob - fuzz/src/chanmon_deser.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 use bitcoin_hashes::sha256d::Hash as Sha256dHash;
   5
   6 use lightning::ln::channelmonitor;
   7 use lightning::util::ser::{ReadableArgs, Writer};
   8
   9 use utils::test_logger;
  10
  11 use std::io::Cursor;
  12 use std::sync::Arc;
  13
  14 struct VecWriter(Vec<u8>);
  15 impl Writer for VecWriter {
  16         fn write_all(&mut self, buf: &[u8]) -> Result<(), ::std::io::Error> {
  17                 self.0.extend_from_slice(buf);
  18                 Ok(())
  19         }
  20         fn size_hint(&mut self, size: usize) {
  21                 self.0.reserve_exact(size);
  22         }
  23 }
  24
  25 #[inline]
  26 pub fn do_test(data: &[u8]) {
  27         let logger = Arc::new(test_logger::TestLogger::new("".to_owned()));
  28         if let Ok((latest_block_hash, monitor)) = <(Sha256dHash, channelmonitor::ChannelMonitor)>::read(&mut Cursor::new(data), logger.clone()) {
  29                 let mut w = VecWriter(Vec::new());
  30                 monitor.write_for_disk(&mut w).unwrap();
  31                 let deserialized_copy = <(Sha256dHash, channelmonitor::ChannelMonitor)>::read(&mut Cursor::new(&w.0), logger.clone()).unwrap();
  32                 assert!(latest_block_hash == deserialized_copy.0);
  33                 assert!(monitor == deserialized_copy.1);
  34                 w.0.clear();
  35                 monitor.write_for_watchtower(&mut w).unwrap();
  36         }
  37 }
  38
  39 #[no_mangle]
  40 pub extern "C" fn chanmon_deser_run(data: *const u8, datalen: usize) {
  41         do_test(unsafe { std::slice::from_raw_parts(data, datalen) });
  42 }
  43
  44 #[cfg(test)]
  45 mod tests {
  46         #[test]
  47         fn duplicate_crash() {
  48                 super::do_test(&::hex::decode("00").unwrap());
  49         }
  50 }