Compatibility with cstruct v4

dissect/hypervisor/backup/c_vma.py

@@ -1,4 +1,4 @@
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 vma_def = """
 #define VMA_BLOCK_BITS          12
@@ -53,8 +53,7 @@
 };
 """
 
-c_vma = cstruct.cstruct(endian=">")
-c_vma.load(vma_def)
+c_vma = cstruct(endian=">").load(vma_def)
 
 
 VMA_MAGIC = b"VMA\x00"

dissect/hypervisor/descriptor/c_hyperv.py

@@ -1,4 +1,4 @@
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 hyperv_def = """
 /* ======== File header ======== */
@@ -112,8 +112,7 @@
 };
 """
 
-c_hyperv = cstruct.cstruct()
-c_hyperv.load(hyperv_def)
+c_hyperv = cstruct().load(hyperv_def)
 
 ObjectEntryType = c_hyperv.ObjectEntryType
 KeyDataType = c_hyperv.KeyDataType

dissect/hypervisor/disk/c_hdd.py

@@ -3,7 +3,7 @@
 # - https://github.com/qemu/qemu/blob/master/docs/interop/parallels.txt
 
 
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 hdd_def = """
 /* Compressed disk v1 signature */
@@ -62,7 +62,6 @@
 };
 """
 
-c_hdd = cstruct.cstruct()
-c_hdd.load(hdd_def)
+c_hdd = cstruct().load(hdd_def)
 
 SECTOR_SIZE = c_hdd.SECTOR_SIZE

dissect/hypervisor/disk/c_qcow2.py

@@ -1,4 +1,4 @@
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 qcow2_def = """
 #define MIN_CLUSTER_BITS 9
@@ -136,8 +136,7 @@
 };
 """
 
-c_qcow2 = cstruct.cstruct(endian=">")
-c_qcow2.load(qcow2_def)
+c_qcow2 = cstruct(endian=">").load(qcow2_def)
 
 QCOW2_MAGIC = 0x514649FB
 QCOW2_MAGIC_BYTES = c_qcow2.uint32.dumps(QCOW2_MAGIC)

dissect/hypervisor/disk/c_vdi.py

@@ -1,4 +1,4 @@
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 # https://www.virtualbox.org/browser/vbox/trunk/src/VBox/Storage/VDICore.h
 # https://forums.virtualbox.org/viewtopic.php?t=8046
@@ -81,8 +81,7 @@
 };
 """
 
-c_vdi = cstruct.cstruct()
-c_vdi.load(vdi_def)
+c_vdi = cstruct().load(vdi_def)
 
 VDI_SIGNATURE = 0xBEDA107F
 

dissect/hypervisor/disk/c_vhd.py

@@ -1,4 +1,4 @@
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 vhd_def = """
 struct footer {
@@ -38,14 +38,13 @@
     uint32          checksum;
     char            parent_unique_id[16];
     uint32          parent_timestamp;
-    uint32          reserved;
+    uint32          reserved1;
     char            parent_unicode_name[512];
     parent_locator  parent_locators[8];
-    char            reserved[256];
+    char            reserved2[256];
 };
 """
 
-c_vhd = cstruct.cstruct(endian=">")
-c_vhd.load(vhd_def)
+c_vhd = cstruct(endian=">").load(vhd_def)
 
 SECTOR_SIZE = 512

dissect/hypervisor/disk/c_vhdx.py

@@ -1,6 +1,6 @@
 from uuid import UUID
 
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 vhdx_def = """
 #define PAYLOAD_BLOCK_NOT_PRESENT           0
@@ -99,8 +99,7 @@
 };
 """
 
-c_vhdx = cstruct.cstruct()
-c_vhdx.load(vhdx_def)
+c_vhdx = cstruct().load(vhdx_def)
 
 ALIGNMENT = 64 * 1024
 MB = 1024 * 1024

dissect/hypervisor/disk/c_vmdk.py

@@ -1,6 +1,6 @@
 import struct
 
-from dissect import cstruct
+from dissect.cstruct import cstruct
 
 vmdk_def = """
 typedef struct {
@@ -132,8 +132,7 @@
 #define GRAIN_MARKER_PROGRESS           4
 """
 
-c_vmdk = cstruct.cstruct()
-c_vmdk.load(vmdk_def)
+c_vmdk = cstruct().load(vmdk_def)
 
 SECTOR_SIZE = 512
 

dissect/hypervisor/util/envelope.py

@@ -25,7 +25,7 @@
 except ImportError:
     HAS_PYCRYPTODOME = False
 
-from dissect import cstruct
+from dissect.cstruct import cstruct
 from dissect.util.stream import RangeStream
 
 c_def = """
@@ -69,8 +69,7 @@
     Bytes = 0xC
 };
 """
-c_envelope = cstruct.cstruct()
-c_envelope.load(c_def)
+c_envelope = cstruct().load(c_def)
 
 FILE_HEADER_MAGIC = b"DataTransformEnvelope"
 FOOTER_AEAD_MAGIC = b"DataTransformAeadFooter"

pyproject.toml

@@ -26,8 +26,8 @@ classifiers = [
 ]
 dependencies = [
     "defusedxml",
-    "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

@@ -11,11 +11,17 @@ minversion = 4.4.3
 requires = virtualenv>=20.16.6
 
 [testenv]
-extras = full
 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 .[full]
 commands =
     pytest --basetemp="{envtmpdir}" {posargs:--color=yes --cov=dissect --cov-report=term-missing -v tests}
     coverage report