mirror of
https://github.com/yuzu-emu/unicorn.git
synced 2025-03-22 14:45:09 +00:00
116 lines
3.5 KiB
C++
116 lines
3.5 KiB
C++
#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;
|
|
}
|