1 /*
2  *    Stack-less Just-In-Time compiler
3  *
4  *    Copyright 2009-2012 Zoltan Herczeg (hzmester@freemail.hu). All rights reserved.
5  *
6  * Redistribution and use in source and binary forms, with or without modification, are
7  * permitted provided that the following conditions are met:
8  *
9  *   1. Redistributions of source code must retain the above copyright notice, this list of
10  *      conditions and the following disclaimer.
11  *
12  *   2. Redistributions in binary form must reproduce the above copyright notice, this list
13  *      of conditions and the following disclaimer in the documentation and/or other materials
14  *      provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER(S) AND CONTRIBUTORS ``AS IS'' AND ANY
17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT
19  * SHALL THE COPYRIGHT HOLDER(S) OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
21  * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
22  * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
23  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
24  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25  */
26 
sljit_get_platform_name(void)27 SLJIT_API_FUNC_ATTRIBUTE SLJIT_CONST char* sljit_get_platform_name(void)
28 {
29 	return "ARM-Thumb2" SLJIT_CPUINFO;
30 }
31 
32 /* Length of an instruction word. */
33 typedef sljit_ui sljit_ins;
34 
35 /* Last register + 1. */
36 #define TMP_REG1	(SLJIT_NUMBER_OF_REGISTERS + 2)
37 #define TMP_REG2	(SLJIT_NUMBER_OF_REGISTERS + 3)
38 #define TMP_REG3	(SLJIT_NUMBER_OF_REGISTERS + 4)
39 #define TMP_PC		(SLJIT_NUMBER_OF_REGISTERS + 5)
40 
41 #define TMP_FREG1	(0)
42 #define TMP_FREG2	(SLJIT_NUMBER_OF_FLOAT_REGISTERS + 1)
43 
44 /* See sljit_emit_enter and sljit_emit_op0 if you want to change them. */
45 static SLJIT_CONST sljit_ub reg_map[SLJIT_NUMBER_OF_REGISTERS + 6] = {
46 	0, 0, 1, 2, 12, 11, 10, 9, 8, 7, 6, 5, 13, 3, 4, 14, 15
47 };
48 
49 #define COPY_BITS(src, from, to, bits) \
50 	((from >= to ? (src >> (from - to)) : (src << (to - from))) & (((1 << bits) - 1) << to))
51 
52 /* Thumb16 encodings. */
53 #define RD3(rd) (reg_map[rd])
54 #define RN3(rn) (reg_map[rn] << 3)
55 #define RM3(rm) (reg_map[rm] << 6)
56 #define RDN3(rdn) (reg_map[rdn] << 8)
57 #define IMM3(imm) (imm << 6)
58 #define IMM8(imm) (imm)
59 
60 /* Thumb16 helpers. */
61 #define SET_REGS44(rd, rn) \
62 	((reg_map[rn] << 3) | (reg_map[rd] & 0x7) | ((reg_map[rd] & 0x8) << 4))
63 #define IS_2_LO_REGS(reg1, reg2) \
64 	(reg_map[reg1] <= 7 && reg_map[reg2] <= 7)
65 #define IS_3_LO_REGS(reg1, reg2, reg3) \
66 	(reg_map[reg1] <= 7 && reg_map[reg2] <= 7 && reg_map[reg3] <= 7)
67 
68 /* Thumb32 encodings. */
69 #define RD4(rd) (reg_map[rd] << 8)
70 #define RN4(rn) (reg_map[rn] << 16)
71 #define RM4(rm) (reg_map[rm])
72 #define RT4(rt) (reg_map[rt] << 12)
73 #define DD4(dd) ((dd) << 12)
74 #define DN4(dn) ((dn) << 16)
75 #define DM4(dm) (dm)
76 #define IMM5(imm) \
77 	(COPY_BITS(imm, 2, 12, 3) | ((imm & 0x3) << 6))
78 #define IMM12(imm) \
79 	(COPY_BITS(imm, 11, 26, 1) | COPY_BITS(imm, 8, 12, 3) | (imm & 0xff))
80 
81 /* --------------------------------------------------------------------- */
82 /*  Instrucion forms                                                     */
83 /* --------------------------------------------------------------------- */
84 
85 /* dot '.' changed to _
86    I immediate form (possibly followed by number of immediate bits). */
87 #define ADCI		0xf1400000
88 #define ADCS		0x4140
89 #define ADC_W		0xeb400000
90 #define ADD		0x4400
91 #define ADDS		0x1800
92 #define ADDSI3		0x1c00
93 #define ADDSI8		0x3000
94 #define ADD_W		0xeb000000
95 #define ADDWI		0xf2000000
96 #define ADD_SP		0xb000
97 #define ADD_W		0xeb000000
98 #define ADD_WI		0xf1000000
99 #define ANDI		0xf0000000
100 #define ANDS		0x4000
101 #define AND_W		0xea000000
102 #define ASRS		0x4100
103 #define ASRSI		0x1000
104 #define ASR_W		0xfa40f000
105 #define ASR_WI		0xea4f0020
106 #define BICI		0xf0200000
107 #define BKPT		0xbe00
108 #define BLX		0x4780
109 #define BX		0x4700
110 #define CLZ		0xfab0f080
111 #define CMPI		0x2800
112 #define CMP_W		0xebb00f00
113 #define EORI		0xf0800000
114 #define EORS		0x4040
115 #define EOR_W		0xea800000
116 #define IT		0xbf00
117 #define LSLS		0x4080
118 #define LSLSI		0x0000
119 #define LSL_W		0xfa00f000
120 #define LSL_WI		0xea4f0000
121 #define LSRS		0x40c0
122 #define LSRSI		0x0800
123 #define LSR_W		0xfa20f000
124 #define LSR_WI		0xea4f0010
125 #define MOV		0x4600
126 #define MOVS		0x0000
127 #define MOVSI		0x2000
128 #define MOVT		0xf2c00000
129 #define MOVW		0xf2400000
130 #define MOV_W		0xea4f0000
131 #define MOV_WI		0xf04f0000
132 #define MUL		0xfb00f000
133 #define MVNS		0x43c0
134 #define MVN_W		0xea6f0000
135 #define MVN_WI		0xf06f0000
136 #define NOP		0xbf00
137 #define ORNI		0xf0600000
138 #define ORRI		0xf0400000
139 #define ORRS		0x4300
140 #define ORR_W		0xea400000
141 #define POP		0xbc00
142 #define POP_W		0xe8bd0000
143 #define PUSH		0xb400
144 #define PUSH_W		0xe92d0000
145 #define RSB_WI		0xf1c00000
146 #define RSBSI		0x4240
147 #define SBCI		0xf1600000
148 #define SBCS		0x4180
149 #define SBC_W		0xeb600000
150 #define SMULL		0xfb800000
151 #define STR_SP		0x9000
152 #define SUBS		0x1a00
153 #define SUBSI3		0x1e00
154 #define SUBSI8		0x3800
155 #define SUB_W		0xeba00000
156 #define SUBWI		0xf2a00000
157 #define SUB_SP		0xb080
158 #define SUB_WI		0xf1a00000
159 #define SXTB		0xb240
160 #define SXTB_W		0xfa4ff080
161 #define SXTH		0xb200
162 #define SXTH_W		0xfa0ff080
163 #define TST		0x4200
164 #define UMULL		0xfba00000
165 #define UXTB		0xb2c0
166 #define UXTB_W		0xfa5ff080
167 #define UXTH		0xb280
168 #define UXTH_W		0xfa1ff080
169 #define VABS_F32	0xeeb00ac0
170 #define VADD_F32	0xee300a00
171 #define VCMP_F32	0xeeb40a40
172 #define VCVT_F32_S32	0xeeb80ac0
173 #define VCVT_F64_F32	0xeeb70ac0
174 #define VCVT_S32_F32	0xeebd0ac0
175 #define VDIV_F32	0xee800a00
176 #define VMOV_F32	0xeeb00a40
177 #define VMOV		0xee000a10
178 #define VMRS		0xeef1fa10
179 #define VMUL_F32	0xee200a00
180 #define VNEG_F32	0xeeb10a40
181 #define VSTR_F32	0xed000a00
182 #define VSUB_F32	0xee300a40
183 
push_inst16(struct sljit_compiler * compiler,sljit_ins inst)184 static sljit_si push_inst16(struct sljit_compiler *compiler, sljit_ins inst)
185 {
186 	sljit_uh *ptr;
187 	SLJIT_ASSERT(!(inst & 0xffff0000));
188 
189 	ptr = (sljit_uh*)ensure_buf(compiler, sizeof(sljit_uh));
190 	FAIL_IF(!ptr);
191 	*ptr = inst;
192 	compiler->size++;
193 	return SLJIT_SUCCESS;
194 }
195 
push_inst32(struct sljit_compiler * compiler,sljit_ins inst)196 static sljit_si push_inst32(struct sljit_compiler *compiler, sljit_ins inst)
197 {
198 	sljit_uh *ptr = (sljit_uh*)ensure_buf(compiler, sizeof(sljit_ins));
199 	FAIL_IF(!ptr);
200 	*ptr++ = inst >> 16;
201 	*ptr = inst;
202 	compiler->size += 2;
203 	return SLJIT_SUCCESS;
204 }
205 
emit_imm32_const(struct sljit_compiler * compiler,sljit_si dst,sljit_uw imm)206 static SLJIT_INLINE sljit_si emit_imm32_const(struct sljit_compiler *compiler, sljit_si dst, sljit_uw imm)
207 {
208 	FAIL_IF(push_inst32(compiler, MOVW | RD4(dst) |
209 		COPY_BITS(imm, 12, 16, 4) | COPY_BITS(imm, 11, 26, 1) | COPY_BITS(imm, 8, 12, 3) | (imm & 0xff)));
210 	return push_inst32(compiler, MOVT | RD4(dst) |
211 		COPY_BITS(imm, 12 + 16, 16, 4) | COPY_BITS(imm, 11 + 16, 26, 1) | COPY_BITS(imm, 8 + 16, 12, 3) | ((imm & 0xff0000) >> 16));
212 }
213 
modify_imm32_const(sljit_uh * inst,sljit_uw new_imm)214 static SLJIT_INLINE void modify_imm32_const(sljit_uh *inst, sljit_uw new_imm)
215 {
216 	sljit_si dst = inst[1] & 0x0f00;
217 	SLJIT_ASSERT(((inst[0] & 0xfbf0) == (MOVW >> 16)) && ((inst[2] & 0xfbf0) == (MOVT >> 16)) && dst == (inst[3] & 0x0f00));
218 	inst[0] = (MOVW >> 16) | COPY_BITS(new_imm, 12, 0, 4) | COPY_BITS(new_imm, 11, 10, 1);
219 	inst[1] = dst | COPY_BITS(new_imm, 8, 12, 3) | (new_imm & 0xff);
220 	inst[2] = (MOVT >> 16) | COPY_BITS(new_imm, 12 + 16, 0, 4) | COPY_BITS(new_imm, 11 + 16, 10, 1);
221 	inst[3] = dst | COPY_BITS(new_imm, 8 + 16, 12, 3) | ((new_imm & 0xff0000) >> 16);
222 }
223 
detect_jump_type(struct sljit_jump * jump,sljit_uh * code_ptr,sljit_uh * code)224 static SLJIT_INLINE sljit_si detect_jump_type(struct sljit_jump *jump, sljit_uh *code_ptr, sljit_uh *code)
225 {
226 	sljit_sw diff;
227 
228 	if (jump->flags & SLJIT_REWRITABLE_JUMP)
229 		return 0;
230 
231 	if (jump->flags & JUMP_ADDR) {
232 		/* Branch to ARM code is not optimized yet. */
233 		if (!(jump->u.target & 0x1))
234 			return 0;
235 		diff = ((sljit_sw)jump->u.target - (sljit_sw)(code_ptr + 2)) >> 1;
236 	}
237 	else {
238 		SLJIT_ASSERT(jump->flags & JUMP_LABEL);
239 		diff = ((sljit_sw)(code + jump->u.label->size) - (sljit_sw)(code_ptr + 2)) >> 1;
240 	}
241 
242 	if (jump->flags & IS_COND) {
243 		SLJIT_ASSERT(!(jump->flags & IS_BL));
244 		if (diff <= 127 && diff >= -128) {
245 			jump->flags |= PATCH_TYPE1;
246 			return 5;
247 		}
248 		if (diff <= 524287 && diff >= -524288) {
249 			jump->flags |= PATCH_TYPE2;
250 			return 4;
251 		}
252 		/* +1 comes from the prefix IT instruction. */
253 		diff--;
254 		if (diff <= 8388607 && diff >= -8388608) {
255 			jump->flags |= PATCH_TYPE3;
256 			return 3;
257 		}
258 	}
259 	else if (jump->flags & IS_BL) {
260 		if (diff <= 8388607 && diff >= -8388608) {
261 			jump->flags |= PATCH_BL;
262 			return 3;
263 		}
264 	}
265 	else {
266 		if (diff <= 1023 && diff >= -1024) {
267 			jump->flags |= PATCH_TYPE4;
268 			return 4;
269 		}
270 		if (diff <= 8388607 && diff >= -8388608) {
271 			jump->flags |= PATCH_TYPE5;
272 			return 3;
273 		}
274 	}
275 
276 	return 0;
277 }
278 
set_jump_instruction(struct sljit_jump * jump)279 static SLJIT_INLINE void set_jump_instruction(struct sljit_jump *jump)
280 {
281 	sljit_si type = (jump->flags >> 4) & 0xf;
282 	sljit_sw diff;
283 	sljit_uh *jump_inst;
284 	sljit_si s, j1, j2;
285 
286 	if (SLJIT_UNLIKELY(type == 0)) {
287 		modify_imm32_const((sljit_uh*)jump->addr, (jump->flags & JUMP_LABEL) ? jump->u.label->addr : jump->u.target);
288 		return;
289 	}
290 
291 	if (jump->flags & JUMP_ADDR) {
292 		SLJIT_ASSERT(jump->u.target & 0x1);
293 		diff = ((sljit_sw)jump->u.target - (sljit_sw)(jump->addr + 4)) >> 1;
294 	}
295 	else
296 		diff = ((sljit_sw)(jump->u.label->addr) - (sljit_sw)(jump->addr + 4)) >> 1;
297 	jump_inst = (sljit_uh*)jump->addr;
298 
299 	switch (type) {
300 	case 1:
301 		/* Encoding T1 of 'B' instruction */
302 		SLJIT_ASSERT(diff <= 127 && diff >= -128 && (jump->flags & IS_COND));
303 		jump_inst[0] = 0xd000 | (jump->flags & 0xf00) | (diff & 0xff);
304 		return;
305 	case 2:
306 		/* Encoding T3 of 'B' instruction */
307 		SLJIT_ASSERT(diff <= 524287 && diff >= -524288 && (jump->flags & IS_COND));
308 		jump_inst[0] = 0xf000 | COPY_BITS(jump->flags, 8, 6, 4) | COPY_BITS(diff, 11, 0, 6) | COPY_BITS(diff, 19, 10, 1);
309 		jump_inst[1] = 0x8000 | COPY_BITS(diff, 17, 13, 1) | COPY_BITS(diff, 18, 11, 1) | (diff & 0x7ff);
310 		return;
311 	case 3:
312 		SLJIT_ASSERT(jump->flags & IS_COND);
313 		*jump_inst++ = IT | ((jump->flags >> 4) & 0xf0) | 0x8;
314 		diff--;
315 		type = 5;
316 		break;
317 	case 4:
318 		/* Encoding T2 of 'B' instruction */
319 		SLJIT_ASSERT(diff <= 1023 && diff >= -1024 && !(jump->flags & IS_COND));
320 		jump_inst[0] = 0xe000 | (diff & 0x7ff);
321 		return;
322 	}
323 
324 	SLJIT_ASSERT(diff <= 8388607 && diff >= -8388608);
325 
326 	/* Really complex instruction form for branches. */
327 	s = (diff >> 23) & 0x1;
328 	j1 = (~(diff >> 21) ^ s) & 0x1;
329 	j2 = (~(diff >> 22) ^ s) & 0x1;
330 	jump_inst[0] = 0xf000 | (s << 10) | COPY_BITS(diff, 11, 0, 10);
331 	jump_inst[1] = (j1 << 13) | (j2 << 11) | (diff & 0x7ff);
332 
333 	/* The others have a common form. */
334 	if (type == 5) /* Encoding T4 of 'B' instruction */
335 		jump_inst[1] |= 0x9000;
336 	else if (type == 6) /* Encoding T1 of 'BL' instruction */
337 		jump_inst[1] |= 0xd000;
338 	else
339 		SLJIT_ASSERT_STOP();
340 }
341 
sljit_generate_code(struct sljit_compiler * compiler)342 SLJIT_API_FUNC_ATTRIBUTE void* sljit_generate_code(struct sljit_compiler *compiler)
343 {
344 	struct sljit_memory_fragment *buf;
345 	sljit_uh *code;
346 	sljit_uh *code_ptr;
347 	sljit_uh *buf_ptr;
348 	sljit_uh *buf_end;
349 	sljit_uw half_count;
350 
351 	struct sljit_label *label;
352 	struct sljit_jump *jump;
353 	struct sljit_const *const_;
354 
355 	CHECK_ERROR_PTR();
356 	CHECK_PTR(check_sljit_generate_code(compiler));
357 	reverse_buf(compiler);
358 
359 	code = (sljit_uh*)SLJIT_MALLOC_EXEC(compiler->size * sizeof(sljit_uh));
360 	PTR_FAIL_WITH_EXEC_IF(code);
361 	buf = compiler->buf;
362 
363 	code_ptr = code;
364 	half_count = 0;
365 	label = compiler->labels;
366 	jump = compiler->jumps;
367 	const_ = compiler->consts;
368 
369 	do {
370 		buf_ptr = (sljit_uh*)buf->memory;
371 		buf_end = buf_ptr + (buf->used_size >> 1);
372 		do {
373 			*code_ptr = *buf_ptr++;
374 			/* These structures are ordered by their address. */
375 			SLJIT_ASSERT(!label || label->size >= half_count);
376 			SLJIT_ASSERT(!jump || jump->addr >= half_count);
377 			SLJIT_ASSERT(!const_ || const_->addr >= half_count);
378 			if (label && label->size == half_count) {
379 				label->addr = ((sljit_uw)code_ptr) | 0x1;
380 				label->size = code_ptr - code;
381 				label = label->next;
382 			}
383 			if (jump && jump->addr == half_count) {
384 					jump->addr = (sljit_uw)code_ptr - ((jump->flags & IS_COND) ? 10 : 8);
385 					code_ptr -= detect_jump_type(jump, code_ptr, code);
386 					jump = jump->next;
387 			}
388 			if (const_ && const_->addr == half_count) {
389 				const_->addr = (sljit_uw)code_ptr;
390 				const_ = const_->next;
391 			}
392 			code_ptr ++;
393 			half_count ++;
394 		} while (buf_ptr < buf_end);
395 
396 		buf = buf->next;
397 	} while (buf);
398 
399 	if (label && label->size == half_count) {
400 		label->addr = ((sljit_uw)code_ptr) | 0x1;
401 		label->size = code_ptr - code;
402 		label = label->next;
403 	}
404 
405 	SLJIT_ASSERT(!label);
406 	SLJIT_ASSERT(!jump);
407 	SLJIT_ASSERT(!const_);
408 	SLJIT_ASSERT(code_ptr - code <= (sljit_sw)compiler->size);
409 
410 	jump = compiler->jumps;
411 	while (jump) {
412 		set_jump_instruction(jump);
413 		jump = jump->next;
414 	}
415 
416 	compiler->error = SLJIT_ERR_COMPILED;
417 	compiler->executable_size = (code_ptr - code) * sizeof(sljit_uh);
418 	SLJIT_CACHE_FLUSH(code, code_ptr);
419 	/* Set thumb mode flag. */
420 	return (void*)((sljit_uw)code | 0x1);
421 }
422 
423 /* --------------------------------------------------------------------- */
424 /*  Core code generator functions.                                       */
425 /* --------------------------------------------------------------------- */
426 
427 #define INVALID_IMM	0x80000000
get_imm(sljit_uw imm)428 static sljit_uw get_imm(sljit_uw imm)
429 {
430 	/* Thumb immediate form. */
431 	sljit_si counter;
432 
433 	if (imm <= 0xff)
434 		return imm;
435 
436 	if ((imm & 0xffff) == (imm >> 16)) {
437 		/* Some special cases. */
438 		if (!(imm & 0xff00))
439 			return (1 << 12) | (imm & 0xff);
440 		if (!(imm & 0xff))
441 			return (2 << 12) | ((imm >> 8) & 0xff);
442 		if ((imm & 0xff00) == ((imm & 0xff) << 8))
443 			return (3 << 12) | (imm & 0xff);
444 	}
445 
446 	/* Assembly optimization: count leading zeroes? */
447 	counter = 8;
448 	if (!(imm & 0xffff0000)) {
449 		counter += 16;
450 		imm <<= 16;
451 	}
452 	if (!(imm & 0xff000000)) {
453 		counter += 8;
454 		imm <<= 8;
455 	}
456 	if (!(imm & 0xf0000000)) {
457 		counter += 4;
458 		imm <<= 4;
459 	}
460 	if (!(imm & 0xc0000000)) {
461 		counter += 2;
462 		imm <<= 2;
463 	}
464 	if (!(imm & 0x80000000)) {
465 		counter += 1;
466 		imm <<= 1;
467 	}
468 	/* Since imm >= 128, this must be true. */
469 	SLJIT_ASSERT(counter <= 31);
470 
471 	if (imm & 0x00ffffff)
472 		return INVALID_IMM; /* Cannot be encoded. */
473 
474 	return ((imm >> 24) & 0x7f) | COPY_BITS(counter, 4, 26, 1) | COPY_BITS(counter, 1, 12, 3) | COPY_BITS(counter, 0, 7, 1);
475 }
476 
load_immediate(struct sljit_compiler * compiler,sljit_si dst,sljit_uw imm)477 static sljit_si load_immediate(struct sljit_compiler *compiler, sljit_si dst, sljit_uw imm)
478 {
479 	sljit_uw tmp;
480 
481 	if (imm >= 0x10000) {
482 		tmp = get_imm(imm);
483 		if (tmp != INVALID_IMM)
484 			return push_inst32(compiler, MOV_WI | RD4(dst) | tmp);
485 		tmp = get_imm(~imm);
486 		if (tmp != INVALID_IMM)
487 			return push_inst32(compiler, MVN_WI | RD4(dst) | tmp);
488 	}
489 
490 	/* set low 16 bits, set hi 16 bits to 0. */
491 	FAIL_IF(push_inst32(compiler, MOVW | RD4(dst) |
492 		COPY_BITS(imm, 12, 16, 4) | COPY_BITS(imm, 11, 26, 1) | COPY_BITS(imm, 8, 12, 3) | (imm & 0xff)));
493 
494 	/* set hi 16 bit if needed. */
495 	if (imm >= 0x10000)
496 		return push_inst32(compiler, MOVT | RD4(dst) |
497 			COPY_BITS(imm, 12 + 16, 16, 4) | COPY_BITS(imm, 11 + 16, 26, 1) | COPY_BITS(imm, 8 + 16, 12, 3) | ((imm & 0xff0000) >> 16));
498 	return SLJIT_SUCCESS;
499 }
500 
501 #define ARG1_IMM	0x0010000
502 #define ARG2_IMM	0x0020000
503 #define KEEP_FLAGS	0x0040000
504 /* SET_FLAGS must be 0x100000 as it is also the value of S bit (can be used for optimization). */
505 #define SET_FLAGS	0x0100000
506 #define UNUSED_RETURN	0x0200000
507 #define SLOW_DEST	0x0400000
508 #define SLOW_SRC1	0x0800000
509 #define SLOW_SRC2	0x1000000
510 
emit_op_imm(struct sljit_compiler * compiler,sljit_si flags,sljit_si dst,sljit_uw arg1,sljit_uw arg2)511 static sljit_si emit_op_imm(struct sljit_compiler *compiler, sljit_si flags, sljit_si dst, sljit_uw arg1, sljit_uw arg2)
512 {
513 	/* dst must be register, TMP_REG1
514 	   arg1 must be register, TMP_REG1, imm
515 	   arg2 must be register, TMP_REG2, imm */
516 	sljit_si reg;
517 	sljit_uw imm, nimm;
518 
519 	if (SLJIT_UNLIKELY((flags & (ARG1_IMM | ARG2_IMM)) == (ARG1_IMM | ARG2_IMM))) {
520 		/* Both are immediates. */
521 		flags &= ~ARG1_IMM;
522 		FAIL_IF(load_immediate(compiler, TMP_REG1, arg1));
523 		arg1 = TMP_REG1;
524 	}
525 
526 	if (flags & (ARG1_IMM | ARG2_IMM)) {
527 		reg = (flags & ARG2_IMM) ? arg1 : arg2;
528 		imm = (flags & ARG2_IMM) ? arg2 : arg1;
529 
530 		switch (flags & 0xffff) {
531 		case SLJIT_CLZ:
532 		case SLJIT_MUL:
533 			/* No form with immediate operand. */
534 			break;
535 		case SLJIT_MOV:
536 			SLJIT_ASSERT(!(flags & SET_FLAGS) && (flags & ARG2_IMM) && arg1 == TMP_REG1);
537 			return load_immediate(compiler, dst, imm);
538 		case SLJIT_NOT:
539 			if (!(flags & SET_FLAGS))
540 				return load_immediate(compiler, dst, ~imm);
541 			/* Since the flags should be set, we just fallback to the register mode.
542 			   Although some clever things could be done here, "NOT IMM" does not worth the efforts. */
543 			break;
544 		case SLJIT_ADD:
545 			nimm = -imm;
546 			if (!(flags & KEEP_FLAGS) && IS_2_LO_REGS(reg, dst)) {
547 				if (imm <= 0x7)
548 					return push_inst16(compiler, ADDSI3 | IMM3(imm) | RD3(dst) | RN3(reg));
549 				if (nimm <= 0x7)
550 					return push_inst16(compiler, SUBSI3 | IMM3(nimm) | RD3(dst) | RN3(reg));
551 				if (reg == dst) {
552 					if (imm <= 0xff)
553 						return push_inst16(compiler, ADDSI8 | IMM8(imm) | RDN3(dst));
554 					if (nimm <= 0xff)
555 						return push_inst16(compiler, SUBSI8 | IMM8(nimm) | RDN3(dst));
556 				}
557 			}
558 			if (!(flags & SET_FLAGS)) {
559 				if (imm <= 0xfff)
560 					return push_inst32(compiler, ADDWI | RD4(dst) | RN4(reg) | IMM12(imm));
561 				if (nimm <= 0xfff)
562 					return push_inst32(compiler, SUBWI | RD4(dst) | RN4(reg) | IMM12(nimm));
563 			}
564 			imm = get_imm(imm);
565 			if (imm != INVALID_IMM)
566 				return push_inst32(compiler, ADD_WI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | imm);
567 			break;
568 		case SLJIT_ADDC:
569 			imm = get_imm(imm);
570 			if (imm != INVALID_IMM)
571 				return push_inst32(compiler, ADCI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | imm);
572 			break;
573 		case SLJIT_SUB:
574 			if (flags & ARG1_IMM) {
575 				if (!(flags & KEEP_FLAGS) && imm == 0 && IS_2_LO_REGS(reg, dst))
576 					return push_inst16(compiler, RSBSI | RD3(dst) | RN3(reg));
577 				imm = get_imm(imm);
578 				if (imm != INVALID_IMM)
579 					return push_inst32(compiler, RSB_WI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | imm);
580 				break;
581 			}
582 			nimm = -imm;
583 			if (!(flags & KEEP_FLAGS) && IS_2_LO_REGS(reg, dst)) {
584 				if (imm <= 0x7)
585 					return push_inst16(compiler, SUBSI3 | IMM3(imm) | RD3(dst) | RN3(reg));
586 				if (nimm <= 0x7)
587 					return push_inst16(compiler, ADDSI3 | IMM3(nimm) | RD3(dst) | RN3(reg));
588 				if (reg == dst) {
589 					if (imm <= 0xff)
590 						return push_inst16(compiler, SUBSI8 | IMM8(imm) | RDN3(dst));
591 					if (nimm <= 0xff)
592 						return push_inst16(compiler, ADDSI8 | IMM8(nimm) | RDN3(dst));
593 				}
594 				if (imm <= 0xff && (flags & UNUSED_RETURN))
595 					return push_inst16(compiler, CMPI | IMM8(imm) | RDN3(reg));
596 			}
597 			if (!(flags & SET_FLAGS)) {
598 				if (imm <= 0xfff)
599 					return push_inst32(compiler, SUBWI | RD4(dst) | RN4(reg) | IMM12(imm));
600 				if (nimm <= 0xfff)
601 					return push_inst32(compiler, ADDWI | RD4(dst) | RN4(reg) | IMM12(nimm));
602 			}
603 			imm = get_imm(imm);
604 			if (imm != INVALID_IMM)
605 				return push_inst32(compiler, SUB_WI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | imm);
606 			break;
607 		case SLJIT_SUBC:
608 			if (flags & ARG1_IMM)
609 				break;
610 			imm = get_imm(imm);
611 			if (imm != INVALID_IMM)
612 				return push_inst32(compiler, SBCI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | imm);
613 			break;
614 		case SLJIT_AND:
615 			nimm = get_imm(imm);
616 			if (nimm != INVALID_IMM)
617 				return push_inst32(compiler, ANDI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | nimm);
618 			imm = get_imm(imm);
619 			if (imm != INVALID_IMM)
620 				return push_inst32(compiler, BICI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | imm);
621 			break;
622 		case SLJIT_OR:
623 			nimm = get_imm(imm);
624 			if (nimm != INVALID_IMM)
625 				return push_inst32(compiler, ORRI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | nimm);
626 			imm = get_imm(imm);
627 			if (imm != INVALID_IMM)
628 				return push_inst32(compiler, ORNI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | imm);
629 			break;
630 		case SLJIT_XOR:
631 			imm = get_imm(imm);
632 			if (imm != INVALID_IMM)
633 				return push_inst32(compiler, EORI | (flags & SET_FLAGS) | RD4(dst) | RN4(reg) | imm);
634 			break;
635 		case SLJIT_SHL:
636 		case SLJIT_LSHR:
637 		case SLJIT_ASHR:
638 			if (flags & ARG1_IMM)
639 				break;
640 			imm &= 0x1f;
641 			if (imm == 0) {
642 				if (!(flags & SET_FLAGS))
643 					return push_inst16(compiler, MOV | SET_REGS44(dst, reg));
644 				if (IS_2_LO_REGS(dst, reg))
645 					return push_inst16(compiler, MOVS | RD3(dst) | RN3(reg));
646 				return push_inst32(compiler, MOV_W | SET_FLAGS | RD4(dst) | RM4(reg));
647 			}
648 			switch (flags & 0xffff) {
649 			case SLJIT_SHL:
650 				if (!(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, reg))
651 					return push_inst16(compiler, LSLSI | RD3(dst) | RN3(reg) | (imm << 6));
652 				return push_inst32(compiler, LSL_WI | (flags & SET_FLAGS) | RD4(dst) | RM4(reg) | IMM5(imm));
653 			case SLJIT_LSHR:
654 				if (!(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, reg))
655 					return push_inst16(compiler, LSRSI | RD3(dst) | RN3(reg) | (imm << 6));
656 				return push_inst32(compiler, LSR_WI | (flags & SET_FLAGS) | RD4(dst) | RM4(reg) | IMM5(imm));
657 			default: /* SLJIT_ASHR */
658 				if (!(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, reg))
659 					return push_inst16(compiler, ASRSI | RD3(dst) | RN3(reg) | (imm << 6));
660 				return push_inst32(compiler, ASR_WI | (flags & SET_FLAGS) | RD4(dst) | RM4(reg) | IMM5(imm));
661 			}
662 		default:
663 			SLJIT_ASSERT_STOP();
664 			break;
665 		}
666 
667 		if (flags & ARG2_IMM) {
668 			FAIL_IF(load_immediate(compiler, TMP_REG2, arg2));
669 			arg2 = TMP_REG2;
670 		}
671 		else {
672 			FAIL_IF(load_immediate(compiler, TMP_REG1, arg1));
673 			arg1 = TMP_REG1;
674 		}
675 	}
676 
677 	/* Both arguments are registers. */
678 	switch (flags & 0xffff) {
679 	case SLJIT_MOV:
680 	case SLJIT_MOV_UI:
681 	case SLJIT_MOV_SI:
682 	case SLJIT_MOV_P:
683 	case SLJIT_MOVU:
684 	case SLJIT_MOVU_UI:
685 	case SLJIT_MOVU_SI:
686 	case SLJIT_MOVU_P:
687 		SLJIT_ASSERT(!(flags & SET_FLAGS) && arg1 == TMP_REG1);
688 		if (dst == arg2)
689 			return SLJIT_SUCCESS;
690 		return push_inst16(compiler, MOV | SET_REGS44(dst, arg2));
691 	case SLJIT_MOV_UB:
692 	case SLJIT_MOVU_UB:
693 		SLJIT_ASSERT(!(flags & SET_FLAGS) && arg1 == TMP_REG1);
694 		if (IS_2_LO_REGS(dst, arg2))
695 			return push_inst16(compiler, UXTB | RD3(dst) | RN3(arg2));
696 		return push_inst32(compiler, UXTB_W | RD4(dst) | RM4(arg2));
697 	case SLJIT_MOV_SB:
698 	case SLJIT_MOVU_SB:
699 		SLJIT_ASSERT(!(flags & SET_FLAGS) && arg1 == TMP_REG1);
700 		if (IS_2_LO_REGS(dst, arg2))
701 			return push_inst16(compiler, SXTB | RD3(dst) | RN3(arg2));
702 		return push_inst32(compiler, SXTB_W | RD4(dst) | RM4(arg2));
703 	case SLJIT_MOV_UH:
704 	case SLJIT_MOVU_UH:
705 		SLJIT_ASSERT(!(flags & SET_FLAGS) && arg1 == TMP_REG1);
706 		if (IS_2_LO_REGS(dst, arg2))
707 			return push_inst16(compiler, UXTH | RD3(dst) | RN3(arg2));
708 		return push_inst32(compiler, UXTH_W | RD4(dst) | RM4(arg2));
709 	case SLJIT_MOV_SH:
710 	case SLJIT_MOVU_SH:
711 		SLJIT_ASSERT(!(flags & SET_FLAGS) && arg1 == TMP_REG1);
712 		if (IS_2_LO_REGS(dst, arg2))
713 			return push_inst16(compiler, SXTH | RD3(dst) | RN3(arg2));
714 		return push_inst32(compiler, SXTH_W | RD4(dst) | RM4(arg2));
715 	case SLJIT_NOT:
716 		SLJIT_ASSERT(arg1 == TMP_REG1);
717 		if (!(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, arg2))
718 			return push_inst16(compiler, MVNS | RD3(dst) | RN3(arg2));
719 		return push_inst32(compiler, MVN_W | (flags & SET_FLAGS) | RD4(dst) | RM4(arg2));
720 	case SLJIT_CLZ:
721 		SLJIT_ASSERT(arg1 == TMP_REG1);
722 		FAIL_IF(push_inst32(compiler, CLZ | RN4(arg2) | RD4(dst) | RM4(arg2)));
723 		if (flags & SET_FLAGS) {
724 			if (reg_map[dst] <= 7)
725 				return push_inst16(compiler, CMPI | RDN3(dst));
726 			return push_inst32(compiler, ADD_WI | SET_FLAGS | RN4(dst) | RD4(dst));
727 		}
728 		return SLJIT_SUCCESS;
729 	case SLJIT_ADD:
730 		if (!(flags & KEEP_FLAGS) && IS_3_LO_REGS(dst, arg1, arg2))
731 			return push_inst16(compiler, ADDS | RD3(dst) | RN3(arg1) | RM3(arg2));
732 		if (dst == arg1 && !(flags & SET_FLAGS))
733 			return push_inst16(compiler, ADD | SET_REGS44(dst, arg2));
734 		return push_inst32(compiler, ADD_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
735 	case SLJIT_ADDC:
736 		if (dst == arg1 && !(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, arg2))
737 			return push_inst16(compiler, ADCS | RD3(dst) | RN3(arg2));
738 		return push_inst32(compiler, ADC_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
739 	case SLJIT_SUB:
740 		if (!(flags & KEEP_FLAGS) && IS_3_LO_REGS(dst, arg1, arg2))
741 			return push_inst16(compiler, SUBS | RD3(dst) | RN3(arg1) | RM3(arg2));
742 		return push_inst32(compiler, SUB_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
743 	case SLJIT_SUBC:
744 		if (dst == arg1 && !(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, arg2))
745 			return push_inst16(compiler, SBCS | RD3(dst) | RN3(arg2));
746 		return push_inst32(compiler, SBC_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
747 	case SLJIT_MUL:
748 		if (!(flags & SET_FLAGS))
749 			return push_inst32(compiler, MUL | RD4(dst) | RN4(arg1) | RM4(arg2));
750 		SLJIT_ASSERT(reg_map[TMP_REG2] <= 7 && dst != TMP_REG2);
751 		FAIL_IF(push_inst32(compiler, SMULL | RT4(dst) | RD4(TMP_REG2) | RN4(arg1) | RM4(arg2)));
752 		/* cmp TMP_REG2, dst asr #31. */
753 		return push_inst32(compiler, CMP_W | RN4(TMP_REG2) | 0x70e0 | RM4(dst));
754 	case SLJIT_AND:
755 		if (!(flags & KEEP_FLAGS)) {
756 			if (dst == arg1 && IS_2_LO_REGS(dst, arg2))
757 				return push_inst16(compiler, ANDS | RD3(dst) | RN3(arg2));
758 			if ((flags & UNUSED_RETURN) && IS_2_LO_REGS(arg1, arg2))
759 				return push_inst16(compiler, TST | RD3(arg1) | RN3(arg2));
760 		}
761 		return push_inst32(compiler, AND_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
762 	case SLJIT_OR:
763 		if (dst == arg1 && !(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, arg2))
764 			return push_inst16(compiler, ORRS | RD3(dst) | RN3(arg2));
765 		return push_inst32(compiler, ORR_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
766 	case SLJIT_XOR:
767 		if (dst == arg1 && !(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, arg2))
768 			return push_inst16(compiler, EORS | RD3(dst) | RN3(arg2));
769 		return push_inst32(compiler, EOR_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
770 	case SLJIT_SHL:
771 		if (dst == arg1 && !(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, arg2))
772 			return push_inst16(compiler, LSLS | RD3(dst) | RN3(arg2));
773 		return push_inst32(compiler, LSL_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
774 	case SLJIT_LSHR:
775 		if (dst == arg1 && !(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, arg2))
776 			return push_inst16(compiler, LSRS | RD3(dst) | RN3(arg2));
777 		return push_inst32(compiler, LSR_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
778 	case SLJIT_ASHR:
779 		if (dst == arg1 && !(flags & KEEP_FLAGS) && IS_2_LO_REGS(dst, arg2))
780 			return push_inst16(compiler, ASRS | RD3(dst) | RN3(arg2));
781 		return push_inst32(compiler, ASR_W | (flags & SET_FLAGS) | RD4(dst) | RN4(arg1) | RM4(arg2));
782 	}
783 
784 	SLJIT_ASSERT_STOP();
785 	return SLJIT_SUCCESS;
786 }
787 
788 #define STORE		0x01
789 #define SIGNED		0x02
790 
791 #define WORD_SIZE	0x00
792 #define BYTE_SIZE	0x04
793 #define HALF_SIZE	0x08
794 
795 #define UPDATE		0x10
796 #define ARG_TEST	0x20
797 
798 #define IS_WORD_SIZE(flags)		(!(flags & (BYTE_SIZE | HALF_SIZE)))
799 #define OFFSET_CHECK(imm, shift)	(!(argw & ~(imm << shift)))
800 
801 /*
802   1st letter:
803   w = word
804   b = byte
805   h = half
806 
807   2nd letter:
808   s = signed
809   u = unsigned
810 
811   3rd letter:
812   l = load
813   s = store
814 */
815 
816 static SLJIT_CONST sljit_ins sljit_mem16[12] = {
817 /* w u l */ 0x5800 /* ldr */,
818 /* w u s */ 0x5000 /* str */,
819 /* w s l */ 0x5800 /* ldr */,
820 /* w s s */ 0x5000 /* str */,
821 
822 /* b u l */ 0x5c00 /* ldrb */,
823 /* b u s */ 0x5400 /* strb */,
824 /* b s l */ 0x5600 /* ldrsb */,
825 /* b s s */ 0x5400 /* strb */,
826 
827 /* h u l */ 0x5a00 /* ldrh */,
828 /* h u s */ 0x5200 /* strh */,
829 /* h s l */ 0x5e00 /* ldrsh */,
830 /* h s s */ 0x5200 /* strh */,
831 };
832 
833 static SLJIT_CONST sljit_ins sljit_mem16_imm5[12] = {
834 /* w u l */ 0x6800 /* ldr imm5 */,
835 /* w u s */ 0x6000 /* str imm5 */,
836 /* w s l */ 0x6800 /* ldr imm5 */,
837 /* w s s */ 0x6000 /* str imm5 */,
838 
839 /* b u l */ 0x7800 /* ldrb imm5 */,
840 /* b u s */ 0x7000 /* strb imm5 */,
841 /* b s l */ 0x0000 /* not allowed */,
842 /* b s s */ 0x7000 /* strb imm5 */,
843 
844 /* h u l */ 0x8800 /* ldrh imm5 */,
845 /* h u s */ 0x8000 /* strh imm5 */,
846 /* h s l */ 0x0000 /* not allowed */,
847 /* h s s */ 0x8000 /* strh imm5 */,
848 };
849 
850 #define MEM_IMM8	0xc00
851 #define MEM_IMM12	0x800000
852 static SLJIT_CONST sljit_ins sljit_mem32[12] = {
853 /* w u l */ 0xf8500000 /* ldr.w */,
854 /* w u s */ 0xf8400000 /* str.w */,
855 /* w s l */ 0xf8500000 /* ldr.w */,
856 /* w s s */ 0xf8400000 /* str.w */,
857 
858 /* b u l */ 0xf8100000 /* ldrb.w */,
859 /* b u s */ 0xf8000000 /* strb.w */,
860 /* b s l */ 0xf9100000 /* ldrsb.w */,
861 /* b s s */ 0xf8000000 /* strb.w */,
862 
863 /* h u l */ 0xf8300000 /* ldrh.w */,
864 /* h u s */ 0xf8200000 /* strsh.w */,
865 /* h s l */ 0xf9300000 /* ldrsh.w */,
866 /* h s s */ 0xf8200000 /* strsh.w */,
867 };
868 
869 /* Helper function. Dst should be reg + value, using at most 1 instruction, flags does not set. */
emit_set_delta(struct sljit_compiler * compiler,sljit_si dst,sljit_si reg,sljit_sw value)870 static sljit_si emit_set_delta(struct sljit_compiler *compiler, sljit_si dst, sljit_si reg, sljit_sw value)
871 {
872 	if (value >= 0) {
873 		if (value <= 0xfff)
874 			return push_inst32(compiler, ADDWI | RD4(dst) | RN4(reg) | IMM12(value));
875 		value = get_imm(value);
876 		if (value != INVALID_IMM)
877 			return push_inst32(compiler, ADD_WI | RD4(dst) | RN4(reg) | value);
878 	}
879 	else {
880 		value = -value;
881 		if (value <= 0xfff)
882 			return push_inst32(compiler, SUBWI | RD4(dst) | RN4(reg) | IMM12(value));
883 		value = get_imm(value);
884 		if (value != INVALID_IMM)
885 			return push_inst32(compiler, SUB_WI | RD4(dst) | RN4(reg) | value);
886 	}
887 	return SLJIT_ERR_UNSUPPORTED;
888 }
889 
890 /* Can perform an operation using at most 1 instruction. */
getput_arg_fast(struct sljit_compiler * compiler,sljit_si flags,sljit_si reg,sljit_si arg,sljit_sw argw)891 static sljit_si getput_arg_fast(struct sljit_compiler *compiler, sljit_si flags, sljit_si reg, sljit_si arg, sljit_sw argw)
892 {
893 	sljit_si other_r, shift;
894 
895 	SLJIT_ASSERT(arg & SLJIT_MEM);
896 
897 	if (SLJIT_UNLIKELY(flags & UPDATE)) {
898 		if ((arg & REG_MASK) && !(arg & OFFS_REG_MASK) && argw <= 0xff && argw >= -0xff) {
899 			if (SLJIT_UNLIKELY(flags & ARG_TEST))
900 				return 1;
901 
902 			flags &= ~UPDATE;
903 			arg &= 0xf;
904 			if (argw >= 0)
905 				argw |= 0x200;
906 			else {
907 				argw = -argw;
908 			}
909 
910 			SLJIT_ASSERT(argw >= 0 && (argw & 0xff) <= 0xff);
911 			FAIL_IF(push_inst32(compiler, sljit_mem32[flags] | MEM_IMM8 | RT4(reg) | RN4(arg) | 0x100 | argw));
912 			return -1;
913 		}
914 		return 0;
915 	}
916 
917 	if (SLJIT_UNLIKELY(arg & OFFS_REG_MASK)) {
918 		if (SLJIT_UNLIKELY(flags & ARG_TEST))
919 			return 1;
920 
921 		argw &= 0x3;
922 		other_r = OFFS_REG(arg);
923 		arg &= 0xf;
924 
925 		if (!argw && IS_3_LO_REGS(reg, arg, other_r))
926 			FAIL_IF(push_inst16(compiler, sljit_mem16[flags] | RD3(reg) | RN3(arg) | RM3(other_r)));
927 		else
928 			FAIL_IF(push_inst32(compiler, sljit_mem32[flags] | RT4(reg) | RN4(arg) | RM4(other_r) | (argw << 4)));
929 		return -1;
930 	}
931 
932 	if (!(arg & REG_MASK) || argw > 0xfff || argw < -0xff)
933 		return 0;
934 
935 	if (SLJIT_UNLIKELY(flags & ARG_TEST))
936 		return 1;
937 
938 	arg &= 0xf;
939 	if (IS_2_LO_REGS(reg, arg) && sljit_mem16_imm5[flags]) {
940 		shift = 3;
941 		if (IS_WORD_SIZE(flags)) {
942 			if (OFFSET_CHECK(0x1f, 2))
943 				shift = 2;
944 		}
945 		else if (flags & BYTE_SIZE)
946 		{
947 			if (OFFSET_CHECK(0x1f, 0))
948 				shift = 0;
949 		}
950 		else {
951 			SLJIT_ASSERT(flags & HALF_SIZE);
952 			if (OFFSET_CHECK(0x1f, 1))
953 				shift = 1;
954 		}
955 
956 		if (shift != 3) {
957 			FAIL_IF(push_inst16(compiler, sljit_mem16_imm5[flags] | RD3(reg) | RN3(arg) | (argw << (6 - shift))));
958 			return -1;
959 		}
960 	}
961 
962 	/* SP based immediate. */
963 	if (SLJIT_UNLIKELY(arg == SLJIT_SP) && OFFSET_CHECK(0xff, 2) && IS_WORD_SIZE(flags) && reg_map[reg] <= 7) {
964 		FAIL_IF(push_inst16(compiler, STR_SP | ((flags & STORE) ? 0 : 0x800) | RDN3(reg) | (argw >> 2)));
965 		return -1;
966 	}
967 
968 	if (argw >= 0)
969 		FAIL_IF(push_inst32(compiler, sljit_mem32[flags] | MEM_IMM12 | RT4(reg) | RN4(arg) | argw));
970 	else
971 		FAIL_IF(push_inst32(compiler, sljit_mem32[flags] | MEM_IMM8 | RT4(reg) | RN4(arg) | -argw));
972 	return -1;
973 }
974 
975 /* see getput_arg below.
976    Note: can_cache is called only for binary operators. Those
977    operators always uses word arguments without write back. */
can_cache(sljit_si arg,sljit_sw argw,sljit_si next_arg,sljit_sw next_argw)978 static sljit_si can_cache(sljit_si arg, sljit_sw argw, sljit_si next_arg, sljit_sw next_argw)
979 {
980 	sljit_sw diff;
981 	if ((arg & OFFS_REG_MASK) || !(next_arg & SLJIT_MEM))
982 		return 0;
983 
984 	if (!(arg & REG_MASK)) {
985 		diff = argw - next_argw;
986 		if (diff <= 0xfff && diff >= -0xfff)
987 			return 1;
988 		return 0;
989 	}
990 
991 	if (argw == next_argw)
992 		return 1;
993 
994 	diff = argw - next_argw;
995 	if (arg == next_arg && diff <= 0xfff && diff >= -0xfff)
996 		return 1;
997 
998 	return 0;
999 }
1000 
1001 /* Emit the necessary instructions. See can_cache above. */
getput_arg(struct sljit_compiler * compiler,sljit_si flags,sljit_si reg,sljit_si arg,sljit_sw argw,sljit_si next_arg,sljit_sw next_argw)1002 static sljit_si getput_arg(struct sljit_compiler *compiler, sljit_si flags, sljit_si reg,
1003 	sljit_si arg, sljit_sw argw, sljit_si next_arg, sljit_sw next_argw)
1004 {
1005 	sljit_si tmp_r, other_r;
1006 	sljit_sw diff;
1007 
1008 	SLJIT_ASSERT(arg & SLJIT_MEM);
1009 	if (!(next_arg & SLJIT_MEM)) {
1010 		next_arg = 0;
1011 		next_argw = 0;
1012 	}
1013 
1014 	tmp_r = (flags & STORE) ? TMP_REG3 : reg;
1015 
1016 	if (SLJIT_UNLIKELY((flags & UPDATE) && (arg & REG_MASK))) {
1017 		/* Update only applies if a base register exists. */
1018 		/* There is no caching here. */
1019 		other_r = OFFS_REG(arg);
1020 		arg &= 0xf;
1021 		flags &= ~UPDATE;
1022 
1023 		if (!other_r) {
1024 			if (!(argw & ~0xfff)) {
1025 				FAIL_IF(push_inst32(compiler, sljit_mem32[flags] | MEM_IMM12 | RT4(reg) | RN4(arg) | argw));
1026 				return push_inst32(compiler, ADDWI | RD4(arg) | RN4(arg) | IMM12(argw));
1027 			}
1028 
1029 			if (compiler->cache_arg == SLJIT_MEM) {
1030 				if (argw == compiler->cache_argw) {
1031 					other_r = TMP_REG3;
1032 					argw = 0;
1033 				}
1034 				else if (emit_set_delta(compiler, TMP_REG3, TMP_REG3, argw - compiler->cache_argw) != SLJIT_ERR_UNSUPPORTED) {
1035 					FAIL_IF(compiler->error);
1036 					compiler->cache_argw = argw;
1037 					other_r = TMP_REG3;
1038 					argw = 0;
1039 				}
1040 			}
1041 
1042 			if (argw) {
1043 				FAIL_IF(load_immediate(compiler, TMP_REG3, argw));
1044 				compiler->cache_arg = SLJIT_MEM;
1045 				compiler->cache_argw = argw;
1046 				other_r = TMP_REG3;
1047 				argw = 0;
1048 			}
1049 		}
1050 
1051 		argw &= 0x3;
1052 		if (!argw && IS_3_LO_REGS(reg, arg, other_r)) {
1053 			FAIL_IF(push_inst16(compiler, sljit_mem16[flags] | RD3(reg) | RN3(arg) | RM3(other_r)));
1054 			return push_inst16(compiler, ADD | SET_REGS44(arg, other_r));
1055 		}
1056 		FAIL_IF(push_inst32(compiler, sljit_mem32[flags] | RT4(reg) | RN4(arg) | RM4(other_r) | (argw << 4)));
1057 		return push_inst32(compiler, ADD_W | RD4(arg) | RN4(arg) | RM4(other_r) | (argw << 6));
1058 	}
1059 	flags &= ~UPDATE;
1060 
1061 	SLJIT_ASSERT(!(arg & OFFS_REG_MASK));
1062 
1063 	if (compiler->cache_arg == arg) {
1064 		diff = argw - compiler->cache_argw;
1065 		if (!(diff & ~0xfff))
1066 			return push_inst32(compiler, sljit_mem32[flags] | MEM_IMM12 | RT4(reg) | RN4(TMP_REG3) | diff);
1067 		if (!((compiler->cache_argw - argw) & ~0xff))
1068 			return push_inst32(compiler, sljit_mem32[flags] | MEM_IMM8 | RT4(reg) | RN4(TMP_REG3) | (compiler->cache_argw - argw));
1069 		if (emit_set_delta(compiler, TMP_REG3, TMP_REG3, diff) != SLJIT_ERR_UNSUPPORTED) {
1070 			FAIL_IF(compiler->error);
1071 			return push_inst32(compiler, sljit_mem32[flags] | MEM_IMM12 | RT4(reg) | RN4(TMP_REG3) | 0);
1072 		}
1073 	}
1074 
1075 	next_arg = (arg & REG_MASK) && (arg == next_arg) && (argw != next_argw);
1076 	arg &= 0xf;
1077 	if (arg && compiler->cache_arg == SLJIT_MEM) {
1078 		if (compiler->cache_argw == argw)
1079 			return push_inst32(compiler, sljit_mem32[flags] | RT4(reg) | RN4(arg) | RM4(TMP_REG3));
1080 		if (emit_set_delta(compiler, TMP_REG3, TMP_REG3, argw - compiler->cache_argw) != SLJIT_ERR_UNSUPPORTED) {
1081 			FAIL_IF(compiler->error);
1082 			compiler->cache_argw = argw;
1083 			return push_inst32(compiler, sljit_mem32[flags] | RT4(reg) | RN4(arg) | RM4(TMP_REG3));
1084 		}
1085 	}
1086 
1087 	compiler->cache_argw = argw;
1088 	if (next_arg && emit_set_delta(compiler, TMP_REG3, arg, argw) != SLJIT_ERR_UNSUPPORTED) {
1089 		FAIL_IF(compiler->error);
1090 		compiler->cache_arg = SLJIT_MEM | arg;
1091 		arg = 0;
1092 	}
1093 	else {
1094 		FAIL_IF(load_immediate(compiler, TMP_REG3, argw));
1095 		compiler->cache_arg = SLJIT_MEM;
1096 
1097 		diff = argw - next_argw;
1098 		if (next_arg && diff <= 0xfff && diff >= -0xfff) {
1099 			FAIL_IF(push_inst16(compiler, ADD | SET_REGS44(TMP_REG3, arg)));
1100 			compiler->cache_arg = SLJIT_MEM | arg;
1101 			arg = 0;
1102 		}
1103 	}
1104 
1105 	if (arg)
1106 		return push_inst32(compiler, sljit_mem32[flags] | RT4(reg) | RN4(arg) | RM4(TMP_REG3));
1107 	return push_inst32(compiler, sljit_mem32[flags] | MEM_IMM12 | RT4(reg) | RN4(TMP_REG3) | 0);
1108 }
1109 
emit_op_mem(struct sljit_compiler * compiler,sljit_si flags,sljit_si reg,sljit_si arg,sljit_sw argw)1110 static SLJIT_INLINE sljit_si emit_op_mem(struct sljit_compiler *compiler, sljit_si flags, sljit_si reg, sljit_si arg, sljit_sw argw)
1111 {
1112 	if (getput_arg_fast(compiler, flags, reg, arg, argw))
1113 		return compiler->error;
1114 	compiler->cache_arg = 0;
1115 	compiler->cache_argw = 0;
1116 	return getput_arg(compiler, flags, reg, arg, argw, 0, 0);
1117 }
1118 
emit_op_mem2(struct sljit_compiler * compiler,sljit_si flags,sljit_si reg,sljit_si arg1,sljit_sw arg1w,sljit_si arg2,sljit_sw arg2w)1119 static SLJIT_INLINE sljit_si emit_op_mem2(struct sljit_compiler *compiler, sljit_si flags, sljit_si reg, sljit_si arg1, sljit_sw arg1w, sljit_si arg2, sljit_sw arg2w)
1120 {
1121 	if (getput_arg_fast(compiler, flags, reg, arg1, arg1w))
1122 		return compiler->error;
1123 	return getput_arg(compiler, flags, reg, arg1, arg1w, arg2, arg2w);
1124 }
1125 
1126 /* --------------------------------------------------------------------- */
1127 /*  Entry, exit                                                          */
1128 /* --------------------------------------------------------------------- */
1129 
sljit_emit_enter(struct sljit_compiler * compiler,sljit_si options,sljit_si args,sljit_si scratches,sljit_si saveds,sljit_si fscratches,sljit_si fsaveds,sljit_si local_size)1130 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_enter(struct sljit_compiler *compiler,
1131 	sljit_si options, sljit_si args, sljit_si scratches, sljit_si saveds,
1132 	sljit_si fscratches, sljit_si fsaveds, sljit_si local_size)
1133 {
1134 	sljit_si size, i, tmp;
1135 	sljit_ins push;
1136 
1137 	CHECK_ERROR();
1138 	CHECK(check_sljit_emit_enter(compiler, options, args, scratches, saveds, fscratches, fsaveds, local_size));
1139 	set_emit_enter(compiler, options, args, scratches, saveds, fscratches, fsaveds, local_size);
1140 
1141 	push = (1 << 4);
1142 
1143 	tmp = saveds < SLJIT_NUMBER_OF_SAVED_REGISTERS ? (SLJIT_S0 + 1 - saveds) : SLJIT_FIRST_SAVED_REG;
1144 	for (i = SLJIT_S0; i >= tmp; i--)
1145 		push |= 1 << reg_map[i];
1146 
1147 	for (i = scratches; i >= SLJIT_FIRST_SAVED_REG; i--)
1148 		push |= 1 << reg_map[i];
1149 
1150 	FAIL_IF((push & 0xff00)
1151 		? push_inst32(compiler, PUSH_W | (1 << 14) | push)
1152 		: push_inst16(compiler, PUSH | (1 << 8) | push));
1153 
1154 	/* Stack must be aligned to 8 bytes: (LR, R4) */
1155 	size = GET_SAVED_REGISTERS_SIZE(scratches, saveds, 2);
1156 	local_size = ((size + local_size + 7) & ~7) - size;
1157 	compiler->local_size = local_size;
1158 	if (local_size > 0) {
1159 		if (local_size <= (127 << 2))
1160 			FAIL_IF(push_inst16(compiler, SUB_SP | (local_size >> 2)));
1161 		else
1162 			FAIL_IF(emit_op_imm(compiler, SLJIT_SUB | ARG2_IMM, SLJIT_SP, SLJIT_SP, local_size));
1163 	}
1164 
1165 	if (args >= 1)
1166 		FAIL_IF(push_inst16(compiler, MOV | SET_REGS44(SLJIT_S0, SLJIT_R0)));
1167 	if (args >= 2)
1168 		FAIL_IF(push_inst16(compiler, MOV | SET_REGS44(SLJIT_S1, SLJIT_R1)));
1169 	if (args >= 3)
1170 		FAIL_IF(push_inst16(compiler, MOV | SET_REGS44(SLJIT_S2, SLJIT_R2)));
1171 
1172 	return SLJIT_SUCCESS;
1173 }
1174 
sljit_set_context(struct sljit_compiler * compiler,sljit_si options,sljit_si args,sljit_si scratches,sljit_si saveds,sljit_si fscratches,sljit_si fsaveds,sljit_si local_size)1175 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_set_context(struct sljit_compiler *compiler,
1176 	sljit_si options, sljit_si args, sljit_si scratches, sljit_si saveds,
1177 	sljit_si fscratches, sljit_si fsaveds, sljit_si local_size)
1178 {
1179 	sljit_si size;
1180 
1181 	CHECK_ERROR();
1182 	CHECK(check_sljit_set_context(compiler, options, args, scratches, saveds, fscratches, fsaveds, local_size));
1183 	set_set_context(compiler, options, args, scratches, saveds, fscratches, fsaveds, local_size);
1184 
1185 	size = GET_SAVED_REGISTERS_SIZE(scratches, saveds, 2);
1186 	compiler->local_size = ((size + local_size + 7) & ~7) - size;
1187 	return SLJIT_SUCCESS;
1188 }
1189 
sljit_emit_return(struct sljit_compiler * compiler,sljit_si op,sljit_si src,sljit_sw srcw)1190 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_return(struct sljit_compiler *compiler, sljit_si op, sljit_si src, sljit_sw srcw)
1191 {
1192 	sljit_si i, tmp;
1193 	sljit_ins pop;
1194 
1195 	CHECK_ERROR();
1196 	CHECK(check_sljit_emit_return(compiler, op, src, srcw));
1197 
1198 	FAIL_IF(emit_mov_before_return(compiler, op, src, srcw));
1199 
1200 	if (compiler->local_size > 0) {
1201 		if (compiler->local_size <= (127 << 2))
1202 			FAIL_IF(push_inst16(compiler, ADD_SP | (compiler->local_size >> 2)));
1203 		else
1204 			FAIL_IF(emit_op_imm(compiler, SLJIT_ADD | ARG2_IMM, SLJIT_SP, SLJIT_SP, compiler->local_size));
1205 	}
1206 
1207 	pop = (1 << 4);
1208 
1209 	tmp = compiler->saveds < SLJIT_NUMBER_OF_SAVED_REGISTERS ? (SLJIT_S0 + 1 - compiler->saveds) : SLJIT_FIRST_SAVED_REG;
1210 	for (i = SLJIT_S0; i >= tmp; i--)
1211 		pop |= 1 << reg_map[i];
1212 
1213 	for (i = compiler->scratches; i >= SLJIT_FIRST_SAVED_REG; i--)
1214 		pop |= 1 << reg_map[i];
1215 
1216 	return (pop & 0xff00)
1217 		? push_inst32(compiler, POP_W | (1 << 15) | pop)
1218 		: push_inst16(compiler, POP | (1 << 8) | pop);
1219 }
1220 
1221 /* --------------------------------------------------------------------- */
1222 /*  Operators                                                            */
1223 /* --------------------------------------------------------------------- */
1224 
1225 #ifdef __cplusplus
1226 extern "C" {
1227 #endif
1228 
1229 #if defined(__GNUC__)
1230 extern unsigned int __aeabi_uidivmod(unsigned int numerator, int unsigned denominator);
1231 extern int __aeabi_idivmod(int numerator, int denominator);
1232 #else
1233 #error "Software divmod functions are needed"
1234 #endif
1235 
1236 #ifdef __cplusplus
1237 }
1238 #endif
1239 
sljit_emit_op0(struct sljit_compiler * compiler,sljit_si op)1240 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_op0(struct sljit_compiler *compiler, sljit_si op)
1241 {
1242 	sljit_sw saved_reg_list[3];
1243 	sljit_sw saved_reg_count;
1244 
1245 	CHECK_ERROR();
1246 	CHECK(check_sljit_emit_op0(compiler, op));
1247 
1248 	op = GET_OPCODE(op);
1249 	switch (op) {
1250 	case SLJIT_BREAKPOINT:
1251 		return push_inst16(compiler, BKPT);
1252 	case SLJIT_NOP:
1253 		return push_inst16(compiler, NOP);
1254 	case SLJIT_LUMUL:
1255 	case SLJIT_LSMUL:
1256 		return push_inst32(compiler, (op == SLJIT_LUMUL ? UMULL : SMULL)
1257 			| (reg_map[SLJIT_R1] << 8)
1258 			| (reg_map[SLJIT_R0] << 12)
1259 			| (reg_map[SLJIT_R0] << 16)
1260 			| reg_map[SLJIT_R1]);
1261 	case SLJIT_UDIVMOD:
1262 	case SLJIT_SDIVMOD:
1263 	case SLJIT_UDIVI:
1264 	case SLJIT_SDIVI:
1265 		SLJIT_COMPILE_ASSERT((SLJIT_UDIVMOD & 0x2) == 0 && SLJIT_UDIVI - 0x2 == SLJIT_UDIVMOD, bad_div_opcode_assignments);
1266 		SLJIT_COMPILE_ASSERT(reg_map[2] == 1 && reg_map[3] == 2 && reg_map[4] == 12, bad_register_mapping);
1267 
1268 		saved_reg_count = 0;
1269 		if (compiler->scratches >= 4)
1270 			saved_reg_list[saved_reg_count++] = 12;
1271 		if (compiler->scratches >= 3)
1272 			saved_reg_list[saved_reg_count++] = 2;
1273 		if (op >= SLJIT_UDIVI)
1274 			saved_reg_list[saved_reg_count++] = 1;
1275 
1276 		if (saved_reg_count > 0) {
1277 			FAIL_IF(push_inst32(compiler, 0xf84d0d00 | (saved_reg_count >= 3 ? 16 : 8)
1278 						| (saved_reg_list[0] << 12) /* str rX, [sp, #-8/-16]! */));
1279 			if (saved_reg_count >= 2) {
1280 				SLJIT_ASSERT(saved_reg_list[1] < 8);
1281 				FAIL_IF(push_inst16(compiler, 0x9001 | (saved_reg_list[1] << 8) /* str rX, [sp, #4] */));
1282 			}
1283 			if (saved_reg_count >= 3) {
1284 				SLJIT_ASSERT(saved_reg_list[2] < 8);
1285 				FAIL_IF(push_inst16(compiler, 0x9002 | (saved_reg_list[2] << 8) /* str rX, [sp, #8] */));
1286 			}
1287 		}
1288 
1289 #if defined(__GNUC__)
1290 		FAIL_IF(sljit_emit_ijump(compiler, SLJIT_FAST_CALL, SLJIT_IMM,
1291 			((op | 0x2) == SLJIT_UDIVI ? SLJIT_FUNC_OFFSET(__aeabi_uidivmod) : SLJIT_FUNC_OFFSET(__aeabi_idivmod))));
1292 #else
1293 #error "Software divmod functions are needed"
1294 #endif
1295 
1296 		if (saved_reg_count > 0) {
1297 			if (saved_reg_count >= 3) {
1298 				SLJIT_ASSERT(saved_reg_list[2] < 8);
1299 				FAIL_IF(push_inst16(compiler, 0x9802 | (saved_reg_list[2] << 8) /* ldr rX, [sp, #8] */));
1300 			}
1301 			if (saved_reg_count >= 2) {
1302 				SLJIT_ASSERT(saved_reg_list[1] < 8);
1303 				FAIL_IF(push_inst16(compiler, 0x9801 | (saved_reg_list[1] << 8) /* ldr rX, [sp, #4] */));
1304 			}
1305 			return push_inst32(compiler, 0xf85d0b00 | (saved_reg_count >= 3 ? 16 : 8)
1306 						| (saved_reg_list[0] << 12) /* ldr rX, [sp], #8/16 */);
1307 		}
1308 		return SLJIT_SUCCESS;
1309 	}
1310 
1311 	return SLJIT_SUCCESS;
1312 }
1313 
sljit_emit_op1(struct sljit_compiler * compiler,sljit_si op,sljit_si dst,sljit_sw dstw,sljit_si src,sljit_sw srcw)1314 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_op1(struct sljit_compiler *compiler, sljit_si op,
1315 	sljit_si dst, sljit_sw dstw,
1316 	sljit_si src, sljit_sw srcw)
1317 {
1318 	sljit_si dst_r, flags;
1319 	sljit_si op_flags = GET_ALL_FLAGS(op);
1320 
1321 	CHECK_ERROR();
1322 	CHECK(check_sljit_emit_op1(compiler, op, dst, dstw, src, srcw));
1323 	ADJUST_LOCAL_OFFSET(dst, dstw);
1324 	ADJUST_LOCAL_OFFSET(src, srcw);
1325 
1326 	compiler->cache_arg = 0;
1327 	compiler->cache_argw = 0;
1328 
1329 	dst_r = SLOW_IS_REG(dst) ? dst : TMP_REG1;
1330 
1331 	op = GET_OPCODE(op);
1332 	if (op >= SLJIT_MOV && op <= SLJIT_MOVU_P) {
1333 		switch (op) {
1334 		case SLJIT_MOV:
1335 		case SLJIT_MOV_UI:
1336 		case SLJIT_MOV_SI:
1337 		case SLJIT_MOV_P:
1338 			flags = WORD_SIZE;
1339 			break;
1340 		case SLJIT_MOV_UB:
1341 			flags = BYTE_SIZE;
1342 			if (src & SLJIT_IMM)
1343 				srcw = (sljit_ub)srcw;
1344 			break;
1345 		case SLJIT_MOV_SB:
1346 			flags = BYTE_SIZE | SIGNED;
1347 			if (src & SLJIT_IMM)
1348 				srcw = (sljit_sb)srcw;
1349 			break;
1350 		case SLJIT_MOV_UH:
1351 			flags = HALF_SIZE;
1352 			if (src & SLJIT_IMM)
1353 				srcw = (sljit_uh)srcw;
1354 			break;
1355 		case SLJIT_MOV_SH:
1356 			flags = HALF_SIZE | SIGNED;
1357 			if (src & SLJIT_IMM)
1358 				srcw = (sljit_sh)srcw;
1359 			break;
1360 		case SLJIT_MOVU:
1361 		case SLJIT_MOVU_UI:
1362 		case SLJIT_MOVU_SI:
1363 		case SLJIT_MOVU_P:
1364 			flags = WORD_SIZE | UPDATE;
1365 			break;
1366 		case SLJIT_MOVU_UB:
1367 			flags = BYTE_SIZE | UPDATE;
1368 			if (src & SLJIT_IMM)
1369 				srcw = (sljit_ub)srcw;
1370 			break;
1371 		case SLJIT_MOVU_SB:
1372 			flags = BYTE_SIZE | SIGNED | UPDATE;
1373 			if (src & SLJIT_IMM)
1374 				srcw = (sljit_sb)srcw;
1375 			break;
1376 		case SLJIT_MOVU_UH:
1377 			flags = HALF_SIZE | UPDATE;
1378 			if (src & SLJIT_IMM)
1379 				srcw = (sljit_uh)srcw;
1380 			break;
1381 		case SLJIT_MOVU_SH:
1382 			flags = HALF_SIZE | SIGNED | UPDATE;
1383 			if (src & SLJIT_IMM)
1384 				srcw = (sljit_sh)srcw;
1385 			break;
1386 		default:
1387 			SLJIT_ASSERT_STOP();
1388 			flags = 0;
1389 			break;
1390 		}
1391 
1392 		if (src & SLJIT_IMM)
1393 			FAIL_IF(emit_op_imm(compiler, SLJIT_MOV | ARG2_IMM, dst_r, TMP_REG1, srcw));
1394 		else if (src & SLJIT_MEM) {
1395 			if (getput_arg_fast(compiler, flags, dst_r, src, srcw))
1396 				FAIL_IF(compiler->error);
1397 			else
1398 				FAIL_IF(getput_arg(compiler, flags, dst_r, src, srcw, dst, dstw));
1399 		} else {
1400 			if (dst_r != TMP_REG1)
1401 				return emit_op_imm(compiler, op, dst_r, TMP_REG1, src);
1402 			dst_r = src;
1403 		}
1404 
1405 		if (dst & SLJIT_MEM) {
1406 			if (getput_arg_fast(compiler, flags | STORE, dst_r, dst, dstw))
1407 				return compiler->error;
1408 			else
1409 				return getput_arg(compiler, flags | STORE, dst_r, dst, dstw, 0, 0);
1410 		}
1411 		return SLJIT_SUCCESS;
1412 	}
1413 
1414 	if (op == SLJIT_NEG) {
1415 #if (defined SLJIT_VERBOSE && SLJIT_VERBOSE) \
1416 			|| (defined SLJIT_ARGUMENT_CHECKS && SLJIT_ARGUMENT_CHECKS)
1417 		compiler->skip_checks = 1;
1418 #endif
1419 		return sljit_emit_op2(compiler, SLJIT_SUB | op_flags, dst, dstw, SLJIT_IMM, 0, src, srcw);
1420 	}
1421 
1422 	flags = (GET_FLAGS(op_flags) ? SET_FLAGS : 0) | ((op_flags & SLJIT_KEEP_FLAGS) ? KEEP_FLAGS : 0);
1423 	if (src & SLJIT_MEM) {
1424 		if (getput_arg_fast(compiler, WORD_SIZE, TMP_REG2, src, srcw))
1425 			FAIL_IF(compiler->error);
1426 		else
1427 			FAIL_IF(getput_arg(compiler, WORD_SIZE, TMP_REG2, src, srcw, dst, dstw));
1428 		src = TMP_REG2;
1429 	}
1430 
1431 	if (src & SLJIT_IMM)
1432 		flags |= ARG2_IMM;
1433 	else
1434 		srcw = src;
1435 
1436 	emit_op_imm(compiler, flags | op, dst_r, TMP_REG1, srcw);
1437 
1438 	if (dst & SLJIT_MEM) {
1439 		if (getput_arg_fast(compiler, flags | STORE, dst_r, dst, dstw))
1440 			return compiler->error;
1441 		else
1442 			return getput_arg(compiler, flags | STORE, dst_r, dst, dstw, 0, 0);
1443 	}
1444 	return SLJIT_SUCCESS;
1445 }
1446 
sljit_emit_op2(struct sljit_compiler * compiler,sljit_si op,sljit_si dst,sljit_sw dstw,sljit_si src1,sljit_sw src1w,sljit_si src2,sljit_sw src2w)1447 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_op2(struct sljit_compiler *compiler, sljit_si op,
1448 	sljit_si dst, sljit_sw dstw,
1449 	sljit_si src1, sljit_sw src1w,
1450 	sljit_si src2, sljit_sw src2w)
1451 {
1452 	sljit_si dst_r, flags;
1453 
1454 	CHECK_ERROR();
1455 	CHECK(check_sljit_emit_op2(compiler, op, dst, dstw, src1, src1w, src2, src2w));
1456 	ADJUST_LOCAL_OFFSET(dst, dstw);
1457 	ADJUST_LOCAL_OFFSET(src1, src1w);
1458 	ADJUST_LOCAL_OFFSET(src2, src2w);
1459 
1460 	compiler->cache_arg = 0;
1461 	compiler->cache_argw = 0;
1462 
1463 	dst_r = SLOW_IS_REG(dst) ? dst : TMP_REG1;
1464 	flags = (GET_FLAGS(op) ? SET_FLAGS : 0) | ((op & SLJIT_KEEP_FLAGS) ? KEEP_FLAGS : 0);
1465 
1466 	if ((dst & SLJIT_MEM) && !getput_arg_fast(compiler, WORD_SIZE | STORE | ARG_TEST, TMP_REG1, dst, dstw))
1467 		flags |= SLOW_DEST;
1468 
1469 	if (src1 & SLJIT_MEM) {
1470 		if (getput_arg_fast(compiler, WORD_SIZE, TMP_REG1, src1, src1w))
1471 			FAIL_IF(compiler->error);
1472 		else
1473 			flags |= SLOW_SRC1;
1474 	}
1475 	if (src2 & SLJIT_MEM) {
1476 		if (getput_arg_fast(compiler, WORD_SIZE, TMP_REG2, src2, src2w))
1477 			FAIL_IF(compiler->error);
1478 		else
1479 			flags |= SLOW_SRC2;
1480 	}
1481 
1482 	if ((flags & (SLOW_SRC1 | SLOW_SRC2)) == (SLOW_SRC1 | SLOW_SRC2)) {
1483 		if (!can_cache(src1, src1w, src2, src2w) && can_cache(src1, src1w, dst, dstw)) {
1484 			FAIL_IF(getput_arg(compiler, WORD_SIZE, TMP_REG2, src2, src2w, src1, src1w));
1485 			FAIL_IF(getput_arg(compiler, WORD_SIZE, TMP_REG1, src1, src1w, dst, dstw));
1486 		}
1487 		else {
1488 			FAIL_IF(getput_arg(compiler, WORD_SIZE, TMP_REG1, src1, src1w, src2, src2w));
1489 			FAIL_IF(getput_arg(compiler, WORD_SIZE, TMP_REG2, src2, src2w, dst, dstw));
1490 		}
1491 	}
1492 	else if (flags & SLOW_SRC1)
1493 		FAIL_IF(getput_arg(compiler, WORD_SIZE, TMP_REG1, src1, src1w, dst, dstw));
1494 	else if (flags & SLOW_SRC2)
1495 		FAIL_IF(getput_arg(compiler, WORD_SIZE, TMP_REG2, src2, src2w, dst, dstw));
1496 
1497 	if (src1 & SLJIT_MEM)
1498 		src1 = TMP_REG1;
1499 	if (src2 & SLJIT_MEM)
1500 		src2 = TMP_REG2;
1501 
1502 	if (src1 & SLJIT_IMM)
1503 		flags |= ARG1_IMM;
1504 	else
1505 		src1w = src1;
1506 	if (src2 & SLJIT_IMM)
1507 		flags |= ARG2_IMM;
1508 	else
1509 		src2w = src2;
1510 
1511 	if (dst == SLJIT_UNUSED)
1512 		flags |= UNUSED_RETURN;
1513 
1514 	emit_op_imm(compiler, flags | GET_OPCODE(op), dst_r, src1w, src2w);
1515 
1516 	if (dst & SLJIT_MEM) {
1517 		if (!(flags & SLOW_DEST)) {
1518 			getput_arg_fast(compiler, WORD_SIZE | STORE, dst_r, dst, dstw);
1519 			return compiler->error;
1520 		}
1521 		return getput_arg(compiler, WORD_SIZE | STORE, TMP_REG1, dst, dstw, 0, 0);
1522 	}
1523 	return SLJIT_SUCCESS;
1524 }
1525 
sljit_get_register_index(sljit_si reg)1526 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_get_register_index(sljit_si reg)
1527 {
1528 	CHECK_REG_INDEX(check_sljit_get_register_index(reg));
1529 	return reg_map[reg];
1530 }
1531 
sljit_get_float_register_index(sljit_si reg)1532 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_get_float_register_index(sljit_si reg)
1533 {
1534 	CHECK_REG_INDEX(check_sljit_get_float_register_index(reg));
1535 	return reg << 1;
1536 }
1537 
sljit_emit_op_custom(struct sljit_compiler * compiler,void * instruction,sljit_si size)1538 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_op_custom(struct sljit_compiler *compiler,
1539 	void *instruction, sljit_si size)
1540 {
1541 	CHECK_ERROR();
1542 	CHECK(check_sljit_emit_op_custom(compiler, instruction, size));
1543 
1544 	if (size == 2)
1545 		return push_inst16(compiler, *(sljit_uh*)instruction);
1546 	return push_inst32(compiler, *(sljit_ins*)instruction);
1547 }
1548 
1549 /* --------------------------------------------------------------------- */
1550 /*  Floating point operators                                             */
1551 /* --------------------------------------------------------------------- */
1552 
sljit_is_fpu_available(void)1553 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_is_fpu_available(void)
1554 {
1555 #ifdef SLJIT_IS_FPU_AVAILABLE
1556 	return SLJIT_IS_FPU_AVAILABLE;
1557 #else
1558 	/* Available by default. */
1559 	return 1;
1560 #endif
1561 }
1562 
1563 #define FPU_LOAD (1 << 20)
1564 
emit_fop_mem(struct sljit_compiler * compiler,sljit_si flags,sljit_si reg,sljit_si arg,sljit_sw argw)1565 static sljit_si emit_fop_mem(struct sljit_compiler *compiler, sljit_si flags, sljit_si reg, sljit_si arg, sljit_sw argw)
1566 {
1567 	sljit_sw tmp;
1568 	sljit_uw imm;
1569 	sljit_sw inst = VSTR_F32 | (flags & (SLJIT_SINGLE_OP | FPU_LOAD));
1570 
1571 	SLJIT_ASSERT(arg & SLJIT_MEM);
1572 
1573 	/* Fast loads and stores. */
1574 	if (SLJIT_UNLIKELY(arg & OFFS_REG_MASK)) {
1575 		FAIL_IF(push_inst32(compiler, ADD_W | RD4(TMP_REG2) | RN4(arg & REG_MASK) | RM4(OFFS_REG(arg)) | ((argw & 0x3) << 6)));
1576 		arg = SLJIT_MEM | TMP_REG2;
1577 		argw = 0;
1578 	}
1579 
1580 	if ((arg & REG_MASK) && (argw & 0x3) == 0) {
1581 		if (!(argw & ~0x3fc))
1582 			return push_inst32(compiler, inst | 0x800000 | RN4(arg & REG_MASK) | DD4(reg) | (argw >> 2));
1583 		if (!(-argw & ~0x3fc))
1584 			return push_inst32(compiler, inst | RN4(arg & REG_MASK) | DD4(reg) | (-argw >> 2));
1585 	}
1586 
1587 	/* Slow cases */
1588 	SLJIT_ASSERT(!(arg & OFFS_REG_MASK));
1589 	if (compiler->cache_arg == arg) {
1590 		tmp = argw - compiler->cache_argw;
1591 		if (!(tmp & ~0x3fc))
1592 			return push_inst32(compiler, inst | 0x800000 | RN4(TMP_REG3) | DD4(reg) | (tmp >> 2));
1593 		if (!(-tmp & ~0x3fc))
1594 			return push_inst32(compiler, inst | RN4(TMP_REG3) | DD4(reg) | (-tmp >> 2));
1595 		if (emit_set_delta(compiler, TMP_REG3, TMP_REG3, tmp) != SLJIT_ERR_UNSUPPORTED) {
1596 			FAIL_IF(compiler->error);
1597 			compiler->cache_argw = argw;
1598 			return push_inst32(compiler, inst | 0x800000 | RN4(TMP_REG3) | DD4(reg));
1599 		}
1600 	}
1601 
1602 	if (arg & REG_MASK) {
1603 		if (emit_set_delta(compiler, TMP_REG1, arg & REG_MASK, argw) != SLJIT_ERR_UNSUPPORTED) {
1604 			FAIL_IF(compiler->error);
1605 			return push_inst32(compiler, inst | 0x800000 | RN4(TMP_REG1) | DD4(reg));
1606 		}
1607 		imm = get_imm(argw & ~0x3fc);
1608 		if (imm != INVALID_IMM) {
1609 			FAIL_IF(push_inst32(compiler, ADD_WI | RD4(TMP_REG1) | RN4(arg & REG_MASK) | imm));
1610 			return push_inst32(compiler, inst | 0x800000 | RN4(TMP_REG1) | DD4(reg) | ((argw & 0x3fc) >> 2));
1611 		}
1612 		imm = get_imm(-argw & ~0x3fc);
1613 		if (imm != INVALID_IMM) {
1614 			argw = -argw;
1615 			FAIL_IF(push_inst32(compiler, SUB_WI | RD4(TMP_REG1) | RN4(arg & REG_MASK) | imm));
1616 			return push_inst32(compiler, inst | RN4(TMP_REG1) | DD4(reg) | ((argw & 0x3fc) >> 2));
1617 		}
1618 	}
1619 
1620 	compiler->cache_arg = arg;
1621 	compiler->cache_argw = argw;
1622 
1623 	FAIL_IF(load_immediate(compiler, TMP_REG3, argw));
1624 	if (arg & REG_MASK)
1625 		FAIL_IF(push_inst16(compiler, ADD | SET_REGS44(TMP_REG3, (arg & REG_MASK))));
1626 	return push_inst32(compiler, inst | 0x800000 | RN4(TMP_REG3) | DD4(reg));
1627 }
1628 
sljit_emit_fop1_convw_fromd(struct sljit_compiler * compiler,sljit_si op,sljit_si dst,sljit_sw dstw,sljit_si src,sljit_sw srcw)1629 static SLJIT_INLINE sljit_si sljit_emit_fop1_convw_fromd(struct sljit_compiler *compiler, sljit_si op,
1630 	sljit_si dst, sljit_sw dstw,
1631 	sljit_si src, sljit_sw srcw)
1632 {
1633 	if (src & SLJIT_MEM) {
1634 		FAIL_IF(emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP) | FPU_LOAD, TMP_FREG1, src, srcw));
1635 		src = TMP_FREG1;
1636 	}
1637 
1638 	FAIL_IF(push_inst32(compiler, VCVT_S32_F32 | (op & SLJIT_SINGLE_OP) | DD4(TMP_FREG1) | DM4(src)));
1639 
1640 	if (dst == SLJIT_UNUSED)
1641 		return SLJIT_SUCCESS;
1642 
1643 	if (FAST_IS_REG(dst))
1644 		return push_inst32(compiler, VMOV | (1 << 20) | RT4(dst) | DN4(TMP_FREG1));
1645 
1646 	/* Store the integer value from a VFP register. */
1647 	return emit_fop_mem(compiler, 0, TMP_FREG1, dst, dstw);
1648 }
1649 
sljit_emit_fop1_convd_fromw(struct sljit_compiler * compiler,sljit_si op,sljit_si dst,sljit_sw dstw,sljit_si src,sljit_sw srcw)1650 static SLJIT_INLINE sljit_si sljit_emit_fop1_convd_fromw(struct sljit_compiler *compiler, sljit_si op,
1651 	sljit_si dst, sljit_sw dstw,
1652 	sljit_si src, sljit_sw srcw)
1653 {
1654 	sljit_si dst_r = FAST_IS_REG(dst) ? dst : TMP_FREG1;
1655 
1656 	if (FAST_IS_REG(src))
1657 		FAIL_IF(push_inst32(compiler, VMOV | RT4(src) | DN4(TMP_FREG1)));
1658 	else if (src & SLJIT_MEM) {
1659 		/* Load the integer value into a VFP register. */
1660 		FAIL_IF(emit_fop_mem(compiler, FPU_LOAD, TMP_FREG1, src, srcw));
1661 	}
1662 	else {
1663 		FAIL_IF(load_immediate(compiler, TMP_REG1, srcw));
1664 		FAIL_IF(push_inst32(compiler, VMOV | RT4(TMP_REG1) | DN4(TMP_FREG1)));
1665 	}
1666 
1667 	FAIL_IF(push_inst32(compiler, VCVT_F32_S32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DM4(TMP_FREG1)));
1668 
1669 	if (dst & SLJIT_MEM)
1670 		return emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP), TMP_FREG1, dst, dstw);
1671 	return SLJIT_SUCCESS;
1672 }
1673 
sljit_emit_fop1_cmp(struct sljit_compiler * compiler,sljit_si op,sljit_si src1,sljit_sw src1w,sljit_si src2,sljit_sw src2w)1674 static SLJIT_INLINE sljit_si sljit_emit_fop1_cmp(struct sljit_compiler *compiler, sljit_si op,
1675 	sljit_si src1, sljit_sw src1w,
1676 	sljit_si src2, sljit_sw src2w)
1677 {
1678 	if (src1 & SLJIT_MEM) {
1679 		emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP) | FPU_LOAD, TMP_FREG1, src1, src1w);
1680 		src1 = TMP_FREG1;
1681 	}
1682 
1683 	if (src2 & SLJIT_MEM) {
1684 		emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP) | FPU_LOAD, TMP_FREG2, src2, src2w);
1685 		src2 = TMP_FREG2;
1686 	}
1687 
1688 	FAIL_IF(push_inst32(compiler, VCMP_F32 | (op & SLJIT_SINGLE_OP) | DD4(src1) | DM4(src2)));
1689 	return push_inst32(compiler, VMRS);
1690 }
1691 
sljit_emit_fop1(struct sljit_compiler * compiler,sljit_si op,sljit_si dst,sljit_sw dstw,sljit_si src,sljit_sw srcw)1692 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_fop1(struct sljit_compiler *compiler, sljit_si op,
1693 	sljit_si dst, sljit_sw dstw,
1694 	sljit_si src, sljit_sw srcw)
1695 {
1696 	sljit_si dst_r;
1697 
1698 	CHECK_ERROR();
1699 	compiler->cache_arg = 0;
1700 	compiler->cache_argw = 0;
1701 	if (GET_OPCODE(op) != SLJIT_CONVD_FROMS)
1702 		op ^= SLJIT_SINGLE_OP;
1703 
1704 	SLJIT_COMPILE_ASSERT((SLJIT_SINGLE_OP == 0x100), float_transfer_bit_error);
1705 	SELECT_FOP1_OPERATION_WITH_CHECKS(compiler, op, dst, dstw, src, srcw);
1706 
1707 	dst_r = FAST_IS_REG(dst) ? dst : TMP_FREG1;
1708 
1709 	if (src & SLJIT_MEM) {
1710 		emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP) | FPU_LOAD, dst_r, src, srcw);
1711 		src = dst_r;
1712 	}
1713 
1714 	switch (GET_OPCODE(op)) {
1715 	case SLJIT_DMOV:
1716 		if (src != dst_r) {
1717 			if (dst_r != TMP_FREG1)
1718 				FAIL_IF(push_inst32(compiler, VMOV_F32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DM4(src)));
1719 			else
1720 				dst_r = src;
1721 		}
1722 		break;
1723 	case SLJIT_DNEG:
1724 		FAIL_IF(push_inst32(compiler, VNEG_F32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DM4(src)));
1725 		break;
1726 	case SLJIT_DABS:
1727 		FAIL_IF(push_inst32(compiler, VABS_F32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DM4(src)));
1728 		break;
1729 	case SLJIT_CONVD_FROMS:
1730 		FAIL_IF(push_inst32(compiler, VCVT_F64_F32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DM4(src)));
1731 		op ^= SLJIT_SINGLE_OP;
1732 		break;
1733 	}
1734 
1735 	if (dst & SLJIT_MEM)
1736 		return emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP), dst_r, dst, dstw);
1737 	return SLJIT_SUCCESS;
1738 }
1739 
sljit_emit_fop2(struct sljit_compiler * compiler,sljit_si op,sljit_si dst,sljit_sw dstw,sljit_si src1,sljit_sw src1w,sljit_si src2,sljit_sw src2w)1740 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_fop2(struct sljit_compiler *compiler, sljit_si op,
1741 	sljit_si dst, sljit_sw dstw,
1742 	sljit_si src1, sljit_sw src1w,
1743 	sljit_si src2, sljit_sw src2w)
1744 {
1745 	sljit_si dst_r;
1746 
1747 	CHECK_ERROR();
1748 	CHECK(check_sljit_emit_fop2(compiler, op, dst, dstw, src1, src1w, src2, src2w));
1749 	ADJUST_LOCAL_OFFSET(dst, dstw);
1750 	ADJUST_LOCAL_OFFSET(src1, src1w);
1751 	ADJUST_LOCAL_OFFSET(src2, src2w);
1752 
1753 	compiler->cache_arg = 0;
1754 	compiler->cache_argw = 0;
1755 	op ^= SLJIT_SINGLE_OP;
1756 
1757 	dst_r = FAST_IS_REG(dst) ? dst : TMP_FREG1;
1758 	if (src1 & SLJIT_MEM) {
1759 		emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP) | FPU_LOAD, TMP_FREG1, src1, src1w);
1760 		src1 = TMP_FREG1;
1761 	}
1762 	if (src2 & SLJIT_MEM) {
1763 		emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP) | FPU_LOAD, TMP_FREG2, src2, src2w);
1764 		src2 = TMP_FREG2;
1765 	}
1766 
1767 	switch (GET_OPCODE(op)) {
1768 	case SLJIT_DADD:
1769 		FAIL_IF(push_inst32(compiler, VADD_F32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DN4(src1) | DM4(src2)));
1770 		break;
1771 	case SLJIT_DSUB:
1772 		FAIL_IF(push_inst32(compiler, VSUB_F32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DN4(src1) | DM4(src2)));
1773 		break;
1774 	case SLJIT_DMUL:
1775 		FAIL_IF(push_inst32(compiler, VMUL_F32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DN4(src1) | DM4(src2)));
1776 		break;
1777 	case SLJIT_DDIV:
1778 		FAIL_IF(push_inst32(compiler, VDIV_F32 | (op & SLJIT_SINGLE_OP) | DD4(dst_r) | DN4(src1) | DM4(src2)));
1779 		break;
1780 	}
1781 
1782 	if (!(dst & SLJIT_MEM))
1783 		return SLJIT_SUCCESS;
1784 	return emit_fop_mem(compiler, (op & SLJIT_SINGLE_OP), TMP_FREG1, dst, dstw);
1785 }
1786 
1787 #undef FPU_LOAD
1788 
1789 /* --------------------------------------------------------------------- */
1790 /*  Other instructions                                                   */
1791 /* --------------------------------------------------------------------- */
1792 
sljit_emit_fast_enter(struct sljit_compiler * compiler,sljit_si dst,sljit_sw dstw)1793 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_fast_enter(struct sljit_compiler *compiler, sljit_si dst, sljit_sw dstw)
1794 {
1795 	CHECK_ERROR();
1796 	CHECK(check_sljit_emit_fast_enter(compiler, dst, dstw));
1797 	ADJUST_LOCAL_OFFSET(dst, dstw);
1798 
1799 	/* For UNUSED dst. Uncommon, but possible. */
1800 	if (dst == SLJIT_UNUSED)
1801 		return SLJIT_SUCCESS;
1802 
1803 	if (FAST_IS_REG(dst))
1804 		return push_inst16(compiler, MOV | SET_REGS44(dst, TMP_REG3));
1805 
1806 	/* Memory. */
1807 	if (getput_arg_fast(compiler, WORD_SIZE | STORE, TMP_REG3, dst, dstw))
1808 		return compiler->error;
1809 	/* TMP_REG3 is used for caching. */
1810 	FAIL_IF(push_inst16(compiler, MOV | SET_REGS44(TMP_REG2, TMP_REG3)));
1811 	compiler->cache_arg = 0;
1812 	compiler->cache_argw = 0;
1813 	return getput_arg(compiler, WORD_SIZE | STORE, TMP_REG2, dst, dstw, 0, 0);
1814 }
1815 
sljit_emit_fast_return(struct sljit_compiler * compiler,sljit_si src,sljit_sw srcw)1816 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_fast_return(struct sljit_compiler *compiler, sljit_si src, sljit_sw srcw)
1817 {
1818 	CHECK_ERROR();
1819 	CHECK(check_sljit_emit_fast_return(compiler, src, srcw));
1820 	ADJUST_LOCAL_OFFSET(src, srcw);
1821 
1822 	if (FAST_IS_REG(src))
1823 		FAIL_IF(push_inst16(compiler, MOV | SET_REGS44(TMP_REG3, src)));
1824 	else if (src & SLJIT_MEM) {
1825 		if (getput_arg_fast(compiler, WORD_SIZE, TMP_REG3, src, srcw))
1826 			FAIL_IF(compiler->error);
1827 		else {
1828 			compiler->cache_arg = 0;
1829 			compiler->cache_argw = 0;
1830 			FAIL_IF(getput_arg(compiler, WORD_SIZE, TMP_REG2, src, srcw, 0, 0));
1831 			FAIL_IF(push_inst16(compiler, MOV | SET_REGS44(TMP_REG3, TMP_REG2)));
1832 		}
1833 	}
1834 	else if (src & SLJIT_IMM)
1835 		FAIL_IF(load_immediate(compiler, TMP_REG3, srcw));
1836 	return push_inst16(compiler, BLX | RN3(TMP_REG3));
1837 }
1838 
1839 /* --------------------------------------------------------------------- */
1840 /*  Conditional instructions                                             */
1841 /* --------------------------------------------------------------------- */
1842 
get_cc(sljit_si type)1843 static sljit_uw get_cc(sljit_si type)
1844 {
1845 	switch (type) {
1846 	case SLJIT_EQUAL:
1847 	case SLJIT_MUL_NOT_OVERFLOW:
1848 	case SLJIT_D_EQUAL:
1849 		return 0x0;
1850 
1851 	case SLJIT_NOT_EQUAL:
1852 	case SLJIT_MUL_OVERFLOW:
1853 	case SLJIT_D_NOT_EQUAL:
1854 		return 0x1;
1855 
1856 	case SLJIT_LESS:
1857 	case SLJIT_D_LESS:
1858 		return 0x3;
1859 
1860 	case SLJIT_GREATER_EQUAL:
1861 	case SLJIT_D_GREATER_EQUAL:
1862 		return 0x2;
1863 
1864 	case SLJIT_GREATER:
1865 	case SLJIT_D_GREATER:
1866 		return 0x8;
1867 
1868 	case SLJIT_LESS_EQUAL:
1869 	case SLJIT_D_LESS_EQUAL:
1870 		return 0x9;
1871 
1872 	case SLJIT_SIG_LESS:
1873 		return 0xb;
1874 
1875 	case SLJIT_SIG_GREATER_EQUAL:
1876 		return 0xa;
1877 
1878 	case SLJIT_SIG_GREATER:
1879 		return 0xc;
1880 
1881 	case SLJIT_SIG_LESS_EQUAL:
1882 		return 0xd;
1883 
1884 	case SLJIT_OVERFLOW:
1885 	case SLJIT_D_UNORDERED:
1886 		return 0x6;
1887 
1888 	case SLJIT_NOT_OVERFLOW:
1889 	case SLJIT_D_ORDERED:
1890 		return 0x7;
1891 
1892 	default: /* SLJIT_JUMP */
1893 		SLJIT_ASSERT_STOP();
1894 		return 0xe;
1895 	}
1896 }
1897 
sljit_emit_label(struct sljit_compiler * compiler)1898 SLJIT_API_FUNC_ATTRIBUTE struct sljit_label* sljit_emit_label(struct sljit_compiler *compiler)
1899 {
1900 	struct sljit_label *label;
1901 
1902 	CHECK_ERROR_PTR();
1903 	CHECK_PTR(check_sljit_emit_label(compiler));
1904 
1905 	if (compiler->last_label && compiler->last_label->size == compiler->size)
1906 		return compiler->last_label;
1907 
1908 	label = (struct sljit_label*)ensure_abuf(compiler, sizeof(struct sljit_label));
1909 	PTR_FAIL_IF(!label);
1910 	set_label(label, compiler);
1911 	return label;
1912 }
1913 
sljit_emit_jump(struct sljit_compiler * compiler,sljit_si type)1914 SLJIT_API_FUNC_ATTRIBUTE struct sljit_jump* sljit_emit_jump(struct sljit_compiler *compiler, sljit_si type)
1915 {
1916 	struct sljit_jump *jump;
1917 	sljit_ins cc;
1918 
1919 	CHECK_ERROR_PTR();
1920 	CHECK_PTR(check_sljit_emit_jump(compiler, type));
1921 
1922 	jump = (struct sljit_jump*)ensure_abuf(compiler, sizeof(struct sljit_jump));
1923 	PTR_FAIL_IF(!jump);
1924 	set_jump(jump, compiler, type & SLJIT_REWRITABLE_JUMP);
1925 	type &= 0xff;
1926 
1927 	/* In ARM, we don't need to touch the arguments. */
1928 	PTR_FAIL_IF(emit_imm32_const(compiler, TMP_REG1, 0));
1929 	if (type < SLJIT_JUMP) {
1930 		jump->flags |= IS_COND;
1931 		cc = get_cc(type);
1932 		jump->flags |= cc << 8;
1933 		PTR_FAIL_IF(push_inst16(compiler, IT | (cc << 4) | 0x8));
1934 	}
1935 
1936 	jump->addr = compiler->size;
1937 	if (type <= SLJIT_JUMP)
1938 		PTR_FAIL_IF(push_inst16(compiler, BX | RN3(TMP_REG1)));
1939 	else {
1940 		jump->flags |= IS_BL;
1941 		PTR_FAIL_IF(push_inst16(compiler, BLX | RN3(TMP_REG1)));
1942 	}
1943 
1944 	return jump;
1945 }
1946 
sljit_emit_ijump(struct sljit_compiler * compiler,sljit_si type,sljit_si src,sljit_sw srcw)1947 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_ijump(struct sljit_compiler *compiler, sljit_si type, sljit_si src, sljit_sw srcw)
1948 {
1949 	struct sljit_jump *jump;
1950 
1951 	CHECK_ERROR();
1952 	CHECK(check_sljit_emit_ijump(compiler, type, src, srcw));
1953 	ADJUST_LOCAL_OFFSET(src, srcw);
1954 
1955 	/* In ARM, we don't need to touch the arguments. */
1956 	if (!(src & SLJIT_IMM)) {
1957 		if (FAST_IS_REG(src))
1958 			return push_inst16(compiler, (type <= SLJIT_JUMP ? BX : BLX) | RN3(src));
1959 
1960 		FAIL_IF(emit_op_mem(compiler, WORD_SIZE, type <= SLJIT_JUMP ? TMP_PC : TMP_REG1, src, srcw));
1961 		if (type >= SLJIT_FAST_CALL)
1962 			return push_inst16(compiler, BLX | RN3(TMP_REG1));
1963 	}
1964 
1965 	jump = (struct sljit_jump*)ensure_abuf(compiler, sizeof(struct sljit_jump));
1966 	FAIL_IF(!jump);
1967 	set_jump(jump, compiler, JUMP_ADDR | ((type >= SLJIT_FAST_CALL) ? IS_BL : 0));
1968 	jump->u.target = srcw;
1969 
1970 	FAIL_IF(emit_imm32_const(compiler, TMP_REG1, 0));
1971 	jump->addr = compiler->size;
1972 	return push_inst16(compiler, (type <= SLJIT_JUMP ? BX : BLX) | RN3(TMP_REG1));
1973 }
1974 
sljit_emit_op_flags(struct sljit_compiler * compiler,sljit_si op,sljit_si dst,sljit_sw dstw,sljit_si src,sljit_sw srcw,sljit_si type)1975 SLJIT_API_FUNC_ATTRIBUTE sljit_si sljit_emit_op_flags(struct sljit_compiler *compiler, sljit_si op,
1976 	sljit_si dst, sljit_sw dstw,
1977 	sljit_si src, sljit_sw srcw,
1978 	sljit_si type)
1979 {
1980 	sljit_si dst_r, flags = GET_ALL_FLAGS(op);
1981 	sljit_ins cc, ins;
1982 
1983 	CHECK_ERROR();
1984 	CHECK(check_sljit_emit_op_flags(compiler, op, dst, dstw, src, srcw, type));
1985 	ADJUST_LOCAL_OFFSET(dst, dstw);
1986 	ADJUST_LOCAL_OFFSET(src, srcw);
1987 
1988 	if (dst == SLJIT_UNUSED)
1989 		return SLJIT_SUCCESS;
1990 
1991 	op = GET_OPCODE(op);
1992 	cc = get_cc(type & 0xff);
1993 	dst_r = FAST_IS_REG(dst) ? dst : TMP_REG2;
1994 
1995 	if (op < SLJIT_ADD) {
1996 		FAIL_IF(push_inst16(compiler, IT | (cc << 4) | (((cc & 0x1) ^ 0x1) << 3) | 0x4));
1997 		if (reg_map[dst_r] > 7) {
1998 			FAIL_IF(push_inst32(compiler, MOV_WI | RD4(dst_r) | 1));
1999 			FAIL_IF(push_inst32(compiler, MOV_WI | RD4(dst_r) | 0));
2000 		} else {
2001 			FAIL_IF(push_inst16(compiler, MOVSI | RDN3(dst_r) | 1));
2002 			FAIL_IF(push_inst16(compiler, MOVSI | RDN3(dst_r) | 0));
2003 		}
2004 		if (dst_r != TMP_REG2)
2005 			return SLJIT_SUCCESS;
2006 		return emit_op_mem(compiler, WORD_SIZE | STORE, TMP_REG2, dst, dstw);
2007 	}
2008 
2009 	ins = (op == SLJIT_AND ? ANDI : (op == SLJIT_OR ? ORRI : EORI));
2010 	if ((op == SLJIT_OR || op == SLJIT_XOR) && FAST_IS_REG(dst) && dst == src) {
2011 		/* Does not change the other bits. */
2012 		FAIL_IF(push_inst16(compiler, IT | (cc << 4) | 0x8));
2013 		FAIL_IF(push_inst32(compiler, ins | RN4(src) | RD4(dst) | 1));
2014 		if (flags & SLJIT_SET_E) {
2015 			/* The condition must always be set, even if the ORRI/EORI is not executed above. */
2016 			if (reg_map[dst] <= 7)
2017 				return push_inst16(compiler, MOVS | RD3(TMP_REG1) | RN3(dst));
2018 			return push_inst32(compiler, MOV_W | SET_FLAGS | RD4(TMP_REG1) | RM4(dst));
2019 		}
2020 		return SLJIT_SUCCESS;
2021 	}
2022 
2023 	compiler->cache_arg = 0;
2024 	compiler->cache_argw = 0;
2025 	if (src & SLJIT_MEM) {
2026 		FAIL_IF(emit_op_mem2(compiler, WORD_SIZE, TMP_REG2, src, srcw, dst, dstw));
2027 		src = TMP_REG2;
2028 		srcw = 0;
2029 	} else if (src & SLJIT_IMM) {
2030 		FAIL_IF(load_immediate(compiler, TMP_REG2, srcw));
2031 		src = TMP_REG2;
2032 		srcw = 0;
2033 	}
2034 
2035 	if (op == SLJIT_AND || src != dst_r) {
2036 		FAIL_IF(push_inst16(compiler, IT | (cc << 4) | (((cc & 0x1) ^ 0x1) << 3) | 0x4));
2037 		FAIL_IF(push_inst32(compiler, ins | RN4(src) | RD4(dst_r) | 1));
2038 		FAIL_IF(push_inst32(compiler, ins | RN4(src) | RD4(dst_r) | 0));
2039 	}
2040 	else {
2041 		FAIL_IF(push_inst16(compiler, IT | (cc << 4) | 0x8));
2042 		FAIL_IF(push_inst32(compiler, ins | RN4(src) | RD4(dst_r) | 1));
2043 	}
2044 
2045 	if (dst_r == TMP_REG2)
2046 		FAIL_IF(emit_op_mem2(compiler, WORD_SIZE | STORE, TMP_REG2, dst, dstw, 0, 0));
2047 
2048 	if (flags & SLJIT_SET_E) {
2049 		/* The condition must always be set, even if the ORR/EORI is not executed above. */
2050 		if (reg_map[dst_r] <= 7)
2051 			return push_inst16(compiler, MOVS | RD3(TMP_REG1) | RN3(dst_r));
2052 		return push_inst32(compiler, MOV_W | SET_FLAGS | RD4(TMP_REG1) | RM4(dst_r));
2053 	}
2054 	return SLJIT_SUCCESS;
2055 }
2056 
sljit_emit_const(struct sljit_compiler * compiler,sljit_si dst,sljit_sw dstw,sljit_sw init_value)2057 SLJIT_API_FUNC_ATTRIBUTE struct sljit_const* sljit_emit_const(struct sljit_compiler *compiler, sljit_si dst, sljit_sw dstw, sljit_sw init_value)
2058 {
2059 	struct sljit_const *const_;
2060 	sljit_si dst_r;
2061 
2062 	CHECK_ERROR_PTR();
2063 	CHECK_PTR(check_sljit_emit_const(compiler, dst, dstw, init_value));
2064 	ADJUST_LOCAL_OFFSET(dst, dstw);
2065 
2066 	const_ = (struct sljit_const*)ensure_abuf(compiler, sizeof(struct sljit_const));
2067 	PTR_FAIL_IF(!const_);
2068 	set_const(const_, compiler);
2069 
2070 	dst_r = SLOW_IS_REG(dst) ? dst : TMP_REG1;
2071 	PTR_FAIL_IF(emit_imm32_const(compiler, dst_r, init_value));
2072 
2073 	if (dst & SLJIT_MEM)
2074 		PTR_FAIL_IF(emit_op_mem(compiler, WORD_SIZE | STORE, dst_r, dst, dstw));
2075 	return const_;
2076 }
2077 
sljit_set_jump_addr(sljit_uw addr,sljit_uw new_addr)2078 SLJIT_API_FUNC_ATTRIBUTE void sljit_set_jump_addr(sljit_uw addr, sljit_uw new_addr)
2079 {
2080 	sljit_uh *inst = (sljit_uh*)addr;
2081 	modify_imm32_const(inst, new_addr);
2082 	SLJIT_CACHE_FLUSH(inst, inst + 4);
2083 }
2084 
sljit_set_const(sljit_uw addr,sljit_sw new_constant)2085 SLJIT_API_FUNC_ATTRIBUTE void sljit_set_const(sljit_uw addr, sljit_sw new_constant)
2086 {
2087 	sljit_uh *inst = (sljit_uh*)addr;
2088 	modify_imm32_const(inst, new_constant);
2089 	SLJIT_CACHE_FLUSH(inst, inst + 4);
2090 }
2091