Merge pull request #907 from onekey-sec/relax-dissect-cstruct-annotation

chore(dissect.cstruct): Make Unblob compatible with version 4.0

Author: vlaci

poetry.lock

@@ -11,7 +11,7 @@ packages = [
 [tool.poetry.dependencies]
 python = "^3.8"
 click = "^8.1.7"
-"dissect.cstruct" = "^2.0"
+"dissect.cstruct" = ">=2.0,<5.0"
 attrs = ">=23.1.0"
 structlog = ">=24.1.0"
 arpy = "^2.3.0"

pyproject.toml

@@ -7,7 +7,14 @@
 from typing import Any
 
 import structlog
-from dissect.cstruct import Instance, dumpstruct
+from dissect.cstruct import dumpstruct
+
+try:
+    # dissect.cstruct >= 4.0
+    from dissect.cstruct import Structure
+except ImportError:
+    # dissect.cstruct == 2.0
+    from dissect.cstruct import Instance as Structure
 
 
 def format_hex(value: int):
@@ -42,7 +49,7 @@ def _format_message(value: Any, extract_root: Path) -> Any:
             new_value = value
         return new_value.as_posix().encode("utf-8", errors="surrogateescape")
 
-    if isinstance(value, Instance):
+    if isinstance(value, Structure):
         return dumpstruct(value, output="string")
 
     if isinstance(value, int):