162306a36Sopenharmony_ci/* SPDX-License-Identifier: GPL-2.0-only */
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Copyright (C) 2012 ARM Ltd.
462306a36Sopenharmony_ci * Author: Marc Zyngier <marc.zyngier@arm.com>
562306a36Sopenharmony_ci */
662306a36Sopenharmony_ci
762306a36Sopenharmony_ci#ifndef __ASM__VIRT_H
862306a36Sopenharmony_ci#define __ASM__VIRT_H
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci/*
1162306a36Sopenharmony_ci * The arm64 hcall implementation uses x0 to specify the hcall
1262306a36Sopenharmony_ci * number. A value less than HVC_STUB_HCALL_NR indicates a special
1362306a36Sopenharmony_ci * hcall, such as set vector. Any other value is handled in a
1462306a36Sopenharmony_ci * hypervisor specific way.
1562306a36Sopenharmony_ci *
1662306a36Sopenharmony_ci * The hypercall is allowed to clobber any of the caller-saved
1762306a36Sopenharmony_ci * registers (x0-x18), so it is advisable to use it through the
1862306a36Sopenharmony_ci * indirection of a function call (as implemented in hyp-stub.S).
1962306a36Sopenharmony_ci */
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci/*
2262306a36Sopenharmony_ci * HVC_SET_VECTORS - Set the value of the vbar_el2 register.
2362306a36Sopenharmony_ci *
2462306a36Sopenharmony_ci * @x1: Physical address of the new vector table.
2562306a36Sopenharmony_ci */
2662306a36Sopenharmony_ci#define HVC_SET_VECTORS 0
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ci/*
2962306a36Sopenharmony_ci * HVC_SOFT_RESTART - CPU soft reset, used by the cpu_soft_restart routine.
3062306a36Sopenharmony_ci */
3162306a36Sopenharmony_ci#define HVC_SOFT_RESTART 1
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci/*
3462306a36Sopenharmony_ci * HVC_RESET_VECTORS - Restore the vectors to the original HYP stubs
3562306a36Sopenharmony_ci */
3662306a36Sopenharmony_ci#define HVC_RESET_VECTORS 2
3762306a36Sopenharmony_ci
3862306a36Sopenharmony_ci/*
3962306a36Sopenharmony_ci * HVC_FINALISE_EL2 - Upgrade the CPU from EL1 to EL2, if possible
4062306a36Sopenharmony_ci */
4162306a36Sopenharmony_ci#define HVC_FINALISE_EL2	3
4262306a36Sopenharmony_ci
4362306a36Sopenharmony_ci/* Max number of HYP stub hypercalls */
4462306a36Sopenharmony_ci#define HVC_STUB_HCALL_NR 4
4562306a36Sopenharmony_ci
4662306a36Sopenharmony_ci/* Error returned when an invalid stub number is passed into x0 */
4762306a36Sopenharmony_ci#define HVC_STUB_ERR	0xbadca11
4862306a36Sopenharmony_ci
4962306a36Sopenharmony_ci#define BOOT_CPU_MODE_EL1	(0xe11)
5062306a36Sopenharmony_ci#define BOOT_CPU_MODE_EL2	(0xe12)
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_ci/*
5362306a36Sopenharmony_ci * Flags returned together with the boot mode, but not preserved in
5462306a36Sopenharmony_ci * __boot_cpu_mode. Used by the idreg override code to work out the
5562306a36Sopenharmony_ci * boot state.
5662306a36Sopenharmony_ci */
5762306a36Sopenharmony_ci#define BOOT_CPU_FLAG_E2H	BIT_ULL(32)
5862306a36Sopenharmony_ci
5962306a36Sopenharmony_ci#ifndef __ASSEMBLY__
6062306a36Sopenharmony_ci
6162306a36Sopenharmony_ci#include <asm/ptrace.h>
6262306a36Sopenharmony_ci#include <asm/sections.h>
6362306a36Sopenharmony_ci#include <asm/sysreg.h>
6462306a36Sopenharmony_ci#include <asm/cpufeature.h>
6562306a36Sopenharmony_ci
6662306a36Sopenharmony_ci/*
6762306a36Sopenharmony_ci * __boot_cpu_mode records what mode CPUs were booted in.
6862306a36Sopenharmony_ci * A correctly-implemented bootloader must start all CPUs in the same mode:
6962306a36Sopenharmony_ci * In this case, both 32bit halves of __boot_cpu_mode will contain the
7062306a36Sopenharmony_ci * same value (either 0 if booted in EL1, BOOT_CPU_MODE_EL2 if booted in EL2).
7162306a36Sopenharmony_ci *
7262306a36Sopenharmony_ci * Should the bootloader fail to do this, the two values will be different.
7362306a36Sopenharmony_ci * This allows the kernel to flag an error when the secondaries have come up.
7462306a36Sopenharmony_ci */
7562306a36Sopenharmony_ciextern u32 __boot_cpu_mode[2];
7662306a36Sopenharmony_ci
7762306a36Sopenharmony_ci#define ARM64_VECTOR_TABLE_LEN	SZ_2K
7862306a36Sopenharmony_ci
7962306a36Sopenharmony_civoid __hyp_set_vectors(phys_addr_t phys_vector_base);
8062306a36Sopenharmony_civoid __hyp_reset_vectors(void);
8162306a36Sopenharmony_cibool is_kvm_arm_initialised(void);
8262306a36Sopenharmony_ci
8362306a36Sopenharmony_ciDECLARE_STATIC_KEY_FALSE(kvm_protected_mode_initialized);
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci/* Reports the availability of HYP mode */
8662306a36Sopenharmony_cistatic inline bool is_hyp_mode_available(void)
8762306a36Sopenharmony_ci{
8862306a36Sopenharmony_ci	/*
8962306a36Sopenharmony_ci	 * If KVM protected mode is initialized, all CPUs must have been booted
9062306a36Sopenharmony_ci	 * in EL2. Avoid checking __boot_cpu_mode as CPUs now come up in EL1.
9162306a36Sopenharmony_ci	 */
9262306a36Sopenharmony_ci	if (IS_ENABLED(CONFIG_KVM) &&
9362306a36Sopenharmony_ci	    static_branch_likely(&kvm_protected_mode_initialized))
9462306a36Sopenharmony_ci		return true;
9562306a36Sopenharmony_ci
9662306a36Sopenharmony_ci	return (__boot_cpu_mode[0] == BOOT_CPU_MODE_EL2 &&
9762306a36Sopenharmony_ci		__boot_cpu_mode[1] == BOOT_CPU_MODE_EL2);
9862306a36Sopenharmony_ci}
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_ci/* Check if the bootloader has booted CPUs in different modes */
10162306a36Sopenharmony_cistatic inline bool is_hyp_mode_mismatched(void)
10262306a36Sopenharmony_ci{
10362306a36Sopenharmony_ci	/*
10462306a36Sopenharmony_ci	 * If KVM protected mode is initialized, all CPUs must have been booted
10562306a36Sopenharmony_ci	 * in EL2. Avoid checking __boot_cpu_mode as CPUs now come up in EL1.
10662306a36Sopenharmony_ci	 */
10762306a36Sopenharmony_ci	if (IS_ENABLED(CONFIG_KVM) &&
10862306a36Sopenharmony_ci	    static_branch_likely(&kvm_protected_mode_initialized))
10962306a36Sopenharmony_ci		return false;
11062306a36Sopenharmony_ci
11162306a36Sopenharmony_ci	return __boot_cpu_mode[0] != __boot_cpu_mode[1];
11262306a36Sopenharmony_ci}
11362306a36Sopenharmony_ci
11462306a36Sopenharmony_cistatic __always_inline bool is_kernel_in_hyp_mode(void)
11562306a36Sopenharmony_ci{
11662306a36Sopenharmony_ci	BUILD_BUG_ON(__is_defined(__KVM_NVHE_HYPERVISOR__) ||
11762306a36Sopenharmony_ci		     __is_defined(__KVM_VHE_HYPERVISOR__));
11862306a36Sopenharmony_ci	return read_sysreg(CurrentEL) == CurrentEL_EL2;
11962306a36Sopenharmony_ci}
12062306a36Sopenharmony_ci
12162306a36Sopenharmony_cistatic __always_inline bool has_vhe(void)
12262306a36Sopenharmony_ci{
12362306a36Sopenharmony_ci	/*
12462306a36Sopenharmony_ci	 * Code only run in VHE/NVHE hyp context can assume VHE is present or
12562306a36Sopenharmony_ci	 * absent. Otherwise fall back to caps.
12662306a36Sopenharmony_ci	 * This allows the compiler to discard VHE-specific code from the
12762306a36Sopenharmony_ci	 * nVHE object, reducing the number of external symbol references
12862306a36Sopenharmony_ci	 * needed to link.
12962306a36Sopenharmony_ci	 */
13062306a36Sopenharmony_ci	if (is_vhe_hyp_code())
13162306a36Sopenharmony_ci		return true;
13262306a36Sopenharmony_ci	else if (is_nvhe_hyp_code())
13362306a36Sopenharmony_ci		return false;
13462306a36Sopenharmony_ci	else
13562306a36Sopenharmony_ci		return cpus_have_final_cap(ARM64_HAS_VIRT_HOST_EXTN);
13662306a36Sopenharmony_ci}
13762306a36Sopenharmony_ci
13862306a36Sopenharmony_cistatic __always_inline bool is_protected_kvm_enabled(void)
13962306a36Sopenharmony_ci{
14062306a36Sopenharmony_ci	if (is_vhe_hyp_code())
14162306a36Sopenharmony_ci		return false;
14262306a36Sopenharmony_ci	else
14362306a36Sopenharmony_ci		return cpus_have_final_cap(ARM64_KVM_PROTECTED_MODE);
14462306a36Sopenharmony_ci}
14562306a36Sopenharmony_ci
14662306a36Sopenharmony_cistatic __always_inline bool has_hvhe(void)
14762306a36Sopenharmony_ci{
14862306a36Sopenharmony_ci	if (is_vhe_hyp_code())
14962306a36Sopenharmony_ci		return false;
15062306a36Sopenharmony_ci
15162306a36Sopenharmony_ci	return cpus_have_final_cap(ARM64_KVM_HVHE);
15262306a36Sopenharmony_ci}
15362306a36Sopenharmony_ci
15462306a36Sopenharmony_cistatic inline bool is_hyp_nvhe(void)
15562306a36Sopenharmony_ci{
15662306a36Sopenharmony_ci	return is_hyp_mode_available() && !is_kernel_in_hyp_mode();
15762306a36Sopenharmony_ci}
15862306a36Sopenharmony_ci
15962306a36Sopenharmony_ci#endif /* __ASSEMBLY__ */
16062306a36Sopenharmony_ci
16162306a36Sopenharmony_ci#endif /* ! __ASM__VIRT_H */
162