include/trace/hooks/dtask.h

/* SPDX-License-Identifier: GPL-2.0 */
#undef TRACE_SYSTEM
#define TRACE_SYSTEM dtask
#define TRACE_INCLUDE_PATH trace/hooks

#if !defined(_TRACE_HOOK_DTASK_H) || defined(TRACE_HEADER_MULTI_READ)
#define _TRACE_HOOK_DTASK_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
 */

struct mutex;
struct rt_mutex;
struct rt_mutex_base;
struct rw_semaphore;
struct task_struct;
struct percpu_rw_semaphore;
struct rq;

DECLARE_HOOK(android_vh_mutex_wait_start,
	TP_PROTO(struct mutex *lock),
	TP_ARGS(lock));
DECLARE_HOOK(android_vh_mutex_wait_finish,
	TP_PROTO(struct mutex *lock),
	TP_ARGS(lock));
DECLARE_HOOK(android_vh_mutex_init,
	TP_PROTO(struct mutex *lock),
	TP_ARGS(lock));
DECLARE_HOOK(android_vh_mutex_opt_spin_start,
	TP_PROTO(struct mutex *lock, bool *time_out, int *cnt),
	TP_ARGS(lock, time_out, cnt));
DECLARE_HOOK(android_vh_mutex_opt_spin_finish,
	TP_PROTO(struct mutex *lock, bool taken),
	TP_ARGS(lock, taken));
DECLARE_HOOK(android_vh_mutex_can_spin_on_owner,
	TP_PROTO(struct mutex *lock, int *retval),
	TP_ARGS(lock, retval));

DECLARE_HOOK(android_vh_rtmutex_wait_start,
	TP_PROTO(struct rt_mutex_base *lock),
	TP_ARGS(lock));
DECLARE_HOOK(android_vh_rtmutex_wait_finish,
	TP_PROTO(struct rt_mutex_base *lock),
	TP_ARGS(lock));

DECLARE_HOOK(android_vh_rwsem_read_wait_start,
	TP_PROTO(struct rw_semaphore *sem),
	TP_ARGS(sem));
DECLARE_HOOK(android_vh_rwsem_read_wait_finish,
	TP_PROTO(struct rw_semaphore *sem),
	TP_ARGS(sem));
DECLARE_HOOK(android_vh_rwsem_write_wait_start,
	TP_PROTO(struct rw_semaphore *sem),
	TP_ARGS(sem));
DECLARE_HOOK(android_vh_rwsem_write_wait_finish,
	TP_PROTO(struct rw_semaphore *sem),
	TP_ARGS(sem));
DECLARE_HOOK(android_vh_rwsem_opt_spin_start,
	TP_PROTO(struct rw_semaphore *sem, bool *time_out, int *cnt, bool chk_only),
	TP_ARGS(sem, time_out, cnt, chk_only));
DECLARE_HOOK(android_vh_rwsem_opt_spin_finish,
	TP_PROTO(struct rw_semaphore *sem, bool taken),
	TP_ARGS(sem, taken));
DECLARE_HOOK(android_vh_rwsem_can_spin_on_owner,
	TP_PROTO(struct rw_semaphore *sem, bool *ret),
	TP_ARGS(sem, ret));

struct device;
DECLARE_HOOK(android_vh_dpm_wait_start,
	TP_PROTO(struct device *dev),
	TP_ARGS(dev));
DECLARE_HOOK(android_vh_dpm_wait_finish,
	TP_PROTO(struct device *dev),
	TP_ARGS(dev));

struct irq_desc;
DECLARE_HOOK(android_vh_sync_irq_wait_start,
	TP_PROTO(struct irq_desc *desc),
	TP_ARGS(desc));
DECLARE_HOOK(android_vh_sync_irq_wait_finish,
	TP_PROTO(struct irq_desc *desc),
	TP_ARGS(desc));

struct workqueue_struct;
DECLARE_HOOK(android_vh_flush_wq_wait_start,
	TP_PROTO(struct workqueue_struct *wq),
	TP_ARGS(wq));
DECLARE_HOOK(android_vh_flush_wq_wait_finish,
	TP_PROTO(struct workqueue_struct *wq),
	TP_ARGS(wq));

struct work_struct;
DECLARE_HOOK(android_vh_flush_work_wait_start,
	TP_PROTO(struct work_struct *work),
	TP_ARGS(work));
DECLARE_HOOK(android_vh_flush_work_wait_finish,
	TP_PROTO(struct work_struct *work),
	TP_ARGS(work));

DECLARE_HOOK(android_vh_sched_show_task,
	TP_PROTO(struct task_struct *task),
	TP_ARGS(task));
DECLARE_HOOK(android_vh_percpu_rwsem_wq_add,
	TP_PROTO(struct percpu_rw_semaphore *sem, bool reader),
	TP_ARGS(sem, reader));
DECLARE_HOOK(android_vh_percpu_rwsem_down_read,
	TP_PROTO(struct percpu_rw_semaphore *sem, bool try, bool *ret),
	TP_ARGS(sem, try, ret));
DECLARE_HOOK(android_vh_percpu_rwsem_up_write,
	TP_PROTO(struct percpu_rw_semaphore *sem),
	TP_ARGS(sem));
DECLARE_RESTRICTED_HOOK(android_rvh_percpu_rwsem_wait_complete,
	TP_PROTO(struct percpu_rw_semaphore *sem, long state, bool *complete),
	TP_ARGS(sem, state, complete), 1);
DECLARE_HOOK(android_vh_percpu_rwsem_init,
	TP_PROTO(struct percpu_rw_semaphore *sem),
	TP_ARGS(sem));

struct mutex_waiter;
DECLARE_HOOK(android_vh_alter_mutex_list_add,
	TP_PROTO(struct mutex *lock,
		struct mutex_waiter *waiter,
		struct list_head *list,
		bool *already_on_list),
	TP_ARGS(lock, waiter, list, already_on_list));
DECLARE_HOOK(android_vh_mutex_unlock_slowpath,
	TP_PROTO(struct mutex *lock),
	TP_ARGS(lock));
DECLARE_HOOK(android_vh_mutex_unlock_slowpath_before_wakeq,
	TP_PROTO(struct mutex *lock),
	TP_ARGS(lock));

DECLARE_HOOK(android_vh_record_mutex_lock_starttime,
	TP_PROTO(struct mutex *lock, unsigned long settime_jiffies),
	TP_ARGS(lock, settime_jiffies));
DECLARE_HOOK(android_vh_record_rtmutex_lock_starttime,
	TP_PROTO(struct rt_mutex *lock, unsigned long settime_jiffies),
	TP_ARGS(lock, settime_jiffies));
DECLARE_HOOK(android_vh_record_rwsem_lock_starttime,
	TP_PROTO(struct rw_semaphore *sem, unsigned long settime_jiffies),
	TP_ARGS(sem, settime_jiffies));
DECLARE_HOOK(android_vh_record_pcpu_rwsem_starttime,
	TP_PROTO(struct percpu_rw_semaphore *sem, unsigned long settime_jiffies),
	TP_ARGS(sem, settime_jiffies));

DECLARE_HOOK(android_vh_exit_check,
	TP_PROTO(struct task_struct *p),
	TP_ARGS(p));

DECLARE_RESTRICTED_HOOK(android_rvh_dpm_prepare,
	TP_PROTO(int flag),
	TP_ARGS(flag), 1);

DECLARE_HOOK(android_vh_set_tsk_need_resched_lazy,
	TP_PROTO(struct task_struct *p, struct rq *rq, int *need_lazy),
	TP_ARGS(p, rq, need_lazy));

DECLARE_HOOK(android_vh_resched_curr_lazy,
	TP_PROTO(struct rq *rq, bool *skip_preempt),
	TP_ARGS(rq, skip_preempt));

DECLARE_HOOK(android_vh_restore_curr_resched,
	TP_PROTO(unsigned long *flags, int *lazy_flage),
	TP_ARGS(flags, lazy_flage));

DECLARE_HOOK(android_vh_clear_curr_lazy,
	TP_PROTO(struct task_struct *tsk),
	TP_ARGS(tsk));
#endif /* _TRACE_HOOK_DTASK_H */

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