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 static __always_inline unsigned long 58 ftrace_regs_get_instruction_pointer(struct ftrace_regs *fregs) 59 { 60 return instruction_pointer(&fregs->regs); 61 } 62 63 static __always_inline void 64 ftrace_regs_set_instruction_pointer(struct ftrace_regs *fregs, unsigned long ip) 65 { 66 regs_set_return_value(&fregs->regs, ip); 67 } 68 69 #define ftrace_regs_get_argument(fregs, n) \ 70 regs_get_kernel_argument(&(fregs)->regs, n) 71 #define ftrace_regs_get_stack_pointer(fregs) \ 72 kernel_stack_pointer(&(fregs)->regs) 73 #define ftrace_regs_return_value(fregs) \ 74 regs_return_value(&(fregs)->regs) 75 #define ftrace_regs_set_return_value(fregs, ret) \ 76 regs_set_return_value(&(fregs)->regs, ret) 77 #define ftrace_override_function_with_return(fregs) \ 78 override_function_with_return(&(fregs)->regs) 79 #define ftrace_regs_query_register_offset(name) \ 80 regs_query_register_offset(name) 81 82 #define ftrace_graph_func ftrace_graph_func 83 void ftrace_graph_func(unsigned long ip, unsigned long parent_ip, 84 struct ftrace_ops *op, struct ftrace_regs *fregs); 85 86 #ifdef CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS 87 static inline void 88 __arch_ftrace_set_direct_caller(struct pt_regs *regs, unsigned long addr) 89 { 90 regs->regs[13] = addr; /* t1 */ 91 } 92 93 #define arch_ftrace_set_direct_caller(fregs, addr) \ 94 __arch_ftrace_set_direct_caller(&(fregs)->regs, addr) 95 #endif /* CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS */ 96 97 #endif 98 99 #endif /* __ASSEMBLY__ */ 100 101 #endif /* CONFIG_FUNCTION_TRACER */ 102 103 #ifndef __ASSEMBLY__ 104 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 105 struct fgraph_ret_regs { 106 /* a0 - a1 */ 107 unsigned long regs[2]; 108 109 unsigned long fp; 110 unsigned long __unused; 111 }; 112 113 static inline unsigned long fgraph_ret_regs_return_value(struct fgraph_ret_regs *ret_regs) 114 { 115 return ret_regs->regs[0]; 116 } 117 118 static inline unsigned long fgraph_ret_regs_frame_pointer(struct fgraph_ret_regs *ret_regs) 119 { 120 return ret_regs->fp; 121 } 122 #endif /* ifdef CONFIG_FUNCTION_GRAPH_TRACER */ 123 #endif 124 125 #endif /* _ASM_LOONGARCH_FTRACE_H */ 126