WARP: Add manual file loading and initial user settings

Also just some general cleanup in the matcher

1 files changed, 52 insertions, 0 deletions

diff --git a/plugins/warp/src/plugin/load.rs b/plugins/warp/src/plugin/load.rs
new file mode 100644
index 00000000..03356444
--- /dev/null
+++ b/plugins/warp/src/plugin/load.rs
@@ -0,0 +1,52 @@
+use crate::matcher::{Matcher, PlatformID, PLAT_MATCHER_CACHE};
+use binaryninja::binaryview::{BinaryView, BinaryViewExt};
+use binaryninja::command::Command;
+pub struct LoadSignatureFile;
+
+impl Command for LoadSignatureFile {
+    fn action(&self, view: &BinaryView) {
+        let Some(platform) = view.default_platform() else {
+            log::error!("Default platform must be set to load signature!");
+            return;
+        };
+
+        let Some(file) =
+            binaryninja::interaction::get_open_filename_input("Load Signature File", "*.sbin")
+        else {
+            return;
+        };
+
+        let Ok(data) = std::fs::read(&file) else {
+            log::error!("Could not read signature file: {:?}", file);
+            return;
+        };
+
+        let Some(data) = warp::signature::Data::from_bytes(&data) else {
+            log::error!("Could not get data from signature file: {:?}", file);
+            return;
+        };
+
+        let new_matcher = Matcher::from_data(data);
+        log::info!(
+            "Loading signature file with {} functions and {} types...",
+            new_matcher.functions.len(),
+            new_matcher.types.len()
+        );
+        let platform_id = PlatformID::from(platform.as_ref());
+        let matcher_cache = PLAT_MATCHER_CACHE.get_or_init(Default::default);
+        match matcher_cache.get_mut(&platform_id) {
+            Some(mut matcher) => matcher.extend_with_matcher(new_matcher),
+            None => {
+                // We still must uphold `from_platform` in case we are running this before the matcher workflow
+                // is kicked off. Other-wise we only will have the `new_matcher` data.
+                let mut matcher = Matcher::from_platform(platform);
+                matcher.extend_with_matcher(new_matcher);
+                matcher_cache.insert(platform_id, matcher);
+            }
+        }
+    }
+
+    fn valid(&self, _view: &BinaryView) -> bool {
+        true
+    }
+}