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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
pub fn read_file_data(path: &PathBuf) -> Vec<u8> {
let mut data = Vec::new();
let mut f = File::open(path).ok().unwrap();
f.read_to_end(&mut data).unwrap();
data
}
const VALIDATOR_CONFIG: Option<ValidatingParserConfig> = Some(ValidatingParserConfig {
operator_config: OperatorValidatorConfig {
enable_threads: true,
enable_reference_types: true,
enable_simd: true,
enable_bulk_memory: true,
enable_multi_value: true,
},
});
#[macro_use]
extern crate criterion;
extern crate wasmparser;
use criterion::Criterion;
use wasmparser::{
validate, OperatorValidatorConfig, Parser, ParserState, ValidatingParser,
ValidatingParserConfig, WasmDecoder,
};
use std::fs::{read_dir, File};
use std::io::Read;
use std::path::PathBuf;
fn read_all_wasm<'a, T>(mut d: T)
where
T: WasmDecoder<'a>,
{
loop {
match *d.read() {
ParserState::Error(e) => panic!("unexpected error {:?}", e),
ParserState::EndWasm => return,
_ => (),
}
}
}
fn it_works_benchmark(c: &mut Criterion) {
let mut data: Vec<Vec<u8>> = vec![];
for entry in read_dir("tests").unwrap() {
let dir = entry.unwrap();
if !dir.file_type().unwrap().is_file() {
continue;
}
data.push(read_file_data(&dir.path()));
}
c.bench_function("it works benchmark", move |b| {
for d in &mut data {
b.iter(|| read_all_wasm(Parser::new(d.as_slice())));
}
});
}
fn validator_not_fails_benchmark(c: &mut Criterion) {
let mut data: Vec<Vec<u8>> = vec![];
for entry in read_dir("tests").unwrap() {
let dir = entry.unwrap();
if !dir.file_type().unwrap().is_file() {
continue;
}
data.push(read_file_data(&dir.path()));
}
c.bench_function("validator no fails benchmark", move |b| {
for d in &mut data {
b.iter(|| read_all_wasm(ValidatingParser::new(d.as_slice(), VALIDATOR_CONFIG)));
}
});
}
fn validate_benchmark(c: &mut Criterion) {
let mut data: Vec<Vec<u8>> = vec![vec![]];
for entry in read_dir("tests").unwrap() {
let dir = entry.unwrap();
if !dir.file_type().unwrap().is_file() {
continue;
}
data.push(read_file_data(&dir.path()));
}
c.bench_function("validate benchmark", move |b| {
for d in &mut data {
b.iter(|| validate(&d, VALIDATOR_CONFIG));
}
});
}
criterion_group!(
benchmark,
it_works_benchmark,
validator_not_fails_benchmark,
validate_benchmark
);
criterion_main!(benchmark);
|
