Rust bindings for custom basic block analysis

1 files changed, 31 insertions, 0 deletions

diff --git a/rust/src/function.rs b/rust/src/function.rs
index c4061328..692eb817 100644
--- a/rust/src/function.rs
+++ b/rust/src/function.rs
@@ -2992,3 +2992,34 @@ unsafe impl CoreArrayProviderInner for Comment {
         }
     }
 }
+
+#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq)]
+pub struct ArchAndAddr {
+    pub arch: CoreArchitecture,
+    pub addr: u64,
+}
+
+impl ArchAndAddr {
+    pub fn new(arch: CoreArchitecture, addr: u64) -> Self {
+        Self { arch, addr }
+    }
+}
+
+impl From<BNArchitectureAndAddress> for ArchAndAddr {
+    fn from(raw: BNArchitectureAndAddress) -> Self {
+        unsafe {
+            let arch = CoreArchitecture::from_raw(raw.arch);
+            let addr = raw.address;
+            ArchAndAddr { arch, addr }
+        }
+    }
+}
+
+impl ArchAndAddr {
+    pub fn into_raw(self) -> BNArchitectureAndAddress {
+        BNArchitectureAndAddress {
+            arch: self.arch.handle,
+            address: self.addr,
+        }
+    }
+}