679af8a3 |
1 | #include <stdarg.h> |
2 | |
3 | // TODO: move |
4 | static int reg_map_g2h[] = { |
5 | -1, -1, -1, -1, |
6 | -1, -1, -1, -1, |
7 | -1, -1, -1, -1, |
8 | -1, -1, -1, -1, |
9 | -1, -1, -1, -1, |
10 | -1, -1, -1, -1, |
11 | }; |
12 | |
13 | enum { xAX = 0, xCX, xDX, xBX, xSP, xBP, xSI, xDI }; |
14 | |
15 | #define EMIT_PTR(ptr, val, type) \ |
16 | *(type *)(ptr) = val |
17 | |
18 | #define EMIT(val, type) { \ |
19 | EMIT_PTR(tcache_ptr, val, type); \ |
20 | tcache_ptr = (char *)tcache_ptr + sizeof(type); \ |
21 | } |
22 | |
23 | #define EMIT_MODRM(mod,r,rm) \ |
24 | EMIT(((mod)<<6) | ((r)<<3) | (rm), u8) |
25 | |
26 | #define EMIT_OP_MODRM(op,mod,r,rm) { \ |
27 | EMIT(op, u8); \ |
28 | EMIT_MODRM(mod, r, rm); \ |
29 | } |
30 | |
31 | #define emith_move_r_r(dst, src) \ |
32 | EMIT_OP_MODRM(0x8b, 3, dst, src) |
33 | |
34 | #define emith_move_r_imm(r, imm) { \ |
35 | EMIT(0xb8 + (r), u8); \ |
36 | EMIT(imm, u32); \ |
37 | } |
38 | |
39 | #define emith_add_r_imm(r, imm) { \ |
40 | EMIT_OP_MODRM(0x81, 3, 0, r); \ |
41 | EMIT(imm, u32); \ |
42 | } |
43 | |
44 | #define emith_sub_r_imm(r, imm) { \ |
45 | EMIT_OP_MODRM(0x81, 3, 5, r); \ |
46 | EMIT(imm, u32); \ |
47 | } |
48 | |
49 | // XXX: offs is 8bit only |
50 | #define emith_ctx_read(r, offs) { \ |
51 | EMIT_OP_MODRM(0x8b, 1, r, 5); \ |
52 | EMIT(offs, u8); /* mov tmp, [ebp+#offs] */ \ |
53 | } |
54 | |
55 | #define emith_ctx_write(r, offs) { \ |
56 | EMIT_OP_MODRM(0x89, 1, r, 5); \ |
57 | EMIT(offs, u8); /* mov [ebp+#offs], tmp */ \ |
58 | } |
59 | |
60 | #define emith_ctx_sub(val, offs) { \ |
61 | EMIT_OP_MODRM(0x81, 1, 5, 5); \ |
62 | EMIT(offs, u8); \ |
63 | EMIT(val, u32); /* sub [ebp+#offs], dword val */ \ |
64 | } |
65 | |
66 | #define emith_test_t() { \ |
67 | if (reg_map_g2h[SHR_SR] == -1) { \ |
68 | EMIT(0xf6, u8); \ |
69 | EMIT_MODRM(1, 0, 5); \ |
70 | EMIT(SHR_SR * 4, u8); \ |
71 | EMIT(0x01, u8); /* test [ebp+SHR_SR], byte 1 */ \ |
72 | } else { \ |
73 | EMIT(0xf7, u8); \ |
74 | EMIT_MODRM(3, 0, reg_map_g2h[SHR_SR]); \ |
75 | EMIT(0x01, u16); /* test <reg>, word 1 */ \ |
76 | } \ |
77 | } |
78 | |
79 | #define emith_jump(ptr) { \ |
80 | u32 disp = (u32)ptr - ((u32)tcache_ptr + 5); \ |
81 | EMIT(0xe9, u8); \ |
82 | EMIT(disp, u32); \ |
83 | } |
84 | |
85 | #define emith_call(ptr) { \ |
86 | u32 disp = (u32)ptr - ((u32)tcache_ptr + 5); \ |
87 | EMIT(0xe8, u8); \ |
88 | EMIT(disp, u32); \ |
89 | } |
90 | |
91 | #define EMIT_CONDITIONAL(code, is_nonzero) { \ |
92 | char *ptr = tcache_ptr; \ |
93 | tcache_ptr = (char *)tcache_ptr + 2; \ |
94 | code; \ |
95 | EMIT_PTR(ptr, ((is_nonzero) ? 0x75 : 0x74), u8); \ |
96 | EMIT_PTR(ptr + 1, ((char *)tcache_ptr - (ptr + 2)), u8); \ |
97 | } |
98 | |
99 | static void emith_pass_arg(int count, ...) |
100 | { |
101 | va_list vl; |
102 | int i; |
103 | |
104 | va_start(vl, count); |
105 | |
106 | for (i = 0; i < count; i++) { |
107 | long av = va_arg(vl, long); |
108 | int r = 7; |
109 | |
110 | switch (i) { |
111 | case 0: r = xAX; break; |
112 | case 1: r = xDX; break; |
113 | case 2: r = xCX; break; |
114 | } |
115 | emith_move_r_imm(r, av); |
116 | } |
117 | |
118 | va_end(vl); |
119 | } |
120 | |