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

Add the unit test from saicao

Browse Source 
Co-authored-by: Sai Cao <1665673333@qq.com>
This commit is contained in:
lazymio
2025-01-04 19:00:22 +08:00
parent 8442eb6feb
commit 8720632764
1 changed files with 49 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
tests/unit/test_arm64.c
View File

@@ -599,6 +599,54 @@ static void test_arm64_mem_prot_regress(void)
OK(uc_close(uc));
}
static bool test_arm64_mem_read_write_cb(uc_engine *uc, int type,
uint64_t address, int size,
int64_t value, void *user_data)
{
uint64_t *count = (uint64_t *)user_data;
switch (type) {
case UC_MEM_READ:
count[0]++;
break;
case UC_MEM_WRITE:
count[1]++;
break;
}
return 0;
}
static void test_arm64_mem_hook_read_write(void)
{
uc_engine *uc;
// ldp x1, x2, [sp]
// stp x1, x2,[sp]
// ldp x1, x2, [sp]
// stp x1, x2,[sp]
const char code[] = {0xe1, 0x0b, 0x40, 0xa9, 0xe1, 0x0b, 0x00, 0xa9,
0xe1, 0x0b, 0x40, 0xa9, 0xe1, 0x0b, 0x00, 0xa9};
uint64_t r_sp;
r_sp = 0x16db6a040;
uc_hook hk;
uint64_t counter[2] = {0, 0};
uc_common_setup(&uc, UC_ARCH_ARM64, UC_MODE_ARM, code, sizeof(code),
UC_CPU_ARM64_A72);
uc_reg_write(uc, UC_ARM64_REG_SP, &r_sp);
uc_mem_map(uc, 0x16db68000, 1024 * 16, UC_PROT_ALL);
OK(uc_hook_add(uc, &hk, UC_HOOK_MEM_READ, test_arm64_mem_read_write_cb,
counter, 1, 0));
OK(uc_hook_add(uc, &hk, UC_HOOK_MEM_WRITE, test_arm64_mem_read_write_cb,
counter, 1, 0));
uc_assert_err(UC_ERR_OK, uc_emu_start(uc, code_start,
code_start + sizeof(code), 0, 0));
TEST_CHECK(counter[0] == 4 && counter[1] == 4);
OK(uc_close(uc));
}
TEST_LIST = {{"test_arm64_until", test_arm64_until},
{"test_arm64_code_patching", test_arm64_code_patching},
{"test_arm64_code_patching_count", test_arm64_code_patching_count},
@@ -615,4 +663,5 @@ TEST_LIST = {{"test_arm64_until", test_arm64_until},
{"test_arm64_mmu", test_arm64_mmu},
{"test_arm64_pc_wrap", test_arm64_pc_wrap},
{"test_arm64_mem_prot_regress", test_arm64_mem_prot_regress},
{"test_arm64_mem_hook_read_write", test_arm64_mem_hook_read_write},
{NULL, NULL}};