mirror of
https://github.com/frida/tinycc
synced 2024-12-22 21:16:49 +03:00
277 lines
7.0 KiB
C
277 lines
7.0 KiB
C
#include <stdio.h>
|
|
#include <assert.h>
|
|
|
|
#ifndef _WIN32
|
|
#define __fastcall __attribute((fastcall))
|
|
#endif
|
|
|
|
#if 1
|
|
#define SYMBOL(x) _##x
|
|
#else
|
|
#define SYMBOL(x) x
|
|
#endif
|
|
|
|
/////////////////////////////////////////////////////////////////////////
|
|
////////// TRAP FRAMEWORK
|
|
/////////////////////////////////////////////////////////////////////////
|
|
// if you cast 'TRAP' to a function pointer and call it,
|
|
// it will save all 8 registers,
|
|
// and jump into C-code (previously set using 'SET_TRAP_HANDLER(x)'),
|
|
// in C-code you can pop DWORDs from stack and modify registers
|
|
//
|
|
|
|
void *SYMBOL(trap_handler);
|
|
|
|
extern unsigned char SYMBOL(trap)[];
|
|
asm (
|
|
".text;"
|
|
"_trap:;"
|
|
"pushl %esp;"
|
|
"pusha;"
|
|
"addl $0x4, 0xc(%esp);"
|
|
"pushl %esp;"
|
|
"call *_trap_handler;"
|
|
"addl $0x4, %esp;"
|
|
"movl 0xc(%esp), %eax;"
|
|
"movl %eax, 0x20(%esp);"
|
|
"popa;"
|
|
"popl %esp;"
|
|
"ret;"
|
|
);
|
|
|
|
struct trapframe {
|
|
unsigned edi, esi, ebp, esp, ebx, edx, ecx, eax;
|
|
};
|
|
|
|
|
|
#define M_FLOAT(addr) (*(float *)(addr))
|
|
#define M_DWORD(addr) (*(unsigned *)(addr))
|
|
#define M_WORD(addr) (*(unsigned short *)(addr))
|
|
#define M_BYTE(addr) (*(unsigned char *)(addr))
|
|
#define R_EAX ((tf)->eax)
|
|
#define R_ECX ((tf)->ecx)
|
|
#define R_EDX ((tf)->edx)
|
|
#define R_EBX ((tf)->ebx)
|
|
#define R_ESP ((tf)->esp)
|
|
#define R_EBP ((tf)->ebp)
|
|
#define R_ESI ((tf)->esi)
|
|
#define R_EDI ((tf)->edi)
|
|
|
|
#define ARG(x) (M_DWORD(R_ESP + (x) * 4))
|
|
|
|
#define RETN(x) do { \
|
|
M_DWORD(R_ESP + (x)) = M_DWORD(R_ESP); \
|
|
R_ESP += (x); \
|
|
} while (0)
|
|
|
|
#define DUMP() do { \
|
|
unsigned i; \
|
|
printf("EAX: %08X\n", R_EAX); \
|
|
printf("ECX: %08X\n", R_ECX); \
|
|
printf("EDX: %08X\n", R_EDX); \
|
|
printf("EBX: %08X\n", R_EBX); \
|
|
printf("ESP: %08X\n", R_ESP); \
|
|
printf("EBP: %08X\n", R_EBP); \
|
|
printf("ESI: %08X\n", R_ESI); \
|
|
printf("EDI: %08X\n", R_EDI); \
|
|
printf("\n"); \
|
|
printf("[RETADDR]: %08X\n", M_DWORD(R_ESP)); \
|
|
for (i = 1; i <= 8; i++) { \
|
|
printf("[ARG%4d]: %08X\n", i, ARG(i)); \
|
|
} \
|
|
} while (0)
|
|
|
|
#define SET_TRAP_HANDLER(x) ((SYMBOL(trap_handler)) = (x))
|
|
#define TRAP ((void *) &SYMBOL(trap))
|
|
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////
|
|
////////// SAFECALL FRAMEWORK
|
|
/////////////////////////////////////////////////////////////////////////
|
|
// this framework will convert any calling convention to cdecl
|
|
// usage: first set call target with 'SET_SAFECALL_TARGET(x)'
|
|
// then cast 'SAFECALL' to target function pointer type and invoke it
|
|
// after calling, 'ESPDIFF' is the difference of old and new esp
|
|
|
|
void *SYMBOL(sc_call_target);
|
|
unsigned SYMBOL(sc_retn_addr);
|
|
unsigned SYMBOL(sc_old_esp);
|
|
unsigned SYMBOL(sc_new_esp);
|
|
|
|
extern unsigned char SYMBOL(safecall)[];
|
|
asm (
|
|
".text;"
|
|
"_safecall:;"
|
|
"popl _sc_retn_addr;"
|
|
"movl %esp, _sc_old_esp;"
|
|
"call *_sc_call_target;"
|
|
"movl %esp, _sc_new_esp;"
|
|
"movl _sc_old_esp, %esp;"
|
|
"jmp *_sc_retn_addr;"
|
|
);
|
|
|
|
#define SET_SAFECALL_TARGET(x) ((SYMBOL(sc_call_target)) = (x))
|
|
#define SAFECALL ((void *) &SYMBOL(safecall))
|
|
#define ESPDIFF (SYMBOL(sc_new_esp) - SYMBOL(sc_old_esp))
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////
|
|
////////// TEST FASTCALL INVOKE
|
|
/////////////////////////////////////////////////////////////////////////
|
|
|
|
void check_fastcall_invoke_0(struct trapframe *tf)
|
|
{
|
|
//DUMP();
|
|
RETN(0);
|
|
}
|
|
|
|
void check_fastcall_invoke_1(struct trapframe *tf)
|
|
{
|
|
//DUMP();
|
|
assert(R_ECX == 0x11111111);
|
|
RETN(0);
|
|
}
|
|
void check_fastcall_invoke_2(struct trapframe *tf)
|
|
{
|
|
//DUMP();
|
|
assert(R_ECX == 0x11111111);
|
|
assert(R_EDX == 0x22222222);
|
|
RETN(0);
|
|
}
|
|
void check_fastcall_invoke_3(struct trapframe *tf)
|
|
{
|
|
//DUMP();
|
|
assert(R_ECX == 0x11111111);
|
|
assert(R_EDX == 0x22222222);
|
|
assert(ARG(1) == 0x33333333);
|
|
RETN(1*4);
|
|
}
|
|
void check_fastcall_invoke_4(struct trapframe *tf)
|
|
{
|
|
//DUMP();
|
|
assert(R_ECX == 0x11111111);
|
|
assert(R_EDX == 0x22222222);
|
|
assert(ARG(1) == 0x33333333);
|
|
assert(ARG(2) == 0x44444444);
|
|
RETN(2*4);
|
|
}
|
|
|
|
void check_fastcall_invoke_5(struct trapframe *tf)
|
|
{
|
|
//DUMP();
|
|
assert(R_ECX == 0x11111111);
|
|
assert(R_EDX == 0x22222222);
|
|
assert(ARG(1) == 0x33333333);
|
|
assert(ARG(2) == 0x44444444);
|
|
assert(ARG(3) == 0x55555555);
|
|
RETN(3*4);
|
|
}
|
|
|
|
void test_fastcall_invoke()
|
|
{
|
|
SET_TRAP_HANDLER(check_fastcall_invoke_0);
|
|
((void __fastcall (*)(void)) TRAP)();
|
|
|
|
SET_TRAP_HANDLER(check_fastcall_invoke_1);
|
|
((void __fastcall (*)(unsigned)) TRAP)(0x11111111);
|
|
|
|
SET_TRAP_HANDLER(check_fastcall_invoke_2);
|
|
((void __fastcall (*)(unsigned, unsigned)) TRAP)(0x11111111, 0x22222222);
|
|
|
|
SET_TRAP_HANDLER(check_fastcall_invoke_3);
|
|
((void __fastcall (*)(unsigned, unsigned, unsigned)) TRAP)(0x11111111, 0x22222222, 0x33333333);
|
|
|
|
SET_TRAP_HANDLER(check_fastcall_invoke_4);
|
|
((void __fastcall (*)(unsigned, unsigned, unsigned, unsigned)) TRAP)(0x11111111, 0x22222222, 0x33333333, 0x44444444);
|
|
|
|
SET_TRAP_HANDLER(check_fastcall_invoke_5);
|
|
((void __fastcall (*)(unsigned, unsigned, unsigned, unsigned, unsigned)) TRAP)(0x11111111, 0x22222222, 0x33333333, 0x44444444, 0x55555555);
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////
|
|
////////// TEST FUNCTION CODE GENERATION
|
|
/////////////////////////////////////////////////////////////////////////
|
|
|
|
int __fastcall check_fastcall_espdiff_0(void)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
int __fastcall check_fastcall_espdiff_1(int a)
|
|
{
|
|
return a;
|
|
}
|
|
|
|
int __fastcall check_fastcall_espdiff_2(int a, int b)
|
|
{
|
|
return a + b;
|
|
}
|
|
|
|
int __fastcall check_fastcall_espdiff_3(int a, int b, int c)
|
|
{
|
|
return a + b + c;
|
|
}
|
|
|
|
int __fastcall check_fastcall_espdiff_4(int a, int b, int c, int d)
|
|
{
|
|
return a + b + c + d;
|
|
}
|
|
|
|
int __fastcall check_fastcall_espdiff_5(int a, int b, int c, int d, int e)
|
|
{
|
|
return a + b + c + d + e;
|
|
}
|
|
|
|
void test_fastcall_espdiff()
|
|
{
|
|
int x;
|
|
SET_SAFECALL_TARGET(check_fastcall_espdiff_0);
|
|
x = ((typeof(&check_fastcall_espdiff_0))SAFECALL)();
|
|
assert(x == 0);
|
|
assert(ESPDIFF == 0);
|
|
|
|
SET_SAFECALL_TARGET(check_fastcall_espdiff_1);
|
|
x = ((typeof(&check_fastcall_espdiff_1))SAFECALL)(1);
|
|
assert(x == 1);
|
|
assert(ESPDIFF == 0);
|
|
|
|
SET_SAFECALL_TARGET(check_fastcall_espdiff_2);
|
|
x = ((typeof(&check_fastcall_espdiff_2))SAFECALL)(1, 2);
|
|
assert(x == 1 + 2);
|
|
assert(ESPDIFF == 0);
|
|
|
|
SET_SAFECALL_TARGET(check_fastcall_espdiff_3);
|
|
x = ((typeof(&check_fastcall_espdiff_3))SAFECALL)(1, 2, 3);
|
|
assert(x == 1 + 2 + 3);
|
|
assert(ESPDIFF == 1*4);
|
|
|
|
SET_SAFECALL_TARGET(check_fastcall_espdiff_4);
|
|
x = ((typeof(&check_fastcall_espdiff_4))SAFECALL)(1, 2, 3, 4);
|
|
assert(x == 1 + 2 + 3 + 4);
|
|
assert(ESPDIFF == 2*4);
|
|
|
|
SET_SAFECALL_TARGET(check_fastcall_espdiff_5);
|
|
x = ((typeof(&check_fastcall_espdiff_5))SAFECALL)(1, 2, 3, 4, 5);
|
|
assert(x == 1 + 2 + 3 + 4 + 5);
|
|
assert(ESPDIFF == 3*4);
|
|
}
|
|
|
|
int main()
|
|
{
|
|
#define N 10000
|
|
int i;
|
|
|
|
for (i = 1; i <= N; i++) {
|
|
test_fastcall_espdiff();
|
|
}
|
|
|
|
for (i = 1; i <= N; i++) {
|
|
test_fastcall_invoke();
|
|
}
|
|
|
|
puts("TEST OK");
|
|
return 0;
|
|
}
|