1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
|
#ifndef AMAL_EXECUTOR_X86_64_ASM_H
#define AMAL_EXECUTOR_X86_64_ASM_H
#include "../../include/std/misc.h"
#include "../../include/std/types.h"
typedef struct {
void *code;
u8 *code_it;
usize allocated_size;
} Asm;
typedef enum {
EAX,
ECX,
EDX,
EBX,
ESP,
EBP,
ESI,
EDI
} Reg32;
typedef enum {
RAX,
RCX,
RDX,
RBX,
RSP,
RBP,
RSI,
RDI
} Reg64;
typedef struct {
Reg64 base;
Reg64 index;
i32 disp;
u8 scale;
} AsmPtr;
void asm_ptr_init(AsmPtr *self, Reg64 base);
void asm_ptr_init_index(AsmPtr *self, Reg64 base, Reg64 index);
void asm_ptr_init_disp(AsmPtr *self, Reg64 base, i32 disp);
void asm_ptr_init_index_disp(AsmPtr *self, Reg64 base, Reg64 index, i32 disp);
CHECK_RESULT int asm_init(Asm *self);
void asm_deinit(Asm *self);
usize asm_get_size(Asm *self);
CHECK_RESULT int asm_execute(Asm *self, u32 offset);
CHECK_RESULT int asm_nop(Asm *self);
CHECK_RESULT int asm_mov_mi(Asm *self, AsmPtr *dst, i32 immediate);
CHECK_RESULT int asm_mov_mr(Asm *self, AsmPtr *dst, Reg64 src);
CHECK_RESULT int asm_mov_rm(Asm *self, Reg64 dst, AsmPtr *src);
CHECK_RESULT int asm_mov_ri(Asm *self, Reg64 dst, i64 immediate);
CHECK_RESULT int asm_mov_rr(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_add_rr(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_sub_rr(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_imul_rr(Asm *self, Reg64 dst, Reg64 src);
/* Sign extend RAX into RDX, this is needed for some operations, such as idiv */
CHECK_RESULT int asm_cqo(Asm *self);
/*
Divide RDX:RAX by @src. Store the quotient in RAX and the remainder in RDX.
@asm_cqo should be called before this, since RAX needs to be sign extended into RDX
*/
CHECK_RESULT int asm_idiv_rr(Asm *self, Reg64 src);
CHECK_RESULT int asm_pushr(Asm *self, Reg64 reg);
CHECK_RESULT int asm_popr(Asm *self, Reg64 reg);
CHECK_RESULT int asm_callr(Asm *self, Reg64 reg);
/*
In x86 assembly, the @relative position starts from the next instruction.
This offset shouldn't be calculated by the caller and is instead managed
by this asm library itself.
*/
CHECK_RESULT int asm_call_rel32(Asm *self, i32 relative);
void asm_override_call_rel32(Asm *self, u32 asm_index, i32 new_relative);
CHECK_RESULT int asm_cmp_rm(Asm *self, Reg64 reg1, AsmPtr *reg2);
/*
Sets the 8 bit memory operand to 1 if the last cmp was equals, otherwise set it to 0.
Note: this instruction doesn't work with AH (RSP), CH (RBP), DH (RSI) and BH (RDI).
TODO: When ST, MM AND XMM registers are implemented, also check for them as they are also invalid
*/
CHECK_RESULT int asm_sete_m(Asm *self, AsmPtr *dst);
CHECK_RESULT int asm_sete_r(Asm *self, Reg64 dst);
/*
In x86 assembly, the @relative position starts from the next instruction.
This offset shouldn't be calculated by the caller and is instead managed
by this asm library itself.
*/
CHECK_RESULT int asm_jz(Asm *self, i32 relative);
/* Override conditional jump target */
void asm_override_jcc_rel32(Asm *self, u32 asm_index, i32 new_relative);
/*
In x86 assembly, the @relative position starts from the next instruction.
This offset shouldn't be calculated by the caller and is instead managed
by this asm library itself.
*/
CHECK_RESULT int asm_jmp(Asm *self, i32 relative);
void asm_override_jmp_rel32(Asm *self, u32 asm_index, i32 new_relative);
CHECK_RESULT int asm_mov_rm32(Asm *self, Reg32 dst, Reg32 src);
CHECK_RESULT int asm_add_rm32(Asm *self, Reg32 dst, Reg32 src);
CHECK_RESULT int asm_sub_rm32(Asm *self, Reg32 dst, Reg32 src);
CHECK_RESULT int asm_and_rm32(Asm *self, Reg32 dst, Reg32 src);
CHECK_RESULT int asm_or_rm32(Asm *self, Reg32 dst, Reg32 src);
CHECK_RESULT int asm_xor_rm32(Asm *self, Reg32 dst, Reg32 src);
CHECK_RESULT int asm_cmp_rm32(Asm *self, Reg32 dst, Reg32 src);
CHECK_RESULT int asm_add_rm32_imm(Asm *self, Reg32 reg, i32 immediate);
CHECK_RESULT int asm_or_rm32_imm(Asm *self, Reg32 reg, i32 immediate);
CHECK_RESULT int asm_adc_rm32_imm(Asm *self, Reg32 reg, i32 immediate);
CHECK_RESULT int asm_sbb_rm32_imm(Asm *self, Reg32 reg, i32 immediate);
CHECK_RESULT int asm_and_rm32_imm(Asm *self, Reg32 reg, i32 immediate);
CHECK_RESULT int asm_sub_rm32_imm(Asm *self, Reg32 reg, i32 immediate);
CHECK_RESULT int asm_xor_rm32_imm(Asm *self, Reg32 reg, i32 immediate);
CHECK_RESULT int asm_cmp_rm32_imm(Asm *self, Reg32 reg, i32 immediate);
CHECK_RESULT int asm_rol_rm32_imm(Asm *self, Reg32 reg, i8 immediate);
CHECK_RESULT int asm_ror_rm32_imm(Asm *self, Reg32 reg, i8 immediate);
CHECK_RESULT int asm_rcl_rm32_imm(Asm *self, Reg32 reg, i8 immediate);
CHECK_RESULT int asm_rcr_rm32_imm(Asm *self, Reg32 reg, i8 immediate);
CHECK_RESULT int asm_shl_rm32_imm(Asm *self, Reg32 reg, i8 immediate);
CHECK_RESULT int asm_shr_rm32_imm(Asm *self, Reg32 reg, i8 immediate);
CHECK_RESULT int asm_sar_rm32_imm(Asm *self, Reg32 reg, i8 immediate);
CHECK_RESULT int asm_mov_rm64(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_add_rm64(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_sub_rm64(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_and_rm64(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_or_rm64(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_xor_rm64(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_cmp_rm64(Asm *self, Reg64 dst, Reg64 src);
CHECK_RESULT int asm_add_rm64_imm(Asm *self, Reg64 reg, i32 immediate);
CHECK_RESULT int asm_or_rm64_imm(Asm *self, Reg64 reg, i32 immediate);
CHECK_RESULT int asm_adc_rm64_imm(Asm *self, Reg64 reg, i32 immediate);
CHECK_RESULT int asm_sbb_rm64_imm(Asm *self, Reg64 reg, i32 immediate);
CHECK_RESULT int asm_and_rm64_imm(Asm *self, Reg64 reg, i32 immediate);
CHECK_RESULT int asm_sub_rm64_imm(Asm *self, Reg64 reg, i32 immediate);
CHECK_RESULT int asm_xor_rm64_imm(Asm *self, Reg64 reg, i32 immediate);
CHECK_RESULT int asm_cmp_rm64_imm(Asm *self, Reg64 reg, i32 immediate);
CHECK_RESULT int asm_rol_rm64_imm(Asm *self, Reg64 reg, i8 immediate);
CHECK_RESULT int asm_ror_rm64_imm(Asm *self, Reg64 reg, i8 immediate);
CHECK_RESULT int asm_rcl_rm64_imm(Asm *self, Reg64 reg, i8 immediate);
CHECK_RESULT int asm_rcr_rm64_imm(Asm *self, Reg64 reg, i8 immediate);
CHECK_RESULT int asm_shl_rm64_imm(Asm *self, Reg64 reg, i8 immediate);
CHECK_RESULT int asm_shr_rm64_imm(Asm *self, Reg64 reg, i8 immediate);
CHECK_RESULT int asm_sar_rm64_imm(Asm *self, Reg64 reg, i8 immediate);
CHECK_RESULT int asm_ret(Asm *self, u16 bytes);
#endif
|
