arch/loongarch/include/asm/ftrace.h

   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 */