1// SPDX-License-Identifier: GPL-2.0
2// Copyright (C) 2018 Hangzhou C-SKY Microsystems co.,ltd.
3
4#include <linux/ftrace.h>
5#include <linux/uaccess.h>
6#include <linux/stop_machine.h>
7#include <asm/cacheflush.h>
8
9#ifdef CONFIG_DYNAMIC_FTRACE
10
11#define NOP 0x4000
12#define NOP32_HI 0xc400
13#define NOP32_LO 0x4820
14#define PUSH_LR 0x14d0
15#define MOVIH_LINK 0xea3a
16#define ORI_LINK 0xef5a
17#define JSR_LINK 0xe8fa
18#define BSR_LINK 0xe000
19
20/*
21 * Gcc-csky with -pg will insert stub in function prologue:
22 * push lr
23 * jbsr _mcount
24 * nop32
25 * nop32
26 *
27 * If the (callee - current_pc) is less then 64MB, we'll use bsr:
28 * push lr
29 * bsr _mcount
30 * nop32
31 * nop32
32 * else we'll use (movih + ori + jsr):
33 * push lr
34 * movih r26, ...
35 * ori r26, ...
36 * jsr r26
37 *
38 * (r26 is our reserved link-reg)
39 *
40 */
41static inline void make_jbsr(unsigned long callee, unsigned long pc,
42 uint16_t *call, bool nolr)
43{
44 long offset;
45
46 call[0] = nolr ? NOP : PUSH_LR;
47
48 offset = (long) callee - (long) pc;
49
50 if (unlikely(offset < -67108864 || offset > 67108864)) {
51 call[1] = MOVIH_LINK;
52 call[2] = callee >> 16;
53 call[3] = ORI_LINK;
54 call[4] = callee & 0xffff;
55 call[5] = JSR_LINK;
56 call[6] = 0;
57 } else {
58 offset = offset >> 1;
59
60 call[1] = BSR_LINK |
61 ((uint16_t)((unsigned long) offset >> 16) & 0x3ff);
62 call[2] = (uint16_t)((unsigned long) offset & 0xffff);
63 call[3] = call[5] = NOP32_HI;
64 call[4] = call[6] = NOP32_LO;
65 }
66}
67
68static uint16_t nops[7] = {NOP, NOP32_HI, NOP32_LO, NOP32_HI, NOP32_LO,
69 NOP32_HI, NOP32_LO};
70static int ftrace_check_current_nop(unsigned long hook)
71{
72 uint16_t olds[7];
73 unsigned long hook_pos = hook - 2;
74
75 if (copy_from_kernel_nofault(dst: (void *)olds, src: (void *)hook_pos,
76 size: sizeof(nops)))
77 return -EFAULT;
78
79 if (memcmp(p: (void *)nops, q: (void *)olds, size: sizeof(nops))) {
80 pr_err("%p: nop but get (%04x %04x %04x %04x %04x %04x %04x)\n",
81 (void *)hook_pos,
82 olds[0], olds[1], olds[2], olds[3], olds[4], olds[5],
83 olds[6]);
84
85 return -EINVAL;
86 }
87
88 return 0;
89}
90
91static int ftrace_modify_code(unsigned long hook, unsigned long target,
92 bool enable, bool nolr)
93{
94 uint16_t call[7];
95
96 unsigned long hook_pos = hook - 2;
97 int ret = 0;
98
99 make_jbsr(callee: target, pc: hook, call, nolr);
100
101 ret = copy_to_kernel_nofault(dst: (void *)hook_pos, src: enable ? call : nops,
102 size: sizeof(nops));
103 if (ret)
104 return -EPERM;
105
106 flush_icache_range(start: hook_pos, end: hook_pos + MCOUNT_INSN_SIZE);
107
108 return 0;
109}
110
111int ftrace_make_call(struct dyn_ftrace *rec, unsigned long addr)
112{
113 int ret = ftrace_check_current_nop(hook: rec->ip);
114
115 if (ret)
116 return ret;
117
118 return ftrace_modify_code(hook: rec->ip, target: addr, enable: true, nolr: false);
119}
120
121int ftrace_make_nop(struct module *mod, struct dyn_ftrace *rec,
122 unsigned long addr)
123{
124 return ftrace_modify_code(hook: rec->ip, target: addr, enable: false, nolr: false);
125}
126
127int ftrace_update_ftrace_func(ftrace_func_t func)
128{
129 int ret = ftrace_modify_code(hook: (unsigned long)&ftrace_call,
130 target: (unsigned long)func, enable: true, nolr: true);
131 if (!ret)
132 ret = ftrace_modify_code(hook: (unsigned long)&ftrace_regs_call,
133 target: (unsigned long)func, enable: true, nolr: true);
134 return ret;
135}
136#endif /* CONFIG_DYNAMIC_FTRACE */
137
138#ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
139int ftrace_modify_call(struct dyn_ftrace *rec, unsigned long old_addr,
140 unsigned long addr)
141{
142 return ftrace_modify_code(hook: rec->ip, target: addr, enable: true, nolr: true);
143}
144#endif
145
146#ifdef CONFIG_FUNCTION_GRAPH_TRACER
147void prepare_ftrace_return(unsigned long *parent, unsigned long self_addr,
148 unsigned long frame_pointer)
149{
150 unsigned long return_hooker = (unsigned long)&return_to_handler;
151 unsigned long old;
152
153 if (unlikely(atomic_read(&current->tracing_graph_pause)))
154 return;
155
156 old = *parent;
157
158 if (!function_graph_enter(ret: old, func: self_addr,
159 frame_pointer: *(unsigned long *)frame_pointer, retp: parent)) {
160 /*
161 * For csky-gcc function has sub-call:
162 * subi sp, sp, 8
163 * stw r8, (sp, 0)
164 * mov r8, sp
165 * st.w r15, (sp, 0x4)
166 * push r15
167 * jl _mcount
168 * We only need set *parent for resume
169 *
170 * For csky-gcc function has no sub-call:
171 * subi sp, sp, 4
172 * stw r8, (sp, 0)
173 * mov r8, sp
174 * push r15
175 * jl _mcount
176 * We need set *parent and *(frame_pointer + 4) for resume,
177 * because lr is resumed twice.
178 */
179 *parent = return_hooker;
180 frame_pointer += 4;
181 if (*(unsigned long *)frame_pointer == old)
182 *(unsigned long *)frame_pointer = return_hooker;
183 }
184}
185
186#ifdef CONFIG_DYNAMIC_FTRACE
187int ftrace_enable_ftrace_graph_caller(void)
188{
189 return ftrace_modify_code(hook: (unsigned long)&ftrace_graph_call,
190 target: (unsigned long)&ftrace_graph_caller, enable: true, nolr: true);
191}
192
193int ftrace_disable_ftrace_graph_caller(void)
194{
195 return ftrace_modify_code(hook: (unsigned long)&ftrace_graph_call,
196 target: (unsigned long)&ftrace_graph_caller, enable: false, nolr: true);
197}
198#endif /* CONFIG_DYNAMIC_FTRACE */
199#endif /* CONFIG_FUNCTION_GRAPH_TRACER */
200
201#ifdef CONFIG_DYNAMIC_FTRACE
202#ifndef CONFIG_CPU_HAS_ICACHE_INS
203struct ftrace_modify_param {
204 int command;
205 atomic_t cpu_count;
206};
207
208static int __ftrace_modify_code(void *data)
209{
210 struct ftrace_modify_param *param = data;
211
212 if (atomic_inc_return(v: &param->cpu_count) == 1) {
213 ftrace_modify_all_code(command: param->command);
214 atomic_inc(v: &param->cpu_count);
215 } else {
216 while (atomic_read(v: &param->cpu_count) <= num_online_cpus())
217 cpu_relax();
218 local_icache_inv_all(NULL);
219 }
220
221 return 0;
222}
223
224void arch_ftrace_update_code(int command)
225{
226 struct ftrace_modify_param param = { command, ATOMIC_INIT(0) };
227
228 stop_machine(fn: __ftrace_modify_code, data: &param, cpu_online_mask);
229}
230#endif
231#endif /* CONFIG_DYNAMIC_FTRACE */
232
233/* _mcount is defined in abi's mcount.S */
234EXPORT_SYMBOL(_mcount);
235

source code of linux/arch/csky/kernel/ftrace.c