arch/x86/kernel/ftrace.c

   1 /*
   2  * Code for replacing ftrace calls with jumps.
   3  *
   4  * Copyright (C) 2007-2008 Steven Rostedt <srostedt@redhat.com>
   5  *
   6  * Thanks goes to Ingo Molnar, for suggesting the idea.
   7  * Mathieu Desnoyers, for suggesting postponing the modifications.
   8  * Arjan van de Ven, for keeping me straight, and explaining to me
   9  * the dangers of modifying code on the run.
  10  */
  11
  12 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  13
  14 #include <linux/spinlock.h>
  15 #include <linux/hardirq.h>
  16 #include <linux/uaccess.h>
  17 #include <linux/ftrace.h>
  18 #include <linux/percpu.h>
  19 #include <linux/sched.h>
  20 #include <linux/init.h>
  21 #include <linux/list.h>
  22 #include <linux/module.h>
  23
  24 #include <trace/syscall.h>
  25
  26 #include <asm/cacheflush.h>
  27 #include <asm/kprobes.h>
  28 #include <asm/ftrace.h>
  29 #include <asm/nops.h>
  30
  31 #ifdef CONFIG_DYNAMIC_FTRACE
  32
  33 int ftrace_arch_code_modify_prepare(void)
  34 {
  35         set_kernel_text_rw();
  36         set_all_modules_text_rw();
  37         return 0;
  38 }
  39
  40 int ftrace_arch_code_modify_post_process(void)
  41 {
  42         set_all_modules_text_ro();
  43         set_kernel_text_ro();
  44         return 0;
  45 }
  46
  47 union ftrace_code_union {
  48         char code[MCOUNT_INSN_SIZE];
  49         struct {
  50                 char e8;
  51                 int offset;
  52         } __attribute__((packed));
  53 };
  54
  55 static int ftrace_calc_offset(long ip, long addr)
  56 {
  57         return (int)(addr - ip);
  58 }
  59
  60 static unsigned char *ftrace_call_replace(unsigned long ip, unsigned long addr)
  61 {
  62         static union ftrace_code_union calc;
  63
  64         calc.e8         = 0xe8;
  65         calc.offset     = ftrace_calc_offset(ip + MCOUNT_INSN_SIZE, addr);
  66
  67         /*
  68          * No locking needed, this must be called via kstop_machine
  69          * which in essence is like running on a uniprocessor machine.
  70          */
  71         return calc.code;
  72 }
  73
  74 static inline int
  75 within(unsigned long addr, unsigned long start, unsigned long end)
  76 {
  77         return addr >= start && addr < end;
  78 }
  79
  80 static unsigned long text_ip_addr(unsigned long ip)
  81 {
  82         /*
  83          * On x86_64, kernel text mappings are mapped read-only with
  84          * CONFIG_DEBUG_RODATA. So we use the kernel identity mapping instead
  85          * of the kernel text mapping to modify the kernel text.
  86          *
  87          * For 32bit kernels, these mappings are same and we can use
  88          * kernel identity mapping to modify code.
  89          */
  90         if (within(ip, (unsigned long)_text, (unsigned long)_etext))
  91                 ip = (unsigned long)__va(__pa_symbol(ip));
  92
  93         return ip;
  94 }
  95
  96 static const unsigned char *ftrace_nop_replace(void)
  97 {
  98         return ideal_nops[NOP_ATOMIC5];
  99 }
 100
 101 static int
 102 ftrace_modify_code_direct(unsigned long ip, unsigned const char *old_code,
 103                    unsigned const char *new_code)
 104 {
 105         unsigned char replaced[MCOUNT_INSN_SIZE];
 106
 107         /*
 108          * Note: Due to modules and __init, code can
 109          *  disappear and change, we need to protect against faulting
 110          *  as well as code changing. We do this by using the
 111          *  probe_kernel_* functions.
 112          *
 113          * No real locking needed, this code is run through
 114          * kstop_machine, or before SMP starts.
 115          */
 116
 117         /* read the text we want to modify */
 118         if (probe_kernel_read(replaced, (void *)ip, MCOUNT_INSN_SIZE))
 119                 return -EFAULT;
 120
 121         /* Make sure it is what we expect it to be */
 122         if (memcmp(replaced, old_code, MCOUNT_INSN_SIZE) != 0)
 123                 return -EINVAL;
 124
 125         ip = text_ip_addr(ip);
 126
 127         /* replace the text with the new text */
 128         if (probe_kernel_write((void *)ip, new_code, MCOUNT_INSN_SIZE))
 129                 return -EPERM;
 130
 131         sync_core();
 132
 133         return 0;
 134 }
 135
 136 int ftrace_make_nop(struct module *mod,
 137                     struct dyn_ftrace *rec, unsigned long addr)
 138 {
 139         unsigned const char *new, *old;
 140         unsigned long ip = rec->ip;
 141
 142         old = ftrace_call_replace(ip, addr);
 143         new = ftrace_nop_replace();
 144
 145         /*
 146          * On boot up, and when modules are loaded, the MCOUNT_ADDR
 147          * is converted to a nop, and will never become MCOUNT_ADDR
 148          * again. This code is either running before SMP (on boot up)
 149          * or before the code will ever be executed (module load).
 150          * We do not want to use the breakpoint version in this case,
 151          * just modify the code directly.
 152          */
 153         if (addr == MCOUNT_ADDR)
 154                 return ftrace_modify_code_direct(rec->ip, old, new);
 155
 156         /* Normal cases use add_brk_on_nop */
 157         WARN_ONCE(1, "invalid use of ftrace_make_nop");
 158         return -EINVAL;
 159 }
 160
 161 int ftrace_make_call(struct dyn_ftrace *rec, unsigned long addr)
 162 {
 163         unsigned const char *new, *old;
 164         unsigned long ip = rec->ip;
 165
 166         old = ftrace_nop_replace();
 167         new = ftrace_call_replace(ip, addr);
 168
 169         /* Should only be called when module is loaded */
 170         return ftrace_modify_code_direct(rec->ip, old, new);
 171 }
 172
 173 /*
 174  * The modifying_ftrace_code is used to tell the breakpoint
 175  * handler to call ftrace_int3_handler(). If it fails to
 176  * call this handler for a breakpoint added by ftrace, then
 177  * the kernel may crash.
 178  *
 179  * As atomic_writes on x86 do not need a barrier, we do not
 180  * need to add smp_mb()s for this to work. It is also considered
 181  * that we can not read the modifying_ftrace_code before
 182  * executing the breakpoint. That would be quite remarkable if
 183  * it could do that. Here's the flow that is required:
 184  *
 185  *   CPU-0                          CPU-1
 186  *
 187  * atomic_inc(mfc);
 188  * write int3s
 189  *                              <trap-int3> // implicit (r)mb
 190  *                              if (atomic_read(mfc))
 191  *                                      call ftrace_int3_handler()
 192  *
 193  * Then when we are finished:
 194  *
 195  * atomic_dec(mfc);
 196  *
 197  * If we hit a breakpoint that was not set by ftrace, it does not
 198  * matter if ftrace_int3_handler() is called or not. It will
 199  * simply be ignored. But it is crucial that a ftrace nop/caller
 200  * breakpoint is handled. No other user should ever place a
 201  * breakpoint on an ftrace nop/caller location. It must only
 202  * be done by this code.
 203  */
 204 atomic_t modifying_ftrace_code __read_mostly;
 205
 206 static int
 207 ftrace_modify_code(unsigned long ip, unsigned const char *old_code,
 208                    unsigned const char *new_code);
 209
 210 /*
 211  * Should never be called:
 212  *  As it is only called by __ftrace_replace_code() which is called by
 213  *  ftrace_replace_code() that x86 overrides, and by ftrace_update_code()
 214  *  which is called to turn mcount into nops or nops into function calls
 215  *  but not to convert a function from not using regs to one that uses
 216  *  regs, which ftrace_modify_call() is for.
 217  */
 218 int ftrace_modify_call(struct dyn_ftrace *rec, unsigned long old_addr,
 219                                  unsigned long addr)
 220 {
 221         WARN_ON(1);
 222         return -EINVAL;
 223 }
 224
 225 static unsigned long ftrace_update_func;
 226
 227 static int update_ftrace_func(unsigned long ip, void *new)
 228 {
 229         unsigned char old[MCOUNT_INSN_SIZE];
 230         int ret;
 231
 232         memcpy(old, (void *)ip, MCOUNT_INSN_SIZE);
 233
 234         ftrace_update_func = ip;
 235         /* Make sure the breakpoints see the ftrace_update_func update */
 236         smp_wmb();
 237
 238         /* See comment above by declaration of modifying_ftrace_code */
 239         atomic_inc(&modifying_ftrace_code);
 240
 241         ret = ftrace_modify_code(ip, old, new);
 242
 243         atomic_dec(&modifying_ftrace_code);
 244
 245         return ret;
 246 }
 247
 248 int ftrace_update_ftrace_func(ftrace_func_t func)
 249 {
 250         unsigned long ip = (unsigned long)(&ftrace_call);
 251         unsigned char *new;
 252         int ret;
 253
 254         new = ftrace_call_replace(ip, (unsigned long)func);
 255         ret = update_ftrace_func(ip, new);
 256
 257         /* Also update the regs callback function */
 258         if (!ret) {
 259                 ip = (unsigned long)(&ftrace_regs_call);
 260                 new = ftrace_call_replace(ip, (unsigned long)func);
 261                 ret = update_ftrace_func(ip, new);
 262         }
 263
 264         return ret;
 265 }
 266
 267 static int is_ftrace_caller(unsigned long ip)
 268 {
 269         if (ip == ftrace_update_func)
 270                 return 1;
 271
 272         return 0;
 273 }
 274
 275 /*
 276  * A breakpoint was added to the code address we are about to
 277  * modify, and this is the handle that will just skip over it.
 278  * We are either changing a nop into a trace call, or a trace
 279  * call to a nop. While the change is taking place, we treat
 280  * it just like it was a nop.
 281  */
 282 int ftrace_int3_handler(struct pt_regs *regs)
 283 {
 284         unsigned long ip;
 285
 286         if (WARN_ON_ONCE(!regs))
 287                 return 0;
 288
 289         ip = regs->ip - 1;
 290         if (!ftrace_location(ip) && !is_ftrace_caller(ip))
 291                 return 0;
 292
 293         regs->ip += MCOUNT_INSN_SIZE - 1;
 294
 295         return 1;
 296 }
 297
 298 static int ftrace_write(unsigned long ip, const char *val, int size)
 299 {
 300         /*
 301          * On x86_64, kernel text mappings are mapped read-only with
 302          * CONFIG_DEBUG_RODATA. So we use the kernel identity mapping instead
 303          * of the kernel text mapping to modify the kernel text.
 304          *
 305          * For 32bit kernels, these mappings are same and we can use
 306          * kernel identity mapping to modify code.
 307          */
 308         if (within(ip, (unsigned long)_text, (unsigned long)_etext))
 309                 ip = (unsigned long)__va(__pa_symbol(ip));
 310
 311         if (probe_kernel_write((void *)ip, val, size))
 312                 return -EPERM;
 313
 314         return 0;
 315 }
 316
 317 static int add_break(unsigned long ip, const char *old)
 318 {
 319         unsigned char replaced[MCOUNT_INSN_SIZE];
 320         unsigned char brk = BREAKPOINT_INSTRUCTION;
 321
 322         if (probe_kernel_read(replaced, (void *)ip, MCOUNT_INSN_SIZE))
 323                 return -EFAULT;
 324
 325         /* Make sure it is what we expect it to be */
 326         if (memcmp(replaced, old, MCOUNT_INSN_SIZE) != 0)
 327                 return -EINVAL;
 328
 329         return ftrace_write(ip, &brk, 1);
 330 }
 331
 332 static int add_brk_on_call(struct dyn_ftrace *rec, unsigned long addr)
 333 {
 334         unsigned const char *old;
 335         unsigned long ip = rec->ip;
 336
 337         old = ftrace_call_replace(ip, addr);
 338
 339         return add_break(rec->ip, old);
 340 }
 341
 342
 343 static int add_brk_on_nop(struct dyn_ftrace *rec)
 344 {
 345         unsigned const char *old;
 346
 347         old = ftrace_nop_replace();
 348
 349         return add_break(rec->ip, old);
 350 }
 351
 352 /*
 353  * If the record has the FTRACE_FL_REGS set, that means that it
 354  * wants to convert to a callback that saves all regs. If FTRACE_FL_REGS
 355  * is not not set, then it wants to convert to the normal callback.
 356  */
 357 static unsigned long get_ftrace_addr(struct dyn_ftrace *rec)
 358 {
 359         if (rec->flags & FTRACE_FL_REGS)
 360                 return (unsigned long)FTRACE_REGS_ADDR;
 361         else
 362                 return (unsigned long)FTRACE_ADDR;
 363 }
 364
 365 /*
 366  * The FTRACE_FL_REGS_EN is set when the record already points to
 367  * a function that saves all the regs. Basically the '_EN' version
 368  * represents the current state of the function.
 369  */
 370 static unsigned long get_ftrace_old_addr(struct dyn_ftrace *rec)
 371 {
 372         if (rec->flags & FTRACE_FL_REGS_EN)
 373                 return (unsigned long)FTRACE_REGS_ADDR;
 374         else
 375                 return (unsigned long)FTRACE_ADDR;
 376 }
 377
 378 static int add_breakpoints(struct dyn_ftrace *rec, int enable)
 379 {
 380         unsigned long ftrace_addr;
 381         int ret;
 382
 383         ftrace_addr = get_ftrace_old_addr(rec);
 384
 385         ret = ftrace_test_record(rec, enable);
 386
 387         switch (ret) {
 388         case FTRACE_UPDATE_IGNORE:
 389                 return 0;
 390
 391         case FTRACE_UPDATE_MAKE_CALL:
 392                 /* converting nop to call */
 393                 return add_brk_on_nop(rec);
 394
 395         case FTRACE_UPDATE_MODIFY_CALL_REGS:
 396         case FTRACE_UPDATE_MODIFY_CALL:
 397         case FTRACE_UPDATE_MAKE_NOP:
 398                 /* converting a call to a nop */
 399                 return add_brk_on_call(rec, ftrace_addr);
 400         }
 401         return 0;
 402 }
 403
 404 /*
 405  * On error, we need to remove breakpoints. This needs to
 406  * be done caefully. If the address does not currently have a
 407  * breakpoint, we know we are done. Otherwise, we look at the
 408  * remaining 4 bytes of the instruction. If it matches a nop
 409  * we replace the breakpoint with the nop. Otherwise we replace
 410  * it with the call instruction.
 411  */
 412 static int remove_breakpoint(struct dyn_ftrace *rec)
 413 {
 414         unsigned char ins[MCOUNT_INSN_SIZE];
 415         unsigned char brk = BREAKPOINT_INSTRUCTION;
 416         const unsigned char *nop;
 417         unsigned long ftrace_addr;
 418         unsigned long ip = rec->ip;
 419
 420         /* If we fail the read, just give up */
 421         if (probe_kernel_read(ins, (void *)ip, MCOUNT_INSN_SIZE))
 422                 return -EFAULT;
 423
 424         /* If this does not have a breakpoint, we are done */
 425         if (ins[0] != brk)
 426                 return 0;
 427
 428         nop = ftrace_nop_replace();
 429
 430         /*
 431          * If the last 4 bytes of the instruction do not match
 432          * a nop, then we assume that this is a call to ftrace_addr.
 433          */
 434         if (memcmp(&ins[1], &nop[1], MCOUNT_INSN_SIZE - 1) != 0) {
 435                 /*
 436                  * For extra paranoidism, we check if the breakpoint is on
 437                  * a call that would actually jump to the ftrace_addr.
 438                  * If not, don't touch the breakpoint, we make just create
 439                  * a disaster.
 440                  */
 441                 ftrace_addr = get_ftrace_addr(rec);
 442                 nop = ftrace_call_replace(ip, ftrace_addr);
 443
 444                 if (memcmp(&ins[1], &nop[1], MCOUNT_INSN_SIZE - 1) == 0)
 445                         goto update;
 446
 447                 /* Check both ftrace_addr and ftrace_old_addr */
 448                 ftrace_addr = get_ftrace_old_addr(rec);
 449                 nop = ftrace_call_replace(ip, ftrace_addr);
 450
 451                 if (memcmp(&ins[1], &nop[1], MCOUNT_INSN_SIZE - 1) != 0)
 452                         return -EINVAL;
 453         }
 454
 455  update:
 456         return ftrace_write(ip, nop, 1);
 457 }
 458
 459 static int add_update_code(unsigned long ip, unsigned const char *new)
 460 {
 461         /* skip breakpoint */
 462         ip++;
 463         new++;
 464         return ftrace_write(ip, new, MCOUNT_INSN_SIZE - 1);
 465 }
 466
 467 static int add_update_call(struct dyn_ftrace *rec, unsigned long addr)
 468 {
 469         unsigned long ip = rec->ip;
 470         unsigned const char *new;
 471
 472         new = ftrace_call_replace(ip, addr);
 473         return add_update_code(ip, new);
 474 }
 475
 476 static int add_update_nop(struct dyn_ftrace *rec)
 477 {
 478         unsigned long ip = rec->ip;
 479         unsigned const char *new;
 480
 481         new = ftrace_nop_replace();
 482         return add_update_code(ip, new);
 483 }
 484
 485 static int add_update(struct dyn_ftrace *rec, int enable)
 486 {
 487         unsigned long ftrace_addr;
 488         int ret;
 489
 490         ret = ftrace_test_record(rec, enable);
 491
 492         ftrace_addr  = get_ftrace_addr(rec);
 493
 494         switch (ret) {
 495         case FTRACE_UPDATE_IGNORE:
 496                 return 0;
 497
 498         case FTRACE_UPDATE_MODIFY_CALL_REGS:
 499         case FTRACE_UPDATE_MODIFY_CALL:
 500         case FTRACE_UPDATE_MAKE_CALL:
 501                 /* converting nop to call */
 502                 return add_update_call(rec, ftrace_addr);
 503
 504         case FTRACE_UPDATE_MAKE_NOP:
 505                 /* converting a call to a nop */
 506                 return add_update_nop(rec);
 507         }
 508
 509         return 0;
 510 }
 511
 512 static int finish_update_call(struct dyn_ftrace *rec, unsigned long addr)
 513 {
 514         unsigned long ip = rec->ip;
 515         unsigned const char *new;
 516
 517         new = ftrace_call_replace(ip, addr);
 518
 519         return ftrace_write(ip, new, 1);
 520 }
 521
 522 static int finish_update_nop(struct dyn_ftrace *rec)
 523 {
 524         unsigned long ip = rec->ip;
 525         unsigned const char *new;
 526
 527         new = ftrace_nop_replace();
 528
 529         return ftrace_write(ip, new, 1);
 530 }
 531
 532 static int finish_update(struct dyn_ftrace *rec, int enable)
 533 {
 534         unsigned long ftrace_addr;
 535         int ret;
 536
 537         ret = ftrace_update_record(rec, enable);
 538
 539         ftrace_addr = get_ftrace_addr(rec);
 540
 541         switch (ret) {
 542         case FTRACE_UPDATE_IGNORE:
 543                 return 0;
 544
 545         case FTRACE_UPDATE_MODIFY_CALL_REGS:
 546         case FTRACE_UPDATE_MODIFY_CALL:
 547         case FTRACE_UPDATE_MAKE_CALL:
 548                 /* converting nop to call */
 549                 return finish_update_call(rec, ftrace_addr);
 550
 551         case FTRACE_UPDATE_MAKE_NOP:
 552                 /* converting a call to a nop */
 553                 return finish_update_nop(rec);
 554         }
 555
 556         return 0;
 557 }
 558
 559 static void do_sync_core(void *data)
 560 {
 561         sync_core();
 562 }
 563
 564 static void run_sync(void)
 565 {
 566         int enable_irqs = irqs_disabled();
 567
 568         /* We may be called with interrupts disbled (on bootup). */
 569         if (enable_irqs)
 570                 local_irq_enable();
 571         on_each_cpu(do_sync_core, NULL, 1);
 572         if (enable_irqs)
 573                 local_irq_disable();
 574 }
 575
 576 void ftrace_replace_code(int enable)
 577 {
 578         struct ftrace_rec_iter *iter;
 579         struct dyn_ftrace *rec;
 580         const char *report = "adding breakpoints";
 581         int count = 0;
 582         int ret;
 583
 584         for_ftrace_rec_iter(iter) {
 585                 rec = ftrace_rec_iter_record(iter);
 586
 587                 ret = add_breakpoints(rec, enable);
 588                 if (ret)
 589                         goto remove_breakpoints;
 590                 count++;
 591         }
 592
 593         run_sync();
 594
 595         report = "updating code";
 596
 597         for_ftrace_rec_iter(iter) {
 598                 rec = ftrace_rec_iter_record(iter);
 599
 600                 ret = add_update(rec, enable);
 601                 if (ret)
 602                         goto remove_breakpoints;
 603         }
 604
 605         run_sync();
 606
 607         report = "removing breakpoints";
 608
 609         for_ftrace_rec_iter(iter) {
 610                 rec = ftrace_rec_iter_record(iter);
 611
 612                 ret = finish_update(rec, enable);
 613                 if (ret)
 614                         goto remove_breakpoints;
 615         }
 616
 617         run_sync();
 618
 619         return;
 620
 621  remove_breakpoints:
 622         pr_warn("Failed on %s (%d):\n", report, count);
 623         ftrace_bug(ret, rec ? rec->ip : 0);
 624         for_ftrace_rec_iter(iter) {
 625                 rec = ftrace_rec_iter_record(iter);
 626                 /*
 627                  * Breakpoints are handled only when this function is in
 628                  * progress. The system could not work with them.
 629                  */
 630                 if (remove_breakpoint(rec))
 631                         BUG();
 632         }
 633         run_sync();
 634 }
 635
 636 static int
 637 ftrace_modify_code(unsigned long ip, unsigned const char *old_code,
 638                    unsigned const char *new_code)
 639 {
 640         int ret;
 641
 642         ret = add_break(ip, old_code);
 643         if (ret)
 644                 goto out;
 645
 646         run_sync();
 647
 648         ret = add_update_code(ip, new_code);
 649         if (ret)
 650                 goto fail_update;
 651
 652         run_sync();
 653
 654         ret = ftrace_write(ip, new_code, 1);
 655         /*
 656          * The breakpoint is handled only when this function is in progress.
 657          * The system could not work if we could not remove it.
 658          */
 659         BUG_ON(ret);
 660  out:
 661         run_sync();
 662         return ret;
 663
 664  fail_update:
 665         /* Also here the system could not work with the breakpoint */
 666         if (ftrace_write(ip, old_code, 1))
 667                 BUG();
 668         goto out;
 669 }
 670
 671 void arch_ftrace_update_code(int command)
 672 {
 673         /* See comment above by declaration of modifying_ftrace_code */
 674         atomic_inc(&modifying_ftrace_code);
 675
 676         ftrace_modify_all_code(command);
 677
 678         atomic_dec(&modifying_ftrace_code);
 679 }
 680
 681 int __init ftrace_dyn_arch_init(void)
 682 {
 683         return 0;
 684 }
 685 #endif
 686
 687 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
 688
 689 #ifdef CONFIG_DYNAMIC_FTRACE
 690 extern void ftrace_graph_call(void);
 691
 692 static unsigned char *ftrace_jmp_replace(unsigned long ip, unsigned long addr)
 693 {
 694         static union ftrace_code_union calc;
 695
 696         /* Jmp not a call (ignore the .e8) */
 697         calc.e8         = 0xe9;
 698         calc.offset     = ftrace_calc_offset(ip + MCOUNT_INSN_SIZE, addr);
 699
 700         /*
 701          * ftrace external locks synchronize the access to the static variable.
 702          */
 703         return calc.code;
 704 }
 705
 706 static int ftrace_mod_jmp(unsigned long ip, void *func)
 707 {
 708         unsigned char *new;
 709
 710         new = ftrace_jmp_replace(ip, (unsigned long)func);
 711
 712         return update_ftrace_func(ip, new);
 713 }
 714
 715 int ftrace_enable_ftrace_graph_caller(void)
 716 {
 717         unsigned long ip = (unsigned long)(&ftrace_graph_call);
 718
 719         return ftrace_mod_jmp(ip, &ftrace_graph_caller);
 720 }
 721
 722 int ftrace_disable_ftrace_graph_caller(void)
 723 {
 724         unsigned long ip = (unsigned long)(&ftrace_graph_call);
 725
 726         return ftrace_mod_jmp(ip, &ftrace_stub);
 727 }
 728
 729 #endif /* !CONFIG_DYNAMIC_FTRACE */
 730
 731 /*
 732  * Hook the return address and push it in the stack of return addrs
 733  * in current thread info.
 734  */
 735 void prepare_ftrace_return(unsigned long *parent, unsigned long self_addr,
 736                            unsigned long frame_pointer)
 737 {
 738         unsigned long old;
 739         int faulted;
 740         struct ftrace_graph_ent trace;
 741         unsigned long return_hooker = (unsigned long)
 742                                 &return_to_handler;
 743
 744         if (unlikely(atomic_read(&current->tracing_graph_pause)))
 745                 return;
 746
 747         /*
 748          * Protect against fault, even if it shouldn't
 749          * happen. This tool is too much intrusive to
 750          * ignore such a protection.
 751          */
 752         asm volatile(
 753                 "1: " _ASM_MOV " (%[parent]), %[old]\n"
 754                 "2: " _ASM_MOV " %[return_hooker], (%[parent])\n"
 755                 "   movl $0, %[faulted]\n"
 756                 "3:\n"
 757
 758                 ".section .fixup, \"ax\"\n"
 759                 "4: movl $1, %[faulted]\n"
 760                 "   jmp 3b\n"
 761                 ".previous\n"
 762
 763                 _ASM_EXTABLE(1b, 4b)
 764                 _ASM_EXTABLE(2b, 4b)
 765
 766                 : [old] "=&r" (old), [faulted] "=r" (faulted)
 767                 : [parent] "r" (parent), [return_hooker] "r" (return_hooker)
 768                 : "memory"
 769         );
 770
 771         if (unlikely(faulted)) {
 772                 ftrace_graph_stop();
 773                 WARN_ON(1);
 774                 return;
 775         }
 776
 777         trace.func = self_addr;
 778         trace.depth = current->curr_ret_stack + 1;
 779
 780         /* Only trace if the calling function expects to */
 781         if (!ftrace_graph_entry(&trace)) {
 782                 *parent = old;
 783                 return;
 784         }
 785
 786         if (ftrace_push_return_trace(old, self_addr, &trace.depth,
 787                     frame_pointer) == -EBUSY) {
 788                 *parent = old;
 789                 return;
 790         }
 791 }
 792 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */