/* SPDX-License-Identifier: GPL-2.0 */
#ifndef _ASM_S390_FTRACE_H
#define _ASM_S390_FTRACE_H

#define ARCH_SUPPORTS_FTRACE_OPS 1
#define MCOUNT_INSN_SIZE        6

#ifndef __ASSEMBLER__
#include <asm/stacktrace.h>

static __always_inline unsigned long return_address(unsigned int n)
{
        struct stack_frame *sf;

        if (!n)
                return (unsigned long)__builtin_return_address(0);

        sf = (struct stack_frame *)current_frame_address();
        do {
                sf = (struct stack_frame *)sf->back_chain;
                if (!sf)
                        return 0;
        } while (--n);
        return sf->gprs[8];
}
#define ftrace_return_address(n) return_address(n)

void ftrace_caller(void);

extern void *ftrace_func;

struct dyn_arch_ftrace { };

#define MCOUNT_ADDR 0
#define FTRACE_ADDR ((unsigned long)ftrace_caller)

#define KPROBE_ON_FTRACE_NOP    0
#define KPROBE_ON_FTRACE_CALL   1

struct module;
struct dyn_ftrace;
struct ftrace_ops;

bool ftrace_need_init_nop(void);
#define ftrace_need_init_nop ftrace_need_init_nop

int ftrace_init_nop(struct module *mod, struct dyn_ftrace *rec);
#define ftrace_init_nop ftrace_init_nop

static inline unsigned long ftrace_call_adjust(unsigned long addr)
{
        return addr;
}
#define ftrace_get_symaddr(fentry_ip) ((unsigned long)(fentry_ip))

#include <linux/ftrace_regs.h>

static __always_inline struct pt_regs *arch_ftrace_get_regs(struct ftrace_regs *fregs)
{
        struct pt_regs *regs = &arch_ftrace_regs(fregs)->regs;

        if (test_pt_regs_flag(regs, PIF_FTRACE_FULL_REGS))
                return regs;
        return NULL;
}

static __always_inline void
ftrace_regs_set_instruction_pointer(struct ftrace_regs *fregs,
                                    unsigned long ip)
{
        arch_ftrace_regs(fregs)->regs.psw.addr = ip;
}

#undef ftrace_regs_get_frame_pointer
static __always_inline unsigned long
ftrace_regs_get_frame_pointer(struct ftrace_regs *fregs)
{
        return ftrace_regs_get_stack_pointer(fregs);
}

static __always_inline unsigned long
ftrace_regs_get_return_address(const struct ftrace_regs *fregs)
{
        return arch_ftrace_regs(fregs)->regs.gprs[14];
}

#define arch_ftrace_fill_perf_regs(fregs, _regs)         do {           \
                (_regs)->psw.mask = 0;                                  \
                (_regs)->psw.addr = arch_ftrace_regs(fregs)->regs.psw.addr;             \
                (_regs)->gprs[15] = arch_ftrace_regs(fregs)->regs.gprs[15];             \
        } while (0)

#ifdef CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS
/*
 * When an ftrace registered caller is tracing a function that is
 * also set by a register_ftrace_direct() call, it needs to be
 * differentiated in the ftrace_caller trampoline. To do this,
 * place the direct caller in the ORIG_GPR2 part of pt_regs. This
 * tells the ftrace_caller that there's a direct caller.
 */
static inline void arch_ftrace_set_direct_caller(struct ftrace_regs *fregs, unsigned long addr)
{
        struct pt_regs *regs = &arch_ftrace_regs(fregs)->regs;
        regs->orig_gpr2 = addr;
}
#endif /* CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS */

#define ARCH_HAS_SYSCALL_MATCH_SYM_NAME
static inline bool arch_syscall_match_sym_name(const char *sym,
                                               const char *name)
{
        /* Skip the __s390x_ prefix. */
        return !strcmp(sym + 7, name) || !strcmp(sym + 8, name);
}

void ftrace_graph_func(unsigned long ip, unsigned long parent_ip,
                       struct ftrace_ops *op, struct ftrace_regs *fregs);
#define ftrace_graph_func ftrace_graph_func

#endif /* __ASSEMBLER__ */

#ifdef CONFIG_FUNCTION_TRACER

#define FTRACE_NOP_INSN .word 0xc004, 0x0000, 0x0000 /* brcl 0,0 */

#ifndef CC_USING_HOTPATCH

#define FTRACE_GEN_MCOUNT_RECORD(name)          \
        .section __mcount_loc, "a", @progbits;  \
        .quad name;                             \
        .previous;

#else /* !CC_USING_HOTPATCH */

#define FTRACE_GEN_MCOUNT_RECORD(name)

#endif /* !CC_USING_HOTPATCH */

#define FTRACE_GEN_NOP_ASM(name)                \
        FTRACE_GEN_MCOUNT_RECORD(name)          \
        FTRACE_NOP_INSN

#else /* CONFIG_FUNCTION_TRACER */

#define FTRACE_GEN_NOP_ASM(name)

#endif /* CONFIG_FUNCTION_TRACER */

#endif /* _ASM_S390_FTRACE_H */