[Rust API] Debug for StructureBuilder

1 files changed, 31 insertions, 1 deletions

diff --git a/rust/src/types.rs b/rust/src/types.rs
index 90abeb41..746142a5 100644
--- a/rust/src/types.rs
+++ b/rust/src/types.rs
@@ -1460,6 +1460,22 @@ impl StructureBuilder {
         self
     }
 
+    pub fn set_alignment<'a>(&'a mut self, alignment: usize) -> &'a mut Self {
+        unsafe {
+            BNSetStructureBuilderAlignment(self.handle, alignment);
+        }
+
+        self
+    }
+
+    pub fn set_packed<'a>(&'a mut self, packed: bool) -> &'a mut Self {
+        unsafe {
+            BNSetStructureBuilderPacked(self.handle, packed);
+        }
+
+        self
+    }
+
     pub fn append<'a, 'b, S: BnStrCompatible, T: Into<Conf<&'b Type>>>(
         &'a mut self,
         t: T,
@@ -1534,11 +1550,19 @@ impl StructureBuilder {
         unsafe { BNGetStructureBuilderWidth(self.handle) }
     }
 
+    pub fn alignment(&self) -> usize {
+        unsafe { BNGetStructureBuilderAlignment(self.handle) }
+    }
+
+    pub fn packed(&self) -> bool {
+        unsafe { BNIsStructureBuilderPacked(self.handle) }
+    }
+
     pub fn structure_type(&self) -> StructureType {
         unsafe { BNGetStructureBuilderType(self.handle) }
     }
 
-    // TODO : The other methods in the python version (alignment, packed, type, members, remove, replace, etc)
+    // TODO : The other methods in the python version (type, members, remove, replace, etc)
 }
 
 impl From<&Structure> for StructureBuilder {
@@ -1547,6 +1571,12 @@ impl From<&Structure> for StructureBuilder {
     }
 }
 
+impl Debug for StructureBuilder {
+    fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
+        write!(f, "StructureBuilder {{ ... }}")
+    }
+}
+
 impl Drop for StructureBuilder {
     fn drop(&mut self) {
         unsafe { BNFreeStructureBuilder(self.handle) };