dissect.cstruct 4.6.dev3__tar.gz → 4.6.dev4__tar.gz

{dissect_cstruct-4.6.dev3 → dissect_cstruct-4.6.dev4}/CHANGELOG.md

@@ -11,6 +11,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/).
 - Optimize storage of field sizes.
 - Rename `_sizes` property of `Structure` to `__sizes__`.
 - Rename `_values` property of `Structure` to `__values__`.
+- Added `load` argument to `cstruct` class, allowing direct initialization with a definition (i.e. `cstruct(cdef)` instead of `cstruct().load(cdef)`. Other arguments to `cstruct` are now keyword only.
 
 ## [4.5] - 20-05-2025
 

{dissect_cstruct-4.6.dev3 → dissect_cstruct-4.6.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dissect.cstruct
-Version: 4.6.dev3
+Version: 4.6.dev4
 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.6.dev3 → dissect_cstruct-4.6.dev4}/dissect/cstruct/cstruct.py

@@ -48,7 +48,7 @@ class cstruct:
     DEF_CSTYLE = 1
     DEF_LEGACY = 2
 
-    def __init__(self, endian: str = "<", pointer: str | None = None):
+    def __init__(self, load: str = "", *, endian: str = "<", pointer: str | None = None):
         self.endian = endian
 
         self.consts = {}
@@ -188,6 +188,9 @@ class cstruct:
         self.pointer: type[BaseType] = self.resolve(pointer)
         self._anonymous_count = 0
 
+        if load:
+            self.load(load)
+
     def __getattr__(self, attr: str) -> Any:
         try:
             return self.consts[attr]

{dissect_cstruct-4.6.dev3 → dissect_cstruct-4.6.dev4}/dissect.cstruct.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dissect.cstruct
-Version: 4.6.dev3
+Version: 4.6.dev4
 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.6.dev3 → dissect_cstruct-4.6.dev4}/tests/test_basic.py

@@ -6,6 +6,7 @@ from typing import TYPE_CHECKING, BinaryIO
 
 import pytest
 
+from dissect.cstruct.cstruct import cstruct
 from dissect.cstruct.exceptions import ArraySizeError, ParserError, ResolveError
 from dissect.cstruct.types import BaseType
 
@@ -14,8 +15,6 @@ from .utils import verify_compiled
 if TYPE_CHECKING:
     from pathlib import Path
 
-    from dissect.cstruct.cstruct import cstruct
-
 
 def test_duplicate_type(cs: cstruct, compiled: bool) -> None:
     cdef = """
@@ -41,6 +40,51 @@ def test_load_file(cs: cstruct, compiled: bool, tmp_path: Path) -> None:
     assert "test" in cs.typedefs
 
 
+def test_load_init() -> None:
+    cdef = """
+    struct test {
+        DWORD   a;
+        QWORD   b;
+    };
+    """
+    # load with first positional argument
+    cs = cstruct(cdef)
+    assert "test" in cs.typedefs
+    assert cs.endian == "<"
+
+    # load from keyword argument and big endian
+    cs = cstruct(load=cdef, endian=">")
+    assert "test" in cs.typedefs
+    a = cs.test(a=0xBADC0DE, b=0xACCE55ED)
+    assert len(bytes(a)) == 12
+    assert bytes(a) == a.dumps()
+    assert bytes(a) == b"\x0b\xad\xc0\xde\x00\x00\x00\x00\xac\xce\x55\xed"
+
+    # load using positional argument and little endian
+    cs = cstruct(cdef, endian="<")
+    assert "test" in cs.typedefs
+    a = cs.test(a=0xBADC0DE, b=0xACCE55ED)
+    assert len(bytes(a)) == 12
+    assert bytes(a) == a.dumps()
+    assert bytes(a) == b"\xde\xc0\xad\x0b\xed\x55\xce\xac\x00\x00\x00\x00"
+
+
+def test_load_init_kwargs_only() -> None:
+    cdef = """
+    struct test {
+        uint32  a;
+    };
+    """
+
+    # kwargs only check
+    with pytest.raises(TypeError, match="takes from .* positional arguments but .* were given"):
+        cs = cstruct(cdef, ">")
+
+    cs = cstruct(cdef, endian=">")
+    assert "test" in cs.typedefs
+    assert cs.endian == ">"
+
+
 def test_read_type_name(cs: cstruct) -> None:
     assert cs.read("uint32", b"\x01\x00\x00\x00") == 1