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
32 #define ftrace_init_nop ftrace_init_nop
33 int ftrace_init_nop(struct module *mod, struct dyn_ftrace *rec);
34
ftrace_call_adjust(unsigned long addr)35 static inline unsigned long ftrace_call_adjust(unsigned long addr)
36 {
37 return addr;
38 }
39
40 void prepare_ftrace_return(unsigned long self_addr, unsigned long *parent);
41
42 #endif /* CONFIG_DYNAMIC_FTRACE */
43
44 #ifdef CONFIG_HAVE_DYNAMIC_FTRACE_WITH_ARGS
45 struct ftrace_ops;
46
47 struct ftrace_regs {
48 struct pt_regs regs;
49 };
50
arch_ftrace_get_regs(struct ftrace_regs * fregs)51 static __always_inline struct pt_regs *arch_ftrace_get_regs(struct ftrace_regs *fregs)
52 {
53 return &fregs->regs;
54 }
55
56 static __always_inline unsigned long
ftrace_regs_get_instruction_pointer(struct ftrace_regs * fregs)57 ftrace_regs_get_instruction_pointer(struct ftrace_regs *fregs)
58 {
59 return instruction_pointer(&fregs->regs);
60 }
61
62 static __always_inline void
ftrace_regs_set_instruction_pointer(struct ftrace_regs * fregs,unsigned long ip)63 ftrace_regs_set_instruction_pointer(struct ftrace_regs *fregs, unsigned long ip)
64 {
65 instruction_pointer_set(&fregs->regs, ip);
66 }
67
68 #define ftrace_regs_get_argument(fregs, n) \
69 regs_get_kernel_argument(&(fregs)->regs, n)
70 #define ftrace_regs_get_stack_pointer(fregs) \
71 kernel_stack_pointer(&(fregs)->regs)
72 #define ftrace_regs_return_value(fregs) \
73 regs_return_value(&(fregs)->regs)
74 #define ftrace_regs_set_return_value(fregs, ret) \
75 regs_set_return_value(&(fregs)->regs, ret)
76 #define ftrace_override_function_with_return(fregs) \
77 override_function_with_return(&(fregs)->regs)
78 #define ftrace_regs_query_register_offset(name) \
79 regs_query_register_offset(name)
80
81 #define ftrace_graph_func ftrace_graph_func
82 void ftrace_graph_func(unsigned long ip, unsigned long parent_ip,
83 struct ftrace_ops *op, struct ftrace_regs *fregs);
84
85 #ifdef CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS
86 static inline void
__arch_ftrace_set_direct_caller(struct pt_regs * regs,unsigned long addr)87 __arch_ftrace_set_direct_caller(struct pt_regs *regs, unsigned long addr)
88 {
89 regs->regs[13] = addr; /* t1 */
90 }
91
92 #define arch_ftrace_set_direct_caller(fregs, addr) \
93 __arch_ftrace_set_direct_caller(&(fregs)->regs, addr)
94 #endif /* CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS */
95
96 #endif
97
98 #endif /* __ASSEMBLY__ */
99
100 #endif /* CONFIG_FUNCTION_TRACER */
101
102 #ifndef __ASSEMBLY__
103 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
104 struct fgraph_ret_regs {
105 /* a0 - a1 */
106 unsigned long regs[2];
107
108 unsigned long fp;
109 unsigned long __unused;
110 };
111
fgraph_ret_regs_return_value(struct fgraph_ret_regs * ret_regs)112 static inline unsigned long fgraph_ret_regs_return_value(struct fgraph_ret_regs *ret_regs)
113 {
114 return ret_regs->regs[0];
115 }
116
fgraph_ret_regs_frame_pointer(struct fgraph_ret_regs * ret_regs)117 static inline unsigned long fgraph_ret_regs_frame_pointer(struct fgraph_ret_regs *ret_regs)
118 {
119 return ret_regs->fp;
120 }
121 #endif /* ifdef CONFIG_FUNCTION_GRAPH_TRACER */
122 #endif
123
124 #endif /* _ASM_LOONGARCH_FTRACE_H */
125