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
45alternative_if_not ARM64_HAS_ADDRESS_AUTH
46	b	.Lno_addr_auth\@
47alternative_else_nop_endif
48	__ptrauth_address_keys_install_kernel \tmp1, \tmp2, \tmp3
49
50.Lno_addr_auth\@:
51alternative_if ARM64_HAS_GENERIC_AUTH
52	__ptrauth_generic_key_install_kernel \tmp1, \tmp2, \tmp3
53alternative_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