#[cfg(feature = "fuzz")]
use anyhow::Result;
macro_rules! impl_fuzz_repr {
($func_name:ident, $func_zip_name:ident, $fuzz_name:ident) => {
#[cfg(feature = "fuzz")]
#[test]
fn $func_name() -> Result<()> {
use arbitrary::Arbitrary;
use std::io::Read;
let dir = format!("fuzz/corpus/{}", stringify!($fuzz_name));
if !std::path::Path::new(&dir).exists() {
eprintln!("The corpus directory {} does not exist", dir);
return Ok(());
}
for file in std::fs::read_dir(&dir)? {
let file = file?;
if file.file_type()?.is_dir() {
continue;
}
let filename = format!("{}/{}", dir, file.file_name().to_string_lossy());
let mut file = std::fs::File::open(&filename)?;
let metadata = std::fs::metadata(&filename)?;
let mut file_bytes = vec![0; metadata.len() as usize];
file.read_exact(&mut file_bytes)?;
let mut unstructured = arbitrary::Unstructured::new(&file_bytes);
let data = webgraph::fuzz::$fuzz_name::FuzzCase::arbitrary(&mut unstructured)?;
webgraph::fuzz::$fuzz_name::harness(data);
}
Ok(())
}
#[cfg(feature = "fuzz")]
#[test]
fn $func_zip_name() -> Result<()> {
use arbitrary::Arbitrary;
use std::io::prelude::*;
let zip_path = format!("tests/corpus/{}.zip", stringify!($fuzz_name));
let zip_file = std::fs::File::open(&zip_path)?;
let mut zip = zip::ZipArchive::new(zip_file)?;
for i in 0..zip.len() {
let mut file = zip.by_index(i)?;
if file.is_dir() {
continue;
}
let mut file_bytes = vec![0; file.size() as usize];
file.read_exact(&mut file_bytes)?;
let mut unstructured = arbitrary::Unstructured::new(&file_bytes);
let data = webgraph::fuzz::$fuzz_name::FuzzCase::arbitrary(&mut unstructured)?;
webgraph::fuzz::$fuzz_name::harness(data);
}
Ok(())
}
};
}
impl_fuzz_repr!(
fuzz_bvcomp_and_read,
fuzz_bvcomp_and_read_zip,
bvcomp_and_read
);