• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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, &not);
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, &not);
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