From e151425087b6c0ee8c4d099bfc2d6d1268201db0 Mon Sep 17 00:00:00 2001
From: Mason Reed <mason@vector35.com>
Date: Wed, 23 Oct 2024 20:47:31 -0400
Subject: Add WARP integration

https://github.com/Vector35/warp/
---
 plugins/warp/benches/function.rs | 42 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 plugins/warp/benches/function.rs

(limited to 'plugins/warp/benches/function.rs')

diff --git a/plugins/warp/benches/function.rs b/plugins/warp/benches/function.rs
new file mode 100644
index 00000000..c6f6d70d
--- /dev/null
+++ b/plugins/warp/benches/function.rs
@@ -0,0 +1,42 @@
+use binaryninja::binaryview::BinaryViewExt;
+use binaryninja::headless::Session;
+use criterion::{criterion_group, criterion_main, Criterion};
+use rayon::prelude::*;
+use warp_ninja::build_function;
+use warp_ninja::cache::FunctionCache;
+
+pub fn function_benchmark(c: &mut Criterion) {
+    let session = Session::new();
+    let bv = session.load(env!("TEST_BIN_LIBRARY_OBJ")).unwrap();
+    let functions = bv.functions();
+    assert_eq!(functions.len(), 6);
+    let mut function_iter = functions.into_iter();
+    let first_function = function_iter.next().unwrap();
+
+    c.bench_function("signature first function", |b| {
+        b.iter(|| {
+            let _ = build_function(&first_function, &first_function.low_level_il().unwrap());
+        })
+    });
+
+    c.bench_function("signature all functions", |b| {
+        b.iter(|| {
+            for func in &functions {
+                let _ = build_function(&func, &func.low_level_il().unwrap());
+            }
+        })
+    });
+
+    let cache = FunctionCache::default();
+    c.bench_function("signature all functions rayon", |b| {
+        b.iter(|| {
+            functions
+                .par_iter()
+                .map_with(cache.clone(), |par_cache, func| par_cache.function(&func))
+                .collect::<Vec<_>>()
+        })
+    });
+}
+
+criterion_group!(benches, function_benchmark);
+criterion_main!(benches);
-- 
cgit v1.3.1