1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 /*
3  * Copyright (c) 2023 Huawei Device Co., Ltd.
4  *
5  * Pointer authentication keys initialisation.
6  */
7 
8 #ifndef __ASM_POINTER_AUTH_KEY_H
9 #define __ASM_POINTER_AUTH_KEY_H
10 
11 #include <asm/alternative.h>
12 #include <asm/asm-offsets.h>
13 #include <asm/cpucaps.h>
14 #include <asm/sysreg.h>
15 
16 	.macro __ptrauth_address_keys_install_kernel tmp1, tmp2, tmp3
17 	ldp	\tmp2, \tmp3, [\tmp1, #PTRAUTH_KERNEL_KEY_APIB]
18 	msr_s	SYS_APIBKEYLO_EL1, \tmp2
19 	msr_s	SYS_APIBKEYHI_EL1, \tmp3
20 
21 	adr_l	\tmp1, kernel_common_keys
22 	ldp	\tmp2, \tmp3, [\tmp1, #PTRAUTH_KERNEL_KEY_APIA]
23 	msr_s	SYS_APIAKEYLO_EL1, \tmp2
24 	msr_s	SYS_APIAKEYHI_EL1, \tmp3
25 
26 	ldp	\tmp2, \tmp3, [\tmp1, #PTRAUTH_KERNEL_KEY_APDA]
27 	msr_s	SYS_APDAKEYLO_EL1, \tmp2
28 	msr_s	SYS_APDAKEYHI_EL1, \tmp3
29 
30 	ldp	\tmp2, \tmp3, [\tmp1, #PTRAUTH_KERNEL_KEY_APDB]
31 	msr_s	SYS_APDBKEYLO_EL1, \tmp2
32 	msr_s	SYS_APDBKEYHI_EL1, \tmp3
33 	.endm
34 
35 	.macro __ptrauth_generic_key_install_kernel tmp1, tmp2, tmp3
36 	ldp	\tmp2, \tmp3, [\tmp1, #PTRAUTH_KERNEL_KEY_APGA]
37 	msr_s	SYS_APGAKEYLO_EL1, \tmp2
38 	msr_s	SYS_APGAKEYHI_EL1, \tmp3
39 	.endm
40 
41 	.macro ptrauth_keys_install_kernel_all tsk, tmp1, tmp2, tmp3
42 	mov	\tmp1, #THREAD_KEYS_KERNEL
43 	add	\tmp1, \tsk, \tmp1
44 
45 alternative_if_not ARM64_HAS_ADDRESS_AUTH
46 	b	.Lno_addr_auth\@
47 alternative_else_nop_endif
48 	__ptrauth_address_keys_install_kernel \tmp1, \tmp2, \tmp3
49 
50 .Lno_addr_auth\@:
51 alternative_if ARM64_HAS_GENERIC_AUTH
52 	__ptrauth_generic_key_install_kernel \tmp1, \tmp2, \tmp3
53 alternative_else_nop_endif
54 	isb
55 	.endm
56 
57 	.macro __ptrauth_keys_install_kernel_all tsk, tmp1, tmp2, tmp3
58 	mov	\tmp1, #THREAD_KEYS_KERNEL
59 	add	\tmp1, \tsk, \tmp1
60 	__ptrauth_address_keys_install_kernel \tmp1, \tmp2, \tmp3
61 	__ptrauth_generic_key_install_kernel \tmp1, \tmp2, \tmp3
62 	.endm
63 
64 #endif
65