Commit b9527b38 authored by Marc Zyngier's avatar Marc Zyngier Committed by Oliver Upton
Browse files

KVM: arm64: nv: Save/Restore vEL2 sysregs 



Whenever we need to restore the guest's system registers to the CPU, we
now need to take care of the EL2 system registers as well. Most of them
are accessed via traps only, but some have an immediate effect and also
a guest running in VHE mode would expect them to be accessible via their
EL1 encoding, which we do not trap.

For vEL2 we write the virtual EL2 registers with an identical format directly
into their EL1 counterpart, and translate the few registers that have a
different format for the same effect on the execution when running a
non-VHE guest guest hypervisor.

Based on an initial patch from Andre Przywara, rewritten many times
since.

Reviewed-by: default avatarAlexandru Elisei <alexandru.elisei@arm.com>
Signed-off-by: default avatarMarc Zyngier <maz@kernel.org>
Link: https://lore.kernel.org/r/20241023145345.1613824-8-maz@kernel.org


Signed-off-by: default avatarOliver Upton <oliver.upton@linux.dev>
parent 164b5e20

arch/arm64/kvm/hyp/include/hyp/sysreg-sr.h

+3 −2
Original line number Diff line number Diff line
@@ -152,9 +152,10 @@ static inline void __sysreg_restore_user_state(struct kvm_cpu_context *ctxt) 
	write_sysreg(ctxt_sys_reg(ctxt, TPIDRRO_EL0),	tpidrro_el0);

}



static inline void __sysreg_restore_el1_state(struct kvm_cpu_context *ctxt)

static inline void __sysreg_restore_el1_state(struct kvm_cpu_context *ctxt,

					      u64 mpidr)

{

	write_sysreg(ctxt_sys_reg(ctxt, MPIDR_EL1),	vmpidr_el2);

	write_sysreg(mpidr,				vmpidr_el2);



	if (has_vhe() ||

	    !cpus_have_final_cap(ARM64_WORKAROUND_SPECULATIVE_AT)) {

arch/arm64/kvm/hyp/nvhe/sysreg-sr.c

+1 −1
Original line number Diff line number Diff line
@@ -28,7 +28,7 @@ void __sysreg_save_state_nvhe(struct kvm_cpu_context *ctxt) 


void __sysreg_restore_state_nvhe(struct kvm_cpu_context *ctxt)

{

	__sysreg_restore_el1_state(ctxt);

	__sysreg_restore_el1_state(ctxt, ctxt_sys_reg(ctxt, MPIDR_EL1));

	__sysreg_restore_common_state(ctxt);

	__sysreg_restore_user_state(ctxt);

	__sysreg_restore_el2_return_state(ctxt);

arch/arm64/kvm/hyp/vhe/sysreg-sr.c

+134 −2
Original line number Diff line number Diff line
@@ -15,6 +15,107 @@ 
#include <asm/kvm_hyp.h>

#include <asm/kvm_nested.h>



static void __sysreg_save_vel2_state(struct kvm_vcpu *vcpu)

{

	/* These registers are common with EL1 */

	__vcpu_sys_reg(vcpu, PAR_EL1)	= read_sysreg(par_el1);

	__vcpu_sys_reg(vcpu, TPIDR_EL1)	= read_sysreg(tpidr_el1);



	__vcpu_sys_reg(vcpu, ESR_EL2)	= read_sysreg_el1(SYS_ESR);

	__vcpu_sys_reg(vcpu, AFSR0_EL2)	= read_sysreg_el1(SYS_AFSR0);

	__vcpu_sys_reg(vcpu, AFSR1_EL2)	= read_sysreg_el1(SYS_AFSR1);

	__vcpu_sys_reg(vcpu, FAR_EL2)	= read_sysreg_el1(SYS_FAR);

	__vcpu_sys_reg(vcpu, MAIR_EL2)	= read_sysreg_el1(SYS_MAIR);

	__vcpu_sys_reg(vcpu, VBAR_EL2)	= read_sysreg_el1(SYS_VBAR);

	__vcpu_sys_reg(vcpu, CONTEXTIDR_EL2) = read_sysreg_el1(SYS_CONTEXTIDR);

	__vcpu_sys_reg(vcpu, AMAIR_EL2)	= read_sysreg_el1(SYS_AMAIR);



	/*

	 * In VHE mode those registers are compatible between EL1 and EL2,

	 * and the guest uses the _EL1 versions on the CPU naturally.

	 * So we save them into their _EL2 versions here.

	 * For nVHE mode we trap accesses to those registers, so our

	 * _EL2 copy in sys_regs[] is always up-to-date and we don't need

	 * to save anything here.

	 */

	if (vcpu_el2_e2h_is_set(vcpu)) {

		u64 val;



		/*

		 * We don't save CPTR_EL2, as accesses to CPACR_EL1

		 * are always trapped, ensuring that the in-memory

		 * copy is always up-to-date. A small blessing...

		 */

		__vcpu_sys_reg(vcpu, SCTLR_EL2)	= read_sysreg_el1(SYS_SCTLR);

		__vcpu_sys_reg(vcpu, TTBR0_EL2)	= read_sysreg_el1(SYS_TTBR0);

		__vcpu_sys_reg(vcpu, TTBR1_EL2)	= read_sysreg_el1(SYS_TTBR1);

		__vcpu_sys_reg(vcpu, TCR_EL2)	= read_sysreg_el1(SYS_TCR);



		/*

		 * The EL1 view of CNTKCTL_EL1 has a bunch of RES0 bits where

		 * the interesting CNTHCTL_EL2 bits live. So preserve these

		 * bits when reading back the guest-visible value.

		 */

		val = read_sysreg_el1(SYS_CNTKCTL);

		val &= CNTKCTL_VALID_BITS;

		__vcpu_sys_reg(vcpu, CNTHCTL_EL2) &= ~CNTKCTL_VALID_BITS;

		__vcpu_sys_reg(vcpu, CNTHCTL_EL2) |= val;

	}



	__vcpu_sys_reg(vcpu, SP_EL2)	= read_sysreg(sp_el1);

	__vcpu_sys_reg(vcpu, ELR_EL2)	= read_sysreg_el1(SYS_ELR);

	__vcpu_sys_reg(vcpu, SPSR_EL2)	= read_sysreg_el1(SYS_SPSR);

}



static void __sysreg_restore_vel2_state(struct kvm_vcpu *vcpu)

{

	u64 val;



	/* These registers are common with EL1 */

	write_sysreg(__vcpu_sys_reg(vcpu, PAR_EL1),	par_el1);

	write_sysreg(__vcpu_sys_reg(vcpu, TPIDR_EL1),	tpidr_el1);



	write_sysreg(__vcpu_sys_reg(vcpu, MPIDR_EL1),		vmpidr_el2);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, MAIR_EL2),	SYS_MAIR);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, VBAR_EL2),	SYS_VBAR);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, CONTEXTIDR_EL2),	SYS_CONTEXTIDR);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, AMAIR_EL2),	SYS_AMAIR);



	if (vcpu_el2_e2h_is_set(vcpu)) {

		/*

		 * In VHE mode those registers are compatible between

		 * EL1 and EL2.

		 */

		write_sysreg_el1(__vcpu_sys_reg(vcpu, SCTLR_EL2),   SYS_SCTLR);

		write_sysreg_el1(__vcpu_sys_reg(vcpu, CPTR_EL2),    SYS_CPACR);

		write_sysreg_el1(__vcpu_sys_reg(vcpu, TTBR0_EL2),   SYS_TTBR0);

		write_sysreg_el1(__vcpu_sys_reg(vcpu, TTBR1_EL2),   SYS_TTBR1);

		write_sysreg_el1(__vcpu_sys_reg(vcpu, TCR_EL2),	    SYS_TCR);

		write_sysreg_el1(__vcpu_sys_reg(vcpu, CNTHCTL_EL2), SYS_CNTKCTL);

	} else {

		/*

		 * CNTHCTL_EL2 only affects EL1 when running nVHE, so

		 * no need to restore it.

		 */

		val = translate_sctlr_el2_to_sctlr_el1(__vcpu_sys_reg(vcpu, SCTLR_EL2));

		write_sysreg_el1(val, SYS_SCTLR);

		val = translate_cptr_el2_to_cpacr_el1(__vcpu_sys_reg(vcpu, CPTR_EL2));

		write_sysreg_el1(val, SYS_CPACR);

		val = translate_ttbr0_el2_to_ttbr0_el1(__vcpu_sys_reg(vcpu, TTBR0_EL2));

		write_sysreg_el1(val, SYS_TTBR0);

		val = translate_tcr_el2_to_tcr_el1(__vcpu_sys_reg(vcpu, TCR_EL2));

		write_sysreg_el1(val, SYS_TCR);

	}



	write_sysreg_el1(__vcpu_sys_reg(vcpu, ESR_EL2),		SYS_ESR);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, AFSR0_EL2),	SYS_AFSR0);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, AFSR1_EL2),	SYS_AFSR1);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, FAR_EL2),		SYS_FAR);

	write_sysreg(__vcpu_sys_reg(vcpu, SP_EL2),		sp_el1);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, ELR_EL2),		SYS_ELR);

	write_sysreg_el1(__vcpu_sys_reg(vcpu, SPSR_EL2),	SYS_SPSR);

}



/*

 * VHE: Host and guest must save mdscr_el1 and sp_el0 (and the PC and

 * pstate, which are handled as part of the el2 return state) on every
@@ -66,6 +167,7 @@ void __vcpu_load_switch_sysregs(struct kvm_vcpu *vcpu) 
{

	struct kvm_cpu_context *guest_ctxt = &vcpu->arch.ctxt;

	struct kvm_cpu_context *host_ctxt;

	u64 mpidr;



	host_ctxt = host_data_ptr(host_ctxt);

	__sysreg_save_user_state(host_ctxt);
@@ -89,7 +191,29 @@ void __vcpu_load_switch_sysregs(struct kvm_vcpu *vcpu) 
	 */

	__sysreg32_restore_state(vcpu);

	__sysreg_restore_user_state(guest_ctxt);

	__sysreg_restore_el1_state(guest_ctxt);



	if (unlikely(__is_hyp_ctxt(guest_ctxt))) {

		__sysreg_restore_vel2_state(vcpu);

	} else {

		if (vcpu_has_nv(vcpu)) {

			/*

			 * Use the guest hypervisor's VPIDR_EL2 when in a

			 * nested state. The hardware value of MIDR_EL1 gets

			 * restored on put.

			 */

			write_sysreg(ctxt_sys_reg(guest_ctxt, VPIDR_EL2), vpidr_el2);



			/*

			 * As we're restoring a nested guest, set the value

			 * provided by the guest hypervisor.

			 */

			mpidr = ctxt_sys_reg(guest_ctxt, VMPIDR_EL2);

		} else {

			mpidr = ctxt_sys_reg(guest_ctxt, MPIDR_EL1);

		}



		__sysreg_restore_el1_state(guest_ctxt, mpidr);

	}



	vcpu_set_flag(vcpu, SYSREGS_ON_CPU);

}
@@ -112,12 +236,20 @@ void __vcpu_put_switch_sysregs(struct kvm_vcpu *vcpu) 


	host_ctxt = host_data_ptr(host_ctxt);



	if (unlikely(__is_hyp_ctxt(guest_ctxt)))

		__sysreg_save_vel2_state(vcpu);

	else

		__sysreg_save_el1_state(guest_ctxt);



	__sysreg_save_user_state(guest_ctxt);

	__sysreg32_save_state(vcpu);



	/* Restore host user state */

	__sysreg_restore_user_state(host_ctxt);



	/* If leaving a nesting guest, restore MIDR_EL1 default view */

	if (vcpu_has_nv(vcpu))

		write_sysreg(read_cpuid_id(),	vpidr_el2);



	vcpu_clear_flag(vcpu, SYSREGS_ON_CPU);

}