/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * Copyright (c) 2026, Microsoft Corporation.
 *
 * Tracepoint declarations for mshv driver.
 */

#undef TRACE_SYSTEM
#define TRACE_SYSTEM mshv

#if !defined(__MSHV_TRACE_H) || defined(TRACE_HEADER_MULTI_READ)
#define _MSHV_TRACE_H_

#include <linux/tracepoint.h>
#include <hyperv/hvhdk.h>

#undef TRACE_INCLUDE_PATH
#define TRACE_INCLUDE_PATH ../../drivers/hv

#undef TRACE_INCLUDE_FILE
#define TRACE_INCLUDE_FILE mshv_trace

TRACE_EVENT(mshv_create_partition,
	    TP_PROTO(u64 partition_id, int vm_fd),
	    TP_ARGS(partition_id, vm_fd),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(int, vm_fd)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vm_fd = vm_fd;
	    ),
	    TP_printk("partition_id=%llu vm_fd=%d",
		    __entry->partition_id,
		    __entry->vm_fd
	    )
);

TRACE_EVENT(mshv_hvcall_create_partition,
	    TP_PROTO(u64 flags, s64 partition_id),
	    TP_ARGS(flags, partition_id),
	    TP_STRUCT__entry(
		    __field(u64, flags)
		    __field(s64, partition_id)
	    ),
	    TP_fast_assign(
		    __entry->flags = flags;
		    __entry->partition_id = partition_id;
	    ),
	    TP_printk("flags=%#llx partition_id=%lld",
		    __entry->flags,
		    __entry->partition_id
	    )
);

TRACE_EVENT(mshv_hvcall_initialize_partition,
	    TP_PROTO(u64 partition_id, u64 status),
	    TP_ARGS(partition_id, status),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u64, status)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->status = status;
	    ),
	    TP_printk("partition_id=%llu status=%#llx",
		    __entry->partition_id,
		    __entry->status
	    )
);

TRACE_EVENT(mshv_partition_release,
	    TP_PROTO(u64 partition_id),
	    TP_ARGS(partition_id),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
	    ),
	    TP_printk("partition_id=%llu",
		    __entry->partition_id
	    )
);

TRACE_EVENT(mshv_destroy_partition,
	    TP_PROTO(u64 partition_id),
	    TP_ARGS(partition_id),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
	    ),
	    TP_printk("partition_id=%llu",
		    __entry->partition_id
	    )
);

TRACE_EVENT(mshv_hvcall_finalize_partition,
	    TP_PROTO(u64 partition_id, u64 status),
	    TP_ARGS(partition_id, status),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u64, status)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->status = status;
	    ),
	    TP_printk("partition_id=%llu status=%#llx ",
		    __entry->partition_id,
		    __entry->status
	    )
);

TRACE_EVENT(mshv_hvcall_withdraw_memory,
	    TP_PROTO(u64 partition_id, u64 withdrawn, u64 status),
	    TP_ARGS(partition_id, withdrawn, status),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u64, withdrawn)
		    __field(u64, status)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->withdrawn = withdrawn;
		    __entry->status = status;
	    ),
	    TP_printk("partition_id=%llu withdrawn=%llu status=%#llx",
		    __entry->partition_id,
		    __entry->withdrawn,
		    __entry->status
	    )
);

TRACE_EVENT(mshv_hvcall_delete_partition,
	    TP_PROTO(u64 partition_id, u64 status),
	    TP_ARGS(partition_id, status),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u64, status)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->status = status;
	    ),
	    TP_printk("partition_id=%llu status=%#llx",
		    __entry->partition_id,
		    __entry->status
	    )
);

TRACE_EVENT(mshv_create_vp,
	    TP_PROTO(u64 partition_id, u32 vp_index, long vp_fd),
	    TP_ARGS(partition_id, vp_index, vp_fd),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(long, vp_fd)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->vp_fd = vp_fd;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u vp_fd=%ld",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->vp_fd
	    )
);

TRACE_EVENT(mshv_hvcall_map_vp_state_page,
	    TP_PROTO(u64 partition_id, u32 vp_index, u32 page_type, u64 status),
	    TP_ARGS(partition_id, vp_index, page_type, status),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(u32, page_type)
		    __field(u64, status)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->page_type = page_type;
		    __entry->status = status;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u page_type=%u status=%#llx",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->page_type,
		    __entry->status
	    )
);

TRACE_EVENT(mshv_drain_vp_signals,
	    TP_PROTO(u64 partition_id, u32 vp_index),
	    TP_ARGS(partition_id, vp_index),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u",
		    __entry->partition_id,
		    __entry->vp_index
	    )
);

TRACE_EVENT(mshv_disable_vp_dispatch,
	    TP_PROTO(u64 partition_id, u32 vp_index, int ret),
	    TP_ARGS(partition_id, vp_index, ret),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(int, ret)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->ret = ret;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u ret=%d",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->ret
	    )
);

TRACE_EVENT(mshv_vp_release,
	    TP_PROTO(u64 partition_id, u32 vp_index),
	    TP_ARGS(partition_id, vp_index),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u",
		    __entry->partition_id,
		    __entry->vp_index
	    )
);

TRACE_EVENT(mshv_run_vp_entry,
	    TP_PROTO(u64 partition_id, u32 vp_index),
	    TP_ARGS(partition_id, vp_index),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u",
		    __entry->partition_id,
		    __entry->vp_index
	    )
);

TRACE_EVENT(mshv_run_vp_exit,
	    TP_PROTO(u64 partition_id, u32 vp_index, u64 hv_message_type, long ret),
	    TP_ARGS(partition_id, vp_index, hv_message_type, ret),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(u64, hv_message_type)
		    __field(long, ret)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->hv_message_type = hv_message_type;
		    __entry->ret = ret;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u hv_message_type=%#llx ret=%ld",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->hv_message_type,
		    __entry->ret
	    )
);

TRACE_EVENT(mshv_vp_clear_explicit_suspend,
	    TP_PROTO(u64 partition_id, u32 vp_index, int ret),
	    TP_ARGS(partition_id, vp_index, ret),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(int, ret)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->ret = ret;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u ret=%d",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->ret
	    )
);

TRACE_EVENT(mshv_xfer_to_guest_mode_work,
	    TP_PROTO(u64 partition_id, u32 vp_index, unsigned long thread_info_flag, long ret),
	    TP_ARGS(partition_id, vp_index, thread_info_flag, ret),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(unsigned long, thread_info_flag)
		    __field(long, ret)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->thread_info_flag = thread_info_flag;
		    __entry->ret = ret;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u thread_info_flag=%#lx ret=%ld",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->thread_info_flag,
		    __entry->ret
	    )
);

TRACE_EVENT(mshv_hvcall_dispatch_vp,
	    TP_PROTO(u64 partition_id, u32 vp_index, u32 flags,
		     u32 dispatch_state, u32 dispatch_event, u64 irq_vectors, u64 status),
	    TP_ARGS(partition_id, vp_index, flags, dispatch_state, dispatch_event, irq_vectors,
		    status),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(u32, flags)
		    __field(u32, dispatch_state)
		    __field(u32, dispatch_event)
		    __field(u64, irq_vectors)
		    __field(u64, status)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->flags = flags;
		    __entry->dispatch_state = dispatch_state;
		    __entry->dispatch_event = dispatch_event;
		    __entry->irq_vectors = irq_vectors;
		    __entry->status = status;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u flags=%#x dispatch_state=%#x dispatch_event=%#x irq_vectors=%#016llx status=%#llx",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->flags,
		    __entry->dispatch_state,
		    __entry->dispatch_event,
		    __entry->irq_vectors,
		    __entry->status
	     )
);

TRACE_EVENT(mshv_update_routing_table,
	    TP_PROTO(u64 partition_id, void *old, void *new, u32 numents),
	    TP_ARGS(partition_id, old, new, numents),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(struct mshv_girq_routing_table *, old)
		    __field(struct mshv_girq_routing_table *, new)
		    __field(u32, numents)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->old = old;
		    __entry->new = new;
		    __entry->numents = numents;
	    ),
	    TP_printk("partition_id=%llu old=%p new=%p numents=%u",
		    __entry->partition_id,
		    __entry->old,
		    __entry->new,
		    __entry->numents
	    )
);

TRACE_EVENT(mshv_map_user_memory,
	    TP_PROTO(u64 partition_id, u64 start_uaddr, u64 start_gfn, u64 nr_pages, u32 map_flags,
		     long ret),
	    TP_ARGS(partition_id, start_uaddr, start_gfn, nr_pages, map_flags, ret),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u64, start_uaddr)
		    __field(u64, start_gfn)
		    __field(u64, nr_pages)
		    __field(u32, map_flags)
		    __field(long, ret)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->start_uaddr = start_uaddr;
		    __entry->start_gfn = start_gfn;
		    __entry->nr_pages = nr_pages;
		    __entry->map_flags = map_flags;
		    __entry->ret = ret;
	    ),
	    TP_printk("partition_id=%llu start_uaddr=%#llx start_gfn=%#llx nr_pages=%llu map_flags=%#x ret=%ld",
		    __entry->partition_id,
		    __entry->start_uaddr,
		    __entry->start_gfn,
		    __entry->nr_pages,
		    __entry->map_flags,
		    __entry->ret
	     )
);

TRACE_EVENT(mshv_assign_ioeventfd,
	    TP_PROTO(u64 partition_id, u64 addr, u64 length, u64 datamatch, bool wildcard,
		     void *eventfd, int ret),
	    TP_ARGS(partition_id, addr, length, datamatch, wildcard, eventfd, ret),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u64, addr)
		    __field(u64, length)
		    __field(u64, datamatch)
		    __field(bool, wildcard)
		    __field(struct eventfd_ctx *, eventfd)
		    __field(int, ret)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->addr = addr;
		    __entry->length = length;
		    __entry->datamatch = datamatch;
		    __entry->wildcard = wildcard;
		    __entry->eventfd = eventfd;
		    __entry->ret = ret;
	    ),
	    TP_printk("partition_id=%llu addr=%#016llx length=%#llx datamatch=%#llx wildcard=%d eventfd=%p ret=%d",
		    __entry->partition_id,
		    __entry->addr,
		    __entry->length,
		    __entry->datamatch,
		    __entry->wildcard,
		    __entry->eventfd,
		    __entry->ret
	     )
);

TRACE_EVENT(mshv_deassign_ioeventfd,
	    TP_PROTO(u64 partition_id, u64 addr, u64 length, u64 datamatch, bool wildcard,
		     void *eventfd),
	    TP_ARGS(partition_id, addr, length, datamatch, wildcard, eventfd),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u64, addr)
		    __field(u64, length)
		    __field(u64, datamatch)
		    __field(bool, wildcard)
		    __field(struct eventfd_ctx *, eventfd)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->addr = addr;
		    __entry->length = length;
		    __entry->datamatch = datamatch;
		    __entry->wildcard = wildcard;
		    __entry->eventfd = eventfd;
	    ),
	    TP_printk("partition_id=%llu addr=%#016llx length=%#llx datamatch=%#llx wildcard=%d eventfd=%p",
		    __entry->partition_id,
		    __entry->addr,
		    __entry->length,
		    __entry->datamatch,
		    __entry->wildcard,
		    __entry->eventfd
	     )
);

TRACE_EVENT(mshv_vp_wait_for_hv_kick,
	    TP_PROTO(u64 partition_id, u32 vp_index, bool kicked_by_hv, bool blocked,
		     bool irq_pending),
	    TP_ARGS(partition_id, vp_index, kicked_by_hv, blocked, irq_pending),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(bool, kicked_by_hv)
		    __field(bool, blocked)
		    __field(bool, irq_pending)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->kicked_by_hv = kicked_by_hv;
		    __entry->blocked = blocked;
		    __entry->irq_pending = irq_pending;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u kicked_by_hv=%d blocked=%d irq_pending=%d",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->kicked_by_hv,
		    __entry->blocked,
		    __entry->irq_pending
	    )
);

TRACE_EVENT(mshv_handle_gpa_intercept,
	    TP_PROTO(u64 partition_id, u32 vp_index, u64 gfn, u8 access_type, bool handled),
	    TP_ARGS(partition_id, vp_index, gfn, access_type, handled),
	    TP_STRUCT__entry(
		    __field(u64, partition_id)
		    __field(u32, vp_index)
		    __field(u64, gfn)
		    __field(u8, access_type)
		    __field(bool, handled)
	    ),
	    TP_fast_assign(
		    __entry->partition_id = partition_id;
		    __entry->vp_index = vp_index;
		    __entry->gfn = gfn;
		    __entry->access_type = access_type == HV_INTERCEPT_ACCESS_READ ? 'R' :
					   (access_type == HV_INTERCEPT_ACCESS_WRITE ? 'W' :
					    (access_type == HV_INTERCEPT_ACCESS_EXECUTE ? 'X' : '?'));
		    __entry->handled = handled;
	    ),
	    TP_printk("partition_id=%llu vp_index=%u gfn=0x%llx access_type=%c handled=%d",
		    __entry->partition_id,
		    __entry->vp_index,
		    __entry->gfn,
		    __entry->access_type,
		    __entry->handled
	    )
);

#endif /* _MSHV_TRACE_H_ */

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