lazymio/unicorn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: Add test to reproduce cow+vtlb error

Browse Source
This commit is contained in:
Mark Giraud
2023-08-24 11:43:54 +02:00
parent e189e1fb8b
commit f081af002a
1 changed files with 61 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
tests/unit/test_mem.c
View File

@@ -315,6 +315,66 @@ static void test_snapshot(void)
OK(uc_close(uc)); OK(uc_close(uc));
} }
static bool test_snapshot_with_vtlb_callback(uc_engine *uc, uint64_t addr,
uc_mem_type type,
uc_tlb_entry *result,
void *user_data)
{
result->paddr = addr - 0x400000000;
result->perms = UC_PROT_ALL;
return true;
}
static void test_snapshot_with_vtlb(void)
{
uc_engine *uc;
uc_context *c0, *c1;
uint32_t mem;
uc_hook hook;
// mov eax, [0x2020]; inc eax; mov [0x2020], eax
char code[] = "\xA1\x20\x20\x00\x00\x04\x00\x00\x00\xFF\xC0\xA3\x20\x20\x00\x00\x04\x00\x00\x00";
OK(uc_open(UC_ARCH_X86, UC_MODE_64, &uc));
// Allocate contexts
OK(uc_context_alloc(uc, &c0));
OK(uc_context_alloc(uc, &c1));
OK(uc_ctl_context_mode(uc, UC_CTL_CONTEXT_MEMORY));
OK(uc_ctl_tlb_mode(uc, UC_TLB_VIRTUAL));
OK(uc_hook_add(uc, &hook, UC_HOOK_TLB_FILL, test_snapshot_with_vtlb_callback, NULL, 1, 0));
// Map physical memory
OK(uc_mem_map(uc, 0x1000, 0x1000, UC_PROT_EXEC | UC_PROT_READ));
OK(uc_mem_write(uc, 0x1000, code, sizeof(code) - 1));
OK(uc_mem_map(uc, 0x2000, 0x1000, UC_PROT_ALL));
// Initial context save
OK(uc_context_save(uc, c0));
OK(uc_emu_start(uc, 0x400000000 + 0x1000, 0x400000000 + 0x1000 + sizeof(code) - 1, 0, 0));
OK(uc_mem_read(uc, 0x2020, &mem, sizeof(mem)));
TEST_CHECK(mem == 1);
OK(uc_context_save(uc, c1));
OK(uc_emu_start(uc, 0x400000000 + 0x1000, 0x400000000 + 0x1000 + sizeof(code) - 1, 0, 0));
OK(uc_mem_read(uc, 0x2020, &mem, sizeof(mem)));
TEST_CHECK(mem == 2);
OK(uc_context_restore(uc, c1));
// TODO check mem
OK(uc_mem_read(uc, 0x2020, &mem, sizeof(mem)));
TEST_CHECK(mem == 1);
OK(uc_context_restore(uc, c0));
OK(uc_mem_read(uc, 0x2020, &mem, sizeof(mem)));
TEST_CHECK(mem == 0);
// TODO check mem
OK(uc_context_free(c0));
OK(uc_context_free(c1));
OK(uc_close(uc));
}
static void test_context_snapshot(void) static void test_context_snapshot(void)
{ {
uc_engine *uc; uc_engine *uc;
@@ -393,6 +453,7 @@ TEST_LIST = {{"test_map_correct", test_map_correct},
{"test_mem_protect_remove_exec", test_mem_protect_remove_exec}, {"test_mem_protect_remove_exec", test_mem_protect_remove_exec},
{"test_mem_protect_mmio", test_mem_protect_mmio}, {"test_mem_protect_mmio", test_mem_protect_mmio},
{"test_snapshot", test_snapshot}, {"test_snapshot", test_snapshot},
{"test_snapshot_with_vtlb", test_snapshot_with_vtlb},
{"test_context_snapshot", test_context_snapshot}, {"test_context_snapshot", test_context_snapshot},
{"test_snapshot_unmap", test_snapshot_unmap}, {"test_snapshot_unmap", test_snapshot_unmap},
{NULL, NULL}}; {NULL, NULL}};