1 files changed, 29 insertions, 0 deletions
diff --git a/third_party/rust/wasm-smith/benches/corpus.rs b/third_party/rust/wasm-smith/benches/corpus.rs
new file mode 100644
index 0000000000..56206c69c4
--- /dev/null
+++ b/third_party/rust/wasm-smith/benches/corpus.rs
@@ -0,0 +1,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);