Add ABI version verification for plugins

2 files changed, 24 insertions, 0 deletions

diff --git a/rust/binaryninjacore-sys/build.rs b/rust/binaryninjacore-sys/build.rs
index 99f499ea..24be4dac 100644
--- a/rust/binaryninjacore-sys/build.rs
+++ b/rust/binaryninjacore-sys/build.rs
@@ -154,6 +154,8 @@ fn main() {
         .size_t_is_usize(true)
         .generate_comments(false)
         .whitelist_function("BN.*")
+        .whitelist_var("BN_CURRENT_CORE_ABI_VERSION")
+        .whitelist_var("BN_MINIMUM_CORE_ABI_VERSION")
         .rustified_enum("BN.*")
         .generate()
         .expect("Unable to generate bindings");
diff --git a/rust/src/lib.rs b/rust/src/lib.rs
index bdef8368..3d287321 100644
--- a/rust/src/lib.rs
+++ b/rust/src/lib.rs
@@ -215,3 +215,25 @@ pub mod logger {
 pub fn version() -> string::BnString {
     unsafe { string::BnString::from_raw(binaryninjacore_sys::BNGetVersionString()) }
 }
+
+pub fn plugin_abi_version() -> u32 {
+    binaryninjacore_sys::BN_CURRENT_CORE_ABI_VERSION
+}
+
+pub fn plugin_abi_minimum_version() -> u32 {
+    binaryninjacore_sys::BN_MINIMUM_CORE_ABI_VERSION
+}
+
+pub fn core_abi_version() -> u32 {
+    unsafe { binaryninjacore_sys::BNGetCurrentCoreABIVersion() }
+}
+
+pub fn core_abi_minimum_version() -> u32 {
+    unsafe { binaryninjacore_sys::BNGetMinimumCoreABIVersion() }
+}
+
+// Provide ABI version automatically so that the core can verify binary compatibility
+#[cfg(not(feature = "headless"))]
+#[no_mangle]
+#[allow(non_snake_case)]
+pub extern "C" fn CorePluginABIVersion() -> u32 { plugin_abi_version() }