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 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 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 57 ftrace_regs_get_instruction_pointer(struct ftrace_regs *fregs) 58 { 59 return instruction_pointer(&fregs->regs); 60 } 61 62 static __always_inline void 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 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 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 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