arm64: __clear_user: handle exceptions on strb
[linux-2.6-block.git] / arch / arm64 / kernel / insn.c
CommitLineData
b11a64a4
JL
1/*
2 * Copyright (C) 2013 Huawei Ltd.
3 * Author: Jiang Liu <liuj97@gmail.com>
4 *
617d2fbc
ZSL
5 * Copyright (C) 2014 Zi Shen Lim <zlim.lnx@gmail.com>
6 *
b11a64a4
JL
7 * This program is free software; you can redistribute it and/or modify
8 * it under the terms of the GNU General Public License version 2 as
9 * published by the Free Software Foundation.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 */
5c5bf25d 19#include <linux/bitops.h>
b11a64a4
JL
20#include <linux/compiler.h>
21#include <linux/kernel.h>
ae164807
JL
22#include <linux/smp.h>
23#include <linux/stop_machine.h>
24#include <linux/uaccess.h>
a9ae04c9 25
ae164807 26#include <asm/cacheflush.h>
a9ae04c9 27#include <asm/debug-monitors.h>
b11a64a4
JL
28#include <asm/insn.h>
29
617d2fbc 30#define AARCH64_INSN_SF_BIT BIT(31)
4a89d2c9 31#define AARCH64_INSN_N_BIT BIT(22)
617d2fbc 32
b11a64a4
JL
33static int aarch64_insn_encoding_class[] = {
34 AARCH64_INSN_CLS_UNKNOWN,
35 AARCH64_INSN_CLS_UNKNOWN,
36 AARCH64_INSN_CLS_UNKNOWN,
37 AARCH64_INSN_CLS_UNKNOWN,
38 AARCH64_INSN_CLS_LDST,
39 AARCH64_INSN_CLS_DP_REG,
40 AARCH64_INSN_CLS_LDST,
41 AARCH64_INSN_CLS_DP_FPSIMD,
42 AARCH64_INSN_CLS_DP_IMM,
43 AARCH64_INSN_CLS_DP_IMM,
44 AARCH64_INSN_CLS_BR_SYS,
45 AARCH64_INSN_CLS_BR_SYS,
46 AARCH64_INSN_CLS_LDST,
47 AARCH64_INSN_CLS_DP_REG,
48 AARCH64_INSN_CLS_LDST,
49 AARCH64_INSN_CLS_DP_FPSIMD,
50};
51
52enum aarch64_insn_encoding_class __kprobes aarch64_get_insn_class(u32 insn)
53{
54 return aarch64_insn_encoding_class[(insn >> 25) & 0xf];
55}
56
57/* NOP is an alias of HINT */
58bool __kprobes aarch64_insn_is_nop(u32 insn)
59{
60 if (!aarch64_insn_is_hint(insn))
61 return false;
62
63 switch (insn & 0xFE0) {
64 case AARCH64_INSN_HINT_YIELD:
65 case AARCH64_INSN_HINT_WFE:
66 case AARCH64_INSN_HINT_WFI:
67 case AARCH64_INSN_HINT_SEV:
68 case AARCH64_INSN_HINT_SEVL:
69 return false;
70 default:
71 return true;
72 }
73}
74
ae164807
JL
75/*
76 * In ARMv8-A, A64 instructions have a fixed length of 32 bits and are always
77 * little-endian.
78 */
79int __kprobes aarch64_insn_read(void *addr, u32 *insnp)
80{
81 int ret;
82 u32 val;
83
84 ret = probe_kernel_read(&val, addr, AARCH64_INSN_SIZE);
85 if (!ret)
86 *insnp = le32_to_cpu(val);
87
88 return ret;
89}
90
91int __kprobes aarch64_insn_write(void *addr, u32 insn)
92{
93 insn = cpu_to_le32(insn);
94 return probe_kernel_write(addr, &insn, AARCH64_INSN_SIZE);
95}
96
b11a64a4
JL
97static bool __kprobes __aarch64_insn_hotpatch_safe(u32 insn)
98{
99 if (aarch64_get_insn_class(insn) != AARCH64_INSN_CLS_BR_SYS)
100 return false;
101
102 return aarch64_insn_is_b(insn) ||
103 aarch64_insn_is_bl(insn) ||
104 aarch64_insn_is_svc(insn) ||
105 aarch64_insn_is_hvc(insn) ||
106 aarch64_insn_is_smc(insn) ||
107 aarch64_insn_is_brk(insn) ||
108 aarch64_insn_is_nop(insn);
109}
110
111/*
112 * ARM Architecture Reference Manual for ARMv8 Profile-A, Issue A.a
113 * Section B2.6.5 "Concurrent modification and execution of instructions":
114 * Concurrent modification and execution of instructions can lead to the
115 * resulting instruction performing any behavior that can be achieved by
116 * executing any sequence of instructions that can be executed from the
117 * same Exception level, except where the instruction before modification
118 * and the instruction after modification is a B, BL, NOP, BKPT, SVC, HVC,
119 * or SMC instruction.
120 */
121bool __kprobes aarch64_insn_hotpatch_safe(u32 old_insn, u32 new_insn)
122{
123 return __aarch64_insn_hotpatch_safe(old_insn) &&
124 __aarch64_insn_hotpatch_safe(new_insn);
125}
ae164807
JL
126
127int __kprobes aarch64_insn_patch_text_nosync(void *addr, u32 insn)
128{
129 u32 *tp = addr;
130 int ret;
131
132 /* A64 instructions must be word aligned */
133 if ((uintptr_t)tp & 0x3)
134 return -EINVAL;
135
136 ret = aarch64_insn_write(tp, insn);
137 if (ret == 0)
138 flush_icache_range((uintptr_t)tp,
139 (uintptr_t)tp + AARCH64_INSN_SIZE);
140
141 return ret;
142}
143
144struct aarch64_insn_patch {
145 void **text_addrs;
146 u32 *new_insns;
147 int insn_cnt;
148 atomic_t cpu_count;
149};
150
151static int __kprobes aarch64_insn_patch_text_cb(void *arg)
152{
153 int i, ret = 0;
154 struct aarch64_insn_patch *pp = arg;
155
156 /* The first CPU becomes master */
157 if (atomic_inc_return(&pp->cpu_count) == 1) {
158 for (i = 0; ret == 0 && i < pp->insn_cnt; i++)
159 ret = aarch64_insn_patch_text_nosync(pp->text_addrs[i],
160 pp->new_insns[i]);
161 /*
162 * aarch64_insn_patch_text_nosync() calls flush_icache_range(),
163 * which ends with "dsb; isb" pair guaranteeing global
164 * visibility.
165 */
166 atomic_set(&pp->cpu_count, -1);
167 } else {
168 while (atomic_read(&pp->cpu_count) != -1)
169 cpu_relax();
170 isb();
171 }
172
173 return ret;
174}
175
176int __kprobes aarch64_insn_patch_text_sync(void *addrs[], u32 insns[], int cnt)
177{
178 struct aarch64_insn_patch patch = {
179 .text_addrs = addrs,
180 .new_insns = insns,
181 .insn_cnt = cnt,
182 .cpu_count = ATOMIC_INIT(0),
183 };
184
185 if (cnt <= 0)
186 return -EINVAL;
187
188 return stop_machine(aarch64_insn_patch_text_cb, &patch,
189 cpu_online_mask);
190}
191
192int __kprobes aarch64_insn_patch_text(void *addrs[], u32 insns[], int cnt)
193{
194 int ret;
195 u32 insn;
196
197 /* Unsafe to patch multiple instructions without synchronizaiton */
198 if (cnt == 1) {
199 ret = aarch64_insn_read(addrs[0], &insn);
200 if (ret)
201 return ret;
202
203 if (aarch64_insn_hotpatch_safe(insn, insns[0])) {
204 /*
205 * ARMv8 architecture doesn't guarantee all CPUs see
206 * the new instruction after returning from function
207 * aarch64_insn_patch_text_nosync(). So send IPIs to
208 * all other CPUs to achieve instruction
209 * synchronization.
210 */
211 ret = aarch64_insn_patch_text_nosync(addrs[0], insns[0]);
212 kick_all_cpus_sync();
213 return ret;
214 }
215 }
216
217 return aarch64_insn_patch_text_sync(addrs, insns, cnt);
218}
c84fced8
JL
219
220u32 __kprobes aarch64_insn_encode_immediate(enum aarch64_insn_imm_type type,
221 u32 insn, u64 imm)
222{
223 u32 immlo, immhi, lomask, himask, mask;
224 int shift;
225
226 switch (type) {
227 case AARCH64_INSN_IMM_ADR:
228 lomask = 0x3;
229 himask = 0x7ffff;
230 immlo = imm & lomask;
231 imm >>= 2;
232 immhi = imm & himask;
233 imm = (immlo << 24) | (immhi);
234 mask = (lomask << 24) | (himask);
235 shift = 5;
236 break;
237 case AARCH64_INSN_IMM_26:
238 mask = BIT(26) - 1;
239 shift = 0;
240 break;
241 case AARCH64_INSN_IMM_19:
242 mask = BIT(19) - 1;
243 shift = 5;
244 break;
245 case AARCH64_INSN_IMM_16:
246 mask = BIT(16) - 1;
247 shift = 5;
248 break;
249 case AARCH64_INSN_IMM_14:
250 mask = BIT(14) - 1;
251 shift = 5;
252 break;
253 case AARCH64_INSN_IMM_12:
254 mask = BIT(12) - 1;
255 shift = 10;
256 break;
257 case AARCH64_INSN_IMM_9:
258 mask = BIT(9) - 1;
259 shift = 12;
260 break;
1bba567d
ZSL
261 case AARCH64_INSN_IMM_7:
262 mask = BIT(7) - 1;
263 shift = 15;
264 break;
5fdc639a 265 case AARCH64_INSN_IMM_6:
4a89d2c9
ZSL
266 case AARCH64_INSN_IMM_S:
267 mask = BIT(6) - 1;
268 shift = 10;
269 break;
270 case AARCH64_INSN_IMM_R:
271 mask = BIT(6) - 1;
272 shift = 16;
273 break;
c84fced8
JL
274 default:
275 pr_err("aarch64_insn_encode_immediate: unknown immediate encoding %d\n",
276 type);
277 return 0;
278 }
279
280 /* Update the immediate field. */
281 insn &= ~(mask << shift);
282 insn |= (imm & mask) << shift;
283
284 return insn;
285}
5c5bf25d 286
617d2fbc
ZSL
287static u32 aarch64_insn_encode_register(enum aarch64_insn_register_type type,
288 u32 insn,
289 enum aarch64_insn_register reg)
290{
291 int shift;
292
293 if (reg < AARCH64_INSN_REG_0 || reg > AARCH64_INSN_REG_SP) {
294 pr_err("%s: unknown register encoding %d\n", __func__, reg);
295 return 0;
296 }
297
298 switch (type) {
299 case AARCH64_INSN_REGTYPE_RT:
9951a157 300 case AARCH64_INSN_REGTYPE_RD:
617d2fbc
ZSL
301 shift = 0;
302 break;
c0cafbae
ZSL
303 case AARCH64_INSN_REGTYPE_RN:
304 shift = 5;
305 break;
1bba567d 306 case AARCH64_INSN_REGTYPE_RT2:
27f95ba5 307 case AARCH64_INSN_REGTYPE_RA:
1bba567d
ZSL
308 shift = 10;
309 break;
17cac179
ZSL
310 case AARCH64_INSN_REGTYPE_RM:
311 shift = 16;
312 break;
617d2fbc
ZSL
313 default:
314 pr_err("%s: unknown register type encoding %d\n", __func__,
315 type);
316 return 0;
317 }
318
319 insn &= ~(GENMASK(4, 0) << shift);
320 insn |= reg << shift;
321
322 return insn;
323}
324
17cac179
ZSL
325static u32 aarch64_insn_encode_ldst_size(enum aarch64_insn_size_type type,
326 u32 insn)
327{
328 u32 size;
329
330 switch (type) {
331 case AARCH64_INSN_SIZE_8:
332 size = 0;
333 break;
334 case AARCH64_INSN_SIZE_16:
335 size = 1;
336 break;
337 case AARCH64_INSN_SIZE_32:
338 size = 2;
339 break;
340 case AARCH64_INSN_SIZE_64:
341 size = 3;
342 break;
343 default:
344 pr_err("%s: unknown size encoding %d\n", __func__, type);
345 return 0;
346 }
347
348 insn &= ~GENMASK(31, 30);
349 insn |= size << 30;
350
351 return insn;
352}
353
617d2fbc
ZSL
354static inline long branch_imm_common(unsigned long pc, unsigned long addr,
355 long range)
5c5bf25d 356{
5c5bf25d
JL
357 long offset;
358
359 /*
360 * PC: A 64-bit Program Counter holding the address of the current
361 * instruction. A64 instructions must be word-aligned.
362 */
363 BUG_ON((pc & 0x3) || (addr & 0x3));
364
617d2fbc
ZSL
365 offset = ((long)addr - (long)pc);
366 BUG_ON(offset < -range || offset >= range);
367
368 return offset;
369}
370
371u32 __kprobes aarch64_insn_gen_branch_imm(unsigned long pc, unsigned long addr,
372 enum aarch64_insn_branch_type type)
373{
374 u32 insn;
375 long offset;
376
5c5bf25d
JL
377 /*
378 * B/BL support [-128M, 128M) offset
379 * ARM64 virtual address arrangement guarantees all kernel and module
380 * texts are within +/-128M.
381 */
617d2fbc 382 offset = branch_imm_common(pc, addr, SZ_128M);
5c5bf25d 383
c0cafbae
ZSL
384 switch (type) {
385 case AARCH64_INSN_BRANCH_LINK:
5c5bf25d 386 insn = aarch64_insn_get_bl_value();
c0cafbae
ZSL
387 break;
388 case AARCH64_INSN_BRANCH_NOLINK:
5c5bf25d 389 insn = aarch64_insn_get_b_value();
c0cafbae
ZSL
390 break;
391 default:
392 BUG_ON(1);
a9ae04c9 393 return AARCH64_BREAK_FAULT;
c0cafbae 394 }
5c5bf25d
JL
395
396 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_26, insn,
397 offset >> 2);
398}
399
617d2fbc
ZSL
400u32 aarch64_insn_gen_comp_branch_imm(unsigned long pc, unsigned long addr,
401 enum aarch64_insn_register reg,
402 enum aarch64_insn_variant variant,
403 enum aarch64_insn_branch_type type)
404{
405 u32 insn;
406 long offset;
407
408 offset = branch_imm_common(pc, addr, SZ_1M);
409
410 switch (type) {
411 case AARCH64_INSN_BRANCH_COMP_ZERO:
412 insn = aarch64_insn_get_cbz_value();
413 break;
414 case AARCH64_INSN_BRANCH_COMP_NONZERO:
415 insn = aarch64_insn_get_cbnz_value();
416 break;
417 default:
418 BUG_ON(1);
a9ae04c9 419 return AARCH64_BREAK_FAULT;
617d2fbc
ZSL
420 }
421
422 switch (variant) {
423 case AARCH64_INSN_VARIANT_32BIT:
424 break;
425 case AARCH64_INSN_VARIANT_64BIT:
426 insn |= AARCH64_INSN_SF_BIT;
427 break;
428 default:
429 BUG_ON(1);
a9ae04c9 430 return AARCH64_BREAK_FAULT;
617d2fbc
ZSL
431 }
432
433 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RT, insn, reg);
434
435 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_19, insn,
436 offset >> 2);
437}
438
345e0d35
ZSL
439u32 aarch64_insn_gen_cond_branch_imm(unsigned long pc, unsigned long addr,
440 enum aarch64_insn_condition cond)
441{
442 u32 insn;
443 long offset;
444
445 offset = branch_imm_common(pc, addr, SZ_1M);
446
447 insn = aarch64_insn_get_bcond_value();
448
449 BUG_ON(cond < AARCH64_INSN_COND_EQ || cond > AARCH64_INSN_COND_AL);
450 insn |= cond;
451
452 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_19, insn,
453 offset >> 2);
454}
455
5c5bf25d
JL
456u32 __kprobes aarch64_insn_gen_hint(enum aarch64_insn_hint_op op)
457{
458 return aarch64_insn_get_hint_value() | op;
459}
460
461u32 __kprobes aarch64_insn_gen_nop(void)
462{
463 return aarch64_insn_gen_hint(AARCH64_INSN_HINT_NOP);
464}
c0cafbae
ZSL
465
466u32 aarch64_insn_gen_branch_reg(enum aarch64_insn_register reg,
467 enum aarch64_insn_branch_type type)
468{
469 u32 insn;
470
471 switch (type) {
472 case AARCH64_INSN_BRANCH_NOLINK:
473 insn = aarch64_insn_get_br_value();
474 break;
475 case AARCH64_INSN_BRANCH_LINK:
476 insn = aarch64_insn_get_blr_value();
477 break;
478 case AARCH64_INSN_BRANCH_RETURN:
479 insn = aarch64_insn_get_ret_value();
480 break;
481 default:
482 BUG_ON(1);
a9ae04c9 483 return AARCH64_BREAK_FAULT;
c0cafbae
ZSL
484 }
485
486 return aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn, reg);
487}
17cac179
ZSL
488
489u32 aarch64_insn_gen_load_store_reg(enum aarch64_insn_register reg,
490 enum aarch64_insn_register base,
491 enum aarch64_insn_register offset,
492 enum aarch64_insn_size_type size,
493 enum aarch64_insn_ldst_type type)
494{
495 u32 insn;
496
497 switch (type) {
498 case AARCH64_INSN_LDST_LOAD_REG_OFFSET:
499 insn = aarch64_insn_get_ldr_reg_value();
500 break;
501 case AARCH64_INSN_LDST_STORE_REG_OFFSET:
502 insn = aarch64_insn_get_str_reg_value();
503 break;
504 default:
505 BUG_ON(1);
a9ae04c9 506 return AARCH64_BREAK_FAULT;
17cac179
ZSL
507 }
508
509 insn = aarch64_insn_encode_ldst_size(size, insn);
510
511 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RT, insn, reg);
512
513 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn,
514 base);
515
516 return aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RM, insn,
517 offset);
518}
1bba567d
ZSL
519
520u32 aarch64_insn_gen_load_store_pair(enum aarch64_insn_register reg1,
521 enum aarch64_insn_register reg2,
522 enum aarch64_insn_register base,
523 int offset,
524 enum aarch64_insn_variant variant,
525 enum aarch64_insn_ldst_type type)
526{
527 u32 insn;
528 int shift;
529
530 switch (type) {
531 case AARCH64_INSN_LDST_LOAD_PAIR_PRE_INDEX:
532 insn = aarch64_insn_get_ldp_pre_value();
533 break;
534 case AARCH64_INSN_LDST_STORE_PAIR_PRE_INDEX:
535 insn = aarch64_insn_get_stp_pre_value();
536 break;
537 case AARCH64_INSN_LDST_LOAD_PAIR_POST_INDEX:
538 insn = aarch64_insn_get_ldp_post_value();
539 break;
540 case AARCH64_INSN_LDST_STORE_PAIR_POST_INDEX:
541 insn = aarch64_insn_get_stp_post_value();
542 break;
543 default:
544 BUG_ON(1);
a9ae04c9 545 return AARCH64_BREAK_FAULT;
1bba567d
ZSL
546 }
547
548 switch (variant) {
549 case AARCH64_INSN_VARIANT_32BIT:
550 /* offset must be multiples of 4 in the range [-256, 252] */
551 BUG_ON(offset & 0x3);
552 BUG_ON(offset < -256 || offset > 252);
553 shift = 2;
554 break;
555 case AARCH64_INSN_VARIANT_64BIT:
556 /* offset must be multiples of 8 in the range [-512, 504] */
557 BUG_ON(offset & 0x7);
558 BUG_ON(offset < -512 || offset > 504);
559 shift = 3;
560 insn |= AARCH64_INSN_SF_BIT;
561 break;
562 default:
563 BUG_ON(1);
a9ae04c9 564 return AARCH64_BREAK_FAULT;
1bba567d
ZSL
565 }
566
567 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RT, insn,
568 reg1);
569
570 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RT2, insn,
571 reg2);
572
573 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn,
574 base);
575
576 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_7, insn,
577 offset >> shift);
578}
9951a157
ZSL
579
580u32 aarch64_insn_gen_add_sub_imm(enum aarch64_insn_register dst,
581 enum aarch64_insn_register src,
582 int imm, enum aarch64_insn_variant variant,
583 enum aarch64_insn_adsb_type type)
584{
585 u32 insn;
586
587 switch (type) {
588 case AARCH64_INSN_ADSB_ADD:
589 insn = aarch64_insn_get_add_imm_value();
590 break;
591 case AARCH64_INSN_ADSB_SUB:
592 insn = aarch64_insn_get_sub_imm_value();
593 break;
594 case AARCH64_INSN_ADSB_ADD_SETFLAGS:
595 insn = aarch64_insn_get_adds_imm_value();
596 break;
597 case AARCH64_INSN_ADSB_SUB_SETFLAGS:
598 insn = aarch64_insn_get_subs_imm_value();
599 break;
600 default:
601 BUG_ON(1);
a9ae04c9 602 return AARCH64_BREAK_FAULT;
9951a157
ZSL
603 }
604
605 switch (variant) {
606 case AARCH64_INSN_VARIANT_32BIT:
607 break;
608 case AARCH64_INSN_VARIANT_64BIT:
609 insn |= AARCH64_INSN_SF_BIT;
610 break;
611 default:
612 BUG_ON(1);
a9ae04c9 613 return AARCH64_BREAK_FAULT;
9951a157
ZSL
614 }
615
616 BUG_ON(imm & ~(SZ_4K - 1));
617
618 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RD, insn, dst);
619
620 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn, src);
621
622 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_12, insn, imm);
623}
4a89d2c9
ZSL
624
625u32 aarch64_insn_gen_bitfield(enum aarch64_insn_register dst,
626 enum aarch64_insn_register src,
627 int immr, int imms,
628 enum aarch64_insn_variant variant,
629 enum aarch64_insn_bitfield_type type)
630{
631 u32 insn;
632 u32 mask;
633
634 switch (type) {
635 case AARCH64_INSN_BITFIELD_MOVE:
636 insn = aarch64_insn_get_bfm_value();
637 break;
638 case AARCH64_INSN_BITFIELD_MOVE_UNSIGNED:
639 insn = aarch64_insn_get_ubfm_value();
640 break;
641 case AARCH64_INSN_BITFIELD_MOVE_SIGNED:
642 insn = aarch64_insn_get_sbfm_value();
643 break;
644 default:
645 BUG_ON(1);
a9ae04c9 646 return AARCH64_BREAK_FAULT;
4a89d2c9
ZSL
647 }
648
649 switch (variant) {
650 case AARCH64_INSN_VARIANT_32BIT:
651 mask = GENMASK(4, 0);
652 break;
653 case AARCH64_INSN_VARIANT_64BIT:
654 insn |= AARCH64_INSN_SF_BIT | AARCH64_INSN_N_BIT;
655 mask = GENMASK(5, 0);
656 break;
657 default:
658 BUG_ON(1);
a9ae04c9 659 return AARCH64_BREAK_FAULT;
4a89d2c9
ZSL
660 }
661
662 BUG_ON(immr & ~mask);
663 BUG_ON(imms & ~mask);
664
665 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RD, insn, dst);
666
667 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn, src);
668
669 insn = aarch64_insn_encode_immediate(AARCH64_INSN_IMM_R, insn, immr);
670
671 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_S, insn, imms);
672}
6098f2d5
ZSL
673
674u32 aarch64_insn_gen_movewide(enum aarch64_insn_register dst,
675 int imm, int shift,
676 enum aarch64_insn_variant variant,
677 enum aarch64_insn_movewide_type type)
678{
679 u32 insn;
680
681 switch (type) {
682 case AARCH64_INSN_MOVEWIDE_ZERO:
683 insn = aarch64_insn_get_movz_value();
684 break;
685 case AARCH64_INSN_MOVEWIDE_KEEP:
686 insn = aarch64_insn_get_movk_value();
687 break;
688 case AARCH64_INSN_MOVEWIDE_INVERSE:
689 insn = aarch64_insn_get_movn_value();
690 break;
691 default:
692 BUG_ON(1);
a9ae04c9 693 return AARCH64_BREAK_FAULT;
6098f2d5
ZSL
694 }
695
696 BUG_ON(imm & ~(SZ_64K - 1));
697
698 switch (variant) {
699 case AARCH64_INSN_VARIANT_32BIT:
700 BUG_ON(shift != 0 && shift != 16);
701 break;
702 case AARCH64_INSN_VARIANT_64BIT:
703 insn |= AARCH64_INSN_SF_BIT;
704 BUG_ON(shift != 0 && shift != 16 && shift != 32 &&
705 shift != 48);
706 break;
707 default:
708 BUG_ON(1);
a9ae04c9 709 return AARCH64_BREAK_FAULT;
6098f2d5
ZSL
710 }
711
712 insn |= (shift >> 4) << 21;
713
714 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RD, insn, dst);
715
716 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_16, insn, imm);
717}
5fdc639a
ZSL
718
719u32 aarch64_insn_gen_add_sub_shifted_reg(enum aarch64_insn_register dst,
720 enum aarch64_insn_register src,
721 enum aarch64_insn_register reg,
722 int shift,
723 enum aarch64_insn_variant variant,
724 enum aarch64_insn_adsb_type type)
725{
726 u32 insn;
727
728 switch (type) {
729 case AARCH64_INSN_ADSB_ADD:
730 insn = aarch64_insn_get_add_value();
731 break;
732 case AARCH64_INSN_ADSB_SUB:
733 insn = aarch64_insn_get_sub_value();
734 break;
735 case AARCH64_INSN_ADSB_ADD_SETFLAGS:
736 insn = aarch64_insn_get_adds_value();
737 break;
738 case AARCH64_INSN_ADSB_SUB_SETFLAGS:
739 insn = aarch64_insn_get_subs_value();
740 break;
741 default:
742 BUG_ON(1);
a9ae04c9 743 return AARCH64_BREAK_FAULT;
5fdc639a
ZSL
744 }
745
746 switch (variant) {
747 case AARCH64_INSN_VARIANT_32BIT:
748 BUG_ON(shift & ~(SZ_32 - 1));
749 break;
750 case AARCH64_INSN_VARIANT_64BIT:
751 insn |= AARCH64_INSN_SF_BIT;
752 BUG_ON(shift & ~(SZ_64 - 1));
753 break;
754 default:
755 BUG_ON(1);
a9ae04c9 756 return AARCH64_BREAK_FAULT;
5fdc639a
ZSL
757 }
758
759
760 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RD, insn, dst);
761
762 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn, src);
763
764 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RM, insn, reg);
765
766 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_6, insn, shift);
767}
546dd36b
ZSL
768
769u32 aarch64_insn_gen_data1(enum aarch64_insn_register dst,
770 enum aarch64_insn_register src,
771 enum aarch64_insn_variant variant,
772 enum aarch64_insn_data1_type type)
773{
774 u32 insn;
775
776 switch (type) {
777 case AARCH64_INSN_DATA1_REVERSE_16:
778 insn = aarch64_insn_get_rev16_value();
779 break;
780 case AARCH64_INSN_DATA1_REVERSE_32:
781 insn = aarch64_insn_get_rev32_value();
782 break;
783 case AARCH64_INSN_DATA1_REVERSE_64:
784 BUG_ON(variant != AARCH64_INSN_VARIANT_64BIT);
785 insn = aarch64_insn_get_rev64_value();
786 break;
787 default:
788 BUG_ON(1);
a9ae04c9 789 return AARCH64_BREAK_FAULT;
546dd36b
ZSL
790 }
791
792 switch (variant) {
793 case AARCH64_INSN_VARIANT_32BIT:
794 break;
795 case AARCH64_INSN_VARIANT_64BIT:
796 insn |= AARCH64_INSN_SF_BIT;
797 break;
798 default:
799 BUG_ON(1);
a9ae04c9 800 return AARCH64_BREAK_FAULT;
546dd36b
ZSL
801 }
802
803 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RD, insn, dst);
804
805 return aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn, src);
806}
64810639
ZSL
807
808u32 aarch64_insn_gen_data2(enum aarch64_insn_register dst,
809 enum aarch64_insn_register src,
810 enum aarch64_insn_register reg,
811 enum aarch64_insn_variant variant,
812 enum aarch64_insn_data2_type type)
813{
814 u32 insn;
815
816 switch (type) {
817 case AARCH64_INSN_DATA2_UDIV:
818 insn = aarch64_insn_get_udiv_value();
819 break;
820 case AARCH64_INSN_DATA2_SDIV:
821 insn = aarch64_insn_get_sdiv_value();
822 break;
823 case AARCH64_INSN_DATA2_LSLV:
824 insn = aarch64_insn_get_lslv_value();
825 break;
826 case AARCH64_INSN_DATA2_LSRV:
827 insn = aarch64_insn_get_lsrv_value();
828 break;
829 case AARCH64_INSN_DATA2_ASRV:
830 insn = aarch64_insn_get_asrv_value();
831 break;
832 case AARCH64_INSN_DATA2_RORV:
833 insn = aarch64_insn_get_rorv_value();
834 break;
835 default:
836 BUG_ON(1);
a9ae04c9 837 return AARCH64_BREAK_FAULT;
64810639
ZSL
838 }
839
840 switch (variant) {
841 case AARCH64_INSN_VARIANT_32BIT:
842 break;
843 case AARCH64_INSN_VARIANT_64BIT:
844 insn |= AARCH64_INSN_SF_BIT;
845 break;
846 default:
847 BUG_ON(1);
a9ae04c9 848 return AARCH64_BREAK_FAULT;
64810639
ZSL
849 }
850
851 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RD, insn, dst);
852
853 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn, src);
854
855 return aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RM, insn, reg);
856}
27f95ba5
ZSL
857
858u32 aarch64_insn_gen_data3(enum aarch64_insn_register dst,
859 enum aarch64_insn_register src,
860 enum aarch64_insn_register reg1,
861 enum aarch64_insn_register reg2,
862 enum aarch64_insn_variant variant,
863 enum aarch64_insn_data3_type type)
864{
865 u32 insn;
866
867 switch (type) {
868 case AARCH64_INSN_DATA3_MADD:
869 insn = aarch64_insn_get_madd_value();
870 break;
871 case AARCH64_INSN_DATA3_MSUB:
872 insn = aarch64_insn_get_msub_value();
873 break;
874 default:
875 BUG_ON(1);
a9ae04c9 876 return AARCH64_BREAK_FAULT;
27f95ba5
ZSL
877 }
878
879 switch (variant) {
880 case AARCH64_INSN_VARIANT_32BIT:
881 break;
882 case AARCH64_INSN_VARIANT_64BIT:
883 insn |= AARCH64_INSN_SF_BIT;
884 break;
885 default:
886 BUG_ON(1);
a9ae04c9 887 return AARCH64_BREAK_FAULT;
27f95ba5
ZSL
888 }
889
890 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RD, insn, dst);
891
892 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RA, insn, src);
893
894 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn,
895 reg1);
896
897 return aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RM, insn,
898 reg2);
899}
5e6e15a2
ZSL
900
901u32 aarch64_insn_gen_logical_shifted_reg(enum aarch64_insn_register dst,
902 enum aarch64_insn_register src,
903 enum aarch64_insn_register reg,
904 int shift,
905 enum aarch64_insn_variant variant,
906 enum aarch64_insn_logic_type type)
907{
908 u32 insn;
909
910 switch (type) {
911 case AARCH64_INSN_LOGIC_AND:
912 insn = aarch64_insn_get_and_value();
913 break;
914 case AARCH64_INSN_LOGIC_BIC:
915 insn = aarch64_insn_get_bic_value();
916 break;
917 case AARCH64_INSN_LOGIC_ORR:
918 insn = aarch64_insn_get_orr_value();
919 break;
920 case AARCH64_INSN_LOGIC_ORN:
921 insn = aarch64_insn_get_orn_value();
922 break;
923 case AARCH64_INSN_LOGIC_EOR:
924 insn = aarch64_insn_get_eor_value();
925 break;
926 case AARCH64_INSN_LOGIC_EON:
927 insn = aarch64_insn_get_eon_value();
928 break;
929 case AARCH64_INSN_LOGIC_AND_SETFLAGS:
930 insn = aarch64_insn_get_ands_value();
931 break;
932 case AARCH64_INSN_LOGIC_BIC_SETFLAGS:
933 insn = aarch64_insn_get_bics_value();
934 break;
935 default:
936 BUG_ON(1);
a9ae04c9 937 return AARCH64_BREAK_FAULT;
5e6e15a2
ZSL
938 }
939
940 switch (variant) {
941 case AARCH64_INSN_VARIANT_32BIT:
942 BUG_ON(shift & ~(SZ_32 - 1));
943 break;
944 case AARCH64_INSN_VARIANT_64BIT:
945 insn |= AARCH64_INSN_SF_BIT;
946 BUG_ON(shift & ~(SZ_64 - 1));
947 break;
948 default:
949 BUG_ON(1);
a9ae04c9 950 return AARCH64_BREAK_FAULT;
5e6e15a2
ZSL
951 }
952
953
954 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RD, insn, dst);
955
956 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RN, insn, src);
957
958 insn = aarch64_insn_encode_register(AARCH64_INSN_REGTYPE_RM, insn, reg);
959
960 return aarch64_insn_encode_immediate(AARCH64_INSN_IMM_6, insn, shift);
961}