1 // SPDX-License-Identifier: GPL-2.0
2 /*
3 * Infrastructure for profiling code inserted by 'gcc -pg'.
4 *
5 * Copyright (C) 2007-2008 Steven Rostedt <srostedt@redhat.com>
6 * Copyright (C) 2004-2008 Ingo Molnar <mingo@redhat.com>
7 *
8 * Originally ported from the -rt patch by:
9 * Copyright (C) 2007 Arnaldo Carvalho de Melo <acme@redhat.com>
10 *
11 * Based on code in the latency_tracer, that is:
12 *
13 * Copyright (C) 2004-2006 Ingo Molnar
14 * Copyright (C) 2004 Nadia Yvette Chambers
15 */
16
17 #include <linux/stop_machine.h>
18 #include <linux/clocksource.h>
19 #include <linux/sched/task.h>
20 #include <linux/kallsyms.h>
21 #include <linux/security.h>
22 #include <linux/seq_file.h>
23 #include <linux/tracefs.h>
24 #include <linux/hardirq.h>
25 #include <linux/kthread.h>
26 #include <linux/uaccess.h>
27 #include <linux/bsearch.h>
28 #include <linux/module.h>
29 #include <linux/ftrace.h>
30 #include <linux/sysctl.h>
31 #include <linux/slab.h>
32 #include <linux/ctype.h>
33 #include <linux/sort.h>
34 #include <linux/list.h>
35 #include <linux/hash.h>
36 #include <linux/rcupdate.h>
37 #include <linux/kprobes.h>
38
39 #include <trace/events/sched.h>
40
41 #include <asm/sections.h>
42 #include <asm/setup.h>
43
44 #include "ftrace_internal.h"
45 #include "trace_output.h"
46 #include "trace_stat.h"
47
48 #define FTRACE_WARN_ON(cond) \
49 ({ \
50 int ___r = cond; \
51 if (WARN_ON(___r)) \
52 ftrace_kill(); \
53 ___r; \
54 })
55
56 #define FTRACE_WARN_ON_ONCE(cond) \
57 ({ \
58 int ___r = cond; \
59 if (WARN_ON_ONCE(___r)) \
60 ftrace_kill(); \
61 ___r; \
62 })
63
64 /* hash bits for specific function selection */
65 #define FTRACE_HASH_DEFAULT_BITS 10
66 #define FTRACE_HASH_MAX_BITS 12
67
68 #ifdef CONFIG_DYNAMIC_FTRACE
69 #define INIT_OPS_HASH(opsname) \
70 .func_hash = &opsname.local_hash, \
71 .local_hash.regex_lock = __MUTEX_INITIALIZER(opsname.local_hash.regex_lock),
72 #else
73 #define INIT_OPS_HASH(opsname)
74 #endif
75
76 enum {
77 FTRACE_MODIFY_ENABLE_FL = (1 << 0),
78 FTRACE_MODIFY_MAY_SLEEP_FL = (1 << 1),
79 };
80
81 struct ftrace_ops ftrace_list_end __read_mostly = {
82 .func = ftrace_stub,
83 .flags = FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_STUB,
84 INIT_OPS_HASH(ftrace_list_end)
85 };
86
87 /* ftrace_enabled is a method to turn ftrace on or off */
88 int ftrace_enabled __read_mostly;
89 static int last_ftrace_enabled;
90
91 /* Current function tracing op */
92 struct ftrace_ops *function_trace_op __read_mostly = &ftrace_list_end;
93 /* What to set function_trace_op to */
94 static struct ftrace_ops *set_function_trace_op;
95
ftrace_pids_enabled(struct ftrace_ops * ops)96 static bool ftrace_pids_enabled(struct ftrace_ops *ops)
97 {
98 struct trace_array *tr;
99
100 if (!(ops->flags & FTRACE_OPS_FL_PID) || !ops->private)
101 return false;
102
103 tr = ops->private;
104
105 return tr->function_pids != NULL || tr->function_no_pids != NULL;
106 }
107
108 static void ftrace_update_trampoline(struct ftrace_ops *ops);
109
110 /*
111 * ftrace_disabled is set when an anomaly is discovered.
112 * ftrace_disabled is much stronger than ftrace_enabled.
113 */
114 static int ftrace_disabled __read_mostly;
115
116 DEFINE_MUTEX(ftrace_lock);
117
118 struct ftrace_ops __rcu *ftrace_ops_list __read_mostly = &ftrace_list_end;
119 ftrace_func_t ftrace_trace_function __read_mostly = ftrace_stub;
120 struct ftrace_ops global_ops;
121
122 #if ARCH_SUPPORTS_FTRACE_OPS
123 static void ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
124 struct ftrace_ops *op, struct pt_regs *regs);
125 #else
126 /* See comment below, where ftrace_ops_list_func is defined */
127 static void ftrace_ops_no_ops(unsigned long ip, unsigned long parent_ip);
128 #define ftrace_ops_list_func ((ftrace_func_t)ftrace_ops_no_ops)
129 #endif
130
ftrace_ops_init(struct ftrace_ops * ops)131 static inline void ftrace_ops_init(struct ftrace_ops *ops)
132 {
133 #ifdef CONFIG_DYNAMIC_FTRACE
134 if (!(ops->flags & FTRACE_OPS_FL_INITIALIZED)) {
135 mutex_init(&ops->local_hash.regex_lock);
136 ops->func_hash = &ops->local_hash;
137 ops->flags |= FTRACE_OPS_FL_INITIALIZED;
138 }
139 #endif
140 }
141
ftrace_pid_func(unsigned long ip,unsigned long parent_ip,struct ftrace_ops * op,struct pt_regs * regs)142 static void ftrace_pid_func(unsigned long ip, unsigned long parent_ip,
143 struct ftrace_ops *op, struct pt_regs *regs)
144 {
145 struct trace_array *tr = op->private;
146 int pid;
147
148 if (tr) {
149 pid = this_cpu_read(tr->array_buffer.data->ftrace_ignore_pid);
150 if (pid == FTRACE_PID_IGNORE)
151 return;
152 if (pid != FTRACE_PID_TRACE &&
153 pid != current->pid)
154 return;
155 }
156
157 op->saved_func(ip, parent_ip, op, regs);
158 }
159
ftrace_sync_ipi(void * data)160 static void ftrace_sync_ipi(void *data)
161 {
162 /* Probably not needed, but do it anyway */
163 smp_rmb();
164 }
165
ftrace_ops_get_list_func(struct ftrace_ops * ops)166 static ftrace_func_t ftrace_ops_get_list_func(struct ftrace_ops *ops)
167 {
168 /*
169 * If this is a dynamic, RCU, or per CPU ops, or we force list func,
170 * then it needs to call the list anyway.
171 */
172 if (ops->flags & (FTRACE_OPS_FL_DYNAMIC | FTRACE_OPS_FL_RCU) ||
173 FTRACE_FORCE_LIST_FUNC)
174 return ftrace_ops_list_func;
175
176 return ftrace_ops_get_func(ops);
177 }
178
update_ftrace_function(void)179 static void update_ftrace_function(void)
180 {
181 ftrace_func_t func;
182
183 /*
184 * Prepare the ftrace_ops that the arch callback will use.
185 * If there's only one ftrace_ops registered, the ftrace_ops_list
186 * will point to the ops we want.
187 */
188 set_function_trace_op = rcu_dereference_protected(ftrace_ops_list,
189 lockdep_is_held(&ftrace_lock));
190
191 /* If there's no ftrace_ops registered, just call the stub function */
192 if (set_function_trace_op == &ftrace_list_end) {
193 func = ftrace_stub;
194
195 /*
196 * If we are at the end of the list and this ops is
197 * recursion safe and not dynamic and the arch supports passing ops,
198 * then have the mcount trampoline call the function directly.
199 */
200 } else if (rcu_dereference_protected(ftrace_ops_list->next,
201 lockdep_is_held(&ftrace_lock)) == &ftrace_list_end) {
202 func = ftrace_ops_get_list_func(ftrace_ops_list);
203
204 } else {
205 /* Just use the default ftrace_ops */
206 set_function_trace_op = &ftrace_list_end;
207 func = ftrace_ops_list_func;
208 }
209
210 update_function_graph_func();
211
212 /* If there's no change, then do nothing more here */
213 if (ftrace_trace_function == func)
214 return;
215
216 /*
217 * If we are using the list function, it doesn't care
218 * about the function_trace_ops.
219 */
220 if (func == ftrace_ops_list_func) {
221 ftrace_trace_function = func;
222 /*
223 * Don't even bother setting function_trace_ops,
224 * it would be racy to do so anyway.
225 */
226 return;
227 }
228
229 #ifndef CONFIG_DYNAMIC_FTRACE
230 /*
231 * For static tracing, we need to be a bit more careful.
232 * The function change takes affect immediately. Thus,
233 * we need to coordinate the setting of the function_trace_ops
234 * with the setting of the ftrace_trace_function.
235 *
236 * Set the function to the list ops, which will call the
237 * function we want, albeit indirectly, but it handles the
238 * ftrace_ops and doesn't depend on function_trace_op.
239 */
240 ftrace_trace_function = ftrace_ops_list_func;
241 /*
242 * Make sure all CPUs see this. Yes this is slow, but static
243 * tracing is slow and nasty to have enabled.
244 */
245 synchronize_rcu_tasks_rude();
246 /* Now all cpus are using the list ops. */
247 function_trace_op = set_function_trace_op;
248 /* Make sure the function_trace_op is visible on all CPUs */
249 smp_wmb();
250 /* Nasty way to force a rmb on all cpus */
251 smp_call_function(ftrace_sync_ipi, NULL, 1);
252 /* OK, we are all set to update the ftrace_trace_function now! */
253 #endif /* !CONFIG_DYNAMIC_FTRACE */
254
255 ftrace_trace_function = func;
256 }
257
add_ftrace_ops(struct ftrace_ops __rcu ** list,struct ftrace_ops * ops)258 static void add_ftrace_ops(struct ftrace_ops __rcu **list,
259 struct ftrace_ops *ops)
260 {
261 rcu_assign_pointer(ops->next, *list);
262
263 /*
264 * We are entering ops into the list but another
265 * CPU might be walking that list. We need to make sure
266 * the ops->next pointer is valid before another CPU sees
267 * the ops pointer included into the list.
268 */
269 rcu_assign_pointer(*list, ops);
270 }
271
remove_ftrace_ops(struct ftrace_ops __rcu ** list,struct ftrace_ops * ops)272 static int remove_ftrace_ops(struct ftrace_ops __rcu **list,
273 struct ftrace_ops *ops)
274 {
275 struct ftrace_ops **p;
276
277 /*
278 * If we are removing the last function, then simply point
279 * to the ftrace_stub.
280 */
281 if (rcu_dereference_protected(*list,
282 lockdep_is_held(&ftrace_lock)) == ops &&
283 rcu_dereference_protected(ops->next,
284 lockdep_is_held(&ftrace_lock)) == &ftrace_list_end) {
285 *list = &ftrace_list_end;
286 return 0;
287 }
288
289 for (p = list; *p != &ftrace_list_end; p = &(*p)->next)
290 if (*p == ops)
291 break;
292
293 if (*p != ops)
294 return -1;
295
296 *p = (*p)->next;
297 return 0;
298 }
299
300 static void ftrace_update_trampoline(struct ftrace_ops *ops);
301
__register_ftrace_function(struct ftrace_ops * ops)302 int __register_ftrace_function(struct ftrace_ops *ops)
303 {
304 if (ops->flags & FTRACE_OPS_FL_DELETED)
305 return -EINVAL;
306
307 if (WARN_ON(ops->flags & FTRACE_OPS_FL_ENABLED))
308 return -EBUSY;
309
310 #ifndef CONFIG_DYNAMIC_FTRACE_WITH_REGS
311 /*
312 * If the ftrace_ops specifies SAVE_REGS, then it only can be used
313 * if the arch supports it, or SAVE_REGS_IF_SUPPORTED is also set.
314 * Setting SAVE_REGS_IF_SUPPORTED makes SAVE_REGS irrelevant.
315 */
316 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS &&
317 !(ops->flags & FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED))
318 return -EINVAL;
319
320 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED)
321 ops->flags |= FTRACE_OPS_FL_SAVE_REGS;
322 #endif
323 if (!ftrace_enabled && (ops->flags & FTRACE_OPS_FL_PERMANENT))
324 return -EBUSY;
325
326 if (!core_kernel_data((unsigned long)ops))
327 ops->flags |= FTRACE_OPS_FL_DYNAMIC;
328
329 add_ftrace_ops(&ftrace_ops_list, ops);
330
331 /* Always save the function, and reset at unregistering */
332 ops->saved_func = ops->func;
333
334 if (ftrace_pids_enabled(ops))
335 ops->func = ftrace_pid_func;
336
337 ftrace_update_trampoline(ops);
338
339 if (ftrace_enabled)
340 update_ftrace_function();
341
342 return 0;
343 }
344
__unregister_ftrace_function(struct ftrace_ops * ops)345 int __unregister_ftrace_function(struct ftrace_ops *ops)
346 {
347 int ret;
348
349 if (WARN_ON(!(ops->flags & FTRACE_OPS_FL_ENABLED)))
350 return -EBUSY;
351
352 ret = remove_ftrace_ops(&ftrace_ops_list, ops);
353
354 if (ret < 0)
355 return ret;
356
357 if (ftrace_enabled)
358 update_ftrace_function();
359
360 ops->func = ops->saved_func;
361
362 return 0;
363 }
364
ftrace_update_pid_func(void)365 static void ftrace_update_pid_func(void)
366 {
367 struct ftrace_ops *op;
368
369 /* Only do something if we are tracing something */
370 if (ftrace_trace_function == ftrace_stub)
371 return;
372
373 do_for_each_ftrace_op(op, ftrace_ops_list) {
374 if (op->flags & FTRACE_OPS_FL_PID) {
375 op->func = ftrace_pids_enabled(op) ?
376 ftrace_pid_func : op->saved_func;
377 ftrace_update_trampoline(op);
378 }
379 } while_for_each_ftrace_op(op);
380
381 update_ftrace_function();
382 }
383
384 #ifdef CONFIG_FUNCTION_PROFILER
385 struct ftrace_profile {
386 struct hlist_node node;
387 unsigned long ip;
388 unsigned long counter;
389 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
390 unsigned long long time;
391 unsigned long long time_squared;
392 #endif
393 };
394
395 struct ftrace_profile_page {
396 struct ftrace_profile_page *next;
397 unsigned long index;
398 struct ftrace_profile records[];
399 };
400
401 struct ftrace_profile_stat {
402 atomic_t disabled;
403 struct hlist_head *hash;
404 struct ftrace_profile_page *pages;
405 struct ftrace_profile_page *start;
406 struct tracer_stat stat;
407 };
408
409 #define PROFILE_RECORDS_SIZE \
410 (PAGE_SIZE - offsetof(struct ftrace_profile_page, records))
411
412 #define PROFILES_PER_PAGE \
413 (PROFILE_RECORDS_SIZE / sizeof(struct ftrace_profile))
414
415 static int ftrace_profile_enabled __read_mostly;
416
417 /* ftrace_profile_lock - synchronize the enable and disable of the profiler */
418 static DEFINE_MUTEX(ftrace_profile_lock);
419
420 static DEFINE_PER_CPU(struct ftrace_profile_stat, ftrace_profile_stats);
421
422 #define FTRACE_PROFILE_HASH_BITS 10
423 #define FTRACE_PROFILE_HASH_SIZE (1 << FTRACE_PROFILE_HASH_BITS)
424
425 static void *
function_stat_next(void * v,int idx)426 function_stat_next(void *v, int idx)
427 {
428 struct ftrace_profile *rec = v;
429 struct ftrace_profile_page *pg;
430
431 pg = (struct ftrace_profile_page *)((unsigned long)rec & PAGE_MASK);
432
433 again:
434 if (idx != 0)
435 rec++;
436
437 if ((void *)rec >= (void *)&pg->records[pg->index]) {
438 pg = pg->next;
439 if (!pg)
440 return NULL;
441 rec = &pg->records[0];
442 if (!rec->counter)
443 goto again;
444 }
445
446 return rec;
447 }
448
function_stat_start(struct tracer_stat * trace)449 static void *function_stat_start(struct tracer_stat *trace)
450 {
451 struct ftrace_profile_stat *stat =
452 container_of(trace, struct ftrace_profile_stat, stat);
453
454 if (!stat || !stat->start)
455 return NULL;
456
457 return function_stat_next(&stat->start->records[0], 0);
458 }
459
460 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
461 /* function graph compares on total time */
function_stat_cmp(const void * p1,const void * p2)462 static int function_stat_cmp(const void *p1, const void *p2)
463 {
464 const struct ftrace_profile *a = p1;
465 const struct ftrace_profile *b = p2;
466
467 if (a->time < b->time)
468 return -1;
469 if (a->time > b->time)
470 return 1;
471 else
472 return 0;
473 }
474 #else
475 /* not function graph compares against hits */
function_stat_cmp(const void * p1,const void * p2)476 static int function_stat_cmp(const void *p1, const void *p2)
477 {
478 const struct ftrace_profile *a = p1;
479 const struct ftrace_profile *b = p2;
480
481 if (a->counter < b->counter)
482 return -1;
483 if (a->counter > b->counter)
484 return 1;
485 else
486 return 0;
487 }
488 #endif
489
function_stat_headers(struct seq_file * m)490 static int function_stat_headers(struct seq_file *m)
491 {
492 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
493 seq_puts(m, " Function "
494 "Hit Time Avg s^2\n"
495 " -------- "
496 "--- ---- --- ---\n");
497 #else
498 seq_puts(m, " Function Hit\n"
499 " -------- ---\n");
500 #endif
501 return 0;
502 }
503
function_stat_show(struct seq_file * m,void * v)504 static int function_stat_show(struct seq_file *m, void *v)
505 {
506 struct ftrace_profile *rec = v;
507 char str[KSYM_SYMBOL_LEN];
508 int ret = 0;
509 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
510 static struct trace_seq s;
511 unsigned long long avg;
512 unsigned long long stddev;
513 #endif
514 mutex_lock(&ftrace_profile_lock);
515
516 /* we raced with function_profile_reset() */
517 if (unlikely(rec->counter == 0)) {
518 ret = -EBUSY;
519 goto out;
520 }
521
522 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
523 avg = div64_ul(rec->time, rec->counter);
524 if (tracing_thresh && (avg < tracing_thresh))
525 goto out;
526 #endif
527
528 kallsyms_lookup(rec->ip, NULL, NULL, NULL, str);
529 seq_printf(m, " %-30.30s %10lu", str, rec->counter);
530
531 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
532 seq_puts(m, " ");
533
534 /* Sample standard deviation (s^2) */
535 if (rec->counter <= 1)
536 stddev = 0;
537 else {
538 /*
539 * Apply Welford's method:
540 * s^2 = 1 / (n * (n-1)) * (n * \Sum (x_i)^2 - (\Sum x_i)^2)
541 */
542 stddev = rec->counter * rec->time_squared -
543 rec->time * rec->time;
544
545 /*
546 * Divide only 1000 for ns^2 -> us^2 conversion.
547 * trace_print_graph_duration will divide 1000 again.
548 */
549 stddev = div64_ul(stddev,
550 rec->counter * (rec->counter - 1) * 1000);
551 }
552
553 trace_seq_init(&s);
554 trace_print_graph_duration(rec->time, &s);
555 trace_seq_puts(&s, " ");
556 trace_print_graph_duration(avg, &s);
557 trace_seq_puts(&s, " ");
558 trace_print_graph_duration(stddev, &s);
559 trace_print_seq(m, &s);
560 #endif
561 seq_putc(m, '\n');
562 out:
563 mutex_unlock(&ftrace_profile_lock);
564
565 return ret;
566 }
567
ftrace_profile_reset(struct ftrace_profile_stat * stat)568 static void ftrace_profile_reset(struct ftrace_profile_stat *stat)
569 {
570 struct ftrace_profile_page *pg;
571
572 pg = stat->pages = stat->start;
573
574 while (pg) {
575 memset(pg->records, 0, PROFILE_RECORDS_SIZE);
576 pg->index = 0;
577 pg = pg->next;
578 }
579
580 memset(stat->hash, 0,
581 FTRACE_PROFILE_HASH_SIZE * sizeof(struct hlist_head));
582 }
583
ftrace_profile_pages_init(struct ftrace_profile_stat * stat)584 int ftrace_profile_pages_init(struct ftrace_profile_stat *stat)
585 {
586 struct ftrace_profile_page *pg;
587 int functions;
588 int pages;
589 int i;
590
591 /* If we already allocated, do nothing */
592 if (stat->pages)
593 return 0;
594
595 stat->pages = (void *)get_zeroed_page(GFP_KERNEL);
596 if (!stat->pages)
597 return -ENOMEM;
598
599 #ifdef CONFIG_DYNAMIC_FTRACE
600 functions = ftrace_update_tot_cnt;
601 #else
602 /*
603 * We do not know the number of functions that exist because
604 * dynamic tracing is what counts them. With past experience
605 * we have around 20K functions. That should be more than enough.
606 * It is highly unlikely we will execute every function in
607 * the kernel.
608 */
609 functions = 20000;
610 #endif
611
612 pg = stat->start = stat->pages;
613
614 pages = DIV_ROUND_UP(functions, PROFILES_PER_PAGE);
615
616 for (i = 1; i < pages; i++) {
617 pg->next = (void *)get_zeroed_page(GFP_KERNEL);
618 if (!pg->next)
619 goto out_free;
620 pg = pg->next;
621 }
622
623 return 0;
624
625 out_free:
626 pg = stat->start;
627 while (pg) {
628 unsigned long tmp = (unsigned long)pg;
629
630 pg = pg->next;
631 free_page(tmp);
632 }
633
634 stat->pages = NULL;
635 stat->start = NULL;
636
637 return -ENOMEM;
638 }
639
ftrace_profile_init_cpu(int cpu)640 static int ftrace_profile_init_cpu(int cpu)
641 {
642 struct ftrace_profile_stat *stat;
643 int size;
644
645 stat = &per_cpu(ftrace_profile_stats, cpu);
646
647 if (stat->hash) {
648 /* If the profile is already created, simply reset it */
649 ftrace_profile_reset(stat);
650 return 0;
651 }
652
653 /*
654 * We are profiling all functions, but usually only a few thousand
655 * functions are hit. We'll make a hash of 1024 items.
656 */
657 size = FTRACE_PROFILE_HASH_SIZE;
658
659 stat->hash = kcalloc(size, sizeof(struct hlist_head), GFP_KERNEL);
660
661 if (!stat->hash)
662 return -ENOMEM;
663
664 /* Preallocate the function profiling pages */
665 if (ftrace_profile_pages_init(stat) < 0) {
666 kfree(stat->hash);
667 stat->hash = NULL;
668 return -ENOMEM;
669 }
670
671 return 0;
672 }
673
ftrace_profile_init(void)674 static int ftrace_profile_init(void)
675 {
676 int cpu;
677 int ret = 0;
678
679 for_each_possible_cpu(cpu) {
680 ret = ftrace_profile_init_cpu(cpu);
681 if (ret)
682 break;
683 }
684
685 return ret;
686 }
687
688 /* interrupts must be disabled */
689 static struct ftrace_profile *
ftrace_find_profiled_func(struct ftrace_profile_stat * stat,unsigned long ip)690 ftrace_find_profiled_func(struct ftrace_profile_stat *stat, unsigned long ip)
691 {
692 struct ftrace_profile *rec;
693 struct hlist_head *hhd;
694 unsigned long key;
695
696 key = hash_long(ip, FTRACE_PROFILE_HASH_BITS);
697 hhd = &stat->hash[key];
698
699 if (hlist_empty(hhd))
700 return NULL;
701
702 hlist_for_each_entry_rcu_notrace(rec, hhd, node) {
703 if (rec->ip == ip)
704 return rec;
705 }
706
707 return NULL;
708 }
709
ftrace_add_profile(struct ftrace_profile_stat * stat,struct ftrace_profile * rec)710 static void ftrace_add_profile(struct ftrace_profile_stat *stat,
711 struct ftrace_profile *rec)
712 {
713 unsigned long key;
714
715 key = hash_long(rec->ip, FTRACE_PROFILE_HASH_BITS);
716 hlist_add_head_rcu(&rec->node, &stat->hash[key]);
717 }
718
719 /*
720 * The memory is already allocated, this simply finds a new record to use.
721 */
722 static struct ftrace_profile *
ftrace_profile_alloc(struct ftrace_profile_stat * stat,unsigned long ip)723 ftrace_profile_alloc(struct ftrace_profile_stat *stat, unsigned long ip)
724 {
725 struct ftrace_profile *rec = NULL;
726
727 /* prevent recursion (from NMIs) */
728 if (atomic_inc_return(&stat->disabled) != 1)
729 goto out;
730
731 /*
732 * Try to find the function again since an NMI
733 * could have added it
734 */
735 rec = ftrace_find_profiled_func(stat, ip);
736 if (rec)
737 goto out;
738
739 if (stat->pages->index == PROFILES_PER_PAGE) {
740 if (!stat->pages->next)
741 goto out;
742 stat->pages = stat->pages->next;
743 }
744
745 rec = &stat->pages->records[stat->pages->index++];
746 rec->ip = ip;
747 ftrace_add_profile(stat, rec);
748
749 out:
750 atomic_dec(&stat->disabled);
751
752 return rec;
753 }
754
755 static void
function_profile_call(unsigned long ip,unsigned long parent_ip,struct ftrace_ops * ops,struct pt_regs * regs)756 function_profile_call(unsigned long ip, unsigned long parent_ip,
757 struct ftrace_ops *ops, struct pt_regs *regs)
758 {
759 struct ftrace_profile_stat *stat;
760 struct ftrace_profile *rec;
761 unsigned long flags;
762
763 if (!ftrace_profile_enabled)
764 return;
765
766 local_irq_save(flags);
767
768 stat = this_cpu_ptr(&ftrace_profile_stats);
769 if (!stat->hash || !ftrace_profile_enabled)
770 goto out;
771
772 rec = ftrace_find_profiled_func(stat, ip);
773 if (!rec) {
774 rec = ftrace_profile_alloc(stat, ip);
775 if (!rec)
776 goto out;
777 }
778
779 rec->counter++;
780 out:
781 local_irq_restore(flags);
782 }
783
784 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
785 static bool fgraph_graph_time = true;
786
ftrace_graph_graph_time_control(bool enable)787 void ftrace_graph_graph_time_control(bool enable)
788 {
789 fgraph_graph_time = enable;
790 }
791
profile_graph_entry(struct ftrace_graph_ent * trace)792 static int profile_graph_entry(struct ftrace_graph_ent *trace)
793 {
794 struct ftrace_ret_stack *ret_stack;
795
796 function_profile_call(trace->func, 0, NULL, NULL);
797
798 /* If function graph is shutting down, ret_stack can be NULL */
799 if (!current->ret_stack)
800 return 0;
801
802 ret_stack = ftrace_graph_get_ret_stack(current, 0);
803 if (ret_stack)
804 ret_stack->subtime = 0;
805
806 return 1;
807 }
808
profile_graph_return(struct ftrace_graph_ret * trace)809 static void profile_graph_return(struct ftrace_graph_ret *trace)
810 {
811 struct ftrace_ret_stack *ret_stack;
812 struct ftrace_profile_stat *stat;
813 unsigned long long calltime;
814 struct ftrace_profile *rec;
815 unsigned long flags;
816
817 local_irq_save(flags);
818 stat = this_cpu_ptr(&ftrace_profile_stats);
819 if (!stat->hash || !ftrace_profile_enabled)
820 goto out;
821
822 /* If the calltime was zero'd ignore it */
823 if (!trace->calltime)
824 goto out;
825
826 calltime = trace->rettime - trace->calltime;
827
828 if (!fgraph_graph_time) {
829
830 /* Append this call time to the parent time to subtract */
831 ret_stack = ftrace_graph_get_ret_stack(current, 1);
832 if (ret_stack)
833 ret_stack->subtime += calltime;
834
835 ret_stack = ftrace_graph_get_ret_stack(current, 0);
836 if (ret_stack && ret_stack->subtime < calltime)
837 calltime -= ret_stack->subtime;
838 else
839 calltime = 0;
840 }
841
842 rec = ftrace_find_profiled_func(stat, trace->func);
843 if (rec) {
844 rec->time += calltime;
845 rec->time_squared += calltime * calltime;
846 }
847
848 out:
849 local_irq_restore(flags);
850 }
851
852 static struct fgraph_ops fprofiler_ops = {
853 .entryfunc = &profile_graph_entry,
854 .retfunc = &profile_graph_return,
855 };
856
register_ftrace_profiler(void)857 static int register_ftrace_profiler(void)
858 {
859 return register_ftrace_graph(&fprofiler_ops);
860 }
861
unregister_ftrace_profiler(void)862 static void unregister_ftrace_profiler(void)
863 {
864 unregister_ftrace_graph(&fprofiler_ops);
865 }
866 #else
867 static struct ftrace_ops ftrace_profile_ops __read_mostly = {
868 .func = function_profile_call,
869 .flags = FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_INITIALIZED,
870 INIT_OPS_HASH(ftrace_profile_ops)
871 };
872
register_ftrace_profiler(void)873 static int register_ftrace_profiler(void)
874 {
875 return register_ftrace_function(&ftrace_profile_ops);
876 }
877
unregister_ftrace_profiler(void)878 static void unregister_ftrace_profiler(void)
879 {
880 unregister_ftrace_function(&ftrace_profile_ops);
881 }
882 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
883
884 static ssize_t
ftrace_profile_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos)885 ftrace_profile_write(struct file *filp, const char __user *ubuf,
886 size_t cnt, loff_t *ppos)
887 {
888 unsigned long val;
889 int ret;
890
891 ret = kstrtoul_from_user(ubuf, cnt, 10, &val);
892 if (ret)
893 return ret;
894
895 val = !!val;
896
897 mutex_lock(&ftrace_profile_lock);
898 if (ftrace_profile_enabled ^ val) {
899 if (val) {
900 ret = ftrace_profile_init();
901 if (ret < 0) {
902 cnt = ret;
903 goto out;
904 }
905
906 ret = register_ftrace_profiler();
907 if (ret < 0) {
908 cnt = ret;
909 goto out;
910 }
911 ftrace_profile_enabled = 1;
912 } else {
913 ftrace_profile_enabled = 0;
914 /*
915 * unregister_ftrace_profiler calls stop_machine
916 * so this acts like an synchronize_rcu.
917 */
918 unregister_ftrace_profiler();
919 }
920 }
921 out:
922 mutex_unlock(&ftrace_profile_lock);
923
924 *ppos += cnt;
925
926 return cnt;
927 }
928
929 static ssize_t
ftrace_profile_read(struct file * filp,char __user * ubuf,size_t cnt,loff_t * ppos)930 ftrace_profile_read(struct file *filp, char __user *ubuf,
931 size_t cnt, loff_t *ppos)
932 {
933 char buf[64]; /* big enough to hold a number */
934 int r;
935
936 r = sprintf(buf, "%u\n", ftrace_profile_enabled);
937 return simple_read_from_buffer(ubuf, cnt, ppos, buf, r);
938 }
939
940 static const struct file_operations ftrace_profile_fops = {
941 .open = tracing_open_generic,
942 .read = ftrace_profile_read,
943 .write = ftrace_profile_write,
944 .llseek = default_llseek,
945 };
946
947 /* used to initialize the real stat files */
948 static struct tracer_stat function_stats __initdata = {
949 .name = "functions",
950 .stat_start = function_stat_start,
951 .stat_next = function_stat_next,
952 .stat_cmp = function_stat_cmp,
953 .stat_headers = function_stat_headers,
954 .stat_show = function_stat_show
955 };
956
ftrace_profile_tracefs(struct dentry * d_tracer)957 static __init void ftrace_profile_tracefs(struct dentry *d_tracer)
958 {
959 struct ftrace_profile_stat *stat;
960 struct dentry *entry;
961 char *name;
962 int ret;
963 int cpu;
964
965 for_each_possible_cpu(cpu) {
966 stat = &per_cpu(ftrace_profile_stats, cpu);
967
968 name = kasprintf(GFP_KERNEL, "function%d", cpu);
969 if (!name) {
970 /*
971 * The files created are permanent, if something happens
972 * we still do not free memory.
973 */
974 WARN(1,
975 "Could not allocate stat file for cpu %d\n",
976 cpu);
977 return;
978 }
979 stat->stat = function_stats;
980 stat->stat.name = name;
981 ret = register_stat_tracer(&stat->stat);
982 if (ret) {
983 WARN(1,
984 "Could not register function stat for cpu %d\n",
985 cpu);
986 kfree(name);
987 return;
988 }
989 }
990
991 entry = tracefs_create_file("function_profile_enabled", 0644,
992 d_tracer, NULL, &ftrace_profile_fops);
993 if (!entry)
994 pr_warn("Could not create tracefs 'function_profile_enabled' entry\n");
995 }
996
997 #else /* CONFIG_FUNCTION_PROFILER */
ftrace_profile_tracefs(struct dentry * d_tracer)998 static __init void ftrace_profile_tracefs(struct dentry *d_tracer)
999 {
1000 }
1001 #endif /* CONFIG_FUNCTION_PROFILER */
1002
1003 #ifdef CONFIG_DYNAMIC_FTRACE
1004
1005 static struct ftrace_ops *removed_ops;
1006
1007 /*
1008 * Set when doing a global update, like enabling all recs or disabling them.
1009 * It is not set when just updating a single ftrace_ops.
1010 */
1011 static bool update_all_ops;
1012
1013 #ifndef CONFIG_FTRACE_MCOUNT_RECORD
1014 # error Dynamic ftrace depends on MCOUNT_RECORD
1015 #endif
1016
1017 struct ftrace_func_probe {
1018 struct ftrace_probe_ops *probe_ops;
1019 struct ftrace_ops ops;
1020 struct trace_array *tr;
1021 struct list_head list;
1022 void *data;
1023 int ref;
1024 };
1025
1026 /*
1027 * We make these constant because no one should touch them,
1028 * but they are used as the default "empty hash", to avoid allocating
1029 * it all the time. These are in a read only section such that if
1030 * anyone does try to modify it, it will cause an exception.
1031 */
1032 static const struct hlist_head empty_buckets[1];
1033 static const struct ftrace_hash empty_hash = {
1034 .buckets = (struct hlist_head *)empty_buckets,
1035 };
1036 #define EMPTY_HASH ((struct ftrace_hash *)&empty_hash)
1037
1038 struct ftrace_ops global_ops = {
1039 .func = ftrace_stub,
1040 .local_hash.notrace_hash = EMPTY_HASH,
1041 .local_hash.filter_hash = EMPTY_HASH,
1042 INIT_OPS_HASH(global_ops)
1043 .flags = FTRACE_OPS_FL_RECURSION_SAFE |
1044 FTRACE_OPS_FL_INITIALIZED |
1045 FTRACE_OPS_FL_PID,
1046 };
1047
1048 /*
1049 * Used by the stack undwinder to know about dynamic ftrace trampolines.
1050 */
ftrace_ops_trampoline(unsigned long addr)1051 struct ftrace_ops *ftrace_ops_trampoline(unsigned long addr)
1052 {
1053 struct ftrace_ops *op = NULL;
1054
1055 /*
1056 * Some of the ops may be dynamically allocated,
1057 * they are freed after a synchronize_rcu().
1058 */
1059 preempt_disable_notrace();
1060
1061 do_for_each_ftrace_op(op, ftrace_ops_list) {
1062 /*
1063 * This is to check for dynamically allocated trampolines.
1064 * Trampolines that are in kernel text will have
1065 * core_kernel_text() return true.
1066 */
1067 if (op->trampoline && op->trampoline_size)
1068 if (addr >= op->trampoline &&
1069 addr < op->trampoline + op->trampoline_size) {
1070 preempt_enable_notrace();
1071 return op;
1072 }
1073 } while_for_each_ftrace_op(op);
1074 preempt_enable_notrace();
1075
1076 return NULL;
1077 }
1078
1079 /*
1080 * This is used by __kernel_text_address() to return true if the
1081 * address is on a dynamically allocated trampoline that would
1082 * not return true for either core_kernel_text() or
1083 * is_module_text_address().
1084 */
is_ftrace_trampoline(unsigned long addr)1085 bool is_ftrace_trampoline(unsigned long addr)
1086 {
1087 return ftrace_ops_trampoline(addr) != NULL;
1088 }
1089
1090 struct ftrace_page {
1091 struct ftrace_page *next;
1092 struct dyn_ftrace *records;
1093 int index;
1094 int order;
1095 };
1096
1097 #define ENTRY_SIZE sizeof(struct dyn_ftrace)
1098 #define ENTRIES_PER_PAGE (PAGE_SIZE / ENTRY_SIZE)
1099
1100 static struct ftrace_page *ftrace_pages_start;
1101 static struct ftrace_page *ftrace_pages;
1102
1103 static __always_inline unsigned long
ftrace_hash_key(struct ftrace_hash * hash,unsigned long ip)1104 ftrace_hash_key(struct ftrace_hash *hash, unsigned long ip)
1105 {
1106 if (hash->size_bits > 0)
1107 return hash_long(ip, hash->size_bits);
1108
1109 return 0;
1110 }
1111
1112 /* Only use this function if ftrace_hash_empty() has already been tested */
1113 static __always_inline struct ftrace_func_entry *
__ftrace_lookup_ip(struct ftrace_hash * hash,unsigned long ip)1114 __ftrace_lookup_ip(struct ftrace_hash *hash, unsigned long ip)
1115 {
1116 unsigned long key;
1117 struct ftrace_func_entry *entry;
1118 struct hlist_head *hhd;
1119
1120 key = ftrace_hash_key(hash, ip);
1121 hhd = &hash->buckets[key];
1122
1123 hlist_for_each_entry_rcu_notrace(entry, hhd, hlist) {
1124 if (entry->ip == ip)
1125 return entry;
1126 }
1127 return NULL;
1128 }
1129
1130 /**
1131 * ftrace_lookup_ip - Test to see if an ip exists in an ftrace_hash
1132 * @hash: The hash to look at
1133 * @ip: The instruction pointer to test
1134 *
1135 * Search a given @hash to see if a given instruction pointer (@ip)
1136 * exists in it.
1137 *
1138 * Returns the entry that holds the @ip if found. NULL otherwise.
1139 */
1140 struct ftrace_func_entry *
ftrace_lookup_ip(struct ftrace_hash * hash,unsigned long ip)1141 ftrace_lookup_ip(struct ftrace_hash *hash, unsigned long ip)
1142 {
1143 if (ftrace_hash_empty(hash))
1144 return NULL;
1145
1146 return __ftrace_lookup_ip(hash, ip);
1147 }
1148
__add_hash_entry(struct ftrace_hash * hash,struct ftrace_func_entry * entry)1149 static void __add_hash_entry(struct ftrace_hash *hash,
1150 struct ftrace_func_entry *entry)
1151 {
1152 struct hlist_head *hhd;
1153 unsigned long key;
1154
1155 key = ftrace_hash_key(hash, entry->ip);
1156 hhd = &hash->buckets[key];
1157 hlist_add_head(&entry->hlist, hhd);
1158 hash->count++;
1159 }
1160
add_hash_entry(struct ftrace_hash * hash,unsigned long ip)1161 static int add_hash_entry(struct ftrace_hash *hash, unsigned long ip)
1162 {
1163 struct ftrace_func_entry *entry;
1164
1165 entry = kmalloc(sizeof(*entry), GFP_KERNEL);
1166 if (!entry)
1167 return -ENOMEM;
1168
1169 entry->ip = ip;
1170 __add_hash_entry(hash, entry);
1171
1172 return 0;
1173 }
1174
1175 static void
free_hash_entry(struct ftrace_hash * hash,struct ftrace_func_entry * entry)1176 free_hash_entry(struct ftrace_hash *hash,
1177 struct ftrace_func_entry *entry)
1178 {
1179 hlist_del(&entry->hlist);
1180 kfree(entry);
1181 hash->count--;
1182 }
1183
1184 static void
remove_hash_entry(struct ftrace_hash * hash,struct ftrace_func_entry * entry)1185 remove_hash_entry(struct ftrace_hash *hash,
1186 struct ftrace_func_entry *entry)
1187 {
1188 hlist_del_rcu(&entry->hlist);
1189 hash->count--;
1190 }
1191
ftrace_hash_clear(struct ftrace_hash * hash)1192 static void ftrace_hash_clear(struct ftrace_hash *hash)
1193 {
1194 struct hlist_head *hhd;
1195 struct hlist_node *tn;
1196 struct ftrace_func_entry *entry;
1197 int size = 1 << hash->size_bits;
1198 int i;
1199
1200 if (!hash->count)
1201 return;
1202
1203 for (i = 0; i < size; i++) {
1204 hhd = &hash->buckets[i];
1205 hlist_for_each_entry_safe(entry, tn, hhd, hlist)
1206 free_hash_entry(hash, entry);
1207 }
1208 FTRACE_WARN_ON(hash->count);
1209 }
1210
free_ftrace_mod(struct ftrace_mod_load * ftrace_mod)1211 static void free_ftrace_mod(struct ftrace_mod_load *ftrace_mod)
1212 {
1213 list_del(&ftrace_mod->list);
1214 kfree(ftrace_mod->module);
1215 kfree(ftrace_mod->func);
1216 kfree(ftrace_mod);
1217 }
1218
clear_ftrace_mod_list(struct list_head * head)1219 static void clear_ftrace_mod_list(struct list_head *head)
1220 {
1221 struct ftrace_mod_load *p, *n;
1222
1223 /* stack tracer isn't supported yet */
1224 if (!head)
1225 return;
1226
1227 mutex_lock(&ftrace_lock);
1228 list_for_each_entry_safe(p, n, head, list)
1229 free_ftrace_mod(p);
1230 mutex_unlock(&ftrace_lock);
1231 }
1232
free_ftrace_hash(struct ftrace_hash * hash)1233 static void free_ftrace_hash(struct ftrace_hash *hash)
1234 {
1235 if (!hash || hash == EMPTY_HASH)
1236 return;
1237 ftrace_hash_clear(hash);
1238 kfree(hash->buckets);
1239 kfree(hash);
1240 }
1241
__free_ftrace_hash_rcu(struct rcu_head * rcu)1242 static void __free_ftrace_hash_rcu(struct rcu_head *rcu)
1243 {
1244 struct ftrace_hash *hash;
1245
1246 hash = container_of(rcu, struct ftrace_hash, rcu);
1247 free_ftrace_hash(hash);
1248 }
1249
free_ftrace_hash_rcu(struct ftrace_hash * hash)1250 static void free_ftrace_hash_rcu(struct ftrace_hash *hash)
1251 {
1252 if (!hash || hash == EMPTY_HASH)
1253 return;
1254 call_rcu(&hash->rcu, __free_ftrace_hash_rcu);
1255 }
1256
ftrace_free_filter(struct ftrace_ops * ops)1257 void ftrace_free_filter(struct ftrace_ops *ops)
1258 {
1259 ftrace_ops_init(ops);
1260 free_ftrace_hash(ops->func_hash->filter_hash);
1261 free_ftrace_hash(ops->func_hash->notrace_hash);
1262 }
1263
alloc_ftrace_hash(int size_bits)1264 static struct ftrace_hash *alloc_ftrace_hash(int size_bits)
1265 {
1266 struct ftrace_hash *hash;
1267 int size;
1268
1269 hash = kzalloc(sizeof(*hash), GFP_KERNEL);
1270 if (!hash)
1271 return NULL;
1272
1273 size = 1 << size_bits;
1274 hash->buckets = kcalloc(size, sizeof(*hash->buckets), GFP_KERNEL);
1275
1276 if (!hash->buckets) {
1277 kfree(hash);
1278 return NULL;
1279 }
1280
1281 hash->size_bits = size_bits;
1282
1283 return hash;
1284 }
1285
1286
ftrace_add_mod(struct trace_array * tr,const char * func,const char * module,int enable)1287 static int ftrace_add_mod(struct trace_array *tr,
1288 const char *func, const char *module,
1289 int enable)
1290 {
1291 struct ftrace_mod_load *ftrace_mod;
1292 struct list_head *mod_head = enable ? &tr->mod_trace : &tr->mod_notrace;
1293
1294 ftrace_mod = kzalloc(sizeof(*ftrace_mod), GFP_KERNEL);
1295 if (!ftrace_mod)
1296 return -ENOMEM;
1297
1298 INIT_LIST_HEAD(&ftrace_mod->list);
1299 ftrace_mod->func = kstrdup(func, GFP_KERNEL);
1300 ftrace_mod->module = kstrdup(module, GFP_KERNEL);
1301 ftrace_mod->enable = enable;
1302
1303 if (!ftrace_mod->func || !ftrace_mod->module)
1304 goto out_free;
1305
1306 list_add(&ftrace_mod->list, mod_head);
1307
1308 return 0;
1309
1310 out_free:
1311 free_ftrace_mod(ftrace_mod);
1312
1313 return -ENOMEM;
1314 }
1315
1316 static struct ftrace_hash *
alloc_and_copy_ftrace_hash(int size_bits,struct ftrace_hash * hash)1317 alloc_and_copy_ftrace_hash(int size_bits, struct ftrace_hash *hash)
1318 {
1319 struct ftrace_func_entry *entry;
1320 struct ftrace_hash *new_hash;
1321 int size;
1322 int ret;
1323 int i;
1324
1325 new_hash = alloc_ftrace_hash(size_bits);
1326 if (!new_hash)
1327 return NULL;
1328
1329 if (hash)
1330 new_hash->flags = hash->flags;
1331
1332 /* Empty hash? */
1333 if (ftrace_hash_empty(hash))
1334 return new_hash;
1335
1336 size = 1 << hash->size_bits;
1337 for (i = 0; i < size; i++) {
1338 hlist_for_each_entry(entry, &hash->buckets[i], hlist) {
1339 ret = add_hash_entry(new_hash, entry->ip);
1340 if (ret < 0)
1341 goto free_hash;
1342 }
1343 }
1344
1345 FTRACE_WARN_ON(new_hash->count != hash->count);
1346
1347 return new_hash;
1348
1349 free_hash:
1350 free_ftrace_hash(new_hash);
1351 return NULL;
1352 }
1353
1354 static void
1355 ftrace_hash_rec_disable_modify(struct ftrace_ops *ops, int filter_hash);
1356 static void
1357 ftrace_hash_rec_enable_modify(struct ftrace_ops *ops, int filter_hash);
1358
1359 static int ftrace_hash_ipmodify_update(struct ftrace_ops *ops,
1360 struct ftrace_hash *new_hash);
1361
dup_hash(struct ftrace_hash * src,int size)1362 static struct ftrace_hash *dup_hash(struct ftrace_hash *src, int size)
1363 {
1364 struct ftrace_func_entry *entry;
1365 struct ftrace_hash *new_hash;
1366 struct hlist_head *hhd;
1367 struct hlist_node *tn;
1368 int bits = 0;
1369 int i;
1370
1371 /*
1372 * Use around half the size (max bit of it), but
1373 * a minimum of 2 is fine (as size of 0 or 1 both give 1 for bits).
1374 */
1375 bits = fls(size / 2);
1376
1377 /* Don't allocate too much */
1378 if (bits > FTRACE_HASH_MAX_BITS)
1379 bits = FTRACE_HASH_MAX_BITS;
1380
1381 new_hash = alloc_ftrace_hash(bits);
1382 if (!new_hash)
1383 return NULL;
1384
1385 new_hash->flags = src->flags;
1386
1387 size = 1 << src->size_bits;
1388 for (i = 0; i < size; i++) {
1389 hhd = &src->buckets[i];
1390 hlist_for_each_entry_safe(entry, tn, hhd, hlist) {
1391 remove_hash_entry(src, entry);
1392 __add_hash_entry(new_hash, entry);
1393 }
1394 }
1395 return new_hash;
1396 }
1397
1398 static struct ftrace_hash *
__ftrace_hash_move(struct ftrace_hash * src)1399 __ftrace_hash_move(struct ftrace_hash *src)
1400 {
1401 int size = src->count;
1402
1403 /*
1404 * If the new source is empty, just return the empty_hash.
1405 */
1406 if (ftrace_hash_empty(src))
1407 return EMPTY_HASH;
1408
1409 return dup_hash(src, size);
1410 }
1411
1412 static int
ftrace_hash_move(struct ftrace_ops * ops,int enable,struct ftrace_hash ** dst,struct ftrace_hash * src)1413 ftrace_hash_move(struct ftrace_ops *ops, int enable,
1414 struct ftrace_hash **dst, struct ftrace_hash *src)
1415 {
1416 struct ftrace_hash *new_hash;
1417 int ret;
1418
1419 /* Reject setting notrace hash on IPMODIFY ftrace_ops */
1420 if (ops->flags & FTRACE_OPS_FL_IPMODIFY && !enable)
1421 return -EINVAL;
1422
1423 new_hash = __ftrace_hash_move(src);
1424 if (!new_hash)
1425 return -ENOMEM;
1426
1427 /* Make sure this can be applied if it is IPMODIFY ftrace_ops */
1428 if (enable) {
1429 /* IPMODIFY should be updated only when filter_hash updating */
1430 ret = ftrace_hash_ipmodify_update(ops, new_hash);
1431 if (ret < 0) {
1432 free_ftrace_hash(new_hash);
1433 return ret;
1434 }
1435 }
1436
1437 /*
1438 * Remove the current set, update the hash and add
1439 * them back.
1440 */
1441 ftrace_hash_rec_disable_modify(ops, enable);
1442
1443 rcu_assign_pointer(*dst, new_hash);
1444
1445 ftrace_hash_rec_enable_modify(ops, enable);
1446
1447 return 0;
1448 }
1449
hash_contains_ip(unsigned long ip,struct ftrace_ops_hash * hash)1450 static bool hash_contains_ip(unsigned long ip,
1451 struct ftrace_ops_hash *hash)
1452 {
1453 /*
1454 * The function record is a match if it exists in the filter
1455 * hash and not in the notrace hash. Note, an empty hash is
1456 * considered a match for the filter hash, but an empty
1457 * notrace hash is considered not in the notrace hash.
1458 */
1459 return (ftrace_hash_empty(hash->filter_hash) ||
1460 __ftrace_lookup_ip(hash->filter_hash, ip)) &&
1461 (ftrace_hash_empty(hash->notrace_hash) ||
1462 !__ftrace_lookup_ip(hash->notrace_hash, ip));
1463 }
1464
1465 /*
1466 * Test the hashes for this ops to see if we want to call
1467 * the ops->func or not.
1468 *
1469 * It's a match if the ip is in the ops->filter_hash or
1470 * the filter_hash does not exist or is empty,
1471 * AND
1472 * the ip is not in the ops->notrace_hash.
1473 *
1474 * This needs to be called with preemption disabled as
1475 * the hashes are freed with call_rcu().
1476 */
1477 int
ftrace_ops_test(struct ftrace_ops * ops,unsigned long ip,void * regs)1478 ftrace_ops_test(struct ftrace_ops *ops, unsigned long ip, void *regs)
1479 {
1480 struct ftrace_ops_hash hash;
1481 int ret;
1482
1483 #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
1484 /*
1485 * There's a small race when adding ops that the ftrace handler
1486 * that wants regs, may be called without them. We can not
1487 * allow that handler to be called if regs is NULL.
1488 */
1489 if (regs == NULL && (ops->flags & FTRACE_OPS_FL_SAVE_REGS))
1490 return 0;
1491 #endif
1492
1493 rcu_assign_pointer(hash.filter_hash, ops->func_hash->filter_hash);
1494 rcu_assign_pointer(hash.notrace_hash, ops->func_hash->notrace_hash);
1495
1496 if (hash_contains_ip(ip, &hash))
1497 ret = 1;
1498 else
1499 ret = 0;
1500
1501 return ret;
1502 }
1503
1504 /*
1505 * This is a double for. Do not use 'break' to break out of the loop,
1506 * you must use a goto.
1507 */
1508 #define do_for_each_ftrace_rec(pg, rec) \
1509 for (pg = ftrace_pages_start; pg; pg = pg->next) { \
1510 int _____i; \
1511 for (_____i = 0; _____i < pg->index; _____i++) { \
1512 rec = &pg->records[_____i];
1513
1514 #define while_for_each_ftrace_rec() \
1515 } \
1516 }
1517
1518
ftrace_cmp_recs(const void * a,const void * b)1519 static int ftrace_cmp_recs(const void *a, const void *b)
1520 {
1521 const struct dyn_ftrace *key = a;
1522 const struct dyn_ftrace *rec = b;
1523
1524 if (key->flags < rec->ip)
1525 return -1;
1526 if (key->ip >= rec->ip + MCOUNT_INSN_SIZE)
1527 return 1;
1528 return 0;
1529 }
1530
lookup_rec(unsigned long start,unsigned long end)1531 static struct dyn_ftrace *lookup_rec(unsigned long start, unsigned long end)
1532 {
1533 struct ftrace_page *pg;
1534 struct dyn_ftrace *rec = NULL;
1535 struct dyn_ftrace key;
1536
1537 key.ip = start;
1538 key.flags = end; /* overload flags, as it is unsigned long */
1539
1540 for (pg = ftrace_pages_start; pg; pg = pg->next) {
1541 if (pg->index == 0 ||
1542 end < pg->records[0].ip ||
1543 start >= (pg->records[pg->index - 1].ip + MCOUNT_INSN_SIZE))
1544 continue;
1545 rec = bsearch(&key, pg->records, pg->index,
1546 sizeof(struct dyn_ftrace),
1547 ftrace_cmp_recs);
1548 if (rec)
1549 break;
1550 }
1551 return rec;
1552 }
1553
1554 /**
1555 * ftrace_location_range - return the first address of a traced location
1556 * if it touches the given ip range
1557 * @start: start of range to search.
1558 * @end: end of range to search (inclusive). @end points to the last byte
1559 * to check.
1560 *
1561 * Returns rec->ip if the related ftrace location is a least partly within
1562 * the given address range. That is, the first address of the instruction
1563 * that is either a NOP or call to the function tracer. It checks the ftrace
1564 * internal tables to determine if the address belongs or not.
1565 */
ftrace_location_range(unsigned long start,unsigned long end)1566 unsigned long ftrace_location_range(unsigned long start, unsigned long end)
1567 {
1568 struct dyn_ftrace *rec;
1569
1570 rec = lookup_rec(start, end);
1571 if (rec)
1572 return rec->ip;
1573
1574 return 0;
1575 }
1576
1577 /**
1578 * ftrace_location - return true if the ip giving is a traced location
1579 * @ip: the instruction pointer to check
1580 *
1581 * Returns rec->ip if @ip given is a pointer to a ftrace location.
1582 * That is, the instruction that is either a NOP or call to
1583 * the function tracer. It checks the ftrace internal tables to
1584 * determine if the address belongs or not.
1585 */
ftrace_location(unsigned long ip)1586 unsigned long ftrace_location(unsigned long ip)
1587 {
1588 return ftrace_location_range(ip, ip);
1589 }
1590
1591 /**
1592 * ftrace_text_reserved - return true if range contains an ftrace location
1593 * @start: start of range to search
1594 * @end: end of range to search (inclusive). @end points to the last byte to check.
1595 *
1596 * Returns 1 if @start and @end contains a ftrace location.
1597 * That is, the instruction that is either a NOP or call to
1598 * the function tracer. It checks the ftrace internal tables to
1599 * determine if the address belongs or not.
1600 */
ftrace_text_reserved(const void * start,const void * end)1601 int ftrace_text_reserved(const void *start, const void *end)
1602 {
1603 unsigned long ret;
1604
1605 ret = ftrace_location_range((unsigned long)start,
1606 (unsigned long)end);
1607
1608 return (int)!!ret;
1609 }
1610
1611 /* Test if ops registered to this rec needs regs */
test_rec_ops_needs_regs(struct dyn_ftrace * rec)1612 static bool test_rec_ops_needs_regs(struct dyn_ftrace *rec)
1613 {
1614 struct ftrace_ops *ops;
1615 bool keep_regs = false;
1616
1617 for (ops = ftrace_ops_list;
1618 ops != &ftrace_list_end; ops = ops->next) {
1619 /* pass rec in as regs to have non-NULL val */
1620 if (ftrace_ops_test(ops, rec->ip, rec)) {
1621 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS) {
1622 keep_regs = true;
1623 break;
1624 }
1625 }
1626 }
1627
1628 return keep_regs;
1629 }
1630
1631 static struct ftrace_ops *
1632 ftrace_find_tramp_ops_any(struct dyn_ftrace *rec);
1633 static struct ftrace_ops *
1634 ftrace_find_tramp_ops_any_other(struct dyn_ftrace *rec, struct ftrace_ops *op_exclude);
1635 static struct ftrace_ops *
1636 ftrace_find_tramp_ops_next(struct dyn_ftrace *rec, struct ftrace_ops *ops);
1637
__ftrace_hash_rec_update(struct ftrace_ops * ops,int filter_hash,bool inc)1638 static bool __ftrace_hash_rec_update(struct ftrace_ops *ops,
1639 int filter_hash,
1640 bool inc)
1641 {
1642 struct ftrace_hash *hash;
1643 struct ftrace_hash *other_hash;
1644 struct ftrace_page *pg;
1645 struct dyn_ftrace *rec;
1646 bool update = false;
1647 int count = 0;
1648 int all = false;
1649
1650 /* Only update if the ops has been registered */
1651 if (!(ops->flags & FTRACE_OPS_FL_ENABLED))
1652 return false;
1653
1654 /*
1655 * In the filter_hash case:
1656 * If the count is zero, we update all records.
1657 * Otherwise we just update the items in the hash.
1658 *
1659 * In the notrace_hash case:
1660 * We enable the update in the hash.
1661 * As disabling notrace means enabling the tracing,
1662 * and enabling notrace means disabling, the inc variable
1663 * gets inversed.
1664 */
1665 if (filter_hash) {
1666 hash = ops->func_hash->filter_hash;
1667 other_hash = ops->func_hash->notrace_hash;
1668 if (ftrace_hash_empty(hash))
1669 all = true;
1670 } else {
1671 inc = !inc;
1672 hash = ops->func_hash->notrace_hash;
1673 other_hash = ops->func_hash->filter_hash;
1674 /*
1675 * If the notrace hash has no items,
1676 * then there's nothing to do.
1677 */
1678 if (ftrace_hash_empty(hash))
1679 return false;
1680 }
1681
1682 do_for_each_ftrace_rec(pg, rec) {
1683 int in_other_hash = 0;
1684 int in_hash = 0;
1685 int match = 0;
1686
1687 if (rec->flags & FTRACE_FL_DISABLED)
1688 continue;
1689
1690 if (all) {
1691 /*
1692 * Only the filter_hash affects all records.
1693 * Update if the record is not in the notrace hash.
1694 */
1695 if (!other_hash || !ftrace_lookup_ip(other_hash, rec->ip))
1696 match = 1;
1697 } else {
1698 in_hash = !!ftrace_lookup_ip(hash, rec->ip);
1699 in_other_hash = !!ftrace_lookup_ip(other_hash, rec->ip);
1700
1701 /*
1702 * If filter_hash is set, we want to match all functions
1703 * that are in the hash but not in the other hash.
1704 *
1705 * If filter_hash is not set, then we are decrementing.
1706 * That means we match anything that is in the hash
1707 * and also in the other_hash. That is, we need to turn
1708 * off functions in the other hash because they are disabled
1709 * by this hash.
1710 */
1711 if (filter_hash && in_hash && !in_other_hash)
1712 match = 1;
1713 else if (!filter_hash && in_hash &&
1714 (in_other_hash || ftrace_hash_empty(other_hash)))
1715 match = 1;
1716 }
1717 if (!match)
1718 continue;
1719
1720 if (inc) {
1721 rec->flags++;
1722 if (FTRACE_WARN_ON(ftrace_rec_count(rec) == FTRACE_REF_MAX))
1723 return false;
1724
1725 if (ops->flags & FTRACE_OPS_FL_DIRECT)
1726 rec->flags |= FTRACE_FL_DIRECT;
1727
1728 /*
1729 * If there's only a single callback registered to a
1730 * function, and the ops has a trampoline registered
1731 * for it, then we can call it directly.
1732 */
1733 if (ftrace_rec_count(rec) == 1 && ops->trampoline)
1734 rec->flags |= FTRACE_FL_TRAMP;
1735 else
1736 /*
1737 * If we are adding another function callback
1738 * to this function, and the previous had a
1739 * custom trampoline in use, then we need to go
1740 * back to the default trampoline.
1741 */
1742 rec->flags &= ~FTRACE_FL_TRAMP;
1743
1744 /*
1745 * If any ops wants regs saved for this function
1746 * then all ops will get saved regs.
1747 */
1748 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS)
1749 rec->flags |= FTRACE_FL_REGS;
1750 } else {
1751 if (FTRACE_WARN_ON(ftrace_rec_count(rec) == 0))
1752 return false;
1753 rec->flags--;
1754
1755 /*
1756 * Only the internal direct_ops should have the
1757 * DIRECT flag set. Thus, if it is removing a
1758 * function, then that function should no longer
1759 * be direct.
1760 */
1761 if (ops->flags & FTRACE_OPS_FL_DIRECT)
1762 rec->flags &= ~FTRACE_FL_DIRECT;
1763
1764 /*
1765 * If the rec had REGS enabled and the ops that is
1766 * being removed had REGS set, then see if there is
1767 * still any ops for this record that wants regs.
1768 * If not, we can stop recording them.
1769 */
1770 if (ftrace_rec_count(rec) > 0 &&
1771 rec->flags & FTRACE_FL_REGS &&
1772 ops->flags & FTRACE_OPS_FL_SAVE_REGS) {
1773 if (!test_rec_ops_needs_regs(rec))
1774 rec->flags &= ~FTRACE_FL_REGS;
1775 }
1776
1777 /*
1778 * The TRAMP needs to be set only if rec count
1779 * is decremented to one, and the ops that is
1780 * left has a trampoline. As TRAMP can only be
1781 * enabled if there is only a single ops attached
1782 * to it.
1783 */
1784 if (ftrace_rec_count(rec) == 1 &&
1785 ftrace_find_tramp_ops_any_other(rec, ops))
1786 rec->flags |= FTRACE_FL_TRAMP;
1787 else
1788 rec->flags &= ~FTRACE_FL_TRAMP;
1789
1790 /*
1791 * flags will be cleared in ftrace_check_record()
1792 * if rec count is zero.
1793 */
1794 }
1795 count++;
1796
1797 /* Must match FTRACE_UPDATE_CALLS in ftrace_modify_all_code() */
1798 update |= ftrace_test_record(rec, true) != FTRACE_UPDATE_IGNORE;
1799
1800 /* Shortcut, if we handled all records, we are done. */
1801 if (!all && count == hash->count)
1802 return update;
1803 } while_for_each_ftrace_rec();
1804
1805 return update;
1806 }
1807
ftrace_hash_rec_disable(struct ftrace_ops * ops,int filter_hash)1808 static bool ftrace_hash_rec_disable(struct ftrace_ops *ops,
1809 int filter_hash)
1810 {
1811 return __ftrace_hash_rec_update(ops, filter_hash, 0);
1812 }
1813
ftrace_hash_rec_enable(struct ftrace_ops * ops,int filter_hash)1814 static bool ftrace_hash_rec_enable(struct ftrace_ops *ops,
1815 int filter_hash)
1816 {
1817 return __ftrace_hash_rec_update(ops, filter_hash, 1);
1818 }
1819
ftrace_hash_rec_update_modify(struct ftrace_ops * ops,int filter_hash,int inc)1820 static void ftrace_hash_rec_update_modify(struct ftrace_ops *ops,
1821 int filter_hash, int inc)
1822 {
1823 struct ftrace_ops *op;
1824
1825 __ftrace_hash_rec_update(ops, filter_hash, inc);
1826
1827 if (ops->func_hash != &global_ops.local_hash)
1828 return;
1829
1830 /*
1831 * If the ops shares the global_ops hash, then we need to update
1832 * all ops that are enabled and use this hash.
1833 */
1834 do_for_each_ftrace_op(op, ftrace_ops_list) {
1835 /* Already done */
1836 if (op == ops)
1837 continue;
1838 if (op->func_hash == &global_ops.local_hash)
1839 __ftrace_hash_rec_update(op, filter_hash, inc);
1840 } while_for_each_ftrace_op(op);
1841 }
1842
ftrace_hash_rec_disable_modify(struct ftrace_ops * ops,int filter_hash)1843 static void ftrace_hash_rec_disable_modify(struct ftrace_ops *ops,
1844 int filter_hash)
1845 {
1846 ftrace_hash_rec_update_modify(ops, filter_hash, 0);
1847 }
1848
ftrace_hash_rec_enable_modify(struct ftrace_ops * ops,int filter_hash)1849 static void ftrace_hash_rec_enable_modify(struct ftrace_ops *ops,
1850 int filter_hash)
1851 {
1852 ftrace_hash_rec_update_modify(ops, filter_hash, 1);
1853 }
1854
1855 /*
1856 * Try to update IPMODIFY flag on each ftrace_rec. Return 0 if it is OK
1857 * or no-needed to update, -EBUSY if it detects a conflict of the flag
1858 * on a ftrace_rec, and -EINVAL if the new_hash tries to trace all recs.
1859 * Note that old_hash and new_hash has below meanings
1860 * - If the hash is NULL, it hits all recs (if IPMODIFY is set, this is rejected)
1861 * - If the hash is EMPTY_HASH, it hits nothing
1862 * - Anything else hits the recs which match the hash entries.
1863 */
__ftrace_hash_update_ipmodify(struct ftrace_ops * ops,struct ftrace_hash * old_hash,struct ftrace_hash * new_hash)1864 static int __ftrace_hash_update_ipmodify(struct ftrace_ops *ops,
1865 struct ftrace_hash *old_hash,
1866 struct ftrace_hash *new_hash)
1867 {
1868 struct ftrace_page *pg;
1869 struct dyn_ftrace *rec, *end = NULL;
1870 int in_old, in_new;
1871
1872 /* Only update if the ops has been registered */
1873 if (!(ops->flags & FTRACE_OPS_FL_ENABLED))
1874 return 0;
1875
1876 if (!(ops->flags & FTRACE_OPS_FL_IPMODIFY))
1877 return 0;
1878
1879 /*
1880 * Since the IPMODIFY is a very address sensitive action, we do not
1881 * allow ftrace_ops to set all functions to new hash.
1882 */
1883 if (!new_hash || !old_hash)
1884 return -EINVAL;
1885
1886 /* Update rec->flags */
1887 do_for_each_ftrace_rec(pg, rec) {
1888
1889 if (rec->flags & FTRACE_FL_DISABLED)
1890 continue;
1891
1892 /* We need to update only differences of filter_hash */
1893 in_old = !!ftrace_lookup_ip(old_hash, rec->ip);
1894 in_new = !!ftrace_lookup_ip(new_hash, rec->ip);
1895 if (in_old == in_new)
1896 continue;
1897
1898 if (in_new) {
1899 /* New entries must ensure no others are using it */
1900 if (rec->flags & FTRACE_FL_IPMODIFY)
1901 goto rollback;
1902 rec->flags |= FTRACE_FL_IPMODIFY;
1903 } else /* Removed entry */
1904 rec->flags &= ~FTRACE_FL_IPMODIFY;
1905 } while_for_each_ftrace_rec();
1906
1907 return 0;
1908
1909 rollback:
1910 end = rec;
1911
1912 /* Roll back what we did above */
1913 do_for_each_ftrace_rec(pg, rec) {
1914
1915 if (rec->flags & FTRACE_FL_DISABLED)
1916 continue;
1917
1918 if (rec == end)
1919 goto err_out;
1920
1921 in_old = !!ftrace_lookup_ip(old_hash, rec->ip);
1922 in_new = !!ftrace_lookup_ip(new_hash, rec->ip);
1923 if (in_old == in_new)
1924 continue;
1925
1926 if (in_new)
1927 rec->flags &= ~FTRACE_FL_IPMODIFY;
1928 else
1929 rec->flags |= FTRACE_FL_IPMODIFY;
1930 } while_for_each_ftrace_rec();
1931
1932 err_out:
1933 return -EBUSY;
1934 }
1935
ftrace_hash_ipmodify_enable(struct ftrace_ops * ops)1936 static int ftrace_hash_ipmodify_enable(struct ftrace_ops *ops)
1937 {
1938 struct ftrace_hash *hash = ops->func_hash->filter_hash;
1939
1940 if (ftrace_hash_empty(hash))
1941 hash = NULL;
1942
1943 return __ftrace_hash_update_ipmodify(ops, EMPTY_HASH, hash);
1944 }
1945
1946 /* Disabling always succeeds */
ftrace_hash_ipmodify_disable(struct ftrace_ops * ops)1947 static void ftrace_hash_ipmodify_disable(struct ftrace_ops *ops)
1948 {
1949 struct ftrace_hash *hash = ops->func_hash->filter_hash;
1950
1951 if (ftrace_hash_empty(hash))
1952 hash = NULL;
1953
1954 __ftrace_hash_update_ipmodify(ops, hash, EMPTY_HASH);
1955 }
1956
ftrace_hash_ipmodify_update(struct ftrace_ops * ops,struct ftrace_hash * new_hash)1957 static int ftrace_hash_ipmodify_update(struct ftrace_ops *ops,
1958 struct ftrace_hash *new_hash)
1959 {
1960 struct ftrace_hash *old_hash = ops->func_hash->filter_hash;
1961
1962 if (ftrace_hash_empty(old_hash))
1963 old_hash = NULL;
1964
1965 if (ftrace_hash_empty(new_hash))
1966 new_hash = NULL;
1967
1968 return __ftrace_hash_update_ipmodify(ops, old_hash, new_hash);
1969 }
1970
print_ip_ins(const char * fmt,const unsigned char * p)1971 static void print_ip_ins(const char *fmt, const unsigned char *p)
1972 {
1973 char ins[MCOUNT_INSN_SIZE];
1974 int i;
1975
1976 if (copy_from_kernel_nofault(ins, p, MCOUNT_INSN_SIZE)) {
1977 printk(KERN_CONT "%s[FAULT] %px\n", fmt, p);
1978 return;
1979 }
1980
1981 printk(KERN_CONT "%s", fmt);
1982
1983 for (i = 0; i < MCOUNT_INSN_SIZE; i++)
1984 printk(KERN_CONT "%s%02x", i ? ":" : "", ins[i]);
1985 }
1986
1987 enum ftrace_bug_type ftrace_bug_type;
1988 const void *ftrace_expected;
1989
print_bug_type(void)1990 static void print_bug_type(void)
1991 {
1992 switch (ftrace_bug_type) {
1993 case FTRACE_BUG_UNKNOWN:
1994 break;
1995 case FTRACE_BUG_INIT:
1996 pr_info("Initializing ftrace call sites\n");
1997 break;
1998 case FTRACE_BUG_NOP:
1999 pr_info("Setting ftrace call site to NOP\n");
2000 break;
2001 case FTRACE_BUG_CALL:
2002 pr_info("Setting ftrace call site to call ftrace function\n");
2003 break;
2004 case FTRACE_BUG_UPDATE:
2005 pr_info("Updating ftrace call site to call a different ftrace function\n");
2006 break;
2007 }
2008 }
2009
2010 /**
2011 * ftrace_bug - report and shutdown function tracer
2012 * @failed: The failed type (EFAULT, EINVAL, EPERM)
2013 * @rec: The record that failed
2014 *
2015 * The arch code that enables or disables the function tracing
2016 * can call ftrace_bug() when it has detected a problem in
2017 * modifying the code. @failed should be one of either:
2018 * EFAULT - if the problem happens on reading the @ip address
2019 * EINVAL - if what is read at @ip is not what was expected
2020 * EPERM - if the problem happens on writing to the @ip address
2021 */
ftrace_bug(int failed,struct dyn_ftrace * rec)2022 void ftrace_bug(int failed, struct dyn_ftrace *rec)
2023 {
2024 unsigned long ip = rec ? rec->ip : 0;
2025
2026 pr_info("------------[ ftrace bug ]------------\n");
2027
2028 switch (failed) {
2029 case -EFAULT:
2030 pr_info("ftrace faulted on modifying ");
2031 print_ip_sym(KERN_INFO, ip);
2032 break;
2033 case -EINVAL:
2034 pr_info("ftrace failed to modify ");
2035 print_ip_sym(KERN_INFO, ip);
2036 print_ip_ins(" actual: ", (unsigned char *)ip);
2037 pr_cont("\n");
2038 if (ftrace_expected) {
2039 print_ip_ins(" expected: ", ftrace_expected);
2040 pr_cont("\n");
2041 }
2042 break;
2043 case -EPERM:
2044 pr_info("ftrace faulted on writing ");
2045 print_ip_sym(KERN_INFO, ip);
2046 break;
2047 default:
2048 pr_info("ftrace faulted on unknown error ");
2049 print_ip_sym(KERN_INFO, ip);
2050 }
2051 print_bug_type();
2052 if (rec) {
2053 struct ftrace_ops *ops = NULL;
2054
2055 pr_info("ftrace record flags: %lx\n", rec->flags);
2056 pr_cont(" (%ld)%s", ftrace_rec_count(rec),
2057 rec->flags & FTRACE_FL_REGS ? " R" : " ");
2058 if (rec->flags & FTRACE_FL_TRAMP_EN) {
2059 ops = ftrace_find_tramp_ops_any(rec);
2060 if (ops) {
2061 do {
2062 pr_cont("\ttramp: %pS (%pS)",
2063 (void *)ops->trampoline,
2064 (void *)ops->func);
2065 ops = ftrace_find_tramp_ops_next(rec, ops);
2066 } while (ops);
2067 } else
2068 pr_cont("\ttramp: ERROR!");
2069
2070 }
2071 ip = ftrace_get_addr_curr(rec);
2072 pr_cont("\n expected tramp: %lx\n", ip);
2073 }
2074
2075 FTRACE_WARN_ON_ONCE(1);
2076 }
2077
ftrace_check_record(struct dyn_ftrace * rec,bool enable,bool update)2078 static int ftrace_check_record(struct dyn_ftrace *rec, bool enable, bool update)
2079 {
2080 unsigned long flag = 0UL;
2081
2082 ftrace_bug_type = FTRACE_BUG_UNKNOWN;
2083
2084 if (rec->flags & FTRACE_FL_DISABLED)
2085 return FTRACE_UPDATE_IGNORE;
2086
2087 /*
2088 * If we are updating calls:
2089 *
2090 * If the record has a ref count, then we need to enable it
2091 * because someone is using it.
2092 *
2093 * Otherwise we make sure its disabled.
2094 *
2095 * If we are disabling calls, then disable all records that
2096 * are enabled.
2097 */
2098 if (enable && ftrace_rec_count(rec))
2099 flag = FTRACE_FL_ENABLED;
2100
2101 /*
2102 * If enabling and the REGS flag does not match the REGS_EN, or
2103 * the TRAMP flag doesn't match the TRAMP_EN, then do not ignore
2104 * this record. Set flags to fail the compare against ENABLED.
2105 * Same for direct calls.
2106 */
2107 if (flag) {
2108 if (!(rec->flags & FTRACE_FL_REGS) !=
2109 !(rec->flags & FTRACE_FL_REGS_EN))
2110 flag |= FTRACE_FL_REGS;
2111
2112 if (!(rec->flags & FTRACE_FL_TRAMP) !=
2113 !(rec->flags & FTRACE_FL_TRAMP_EN))
2114 flag |= FTRACE_FL_TRAMP;
2115
2116 /*
2117 * Direct calls are special, as count matters.
2118 * We must test the record for direct, if the
2119 * DIRECT and DIRECT_EN do not match, but only
2120 * if the count is 1. That's because, if the
2121 * count is something other than one, we do not
2122 * want the direct enabled (it will be done via the
2123 * direct helper). But if DIRECT_EN is set, and
2124 * the count is not one, we need to clear it.
2125 */
2126 if (ftrace_rec_count(rec) == 1) {
2127 if (!(rec->flags & FTRACE_FL_DIRECT) !=
2128 !(rec->flags & FTRACE_FL_DIRECT_EN))
2129 flag |= FTRACE_FL_DIRECT;
2130 } else if (rec->flags & FTRACE_FL_DIRECT_EN) {
2131 flag |= FTRACE_FL_DIRECT;
2132 }
2133 }
2134
2135 /* If the state of this record hasn't changed, then do nothing */
2136 if ((rec->flags & FTRACE_FL_ENABLED) == flag)
2137 return FTRACE_UPDATE_IGNORE;
2138
2139 if (flag) {
2140 /* Save off if rec is being enabled (for return value) */
2141 flag ^= rec->flags & FTRACE_FL_ENABLED;
2142
2143 if (update) {
2144 rec->flags |= FTRACE_FL_ENABLED;
2145 if (flag & FTRACE_FL_REGS) {
2146 if (rec->flags & FTRACE_FL_REGS)
2147 rec->flags |= FTRACE_FL_REGS_EN;
2148 else
2149 rec->flags &= ~FTRACE_FL_REGS_EN;
2150 }
2151 if (flag & FTRACE_FL_TRAMP) {
2152 if (rec->flags & FTRACE_FL_TRAMP)
2153 rec->flags |= FTRACE_FL_TRAMP_EN;
2154 else
2155 rec->flags &= ~FTRACE_FL_TRAMP_EN;
2156 }
2157 if (flag & FTRACE_FL_DIRECT) {
2158 /*
2159 * If there's only one user (direct_ops helper)
2160 * then we can call the direct function
2161 * directly (no ftrace trampoline).
2162 */
2163 if (ftrace_rec_count(rec) == 1) {
2164 if (rec->flags & FTRACE_FL_DIRECT)
2165 rec->flags |= FTRACE_FL_DIRECT_EN;
2166 else
2167 rec->flags &= ~FTRACE_FL_DIRECT_EN;
2168 } else {
2169 /*
2170 * Can only call directly if there's
2171 * only one callback to the function.
2172 */
2173 rec->flags &= ~FTRACE_FL_DIRECT_EN;
2174 }
2175 }
2176 }
2177
2178 /*
2179 * If this record is being updated from a nop, then
2180 * return UPDATE_MAKE_CALL.
2181 * Otherwise,
2182 * return UPDATE_MODIFY_CALL to tell the caller to convert
2183 * from the save regs, to a non-save regs function or
2184 * vice versa, or from a trampoline call.
2185 */
2186 if (flag & FTRACE_FL_ENABLED) {
2187 ftrace_bug_type = FTRACE_BUG_CALL;
2188 return FTRACE_UPDATE_MAKE_CALL;
2189 }
2190
2191 ftrace_bug_type = FTRACE_BUG_UPDATE;
2192 return FTRACE_UPDATE_MODIFY_CALL;
2193 }
2194
2195 if (update) {
2196 /* If there's no more users, clear all flags */
2197 if (!ftrace_rec_count(rec))
2198 rec->flags = 0;
2199 else
2200 /*
2201 * Just disable the record, but keep the ops TRAMP
2202 * and REGS states. The _EN flags must be disabled though.
2203 */
2204 rec->flags &= ~(FTRACE_FL_ENABLED | FTRACE_FL_TRAMP_EN |
2205 FTRACE_FL_REGS_EN | FTRACE_FL_DIRECT_EN);
2206 }
2207
2208 ftrace_bug_type = FTRACE_BUG_NOP;
2209 return FTRACE_UPDATE_MAKE_NOP;
2210 }
2211
2212 /**
2213 * ftrace_update_record, set a record that now is tracing or not
2214 * @rec: the record to update
2215 * @enable: set to true if the record is tracing, false to force disable
2216 *
2217 * The records that represent all functions that can be traced need
2218 * to be updated when tracing has been enabled.
2219 */
ftrace_update_record(struct dyn_ftrace * rec,bool enable)2220 int ftrace_update_record(struct dyn_ftrace *rec, bool enable)
2221 {
2222 return ftrace_check_record(rec, enable, true);
2223 }
2224
2225 /**
2226 * ftrace_test_record, check if the record has been enabled or not
2227 * @rec: the record to test
2228 * @enable: set to true to check if enabled, false if it is disabled
2229 *
2230 * The arch code may need to test if a record is already set to
2231 * tracing to determine how to modify the function code that it
2232 * represents.
2233 */
ftrace_test_record(struct dyn_ftrace * rec,bool enable)2234 int ftrace_test_record(struct dyn_ftrace *rec, bool enable)
2235 {
2236 return ftrace_check_record(rec, enable, false);
2237 }
2238
2239 static struct ftrace_ops *
ftrace_find_tramp_ops_any(struct dyn_ftrace * rec)2240 ftrace_find_tramp_ops_any(struct dyn_ftrace *rec)
2241 {
2242 struct ftrace_ops *op;
2243 unsigned long ip = rec->ip;
2244
2245 do_for_each_ftrace_op(op, ftrace_ops_list) {
2246
2247 if (!op->trampoline)
2248 continue;
2249
2250 if (hash_contains_ip(ip, op->func_hash))
2251 return op;
2252 } while_for_each_ftrace_op(op);
2253
2254 return NULL;
2255 }
2256
2257 static struct ftrace_ops *
ftrace_find_tramp_ops_any_other(struct dyn_ftrace * rec,struct ftrace_ops * op_exclude)2258 ftrace_find_tramp_ops_any_other(struct dyn_ftrace *rec, struct ftrace_ops *op_exclude)
2259 {
2260 struct ftrace_ops *op;
2261 unsigned long ip = rec->ip;
2262
2263 do_for_each_ftrace_op(op, ftrace_ops_list) {
2264
2265 if (op == op_exclude || !op->trampoline)
2266 continue;
2267
2268 if (hash_contains_ip(ip, op->func_hash))
2269 return op;
2270 } while_for_each_ftrace_op(op);
2271
2272 return NULL;
2273 }
2274
2275 static struct ftrace_ops *
ftrace_find_tramp_ops_next(struct dyn_ftrace * rec,struct ftrace_ops * op)2276 ftrace_find_tramp_ops_next(struct dyn_ftrace *rec,
2277 struct ftrace_ops *op)
2278 {
2279 unsigned long ip = rec->ip;
2280
2281 while_for_each_ftrace_op(op) {
2282
2283 if (!op->trampoline)
2284 continue;
2285
2286 if (hash_contains_ip(ip, op->func_hash))
2287 return op;
2288 }
2289
2290 return NULL;
2291 }
2292
2293 static struct ftrace_ops *
ftrace_find_tramp_ops_curr(struct dyn_ftrace * rec)2294 ftrace_find_tramp_ops_curr(struct dyn_ftrace *rec)
2295 {
2296 struct ftrace_ops *op;
2297 unsigned long ip = rec->ip;
2298
2299 /*
2300 * Need to check removed ops first.
2301 * If they are being removed, and this rec has a tramp,
2302 * and this rec is in the ops list, then it would be the
2303 * one with the tramp.
2304 */
2305 if (removed_ops) {
2306 if (hash_contains_ip(ip, &removed_ops->old_hash))
2307 return removed_ops;
2308 }
2309
2310 /*
2311 * Need to find the current trampoline for a rec.
2312 * Now, a trampoline is only attached to a rec if there
2313 * was a single 'ops' attached to it. But this can be called
2314 * when we are adding another op to the rec or removing the
2315 * current one. Thus, if the op is being added, we can
2316 * ignore it because it hasn't attached itself to the rec
2317 * yet.
2318 *
2319 * If an ops is being modified (hooking to different functions)
2320 * then we don't care about the new functions that are being
2321 * added, just the old ones (that are probably being removed).
2322 *
2323 * If we are adding an ops to a function that already is using
2324 * a trampoline, it needs to be removed (trampolines are only
2325 * for single ops connected), then an ops that is not being
2326 * modified also needs to be checked.
2327 */
2328 do_for_each_ftrace_op(op, ftrace_ops_list) {
2329
2330 if (!op->trampoline)
2331 continue;
2332
2333 /*
2334 * If the ops is being added, it hasn't gotten to
2335 * the point to be removed from this tree yet.
2336 */
2337 if (op->flags & FTRACE_OPS_FL_ADDING)
2338 continue;
2339
2340
2341 /*
2342 * If the ops is being modified and is in the old
2343 * hash, then it is probably being removed from this
2344 * function.
2345 */
2346 if ((op->flags & FTRACE_OPS_FL_MODIFYING) &&
2347 hash_contains_ip(ip, &op->old_hash))
2348 return op;
2349 /*
2350 * If the ops is not being added or modified, and it's
2351 * in its normal filter hash, then this must be the one
2352 * we want!
2353 */
2354 if (!(op->flags & FTRACE_OPS_FL_MODIFYING) &&
2355 hash_contains_ip(ip, op->func_hash))
2356 return op;
2357
2358 } while_for_each_ftrace_op(op);
2359
2360 return NULL;
2361 }
2362
2363 static struct ftrace_ops *
ftrace_find_tramp_ops_new(struct dyn_ftrace * rec)2364 ftrace_find_tramp_ops_new(struct dyn_ftrace *rec)
2365 {
2366 struct ftrace_ops *op;
2367 unsigned long ip = rec->ip;
2368
2369 do_for_each_ftrace_op(op, ftrace_ops_list) {
2370 /* pass rec in as regs to have non-NULL val */
2371 if (hash_contains_ip(ip, op->func_hash))
2372 return op;
2373 } while_for_each_ftrace_op(op);
2374
2375 return NULL;
2376 }
2377
2378 #ifdef CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS
2379 /* Protected by rcu_tasks for reading, and direct_mutex for writing */
2380 static struct ftrace_hash *direct_functions = EMPTY_HASH;
2381 static DEFINE_MUTEX(direct_mutex);
2382 int ftrace_direct_func_count;
2383
2384 /*
2385 * Search the direct_functions hash to see if the given instruction pointer
2386 * has a direct caller attached to it.
2387 */
ftrace_find_rec_direct(unsigned long ip)2388 unsigned long ftrace_find_rec_direct(unsigned long ip)
2389 {
2390 struct ftrace_func_entry *entry;
2391
2392 entry = __ftrace_lookup_ip(direct_functions, ip);
2393 if (!entry)
2394 return 0;
2395
2396 return entry->direct;
2397 }
2398
call_direct_funcs(unsigned long ip,unsigned long pip,struct ftrace_ops * ops,struct pt_regs * regs)2399 static void call_direct_funcs(unsigned long ip, unsigned long pip,
2400 struct ftrace_ops *ops, struct pt_regs *regs)
2401 {
2402 unsigned long addr;
2403
2404 addr = ftrace_find_rec_direct(ip);
2405 if (!addr)
2406 return;
2407
2408 arch_ftrace_set_direct_caller(regs, addr);
2409 }
2410
2411 struct ftrace_ops direct_ops = {
2412 .func = call_direct_funcs,
2413 .flags = FTRACE_OPS_FL_IPMODIFY | FTRACE_OPS_FL_RECURSION_SAFE
2414 | FTRACE_OPS_FL_DIRECT | FTRACE_OPS_FL_SAVE_REGS
2415 | FTRACE_OPS_FL_PERMANENT,
2416 /*
2417 * By declaring the main trampoline as this trampoline
2418 * it will never have one allocated for it. Allocated
2419 * trampolines should not call direct functions.
2420 * The direct_ops should only be called by the builtin
2421 * ftrace_regs_caller trampoline.
2422 */
2423 .trampoline = FTRACE_REGS_ADDR,
2424 };
2425 #endif /* CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS */
2426
2427 /**
2428 * ftrace_get_addr_new - Get the call address to set to
2429 * @rec: The ftrace record descriptor
2430 *
2431 * If the record has the FTRACE_FL_REGS set, that means that it
2432 * wants to convert to a callback that saves all regs. If FTRACE_FL_REGS
2433 * is not set, then it wants to convert to the normal callback.
2434 *
2435 * Returns the address of the trampoline to set to
2436 */
ftrace_get_addr_new(struct dyn_ftrace * rec)2437 unsigned long ftrace_get_addr_new(struct dyn_ftrace *rec)
2438 {
2439 struct ftrace_ops *ops;
2440 unsigned long addr;
2441
2442 if ((rec->flags & FTRACE_FL_DIRECT) &&
2443 (ftrace_rec_count(rec) == 1)) {
2444 addr = ftrace_find_rec_direct(rec->ip);
2445 if (addr)
2446 return addr;
2447 WARN_ON_ONCE(1);
2448 }
2449
2450 /* Trampolines take precedence over regs */
2451 if (rec->flags & FTRACE_FL_TRAMP) {
2452 ops = ftrace_find_tramp_ops_new(rec);
2453 if (FTRACE_WARN_ON(!ops || !ops->trampoline)) {
2454 pr_warn("Bad trampoline accounting at: %p (%pS) (%lx)\n",
2455 (void *)rec->ip, (void *)rec->ip, rec->flags);
2456 /* Ftrace is shutting down, return anything */
2457 return (unsigned long)FTRACE_ADDR;
2458 }
2459 return ops->trampoline;
2460 }
2461
2462 if (rec->flags & FTRACE_FL_REGS)
2463 return (unsigned long)FTRACE_REGS_ADDR;
2464 else
2465 return (unsigned long)FTRACE_ADDR;
2466 }
2467
2468 /**
2469 * ftrace_get_addr_curr - Get the call address that is already there
2470 * @rec: The ftrace record descriptor
2471 *
2472 * The FTRACE_FL_REGS_EN is set when the record already points to
2473 * a function that saves all the regs. Basically the '_EN' version
2474 * represents the current state of the function.
2475 *
2476 * Returns the address of the trampoline that is currently being called
2477 */
ftrace_get_addr_curr(struct dyn_ftrace * rec)2478 unsigned long ftrace_get_addr_curr(struct dyn_ftrace *rec)
2479 {
2480 struct ftrace_ops *ops;
2481 unsigned long addr;
2482
2483 /* Direct calls take precedence over trampolines */
2484 if (rec->flags & FTRACE_FL_DIRECT_EN) {
2485 addr = ftrace_find_rec_direct(rec->ip);
2486 if (addr)
2487 return addr;
2488 WARN_ON_ONCE(1);
2489 }
2490
2491 /* Trampolines take precedence over regs */
2492 if (rec->flags & FTRACE_FL_TRAMP_EN) {
2493 ops = ftrace_find_tramp_ops_curr(rec);
2494 if (FTRACE_WARN_ON(!ops)) {
2495 pr_warn("Bad trampoline accounting at: %p (%pS)\n",
2496 (void *)rec->ip, (void *)rec->ip);
2497 /* Ftrace is shutting down, return anything */
2498 return (unsigned long)FTRACE_ADDR;
2499 }
2500 return ops->trampoline;
2501 }
2502
2503 if (rec->flags & FTRACE_FL_REGS_EN)
2504 return (unsigned long)FTRACE_REGS_ADDR;
2505 else
2506 return (unsigned long)FTRACE_ADDR;
2507 }
2508
2509 static int
__ftrace_replace_code(struct dyn_ftrace * rec,bool enable)2510 __ftrace_replace_code(struct dyn_ftrace *rec, bool enable)
2511 {
2512 unsigned long ftrace_old_addr;
2513 unsigned long ftrace_addr;
2514 int ret;
2515
2516 ftrace_addr = ftrace_get_addr_new(rec);
2517
2518 /* This needs to be done before we call ftrace_update_record */
2519 ftrace_old_addr = ftrace_get_addr_curr(rec);
2520
2521 ret = ftrace_update_record(rec, enable);
2522
2523 ftrace_bug_type = FTRACE_BUG_UNKNOWN;
2524
2525 switch (ret) {
2526 case FTRACE_UPDATE_IGNORE:
2527 return 0;
2528
2529 case FTRACE_UPDATE_MAKE_CALL:
2530 ftrace_bug_type = FTRACE_BUG_CALL;
2531 return ftrace_make_call(rec, ftrace_addr);
2532
2533 case FTRACE_UPDATE_MAKE_NOP:
2534 ftrace_bug_type = FTRACE_BUG_NOP;
2535 return ftrace_make_nop(NULL, rec, ftrace_old_addr);
2536
2537 case FTRACE_UPDATE_MODIFY_CALL:
2538 ftrace_bug_type = FTRACE_BUG_UPDATE;
2539 return ftrace_modify_call(rec, ftrace_old_addr, ftrace_addr);
2540 }
2541
2542 return -1; /* unknown ftrace bug */
2543 }
2544
ftrace_replace_code(int mod_flags)2545 void __weak ftrace_replace_code(int mod_flags)
2546 {
2547 struct dyn_ftrace *rec;
2548 struct ftrace_page *pg;
2549 bool enable = mod_flags & FTRACE_MODIFY_ENABLE_FL;
2550 int schedulable = mod_flags & FTRACE_MODIFY_MAY_SLEEP_FL;
2551 int failed;
2552
2553 if (unlikely(ftrace_disabled))
2554 return;
2555
2556 do_for_each_ftrace_rec(pg, rec) {
2557
2558 if (rec->flags & FTRACE_FL_DISABLED)
2559 continue;
2560
2561 failed = __ftrace_replace_code(rec, enable);
2562 if (failed) {
2563 ftrace_bug(failed, rec);
2564 /* Stop processing */
2565 return;
2566 }
2567 if (schedulable)
2568 cond_resched();
2569 } while_for_each_ftrace_rec();
2570 }
2571
2572 struct ftrace_rec_iter {
2573 struct ftrace_page *pg;
2574 int index;
2575 };
2576
2577 /**
2578 * ftrace_rec_iter_start, start up iterating over traced functions
2579 *
2580 * Returns an iterator handle that is used to iterate over all
2581 * the records that represent address locations where functions
2582 * are traced.
2583 *
2584 * May return NULL if no records are available.
2585 */
ftrace_rec_iter_start(void)2586 struct ftrace_rec_iter *ftrace_rec_iter_start(void)
2587 {
2588 /*
2589 * We only use a single iterator.
2590 * Protected by the ftrace_lock mutex.
2591 */
2592 static struct ftrace_rec_iter ftrace_rec_iter;
2593 struct ftrace_rec_iter *iter = &ftrace_rec_iter;
2594
2595 iter->pg = ftrace_pages_start;
2596 iter->index = 0;
2597
2598 /* Could have empty pages */
2599 while (iter->pg && !iter->pg->index)
2600 iter->pg = iter->pg->next;
2601
2602 if (!iter->pg)
2603 return NULL;
2604
2605 return iter;
2606 }
2607
2608 /**
2609 * ftrace_rec_iter_next, get the next record to process.
2610 * @iter: The handle to the iterator.
2611 *
2612 * Returns the next iterator after the given iterator @iter.
2613 */
ftrace_rec_iter_next(struct ftrace_rec_iter * iter)2614 struct ftrace_rec_iter *ftrace_rec_iter_next(struct ftrace_rec_iter *iter)
2615 {
2616 iter->index++;
2617
2618 if (iter->index >= iter->pg->index) {
2619 iter->pg = iter->pg->next;
2620 iter->index = 0;
2621
2622 /* Could have empty pages */
2623 while (iter->pg && !iter->pg->index)
2624 iter->pg = iter->pg->next;
2625 }
2626
2627 if (!iter->pg)
2628 return NULL;
2629
2630 return iter;
2631 }
2632
2633 /**
2634 * ftrace_rec_iter_record, get the record at the iterator location
2635 * @iter: The current iterator location
2636 *
2637 * Returns the record that the current @iter is at.
2638 */
ftrace_rec_iter_record(struct ftrace_rec_iter * iter)2639 struct dyn_ftrace *ftrace_rec_iter_record(struct ftrace_rec_iter *iter)
2640 {
2641 return &iter->pg->records[iter->index];
2642 }
2643
2644 static int
ftrace_nop_initialize(struct module * mod,struct dyn_ftrace * rec)2645 ftrace_nop_initialize(struct module *mod, struct dyn_ftrace *rec)
2646 {
2647 int ret;
2648
2649 if (unlikely(ftrace_disabled))
2650 return 0;
2651
2652 ret = ftrace_init_nop(mod, rec);
2653 if (ret) {
2654 ftrace_bug_type = FTRACE_BUG_INIT;
2655 ftrace_bug(ret, rec);
2656 return 0;
2657 }
2658 return 1;
2659 }
2660
2661 /*
2662 * archs can override this function if they must do something
2663 * before the modifying code is performed.
2664 */
ftrace_arch_code_modify_prepare(void)2665 int __weak ftrace_arch_code_modify_prepare(void)
2666 {
2667 return 0;
2668 }
2669
2670 /*
2671 * archs can override this function if they must do something
2672 * after the modifying code is performed.
2673 */
ftrace_arch_code_modify_post_process(void)2674 int __weak ftrace_arch_code_modify_post_process(void)
2675 {
2676 return 0;
2677 }
2678
ftrace_modify_all_code(int command)2679 void ftrace_modify_all_code(int command)
2680 {
2681 int update = command & FTRACE_UPDATE_TRACE_FUNC;
2682 int mod_flags = 0;
2683 int err = 0;
2684
2685 if (command & FTRACE_MAY_SLEEP)
2686 mod_flags = FTRACE_MODIFY_MAY_SLEEP_FL;
2687
2688 /*
2689 * If the ftrace_caller calls a ftrace_ops func directly,
2690 * we need to make sure that it only traces functions it
2691 * expects to trace. When doing the switch of functions,
2692 * we need to update to the ftrace_ops_list_func first
2693 * before the transition between old and new calls are set,
2694 * as the ftrace_ops_list_func will check the ops hashes
2695 * to make sure the ops are having the right functions
2696 * traced.
2697 */
2698 if (update) {
2699 err = ftrace_update_ftrace_func(ftrace_ops_list_func);
2700 if (FTRACE_WARN_ON(err))
2701 return;
2702 }
2703
2704 if (command & FTRACE_UPDATE_CALLS)
2705 ftrace_replace_code(mod_flags | FTRACE_MODIFY_ENABLE_FL);
2706 else if (command & FTRACE_DISABLE_CALLS)
2707 ftrace_replace_code(mod_flags);
2708
2709 if (update && ftrace_trace_function != ftrace_ops_list_func) {
2710 function_trace_op = set_function_trace_op;
2711 smp_wmb();
2712 /* If irqs are disabled, we are in stop machine */
2713 if (!irqs_disabled())
2714 smp_call_function(ftrace_sync_ipi, NULL, 1);
2715 err = ftrace_update_ftrace_func(ftrace_trace_function);
2716 if (FTRACE_WARN_ON(err))
2717 return;
2718 }
2719
2720 if (command & FTRACE_START_FUNC_RET)
2721 err = ftrace_enable_ftrace_graph_caller();
2722 else if (command & FTRACE_STOP_FUNC_RET)
2723 err = ftrace_disable_ftrace_graph_caller();
2724 FTRACE_WARN_ON(err);
2725 }
2726
__ftrace_modify_code(void * data)2727 static int __ftrace_modify_code(void *data)
2728 {
2729 int *command = data;
2730
2731 ftrace_modify_all_code(*command);
2732
2733 return 0;
2734 }
2735
2736 /**
2737 * ftrace_run_stop_machine, go back to the stop machine method
2738 * @command: The command to tell ftrace what to do
2739 *
2740 * If an arch needs to fall back to the stop machine method, the
2741 * it can call this function.
2742 */
ftrace_run_stop_machine(int command)2743 void ftrace_run_stop_machine(int command)
2744 {
2745 stop_machine(__ftrace_modify_code, &command, NULL);
2746 }
2747
2748 /**
2749 * arch_ftrace_update_code, modify the code to trace or not trace
2750 * @command: The command that needs to be done
2751 *
2752 * Archs can override this function if it does not need to
2753 * run stop_machine() to modify code.
2754 */
arch_ftrace_update_code(int command)2755 void __weak arch_ftrace_update_code(int command)
2756 {
2757 ftrace_run_stop_machine(command);
2758 }
2759
ftrace_run_update_code(int command)2760 static void ftrace_run_update_code(int command)
2761 {
2762 int ret;
2763
2764 ret = ftrace_arch_code_modify_prepare();
2765 FTRACE_WARN_ON(ret);
2766 if (ret)
2767 return;
2768
2769 /*
2770 * By default we use stop_machine() to modify the code.
2771 * But archs can do what ever they want as long as it
2772 * is safe. The stop_machine() is the safest, but also
2773 * produces the most overhead.
2774 */
2775 arch_ftrace_update_code(command);
2776
2777 ret = ftrace_arch_code_modify_post_process();
2778 FTRACE_WARN_ON(ret);
2779 }
2780
ftrace_run_modify_code(struct ftrace_ops * ops,int command,struct ftrace_ops_hash * old_hash)2781 static void ftrace_run_modify_code(struct ftrace_ops *ops, int command,
2782 struct ftrace_ops_hash *old_hash)
2783 {
2784 ops->flags |= FTRACE_OPS_FL_MODIFYING;
2785 ops->old_hash.filter_hash = old_hash->filter_hash;
2786 ops->old_hash.notrace_hash = old_hash->notrace_hash;
2787 ftrace_run_update_code(command);
2788 ops->old_hash.filter_hash = NULL;
2789 ops->old_hash.notrace_hash = NULL;
2790 ops->flags &= ~FTRACE_OPS_FL_MODIFYING;
2791 }
2792
2793 static ftrace_func_t saved_ftrace_func;
2794 static int ftrace_start_up;
2795
arch_ftrace_trampoline_free(struct ftrace_ops * ops)2796 void __weak arch_ftrace_trampoline_free(struct ftrace_ops *ops)
2797 {
2798 }
2799
2800 /* List of trace_ops that have allocated trampolines */
2801 static LIST_HEAD(ftrace_ops_trampoline_list);
2802
ftrace_add_trampoline_to_kallsyms(struct ftrace_ops * ops)2803 static void ftrace_add_trampoline_to_kallsyms(struct ftrace_ops *ops)
2804 {
2805 lockdep_assert_held(&ftrace_lock);
2806 list_add_rcu(&ops->list, &ftrace_ops_trampoline_list);
2807 }
2808
ftrace_remove_trampoline_from_kallsyms(struct ftrace_ops * ops)2809 static void ftrace_remove_trampoline_from_kallsyms(struct ftrace_ops *ops)
2810 {
2811 lockdep_assert_held(&ftrace_lock);
2812 list_del_rcu(&ops->list);
2813 synchronize_rcu();
2814 }
2815
2816 /*
2817 * "__builtin__ftrace" is used as a module name in /proc/kallsyms for symbols
2818 * for pages allocated for ftrace purposes, even though "__builtin__ftrace" is
2819 * not a module.
2820 */
2821 #define FTRACE_TRAMPOLINE_MOD "__builtin__ftrace"
2822 #define FTRACE_TRAMPOLINE_SYM "ftrace_trampoline"
2823
ftrace_trampoline_free(struct ftrace_ops * ops)2824 static void ftrace_trampoline_free(struct ftrace_ops *ops)
2825 {
2826 if (ops && (ops->flags & FTRACE_OPS_FL_ALLOC_TRAMP) &&
2827 ops->trampoline) {
2828 /*
2829 * Record the text poke event before the ksymbol unregister
2830 * event.
2831 */
2832 perf_event_text_poke((void *)ops->trampoline,
2833 (void *)ops->trampoline,
2834 ops->trampoline_size, NULL, 0);
2835 perf_event_ksymbol(PERF_RECORD_KSYMBOL_TYPE_OOL,
2836 ops->trampoline, ops->trampoline_size,
2837 true, FTRACE_TRAMPOLINE_SYM);
2838 /* Remove from kallsyms after the perf events */
2839 ftrace_remove_trampoline_from_kallsyms(ops);
2840 }
2841
2842 arch_ftrace_trampoline_free(ops);
2843 }
2844
ftrace_startup_enable(int command)2845 static void ftrace_startup_enable(int command)
2846 {
2847 if (saved_ftrace_func != ftrace_trace_function) {
2848 saved_ftrace_func = ftrace_trace_function;
2849 command |= FTRACE_UPDATE_TRACE_FUNC;
2850 }
2851
2852 if (!command || !ftrace_enabled)
2853 return;
2854
2855 ftrace_run_update_code(command);
2856 }
2857
ftrace_startup_all(int command)2858 static void ftrace_startup_all(int command)
2859 {
2860 update_all_ops = true;
2861 ftrace_startup_enable(command);
2862 update_all_ops = false;
2863 }
2864
ftrace_startup(struct ftrace_ops * ops,int command)2865 int ftrace_startup(struct ftrace_ops *ops, int command)
2866 {
2867 int ret;
2868
2869 if (unlikely(ftrace_disabled))
2870 return -ENODEV;
2871
2872 ret = __register_ftrace_function(ops);
2873 if (ret)
2874 return ret;
2875
2876 ftrace_start_up++;
2877
2878 /*
2879 * Note that ftrace probes uses this to start up
2880 * and modify functions it will probe. But we still
2881 * set the ADDING flag for modification, as probes
2882 * do not have trampolines. If they add them in the
2883 * future, then the probes will need to distinguish
2884 * between adding and updating probes.
2885 */
2886 ops->flags |= FTRACE_OPS_FL_ENABLED | FTRACE_OPS_FL_ADDING;
2887
2888 ret = ftrace_hash_ipmodify_enable(ops);
2889 if (ret < 0) {
2890 /* Rollback registration process */
2891 __unregister_ftrace_function(ops);
2892 ftrace_start_up--;
2893 ops->flags &= ~FTRACE_OPS_FL_ENABLED;
2894 if (ops->flags & FTRACE_OPS_FL_DYNAMIC)
2895 ftrace_trampoline_free(ops);
2896 return ret;
2897 }
2898
2899 if (ftrace_hash_rec_enable(ops, 1))
2900 command |= FTRACE_UPDATE_CALLS;
2901
2902 ftrace_startup_enable(command);
2903
2904 /*
2905 * If ftrace is in an undefined state, we just remove ops from list
2906 * to prevent the NULL pointer, instead of totally rolling it back and
2907 * free trampoline, because those actions could cause further damage.
2908 */
2909 if (unlikely(ftrace_disabled)) {
2910 __unregister_ftrace_function(ops);
2911 return -ENODEV;
2912 }
2913
2914 ops->flags &= ~FTRACE_OPS_FL_ADDING;
2915
2916 return 0;
2917 }
2918
ftrace_shutdown(struct ftrace_ops * ops,int command)2919 int ftrace_shutdown(struct ftrace_ops *ops, int command)
2920 {
2921 int ret;
2922
2923 if (unlikely(ftrace_disabled))
2924 return -ENODEV;
2925
2926 ret = __unregister_ftrace_function(ops);
2927 if (ret)
2928 return ret;
2929
2930 ftrace_start_up--;
2931 /*
2932 * Just warn in case of unbalance, no need to kill ftrace, it's not
2933 * critical but the ftrace_call callers may be never nopped again after
2934 * further ftrace uses.
2935 */
2936 WARN_ON_ONCE(ftrace_start_up < 0);
2937
2938 /* Disabling ipmodify never fails */
2939 ftrace_hash_ipmodify_disable(ops);
2940
2941 if (ftrace_hash_rec_disable(ops, 1))
2942 command |= FTRACE_UPDATE_CALLS;
2943
2944 ops->flags &= ~FTRACE_OPS_FL_ENABLED;
2945
2946 if (saved_ftrace_func != ftrace_trace_function) {
2947 saved_ftrace_func = ftrace_trace_function;
2948 command |= FTRACE_UPDATE_TRACE_FUNC;
2949 }
2950
2951 if (!command || !ftrace_enabled)
2952 goto out;
2953
2954 /*
2955 * If the ops uses a trampoline, then it needs to be
2956 * tested first on update.
2957 */
2958 ops->flags |= FTRACE_OPS_FL_REMOVING;
2959 removed_ops = ops;
2960
2961 /* The trampoline logic checks the old hashes */
2962 ops->old_hash.filter_hash = ops->func_hash->filter_hash;
2963 ops->old_hash.notrace_hash = ops->func_hash->notrace_hash;
2964
2965 ftrace_run_update_code(command);
2966
2967 /*
2968 * If there's no more ops registered with ftrace, run a
2969 * sanity check to make sure all rec flags are cleared.
2970 */
2971 if (rcu_dereference_protected(ftrace_ops_list,
2972 lockdep_is_held(&ftrace_lock)) == &ftrace_list_end) {
2973 struct ftrace_page *pg;
2974 struct dyn_ftrace *rec;
2975
2976 do_for_each_ftrace_rec(pg, rec) {
2977 if (FTRACE_WARN_ON_ONCE(rec->flags & ~FTRACE_FL_DISABLED))
2978 pr_warn(" %pS flags:%lx\n",
2979 (void *)rec->ip, rec->flags);
2980 } while_for_each_ftrace_rec();
2981 }
2982
2983 ops->old_hash.filter_hash = NULL;
2984 ops->old_hash.notrace_hash = NULL;
2985
2986 removed_ops = NULL;
2987 ops->flags &= ~FTRACE_OPS_FL_REMOVING;
2988
2989 out:
2990 /*
2991 * Dynamic ops may be freed, we must make sure that all
2992 * callers are done before leaving this function.
2993 * The same goes for freeing the per_cpu data of the per_cpu
2994 * ops.
2995 */
2996 if (ops->flags & FTRACE_OPS_FL_DYNAMIC) {
2997 /*
2998 * We need to do a hard force of sched synchronization.
2999 * This is because we use preempt_disable() to do RCU, but
3000 * the function tracers can be called where RCU is not watching
3001 * (like before user_exit()). We can not rely on the RCU
3002 * infrastructure to do the synchronization, thus we must do it
3003 * ourselves.
3004 */
3005 synchronize_rcu_tasks_rude();
3006
3007 /*
3008 * When the kernel is preemptive, tasks can be preempted
3009 * while on a ftrace trampoline. Just scheduling a task on
3010 * a CPU is not good enough to flush them. Calling
3011 * synchornize_rcu_tasks() will wait for those tasks to
3012 * execute and either schedule voluntarily or enter user space.
3013 */
3014 if (IS_ENABLED(CONFIG_PREEMPTION))
3015 synchronize_rcu_tasks();
3016
3017 ftrace_trampoline_free(ops);
3018 }
3019
3020 return 0;
3021 }
3022
ftrace_startup_sysctl(void)3023 static void ftrace_startup_sysctl(void)
3024 {
3025 int command;
3026
3027 if (unlikely(ftrace_disabled))
3028 return;
3029
3030 /* Force update next time */
3031 saved_ftrace_func = NULL;
3032 /* ftrace_start_up is true if we want ftrace running */
3033 if (ftrace_start_up) {
3034 command = FTRACE_UPDATE_CALLS;
3035 if (ftrace_graph_active)
3036 command |= FTRACE_START_FUNC_RET;
3037 ftrace_startup_enable(command);
3038 }
3039 }
3040
ftrace_shutdown_sysctl(void)3041 static void ftrace_shutdown_sysctl(void)
3042 {
3043 int command;
3044
3045 if (unlikely(ftrace_disabled))
3046 return;
3047
3048 /* ftrace_start_up is true if ftrace is running */
3049 if (ftrace_start_up) {
3050 command = FTRACE_DISABLE_CALLS;
3051 if (ftrace_graph_active)
3052 command |= FTRACE_STOP_FUNC_RET;
3053 ftrace_run_update_code(command);
3054 }
3055 }
3056
3057 static u64 ftrace_update_time;
3058 unsigned long ftrace_update_tot_cnt;
3059 unsigned long ftrace_number_of_pages;
3060 unsigned long ftrace_number_of_groups;
3061
ops_traces_mod(struct ftrace_ops * ops)3062 static inline int ops_traces_mod(struct ftrace_ops *ops)
3063 {
3064 /*
3065 * Filter_hash being empty will default to trace module.
3066 * But notrace hash requires a test of individual module functions.
3067 */
3068 return ftrace_hash_empty(ops->func_hash->filter_hash) &&
3069 ftrace_hash_empty(ops->func_hash->notrace_hash);
3070 }
3071
3072 /*
3073 * Check if the current ops references the record.
3074 *
3075 * If the ops traces all functions, then it was already accounted for.
3076 * If the ops does not trace the current record function, skip it.
3077 * If the ops ignores the function via notrace filter, skip it.
3078 */
3079 static inline bool
ops_references_rec(struct ftrace_ops * ops,struct dyn_ftrace * rec)3080 ops_references_rec(struct ftrace_ops *ops, struct dyn_ftrace *rec)
3081 {
3082 /* If ops isn't enabled, ignore it */
3083 if (!(ops->flags & FTRACE_OPS_FL_ENABLED))
3084 return false;
3085
3086 /* If ops traces all then it includes this function */
3087 if (ops_traces_mod(ops))
3088 return true;
3089
3090 /* The function must be in the filter */
3091 if (!ftrace_hash_empty(ops->func_hash->filter_hash) &&
3092 !__ftrace_lookup_ip(ops->func_hash->filter_hash, rec->ip))
3093 return false;
3094
3095 /* If in notrace hash, we ignore it too */
3096 if (ftrace_lookup_ip(ops->func_hash->notrace_hash, rec->ip))
3097 return false;
3098
3099 return true;
3100 }
3101
ftrace_update_code(struct module * mod,struct ftrace_page * new_pgs)3102 static int ftrace_update_code(struct module *mod, struct ftrace_page *new_pgs)
3103 {
3104 struct ftrace_page *pg;
3105 struct dyn_ftrace *p;
3106 u64 start, stop;
3107 unsigned long update_cnt = 0;
3108 unsigned long rec_flags = 0;
3109 int i;
3110
3111 start = ftrace_now(raw_smp_processor_id());
3112
3113 /*
3114 * When a module is loaded, this function is called to convert
3115 * the calls to mcount in its text to nops, and also to create
3116 * an entry in the ftrace data. Now, if ftrace is activated
3117 * after this call, but before the module sets its text to
3118 * read-only, the modification of enabling ftrace can fail if
3119 * the read-only is done while ftrace is converting the calls.
3120 * To prevent this, the module's records are set as disabled
3121 * and will be enabled after the call to set the module's text
3122 * to read-only.
3123 */
3124 if (mod)
3125 rec_flags |= FTRACE_FL_DISABLED;
3126
3127 for (pg = new_pgs; pg; pg = pg->next) {
3128
3129 for (i = 0; i < pg->index; i++) {
3130
3131 /* If something went wrong, bail without enabling anything */
3132 if (unlikely(ftrace_disabled))
3133 return -1;
3134
3135 p = &pg->records[i];
3136 p->flags = rec_flags;
3137
3138 /*
3139 * Do the initial record conversion from mcount jump
3140 * to the NOP instructions.
3141 */
3142 if (!__is_defined(CC_USING_NOP_MCOUNT) &&
3143 !ftrace_nop_initialize(mod, p))
3144 break;
3145
3146 update_cnt++;
3147 }
3148 }
3149
3150 stop = ftrace_now(raw_smp_processor_id());
3151 ftrace_update_time = stop - start;
3152 ftrace_update_tot_cnt += update_cnt;
3153
3154 return 0;
3155 }
3156
ftrace_allocate_records(struct ftrace_page * pg,int count)3157 static int ftrace_allocate_records(struct ftrace_page *pg, int count)
3158 {
3159 int order;
3160 int pages;
3161 int cnt;
3162
3163 if (WARN_ON(!count))
3164 return -EINVAL;
3165
3166 pages = DIV_ROUND_UP(count, ENTRIES_PER_PAGE);
3167 order = get_count_order(pages);
3168
3169 /*
3170 * We want to fill as much as possible. No more than a page
3171 * may be empty.
3172 */
3173 if (!is_power_of_2(pages))
3174 order--;
3175
3176 again:
3177 pg->records = (void *)__get_free_pages(GFP_KERNEL | __GFP_ZERO, order);
3178
3179 if (!pg->records) {
3180 /* if we can't allocate this size, try something smaller */
3181 if (!order)
3182 return -ENOMEM;
3183 order--;
3184 goto again;
3185 }
3186
3187 ftrace_number_of_pages += 1 << order;
3188 ftrace_number_of_groups++;
3189
3190 cnt = (PAGE_SIZE << order) / ENTRY_SIZE;
3191 pg->order = order;
3192
3193 if (cnt > count)
3194 cnt = count;
3195
3196 return cnt;
3197 }
3198
ftrace_free_pages(struct ftrace_page * pages)3199 static void ftrace_free_pages(struct ftrace_page *pages)
3200 {
3201 struct ftrace_page *pg = pages;
3202
3203 while (pg) {
3204 if (pg->records) {
3205 free_pages((unsigned long)pg->records, pg->order);
3206 ftrace_number_of_pages -= 1 << pg->order;
3207 }
3208 pages = pg->next;
3209 kfree(pg);
3210 pg = pages;
3211 ftrace_number_of_groups--;
3212 }
3213 }
3214
3215 static struct ftrace_page *
ftrace_allocate_pages(unsigned long num_to_init)3216 ftrace_allocate_pages(unsigned long num_to_init)
3217 {
3218 struct ftrace_page *start_pg;
3219 struct ftrace_page *pg;
3220 int cnt;
3221
3222 if (!num_to_init)
3223 return NULL;
3224
3225 start_pg = pg = kzalloc(sizeof(*pg), GFP_KERNEL);
3226 if (!pg)
3227 return NULL;
3228
3229 /*
3230 * Try to allocate as much as possible in one continues
3231 * location that fills in all of the space. We want to
3232 * waste as little space as possible.
3233 */
3234 for (;;) {
3235 cnt = ftrace_allocate_records(pg, num_to_init);
3236 if (cnt < 0)
3237 goto free_pages;
3238
3239 num_to_init -= cnt;
3240 if (!num_to_init)
3241 break;
3242
3243 pg->next = kzalloc(sizeof(*pg), GFP_KERNEL);
3244 if (!pg->next)
3245 goto free_pages;
3246
3247 pg = pg->next;
3248 }
3249
3250 return start_pg;
3251
3252 free_pages:
3253 ftrace_free_pages(start_pg);
3254 pr_info("ftrace: FAILED to allocate memory for functions\n");
3255 return NULL;
3256 }
3257
3258 #define FTRACE_BUFF_MAX (KSYM_SYMBOL_LEN+4) /* room for wildcards */
3259
3260 struct ftrace_iterator {
3261 loff_t pos;
3262 loff_t func_pos;
3263 loff_t mod_pos;
3264 struct ftrace_page *pg;
3265 struct dyn_ftrace *func;
3266 struct ftrace_func_probe *probe;
3267 struct ftrace_func_entry *probe_entry;
3268 struct trace_parser parser;
3269 struct ftrace_hash *hash;
3270 struct ftrace_ops *ops;
3271 struct trace_array *tr;
3272 struct list_head *mod_list;
3273 int pidx;
3274 int idx;
3275 unsigned flags;
3276 };
3277
3278 static void *
t_probe_next(struct seq_file * m,loff_t * pos)3279 t_probe_next(struct seq_file *m, loff_t *pos)
3280 {
3281 struct ftrace_iterator *iter = m->private;
3282 struct trace_array *tr = iter->ops->private;
3283 struct list_head *func_probes;
3284 struct ftrace_hash *hash;
3285 struct list_head *next;
3286 struct hlist_node *hnd = NULL;
3287 struct hlist_head *hhd;
3288 int size;
3289
3290 (*pos)++;
3291 iter->pos = *pos;
3292
3293 if (!tr)
3294 return NULL;
3295
3296 func_probes = &tr->func_probes;
3297 if (list_empty(func_probes))
3298 return NULL;
3299
3300 if (!iter->probe) {
3301 next = func_probes->next;
3302 iter->probe = list_entry(next, struct ftrace_func_probe, list);
3303 }
3304
3305 if (iter->probe_entry)
3306 hnd = &iter->probe_entry->hlist;
3307
3308 hash = iter->probe->ops.func_hash->filter_hash;
3309
3310 /*
3311 * A probe being registered may temporarily have an empty hash
3312 * and it's at the end of the func_probes list.
3313 */
3314 if (!hash || hash == EMPTY_HASH)
3315 return NULL;
3316
3317 size = 1 << hash->size_bits;
3318
3319 retry:
3320 if (iter->pidx >= size) {
3321 if (iter->probe->list.next == func_probes)
3322 return NULL;
3323 next = iter->probe->list.next;
3324 iter->probe = list_entry(next, struct ftrace_func_probe, list);
3325 hash = iter->probe->ops.func_hash->filter_hash;
3326 size = 1 << hash->size_bits;
3327 iter->pidx = 0;
3328 }
3329
3330 hhd = &hash->buckets[iter->pidx];
3331
3332 if (hlist_empty(hhd)) {
3333 iter->pidx++;
3334 hnd = NULL;
3335 goto retry;
3336 }
3337
3338 if (!hnd)
3339 hnd = hhd->first;
3340 else {
3341 hnd = hnd->next;
3342 if (!hnd) {
3343 iter->pidx++;
3344 goto retry;
3345 }
3346 }
3347
3348 if (WARN_ON_ONCE(!hnd))
3349 return NULL;
3350
3351 iter->probe_entry = hlist_entry(hnd, struct ftrace_func_entry, hlist);
3352
3353 return iter;
3354 }
3355
t_probe_start(struct seq_file * m,loff_t * pos)3356 static void *t_probe_start(struct seq_file *m, loff_t *pos)
3357 {
3358 struct ftrace_iterator *iter = m->private;
3359 void *p = NULL;
3360 loff_t l;
3361
3362 if (!(iter->flags & FTRACE_ITER_DO_PROBES))
3363 return NULL;
3364
3365 if (iter->mod_pos > *pos)
3366 return NULL;
3367
3368 iter->probe = NULL;
3369 iter->probe_entry = NULL;
3370 iter->pidx = 0;
3371 for (l = 0; l <= (*pos - iter->mod_pos); ) {
3372 p = t_probe_next(m, &l);
3373 if (!p)
3374 break;
3375 }
3376 if (!p)
3377 return NULL;
3378
3379 /* Only set this if we have an item */
3380 iter->flags |= FTRACE_ITER_PROBE;
3381
3382 return iter;
3383 }
3384
3385 static int
t_probe_show(struct seq_file * m,struct ftrace_iterator * iter)3386 t_probe_show(struct seq_file *m, struct ftrace_iterator *iter)
3387 {
3388 struct ftrace_func_entry *probe_entry;
3389 struct ftrace_probe_ops *probe_ops;
3390 struct ftrace_func_probe *probe;
3391
3392 probe = iter->probe;
3393 probe_entry = iter->probe_entry;
3394
3395 if (WARN_ON_ONCE(!probe || !probe_entry))
3396 return -EIO;
3397
3398 probe_ops = probe->probe_ops;
3399
3400 if (probe_ops->print)
3401 return probe_ops->print(m, probe_entry->ip, probe_ops, probe->data);
3402
3403 seq_printf(m, "%ps:%ps\n", (void *)probe_entry->ip,
3404 (void *)probe_ops->func);
3405
3406 return 0;
3407 }
3408
3409 static void *
t_mod_next(struct seq_file * m,loff_t * pos)3410 t_mod_next(struct seq_file *m, loff_t *pos)
3411 {
3412 struct ftrace_iterator *iter = m->private;
3413 struct trace_array *tr = iter->tr;
3414
3415 (*pos)++;
3416 iter->pos = *pos;
3417
3418 iter->mod_list = iter->mod_list->next;
3419
3420 if (iter->mod_list == &tr->mod_trace ||
3421 iter->mod_list == &tr->mod_notrace) {
3422 iter->flags &= ~FTRACE_ITER_MOD;
3423 return NULL;
3424 }
3425
3426 iter->mod_pos = *pos;
3427
3428 return iter;
3429 }
3430
t_mod_start(struct seq_file * m,loff_t * pos)3431 static void *t_mod_start(struct seq_file *m, loff_t *pos)
3432 {
3433 struct ftrace_iterator *iter = m->private;
3434 void *p = NULL;
3435 loff_t l;
3436
3437 if (iter->func_pos > *pos)
3438 return NULL;
3439
3440 iter->mod_pos = iter->func_pos;
3441
3442 /* probes are only available if tr is set */
3443 if (!iter->tr)
3444 return NULL;
3445
3446 for (l = 0; l <= (*pos - iter->func_pos); ) {
3447 p = t_mod_next(m, &l);
3448 if (!p)
3449 break;
3450 }
3451 if (!p) {
3452 iter->flags &= ~FTRACE_ITER_MOD;
3453 return t_probe_start(m, pos);
3454 }
3455
3456 /* Only set this if we have an item */
3457 iter->flags |= FTRACE_ITER_MOD;
3458
3459 return iter;
3460 }
3461
3462 static int
t_mod_show(struct seq_file * m,struct ftrace_iterator * iter)3463 t_mod_show(struct seq_file *m, struct ftrace_iterator *iter)
3464 {
3465 struct ftrace_mod_load *ftrace_mod;
3466 struct trace_array *tr = iter->tr;
3467
3468 if (WARN_ON_ONCE(!iter->mod_list) ||
3469 iter->mod_list == &tr->mod_trace ||
3470 iter->mod_list == &tr->mod_notrace)
3471 return -EIO;
3472
3473 ftrace_mod = list_entry(iter->mod_list, struct ftrace_mod_load, list);
3474
3475 if (ftrace_mod->func)
3476 seq_printf(m, "%s", ftrace_mod->func);
3477 else
3478 seq_putc(m, '*');
3479
3480 seq_printf(m, ":mod:%s\n", ftrace_mod->module);
3481
3482 return 0;
3483 }
3484
3485 static void *
t_func_next(struct seq_file * m,loff_t * pos)3486 t_func_next(struct seq_file *m, loff_t *pos)
3487 {
3488 struct ftrace_iterator *iter = m->private;
3489 struct dyn_ftrace *rec = NULL;
3490
3491 (*pos)++;
3492
3493 retry:
3494 if (iter->idx >= iter->pg->index) {
3495 if (iter->pg->next) {
3496 iter->pg = iter->pg->next;
3497 iter->idx = 0;
3498 goto retry;
3499 }
3500 } else {
3501 rec = &iter->pg->records[iter->idx++];
3502 if (((iter->flags & (FTRACE_ITER_FILTER | FTRACE_ITER_NOTRACE)) &&
3503 !ftrace_lookup_ip(iter->hash, rec->ip)) ||
3504
3505 ((iter->flags & FTRACE_ITER_ENABLED) &&
3506 !(rec->flags & FTRACE_FL_ENABLED))) {
3507
3508 rec = NULL;
3509 goto retry;
3510 }
3511 }
3512
3513 if (!rec)
3514 return NULL;
3515
3516 iter->pos = iter->func_pos = *pos;
3517 iter->func = rec;
3518
3519 return iter;
3520 }
3521
3522 static void *
t_next(struct seq_file * m,void * v,loff_t * pos)3523 t_next(struct seq_file *m, void *v, loff_t *pos)
3524 {
3525 struct ftrace_iterator *iter = m->private;
3526 loff_t l = *pos; /* t_probe_start() must use original pos */
3527 void *ret;
3528
3529 if (unlikely(ftrace_disabled))
3530 return NULL;
3531
3532 if (iter->flags & FTRACE_ITER_PROBE)
3533 return t_probe_next(m, pos);
3534
3535 if (iter->flags & FTRACE_ITER_MOD)
3536 return t_mod_next(m, pos);
3537
3538 if (iter->flags & FTRACE_ITER_PRINTALL) {
3539 /* next must increment pos, and t_probe_start does not */
3540 (*pos)++;
3541 return t_mod_start(m, &l);
3542 }
3543
3544 ret = t_func_next(m, pos);
3545
3546 if (!ret)
3547 return t_mod_start(m, &l);
3548
3549 return ret;
3550 }
3551
reset_iter_read(struct ftrace_iterator * iter)3552 static void reset_iter_read(struct ftrace_iterator *iter)
3553 {
3554 iter->pos = 0;
3555 iter->func_pos = 0;
3556 iter->flags &= ~(FTRACE_ITER_PRINTALL | FTRACE_ITER_PROBE | FTRACE_ITER_MOD);
3557 }
3558
t_start(struct seq_file * m,loff_t * pos)3559 static void *t_start(struct seq_file *m, loff_t *pos)
3560 {
3561 struct ftrace_iterator *iter = m->private;
3562 void *p = NULL;
3563 loff_t l;
3564
3565 mutex_lock(&ftrace_lock);
3566
3567 if (unlikely(ftrace_disabled))
3568 return NULL;
3569
3570 /*
3571 * If an lseek was done, then reset and start from beginning.
3572 */
3573 if (*pos < iter->pos)
3574 reset_iter_read(iter);
3575
3576 /*
3577 * For set_ftrace_filter reading, if we have the filter
3578 * off, we can short cut and just print out that all
3579 * functions are enabled.
3580 */
3581 if ((iter->flags & (FTRACE_ITER_FILTER | FTRACE_ITER_NOTRACE)) &&
3582 ftrace_hash_empty(iter->hash)) {
3583 iter->func_pos = 1; /* Account for the message */
3584 if (*pos > 0)
3585 return t_mod_start(m, pos);
3586 iter->flags |= FTRACE_ITER_PRINTALL;
3587 /* reset in case of seek/pread */
3588 iter->flags &= ~FTRACE_ITER_PROBE;
3589 return iter;
3590 }
3591
3592 if (iter->flags & FTRACE_ITER_MOD)
3593 return t_mod_start(m, pos);
3594
3595 /*
3596 * Unfortunately, we need to restart at ftrace_pages_start
3597 * every time we let go of the ftrace_mutex. This is because
3598 * those pointers can change without the lock.
3599 */
3600 iter->pg = ftrace_pages_start;
3601 iter->idx = 0;
3602 for (l = 0; l <= *pos; ) {
3603 p = t_func_next(m, &l);
3604 if (!p)
3605 break;
3606 }
3607
3608 if (!p)
3609 return t_mod_start(m, pos);
3610
3611 return iter;
3612 }
3613
t_stop(struct seq_file * m,void * p)3614 static void t_stop(struct seq_file *m, void *p)
3615 {
3616 mutex_unlock(&ftrace_lock);
3617 }
3618
3619 void * __weak
arch_ftrace_trampoline_func(struct ftrace_ops * ops,struct dyn_ftrace * rec)3620 arch_ftrace_trampoline_func(struct ftrace_ops *ops, struct dyn_ftrace *rec)
3621 {
3622 return NULL;
3623 }
3624
add_trampoline_func(struct seq_file * m,struct ftrace_ops * ops,struct dyn_ftrace * rec)3625 static void add_trampoline_func(struct seq_file *m, struct ftrace_ops *ops,
3626 struct dyn_ftrace *rec)
3627 {
3628 void *ptr;
3629
3630 ptr = arch_ftrace_trampoline_func(ops, rec);
3631 if (ptr)
3632 seq_printf(m, " ->%pS", ptr);
3633 }
3634
t_show(struct seq_file * m,void * v)3635 static int t_show(struct seq_file *m, void *v)
3636 {
3637 struct ftrace_iterator *iter = m->private;
3638 struct dyn_ftrace *rec;
3639
3640 if (iter->flags & FTRACE_ITER_PROBE)
3641 return t_probe_show(m, iter);
3642
3643 if (iter->flags & FTRACE_ITER_MOD)
3644 return t_mod_show(m, iter);
3645
3646 if (iter->flags & FTRACE_ITER_PRINTALL) {
3647 if (iter->flags & FTRACE_ITER_NOTRACE)
3648 seq_puts(m, "#### no functions disabled ####\n");
3649 else
3650 seq_puts(m, "#### all functions enabled ####\n");
3651 return 0;
3652 }
3653
3654 rec = iter->func;
3655
3656 if (!rec)
3657 return 0;
3658
3659 seq_printf(m, "%ps", (void *)rec->ip);
3660 if (iter->flags & FTRACE_ITER_ENABLED) {
3661 struct ftrace_ops *ops;
3662
3663 seq_printf(m, " (%ld)%s%s%s",
3664 ftrace_rec_count(rec),
3665 rec->flags & FTRACE_FL_REGS ? " R" : " ",
3666 rec->flags & FTRACE_FL_IPMODIFY ? " I" : " ",
3667 rec->flags & FTRACE_FL_DIRECT ? " D" : " ");
3668 if (rec->flags & FTRACE_FL_TRAMP_EN) {
3669 ops = ftrace_find_tramp_ops_any(rec);
3670 if (ops) {
3671 do {
3672 seq_printf(m, "\ttramp: %pS (%pS)",
3673 (void *)ops->trampoline,
3674 (void *)ops->func);
3675 add_trampoline_func(m, ops, rec);
3676 ops = ftrace_find_tramp_ops_next(rec, ops);
3677 } while (ops);
3678 } else
3679 seq_puts(m, "\ttramp: ERROR!");
3680 } else {
3681 add_trampoline_func(m, NULL, rec);
3682 }
3683 if (rec->flags & FTRACE_FL_DIRECT) {
3684 unsigned long direct;
3685
3686 direct = ftrace_find_rec_direct(rec->ip);
3687 if (direct)
3688 seq_printf(m, "\n\tdirect-->%pS", (void *)direct);
3689 }
3690 }
3691
3692 seq_putc(m, '\n');
3693
3694 return 0;
3695 }
3696
3697 static const struct seq_operations show_ftrace_seq_ops = {
3698 .start = t_start,
3699 .next = t_next,
3700 .stop = t_stop,
3701 .show = t_show,
3702 };
3703
3704 static int
ftrace_avail_open(struct inode * inode,struct file * file)3705 ftrace_avail_open(struct inode *inode, struct file *file)
3706 {
3707 struct ftrace_iterator *iter;
3708 int ret;
3709
3710 ret = security_locked_down(LOCKDOWN_TRACEFS);
3711 if (ret)
3712 return ret;
3713
3714 if (unlikely(ftrace_disabled))
3715 return -ENODEV;
3716
3717 iter = __seq_open_private(file, &show_ftrace_seq_ops, sizeof(*iter));
3718 if (!iter)
3719 return -ENOMEM;
3720
3721 iter->pg = ftrace_pages_start;
3722 iter->ops = &global_ops;
3723
3724 return 0;
3725 }
3726
3727 static int
ftrace_enabled_open(struct inode * inode,struct file * file)3728 ftrace_enabled_open(struct inode *inode, struct file *file)
3729 {
3730 struct ftrace_iterator *iter;
3731
3732 /*
3733 * This shows us what functions are currently being
3734 * traced and by what. Not sure if we want lockdown
3735 * to hide such critical information for an admin.
3736 * Although, perhaps it can show information we don't
3737 * want people to see, but if something is tracing
3738 * something, we probably want to know about it.
3739 */
3740
3741 iter = __seq_open_private(file, &show_ftrace_seq_ops, sizeof(*iter));
3742 if (!iter)
3743 return -ENOMEM;
3744
3745 iter->pg = ftrace_pages_start;
3746 iter->flags = FTRACE_ITER_ENABLED;
3747 iter->ops = &global_ops;
3748
3749 return 0;
3750 }
3751
3752 /**
3753 * ftrace_regex_open - initialize function tracer filter files
3754 * @ops: The ftrace_ops that hold the hash filters
3755 * @flag: The type of filter to process
3756 * @inode: The inode, usually passed in to your open routine
3757 * @file: The file, usually passed in to your open routine
3758 *
3759 * ftrace_regex_open() initializes the filter files for the
3760 * @ops. Depending on @flag it may process the filter hash or
3761 * the notrace hash of @ops. With this called from the open
3762 * routine, you can use ftrace_filter_write() for the write
3763 * routine if @flag has FTRACE_ITER_FILTER set, or
3764 * ftrace_notrace_write() if @flag has FTRACE_ITER_NOTRACE set.
3765 * tracing_lseek() should be used as the lseek routine, and
3766 * release must call ftrace_regex_release().
3767 */
3768 int
ftrace_regex_open(struct ftrace_ops * ops,int flag,struct inode * inode,struct file * file)3769 ftrace_regex_open(struct ftrace_ops *ops, int flag,
3770 struct inode *inode, struct file *file)
3771 {
3772 struct ftrace_iterator *iter;
3773 struct ftrace_hash *hash;
3774 struct list_head *mod_head;
3775 struct trace_array *tr = ops->private;
3776 int ret = -ENOMEM;
3777
3778 ftrace_ops_init(ops);
3779
3780 if (unlikely(ftrace_disabled))
3781 return -ENODEV;
3782
3783 if (tracing_check_open_get_tr(tr))
3784 return -ENODEV;
3785
3786 iter = kzalloc(sizeof(*iter), GFP_KERNEL);
3787 if (!iter)
3788 goto out;
3789
3790 if (trace_parser_get_init(&iter->parser, FTRACE_BUFF_MAX))
3791 goto out;
3792
3793 iter->ops = ops;
3794 iter->flags = flag;
3795 iter->tr = tr;
3796
3797 mutex_lock(&ops->func_hash->regex_lock);
3798
3799 if (flag & FTRACE_ITER_NOTRACE) {
3800 hash = ops->func_hash->notrace_hash;
3801 mod_head = tr ? &tr->mod_notrace : NULL;
3802 } else {
3803 hash = ops->func_hash->filter_hash;
3804 mod_head = tr ? &tr->mod_trace : NULL;
3805 }
3806
3807 iter->mod_list = mod_head;
3808
3809 if (file->f_mode & FMODE_WRITE) {
3810 const int size_bits = FTRACE_HASH_DEFAULT_BITS;
3811
3812 if (file->f_flags & O_TRUNC) {
3813 iter->hash = alloc_ftrace_hash(size_bits);
3814 clear_ftrace_mod_list(mod_head);
3815 } else {
3816 iter->hash = alloc_and_copy_ftrace_hash(size_bits, hash);
3817 }
3818
3819 if (!iter->hash) {
3820 trace_parser_put(&iter->parser);
3821 goto out_unlock;
3822 }
3823 } else
3824 iter->hash = hash;
3825
3826 ret = 0;
3827
3828 if (file->f_mode & FMODE_READ) {
3829 iter->pg = ftrace_pages_start;
3830
3831 ret = seq_open(file, &show_ftrace_seq_ops);
3832 if (!ret) {
3833 struct seq_file *m = file->private_data;
3834 m->private = iter;
3835 } else {
3836 /* Failed */
3837 free_ftrace_hash(iter->hash);
3838 trace_parser_put(&iter->parser);
3839 }
3840 } else
3841 file->private_data = iter;
3842
3843 out_unlock:
3844 mutex_unlock(&ops->func_hash->regex_lock);
3845
3846 out:
3847 if (ret) {
3848 kfree(iter);
3849 if (tr)
3850 trace_array_put(tr);
3851 }
3852
3853 return ret;
3854 }
3855
3856 static int
ftrace_filter_open(struct inode * inode,struct file * file)3857 ftrace_filter_open(struct inode *inode, struct file *file)
3858 {
3859 struct ftrace_ops *ops = inode->i_private;
3860
3861 /* Checks for tracefs lockdown */
3862 return ftrace_regex_open(ops,
3863 FTRACE_ITER_FILTER | FTRACE_ITER_DO_PROBES,
3864 inode, file);
3865 }
3866
3867 static int
ftrace_notrace_open(struct inode * inode,struct file * file)3868 ftrace_notrace_open(struct inode *inode, struct file *file)
3869 {
3870 struct ftrace_ops *ops = inode->i_private;
3871
3872 /* Checks for tracefs lockdown */
3873 return ftrace_regex_open(ops, FTRACE_ITER_NOTRACE,
3874 inode, file);
3875 }
3876
3877 /* Type for quick search ftrace basic regexes (globs) from filter_parse_regex */
3878 struct ftrace_glob {
3879 char *search;
3880 unsigned len;
3881 int type;
3882 };
3883
3884 /*
3885 * If symbols in an architecture don't correspond exactly to the user-visible
3886 * name of what they represent, it is possible to define this function to
3887 * perform the necessary adjustments.
3888 */
arch_ftrace_match_adjust(char * str,const char * search)3889 char * __weak arch_ftrace_match_adjust(char *str, const char *search)
3890 {
3891 return str;
3892 }
3893
ftrace_match(char * str,struct ftrace_glob * g)3894 static int ftrace_match(char *str, struct ftrace_glob *g)
3895 {
3896 int matched = 0;
3897 int slen;
3898
3899 str = arch_ftrace_match_adjust(str, g->search);
3900
3901 switch (g->type) {
3902 case MATCH_FULL:
3903 if (strcmp(str, g->search) == 0)
3904 matched = 1;
3905 break;
3906 case MATCH_FRONT_ONLY:
3907 if (strncmp(str, g->search, g->len) == 0)
3908 matched = 1;
3909 break;
3910 case MATCH_MIDDLE_ONLY:
3911 if (strstr(str, g->search))
3912 matched = 1;
3913 break;
3914 case MATCH_END_ONLY:
3915 slen = strlen(str);
3916 if (slen >= g->len &&
3917 memcmp(str + slen - g->len, g->search, g->len) == 0)
3918 matched = 1;
3919 break;
3920 case MATCH_GLOB:
3921 if (glob_match(g->search, str))
3922 matched = 1;
3923 break;
3924 }
3925
3926 return matched;
3927 }
3928
3929 static int
enter_record(struct ftrace_hash * hash,struct dyn_ftrace * rec,int clear_filter)3930 enter_record(struct ftrace_hash *hash, struct dyn_ftrace *rec, int clear_filter)
3931 {
3932 struct ftrace_func_entry *entry;
3933 int ret = 0;
3934
3935 entry = ftrace_lookup_ip(hash, rec->ip);
3936 if (clear_filter) {
3937 /* Do nothing if it doesn't exist */
3938 if (!entry)
3939 return 0;
3940
3941 free_hash_entry(hash, entry);
3942 } else {
3943 /* Do nothing if it exists */
3944 if (entry)
3945 return 0;
3946
3947 ret = add_hash_entry(hash, rec->ip);
3948 }
3949 return ret;
3950 }
3951
3952 static int
add_rec_by_index(struct ftrace_hash * hash,struct ftrace_glob * func_g,int clear_filter)3953 add_rec_by_index(struct ftrace_hash *hash, struct ftrace_glob *func_g,
3954 int clear_filter)
3955 {
3956 long index = simple_strtoul(func_g->search, NULL, 0);
3957 struct ftrace_page *pg;
3958 struct dyn_ftrace *rec;
3959
3960 /* The index starts at 1 */
3961 if (--index < 0)
3962 return 0;
3963
3964 do_for_each_ftrace_rec(pg, rec) {
3965 if (pg->index <= index) {
3966 index -= pg->index;
3967 /* this is a double loop, break goes to the next page */
3968 break;
3969 }
3970 rec = &pg->records[index];
3971 enter_record(hash, rec, clear_filter);
3972 return 1;
3973 } while_for_each_ftrace_rec();
3974 return 0;
3975 }
3976
3977 static int
ftrace_match_record(struct dyn_ftrace * rec,struct ftrace_glob * func_g,struct ftrace_glob * mod_g,int exclude_mod)3978 ftrace_match_record(struct dyn_ftrace *rec, struct ftrace_glob *func_g,
3979 struct ftrace_glob *mod_g, int exclude_mod)
3980 {
3981 char str[KSYM_SYMBOL_LEN];
3982 char *modname;
3983
3984 kallsyms_lookup(rec->ip, NULL, NULL, &modname, str);
3985
3986 if (mod_g) {
3987 int mod_matches = (modname) ? ftrace_match(modname, mod_g) : 0;
3988
3989 /* blank module name to match all modules */
3990 if (!mod_g->len) {
3991 /* blank module globbing: modname xor exclude_mod */
3992 if (!exclude_mod != !modname)
3993 goto func_match;
3994 return 0;
3995 }
3996
3997 /*
3998 * exclude_mod is set to trace everything but the given
3999 * module. If it is set and the module matches, then
4000 * return 0. If it is not set, and the module doesn't match
4001 * also return 0. Otherwise, check the function to see if
4002 * that matches.
4003 */
4004 if (!mod_matches == !exclude_mod)
4005 return 0;
4006 func_match:
4007 /* blank search means to match all funcs in the mod */
4008 if (!func_g->len)
4009 return 1;
4010 }
4011
4012 return ftrace_match(str, func_g);
4013 }
4014
4015 static int
match_records(struct ftrace_hash * hash,char * func,int len,char * mod)4016 match_records(struct ftrace_hash *hash, char *func, int len, char *mod)
4017 {
4018 struct ftrace_page *pg;
4019 struct dyn_ftrace *rec;
4020 struct ftrace_glob func_g = { .type = MATCH_FULL };
4021 struct ftrace_glob mod_g = { .type = MATCH_FULL };
4022 struct ftrace_glob *mod_match = (mod) ? &mod_g : NULL;
4023 int exclude_mod = 0;
4024 int found = 0;
4025 int ret;
4026 int clear_filter = 0;
4027
4028 if (func) {
4029 func_g.type = filter_parse_regex(func, len, &func_g.search,
4030 &clear_filter);
4031 func_g.len = strlen(func_g.search);
4032 }
4033
4034 if (mod) {
4035 mod_g.type = filter_parse_regex(mod, strlen(mod),
4036 &mod_g.search, &exclude_mod);
4037 mod_g.len = strlen(mod_g.search);
4038 }
4039
4040 mutex_lock(&ftrace_lock);
4041
4042 if (unlikely(ftrace_disabled))
4043 goto out_unlock;
4044
4045 if (func_g.type == MATCH_INDEX) {
4046 found = add_rec_by_index(hash, &func_g, clear_filter);
4047 goto out_unlock;
4048 }
4049
4050 do_for_each_ftrace_rec(pg, rec) {
4051
4052 if (rec->flags & FTRACE_FL_DISABLED)
4053 continue;
4054
4055 if (ftrace_match_record(rec, &func_g, mod_match, exclude_mod)) {
4056 ret = enter_record(hash, rec, clear_filter);
4057 if (ret < 0) {
4058 found = ret;
4059 goto out_unlock;
4060 }
4061 found = 1;
4062 }
4063 } while_for_each_ftrace_rec();
4064 out_unlock:
4065 mutex_unlock(&ftrace_lock);
4066
4067 return found;
4068 }
4069
4070 static int
ftrace_match_records(struct ftrace_hash * hash,char * buff,int len)4071 ftrace_match_records(struct ftrace_hash *hash, char *buff, int len)
4072 {
4073 return match_records(hash, buff, len, NULL);
4074 }
4075
ftrace_ops_update_code(struct ftrace_ops * ops,struct ftrace_ops_hash * old_hash)4076 static void ftrace_ops_update_code(struct ftrace_ops *ops,
4077 struct ftrace_ops_hash *old_hash)
4078 {
4079 struct ftrace_ops *op;
4080
4081 if (!ftrace_enabled)
4082 return;
4083
4084 if (ops->flags & FTRACE_OPS_FL_ENABLED) {
4085 ftrace_run_modify_code(ops, FTRACE_UPDATE_CALLS, old_hash);
4086 return;
4087 }
4088
4089 /*
4090 * If this is the shared global_ops filter, then we need to
4091 * check if there is another ops that shares it, is enabled.
4092 * If so, we still need to run the modify code.
4093 */
4094 if (ops->func_hash != &global_ops.local_hash)
4095 return;
4096
4097 do_for_each_ftrace_op(op, ftrace_ops_list) {
4098 if (op->func_hash == &global_ops.local_hash &&
4099 op->flags & FTRACE_OPS_FL_ENABLED) {
4100 ftrace_run_modify_code(op, FTRACE_UPDATE_CALLS, old_hash);
4101 /* Only need to do this once */
4102 return;
4103 }
4104 } while_for_each_ftrace_op(op);
4105 }
4106
ftrace_hash_move_and_update_ops(struct ftrace_ops * ops,struct ftrace_hash ** orig_hash,struct ftrace_hash * hash,int enable)4107 static int ftrace_hash_move_and_update_ops(struct ftrace_ops *ops,
4108 struct ftrace_hash **orig_hash,
4109 struct ftrace_hash *hash,
4110 int enable)
4111 {
4112 struct ftrace_ops_hash old_hash_ops;
4113 struct ftrace_hash *old_hash;
4114 int ret;
4115
4116 old_hash = *orig_hash;
4117 old_hash_ops.filter_hash = ops->func_hash->filter_hash;
4118 old_hash_ops.notrace_hash = ops->func_hash->notrace_hash;
4119 ret = ftrace_hash_move(ops, enable, orig_hash, hash);
4120 if (!ret) {
4121 ftrace_ops_update_code(ops, &old_hash_ops);
4122 free_ftrace_hash_rcu(old_hash);
4123 }
4124 return ret;
4125 }
4126
module_exists(const char * module)4127 static bool module_exists(const char *module)
4128 {
4129 /* All modules have the symbol __this_module */
4130 static const char this_mod[] = "__this_module";
4131 char modname[MAX_PARAM_PREFIX_LEN + sizeof(this_mod) + 2];
4132 unsigned long val;
4133 int n;
4134
4135 n = snprintf(modname, sizeof(modname), "%s:%s", module, this_mod);
4136
4137 if (n > sizeof(modname) - 1)
4138 return false;
4139
4140 val = module_kallsyms_lookup_name(modname);
4141 return val != 0;
4142 }
4143
cache_mod(struct trace_array * tr,const char * func,char * module,int enable)4144 static int cache_mod(struct trace_array *tr,
4145 const char *func, char *module, int enable)
4146 {
4147 struct ftrace_mod_load *ftrace_mod, *n;
4148 struct list_head *head = enable ? &tr->mod_trace : &tr->mod_notrace;
4149 int ret;
4150
4151 mutex_lock(&ftrace_lock);
4152
4153 /* We do not cache inverse filters */
4154 if (func[0] == '!') {
4155 func++;
4156 ret = -EINVAL;
4157
4158 /* Look to remove this hash */
4159 list_for_each_entry_safe(ftrace_mod, n, head, list) {
4160 if (strcmp(ftrace_mod->module, module) != 0)
4161 continue;
4162
4163 /* no func matches all */
4164 if (strcmp(func, "*") == 0 ||
4165 (ftrace_mod->func &&
4166 strcmp(ftrace_mod->func, func) == 0)) {
4167 ret = 0;
4168 free_ftrace_mod(ftrace_mod);
4169 continue;
4170 }
4171 }
4172 goto out;
4173 }
4174
4175 ret = -EINVAL;
4176 /* We only care about modules that have not been loaded yet */
4177 if (module_exists(module))
4178 goto out;
4179
4180 /* Save this string off, and execute it when the module is loaded */
4181 ret = ftrace_add_mod(tr, func, module, enable);
4182 out:
4183 mutex_unlock(&ftrace_lock);
4184
4185 return ret;
4186 }
4187
4188 static int
4189 ftrace_set_regex(struct ftrace_ops *ops, unsigned char *buf, int len,
4190 int reset, int enable);
4191
4192 #ifdef CONFIG_MODULES
process_mod_list(struct list_head * head,struct ftrace_ops * ops,char * mod,bool enable)4193 static void process_mod_list(struct list_head *head, struct ftrace_ops *ops,
4194 char *mod, bool enable)
4195 {
4196 struct ftrace_mod_load *ftrace_mod, *n;
4197 struct ftrace_hash **orig_hash, *new_hash;
4198 LIST_HEAD(process_mods);
4199 char *func;
4200 int ret;
4201
4202 mutex_lock(&ops->func_hash->regex_lock);
4203
4204 if (enable)
4205 orig_hash = &ops->func_hash->filter_hash;
4206 else
4207 orig_hash = &ops->func_hash->notrace_hash;
4208
4209 new_hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS,
4210 *orig_hash);
4211 if (!new_hash)
4212 goto out; /* warn? */
4213
4214 mutex_lock(&ftrace_lock);
4215
4216 list_for_each_entry_safe(ftrace_mod, n, head, list) {
4217
4218 if (strcmp(ftrace_mod->module, mod) != 0)
4219 continue;
4220
4221 if (ftrace_mod->func)
4222 func = kstrdup(ftrace_mod->func, GFP_KERNEL);
4223 else
4224 func = kstrdup("*", GFP_KERNEL);
4225
4226 if (!func) /* warn? */
4227 continue;
4228
4229 list_del(&ftrace_mod->list);
4230 list_add(&ftrace_mod->list, &process_mods);
4231
4232 /* Use the newly allocated func, as it may be "*" */
4233 kfree(ftrace_mod->func);
4234 ftrace_mod->func = func;
4235 }
4236
4237 mutex_unlock(&ftrace_lock);
4238
4239 list_for_each_entry_safe(ftrace_mod, n, &process_mods, list) {
4240
4241 func = ftrace_mod->func;
4242
4243 /* Grabs ftrace_lock, which is why we have this extra step */
4244 match_records(new_hash, func, strlen(func), mod);
4245 free_ftrace_mod(ftrace_mod);
4246 }
4247
4248 if (enable && list_empty(head))
4249 new_hash->flags &= ~FTRACE_HASH_FL_MOD;
4250
4251 mutex_lock(&ftrace_lock);
4252
4253 ret = ftrace_hash_move_and_update_ops(ops, orig_hash,
4254 new_hash, enable);
4255 mutex_unlock(&ftrace_lock);
4256
4257 out:
4258 mutex_unlock(&ops->func_hash->regex_lock);
4259
4260 free_ftrace_hash(new_hash);
4261 }
4262
process_cached_mods(const char * mod_name)4263 static void process_cached_mods(const char *mod_name)
4264 {
4265 struct trace_array *tr;
4266 char *mod;
4267
4268 mod = kstrdup(mod_name, GFP_KERNEL);
4269 if (!mod)
4270 return;
4271
4272 mutex_lock(&trace_types_lock);
4273 list_for_each_entry(tr, &ftrace_trace_arrays, list) {
4274 if (!list_empty(&tr->mod_trace))
4275 process_mod_list(&tr->mod_trace, tr->ops, mod, true);
4276 if (!list_empty(&tr->mod_notrace))
4277 process_mod_list(&tr->mod_notrace, tr->ops, mod, false);
4278 }
4279 mutex_unlock(&trace_types_lock);
4280
4281 kfree(mod);
4282 }
4283 #endif
4284
4285 /*
4286 * We register the module command as a template to show others how
4287 * to register the a command as well.
4288 */
4289
4290 static int
ftrace_mod_callback(struct trace_array * tr,struct ftrace_hash * hash,char * func_orig,char * cmd,char * module,int enable)4291 ftrace_mod_callback(struct trace_array *tr, struct ftrace_hash *hash,
4292 char *func_orig, char *cmd, char *module, int enable)
4293 {
4294 char *func;
4295 int ret;
4296
4297 /* match_records() modifies func, and we need the original */
4298 func = kstrdup(func_orig, GFP_KERNEL);
4299 if (!func)
4300 return -ENOMEM;
4301
4302 /*
4303 * cmd == 'mod' because we only registered this func
4304 * for the 'mod' ftrace_func_command.
4305 * But if you register one func with multiple commands,
4306 * you can tell which command was used by the cmd
4307 * parameter.
4308 */
4309 ret = match_records(hash, func, strlen(func), module);
4310 kfree(func);
4311
4312 if (!ret)
4313 return cache_mod(tr, func_orig, module, enable);
4314 if (ret < 0)
4315 return ret;
4316 return 0;
4317 }
4318
4319 static struct ftrace_func_command ftrace_mod_cmd = {
4320 .name = "mod",
4321 .func = ftrace_mod_callback,
4322 };
4323
ftrace_mod_cmd_init(void)4324 static int __init ftrace_mod_cmd_init(void)
4325 {
4326 return register_ftrace_command(&ftrace_mod_cmd);
4327 }
4328 core_initcall(ftrace_mod_cmd_init);
4329
function_trace_probe_call(unsigned long ip,unsigned long parent_ip,struct ftrace_ops * op,struct pt_regs * pt_regs)4330 static void function_trace_probe_call(unsigned long ip, unsigned long parent_ip,
4331 struct ftrace_ops *op, struct pt_regs *pt_regs)
4332 {
4333 struct ftrace_probe_ops *probe_ops;
4334 struct ftrace_func_probe *probe;
4335
4336 probe = container_of(op, struct ftrace_func_probe, ops);
4337 probe_ops = probe->probe_ops;
4338
4339 /*
4340 * Disable preemption for these calls to prevent a RCU grace
4341 * period. This syncs the hash iteration and freeing of items
4342 * on the hash. rcu_read_lock is too dangerous here.
4343 */
4344 preempt_disable_notrace();
4345 probe_ops->func(ip, parent_ip, probe->tr, probe_ops, probe->data);
4346 preempt_enable_notrace();
4347 }
4348
4349 struct ftrace_func_map {
4350 struct ftrace_func_entry entry;
4351 void *data;
4352 };
4353
4354 struct ftrace_func_mapper {
4355 struct ftrace_hash hash;
4356 };
4357
4358 /**
4359 * allocate_ftrace_func_mapper - allocate a new ftrace_func_mapper
4360 *
4361 * Returns a ftrace_func_mapper descriptor that can be used to map ips to data.
4362 */
allocate_ftrace_func_mapper(void)4363 struct ftrace_func_mapper *allocate_ftrace_func_mapper(void)
4364 {
4365 struct ftrace_hash *hash;
4366
4367 /*
4368 * The mapper is simply a ftrace_hash, but since the entries
4369 * in the hash are not ftrace_func_entry type, we define it
4370 * as a separate structure.
4371 */
4372 hash = alloc_ftrace_hash(FTRACE_HASH_DEFAULT_BITS);
4373 return (struct ftrace_func_mapper *)hash;
4374 }
4375
4376 /**
4377 * ftrace_func_mapper_find_ip - Find some data mapped to an ip
4378 * @mapper: The mapper that has the ip maps
4379 * @ip: the instruction pointer to find the data for
4380 *
4381 * Returns the data mapped to @ip if found otherwise NULL. The return
4382 * is actually the address of the mapper data pointer. The address is
4383 * returned for use cases where the data is no bigger than a long, and
4384 * the user can use the data pointer as its data instead of having to
4385 * allocate more memory for the reference.
4386 */
ftrace_func_mapper_find_ip(struct ftrace_func_mapper * mapper,unsigned long ip)4387 void **ftrace_func_mapper_find_ip(struct ftrace_func_mapper *mapper,
4388 unsigned long ip)
4389 {
4390 struct ftrace_func_entry *entry;
4391 struct ftrace_func_map *map;
4392
4393 entry = ftrace_lookup_ip(&mapper->hash, ip);
4394 if (!entry)
4395 return NULL;
4396
4397 map = (struct ftrace_func_map *)entry;
4398 return &map->data;
4399 }
4400
4401 /**
4402 * ftrace_func_mapper_add_ip - Map some data to an ip
4403 * @mapper: The mapper that has the ip maps
4404 * @ip: The instruction pointer address to map @data to
4405 * @data: The data to map to @ip
4406 *
4407 * Returns 0 on success otherwise an error.
4408 */
ftrace_func_mapper_add_ip(struct ftrace_func_mapper * mapper,unsigned long ip,void * data)4409 int ftrace_func_mapper_add_ip(struct ftrace_func_mapper *mapper,
4410 unsigned long ip, void *data)
4411 {
4412 struct ftrace_func_entry *entry;
4413 struct ftrace_func_map *map;
4414
4415 entry = ftrace_lookup_ip(&mapper->hash, ip);
4416 if (entry)
4417 return -EBUSY;
4418
4419 map = kmalloc(sizeof(*map), GFP_KERNEL);
4420 if (!map)
4421 return -ENOMEM;
4422
4423 map->entry.ip = ip;
4424 map->data = data;
4425
4426 __add_hash_entry(&mapper->hash, &map->entry);
4427
4428 return 0;
4429 }
4430
4431 /**
4432 * ftrace_func_mapper_remove_ip - Remove an ip from the mapping
4433 * @mapper: The mapper that has the ip maps
4434 * @ip: The instruction pointer address to remove the data from
4435 *
4436 * Returns the data if it is found, otherwise NULL.
4437 * Note, if the data pointer is used as the data itself, (see
4438 * ftrace_func_mapper_find_ip(), then the return value may be meaningless,
4439 * if the data pointer was set to zero.
4440 */
ftrace_func_mapper_remove_ip(struct ftrace_func_mapper * mapper,unsigned long ip)4441 void *ftrace_func_mapper_remove_ip(struct ftrace_func_mapper *mapper,
4442 unsigned long ip)
4443 {
4444 struct ftrace_func_entry *entry;
4445 struct ftrace_func_map *map;
4446 void *data;
4447
4448 entry = ftrace_lookup_ip(&mapper->hash, ip);
4449 if (!entry)
4450 return NULL;
4451
4452 map = (struct ftrace_func_map *)entry;
4453 data = map->data;
4454
4455 remove_hash_entry(&mapper->hash, entry);
4456 kfree(entry);
4457
4458 return data;
4459 }
4460
4461 /**
4462 * free_ftrace_func_mapper - free a mapping of ips and data
4463 * @mapper: The mapper that has the ip maps
4464 * @free_func: A function to be called on each data item.
4465 *
4466 * This is used to free the function mapper. The @free_func is optional
4467 * and can be used if the data needs to be freed as well.
4468 */
free_ftrace_func_mapper(struct ftrace_func_mapper * mapper,ftrace_mapper_func free_func)4469 void free_ftrace_func_mapper(struct ftrace_func_mapper *mapper,
4470 ftrace_mapper_func free_func)
4471 {
4472 struct ftrace_func_entry *entry;
4473 struct ftrace_func_map *map;
4474 struct hlist_head *hhd;
4475 int size, i;
4476
4477 if (!mapper)
4478 return;
4479
4480 if (free_func && mapper->hash.count) {
4481 size = 1 << mapper->hash.size_bits;
4482 for (i = 0; i < size; i++) {
4483 hhd = &mapper->hash.buckets[i];
4484 hlist_for_each_entry(entry, hhd, hlist) {
4485 map = (struct ftrace_func_map *)entry;
4486 free_func(map);
4487 }
4488 }
4489 }
4490 free_ftrace_hash(&mapper->hash);
4491 }
4492
release_probe(struct ftrace_func_probe * probe)4493 static void release_probe(struct ftrace_func_probe *probe)
4494 {
4495 struct ftrace_probe_ops *probe_ops;
4496
4497 mutex_lock(&ftrace_lock);
4498
4499 WARN_ON(probe->ref <= 0);
4500
4501 /* Subtract the ref that was used to protect this instance */
4502 probe->ref--;
4503
4504 if (!probe->ref) {
4505 probe_ops = probe->probe_ops;
4506 /*
4507 * Sending zero as ip tells probe_ops to free
4508 * the probe->data itself
4509 */
4510 if (probe_ops->free)
4511 probe_ops->free(probe_ops, probe->tr, 0, probe->data);
4512 list_del(&probe->list);
4513 kfree(probe);
4514 }
4515 mutex_unlock(&ftrace_lock);
4516 }
4517
acquire_probe_locked(struct ftrace_func_probe * probe)4518 static void acquire_probe_locked(struct ftrace_func_probe *probe)
4519 {
4520 /*
4521 * Add one ref to keep it from being freed when releasing the
4522 * ftrace_lock mutex.
4523 */
4524 probe->ref++;
4525 }
4526
4527 int
register_ftrace_function_probe(char * glob,struct trace_array * tr,struct ftrace_probe_ops * probe_ops,void * data)4528 register_ftrace_function_probe(char *glob, struct trace_array *tr,
4529 struct ftrace_probe_ops *probe_ops,
4530 void *data)
4531 {
4532 struct ftrace_func_entry *entry;
4533 struct ftrace_func_probe *probe;
4534 struct ftrace_hash **orig_hash;
4535 struct ftrace_hash *old_hash;
4536 struct ftrace_hash *hash;
4537 int count = 0;
4538 int size;
4539 int ret;
4540 int i;
4541
4542 if (WARN_ON(!tr))
4543 return -EINVAL;
4544
4545 /* We do not support '!' for function probes */
4546 if (WARN_ON(glob[0] == '!'))
4547 return -EINVAL;
4548
4549
4550 mutex_lock(&ftrace_lock);
4551 /* Check if the probe_ops is already registered */
4552 list_for_each_entry(probe, &tr->func_probes, list) {
4553 if (probe->probe_ops == probe_ops)
4554 break;
4555 }
4556 if (&probe->list == &tr->func_probes) {
4557 probe = kzalloc(sizeof(*probe), GFP_KERNEL);
4558 if (!probe) {
4559 mutex_unlock(&ftrace_lock);
4560 return -ENOMEM;
4561 }
4562 probe->probe_ops = probe_ops;
4563 probe->ops.func = function_trace_probe_call;
4564 probe->tr = tr;
4565 ftrace_ops_init(&probe->ops);
4566 list_add(&probe->list, &tr->func_probes);
4567 }
4568
4569 acquire_probe_locked(probe);
4570
4571 mutex_unlock(&ftrace_lock);
4572
4573 /*
4574 * Note, there's a small window here that the func_hash->filter_hash
4575 * may be NULL or empty. Need to be careful when reading the loop.
4576 */
4577 mutex_lock(&probe->ops.func_hash->regex_lock);
4578
4579 orig_hash = &probe->ops.func_hash->filter_hash;
4580 old_hash = *orig_hash;
4581 hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, old_hash);
4582
4583 if (!hash) {
4584 ret = -ENOMEM;
4585 goto out;
4586 }
4587
4588 ret = ftrace_match_records(hash, glob, strlen(glob));
4589
4590 /* Nothing found? */
4591 if (!ret)
4592 ret = -EINVAL;
4593
4594 if (ret < 0)
4595 goto out;
4596
4597 size = 1 << hash->size_bits;
4598 for (i = 0; i < size; i++) {
4599 hlist_for_each_entry(entry, &hash->buckets[i], hlist) {
4600 if (ftrace_lookup_ip(old_hash, entry->ip))
4601 continue;
4602 /*
4603 * The caller might want to do something special
4604 * for each function we find. We call the callback
4605 * to give the caller an opportunity to do so.
4606 */
4607 if (probe_ops->init) {
4608 ret = probe_ops->init(probe_ops, tr,
4609 entry->ip, data,
4610 &probe->data);
4611 if (ret < 0) {
4612 if (probe_ops->free && count)
4613 probe_ops->free(probe_ops, tr,
4614 0, probe->data);
4615 probe->data = NULL;
4616 goto out;
4617 }
4618 }
4619 count++;
4620 }
4621 }
4622
4623 mutex_lock(&ftrace_lock);
4624
4625 if (!count) {
4626 /* Nothing was added? */
4627 ret = -EINVAL;
4628 goto out_unlock;
4629 }
4630
4631 ret = ftrace_hash_move_and_update_ops(&probe->ops, orig_hash,
4632 hash, 1);
4633 if (ret < 0)
4634 goto err_unlock;
4635
4636 /* One ref for each new function traced */
4637 probe->ref += count;
4638
4639 if (!(probe->ops.flags & FTRACE_OPS_FL_ENABLED))
4640 ret = ftrace_startup(&probe->ops, 0);
4641
4642 out_unlock:
4643 mutex_unlock(&ftrace_lock);
4644
4645 if (!ret)
4646 ret = count;
4647 out:
4648 mutex_unlock(&probe->ops.func_hash->regex_lock);
4649 free_ftrace_hash(hash);
4650
4651 release_probe(probe);
4652
4653 return ret;
4654
4655 err_unlock:
4656 if (!probe_ops->free || !count)
4657 goto out_unlock;
4658
4659 /* Failed to do the move, need to call the free functions */
4660 for (i = 0; i < size; i++) {
4661 hlist_for_each_entry(entry, &hash->buckets[i], hlist) {
4662 if (ftrace_lookup_ip(old_hash, entry->ip))
4663 continue;
4664 probe_ops->free(probe_ops, tr, entry->ip, probe->data);
4665 }
4666 }
4667 goto out_unlock;
4668 }
4669
4670 int
unregister_ftrace_function_probe_func(char * glob,struct trace_array * tr,struct ftrace_probe_ops * probe_ops)4671 unregister_ftrace_function_probe_func(char *glob, struct trace_array *tr,
4672 struct ftrace_probe_ops *probe_ops)
4673 {
4674 struct ftrace_ops_hash old_hash_ops;
4675 struct ftrace_func_entry *entry;
4676 struct ftrace_func_probe *probe;
4677 struct ftrace_glob func_g;
4678 struct ftrace_hash **orig_hash;
4679 struct ftrace_hash *old_hash;
4680 struct ftrace_hash *hash = NULL;
4681 struct hlist_node *tmp;
4682 struct hlist_head hhd;
4683 char str[KSYM_SYMBOL_LEN];
4684 int count = 0;
4685 int i, ret = -ENODEV;
4686 int size;
4687
4688 if (!glob || !strlen(glob) || !strcmp(glob, "*"))
4689 func_g.search = NULL;
4690 else {
4691 int not;
4692
4693 func_g.type = filter_parse_regex(glob, strlen(glob),
4694 &func_g.search, ¬);
4695 func_g.len = strlen(func_g.search);
4696
4697 /* we do not support '!' for function probes */
4698 if (WARN_ON(not))
4699 return -EINVAL;
4700 }
4701
4702 mutex_lock(&ftrace_lock);
4703 /* Check if the probe_ops is already registered */
4704 list_for_each_entry(probe, &tr->func_probes, list) {
4705 if (probe->probe_ops == probe_ops)
4706 break;
4707 }
4708 if (&probe->list == &tr->func_probes)
4709 goto err_unlock_ftrace;
4710
4711 ret = -EINVAL;
4712 if (!(probe->ops.flags & FTRACE_OPS_FL_INITIALIZED))
4713 goto err_unlock_ftrace;
4714
4715 acquire_probe_locked(probe);
4716
4717 mutex_unlock(&ftrace_lock);
4718
4719 mutex_lock(&probe->ops.func_hash->regex_lock);
4720
4721 orig_hash = &probe->ops.func_hash->filter_hash;
4722 old_hash = *orig_hash;
4723
4724 if (ftrace_hash_empty(old_hash))
4725 goto out_unlock;
4726
4727 old_hash_ops.filter_hash = old_hash;
4728 /* Probes only have filters */
4729 old_hash_ops.notrace_hash = NULL;
4730
4731 ret = -ENOMEM;
4732 hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, old_hash);
4733 if (!hash)
4734 goto out_unlock;
4735
4736 INIT_HLIST_HEAD(&hhd);
4737
4738 size = 1 << hash->size_bits;
4739 for (i = 0; i < size; i++) {
4740 hlist_for_each_entry_safe(entry, tmp, &hash->buckets[i], hlist) {
4741
4742 if (func_g.search) {
4743 kallsyms_lookup(entry->ip, NULL, NULL,
4744 NULL, str);
4745 if (!ftrace_match(str, &func_g))
4746 continue;
4747 }
4748 count++;
4749 remove_hash_entry(hash, entry);
4750 hlist_add_head(&entry->hlist, &hhd);
4751 }
4752 }
4753
4754 /* Nothing found? */
4755 if (!count) {
4756 ret = -EINVAL;
4757 goto out_unlock;
4758 }
4759
4760 mutex_lock(&ftrace_lock);
4761
4762 WARN_ON(probe->ref < count);
4763
4764 probe->ref -= count;
4765
4766 if (ftrace_hash_empty(hash))
4767 ftrace_shutdown(&probe->ops, 0);
4768
4769 ret = ftrace_hash_move_and_update_ops(&probe->ops, orig_hash,
4770 hash, 1);
4771
4772 /* still need to update the function call sites */
4773 if (ftrace_enabled && !ftrace_hash_empty(hash))
4774 ftrace_run_modify_code(&probe->ops, FTRACE_UPDATE_CALLS,
4775 &old_hash_ops);
4776 synchronize_rcu();
4777
4778 hlist_for_each_entry_safe(entry, tmp, &hhd, hlist) {
4779 hlist_del(&entry->hlist);
4780 if (probe_ops->free)
4781 probe_ops->free(probe_ops, tr, entry->ip, probe->data);
4782 kfree(entry);
4783 }
4784 mutex_unlock(&ftrace_lock);
4785
4786 out_unlock:
4787 mutex_unlock(&probe->ops.func_hash->regex_lock);
4788 free_ftrace_hash(hash);
4789
4790 release_probe(probe);
4791
4792 return ret;
4793
4794 err_unlock_ftrace:
4795 mutex_unlock(&ftrace_lock);
4796 return ret;
4797 }
4798
clear_ftrace_function_probes(struct trace_array * tr)4799 void clear_ftrace_function_probes(struct trace_array *tr)
4800 {
4801 struct ftrace_func_probe *probe, *n;
4802
4803 list_for_each_entry_safe(probe, n, &tr->func_probes, list)
4804 unregister_ftrace_function_probe_func(NULL, tr, probe->probe_ops);
4805 }
4806
4807 static LIST_HEAD(ftrace_commands);
4808 static DEFINE_MUTEX(ftrace_cmd_mutex);
4809
4810 /*
4811 * Currently we only register ftrace commands from __init, so mark this
4812 * __init too.
4813 */
register_ftrace_command(struct ftrace_func_command * cmd)4814 __init int register_ftrace_command(struct ftrace_func_command *cmd)
4815 {
4816 struct ftrace_func_command *p;
4817 int ret = 0;
4818
4819 mutex_lock(&ftrace_cmd_mutex);
4820 list_for_each_entry(p, &ftrace_commands, list) {
4821 if (strcmp(cmd->name, p->name) == 0) {
4822 ret = -EBUSY;
4823 goto out_unlock;
4824 }
4825 }
4826 list_add(&cmd->list, &ftrace_commands);
4827 out_unlock:
4828 mutex_unlock(&ftrace_cmd_mutex);
4829
4830 return ret;
4831 }
4832
4833 /*
4834 * Currently we only unregister ftrace commands from __init, so mark
4835 * this __init too.
4836 */
unregister_ftrace_command(struct ftrace_func_command * cmd)4837 __init int unregister_ftrace_command(struct ftrace_func_command *cmd)
4838 {
4839 struct ftrace_func_command *p, *n;
4840 int ret = -ENODEV;
4841
4842 mutex_lock(&ftrace_cmd_mutex);
4843 list_for_each_entry_safe(p, n, &ftrace_commands, list) {
4844 if (strcmp(cmd->name, p->name) == 0) {
4845 ret = 0;
4846 list_del_init(&p->list);
4847 goto out_unlock;
4848 }
4849 }
4850 out_unlock:
4851 mutex_unlock(&ftrace_cmd_mutex);
4852
4853 return ret;
4854 }
4855
ftrace_process_regex(struct ftrace_iterator * iter,char * buff,int len,int enable)4856 static int ftrace_process_regex(struct ftrace_iterator *iter,
4857 char *buff, int len, int enable)
4858 {
4859 struct ftrace_hash *hash = iter->hash;
4860 struct trace_array *tr = iter->ops->private;
4861 char *func, *command, *next = buff;
4862 struct ftrace_func_command *p;
4863 int ret = -EINVAL;
4864
4865 func = strsep(&next, ":");
4866
4867 if (!next) {
4868 ret = ftrace_match_records(hash, func, len);
4869 if (!ret)
4870 ret = -EINVAL;
4871 if (ret < 0)
4872 return ret;
4873 return 0;
4874 }
4875
4876 /* command found */
4877
4878 command = strsep(&next, ":");
4879
4880 mutex_lock(&ftrace_cmd_mutex);
4881 list_for_each_entry(p, &ftrace_commands, list) {
4882 if (strcmp(p->name, command) == 0) {
4883 ret = p->func(tr, hash, func, command, next, enable);
4884 goto out_unlock;
4885 }
4886 }
4887 out_unlock:
4888 mutex_unlock(&ftrace_cmd_mutex);
4889
4890 return ret;
4891 }
4892
4893 static ssize_t
ftrace_regex_write(struct file * file,const char __user * ubuf,size_t cnt,loff_t * ppos,int enable)4894 ftrace_regex_write(struct file *file, const char __user *ubuf,
4895 size_t cnt, loff_t *ppos, int enable)
4896 {
4897 struct ftrace_iterator *iter;
4898 struct trace_parser *parser;
4899 ssize_t ret, read;
4900
4901 if (!cnt)
4902 return 0;
4903
4904 if (file->f_mode & FMODE_READ) {
4905 struct seq_file *m = file->private_data;
4906 iter = m->private;
4907 } else
4908 iter = file->private_data;
4909
4910 if (unlikely(ftrace_disabled))
4911 return -ENODEV;
4912
4913 /* iter->hash is a local copy, so we don't need regex_lock */
4914
4915 parser = &iter->parser;
4916 read = trace_get_user(parser, ubuf, cnt, ppos);
4917
4918 if (read >= 0 && trace_parser_loaded(parser) &&
4919 !trace_parser_cont(parser)) {
4920 ret = ftrace_process_regex(iter, parser->buffer,
4921 parser->idx, enable);
4922 trace_parser_clear(parser);
4923 if (ret < 0)
4924 goto out;
4925 }
4926
4927 ret = read;
4928 out:
4929 return ret;
4930 }
4931
4932 ssize_t
ftrace_filter_write(struct file * file,const char __user * ubuf,size_t cnt,loff_t * ppos)4933 ftrace_filter_write(struct file *file, const char __user *ubuf,
4934 size_t cnt, loff_t *ppos)
4935 {
4936 return ftrace_regex_write(file, ubuf, cnt, ppos, 1);
4937 }
4938
4939 ssize_t
ftrace_notrace_write(struct file * file,const char __user * ubuf,size_t cnt,loff_t * ppos)4940 ftrace_notrace_write(struct file *file, const char __user *ubuf,
4941 size_t cnt, loff_t *ppos)
4942 {
4943 return ftrace_regex_write(file, ubuf, cnt, ppos, 0);
4944 }
4945
4946 static int
ftrace_match_addr(struct ftrace_hash * hash,unsigned long ip,int remove)4947 ftrace_match_addr(struct ftrace_hash *hash, unsigned long ip, int remove)
4948 {
4949 struct ftrace_func_entry *entry;
4950
4951 if (!ftrace_location(ip))
4952 return -EINVAL;
4953
4954 if (remove) {
4955 entry = ftrace_lookup_ip(hash, ip);
4956 if (!entry)
4957 return -ENOENT;
4958 free_hash_entry(hash, entry);
4959 return 0;
4960 }
4961
4962 return add_hash_entry(hash, ip);
4963 }
4964
4965 static int
ftrace_set_hash(struct ftrace_ops * ops,unsigned char * buf,int len,unsigned long ip,int remove,int reset,int enable)4966 ftrace_set_hash(struct ftrace_ops *ops, unsigned char *buf, int len,
4967 unsigned long ip, int remove, int reset, int enable)
4968 {
4969 struct ftrace_hash **orig_hash;
4970 struct ftrace_hash *hash;
4971 int ret;
4972
4973 if (unlikely(ftrace_disabled))
4974 return -ENODEV;
4975
4976 mutex_lock(&ops->func_hash->regex_lock);
4977
4978 if (enable)
4979 orig_hash = &ops->func_hash->filter_hash;
4980 else
4981 orig_hash = &ops->func_hash->notrace_hash;
4982
4983 if (reset)
4984 hash = alloc_ftrace_hash(FTRACE_HASH_DEFAULT_BITS);
4985 else
4986 hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, *orig_hash);
4987
4988 if (!hash) {
4989 ret = -ENOMEM;
4990 goto out_regex_unlock;
4991 }
4992
4993 if (buf && !ftrace_match_records(hash, buf, len)) {
4994 ret = -EINVAL;
4995 goto out_regex_unlock;
4996 }
4997 if (ip) {
4998 ret = ftrace_match_addr(hash, ip, remove);
4999 if (ret < 0)
5000 goto out_regex_unlock;
5001 }
5002
5003 mutex_lock(&ftrace_lock);
5004 ret = ftrace_hash_move_and_update_ops(ops, orig_hash, hash, enable);
5005 mutex_unlock(&ftrace_lock);
5006
5007 out_regex_unlock:
5008 mutex_unlock(&ops->func_hash->regex_lock);
5009
5010 free_ftrace_hash(hash);
5011 return ret;
5012 }
5013
5014 static int
ftrace_set_addr(struct ftrace_ops * ops,unsigned long ip,int remove,int reset,int enable)5015 ftrace_set_addr(struct ftrace_ops *ops, unsigned long ip, int remove,
5016 int reset, int enable)
5017 {
5018 return ftrace_set_hash(ops, NULL, 0, ip, remove, reset, enable);
5019 }
5020
5021 #ifdef CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS
5022
5023 struct ftrace_direct_func {
5024 struct list_head next;
5025 unsigned long addr;
5026 int count;
5027 };
5028
5029 static LIST_HEAD(ftrace_direct_funcs);
5030
5031 /**
5032 * ftrace_find_direct_func - test an address if it is a registered direct caller
5033 * @addr: The address of a registered direct caller
5034 *
5035 * This searches to see if a ftrace direct caller has been registered
5036 * at a specific address, and if so, it returns a descriptor for it.
5037 *
5038 * This can be used by architecture code to see if an address is
5039 * a direct caller (trampoline) attached to a fentry/mcount location.
5040 * This is useful for the function_graph tracer, as it may need to
5041 * do adjustments if it traced a location that also has a direct
5042 * trampoline attached to it.
5043 */
ftrace_find_direct_func(unsigned long addr)5044 struct ftrace_direct_func *ftrace_find_direct_func(unsigned long addr)
5045 {
5046 struct ftrace_direct_func *entry;
5047 bool found = false;
5048
5049 /* May be called by fgraph trampoline (protected by rcu tasks) */
5050 list_for_each_entry_rcu(entry, &ftrace_direct_funcs, next) {
5051 if (entry->addr == addr) {
5052 found = true;
5053 break;
5054 }
5055 }
5056 if (found)
5057 return entry;
5058
5059 return NULL;
5060 }
5061
ftrace_alloc_direct_func(unsigned long addr)5062 static struct ftrace_direct_func *ftrace_alloc_direct_func(unsigned long addr)
5063 {
5064 struct ftrace_direct_func *direct;
5065
5066 direct = kmalloc(sizeof(*direct), GFP_KERNEL);
5067 if (!direct)
5068 return NULL;
5069 direct->addr = addr;
5070 direct->count = 0;
5071 list_add_rcu(&direct->next, &ftrace_direct_funcs);
5072 ftrace_direct_func_count++;
5073 return direct;
5074 }
5075
5076 /**
5077 * register_ftrace_direct - Call a custom trampoline directly
5078 * @ip: The address of the nop at the beginning of a function
5079 * @addr: The address of the trampoline to call at @ip
5080 *
5081 * This is used to connect a direct call from the nop location (@ip)
5082 * at the start of ftrace traced functions. The location that it calls
5083 * (@addr) must be able to handle a direct call, and save the parameters
5084 * of the function being traced, and restore them (or inject new ones
5085 * if needed), before returning.
5086 *
5087 * Returns:
5088 * 0 on success
5089 * -EBUSY - Another direct function is already attached (there can be only one)
5090 * -ENODEV - @ip does not point to a ftrace nop location (or not supported)
5091 * -ENOMEM - There was an allocation failure.
5092 */
register_ftrace_direct(unsigned long ip,unsigned long addr)5093 int register_ftrace_direct(unsigned long ip, unsigned long addr)
5094 {
5095 struct ftrace_direct_func *direct;
5096 struct ftrace_func_entry *entry;
5097 struct ftrace_hash *free_hash = NULL;
5098 struct dyn_ftrace *rec;
5099 int ret = -EBUSY;
5100
5101 mutex_lock(&direct_mutex);
5102
5103 /* See if there's a direct function at @ip already */
5104 if (ftrace_find_rec_direct(ip))
5105 goto out_unlock;
5106
5107 ret = -ENODEV;
5108 rec = lookup_rec(ip, ip);
5109 if (!rec)
5110 goto out_unlock;
5111
5112 /*
5113 * Check if the rec says it has a direct call but we didn't
5114 * find one earlier?
5115 */
5116 if (WARN_ON(rec->flags & FTRACE_FL_DIRECT))
5117 goto out_unlock;
5118
5119 /* Make sure the ip points to the exact record */
5120 if (ip != rec->ip) {
5121 ip = rec->ip;
5122 /* Need to check this ip for a direct. */
5123 if (ftrace_find_rec_direct(ip))
5124 goto out_unlock;
5125 }
5126
5127 ret = -ENOMEM;
5128 if (ftrace_hash_empty(direct_functions) ||
5129 direct_functions->count > 2 * (1 << direct_functions->size_bits)) {
5130 struct ftrace_hash *new_hash;
5131 int size = ftrace_hash_empty(direct_functions) ? 0 :
5132 direct_functions->count + 1;
5133
5134 if (size < 32)
5135 size = 32;
5136
5137 new_hash = dup_hash(direct_functions, size);
5138 if (!new_hash)
5139 goto out_unlock;
5140
5141 free_hash = direct_functions;
5142 direct_functions = new_hash;
5143 }
5144
5145 entry = kmalloc(sizeof(*entry), GFP_KERNEL);
5146 if (!entry)
5147 goto out_unlock;
5148
5149 direct = ftrace_find_direct_func(addr);
5150 if (!direct) {
5151 direct = ftrace_alloc_direct_func(addr);
5152 if (!direct) {
5153 kfree(entry);
5154 goto out_unlock;
5155 }
5156 }
5157
5158 entry->ip = ip;
5159 entry->direct = addr;
5160 __add_hash_entry(direct_functions, entry);
5161
5162 ret = ftrace_set_filter_ip(&direct_ops, ip, 0, 0);
5163
5164 if (!ret && !(direct_ops.flags & FTRACE_OPS_FL_ENABLED)) {
5165 ret = register_ftrace_function(&direct_ops);
5166 if (ret)
5167 ftrace_set_filter_ip(&direct_ops, ip, 1, 0);
5168 }
5169
5170 if (ret) {
5171 remove_hash_entry(direct_functions, entry);
5172 kfree(entry);
5173 if (!direct->count) {
5174 list_del_rcu(&direct->next);
5175 synchronize_rcu_tasks();
5176 kfree(direct);
5177 if (free_hash)
5178 free_ftrace_hash(free_hash);
5179 free_hash = NULL;
5180 ftrace_direct_func_count--;
5181 }
5182 } else {
5183 direct->count++;
5184 }
5185 out_unlock:
5186 mutex_unlock(&direct_mutex);
5187
5188 if (free_hash) {
5189 synchronize_rcu_tasks();
5190 free_ftrace_hash(free_hash);
5191 }
5192
5193 return ret;
5194 }
5195 EXPORT_SYMBOL_GPL(register_ftrace_direct);
5196
find_direct_entry(unsigned long * ip,struct dyn_ftrace ** recp)5197 static struct ftrace_func_entry *find_direct_entry(unsigned long *ip,
5198 struct dyn_ftrace **recp)
5199 {
5200 struct ftrace_func_entry *entry;
5201 struct dyn_ftrace *rec;
5202
5203 rec = lookup_rec(*ip, *ip);
5204 if (!rec)
5205 return NULL;
5206
5207 entry = __ftrace_lookup_ip(direct_functions, rec->ip);
5208 if (!entry) {
5209 WARN_ON(rec->flags & FTRACE_FL_DIRECT);
5210 return NULL;
5211 }
5212
5213 WARN_ON(!(rec->flags & FTRACE_FL_DIRECT));
5214
5215 /* Passed in ip just needs to be on the call site */
5216 *ip = rec->ip;
5217
5218 if (recp)
5219 *recp = rec;
5220
5221 return entry;
5222 }
5223
unregister_ftrace_direct(unsigned long ip,unsigned long addr)5224 int unregister_ftrace_direct(unsigned long ip, unsigned long addr)
5225 {
5226 struct ftrace_direct_func *direct;
5227 struct ftrace_func_entry *entry;
5228 int ret = -ENODEV;
5229
5230 mutex_lock(&direct_mutex);
5231
5232 entry = find_direct_entry(&ip, NULL);
5233 if (!entry)
5234 goto out_unlock;
5235
5236 if (direct_functions->count == 1)
5237 unregister_ftrace_function(&direct_ops);
5238
5239 ret = ftrace_set_filter_ip(&direct_ops, ip, 1, 0);
5240
5241 WARN_ON(ret);
5242
5243 remove_hash_entry(direct_functions, entry);
5244
5245 direct = ftrace_find_direct_func(addr);
5246 if (!WARN_ON(!direct)) {
5247 /* This is the good path (see the ! before WARN) */
5248 direct->count--;
5249 WARN_ON(direct->count < 0);
5250 if (!direct->count) {
5251 list_del_rcu(&direct->next);
5252 synchronize_rcu_tasks();
5253 kfree(direct);
5254 kfree(entry);
5255 ftrace_direct_func_count--;
5256 }
5257 }
5258 out_unlock:
5259 mutex_unlock(&direct_mutex);
5260
5261 return ret;
5262 }
5263 EXPORT_SYMBOL_GPL(unregister_ftrace_direct);
5264
5265 static struct ftrace_ops stub_ops = {
5266 .func = ftrace_stub,
5267 };
5268
5269 /**
5270 * ftrace_modify_direct_caller - modify ftrace nop directly
5271 * @entry: The ftrace hash entry of the direct helper for @rec
5272 * @rec: The record representing the function site to patch
5273 * @old_addr: The location that the site at @rec->ip currently calls
5274 * @new_addr: The location that the site at @rec->ip should call
5275 *
5276 * An architecture may overwrite this function to optimize the
5277 * changing of the direct callback on an ftrace nop location.
5278 * This is called with the ftrace_lock mutex held, and no other
5279 * ftrace callbacks are on the associated record (@rec). Thus,
5280 * it is safe to modify the ftrace record, where it should be
5281 * currently calling @old_addr directly, to call @new_addr.
5282 *
5283 * Safety checks should be made to make sure that the code at
5284 * @rec->ip is currently calling @old_addr. And this must
5285 * also update entry->direct to @new_addr.
5286 */
ftrace_modify_direct_caller(struct ftrace_func_entry * entry,struct dyn_ftrace * rec,unsigned long old_addr,unsigned long new_addr)5287 int __weak ftrace_modify_direct_caller(struct ftrace_func_entry *entry,
5288 struct dyn_ftrace *rec,
5289 unsigned long old_addr,
5290 unsigned long new_addr)
5291 {
5292 unsigned long ip = rec->ip;
5293 int ret;
5294
5295 /*
5296 * The ftrace_lock was used to determine if the record
5297 * had more than one registered user to it. If it did,
5298 * we needed to prevent that from changing to do the quick
5299 * switch. But if it did not (only a direct caller was attached)
5300 * then this function is called. But this function can deal
5301 * with attached callers to the rec that we care about, and
5302 * since this function uses standard ftrace calls that take
5303 * the ftrace_lock mutex, we need to release it.
5304 */
5305 mutex_unlock(&ftrace_lock);
5306
5307 /*
5308 * By setting a stub function at the same address, we force
5309 * the code to call the iterator and the direct_ops helper.
5310 * This means that @ip does not call the direct call, and
5311 * we can simply modify it.
5312 */
5313 ret = ftrace_set_filter_ip(&stub_ops, ip, 0, 0);
5314 if (ret)
5315 goto out_lock;
5316
5317 ret = register_ftrace_function(&stub_ops);
5318 if (ret) {
5319 ftrace_set_filter_ip(&stub_ops, ip, 1, 0);
5320 goto out_lock;
5321 }
5322
5323 entry->direct = new_addr;
5324
5325 /*
5326 * By removing the stub, we put back the direct call, calling
5327 * the @new_addr.
5328 */
5329 unregister_ftrace_function(&stub_ops);
5330 ftrace_set_filter_ip(&stub_ops, ip, 1, 0);
5331
5332 out_lock:
5333 mutex_lock(&ftrace_lock);
5334
5335 return ret;
5336 }
5337
5338 /**
5339 * modify_ftrace_direct - Modify an existing direct call to call something else
5340 * @ip: The instruction pointer to modify
5341 * @old_addr: The address that the current @ip calls directly
5342 * @new_addr: The address that the @ip should call
5343 *
5344 * This modifies a ftrace direct caller at an instruction pointer without
5345 * having to disable it first. The direct call will switch over to the
5346 * @new_addr without missing anything.
5347 *
5348 * Returns: zero on success. Non zero on error, which includes:
5349 * -ENODEV : the @ip given has no direct caller attached
5350 * -EINVAL : the @old_addr does not match the current direct caller
5351 */
modify_ftrace_direct(unsigned long ip,unsigned long old_addr,unsigned long new_addr)5352 int modify_ftrace_direct(unsigned long ip,
5353 unsigned long old_addr, unsigned long new_addr)
5354 {
5355 struct ftrace_direct_func *direct, *new_direct = NULL;
5356 struct ftrace_func_entry *entry;
5357 struct dyn_ftrace *rec;
5358 int ret = -ENODEV;
5359
5360 mutex_lock(&direct_mutex);
5361
5362 mutex_lock(&ftrace_lock);
5363 entry = find_direct_entry(&ip, &rec);
5364 if (!entry)
5365 goto out_unlock;
5366
5367 ret = -EINVAL;
5368 if (entry->direct != old_addr)
5369 goto out_unlock;
5370
5371 direct = ftrace_find_direct_func(old_addr);
5372 if (WARN_ON(!direct))
5373 goto out_unlock;
5374 if (direct->count > 1) {
5375 ret = -ENOMEM;
5376 new_direct = ftrace_alloc_direct_func(new_addr);
5377 if (!new_direct)
5378 goto out_unlock;
5379 direct->count--;
5380 new_direct->count++;
5381 } else {
5382 direct->addr = new_addr;
5383 }
5384
5385 /*
5386 * If there's no other ftrace callback on the rec->ip location,
5387 * then it can be changed directly by the architecture.
5388 * If there is another caller, then we just need to change the
5389 * direct caller helper to point to @new_addr.
5390 */
5391 if (ftrace_rec_count(rec) == 1) {
5392 ret = ftrace_modify_direct_caller(entry, rec, old_addr, new_addr);
5393 } else {
5394 entry->direct = new_addr;
5395 ret = 0;
5396 }
5397
5398 if (ret) {
5399 direct->addr = old_addr;
5400 if (unlikely(new_direct)) {
5401 direct->count++;
5402 list_del_rcu(&new_direct->next);
5403 synchronize_rcu_tasks();
5404 kfree(new_direct);
5405 ftrace_direct_func_count--;
5406 }
5407 }
5408
5409 out_unlock:
5410 mutex_unlock(&ftrace_lock);
5411 mutex_unlock(&direct_mutex);
5412 return ret;
5413 }
5414 EXPORT_SYMBOL_GPL(modify_ftrace_direct);
5415 #endif /* CONFIG_DYNAMIC_FTRACE_WITH_DIRECT_CALLS */
5416
5417 /**
5418 * ftrace_set_filter_ip - set a function to filter on in ftrace by address
5419 * @ops - the ops to set the filter with
5420 * @ip - the address to add to or remove from the filter.
5421 * @remove - non zero to remove the ip from the filter
5422 * @reset - non zero to reset all filters before applying this filter.
5423 *
5424 * Filters denote which functions should be enabled when tracing is enabled
5425 * If @ip is NULL, it failes to update filter.
5426 */
ftrace_set_filter_ip(struct ftrace_ops * ops,unsigned long ip,int remove,int reset)5427 int ftrace_set_filter_ip(struct ftrace_ops *ops, unsigned long ip,
5428 int remove, int reset)
5429 {
5430 ftrace_ops_init(ops);
5431 return ftrace_set_addr(ops, ip, remove, reset, 1);
5432 }
5433 EXPORT_SYMBOL_GPL(ftrace_set_filter_ip);
5434
5435 /**
5436 * ftrace_ops_set_global_filter - setup ops to use global filters
5437 * @ops - the ops which will use the global filters
5438 *
5439 * ftrace users who need global function trace filtering should call this.
5440 * It can set the global filter only if ops were not initialized before.
5441 */
ftrace_ops_set_global_filter(struct ftrace_ops * ops)5442 void ftrace_ops_set_global_filter(struct ftrace_ops *ops)
5443 {
5444 if (ops->flags & FTRACE_OPS_FL_INITIALIZED)
5445 return;
5446
5447 ftrace_ops_init(ops);
5448 ops->func_hash = &global_ops.local_hash;
5449 }
5450 EXPORT_SYMBOL_GPL(ftrace_ops_set_global_filter);
5451
5452 static int
ftrace_set_regex(struct ftrace_ops * ops,unsigned char * buf,int len,int reset,int enable)5453 ftrace_set_regex(struct ftrace_ops *ops, unsigned char *buf, int len,
5454 int reset, int enable)
5455 {
5456 return ftrace_set_hash(ops, buf, len, 0, 0, reset, enable);
5457 }
5458
5459 /**
5460 * ftrace_set_filter - set a function to filter on in ftrace
5461 * @ops - the ops to set the filter with
5462 * @buf - the string that holds the function filter text.
5463 * @len - the length of the string.
5464 * @reset - non zero to reset all filters before applying this filter.
5465 *
5466 * Filters denote which functions should be enabled when tracing is enabled.
5467 * If @buf is NULL and reset is set, all functions will be enabled for tracing.
5468 */
ftrace_set_filter(struct ftrace_ops * ops,unsigned char * buf,int len,int reset)5469 int ftrace_set_filter(struct ftrace_ops *ops, unsigned char *buf,
5470 int len, int reset)
5471 {
5472 ftrace_ops_init(ops);
5473 return ftrace_set_regex(ops, buf, len, reset, 1);
5474 }
5475 EXPORT_SYMBOL_GPL(ftrace_set_filter);
5476
5477 /**
5478 * ftrace_set_notrace - set a function to not trace in ftrace
5479 * @ops - the ops to set the notrace filter with
5480 * @buf - the string that holds the function notrace text.
5481 * @len - the length of the string.
5482 * @reset - non zero to reset all filters before applying this filter.
5483 *
5484 * Notrace Filters denote which functions should not be enabled when tracing
5485 * is enabled. If @buf is NULL and reset is set, all functions will be enabled
5486 * for tracing.
5487 */
ftrace_set_notrace(struct ftrace_ops * ops,unsigned char * buf,int len,int reset)5488 int ftrace_set_notrace(struct ftrace_ops *ops, unsigned char *buf,
5489 int len, int reset)
5490 {
5491 ftrace_ops_init(ops);
5492 return ftrace_set_regex(ops, buf, len, reset, 0);
5493 }
5494 EXPORT_SYMBOL_GPL(ftrace_set_notrace);
5495 /**
5496 * ftrace_set_global_filter - set a function to filter on with global tracers
5497 * @buf - the string that holds the function filter text.
5498 * @len - the length of the string.
5499 * @reset - non zero to reset all filters before applying this filter.
5500 *
5501 * Filters denote which functions should be enabled when tracing is enabled.
5502 * If @buf is NULL and reset is set, all functions will be enabled for tracing.
5503 */
ftrace_set_global_filter(unsigned char * buf,int len,int reset)5504 void ftrace_set_global_filter(unsigned char *buf, int len, int reset)
5505 {
5506 ftrace_set_regex(&global_ops, buf, len, reset, 1);
5507 }
5508 EXPORT_SYMBOL_GPL(ftrace_set_global_filter);
5509
5510 /**
5511 * ftrace_set_global_notrace - set a function to not trace with global tracers
5512 * @buf - the string that holds the function notrace text.
5513 * @len - the length of the string.
5514 * @reset - non zero to reset all filters before applying this filter.
5515 *
5516 * Notrace Filters denote which functions should not be enabled when tracing
5517 * is enabled. If @buf is NULL and reset is set, all functions will be enabled
5518 * for tracing.
5519 */
ftrace_set_global_notrace(unsigned char * buf,int len,int reset)5520 void ftrace_set_global_notrace(unsigned char *buf, int len, int reset)
5521 {
5522 ftrace_set_regex(&global_ops, buf, len, reset, 0);
5523 }
5524 EXPORT_SYMBOL_GPL(ftrace_set_global_notrace);
5525
5526 /*
5527 * command line interface to allow users to set filters on boot up.
5528 */
5529 #define FTRACE_FILTER_SIZE COMMAND_LINE_SIZE
5530 static char ftrace_notrace_buf[FTRACE_FILTER_SIZE] __initdata;
5531 static char ftrace_filter_buf[FTRACE_FILTER_SIZE] __initdata;
5532
5533 /* Used by function selftest to not test if filter is set */
5534 bool ftrace_filter_param __initdata;
5535
set_ftrace_notrace(char * str)5536 static int __init set_ftrace_notrace(char *str)
5537 {
5538 ftrace_filter_param = true;
5539 strlcpy(ftrace_notrace_buf, str, FTRACE_FILTER_SIZE);
5540 return 1;
5541 }
5542 __setup("ftrace_notrace=", set_ftrace_notrace);
5543
set_ftrace_filter(char * str)5544 static int __init set_ftrace_filter(char *str)
5545 {
5546 ftrace_filter_param = true;
5547 strlcpy(ftrace_filter_buf, str, FTRACE_FILTER_SIZE);
5548 return 1;
5549 }
5550 __setup("ftrace_filter=", set_ftrace_filter);
5551
5552 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
5553 static char ftrace_graph_buf[FTRACE_FILTER_SIZE] __initdata;
5554 static char ftrace_graph_notrace_buf[FTRACE_FILTER_SIZE] __initdata;
5555 static int ftrace_graph_set_hash(struct ftrace_hash *hash, char *buffer);
5556
set_graph_function(char * str)5557 static int __init set_graph_function(char *str)
5558 {
5559 strlcpy(ftrace_graph_buf, str, FTRACE_FILTER_SIZE);
5560 return 1;
5561 }
5562 __setup("ftrace_graph_filter=", set_graph_function);
5563
set_graph_notrace_function(char * str)5564 static int __init set_graph_notrace_function(char *str)
5565 {
5566 strlcpy(ftrace_graph_notrace_buf, str, FTRACE_FILTER_SIZE);
5567 return 1;
5568 }
5569 __setup("ftrace_graph_notrace=", set_graph_notrace_function);
5570
set_graph_max_depth_function(char * str)5571 static int __init set_graph_max_depth_function(char *str)
5572 {
5573 if (!str)
5574 return 0;
5575 fgraph_max_depth = simple_strtoul(str, NULL, 0);
5576 return 1;
5577 }
5578 __setup("ftrace_graph_max_depth=", set_graph_max_depth_function);
5579
set_ftrace_early_graph(char * buf,int enable)5580 static void __init set_ftrace_early_graph(char *buf, int enable)
5581 {
5582 int ret;
5583 char *func;
5584 struct ftrace_hash *hash;
5585
5586 hash = alloc_ftrace_hash(FTRACE_HASH_DEFAULT_BITS);
5587 if (MEM_FAIL(!hash, "Failed to allocate hash\n"))
5588 return;
5589
5590 while (buf) {
5591 func = strsep(&buf, ",");
5592 /* we allow only one expression at a time */
5593 ret = ftrace_graph_set_hash(hash, func);
5594 if (ret)
5595 printk(KERN_DEBUG "ftrace: function %s not "
5596 "traceable\n", func);
5597 }
5598
5599 if (enable)
5600 ftrace_graph_hash = hash;
5601 else
5602 ftrace_graph_notrace_hash = hash;
5603 }
5604 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
5605
5606 void __init
ftrace_set_early_filter(struct ftrace_ops * ops,char * buf,int enable)5607 ftrace_set_early_filter(struct ftrace_ops *ops, char *buf, int enable)
5608 {
5609 char *func;
5610
5611 ftrace_ops_init(ops);
5612
5613 while (buf) {
5614 func = strsep(&buf, ",");
5615 ftrace_set_regex(ops, func, strlen(func), 0, enable);
5616 }
5617 }
5618
set_ftrace_early_filters(void)5619 static void __init set_ftrace_early_filters(void)
5620 {
5621 if (ftrace_filter_buf[0])
5622 ftrace_set_early_filter(&global_ops, ftrace_filter_buf, 1);
5623 if (ftrace_notrace_buf[0])
5624 ftrace_set_early_filter(&global_ops, ftrace_notrace_buf, 0);
5625 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
5626 if (ftrace_graph_buf[0])
5627 set_ftrace_early_graph(ftrace_graph_buf, 1);
5628 if (ftrace_graph_notrace_buf[0])
5629 set_ftrace_early_graph(ftrace_graph_notrace_buf, 0);
5630 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
5631 }
5632
ftrace_regex_release(struct inode * inode,struct file * file)5633 int ftrace_regex_release(struct inode *inode, struct file *file)
5634 {
5635 struct seq_file *m = (struct seq_file *)file->private_data;
5636 struct ftrace_iterator *iter;
5637 struct ftrace_hash **orig_hash;
5638 struct trace_parser *parser;
5639 int filter_hash;
5640 int ret;
5641
5642 if (file->f_mode & FMODE_READ) {
5643 iter = m->private;
5644 seq_release(inode, file);
5645 } else
5646 iter = file->private_data;
5647
5648 parser = &iter->parser;
5649 if (trace_parser_loaded(parser)) {
5650 int enable = !(iter->flags & FTRACE_ITER_NOTRACE);
5651
5652 ftrace_process_regex(iter, parser->buffer,
5653 parser->idx, enable);
5654 }
5655
5656 trace_parser_put(parser);
5657
5658 mutex_lock(&iter->ops->func_hash->regex_lock);
5659
5660 if (file->f_mode & FMODE_WRITE) {
5661 filter_hash = !!(iter->flags & FTRACE_ITER_FILTER);
5662
5663 if (filter_hash) {
5664 orig_hash = &iter->ops->func_hash->filter_hash;
5665 if (iter->tr) {
5666 if (list_empty(&iter->tr->mod_trace))
5667 iter->hash->flags &= ~FTRACE_HASH_FL_MOD;
5668 else
5669 iter->hash->flags |= FTRACE_HASH_FL_MOD;
5670 }
5671 } else
5672 orig_hash = &iter->ops->func_hash->notrace_hash;
5673
5674 mutex_lock(&ftrace_lock);
5675 ret = ftrace_hash_move_and_update_ops(iter->ops, orig_hash,
5676 iter->hash, filter_hash);
5677 mutex_unlock(&ftrace_lock);
5678 } else {
5679 /* For read only, the hash is the ops hash */
5680 iter->hash = NULL;
5681 }
5682
5683 mutex_unlock(&iter->ops->func_hash->regex_lock);
5684 free_ftrace_hash(iter->hash);
5685 if (iter->tr)
5686 trace_array_put(iter->tr);
5687 kfree(iter);
5688
5689 return 0;
5690 }
5691
5692 static const struct file_operations ftrace_avail_fops = {
5693 .open = ftrace_avail_open,
5694 .read = seq_read,
5695 .llseek = seq_lseek,
5696 .release = seq_release_private,
5697 };
5698
5699 static const struct file_operations ftrace_enabled_fops = {
5700 .open = ftrace_enabled_open,
5701 .read = seq_read,
5702 .llseek = seq_lseek,
5703 .release = seq_release_private,
5704 };
5705
5706 static const struct file_operations ftrace_filter_fops = {
5707 .open = ftrace_filter_open,
5708 .read = seq_read,
5709 .write = ftrace_filter_write,
5710 .llseek = tracing_lseek,
5711 .release = ftrace_regex_release,
5712 };
5713
5714 static const struct file_operations ftrace_notrace_fops = {
5715 .open = ftrace_notrace_open,
5716 .read = seq_read,
5717 .write = ftrace_notrace_write,
5718 .llseek = tracing_lseek,
5719 .release = ftrace_regex_release,
5720 };
5721
5722 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
5723
5724 static DEFINE_MUTEX(graph_lock);
5725
5726 struct ftrace_hash __rcu *ftrace_graph_hash = EMPTY_HASH;
5727 struct ftrace_hash __rcu *ftrace_graph_notrace_hash = EMPTY_HASH;
5728
5729 enum graph_filter_type {
5730 GRAPH_FILTER_NOTRACE = 0,
5731 GRAPH_FILTER_FUNCTION,
5732 };
5733
5734 #define FTRACE_GRAPH_EMPTY ((void *)1)
5735
5736 struct ftrace_graph_data {
5737 struct ftrace_hash *hash;
5738 struct ftrace_func_entry *entry;
5739 int idx; /* for hash table iteration */
5740 enum graph_filter_type type;
5741 struct ftrace_hash *new_hash;
5742 const struct seq_operations *seq_ops;
5743 struct trace_parser parser;
5744 };
5745
5746 static void *
__g_next(struct seq_file * m,loff_t * pos)5747 __g_next(struct seq_file *m, loff_t *pos)
5748 {
5749 struct ftrace_graph_data *fgd = m->private;
5750 struct ftrace_func_entry *entry = fgd->entry;
5751 struct hlist_head *head;
5752 int i, idx = fgd->idx;
5753
5754 if (*pos >= fgd->hash->count)
5755 return NULL;
5756
5757 if (entry) {
5758 hlist_for_each_entry_continue(entry, hlist) {
5759 fgd->entry = entry;
5760 return entry;
5761 }
5762
5763 idx++;
5764 }
5765
5766 for (i = idx; i < 1 << fgd->hash->size_bits; i++) {
5767 head = &fgd->hash->buckets[i];
5768 hlist_for_each_entry(entry, head, hlist) {
5769 fgd->entry = entry;
5770 fgd->idx = i;
5771 return entry;
5772 }
5773 }
5774 return NULL;
5775 }
5776
5777 static void *
g_next(struct seq_file * m,void * v,loff_t * pos)5778 g_next(struct seq_file *m, void *v, loff_t *pos)
5779 {
5780 (*pos)++;
5781 return __g_next(m, pos);
5782 }
5783
g_start(struct seq_file * m,loff_t * pos)5784 static void *g_start(struct seq_file *m, loff_t *pos)
5785 {
5786 struct ftrace_graph_data *fgd = m->private;
5787
5788 mutex_lock(&graph_lock);
5789
5790 if (fgd->type == GRAPH_FILTER_FUNCTION)
5791 fgd->hash = rcu_dereference_protected(ftrace_graph_hash,
5792 lockdep_is_held(&graph_lock));
5793 else
5794 fgd->hash = rcu_dereference_protected(ftrace_graph_notrace_hash,
5795 lockdep_is_held(&graph_lock));
5796
5797 /* Nothing, tell g_show to print all functions are enabled */
5798 if (ftrace_hash_empty(fgd->hash) && !*pos)
5799 return FTRACE_GRAPH_EMPTY;
5800
5801 fgd->idx = 0;
5802 fgd->entry = NULL;
5803 return __g_next(m, pos);
5804 }
5805
g_stop(struct seq_file * m,void * p)5806 static void g_stop(struct seq_file *m, void *p)
5807 {
5808 mutex_unlock(&graph_lock);
5809 }
5810
g_show(struct seq_file * m,void * v)5811 static int g_show(struct seq_file *m, void *v)
5812 {
5813 struct ftrace_func_entry *entry = v;
5814
5815 if (!entry)
5816 return 0;
5817
5818 if (entry == FTRACE_GRAPH_EMPTY) {
5819 struct ftrace_graph_data *fgd = m->private;
5820
5821 if (fgd->type == GRAPH_FILTER_FUNCTION)
5822 seq_puts(m, "#### all functions enabled ####\n");
5823 else
5824 seq_puts(m, "#### no functions disabled ####\n");
5825 return 0;
5826 }
5827
5828 seq_printf(m, "%ps\n", (void *)entry->ip);
5829
5830 return 0;
5831 }
5832
5833 static const struct seq_operations ftrace_graph_seq_ops = {
5834 .start = g_start,
5835 .next = g_next,
5836 .stop = g_stop,
5837 .show = g_show,
5838 };
5839
5840 static int
__ftrace_graph_open(struct inode * inode,struct file * file,struct ftrace_graph_data * fgd)5841 __ftrace_graph_open(struct inode *inode, struct file *file,
5842 struct ftrace_graph_data *fgd)
5843 {
5844 int ret;
5845 struct ftrace_hash *new_hash = NULL;
5846
5847 ret = security_locked_down(LOCKDOWN_TRACEFS);
5848 if (ret)
5849 return ret;
5850
5851 if (file->f_mode & FMODE_WRITE) {
5852 const int size_bits = FTRACE_HASH_DEFAULT_BITS;
5853
5854 if (trace_parser_get_init(&fgd->parser, FTRACE_BUFF_MAX))
5855 return -ENOMEM;
5856
5857 if (file->f_flags & O_TRUNC)
5858 new_hash = alloc_ftrace_hash(size_bits);
5859 else
5860 new_hash = alloc_and_copy_ftrace_hash(size_bits,
5861 fgd->hash);
5862 if (!new_hash) {
5863 ret = -ENOMEM;
5864 goto out;
5865 }
5866 }
5867
5868 if (file->f_mode & FMODE_READ) {
5869 ret = seq_open(file, &ftrace_graph_seq_ops);
5870 if (!ret) {
5871 struct seq_file *m = file->private_data;
5872 m->private = fgd;
5873 } else {
5874 /* Failed */
5875 free_ftrace_hash(new_hash);
5876 new_hash = NULL;
5877 }
5878 } else
5879 file->private_data = fgd;
5880
5881 out:
5882 if (ret < 0 && file->f_mode & FMODE_WRITE)
5883 trace_parser_put(&fgd->parser);
5884
5885 fgd->new_hash = new_hash;
5886
5887 /*
5888 * All uses of fgd->hash must be taken with the graph_lock
5889 * held. The graph_lock is going to be released, so force
5890 * fgd->hash to be reinitialized when it is taken again.
5891 */
5892 fgd->hash = NULL;
5893
5894 return ret;
5895 }
5896
5897 static int
ftrace_graph_open(struct inode * inode,struct file * file)5898 ftrace_graph_open(struct inode *inode, struct file *file)
5899 {
5900 struct ftrace_graph_data *fgd;
5901 int ret;
5902
5903 if (unlikely(ftrace_disabled))
5904 return -ENODEV;
5905
5906 fgd = kmalloc(sizeof(*fgd), GFP_KERNEL);
5907 if (fgd == NULL)
5908 return -ENOMEM;
5909
5910 mutex_lock(&graph_lock);
5911
5912 fgd->hash = rcu_dereference_protected(ftrace_graph_hash,
5913 lockdep_is_held(&graph_lock));
5914 fgd->type = GRAPH_FILTER_FUNCTION;
5915 fgd->seq_ops = &ftrace_graph_seq_ops;
5916
5917 ret = __ftrace_graph_open(inode, file, fgd);
5918 if (ret < 0)
5919 kfree(fgd);
5920
5921 mutex_unlock(&graph_lock);
5922 return ret;
5923 }
5924
5925 static int
ftrace_graph_notrace_open(struct inode * inode,struct file * file)5926 ftrace_graph_notrace_open(struct inode *inode, struct file *file)
5927 {
5928 struct ftrace_graph_data *fgd;
5929 int ret;
5930
5931 if (unlikely(ftrace_disabled))
5932 return -ENODEV;
5933
5934 fgd = kmalloc(sizeof(*fgd), GFP_KERNEL);
5935 if (fgd == NULL)
5936 return -ENOMEM;
5937
5938 mutex_lock(&graph_lock);
5939
5940 fgd->hash = rcu_dereference_protected(ftrace_graph_notrace_hash,
5941 lockdep_is_held(&graph_lock));
5942 fgd->type = GRAPH_FILTER_NOTRACE;
5943 fgd->seq_ops = &ftrace_graph_seq_ops;
5944
5945 ret = __ftrace_graph_open(inode, file, fgd);
5946 if (ret < 0)
5947 kfree(fgd);
5948
5949 mutex_unlock(&graph_lock);
5950 return ret;
5951 }
5952
5953 static int
ftrace_graph_release(struct inode * inode,struct file * file)5954 ftrace_graph_release(struct inode *inode, struct file *file)
5955 {
5956 struct ftrace_graph_data *fgd;
5957 struct ftrace_hash *old_hash, *new_hash;
5958 struct trace_parser *parser;
5959 int ret = 0;
5960
5961 if (file->f_mode & FMODE_READ) {
5962 struct seq_file *m = file->private_data;
5963
5964 fgd = m->private;
5965 seq_release(inode, file);
5966 } else {
5967 fgd = file->private_data;
5968 }
5969
5970
5971 if (file->f_mode & FMODE_WRITE) {
5972
5973 parser = &fgd->parser;
5974
5975 if (trace_parser_loaded((parser))) {
5976 ret = ftrace_graph_set_hash(fgd->new_hash,
5977 parser->buffer);
5978 }
5979
5980 trace_parser_put(parser);
5981
5982 new_hash = __ftrace_hash_move(fgd->new_hash);
5983 if (!new_hash) {
5984 ret = -ENOMEM;
5985 goto out;
5986 }
5987
5988 mutex_lock(&graph_lock);
5989
5990 if (fgd->type == GRAPH_FILTER_FUNCTION) {
5991 old_hash = rcu_dereference_protected(ftrace_graph_hash,
5992 lockdep_is_held(&graph_lock));
5993 rcu_assign_pointer(ftrace_graph_hash, new_hash);
5994 } else {
5995 old_hash = rcu_dereference_protected(ftrace_graph_notrace_hash,
5996 lockdep_is_held(&graph_lock));
5997 rcu_assign_pointer(ftrace_graph_notrace_hash, new_hash);
5998 }
5999
6000 mutex_unlock(&graph_lock);
6001
6002 /*
6003 * We need to do a hard force of sched synchronization.
6004 * This is because we use preempt_disable() to do RCU, but
6005 * the function tracers can be called where RCU is not watching
6006 * (like before user_exit()). We can not rely on the RCU
6007 * infrastructure to do the synchronization, thus we must do it
6008 * ourselves.
6009 */
6010 synchronize_rcu_tasks_rude();
6011
6012 free_ftrace_hash(old_hash);
6013 }
6014
6015 out:
6016 free_ftrace_hash(fgd->new_hash);
6017 kfree(fgd);
6018
6019 return ret;
6020 }
6021
6022 static int
ftrace_graph_set_hash(struct ftrace_hash * hash,char * buffer)6023 ftrace_graph_set_hash(struct ftrace_hash *hash, char *buffer)
6024 {
6025 struct ftrace_glob func_g;
6026 struct dyn_ftrace *rec;
6027 struct ftrace_page *pg;
6028 struct ftrace_func_entry *entry;
6029 int fail = 1;
6030 int not;
6031
6032 /* decode regex */
6033 func_g.type = filter_parse_regex(buffer, strlen(buffer),
6034 &func_g.search, ¬);
6035
6036 func_g.len = strlen(func_g.search);
6037
6038 mutex_lock(&ftrace_lock);
6039
6040 if (unlikely(ftrace_disabled)) {
6041 mutex_unlock(&ftrace_lock);
6042 return -ENODEV;
6043 }
6044
6045 do_for_each_ftrace_rec(pg, rec) {
6046
6047 if (rec->flags & FTRACE_FL_DISABLED)
6048 continue;
6049
6050 if (ftrace_match_record(rec, &func_g, NULL, 0)) {
6051 entry = ftrace_lookup_ip(hash, rec->ip);
6052
6053 if (!not) {
6054 fail = 0;
6055
6056 if (entry)
6057 continue;
6058 if (add_hash_entry(hash, rec->ip) < 0)
6059 goto out;
6060 } else {
6061 if (entry) {
6062 free_hash_entry(hash, entry);
6063 fail = 0;
6064 }
6065 }
6066 }
6067 } while_for_each_ftrace_rec();
6068 out:
6069 mutex_unlock(&ftrace_lock);
6070
6071 if (fail)
6072 return -EINVAL;
6073
6074 return 0;
6075 }
6076
6077 static ssize_t
ftrace_graph_write(struct file * file,const char __user * ubuf,size_t cnt,loff_t * ppos)6078 ftrace_graph_write(struct file *file, const char __user *ubuf,
6079 size_t cnt, loff_t *ppos)
6080 {
6081 ssize_t read, ret = 0;
6082 struct ftrace_graph_data *fgd = file->private_data;
6083 struct trace_parser *parser;
6084
6085 if (!cnt)
6086 return 0;
6087
6088 /* Read mode uses seq functions */
6089 if (file->f_mode & FMODE_READ) {
6090 struct seq_file *m = file->private_data;
6091 fgd = m->private;
6092 }
6093
6094 parser = &fgd->parser;
6095
6096 read = trace_get_user(parser, ubuf, cnt, ppos);
6097
6098 if (read >= 0 && trace_parser_loaded(parser) &&
6099 !trace_parser_cont(parser)) {
6100
6101 ret = ftrace_graph_set_hash(fgd->new_hash,
6102 parser->buffer);
6103 trace_parser_clear(parser);
6104 }
6105
6106 if (!ret)
6107 ret = read;
6108
6109 return ret;
6110 }
6111
6112 static const struct file_operations ftrace_graph_fops = {
6113 .open = ftrace_graph_open,
6114 .read = seq_read,
6115 .write = ftrace_graph_write,
6116 .llseek = tracing_lseek,
6117 .release = ftrace_graph_release,
6118 };
6119
6120 static const struct file_operations ftrace_graph_notrace_fops = {
6121 .open = ftrace_graph_notrace_open,
6122 .read = seq_read,
6123 .write = ftrace_graph_write,
6124 .llseek = tracing_lseek,
6125 .release = ftrace_graph_release,
6126 };
6127 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
6128
ftrace_create_filter_files(struct ftrace_ops * ops,struct dentry * parent)6129 void ftrace_create_filter_files(struct ftrace_ops *ops,
6130 struct dentry *parent)
6131 {
6132
6133 trace_create_file("set_ftrace_filter", 0644, parent,
6134 ops, &ftrace_filter_fops);
6135
6136 trace_create_file("set_ftrace_notrace", 0644, parent,
6137 ops, &ftrace_notrace_fops);
6138 }
6139
6140 /*
6141 * The name "destroy_filter_files" is really a misnomer. Although
6142 * in the future, it may actually delete the files, but this is
6143 * really intended to make sure the ops passed in are disabled
6144 * and that when this function returns, the caller is free to
6145 * free the ops.
6146 *
6147 * The "destroy" name is only to match the "create" name that this
6148 * should be paired with.
6149 */
ftrace_destroy_filter_files(struct ftrace_ops * ops)6150 void ftrace_destroy_filter_files(struct ftrace_ops *ops)
6151 {
6152 mutex_lock(&ftrace_lock);
6153 if (ops->flags & FTRACE_OPS_FL_ENABLED)
6154 ftrace_shutdown(ops, 0);
6155 ops->flags |= FTRACE_OPS_FL_DELETED;
6156 ftrace_free_filter(ops);
6157 mutex_unlock(&ftrace_lock);
6158 }
6159
ftrace_init_dyn_tracefs(struct dentry * d_tracer)6160 static __init int ftrace_init_dyn_tracefs(struct dentry *d_tracer)
6161 {
6162
6163 trace_create_file("available_filter_functions", 0444,
6164 d_tracer, NULL, &ftrace_avail_fops);
6165
6166 trace_create_file("enabled_functions", 0444,
6167 d_tracer, NULL, &ftrace_enabled_fops);
6168
6169 ftrace_create_filter_files(&global_ops, d_tracer);
6170
6171 #ifdef CONFIG_FUNCTION_GRAPH_TRACER
6172 trace_create_file("set_graph_function", 0644, d_tracer,
6173 NULL,
6174 &ftrace_graph_fops);
6175 trace_create_file("set_graph_notrace", 0644, d_tracer,
6176 NULL,
6177 &ftrace_graph_notrace_fops);
6178 #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
6179
6180 return 0;
6181 }
6182
ftrace_cmp_ips(const void * a,const void * b)6183 static int ftrace_cmp_ips(const void *a, const void *b)
6184 {
6185 const unsigned long *ipa = a;
6186 const unsigned long *ipb = b;
6187
6188 if (*ipa > *ipb)
6189 return 1;
6190 if (*ipa < *ipb)
6191 return -1;
6192 return 0;
6193 }
6194
ftrace_process_locs(struct module * mod,unsigned long * start,unsigned long * end)6195 static int ftrace_process_locs(struct module *mod,
6196 unsigned long *start,
6197 unsigned long *end)
6198 {
6199 struct ftrace_page *pg_unuse = NULL;
6200 struct ftrace_page *start_pg;
6201 struct ftrace_page *pg;
6202 struct dyn_ftrace *rec;
6203 unsigned long skipped = 0;
6204 unsigned long count;
6205 unsigned long *p;
6206 unsigned long addr;
6207 unsigned long flags = 0; /* Shut up gcc */
6208 int ret = -ENOMEM;
6209
6210 count = end - start;
6211
6212 if (!count)
6213 return 0;
6214
6215 sort(start, count, sizeof(*start),
6216 ftrace_cmp_ips, NULL);
6217
6218 start_pg = ftrace_allocate_pages(count);
6219 if (!start_pg)
6220 return -ENOMEM;
6221
6222 mutex_lock(&ftrace_lock);
6223
6224 /*
6225 * Core and each module needs their own pages, as
6226 * modules will free them when they are removed.
6227 * Force a new page to be allocated for modules.
6228 */
6229 if (!mod) {
6230 WARN_ON(ftrace_pages || ftrace_pages_start);
6231 /* First initialization */
6232 ftrace_pages = ftrace_pages_start = start_pg;
6233 } else {
6234 if (!ftrace_pages)
6235 goto out;
6236
6237 if (WARN_ON(ftrace_pages->next)) {
6238 /* Hmm, we have free pages? */
6239 while (ftrace_pages->next)
6240 ftrace_pages = ftrace_pages->next;
6241 }
6242
6243 ftrace_pages->next = start_pg;
6244 }
6245
6246 p = start;
6247 pg = start_pg;
6248 while (p < end) {
6249 unsigned long end_offset;
6250 addr = ftrace_call_adjust(*p++);
6251 /*
6252 * Some architecture linkers will pad between
6253 * the different mcount_loc sections of different
6254 * object files to satisfy alignments.
6255 * Skip any NULL pointers.
6256 */
6257 if (!addr) {
6258 skipped++;
6259 continue;
6260 }
6261
6262 end_offset = (pg->index+1) * sizeof(pg->records[0]);
6263 if (end_offset > PAGE_SIZE << pg->order) {
6264 /* We should have allocated enough */
6265 if (WARN_ON(!pg->next))
6266 break;
6267 pg = pg->next;
6268 }
6269
6270 rec = &pg->records[pg->index++];
6271 rec->ip = addr;
6272 }
6273
6274 if (pg->next) {
6275 pg_unuse = pg->next;
6276 pg->next = NULL;
6277 }
6278
6279 /* Assign the last page to ftrace_pages */
6280 ftrace_pages = pg;
6281
6282 /*
6283 * We only need to disable interrupts on start up
6284 * because we are modifying code that an interrupt
6285 * may execute, and the modification is not atomic.
6286 * But for modules, nothing runs the code we modify
6287 * until we are finished with it, and there's no
6288 * reason to cause large interrupt latencies while we do it.
6289 */
6290 if (!mod)
6291 local_irq_save(flags);
6292 ftrace_update_code(mod, start_pg);
6293 if (!mod)
6294 local_irq_restore(flags);
6295 ret = 0;
6296 out:
6297 mutex_unlock(&ftrace_lock);
6298
6299 /* We should have used all pages unless we skipped some */
6300 if (pg_unuse) {
6301 WARN_ON(!skipped);
6302 ftrace_free_pages(pg_unuse);
6303 }
6304 return ret;
6305 }
6306
6307 struct ftrace_mod_func {
6308 struct list_head list;
6309 char *name;
6310 unsigned long ip;
6311 unsigned int size;
6312 };
6313
6314 struct ftrace_mod_map {
6315 struct rcu_head rcu;
6316 struct list_head list;
6317 struct module *mod;
6318 unsigned long start_addr;
6319 unsigned long end_addr;
6320 struct list_head funcs;
6321 unsigned int num_funcs;
6322 };
6323
ftrace_get_trampoline_kallsym(unsigned int symnum,unsigned long * value,char * type,char * name,char * module_name,int * exported)6324 static int ftrace_get_trampoline_kallsym(unsigned int symnum,
6325 unsigned long *value, char *type,
6326 char *name, char *module_name,
6327 int *exported)
6328 {
6329 struct ftrace_ops *op;
6330
6331 list_for_each_entry_rcu(op, &ftrace_ops_trampoline_list, list) {
6332 if (!op->trampoline || symnum--)
6333 continue;
6334 *value = op->trampoline;
6335 *type = 't';
6336 strlcpy(name, FTRACE_TRAMPOLINE_SYM, KSYM_NAME_LEN);
6337 strlcpy(module_name, FTRACE_TRAMPOLINE_MOD, MODULE_NAME_LEN);
6338 *exported = 0;
6339 return 0;
6340 }
6341
6342 return -ERANGE;
6343 }
6344
6345 #ifdef CONFIG_MODULES
6346
6347 #define next_to_ftrace_page(p) container_of(p, struct ftrace_page, next)
6348
6349 static LIST_HEAD(ftrace_mod_maps);
6350
referenced_filters(struct dyn_ftrace * rec)6351 static int referenced_filters(struct dyn_ftrace *rec)
6352 {
6353 struct ftrace_ops *ops;
6354 int cnt = 0;
6355
6356 for (ops = ftrace_ops_list; ops != &ftrace_list_end; ops = ops->next) {
6357 if (ops_references_rec(ops, rec)) {
6358 if (WARN_ON_ONCE(ops->flags & FTRACE_OPS_FL_DIRECT))
6359 continue;
6360 if (WARN_ON_ONCE(ops->flags & FTRACE_OPS_FL_IPMODIFY))
6361 continue;
6362 cnt++;
6363 if (ops->flags & FTRACE_OPS_FL_SAVE_REGS)
6364 rec->flags |= FTRACE_FL_REGS;
6365 if (cnt == 1 && ops->trampoline)
6366 rec->flags |= FTRACE_FL_TRAMP;
6367 else
6368 rec->flags &= ~FTRACE_FL_TRAMP;
6369 }
6370 }
6371
6372 return cnt;
6373 }
6374
6375 static void
clear_mod_from_hash(struct ftrace_page * pg,struct ftrace_hash * hash)6376 clear_mod_from_hash(struct ftrace_page *pg, struct ftrace_hash *hash)
6377 {
6378 struct ftrace_func_entry *entry;
6379 struct dyn_ftrace *rec;
6380 int i;
6381
6382 if (ftrace_hash_empty(hash))
6383 return;
6384
6385 for (i = 0; i < pg->index; i++) {
6386 rec = &pg->records[i];
6387 entry = __ftrace_lookup_ip(hash, rec->ip);
6388 /*
6389 * Do not allow this rec to match again.
6390 * Yeah, it may waste some memory, but will be removed
6391 * if/when the hash is modified again.
6392 */
6393 if (entry)
6394 entry->ip = 0;
6395 }
6396 }
6397
6398 /* Clear any records from hashs */
clear_mod_from_hashes(struct ftrace_page * pg)6399 static void clear_mod_from_hashes(struct ftrace_page *pg)
6400 {
6401 struct trace_array *tr;
6402
6403 mutex_lock(&trace_types_lock);
6404 list_for_each_entry(tr, &ftrace_trace_arrays, list) {
6405 if (!tr->ops || !tr->ops->func_hash)
6406 continue;
6407 mutex_lock(&tr->ops->func_hash->regex_lock);
6408 clear_mod_from_hash(pg, tr->ops->func_hash->filter_hash);
6409 clear_mod_from_hash(pg, tr->ops->func_hash->notrace_hash);
6410 mutex_unlock(&tr->ops->func_hash->regex_lock);
6411 }
6412 mutex_unlock(&trace_types_lock);
6413 }
6414
ftrace_free_mod_map(struct rcu_head * rcu)6415 static void ftrace_free_mod_map(struct rcu_head *rcu)
6416 {
6417 struct ftrace_mod_map *mod_map = container_of(rcu, struct ftrace_mod_map, rcu);
6418 struct ftrace_mod_func *mod_func;
6419 struct ftrace_mod_func *n;
6420
6421 /* All the contents of mod_map are now not visible to readers */
6422 list_for_each_entry_safe(mod_func, n, &mod_map->funcs, list) {
6423 kfree(mod_func->name);
6424 list_del(&mod_func->list);
6425 kfree(mod_func);
6426 }
6427
6428 kfree(mod_map);
6429 }
6430
ftrace_release_mod(struct module * mod)6431 void ftrace_release_mod(struct module *mod)
6432 {
6433 struct ftrace_mod_map *mod_map;
6434 struct ftrace_mod_map *n;
6435 struct dyn_ftrace *rec;
6436 struct ftrace_page **last_pg;
6437 struct ftrace_page *tmp_page = NULL;
6438 struct ftrace_page *pg;
6439
6440 mutex_lock(&ftrace_lock);
6441
6442 if (ftrace_disabled)
6443 goto out_unlock;
6444
6445 list_for_each_entry_safe(mod_map, n, &ftrace_mod_maps, list) {
6446 if (mod_map->mod == mod) {
6447 list_del_rcu(&mod_map->list);
6448 call_rcu(&mod_map->rcu, ftrace_free_mod_map);
6449 break;
6450 }
6451 }
6452
6453 /*
6454 * Each module has its own ftrace_pages, remove
6455 * them from the list.
6456 */
6457 last_pg = &ftrace_pages_start;
6458 for (pg = ftrace_pages_start; pg; pg = *last_pg) {
6459 rec = &pg->records[0];
6460 if (within_module_core(rec->ip, mod) ||
6461 within_module_init(rec->ip, mod)) {
6462 /*
6463 * As core pages are first, the first
6464 * page should never be a module page.
6465 */
6466 if (WARN_ON(pg == ftrace_pages_start))
6467 goto out_unlock;
6468
6469 /* Check if we are deleting the last page */
6470 if (pg == ftrace_pages)
6471 ftrace_pages = next_to_ftrace_page(last_pg);
6472
6473 ftrace_update_tot_cnt -= pg->index;
6474 *last_pg = pg->next;
6475
6476 pg->next = tmp_page;
6477 tmp_page = pg;
6478 } else
6479 last_pg = &pg->next;
6480 }
6481 out_unlock:
6482 mutex_unlock(&ftrace_lock);
6483
6484 for (pg = tmp_page; pg; pg = tmp_page) {
6485
6486 /* Needs to be called outside of ftrace_lock */
6487 clear_mod_from_hashes(pg);
6488
6489 if (pg->records) {
6490 free_pages((unsigned long)pg->records, pg->order);
6491 ftrace_number_of_pages -= 1 << pg->order;
6492 }
6493 tmp_page = pg->next;
6494 kfree(pg);
6495 ftrace_number_of_groups--;
6496 }
6497 }
6498
ftrace_module_enable(struct module * mod)6499 void ftrace_module_enable(struct module *mod)
6500 {
6501 struct dyn_ftrace *rec;
6502 struct ftrace_page *pg;
6503
6504 mutex_lock(&ftrace_lock);
6505
6506 if (ftrace_disabled)
6507 goto out_unlock;
6508
6509 /*
6510 * If the tracing is enabled, go ahead and enable the record.
6511 *
6512 * The reason not to enable the record immediately is the
6513 * inherent check of ftrace_make_nop/ftrace_make_call for
6514 * correct previous instructions. Making first the NOP
6515 * conversion puts the module to the correct state, thus
6516 * passing the ftrace_make_call check.
6517 *
6518 * We also delay this to after the module code already set the
6519 * text to read-only, as we now need to set it back to read-write
6520 * so that we can modify the text.
6521 */
6522 if (ftrace_start_up)
6523 ftrace_arch_code_modify_prepare();
6524
6525 do_for_each_ftrace_rec(pg, rec) {
6526 int cnt;
6527 /*
6528 * do_for_each_ftrace_rec() is a double loop.
6529 * module text shares the pg. If a record is
6530 * not part of this module, then skip this pg,
6531 * which the "break" will do.
6532 */
6533 if (!within_module_core(rec->ip, mod) &&
6534 !within_module_init(rec->ip, mod))
6535 break;
6536
6537 cnt = 0;
6538
6539 /*
6540 * When adding a module, we need to check if tracers are
6541 * currently enabled and if they are, and can trace this record,
6542 * we need to enable the module functions as well as update the
6543 * reference counts for those function records.
6544 */
6545 if (ftrace_start_up)
6546 cnt += referenced_filters(rec);
6547
6548 rec->flags &= ~FTRACE_FL_DISABLED;
6549 rec->flags += cnt;
6550
6551 if (ftrace_start_up && cnt) {
6552 int failed = __ftrace_replace_code(rec, 1);
6553 if (failed) {
6554 ftrace_bug(failed, rec);
6555 goto out_loop;
6556 }
6557 }
6558
6559 } while_for_each_ftrace_rec();
6560
6561 out_loop:
6562 if (ftrace_start_up)
6563 ftrace_arch_code_modify_post_process();
6564
6565 out_unlock:
6566 mutex_unlock(&ftrace_lock);
6567
6568 process_cached_mods(mod->name);
6569 }
6570
ftrace_module_init(struct module * mod)6571 void ftrace_module_init(struct module *mod)
6572 {
6573 if (ftrace_disabled || !mod->num_ftrace_callsites)
6574 return;
6575
6576 ftrace_process_locs(mod, mod->ftrace_callsites,
6577 mod->ftrace_callsites + mod->num_ftrace_callsites);
6578 }
6579
save_ftrace_mod_rec(struct ftrace_mod_map * mod_map,struct dyn_ftrace * rec)6580 static void save_ftrace_mod_rec(struct ftrace_mod_map *mod_map,
6581 struct dyn_ftrace *rec)
6582 {
6583 struct ftrace_mod_func *mod_func;
6584 unsigned long symsize;
6585 unsigned long offset;
6586 char str[KSYM_SYMBOL_LEN];
6587 char *modname;
6588 const char *ret;
6589
6590 ret = kallsyms_lookup(rec->ip, &symsize, &offset, &modname, str);
6591 if (!ret)
6592 return;
6593
6594 mod_func = kmalloc(sizeof(*mod_func), GFP_KERNEL);
6595 if (!mod_func)
6596 return;
6597
6598 mod_func->name = kstrdup(str, GFP_KERNEL);
6599 if (!mod_func->name) {
6600 kfree(mod_func);
6601 return;
6602 }
6603
6604 mod_func->ip = rec->ip - offset;
6605 mod_func->size = symsize;
6606
6607 mod_map->num_funcs++;
6608
6609 list_add_rcu(&mod_func->list, &mod_map->funcs);
6610 }
6611
6612 static struct ftrace_mod_map *
allocate_ftrace_mod_map(struct module * mod,unsigned long start,unsigned long end)6613 allocate_ftrace_mod_map(struct module *mod,
6614 unsigned long start, unsigned long end)
6615 {
6616 struct ftrace_mod_map *mod_map;
6617
6618 mod_map = kmalloc(sizeof(*mod_map), GFP_KERNEL);
6619 if (!mod_map)
6620 return NULL;
6621
6622 mod_map->mod = mod;
6623 mod_map->start_addr = start;
6624 mod_map->end_addr = end;
6625 mod_map->num_funcs = 0;
6626
6627 INIT_LIST_HEAD_RCU(&mod_map->funcs);
6628
6629 list_add_rcu(&mod_map->list, &ftrace_mod_maps);
6630
6631 return mod_map;
6632 }
6633
6634 static const char *
ftrace_func_address_lookup(struct ftrace_mod_map * mod_map,unsigned long addr,unsigned long * size,unsigned long * off,char * sym)6635 ftrace_func_address_lookup(struct ftrace_mod_map *mod_map,
6636 unsigned long addr, unsigned long *size,
6637 unsigned long *off, char *sym)
6638 {
6639 struct ftrace_mod_func *found_func = NULL;
6640 struct ftrace_mod_func *mod_func;
6641
6642 list_for_each_entry_rcu(mod_func, &mod_map->funcs, list) {
6643 if (addr >= mod_func->ip &&
6644 addr < mod_func->ip + mod_func->size) {
6645 found_func = mod_func;
6646 break;
6647 }
6648 }
6649
6650 if (found_func) {
6651 if (size)
6652 *size = found_func->size;
6653 if (off)
6654 *off = addr - found_func->ip;
6655 if (sym)
6656 strlcpy(sym, found_func->name, KSYM_NAME_LEN);
6657
6658 return found_func->name;
6659 }
6660
6661 return NULL;
6662 }
6663
6664 const char *
ftrace_mod_address_lookup(unsigned long addr,unsigned long * size,unsigned long * off,char ** modname,char * sym)6665 ftrace_mod_address_lookup(unsigned long addr, unsigned long *size,
6666 unsigned long *off, char **modname, char *sym)
6667 {
6668 struct ftrace_mod_map *mod_map;
6669 const char *ret = NULL;
6670
6671 /* mod_map is freed via call_rcu() */
6672 preempt_disable();
6673 list_for_each_entry_rcu(mod_map, &ftrace_mod_maps, list) {
6674 ret = ftrace_func_address_lookup(mod_map, addr, size, off, sym);
6675 if (ret) {
6676 if (modname)
6677 *modname = mod_map->mod->name;
6678 break;
6679 }
6680 }
6681 preempt_enable();
6682
6683 return ret;
6684 }
6685
ftrace_mod_get_kallsym(unsigned int symnum,unsigned long * value,char * type,char * name,char * module_name,int * exported)6686 int ftrace_mod_get_kallsym(unsigned int symnum, unsigned long *value,
6687 char *type, char *name,
6688 char *module_name, int *exported)
6689 {
6690 struct ftrace_mod_map *mod_map;
6691 struct ftrace_mod_func *mod_func;
6692 int ret;
6693
6694 preempt_disable();
6695 list_for_each_entry_rcu(mod_map, &ftrace_mod_maps, list) {
6696
6697 if (symnum >= mod_map->num_funcs) {
6698 symnum -= mod_map->num_funcs;
6699 continue;
6700 }
6701
6702 list_for_each_entry_rcu(mod_func, &mod_map->funcs, list) {
6703 if (symnum > 1) {
6704 symnum--;
6705 continue;
6706 }
6707
6708 *value = mod_func->ip;
6709 *type = 'T';
6710 strlcpy(name, mod_func->name, KSYM_NAME_LEN);
6711 strlcpy(module_name, mod_map->mod->name, MODULE_NAME_LEN);
6712 *exported = 1;
6713 preempt_enable();
6714 return 0;
6715 }
6716 WARN_ON(1);
6717 break;
6718 }
6719 ret = ftrace_get_trampoline_kallsym(symnum, value, type, name,
6720 module_name, exported);
6721 preempt_enable();
6722 return ret;
6723 }
6724
6725 #else
save_ftrace_mod_rec(struct ftrace_mod_map * mod_map,struct dyn_ftrace * rec)6726 static void save_ftrace_mod_rec(struct ftrace_mod_map *mod_map,
6727 struct dyn_ftrace *rec) { }
6728 static inline struct ftrace_mod_map *
allocate_ftrace_mod_map(struct module * mod,unsigned long start,unsigned long end)6729 allocate_ftrace_mod_map(struct module *mod,
6730 unsigned long start, unsigned long end)
6731 {
6732 return NULL;
6733 }
ftrace_mod_get_kallsym(unsigned int symnum,unsigned long * value,char * type,char * name,char * module_name,int * exported)6734 int ftrace_mod_get_kallsym(unsigned int symnum, unsigned long *value,
6735 char *type, char *name, char *module_name,
6736 int *exported)
6737 {
6738 int ret;
6739
6740 preempt_disable();
6741 ret = ftrace_get_trampoline_kallsym(symnum, value, type, name,
6742 module_name, exported);
6743 preempt_enable();
6744 return ret;
6745 }
6746 #endif /* CONFIG_MODULES */
6747
6748 struct ftrace_init_func {
6749 struct list_head list;
6750 unsigned long ip;
6751 };
6752
6753 /* Clear any init ips from hashes */
6754 static void
clear_func_from_hash(struct ftrace_init_func * func,struct ftrace_hash * hash)6755 clear_func_from_hash(struct ftrace_init_func *func, struct ftrace_hash *hash)
6756 {
6757 struct ftrace_func_entry *entry;
6758
6759 entry = ftrace_lookup_ip(hash, func->ip);
6760 /*
6761 * Do not allow this rec to match again.
6762 * Yeah, it may waste some memory, but will be removed
6763 * if/when the hash is modified again.
6764 */
6765 if (entry)
6766 entry->ip = 0;
6767 }
6768
6769 static void
clear_func_from_hashes(struct ftrace_init_func * func)6770 clear_func_from_hashes(struct ftrace_init_func *func)
6771 {
6772 struct trace_array *tr;
6773
6774 mutex_lock(&trace_types_lock);
6775 list_for_each_entry(tr, &ftrace_trace_arrays, list) {
6776 if (!tr->ops || !tr->ops->func_hash)
6777 continue;
6778 mutex_lock(&tr->ops->func_hash->regex_lock);
6779 clear_func_from_hash(func, tr->ops->func_hash->filter_hash);
6780 clear_func_from_hash(func, tr->ops->func_hash->notrace_hash);
6781 mutex_unlock(&tr->ops->func_hash->regex_lock);
6782 }
6783 mutex_unlock(&trace_types_lock);
6784 }
6785
add_to_clear_hash_list(struct list_head * clear_list,struct dyn_ftrace * rec)6786 static void add_to_clear_hash_list(struct list_head *clear_list,
6787 struct dyn_ftrace *rec)
6788 {
6789 struct ftrace_init_func *func;
6790
6791 func = kmalloc(sizeof(*func), GFP_KERNEL);
6792 if (!func) {
6793 MEM_FAIL(1, "alloc failure, ftrace filter could be stale\n");
6794 return;
6795 }
6796
6797 func->ip = rec->ip;
6798 list_add(&func->list, clear_list);
6799 }
6800
ftrace_free_mem(struct module * mod,void * start_ptr,void * end_ptr)6801 void ftrace_free_mem(struct module *mod, void *start_ptr, void *end_ptr)
6802 {
6803 unsigned long start = (unsigned long)(start_ptr);
6804 unsigned long end = (unsigned long)(end_ptr);
6805 struct ftrace_page **last_pg = &ftrace_pages_start;
6806 struct ftrace_page *pg;
6807 struct dyn_ftrace *rec;
6808 struct dyn_ftrace key;
6809 struct ftrace_mod_map *mod_map = NULL;
6810 struct ftrace_init_func *func, *func_next;
6811 struct list_head clear_hash;
6812
6813 INIT_LIST_HEAD(&clear_hash);
6814
6815 key.ip = start;
6816 key.flags = end; /* overload flags, as it is unsigned long */
6817
6818 mutex_lock(&ftrace_lock);
6819
6820 /*
6821 * If we are freeing module init memory, then check if
6822 * any tracer is active. If so, we need to save a mapping of
6823 * the module functions being freed with the address.
6824 */
6825 if (mod && ftrace_ops_list != &ftrace_list_end)
6826 mod_map = allocate_ftrace_mod_map(mod, start, end);
6827
6828 for (pg = ftrace_pages_start; pg; last_pg = &pg->next, pg = *last_pg) {
6829 if (end < pg->records[0].ip ||
6830 start >= (pg->records[pg->index - 1].ip + MCOUNT_INSN_SIZE))
6831 continue;
6832 again:
6833 rec = bsearch(&key, pg->records, pg->index,
6834 sizeof(struct dyn_ftrace),
6835 ftrace_cmp_recs);
6836 if (!rec)
6837 continue;
6838
6839 /* rec will be cleared from hashes after ftrace_lock unlock */
6840 add_to_clear_hash_list(&clear_hash, rec);
6841
6842 if (mod_map)
6843 save_ftrace_mod_rec(mod_map, rec);
6844
6845 pg->index--;
6846 ftrace_update_tot_cnt--;
6847 if (!pg->index) {
6848 *last_pg = pg->next;
6849 if (pg->records) {
6850 free_pages((unsigned long)pg->records, pg->order);
6851 ftrace_number_of_pages -= 1 << pg->order;
6852 }
6853 ftrace_number_of_groups--;
6854 kfree(pg);
6855 pg = container_of(last_pg, struct ftrace_page, next);
6856 if (!(*last_pg))
6857 ftrace_pages = pg;
6858 continue;
6859 }
6860 memmove(rec, rec + 1,
6861 (pg->index - (rec - pg->records)) * sizeof(*rec));
6862 /* More than one function may be in this block */
6863 goto again;
6864 }
6865 mutex_unlock(&ftrace_lock);
6866
6867 list_for_each_entry_safe(func, func_next, &clear_hash, list) {
6868 clear_func_from_hashes(func);
6869 kfree(func);
6870 }
6871 }
6872
ftrace_free_init_mem(void)6873 void __init ftrace_free_init_mem(void)
6874 {
6875 void *start = (void *)(&__init_begin);
6876 void *end = (void *)(&__init_end);
6877
6878 ftrace_free_mem(NULL, start, end);
6879 }
6880
ftrace_init(void)6881 void __init ftrace_init(void)
6882 {
6883 extern unsigned long __start_mcount_loc[];
6884 extern unsigned long __stop_mcount_loc[];
6885 unsigned long count, flags;
6886 int ret;
6887
6888 local_irq_save(flags);
6889 ret = ftrace_dyn_arch_init();
6890 local_irq_restore(flags);
6891 if (ret)
6892 goto failed;
6893
6894 count = __stop_mcount_loc - __start_mcount_loc;
6895 if (!count) {
6896 pr_info("ftrace: No functions to be traced?\n");
6897 goto failed;
6898 }
6899
6900 pr_info("ftrace: allocating %ld entries in %ld pages\n",
6901 count, DIV_ROUND_UP(count, ENTRIES_PER_PAGE));
6902
6903 last_ftrace_enabled = ftrace_enabled = 1;
6904
6905 ret = ftrace_process_locs(NULL,
6906 __start_mcount_loc,
6907 __stop_mcount_loc);
6908
6909 pr_info("ftrace: allocated %ld pages with %ld groups\n",
6910 ftrace_number_of_pages, ftrace_number_of_groups);
6911
6912 set_ftrace_early_filters();
6913
6914 return;
6915 failed:
6916 ftrace_disabled = 1;
6917 }
6918
6919 /* Do nothing if arch does not support this */
arch_ftrace_update_trampoline(struct ftrace_ops * ops)6920 void __weak arch_ftrace_update_trampoline(struct ftrace_ops *ops)
6921 {
6922 }
6923
ftrace_update_trampoline(struct ftrace_ops * ops)6924 static void ftrace_update_trampoline(struct ftrace_ops *ops)
6925 {
6926 unsigned long trampoline = ops->trampoline;
6927
6928 arch_ftrace_update_trampoline(ops);
6929 if (ops->trampoline && ops->trampoline != trampoline &&
6930 (ops->flags & FTRACE_OPS_FL_ALLOC_TRAMP)) {
6931 /* Add to kallsyms before the perf events */
6932 ftrace_add_trampoline_to_kallsyms(ops);
6933 perf_event_ksymbol(PERF_RECORD_KSYMBOL_TYPE_OOL,
6934 ops->trampoline, ops->trampoline_size, false,
6935 FTRACE_TRAMPOLINE_SYM);
6936 /*
6937 * Record the perf text poke event after the ksymbol register
6938 * event.
6939 */
6940 perf_event_text_poke((void *)ops->trampoline, NULL, 0,
6941 (void *)ops->trampoline,
6942 ops->trampoline_size);
6943 }
6944 }
6945
ftrace_init_trace_array(struct trace_array * tr)6946 void ftrace_init_trace_array(struct trace_array *tr)
6947 {
6948 INIT_LIST_HEAD(&tr->func_probes);
6949 INIT_LIST_HEAD(&tr->mod_trace);
6950 INIT_LIST_HEAD(&tr->mod_notrace);
6951 }
6952 #else
6953
6954 struct ftrace_ops global_ops = {
6955 .func = ftrace_stub,
6956 .flags = FTRACE_OPS_FL_RECURSION_SAFE |
6957 FTRACE_OPS_FL_INITIALIZED |
6958 FTRACE_OPS_FL_PID,
6959 };
6960
ftrace_nodyn_init(void)6961 static int __init ftrace_nodyn_init(void)
6962 {
6963 ftrace_enabled = 1;
6964 return 0;
6965 }
6966 core_initcall(ftrace_nodyn_init);
6967
ftrace_init_dyn_tracefs(struct dentry * d_tracer)6968 static inline int ftrace_init_dyn_tracefs(struct dentry *d_tracer) { return 0; }
ftrace_startup_enable(int command)6969 static inline void ftrace_startup_enable(int command) { }
ftrace_startup_all(int command)6970 static inline void ftrace_startup_all(int command) { }
6971
6972 # define ftrace_startup_sysctl() do { } while (0)
6973 # define ftrace_shutdown_sysctl() do { } while (0)
6974
ftrace_update_trampoline(struct ftrace_ops * ops)6975 static void ftrace_update_trampoline(struct ftrace_ops *ops)
6976 {
6977 }
6978
6979 #endif /* CONFIG_DYNAMIC_FTRACE */
6980
ftrace_init_global_array_ops(struct trace_array * tr)6981 __init void ftrace_init_global_array_ops(struct trace_array *tr)
6982 {
6983 tr->ops = &global_ops;
6984 tr->ops->private = tr;
6985 ftrace_init_trace_array(tr);
6986 }
6987
ftrace_init_array_ops(struct trace_array * tr,ftrace_func_t func)6988 void ftrace_init_array_ops(struct trace_array *tr, ftrace_func_t func)
6989 {
6990 /* If we filter on pids, update to use the pid function */
6991 if (tr->flags & TRACE_ARRAY_FL_GLOBAL) {
6992 if (WARN_ON(tr->ops->func != ftrace_stub))
6993 printk("ftrace ops had %pS for function\n",
6994 tr->ops->func);
6995 }
6996 tr->ops->func = func;
6997 tr->ops->private = tr;
6998 }
6999
ftrace_reset_array_ops(struct trace_array * tr)7000 void ftrace_reset_array_ops(struct trace_array *tr)
7001 {
7002 tr->ops->func = ftrace_stub;
7003 }
7004
7005 static nokprobe_inline void
__ftrace_ops_list_func(unsigned long ip,unsigned long parent_ip,struct ftrace_ops * ignored,struct pt_regs * regs)7006 __ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
7007 struct ftrace_ops *ignored, struct pt_regs *regs)
7008 {
7009 struct ftrace_ops *op;
7010 int bit;
7011
7012 bit = trace_test_and_set_recursion(TRACE_LIST_START);
7013 if (bit < 0)
7014 return;
7015
7016 /*
7017 * Some of the ops may be dynamically allocated,
7018 * they must be freed after a synchronize_rcu().
7019 */
7020 preempt_disable_notrace();
7021
7022 do_for_each_ftrace_op(op, ftrace_ops_list) {
7023 /* Stub functions don't need to be called nor tested */
7024 if (op->flags & FTRACE_OPS_FL_STUB)
7025 continue;
7026 /*
7027 * Check the following for each ops before calling their func:
7028 * if RCU flag is set, then rcu_is_watching() must be true
7029 * if PER_CPU is set, then ftrace_function_local_disable()
7030 * must be false
7031 * Otherwise test if the ip matches the ops filter
7032 *
7033 * If any of the above fails then the op->func() is not executed.
7034 */
7035 if ((!(op->flags & FTRACE_OPS_FL_RCU) || rcu_is_watching()) &&
7036 ftrace_ops_test(op, ip, regs)) {
7037 if (FTRACE_WARN_ON(!op->func)) {
7038 pr_warn("op=%p %pS\n", op, op);
7039 goto out;
7040 }
7041 op->func(ip, parent_ip, op, regs);
7042 }
7043 } while_for_each_ftrace_op(op);
7044 out:
7045 preempt_enable_notrace();
7046 trace_clear_recursion(bit);
7047 }
7048
7049 /*
7050 * Some archs only support passing ip and parent_ip. Even though
7051 * the list function ignores the op parameter, we do not want any
7052 * C side effects, where a function is called without the caller
7053 * sending a third parameter.
7054 * Archs are to support both the regs and ftrace_ops at the same time.
7055 * If they support ftrace_ops, it is assumed they support regs.
7056 * If call backs want to use regs, they must either check for regs
7057 * being NULL, or CONFIG_DYNAMIC_FTRACE_WITH_REGS.
7058 * Note, CONFIG_DYNAMIC_FTRACE_WITH_REGS expects a full regs to be saved.
7059 * An architecture can pass partial regs with ftrace_ops and still
7060 * set the ARCH_SUPPORTS_FTRACE_OPS.
7061 */
7062 #if ARCH_SUPPORTS_FTRACE_OPS
ftrace_ops_list_func(unsigned long ip,unsigned long parent_ip,struct ftrace_ops * op,struct pt_regs * regs)7063 static void ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
7064 struct ftrace_ops *op, struct pt_regs *regs)
7065 {
7066 __ftrace_ops_list_func(ip, parent_ip, NULL, regs);
7067 }
7068 NOKPROBE_SYMBOL(ftrace_ops_list_func);
7069 #else
ftrace_ops_no_ops(unsigned long ip,unsigned long parent_ip)7070 static void ftrace_ops_no_ops(unsigned long ip, unsigned long parent_ip)
7071 {
7072 __ftrace_ops_list_func(ip, parent_ip, NULL, NULL);
7073 }
7074 NOKPROBE_SYMBOL(ftrace_ops_no_ops);
7075 #endif
7076
7077 /*
7078 * If there's only one function registered but it does not support
7079 * recursion, needs RCU protection and/or requires per cpu handling, then
7080 * this function will be called by the mcount trampoline.
7081 */
ftrace_ops_assist_func(unsigned long ip,unsigned long parent_ip,struct ftrace_ops * op,struct pt_regs * regs)7082 static void ftrace_ops_assist_func(unsigned long ip, unsigned long parent_ip,
7083 struct ftrace_ops *op, struct pt_regs *regs)
7084 {
7085 int bit;
7086
7087 bit = trace_test_and_set_recursion(TRACE_LIST_START);
7088 if (bit < 0)
7089 return;
7090
7091 preempt_disable_notrace();
7092
7093 if (!(op->flags & FTRACE_OPS_FL_RCU) || rcu_is_watching())
7094 op->func(ip, parent_ip, op, regs);
7095
7096 preempt_enable_notrace();
7097 trace_clear_recursion(bit);
7098 }
7099 NOKPROBE_SYMBOL(ftrace_ops_assist_func);
7100
7101 /**
7102 * ftrace_ops_get_func - get the function a trampoline should call
7103 * @ops: the ops to get the function for
7104 *
7105 * Normally the mcount trampoline will call the ops->func, but there
7106 * are times that it should not. For example, if the ops does not
7107 * have its own recursion protection, then it should call the
7108 * ftrace_ops_assist_func() instead.
7109 *
7110 * Returns the function that the trampoline should call for @ops.
7111 */
ftrace_ops_get_func(struct ftrace_ops * ops)7112 ftrace_func_t ftrace_ops_get_func(struct ftrace_ops *ops)
7113 {
7114 /*
7115 * If the function does not handle recursion, needs to be RCU safe,
7116 * or does per cpu logic, then we need to call the assist handler.
7117 */
7118 if (!(ops->flags & FTRACE_OPS_FL_RECURSION_SAFE) ||
7119 ops->flags & FTRACE_OPS_FL_RCU)
7120 return ftrace_ops_assist_func;
7121
7122 return ops->func;
7123 }
7124
7125 static void
ftrace_filter_pid_sched_switch_probe(void * data,bool preempt,struct task_struct * prev,struct task_struct * next)7126 ftrace_filter_pid_sched_switch_probe(void *data, bool preempt,
7127 struct task_struct *prev, struct task_struct *next)
7128 {
7129 struct trace_array *tr = data;
7130 struct trace_pid_list *pid_list;
7131 struct trace_pid_list *no_pid_list;
7132
7133 pid_list = rcu_dereference_sched(tr->function_pids);
7134 no_pid_list = rcu_dereference_sched(tr->function_no_pids);
7135
7136 if (trace_ignore_this_task(pid_list, no_pid_list, next))
7137 this_cpu_write(tr->array_buffer.data->ftrace_ignore_pid,
7138 FTRACE_PID_IGNORE);
7139 else
7140 this_cpu_write(tr->array_buffer.data->ftrace_ignore_pid,
7141 next->pid);
7142 }
7143
7144 static void
ftrace_pid_follow_sched_process_fork(void * data,struct task_struct * self,struct task_struct * task)7145 ftrace_pid_follow_sched_process_fork(void *data,
7146 struct task_struct *self,
7147 struct task_struct *task)
7148 {
7149 struct trace_pid_list *pid_list;
7150 struct trace_array *tr = data;
7151
7152 pid_list = rcu_dereference_sched(tr->function_pids);
7153 trace_filter_add_remove_task(pid_list, self, task);
7154
7155 pid_list = rcu_dereference_sched(tr->function_no_pids);
7156 trace_filter_add_remove_task(pid_list, self, task);
7157 }
7158
7159 static void
ftrace_pid_follow_sched_process_exit(void * data,struct task_struct * task)7160 ftrace_pid_follow_sched_process_exit(void *data, struct task_struct *task)
7161 {
7162 struct trace_pid_list *pid_list;
7163 struct trace_array *tr = data;
7164
7165 pid_list = rcu_dereference_sched(tr->function_pids);
7166 trace_filter_add_remove_task(pid_list, NULL, task);
7167
7168 pid_list = rcu_dereference_sched(tr->function_no_pids);
7169 trace_filter_add_remove_task(pid_list, NULL, task);
7170 }
7171
ftrace_pid_follow_fork(struct trace_array * tr,bool enable)7172 void ftrace_pid_follow_fork(struct trace_array *tr, bool enable)
7173 {
7174 if (enable) {
7175 register_trace_sched_process_fork(ftrace_pid_follow_sched_process_fork,
7176 tr);
7177 register_trace_sched_process_free(ftrace_pid_follow_sched_process_exit,
7178 tr);
7179 } else {
7180 unregister_trace_sched_process_fork(ftrace_pid_follow_sched_process_fork,
7181 tr);
7182 unregister_trace_sched_process_free(ftrace_pid_follow_sched_process_exit,
7183 tr);
7184 }
7185 }
7186
clear_ftrace_pids(struct trace_array * tr,int type)7187 static void clear_ftrace_pids(struct trace_array *tr, int type)
7188 {
7189 struct trace_pid_list *pid_list;
7190 struct trace_pid_list *no_pid_list;
7191 int cpu;
7192
7193 pid_list = rcu_dereference_protected(tr->function_pids,
7194 lockdep_is_held(&ftrace_lock));
7195 no_pid_list = rcu_dereference_protected(tr->function_no_pids,
7196 lockdep_is_held(&ftrace_lock));
7197
7198 /* Make sure there's something to do */
7199 if (!pid_type_enabled(type, pid_list, no_pid_list))
7200 return;
7201
7202 /* See if the pids still need to be checked after this */
7203 if (!still_need_pid_events(type, pid_list, no_pid_list)) {
7204 unregister_trace_sched_switch(ftrace_filter_pid_sched_switch_probe, tr);
7205 for_each_possible_cpu(cpu)
7206 per_cpu_ptr(tr->array_buffer.data, cpu)->ftrace_ignore_pid = FTRACE_PID_TRACE;
7207 }
7208
7209 if (type & TRACE_PIDS)
7210 rcu_assign_pointer(tr->function_pids, NULL);
7211
7212 if (type & TRACE_NO_PIDS)
7213 rcu_assign_pointer(tr->function_no_pids, NULL);
7214
7215 /* Wait till all users are no longer using pid filtering */
7216 synchronize_rcu();
7217
7218 if ((type & TRACE_PIDS) && pid_list)
7219 trace_free_pid_list(pid_list);
7220
7221 if ((type & TRACE_NO_PIDS) && no_pid_list)
7222 trace_free_pid_list(no_pid_list);
7223 }
7224
ftrace_clear_pids(struct trace_array * tr)7225 void ftrace_clear_pids(struct trace_array *tr)
7226 {
7227 mutex_lock(&ftrace_lock);
7228
7229 clear_ftrace_pids(tr, TRACE_PIDS | TRACE_NO_PIDS);
7230
7231 mutex_unlock(&ftrace_lock);
7232 }
7233
ftrace_pid_reset(struct trace_array * tr,int type)7234 static void ftrace_pid_reset(struct trace_array *tr, int type)
7235 {
7236 mutex_lock(&ftrace_lock);
7237 clear_ftrace_pids(tr, type);
7238
7239 ftrace_update_pid_func();
7240 ftrace_startup_all(0);
7241
7242 mutex_unlock(&ftrace_lock);
7243 }
7244
7245 /* Greater than any max PID */
7246 #define FTRACE_NO_PIDS (void *)(PID_MAX_LIMIT + 1)
7247
fpid_start(struct seq_file * m,loff_t * pos)7248 static void *fpid_start(struct seq_file *m, loff_t *pos)
7249 __acquires(RCU)
7250 {
7251 struct trace_pid_list *pid_list;
7252 struct trace_array *tr = m->private;
7253
7254 mutex_lock(&ftrace_lock);
7255 rcu_read_lock_sched();
7256
7257 pid_list = rcu_dereference_sched(tr->function_pids);
7258
7259 if (!pid_list)
7260 return !(*pos) ? FTRACE_NO_PIDS : NULL;
7261
7262 return trace_pid_start(pid_list, pos);
7263 }
7264
fpid_next(struct seq_file * m,void * v,loff_t * pos)7265 static void *fpid_next(struct seq_file *m, void *v, loff_t *pos)
7266 {
7267 struct trace_array *tr = m->private;
7268 struct trace_pid_list *pid_list = rcu_dereference_sched(tr->function_pids);
7269
7270 if (v == FTRACE_NO_PIDS) {
7271 (*pos)++;
7272 return NULL;
7273 }
7274 return trace_pid_next(pid_list, v, pos);
7275 }
7276
fpid_stop(struct seq_file * m,void * p)7277 static void fpid_stop(struct seq_file *m, void *p)
7278 __releases(RCU)
7279 {
7280 rcu_read_unlock_sched();
7281 mutex_unlock(&ftrace_lock);
7282 }
7283
fpid_show(struct seq_file * m,void * v)7284 static int fpid_show(struct seq_file *m, void *v)
7285 {
7286 if (v == FTRACE_NO_PIDS) {
7287 seq_puts(m, "no pid\n");
7288 return 0;
7289 }
7290
7291 return trace_pid_show(m, v);
7292 }
7293
7294 static const struct seq_operations ftrace_pid_sops = {
7295 .start = fpid_start,
7296 .next = fpid_next,
7297 .stop = fpid_stop,
7298 .show = fpid_show,
7299 };
7300
fnpid_start(struct seq_file * m,loff_t * pos)7301 static void *fnpid_start(struct seq_file *m, loff_t *pos)
7302 __acquires(RCU)
7303 {
7304 struct trace_pid_list *pid_list;
7305 struct trace_array *tr = m->private;
7306
7307 mutex_lock(&ftrace_lock);
7308 rcu_read_lock_sched();
7309
7310 pid_list = rcu_dereference_sched(tr->function_no_pids);
7311
7312 if (!pid_list)
7313 return !(*pos) ? FTRACE_NO_PIDS : NULL;
7314
7315 return trace_pid_start(pid_list, pos);
7316 }
7317
fnpid_next(struct seq_file * m,void * v,loff_t * pos)7318 static void *fnpid_next(struct seq_file *m, void *v, loff_t *pos)
7319 {
7320 struct trace_array *tr = m->private;
7321 struct trace_pid_list *pid_list = rcu_dereference_sched(tr->function_no_pids);
7322
7323 if (v == FTRACE_NO_PIDS) {
7324 (*pos)++;
7325 return NULL;
7326 }
7327 return trace_pid_next(pid_list, v, pos);
7328 }
7329
7330 static const struct seq_operations ftrace_no_pid_sops = {
7331 .start = fnpid_start,
7332 .next = fnpid_next,
7333 .stop = fpid_stop,
7334 .show = fpid_show,
7335 };
7336
pid_open(struct inode * inode,struct file * file,int type)7337 static int pid_open(struct inode *inode, struct file *file, int type)
7338 {
7339 const struct seq_operations *seq_ops;
7340 struct trace_array *tr = inode->i_private;
7341 struct seq_file *m;
7342 int ret = 0;
7343
7344 ret = tracing_check_open_get_tr(tr);
7345 if (ret)
7346 return ret;
7347
7348 if ((file->f_mode & FMODE_WRITE) &&
7349 (file->f_flags & O_TRUNC))
7350 ftrace_pid_reset(tr, type);
7351
7352 switch (type) {
7353 case TRACE_PIDS:
7354 seq_ops = &ftrace_pid_sops;
7355 break;
7356 case TRACE_NO_PIDS:
7357 seq_ops = &ftrace_no_pid_sops;
7358 break;
7359 default:
7360 trace_array_put(tr);
7361 WARN_ON_ONCE(1);
7362 return -EINVAL;
7363 }
7364
7365 ret = seq_open(file, seq_ops);
7366 if (ret < 0) {
7367 trace_array_put(tr);
7368 } else {
7369 m = file->private_data;
7370 /* copy tr over to seq ops */
7371 m->private = tr;
7372 }
7373
7374 return ret;
7375 }
7376
7377 static int
ftrace_pid_open(struct inode * inode,struct file * file)7378 ftrace_pid_open(struct inode *inode, struct file *file)
7379 {
7380 return pid_open(inode, file, TRACE_PIDS);
7381 }
7382
7383 static int
ftrace_no_pid_open(struct inode * inode,struct file * file)7384 ftrace_no_pid_open(struct inode *inode, struct file *file)
7385 {
7386 return pid_open(inode, file, TRACE_NO_PIDS);
7387 }
7388
ignore_task_cpu(void * data)7389 static void ignore_task_cpu(void *data)
7390 {
7391 struct trace_array *tr = data;
7392 struct trace_pid_list *pid_list;
7393 struct trace_pid_list *no_pid_list;
7394
7395 /*
7396 * This function is called by on_each_cpu() while the
7397 * event_mutex is held.
7398 */
7399 pid_list = rcu_dereference_protected(tr->function_pids,
7400 mutex_is_locked(&ftrace_lock));
7401 no_pid_list = rcu_dereference_protected(tr->function_no_pids,
7402 mutex_is_locked(&ftrace_lock));
7403
7404 if (trace_ignore_this_task(pid_list, no_pid_list, current))
7405 this_cpu_write(tr->array_buffer.data->ftrace_ignore_pid,
7406 FTRACE_PID_IGNORE);
7407 else
7408 this_cpu_write(tr->array_buffer.data->ftrace_ignore_pid,
7409 current->pid);
7410 }
7411
7412 static ssize_t
pid_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos,int type)7413 pid_write(struct file *filp, const char __user *ubuf,
7414 size_t cnt, loff_t *ppos, int type)
7415 {
7416 struct seq_file *m = filp->private_data;
7417 struct trace_array *tr = m->private;
7418 struct trace_pid_list *filtered_pids;
7419 struct trace_pid_list *other_pids;
7420 struct trace_pid_list *pid_list;
7421 ssize_t ret;
7422
7423 if (!cnt)
7424 return 0;
7425
7426 mutex_lock(&ftrace_lock);
7427
7428 switch (type) {
7429 case TRACE_PIDS:
7430 filtered_pids = rcu_dereference_protected(tr->function_pids,
7431 lockdep_is_held(&ftrace_lock));
7432 other_pids = rcu_dereference_protected(tr->function_no_pids,
7433 lockdep_is_held(&ftrace_lock));
7434 break;
7435 case TRACE_NO_PIDS:
7436 filtered_pids = rcu_dereference_protected(tr->function_no_pids,
7437 lockdep_is_held(&ftrace_lock));
7438 other_pids = rcu_dereference_protected(tr->function_pids,
7439 lockdep_is_held(&ftrace_lock));
7440 break;
7441 default:
7442 ret = -EINVAL;
7443 WARN_ON_ONCE(1);
7444 goto out;
7445 }
7446
7447 ret = trace_pid_write(filtered_pids, &pid_list, ubuf, cnt);
7448 if (ret < 0)
7449 goto out;
7450
7451 switch (type) {
7452 case TRACE_PIDS:
7453 rcu_assign_pointer(tr->function_pids, pid_list);
7454 break;
7455 case TRACE_NO_PIDS:
7456 rcu_assign_pointer(tr->function_no_pids, pid_list);
7457 break;
7458 }
7459
7460
7461 if (filtered_pids) {
7462 synchronize_rcu();
7463 trace_free_pid_list(filtered_pids);
7464 } else if (pid_list && !other_pids) {
7465 /* Register a probe to set whether to ignore the tracing of a task */
7466 register_trace_sched_switch(ftrace_filter_pid_sched_switch_probe, tr);
7467 }
7468
7469 /*
7470 * Ignoring of pids is done at task switch. But we have to
7471 * check for those tasks that are currently running.
7472 * Always do this in case a pid was appended or removed.
7473 */
7474 on_each_cpu(ignore_task_cpu, tr, 1);
7475
7476 ftrace_update_pid_func();
7477 ftrace_startup_all(0);
7478 out:
7479 mutex_unlock(&ftrace_lock);
7480
7481 if (ret > 0)
7482 *ppos += ret;
7483
7484 return ret;
7485 }
7486
7487 static ssize_t
ftrace_pid_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos)7488 ftrace_pid_write(struct file *filp, const char __user *ubuf,
7489 size_t cnt, loff_t *ppos)
7490 {
7491 return pid_write(filp, ubuf, cnt, ppos, TRACE_PIDS);
7492 }
7493
7494 static ssize_t
ftrace_no_pid_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos)7495 ftrace_no_pid_write(struct file *filp, const char __user *ubuf,
7496 size_t cnt, loff_t *ppos)
7497 {
7498 return pid_write(filp, ubuf, cnt, ppos, TRACE_NO_PIDS);
7499 }
7500
7501 static int
ftrace_pid_release(struct inode * inode,struct file * file)7502 ftrace_pid_release(struct inode *inode, struct file *file)
7503 {
7504 struct trace_array *tr = inode->i_private;
7505
7506 trace_array_put(tr);
7507
7508 return seq_release(inode, file);
7509 }
7510
7511 static const struct file_operations ftrace_pid_fops = {
7512 .open = ftrace_pid_open,
7513 .write = ftrace_pid_write,
7514 .read = seq_read,
7515 .llseek = tracing_lseek,
7516 .release = ftrace_pid_release,
7517 };
7518
7519 static const struct file_operations ftrace_no_pid_fops = {
7520 .open = ftrace_no_pid_open,
7521 .write = ftrace_no_pid_write,
7522 .read = seq_read,
7523 .llseek = tracing_lseek,
7524 .release = ftrace_pid_release,
7525 };
7526
ftrace_init_tracefs(struct trace_array * tr,struct dentry * d_tracer)7527 void ftrace_init_tracefs(struct trace_array *tr, struct dentry *d_tracer)
7528 {
7529 trace_create_file("set_ftrace_pid", 0644, d_tracer,
7530 tr, &ftrace_pid_fops);
7531 trace_create_file("set_ftrace_notrace_pid", 0644, d_tracer,
7532 tr, &ftrace_no_pid_fops);
7533 }
7534
ftrace_init_tracefs_toplevel(struct trace_array * tr,struct dentry * d_tracer)7535 void __init ftrace_init_tracefs_toplevel(struct trace_array *tr,
7536 struct dentry *d_tracer)
7537 {
7538 /* Only the top level directory has the dyn_tracefs and profile */
7539 WARN_ON(!(tr->flags & TRACE_ARRAY_FL_GLOBAL));
7540
7541 ftrace_init_dyn_tracefs(d_tracer);
7542 ftrace_profile_tracefs(d_tracer);
7543 }
7544
7545 /**
7546 * ftrace_kill - kill ftrace
7547 *
7548 * This function should be used by panic code. It stops ftrace
7549 * but in a not so nice way. If you need to simply kill ftrace
7550 * from a non-atomic section, use ftrace_kill.
7551 */
ftrace_kill(void)7552 void ftrace_kill(void)
7553 {
7554 ftrace_disabled = 1;
7555 ftrace_enabled = 0;
7556 ftrace_trace_function = ftrace_stub;
7557 }
7558
7559 /**
7560 * Test if ftrace is dead or not.
7561 */
ftrace_is_dead(void)7562 int ftrace_is_dead(void)
7563 {
7564 return ftrace_disabled;
7565 }
7566
7567 /**
7568 * register_ftrace_function - register a function for profiling
7569 * @ops - ops structure that holds the function for profiling.
7570 *
7571 * Register a function to be called by all functions in the
7572 * kernel.
7573 *
7574 * Note: @ops->func and all the functions it calls must be labeled
7575 * with "notrace", otherwise it will go into a
7576 * recursive loop.
7577 */
register_ftrace_function(struct ftrace_ops * ops)7578 int register_ftrace_function(struct ftrace_ops *ops)
7579 {
7580 int ret = -1;
7581
7582 ftrace_ops_init(ops);
7583
7584 mutex_lock(&ftrace_lock);
7585
7586 ret = ftrace_startup(ops, 0);
7587
7588 mutex_unlock(&ftrace_lock);
7589
7590 return ret;
7591 }
7592 EXPORT_SYMBOL_GPL(register_ftrace_function);
7593
7594 /**
7595 * unregister_ftrace_function - unregister a function for profiling.
7596 * @ops - ops structure that holds the function to unregister
7597 *
7598 * Unregister a function that was added to be called by ftrace profiling.
7599 */
unregister_ftrace_function(struct ftrace_ops * ops)7600 int unregister_ftrace_function(struct ftrace_ops *ops)
7601 {
7602 int ret;
7603
7604 mutex_lock(&ftrace_lock);
7605 ret = ftrace_shutdown(ops, 0);
7606 mutex_unlock(&ftrace_lock);
7607
7608 return ret;
7609 }
7610 EXPORT_SYMBOL_GPL(unregister_ftrace_function);
7611
is_permanent_ops_registered(void)7612 static bool is_permanent_ops_registered(void)
7613 {
7614 struct ftrace_ops *op;
7615
7616 do_for_each_ftrace_op(op, ftrace_ops_list) {
7617 if (op->flags & FTRACE_OPS_FL_PERMANENT)
7618 return true;
7619 } while_for_each_ftrace_op(op);
7620
7621 return false;
7622 }
7623
7624 int
ftrace_enable_sysctl(struct ctl_table * table,int write,void * buffer,size_t * lenp,loff_t * ppos)7625 ftrace_enable_sysctl(struct ctl_table *table, int write,
7626 void *buffer, size_t *lenp, loff_t *ppos)
7627 {
7628 int ret = -ENODEV;
7629
7630 mutex_lock(&ftrace_lock);
7631
7632 if (unlikely(ftrace_disabled))
7633 goto out;
7634
7635 ret = proc_dointvec(table, write, buffer, lenp, ppos);
7636
7637 if (ret || !write || (last_ftrace_enabled == !!ftrace_enabled))
7638 goto out;
7639
7640 if (ftrace_enabled) {
7641
7642 /* we are starting ftrace again */
7643 if (rcu_dereference_protected(ftrace_ops_list,
7644 lockdep_is_held(&ftrace_lock)) != &ftrace_list_end)
7645 update_ftrace_function();
7646
7647 ftrace_startup_sysctl();
7648
7649 } else {
7650 if (is_permanent_ops_registered()) {
7651 ftrace_enabled = true;
7652 ret = -EBUSY;
7653 goto out;
7654 }
7655
7656 /* stopping ftrace calls (just send to ftrace_stub) */
7657 ftrace_trace_function = ftrace_stub;
7658
7659 ftrace_shutdown_sysctl();
7660 }
7661
7662 last_ftrace_enabled = !!ftrace_enabled;
7663 out:
7664 mutex_unlock(&ftrace_lock);
7665 return ret;
7666 }
7667