Merge pull request #8 from ACRIOS-Systems/feature/allowArrayOfStructures

This enables dumping structure data into bytearray for array of structures

dissect/cstruct/cstruct.py

@@ -965,6 +965,13 @@ def _write(self, stream, data):
 
         return num
 
+    def _write_0(self, stream, data):
+        num=0
+        for d in data:
+            num+=self._write(stream, d)
+            pass
+        return num
+
     def add_fields(self, name, type_, offset=None, commentAttributes:typeCommentAttributes={}):
         """Add a field to this structure.
 

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="dissect.cstruct",
-    version="1.0.0",
+    version="1.2.1",
     author="Fox-IT",
     description="Structure parsing in Python made easy.",
     long_description=long_description,

tests/test_basic.py

@@ -669,6 +669,10 @@ def test_struct_definitions():
         // uint32 comment
         uint32  b;
     } test, test1;
+
+    struct test_struct {
+        structure _test structure[];
+    };
     """, compiled=False)
 
     assert c._test == c.test == c.test1
@@ -678,6 +682,10 @@ def test_struct_definitions():
     assert 'a' in c.test.lookup
     assert 'b' in c.test.lookup
 
+    test_struct_data = b'\x01\x00\x00\x00\x02\x00\x00\x00\x03\x00\x00\x00\x04\x00\x00\x00'
+    test_struct = c.test_struct(structure=c.test[2](test_struct_data))
+    assert test_struct_data == test_struct.dumps()
+
     with pytest.raises(cstruct.ParserError):
         c.load("""
         struct {