xref: /linux-6.15/kernel/trace/ftrace.c (revision 345ddcc8)
116444a8aSArnaldo Carvalho de Melo /*
216444a8aSArnaldo Carvalho de Melo  * Infrastructure for profiling code inserted by 'gcc -pg'.
316444a8aSArnaldo Carvalho de Melo  *
416444a8aSArnaldo Carvalho de Melo  * Copyright (C) 2007-2008 Steven Rostedt <[email protected]>
516444a8aSArnaldo Carvalho de Melo  * Copyright (C) 2004-2008 Ingo Molnar <[email protected]>
616444a8aSArnaldo Carvalho de Melo  *
716444a8aSArnaldo Carvalho de Melo  * Originally ported from the -rt patch by:
816444a8aSArnaldo Carvalho de Melo  *   Copyright (C) 2007 Arnaldo Carvalho de Melo <[email protected]>
916444a8aSArnaldo Carvalho de Melo  *
1016444a8aSArnaldo Carvalho de Melo  * Based on code in the latency_tracer, that is:
1116444a8aSArnaldo Carvalho de Melo  *
1216444a8aSArnaldo Carvalho de Melo  *  Copyright (C) 2004-2006 Ingo Molnar
136d49e352SNadia Yvette Chambers  *  Copyright (C) 2004 Nadia Yvette Chambers
1416444a8aSArnaldo Carvalho de Melo  */
1516444a8aSArnaldo Carvalho de Melo 
163d083395SSteven Rostedt #include <linux/stop_machine.h>
173d083395SSteven Rostedt #include <linux/clocksource.h>
183d083395SSteven Rostedt #include <linux/kallsyms.h>
195072c59fSSteven Rostedt #include <linux/seq_file.h>
204a2b8ddaSFrederic Weisbecker #include <linux/suspend.h>
218434dc93SSteven Rostedt (Red Hat) #include <linux/tracefs.h>
223d083395SSteven Rostedt #include <linux/hardirq.h>
232d8b820bSIngo Molnar #include <linux/kthread.h>
245072c59fSSteven Rostedt #include <linux/uaccess.h>
255855feadSSteven Rostedt #include <linux/bsearch.h>
2656d82e00SPaul Gortmaker #include <linux/module.h>
272d8b820bSIngo Molnar #include <linux/ftrace.h>
28b0fc494fSSteven Rostedt #include <linux/sysctl.h>
295a0e3ad6STejun Heo #include <linux/slab.h>
305072c59fSSteven Rostedt #include <linux/ctype.h>
3168950619SSteven Rostedt #include <linux/sort.h>
323d083395SSteven Rostedt #include <linux/list.h>
3359df055fSSteven Rostedt #include <linux/hash.h>
343f379b03SPaul E. McKenney #include <linux/rcupdate.h>
3516444a8aSArnaldo Carvalho de Melo 
36ad8d75ffSSteven Rostedt #include <trace/events/sched.h>
378aef2d28SSteven Rostedt 
382af15d6aSSteven Rostedt #include <asm/setup.h>
39395a59d0SAbhishek Sagar 
400706f1c4SSteven Rostedt #include "trace_output.h"
41bac429f0SSteven Rostedt #include "trace_stat.h"
423d083395SSteven Rostedt 
436912896eSSteven Rostedt #define FTRACE_WARN_ON(cond)			\
440778d9adSSteven Rostedt 	({					\
450778d9adSSteven Rostedt 		int ___r = cond;		\
460778d9adSSteven Rostedt 		if (WARN_ON(___r))		\
476912896eSSteven Rostedt 			ftrace_kill();		\
480778d9adSSteven Rostedt 		___r;				\
490778d9adSSteven Rostedt 	})
506912896eSSteven Rostedt 
516912896eSSteven Rostedt #define FTRACE_WARN_ON_ONCE(cond)		\
520778d9adSSteven Rostedt 	({					\
530778d9adSSteven Rostedt 		int ___r = cond;		\
540778d9adSSteven Rostedt 		if (WARN_ON_ONCE(___r))		\
556912896eSSteven Rostedt 			ftrace_kill();		\
560778d9adSSteven Rostedt 		___r;				\
570778d9adSSteven Rostedt 	})
586912896eSSteven Rostedt 
598fc0c701SSteven Rostedt /* hash bits for specific function selection */
608fc0c701SSteven Rostedt #define FTRACE_HASH_BITS 7
618fc0c701SSteven Rostedt #define FTRACE_FUNC_HASHSIZE (1 << FTRACE_HASH_BITS)
6233dc9b12SSteven Rostedt #define FTRACE_HASH_DEFAULT_BITS 10
6333dc9b12SSteven Rostedt #define FTRACE_HASH_MAX_BITS 12
648fc0c701SSteven Rostedt 
65f04f24fbSMasami Hiramatsu #ifdef CONFIG_DYNAMIC_FTRACE
6633b7f99cSSteven Rostedt (Red Hat) #define INIT_OPS_HASH(opsname)	\
6733b7f99cSSteven Rostedt (Red Hat) 	.func_hash		= &opsname.local_hash,			\
6833b7f99cSSteven Rostedt (Red Hat) 	.local_hash.regex_lock	= __MUTEX_INITIALIZER(opsname.local_hash.regex_lock),
695f151b24SSteven Rostedt (Red Hat) #define ASSIGN_OPS_HASH(opsname, val) \
705f151b24SSteven Rostedt (Red Hat) 	.func_hash		= val, \
715f151b24SSteven Rostedt (Red Hat) 	.local_hash.regex_lock	= __MUTEX_INITIALIZER(opsname.local_hash.regex_lock),
72f04f24fbSMasami Hiramatsu #else
7333b7f99cSSteven Rostedt (Red Hat) #define INIT_OPS_HASH(opsname)
745f151b24SSteven Rostedt (Red Hat) #define ASSIGN_OPS_HASH(opsname, val)
75f04f24fbSMasami Hiramatsu #endif
76f04f24fbSMasami Hiramatsu 
772f5f6ad9SSteven Rostedt static struct ftrace_ops ftrace_list_end __read_mostly = {
782f5f6ad9SSteven Rostedt 	.func		= ftrace_stub,
79395b97a3SSteven Rostedt (Red Hat) 	.flags		= FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_STUB,
8033b7f99cSSteven Rostedt (Red Hat) 	INIT_OPS_HASH(ftrace_list_end)
812f5f6ad9SSteven Rostedt };
822f5f6ad9SSteven Rostedt 
834eebcc81SSteven Rostedt /* ftrace_enabled is a method to turn ftrace on or off */
844eebcc81SSteven Rostedt int ftrace_enabled __read_mostly;
85d61f82d0SSteven Rostedt static int last_ftrace_enabled;
86b0fc494fSSteven Rostedt 
872f5f6ad9SSteven Rostedt /* Current function tracing op */
882f5f6ad9SSteven Rostedt struct ftrace_ops *function_trace_op __read_mostly = &ftrace_list_end;
89405e1d83SSteven Rostedt (Red Hat) /* What to set function_trace_op to */
90405e1d83SSteven Rostedt (Red Hat) static struct ftrace_ops *set_function_trace_op;
9160a7ecf4SSteven Rostedt 
92*345ddcc8SSteven Rostedt (Red Hat) static bool ftrace_pids_enabled(struct ftrace_ops *ops)
93e3eea140SSteven Rostedt (Red Hat) {
94*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr;
95*345ddcc8SSteven Rostedt (Red Hat) 
96*345ddcc8SSteven Rostedt (Red Hat) 	if (!(ops->flags & FTRACE_OPS_FL_PID) || !ops->private)
97*345ddcc8SSteven Rostedt (Red Hat) 		return false;
98*345ddcc8SSteven Rostedt (Red Hat) 
99*345ddcc8SSteven Rostedt (Red Hat) 	tr = ops->private;
100*345ddcc8SSteven Rostedt (Red Hat) 
101*345ddcc8SSteven Rostedt (Red Hat) 	return tr->function_pids != NULL;
102e3eea140SSteven Rostedt (Red Hat) }
103e3eea140SSteven Rostedt (Red Hat) 
104e3eea140SSteven Rostedt (Red Hat) static void ftrace_update_trampoline(struct ftrace_ops *ops);
105e3eea140SSteven Rostedt (Red Hat) 
1064eebcc81SSteven Rostedt /*
1074eebcc81SSteven Rostedt  * ftrace_disabled is set when an anomaly is discovered.
1084eebcc81SSteven Rostedt  * ftrace_disabled is much stronger than ftrace_enabled.
1094eebcc81SSteven Rostedt  */
1104eebcc81SSteven Rostedt static int ftrace_disabled __read_mostly;
1114eebcc81SSteven Rostedt 
11252baf119SSteven Rostedt static DEFINE_MUTEX(ftrace_lock);
113b0fc494fSSteven Rostedt 
114b848914cSSteven Rostedt static struct ftrace_ops *ftrace_ops_list __read_mostly = &ftrace_list_end;
11516444a8aSArnaldo Carvalho de Melo ftrace_func_t ftrace_trace_function __read_mostly = ftrace_stub;
1162b499381SSteven Rostedt static struct ftrace_ops global_ops;
11716444a8aSArnaldo Carvalho de Melo 
1182f5f6ad9SSteven Rostedt #if ARCH_SUPPORTS_FTRACE_OPS
1192f5f6ad9SSteven Rostedt static void ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
120a1e2e31dSSteven Rostedt 				 struct ftrace_ops *op, struct pt_regs *regs);
1212f5f6ad9SSteven Rostedt #else
1222f5f6ad9SSteven Rostedt /* See comment below, where ftrace_ops_list_func is defined */
1232f5f6ad9SSteven Rostedt static void ftrace_ops_no_ops(unsigned long ip, unsigned long parent_ip);
1242f5f6ad9SSteven Rostedt #define ftrace_ops_list_func ((ftrace_func_t)ftrace_ops_no_ops)
1252f5f6ad9SSteven Rostedt #endif
126b848914cSSteven Rostedt 
1270a016409SSteven Rostedt /*
1280a016409SSteven Rostedt  * Traverse the ftrace_global_list, invoking all entries.  The reason that we
1291bb539caSSteven Rostedt  * can use rcu_dereference_raw_notrace() is that elements removed from this list
1300a016409SSteven Rostedt  * are simply leaked, so there is no need to interact with a grace-period
1311bb539caSSteven Rostedt  * mechanism.  The rcu_dereference_raw_notrace() calls are needed to handle
1320a016409SSteven Rostedt  * concurrent insertions into the ftrace_global_list.
1330a016409SSteven Rostedt  *
1340a016409SSteven Rostedt  * Silly Alpha and silly pointer-speculation compiler optimizations!
1350a016409SSteven Rostedt  */
1360a016409SSteven Rostedt #define do_for_each_ftrace_op(op, list)			\
1371bb539caSSteven Rostedt 	op = rcu_dereference_raw_notrace(list);			\
1380a016409SSteven Rostedt 	do
1390a016409SSteven Rostedt 
1400a016409SSteven Rostedt /*
1410a016409SSteven Rostedt  * Optimized for just a single item in the list (as that is the normal case).
1420a016409SSteven Rostedt  */
1430a016409SSteven Rostedt #define while_for_each_ftrace_op(op)				\
1441bb539caSSteven Rostedt 	while (likely(op = rcu_dereference_raw_notrace((op)->next)) &&	\
1450a016409SSteven Rostedt 	       unlikely((op) != &ftrace_list_end))
1460a016409SSteven Rostedt 
147f04f24fbSMasami Hiramatsu static inline void ftrace_ops_init(struct ftrace_ops *ops)
148f04f24fbSMasami Hiramatsu {
149f04f24fbSMasami Hiramatsu #ifdef CONFIG_DYNAMIC_FTRACE
150f04f24fbSMasami Hiramatsu 	if (!(ops->flags & FTRACE_OPS_FL_INITIALIZED)) {
15133b7f99cSSteven Rostedt (Red Hat) 		mutex_init(&ops->local_hash.regex_lock);
15233b7f99cSSteven Rostedt (Red Hat) 		ops->func_hash = &ops->local_hash;
153f04f24fbSMasami Hiramatsu 		ops->flags |= FTRACE_OPS_FL_INITIALIZED;
154f04f24fbSMasami Hiramatsu 	}
155f04f24fbSMasami Hiramatsu #endif
156f04f24fbSMasami Hiramatsu }
157f04f24fbSMasami Hiramatsu 
158ea701f11SSteven Rostedt /**
159ea701f11SSteven Rostedt  * ftrace_nr_registered_ops - return number of ops registered
160ea701f11SSteven Rostedt  *
161ea701f11SSteven Rostedt  * Returns the number of ftrace_ops registered and tracing functions
162ea701f11SSteven Rostedt  */
163ea701f11SSteven Rostedt int ftrace_nr_registered_ops(void)
164ea701f11SSteven Rostedt {
165ea701f11SSteven Rostedt 	struct ftrace_ops *ops;
166ea701f11SSteven Rostedt 	int cnt = 0;
167ea701f11SSteven Rostedt 
168ea701f11SSteven Rostedt 	mutex_lock(&ftrace_lock);
169ea701f11SSteven Rostedt 
170ea701f11SSteven Rostedt 	for (ops = ftrace_ops_list;
171ea701f11SSteven Rostedt 	     ops != &ftrace_list_end; ops = ops->next)
172ea701f11SSteven Rostedt 		cnt++;
173ea701f11SSteven Rostedt 
174ea701f11SSteven Rostedt 	mutex_unlock(&ftrace_lock);
175ea701f11SSteven Rostedt 
176ea701f11SSteven Rostedt 	return cnt;
177ea701f11SSteven Rostedt }
178ea701f11SSteven Rostedt 
1792f5f6ad9SSteven Rostedt static void ftrace_pid_func(unsigned long ip, unsigned long parent_ip,
180a1e2e31dSSteven Rostedt 			    struct ftrace_ops *op, struct pt_regs *regs)
181df4fc315SSteven Rostedt {
182*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr = op->private;
183*345ddcc8SSteven Rostedt (Red Hat) 
184*345ddcc8SSteven Rostedt (Red Hat) 	if (tr && this_cpu_read(tr->trace_buffer.data->ftrace_ignore_pid))
185df4fc315SSteven Rostedt 		return;
186df4fc315SSteven Rostedt 
187e3eea140SSteven Rostedt (Red Hat) 	op->saved_func(ip, parent_ip, op, regs);
188df4fc315SSteven Rostedt }
189df4fc315SSteven Rostedt 
19016444a8aSArnaldo Carvalho de Melo /**
1913d083395SSteven Rostedt  * clear_ftrace_function - reset the ftrace function
19216444a8aSArnaldo Carvalho de Melo  *
1933d083395SSteven Rostedt  * This NULLs the ftrace function and in essence stops
1943d083395SSteven Rostedt  * tracing.  There may be lag
19516444a8aSArnaldo Carvalho de Melo  */
1963d083395SSteven Rostedt void clear_ftrace_function(void)
19716444a8aSArnaldo Carvalho de Melo {
1983d083395SSteven Rostedt 	ftrace_trace_function = ftrace_stub;
1993d083395SSteven Rostedt }
20016444a8aSArnaldo Carvalho de Melo 
201ba27f2bcSSteven Rostedt (Red Hat) static void per_cpu_ops_disable_all(struct ftrace_ops *ops)
202e248491aSJiri Olsa {
203e248491aSJiri Olsa 	int cpu;
204e248491aSJiri Olsa 
205e248491aSJiri Olsa 	for_each_possible_cpu(cpu)
206e248491aSJiri Olsa 		*per_cpu_ptr(ops->disabled, cpu) = 1;
207e248491aSJiri Olsa }
208e248491aSJiri Olsa 
209ba27f2bcSSteven Rostedt (Red Hat) static int per_cpu_ops_alloc(struct ftrace_ops *ops)
210e248491aSJiri Olsa {
211e248491aSJiri Olsa 	int __percpu *disabled;
212e248491aSJiri Olsa 
213ba27f2bcSSteven Rostedt (Red Hat) 	if (WARN_ON_ONCE(!(ops->flags & FTRACE_OPS_FL_PER_CPU)))
214ba27f2bcSSteven Rostedt (Red Hat) 		return -EINVAL;
215ba27f2bcSSteven Rostedt (Red Hat) 
216e248491aSJiri Olsa 	disabled = alloc_percpu(int);
217e248491aSJiri Olsa 	if (!disabled)
218e248491aSJiri Olsa 		return -ENOMEM;
219e248491aSJiri Olsa 
220e248491aSJiri Olsa 	ops->disabled = disabled;
221ba27f2bcSSteven Rostedt (Red Hat) 	per_cpu_ops_disable_all(ops);
222e248491aSJiri Olsa 	return 0;
223e248491aSJiri Olsa }
224e248491aSJiri Olsa 
225405e1d83SSteven Rostedt (Red Hat) static void ftrace_sync(struct work_struct *work)
226405e1d83SSteven Rostedt (Red Hat) {
227405e1d83SSteven Rostedt (Red Hat) 	/*
228405e1d83SSteven Rostedt (Red Hat) 	 * This function is just a stub to implement a hard force
229405e1d83SSteven Rostedt (Red Hat) 	 * of synchronize_sched(). This requires synchronizing
230405e1d83SSteven Rostedt (Red Hat) 	 * tasks even in userspace and idle.
231405e1d83SSteven Rostedt (Red Hat) 	 *
232405e1d83SSteven Rostedt (Red Hat) 	 * Yes, function tracing is rude.
233405e1d83SSteven Rostedt (Red Hat) 	 */
234405e1d83SSteven Rostedt (Red Hat) }
235405e1d83SSteven Rostedt (Red Hat) 
236405e1d83SSteven Rostedt (Red Hat) static void ftrace_sync_ipi(void *data)
237405e1d83SSteven Rostedt (Red Hat) {
238405e1d83SSteven Rostedt (Red Hat) 	/* Probably not needed, but do it anyway */
239405e1d83SSteven Rostedt (Red Hat) 	smp_rmb();
240405e1d83SSteven Rostedt (Red Hat) }
241405e1d83SSteven Rostedt (Red Hat) 
24223a8e844SSteven Rostedt (Red Hat) #ifdef CONFIG_FUNCTION_GRAPH_TRACER
24323a8e844SSteven Rostedt (Red Hat) static void update_function_graph_func(void);
24455577204SSteven Rostedt (Red Hat) 
24555577204SSteven Rostedt (Red Hat) /* Both enabled by default (can be cleared by function_graph tracer flags */
24655577204SSteven Rostedt (Red Hat) static bool fgraph_sleep_time = true;
24755577204SSteven Rostedt (Red Hat) static bool fgraph_graph_time = true;
24855577204SSteven Rostedt (Red Hat) 
24923a8e844SSteven Rostedt (Red Hat) #else
25023a8e844SSteven Rostedt (Red Hat) static inline void update_function_graph_func(void) { }
25123a8e844SSteven Rostedt (Red Hat) #endif
25223a8e844SSteven Rostedt (Red Hat) 
25300ccbf2fSSteven Rostedt (Red Hat) 
25400ccbf2fSSteven Rostedt (Red Hat) static ftrace_func_t ftrace_ops_get_list_func(struct ftrace_ops *ops)
25500ccbf2fSSteven Rostedt (Red Hat) {
25600ccbf2fSSteven Rostedt (Red Hat) 	/*
257ba27f2bcSSteven Rostedt (Red Hat) 	 * If this is a dynamic, RCU, or per CPU ops, or we force list func,
25800ccbf2fSSteven Rostedt (Red Hat) 	 * then it needs to call the list anyway.
25900ccbf2fSSteven Rostedt (Red Hat) 	 */
260ba27f2bcSSteven Rostedt (Red Hat) 	if (ops->flags & (FTRACE_OPS_FL_DYNAMIC | FTRACE_OPS_FL_PER_CPU |
261ba27f2bcSSteven Rostedt (Red Hat) 			  FTRACE_OPS_FL_RCU) || FTRACE_FORCE_LIST_FUNC)
26200ccbf2fSSteven Rostedt (Red Hat) 		return ftrace_ops_list_func;
26300ccbf2fSSteven Rostedt (Red Hat) 
26400ccbf2fSSteven Rostedt (Red Hat) 	return ftrace_ops_get_func(ops);
26500ccbf2fSSteven Rostedt (Red Hat) }
26600ccbf2fSSteven Rostedt (Red Hat) 
2672b499381SSteven Rostedt static void update_ftrace_function(void)
2682b499381SSteven Rostedt {
2692b499381SSteven Rostedt 	ftrace_func_t func;
2702b499381SSteven Rostedt 
271cdbe61bfSSteven Rostedt 	/*
272f7aad4e1SSteven Rostedt (Red Hat) 	 * Prepare the ftrace_ops that the arch callback will use.
273f7aad4e1SSteven Rostedt (Red Hat) 	 * If there's only one ftrace_ops registered, the ftrace_ops_list
274f7aad4e1SSteven Rostedt (Red Hat) 	 * will point to the ops we want.
275f7aad4e1SSteven Rostedt (Red Hat) 	 */
276f7aad4e1SSteven Rostedt (Red Hat) 	set_function_trace_op = ftrace_ops_list;
277f7aad4e1SSteven Rostedt (Red Hat) 
278f7aad4e1SSteven Rostedt (Red Hat) 	/* If there's no ftrace_ops registered, just call the stub function */
279f7aad4e1SSteven Rostedt (Red Hat) 	if (ftrace_ops_list == &ftrace_list_end) {
280f7aad4e1SSteven Rostedt (Red Hat) 		func = ftrace_stub;
281f7aad4e1SSteven Rostedt (Red Hat) 
282f7aad4e1SSteven Rostedt (Red Hat) 	/*
283cdbe61bfSSteven Rostedt 	 * If we are at the end of the list and this ops is
2844740974aSSteven Rostedt 	 * recursion safe and not dynamic and the arch supports passing ops,
2854740974aSSteven Rostedt 	 * then have the mcount trampoline call the function directly.
286cdbe61bfSSteven Rostedt 	 */
287f7aad4e1SSteven Rostedt (Red Hat) 	} else if (ftrace_ops_list->next == &ftrace_list_end) {
28800ccbf2fSSteven Rostedt (Red Hat) 		func = ftrace_ops_get_list_func(ftrace_ops_list);
289f7aad4e1SSteven Rostedt (Red Hat) 
2902f5f6ad9SSteven Rostedt 	} else {
2912f5f6ad9SSteven Rostedt 		/* Just use the default ftrace_ops */
292405e1d83SSteven Rostedt (Red Hat) 		set_function_trace_op = &ftrace_list_end;
293b848914cSSteven Rostedt 		func = ftrace_ops_list_func;
2942f5f6ad9SSteven Rostedt 	}
2952b499381SSteven Rostedt 
2965f8bf2d2SSteven Rostedt (Red Hat) 	update_function_graph_func();
2975f8bf2d2SSteven Rostedt (Red Hat) 
298405e1d83SSteven Rostedt (Red Hat) 	/* If there's no change, then do nothing more here */
299405e1d83SSteven Rostedt (Red Hat) 	if (ftrace_trace_function == func)
300405e1d83SSteven Rostedt (Red Hat) 		return;
301405e1d83SSteven Rostedt (Red Hat) 
302405e1d83SSteven Rostedt (Red Hat) 	/*
303405e1d83SSteven Rostedt (Red Hat) 	 * If we are using the list function, it doesn't care
304405e1d83SSteven Rostedt (Red Hat) 	 * about the function_trace_ops.
305405e1d83SSteven Rostedt (Red Hat) 	 */
306405e1d83SSteven Rostedt (Red Hat) 	if (func == ftrace_ops_list_func) {
307405e1d83SSteven Rostedt (Red Hat) 		ftrace_trace_function = func;
308405e1d83SSteven Rostedt (Red Hat) 		/*
309405e1d83SSteven Rostedt (Red Hat) 		 * Don't even bother setting function_trace_ops,
310405e1d83SSteven Rostedt (Red Hat) 		 * it would be racy to do so anyway.
311405e1d83SSteven Rostedt (Red Hat) 		 */
312405e1d83SSteven Rostedt (Red Hat) 		return;
313405e1d83SSteven Rostedt (Red Hat) 	}
314405e1d83SSteven Rostedt (Red Hat) 
315405e1d83SSteven Rostedt (Red Hat) #ifndef CONFIG_DYNAMIC_FTRACE
316405e1d83SSteven Rostedt (Red Hat) 	/*
317405e1d83SSteven Rostedt (Red Hat) 	 * For static tracing, we need to be a bit more careful.
318405e1d83SSteven Rostedt (Red Hat) 	 * The function change takes affect immediately. Thus,
319405e1d83SSteven Rostedt (Red Hat) 	 * we need to coorditate the setting of the function_trace_ops
320405e1d83SSteven Rostedt (Red Hat) 	 * with the setting of the ftrace_trace_function.
321405e1d83SSteven Rostedt (Red Hat) 	 *
322405e1d83SSteven Rostedt (Red Hat) 	 * Set the function to the list ops, which will call the
323405e1d83SSteven Rostedt (Red Hat) 	 * function we want, albeit indirectly, but it handles the
324405e1d83SSteven Rostedt (Red Hat) 	 * ftrace_ops and doesn't depend on function_trace_op.
325405e1d83SSteven Rostedt (Red Hat) 	 */
326405e1d83SSteven Rostedt (Red Hat) 	ftrace_trace_function = ftrace_ops_list_func;
327405e1d83SSteven Rostedt (Red Hat) 	/*
328405e1d83SSteven Rostedt (Red Hat) 	 * Make sure all CPUs see this. Yes this is slow, but static
329405e1d83SSteven Rostedt (Red Hat) 	 * tracing is slow and nasty to have enabled.
330405e1d83SSteven Rostedt (Red Hat) 	 */
331405e1d83SSteven Rostedt (Red Hat) 	schedule_on_each_cpu(ftrace_sync);
332405e1d83SSteven Rostedt (Red Hat) 	/* Now all cpus are using the list ops. */
333405e1d83SSteven Rostedt (Red Hat) 	function_trace_op = set_function_trace_op;
334405e1d83SSteven Rostedt (Red Hat) 	/* Make sure the function_trace_op is visible on all CPUs */
335405e1d83SSteven Rostedt (Red Hat) 	smp_wmb();
336405e1d83SSteven Rostedt (Red Hat) 	/* Nasty way to force a rmb on all cpus */
337405e1d83SSteven Rostedt (Red Hat) 	smp_call_function(ftrace_sync_ipi, NULL, 1);
338405e1d83SSteven Rostedt (Red Hat) 	/* OK, we are all set to update the ftrace_trace_function now! */
339405e1d83SSteven Rostedt (Red Hat) #endif /* !CONFIG_DYNAMIC_FTRACE */
340405e1d83SSteven Rostedt (Red Hat) 
341491d0dcfSSteven Rostedt 	ftrace_trace_function = func;
342491d0dcfSSteven Rostedt }
343491d0dcfSSteven Rostedt 
3447eea4fceSJiaxing Wang int using_ftrace_ops_list_func(void)
3457eea4fceSJiaxing Wang {
3467eea4fceSJiaxing Wang 	return ftrace_trace_function == ftrace_ops_list_func;
3477eea4fceSJiaxing Wang }
3487eea4fceSJiaxing Wang 
3492b499381SSteven Rostedt static void add_ftrace_ops(struct ftrace_ops **list, struct ftrace_ops *ops)
3503d083395SSteven Rostedt {
3512b499381SSteven Rostedt 	ops->next = *list;
35216444a8aSArnaldo Carvalho de Melo 	/*
353b848914cSSteven Rostedt 	 * We are entering ops into the list but another
35416444a8aSArnaldo Carvalho de Melo 	 * CPU might be walking that list. We need to make sure
35516444a8aSArnaldo Carvalho de Melo 	 * the ops->next pointer is valid before another CPU sees
356b848914cSSteven Rostedt 	 * the ops pointer included into the list.
35716444a8aSArnaldo Carvalho de Melo 	 */
3582b499381SSteven Rostedt 	rcu_assign_pointer(*list, ops);
3592b499381SSteven Rostedt }
3603d083395SSteven Rostedt 
3612b499381SSteven Rostedt static int remove_ftrace_ops(struct ftrace_ops **list, struct ftrace_ops *ops)
3622b499381SSteven Rostedt {
3632b499381SSteven Rostedt 	struct ftrace_ops **p;
3642b499381SSteven Rostedt 
3652b499381SSteven Rostedt 	/*
3662b499381SSteven Rostedt 	 * If we are removing the last function, then simply point
3672b499381SSteven Rostedt 	 * to the ftrace_stub.
3682b499381SSteven Rostedt 	 */
3692b499381SSteven Rostedt 	if (*list == ops && ops->next == &ftrace_list_end) {
3702b499381SSteven Rostedt 		*list = &ftrace_list_end;
3712b499381SSteven Rostedt 		return 0;
3722b499381SSteven Rostedt 	}
3732b499381SSteven Rostedt 
3742b499381SSteven Rostedt 	for (p = list; *p != &ftrace_list_end; p = &(*p)->next)
3752b499381SSteven Rostedt 		if (*p == ops)
3762b499381SSteven Rostedt 			break;
3772b499381SSteven Rostedt 
3782b499381SSteven Rostedt 	if (*p != ops)
3792b499381SSteven Rostedt 		return -1;
3802b499381SSteven Rostedt 
3812b499381SSteven Rostedt 	*p = (*p)->next;
3822b499381SSteven Rostedt 	return 0;
3832b499381SSteven Rostedt }
3842b499381SSteven Rostedt 
385f3bea491SSteven Rostedt (Red Hat) static void ftrace_update_trampoline(struct ftrace_ops *ops);
386f3bea491SSteven Rostedt (Red Hat) 
3872b499381SSteven Rostedt static int __register_ftrace_function(struct ftrace_ops *ops)
3882b499381SSteven Rostedt {
389591dffdaSSteven Rostedt (Red Hat) 	if (ops->flags & FTRACE_OPS_FL_DELETED)
390591dffdaSSteven Rostedt (Red Hat) 		return -EINVAL;
391591dffdaSSteven Rostedt (Red Hat) 
392b848914cSSteven Rostedt 	if (WARN_ON(ops->flags & FTRACE_OPS_FL_ENABLED))
393b848914cSSteven Rostedt 		return -EBUSY;
394b848914cSSteven Rostedt 
39506aeaaeaSMasami Hiramatsu #ifndef CONFIG_DYNAMIC_FTRACE_WITH_REGS
39608f6fba5SSteven Rostedt 	/*
39708f6fba5SSteven Rostedt 	 * If the ftrace_ops specifies SAVE_REGS, then it only can be used
39808f6fba5SSteven Rostedt 	 * if the arch supports it, or SAVE_REGS_IF_SUPPORTED is also set.
39908f6fba5SSteven Rostedt 	 * Setting SAVE_REGS_IF_SUPPORTED makes SAVE_REGS irrelevant.
40008f6fba5SSteven Rostedt 	 */
40108f6fba5SSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_SAVE_REGS &&
40208f6fba5SSteven Rostedt 	    !(ops->flags & FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED))
40308f6fba5SSteven Rostedt 		return -EINVAL;
40408f6fba5SSteven Rostedt 
40508f6fba5SSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED)
40608f6fba5SSteven Rostedt 		ops->flags |= FTRACE_OPS_FL_SAVE_REGS;
40708f6fba5SSteven Rostedt #endif
40808f6fba5SSteven Rostedt 
409cdbe61bfSSteven Rostedt 	if (!core_kernel_data((unsigned long)ops))
410cdbe61bfSSteven Rostedt 		ops->flags |= FTRACE_OPS_FL_DYNAMIC;
411cdbe61bfSSteven Rostedt 
412ba27f2bcSSteven Rostedt (Red Hat) 	if (ops->flags & FTRACE_OPS_FL_PER_CPU) {
413ba27f2bcSSteven Rostedt (Red Hat) 		if (per_cpu_ops_alloc(ops))
414e248491aSJiri Olsa 			return -ENOMEM;
415ba27f2bcSSteven Rostedt (Red Hat) 	}
416ba27f2bcSSteven Rostedt (Red Hat) 
417b848914cSSteven Rostedt 	add_ftrace_ops(&ftrace_ops_list, ops);
418b848914cSSteven Rostedt 
419e3eea140SSteven Rostedt (Red Hat) 	/* Always save the function, and reset at unregistering */
420e3eea140SSteven Rostedt (Red Hat) 	ops->saved_func = ops->func;
421e3eea140SSteven Rostedt (Red Hat) 
422*345ddcc8SSteven Rostedt (Red Hat) 	if (ftrace_pids_enabled(ops))
423e3eea140SSteven Rostedt (Red Hat) 		ops->func = ftrace_pid_func;
424e3eea140SSteven Rostedt (Red Hat) 
425f3bea491SSteven Rostedt (Red Hat) 	ftrace_update_trampoline(ops);
426f3bea491SSteven Rostedt (Red Hat) 
427491d0dcfSSteven Rostedt 	if (ftrace_enabled)
428491d0dcfSSteven Rostedt 		update_ftrace_function();
4293d083395SSteven Rostedt 
43016444a8aSArnaldo Carvalho de Melo 	return 0;
43116444a8aSArnaldo Carvalho de Melo }
43216444a8aSArnaldo Carvalho de Melo 
433e309b41dSIngo Molnar static int __unregister_ftrace_function(struct ftrace_ops *ops)
43416444a8aSArnaldo Carvalho de Melo {
4352b499381SSteven Rostedt 	int ret;
43616444a8aSArnaldo Carvalho de Melo 
437b848914cSSteven Rostedt 	if (WARN_ON(!(ops->flags & FTRACE_OPS_FL_ENABLED)))
438b848914cSSteven Rostedt 		return -EBUSY;
439b848914cSSteven Rostedt 
440b848914cSSteven Rostedt 	ret = remove_ftrace_ops(&ftrace_ops_list, ops);
441b848914cSSteven Rostedt 
4422b499381SSteven Rostedt 	if (ret < 0)
4432b499381SSteven Rostedt 		return ret;
444b848914cSSteven Rostedt 
445491d0dcfSSteven Rostedt 	if (ftrace_enabled)
446491d0dcfSSteven Rostedt 		update_ftrace_function();
44716444a8aSArnaldo Carvalho de Melo 
448e3eea140SSteven Rostedt (Red Hat) 	ops->func = ops->saved_func;
449e3eea140SSteven Rostedt (Red Hat) 
450e6ea44e9SSteven Rostedt 	return 0;
4513d083395SSteven Rostedt }
4523d083395SSteven Rostedt 
453df4fc315SSteven Rostedt static void ftrace_update_pid_func(void)
454df4fc315SSteven Rostedt {
455e3eea140SSteven Rostedt (Red Hat) 	struct ftrace_ops *op;
456e3eea140SSteven Rostedt (Red Hat) 
457491d0dcfSSteven Rostedt 	/* Only do something if we are tracing something */
458df4fc315SSteven Rostedt 	if (ftrace_trace_function == ftrace_stub)
45910dd3ebeSKOSAKI Motohiro 		return;
460df4fc315SSteven Rostedt 
461e3eea140SSteven Rostedt (Red Hat) 	do_for_each_ftrace_op(op, ftrace_ops_list) {
462e3eea140SSteven Rostedt (Red Hat) 		if (op->flags & FTRACE_OPS_FL_PID) {
463*345ddcc8SSteven Rostedt (Red Hat) 			op->func = ftrace_pids_enabled(op) ?
464*345ddcc8SSteven Rostedt (Red Hat) 				ftrace_pid_func : op->saved_func;
465e3eea140SSteven Rostedt (Red Hat) 			ftrace_update_trampoline(op);
466e3eea140SSteven Rostedt (Red Hat) 		}
467e3eea140SSteven Rostedt (Red Hat) 	} while_for_each_ftrace_op(op);
468e3eea140SSteven Rostedt (Red Hat) 
469491d0dcfSSteven Rostedt 	update_ftrace_function();
470df4fc315SSteven Rostedt }
471df4fc315SSteven Rostedt 
472493762fcSSteven Rostedt #ifdef CONFIG_FUNCTION_PROFILER
473493762fcSSteven Rostedt struct ftrace_profile {
474493762fcSSteven Rostedt 	struct hlist_node		node;
475493762fcSSteven Rostedt 	unsigned long			ip;
476493762fcSSteven Rostedt 	unsigned long			counter;
4770706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
4780706f1c4SSteven Rostedt 	unsigned long long		time;
479e330b3bcSChase Douglas 	unsigned long long		time_squared;
4800706f1c4SSteven Rostedt #endif
481493762fcSSteven Rostedt };
482493762fcSSteven Rostedt 
483493762fcSSteven Rostedt struct ftrace_profile_page {
484493762fcSSteven Rostedt 	struct ftrace_profile_page	*next;
485493762fcSSteven Rostedt 	unsigned long			index;
486493762fcSSteven Rostedt 	struct ftrace_profile		records[];
487493762fcSSteven Rostedt };
488493762fcSSteven Rostedt 
489cafb168aSSteven Rostedt struct ftrace_profile_stat {
490cafb168aSSteven Rostedt 	atomic_t			disabled;
491cafb168aSSteven Rostedt 	struct hlist_head		*hash;
492cafb168aSSteven Rostedt 	struct ftrace_profile_page	*pages;
493cafb168aSSteven Rostedt 	struct ftrace_profile_page	*start;
494cafb168aSSteven Rostedt 	struct tracer_stat		stat;
495cafb168aSSteven Rostedt };
496cafb168aSSteven Rostedt 
497493762fcSSteven Rostedt #define PROFILE_RECORDS_SIZE						\
498493762fcSSteven Rostedt 	(PAGE_SIZE - offsetof(struct ftrace_profile_page, records))
499493762fcSSteven Rostedt 
500493762fcSSteven Rostedt #define PROFILES_PER_PAGE					\
501493762fcSSteven Rostedt 	(PROFILE_RECORDS_SIZE / sizeof(struct ftrace_profile))
502493762fcSSteven Rostedt 
503fb9fb015SSteven Rostedt static int ftrace_profile_enabled __read_mostly;
504fb9fb015SSteven Rostedt 
505fb9fb015SSteven Rostedt /* ftrace_profile_lock - synchronize the enable and disable of the profiler */
506493762fcSSteven Rostedt static DEFINE_MUTEX(ftrace_profile_lock);
507493762fcSSteven Rostedt 
508cafb168aSSteven Rostedt static DEFINE_PER_CPU(struct ftrace_profile_stat, ftrace_profile_stats);
509493762fcSSteven Rostedt 
51020079ebeSNamhyung Kim #define FTRACE_PROFILE_HASH_BITS 10
51120079ebeSNamhyung Kim #define FTRACE_PROFILE_HASH_SIZE (1 << FTRACE_PROFILE_HASH_BITS)
512493762fcSSteven Rostedt 
513493762fcSSteven Rostedt static void *
514493762fcSSteven Rostedt function_stat_next(void *v, int idx)
515493762fcSSteven Rostedt {
516493762fcSSteven Rostedt 	struct ftrace_profile *rec = v;
517493762fcSSteven Rostedt 	struct ftrace_profile_page *pg;
518493762fcSSteven Rostedt 
519493762fcSSteven Rostedt 	pg = (struct ftrace_profile_page *)((unsigned long)rec & PAGE_MASK);
520493762fcSSteven Rostedt 
521493762fcSSteven Rostedt  again:
5220296e425SLi Zefan 	if (idx != 0)
523493762fcSSteven Rostedt 		rec++;
5240296e425SLi Zefan 
525493762fcSSteven Rostedt 	if ((void *)rec >= (void *)&pg->records[pg->index]) {
526493762fcSSteven Rostedt 		pg = pg->next;
527493762fcSSteven Rostedt 		if (!pg)
528493762fcSSteven Rostedt 			return NULL;
529493762fcSSteven Rostedt 		rec = &pg->records[0];
530493762fcSSteven Rostedt 		if (!rec->counter)
531493762fcSSteven Rostedt 			goto again;
532493762fcSSteven Rostedt 	}
533493762fcSSteven Rostedt 
534493762fcSSteven Rostedt 	return rec;
535493762fcSSteven Rostedt }
536493762fcSSteven Rostedt 
537493762fcSSteven Rostedt static void *function_stat_start(struct tracer_stat *trace)
538493762fcSSteven Rostedt {
539cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat =
540cafb168aSSteven Rostedt 		container_of(trace, struct ftrace_profile_stat, stat);
541cafb168aSSteven Rostedt 
542cafb168aSSteven Rostedt 	if (!stat || !stat->start)
543cafb168aSSteven Rostedt 		return NULL;
544cafb168aSSteven Rostedt 
545cafb168aSSteven Rostedt 	return function_stat_next(&stat->start->records[0], 0);
546493762fcSSteven Rostedt }
547493762fcSSteven Rostedt 
5480706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
5490706f1c4SSteven Rostedt /* function graph compares on total time */
5500706f1c4SSteven Rostedt static int function_stat_cmp(void *p1, void *p2)
5510706f1c4SSteven Rostedt {
5520706f1c4SSteven Rostedt 	struct ftrace_profile *a = p1;
5530706f1c4SSteven Rostedt 	struct ftrace_profile *b = p2;
5540706f1c4SSteven Rostedt 
5550706f1c4SSteven Rostedt 	if (a->time < b->time)
5560706f1c4SSteven Rostedt 		return -1;
5570706f1c4SSteven Rostedt 	if (a->time > b->time)
5580706f1c4SSteven Rostedt 		return 1;
5590706f1c4SSteven Rostedt 	else
5600706f1c4SSteven Rostedt 		return 0;
5610706f1c4SSteven Rostedt }
5620706f1c4SSteven Rostedt #else
5630706f1c4SSteven Rostedt /* not function graph compares against hits */
564493762fcSSteven Rostedt static int function_stat_cmp(void *p1, void *p2)
565493762fcSSteven Rostedt {
566493762fcSSteven Rostedt 	struct ftrace_profile *a = p1;
567493762fcSSteven Rostedt 	struct ftrace_profile *b = p2;
568493762fcSSteven Rostedt 
569493762fcSSteven Rostedt 	if (a->counter < b->counter)
570493762fcSSteven Rostedt 		return -1;
571493762fcSSteven Rostedt 	if (a->counter > b->counter)
572493762fcSSteven Rostedt 		return 1;
573493762fcSSteven Rostedt 	else
574493762fcSSteven Rostedt 		return 0;
575493762fcSSteven Rostedt }
5760706f1c4SSteven Rostedt #endif
577493762fcSSteven Rostedt 
578493762fcSSteven Rostedt static int function_stat_headers(struct seq_file *m)
579493762fcSSteven Rostedt {
5800706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
581fa6f0cc7SRasmus Villemoes 	seq_puts(m, "  Function                               "
582e330b3bcSChase Douglas 		 "Hit    Time            Avg             s^2\n"
58334886c8bSSteven Rostedt 		    "  --------                               "
584e330b3bcSChase Douglas 		 "---    ----            ---             ---\n");
5850706f1c4SSteven Rostedt #else
586fa6f0cc7SRasmus Villemoes 	seq_puts(m, "  Function                               Hit\n"
587493762fcSSteven Rostedt 		    "  --------                               ---\n");
5880706f1c4SSteven Rostedt #endif
589493762fcSSteven Rostedt 	return 0;
590493762fcSSteven Rostedt }
591493762fcSSteven Rostedt 
592493762fcSSteven Rostedt static int function_stat_show(struct seq_file *m, void *v)
593493762fcSSteven Rostedt {
594493762fcSSteven Rostedt 	struct ftrace_profile *rec = v;
595493762fcSSteven Rostedt 	char str[KSYM_SYMBOL_LEN];
5963aaba20fSLi Zefan 	int ret = 0;
5970706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
59834886c8bSSteven Rostedt 	static struct trace_seq s;
59934886c8bSSteven Rostedt 	unsigned long long avg;
600e330b3bcSChase Douglas 	unsigned long long stddev;
6010706f1c4SSteven Rostedt #endif
6023aaba20fSLi Zefan 	mutex_lock(&ftrace_profile_lock);
6033aaba20fSLi Zefan 
6043aaba20fSLi Zefan 	/* we raced with function_profile_reset() */
6053aaba20fSLi Zefan 	if (unlikely(rec->counter == 0)) {
6063aaba20fSLi Zefan 		ret = -EBUSY;
6073aaba20fSLi Zefan 		goto out;
6083aaba20fSLi Zefan 	}
609493762fcSSteven Rostedt 
6108e436ca0SUmesh Tiwari #ifdef CONFIG_FUNCTION_GRAPH_TRACER
6118e436ca0SUmesh Tiwari 	avg = rec->time;
6128e436ca0SUmesh Tiwari 	do_div(avg, rec->counter);
6138e436ca0SUmesh Tiwari 	if (tracing_thresh && (avg < tracing_thresh))
6148e436ca0SUmesh Tiwari 		goto out;
6158e436ca0SUmesh Tiwari #endif
6168e436ca0SUmesh Tiwari 
617493762fcSSteven Rostedt 	kallsyms_lookup(rec->ip, NULL, NULL, NULL, str);
6180706f1c4SSteven Rostedt 	seq_printf(m, "  %-30.30s  %10lu", str, rec->counter);
619493762fcSSteven Rostedt 
6200706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
621fa6f0cc7SRasmus Villemoes 	seq_puts(m, "    ");
62234886c8bSSteven Rostedt 
623e330b3bcSChase Douglas 	/* Sample standard deviation (s^2) */
624e330b3bcSChase Douglas 	if (rec->counter <= 1)
625e330b3bcSChase Douglas 		stddev = 0;
626e330b3bcSChase Douglas 	else {
62752d85d76SJuri Lelli 		/*
62852d85d76SJuri Lelli 		 * Apply Welford's method:
62952d85d76SJuri Lelli 		 * s^2 = 1 / (n * (n-1)) * (n * \Sum (x_i)^2 - (\Sum x_i)^2)
63052d85d76SJuri Lelli 		 */
63152d85d76SJuri Lelli 		stddev = rec->counter * rec->time_squared -
63252d85d76SJuri Lelli 			 rec->time * rec->time;
63352d85d76SJuri Lelli 
634e330b3bcSChase Douglas 		/*
635e330b3bcSChase Douglas 		 * Divide only 1000 for ns^2 -> us^2 conversion.
636e330b3bcSChase Douglas 		 * trace_print_graph_duration will divide 1000 again.
637e330b3bcSChase Douglas 		 */
63852d85d76SJuri Lelli 		do_div(stddev, rec->counter * (rec->counter - 1) * 1000);
639e330b3bcSChase Douglas 	}
640e330b3bcSChase Douglas 
64134886c8bSSteven Rostedt 	trace_seq_init(&s);
64234886c8bSSteven Rostedt 	trace_print_graph_duration(rec->time, &s);
64334886c8bSSteven Rostedt 	trace_seq_puts(&s, "    ");
64434886c8bSSteven Rostedt 	trace_print_graph_duration(avg, &s);
645e330b3bcSChase Douglas 	trace_seq_puts(&s, "    ");
646e330b3bcSChase Douglas 	trace_print_graph_duration(stddev, &s);
6470706f1c4SSteven Rostedt 	trace_print_seq(m, &s);
6480706f1c4SSteven Rostedt #endif
6490706f1c4SSteven Rostedt 	seq_putc(m, '\n');
6503aaba20fSLi Zefan out:
6513aaba20fSLi Zefan 	mutex_unlock(&ftrace_profile_lock);
6520706f1c4SSteven Rostedt 
6533aaba20fSLi Zefan 	return ret;
654493762fcSSteven Rostedt }
655493762fcSSteven Rostedt 
656cafb168aSSteven Rostedt static void ftrace_profile_reset(struct ftrace_profile_stat *stat)
657493762fcSSteven Rostedt {
658493762fcSSteven Rostedt 	struct ftrace_profile_page *pg;
659493762fcSSteven Rostedt 
660cafb168aSSteven Rostedt 	pg = stat->pages = stat->start;
661493762fcSSteven Rostedt 
662493762fcSSteven Rostedt 	while (pg) {
663493762fcSSteven Rostedt 		memset(pg->records, 0, PROFILE_RECORDS_SIZE);
664493762fcSSteven Rostedt 		pg->index = 0;
665493762fcSSteven Rostedt 		pg = pg->next;
666493762fcSSteven Rostedt 	}
667493762fcSSteven Rostedt 
668cafb168aSSteven Rostedt 	memset(stat->hash, 0,
669493762fcSSteven Rostedt 	       FTRACE_PROFILE_HASH_SIZE * sizeof(struct hlist_head));
670493762fcSSteven Rostedt }
671493762fcSSteven Rostedt 
672cafb168aSSteven Rostedt int ftrace_profile_pages_init(struct ftrace_profile_stat *stat)
673493762fcSSteven Rostedt {
674493762fcSSteven Rostedt 	struct ftrace_profile_page *pg;
675318e0a73SSteven Rostedt 	int functions;
676318e0a73SSteven Rostedt 	int pages;
677493762fcSSteven Rostedt 	int i;
678493762fcSSteven Rostedt 
679493762fcSSteven Rostedt 	/* If we already allocated, do nothing */
680cafb168aSSteven Rostedt 	if (stat->pages)
681493762fcSSteven Rostedt 		return 0;
682493762fcSSteven Rostedt 
683cafb168aSSteven Rostedt 	stat->pages = (void *)get_zeroed_page(GFP_KERNEL);
684cafb168aSSteven Rostedt 	if (!stat->pages)
685493762fcSSteven Rostedt 		return -ENOMEM;
686493762fcSSteven Rostedt 
687318e0a73SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
688318e0a73SSteven Rostedt 	functions = ftrace_update_tot_cnt;
689318e0a73SSteven Rostedt #else
690318e0a73SSteven Rostedt 	/*
691318e0a73SSteven Rostedt 	 * We do not know the number of functions that exist because
692318e0a73SSteven Rostedt 	 * dynamic tracing is what counts them. With past experience
693318e0a73SSteven Rostedt 	 * we have around 20K functions. That should be more than enough.
694318e0a73SSteven Rostedt 	 * It is highly unlikely we will execute every function in
695318e0a73SSteven Rostedt 	 * the kernel.
696318e0a73SSteven Rostedt 	 */
697318e0a73SSteven Rostedt 	functions = 20000;
698318e0a73SSteven Rostedt #endif
699318e0a73SSteven Rostedt 
700cafb168aSSteven Rostedt 	pg = stat->start = stat->pages;
701493762fcSSteven Rostedt 
702318e0a73SSteven Rostedt 	pages = DIV_ROUND_UP(functions, PROFILES_PER_PAGE);
703318e0a73SSteven Rostedt 
70439e30cd1SNamhyung Kim 	for (i = 1; i < pages; i++) {
705493762fcSSteven Rostedt 		pg->next = (void *)get_zeroed_page(GFP_KERNEL);
706493762fcSSteven Rostedt 		if (!pg->next)
707318e0a73SSteven Rostedt 			goto out_free;
708493762fcSSteven Rostedt 		pg = pg->next;
709493762fcSSteven Rostedt 	}
710493762fcSSteven Rostedt 
711493762fcSSteven Rostedt 	return 0;
712318e0a73SSteven Rostedt 
713318e0a73SSteven Rostedt  out_free:
714318e0a73SSteven Rostedt 	pg = stat->start;
715318e0a73SSteven Rostedt 	while (pg) {
716318e0a73SSteven Rostedt 		unsigned long tmp = (unsigned long)pg;
717318e0a73SSteven Rostedt 
718318e0a73SSteven Rostedt 		pg = pg->next;
719318e0a73SSteven Rostedt 		free_page(tmp);
720318e0a73SSteven Rostedt 	}
721318e0a73SSteven Rostedt 
722318e0a73SSteven Rostedt 	stat->pages = NULL;
723318e0a73SSteven Rostedt 	stat->start = NULL;
724318e0a73SSteven Rostedt 
725318e0a73SSteven Rostedt 	return -ENOMEM;
726493762fcSSteven Rostedt }
727493762fcSSteven Rostedt 
728cafb168aSSteven Rostedt static int ftrace_profile_init_cpu(int cpu)
729493762fcSSteven Rostedt {
730cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat;
731493762fcSSteven Rostedt 	int size;
732493762fcSSteven Rostedt 
733cafb168aSSteven Rostedt 	stat = &per_cpu(ftrace_profile_stats, cpu);
734cafb168aSSteven Rostedt 
735cafb168aSSteven Rostedt 	if (stat->hash) {
736493762fcSSteven Rostedt 		/* If the profile is already created, simply reset it */
737cafb168aSSteven Rostedt 		ftrace_profile_reset(stat);
738493762fcSSteven Rostedt 		return 0;
739493762fcSSteven Rostedt 	}
740493762fcSSteven Rostedt 
741493762fcSSteven Rostedt 	/*
742493762fcSSteven Rostedt 	 * We are profiling all functions, but usually only a few thousand
743493762fcSSteven Rostedt 	 * functions are hit. We'll make a hash of 1024 items.
744493762fcSSteven Rostedt 	 */
745493762fcSSteven Rostedt 	size = FTRACE_PROFILE_HASH_SIZE;
746493762fcSSteven Rostedt 
747cafb168aSSteven Rostedt 	stat->hash = kzalloc(sizeof(struct hlist_head) * size, GFP_KERNEL);
748493762fcSSteven Rostedt 
749cafb168aSSteven Rostedt 	if (!stat->hash)
750493762fcSSteven Rostedt 		return -ENOMEM;
751493762fcSSteven Rostedt 
752318e0a73SSteven Rostedt 	/* Preallocate the function profiling pages */
753cafb168aSSteven Rostedt 	if (ftrace_profile_pages_init(stat) < 0) {
754cafb168aSSteven Rostedt 		kfree(stat->hash);
755cafb168aSSteven Rostedt 		stat->hash = NULL;
756493762fcSSteven Rostedt 		return -ENOMEM;
757493762fcSSteven Rostedt 	}
758493762fcSSteven Rostedt 
759493762fcSSteven Rostedt 	return 0;
760493762fcSSteven Rostedt }
761493762fcSSteven Rostedt 
762cafb168aSSteven Rostedt static int ftrace_profile_init(void)
763cafb168aSSteven Rostedt {
764cafb168aSSteven Rostedt 	int cpu;
765cafb168aSSteven Rostedt 	int ret = 0;
766cafb168aSSteven Rostedt 
767c4602c1cSMiao Xie 	for_each_possible_cpu(cpu) {
768cafb168aSSteven Rostedt 		ret = ftrace_profile_init_cpu(cpu);
769cafb168aSSteven Rostedt 		if (ret)
770cafb168aSSteven Rostedt 			break;
771cafb168aSSteven Rostedt 	}
772cafb168aSSteven Rostedt 
773cafb168aSSteven Rostedt 	return ret;
774cafb168aSSteven Rostedt }
775cafb168aSSteven Rostedt 
776493762fcSSteven Rostedt /* interrupts must be disabled */
777cafb168aSSteven Rostedt static struct ftrace_profile *
778cafb168aSSteven Rostedt ftrace_find_profiled_func(struct ftrace_profile_stat *stat, unsigned long ip)
779493762fcSSteven Rostedt {
780493762fcSSteven Rostedt 	struct ftrace_profile *rec;
781493762fcSSteven Rostedt 	struct hlist_head *hhd;
782493762fcSSteven Rostedt 	unsigned long key;
783493762fcSSteven Rostedt 
78420079ebeSNamhyung Kim 	key = hash_long(ip, FTRACE_PROFILE_HASH_BITS);
785cafb168aSSteven Rostedt 	hhd = &stat->hash[key];
786493762fcSSteven Rostedt 
787493762fcSSteven Rostedt 	if (hlist_empty(hhd))
788493762fcSSteven Rostedt 		return NULL;
789493762fcSSteven Rostedt 
7901bb539caSSteven Rostedt 	hlist_for_each_entry_rcu_notrace(rec, hhd, node) {
791493762fcSSteven Rostedt 		if (rec->ip == ip)
792493762fcSSteven Rostedt 			return rec;
793493762fcSSteven Rostedt 	}
794493762fcSSteven Rostedt 
795493762fcSSteven Rostedt 	return NULL;
796493762fcSSteven Rostedt }
797493762fcSSteven Rostedt 
798cafb168aSSteven Rostedt static void ftrace_add_profile(struct ftrace_profile_stat *stat,
799cafb168aSSteven Rostedt 			       struct ftrace_profile *rec)
800493762fcSSteven Rostedt {
801493762fcSSteven Rostedt 	unsigned long key;
802493762fcSSteven Rostedt 
80320079ebeSNamhyung Kim 	key = hash_long(rec->ip, FTRACE_PROFILE_HASH_BITS);
804cafb168aSSteven Rostedt 	hlist_add_head_rcu(&rec->node, &stat->hash[key]);
805493762fcSSteven Rostedt }
806493762fcSSteven Rostedt 
807318e0a73SSteven Rostedt /*
808318e0a73SSteven Rostedt  * The memory is already allocated, this simply finds a new record to use.
809318e0a73SSteven Rostedt  */
810493762fcSSteven Rostedt static struct ftrace_profile *
811318e0a73SSteven Rostedt ftrace_profile_alloc(struct ftrace_profile_stat *stat, unsigned long ip)
812493762fcSSteven Rostedt {
813493762fcSSteven Rostedt 	struct ftrace_profile *rec = NULL;
814493762fcSSteven Rostedt 
815318e0a73SSteven Rostedt 	/* prevent recursion (from NMIs) */
816cafb168aSSteven Rostedt 	if (atomic_inc_return(&stat->disabled) != 1)
817493762fcSSteven Rostedt 		goto out;
818493762fcSSteven Rostedt 
819493762fcSSteven Rostedt 	/*
820318e0a73SSteven Rostedt 	 * Try to find the function again since an NMI
821318e0a73SSteven Rostedt 	 * could have added it
822493762fcSSteven Rostedt 	 */
823cafb168aSSteven Rostedt 	rec = ftrace_find_profiled_func(stat, ip);
824493762fcSSteven Rostedt 	if (rec)
825cafb168aSSteven Rostedt 		goto out;
826493762fcSSteven Rostedt 
827cafb168aSSteven Rostedt 	if (stat->pages->index == PROFILES_PER_PAGE) {
828cafb168aSSteven Rostedt 		if (!stat->pages->next)
829cafb168aSSteven Rostedt 			goto out;
830cafb168aSSteven Rostedt 		stat->pages = stat->pages->next;
831493762fcSSteven Rostedt 	}
832493762fcSSteven Rostedt 
833cafb168aSSteven Rostedt 	rec = &stat->pages->records[stat->pages->index++];
834493762fcSSteven Rostedt 	rec->ip = ip;
835cafb168aSSteven Rostedt 	ftrace_add_profile(stat, rec);
836493762fcSSteven Rostedt 
837493762fcSSteven Rostedt  out:
838cafb168aSSteven Rostedt 	atomic_dec(&stat->disabled);
839493762fcSSteven Rostedt 
840493762fcSSteven Rostedt 	return rec;
841493762fcSSteven Rostedt }
842493762fcSSteven Rostedt 
843493762fcSSteven Rostedt static void
8442f5f6ad9SSteven Rostedt function_profile_call(unsigned long ip, unsigned long parent_ip,
845a1e2e31dSSteven Rostedt 		      struct ftrace_ops *ops, struct pt_regs *regs)
846493762fcSSteven Rostedt {
847cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat;
848493762fcSSteven Rostedt 	struct ftrace_profile *rec;
849493762fcSSteven Rostedt 	unsigned long flags;
850493762fcSSteven Rostedt 
851493762fcSSteven Rostedt 	if (!ftrace_profile_enabled)
852493762fcSSteven Rostedt 		return;
853493762fcSSteven Rostedt 
854493762fcSSteven Rostedt 	local_irq_save(flags);
855cafb168aSSteven Rostedt 
856bdffd893SChristoph Lameter 	stat = this_cpu_ptr(&ftrace_profile_stats);
8570f6ce3deSSteven Rostedt 	if (!stat->hash || !ftrace_profile_enabled)
858cafb168aSSteven Rostedt 		goto out;
859cafb168aSSteven Rostedt 
860cafb168aSSteven Rostedt 	rec = ftrace_find_profiled_func(stat, ip);
861493762fcSSteven Rostedt 	if (!rec) {
862318e0a73SSteven Rostedt 		rec = ftrace_profile_alloc(stat, ip);
863493762fcSSteven Rostedt 		if (!rec)
864493762fcSSteven Rostedt 			goto out;
865493762fcSSteven Rostedt 	}
866493762fcSSteven Rostedt 
867493762fcSSteven Rostedt 	rec->counter++;
868493762fcSSteven Rostedt  out:
869493762fcSSteven Rostedt 	local_irq_restore(flags);
870493762fcSSteven Rostedt }
871493762fcSSteven Rostedt 
8720706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
8730706f1c4SSteven Rostedt static int profile_graph_entry(struct ftrace_graph_ent *trace)
8740706f1c4SSteven Rostedt {
875a1e2e31dSSteven Rostedt 	function_profile_call(trace->func, 0, NULL, NULL);
8760706f1c4SSteven Rostedt 	return 1;
8770706f1c4SSteven Rostedt }
8780706f1c4SSteven Rostedt 
8790706f1c4SSteven Rostedt static void profile_graph_return(struct ftrace_graph_ret *trace)
8800706f1c4SSteven Rostedt {
881cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat;
882a2a16d6aSSteven Rostedt 	unsigned long long calltime;
8830706f1c4SSteven Rostedt 	struct ftrace_profile *rec;
884cafb168aSSteven Rostedt 	unsigned long flags;
8850706f1c4SSteven Rostedt 
8860706f1c4SSteven Rostedt 	local_irq_save(flags);
887bdffd893SChristoph Lameter 	stat = this_cpu_ptr(&ftrace_profile_stats);
8880f6ce3deSSteven Rostedt 	if (!stat->hash || !ftrace_profile_enabled)
889cafb168aSSteven Rostedt 		goto out;
890cafb168aSSteven Rostedt 
89137e44bc5SSteven Rostedt 	/* If the calltime was zero'd ignore it */
89237e44bc5SSteven Rostedt 	if (!trace->calltime)
89337e44bc5SSteven Rostedt 		goto out;
89437e44bc5SSteven Rostedt 
895a2a16d6aSSteven Rostedt 	calltime = trace->rettime - trace->calltime;
896a2a16d6aSSteven Rostedt 
89755577204SSteven Rostedt (Red Hat) 	if (!fgraph_graph_time) {
898a2a16d6aSSteven Rostedt 		int index;
899a2a16d6aSSteven Rostedt 
900a2a16d6aSSteven Rostedt 		index = trace->depth;
901a2a16d6aSSteven Rostedt 
902a2a16d6aSSteven Rostedt 		/* Append this call time to the parent time to subtract */
903a2a16d6aSSteven Rostedt 		if (index)
904a2a16d6aSSteven Rostedt 			current->ret_stack[index - 1].subtime += calltime;
905a2a16d6aSSteven Rostedt 
906a2a16d6aSSteven Rostedt 		if (current->ret_stack[index].subtime < calltime)
907a2a16d6aSSteven Rostedt 			calltime -= current->ret_stack[index].subtime;
908a2a16d6aSSteven Rostedt 		else
909a2a16d6aSSteven Rostedt 			calltime = 0;
910a2a16d6aSSteven Rostedt 	}
911a2a16d6aSSteven Rostedt 
912cafb168aSSteven Rostedt 	rec = ftrace_find_profiled_func(stat, trace->func);
913e330b3bcSChase Douglas 	if (rec) {
914a2a16d6aSSteven Rostedt 		rec->time += calltime;
915e330b3bcSChase Douglas 		rec->time_squared += calltime * calltime;
916e330b3bcSChase Douglas 	}
917a2a16d6aSSteven Rostedt 
918cafb168aSSteven Rostedt  out:
9190706f1c4SSteven Rostedt 	local_irq_restore(flags);
9200706f1c4SSteven Rostedt }
9210706f1c4SSteven Rostedt 
9220706f1c4SSteven Rostedt static int register_ftrace_profiler(void)
9230706f1c4SSteven Rostedt {
9240706f1c4SSteven Rostedt 	return register_ftrace_graph(&profile_graph_return,
9250706f1c4SSteven Rostedt 				     &profile_graph_entry);
9260706f1c4SSteven Rostedt }
9270706f1c4SSteven Rostedt 
9280706f1c4SSteven Rostedt static void unregister_ftrace_profiler(void)
9290706f1c4SSteven Rostedt {
9300706f1c4SSteven Rostedt 	unregister_ftrace_graph();
9310706f1c4SSteven Rostedt }
9320706f1c4SSteven Rostedt #else
933bd38c0e6SPaul McQuade static struct ftrace_ops ftrace_profile_ops __read_mostly = {
934493762fcSSteven Rostedt 	.func		= function_profile_call,
935f04f24fbSMasami Hiramatsu 	.flags		= FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_INITIALIZED,
93633b7f99cSSteven Rostedt (Red Hat) 	INIT_OPS_HASH(ftrace_profile_ops)
937493762fcSSteven Rostedt };
938493762fcSSteven Rostedt 
9390706f1c4SSteven Rostedt static int register_ftrace_profiler(void)
9400706f1c4SSteven Rostedt {
9410706f1c4SSteven Rostedt 	return register_ftrace_function(&ftrace_profile_ops);
9420706f1c4SSteven Rostedt }
9430706f1c4SSteven Rostedt 
9440706f1c4SSteven Rostedt static void unregister_ftrace_profiler(void)
9450706f1c4SSteven Rostedt {
9460706f1c4SSteven Rostedt 	unregister_ftrace_function(&ftrace_profile_ops);
9470706f1c4SSteven Rostedt }
9480706f1c4SSteven Rostedt #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
9490706f1c4SSteven Rostedt 
950493762fcSSteven Rostedt static ssize_t
951493762fcSSteven Rostedt ftrace_profile_write(struct file *filp, const char __user *ubuf,
952493762fcSSteven Rostedt 		     size_t cnt, loff_t *ppos)
953493762fcSSteven Rostedt {
954493762fcSSteven Rostedt 	unsigned long val;
955493762fcSSteven Rostedt 	int ret;
956493762fcSSteven Rostedt 
95722fe9b54SPeter Huewe 	ret = kstrtoul_from_user(ubuf, cnt, 10, &val);
95822fe9b54SPeter Huewe 	if (ret)
959493762fcSSteven Rostedt 		return ret;
960493762fcSSteven Rostedt 
961493762fcSSteven Rostedt 	val = !!val;
962493762fcSSteven Rostedt 
963493762fcSSteven Rostedt 	mutex_lock(&ftrace_profile_lock);
964493762fcSSteven Rostedt 	if (ftrace_profile_enabled ^ val) {
965493762fcSSteven Rostedt 		if (val) {
966493762fcSSteven Rostedt 			ret = ftrace_profile_init();
967493762fcSSteven Rostedt 			if (ret < 0) {
968493762fcSSteven Rostedt 				cnt = ret;
969493762fcSSteven Rostedt 				goto out;
970493762fcSSteven Rostedt 			}
971493762fcSSteven Rostedt 
9720706f1c4SSteven Rostedt 			ret = register_ftrace_profiler();
9730706f1c4SSteven Rostedt 			if (ret < 0) {
9740706f1c4SSteven Rostedt 				cnt = ret;
9750706f1c4SSteven Rostedt 				goto out;
9760706f1c4SSteven Rostedt 			}
977493762fcSSteven Rostedt 			ftrace_profile_enabled = 1;
978493762fcSSteven Rostedt 		} else {
979493762fcSSteven Rostedt 			ftrace_profile_enabled = 0;
9800f6ce3deSSteven Rostedt 			/*
9810f6ce3deSSteven Rostedt 			 * unregister_ftrace_profiler calls stop_machine
9820f6ce3deSSteven Rostedt 			 * so this acts like an synchronize_sched.
9830f6ce3deSSteven Rostedt 			 */
9840706f1c4SSteven Rostedt 			unregister_ftrace_profiler();
985493762fcSSteven Rostedt 		}
986493762fcSSteven Rostedt 	}
987493762fcSSteven Rostedt  out:
988493762fcSSteven Rostedt 	mutex_unlock(&ftrace_profile_lock);
989493762fcSSteven Rostedt 
990cf8517cfSJiri Olsa 	*ppos += cnt;
991493762fcSSteven Rostedt 
992493762fcSSteven Rostedt 	return cnt;
993493762fcSSteven Rostedt }
994493762fcSSteven Rostedt 
995493762fcSSteven Rostedt static ssize_t
996493762fcSSteven Rostedt ftrace_profile_read(struct file *filp, char __user *ubuf,
997493762fcSSteven Rostedt 		     size_t cnt, loff_t *ppos)
998493762fcSSteven Rostedt {
999fb9fb015SSteven Rostedt 	char buf[64];		/* big enough to hold a number */
1000493762fcSSteven Rostedt 	int r;
1001493762fcSSteven Rostedt 
1002493762fcSSteven Rostedt 	r = sprintf(buf, "%u\n", ftrace_profile_enabled);
1003493762fcSSteven Rostedt 	return simple_read_from_buffer(ubuf, cnt, ppos, buf, r);
1004493762fcSSteven Rostedt }
1005493762fcSSteven Rostedt 
1006493762fcSSteven Rostedt static const struct file_operations ftrace_profile_fops = {
1007493762fcSSteven Rostedt 	.open		= tracing_open_generic,
1008493762fcSSteven Rostedt 	.read		= ftrace_profile_read,
1009493762fcSSteven Rostedt 	.write		= ftrace_profile_write,
10106038f373SArnd Bergmann 	.llseek		= default_llseek,
1011493762fcSSteven Rostedt };
1012493762fcSSteven Rostedt 
1013cafb168aSSteven Rostedt /* used to initialize the real stat files */
1014cafb168aSSteven Rostedt static struct tracer_stat function_stats __initdata = {
1015cafb168aSSteven Rostedt 	.name		= "functions",
1016cafb168aSSteven Rostedt 	.stat_start	= function_stat_start,
1017cafb168aSSteven Rostedt 	.stat_next	= function_stat_next,
1018cafb168aSSteven Rostedt 	.stat_cmp	= function_stat_cmp,
1019cafb168aSSteven Rostedt 	.stat_headers	= function_stat_headers,
1020cafb168aSSteven Rostedt 	.stat_show	= function_stat_show
1021cafb168aSSteven Rostedt };
1022cafb168aSSteven Rostedt 
10238434dc93SSteven Rostedt (Red Hat) static __init void ftrace_profile_tracefs(struct dentry *d_tracer)
1024493762fcSSteven Rostedt {
1025cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat;
1026493762fcSSteven Rostedt 	struct dentry *entry;
1027cafb168aSSteven Rostedt 	char *name;
1028493762fcSSteven Rostedt 	int ret;
1029cafb168aSSteven Rostedt 	int cpu;
1030493762fcSSteven Rostedt 
1031cafb168aSSteven Rostedt 	for_each_possible_cpu(cpu) {
1032cafb168aSSteven Rostedt 		stat = &per_cpu(ftrace_profile_stats, cpu);
1033cafb168aSSteven Rostedt 
10346363c6b5SGeliang Tang 		name = kasprintf(GFP_KERNEL, "function%d", cpu);
1035cafb168aSSteven Rostedt 		if (!name) {
1036cafb168aSSteven Rostedt 			/*
1037cafb168aSSteven Rostedt 			 * The files created are permanent, if something happens
1038cafb168aSSteven Rostedt 			 * we still do not free memory.
1039cafb168aSSteven Rostedt 			 */
1040cafb168aSSteven Rostedt 			WARN(1,
1041cafb168aSSteven Rostedt 			     "Could not allocate stat file for cpu %d\n",
1042cafb168aSSteven Rostedt 			     cpu);
1043493762fcSSteven Rostedt 			return;
1044493762fcSSteven Rostedt 		}
1045cafb168aSSteven Rostedt 		stat->stat = function_stats;
1046cafb168aSSteven Rostedt 		stat->stat.name = name;
1047cafb168aSSteven Rostedt 		ret = register_stat_tracer(&stat->stat);
1048cafb168aSSteven Rostedt 		if (ret) {
1049cafb168aSSteven Rostedt 			WARN(1,
1050cafb168aSSteven Rostedt 			     "Could not register function stat for cpu %d\n",
1051cafb168aSSteven Rostedt 			     cpu);
1052cafb168aSSteven Rostedt 			kfree(name);
1053cafb168aSSteven Rostedt 			return;
1054cafb168aSSteven Rostedt 		}
1055cafb168aSSteven Rostedt 	}
1056493762fcSSteven Rostedt 
10578434dc93SSteven Rostedt (Red Hat) 	entry = tracefs_create_file("function_profile_enabled", 0644,
1058493762fcSSteven Rostedt 				    d_tracer, NULL, &ftrace_profile_fops);
1059493762fcSSteven Rostedt 	if (!entry)
1060a395d6a7SJoe Perches 		pr_warn("Could not create tracefs 'function_profile_enabled' entry\n");
1061493762fcSSteven Rostedt }
1062493762fcSSteven Rostedt 
1063493762fcSSteven Rostedt #else /* CONFIG_FUNCTION_PROFILER */
10648434dc93SSteven Rostedt (Red Hat) static __init void ftrace_profile_tracefs(struct dentry *d_tracer)
1065493762fcSSteven Rostedt {
1066493762fcSSteven Rostedt }
1067493762fcSSteven Rostedt #endif /* CONFIG_FUNCTION_PROFILER */
1068493762fcSSteven Rostedt 
106973d3fd96SIngo Molnar static struct pid * const ftrace_swapper_pid = &init_struct_pid;
107073d3fd96SIngo Molnar 
10711619dc3fSPratyush Anand #ifdef CONFIG_FUNCTION_GRAPH_TRACER
10721619dc3fSPratyush Anand static int ftrace_graph_active;
10731619dc3fSPratyush Anand #else
10741619dc3fSPratyush Anand # define ftrace_graph_active 0
10751619dc3fSPratyush Anand #endif
10761619dc3fSPratyush Anand 
10773d083395SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
107873d3fd96SIngo Molnar 
107979922b80SSteven Rostedt (Red Hat) static struct ftrace_ops *removed_ops;
108079922b80SSteven Rostedt (Red Hat) 
1081e1effa01SSteven Rostedt (Red Hat) /*
1082e1effa01SSteven Rostedt (Red Hat)  * Set when doing a global update, like enabling all recs or disabling them.
1083e1effa01SSteven Rostedt (Red Hat)  * It is not set when just updating a single ftrace_ops.
1084e1effa01SSteven Rostedt (Red Hat)  */
1085e1effa01SSteven Rostedt (Red Hat) static bool update_all_ops;
1086e1effa01SSteven Rostedt (Red Hat) 
108799ecdc43SSteven Rostedt #ifndef CONFIG_FTRACE_MCOUNT_RECORD
1088cb7be3b2SSteven Rostedt # error Dynamic ftrace depends on MCOUNT_RECORD
108999ecdc43SSteven Rostedt #endif
109099ecdc43SSteven Rostedt 
10918fc0c701SSteven Rostedt static struct hlist_head ftrace_func_hash[FTRACE_FUNC_HASHSIZE] __read_mostly;
10928fc0c701SSteven Rostedt 
1093b6887d79SSteven Rostedt struct ftrace_func_probe {
10948fc0c701SSteven Rostedt 	struct hlist_node	node;
1095b6887d79SSteven Rostedt 	struct ftrace_probe_ops	*ops;
10968fc0c701SSteven Rostedt 	unsigned long		flags;
10978fc0c701SSteven Rostedt 	unsigned long		ip;
10988fc0c701SSteven Rostedt 	void			*data;
10997818b388SSteven Rostedt (Red Hat) 	struct list_head	free_list;
11008fc0c701SSteven Rostedt };
11018fc0c701SSteven Rostedt 
1102b448c4e3SSteven Rostedt struct ftrace_func_entry {
1103b448c4e3SSteven Rostedt 	struct hlist_node hlist;
1104b448c4e3SSteven Rostedt 	unsigned long ip;
1105b448c4e3SSteven Rostedt };
1106b448c4e3SSteven Rostedt 
1107b448c4e3SSteven Rostedt struct ftrace_hash {
1108b448c4e3SSteven Rostedt 	unsigned long		size_bits;
1109b448c4e3SSteven Rostedt 	struct hlist_head	*buckets;
1110b448c4e3SSteven Rostedt 	unsigned long		count;
111107fd5515SSteven Rostedt 	struct rcu_head		rcu;
1112b448c4e3SSteven Rostedt };
1113b448c4e3SSteven Rostedt 
111433dc9b12SSteven Rostedt /*
111533dc9b12SSteven Rostedt  * We make these constant because no one should touch them,
111633dc9b12SSteven Rostedt  * but they are used as the default "empty hash", to avoid allocating
111733dc9b12SSteven Rostedt  * it all the time. These are in a read only section such that if
111833dc9b12SSteven Rostedt  * anyone does try to modify it, it will cause an exception.
111933dc9b12SSteven Rostedt  */
112033dc9b12SSteven Rostedt static const struct hlist_head empty_buckets[1];
112133dc9b12SSteven Rostedt static const struct ftrace_hash empty_hash = {
112233dc9b12SSteven Rostedt 	.buckets = (struct hlist_head *)empty_buckets,
1123b448c4e3SSteven Rostedt };
112433dc9b12SSteven Rostedt #define EMPTY_HASH	((struct ftrace_hash *)&empty_hash)
11255072c59fSSteven Rostedt 
11262b499381SSteven Rostedt static struct ftrace_ops global_ops = {
1127f45948e8SSteven Rostedt 	.func				= ftrace_stub,
112833b7f99cSSteven Rostedt (Red Hat) 	.local_hash.notrace_hash	= EMPTY_HASH,
112933b7f99cSSteven Rostedt (Red Hat) 	.local_hash.filter_hash		= EMPTY_HASH,
113033b7f99cSSteven Rostedt (Red Hat) 	INIT_OPS_HASH(global_ops)
113133b7f99cSSteven Rostedt (Red Hat) 	.flags				= FTRACE_OPS_FL_RECURSION_SAFE |
1132e3eea140SSteven Rostedt (Red Hat) 					  FTRACE_OPS_FL_INITIALIZED |
1133e3eea140SSteven Rostedt (Red Hat) 					  FTRACE_OPS_FL_PID,
1134f45948e8SSteven Rostedt };
1135f45948e8SSteven Rostedt 
1136aec0be2dSSteven Rostedt (Red Hat) /*
1137aec0be2dSSteven Rostedt (Red Hat)  * This is used by __kernel_text_address() to return true if the
11380af26492SSteven Rostedt (Red Hat)  * address is on a dynamically allocated trampoline that would
1139aec0be2dSSteven Rostedt (Red Hat)  * not return true for either core_kernel_text() or
1140aec0be2dSSteven Rostedt (Red Hat)  * is_module_text_address().
1141aec0be2dSSteven Rostedt (Red Hat)  */
1142aec0be2dSSteven Rostedt (Red Hat) bool is_ftrace_trampoline(unsigned long addr)
1143aec0be2dSSteven Rostedt (Red Hat) {
1144aec0be2dSSteven Rostedt (Red Hat) 	struct ftrace_ops *op;
1145aec0be2dSSteven Rostedt (Red Hat) 	bool ret = false;
1146aec0be2dSSteven Rostedt (Red Hat) 
1147aec0be2dSSteven Rostedt (Red Hat) 	/*
1148aec0be2dSSteven Rostedt (Red Hat) 	 * Some of the ops may be dynamically allocated,
1149aec0be2dSSteven Rostedt (Red Hat) 	 * they are freed after a synchronize_sched().
1150aec0be2dSSteven Rostedt (Red Hat) 	 */
1151aec0be2dSSteven Rostedt (Red Hat) 	preempt_disable_notrace();
1152aec0be2dSSteven Rostedt (Red Hat) 
1153aec0be2dSSteven Rostedt (Red Hat) 	do_for_each_ftrace_op(op, ftrace_ops_list) {
1154aec0be2dSSteven Rostedt (Red Hat) 		/*
1155aec0be2dSSteven Rostedt (Red Hat) 		 * This is to check for dynamically allocated trampolines.
1156aec0be2dSSteven Rostedt (Red Hat) 		 * Trampolines that are in kernel text will have
1157aec0be2dSSteven Rostedt (Red Hat) 		 * core_kernel_text() return true.
1158aec0be2dSSteven Rostedt (Red Hat) 		 */
1159aec0be2dSSteven Rostedt (Red Hat) 		if (op->trampoline && op->trampoline_size)
1160aec0be2dSSteven Rostedt (Red Hat) 			if (addr >= op->trampoline &&
1161aec0be2dSSteven Rostedt (Red Hat) 			    addr < op->trampoline + op->trampoline_size) {
1162aec0be2dSSteven Rostedt (Red Hat) 				ret = true;
1163aec0be2dSSteven Rostedt (Red Hat) 				goto out;
1164aec0be2dSSteven Rostedt (Red Hat) 			}
1165aec0be2dSSteven Rostedt (Red Hat) 	} while_for_each_ftrace_op(op);
1166aec0be2dSSteven Rostedt (Red Hat) 
1167aec0be2dSSteven Rostedt (Red Hat)  out:
1168aec0be2dSSteven Rostedt (Red Hat) 	preempt_enable_notrace();
1169aec0be2dSSteven Rostedt (Red Hat) 
1170aec0be2dSSteven Rostedt (Red Hat) 	return ret;
1171aec0be2dSSteven Rostedt (Red Hat) }
1172aec0be2dSSteven Rostedt (Red Hat) 
11733c1720f0SSteven Rostedt struct ftrace_page {
11743c1720f0SSteven Rostedt 	struct ftrace_page	*next;
1175a7900875SSteven Rostedt 	struct dyn_ftrace	*records;
1176431aa3fbSSteven Rostedt 	int			index;
1177a7900875SSteven Rostedt 	int			size;
1178aa5e5ceaSDavid Miller };
11793c1720f0SSteven Rostedt 
1180a7900875SSteven Rostedt #define ENTRY_SIZE sizeof(struct dyn_ftrace)
1181a7900875SSteven Rostedt #define ENTRIES_PER_PAGE (PAGE_SIZE / ENTRY_SIZE)
11823c1720f0SSteven Rostedt 
11833c1720f0SSteven Rostedt /* estimate from running different kernels */
11843c1720f0SSteven Rostedt #define NR_TO_INIT		10000
11853c1720f0SSteven Rostedt 
11863c1720f0SSteven Rostedt static struct ftrace_page	*ftrace_pages_start;
11873c1720f0SSteven Rostedt static struct ftrace_page	*ftrace_pages;
11883c1720f0SSteven Rostedt 
118968f40969SSteven Rostedt (Red Hat) static bool __always_inline ftrace_hash_empty(struct ftrace_hash *hash)
119006a51d93SSteven Rostedt {
119106a51d93SSteven Rostedt 	return !hash || !hash->count;
119206a51d93SSteven Rostedt }
119306a51d93SSteven Rostedt 
1194b448c4e3SSteven Rostedt static struct ftrace_func_entry *
1195b448c4e3SSteven Rostedt ftrace_lookup_ip(struct ftrace_hash *hash, unsigned long ip)
1196b448c4e3SSteven Rostedt {
1197b448c4e3SSteven Rostedt 	unsigned long key;
1198b448c4e3SSteven Rostedt 	struct ftrace_func_entry *entry;
1199b448c4e3SSteven Rostedt 	struct hlist_head *hhd;
1200b448c4e3SSteven Rostedt 
120106a51d93SSteven Rostedt 	if (ftrace_hash_empty(hash))
1202b448c4e3SSteven Rostedt 		return NULL;
1203b448c4e3SSteven Rostedt 
1204b448c4e3SSteven Rostedt 	if (hash->size_bits > 0)
1205b448c4e3SSteven Rostedt 		key = hash_long(ip, hash->size_bits);
1206b448c4e3SSteven Rostedt 	else
1207b448c4e3SSteven Rostedt 		key = 0;
1208b448c4e3SSteven Rostedt 
1209b448c4e3SSteven Rostedt 	hhd = &hash->buckets[key];
1210b448c4e3SSteven Rostedt 
12111bb539caSSteven Rostedt 	hlist_for_each_entry_rcu_notrace(entry, hhd, hlist) {
1212b448c4e3SSteven Rostedt 		if (entry->ip == ip)
1213b448c4e3SSteven Rostedt 			return entry;
1214b448c4e3SSteven Rostedt 	}
1215b448c4e3SSteven Rostedt 	return NULL;
1216b448c4e3SSteven Rostedt }
1217b448c4e3SSteven Rostedt 
121833dc9b12SSteven Rostedt static void __add_hash_entry(struct ftrace_hash *hash,
121933dc9b12SSteven Rostedt 			     struct ftrace_func_entry *entry)
122033dc9b12SSteven Rostedt {
122133dc9b12SSteven Rostedt 	struct hlist_head *hhd;
122233dc9b12SSteven Rostedt 	unsigned long key;
122333dc9b12SSteven Rostedt 
122433dc9b12SSteven Rostedt 	if (hash->size_bits)
122533dc9b12SSteven Rostedt 		key = hash_long(entry->ip, hash->size_bits);
122633dc9b12SSteven Rostedt 	else
122733dc9b12SSteven Rostedt 		key = 0;
122833dc9b12SSteven Rostedt 
122933dc9b12SSteven Rostedt 	hhd = &hash->buckets[key];
123033dc9b12SSteven Rostedt 	hlist_add_head(&entry->hlist, hhd);
123133dc9b12SSteven Rostedt 	hash->count++;
123233dc9b12SSteven Rostedt }
123333dc9b12SSteven Rostedt 
1234b448c4e3SSteven Rostedt static int add_hash_entry(struct ftrace_hash *hash, unsigned long ip)
1235b448c4e3SSteven Rostedt {
1236b448c4e3SSteven Rostedt 	struct ftrace_func_entry *entry;
1237b448c4e3SSteven Rostedt 
1238b448c4e3SSteven Rostedt 	entry = kmalloc(sizeof(*entry), GFP_KERNEL);
1239b448c4e3SSteven Rostedt 	if (!entry)
1240b448c4e3SSteven Rostedt 		return -ENOMEM;
1241b448c4e3SSteven Rostedt 
1242b448c4e3SSteven Rostedt 	entry->ip = ip;
124333dc9b12SSteven Rostedt 	__add_hash_entry(hash, entry);
1244b448c4e3SSteven Rostedt 
1245b448c4e3SSteven Rostedt 	return 0;
1246b448c4e3SSteven Rostedt }
1247b448c4e3SSteven Rostedt 
1248b448c4e3SSteven Rostedt static void
124933dc9b12SSteven Rostedt free_hash_entry(struct ftrace_hash *hash,
125033dc9b12SSteven Rostedt 		  struct ftrace_func_entry *entry)
125133dc9b12SSteven Rostedt {
125233dc9b12SSteven Rostedt 	hlist_del(&entry->hlist);
125333dc9b12SSteven Rostedt 	kfree(entry);
125433dc9b12SSteven Rostedt 	hash->count--;
125533dc9b12SSteven Rostedt }
125633dc9b12SSteven Rostedt 
125733dc9b12SSteven Rostedt static void
1258b448c4e3SSteven Rostedt remove_hash_entry(struct ftrace_hash *hash,
1259b448c4e3SSteven Rostedt 		  struct ftrace_func_entry *entry)
1260b448c4e3SSteven Rostedt {
1261b448c4e3SSteven Rostedt 	hlist_del(&entry->hlist);
1262b448c4e3SSteven Rostedt 	hash->count--;
1263b448c4e3SSteven Rostedt }
1264b448c4e3SSteven Rostedt 
1265b448c4e3SSteven Rostedt static void ftrace_hash_clear(struct ftrace_hash *hash)
1266b448c4e3SSteven Rostedt {
1267b448c4e3SSteven Rostedt 	struct hlist_head *hhd;
1268b67bfe0dSSasha Levin 	struct hlist_node *tn;
1269b448c4e3SSteven Rostedt 	struct ftrace_func_entry *entry;
1270b448c4e3SSteven Rostedt 	int size = 1 << hash->size_bits;
1271b448c4e3SSteven Rostedt 	int i;
1272b448c4e3SSteven Rostedt 
127333dc9b12SSteven Rostedt 	if (!hash->count)
127433dc9b12SSteven Rostedt 		return;
127533dc9b12SSteven Rostedt 
1276b448c4e3SSteven Rostedt 	for (i = 0; i < size; i++) {
1277b448c4e3SSteven Rostedt 		hhd = &hash->buckets[i];
1278b67bfe0dSSasha Levin 		hlist_for_each_entry_safe(entry, tn, hhd, hlist)
127933dc9b12SSteven Rostedt 			free_hash_entry(hash, entry);
1280b448c4e3SSteven Rostedt 	}
1281b448c4e3SSteven Rostedt 	FTRACE_WARN_ON(hash->count);
1282b448c4e3SSteven Rostedt }
1283b448c4e3SSteven Rostedt 
128433dc9b12SSteven Rostedt static void free_ftrace_hash(struct ftrace_hash *hash)
128533dc9b12SSteven Rostedt {
128633dc9b12SSteven Rostedt 	if (!hash || hash == EMPTY_HASH)
128733dc9b12SSteven Rostedt 		return;
128833dc9b12SSteven Rostedt 	ftrace_hash_clear(hash);
128933dc9b12SSteven Rostedt 	kfree(hash->buckets);
129033dc9b12SSteven Rostedt 	kfree(hash);
129133dc9b12SSteven Rostedt }
129233dc9b12SSteven Rostedt 
129307fd5515SSteven Rostedt static void __free_ftrace_hash_rcu(struct rcu_head *rcu)
129407fd5515SSteven Rostedt {
129507fd5515SSteven Rostedt 	struct ftrace_hash *hash;
129607fd5515SSteven Rostedt 
129707fd5515SSteven Rostedt 	hash = container_of(rcu, struct ftrace_hash, rcu);
129807fd5515SSteven Rostedt 	free_ftrace_hash(hash);
129907fd5515SSteven Rostedt }
130007fd5515SSteven Rostedt 
130107fd5515SSteven Rostedt static void free_ftrace_hash_rcu(struct ftrace_hash *hash)
130207fd5515SSteven Rostedt {
130307fd5515SSteven Rostedt 	if (!hash || hash == EMPTY_HASH)
130407fd5515SSteven Rostedt 		return;
130507fd5515SSteven Rostedt 	call_rcu_sched(&hash->rcu, __free_ftrace_hash_rcu);
130607fd5515SSteven Rostedt }
130707fd5515SSteven Rostedt 
13085500fa51SJiri Olsa void ftrace_free_filter(struct ftrace_ops *ops)
13095500fa51SJiri Olsa {
1310f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
131133b7f99cSSteven Rostedt (Red Hat) 	free_ftrace_hash(ops->func_hash->filter_hash);
131233b7f99cSSteven Rostedt (Red Hat) 	free_ftrace_hash(ops->func_hash->notrace_hash);
13135500fa51SJiri Olsa }
13145500fa51SJiri Olsa 
131533dc9b12SSteven Rostedt static struct ftrace_hash *alloc_ftrace_hash(int size_bits)
131633dc9b12SSteven Rostedt {
131733dc9b12SSteven Rostedt 	struct ftrace_hash *hash;
131833dc9b12SSteven Rostedt 	int size;
131933dc9b12SSteven Rostedt 
132033dc9b12SSteven Rostedt 	hash = kzalloc(sizeof(*hash), GFP_KERNEL);
132133dc9b12SSteven Rostedt 	if (!hash)
132233dc9b12SSteven Rostedt 		return NULL;
132333dc9b12SSteven Rostedt 
132433dc9b12SSteven Rostedt 	size = 1 << size_bits;
132547b0edcbSThomas Meyer 	hash->buckets = kcalloc(size, sizeof(*hash->buckets), GFP_KERNEL);
132633dc9b12SSteven Rostedt 
132733dc9b12SSteven Rostedt 	if (!hash->buckets) {
132833dc9b12SSteven Rostedt 		kfree(hash);
132933dc9b12SSteven Rostedt 		return NULL;
133033dc9b12SSteven Rostedt 	}
133133dc9b12SSteven Rostedt 
133233dc9b12SSteven Rostedt 	hash->size_bits = size_bits;
133333dc9b12SSteven Rostedt 
133433dc9b12SSteven Rostedt 	return hash;
133533dc9b12SSteven Rostedt }
133633dc9b12SSteven Rostedt 
133733dc9b12SSteven Rostedt static struct ftrace_hash *
133833dc9b12SSteven Rostedt alloc_and_copy_ftrace_hash(int size_bits, struct ftrace_hash *hash)
133933dc9b12SSteven Rostedt {
134033dc9b12SSteven Rostedt 	struct ftrace_func_entry *entry;
134133dc9b12SSteven Rostedt 	struct ftrace_hash *new_hash;
134233dc9b12SSteven Rostedt 	int size;
134333dc9b12SSteven Rostedt 	int ret;
134433dc9b12SSteven Rostedt 	int i;
134533dc9b12SSteven Rostedt 
134633dc9b12SSteven Rostedt 	new_hash = alloc_ftrace_hash(size_bits);
134733dc9b12SSteven Rostedt 	if (!new_hash)
134833dc9b12SSteven Rostedt 		return NULL;
134933dc9b12SSteven Rostedt 
135033dc9b12SSteven Rostedt 	/* Empty hash? */
135106a51d93SSteven Rostedt 	if (ftrace_hash_empty(hash))
135233dc9b12SSteven Rostedt 		return new_hash;
135333dc9b12SSteven Rostedt 
135433dc9b12SSteven Rostedt 	size = 1 << hash->size_bits;
135533dc9b12SSteven Rostedt 	for (i = 0; i < size; i++) {
1356b67bfe0dSSasha Levin 		hlist_for_each_entry(entry, &hash->buckets[i], hlist) {
135733dc9b12SSteven Rostedt 			ret = add_hash_entry(new_hash, entry->ip);
135833dc9b12SSteven Rostedt 			if (ret < 0)
135933dc9b12SSteven Rostedt 				goto free_hash;
136033dc9b12SSteven Rostedt 		}
136133dc9b12SSteven Rostedt 	}
136233dc9b12SSteven Rostedt 
136333dc9b12SSteven Rostedt 	FTRACE_WARN_ON(new_hash->count != hash->count);
136433dc9b12SSteven Rostedt 
136533dc9b12SSteven Rostedt 	return new_hash;
136633dc9b12SSteven Rostedt 
136733dc9b12SSteven Rostedt  free_hash:
136833dc9b12SSteven Rostedt 	free_ftrace_hash(new_hash);
136933dc9b12SSteven Rostedt 	return NULL;
137033dc9b12SSteven Rostedt }
137133dc9b12SSteven Rostedt 
137241fb61c2SSteven Rostedt static void
137384261912SSteven Rostedt (Red Hat) ftrace_hash_rec_disable_modify(struct ftrace_ops *ops, int filter_hash);
137441fb61c2SSteven Rostedt static void
137584261912SSteven Rostedt (Red Hat) ftrace_hash_rec_enable_modify(struct ftrace_ops *ops, int filter_hash);
137641fb61c2SSteven Rostedt 
1377f8b8be8aSMasami Hiramatsu static int ftrace_hash_ipmodify_update(struct ftrace_ops *ops,
1378f8b8be8aSMasami Hiramatsu 				       struct ftrace_hash *new_hash);
1379f8b8be8aSMasami Hiramatsu 
138033dc9b12SSteven Rostedt static int
138141fb61c2SSteven Rostedt ftrace_hash_move(struct ftrace_ops *ops, int enable,
138241fb61c2SSteven Rostedt 		 struct ftrace_hash **dst, struct ftrace_hash *src)
138333dc9b12SSteven Rostedt {
138433dc9b12SSteven Rostedt 	struct ftrace_func_entry *entry;
1385b67bfe0dSSasha Levin 	struct hlist_node *tn;
138633dc9b12SSteven Rostedt 	struct hlist_head *hhd;
138707fd5515SSteven Rostedt 	struct ftrace_hash *new_hash;
138833dc9b12SSteven Rostedt 	int size = src->count;
138933dc9b12SSteven Rostedt 	int bits = 0;
1390f8b8be8aSMasami Hiramatsu 	int ret;
139133dc9b12SSteven Rostedt 	int i;
139233dc9b12SSteven Rostedt 
1393f8b8be8aSMasami Hiramatsu 	/* Reject setting notrace hash on IPMODIFY ftrace_ops */
1394f8b8be8aSMasami Hiramatsu 	if (ops->flags & FTRACE_OPS_FL_IPMODIFY && !enable)
1395f8b8be8aSMasami Hiramatsu 		return -EINVAL;
1396f8b8be8aSMasami Hiramatsu 
139733dc9b12SSteven Rostedt 	/*
139833dc9b12SSteven Rostedt 	 * If the new source is empty, just free dst and assign it
139933dc9b12SSteven Rostedt 	 * the empty_hash.
140033dc9b12SSteven Rostedt 	 */
140133dc9b12SSteven Rostedt 	if (!src->count) {
14025c27c775SMasami Hiramatsu 		new_hash = EMPTY_HASH;
14035c27c775SMasami Hiramatsu 		goto update;
140433dc9b12SSteven Rostedt 	}
140533dc9b12SSteven Rostedt 
140633dc9b12SSteven Rostedt 	/*
140733dc9b12SSteven Rostedt 	 * Make the hash size about 1/2 the # found
140833dc9b12SSteven Rostedt 	 */
140933dc9b12SSteven Rostedt 	for (size /= 2; size; size >>= 1)
141033dc9b12SSteven Rostedt 		bits++;
141133dc9b12SSteven Rostedt 
141233dc9b12SSteven Rostedt 	/* Don't allocate too much */
141333dc9b12SSteven Rostedt 	if (bits > FTRACE_HASH_MAX_BITS)
141433dc9b12SSteven Rostedt 		bits = FTRACE_HASH_MAX_BITS;
141533dc9b12SSteven Rostedt 
141607fd5515SSteven Rostedt 	new_hash = alloc_ftrace_hash(bits);
141707fd5515SSteven Rostedt 	if (!new_hash)
14185c27c775SMasami Hiramatsu 		return -ENOMEM;
141933dc9b12SSteven Rostedt 
142033dc9b12SSteven Rostedt 	size = 1 << src->size_bits;
142133dc9b12SSteven Rostedt 	for (i = 0; i < size; i++) {
142233dc9b12SSteven Rostedt 		hhd = &src->buckets[i];
1423b67bfe0dSSasha Levin 		hlist_for_each_entry_safe(entry, tn, hhd, hlist) {
142433dc9b12SSteven Rostedt 			remove_hash_entry(src, entry);
142507fd5515SSteven Rostedt 			__add_hash_entry(new_hash, entry);
142633dc9b12SSteven Rostedt 		}
142733dc9b12SSteven Rostedt 	}
142833dc9b12SSteven Rostedt 
14295c27c775SMasami Hiramatsu update:
1430f8b8be8aSMasami Hiramatsu 	/* Make sure this can be applied if it is IPMODIFY ftrace_ops */
1431f8b8be8aSMasami Hiramatsu 	if (enable) {
1432f8b8be8aSMasami Hiramatsu 		/* IPMODIFY should be updated only when filter_hash updating */
1433f8b8be8aSMasami Hiramatsu 		ret = ftrace_hash_ipmodify_update(ops, new_hash);
1434f8b8be8aSMasami Hiramatsu 		if (ret < 0) {
1435f8b8be8aSMasami Hiramatsu 			free_ftrace_hash(new_hash);
1436f8b8be8aSMasami Hiramatsu 			return ret;
1437f8b8be8aSMasami Hiramatsu 		}
1438f8b8be8aSMasami Hiramatsu 	}
1439f8b8be8aSMasami Hiramatsu 
14405c27c775SMasami Hiramatsu 	/*
14415c27c775SMasami Hiramatsu 	 * Remove the current set, update the hash and add
14425c27c775SMasami Hiramatsu 	 * them back.
14435c27c775SMasami Hiramatsu 	 */
144484261912SSteven Rostedt (Red Hat) 	ftrace_hash_rec_disable_modify(ops, enable);
14455c27c775SMasami Hiramatsu 
144607fd5515SSteven Rostedt 	rcu_assign_pointer(*dst, new_hash);
144707fd5515SSteven Rostedt 
144884261912SSteven Rostedt (Red Hat) 	ftrace_hash_rec_enable_modify(ops, enable);
144941fb61c2SSteven Rostedt 
14505c27c775SMasami Hiramatsu 	return 0;
145133dc9b12SSteven Rostedt }
145233dc9b12SSteven Rostedt 
1453fef5aeeeSSteven Rostedt (Red Hat) static bool hash_contains_ip(unsigned long ip,
1454fef5aeeeSSteven Rostedt (Red Hat) 			     struct ftrace_ops_hash *hash)
1455fef5aeeeSSteven Rostedt (Red Hat) {
1456fef5aeeeSSteven Rostedt (Red Hat) 	/*
1457fef5aeeeSSteven Rostedt (Red Hat) 	 * The function record is a match if it exists in the filter
1458fef5aeeeSSteven Rostedt (Red Hat) 	 * hash and not in the notrace hash. Note, an emty hash is
1459fef5aeeeSSteven Rostedt (Red Hat) 	 * considered a match for the filter hash, but an empty
1460fef5aeeeSSteven Rostedt (Red Hat) 	 * notrace hash is considered not in the notrace hash.
1461fef5aeeeSSteven Rostedt (Red Hat) 	 */
1462fef5aeeeSSteven Rostedt (Red Hat) 	return (ftrace_hash_empty(hash->filter_hash) ||
1463fef5aeeeSSteven Rostedt (Red Hat) 		ftrace_lookup_ip(hash->filter_hash, ip)) &&
1464fef5aeeeSSteven Rostedt (Red Hat) 		(ftrace_hash_empty(hash->notrace_hash) ||
1465fef5aeeeSSteven Rostedt (Red Hat) 		 !ftrace_lookup_ip(hash->notrace_hash, ip));
1466fef5aeeeSSteven Rostedt (Red Hat) }
1467fef5aeeeSSteven Rostedt (Red Hat) 
1468265c831cSSteven Rostedt /*
1469b848914cSSteven Rostedt  * Test the hashes for this ops to see if we want to call
1470b848914cSSteven Rostedt  * the ops->func or not.
1471b848914cSSteven Rostedt  *
1472b848914cSSteven Rostedt  * It's a match if the ip is in the ops->filter_hash or
1473b848914cSSteven Rostedt  * the filter_hash does not exist or is empty,
1474b848914cSSteven Rostedt  *  AND
1475b848914cSSteven Rostedt  * the ip is not in the ops->notrace_hash.
1476cdbe61bfSSteven Rostedt  *
1477cdbe61bfSSteven Rostedt  * This needs to be called with preemption disabled as
1478cdbe61bfSSteven Rostedt  * the hashes are freed with call_rcu_sched().
1479b848914cSSteven Rostedt  */
1480b848914cSSteven Rostedt static int
1481195a8afcSSteven Rostedt (Red Hat) ftrace_ops_test(struct ftrace_ops *ops, unsigned long ip, void *regs)
1482b848914cSSteven Rostedt {
1483fef5aeeeSSteven Rostedt (Red Hat) 	struct ftrace_ops_hash hash;
1484b848914cSSteven Rostedt 	int ret;
1485b848914cSSteven Rostedt 
1486195a8afcSSteven Rostedt (Red Hat) #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
1487195a8afcSSteven Rostedt (Red Hat) 	/*
1488195a8afcSSteven Rostedt (Red Hat) 	 * There's a small race when adding ops that the ftrace handler
1489195a8afcSSteven Rostedt (Red Hat) 	 * that wants regs, may be called without them. We can not
1490195a8afcSSteven Rostedt (Red Hat) 	 * allow that handler to be called if regs is NULL.
1491195a8afcSSteven Rostedt (Red Hat) 	 */
1492195a8afcSSteven Rostedt (Red Hat) 	if (regs == NULL && (ops->flags & FTRACE_OPS_FL_SAVE_REGS))
1493195a8afcSSteven Rostedt (Red Hat) 		return 0;
1494195a8afcSSteven Rostedt (Red Hat) #endif
1495195a8afcSSteven Rostedt (Red Hat) 
1496fef5aeeeSSteven Rostedt (Red Hat) 	hash.filter_hash = rcu_dereference_raw_notrace(ops->func_hash->filter_hash);
1497fef5aeeeSSteven Rostedt (Red Hat) 	hash.notrace_hash = rcu_dereference_raw_notrace(ops->func_hash->notrace_hash);
1498b848914cSSteven Rostedt 
1499fef5aeeeSSteven Rostedt (Red Hat) 	if (hash_contains_ip(ip, &hash))
1500b848914cSSteven Rostedt 		ret = 1;
1501b848914cSSteven Rostedt 	else
1502b848914cSSteven Rostedt 		ret = 0;
1503b848914cSSteven Rostedt 
1504b848914cSSteven Rostedt 	return ret;
1505b848914cSSteven Rostedt }
1506b848914cSSteven Rostedt 
1507b848914cSSteven Rostedt /*
1508265c831cSSteven Rostedt  * This is a double for. Do not use 'break' to break out of the loop,
1509265c831cSSteven Rostedt  * you must use a goto.
1510265c831cSSteven Rostedt  */
1511265c831cSSteven Rostedt #define do_for_each_ftrace_rec(pg, rec)					\
1512265c831cSSteven Rostedt 	for (pg = ftrace_pages_start; pg; pg = pg->next) {		\
1513265c831cSSteven Rostedt 		int _____i;						\
1514265c831cSSteven Rostedt 		for (_____i = 0; _____i < pg->index; _____i++) {	\
1515265c831cSSteven Rostedt 			rec = &pg->records[_____i];
1516265c831cSSteven Rostedt 
1517265c831cSSteven Rostedt #define while_for_each_ftrace_rec()		\
1518265c831cSSteven Rostedt 		}				\
1519265c831cSSteven Rostedt 	}
1520ecea656dSAbhishek Sagar 
15215855feadSSteven Rostedt 
15225855feadSSteven Rostedt static int ftrace_cmp_recs(const void *a, const void *b)
15235855feadSSteven Rostedt {
1524a650e02aSSteven Rostedt 	const struct dyn_ftrace *key = a;
1525a650e02aSSteven Rostedt 	const struct dyn_ftrace *rec = b;
15265855feadSSteven Rostedt 
1527a650e02aSSteven Rostedt 	if (key->flags < rec->ip)
15285855feadSSteven Rostedt 		return -1;
1529a650e02aSSteven Rostedt 	if (key->ip >= rec->ip + MCOUNT_INSN_SIZE)
1530a650e02aSSteven Rostedt 		return 1;
1531a650e02aSSteven Rostedt 	return 0;
1532a650e02aSSteven Rostedt }
1533a650e02aSSteven Rostedt 
153404cf31a7SMichael Ellerman /**
153504cf31a7SMichael Ellerman  * ftrace_location_range - return the first address of a traced location
153604cf31a7SMichael Ellerman  *	if it touches the given ip range
153704cf31a7SMichael Ellerman  * @start: start of range to search.
153804cf31a7SMichael Ellerman  * @end: end of range to search (inclusive). @end points to the last byte
153904cf31a7SMichael Ellerman  *	to check.
154004cf31a7SMichael Ellerman  *
154104cf31a7SMichael Ellerman  * Returns rec->ip if the related ftrace location is a least partly within
154204cf31a7SMichael Ellerman  * the given address range. That is, the first address of the instruction
154304cf31a7SMichael Ellerman  * that is either a NOP or call to the function tracer. It checks the ftrace
154404cf31a7SMichael Ellerman  * internal tables to determine if the address belongs or not.
154504cf31a7SMichael Ellerman  */
154604cf31a7SMichael Ellerman unsigned long ftrace_location_range(unsigned long start, unsigned long end)
1547a650e02aSSteven Rostedt {
1548a650e02aSSteven Rostedt 	struct ftrace_page *pg;
1549a650e02aSSteven Rostedt 	struct dyn_ftrace *rec;
1550a650e02aSSteven Rostedt 	struct dyn_ftrace key;
1551a650e02aSSteven Rostedt 
1552a650e02aSSteven Rostedt 	key.ip = start;
1553a650e02aSSteven Rostedt 	key.flags = end;	/* overload flags, as it is unsigned long */
1554a650e02aSSteven Rostedt 
1555a650e02aSSteven Rostedt 	for (pg = ftrace_pages_start; pg; pg = pg->next) {
1556a650e02aSSteven Rostedt 		if (end < pg->records[0].ip ||
1557a650e02aSSteven Rostedt 		    start >= (pg->records[pg->index - 1].ip + MCOUNT_INSN_SIZE))
1558a650e02aSSteven Rostedt 			continue;
1559a650e02aSSteven Rostedt 		rec = bsearch(&key, pg->records, pg->index,
1560a650e02aSSteven Rostedt 			      sizeof(struct dyn_ftrace),
1561a650e02aSSteven Rostedt 			      ftrace_cmp_recs);
1562a650e02aSSteven Rostedt 		if (rec)
1563f0cf973aSSteven Rostedt 			return rec->ip;
1564a650e02aSSteven Rostedt 	}
1565a650e02aSSteven Rostedt 
15665855feadSSteven Rostedt 	return 0;
15675855feadSSteven Rostedt }
15685855feadSSteven Rostedt 
1569c88fd863SSteven Rostedt /**
1570c88fd863SSteven Rostedt  * ftrace_location - return true if the ip giving is a traced location
1571c88fd863SSteven Rostedt  * @ip: the instruction pointer to check
1572c88fd863SSteven Rostedt  *
1573f0cf973aSSteven Rostedt  * Returns rec->ip if @ip given is a pointer to a ftrace location.
1574c88fd863SSteven Rostedt  * That is, the instruction that is either a NOP or call to
1575c88fd863SSteven Rostedt  * the function tracer. It checks the ftrace internal tables to
1576c88fd863SSteven Rostedt  * determine if the address belongs or not.
1577c88fd863SSteven Rostedt  */
1578f0cf973aSSteven Rostedt unsigned long ftrace_location(unsigned long ip)
1579c88fd863SSteven Rostedt {
1580a650e02aSSteven Rostedt 	return ftrace_location_range(ip, ip);
15815855feadSSteven Rostedt }
1582c88fd863SSteven Rostedt 
1583a650e02aSSteven Rostedt /**
1584a650e02aSSteven Rostedt  * ftrace_text_reserved - return true if range contains an ftrace location
1585a650e02aSSteven Rostedt  * @start: start of range to search
1586a650e02aSSteven Rostedt  * @end: end of range to search (inclusive). @end points to the last byte to check.
1587a650e02aSSteven Rostedt  *
1588a650e02aSSteven Rostedt  * Returns 1 if @start and @end contains a ftrace location.
1589a650e02aSSteven Rostedt  * That is, the instruction that is either a NOP or call to
1590a650e02aSSteven Rostedt  * the function tracer. It checks the ftrace internal tables to
1591a650e02aSSteven Rostedt  * determine if the address belongs or not.
1592a650e02aSSteven Rostedt  */
1593d88471cbSSasha Levin int ftrace_text_reserved(const void *start, const void *end)
1594a650e02aSSteven Rostedt {
1595f0cf973aSSteven Rostedt 	unsigned long ret;
1596f0cf973aSSteven Rostedt 
1597f0cf973aSSteven Rostedt 	ret = ftrace_location_range((unsigned long)start,
1598a650e02aSSteven Rostedt 				    (unsigned long)end);
1599f0cf973aSSteven Rostedt 
1600f0cf973aSSteven Rostedt 	return (int)!!ret;
1601c88fd863SSteven Rostedt }
1602c88fd863SSteven Rostedt 
16034fbb48cbSSteven Rostedt (Red Hat) /* Test if ops registered to this rec needs regs */
16044fbb48cbSSteven Rostedt (Red Hat) static bool test_rec_ops_needs_regs(struct dyn_ftrace *rec)
16054fbb48cbSSteven Rostedt (Red Hat) {
16064fbb48cbSSteven Rostedt (Red Hat) 	struct ftrace_ops *ops;
16074fbb48cbSSteven Rostedt (Red Hat) 	bool keep_regs = false;
16084fbb48cbSSteven Rostedt (Red Hat) 
16094fbb48cbSSteven Rostedt (Red Hat) 	for (ops = ftrace_ops_list;
16104fbb48cbSSteven Rostedt (Red Hat) 	     ops != &ftrace_list_end; ops = ops->next) {
16114fbb48cbSSteven Rostedt (Red Hat) 		/* pass rec in as regs to have non-NULL val */
16124fbb48cbSSteven Rostedt (Red Hat) 		if (ftrace_ops_test(ops, rec->ip, rec)) {
16134fbb48cbSSteven Rostedt (Red Hat) 			if (ops->flags & FTRACE_OPS_FL_SAVE_REGS) {
16144fbb48cbSSteven Rostedt (Red Hat) 				keep_regs = true;
16154fbb48cbSSteven Rostedt (Red Hat) 				break;
16164fbb48cbSSteven Rostedt (Red Hat) 			}
16174fbb48cbSSteven Rostedt (Red Hat) 		}
16184fbb48cbSSteven Rostedt (Red Hat) 	}
16194fbb48cbSSteven Rostedt (Red Hat) 
16204fbb48cbSSteven Rostedt (Red Hat) 	return  keep_regs;
16214fbb48cbSSteven Rostedt (Red Hat) }
16224fbb48cbSSteven Rostedt (Red Hat) 
162384b6d3e6SJiri Olsa static bool __ftrace_hash_rec_update(struct ftrace_ops *ops,
1624ed926f9bSSteven Rostedt 				     int filter_hash,
1625ed926f9bSSteven Rostedt 				     bool inc)
1626ed926f9bSSteven Rostedt {
1627ed926f9bSSteven Rostedt 	struct ftrace_hash *hash;
1628ed926f9bSSteven Rostedt 	struct ftrace_hash *other_hash;
1629ed926f9bSSteven Rostedt 	struct ftrace_page *pg;
1630ed926f9bSSteven Rostedt 	struct dyn_ftrace *rec;
163184b6d3e6SJiri Olsa 	bool update = false;
1632ed926f9bSSteven Rostedt 	int count = 0;
1633ed926f9bSSteven Rostedt 	int all = 0;
1634ed926f9bSSteven Rostedt 
1635ed926f9bSSteven Rostedt 	/* Only update if the ops has been registered */
1636ed926f9bSSteven Rostedt 	if (!(ops->flags & FTRACE_OPS_FL_ENABLED))
163784b6d3e6SJiri Olsa 		return false;
1638ed926f9bSSteven Rostedt 
1639ed926f9bSSteven Rostedt 	/*
1640ed926f9bSSteven Rostedt 	 * In the filter_hash case:
1641ed926f9bSSteven Rostedt 	 *   If the count is zero, we update all records.
1642ed926f9bSSteven Rostedt 	 *   Otherwise we just update the items in the hash.
1643ed926f9bSSteven Rostedt 	 *
1644ed926f9bSSteven Rostedt 	 * In the notrace_hash case:
1645ed926f9bSSteven Rostedt 	 *   We enable the update in the hash.
1646ed926f9bSSteven Rostedt 	 *   As disabling notrace means enabling the tracing,
1647ed926f9bSSteven Rostedt 	 *   and enabling notrace means disabling, the inc variable
1648ed926f9bSSteven Rostedt 	 *   gets inversed.
1649ed926f9bSSteven Rostedt 	 */
1650ed926f9bSSteven Rostedt 	if (filter_hash) {
165133b7f99cSSteven Rostedt (Red Hat) 		hash = ops->func_hash->filter_hash;
165233b7f99cSSteven Rostedt (Red Hat) 		other_hash = ops->func_hash->notrace_hash;
165306a51d93SSteven Rostedt 		if (ftrace_hash_empty(hash))
1654ed926f9bSSteven Rostedt 			all = 1;
1655ed926f9bSSteven Rostedt 	} else {
1656ed926f9bSSteven Rostedt 		inc = !inc;
165733b7f99cSSteven Rostedt (Red Hat) 		hash = ops->func_hash->notrace_hash;
165833b7f99cSSteven Rostedt (Red Hat) 		other_hash = ops->func_hash->filter_hash;
1659ed926f9bSSteven Rostedt 		/*
1660ed926f9bSSteven Rostedt 		 * If the notrace hash has no items,
1661ed926f9bSSteven Rostedt 		 * then there's nothing to do.
1662ed926f9bSSteven Rostedt 		 */
166306a51d93SSteven Rostedt 		if (ftrace_hash_empty(hash))
166484b6d3e6SJiri Olsa 			return false;
1665ed926f9bSSteven Rostedt 	}
1666ed926f9bSSteven Rostedt 
1667ed926f9bSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
1668ed926f9bSSteven Rostedt 		int in_other_hash = 0;
1669ed926f9bSSteven Rostedt 		int in_hash = 0;
1670ed926f9bSSteven Rostedt 		int match = 0;
1671ed926f9bSSteven Rostedt 
1672b7ffffbbSSteven Rostedt (Red Hat) 		if (rec->flags & FTRACE_FL_DISABLED)
1673b7ffffbbSSteven Rostedt (Red Hat) 			continue;
1674b7ffffbbSSteven Rostedt (Red Hat) 
1675ed926f9bSSteven Rostedt 		if (all) {
1676ed926f9bSSteven Rostedt 			/*
1677ed926f9bSSteven Rostedt 			 * Only the filter_hash affects all records.
1678ed926f9bSSteven Rostedt 			 * Update if the record is not in the notrace hash.
1679ed926f9bSSteven Rostedt 			 */
1680b848914cSSteven Rostedt 			if (!other_hash || !ftrace_lookup_ip(other_hash, rec->ip))
1681ed926f9bSSteven Rostedt 				match = 1;
1682ed926f9bSSteven Rostedt 		} else {
168306a51d93SSteven Rostedt 			in_hash = !!ftrace_lookup_ip(hash, rec->ip);
168406a51d93SSteven Rostedt 			in_other_hash = !!ftrace_lookup_ip(other_hash, rec->ip);
1685ed926f9bSSteven Rostedt 
1686ed926f9bSSteven Rostedt 			/*
168719eab4a4SSteven Rostedt (Red Hat) 			 * If filter_hash is set, we want to match all functions
168819eab4a4SSteven Rostedt (Red Hat) 			 * that are in the hash but not in the other hash.
1689ed926f9bSSteven Rostedt 			 *
169019eab4a4SSteven Rostedt (Red Hat) 			 * If filter_hash is not set, then we are decrementing.
169119eab4a4SSteven Rostedt (Red Hat) 			 * That means we match anything that is in the hash
169219eab4a4SSteven Rostedt (Red Hat) 			 * and also in the other_hash. That is, we need to turn
169319eab4a4SSteven Rostedt (Red Hat) 			 * off functions in the other hash because they are disabled
169419eab4a4SSteven Rostedt (Red Hat) 			 * by this hash.
1695ed926f9bSSteven Rostedt 			 */
1696ed926f9bSSteven Rostedt 			if (filter_hash && in_hash && !in_other_hash)
1697ed926f9bSSteven Rostedt 				match = 1;
1698ed926f9bSSteven Rostedt 			else if (!filter_hash && in_hash &&
169906a51d93SSteven Rostedt 				 (in_other_hash || ftrace_hash_empty(other_hash)))
1700ed926f9bSSteven Rostedt 				match = 1;
1701ed926f9bSSteven Rostedt 		}
1702ed926f9bSSteven Rostedt 		if (!match)
1703ed926f9bSSteven Rostedt 			continue;
1704ed926f9bSSteven Rostedt 
1705ed926f9bSSteven Rostedt 		if (inc) {
1706ed926f9bSSteven Rostedt 			rec->flags++;
17070376bde1SSteven Rostedt (Red Hat) 			if (FTRACE_WARN_ON(ftrace_rec_count(rec) == FTRACE_REF_MAX))
170884b6d3e6SJiri Olsa 				return false;
170979922b80SSteven Rostedt (Red Hat) 
171079922b80SSteven Rostedt (Red Hat) 			/*
171179922b80SSteven Rostedt (Red Hat) 			 * If there's only a single callback registered to a
171279922b80SSteven Rostedt (Red Hat) 			 * function, and the ops has a trampoline registered
171379922b80SSteven Rostedt (Red Hat) 			 * for it, then we can call it directly.
171479922b80SSteven Rostedt (Red Hat) 			 */
1715fef5aeeeSSteven Rostedt (Red Hat) 			if (ftrace_rec_count(rec) == 1 && ops->trampoline)
171679922b80SSteven Rostedt (Red Hat) 				rec->flags |= FTRACE_FL_TRAMP;
1717fef5aeeeSSteven Rostedt (Red Hat) 			else
171879922b80SSteven Rostedt (Red Hat) 				/*
171979922b80SSteven Rostedt (Red Hat) 				 * If we are adding another function callback
172079922b80SSteven Rostedt (Red Hat) 				 * to this function, and the previous had a
1721bce0b6c5SSteven Rostedt (Red Hat) 				 * custom trampoline in use, then we need to go
1722bce0b6c5SSteven Rostedt (Red Hat) 				 * back to the default trampoline.
172379922b80SSteven Rostedt (Red Hat) 				 */
1724fef5aeeeSSteven Rostedt (Red Hat) 				rec->flags &= ~FTRACE_FL_TRAMP;
172579922b80SSteven Rostedt (Red Hat) 
172608f6fba5SSteven Rostedt 			/*
172708f6fba5SSteven Rostedt 			 * If any ops wants regs saved for this function
172808f6fba5SSteven Rostedt 			 * then all ops will get saved regs.
172908f6fba5SSteven Rostedt 			 */
173008f6fba5SSteven Rostedt 			if (ops->flags & FTRACE_OPS_FL_SAVE_REGS)
173108f6fba5SSteven Rostedt 				rec->flags |= FTRACE_FL_REGS;
1732ed926f9bSSteven Rostedt 		} else {
17330376bde1SSteven Rostedt (Red Hat) 			if (FTRACE_WARN_ON(ftrace_rec_count(rec) == 0))
173484b6d3e6SJiri Olsa 				return false;
1735ed926f9bSSteven Rostedt 			rec->flags--;
173679922b80SSteven Rostedt (Red Hat) 
17374fbb48cbSSteven Rostedt (Red Hat) 			/*
17384fbb48cbSSteven Rostedt (Red Hat) 			 * If the rec had REGS enabled and the ops that is
17394fbb48cbSSteven Rostedt (Red Hat) 			 * being removed had REGS set, then see if there is
17404fbb48cbSSteven Rostedt (Red Hat) 			 * still any ops for this record that wants regs.
17414fbb48cbSSteven Rostedt (Red Hat) 			 * If not, we can stop recording them.
17424fbb48cbSSteven Rostedt (Red Hat) 			 */
17430376bde1SSteven Rostedt (Red Hat) 			if (ftrace_rec_count(rec) > 0 &&
17444fbb48cbSSteven Rostedt (Red Hat) 			    rec->flags & FTRACE_FL_REGS &&
17454fbb48cbSSteven Rostedt (Red Hat) 			    ops->flags & FTRACE_OPS_FL_SAVE_REGS) {
17464fbb48cbSSteven Rostedt (Red Hat) 				if (!test_rec_ops_needs_regs(rec))
17474fbb48cbSSteven Rostedt (Red Hat) 					rec->flags &= ~FTRACE_FL_REGS;
17484fbb48cbSSteven Rostedt (Red Hat) 			}
174979922b80SSteven Rostedt (Red Hat) 
175079922b80SSteven Rostedt (Red Hat) 			/*
1751fef5aeeeSSteven Rostedt (Red Hat) 			 * If the rec had TRAMP enabled, then it needs to
1752fef5aeeeSSteven Rostedt (Red Hat) 			 * be cleared. As TRAMP can only be enabled iff
1753fef5aeeeSSteven Rostedt (Red Hat) 			 * there is only a single ops attached to it.
1754fef5aeeeSSteven Rostedt (Red Hat) 			 * In otherwords, always disable it on decrementing.
1755fef5aeeeSSteven Rostedt (Red Hat) 			 * In the future, we may set it if rec count is
1756fef5aeeeSSteven Rostedt (Red Hat) 			 * decremented to one, and the ops that is left
1757fef5aeeeSSteven Rostedt (Red Hat) 			 * has a trampoline.
1758fef5aeeeSSteven Rostedt (Red Hat) 			 */
1759fef5aeeeSSteven Rostedt (Red Hat) 			rec->flags &= ~FTRACE_FL_TRAMP;
1760fef5aeeeSSteven Rostedt (Red Hat) 
1761fef5aeeeSSteven Rostedt (Red Hat) 			/*
176279922b80SSteven Rostedt (Red Hat) 			 * flags will be cleared in ftrace_check_record()
176379922b80SSteven Rostedt (Red Hat) 			 * if rec count is zero.
176479922b80SSteven Rostedt (Red Hat) 			 */
1765ed926f9bSSteven Rostedt 		}
1766ed926f9bSSteven Rostedt 		count++;
176784b6d3e6SJiri Olsa 
176884b6d3e6SJiri Olsa 		/* Must match FTRACE_UPDATE_CALLS in ftrace_modify_all_code() */
176984b6d3e6SJiri Olsa 		update |= ftrace_test_record(rec, 1) != FTRACE_UPDATE_IGNORE;
177084b6d3e6SJiri Olsa 
1771ed926f9bSSteven Rostedt 		/* Shortcut, if we handled all records, we are done. */
1772ed926f9bSSteven Rostedt 		if (!all && count == hash->count)
177384b6d3e6SJiri Olsa 			return update;
1774ed926f9bSSteven Rostedt 	} while_for_each_ftrace_rec();
177584b6d3e6SJiri Olsa 
177684b6d3e6SJiri Olsa 	return update;
1777ed926f9bSSteven Rostedt }
1778ed926f9bSSteven Rostedt 
177984b6d3e6SJiri Olsa static bool ftrace_hash_rec_disable(struct ftrace_ops *ops,
1780ed926f9bSSteven Rostedt 				    int filter_hash)
1781ed926f9bSSteven Rostedt {
178284b6d3e6SJiri Olsa 	return __ftrace_hash_rec_update(ops, filter_hash, 0);
1783ed926f9bSSteven Rostedt }
1784ed926f9bSSteven Rostedt 
178584b6d3e6SJiri Olsa static bool ftrace_hash_rec_enable(struct ftrace_ops *ops,
1786ed926f9bSSteven Rostedt 				   int filter_hash)
1787ed926f9bSSteven Rostedt {
178884b6d3e6SJiri Olsa 	return __ftrace_hash_rec_update(ops, filter_hash, 1);
1789ed926f9bSSteven Rostedt }
1790ed926f9bSSteven Rostedt 
179184261912SSteven Rostedt (Red Hat) static void ftrace_hash_rec_update_modify(struct ftrace_ops *ops,
179284261912SSteven Rostedt (Red Hat) 					  int filter_hash, int inc)
179384261912SSteven Rostedt (Red Hat) {
179484261912SSteven Rostedt (Red Hat) 	struct ftrace_ops *op;
179584261912SSteven Rostedt (Red Hat) 
179684261912SSteven Rostedt (Red Hat) 	__ftrace_hash_rec_update(ops, filter_hash, inc);
179784261912SSteven Rostedt (Red Hat) 
179884261912SSteven Rostedt (Red Hat) 	if (ops->func_hash != &global_ops.local_hash)
179984261912SSteven Rostedt (Red Hat) 		return;
180084261912SSteven Rostedt (Red Hat) 
180184261912SSteven Rostedt (Red Hat) 	/*
180284261912SSteven Rostedt (Red Hat) 	 * If the ops shares the global_ops hash, then we need to update
180384261912SSteven Rostedt (Red Hat) 	 * all ops that are enabled and use this hash.
180484261912SSteven Rostedt (Red Hat) 	 */
180584261912SSteven Rostedt (Red Hat) 	do_for_each_ftrace_op(op, ftrace_ops_list) {
180684261912SSteven Rostedt (Red Hat) 		/* Already done */
180784261912SSteven Rostedt (Red Hat) 		if (op == ops)
180884261912SSteven Rostedt (Red Hat) 			continue;
180984261912SSteven Rostedt (Red Hat) 		if (op->func_hash == &global_ops.local_hash)
181084261912SSteven Rostedt (Red Hat) 			__ftrace_hash_rec_update(op, filter_hash, inc);
181184261912SSteven Rostedt (Red Hat) 	} while_for_each_ftrace_op(op);
181284261912SSteven Rostedt (Red Hat) }
181384261912SSteven Rostedt (Red Hat) 
181484261912SSteven Rostedt (Red Hat) static void ftrace_hash_rec_disable_modify(struct ftrace_ops *ops,
181584261912SSteven Rostedt (Red Hat) 					   int filter_hash)
181684261912SSteven Rostedt (Red Hat) {
181784261912SSteven Rostedt (Red Hat) 	ftrace_hash_rec_update_modify(ops, filter_hash, 0);
181884261912SSteven Rostedt (Red Hat) }
181984261912SSteven Rostedt (Red Hat) 
182084261912SSteven Rostedt (Red Hat) static void ftrace_hash_rec_enable_modify(struct ftrace_ops *ops,
182184261912SSteven Rostedt (Red Hat) 					  int filter_hash)
182284261912SSteven Rostedt (Red Hat) {
182384261912SSteven Rostedt (Red Hat) 	ftrace_hash_rec_update_modify(ops, filter_hash, 1);
182484261912SSteven Rostedt (Red Hat) }
182584261912SSteven Rostedt (Red Hat) 
1826f8b8be8aSMasami Hiramatsu /*
1827f8b8be8aSMasami Hiramatsu  * Try to update IPMODIFY flag on each ftrace_rec. Return 0 if it is OK
1828f8b8be8aSMasami Hiramatsu  * or no-needed to update, -EBUSY if it detects a conflict of the flag
1829f8b8be8aSMasami Hiramatsu  * on a ftrace_rec, and -EINVAL if the new_hash tries to trace all recs.
1830f8b8be8aSMasami Hiramatsu  * Note that old_hash and new_hash has below meanings
1831f8b8be8aSMasami Hiramatsu  *  - If the hash is NULL, it hits all recs (if IPMODIFY is set, this is rejected)
1832f8b8be8aSMasami Hiramatsu  *  - If the hash is EMPTY_HASH, it hits nothing
1833f8b8be8aSMasami Hiramatsu  *  - Anything else hits the recs which match the hash entries.
1834f8b8be8aSMasami Hiramatsu  */
1835f8b8be8aSMasami Hiramatsu static int __ftrace_hash_update_ipmodify(struct ftrace_ops *ops,
1836f8b8be8aSMasami Hiramatsu 					 struct ftrace_hash *old_hash,
1837f8b8be8aSMasami Hiramatsu 					 struct ftrace_hash *new_hash)
1838f8b8be8aSMasami Hiramatsu {
1839f8b8be8aSMasami Hiramatsu 	struct ftrace_page *pg;
1840f8b8be8aSMasami Hiramatsu 	struct dyn_ftrace *rec, *end = NULL;
1841f8b8be8aSMasami Hiramatsu 	int in_old, in_new;
1842f8b8be8aSMasami Hiramatsu 
1843f8b8be8aSMasami Hiramatsu 	/* Only update if the ops has been registered */
1844f8b8be8aSMasami Hiramatsu 	if (!(ops->flags & FTRACE_OPS_FL_ENABLED))
1845f8b8be8aSMasami Hiramatsu 		return 0;
1846f8b8be8aSMasami Hiramatsu 
1847f8b8be8aSMasami Hiramatsu 	if (!(ops->flags & FTRACE_OPS_FL_IPMODIFY))
1848f8b8be8aSMasami Hiramatsu 		return 0;
1849f8b8be8aSMasami Hiramatsu 
1850f8b8be8aSMasami Hiramatsu 	/*
1851f8b8be8aSMasami Hiramatsu 	 * Since the IPMODIFY is a very address sensitive action, we do not
1852f8b8be8aSMasami Hiramatsu 	 * allow ftrace_ops to set all functions to new hash.
1853f8b8be8aSMasami Hiramatsu 	 */
1854f8b8be8aSMasami Hiramatsu 	if (!new_hash || !old_hash)
1855f8b8be8aSMasami Hiramatsu 		return -EINVAL;
1856f8b8be8aSMasami Hiramatsu 
1857f8b8be8aSMasami Hiramatsu 	/* Update rec->flags */
1858f8b8be8aSMasami Hiramatsu 	do_for_each_ftrace_rec(pg, rec) {
1859f8b8be8aSMasami Hiramatsu 		/* We need to update only differences of filter_hash */
1860f8b8be8aSMasami Hiramatsu 		in_old = !!ftrace_lookup_ip(old_hash, rec->ip);
1861f8b8be8aSMasami Hiramatsu 		in_new = !!ftrace_lookup_ip(new_hash, rec->ip);
1862f8b8be8aSMasami Hiramatsu 		if (in_old == in_new)
1863f8b8be8aSMasami Hiramatsu 			continue;
1864f8b8be8aSMasami Hiramatsu 
1865f8b8be8aSMasami Hiramatsu 		if (in_new) {
1866f8b8be8aSMasami Hiramatsu 			/* New entries must ensure no others are using it */
1867f8b8be8aSMasami Hiramatsu 			if (rec->flags & FTRACE_FL_IPMODIFY)
1868f8b8be8aSMasami Hiramatsu 				goto rollback;
1869f8b8be8aSMasami Hiramatsu 			rec->flags |= FTRACE_FL_IPMODIFY;
1870f8b8be8aSMasami Hiramatsu 		} else /* Removed entry */
1871f8b8be8aSMasami Hiramatsu 			rec->flags &= ~FTRACE_FL_IPMODIFY;
1872f8b8be8aSMasami Hiramatsu 	} while_for_each_ftrace_rec();
1873f8b8be8aSMasami Hiramatsu 
1874f8b8be8aSMasami Hiramatsu 	return 0;
1875f8b8be8aSMasami Hiramatsu 
1876f8b8be8aSMasami Hiramatsu rollback:
1877f8b8be8aSMasami Hiramatsu 	end = rec;
1878f8b8be8aSMasami Hiramatsu 
1879f8b8be8aSMasami Hiramatsu 	/* Roll back what we did above */
1880f8b8be8aSMasami Hiramatsu 	do_for_each_ftrace_rec(pg, rec) {
1881f8b8be8aSMasami Hiramatsu 		if (rec == end)
1882f8b8be8aSMasami Hiramatsu 			goto err_out;
1883f8b8be8aSMasami Hiramatsu 
1884f8b8be8aSMasami Hiramatsu 		in_old = !!ftrace_lookup_ip(old_hash, rec->ip);
1885f8b8be8aSMasami Hiramatsu 		in_new = !!ftrace_lookup_ip(new_hash, rec->ip);
1886f8b8be8aSMasami Hiramatsu 		if (in_old == in_new)
1887f8b8be8aSMasami Hiramatsu 			continue;
1888f8b8be8aSMasami Hiramatsu 
1889f8b8be8aSMasami Hiramatsu 		if (in_new)
1890f8b8be8aSMasami Hiramatsu 			rec->flags &= ~FTRACE_FL_IPMODIFY;
1891f8b8be8aSMasami Hiramatsu 		else
1892f8b8be8aSMasami Hiramatsu 			rec->flags |= FTRACE_FL_IPMODIFY;
1893f8b8be8aSMasami Hiramatsu 	} while_for_each_ftrace_rec();
1894f8b8be8aSMasami Hiramatsu 
1895f8b8be8aSMasami Hiramatsu err_out:
1896f8b8be8aSMasami Hiramatsu 	return -EBUSY;
1897f8b8be8aSMasami Hiramatsu }
1898f8b8be8aSMasami Hiramatsu 
1899f8b8be8aSMasami Hiramatsu static int ftrace_hash_ipmodify_enable(struct ftrace_ops *ops)
1900f8b8be8aSMasami Hiramatsu {
1901f8b8be8aSMasami Hiramatsu 	struct ftrace_hash *hash = ops->func_hash->filter_hash;
1902f8b8be8aSMasami Hiramatsu 
1903f8b8be8aSMasami Hiramatsu 	if (ftrace_hash_empty(hash))
1904f8b8be8aSMasami Hiramatsu 		hash = NULL;
1905f8b8be8aSMasami Hiramatsu 
1906f8b8be8aSMasami Hiramatsu 	return __ftrace_hash_update_ipmodify(ops, EMPTY_HASH, hash);
1907f8b8be8aSMasami Hiramatsu }
1908f8b8be8aSMasami Hiramatsu 
1909f8b8be8aSMasami Hiramatsu /* Disabling always succeeds */
1910f8b8be8aSMasami Hiramatsu static void ftrace_hash_ipmodify_disable(struct ftrace_ops *ops)
1911f8b8be8aSMasami Hiramatsu {
1912f8b8be8aSMasami Hiramatsu 	struct ftrace_hash *hash = ops->func_hash->filter_hash;
1913f8b8be8aSMasami Hiramatsu 
1914f8b8be8aSMasami Hiramatsu 	if (ftrace_hash_empty(hash))
1915f8b8be8aSMasami Hiramatsu 		hash = NULL;
1916f8b8be8aSMasami Hiramatsu 
1917f8b8be8aSMasami Hiramatsu 	__ftrace_hash_update_ipmodify(ops, hash, EMPTY_HASH);
1918f8b8be8aSMasami Hiramatsu }
1919f8b8be8aSMasami Hiramatsu 
1920f8b8be8aSMasami Hiramatsu static int ftrace_hash_ipmodify_update(struct ftrace_ops *ops,
1921f8b8be8aSMasami Hiramatsu 				       struct ftrace_hash *new_hash)
1922f8b8be8aSMasami Hiramatsu {
1923f8b8be8aSMasami Hiramatsu 	struct ftrace_hash *old_hash = ops->func_hash->filter_hash;
1924f8b8be8aSMasami Hiramatsu 
1925f8b8be8aSMasami Hiramatsu 	if (ftrace_hash_empty(old_hash))
1926f8b8be8aSMasami Hiramatsu 		old_hash = NULL;
1927f8b8be8aSMasami Hiramatsu 
1928f8b8be8aSMasami Hiramatsu 	if (ftrace_hash_empty(new_hash))
1929f8b8be8aSMasami Hiramatsu 		new_hash = NULL;
1930f8b8be8aSMasami Hiramatsu 
1931f8b8be8aSMasami Hiramatsu 	return __ftrace_hash_update_ipmodify(ops, old_hash, new_hash);
1932f8b8be8aSMasami Hiramatsu }
1933f8b8be8aSMasami Hiramatsu 
1934b05086c7SSteven Rostedt (Red Hat) static void print_ip_ins(const char *fmt, const unsigned char *p)
193505736a42SSteven Rostedt {
193605736a42SSteven Rostedt 	int i;
193705736a42SSteven Rostedt 
193805736a42SSteven Rostedt 	printk(KERN_CONT "%s", fmt);
193905736a42SSteven Rostedt 
194005736a42SSteven Rostedt 	for (i = 0; i < MCOUNT_INSN_SIZE; i++)
194105736a42SSteven Rostedt 		printk(KERN_CONT "%s%02x", i ? ":" : "", p[i]);
194205736a42SSteven Rostedt }
194305736a42SSteven Rostedt 
19444fd3279bSSteven Rostedt (Red Hat) static struct ftrace_ops *
19454fd3279bSSteven Rostedt (Red Hat) ftrace_find_tramp_ops_any(struct dyn_ftrace *rec);
194639daa7b9SSteven Rostedt (Red Hat) static struct ftrace_ops *
194739daa7b9SSteven Rostedt (Red Hat) ftrace_find_tramp_ops_next(struct dyn_ftrace *rec, struct ftrace_ops *ops);
19484fd3279bSSteven Rostedt (Red Hat) 
194902a392a0SSteven Rostedt (Red Hat) enum ftrace_bug_type ftrace_bug_type;
1950b05086c7SSteven Rostedt (Red Hat) const void *ftrace_expected;
195102a392a0SSteven Rostedt (Red Hat) 
195202a392a0SSteven Rostedt (Red Hat) static void print_bug_type(void)
195302a392a0SSteven Rostedt (Red Hat) {
195402a392a0SSteven Rostedt (Red Hat) 	switch (ftrace_bug_type) {
195502a392a0SSteven Rostedt (Red Hat) 	case FTRACE_BUG_UNKNOWN:
195602a392a0SSteven Rostedt (Red Hat) 		break;
195702a392a0SSteven Rostedt (Red Hat) 	case FTRACE_BUG_INIT:
195802a392a0SSteven Rostedt (Red Hat) 		pr_info("Initializing ftrace call sites\n");
195902a392a0SSteven Rostedt (Red Hat) 		break;
196002a392a0SSteven Rostedt (Red Hat) 	case FTRACE_BUG_NOP:
196102a392a0SSteven Rostedt (Red Hat) 		pr_info("Setting ftrace call site to NOP\n");
196202a392a0SSteven Rostedt (Red Hat) 		break;
196302a392a0SSteven Rostedt (Red Hat) 	case FTRACE_BUG_CALL:
196402a392a0SSteven Rostedt (Red Hat) 		pr_info("Setting ftrace call site to call ftrace function\n");
196502a392a0SSteven Rostedt (Red Hat) 		break;
196602a392a0SSteven Rostedt (Red Hat) 	case FTRACE_BUG_UPDATE:
196702a392a0SSteven Rostedt (Red Hat) 		pr_info("Updating ftrace call site to call a different ftrace function\n");
196802a392a0SSteven Rostedt (Red Hat) 		break;
196902a392a0SSteven Rostedt (Red Hat) 	}
197002a392a0SSteven Rostedt (Red Hat) }
197102a392a0SSteven Rostedt (Red Hat) 
1972c88fd863SSteven Rostedt /**
1973c88fd863SSteven Rostedt  * ftrace_bug - report and shutdown function tracer
1974c88fd863SSteven Rostedt  * @failed: The failed type (EFAULT, EINVAL, EPERM)
19754fd3279bSSteven Rostedt (Red Hat)  * @rec: The record that failed
1976c88fd863SSteven Rostedt  *
1977c88fd863SSteven Rostedt  * The arch code that enables or disables the function tracing
1978c88fd863SSteven Rostedt  * can call ftrace_bug() when it has detected a problem in
1979c88fd863SSteven Rostedt  * modifying the code. @failed should be one of either:
1980c88fd863SSteven Rostedt  * EFAULT - if the problem happens on reading the @ip address
1981c88fd863SSteven Rostedt  * EINVAL - if what is read at @ip is not what was expected
1982c88fd863SSteven Rostedt  * EPERM - if the problem happens on writting to the @ip address
1983c88fd863SSteven Rostedt  */
19844fd3279bSSteven Rostedt (Red Hat) void ftrace_bug(int failed, struct dyn_ftrace *rec)
19853c1720f0SSteven Rostedt {
19864fd3279bSSteven Rostedt (Red Hat) 	unsigned long ip = rec ? rec->ip : 0;
19874fd3279bSSteven Rostedt (Red Hat) 
1988b17e8a37SSteven Rostedt 	switch (failed) {
1989593eb8a2SSteven Rostedt 	case -EFAULT:
19906912896eSSteven Rostedt 		FTRACE_WARN_ON_ONCE(1);
199105736a42SSteven Rostedt 		pr_info("ftrace faulted on modifying ");
199205736a42SSteven Rostedt 		print_ip_sym(ip);
199305736a42SSteven Rostedt 		break;
1994593eb8a2SSteven Rostedt 	case -EINVAL:
19956912896eSSteven Rostedt 		FTRACE_WARN_ON_ONCE(1);
199605736a42SSteven Rostedt 		pr_info("ftrace failed to modify ");
199705736a42SSteven Rostedt 		print_ip_sym(ip);
199805736a42SSteven Rostedt 		print_ip_ins(" actual:   ", (unsigned char *)ip);
19994fd3279bSSteven Rostedt (Red Hat) 		pr_cont("\n");
2000b05086c7SSteven Rostedt (Red Hat) 		if (ftrace_expected) {
2001b05086c7SSteven Rostedt (Red Hat) 			print_ip_ins(" expected: ", ftrace_expected);
2002b05086c7SSteven Rostedt (Red Hat) 			pr_cont("\n");
2003b05086c7SSteven Rostedt (Red Hat) 		}
200405736a42SSteven Rostedt 		break;
2005593eb8a2SSteven Rostedt 	case -EPERM:
20066912896eSSteven Rostedt 		FTRACE_WARN_ON_ONCE(1);
2007593eb8a2SSteven Rostedt 		pr_info("ftrace faulted on writing ");
2008593eb8a2SSteven Rostedt 		print_ip_sym(ip);
2009593eb8a2SSteven Rostedt 		break;
2010593eb8a2SSteven Rostedt 	default:
20116912896eSSteven Rostedt 		FTRACE_WARN_ON_ONCE(1);
2012593eb8a2SSteven Rostedt 		pr_info("ftrace faulted on unknown error ");
2013593eb8a2SSteven Rostedt 		print_ip_sym(ip);
201405736a42SSteven Rostedt 	}
201502a392a0SSteven Rostedt (Red Hat) 	print_bug_type();
20164fd3279bSSteven Rostedt (Red Hat) 	if (rec) {
20174fd3279bSSteven Rostedt (Red Hat) 		struct ftrace_ops *ops = NULL;
20184fd3279bSSteven Rostedt (Red Hat) 
20194fd3279bSSteven Rostedt (Red Hat) 		pr_info("ftrace record flags: %lx\n", rec->flags);
20204fd3279bSSteven Rostedt (Red Hat) 		pr_cont(" (%ld)%s", ftrace_rec_count(rec),
20214fd3279bSSteven Rostedt (Red Hat) 			rec->flags & FTRACE_FL_REGS ? " R" : "  ");
20224fd3279bSSteven Rostedt (Red Hat) 		if (rec->flags & FTRACE_FL_TRAMP_EN) {
20234fd3279bSSteven Rostedt (Red Hat) 			ops = ftrace_find_tramp_ops_any(rec);
202439daa7b9SSteven Rostedt (Red Hat) 			if (ops) {
202539daa7b9SSteven Rostedt (Red Hat) 				do {
202639daa7b9SSteven Rostedt (Red Hat) 					pr_cont("\ttramp: %pS (%pS)",
202739daa7b9SSteven Rostedt (Red Hat) 						(void *)ops->trampoline,
202839daa7b9SSteven Rostedt (Red Hat) 						(void *)ops->func);
202939daa7b9SSteven Rostedt (Red Hat) 					ops = ftrace_find_tramp_ops_next(rec, ops);
203039daa7b9SSteven Rostedt (Red Hat) 				} while (ops);
203139daa7b9SSteven Rostedt (Red Hat) 			} else
20324fd3279bSSteven Rostedt (Red Hat) 				pr_cont("\ttramp: ERROR!");
20334fd3279bSSteven Rostedt (Red Hat) 
20344fd3279bSSteven Rostedt (Red Hat) 		}
20354fd3279bSSteven Rostedt (Red Hat) 		ip = ftrace_get_addr_curr(rec);
203639daa7b9SSteven Rostedt (Red Hat) 		pr_cont("\n expected tramp: %lx\n", ip);
20374fd3279bSSteven Rostedt (Red Hat) 	}
2038b17e8a37SSteven Rostedt }
203905736a42SSteven Rostedt 
2040c88fd863SSteven Rostedt static int ftrace_check_record(struct dyn_ftrace *rec, int enable, int update)
20413d083395SSteven Rostedt {
204264fbcd16SXiao Guangrong 	unsigned long flag = 0UL;
2043e7d3737eSFrederic Weisbecker 
204402a392a0SSteven Rostedt (Red Hat) 	ftrace_bug_type = FTRACE_BUG_UNKNOWN;
204502a392a0SSteven Rostedt (Red Hat) 
2046b7ffffbbSSteven Rostedt (Red Hat) 	if (rec->flags & FTRACE_FL_DISABLED)
2047b7ffffbbSSteven Rostedt (Red Hat) 		return FTRACE_UPDATE_IGNORE;
2048b7ffffbbSSteven Rostedt (Red Hat) 
20493d083395SSteven Rostedt 	/*
205030fb6aa7SJiri Olsa 	 * If we are updating calls:
20513d083395SSteven Rostedt 	 *
2052ed926f9bSSteven Rostedt 	 *   If the record has a ref count, then we need to enable it
2053ed926f9bSSteven Rostedt 	 *   because someone is using it.
20543d083395SSteven Rostedt 	 *
2055ed926f9bSSteven Rostedt 	 *   Otherwise we make sure its disabled.
2056ed926f9bSSteven Rostedt 	 *
205730fb6aa7SJiri Olsa 	 * If we are disabling calls, then disable all records that
2058ed926f9bSSteven Rostedt 	 * are enabled.
20593d083395SSteven Rostedt 	 */
20600376bde1SSteven Rostedt (Red Hat) 	if (enable && ftrace_rec_count(rec))
206164fbcd16SXiao Guangrong 		flag = FTRACE_FL_ENABLED;
20623d083395SSteven Rostedt 
206308f6fba5SSteven Rostedt 	/*
206479922b80SSteven Rostedt (Red Hat) 	 * If enabling and the REGS flag does not match the REGS_EN, or
206579922b80SSteven Rostedt (Red Hat) 	 * the TRAMP flag doesn't match the TRAMP_EN, then do not ignore
206679922b80SSteven Rostedt (Red Hat) 	 * this record. Set flags to fail the compare against ENABLED.
206708f6fba5SSteven Rostedt 	 */
206879922b80SSteven Rostedt (Red Hat) 	if (flag) {
206979922b80SSteven Rostedt (Red Hat) 		if (!(rec->flags & FTRACE_FL_REGS) !=
207079922b80SSteven Rostedt (Red Hat) 		    !(rec->flags & FTRACE_FL_REGS_EN))
207108f6fba5SSteven Rostedt 			flag |= FTRACE_FL_REGS;
207208f6fba5SSteven Rostedt 
207379922b80SSteven Rostedt (Red Hat) 		if (!(rec->flags & FTRACE_FL_TRAMP) !=
207479922b80SSteven Rostedt (Red Hat) 		    !(rec->flags & FTRACE_FL_TRAMP_EN))
207579922b80SSteven Rostedt (Red Hat) 			flag |= FTRACE_FL_TRAMP;
207679922b80SSteven Rostedt (Red Hat) 	}
207779922b80SSteven Rostedt (Red Hat) 
207864fbcd16SXiao Guangrong 	/* If the state of this record hasn't changed, then do nothing */
207964fbcd16SXiao Guangrong 	if ((rec->flags & FTRACE_FL_ENABLED) == flag)
2080c88fd863SSteven Rostedt 		return FTRACE_UPDATE_IGNORE;
208164fbcd16SXiao Guangrong 
208264fbcd16SXiao Guangrong 	if (flag) {
208308f6fba5SSteven Rostedt 		/* Save off if rec is being enabled (for return value) */
208408f6fba5SSteven Rostedt 		flag ^= rec->flags & FTRACE_FL_ENABLED;
208508f6fba5SSteven Rostedt 
208608f6fba5SSteven Rostedt 		if (update) {
208764fbcd16SXiao Guangrong 			rec->flags |= FTRACE_FL_ENABLED;
208808f6fba5SSteven Rostedt 			if (flag & FTRACE_FL_REGS) {
208908f6fba5SSteven Rostedt 				if (rec->flags & FTRACE_FL_REGS)
209008f6fba5SSteven Rostedt 					rec->flags |= FTRACE_FL_REGS_EN;
209108f6fba5SSteven Rostedt 				else
209208f6fba5SSteven Rostedt 					rec->flags &= ~FTRACE_FL_REGS_EN;
209308f6fba5SSteven Rostedt 			}
209479922b80SSteven Rostedt (Red Hat) 			if (flag & FTRACE_FL_TRAMP) {
209579922b80SSteven Rostedt (Red Hat) 				if (rec->flags & FTRACE_FL_TRAMP)
209679922b80SSteven Rostedt (Red Hat) 					rec->flags |= FTRACE_FL_TRAMP_EN;
209779922b80SSteven Rostedt (Red Hat) 				else
209879922b80SSteven Rostedt (Red Hat) 					rec->flags &= ~FTRACE_FL_TRAMP_EN;
209979922b80SSteven Rostedt (Red Hat) 			}
210064fbcd16SXiao Guangrong 		}
210164fbcd16SXiao Guangrong 
210208f6fba5SSteven Rostedt 		/*
210308f6fba5SSteven Rostedt 		 * If this record is being updated from a nop, then
210408f6fba5SSteven Rostedt 		 *   return UPDATE_MAKE_CALL.
210508f6fba5SSteven Rostedt 		 * Otherwise,
210608f6fba5SSteven Rostedt 		 *   return UPDATE_MODIFY_CALL to tell the caller to convert
2107f1b2f2bdSSteven Rostedt (Red Hat) 		 *   from the save regs, to a non-save regs function or
210879922b80SSteven Rostedt (Red Hat) 		 *   vice versa, or from a trampoline call.
210908f6fba5SSteven Rostedt 		 */
211002a392a0SSteven Rostedt (Red Hat) 		if (flag & FTRACE_FL_ENABLED) {
211102a392a0SSteven Rostedt (Red Hat) 			ftrace_bug_type = FTRACE_BUG_CALL;
211208f6fba5SSteven Rostedt 			return FTRACE_UPDATE_MAKE_CALL;
211302a392a0SSteven Rostedt (Red Hat) 		}
2114f1b2f2bdSSteven Rostedt (Red Hat) 
211502a392a0SSteven Rostedt (Red Hat) 		ftrace_bug_type = FTRACE_BUG_UPDATE;
211608f6fba5SSteven Rostedt 		return FTRACE_UPDATE_MODIFY_CALL;
211708f6fba5SSteven Rostedt 	}
211808f6fba5SSteven Rostedt 
211908f6fba5SSteven Rostedt 	if (update) {
212008f6fba5SSteven Rostedt 		/* If there's no more users, clear all flags */
21210376bde1SSteven Rostedt (Red Hat) 		if (!ftrace_rec_count(rec))
212208f6fba5SSteven Rostedt 			rec->flags = 0;
212308f6fba5SSteven Rostedt 		else
2124b24d443bSSteven Rostedt (Red Hat) 			/*
2125b24d443bSSteven Rostedt (Red Hat) 			 * Just disable the record, but keep the ops TRAMP
2126b24d443bSSteven Rostedt (Red Hat) 			 * and REGS states. The _EN flags must be disabled though.
2127b24d443bSSteven Rostedt (Red Hat) 			 */
2128b24d443bSSteven Rostedt (Red Hat) 			rec->flags &= ~(FTRACE_FL_ENABLED | FTRACE_FL_TRAMP_EN |
2129b24d443bSSteven Rostedt (Red Hat) 					FTRACE_FL_REGS_EN);
213008f6fba5SSteven Rostedt 	}
2131c88fd863SSteven Rostedt 
213202a392a0SSteven Rostedt (Red Hat) 	ftrace_bug_type = FTRACE_BUG_NOP;
2133c88fd863SSteven Rostedt 	return FTRACE_UPDATE_MAKE_NOP;
2134c88fd863SSteven Rostedt }
2135c88fd863SSteven Rostedt 
2136c88fd863SSteven Rostedt /**
2137c88fd863SSteven Rostedt  * ftrace_update_record, set a record that now is tracing or not
2138c88fd863SSteven Rostedt  * @rec: the record to update
2139c88fd863SSteven Rostedt  * @enable: set to 1 if the record is tracing, zero to force disable
2140c88fd863SSteven Rostedt  *
2141c88fd863SSteven Rostedt  * The records that represent all functions that can be traced need
2142c88fd863SSteven Rostedt  * to be updated when tracing has been enabled.
2143c88fd863SSteven Rostedt  */
2144c88fd863SSteven Rostedt int ftrace_update_record(struct dyn_ftrace *rec, int enable)
2145c88fd863SSteven Rostedt {
2146c88fd863SSteven Rostedt 	return ftrace_check_record(rec, enable, 1);
2147c88fd863SSteven Rostedt }
2148c88fd863SSteven Rostedt 
2149c88fd863SSteven Rostedt /**
2150c88fd863SSteven Rostedt  * ftrace_test_record, check if the record has been enabled or not
2151c88fd863SSteven Rostedt  * @rec: the record to test
2152c88fd863SSteven Rostedt  * @enable: set to 1 to check if enabled, 0 if it is disabled
2153c88fd863SSteven Rostedt  *
2154c88fd863SSteven Rostedt  * The arch code may need to test if a record is already set to
2155c88fd863SSteven Rostedt  * tracing to determine how to modify the function code that it
2156c88fd863SSteven Rostedt  * represents.
2157c88fd863SSteven Rostedt  */
2158c88fd863SSteven Rostedt int ftrace_test_record(struct dyn_ftrace *rec, int enable)
2159c88fd863SSteven Rostedt {
2160c88fd863SSteven Rostedt 	return ftrace_check_record(rec, enable, 0);
2161c88fd863SSteven Rostedt }
2162c88fd863SSteven Rostedt 
216379922b80SSteven Rostedt (Red Hat) static struct ftrace_ops *
21645fecaa04SSteven Rostedt (Red Hat) ftrace_find_tramp_ops_any(struct dyn_ftrace *rec)
21655fecaa04SSteven Rostedt (Red Hat) {
21665fecaa04SSteven Rostedt (Red Hat) 	struct ftrace_ops *op;
2167fef5aeeeSSteven Rostedt (Red Hat) 	unsigned long ip = rec->ip;
21685fecaa04SSteven Rostedt (Red Hat) 
21695fecaa04SSteven Rostedt (Red Hat) 	do_for_each_ftrace_op(op, ftrace_ops_list) {
21705fecaa04SSteven Rostedt (Red Hat) 
21715fecaa04SSteven Rostedt (Red Hat) 		if (!op->trampoline)
21725fecaa04SSteven Rostedt (Red Hat) 			continue;
21735fecaa04SSteven Rostedt (Red Hat) 
2174fef5aeeeSSteven Rostedt (Red Hat) 		if (hash_contains_ip(ip, op->func_hash))
21755fecaa04SSteven Rostedt (Red Hat) 			return op;
21765fecaa04SSteven Rostedt (Red Hat) 	} while_for_each_ftrace_op(op);
21775fecaa04SSteven Rostedt (Red Hat) 
21785fecaa04SSteven Rostedt (Red Hat) 	return NULL;
21795fecaa04SSteven Rostedt (Red Hat) }
21805fecaa04SSteven Rostedt (Red Hat) 
21815fecaa04SSteven Rostedt (Red Hat) static struct ftrace_ops *
218239daa7b9SSteven Rostedt (Red Hat) ftrace_find_tramp_ops_next(struct dyn_ftrace *rec,
218339daa7b9SSteven Rostedt (Red Hat) 			   struct ftrace_ops *op)
218439daa7b9SSteven Rostedt (Red Hat) {
218539daa7b9SSteven Rostedt (Red Hat) 	unsigned long ip = rec->ip;
218639daa7b9SSteven Rostedt (Red Hat) 
218739daa7b9SSteven Rostedt (Red Hat) 	while_for_each_ftrace_op(op) {
218839daa7b9SSteven Rostedt (Red Hat) 
218939daa7b9SSteven Rostedt (Red Hat) 		if (!op->trampoline)
219039daa7b9SSteven Rostedt (Red Hat) 			continue;
219139daa7b9SSteven Rostedt (Red Hat) 
219239daa7b9SSteven Rostedt (Red Hat) 		if (hash_contains_ip(ip, op->func_hash))
219339daa7b9SSteven Rostedt (Red Hat) 			return op;
219439daa7b9SSteven Rostedt (Red Hat) 	}
219539daa7b9SSteven Rostedt (Red Hat) 
219639daa7b9SSteven Rostedt (Red Hat) 	return NULL;
219739daa7b9SSteven Rostedt (Red Hat) }
219839daa7b9SSteven Rostedt (Red Hat) 
219939daa7b9SSteven Rostedt (Red Hat) static struct ftrace_ops *
220079922b80SSteven Rostedt (Red Hat) ftrace_find_tramp_ops_curr(struct dyn_ftrace *rec)
220179922b80SSteven Rostedt (Red Hat) {
220279922b80SSteven Rostedt (Red Hat) 	struct ftrace_ops *op;
2203fef5aeeeSSteven Rostedt (Red Hat) 	unsigned long ip = rec->ip;
220479922b80SSteven Rostedt (Red Hat) 
2205fef5aeeeSSteven Rostedt (Red Hat) 	/*
2206fef5aeeeSSteven Rostedt (Red Hat) 	 * Need to check removed ops first.
2207fef5aeeeSSteven Rostedt (Red Hat) 	 * If they are being removed, and this rec has a tramp,
2208fef5aeeeSSteven Rostedt (Red Hat) 	 * and this rec is in the ops list, then it would be the
2209fef5aeeeSSteven Rostedt (Red Hat) 	 * one with the tramp.
2210fef5aeeeSSteven Rostedt (Red Hat) 	 */
2211fef5aeeeSSteven Rostedt (Red Hat) 	if (removed_ops) {
2212fef5aeeeSSteven Rostedt (Red Hat) 		if (hash_contains_ip(ip, &removed_ops->old_hash))
221379922b80SSteven Rostedt (Red Hat) 			return removed_ops;
221479922b80SSteven Rostedt (Red Hat) 	}
221579922b80SSteven Rostedt (Red Hat) 
2216fef5aeeeSSteven Rostedt (Red Hat) 	/*
2217fef5aeeeSSteven Rostedt (Red Hat) 	 * Need to find the current trampoline for a rec.
2218fef5aeeeSSteven Rostedt (Red Hat) 	 * Now, a trampoline is only attached to a rec if there
2219fef5aeeeSSteven Rostedt (Red Hat) 	 * was a single 'ops' attached to it. But this can be called
2220fef5aeeeSSteven Rostedt (Red Hat) 	 * when we are adding another op to the rec or removing the
2221fef5aeeeSSteven Rostedt (Red Hat) 	 * current one. Thus, if the op is being added, we can
2222fef5aeeeSSteven Rostedt (Red Hat) 	 * ignore it because it hasn't attached itself to the rec
22234fc40904SSteven Rostedt (Red Hat) 	 * yet.
22244fc40904SSteven Rostedt (Red Hat) 	 *
22254fc40904SSteven Rostedt (Red Hat) 	 * If an ops is being modified (hooking to different functions)
22264fc40904SSteven Rostedt (Red Hat) 	 * then we don't care about the new functions that are being
22274fc40904SSteven Rostedt (Red Hat) 	 * added, just the old ones (that are probably being removed).
22284fc40904SSteven Rostedt (Red Hat) 	 *
22294fc40904SSteven Rostedt (Red Hat) 	 * If we are adding an ops to a function that already is using
22304fc40904SSteven Rostedt (Red Hat) 	 * a trampoline, it needs to be removed (trampolines are only
22314fc40904SSteven Rostedt (Red Hat) 	 * for single ops connected), then an ops that is not being
22324fc40904SSteven Rostedt (Red Hat) 	 * modified also needs to be checked.
2233fef5aeeeSSteven Rostedt (Red Hat) 	 */
223479922b80SSteven Rostedt (Red Hat) 	do_for_each_ftrace_op(op, ftrace_ops_list) {
2235fef5aeeeSSteven Rostedt (Red Hat) 
2236fef5aeeeSSteven Rostedt (Red Hat) 		if (!op->trampoline)
223779922b80SSteven Rostedt (Red Hat) 			continue;
223879922b80SSteven Rostedt (Red Hat) 
2239fef5aeeeSSteven Rostedt (Red Hat) 		/*
2240fef5aeeeSSteven Rostedt (Red Hat) 		 * If the ops is being added, it hasn't gotten to
2241fef5aeeeSSteven Rostedt (Red Hat) 		 * the point to be removed from this tree yet.
2242fef5aeeeSSteven Rostedt (Red Hat) 		 */
2243fef5aeeeSSteven Rostedt (Red Hat) 		if (op->flags & FTRACE_OPS_FL_ADDING)
2244fef5aeeeSSteven Rostedt (Red Hat) 			continue;
2245fef5aeeeSSteven Rostedt (Red Hat) 
2246fef5aeeeSSteven Rostedt (Red Hat) 
22474fc40904SSteven Rostedt (Red Hat) 		/*
22484fc40904SSteven Rostedt (Red Hat) 		 * If the ops is being modified and is in the old
22494fc40904SSteven Rostedt (Red Hat) 		 * hash, then it is probably being removed from this
22504fc40904SSteven Rostedt (Red Hat) 		 * function.
22514fc40904SSteven Rostedt (Red Hat) 		 */
2252fef5aeeeSSteven Rostedt (Red Hat) 		if ((op->flags & FTRACE_OPS_FL_MODIFYING) &&
2253fef5aeeeSSteven Rostedt (Red Hat) 		    hash_contains_ip(ip, &op->old_hash))
225479922b80SSteven Rostedt (Red Hat) 			return op;
22554fc40904SSteven Rostedt (Red Hat) 		/*
22564fc40904SSteven Rostedt (Red Hat) 		 * If the ops is not being added or modified, and it's
22574fc40904SSteven Rostedt (Red Hat) 		 * in its normal filter hash, then this must be the one
22584fc40904SSteven Rostedt (Red Hat) 		 * we want!
22594fc40904SSteven Rostedt (Red Hat) 		 */
22604fc40904SSteven Rostedt (Red Hat) 		if (!(op->flags & FTRACE_OPS_FL_MODIFYING) &&
22614fc40904SSteven Rostedt (Red Hat) 		    hash_contains_ip(ip, op->func_hash))
22624fc40904SSteven Rostedt (Red Hat) 			return op;
226379922b80SSteven Rostedt (Red Hat) 
226479922b80SSteven Rostedt (Red Hat) 	} while_for_each_ftrace_op(op);
226579922b80SSteven Rostedt (Red Hat) 
226679922b80SSteven Rostedt (Red Hat) 	return NULL;
226779922b80SSteven Rostedt (Red Hat) }
226879922b80SSteven Rostedt (Red Hat) 
226979922b80SSteven Rostedt (Red Hat) static struct ftrace_ops *
227079922b80SSteven Rostedt (Red Hat) ftrace_find_tramp_ops_new(struct dyn_ftrace *rec)
227179922b80SSteven Rostedt (Red Hat) {
227279922b80SSteven Rostedt (Red Hat) 	struct ftrace_ops *op;
2273fef5aeeeSSteven Rostedt (Red Hat) 	unsigned long ip = rec->ip;
227479922b80SSteven Rostedt (Red Hat) 
227579922b80SSteven Rostedt (Red Hat) 	do_for_each_ftrace_op(op, ftrace_ops_list) {
227679922b80SSteven Rostedt (Red Hat) 		/* pass rec in as regs to have non-NULL val */
2277fef5aeeeSSteven Rostedt (Red Hat) 		if (hash_contains_ip(ip, op->func_hash))
227879922b80SSteven Rostedt (Red Hat) 			return op;
227979922b80SSteven Rostedt (Red Hat) 	} while_for_each_ftrace_op(op);
228079922b80SSteven Rostedt (Red Hat) 
228179922b80SSteven Rostedt (Red Hat) 	return NULL;
228279922b80SSteven Rostedt (Red Hat) }
228379922b80SSteven Rostedt (Red Hat) 
22847413af1fSSteven Rostedt (Red Hat) /**
22857413af1fSSteven Rostedt (Red Hat)  * ftrace_get_addr_new - Get the call address to set to
22867413af1fSSteven Rostedt (Red Hat)  * @rec:  The ftrace record descriptor
22877413af1fSSteven Rostedt (Red Hat)  *
22887413af1fSSteven Rostedt (Red Hat)  * If the record has the FTRACE_FL_REGS set, that means that it
22897413af1fSSteven Rostedt (Red Hat)  * wants to convert to a callback that saves all regs. If FTRACE_FL_REGS
22907413af1fSSteven Rostedt (Red Hat)  * is not not set, then it wants to convert to the normal callback.
22917413af1fSSteven Rostedt (Red Hat)  *
22927413af1fSSteven Rostedt (Red Hat)  * Returns the address of the trampoline to set to
22937413af1fSSteven Rostedt (Red Hat)  */
22947413af1fSSteven Rostedt (Red Hat) unsigned long ftrace_get_addr_new(struct dyn_ftrace *rec)
22957413af1fSSteven Rostedt (Red Hat) {
229679922b80SSteven Rostedt (Red Hat) 	struct ftrace_ops *ops;
229779922b80SSteven Rostedt (Red Hat) 
229879922b80SSteven Rostedt (Red Hat) 	/* Trampolines take precedence over regs */
229979922b80SSteven Rostedt (Red Hat) 	if (rec->flags & FTRACE_FL_TRAMP) {
230079922b80SSteven Rostedt (Red Hat) 		ops = ftrace_find_tramp_ops_new(rec);
230179922b80SSteven Rostedt (Red Hat) 		if (FTRACE_WARN_ON(!ops || !ops->trampoline)) {
2302bce0b6c5SSteven Rostedt (Red Hat) 			pr_warn("Bad trampoline accounting at: %p (%pS) (%lx)\n",
2303bce0b6c5SSteven Rostedt (Red Hat) 				(void *)rec->ip, (void *)rec->ip, rec->flags);
230479922b80SSteven Rostedt (Red Hat) 			/* Ftrace is shutting down, return anything */
230579922b80SSteven Rostedt (Red Hat) 			return (unsigned long)FTRACE_ADDR;
230679922b80SSteven Rostedt (Red Hat) 		}
230779922b80SSteven Rostedt (Red Hat) 		return ops->trampoline;
230879922b80SSteven Rostedt (Red Hat) 	}
230979922b80SSteven Rostedt (Red Hat) 
23107413af1fSSteven Rostedt (Red Hat) 	if (rec->flags & FTRACE_FL_REGS)
23117413af1fSSteven Rostedt (Red Hat) 		return (unsigned long)FTRACE_REGS_ADDR;
23127413af1fSSteven Rostedt (Red Hat) 	else
23137413af1fSSteven Rostedt (Red Hat) 		return (unsigned long)FTRACE_ADDR;
23147413af1fSSteven Rostedt (Red Hat) }
23157413af1fSSteven Rostedt (Red Hat) 
23167413af1fSSteven Rostedt (Red Hat) /**
23177413af1fSSteven Rostedt (Red Hat)  * ftrace_get_addr_curr - Get the call address that is already there
23187413af1fSSteven Rostedt (Red Hat)  * @rec:  The ftrace record descriptor
23197413af1fSSteven Rostedt (Red Hat)  *
23207413af1fSSteven Rostedt (Red Hat)  * The FTRACE_FL_REGS_EN is set when the record already points to
23217413af1fSSteven Rostedt (Red Hat)  * a function that saves all the regs. Basically the '_EN' version
23227413af1fSSteven Rostedt (Red Hat)  * represents the current state of the function.
23237413af1fSSteven Rostedt (Red Hat)  *
23247413af1fSSteven Rostedt (Red Hat)  * Returns the address of the trampoline that is currently being called
23257413af1fSSteven Rostedt (Red Hat)  */
23267413af1fSSteven Rostedt (Red Hat) unsigned long ftrace_get_addr_curr(struct dyn_ftrace *rec)
23277413af1fSSteven Rostedt (Red Hat) {
232879922b80SSteven Rostedt (Red Hat) 	struct ftrace_ops *ops;
232979922b80SSteven Rostedt (Red Hat) 
233079922b80SSteven Rostedt (Red Hat) 	/* Trampolines take precedence over regs */
233179922b80SSteven Rostedt (Red Hat) 	if (rec->flags & FTRACE_FL_TRAMP_EN) {
233279922b80SSteven Rostedt (Red Hat) 		ops = ftrace_find_tramp_ops_curr(rec);
233379922b80SSteven Rostedt (Red Hat) 		if (FTRACE_WARN_ON(!ops)) {
2334a395d6a7SJoe Perches 			pr_warn("Bad trampoline accounting at: %p (%pS)\n",
233579922b80SSteven Rostedt (Red Hat) 				(void *)rec->ip, (void *)rec->ip);
233679922b80SSteven Rostedt (Red Hat) 			/* Ftrace is shutting down, return anything */
233779922b80SSteven Rostedt (Red Hat) 			return (unsigned long)FTRACE_ADDR;
233879922b80SSteven Rostedt (Red Hat) 		}
233979922b80SSteven Rostedt (Red Hat) 		return ops->trampoline;
234079922b80SSteven Rostedt (Red Hat) 	}
234179922b80SSteven Rostedt (Red Hat) 
23427413af1fSSteven Rostedt (Red Hat) 	if (rec->flags & FTRACE_FL_REGS_EN)
23437413af1fSSteven Rostedt (Red Hat) 		return (unsigned long)FTRACE_REGS_ADDR;
23447413af1fSSteven Rostedt (Red Hat) 	else
23457413af1fSSteven Rostedt (Red Hat) 		return (unsigned long)FTRACE_ADDR;
23467413af1fSSteven Rostedt (Red Hat) }
23477413af1fSSteven Rostedt (Red Hat) 
2348c88fd863SSteven Rostedt static int
2349c88fd863SSteven Rostedt __ftrace_replace_code(struct dyn_ftrace *rec, int enable)
2350c88fd863SSteven Rostedt {
235108f6fba5SSteven Rostedt 	unsigned long ftrace_old_addr;
2352c88fd863SSteven Rostedt 	unsigned long ftrace_addr;
2353c88fd863SSteven Rostedt 	int ret;
2354c88fd863SSteven Rostedt 
23557c0868e0SSteven Rostedt (Red Hat) 	ftrace_addr = ftrace_get_addr_new(rec);
2356c88fd863SSteven Rostedt 
23577c0868e0SSteven Rostedt (Red Hat) 	/* This needs to be done before we call ftrace_update_record */
23587c0868e0SSteven Rostedt (Red Hat) 	ftrace_old_addr = ftrace_get_addr_curr(rec);
23597c0868e0SSteven Rostedt (Red Hat) 
23607c0868e0SSteven Rostedt (Red Hat) 	ret = ftrace_update_record(rec, enable);
236108f6fba5SSteven Rostedt 
236202a392a0SSteven Rostedt (Red Hat) 	ftrace_bug_type = FTRACE_BUG_UNKNOWN;
236302a392a0SSteven Rostedt (Red Hat) 
2364c88fd863SSteven Rostedt 	switch (ret) {
2365c88fd863SSteven Rostedt 	case FTRACE_UPDATE_IGNORE:
2366c88fd863SSteven Rostedt 		return 0;
2367c88fd863SSteven Rostedt 
2368c88fd863SSteven Rostedt 	case FTRACE_UPDATE_MAKE_CALL:
236902a392a0SSteven Rostedt (Red Hat) 		ftrace_bug_type = FTRACE_BUG_CALL;
2370c88fd863SSteven Rostedt 		return ftrace_make_call(rec, ftrace_addr);
2371c88fd863SSteven Rostedt 
2372c88fd863SSteven Rostedt 	case FTRACE_UPDATE_MAKE_NOP:
237302a392a0SSteven Rostedt (Red Hat) 		ftrace_bug_type = FTRACE_BUG_NOP;
237439b5552cSSteven Rostedt (Red Hat) 		return ftrace_make_nop(NULL, rec, ftrace_old_addr);
237508f6fba5SSteven Rostedt 
237608f6fba5SSteven Rostedt 	case FTRACE_UPDATE_MODIFY_CALL:
237702a392a0SSteven Rostedt (Red Hat) 		ftrace_bug_type = FTRACE_BUG_UPDATE;
237808f6fba5SSteven Rostedt 		return ftrace_modify_call(rec, ftrace_old_addr, ftrace_addr);
23793d083395SSteven Rostedt 	}
23803d083395SSteven Rostedt 
2381c88fd863SSteven Rostedt 	return -1; /* unknow ftrace bug */
2382c88fd863SSteven Rostedt }
2383c88fd863SSteven Rostedt 
2384e4f5d544SSteven Rostedt void __weak ftrace_replace_code(int enable)
23853d083395SSteven Rostedt {
23863d083395SSteven Rostedt 	struct dyn_ftrace *rec;
23873d083395SSteven Rostedt 	struct ftrace_page *pg;
23886a24a244SSteven Rostedt 	int failed;
23893d083395SSteven Rostedt 
239045a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
239145a4a237SSteven Rostedt 		return;
239245a4a237SSteven Rostedt 
2393265c831cSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
2394e4f5d544SSteven Rostedt 		failed = __ftrace_replace_code(rec, enable);
2395fa9d13cfSZhaolei 		if (failed) {
23964fd3279bSSteven Rostedt (Red Hat) 			ftrace_bug(failed, rec);
23974377245aSSteven Rostedt 			/* Stop processing */
23984377245aSSteven Rostedt 			return;
23994377245aSSteven Rostedt 		}
2400265c831cSSteven Rostedt 	} while_for_each_ftrace_rec();
24013d083395SSteven Rostedt }
24023c1720f0SSteven Rostedt 
2403c88fd863SSteven Rostedt struct ftrace_rec_iter {
2404c88fd863SSteven Rostedt 	struct ftrace_page	*pg;
2405c88fd863SSteven Rostedt 	int			index;
2406c88fd863SSteven Rostedt };
2407c88fd863SSteven Rostedt 
2408c88fd863SSteven Rostedt /**
2409c88fd863SSteven Rostedt  * ftrace_rec_iter_start, start up iterating over traced functions
2410c88fd863SSteven Rostedt  *
2411c88fd863SSteven Rostedt  * Returns an iterator handle that is used to iterate over all
2412c88fd863SSteven Rostedt  * the records that represent address locations where functions
2413c88fd863SSteven Rostedt  * are traced.
2414c88fd863SSteven Rostedt  *
2415c88fd863SSteven Rostedt  * May return NULL if no records are available.
2416c88fd863SSteven Rostedt  */
2417c88fd863SSteven Rostedt struct ftrace_rec_iter *ftrace_rec_iter_start(void)
2418c88fd863SSteven Rostedt {
2419c88fd863SSteven Rostedt 	/*
2420c88fd863SSteven Rostedt 	 * We only use a single iterator.
2421c88fd863SSteven Rostedt 	 * Protected by the ftrace_lock mutex.
2422c88fd863SSteven Rostedt 	 */
2423c88fd863SSteven Rostedt 	static struct ftrace_rec_iter ftrace_rec_iter;
2424c88fd863SSteven Rostedt 	struct ftrace_rec_iter *iter = &ftrace_rec_iter;
2425c88fd863SSteven Rostedt 
2426c88fd863SSteven Rostedt 	iter->pg = ftrace_pages_start;
2427c88fd863SSteven Rostedt 	iter->index = 0;
2428c88fd863SSteven Rostedt 
2429c88fd863SSteven Rostedt 	/* Could have empty pages */
2430c88fd863SSteven Rostedt 	while (iter->pg && !iter->pg->index)
2431c88fd863SSteven Rostedt 		iter->pg = iter->pg->next;
2432c88fd863SSteven Rostedt 
2433c88fd863SSteven Rostedt 	if (!iter->pg)
2434c88fd863SSteven Rostedt 		return NULL;
2435c88fd863SSteven Rostedt 
2436c88fd863SSteven Rostedt 	return iter;
2437c88fd863SSteven Rostedt }
2438c88fd863SSteven Rostedt 
2439c88fd863SSteven Rostedt /**
2440c88fd863SSteven Rostedt  * ftrace_rec_iter_next, get the next record to process.
2441c88fd863SSteven Rostedt  * @iter: The handle to the iterator.
2442c88fd863SSteven Rostedt  *
2443c88fd863SSteven Rostedt  * Returns the next iterator after the given iterator @iter.
2444c88fd863SSteven Rostedt  */
2445c88fd863SSteven Rostedt struct ftrace_rec_iter *ftrace_rec_iter_next(struct ftrace_rec_iter *iter)
2446c88fd863SSteven Rostedt {
2447c88fd863SSteven Rostedt 	iter->index++;
2448c88fd863SSteven Rostedt 
2449c88fd863SSteven Rostedt 	if (iter->index >= iter->pg->index) {
2450c88fd863SSteven Rostedt 		iter->pg = iter->pg->next;
2451c88fd863SSteven Rostedt 		iter->index = 0;
2452c88fd863SSteven Rostedt 
2453c88fd863SSteven Rostedt 		/* Could have empty pages */
2454c88fd863SSteven Rostedt 		while (iter->pg && !iter->pg->index)
2455c88fd863SSteven Rostedt 			iter->pg = iter->pg->next;
2456c88fd863SSteven Rostedt 	}
2457c88fd863SSteven Rostedt 
2458c88fd863SSteven Rostedt 	if (!iter->pg)
2459c88fd863SSteven Rostedt 		return NULL;
2460c88fd863SSteven Rostedt 
2461c88fd863SSteven Rostedt 	return iter;
2462c88fd863SSteven Rostedt }
2463c88fd863SSteven Rostedt 
2464c88fd863SSteven Rostedt /**
2465c88fd863SSteven Rostedt  * ftrace_rec_iter_record, get the record at the iterator location
2466c88fd863SSteven Rostedt  * @iter: The current iterator location
2467c88fd863SSteven Rostedt  *
2468c88fd863SSteven Rostedt  * Returns the record that the current @iter is at.
2469c88fd863SSteven Rostedt  */
2470c88fd863SSteven Rostedt struct dyn_ftrace *ftrace_rec_iter_record(struct ftrace_rec_iter *iter)
2471c88fd863SSteven Rostedt {
2472c88fd863SSteven Rostedt 	return &iter->pg->records[iter->index];
2473c88fd863SSteven Rostedt }
2474c88fd863SSteven Rostedt 
24753b47bfc1SSteven Rostedt static int
247631e88909SSteven Rostedt ftrace_code_disable(struct module *mod, struct dyn_ftrace *rec)
24773c1720f0SSteven Rostedt {
2478492a7ea5SAbhishek Sagar 	int ret;
247937ad5084SSteven Rostedt 
248045a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
248145a4a237SSteven Rostedt 		return 0;
248245a4a237SSteven Rostedt 
248325aac9dcSShaohua Li 	ret = ftrace_make_nop(mod, rec, MCOUNT_ADDR);
24843d083395SSteven Rostedt 	if (ret) {
248502a392a0SSteven Rostedt (Red Hat) 		ftrace_bug_type = FTRACE_BUG_INIT;
24864fd3279bSSteven Rostedt (Red Hat) 		ftrace_bug(ret, rec);
24873d083395SSteven Rostedt 		return 0;
2488d61f82d0SSteven Rostedt 	}
2489d61f82d0SSteven Rostedt 	return 1;
24903d083395SSteven Rostedt }
24913d083395SSteven Rostedt 
2492000ab691SSteven Rostedt /*
2493000ab691SSteven Rostedt  * archs can override this function if they must do something
2494000ab691SSteven Rostedt  * before the modifying code is performed.
2495000ab691SSteven Rostedt  */
2496000ab691SSteven Rostedt int __weak ftrace_arch_code_modify_prepare(void)
2497000ab691SSteven Rostedt {
2498000ab691SSteven Rostedt 	return 0;
2499000ab691SSteven Rostedt }
2500000ab691SSteven Rostedt 
2501000ab691SSteven Rostedt /*
2502000ab691SSteven Rostedt  * archs can override this function if they must do something
2503000ab691SSteven Rostedt  * after the modifying code is performed.
2504000ab691SSteven Rostedt  */
2505000ab691SSteven Rostedt int __weak ftrace_arch_code_modify_post_process(void)
2506000ab691SSteven Rostedt {
2507000ab691SSteven Rostedt 	return 0;
2508000ab691SSteven Rostedt }
2509000ab691SSteven Rostedt 
25108ed3e2cfSSteven Rostedt void ftrace_modify_all_code(int command)
25118ed3e2cfSSteven Rostedt {
251259338f75SSteven Rostedt (Red Hat) 	int update = command & FTRACE_UPDATE_TRACE_FUNC;
2513cd21067fSPetr Mladek 	int err = 0;
251459338f75SSteven Rostedt (Red Hat) 
251559338f75SSteven Rostedt (Red Hat) 	/*
251659338f75SSteven Rostedt (Red Hat) 	 * If the ftrace_caller calls a ftrace_ops func directly,
251759338f75SSteven Rostedt (Red Hat) 	 * we need to make sure that it only traces functions it
251859338f75SSteven Rostedt (Red Hat) 	 * expects to trace. When doing the switch of functions,
251959338f75SSteven Rostedt (Red Hat) 	 * we need to update to the ftrace_ops_list_func first
252059338f75SSteven Rostedt (Red Hat) 	 * before the transition between old and new calls are set,
252159338f75SSteven Rostedt (Red Hat) 	 * as the ftrace_ops_list_func will check the ops hashes
252259338f75SSteven Rostedt (Red Hat) 	 * to make sure the ops are having the right functions
252359338f75SSteven Rostedt (Red Hat) 	 * traced.
252459338f75SSteven Rostedt (Red Hat) 	 */
2525cd21067fSPetr Mladek 	if (update) {
2526cd21067fSPetr Mladek 		err = ftrace_update_ftrace_func(ftrace_ops_list_func);
2527cd21067fSPetr Mladek 		if (FTRACE_WARN_ON(err))
2528cd21067fSPetr Mladek 			return;
2529cd21067fSPetr Mladek 	}
253059338f75SSteven Rostedt (Red Hat) 
25318ed3e2cfSSteven Rostedt 	if (command & FTRACE_UPDATE_CALLS)
25328ed3e2cfSSteven Rostedt 		ftrace_replace_code(1);
25338ed3e2cfSSteven Rostedt 	else if (command & FTRACE_DISABLE_CALLS)
25348ed3e2cfSSteven Rostedt 		ftrace_replace_code(0);
25358ed3e2cfSSteven Rostedt 
2536405e1d83SSteven Rostedt (Red Hat) 	if (update && ftrace_trace_function != ftrace_ops_list_func) {
2537405e1d83SSteven Rostedt (Red Hat) 		function_trace_op = set_function_trace_op;
2538405e1d83SSteven Rostedt (Red Hat) 		smp_wmb();
2539405e1d83SSteven Rostedt (Red Hat) 		/* If irqs are disabled, we are in stop machine */
2540405e1d83SSteven Rostedt (Red Hat) 		if (!irqs_disabled())
2541405e1d83SSteven Rostedt (Red Hat) 			smp_call_function(ftrace_sync_ipi, NULL, 1);
2542cd21067fSPetr Mladek 		err = ftrace_update_ftrace_func(ftrace_trace_function);
2543cd21067fSPetr Mladek 		if (FTRACE_WARN_ON(err))
2544cd21067fSPetr Mladek 			return;
2545405e1d83SSteven Rostedt (Red Hat) 	}
25468ed3e2cfSSteven Rostedt 
25478ed3e2cfSSteven Rostedt 	if (command & FTRACE_START_FUNC_RET)
2548cd21067fSPetr Mladek 		err = ftrace_enable_ftrace_graph_caller();
25498ed3e2cfSSteven Rostedt 	else if (command & FTRACE_STOP_FUNC_RET)
2550cd21067fSPetr Mladek 		err = ftrace_disable_ftrace_graph_caller();
2551cd21067fSPetr Mladek 	FTRACE_WARN_ON(err);
25528ed3e2cfSSteven Rostedt }
25538ed3e2cfSSteven Rostedt 
25543d083395SSteven Rostedt static int __ftrace_modify_code(void *data)
25553d083395SSteven Rostedt {
25563d083395SSteven Rostedt 	int *command = data;
25573d083395SSteven Rostedt 
25588ed3e2cfSSteven Rostedt 	ftrace_modify_all_code(*command);
25595a45cfe1SSteven Rostedt 
2560c88fd863SSteven Rostedt 	return 0;
2561c88fd863SSteven Rostedt }
2562c88fd863SSteven Rostedt 
2563c88fd863SSteven Rostedt /**
2564c88fd863SSteven Rostedt  * ftrace_run_stop_machine, go back to the stop machine method
2565c88fd863SSteven Rostedt  * @command: The command to tell ftrace what to do
2566c88fd863SSteven Rostedt  *
2567c88fd863SSteven Rostedt  * If an arch needs to fall back to the stop machine method, the
2568c88fd863SSteven Rostedt  * it can call this function.
2569c88fd863SSteven Rostedt  */
2570c88fd863SSteven Rostedt void ftrace_run_stop_machine(int command)
2571c88fd863SSteven Rostedt {
2572c88fd863SSteven Rostedt 	stop_machine(__ftrace_modify_code, &command, NULL);
2573c88fd863SSteven Rostedt }
2574c88fd863SSteven Rostedt 
2575c88fd863SSteven Rostedt /**
2576c88fd863SSteven Rostedt  * arch_ftrace_update_code, modify the code to trace or not trace
2577c88fd863SSteven Rostedt  * @command: The command that needs to be done
2578c88fd863SSteven Rostedt  *
2579c88fd863SSteven Rostedt  * Archs can override this function if it does not need to
2580c88fd863SSteven Rostedt  * run stop_machine() to modify code.
2581c88fd863SSteven Rostedt  */
2582c88fd863SSteven Rostedt void __weak arch_ftrace_update_code(int command)
2583c88fd863SSteven Rostedt {
2584c88fd863SSteven Rostedt 	ftrace_run_stop_machine(command);
2585c88fd863SSteven Rostedt }
2586c88fd863SSteven Rostedt 
2587c88fd863SSteven Rostedt static void ftrace_run_update_code(int command)
2588c88fd863SSteven Rostedt {
2589c88fd863SSteven Rostedt 	int ret;
2590c88fd863SSteven Rostedt 
2591c88fd863SSteven Rostedt 	ret = ftrace_arch_code_modify_prepare();
2592c88fd863SSteven Rostedt 	FTRACE_WARN_ON(ret);
2593c88fd863SSteven Rostedt 	if (ret)
2594c88fd863SSteven Rostedt 		return;
2595c88fd863SSteven Rostedt 
2596c88fd863SSteven Rostedt 	/*
2597c88fd863SSteven Rostedt 	 * By default we use stop_machine() to modify the code.
2598c88fd863SSteven Rostedt 	 * But archs can do what ever they want as long as it
2599c88fd863SSteven Rostedt 	 * is safe. The stop_machine() is the safest, but also
2600c88fd863SSteven Rostedt 	 * produces the most overhead.
2601c88fd863SSteven Rostedt 	 */
2602c88fd863SSteven Rostedt 	arch_ftrace_update_code(command);
2603c88fd863SSteven Rostedt 
2604000ab691SSteven Rostedt 	ret = ftrace_arch_code_modify_post_process();
2605000ab691SSteven Rostedt 	FTRACE_WARN_ON(ret);
26063d083395SSteven Rostedt }
26073d083395SSteven Rostedt 
26088252ecf3SSteven Rostedt (Red Hat) static void ftrace_run_modify_code(struct ftrace_ops *ops, int command,
26097485058eSSteven Rostedt (Red Hat) 				   struct ftrace_ops_hash *old_hash)
2610e1effa01SSteven Rostedt (Red Hat) {
2611e1effa01SSteven Rostedt (Red Hat) 	ops->flags |= FTRACE_OPS_FL_MODIFYING;
26127485058eSSteven Rostedt (Red Hat) 	ops->old_hash.filter_hash = old_hash->filter_hash;
26137485058eSSteven Rostedt (Red Hat) 	ops->old_hash.notrace_hash = old_hash->notrace_hash;
2614e1effa01SSteven Rostedt (Red Hat) 	ftrace_run_update_code(command);
26158252ecf3SSteven Rostedt (Red Hat) 	ops->old_hash.filter_hash = NULL;
26167485058eSSteven Rostedt (Red Hat) 	ops->old_hash.notrace_hash = NULL;
2617e1effa01SSteven Rostedt (Red Hat) 	ops->flags &= ~FTRACE_OPS_FL_MODIFYING;
2618e1effa01SSteven Rostedt (Red Hat) }
2619e1effa01SSteven Rostedt (Red Hat) 
26203d083395SSteven Rostedt static ftrace_func_t saved_ftrace_func;
262160a7ecf4SSteven Rostedt static int ftrace_start_up;
2622df4fc315SSteven Rostedt 
262312cce594SSteven Rostedt (Red Hat) void __weak arch_ftrace_trampoline_free(struct ftrace_ops *ops)
262412cce594SSteven Rostedt (Red Hat) {
262512cce594SSteven Rostedt (Red Hat) }
262612cce594SSteven Rostedt (Red Hat) 
2627ba27f2bcSSteven Rostedt (Red Hat) static void per_cpu_ops_free(struct ftrace_ops *ops)
2628db0fbadcSJiri Slaby {
2629db0fbadcSJiri Slaby 	free_percpu(ops->disabled);
2630db0fbadcSJiri Slaby }
2631db0fbadcSJiri Slaby 
2632df4fc315SSteven Rostedt static void ftrace_startup_enable(int command)
2633df4fc315SSteven Rostedt {
2634df4fc315SSteven Rostedt 	if (saved_ftrace_func != ftrace_trace_function) {
2635df4fc315SSteven Rostedt 		saved_ftrace_func = ftrace_trace_function;
2636df4fc315SSteven Rostedt 		command |= FTRACE_UPDATE_TRACE_FUNC;
2637df4fc315SSteven Rostedt 	}
2638df4fc315SSteven Rostedt 
2639df4fc315SSteven Rostedt 	if (!command || !ftrace_enabled)
2640df4fc315SSteven Rostedt 		return;
2641df4fc315SSteven Rostedt 
2642df4fc315SSteven Rostedt 	ftrace_run_update_code(command);
2643df4fc315SSteven Rostedt }
26443d083395SSteven Rostedt 
2645e1effa01SSteven Rostedt (Red Hat) static void ftrace_startup_all(int command)
2646e1effa01SSteven Rostedt (Red Hat) {
2647e1effa01SSteven Rostedt (Red Hat) 	update_all_ops = true;
2648e1effa01SSteven Rostedt (Red Hat) 	ftrace_startup_enable(command);
2649e1effa01SSteven Rostedt (Red Hat) 	update_all_ops = false;
2650e1effa01SSteven Rostedt (Red Hat) }
2651e1effa01SSteven Rostedt (Red Hat) 
2652a1cd6173SSteven Rostedt static int ftrace_startup(struct ftrace_ops *ops, int command)
26533d083395SSteven Rostedt {
26548a56d776SSteven Rostedt (Red Hat) 	int ret;
2655b848914cSSteven Rostedt 
26564eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
2657a1cd6173SSteven Rostedt 		return -ENODEV;
26584eebcc81SSteven Rostedt 
26598a56d776SSteven Rostedt (Red Hat) 	ret = __register_ftrace_function(ops);
26608a56d776SSteven Rostedt (Red Hat) 	if (ret)
26618a56d776SSteven Rostedt (Red Hat) 		return ret;
26628a56d776SSteven Rostedt (Red Hat) 
266360a7ecf4SSteven Rostedt 	ftrace_start_up++;
26643d083395SSteven Rostedt 
2665e1effa01SSteven Rostedt (Red Hat) 	/*
2666e1effa01SSteven Rostedt (Red Hat) 	 * Note that ftrace probes uses this to start up
2667e1effa01SSteven Rostedt (Red Hat) 	 * and modify functions it will probe. But we still
2668e1effa01SSteven Rostedt (Red Hat) 	 * set the ADDING flag for modification, as probes
2669e1effa01SSteven Rostedt (Red Hat) 	 * do not have trampolines. If they add them in the
2670e1effa01SSteven Rostedt (Red Hat) 	 * future, then the probes will need to distinguish
2671e1effa01SSteven Rostedt (Red Hat) 	 * between adding and updating probes.
2672e1effa01SSteven Rostedt (Red Hat) 	 */
2673e1effa01SSteven Rostedt (Red Hat) 	ops->flags |= FTRACE_OPS_FL_ENABLED | FTRACE_OPS_FL_ADDING;
267466209a5bSSteven Rostedt (Red Hat) 
2675f8b8be8aSMasami Hiramatsu 	ret = ftrace_hash_ipmodify_enable(ops);
2676f8b8be8aSMasami Hiramatsu 	if (ret < 0) {
2677f8b8be8aSMasami Hiramatsu 		/* Rollback registration process */
2678f8b8be8aSMasami Hiramatsu 		__unregister_ftrace_function(ops);
2679f8b8be8aSMasami Hiramatsu 		ftrace_start_up--;
2680f8b8be8aSMasami Hiramatsu 		ops->flags &= ~FTRACE_OPS_FL_ENABLED;
2681f8b8be8aSMasami Hiramatsu 		return ret;
2682f8b8be8aSMasami Hiramatsu 	}
2683f8b8be8aSMasami Hiramatsu 
26847f50d06bSJiri Olsa 	if (ftrace_hash_rec_enable(ops, 1))
26857f50d06bSJiri Olsa 		command |= FTRACE_UPDATE_CALLS;
2686ed926f9bSSteven Rostedt 
2687df4fc315SSteven Rostedt 	ftrace_startup_enable(command);
2688a1cd6173SSteven Rostedt 
2689e1effa01SSteven Rostedt (Red Hat) 	ops->flags &= ~FTRACE_OPS_FL_ADDING;
2690e1effa01SSteven Rostedt (Red Hat) 
2691a1cd6173SSteven Rostedt 	return 0;
26923d083395SSteven Rostedt }
26933d083395SSteven Rostedt 
26948a56d776SSteven Rostedt (Red Hat) static int ftrace_shutdown(struct ftrace_ops *ops, int command)
26953d083395SSteven Rostedt {
26968a56d776SSteven Rostedt (Red Hat) 	int ret;
2697b848914cSSteven Rostedt 
26984eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
26998a56d776SSteven Rostedt (Red Hat) 		return -ENODEV;
27008a56d776SSteven Rostedt (Red Hat) 
27018a56d776SSteven Rostedt (Red Hat) 	ret = __unregister_ftrace_function(ops);
27028a56d776SSteven Rostedt (Red Hat) 	if (ret)
27038a56d776SSteven Rostedt (Red Hat) 		return ret;
27044eebcc81SSteven Rostedt 
270560a7ecf4SSteven Rostedt 	ftrace_start_up--;
27069ea1a153SFrederic Weisbecker 	/*
27079ea1a153SFrederic Weisbecker 	 * Just warn in case of unbalance, no need to kill ftrace, it's not
27089ea1a153SFrederic Weisbecker 	 * critical but the ftrace_call callers may be never nopped again after
27099ea1a153SFrederic Weisbecker 	 * further ftrace uses.
27109ea1a153SFrederic Weisbecker 	 */
27119ea1a153SFrederic Weisbecker 	WARN_ON_ONCE(ftrace_start_up < 0);
27129ea1a153SFrederic Weisbecker 
2713f8b8be8aSMasami Hiramatsu 	/* Disabling ipmodify never fails */
2714f8b8be8aSMasami Hiramatsu 	ftrace_hash_ipmodify_disable(ops);
27157f50d06bSJiri Olsa 
27167f50d06bSJiri Olsa 	if (ftrace_hash_rec_disable(ops, 1))
27177f50d06bSJiri Olsa 		command |= FTRACE_UPDATE_CALLS;
2718ed926f9bSSteven Rostedt 
2719ed926f9bSSteven Rostedt 	ops->flags &= ~FTRACE_OPS_FL_ENABLED;
2720b848914cSSteven Rostedt 
2721d61f82d0SSteven Rostedt 	if (saved_ftrace_func != ftrace_trace_function) {
2722d61f82d0SSteven Rostedt 		saved_ftrace_func = ftrace_trace_function;
2723d61f82d0SSteven Rostedt 		command |= FTRACE_UPDATE_TRACE_FUNC;
2724d61f82d0SSteven Rostedt 	}
2725d61f82d0SSteven Rostedt 
2726a4c35ed2SSteven Rostedt (Red Hat) 	if (!command || !ftrace_enabled) {
2727a4c35ed2SSteven Rostedt (Red Hat) 		/*
2728ba27f2bcSSteven Rostedt (Red Hat) 		 * If these are per_cpu ops, they still need their
2729a4c35ed2SSteven Rostedt (Red Hat) 		 * per_cpu field freed. Since, function tracing is
2730a4c35ed2SSteven Rostedt (Red Hat) 		 * not currently active, we can just free them
2731a4c35ed2SSteven Rostedt (Red Hat) 		 * without synchronizing all CPUs.
2732a4c35ed2SSteven Rostedt (Red Hat) 		 */
2733ba27f2bcSSteven Rostedt (Red Hat) 		if (ops->flags & FTRACE_OPS_FL_PER_CPU)
2734ba27f2bcSSteven Rostedt (Red Hat) 			per_cpu_ops_free(ops);
27358a56d776SSteven Rostedt (Red Hat) 		return 0;
2736a4c35ed2SSteven Rostedt (Red Hat) 	}
27373d083395SSteven Rostedt 
273879922b80SSteven Rostedt (Red Hat) 	/*
273979922b80SSteven Rostedt (Red Hat) 	 * If the ops uses a trampoline, then it needs to be
274079922b80SSteven Rostedt (Red Hat) 	 * tested first on update.
274179922b80SSteven Rostedt (Red Hat) 	 */
2742e1effa01SSteven Rostedt (Red Hat) 	ops->flags |= FTRACE_OPS_FL_REMOVING;
274379922b80SSteven Rostedt (Red Hat) 	removed_ops = ops;
274479922b80SSteven Rostedt (Red Hat) 
2745fef5aeeeSSteven Rostedt (Red Hat) 	/* The trampoline logic checks the old hashes */
2746fef5aeeeSSteven Rostedt (Red Hat) 	ops->old_hash.filter_hash = ops->func_hash->filter_hash;
2747fef5aeeeSSteven Rostedt (Red Hat) 	ops->old_hash.notrace_hash = ops->func_hash->notrace_hash;
2748fef5aeeeSSteven Rostedt (Red Hat) 
2749d61f82d0SSteven Rostedt 	ftrace_run_update_code(command);
2750a4c35ed2SSteven Rostedt (Red Hat) 
275184bde62cSSteven Rostedt (Red Hat) 	/*
275284bde62cSSteven Rostedt (Red Hat) 	 * If there's no more ops registered with ftrace, run a
275384bde62cSSteven Rostedt (Red Hat) 	 * sanity check to make sure all rec flags are cleared.
275484bde62cSSteven Rostedt (Red Hat) 	 */
275584bde62cSSteven Rostedt (Red Hat) 	if (ftrace_ops_list == &ftrace_list_end) {
275684bde62cSSteven Rostedt (Red Hat) 		struct ftrace_page *pg;
275784bde62cSSteven Rostedt (Red Hat) 		struct dyn_ftrace *rec;
275884bde62cSSteven Rostedt (Red Hat) 
275984bde62cSSteven Rostedt (Red Hat) 		do_for_each_ftrace_rec(pg, rec) {
276084bde62cSSteven Rostedt (Red Hat) 			if (FTRACE_WARN_ON_ONCE(rec->flags))
276184bde62cSSteven Rostedt (Red Hat) 				pr_warn("  %pS flags:%lx\n",
276284bde62cSSteven Rostedt (Red Hat) 					(void *)rec->ip, rec->flags);
276384bde62cSSteven Rostedt (Red Hat) 		} while_for_each_ftrace_rec();
276484bde62cSSteven Rostedt (Red Hat) 	}
276584bde62cSSteven Rostedt (Red Hat) 
2766fef5aeeeSSteven Rostedt (Red Hat) 	ops->old_hash.filter_hash = NULL;
2767fef5aeeeSSteven Rostedt (Red Hat) 	ops->old_hash.notrace_hash = NULL;
2768fef5aeeeSSteven Rostedt (Red Hat) 
2769fef5aeeeSSteven Rostedt (Red Hat) 	removed_ops = NULL;
2770e1effa01SSteven Rostedt (Red Hat) 	ops->flags &= ~FTRACE_OPS_FL_REMOVING;
277179922b80SSteven Rostedt (Red Hat) 
2772a4c35ed2SSteven Rostedt (Red Hat) 	/*
2773a4c35ed2SSteven Rostedt (Red Hat) 	 * Dynamic ops may be freed, we must make sure that all
2774a4c35ed2SSteven Rostedt (Red Hat) 	 * callers are done before leaving this function.
2775ba27f2bcSSteven Rostedt (Red Hat) 	 * The same goes for freeing the per_cpu data of the per_cpu
2776a4c35ed2SSteven Rostedt (Red Hat) 	 * ops.
2777a4c35ed2SSteven Rostedt (Red Hat) 	 *
2778a4c35ed2SSteven Rostedt (Red Hat) 	 * Again, normal synchronize_sched() is not good enough.
2779a4c35ed2SSteven Rostedt (Red Hat) 	 * We need to do a hard force of sched synchronization.
2780a4c35ed2SSteven Rostedt (Red Hat) 	 * This is because we use preempt_disable() to do RCU, but
2781a4c35ed2SSteven Rostedt (Red Hat) 	 * the function tracers can be called where RCU is not watching
2782a4c35ed2SSteven Rostedt (Red Hat) 	 * (like before user_exit()). We can not rely on the RCU
2783a4c35ed2SSteven Rostedt (Red Hat) 	 * infrastructure to do the synchronization, thus we must do it
2784a4c35ed2SSteven Rostedt (Red Hat) 	 * ourselves.
2785a4c35ed2SSteven Rostedt (Red Hat) 	 */
2786ba27f2bcSSteven Rostedt (Red Hat) 	if (ops->flags & (FTRACE_OPS_FL_DYNAMIC | FTRACE_OPS_FL_PER_CPU)) {
2787a4c35ed2SSteven Rostedt (Red Hat) 		schedule_on_each_cpu(ftrace_sync);
2788a4c35ed2SSteven Rostedt (Red Hat) 
278912cce594SSteven Rostedt (Red Hat) 		arch_ftrace_trampoline_free(ops);
279012cce594SSteven Rostedt (Red Hat) 
2791ba27f2bcSSteven Rostedt (Red Hat) 		if (ops->flags & FTRACE_OPS_FL_PER_CPU)
2792ba27f2bcSSteven Rostedt (Red Hat) 			per_cpu_ops_free(ops);
2793a4c35ed2SSteven Rostedt (Red Hat) 	}
2794a4c35ed2SSteven Rostedt (Red Hat) 
27958a56d776SSteven Rostedt (Red Hat) 	return 0;
27963d083395SSteven Rostedt }
27973d083395SSteven Rostedt 
2798e309b41dSIngo Molnar static void ftrace_startup_sysctl(void)
2799b0fc494fSSteven Rostedt {
28001619dc3fSPratyush Anand 	int command;
28011619dc3fSPratyush Anand 
28024eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
28034eebcc81SSteven Rostedt 		return;
28044eebcc81SSteven Rostedt 
2805d61f82d0SSteven Rostedt 	/* Force update next time */
2806d61f82d0SSteven Rostedt 	saved_ftrace_func = NULL;
280760a7ecf4SSteven Rostedt 	/* ftrace_start_up is true if we want ftrace running */
28081619dc3fSPratyush Anand 	if (ftrace_start_up) {
28091619dc3fSPratyush Anand 		command = FTRACE_UPDATE_CALLS;
28101619dc3fSPratyush Anand 		if (ftrace_graph_active)
28111619dc3fSPratyush Anand 			command |= FTRACE_START_FUNC_RET;
2812524a3868SSteven Rostedt (Red Hat) 		ftrace_startup_enable(command);
28131619dc3fSPratyush Anand 	}
2814b0fc494fSSteven Rostedt }
2815b0fc494fSSteven Rostedt 
2816e309b41dSIngo Molnar static void ftrace_shutdown_sysctl(void)
2817b0fc494fSSteven Rostedt {
28181619dc3fSPratyush Anand 	int command;
28191619dc3fSPratyush Anand 
28204eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
28214eebcc81SSteven Rostedt 		return;
28224eebcc81SSteven Rostedt 
282360a7ecf4SSteven Rostedt 	/* ftrace_start_up is true if ftrace is running */
28241619dc3fSPratyush Anand 	if (ftrace_start_up) {
28251619dc3fSPratyush Anand 		command = FTRACE_DISABLE_CALLS;
28261619dc3fSPratyush Anand 		if (ftrace_graph_active)
28271619dc3fSPratyush Anand 			command |= FTRACE_STOP_FUNC_RET;
28281619dc3fSPratyush Anand 		ftrace_run_update_code(command);
28291619dc3fSPratyush Anand 	}
2830b0fc494fSSteven Rostedt }
2831b0fc494fSSteven Rostedt 
28323d083395SSteven Rostedt static cycle_t		ftrace_update_time;
28333d083395SSteven Rostedt unsigned long		ftrace_update_tot_cnt;
28343d083395SSteven Rostedt 
28358c4f3c3fSSteven Rostedt (Red Hat) static inline int ops_traces_mod(struct ftrace_ops *ops)
2836f7bc8b61SSteven Rostedt {
28378c4f3c3fSSteven Rostedt (Red Hat) 	/*
28388c4f3c3fSSteven Rostedt (Red Hat) 	 * Filter_hash being empty will default to trace module.
28398c4f3c3fSSteven Rostedt (Red Hat) 	 * But notrace hash requires a test of individual module functions.
28408c4f3c3fSSteven Rostedt (Red Hat) 	 */
284133b7f99cSSteven Rostedt (Red Hat) 	return ftrace_hash_empty(ops->func_hash->filter_hash) &&
284233b7f99cSSteven Rostedt (Red Hat) 		ftrace_hash_empty(ops->func_hash->notrace_hash);
28438c4f3c3fSSteven Rostedt (Red Hat) }
2844f7bc8b61SSteven Rostedt 
28458c4f3c3fSSteven Rostedt (Red Hat) /*
28468c4f3c3fSSteven Rostedt (Red Hat)  * Check if the current ops references the record.
28478c4f3c3fSSteven Rostedt (Red Hat)  *
28488c4f3c3fSSteven Rostedt (Red Hat)  * If the ops traces all functions, then it was already accounted for.
28498c4f3c3fSSteven Rostedt (Red Hat)  * If the ops does not trace the current record function, skip it.
28508c4f3c3fSSteven Rostedt (Red Hat)  * If the ops ignores the function via notrace filter, skip it.
28518c4f3c3fSSteven Rostedt (Red Hat)  */
28528c4f3c3fSSteven Rostedt (Red Hat) static inline bool
28538c4f3c3fSSteven Rostedt (Red Hat) ops_references_rec(struct ftrace_ops *ops, struct dyn_ftrace *rec)
28548c4f3c3fSSteven Rostedt (Red Hat) {
28558c4f3c3fSSteven Rostedt (Red Hat) 	/* If ops isn't enabled, ignore it */
28568c4f3c3fSSteven Rostedt (Red Hat) 	if (!(ops->flags & FTRACE_OPS_FL_ENABLED))
28578c4f3c3fSSteven Rostedt (Red Hat) 		return 0;
28588c4f3c3fSSteven Rostedt (Red Hat) 
2859b7ffffbbSSteven Rostedt (Red Hat) 	/* If ops traces all then it includes this function */
28608c4f3c3fSSteven Rostedt (Red Hat) 	if (ops_traces_mod(ops))
2861b7ffffbbSSteven Rostedt (Red Hat) 		return 1;
28628c4f3c3fSSteven Rostedt (Red Hat) 
28638c4f3c3fSSteven Rostedt (Red Hat) 	/* The function must be in the filter */
286433b7f99cSSteven Rostedt (Red Hat) 	if (!ftrace_hash_empty(ops->func_hash->filter_hash) &&
286533b7f99cSSteven Rostedt (Red Hat) 	    !ftrace_lookup_ip(ops->func_hash->filter_hash, rec->ip))
28668c4f3c3fSSteven Rostedt (Red Hat) 		return 0;
28678c4f3c3fSSteven Rostedt (Red Hat) 
28688c4f3c3fSSteven Rostedt (Red Hat) 	/* If in notrace hash, we ignore it too */
286933b7f99cSSteven Rostedt (Red Hat) 	if (ftrace_lookup_ip(ops->func_hash->notrace_hash, rec->ip))
28708c4f3c3fSSteven Rostedt (Red Hat) 		return 0;
28718c4f3c3fSSteven Rostedt (Red Hat) 
28728c4f3c3fSSteven Rostedt (Red Hat) 	return 1;
28738c4f3c3fSSteven Rostedt (Red Hat) }
28748c4f3c3fSSteven Rostedt (Red Hat) 
28751dc43cf0SJiri Slaby static int ftrace_update_code(struct module *mod, struct ftrace_page *new_pgs)
28763d083395SSteven Rostedt {
287785ae32aeSSteven Rostedt 	struct ftrace_page *pg;
2878e94142a6SLai Jiangshan 	struct dyn_ftrace *p;
2879f22f9a89SAbhishek Sagar 	cycle_t start, stop;
28801dc43cf0SJiri Slaby 	unsigned long update_cnt = 0;
2881b7ffffbbSSteven Rostedt (Red Hat) 	unsigned long rec_flags = 0;
288285ae32aeSSteven Rostedt 	int i;
2883f7bc8b61SSteven Rostedt 
2884750ed1a4SIngo Molnar 	start = ftrace_now(raw_smp_processor_id());
28853d083395SSteven Rostedt 
2886b7ffffbbSSteven Rostedt (Red Hat) 	/*
2887b7ffffbbSSteven Rostedt (Red Hat) 	 * When a module is loaded, this function is called to convert
2888b7ffffbbSSteven Rostedt (Red Hat) 	 * the calls to mcount in its text to nops, and also to create
2889b7ffffbbSSteven Rostedt (Red Hat) 	 * an entry in the ftrace data. Now, if ftrace is activated
2890b7ffffbbSSteven Rostedt (Red Hat) 	 * after this call, but before the module sets its text to
2891b7ffffbbSSteven Rostedt (Red Hat) 	 * read-only, the modification of enabling ftrace can fail if
2892b7ffffbbSSteven Rostedt (Red Hat) 	 * the read-only is done while ftrace is converting the calls.
2893b7ffffbbSSteven Rostedt (Red Hat) 	 * To prevent this, the module's records are set as disabled
2894b7ffffbbSSteven Rostedt (Red Hat) 	 * and will be enabled after the call to set the module's text
2895b7ffffbbSSteven Rostedt (Red Hat) 	 * to read-only.
2896b7ffffbbSSteven Rostedt (Red Hat) 	 */
2897b7ffffbbSSteven Rostedt (Red Hat) 	if (mod)
2898b7ffffbbSSteven Rostedt (Red Hat) 		rec_flags |= FTRACE_FL_DISABLED;
2899b7ffffbbSSteven Rostedt (Red Hat) 
29001dc43cf0SJiri Slaby 	for (pg = new_pgs; pg; pg = pg->next) {
2901f22f9a89SAbhishek Sagar 
290285ae32aeSSteven Rostedt 		for (i = 0; i < pg->index; i++) {
29038c4f3c3fSSteven Rostedt (Red Hat) 
290408f5ac90SSteven Rostedt 			/* If something went wrong, bail without enabling anything */
290508f5ac90SSteven Rostedt 			if (unlikely(ftrace_disabled))
290608f5ac90SSteven Rostedt 				return -1;
29073d083395SSteven Rostedt 
290885ae32aeSSteven Rostedt 			p = &pg->records[i];
2909b7ffffbbSSteven Rostedt (Red Hat) 			p->flags = rec_flags;
2910f22f9a89SAbhishek Sagar 
29115cb084bbSJiri Olsa 			/*
291225985edcSLucas De Marchi 			 * Do the initial record conversion from mcount jump
29135cb084bbSJiri Olsa 			 * to the NOP instructions.
29145cb084bbSJiri Olsa 			 */
291532082309SSteven Rostedt 			if (!ftrace_code_disable(mod, p))
2916d2c8c3eaSSteven Rostedt 				break;
29175cb084bbSJiri Olsa 
29181dc43cf0SJiri Slaby 			update_cnt++;
29195cb084bbSJiri Olsa 		}
292085ae32aeSSteven Rostedt 	}
292185ae32aeSSteven Rostedt 
2922750ed1a4SIngo Molnar 	stop = ftrace_now(raw_smp_processor_id());
29233d083395SSteven Rostedt 	ftrace_update_time = stop - start;
29241dc43cf0SJiri Slaby 	ftrace_update_tot_cnt += update_cnt;
29253d083395SSteven Rostedt 
292616444a8aSArnaldo Carvalho de Melo 	return 0;
292716444a8aSArnaldo Carvalho de Melo }
292816444a8aSArnaldo Carvalho de Melo 
2929a7900875SSteven Rostedt static int ftrace_allocate_records(struct ftrace_page *pg, int count)
29303c1720f0SSteven Rostedt {
2931a7900875SSteven Rostedt 	int order;
29323c1720f0SSteven Rostedt 	int cnt;
29333c1720f0SSteven Rostedt 
2934a7900875SSteven Rostedt 	if (WARN_ON(!count))
2935a7900875SSteven Rostedt 		return -EINVAL;
2936a7900875SSteven Rostedt 
2937a7900875SSteven Rostedt 	order = get_count_order(DIV_ROUND_UP(count, ENTRIES_PER_PAGE));
29383c1720f0SSteven Rostedt 
29393c1720f0SSteven Rostedt 	/*
2940a7900875SSteven Rostedt 	 * We want to fill as much as possible. No more than a page
2941a7900875SSteven Rostedt 	 * may be empty.
29423c1720f0SSteven Rostedt 	 */
2943a7900875SSteven Rostedt 	while ((PAGE_SIZE << order) / ENTRY_SIZE >= count + ENTRIES_PER_PAGE)
2944a7900875SSteven Rostedt 		order--;
29453c1720f0SSteven Rostedt 
2946a7900875SSteven Rostedt  again:
2947a7900875SSteven Rostedt 	pg->records = (void *)__get_free_pages(GFP_KERNEL | __GFP_ZERO, order);
2948a7900875SSteven Rostedt 
2949a7900875SSteven Rostedt 	if (!pg->records) {
2950a7900875SSteven Rostedt 		/* if we can't allocate this size, try something smaller */
2951a7900875SSteven Rostedt 		if (!order)
2952a7900875SSteven Rostedt 			return -ENOMEM;
2953a7900875SSteven Rostedt 		order >>= 1;
2954a7900875SSteven Rostedt 		goto again;
2955a7900875SSteven Rostedt 	}
2956a7900875SSteven Rostedt 
2957a7900875SSteven Rostedt 	cnt = (PAGE_SIZE << order) / ENTRY_SIZE;
2958a7900875SSteven Rostedt 	pg->size = cnt;
2959a7900875SSteven Rostedt 
2960a7900875SSteven Rostedt 	if (cnt > count)
2961a7900875SSteven Rostedt 		cnt = count;
2962a7900875SSteven Rostedt 
2963a7900875SSteven Rostedt 	return cnt;
2964a7900875SSteven Rostedt }
2965a7900875SSteven Rostedt 
2966a7900875SSteven Rostedt static struct ftrace_page *
2967a7900875SSteven Rostedt ftrace_allocate_pages(unsigned long num_to_init)
2968a7900875SSteven Rostedt {
2969a7900875SSteven Rostedt 	struct ftrace_page *start_pg;
2970a7900875SSteven Rostedt 	struct ftrace_page *pg;
2971a7900875SSteven Rostedt 	int order;
2972a7900875SSteven Rostedt 	int cnt;
2973a7900875SSteven Rostedt 
2974a7900875SSteven Rostedt 	if (!num_to_init)
2975a7900875SSteven Rostedt 		return 0;
2976a7900875SSteven Rostedt 
2977a7900875SSteven Rostedt 	start_pg = pg = kzalloc(sizeof(*pg), GFP_KERNEL);
2978a7900875SSteven Rostedt 	if (!pg)
2979a7900875SSteven Rostedt 		return NULL;
2980a7900875SSteven Rostedt 
2981a7900875SSteven Rostedt 	/*
2982a7900875SSteven Rostedt 	 * Try to allocate as much as possible in one continues
2983a7900875SSteven Rostedt 	 * location that fills in all of the space. We want to
2984a7900875SSteven Rostedt 	 * waste as little space as possible.
2985a7900875SSteven Rostedt 	 */
2986a7900875SSteven Rostedt 	for (;;) {
2987a7900875SSteven Rostedt 		cnt = ftrace_allocate_records(pg, num_to_init);
2988a7900875SSteven Rostedt 		if (cnt < 0)
2989a7900875SSteven Rostedt 			goto free_pages;
2990a7900875SSteven Rostedt 
2991a7900875SSteven Rostedt 		num_to_init -= cnt;
2992a7900875SSteven Rostedt 		if (!num_to_init)
2993a7900875SSteven Rostedt 			break;
2994a7900875SSteven Rostedt 
2995a7900875SSteven Rostedt 		pg->next = kzalloc(sizeof(*pg), GFP_KERNEL);
2996a7900875SSteven Rostedt 		if (!pg->next)
2997a7900875SSteven Rostedt 			goto free_pages;
2998a7900875SSteven Rostedt 
2999a7900875SSteven Rostedt 		pg = pg->next;
3000a7900875SSteven Rostedt 	}
3001a7900875SSteven Rostedt 
3002a7900875SSteven Rostedt 	return start_pg;
3003a7900875SSteven Rostedt 
3004a7900875SSteven Rostedt  free_pages:
30051f61be00SNamhyung Kim 	pg = start_pg;
30061f61be00SNamhyung Kim 	while (pg) {
3007a7900875SSteven Rostedt 		order = get_count_order(pg->size / ENTRIES_PER_PAGE);
3008a7900875SSteven Rostedt 		free_pages((unsigned long)pg->records, order);
3009a7900875SSteven Rostedt 		start_pg = pg->next;
3010a7900875SSteven Rostedt 		kfree(pg);
3011a7900875SSteven Rostedt 		pg = start_pg;
3012a7900875SSteven Rostedt 	}
3013a7900875SSteven Rostedt 	pr_info("ftrace: FAILED to allocate memory for functions\n");
3014a7900875SSteven Rostedt 	return NULL;
3015a7900875SSteven Rostedt }
3016a7900875SSteven Rostedt 
30175072c59fSSteven Rostedt #define FTRACE_BUFF_MAX (KSYM_SYMBOL_LEN+4) /* room for wildcards */
30185072c59fSSteven Rostedt 
30195072c59fSSteven Rostedt struct ftrace_iterator {
302098c4fd04SSteven Rostedt 	loff_t				pos;
30212bccfffdSSteven Rostedt 	loff_t				func_pos;
30225072c59fSSteven Rostedt 	struct ftrace_page		*pg;
30234aeb6967SSteven Rostedt 	struct dyn_ftrace		*func;
30244aeb6967SSteven Rostedt 	struct ftrace_func_probe	*probe;
30254aeb6967SSteven Rostedt 	struct trace_parser		parser;
30261cf41dd7SSteven Rostedt 	struct ftrace_hash		*hash;
302733dc9b12SSteven Rostedt 	struct ftrace_ops		*ops;
30288fc0c701SSteven Rostedt 	int				hidx;
3029431aa3fbSSteven Rostedt 	int				idx;
30305072c59fSSteven Rostedt 	unsigned			flags;
30315072c59fSSteven Rostedt };
30325072c59fSSteven Rostedt 
3033e309b41dSIngo Molnar static void *
30344aeb6967SSteven Rostedt t_hash_next(struct seq_file *m, loff_t *pos)
30358fc0c701SSteven Rostedt {
30368fc0c701SSteven Rostedt 	struct ftrace_iterator *iter = m->private;
30374aeb6967SSteven Rostedt 	struct hlist_node *hnd = NULL;
30388fc0c701SSteven Rostedt 	struct hlist_head *hhd;
30398fc0c701SSteven Rostedt 
30408fc0c701SSteven Rostedt 	(*pos)++;
304198c4fd04SSteven Rostedt 	iter->pos = *pos;
30428fc0c701SSteven Rostedt 
30434aeb6967SSteven Rostedt 	if (iter->probe)
30444aeb6967SSteven Rostedt 		hnd = &iter->probe->node;
30458fc0c701SSteven Rostedt  retry:
30468fc0c701SSteven Rostedt 	if (iter->hidx >= FTRACE_FUNC_HASHSIZE)
30478fc0c701SSteven Rostedt 		return NULL;
30488fc0c701SSteven Rostedt 
30498fc0c701SSteven Rostedt 	hhd = &ftrace_func_hash[iter->hidx];
30508fc0c701SSteven Rostedt 
30518fc0c701SSteven Rostedt 	if (hlist_empty(hhd)) {
30528fc0c701SSteven Rostedt 		iter->hidx++;
30538fc0c701SSteven Rostedt 		hnd = NULL;
30548fc0c701SSteven Rostedt 		goto retry;
30558fc0c701SSteven Rostedt 	}
30568fc0c701SSteven Rostedt 
30578fc0c701SSteven Rostedt 	if (!hnd)
30588fc0c701SSteven Rostedt 		hnd = hhd->first;
30598fc0c701SSteven Rostedt 	else {
30608fc0c701SSteven Rostedt 		hnd = hnd->next;
30618fc0c701SSteven Rostedt 		if (!hnd) {
30628fc0c701SSteven Rostedt 			iter->hidx++;
30638fc0c701SSteven Rostedt 			goto retry;
30648fc0c701SSteven Rostedt 		}
30658fc0c701SSteven Rostedt 	}
30668fc0c701SSteven Rostedt 
30674aeb6967SSteven Rostedt 	if (WARN_ON_ONCE(!hnd))
30684aeb6967SSteven Rostedt 		return NULL;
30694aeb6967SSteven Rostedt 
30704aeb6967SSteven Rostedt 	iter->probe = hlist_entry(hnd, struct ftrace_func_probe, node);
30714aeb6967SSteven Rostedt 
30724aeb6967SSteven Rostedt 	return iter;
30738fc0c701SSteven Rostedt }
30748fc0c701SSteven Rostedt 
30758fc0c701SSteven Rostedt static void *t_hash_start(struct seq_file *m, loff_t *pos)
30768fc0c701SSteven Rostedt {
30778fc0c701SSteven Rostedt 	struct ftrace_iterator *iter = m->private;
30788fc0c701SSteven Rostedt 	void *p = NULL;
3079d82d6244SLi Zefan 	loff_t l;
3080d82d6244SLi Zefan 
308169a3083cSSteven Rostedt 	if (!(iter->flags & FTRACE_ITER_DO_HASH))
308269a3083cSSteven Rostedt 		return NULL;
308369a3083cSSteven Rostedt 
30842bccfffdSSteven Rostedt 	if (iter->func_pos > *pos)
30852bccfffdSSteven Rostedt 		return NULL;
30868fc0c701SSteven Rostedt 
3087d82d6244SLi Zefan 	iter->hidx = 0;
30882bccfffdSSteven Rostedt 	for (l = 0; l <= (*pos - iter->func_pos); ) {
30894aeb6967SSteven Rostedt 		p = t_hash_next(m, &l);
3090d82d6244SLi Zefan 		if (!p)
3091d82d6244SLi Zefan 			break;
3092d82d6244SLi Zefan 	}
30934aeb6967SSteven Rostedt 	if (!p)
30944aeb6967SSteven Rostedt 		return NULL;
30954aeb6967SSteven Rostedt 
309698c4fd04SSteven Rostedt 	/* Only set this if we have an item */
309798c4fd04SSteven Rostedt 	iter->flags |= FTRACE_ITER_HASH;
309898c4fd04SSteven Rostedt 
30994aeb6967SSteven Rostedt 	return iter;
31008fc0c701SSteven Rostedt }
31018fc0c701SSteven Rostedt 
31024aeb6967SSteven Rostedt static int
31034aeb6967SSteven Rostedt t_hash_show(struct seq_file *m, struct ftrace_iterator *iter)
31048fc0c701SSteven Rostedt {
3105b6887d79SSteven Rostedt 	struct ftrace_func_probe *rec;
31068fc0c701SSteven Rostedt 
31074aeb6967SSteven Rostedt 	rec = iter->probe;
31084aeb6967SSteven Rostedt 	if (WARN_ON_ONCE(!rec))
31094aeb6967SSteven Rostedt 		return -EIO;
31108fc0c701SSteven Rostedt 
3111809dcf29SSteven Rostedt 	if (rec->ops->print)
3112809dcf29SSteven Rostedt 		return rec->ops->print(m, rec->ip, rec->ops, rec->data);
3113809dcf29SSteven Rostedt 
3114b375a11aSSteven Rostedt 	seq_printf(m, "%ps:%ps", (void *)rec->ip, (void *)rec->ops->func);
31158fc0c701SSteven Rostedt 
31168fc0c701SSteven Rostedt 	if (rec->data)
31178fc0c701SSteven Rostedt 		seq_printf(m, ":%p", rec->data);
31188fc0c701SSteven Rostedt 	seq_putc(m, '\n');
31198fc0c701SSteven Rostedt 
31208fc0c701SSteven Rostedt 	return 0;
31218fc0c701SSteven Rostedt }
31228fc0c701SSteven Rostedt 
31238fc0c701SSteven Rostedt static void *
31245072c59fSSteven Rostedt t_next(struct seq_file *m, void *v, loff_t *pos)
31255072c59fSSteven Rostedt {
31265072c59fSSteven Rostedt 	struct ftrace_iterator *iter = m->private;
3127fc13cb0cSSteven Rostedt 	struct ftrace_ops *ops = iter->ops;
31285072c59fSSteven Rostedt 	struct dyn_ftrace *rec = NULL;
31295072c59fSSteven Rostedt 
313045a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
313145a4a237SSteven Rostedt 		return NULL;
313245a4a237SSteven Rostedt 
31338fc0c701SSteven Rostedt 	if (iter->flags & FTRACE_ITER_HASH)
31344aeb6967SSteven Rostedt 		return t_hash_next(m, pos);
31358fc0c701SSteven Rostedt 
31365072c59fSSteven Rostedt 	(*pos)++;
31371106b699SJiri Olsa 	iter->pos = iter->func_pos = *pos;
31385072c59fSSteven Rostedt 
31390c75a3edSSteven Rostedt 	if (iter->flags & FTRACE_ITER_PRINTALL)
314057c072c7SSteven Rostedt 		return t_hash_start(m, pos);
31410c75a3edSSteven Rostedt 
31425072c59fSSteven Rostedt  retry:
31435072c59fSSteven Rostedt 	if (iter->idx >= iter->pg->index) {
31445072c59fSSteven Rostedt 		if (iter->pg->next) {
31455072c59fSSteven Rostedt 			iter->pg = iter->pg->next;
31465072c59fSSteven Rostedt 			iter->idx = 0;
31475072c59fSSteven Rostedt 			goto retry;
31485072c59fSSteven Rostedt 		}
31495072c59fSSteven Rostedt 	} else {
31505072c59fSSteven Rostedt 		rec = &iter->pg->records[iter->idx++];
315132082309SSteven Rostedt 		if (((iter->flags & FTRACE_ITER_FILTER) &&
315233b7f99cSSteven Rostedt (Red Hat) 		     !(ftrace_lookup_ip(ops->func_hash->filter_hash, rec->ip))) ||
31530183fb1cSSteven Rostedt 
315441c52c0dSSteven Rostedt 		    ((iter->flags & FTRACE_ITER_NOTRACE) &&
315533b7f99cSSteven Rostedt (Red Hat) 		     !ftrace_lookup_ip(ops->func_hash->notrace_hash, rec->ip)) ||
3156647bcd03SSteven Rostedt 
3157647bcd03SSteven Rostedt 		    ((iter->flags & FTRACE_ITER_ENABLED) &&
315823ea9c4dSSteven Rostedt (Red Hat) 		     !(rec->flags & FTRACE_FL_ENABLED))) {
3159647bcd03SSteven Rostedt 
31605072c59fSSteven Rostedt 			rec = NULL;
31615072c59fSSteven Rostedt 			goto retry;
31625072c59fSSteven Rostedt 		}
31635072c59fSSteven Rostedt 	}
31645072c59fSSteven Rostedt 
31654aeb6967SSteven Rostedt 	if (!rec)
316657c072c7SSteven Rostedt 		return t_hash_start(m, pos);
31674aeb6967SSteven Rostedt 
31684aeb6967SSteven Rostedt 	iter->func = rec;
31694aeb6967SSteven Rostedt 
31704aeb6967SSteven Rostedt 	return iter;
31715072c59fSSteven Rostedt }
31725072c59fSSteven Rostedt 
317398c4fd04SSteven Rostedt static void reset_iter_read(struct ftrace_iterator *iter)
317498c4fd04SSteven Rostedt {
317598c4fd04SSteven Rostedt 	iter->pos = 0;
317698c4fd04SSteven Rostedt 	iter->func_pos = 0;
317770f77b3fSDan Carpenter 	iter->flags &= ~(FTRACE_ITER_PRINTALL | FTRACE_ITER_HASH);
31785072c59fSSteven Rostedt }
31795072c59fSSteven Rostedt 
31805072c59fSSteven Rostedt static void *t_start(struct seq_file *m, loff_t *pos)
31815072c59fSSteven Rostedt {
31825072c59fSSteven Rostedt 	struct ftrace_iterator *iter = m->private;
3183fc13cb0cSSteven Rostedt 	struct ftrace_ops *ops = iter->ops;
31845072c59fSSteven Rostedt 	void *p = NULL;
3185694ce0a5SLi Zefan 	loff_t l;
31865072c59fSSteven Rostedt 
31878fc0c701SSteven Rostedt 	mutex_lock(&ftrace_lock);
318845a4a237SSteven Rostedt 
318945a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
319045a4a237SSteven Rostedt 		return NULL;
319145a4a237SSteven Rostedt 
31920c75a3edSSteven Rostedt 	/*
319398c4fd04SSteven Rostedt 	 * If an lseek was done, then reset and start from beginning.
319498c4fd04SSteven Rostedt 	 */
319598c4fd04SSteven Rostedt 	if (*pos < iter->pos)
319698c4fd04SSteven Rostedt 		reset_iter_read(iter);
319798c4fd04SSteven Rostedt 
319898c4fd04SSteven Rostedt 	/*
31990c75a3edSSteven Rostedt 	 * For set_ftrace_filter reading, if we have the filter
32000c75a3edSSteven Rostedt 	 * off, we can short cut and just print out that all
32010c75a3edSSteven Rostedt 	 * functions are enabled.
32020c75a3edSSteven Rostedt 	 */
32038c006cf7SNamhyung Kim 	if ((iter->flags & FTRACE_ITER_FILTER &&
320433b7f99cSSteven Rostedt (Red Hat) 	     ftrace_hash_empty(ops->func_hash->filter_hash)) ||
32058c006cf7SNamhyung Kim 	    (iter->flags & FTRACE_ITER_NOTRACE &&
320633b7f99cSSteven Rostedt (Red Hat) 	     ftrace_hash_empty(ops->func_hash->notrace_hash))) {
32070c75a3edSSteven Rostedt 		if (*pos > 0)
32088fc0c701SSteven Rostedt 			return t_hash_start(m, pos);
32090c75a3edSSteven Rostedt 		iter->flags |= FTRACE_ITER_PRINTALL;
3210df091625SChris Wright 		/* reset in case of seek/pread */
3211df091625SChris Wright 		iter->flags &= ~FTRACE_ITER_HASH;
32120c75a3edSSteven Rostedt 		return iter;
32130c75a3edSSteven Rostedt 	}
32140c75a3edSSteven Rostedt 
32158fc0c701SSteven Rostedt 	if (iter->flags & FTRACE_ITER_HASH)
32168fc0c701SSteven Rostedt 		return t_hash_start(m, pos);
32178fc0c701SSteven Rostedt 
321898c4fd04SSteven Rostedt 	/*
321998c4fd04SSteven Rostedt 	 * Unfortunately, we need to restart at ftrace_pages_start
322098c4fd04SSteven Rostedt 	 * every time we let go of the ftrace_mutex. This is because
322198c4fd04SSteven Rostedt 	 * those pointers can change without the lock.
322298c4fd04SSteven Rostedt 	 */
3223694ce0a5SLi Zefan 	iter->pg = ftrace_pages_start;
3224694ce0a5SLi Zefan 	iter->idx = 0;
3225694ce0a5SLi Zefan 	for (l = 0; l <= *pos; ) {
3226694ce0a5SLi Zefan 		p = t_next(m, p, &l);
3227694ce0a5SLi Zefan 		if (!p)
3228694ce0a5SLi Zefan 			break;
322950cdaf08SLiming Wang 	}
32305821e1b7Swalimis 
323169a3083cSSteven Rostedt 	if (!p)
32328fc0c701SSteven Rostedt 		return t_hash_start(m, pos);
32338fc0c701SSteven Rostedt 
32344aeb6967SSteven Rostedt 	return iter;
32355072c59fSSteven Rostedt }
32365072c59fSSteven Rostedt 
32375072c59fSSteven Rostedt static void t_stop(struct seq_file *m, void *p)
32385072c59fSSteven Rostedt {
32398fc0c701SSteven Rostedt 	mutex_unlock(&ftrace_lock);
32405072c59fSSteven Rostedt }
32415072c59fSSteven Rostedt 
324215d5b02cSSteven Rostedt (Red Hat) void * __weak
324315d5b02cSSteven Rostedt (Red Hat) arch_ftrace_trampoline_func(struct ftrace_ops *ops, struct dyn_ftrace *rec)
324415d5b02cSSteven Rostedt (Red Hat) {
324515d5b02cSSteven Rostedt (Red Hat) 	return NULL;
324615d5b02cSSteven Rostedt (Red Hat) }
324715d5b02cSSteven Rostedt (Red Hat) 
324815d5b02cSSteven Rostedt (Red Hat) static void add_trampoline_func(struct seq_file *m, struct ftrace_ops *ops,
324915d5b02cSSteven Rostedt (Red Hat) 				struct dyn_ftrace *rec)
325015d5b02cSSteven Rostedt (Red Hat) {
325115d5b02cSSteven Rostedt (Red Hat) 	void *ptr;
325215d5b02cSSteven Rostedt (Red Hat) 
325315d5b02cSSteven Rostedt (Red Hat) 	ptr = arch_ftrace_trampoline_func(ops, rec);
325415d5b02cSSteven Rostedt (Red Hat) 	if (ptr)
325515d5b02cSSteven Rostedt (Red Hat) 		seq_printf(m, " ->%pS", ptr);
325615d5b02cSSteven Rostedt (Red Hat) }
325715d5b02cSSteven Rostedt (Red Hat) 
32585072c59fSSteven Rostedt static int t_show(struct seq_file *m, void *v)
32595072c59fSSteven Rostedt {
32600c75a3edSSteven Rostedt 	struct ftrace_iterator *iter = m->private;
32614aeb6967SSteven Rostedt 	struct dyn_ftrace *rec;
32625072c59fSSteven Rostedt 
32638fc0c701SSteven Rostedt 	if (iter->flags & FTRACE_ITER_HASH)
32644aeb6967SSteven Rostedt 		return t_hash_show(m, iter);
32658fc0c701SSteven Rostedt 
32660c75a3edSSteven Rostedt 	if (iter->flags & FTRACE_ITER_PRINTALL) {
32678c006cf7SNamhyung Kim 		if (iter->flags & FTRACE_ITER_NOTRACE)
3268fa6f0cc7SRasmus Villemoes 			seq_puts(m, "#### no functions disabled ####\n");
32698c006cf7SNamhyung Kim 		else
3270fa6f0cc7SRasmus Villemoes 			seq_puts(m, "#### all functions enabled ####\n");
32710c75a3edSSteven Rostedt 		return 0;
32720c75a3edSSteven Rostedt 	}
32730c75a3edSSteven Rostedt 
32744aeb6967SSteven Rostedt 	rec = iter->func;
32754aeb6967SSteven Rostedt 
32765072c59fSSteven Rostedt 	if (!rec)
32775072c59fSSteven Rostedt 		return 0;
32785072c59fSSteven Rostedt 
3279647bcd03SSteven Rostedt 	seq_printf(m, "%ps", (void *)rec->ip);
32809674b2faSSteven Rostedt (Red Hat) 	if (iter->flags & FTRACE_ITER_ENABLED) {
3281030f4e1cSSteven Rostedt (Red Hat) 		struct ftrace_ops *ops;
328215d5b02cSSteven Rostedt (Red Hat) 
3283f8b8be8aSMasami Hiramatsu 		seq_printf(m, " (%ld)%s%s",
32840376bde1SSteven Rostedt (Red Hat) 			   ftrace_rec_count(rec),
3285f8b8be8aSMasami Hiramatsu 			   rec->flags & FTRACE_FL_REGS ? " R" : "  ",
3286f8b8be8aSMasami Hiramatsu 			   rec->flags & FTRACE_FL_IPMODIFY ? " I" : "  ");
32879674b2faSSteven Rostedt (Red Hat) 		if (rec->flags & FTRACE_FL_TRAMP_EN) {
32885fecaa04SSteven Rostedt (Red Hat) 			ops = ftrace_find_tramp_ops_any(rec);
328939daa7b9SSteven Rostedt (Red Hat) 			if (ops) {
329039daa7b9SSteven Rostedt (Red Hat) 				do {
329139daa7b9SSteven Rostedt (Red Hat) 					seq_printf(m, "\ttramp: %pS (%pS)",
329239daa7b9SSteven Rostedt (Red Hat) 						   (void *)ops->trampoline,
329339daa7b9SSteven Rostedt (Red Hat) 						   (void *)ops->func);
3294030f4e1cSSteven Rostedt (Red Hat) 					add_trampoline_func(m, ops, rec);
329539daa7b9SSteven Rostedt (Red Hat) 					ops = ftrace_find_tramp_ops_next(rec, ops);
329639daa7b9SSteven Rostedt (Red Hat) 				} while (ops);
329739daa7b9SSteven Rostedt (Red Hat) 			} else
3298fa6f0cc7SRasmus Villemoes 				seq_puts(m, "\ttramp: ERROR!");
3299030f4e1cSSteven Rostedt (Red Hat) 		} else {
3300030f4e1cSSteven Rostedt (Red Hat) 			add_trampoline_func(m, NULL, rec);
33019674b2faSSteven Rostedt (Red Hat) 		}
33029674b2faSSteven Rostedt (Red Hat) 	}
33039674b2faSSteven Rostedt (Red Hat) 
3304fa6f0cc7SRasmus Villemoes 	seq_putc(m, '\n');
33055072c59fSSteven Rostedt 
33065072c59fSSteven Rostedt 	return 0;
33075072c59fSSteven Rostedt }
33085072c59fSSteven Rostedt 
330988e9d34cSJames Morris static const struct seq_operations show_ftrace_seq_ops = {
33105072c59fSSteven Rostedt 	.start = t_start,
33115072c59fSSteven Rostedt 	.next = t_next,
33125072c59fSSteven Rostedt 	.stop = t_stop,
33135072c59fSSteven Rostedt 	.show = t_show,
33145072c59fSSteven Rostedt };
33155072c59fSSteven Rostedt 
3316e309b41dSIngo Molnar static int
33175072c59fSSteven Rostedt ftrace_avail_open(struct inode *inode, struct file *file)
33185072c59fSSteven Rostedt {
33195072c59fSSteven Rostedt 	struct ftrace_iterator *iter;
33205072c59fSSteven Rostedt 
33214eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
33224eebcc81SSteven Rostedt 		return -ENODEV;
33234eebcc81SSteven Rostedt 
332450e18b94SJiri Olsa 	iter = __seq_open_private(file, &show_ftrace_seq_ops, sizeof(*iter));
332550e18b94SJiri Olsa 	if (iter) {
33265072c59fSSteven Rostedt 		iter->pg = ftrace_pages_start;
3327fc13cb0cSSteven Rostedt 		iter->ops = &global_ops;
33284bf39a94SIngo Molnar 	}
33295072c59fSSteven Rostedt 
333050e18b94SJiri Olsa 	return iter ? 0 : -ENOMEM;
33315072c59fSSteven Rostedt }
33325072c59fSSteven Rostedt 
3333647bcd03SSteven Rostedt static int
3334647bcd03SSteven Rostedt ftrace_enabled_open(struct inode *inode, struct file *file)
3335647bcd03SSteven Rostedt {
3336647bcd03SSteven Rostedt 	struct ftrace_iterator *iter;
3337647bcd03SSteven Rostedt 
333850e18b94SJiri Olsa 	iter = __seq_open_private(file, &show_ftrace_seq_ops, sizeof(*iter));
333950e18b94SJiri Olsa 	if (iter) {
3340647bcd03SSteven Rostedt 		iter->pg = ftrace_pages_start;
3341647bcd03SSteven Rostedt 		iter->flags = FTRACE_ITER_ENABLED;
3342fc13cb0cSSteven Rostedt 		iter->ops = &global_ops;
3343647bcd03SSteven Rostedt 	}
3344647bcd03SSteven Rostedt 
334550e18b94SJiri Olsa 	return iter ? 0 : -ENOMEM;
3346647bcd03SSteven Rostedt }
3347647bcd03SSteven Rostedt 
3348fc13cb0cSSteven Rostedt /**
3349fc13cb0cSSteven Rostedt  * ftrace_regex_open - initialize function tracer filter files
3350fc13cb0cSSteven Rostedt  * @ops: The ftrace_ops that hold the hash filters
3351fc13cb0cSSteven Rostedt  * @flag: The type of filter to process
3352fc13cb0cSSteven Rostedt  * @inode: The inode, usually passed in to your open routine
3353fc13cb0cSSteven Rostedt  * @file: The file, usually passed in to your open routine
3354fc13cb0cSSteven Rostedt  *
3355fc13cb0cSSteven Rostedt  * ftrace_regex_open() initializes the filter files for the
3356fc13cb0cSSteven Rostedt  * @ops. Depending on @flag it may process the filter hash or
3357fc13cb0cSSteven Rostedt  * the notrace hash of @ops. With this called from the open
3358fc13cb0cSSteven Rostedt  * routine, you can use ftrace_filter_write() for the write
3359fc13cb0cSSteven Rostedt  * routine if @flag has FTRACE_ITER_FILTER set, or
3360fc13cb0cSSteven Rostedt  * ftrace_notrace_write() if @flag has FTRACE_ITER_NOTRACE set.
3361098c879eSSteven Rostedt (Red Hat)  * tracing_lseek() should be used as the lseek routine, and
3362fc13cb0cSSteven Rostedt  * release must call ftrace_regex_release().
3363fc13cb0cSSteven Rostedt  */
3364fc13cb0cSSteven Rostedt int
3365f45948e8SSteven Rostedt ftrace_regex_open(struct ftrace_ops *ops, int flag,
33661cf41dd7SSteven Rostedt 		  struct inode *inode, struct file *file)
33675072c59fSSteven Rostedt {
33685072c59fSSteven Rostedt 	struct ftrace_iterator *iter;
3369f45948e8SSteven Rostedt 	struct ftrace_hash *hash;
33705072c59fSSteven Rostedt 	int ret = 0;
33715072c59fSSteven Rostedt 
3372f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
3373f04f24fbSMasami Hiramatsu 
33744eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
33754eebcc81SSteven Rostedt 		return -ENODEV;
33764eebcc81SSteven Rostedt 
33775072c59fSSteven Rostedt 	iter = kzalloc(sizeof(*iter), GFP_KERNEL);
33785072c59fSSteven Rostedt 	if (!iter)
33795072c59fSSteven Rostedt 		return -ENOMEM;
33805072c59fSSteven Rostedt 
3381689fd8b6S[email protected] 	if (trace_parser_get_init(&iter->parser, FTRACE_BUFF_MAX)) {
3382689fd8b6S[email protected] 		kfree(iter);
3383689fd8b6S[email protected] 		return -ENOMEM;
3384689fd8b6S[email protected] 	}
3385689fd8b6S[email protected] 
33863f2367baSMasami Hiramatsu 	iter->ops = ops;
33873f2367baSMasami Hiramatsu 	iter->flags = flag;
33883f2367baSMasami Hiramatsu 
338933b7f99cSSteven Rostedt (Red Hat) 	mutex_lock(&ops->func_hash->regex_lock);
33903f2367baSMasami Hiramatsu 
3391f45948e8SSteven Rostedt 	if (flag & FTRACE_ITER_NOTRACE)
339233b7f99cSSteven Rostedt (Red Hat) 		hash = ops->func_hash->notrace_hash;
3393f45948e8SSteven Rostedt 	else
339433b7f99cSSteven Rostedt (Red Hat) 		hash = ops->func_hash->filter_hash;
3395f45948e8SSteven Rostedt 
339633dc9b12SSteven Rostedt 	if (file->f_mode & FMODE_WRITE) {
3397ef2fbe16SNamhyung Kim 		const int size_bits = FTRACE_HASH_DEFAULT_BITS;
3398ef2fbe16SNamhyung Kim 
3399ef2fbe16SNamhyung Kim 		if (file->f_flags & O_TRUNC)
3400ef2fbe16SNamhyung Kim 			iter->hash = alloc_ftrace_hash(size_bits);
3401ef2fbe16SNamhyung Kim 		else
3402ef2fbe16SNamhyung Kim 			iter->hash = alloc_and_copy_ftrace_hash(size_bits, hash);
3403ef2fbe16SNamhyung Kim 
340433dc9b12SSteven Rostedt 		if (!iter->hash) {
340533dc9b12SSteven Rostedt 			trace_parser_put(&iter->parser);
340633dc9b12SSteven Rostedt 			kfree(iter);
34073f2367baSMasami Hiramatsu 			ret = -ENOMEM;
34083f2367baSMasami Hiramatsu 			goto out_unlock;
340933dc9b12SSteven Rostedt 		}
341033dc9b12SSteven Rostedt 	}
34111cf41dd7SSteven Rostedt 
34125072c59fSSteven Rostedt 	if (file->f_mode & FMODE_READ) {
34135072c59fSSteven Rostedt 		iter->pg = ftrace_pages_start;
34145072c59fSSteven Rostedt 
34155072c59fSSteven Rostedt 		ret = seq_open(file, &show_ftrace_seq_ops);
34165072c59fSSteven Rostedt 		if (!ret) {
34175072c59fSSteven Rostedt 			struct seq_file *m = file->private_data;
34185072c59fSSteven Rostedt 			m->private = iter;
341979fe249cSLi Zefan 		} else {
342033dc9b12SSteven Rostedt 			/* Failed */
342133dc9b12SSteven Rostedt 			free_ftrace_hash(iter->hash);
342279fe249cSLi Zefan 			trace_parser_put(&iter->parser);
34235072c59fSSteven Rostedt 			kfree(iter);
342479fe249cSLi Zefan 		}
34255072c59fSSteven Rostedt 	} else
34265072c59fSSteven Rostedt 		file->private_data = iter;
34273f2367baSMasami Hiramatsu 
34283f2367baSMasami Hiramatsu  out_unlock:
342933b7f99cSSteven Rostedt (Red Hat) 	mutex_unlock(&ops->func_hash->regex_lock);
34305072c59fSSteven Rostedt 
34315072c59fSSteven Rostedt 	return ret;
34325072c59fSSteven Rostedt }
34335072c59fSSteven Rostedt 
343441c52c0dSSteven Rostedt static int
343541c52c0dSSteven Rostedt ftrace_filter_open(struct inode *inode, struct file *file)
343641c52c0dSSteven Rostedt {
3437e3b3e2e8SSteven Rostedt (Red Hat) 	struct ftrace_ops *ops = inode->i_private;
3438e3b3e2e8SSteven Rostedt (Red Hat) 
3439e3b3e2e8SSteven Rostedt (Red Hat) 	return ftrace_regex_open(ops,
344069a3083cSSteven Rostedt 			FTRACE_ITER_FILTER | FTRACE_ITER_DO_HASH,
34411cf41dd7SSteven Rostedt 			inode, file);
344241c52c0dSSteven Rostedt }
344341c52c0dSSteven Rostedt 
344441c52c0dSSteven Rostedt static int
344541c52c0dSSteven Rostedt ftrace_notrace_open(struct inode *inode, struct file *file)
344641c52c0dSSteven Rostedt {
3447e3b3e2e8SSteven Rostedt (Red Hat) 	struct ftrace_ops *ops = inode->i_private;
3448e3b3e2e8SSteven Rostedt (Red Hat) 
3449e3b3e2e8SSteven Rostedt (Red Hat) 	return ftrace_regex_open(ops, FTRACE_ITER_NOTRACE,
34501cf41dd7SSteven Rostedt 				 inode, file);
345141c52c0dSSteven Rostedt }
345241c52c0dSSteven Rostedt 
34533ba00929SDmitry Safonov /* Type for quick search ftrace basic regexes (globs) from filter_parse_regex */
34543ba00929SDmitry Safonov struct ftrace_glob {
34553ba00929SDmitry Safonov 	char *search;
34563ba00929SDmitry Safonov 	unsigned len;
34573ba00929SDmitry Safonov 	int type;
34583ba00929SDmitry Safonov };
34593ba00929SDmitry Safonov 
34607132e2d6SThiago Jung Bauermann /*
34617132e2d6SThiago Jung Bauermann  * If symbols in an architecture don't correspond exactly to the user-visible
34627132e2d6SThiago Jung Bauermann  * name of what they represent, it is possible to define this function to
34637132e2d6SThiago Jung Bauermann  * perform the necessary adjustments.
34647132e2d6SThiago Jung Bauermann */
34657132e2d6SThiago Jung Bauermann char * __weak arch_ftrace_match_adjust(char *str, const char *search)
34667132e2d6SThiago Jung Bauermann {
34677132e2d6SThiago Jung Bauermann 	return str;
34687132e2d6SThiago Jung Bauermann }
34697132e2d6SThiago Jung Bauermann 
34703ba00929SDmitry Safonov static int ftrace_match(char *str, struct ftrace_glob *g)
34719f4801e3SSteven Rostedt {
34729f4801e3SSteven Rostedt 	int matched = 0;
3473751e9983SLi Zefan 	int slen;
34749f4801e3SSteven Rostedt 
34757132e2d6SThiago Jung Bauermann 	str = arch_ftrace_match_adjust(str, g->search);
34767132e2d6SThiago Jung Bauermann 
34773ba00929SDmitry Safonov 	switch (g->type) {
34789f4801e3SSteven Rostedt 	case MATCH_FULL:
34793ba00929SDmitry Safonov 		if (strcmp(str, g->search) == 0)
34809f4801e3SSteven Rostedt 			matched = 1;
34819f4801e3SSteven Rostedt 		break;
34829f4801e3SSteven Rostedt 	case MATCH_FRONT_ONLY:
34833ba00929SDmitry Safonov 		if (strncmp(str, g->search, g->len) == 0)
34849f4801e3SSteven Rostedt 			matched = 1;
34859f4801e3SSteven Rostedt 		break;
34869f4801e3SSteven Rostedt 	case MATCH_MIDDLE_ONLY:
34873ba00929SDmitry Safonov 		if (strstr(str, g->search))
34889f4801e3SSteven Rostedt 			matched = 1;
34899f4801e3SSteven Rostedt 		break;
34909f4801e3SSteven Rostedt 	case MATCH_END_ONLY:
3491751e9983SLi Zefan 		slen = strlen(str);
34923ba00929SDmitry Safonov 		if (slen >= g->len &&
34933ba00929SDmitry Safonov 		    memcmp(str + slen - g->len, g->search, g->len) == 0)
34949f4801e3SSteven Rostedt 			matched = 1;
34959f4801e3SSteven Rostedt 		break;
34969f4801e3SSteven Rostedt 	}
34979f4801e3SSteven Rostedt 
34989f4801e3SSteven Rostedt 	return matched;
34999f4801e3SSteven Rostedt }
35009f4801e3SSteven Rostedt 
3501b448c4e3SSteven Rostedt static int
3502f0a3b154SDmitry Safonov enter_record(struct ftrace_hash *hash, struct dyn_ftrace *rec, int clear_filter)
350364e7c440SSteven Rostedt {
3504b448c4e3SSteven Rostedt 	struct ftrace_func_entry *entry;
3505b448c4e3SSteven Rostedt 	int ret = 0;
3506b448c4e3SSteven Rostedt 
35071cf41dd7SSteven Rostedt 	entry = ftrace_lookup_ip(hash, rec->ip);
3508f0a3b154SDmitry Safonov 	if (clear_filter) {
3509b448c4e3SSteven Rostedt 		/* Do nothing if it doesn't exist */
3510b448c4e3SSteven Rostedt 		if (!entry)
3511b448c4e3SSteven Rostedt 			return 0;
3512b448c4e3SSteven Rostedt 
351333dc9b12SSteven Rostedt 		free_hash_entry(hash, entry);
3514b448c4e3SSteven Rostedt 	} else {
3515b448c4e3SSteven Rostedt 		/* Do nothing if it exists */
3516b448c4e3SSteven Rostedt 		if (entry)
3517b448c4e3SSteven Rostedt 			return 0;
3518b448c4e3SSteven Rostedt 
3519b448c4e3SSteven Rostedt 		ret = add_hash_entry(hash, rec->ip);
3520b448c4e3SSteven Rostedt 	}
3521b448c4e3SSteven Rostedt 	return ret;
3522996e87beSSteven Rostedt }
3523996e87beSSteven Rostedt 
352464e7c440SSteven Rostedt static int
35250b507e1eSDmitry Safonov ftrace_match_record(struct dyn_ftrace *rec, struct ftrace_glob *func_g,
35260b507e1eSDmitry Safonov 		struct ftrace_glob *mod_g, int exclude_mod)
352764e7c440SSteven Rostedt {
352864e7c440SSteven Rostedt 	char str[KSYM_SYMBOL_LEN];
3529b9df92d2SSteven Rostedt 	char *modname;
353064e7c440SSteven Rostedt 
3531b9df92d2SSteven Rostedt 	kallsyms_lookup(rec->ip, NULL, NULL, &modname, str);
3532b9df92d2SSteven Rostedt 
35330b507e1eSDmitry Safonov 	if (mod_g) {
35340b507e1eSDmitry Safonov 		int mod_matches = (modname) ? ftrace_match(modname, mod_g) : 0;
35350b507e1eSDmitry Safonov 
35360b507e1eSDmitry Safonov 		/* blank module name to match all modules */
35370b507e1eSDmitry Safonov 		if (!mod_g->len) {
35380b507e1eSDmitry Safonov 			/* blank module globbing: modname xor exclude_mod */
35390b507e1eSDmitry Safonov 			if ((!exclude_mod) != (!modname))
35400b507e1eSDmitry Safonov 				goto func_match;
35410b507e1eSDmitry Safonov 			return 0;
35420b507e1eSDmitry Safonov 		}
35430b507e1eSDmitry Safonov 
35440b507e1eSDmitry Safonov 		/* not matching the module */
35450b507e1eSDmitry Safonov 		if (!modname || !mod_matches) {
35460b507e1eSDmitry Safonov 			if (exclude_mod)
35470b507e1eSDmitry Safonov 				goto func_match;
35480b507e1eSDmitry Safonov 			else
35490b507e1eSDmitry Safonov 				return 0;
35500b507e1eSDmitry Safonov 		}
35510b507e1eSDmitry Safonov 
35520b507e1eSDmitry Safonov 		if (mod_matches && exclude_mod)
3553b9df92d2SSteven Rostedt 			return 0;
3554b9df92d2SSteven Rostedt 
35550b507e1eSDmitry Safonov func_match:
3556b9df92d2SSteven Rostedt 		/* blank search means to match all funcs in the mod */
35573ba00929SDmitry Safonov 		if (!func_g->len)
3558b9df92d2SSteven Rostedt 			return 1;
3559b9df92d2SSteven Rostedt 	}
3560b9df92d2SSteven Rostedt 
35613ba00929SDmitry Safonov 	return ftrace_match(str, func_g);
356264e7c440SSteven Rostedt }
356364e7c440SSteven Rostedt 
35641cf41dd7SSteven Rostedt static int
35653ba00929SDmitry Safonov match_records(struct ftrace_hash *hash, char *func, int len, char *mod)
35669f4801e3SSteven Rostedt {
35679f4801e3SSteven Rostedt 	struct ftrace_page *pg;
35689f4801e3SSteven Rostedt 	struct dyn_ftrace *rec;
35693ba00929SDmitry Safonov 	struct ftrace_glob func_g = { .type = MATCH_FULL };
35700b507e1eSDmitry Safonov 	struct ftrace_glob mod_g = { .type = MATCH_FULL };
35710b507e1eSDmitry Safonov 	struct ftrace_glob *mod_match = (mod) ? &mod_g : NULL;
35720b507e1eSDmitry Safonov 	int exclude_mod = 0;
3573311d16daSLi Zefan 	int found = 0;
3574b448c4e3SSteven Rostedt 	int ret;
3575f0a3b154SDmitry Safonov 	int clear_filter;
35769f4801e3SSteven Rostedt 
35770b507e1eSDmitry Safonov 	if (func) {
35783ba00929SDmitry Safonov 		func_g.type = filter_parse_regex(func, len, &func_g.search,
35793ba00929SDmitry Safonov 						 &clear_filter);
35803ba00929SDmitry Safonov 		func_g.len = strlen(func_g.search);
3581b9df92d2SSteven Rostedt 	}
3582b9df92d2SSteven Rostedt 
35830b507e1eSDmitry Safonov 	if (mod) {
35840b507e1eSDmitry Safonov 		mod_g.type = filter_parse_regex(mod, strlen(mod),
35850b507e1eSDmitry Safonov 				&mod_g.search, &exclude_mod);
35860b507e1eSDmitry Safonov 		mod_g.len = strlen(mod_g.search);
3587b9df92d2SSteven Rostedt 	}
3588b9df92d2SSteven Rostedt 
358952baf119SSteven Rostedt 	mutex_lock(&ftrace_lock);
3590b9df92d2SSteven Rostedt 
3591b9df92d2SSteven Rostedt 	if (unlikely(ftrace_disabled))
3592b9df92d2SSteven Rostedt 		goto out_unlock;
3593b9df92d2SSteven Rostedt 
3594265c831cSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
35950b507e1eSDmitry Safonov 		if (ftrace_match_record(rec, &func_g, mod_match, exclude_mod)) {
3596f0a3b154SDmitry Safonov 			ret = enter_record(hash, rec, clear_filter);
3597b448c4e3SSteven Rostedt 			if (ret < 0) {
3598b448c4e3SSteven Rostedt 				found = ret;
3599b448c4e3SSteven Rostedt 				goto out_unlock;
3600b448c4e3SSteven Rostedt 			}
3601311d16daSLi Zefan 			found = 1;
36025072c59fSSteven Rostedt 		}
3603265c831cSSteven Rostedt 	} while_for_each_ftrace_rec();
3604b9df92d2SSteven Rostedt  out_unlock:
360552baf119SSteven Rostedt 	mutex_unlock(&ftrace_lock);
3606311d16daSLi Zefan 
3607311d16daSLi Zefan 	return found;
36085072c59fSSteven Rostedt }
36095072c59fSSteven Rostedt 
361064e7c440SSteven Rostedt static int
36111cf41dd7SSteven Rostedt ftrace_match_records(struct ftrace_hash *hash, char *buff, int len)
361264e7c440SSteven Rostedt {
3613f0a3b154SDmitry Safonov 	return match_records(hash, buff, len, NULL);
361464e7c440SSteven Rostedt }
361564e7c440SSteven Rostedt 
361664e7c440SSteven Rostedt 
3617f6180773SSteven Rostedt /*
3618f6180773SSteven Rostedt  * We register the module command as a template to show others how
3619f6180773SSteven Rostedt  * to register the a command as well.
3620f6180773SSteven Rostedt  */
3621f6180773SSteven Rostedt 
3622f6180773SSteven Rostedt static int
362343dd61c9SSteven Rostedt ftrace_mod_callback(struct ftrace_hash *hash,
3624f0a3b154SDmitry Safonov 		    char *func, char *cmd, char *module, int enable)
3625f6180773SSteven Rostedt {
36265e3949f0SDmitry Safonov 	int ret;
3627f6180773SSteven Rostedt 
3628f6180773SSteven Rostedt 	/*
3629f6180773SSteven Rostedt 	 * cmd == 'mod' because we only registered this func
3630f6180773SSteven Rostedt 	 * for the 'mod' ftrace_func_command.
3631f6180773SSteven Rostedt 	 * But if you register one func with multiple commands,
3632f6180773SSteven Rostedt 	 * you can tell which command was used by the cmd
3633f6180773SSteven Rostedt 	 * parameter.
3634f6180773SSteven Rostedt 	 */
3635f0a3b154SDmitry Safonov 	ret = match_records(hash, func, strlen(func), module);
3636b448c4e3SSteven Rostedt 	if (!ret)
36375e3949f0SDmitry Safonov 		return -EINVAL;
3638b448c4e3SSteven Rostedt 	if (ret < 0)
3639b448c4e3SSteven Rostedt 		return ret;
3640f6180773SSteven Rostedt 	return 0;
3641f6180773SSteven Rostedt }
3642f6180773SSteven Rostedt 
3643f6180773SSteven Rostedt static struct ftrace_func_command ftrace_mod_cmd = {
3644f6180773SSteven Rostedt 	.name			= "mod",
3645f6180773SSteven Rostedt 	.func			= ftrace_mod_callback,
3646f6180773SSteven Rostedt };
3647f6180773SSteven Rostedt 
3648f6180773SSteven Rostedt static int __init ftrace_mod_cmd_init(void)
3649f6180773SSteven Rostedt {
3650f6180773SSteven Rostedt 	return register_ftrace_command(&ftrace_mod_cmd);
3651f6180773SSteven Rostedt }
36526f415672SSteven Rostedt core_initcall(ftrace_mod_cmd_init);
3653f6180773SSteven Rostedt 
36542f5f6ad9SSteven Rostedt static void function_trace_probe_call(unsigned long ip, unsigned long parent_ip,
3655a1e2e31dSSteven Rostedt 				      struct ftrace_ops *op, struct pt_regs *pt_regs)
365659df055fSSteven Rostedt {
3657b6887d79SSteven Rostedt 	struct ftrace_func_probe *entry;
365859df055fSSteven Rostedt 	struct hlist_head *hhd;
365959df055fSSteven Rostedt 	unsigned long key;
366059df055fSSteven Rostedt 
366159df055fSSteven Rostedt 	key = hash_long(ip, FTRACE_HASH_BITS);
366259df055fSSteven Rostedt 
366359df055fSSteven Rostedt 	hhd = &ftrace_func_hash[key];
366459df055fSSteven Rostedt 
366559df055fSSteven Rostedt 	if (hlist_empty(hhd))
366659df055fSSteven Rostedt 		return;
366759df055fSSteven Rostedt 
366859df055fSSteven Rostedt 	/*
366959df055fSSteven Rostedt 	 * Disable preemption for these calls to prevent a RCU grace
367059df055fSSteven Rostedt 	 * period. This syncs the hash iteration and freeing of items
367159df055fSSteven Rostedt 	 * on the hash. rcu_read_lock is too dangerous here.
367259df055fSSteven Rostedt 	 */
36735168ae50SSteven Rostedt 	preempt_disable_notrace();
36741bb539caSSteven Rostedt 	hlist_for_each_entry_rcu_notrace(entry, hhd, node) {
367559df055fSSteven Rostedt 		if (entry->ip == ip)
367659df055fSSteven Rostedt 			entry->ops->func(ip, parent_ip, &entry->data);
367759df055fSSteven Rostedt 	}
36785168ae50SSteven Rostedt 	preempt_enable_notrace();
367959df055fSSteven Rostedt }
368059df055fSSteven Rostedt 
3681b6887d79SSteven Rostedt static struct ftrace_ops trace_probe_ops __read_mostly =
368259df055fSSteven Rostedt {
3683b6887d79SSteven Rostedt 	.func		= function_trace_probe_call,
3684f04f24fbSMasami Hiramatsu 	.flags		= FTRACE_OPS_FL_INITIALIZED,
368533b7f99cSSteven Rostedt (Red Hat) 	INIT_OPS_HASH(trace_probe_ops)
368659df055fSSteven Rostedt };
368759df055fSSteven Rostedt 
3688b6887d79SSteven Rostedt static int ftrace_probe_registered;
368959df055fSSteven Rostedt 
36907485058eSSteven Rostedt (Red Hat) static void __enable_ftrace_function_probe(struct ftrace_ops_hash *old_hash)
369159df055fSSteven Rostedt {
3692b848914cSSteven Rostedt 	int ret;
369359df055fSSteven Rostedt 	int i;
369459df055fSSteven Rostedt 
369519dd603eSSteven Rostedt (Red Hat) 	if (ftrace_probe_registered) {
369619dd603eSSteven Rostedt (Red Hat) 		/* still need to update the function call sites */
369719dd603eSSteven Rostedt (Red Hat) 		if (ftrace_enabled)
36988252ecf3SSteven Rostedt (Red Hat) 			ftrace_run_modify_code(&trace_probe_ops, FTRACE_UPDATE_CALLS,
36998252ecf3SSteven Rostedt (Red Hat) 					       old_hash);
370059df055fSSteven Rostedt 		return;
370119dd603eSSteven Rostedt (Red Hat) 	}
370259df055fSSteven Rostedt 
370359df055fSSteven Rostedt 	for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) {
370459df055fSSteven Rostedt 		struct hlist_head *hhd = &ftrace_func_hash[i];
370559df055fSSteven Rostedt 		if (hhd->first)
370659df055fSSteven Rostedt 			break;
370759df055fSSteven Rostedt 	}
370859df055fSSteven Rostedt 	/* Nothing registered? */
370959df055fSSteven Rostedt 	if (i == FTRACE_FUNC_HASHSIZE)
371059df055fSSteven Rostedt 		return;
371159df055fSSteven Rostedt 
3712a1cd6173SSteven Rostedt 	ret = ftrace_startup(&trace_probe_ops, 0);
3713b848914cSSteven Rostedt 
3714b6887d79SSteven Rostedt 	ftrace_probe_registered = 1;
371559df055fSSteven Rostedt }
371659df055fSSteven Rostedt 
3717b6887d79SSteven Rostedt static void __disable_ftrace_function_probe(void)
371859df055fSSteven Rostedt {
371959df055fSSteven Rostedt 	int i;
372059df055fSSteven Rostedt 
3721b6887d79SSteven Rostedt 	if (!ftrace_probe_registered)
372259df055fSSteven Rostedt 		return;
372359df055fSSteven Rostedt 
372459df055fSSteven Rostedt 	for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) {
372559df055fSSteven Rostedt 		struct hlist_head *hhd = &ftrace_func_hash[i];
372659df055fSSteven Rostedt 		if (hhd->first)
372759df055fSSteven Rostedt 			return;
372859df055fSSteven Rostedt 	}
372959df055fSSteven Rostedt 
373059df055fSSteven Rostedt 	/* no more funcs left */
3731b848914cSSteven Rostedt 	ftrace_shutdown(&trace_probe_ops, 0);
3732b848914cSSteven Rostedt 
3733b6887d79SSteven Rostedt 	ftrace_probe_registered = 0;
373459df055fSSteven Rostedt }
373559df055fSSteven Rostedt 
373659df055fSSteven Rostedt 
37377818b388SSteven Rostedt (Red Hat) static void ftrace_free_entry(struct ftrace_func_probe *entry)
373859df055fSSteven Rostedt {
373959df055fSSteven Rostedt 	if (entry->ops->free)
3740e67efb93SSteven Rostedt (Red Hat) 		entry->ops->free(entry->ops, entry->ip, &entry->data);
374159df055fSSteven Rostedt 	kfree(entry);
374259df055fSSteven Rostedt }
374359df055fSSteven Rostedt 
374459df055fSSteven Rostedt int
3745b6887d79SSteven Rostedt register_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops,
374659df055fSSteven Rostedt 			      void *data)
374759df055fSSteven Rostedt {
37487485058eSSteven Rostedt (Red Hat) 	struct ftrace_ops_hash old_hash_ops;
3749b6887d79SSteven Rostedt 	struct ftrace_func_probe *entry;
37503ba00929SDmitry Safonov 	struct ftrace_glob func_g;
375133b7f99cSSteven Rostedt (Red Hat) 	struct ftrace_hash **orig_hash = &trace_probe_ops.func_hash->filter_hash;
37523296fc4eSSteven Rostedt (Red Hat) 	struct ftrace_hash *old_hash = *orig_hash;
3753e1df4cb6SSteven Rostedt (Red Hat) 	struct ftrace_hash *hash;
375459df055fSSteven Rostedt 	struct ftrace_page *pg;
375559df055fSSteven Rostedt 	struct dyn_ftrace *rec;
37563ba00929SDmitry Safonov 	int not;
37576a24a244SSteven Rostedt 	unsigned long key;
375859df055fSSteven Rostedt 	int count = 0;
3759e1df4cb6SSteven Rostedt (Red Hat) 	int ret;
376059df055fSSteven Rostedt 
37613ba00929SDmitry Safonov 	func_g.type = filter_parse_regex(glob, strlen(glob),
37623ba00929SDmitry Safonov 			&func_g.search, &not);
37633ba00929SDmitry Safonov 	func_g.len = strlen(func_g.search);
376459df055fSSteven Rostedt 
3765b6887d79SSteven Rostedt 	/* we do not support '!' for function probes */
376659df055fSSteven Rostedt 	if (WARN_ON(not))
376759df055fSSteven Rostedt 		return -EINVAL;
376859df055fSSteven Rostedt 
376933b7f99cSSteven Rostedt (Red Hat) 	mutex_lock(&trace_probe_ops.func_hash->regex_lock);
377059df055fSSteven Rostedt 
37717485058eSSteven Rostedt (Red Hat) 	old_hash_ops.filter_hash = old_hash;
37727485058eSSteven Rostedt (Red Hat) 	/* Probes only have filters */
37737485058eSSteven Rostedt (Red Hat) 	old_hash_ops.notrace_hash = NULL;
37747485058eSSteven Rostedt (Red Hat) 
37753296fc4eSSteven Rostedt (Red Hat) 	hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, old_hash);
3776e1df4cb6SSteven Rostedt (Red Hat) 	if (!hash) {
3777e1df4cb6SSteven Rostedt (Red Hat) 		count = -ENOMEM;
37785ae0bf59SSteven Rostedt (Red Hat) 		goto out;
3779e1df4cb6SSteven Rostedt (Red Hat) 	}
3780e1df4cb6SSteven Rostedt (Red Hat) 
3781e1df4cb6SSteven Rostedt (Red Hat) 	if (unlikely(ftrace_disabled)) {
3782e1df4cb6SSteven Rostedt (Red Hat) 		count = -ENODEV;
37835ae0bf59SSteven Rostedt (Red Hat) 		goto out;
3784e1df4cb6SSteven Rostedt (Red Hat) 	}
378545a4a237SSteven Rostedt 
37865ae0bf59SSteven Rostedt (Red Hat) 	mutex_lock(&ftrace_lock);
37875ae0bf59SSteven Rostedt (Red Hat) 
378859df055fSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
378959df055fSSteven Rostedt 
37900b507e1eSDmitry Safonov 		if (!ftrace_match_record(rec, &func_g, NULL, 0))
379159df055fSSteven Rostedt 			continue;
379259df055fSSteven Rostedt 
379359df055fSSteven Rostedt 		entry = kmalloc(sizeof(*entry), GFP_KERNEL);
379459df055fSSteven Rostedt 		if (!entry) {
3795b6887d79SSteven Rostedt 			/* If we did not process any, then return error */
379659df055fSSteven Rostedt 			if (!count)
379759df055fSSteven Rostedt 				count = -ENOMEM;
379859df055fSSteven Rostedt 			goto out_unlock;
379959df055fSSteven Rostedt 		}
380059df055fSSteven Rostedt 
380159df055fSSteven Rostedt 		count++;
380259df055fSSteven Rostedt 
380359df055fSSteven Rostedt 		entry->data = data;
380459df055fSSteven Rostedt 
380559df055fSSteven Rostedt 		/*
380659df055fSSteven Rostedt 		 * The caller might want to do something special
380759df055fSSteven Rostedt 		 * for each function we find. We call the callback
380859df055fSSteven Rostedt 		 * to give the caller an opportunity to do so.
380959df055fSSteven Rostedt 		 */
3810e67efb93SSteven Rostedt (Red Hat) 		if (ops->init) {
3811e67efb93SSteven Rostedt (Red Hat) 			if (ops->init(ops, rec->ip, &entry->data) < 0) {
381259df055fSSteven Rostedt 				/* caller does not like this func */
381359df055fSSteven Rostedt 				kfree(entry);
381459df055fSSteven Rostedt 				continue;
381559df055fSSteven Rostedt 			}
381659df055fSSteven Rostedt 		}
381759df055fSSteven Rostedt 
3818e1df4cb6SSteven Rostedt (Red Hat) 		ret = enter_record(hash, rec, 0);
3819e1df4cb6SSteven Rostedt (Red Hat) 		if (ret < 0) {
3820e1df4cb6SSteven Rostedt (Red Hat) 			kfree(entry);
3821e1df4cb6SSteven Rostedt (Red Hat) 			count = ret;
3822e1df4cb6SSteven Rostedt (Red Hat) 			goto out_unlock;
3823e1df4cb6SSteven Rostedt (Red Hat) 		}
3824e1df4cb6SSteven Rostedt (Red Hat) 
382559df055fSSteven Rostedt 		entry->ops = ops;
382659df055fSSteven Rostedt 		entry->ip = rec->ip;
382759df055fSSteven Rostedt 
382859df055fSSteven Rostedt 		key = hash_long(entry->ip, FTRACE_HASH_BITS);
382959df055fSSteven Rostedt 		hlist_add_head_rcu(&entry->node, &ftrace_func_hash[key]);
383059df055fSSteven Rostedt 
383159df055fSSteven Rostedt 	} while_for_each_ftrace_rec();
3832e1df4cb6SSteven Rostedt (Red Hat) 
3833e1df4cb6SSteven Rostedt (Red Hat) 	ret = ftrace_hash_move(&trace_probe_ops, 1, orig_hash, hash);
38348252ecf3SSteven Rostedt (Red Hat) 
38357485058eSSteven Rostedt (Red Hat) 	__enable_ftrace_function_probe(&old_hash_ops);
38368252ecf3SSteven Rostedt (Red Hat) 
38373296fc4eSSteven Rostedt (Red Hat) 	if (!ret)
38383296fc4eSSteven Rostedt (Red Hat) 		free_ftrace_hash_rcu(old_hash);
38393296fc4eSSteven Rostedt (Red Hat) 	else
3840e1df4cb6SSteven Rostedt (Red Hat) 		count = ret;
3841e1df4cb6SSteven Rostedt (Red Hat) 
384259df055fSSteven Rostedt  out_unlock:
38435ae0bf59SSteven Rostedt (Red Hat) 	mutex_unlock(&ftrace_lock);
38445ae0bf59SSteven Rostedt (Red Hat)  out:
384533b7f99cSSteven Rostedt (Red Hat) 	mutex_unlock(&trace_probe_ops.func_hash->regex_lock);
3846e1df4cb6SSteven Rostedt (Red Hat) 	free_ftrace_hash(hash);
384759df055fSSteven Rostedt 
384859df055fSSteven Rostedt 	return count;
384959df055fSSteven Rostedt }
385059df055fSSteven Rostedt 
385159df055fSSteven Rostedt enum {
3852b6887d79SSteven Rostedt 	PROBE_TEST_FUNC		= 1,
3853b6887d79SSteven Rostedt 	PROBE_TEST_DATA		= 2
385459df055fSSteven Rostedt };
385559df055fSSteven Rostedt 
385659df055fSSteven Rostedt static void
3857b6887d79SSteven Rostedt __unregister_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops,
385859df055fSSteven Rostedt 				  void *data, int flags)
385959df055fSSteven Rostedt {
3860e1df4cb6SSteven Rostedt (Red Hat) 	struct ftrace_func_entry *rec_entry;
3861b6887d79SSteven Rostedt 	struct ftrace_func_probe *entry;
38627818b388SSteven Rostedt (Red Hat) 	struct ftrace_func_probe *p;
38633ba00929SDmitry Safonov 	struct ftrace_glob func_g;
386433b7f99cSSteven Rostedt (Red Hat) 	struct ftrace_hash **orig_hash = &trace_probe_ops.func_hash->filter_hash;
38653296fc4eSSteven Rostedt (Red Hat) 	struct ftrace_hash *old_hash = *orig_hash;
38667818b388SSteven Rostedt (Red Hat) 	struct list_head free_list;
3867e1df4cb6SSteven Rostedt (Red Hat) 	struct ftrace_hash *hash;
3868b67bfe0dSSasha Levin 	struct hlist_node *tmp;
386959df055fSSteven Rostedt 	char str[KSYM_SYMBOL_LEN];
38703ba00929SDmitry Safonov 	int i, ret;
387159df055fSSteven Rostedt 
3872b36461daSAtsushi Tsuji 	if (glob && (strcmp(glob, "*") == 0 || !strlen(glob)))
38733ba00929SDmitry Safonov 		func_g.search = NULL;
3874b36461daSAtsushi Tsuji 	else if (glob) {
387559df055fSSteven Rostedt 		int not;
387659df055fSSteven Rostedt 
38773ba00929SDmitry Safonov 		func_g.type = filter_parse_regex(glob, strlen(glob),
38783ba00929SDmitry Safonov 						 &func_g.search, &not);
38793ba00929SDmitry Safonov 		func_g.len = strlen(func_g.search);
38803ba00929SDmitry Safonov 		func_g.search = glob;
388159df055fSSteven Rostedt 
3882b6887d79SSteven Rostedt 		/* we do not support '!' for function probes */
388359df055fSSteven Rostedt 		if (WARN_ON(not))
388459df055fSSteven Rostedt 			return;
388559df055fSSteven Rostedt 	}
388659df055fSSteven Rostedt 
388733b7f99cSSteven Rostedt (Red Hat) 	mutex_lock(&trace_probe_ops.func_hash->regex_lock);
3888e1df4cb6SSteven Rostedt (Red Hat) 
3889e1df4cb6SSteven Rostedt (Red Hat) 	hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, *orig_hash);
3890e1df4cb6SSteven Rostedt (Red Hat) 	if (!hash)
3891e1df4cb6SSteven Rostedt (Red Hat) 		/* Hmm, should report this somehow */
3892e1df4cb6SSteven Rostedt (Red Hat) 		goto out_unlock;
3893e1df4cb6SSteven Rostedt (Red Hat) 
38947818b388SSteven Rostedt (Red Hat) 	INIT_LIST_HEAD(&free_list);
38957818b388SSteven Rostedt (Red Hat) 
389659df055fSSteven Rostedt 	for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) {
389759df055fSSteven Rostedt 		struct hlist_head *hhd = &ftrace_func_hash[i];
389859df055fSSteven Rostedt 
3899b67bfe0dSSasha Levin 		hlist_for_each_entry_safe(entry, tmp, hhd, node) {
390059df055fSSteven Rostedt 
390159df055fSSteven Rostedt 			/* break up if statements for readability */
3902b6887d79SSteven Rostedt 			if ((flags & PROBE_TEST_FUNC) && entry->ops != ops)
390359df055fSSteven Rostedt 				continue;
390459df055fSSteven Rostedt 
3905b6887d79SSteven Rostedt 			if ((flags & PROBE_TEST_DATA) && entry->data != data)
390659df055fSSteven Rostedt 				continue;
390759df055fSSteven Rostedt 
390859df055fSSteven Rostedt 			/* do this last, since it is the most expensive */
39093ba00929SDmitry Safonov 			if (func_g.search) {
391059df055fSSteven Rostedt 				kallsyms_lookup(entry->ip, NULL, NULL,
391159df055fSSteven Rostedt 						NULL, str);
39123ba00929SDmitry Safonov 				if (!ftrace_match(str, &func_g))
391359df055fSSteven Rostedt 					continue;
391459df055fSSteven Rostedt 			}
391559df055fSSteven Rostedt 
3916e1df4cb6SSteven Rostedt (Red Hat) 			rec_entry = ftrace_lookup_ip(hash, entry->ip);
3917e1df4cb6SSteven Rostedt (Red Hat) 			/* It is possible more than one entry had this ip */
3918e1df4cb6SSteven Rostedt (Red Hat) 			if (rec_entry)
3919e1df4cb6SSteven Rostedt (Red Hat) 				free_hash_entry(hash, rec_entry);
3920e1df4cb6SSteven Rostedt (Red Hat) 
3921740466bcSSteven Rostedt (Red Hat) 			hlist_del_rcu(&entry->node);
39227818b388SSteven Rostedt (Red Hat) 			list_add(&entry->free_list, &free_list);
392359df055fSSteven Rostedt 		}
392459df055fSSteven Rostedt 	}
39253f2367baSMasami Hiramatsu 	mutex_lock(&ftrace_lock);
3926b6887d79SSteven Rostedt 	__disable_ftrace_function_probe();
3927e1df4cb6SSteven Rostedt (Red Hat) 	/*
3928e1df4cb6SSteven Rostedt (Red Hat) 	 * Remove after the disable is called. Otherwise, if the last
3929e1df4cb6SSteven Rostedt (Red Hat) 	 * probe is removed, a null hash means *all enabled*.
3930e1df4cb6SSteven Rostedt (Red Hat) 	 */
39313296fc4eSSteven Rostedt (Red Hat) 	ret = ftrace_hash_move(&trace_probe_ops, 1, orig_hash, hash);
39327818b388SSteven Rostedt (Red Hat) 	synchronize_sched();
39333296fc4eSSteven Rostedt (Red Hat) 	if (!ret)
39343296fc4eSSteven Rostedt (Red Hat) 		free_ftrace_hash_rcu(old_hash);
39353296fc4eSSteven Rostedt (Red Hat) 
39367818b388SSteven Rostedt (Red Hat) 	list_for_each_entry_safe(entry, p, &free_list, free_list) {
39377818b388SSteven Rostedt (Red Hat) 		list_del(&entry->free_list);
39387818b388SSteven Rostedt (Red Hat) 		ftrace_free_entry(entry);
39397818b388SSteven Rostedt (Red Hat) 	}
39403f2367baSMasami Hiramatsu 	mutex_unlock(&ftrace_lock);
39417818b388SSteven Rostedt (Red Hat) 
3942e1df4cb6SSteven Rostedt (Red Hat)  out_unlock:
394333b7f99cSSteven Rostedt (Red Hat) 	mutex_unlock(&trace_probe_ops.func_hash->regex_lock);
3944e1df4cb6SSteven Rostedt (Red Hat) 	free_ftrace_hash(hash);
394559df055fSSteven Rostedt }
394659df055fSSteven Rostedt 
394759df055fSSteven Rostedt void
3948b6887d79SSteven Rostedt unregister_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops,
394959df055fSSteven Rostedt 				void *data)
395059df055fSSteven Rostedt {
3951b6887d79SSteven Rostedt 	__unregister_ftrace_function_probe(glob, ops, data,
3952b6887d79SSteven Rostedt 					  PROBE_TEST_FUNC | PROBE_TEST_DATA);
395359df055fSSteven Rostedt }
395459df055fSSteven Rostedt 
395559df055fSSteven Rostedt void
3956b6887d79SSteven Rostedt unregister_ftrace_function_probe_func(char *glob, struct ftrace_probe_ops *ops)
395759df055fSSteven Rostedt {
3958b6887d79SSteven Rostedt 	__unregister_ftrace_function_probe(glob, ops, NULL, PROBE_TEST_FUNC);
395959df055fSSteven Rostedt }
396059df055fSSteven Rostedt 
3961b6887d79SSteven Rostedt void unregister_ftrace_function_probe_all(char *glob)
396259df055fSSteven Rostedt {
3963b6887d79SSteven Rostedt 	__unregister_ftrace_function_probe(glob, NULL, NULL, 0);
396459df055fSSteven Rostedt }
396559df055fSSteven Rostedt 
3966f6180773SSteven Rostedt static LIST_HEAD(ftrace_commands);
3967f6180773SSteven Rostedt static DEFINE_MUTEX(ftrace_cmd_mutex);
3968f6180773SSteven Rostedt 
396938de93abSTom Zanussi /*
397038de93abSTom Zanussi  * Currently we only register ftrace commands from __init, so mark this
397138de93abSTom Zanussi  * __init too.
397238de93abSTom Zanussi  */
397338de93abSTom Zanussi __init int register_ftrace_command(struct ftrace_func_command *cmd)
3974f6180773SSteven Rostedt {
3975f6180773SSteven Rostedt 	struct ftrace_func_command *p;
3976f6180773SSteven Rostedt 	int ret = 0;
3977f6180773SSteven Rostedt 
3978f6180773SSteven Rostedt 	mutex_lock(&ftrace_cmd_mutex);
3979f6180773SSteven Rostedt 	list_for_each_entry(p, &ftrace_commands, list) {
3980f6180773SSteven Rostedt 		if (strcmp(cmd->name, p->name) == 0) {
3981f6180773SSteven Rostedt 			ret = -EBUSY;
3982f6180773SSteven Rostedt 			goto out_unlock;
3983f6180773SSteven Rostedt 		}
3984f6180773SSteven Rostedt 	}
3985f6180773SSteven Rostedt 	list_add(&cmd->list, &ftrace_commands);
3986f6180773SSteven Rostedt  out_unlock:
3987f6180773SSteven Rostedt 	mutex_unlock(&ftrace_cmd_mutex);
3988f6180773SSteven Rostedt 
3989f6180773SSteven Rostedt 	return ret;
3990f6180773SSteven Rostedt }
3991f6180773SSteven Rostedt 
399238de93abSTom Zanussi /*
399338de93abSTom Zanussi  * Currently we only unregister ftrace commands from __init, so mark
399438de93abSTom Zanussi  * this __init too.
399538de93abSTom Zanussi  */
399638de93abSTom Zanussi __init int unregister_ftrace_command(struct ftrace_func_command *cmd)
3997f6180773SSteven Rostedt {
3998f6180773SSteven Rostedt 	struct ftrace_func_command *p, *n;
3999f6180773SSteven Rostedt 	int ret = -ENODEV;
4000f6180773SSteven Rostedt 
4001f6180773SSteven Rostedt 	mutex_lock(&ftrace_cmd_mutex);
4002f6180773SSteven Rostedt 	list_for_each_entry_safe(p, n, &ftrace_commands, list) {
4003f6180773SSteven Rostedt 		if (strcmp(cmd->name, p->name) == 0) {
4004f6180773SSteven Rostedt 			ret = 0;
4005f6180773SSteven Rostedt 			list_del_init(&p->list);
4006f6180773SSteven Rostedt 			goto out_unlock;
4007f6180773SSteven Rostedt 		}
4008f6180773SSteven Rostedt 	}
4009f6180773SSteven Rostedt  out_unlock:
4010f6180773SSteven Rostedt 	mutex_unlock(&ftrace_cmd_mutex);
4011f6180773SSteven Rostedt 
4012f6180773SSteven Rostedt 	return ret;
4013f6180773SSteven Rostedt }
4014f6180773SSteven Rostedt 
401533dc9b12SSteven Rostedt static int ftrace_process_regex(struct ftrace_hash *hash,
401633dc9b12SSteven Rostedt 				char *buff, int len, int enable)
401764e7c440SSteven Rostedt {
4018f6180773SSteven Rostedt 	char *func, *command, *next = buff;
40196a24a244SSteven Rostedt 	struct ftrace_func_command *p;
40200aff1c0cSGuoWen Li 	int ret = -EINVAL;
402164e7c440SSteven Rostedt 
402264e7c440SSteven Rostedt 	func = strsep(&next, ":");
402364e7c440SSteven Rostedt 
402464e7c440SSteven Rostedt 	if (!next) {
40251cf41dd7SSteven Rostedt 		ret = ftrace_match_records(hash, func, len);
4026b448c4e3SSteven Rostedt 		if (!ret)
4027b448c4e3SSteven Rostedt 			ret = -EINVAL;
4028b448c4e3SSteven Rostedt 		if (ret < 0)
4029311d16daSLi Zefan 			return ret;
4030b448c4e3SSteven Rostedt 		return 0;
403164e7c440SSteven Rostedt 	}
403264e7c440SSteven Rostedt 
4033f6180773SSteven Rostedt 	/* command found */
403464e7c440SSteven Rostedt 
403564e7c440SSteven Rostedt 	command = strsep(&next, ":");
403664e7c440SSteven Rostedt 
4037f6180773SSteven Rostedt 	mutex_lock(&ftrace_cmd_mutex);
4038f6180773SSteven Rostedt 	list_for_each_entry(p, &ftrace_commands, list) {
4039f6180773SSteven Rostedt 		if (strcmp(p->name, command) == 0) {
404043dd61c9SSteven Rostedt 			ret = p->func(hash, func, command, next, enable);
4041f6180773SSteven Rostedt 			goto out_unlock;
404264e7c440SSteven Rostedt 		}
4043f6180773SSteven Rostedt 	}
4044f6180773SSteven Rostedt  out_unlock:
4045f6180773SSteven Rostedt 	mutex_unlock(&ftrace_cmd_mutex);
404664e7c440SSteven Rostedt 
4047f6180773SSteven Rostedt 	return ret;
404864e7c440SSteven Rostedt }
404964e7c440SSteven Rostedt 
4050e309b41dSIngo Molnar static ssize_t
405141c52c0dSSteven Rostedt ftrace_regex_write(struct file *file, const char __user *ubuf,
405241c52c0dSSteven Rostedt 		   size_t cnt, loff_t *ppos, int enable)
40535072c59fSSteven Rostedt {
40545072c59fSSteven Rostedt 	struct ftrace_iterator *iter;
4055689fd8b6S[email protected] 	struct trace_parser *parser;
4056689fd8b6S[email protected] 	ssize_t ret, read;
40575072c59fSSteven Rostedt 
40584ba7978eSLi Zefan 	if (!cnt)
40595072c59fSSteven Rostedt 		return 0;
40605072c59fSSteven Rostedt 
40615072c59fSSteven Rostedt 	if (file->f_mode & FMODE_READ) {
40625072c59fSSteven Rostedt 		struct seq_file *m = file->private_data;
40635072c59fSSteven Rostedt 		iter = m->private;
40645072c59fSSteven Rostedt 	} else
40655072c59fSSteven Rostedt 		iter = file->private_data;
40665072c59fSSteven Rostedt 
4067f04f24fbSMasami Hiramatsu 	if (unlikely(ftrace_disabled))
40683f2367baSMasami Hiramatsu 		return -ENODEV;
40693f2367baSMasami Hiramatsu 
40703f2367baSMasami Hiramatsu 	/* iter->hash is a local copy, so we don't need regex_lock */
4071f04f24fbSMasami Hiramatsu 
4072689fd8b6S[email protected] 	parser = &iter->parser;
4073689fd8b6S[email protected] 	read = trace_get_user(parser, ubuf, cnt, ppos);
40745072c59fSSteven Rostedt 
40754ba7978eSLi Zefan 	if (read >= 0 && trace_parser_loaded(parser) &&
4076689fd8b6S[email protected] 	    !trace_parser_cont(parser)) {
407733dc9b12SSteven Rostedt 		ret = ftrace_process_regex(iter->hash, parser->buffer,
4078689fd8b6S[email protected] 					   parser->idx, enable);
4079313254a9SLi Zefan 		trace_parser_clear(parser);
40807c088b51SSteven Rostedt (Red Hat) 		if (ret < 0)
40813f2367baSMasami Hiramatsu 			goto out;
40825072c59fSSteven Rostedt 	}
40835072c59fSSteven Rostedt 
40845072c59fSSteven Rostedt 	ret = read;
40853f2367baSMasami Hiramatsu  out:
40865072c59fSSteven Rostedt 	return ret;
40875072c59fSSteven Rostedt }
40885072c59fSSteven Rostedt 
4089fc13cb0cSSteven Rostedt ssize_t
409041c52c0dSSteven Rostedt ftrace_filter_write(struct file *file, const char __user *ubuf,
409141c52c0dSSteven Rostedt 		    size_t cnt, loff_t *ppos)
409241c52c0dSSteven Rostedt {
409341c52c0dSSteven Rostedt 	return ftrace_regex_write(file, ubuf, cnt, ppos, 1);
409441c52c0dSSteven Rostedt }
409541c52c0dSSteven Rostedt 
4096fc13cb0cSSteven Rostedt ssize_t
409741c52c0dSSteven Rostedt ftrace_notrace_write(struct file *file, const char __user *ubuf,
409841c52c0dSSteven Rostedt 		     size_t cnt, loff_t *ppos)
409941c52c0dSSteven Rostedt {
410041c52c0dSSteven Rostedt 	return ftrace_regex_write(file, ubuf, cnt, ppos, 0);
410141c52c0dSSteven Rostedt }
410241c52c0dSSteven Rostedt 
410333dc9b12SSteven Rostedt static int
4104647664eaSMasami Hiramatsu ftrace_match_addr(struct ftrace_hash *hash, unsigned long ip, int remove)
4105647664eaSMasami Hiramatsu {
4106647664eaSMasami Hiramatsu 	struct ftrace_func_entry *entry;
4107647664eaSMasami Hiramatsu 
4108647664eaSMasami Hiramatsu 	if (!ftrace_location(ip))
4109647664eaSMasami Hiramatsu 		return -EINVAL;
4110647664eaSMasami Hiramatsu 
4111647664eaSMasami Hiramatsu 	if (remove) {
4112647664eaSMasami Hiramatsu 		entry = ftrace_lookup_ip(hash, ip);
4113647664eaSMasami Hiramatsu 		if (!entry)
4114647664eaSMasami Hiramatsu 			return -ENOENT;
4115647664eaSMasami Hiramatsu 		free_hash_entry(hash, entry);
4116647664eaSMasami Hiramatsu 		return 0;
4117647664eaSMasami Hiramatsu 	}
4118647664eaSMasami Hiramatsu 
4119647664eaSMasami Hiramatsu 	return add_hash_entry(hash, ip);
4120647664eaSMasami Hiramatsu }
4121647664eaSMasami Hiramatsu 
41228252ecf3SSteven Rostedt (Red Hat) static void ftrace_ops_update_code(struct ftrace_ops *ops,
41237485058eSSteven Rostedt (Red Hat) 				   struct ftrace_ops_hash *old_hash)
41241c80c432SSteven Rostedt (Red Hat) {
41258f86f837SSteven Rostedt (Red Hat) 	struct ftrace_ops *op;
41268f86f837SSteven Rostedt (Red Hat) 
41278f86f837SSteven Rostedt (Red Hat) 	if (!ftrace_enabled)
41288f86f837SSteven Rostedt (Red Hat) 		return;
41298f86f837SSteven Rostedt (Red Hat) 
41308f86f837SSteven Rostedt (Red Hat) 	if (ops->flags & FTRACE_OPS_FL_ENABLED) {
41318252ecf3SSteven Rostedt (Red Hat) 		ftrace_run_modify_code(ops, FTRACE_UPDATE_CALLS, old_hash);
41328f86f837SSteven Rostedt (Red Hat) 		return;
41338f86f837SSteven Rostedt (Red Hat) 	}
41348f86f837SSteven Rostedt (Red Hat) 
41358f86f837SSteven Rostedt (Red Hat) 	/*
41368f86f837SSteven Rostedt (Red Hat) 	 * If this is the shared global_ops filter, then we need to
41378f86f837SSteven Rostedt (Red Hat) 	 * check if there is another ops that shares it, is enabled.
41388f86f837SSteven Rostedt (Red Hat) 	 * If so, we still need to run the modify code.
41398f86f837SSteven Rostedt (Red Hat) 	 */
41408f86f837SSteven Rostedt (Red Hat) 	if (ops->func_hash != &global_ops.local_hash)
41418f86f837SSteven Rostedt (Red Hat) 		return;
41428f86f837SSteven Rostedt (Red Hat) 
41438f86f837SSteven Rostedt (Red Hat) 	do_for_each_ftrace_op(op, ftrace_ops_list) {
41448f86f837SSteven Rostedt (Red Hat) 		if (op->func_hash == &global_ops.local_hash &&
41458f86f837SSteven Rostedt (Red Hat) 		    op->flags & FTRACE_OPS_FL_ENABLED) {
41468f86f837SSteven Rostedt (Red Hat) 			ftrace_run_modify_code(op, FTRACE_UPDATE_CALLS, old_hash);
41478f86f837SSteven Rostedt (Red Hat) 			/* Only need to do this once */
41488f86f837SSteven Rostedt (Red Hat) 			return;
41498f86f837SSteven Rostedt (Red Hat) 		}
41508f86f837SSteven Rostedt (Red Hat) 	} while_for_each_ftrace_op(op);
41511c80c432SSteven Rostedt (Red Hat) }
41521c80c432SSteven Rostedt (Red Hat) 
4153647664eaSMasami Hiramatsu static int
4154647664eaSMasami Hiramatsu ftrace_set_hash(struct ftrace_ops *ops, unsigned char *buf, int len,
4155647664eaSMasami Hiramatsu 		unsigned long ip, int remove, int reset, int enable)
415641c52c0dSSteven Rostedt {
415733dc9b12SSteven Rostedt 	struct ftrace_hash **orig_hash;
41587485058eSSteven Rostedt (Red Hat) 	struct ftrace_ops_hash old_hash_ops;
41593296fc4eSSteven Rostedt (Red Hat) 	struct ftrace_hash *old_hash;
4160f45948e8SSteven Rostedt 	struct ftrace_hash *hash;
416133dc9b12SSteven Rostedt 	int ret;
4162f45948e8SSteven Rostedt 
416341c52c0dSSteven Rostedt 	if (unlikely(ftrace_disabled))
416433dc9b12SSteven Rostedt 		return -ENODEV;
416541c52c0dSSteven Rostedt 
416633b7f99cSSteven Rostedt (Red Hat) 	mutex_lock(&ops->func_hash->regex_lock);
41673f2367baSMasami Hiramatsu 
4168f45948e8SSteven Rostedt 	if (enable)
416933b7f99cSSteven Rostedt (Red Hat) 		orig_hash = &ops->func_hash->filter_hash;
4170f45948e8SSteven Rostedt 	else
417133b7f99cSSteven Rostedt (Red Hat) 		orig_hash = &ops->func_hash->notrace_hash;
417233dc9b12SSteven Rostedt 
4173b972cc58SWang Nan 	if (reset)
4174b972cc58SWang Nan 		hash = alloc_ftrace_hash(FTRACE_HASH_DEFAULT_BITS);
4175b972cc58SWang Nan 	else
417633dc9b12SSteven Rostedt 		hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, *orig_hash);
4177b972cc58SWang Nan 
41783f2367baSMasami Hiramatsu 	if (!hash) {
41793f2367baSMasami Hiramatsu 		ret = -ENOMEM;
41803f2367baSMasami Hiramatsu 		goto out_regex_unlock;
41813f2367baSMasami Hiramatsu 	}
4182f45948e8SSteven Rostedt 
4183ac483c44SJiri Olsa 	if (buf && !ftrace_match_records(hash, buf, len)) {
4184ac483c44SJiri Olsa 		ret = -EINVAL;
4185ac483c44SJiri Olsa 		goto out_regex_unlock;
4186ac483c44SJiri Olsa 	}
4187647664eaSMasami Hiramatsu 	if (ip) {
4188647664eaSMasami Hiramatsu 		ret = ftrace_match_addr(hash, ip, remove);
4189647664eaSMasami Hiramatsu 		if (ret < 0)
4190647664eaSMasami Hiramatsu 			goto out_regex_unlock;
4191647664eaSMasami Hiramatsu 	}
419233dc9b12SSteven Rostedt 
419333dc9b12SSteven Rostedt 	mutex_lock(&ftrace_lock);
41943296fc4eSSteven Rostedt (Red Hat) 	old_hash = *orig_hash;
41957485058eSSteven Rostedt (Red Hat) 	old_hash_ops.filter_hash = ops->func_hash->filter_hash;
41967485058eSSteven Rostedt (Red Hat) 	old_hash_ops.notrace_hash = ops->func_hash->notrace_hash;
419741fb61c2SSteven Rostedt 	ret = ftrace_hash_move(ops, enable, orig_hash, hash);
41983296fc4eSSteven Rostedt (Red Hat) 	if (!ret) {
41997485058eSSteven Rostedt (Red Hat) 		ftrace_ops_update_code(ops, &old_hash_ops);
42003296fc4eSSteven Rostedt (Red Hat) 		free_ftrace_hash_rcu(old_hash);
42013296fc4eSSteven Rostedt (Red Hat) 	}
420233dc9b12SSteven Rostedt 	mutex_unlock(&ftrace_lock);
420333dc9b12SSteven Rostedt 
4204ac483c44SJiri Olsa  out_regex_unlock:
420533b7f99cSSteven Rostedt (Red Hat) 	mutex_unlock(&ops->func_hash->regex_lock);
420633dc9b12SSteven Rostedt 
420733dc9b12SSteven Rostedt 	free_ftrace_hash(hash);
420833dc9b12SSteven Rostedt 	return ret;
420941c52c0dSSteven Rostedt }
421041c52c0dSSteven Rostedt 
4211647664eaSMasami Hiramatsu static int
4212647664eaSMasami Hiramatsu ftrace_set_addr(struct ftrace_ops *ops, unsigned long ip, int remove,
4213647664eaSMasami Hiramatsu 		int reset, int enable)
4214647664eaSMasami Hiramatsu {
4215647664eaSMasami Hiramatsu 	return ftrace_set_hash(ops, 0, 0, ip, remove, reset, enable);
4216647664eaSMasami Hiramatsu }
4217647664eaSMasami Hiramatsu 
4218647664eaSMasami Hiramatsu /**
4219647664eaSMasami Hiramatsu  * ftrace_set_filter_ip - set a function to filter on in ftrace by address
4220647664eaSMasami Hiramatsu  * @ops - the ops to set the filter with
4221647664eaSMasami Hiramatsu  * @ip - the address to add to or remove from the filter.
4222647664eaSMasami Hiramatsu  * @remove - non zero to remove the ip from the filter
4223647664eaSMasami Hiramatsu  * @reset - non zero to reset all filters before applying this filter.
4224647664eaSMasami Hiramatsu  *
4225647664eaSMasami Hiramatsu  * Filters denote which functions should be enabled when tracing is enabled
4226647664eaSMasami Hiramatsu  * If @ip is NULL, it failes to update filter.
4227647664eaSMasami Hiramatsu  */
4228647664eaSMasami Hiramatsu int ftrace_set_filter_ip(struct ftrace_ops *ops, unsigned long ip,
4229647664eaSMasami Hiramatsu 			 int remove, int reset)
4230647664eaSMasami Hiramatsu {
4231f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
4232647664eaSMasami Hiramatsu 	return ftrace_set_addr(ops, ip, remove, reset, 1);
4233647664eaSMasami Hiramatsu }
4234647664eaSMasami Hiramatsu EXPORT_SYMBOL_GPL(ftrace_set_filter_ip);
4235647664eaSMasami Hiramatsu 
4236647664eaSMasami Hiramatsu static int
4237647664eaSMasami Hiramatsu ftrace_set_regex(struct ftrace_ops *ops, unsigned char *buf, int len,
4238647664eaSMasami Hiramatsu 		 int reset, int enable)
4239647664eaSMasami Hiramatsu {
4240647664eaSMasami Hiramatsu 	return ftrace_set_hash(ops, buf, len, 0, 0, reset, enable);
4241647664eaSMasami Hiramatsu }
4242647664eaSMasami Hiramatsu 
424377a2b37dSSteven Rostedt /**
424477a2b37dSSteven Rostedt  * ftrace_set_filter - set a function to filter on in ftrace
4245936e074bSSteven Rostedt  * @ops - the ops to set the filter with
424677a2b37dSSteven Rostedt  * @buf - the string that holds the function filter text.
424777a2b37dSSteven Rostedt  * @len - the length of the string.
424877a2b37dSSteven Rostedt  * @reset - non zero to reset all filters before applying this filter.
424977a2b37dSSteven Rostedt  *
425077a2b37dSSteven Rostedt  * Filters denote which functions should be enabled when tracing is enabled.
425177a2b37dSSteven Rostedt  * If @buf is NULL and reset is set, all functions will be enabled for tracing.
425277a2b37dSSteven Rostedt  */
4253ac483c44SJiri Olsa int ftrace_set_filter(struct ftrace_ops *ops, unsigned char *buf,
4254936e074bSSteven Rostedt 		       int len, int reset)
425577a2b37dSSteven Rostedt {
4256f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
4257ac483c44SJiri Olsa 	return ftrace_set_regex(ops, buf, len, reset, 1);
425841c52c0dSSteven Rostedt }
4259936e074bSSteven Rostedt EXPORT_SYMBOL_GPL(ftrace_set_filter);
42604eebcc81SSteven Rostedt 
426141c52c0dSSteven Rostedt /**
426241c52c0dSSteven Rostedt  * ftrace_set_notrace - set a function to not trace in ftrace
4263936e074bSSteven Rostedt  * @ops - the ops to set the notrace filter with
426441c52c0dSSteven Rostedt  * @buf - the string that holds the function notrace text.
426541c52c0dSSteven Rostedt  * @len - the length of the string.
426641c52c0dSSteven Rostedt  * @reset - non zero to reset all filters before applying this filter.
426741c52c0dSSteven Rostedt  *
426841c52c0dSSteven Rostedt  * Notrace Filters denote which functions should not be enabled when tracing
426941c52c0dSSteven Rostedt  * is enabled. If @buf is NULL and reset is set, all functions will be enabled
427041c52c0dSSteven Rostedt  * for tracing.
427141c52c0dSSteven Rostedt  */
4272ac483c44SJiri Olsa int ftrace_set_notrace(struct ftrace_ops *ops, unsigned char *buf,
4273936e074bSSteven Rostedt 			int len, int reset)
4274936e074bSSteven Rostedt {
4275f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
4276ac483c44SJiri Olsa 	return ftrace_set_regex(ops, buf, len, reset, 0);
4277936e074bSSteven Rostedt }
4278936e074bSSteven Rostedt EXPORT_SYMBOL_GPL(ftrace_set_notrace);
4279936e074bSSteven Rostedt /**
42808d1b065dSJiaxing Wang  * ftrace_set_global_filter - set a function to filter on with global tracers
4281936e074bSSteven Rostedt  * @buf - the string that holds the function filter text.
4282936e074bSSteven Rostedt  * @len - the length of the string.
4283936e074bSSteven Rostedt  * @reset - non zero to reset all filters before applying this filter.
4284936e074bSSteven Rostedt  *
4285936e074bSSteven Rostedt  * Filters denote which functions should be enabled when tracing is enabled.
4286936e074bSSteven Rostedt  * If @buf is NULL and reset is set, all functions will be enabled for tracing.
4287936e074bSSteven Rostedt  */
4288936e074bSSteven Rostedt void ftrace_set_global_filter(unsigned char *buf, int len, int reset)
4289936e074bSSteven Rostedt {
4290936e074bSSteven Rostedt 	ftrace_set_regex(&global_ops, buf, len, reset, 1);
4291936e074bSSteven Rostedt }
4292936e074bSSteven Rostedt EXPORT_SYMBOL_GPL(ftrace_set_global_filter);
4293936e074bSSteven Rostedt 
4294936e074bSSteven Rostedt /**
42958d1b065dSJiaxing Wang  * ftrace_set_global_notrace - set a function to not trace with global tracers
4296936e074bSSteven Rostedt  * @buf - the string that holds the function notrace text.
4297936e074bSSteven Rostedt  * @len - the length of the string.
4298936e074bSSteven Rostedt  * @reset - non zero to reset all filters before applying this filter.
4299936e074bSSteven Rostedt  *
4300936e074bSSteven Rostedt  * Notrace Filters denote which functions should not be enabled when tracing
4301936e074bSSteven Rostedt  * is enabled. If @buf is NULL and reset is set, all functions will be enabled
4302936e074bSSteven Rostedt  * for tracing.
4303936e074bSSteven Rostedt  */
4304936e074bSSteven Rostedt void ftrace_set_global_notrace(unsigned char *buf, int len, int reset)
430541c52c0dSSteven Rostedt {
4306f45948e8SSteven Rostedt 	ftrace_set_regex(&global_ops, buf, len, reset, 0);
430777a2b37dSSteven Rostedt }
4308936e074bSSteven Rostedt EXPORT_SYMBOL_GPL(ftrace_set_global_notrace);
430977a2b37dSSteven Rostedt 
43102af15d6aSSteven Rostedt /*
43112af15d6aSSteven Rostedt  * command line interface to allow users to set filters on boot up.
43122af15d6aSSteven Rostedt  */
43132af15d6aSSteven Rostedt #define FTRACE_FILTER_SIZE		COMMAND_LINE_SIZE
43142af15d6aSSteven Rostedt static char ftrace_notrace_buf[FTRACE_FILTER_SIZE] __initdata;
43152af15d6aSSteven Rostedt static char ftrace_filter_buf[FTRACE_FILTER_SIZE] __initdata;
43162af15d6aSSteven Rostedt 
4317f1ed7c74SSteven Rostedt (Red Hat) /* Used by function selftest to not test if filter is set */
4318f1ed7c74SSteven Rostedt (Red Hat) bool ftrace_filter_param __initdata;
4319f1ed7c74SSteven Rostedt (Red Hat) 
43202af15d6aSSteven Rostedt static int __init set_ftrace_notrace(char *str)
43212af15d6aSSteven Rostedt {
4322f1ed7c74SSteven Rostedt (Red Hat) 	ftrace_filter_param = true;
432375761cc1SChen Gang 	strlcpy(ftrace_notrace_buf, str, FTRACE_FILTER_SIZE);
43242af15d6aSSteven Rostedt 	return 1;
43252af15d6aSSteven Rostedt }
43262af15d6aSSteven Rostedt __setup("ftrace_notrace=", set_ftrace_notrace);
43272af15d6aSSteven Rostedt 
43282af15d6aSSteven Rostedt static int __init set_ftrace_filter(char *str)
43292af15d6aSSteven Rostedt {
4330f1ed7c74SSteven Rostedt (Red Hat) 	ftrace_filter_param = true;
433175761cc1SChen Gang 	strlcpy(ftrace_filter_buf, str, FTRACE_FILTER_SIZE);
43322af15d6aSSteven Rostedt 	return 1;
43332af15d6aSSteven Rostedt }
43342af15d6aSSteven Rostedt __setup("ftrace_filter=", set_ftrace_filter);
43352af15d6aSSteven Rostedt 
4336369bc18fSStefan Assmann #ifdef CONFIG_FUNCTION_GRAPH_TRACER
4337f6060f46SLai Jiangshan static char ftrace_graph_buf[FTRACE_FILTER_SIZE] __initdata;
43380d7d9a16SNamhyung Kim static char ftrace_graph_notrace_buf[FTRACE_FILTER_SIZE] __initdata;
4339faf982a6SNamhyung Kim static int ftrace_set_func(unsigned long *array, int *idx, int size, char *buffer);
4340801c29fdSSteven Rostedt 
4341f3bea491SSteven Rostedt (Red Hat) static unsigned long save_global_trampoline;
4342f3bea491SSteven Rostedt (Red Hat) static unsigned long save_global_flags;
4343f3bea491SSteven Rostedt (Red Hat) 
4344369bc18fSStefan Assmann static int __init set_graph_function(char *str)
4345369bc18fSStefan Assmann {
434606f43d66SFrederic Weisbecker 	strlcpy(ftrace_graph_buf, str, FTRACE_FILTER_SIZE);
4347369bc18fSStefan Assmann 	return 1;
4348369bc18fSStefan Assmann }
4349369bc18fSStefan Assmann __setup("ftrace_graph_filter=", set_graph_function);
4350369bc18fSStefan Assmann 
43510d7d9a16SNamhyung Kim static int __init set_graph_notrace_function(char *str)
43520d7d9a16SNamhyung Kim {
43530d7d9a16SNamhyung Kim 	strlcpy(ftrace_graph_notrace_buf, str, FTRACE_FILTER_SIZE);
43540d7d9a16SNamhyung Kim 	return 1;
43550d7d9a16SNamhyung Kim }
43560d7d9a16SNamhyung Kim __setup("ftrace_graph_notrace=", set_graph_notrace_function);
43570d7d9a16SNamhyung Kim 
43580d7d9a16SNamhyung Kim static void __init set_ftrace_early_graph(char *buf, int enable)
4359369bc18fSStefan Assmann {
4360369bc18fSStefan Assmann 	int ret;
4361369bc18fSStefan Assmann 	char *func;
43620d7d9a16SNamhyung Kim 	unsigned long *table = ftrace_graph_funcs;
43630d7d9a16SNamhyung Kim 	int *count = &ftrace_graph_count;
43640d7d9a16SNamhyung Kim 
43650d7d9a16SNamhyung Kim 	if (!enable) {
43660d7d9a16SNamhyung Kim 		table = ftrace_graph_notrace_funcs;
43670d7d9a16SNamhyung Kim 		count = &ftrace_graph_notrace_count;
43680d7d9a16SNamhyung Kim 	}
4369369bc18fSStefan Assmann 
4370369bc18fSStefan Assmann 	while (buf) {
4371369bc18fSStefan Assmann 		func = strsep(&buf, ",");
4372369bc18fSStefan Assmann 		/* we allow only one expression at a time */
43730d7d9a16SNamhyung Kim 		ret = ftrace_set_func(table, count, FTRACE_GRAPH_MAX_FUNCS, func);
4374369bc18fSStefan Assmann 		if (ret)
4375369bc18fSStefan Assmann 			printk(KERN_DEBUG "ftrace: function %s not "
4376369bc18fSStefan Assmann 					  "traceable\n", func);
4377369bc18fSStefan Assmann 	}
4378369bc18fSStefan Assmann }
4379369bc18fSStefan Assmann #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
4380369bc18fSStefan Assmann 
43812a85a37fSSteven Rostedt void __init
43822a85a37fSSteven Rostedt ftrace_set_early_filter(struct ftrace_ops *ops, char *buf, int enable)
43832af15d6aSSteven Rostedt {
43842af15d6aSSteven Rostedt 	char *func;
43852af15d6aSSteven Rostedt 
4386f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
4387f04f24fbSMasami Hiramatsu 
43882af15d6aSSteven Rostedt 	while (buf) {
43892af15d6aSSteven Rostedt 		func = strsep(&buf, ",");
4390f45948e8SSteven Rostedt 		ftrace_set_regex(ops, func, strlen(func), 0, enable);
43912af15d6aSSteven Rostedt 	}
43922af15d6aSSteven Rostedt }
43932af15d6aSSteven Rostedt 
43942af15d6aSSteven Rostedt static void __init set_ftrace_early_filters(void)
43952af15d6aSSteven Rostedt {
43962af15d6aSSteven Rostedt 	if (ftrace_filter_buf[0])
43972a85a37fSSteven Rostedt 		ftrace_set_early_filter(&global_ops, ftrace_filter_buf, 1);
43982af15d6aSSteven Rostedt 	if (ftrace_notrace_buf[0])
43992a85a37fSSteven Rostedt 		ftrace_set_early_filter(&global_ops, ftrace_notrace_buf, 0);
4400369bc18fSStefan Assmann #ifdef CONFIG_FUNCTION_GRAPH_TRACER
4401369bc18fSStefan Assmann 	if (ftrace_graph_buf[0])
44020d7d9a16SNamhyung Kim 		set_ftrace_early_graph(ftrace_graph_buf, 1);
44030d7d9a16SNamhyung Kim 	if (ftrace_graph_notrace_buf[0])
44040d7d9a16SNamhyung Kim 		set_ftrace_early_graph(ftrace_graph_notrace_buf, 0);
4405369bc18fSStefan Assmann #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
44062af15d6aSSteven Rostedt }
44072af15d6aSSteven Rostedt 
4408fc13cb0cSSteven Rostedt int ftrace_regex_release(struct inode *inode, struct file *file)
44095072c59fSSteven Rostedt {
44105072c59fSSteven Rostedt 	struct seq_file *m = (struct seq_file *)file->private_data;
44117485058eSSteven Rostedt (Red Hat) 	struct ftrace_ops_hash old_hash_ops;
44125072c59fSSteven Rostedt 	struct ftrace_iterator *iter;
441333dc9b12SSteven Rostedt 	struct ftrace_hash **orig_hash;
44143296fc4eSSteven Rostedt (Red Hat) 	struct ftrace_hash *old_hash;
4415689fd8b6S[email protected] 	struct trace_parser *parser;
4416ed926f9bSSteven Rostedt 	int filter_hash;
441733dc9b12SSteven Rostedt 	int ret;
44185072c59fSSteven Rostedt 
44195072c59fSSteven Rostedt 	if (file->f_mode & FMODE_READ) {
44205072c59fSSteven Rostedt 		iter = m->private;
44215072c59fSSteven Rostedt 		seq_release(inode, file);
44225072c59fSSteven Rostedt 	} else
44235072c59fSSteven Rostedt 		iter = file->private_data;
44245072c59fSSteven Rostedt 
4425689fd8b6S[email protected] 	parser = &iter->parser;
4426689fd8b6S[email protected] 	if (trace_parser_loaded(parser)) {
4427689fd8b6S[email protected] 		parser->buffer[parser->idx] = 0;
44281cf41dd7SSteven Rostedt 		ftrace_match_records(iter->hash, parser->buffer, parser->idx);
44295072c59fSSteven Rostedt 	}
44305072c59fSSteven Rostedt 
4431058e297dSSteven Rostedt 	trace_parser_put(parser);
4432058e297dSSteven Rostedt 
443333b7f99cSSteven Rostedt (Red Hat) 	mutex_lock(&iter->ops->func_hash->regex_lock);
44343f2367baSMasami Hiramatsu 
4435058e297dSSteven Rostedt 	if (file->f_mode & FMODE_WRITE) {
4436ed926f9bSSteven Rostedt 		filter_hash = !!(iter->flags & FTRACE_ITER_FILTER);
4437ed926f9bSSteven Rostedt 
4438ed926f9bSSteven Rostedt 		if (filter_hash)
443933b7f99cSSteven Rostedt (Red Hat) 			orig_hash = &iter->ops->func_hash->filter_hash;
4440ed926f9bSSteven Rostedt 		else
444133b7f99cSSteven Rostedt (Red Hat) 			orig_hash = &iter->ops->func_hash->notrace_hash;
444233dc9b12SSteven Rostedt 
4443e6ea44e9SSteven Rostedt 		mutex_lock(&ftrace_lock);
44443296fc4eSSteven Rostedt (Red Hat) 		old_hash = *orig_hash;
44457485058eSSteven Rostedt (Red Hat) 		old_hash_ops.filter_hash = iter->ops->func_hash->filter_hash;
44467485058eSSteven Rostedt (Red Hat) 		old_hash_ops.notrace_hash = iter->ops->func_hash->notrace_hash;
444741fb61c2SSteven Rostedt 		ret = ftrace_hash_move(iter->ops, filter_hash,
444841fb61c2SSteven Rostedt 				       orig_hash, iter->hash);
44493296fc4eSSteven Rostedt (Red Hat) 		if (!ret) {
44507485058eSSteven Rostedt (Red Hat) 			ftrace_ops_update_code(iter->ops, &old_hash_ops);
44513296fc4eSSteven Rostedt (Red Hat) 			free_ftrace_hash_rcu(old_hash);
44523296fc4eSSteven Rostedt (Red Hat) 		}
4453e6ea44e9SSteven Rostedt 		mutex_unlock(&ftrace_lock);
4454058e297dSSteven Rostedt 	}
44553f2367baSMasami Hiramatsu 
445633b7f99cSSteven Rostedt (Red Hat) 	mutex_unlock(&iter->ops->func_hash->regex_lock);
445733dc9b12SSteven Rostedt 	free_ftrace_hash(iter->hash);
445833dc9b12SSteven Rostedt 	kfree(iter);
4459689fd8b6S[email protected] 
44605072c59fSSteven Rostedt 	return 0;
44615072c59fSSteven Rostedt }
44625072c59fSSteven Rostedt 
44635e2336a0SSteven Rostedt static const struct file_operations ftrace_avail_fops = {
44645072c59fSSteven Rostedt 	.open = ftrace_avail_open,
44655072c59fSSteven Rostedt 	.read = seq_read,
44665072c59fSSteven Rostedt 	.llseek = seq_lseek,
44673be04b47SLi Zefan 	.release = seq_release_private,
44685072c59fSSteven Rostedt };
44695072c59fSSteven Rostedt 
4470647bcd03SSteven Rostedt static const struct file_operations ftrace_enabled_fops = {
4471647bcd03SSteven Rostedt 	.open = ftrace_enabled_open,
4472647bcd03SSteven Rostedt 	.read = seq_read,
4473647bcd03SSteven Rostedt 	.llseek = seq_lseek,
4474647bcd03SSteven Rostedt 	.release = seq_release_private,
4475647bcd03SSteven Rostedt };
4476647bcd03SSteven Rostedt 
44775e2336a0SSteven Rostedt static const struct file_operations ftrace_filter_fops = {
44785072c59fSSteven Rostedt 	.open = ftrace_filter_open,
4479850a80cfSLai Jiangshan 	.read = seq_read,
44805072c59fSSteven Rostedt 	.write = ftrace_filter_write,
4481098c879eSSteven Rostedt (Red Hat) 	.llseek = tracing_lseek,
44821cf41dd7SSteven Rostedt 	.release = ftrace_regex_release,
44835072c59fSSteven Rostedt };
44845072c59fSSteven Rostedt 
44855e2336a0SSteven Rostedt static const struct file_operations ftrace_notrace_fops = {
448641c52c0dSSteven Rostedt 	.open = ftrace_notrace_open,
4487850a80cfSLai Jiangshan 	.read = seq_read,
448841c52c0dSSteven Rostedt 	.write = ftrace_notrace_write,
4489098c879eSSteven Rostedt (Red Hat) 	.llseek = tracing_lseek,
44901cf41dd7SSteven Rostedt 	.release = ftrace_regex_release,
449141c52c0dSSteven Rostedt };
449241c52c0dSSteven Rostedt 
4493ea4e2bc4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
4494ea4e2bc4SSteven Rostedt 
4495ea4e2bc4SSteven Rostedt static DEFINE_MUTEX(graph_lock);
4496ea4e2bc4SSteven Rostedt 
4497ea4e2bc4SSteven Rostedt int ftrace_graph_count;
449829ad23b0SNamhyung Kim int ftrace_graph_notrace_count;
4499ea4e2bc4SSteven Rostedt unsigned long ftrace_graph_funcs[FTRACE_GRAPH_MAX_FUNCS] __read_mostly;
450029ad23b0SNamhyung Kim unsigned long ftrace_graph_notrace_funcs[FTRACE_GRAPH_MAX_FUNCS] __read_mostly;
4501ea4e2bc4SSteven Rostedt 
4502faf982a6SNamhyung Kim struct ftrace_graph_data {
4503faf982a6SNamhyung Kim 	unsigned long *table;
4504faf982a6SNamhyung Kim 	size_t size;
4505faf982a6SNamhyung Kim 	int *count;
4506faf982a6SNamhyung Kim 	const struct seq_operations *seq_ops;
4507faf982a6SNamhyung Kim };
4508faf982a6SNamhyung Kim 
4509ea4e2bc4SSteven Rostedt static void *
451085951842SLi Zefan __g_next(struct seq_file *m, loff_t *pos)
4511ea4e2bc4SSteven Rostedt {
4512faf982a6SNamhyung Kim 	struct ftrace_graph_data *fgd = m->private;
4513faf982a6SNamhyung Kim 
4514faf982a6SNamhyung Kim 	if (*pos >= *fgd->count)
4515ea4e2bc4SSteven Rostedt 		return NULL;
4516faf982a6SNamhyung Kim 	return &fgd->table[*pos];
451785951842SLi Zefan }
4518ea4e2bc4SSteven Rostedt 
451985951842SLi Zefan static void *
452085951842SLi Zefan g_next(struct seq_file *m, void *v, loff_t *pos)
452185951842SLi Zefan {
452285951842SLi Zefan 	(*pos)++;
452385951842SLi Zefan 	return __g_next(m, pos);
4524ea4e2bc4SSteven Rostedt }
4525ea4e2bc4SSteven Rostedt 
4526ea4e2bc4SSteven Rostedt static void *g_start(struct seq_file *m, loff_t *pos)
4527ea4e2bc4SSteven Rostedt {
4528faf982a6SNamhyung Kim 	struct ftrace_graph_data *fgd = m->private;
4529faf982a6SNamhyung Kim 
4530ea4e2bc4SSteven Rostedt 	mutex_lock(&graph_lock);
4531ea4e2bc4SSteven Rostedt 
4532f9349a8fSFrederic Weisbecker 	/* Nothing, tell g_show to print all functions are enabled */
4533faf982a6SNamhyung Kim 	if (!*fgd->count && !*pos)
4534f9349a8fSFrederic Weisbecker 		return (void *)1;
4535f9349a8fSFrederic Weisbecker 
453685951842SLi Zefan 	return __g_next(m, pos);
4537ea4e2bc4SSteven Rostedt }
4538ea4e2bc4SSteven Rostedt 
4539ea4e2bc4SSteven Rostedt static void g_stop(struct seq_file *m, void *p)
4540ea4e2bc4SSteven Rostedt {
4541ea4e2bc4SSteven Rostedt 	mutex_unlock(&graph_lock);
4542ea4e2bc4SSteven Rostedt }
4543ea4e2bc4SSteven Rostedt 
4544ea4e2bc4SSteven Rostedt static int g_show(struct seq_file *m, void *v)
4545ea4e2bc4SSteven Rostedt {
4546ea4e2bc4SSteven Rostedt 	unsigned long *ptr = v;
4547ea4e2bc4SSteven Rostedt 
4548ea4e2bc4SSteven Rostedt 	if (!ptr)
4549ea4e2bc4SSteven Rostedt 		return 0;
4550ea4e2bc4SSteven Rostedt 
4551f9349a8fSFrederic Weisbecker 	if (ptr == (unsigned long *)1) {
4552280d1429SNamhyung Kim 		struct ftrace_graph_data *fgd = m->private;
4553280d1429SNamhyung Kim 
4554280d1429SNamhyung Kim 		if (fgd->table == ftrace_graph_funcs)
4555fa6f0cc7SRasmus Villemoes 			seq_puts(m, "#### all functions enabled ####\n");
4556280d1429SNamhyung Kim 		else
4557fa6f0cc7SRasmus Villemoes 			seq_puts(m, "#### no functions disabled ####\n");
4558f9349a8fSFrederic Weisbecker 		return 0;
4559f9349a8fSFrederic Weisbecker 	}
4560f9349a8fSFrederic Weisbecker 
4561b375a11aSSteven Rostedt 	seq_printf(m, "%ps\n", (void *)*ptr);
4562ea4e2bc4SSteven Rostedt 
4563ea4e2bc4SSteven Rostedt 	return 0;
4564ea4e2bc4SSteven Rostedt }
4565ea4e2bc4SSteven Rostedt 
456688e9d34cSJames Morris static const struct seq_operations ftrace_graph_seq_ops = {
4567ea4e2bc4SSteven Rostedt 	.start = g_start,
4568ea4e2bc4SSteven Rostedt 	.next = g_next,
4569ea4e2bc4SSteven Rostedt 	.stop = g_stop,
4570ea4e2bc4SSteven Rostedt 	.show = g_show,
4571ea4e2bc4SSteven Rostedt };
4572ea4e2bc4SSteven Rostedt 
4573ea4e2bc4SSteven Rostedt static int
4574faf982a6SNamhyung Kim __ftrace_graph_open(struct inode *inode, struct file *file,
4575faf982a6SNamhyung Kim 		    struct ftrace_graph_data *fgd)
4576ea4e2bc4SSteven Rostedt {
4577ea4e2bc4SSteven Rostedt 	int ret = 0;
4578ea4e2bc4SSteven Rostedt 
4579ea4e2bc4SSteven Rostedt 	mutex_lock(&graph_lock);
4580ea4e2bc4SSteven Rostedt 	if ((file->f_mode & FMODE_WRITE) &&
45818650ae32SSteven Rostedt 	    (file->f_flags & O_TRUNC)) {
4582faf982a6SNamhyung Kim 		*fgd->count = 0;
4583faf982a6SNamhyung Kim 		memset(fgd->table, 0, fgd->size * sizeof(*fgd->table));
4584ea4e2bc4SSteven Rostedt 	}
4585ea4e2bc4SSteven Rostedt 	mutex_unlock(&graph_lock);
4586ea4e2bc4SSteven Rostedt 
4587faf982a6SNamhyung Kim 	if (file->f_mode & FMODE_READ) {
4588faf982a6SNamhyung Kim 		ret = seq_open(file, fgd->seq_ops);
4589faf982a6SNamhyung Kim 		if (!ret) {
4590faf982a6SNamhyung Kim 			struct seq_file *m = file->private_data;
4591faf982a6SNamhyung Kim 			m->private = fgd;
4592faf982a6SNamhyung Kim 		}
4593faf982a6SNamhyung Kim 	} else
4594faf982a6SNamhyung Kim 		file->private_data = fgd;
4595a4ec5e0cSLi Zefan 
4596ea4e2bc4SSteven Rostedt 	return ret;
4597ea4e2bc4SSteven Rostedt }
4598ea4e2bc4SSteven Rostedt 
4599ea4e2bc4SSteven Rostedt static int
4600faf982a6SNamhyung Kim ftrace_graph_open(struct inode *inode, struct file *file)
4601faf982a6SNamhyung Kim {
4602faf982a6SNamhyung Kim 	struct ftrace_graph_data *fgd;
4603faf982a6SNamhyung Kim 
4604faf982a6SNamhyung Kim 	if (unlikely(ftrace_disabled))
4605faf982a6SNamhyung Kim 		return -ENODEV;
4606faf982a6SNamhyung Kim 
4607faf982a6SNamhyung Kim 	fgd = kmalloc(sizeof(*fgd), GFP_KERNEL);
4608faf982a6SNamhyung Kim 	if (fgd == NULL)
4609faf982a6SNamhyung Kim 		return -ENOMEM;
4610faf982a6SNamhyung Kim 
4611faf982a6SNamhyung Kim 	fgd->table = ftrace_graph_funcs;
4612faf982a6SNamhyung Kim 	fgd->size = FTRACE_GRAPH_MAX_FUNCS;
4613faf982a6SNamhyung Kim 	fgd->count = &ftrace_graph_count;
4614faf982a6SNamhyung Kim 	fgd->seq_ops = &ftrace_graph_seq_ops;
4615faf982a6SNamhyung Kim 
4616faf982a6SNamhyung Kim 	return __ftrace_graph_open(inode, file, fgd);
4617faf982a6SNamhyung Kim }
4618faf982a6SNamhyung Kim 
4619faf982a6SNamhyung Kim static int
462029ad23b0SNamhyung Kim ftrace_graph_notrace_open(struct inode *inode, struct file *file)
462129ad23b0SNamhyung Kim {
462229ad23b0SNamhyung Kim 	struct ftrace_graph_data *fgd;
462329ad23b0SNamhyung Kim 
462429ad23b0SNamhyung Kim 	if (unlikely(ftrace_disabled))
462529ad23b0SNamhyung Kim 		return -ENODEV;
462629ad23b0SNamhyung Kim 
462729ad23b0SNamhyung Kim 	fgd = kmalloc(sizeof(*fgd), GFP_KERNEL);
462829ad23b0SNamhyung Kim 	if (fgd == NULL)
462929ad23b0SNamhyung Kim 		return -ENOMEM;
463029ad23b0SNamhyung Kim 
463129ad23b0SNamhyung Kim 	fgd->table = ftrace_graph_notrace_funcs;
463229ad23b0SNamhyung Kim 	fgd->size = FTRACE_GRAPH_MAX_FUNCS;
463329ad23b0SNamhyung Kim 	fgd->count = &ftrace_graph_notrace_count;
463429ad23b0SNamhyung Kim 	fgd->seq_ops = &ftrace_graph_seq_ops;
463529ad23b0SNamhyung Kim 
463629ad23b0SNamhyung Kim 	return __ftrace_graph_open(inode, file, fgd);
463729ad23b0SNamhyung Kim }
463829ad23b0SNamhyung Kim 
463929ad23b0SNamhyung Kim static int
464087827111SLi Zefan ftrace_graph_release(struct inode *inode, struct file *file)
464187827111SLi Zefan {
4642faf982a6SNamhyung Kim 	if (file->f_mode & FMODE_READ) {
4643faf982a6SNamhyung Kim 		struct seq_file *m = file->private_data;
4644faf982a6SNamhyung Kim 
4645faf982a6SNamhyung Kim 		kfree(m->private);
464687827111SLi Zefan 		seq_release(inode, file);
4647faf982a6SNamhyung Kim 	} else {
4648faf982a6SNamhyung Kim 		kfree(file->private_data);
4649faf982a6SNamhyung Kim 	}
4650faf982a6SNamhyung Kim 
465187827111SLi Zefan 	return 0;
465287827111SLi Zefan }
465387827111SLi Zefan 
465487827111SLi Zefan static int
4655faf982a6SNamhyung Kim ftrace_set_func(unsigned long *array, int *idx, int size, char *buffer)
4656ea4e2bc4SSteven Rostedt {
46573ba00929SDmitry Safonov 	struct ftrace_glob func_g;
4658ea4e2bc4SSteven Rostedt 	struct dyn_ftrace *rec;
4659ea4e2bc4SSteven Rostedt 	struct ftrace_page *pg;
4660c7c6b1feSLi Zefan 	int fail = 1;
46613ba00929SDmitry Safonov 	int not;
4662f9349a8fSFrederic Weisbecker 	bool exists;
4663f9349a8fSFrederic Weisbecker 	int i;
4664ea4e2bc4SSteven Rostedt 
4665f9349a8fSFrederic Weisbecker 	/* decode regex */
46663ba00929SDmitry Safonov 	func_g.type = filter_parse_regex(buffer, strlen(buffer),
46673ba00929SDmitry Safonov 					 &func_g.search, &not);
4668faf982a6SNamhyung Kim 	if (!not && *idx >= size)
4669c7c6b1feSLi Zefan 		return -EBUSY;
4670f9349a8fSFrederic Weisbecker 
46713ba00929SDmitry Safonov 	func_g.len = strlen(func_g.search);
4672f9349a8fSFrederic Weisbecker 
467352baf119SSteven Rostedt 	mutex_lock(&ftrace_lock);
467445a4a237SSteven Rostedt 
467545a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled)) {
467645a4a237SSteven Rostedt 		mutex_unlock(&ftrace_lock);
467745a4a237SSteven Rostedt 		return -ENODEV;
467845a4a237SSteven Rostedt 	}
467945a4a237SSteven Rostedt 
4680265c831cSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
4681ea4e2bc4SSteven Rostedt 
46820b507e1eSDmitry Safonov 		if (ftrace_match_record(rec, &func_g, NULL, 0)) {
4683c7c6b1feSLi Zefan 			/* if it is in the array */
4684f9349a8fSFrederic Weisbecker 			exists = false;
4685c7c6b1feSLi Zefan 			for (i = 0; i < *idx; i++) {
4686f9349a8fSFrederic Weisbecker 				if (array[i] == rec->ip) {
4687f9349a8fSFrederic Weisbecker 					exists = true;
4688faec2ec5SLiming Wang 					break;
4689faec2ec5SLiming Wang 				}
4690c7c6b1feSLi Zefan 			}
4691c7c6b1feSLi Zefan 
4692c7c6b1feSLi Zefan 			if (!not) {
4693c7c6b1feSLi Zefan 				fail = 0;
4694c7c6b1feSLi Zefan 				if (!exists) {
4695f9349a8fSFrederic Weisbecker 					array[(*idx)++] = rec->ip;
4696faf982a6SNamhyung Kim 					if (*idx >= size)
4697c7c6b1feSLi Zefan 						goto out;
4698c7c6b1feSLi Zefan 				}
4699c7c6b1feSLi Zefan 			} else {
4700c7c6b1feSLi Zefan 				if (exists) {
4701c7c6b1feSLi Zefan 					array[i] = array[--(*idx)];
4702c7c6b1feSLi Zefan 					array[*idx] = 0;
4703c7c6b1feSLi Zefan 					fail = 0;
4704c7c6b1feSLi Zefan 				}
4705c7c6b1feSLi Zefan 			}
4706f9349a8fSFrederic Weisbecker 		}
4707265c831cSSteven Rostedt 	} while_for_each_ftrace_rec();
4708c7c6b1feSLi Zefan out:
470952baf119SSteven Rostedt 	mutex_unlock(&ftrace_lock);
4710ea4e2bc4SSteven Rostedt 
4711c7c6b1feSLi Zefan 	if (fail)
4712c7c6b1feSLi Zefan 		return -EINVAL;
4713c7c6b1feSLi Zefan 
4714c7c6b1feSLi Zefan 	return 0;
4715ea4e2bc4SSteven Rostedt }
4716ea4e2bc4SSteven Rostedt 
4717ea4e2bc4SSteven Rostedt static ssize_t
4718ea4e2bc4SSteven Rostedt ftrace_graph_write(struct file *file, const char __user *ubuf,
4719ea4e2bc4SSteven Rostedt 		   size_t cnt, loff_t *ppos)
4720ea4e2bc4SSteven Rostedt {
4721689fd8b6S[email protected] 	struct trace_parser parser;
47226a10108bSNamhyung Kim 	ssize_t read, ret = 0;
4723faf982a6SNamhyung Kim 	struct ftrace_graph_data *fgd = file->private_data;
4724ea4e2bc4SSteven Rostedt 
4725c7c6b1feSLi Zefan 	if (!cnt)
4726ea4e2bc4SSteven Rostedt 		return 0;
4727ea4e2bc4SSteven Rostedt 
47286a10108bSNamhyung Kim 	if (trace_parser_get_init(&parser, FTRACE_BUFF_MAX))
47296a10108bSNamhyung Kim 		return -ENOMEM;
4730ea4e2bc4SSteven Rostedt 
4731689fd8b6S[email protected] 	read = trace_get_user(&parser, ubuf, cnt, ppos);
4732689fd8b6S[email protected] 
47334ba7978eSLi Zefan 	if (read >= 0 && trace_parser_loaded((&parser))) {
4734689fd8b6S[email protected] 		parser.buffer[parser.idx] = 0;
4735ea4e2bc4SSteven Rostedt 
47366a10108bSNamhyung Kim 		mutex_lock(&graph_lock);
47376a10108bSNamhyung Kim 
4738f9349a8fSFrederic Weisbecker 		/* we allow only one expression at a time */
4739faf982a6SNamhyung Kim 		ret = ftrace_set_func(fgd->table, fgd->count, fgd->size,
4740689fd8b6S[email protected] 				      parser.buffer);
47416a10108bSNamhyung Kim 
47426a10108bSNamhyung Kim 		mutex_unlock(&graph_lock);
4743689fd8b6S[email protected] 	}
4744ea4e2bc4SSteven Rostedt 
47456a10108bSNamhyung Kim 	if (!ret)
4746ea4e2bc4SSteven Rostedt 		ret = read;
47471eb90f13SLi Zefan 
4748689fd8b6S[email protected] 	trace_parser_put(&parser);
4749ea4e2bc4SSteven Rostedt 
4750ea4e2bc4SSteven Rostedt 	return ret;
4751ea4e2bc4SSteven Rostedt }
4752ea4e2bc4SSteven Rostedt 
4753ea4e2bc4SSteven Rostedt static const struct file_operations ftrace_graph_fops = {
4754ea4e2bc4SSteven Rostedt 	.open		= ftrace_graph_open,
4755850a80cfSLai Jiangshan 	.read		= seq_read,
4756ea4e2bc4SSteven Rostedt 	.write		= ftrace_graph_write,
4757098c879eSSteven Rostedt (Red Hat) 	.llseek		= tracing_lseek,
475887827111SLi Zefan 	.release	= ftrace_graph_release,
4759ea4e2bc4SSteven Rostedt };
476029ad23b0SNamhyung Kim 
476129ad23b0SNamhyung Kim static const struct file_operations ftrace_graph_notrace_fops = {
476229ad23b0SNamhyung Kim 	.open		= ftrace_graph_notrace_open,
476329ad23b0SNamhyung Kim 	.read		= seq_read,
476429ad23b0SNamhyung Kim 	.write		= ftrace_graph_write,
4765098c879eSSteven Rostedt (Red Hat) 	.llseek		= tracing_lseek,
476629ad23b0SNamhyung Kim 	.release	= ftrace_graph_release,
476729ad23b0SNamhyung Kim };
4768ea4e2bc4SSteven Rostedt #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
4769ea4e2bc4SSteven Rostedt 
4770591dffdaSSteven Rostedt (Red Hat) void ftrace_create_filter_files(struct ftrace_ops *ops,
4771591dffdaSSteven Rostedt (Red Hat) 				struct dentry *parent)
4772591dffdaSSteven Rostedt (Red Hat) {
4773591dffdaSSteven Rostedt (Red Hat) 
4774591dffdaSSteven Rostedt (Red Hat) 	trace_create_file("set_ftrace_filter", 0644, parent,
4775591dffdaSSteven Rostedt (Red Hat) 			  ops, &ftrace_filter_fops);
4776591dffdaSSteven Rostedt (Red Hat) 
4777591dffdaSSteven Rostedt (Red Hat) 	trace_create_file("set_ftrace_notrace", 0644, parent,
4778591dffdaSSteven Rostedt (Red Hat) 			  ops, &ftrace_notrace_fops);
4779591dffdaSSteven Rostedt (Red Hat) }
4780591dffdaSSteven Rostedt (Red Hat) 
4781591dffdaSSteven Rostedt (Red Hat) /*
4782591dffdaSSteven Rostedt (Red Hat)  * The name "destroy_filter_files" is really a misnomer. Although
4783591dffdaSSteven Rostedt (Red Hat)  * in the future, it may actualy delete the files, but this is
4784591dffdaSSteven Rostedt (Red Hat)  * really intended to make sure the ops passed in are disabled
4785591dffdaSSteven Rostedt (Red Hat)  * and that when this function returns, the caller is free to
4786591dffdaSSteven Rostedt (Red Hat)  * free the ops.
4787591dffdaSSteven Rostedt (Red Hat)  *
4788591dffdaSSteven Rostedt (Red Hat)  * The "destroy" name is only to match the "create" name that this
4789591dffdaSSteven Rostedt (Red Hat)  * should be paired with.
4790591dffdaSSteven Rostedt (Red Hat)  */
4791591dffdaSSteven Rostedt (Red Hat) void ftrace_destroy_filter_files(struct ftrace_ops *ops)
4792591dffdaSSteven Rostedt (Red Hat) {
4793591dffdaSSteven Rostedt (Red Hat) 	mutex_lock(&ftrace_lock);
4794591dffdaSSteven Rostedt (Red Hat) 	if (ops->flags & FTRACE_OPS_FL_ENABLED)
4795591dffdaSSteven Rostedt (Red Hat) 		ftrace_shutdown(ops, 0);
4796591dffdaSSteven Rostedt (Red Hat) 	ops->flags |= FTRACE_OPS_FL_DELETED;
4797591dffdaSSteven Rostedt (Red Hat) 	mutex_unlock(&ftrace_lock);
4798591dffdaSSteven Rostedt (Red Hat) }
4799591dffdaSSteven Rostedt (Red Hat) 
48008434dc93SSteven Rostedt (Red Hat) static __init int ftrace_init_dyn_tracefs(struct dentry *d_tracer)
48015072c59fSSteven Rostedt {
48025072c59fSSteven Rostedt 
48035452af66SFrederic Weisbecker 	trace_create_file("available_filter_functions", 0444,
48045072c59fSSteven Rostedt 			d_tracer, NULL, &ftrace_avail_fops);
48055072c59fSSteven Rostedt 
4806647bcd03SSteven Rostedt 	trace_create_file("enabled_functions", 0444,
4807647bcd03SSteven Rostedt 			d_tracer, NULL, &ftrace_enabled_fops);
4808647bcd03SSteven Rostedt 
4809591dffdaSSteven Rostedt (Red Hat) 	ftrace_create_filter_files(&global_ops, d_tracer);
4810ad90c0e3SSteven Rostedt 
4811ea4e2bc4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
48125452af66SFrederic Weisbecker 	trace_create_file("set_graph_function", 0444, d_tracer,
4813ea4e2bc4SSteven Rostedt 				    NULL,
4814ea4e2bc4SSteven Rostedt 				    &ftrace_graph_fops);
481529ad23b0SNamhyung Kim 	trace_create_file("set_graph_notrace", 0444, d_tracer,
481629ad23b0SNamhyung Kim 				    NULL,
481729ad23b0SNamhyung Kim 				    &ftrace_graph_notrace_fops);
4818ea4e2bc4SSteven Rostedt #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
4819ea4e2bc4SSteven Rostedt 
48205072c59fSSteven Rostedt 	return 0;
48215072c59fSSteven Rostedt }
48225072c59fSSteven Rostedt 
48239fd49328SSteven Rostedt static int ftrace_cmp_ips(const void *a, const void *b)
482468950619SSteven Rostedt {
48259fd49328SSteven Rostedt 	const unsigned long *ipa = a;
48269fd49328SSteven Rostedt 	const unsigned long *ipb = b;
482768950619SSteven Rostedt 
48289fd49328SSteven Rostedt 	if (*ipa > *ipb)
48299fd49328SSteven Rostedt 		return 1;
48309fd49328SSteven Rostedt 	if (*ipa < *ipb)
48319fd49328SSteven Rostedt 		return -1;
48329fd49328SSteven Rostedt 	return 0;
48339fd49328SSteven Rostedt }
48349fd49328SSteven Rostedt 
48355cb084bbSJiri Olsa static int ftrace_process_locs(struct module *mod,
483631e88909SSteven Rostedt 			       unsigned long *start,
483768bf21aaSSteven Rostedt 			       unsigned long *end)
483868bf21aaSSteven Rostedt {
4839706c81f8SSteven Rostedt 	struct ftrace_page *start_pg;
4840a7900875SSteven Rostedt 	struct ftrace_page *pg;
4841706c81f8SSteven Rostedt 	struct dyn_ftrace *rec;
4842a7900875SSteven Rostedt 	unsigned long count;
484368bf21aaSSteven Rostedt 	unsigned long *p;
484468bf21aaSSteven Rostedt 	unsigned long addr;
48454376cac6SSteven Rostedt 	unsigned long flags = 0; /* Shut up gcc */
4846a7900875SSteven Rostedt 	int ret = -ENOMEM;
4847a7900875SSteven Rostedt 
4848a7900875SSteven Rostedt 	count = end - start;
4849a7900875SSteven Rostedt 
4850a7900875SSteven Rostedt 	if (!count)
4851a7900875SSteven Rostedt 		return 0;
4852a7900875SSteven Rostedt 
48539fd49328SSteven Rostedt 	sort(start, count, sizeof(*start),
48546db02903SRasmus Villemoes 	     ftrace_cmp_ips, NULL);
48559fd49328SSteven Rostedt 
4856706c81f8SSteven Rostedt 	start_pg = ftrace_allocate_pages(count);
4857706c81f8SSteven Rostedt 	if (!start_pg)
4858a7900875SSteven Rostedt 		return -ENOMEM;
485968bf21aaSSteven Rostedt 
4860e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
4861a7900875SSteven Rostedt 
486232082309SSteven Rostedt 	/*
486332082309SSteven Rostedt 	 * Core and each module needs their own pages, as
486432082309SSteven Rostedt 	 * modules will free them when they are removed.
486532082309SSteven Rostedt 	 * Force a new page to be allocated for modules.
486632082309SSteven Rostedt 	 */
4867a7900875SSteven Rostedt 	if (!mod) {
4868a7900875SSteven Rostedt 		WARN_ON(ftrace_pages || ftrace_pages_start);
4869a7900875SSteven Rostedt 		/* First initialization */
4870706c81f8SSteven Rostedt 		ftrace_pages = ftrace_pages_start = start_pg;
4871a7900875SSteven Rostedt 	} else {
487232082309SSteven Rostedt 		if (!ftrace_pages)
4873a7900875SSteven Rostedt 			goto out;
487432082309SSteven Rostedt 
4875a7900875SSteven Rostedt 		if (WARN_ON(ftrace_pages->next)) {
4876a7900875SSteven Rostedt 			/* Hmm, we have free pages? */
4877a7900875SSteven Rostedt 			while (ftrace_pages->next)
487832082309SSteven Rostedt 				ftrace_pages = ftrace_pages->next;
487932082309SSteven Rostedt 		}
4880a7900875SSteven Rostedt 
4881706c81f8SSteven Rostedt 		ftrace_pages->next = start_pg;
488232082309SSteven Rostedt 	}
488332082309SSteven Rostedt 
488468bf21aaSSteven Rostedt 	p = start;
4885706c81f8SSteven Rostedt 	pg = start_pg;
488668bf21aaSSteven Rostedt 	while (p < end) {
488768bf21aaSSteven Rostedt 		addr = ftrace_call_adjust(*p++);
488820e5227eSSteven Rostedt 		/*
488920e5227eSSteven Rostedt 		 * Some architecture linkers will pad between
489020e5227eSSteven Rostedt 		 * the different mcount_loc sections of different
489120e5227eSSteven Rostedt 		 * object files to satisfy alignments.
489220e5227eSSteven Rostedt 		 * Skip any NULL pointers.
489320e5227eSSteven Rostedt 		 */
489420e5227eSSteven Rostedt 		if (!addr)
489520e5227eSSteven Rostedt 			continue;
4896706c81f8SSteven Rostedt 
4897706c81f8SSteven Rostedt 		if (pg->index == pg->size) {
4898706c81f8SSteven Rostedt 			/* We should have allocated enough */
4899706c81f8SSteven Rostedt 			if (WARN_ON(!pg->next))
4900a7900875SSteven Rostedt 				break;
4901706c81f8SSteven Rostedt 			pg = pg->next;
490268bf21aaSSteven Rostedt 		}
490368bf21aaSSteven Rostedt 
4904706c81f8SSteven Rostedt 		rec = &pg->records[pg->index++];
4905706c81f8SSteven Rostedt 		rec->ip = addr;
4906706c81f8SSteven Rostedt 	}
4907706c81f8SSteven Rostedt 
4908706c81f8SSteven Rostedt 	/* We should have used all pages */
4909706c81f8SSteven Rostedt 	WARN_ON(pg->next);
4910706c81f8SSteven Rostedt 
4911706c81f8SSteven Rostedt 	/* Assign the last page to ftrace_pages */
4912706c81f8SSteven Rostedt 	ftrace_pages = pg;
4913706c81f8SSteven Rostedt 
4914a4f18ed1SSteven Rostedt 	/*
49154376cac6SSteven Rostedt 	 * We only need to disable interrupts on start up
49164376cac6SSteven Rostedt 	 * because we are modifying code that an interrupt
49174376cac6SSteven Rostedt 	 * may execute, and the modification is not atomic.
49184376cac6SSteven Rostedt 	 * But for modules, nothing runs the code we modify
49194376cac6SSteven Rostedt 	 * until we are finished with it, and there's no
49204376cac6SSteven Rostedt 	 * reason to cause large interrupt latencies while we do it.
4921a4f18ed1SSteven Rostedt 	 */
49224376cac6SSteven Rostedt 	if (!mod)
4923a4f18ed1SSteven Rostedt 		local_irq_save(flags);
49241dc43cf0SJiri Slaby 	ftrace_update_code(mod, start_pg);
49254376cac6SSteven Rostedt 	if (!mod)
4926a4f18ed1SSteven Rostedt 		local_irq_restore(flags);
4927a7900875SSteven Rostedt 	ret = 0;
4928a7900875SSteven Rostedt  out:
4929e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
493068bf21aaSSteven Rostedt 
4931a7900875SSteven Rostedt 	return ret;
493268bf21aaSSteven Rostedt }
493368bf21aaSSteven Rostedt 
493493eb677dSSteven Rostedt #ifdef CONFIG_MODULES
493532082309SSteven Rostedt 
493632082309SSteven Rostedt #define next_to_ftrace_page(p) container_of(p, struct ftrace_page, next)
493732082309SSteven Rostedt 
4938b7ffffbbSSteven Rostedt (Red Hat) static int referenced_filters(struct dyn_ftrace *rec)
4939b7ffffbbSSteven Rostedt (Red Hat) {
4940b7ffffbbSSteven Rostedt (Red Hat) 	struct ftrace_ops *ops;
4941b7ffffbbSSteven Rostedt (Red Hat) 	int cnt = 0;
4942b7ffffbbSSteven Rostedt (Red Hat) 
4943b7ffffbbSSteven Rostedt (Red Hat) 	for (ops = ftrace_ops_list; ops != &ftrace_list_end; ops = ops->next) {
4944b7ffffbbSSteven Rostedt (Red Hat) 		if (ops_references_rec(ops, rec))
4945b7ffffbbSSteven Rostedt (Red Hat) 		    cnt++;
4946b7ffffbbSSteven Rostedt (Red Hat) 	}
4947b7ffffbbSSteven Rostedt (Red Hat) 
4948b7ffffbbSSteven Rostedt (Red Hat) 	return cnt;
4949b7ffffbbSSteven Rostedt (Red Hat) }
4950b7ffffbbSSteven Rostedt (Red Hat) 
4951e7247a15S[email protected] void ftrace_release_mod(struct module *mod)
495293eb677dSSteven Rostedt {
495393eb677dSSteven Rostedt 	struct dyn_ftrace *rec;
495432082309SSteven Rostedt 	struct ftrace_page **last_pg;
495593eb677dSSteven Rostedt 	struct ftrace_page *pg;
4956a7900875SSteven Rostedt 	int order;
495793eb677dSSteven Rostedt 
495893eb677dSSteven Rostedt 	mutex_lock(&ftrace_lock);
495945a4a237SSteven Rostedt 
496045a4a237SSteven Rostedt 	if (ftrace_disabled)
496145a4a237SSteven Rostedt 		goto out_unlock;
496245a4a237SSteven Rostedt 
496332082309SSteven Rostedt 	/*
496432082309SSteven Rostedt 	 * Each module has its own ftrace_pages, remove
496532082309SSteven Rostedt 	 * them from the list.
496632082309SSteven Rostedt 	 */
496732082309SSteven Rostedt 	last_pg = &ftrace_pages_start;
496832082309SSteven Rostedt 	for (pg = ftrace_pages_start; pg; pg = *last_pg) {
496932082309SSteven Rostedt 		rec = &pg->records[0];
4970e7247a15S[email protected] 		if (within_module_core(rec->ip, mod)) {
497193eb677dSSteven Rostedt 			/*
497232082309SSteven Rostedt 			 * As core pages are first, the first
497332082309SSteven Rostedt 			 * page should never be a module page.
497493eb677dSSteven Rostedt 			 */
497532082309SSteven Rostedt 			if (WARN_ON(pg == ftrace_pages_start))
497632082309SSteven Rostedt 				goto out_unlock;
497732082309SSteven Rostedt 
497832082309SSteven Rostedt 			/* Check if we are deleting the last page */
497932082309SSteven Rostedt 			if (pg == ftrace_pages)
498032082309SSteven Rostedt 				ftrace_pages = next_to_ftrace_page(last_pg);
498132082309SSteven Rostedt 
498232082309SSteven Rostedt 			*last_pg = pg->next;
4983a7900875SSteven Rostedt 			order = get_count_order(pg->size / ENTRIES_PER_PAGE);
4984a7900875SSteven Rostedt 			free_pages((unsigned long)pg->records, order);
4985a7900875SSteven Rostedt 			kfree(pg);
498632082309SSteven Rostedt 		} else
498732082309SSteven Rostedt 			last_pg = &pg->next;
498893eb677dSSteven Rostedt 	}
498945a4a237SSteven Rostedt  out_unlock:
499093eb677dSSteven Rostedt 	mutex_unlock(&ftrace_lock);
499193eb677dSSteven Rostedt }
499293eb677dSSteven Rostedt 
49937dcd182bSJessica Yu void ftrace_module_enable(struct module *mod)
4994b7ffffbbSSteven Rostedt (Red Hat) {
4995b7ffffbbSSteven Rostedt (Red Hat) 	struct dyn_ftrace *rec;
4996b7ffffbbSSteven Rostedt (Red Hat) 	struct ftrace_page *pg;
4997b7ffffbbSSteven Rostedt (Red Hat) 
4998b7ffffbbSSteven Rostedt (Red Hat) 	mutex_lock(&ftrace_lock);
4999b7ffffbbSSteven Rostedt (Red Hat) 
5000b7ffffbbSSteven Rostedt (Red Hat) 	if (ftrace_disabled)
5001b7ffffbbSSteven Rostedt (Red Hat) 		goto out_unlock;
5002b7ffffbbSSteven Rostedt (Red Hat) 
5003b7ffffbbSSteven Rostedt (Red Hat) 	/*
5004b7ffffbbSSteven Rostedt (Red Hat) 	 * If the tracing is enabled, go ahead and enable the record.
5005b7ffffbbSSteven Rostedt (Red Hat) 	 *
5006b7ffffbbSSteven Rostedt (Red Hat) 	 * The reason not to enable the record immediatelly is the
5007b7ffffbbSSteven Rostedt (Red Hat) 	 * inherent check of ftrace_make_nop/ftrace_make_call for
5008b7ffffbbSSteven Rostedt (Red Hat) 	 * correct previous instructions.  Making first the NOP
5009b7ffffbbSSteven Rostedt (Red Hat) 	 * conversion puts the module to the correct state, thus
5010b7ffffbbSSteven Rostedt (Red Hat) 	 * passing the ftrace_make_call check.
5011b7ffffbbSSteven Rostedt (Red Hat) 	 *
5012b7ffffbbSSteven Rostedt (Red Hat) 	 * We also delay this to after the module code already set the
5013b7ffffbbSSteven Rostedt (Red Hat) 	 * text to read-only, as we now need to set it back to read-write
5014b7ffffbbSSteven Rostedt (Red Hat) 	 * so that we can modify the text.
5015b7ffffbbSSteven Rostedt (Red Hat) 	 */
5016b7ffffbbSSteven Rostedt (Red Hat) 	if (ftrace_start_up)
5017b7ffffbbSSteven Rostedt (Red Hat) 		ftrace_arch_code_modify_prepare();
5018b7ffffbbSSteven Rostedt (Red Hat) 
5019b7ffffbbSSteven Rostedt (Red Hat) 	do_for_each_ftrace_rec(pg, rec) {
5020b7ffffbbSSteven Rostedt (Red Hat) 		int cnt;
5021b7ffffbbSSteven Rostedt (Red Hat) 		/*
5022b7ffffbbSSteven Rostedt (Red Hat) 		 * do_for_each_ftrace_rec() is a double loop.
5023b7ffffbbSSteven Rostedt (Red Hat) 		 * module text shares the pg. If a record is
5024b7ffffbbSSteven Rostedt (Red Hat) 		 * not part of this module, then skip this pg,
5025b7ffffbbSSteven Rostedt (Red Hat) 		 * which the "break" will do.
5026b7ffffbbSSteven Rostedt (Red Hat) 		 */
5027b7ffffbbSSteven Rostedt (Red Hat) 		if (!within_module_core(rec->ip, mod))
5028b7ffffbbSSteven Rostedt (Red Hat) 			break;
5029b7ffffbbSSteven Rostedt (Red Hat) 
5030b7ffffbbSSteven Rostedt (Red Hat) 		cnt = 0;
5031b7ffffbbSSteven Rostedt (Red Hat) 
5032b7ffffbbSSteven Rostedt (Red Hat) 		/*
5033b7ffffbbSSteven Rostedt (Red Hat) 		 * When adding a module, we need to check if tracers are
5034b7ffffbbSSteven Rostedt (Red Hat) 		 * currently enabled and if they are, and can trace this record,
5035b7ffffbbSSteven Rostedt (Red Hat) 		 * we need to enable the module functions as well as update the
5036b7ffffbbSSteven Rostedt (Red Hat) 		 * reference counts for those function records.
5037b7ffffbbSSteven Rostedt (Red Hat) 		 */
5038b7ffffbbSSteven Rostedt (Red Hat) 		if (ftrace_start_up)
5039b7ffffbbSSteven Rostedt (Red Hat) 			cnt += referenced_filters(rec);
5040b7ffffbbSSteven Rostedt (Red Hat) 
5041b7ffffbbSSteven Rostedt (Red Hat) 		/* This clears FTRACE_FL_DISABLED */
5042b7ffffbbSSteven Rostedt (Red Hat) 		rec->flags = cnt;
5043b7ffffbbSSteven Rostedt (Red Hat) 
5044b7ffffbbSSteven Rostedt (Red Hat) 		if (ftrace_start_up && cnt) {
5045b7ffffbbSSteven Rostedt (Red Hat) 			int failed = __ftrace_replace_code(rec, 1);
5046b7ffffbbSSteven Rostedt (Red Hat) 			if (failed) {
5047b7ffffbbSSteven Rostedt (Red Hat) 				ftrace_bug(failed, rec);
5048b7ffffbbSSteven Rostedt (Red Hat) 				goto out_loop;
5049b7ffffbbSSteven Rostedt (Red Hat) 			}
5050b7ffffbbSSteven Rostedt (Red Hat) 		}
5051b7ffffbbSSteven Rostedt (Red Hat) 
5052b7ffffbbSSteven Rostedt (Red Hat) 	} while_for_each_ftrace_rec();
5053b7ffffbbSSteven Rostedt (Red Hat) 
5054b7ffffbbSSteven Rostedt (Red Hat)  out_loop:
5055b7ffffbbSSteven Rostedt (Red Hat) 	if (ftrace_start_up)
5056b7ffffbbSSteven Rostedt (Red Hat) 		ftrace_arch_code_modify_post_process();
5057b7ffffbbSSteven Rostedt (Red Hat) 
5058b7ffffbbSSteven Rostedt (Red Hat)  out_unlock:
5059b7ffffbbSSteven Rostedt (Red Hat) 	mutex_unlock(&ftrace_lock);
5060b7ffffbbSSteven Rostedt (Red Hat) }
5061b7ffffbbSSteven Rostedt (Red Hat) 
5062a949ae56SSteven Rostedt (Red Hat) void ftrace_module_init(struct module *mod)
506393eb677dSSteven Rostedt {
506497e9b4fcSSteven Rostedt (Red Hat) 	if (ftrace_disabled || !mod->num_ftrace_callsites)
5065b6b71f66SAbel Vesa 		return;
5066b6b71f66SAbel Vesa 
506797e9b4fcSSteven Rostedt (Red Hat) 	ftrace_process_locs(mod, mod->ftrace_callsites,
506897e9b4fcSSteven Rostedt (Red Hat) 			    mod->ftrace_callsites + mod->num_ftrace_callsites);
506993eb677dSSteven Rostedt }
507093eb677dSSteven Rostedt #endif /* CONFIG_MODULES */
507193eb677dSSteven Rostedt 
507268bf21aaSSteven Rostedt void __init ftrace_init(void)
507368bf21aaSSteven Rostedt {
50741dc43cf0SJiri Slaby 	extern unsigned long __start_mcount_loc[];
50751dc43cf0SJiri Slaby 	extern unsigned long __stop_mcount_loc[];
50763a36cb11SJiri Slaby 	unsigned long count, flags;
507768bf21aaSSteven Rostedt 	int ret;
507868bf21aaSSteven Rostedt 
507968bf21aaSSteven Rostedt 	local_irq_save(flags);
50803a36cb11SJiri Slaby 	ret = ftrace_dyn_arch_init();
508168bf21aaSSteven Rostedt 	local_irq_restore(flags);
5082af64a7cbSJiri Slaby 	if (ret)
508368bf21aaSSteven Rostedt 		goto failed;
508468bf21aaSSteven Rostedt 
508568bf21aaSSteven Rostedt 	count = __stop_mcount_loc - __start_mcount_loc;
5086c867ccd8SJiri Slaby 	if (!count) {
5087c867ccd8SJiri Slaby 		pr_info("ftrace: No functions to be traced?\n");
508868bf21aaSSteven Rostedt 		goto failed;
5089c867ccd8SJiri Slaby 	}
5090c867ccd8SJiri Slaby 
5091c867ccd8SJiri Slaby 	pr_info("ftrace: allocating %ld entries in %ld pages\n",
5092c867ccd8SJiri Slaby 		count, count / ENTRIES_PER_PAGE + 1);
509368bf21aaSSteven Rostedt 
509468bf21aaSSteven Rostedt 	last_ftrace_enabled = ftrace_enabled = 1;
509568bf21aaSSteven Rostedt 
50965cb084bbSJiri Olsa 	ret = ftrace_process_locs(NULL,
509731e88909SSteven Rostedt 				  __start_mcount_loc,
509868bf21aaSSteven Rostedt 				  __stop_mcount_loc);
509968bf21aaSSteven Rostedt 
51002af15d6aSSteven Rostedt 	set_ftrace_early_filters();
51012af15d6aSSteven Rostedt 
510268bf21aaSSteven Rostedt 	return;
510368bf21aaSSteven Rostedt  failed:
510468bf21aaSSteven Rostedt 	ftrace_disabled = 1;
510568bf21aaSSteven Rostedt }
510668bf21aaSSteven Rostedt 
5107f3bea491SSteven Rostedt (Red Hat) /* Do nothing if arch does not support this */
5108f3bea491SSteven Rostedt (Red Hat) void __weak arch_ftrace_update_trampoline(struct ftrace_ops *ops)
5109f3bea491SSteven Rostedt (Red Hat) {
5110f3bea491SSteven Rostedt (Red Hat) }
5111f3bea491SSteven Rostedt (Red Hat) 
5112f3bea491SSteven Rostedt (Red Hat) static void ftrace_update_trampoline(struct ftrace_ops *ops)
5113f3bea491SSteven Rostedt (Red Hat) {
511412cce594SSteven Rostedt (Red Hat) 
511512cce594SSteven Rostedt (Red Hat) /*
511612cce594SSteven Rostedt (Red Hat)  * Currently there's no safe way to free a trampoline when the kernel
511712cce594SSteven Rostedt (Red Hat)  * is configured with PREEMPT. That is because a task could be preempted
511812cce594SSteven Rostedt (Red Hat)  * when it jumped to the trampoline, it may be preempted for a long time
511912cce594SSteven Rostedt (Red Hat)  * depending on the system load, and currently there's no way to know
512012cce594SSteven Rostedt (Red Hat)  * when it will be off the trampoline. If the trampoline is freed
512112cce594SSteven Rostedt (Red Hat)  * too early, when the task runs again, it will be executing on freed
512212cce594SSteven Rostedt (Red Hat)  * memory and crash.
512312cce594SSteven Rostedt (Red Hat)  */
512412cce594SSteven Rostedt (Red Hat) #ifdef CONFIG_PREEMPT
5125f3bea491SSteven Rostedt (Red Hat) 	/* Currently, only non dynamic ops can have a trampoline */
5126f3bea491SSteven Rostedt (Red Hat) 	if (ops->flags & FTRACE_OPS_FL_DYNAMIC)
5127f3bea491SSteven Rostedt (Red Hat) 		return;
512812cce594SSteven Rostedt (Red Hat) #endif
5129f3bea491SSteven Rostedt (Red Hat) 
5130f3bea491SSteven Rostedt (Red Hat) 	arch_ftrace_update_trampoline(ops);
5131f3bea491SSteven Rostedt (Red Hat) }
5132f3bea491SSteven Rostedt (Red Hat) 
51333d083395SSteven Rostedt #else
51340b6e4d56SFrederic Weisbecker 
51352b499381SSteven Rostedt static struct ftrace_ops global_ops = {
5136bd69c30bSSteven Rostedt 	.func			= ftrace_stub,
5137e3eea140SSteven Rostedt (Red Hat) 	.flags			= FTRACE_OPS_FL_RECURSION_SAFE |
5138e3eea140SSteven Rostedt (Red Hat) 				  FTRACE_OPS_FL_INITIALIZED |
5139e3eea140SSteven Rostedt (Red Hat) 				  FTRACE_OPS_FL_PID,
5140bd69c30bSSteven Rostedt };
5141bd69c30bSSteven Rostedt 
51420b6e4d56SFrederic Weisbecker static int __init ftrace_nodyn_init(void)
51430b6e4d56SFrederic Weisbecker {
51440b6e4d56SFrederic Weisbecker 	ftrace_enabled = 1;
51450b6e4d56SFrederic Weisbecker 	return 0;
51460b6e4d56SFrederic Weisbecker }
51476f415672SSteven Rostedt core_initcall(ftrace_nodyn_init);
51480b6e4d56SFrederic Weisbecker 
51498434dc93SSteven Rostedt (Red Hat) static inline int ftrace_init_dyn_tracefs(struct dentry *d_tracer) { return 0; }
5150df4fc315SSteven Rostedt static inline void ftrace_startup_enable(int command) { }
5151e1effa01SSteven Rostedt (Red Hat) static inline void ftrace_startup_all(int command) { }
51525a45cfe1SSteven Rostedt /* Keep as macros so we do not need to define the commands */
51533b6cfdb1SSteven Rostedt # define ftrace_startup(ops, command)					\
51543b6cfdb1SSteven Rostedt 	({								\
51558a56d776SSteven Rostedt (Red Hat) 		int ___ret = __register_ftrace_function(ops);		\
51568a56d776SSteven Rostedt (Red Hat) 		if (!___ret)						\
51573b6cfdb1SSteven Rostedt 			(ops)->flags |= FTRACE_OPS_FL_ENABLED;		\
51588a56d776SSteven Rostedt (Red Hat) 		___ret;							\
51593b6cfdb1SSteven Rostedt 	})
51601fcc1553SSteven Rostedt (Red Hat) # define ftrace_shutdown(ops, command)					\
51611fcc1553SSteven Rostedt (Red Hat) 	({								\
51621fcc1553SSteven Rostedt (Red Hat) 		int ___ret = __unregister_ftrace_function(ops);		\
51631fcc1553SSteven Rostedt (Red Hat) 		if (!___ret)						\
51641fcc1553SSteven Rostedt (Red Hat) 			(ops)->flags &= ~FTRACE_OPS_FL_ENABLED;		\
51651fcc1553SSteven Rostedt (Red Hat) 		___ret;							\
51661fcc1553SSteven Rostedt (Red Hat) 	})
51678a56d776SSteven Rostedt (Red Hat) 
5168b0fc494fSSteven Rostedt # define ftrace_startup_sysctl()	do { } while (0)
5169b0fc494fSSteven Rostedt # define ftrace_shutdown_sysctl()	do { } while (0)
5170b848914cSSteven Rostedt 
5171b848914cSSteven Rostedt static inline int
5172195a8afcSSteven Rostedt (Red Hat) ftrace_ops_test(struct ftrace_ops *ops, unsigned long ip, void *regs)
5173b848914cSSteven Rostedt {
5174b848914cSSteven Rostedt 	return 1;
5175b848914cSSteven Rostedt }
5176b848914cSSteven Rostedt 
5177f3bea491SSteven Rostedt (Red Hat) static void ftrace_update_trampoline(struct ftrace_ops *ops)
5178f3bea491SSteven Rostedt (Red Hat) {
5179f3bea491SSteven Rostedt (Red Hat) }
5180f3bea491SSteven Rostedt (Red Hat) 
51813d083395SSteven Rostedt #endif /* CONFIG_DYNAMIC_FTRACE */
51823d083395SSteven Rostedt 
51834104d326SSteven Rostedt (Red Hat) __init void ftrace_init_global_array_ops(struct trace_array *tr)
51844104d326SSteven Rostedt (Red Hat) {
51854104d326SSteven Rostedt (Red Hat) 	tr->ops = &global_ops;
51864104d326SSteven Rostedt (Red Hat) 	tr->ops->private = tr;
51874104d326SSteven Rostedt (Red Hat) }
51884104d326SSteven Rostedt (Red Hat) 
51894104d326SSteven Rostedt (Red Hat) void ftrace_init_array_ops(struct trace_array *tr, ftrace_func_t func)
51904104d326SSteven Rostedt (Red Hat) {
51914104d326SSteven Rostedt (Red Hat) 	/* If we filter on pids, update to use the pid function */
51924104d326SSteven Rostedt (Red Hat) 	if (tr->flags & TRACE_ARRAY_FL_GLOBAL) {
51934104d326SSteven Rostedt (Red Hat) 		if (WARN_ON(tr->ops->func != ftrace_stub))
51944104d326SSteven Rostedt (Red Hat) 			printk("ftrace ops had %pS for function\n",
51954104d326SSteven Rostedt (Red Hat) 			       tr->ops->func);
51964104d326SSteven Rostedt (Red Hat) 	}
51974104d326SSteven Rostedt (Red Hat) 	tr->ops->func = func;
51984104d326SSteven Rostedt (Red Hat) 	tr->ops->private = tr;
51994104d326SSteven Rostedt (Red Hat) }
52004104d326SSteven Rostedt (Red Hat) 
52014104d326SSteven Rostedt (Red Hat) void ftrace_reset_array_ops(struct trace_array *tr)
52024104d326SSteven Rostedt (Red Hat) {
52034104d326SSteven Rostedt (Red Hat) 	tr->ops->func = ftrace_stub;
52044104d326SSteven Rostedt (Red Hat) }
52054104d326SSteven Rostedt (Red Hat) 
52062f5f6ad9SSteven Rostedt static inline void
52072f5f6ad9SSteven Rostedt __ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
5208a1e2e31dSSteven Rostedt 		       struct ftrace_ops *ignored, struct pt_regs *regs)
5209b848914cSSteven Rostedt {
5210cdbe61bfSSteven Rostedt 	struct ftrace_ops *op;
5211edc15cafSSteven Rostedt 	int bit;
5212b848914cSSteven Rostedt 
5213edc15cafSSteven Rostedt 	bit = trace_test_and_set_recursion(TRACE_LIST_START, TRACE_LIST_MAX);
5214edc15cafSSteven Rostedt 	if (bit < 0)
5215b1cff0adSSteven Rostedt 		return;
5216b1cff0adSSteven Rostedt 
5217cdbe61bfSSteven Rostedt 	/*
5218cdbe61bfSSteven Rostedt 	 * Some of the ops may be dynamically allocated,
5219cdbe61bfSSteven Rostedt 	 * they must be freed after a synchronize_sched().
5220cdbe61bfSSteven Rostedt 	 */
5221cdbe61bfSSteven Rostedt 	preempt_disable_notrace();
5222ba27f2bcSSteven Rostedt (Red Hat) 
52230a016409SSteven Rostedt 	do_for_each_ftrace_op(op, ftrace_ops_list) {
5224ba27f2bcSSteven Rostedt (Red Hat) 		/*
5225ba27f2bcSSteven Rostedt (Red Hat) 		 * Check the following for each ops before calling their func:
5226ba27f2bcSSteven Rostedt (Red Hat) 		 *  if RCU flag is set, then rcu_is_watching() must be true
5227ba27f2bcSSteven Rostedt (Red Hat) 		 *  if PER_CPU is set, then ftrace_function_local_disable()
5228ba27f2bcSSteven Rostedt (Red Hat) 		 *                          must be false
5229ba27f2bcSSteven Rostedt (Red Hat) 		 *  Otherwise test if the ip matches the ops filter
5230ba27f2bcSSteven Rostedt (Red Hat) 		 *
5231ba27f2bcSSteven Rostedt (Red Hat) 		 * If any of the above fails then the op->func() is not executed.
5232ba27f2bcSSteven Rostedt (Red Hat) 		 */
5233ba27f2bcSSteven Rostedt (Red Hat) 		if ((!(op->flags & FTRACE_OPS_FL_RCU) || rcu_is_watching()) &&
5234ba27f2bcSSteven Rostedt (Red Hat) 		    (!(op->flags & FTRACE_OPS_FL_PER_CPU) ||
5235ba27f2bcSSteven Rostedt (Red Hat) 		     !ftrace_function_local_disabled(op)) &&
5236ba27f2bcSSteven Rostedt (Red Hat) 		    ftrace_ops_test(op, ip, regs)) {
5237ba27f2bcSSteven Rostedt (Red Hat) 
52381d48d596SSteven Rostedt (Red Hat) 			if (FTRACE_WARN_ON(!op->func)) {
52391d48d596SSteven Rostedt (Red Hat) 				pr_warn("op=%p %pS\n", op, op);
52404104d326SSteven Rostedt (Red Hat) 				goto out;
52414104d326SSteven Rostedt (Red Hat) 			}
5242a1e2e31dSSteven Rostedt 			op->func(ip, parent_ip, op, regs);
52434104d326SSteven Rostedt (Red Hat) 		}
52440a016409SSteven Rostedt 	} while_for_each_ftrace_op(op);
52454104d326SSteven Rostedt (Red Hat) out:
5246cdbe61bfSSteven Rostedt 	preempt_enable_notrace();
5247edc15cafSSteven Rostedt 	trace_clear_recursion(bit);
5248b848914cSSteven Rostedt }
5249b848914cSSteven Rostedt 
52502f5f6ad9SSteven Rostedt /*
52512f5f6ad9SSteven Rostedt  * Some archs only support passing ip and parent_ip. Even though
52522f5f6ad9SSteven Rostedt  * the list function ignores the op parameter, we do not want any
52532f5f6ad9SSteven Rostedt  * C side effects, where a function is called without the caller
52542f5f6ad9SSteven Rostedt  * sending a third parameter.
5255a1e2e31dSSteven Rostedt  * Archs are to support both the regs and ftrace_ops at the same time.
5256a1e2e31dSSteven Rostedt  * If they support ftrace_ops, it is assumed they support regs.
5257a1e2e31dSSteven Rostedt  * If call backs want to use regs, they must either check for regs
525806aeaaeaSMasami Hiramatsu  * being NULL, or CONFIG_DYNAMIC_FTRACE_WITH_REGS.
525906aeaaeaSMasami Hiramatsu  * Note, CONFIG_DYNAMIC_FTRACE_WITH_REGS expects a full regs to be saved.
5260a1e2e31dSSteven Rostedt  * An architecture can pass partial regs with ftrace_ops and still
5261b8ec330aSLi Bin  * set the ARCH_SUPPORTS_FTRACE_OPS.
52622f5f6ad9SSteven Rostedt  */
52632f5f6ad9SSteven Rostedt #if ARCH_SUPPORTS_FTRACE_OPS
52642f5f6ad9SSteven Rostedt static void ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
5265a1e2e31dSSteven Rostedt 				 struct ftrace_ops *op, struct pt_regs *regs)
52662f5f6ad9SSteven Rostedt {
5267a1e2e31dSSteven Rostedt 	__ftrace_ops_list_func(ip, parent_ip, NULL, regs);
52682f5f6ad9SSteven Rostedt }
52692f5f6ad9SSteven Rostedt #else
52702f5f6ad9SSteven Rostedt static void ftrace_ops_no_ops(unsigned long ip, unsigned long parent_ip)
52712f5f6ad9SSteven Rostedt {
5272a1e2e31dSSteven Rostedt 	__ftrace_ops_list_func(ip, parent_ip, NULL, NULL);
52732f5f6ad9SSteven Rostedt }
52742f5f6ad9SSteven Rostedt #endif
52752f5f6ad9SSteven Rostedt 
5276f1ff6348SSteven Rostedt (Red Hat) /*
5277f1ff6348SSteven Rostedt (Red Hat)  * If there's only one function registered but it does not support
5278c68c0fa2SSteven Rostedt (Red Hat)  * recursion, needs RCU protection and/or requires per cpu handling, then
5279c68c0fa2SSteven Rostedt (Red Hat)  * this function will be called by the mcount trampoline.
5280f1ff6348SSteven Rostedt (Red Hat)  */
5281c68c0fa2SSteven Rostedt (Red Hat) static void ftrace_ops_assist_func(unsigned long ip, unsigned long parent_ip,
5282f1ff6348SSteven Rostedt (Red Hat) 				   struct ftrace_ops *op, struct pt_regs *regs)
5283f1ff6348SSteven Rostedt (Red Hat) {
5284f1ff6348SSteven Rostedt (Red Hat) 	int bit;
5285f1ff6348SSteven Rostedt (Red Hat) 
5286c68c0fa2SSteven Rostedt (Red Hat) 	if ((op->flags & FTRACE_OPS_FL_RCU) && !rcu_is_watching())
5287c68c0fa2SSteven Rostedt (Red Hat) 		return;
5288c68c0fa2SSteven Rostedt (Red Hat) 
5289f1ff6348SSteven Rostedt (Red Hat) 	bit = trace_test_and_set_recursion(TRACE_LIST_START, TRACE_LIST_MAX);
5290f1ff6348SSteven Rostedt (Red Hat) 	if (bit < 0)
5291f1ff6348SSteven Rostedt (Red Hat) 		return;
5292f1ff6348SSteven Rostedt (Red Hat) 
5293c68c0fa2SSteven Rostedt (Red Hat) 	preempt_disable_notrace();
5294f1ff6348SSteven Rostedt (Red Hat) 
5295c68c0fa2SSteven Rostedt (Red Hat) 	if (!(op->flags & FTRACE_OPS_FL_PER_CPU) ||
5296c68c0fa2SSteven Rostedt (Red Hat) 	    !ftrace_function_local_disabled(op)) {
5297c68c0fa2SSteven Rostedt (Red Hat) 		op->func(ip, parent_ip, op, regs);
5298c68c0fa2SSteven Rostedt (Red Hat) 	}
5299c68c0fa2SSteven Rostedt (Red Hat) 
5300c68c0fa2SSteven Rostedt (Red Hat) 	preempt_enable_notrace();
5301f1ff6348SSteven Rostedt (Red Hat) 	trace_clear_recursion(bit);
5302f1ff6348SSteven Rostedt (Red Hat) }
5303f1ff6348SSteven Rostedt (Red Hat) 
530487354059SSteven Rostedt (Red Hat) /**
530587354059SSteven Rostedt (Red Hat)  * ftrace_ops_get_func - get the function a trampoline should call
530687354059SSteven Rostedt (Red Hat)  * @ops: the ops to get the function for
530787354059SSteven Rostedt (Red Hat)  *
530887354059SSteven Rostedt (Red Hat)  * Normally the mcount trampoline will call the ops->func, but there
530987354059SSteven Rostedt (Red Hat)  * are times that it should not. For example, if the ops does not
531087354059SSteven Rostedt (Red Hat)  * have its own recursion protection, then it should call the
531187354059SSteven Rostedt (Red Hat)  * ftrace_ops_recurs_func() instead.
531287354059SSteven Rostedt (Red Hat)  *
531387354059SSteven Rostedt (Red Hat)  * Returns the function that the trampoline should call for @ops.
531487354059SSteven Rostedt (Red Hat)  */
531587354059SSteven Rostedt (Red Hat) ftrace_func_t ftrace_ops_get_func(struct ftrace_ops *ops)
531687354059SSteven Rostedt (Red Hat) {
531787354059SSteven Rostedt (Red Hat) 	/*
5318c68c0fa2SSteven Rostedt (Red Hat) 	 * If the function does not handle recursion, needs to be RCU safe,
5319c68c0fa2SSteven Rostedt (Red Hat) 	 * or does per cpu logic, then we need to call the assist handler.
532087354059SSteven Rostedt (Red Hat) 	 */
5321c68c0fa2SSteven Rostedt (Red Hat) 	if (!(ops->flags & FTRACE_OPS_FL_RECURSION_SAFE) ||
5322c68c0fa2SSteven Rostedt (Red Hat) 	    ops->flags & (FTRACE_OPS_FL_RCU | FTRACE_OPS_FL_PER_CPU))
5323c68c0fa2SSteven Rostedt (Red Hat) 		return ftrace_ops_assist_func;
532487354059SSteven Rostedt (Red Hat) 
532587354059SSteven Rostedt (Red Hat) 	return ops->func;
532687354059SSteven Rostedt (Red Hat) }
532787354059SSteven Rostedt (Red Hat) 
5328*345ddcc8SSteven Rostedt (Red Hat) static void
5329*345ddcc8SSteven Rostedt (Red Hat) ftrace_filter_pid_sched_switch_probe(void *data, bool preempt,
5330*345ddcc8SSteven Rostedt (Red Hat) 		    struct task_struct *prev, struct task_struct *next)
5331e32d8956SSteven Rostedt {
5332*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr = data;
5333*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_pid_list *pid_list;
5334*345ddcc8SSteven Rostedt (Red Hat) 
5335*345ddcc8SSteven Rostedt (Red Hat) 	pid_list = rcu_dereference_sched(tr->function_pids);
5336*345ddcc8SSteven Rostedt (Red Hat) 
5337*345ddcc8SSteven Rostedt (Red Hat) 	this_cpu_write(tr->trace_buffer.data->ftrace_ignore_pid,
5338*345ddcc8SSteven Rostedt (Red Hat) 		       trace_ignore_this_task(pid_list, next));
5339*345ddcc8SSteven Rostedt (Red Hat) }
5340*345ddcc8SSteven Rostedt (Red Hat) 
5341*345ddcc8SSteven Rostedt (Red Hat) static void clear_ftrace_pids(struct trace_array *tr)
5342*345ddcc8SSteven Rostedt (Red Hat) {
5343*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_pid_list *pid_list;
5344e32d8956SSteven Rostedt 	int cpu;
5345e32d8956SSteven Rostedt 
5346*345ddcc8SSteven Rostedt (Red Hat) 	pid_list = rcu_dereference_protected(tr->function_pids,
5347*345ddcc8SSteven Rostedt (Red Hat) 					     lockdep_is_held(&ftrace_lock));
5348*345ddcc8SSteven Rostedt (Red Hat) 	if (!pid_list)
5349*345ddcc8SSteven Rostedt (Red Hat) 		return;
5350*345ddcc8SSteven Rostedt (Red Hat) 
5351*345ddcc8SSteven Rostedt (Red Hat) 	unregister_trace_sched_switch(ftrace_filter_pid_sched_switch_probe, tr);
5352*345ddcc8SSteven Rostedt (Red Hat) 
5353*345ddcc8SSteven Rostedt (Red Hat) 	for_each_possible_cpu(cpu)
5354*345ddcc8SSteven Rostedt (Red Hat) 		per_cpu_ptr(tr->trace_buffer.data, cpu)->ftrace_ignore_pid = false;
5355*345ddcc8SSteven Rostedt (Red Hat) 
5356*345ddcc8SSteven Rostedt (Red Hat) 	rcu_assign_pointer(tr->function_pids, NULL);
5357*345ddcc8SSteven Rostedt (Red Hat) 
5358*345ddcc8SSteven Rostedt (Red Hat) 	/* Wait till all users are no longer using pid filtering */
5359*345ddcc8SSteven Rostedt (Red Hat) 	synchronize_sched();
5360*345ddcc8SSteven Rostedt (Red Hat) 
5361*345ddcc8SSteven Rostedt (Red Hat) 	trace_free_pid_list(pid_list);
5362e32d8956SSteven Rostedt }
5363e32d8956SSteven Rostedt 
5364*345ddcc8SSteven Rostedt (Red Hat) static void ftrace_pid_reset(struct trace_array *tr)
5365e32d8956SSteven Rostedt {
5366756d17eeS[email protected] 	mutex_lock(&ftrace_lock);
5367*345ddcc8SSteven Rostedt (Red Hat) 	clear_ftrace_pids(tr);
5368756d17eeS[email protected] 
5369756d17eeS[email protected] 	ftrace_update_pid_func();
5370e1effa01SSteven Rostedt (Red Hat) 	ftrace_startup_all(0);
5371756d17eeS[email protected] 
5372756d17eeS[email protected] 	mutex_unlock(&ftrace_lock);
5373756d17eeS[email protected] }
5374756d17eeS[email protected] 
5375*345ddcc8SSteven Rostedt (Red Hat) /* Greater than any max PID */
5376*345ddcc8SSteven Rostedt (Red Hat) #define FTRACE_NO_PIDS		(void *)(PID_MAX_LIMIT + 1)
5377*345ddcc8SSteven Rostedt (Red Hat) 
5378756d17eeS[email protected] static void *fpid_start(struct seq_file *m, loff_t *pos)
5379*345ddcc8SSteven Rostedt (Red Hat) 	__acquires(RCU)
5380756d17eeS[email protected] {
5381*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_pid_list *pid_list;
5382*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr = m->private;
5383*345ddcc8SSteven Rostedt (Red Hat) 
5384756d17eeS[email protected] 	mutex_lock(&ftrace_lock);
5385*345ddcc8SSteven Rostedt (Red Hat) 	rcu_read_lock_sched();
5386756d17eeS[email protected] 
5387*345ddcc8SSteven Rostedt (Red Hat) 	pid_list = rcu_dereference_sched(tr->function_pids);
5388756d17eeS[email protected] 
5389*345ddcc8SSteven Rostedt (Red Hat) 	if (!pid_list)
5390*345ddcc8SSteven Rostedt (Red Hat) 		return !(*pos) ? FTRACE_NO_PIDS : NULL;
5391*345ddcc8SSteven Rostedt (Red Hat) 
5392*345ddcc8SSteven Rostedt (Red Hat) 	return trace_pid_start(pid_list, pos);
5393756d17eeS[email protected] }
5394756d17eeS[email protected] 
5395756d17eeS[email protected] static void *fpid_next(struct seq_file *m, void *v, loff_t *pos)
5396756d17eeS[email protected] {
5397*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr = m->private;
5398*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_pid_list *pid_list = rcu_dereference_sched(tr->function_pids);
5399*345ddcc8SSteven Rostedt (Red Hat) 
5400*345ddcc8SSteven Rostedt (Red Hat) 	if (v == FTRACE_NO_PIDS)
5401756d17eeS[email protected] 		return NULL;
5402756d17eeS[email protected] 
5403*345ddcc8SSteven Rostedt (Red Hat) 	return trace_pid_next(pid_list, v, pos);
5404756d17eeS[email protected] }
5405756d17eeS[email protected] 
5406756d17eeS[email protected] static void fpid_stop(struct seq_file *m, void *p)
5407*345ddcc8SSteven Rostedt (Red Hat) 	__releases(RCU)
5408756d17eeS[email protected] {
5409*345ddcc8SSteven Rostedt (Red Hat) 	rcu_read_unlock_sched();
5410756d17eeS[email protected] 	mutex_unlock(&ftrace_lock);
5411756d17eeS[email protected] }
5412756d17eeS[email protected] 
5413756d17eeS[email protected] static int fpid_show(struct seq_file *m, void *v)
5414756d17eeS[email protected] {
5415*345ddcc8SSteven Rostedt (Red Hat) 	if (v == FTRACE_NO_PIDS) {
5416fa6f0cc7SRasmus Villemoes 		seq_puts(m, "no pid\n");
5417756d17eeS[email protected] 		return 0;
5418756d17eeS[email protected] 	}
5419756d17eeS[email protected] 
5420*345ddcc8SSteven Rostedt (Red Hat) 	return trace_pid_show(m, v);
5421756d17eeS[email protected] }
5422756d17eeS[email protected] 
5423756d17eeS[email protected] static const struct seq_operations ftrace_pid_sops = {
5424756d17eeS[email protected] 	.start = fpid_start,
5425756d17eeS[email protected] 	.next = fpid_next,
5426756d17eeS[email protected] 	.stop = fpid_stop,
5427756d17eeS[email protected] 	.show = fpid_show,
5428756d17eeS[email protected] };
5429756d17eeS[email protected] 
5430756d17eeS[email protected] static int
5431756d17eeS[email protected] ftrace_pid_open(struct inode *inode, struct file *file)
5432756d17eeS[email protected] {
5433*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr = inode->i_private;
5434*345ddcc8SSteven Rostedt (Red Hat) 	struct seq_file *m;
5435756d17eeS[email protected] 	int ret = 0;
5436756d17eeS[email protected] 
5437*345ddcc8SSteven Rostedt (Red Hat) 	if (trace_array_get(tr) < 0)
5438*345ddcc8SSteven Rostedt (Red Hat) 		return -ENODEV;
5439*345ddcc8SSteven Rostedt (Red Hat) 
5440756d17eeS[email protected] 	if ((file->f_mode & FMODE_WRITE) &&
5441756d17eeS[email protected] 	    (file->f_flags & O_TRUNC))
5442*345ddcc8SSteven Rostedt (Red Hat) 		ftrace_pid_reset(tr);
5443756d17eeS[email protected] 
5444756d17eeS[email protected] 	ret = seq_open(file, &ftrace_pid_sops);
5445*345ddcc8SSteven Rostedt (Red Hat) 	if (ret < 0) {
5446*345ddcc8SSteven Rostedt (Red Hat) 		trace_array_put(tr);
5447*345ddcc8SSteven Rostedt (Red Hat) 	} else {
5448*345ddcc8SSteven Rostedt (Red Hat) 		m = file->private_data;
5449*345ddcc8SSteven Rostedt (Red Hat) 		/* copy tr over to seq ops */
5450*345ddcc8SSteven Rostedt (Red Hat) 		m->private = tr;
5451*345ddcc8SSteven Rostedt (Red Hat) 	}
5452756d17eeS[email protected] 
5453756d17eeS[email protected] 	return ret;
5454756d17eeS[email protected] }
5455756d17eeS[email protected] 
5456*345ddcc8SSteven Rostedt (Red Hat) static void ignore_task_cpu(void *data)
5457*345ddcc8SSteven Rostedt (Red Hat) {
5458*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr = data;
5459*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_pid_list *pid_list;
5460*345ddcc8SSteven Rostedt (Red Hat) 
5461*345ddcc8SSteven Rostedt (Red Hat) 	/*
5462*345ddcc8SSteven Rostedt (Red Hat) 	 * This function is called by on_each_cpu() while the
5463*345ddcc8SSteven Rostedt (Red Hat) 	 * event_mutex is held.
5464*345ddcc8SSteven Rostedt (Red Hat) 	 */
5465*345ddcc8SSteven Rostedt (Red Hat) 	pid_list = rcu_dereference_protected(tr->function_pids,
5466*345ddcc8SSteven Rostedt (Red Hat) 					     mutex_is_locked(&ftrace_lock));
5467*345ddcc8SSteven Rostedt (Red Hat) 
5468*345ddcc8SSteven Rostedt (Red Hat) 	this_cpu_write(tr->trace_buffer.data->ftrace_ignore_pid,
5469*345ddcc8SSteven Rostedt (Red Hat) 		       trace_ignore_this_task(pid_list, current));
5470*345ddcc8SSteven Rostedt (Red Hat) }
5471*345ddcc8SSteven Rostedt (Red Hat) 
5472df4fc315SSteven Rostedt static ssize_t
5473df4fc315SSteven Rostedt ftrace_pid_write(struct file *filp, const char __user *ubuf,
5474df4fc315SSteven Rostedt 		   size_t cnt, loff_t *ppos)
5475df4fc315SSteven Rostedt {
5476*345ddcc8SSteven Rostedt (Red Hat) 	struct seq_file *m = filp->private_data;
5477*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr = m->private;
5478*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_pid_list *filtered_pids = NULL;
5479*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_pid_list *pid_list;
5480*345ddcc8SSteven Rostedt (Red Hat) 	ssize_t ret;
5481df4fc315SSteven Rostedt 
5482*345ddcc8SSteven Rostedt (Red Hat) 	if (!cnt)
5483*345ddcc8SSteven Rostedt (Red Hat) 		return 0;
5484df4fc315SSteven Rostedt 
5485*345ddcc8SSteven Rostedt (Red Hat) 	mutex_lock(&ftrace_lock);
5486df4fc315SSteven Rostedt 
5487*345ddcc8SSteven Rostedt (Red Hat) 	filtered_pids = rcu_dereference_protected(tr->function_pids,
5488*345ddcc8SSteven Rostedt (Red Hat) 					     lockdep_is_held(&ftrace_lock));
5489*345ddcc8SSteven Rostedt (Red Hat) 
5490*345ddcc8SSteven Rostedt (Red Hat) 	ret = trace_pid_write(filtered_pids, &pid_list, ubuf, cnt);
5491*345ddcc8SSteven Rostedt (Red Hat) 	if (ret < 0)
5492*345ddcc8SSteven Rostedt (Red Hat) 		goto out;
5493*345ddcc8SSteven Rostedt (Red Hat) 
5494*345ddcc8SSteven Rostedt (Red Hat) 	rcu_assign_pointer(tr->function_pids, pid_list);
5495*345ddcc8SSteven Rostedt (Red Hat) 
5496*345ddcc8SSteven Rostedt (Red Hat) 	if (filtered_pids) {
5497*345ddcc8SSteven Rostedt (Red Hat) 		synchronize_sched();
5498*345ddcc8SSteven Rostedt (Red Hat) 		trace_free_pid_list(filtered_pids);
5499*345ddcc8SSteven Rostedt (Red Hat) 	} else if (pid_list) {
5500*345ddcc8SSteven Rostedt (Red Hat) 		/* Register a probe to set whether to ignore the tracing of a task */
5501*345ddcc8SSteven Rostedt (Red Hat) 		register_trace_sched_switch(ftrace_filter_pid_sched_switch_probe, tr);
5502*345ddcc8SSteven Rostedt (Red Hat) 	}
5503df4fc315SSteven Rostedt 
5504756d17eeS[email protected] 	/*
5505*345ddcc8SSteven Rostedt (Red Hat) 	 * Ignoring of pids is done at task switch. But we have to
5506*345ddcc8SSteven Rostedt (Red Hat) 	 * check for those tasks that are currently running.
5507*345ddcc8SSteven Rostedt (Red Hat) 	 * Always do this in case a pid was appended or removed.
5508756d17eeS[email protected] 	 */
5509*345ddcc8SSteven Rostedt (Red Hat) 	on_each_cpu(ignore_task_cpu, tr, 1);
5510756d17eeS[email protected] 
5511*345ddcc8SSteven Rostedt (Red Hat) 	ftrace_update_pid_func();
5512*345ddcc8SSteven Rostedt (Red Hat) 	ftrace_startup_all(0);
5513*345ddcc8SSteven Rostedt (Red Hat)  out:
5514*345ddcc8SSteven Rostedt (Red Hat) 	mutex_unlock(&ftrace_lock);
5515*345ddcc8SSteven Rostedt (Red Hat) 
5516*345ddcc8SSteven Rostedt (Red Hat) 	if (ret > 0)
5517*345ddcc8SSteven Rostedt (Red Hat) 		*ppos += ret;
5518*345ddcc8SSteven Rostedt (Red Hat) 
5519df4fc315SSteven Rostedt 	return ret;
5520e32d8956SSteven Rostedt }
5521978f3a45SSteven Rostedt 
5522756d17eeS[email protected] static int
5523756d17eeS[email protected] ftrace_pid_release(struct inode *inode, struct file *file)
5524756d17eeS[email protected] {
5525*345ddcc8SSteven Rostedt (Red Hat) 	struct trace_array *tr = inode->i_private;
5526978f3a45SSteven Rostedt 
5527*345ddcc8SSteven Rostedt (Red Hat) 	trace_array_put(tr);
5528*345ddcc8SSteven Rostedt (Red Hat) 
5529*345ddcc8SSteven Rostedt (Red Hat) 	return seq_release(inode, file);
5530df4fc315SSteven Rostedt }
5531df4fc315SSteven Rostedt 
55325e2336a0SSteven Rostedt static const struct file_operations ftrace_pid_fops = {
5533756d17eeS[email protected] 	.open		= ftrace_pid_open,
5534df4fc315SSteven Rostedt 	.write		= ftrace_pid_write,
5535756d17eeS[email protected] 	.read		= seq_read,
5536098c879eSSteven Rostedt (Red Hat) 	.llseek		= tracing_lseek,
5537756d17eeS[email protected] 	.release	= ftrace_pid_release,
5538df4fc315SSteven Rostedt };
5539df4fc315SSteven Rostedt 
5540*345ddcc8SSteven Rostedt (Red Hat) void ftrace_init_tracefs(struct trace_array *tr, struct dentry *d_tracer)
5541df4fc315SSteven Rostedt {
5542*345ddcc8SSteven Rostedt (Red Hat) 	/* Only the top level directory has the dyn_tracefs and profile */
5543*345ddcc8SSteven Rostedt (Red Hat) 	if (tr->flags & TRACE_ARRAY_FL_GLOBAL) {
55448434dc93SSteven Rostedt (Red Hat) 		ftrace_init_dyn_tracefs(d_tracer);
5545*345ddcc8SSteven Rostedt (Red Hat) 		ftrace_profile_tracefs(d_tracer);
5546*345ddcc8SSteven Rostedt (Red Hat) 	}
5547df4fc315SSteven Rostedt 
55485452af66SFrederic Weisbecker 	trace_create_file("set_ftrace_pid", 0644, d_tracer,
5549*345ddcc8SSteven Rostedt (Red Hat) 			    tr, &ftrace_pid_fops);
5550df4fc315SSteven Rostedt }
5551df4fc315SSteven Rostedt 
55523d083395SSteven Rostedt /**
555381adbdc0SSteven Rostedt  * ftrace_kill - kill ftrace
5554a2bb6a3dSSteven Rostedt  *
5555a2bb6a3dSSteven Rostedt  * This function should be used by panic code. It stops ftrace
5556a2bb6a3dSSteven Rostedt  * but in a not so nice way. If you need to simply kill ftrace
5557a2bb6a3dSSteven Rostedt  * from a non-atomic section, use ftrace_kill.
5558a2bb6a3dSSteven Rostedt  */
555981adbdc0SSteven Rostedt void ftrace_kill(void)
5560a2bb6a3dSSteven Rostedt {
5561a2bb6a3dSSteven Rostedt 	ftrace_disabled = 1;
5562a2bb6a3dSSteven Rostedt 	ftrace_enabled = 0;
5563a2bb6a3dSSteven Rostedt 	clear_ftrace_function();
5564a2bb6a3dSSteven Rostedt }
5565a2bb6a3dSSteven Rostedt 
5566a2bb6a3dSSteven Rostedt /**
5567e0a413f6SSteven Rostedt  * Test if ftrace is dead or not.
5568e0a413f6SSteven Rostedt  */
5569e0a413f6SSteven Rostedt int ftrace_is_dead(void)
5570e0a413f6SSteven Rostedt {
5571e0a413f6SSteven Rostedt 	return ftrace_disabled;
5572e0a413f6SSteven Rostedt }
5573e0a413f6SSteven Rostedt 
5574e0a413f6SSteven Rostedt /**
55753d083395SSteven Rostedt  * register_ftrace_function - register a function for profiling
55763d083395SSteven Rostedt  * @ops - ops structure that holds the function for profiling.
55773d083395SSteven Rostedt  *
55783d083395SSteven Rostedt  * Register a function to be called by all functions in the
55793d083395SSteven Rostedt  * kernel.
55803d083395SSteven Rostedt  *
55813d083395SSteven Rostedt  * Note: @ops->func and all the functions it calls must be labeled
55823d083395SSteven Rostedt  *       with "notrace", otherwise it will go into a
55833d083395SSteven Rostedt  *       recursive loop.
55843d083395SSteven Rostedt  */
55853d083395SSteven Rostedt int register_ftrace_function(struct ftrace_ops *ops)
55863d083395SSteven Rostedt {
558745a4a237SSteven Rostedt 	int ret = -1;
55884eebcc81SSteven Rostedt 
5589f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
5590f04f24fbSMasami Hiramatsu 
5591e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
5592e7d3737eSFrederic Weisbecker 
5593a1cd6173SSteven Rostedt 	ret = ftrace_startup(ops, 0);
5594b848914cSSteven Rostedt 
5595e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
55968d240dd8SBorislav Petkov 
5597b0fc494fSSteven Rostedt 	return ret;
55983d083395SSteven Rostedt }
5599cdbe61bfSSteven Rostedt EXPORT_SYMBOL_GPL(register_ftrace_function);
56003d083395SSteven Rostedt 
56013d083395SSteven Rostedt /**
560232632920SUwe Kleine-Koenig  * unregister_ftrace_function - unregister a function for profiling.
56033d083395SSteven Rostedt  * @ops - ops structure that holds the function to unregister
56043d083395SSteven Rostedt  *
56053d083395SSteven Rostedt  * Unregister a function that was added to be called by ftrace profiling.
56063d083395SSteven Rostedt  */
56073d083395SSteven Rostedt int unregister_ftrace_function(struct ftrace_ops *ops)
56083d083395SSteven Rostedt {
56093d083395SSteven Rostedt 	int ret;
56103d083395SSteven Rostedt 
5611e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
56128a56d776SSteven Rostedt (Red Hat) 	ret = ftrace_shutdown(ops, 0);
5613e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
5614b0fc494fSSteven Rostedt 
5615b0fc494fSSteven Rostedt 	return ret;
5616b0fc494fSSteven Rostedt }
5617cdbe61bfSSteven Rostedt EXPORT_SYMBOL_GPL(unregister_ftrace_function);
5618b0fc494fSSteven Rostedt 
5619e309b41dSIngo Molnar int
5620b0fc494fSSteven Rostedt ftrace_enable_sysctl(struct ctl_table *table, int write,
56218d65af78SAlexey Dobriyan 		     void __user *buffer, size_t *lenp,
5622b0fc494fSSteven Rostedt 		     loff_t *ppos)
5623b0fc494fSSteven Rostedt {
562445a4a237SSteven Rostedt 	int ret = -ENODEV;
56254eebcc81SSteven Rostedt 
5626e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
5627b0fc494fSSteven Rostedt 
562845a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
562945a4a237SSteven Rostedt 		goto out;
563045a4a237SSteven Rostedt 
56318d65af78SAlexey Dobriyan 	ret = proc_dointvec(table, write, buffer, lenp, ppos);
5632b0fc494fSSteven Rostedt 
5633a32c7765SLi Zefan 	if (ret || !write || (last_ftrace_enabled == !!ftrace_enabled))
5634b0fc494fSSteven Rostedt 		goto out;
5635b0fc494fSSteven Rostedt 
5636a32c7765SLi Zefan 	last_ftrace_enabled = !!ftrace_enabled;
5637b0fc494fSSteven Rostedt 
5638b0fc494fSSteven Rostedt 	if (ftrace_enabled) {
5639b0fc494fSSteven Rostedt 
5640b0fc494fSSteven Rostedt 		/* we are starting ftrace again */
56415000c418SJan Kiszka 		if (ftrace_ops_list != &ftrace_list_end)
56425000c418SJan Kiszka 			update_ftrace_function();
5643b0fc494fSSteven Rostedt 
5644524a3868SSteven Rostedt (Red Hat) 		ftrace_startup_sysctl();
5645524a3868SSteven Rostedt (Red Hat) 
5646b0fc494fSSteven Rostedt 	} else {
5647b0fc494fSSteven Rostedt 		/* stopping ftrace calls (just send to ftrace_stub) */
5648b0fc494fSSteven Rostedt 		ftrace_trace_function = ftrace_stub;
5649b0fc494fSSteven Rostedt 
5650b0fc494fSSteven Rostedt 		ftrace_shutdown_sysctl();
5651b0fc494fSSteven Rostedt 	}
5652b0fc494fSSteven Rostedt 
5653b0fc494fSSteven Rostedt  out:
5654e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
56553d083395SSteven Rostedt 	return ret;
565616444a8aSArnaldo Carvalho de Melo }
5657f17845e5SIngo Molnar 
5658fb52607aSFrederic Weisbecker #ifdef CONFIG_FUNCTION_GRAPH_TRACER
5659e7d3737eSFrederic Weisbecker 
56605f151b24SSteven Rostedt (Red Hat) static struct ftrace_ops graph_ops = {
56615f151b24SSteven Rostedt (Red Hat) 	.func			= ftrace_stub,
56625f151b24SSteven Rostedt (Red Hat) 	.flags			= FTRACE_OPS_FL_RECURSION_SAFE |
56635f151b24SSteven Rostedt (Red Hat) 				   FTRACE_OPS_FL_INITIALIZED |
5664e3eea140SSteven Rostedt (Red Hat) 				   FTRACE_OPS_FL_PID |
56655f151b24SSteven Rostedt (Red Hat) 				   FTRACE_OPS_FL_STUB,
56665f151b24SSteven Rostedt (Red Hat) #ifdef FTRACE_GRAPH_TRAMP_ADDR
56675f151b24SSteven Rostedt (Red Hat) 	.trampoline		= FTRACE_GRAPH_TRAMP_ADDR,
5668aec0be2dSSteven Rostedt (Red Hat) 	/* trampoline_size is only needed for dynamically allocated tramps */
56695f151b24SSteven Rostedt (Red Hat) #endif
56705f151b24SSteven Rostedt (Red Hat) 	ASSIGN_OPS_HASH(graph_ops, &global_ops.local_hash)
56715f151b24SSteven Rostedt (Red Hat) };
56725f151b24SSteven Rostedt (Red Hat) 
567355577204SSteven Rostedt (Red Hat) void ftrace_graph_sleep_time_control(bool enable)
567455577204SSteven Rostedt (Red Hat) {
567555577204SSteven Rostedt (Red Hat) 	fgraph_sleep_time = enable;
567655577204SSteven Rostedt (Red Hat) }
567755577204SSteven Rostedt (Red Hat) 
567855577204SSteven Rostedt (Red Hat) void ftrace_graph_graph_time_control(bool enable)
567955577204SSteven Rostedt (Red Hat) {
568055577204SSteven Rostedt (Red Hat) 	fgraph_graph_time = enable;
568155577204SSteven Rostedt (Red Hat) }
568255577204SSteven Rostedt (Red Hat) 
5683e49dc19cSSteven Rostedt int ftrace_graph_entry_stub(struct ftrace_graph_ent *trace)
5684e49dc19cSSteven Rostedt {
5685e49dc19cSSteven Rostedt 	return 0;
5686e49dc19cSSteven Rostedt }
5687e49dc19cSSteven Rostedt 
5688287b6e68SFrederic Weisbecker /* The callbacks that hook a function */
5689287b6e68SFrederic Weisbecker trace_func_graph_ret_t ftrace_graph_return =
5690287b6e68SFrederic Weisbecker 			(trace_func_graph_ret_t)ftrace_stub;
5691e49dc19cSSteven Rostedt trace_func_graph_ent_t ftrace_graph_entry = ftrace_graph_entry_stub;
569223a8e844SSteven Rostedt (Red Hat) static trace_func_graph_ent_t __ftrace_graph_entry = ftrace_graph_entry_stub;
5693f201ae23SFrederic Weisbecker 
5694f201ae23SFrederic Weisbecker /* Try to assign a return stack array on FTRACE_RETSTACK_ALLOC_SIZE tasks. */
5695f201ae23SFrederic Weisbecker static int alloc_retstack_tasklist(struct ftrace_ret_stack **ret_stack_list)
5696f201ae23SFrederic Weisbecker {
5697f201ae23SFrederic Weisbecker 	int i;
5698f201ae23SFrederic Weisbecker 	int ret = 0;
5699f201ae23SFrederic Weisbecker 	int start = 0, end = FTRACE_RETSTACK_ALLOC_SIZE;
5700f201ae23SFrederic Weisbecker 	struct task_struct *g, *t;
5701f201ae23SFrederic Weisbecker 
5702f201ae23SFrederic Weisbecker 	for (i = 0; i < FTRACE_RETSTACK_ALLOC_SIZE; i++) {
5703f201ae23SFrederic Weisbecker 		ret_stack_list[i] = kmalloc(FTRACE_RETFUNC_DEPTH
5704f201ae23SFrederic Weisbecker 					* sizeof(struct ftrace_ret_stack),
5705f201ae23SFrederic Weisbecker 					GFP_KERNEL);
5706f201ae23SFrederic Weisbecker 		if (!ret_stack_list[i]) {
5707f201ae23SFrederic Weisbecker 			start = 0;
5708f201ae23SFrederic Weisbecker 			end = i;
5709f201ae23SFrederic Weisbecker 			ret = -ENOMEM;
5710f201ae23SFrederic Weisbecker 			goto free;
5711f201ae23SFrederic Weisbecker 		}
5712f201ae23SFrederic Weisbecker 	}
5713f201ae23SFrederic Weisbecker 
57146112a300SSoumya PN 	read_lock(&tasklist_lock);
5715f201ae23SFrederic Weisbecker 	do_each_thread(g, t) {
5716f201ae23SFrederic Weisbecker 		if (start == end) {
5717f201ae23SFrederic Weisbecker 			ret = -EAGAIN;
5718f201ae23SFrederic Weisbecker 			goto unlock;
5719f201ae23SFrederic Weisbecker 		}
5720f201ae23SFrederic Weisbecker 
5721f201ae23SFrederic Weisbecker 		if (t->ret_stack == NULL) {
5722380c4b14SFrederic Weisbecker 			atomic_set(&t->tracing_graph_pause, 0);
5723f201ae23SFrederic Weisbecker 			atomic_set(&t->trace_overrun, 0);
572426c01624SSteven Rostedt 			t->curr_ret_stack = -1;
572526c01624SSteven Rostedt 			/* Make sure the tasks see the -1 first: */
572626c01624SSteven Rostedt 			smp_wmb();
572726c01624SSteven Rostedt 			t->ret_stack = ret_stack_list[start++];
5728f201ae23SFrederic Weisbecker 		}
5729f201ae23SFrederic Weisbecker 	} while_each_thread(g, t);
5730f201ae23SFrederic Weisbecker 
5731f201ae23SFrederic Weisbecker unlock:
57326112a300SSoumya PN 	read_unlock(&tasklist_lock);
5733f201ae23SFrederic Weisbecker free:
5734f201ae23SFrederic Weisbecker 	for (i = start; i < end; i++)
5735f201ae23SFrederic Weisbecker 		kfree(ret_stack_list[i]);
5736f201ae23SFrederic Weisbecker 	return ret;
5737f201ae23SFrederic Weisbecker }
5738f201ae23SFrederic Weisbecker 
57398aef2d28SSteven Rostedt static void
5740c73464b1SPeter Zijlstra ftrace_graph_probe_sched_switch(void *ignore, bool preempt,
574138516ab5SSteven Rostedt 			struct task_struct *prev, struct task_struct *next)
57428aef2d28SSteven Rostedt {
57438aef2d28SSteven Rostedt 	unsigned long long timestamp;
57448aef2d28SSteven Rostedt 	int index;
57458aef2d28SSteven Rostedt 
5746be6f164aSSteven Rostedt 	/*
5747be6f164aSSteven Rostedt 	 * Does the user want to count the time a function was asleep.
5748be6f164aSSteven Rostedt 	 * If so, do not update the time stamps.
5749be6f164aSSteven Rostedt 	 */
575055577204SSteven Rostedt (Red Hat) 	if (fgraph_sleep_time)
5751be6f164aSSteven Rostedt 		return;
5752be6f164aSSteven Rostedt 
57538aef2d28SSteven Rostedt 	timestamp = trace_clock_local();
57548aef2d28SSteven Rostedt 
57558aef2d28SSteven Rostedt 	prev->ftrace_timestamp = timestamp;
57568aef2d28SSteven Rostedt 
57578aef2d28SSteven Rostedt 	/* only process tasks that we timestamped */
57588aef2d28SSteven Rostedt 	if (!next->ftrace_timestamp)
57598aef2d28SSteven Rostedt 		return;
57608aef2d28SSteven Rostedt 
57618aef2d28SSteven Rostedt 	/*
57628aef2d28SSteven Rostedt 	 * Update all the counters in next to make up for the
57638aef2d28SSteven Rostedt 	 * time next was sleeping.
57648aef2d28SSteven Rostedt 	 */
57658aef2d28SSteven Rostedt 	timestamp -= next->ftrace_timestamp;
57668aef2d28SSteven Rostedt 
57678aef2d28SSteven Rostedt 	for (index = next->curr_ret_stack; index >= 0; index--)
57688aef2d28SSteven Rostedt 		next->ret_stack[index].calltime += timestamp;
57698aef2d28SSteven Rostedt }
57708aef2d28SSteven Rostedt 
5771f201ae23SFrederic Weisbecker /* Allocate a return stack for each task */
5772fb52607aSFrederic Weisbecker static int start_graph_tracing(void)
5773f201ae23SFrederic Weisbecker {
5774f201ae23SFrederic Weisbecker 	struct ftrace_ret_stack **ret_stack_list;
57755b058bcdSFrederic Weisbecker 	int ret, cpu;
5776f201ae23SFrederic Weisbecker 
5777f201ae23SFrederic Weisbecker 	ret_stack_list = kmalloc(FTRACE_RETSTACK_ALLOC_SIZE *
5778f201ae23SFrederic Weisbecker 				sizeof(struct ftrace_ret_stack *),
5779f201ae23SFrederic Weisbecker 				GFP_KERNEL);
5780f201ae23SFrederic Weisbecker 
5781f201ae23SFrederic Weisbecker 	if (!ret_stack_list)
5782f201ae23SFrederic Weisbecker 		return -ENOMEM;
5783f201ae23SFrederic Weisbecker 
57845b058bcdSFrederic Weisbecker 	/* The cpu_boot init_task->ret_stack will never be freed */
5785179c498aSSteven Rostedt 	for_each_online_cpu(cpu) {
5786179c498aSSteven Rostedt 		if (!idle_task(cpu)->ret_stack)
5787868baf07SSteven Rostedt 			ftrace_graph_init_idle_task(idle_task(cpu), cpu);
5788179c498aSSteven Rostedt 	}
57895b058bcdSFrederic Weisbecker 
5790f201ae23SFrederic Weisbecker 	do {
5791f201ae23SFrederic Weisbecker 		ret = alloc_retstack_tasklist(ret_stack_list);
5792f201ae23SFrederic Weisbecker 	} while (ret == -EAGAIN);
5793f201ae23SFrederic Weisbecker 
57948aef2d28SSteven Rostedt 	if (!ret) {
579538516ab5SSteven Rostedt 		ret = register_trace_sched_switch(ftrace_graph_probe_sched_switch, NULL);
57968aef2d28SSteven Rostedt 		if (ret)
57978aef2d28SSteven Rostedt 			pr_info("ftrace_graph: Couldn't activate tracepoint"
57988aef2d28SSteven Rostedt 				" probe to kernel_sched_switch\n");
57998aef2d28SSteven Rostedt 	}
58008aef2d28SSteven Rostedt 
5801f201ae23SFrederic Weisbecker 	kfree(ret_stack_list);
5802f201ae23SFrederic Weisbecker 	return ret;
5803f201ae23SFrederic Weisbecker }
5804f201ae23SFrederic Weisbecker 
58054a2b8ddaSFrederic Weisbecker /*
58064a2b8ddaSFrederic Weisbecker  * Hibernation protection.
58074a2b8ddaSFrederic Weisbecker  * The state of the current task is too much unstable during
58084a2b8ddaSFrederic Weisbecker  * suspend/restore to disk. We want to protect against that.
58094a2b8ddaSFrederic Weisbecker  */
58104a2b8ddaSFrederic Weisbecker static int
58114a2b8ddaSFrederic Weisbecker ftrace_suspend_notifier_call(struct notifier_block *bl, unsigned long state,
58124a2b8ddaSFrederic Weisbecker 							void *unused)
58134a2b8ddaSFrederic Weisbecker {
58144a2b8ddaSFrederic Weisbecker 	switch (state) {
58154a2b8ddaSFrederic Weisbecker 	case PM_HIBERNATION_PREPARE:
58164a2b8ddaSFrederic Weisbecker 		pause_graph_tracing();
58174a2b8ddaSFrederic Weisbecker 		break;
58184a2b8ddaSFrederic Weisbecker 
58194a2b8ddaSFrederic Weisbecker 	case PM_POST_HIBERNATION:
58204a2b8ddaSFrederic Weisbecker 		unpause_graph_tracing();
58214a2b8ddaSFrederic Weisbecker 		break;
58224a2b8ddaSFrederic Weisbecker 	}
58234a2b8ddaSFrederic Weisbecker 	return NOTIFY_DONE;
58244a2b8ddaSFrederic Weisbecker }
58254a2b8ddaSFrederic Weisbecker 
582623a8e844SSteven Rostedt (Red Hat) static int ftrace_graph_entry_test(struct ftrace_graph_ent *trace)
582723a8e844SSteven Rostedt (Red Hat) {
582823a8e844SSteven Rostedt (Red Hat) 	if (!ftrace_ops_test(&global_ops, trace->func, NULL))
582923a8e844SSteven Rostedt (Red Hat) 		return 0;
583023a8e844SSteven Rostedt (Red Hat) 	return __ftrace_graph_entry(trace);
583123a8e844SSteven Rostedt (Red Hat) }
583223a8e844SSteven Rostedt (Red Hat) 
583323a8e844SSteven Rostedt (Red Hat) /*
583423a8e844SSteven Rostedt (Red Hat)  * The function graph tracer should only trace the functions defined
583523a8e844SSteven Rostedt (Red Hat)  * by set_ftrace_filter and set_ftrace_notrace. If another function
583623a8e844SSteven Rostedt (Red Hat)  * tracer ops is registered, the graph tracer requires testing the
583723a8e844SSteven Rostedt (Red Hat)  * function against the global ops, and not just trace any function
583823a8e844SSteven Rostedt (Red Hat)  * that any ftrace_ops registered.
583923a8e844SSteven Rostedt (Red Hat)  */
584023a8e844SSteven Rostedt (Red Hat) static void update_function_graph_func(void)
584123a8e844SSteven Rostedt (Red Hat) {
58425f151b24SSteven Rostedt (Red Hat) 	struct ftrace_ops *op;
58435f151b24SSteven Rostedt (Red Hat) 	bool do_test = false;
58445f151b24SSteven Rostedt (Red Hat) 
58455f151b24SSteven Rostedt (Red Hat) 	/*
58465f151b24SSteven Rostedt (Red Hat) 	 * The graph and global ops share the same set of functions
58475f151b24SSteven Rostedt (Red Hat) 	 * to test. If any other ops is on the list, then
58485f151b24SSteven Rostedt (Red Hat) 	 * the graph tracing needs to test if its the function
58495f151b24SSteven Rostedt (Red Hat) 	 * it should call.
58505f151b24SSteven Rostedt (Red Hat) 	 */
58515f151b24SSteven Rostedt (Red Hat) 	do_for_each_ftrace_op(op, ftrace_ops_list) {
58525f151b24SSteven Rostedt (Red Hat) 		if (op != &global_ops && op != &graph_ops &&
58535f151b24SSteven Rostedt (Red Hat) 		    op != &ftrace_list_end) {
58545f151b24SSteven Rostedt (Red Hat) 			do_test = true;
58555f151b24SSteven Rostedt (Red Hat) 			/* in double loop, break out with goto */
58565f151b24SSteven Rostedt (Red Hat) 			goto out;
58575f151b24SSteven Rostedt (Red Hat) 		}
58585f151b24SSteven Rostedt (Red Hat) 	} while_for_each_ftrace_op(op);
58595f151b24SSteven Rostedt (Red Hat)  out:
58605f151b24SSteven Rostedt (Red Hat) 	if (do_test)
586123a8e844SSteven Rostedt (Red Hat) 		ftrace_graph_entry = ftrace_graph_entry_test;
58625f151b24SSteven Rostedt (Red Hat) 	else
58635f151b24SSteven Rostedt (Red Hat) 		ftrace_graph_entry = __ftrace_graph_entry;
586423a8e844SSteven Rostedt (Red Hat) }
586523a8e844SSteven Rostedt (Red Hat) 
58668275f69fSMathias Krause static struct notifier_block ftrace_suspend_notifier = {
58678275f69fSMathias Krause 	.notifier_call = ftrace_suspend_notifier_call,
58688275f69fSMathias Krause };
58698275f69fSMathias Krause 
5870287b6e68SFrederic Weisbecker int register_ftrace_graph(trace_func_graph_ret_t retfunc,
5871287b6e68SFrederic Weisbecker 			trace_func_graph_ent_t entryfunc)
587215e6cb36SFrederic Weisbecker {
5873e7d3737eSFrederic Weisbecker 	int ret = 0;
5874e7d3737eSFrederic Weisbecker 
5875e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
5876e7d3737eSFrederic Weisbecker 
587705ce5818SSteven Rostedt 	/* we currently allow only one tracer registered at a time */
5878597af815SSteven Rostedt 	if (ftrace_graph_active) {
587905ce5818SSteven Rostedt 		ret = -EBUSY;
588005ce5818SSteven Rostedt 		goto out;
588105ce5818SSteven Rostedt 	}
588205ce5818SSteven Rostedt 
58834a2b8ddaSFrederic Weisbecker 	register_pm_notifier(&ftrace_suspend_notifier);
58844a2b8ddaSFrederic Weisbecker 
5885597af815SSteven Rostedt 	ftrace_graph_active++;
5886fb52607aSFrederic Weisbecker 	ret = start_graph_tracing();
5887f201ae23SFrederic Weisbecker 	if (ret) {
5888597af815SSteven Rostedt 		ftrace_graph_active--;
5889f201ae23SFrederic Weisbecker 		goto out;
5890f201ae23SFrederic Weisbecker 	}
5891e53a6319SSteven Rostedt 
5892287b6e68SFrederic Weisbecker 	ftrace_graph_return = retfunc;
589323a8e844SSteven Rostedt (Red Hat) 
589423a8e844SSteven Rostedt (Red Hat) 	/*
589523a8e844SSteven Rostedt (Red Hat) 	 * Update the indirect function to the entryfunc, and the
589623a8e844SSteven Rostedt (Red Hat) 	 * function that gets called to the entry_test first. Then
589723a8e844SSteven Rostedt (Red Hat) 	 * call the update fgraph entry function to determine if
589823a8e844SSteven Rostedt (Red Hat) 	 * the entryfunc should be called directly or not.
589923a8e844SSteven Rostedt (Red Hat) 	 */
590023a8e844SSteven Rostedt (Red Hat) 	__ftrace_graph_entry = entryfunc;
590123a8e844SSteven Rostedt (Red Hat) 	ftrace_graph_entry = ftrace_graph_entry_test;
590223a8e844SSteven Rostedt (Red Hat) 	update_function_graph_func();
5903e53a6319SSteven Rostedt 
59045f151b24SSteven Rostedt (Red Hat) 	ret = ftrace_startup(&graph_ops, FTRACE_START_FUNC_RET);
5905e7d3737eSFrederic Weisbecker out:
5906e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
5907e7d3737eSFrederic Weisbecker 	return ret;
590815e6cb36SFrederic Weisbecker }
590915e6cb36SFrederic Weisbecker 
5910fb52607aSFrederic Weisbecker void unregister_ftrace_graph(void)
591115e6cb36SFrederic Weisbecker {
5912e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
5913e7d3737eSFrederic Weisbecker 
5914597af815SSteven Rostedt 	if (unlikely(!ftrace_graph_active))
59152aad1b76SSteven Rostedt 		goto out;
59162aad1b76SSteven Rostedt 
5917597af815SSteven Rostedt 	ftrace_graph_active--;
5918287b6e68SFrederic Weisbecker 	ftrace_graph_return = (trace_func_graph_ret_t)ftrace_stub;
5919e49dc19cSSteven Rostedt 	ftrace_graph_entry = ftrace_graph_entry_stub;
592023a8e844SSteven Rostedt (Red Hat) 	__ftrace_graph_entry = ftrace_graph_entry_stub;
59215f151b24SSteven Rostedt (Red Hat) 	ftrace_shutdown(&graph_ops, FTRACE_STOP_FUNC_RET);
59224a2b8ddaSFrederic Weisbecker 	unregister_pm_notifier(&ftrace_suspend_notifier);
592338516ab5SSteven Rostedt 	unregister_trace_sched_switch(ftrace_graph_probe_sched_switch, NULL);
5924e7d3737eSFrederic Weisbecker 
5925f3bea491SSteven Rostedt (Red Hat) #ifdef CONFIG_DYNAMIC_FTRACE
5926f3bea491SSteven Rostedt (Red Hat) 	/*
5927f3bea491SSteven Rostedt (Red Hat) 	 * Function graph does not allocate the trampoline, but
5928f3bea491SSteven Rostedt (Red Hat) 	 * other global_ops do. We need to reset the ALLOC_TRAMP flag
5929f3bea491SSteven Rostedt (Red Hat) 	 * if one was used.
5930f3bea491SSteven Rostedt (Red Hat) 	 */
5931f3bea491SSteven Rostedt (Red Hat) 	global_ops.trampoline = save_global_trampoline;
5932f3bea491SSteven Rostedt (Red Hat) 	if (save_global_flags & FTRACE_OPS_FL_ALLOC_TRAMP)
5933f3bea491SSteven Rostedt (Red Hat) 		global_ops.flags |= FTRACE_OPS_FL_ALLOC_TRAMP;
5934f3bea491SSteven Rostedt (Red Hat) #endif
5935f3bea491SSteven Rostedt (Red Hat) 
59362aad1b76SSteven Rostedt  out:
5937e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
593815e6cb36SFrederic Weisbecker }
5939f201ae23SFrederic Weisbecker 
5940868baf07SSteven Rostedt static DEFINE_PER_CPU(struct ftrace_ret_stack *, idle_ret_stack);
5941868baf07SSteven Rostedt 
5942868baf07SSteven Rostedt static void
5943868baf07SSteven Rostedt graph_init_task(struct task_struct *t, struct ftrace_ret_stack *ret_stack)
5944868baf07SSteven Rostedt {
5945868baf07SSteven Rostedt 	atomic_set(&t->tracing_graph_pause, 0);
5946868baf07SSteven Rostedt 	atomic_set(&t->trace_overrun, 0);
5947868baf07SSteven Rostedt 	t->ftrace_timestamp = 0;
594825985edcSLucas De Marchi 	/* make curr_ret_stack visible before we add the ret_stack */
5949868baf07SSteven Rostedt 	smp_wmb();
5950868baf07SSteven Rostedt 	t->ret_stack = ret_stack;
5951868baf07SSteven Rostedt }
5952868baf07SSteven Rostedt 
5953868baf07SSteven Rostedt /*
5954868baf07SSteven Rostedt  * Allocate a return stack for the idle task. May be the first
5955868baf07SSteven Rostedt  * time through, or it may be done by CPU hotplug online.
5956868baf07SSteven Rostedt  */
5957868baf07SSteven Rostedt void ftrace_graph_init_idle_task(struct task_struct *t, int cpu)
5958868baf07SSteven Rostedt {
5959868baf07SSteven Rostedt 	t->curr_ret_stack = -1;
5960868baf07SSteven Rostedt 	/*
5961868baf07SSteven Rostedt 	 * The idle task has no parent, it either has its own
5962868baf07SSteven Rostedt 	 * stack or no stack at all.
5963868baf07SSteven Rostedt 	 */
5964868baf07SSteven Rostedt 	if (t->ret_stack)
5965868baf07SSteven Rostedt 		WARN_ON(t->ret_stack != per_cpu(idle_ret_stack, cpu));
5966868baf07SSteven Rostedt 
5967868baf07SSteven Rostedt 	if (ftrace_graph_active) {
5968868baf07SSteven Rostedt 		struct ftrace_ret_stack *ret_stack;
5969868baf07SSteven Rostedt 
5970868baf07SSteven Rostedt 		ret_stack = per_cpu(idle_ret_stack, cpu);
5971868baf07SSteven Rostedt 		if (!ret_stack) {
5972868baf07SSteven Rostedt 			ret_stack = kmalloc(FTRACE_RETFUNC_DEPTH
5973868baf07SSteven Rostedt 					    * sizeof(struct ftrace_ret_stack),
5974868baf07SSteven Rostedt 					    GFP_KERNEL);
5975868baf07SSteven Rostedt 			if (!ret_stack)
5976868baf07SSteven Rostedt 				return;
5977868baf07SSteven Rostedt 			per_cpu(idle_ret_stack, cpu) = ret_stack;
5978868baf07SSteven Rostedt 		}
5979868baf07SSteven Rostedt 		graph_init_task(t, ret_stack);
5980868baf07SSteven Rostedt 	}
5981868baf07SSteven Rostedt }
5982868baf07SSteven Rostedt 
5983f201ae23SFrederic Weisbecker /* Allocate a return stack for newly created task */
5984fb52607aSFrederic Weisbecker void ftrace_graph_init_task(struct task_struct *t)
5985f201ae23SFrederic Weisbecker {
598684047e36SSteven Rostedt 	/* Make sure we do not use the parent ret_stack */
598784047e36SSteven Rostedt 	t->ret_stack = NULL;
5988ea14eb71SSteven Rostedt 	t->curr_ret_stack = -1;
598984047e36SSteven Rostedt 
5990597af815SSteven Rostedt 	if (ftrace_graph_active) {
599182310a32SSteven Rostedt 		struct ftrace_ret_stack *ret_stack;
599282310a32SSteven Rostedt 
599382310a32SSteven Rostedt 		ret_stack = kmalloc(FTRACE_RETFUNC_DEPTH
5994f201ae23SFrederic Weisbecker 				* sizeof(struct ftrace_ret_stack),
5995f201ae23SFrederic Weisbecker 				GFP_KERNEL);
599682310a32SSteven Rostedt 		if (!ret_stack)
5997f201ae23SFrederic Weisbecker 			return;
5998868baf07SSteven Rostedt 		graph_init_task(t, ret_stack);
599984047e36SSteven Rostedt 	}
6000f201ae23SFrederic Weisbecker }
6001f201ae23SFrederic Weisbecker 
6002fb52607aSFrederic Weisbecker void ftrace_graph_exit_task(struct task_struct *t)
6003f201ae23SFrederic Weisbecker {
6004eae849caSFrederic Weisbecker 	struct ftrace_ret_stack	*ret_stack = t->ret_stack;
6005eae849caSFrederic Weisbecker 
6006f201ae23SFrederic Weisbecker 	t->ret_stack = NULL;
6007eae849caSFrederic Weisbecker 	/* NULL must become visible to IRQs before we free it: */
6008eae849caSFrederic Weisbecker 	barrier();
6009eae849caSFrederic Weisbecker 
6010eae849caSFrederic Weisbecker 	kfree(ret_stack);
6011f201ae23SFrederic Weisbecker }
601215e6cb36SFrederic Weisbecker #endif
6013