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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
|
import unittest
import time
import platform
import os
import tempfile
import zipfile
import urllib
import subprocess
import getpass
from binaryninja.setting import Setting
from binaryninja.metadata import Metadata
from binaryninja.demangle import demangle_gnu3, get_qualified_name
from binaryninja.architecture import Architecture
from binaryninja.binaryview import BinaryViewType
class SettingsAPI(unittest.TestCase):
@classmethod
def setUpClass(cls):
pass
@classmethod
def tearDownClass(cls):
setting = Setting("test")
setting.remove_setting_group("test")
def test_bool_settings(self):
setting = Setting("test")
setting.set("bool_test_true", True)
setting.set("bool_test_false", False)
assert not setting.get_bool("bool_test_false"), "bool_test_false failed"
assert setting.get_bool("bool_test_true"), "bool_test_true failed"
assert setting.get_bool("bool_test_default_True", True), "bool_test_default_True failed"
assert not setting.get_bool("bool_test_default_False", False), "bool_test_default_False failed"
def test_int_settings(self):
setting = Setting("test")
setting.set("int_test1", 0x100)
setting.set("int_test2", 0)
setting.set("int_test3", -1)
assert setting.get_integer("int_test1") == 0x100, "int_test1 failed"
assert setting.get_integer("int_test2") == 0, "int_test2 failed"
assert setting.get_integer("int_test3") == -1, "int_test3 failed"
assert setting.get_integer("int_test_default_1", 1) == 1, "int_test_default_1 failed"
def test_float_settings(self):
setting = Setting("test")
setting.set("float_test1", 10.5)
setting.set("float_test2", -0.5)
assert setting.get_double("float_test1") == 10.5, "float_test1 failed"
assert setting.get_double("float_test2") == -0.5, "float_test1 failed"
assert setting.get_double("float_test_default", -5.5), "float_test_default failed"
def test_str_settings(self):
setting = Setting("test")
setting.set("str_test1", "hi")
setting.set("str_test2", "")
setting.set("str_test3", "A" * 1000)
assert setting.get_string("str_test1") == "hi", "str_test1 failed"
assert setting.get_string("str_test2") == "", "str_test2 failed"
assert setting.get_string("str_test3") == "A" * 1000, "str_test3 failed"
assert setting.get_string("str_test_default", "hi") == "hi", "str_test_default failed"
def test_int_list_settings(self):
setting = Setting("test")
setting.set("int_list_test1", [0x100])
setting.set("int_list_test2", [1, 2])
setting.set("int_list_test3", [])
assert setting.get_integer_list("int_list_test1") == [0x100], "int_list_test1 failed"
assert setting.get_integer_list("int_list_test2") == [1, 2], "int_list_test2 failed"
assert setting.get_integer_list("int_list_test3") == [], "int_list_test3 failed"
assert setting.get_integer_list("int_list_test_default", [2, 3]), "int_list_test_default failed"
def test_str_list_settings(self):
setting = Setting("test")
setting.set("str_list_test1", ["hi"])
setting.set("str_list_test2", ["hello", "world"])
setting.set("str_list_test3", [])
assert setting.get_string_list("str_list_test1") == ["hi"], "str_list_test1 failed"
assert setting.get_string_list("str_list_test2") == ["hello", "world"], "str_list_test2 failed"
assert setting.get_string_list("str_list_test3") == [], "str_list_test3 failed"
assert setting.get_string_list("str_list_test_default", ["hi", "there"]), "str_list_test_default failed"
class MetaddataAPI(unittest.TestCase):
def test_metadata_basic_types(self):
# Core is tested thoroughly through the C++ unit tests here we focus on the python api side
md = Metadata(1)
assert md.is_integer
assert int(md) == 1
assert md.value == 1
md = Metadata(-1, signed=True)
assert md.is_signed_integer
assert int(md) == -1
assert md.value == -1
md = Metadata(1, signed=False)
assert md.is_unsigned_integer
assert int(md) == 1
md = Metadata(3.14)
assert md.is_float
assert float(md) == 3.14
assert md.value == 3.14
md = Metadata("asdf")
assert md.is_string
assert str(md) == "asdf"
assert len(md) == 4
assert md.value == "asdf"
md = Metadata("\x00\x00\x41\x00", raw=True)
assert md.is_raw
assert len(md) == 4
assert str(md) == "\x00\x00\x41\x00"
assert md.value == "\x00\x00\x41\x00"
def test_metadata_compound_types(self):
md = Metadata([1, 2, 3])
assert md.is_array
assert md.value == [1, 2, 3]
assert len(md) == 3
assert md[0] == 1
assert md[1] == 2
assert md[2] == 3
assert isinstance(list(md), list)
md.remove(0)
assert len(md) == 2
assert md == [2, 3]
md = Metadata({"a": 1, "b": 2})
assert md.is_dict
assert len(md) == 2
assert md.value == {"a": 1, "b": 2}
assert md["a"] == 1
assert md["b"] == 2
md.remove("a")
assert len(md) == 1
assert md == {"b": 2}
def test_metadata_equality(self):
assert Metadata(1) == 1
assert Metadata(1) != 0
assert Metadata(1) == Metadata(1)
assert Metadata(1) != Metadata(0)
assert Metadata(3.14) == 3.14
assert Metadata(3.14) == Metadata(3.14)
assert Metadata(3.14) != 3.1
assert Metadata(3.14) != Metadata(3.1)
assert Metadata("asdf") == "asdf"
assert Metadata("asdf") == Metadata("asdf")
assert Metadata("asdf") != "qwer"
assert Metadata("asdf") != Metadata("qwer")
assert Metadata("as\x00df", raw=True) == "as\x00df"
assert Metadata("as\x00df", raw=True) == Metadata("as\x00df", raw=True)
assert Metadata("as\x00df", raw=True) != "qw\x00er"
assert Metadata("as\x00df", raw=True) != Metadata("qw\x00er", raw=True)
assert Metadata([1, 2, 3]) == [1, 2, 3]
assert Metadata([1, 2, 3]) == Metadata([1, 2, 3])
assert Metadata([1, 2, 3]) != [1, 2]
assert Metadata([1, 2, 3]) != Metadata([1, 2])
assert Metadata({"a": 1, "b": 2}) == {"a": 1, "b": 2}
assert Metadata({"a": 1, "b": 2}) == Metadata({"a": 1, "b": 2})
assert Metadata({"a": 1, "b": 2}) != {"a": 1}
assert Metadata({"a": 1, "b": 2}) != Metadata({"a": 1})
class DemanglerTest(unittest.TestCase):
def get_type_string(self, t, n):
out = ""
if t is not None:
out = str(t.get_string_before_name())
if len(out) > 1 and out[-1] != ' ':
out += " "
out += get_qualified_name(n)
out += str(t.get_string_after_name())
return out
def test_demangle_gnu3(self):
tests = ("__ZN15BinaryNinjaCore12BinaryReader5Read8Ev",
"__ZN5QListIP18QAbstractAnimationE18detach_helper_growEii",
"__ZN13QStatePrivate22emitPropertiesAssignedEv",
"__ZN17QtMetaTypePrivate23QMetaTypeFunctionHelperI14QItemSelectionLb1EE9ConstructEPvPKv",
"__ZN18QSharedDataPointerI16QFileInfoPrivateE4dataEv",
"__ZN26QAbstractNativeEventFilterD2Ev",
"__ZN5QListIP14QAbstractStateE3endEv",
"__ZNK15BinaryNinjaCore19ArchitectureWrapper22GetOpcodeDisplayLengthEv",
"__ZN15BinaryNinjaCore17ScriptingInstance19SetCurrentSelectionEyy",
"__ZL32qt_meta_stringdata_QHistoryState",
"__ZN12_GLOBAL__N_114TypeDestructor14DestructorImplI11QStringListLb1EE8DestructEiPv",
"__ZN13QGb18030Codec5_nameEv",
"__ZN5QListIP7QObjectE6detachEv",
"__ZN19QBasicAtomicPointerI9QFreeListI13QMutexPrivateN12_GLOBAL__N_117FreeListConstantsEEE17testAndSetReleaseEPS4_S6_",
"__ZN12QJsonPrivate6Parser12reserveSpaceEi",
"__ZN20QStateMachinePrivate12endMacrostepEb",
"__ZN14QScopedPointerI20QTemporaryDirPrivate21QScopedPointerDeleterIS0_EED2Ev",
"__ZN14QVariantIsNullIN12_GLOBAL__N_115CoreTypesFilterEE8delegateI10QMatrix4x4EEbPKT_",
"__ZN26QAbstractProxyModelPrivateC2Ev",
"__ZNSt3__110__function6__funcIZ26BNWorkerInteractiveEnqueueE4$_16NS_9allocatorIS2_EEFvvEEclEv")
results = ("int32_t BinaryNinjaCore::BinaryReader::Read8()",
"int32_t QList<QAbstractAnimation*>::detach_helper_grow(int32_t, int32_t)",
"int32_t QStatePrivate::emitPropertiesAssigned()",
"int32_t QtMetaTypePrivate::QMetaTypeFunctionHelper<QItemSelection, true>::Construct(void*, void const*)",
"int32_t QSharedDataPointer<QFileInfoPrivate>::data()",
"void QAbstractNativeEventFilter::~QAbstractNativeEventFilter()",
"int32_t QList<QAbstractState*>::end()",
"int32_t BinaryNinjaCore::ArchitectureWrapper::GetOpcodeDisplayLength() const",
"int32_t BinaryNinjaCore::ScriptingInstance::SetCurrentSelection(uint64_t, uint64_t)",
"qt_meta_stringdata_QHistoryState",
"int32_t (anonymous namespace)::TypeDestructor::DestructorImpl<QStringList, true>::Destruct(int32_t, void*)",
"int32_t QGb18030Codec::_name()",
"int32_t QList<QObject*>::detach()",
"int32_t QBasicAtomicPointer<QFreeList<QMutexPrivate, (anonymous namespace)::FreeListConstants> >::testAndSetRelease(QFreeList<QMutexPrivate, (anonymous namespace)::FreeListConstants>*, QFreeList<QMutexPrivate, (anonymous namespace)::FreeListConstants>*)",
"int32_t QJsonPrivate::Parser::reserveSpace(int32_t)",
"int32_t QStateMachinePrivate::endMacrostep(bool)",
"void QScopedPointer<QTemporaryDirPrivate, QScopedPointerDeleter<QTemporaryDirPrivate> >::~QScopedPointer()",
"bool QVariantIsNull<(anonymous namespace)::CoreTypesFilter>::delegate<QMatrix4x4>(QMatrix4x4 const*)",
"void QAbstractProxyModelPrivate::QAbstractProxyModelPrivate()",
"int32_t std::__1::__function::__func<BNWorkerInteractiveEnqueue::$_16, std::__1::allocator<BNWorkerInteractiveEnqueue::$_16>, void ()>::operator()()")
for i, test in enumerate(tests):
t, n = demangle_gnu3(Architecture['x86'], test)
assert self.get_type_string(t, n) == results[i]
class TimingTest(unittest.TestCase):
def unpackage_file(self, file):
if not os.path.exists(file):
with zipfile.ZipFile(file + ".zip", "r") as zf:
zf.extractall()
assert os.path.exists(file)
def test_analysis_time(self):
if platform.system() != "Linux" or getpass.getuser() != "jenkins":
return
start_time = time.time()
file_names = [os.path.join(os.path.dirname(__file__), "binaries", "quick3dcoreplugin.dll"),
os.path.join(os.path.dirname(__file__), "binaries", "md5"),
os.path.join(os.path.dirname(__file__), "binaries", "ls")]
for file_name in file_names:
temp_name = next(tempfile._get_candidate_names()) + ".bndb"
self.unpackage_file(file_name)
try:
bv = BinaryViewType.get_view_of_file(file_name)
bv.create_database(temp_name)
bv.file.close()
bv = BinaryViewType.get_view_of_file(temp_name)
bv.file.close()
finally:
if os.path.exists(file_name):
os.unlink(file_name)
if os.path.exists(temp_name):
os.unlink(temp_name)
time_s = time.time() - start_time
commit = subprocess.check_output(["git", "rev-parse", "HEAD"])[:-1]
conn = urllib.urlopen("https://script.google.com/macros/s/AKfycbxrZtlgLaWt3l95_7RyH9ceDdFIBm0VaR6jG1-4UDt6CFFCFzQ/exec?BuildID=%s&Test1=%.2f" % (commit, time_s))
|
