dissect.cstruct 4.0.dev7__tar.gz → 4.1.dev1__tar.gz

{dissect_cstruct-4.0.dev7/dissect.cstruct.egg-info → dissect_cstruct-4.1.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dissect.cstruct
-Version: 4.0.dev7
+Version: 4.1.dev1
 Summary: A Dissect module implementing a parser for C-like structures: structure parsing in Python made easy
 Author-email: Dissect Team <dissect@fox-it.com>
 License: Apache License 2.0

{dissect_cstruct-4.0.dev7 → dissect_cstruct-4.1.dev1}/dissect/cstruct/parser.py

@@ -251,8 +251,10 @@ class TokenParser(Parser):
             type_ = self.cstruct.resolve(self._identifier(tokens))
         elif tokens.next == self.TOK.STRUCT:
             type_ = self._struct(tokens)
+
             if tokens.next != self.TOK.NAME:
-                return Field(type_.__name__, type_)
+                type_, name, bits = self._parse_field_type(type_, type_.__name__)
+                return Field(name.strip(), type_, bits)
 
         if tokens.next != self.TOK.NAME:
             raise ParserError(f"line {self._lineno(tokens.next)}: expected name")

{dissect_cstruct-4.0.dev7 → dissect_cstruct-4.1.dev1/dissect.cstruct.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dissect.cstruct
-Version: 4.0.dev7
+Version: 4.1.dev1
 Summary: A Dissect module implementing a parser for C-like structures: structure parsing in Python made easy
 Author-email: Dissect Team <dissect@fox-it.com>
 License: Apache License 2.0

{dissect_cstruct-4.0.dev7 → dissect_cstruct-4.1.dev1}/tests/test_parser.py

@@ -8,6 +8,22 @@ from dissect.cstruct.parser import TokenParser
 from dissect.cstruct.types import ArrayMetaType, Pointer
 
 
+def test_nested_structs(cs: cstruct, compiled: bool) -> None:
+    cdef = """
+    struct nest {
+        struct {
+            uint32 b;
+        } a[4];
+    };
+    """
+
+    cs.load(cdef, compiled=compiled)
+    data = b"\x00\x00\x00\x00\x01\x00\x00\x00\x02\x00\x00\x00\x03\x00\x00\x00"
+    obj = cs.nest(data)
+    for i in range(0, 4):
+        assert obj.a[i].b == i
+
+
 def test_preserve_comment_newlines() -> None:
     cdef = """
     // normal comment