| // SPDX-License-Identifier: GPL-2.0 | 
 | #include <linux/spinlock.h> | 
 | #include <linux/hardirq.h> | 
 | #include <linux/ftrace.h> | 
 | #include <linux/percpu.h> | 
 | #include <linux/init.h> | 
 | #include <linux/list.h> | 
 | #include <trace/syscall.h> | 
 |  | 
 | #include <asm/ftrace.h> | 
 |  | 
 | #ifdef CONFIG_DYNAMIC_FTRACE | 
 | static const u32 ftrace_nop = 0x01000000; | 
 |  | 
 | static u32 ftrace_call_replace(unsigned long ip, unsigned long addr) | 
 | { | 
 | 	u32 call; | 
 | 	s32 off; | 
 |  | 
 | 	off = ((s32)addr - (s32)ip); | 
 | 	call = 0x40000000 | ((u32)off >> 2); | 
 |  | 
 | 	return call; | 
 | } | 
 |  | 
 | static int ftrace_modify_code(unsigned long ip, u32 old, u32 new) | 
 | { | 
 | 	u32 replaced; | 
 | 	int faulted; | 
 |  | 
 | 	__asm__ __volatile__( | 
 | 	"1:	cas	[%[ip]], %[old], %[new]\n" | 
 | 	"	flush	%[ip]\n" | 
 | 	"	mov	0, %[faulted]\n" | 
 | 	"2:\n" | 
 | 	"	.section .fixup,#alloc,#execinstr\n" | 
 | 	"	.align	4\n" | 
 | 	"3:	sethi	%%hi(2b), %[faulted]\n" | 
 | 	"	jmpl	%[faulted] + %%lo(2b), %%g0\n" | 
 | 	"	 mov	1, %[faulted]\n" | 
 | 	"	.previous\n" | 
 | 	"	.section __ex_table,\"a\"\n" | 
 | 	"	.align	4\n" | 
 | 	"	.word	1b, 3b\n" | 
 | 	"	.previous\n" | 
 | 	: "=r" (replaced), [faulted] "=r" (faulted) | 
 | 	: [new] "0" (new), [old] "r" (old), [ip] "r" (ip) | 
 | 	: "memory"); | 
 |  | 
 | 	if (replaced != old && replaced != new) | 
 | 		faulted = 2; | 
 |  | 
 | 	return faulted; | 
 | } | 
 |  | 
 | int ftrace_make_nop(struct module *mod, struct dyn_ftrace *rec, unsigned long addr) | 
 | { | 
 | 	unsigned long ip = rec->ip; | 
 | 	u32 old, new; | 
 |  | 
 | 	old = ftrace_call_replace(ip, addr); | 
 | 	new = ftrace_nop; | 
 | 	return ftrace_modify_code(ip, old, new); | 
 | } | 
 |  | 
 | int ftrace_make_call(struct dyn_ftrace *rec, unsigned long addr) | 
 | { | 
 | 	unsigned long ip = rec->ip; | 
 | 	u32 old, new; | 
 |  | 
 | 	old = ftrace_nop; | 
 | 	new = ftrace_call_replace(ip, addr); | 
 | 	return ftrace_modify_code(ip, old, new); | 
 | } | 
 |  | 
 | int ftrace_update_ftrace_func(ftrace_func_t func) | 
 | { | 
 | 	unsigned long ip = (unsigned long)(&ftrace_call); | 
 | 	u32 old, new; | 
 |  | 
 | 	old = *(u32 *) &ftrace_call; | 
 | 	new = ftrace_call_replace(ip, (unsigned long)func); | 
 | 	return ftrace_modify_code(ip, old, new); | 
 | } | 
 |  | 
 | int __init ftrace_dyn_arch_init(void) | 
 | { | 
 | 	return 0; | 
 | } | 
 | #endif | 
 |  | 
 | #ifdef CONFIG_FUNCTION_GRAPH_TRACER | 
 |  | 
 | #ifdef CONFIG_DYNAMIC_FTRACE | 
 | extern void ftrace_graph_call(void); | 
 |  | 
 | int ftrace_enable_ftrace_graph_caller(void) | 
 | { | 
 | 	unsigned long ip = (unsigned long)(&ftrace_graph_call); | 
 | 	u32 old, new; | 
 |  | 
 | 	old = *(u32 *) &ftrace_graph_call; | 
 | 	new = ftrace_call_replace(ip, (unsigned long) &ftrace_graph_caller); | 
 | 	return ftrace_modify_code(ip, old, new); | 
 | } | 
 |  | 
 | int ftrace_disable_ftrace_graph_caller(void) | 
 | { | 
 | 	unsigned long ip = (unsigned long)(&ftrace_graph_call); | 
 | 	u32 old, new; | 
 |  | 
 | 	old = *(u32 *) &ftrace_graph_call; | 
 | 	new = ftrace_call_replace(ip, (unsigned long) &ftrace_stub); | 
 |  | 
 | 	return ftrace_modify_code(ip, old, new); | 
 | } | 
 |  | 
 | #endif /* !CONFIG_DYNAMIC_FTRACE */ | 
 |  | 
 | /* | 
 |  * Hook the return address and push it in the stack of return addrs | 
 |  * in current thread info. | 
 |  */ | 
 | unsigned long prepare_ftrace_return(unsigned long parent, | 
 | 				    unsigned long self_addr, | 
 | 				    unsigned long frame_pointer) | 
 | { | 
 | 	unsigned long return_hooker = (unsigned long) &return_to_handler; | 
 | 	struct ftrace_graph_ent trace; | 
 |  | 
 | 	if (unlikely(atomic_read(¤t->tracing_graph_pause))) | 
 | 		return parent + 8UL; | 
 |  | 
 | 	trace.func = self_addr; | 
 | 	trace.depth = current->curr_ret_stack + 1; | 
 |  | 
 | 	/* Only trace if the calling function expects to */ | 
 | 	if (!ftrace_graph_entry(&trace)) | 
 | 		return parent + 8UL; | 
 |  | 
 | 	if (ftrace_push_return_trace(parent, self_addr, &trace.depth, | 
 | 				     frame_pointer, NULL) == -EBUSY) | 
 | 		return parent + 8UL; | 
 |  | 
 | 	return return_hooker; | 
 | } | 
 | #endif /* CONFIG_FUNCTION_GRAPH_TRACER */ |