e67d406f2402c52ba528bd750468b9abb846c38a
[pcsx_rearmed.git] / deps / lightrec / lightrec-private.h
1 /* SPDX-License-Identifier: LGPL-2.1-or-later */
2 /*
3  * Copyright (C) 2016-2021 Paul Cercueil <paul@crapouillou.net>
4  */
5
6 #ifndef __LIGHTREC_PRIVATE_H__
7 #define __LIGHTREC_PRIVATE_H__
8
9 #include "lightning-wrapper.h"
10 #include "lightrec-config.h"
11 #include "disassembler.h"
12 #include "lightrec.h"
13 #include "regcache.h"
14
15 #if ENABLE_THREADED_COMPILER
16 #include <stdatomic.h>
17 #endif
18
19 #ifdef _MSC_BUILD
20 #include <immintrin.h>
21 #endif
22
23 #define ARRAY_SIZE(x) (sizeof(x) ? sizeof(x) / sizeof((x)[0]) : 0)
24
25 #define GENMASK(h, l) \
26         (((uintptr_t)-1 << (l)) & ((uintptr_t)-1 >> (__WORDSIZE - 1 - (h))))
27
28 #ifdef __GNUC__
29 #       define likely(x)       __builtin_expect(!!(x),1)
30 #       define unlikely(x)     __builtin_expect(!!(x),0)
31 #else
32 #       define likely(x)       (x)
33 #       define unlikely(x)     (x)
34 #endif
35
36 #if __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
37 #       define LE32TOH(x)       __builtin_bswap32(x)
38 #       define HTOLE32(x)       __builtin_bswap32(x)
39 #       define LE16TOH(x)       __builtin_bswap16(x)
40 #       define HTOLE16(x)       __builtin_bswap16(x)
41 #else
42 #       define LE32TOH(x)       (x)
43 #       define HTOLE32(x)       (x)
44 #       define LE16TOH(x)       (x)
45 #       define HTOLE16(x)       (x)
46 #endif
47
48 #if HAS_DEFAULT_ELM
49 #define SET_DEFAULT_ELM(table, value) [0 ... ARRAY_SIZE(table) - 1] = value
50 #else
51 #define SET_DEFAULT_ELM(table, value) [0] = NULL
52 #endif
53
54 #define fallthrough do {} while (0) /* fall-through */
55
56 #define container_of(ptr, type, member) \
57         ((type *)((void *)(ptr) - offsetof(type, member)))
58
59 #ifdef _MSC_BUILD
60 #       define popcount32(x)    __popcnt(x)
61 #       define clz32(x)         _lzcnt_u32(x)
62 #       define ctz32(x)         _tzcnt_u32(x)
63 #else
64 #       define popcount32(x)    __builtin_popcount(x)
65 #       define clz32(x)         __builtin_clz(x)
66 #       define ctz32(x)         __builtin_ctz(x)
67 #endif
68
69 /* Flags for (struct block *)->flags */
70 #define BLOCK_NEVER_COMPILE     BIT(0)
71 #define BLOCK_SHOULD_RECOMPILE  BIT(1)
72 #define BLOCK_FULLY_TAGGED      BIT(2)
73 #define BLOCK_IS_DEAD           BIT(3)
74 #define BLOCK_IS_MEMSET         BIT(4)
75 #define BLOCK_NO_OPCODE_LIST    BIT(5)
76
77 #define RAM_SIZE        0x200000
78 #define BIOS_SIZE       0x80000
79
80 #define CODE_LUT_SIZE   ((RAM_SIZE + BIOS_SIZE) >> 2)
81
82 #define REG_LO 32
83 #define REG_HI 33
84 #define REG_CP2_TEMP (offsetof(struct lightrec_state, cp2_temp_reg) / sizeof(u32))
85
86 /* Definition of jit_state_t (avoids inclusion of <lightning.h>) */
87 struct jit_node;
88 struct jit_state;
89 typedef struct jit_state jit_state_t;
90
91 struct blockcache;
92 struct recompiler;
93 struct regcache;
94 struct opcode;
95 struct reaper;
96
97 struct u16x2 {
98 #if __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
99         u16 h, l;
100 #else
101         u16 l, h;
102 #endif
103 };
104
105 struct block {
106         jit_state_t *_jit;
107         struct opcode *opcode_list;
108         void (*function)(void);
109         const u32 *code;
110         struct block *next;
111         u32 pc;
112         u32 hash;
113         u32 precompile_date;
114         unsigned int code_size;
115         u16 nb_ops;
116 #if ENABLE_THREADED_COMPILER
117         _Atomic u8 flags;
118 #else
119         u8 flags;
120 #endif
121 };
122
123 struct lightrec_branch {
124         struct jit_node *branch;
125         u32 target;
126 };
127
128 struct lightrec_branch_target {
129         struct jit_node *label;
130         u32 offset;
131 };
132
133 enum c_wrappers {
134         C_WRAPPER_RW,
135         C_WRAPPER_RW_GENERIC,
136         C_WRAPPER_MFC,
137         C_WRAPPER_MTC,
138         C_WRAPPER_CP,
139         C_WRAPPERS_COUNT,
140 };
141
142 struct lightrec_cstate {
143         struct lightrec_state *state;
144
145         struct lightrec_branch local_branches[512];
146         struct lightrec_branch_target targets[512];
147         unsigned int nb_local_branches;
148         unsigned int nb_targets;
149         unsigned int cycles;
150
151         struct regcache *reg_cache;
152 };
153
154 struct lightrec_state {
155         struct lightrec_registers regs;
156         u32 cp2_temp_reg;
157         u32 next_pc;
158         uintptr_t wrapper_regs[NUM_TEMPS];
159         u32 current_cycle;
160         u32 target_cycle;
161         u32 exit_flags;
162         u32 old_cycle_counter;
163         struct block *dispatcher, *c_wrapper_block;
164         void *c_wrappers[C_WRAPPERS_COUNT];
165         void *wrappers_eps[C_WRAPPERS_COUNT];
166         struct blockcache *block_cache;
167         struct recompiler *rec;
168         struct lightrec_cstate *cstate;
169         struct reaper *reaper;
170         void *tlsf;
171         void (*eob_wrapper_func)(void);
172         void (*memset_func)(void);
173         void (*get_next_block)(void);
174         struct lightrec_ops ops;
175         unsigned int nb_precompile;
176         unsigned int nb_maps;
177         const struct lightrec_mem_map *maps;
178         uintptr_t offset_ram, offset_bios, offset_scratch, offset_io;
179         _Bool with_32bit_lut;
180         _Bool mirrors_mapped;
181         _Bool invalidate_from_dma_only;
182         void *code_lut[];
183 };
184
185 u32 lightrec_rw(struct lightrec_state *state, union code op,
186                 u32 addr, u32 data, u32 *flags,
187                 struct block *block);
188
189 void lightrec_free_block(struct lightrec_state *state, struct block *block);
190
191 void remove_from_code_lut(struct blockcache *cache, struct block *block);
192
193 const struct lightrec_mem_map *
194 lightrec_get_map(struct lightrec_state *state, void **host, u32 kaddr);
195
196 static inline u32 kunseg(u32 addr)
197 {
198         if (unlikely(addr >= 0xa0000000))
199                 return addr - 0xa0000000;
200         else
201                 return addr &~ 0x80000000;
202 }
203
204 static inline u32 lut_offset(u32 pc)
205 {
206         if (pc & BIT(28))
207                 return ((pc & (BIOS_SIZE - 1)) + RAM_SIZE) >> 2; // BIOS
208         else
209                 return (pc & (RAM_SIZE - 1)) >> 2; // RAM
210 }
211
212 static inline _Bool is_big_endian(void)
213 {
214         return __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__;
215 }
216
217 static inline _Bool lut_is_32bit(const struct lightrec_state *state)
218 {
219         return __WORDSIZE == 32 ||
220                 (ENABLE_CODE_BUFFER && state->with_32bit_lut);
221 }
222
223 static inline size_t lut_elm_size(const struct lightrec_state *state)
224 {
225         return lut_is_32bit(state) ? 4 : sizeof(void *);
226 }
227
228 static inline void ** lut_address(struct lightrec_state *state, u32 offset)
229 {
230         if (lut_is_32bit(state))
231                 return (void **) ((uintptr_t) state->code_lut + offset * 4);
232         else
233                 return &state->code_lut[offset];
234 }
235
236 static inline void * lut_read(struct lightrec_state *state, u32 offset)
237 {
238         void **lut_entry = lut_address(state, offset);
239
240         if (lut_is_32bit(state))
241                 return (void *)(uintptr_t) *(u32 *) lut_entry;
242         else
243                 return *lut_entry;
244 }
245
246 static inline void lut_write(struct lightrec_state *state, u32 offset, void *ptr)
247 {
248         void **lut_entry = lut_address(state, offset);
249
250         if (lut_is_32bit(state))
251                 *(u32 *) lut_entry = (u32)(uintptr_t) ptr;
252         else
253                 *lut_entry = ptr;
254 }
255
256 static inline u32 get_ds_pc(const struct block *block, u16 offset, s16 imm)
257 {
258         u16 flags = block->opcode_list[offset].flags;
259
260         offset += op_flag_no_ds(flags);
261
262         return block->pc + (offset + imm << 2);
263 }
264
265 static inline u32 get_branch_pc(const struct block *block, u16 offset, s16 imm)
266 {
267         u16 flags = block->opcode_list[offset].flags;
268
269         offset -= op_flag_no_ds(flags);
270
271         return block->pc + (offset + imm << 2);
272 }
273
274 void lightrec_mtc(struct lightrec_state *state, union code op, u8 reg, u32 data);
275 u32 lightrec_mfc(struct lightrec_state *state, union code op);
276 void lightrec_rfe(struct lightrec_state *state);
277 void lightrec_cp(struct lightrec_state *state, union code op);
278
279 struct lightrec_cstate * lightrec_create_cstate(struct lightrec_state *state);
280 void lightrec_free_cstate(struct lightrec_cstate *cstate);
281
282 union code lightrec_read_opcode(struct lightrec_state *state, u32 pc);
283
284 int lightrec_compile_block(struct lightrec_cstate *cstate, struct block *block);
285 void lightrec_free_opcode_list(struct lightrec_state *state,
286                                struct opcode *list);
287
288 unsigned int lightrec_cycles_of_opcode(union code code);
289
290 static inline u8 get_mult_div_lo(union code c)
291 {
292         return (OPT_FLAG_MULT_DIV && c.r.rd) ? c.r.rd : REG_LO;
293 }
294
295 static inline u8 get_mult_div_hi(union code c)
296 {
297         return (OPT_FLAG_MULT_DIV && c.r.imm) ? c.r.imm : REG_HI;
298 }
299
300 static inline s16 s16_max(s16 a, s16 b)
301 {
302         return a > b ? a : b;
303 }
304
305 static inline _Bool block_has_flag(struct block *block, u8 flag)
306 {
307 #if ENABLE_THREADED_COMPILER
308         return atomic_load_explicit(&block->flags, memory_order_relaxed) & flag;
309 #else
310         return block->flags & flag;
311 #endif
312 }
313
314 static inline u8 block_set_flags(struct block *block, u8 mask)
315 {
316 #if ENABLE_THREADED_COMPILER
317         return atomic_fetch_or_explicit(&block->flags, mask,
318                                         memory_order_relaxed);
319 #else
320         u8 flags = block->flags;
321
322         block->flags |= mask;
323
324         return flags;
325 #endif
326 }
327
328 static inline u8 block_clear_flags(struct block *block, u8 mask)
329 {
330 #if ENABLE_THREADED_COMPILER
331         return atomic_fetch_and_explicit(&block->flags, ~mask,
332                                          memory_order_relaxed);
333 #else
334         u8 flags = block->flags;
335
336         block->flags &= ~mask;
337
338         return flags;
339 #endif
340 }
341
342 static inline _Bool can_sign_extend(s32 value, u8 order)
343 {
344       return (u32)(value >> order - 1) + 1 < 2;
345 }
346
347 static inline _Bool can_zero_extend(u32 value, u8 order)
348 {
349       return (value >> order) == 0;
350 }
351
352 #endif /* __LIGHTREC_PRIVATE_H__ */