-
Notifications
You must be signed in to change notification settings - Fork 9
/
__init__.py
99 lines (71 loc) · 2.3 KB
/
__init__.py
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
from pathlib import Path
from sys import platform
from cffi import FFI
from platform import architecture, machine
_ffi_def = """
extern char* ffiverify(char* proofQREncoded, char* configpath);
extern void freeCString(char* s);
"""
def _libpath():
return Path(__file__).parent.absolute()
def listlibs():
return [x.name for x in list(_libpath().glob('*.so')) + list(_libpath().glob('*.dll'))]
def loadlib(lib='auto'):
if lib == 'auto':
lib = _autodetect()
lib = _libpath() / lib
if not lib.is_file():
raise ValueError(f'Could not find verifier lib {lib.name} choose one of: {", ".join(listlibs())}')
ffi = FFI()
ffi.cdef(_ffi_def)
verifier = ffi.dlopen(str(lib.absolute()))
return verifier, ffi
def _autodetect():
libos = None
libext = '.so'
if platform.startswith("linux"):
libos = 'linux'
elif platform.startswith("darwin"):
libos = 'darwin'
elif platform.startswith("win32"):
libos = "windows"
libext = '.dll'
if libos is None:
raise ValueError(f'Auto detect failed OS unknown: {platform}')
libarch = None
arch = architecture()
mach = machine().lower()
if 'arm' in mach:
if 'v7' in mach:
libarch = 'armv7'
elif 'v6' in mach:
libarch = 'armv6'
# detect rpi
if libos == 'linux':
chips = ('BCM2835', )
with open('/proc/cpuinfo', 'r') as cpuinfo:
for line in cpuinfo:
if line.startswith('Hardware') and any([chip in line for chip in chips]):
libarch = 'armv6l'
elif 'v5' in mach:
libarch = 'armv5'
elif '64' in mach:
libarch = 'arm64'
if 'aarch64' in mach:
libarch = 'arm64'
if 'x86' in mach:
if '64' in mach:
libarch = 'amd64'
else:
libarch = '386'
if 'amd64' in mach:
libarch = 'amd64'
b32arches = ['i686', 'i386', 'i486']
if mach.lower() in b32arches:
if '64' in mach:
libarch = 'amd64'
else:
libarch = '386'
if libarch is None:
raise ValueError(f'Auto detect failed CPU Architecture unknown: {arch} {mach}')
return f"verifier-{libos}-{libarch}{libext}"