1 /* 2 * Infrastructure for profiling code inserted by 'gcc -pg'. 3 * 4 * Copyright (C) 2007-2008 Steven Rostedt <srostedt@redhat.com> 5 * Copyright (C) 2004-2008 Ingo Molnar <mingo@redhat.com> 6 * 7 * Originally ported from the -rt patch by: 8 * Copyright (C) 2007 Arnaldo Carvalho de Melo <acme@redhat.com> 9 * 10 * Based on code in the latency_tracer, that is: 11 * 12 * Copyright (C) 2004-2006 Ingo Molnar 13 * Copyright (C) 2004 Nadia Yvette Chambers 14 */ 15 16 #include <linux/stop_machine.h> 17 #include <linux/clocksource.h> 18 #include <linux/kallsyms.h> 19 #include <linux/seq_file.h> 20 #include <linux/suspend.h> 21 #include <linux/tracefs.h> 22 #include <linux/hardirq.h> 23 #include <linux/kthread.h> 24 #include <linux/uaccess.h> 25 #include <linux/bsearch.h> 26 #include <linux/module.h> 27 #include <linux/ftrace.h> 28 #include <linux/sysctl.h> 29 #include <linux/slab.h> 30 #include <linux/ctype.h> 31 #include <linux/sort.h> 32 #include <linux/list.h> 33 #include <linux/hash.h> 34 #include <linux/rcupdate.h> 35 36 #include <trace/events/sched.h> 37 38 #include <asm/setup.h> 39 40 #include "trace_output.h" 41 #include "trace_stat.h" 42 43 #define FTRACE_WARN_ON(cond) \ 44 ({ \ 45 int ___r = cond; \ 46 if (WARN_ON(___r)) \ 47 ftrace_kill(); \ 48 ___r; \ 49 }) 50 51 #define FTRACE_WARN_ON_ONCE(cond) \ 52 ({ \ 53 int ___r = cond; \ 54 if (WARN_ON_ONCE(___r)) \ 55 ftrace_kill(); \ 56 ___r; \ 57 }) 58 59 /* hash bits for specific function selection */ 60 #define FTRACE_HASH_BITS 7 61 #define FTRACE_FUNC_HASHSIZE (1 << FTRACE_HASH_BITS) 62 #define FTRACE_HASH_DEFAULT_BITS 10 63 #define FTRACE_HASH_MAX_BITS 12 64 65 #ifdef CONFIG_DYNAMIC_FTRACE 66 #define INIT_OPS_HASH(opsname) \ 67 .func_hash = &opsname.local_hash, \ 68 .local_hash.regex_lock = __MUTEX_INITIALIZER(opsname.local_hash.regex_lock), 69 #define ASSIGN_OPS_HASH(opsname, val) \ 70 .func_hash = val, \ 71 .local_hash.regex_lock = __MUTEX_INITIALIZER(opsname.local_hash.regex_lock), 72 #else 73 #define INIT_OPS_HASH(opsname) 74 #define ASSIGN_OPS_HASH(opsname, val) 75 #endif 76 77 static struct ftrace_ops ftrace_list_end __read_mostly = { 78 .func = ftrace_stub, 79 .flags = FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_STUB, 80 INIT_OPS_HASH(ftrace_list_end) 81 }; 82 83 /* ftrace_enabled is a method to turn ftrace on or off */ 84 int ftrace_enabled __read_mostly; 85 static int last_ftrace_enabled; 86 87 /* Current function tracing op */ 88 struct ftrace_ops *function_trace_op __read_mostly = &ftrace_list_end; 89 /* What to set function_trace_op to */ 90 static struct ftrace_ops *set_function_trace_op; 91 92 /* List for set_ftrace_pid's pids. */ 93 LIST_HEAD(ftrace_pids); 94 struct ftrace_pid { 95 struct list_head list; 96 struct pid *pid; 97 }; 98 99 static bool ftrace_pids_enabled(void) 100 { 101 return !list_empty(&ftrace_pids); 102 } 103 104 static void ftrace_update_trampoline(struct ftrace_ops *ops); 105 106 /* 107 * ftrace_disabled is set when an anomaly is discovered. 108 * ftrace_disabled is much stronger than ftrace_enabled. 109 */ 110 static int ftrace_disabled __read_mostly; 111 112 static DEFINE_MUTEX(ftrace_lock); 113 114 static struct ftrace_ops *ftrace_ops_list __read_mostly = &ftrace_list_end; 115 ftrace_func_t ftrace_trace_function __read_mostly = ftrace_stub; 116 static struct ftrace_ops global_ops; 117 118 #if ARCH_SUPPORTS_FTRACE_OPS 119 static void ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip, 120 struct ftrace_ops *op, struct pt_regs *regs); 121 #else 122 /* See comment below, where ftrace_ops_list_func is defined */ 123 static void ftrace_ops_no_ops(unsigned long ip, unsigned long parent_ip); 124 #define ftrace_ops_list_func ((ftrace_func_t)ftrace_ops_no_ops) 125 #endif 126 127 /* 128 * Traverse the ftrace_global_list, invoking all entries. The reason that we 129 * can use rcu_dereference_raw_notrace() is that elements removed from this list 130 * are simply leaked, so there is no need to interact with a grace-period 131 * mechanism. The rcu_dereference_raw_notrace() calls are needed to handle 132 * concurrent insertions into the ftrace_global_list. 133 * 134 * Silly Alpha and silly pointer-speculation compiler optimizations! 135 */ 136 #define do_for_each_ftrace_op(op, list) \ 137 op = rcu_dereference_raw_notrace(list); \ 138 do 139 140 /* 141 * Optimized for just a single item in the list (as that is the normal case). 142 */ 143 #define while_for_each_ftrace_op(op) \ 144 while (likely(op = rcu_dereference_raw_notrace((op)->next)) && \ 145 unlikely((op) != &ftrace_list_end)) 146 147 static inline void ftrace_ops_init(struct ftrace_ops *ops) 148 { 149 #ifdef CONFIG_DYNAMIC_FTRACE 150 if (!(ops->flags & FTRACE_OPS_FL_INITIALIZED)) { 151 mutex_init(&ops->local_hash.regex_lock); 152 ops->func_hash = &ops->local_hash; 153 ops->flags |= FTRACE_OPS_FL_INITIALIZED; 154 } 155 #endif 156 } 157 158 /** 159 * ftrace_nr_registered_ops - return number of ops registered 160 * 161 * Returns the number of ftrace_ops registered and tracing functions 162 */ 163 int ftrace_nr_registered_ops(void) 164 { 165 struct ftrace_ops *ops; 166 int cnt = 0; 167 168 mutex_lock(&ftrace_lock); 169 170 for (ops = ftrace_ops_list; 171 ops != &ftrace_list_end; ops = ops->next) 172 cnt++; 173 174 mutex_unlock(&ftrace_lock); 175 176 return cnt; 177 } 178 179 static void ftrace_pid_func(unsigned long ip, unsigned long parent_ip, 180 struct ftrace_ops *op, struct pt_regs *regs) 181 { 182 if (!test_tsk_trace_trace(current)) 183 return; 184 185 op->saved_func(ip, parent_ip, op, regs); 186 } 187 188 /** 189 * clear_ftrace_function - reset the ftrace function 190 * 191 * This NULLs the ftrace function and in essence stops 192 * tracing. There may be lag 193 */ 194 void clear_ftrace_function(void) 195 { 196 ftrace_trace_function = ftrace_stub; 197 } 198 199 static void per_cpu_ops_disable_all(struct ftrace_ops *ops) 200 { 201 int cpu; 202 203 for_each_possible_cpu(cpu) 204 *per_cpu_ptr(ops->disabled, cpu) = 1; 205 } 206 207 static int per_cpu_ops_alloc(struct ftrace_ops *ops) 208 { 209 int __percpu *disabled; 210 211 if (WARN_ON_ONCE(!(ops->flags & FTRACE_OPS_FL_PER_CPU))) 212 return -EINVAL; 213 214 disabled = alloc_percpu(int); 215 if (!disabled) 216 return -ENOMEM; 217 218 ops->disabled = disabled; 219 per_cpu_ops_disable_all(ops); 220 return 0; 221 } 222 223 static void ftrace_sync(struct work_struct *work) 224 { 225 /* 226 * This function is just a stub to implement a hard force 227 * of synchronize_sched(). This requires synchronizing 228 * tasks even in userspace and idle. 229 * 230 * Yes, function tracing is rude. 231 */ 232 } 233 234 static void ftrace_sync_ipi(void *data) 235 { 236 /* Probably not needed, but do it anyway */ 237 smp_rmb(); 238 } 239 240 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 241 static void update_function_graph_func(void); 242 243 /* Both enabled by default (can be cleared by function_graph tracer flags */ 244 static bool fgraph_sleep_time = true; 245 static bool fgraph_graph_time = true; 246 247 #else 248 static inline void update_function_graph_func(void) { } 249 #endif 250 251 252 static ftrace_func_t ftrace_ops_get_list_func(struct ftrace_ops *ops) 253 { 254 /* 255 * If this is a dynamic, RCU, or per CPU ops, or we force list func, 256 * then it needs to call the list anyway. 257 */ 258 if (ops->flags & (FTRACE_OPS_FL_DYNAMIC | FTRACE_OPS_FL_PER_CPU | 259 FTRACE_OPS_FL_RCU) || FTRACE_FORCE_LIST_FUNC) 260 return ftrace_ops_list_func; 261 262 return ftrace_ops_get_func(ops); 263 } 264 265 static void update_ftrace_function(void) 266 { 267 ftrace_func_t func; 268 269 /* 270 * Prepare the ftrace_ops that the arch callback will use. 271 * If there's only one ftrace_ops registered, the ftrace_ops_list 272 * will point to the ops we want. 273 */ 274 set_function_trace_op = ftrace_ops_list; 275 276 /* If there's no ftrace_ops registered, just call the stub function */ 277 if (ftrace_ops_list == &ftrace_list_end) { 278 func = ftrace_stub; 279 280 /* 281 * If we are at the end of the list and this ops is 282 * recursion safe and not dynamic and the arch supports passing ops, 283 * then have the mcount trampoline call the function directly. 284 */ 285 } else if (ftrace_ops_list->next == &ftrace_list_end) { 286 func = ftrace_ops_get_list_func(ftrace_ops_list); 287 288 } else { 289 /* Just use the default ftrace_ops */ 290 set_function_trace_op = &ftrace_list_end; 291 func = ftrace_ops_list_func; 292 } 293 294 update_function_graph_func(); 295 296 /* If there's no change, then do nothing more here */ 297 if (ftrace_trace_function == func) 298 return; 299 300 /* 301 * If we are using the list function, it doesn't care 302 * about the function_trace_ops. 303 */ 304 if (func == ftrace_ops_list_func) { 305 ftrace_trace_function = func; 306 /* 307 * Don't even bother setting function_trace_ops, 308 * it would be racy to do so anyway. 309 */ 310 return; 311 } 312 313 #ifndef CONFIG_DYNAMIC_FTRACE 314 /* 315 * For static tracing, we need to be a bit more careful. 316 * The function change takes affect immediately. Thus, 317 * we need to coorditate the setting of the function_trace_ops 318 * with the setting of the ftrace_trace_function. 319 * 320 * Set the function to the list ops, which will call the 321 * function we want, albeit indirectly, but it handles the 322 * ftrace_ops and doesn't depend on function_trace_op. 323 */ 324 ftrace_trace_function = ftrace_ops_list_func; 325 /* 326 * Make sure all CPUs see this. Yes this is slow, but static 327 * tracing is slow and nasty to have enabled. 328 */ 329 schedule_on_each_cpu(ftrace_sync); 330 /* Now all cpus are using the list ops. */ 331 function_trace_op = set_function_trace_op; 332 /* Make sure the function_trace_op is visible on all CPUs */ 333 smp_wmb(); 334 /* Nasty way to force a rmb on all cpus */ 335 smp_call_function(ftrace_sync_ipi, NULL, 1); 336 /* OK, we are all set to update the ftrace_trace_function now! */ 337 #endif /* !CONFIG_DYNAMIC_FTRACE */ 338 339 ftrace_trace_function = func; 340 } 341 342 int using_ftrace_ops_list_func(void) 343 { 344 return ftrace_trace_function == ftrace_ops_list_func; 345 } 346 347 static void add_ftrace_ops(struct ftrace_ops **list, struct ftrace_ops *ops) 348 { 349 ops->next = *list; 350 /* 351 * We are entering ops into the list but another 352 * CPU might be walking that list. We need to make sure 353 * the ops->next pointer is valid before another CPU sees 354 * the ops pointer included into the list. 355 */ 356 rcu_assign_pointer(*list, ops); 357 } 358 359 static int remove_ftrace_ops(struct ftrace_ops **list, struct ftrace_ops *ops) 360 { 361 struct ftrace_ops **p; 362 363 /* 364 * If we are removing the last function, then simply point 365 * to the ftrace_stub. 366 */ 367 if (*list == ops && ops->next == &ftrace_list_end) { 368 *list = &ftrace_list_end; 369 return 0; 370 } 371 372 for (p = list; *p != &ftrace_list_end; p = &(*p)->next) 373 if (*p == ops) 374 break; 375 376 if (*p != ops) 377 return -1; 378 379 *p = (*p)->next; 380 return 0; 381 } 382 383 static void ftrace_update_trampoline(struct ftrace_ops *ops); 384 385 static int __register_ftrace_function(struct ftrace_ops *ops) 386 { 387 if (ops->flags & FTRACE_OPS_FL_DELETED) 388 return -EINVAL; 389 390 if (WARN_ON(ops->flags & FTRACE_OPS_FL_ENABLED)) 391 return -EBUSY; 392 393 #ifndef CONFIG_DYNAMIC_FTRACE_WITH_REGS 394 /* 395 * If the ftrace_ops specifies SAVE_REGS, then it only can be used 396 * if the arch supports it, or SAVE_REGS_IF_SUPPORTED is also set. 397 * Setting SAVE_REGS_IF_SUPPORTED makes SAVE_REGS irrelevant. 398 */ 399 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS && 400 !(ops->flags & FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED)) 401 return -EINVAL; 402 403 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED) 404 ops->flags |= FTRACE_OPS_FL_SAVE_REGS; 405 #endif 406 407 if (!core_kernel_data((unsigned long)ops)) 408 ops->flags |= FTRACE_OPS_FL_DYNAMIC; 409 410 if (ops->flags & FTRACE_OPS_FL_PER_CPU) { 411 if (per_cpu_ops_alloc(ops)) 412 return -ENOMEM; 413 } 414 415 add_ftrace_ops(&ftrace_ops_list, ops); 416 417 /* Always save the function, and reset at unregistering */ 418 ops->saved_func = ops->func; 419 420 if (ops->flags & FTRACE_OPS_FL_PID && ftrace_pids_enabled()) 421 ops->func = ftrace_pid_func; 422 423 ftrace_update_trampoline(ops); 424 425 if (ftrace_enabled) 426 update_ftrace_function(); 427 428 return 0; 429 } 430 431 static int __unregister_ftrace_function(struct ftrace_ops *ops) 432 { 433 int ret; 434 435 if (WARN_ON(!(ops->flags & FTRACE_OPS_FL_ENABLED))) 436 return -EBUSY; 437 438 ret = remove_ftrace_ops(&ftrace_ops_list, ops); 439 440 if (ret < 0) 441 return ret; 442 443 if (ftrace_enabled) 444 update_ftrace_function(); 445 446 ops->func = ops->saved_func; 447 448 return 0; 449 } 450 451 static void ftrace_update_pid_func(void) 452 { 453 bool enabled = ftrace_pids_enabled(); 454 struct ftrace_ops *op; 455 456 /* Only do something if we are tracing something */ 457 if (ftrace_trace_function == ftrace_stub) 458 return; 459 460 do_for_each_ftrace_op(op, ftrace_ops_list) { 461 if (op->flags & FTRACE_OPS_FL_PID) { 462 op->func = enabled ? ftrace_pid_func : 463 op->saved_func; 464 ftrace_update_trampoline(op); 465 } 466 } while_for_each_ftrace_op(op); 467 468 update_ftrace_function(); 469 } 470 471 #ifdef CONFIG_FUNCTION_PROFILER 472 struct ftrace_profile { 473 struct hlist_node node; 474 unsigned long ip; 475 unsigned long counter; 476 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 477 unsigned long long time; 478 unsigned long long time_squared; 479 #endif 480 }; 481 482 struct ftrace_profile_page { 483 struct ftrace_profile_page *next; 484 unsigned long index; 485 struct ftrace_profile records[]; 486 }; 487 488 struct ftrace_profile_stat { 489 atomic_t disabled; 490 struct hlist_head *hash; 491 struct ftrace_profile_page *pages; 492 struct ftrace_profile_page *start; 493 struct tracer_stat stat; 494 }; 495 496 #define PROFILE_RECORDS_SIZE \ 497 (PAGE_SIZE - offsetof(struct ftrace_profile_page, records)) 498 499 #define PROFILES_PER_PAGE \ 500 (PROFILE_RECORDS_SIZE / sizeof(struct ftrace_profile)) 501 502 static int ftrace_profile_enabled __read_mostly; 503 504 /* ftrace_profile_lock - synchronize the enable and disable of the profiler */ 505 static DEFINE_MUTEX(ftrace_profile_lock); 506 507 static DEFINE_PER_CPU(struct ftrace_profile_stat, ftrace_profile_stats); 508 509 #define FTRACE_PROFILE_HASH_BITS 10 510 #define FTRACE_PROFILE_HASH_SIZE (1 << FTRACE_PROFILE_HASH_BITS) 511 512 static void * 513 function_stat_next(void *v, int idx) 514 { 515 struct ftrace_profile *rec = v; 516 struct ftrace_profile_page *pg; 517 518 pg = (struct ftrace_profile_page *)((unsigned long)rec & PAGE_MASK); 519 520 again: 521 if (idx != 0) 522 rec++; 523 524 if ((void *)rec >= (void *)&pg->records[pg->index]) { 525 pg = pg->next; 526 if (!pg) 527 return NULL; 528 rec = &pg->records[0]; 529 if (!rec->counter) 530 goto again; 531 } 532 533 return rec; 534 } 535 536 static void *function_stat_start(struct tracer_stat *trace) 537 { 538 struct ftrace_profile_stat *stat = 539 container_of(trace, struct ftrace_profile_stat, stat); 540 541 if (!stat || !stat->start) 542 return NULL; 543 544 return function_stat_next(&stat->start->records[0], 0); 545 } 546 547 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 548 /* function graph compares on total time */ 549 static int function_stat_cmp(void *p1, void *p2) 550 { 551 struct ftrace_profile *a = p1; 552 struct ftrace_profile *b = p2; 553 554 if (a->time < b->time) 555 return -1; 556 if (a->time > b->time) 557 return 1; 558 else 559 return 0; 560 } 561 #else 562 /* not function graph compares against hits */ 563 static int function_stat_cmp(void *p1, void *p2) 564 { 565 struct ftrace_profile *a = p1; 566 struct ftrace_profile *b = p2; 567 568 if (a->counter < b->counter) 569 return -1; 570 if (a->counter > b->counter) 571 return 1; 572 else 573 return 0; 574 } 575 #endif 576 577 static int function_stat_headers(struct seq_file *m) 578 { 579 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 580 seq_puts(m, " Function " 581 "Hit Time Avg s^2\n" 582 " -------- " 583 "--- ---- --- ---\n"); 584 #else 585 seq_puts(m, " Function Hit\n" 586 " -------- ---\n"); 587 #endif 588 return 0; 589 } 590 591 static int function_stat_show(struct seq_file *m, void *v) 592 { 593 struct ftrace_profile *rec = v; 594 char str[KSYM_SYMBOL_LEN]; 595 int ret = 0; 596 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 597 static struct trace_seq s; 598 unsigned long long avg; 599 unsigned long long stddev; 600 #endif 601 mutex_lock(&ftrace_profile_lock); 602 603 /* we raced with function_profile_reset() */ 604 if (unlikely(rec->counter == 0)) { 605 ret = -EBUSY; 606 goto out; 607 } 608 609 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 610 avg = rec->time; 611 do_div(avg, rec->counter); 612 if (tracing_thresh && (avg < tracing_thresh)) 613 goto out; 614 #endif 615 616 kallsyms_lookup(rec->ip, NULL, NULL, NULL, str); 617 seq_printf(m, " %-30.30s %10lu", str, rec->counter); 618 619 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 620 seq_puts(m, " "); 621 622 /* Sample standard deviation (s^2) */ 623 if (rec->counter <= 1) 624 stddev = 0; 625 else { 626 /* 627 * Apply Welford's method: 628 * s^2 = 1 / (n * (n-1)) * (n * \Sum (x_i)^2 - (\Sum x_i)^2) 629 */ 630 stddev = rec->counter * rec->time_squared - 631 rec->time * rec->time; 632 633 /* 634 * Divide only 1000 for ns^2 -> us^2 conversion. 635 * trace_print_graph_duration will divide 1000 again. 636 */ 637 do_div(stddev, rec->counter * (rec->counter - 1) * 1000); 638 } 639 640 trace_seq_init(&s); 641 trace_print_graph_duration(rec->time, &s); 642 trace_seq_puts(&s, " "); 643 trace_print_graph_duration(avg, &s); 644 trace_seq_puts(&s, " "); 645 trace_print_graph_duration(stddev, &s); 646 trace_print_seq(m, &s); 647 #endif 648 seq_putc(m, '\n'); 649 out: 650 mutex_unlock(&ftrace_profile_lock); 651 652 return ret; 653 } 654 655 static void ftrace_profile_reset(struct ftrace_profile_stat *stat) 656 { 657 struct ftrace_profile_page *pg; 658 659 pg = stat->pages = stat->start; 660 661 while (pg) { 662 memset(pg->records, 0, PROFILE_RECORDS_SIZE); 663 pg->index = 0; 664 pg = pg->next; 665 } 666 667 memset(stat->hash, 0, 668 FTRACE_PROFILE_HASH_SIZE * sizeof(struct hlist_head)); 669 } 670 671 int ftrace_profile_pages_init(struct ftrace_profile_stat *stat) 672 { 673 struct ftrace_profile_page *pg; 674 int functions; 675 int pages; 676 int i; 677 678 /* If we already allocated, do nothing */ 679 if (stat->pages) 680 return 0; 681 682 stat->pages = (void *)get_zeroed_page(GFP_KERNEL); 683 if (!stat->pages) 684 return -ENOMEM; 685 686 #ifdef CONFIG_DYNAMIC_FTRACE 687 functions = ftrace_update_tot_cnt; 688 #else 689 /* 690 * We do not know the number of functions that exist because 691 * dynamic tracing is what counts them. With past experience 692 * we have around 20K functions. That should be more than enough. 693 * It is highly unlikely we will execute every function in 694 * the kernel. 695 */ 696 functions = 20000; 697 #endif 698 699 pg = stat->start = stat->pages; 700 701 pages = DIV_ROUND_UP(functions, PROFILES_PER_PAGE); 702 703 for (i = 1; i < pages; i++) { 704 pg->next = (void *)get_zeroed_page(GFP_KERNEL); 705 if (!pg->next) 706 goto out_free; 707 pg = pg->next; 708 } 709 710 return 0; 711 712 out_free: 713 pg = stat->start; 714 while (pg) { 715 unsigned long tmp = (unsigned long)pg; 716 717 pg = pg->next; 718 free_page(tmp); 719 } 720 721 stat->pages = NULL; 722 stat->start = NULL; 723 724 return -ENOMEM; 725 } 726 727 static int ftrace_profile_init_cpu(int cpu) 728 { 729 struct ftrace_profile_stat *stat; 730 int size; 731 732 stat = &per_cpu(ftrace_profile_stats, cpu); 733 734 if (stat->hash) { 735 /* If the profile is already created, simply reset it */ 736 ftrace_profile_reset(stat); 737 return 0; 738 } 739 740 /* 741 * We are profiling all functions, but usually only a few thousand 742 * functions are hit. We'll make a hash of 1024 items. 743 */ 744 size = FTRACE_PROFILE_HASH_SIZE; 745 746 stat->hash = kzalloc(sizeof(struct hlist_head) * size, GFP_KERNEL); 747 748 if (!stat->hash) 749 return -ENOMEM; 750 751 /* Preallocate the function profiling pages */ 752 if (ftrace_profile_pages_init(stat) < 0) { 753 kfree(stat->hash); 754 stat->hash = NULL; 755 return -ENOMEM; 756 } 757 758 return 0; 759 } 760 761 static int ftrace_profile_init(void) 762 { 763 int cpu; 764 int ret = 0; 765 766 for_each_possible_cpu(cpu) { 767 ret = ftrace_profile_init_cpu(cpu); 768 if (ret) 769 break; 770 } 771 772 return ret; 773 } 774 775 /* interrupts must be disabled */ 776 static struct ftrace_profile * 777 ftrace_find_profiled_func(struct ftrace_profile_stat *stat, unsigned long ip) 778 { 779 struct ftrace_profile *rec; 780 struct hlist_head *hhd; 781 unsigned long key; 782 783 key = hash_long(ip, FTRACE_PROFILE_HASH_BITS); 784 hhd = &stat->hash[key]; 785 786 if (hlist_empty(hhd)) 787 return NULL; 788 789 hlist_for_each_entry_rcu_notrace(rec, hhd, node) { 790 if (rec->ip == ip) 791 return rec; 792 } 793 794 return NULL; 795 } 796 797 static void ftrace_add_profile(struct ftrace_profile_stat *stat, 798 struct ftrace_profile *rec) 799 { 800 unsigned long key; 801 802 key = hash_long(rec->ip, FTRACE_PROFILE_HASH_BITS); 803 hlist_add_head_rcu(&rec->node, &stat->hash[key]); 804 } 805 806 /* 807 * The memory is already allocated, this simply finds a new record to use. 808 */ 809 static struct ftrace_profile * 810 ftrace_profile_alloc(struct ftrace_profile_stat *stat, unsigned long ip) 811 { 812 struct ftrace_profile *rec = NULL; 813 814 /* prevent recursion (from NMIs) */ 815 if (atomic_inc_return(&stat->disabled) != 1) 816 goto out; 817 818 /* 819 * Try to find the function again since an NMI 820 * could have added it 821 */ 822 rec = ftrace_find_profiled_func(stat, ip); 823 if (rec) 824 goto out; 825 826 if (stat->pages->index == PROFILES_PER_PAGE) { 827 if (!stat->pages->next) 828 goto out; 829 stat->pages = stat->pages->next; 830 } 831 832 rec = &stat->pages->records[stat->pages->index++]; 833 rec->ip = ip; 834 ftrace_add_profile(stat, rec); 835 836 out: 837 atomic_dec(&stat->disabled); 838 839 return rec; 840 } 841 842 static void 843 function_profile_call(unsigned long ip, unsigned long parent_ip, 844 struct ftrace_ops *ops, struct pt_regs *regs) 845 { 846 struct ftrace_profile_stat *stat; 847 struct ftrace_profile *rec; 848 unsigned long flags; 849 850 if (!ftrace_profile_enabled) 851 return; 852 853 local_irq_save(flags); 854 855 stat = this_cpu_ptr(&ftrace_profile_stats); 856 if (!stat->hash || !ftrace_profile_enabled) 857 goto out; 858 859 rec = ftrace_find_profiled_func(stat, ip); 860 if (!rec) { 861 rec = ftrace_profile_alloc(stat, ip); 862 if (!rec) 863 goto out; 864 } 865 866 rec->counter++; 867 out: 868 local_irq_restore(flags); 869 } 870 871 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 872 static int profile_graph_entry(struct ftrace_graph_ent *trace) 873 { 874 function_profile_call(trace->func, 0, NULL, NULL); 875 return 1; 876 } 877 878 static void profile_graph_return(struct ftrace_graph_ret *trace) 879 { 880 struct ftrace_profile_stat *stat; 881 unsigned long long calltime; 882 struct ftrace_profile *rec; 883 unsigned long flags; 884 885 local_irq_save(flags); 886 stat = this_cpu_ptr(&ftrace_profile_stats); 887 if (!stat->hash || !ftrace_profile_enabled) 888 goto out; 889 890 /* If the calltime was zero'd ignore it */ 891 if (!trace->calltime) 892 goto out; 893 894 calltime = trace->rettime - trace->calltime; 895 896 if (!fgraph_graph_time) { 897 int index; 898 899 index = trace->depth; 900 901 /* Append this call time to the parent time to subtract */ 902 if (index) 903 current->ret_stack[index - 1].subtime += calltime; 904 905 if (current->ret_stack[index].subtime < calltime) 906 calltime -= current->ret_stack[index].subtime; 907 else 908 calltime = 0; 909 } 910 911 rec = ftrace_find_profiled_func(stat, trace->func); 912 if (rec) { 913 rec->time += calltime; 914 rec->time_squared += calltime * calltime; 915 } 916 917 out: 918 local_irq_restore(flags); 919 } 920 921 static int register_ftrace_profiler(void) 922 { 923 return register_ftrace_graph(&profile_graph_return, 924 &profile_graph_entry); 925 } 926 927 static void unregister_ftrace_profiler(void) 928 { 929 unregister_ftrace_graph(); 930 } 931 #else 932 static struct ftrace_ops ftrace_profile_ops __read_mostly = { 933 .func = function_profile_call, 934 .flags = FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_INITIALIZED, 935 INIT_OPS_HASH(ftrace_profile_ops) 936 }; 937 938 static int register_ftrace_profiler(void) 939 { 940 return register_ftrace_function(&ftrace_profile_ops); 941 } 942 943 static void unregister_ftrace_profiler(void) 944 { 945 unregister_ftrace_function(&ftrace_profile_ops); 946 } 947 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */ 948 949 static ssize_t 950 ftrace_profile_write(struct file *filp, const char __user *ubuf, 951 size_t cnt, loff_t *ppos) 952 { 953 unsigned long val; 954 int ret; 955 956 ret = kstrtoul_from_user(ubuf, cnt, 10, &val); 957 if (ret) 958 return ret; 959 960 val = !!val; 961 962 mutex_lock(&ftrace_profile_lock); 963 if (ftrace_profile_enabled ^ val) { 964 if (val) { 965 ret = ftrace_profile_init(); 966 if (ret < 0) { 967 cnt = ret; 968 goto out; 969 } 970 971 ret = register_ftrace_profiler(); 972 if (ret < 0) { 973 cnt = ret; 974 goto out; 975 } 976 ftrace_profile_enabled = 1; 977 } else { 978 ftrace_profile_enabled = 0; 979 /* 980 * unregister_ftrace_profiler calls stop_machine 981 * so this acts like an synchronize_sched. 982 */ 983 unregister_ftrace_profiler(); 984 } 985 } 986 out: 987 mutex_unlock(&ftrace_profile_lock); 988 989 *ppos += cnt; 990 991 return cnt; 992 } 993 994 static ssize_t 995 ftrace_profile_read(struct file *filp, char __user *ubuf, 996 size_t cnt, loff_t *ppos) 997 { 998 char buf[64]; /* big enough to hold a number */ 999 int r; 1000 1001 r = sprintf(buf, "%u\n", ftrace_profile_enabled); 1002 return simple_read_from_buffer(ubuf, cnt, ppos, buf, r); 1003 } 1004 1005 static const struct file_operations ftrace_profile_fops = { 1006 .open = tracing_open_generic, 1007 .read = ftrace_profile_read, 1008 .write = ftrace_profile_write, 1009 .llseek = default_llseek, 1010 }; 1011 1012 /* used to initialize the real stat files */ 1013 static struct tracer_stat function_stats __initdata = { 1014 .name = "functions", 1015 .stat_start = function_stat_start, 1016 .stat_next = function_stat_next, 1017 .stat_cmp = function_stat_cmp, 1018 .stat_headers = function_stat_headers, 1019 .stat_show = function_stat_show 1020 }; 1021 1022 static __init void ftrace_profile_tracefs(struct dentry *d_tracer) 1023 { 1024 struct ftrace_profile_stat *stat; 1025 struct dentry *entry; 1026 char *name; 1027 int ret; 1028 int cpu; 1029 1030 for_each_possible_cpu(cpu) { 1031 stat = &per_cpu(ftrace_profile_stats, cpu); 1032 1033 name = kasprintf(GFP_KERNEL, "function%d", cpu); 1034 if (!name) { 1035 /* 1036 * The files created are permanent, if something happens 1037 * we still do not free memory. 1038 */ 1039 WARN(1, 1040 "Could not allocate stat file for cpu %d\n", 1041 cpu); 1042 return; 1043 } 1044 stat->stat = function_stats; 1045 stat->stat.name = name; 1046 ret = register_stat_tracer(&stat->stat); 1047 if (ret) { 1048 WARN(1, 1049 "Could not register function stat for cpu %d\n", 1050 cpu); 1051 kfree(name); 1052 return; 1053 } 1054 } 1055 1056 entry = tracefs_create_file("function_profile_enabled", 0644, 1057 d_tracer, NULL, &ftrace_profile_fops); 1058 if (!entry) 1059 pr_warn("Could not create tracefs 'function_profile_enabled' entry\n"); 1060 } 1061 1062 #else /* CONFIG_FUNCTION_PROFILER */ 1063 static __init void ftrace_profile_tracefs(struct dentry *d_tracer) 1064 { 1065 } 1066 #endif /* CONFIG_FUNCTION_PROFILER */ 1067 1068 static struct pid * const ftrace_swapper_pid = &init_struct_pid; 1069 1070 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 1071 static int ftrace_graph_active; 1072 #else 1073 # define ftrace_graph_active 0 1074 #endif 1075 1076 #ifdef CONFIG_DYNAMIC_FTRACE 1077 1078 static struct ftrace_ops *removed_ops; 1079 1080 /* 1081 * Set when doing a global update, like enabling all recs or disabling them. 1082 * It is not set when just updating a single ftrace_ops. 1083 */ 1084 static bool update_all_ops; 1085 1086 #ifndef CONFIG_FTRACE_MCOUNT_RECORD 1087 # error Dynamic ftrace depends on MCOUNT_RECORD 1088 #endif 1089 1090 static struct hlist_head ftrace_func_hash[FTRACE_FUNC_HASHSIZE] __read_mostly; 1091 1092 struct ftrace_func_probe { 1093 struct hlist_node node; 1094 struct ftrace_probe_ops *ops; 1095 unsigned long flags; 1096 unsigned long ip; 1097 void *data; 1098 struct list_head free_list; 1099 }; 1100 1101 struct ftrace_func_entry { 1102 struct hlist_node hlist; 1103 unsigned long ip; 1104 }; 1105 1106 struct ftrace_hash { 1107 unsigned long size_bits; 1108 struct hlist_head *buckets; 1109 unsigned long count; 1110 struct rcu_head rcu; 1111 }; 1112 1113 /* 1114 * We make these constant because no one should touch them, 1115 * but they are used as the default "empty hash", to avoid allocating 1116 * it all the time. These are in a read only section such that if 1117 * anyone does try to modify it, it will cause an exception. 1118 */ 1119 static const struct hlist_head empty_buckets[1]; 1120 static const struct ftrace_hash empty_hash = { 1121 .buckets = (struct hlist_head *)empty_buckets, 1122 }; 1123 #define EMPTY_HASH ((struct ftrace_hash *)&empty_hash) 1124 1125 static struct ftrace_ops global_ops = { 1126 .func = ftrace_stub, 1127 .local_hash.notrace_hash = EMPTY_HASH, 1128 .local_hash.filter_hash = EMPTY_HASH, 1129 INIT_OPS_HASH(global_ops) 1130 .flags = FTRACE_OPS_FL_RECURSION_SAFE | 1131 FTRACE_OPS_FL_INITIALIZED | 1132 FTRACE_OPS_FL_PID, 1133 }; 1134 1135 /* 1136 * This is used by __kernel_text_address() to return true if the 1137 * address is on a dynamically allocated trampoline that would 1138 * not return true for either core_kernel_text() or 1139 * is_module_text_address(). 1140 */ 1141 bool is_ftrace_trampoline(unsigned long addr) 1142 { 1143 struct ftrace_ops *op; 1144 bool ret = false; 1145 1146 /* 1147 * Some of the ops may be dynamically allocated, 1148 * they are freed after a synchronize_sched(). 1149 */ 1150 preempt_disable_notrace(); 1151 1152 do_for_each_ftrace_op(op, ftrace_ops_list) { 1153 /* 1154 * This is to check for dynamically allocated trampolines. 1155 * Trampolines that are in kernel text will have 1156 * core_kernel_text() return true. 1157 */ 1158 if (op->trampoline && op->trampoline_size) 1159 if (addr >= op->trampoline && 1160 addr < op->trampoline + op->trampoline_size) { 1161 ret = true; 1162 goto out; 1163 } 1164 } while_for_each_ftrace_op(op); 1165 1166 out: 1167 preempt_enable_notrace(); 1168 1169 return ret; 1170 } 1171 1172 struct ftrace_page { 1173 struct ftrace_page *next; 1174 struct dyn_ftrace *records; 1175 int index; 1176 int size; 1177 }; 1178 1179 #define ENTRY_SIZE sizeof(struct dyn_ftrace) 1180 #define ENTRIES_PER_PAGE (PAGE_SIZE / ENTRY_SIZE) 1181 1182 /* estimate from running different kernels */ 1183 #define NR_TO_INIT 10000 1184 1185 static struct ftrace_page *ftrace_pages_start; 1186 static struct ftrace_page *ftrace_pages; 1187 1188 static bool __always_inline ftrace_hash_empty(struct ftrace_hash *hash) 1189 { 1190 return !hash || !hash->count; 1191 } 1192 1193 static struct ftrace_func_entry * 1194 ftrace_lookup_ip(struct ftrace_hash *hash, unsigned long ip) 1195 { 1196 unsigned long key; 1197 struct ftrace_func_entry *entry; 1198 struct hlist_head *hhd; 1199 1200 if (ftrace_hash_empty(hash)) 1201 return NULL; 1202 1203 if (hash->size_bits > 0) 1204 key = hash_long(ip, hash->size_bits); 1205 else 1206 key = 0; 1207 1208 hhd = &hash->buckets[key]; 1209 1210 hlist_for_each_entry_rcu_notrace(entry, hhd, hlist) { 1211 if (entry->ip == ip) 1212 return entry; 1213 } 1214 return NULL; 1215 } 1216 1217 static void __add_hash_entry(struct ftrace_hash *hash, 1218 struct ftrace_func_entry *entry) 1219 { 1220 struct hlist_head *hhd; 1221 unsigned long key; 1222 1223 if (hash->size_bits) 1224 key = hash_long(entry->ip, hash->size_bits); 1225 else 1226 key = 0; 1227 1228 hhd = &hash->buckets[key]; 1229 hlist_add_head(&entry->hlist, hhd); 1230 hash->count++; 1231 } 1232 1233 static int add_hash_entry(struct ftrace_hash *hash, unsigned long ip) 1234 { 1235 struct ftrace_func_entry *entry; 1236 1237 entry = kmalloc(sizeof(*entry), GFP_KERNEL); 1238 if (!entry) 1239 return -ENOMEM; 1240 1241 entry->ip = ip; 1242 __add_hash_entry(hash, entry); 1243 1244 return 0; 1245 } 1246 1247 static void 1248 free_hash_entry(struct ftrace_hash *hash, 1249 struct ftrace_func_entry *entry) 1250 { 1251 hlist_del(&entry->hlist); 1252 kfree(entry); 1253 hash->count--; 1254 } 1255 1256 static void 1257 remove_hash_entry(struct ftrace_hash *hash, 1258 struct ftrace_func_entry *entry) 1259 { 1260 hlist_del(&entry->hlist); 1261 hash->count--; 1262 } 1263 1264 static void ftrace_hash_clear(struct ftrace_hash *hash) 1265 { 1266 struct hlist_head *hhd; 1267 struct hlist_node *tn; 1268 struct ftrace_func_entry *entry; 1269 int size = 1 << hash->size_bits; 1270 int i; 1271 1272 if (!hash->count) 1273 return; 1274 1275 for (i = 0; i < size; i++) { 1276 hhd = &hash->buckets[i]; 1277 hlist_for_each_entry_safe(entry, tn, hhd, hlist) 1278 free_hash_entry(hash, entry); 1279 } 1280 FTRACE_WARN_ON(hash->count); 1281 } 1282 1283 static void free_ftrace_hash(struct ftrace_hash *hash) 1284 { 1285 if (!hash || hash == EMPTY_HASH) 1286 return; 1287 ftrace_hash_clear(hash); 1288 kfree(hash->buckets); 1289 kfree(hash); 1290 } 1291 1292 static void __free_ftrace_hash_rcu(struct rcu_head *rcu) 1293 { 1294 struct ftrace_hash *hash; 1295 1296 hash = container_of(rcu, struct ftrace_hash, rcu); 1297 free_ftrace_hash(hash); 1298 } 1299 1300 static void free_ftrace_hash_rcu(struct ftrace_hash *hash) 1301 { 1302 if (!hash || hash == EMPTY_HASH) 1303 return; 1304 call_rcu_sched(&hash->rcu, __free_ftrace_hash_rcu); 1305 } 1306 1307 void ftrace_free_filter(struct ftrace_ops *ops) 1308 { 1309 ftrace_ops_init(ops); 1310 free_ftrace_hash(ops->func_hash->filter_hash); 1311 free_ftrace_hash(ops->func_hash->notrace_hash); 1312 } 1313 1314 static struct ftrace_hash *alloc_ftrace_hash(int size_bits) 1315 { 1316 struct ftrace_hash *hash; 1317 int size; 1318 1319 hash = kzalloc(sizeof(*hash), GFP_KERNEL); 1320 if (!hash) 1321 return NULL; 1322 1323 size = 1 << size_bits; 1324 hash->buckets = kcalloc(size, sizeof(*hash->buckets), GFP_KERNEL); 1325 1326 if (!hash->buckets) { 1327 kfree(hash); 1328 return NULL; 1329 } 1330 1331 hash->size_bits = size_bits; 1332 1333 return hash; 1334 } 1335 1336 static struct ftrace_hash * 1337 alloc_and_copy_ftrace_hash(int size_bits, struct ftrace_hash *hash) 1338 { 1339 struct ftrace_func_entry *entry; 1340 struct ftrace_hash *new_hash; 1341 int size; 1342 int ret; 1343 int i; 1344 1345 new_hash = alloc_ftrace_hash(size_bits); 1346 if (!new_hash) 1347 return NULL; 1348 1349 /* Empty hash? */ 1350 if (ftrace_hash_empty(hash)) 1351 return new_hash; 1352 1353 size = 1 << hash->size_bits; 1354 for (i = 0; i < size; i++) { 1355 hlist_for_each_entry(entry, &hash->buckets[i], hlist) { 1356 ret = add_hash_entry(new_hash, entry->ip); 1357 if (ret < 0) 1358 goto free_hash; 1359 } 1360 } 1361 1362 FTRACE_WARN_ON(new_hash->count != hash->count); 1363 1364 return new_hash; 1365 1366 free_hash: 1367 free_ftrace_hash(new_hash); 1368 return NULL; 1369 } 1370 1371 static void 1372 ftrace_hash_rec_disable_modify(struct ftrace_ops *ops, int filter_hash); 1373 static void 1374 ftrace_hash_rec_enable_modify(struct ftrace_ops *ops, int filter_hash); 1375 1376 static int ftrace_hash_ipmodify_update(struct ftrace_ops *ops, 1377 struct ftrace_hash *new_hash); 1378 1379 static int 1380 ftrace_hash_move(struct ftrace_ops *ops, int enable, 1381 struct ftrace_hash **dst, struct ftrace_hash *src) 1382 { 1383 struct ftrace_func_entry *entry; 1384 struct hlist_node *tn; 1385 struct hlist_head *hhd; 1386 struct ftrace_hash *new_hash; 1387 int size = src->count; 1388 int bits = 0; 1389 int ret; 1390 int i; 1391 1392 /* Reject setting notrace hash on IPMODIFY ftrace_ops */ 1393 if (ops->flags & FTRACE_OPS_FL_IPMODIFY && !enable) 1394 return -EINVAL; 1395 1396 /* 1397 * If the new source is empty, just free dst and assign it 1398 * the empty_hash. 1399 */ 1400 if (!src->count) { 1401 new_hash = EMPTY_HASH; 1402 goto update; 1403 } 1404 1405 /* 1406 * Make the hash size about 1/2 the # found 1407 */ 1408 for (size /= 2; size; size >>= 1) 1409 bits++; 1410 1411 /* Don't allocate too much */ 1412 if (bits > FTRACE_HASH_MAX_BITS) 1413 bits = FTRACE_HASH_MAX_BITS; 1414 1415 new_hash = alloc_ftrace_hash(bits); 1416 if (!new_hash) 1417 return -ENOMEM; 1418 1419 size = 1 << src->size_bits; 1420 for (i = 0; i < size; i++) { 1421 hhd = &src->buckets[i]; 1422 hlist_for_each_entry_safe(entry, tn, hhd, hlist) { 1423 remove_hash_entry(src, entry); 1424 __add_hash_entry(new_hash, entry); 1425 } 1426 } 1427 1428 update: 1429 /* Make sure this can be applied if it is IPMODIFY ftrace_ops */ 1430 if (enable) { 1431 /* IPMODIFY should be updated only when filter_hash updating */ 1432 ret = ftrace_hash_ipmodify_update(ops, new_hash); 1433 if (ret < 0) { 1434 free_ftrace_hash(new_hash); 1435 return ret; 1436 } 1437 } 1438 1439 /* 1440 * Remove the current set, update the hash and add 1441 * them back. 1442 */ 1443 ftrace_hash_rec_disable_modify(ops, enable); 1444 1445 rcu_assign_pointer(*dst, new_hash); 1446 1447 ftrace_hash_rec_enable_modify(ops, enable); 1448 1449 return 0; 1450 } 1451 1452 static bool hash_contains_ip(unsigned long ip, 1453 struct ftrace_ops_hash *hash) 1454 { 1455 /* 1456 * The function record is a match if it exists in the filter 1457 * hash and not in the notrace hash. Note, an emty hash is 1458 * considered a match for the filter hash, but an empty 1459 * notrace hash is considered not in the notrace hash. 1460 */ 1461 return (ftrace_hash_empty(hash->filter_hash) || 1462 ftrace_lookup_ip(hash->filter_hash, ip)) && 1463 (ftrace_hash_empty(hash->notrace_hash) || 1464 !ftrace_lookup_ip(hash->notrace_hash, ip)); 1465 } 1466 1467 /* 1468 * Test the hashes for this ops to see if we want to call 1469 * the ops->func or not. 1470 * 1471 * It's a match if the ip is in the ops->filter_hash or 1472 * the filter_hash does not exist or is empty, 1473 * AND 1474 * the ip is not in the ops->notrace_hash. 1475 * 1476 * This needs to be called with preemption disabled as 1477 * the hashes are freed with call_rcu_sched(). 1478 */ 1479 static int 1480 ftrace_ops_test(struct ftrace_ops *ops, unsigned long ip, void *regs) 1481 { 1482 struct ftrace_ops_hash hash; 1483 int ret; 1484 1485 #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS 1486 /* 1487 * There's a small race when adding ops that the ftrace handler 1488 * that wants regs, may be called without them. We can not 1489 * allow that handler to be called if regs is NULL. 1490 */ 1491 if (regs == NULL && (ops->flags & FTRACE_OPS_FL_SAVE_REGS)) 1492 return 0; 1493 #endif 1494 1495 hash.filter_hash = rcu_dereference_raw_notrace(ops->func_hash->filter_hash); 1496 hash.notrace_hash = rcu_dereference_raw_notrace(ops->func_hash->notrace_hash); 1497 1498 if (hash_contains_ip(ip, &hash)) 1499 ret = 1; 1500 else 1501 ret = 0; 1502 1503 return ret; 1504 } 1505 1506 /* 1507 * This is a double for. Do not use 'break' to break out of the loop, 1508 * you must use a goto. 1509 */ 1510 #define do_for_each_ftrace_rec(pg, rec) \ 1511 for (pg = ftrace_pages_start; pg; pg = pg->next) { \ 1512 int _____i; \ 1513 for (_____i = 0; _____i < pg->index; _____i++) { \ 1514 rec = &pg->records[_____i]; 1515 1516 #define while_for_each_ftrace_rec() \ 1517 } \ 1518 } 1519 1520 1521 static int ftrace_cmp_recs(const void *a, const void *b) 1522 { 1523 const struct dyn_ftrace *key = a; 1524 const struct dyn_ftrace *rec = b; 1525 1526 if (key->flags < rec->ip) 1527 return -1; 1528 if (key->ip >= rec->ip + MCOUNT_INSN_SIZE) 1529 return 1; 1530 return 0; 1531 } 1532 1533 /** 1534 * ftrace_location_range - return the first address of a traced location 1535 * if it touches the given ip range 1536 * @start: start of range to search. 1537 * @end: end of range to search (inclusive). @end points to the last byte 1538 * to check. 1539 * 1540 * Returns rec->ip if the related ftrace location is a least partly within 1541 * the given address range. That is, the first address of the instruction 1542 * that is either a NOP or call to the function tracer. It checks the ftrace 1543 * internal tables to determine if the address belongs or not. 1544 */ 1545 unsigned long ftrace_location_range(unsigned long start, unsigned long end) 1546 { 1547 struct ftrace_page *pg; 1548 struct dyn_ftrace *rec; 1549 struct dyn_ftrace key; 1550 1551 key.ip = start; 1552 key.flags = end; /* overload flags, as it is unsigned long */ 1553 1554 for (pg = ftrace_pages_start; pg; pg = pg->next) { 1555 if (end < pg->records[0].ip || 1556 start >= (pg->records[pg->index - 1].ip + MCOUNT_INSN_SIZE)) 1557 continue; 1558 rec = bsearch(&key, pg->records, pg->index, 1559 sizeof(struct dyn_ftrace), 1560 ftrace_cmp_recs); 1561 if (rec) 1562 return rec->ip; 1563 } 1564 1565 return 0; 1566 } 1567 1568 /** 1569 * ftrace_location - return true if the ip giving is a traced location 1570 * @ip: the instruction pointer to check 1571 * 1572 * Returns rec->ip if @ip given is a pointer to a ftrace location. 1573 * That is, the instruction that is either a NOP or call to 1574 * the function tracer. It checks the ftrace internal tables to 1575 * determine if the address belongs or not. 1576 */ 1577 unsigned long ftrace_location(unsigned long ip) 1578 { 1579 return ftrace_location_range(ip, ip); 1580 } 1581 1582 /** 1583 * ftrace_text_reserved - return true if range contains an ftrace location 1584 * @start: start of range to search 1585 * @end: end of range to search (inclusive). @end points to the last byte to check. 1586 * 1587 * Returns 1 if @start and @end contains a ftrace location. 1588 * That is, the instruction that is either a NOP or call to 1589 * the function tracer. It checks the ftrace internal tables to 1590 * determine if the address belongs or not. 1591 */ 1592 int ftrace_text_reserved(const void *start, const void *end) 1593 { 1594 unsigned long ret; 1595 1596 ret = ftrace_location_range((unsigned long)start, 1597 (unsigned long)end); 1598 1599 return (int)!!ret; 1600 } 1601 1602 /* Test if ops registered to this rec needs regs */ 1603 static bool test_rec_ops_needs_regs(struct dyn_ftrace *rec) 1604 { 1605 struct ftrace_ops *ops; 1606 bool keep_regs = false; 1607 1608 for (ops = ftrace_ops_list; 1609 ops != &ftrace_list_end; ops = ops->next) { 1610 /* pass rec in as regs to have non-NULL val */ 1611 if (ftrace_ops_test(ops, rec->ip, rec)) { 1612 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS) { 1613 keep_regs = true; 1614 break; 1615 } 1616 } 1617 } 1618 1619 return keep_regs; 1620 } 1621 1622 static bool __ftrace_hash_rec_update(struct ftrace_ops *ops, 1623 int filter_hash, 1624 bool inc) 1625 { 1626 struct ftrace_hash *hash; 1627 struct ftrace_hash *other_hash; 1628 struct ftrace_page *pg; 1629 struct dyn_ftrace *rec; 1630 bool update = false; 1631 int count = 0; 1632 int all = 0; 1633 1634 /* Only update if the ops has been registered */ 1635 if (!(ops->flags & FTRACE_OPS_FL_ENABLED)) 1636 return false; 1637 1638 /* 1639 * In the filter_hash case: 1640 * If the count is zero, we update all records. 1641 * Otherwise we just update the items in the hash. 1642 * 1643 * In the notrace_hash case: 1644 * We enable the update in the hash. 1645 * As disabling notrace means enabling the tracing, 1646 * and enabling notrace means disabling, the inc variable 1647 * gets inversed. 1648 */ 1649 if (filter_hash) { 1650 hash = ops->func_hash->filter_hash; 1651 other_hash = ops->func_hash->notrace_hash; 1652 if (ftrace_hash_empty(hash)) 1653 all = 1; 1654 } else { 1655 inc = !inc; 1656 hash = ops->func_hash->notrace_hash; 1657 other_hash = ops->func_hash->filter_hash; 1658 /* 1659 * If the notrace hash has no items, 1660 * then there's nothing to do. 1661 */ 1662 if (ftrace_hash_empty(hash)) 1663 return false; 1664 } 1665 1666 do_for_each_ftrace_rec(pg, rec) { 1667 int in_other_hash = 0; 1668 int in_hash = 0; 1669 int match = 0; 1670 1671 if (rec->flags & FTRACE_FL_DISABLED) 1672 continue; 1673 1674 if (all) { 1675 /* 1676 * Only the filter_hash affects all records. 1677 * Update if the record is not in the notrace hash. 1678 */ 1679 if (!other_hash || !ftrace_lookup_ip(other_hash, rec->ip)) 1680 match = 1; 1681 } else { 1682 in_hash = !!ftrace_lookup_ip(hash, rec->ip); 1683 in_other_hash = !!ftrace_lookup_ip(other_hash, rec->ip); 1684 1685 /* 1686 * If filter_hash is set, we want to match all functions 1687 * that are in the hash but not in the other hash. 1688 * 1689 * If filter_hash is not set, then we are decrementing. 1690 * That means we match anything that is in the hash 1691 * and also in the other_hash. That is, we need to turn 1692 * off functions in the other hash because they are disabled 1693 * by this hash. 1694 */ 1695 if (filter_hash && in_hash && !in_other_hash) 1696 match = 1; 1697 else if (!filter_hash && in_hash && 1698 (in_other_hash || ftrace_hash_empty(other_hash))) 1699 match = 1; 1700 } 1701 if (!match) 1702 continue; 1703 1704 if (inc) { 1705 rec->flags++; 1706 if (FTRACE_WARN_ON(ftrace_rec_count(rec) == FTRACE_REF_MAX)) 1707 return false; 1708 1709 /* 1710 * If there's only a single callback registered to a 1711 * function, and the ops has a trampoline registered 1712 * for it, then we can call it directly. 1713 */ 1714 if (ftrace_rec_count(rec) == 1 && ops->trampoline) 1715 rec->flags |= FTRACE_FL_TRAMP; 1716 else 1717 /* 1718 * If we are adding another function callback 1719 * to this function, and the previous had a 1720 * custom trampoline in use, then we need to go 1721 * back to the default trampoline. 1722 */ 1723 rec->flags &= ~FTRACE_FL_TRAMP; 1724 1725 /* 1726 * If any ops wants regs saved for this function 1727 * then all ops will get saved regs. 1728 */ 1729 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS) 1730 rec->flags |= FTRACE_FL_REGS; 1731 } else { 1732 if (FTRACE_WARN_ON(ftrace_rec_count(rec) == 0)) 1733 return false; 1734 rec->flags--; 1735 1736 /* 1737 * If the rec had REGS enabled and the ops that is 1738 * being removed had REGS set, then see if there is 1739 * still any ops for this record that wants regs. 1740 * If not, we can stop recording them. 1741 */ 1742 if (ftrace_rec_count(rec) > 0 && 1743 rec->flags & FTRACE_FL_REGS && 1744 ops->flags & FTRACE_OPS_FL_SAVE_REGS) { 1745 if (!test_rec_ops_needs_regs(rec)) 1746 rec->flags &= ~FTRACE_FL_REGS; 1747 } 1748 1749 /* 1750 * If the rec had TRAMP enabled, then it needs to 1751 * be cleared. As TRAMP can only be enabled iff 1752 * there is only a single ops attached to it. 1753 * In otherwords, always disable it on decrementing. 1754 * In the future, we may set it if rec count is 1755 * decremented to one, and the ops that is left 1756 * has a trampoline. 1757 */ 1758 rec->flags &= ~FTRACE_FL_TRAMP; 1759 1760 /* 1761 * flags will be cleared in ftrace_check_record() 1762 * if rec count is zero. 1763 */ 1764 } 1765 count++; 1766 1767 /* Must match FTRACE_UPDATE_CALLS in ftrace_modify_all_code() */ 1768 update |= ftrace_test_record(rec, 1) != FTRACE_UPDATE_IGNORE; 1769 1770 /* Shortcut, if we handled all records, we are done. */ 1771 if (!all && count == hash->count) 1772 return update; 1773 } while_for_each_ftrace_rec(); 1774 1775 return update; 1776 } 1777 1778 static bool ftrace_hash_rec_disable(struct ftrace_ops *ops, 1779 int filter_hash) 1780 { 1781 return __ftrace_hash_rec_update(ops, filter_hash, 0); 1782 } 1783 1784 static bool ftrace_hash_rec_enable(struct ftrace_ops *ops, 1785 int filter_hash) 1786 { 1787 return __ftrace_hash_rec_update(ops, filter_hash, 1); 1788 } 1789 1790 static void ftrace_hash_rec_update_modify(struct ftrace_ops *ops, 1791 int filter_hash, int inc) 1792 { 1793 struct ftrace_ops *op; 1794 1795 __ftrace_hash_rec_update(ops, filter_hash, inc); 1796 1797 if (ops->func_hash != &global_ops.local_hash) 1798 return; 1799 1800 /* 1801 * If the ops shares the global_ops hash, then we need to update 1802 * all ops that are enabled and use this hash. 1803 */ 1804 do_for_each_ftrace_op(op, ftrace_ops_list) { 1805 /* Already done */ 1806 if (op == ops) 1807 continue; 1808 if (op->func_hash == &global_ops.local_hash) 1809 __ftrace_hash_rec_update(op, filter_hash, inc); 1810 } while_for_each_ftrace_op(op); 1811 } 1812 1813 static void ftrace_hash_rec_disable_modify(struct ftrace_ops *ops, 1814 int filter_hash) 1815 { 1816 ftrace_hash_rec_update_modify(ops, filter_hash, 0); 1817 } 1818 1819 static void ftrace_hash_rec_enable_modify(struct ftrace_ops *ops, 1820 int filter_hash) 1821 { 1822 ftrace_hash_rec_update_modify(ops, filter_hash, 1); 1823 } 1824 1825 /* 1826 * Try to update IPMODIFY flag on each ftrace_rec. Return 0 if it is OK 1827 * or no-needed to update, -EBUSY if it detects a conflict of the flag 1828 * on a ftrace_rec, and -EINVAL if the new_hash tries to trace all recs. 1829 * Note that old_hash and new_hash has below meanings 1830 * - If the hash is NULL, it hits all recs (if IPMODIFY is set, this is rejected) 1831 * - If the hash is EMPTY_HASH, it hits nothing 1832 * - Anything else hits the recs which match the hash entries. 1833 */ 1834 static int __ftrace_hash_update_ipmodify(struct ftrace_ops *ops, 1835 struct ftrace_hash *old_hash, 1836 struct ftrace_hash *new_hash) 1837 { 1838 struct ftrace_page *pg; 1839 struct dyn_ftrace *rec, *end = NULL; 1840 int in_old, in_new; 1841 1842 /* Only update if the ops has been registered */ 1843 if (!(ops->flags & FTRACE_OPS_FL_ENABLED)) 1844 return 0; 1845 1846 if (!(ops->flags & FTRACE_OPS_FL_IPMODIFY)) 1847 return 0; 1848 1849 /* 1850 * Since the IPMODIFY is a very address sensitive action, we do not 1851 * allow ftrace_ops to set all functions to new hash. 1852 */ 1853 if (!new_hash || !old_hash) 1854 return -EINVAL; 1855 1856 /* Update rec->flags */ 1857 do_for_each_ftrace_rec(pg, rec) { 1858 /* We need to update only differences of filter_hash */ 1859 in_old = !!ftrace_lookup_ip(old_hash, rec->ip); 1860 in_new = !!ftrace_lookup_ip(new_hash, rec->ip); 1861 if (in_old == in_new) 1862 continue; 1863 1864 if (in_new) { 1865 /* New entries must ensure no others are using it */ 1866 if (rec->flags & FTRACE_FL_IPMODIFY) 1867 goto rollback; 1868 rec->flags |= FTRACE_FL_IPMODIFY; 1869 } else /* Removed entry */ 1870 rec->flags &= ~FTRACE_FL_IPMODIFY; 1871 } while_for_each_ftrace_rec(); 1872 1873 return 0; 1874 1875 rollback: 1876 end = rec; 1877 1878 /* Roll back what we did above */ 1879 do_for_each_ftrace_rec(pg, rec) { 1880 if (rec == end) 1881 goto err_out; 1882 1883 in_old = !!ftrace_lookup_ip(old_hash, rec->ip); 1884 in_new = !!ftrace_lookup_ip(new_hash, rec->ip); 1885 if (in_old == in_new) 1886 continue; 1887 1888 if (in_new) 1889 rec->flags &= ~FTRACE_FL_IPMODIFY; 1890 else 1891 rec->flags |= FTRACE_FL_IPMODIFY; 1892 } while_for_each_ftrace_rec(); 1893 1894 err_out: 1895 return -EBUSY; 1896 } 1897 1898 static int ftrace_hash_ipmodify_enable(struct ftrace_ops *ops) 1899 { 1900 struct ftrace_hash *hash = ops->func_hash->filter_hash; 1901 1902 if (ftrace_hash_empty(hash)) 1903 hash = NULL; 1904 1905 return __ftrace_hash_update_ipmodify(ops, EMPTY_HASH, hash); 1906 } 1907 1908 /* Disabling always succeeds */ 1909 static void ftrace_hash_ipmodify_disable(struct ftrace_ops *ops) 1910 { 1911 struct ftrace_hash *hash = ops->func_hash->filter_hash; 1912 1913 if (ftrace_hash_empty(hash)) 1914 hash = NULL; 1915 1916 __ftrace_hash_update_ipmodify(ops, hash, EMPTY_HASH); 1917 } 1918 1919 static int ftrace_hash_ipmodify_update(struct ftrace_ops *ops, 1920 struct ftrace_hash *new_hash) 1921 { 1922 struct ftrace_hash *old_hash = ops->func_hash->filter_hash; 1923 1924 if (ftrace_hash_empty(old_hash)) 1925 old_hash = NULL; 1926 1927 if (ftrace_hash_empty(new_hash)) 1928 new_hash = NULL; 1929 1930 return __ftrace_hash_update_ipmodify(ops, old_hash, new_hash); 1931 } 1932 1933 static void print_ip_ins(const char *fmt, const unsigned char *p) 1934 { 1935 int i; 1936 1937 printk(KERN_CONT "%s", fmt); 1938 1939 for (i = 0; i < MCOUNT_INSN_SIZE; i++) 1940 printk(KERN_CONT "%s%02x", i ? ":" : "", p[i]); 1941 } 1942 1943 static struct ftrace_ops * 1944 ftrace_find_tramp_ops_any(struct dyn_ftrace *rec); 1945 static struct ftrace_ops * 1946 ftrace_find_tramp_ops_next(struct dyn_ftrace *rec, struct ftrace_ops *ops); 1947 1948 enum ftrace_bug_type ftrace_bug_type; 1949 const void *ftrace_expected; 1950 1951 static void print_bug_type(void) 1952 { 1953 switch (ftrace_bug_type) { 1954 case FTRACE_BUG_UNKNOWN: 1955 break; 1956 case FTRACE_BUG_INIT: 1957 pr_info("Initializing ftrace call sites\n"); 1958 break; 1959 case FTRACE_BUG_NOP: 1960 pr_info("Setting ftrace call site to NOP\n"); 1961 break; 1962 case FTRACE_BUG_CALL: 1963 pr_info("Setting ftrace call site to call ftrace function\n"); 1964 break; 1965 case FTRACE_BUG_UPDATE: 1966 pr_info("Updating ftrace call site to call a different ftrace function\n"); 1967 break; 1968 } 1969 } 1970 1971 /** 1972 * ftrace_bug - report and shutdown function tracer 1973 * @failed: The failed type (EFAULT, EINVAL, EPERM) 1974 * @rec: The record that failed 1975 * 1976 * The arch code that enables or disables the function tracing 1977 * can call ftrace_bug() when it has detected a problem in 1978 * modifying the code. @failed should be one of either: 1979 * EFAULT - if the problem happens on reading the @ip address 1980 * EINVAL - if what is read at @ip is not what was expected 1981 * EPERM - if the problem happens on writting to the @ip address 1982 */ 1983 void ftrace_bug(int failed, struct dyn_ftrace *rec) 1984 { 1985 unsigned long ip = rec ? rec->ip : 0; 1986 1987 switch (failed) { 1988 case -EFAULT: 1989 FTRACE_WARN_ON_ONCE(1); 1990 pr_info("ftrace faulted on modifying "); 1991 print_ip_sym(ip); 1992 break; 1993 case -EINVAL: 1994 FTRACE_WARN_ON_ONCE(1); 1995 pr_info("ftrace failed to modify "); 1996 print_ip_sym(ip); 1997 print_ip_ins(" actual: ", (unsigned char *)ip); 1998 pr_cont("\n"); 1999 if (ftrace_expected) { 2000 print_ip_ins(" expected: ", ftrace_expected); 2001 pr_cont("\n"); 2002 } 2003 break; 2004 case -EPERM: 2005 FTRACE_WARN_ON_ONCE(1); 2006 pr_info("ftrace faulted on writing "); 2007 print_ip_sym(ip); 2008 break; 2009 default: 2010 FTRACE_WARN_ON_ONCE(1); 2011 pr_info("ftrace faulted on unknown error "); 2012 print_ip_sym(ip); 2013 } 2014 print_bug_type(); 2015 if (rec) { 2016 struct ftrace_ops *ops = NULL; 2017 2018 pr_info("ftrace record flags: %lx\n", rec->flags); 2019 pr_cont(" (%ld)%s", ftrace_rec_count(rec), 2020 rec->flags & FTRACE_FL_REGS ? " R" : " "); 2021 if (rec->flags & FTRACE_FL_TRAMP_EN) { 2022 ops = ftrace_find_tramp_ops_any(rec); 2023 if (ops) { 2024 do { 2025 pr_cont("\ttramp: %pS (%pS)", 2026 (void *)ops->trampoline, 2027 (void *)ops->func); 2028 ops = ftrace_find_tramp_ops_next(rec, ops); 2029 } while (ops); 2030 } else 2031 pr_cont("\ttramp: ERROR!"); 2032 2033 } 2034 ip = ftrace_get_addr_curr(rec); 2035 pr_cont("\n expected tramp: %lx\n", ip); 2036 } 2037 } 2038 2039 static int ftrace_check_record(struct dyn_ftrace *rec, int enable, int update) 2040 { 2041 unsigned long flag = 0UL; 2042 2043 ftrace_bug_type = FTRACE_BUG_UNKNOWN; 2044 2045 if (rec->flags & FTRACE_FL_DISABLED) 2046 return FTRACE_UPDATE_IGNORE; 2047 2048 /* 2049 * If we are updating calls: 2050 * 2051 * If the record has a ref count, then we need to enable it 2052 * because someone is using it. 2053 * 2054 * Otherwise we make sure its disabled. 2055 * 2056 * If we are disabling calls, then disable all records that 2057 * are enabled. 2058 */ 2059 if (enable && ftrace_rec_count(rec)) 2060 flag = FTRACE_FL_ENABLED; 2061 2062 /* 2063 * If enabling and the REGS flag does not match the REGS_EN, or 2064 * the TRAMP flag doesn't match the TRAMP_EN, then do not ignore 2065 * this record. Set flags to fail the compare against ENABLED. 2066 */ 2067 if (flag) { 2068 if (!(rec->flags & FTRACE_FL_REGS) != 2069 !(rec->flags & FTRACE_FL_REGS_EN)) 2070 flag |= FTRACE_FL_REGS; 2071 2072 if (!(rec->flags & FTRACE_FL_TRAMP) != 2073 !(rec->flags & FTRACE_FL_TRAMP_EN)) 2074 flag |= FTRACE_FL_TRAMP; 2075 } 2076 2077 /* If the state of this record hasn't changed, then do nothing */ 2078 if ((rec->flags & FTRACE_FL_ENABLED) == flag) 2079 return FTRACE_UPDATE_IGNORE; 2080 2081 if (flag) { 2082 /* Save off if rec is being enabled (for return value) */ 2083 flag ^= rec->flags & FTRACE_FL_ENABLED; 2084 2085 if (update) { 2086 rec->flags |= FTRACE_FL_ENABLED; 2087 if (flag & FTRACE_FL_REGS) { 2088 if (rec->flags & FTRACE_FL_REGS) 2089 rec->flags |= FTRACE_FL_REGS_EN; 2090 else 2091 rec->flags &= ~FTRACE_FL_REGS_EN; 2092 } 2093 if (flag & FTRACE_FL_TRAMP) { 2094 if (rec->flags & FTRACE_FL_TRAMP) 2095 rec->flags |= FTRACE_FL_TRAMP_EN; 2096 else 2097 rec->flags &= ~FTRACE_FL_TRAMP_EN; 2098 } 2099 } 2100 2101 /* 2102 * If this record is being updated from a nop, then 2103 * return UPDATE_MAKE_CALL. 2104 * Otherwise, 2105 * return UPDATE_MODIFY_CALL to tell the caller to convert 2106 * from the save regs, to a non-save regs function or 2107 * vice versa, or from a trampoline call. 2108 */ 2109 if (flag & FTRACE_FL_ENABLED) { 2110 ftrace_bug_type = FTRACE_BUG_CALL; 2111 return FTRACE_UPDATE_MAKE_CALL; 2112 } 2113 2114 ftrace_bug_type = FTRACE_BUG_UPDATE; 2115 return FTRACE_UPDATE_MODIFY_CALL; 2116 } 2117 2118 if (update) { 2119 /* If there's no more users, clear all flags */ 2120 if (!ftrace_rec_count(rec)) 2121 rec->flags = 0; 2122 else 2123 /* 2124 * Just disable the record, but keep the ops TRAMP 2125 * and REGS states. The _EN flags must be disabled though. 2126 */ 2127 rec->flags &= ~(FTRACE_FL_ENABLED | FTRACE_FL_TRAMP_EN | 2128 FTRACE_FL_REGS_EN); 2129 } 2130 2131 ftrace_bug_type = FTRACE_BUG_NOP; 2132 return FTRACE_UPDATE_MAKE_NOP; 2133 } 2134 2135 /** 2136 * ftrace_update_record, set a record that now is tracing or not 2137 * @rec: the record to update 2138 * @enable: set to 1 if the record is tracing, zero to force disable 2139 * 2140 * The records that represent all functions that can be traced need 2141 * to be updated when tracing has been enabled. 2142 */ 2143 int ftrace_update_record(struct dyn_ftrace *rec, int enable) 2144 { 2145 return ftrace_check_record(rec, enable, 1); 2146 } 2147 2148 /** 2149 * ftrace_test_record, check if the record has been enabled or not 2150 * @rec: the record to test 2151 * @enable: set to 1 to check if enabled, 0 if it is disabled 2152 * 2153 * The arch code may need to test if a record is already set to 2154 * tracing to determine how to modify the function code that it 2155 * represents. 2156 */ 2157 int ftrace_test_record(struct dyn_ftrace *rec, int enable) 2158 { 2159 return ftrace_check_record(rec, enable, 0); 2160 } 2161 2162 static struct ftrace_ops * 2163 ftrace_find_tramp_ops_any(struct dyn_ftrace *rec) 2164 { 2165 struct ftrace_ops *op; 2166 unsigned long ip = rec->ip; 2167 2168 do_for_each_ftrace_op(op, ftrace_ops_list) { 2169 2170 if (!op->trampoline) 2171 continue; 2172 2173 if (hash_contains_ip(ip, op->func_hash)) 2174 return op; 2175 } while_for_each_ftrace_op(op); 2176 2177 return NULL; 2178 } 2179 2180 static struct ftrace_ops * 2181 ftrace_find_tramp_ops_next(struct dyn_ftrace *rec, 2182 struct ftrace_ops *op) 2183 { 2184 unsigned long ip = rec->ip; 2185 2186 while_for_each_ftrace_op(op) { 2187 2188 if (!op->trampoline) 2189 continue; 2190 2191 if (hash_contains_ip(ip, op->func_hash)) 2192 return op; 2193 } 2194 2195 return NULL; 2196 } 2197 2198 static struct ftrace_ops * 2199 ftrace_find_tramp_ops_curr(struct dyn_ftrace *rec) 2200 { 2201 struct ftrace_ops *op; 2202 unsigned long ip = rec->ip; 2203 2204 /* 2205 * Need to check removed ops first. 2206 * If they are being removed, and this rec has a tramp, 2207 * and this rec is in the ops list, then it would be the 2208 * one with the tramp. 2209 */ 2210 if (removed_ops) { 2211 if (hash_contains_ip(ip, &removed_ops->old_hash)) 2212 return removed_ops; 2213 } 2214 2215 /* 2216 * Need to find the current trampoline for a rec. 2217 * Now, a trampoline is only attached to a rec if there 2218 * was a single 'ops' attached to it. But this can be called 2219 * when we are adding another op to the rec or removing the 2220 * current one. Thus, if the op is being added, we can 2221 * ignore it because it hasn't attached itself to the rec 2222 * yet. 2223 * 2224 * If an ops is being modified (hooking to different functions) 2225 * then we don't care about the new functions that are being 2226 * added, just the old ones (that are probably being removed). 2227 * 2228 * If we are adding an ops to a function that already is using 2229 * a trampoline, it needs to be removed (trampolines are only 2230 * for single ops connected), then an ops that is not being 2231 * modified also needs to be checked. 2232 */ 2233 do_for_each_ftrace_op(op, ftrace_ops_list) { 2234 2235 if (!op->trampoline) 2236 continue; 2237 2238 /* 2239 * If the ops is being added, it hasn't gotten to 2240 * the point to be removed from this tree yet. 2241 */ 2242 if (op->flags & FTRACE_OPS_FL_ADDING) 2243 continue; 2244 2245 2246 /* 2247 * If the ops is being modified and is in the old 2248 * hash, then it is probably being removed from this 2249 * function. 2250 */ 2251 if ((op->flags & FTRACE_OPS_FL_MODIFYING) && 2252 hash_contains_ip(ip, &op->old_hash)) 2253 return op; 2254 /* 2255 * If the ops is not being added or modified, and it's 2256 * in its normal filter hash, then this must be the one 2257 * we want! 2258 */ 2259 if (!(op->flags & FTRACE_OPS_FL_MODIFYING) && 2260 hash_contains_ip(ip, op->func_hash)) 2261 return op; 2262 2263 } while_for_each_ftrace_op(op); 2264 2265 return NULL; 2266 } 2267 2268 static struct ftrace_ops * 2269 ftrace_find_tramp_ops_new(struct dyn_ftrace *rec) 2270 { 2271 struct ftrace_ops *op; 2272 unsigned long ip = rec->ip; 2273 2274 do_for_each_ftrace_op(op, ftrace_ops_list) { 2275 /* pass rec in as regs to have non-NULL val */ 2276 if (hash_contains_ip(ip, op->func_hash)) 2277 return op; 2278 } while_for_each_ftrace_op(op); 2279 2280 return NULL; 2281 } 2282 2283 /** 2284 * ftrace_get_addr_new - Get the call address to set to 2285 * @rec: The ftrace record descriptor 2286 * 2287 * If the record has the FTRACE_FL_REGS set, that means that it 2288 * wants to convert to a callback that saves all regs. If FTRACE_FL_REGS 2289 * is not not set, then it wants to convert to the normal callback. 2290 * 2291 * Returns the address of the trampoline to set to 2292 */ 2293 unsigned long ftrace_get_addr_new(struct dyn_ftrace *rec) 2294 { 2295 struct ftrace_ops *ops; 2296 2297 /* Trampolines take precedence over regs */ 2298 if (rec->flags & FTRACE_FL_TRAMP) { 2299 ops = ftrace_find_tramp_ops_new(rec); 2300 if (FTRACE_WARN_ON(!ops || !ops->trampoline)) { 2301 pr_warn("Bad trampoline accounting at: %p (%pS) (%lx)\n", 2302 (void *)rec->ip, (void *)rec->ip, rec->flags); 2303 /* Ftrace is shutting down, return anything */ 2304 return (unsigned long)FTRACE_ADDR; 2305 } 2306 return ops->trampoline; 2307 } 2308 2309 if (rec->flags & FTRACE_FL_REGS) 2310 return (unsigned long)FTRACE_REGS_ADDR; 2311 else 2312 return (unsigned long)FTRACE_ADDR; 2313 } 2314 2315 /** 2316 * ftrace_get_addr_curr - Get the call address that is already there 2317 * @rec: The ftrace record descriptor 2318 * 2319 * The FTRACE_FL_REGS_EN is set when the record already points to 2320 * a function that saves all the regs. Basically the '_EN' version 2321 * represents the current state of the function. 2322 * 2323 * Returns the address of the trampoline that is currently being called 2324 */ 2325 unsigned long ftrace_get_addr_curr(struct dyn_ftrace *rec) 2326 { 2327 struct ftrace_ops *ops; 2328 2329 /* Trampolines take precedence over regs */ 2330 if (rec->flags & FTRACE_FL_TRAMP_EN) { 2331 ops = ftrace_find_tramp_ops_curr(rec); 2332 if (FTRACE_WARN_ON(!ops)) { 2333 pr_warn("Bad trampoline accounting at: %p (%pS)\n", 2334 (void *)rec->ip, (void *)rec->ip); 2335 /* Ftrace is shutting down, return anything */ 2336 return (unsigned long)FTRACE_ADDR; 2337 } 2338 return ops->trampoline; 2339 } 2340 2341 if (rec->flags & FTRACE_FL_REGS_EN) 2342 return (unsigned long)FTRACE_REGS_ADDR; 2343 else 2344 return (unsigned long)FTRACE_ADDR; 2345 } 2346 2347 static int 2348 __ftrace_replace_code(struct dyn_ftrace *rec, int enable) 2349 { 2350 unsigned long ftrace_old_addr; 2351 unsigned long ftrace_addr; 2352 int ret; 2353 2354 ftrace_addr = ftrace_get_addr_new(rec); 2355 2356 /* This needs to be done before we call ftrace_update_record */ 2357 ftrace_old_addr = ftrace_get_addr_curr(rec); 2358 2359 ret = ftrace_update_record(rec, enable); 2360 2361 ftrace_bug_type = FTRACE_BUG_UNKNOWN; 2362 2363 switch (ret) { 2364 case FTRACE_UPDATE_IGNORE: 2365 return 0; 2366 2367 case FTRACE_UPDATE_MAKE_CALL: 2368 ftrace_bug_type = FTRACE_BUG_CALL; 2369 return ftrace_make_call(rec, ftrace_addr); 2370 2371 case FTRACE_UPDATE_MAKE_NOP: 2372 ftrace_bug_type = FTRACE_BUG_NOP; 2373 return ftrace_make_nop(NULL, rec, ftrace_old_addr); 2374 2375 case FTRACE_UPDATE_MODIFY_CALL: 2376 ftrace_bug_type = FTRACE_BUG_UPDATE; 2377 return ftrace_modify_call(rec, ftrace_old_addr, ftrace_addr); 2378 } 2379 2380 return -1; /* unknow ftrace bug */ 2381 } 2382 2383 void __weak ftrace_replace_code(int enable) 2384 { 2385 struct dyn_ftrace *rec; 2386 struct ftrace_page *pg; 2387 int failed; 2388 2389 if (unlikely(ftrace_disabled)) 2390 return; 2391 2392 do_for_each_ftrace_rec(pg, rec) { 2393 failed = __ftrace_replace_code(rec, enable); 2394 if (failed) { 2395 ftrace_bug(failed, rec); 2396 /* Stop processing */ 2397 return; 2398 } 2399 } while_for_each_ftrace_rec(); 2400 } 2401 2402 struct ftrace_rec_iter { 2403 struct ftrace_page *pg; 2404 int index; 2405 }; 2406 2407 /** 2408 * ftrace_rec_iter_start, start up iterating over traced functions 2409 * 2410 * Returns an iterator handle that is used to iterate over all 2411 * the records that represent address locations where functions 2412 * are traced. 2413 * 2414 * May return NULL if no records are available. 2415 */ 2416 struct ftrace_rec_iter *ftrace_rec_iter_start(void) 2417 { 2418 /* 2419 * We only use a single iterator. 2420 * Protected by the ftrace_lock mutex. 2421 */ 2422 static struct ftrace_rec_iter ftrace_rec_iter; 2423 struct ftrace_rec_iter *iter = &ftrace_rec_iter; 2424 2425 iter->pg = ftrace_pages_start; 2426 iter->index = 0; 2427 2428 /* Could have empty pages */ 2429 while (iter->pg && !iter->pg->index) 2430 iter->pg = iter->pg->next; 2431 2432 if (!iter->pg) 2433 return NULL; 2434 2435 return iter; 2436 } 2437 2438 /** 2439 * ftrace_rec_iter_next, get the next record to process. 2440 * @iter: The handle to the iterator. 2441 * 2442 * Returns the next iterator after the given iterator @iter. 2443 */ 2444 struct ftrace_rec_iter *ftrace_rec_iter_next(struct ftrace_rec_iter *iter) 2445 { 2446 iter->index++; 2447 2448 if (iter->index >= iter->pg->index) { 2449 iter->pg = iter->pg->next; 2450 iter->index = 0; 2451 2452 /* Could have empty pages */ 2453 while (iter->pg && !iter->pg->index) 2454 iter->pg = iter->pg->next; 2455 } 2456 2457 if (!iter->pg) 2458 return NULL; 2459 2460 return iter; 2461 } 2462 2463 /** 2464 * ftrace_rec_iter_record, get the record at the iterator location 2465 * @iter: The current iterator location 2466 * 2467 * Returns the record that the current @iter is at. 2468 */ 2469 struct dyn_ftrace *ftrace_rec_iter_record(struct ftrace_rec_iter *iter) 2470 { 2471 return &iter->pg->records[iter->index]; 2472 } 2473 2474 static int 2475 ftrace_code_disable(struct module *mod, struct dyn_ftrace *rec) 2476 { 2477 int ret; 2478 2479 if (unlikely(ftrace_disabled)) 2480 return 0; 2481 2482 ret = ftrace_make_nop(mod, rec, MCOUNT_ADDR); 2483 if (ret) { 2484 ftrace_bug_type = FTRACE_BUG_INIT; 2485 ftrace_bug(ret, rec); 2486 return 0; 2487 } 2488 return 1; 2489 } 2490 2491 /* 2492 * archs can override this function if they must do something 2493 * before the modifying code is performed. 2494 */ 2495 int __weak ftrace_arch_code_modify_prepare(void) 2496 { 2497 return 0; 2498 } 2499 2500 /* 2501 * archs can override this function if they must do something 2502 * after the modifying code is performed. 2503 */ 2504 int __weak ftrace_arch_code_modify_post_process(void) 2505 { 2506 return 0; 2507 } 2508 2509 void ftrace_modify_all_code(int command) 2510 { 2511 int update = command & FTRACE_UPDATE_TRACE_FUNC; 2512 int err = 0; 2513 2514 /* 2515 * If the ftrace_caller calls a ftrace_ops func directly, 2516 * we need to make sure that it only traces functions it 2517 * expects to trace. When doing the switch of functions, 2518 * we need to update to the ftrace_ops_list_func first 2519 * before the transition between old and new calls are set, 2520 * as the ftrace_ops_list_func will check the ops hashes 2521 * to make sure the ops are having the right functions 2522 * traced. 2523 */ 2524 if (update) { 2525 err = ftrace_update_ftrace_func(ftrace_ops_list_func); 2526 if (FTRACE_WARN_ON(err)) 2527 return; 2528 } 2529 2530 if (command & FTRACE_UPDATE_CALLS) 2531 ftrace_replace_code(1); 2532 else if (command & FTRACE_DISABLE_CALLS) 2533 ftrace_replace_code(0); 2534 2535 if (update && ftrace_trace_function != ftrace_ops_list_func) { 2536 function_trace_op = set_function_trace_op; 2537 smp_wmb(); 2538 /* If irqs are disabled, we are in stop machine */ 2539 if (!irqs_disabled()) 2540 smp_call_function(ftrace_sync_ipi, NULL, 1); 2541 err = ftrace_update_ftrace_func(ftrace_trace_function); 2542 if (FTRACE_WARN_ON(err)) 2543 return; 2544 } 2545 2546 if (command & FTRACE_START_FUNC_RET) 2547 err = ftrace_enable_ftrace_graph_caller(); 2548 else if (command & FTRACE_STOP_FUNC_RET) 2549 err = ftrace_disable_ftrace_graph_caller(); 2550 FTRACE_WARN_ON(err); 2551 } 2552 2553 static int __ftrace_modify_code(void *data) 2554 { 2555 int *command = data; 2556 2557 ftrace_modify_all_code(*command); 2558 2559 return 0; 2560 } 2561 2562 /** 2563 * ftrace_run_stop_machine, go back to the stop machine method 2564 * @command: The command to tell ftrace what to do 2565 * 2566 * If an arch needs to fall back to the stop machine method, the 2567 * it can call this function. 2568 */ 2569 void ftrace_run_stop_machine(int command) 2570 { 2571 stop_machine(__ftrace_modify_code, &command, NULL); 2572 } 2573 2574 /** 2575 * arch_ftrace_update_code, modify the code to trace or not trace 2576 * @command: The command that needs to be done 2577 * 2578 * Archs can override this function if it does not need to 2579 * run stop_machine() to modify code. 2580 */ 2581 void __weak arch_ftrace_update_code(int command) 2582 { 2583 ftrace_run_stop_machine(command); 2584 } 2585 2586 static void ftrace_run_update_code(int command) 2587 { 2588 int ret; 2589 2590 ret = ftrace_arch_code_modify_prepare(); 2591 FTRACE_WARN_ON(ret); 2592 if (ret) 2593 return; 2594 2595 /* 2596 * By default we use stop_machine() to modify the code. 2597 * But archs can do what ever they want as long as it 2598 * is safe. The stop_machine() is the safest, but also 2599 * produces the most overhead. 2600 */ 2601 arch_ftrace_update_code(command); 2602 2603 ret = ftrace_arch_code_modify_post_process(); 2604 FTRACE_WARN_ON(ret); 2605 } 2606 2607 static void ftrace_run_modify_code(struct ftrace_ops *ops, int command, 2608 struct ftrace_ops_hash *old_hash) 2609 { 2610 ops->flags |= FTRACE_OPS_FL_MODIFYING; 2611 ops->old_hash.filter_hash = old_hash->filter_hash; 2612 ops->old_hash.notrace_hash = old_hash->notrace_hash; 2613 ftrace_run_update_code(command); 2614 ops->old_hash.filter_hash = NULL; 2615 ops->old_hash.notrace_hash = NULL; 2616 ops->flags &= ~FTRACE_OPS_FL_MODIFYING; 2617 } 2618 2619 static ftrace_func_t saved_ftrace_func; 2620 static int ftrace_start_up; 2621 2622 void __weak arch_ftrace_trampoline_free(struct ftrace_ops *ops) 2623 { 2624 } 2625 2626 static void per_cpu_ops_free(struct ftrace_ops *ops) 2627 { 2628 free_percpu(ops->disabled); 2629 } 2630 2631 static void ftrace_startup_enable(int command) 2632 { 2633 if (saved_ftrace_func != ftrace_trace_function) { 2634 saved_ftrace_func = ftrace_trace_function; 2635 command |= FTRACE_UPDATE_TRACE_FUNC; 2636 } 2637 2638 if (!command || !ftrace_enabled) 2639 return; 2640 2641 ftrace_run_update_code(command); 2642 } 2643 2644 static void ftrace_startup_all(int command) 2645 { 2646 update_all_ops = true; 2647 ftrace_startup_enable(command); 2648 update_all_ops = false; 2649 } 2650 2651 static int ftrace_startup(struct ftrace_ops *ops, int command) 2652 { 2653 int ret; 2654 2655 if (unlikely(ftrace_disabled)) 2656 return -ENODEV; 2657 2658 ret = __register_ftrace_function(ops); 2659 if (ret) 2660 return ret; 2661 2662 ftrace_start_up++; 2663 2664 /* 2665 * Note that ftrace probes uses this to start up 2666 * and modify functions it will probe. But we still 2667 * set the ADDING flag for modification, as probes 2668 * do not have trampolines. If they add them in the 2669 * future, then the probes will need to distinguish 2670 * between adding and updating probes. 2671 */ 2672 ops->flags |= FTRACE_OPS_FL_ENABLED | FTRACE_OPS_FL_ADDING; 2673 2674 ret = ftrace_hash_ipmodify_enable(ops); 2675 if (ret < 0) { 2676 /* Rollback registration process */ 2677 __unregister_ftrace_function(ops); 2678 ftrace_start_up--; 2679 ops->flags &= ~FTRACE_OPS_FL_ENABLED; 2680 return ret; 2681 } 2682 2683 if (ftrace_hash_rec_enable(ops, 1)) 2684 command |= FTRACE_UPDATE_CALLS; 2685 2686 ftrace_startup_enable(command); 2687 2688 ops->flags &= ~FTRACE_OPS_FL_ADDING; 2689 2690 return 0; 2691 } 2692 2693 static int ftrace_shutdown(struct ftrace_ops *ops, int command) 2694 { 2695 int ret; 2696 2697 if (unlikely(ftrace_disabled)) 2698 return -ENODEV; 2699 2700 ret = __unregister_ftrace_function(ops); 2701 if (ret) 2702 return ret; 2703 2704 ftrace_start_up--; 2705 /* 2706 * Just warn in case of unbalance, no need to kill ftrace, it's not 2707 * critical but the ftrace_call callers may be never nopped again after 2708 * further ftrace uses. 2709 */ 2710 WARN_ON_ONCE(ftrace_start_up < 0); 2711 2712 /* Disabling ipmodify never fails */ 2713 ftrace_hash_ipmodify_disable(ops); 2714 2715 if (ftrace_hash_rec_disable(ops, 1)) 2716 command |= FTRACE_UPDATE_CALLS; 2717 2718 ops->flags &= ~FTRACE_OPS_FL_ENABLED; 2719 2720 if (saved_ftrace_func != ftrace_trace_function) { 2721 saved_ftrace_func = ftrace_trace_function; 2722 command |= FTRACE_UPDATE_TRACE_FUNC; 2723 } 2724 2725 if (!command || !ftrace_enabled) { 2726 /* 2727 * If these are per_cpu ops, they still need their 2728 * per_cpu field freed. Since, function tracing is 2729 * not currently active, we can just free them 2730 * without synchronizing all CPUs. 2731 */ 2732 if (ops->flags & FTRACE_OPS_FL_PER_CPU) 2733 per_cpu_ops_free(ops); 2734 return 0; 2735 } 2736 2737 /* 2738 * If the ops uses a trampoline, then it needs to be 2739 * tested first on update. 2740 */ 2741 ops->flags |= FTRACE_OPS_FL_REMOVING; 2742 removed_ops = ops; 2743 2744 /* The trampoline logic checks the old hashes */ 2745 ops->old_hash.filter_hash = ops->func_hash->filter_hash; 2746 ops->old_hash.notrace_hash = ops->func_hash->notrace_hash; 2747 2748 ftrace_run_update_code(command); 2749 2750 /* 2751 * If there's no more ops registered with ftrace, run a 2752 * sanity check to make sure all rec flags are cleared. 2753 */ 2754 if (ftrace_ops_list == &ftrace_list_end) { 2755 struct ftrace_page *pg; 2756 struct dyn_ftrace *rec; 2757 2758 do_for_each_ftrace_rec(pg, rec) { 2759 if (FTRACE_WARN_ON_ONCE(rec->flags)) 2760 pr_warn(" %pS flags:%lx\n", 2761 (void *)rec->ip, rec->flags); 2762 } while_for_each_ftrace_rec(); 2763 } 2764 2765 ops->old_hash.filter_hash = NULL; 2766 ops->old_hash.notrace_hash = NULL; 2767 2768 removed_ops = NULL; 2769 ops->flags &= ~FTRACE_OPS_FL_REMOVING; 2770 2771 /* 2772 * Dynamic ops may be freed, we must make sure that all 2773 * callers are done before leaving this function. 2774 * The same goes for freeing the per_cpu data of the per_cpu 2775 * ops. 2776 * 2777 * Again, normal synchronize_sched() is not good enough. 2778 * We need to do a hard force of sched synchronization. 2779 * This is because we use preempt_disable() to do RCU, but 2780 * the function tracers can be called where RCU is not watching 2781 * (like before user_exit()). We can not rely on the RCU 2782 * infrastructure to do the synchronization, thus we must do it 2783 * ourselves. 2784 */ 2785 if (ops->flags & (FTRACE_OPS_FL_DYNAMIC | FTRACE_OPS_FL_PER_CPU)) { 2786 schedule_on_each_cpu(ftrace_sync); 2787 2788 arch_ftrace_trampoline_free(ops); 2789 2790 if (ops->flags & FTRACE_OPS_FL_PER_CPU) 2791 per_cpu_ops_free(ops); 2792 } 2793 2794 return 0; 2795 } 2796 2797 static void ftrace_startup_sysctl(void) 2798 { 2799 int command; 2800 2801 if (unlikely(ftrace_disabled)) 2802 return; 2803 2804 /* Force update next time */ 2805 saved_ftrace_func = NULL; 2806 /* ftrace_start_up is true if we want ftrace running */ 2807 if (ftrace_start_up) { 2808 command = FTRACE_UPDATE_CALLS; 2809 if (ftrace_graph_active) 2810 command |= FTRACE_START_FUNC_RET; 2811 ftrace_startup_enable(command); 2812 } 2813 } 2814 2815 static void ftrace_shutdown_sysctl(void) 2816 { 2817 int command; 2818 2819 if (unlikely(ftrace_disabled)) 2820 return; 2821 2822 /* ftrace_start_up is true if ftrace is running */ 2823 if (ftrace_start_up) { 2824 command = FTRACE_DISABLE_CALLS; 2825 if (ftrace_graph_active) 2826 command |= FTRACE_STOP_FUNC_RET; 2827 ftrace_run_update_code(command); 2828 } 2829 } 2830 2831 static cycle_t ftrace_update_time; 2832 unsigned long ftrace_update_tot_cnt; 2833 2834 static inline int ops_traces_mod(struct ftrace_ops *ops) 2835 { 2836 /* 2837 * Filter_hash being empty will default to trace module. 2838 * But notrace hash requires a test of individual module functions. 2839 */ 2840 return ftrace_hash_empty(ops->func_hash->filter_hash) && 2841 ftrace_hash_empty(ops->func_hash->notrace_hash); 2842 } 2843 2844 /* 2845 * Check if the current ops references the record. 2846 * 2847 * If the ops traces all functions, then it was already accounted for. 2848 * If the ops does not trace the current record function, skip it. 2849 * If the ops ignores the function via notrace filter, skip it. 2850 */ 2851 static inline bool 2852 ops_references_rec(struct ftrace_ops *ops, struct dyn_ftrace *rec) 2853 { 2854 /* If ops isn't enabled, ignore it */ 2855 if (!(ops->flags & FTRACE_OPS_FL_ENABLED)) 2856 return 0; 2857 2858 /* If ops traces all then it includes this function */ 2859 if (ops_traces_mod(ops)) 2860 return 1; 2861 2862 /* The function must be in the filter */ 2863 if (!ftrace_hash_empty(ops->func_hash->filter_hash) && 2864 !ftrace_lookup_ip(ops->func_hash->filter_hash, rec->ip)) 2865 return 0; 2866 2867 /* If in notrace hash, we ignore it too */ 2868 if (ftrace_lookup_ip(ops->func_hash->notrace_hash, rec->ip)) 2869 return 0; 2870 2871 return 1; 2872 } 2873 2874 static int ftrace_update_code(struct module *mod, struct ftrace_page *new_pgs) 2875 { 2876 struct ftrace_page *pg; 2877 struct dyn_ftrace *p; 2878 cycle_t start, stop; 2879 unsigned long update_cnt = 0; 2880 unsigned long rec_flags = 0; 2881 int i; 2882 2883 start = ftrace_now(raw_smp_processor_id()); 2884 2885 /* 2886 * When a module is loaded, this function is called to convert 2887 * the calls to mcount in its text to nops, and also to create 2888 * an entry in the ftrace data. Now, if ftrace is activated 2889 * after this call, but before the module sets its text to 2890 * read-only, the modification of enabling ftrace can fail if 2891 * the read-only is done while ftrace is converting the calls. 2892 * To prevent this, the module's records are set as disabled 2893 * and will be enabled after the call to set the module's text 2894 * to read-only. 2895 */ 2896 if (mod) 2897 rec_flags |= FTRACE_FL_DISABLED; 2898 2899 for (pg = new_pgs; pg; pg = pg->next) { 2900 2901 for (i = 0; i < pg->index; i++) { 2902 2903 /* If something went wrong, bail without enabling anything */ 2904 if (unlikely(ftrace_disabled)) 2905 return -1; 2906 2907 p = &pg->records[i]; 2908 p->flags = rec_flags; 2909 2910 /* 2911 * Do the initial record conversion from mcount jump 2912 * to the NOP instructions. 2913 */ 2914 if (!ftrace_code_disable(mod, p)) 2915 break; 2916 2917 update_cnt++; 2918 } 2919 } 2920 2921 stop = ftrace_now(raw_smp_processor_id()); 2922 ftrace_update_time = stop - start; 2923 ftrace_update_tot_cnt += update_cnt; 2924 2925 return 0; 2926 } 2927 2928 static int ftrace_allocate_records(struct ftrace_page *pg, int count) 2929 { 2930 int order; 2931 int cnt; 2932 2933 if (WARN_ON(!count)) 2934 return -EINVAL; 2935 2936 order = get_count_order(DIV_ROUND_UP(count, ENTRIES_PER_PAGE)); 2937 2938 /* 2939 * We want to fill as much as possible. No more than a page 2940 * may be empty. 2941 */ 2942 while ((PAGE_SIZE << order) / ENTRY_SIZE >= count + ENTRIES_PER_PAGE) 2943 order--; 2944 2945 again: 2946 pg->records = (void *)__get_free_pages(GFP_KERNEL | __GFP_ZERO, order); 2947 2948 if (!pg->records) { 2949 /* if we can't allocate this size, try something smaller */ 2950 if (!order) 2951 return -ENOMEM; 2952 order >>= 1; 2953 goto again; 2954 } 2955 2956 cnt = (PAGE_SIZE << order) / ENTRY_SIZE; 2957 pg->size = cnt; 2958 2959 if (cnt > count) 2960 cnt = count; 2961 2962 return cnt; 2963 } 2964 2965 static struct ftrace_page * 2966 ftrace_allocate_pages(unsigned long num_to_init) 2967 { 2968 struct ftrace_page *start_pg; 2969 struct ftrace_page *pg; 2970 int order; 2971 int cnt; 2972 2973 if (!num_to_init) 2974 return 0; 2975 2976 start_pg = pg = kzalloc(sizeof(*pg), GFP_KERNEL); 2977 if (!pg) 2978 return NULL; 2979 2980 /* 2981 * Try to allocate as much as possible in one continues 2982 * location that fills in all of the space. We want to 2983 * waste as little space as possible. 2984 */ 2985 for (;;) { 2986 cnt = ftrace_allocate_records(pg, num_to_init); 2987 if (cnt < 0) 2988 goto free_pages; 2989 2990 num_to_init -= cnt; 2991 if (!num_to_init) 2992 break; 2993 2994 pg->next = kzalloc(sizeof(*pg), GFP_KERNEL); 2995 if (!pg->next) 2996 goto free_pages; 2997 2998 pg = pg->next; 2999 } 3000 3001 return start_pg; 3002 3003 free_pages: 3004 pg = start_pg; 3005 while (pg) { 3006 order = get_count_order(pg->size / ENTRIES_PER_PAGE); 3007 free_pages((unsigned long)pg->records, order); 3008 start_pg = pg->next; 3009 kfree(pg); 3010 pg = start_pg; 3011 } 3012 pr_info("ftrace: FAILED to allocate memory for functions\n"); 3013 return NULL; 3014 } 3015 3016 #define FTRACE_BUFF_MAX (KSYM_SYMBOL_LEN+4) /* room for wildcards */ 3017 3018 struct ftrace_iterator { 3019 loff_t pos; 3020 loff_t func_pos; 3021 struct ftrace_page *pg; 3022 struct dyn_ftrace *func; 3023 struct ftrace_func_probe *probe; 3024 struct trace_parser parser; 3025 struct ftrace_hash *hash; 3026 struct ftrace_ops *ops; 3027 int hidx; 3028 int idx; 3029 unsigned flags; 3030 }; 3031 3032 static void * 3033 t_hash_next(struct seq_file *m, loff_t *pos) 3034 { 3035 struct ftrace_iterator *iter = m->private; 3036 struct hlist_node *hnd = NULL; 3037 struct hlist_head *hhd; 3038 3039 (*pos)++; 3040 iter->pos = *pos; 3041 3042 if (iter->probe) 3043 hnd = &iter->probe->node; 3044 retry: 3045 if (iter->hidx >= FTRACE_FUNC_HASHSIZE) 3046 return NULL; 3047 3048 hhd = &ftrace_func_hash[iter->hidx]; 3049 3050 if (hlist_empty(hhd)) { 3051 iter->hidx++; 3052 hnd = NULL; 3053 goto retry; 3054 } 3055 3056 if (!hnd) 3057 hnd = hhd->first; 3058 else { 3059 hnd = hnd->next; 3060 if (!hnd) { 3061 iter->hidx++; 3062 goto retry; 3063 } 3064 } 3065 3066 if (WARN_ON_ONCE(!hnd)) 3067 return NULL; 3068 3069 iter->probe = hlist_entry(hnd, struct ftrace_func_probe, node); 3070 3071 return iter; 3072 } 3073 3074 static void *t_hash_start(struct seq_file *m, loff_t *pos) 3075 { 3076 struct ftrace_iterator *iter = m->private; 3077 void *p = NULL; 3078 loff_t l; 3079 3080 if (!(iter->flags & FTRACE_ITER_DO_HASH)) 3081 return NULL; 3082 3083 if (iter->func_pos > *pos) 3084 return NULL; 3085 3086 iter->hidx = 0; 3087 for (l = 0; l <= (*pos - iter->func_pos); ) { 3088 p = t_hash_next(m, &l); 3089 if (!p) 3090 break; 3091 } 3092 if (!p) 3093 return NULL; 3094 3095 /* Only set this if we have an item */ 3096 iter->flags |= FTRACE_ITER_HASH; 3097 3098 return iter; 3099 } 3100 3101 static int 3102 t_hash_show(struct seq_file *m, struct ftrace_iterator *iter) 3103 { 3104 struct ftrace_func_probe *rec; 3105 3106 rec = iter->probe; 3107 if (WARN_ON_ONCE(!rec)) 3108 return -EIO; 3109 3110 if (rec->ops->print) 3111 return rec->ops->print(m, rec->ip, rec->ops, rec->data); 3112 3113 seq_printf(m, "%ps:%ps", (void *)rec->ip, (void *)rec->ops->func); 3114 3115 if (rec->data) 3116 seq_printf(m, ":%p", rec->data); 3117 seq_putc(m, '\n'); 3118 3119 return 0; 3120 } 3121 3122 static void * 3123 t_next(struct seq_file *m, void *v, loff_t *pos) 3124 { 3125 struct ftrace_iterator *iter = m->private; 3126 struct ftrace_ops *ops = iter->ops; 3127 struct dyn_ftrace *rec = NULL; 3128 3129 if (unlikely(ftrace_disabled)) 3130 return NULL; 3131 3132 if (iter->flags & FTRACE_ITER_HASH) 3133 return t_hash_next(m, pos); 3134 3135 (*pos)++; 3136 iter->pos = iter->func_pos = *pos; 3137 3138 if (iter->flags & FTRACE_ITER_PRINTALL) 3139 return t_hash_start(m, pos); 3140 3141 retry: 3142 if (iter->idx >= iter->pg->index) { 3143 if (iter->pg->next) { 3144 iter->pg = iter->pg->next; 3145 iter->idx = 0; 3146 goto retry; 3147 } 3148 } else { 3149 rec = &iter->pg->records[iter->idx++]; 3150 if (((iter->flags & FTRACE_ITER_FILTER) && 3151 !(ftrace_lookup_ip(ops->func_hash->filter_hash, rec->ip))) || 3152 3153 ((iter->flags & FTRACE_ITER_NOTRACE) && 3154 !ftrace_lookup_ip(ops->func_hash->notrace_hash, rec->ip)) || 3155 3156 ((iter->flags & FTRACE_ITER_ENABLED) && 3157 !(rec->flags & FTRACE_FL_ENABLED))) { 3158 3159 rec = NULL; 3160 goto retry; 3161 } 3162 } 3163 3164 if (!rec) 3165 return t_hash_start(m, pos); 3166 3167 iter->func = rec; 3168 3169 return iter; 3170 } 3171 3172 static void reset_iter_read(struct ftrace_iterator *iter) 3173 { 3174 iter->pos = 0; 3175 iter->func_pos = 0; 3176 iter->flags &= ~(FTRACE_ITER_PRINTALL | FTRACE_ITER_HASH); 3177 } 3178 3179 static void *t_start(struct seq_file *m, loff_t *pos) 3180 { 3181 struct ftrace_iterator *iter = m->private; 3182 struct ftrace_ops *ops = iter->ops; 3183 void *p = NULL; 3184 loff_t l; 3185 3186 mutex_lock(&ftrace_lock); 3187 3188 if (unlikely(ftrace_disabled)) 3189 return NULL; 3190 3191 /* 3192 * If an lseek was done, then reset and start from beginning. 3193 */ 3194 if (*pos < iter->pos) 3195 reset_iter_read(iter); 3196 3197 /* 3198 * For set_ftrace_filter reading, if we have the filter 3199 * off, we can short cut and just print out that all 3200 * functions are enabled. 3201 */ 3202 if ((iter->flags & FTRACE_ITER_FILTER && 3203 ftrace_hash_empty(ops->func_hash->filter_hash)) || 3204 (iter->flags & FTRACE_ITER_NOTRACE && 3205 ftrace_hash_empty(ops->func_hash->notrace_hash))) { 3206 if (*pos > 0) 3207 return t_hash_start(m, pos); 3208 iter->flags |= FTRACE_ITER_PRINTALL; 3209 /* reset in case of seek/pread */ 3210 iter->flags &= ~FTRACE_ITER_HASH; 3211 return iter; 3212 } 3213 3214 if (iter->flags & FTRACE_ITER_HASH) 3215 return t_hash_start(m, pos); 3216 3217 /* 3218 * Unfortunately, we need to restart at ftrace_pages_start 3219 * every time we let go of the ftrace_mutex. This is because 3220 * those pointers can change without the lock. 3221 */ 3222 iter->pg = ftrace_pages_start; 3223 iter->idx = 0; 3224 for (l = 0; l <= *pos; ) { 3225 p = t_next(m, p, &l); 3226 if (!p) 3227 break; 3228 } 3229 3230 if (!p) 3231 return t_hash_start(m, pos); 3232 3233 return iter; 3234 } 3235 3236 static void t_stop(struct seq_file *m, void *p) 3237 { 3238 mutex_unlock(&ftrace_lock); 3239 } 3240 3241 void * __weak 3242 arch_ftrace_trampoline_func(struct ftrace_ops *ops, struct dyn_ftrace *rec) 3243 { 3244 return NULL; 3245 } 3246 3247 static void add_trampoline_func(struct seq_file *m, struct ftrace_ops *ops, 3248 struct dyn_ftrace *rec) 3249 { 3250 void *ptr; 3251 3252 ptr = arch_ftrace_trampoline_func(ops, rec); 3253 if (ptr) 3254 seq_printf(m, " ->%pS", ptr); 3255 } 3256 3257 static int t_show(struct seq_file *m, void *v) 3258 { 3259 struct ftrace_iterator *iter = m->private; 3260 struct dyn_ftrace *rec; 3261 3262 if (iter->flags & FTRACE_ITER_HASH) 3263 return t_hash_show(m, iter); 3264 3265 if (iter->flags & FTRACE_ITER_PRINTALL) { 3266 if (iter->flags & FTRACE_ITER_NOTRACE) 3267 seq_puts(m, "#### no functions disabled ####\n"); 3268 else 3269 seq_puts(m, "#### all functions enabled ####\n"); 3270 return 0; 3271 } 3272 3273 rec = iter->func; 3274 3275 if (!rec) 3276 return 0; 3277 3278 seq_printf(m, "%ps", (void *)rec->ip); 3279 if (iter->flags & FTRACE_ITER_ENABLED) { 3280 struct ftrace_ops *ops; 3281 3282 seq_printf(m, " (%ld)%s%s", 3283 ftrace_rec_count(rec), 3284 rec->flags & FTRACE_FL_REGS ? " R" : " ", 3285 rec->flags & FTRACE_FL_IPMODIFY ? " I" : " "); 3286 if (rec->flags & FTRACE_FL_TRAMP_EN) { 3287 ops = ftrace_find_tramp_ops_any(rec); 3288 if (ops) { 3289 do { 3290 seq_printf(m, "\ttramp: %pS (%pS)", 3291 (void *)ops->trampoline, 3292 (void *)ops->func); 3293 add_trampoline_func(m, ops, rec); 3294 ops = ftrace_find_tramp_ops_next(rec, ops); 3295 } while (ops); 3296 } else 3297 seq_puts(m, "\ttramp: ERROR!"); 3298 } else { 3299 add_trampoline_func(m, NULL, rec); 3300 } 3301 } 3302 3303 seq_putc(m, '\n'); 3304 3305 return 0; 3306 } 3307 3308 static const struct seq_operations show_ftrace_seq_ops = { 3309 .start = t_start, 3310 .next = t_next, 3311 .stop = t_stop, 3312 .show = t_show, 3313 }; 3314 3315 static int 3316 ftrace_avail_open(struct inode *inode, struct file *file) 3317 { 3318 struct ftrace_iterator *iter; 3319 3320 if (unlikely(ftrace_disabled)) 3321 return -ENODEV; 3322 3323 iter = __seq_open_private(file, &show_ftrace_seq_ops, sizeof(*iter)); 3324 if (iter) { 3325 iter->pg = ftrace_pages_start; 3326 iter->ops = &global_ops; 3327 } 3328 3329 return iter ? 0 : -ENOMEM; 3330 } 3331 3332 static int 3333 ftrace_enabled_open(struct inode *inode, struct file *file) 3334 { 3335 struct ftrace_iterator *iter; 3336 3337 iter = __seq_open_private(file, &show_ftrace_seq_ops, sizeof(*iter)); 3338 if (iter) { 3339 iter->pg = ftrace_pages_start; 3340 iter->flags = FTRACE_ITER_ENABLED; 3341 iter->ops = &global_ops; 3342 } 3343 3344 return iter ? 0 : -ENOMEM; 3345 } 3346 3347 /** 3348 * ftrace_regex_open - initialize function tracer filter files 3349 * @ops: The ftrace_ops that hold the hash filters 3350 * @flag: The type of filter to process 3351 * @inode: The inode, usually passed in to your open routine 3352 * @file: The file, usually passed in to your open routine 3353 * 3354 * ftrace_regex_open() initializes the filter files for the 3355 * @ops. Depending on @flag it may process the filter hash or 3356 * the notrace hash of @ops. With this called from the open 3357 * routine, you can use ftrace_filter_write() for the write 3358 * routine if @flag has FTRACE_ITER_FILTER set, or 3359 * ftrace_notrace_write() if @flag has FTRACE_ITER_NOTRACE set. 3360 * tracing_lseek() should be used as the lseek routine, and 3361 * release must call ftrace_regex_release(). 3362 */ 3363 int 3364 ftrace_regex_open(struct ftrace_ops *ops, int flag, 3365 struct inode *inode, struct file *file) 3366 { 3367 struct ftrace_iterator *iter; 3368 struct ftrace_hash *hash; 3369 int ret = 0; 3370 3371 ftrace_ops_init(ops); 3372 3373 if (unlikely(ftrace_disabled)) 3374 return -ENODEV; 3375 3376 iter = kzalloc(sizeof(*iter), GFP_KERNEL); 3377 if (!iter) 3378 return -ENOMEM; 3379 3380 if (trace_parser_get_init(&iter->parser, FTRACE_BUFF_MAX)) { 3381 kfree(iter); 3382 return -ENOMEM; 3383 } 3384 3385 iter->ops = ops; 3386 iter->flags = flag; 3387 3388 mutex_lock(&ops->func_hash->regex_lock); 3389 3390 if (flag & FTRACE_ITER_NOTRACE) 3391 hash = ops->func_hash->notrace_hash; 3392 else 3393 hash = ops->func_hash->filter_hash; 3394 3395 if (file->f_mode & FMODE_WRITE) { 3396 const int size_bits = FTRACE_HASH_DEFAULT_BITS; 3397 3398 if (file->f_flags & O_TRUNC) 3399 iter->hash = alloc_ftrace_hash(size_bits); 3400 else 3401 iter->hash = alloc_and_copy_ftrace_hash(size_bits, hash); 3402 3403 if (!iter->hash) { 3404 trace_parser_put(&iter->parser); 3405 kfree(iter); 3406 ret = -ENOMEM; 3407 goto out_unlock; 3408 } 3409 } 3410 3411 if (file->f_mode & FMODE_READ) { 3412 iter->pg = ftrace_pages_start; 3413 3414 ret = seq_open(file, &show_ftrace_seq_ops); 3415 if (!ret) { 3416 struct seq_file *m = file->private_data; 3417 m->private = iter; 3418 } else { 3419 /* Failed */ 3420 free_ftrace_hash(iter->hash); 3421 trace_parser_put(&iter->parser); 3422 kfree(iter); 3423 } 3424 } else 3425 file->private_data = iter; 3426 3427 out_unlock: 3428 mutex_unlock(&ops->func_hash->regex_lock); 3429 3430 return ret; 3431 } 3432 3433 static int 3434 ftrace_filter_open(struct inode *inode, struct file *file) 3435 { 3436 struct ftrace_ops *ops = inode->i_private; 3437 3438 return ftrace_regex_open(ops, 3439 FTRACE_ITER_FILTER | FTRACE_ITER_DO_HASH, 3440 inode, file); 3441 } 3442 3443 static int 3444 ftrace_notrace_open(struct inode *inode, struct file *file) 3445 { 3446 struct ftrace_ops *ops = inode->i_private; 3447 3448 return ftrace_regex_open(ops, FTRACE_ITER_NOTRACE, 3449 inode, file); 3450 } 3451 3452 /* Type for quick search ftrace basic regexes (globs) from filter_parse_regex */ 3453 struct ftrace_glob { 3454 char *search; 3455 unsigned len; 3456 int type; 3457 }; 3458 3459 /* 3460 * If symbols in an architecture don't correspond exactly to the user-visible 3461 * name of what they represent, it is possible to define this function to 3462 * perform the necessary adjustments. 3463 */ 3464 char * __weak arch_ftrace_match_adjust(char *str, const char *search) 3465 { 3466 return str; 3467 } 3468 3469 static int ftrace_match(char *str, struct ftrace_glob *g) 3470 { 3471 int matched = 0; 3472 int slen; 3473 3474 str = arch_ftrace_match_adjust(str, g->search); 3475 3476 switch (g->type) { 3477 case MATCH_FULL: 3478 if (strcmp(str, g->search) == 0) 3479 matched = 1; 3480 break; 3481 case MATCH_FRONT_ONLY: 3482 if (strncmp(str, g->search, g->len) == 0) 3483 matched = 1; 3484 break; 3485 case MATCH_MIDDLE_ONLY: 3486 if (strstr(str, g->search)) 3487 matched = 1; 3488 break; 3489 case MATCH_END_ONLY: 3490 slen = strlen(str); 3491 if (slen >= g->len && 3492 memcmp(str + slen - g->len, g->search, g->len) == 0) 3493 matched = 1; 3494 break; 3495 } 3496 3497 return matched; 3498 } 3499 3500 static int 3501 enter_record(struct ftrace_hash *hash, struct dyn_ftrace *rec, int clear_filter) 3502 { 3503 struct ftrace_func_entry *entry; 3504 int ret = 0; 3505 3506 entry = ftrace_lookup_ip(hash, rec->ip); 3507 if (clear_filter) { 3508 /* Do nothing if it doesn't exist */ 3509 if (!entry) 3510 return 0; 3511 3512 free_hash_entry(hash, entry); 3513 } else { 3514 /* Do nothing if it exists */ 3515 if (entry) 3516 return 0; 3517 3518 ret = add_hash_entry(hash, rec->ip); 3519 } 3520 return ret; 3521 } 3522 3523 static int 3524 ftrace_match_record(struct dyn_ftrace *rec, struct ftrace_glob *func_g, 3525 struct ftrace_glob *mod_g, int exclude_mod) 3526 { 3527 char str[KSYM_SYMBOL_LEN]; 3528 char *modname; 3529 3530 kallsyms_lookup(rec->ip, NULL, NULL, &modname, str); 3531 3532 if (mod_g) { 3533 int mod_matches = (modname) ? ftrace_match(modname, mod_g) : 0; 3534 3535 /* blank module name to match all modules */ 3536 if (!mod_g->len) { 3537 /* blank module globbing: modname xor exclude_mod */ 3538 if ((!exclude_mod) != (!modname)) 3539 goto func_match; 3540 return 0; 3541 } 3542 3543 /* not matching the module */ 3544 if (!modname || !mod_matches) { 3545 if (exclude_mod) 3546 goto func_match; 3547 else 3548 return 0; 3549 } 3550 3551 if (mod_matches && exclude_mod) 3552 return 0; 3553 3554 func_match: 3555 /* blank search means to match all funcs in the mod */ 3556 if (!func_g->len) 3557 return 1; 3558 } 3559 3560 return ftrace_match(str, func_g); 3561 } 3562 3563 static int 3564 match_records(struct ftrace_hash *hash, char *func, int len, char *mod) 3565 { 3566 struct ftrace_page *pg; 3567 struct dyn_ftrace *rec; 3568 struct ftrace_glob func_g = { .type = MATCH_FULL }; 3569 struct ftrace_glob mod_g = { .type = MATCH_FULL }; 3570 struct ftrace_glob *mod_match = (mod) ? &mod_g : NULL; 3571 int exclude_mod = 0; 3572 int found = 0; 3573 int ret; 3574 int clear_filter; 3575 3576 if (func) { 3577 func_g.type = filter_parse_regex(func, len, &func_g.search, 3578 &clear_filter); 3579 func_g.len = strlen(func_g.search); 3580 } 3581 3582 if (mod) { 3583 mod_g.type = filter_parse_regex(mod, strlen(mod), 3584 &mod_g.search, &exclude_mod); 3585 mod_g.len = strlen(mod_g.search); 3586 } 3587 3588 mutex_lock(&ftrace_lock); 3589 3590 if (unlikely(ftrace_disabled)) 3591 goto out_unlock; 3592 3593 do_for_each_ftrace_rec(pg, rec) { 3594 if (ftrace_match_record(rec, &func_g, mod_match, exclude_mod)) { 3595 ret = enter_record(hash, rec, clear_filter); 3596 if (ret < 0) { 3597 found = ret; 3598 goto out_unlock; 3599 } 3600 found = 1; 3601 } 3602 } while_for_each_ftrace_rec(); 3603 out_unlock: 3604 mutex_unlock(&ftrace_lock); 3605 3606 return found; 3607 } 3608 3609 static int 3610 ftrace_match_records(struct ftrace_hash *hash, char *buff, int len) 3611 { 3612 return match_records(hash, buff, len, NULL); 3613 } 3614 3615 3616 /* 3617 * We register the module command as a template to show others how 3618 * to register the a command as well. 3619 */ 3620 3621 static int 3622 ftrace_mod_callback(struct ftrace_hash *hash, 3623 char *func, char *cmd, char *module, int enable) 3624 { 3625 int ret; 3626 3627 /* 3628 * cmd == 'mod' because we only registered this func 3629 * for the 'mod' ftrace_func_command. 3630 * But if you register one func with multiple commands, 3631 * you can tell which command was used by the cmd 3632 * parameter. 3633 */ 3634 ret = match_records(hash, func, strlen(func), module); 3635 if (!ret) 3636 return -EINVAL; 3637 if (ret < 0) 3638 return ret; 3639 return 0; 3640 } 3641 3642 static struct ftrace_func_command ftrace_mod_cmd = { 3643 .name = "mod", 3644 .func = ftrace_mod_callback, 3645 }; 3646 3647 static int __init ftrace_mod_cmd_init(void) 3648 { 3649 return register_ftrace_command(&ftrace_mod_cmd); 3650 } 3651 core_initcall(ftrace_mod_cmd_init); 3652 3653 static void function_trace_probe_call(unsigned long ip, unsigned long parent_ip, 3654 struct ftrace_ops *op, struct pt_regs *pt_regs) 3655 { 3656 struct ftrace_func_probe *entry; 3657 struct hlist_head *hhd; 3658 unsigned long key; 3659 3660 key = hash_long(ip, FTRACE_HASH_BITS); 3661 3662 hhd = &ftrace_func_hash[key]; 3663 3664 if (hlist_empty(hhd)) 3665 return; 3666 3667 /* 3668 * Disable preemption for these calls to prevent a RCU grace 3669 * period. This syncs the hash iteration and freeing of items 3670 * on the hash. rcu_read_lock is too dangerous here. 3671 */ 3672 preempt_disable_notrace(); 3673 hlist_for_each_entry_rcu_notrace(entry, hhd, node) { 3674 if (entry->ip == ip) 3675 entry->ops->func(ip, parent_ip, &entry->data); 3676 } 3677 preempt_enable_notrace(); 3678 } 3679 3680 static struct ftrace_ops trace_probe_ops __read_mostly = 3681 { 3682 .func = function_trace_probe_call, 3683 .flags = FTRACE_OPS_FL_INITIALIZED, 3684 INIT_OPS_HASH(trace_probe_ops) 3685 }; 3686 3687 static int ftrace_probe_registered; 3688 3689 static void __enable_ftrace_function_probe(struct ftrace_ops_hash *old_hash) 3690 { 3691 int ret; 3692 int i; 3693 3694 if (ftrace_probe_registered) { 3695 /* still need to update the function call sites */ 3696 if (ftrace_enabled) 3697 ftrace_run_modify_code(&trace_probe_ops, FTRACE_UPDATE_CALLS, 3698 old_hash); 3699 return; 3700 } 3701 3702 for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) { 3703 struct hlist_head *hhd = &ftrace_func_hash[i]; 3704 if (hhd->first) 3705 break; 3706 } 3707 /* Nothing registered? */ 3708 if (i == FTRACE_FUNC_HASHSIZE) 3709 return; 3710 3711 ret = ftrace_startup(&trace_probe_ops, 0); 3712 3713 ftrace_probe_registered = 1; 3714 } 3715 3716 static void __disable_ftrace_function_probe(void) 3717 { 3718 int i; 3719 3720 if (!ftrace_probe_registered) 3721 return; 3722 3723 for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) { 3724 struct hlist_head *hhd = &ftrace_func_hash[i]; 3725 if (hhd->first) 3726 return; 3727 } 3728 3729 /* no more funcs left */ 3730 ftrace_shutdown(&trace_probe_ops, 0); 3731 3732 ftrace_probe_registered = 0; 3733 } 3734 3735 3736 static void ftrace_free_entry(struct ftrace_func_probe *entry) 3737 { 3738 if (entry->ops->free) 3739 entry->ops->free(entry->ops, entry->ip, &entry->data); 3740 kfree(entry); 3741 } 3742 3743 int 3744 register_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops, 3745 void *data) 3746 { 3747 struct ftrace_ops_hash old_hash_ops; 3748 struct ftrace_func_probe *entry; 3749 struct ftrace_glob func_g; 3750 struct ftrace_hash **orig_hash = &trace_probe_ops.func_hash->filter_hash; 3751 struct ftrace_hash *old_hash = *orig_hash; 3752 struct ftrace_hash *hash; 3753 struct ftrace_page *pg; 3754 struct dyn_ftrace *rec; 3755 int not; 3756 unsigned long key; 3757 int count = 0; 3758 int ret; 3759 3760 func_g.type = filter_parse_regex(glob, strlen(glob), 3761 &func_g.search, ¬); 3762 func_g.len = strlen(func_g.search); 3763 3764 /* we do not support '!' for function probes */ 3765 if (WARN_ON(not)) 3766 return -EINVAL; 3767 3768 mutex_lock(&trace_probe_ops.func_hash->regex_lock); 3769 3770 old_hash_ops.filter_hash = old_hash; 3771 /* Probes only have filters */ 3772 old_hash_ops.notrace_hash = NULL; 3773 3774 hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, old_hash); 3775 if (!hash) { 3776 count = -ENOMEM; 3777 goto out; 3778 } 3779 3780 if (unlikely(ftrace_disabled)) { 3781 count = -ENODEV; 3782 goto out; 3783 } 3784 3785 mutex_lock(&ftrace_lock); 3786 3787 do_for_each_ftrace_rec(pg, rec) { 3788 3789 if (!ftrace_match_record(rec, &func_g, NULL, 0)) 3790 continue; 3791 3792 entry = kmalloc(sizeof(*entry), GFP_KERNEL); 3793 if (!entry) { 3794 /* If we did not process any, then return error */ 3795 if (!count) 3796 count = -ENOMEM; 3797 goto out_unlock; 3798 } 3799 3800 count++; 3801 3802 entry->data = data; 3803 3804 /* 3805 * The caller might want to do something special 3806 * for each function we find. We call the callback 3807 * to give the caller an opportunity to do so. 3808 */ 3809 if (ops->init) { 3810 if (ops->init(ops, rec->ip, &entry->data) < 0) { 3811 /* caller does not like this func */ 3812 kfree(entry); 3813 continue; 3814 } 3815 } 3816 3817 ret = enter_record(hash, rec, 0); 3818 if (ret < 0) { 3819 kfree(entry); 3820 count = ret; 3821 goto out_unlock; 3822 } 3823 3824 entry->ops = ops; 3825 entry->ip = rec->ip; 3826 3827 key = hash_long(entry->ip, FTRACE_HASH_BITS); 3828 hlist_add_head_rcu(&entry->node, &ftrace_func_hash[key]); 3829 3830 } while_for_each_ftrace_rec(); 3831 3832 ret = ftrace_hash_move(&trace_probe_ops, 1, orig_hash, hash); 3833 3834 __enable_ftrace_function_probe(&old_hash_ops); 3835 3836 if (!ret) 3837 free_ftrace_hash_rcu(old_hash); 3838 else 3839 count = ret; 3840 3841 out_unlock: 3842 mutex_unlock(&ftrace_lock); 3843 out: 3844 mutex_unlock(&trace_probe_ops.func_hash->regex_lock); 3845 free_ftrace_hash(hash); 3846 3847 return count; 3848 } 3849 3850 enum { 3851 PROBE_TEST_FUNC = 1, 3852 PROBE_TEST_DATA = 2 3853 }; 3854 3855 static void 3856 __unregister_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops, 3857 void *data, int flags) 3858 { 3859 struct ftrace_func_entry *rec_entry; 3860 struct ftrace_func_probe *entry; 3861 struct ftrace_func_probe *p; 3862 struct ftrace_glob func_g; 3863 struct ftrace_hash **orig_hash = &trace_probe_ops.func_hash->filter_hash; 3864 struct ftrace_hash *old_hash = *orig_hash; 3865 struct list_head free_list; 3866 struct ftrace_hash *hash; 3867 struct hlist_node *tmp; 3868 char str[KSYM_SYMBOL_LEN]; 3869 int i, ret; 3870 3871 if (glob && (strcmp(glob, "*") == 0 || !strlen(glob))) 3872 func_g.search = NULL; 3873 else if (glob) { 3874 int not; 3875 3876 func_g.type = filter_parse_regex(glob, strlen(glob), 3877 &func_g.search, ¬); 3878 func_g.len = strlen(func_g.search); 3879 func_g.search = glob; 3880 3881 /* we do not support '!' for function probes */ 3882 if (WARN_ON(not)) 3883 return; 3884 } 3885 3886 mutex_lock(&trace_probe_ops.func_hash->regex_lock); 3887 3888 hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, *orig_hash); 3889 if (!hash) 3890 /* Hmm, should report this somehow */ 3891 goto out_unlock; 3892 3893 INIT_LIST_HEAD(&free_list); 3894 3895 for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) { 3896 struct hlist_head *hhd = &ftrace_func_hash[i]; 3897 3898 hlist_for_each_entry_safe(entry, tmp, hhd, node) { 3899 3900 /* break up if statements for readability */ 3901 if ((flags & PROBE_TEST_FUNC) && entry->ops != ops) 3902 continue; 3903 3904 if ((flags & PROBE_TEST_DATA) && entry->data != data) 3905 continue; 3906 3907 /* do this last, since it is the most expensive */ 3908 if (func_g.search) { 3909 kallsyms_lookup(entry->ip, NULL, NULL, 3910 NULL, str); 3911 if (!ftrace_match(str, &func_g)) 3912 continue; 3913 } 3914 3915 rec_entry = ftrace_lookup_ip(hash, entry->ip); 3916 /* It is possible more than one entry had this ip */ 3917 if (rec_entry) 3918 free_hash_entry(hash, rec_entry); 3919 3920 hlist_del_rcu(&entry->node); 3921 list_add(&entry->free_list, &free_list); 3922 } 3923 } 3924 mutex_lock(&ftrace_lock); 3925 __disable_ftrace_function_probe(); 3926 /* 3927 * Remove after the disable is called. Otherwise, if the last 3928 * probe is removed, a null hash means *all enabled*. 3929 */ 3930 ret = ftrace_hash_move(&trace_probe_ops, 1, orig_hash, hash); 3931 synchronize_sched(); 3932 if (!ret) 3933 free_ftrace_hash_rcu(old_hash); 3934 3935 list_for_each_entry_safe(entry, p, &free_list, free_list) { 3936 list_del(&entry->free_list); 3937 ftrace_free_entry(entry); 3938 } 3939 mutex_unlock(&ftrace_lock); 3940 3941 out_unlock: 3942 mutex_unlock(&trace_probe_ops.func_hash->regex_lock); 3943 free_ftrace_hash(hash); 3944 } 3945 3946 void 3947 unregister_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops, 3948 void *data) 3949 { 3950 __unregister_ftrace_function_probe(glob, ops, data, 3951 PROBE_TEST_FUNC | PROBE_TEST_DATA); 3952 } 3953 3954 void 3955 unregister_ftrace_function_probe_func(char *glob, struct ftrace_probe_ops *ops) 3956 { 3957 __unregister_ftrace_function_probe(glob, ops, NULL, PROBE_TEST_FUNC); 3958 } 3959 3960 void unregister_ftrace_function_probe_all(char *glob) 3961 { 3962 __unregister_ftrace_function_probe(glob, NULL, NULL, 0); 3963 } 3964 3965 static LIST_HEAD(ftrace_commands); 3966 static DEFINE_MUTEX(ftrace_cmd_mutex); 3967 3968 /* 3969 * Currently we only register ftrace commands from __init, so mark this 3970 * __init too. 3971 */ 3972 __init int register_ftrace_command(struct ftrace_func_command *cmd) 3973 { 3974 struct ftrace_func_command *p; 3975 int ret = 0; 3976 3977 mutex_lock(&ftrace_cmd_mutex); 3978 list_for_each_entry(p, &ftrace_commands, list) { 3979 if (strcmp(cmd->name, p->name) == 0) { 3980 ret = -EBUSY; 3981 goto out_unlock; 3982 } 3983 } 3984 list_add(&cmd->list, &ftrace_commands); 3985 out_unlock: 3986 mutex_unlock(&ftrace_cmd_mutex); 3987 3988 return ret; 3989 } 3990 3991 /* 3992 * Currently we only unregister ftrace commands from __init, so mark 3993 * this __init too. 3994 */ 3995 __init int unregister_ftrace_command(struct ftrace_func_command *cmd) 3996 { 3997 struct ftrace_func_command *p, *n; 3998 int ret = -ENODEV; 3999 4000 mutex_lock(&ftrace_cmd_mutex); 4001 list_for_each_entry_safe(p, n, &ftrace_commands, list) { 4002 if (strcmp(cmd->name, p->name) == 0) { 4003 ret = 0; 4004 list_del_init(&p->list); 4005 goto out_unlock; 4006 } 4007 } 4008 out_unlock: 4009 mutex_unlock(&ftrace_cmd_mutex); 4010 4011 return ret; 4012 } 4013 4014 static int ftrace_process_regex(struct ftrace_hash *hash, 4015 char *buff, int len, int enable) 4016 { 4017 char *func, *command, *next = buff; 4018 struct ftrace_func_command *p; 4019 int ret = -EINVAL; 4020 4021 func = strsep(&next, ":"); 4022 4023 if (!next) { 4024 ret = ftrace_match_records(hash, func, len); 4025 if (!ret) 4026 ret = -EINVAL; 4027 if (ret < 0) 4028 return ret; 4029 return 0; 4030 } 4031 4032 /* command found */ 4033 4034 command = strsep(&next, ":"); 4035 4036 mutex_lock(&ftrace_cmd_mutex); 4037 list_for_each_entry(p, &ftrace_commands, list) { 4038 if (strcmp(p->name, command) == 0) { 4039 ret = p->func(hash, func, command, next, enable); 4040 goto out_unlock; 4041 } 4042 } 4043 out_unlock: 4044 mutex_unlock(&ftrace_cmd_mutex); 4045 4046 return ret; 4047 } 4048 4049 static ssize_t 4050 ftrace_regex_write(struct file *file, const char __user *ubuf, 4051 size_t cnt, loff_t *ppos, int enable) 4052 { 4053 struct ftrace_iterator *iter; 4054 struct trace_parser *parser; 4055 ssize_t ret, read; 4056 4057 if (!cnt) 4058 return 0; 4059 4060 if (file->f_mode & FMODE_READ) { 4061 struct seq_file *m = file->private_data; 4062 iter = m->private; 4063 } else 4064 iter = file->private_data; 4065 4066 if (unlikely(ftrace_disabled)) 4067 return -ENODEV; 4068 4069 /* iter->hash is a local copy, so we don't need regex_lock */ 4070 4071 parser = &iter->parser; 4072 read = trace_get_user(parser, ubuf, cnt, ppos); 4073 4074 if (read >= 0 && trace_parser_loaded(parser) && 4075 !trace_parser_cont(parser)) { 4076 ret = ftrace_process_regex(iter->hash, parser->buffer, 4077 parser->idx, enable); 4078 trace_parser_clear(parser); 4079 if (ret < 0) 4080 goto out; 4081 } 4082 4083 ret = read; 4084 out: 4085 return ret; 4086 } 4087 4088 ssize_t 4089 ftrace_filter_write(struct file *file, const char __user *ubuf, 4090 size_t cnt, loff_t *ppos) 4091 { 4092 return ftrace_regex_write(file, ubuf, cnt, ppos, 1); 4093 } 4094 4095 ssize_t 4096 ftrace_notrace_write(struct file *file, const char __user *ubuf, 4097 size_t cnt, loff_t *ppos) 4098 { 4099 return ftrace_regex_write(file, ubuf, cnt, ppos, 0); 4100 } 4101 4102 static int 4103 ftrace_match_addr(struct ftrace_hash *hash, unsigned long ip, int remove) 4104 { 4105 struct ftrace_func_entry *entry; 4106 4107 if (!ftrace_location(ip)) 4108 return -EINVAL; 4109 4110 if (remove) { 4111 entry = ftrace_lookup_ip(hash, ip); 4112 if (!entry) 4113 return -ENOENT; 4114 free_hash_entry(hash, entry); 4115 return 0; 4116 } 4117 4118 return add_hash_entry(hash, ip); 4119 } 4120 4121 static void ftrace_ops_update_code(struct ftrace_ops *ops, 4122 struct ftrace_ops_hash *old_hash) 4123 { 4124 struct ftrace_ops *op; 4125 4126 if (!ftrace_enabled) 4127 return; 4128 4129 if (ops->flags & FTRACE_OPS_FL_ENABLED) { 4130 ftrace_run_modify_code(ops, FTRACE_UPDATE_CALLS, old_hash); 4131 return; 4132 } 4133 4134 /* 4135 * If this is the shared global_ops filter, then we need to 4136 * check if there is another ops that shares it, is enabled. 4137 * If so, we still need to run the modify code. 4138 */ 4139 if (ops->func_hash != &global_ops.local_hash) 4140 return; 4141 4142 do_for_each_ftrace_op(op, ftrace_ops_list) { 4143 if (op->func_hash == &global_ops.local_hash && 4144 op->flags & FTRACE_OPS_FL_ENABLED) { 4145 ftrace_run_modify_code(op, FTRACE_UPDATE_CALLS, old_hash); 4146 /* Only need to do this once */ 4147 return; 4148 } 4149 } while_for_each_ftrace_op(op); 4150 } 4151 4152 static int 4153 ftrace_set_hash(struct ftrace_ops *ops, unsigned char *buf, int len, 4154 unsigned long ip, int remove, int reset, int enable) 4155 { 4156 struct ftrace_hash **orig_hash; 4157 struct ftrace_ops_hash old_hash_ops; 4158 struct ftrace_hash *old_hash; 4159 struct ftrace_hash *hash; 4160 int ret; 4161 4162 if (unlikely(ftrace_disabled)) 4163 return -ENODEV; 4164 4165 mutex_lock(&ops->func_hash->regex_lock); 4166 4167 if (enable) 4168 orig_hash = &ops->func_hash->filter_hash; 4169 else 4170 orig_hash = &ops->func_hash->notrace_hash; 4171 4172 if (reset) 4173 hash = alloc_ftrace_hash(FTRACE_HASH_DEFAULT_BITS); 4174 else 4175 hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, *orig_hash); 4176 4177 if (!hash) { 4178 ret = -ENOMEM; 4179 goto out_regex_unlock; 4180 } 4181 4182 if (buf && !ftrace_match_records(hash, buf, len)) { 4183 ret = -EINVAL; 4184 goto out_regex_unlock; 4185 } 4186 if (ip) { 4187 ret = ftrace_match_addr(hash, ip, remove); 4188 if (ret < 0) 4189 goto out_regex_unlock; 4190 } 4191 4192 mutex_lock(&ftrace_lock); 4193 old_hash = *orig_hash; 4194 old_hash_ops.filter_hash = ops->func_hash->filter_hash; 4195 old_hash_ops.notrace_hash = ops->func_hash->notrace_hash; 4196 ret = ftrace_hash_move(ops, enable, orig_hash, hash); 4197 if (!ret) { 4198 ftrace_ops_update_code(ops, &old_hash_ops); 4199 free_ftrace_hash_rcu(old_hash); 4200 } 4201 mutex_unlock(&ftrace_lock); 4202 4203 out_regex_unlock: 4204 mutex_unlock(&ops->func_hash->regex_lock); 4205 4206 free_ftrace_hash(hash); 4207 return ret; 4208 } 4209 4210 static int 4211 ftrace_set_addr(struct ftrace_ops *ops, unsigned long ip, int remove, 4212 int reset, int enable) 4213 { 4214 return ftrace_set_hash(ops, 0, 0, ip, remove, reset, enable); 4215 } 4216 4217 /** 4218 * ftrace_set_filter_ip - set a function to filter on in ftrace by address 4219 * @ops - the ops to set the filter with 4220 * @ip - the address to add to or remove from the filter. 4221 * @remove - non zero to remove the ip from the filter 4222 * @reset - non zero to reset all filters before applying this filter. 4223 * 4224 * Filters denote which functions should be enabled when tracing is enabled 4225 * If @ip is NULL, it failes to update filter. 4226 */ 4227 int ftrace_set_filter_ip(struct ftrace_ops *ops, unsigned long ip, 4228 int remove, int reset) 4229 { 4230 ftrace_ops_init(ops); 4231 return ftrace_set_addr(ops, ip, remove, reset, 1); 4232 } 4233 EXPORT_SYMBOL_GPL(ftrace_set_filter_ip); 4234 4235 static int 4236 ftrace_set_regex(struct ftrace_ops *ops, unsigned char *buf, int len, 4237 int reset, int enable) 4238 { 4239 return ftrace_set_hash(ops, buf, len, 0, 0, reset, enable); 4240 } 4241 4242 /** 4243 * ftrace_set_filter - set a function to filter on in ftrace 4244 * @ops - the ops to set the filter with 4245 * @buf - the string that holds the function filter text. 4246 * @len - the length of the string. 4247 * @reset - non zero to reset all filters before applying this filter. 4248 * 4249 * Filters denote which functions should be enabled when tracing is enabled. 4250 * If @buf is NULL and reset is set, all functions will be enabled for tracing. 4251 */ 4252 int ftrace_set_filter(struct ftrace_ops *ops, unsigned char *buf, 4253 int len, int reset) 4254 { 4255 ftrace_ops_init(ops); 4256 return ftrace_set_regex(ops, buf, len, reset, 1); 4257 } 4258 EXPORT_SYMBOL_GPL(ftrace_set_filter); 4259 4260 /** 4261 * ftrace_set_notrace - set a function to not trace in ftrace 4262 * @ops - the ops to set the notrace filter with 4263 * @buf - the string that holds the function notrace text. 4264 * @len - the length of the string. 4265 * @reset - non zero to reset all filters before applying this filter. 4266 * 4267 * Notrace Filters denote which functions should not be enabled when tracing 4268 * is enabled. If @buf is NULL and reset is set, all functions will be enabled 4269 * for tracing. 4270 */ 4271 int ftrace_set_notrace(struct ftrace_ops *ops, unsigned char *buf, 4272 int len, int reset) 4273 { 4274 ftrace_ops_init(ops); 4275 return ftrace_set_regex(ops, buf, len, reset, 0); 4276 } 4277 EXPORT_SYMBOL_GPL(ftrace_set_notrace); 4278 /** 4279 * ftrace_set_global_filter - set a function to filter on with global tracers 4280 * @buf - the string that holds the function filter text. 4281 * @len - the length of the string. 4282 * @reset - non zero to reset all filters before applying this filter. 4283 * 4284 * Filters denote which functions should be enabled when tracing is enabled. 4285 * If @buf is NULL and reset is set, all functions will be enabled for tracing. 4286 */ 4287 void ftrace_set_global_filter(unsigned char *buf, int len, int reset) 4288 { 4289 ftrace_set_regex(&global_ops, buf, len, reset, 1); 4290 } 4291 EXPORT_SYMBOL_GPL(ftrace_set_global_filter); 4292 4293 /** 4294 * ftrace_set_global_notrace - set a function to not trace with global tracers 4295 * @buf - the string that holds the function notrace text. 4296 * @len - the length of the string. 4297 * @reset - non zero to reset all filters before applying this filter. 4298 * 4299 * Notrace Filters denote which functions should not be enabled when tracing 4300 * is enabled. If @buf is NULL and reset is set, all functions will be enabled 4301 * for tracing. 4302 */ 4303 void ftrace_set_global_notrace(unsigned char *buf, int len, int reset) 4304 { 4305 ftrace_set_regex(&global_ops, buf, len, reset, 0); 4306 } 4307 EXPORT_SYMBOL_GPL(ftrace_set_global_notrace); 4308 4309 /* 4310 * command line interface to allow users to set filters on boot up. 4311 */ 4312 #define FTRACE_FILTER_SIZE COMMAND_LINE_SIZE 4313 static char ftrace_notrace_buf[FTRACE_FILTER_SIZE] __initdata; 4314 static char ftrace_filter_buf[FTRACE_FILTER_SIZE] __initdata; 4315 4316 /* Used by function selftest to not test if filter is set */ 4317 bool ftrace_filter_param __initdata; 4318 4319 static int __init set_ftrace_notrace(char *str) 4320 { 4321 ftrace_filter_param = true; 4322 strlcpy(ftrace_notrace_buf, str, FTRACE_FILTER_SIZE); 4323 return 1; 4324 } 4325 __setup("ftrace_notrace=", set_ftrace_notrace); 4326 4327 static int __init set_ftrace_filter(char *str) 4328 { 4329 ftrace_filter_param = true; 4330 strlcpy(ftrace_filter_buf, str, FTRACE_FILTER_SIZE); 4331 return 1; 4332 } 4333 __setup("ftrace_filter=", set_ftrace_filter); 4334 4335 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 4336 static char ftrace_graph_buf[FTRACE_FILTER_SIZE] __initdata; 4337 static char ftrace_graph_notrace_buf[FTRACE_FILTER_SIZE] __initdata; 4338 static int ftrace_set_func(unsigned long *array, int *idx, int size, char *buffer); 4339 4340 static unsigned long save_global_trampoline; 4341 static unsigned long save_global_flags; 4342 4343 static int __init set_graph_function(char *str) 4344 { 4345 strlcpy(ftrace_graph_buf, str, FTRACE_FILTER_SIZE); 4346 return 1; 4347 } 4348 __setup("ftrace_graph_filter=", set_graph_function); 4349 4350 static int __init set_graph_notrace_function(char *str) 4351 { 4352 strlcpy(ftrace_graph_notrace_buf, str, FTRACE_FILTER_SIZE); 4353 return 1; 4354 } 4355 __setup("ftrace_graph_notrace=", set_graph_notrace_function); 4356 4357 static void __init set_ftrace_early_graph(char *buf, int enable) 4358 { 4359 int ret; 4360 char *func; 4361 unsigned long *table = ftrace_graph_funcs; 4362 int *count = &ftrace_graph_count; 4363 4364 if (!enable) { 4365 table = ftrace_graph_notrace_funcs; 4366 count = &ftrace_graph_notrace_count; 4367 } 4368 4369 while (buf) { 4370 func = strsep(&buf, ","); 4371 /* we allow only one expression at a time */ 4372 ret = ftrace_set_func(table, count, FTRACE_GRAPH_MAX_FUNCS, func); 4373 if (ret) 4374 printk(KERN_DEBUG "ftrace: function %s not " 4375 "traceable\n", func); 4376 } 4377 } 4378 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */ 4379 4380 void __init 4381 ftrace_set_early_filter(struct ftrace_ops *ops, char *buf, int enable) 4382 { 4383 char *func; 4384 4385 ftrace_ops_init(ops); 4386 4387 while (buf) { 4388 func = strsep(&buf, ","); 4389 ftrace_set_regex(ops, func, strlen(func), 0, enable); 4390 } 4391 } 4392 4393 static void __init set_ftrace_early_filters(void) 4394 { 4395 if (ftrace_filter_buf[0]) 4396 ftrace_set_early_filter(&global_ops, ftrace_filter_buf, 1); 4397 if (ftrace_notrace_buf[0]) 4398 ftrace_set_early_filter(&global_ops, ftrace_notrace_buf, 0); 4399 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 4400 if (ftrace_graph_buf[0]) 4401 set_ftrace_early_graph(ftrace_graph_buf, 1); 4402 if (ftrace_graph_notrace_buf[0]) 4403 set_ftrace_early_graph(ftrace_graph_notrace_buf, 0); 4404 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */ 4405 } 4406 4407 int ftrace_regex_release(struct inode *inode, struct file *file) 4408 { 4409 struct seq_file *m = (struct seq_file *)file->private_data; 4410 struct ftrace_ops_hash old_hash_ops; 4411 struct ftrace_iterator *iter; 4412 struct ftrace_hash **orig_hash; 4413 struct ftrace_hash *old_hash; 4414 struct trace_parser *parser; 4415 int filter_hash; 4416 int ret; 4417 4418 if (file->f_mode & FMODE_READ) { 4419 iter = m->private; 4420 seq_release(inode, file); 4421 } else 4422 iter = file->private_data; 4423 4424 parser = &iter->parser; 4425 if (trace_parser_loaded(parser)) { 4426 parser->buffer[parser->idx] = 0; 4427 ftrace_match_records(iter->hash, parser->buffer, parser->idx); 4428 } 4429 4430 trace_parser_put(parser); 4431 4432 mutex_lock(&iter->ops->func_hash->regex_lock); 4433 4434 if (file->f_mode & FMODE_WRITE) { 4435 filter_hash = !!(iter->flags & FTRACE_ITER_FILTER); 4436 4437 if (filter_hash) 4438 orig_hash = &iter->ops->func_hash->filter_hash; 4439 else 4440 orig_hash = &iter->ops->func_hash->notrace_hash; 4441 4442 mutex_lock(&ftrace_lock); 4443 old_hash = *orig_hash; 4444 old_hash_ops.filter_hash = iter->ops->func_hash->filter_hash; 4445 old_hash_ops.notrace_hash = iter->ops->func_hash->notrace_hash; 4446 ret = ftrace_hash_move(iter->ops, filter_hash, 4447 orig_hash, iter->hash); 4448 if (!ret) { 4449 ftrace_ops_update_code(iter->ops, &old_hash_ops); 4450 free_ftrace_hash_rcu(old_hash); 4451 } 4452 mutex_unlock(&ftrace_lock); 4453 } 4454 4455 mutex_unlock(&iter->ops->func_hash->regex_lock); 4456 free_ftrace_hash(iter->hash); 4457 kfree(iter); 4458 4459 return 0; 4460 } 4461 4462 static const struct file_operations ftrace_avail_fops = { 4463 .open = ftrace_avail_open, 4464 .read = seq_read, 4465 .llseek = seq_lseek, 4466 .release = seq_release_private, 4467 }; 4468 4469 static const struct file_operations ftrace_enabled_fops = { 4470 .open = ftrace_enabled_open, 4471 .read = seq_read, 4472 .llseek = seq_lseek, 4473 .release = seq_release_private, 4474 }; 4475 4476 static const struct file_operations ftrace_filter_fops = { 4477 .open = ftrace_filter_open, 4478 .read = seq_read, 4479 .write = ftrace_filter_write, 4480 .llseek = tracing_lseek, 4481 .release = ftrace_regex_release, 4482 }; 4483 4484 static const struct file_operations ftrace_notrace_fops = { 4485 .open = ftrace_notrace_open, 4486 .read = seq_read, 4487 .write = ftrace_notrace_write, 4488 .llseek = tracing_lseek, 4489 .release = ftrace_regex_release, 4490 }; 4491 4492 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 4493 4494 static DEFINE_MUTEX(graph_lock); 4495 4496 int ftrace_graph_count; 4497 int ftrace_graph_notrace_count; 4498 unsigned long ftrace_graph_funcs[FTRACE_GRAPH_MAX_FUNCS] __read_mostly; 4499 unsigned long ftrace_graph_notrace_funcs[FTRACE_GRAPH_MAX_FUNCS] __read_mostly; 4500 4501 struct ftrace_graph_data { 4502 unsigned long *table; 4503 size_t size; 4504 int *count; 4505 const struct seq_operations *seq_ops; 4506 }; 4507 4508 static void * 4509 __g_next(struct seq_file *m, loff_t *pos) 4510 { 4511 struct ftrace_graph_data *fgd = m->private; 4512 4513 if (*pos >= *fgd->count) 4514 return NULL; 4515 return &fgd->table[*pos]; 4516 } 4517 4518 static void * 4519 g_next(struct seq_file *m, void *v, loff_t *pos) 4520 { 4521 (*pos)++; 4522 return __g_next(m, pos); 4523 } 4524 4525 static void *g_start(struct seq_file *m, loff_t *pos) 4526 { 4527 struct ftrace_graph_data *fgd = m->private; 4528 4529 mutex_lock(&graph_lock); 4530 4531 /* Nothing, tell g_show to print all functions are enabled */ 4532 if (!*fgd->count && !*pos) 4533 return (void *)1; 4534 4535 return __g_next(m, pos); 4536 } 4537 4538 static void g_stop(struct seq_file *m, void *p) 4539 { 4540 mutex_unlock(&graph_lock); 4541 } 4542 4543 static int g_show(struct seq_file *m, void *v) 4544 { 4545 unsigned long *ptr = v; 4546 4547 if (!ptr) 4548 return 0; 4549 4550 if (ptr == (unsigned long *)1) { 4551 struct ftrace_graph_data *fgd = m->private; 4552 4553 if (fgd->table == ftrace_graph_funcs) 4554 seq_puts(m, "#### all functions enabled ####\n"); 4555 else 4556 seq_puts(m, "#### no functions disabled ####\n"); 4557 return 0; 4558 } 4559 4560 seq_printf(m, "%ps\n", (void *)*ptr); 4561 4562 return 0; 4563 } 4564 4565 static const struct seq_operations ftrace_graph_seq_ops = { 4566 .start = g_start, 4567 .next = g_next, 4568 .stop = g_stop, 4569 .show = g_show, 4570 }; 4571 4572 static int 4573 __ftrace_graph_open(struct inode *inode, struct file *file, 4574 struct ftrace_graph_data *fgd) 4575 { 4576 int ret = 0; 4577 4578 mutex_lock(&graph_lock); 4579 if ((file->f_mode & FMODE_WRITE) && 4580 (file->f_flags & O_TRUNC)) { 4581 *fgd->count = 0; 4582 memset(fgd->table, 0, fgd->size * sizeof(*fgd->table)); 4583 } 4584 mutex_unlock(&graph_lock); 4585 4586 if (file->f_mode & FMODE_READ) { 4587 ret = seq_open(file, fgd->seq_ops); 4588 if (!ret) { 4589 struct seq_file *m = file->private_data; 4590 m->private = fgd; 4591 } 4592 } else 4593 file->private_data = fgd; 4594 4595 return ret; 4596 } 4597 4598 static int 4599 ftrace_graph_open(struct inode *inode, struct file *file) 4600 { 4601 struct ftrace_graph_data *fgd; 4602 4603 if (unlikely(ftrace_disabled)) 4604 return -ENODEV; 4605 4606 fgd = kmalloc(sizeof(*fgd), GFP_KERNEL); 4607 if (fgd == NULL) 4608 return -ENOMEM; 4609 4610 fgd->table = ftrace_graph_funcs; 4611 fgd->size = FTRACE_GRAPH_MAX_FUNCS; 4612 fgd->count = &ftrace_graph_count; 4613 fgd->seq_ops = &ftrace_graph_seq_ops; 4614 4615 return __ftrace_graph_open(inode, file, fgd); 4616 } 4617 4618 static int 4619 ftrace_graph_notrace_open(struct inode *inode, struct file *file) 4620 { 4621 struct ftrace_graph_data *fgd; 4622 4623 if (unlikely(ftrace_disabled)) 4624 return -ENODEV; 4625 4626 fgd = kmalloc(sizeof(*fgd), GFP_KERNEL); 4627 if (fgd == NULL) 4628 return -ENOMEM; 4629 4630 fgd->table = ftrace_graph_notrace_funcs; 4631 fgd->size = FTRACE_GRAPH_MAX_FUNCS; 4632 fgd->count = &ftrace_graph_notrace_count; 4633 fgd->seq_ops = &ftrace_graph_seq_ops; 4634 4635 return __ftrace_graph_open(inode, file, fgd); 4636 } 4637 4638 static int 4639 ftrace_graph_release(struct inode *inode, struct file *file) 4640 { 4641 if (file->f_mode & FMODE_READ) { 4642 struct seq_file *m = file->private_data; 4643 4644 kfree(m->private); 4645 seq_release(inode, file); 4646 } else { 4647 kfree(file->private_data); 4648 } 4649 4650 return 0; 4651 } 4652 4653 static int 4654 ftrace_set_func(unsigned long *array, int *idx, int size, char *buffer) 4655 { 4656 struct ftrace_glob func_g; 4657 struct dyn_ftrace *rec; 4658 struct ftrace_page *pg; 4659 int fail = 1; 4660 int not; 4661 bool exists; 4662 int i; 4663 4664 /* decode regex */ 4665 func_g.type = filter_parse_regex(buffer, strlen(buffer), 4666 &func_g.search, ¬); 4667 if (!not && *idx >= size) 4668 return -EBUSY; 4669 4670 func_g.len = strlen(func_g.search); 4671 4672 mutex_lock(&ftrace_lock); 4673 4674 if (unlikely(ftrace_disabled)) { 4675 mutex_unlock(&ftrace_lock); 4676 return -ENODEV; 4677 } 4678 4679 do_for_each_ftrace_rec(pg, rec) { 4680 4681 if (ftrace_match_record(rec, &func_g, NULL, 0)) { 4682 /* if it is in the array */ 4683 exists = false; 4684 for (i = 0; i < *idx; i++) { 4685 if (array[i] == rec->ip) { 4686 exists = true; 4687 break; 4688 } 4689 } 4690 4691 if (!not) { 4692 fail = 0; 4693 if (!exists) { 4694 array[(*idx)++] = rec->ip; 4695 if (*idx >= size) 4696 goto out; 4697 } 4698 } else { 4699 if (exists) { 4700 array[i] = array[--(*idx)]; 4701 array[*idx] = 0; 4702 fail = 0; 4703 } 4704 } 4705 } 4706 } while_for_each_ftrace_rec(); 4707 out: 4708 mutex_unlock(&ftrace_lock); 4709 4710 if (fail) 4711 return -EINVAL; 4712 4713 return 0; 4714 } 4715 4716 static ssize_t 4717 ftrace_graph_write(struct file *file, const char __user *ubuf, 4718 size_t cnt, loff_t *ppos) 4719 { 4720 struct trace_parser parser; 4721 ssize_t read, ret = 0; 4722 struct ftrace_graph_data *fgd = file->private_data; 4723 4724 if (!cnt) 4725 return 0; 4726 4727 if (trace_parser_get_init(&parser, FTRACE_BUFF_MAX)) 4728 return -ENOMEM; 4729 4730 read = trace_get_user(&parser, ubuf, cnt, ppos); 4731 4732 if (read >= 0 && trace_parser_loaded((&parser))) { 4733 parser.buffer[parser.idx] = 0; 4734 4735 mutex_lock(&graph_lock); 4736 4737 /* we allow only one expression at a time */ 4738 ret = ftrace_set_func(fgd->table, fgd->count, fgd->size, 4739 parser.buffer); 4740 4741 mutex_unlock(&graph_lock); 4742 } 4743 4744 if (!ret) 4745 ret = read; 4746 4747 trace_parser_put(&parser); 4748 4749 return ret; 4750 } 4751 4752 static const struct file_operations ftrace_graph_fops = { 4753 .open = ftrace_graph_open, 4754 .read = seq_read, 4755 .write = ftrace_graph_write, 4756 .llseek = tracing_lseek, 4757 .release = ftrace_graph_release, 4758 }; 4759 4760 static const struct file_operations ftrace_graph_notrace_fops = { 4761 .open = ftrace_graph_notrace_open, 4762 .read = seq_read, 4763 .write = ftrace_graph_write, 4764 .llseek = tracing_lseek, 4765 .release = ftrace_graph_release, 4766 }; 4767 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */ 4768 4769 void ftrace_create_filter_files(struct ftrace_ops *ops, 4770 struct dentry *parent) 4771 { 4772 4773 trace_create_file("set_ftrace_filter", 0644, parent, 4774 ops, &ftrace_filter_fops); 4775 4776 trace_create_file("set_ftrace_notrace", 0644, parent, 4777 ops, &ftrace_notrace_fops); 4778 } 4779 4780 /* 4781 * The name "destroy_filter_files" is really a misnomer. Although 4782 * in the future, it may actualy delete the files, but this is 4783 * really intended to make sure the ops passed in are disabled 4784 * and that when this function returns, the caller is free to 4785 * free the ops. 4786 * 4787 * The "destroy" name is only to match the "create" name that this 4788 * should be paired with. 4789 */ 4790 void ftrace_destroy_filter_files(struct ftrace_ops *ops) 4791 { 4792 mutex_lock(&ftrace_lock); 4793 if (ops->flags & FTRACE_OPS_FL_ENABLED) 4794 ftrace_shutdown(ops, 0); 4795 ops->flags |= FTRACE_OPS_FL_DELETED; 4796 mutex_unlock(&ftrace_lock); 4797 } 4798 4799 static __init int ftrace_init_dyn_tracefs(struct dentry *d_tracer) 4800 { 4801 4802 trace_create_file("available_filter_functions", 0444, 4803 d_tracer, NULL, &ftrace_avail_fops); 4804 4805 trace_create_file("enabled_functions", 0444, 4806 d_tracer, NULL, &ftrace_enabled_fops); 4807 4808 ftrace_create_filter_files(&global_ops, d_tracer); 4809 4810 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 4811 trace_create_file("set_graph_function", 0444, d_tracer, 4812 NULL, 4813 &ftrace_graph_fops); 4814 trace_create_file("set_graph_notrace", 0444, d_tracer, 4815 NULL, 4816 &ftrace_graph_notrace_fops); 4817 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */ 4818 4819 return 0; 4820 } 4821 4822 static int ftrace_cmp_ips(const void *a, const void *b) 4823 { 4824 const unsigned long *ipa = a; 4825 const unsigned long *ipb = b; 4826 4827 if (*ipa > *ipb) 4828 return 1; 4829 if (*ipa < *ipb) 4830 return -1; 4831 return 0; 4832 } 4833 4834 static int ftrace_process_locs(struct module *mod, 4835 unsigned long *start, 4836 unsigned long *end) 4837 { 4838 struct ftrace_page *start_pg; 4839 struct ftrace_page *pg; 4840 struct dyn_ftrace *rec; 4841 unsigned long count; 4842 unsigned long *p; 4843 unsigned long addr; 4844 unsigned long flags = 0; /* Shut up gcc */ 4845 int ret = -ENOMEM; 4846 4847 count = end - start; 4848 4849 if (!count) 4850 return 0; 4851 4852 sort(start, count, sizeof(*start), 4853 ftrace_cmp_ips, NULL); 4854 4855 start_pg = ftrace_allocate_pages(count); 4856 if (!start_pg) 4857 return -ENOMEM; 4858 4859 mutex_lock(&ftrace_lock); 4860 4861 /* 4862 * Core and each module needs their own pages, as 4863 * modules will free them when they are removed. 4864 * Force a new page to be allocated for modules. 4865 */ 4866 if (!mod) { 4867 WARN_ON(ftrace_pages || ftrace_pages_start); 4868 /* First initialization */ 4869 ftrace_pages = ftrace_pages_start = start_pg; 4870 } else { 4871 if (!ftrace_pages) 4872 goto out; 4873 4874 if (WARN_ON(ftrace_pages->next)) { 4875 /* Hmm, we have free pages? */ 4876 while (ftrace_pages->next) 4877 ftrace_pages = ftrace_pages->next; 4878 } 4879 4880 ftrace_pages->next = start_pg; 4881 } 4882 4883 p = start; 4884 pg = start_pg; 4885 while (p < end) { 4886 addr = ftrace_call_adjust(*p++); 4887 /* 4888 * Some architecture linkers will pad between 4889 * the different mcount_loc sections of different 4890 * object files to satisfy alignments. 4891 * Skip any NULL pointers. 4892 */ 4893 if (!addr) 4894 continue; 4895 4896 if (pg->index == pg->size) { 4897 /* We should have allocated enough */ 4898 if (WARN_ON(!pg->next)) 4899 break; 4900 pg = pg->next; 4901 } 4902 4903 rec = &pg->records[pg->index++]; 4904 rec->ip = addr; 4905 } 4906 4907 /* We should have used all pages */ 4908 WARN_ON(pg->next); 4909 4910 /* Assign the last page to ftrace_pages */ 4911 ftrace_pages = pg; 4912 4913 /* 4914 * We only need to disable interrupts on start up 4915 * because we are modifying code that an interrupt 4916 * may execute, and the modification is not atomic. 4917 * But for modules, nothing runs the code we modify 4918 * until we are finished with it, and there's no 4919 * reason to cause large interrupt latencies while we do it. 4920 */ 4921 if (!mod) 4922 local_irq_save(flags); 4923 ftrace_update_code(mod, start_pg); 4924 if (!mod) 4925 local_irq_restore(flags); 4926 ret = 0; 4927 out: 4928 mutex_unlock(&ftrace_lock); 4929 4930 return ret; 4931 } 4932 4933 #ifdef CONFIG_MODULES 4934 4935 #define next_to_ftrace_page(p) container_of(p, struct ftrace_page, next) 4936 4937 static int referenced_filters(struct dyn_ftrace *rec) 4938 { 4939 struct ftrace_ops *ops; 4940 int cnt = 0; 4941 4942 for (ops = ftrace_ops_list; ops != &ftrace_list_end; ops = ops->next) { 4943 if (ops_references_rec(ops, rec)) 4944 cnt++; 4945 } 4946 4947 return cnt; 4948 } 4949 4950 void ftrace_release_mod(struct module *mod) 4951 { 4952 struct dyn_ftrace *rec; 4953 struct ftrace_page **last_pg; 4954 struct ftrace_page *pg; 4955 int order; 4956 4957 mutex_lock(&ftrace_lock); 4958 4959 if (ftrace_disabled) 4960 goto out_unlock; 4961 4962 /* 4963 * Each module has its own ftrace_pages, remove 4964 * them from the list. 4965 */ 4966 last_pg = &ftrace_pages_start; 4967 for (pg = ftrace_pages_start; pg; pg = *last_pg) { 4968 rec = &pg->records[0]; 4969 if (within_module_core(rec->ip, mod)) { 4970 /* 4971 * As core pages are first, the first 4972 * page should never be a module page. 4973 */ 4974 if (WARN_ON(pg == ftrace_pages_start)) 4975 goto out_unlock; 4976 4977 /* Check if we are deleting the last page */ 4978 if (pg == ftrace_pages) 4979 ftrace_pages = next_to_ftrace_page(last_pg); 4980 4981 *last_pg = pg->next; 4982 order = get_count_order(pg->size / ENTRIES_PER_PAGE); 4983 free_pages((unsigned long)pg->records, order); 4984 kfree(pg); 4985 } else 4986 last_pg = &pg->next; 4987 } 4988 out_unlock: 4989 mutex_unlock(&ftrace_lock); 4990 } 4991 4992 void ftrace_module_enable(struct module *mod) 4993 { 4994 struct dyn_ftrace *rec; 4995 struct ftrace_page *pg; 4996 4997 mutex_lock(&ftrace_lock); 4998 4999 if (ftrace_disabled) 5000 goto out_unlock; 5001 5002 /* 5003 * If the tracing is enabled, go ahead and enable the record. 5004 * 5005 * The reason not to enable the record immediatelly is the 5006 * inherent check of ftrace_make_nop/ftrace_make_call for 5007 * correct previous instructions. Making first the NOP 5008 * conversion puts the module to the correct state, thus 5009 * passing the ftrace_make_call check. 5010 * 5011 * We also delay this to after the module code already set the 5012 * text to read-only, as we now need to set it back to read-write 5013 * so that we can modify the text. 5014 */ 5015 if (ftrace_start_up) 5016 ftrace_arch_code_modify_prepare(); 5017 5018 do_for_each_ftrace_rec(pg, rec) { 5019 int cnt; 5020 /* 5021 * do_for_each_ftrace_rec() is a double loop. 5022 * module text shares the pg. If a record is 5023 * not part of this module, then skip this pg, 5024 * which the "break" will do. 5025 */ 5026 if (!within_module_core(rec->ip, mod)) 5027 break; 5028 5029 cnt = 0; 5030 5031 /* 5032 * When adding a module, we need to check if tracers are 5033 * currently enabled and if they are, and can trace this record, 5034 * we need to enable the module functions as well as update the 5035 * reference counts for those function records. 5036 */ 5037 if (ftrace_start_up) 5038 cnt += referenced_filters(rec); 5039 5040 /* This clears FTRACE_FL_DISABLED */ 5041 rec->flags = cnt; 5042 5043 if (ftrace_start_up && cnt) { 5044 int failed = __ftrace_replace_code(rec, 1); 5045 if (failed) { 5046 ftrace_bug(failed, rec); 5047 goto out_loop; 5048 } 5049 } 5050 5051 } while_for_each_ftrace_rec(); 5052 5053 out_loop: 5054 if (ftrace_start_up) 5055 ftrace_arch_code_modify_post_process(); 5056 5057 out_unlock: 5058 mutex_unlock(&ftrace_lock); 5059 } 5060 5061 void ftrace_module_init(struct module *mod) 5062 { 5063 if (ftrace_disabled || !mod->num_ftrace_callsites) 5064 return; 5065 5066 ftrace_process_locs(mod, mod->ftrace_callsites, 5067 mod->ftrace_callsites + mod->num_ftrace_callsites); 5068 } 5069 #endif /* CONFIG_MODULES */ 5070 5071 void __init ftrace_init(void) 5072 { 5073 extern unsigned long __start_mcount_loc[]; 5074 extern unsigned long __stop_mcount_loc[]; 5075 unsigned long count, flags; 5076 int ret; 5077 5078 local_irq_save(flags); 5079 ret = ftrace_dyn_arch_init(); 5080 local_irq_restore(flags); 5081 if (ret) 5082 goto failed; 5083 5084 count = __stop_mcount_loc - __start_mcount_loc; 5085 if (!count) { 5086 pr_info("ftrace: No functions to be traced?\n"); 5087 goto failed; 5088 } 5089 5090 pr_info("ftrace: allocating %ld entries in %ld pages\n", 5091 count, count / ENTRIES_PER_PAGE + 1); 5092 5093 last_ftrace_enabled = ftrace_enabled = 1; 5094 5095 ret = ftrace_process_locs(NULL, 5096 __start_mcount_loc, 5097 __stop_mcount_loc); 5098 5099 set_ftrace_early_filters(); 5100 5101 return; 5102 failed: 5103 ftrace_disabled = 1; 5104 } 5105 5106 /* Do nothing if arch does not support this */ 5107 void __weak arch_ftrace_update_trampoline(struct ftrace_ops *ops) 5108 { 5109 } 5110 5111 static void ftrace_update_trampoline(struct ftrace_ops *ops) 5112 { 5113 5114 /* 5115 * Currently there's no safe way to free a trampoline when the kernel 5116 * is configured with PREEMPT. That is because a task could be preempted 5117 * when it jumped to the trampoline, it may be preempted for a long time 5118 * depending on the system load, and currently there's no way to know 5119 * when it will be off the trampoline. If the trampoline is freed 5120 * too early, when the task runs again, it will be executing on freed 5121 * memory and crash. 5122 */ 5123 #ifdef CONFIG_PREEMPT 5124 /* Currently, only non dynamic ops can have a trampoline */ 5125 if (ops->flags & FTRACE_OPS_FL_DYNAMIC) 5126 return; 5127 #endif 5128 5129 arch_ftrace_update_trampoline(ops); 5130 } 5131 5132 #else 5133 5134 static struct ftrace_ops global_ops = { 5135 .func = ftrace_stub, 5136 .flags = FTRACE_OPS_FL_RECURSION_SAFE | 5137 FTRACE_OPS_FL_INITIALIZED | 5138 FTRACE_OPS_FL_PID, 5139 }; 5140 5141 static int __init ftrace_nodyn_init(void) 5142 { 5143 ftrace_enabled = 1; 5144 return 0; 5145 } 5146 core_initcall(ftrace_nodyn_init); 5147 5148 static inline int ftrace_init_dyn_tracefs(struct dentry *d_tracer) { return 0; } 5149 static inline void ftrace_startup_enable(int command) { } 5150 static inline void ftrace_startup_all(int command) { } 5151 /* Keep as macros so we do not need to define the commands */ 5152 # define ftrace_startup(ops, command) \ 5153 ({ \ 5154 int ___ret = __register_ftrace_function(ops); \ 5155 if (!___ret) \ 5156 (ops)->flags |= FTRACE_OPS_FL_ENABLED; \ 5157 ___ret; \ 5158 }) 5159 # define ftrace_shutdown(ops, command) \ 5160 ({ \ 5161 int ___ret = __unregister_ftrace_function(ops); \ 5162 if (!___ret) \ 5163 (ops)->flags &= ~FTRACE_OPS_FL_ENABLED; \ 5164 ___ret; \ 5165 }) 5166 5167 # define ftrace_startup_sysctl() do { } while (0) 5168 # define ftrace_shutdown_sysctl() do { } while (0) 5169 5170 static inline int 5171 ftrace_ops_test(struct ftrace_ops *ops, unsigned long ip, void *regs) 5172 { 5173 return 1; 5174 } 5175 5176 static void ftrace_update_trampoline(struct ftrace_ops *ops) 5177 { 5178 } 5179 5180 #endif /* CONFIG_DYNAMIC_FTRACE */ 5181 5182 __init void ftrace_init_global_array_ops(struct trace_array *tr) 5183 { 5184 tr->ops = &global_ops; 5185 tr->ops->private = tr; 5186 } 5187 5188 void ftrace_init_array_ops(struct trace_array *tr, ftrace_func_t func) 5189 { 5190 /* If we filter on pids, update to use the pid function */ 5191 if (tr->flags & TRACE_ARRAY_FL_GLOBAL) { 5192 if (WARN_ON(tr->ops->func != ftrace_stub)) 5193 printk("ftrace ops had %pS for function\n", 5194 tr->ops->func); 5195 } 5196 tr->ops->func = func; 5197 tr->ops->private = tr; 5198 } 5199 5200 void ftrace_reset_array_ops(struct trace_array *tr) 5201 { 5202 tr->ops->func = ftrace_stub; 5203 } 5204 5205 static inline void 5206 __ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip, 5207 struct ftrace_ops *ignored, struct pt_regs *regs) 5208 { 5209 struct ftrace_ops *op; 5210 int bit; 5211 5212 bit = trace_test_and_set_recursion(TRACE_LIST_START, TRACE_LIST_MAX); 5213 if (bit < 0) 5214 return; 5215 5216 /* 5217 * Some of the ops may be dynamically allocated, 5218 * they must be freed after a synchronize_sched(). 5219 */ 5220 preempt_disable_notrace(); 5221 5222 do_for_each_ftrace_op(op, ftrace_ops_list) { 5223 /* 5224 * Check the following for each ops before calling their func: 5225 * if RCU flag is set, then rcu_is_watching() must be true 5226 * if PER_CPU is set, then ftrace_function_local_disable() 5227 * must be false 5228 * Otherwise test if the ip matches the ops filter 5229 * 5230 * If any of the above fails then the op->func() is not executed. 5231 */ 5232 if ((!(op->flags & FTRACE_OPS_FL_RCU) || rcu_is_watching()) && 5233 (!(op->flags & FTRACE_OPS_FL_PER_CPU) || 5234 !ftrace_function_local_disabled(op)) && 5235 ftrace_ops_test(op, ip, regs)) { 5236 5237 if (FTRACE_WARN_ON(!op->func)) { 5238 pr_warn("op=%p %pS\n", op, op); 5239 goto out; 5240 } 5241 op->func(ip, parent_ip, op, regs); 5242 } 5243 } while_for_each_ftrace_op(op); 5244 out: 5245 preempt_enable_notrace(); 5246 trace_clear_recursion(bit); 5247 } 5248 5249 /* 5250 * Some archs only support passing ip and parent_ip. Even though 5251 * the list function ignores the op parameter, we do not want any 5252 * C side effects, where a function is called without the caller 5253 * sending a third parameter. 5254 * Archs are to support both the regs and ftrace_ops at the same time. 5255 * If they support ftrace_ops, it is assumed they support regs. 5256 * If call backs want to use regs, they must either check for regs 5257 * being NULL, or CONFIG_DYNAMIC_FTRACE_WITH_REGS. 5258 * Note, CONFIG_DYNAMIC_FTRACE_WITH_REGS expects a full regs to be saved. 5259 * An architecture can pass partial regs with ftrace_ops and still 5260 * set the ARCH_SUPPORTS_FTRACE_OPS. 5261 */ 5262 #if ARCH_SUPPORTS_FTRACE_OPS 5263 static void ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip, 5264 struct ftrace_ops *op, struct pt_regs *regs) 5265 { 5266 __ftrace_ops_list_func(ip, parent_ip, NULL, regs); 5267 } 5268 #else 5269 static void ftrace_ops_no_ops(unsigned long ip, unsigned long parent_ip) 5270 { 5271 __ftrace_ops_list_func(ip, parent_ip, NULL, NULL); 5272 } 5273 #endif 5274 5275 /* 5276 * If there's only one function registered but it does not support 5277 * recursion, needs RCU protection and/or requires per cpu handling, then 5278 * this function will be called by the mcount trampoline. 5279 */ 5280 static void ftrace_ops_assist_func(unsigned long ip, unsigned long parent_ip, 5281 struct ftrace_ops *op, struct pt_regs *regs) 5282 { 5283 int bit; 5284 5285 if ((op->flags & FTRACE_OPS_FL_RCU) && !rcu_is_watching()) 5286 return; 5287 5288 bit = trace_test_and_set_recursion(TRACE_LIST_START, TRACE_LIST_MAX); 5289 if (bit < 0) 5290 return; 5291 5292 preempt_disable_notrace(); 5293 5294 if (!(op->flags & FTRACE_OPS_FL_PER_CPU) || 5295 !ftrace_function_local_disabled(op)) { 5296 op->func(ip, parent_ip, op, regs); 5297 } 5298 5299 preempt_enable_notrace(); 5300 trace_clear_recursion(bit); 5301 } 5302 5303 /** 5304 * ftrace_ops_get_func - get the function a trampoline should call 5305 * @ops: the ops to get the function for 5306 * 5307 * Normally the mcount trampoline will call the ops->func, but there 5308 * are times that it should not. For example, if the ops does not 5309 * have its own recursion protection, then it should call the 5310 * ftrace_ops_recurs_func() instead. 5311 * 5312 * Returns the function that the trampoline should call for @ops. 5313 */ 5314 ftrace_func_t ftrace_ops_get_func(struct ftrace_ops *ops) 5315 { 5316 /* 5317 * If the function does not handle recursion, needs to be RCU safe, 5318 * or does per cpu logic, then we need to call the assist handler. 5319 */ 5320 if (!(ops->flags & FTRACE_OPS_FL_RECURSION_SAFE) || 5321 ops->flags & (FTRACE_OPS_FL_RCU | FTRACE_OPS_FL_PER_CPU)) 5322 return ftrace_ops_assist_func; 5323 5324 return ops->func; 5325 } 5326 5327 static void clear_ftrace_swapper(void) 5328 { 5329 struct task_struct *p; 5330 int cpu; 5331 5332 get_online_cpus(); 5333 for_each_online_cpu(cpu) { 5334 p = idle_task(cpu); 5335 clear_tsk_trace_trace(p); 5336 } 5337 put_online_cpus(); 5338 } 5339 5340 static void set_ftrace_swapper(void) 5341 { 5342 struct task_struct *p; 5343 int cpu; 5344 5345 get_online_cpus(); 5346 for_each_online_cpu(cpu) { 5347 p = idle_task(cpu); 5348 set_tsk_trace_trace(p); 5349 } 5350 put_online_cpus(); 5351 } 5352 5353 static void clear_ftrace_pid(struct pid *pid) 5354 { 5355 struct task_struct *p; 5356 5357 rcu_read_lock(); 5358 do_each_pid_task(pid, PIDTYPE_PID, p) { 5359 clear_tsk_trace_trace(p); 5360 } while_each_pid_task(pid, PIDTYPE_PID, p); 5361 rcu_read_unlock(); 5362 5363 put_pid(pid); 5364 } 5365 5366 static void set_ftrace_pid(struct pid *pid) 5367 { 5368 struct task_struct *p; 5369 5370 rcu_read_lock(); 5371 do_each_pid_task(pid, PIDTYPE_PID, p) { 5372 set_tsk_trace_trace(p); 5373 } while_each_pid_task(pid, PIDTYPE_PID, p); 5374 rcu_read_unlock(); 5375 } 5376 5377 static void clear_ftrace_pid_task(struct pid *pid) 5378 { 5379 if (pid == ftrace_swapper_pid) 5380 clear_ftrace_swapper(); 5381 else 5382 clear_ftrace_pid(pid); 5383 } 5384 5385 static void set_ftrace_pid_task(struct pid *pid) 5386 { 5387 if (pid == ftrace_swapper_pid) 5388 set_ftrace_swapper(); 5389 else 5390 set_ftrace_pid(pid); 5391 } 5392 5393 static int ftrace_pid_add(int p) 5394 { 5395 struct pid *pid; 5396 struct ftrace_pid *fpid; 5397 int ret = -EINVAL; 5398 5399 mutex_lock(&ftrace_lock); 5400 5401 if (!p) 5402 pid = ftrace_swapper_pid; 5403 else 5404 pid = find_get_pid(p); 5405 5406 if (!pid) 5407 goto out; 5408 5409 ret = 0; 5410 5411 list_for_each_entry(fpid, &ftrace_pids, list) 5412 if (fpid->pid == pid) 5413 goto out_put; 5414 5415 ret = -ENOMEM; 5416 5417 fpid = kmalloc(sizeof(*fpid), GFP_KERNEL); 5418 if (!fpid) 5419 goto out_put; 5420 5421 list_add(&fpid->list, &ftrace_pids); 5422 fpid->pid = pid; 5423 5424 set_ftrace_pid_task(pid); 5425 5426 ftrace_update_pid_func(); 5427 5428 ftrace_startup_all(0); 5429 5430 mutex_unlock(&ftrace_lock); 5431 return 0; 5432 5433 out_put: 5434 if (pid != ftrace_swapper_pid) 5435 put_pid(pid); 5436 5437 out: 5438 mutex_unlock(&ftrace_lock); 5439 return ret; 5440 } 5441 5442 static void ftrace_pid_reset(void) 5443 { 5444 struct ftrace_pid *fpid, *safe; 5445 5446 mutex_lock(&ftrace_lock); 5447 list_for_each_entry_safe(fpid, safe, &ftrace_pids, list) { 5448 struct pid *pid = fpid->pid; 5449 5450 clear_ftrace_pid_task(pid); 5451 5452 list_del(&fpid->list); 5453 kfree(fpid); 5454 } 5455 5456 ftrace_update_pid_func(); 5457 ftrace_startup_all(0); 5458 5459 mutex_unlock(&ftrace_lock); 5460 } 5461 5462 static void *fpid_start(struct seq_file *m, loff_t *pos) 5463 { 5464 mutex_lock(&ftrace_lock); 5465 5466 if (!ftrace_pids_enabled() && (!*pos)) 5467 return (void *) 1; 5468 5469 return seq_list_start(&ftrace_pids, *pos); 5470 } 5471 5472 static void *fpid_next(struct seq_file *m, void *v, loff_t *pos) 5473 { 5474 if (v == (void *)1) 5475 return NULL; 5476 5477 return seq_list_next(v, &ftrace_pids, pos); 5478 } 5479 5480 static void fpid_stop(struct seq_file *m, void *p) 5481 { 5482 mutex_unlock(&ftrace_lock); 5483 } 5484 5485 static int fpid_show(struct seq_file *m, void *v) 5486 { 5487 const struct ftrace_pid *fpid = list_entry(v, struct ftrace_pid, list); 5488 5489 if (v == (void *)1) { 5490 seq_puts(m, "no pid\n"); 5491 return 0; 5492 } 5493 5494 if (fpid->pid == ftrace_swapper_pid) 5495 seq_puts(m, "swapper tasks\n"); 5496 else 5497 seq_printf(m, "%u\n", pid_vnr(fpid->pid)); 5498 5499 return 0; 5500 } 5501 5502 static const struct seq_operations ftrace_pid_sops = { 5503 .start = fpid_start, 5504 .next = fpid_next, 5505 .stop = fpid_stop, 5506 .show = fpid_show, 5507 }; 5508 5509 static int 5510 ftrace_pid_open(struct inode *inode, struct file *file) 5511 { 5512 int ret = 0; 5513 5514 if ((file->f_mode & FMODE_WRITE) && 5515 (file->f_flags & O_TRUNC)) 5516 ftrace_pid_reset(); 5517 5518 if (file->f_mode & FMODE_READ) 5519 ret = seq_open(file, &ftrace_pid_sops); 5520 5521 return ret; 5522 } 5523 5524 static ssize_t 5525 ftrace_pid_write(struct file *filp, const char __user *ubuf, 5526 size_t cnt, loff_t *ppos) 5527 { 5528 char buf[64], *tmp; 5529 long val; 5530 int ret; 5531 5532 if (cnt >= sizeof(buf)) 5533 return -EINVAL; 5534 5535 if (copy_from_user(&buf, ubuf, cnt)) 5536 return -EFAULT; 5537 5538 buf[cnt] = 0; 5539 5540 /* 5541 * Allow "echo > set_ftrace_pid" or "echo -n '' > set_ftrace_pid" 5542 * to clean the filter quietly. 5543 */ 5544 tmp = strstrip(buf); 5545 if (strlen(tmp) == 0) 5546 return 1; 5547 5548 ret = kstrtol(tmp, 10, &val); 5549 if (ret < 0) 5550 return ret; 5551 5552 ret = ftrace_pid_add(val); 5553 5554 return ret ? ret : cnt; 5555 } 5556 5557 static int 5558 ftrace_pid_release(struct inode *inode, struct file *file) 5559 { 5560 if (file->f_mode & FMODE_READ) 5561 seq_release(inode, file); 5562 5563 return 0; 5564 } 5565 5566 static const struct file_operations ftrace_pid_fops = { 5567 .open = ftrace_pid_open, 5568 .write = ftrace_pid_write, 5569 .read = seq_read, 5570 .llseek = tracing_lseek, 5571 .release = ftrace_pid_release, 5572 }; 5573 5574 static __init int ftrace_init_tracefs(void) 5575 { 5576 struct dentry *d_tracer; 5577 5578 d_tracer = tracing_init_dentry(); 5579 if (IS_ERR(d_tracer)) 5580 return 0; 5581 5582 ftrace_init_dyn_tracefs(d_tracer); 5583 5584 trace_create_file("set_ftrace_pid", 0644, d_tracer, 5585 NULL, &ftrace_pid_fops); 5586 5587 ftrace_profile_tracefs(d_tracer); 5588 5589 return 0; 5590 } 5591 fs_initcall(ftrace_init_tracefs); 5592 5593 /** 5594 * ftrace_kill - kill ftrace 5595 * 5596 * This function should be used by panic code. It stops ftrace 5597 * but in a not so nice way. If you need to simply kill ftrace 5598 * from a non-atomic section, use ftrace_kill. 5599 */ 5600 void ftrace_kill(void) 5601 { 5602 ftrace_disabled = 1; 5603 ftrace_enabled = 0; 5604 clear_ftrace_function(); 5605 } 5606 5607 /** 5608 * Test if ftrace is dead or not. 5609 */ 5610 int ftrace_is_dead(void) 5611 { 5612 return ftrace_disabled; 5613 } 5614 5615 /** 5616 * register_ftrace_function - register a function for profiling 5617 * @ops - ops structure that holds the function for profiling. 5618 * 5619 * Register a function to be called by all functions in the 5620 * kernel. 5621 * 5622 * Note: @ops->func and all the functions it calls must be labeled 5623 * with "notrace", otherwise it will go into a 5624 * recursive loop. 5625 */ 5626 int register_ftrace_function(struct ftrace_ops *ops) 5627 { 5628 int ret = -1; 5629 5630 ftrace_ops_init(ops); 5631 5632 mutex_lock(&ftrace_lock); 5633 5634 ret = ftrace_startup(ops, 0); 5635 5636 mutex_unlock(&ftrace_lock); 5637 5638 return ret; 5639 } 5640 EXPORT_SYMBOL_GPL(register_ftrace_function); 5641 5642 /** 5643 * unregister_ftrace_function - unregister a function for profiling. 5644 * @ops - ops structure that holds the function to unregister 5645 * 5646 * Unregister a function that was added to be called by ftrace profiling. 5647 */ 5648 int unregister_ftrace_function(struct ftrace_ops *ops) 5649 { 5650 int ret; 5651 5652 mutex_lock(&ftrace_lock); 5653 ret = ftrace_shutdown(ops, 0); 5654 mutex_unlock(&ftrace_lock); 5655 5656 return ret; 5657 } 5658 EXPORT_SYMBOL_GPL(unregister_ftrace_function); 5659 5660 int 5661 ftrace_enable_sysctl(struct ctl_table *table, int write, 5662 void __user *buffer, size_t *lenp, 5663 loff_t *ppos) 5664 { 5665 int ret = -ENODEV; 5666 5667 mutex_lock(&ftrace_lock); 5668 5669 if (unlikely(ftrace_disabled)) 5670 goto out; 5671 5672 ret = proc_dointvec(table, write, buffer, lenp, ppos); 5673 5674 if (ret || !write || (last_ftrace_enabled == !!ftrace_enabled)) 5675 goto out; 5676 5677 last_ftrace_enabled = !!ftrace_enabled; 5678 5679 if (ftrace_enabled) { 5680 5681 /* we are starting ftrace again */ 5682 if (ftrace_ops_list != &ftrace_list_end) 5683 update_ftrace_function(); 5684 5685 ftrace_startup_sysctl(); 5686 5687 } else { 5688 /* stopping ftrace calls (just send to ftrace_stub) */ 5689 ftrace_trace_function = ftrace_stub; 5690 5691 ftrace_shutdown_sysctl(); 5692 } 5693 5694 out: 5695 mutex_unlock(&ftrace_lock); 5696 return ret; 5697 } 5698 5699 #ifdef CONFIG_FUNCTION_GRAPH_TRACER 5700 5701 static struct ftrace_ops graph_ops = { 5702 .func = ftrace_stub, 5703 .flags = FTRACE_OPS_FL_RECURSION_SAFE | 5704 FTRACE_OPS_FL_INITIALIZED | 5705 FTRACE_OPS_FL_PID | 5706 FTRACE_OPS_FL_STUB, 5707 #ifdef FTRACE_GRAPH_TRAMP_ADDR 5708 .trampoline = FTRACE_GRAPH_TRAMP_ADDR, 5709 /* trampoline_size is only needed for dynamically allocated tramps */ 5710 #endif 5711 ASSIGN_OPS_HASH(graph_ops, &global_ops.local_hash) 5712 }; 5713 5714 void ftrace_graph_sleep_time_control(bool enable) 5715 { 5716 fgraph_sleep_time = enable; 5717 } 5718 5719 void ftrace_graph_graph_time_control(bool enable) 5720 { 5721 fgraph_graph_time = enable; 5722 } 5723 5724 int ftrace_graph_entry_stub(struct ftrace_graph_ent *trace) 5725 { 5726 return 0; 5727 } 5728 5729 /* The callbacks that hook a function */ 5730 trace_func_graph_ret_t ftrace_graph_return = 5731 (trace_func_graph_ret_t)ftrace_stub; 5732 trace_func_graph_ent_t ftrace_graph_entry = ftrace_graph_entry_stub; 5733 static trace_func_graph_ent_t __ftrace_graph_entry = ftrace_graph_entry_stub; 5734 5735 /* Try to assign a return stack array on FTRACE_RETSTACK_ALLOC_SIZE tasks. */ 5736 static int alloc_retstack_tasklist(struct ftrace_ret_stack **ret_stack_list) 5737 { 5738 int i; 5739 int ret = 0; 5740 int start = 0, end = FTRACE_RETSTACK_ALLOC_SIZE; 5741 struct task_struct *g, *t; 5742 5743 for (i = 0; i < FTRACE_RETSTACK_ALLOC_SIZE; i++) { 5744 ret_stack_list[i] = kmalloc(FTRACE_RETFUNC_DEPTH 5745 * sizeof(struct ftrace_ret_stack), 5746 GFP_KERNEL); 5747 if (!ret_stack_list[i]) { 5748 start = 0; 5749 end = i; 5750 ret = -ENOMEM; 5751 goto free; 5752 } 5753 } 5754 5755 read_lock(&tasklist_lock); 5756 do_each_thread(g, t) { 5757 if (start == end) { 5758 ret = -EAGAIN; 5759 goto unlock; 5760 } 5761 5762 if (t->ret_stack == NULL) { 5763 atomic_set(&t->tracing_graph_pause, 0); 5764 atomic_set(&t->trace_overrun, 0); 5765 t->curr_ret_stack = -1; 5766 /* Make sure the tasks see the -1 first: */ 5767 smp_wmb(); 5768 t->ret_stack = ret_stack_list[start++]; 5769 } 5770 } while_each_thread(g, t); 5771 5772 unlock: 5773 read_unlock(&tasklist_lock); 5774 free: 5775 for (i = start; i < end; i++) 5776 kfree(ret_stack_list[i]); 5777 return ret; 5778 } 5779 5780 static void 5781 ftrace_graph_probe_sched_switch(void *ignore, bool preempt, 5782 struct task_struct *prev, struct task_struct *next) 5783 { 5784 unsigned long long timestamp; 5785 int index; 5786 5787 /* 5788 * Does the user want to count the time a function was asleep. 5789 * If so, do not update the time stamps. 5790 */ 5791 if (fgraph_sleep_time) 5792 return; 5793 5794 timestamp = trace_clock_local(); 5795 5796 prev->ftrace_timestamp = timestamp; 5797 5798 /* only process tasks that we timestamped */ 5799 if (!next->ftrace_timestamp) 5800 return; 5801 5802 /* 5803 * Update all the counters in next to make up for the 5804 * time next was sleeping. 5805 */ 5806 timestamp -= next->ftrace_timestamp; 5807 5808 for (index = next->curr_ret_stack; index >= 0; index--) 5809 next->ret_stack[index].calltime += timestamp; 5810 } 5811 5812 /* Allocate a return stack for each task */ 5813 static int start_graph_tracing(void) 5814 { 5815 struct ftrace_ret_stack **ret_stack_list; 5816 int ret, cpu; 5817 5818 ret_stack_list = kmalloc(FTRACE_RETSTACK_ALLOC_SIZE * 5819 sizeof(struct ftrace_ret_stack *), 5820 GFP_KERNEL); 5821 5822 if (!ret_stack_list) 5823 return -ENOMEM; 5824 5825 /* The cpu_boot init_task->ret_stack will never be freed */ 5826 for_each_online_cpu(cpu) { 5827 if (!idle_task(cpu)->ret_stack) 5828 ftrace_graph_init_idle_task(idle_task(cpu), cpu); 5829 } 5830 5831 do { 5832 ret = alloc_retstack_tasklist(ret_stack_list); 5833 } while (ret == -EAGAIN); 5834 5835 if (!ret) { 5836 ret = register_trace_sched_switch(ftrace_graph_probe_sched_switch, NULL); 5837 if (ret) 5838 pr_info("ftrace_graph: Couldn't activate tracepoint" 5839 " probe to kernel_sched_switch\n"); 5840 } 5841 5842 kfree(ret_stack_list); 5843 return ret; 5844 } 5845 5846 /* 5847 * Hibernation protection. 5848 * The state of the current task is too much unstable during 5849 * suspend/restore to disk. We want to protect against that. 5850 */ 5851 static int 5852 ftrace_suspend_notifier_call(struct notifier_block *bl, unsigned long state, 5853 void *unused) 5854 { 5855 switch (state) { 5856 case PM_HIBERNATION_PREPARE: 5857 pause_graph_tracing(); 5858 break; 5859 5860 case PM_POST_HIBERNATION: 5861 unpause_graph_tracing(); 5862 break; 5863 } 5864 return NOTIFY_DONE; 5865 } 5866 5867 static int ftrace_graph_entry_test(struct ftrace_graph_ent *trace) 5868 { 5869 if (!ftrace_ops_test(&global_ops, trace->func, NULL)) 5870 return 0; 5871 return __ftrace_graph_entry(trace); 5872 } 5873 5874 /* 5875 * The function graph tracer should only trace the functions defined 5876 * by set_ftrace_filter and set_ftrace_notrace. If another function 5877 * tracer ops is registered, the graph tracer requires testing the 5878 * function against the global ops, and not just trace any function 5879 * that any ftrace_ops registered. 5880 */ 5881 static void update_function_graph_func(void) 5882 { 5883 struct ftrace_ops *op; 5884 bool do_test = false; 5885 5886 /* 5887 * The graph and global ops share the same set of functions 5888 * to test. If any other ops is on the list, then 5889 * the graph tracing needs to test if its the function 5890 * it should call. 5891 */ 5892 do_for_each_ftrace_op(op, ftrace_ops_list) { 5893 if (op != &global_ops && op != &graph_ops && 5894 op != &ftrace_list_end) { 5895 do_test = true; 5896 /* in double loop, break out with goto */ 5897 goto out; 5898 } 5899 } while_for_each_ftrace_op(op); 5900 out: 5901 if (do_test) 5902 ftrace_graph_entry = ftrace_graph_entry_test; 5903 else 5904 ftrace_graph_entry = __ftrace_graph_entry; 5905 } 5906 5907 static struct notifier_block ftrace_suspend_notifier = { 5908 .notifier_call = ftrace_suspend_notifier_call, 5909 }; 5910 5911 int register_ftrace_graph(trace_func_graph_ret_t retfunc, 5912 trace_func_graph_ent_t entryfunc) 5913 { 5914 int ret = 0; 5915 5916 mutex_lock(&ftrace_lock); 5917 5918 /* we currently allow only one tracer registered at a time */ 5919 if (ftrace_graph_active) { 5920 ret = -EBUSY; 5921 goto out; 5922 } 5923 5924 register_pm_notifier(&ftrace_suspend_notifier); 5925 5926 ftrace_graph_active++; 5927 ret = start_graph_tracing(); 5928 if (ret) { 5929 ftrace_graph_active--; 5930 goto out; 5931 } 5932 5933 ftrace_graph_return = retfunc; 5934 5935 /* 5936 * Update the indirect function to the entryfunc, and the 5937 * function that gets called to the entry_test first. Then 5938 * call the update fgraph entry function to determine if 5939 * the entryfunc should be called directly or not. 5940 */ 5941 __ftrace_graph_entry = entryfunc; 5942 ftrace_graph_entry = ftrace_graph_entry_test; 5943 update_function_graph_func(); 5944 5945 ret = ftrace_startup(&graph_ops, FTRACE_START_FUNC_RET); 5946 out: 5947 mutex_unlock(&ftrace_lock); 5948 return ret; 5949 } 5950 5951 void unregister_ftrace_graph(void) 5952 { 5953 mutex_lock(&ftrace_lock); 5954 5955 if (unlikely(!ftrace_graph_active)) 5956 goto out; 5957 5958 ftrace_graph_active--; 5959 ftrace_graph_return = (trace_func_graph_ret_t)ftrace_stub; 5960 ftrace_graph_entry = ftrace_graph_entry_stub; 5961 __ftrace_graph_entry = ftrace_graph_entry_stub; 5962 ftrace_shutdown(&graph_ops, FTRACE_STOP_FUNC_RET); 5963 unregister_pm_notifier(&ftrace_suspend_notifier); 5964 unregister_trace_sched_switch(ftrace_graph_probe_sched_switch, NULL); 5965 5966 #ifdef CONFIG_DYNAMIC_FTRACE 5967 /* 5968 * Function graph does not allocate the trampoline, but 5969 * other global_ops do. We need to reset the ALLOC_TRAMP flag 5970 * if one was used. 5971 */ 5972 global_ops.trampoline = save_global_trampoline; 5973 if (save_global_flags & FTRACE_OPS_FL_ALLOC_TRAMP) 5974 global_ops.flags |= FTRACE_OPS_FL_ALLOC_TRAMP; 5975 #endif 5976 5977 out: 5978 mutex_unlock(&ftrace_lock); 5979 } 5980 5981 static DEFINE_PER_CPU(struct ftrace_ret_stack *, idle_ret_stack); 5982 5983 static void 5984 graph_init_task(struct task_struct *t, struct ftrace_ret_stack *ret_stack) 5985 { 5986 atomic_set(&t->tracing_graph_pause, 0); 5987 atomic_set(&t->trace_overrun, 0); 5988 t->ftrace_timestamp = 0; 5989 /* make curr_ret_stack visible before we add the ret_stack */ 5990 smp_wmb(); 5991 t->ret_stack = ret_stack; 5992 } 5993 5994 /* 5995 * Allocate a return stack for the idle task. May be the first 5996 * time through, or it may be done by CPU hotplug online. 5997 */ 5998 void ftrace_graph_init_idle_task(struct task_struct *t, int cpu) 5999 { 6000 t->curr_ret_stack = -1; 6001 /* 6002 * The idle task has no parent, it either has its own 6003 * stack or no stack at all. 6004 */ 6005 if (t->ret_stack) 6006 WARN_ON(t->ret_stack != per_cpu(idle_ret_stack, cpu)); 6007 6008 if (ftrace_graph_active) { 6009 struct ftrace_ret_stack *ret_stack; 6010 6011 ret_stack = per_cpu(idle_ret_stack, cpu); 6012 if (!ret_stack) { 6013 ret_stack = kmalloc(FTRACE_RETFUNC_DEPTH 6014 * sizeof(struct ftrace_ret_stack), 6015 GFP_KERNEL); 6016 if (!ret_stack) 6017 return; 6018 per_cpu(idle_ret_stack, cpu) = ret_stack; 6019 } 6020 graph_init_task(t, ret_stack); 6021 } 6022 } 6023 6024 /* Allocate a return stack for newly created task */ 6025 void ftrace_graph_init_task(struct task_struct *t) 6026 { 6027 /* Make sure we do not use the parent ret_stack */ 6028 t->ret_stack = NULL; 6029 t->curr_ret_stack = -1; 6030 6031 if (ftrace_graph_active) { 6032 struct ftrace_ret_stack *ret_stack; 6033 6034 ret_stack = kmalloc(FTRACE_RETFUNC_DEPTH 6035 * sizeof(struct ftrace_ret_stack), 6036 GFP_KERNEL); 6037 if (!ret_stack) 6038 return; 6039 graph_init_task(t, ret_stack); 6040 } 6041 } 6042 6043 void ftrace_graph_exit_task(struct task_struct *t) 6044 { 6045 struct ftrace_ret_stack *ret_stack = t->ret_stack; 6046 6047 t->ret_stack = NULL; 6048 /* NULL must become visible to IRQs before we free it: */ 6049 barrier(); 6050 6051 kfree(ret_stack); 6052 } 6053 #endif 6054