From e7895a2c36189bc1c3010a48b47656e861607640 Mon Sep 17 00:00:00 2001
From: Mason Reed <mason@vector35.com>
Date: Sat, 19 Oct 2024 19:12:30 -0400
Subject: Add way to set the id of the defined type in rust api

---
 rust/src/binaryview.rs | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

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

diff --git a/rust/src/binaryview.rs b/rust/src/binaryview.rs
index f1006d57..f04d7da5 100644
--- a/rust/src/binaryview.rs
+++ b/rust/src/binaryview.rs
@@ -620,6 +620,25 @@ pub trait BinaryViewExt: BinaryViewBase {
         QualifiedName(name_handle)
     }
 
+    fn define_auto_type_with_id<S: BnStrCompatible>(
+        &self,
+        name: S,
+        id: S,
+        type_obj: &Type,
+    ) -> QualifiedName {
+        let mut qualified_name = QualifiedName::from(name);
+        let id_str = id.into_bytes_with_nul();
+        let name_handle = unsafe {
+            BNDefineAnalysisType(
+                self.as_ref().handle,
+                id_str.as_ref().as_ptr() as *const _,
+                &mut qualified_name.0,
+                type_obj.handle,
+            )
+        };
+        QualifiedName(name_handle)
+    }
+
     fn define_user_type<S: BnStrCompatible>(&self, name: S, type_obj: &Type) {
         let mut qualified_name = QualifiedName::from(name);
         unsafe {
-- 
cgit v1.3.1