[pull] main from VirusTotal:main

lib/Cargo.toml

@@ -359,4 +359,8 @@ zip = { workspace = true }
 
 [[bench]]
 name = "dex"
+harness = false
+
+[[bench]]
+name = "pe"
 harness = false

lib/benches/dex.rs

@@ -1,4 +1,4 @@
-use criterion::{black_box, criterion_group, criterion_main, Criterion};
+use criterion::{Criterion, black_box, criterion_group, criterion_main};
 
 mod commons;
 
@@ -13,7 +13,9 @@ fn bench_dex(c: &mut Criterion) {
 
     group.bench_function("parse", |b| {
         b.iter(|| {
-            let _ = black_box(yara_x::mods::invoke::<yara_x::mods::Dex>(black_box(&data)));
+            let _ = black_box(yara_x::mods::invoke::<yara_x::mods::Dex>(
+                black_box(&data),
+            ));
         });
     });
 

lib/benches/pe.rs

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

lib/src/modules/dex/mod.rs

@@ -27,7 +27,7 @@ fn main(data: &[u8], _meta: Option<&[u8]>) -> Result<Dex, ModuleError> {
     SIGNATURE_CACHE.with(|cache| *cache.borrow_mut() = None);
 
     match parser::Dex::parse(data) {
-        Ok(dex) => Ok(dex.into()),
+        Ok(dex) => Ok(dex),
         Err(_) => {
             let mut dex = Dex::new();
             dex.set_is_dex(false);