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
|
#!/usr/bin/env python
# this tests disassembly through the binja API, ensuring binja <- arch <- disassembler
#
# that's an important distinction versus testing just the disassembler
import re, struct, os, sys, ctypes
import binaryninja
print("binaryninja.__file__:", binaryninja.__file__)
sys.path.append('./disassembler')
import disasm_test
arch = None
def disassemble(addr, insnum):
global arch
if not arch:
arch = binaryninja.Architecture['aarch64']
data = struct.pack('<I', insnum)
(tokens, length) = arch.get_instruction_text(data, addr)
if not tokens or length==0:
return None
return disasm_test.normalize(''.join([x.text for x in tokens]))
def main():
if sys.argv[1:]:
if sys.argv[1] in ['strain', 'strainer', 'stress', 'stresser']:
for insnum in range(0, 2**32):
result = disassemble(0, insnum)
if insnum & 0xFFFFFF == 0:
print('%08X: %s' % (insnum, result))
else:
insnum = int(sys.argv[1], 16)
print(disassemble(0, insnum))
else:
with open('./disassembler/test_cases.txt') as fp:
lines = fp.readlines()
for (i,line) in enumerate(lines):
if line.startswith('// '): continue
assert line[8] == ' '
insnum = int(line[0:8], 16)
actual = disassemble(disasm_test.ADDRESS_TEST, insnum)
expected = line[9:].rstrip()
print('0x%08X -%s- vs. -%s-' % (insnum, actual, expected))
if disasm_test.compare_disassembly(actual, expected):
if actual and disasm_test.excusable_difference(actual, expected):
continue
print('line %d/%d (%.2f%%)' % (i, len(lines), i/len(lines)*100))
sys.exit(-1)
print('PASS')
if __name__ == '__main__':
main()
|
