From 01ef88f28a999ab16508483bd5a442f26876a81e Mon Sep 17 00:00:00 2001
From: Rubens Brandao <git@rubens.io>
Date: Thu, 20 Jun 2024 15:34:39 -0300
Subject: Implement Rust BaseAddressDetection

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

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

diff --git a/rust/src/binary_view.rs b/rust/src/binary_view.rs
index af4709e7..396c04ce 100644
--- a/rust/src/binary_view.rs
+++ b/rust/src/binary_view.rs
@@ -24,6 +24,7 @@
 use binaryninjacore_sys::*;
 
 use crate::architecture::{Architecture, CoreArchitecture};
+use crate::base_detection::BaseAddressDetection;
 use crate::basic_block::BasicBlock;
 use crate::component::{Component, IntoComponentGuid};
 use crate::confidence::Conf;
@@ -369,6 +370,13 @@ pub trait BinaryViewExt: BinaryViewBase {
         }
     }
 
+    fn base_address_detection(&self) -> Option<BaseAddressDetection> {
+        unsafe {
+            let handle = BNCreateBaseAddressDetection(self.as_ref().handle);
+            NonNull::new(handle).map(|base| BaseAddressDetection::from_raw(base))
+        }
+    }
+
     fn instruction_len<A: Architecture>(&self, arch: &A, addr: u64) -> Option<usize> {
         unsafe {
             let size = BNGetInstructionLength(self.as_ref().handle, arch.as_ref().handle, addr);
-- 
cgit v1.3.1