arch/mips/kernel/ftrace.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Code for replacing ftrace calls with jumps.
   4  *
   5  * Copyright (C) 2007-2008 Steven Rostedt <srostedt@redhat.com>
   6  * Copyright (C) 2009, 2010 DSLab, Lanzhou University, China
   7  * Author: Wu Zhangjin <wuzhangjin@gmail.com>
   8  *
   9  * Thanks goes to Steven Rostedt for writing the original x86 version.
  10  */
  11
  12 #include <linux/uaccess.h>
  13 #include <linux/init.h>
  14 #include <linux/ftrace.h>
  15 #include <linux/syscalls.h>
  16
  17 #include <asm/asm.h>
  18 #include <asm/asm-offsets.h>
  19 #include <asm/cacheflush.h>
  20 #include <asm/syscall.h>
  21 #include <asm/uasm.h>
  22 #include <asm/unistd.h>
  23
  24 #include <asm-generic/sections.h>
  25
  26 #if defined(KBUILD_MCOUNT_RA_ADDRESS) && defined(CONFIG_32BIT)
  27 #define MCOUNT_OFFSET_INSNS 5
  28 #else
  29 #define MCOUNT_OFFSET_INSNS 4
  30 #endif
  31
  32 #ifdef CONFIG_DYNAMIC_FTRACE
  33
  34 /* Arch override because MIPS doesn't need to run this from stop_machine() */
  35 void arch_ftrace_update_code(int command)
  36 {
  37         ftrace_modify_all_code(command);
  38 }
  39
  40 #endif
  41
  42 #ifdef CONFIG_DYNAMIC_FTRACE
  43
  44 #define JAL 0x0c000000          /* jump & link: ip --> ra, jump to target */
  45 #define ADDR_MASK 0x03ffffff    /*  op_code|addr : 31...26|25 ....0 */
  46 #define JUMP_RANGE_MASK ((1UL << 28) - 1)
  47
  48 #define INSN_NOP 0x00000000     /* nop */
  49 #define INSN_JAL(addr)  \
  50         ((unsigned int)(JAL | (((addr) >> 2) & ADDR_MASK)))
  51
  52 static unsigned int insn_jal_ftrace_caller __read_mostly;
  53 static unsigned int insn_la_mcount[2] __read_mostly;
  54 static unsigned int insn_j_ftrace_graph_caller __maybe_unused __read_mostly;
  55
  56 static inline void ftrace_dyn_arch_init_insns(void)
  57 {
  58         u32 *buf;
  59         unsigned int v1;
  60
  61         /* la v1, _mcount */
  62         v1 = 3;
  63         buf = (u32 *)&insn_la_mcount[0];
  64         UASM_i_LA(&buf, v1, MCOUNT_ADDR);
  65
  66         /* jal (ftrace_caller + 8), jump over the first two instruction */
  67         buf = (u32 *)&insn_jal_ftrace_caller;
  68         uasm_i_jal(&buf, (FTRACE_ADDR + 8) & JUMP_RANGE_MASK);
  69
  70 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
  71         /* j ftrace_graph_caller */
  72         buf = (u32 *)&insn_j_ftrace_graph_caller;
  73         uasm_i_j(&buf, (unsigned long)ftrace_graph_caller & JUMP_RANGE_MASK);
  74 #endif
  75 }
  76
  77 static int ftrace_modify_code(unsigned long ip, unsigned int new_code)
  78 {
  79         int faulted;
  80         mm_segment_t old_fs;
  81
  82         /* *(unsigned int *)ip = new_code; */
  83         safe_store_code(new_code, ip, faulted);
  84
  85         if (unlikely(faulted))
  86                 return -EFAULT;
  87
  88         old_fs = get_fs();
  89         set_fs(get_ds());
  90         flush_icache_range(ip, ip + 8);
  91         set_fs(old_fs);
  92
  93         return 0;
  94 }
  95
  96 #ifndef CONFIG_64BIT
  97 static int ftrace_modify_code_2(unsigned long ip, unsigned int new_code1,
  98                                 unsigned int new_code2)
  99 {
 100         int faulted;
 101         mm_segment_t old_fs;
 102
 103         safe_store_code(new_code1, ip, faulted);
 104         if (unlikely(faulted))
 105                 return -EFAULT;
 106
 107         ip += 4;
 108         safe_store_code(new_code2, ip, faulted);
 109         if (unlikely(faulted))
 110                 return -EFAULT;
 111
 112         ip -= 4;
 113         old_fs = get_fs();
 114         set_fs(get_ds());
 115         flush_icache_range(ip, ip + 8);
 116         set_fs(old_fs);
 117
 118         return 0;
 119 }
 120
 121 static int ftrace_modify_code_2r(unsigned long ip, unsigned int new_code1,
 122                                  unsigned int new_code2)
 123 {
 124         int faulted;
 125         mm_segment_t old_fs;
 126
 127         ip += 4;
 128         safe_store_code(new_code2, ip, faulted);
 129         if (unlikely(faulted))
 130                 return -EFAULT;
 131
 132         ip -= 4;
 133         safe_store_code(new_code1, ip, faulted);
 134         if (unlikely(faulted))
 135                 return -EFAULT;
 136
 137         old_fs = get_fs();
 138         set_fs(get_ds());
 139         flush_icache_range(ip, ip + 8);
 140         set_fs(old_fs);
 141
 142         return 0;
 143 }
 144 #endif
 145
 146 /*
 147  * The details about the calling site of mcount on MIPS
 148  *
 149  * 1. For kernel:
 150  *
 151  * move at, ra
 152  * jal _mcount          --> nop
 153  *  sub sp, sp, 8       --> nop  (CONFIG_32BIT)
 154  *
 155  * 2. For modules:
 156  *
 157  * 2.1 For KBUILD_MCOUNT_RA_ADDRESS and CONFIG_32BIT
 158  *
 159  * lui v1, hi_16bit_of_mcount        --> b 1f (0x10000005)
 160  * addiu v1, v1, low_16bit_of_mcount --> nop  (CONFIG_32BIT)
 161  * move at, ra
 162  * move $12, ra_address
 163  * jalr v1
 164  *  sub sp, sp, 8
 165  *                                  1: offset = 5 instructions
 166  * 2.2 For the Other situations
 167  *
 168  * lui v1, hi_16bit_of_mcount        --> b 1f (0x10000004)
 169  * addiu v1, v1, low_16bit_of_mcount --> nop  (CONFIG_32BIT)
 170  * move at, ra
 171  * jalr v1
 172  *  nop | move $12, ra_address | sub sp, sp, 8
 173  *                                  1: offset = 4 instructions
 174  */
 175
 176 #define INSN_B_1F (0x10000000 | MCOUNT_OFFSET_INSNS)
 177
 178 int ftrace_make_nop(struct module *mod,
 179                     struct dyn_ftrace *rec, unsigned long addr)
 180 {
 181         unsigned int new;
 182         unsigned long ip = rec->ip;
 183
 184         /*
 185          * If ip is in kernel space, no long call, otherwise, long call is
 186          * needed.
 187          */
 188         new = core_kernel_text(ip) ? INSN_NOP : INSN_B_1F;
 189 #ifdef CONFIG_64BIT
 190         return ftrace_modify_code(ip, new);
 191 #else
 192         /*
 193          * On 32 bit MIPS platforms, gcc adds a stack adjust
 194          * instruction in the delay slot after the branch to
 195          * mcount and expects mcount to restore the sp on return.
 196          * This is based on a legacy API and does nothing but
 197          * waste instructions so it's being removed at runtime.
 198          */
 199         return ftrace_modify_code_2(ip, new, INSN_NOP);
 200 #endif
 201 }
 202
 203 int ftrace_make_call(struct dyn_ftrace *rec, unsigned long addr)
 204 {
 205         unsigned int new;
 206         unsigned long ip = rec->ip;
 207
 208         new = core_kernel_text(ip) ? insn_jal_ftrace_caller : insn_la_mcount[0];
 209
 210 #ifdef CONFIG_64BIT
 211         return ftrace_modify_code(ip, new);
 212 #else
 213         return ftrace_modify_code_2r(ip, new, core_kernel_text(ip) ?
 214                                                 INSN_NOP : insn_la_mcount[1]);
 215 #endif
 216 }
 217
 218 #define FTRACE_CALL_IP ((unsigned long)(&ftrace_call))
 219
 220 int ftrace_update_ftrace_func(ftrace_func_t func)
 221 {
 222         unsigned int new;
 223
 224         new = INSN_JAL((unsigned long)func);
 225
 226         return ftrace_modify_code(FTRACE_CALL_IP, new);
 227 }
 228
 229 int __init ftrace_dyn_arch_init(void)
 230 {
 231         /* Encode the instructions when booting */
 232         ftrace_dyn_arch_init_insns();
 233
 234         /* Remove "b ftrace_stub" to ensure ftrace_caller() is executed */
 235         ftrace_modify_code(MCOUNT_ADDR, INSN_NOP);
 236
 237         return 0;
 238 }
 239 #endif  /* CONFIG_DYNAMIC_FTRACE */
 240
 241 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
 242
 243 #ifdef CONFIG_DYNAMIC_FTRACE
 244
 245 extern void ftrace_graph_call(void);
 246 #define FTRACE_GRAPH_CALL_IP    ((unsigned long)(&ftrace_graph_call))
 247
 248 int ftrace_enable_ftrace_graph_caller(void)
 249 {
 250         return ftrace_modify_code(FTRACE_GRAPH_CALL_IP,
 251                         insn_j_ftrace_graph_caller);
 252 }
 253
 254 int ftrace_disable_ftrace_graph_caller(void)
 255 {
 256         return ftrace_modify_code(FTRACE_GRAPH_CALL_IP, INSN_NOP);
 257 }
 258
 259 #endif  /* CONFIG_DYNAMIC_FTRACE */
 260
 261 #ifndef KBUILD_MCOUNT_RA_ADDRESS
 262
 263 #define S_RA_SP (0xafbf << 16)  /* s{d,w} ra, offset(sp) */
 264 #define S_R_SP  (0xafb0 << 16)  /* s{d,w} R, offset(sp) */
 265 #define OFFSET_MASK     0xffff  /* stack offset range: 0 ~ PT_SIZE */
 266
 267 unsigned long ftrace_get_parent_ra_addr(unsigned long self_ra, unsigned long
 268                 old_parent_ra, unsigned long parent_ra_addr, unsigned long fp)
 269 {
 270         unsigned long sp, ip, tmp;
 271         unsigned int code;
 272         int faulted;
 273
 274         /*
 275          * For module, move the ip from the return address after the
 276          * instruction "lui v1, hi_16bit_of_mcount"(offset is 24), but for
 277          * kernel, move after the instruction "move ra, at"(offset is 16)
 278          */
 279         ip = self_ra - (core_kernel_text(self_ra) ? 16 : 24);
 280
 281         /*
 282          * search the text until finding the non-store instruction or "s{d,w}
 283          * ra, offset(sp)" instruction
 284          */
 285         do {
 286                 /* get the code at "ip": code = *(unsigned int *)ip; */
 287                 safe_load_code(code, ip, faulted);
 288
 289                 if (unlikely(faulted))
 290                         return 0;
 291                 /*
 292                  * If we hit the non-store instruction before finding where the
 293                  * ra is stored, then this is a leaf function and it does not
 294                  * store the ra on the stack
 295                  */
 296                 if ((code & S_R_SP) != S_R_SP)
 297                         return parent_ra_addr;
 298
 299                 /* Move to the next instruction */
 300                 ip -= 4;
 301         } while ((code & S_RA_SP) != S_RA_SP);
 302
 303         sp = fp + (code & OFFSET_MASK);
 304
 305         /* tmp = *(unsigned long *)sp; */
 306         safe_load_stack(tmp, sp, faulted);
 307         if (unlikely(faulted))
 308                 return 0;
 309
 310         if (tmp == old_parent_ra)
 311                 return sp;
 312         return 0;
 313 }
 314
 315 #endif  /* !KBUILD_MCOUNT_RA_ADDRESS */
 316
 317 /*
 318  * Hook the return address and push it in the stack of return addrs
 319  * in current thread info.
 320  */
 321 void prepare_ftrace_return(unsigned long *parent_ra_addr, unsigned long self_ra,
 322                            unsigned long fp)
 323 {
 324         unsigned long old_parent_ra;
 325         unsigned long return_hooker = (unsigned long)
 326             &return_to_handler;
 327         int faulted, insns;
 328
 329         if (unlikely(ftrace_graph_is_dead()))
 330                 return;
 331
 332         if (unlikely(atomic_read(&current->tracing_graph_pause)))
 333                 return;
 334
 335         /*
 336          * "parent_ra_addr" is the stack address saved the return address of
 337          * the caller of _mcount.
 338          *
 339          * if the gcc < 4.5, a leaf function does not save the return address
 340          * in the stack address, so, we "emulate" one in _mcount's stack space,
 341          * and hijack it directly, but for a non-leaf function, it save the
 342          * return address to the its own stack space, we can not hijack it
 343          * directly, but need to find the real stack address,
 344          * ftrace_get_parent_addr() does it!
 345          *
 346          * if gcc>= 4.5, with the new -mmcount-ra-address option, for a
 347          * non-leaf function, the location of the return address will be saved
 348          * to $12 for us, and for a leaf function, only put a zero into $12. we
 349          * do it in ftrace_graph_caller of mcount.S.
 350          */
 351
 352         /* old_parent_ra = *parent_ra_addr; */
 353         safe_load_stack(old_parent_ra, parent_ra_addr, faulted);
 354         if (unlikely(faulted))
 355                 goto out;
 356 #ifndef KBUILD_MCOUNT_RA_ADDRESS
 357         parent_ra_addr = (unsigned long *)ftrace_get_parent_ra_addr(self_ra,
 358                         old_parent_ra, (unsigned long)parent_ra_addr, fp);
 359         /*
 360          * If fails when getting the stack address of the non-leaf function's
 361          * ra, stop function graph tracer and return
 362          */
 363         if (parent_ra_addr == NULL)
 364                 goto out;
 365 #endif
 366         /* *parent_ra_addr = return_hooker; */
 367         safe_store_stack(return_hooker, parent_ra_addr, faulted);
 368         if (unlikely(faulted))
 369                 goto out;
 370
 371         /*
 372          * Get the recorded ip of the current mcount calling site in the
 373          * __mcount_loc section, which will be used to filter the function
 374          * entries configured through the tracing/set_graph_function interface.
 375          */
 376
 377         insns = core_kernel_text(self_ra) ? 2 : MCOUNT_OFFSET_INSNS + 1;
 378         self_ra -= (MCOUNT_INSN_SIZE * insns);
 379
 380         if (function_graph_enter(old_parent_ra, self_ra, fp, NULL))
 381                 *parent_ra_addr = old_parent_ra;
 382         return;
 383 out:
 384         ftrace_graph_stop();
 385         WARN_ON(1);
 386 }
 387 #endif  /* CONFIG_FUNCTION_GRAPH_TRACER */
 388
 389 #ifdef CONFIG_FTRACE_SYSCALLS
 390
 391 #ifdef CONFIG_32BIT
 392 unsigned long __init arch_syscall_addr(int nr)
 393 {
 394         return (unsigned long)sys_call_table[nr - __NR_O32_Linux];
 395 }
 396 #endif
 397
 398 #ifdef CONFIG_64BIT
 399
 400 unsigned long __init arch_syscall_addr(int nr)
 401 {
 402 #ifdef CONFIG_MIPS32_N32
 403         if (nr >= __NR_N32_Linux && nr <= __NR_N32_Linux + __NR_N32_Linux_syscalls)
 404                 return (unsigned long)sysn32_call_table[nr - __NR_N32_Linux];
 405 #endif
 406         if (nr >= __NR_64_Linux  && nr <= __NR_64_Linux + __NR_64_Linux_syscalls)
 407                 return (unsigned long)sys_call_table[nr - __NR_64_Linux];
 408 #ifdef CONFIG_MIPS32_O32
 409         if (nr >= __NR_O32_Linux && nr <= __NR_O32_Linux + __NR_O32_Linux_syscalls)
 410                 return (unsigned long)sys32_call_table[nr - __NR_O32_Linux];
 411 #endif
 412
 413         return (unsigned long) &sys_ni_syscall;
 414 }
 415 #endif
 416
 417 #endif /* CONFIG_FTRACE_SYSCALLS */