// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2012-2015 - ARM Ltd
 * Author: Marc Zyngier <marc.zyngier@arm.com>
 */

#include <clocksource/arm_arch_timer.h>
#include <linux/compiler.h>
#include <linux/kvm_host.h>

#include <asm/kvm_hyp.h>
#include <asm/kvm_mmu.h>

void __kvm_timer_set_cntvoff(u64 cntvoff)
{
        write_sysreg(cntvoff, cntvoff_el2);
}

/*
 * Should only be called on non-VHE or hVHE setups.
 * VHE systems use EL2 timers and configure EL1 timers in kvm_timer_init_vhe().
 */
void __timer_disable_traps(struct kvm_vcpu *vcpu)
{
        u64 set, clr, shift = 0;

        if (has_hvhe())
                shift = 10;

        /* Allow physical timer/counter access for the host */
        set = (CNTHCTL_EL1PCTEN | CNTHCTL_EL1PCEN) << shift;
        clr = CNTHCTL_EL1TVT | CNTHCTL_EL1TVCT;

        sysreg_clear_set(cnthctl_el2, clr, set);
}

/*
 * Should only be called on non-VHE or hVHE setups.
 * VHE systems use EL2 timers and configure EL1 timers in kvm_timer_init_vhe().
 */
void __timer_enable_traps(struct kvm_vcpu *vcpu)
{
        u64 clr = 0, set = 0;

        /*
         * Disallow physical timer access for the guest
         * Physical counter access is allowed if no offset is enforced
         * or running protected (we don't offset anything in this case).
         */
        clr = CNTHCTL_EL1PCEN;
        if (is_protected_kvm_enabled() ||
            !kern_hyp_va(vcpu->kvm)->arch.timer_data.poffset)
                set |= CNTHCTL_EL1PCTEN;
        else
                clr |= CNTHCTL_EL1PCTEN;

        if (has_hvhe()) {
                clr <<= 10;
                set <<= 10;
        }

        /*
         * Trap the virtual counter/timer if we have a broken cntvoff
         * implementation.
         */
        if (has_broken_cntvoff())
                set |= CNTHCTL_EL1TVT | CNTHCTL_EL1TVCT;

        sysreg_clear_set(cnthctl_el2, clr, set);
}