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