1 | #include <stdio.h>
|
---|
2 | #include <assert.h>
|
---|
3 |
|
---|
4 | #ifndef _WIN32
|
---|
5 | #define __fastcall __attribute((fastcall))
|
---|
6 | #endif
|
---|
7 |
|
---|
8 | #if 1
|
---|
9 | #define SYMBOL(x) _##x
|
---|
10 | #else
|
---|
11 | #define SYMBOL(x) x
|
---|
12 | #endif
|
---|
13 |
|
---|
14 | /////////////////////////////////////////////////////////////////////////
|
---|
15 | ////////// TRAP FRAMEWORK
|
---|
16 | /////////////////////////////////////////////////////////////////////////
|
---|
17 | // if you cast 'TRAP' to a function pointer and call it,
|
---|
18 | // it will save all 8 registers,
|
---|
19 | // and jump into C-code (previously set using 'SET_TRAP_HANDLER(x)'),
|
---|
20 | // in C-code you can pop DWORDs from stack and modify registers
|
---|
21 | //
|
---|
22 |
|
---|
23 | void *SYMBOL(trap_handler);
|
---|
24 |
|
---|
25 | extern unsigned char SYMBOL(trap)[];
|
---|
26 | asm (
|
---|
27 | ".text;"
|
---|
28 | "_trap:;"
|
---|
29 | "pushl %esp;"
|
---|
30 | "pusha;"
|
---|
31 | "addl $0x4, 0xc(%esp);"
|
---|
32 | "pushl %esp;"
|
---|
33 | "call *_trap_handler;"
|
---|
34 | "addl $0x4, %esp;"
|
---|
35 | "movl 0xc(%esp), %eax;"
|
---|
36 | "movl %eax, 0x20(%esp);"
|
---|
37 | "popa;"
|
---|
38 | "popl %esp;"
|
---|
39 | "ret;"
|
---|
40 | );
|
---|
41 |
|
---|
42 | struct trapframe {
|
---|
43 | unsigned edi, esi, ebp, esp, ebx, edx, ecx, eax;
|
---|
44 | };
|
---|
45 |
|
---|
46 |
|
---|
47 | #define M_FLOAT(addr) (*(float *)(addr))
|
---|
48 | #define M_DWORD(addr) (*(unsigned *)(addr))
|
---|
49 | #define M_WORD(addr) (*(unsigned short *)(addr))
|
---|
50 | #define M_BYTE(addr) (*(unsigned char *)(addr))
|
---|
51 | #define R_EAX ((tf)->eax)
|
---|
52 | #define R_ECX ((tf)->ecx)
|
---|
53 | #define R_EDX ((tf)->edx)
|
---|
54 | #define R_EBX ((tf)->ebx)
|
---|
55 | #define R_ESP ((tf)->esp)
|
---|
56 | #define R_EBP ((tf)->ebp)
|
---|
57 | #define R_ESI ((tf)->esi)
|
---|
58 | #define R_EDI ((tf)->edi)
|
---|
59 |
|
---|
60 | #define ARG(x) (M_DWORD(R_ESP + (x) * 4))
|
---|
61 |
|
---|
62 | #define RETN(x) do { \
|
---|
63 | M_DWORD(R_ESP + (x)) = M_DWORD(R_ESP); \
|
---|
64 | R_ESP += (x); \
|
---|
65 | } while (0)
|
---|
66 |
|
---|
67 | #define DUMP() do { \
|
---|
68 | unsigned i; \
|
---|
69 | printf("EAX: %08X\n", R_EAX); \
|
---|
70 | printf("ECX: %08X\n", R_ECX); \
|
---|
71 | printf("EDX: %08X\n", R_EDX); \
|
---|
72 | printf("EBX: %08X\n", R_EBX); \
|
---|
73 | printf("ESP: %08X\n", R_ESP); \
|
---|
74 | printf("EBP: %08X\n", R_EBP); \
|
---|
75 | printf("ESI: %08X\n", R_ESI); \
|
---|
76 | printf("EDI: %08X\n", R_EDI); \
|
---|
77 | printf("\n"); \
|
---|
78 | printf("[RETADDR]: %08X\n", M_DWORD(R_ESP)); \
|
---|
79 | for (i = 1; i <= 8; i++) { \
|
---|
80 | printf("[ARG%4d]: %08X\n", i, ARG(i)); \
|
---|
81 | } \
|
---|
82 | } while (0)
|
---|
83 |
|
---|
84 | #define SET_TRAP_HANDLER(x) ((SYMBOL(trap_handler)) = (x))
|
---|
85 | #define TRAP ((void *) &SYMBOL(trap))
|
---|
86 |
|
---|
87 |
|
---|
88 |
|
---|
89 | /////////////////////////////////////////////////////////////////////////
|
---|
90 | ////////// SAFECALL FRAMEWORK
|
---|
91 | /////////////////////////////////////////////////////////////////////////
|
---|
92 | // this framework will convert any calling convention to cdecl
|
---|
93 | // usage: first set call target with 'SET_SAFECALL_TARGET(x)'
|
---|
94 | // then cast 'SAFECALL' to target function pointer type and invoke it
|
---|
95 | // after calling, 'ESPDIFF' is the difference of old and new esp
|
---|
96 |
|
---|
97 | void *SYMBOL(sc_call_target);
|
---|
98 | unsigned SYMBOL(sc_retn_addr);
|
---|
99 | unsigned SYMBOL(sc_old_esp);
|
---|
100 | unsigned SYMBOL(sc_new_esp);
|
---|
101 |
|
---|
102 | extern unsigned char SYMBOL(safecall)[];
|
---|
103 | asm (
|
---|
104 | ".text;"
|
---|
105 | "_safecall:;"
|
---|
106 | "popl _sc_retn_addr;"
|
---|
107 | "movl %esp, _sc_old_esp;"
|
---|
108 | "call *_sc_call_target;"
|
---|
109 | "movl %esp, _sc_new_esp;"
|
---|
110 | "movl _sc_old_esp, %esp;"
|
---|
111 | "jmp *_sc_retn_addr;"
|
---|
112 | );
|
---|
113 |
|
---|
114 | #define SET_SAFECALL_TARGET(x) ((SYMBOL(sc_call_target)) = (x))
|
---|
115 | #define SAFECALL ((void *) &SYMBOL(safecall))
|
---|
116 | #define ESPDIFF (SYMBOL(sc_new_esp) - SYMBOL(sc_old_esp))
|
---|
117 |
|
---|
118 |
|
---|
119 | /////////////////////////////////////////////////////////////////////////
|
---|
120 | ////////// TEST FASTCALL INVOKE
|
---|
121 | /////////////////////////////////////////////////////////////////////////
|
---|
122 |
|
---|
123 | void check_fastcall_invoke_0(struct trapframe *tf)
|
---|
124 | {
|
---|
125 | //DUMP();
|
---|
126 | RETN(0);
|
---|
127 | }
|
---|
128 |
|
---|
129 | void check_fastcall_invoke_1(struct trapframe *tf)
|
---|
130 | {
|
---|
131 | //DUMP();
|
---|
132 | assert(R_ECX == 0x11111111);
|
---|
133 | RETN(0);
|
---|
134 | }
|
---|
135 | void check_fastcall_invoke_2(struct trapframe *tf)
|
---|
136 | {
|
---|
137 | //DUMP();
|
---|
138 | assert(R_ECX == 0x11111111);
|
---|
139 | assert(R_EDX == 0x22222222);
|
---|
140 | RETN(0);
|
---|
141 | }
|
---|
142 | void check_fastcall_invoke_3(struct trapframe *tf)
|
---|
143 | {
|
---|
144 | //DUMP();
|
---|
145 | assert(R_ECX == 0x11111111);
|
---|
146 | assert(R_EDX == 0x22222222);
|
---|
147 | assert(ARG(1) == 0x33333333);
|
---|
148 | RETN(1*4);
|
---|
149 | }
|
---|
150 | void check_fastcall_invoke_4(struct trapframe *tf)
|
---|
151 | {
|
---|
152 | //DUMP();
|
---|
153 | assert(R_ECX == 0x11111111);
|
---|
154 | assert(R_EDX == 0x22222222);
|
---|
155 | assert(ARG(1) == 0x33333333);
|
---|
156 | assert(ARG(2) == 0x44444444);
|
---|
157 | RETN(2*4);
|
---|
158 | }
|
---|
159 |
|
---|
160 | void check_fastcall_invoke_5(struct trapframe *tf)
|
---|
161 | {
|
---|
162 | //DUMP();
|
---|
163 | assert(R_ECX == 0x11111111);
|
---|
164 | assert(R_EDX == 0x22222222);
|
---|
165 | assert(ARG(1) == 0x33333333);
|
---|
166 | assert(ARG(2) == 0x44444444);
|
---|
167 | assert(ARG(3) == 0x55555555);
|
---|
168 | RETN(3*4);
|
---|
169 | }
|
---|
170 |
|
---|
171 | void test_fastcall_invoke()
|
---|
172 | {
|
---|
173 | SET_TRAP_HANDLER(check_fastcall_invoke_0);
|
---|
174 | ((void __fastcall (*)(void)) TRAP)();
|
---|
175 |
|
---|
176 | SET_TRAP_HANDLER(check_fastcall_invoke_1);
|
---|
177 | ((void __fastcall (*)(unsigned)) TRAP)(0x11111111);
|
---|
178 |
|
---|
179 | SET_TRAP_HANDLER(check_fastcall_invoke_2);
|
---|
180 | ((void __fastcall (*)(unsigned, unsigned)) TRAP)(0x11111111, 0x22222222);
|
---|
181 |
|
---|
182 | SET_TRAP_HANDLER(check_fastcall_invoke_3);
|
---|
183 | ((void __fastcall (*)(unsigned, unsigned, unsigned)) TRAP)(0x11111111, 0x22222222, 0x33333333);
|
---|
184 |
|
---|
185 | SET_TRAP_HANDLER(check_fastcall_invoke_4);
|
---|
186 | ((void __fastcall (*)(unsigned, unsigned, unsigned, unsigned)) TRAP)(0x11111111, 0x22222222, 0x33333333, 0x44444444);
|
---|
187 |
|
---|
188 | SET_TRAP_HANDLER(check_fastcall_invoke_5);
|
---|
189 | ((void __fastcall (*)(unsigned, unsigned, unsigned, unsigned, unsigned)) TRAP)(0x11111111, 0x22222222, 0x33333333, 0x44444444, 0x55555555);
|
---|
190 | }
|
---|
191 |
|
---|
192 |
|
---|
193 | /////////////////////////////////////////////////////////////////////////
|
---|
194 | ////////// TEST FUNCTION CODE GENERATION
|
---|
195 | /////////////////////////////////////////////////////////////////////////
|
---|
196 |
|
---|
197 | int __fastcall check_fastcall_espdiff_0(void)
|
---|
198 | {
|
---|
199 | return 0;
|
---|
200 | }
|
---|
201 |
|
---|
202 | int __fastcall check_fastcall_espdiff_1(int a)
|
---|
203 | {
|
---|
204 | return a;
|
---|
205 | }
|
---|
206 |
|
---|
207 | int __fastcall check_fastcall_espdiff_2(int a, int b)
|
---|
208 | {
|
---|
209 | return a + b;
|
---|
210 | }
|
---|
211 |
|
---|
212 | int __fastcall check_fastcall_espdiff_3(int a, int b, int c)
|
---|
213 | {
|
---|
214 | return a + b + c;
|
---|
215 | }
|
---|
216 |
|
---|
217 | int __fastcall check_fastcall_espdiff_4(int a, int b, int c, int d)
|
---|
218 | {
|
---|
219 | return a + b + c + d;
|
---|
220 | }
|
---|
221 |
|
---|
222 | int __fastcall check_fastcall_espdiff_5(int a, int b, int c, int d, int e)
|
---|
223 | {
|
---|
224 | return a + b + c + d + e;
|
---|
225 | }
|
---|
226 |
|
---|
227 | void test_fastcall_espdiff()
|
---|
228 | {
|
---|
229 | int x;
|
---|
230 | SET_SAFECALL_TARGET(check_fastcall_espdiff_0);
|
---|
231 | x = ((typeof(&check_fastcall_espdiff_0))SAFECALL)();
|
---|
232 | assert(x == 0);
|
---|
233 | assert(ESPDIFF == 0);
|
---|
234 |
|
---|
235 | SET_SAFECALL_TARGET(check_fastcall_espdiff_1);
|
---|
236 | x = ((typeof(&check_fastcall_espdiff_1))SAFECALL)(1);
|
---|
237 | assert(x == 1);
|
---|
238 | assert(ESPDIFF == 0);
|
---|
239 |
|
---|
240 | SET_SAFECALL_TARGET(check_fastcall_espdiff_2);
|
---|
241 | x = ((typeof(&check_fastcall_espdiff_2))SAFECALL)(1, 2);
|
---|
242 | assert(x == 1 + 2);
|
---|
243 | assert(ESPDIFF == 0);
|
---|
244 |
|
---|
245 | SET_SAFECALL_TARGET(check_fastcall_espdiff_3);
|
---|
246 | x = ((typeof(&check_fastcall_espdiff_3))SAFECALL)(1, 2, 3);
|
---|
247 | assert(x == 1 + 2 + 3);
|
---|
248 | assert(ESPDIFF == 1*4);
|
---|
249 |
|
---|
250 | SET_SAFECALL_TARGET(check_fastcall_espdiff_4);
|
---|
251 | x = ((typeof(&check_fastcall_espdiff_4))SAFECALL)(1, 2, 3, 4);
|
---|
252 | assert(x == 1 + 2 + 3 + 4);
|
---|
253 | assert(ESPDIFF == 2*4);
|
---|
254 |
|
---|
255 | SET_SAFECALL_TARGET(check_fastcall_espdiff_5);
|
---|
256 | x = ((typeof(&check_fastcall_espdiff_5))SAFECALL)(1, 2, 3, 4, 5);
|
---|
257 | assert(x == 1 + 2 + 3 + 4 + 5);
|
---|
258 | assert(ESPDIFF == 3*4);
|
---|
259 | }
|
---|
260 |
|
---|
261 | int main()
|
---|
262 | {
|
---|
263 | #define N 10000
|
---|
264 | int i;
|
---|
265 |
|
---|
266 | for (i = 1; i <= N; i++) {
|
---|
267 | test_fastcall_espdiff();
|
---|
268 | }
|
---|
269 |
|
---|
270 | for (i = 1; i <= N; i++) {
|
---|
271 | test_fastcall_invoke();
|
---|
272 | }
|
---|
273 |
|
---|
274 | puts("TEST OK");
|
---|
275 | return 0;
|
---|
276 | }
|
---|