corpus.rs - mozsearch

Enable keyboard shortcuts

use arbitrary::{Arbitrary, Unstructured};

use criterion::{black_box, criterion_group, criterion_main, Criterion};

use wasm_smith::Module;

pub fn benchmark_corpus(c: &mut Criterion) {

    let mut corpus = Vec::with_capacity(2000);

    let entries = std::fs::read_dir("./benches/corpus").expect("failed to read dir");

    for e in entries {

        let e = e.expect("failed to read dir entry");

        let seed = std::fs::read(e.path()).expect("failed to read seed file");

        corpus.push(seed);

    // Benchmark how long it takes to generate a module for every seed in our

    // corpus (taken from the `validate` fuzz target).

    c.bench_function("corpus", |b| {

        b.iter(|| {

            for seed in &corpus {

                let seed = black_box(seed);

                let mut u = Unstructured::new(seed);

                let result = Module::arbitrary(&mut u);

                let _ = black_box(result);

})

});

criterion_group!(benches, benchmark_corpus);

criterion_main!(benches);