chore: write a benchmark for measuring performance of dex module.

Author: plusvic

Cargo.lock

@@ -348,10 +348,15 @@ protobuf-parse = { workspace = true, optional = true }
 yara-x-proto = { workspace = true, optional = true }
 
 [dev-dependencies]
+criterion = "0.5"
 globwalk = { workspace = true }
 goldenfile = { workspace = true }
 ihex = { workspace = true }
 pretty_assertions = { workspace = true }
 rayon = { workspace = true }
 yara-x-proto-yaml = { workspace = true }
-zip = { workspace = true }
+zip = { workspace = true }
+
+[[bench]]
+name = "dex"
+harness = false

lib/Cargo.toml

@@ -0,0 +1,41 @@
+use std::fs::File;
+use std::io::Read;
+use std::path::Path;
+
+pub fn create_binary_from_ihex(ihex: &str) -> Vec<u8> {
+    let mut reader = ihex::Reader::new(ihex);
+    let mut data = Vec::new();
+    while let Some(Ok(record)) = reader.next() {
+        if let ihex::Record::Data { value, .. } = record {
+            data.extend(value);
+        }
+    }
+    data
+}
+
+pub fn create_binary_from_zipped_ihex<P: AsRef<Path>>(path: P) -> Vec<u8> {
+    let path = path.as_ref();
+    let f = File::open(path)
+        .unwrap_or_else(|_| panic!("can not open file: {:?}", &path));
+
+    let mut zip = zip::ZipArchive::new(f)
+        .unwrap_or_else(|_| panic!("can not unzip file: {:?}", &path));
+
+    let path_without_zip_ext = path.with_extension("");
+    let inner_file_name =
+        path_without_zip_ext.file_name().unwrap().to_str().unwrap();
+
+    let mut inner_file = zip.by_name(inner_file_name).unwrap_or_else(|_| {
+        panic!(
+            "ZIP archive {:?} doesn't contain file: {:?}",
+            &path, &inner_file_name
+        )
+    });
+
+    let mut ihex = String::new();
+    inner_file.read_to_string(&mut ihex).unwrap_or_else(|_| {
+        panic!("can not read ihex content from : {:?}", &path)
+    });
+
+    create_binary_from_ihex(ihex.as_str())
+}

lib/benches/commons/mod.rs

@@ -0,0 +1,24 @@
+use criterion::{black_box, criterion_group, criterion_main, Criterion};
+
+mod commons;
+
+use commons::create_binary_from_zipped_ihex;
+
+fn bench_dex(c: &mut Criterion) {
+    let data = create_binary_from_zipped_ihex(
+        "src/modules/dex/tests/testdata/c14c75d58399825287e0ee0fcfede6ec06f93489fb52f70bca2736fae5fceab2.in.zip",
+    );
+
+    let mut group = c.benchmark_group("dex");
+
+    group.bench_function("parse", |b| {
+        b.iter(|| {
+            let _ = black_box(yara_x::mods::invoke::<yara_x::mods::Dex>(black_box(&data)));
+        });
+    });
+
+    group.finish();
+}
+
+criterion_group!(benches, bench_dex);
+criterion_main!(benches);