Revamp Python regression tests suite (#2022)

* Fix Python regression test suite (partial)

* Fix Python regression test suite

* Add a test for mapping at high addresses

* Add ctl tests
This commit is contained in:
Eli
2024-10-13 08:14:10 +03:00
committed by GitHub
parent 78580ca8f9
commit 9f578946d5
58 changed files with 1903 additions and 1455 deletions

View File

@@ -1,35 +1,39 @@
#!/usr/bin/python
# By Ryan Hileman, issue #3
from capstone import *
import regress
from capstone import Cs, CS_ARCH_X86, CS_MODE_64
from unicorn import *
from unicorn.x86_const import *
import regress
code = 'f20f1005aa120000'.decode('hex')
def dis(mem, addr):
md = Cs(CS_ARCH_X86, CS_MODE_64)
return '\n'.join([
'%s %s' % (i.mnemonic, i.op_str)
for i in md.disasm(str(mem), addr)
])
CODE = b'\xf2\x0f\x10\x05\xaa\x12\x00\x00'
def hook_code(uc, addr, size, user_data):
def dis(md, mem, addr):
return '\n'.join(('%-16s %s' % (insn.mnemonic, insn.op_str) for insn in md.disasm(mem, addr)))
def hook_code(uc, addr, size, md):
mem = uc.mem_read(addr, size)
print 'instruction size:', size
print 'instruction:', str(mem).encode('hex'), dis(mem, addr)
print 'reference: ', code.encode('hex'), dis(code, addr)
regress.logger.debug('instruction size: %d', size)
regress.logger.debug('instruction: %s %s', mem, dis(md, mem, addr))
regress.logger.debug('reference: %s %s', CODE, dis(md, CODE, addr))
class Movsd(regress.RegressTest):
def runTest(self):
addr = 0x400000
mu = Uc(UC_ARCH_X86, UC_MODE_64)
mu.hook_add(UC_HOOK_CODE, hook_code)
md = Cs(CS_ARCH_X86, CS_MODE_64)
mu.hook_add(UC_HOOK_CODE, hook_code, md)
mu.mem_map(addr, 8 * 1024 * 1024)
mu.mem_write(addr, code)
mu.emu_start(addr, addr + len(code))
mu.mem_write(addr, CODE)
mu.emu_start(addr, addr + len(CODE))
if __name__ == '__main__':
regress.main()