[Rust] Provide setters useful for creating NTR references

Need to expose these setters, even if there usage is limited when dealing with NTR reference, something in the core might be unconditionally
retrieving the width or alignment of the type without trying to resolve NTR.

1 files changed, 17 insertions, 0 deletions

diff --git a/rust/src/types.rs b/rust/src/types.rs
index 4f6f82d2..bb5b3883 100644
--- a/rust/src/types.rs
+++ b/rust/src/types.rs
@@ -123,6 +123,22 @@ impl TypeBuilder {
         self
     }
 
+    /// Set the width of the type.
+    ///
+    /// Typically only done for named type references, which will not have their width set otherwise.
+    pub fn set_width(&self, width: usize) -> &Self {
+        unsafe { BNTypeBuilderSetWidth(self.handle, width) }
+        self
+    }
+
+    /// Set the alignment of the type.
+    ///
+    /// Typically only done for named type references, which will not have their alignment set otherwise.
+    pub fn set_alignment(&self, alignment: usize) -> &Self {
+        unsafe { BNTypeBuilderSetAlignment(self.handle, alignment) }
+        self
+    }
+
     pub fn set_pointer_base(&self, base_type: PointerBaseType, base_offset: i64) -> &Self {
         unsafe { BNSetTypeBuilderPointerBase(self.handle, base_type, base_offset) }
         self
@@ -541,6 +557,7 @@ impl Type {
 
     // TODO: We need to decide on a public type to represent type width.
     // TODO: The api uses both `u64` and `usize`, pick one or a new type!
+    /// The size of the type in bytes.
     pub fn width(&self) -> u64 {
         unsafe { BNGetTypeWidth(self.handle) }
     }