/* SPDX-License-Identifier: GPL-2.0 */
#undef TRACE_SYSTEM
#define TRACE_SYSTEM lock

#if !defined(_TRACE_LOCK_H) || defined(TRACE_HEADER_MULTI_READ)
#define _TRACE_LOCK_H

#include <linux/sched.h>
#include <linux/tracepoint.h>

/* flags for lock:contention_begin */
#define LCB_F_SPIN      (1U << 0)
#define LCB_F_READ      (1U << 1)
#define LCB_F_WRITE     (1U << 2)
#define LCB_F_RT        (1U << 3)
#define LCB_F_PERCPU    (1U << 4)
#define LCB_F_MUTEX     (1U << 5)


#ifdef CONFIG_LOCKDEP

#include <linux/lockdep.h>

TRACE_EVENT(lock_acquire,

        TP_PROTO(struct lockdep_map *lock, unsigned int subclass,
                int trylock, int read, int check,
                struct lockdep_map *next_lock, unsigned long ip),

        TP_ARGS(lock, subclass, trylock, read, check, next_lock, ip),

        TP_STRUCT__entry(
                __field(unsigned int, flags)
                __string(name, lock->name)
                __field(void *, lockdep_addr)
        ),

        TP_fast_assign(
                __entry->flags = (trylock ? 1 : 0) | (read ? 2 : 0);
                __assign_str(name);
                __entry->lockdep_addr = lock;
        ),

        TP_printk("%p %s%s%s", __entry->lockdep_addr,
                  (__entry->flags & 1) ? "try " : "",
                  (__entry->flags & 2) ? "read " : "",
                  __get_str(name))
);

DECLARE_EVENT_CLASS(lock,

        TP_PROTO(struct lockdep_map *lock, unsigned long ip),

        TP_ARGS(lock, ip),

        TP_STRUCT__entry(
                __string(       name,   lock->name      )
                __field(        void *, lockdep_addr    )
        ),

        TP_fast_assign(
                __assign_str(name);
                __entry->lockdep_addr = lock;
        ),

        TP_printk("%p %s",  __entry->lockdep_addr, __get_str(name))
);

DEFINE_EVENT(lock, lock_release,

        TP_PROTO(struct lockdep_map *lock, unsigned long ip),

        TP_ARGS(lock, ip)
);

#ifdef CONFIG_LOCK_STAT

DEFINE_EVENT(lock, lock_contended,

        TP_PROTO(struct lockdep_map *lock, unsigned long ip),

        TP_ARGS(lock, ip)
);

DEFINE_EVENT(lock, lock_acquired,

        TP_PROTO(struct lockdep_map *lock, unsigned long ip),

        TP_ARGS(lock, ip)
);

#endif /* CONFIG_LOCK_STAT */
#endif /* CONFIG_LOCKDEP */

TRACE_EVENT(contention_begin,

        TP_PROTO(void *lock, unsigned int flags),

        TP_ARGS(lock, flags),

        TP_STRUCT__entry(
                __field(void *, lock_addr)
                __field(unsigned int, flags)
        ),

        TP_fast_assign(
                __entry->lock_addr = lock;
                __entry->flags = flags;
        ),

        TP_printk("%p (flags=%s)", __entry->lock_addr,
                  __print_flags(__entry->flags, "|",
                                { LCB_F_SPIN,           "SPIN" },
                                { LCB_F_READ,           "READ" },
                                { LCB_F_WRITE,          "WRITE" },
                                { LCB_F_RT,             "RT" },
                                { LCB_F_PERCPU,         "PERCPU" },
                                { LCB_F_MUTEX,          "MUTEX" }
                          ))
);

TRACE_EVENT(contention_end,

        TP_PROTO(void *lock, int ret),

        TP_ARGS(lock, ret),

        TP_STRUCT__entry(
                __field(void *, lock_addr)
                __field(int, ret)
        ),

        TP_fast_assign(
                __entry->lock_addr = lock;
                __entry->ret = ret;
        ),

        TP_printk("%p (ret=%d)", __entry->lock_addr, __entry->ret)
);

#endif /* _TRACE_LOCK_H */

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