1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
from struct import unpack
from .constants import *
from .logging import *
def calculate_from_build_dir(root):
candidates = [
root / PYTHON_DLL_NAME,
root / FREETHREADED_PYTHON_DLL_NAME,
*root.glob("*.dll"),
*root.glob("*.pyd"),
# Check EXE last because it's easier to have cross-platform EXE
*root.glob("*.exe"),
]
ARCHS = {
b"PE\0\0\x4c\x01": "win32",
b"PE\0\0\x64\x86": "amd64",
b"PE\0\0\x64\xAA": "arm64"
}
first_exc = None
for pe in candidates:
try:
# Read the PE header to grab the machine type
with open(pe, "rb") as f:
f.seek(0x3C)
offset = int.from_bytes(f.read(4), "little")
f.seek(offset)
arch = ARCHS[f.read(6)]
except (FileNotFoundError, PermissionError, LookupError) as ex:
log_debug("Failed to open {}: {}", pe, ex)
continue
log_info("Inferred architecture {} from {}", arch, pe)
return arch
|
