You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
41 lines
1003 B
41 lines
1003 B
#!/usr/bin/python
|
|
|
|
# Test BL <symbol> instruction with KS_OPT_SYM_RESOLVER for ARM32
|
|
|
|
# Github issue: #355
|
|
# Author: jan2642
|
|
|
|
from keystone import *
|
|
|
|
import regress
|
|
|
|
class TestARM(regress.RegressTest):
|
|
def runTest(self):
|
|
def sym_resolver(symbol, value):
|
|
if symbol == "symBackward":
|
|
value[0] = 0x0
|
|
return True
|
|
|
|
if symbol == "symForward":
|
|
value[0] = 0x20
|
|
return True
|
|
|
|
return False
|
|
|
|
# Initialize Keystone engine
|
|
ks = Ks(KS_ARCH_ARM, KS_MODE_ARM)
|
|
ks.sym_resolver = sym_resolver
|
|
|
|
encoding, _ = ks.asm(b"""
|
|
sub r0, r0, r0
|
|
sub r1, r1, r1
|
|
bl symBackward
|
|
""")
|
|
self.assertEqual(encoding[-4:], [ 0xfc, 0xff, 0xff, 0xeb ])
|
|
|
|
encoding, _ = ks.asm(b"bl symForward")
|
|
self.assertEqual(encoding, [ 0x06, 0x00, 0x00, 0xeb ])
|
|
|
|
if __name__ == '__main__':
|
|
regress.main()
|