18c2ecf20Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
28c2ecf20Sopenharmony_ci/*
38c2ecf20Sopenharmony_ci * Copyright (C) 2012-2015 - ARM Ltd
48c2ecf20Sopenharmony_ci * Author: Marc Zyngier <marc.zyngier@arm.com>
58c2ecf20Sopenharmony_ci */
68c2ecf20Sopenharmony_ci
78c2ecf20Sopenharmony_ci#include <hyp/sysreg-sr.h>
88c2ecf20Sopenharmony_ci
98c2ecf20Sopenharmony_ci#include <linux/compiler.h>
108c2ecf20Sopenharmony_ci#include <linux/kvm_host.h>
118c2ecf20Sopenharmony_ci
128c2ecf20Sopenharmony_ci#include <asm/kprobes.h>
138c2ecf20Sopenharmony_ci#include <asm/kvm_asm.h>
148c2ecf20Sopenharmony_ci#include <asm/kvm_emulate.h>
158c2ecf20Sopenharmony_ci#include <asm/kvm_hyp.h>
168c2ecf20Sopenharmony_ci
178c2ecf20Sopenharmony_ci/*
188c2ecf20Sopenharmony_ci * Non-VHE: Both host and guest must save everything.
198c2ecf20Sopenharmony_ci */
208c2ecf20Sopenharmony_ci
218c2ecf20Sopenharmony_civoid __sysreg_save_state_nvhe(struct kvm_cpu_context *ctxt)
228c2ecf20Sopenharmony_ci{
238c2ecf20Sopenharmony_ci	__sysreg_save_el1_state(ctxt);
248c2ecf20Sopenharmony_ci	__sysreg_save_common_state(ctxt);
258c2ecf20Sopenharmony_ci	__sysreg_save_user_state(ctxt);
268c2ecf20Sopenharmony_ci	__sysreg_save_el2_return_state(ctxt);
278c2ecf20Sopenharmony_ci}
288c2ecf20Sopenharmony_ci
298c2ecf20Sopenharmony_civoid __sysreg_restore_state_nvhe(struct kvm_cpu_context *ctxt)
308c2ecf20Sopenharmony_ci{
318c2ecf20Sopenharmony_ci	__sysreg_restore_el1_state(ctxt);
328c2ecf20Sopenharmony_ci	__sysreg_restore_common_state(ctxt);
338c2ecf20Sopenharmony_ci	__sysreg_restore_user_state(ctxt);
348c2ecf20Sopenharmony_ci	__sysreg_restore_el2_return_state(ctxt);
358c2ecf20Sopenharmony_ci}
368c2ecf20Sopenharmony_ci
378c2ecf20Sopenharmony_civoid __kvm_enable_ssbs(void)
388c2ecf20Sopenharmony_ci{
398c2ecf20Sopenharmony_ci	u64 tmp;
408c2ecf20Sopenharmony_ci
418c2ecf20Sopenharmony_ci	asm volatile(
428c2ecf20Sopenharmony_ci	"mrs	%0, sctlr_el2\n"
438c2ecf20Sopenharmony_ci	"orr	%0, %0, %1\n"
448c2ecf20Sopenharmony_ci	"msr	sctlr_el2, %0"
458c2ecf20Sopenharmony_ci	: "=&r" (tmp) : "L" (SCTLR_ELx_DSSBS));
468c2ecf20Sopenharmony_ci}
47