Regress python testcases must define expected value via unittest

regress/fpu_mem_write.py

@@ -2,6 +2,8 @@
 from unicorn import *
 from unicorn.x86_const import *
 
+import regress
+
 ESP = 0x2000
 PAGE_SIZE = 1 * 1024 * 1024
 
@@ -9,24 +11,27 @@ PAGE_SIZE = 1 * 1024 * 1024
 #	pop ecx
 CODE = b'\x9B\xD9\x3C\x24\x59'
 
-def mem_reader(addr, size):
-	tmp = mu.mem_read(addr, size)
-
-	for i in tmp:
-		print(" 0x%x" % i),
-	print("")
-
 def hook_mem_write(uc, access, address, size, value, user_data):
 	print("mem WRITE: 0x%x, data size = %u, data value = 0x%x" % (address, size, value))
 	return True
 
-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_ESP, ESP)
-mu.hook_add(UC_HOOK_MEM_WRITE, hook_mem_write)
+class FpuWrite(regress.RegressTest):
 
-mu.emu_start(0x0, 0, 0, 2)
-esp = mu.reg_read(UC_X86_REG_ESP)
-print("value at ESP [0x%X]: " % esp)
-mem_reader(esp, 10)
+    def mem_reader(self, mu, addr, size, expected):
+        tmp = mu.mem_read(addr, size)
+        for i, e in zip(tmp, expected):
+            self.assertEquals(e, i)
+
+    def runTest(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_ESP, ESP)
+
+        mu.hook_add(UC_HOOK_MEM_WRITE, hook_mem_write)
+        mu.emu_start(0x0, 0, 0, 2)
+        esp = mu.reg_read(UC_X86_REG_ESP)
+        self.mem_reader(mu, esp, 10, [0] * 10)
+
+if __name__ == '__main__':
+    regress.main()