xref: /qemu/target/arm/tcg/pauth_helper.c (revision 370ed600)
1 /*
2  * ARM v8.3-PAuth Operations
3  *
4  * Copyright (c) 2019 Linaro, Ltd.
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library 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 GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, see <http://www.gnu.org/licenses/>.
18  */
19 
20 #include "qemu/osdep.h"
21 #include "cpu.h"
22 #include "internals.h"
23 #include "exec/exec-all.h"
24 #include "exec/cpu_ldst.h"
25 #include "exec/helper-proto.h"
26 #include "tcg/tcg-gvec-desc.h"
27 #include "qemu/xxhash.h"
28 
29 
30 static uint64_t pac_cell_shuffle(uint64_t i)
31 {
32     uint64_t o = 0;
33 
34     o |= extract64(i, 52, 4);
35     o |= extract64(i, 24, 4) << 4;
36     o |= extract64(i, 44, 4) << 8;
37     o |= extract64(i,  0, 4) << 12;
38 
39     o |= extract64(i, 28, 4) << 16;
40     o |= extract64(i, 48, 4) << 20;
41     o |= extract64(i,  4, 4) << 24;
42     o |= extract64(i, 40, 4) << 28;
43 
44     o |= extract64(i, 32, 4) << 32;
45     o |= extract64(i, 12, 4) << 36;
46     o |= extract64(i, 56, 4) << 40;
47     o |= extract64(i, 20, 4) << 44;
48 
49     o |= extract64(i,  8, 4) << 48;
50     o |= extract64(i, 36, 4) << 52;
51     o |= extract64(i, 16, 4) << 56;
52     o |= extract64(i, 60, 4) << 60;
53 
54     return o;
55 }
56 
57 static uint64_t pac_cell_inv_shuffle(uint64_t i)
58 {
59     uint64_t o = 0;
60 
61     o |= extract64(i, 12, 4);
62     o |= extract64(i, 24, 4) << 4;
63     o |= extract64(i, 48, 4) << 8;
64     o |= extract64(i, 36, 4) << 12;
65 
66     o |= extract64(i, 56, 4) << 16;
67     o |= extract64(i, 44, 4) << 20;
68     o |= extract64(i,  4, 4) << 24;
69     o |= extract64(i, 16, 4) << 28;
70 
71     o |= i & MAKE_64BIT_MASK(32, 4);
72     o |= extract64(i, 52, 4) << 36;
73     o |= extract64(i, 28, 4) << 40;
74     o |= extract64(i,  8, 4) << 44;
75 
76     o |= extract64(i, 20, 4) << 48;
77     o |= extract64(i,  0, 4) << 52;
78     o |= extract64(i, 40, 4) << 56;
79     o |= i & MAKE_64BIT_MASK(60, 4);
80 
81     return o;
82 }
83 
84 static uint64_t pac_sub(uint64_t i)
85 {
86     static const uint8_t sub[16] = {
87         0xb, 0x6, 0x8, 0xf, 0xc, 0x0, 0x9, 0xe,
88         0x3, 0x7, 0x4, 0x5, 0xd, 0x2, 0x1, 0xa,
89     };
90     uint64_t o = 0;
91     int b;
92 
93     for (b = 0; b < 64; b += 4) {
94         o |= (uint64_t)sub[(i >> b) & 0xf] << b;
95     }
96     return o;
97 }
98 
99 static uint64_t pac_inv_sub(uint64_t i)
100 {
101     static const uint8_t inv_sub[16] = {
102         0x5, 0xe, 0xd, 0x8, 0xa, 0xb, 0x1, 0x9,
103         0x2, 0x6, 0xf, 0x0, 0x4, 0xc, 0x7, 0x3,
104     };
105     uint64_t o = 0;
106     int b;
107 
108     for (b = 0; b < 64; b += 4) {
109         o |= (uint64_t)inv_sub[(i >> b) & 0xf] << b;
110     }
111     return o;
112 }
113 
114 static int rot_cell(int cell, int n)
115 {
116     /* 4-bit rotate left by n.  */
117     cell |= cell << 4;
118     return extract32(cell, 4 - n, 4);
119 }
120 
121 static uint64_t pac_mult(uint64_t i)
122 {
123     uint64_t o = 0;
124     int b;
125 
126     for (b = 0; b < 4 * 4; b += 4) {
127         int i0, i4, i8, ic, t0, t1, t2, t3;
128 
129         i0 = extract64(i, b, 4);
130         i4 = extract64(i, b + 4 * 4, 4);
131         i8 = extract64(i, b + 8 * 4, 4);
132         ic = extract64(i, b + 12 * 4, 4);
133 
134         t0 = rot_cell(i8, 1) ^ rot_cell(i4, 2) ^ rot_cell(i0, 1);
135         t1 = rot_cell(ic, 1) ^ rot_cell(i4, 1) ^ rot_cell(i0, 2);
136         t2 = rot_cell(ic, 2) ^ rot_cell(i8, 1) ^ rot_cell(i0, 1);
137         t3 = rot_cell(ic, 1) ^ rot_cell(i8, 2) ^ rot_cell(i4, 1);
138 
139         o |= (uint64_t)t3 << b;
140         o |= (uint64_t)t2 << (b + 4 * 4);
141         o |= (uint64_t)t1 << (b + 8 * 4);
142         o |= (uint64_t)t0 << (b + 12 * 4);
143     }
144     return o;
145 }
146 
147 static uint64_t tweak_cell_rot(uint64_t cell)
148 {
149     return (cell >> 1) | (((cell ^ (cell >> 1)) & 1) << 3);
150 }
151 
152 static uint64_t tweak_shuffle(uint64_t i)
153 {
154     uint64_t o = 0;
155 
156     o |= extract64(i, 16, 4) << 0;
157     o |= extract64(i, 20, 4) << 4;
158     o |= tweak_cell_rot(extract64(i, 24, 4)) << 8;
159     o |= extract64(i, 28, 4) << 12;
160 
161     o |= tweak_cell_rot(extract64(i, 44, 4)) << 16;
162     o |= extract64(i,  8, 4) << 20;
163     o |= extract64(i, 12, 4) << 24;
164     o |= tweak_cell_rot(extract64(i, 32, 4)) << 28;
165 
166     o |= extract64(i, 48, 4) << 32;
167     o |= extract64(i, 52, 4) << 36;
168     o |= extract64(i, 56, 4) << 40;
169     o |= tweak_cell_rot(extract64(i, 60, 4)) << 44;
170 
171     o |= tweak_cell_rot(extract64(i,  0, 4)) << 48;
172     o |= extract64(i,  4, 4) << 52;
173     o |= tweak_cell_rot(extract64(i, 40, 4)) << 56;
174     o |= tweak_cell_rot(extract64(i, 36, 4)) << 60;
175 
176     return o;
177 }
178 
179 static uint64_t tweak_cell_inv_rot(uint64_t cell)
180 {
181     return ((cell << 1) & 0xf) | ((cell & 1) ^ (cell >> 3));
182 }
183 
184 static uint64_t tweak_inv_shuffle(uint64_t i)
185 {
186     uint64_t o = 0;
187 
188     o |= tweak_cell_inv_rot(extract64(i, 48, 4));
189     o |= extract64(i, 52, 4) << 4;
190     o |= extract64(i, 20, 4) << 8;
191     o |= extract64(i, 24, 4) << 12;
192 
193     o |= extract64(i,  0, 4) << 16;
194     o |= extract64(i,  4, 4) << 20;
195     o |= tweak_cell_inv_rot(extract64(i,  8, 4)) << 24;
196     o |= extract64(i, 12, 4) << 28;
197 
198     o |= tweak_cell_inv_rot(extract64(i, 28, 4)) << 32;
199     o |= tweak_cell_inv_rot(extract64(i, 60, 4)) << 36;
200     o |= tweak_cell_inv_rot(extract64(i, 56, 4)) << 40;
201     o |= tweak_cell_inv_rot(extract64(i, 16, 4)) << 44;
202 
203     o |= extract64(i, 32, 4) << 48;
204     o |= extract64(i, 36, 4) << 52;
205     o |= extract64(i, 40, 4) << 56;
206     o |= tweak_cell_inv_rot(extract64(i, 44, 4)) << 60;
207 
208     return o;
209 }
210 
211 static uint64_t pauth_computepac_architected(uint64_t data, uint64_t modifier,
212                                              ARMPACKey key)
213 {
214     static const uint64_t RC[5] = {
215         0x0000000000000000ull,
216         0x13198A2E03707344ull,
217         0xA4093822299F31D0ull,
218         0x082EFA98EC4E6C89ull,
219         0x452821E638D01377ull,
220     };
221     const uint64_t alpha = 0xC0AC29B7C97C50DDull;
222     /*
223      * Note that in the ARM pseudocode, key0 contains bits <127:64>
224      * and key1 contains bits <63:0> of the 128-bit key.
225      */
226     uint64_t key0 = key.hi, key1 = key.lo;
227     uint64_t workingval, runningmod, roundkey, modk0;
228     int i;
229 
230     modk0 = (key0 << 63) | ((key0 >> 1) ^ (key0 >> 63));
231     runningmod = modifier;
232     workingval = data ^ key0;
233 
234     for (i = 0; i <= 4; ++i) {
235         roundkey = key1 ^ runningmod;
236         workingval ^= roundkey;
237         workingval ^= RC[i];
238         if (i > 0) {
239             workingval = pac_cell_shuffle(workingval);
240             workingval = pac_mult(workingval);
241         }
242         workingval = pac_sub(workingval);
243         runningmod = tweak_shuffle(runningmod);
244     }
245     roundkey = modk0 ^ runningmod;
246     workingval ^= roundkey;
247     workingval = pac_cell_shuffle(workingval);
248     workingval = pac_mult(workingval);
249     workingval = pac_sub(workingval);
250     workingval = pac_cell_shuffle(workingval);
251     workingval = pac_mult(workingval);
252     workingval ^= key1;
253     workingval = pac_cell_inv_shuffle(workingval);
254     workingval = pac_inv_sub(workingval);
255     workingval = pac_mult(workingval);
256     workingval = pac_cell_inv_shuffle(workingval);
257     workingval ^= key0;
258     workingval ^= runningmod;
259     for (i = 0; i <= 4; ++i) {
260         workingval = pac_inv_sub(workingval);
261         if (i < 4) {
262             workingval = pac_mult(workingval);
263             workingval = pac_cell_inv_shuffle(workingval);
264         }
265         runningmod = tweak_inv_shuffle(runningmod);
266         roundkey = key1 ^ runningmod;
267         workingval ^= RC[4 - i];
268         workingval ^= roundkey;
269         workingval ^= alpha;
270     }
271     workingval ^= modk0;
272 
273     return workingval;
274 }
275 
276 static uint64_t pauth_computepac_impdef(uint64_t data, uint64_t modifier,
277                                         ARMPACKey key)
278 {
279     return qemu_xxhash64_4(data, modifier, key.lo, key.hi);
280 }
281 
282 static uint64_t pauth_computepac(CPUARMState *env, uint64_t data,
283                                  uint64_t modifier, ARMPACKey key)
284 {
285     if (cpu_isar_feature(aa64_pauth_arch, env_archcpu(env))) {
286         return pauth_computepac_architected(data, modifier, key);
287     } else {
288         return pauth_computepac_impdef(data, modifier, key);
289     }
290 }
291 
292 static uint64_t pauth_addpac(CPUARMState *env, uint64_t ptr, uint64_t modifier,
293                              ARMPACKey *key, bool data)
294 {
295     ARMMMUIdx mmu_idx = arm_stage1_mmu_idx(env);
296     ARMVAParameters param = aa64_va_parameters(env, ptr, mmu_idx, data);
297     uint64_t pac, ext_ptr, ext, test;
298     int bot_bit, top_bit;
299 
300     /* If tagged pointers are in use, use ptr<55>, otherwise ptr<63>.  */
301     if (param.tbi) {
302         ext = sextract64(ptr, 55, 1);
303     } else {
304         ext = sextract64(ptr, 63, 1);
305     }
306 
307     /* Build a pointer with known good extension bits.  */
308     top_bit = 64 - 8 * param.tbi;
309     bot_bit = 64 - param.tsz;
310     ext_ptr = deposit64(ptr, bot_bit, top_bit - bot_bit, ext);
311 
312     pac = pauth_computepac(env, ext_ptr, modifier, *key);
313 
314     /*
315      * Check if the ptr has good extension bits and corrupt the
316      * pointer authentication code if not.
317      */
318     test = sextract64(ptr, bot_bit, top_bit - bot_bit);
319     if (test != 0 && test != -1) {
320         /*
321          * Note that our top_bit is one greater than the pseudocode's
322          * version, hence "- 2" here.
323          */
324         pac ^= MAKE_64BIT_MASK(top_bit - 2, 1);
325     }
326 
327     /*
328      * Preserve the determination between upper and lower at bit 55,
329      * and insert pointer authentication code.
330      */
331     if (param.tbi) {
332         ptr &= ~MAKE_64BIT_MASK(bot_bit, 55 - bot_bit + 1);
333         pac &= MAKE_64BIT_MASK(bot_bit, 54 - bot_bit + 1);
334     } else {
335         ptr &= MAKE_64BIT_MASK(0, bot_bit);
336         pac &= ~(MAKE_64BIT_MASK(55, 1) | MAKE_64BIT_MASK(0, bot_bit));
337     }
338     ext &= MAKE_64BIT_MASK(55, 1);
339     return pac | ext | ptr;
340 }
341 
342 static uint64_t pauth_original_ptr(uint64_t ptr, ARMVAParameters param)
343 {
344     uint64_t mask = pauth_ptr_mask(param);
345 
346     /* Note that bit 55 is used whether or not the regime has 2 ranges. */
347     if (extract64(ptr, 55, 1)) {
348         return ptr | mask;
349     } else {
350         return ptr & ~mask;
351     }
352 }
353 
354 static uint64_t pauth_auth(CPUARMState *env, uint64_t ptr, uint64_t modifier,
355                            ARMPACKey *key, bool data, int keynumber)
356 {
357     ARMMMUIdx mmu_idx = arm_stage1_mmu_idx(env);
358     ARMVAParameters param = aa64_va_parameters(env, ptr, mmu_idx, data);
359     int bot_bit, top_bit;
360     uint64_t pac, orig_ptr, test;
361 
362     orig_ptr = pauth_original_ptr(ptr, param);
363     pac = pauth_computepac(env, orig_ptr, modifier, *key);
364     bot_bit = 64 - param.tsz;
365     top_bit = 64 - 8 * param.tbi;
366 
367     test = (pac ^ ptr) & ~MAKE_64BIT_MASK(55, 1);
368     if (unlikely(extract64(test, bot_bit, top_bit - bot_bit))) {
369         int error_code = (keynumber << 1) | (keynumber ^ 1);
370         if (param.tbi) {
371             return deposit64(orig_ptr, 53, 2, error_code);
372         } else {
373             return deposit64(orig_ptr, 61, 2, error_code);
374         }
375     }
376     return orig_ptr;
377 }
378 
379 static uint64_t pauth_strip(CPUARMState *env, uint64_t ptr, bool data)
380 {
381     ARMMMUIdx mmu_idx = arm_stage1_mmu_idx(env);
382     ARMVAParameters param = aa64_va_parameters(env, ptr, mmu_idx, data);
383 
384     return pauth_original_ptr(ptr, param);
385 }
386 
387 static G_NORETURN
388 void pauth_trap(CPUARMState *env, int target_el, uintptr_t ra)
389 {
390     raise_exception_ra(env, EXCP_UDEF, syn_pactrap(), target_el, ra);
391 }
392 
393 static void pauth_check_trap(CPUARMState *env, int el, uintptr_t ra)
394 {
395     if (el < 2 && arm_is_el2_enabled(env)) {
396         uint64_t hcr = arm_hcr_el2_eff(env);
397         bool trap = !(hcr & HCR_API);
398         if (el == 0) {
399             /* Trap only applies to EL1&0 regime.  */
400             trap &= (hcr & (HCR_E2H | HCR_TGE)) != (HCR_E2H | HCR_TGE);
401         }
402         /* FIXME: ARMv8.3-NV: HCR_NV trap takes precedence for ERETA[AB].  */
403         if (trap) {
404             pauth_trap(env, 2, ra);
405         }
406     }
407     if (el < 3 && arm_feature(env, ARM_FEATURE_EL3)) {
408         if (!(env->cp15.scr_el3 & SCR_API)) {
409             pauth_trap(env, 3, ra);
410         }
411     }
412 }
413 
414 static bool pauth_key_enabled(CPUARMState *env, int el, uint32_t bit)
415 {
416     return (arm_sctlr(env, el) & bit) != 0;
417 }
418 
419 uint64_t HELPER(pacia)(CPUARMState *env, uint64_t x, uint64_t y)
420 {
421     int el = arm_current_el(env);
422     if (!pauth_key_enabled(env, el, SCTLR_EnIA)) {
423         return x;
424     }
425     pauth_check_trap(env, el, GETPC());
426     return pauth_addpac(env, x, y, &env->keys.apia, false);
427 }
428 
429 uint64_t HELPER(pacib)(CPUARMState *env, uint64_t x, uint64_t y)
430 {
431     int el = arm_current_el(env);
432     if (!pauth_key_enabled(env, el, SCTLR_EnIB)) {
433         return x;
434     }
435     pauth_check_trap(env, el, GETPC());
436     return pauth_addpac(env, x, y, &env->keys.apib, false);
437 }
438 
439 uint64_t HELPER(pacda)(CPUARMState *env, uint64_t x, uint64_t y)
440 {
441     int el = arm_current_el(env);
442     if (!pauth_key_enabled(env, el, SCTLR_EnDA)) {
443         return x;
444     }
445     pauth_check_trap(env, el, GETPC());
446     return pauth_addpac(env, x, y, &env->keys.apda, true);
447 }
448 
449 uint64_t HELPER(pacdb)(CPUARMState *env, uint64_t x, uint64_t y)
450 {
451     int el = arm_current_el(env);
452     if (!pauth_key_enabled(env, el, SCTLR_EnDB)) {
453         return x;
454     }
455     pauth_check_trap(env, el, GETPC());
456     return pauth_addpac(env, x, y, &env->keys.apdb, true);
457 }
458 
459 uint64_t HELPER(pacga)(CPUARMState *env, uint64_t x, uint64_t y)
460 {
461     uint64_t pac;
462 
463     pauth_check_trap(env, arm_current_el(env), GETPC());
464     pac = pauth_computepac(env, x, y, env->keys.apga);
465 
466     return pac & 0xffffffff00000000ull;
467 }
468 
469 uint64_t HELPER(autia)(CPUARMState *env, uint64_t x, uint64_t y)
470 {
471     int el = arm_current_el(env);
472     if (!pauth_key_enabled(env, el, SCTLR_EnIA)) {
473         return x;
474     }
475     pauth_check_trap(env, el, GETPC());
476     return pauth_auth(env, x, y, &env->keys.apia, false, 0);
477 }
478 
479 uint64_t HELPER(autib)(CPUARMState *env, uint64_t x, uint64_t y)
480 {
481     int el = arm_current_el(env);
482     if (!pauth_key_enabled(env, el, SCTLR_EnIB)) {
483         return x;
484     }
485     pauth_check_trap(env, el, GETPC());
486     return pauth_auth(env, x, y, &env->keys.apib, false, 1);
487 }
488 
489 uint64_t HELPER(autda)(CPUARMState *env, uint64_t x, uint64_t y)
490 {
491     int el = arm_current_el(env);
492     if (!pauth_key_enabled(env, el, SCTLR_EnDA)) {
493         return x;
494     }
495     pauth_check_trap(env, el, GETPC());
496     return pauth_auth(env, x, y, &env->keys.apda, true, 0);
497 }
498 
499 uint64_t HELPER(autdb)(CPUARMState *env, uint64_t x, uint64_t y)
500 {
501     int el = arm_current_el(env);
502     if (!pauth_key_enabled(env, el, SCTLR_EnDB)) {
503         return x;
504     }
505     pauth_check_trap(env, el, GETPC());
506     return pauth_auth(env, x, y, &env->keys.apdb, true, 1);
507 }
508 
509 uint64_t HELPER(xpaci)(CPUARMState *env, uint64_t a)
510 {
511     return pauth_strip(env, a, false);
512 }
513 
514 uint64_t HELPER(xpacd)(CPUARMState *env, uint64_t a)
515 {
516     return pauth_strip(env, a, true);
517 }
518