Compatibility with cstruct v4

dissect/extfs/c_ext.py

@@ -1,6 +1,6 @@
 import stat
 
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 ext_def = """
 #define EXT2_SBOFF              1024        // offset to superblock
@@ -397,8 +397,7 @@
 };
 """
 
-c_ext = cstruct.cstruct()
-c_ext.load(ext_def)
+c_ext = cstruct().load(ext_def)
 
 EXT2 = 2
 EXT3 = 3

dissect/extfs/c_jdb2.py

@@ -1,4 +1,4 @@
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 jdb2_def = """
 #define JBD2_MAGIC_NUMBER                   0xC03B3998
@@ -87,5 +87,4 @@
 };
 """
 
-c_jdb2 = cstruct.cstruct(endian=">")
-c_jdb2.load(jdb2_def)
+c_jdb2 = cstruct(endian=">").load(jdb2_def)

dissect/extfs/journal.py

@@ -4,7 +4,6 @@
 import io
 from typing import BinaryIO, Iterator, Optional
 
-from dissect.cstruct import Instance
 from dissect.util.stream import RangeStream
 
 from dissect.extfs.c_jdb2 import c_jdb2
@@ -83,7 +82,7 @@ def walk(self) -> Iterator[CommitBlock]:
 
 
 class DescriptorBlock:
-    def __init__(self, jdb2: JDB2, header: Instance, block: int):
+    def __init__(self, jdb2: JDB2, header: c_jdb2.journal_header, block: int):
         self.jdb2 = jdb2
         self.header = header
         self.journal_block = block
@@ -110,7 +109,9 @@ def tags(self) -> Iterator[DescriptorBlockTag]:
 
 
 class DescriptorBlockTag:
-    def __init__(self, descriptor: DescriptorBlock, tag: Instance, journal_block: int):
+    def __init__(
+        self, descriptor: DescriptorBlock, tag: c_jdb2.journal_block_tag | c_jdb2.journal_block_tag3, journal_block: int
+    ):
         self.descriptor = descriptor
         self.tag = tag
         self.journal_block = journal_block
@@ -127,7 +128,11 @@ def open(self) -> BinaryIO:
 
 class CommitBlock:
     def __init__(
-        self, jdb2: JDB2, header: Instance, journal_block: int, descriptors: Optional[list[DescriptorBlock]] = None
+        self,
+        jdb2: JDB2,
+        header: c_jdb2.commit_header,
+        journal_block: int,
+        descriptors: Optional[list[DescriptorBlock]] = None,
     ):
         self.jdb2 = jdb2
         self.header = header

pyproject.toml

@@ -25,8 +25,8 @@ classifiers = [
   "Topic :: Utilities",
 ]
 dependencies = [
-    "dissect.cstruct>=3.0.dev,<4.0.dev",
-    "dissect.util>=3.0.dev,<4.0.dev",
+    "dissect.cstruct>3,<5",
+    "dissect.util>2,<4",
 ]
 dynamic = ["version"]
 

tox.ini

@@ -15,6 +15,13 @@ deps =
     pytest
     pytest-cov
     coverage
+# Unfortunately, tox does not allow separate installation flags for the project
+# dependencies and the test dependencies. When running tox, we want to install the
+# project dependencies with the --pre flag, so that we get the latest version of all
+# dependencies. We do the installation step ourselves for this reason.
+skip_install = true
+commands_pre =
+    pip install --pre -e .
 commands =
     pytest --basetemp="{envtmpdir}" {posargs:--color=yes --cov=dissect --cov-report=term-missing -v tests}
     coverage report