dissect.cstruct 4.8.dev1__tar.gz → 4.8.dev3__tar.gz

{dissect_cstruct-4.8.dev1 → dissect_cstruct-4.8.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dissect.cstruct
-Version: 4.8.dev1
+Version: 4.8.dev3
 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-Expression: Apache-2.0

{dissect_cstruct-4.8.dev1 → dissect_cstruct-4.8.dev3}/dissect.cstruct.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dissect.cstruct
-Version: 4.8.dev1
+Version: 4.8.dev3
 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-Expression: Apache-2.0

{dissect_cstruct-4.8.dev1 → dissect_cstruct-4.8.dev3}/pyproject.toml

@@ -106,6 +106,7 @@ select = [
 ignore = ["E203", "B904", "UP024", "ANN002", "ANN003", "ANN204", "ANN401", "SIM105", "TRY003"]
 
 [tool.ruff.lint.per-file-ignores]
+"tests/**" = ["S101", "PLC0415"]
 "tests/_docs/**" = ["INP001"]
 
 [tool.ruff.lint.isort]

{dissect_cstruct-4.8.dev1 → dissect_cstruct-4.8.dev3}/tests/test_annotations.py

@@ -21,7 +21,7 @@ def test_cstruct_type_annotation(name: str, monkeypatch: pytest.MonkeyPatch) ->
         for module in [module for module in sys.modules if module in ("dissect.cstruct.cstruct")]:
             monkeypatch.delitem(sys.modules, module)
 
-        from dissect.cstruct import cstruct  # noqa: PLC0415
+        from dissect.cstruct import cstruct
 
         if name.startswith("__"):
             name = f"_cstruct{name}"

{dissect_cstruct-4.8.dev1 → dissect_cstruct-4.8.dev3}/tests/test_benchmark.py

@@ -1,9 +1,12 @@
 from __future__ import annotations
 
+from functools import partial
 from typing import TYPE_CHECKING
 
 import pytest
 
+from dissect.cstruct.expression import Expression
+
 if TYPE_CHECKING:
     from pytest_benchmark.fixture import BenchmarkFixture
 
@@ -85,3 +88,89 @@ def test_benchmark_getattr_typedefs(cs: cstruct, benchmark: BenchmarkFixture) ->
     cs.load(cdef)
 
     benchmark(lambda: cs.my_uint8)
+
+
+@pytest.mark.benchmark
+def test_benchmark_expression_parse(cs: cstruct, benchmark: BenchmarkFixture) -> None:
+    """Benchmark the parsing of expressions."""
+    cs.load("#define a 5\n#define b 10")
+
+    benchmark(lambda: Expression("a * 2 + b * (3 + 4) >> 1"))
+
+
+@pytest.mark.benchmark
+def test_benchmark_expression_evaluate(cs: cstruct, benchmark: BenchmarkFixture) -> None:
+    """Benchmark the evaluation of expressions."""
+    cs.load("#define a 5\n#define b 10")
+
+    expression = Expression("a * 2 + b * (3 + 4) >> 1")
+    benchmark(lambda: expression.evaluate(cs))
+
+
+@pytest.mark.benchmark
+def test_benchmark_expression_parse_and_evaluate(cs: cstruct, benchmark: BenchmarkFixture) -> None:
+    """Benchmark the parsing and evaluation of expressions."""
+    cs.load("#define a 5\n#define b 10")
+
+    benchmark(lambda: Expression("a * 2 + b * (3 + 4) >> 1").evaluate(cs))
+
+
+_BENCHMARK_CDEF = """
+struct SECURITY_DESCRIPTOR {
+    uint8   Revision;
+    uint8   Sbz1;
+    uint16  Control;
+    uint32  OffsetOwner;
+    uint32  OffsetGroup;
+    uint32  OffsetSacl;
+    uint32  OffsetDacl;
+};
+
+struct LDAP_SID_IDENTIFIER_AUTHORITY {
+    char    Value[6];
+};
+
+struct LDAP_SID {
+    uint8   Revision;
+    uint8   SubAuthorityCount;
+    LDAP_SID_IDENTIFIER_AUTHORITY   IdentifierAuthority;
+    uint32  SubAuthority[SubAuthorityCount];
+};
+
+struct ACL {
+    uint8   AclRevision;
+    uint8   Sbz1;
+    uint16  AclSize;
+    uint16  AceCount;
+    uint16  Sbz2;
+    char    Data[AclSize - 8];
+};
+
+struct ACE {
+    uint8   AceType;
+    uint8   AceFlags;
+    uint16  AceSize;
+    char    Data[AceSize - 4];
+};
+
+struct ACCESS_ALLOWED_ACE {
+    uint16  Mask;
+    LDAP_SID Sid;
+};
+
+struct ACCESS_ALLOWED_OBJECT_ACE {
+    uint32  Mask;
+    uint32  Flags;
+    char    ObjectType[(Flags & 1) * 16];
+    char    InheritedObjectType[(Flags & 2) * 8];
+    LDAP_SID Sid;
+};
+"""
+
+
+@pytest.mark.benchmark
+def test_benchmark_lexer_and_parser(cs: cstruct, benchmark: BenchmarkFixture) -> None:
+    """Benchmark tokenizing and parsing a realistic set of struct definitions."""
+    cs.add_type = partial(cs.add_type, replace=True)
+
+    benchmark(lambda: cs.load(_BENCHMARK_CDEF))