WARP: Add ability to save single function to file

This command will update or create a signature file with the selected function

1 files changed, 7 insertions, 0 deletions

diff --git a/plugins/warp/src/plugin.rs b/plugins/warp/src/plugin.rs
index 8500a016..224887ef 100644
--- a/plugins/warp/src/plugin.rs
+++ b/plugins/warp/src/plugin.rs
@@ -17,6 +17,7 @@ use log::LevelFilter;
 use warp::signature::function::constraints::FunctionConstraint;
 use warp::signature::function::Function as WarpFunction;
 
+mod add;
 mod copy;
 mod create;
 mod find;
@@ -245,5 +246,11 @@ pub extern "C" fn CorePluginInit() -> bool {
         create::CreateSignatureFile {},
     );
 
+    binaryninja::command::register_for_function(
+        "WARP\\Add Function Signature to File",
+        "Stores the signature for the function in the signature file",
+        add::AddFunctionSignature {},
+    );
+
     true
 }