/* SPDX-License-Identifier: GPL-2.0 */
#if !defined(_TRACE_HANDLE_EXIT_ARM64_KVM_H) || defined(TRACE_HEADER_MULTI_READ)
#define _TRACE_HANDLE_EXIT_ARM64_KVM_H

#include <linux/tracepoint.h>
#include "sys_regs.h"

#undef TRACE_SYSTEM
#define TRACE_SYSTEM kvm

TRACE_EVENT(kvm_wfx_arm64,
        TP_PROTO(unsigned long vcpu_pc, bool is_wfe),
        TP_ARGS(vcpu_pc, is_wfe),

        TP_STRUCT__entry(
                __field(unsigned long,  vcpu_pc)
                __field(bool,           is_wfe)
        ),

        TP_fast_assign(
                __entry->vcpu_pc = vcpu_pc;
                __entry->is_wfe  = is_wfe;
        ),

        TP_printk("guest executed wf%c at: 0x%016lx",
                  __entry->is_wfe ? 'e' : 'i', __entry->vcpu_pc)
);

TRACE_EVENT(kvm_hvc_arm64,
        TP_PROTO(unsigned long vcpu_pc, unsigned long r0, unsigned long imm),
        TP_ARGS(vcpu_pc, r0, imm),

        TP_STRUCT__entry(
                __field(unsigned long, vcpu_pc)
                __field(unsigned long, r0)
                __field(unsigned long, imm)
        ),

        TP_fast_assign(
                __entry->vcpu_pc = vcpu_pc;
                __entry->r0 = r0;
                __entry->imm = imm;
        ),

        TP_printk("HVC at 0x%016lx (r0: 0x%016lx, imm: 0x%lx)",
                  __entry->vcpu_pc, __entry->r0, __entry->imm)
);

/*
 * The dreg32 name is a leftover from a distant past. This will really
 * output a 64bit value...
 */
TRACE_EVENT(kvm_arm_set_dreg32,
        TP_PROTO(const char *name, __u64 value),
        TP_ARGS(name, value),

        TP_STRUCT__entry(
                __field(const char *, name)
                __field(__u64, value)
        ),

        TP_fast_assign(
                __entry->name = name;
                __entry->value = value;
        ),

        TP_printk("%s: 0x%llx", __entry->name, __entry->value)
);

TRACE_EVENT(kvm_handle_sys_reg,
        TP_PROTO(unsigned long hsr),
        TP_ARGS(hsr),

        TP_STRUCT__entry(
                __field(unsigned long,  hsr)
        ),

        TP_fast_assign(
                __entry->hsr = hsr;
        ),

        TP_printk("HSR 0x%08lx", __entry->hsr)
);

TRACE_EVENT(kvm_sys_access,
        TP_PROTO(unsigned long vcpu_pc, struct sys_reg_params *params, const struct sys_reg_desc *reg),
        TP_ARGS(vcpu_pc, params, reg),

        TP_STRUCT__entry(
                __field(unsigned long,                  vcpu_pc)
                __field(bool,                           is_write)
                __field(const char *,                   name)
                __field(u8,                             Op0)
                __field(u8,                             Op1)
                __field(u8,                             CRn)
                __field(u8,                             CRm)
                __field(u8,                             Op2)
        ),

        TP_fast_assign(
                __entry->vcpu_pc = vcpu_pc;
                __entry->is_write = params->is_write;
                __entry->name = reg->name;
                __entry->Op0 = reg->Op0;
                __entry->Op0 = reg->Op0;
                __entry->Op1 = reg->Op1;
                __entry->CRn = reg->CRn;
                __entry->CRm = reg->CRm;
                __entry->Op2 = reg->Op2;
        ),

        TP_printk("PC: %lx %s (%d,%d,%d,%d,%d) %s",
                  __entry->vcpu_pc, __entry->name ?: "UNKN",
                  __entry->Op0, __entry->Op1, __entry->CRn,
                  __entry->CRm, __entry->Op2,
                  str_write_read(__entry->is_write))
);

TRACE_EVENT(kvm_set_guest_debug,
        TP_PROTO(struct kvm_vcpu *vcpu, __u32 guest_debug),
        TP_ARGS(vcpu, guest_debug),

        TP_STRUCT__entry(
                __field(struct kvm_vcpu *, vcpu)
                __field(__u32, guest_debug)
        ),

        TP_fast_assign(
                __entry->vcpu = vcpu;
                __entry->guest_debug = guest_debug;
        ),

        TP_printk("vcpu: %p, flags: 0x%08x", __entry->vcpu, __entry->guest_debug)
);

#endif /* _TRACE_HANDLE_EXIT_ARM64_KVM_H */

#undef TRACE_INCLUDE_PATH
#define TRACE_INCLUDE_PATH .
#undef TRACE_INCLUDE_FILE
#define TRACE_INCLUDE_FILE trace_handle_exit

/* This part must be outside protection */
#include <trace/define_trace.h>