From 80b07dfa158bb093cc93e602363655b62ae595f8 Mon Sep 17 00:00:00 2001
From: Rusty Wagner <rusty.wagner@gmail.com>
Date: Fri, 22 Dec 2023 13:46:10 -0700
Subject: Support function recognizers in Rust architecture plugins

---
 rust/src/architecture.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'rust/src/architecture.rs')

diff --git a/rust/src/architecture.rs b/rust/src/architecture.rs
index 2c9843d9..edc16209 100644
--- a/rust/src/architecture.rs
+++ b/rust/src/architecture.rs
@@ -175,6 +175,7 @@ impl InstructionInfo {
     }
 }
 
+use crate::functionrecognizer::FunctionRecognizer;
 use crate::relocation::{CustomRelocationHandlerHandle, RelocationHandler};
 pub use binaryninjacore_sys::BNFlagRole as FlagRole;
 pub use binaryninjacore_sys::BNImplicitRegisterExtend as ImplicitRegisterExtend;
@@ -1646,6 +1647,13 @@ pub trait ArchitectureExt: Architecture {
     {
         crate::relocation::register_relocation_handler(self.as_ref(), name, func);
     }
+
+    fn register_function_recognizer<R>(&self, recognizer: R)
+    where
+        R: 'static + FunctionRecognizer + Send + Sync + Sized,
+    {
+        crate::functionrecognizer::register_arch_function_recognizer(self.as_ref(), recognizer);
+    }
 }
 
 impl<T: Architecture> ArchitectureExt for T {}
-- 
cgit v1.3.1