Add way to set the id of the defined type in rust api

1 files changed, 19 insertions, 0 deletions

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 {