67 lines
2.2 KiB
C
67 lines
2.2 KiB
C
#include "unicorn_test.h"
|
|
|
|
static void test_map_correct()
|
|
{
|
|
uc_engine *uc;
|
|
|
|
OK(uc_open(UC_ARCH_X86, UC_MODE_64, &uc));
|
|
OK(uc_mem_map(uc, 0x40000, 0x1000 * 16, UC_PROT_ALL)); // [0x40000, 0x50000]
|
|
OK(uc_mem_map(uc, 0x60000, 0x1000 * 16, UC_PROT_ALL)); // [0x60000, 0x70000]
|
|
OK(uc_mem_map(uc, 0x20000, 0x1000 * 16, UC_PROT_ALL)); // [0x20000, 0x30000]
|
|
uc_assert_err(UC_ERR_MAP,
|
|
uc_mem_map(uc, 0x10000, 0x2000 * 16, UC_PROT_ALL));
|
|
uc_assert_err(UC_ERR_MAP,
|
|
uc_mem_map(uc, 0x25000, 0x1000 * 16, UC_PROT_ALL));
|
|
uc_assert_err(UC_ERR_MAP,
|
|
uc_mem_map(uc, 0x35000, 0x1000 * 16, UC_PROT_ALL));
|
|
uc_assert_err(UC_ERR_MAP,
|
|
uc_mem_map(uc, 0x45000, 0x1000 * 16, UC_PROT_ALL));
|
|
uc_assert_err(UC_ERR_MAP,
|
|
uc_mem_map(uc, 0x55000, 0x2000 * 16, UC_PROT_ALL));
|
|
OK(uc_mem_map(uc, 0x35000, 0x5000, UC_PROT_ALL));
|
|
OK(uc_mem_map(uc, 0x50000, 0x5000, UC_PROT_ALL));
|
|
|
|
OK(uc_close(uc));
|
|
}
|
|
|
|
static void test_map_wrapping()
|
|
{
|
|
uc_engine *uc;
|
|
|
|
OK(uc_open(UC_ARCH_X86, UC_MODE_64, &uc));
|
|
uc_assert_err(UC_ERR_ARG, uc_mem_map(uc, (~0ll - 0x4000) & ~0xfff, 0x8000,
|
|
UC_PROT_ALL));
|
|
|
|
OK(uc_close(uc));
|
|
}
|
|
|
|
static void test_mem_protect()
|
|
{
|
|
uc_engine *qc;
|
|
int r_eax = 0x2000;
|
|
int r_esi = 0xdeadbeef;
|
|
uint32_t mem;
|
|
// add [eax + 4], esi
|
|
char code[] = {0x01, 0x70, 0x04};
|
|
|
|
OK(uc_open(UC_ARCH_X86, UC_MODE_32, &qc));
|
|
OK(uc_reg_write(qc, UC_X86_REG_EAX, &r_eax));
|
|
OK(uc_reg_write(qc, UC_X86_REG_ESI, &r_esi));
|
|
OK(uc_mem_map(qc, 0x1000, 0x1000, UC_PROT_READ | UC_PROT_EXEC));
|
|
OK(uc_mem_map(qc, 0x2000, 0x1000, UC_PROT_READ));
|
|
OK(uc_mem_protect(qc, 0x2000, 0x1000, UC_PROT_READ | UC_PROT_WRITE));
|
|
OK(uc_mem_write(qc, 0x1000, code, sizeof(code)));
|
|
|
|
OK(uc_emu_start(qc, 0x1000, 0x1000 + sizeof(code) - 1, 0, 1));
|
|
OK(uc_mem_read(qc, 0x2000 + 4, &mem, 4));
|
|
|
|
TEST_CHECK(mem == 0xdeadbeef);
|
|
|
|
OK(uc_close(qc));
|
|
}
|
|
|
|
TEST_LIST = {{"test_map_correct", test_map_correct},
|
|
{"test_map_wrapping", test_map_wrapping},
|
|
{"test_mem_protect", test_mem_protect},
|
|
{NULL, NULL}};
|