pyxcp 0.25.0__cp313-cp313-win_arm64.whl

pyxcp/dllif.py

@@ -0,0 +1,100 @@
+#!/usr/bin/env python
+import binascii
+import ctypes
+import enum
+import platform
+import re
+import subprocess  # nosec
+import sys
+from pathlib import Path
+
+
+class SeedNKeyResult(enum.IntEnum):
+    ACK = 0  # o.k.
+    ERR_PRIVILEGE_NOT_AVAILABLE = 1  # the requested privilege can not be unlocked with this DLL
+    ERR_INVALID_SEED_LENGTH = 2  # the seed length is wrong, key could not be computed
+    ERR_UNSUFFICIENT_KEY_LENGTH = 3  # the space for the key is too small
+
+    ERR_COULD_NOT_LOAD_DLL = 16
+    ERR_COULD_NOT_LOAD_FUNC = 17
+
+
+class SeedNKeyError(Exception):
+    """"""
+
+
+LOADER = Path(str(sys.modules["pyxcp"].__file__)).parent / "asamkeydll"  # Absolute path to DLL loader.
+
+bwidth, _ = platform.architecture()
+
+if sys.platform in ("win32", "linux", "darwin"):
+    if bwidth == "64bit":
+        use_ctypes = False
+    elif bwidth == "32bit":
+        use_ctypes = True
+else:
+    raise RuntimeError(f"Platform {sys.platform!r} currently not supported.")
+
+
+def getKey(logger, loader_cfg: str, dllName: str, privilege: int, seed: bytes, assume_same_bit_width: bool):
+    dllName = str(Path(dllName).absolute())  # Fix loader issues.
+
+    if loader_cfg is not None:
+        loader_exe = loader_cfg
+    else:
+        loader_exe = LOADER
+
+    use_ctypes: bool = False
+    if assume_same_bit_width:
+        use_ctypes = True
+    if use_ctypes:
+        try:
+            lib: ctypes.CDLL = ctypes.cdll.LoadLibrary(dllName)
+        except OSError:
+            logger.error(f"Could not load DLL {dllName!r} -- Probably an 64bit vs 32bit issue?")
+            return (SeedNKeyResult.ERR_COULD_NOT_LOAD_DLL, None)
+        func = lib.XCP_ComputeKeyFromSeed
+        func.restype = ctypes.c_uint32
+        func.argtypes = [
+            ctypes.c_uint8,
+            ctypes.c_uint8,
+            ctypes.c_char_p,
+            ctypes.POINTER(ctypes.c_uint8),
+            ctypes.c_char_p,
+        ]
+        key_buffer: ctypes.Array[ctypes.c_char] = ctypes.create_string_buffer(b"\000" * 128)
+        key_length: ctypes.c_uint8 = ctypes.c_uint8(128)
+        ret_code: int = func(
+            privilege,
+            len(seed),
+            ctypes.c_char_p(seed),
+            ctypes.byref(key_length),
+            key_buffer,
+        )
+        return (ret_code, key_buffer.raw[0 : key_length.value])
+    else:
+        try:
+            p0 = subprocess.Popen(
+                [loader_exe, dllName, str(privilege), binascii.hexlify(seed).decode("ascii")],
+                stdout=subprocess.PIPE,
+                shell=False,
+            )  # nosec
+        except FileNotFoundError as exc:
+            logger.error(f"Could not find executable {loader_exe!r} -- {exc}")
+            return (SeedNKeyResult.ERR_COULD_NOT_LOAD_DLL, None)
+        except OSError as exc:
+            logger.error(f"Cannot execute {loader_exe!r} -- {exc}")
+            return (SeedNKeyResult.ERR_COULD_NOT_LOAD_DLL, None)
+        key: bytes = b""
+        if p0.stdout:
+            key = p0.stdout.read()
+            p0.stdout.close()
+        p0.kill()
+        p0.wait()
+        if not key:
+            logger.error(f"Something went wrong while calling seed-and-key-DLL {dllName!r}")
+            return (SeedNKeyResult.ERR_COULD_NOT_LOAD_DLL, None)
+        res = re.split(b"\r?\n", key)
+        returnCode = int(res[0])
+        key = binascii.unhexlify(res[1])
+    return (returnCode, key)