Added support for commentAttributes within the add_fields() rooutine

Signed-off-by: qcabrol

Author: cabrol

dissect/cstruct/cstruct.py

@@ -81,7 +81,7 @@ def _read(self, stream):
 
         return Instance(self, r, sizes)
 
-    def add_fields(self, name, type_, offset=None):
+    def add_fields(self, name, type_, offset=None, commentAttributes={}):
         raise NotImplementedError("Can't add fields to a compiled structure")
 
     def __repr__(self):
@@ -877,15 +877,15 @@ def _write(self, stream, data):
 
         return num
 
-    def add_field(self, name, type_, offset=None):
+    def add_fields(self, name, type_, offset=None, commentAttributes={}):
         """Add a field to this structure.
 
         Args:
             name: The field name.
             type_: The field type.
             offset: The field offset.
         """
-        field = Field(name, type_, offset=offset)
+        field = Field(name, type_, offset=offset, commentAttributes=commentAttributes)
         self.fields.append(field)
         self.lookup[name] = field
         self.size = None

tests/test_basic.py

@@ -731,9 +731,9 @@ def test_struct_sizes():
     """, compiled=False)
 
     assert len(c.static) == 4
-    c.static.add_field("another", c.uint32)
+    c.static.add_fields("another", c.uint32)
     assert len(c.static) == 8
-    c.static.add_field("atoffset", c.uint32, 12)
+    c.static.add_fields("atoffset", c.uint32, 12)
     assert len(c.static) == 16
 
     a = c.static(b'\x01\x00\x00\x00\x02\x00\x00\x00\x00\x00\x00\x00\x03\x00\x00\x00')