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

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

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

#ifdef CONFIG_PPC_BOOK3S_64
DECLARE_EVENT_CLASS(hugepage_set,

            TP_PROTO(unsigned long addr, unsigned long pte),
            TP_ARGS(addr, pte),
            TP_STRUCT__entry(
                    __field(unsigned long, addr)
                    __field(unsigned long, pte)
                    ),

            TP_fast_assign(
                    __entry->addr = addr;
                    __entry->pte = pte;
                    ),

            TP_printk("Set page table entry with 0x%lx with 0x%lx", __entry->addr, __entry->pte)
);

DEFINE_EVENT(hugepage_set, hugepage_set_pmd,
            TP_PROTO(unsigned long addr, unsigned long pmd),
            TP_ARGS(addr, pmd)
);

DEFINE_EVENT(hugepage_set, hugepage_set_pud,
            TP_PROTO(unsigned long addr, unsigned long pud),
            TP_ARGS(addr, pud)
);

DECLARE_EVENT_CLASS(hugepage_update,

            TP_PROTO(unsigned long addr, unsigned long pte, unsigned long clr, unsigned long set),
            TP_ARGS(addr, pte, clr, set),
            TP_STRUCT__entry(
                    __field(unsigned long, addr)
                    __field(unsigned long, pte)
                    __field(unsigned long, clr)
                    __field(unsigned long, set)
                    ),

            TP_fast_assign(
                    __entry->addr = addr;
                    __entry->pte = pte;
                    __entry->clr = clr;
                    __entry->set = set;

                    ),

            TP_printk("hugepage update at addr 0x%lx and pte = 0x%lx clr = 0x%lx, set = 0x%lx", __entry->addr, __entry->pte, __entry->clr, __entry->set)
);

DEFINE_EVENT(hugepage_update, hugepage_update_pmd,
            TP_PROTO(unsigned long addr, unsigned long pmd, unsigned long clr, unsigned long set),
            TP_ARGS(addr, pmd, clr, set)
);

DEFINE_EVENT(hugepage_update, hugepage_update_pud,
            TP_PROTO(unsigned long addr, unsigned long pud, unsigned long clr, unsigned long set),
            TP_ARGS(addr, pud, clr, set)
);
#endif /* CONFIG_PPC_BOOK3S_64 */

DECLARE_EVENT_CLASS(migration_pmd,

                TP_PROTO(unsigned long addr, unsigned long pmd),

                TP_ARGS(addr, pmd),

                TP_STRUCT__entry(
                        __field(unsigned long, addr)
                        __field(unsigned long, pmd)
                ),

                TP_fast_assign(
                        __entry->addr = addr;
                        __entry->pmd = pmd;
                ),
                TP_printk("addr=%lx, pmd=%lx", __entry->addr, __entry->pmd)
);

DEFINE_EVENT(migration_pmd, set_migration_pmd,
        TP_PROTO(unsigned long addr, unsigned long pmd),
        TP_ARGS(addr, pmd)
);

DEFINE_EVENT(migration_pmd, remove_migration_pmd,
        TP_PROTO(unsigned long addr, unsigned long pmd),
        TP_ARGS(addr, pmd)
);
#endif /* _TRACE_THP_H */

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