/* SPDX-License-Identifier: GPL-2.0 */
#undef TRACE_SYSTEM
#define TRACE_SYSTEM ufshcd
#define TRACE_INCLUDE_PATH trace/hooks
#if !defined(_TRACE_HOOK_UFSHCD_H) || defined(TRACE_HEADER_MULTI_READ)
#define _TRACE_HOOK_UFSHCD_H
#include <trace/hooks/vendor_hooks.h>
/*
 * Following tracepoints are not exported in tracefs and provide a
 * mechanism for vendor modules to hook and extend functionality
 */
#ifdef __GENKSYMS__
#include <../drivers/scsi/ufs/ufshcd.h>
#include <linux/blkdev.h>
#include <scsi/scsi_device.h>
#endif

struct ufs_hba;
struct ufshcd_lrb;
struct uic_command;
struct request;
struct scsi_device;

DECLARE_HOOK(android_vh_ufs_fill_prdt,
	TP_PROTO(struct ufs_hba *hba, struct ufshcd_lrb *lrbp,
		 unsigned int segments, int *err),
	TP_ARGS(hba, lrbp, segments, err));

DECLARE_RESTRICTED_HOOK(android_rvh_ufs_complete_init,
			TP_PROTO(struct ufs_hba *hba),
			TP_ARGS(hba), 1);

DECLARE_RESTRICTED_HOOK(android_rvh_ufs_reprogram_all_keys,
			TP_PROTO(struct ufs_hba *hba, int *err),
			TP_ARGS(hba, err), 1);

DECLARE_HOOK(android_vh_ufs_prepare_command,
	TP_PROTO(struct ufs_hba *hba, struct request *rq,
		 struct ufshcd_lrb *lrbp, int *err),
	TP_ARGS(hba, rq, lrbp, err));

DECLARE_HOOK(android_vh_ufs_update_sysfs,
	TP_PROTO(struct ufs_hba *hba),
	TP_ARGS(hba));

DECLARE_HOOK(android_vh_ufs_send_command,
	TP_PROTO(struct ufs_hba *hba, struct ufshcd_lrb *lrbp),
	TP_ARGS(hba, lrbp));

DECLARE_HOOK(android_vh_ufs_compl_command,
	TP_PROTO(struct ufs_hba *hba, struct ufshcd_lrb *lrbp),
	TP_ARGS(hba, lrbp));

DECLARE_HOOK(android_vh_ufs_send_uic_command,
	TP_PROTO(struct ufs_hba *hba, struct uic_command *ucmd, int str_t),
	TP_ARGS(hba, ucmd, str_t));

DECLARE_HOOK(android_vh_ufs_send_tm_command,
	TP_PROTO(struct ufs_hba *hba, int tag, int str_t),
	TP_ARGS(hba, tag, str_t));

DECLARE_HOOK(android_vh_ufs_check_int_errors,
	TP_PROTO(struct ufs_hba *hba, bool queue_eh_work),
	TP_ARGS(hba, queue_eh_work));

DECLARE_HOOK(android_vh_ufs_update_sdev,
	TP_PROTO(struct scsi_device *sdev),
	TP_ARGS(sdev));

DECLARE_HOOK(android_vh_ufs_clock_scaling,
		TP_PROTO(struct ufs_hba *hba, bool *force_out, bool *force_scaling, bool *scale_up),
		TP_ARGS(hba, force_out, force_scaling, scale_up));

DECLARE_HOOK(android_vh_ufs_use_mcq_hooks,
	TP_PROTO(struct ufs_hba *hba, bool *use_mcq),
	TP_ARGS(hba, use_mcq));

DECLARE_HOOK(android_vh_ufs_mcq_max_tag,
	TP_PROTO(struct ufs_hba *hba, int *max_tag),
	TP_ARGS(hba, max_tag));

DECLARE_HOOK(android_vh_ufs_mcq_map_tag,
	TP_PROTO(struct ufs_hba *hba, int index, int *tag),
	TP_ARGS(hba, index, tag));

DECLARE_HOOK(android_vh_ufs_mcq_set_sqid,
	TP_PROTO(struct ufs_hba *hba, int index, struct ufshcd_lrb *lrbp),
	TP_ARGS(hba, index, lrbp));

DECLARE_HOOK(android_vh_ufs_mcq_handler,
	TP_PROTO(struct ufs_hba *hba, u32 intr_status, irqreturn_t *retval),
	TP_ARGS(hba, intr_status, retval));

DECLARE_HOOK(android_vh_ufs_mcq_make_hba_operational,
	TP_PROTO(struct ufs_hba *hba, int *err),
	TP_ARGS(hba, err));

DECLARE_HOOK(android_vh_ufs_mcq_hba_capabilities,
	TP_PROTO(struct ufs_hba *hba, int *err),
	TP_ARGS(hba, err));

DECLARE_HOOK(android_vh_ufs_mcq_print_trs,
	TP_PROTO(struct ufs_hba *hba, bool pr_prdt),
	TP_ARGS(hba, pr_prdt));

DECLARE_HOOK(android_vh_ufs_mcq_send_command,
	TP_PROTO(struct ufs_hba *hba, unsigned int task_tag),
	TP_ARGS(hba, task_tag));

DECLARE_HOOK(android_vh_ufs_mcq_config,
	TP_PROTO(struct ufs_hba *hba, int *err),
	TP_ARGS(hba, err));

DECLARE_HOOK(android_vh_ufs_mcq_has_oustanding_reqs,
	TP_PROTO(struct ufs_hba *hba, bool *ret),
	TP_ARGS(hba, ret));

DECLARE_HOOK(android_vh_ufs_mcq_get_outstanding_reqs,
	TP_PROTO(struct ufs_hba *hba, unsigned long **outstanding, int *nr_tag),
	TP_ARGS(hba, outstanding, nr_tag));

struct scsi_cmnd;
DECLARE_HOOK(android_vh_ufs_mcq_abort,
	TP_PROTO(struct scsi_cmnd *cmd, int *ret),
	TP_ARGS(cmd, ret));

DECLARE_HOOK(android_vh_ufs_mcq_clear_cmd,
	TP_PROTO(struct ufs_hba *hba, int tag, int *ret),
	TP_ARGS(hba, tag, ret));

DECLARE_HOOK(android_vh_ufs_mcq_clear_pending,
	TP_PROTO(struct ufs_hba *hba, int *ret),
	TP_ARGS(hba, ret));

DECLARE_HOOK(android_vh_ufs_mcq_retry_complete,
	TP_PROTO(struct ufs_hba *hba),
	TP_ARGS(hba));

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