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

unit: fix test_hookcounts.c

Browse Source
This commit is contained in:
Nguyen Anh Quynh 2016-03-20 01:54:37 +08:00
parent 5d22d6c7f4
commit ad6bb8c9f8
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/unit/test_hookcounts.c
View File

@ -24,7 +24,7 @@ static void test_code_hook(uc_engine *uc,
{ {
++total_instructions; ++total_instructions;
if (total_instructions > expected_instructions) if (total_instructions == expected_instructions)
{ {
uc_emu_stop(uc); uc_emu_stop(uc);
} }
@ -78,13 +78,13 @@ test_hook_count(uc_engine *uc,
const uint8_t *code, const uint8_t *code,
int start_offset, int start_offset,
int code_length, int code_length,
int expected_instructions) int count)
{ {
#define BASEADDR 0x1000000 #define BASEADDR 0x1000000
#define MEMSIZE (2 * 1024 * 1024) #define MEMSIZE (2 * 1024 * 1024)
uint64_t address = BASEADDR + (expected_instructions * MEMSIZE); uint64_t address = BASEADDR + (count * MEMSIZE);
total_instructions = 0; total_instructions = 0;
#undef BASEADDR #undef BASEADDR
@ -99,13 +99,14 @@ test_hook_count(uc_engine *uc,
printf("Address: %"PRIx64"\n", address); printf("Address: %"PRIx64"\n", address);
printf("Start : %"PRIx64"\n", address + start_offset); printf("Start : %"PRIx64"\n", address + start_offset);
printf("End : %"PRIx64"\n", address + code_length - 1); printf("End : %"PRIx64"\n", address + code_length - 1);
printf("Count : %d\n", expected_instructions); printf("Count : %d\n", count);
#endif #endif
expected_instructions = count;
OK(uc_emu_start(uc, OK(uc_emu_start(uc,
address+start_offset, address+start_offset,
address+code_length, address+code_length,
0, 0,
expected_instructions)); count));
assert_int_equal(expected_instructions, total_instructions); assert_int_equal(expected_instructions, total_instructions);