added regress test for X86

Test a bug with the SHL instruction.
2016-07-28 22:35:58 +02:00 · 2016-07-28 22:35:58 +02:00 · 8ec8c85e0b
parent 608cb1d943
commit 8ec8c85e0b
1 changed files with 115 additions and 0 deletions
--- a/test_x86_shl.cpp
+++ b/test_x86_shl.cpp
@ -0,0 +1,115 @@
+#include <stdio.h>
+#include <stdint.h>
+#include <unicorn/unicorn.h>
+
+
+void hook_code(uc_engine *uc, uint64_t address, uint32_t size, void *user_data)
+{
+	uint32_t eax = 0, eflags = 0;
+	uc_reg_read(uc, UC_X86_REG_EAX, &eax);
+	uc_reg_read(uc, UC_X86_REG_EFLAGS, &eflags);
+	printf("eip=%08x - eax=%08x -  eflags=%08X\n", (uint32_t)address, eax, eflags);
+}
+
+const char code32_prob[] = { 
+	0xB8, 0x3C, 0x00, 0x00, 0x00,	//0x100000: mov     eax, 3Ch
+	0xB1, 0x02,						//0x100005: mov     cl, 2
+	0xD3, 0xE0,						//0x100007: shl     eax, cl  <- do not set PF
+	0x33, 0xC0						//0x100009: xor     eax, eax
+};
+
+const char code32_ok[] = {
+	0xB8, 0x3C, 0x00, 0x00, 0x00,	//0x100000: mov     eax, 3Ch
+	0xC1, 0xE0, 0x02,				//0x100005: shl     eax, 2  <- set PF correctly
+	0x33, 0xC0						//0x100007: xor     eax, eax
+};
+
+const char code16_prob[] = {
+	0x66, 0xB8, 0x3C, 0x00,			//0x100000: mov     ax, 3Ch
+	0xB1, 0x02,						//0x100004: mov     cl, 2
+	0x66, 0xD3, 0xE0,				//0x100006: shl     ax, cl
+	0x33, 0xC0						//0x100009: xor     eax, eax
+};
+
+const char code16_ok[] = {
+	0x66, 0xB8, 0x3C, 0x00,			//0x100000: mov     ax, 3Ch
+	0x66, 0xC1, 0xE0, 0x02,			//0x100004: shl     ax, 2
+	0x33, 0xC0						//0x100008: xor     eax, eax
+};
+
+const char code8_prob[] = {
+	0xB0, 0x3C,						//0x100000: mov     al, 3Ch
+	0xB1, 0x02,						//0x100002: mov     cl, 2
+	0xD2, 0xE0,						//0x100004: shl     al, 2
+	0x33, 0xC0						//0x100006: xor     eax, eax
+};
+
+const char code8_ok[] = {
+	0xB0, 0x3C,						//0x100000: mov     al, 3Ch
+	0xC0, 0xE0, 0x02,				//0x100002: shl     al, 2
+	0x33, 0xC0						//0x100005: xor     eax, eax
+};
+
+#define ADDR_START 0x100000
+
+#define TEST(X)		if((X) != UC_ERR_OK) { \
+											printf("error: '" #X "' failed\n"); \
+											return 1; \
+										 }
+
+int main()
+{
+	uc_engine *	uc = NULL;
+
+	TEST(uc_open(UC_ARCH_X86, UC_MODE_32, &uc));
+
+	uc_hook trace_hook_code;
+	TEST(uc_hook_add(uc, &trace_hook_code, UC_HOOK_CODE, hook_code, NULL, 1, 0));
+	TEST(uc_mem_map(uc, ADDR_START, 0x1000, UC_PROT_READ | UC_PROT_EXEC));
+
+	//32 bits
+	TEST(uc_mem_write(uc, ADDR_START, code32_prob, sizeof(code32_prob)));
+	printf("running code_prob\n");
+	TEST(uc_emu_start(uc, ADDR_START, ADDR_START + sizeof(code32_prob) - 1, 0, 4));
+
+	uint32_t eflags = 0;
+	uc_reg_write(uc, UC_X86_REG_EFLAGS, &eflags);
+
+	TEST(uc_mem_write(uc, ADDR_START, code32_ok, sizeof(code32_ok)));
+	printf("running code_ok\n");
+	TEST(uc_emu_start(uc, ADDR_START, ADDR_START + sizeof(code32_ok) - 1, 0, 3));
+
+	//16 bits
+	eflags = 0;
+	uc_reg_write(uc, UC_X86_REG_EFLAGS, &eflags);
+
+	TEST(uc_mem_write(uc, ADDR_START, code16_prob, sizeof(code16_prob)));
+	printf("running code16_prob\n");
+	TEST(uc_emu_start(uc, ADDR_START, ADDR_START + sizeof(code16_prob) - 1, 0, 4));
+	
+	eflags = 0;
+	uc_reg_write(uc, UC_X86_REG_EFLAGS, &eflags);
+
+	TEST(uc_mem_write(uc, ADDR_START, code16_ok, sizeof(code16_ok)));
+	printf("running code16_ok\n");
+	TEST(uc_emu_start(uc, ADDR_START, ADDR_START + sizeof(code16_ok) - 1, 0, 3));
+
+	//8 bits
+	eflags = 0;
+	uc_reg_write(uc, UC_X86_REG_EFLAGS, &eflags);
+
+	TEST(uc_mem_write(uc, ADDR_START, code8_prob, sizeof(code8_prob)));
+	printf("running code8_prob\n");
+	TEST(uc_emu_start(uc, ADDR_START, ADDR_START + sizeof(code8_prob) - 1, 0, 4));
+
+	eflags = 0;
+	uc_reg_write(uc, UC_X86_REG_EFLAGS, &eflags);
+
+	TEST(uc_mem_write(uc, ADDR_START, code8_ok, sizeof(code8_ok)));
+	printf("running code8_ok\n");
+	TEST(uc_emu_start(uc, ADDR_START, ADDR_START + sizeof(code8_ok) - 1, 0, 3));
+
+	TEST(uc_close(uc));
+
+	return 0;
+}