9f578946d5
* Fix Python regression test suite (partial) * Fix Python regression test suite * Add a test for mapping at high addresses * Add ctl tests
52 lines
1.3 KiB
Python
Executable File
52 lines
1.3 KiB
Python
Executable File
#!/usr/bin/python
|
|
|
|
'''https://github.com/unicorn-engine/unicorn/issues/334'''
|
|
|
|
from __future__ import print_function
|
|
import regress
|
|
|
|
from unicorn import *
|
|
from unicorn.x86_const import *
|
|
|
|
ADDRESS = 0x8048000
|
|
STACK_ADDRESS = 0xffff000
|
|
STACK_SIZE = 4096
|
|
|
|
CODE = (
|
|
b'\x31\xDB' # xor ebx, ebx
|
|
b'\x53' # push ebx
|
|
b'\x43' # inc ebx
|
|
b'\x53' # push ebx
|
|
b'\x6A\x02' # push 2
|
|
b'\x6A\x66' # push 66h
|
|
b'\x58' # pop eax
|
|
b'\x89\xE1' # mov ecx, esp
|
|
b'\xCD\x80' # int 80h
|
|
)
|
|
|
|
EP = ADDRESS + 0x54
|
|
|
|
def hook_code(mu, address, size, user_data):
|
|
regress.logger.debug(">>> Tracing instruction at %#x, instruction size = %u", address, size)
|
|
|
|
|
|
class HookCodeAddDelTest(regress.RegressTest):
|
|
def runTest(self):
|
|
emu = Uc(UC_ARCH_X86, UC_MODE_32)
|
|
emu.mem_map(ADDRESS, 0x1000)
|
|
emu.mem_write(EP, CODE)
|
|
|
|
emu.mem_map(STACK_ADDRESS, STACK_SIZE)
|
|
emu.reg_write(UC_X86_REG_ESP, STACK_ADDRESS + STACK_SIZE)
|
|
|
|
# UC_HOOK_CODE hook will work even after deletion
|
|
i = emu.hook_add(UC_HOOK_CODE, hook_code, None)
|
|
emu.hook_del(i)
|
|
|
|
emu.emu_start(EP, EP + len(CODE), count = 3)
|
|
regress.logger.debug("EIP: %#x", emu.reg_read(UC_X86_REG_EIP))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
regress.main()
|