yuzu/unicorn
1
0
Fork 0
mirror of https://github.com/yuzu-emu/unicorn.git synced 2025-01-10 20:35:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

samples: cleanup & put mem_apis into sample_all.sh

Browse source
This commit is contained in:
Nguyen Anh Quynh 2015-09-05 09:42:47 +08:00
parent 7b5bb95147
commit 74c069dc22
2 changed files with 68 additions and 54 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

120
samples/mem_apis.c
View file

@ -35,7 +35,9 @@ static void hook_code(ucengine *uc, uint64_t addr, uint32_t size, void *user_dat
{ {
uint8_t opcode; uint8_t opcode;
unsigned char buf[256]; unsigned char buf[256];
insts_executed++; insts_executed++;
if (uc_mem_read(uc, addr, buf, size) != UC_ERR_OK) { if (uc_mem_read(uc, addr, buf, size) != UC_ERR_OK) {
printf("not ok - uc_mem_read fail during hook_code callback, addr: 0x%" PRIx64 "\n", addr); printf("not ok - uc_mem_read fail during hook_code callback, addr: 0x%" PRIx64 "\n", addr);
if (uc_emu_stop(uc) != UC_ERR_OK) { if (uc_emu_stop(uc) != UC_ERR_OK) {
@ -43,27 +45,28 @@ static void hook_code(ucengine *uc, uint64_t addr, uint32_t size, void *user_dat
_exit(-1); _exit(-1);
} }
} }
opcode = buf[0]; opcode = buf[0];
switch (opcode) { switch (opcode) {
case 0x41: //inc ecx case 0x41: // inc ecx
if (uc_mem_protect(uc, 0x101000, 0x1000, UC_PROT_READ) != UC_ERR_OK) { if (uc_mem_protect(uc, 0x101000, 0x1000, UC_PROT_READ) != UC_ERR_OK) {
printf("not ok - uc_mem_protect fail during hook_code callback, addr: 0x%" PRIx64 "\n", addr); printf("not ok - uc_mem_protect fail during hook_code callback, addr: 0x%" PRIx64 "\n", addr);
_exit(-1); _exit(-1);
} }
break; break;
case 0x42: //inc edx case 0x42: // inc edx
if (uc_mem_unmap(uc, 0x101000, 0x1000) != UC_ERR_OK) { if (uc_mem_unmap(uc, 0x101000, 0x1000) != UC_ERR_OK) {
printf("not ok - uc_mem_unmap fail during hook_code callback, addr: 0x%" PRIx64 "\n", addr); printf("not ok - uc_mem_unmap fail during hook_code callback, addr: 0x%" PRIx64 "\n", addr);
_exit(-1); _exit(-1);
} }
break; break;
case 0xf4: //hlt case 0xf4: // hlt
if (uc_emu_stop(uc) != UC_ERR_OK) { if (uc_emu_stop(uc) != UC_ERR_OK) {
printf("not ok - uc_emu_stop fail during hook_code callback, addr: 0x%" PRIx64 "\n", addr); printf("not ok - uc_emu_stop fail during hook_code callback, addr: 0x%" PRIx64 "\n", addr);
_exit(-1); _exit(-1);
} }
break; break;
default: //all others default: // all others
break; break;
} }
} }
@ -115,24 +118,25 @@ static void do_nx_demo(bool cause_fault)
uc_mem_map(uc, 0x100000, 0x3000, UC_PROT_READ | UC_PROT_EXEC); uc_mem_map(uc, 0x100000, 0x3000, UC_PROT_READ | UC_PROT_EXEC);
/* /*
bits 32 bits 32
page0: page0:
times 4091 inc eax times 4091 inc eax
jmp page2 jmp page2
page1: page1:
times 4095 inc eax times 4095 inc eax
hlt hlt
page2: page2:
jmp page1 jmp page1
*/ */
memset(code_buf, 0x40, sizeof(code_buf)); //fill with inc eax memset(code_buf, 0x40, sizeof(code_buf)); // fill with inc eax
memcpy(code_buf + 0x1000 - 5, "\xe9\x00\x10\x00\x00", 5); //jump to 0x102000 memcpy(code_buf + 0x1000 - 5, "\xe9\x00\x10\x00\x00", 5); // jump to 0x102000
memcpy(code_buf + 0x2000, "\xe9\xfb\xef\xff\xff", 5); //jump to 0x101000 memcpy(code_buf + 0x2000, "\xe9\xfb\xef\xff\xff", 5); // jump to 0x101000
code_buf[0x1fff] = 0xf4; //hlt code_buf[0x1fff] = 0xf4; //hlt
if (cause_fault) { if (cause_fault) {
//insert instruction to trigger U_PROT_EXEC change (see hook_code function) // insert instruction to trigger U_PROT_EXEC change (see hook_code function)
code_buf[0x1000] = 0x41; //inc ecx at page1 code_buf[0x1000] = 0x41; // inc ecx at page1
} }
// write machine code to be emulated to memory // write machine code to be emulated to memory
@ -143,8 +147,7 @@ page2:
// intercept code and invalid memory events // intercept code and invalid memory events
if (uc_hook_add(uc, &trace2, UC_HOOK_CODE, hook_code, NULL, (uint64_t)1, (uint64_t)0) != UC_ERR_OK || if (uc_hook_add(uc, &trace2, UC_HOOK_CODE, hook_code, NULL, (uint64_t)1, (uint64_t)0) != UC_ERR_OK ||
uc_hook_add(uc, &trace1, UC_HOOK_MEM_INVALID, hook_mem_invalid, NULL) != UC_ERR_OK) { uc_hook_add(uc, &trace1, UC_HOOK_MEM_INVALID, hook_mem_invalid, NULL) != UC_ERR_OK) {
printf("not ok - Failed to install hooks\n"); printf("not ok - Failed to install hooks\n");
return; return;
} }
@ -158,11 +161,13 @@ page2:
} else { } else {
printf("SUCCESSFUL EXECUTION\n"); printf("SUCCESSFUL EXECUTION\n");
} }
printf("Executed %d instructions\n\n", insts_executed); printf("Executed %d instructions\n\n", insts_executed);
uc_close(uc); uc_close(uc);
} }
void nx_test() static void nx_test()
{ {
printf("NX demo - step 1: show that code runs to completion\n"); printf("NX demo - step 1: show that code runs to completion\n");
do_nx_demo(false); do_nx_demo(false);
@ -170,9 +175,9 @@ void nx_test()
do_nx_demo(true); do_nx_demo(true);
} }
const uint8_t WRITE_DEMO[] = static const uint8_t WRITE_DEMO[] =
"\x90\xc7\x05\x00\x20\x10\x00\x78\x56\x34\x12\xc7\x05\xfc\x0f\x10" "\x90\xc7\x05\x00\x20\x10\x00\x78\x56\x34\x12\xc7\x05\xfc\x0f\x10"
"\x00\x78\x56\x34\x12\xc7\x05\x00\x10\x10\x00\x21\x43\x65\x87"; "\x00\x78\x56\x34\x12\xc7\x05\x00\x10\x10\x00\x21\x43\x65\x87";
static void do_perms_demo(bool change_perms) static void do_perms_demo(bool change_perms)
{ {
@ -195,20 +200,22 @@ static void do_perms_demo(bool change_perms)
uc_mem_map(uc, 0x100000, 0x3000, UC_PROT_ALL); uc_mem_map(uc, 0x100000, 0x3000, UC_PROT_ALL);
/* /*
bits 32 bits 32
nop nop
mov dword [0x102000], 0x12345678 mov dword [0x102000], 0x12345678
mov dword [0x100ffc], 0x12345678 mov dword [0x100ffc], 0x12345678
mov dword [0x101000], 0x87654321 ; crashing case crashes here mov dword [0x101000], 0x87654321 ; crashing case crashes here
times 1000 nop times 1000 nop
hlt hlt
*/ */
memcpy(code_buf, WRITE_DEMO, sizeof(WRITE_DEMO) - 1); memcpy(code_buf, WRITE_DEMO, sizeof(WRITE_DEMO) - 1);
memset(code_buf + sizeof(WRITE_DEMO) - 1, 0x90, 1000); memset(code_buf + sizeof(WRITE_DEMO) - 1, 0x90, 1000);
code_buf[sizeof(WRITE_DEMO) - 1 + 1000] = 0xf4; //hlt code_buf[sizeof(WRITE_DEMO) - 1 + 1000] = 0xf4; // hlt
if (change_perms) { if (change_perms) {
code_buf[0] = 0x41; //inc ecx (see hook_code function) // write protect memory area [0x101000, 0x101fff]. see hook_code function
code_buf[0] = 0x41; // inc ecx
} }
// write machine code to be emulated to memory // write machine code to be emulated to memory
@ -219,8 +226,7 @@ bits 32
// intercept code and invalid memory events // intercept code and invalid memory events
if (uc_hook_add(uc, &trace2, UC_HOOK_CODE, hook_code, NULL, (uint64_t)1, (uint64_t)0) != UC_ERR_OK || if (uc_hook_add(uc, &trace2, UC_HOOK_CODE, hook_code, NULL, (uint64_t)1, (uint64_t)0) != UC_ERR_OK ||
uc_hook_add(uc, &trace1, UC_HOOK_MEM_INVALID, hook_mem_invalid, NULL) != UC_ERR_OK) { uc_hook_add(uc, &trace1, UC_HOOK_MEM_INVALID, hook_mem_invalid, NULL) != UC_ERR_OK) {
printf("not ok - Failed to install hooks\n"); printf("not ok - Failed to install hooks\n");
return; return;
} }
@ -234,11 +240,13 @@ bits 32
} else { } else {
printf("SUCCESSFUL EXECUTION\n"); printf("SUCCESSFUL EXECUTION\n");
} }
printf("Executed %d instructions\n\n", insts_executed); printf("Executed %d instructions\n\n", insts_executed);
uc_close(uc); uc_close(uc);
} }
void perms_test() static void perms_test()
{ {
printf("Permissions demo - step 1: show that area is writeable\n"); printf("Permissions demo - step 1: show that area is writeable\n");
do_perms_demo(false); do_perms_demo(false);
@ -268,20 +276,22 @@ static void do_unmap_demo(bool do_unmap)
uc_mem_map(uc, 0x100000, 0x3000, UC_PROT_ALL); uc_mem_map(uc, 0x100000, 0x3000, UC_PROT_ALL);
/* /*
bits 32 bits 32
nop nop
mov dword [0x102000], 0x12345678 mov dword [0x102000], 0x12345678
mov dword [0x100ffc], 0x12345678 mov dword [0x100ffc], 0x12345678
mov dword [0x101000], 0x87654321 ; crashing case crashes here mov dword [0x101000], 0x87654321 ; crashing case crashes here
times 1000 nop times 1000 nop
hlt hlt
*/ */
memcpy(code_buf, WRITE_DEMO, sizeof(WRITE_DEMO) - 1); memcpy(code_buf, WRITE_DEMO, sizeof(WRITE_DEMO) - 1);
memset(code_buf + sizeof(WRITE_DEMO) - 1, 0x90, 1000); memset(code_buf + sizeof(WRITE_DEMO) - 1, 0x90, 1000);
code_buf[sizeof(WRITE_DEMO) - 1 + 1000] = 0xf4; //hlt code_buf[sizeof(WRITE_DEMO) - 1 + 1000] = 0xf4; // hlt
if (do_unmap) { if (do_unmap) {
code_buf[0] = 0x42; //inc edx (see hook_code function) // unmap memory area [0x101000, 0x101fff]. see hook_code function
code_buf[0] = 0x42; // inc edx (see hook_code function)
} }
// write machine code to be emulated to memory // write machine code to be emulated to memory
@ -292,8 +302,7 @@ bits 32
// intercept code and invalid memory events // intercept code and invalid memory events
if (uc_hook_add(uc, &trace2, UC_HOOK_CODE, hook_code, NULL, (uint64_t)1, (uint64_t)0) != UC_ERR_OK || if (uc_hook_add(uc, &trace2, UC_HOOK_CODE, hook_code, NULL, (uint64_t)1, (uint64_t)0) != UC_ERR_OK ||
uc_hook_add(uc, &trace1, UC_HOOK_MEM_INVALID, hook_mem_invalid, NULL) != UC_ERR_OK) { uc_hook_add(uc, &trace1, UC_HOOK_MEM_INVALID, hook_mem_invalid, NULL) != UC_ERR_OK) {
printf("not ok - Failed to install hooks\n"); printf("not ok - Failed to install hooks\n");
return; return;
} }
@ -307,11 +316,13 @@ bits 32
} else { } else {
printf("SUCCESSFUL EXECUTION\n"); printf("SUCCESSFUL EXECUTION\n");
} }
printf("Executed %d instructions\n\n", insts_executed); printf("Executed %d instructions\n\n", insts_executed);
uc_close(uc); uc_close(uc);
} }
void unmap_test() static void unmap_test()
{ {
printf("Unmap demo - step 1: show that area is writeable\n"); printf("Unmap demo - step 1: show that area is writeable\n");
do_unmap_demo(false); do_unmap_demo(false);
@ -324,5 +335,6 @@ int main(int argc, char **argv, char **envp)
nx_test(); nx_test();
perms_test(); perms_test();
unmap_test(); unmap_test();
return 0; return 0;
} }

2
samples/sample_all.sh
View file

@ -19,6 +19,8 @@ if test -e $DIR/sample_x86; then
$DIR/sample_x86 -16 $DIR/sample_x86 -16
echo "==========================" echo "=========================="
$DIR/shellcode -32 $DIR/shellcode -32
echo "=========================="
$DIR/mem_apis
fi fi
if test -e $DIR/sample_arm; then if test -e $DIR/sample_arm; then
echo "==========================" echo "=========================="