regress: add rep_hook.py

tests/regress/rep_hook.py

@@ -0,0 +1,30 @@
+#!/usr/bin/python
+from unicorn import *
+from unicorn.x86_const import *
+import regress
+
+PAGE_SIZE = 4 * 1024
+
+CODE = b'\xf3\xaa'  # rep stosb
+
+
+def hook_code(uc, addr, size, user_data):
+    print("hook called at %x" %addr)
+
+class REP(regress.RegressTest):
+
+    def test_rep(self):
+        mu = Uc(UC_ARCH_X86, UC_MODE_32)
+
+        mu.mem_map(0, PAGE_SIZE)
+        mu.mem_write(0, CODE)
+        mu.reg_write(UC_X86_REG_ECX, 3)
+        mu.reg_write(UC_X86_REG_EDI, 0x100)
+        mu.hook_add(UC_HOOK_CODE, hook_code)
+
+        mu.emu_start(0, len(CODE))
+        self.assertEqual(0, mu.reg_read(UC_X86_REG_ECX))
+
+
+if __name__ == '__main__':
+    regress.main()