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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
#!/usr/bin/env python3
#
# Thanks to @withzombies for letting us adapt his script
#
import importlib.util
import os
import sys
from site import check_enableusersite
# Handle both normal environments and virtualenvs
try:
from site import getusersitepackages, getsitepackages
except ImportError:
from sysconfig import get_path
getsitepackages = lambda: get_path('purelib')
getusersitepackages = getsitepackages
# Hacky command line parsing to accept a silent-install -s flag like linux-setup.sh:
INTERACTIVE = True
if '-s' in sys.argv[1:]:
INTERACTIVE = False
INSTALL_VENV = False
if '-v' in sys.argv[1:]:
if os.environ.get('VIRTUAL_ENV'):
INSTALL_VENV = True
else:
print("Error: venv installation requested without an active python3 venv.")
sys.exit(1)
try:
binaryninja = importlib.util.find_spec('binaryninja')
assert binaryninja is not None
binaryninjaui = importlib.util.find_spec('binaryninjaui')
assert binaryninjaui is not None
if '-u' in sys.argv[1:]:
#Uninstall mode
userpth = os.path.join(getusersitepackages(), 'binaryninja.pth')
if os.path.exists(userpth):
print(f"Removing {userpth}")
os.unlink(userpth)
sitepth = os.path.join(getsitepackages()[0], 'binaryninja.pth')
if os.path.exists(sitepth):
print(f"Removing {sitepth}")
try:
os.unlink(sitepth)
except:
print("Unable to unlink, please re-run with appropriate permissions.")
sys.exit(0)
print("Binary Ninja API already in the path")
sys.exit(1)
except ModuleNotFoundError:
pass
except AssertionError:
pass
if '-u' in sys.argv[1:]:
print("Uninstall specified, but binaryninja not in the current path")
sys.exit(1)
dir_name = os.path.dirname(os.path.abspath(__file__))
api_path = os.path.abspath(os.path.join(dir_name, "..", "python"))
if (os.path.isdir(api_path)):
print(f"Found install folder of {api_path}")
else:
print("Failed to find installed python expected at {}".format(api_path))
sys.exit(1)
def validate_path(path):
try:
os.stat(path)
except OSError:
return False
old_path = sys.path
sys.path.append(path)
try:
from binaryninja import core_version
print("Found Binary Ninja core version: {}".format(core_version()))
except ImportError:
sys.path = old_path
return False
return True
while not validate_path(api_path):
print("\nBinary Ninja not found.")
if not INTERACTIVE:
print("Non-interactive mode selected, failing.")
sys.exit(-1)
print("Please provide the path to Binary Ninja's install directory: \n [{}] : ".format(api_path))
new_path = sys.stdin.readline().strip()
if len(new_path) == 0:
print("\nInvalid path")
continue
if not new_path.endswith('python'):
new_path = os.path.join(new_path, 'python')
api_path = new_path
if (len(sys.argv) > 1 and sys.argv[1].lower() == "root"):
#write to root site
install_path = getsitepackages()[0]
if not os.access(install_path, os.W_OK):
print("Root install specified but cannot write to {}".format(install_path))
sys.exit(1)
elif INSTALL_VENV:
install_path = getsitepackages()[0]
else:
if check_enableusersite():
install_path = getusersitepackages()
if not os.path.exists(install_path):
os.makedirs(install_path)
else:
print("Warning, trying to write to user site packages, but check_enableusersite fails.")
sys.exit(1)
binaryninja_pth_path = os.path.join(install_path, 'binaryninja.pth')
with open(binaryninja_pth_path, 'wb') as pth_file:
pth_file.write((api_path + "\n").encode('charmap'))
pth_file.write((api_path + sys.version[0] + "\n").encode('charmap')) #support for QT bindings
print("Binary Ninja API installed using {}".format(binaryninja_pth_path))
|
