// SPDX-License-Identifier: GPL-2.0-only
/*
 * Guest PC manipulation helpers
 *
 * Copyright (C) 2012,2013 - ARM Ltd
 * Copyright (C) 2020 - Google LLC
 * Author: Marc Zyngier <maz@kernel.org>
 */

#ifndef __ARM64_KVM_HYP_ADJUST_PC_H__
#define __ARM64_KVM_HYP_ADJUST_PC_H__

#include <asm/kvm_emulate.h>
#include <asm/kvm_host.h>

static inline void kvm_skip_instr(struct kvm_vcpu *vcpu)
{
        if (vcpu_mode_is_32bit(vcpu)) {
                kvm_skip_instr32(vcpu);
        } else {
                *vcpu_pc(vcpu) += 4;
                *vcpu_cpsr(vcpu) &= ~PSR_BTYPE_MASK;
        }

        /* advance the singlestep state machine */
        *vcpu_cpsr(vcpu) &= ~DBG_SPSR_SS;
}

/*
 * Skip an instruction which has been emulated at hyp while most guest sysregs
 * are live.
 */
static inline void __kvm_skip_instr(struct kvm_vcpu *vcpu)
{
        *vcpu_pc(vcpu) = read_sysreg_el2(SYS_ELR);
        vcpu_gp_regs(vcpu)->pstate = read_sysreg_el2(SYS_SPSR);

        kvm_skip_instr(vcpu);

        write_sysreg_el2(vcpu_gp_regs(vcpu)->pstate, SYS_SPSR);
        write_sysreg_el2(*vcpu_pc(vcpu), SYS_ELR);
}

/*
 * Skip an instruction while host sysregs are live.
 * Assumes host is always 64-bit.
 */
static inline void kvm_skip_host_instr(void)
{
        write_sysreg_el2(read_sysreg_el2(SYS_ELR) + 4, SYS_ELR);
}

#endif