1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3  * Copyright (C) 2022 Loongson Technology Corporation Limited
4  */
5 
6 #ifndef _ASM_LOONGARCH_FTRACE_H
7 #define _ASM_LOONGARCH_FTRACE_H
8 
9 #define FTRACE_PLT_IDX		0
10 #define FTRACE_REGS_PLT_IDX	1
11 #define NR_FTRACE_PLTS		2
12 
13 #ifdef CONFIG_FUNCTION_TRACER
14 
15 #define MCOUNT_INSN_SIZE 4		/* sizeof mcount call */
16 
17 #ifndef __ASSEMBLY__
18 
19 #ifndef CONFIG_DYNAMIC_FTRACE
20 
21 #define mcount _mcount
22 extern void _mcount(void);
23 extern void prepare_ftrace_return(unsigned long self_addr, unsigned long callsite_sp, unsigned long old);
24 
25 #else
26 
27 struct dyn_ftrace;
28 struct dyn_arch_ftrace { };
29 
30 #define ARCH_SUPPORTS_FTRACE_OPS 1
31 #define HAVE_FUNCTION_GRAPH_RET_ADDR_PTR
32 
33 #define ftrace_init_nop ftrace_init_nop
34 int ftrace_init_nop(struct module *mod, struct dyn_ftrace *rec);
35 
36 static inline unsigned long ftrace_call_adjust(unsigned long addr)
37 {
38 	return addr;
39 }
40 
41 void prepare_ftrace_return(unsigned long self_addr, unsigned long *parent);
42 
43 #endif /* CONFIG_DYNAMIC_FTRACE */
44 
45 #ifdef CONFIG_HAVE_DYNAMIC_FTRACE_WITH_ARGS
46 struct ftrace_ops;
47 
48 struct ftrace_regs {
49 	struct pt_regs regs;
50 };
51 
52 static __always_inline struct pt_regs *arch_ftrace_get_regs(struct ftrace_regs *fregs)
53 {
54 	return &fregs->regs;
55 }
56 
57 #define ftrace_graph_func ftrace_graph_func
58 void ftrace_graph_func(unsigned long ip, unsigned long parent_ip,
59 		       struct ftrace_ops *op, struct ftrace_regs *fregs);
60 #endif
61 
62 #endif /* __ASSEMBLY__ */
63 
64 #endif /* CONFIG_FUNCTION_TRACER */
65 
66 #endif /* _ASM_LOONGARCH_FTRACE_H */
67