9f578946d5
* Fix Python regression test suite (partial) * Fix Python regression test suite * Add a test for mapping at high addresses * Add ctl tests
39 lines
790 B
Python
Executable File
39 lines
790 B
Python
Executable File
#!/usr/bin/python
|
|
|
|
import regress
|
|
|
|
from unicorn import *
|
|
from unicorn.x86_const import *
|
|
|
|
|
|
PAGE_SIZE = 0x1000
|
|
|
|
CODE = b'\xf3\xaa' # rep stosb
|
|
BASE = 0x00000000
|
|
|
|
|
|
class TestRep(regress.RegressTest):
|
|
|
|
def runTest(self):
|
|
mu = Uc(UC_ARCH_X86, UC_MODE_32)
|
|
|
|
mu.mem_map(BASE, PAGE_SIZE)
|
|
mu.mem_write(BASE, CODE)
|
|
|
|
mu.reg_write(UC_X86_REG_ECX, 8)
|
|
mu.reg_write(UC_X86_REG_ESI, 0x10)
|
|
mu.reg_write(UC_X86_REG_EDI, 0x20)
|
|
|
|
def __hook_code(uc, addr, size, ud):
|
|
regress.logger.debug('iterations remaining: %d', uc.reg_read(UC_X86_REG_ECX))
|
|
|
|
mu.hook_add(UC_HOOK_CODE, __hook_code)
|
|
|
|
mu.emu_start(BASE, len(CODE))
|
|
|
|
self.assertEqual(0, mu.reg_read(UC_X86_REG_ECX))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
regress.main()
|