summaryrefslogtreecommitdiffstats
path: root/third_party/rust/wasm-smith/benches/corpus.rs
blob: 56206c69c481a532331ac7ea6052c72c875270e2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
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);