arch/x86/kernel/ftrace.c

   1 // SPDX-License-Identifier: GPL-2.0
   2 /*
   3  * Dynamic function tracing support.
   4  *
   5  * Copyright (C) 2007-2008 Steven Rostedt <srostedt@redhat.com>
   6  *
   7  * Thanks goes to Ingo Molnar, for suggesting the idea.
   8  * Mathieu Desnoyers, for suggesting postponing the modifications.
   9  * Arjan van de Ven, for keeping me straight, and explaining to me
  10  * the dangers of modifying code on the run.
  11  */
  12
  13 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  14
  15 #include <linux/spinlock.h>
  16 #include <linux/hardirq.h>
  17 #include <linux/uaccess.h>
  18 #include <linux/ftrace.h>
  19 #include <linux/percpu.h>
  20 #include <linux/sched.h>
  21 #include <linux/slab.h>
  22 #include <linux/init.h>
  23 #include <linux/list.h>
  24 #include <linux/module.h>
  25 #include <linux/memory.h>
  26 #include <linux/vmalloc.h>
  27
  28 #include <trace/syscall.h>
  29
  30 #include <asm/set_memory.h>
  31 #include <asm/kprobes.h>
  32 #include <asm/ftrace.h>
  33 #include <asm/nops.h>
  34 #include <asm/text-patching.h>
  35
  36 #ifdef CONFIG_DYNAMIC_FTRACE
  37
  38 static int ftrace_poke_late = 0;
  39
  40 int ftrace_arch_code_modify_prepare(void)
  41     __acquires(&text_mutex)
  42 {
  43         /*
  44          * Need to grab text_mutex to prevent a race from module loading
  45          * and live kernel patching from changing the text permissions while
  46          * ftrace has it set to "read/write".
  47          */
  48         mutex_lock(&text_mutex);
  49         ftrace_poke_late = 1;
  50         return 0;
  51 }
  52
  53 int ftrace_arch_code_modify_post_process(void)
  54     __releases(&text_mutex)
  55 {
  56         /*
  57          * ftrace_make_{call,nop}() may be called during
  58          * module load, and we need to finish the text_poke_queue()
  59          * that they do, here.
  60          */
  61         text_poke_finish();
  62         ftrace_poke_late = 0;
  63         mutex_unlock(&text_mutex);
  64         return 0;
  65 }
  66
  67 static const char *ftrace_nop_replace(void)
  68 {
  69         return ideal_nops[NOP_ATOMIC5];
  70 }
  71
  72 static const char *ftrace_call_replace(unsigned long ip, unsigned long addr)
  73 {
  74         return text_gen_insn(CALL_INSN_OPCODE, (void *)ip, (void *)addr);
  75 }
  76
  77 static int ftrace_verify_code(unsigned long ip, const char *old_code)
  78 {
  79         char cur_code[MCOUNT_INSN_SIZE];
  80
  81         /*
  82          * Note:
  83          * We are paranoid about modifying text, as if a bug was to happen, it
  84          * could cause us to read or write to someplace that could cause harm.
  85          * Carefully read and modify the code with probe_kernel_*(), and make
  86          * sure what we read is what we expected it to be before modifying it.
  87          */
  88         /* read the text we want to modify */
  89         if (probe_kernel_read(cur_code, (void *)ip, MCOUNT_INSN_SIZE)) {
  90                 WARN_ON(1);
  91                 return -EFAULT;
  92         }
  93
  94         /* Make sure it is what we expect it to be */
  95         if (memcmp(cur_code, old_code, MCOUNT_INSN_SIZE) != 0) {
  96                 WARN_ON(1);
  97                 return -EINVAL;
  98         }
  99
 100         return 0;
 101 }
 102
 103 /*
 104  * Marked __ref because it calls text_poke_early() which is .init.text. That is
 105  * ok because that call will happen early, during boot, when .init sections are
 106  * still present.
 107  */
 108 static int __ref
 109 ftrace_modify_code_direct(unsigned long ip, const char *old_code,
 110                           const char *new_code)
 111 {
 112         int ret = ftrace_verify_code(ip, old_code);
 113         if (ret)
 114                 return ret;
 115
 116         /* replace the text with the new text */
 117         if (ftrace_poke_late)
 118                 text_poke_queue((void *)ip, new_code, MCOUNT_INSN_SIZE, NULL);
 119         else
 120                 text_poke_early((void *)ip, new_code, MCOUNT_INSN_SIZE);
 121         return 0;
 122 }
 123
 124 int ftrace_make_nop(struct module *mod, struct dyn_ftrace *rec, unsigned long addr)
 125 {
 126         unsigned long ip = rec->ip;
 127         const char *new, *old;
 128
 129         old = ftrace_call_replace(ip, addr);
 130         new = ftrace_nop_replace();
 131
 132         /*
 133          * On boot up, and when modules are loaded, the MCOUNT_ADDR
 134          * is converted to a nop, and will never become MCOUNT_ADDR
 135          * again. This code is either running before SMP (on boot up)
 136          * or before the code will ever be executed (module load).
 137          * We do not want to use the breakpoint version in this case,
 138          * just modify the code directly.
 139          */
 140         if (addr == MCOUNT_ADDR)
 141                 return ftrace_modify_code_direct(ip, old, new);
 142
 143         /*
 144          * x86 overrides ftrace_replace_code -- this function will never be used
 145          * in this case.
 146          */
 147         WARN_ONCE(1, "invalid use of ftrace_make_nop");
 148         return -EINVAL;
 149 }
 150
 151 int ftrace_make_call(struct dyn_ftrace *rec, unsigned long addr)
 152 {
 153         unsigned long ip = rec->ip;
 154         const char *new, *old;
 155
 156         old = ftrace_nop_replace();
 157         new = ftrace_call_replace(ip, addr);
 158
 159         /* Should only be called when module is loaded */
 160         return ftrace_modify_code_direct(rec->ip, old, new);
 161 }
 162
 163 /*
 164  * Should never be called:
 165  *  As it is only called by __ftrace_replace_code() which is called by
 166  *  ftrace_replace_code() that x86 overrides, and by ftrace_update_code()
 167  *  which is called to turn mcount into nops or nops into function calls
 168  *  but not to convert a function from not using regs to one that uses
 169  *  regs, which ftrace_modify_call() is for.
 170  */
 171 int ftrace_modify_call(struct dyn_ftrace *rec, unsigned long old_addr,
 172                                  unsigned long addr)
 173 {
 174         WARN_ON(1);
 175         return -EINVAL;
 176 }
 177
 178 int ftrace_update_ftrace_func(ftrace_func_t func)
 179 {
 180         unsigned long ip;
 181         const char *new;
 182
 183         ip = (unsigned long)(&ftrace_call);
 184         new = ftrace_call_replace(ip, (unsigned long)func);
 185         text_poke_bp((void *)ip, new, MCOUNT_INSN_SIZE, NULL);
 186
 187         ip = (unsigned long)(&ftrace_regs_call);
 188         new = ftrace_call_replace(ip, (unsigned long)func);
 189         text_poke_bp((void *)ip, new, MCOUNT_INSN_SIZE, NULL);
 190
 191         return 0;
 192 }
 193
 194 void ftrace_replace_code(int enable)
 195 {
 196         struct ftrace_rec_iter *iter;
 197         struct dyn_ftrace *rec;
 198         const char *new, *old;
 199         int ret;
 200
 201         for_ftrace_rec_iter(iter) {
 202                 rec = ftrace_rec_iter_record(iter);
 203
 204                 switch (ftrace_test_record(rec, enable)) {
 205                 case FTRACE_UPDATE_IGNORE:
 206                 default:
 207                         continue;
 208
 209                 case FTRACE_UPDATE_MAKE_CALL:
 210                         old = ftrace_nop_replace();
 211                         break;
 212
 213                 case FTRACE_UPDATE_MODIFY_CALL:
 214                 case FTRACE_UPDATE_MAKE_NOP:
 215                         old = ftrace_call_replace(rec->ip, ftrace_get_addr_curr(rec));
 216                         break;
 217                 }
 218
 219                 ret = ftrace_verify_code(rec->ip, old);
 220                 if (ret) {
 221                         ftrace_bug(ret, rec);
 222                         return;
 223                 }
 224         }
 225
 226         for_ftrace_rec_iter(iter) {
 227                 rec = ftrace_rec_iter_record(iter);
 228
 229                 switch (ftrace_test_record(rec, enable)) {
 230                 case FTRACE_UPDATE_IGNORE:
 231                 default:
 232                         continue;
 233
 234                 case FTRACE_UPDATE_MAKE_CALL:
 235                 case FTRACE_UPDATE_MODIFY_CALL:
 236                         new = ftrace_call_replace(rec->ip, ftrace_get_addr_new(rec));
 237                         break;
 238
 239                 case FTRACE_UPDATE_MAKE_NOP:
 240                         new = ftrace_nop_replace();
 241                         break;
 242                 }
 243
 244                 text_poke_queue((void *)rec->ip, new, MCOUNT_INSN_SIZE, NULL);
 245                 ftrace_update_record(rec, enable);
 246         }
 247         text_poke_finish();
 248 }
 249
 250 void arch_ftrace_update_code(int command)
 251 {
 252         ftrace_modify_all_code(command);
 253 }
 254
 255 int __init ftrace_dyn_arch_init(void)
 256 {
 257         return 0;
 258 }
 259
 260 /* Currently only x86_64 supports dynamic trampolines */
 261 #ifdef CONFIG_X86_64
 262
 263 #ifdef CONFIG_MODULES
 264 #include <linux/moduleloader.h>
 265 /* Module allocation simplifies allocating memory for code */
 266 static inline void *alloc_tramp(unsigned long size)
 267 {
 268         return module_alloc(size);
 269 }
 270 static inline void tramp_free(void *tramp)
 271 {
 272         module_memfree(tramp);
 273 }
 274 #else
 275 /* Trampolines can only be created if modules are supported */
 276 static inline void *alloc_tramp(unsigned long size)
 277 {
 278         return NULL;
 279 }
 280 static inline void tramp_free(void *tramp) { }
 281 #endif
 282
 283 /* Defined as markers to the end of the ftrace default trampolines */
 284 extern void ftrace_regs_caller_end(void);
 285 extern void ftrace_epilogue(void);
 286 extern void ftrace_caller_op_ptr(void);
 287 extern void ftrace_regs_caller_op_ptr(void);
 288
 289 /* movq function_trace_op(%rip), %rdx */
 290 /* 0x48 0x8b 0x15 <offset-to-ftrace_trace_op (4 bytes)> */
 291 #define OP_REF_SIZE     7
 292
 293 /*
 294  * The ftrace_ops is passed to the function callback. Since the
 295  * trampoline only services a single ftrace_ops, we can pass in
 296  * that ops directly.
 297  *
 298  * The ftrace_op_code_union is used to create a pointer to the
 299  * ftrace_ops that will be passed to the callback function.
 300  */
 301 union ftrace_op_code_union {
 302         char code[OP_REF_SIZE];
 303         struct {
 304                 char op[3];
 305                 int offset;
 306         } __attribute__((packed));
 307 };
 308
 309 #define RET_SIZE                1
 310
 311 static unsigned long
 312 create_trampoline(struct ftrace_ops *ops, unsigned int *tramp_size)
 313 {
 314         unsigned long start_offset;
 315         unsigned long end_offset;
 316         unsigned long op_offset;
 317         unsigned long call_offset;
 318         unsigned long offset;
 319         unsigned long npages;
 320         unsigned long size;
 321         unsigned long retq;
 322         unsigned long *ptr;
 323         void *trampoline;
 324         void *ip;
 325         /* 48 8b 15 <offset> is movq <offset>(%rip), %rdx */
 326         unsigned const char op_ref[] = { 0x48, 0x8b, 0x15 };
 327         union ftrace_op_code_union op_ptr;
 328         int ret;
 329
 330         if (ops->flags & FTRACE_OPS_FL_SAVE_REGS) {
 331                 start_offset = (unsigned long)ftrace_regs_caller;
 332                 end_offset = (unsigned long)ftrace_regs_caller_end;
 333                 op_offset = (unsigned long)ftrace_regs_caller_op_ptr;
 334                 call_offset = (unsigned long)ftrace_regs_call;
 335         } else {
 336                 start_offset = (unsigned long)ftrace_caller;
 337                 end_offset = (unsigned long)ftrace_epilogue;
 338                 op_offset = (unsigned long)ftrace_caller_op_ptr;
 339                 call_offset = (unsigned long)ftrace_call;
 340         }
 341
 342         size = end_offset - start_offset;
 343
 344         /*
 345          * Allocate enough size to store the ftrace_caller code,
 346          * the iret , as well as the address of the ftrace_ops this
 347          * trampoline is used for.
 348          */
 349         trampoline = alloc_tramp(size + RET_SIZE + sizeof(void *));
 350         if (!trampoline)
 351                 return 0;
 352
 353         *tramp_size = size + RET_SIZE + sizeof(void *);
 354         npages = DIV_ROUND_UP(*tramp_size, PAGE_SIZE);
 355
 356         /* Copy ftrace_caller onto the trampoline memory */
 357         ret = probe_kernel_read(trampoline, (void *)start_offset, size);
 358         if (WARN_ON(ret < 0))
 359                 goto fail;
 360
 361         ip = trampoline + size;
 362
 363         /* The trampoline ends with ret(q) */
 364         retq = (unsigned long)ftrace_stub;
 365         ret = probe_kernel_read(ip, (void *)retq, RET_SIZE);
 366         if (WARN_ON(ret < 0))
 367                 goto fail;
 368
 369         /*
 370          * The address of the ftrace_ops that is used for this trampoline
 371          * is stored at the end of the trampoline. This will be used to
 372          * load the third parameter for the callback. Basically, that
 373          * location at the end of the trampoline takes the place of
 374          * the global function_trace_op variable.
 375          */
 376
 377         ptr = (unsigned long *)(trampoline + size + RET_SIZE);
 378         *ptr = (unsigned long)ops;
 379
 380         op_offset -= start_offset;
 381         memcpy(&op_ptr, trampoline + op_offset, OP_REF_SIZE);
 382
 383         /* Are we pointing to the reference? */
 384         if (WARN_ON(memcmp(op_ptr.op, op_ref, 3) != 0))
 385                 goto fail;
 386
 387         /* Load the contents of ptr into the callback parameter */
 388         offset = (unsigned long)ptr;
 389         offset -= (unsigned long)trampoline + op_offset + OP_REF_SIZE;
 390
 391         op_ptr.offset = offset;
 392
 393         /* put in the new offset to the ftrace_ops */
 394         memcpy(trampoline + op_offset, &op_ptr, OP_REF_SIZE);
 395
 396         /* put in the call to the function */
 397         mutex_lock(&text_mutex);
 398         call_offset -= start_offset;
 399         memcpy(trampoline + call_offset,
 400                text_gen_insn(CALL_INSN_OPCODE,
 401                              trampoline + call_offset,
 402                              ftrace_ops_get_func(ops)), CALL_INSN_SIZE);
 403         mutex_unlock(&text_mutex);
 404
 405         /* ALLOC_TRAMP flags lets us know we created it */
 406         ops->flags |= FTRACE_OPS_FL_ALLOC_TRAMP;
 407
 408         set_vm_flush_reset_perms(trampoline);
 409
 410         set_memory_ro((unsigned long)trampoline, npages);
 411         set_memory_x((unsigned long)trampoline, npages);
 412         return (unsigned long)trampoline;
 413 fail:
 414         tramp_free(trampoline);
 415         return 0;
 416 }
 417
 418 static unsigned long calc_trampoline_call_offset(bool save_regs)
 419 {
 420         unsigned long start_offset;
 421         unsigned long call_offset;
 422
 423         if (save_regs) {
 424                 start_offset = (unsigned long)ftrace_regs_caller;
 425                 call_offset = (unsigned long)ftrace_regs_call;
 426         } else {
 427                 start_offset = (unsigned long)ftrace_caller;
 428                 call_offset = (unsigned long)ftrace_call;
 429         }
 430
 431         return call_offset - start_offset;
 432 }
 433
 434 void arch_ftrace_update_trampoline(struct ftrace_ops *ops)
 435 {
 436         ftrace_func_t func;
 437         unsigned long offset;
 438         unsigned long ip;
 439         unsigned int size;
 440         const char *new;
 441
 442         if (!ops->trampoline) {
 443                 ops->trampoline = create_trampoline(ops, &size);
 444                 if (!ops->trampoline)
 445                         return;
 446                 ops->trampoline_size = size;
 447                 return;
 448         }
 449
 450         /*
 451          * The ftrace_ops caller may set up its own trampoline.
 452          * In such a case, this code must not modify it.
 453          */
 454         if (!(ops->flags & FTRACE_OPS_FL_ALLOC_TRAMP))
 455                 return;
 456
 457         offset = calc_trampoline_call_offset(ops->flags & FTRACE_OPS_FL_SAVE_REGS);
 458         ip = ops->trampoline + offset;
 459         func = ftrace_ops_get_func(ops);
 460
 461         mutex_lock(&text_mutex);
 462         /* Do a safe modify in case the trampoline is executing */
 463         new = ftrace_call_replace(ip, (unsigned long)func);
 464         text_poke_bp((void *)ip, new, MCOUNT_INSN_SIZE, NULL);
 465         mutex_unlock(&text_mutex);
 466 }
 467
 468 /* Return the address of the function the trampoline calls */
 469 static void *addr_from_call(void *ptr)
 470 {
 471         union text_poke_insn call;
 472         int ret;
 473
 474         ret = probe_kernel_read(&call, ptr, CALL_INSN_SIZE);
 475         if (WARN_ON_ONCE(ret < 0))
 476                 return NULL;
 477
 478         /* Make sure this is a call */
 479         if (WARN_ON_ONCE(call.opcode != CALL_INSN_OPCODE)) {
 480                 pr_warn("Expected E8, got %x\n", call.opcode);
 481                 return NULL;
 482         }
 483
 484         return ptr + CALL_INSN_SIZE + call.disp;
 485 }
 486
 487 void prepare_ftrace_return(unsigned long self_addr, unsigned long *parent,
 488                            unsigned long frame_pointer);
 489
 490 /*
 491  * If the ops->trampoline was not allocated, then it probably
 492  * has a static trampoline func, or is the ftrace caller itself.
 493  */
 494 static void *static_tramp_func(struct ftrace_ops *ops, struct dyn_ftrace *rec)
 495 {
 496         unsigned long offset;
 497         bool save_regs = rec->flags & FTRACE_FL_REGS_EN;
 498         void *ptr;
 499
 500         if (ops && ops->trampoline) {
 501 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
 502                 /*
 503                  * We only know about function graph tracer setting as static
 504                  * trampoline.
 505                  */
 506                 if (ops->trampoline == FTRACE_GRAPH_ADDR)
 507                         return (void *)prepare_ftrace_return;
 508 #endif
 509                 return NULL;
 510         }
 511
 512         offset = calc_trampoline_call_offset(save_regs);
 513
 514         if (save_regs)
 515                 ptr = (void *)FTRACE_REGS_ADDR + offset;
 516         else
 517                 ptr = (void *)FTRACE_ADDR + offset;
 518
 519         return addr_from_call(ptr);
 520 }
 521
 522 void *arch_ftrace_trampoline_func(struct ftrace_ops *ops, struct dyn_ftrace *rec)
 523 {
 524         unsigned long offset;
 525
 526         /* If we didn't allocate this trampoline, consider it static */
 527         if (!ops || !(ops->flags & FTRACE_OPS_FL_ALLOC_TRAMP))
 528                 return static_tramp_func(ops, rec);
 529
 530         offset = calc_trampoline_call_offset(ops->flags & FTRACE_OPS_FL_SAVE_REGS);
 531         return addr_from_call((void *)ops->trampoline + offset);
 532 }
 533
 534 void arch_ftrace_trampoline_free(struct ftrace_ops *ops)
 535 {
 536         if (!ops || !(ops->flags & FTRACE_OPS_FL_ALLOC_TRAMP))
 537                 return;
 538
 539         tramp_free((void *)ops->trampoline);
 540         ops->trampoline = 0;
 541 }
 542
 543 #endif /* CONFIG_X86_64 */
 544 #endif /* CONFIG_DYNAMIC_FTRACE */
 545
 546 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
 547
 548 #ifdef CONFIG_DYNAMIC_FTRACE
 549 extern void ftrace_graph_call(void);
 550
 551 static const char *ftrace_jmp_replace(unsigned long ip, unsigned long addr)
 552 {
 553         return text_gen_insn(JMP32_INSN_OPCODE, (void *)ip, (void *)addr);
 554 }
 555
 556 static int ftrace_mod_jmp(unsigned long ip, void *func)
 557 {
 558         const char *new;
 559
 560         new = ftrace_jmp_replace(ip, (unsigned long)func);
 561         text_poke_bp((void *)ip, new, MCOUNT_INSN_SIZE, NULL);
 562         return 0;
 563 }
 564
 565 int ftrace_enable_ftrace_graph_caller(void)
 566 {
 567         unsigned long ip = (unsigned long)(&ftrace_graph_call);
 568
 569         return ftrace_mod_jmp(ip, &ftrace_graph_caller);
 570 }
 571
 572 int ftrace_disable_ftrace_graph_caller(void)
 573 {
 574         unsigned long ip = (unsigned long)(&ftrace_graph_call);
 575
 576         return ftrace_mod_jmp(ip, &ftrace_stub);
 577 }
 578
 579 #endif /* !CONFIG_DYNAMIC_FTRACE */
 580
 581 /*
 582  * Hook the return address and push it in the stack of return addrs
 583  * in current thread info.
 584  */
 585 void prepare_ftrace_return(unsigned long self_addr, unsigned long *parent,
 586                            unsigned long frame_pointer)
 587 {
 588         unsigned long return_hooker = (unsigned long)&return_to_handler;
 589         unsigned long old;
 590         int faulted;
 591
 592         /*
 593          * When resuming from suspend-to-ram, this function can be indirectly
 594          * called from early CPU startup code while the CPU is in real mode,
 595          * which would fail miserably.  Make sure the stack pointer is a
 596          * virtual address.
 597          *
 598          * This check isn't as accurate as virt_addr_valid(), but it should be
 599          * good enough for this purpose, and it's fast.
 600          */
 601         if (unlikely((long)__builtin_frame_address(0) >= 0))
 602                 return;
 603
 604         if (unlikely(ftrace_graph_is_dead()))
 605                 return;
 606
 607         if (unlikely(atomic_read(&current->tracing_graph_pause)))
 608                 return;
 609
 610         /*
 611          * Protect against fault, even if it shouldn't
 612          * happen. This tool is too much intrusive to
 613          * ignore such a protection.
 614          */
 615         asm volatile(
 616                 "1: " _ASM_MOV " (%[parent]), %[old]\n"
 617                 "2: " _ASM_MOV " %[return_hooker], (%[parent])\n"
 618                 "   movl $0, %[faulted]\n"
 619                 "3:\n"
 620
 621                 ".section .fixup, \"ax\"\n"
 622                 "4: movl $1, %[faulted]\n"
 623                 "   jmp 3b\n"
 624                 ".previous\n"
 625
 626                 _ASM_EXTABLE(1b, 4b)
 627                 _ASM_EXTABLE(2b, 4b)
 628
 629                 : [old] "=&r" (old), [faulted] "=r" (faulted)
 630                 : [parent] "r" (parent), [return_hooker] "r" (return_hooker)
 631                 : "memory"
 632         );
 633
 634         if (unlikely(faulted)) {
 635                 ftrace_graph_stop();
 636                 WARN_ON(1);
 637                 return;
 638         }
 639
 640         if (function_graph_enter(old, self_addr, frame_pointer, parent))
 641                 *parent = old;
 642 }
 643 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */