xref: /linux-6.15/kernel/trace/ftrace.c (revision 3a36cb11)
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>
215072c59fSSteven Rostedt #include <linux/debugfs.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 
65e248491aSJiri Olsa #define FL_GLOBAL_CONTROL_MASK (FTRACE_OPS_FL_GLOBAL | FTRACE_OPS_FL_CONTROL)
66e248491aSJiri Olsa 
67f04f24fbSMasami Hiramatsu #ifdef CONFIG_DYNAMIC_FTRACE
68f04f24fbSMasami Hiramatsu #define INIT_REGEX_LOCK(opsname)	\
69f04f24fbSMasami Hiramatsu 	.regex_lock	= __MUTEX_INITIALIZER(opsname.regex_lock),
70f04f24fbSMasami Hiramatsu #else
71f04f24fbSMasami Hiramatsu #define INIT_REGEX_LOCK(opsname)
72f04f24fbSMasami Hiramatsu #endif
73f04f24fbSMasami Hiramatsu 
742f5f6ad9SSteven Rostedt static struct ftrace_ops ftrace_list_end __read_mostly = {
752f5f6ad9SSteven Rostedt 	.func		= ftrace_stub,
76395b97a3SSteven Rostedt (Red Hat) 	.flags		= FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_STUB,
772f5f6ad9SSteven Rostedt };
782f5f6ad9SSteven Rostedt 
794eebcc81SSteven Rostedt /* ftrace_enabled is a method to turn ftrace on or off */
804eebcc81SSteven Rostedt int ftrace_enabled __read_mostly;
81d61f82d0SSteven Rostedt static int last_ftrace_enabled;
82b0fc494fSSteven Rostedt 
8360a7ecf4SSteven Rostedt /* Quick disabling of function tracer. */
842f5f6ad9SSteven Rostedt int function_trace_stop __read_mostly;
852f5f6ad9SSteven Rostedt 
862f5f6ad9SSteven Rostedt /* Current function tracing op */
872f5f6ad9SSteven Rostedt struct ftrace_ops *function_trace_op __read_mostly = &ftrace_list_end;
88405e1d83SSteven Rostedt (Red Hat) /* What to set function_trace_op to */
89405e1d83SSteven Rostedt (Red Hat) static struct ftrace_ops *set_function_trace_op;
9060a7ecf4SSteven Rostedt 
91756d17eeS[email protected] /* List for set_ftrace_pid's pids. */
92756d17eeS[email protected] LIST_HEAD(ftrace_pids);
93756d17eeS[email protected] struct ftrace_pid {
94756d17eeS[email protected] 	struct list_head list;
95756d17eeS[email protected] 	struct pid *pid;
96756d17eeS[email protected] };
97756d17eeS[email protected] 
984eebcc81SSteven Rostedt /*
994eebcc81SSteven Rostedt  * ftrace_disabled is set when an anomaly is discovered.
1004eebcc81SSteven Rostedt  * ftrace_disabled is much stronger than ftrace_enabled.
1014eebcc81SSteven Rostedt  */
1024eebcc81SSteven Rostedt static int ftrace_disabled __read_mostly;
1034eebcc81SSteven Rostedt 
10452baf119SSteven Rostedt static DEFINE_MUTEX(ftrace_lock);
105b0fc494fSSteven Rostedt 
106b848914cSSteven Rostedt static struct ftrace_ops *ftrace_global_list __read_mostly = &ftrace_list_end;
107e248491aSJiri Olsa static struct ftrace_ops *ftrace_control_list __read_mostly = &ftrace_list_end;
108b848914cSSteven Rostedt static struct ftrace_ops *ftrace_ops_list __read_mostly = &ftrace_list_end;
10916444a8aSArnaldo Carvalho de Melo ftrace_func_t ftrace_trace_function __read_mostly = ftrace_stub;
110df4fc315SSteven Rostedt ftrace_func_t ftrace_pid_function __read_mostly = ftrace_stub;
1112b499381SSteven Rostedt static struct ftrace_ops global_ops;
112e248491aSJiri Olsa static struct ftrace_ops control_ops;
11316444a8aSArnaldo Carvalho de Melo 
1142f5f6ad9SSteven Rostedt #if ARCH_SUPPORTS_FTRACE_OPS
1152f5f6ad9SSteven Rostedt static void ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
116a1e2e31dSSteven Rostedt 				 struct ftrace_ops *op, struct pt_regs *regs);
1172f5f6ad9SSteven Rostedt #else
1182f5f6ad9SSteven Rostedt /* See comment below, where ftrace_ops_list_func is defined */
1192f5f6ad9SSteven Rostedt static void ftrace_ops_no_ops(unsigned long ip, unsigned long parent_ip);
1202f5f6ad9SSteven Rostedt #define ftrace_ops_list_func ((ftrace_func_t)ftrace_ops_no_ops)
1212f5f6ad9SSteven Rostedt #endif
122b848914cSSteven Rostedt 
1230a016409SSteven Rostedt /*
1240a016409SSteven Rostedt  * Traverse the ftrace_global_list, invoking all entries.  The reason that we
1251bb539caSSteven Rostedt  * can use rcu_dereference_raw_notrace() is that elements removed from this list
1260a016409SSteven Rostedt  * are simply leaked, so there is no need to interact with a grace-period
1271bb539caSSteven Rostedt  * mechanism.  The rcu_dereference_raw_notrace() calls are needed to handle
1280a016409SSteven Rostedt  * concurrent insertions into the ftrace_global_list.
1290a016409SSteven Rostedt  *
1300a016409SSteven Rostedt  * Silly Alpha and silly pointer-speculation compiler optimizations!
1310a016409SSteven Rostedt  */
1320a016409SSteven Rostedt #define do_for_each_ftrace_op(op, list)			\
1331bb539caSSteven Rostedt 	op = rcu_dereference_raw_notrace(list);			\
1340a016409SSteven Rostedt 	do
1350a016409SSteven Rostedt 
1360a016409SSteven Rostedt /*
1370a016409SSteven Rostedt  * Optimized for just a single item in the list (as that is the normal case).
1380a016409SSteven Rostedt  */
1390a016409SSteven Rostedt #define while_for_each_ftrace_op(op)				\
1401bb539caSSteven Rostedt 	while (likely(op = rcu_dereference_raw_notrace((op)->next)) &&	\
1410a016409SSteven Rostedt 	       unlikely((op) != &ftrace_list_end))
1420a016409SSteven Rostedt 
143f04f24fbSMasami Hiramatsu static inline void ftrace_ops_init(struct ftrace_ops *ops)
144f04f24fbSMasami Hiramatsu {
145f04f24fbSMasami Hiramatsu #ifdef CONFIG_DYNAMIC_FTRACE
146f04f24fbSMasami Hiramatsu 	if (!(ops->flags & FTRACE_OPS_FL_INITIALIZED)) {
147f04f24fbSMasami Hiramatsu 		mutex_init(&ops->regex_lock);
148f04f24fbSMasami Hiramatsu 		ops->flags |= FTRACE_OPS_FL_INITIALIZED;
149f04f24fbSMasami Hiramatsu 	}
150f04f24fbSMasami Hiramatsu #endif
151f04f24fbSMasami Hiramatsu }
152f04f24fbSMasami Hiramatsu 
153ea701f11SSteven Rostedt /**
154ea701f11SSteven Rostedt  * ftrace_nr_registered_ops - return number of ops registered
155ea701f11SSteven Rostedt  *
156ea701f11SSteven Rostedt  * Returns the number of ftrace_ops registered and tracing functions
157ea701f11SSteven Rostedt  */
158ea701f11SSteven Rostedt int ftrace_nr_registered_ops(void)
159ea701f11SSteven Rostedt {
160ea701f11SSteven Rostedt 	struct ftrace_ops *ops;
161ea701f11SSteven Rostedt 	int cnt = 0;
162ea701f11SSteven Rostedt 
163ea701f11SSteven Rostedt 	mutex_lock(&ftrace_lock);
164ea701f11SSteven Rostedt 
165ea701f11SSteven Rostedt 	for (ops = ftrace_ops_list;
166ea701f11SSteven Rostedt 	     ops != &ftrace_list_end; ops = ops->next)
167ea701f11SSteven Rostedt 		cnt++;
168ea701f11SSteven Rostedt 
169ea701f11SSteven Rostedt 	mutex_unlock(&ftrace_lock);
170ea701f11SSteven Rostedt 
171ea701f11SSteven Rostedt 	return cnt;
172ea701f11SSteven Rostedt }
173ea701f11SSteven Rostedt 
1742f5f6ad9SSteven Rostedt static void
1752f5f6ad9SSteven Rostedt ftrace_global_list_func(unsigned long ip, unsigned long parent_ip,
176a1e2e31dSSteven Rostedt 			struct ftrace_ops *op, struct pt_regs *regs)
17716444a8aSArnaldo Carvalho de Melo {
178c29f122cSSteven Rostedt 	int bit;
179c29f122cSSteven Rostedt 
180edc15cafSSteven Rostedt 	bit = trace_test_and_set_recursion(TRACE_GLOBAL_START, TRACE_GLOBAL_MAX);
181edc15cafSSteven Rostedt 	if (bit < 0)
182b1cff0adSSteven Rostedt 		return;
183b1cff0adSSteven Rostedt 
1840a016409SSteven Rostedt 	do_for_each_ftrace_op(op, ftrace_global_list) {
185a1e2e31dSSteven Rostedt 		op->func(ip, parent_ip, op, regs);
1860a016409SSteven Rostedt 	} while_for_each_ftrace_op(op);
187edc15cafSSteven Rostedt 
188edc15cafSSteven Rostedt 	trace_clear_recursion(bit);
18916444a8aSArnaldo Carvalho de Melo }
19016444a8aSArnaldo Carvalho de Melo 
1912f5f6ad9SSteven Rostedt static void ftrace_pid_func(unsigned long ip, unsigned long parent_ip,
192a1e2e31dSSteven Rostedt 			    struct ftrace_ops *op, struct pt_regs *regs)
193df4fc315SSteven Rostedt {
1940ef8cde5SSteven Rostedt 	if (!test_tsk_trace_trace(current))
195df4fc315SSteven Rostedt 		return;
196df4fc315SSteven Rostedt 
197a1e2e31dSSteven Rostedt 	ftrace_pid_function(ip, parent_ip, op, regs);
198df4fc315SSteven Rostedt }
199df4fc315SSteven Rostedt 
200df4fc315SSteven Rostedt static void set_ftrace_pid_function(ftrace_func_t func)
201df4fc315SSteven Rostedt {
202df4fc315SSteven Rostedt 	/* do not set ftrace_pid_function to itself! */
203df4fc315SSteven Rostedt 	if (func != ftrace_pid_func)
204df4fc315SSteven Rostedt 		ftrace_pid_function = func;
205df4fc315SSteven Rostedt }
206df4fc315SSteven Rostedt 
20716444a8aSArnaldo Carvalho de Melo /**
2083d083395SSteven Rostedt  * clear_ftrace_function - reset the ftrace function
20916444a8aSArnaldo Carvalho de Melo  *
2103d083395SSteven Rostedt  * This NULLs the ftrace function and in essence stops
2113d083395SSteven Rostedt  * tracing.  There may be lag
21216444a8aSArnaldo Carvalho de Melo  */
2133d083395SSteven Rostedt void clear_ftrace_function(void)
21416444a8aSArnaldo Carvalho de Melo {
2153d083395SSteven Rostedt 	ftrace_trace_function = ftrace_stub;
216df4fc315SSteven Rostedt 	ftrace_pid_function = ftrace_stub;
2173d083395SSteven Rostedt }
21816444a8aSArnaldo Carvalho de Melo 
219e248491aSJiri Olsa static void control_ops_disable_all(struct ftrace_ops *ops)
220e248491aSJiri Olsa {
221e248491aSJiri Olsa 	int cpu;
222e248491aSJiri Olsa 
223e248491aSJiri Olsa 	for_each_possible_cpu(cpu)
224e248491aSJiri Olsa 		*per_cpu_ptr(ops->disabled, cpu) = 1;
225e248491aSJiri Olsa }
226e248491aSJiri Olsa 
227e248491aSJiri Olsa static int control_ops_alloc(struct ftrace_ops *ops)
228e248491aSJiri Olsa {
229e248491aSJiri Olsa 	int __percpu *disabled;
230e248491aSJiri Olsa 
231e248491aSJiri Olsa 	disabled = alloc_percpu(int);
232e248491aSJiri Olsa 	if (!disabled)
233e248491aSJiri Olsa 		return -ENOMEM;
234e248491aSJiri Olsa 
235e248491aSJiri Olsa 	ops->disabled = disabled;
236e248491aSJiri Olsa 	control_ops_disable_all(ops);
237e248491aSJiri Olsa 	return 0;
238e248491aSJiri Olsa }
239e248491aSJiri Olsa 
240e248491aSJiri Olsa static void control_ops_free(struct ftrace_ops *ops)
241e248491aSJiri Olsa {
242e248491aSJiri Olsa 	free_percpu(ops->disabled);
243e248491aSJiri Olsa }
244e248491aSJiri Olsa 
2452b499381SSteven Rostedt static void update_global_ops(void)
246491d0dcfSSteven Rostedt {
247e6435e96SSteven Rostedt (Red Hat) 	ftrace_func_t func = ftrace_global_list_func;
248e6435e96SSteven Rostedt (Red Hat) 	void *private = NULL;
249e6435e96SSteven Rostedt (Red Hat) 
250e6435e96SSteven Rostedt (Red Hat) 	/* The list has its own recursion protection. */
251e6435e96SSteven Rostedt (Red Hat) 	global_ops.flags |= FTRACE_OPS_FL_RECURSION_SAFE;
252491d0dcfSSteven Rostedt 
253491d0dcfSSteven Rostedt 	/*
254491d0dcfSSteven Rostedt 	 * If there's only one function registered, then call that
255491d0dcfSSteven Rostedt 	 * function directly. Otherwise, we need to iterate over the
256491d0dcfSSteven Rostedt 	 * registered callers.
257491d0dcfSSteven Rostedt 	 */
258b848914cSSteven Rostedt 	if (ftrace_global_list == &ftrace_list_end ||
25963503794SSteven Rostedt 	    ftrace_global_list->next == &ftrace_list_end) {
260b848914cSSteven Rostedt 		func = ftrace_global_list->func;
261e6435e96SSteven Rostedt (Red Hat) 		private = ftrace_global_list->private;
26263503794SSteven Rostedt 		/*
26363503794SSteven Rostedt 		 * As we are calling the function directly.
26463503794SSteven Rostedt 		 * If it does not have recursion protection,
26563503794SSteven Rostedt 		 * the function_trace_op needs to be updated
26663503794SSteven Rostedt 		 * accordingly.
26763503794SSteven Rostedt 		 */
268e6435e96SSteven Rostedt (Red Hat) 		if (!(ftrace_global_list->flags & FTRACE_OPS_FL_RECURSION_SAFE))
26963503794SSteven Rostedt 			global_ops.flags &= ~FTRACE_OPS_FL_RECURSION_SAFE;
27063503794SSteven Rostedt 	}
27163503794SSteven Rostedt 
272491d0dcfSSteven Rostedt 	/* If we filter on pids, update to use the pid function */
273491d0dcfSSteven Rostedt 	if (!list_empty(&ftrace_pids)) {
274491d0dcfSSteven Rostedt 		set_ftrace_pid_function(func);
275491d0dcfSSteven Rostedt 		func = ftrace_pid_func;
276491d0dcfSSteven Rostedt 	}
2772b499381SSteven Rostedt 
2782b499381SSteven Rostedt 	global_ops.func = func;
279e6435e96SSteven Rostedt (Red Hat) 	global_ops.private = private;
2802b499381SSteven Rostedt }
2812b499381SSteven Rostedt 
282405e1d83SSteven Rostedt (Red Hat) static void ftrace_sync(struct work_struct *work)
283405e1d83SSteven Rostedt (Red Hat) {
284405e1d83SSteven Rostedt (Red Hat) 	/*
285405e1d83SSteven Rostedt (Red Hat) 	 * This function is just a stub to implement a hard force
286405e1d83SSteven Rostedt (Red Hat) 	 * of synchronize_sched(). This requires synchronizing
287405e1d83SSteven Rostedt (Red Hat) 	 * tasks even in userspace and idle.
288405e1d83SSteven Rostedt (Red Hat) 	 *
289405e1d83SSteven Rostedt (Red Hat) 	 * Yes, function tracing is rude.
290405e1d83SSteven Rostedt (Red Hat) 	 */
291405e1d83SSteven Rostedt (Red Hat) }
292405e1d83SSteven Rostedt (Red Hat) 
293405e1d83SSteven Rostedt (Red Hat) static void ftrace_sync_ipi(void *data)
294405e1d83SSteven Rostedt (Red Hat) {
295405e1d83SSteven Rostedt (Red Hat) 	/* Probably not needed, but do it anyway */
296405e1d83SSteven Rostedt (Red Hat) 	smp_rmb();
297405e1d83SSteven Rostedt (Red Hat) }
298405e1d83SSteven Rostedt (Red Hat) 
29923a8e844SSteven Rostedt (Red Hat) #ifdef CONFIG_FUNCTION_GRAPH_TRACER
30023a8e844SSteven Rostedt (Red Hat) static void update_function_graph_func(void);
30123a8e844SSteven Rostedt (Red Hat) #else
30223a8e844SSteven Rostedt (Red Hat) static inline void update_function_graph_func(void) { }
30323a8e844SSteven Rostedt (Red Hat) #endif
30423a8e844SSteven Rostedt (Red Hat) 
3052b499381SSteven Rostedt static void update_ftrace_function(void)
3062b499381SSteven Rostedt {
3072b499381SSteven Rostedt 	ftrace_func_t func;
3082b499381SSteven Rostedt 
3092b499381SSteven Rostedt 	update_global_ops();
3102b499381SSteven Rostedt 
311cdbe61bfSSteven Rostedt 	/*
312cdbe61bfSSteven Rostedt 	 * If we are at the end of the list and this ops is
3134740974aSSteven Rostedt 	 * recursion safe and not dynamic and the arch supports passing ops,
3144740974aSSteven Rostedt 	 * then have the mcount trampoline call the function directly.
315cdbe61bfSSteven Rostedt 	 */
316b848914cSSteven Rostedt 	if (ftrace_ops_list == &ftrace_list_end ||
317cdbe61bfSSteven Rostedt 	    (ftrace_ops_list->next == &ftrace_list_end &&
3182f5f6ad9SSteven Rostedt 	     !(ftrace_ops_list->flags & FTRACE_OPS_FL_DYNAMIC) &&
3194740974aSSteven Rostedt 	     (ftrace_ops_list->flags & FTRACE_OPS_FL_RECURSION_SAFE) &&
320ccf3672dSSteven Rostedt 	     !FTRACE_FORCE_LIST_FUNC)) {
3212f5f6ad9SSteven Rostedt 		/* Set the ftrace_ops that the arch callback uses */
3222f5f6ad9SSteven Rostedt 		if (ftrace_ops_list == &global_ops)
323405e1d83SSteven Rostedt (Red Hat) 			set_function_trace_op = ftrace_global_list;
324b848914cSSteven Rostedt 		else
325405e1d83SSteven Rostedt (Red Hat) 			set_function_trace_op = ftrace_ops_list;
3262f5f6ad9SSteven Rostedt 		func = ftrace_ops_list->func;
3272f5f6ad9SSteven Rostedt 	} else {
3282f5f6ad9SSteven Rostedt 		/* Just use the default ftrace_ops */
329405e1d83SSteven Rostedt (Red Hat) 		set_function_trace_op = &ftrace_list_end;
330b848914cSSteven Rostedt 		func = ftrace_ops_list_func;
3312f5f6ad9SSteven Rostedt 	}
3322b499381SSteven Rostedt 
333405e1d83SSteven Rostedt (Red Hat) 	/* If there's no change, then do nothing more here */
334405e1d83SSteven Rostedt (Red Hat) 	if (ftrace_trace_function == func)
335405e1d83SSteven Rostedt (Red Hat) 		return;
336405e1d83SSteven Rostedt (Red Hat) 
33723a8e844SSteven Rostedt (Red Hat) 	update_function_graph_func();
33823a8e844SSteven Rostedt (Red Hat) 
339405e1d83SSteven Rostedt (Red Hat) 	/*
340405e1d83SSteven Rostedt (Red Hat) 	 * If we are using the list function, it doesn't care
341405e1d83SSteven Rostedt (Red Hat) 	 * about the function_trace_ops.
342405e1d83SSteven Rostedt (Red Hat) 	 */
343405e1d83SSteven Rostedt (Red Hat) 	if (func == ftrace_ops_list_func) {
344405e1d83SSteven Rostedt (Red Hat) 		ftrace_trace_function = func;
345405e1d83SSteven Rostedt (Red Hat) 		/*
346405e1d83SSteven Rostedt (Red Hat) 		 * Don't even bother setting function_trace_ops,
347405e1d83SSteven Rostedt (Red Hat) 		 * it would be racy to do so anyway.
348405e1d83SSteven Rostedt (Red Hat) 		 */
349405e1d83SSteven Rostedt (Red Hat) 		return;
350405e1d83SSteven Rostedt (Red Hat) 	}
351405e1d83SSteven Rostedt (Red Hat) 
352405e1d83SSteven Rostedt (Red Hat) #ifndef CONFIG_DYNAMIC_FTRACE
353405e1d83SSteven Rostedt (Red Hat) 	/*
354405e1d83SSteven Rostedt (Red Hat) 	 * For static tracing, we need to be a bit more careful.
355405e1d83SSteven Rostedt (Red Hat) 	 * The function change takes affect immediately. Thus,
356405e1d83SSteven Rostedt (Red Hat) 	 * we need to coorditate the setting of the function_trace_ops
357405e1d83SSteven Rostedt (Red Hat) 	 * with the setting of the ftrace_trace_function.
358405e1d83SSteven Rostedt (Red Hat) 	 *
359405e1d83SSteven Rostedt (Red Hat) 	 * Set the function to the list ops, which will call the
360405e1d83SSteven Rostedt (Red Hat) 	 * function we want, albeit indirectly, but it handles the
361405e1d83SSteven Rostedt (Red Hat) 	 * ftrace_ops and doesn't depend on function_trace_op.
362405e1d83SSteven Rostedt (Red Hat) 	 */
363405e1d83SSteven Rostedt (Red Hat) 	ftrace_trace_function = ftrace_ops_list_func;
364405e1d83SSteven Rostedt (Red Hat) 	/*
365405e1d83SSteven Rostedt (Red Hat) 	 * Make sure all CPUs see this. Yes this is slow, but static
366405e1d83SSteven Rostedt (Red Hat) 	 * tracing is slow and nasty to have enabled.
367405e1d83SSteven Rostedt (Red Hat) 	 */
368405e1d83SSteven Rostedt (Red Hat) 	schedule_on_each_cpu(ftrace_sync);
369405e1d83SSteven Rostedt (Red Hat) 	/* Now all cpus are using the list ops. */
370405e1d83SSteven Rostedt (Red Hat) 	function_trace_op = set_function_trace_op;
371405e1d83SSteven Rostedt (Red Hat) 	/* Make sure the function_trace_op is visible on all CPUs */
372405e1d83SSteven Rostedt (Red Hat) 	smp_wmb();
373405e1d83SSteven Rostedt (Red Hat) 	/* Nasty way to force a rmb on all cpus */
374405e1d83SSteven Rostedt (Red Hat) 	smp_call_function(ftrace_sync_ipi, NULL, 1);
375405e1d83SSteven Rostedt (Red Hat) 	/* OK, we are all set to update the ftrace_trace_function now! */
376405e1d83SSteven Rostedt (Red Hat) #endif /* !CONFIG_DYNAMIC_FTRACE */
377405e1d83SSteven Rostedt (Red Hat) 
378491d0dcfSSteven Rostedt 	ftrace_trace_function = func;
379491d0dcfSSteven Rostedt }
380491d0dcfSSteven Rostedt 
3812b499381SSteven Rostedt static void add_ftrace_ops(struct ftrace_ops **list, struct ftrace_ops *ops)
3823d083395SSteven Rostedt {
3832b499381SSteven Rostedt 	ops->next = *list;
38416444a8aSArnaldo Carvalho de Melo 	/*
385b848914cSSteven Rostedt 	 * We are entering ops into the list but another
38616444a8aSArnaldo Carvalho de Melo 	 * CPU might be walking that list. We need to make sure
38716444a8aSArnaldo Carvalho de Melo 	 * the ops->next pointer is valid before another CPU sees
388b848914cSSteven Rostedt 	 * the ops pointer included into the list.
38916444a8aSArnaldo Carvalho de Melo 	 */
3902b499381SSteven Rostedt 	rcu_assign_pointer(*list, ops);
3912b499381SSteven Rostedt }
3923d083395SSteven Rostedt 
3932b499381SSteven Rostedt static int remove_ftrace_ops(struct ftrace_ops **list, struct ftrace_ops *ops)
3942b499381SSteven Rostedt {
3952b499381SSteven Rostedt 	struct ftrace_ops **p;
3962b499381SSteven Rostedt 
3972b499381SSteven Rostedt 	/*
3982b499381SSteven Rostedt 	 * If we are removing the last function, then simply point
3992b499381SSteven Rostedt 	 * to the ftrace_stub.
4002b499381SSteven Rostedt 	 */
4012b499381SSteven Rostedt 	if (*list == ops && ops->next == &ftrace_list_end) {
4022b499381SSteven Rostedt 		*list = &ftrace_list_end;
4032b499381SSteven Rostedt 		return 0;
4042b499381SSteven Rostedt 	}
4052b499381SSteven Rostedt 
4062b499381SSteven Rostedt 	for (p = list; *p != &ftrace_list_end; p = &(*p)->next)
4072b499381SSteven Rostedt 		if (*p == ops)
4082b499381SSteven Rostedt 			break;
4092b499381SSteven Rostedt 
4102b499381SSteven Rostedt 	if (*p != ops)
4112b499381SSteven Rostedt 		return -1;
4122b499381SSteven Rostedt 
4132b499381SSteven Rostedt 	*p = (*p)->next;
4142b499381SSteven Rostedt 	return 0;
4152b499381SSteven Rostedt }
4162b499381SSteven Rostedt 
417e248491aSJiri Olsa static void add_ftrace_list_ops(struct ftrace_ops **list,
418e248491aSJiri Olsa 				struct ftrace_ops *main_ops,
419e248491aSJiri Olsa 				struct ftrace_ops *ops)
420e248491aSJiri Olsa {
421e248491aSJiri Olsa 	int first = *list == &ftrace_list_end;
422e248491aSJiri Olsa 	add_ftrace_ops(list, ops);
423e248491aSJiri Olsa 	if (first)
424e248491aSJiri Olsa 		add_ftrace_ops(&ftrace_ops_list, main_ops);
425e248491aSJiri Olsa }
426e248491aSJiri Olsa 
427e248491aSJiri Olsa static int remove_ftrace_list_ops(struct ftrace_ops **list,
428e248491aSJiri Olsa 				  struct ftrace_ops *main_ops,
429e248491aSJiri Olsa 				  struct ftrace_ops *ops)
430e248491aSJiri Olsa {
431e248491aSJiri Olsa 	int ret = remove_ftrace_ops(list, ops);
432e248491aSJiri Olsa 	if (!ret && *list == &ftrace_list_end)
433e248491aSJiri Olsa 		ret = remove_ftrace_ops(&ftrace_ops_list, main_ops);
434e248491aSJiri Olsa 	return ret;
435e248491aSJiri Olsa }
436e248491aSJiri Olsa 
4372b499381SSteven Rostedt static int __register_ftrace_function(struct ftrace_ops *ops)
4382b499381SSteven Rostedt {
439591dffdaSSteven Rostedt (Red Hat) 	if (ops->flags & FTRACE_OPS_FL_DELETED)
440591dffdaSSteven Rostedt (Red Hat) 		return -EINVAL;
441591dffdaSSteven Rostedt (Red Hat) 
4422b499381SSteven Rostedt 	if (FTRACE_WARN_ON(ops == &global_ops))
4432b499381SSteven Rostedt 		return -EINVAL;
4442b499381SSteven Rostedt 
445b848914cSSteven Rostedt 	if (WARN_ON(ops->flags & FTRACE_OPS_FL_ENABLED))
446b848914cSSteven Rostedt 		return -EBUSY;
447b848914cSSteven Rostedt 
448e248491aSJiri Olsa 	/* We don't support both control and global flags set. */
449e248491aSJiri Olsa 	if ((ops->flags & FL_GLOBAL_CONTROL_MASK) == FL_GLOBAL_CONTROL_MASK)
450e248491aSJiri Olsa 		return -EINVAL;
451e248491aSJiri Olsa 
45206aeaaeaSMasami Hiramatsu #ifndef CONFIG_DYNAMIC_FTRACE_WITH_REGS
45308f6fba5SSteven Rostedt 	/*
45408f6fba5SSteven Rostedt 	 * If the ftrace_ops specifies SAVE_REGS, then it only can be used
45508f6fba5SSteven Rostedt 	 * if the arch supports it, or SAVE_REGS_IF_SUPPORTED is also set.
45608f6fba5SSteven Rostedt 	 * Setting SAVE_REGS_IF_SUPPORTED makes SAVE_REGS irrelevant.
45708f6fba5SSteven Rostedt 	 */
45808f6fba5SSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_SAVE_REGS &&
45908f6fba5SSteven Rostedt 	    !(ops->flags & FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED))
46008f6fba5SSteven Rostedt 		return -EINVAL;
46108f6fba5SSteven Rostedt 
46208f6fba5SSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_SAVE_REGS_IF_SUPPORTED)
46308f6fba5SSteven Rostedt 		ops->flags |= FTRACE_OPS_FL_SAVE_REGS;
46408f6fba5SSteven Rostedt #endif
46508f6fba5SSteven Rostedt 
466cdbe61bfSSteven Rostedt 	if (!core_kernel_data((unsigned long)ops))
467cdbe61bfSSteven Rostedt 		ops->flags |= FTRACE_OPS_FL_DYNAMIC;
468cdbe61bfSSteven Rostedt 
469b848914cSSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_GLOBAL) {
470e248491aSJiri Olsa 		add_ftrace_list_ops(&ftrace_global_list, &global_ops, ops);
471b848914cSSteven Rostedt 		ops->flags |= FTRACE_OPS_FL_ENABLED;
472e248491aSJiri Olsa 	} else if (ops->flags & FTRACE_OPS_FL_CONTROL) {
473e248491aSJiri Olsa 		if (control_ops_alloc(ops))
474e248491aSJiri Olsa 			return -ENOMEM;
475e248491aSJiri Olsa 		add_ftrace_list_ops(&ftrace_control_list, &control_ops, ops);
476b848914cSSteven Rostedt 	} else
477b848914cSSteven Rostedt 		add_ftrace_ops(&ftrace_ops_list, ops);
478b848914cSSteven Rostedt 
479491d0dcfSSteven Rostedt 	if (ftrace_enabled)
480491d0dcfSSteven Rostedt 		update_ftrace_function();
4813d083395SSteven Rostedt 
48216444a8aSArnaldo Carvalho de Melo 	return 0;
48316444a8aSArnaldo Carvalho de Melo }
48416444a8aSArnaldo Carvalho de Melo 
485e309b41dSIngo Molnar static int __unregister_ftrace_function(struct ftrace_ops *ops)
48616444a8aSArnaldo Carvalho de Melo {
4872b499381SSteven Rostedt 	int ret;
48816444a8aSArnaldo Carvalho de Melo 
489b848914cSSteven Rostedt 	if (WARN_ON(!(ops->flags & FTRACE_OPS_FL_ENABLED)))
490b848914cSSteven Rostedt 		return -EBUSY;
491b848914cSSteven Rostedt 
4922b499381SSteven Rostedt 	if (FTRACE_WARN_ON(ops == &global_ops))
4932b499381SSteven Rostedt 		return -EINVAL;
49416444a8aSArnaldo Carvalho de Melo 
495b848914cSSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_GLOBAL) {
496e248491aSJiri Olsa 		ret = remove_ftrace_list_ops(&ftrace_global_list,
497e248491aSJiri Olsa 					     &global_ops, ops);
498b848914cSSteven Rostedt 		if (!ret)
499b848914cSSteven Rostedt 			ops->flags &= ~FTRACE_OPS_FL_ENABLED;
500e248491aSJiri Olsa 	} else if (ops->flags & FTRACE_OPS_FL_CONTROL) {
501e248491aSJiri Olsa 		ret = remove_ftrace_list_ops(&ftrace_control_list,
502e248491aSJiri Olsa 					     &control_ops, ops);
503b848914cSSteven Rostedt 	} else
504b848914cSSteven Rostedt 		ret = remove_ftrace_ops(&ftrace_ops_list, ops);
505b848914cSSteven Rostedt 
5062b499381SSteven Rostedt 	if (ret < 0)
5072b499381SSteven Rostedt 		return ret;
508b848914cSSteven Rostedt 
509491d0dcfSSteven Rostedt 	if (ftrace_enabled)
510491d0dcfSSteven Rostedt 		update_ftrace_function();
51116444a8aSArnaldo Carvalho de Melo 
512e6ea44e9SSteven Rostedt 	return 0;
5133d083395SSteven Rostedt }
5143d083395SSteven Rostedt 
515df4fc315SSteven Rostedt static void ftrace_update_pid_func(void)
516df4fc315SSteven Rostedt {
517491d0dcfSSteven Rostedt 	/* Only do something if we are tracing something */
518df4fc315SSteven Rostedt 	if (ftrace_trace_function == ftrace_stub)
51910dd3ebeSKOSAKI Motohiro 		return;
520df4fc315SSteven Rostedt 
521491d0dcfSSteven Rostedt 	update_ftrace_function();
522df4fc315SSteven Rostedt }
523df4fc315SSteven Rostedt 
524493762fcSSteven Rostedt #ifdef CONFIG_FUNCTION_PROFILER
525493762fcSSteven Rostedt struct ftrace_profile {
526493762fcSSteven Rostedt 	struct hlist_node		node;
527493762fcSSteven Rostedt 	unsigned long			ip;
528493762fcSSteven Rostedt 	unsigned long			counter;
5290706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
5300706f1c4SSteven Rostedt 	unsigned long long		time;
531e330b3bcSChase Douglas 	unsigned long long		time_squared;
5320706f1c4SSteven Rostedt #endif
533493762fcSSteven Rostedt };
534493762fcSSteven Rostedt 
535493762fcSSteven Rostedt struct ftrace_profile_page {
536493762fcSSteven Rostedt 	struct ftrace_profile_page	*next;
537493762fcSSteven Rostedt 	unsigned long			index;
538493762fcSSteven Rostedt 	struct ftrace_profile		records[];
539493762fcSSteven Rostedt };
540493762fcSSteven Rostedt 
541cafb168aSSteven Rostedt struct ftrace_profile_stat {
542cafb168aSSteven Rostedt 	atomic_t			disabled;
543cafb168aSSteven Rostedt 	struct hlist_head		*hash;
544cafb168aSSteven Rostedt 	struct ftrace_profile_page	*pages;
545cafb168aSSteven Rostedt 	struct ftrace_profile_page	*start;
546cafb168aSSteven Rostedt 	struct tracer_stat		stat;
547cafb168aSSteven Rostedt };
548cafb168aSSteven Rostedt 
549493762fcSSteven Rostedt #define PROFILE_RECORDS_SIZE						\
550493762fcSSteven Rostedt 	(PAGE_SIZE - offsetof(struct ftrace_profile_page, records))
551493762fcSSteven Rostedt 
552493762fcSSteven Rostedt #define PROFILES_PER_PAGE					\
553493762fcSSteven Rostedt 	(PROFILE_RECORDS_SIZE / sizeof(struct ftrace_profile))
554493762fcSSteven Rostedt 
555fb9fb015SSteven Rostedt static int ftrace_profile_enabled __read_mostly;
556fb9fb015SSteven Rostedt 
557fb9fb015SSteven Rostedt /* ftrace_profile_lock - synchronize the enable and disable of the profiler */
558493762fcSSteven Rostedt static DEFINE_MUTEX(ftrace_profile_lock);
559493762fcSSteven Rostedt 
560cafb168aSSteven Rostedt static DEFINE_PER_CPU(struct ftrace_profile_stat, ftrace_profile_stats);
561493762fcSSteven Rostedt 
56220079ebeSNamhyung Kim #define FTRACE_PROFILE_HASH_BITS 10
56320079ebeSNamhyung Kim #define FTRACE_PROFILE_HASH_SIZE (1 << FTRACE_PROFILE_HASH_BITS)
564493762fcSSteven Rostedt 
565493762fcSSteven Rostedt static void *
566493762fcSSteven Rostedt function_stat_next(void *v, int idx)
567493762fcSSteven Rostedt {
568493762fcSSteven Rostedt 	struct ftrace_profile *rec = v;
569493762fcSSteven Rostedt 	struct ftrace_profile_page *pg;
570493762fcSSteven Rostedt 
571493762fcSSteven Rostedt 	pg = (struct ftrace_profile_page *)((unsigned long)rec & PAGE_MASK);
572493762fcSSteven Rostedt 
573493762fcSSteven Rostedt  again:
5740296e425SLi Zefan 	if (idx != 0)
575493762fcSSteven Rostedt 		rec++;
5760296e425SLi Zefan 
577493762fcSSteven Rostedt 	if ((void *)rec >= (void *)&pg->records[pg->index]) {
578493762fcSSteven Rostedt 		pg = pg->next;
579493762fcSSteven Rostedt 		if (!pg)
580493762fcSSteven Rostedt 			return NULL;
581493762fcSSteven Rostedt 		rec = &pg->records[0];
582493762fcSSteven Rostedt 		if (!rec->counter)
583493762fcSSteven Rostedt 			goto again;
584493762fcSSteven Rostedt 	}
585493762fcSSteven Rostedt 
586493762fcSSteven Rostedt 	return rec;
587493762fcSSteven Rostedt }
588493762fcSSteven Rostedt 
589493762fcSSteven Rostedt static void *function_stat_start(struct tracer_stat *trace)
590493762fcSSteven Rostedt {
591cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat =
592cafb168aSSteven Rostedt 		container_of(trace, struct ftrace_profile_stat, stat);
593cafb168aSSteven Rostedt 
594cafb168aSSteven Rostedt 	if (!stat || !stat->start)
595cafb168aSSteven Rostedt 		return NULL;
596cafb168aSSteven Rostedt 
597cafb168aSSteven Rostedt 	return function_stat_next(&stat->start->records[0], 0);
598493762fcSSteven Rostedt }
599493762fcSSteven Rostedt 
6000706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
6010706f1c4SSteven Rostedt /* function graph compares on total time */
6020706f1c4SSteven Rostedt static int function_stat_cmp(void *p1, void *p2)
6030706f1c4SSteven Rostedt {
6040706f1c4SSteven Rostedt 	struct ftrace_profile *a = p1;
6050706f1c4SSteven Rostedt 	struct ftrace_profile *b = p2;
6060706f1c4SSteven Rostedt 
6070706f1c4SSteven Rostedt 	if (a->time < b->time)
6080706f1c4SSteven Rostedt 		return -1;
6090706f1c4SSteven Rostedt 	if (a->time > b->time)
6100706f1c4SSteven Rostedt 		return 1;
6110706f1c4SSteven Rostedt 	else
6120706f1c4SSteven Rostedt 		return 0;
6130706f1c4SSteven Rostedt }
6140706f1c4SSteven Rostedt #else
6150706f1c4SSteven Rostedt /* not function graph compares against hits */
616493762fcSSteven Rostedt static int function_stat_cmp(void *p1, void *p2)
617493762fcSSteven Rostedt {
618493762fcSSteven Rostedt 	struct ftrace_profile *a = p1;
619493762fcSSteven Rostedt 	struct ftrace_profile *b = p2;
620493762fcSSteven Rostedt 
621493762fcSSteven Rostedt 	if (a->counter < b->counter)
622493762fcSSteven Rostedt 		return -1;
623493762fcSSteven Rostedt 	if (a->counter > b->counter)
624493762fcSSteven Rostedt 		return 1;
625493762fcSSteven Rostedt 	else
626493762fcSSteven Rostedt 		return 0;
627493762fcSSteven Rostedt }
6280706f1c4SSteven Rostedt #endif
629493762fcSSteven Rostedt 
630493762fcSSteven Rostedt static int function_stat_headers(struct seq_file *m)
631493762fcSSteven Rostedt {
6320706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
63334886c8bSSteven Rostedt 	seq_printf(m, "  Function                               "
634e330b3bcSChase Douglas 		   "Hit    Time            Avg             s^2\n"
63534886c8bSSteven Rostedt 		      "  --------                               "
636e330b3bcSChase Douglas 		   "---    ----            ---             ---\n");
6370706f1c4SSteven Rostedt #else
638493762fcSSteven Rostedt 	seq_printf(m, "  Function                               Hit\n"
639493762fcSSteven Rostedt 		      "  --------                               ---\n");
6400706f1c4SSteven Rostedt #endif
641493762fcSSteven Rostedt 	return 0;
642493762fcSSteven Rostedt }
643493762fcSSteven Rostedt 
644493762fcSSteven Rostedt static int function_stat_show(struct seq_file *m, void *v)
645493762fcSSteven Rostedt {
646493762fcSSteven Rostedt 	struct ftrace_profile *rec = v;
647493762fcSSteven Rostedt 	char str[KSYM_SYMBOL_LEN];
6483aaba20fSLi Zefan 	int ret = 0;
6490706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
65034886c8bSSteven Rostedt 	static struct trace_seq s;
65134886c8bSSteven Rostedt 	unsigned long long avg;
652e330b3bcSChase Douglas 	unsigned long long stddev;
6530706f1c4SSteven Rostedt #endif
6543aaba20fSLi Zefan 	mutex_lock(&ftrace_profile_lock);
6553aaba20fSLi Zefan 
6563aaba20fSLi Zefan 	/* we raced with function_profile_reset() */
6573aaba20fSLi Zefan 	if (unlikely(rec->counter == 0)) {
6583aaba20fSLi Zefan 		ret = -EBUSY;
6593aaba20fSLi Zefan 		goto out;
6603aaba20fSLi Zefan 	}
661493762fcSSteven Rostedt 
662493762fcSSteven Rostedt 	kallsyms_lookup(rec->ip, NULL, NULL, NULL, str);
6630706f1c4SSteven Rostedt 	seq_printf(m, "  %-30.30s  %10lu", str, rec->counter);
664493762fcSSteven Rostedt 
6650706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
6660706f1c4SSteven Rostedt 	seq_printf(m, "    ");
66734886c8bSSteven Rostedt 	avg = rec->time;
66834886c8bSSteven Rostedt 	do_div(avg, rec->counter);
66934886c8bSSteven Rostedt 
670e330b3bcSChase Douglas 	/* Sample standard deviation (s^2) */
671e330b3bcSChase Douglas 	if (rec->counter <= 1)
672e330b3bcSChase Douglas 		stddev = 0;
673e330b3bcSChase Douglas 	else {
67452d85d76SJuri Lelli 		/*
67552d85d76SJuri Lelli 		 * Apply Welford's method:
67652d85d76SJuri Lelli 		 * s^2 = 1 / (n * (n-1)) * (n * \Sum (x_i)^2 - (\Sum x_i)^2)
67752d85d76SJuri Lelli 		 */
67852d85d76SJuri Lelli 		stddev = rec->counter * rec->time_squared -
67952d85d76SJuri Lelli 			 rec->time * rec->time;
68052d85d76SJuri Lelli 
681e330b3bcSChase Douglas 		/*
682e330b3bcSChase Douglas 		 * Divide only 1000 for ns^2 -> us^2 conversion.
683e330b3bcSChase Douglas 		 * trace_print_graph_duration will divide 1000 again.
684e330b3bcSChase Douglas 		 */
68552d85d76SJuri Lelli 		do_div(stddev, rec->counter * (rec->counter - 1) * 1000);
686e330b3bcSChase Douglas 	}
687e330b3bcSChase Douglas 
68834886c8bSSteven Rostedt 	trace_seq_init(&s);
68934886c8bSSteven Rostedt 	trace_print_graph_duration(rec->time, &s);
69034886c8bSSteven Rostedt 	trace_seq_puts(&s, "    ");
69134886c8bSSteven Rostedt 	trace_print_graph_duration(avg, &s);
692e330b3bcSChase Douglas 	trace_seq_puts(&s, "    ");
693e330b3bcSChase Douglas 	trace_print_graph_duration(stddev, &s);
6940706f1c4SSteven Rostedt 	trace_print_seq(m, &s);
6950706f1c4SSteven Rostedt #endif
6960706f1c4SSteven Rostedt 	seq_putc(m, '\n');
6973aaba20fSLi Zefan out:
6983aaba20fSLi Zefan 	mutex_unlock(&ftrace_profile_lock);
6990706f1c4SSteven Rostedt 
7003aaba20fSLi Zefan 	return ret;
701493762fcSSteven Rostedt }
702493762fcSSteven Rostedt 
703cafb168aSSteven Rostedt static void ftrace_profile_reset(struct ftrace_profile_stat *stat)
704493762fcSSteven Rostedt {
705493762fcSSteven Rostedt 	struct ftrace_profile_page *pg;
706493762fcSSteven Rostedt 
707cafb168aSSteven Rostedt 	pg = stat->pages = stat->start;
708493762fcSSteven Rostedt 
709493762fcSSteven Rostedt 	while (pg) {
710493762fcSSteven Rostedt 		memset(pg->records, 0, PROFILE_RECORDS_SIZE);
711493762fcSSteven Rostedt 		pg->index = 0;
712493762fcSSteven Rostedt 		pg = pg->next;
713493762fcSSteven Rostedt 	}
714493762fcSSteven Rostedt 
715cafb168aSSteven Rostedt 	memset(stat->hash, 0,
716493762fcSSteven Rostedt 	       FTRACE_PROFILE_HASH_SIZE * sizeof(struct hlist_head));
717493762fcSSteven Rostedt }
718493762fcSSteven Rostedt 
719cafb168aSSteven Rostedt int ftrace_profile_pages_init(struct ftrace_profile_stat *stat)
720493762fcSSteven Rostedt {
721493762fcSSteven Rostedt 	struct ftrace_profile_page *pg;
722318e0a73SSteven Rostedt 	int functions;
723318e0a73SSteven Rostedt 	int pages;
724493762fcSSteven Rostedt 	int i;
725493762fcSSteven Rostedt 
726493762fcSSteven Rostedt 	/* If we already allocated, do nothing */
727cafb168aSSteven Rostedt 	if (stat->pages)
728493762fcSSteven Rostedt 		return 0;
729493762fcSSteven Rostedt 
730cafb168aSSteven Rostedt 	stat->pages = (void *)get_zeroed_page(GFP_KERNEL);
731cafb168aSSteven Rostedt 	if (!stat->pages)
732493762fcSSteven Rostedt 		return -ENOMEM;
733493762fcSSteven Rostedt 
734318e0a73SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
735318e0a73SSteven Rostedt 	functions = ftrace_update_tot_cnt;
736318e0a73SSteven Rostedt #else
737318e0a73SSteven Rostedt 	/*
738318e0a73SSteven Rostedt 	 * We do not know the number of functions that exist because
739318e0a73SSteven Rostedt 	 * dynamic tracing is what counts them. With past experience
740318e0a73SSteven Rostedt 	 * we have around 20K functions. That should be more than enough.
741318e0a73SSteven Rostedt 	 * It is highly unlikely we will execute every function in
742318e0a73SSteven Rostedt 	 * the kernel.
743318e0a73SSteven Rostedt 	 */
744318e0a73SSteven Rostedt 	functions = 20000;
745318e0a73SSteven Rostedt #endif
746318e0a73SSteven Rostedt 
747cafb168aSSteven Rostedt 	pg = stat->start = stat->pages;
748493762fcSSteven Rostedt 
749318e0a73SSteven Rostedt 	pages = DIV_ROUND_UP(functions, PROFILES_PER_PAGE);
750318e0a73SSteven Rostedt 
75139e30cd1SNamhyung Kim 	for (i = 1; i < pages; i++) {
752493762fcSSteven Rostedt 		pg->next = (void *)get_zeroed_page(GFP_KERNEL);
753493762fcSSteven Rostedt 		if (!pg->next)
754318e0a73SSteven Rostedt 			goto out_free;
755493762fcSSteven Rostedt 		pg = pg->next;
756493762fcSSteven Rostedt 	}
757493762fcSSteven Rostedt 
758493762fcSSteven Rostedt 	return 0;
759318e0a73SSteven Rostedt 
760318e0a73SSteven Rostedt  out_free:
761318e0a73SSteven Rostedt 	pg = stat->start;
762318e0a73SSteven Rostedt 	while (pg) {
763318e0a73SSteven Rostedt 		unsigned long tmp = (unsigned long)pg;
764318e0a73SSteven Rostedt 
765318e0a73SSteven Rostedt 		pg = pg->next;
766318e0a73SSteven Rostedt 		free_page(tmp);
767318e0a73SSteven Rostedt 	}
768318e0a73SSteven Rostedt 
769318e0a73SSteven Rostedt 	stat->pages = NULL;
770318e0a73SSteven Rostedt 	stat->start = NULL;
771318e0a73SSteven Rostedt 
772318e0a73SSteven Rostedt 	return -ENOMEM;
773493762fcSSteven Rostedt }
774493762fcSSteven Rostedt 
775cafb168aSSteven Rostedt static int ftrace_profile_init_cpu(int cpu)
776493762fcSSteven Rostedt {
777cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat;
778493762fcSSteven Rostedt 	int size;
779493762fcSSteven Rostedt 
780cafb168aSSteven Rostedt 	stat = &per_cpu(ftrace_profile_stats, cpu);
781cafb168aSSteven Rostedt 
782cafb168aSSteven Rostedt 	if (stat->hash) {
783493762fcSSteven Rostedt 		/* If the profile is already created, simply reset it */
784cafb168aSSteven Rostedt 		ftrace_profile_reset(stat);
785493762fcSSteven Rostedt 		return 0;
786493762fcSSteven Rostedt 	}
787493762fcSSteven Rostedt 
788493762fcSSteven Rostedt 	/*
789493762fcSSteven Rostedt 	 * We are profiling all functions, but usually only a few thousand
790493762fcSSteven Rostedt 	 * functions are hit. We'll make a hash of 1024 items.
791493762fcSSteven Rostedt 	 */
792493762fcSSteven Rostedt 	size = FTRACE_PROFILE_HASH_SIZE;
793493762fcSSteven Rostedt 
794cafb168aSSteven Rostedt 	stat->hash = kzalloc(sizeof(struct hlist_head) * size, GFP_KERNEL);
795493762fcSSteven Rostedt 
796cafb168aSSteven Rostedt 	if (!stat->hash)
797493762fcSSteven Rostedt 		return -ENOMEM;
798493762fcSSteven Rostedt 
799318e0a73SSteven Rostedt 	/* Preallocate the function profiling pages */
800cafb168aSSteven Rostedt 	if (ftrace_profile_pages_init(stat) < 0) {
801cafb168aSSteven Rostedt 		kfree(stat->hash);
802cafb168aSSteven Rostedt 		stat->hash = NULL;
803493762fcSSteven Rostedt 		return -ENOMEM;
804493762fcSSteven Rostedt 	}
805493762fcSSteven Rostedt 
806493762fcSSteven Rostedt 	return 0;
807493762fcSSteven Rostedt }
808493762fcSSteven Rostedt 
809cafb168aSSteven Rostedt static int ftrace_profile_init(void)
810cafb168aSSteven Rostedt {
811cafb168aSSteven Rostedt 	int cpu;
812cafb168aSSteven Rostedt 	int ret = 0;
813cafb168aSSteven Rostedt 
814c4602c1cSMiao Xie 	for_each_possible_cpu(cpu) {
815cafb168aSSteven Rostedt 		ret = ftrace_profile_init_cpu(cpu);
816cafb168aSSteven Rostedt 		if (ret)
817cafb168aSSteven Rostedt 			break;
818cafb168aSSteven Rostedt 	}
819cafb168aSSteven Rostedt 
820cafb168aSSteven Rostedt 	return ret;
821cafb168aSSteven Rostedt }
822cafb168aSSteven Rostedt 
823493762fcSSteven Rostedt /* interrupts must be disabled */
824cafb168aSSteven Rostedt static struct ftrace_profile *
825cafb168aSSteven Rostedt ftrace_find_profiled_func(struct ftrace_profile_stat *stat, unsigned long ip)
826493762fcSSteven Rostedt {
827493762fcSSteven Rostedt 	struct ftrace_profile *rec;
828493762fcSSteven Rostedt 	struct hlist_head *hhd;
829493762fcSSteven Rostedt 	unsigned long key;
830493762fcSSteven Rostedt 
83120079ebeSNamhyung Kim 	key = hash_long(ip, FTRACE_PROFILE_HASH_BITS);
832cafb168aSSteven Rostedt 	hhd = &stat->hash[key];
833493762fcSSteven Rostedt 
834493762fcSSteven Rostedt 	if (hlist_empty(hhd))
835493762fcSSteven Rostedt 		return NULL;
836493762fcSSteven Rostedt 
8371bb539caSSteven Rostedt 	hlist_for_each_entry_rcu_notrace(rec, hhd, node) {
838493762fcSSteven Rostedt 		if (rec->ip == ip)
839493762fcSSteven Rostedt 			return rec;
840493762fcSSteven Rostedt 	}
841493762fcSSteven Rostedt 
842493762fcSSteven Rostedt 	return NULL;
843493762fcSSteven Rostedt }
844493762fcSSteven Rostedt 
845cafb168aSSteven Rostedt static void ftrace_add_profile(struct ftrace_profile_stat *stat,
846cafb168aSSteven Rostedt 			       struct ftrace_profile *rec)
847493762fcSSteven Rostedt {
848493762fcSSteven Rostedt 	unsigned long key;
849493762fcSSteven Rostedt 
85020079ebeSNamhyung Kim 	key = hash_long(rec->ip, FTRACE_PROFILE_HASH_BITS);
851cafb168aSSteven Rostedt 	hlist_add_head_rcu(&rec->node, &stat->hash[key]);
852493762fcSSteven Rostedt }
853493762fcSSteven Rostedt 
854318e0a73SSteven Rostedt /*
855318e0a73SSteven Rostedt  * The memory is already allocated, this simply finds a new record to use.
856318e0a73SSteven Rostedt  */
857493762fcSSteven Rostedt static struct ftrace_profile *
858318e0a73SSteven Rostedt ftrace_profile_alloc(struct ftrace_profile_stat *stat, unsigned long ip)
859493762fcSSteven Rostedt {
860493762fcSSteven Rostedt 	struct ftrace_profile *rec = NULL;
861493762fcSSteven Rostedt 
862318e0a73SSteven Rostedt 	/* prevent recursion (from NMIs) */
863cafb168aSSteven Rostedt 	if (atomic_inc_return(&stat->disabled) != 1)
864493762fcSSteven Rostedt 		goto out;
865493762fcSSteven Rostedt 
866493762fcSSteven Rostedt 	/*
867318e0a73SSteven Rostedt 	 * Try to find the function again since an NMI
868318e0a73SSteven Rostedt 	 * could have added it
869493762fcSSteven Rostedt 	 */
870cafb168aSSteven Rostedt 	rec = ftrace_find_profiled_func(stat, ip);
871493762fcSSteven Rostedt 	if (rec)
872cafb168aSSteven Rostedt 		goto out;
873493762fcSSteven Rostedt 
874cafb168aSSteven Rostedt 	if (stat->pages->index == PROFILES_PER_PAGE) {
875cafb168aSSteven Rostedt 		if (!stat->pages->next)
876cafb168aSSteven Rostedt 			goto out;
877cafb168aSSteven Rostedt 		stat->pages = stat->pages->next;
878493762fcSSteven Rostedt 	}
879493762fcSSteven Rostedt 
880cafb168aSSteven Rostedt 	rec = &stat->pages->records[stat->pages->index++];
881493762fcSSteven Rostedt 	rec->ip = ip;
882cafb168aSSteven Rostedt 	ftrace_add_profile(stat, rec);
883493762fcSSteven Rostedt 
884493762fcSSteven Rostedt  out:
885cafb168aSSteven Rostedt 	atomic_dec(&stat->disabled);
886493762fcSSteven Rostedt 
887493762fcSSteven Rostedt 	return rec;
888493762fcSSteven Rostedt }
889493762fcSSteven Rostedt 
890493762fcSSteven Rostedt static void
8912f5f6ad9SSteven Rostedt function_profile_call(unsigned long ip, unsigned long parent_ip,
892a1e2e31dSSteven Rostedt 		      struct ftrace_ops *ops, struct pt_regs *regs)
893493762fcSSteven Rostedt {
894cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat;
895493762fcSSteven Rostedt 	struct ftrace_profile *rec;
896493762fcSSteven Rostedt 	unsigned long flags;
897493762fcSSteven Rostedt 
898493762fcSSteven Rostedt 	if (!ftrace_profile_enabled)
899493762fcSSteven Rostedt 		return;
900493762fcSSteven Rostedt 
901493762fcSSteven Rostedt 	local_irq_save(flags);
902cafb168aSSteven Rostedt 
903cafb168aSSteven Rostedt 	stat = &__get_cpu_var(ftrace_profile_stats);
9040f6ce3deSSteven Rostedt 	if (!stat->hash || !ftrace_profile_enabled)
905cafb168aSSteven Rostedt 		goto out;
906cafb168aSSteven Rostedt 
907cafb168aSSteven Rostedt 	rec = ftrace_find_profiled_func(stat, ip);
908493762fcSSteven Rostedt 	if (!rec) {
909318e0a73SSteven Rostedt 		rec = ftrace_profile_alloc(stat, ip);
910493762fcSSteven Rostedt 		if (!rec)
911493762fcSSteven Rostedt 			goto out;
912493762fcSSteven Rostedt 	}
913493762fcSSteven Rostedt 
914493762fcSSteven Rostedt 	rec->counter++;
915493762fcSSteven Rostedt  out:
916493762fcSSteven Rostedt 	local_irq_restore(flags);
917493762fcSSteven Rostedt }
918493762fcSSteven Rostedt 
9190706f1c4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
9200706f1c4SSteven Rostedt static int profile_graph_entry(struct ftrace_graph_ent *trace)
9210706f1c4SSteven Rostedt {
922a1e2e31dSSteven Rostedt 	function_profile_call(trace->func, 0, NULL, NULL);
9230706f1c4SSteven Rostedt 	return 1;
9240706f1c4SSteven Rostedt }
9250706f1c4SSteven Rostedt 
9260706f1c4SSteven Rostedt static void profile_graph_return(struct ftrace_graph_ret *trace)
9270706f1c4SSteven Rostedt {
928cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat;
929a2a16d6aSSteven Rostedt 	unsigned long long calltime;
9300706f1c4SSteven Rostedt 	struct ftrace_profile *rec;
931cafb168aSSteven Rostedt 	unsigned long flags;
9320706f1c4SSteven Rostedt 
9330706f1c4SSteven Rostedt 	local_irq_save(flags);
934cafb168aSSteven Rostedt 	stat = &__get_cpu_var(ftrace_profile_stats);
9350f6ce3deSSteven Rostedt 	if (!stat->hash || !ftrace_profile_enabled)
936cafb168aSSteven Rostedt 		goto out;
937cafb168aSSteven Rostedt 
93837e44bc5SSteven Rostedt 	/* If the calltime was zero'd ignore it */
93937e44bc5SSteven Rostedt 	if (!trace->calltime)
94037e44bc5SSteven Rostedt 		goto out;
94137e44bc5SSteven Rostedt 
942a2a16d6aSSteven Rostedt 	calltime = trace->rettime - trace->calltime;
943a2a16d6aSSteven Rostedt 
944a2a16d6aSSteven Rostedt 	if (!(trace_flags & TRACE_ITER_GRAPH_TIME)) {
945a2a16d6aSSteven Rostedt 		int index;
946a2a16d6aSSteven Rostedt 
947a2a16d6aSSteven Rostedt 		index = trace->depth;
948a2a16d6aSSteven Rostedt 
949a2a16d6aSSteven Rostedt 		/* Append this call time to the parent time to subtract */
950a2a16d6aSSteven Rostedt 		if (index)
951a2a16d6aSSteven Rostedt 			current->ret_stack[index - 1].subtime += calltime;
952a2a16d6aSSteven Rostedt 
953a2a16d6aSSteven Rostedt 		if (current->ret_stack[index].subtime < calltime)
954a2a16d6aSSteven Rostedt 			calltime -= current->ret_stack[index].subtime;
955a2a16d6aSSteven Rostedt 		else
956a2a16d6aSSteven Rostedt 			calltime = 0;
957a2a16d6aSSteven Rostedt 	}
958a2a16d6aSSteven Rostedt 
959cafb168aSSteven Rostedt 	rec = ftrace_find_profiled_func(stat, trace->func);
960e330b3bcSChase Douglas 	if (rec) {
961a2a16d6aSSteven Rostedt 		rec->time += calltime;
962e330b3bcSChase Douglas 		rec->time_squared += calltime * calltime;
963e330b3bcSChase Douglas 	}
964a2a16d6aSSteven Rostedt 
965cafb168aSSteven Rostedt  out:
9660706f1c4SSteven Rostedt 	local_irq_restore(flags);
9670706f1c4SSteven Rostedt }
9680706f1c4SSteven Rostedt 
9690706f1c4SSteven Rostedt static int register_ftrace_profiler(void)
9700706f1c4SSteven Rostedt {
9710706f1c4SSteven Rostedt 	return register_ftrace_graph(&profile_graph_return,
9720706f1c4SSteven Rostedt 				     &profile_graph_entry);
9730706f1c4SSteven Rostedt }
9740706f1c4SSteven Rostedt 
9750706f1c4SSteven Rostedt static void unregister_ftrace_profiler(void)
9760706f1c4SSteven Rostedt {
9770706f1c4SSteven Rostedt 	unregister_ftrace_graph();
9780706f1c4SSteven Rostedt }
9790706f1c4SSteven Rostedt #else
980bd38c0e6SPaul McQuade static struct ftrace_ops ftrace_profile_ops __read_mostly = {
981493762fcSSteven Rostedt 	.func		= function_profile_call,
982f04f24fbSMasami Hiramatsu 	.flags		= FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_INITIALIZED,
983f04f24fbSMasami Hiramatsu 	INIT_REGEX_LOCK(ftrace_profile_ops)
984493762fcSSteven Rostedt };
985493762fcSSteven Rostedt 
9860706f1c4SSteven Rostedt static int register_ftrace_profiler(void)
9870706f1c4SSteven Rostedt {
9880706f1c4SSteven Rostedt 	return register_ftrace_function(&ftrace_profile_ops);
9890706f1c4SSteven Rostedt }
9900706f1c4SSteven Rostedt 
9910706f1c4SSteven Rostedt static void unregister_ftrace_profiler(void)
9920706f1c4SSteven Rostedt {
9930706f1c4SSteven Rostedt 	unregister_ftrace_function(&ftrace_profile_ops);
9940706f1c4SSteven Rostedt }
9950706f1c4SSteven Rostedt #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
9960706f1c4SSteven Rostedt 
997493762fcSSteven Rostedt static ssize_t
998493762fcSSteven Rostedt ftrace_profile_write(struct file *filp, const char __user *ubuf,
999493762fcSSteven Rostedt 		     size_t cnt, loff_t *ppos)
1000493762fcSSteven Rostedt {
1001493762fcSSteven Rostedt 	unsigned long val;
1002493762fcSSteven Rostedt 	int ret;
1003493762fcSSteven Rostedt 
100422fe9b54SPeter Huewe 	ret = kstrtoul_from_user(ubuf, cnt, 10, &val);
100522fe9b54SPeter Huewe 	if (ret)
1006493762fcSSteven Rostedt 		return ret;
1007493762fcSSteven Rostedt 
1008493762fcSSteven Rostedt 	val = !!val;
1009493762fcSSteven Rostedt 
1010493762fcSSteven Rostedt 	mutex_lock(&ftrace_profile_lock);
1011493762fcSSteven Rostedt 	if (ftrace_profile_enabled ^ val) {
1012493762fcSSteven Rostedt 		if (val) {
1013493762fcSSteven Rostedt 			ret = ftrace_profile_init();
1014493762fcSSteven Rostedt 			if (ret < 0) {
1015493762fcSSteven Rostedt 				cnt = ret;
1016493762fcSSteven Rostedt 				goto out;
1017493762fcSSteven Rostedt 			}
1018493762fcSSteven Rostedt 
10190706f1c4SSteven Rostedt 			ret = register_ftrace_profiler();
10200706f1c4SSteven Rostedt 			if (ret < 0) {
10210706f1c4SSteven Rostedt 				cnt = ret;
10220706f1c4SSteven Rostedt 				goto out;
10230706f1c4SSteven Rostedt 			}
1024493762fcSSteven Rostedt 			ftrace_profile_enabled = 1;
1025493762fcSSteven Rostedt 		} else {
1026493762fcSSteven Rostedt 			ftrace_profile_enabled = 0;
10270f6ce3deSSteven Rostedt 			/*
10280f6ce3deSSteven Rostedt 			 * unregister_ftrace_profiler calls stop_machine
10290f6ce3deSSteven Rostedt 			 * so this acts like an synchronize_sched.
10300f6ce3deSSteven Rostedt 			 */
10310706f1c4SSteven Rostedt 			unregister_ftrace_profiler();
1032493762fcSSteven Rostedt 		}
1033493762fcSSteven Rostedt 	}
1034493762fcSSteven Rostedt  out:
1035493762fcSSteven Rostedt 	mutex_unlock(&ftrace_profile_lock);
1036493762fcSSteven Rostedt 
1037cf8517cfSJiri Olsa 	*ppos += cnt;
1038493762fcSSteven Rostedt 
1039493762fcSSteven Rostedt 	return cnt;
1040493762fcSSteven Rostedt }
1041493762fcSSteven Rostedt 
1042493762fcSSteven Rostedt static ssize_t
1043493762fcSSteven Rostedt ftrace_profile_read(struct file *filp, char __user *ubuf,
1044493762fcSSteven Rostedt 		     size_t cnt, loff_t *ppos)
1045493762fcSSteven Rostedt {
1046fb9fb015SSteven Rostedt 	char buf[64];		/* big enough to hold a number */
1047493762fcSSteven Rostedt 	int r;
1048493762fcSSteven Rostedt 
1049493762fcSSteven Rostedt 	r = sprintf(buf, "%u\n", ftrace_profile_enabled);
1050493762fcSSteven Rostedt 	return simple_read_from_buffer(ubuf, cnt, ppos, buf, r);
1051493762fcSSteven Rostedt }
1052493762fcSSteven Rostedt 
1053493762fcSSteven Rostedt static const struct file_operations ftrace_profile_fops = {
1054493762fcSSteven Rostedt 	.open		= tracing_open_generic,
1055493762fcSSteven Rostedt 	.read		= ftrace_profile_read,
1056493762fcSSteven Rostedt 	.write		= ftrace_profile_write,
10576038f373SArnd Bergmann 	.llseek		= default_llseek,
1058493762fcSSteven Rostedt };
1059493762fcSSteven Rostedt 
1060cafb168aSSteven Rostedt /* used to initialize the real stat files */
1061cafb168aSSteven Rostedt static struct tracer_stat function_stats __initdata = {
1062cafb168aSSteven Rostedt 	.name		= "functions",
1063cafb168aSSteven Rostedt 	.stat_start	= function_stat_start,
1064cafb168aSSteven Rostedt 	.stat_next	= function_stat_next,
1065cafb168aSSteven Rostedt 	.stat_cmp	= function_stat_cmp,
1066cafb168aSSteven Rostedt 	.stat_headers	= function_stat_headers,
1067cafb168aSSteven Rostedt 	.stat_show	= function_stat_show
1068cafb168aSSteven Rostedt };
1069cafb168aSSteven Rostedt 
10706ab5d668SSteven Rostedt static __init void ftrace_profile_debugfs(struct dentry *d_tracer)
1071493762fcSSteven Rostedt {
1072cafb168aSSteven Rostedt 	struct ftrace_profile_stat *stat;
1073493762fcSSteven Rostedt 	struct dentry *entry;
1074cafb168aSSteven Rostedt 	char *name;
1075493762fcSSteven Rostedt 	int ret;
1076cafb168aSSteven Rostedt 	int cpu;
1077493762fcSSteven Rostedt 
1078cafb168aSSteven Rostedt 	for_each_possible_cpu(cpu) {
1079cafb168aSSteven Rostedt 		stat = &per_cpu(ftrace_profile_stats, cpu);
1080cafb168aSSteven Rostedt 
1081cafb168aSSteven Rostedt 		/* allocate enough for function name + cpu number */
1082cafb168aSSteven Rostedt 		name = kmalloc(32, GFP_KERNEL);
1083cafb168aSSteven Rostedt 		if (!name) {
1084cafb168aSSteven Rostedt 			/*
1085cafb168aSSteven Rostedt 			 * The files created are permanent, if something happens
1086cafb168aSSteven Rostedt 			 * we still do not free memory.
1087cafb168aSSteven Rostedt 			 */
1088cafb168aSSteven Rostedt 			WARN(1,
1089cafb168aSSteven Rostedt 			     "Could not allocate stat file for cpu %d\n",
1090cafb168aSSteven Rostedt 			     cpu);
1091493762fcSSteven Rostedt 			return;
1092493762fcSSteven Rostedt 		}
1093cafb168aSSteven Rostedt 		stat->stat = function_stats;
1094cafb168aSSteven Rostedt 		snprintf(name, 32, "function%d", cpu);
1095cafb168aSSteven Rostedt 		stat->stat.name = name;
1096cafb168aSSteven Rostedt 		ret = register_stat_tracer(&stat->stat);
1097cafb168aSSteven Rostedt 		if (ret) {
1098cafb168aSSteven Rostedt 			WARN(1,
1099cafb168aSSteven Rostedt 			     "Could not register function stat for cpu %d\n",
1100cafb168aSSteven Rostedt 			     cpu);
1101cafb168aSSteven Rostedt 			kfree(name);
1102cafb168aSSteven Rostedt 			return;
1103cafb168aSSteven Rostedt 		}
1104cafb168aSSteven Rostedt 	}
1105493762fcSSteven Rostedt 
1106493762fcSSteven Rostedt 	entry = debugfs_create_file("function_profile_enabled", 0644,
1107493762fcSSteven Rostedt 				    d_tracer, NULL, &ftrace_profile_fops);
1108493762fcSSteven Rostedt 	if (!entry)
1109493762fcSSteven Rostedt 		pr_warning("Could not create debugfs "
1110493762fcSSteven Rostedt 			   "'function_profile_enabled' entry\n");
1111493762fcSSteven Rostedt }
1112493762fcSSteven Rostedt 
1113493762fcSSteven Rostedt #else /* CONFIG_FUNCTION_PROFILER */
11146ab5d668SSteven Rostedt static __init void ftrace_profile_debugfs(struct dentry *d_tracer)
1115493762fcSSteven Rostedt {
1116493762fcSSteven Rostedt }
1117493762fcSSteven Rostedt #endif /* CONFIG_FUNCTION_PROFILER */
1118493762fcSSteven Rostedt 
111973d3fd96SIngo Molnar static struct pid * const ftrace_swapper_pid = &init_struct_pid;
112073d3fd96SIngo Molnar 
11213d083395SSteven Rostedt #ifdef CONFIG_DYNAMIC_FTRACE
112273d3fd96SIngo Molnar 
112399ecdc43SSteven Rostedt #ifndef CONFIG_FTRACE_MCOUNT_RECORD
1124cb7be3b2SSteven Rostedt # error Dynamic ftrace depends on MCOUNT_RECORD
112599ecdc43SSteven Rostedt #endif
112699ecdc43SSteven Rostedt 
11278fc0c701SSteven Rostedt static struct hlist_head ftrace_func_hash[FTRACE_FUNC_HASHSIZE] __read_mostly;
11288fc0c701SSteven Rostedt 
1129b6887d79SSteven Rostedt struct ftrace_func_probe {
11308fc0c701SSteven Rostedt 	struct hlist_node	node;
1131b6887d79SSteven Rostedt 	struct ftrace_probe_ops	*ops;
11328fc0c701SSteven Rostedt 	unsigned long		flags;
11338fc0c701SSteven Rostedt 	unsigned long		ip;
11348fc0c701SSteven Rostedt 	void			*data;
11357818b388SSteven Rostedt (Red Hat) 	struct list_head	free_list;
11368fc0c701SSteven Rostedt };
11378fc0c701SSteven Rostedt 
1138b448c4e3SSteven Rostedt struct ftrace_func_entry {
1139b448c4e3SSteven Rostedt 	struct hlist_node hlist;
1140b448c4e3SSteven Rostedt 	unsigned long ip;
1141b448c4e3SSteven Rostedt };
1142b448c4e3SSteven Rostedt 
1143b448c4e3SSteven Rostedt struct ftrace_hash {
1144b448c4e3SSteven Rostedt 	unsigned long		size_bits;
1145b448c4e3SSteven Rostedt 	struct hlist_head	*buckets;
1146b448c4e3SSteven Rostedt 	unsigned long		count;
114707fd5515SSteven Rostedt 	struct rcu_head		rcu;
1148b448c4e3SSteven Rostedt };
1149b448c4e3SSteven Rostedt 
115033dc9b12SSteven Rostedt /*
115133dc9b12SSteven Rostedt  * We make these constant because no one should touch them,
115233dc9b12SSteven Rostedt  * but they are used as the default "empty hash", to avoid allocating
115333dc9b12SSteven Rostedt  * it all the time. These are in a read only section such that if
115433dc9b12SSteven Rostedt  * anyone does try to modify it, it will cause an exception.
115533dc9b12SSteven Rostedt  */
115633dc9b12SSteven Rostedt static const struct hlist_head empty_buckets[1];
115733dc9b12SSteven Rostedt static const struct ftrace_hash empty_hash = {
115833dc9b12SSteven Rostedt 	.buckets = (struct hlist_head *)empty_buckets,
1159b448c4e3SSteven Rostedt };
116033dc9b12SSteven Rostedt #define EMPTY_HASH	((struct ftrace_hash *)&empty_hash)
11615072c59fSSteven Rostedt 
11622b499381SSteven Rostedt static struct ftrace_ops global_ops = {
1163f45948e8SSteven Rostedt 	.func			= ftrace_stub,
116433dc9b12SSteven Rostedt 	.notrace_hash		= EMPTY_HASH,
116533dc9b12SSteven Rostedt 	.filter_hash		= EMPTY_HASH,
1166f04f24fbSMasami Hiramatsu 	.flags			= FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_INITIALIZED,
1167f04f24fbSMasami Hiramatsu 	INIT_REGEX_LOCK(global_ops)
1168f45948e8SSteven Rostedt };
1169f45948e8SSteven Rostedt 
11703c1720f0SSteven Rostedt struct ftrace_page {
11713c1720f0SSteven Rostedt 	struct ftrace_page	*next;
1172a7900875SSteven Rostedt 	struct dyn_ftrace	*records;
1173431aa3fbSSteven Rostedt 	int			index;
1174a7900875SSteven Rostedt 	int			size;
1175aa5e5ceaSDavid Miller };
11763c1720f0SSteven Rostedt 
1177a7900875SSteven Rostedt #define ENTRY_SIZE sizeof(struct dyn_ftrace)
1178a7900875SSteven Rostedt #define ENTRIES_PER_PAGE (PAGE_SIZE / ENTRY_SIZE)
11793c1720f0SSteven Rostedt 
11803c1720f0SSteven Rostedt /* estimate from running different kernels */
11813c1720f0SSteven Rostedt #define NR_TO_INIT		10000
11823c1720f0SSteven Rostedt 
11833c1720f0SSteven Rostedt static struct ftrace_page	*ftrace_pages_start;
11843c1720f0SSteven Rostedt static struct ftrace_page	*ftrace_pages;
11853c1720f0SSteven Rostedt 
118606a51d93SSteven Rostedt static bool ftrace_hash_empty(struct ftrace_hash *hash)
118706a51d93SSteven Rostedt {
118806a51d93SSteven Rostedt 	return !hash || !hash->count;
118906a51d93SSteven Rostedt }
119006a51d93SSteven Rostedt 
1191b448c4e3SSteven Rostedt static struct ftrace_func_entry *
1192b448c4e3SSteven Rostedt ftrace_lookup_ip(struct ftrace_hash *hash, unsigned long ip)
1193b448c4e3SSteven Rostedt {
1194b448c4e3SSteven Rostedt 	unsigned long key;
1195b448c4e3SSteven Rostedt 	struct ftrace_func_entry *entry;
1196b448c4e3SSteven Rostedt 	struct hlist_head *hhd;
1197b448c4e3SSteven Rostedt 
119806a51d93SSteven Rostedt 	if (ftrace_hash_empty(hash))
1199b448c4e3SSteven Rostedt 		return NULL;
1200b448c4e3SSteven Rostedt 
1201b448c4e3SSteven Rostedt 	if (hash->size_bits > 0)
1202b448c4e3SSteven Rostedt 		key = hash_long(ip, hash->size_bits);
1203b448c4e3SSteven Rostedt 	else
1204b448c4e3SSteven Rostedt 		key = 0;
1205b448c4e3SSteven Rostedt 
1206b448c4e3SSteven Rostedt 	hhd = &hash->buckets[key];
1207b448c4e3SSteven Rostedt 
12081bb539caSSteven Rostedt 	hlist_for_each_entry_rcu_notrace(entry, hhd, hlist) {
1209b448c4e3SSteven Rostedt 		if (entry->ip == ip)
1210b448c4e3SSteven Rostedt 			return entry;
1211b448c4e3SSteven Rostedt 	}
1212b448c4e3SSteven Rostedt 	return NULL;
1213b448c4e3SSteven Rostedt }
1214b448c4e3SSteven Rostedt 
121533dc9b12SSteven Rostedt static void __add_hash_entry(struct ftrace_hash *hash,
121633dc9b12SSteven Rostedt 			     struct ftrace_func_entry *entry)
121733dc9b12SSteven Rostedt {
121833dc9b12SSteven Rostedt 	struct hlist_head *hhd;
121933dc9b12SSteven Rostedt 	unsigned long key;
122033dc9b12SSteven Rostedt 
122133dc9b12SSteven Rostedt 	if (hash->size_bits)
122233dc9b12SSteven Rostedt 		key = hash_long(entry->ip, hash->size_bits);
122333dc9b12SSteven Rostedt 	else
122433dc9b12SSteven Rostedt 		key = 0;
122533dc9b12SSteven Rostedt 
122633dc9b12SSteven Rostedt 	hhd = &hash->buckets[key];
122733dc9b12SSteven Rostedt 	hlist_add_head(&entry->hlist, hhd);
122833dc9b12SSteven Rostedt 	hash->count++;
122933dc9b12SSteven Rostedt }
123033dc9b12SSteven Rostedt 
1231b448c4e3SSteven Rostedt static int add_hash_entry(struct ftrace_hash *hash, unsigned long ip)
1232b448c4e3SSteven Rostedt {
1233b448c4e3SSteven Rostedt 	struct ftrace_func_entry *entry;
1234b448c4e3SSteven Rostedt 
1235b448c4e3SSteven Rostedt 	entry = kmalloc(sizeof(*entry), GFP_KERNEL);
1236b448c4e3SSteven Rostedt 	if (!entry)
1237b448c4e3SSteven Rostedt 		return -ENOMEM;
1238b448c4e3SSteven Rostedt 
1239b448c4e3SSteven Rostedt 	entry->ip = ip;
124033dc9b12SSteven Rostedt 	__add_hash_entry(hash, entry);
1241b448c4e3SSteven Rostedt 
1242b448c4e3SSteven Rostedt 	return 0;
1243b448c4e3SSteven Rostedt }
1244b448c4e3SSteven Rostedt 
1245b448c4e3SSteven Rostedt static void
124633dc9b12SSteven Rostedt free_hash_entry(struct ftrace_hash *hash,
124733dc9b12SSteven Rostedt 		  struct ftrace_func_entry *entry)
124833dc9b12SSteven Rostedt {
124933dc9b12SSteven Rostedt 	hlist_del(&entry->hlist);
125033dc9b12SSteven Rostedt 	kfree(entry);
125133dc9b12SSteven Rostedt 	hash->count--;
125233dc9b12SSteven Rostedt }
125333dc9b12SSteven Rostedt 
125433dc9b12SSteven Rostedt static void
1255b448c4e3SSteven Rostedt remove_hash_entry(struct ftrace_hash *hash,
1256b448c4e3SSteven Rostedt 		  struct ftrace_func_entry *entry)
1257b448c4e3SSteven Rostedt {
1258b448c4e3SSteven Rostedt 	hlist_del(&entry->hlist);
1259b448c4e3SSteven Rostedt 	hash->count--;
1260b448c4e3SSteven Rostedt }
1261b448c4e3SSteven Rostedt 
1262b448c4e3SSteven Rostedt static void ftrace_hash_clear(struct ftrace_hash *hash)
1263b448c4e3SSteven Rostedt {
1264b448c4e3SSteven Rostedt 	struct hlist_head *hhd;
1265b67bfe0dSSasha Levin 	struct hlist_node *tn;
1266b448c4e3SSteven Rostedt 	struct ftrace_func_entry *entry;
1267b448c4e3SSteven Rostedt 	int size = 1 << hash->size_bits;
1268b448c4e3SSteven Rostedt 	int i;
1269b448c4e3SSteven Rostedt 
127033dc9b12SSteven Rostedt 	if (!hash->count)
127133dc9b12SSteven Rostedt 		return;
127233dc9b12SSteven Rostedt 
1273b448c4e3SSteven Rostedt 	for (i = 0; i < size; i++) {
1274b448c4e3SSteven Rostedt 		hhd = &hash->buckets[i];
1275b67bfe0dSSasha Levin 		hlist_for_each_entry_safe(entry, tn, hhd, hlist)
127633dc9b12SSteven Rostedt 			free_hash_entry(hash, entry);
1277b448c4e3SSteven Rostedt 	}
1278b448c4e3SSteven Rostedt 	FTRACE_WARN_ON(hash->count);
1279b448c4e3SSteven Rostedt }
1280b448c4e3SSteven Rostedt 
128133dc9b12SSteven Rostedt static void free_ftrace_hash(struct ftrace_hash *hash)
128233dc9b12SSteven Rostedt {
128333dc9b12SSteven Rostedt 	if (!hash || hash == EMPTY_HASH)
128433dc9b12SSteven Rostedt 		return;
128533dc9b12SSteven Rostedt 	ftrace_hash_clear(hash);
128633dc9b12SSteven Rostedt 	kfree(hash->buckets);
128733dc9b12SSteven Rostedt 	kfree(hash);
128833dc9b12SSteven Rostedt }
128933dc9b12SSteven Rostedt 
129007fd5515SSteven Rostedt static void __free_ftrace_hash_rcu(struct rcu_head *rcu)
129107fd5515SSteven Rostedt {
129207fd5515SSteven Rostedt 	struct ftrace_hash *hash;
129307fd5515SSteven Rostedt 
129407fd5515SSteven Rostedt 	hash = container_of(rcu, struct ftrace_hash, rcu);
129507fd5515SSteven Rostedt 	free_ftrace_hash(hash);
129607fd5515SSteven Rostedt }
129707fd5515SSteven Rostedt 
129807fd5515SSteven Rostedt static void free_ftrace_hash_rcu(struct ftrace_hash *hash)
129907fd5515SSteven Rostedt {
130007fd5515SSteven Rostedt 	if (!hash || hash == EMPTY_HASH)
130107fd5515SSteven Rostedt 		return;
130207fd5515SSteven Rostedt 	call_rcu_sched(&hash->rcu, __free_ftrace_hash_rcu);
130307fd5515SSteven Rostedt }
130407fd5515SSteven Rostedt 
13055500fa51SJiri Olsa void ftrace_free_filter(struct ftrace_ops *ops)
13065500fa51SJiri Olsa {
1307f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
13085500fa51SJiri Olsa 	free_ftrace_hash(ops->filter_hash);
13095500fa51SJiri Olsa 	free_ftrace_hash(ops->notrace_hash);
13105500fa51SJiri Olsa }
13115500fa51SJiri Olsa 
131233dc9b12SSteven Rostedt static struct ftrace_hash *alloc_ftrace_hash(int size_bits)
131333dc9b12SSteven Rostedt {
131433dc9b12SSteven Rostedt 	struct ftrace_hash *hash;
131533dc9b12SSteven Rostedt 	int size;
131633dc9b12SSteven Rostedt 
131733dc9b12SSteven Rostedt 	hash = kzalloc(sizeof(*hash), GFP_KERNEL);
131833dc9b12SSteven Rostedt 	if (!hash)
131933dc9b12SSteven Rostedt 		return NULL;
132033dc9b12SSteven Rostedt 
132133dc9b12SSteven Rostedt 	size = 1 << size_bits;
132247b0edcbSThomas Meyer 	hash->buckets = kcalloc(size, sizeof(*hash->buckets), GFP_KERNEL);
132333dc9b12SSteven Rostedt 
132433dc9b12SSteven Rostedt 	if (!hash->buckets) {
132533dc9b12SSteven Rostedt 		kfree(hash);
132633dc9b12SSteven Rostedt 		return NULL;
132733dc9b12SSteven Rostedt 	}
132833dc9b12SSteven Rostedt 
132933dc9b12SSteven Rostedt 	hash->size_bits = size_bits;
133033dc9b12SSteven Rostedt 
133133dc9b12SSteven Rostedt 	return hash;
133233dc9b12SSteven Rostedt }
133333dc9b12SSteven Rostedt 
133433dc9b12SSteven Rostedt static struct ftrace_hash *
133533dc9b12SSteven Rostedt alloc_and_copy_ftrace_hash(int size_bits, struct ftrace_hash *hash)
133633dc9b12SSteven Rostedt {
133733dc9b12SSteven Rostedt 	struct ftrace_func_entry *entry;
133833dc9b12SSteven Rostedt 	struct ftrace_hash *new_hash;
133933dc9b12SSteven Rostedt 	int size;
134033dc9b12SSteven Rostedt 	int ret;
134133dc9b12SSteven Rostedt 	int i;
134233dc9b12SSteven Rostedt 
134333dc9b12SSteven Rostedt 	new_hash = alloc_ftrace_hash(size_bits);
134433dc9b12SSteven Rostedt 	if (!new_hash)
134533dc9b12SSteven Rostedt 		return NULL;
134633dc9b12SSteven Rostedt 
134733dc9b12SSteven Rostedt 	/* Empty hash? */
134806a51d93SSteven Rostedt 	if (ftrace_hash_empty(hash))
134933dc9b12SSteven Rostedt 		return new_hash;
135033dc9b12SSteven Rostedt 
135133dc9b12SSteven Rostedt 	size = 1 << hash->size_bits;
135233dc9b12SSteven Rostedt 	for (i = 0; i < size; i++) {
1353b67bfe0dSSasha Levin 		hlist_for_each_entry(entry, &hash->buckets[i], hlist) {
135433dc9b12SSteven Rostedt 			ret = add_hash_entry(new_hash, entry->ip);
135533dc9b12SSteven Rostedt 			if (ret < 0)
135633dc9b12SSteven Rostedt 				goto free_hash;
135733dc9b12SSteven Rostedt 		}
135833dc9b12SSteven Rostedt 	}
135933dc9b12SSteven Rostedt 
136033dc9b12SSteven Rostedt 	FTRACE_WARN_ON(new_hash->count != hash->count);
136133dc9b12SSteven Rostedt 
136233dc9b12SSteven Rostedt 	return new_hash;
136333dc9b12SSteven Rostedt 
136433dc9b12SSteven Rostedt  free_hash:
136533dc9b12SSteven Rostedt 	free_ftrace_hash(new_hash);
136633dc9b12SSteven Rostedt 	return NULL;
136733dc9b12SSteven Rostedt }
136833dc9b12SSteven Rostedt 
136941fb61c2SSteven Rostedt static void
137041fb61c2SSteven Rostedt ftrace_hash_rec_disable(struct ftrace_ops *ops, int filter_hash);
137141fb61c2SSteven Rostedt static void
137241fb61c2SSteven Rostedt ftrace_hash_rec_enable(struct ftrace_ops *ops, int filter_hash);
137341fb61c2SSteven Rostedt 
137433dc9b12SSteven Rostedt static int
137541fb61c2SSteven Rostedt ftrace_hash_move(struct ftrace_ops *ops, int enable,
137641fb61c2SSteven Rostedt 		 struct ftrace_hash **dst, struct ftrace_hash *src)
137733dc9b12SSteven Rostedt {
137833dc9b12SSteven Rostedt 	struct ftrace_func_entry *entry;
1379b67bfe0dSSasha Levin 	struct hlist_node *tn;
138033dc9b12SSteven Rostedt 	struct hlist_head *hhd;
138107fd5515SSteven Rostedt 	struct ftrace_hash *old_hash;
138207fd5515SSteven Rostedt 	struct ftrace_hash *new_hash;
138333dc9b12SSteven Rostedt 	int size = src->count;
138433dc9b12SSteven Rostedt 	int bits = 0;
138541fb61c2SSteven Rostedt 	int ret;
138633dc9b12SSteven Rostedt 	int i;
138733dc9b12SSteven Rostedt 
138833dc9b12SSteven Rostedt 	/*
138941fb61c2SSteven Rostedt 	 * Remove the current set, update the hash and add
139041fb61c2SSteven Rostedt 	 * them back.
139141fb61c2SSteven Rostedt 	 */
139241fb61c2SSteven Rostedt 	ftrace_hash_rec_disable(ops, enable);
139341fb61c2SSteven Rostedt 
139441fb61c2SSteven Rostedt 	/*
139533dc9b12SSteven Rostedt 	 * If the new source is empty, just free dst and assign it
139633dc9b12SSteven Rostedt 	 * the empty_hash.
139733dc9b12SSteven Rostedt 	 */
139833dc9b12SSteven Rostedt 	if (!src->count) {
139907fd5515SSteven Rostedt 		free_ftrace_hash_rcu(*dst);
140007fd5515SSteven Rostedt 		rcu_assign_pointer(*dst, EMPTY_HASH);
1401d4d34b98SSteven Rostedt 		/* still need to update the function records */
1402d4d34b98SSteven Rostedt 		ret = 0;
1403d4d34b98SSteven Rostedt 		goto out;
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 
141641fb61c2SSteven Rostedt 	ret = -ENOMEM;
141707fd5515SSteven Rostedt 	new_hash = alloc_ftrace_hash(bits);
141807fd5515SSteven Rostedt 	if (!new_hash)
141941fb61c2SSteven Rostedt 		goto out;
142033dc9b12SSteven Rostedt 
142133dc9b12SSteven Rostedt 	size = 1 << src->size_bits;
142233dc9b12SSteven Rostedt 	for (i = 0; i < size; i++) {
142333dc9b12SSteven Rostedt 		hhd = &src->buckets[i];
1424b67bfe0dSSasha Levin 		hlist_for_each_entry_safe(entry, tn, hhd, hlist) {
142533dc9b12SSteven Rostedt 			remove_hash_entry(src, entry);
142607fd5515SSteven Rostedt 			__add_hash_entry(new_hash, entry);
142733dc9b12SSteven Rostedt 		}
142833dc9b12SSteven Rostedt 	}
142933dc9b12SSteven Rostedt 
143007fd5515SSteven Rostedt 	old_hash = *dst;
143107fd5515SSteven Rostedt 	rcu_assign_pointer(*dst, new_hash);
143207fd5515SSteven Rostedt 	free_ftrace_hash_rcu(old_hash);
143307fd5515SSteven Rostedt 
143441fb61c2SSteven Rostedt 	ret = 0;
143541fb61c2SSteven Rostedt  out:
143641fb61c2SSteven Rostedt 	/*
143741fb61c2SSteven Rostedt 	 * Enable regardless of ret:
143841fb61c2SSteven Rostedt 	 *  On success, we enable the new hash.
143941fb61c2SSteven Rostedt 	 *  On failure, we re-enable the original hash.
144041fb61c2SSteven Rostedt 	 */
144141fb61c2SSteven Rostedt 	ftrace_hash_rec_enable(ops, enable);
144241fb61c2SSteven Rostedt 
144341fb61c2SSteven Rostedt 	return ret;
144433dc9b12SSteven Rostedt }
144533dc9b12SSteven Rostedt 
1446265c831cSSteven Rostedt /*
1447b848914cSSteven Rostedt  * Test the hashes for this ops to see if we want to call
1448b848914cSSteven Rostedt  * the ops->func or not.
1449b848914cSSteven Rostedt  *
1450b848914cSSteven Rostedt  * It's a match if the ip is in the ops->filter_hash or
1451b848914cSSteven Rostedt  * the filter_hash does not exist or is empty,
1452b848914cSSteven Rostedt  *  AND
1453b848914cSSteven Rostedt  * the ip is not in the ops->notrace_hash.
1454cdbe61bfSSteven Rostedt  *
1455cdbe61bfSSteven Rostedt  * This needs to be called with preemption disabled as
1456cdbe61bfSSteven Rostedt  * the hashes are freed with call_rcu_sched().
1457b848914cSSteven Rostedt  */
1458b848914cSSteven Rostedt static int
1459195a8afcSSteven Rostedt (Red Hat) ftrace_ops_test(struct ftrace_ops *ops, unsigned long ip, void *regs)
1460b848914cSSteven Rostedt {
1461b848914cSSteven Rostedt 	struct ftrace_hash *filter_hash;
1462b848914cSSteven Rostedt 	struct ftrace_hash *notrace_hash;
1463b848914cSSteven Rostedt 	int ret;
1464b848914cSSteven Rostedt 
1465195a8afcSSteven Rostedt (Red Hat) #ifdef CONFIG_DYNAMIC_FTRACE_WITH_REGS
1466195a8afcSSteven Rostedt (Red Hat) 	/*
1467195a8afcSSteven Rostedt (Red Hat) 	 * There's a small race when adding ops that the ftrace handler
1468195a8afcSSteven Rostedt (Red Hat) 	 * that wants regs, may be called without them. We can not
1469195a8afcSSteven Rostedt (Red Hat) 	 * allow that handler to be called if regs is NULL.
1470195a8afcSSteven Rostedt (Red Hat) 	 */
1471195a8afcSSteven Rostedt (Red Hat) 	if (regs == NULL && (ops->flags & FTRACE_OPS_FL_SAVE_REGS))
1472195a8afcSSteven Rostedt (Red Hat) 		return 0;
1473195a8afcSSteven Rostedt (Red Hat) #endif
1474195a8afcSSteven Rostedt (Red Hat) 
14751bb539caSSteven Rostedt 	filter_hash = rcu_dereference_raw_notrace(ops->filter_hash);
14761bb539caSSteven Rostedt 	notrace_hash = rcu_dereference_raw_notrace(ops->notrace_hash);
1477b848914cSSteven Rostedt 
147806a51d93SSteven Rostedt 	if ((ftrace_hash_empty(filter_hash) ||
1479b848914cSSteven Rostedt 	     ftrace_lookup_ip(filter_hash, ip)) &&
148006a51d93SSteven Rostedt 	    (ftrace_hash_empty(notrace_hash) ||
1481b848914cSSteven Rostedt 	     !ftrace_lookup_ip(notrace_hash, ip)))
1482b848914cSSteven Rostedt 		ret = 1;
1483b848914cSSteven Rostedt 	else
1484b848914cSSteven Rostedt 		ret = 0;
1485b848914cSSteven Rostedt 
1486b848914cSSteven Rostedt 	return ret;
1487b848914cSSteven Rostedt }
1488b848914cSSteven Rostedt 
1489b848914cSSteven Rostedt /*
1490265c831cSSteven Rostedt  * This is a double for. Do not use 'break' to break out of the loop,
1491265c831cSSteven Rostedt  * you must use a goto.
1492265c831cSSteven Rostedt  */
1493265c831cSSteven Rostedt #define do_for_each_ftrace_rec(pg, rec)					\
1494265c831cSSteven Rostedt 	for (pg = ftrace_pages_start; pg; pg = pg->next) {		\
1495265c831cSSteven Rostedt 		int _____i;						\
1496265c831cSSteven Rostedt 		for (_____i = 0; _____i < pg->index; _____i++) {	\
1497265c831cSSteven Rostedt 			rec = &pg->records[_____i];
1498265c831cSSteven Rostedt 
1499265c831cSSteven Rostedt #define while_for_each_ftrace_rec()		\
1500265c831cSSteven Rostedt 		}				\
1501265c831cSSteven Rostedt 	}
1502ecea656dSAbhishek Sagar 
15035855feadSSteven Rostedt 
15045855feadSSteven Rostedt static int ftrace_cmp_recs(const void *a, const void *b)
15055855feadSSteven Rostedt {
1506a650e02aSSteven Rostedt 	const struct dyn_ftrace *key = a;
1507a650e02aSSteven Rostedt 	const struct dyn_ftrace *rec = b;
15085855feadSSteven Rostedt 
1509a650e02aSSteven Rostedt 	if (key->flags < rec->ip)
15105855feadSSteven Rostedt 		return -1;
1511a650e02aSSteven Rostedt 	if (key->ip >= rec->ip + MCOUNT_INSN_SIZE)
1512a650e02aSSteven Rostedt 		return 1;
1513a650e02aSSteven Rostedt 	return 0;
1514a650e02aSSteven Rostedt }
1515a650e02aSSteven Rostedt 
1516f0cf973aSSteven Rostedt static unsigned long ftrace_location_range(unsigned long start, unsigned long end)
1517a650e02aSSteven Rostedt {
1518a650e02aSSteven Rostedt 	struct ftrace_page *pg;
1519a650e02aSSteven Rostedt 	struct dyn_ftrace *rec;
1520a650e02aSSteven Rostedt 	struct dyn_ftrace key;
1521a650e02aSSteven Rostedt 
1522a650e02aSSteven Rostedt 	key.ip = start;
1523a650e02aSSteven Rostedt 	key.flags = end;	/* overload flags, as it is unsigned long */
1524a650e02aSSteven Rostedt 
1525a650e02aSSteven Rostedt 	for (pg = ftrace_pages_start; pg; pg = pg->next) {
1526a650e02aSSteven Rostedt 		if (end < pg->records[0].ip ||
1527a650e02aSSteven Rostedt 		    start >= (pg->records[pg->index - 1].ip + MCOUNT_INSN_SIZE))
1528a650e02aSSteven Rostedt 			continue;
1529a650e02aSSteven Rostedt 		rec = bsearch(&key, pg->records, pg->index,
1530a650e02aSSteven Rostedt 			      sizeof(struct dyn_ftrace),
1531a650e02aSSteven Rostedt 			      ftrace_cmp_recs);
1532a650e02aSSteven Rostedt 		if (rec)
1533f0cf973aSSteven Rostedt 			return rec->ip;
1534a650e02aSSteven Rostedt 	}
1535a650e02aSSteven Rostedt 
15365855feadSSteven Rostedt 	return 0;
15375855feadSSteven Rostedt }
15385855feadSSteven Rostedt 
1539c88fd863SSteven Rostedt /**
1540c88fd863SSteven Rostedt  * ftrace_location - return true if the ip giving is a traced location
1541c88fd863SSteven Rostedt  * @ip: the instruction pointer to check
1542c88fd863SSteven Rostedt  *
1543f0cf973aSSteven Rostedt  * Returns rec->ip if @ip given is a pointer to a ftrace location.
1544c88fd863SSteven Rostedt  * That is, the instruction that is either a NOP or call to
1545c88fd863SSteven Rostedt  * the function tracer. It checks the ftrace internal tables to
1546c88fd863SSteven Rostedt  * determine if the address belongs or not.
1547c88fd863SSteven Rostedt  */
1548f0cf973aSSteven Rostedt unsigned long ftrace_location(unsigned long ip)
1549c88fd863SSteven Rostedt {
1550a650e02aSSteven Rostedt 	return ftrace_location_range(ip, ip);
15515855feadSSteven Rostedt }
1552c88fd863SSteven Rostedt 
1553a650e02aSSteven Rostedt /**
1554a650e02aSSteven Rostedt  * ftrace_text_reserved - return true if range contains an ftrace location
1555a650e02aSSteven Rostedt  * @start: start of range to search
1556a650e02aSSteven Rostedt  * @end: end of range to search (inclusive). @end points to the last byte to check.
1557a650e02aSSteven Rostedt  *
1558a650e02aSSteven Rostedt  * Returns 1 if @start and @end contains a ftrace location.
1559a650e02aSSteven Rostedt  * That is, the instruction that is either a NOP or call to
1560a650e02aSSteven Rostedt  * the function tracer. It checks the ftrace internal tables to
1561a650e02aSSteven Rostedt  * determine if the address belongs or not.
1562a650e02aSSteven Rostedt  */
1563a650e02aSSteven Rostedt int ftrace_text_reserved(void *start, void *end)
1564a650e02aSSteven Rostedt {
1565f0cf973aSSteven Rostedt 	unsigned long ret;
1566f0cf973aSSteven Rostedt 
1567f0cf973aSSteven Rostedt 	ret = ftrace_location_range((unsigned long)start,
1568a650e02aSSteven Rostedt 				    (unsigned long)end);
1569f0cf973aSSteven Rostedt 
1570f0cf973aSSteven Rostedt 	return (int)!!ret;
1571c88fd863SSteven Rostedt }
1572c88fd863SSteven Rostedt 
1573ed926f9bSSteven Rostedt static void __ftrace_hash_rec_update(struct ftrace_ops *ops,
1574ed926f9bSSteven Rostedt 				     int filter_hash,
1575ed926f9bSSteven Rostedt 				     bool inc)
1576ed926f9bSSteven Rostedt {
1577ed926f9bSSteven Rostedt 	struct ftrace_hash *hash;
1578ed926f9bSSteven Rostedt 	struct ftrace_hash *other_hash;
1579ed926f9bSSteven Rostedt 	struct ftrace_page *pg;
1580ed926f9bSSteven Rostedt 	struct dyn_ftrace *rec;
1581ed926f9bSSteven Rostedt 	int count = 0;
1582ed926f9bSSteven Rostedt 	int all = 0;
1583ed926f9bSSteven Rostedt 
1584ed926f9bSSteven Rostedt 	/* Only update if the ops has been registered */
1585ed926f9bSSteven Rostedt 	if (!(ops->flags & FTRACE_OPS_FL_ENABLED))
1586ed926f9bSSteven Rostedt 		return;
1587ed926f9bSSteven Rostedt 
1588ed926f9bSSteven Rostedt 	/*
1589ed926f9bSSteven Rostedt 	 * In the filter_hash case:
1590ed926f9bSSteven Rostedt 	 *   If the count is zero, we update all records.
1591ed926f9bSSteven Rostedt 	 *   Otherwise we just update the items in the hash.
1592ed926f9bSSteven Rostedt 	 *
1593ed926f9bSSteven Rostedt 	 * In the notrace_hash case:
1594ed926f9bSSteven Rostedt 	 *   We enable the update in the hash.
1595ed926f9bSSteven Rostedt 	 *   As disabling notrace means enabling the tracing,
1596ed926f9bSSteven Rostedt 	 *   and enabling notrace means disabling, the inc variable
1597ed926f9bSSteven Rostedt 	 *   gets inversed.
1598ed926f9bSSteven Rostedt 	 */
1599ed926f9bSSteven Rostedt 	if (filter_hash) {
1600ed926f9bSSteven Rostedt 		hash = ops->filter_hash;
1601ed926f9bSSteven Rostedt 		other_hash = ops->notrace_hash;
160206a51d93SSteven Rostedt 		if (ftrace_hash_empty(hash))
1603ed926f9bSSteven Rostedt 			all = 1;
1604ed926f9bSSteven Rostedt 	} else {
1605ed926f9bSSteven Rostedt 		inc = !inc;
1606ed926f9bSSteven Rostedt 		hash = ops->notrace_hash;
1607ed926f9bSSteven Rostedt 		other_hash = ops->filter_hash;
1608ed926f9bSSteven Rostedt 		/*
1609ed926f9bSSteven Rostedt 		 * If the notrace hash has no items,
1610ed926f9bSSteven Rostedt 		 * then there's nothing to do.
1611ed926f9bSSteven Rostedt 		 */
161206a51d93SSteven Rostedt 		if (ftrace_hash_empty(hash))
1613ed926f9bSSteven Rostedt 			return;
1614ed926f9bSSteven Rostedt 	}
1615ed926f9bSSteven Rostedt 
1616ed926f9bSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
1617ed926f9bSSteven Rostedt 		int in_other_hash = 0;
1618ed926f9bSSteven Rostedt 		int in_hash = 0;
1619ed926f9bSSteven Rostedt 		int match = 0;
1620ed926f9bSSteven Rostedt 
1621ed926f9bSSteven Rostedt 		if (all) {
1622ed926f9bSSteven Rostedt 			/*
1623ed926f9bSSteven Rostedt 			 * Only the filter_hash affects all records.
1624ed926f9bSSteven Rostedt 			 * Update if the record is not in the notrace hash.
1625ed926f9bSSteven Rostedt 			 */
1626b848914cSSteven Rostedt 			if (!other_hash || !ftrace_lookup_ip(other_hash, rec->ip))
1627ed926f9bSSteven Rostedt 				match = 1;
1628ed926f9bSSteven Rostedt 		} else {
162906a51d93SSteven Rostedt 			in_hash = !!ftrace_lookup_ip(hash, rec->ip);
163006a51d93SSteven Rostedt 			in_other_hash = !!ftrace_lookup_ip(other_hash, rec->ip);
1631ed926f9bSSteven Rostedt 
1632ed926f9bSSteven Rostedt 			/*
1633ed926f9bSSteven Rostedt 			 *
1634ed926f9bSSteven Rostedt 			 */
1635ed926f9bSSteven Rostedt 			if (filter_hash && in_hash && !in_other_hash)
1636ed926f9bSSteven Rostedt 				match = 1;
1637ed926f9bSSteven Rostedt 			else if (!filter_hash && in_hash &&
163806a51d93SSteven Rostedt 				 (in_other_hash || ftrace_hash_empty(other_hash)))
1639ed926f9bSSteven Rostedt 				match = 1;
1640ed926f9bSSteven Rostedt 		}
1641ed926f9bSSteven Rostedt 		if (!match)
1642ed926f9bSSteven Rostedt 			continue;
1643ed926f9bSSteven Rostedt 
1644ed926f9bSSteven Rostedt 		if (inc) {
1645ed926f9bSSteven Rostedt 			rec->flags++;
1646ed926f9bSSteven Rostedt 			if (FTRACE_WARN_ON((rec->flags & ~FTRACE_FL_MASK) == FTRACE_REF_MAX))
1647ed926f9bSSteven Rostedt 				return;
164808f6fba5SSteven Rostedt 			/*
164908f6fba5SSteven Rostedt 			 * If any ops wants regs saved for this function
165008f6fba5SSteven Rostedt 			 * then all ops will get saved regs.
165108f6fba5SSteven Rostedt 			 */
165208f6fba5SSteven Rostedt 			if (ops->flags & FTRACE_OPS_FL_SAVE_REGS)
165308f6fba5SSteven Rostedt 				rec->flags |= FTRACE_FL_REGS;
1654ed926f9bSSteven Rostedt 		} else {
1655ed926f9bSSteven Rostedt 			if (FTRACE_WARN_ON((rec->flags & ~FTRACE_FL_MASK) == 0))
1656ed926f9bSSteven Rostedt 				return;
1657ed926f9bSSteven Rostedt 			rec->flags--;
1658ed926f9bSSteven Rostedt 		}
1659ed926f9bSSteven Rostedt 		count++;
1660ed926f9bSSteven Rostedt 		/* Shortcut, if we handled all records, we are done. */
1661ed926f9bSSteven Rostedt 		if (!all && count == hash->count)
1662ed926f9bSSteven Rostedt 			return;
1663ed926f9bSSteven Rostedt 	} while_for_each_ftrace_rec();
1664ed926f9bSSteven Rostedt }
1665ed926f9bSSteven Rostedt 
1666ed926f9bSSteven Rostedt static void ftrace_hash_rec_disable(struct ftrace_ops *ops,
1667ed926f9bSSteven Rostedt 				    int filter_hash)
1668ed926f9bSSteven Rostedt {
1669ed926f9bSSteven Rostedt 	__ftrace_hash_rec_update(ops, filter_hash, 0);
1670ed926f9bSSteven Rostedt }
1671ed926f9bSSteven Rostedt 
1672ed926f9bSSteven Rostedt static void ftrace_hash_rec_enable(struct ftrace_ops *ops,
1673ed926f9bSSteven Rostedt 				   int filter_hash)
1674ed926f9bSSteven Rostedt {
1675ed926f9bSSteven Rostedt 	__ftrace_hash_rec_update(ops, filter_hash, 1);
1676ed926f9bSSteven Rostedt }
1677ed926f9bSSteven Rostedt 
167805736a42SSteven Rostedt static void print_ip_ins(const char *fmt, unsigned char *p)
167905736a42SSteven Rostedt {
168005736a42SSteven Rostedt 	int i;
168105736a42SSteven Rostedt 
168205736a42SSteven Rostedt 	printk(KERN_CONT "%s", fmt);
168305736a42SSteven Rostedt 
168405736a42SSteven Rostedt 	for (i = 0; i < MCOUNT_INSN_SIZE; i++)
168505736a42SSteven Rostedt 		printk(KERN_CONT "%s%02x", i ? ":" : "", p[i]);
168605736a42SSteven Rostedt }
168705736a42SSteven Rostedt 
1688c88fd863SSteven Rostedt /**
1689c88fd863SSteven Rostedt  * ftrace_bug - report and shutdown function tracer
1690c88fd863SSteven Rostedt  * @failed: The failed type (EFAULT, EINVAL, EPERM)
1691c88fd863SSteven Rostedt  * @ip: The address that failed
1692c88fd863SSteven Rostedt  *
1693c88fd863SSteven Rostedt  * The arch code that enables or disables the function tracing
1694c88fd863SSteven Rostedt  * can call ftrace_bug() when it has detected a problem in
1695c88fd863SSteven Rostedt  * modifying the code. @failed should be one of either:
1696c88fd863SSteven Rostedt  * EFAULT - if the problem happens on reading the @ip address
1697c88fd863SSteven Rostedt  * EINVAL - if what is read at @ip is not what was expected
1698c88fd863SSteven Rostedt  * EPERM - if the problem happens on writting to the @ip address
1699c88fd863SSteven Rostedt  */
1700c88fd863SSteven Rostedt void ftrace_bug(int failed, unsigned long ip)
17013c1720f0SSteven Rostedt {
1702b17e8a37SSteven Rostedt 	switch (failed) {
1703593eb8a2SSteven Rostedt 	case -EFAULT:
17046912896eSSteven Rostedt 		FTRACE_WARN_ON_ONCE(1);
170505736a42SSteven Rostedt 		pr_info("ftrace faulted on modifying ");
170605736a42SSteven Rostedt 		print_ip_sym(ip);
170705736a42SSteven Rostedt 		break;
1708593eb8a2SSteven Rostedt 	case -EINVAL:
17096912896eSSteven Rostedt 		FTRACE_WARN_ON_ONCE(1);
171005736a42SSteven Rostedt 		pr_info("ftrace failed to modify ");
171105736a42SSteven Rostedt 		print_ip_sym(ip);
171205736a42SSteven Rostedt 		print_ip_ins(" actual: ", (unsigned char *)ip);
171305736a42SSteven Rostedt 		printk(KERN_CONT "\n");
171405736a42SSteven Rostedt 		break;
1715593eb8a2SSteven Rostedt 	case -EPERM:
17166912896eSSteven Rostedt 		FTRACE_WARN_ON_ONCE(1);
1717593eb8a2SSteven Rostedt 		pr_info("ftrace faulted on writing ");
1718593eb8a2SSteven Rostedt 		print_ip_sym(ip);
1719593eb8a2SSteven Rostedt 		break;
1720593eb8a2SSteven Rostedt 	default:
17216912896eSSteven Rostedt 		FTRACE_WARN_ON_ONCE(1);
1722593eb8a2SSteven Rostedt 		pr_info("ftrace faulted on unknown error ");
1723593eb8a2SSteven Rostedt 		print_ip_sym(ip);
172405736a42SSteven Rostedt 	}
1725b17e8a37SSteven Rostedt }
172605736a42SSteven Rostedt 
1727c88fd863SSteven Rostedt static int ftrace_check_record(struct dyn_ftrace *rec, int enable, int update)
17283d083395SSteven Rostedt {
172964fbcd16SXiao Guangrong 	unsigned long flag = 0UL;
1730e7d3737eSFrederic Weisbecker 
17313d083395SSteven Rostedt 	/*
173230fb6aa7SJiri Olsa 	 * If we are updating calls:
17333d083395SSteven Rostedt 	 *
1734ed926f9bSSteven Rostedt 	 *   If the record has a ref count, then we need to enable it
1735ed926f9bSSteven Rostedt 	 *   because someone is using it.
17363d083395SSteven Rostedt 	 *
1737ed926f9bSSteven Rostedt 	 *   Otherwise we make sure its disabled.
1738ed926f9bSSteven Rostedt 	 *
173930fb6aa7SJiri Olsa 	 * If we are disabling calls, then disable all records that
1740ed926f9bSSteven Rostedt 	 * are enabled.
17413d083395SSteven Rostedt 	 */
1742c88fd863SSteven Rostedt 	if (enable && (rec->flags & ~FTRACE_FL_MASK))
174364fbcd16SXiao Guangrong 		flag = FTRACE_FL_ENABLED;
17443d083395SSteven Rostedt 
174508f6fba5SSteven Rostedt 	/*
174608f6fba5SSteven Rostedt 	 * If enabling and the REGS flag does not match the REGS_EN, then
174708f6fba5SSteven Rostedt 	 * do not ignore this record. Set flags to fail the compare against
174808f6fba5SSteven Rostedt 	 * ENABLED.
174908f6fba5SSteven Rostedt 	 */
175008f6fba5SSteven Rostedt 	if (flag &&
175108f6fba5SSteven Rostedt 	    (!(rec->flags & FTRACE_FL_REGS) != !(rec->flags & FTRACE_FL_REGS_EN)))
175208f6fba5SSteven Rostedt 		flag |= FTRACE_FL_REGS;
175308f6fba5SSteven Rostedt 
175464fbcd16SXiao Guangrong 	/* If the state of this record hasn't changed, then do nothing */
175564fbcd16SXiao Guangrong 	if ((rec->flags & FTRACE_FL_ENABLED) == flag)
1756c88fd863SSteven Rostedt 		return FTRACE_UPDATE_IGNORE;
175764fbcd16SXiao Guangrong 
175864fbcd16SXiao Guangrong 	if (flag) {
175908f6fba5SSteven Rostedt 		/* Save off if rec is being enabled (for return value) */
176008f6fba5SSteven Rostedt 		flag ^= rec->flags & FTRACE_FL_ENABLED;
176108f6fba5SSteven Rostedt 
176208f6fba5SSteven Rostedt 		if (update) {
176364fbcd16SXiao Guangrong 			rec->flags |= FTRACE_FL_ENABLED;
176408f6fba5SSteven Rostedt 			if (flag & FTRACE_FL_REGS) {
176508f6fba5SSteven Rostedt 				if (rec->flags & FTRACE_FL_REGS)
176608f6fba5SSteven Rostedt 					rec->flags |= FTRACE_FL_REGS_EN;
176708f6fba5SSteven Rostedt 				else
176808f6fba5SSteven Rostedt 					rec->flags &= ~FTRACE_FL_REGS_EN;
176908f6fba5SSteven Rostedt 			}
177064fbcd16SXiao Guangrong 		}
177164fbcd16SXiao Guangrong 
177208f6fba5SSteven Rostedt 		/*
177308f6fba5SSteven Rostedt 		 * If this record is being updated from a nop, then
177408f6fba5SSteven Rostedt 		 *   return UPDATE_MAKE_CALL.
177508f6fba5SSteven Rostedt 		 * Otherwise, if the EN flag is set, then return
177608f6fba5SSteven Rostedt 		 *   UPDATE_MODIFY_CALL_REGS to tell the caller to convert
177708f6fba5SSteven Rostedt 		 *   from the non-save regs, to a save regs function.
177808f6fba5SSteven Rostedt 		 * Otherwise,
177908f6fba5SSteven Rostedt 		 *   return UPDATE_MODIFY_CALL to tell the caller to convert
178008f6fba5SSteven Rostedt 		 *   from the save regs, to a non-save regs function.
178108f6fba5SSteven Rostedt 		 */
178208f6fba5SSteven Rostedt 		if (flag & FTRACE_FL_ENABLED)
178308f6fba5SSteven Rostedt 			return FTRACE_UPDATE_MAKE_CALL;
178408f6fba5SSteven Rostedt 		else if (rec->flags & FTRACE_FL_REGS_EN)
178508f6fba5SSteven Rostedt 			return FTRACE_UPDATE_MODIFY_CALL_REGS;
178608f6fba5SSteven Rostedt 		else
178708f6fba5SSteven Rostedt 			return FTRACE_UPDATE_MODIFY_CALL;
178808f6fba5SSteven Rostedt 	}
178908f6fba5SSteven Rostedt 
179008f6fba5SSteven Rostedt 	if (update) {
179108f6fba5SSteven Rostedt 		/* If there's no more users, clear all flags */
179208f6fba5SSteven Rostedt 		if (!(rec->flags & ~FTRACE_FL_MASK))
179308f6fba5SSteven Rostedt 			rec->flags = 0;
179408f6fba5SSteven Rostedt 		else
179508f6fba5SSteven Rostedt 			/* Just disable the record (keep REGS state) */
179664fbcd16SXiao Guangrong 			rec->flags &= ~FTRACE_FL_ENABLED;
179708f6fba5SSteven Rostedt 	}
1798c88fd863SSteven Rostedt 
1799c88fd863SSteven Rostedt 	return FTRACE_UPDATE_MAKE_NOP;
1800c88fd863SSteven Rostedt }
1801c88fd863SSteven Rostedt 
1802c88fd863SSteven Rostedt /**
1803c88fd863SSteven Rostedt  * ftrace_update_record, set a record that now is tracing or not
1804c88fd863SSteven Rostedt  * @rec: the record to update
1805c88fd863SSteven Rostedt  * @enable: set to 1 if the record is tracing, zero to force disable
1806c88fd863SSteven Rostedt  *
1807c88fd863SSteven Rostedt  * The records that represent all functions that can be traced need
1808c88fd863SSteven Rostedt  * to be updated when tracing has been enabled.
1809c88fd863SSteven Rostedt  */
1810c88fd863SSteven Rostedt int ftrace_update_record(struct dyn_ftrace *rec, int enable)
1811c88fd863SSteven Rostedt {
1812c88fd863SSteven Rostedt 	return ftrace_check_record(rec, enable, 1);
1813c88fd863SSteven Rostedt }
1814c88fd863SSteven Rostedt 
1815c88fd863SSteven Rostedt /**
1816c88fd863SSteven Rostedt  * ftrace_test_record, check if the record has been enabled or not
1817c88fd863SSteven Rostedt  * @rec: the record to test
1818c88fd863SSteven Rostedt  * @enable: set to 1 to check if enabled, 0 if it is disabled
1819c88fd863SSteven Rostedt  *
1820c88fd863SSteven Rostedt  * The arch code may need to test if a record is already set to
1821c88fd863SSteven Rostedt  * tracing to determine how to modify the function code that it
1822c88fd863SSteven Rostedt  * represents.
1823c88fd863SSteven Rostedt  */
1824c88fd863SSteven Rostedt int ftrace_test_record(struct dyn_ftrace *rec, int enable)
1825c88fd863SSteven Rostedt {
1826c88fd863SSteven Rostedt 	return ftrace_check_record(rec, enable, 0);
1827c88fd863SSteven Rostedt }
1828c88fd863SSteven Rostedt 
1829c88fd863SSteven Rostedt static int
1830c88fd863SSteven Rostedt __ftrace_replace_code(struct dyn_ftrace *rec, int enable)
1831c88fd863SSteven Rostedt {
183208f6fba5SSteven Rostedt 	unsigned long ftrace_old_addr;
1833c88fd863SSteven Rostedt 	unsigned long ftrace_addr;
1834c88fd863SSteven Rostedt 	int ret;
1835c88fd863SSteven Rostedt 
1836c88fd863SSteven Rostedt 	ret = ftrace_update_record(rec, enable);
1837c88fd863SSteven Rostedt 
183808f6fba5SSteven Rostedt 	if (rec->flags & FTRACE_FL_REGS)
183908f6fba5SSteven Rostedt 		ftrace_addr = (unsigned long)FTRACE_REGS_ADDR;
184008f6fba5SSteven Rostedt 	else
184108f6fba5SSteven Rostedt 		ftrace_addr = (unsigned long)FTRACE_ADDR;
184208f6fba5SSteven Rostedt 
1843c88fd863SSteven Rostedt 	switch (ret) {
1844c88fd863SSteven Rostedt 	case FTRACE_UPDATE_IGNORE:
1845c88fd863SSteven Rostedt 		return 0;
1846c88fd863SSteven Rostedt 
1847c88fd863SSteven Rostedt 	case FTRACE_UPDATE_MAKE_CALL:
1848c88fd863SSteven Rostedt 		return ftrace_make_call(rec, ftrace_addr);
1849c88fd863SSteven Rostedt 
1850c88fd863SSteven Rostedt 	case FTRACE_UPDATE_MAKE_NOP:
1851e7d3737eSFrederic Weisbecker 		return ftrace_make_nop(NULL, rec, ftrace_addr);
185208f6fba5SSteven Rostedt 
185308f6fba5SSteven Rostedt 	case FTRACE_UPDATE_MODIFY_CALL_REGS:
185408f6fba5SSteven Rostedt 	case FTRACE_UPDATE_MODIFY_CALL:
185508f6fba5SSteven Rostedt 		if (rec->flags & FTRACE_FL_REGS)
185608f6fba5SSteven Rostedt 			ftrace_old_addr = (unsigned long)FTRACE_ADDR;
185708f6fba5SSteven Rostedt 		else
185808f6fba5SSteven Rostedt 			ftrace_old_addr = (unsigned long)FTRACE_REGS_ADDR;
185908f6fba5SSteven Rostedt 
186008f6fba5SSteven Rostedt 		return ftrace_modify_call(rec, ftrace_old_addr, ftrace_addr);
18613d083395SSteven Rostedt 	}
18623d083395SSteven Rostedt 
1863c88fd863SSteven Rostedt 	return -1; /* unknow ftrace bug */
1864c88fd863SSteven Rostedt }
1865c88fd863SSteven Rostedt 
1866e4f5d544SSteven Rostedt void __weak ftrace_replace_code(int enable)
18673d083395SSteven Rostedt {
18683d083395SSteven Rostedt 	struct dyn_ftrace *rec;
18693d083395SSteven Rostedt 	struct ftrace_page *pg;
18706a24a244SSteven Rostedt 	int failed;
18713d083395SSteven Rostedt 
187245a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
187345a4a237SSteven Rostedt 		return;
187445a4a237SSteven Rostedt 
1875265c831cSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
1876e4f5d544SSteven Rostedt 		failed = __ftrace_replace_code(rec, enable);
1877fa9d13cfSZhaolei 		if (failed) {
187831e88909SSteven Rostedt 			ftrace_bug(failed, rec->ip);
18794377245aSSteven Rostedt 			/* Stop processing */
18804377245aSSteven Rostedt 			return;
18814377245aSSteven Rostedt 		}
1882265c831cSSteven Rostedt 	} while_for_each_ftrace_rec();
18833d083395SSteven Rostedt }
18843c1720f0SSteven Rostedt 
1885c88fd863SSteven Rostedt struct ftrace_rec_iter {
1886c88fd863SSteven Rostedt 	struct ftrace_page	*pg;
1887c88fd863SSteven Rostedt 	int			index;
1888c88fd863SSteven Rostedt };
1889c88fd863SSteven Rostedt 
1890c88fd863SSteven Rostedt /**
1891c88fd863SSteven Rostedt  * ftrace_rec_iter_start, start up iterating over traced functions
1892c88fd863SSteven Rostedt  *
1893c88fd863SSteven Rostedt  * Returns an iterator handle that is used to iterate over all
1894c88fd863SSteven Rostedt  * the records that represent address locations where functions
1895c88fd863SSteven Rostedt  * are traced.
1896c88fd863SSteven Rostedt  *
1897c88fd863SSteven Rostedt  * May return NULL if no records are available.
1898c88fd863SSteven Rostedt  */
1899c88fd863SSteven Rostedt struct ftrace_rec_iter *ftrace_rec_iter_start(void)
1900c88fd863SSteven Rostedt {
1901c88fd863SSteven Rostedt 	/*
1902c88fd863SSteven Rostedt 	 * We only use a single iterator.
1903c88fd863SSteven Rostedt 	 * Protected by the ftrace_lock mutex.
1904c88fd863SSteven Rostedt 	 */
1905c88fd863SSteven Rostedt 	static struct ftrace_rec_iter ftrace_rec_iter;
1906c88fd863SSteven Rostedt 	struct ftrace_rec_iter *iter = &ftrace_rec_iter;
1907c88fd863SSteven Rostedt 
1908c88fd863SSteven Rostedt 	iter->pg = ftrace_pages_start;
1909c88fd863SSteven Rostedt 	iter->index = 0;
1910c88fd863SSteven Rostedt 
1911c88fd863SSteven Rostedt 	/* Could have empty pages */
1912c88fd863SSteven Rostedt 	while (iter->pg && !iter->pg->index)
1913c88fd863SSteven Rostedt 		iter->pg = iter->pg->next;
1914c88fd863SSteven Rostedt 
1915c88fd863SSteven Rostedt 	if (!iter->pg)
1916c88fd863SSteven Rostedt 		return NULL;
1917c88fd863SSteven Rostedt 
1918c88fd863SSteven Rostedt 	return iter;
1919c88fd863SSteven Rostedt }
1920c88fd863SSteven Rostedt 
1921c88fd863SSteven Rostedt /**
1922c88fd863SSteven Rostedt  * ftrace_rec_iter_next, get the next record to process.
1923c88fd863SSteven Rostedt  * @iter: The handle to the iterator.
1924c88fd863SSteven Rostedt  *
1925c88fd863SSteven Rostedt  * Returns the next iterator after the given iterator @iter.
1926c88fd863SSteven Rostedt  */
1927c88fd863SSteven Rostedt struct ftrace_rec_iter *ftrace_rec_iter_next(struct ftrace_rec_iter *iter)
1928c88fd863SSteven Rostedt {
1929c88fd863SSteven Rostedt 	iter->index++;
1930c88fd863SSteven Rostedt 
1931c88fd863SSteven Rostedt 	if (iter->index >= iter->pg->index) {
1932c88fd863SSteven Rostedt 		iter->pg = iter->pg->next;
1933c88fd863SSteven Rostedt 		iter->index = 0;
1934c88fd863SSteven Rostedt 
1935c88fd863SSteven Rostedt 		/* Could have empty pages */
1936c88fd863SSteven Rostedt 		while (iter->pg && !iter->pg->index)
1937c88fd863SSteven Rostedt 			iter->pg = iter->pg->next;
1938c88fd863SSteven Rostedt 	}
1939c88fd863SSteven Rostedt 
1940c88fd863SSteven Rostedt 	if (!iter->pg)
1941c88fd863SSteven Rostedt 		return NULL;
1942c88fd863SSteven Rostedt 
1943c88fd863SSteven Rostedt 	return iter;
1944c88fd863SSteven Rostedt }
1945c88fd863SSteven Rostedt 
1946c88fd863SSteven Rostedt /**
1947c88fd863SSteven Rostedt  * ftrace_rec_iter_record, get the record at the iterator location
1948c88fd863SSteven Rostedt  * @iter: The current iterator location
1949c88fd863SSteven Rostedt  *
1950c88fd863SSteven Rostedt  * Returns the record that the current @iter is at.
1951c88fd863SSteven Rostedt  */
1952c88fd863SSteven Rostedt struct dyn_ftrace *ftrace_rec_iter_record(struct ftrace_rec_iter *iter)
1953c88fd863SSteven Rostedt {
1954c88fd863SSteven Rostedt 	return &iter->pg->records[iter->index];
1955c88fd863SSteven Rostedt }
1956c88fd863SSteven Rostedt 
19573b47bfc1SSteven Rostedt static int
195831e88909SSteven Rostedt ftrace_code_disable(struct module *mod, struct dyn_ftrace *rec)
19593c1720f0SSteven Rostedt {
196037ad5084SSteven Rostedt 	unsigned long ip;
1961492a7ea5SAbhishek Sagar 	int ret;
196237ad5084SSteven Rostedt 
1963492a7ea5SAbhishek Sagar 	ip = rec->ip;
19643c1720f0SSteven Rostedt 
196545a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
196645a4a237SSteven Rostedt 		return 0;
196745a4a237SSteven Rostedt 
196825aac9dcSShaohua Li 	ret = ftrace_make_nop(mod, rec, MCOUNT_ADDR);
19693d083395SSteven Rostedt 	if (ret) {
197031e88909SSteven Rostedt 		ftrace_bug(ret, ip);
19713d083395SSteven Rostedt 		return 0;
1972d61f82d0SSteven Rostedt 	}
1973d61f82d0SSteven Rostedt 	return 1;
19743d083395SSteven Rostedt }
19753d083395SSteven Rostedt 
1976000ab691SSteven Rostedt /*
1977000ab691SSteven Rostedt  * archs can override this function if they must do something
1978000ab691SSteven Rostedt  * before the modifying code is performed.
1979000ab691SSteven Rostedt  */
1980000ab691SSteven Rostedt int __weak ftrace_arch_code_modify_prepare(void)
1981000ab691SSteven Rostedt {
1982000ab691SSteven Rostedt 	return 0;
1983000ab691SSteven Rostedt }
1984000ab691SSteven Rostedt 
1985000ab691SSteven Rostedt /*
1986000ab691SSteven Rostedt  * archs can override this function if they must do something
1987000ab691SSteven Rostedt  * after the modifying code is performed.
1988000ab691SSteven Rostedt  */
1989000ab691SSteven Rostedt int __weak ftrace_arch_code_modify_post_process(void)
1990000ab691SSteven Rostedt {
1991000ab691SSteven Rostedt 	return 0;
1992000ab691SSteven Rostedt }
1993000ab691SSteven Rostedt 
19948ed3e2cfSSteven Rostedt void ftrace_modify_all_code(int command)
19958ed3e2cfSSteven Rostedt {
199659338f75SSteven Rostedt (Red Hat) 	int update = command & FTRACE_UPDATE_TRACE_FUNC;
199759338f75SSteven Rostedt (Red Hat) 
199859338f75SSteven Rostedt (Red Hat) 	/*
199959338f75SSteven Rostedt (Red Hat) 	 * If the ftrace_caller calls a ftrace_ops func directly,
200059338f75SSteven Rostedt (Red Hat) 	 * we need to make sure that it only traces functions it
200159338f75SSteven Rostedt (Red Hat) 	 * expects to trace. When doing the switch of functions,
200259338f75SSteven Rostedt (Red Hat) 	 * we need to update to the ftrace_ops_list_func first
200359338f75SSteven Rostedt (Red Hat) 	 * before the transition between old and new calls are set,
200459338f75SSteven Rostedt (Red Hat) 	 * as the ftrace_ops_list_func will check the ops hashes
200559338f75SSteven Rostedt (Red Hat) 	 * to make sure the ops are having the right functions
200659338f75SSteven Rostedt (Red Hat) 	 * traced.
200759338f75SSteven Rostedt (Red Hat) 	 */
200859338f75SSteven Rostedt (Red Hat) 	if (update)
200959338f75SSteven Rostedt (Red Hat) 		ftrace_update_ftrace_func(ftrace_ops_list_func);
201059338f75SSteven Rostedt (Red Hat) 
20118ed3e2cfSSteven Rostedt 	if (command & FTRACE_UPDATE_CALLS)
20128ed3e2cfSSteven Rostedt 		ftrace_replace_code(1);
20138ed3e2cfSSteven Rostedt 	else if (command & FTRACE_DISABLE_CALLS)
20148ed3e2cfSSteven Rostedt 		ftrace_replace_code(0);
20158ed3e2cfSSteven Rostedt 
2016405e1d83SSteven Rostedt (Red Hat) 	if (update && ftrace_trace_function != ftrace_ops_list_func) {
2017405e1d83SSteven Rostedt (Red Hat) 		function_trace_op = set_function_trace_op;
2018405e1d83SSteven Rostedt (Red Hat) 		smp_wmb();
2019405e1d83SSteven Rostedt (Red Hat) 		/* If irqs are disabled, we are in stop machine */
2020405e1d83SSteven Rostedt (Red Hat) 		if (!irqs_disabled())
2021405e1d83SSteven Rostedt (Red Hat) 			smp_call_function(ftrace_sync_ipi, NULL, 1);
20228ed3e2cfSSteven Rostedt 		ftrace_update_ftrace_func(ftrace_trace_function);
2023405e1d83SSteven Rostedt (Red Hat) 	}
20248ed3e2cfSSteven Rostedt 
20258ed3e2cfSSteven Rostedt 	if (command & FTRACE_START_FUNC_RET)
20268ed3e2cfSSteven Rostedt 		ftrace_enable_ftrace_graph_caller();
20278ed3e2cfSSteven Rostedt 	else if (command & FTRACE_STOP_FUNC_RET)
20288ed3e2cfSSteven Rostedt 		ftrace_disable_ftrace_graph_caller();
20298ed3e2cfSSteven Rostedt }
20308ed3e2cfSSteven Rostedt 
20313d083395SSteven Rostedt static int __ftrace_modify_code(void *data)
20323d083395SSteven Rostedt {
20333d083395SSteven Rostedt 	int *command = data;
20343d083395SSteven Rostedt 
20358ed3e2cfSSteven Rostedt 	ftrace_modify_all_code(*command);
20365a45cfe1SSteven Rostedt 
2037c88fd863SSteven Rostedt 	return 0;
2038c88fd863SSteven Rostedt }
2039c88fd863SSteven Rostedt 
2040c88fd863SSteven Rostedt /**
2041c88fd863SSteven Rostedt  * ftrace_run_stop_machine, go back to the stop machine method
2042c88fd863SSteven Rostedt  * @command: The command to tell ftrace what to do
2043c88fd863SSteven Rostedt  *
2044c88fd863SSteven Rostedt  * If an arch needs to fall back to the stop machine method, the
2045c88fd863SSteven Rostedt  * it can call this function.
2046c88fd863SSteven Rostedt  */
2047c88fd863SSteven Rostedt void ftrace_run_stop_machine(int command)
2048c88fd863SSteven Rostedt {
2049c88fd863SSteven Rostedt 	stop_machine(__ftrace_modify_code, &command, NULL);
2050c88fd863SSteven Rostedt }
2051c88fd863SSteven Rostedt 
2052c88fd863SSteven Rostedt /**
2053c88fd863SSteven Rostedt  * arch_ftrace_update_code, modify the code to trace or not trace
2054c88fd863SSteven Rostedt  * @command: The command that needs to be done
2055c88fd863SSteven Rostedt  *
2056c88fd863SSteven Rostedt  * Archs can override this function if it does not need to
2057c88fd863SSteven Rostedt  * run stop_machine() to modify code.
2058c88fd863SSteven Rostedt  */
2059c88fd863SSteven Rostedt void __weak arch_ftrace_update_code(int command)
2060c88fd863SSteven Rostedt {
2061c88fd863SSteven Rostedt 	ftrace_run_stop_machine(command);
2062c88fd863SSteven Rostedt }
2063c88fd863SSteven Rostedt 
2064c88fd863SSteven Rostedt static void ftrace_run_update_code(int command)
2065c88fd863SSteven Rostedt {
2066c88fd863SSteven Rostedt 	int ret;
2067c88fd863SSteven Rostedt 
2068c88fd863SSteven Rostedt 	ret = ftrace_arch_code_modify_prepare();
2069c88fd863SSteven Rostedt 	FTRACE_WARN_ON(ret);
2070c88fd863SSteven Rostedt 	if (ret)
2071c88fd863SSteven Rostedt 		return;
2072c88fd863SSteven Rostedt 	/*
2073c88fd863SSteven Rostedt 	 * Do not call function tracer while we update the code.
2074c88fd863SSteven Rostedt 	 * We are in stop machine.
2075c88fd863SSteven Rostedt 	 */
2076c88fd863SSteven Rostedt 	function_trace_stop++;
2077c88fd863SSteven Rostedt 
2078c88fd863SSteven Rostedt 	/*
2079c88fd863SSteven Rostedt 	 * By default we use stop_machine() to modify the code.
2080c88fd863SSteven Rostedt 	 * But archs can do what ever they want as long as it
2081c88fd863SSteven Rostedt 	 * is safe. The stop_machine() is the safest, but also
2082c88fd863SSteven Rostedt 	 * produces the most overhead.
2083c88fd863SSteven Rostedt 	 */
2084c88fd863SSteven Rostedt 	arch_ftrace_update_code(command);
2085c88fd863SSteven Rostedt 
20866331c28cSSteven Rostedt 	function_trace_stop--;
20876331c28cSSteven Rostedt 
2088000ab691SSteven Rostedt 	ret = ftrace_arch_code_modify_post_process();
2089000ab691SSteven Rostedt 	FTRACE_WARN_ON(ret);
20903d083395SSteven Rostedt }
20913d083395SSteven Rostedt 
20923d083395SSteven Rostedt static ftrace_func_t saved_ftrace_func;
209360a7ecf4SSteven Rostedt static int ftrace_start_up;
2094b848914cSSteven Rostedt static int global_start_up;
2095df4fc315SSteven Rostedt 
2096df4fc315SSteven Rostedt static void ftrace_startup_enable(int command)
2097df4fc315SSteven Rostedt {
2098df4fc315SSteven Rostedt 	if (saved_ftrace_func != ftrace_trace_function) {
2099df4fc315SSteven Rostedt 		saved_ftrace_func = ftrace_trace_function;
2100df4fc315SSteven Rostedt 		command |= FTRACE_UPDATE_TRACE_FUNC;
2101df4fc315SSteven Rostedt 	}
2102df4fc315SSteven Rostedt 
2103df4fc315SSteven Rostedt 	if (!command || !ftrace_enabled)
2104df4fc315SSteven Rostedt 		return;
2105df4fc315SSteven Rostedt 
2106df4fc315SSteven Rostedt 	ftrace_run_update_code(command);
2107df4fc315SSteven Rostedt }
21083d083395SSteven Rostedt 
2109a1cd6173SSteven Rostedt static int ftrace_startup(struct ftrace_ops *ops, int command)
21103d083395SSteven Rostedt {
2111b848914cSSteven Rostedt 	bool hash_enable = true;
21128a56d776SSteven Rostedt (Red Hat) 	int ret;
2113b848914cSSteven Rostedt 
21144eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
2115a1cd6173SSteven Rostedt 		return -ENODEV;
21164eebcc81SSteven Rostedt 
21178a56d776SSteven Rostedt (Red Hat) 	ret = __register_ftrace_function(ops);
21188a56d776SSteven Rostedt (Red Hat) 	if (ret)
21198a56d776SSteven Rostedt (Red Hat) 		return ret;
21208a56d776SSteven Rostedt (Red Hat) 
212160a7ecf4SSteven Rostedt 	ftrace_start_up++;
212230fb6aa7SJiri Olsa 	command |= FTRACE_UPDATE_CALLS;
21233d083395SSteven Rostedt 
2124b848914cSSteven Rostedt 	/* ops marked global share the filter hashes */
2125b848914cSSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_GLOBAL) {
2126b848914cSSteven Rostedt 		ops = &global_ops;
2127b848914cSSteven Rostedt 		/* Don't update hash if global is already set */
2128b848914cSSteven Rostedt 		if (global_start_up)
2129b848914cSSteven Rostedt 			hash_enable = false;
2130b848914cSSteven Rostedt 		global_start_up++;
2131b848914cSSteven Rostedt 	}
2132b848914cSSteven Rostedt 
2133ed926f9bSSteven Rostedt 	ops->flags |= FTRACE_OPS_FL_ENABLED;
2134b848914cSSteven Rostedt 	if (hash_enable)
2135ed926f9bSSteven Rostedt 		ftrace_hash_rec_enable(ops, 1);
2136ed926f9bSSteven Rostedt 
2137df4fc315SSteven Rostedt 	ftrace_startup_enable(command);
2138a1cd6173SSteven Rostedt 
2139a1cd6173SSteven Rostedt 	return 0;
21403d083395SSteven Rostedt }
21413d083395SSteven Rostedt 
21428a56d776SSteven Rostedt (Red Hat) static int ftrace_shutdown(struct ftrace_ops *ops, int command)
21433d083395SSteven Rostedt {
2144b848914cSSteven Rostedt 	bool hash_disable = true;
21458a56d776SSteven Rostedt (Red Hat) 	int ret;
2146b848914cSSteven Rostedt 
21474eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
21488a56d776SSteven Rostedt (Red Hat) 		return -ENODEV;
21498a56d776SSteven Rostedt (Red Hat) 
21508a56d776SSteven Rostedt (Red Hat) 	ret = __unregister_ftrace_function(ops);
21518a56d776SSteven Rostedt (Red Hat) 	if (ret)
21528a56d776SSteven Rostedt (Red Hat) 		return ret;
21534eebcc81SSteven Rostedt 
215460a7ecf4SSteven Rostedt 	ftrace_start_up--;
21559ea1a153SFrederic Weisbecker 	/*
21569ea1a153SFrederic Weisbecker 	 * Just warn in case of unbalance, no need to kill ftrace, it's not
21579ea1a153SFrederic Weisbecker 	 * critical but the ftrace_call callers may be never nopped again after
21589ea1a153SFrederic Weisbecker 	 * further ftrace uses.
21599ea1a153SFrederic Weisbecker 	 */
21609ea1a153SFrederic Weisbecker 	WARN_ON_ONCE(ftrace_start_up < 0);
21619ea1a153SFrederic Weisbecker 
2162b848914cSSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_GLOBAL) {
2163b848914cSSteven Rostedt 		ops = &global_ops;
2164b848914cSSteven Rostedt 		global_start_up--;
2165b848914cSSteven Rostedt 		WARN_ON_ONCE(global_start_up < 0);
2166b848914cSSteven Rostedt 		/* Don't update hash if global still has users */
2167b848914cSSteven Rostedt 		if (global_start_up) {
2168b848914cSSteven Rostedt 			WARN_ON_ONCE(!ftrace_start_up);
2169b848914cSSteven Rostedt 			hash_disable = false;
2170b848914cSSteven Rostedt 		}
2171b848914cSSteven Rostedt 	}
2172b848914cSSteven Rostedt 
2173b848914cSSteven Rostedt 	if (hash_disable)
2174ed926f9bSSteven Rostedt 		ftrace_hash_rec_disable(ops, 1);
2175ed926f9bSSteven Rostedt 
2176b848914cSSteven Rostedt 	if (ops != &global_ops || !global_start_up)
2177ed926f9bSSteven Rostedt 		ops->flags &= ~FTRACE_OPS_FL_ENABLED;
2178b848914cSSteven Rostedt 
217930fb6aa7SJiri Olsa 	command |= FTRACE_UPDATE_CALLS;
2180d61f82d0SSteven Rostedt 
2181d61f82d0SSteven Rostedt 	if (saved_ftrace_func != ftrace_trace_function) {
2182d61f82d0SSteven Rostedt 		saved_ftrace_func = ftrace_trace_function;
2183d61f82d0SSteven Rostedt 		command |= FTRACE_UPDATE_TRACE_FUNC;
2184d61f82d0SSteven Rostedt 	}
2185d61f82d0SSteven Rostedt 
2186a4c35ed2SSteven Rostedt (Red Hat) 	if (!command || !ftrace_enabled) {
2187a4c35ed2SSteven Rostedt (Red Hat) 		/*
2188a4c35ed2SSteven Rostedt (Red Hat) 		 * If these are control ops, they still need their
2189a4c35ed2SSteven Rostedt (Red Hat) 		 * per_cpu field freed. Since, function tracing is
2190a4c35ed2SSteven Rostedt (Red Hat) 		 * not currently active, we can just free them
2191a4c35ed2SSteven Rostedt (Red Hat) 		 * without synchronizing all CPUs.
2192a4c35ed2SSteven Rostedt (Red Hat) 		 */
2193a4c35ed2SSteven Rostedt (Red Hat) 		if (ops->flags & FTRACE_OPS_FL_CONTROL)
2194a4c35ed2SSteven Rostedt (Red Hat) 			control_ops_free(ops);
21958a56d776SSteven Rostedt (Red Hat) 		return 0;
2196a4c35ed2SSteven Rostedt (Red Hat) 	}
21973d083395SSteven Rostedt 
2198d61f82d0SSteven Rostedt 	ftrace_run_update_code(command);
2199a4c35ed2SSteven Rostedt (Red Hat) 
2200a4c35ed2SSteven Rostedt (Red Hat) 	/*
2201a4c35ed2SSteven Rostedt (Red Hat) 	 * Dynamic ops may be freed, we must make sure that all
2202a4c35ed2SSteven Rostedt (Red Hat) 	 * callers are done before leaving this function.
2203a4c35ed2SSteven Rostedt (Red Hat) 	 * The same goes for freeing the per_cpu data of the control
2204a4c35ed2SSteven Rostedt (Red Hat) 	 * ops.
2205a4c35ed2SSteven Rostedt (Red Hat) 	 *
2206a4c35ed2SSteven Rostedt (Red Hat) 	 * Again, normal synchronize_sched() is not good enough.
2207a4c35ed2SSteven Rostedt (Red Hat) 	 * We need to do a hard force of sched synchronization.
2208a4c35ed2SSteven Rostedt (Red Hat) 	 * This is because we use preempt_disable() to do RCU, but
2209a4c35ed2SSteven Rostedt (Red Hat) 	 * the function tracers can be called where RCU is not watching
2210a4c35ed2SSteven Rostedt (Red Hat) 	 * (like before user_exit()). We can not rely on the RCU
2211a4c35ed2SSteven Rostedt (Red Hat) 	 * infrastructure to do the synchronization, thus we must do it
2212a4c35ed2SSteven Rostedt (Red Hat) 	 * ourselves.
2213a4c35ed2SSteven Rostedt (Red Hat) 	 */
2214a4c35ed2SSteven Rostedt (Red Hat) 	if (ops->flags & (FTRACE_OPS_FL_DYNAMIC | FTRACE_OPS_FL_CONTROL)) {
2215a4c35ed2SSteven Rostedt (Red Hat) 		schedule_on_each_cpu(ftrace_sync);
2216a4c35ed2SSteven Rostedt (Red Hat) 
2217a4c35ed2SSteven Rostedt (Red Hat) 		if (ops->flags & FTRACE_OPS_FL_CONTROL)
2218a4c35ed2SSteven Rostedt (Red Hat) 			control_ops_free(ops);
2219a4c35ed2SSteven Rostedt (Red Hat) 	}
2220a4c35ed2SSteven Rostedt (Red Hat) 
22218a56d776SSteven Rostedt (Red Hat) 	return 0;
22223d083395SSteven Rostedt }
22233d083395SSteven Rostedt 
2224e309b41dSIngo Molnar static void ftrace_startup_sysctl(void)
2225b0fc494fSSteven Rostedt {
22264eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
22274eebcc81SSteven Rostedt 		return;
22284eebcc81SSteven Rostedt 
2229d61f82d0SSteven Rostedt 	/* Force update next time */
2230d61f82d0SSteven Rostedt 	saved_ftrace_func = NULL;
223160a7ecf4SSteven Rostedt 	/* ftrace_start_up is true if we want ftrace running */
223260a7ecf4SSteven Rostedt 	if (ftrace_start_up)
223330fb6aa7SJiri Olsa 		ftrace_run_update_code(FTRACE_UPDATE_CALLS);
2234b0fc494fSSteven Rostedt }
2235b0fc494fSSteven Rostedt 
2236e309b41dSIngo Molnar static void ftrace_shutdown_sysctl(void)
2237b0fc494fSSteven Rostedt {
22384eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
22394eebcc81SSteven Rostedt 		return;
22404eebcc81SSteven Rostedt 
224160a7ecf4SSteven Rostedt 	/* ftrace_start_up is true if ftrace is running */
224260a7ecf4SSteven Rostedt 	if (ftrace_start_up)
224379e406d7SSteven Rostedt 		ftrace_run_update_code(FTRACE_DISABLE_CALLS);
2244b0fc494fSSteven Rostedt }
2245b0fc494fSSteven Rostedt 
22463d083395SSteven Rostedt static cycle_t		ftrace_update_time;
22473d083395SSteven Rostedt unsigned long		ftrace_update_tot_cnt;
22483d083395SSteven Rostedt 
22498c4f3c3fSSteven Rostedt (Red Hat) static inline int ops_traces_mod(struct ftrace_ops *ops)
2250f7bc8b61SSteven Rostedt {
22518c4f3c3fSSteven Rostedt (Red Hat) 	/*
22528c4f3c3fSSteven Rostedt (Red Hat) 	 * Filter_hash being empty will default to trace module.
22538c4f3c3fSSteven Rostedt (Red Hat) 	 * But notrace hash requires a test of individual module functions.
22548c4f3c3fSSteven Rostedt (Red Hat) 	 */
22558c4f3c3fSSteven Rostedt (Red Hat) 	return ftrace_hash_empty(ops->filter_hash) &&
22568c4f3c3fSSteven Rostedt (Red Hat) 		ftrace_hash_empty(ops->notrace_hash);
22578c4f3c3fSSteven Rostedt (Red Hat) }
2258f7bc8b61SSteven Rostedt 
22598c4f3c3fSSteven Rostedt (Red Hat) /*
22608c4f3c3fSSteven Rostedt (Red Hat)  * Check if the current ops references the record.
22618c4f3c3fSSteven Rostedt (Red Hat)  *
22628c4f3c3fSSteven Rostedt (Red Hat)  * If the ops traces all functions, then it was already accounted for.
22638c4f3c3fSSteven Rostedt (Red Hat)  * If the ops does not trace the current record function, skip it.
22648c4f3c3fSSteven Rostedt (Red Hat)  * If the ops ignores the function via notrace filter, skip it.
22658c4f3c3fSSteven Rostedt (Red Hat)  */
22668c4f3c3fSSteven Rostedt (Red Hat) static inline bool
22678c4f3c3fSSteven Rostedt (Red Hat) ops_references_rec(struct ftrace_ops *ops, struct dyn_ftrace *rec)
22688c4f3c3fSSteven Rostedt (Red Hat) {
22698c4f3c3fSSteven Rostedt (Red Hat) 	/* If ops isn't enabled, ignore it */
22708c4f3c3fSSteven Rostedt (Red Hat) 	if (!(ops->flags & FTRACE_OPS_FL_ENABLED))
22718c4f3c3fSSteven Rostedt (Red Hat) 		return 0;
22728c4f3c3fSSteven Rostedt (Red Hat) 
22738c4f3c3fSSteven Rostedt (Red Hat) 	/* If ops traces all mods, we already accounted for it */
22748c4f3c3fSSteven Rostedt (Red Hat) 	if (ops_traces_mod(ops))
22758c4f3c3fSSteven Rostedt (Red Hat) 		return 0;
22768c4f3c3fSSteven Rostedt (Red Hat) 
22778c4f3c3fSSteven Rostedt (Red Hat) 	/* The function must be in the filter */
22788c4f3c3fSSteven Rostedt (Red Hat) 	if (!ftrace_hash_empty(ops->filter_hash) &&
22798c4f3c3fSSteven Rostedt (Red Hat) 	    !ftrace_lookup_ip(ops->filter_hash, rec->ip))
22808c4f3c3fSSteven Rostedt (Red Hat) 		return 0;
22818c4f3c3fSSteven Rostedt (Red Hat) 
22828c4f3c3fSSteven Rostedt (Red Hat) 	/* If in notrace hash, we ignore it too */
22838c4f3c3fSSteven Rostedt (Red Hat) 	if (ftrace_lookup_ip(ops->notrace_hash, rec->ip))
22848c4f3c3fSSteven Rostedt (Red Hat) 		return 0;
22858c4f3c3fSSteven Rostedt (Red Hat) 
22868c4f3c3fSSteven Rostedt (Red Hat) 	return 1;
22878c4f3c3fSSteven Rostedt (Red Hat) }
22888c4f3c3fSSteven Rostedt (Red Hat) 
22898c4f3c3fSSteven Rostedt (Red Hat) static int referenced_filters(struct dyn_ftrace *rec)
22908c4f3c3fSSteven Rostedt (Red Hat) {
22918c4f3c3fSSteven Rostedt (Red Hat) 	struct ftrace_ops *ops;
22928c4f3c3fSSteven Rostedt (Red Hat) 	int cnt = 0;
22938c4f3c3fSSteven Rostedt (Red Hat) 
22948c4f3c3fSSteven Rostedt (Red Hat) 	for (ops = ftrace_ops_list; ops != &ftrace_list_end; ops = ops->next) {
22958c4f3c3fSSteven Rostedt (Red Hat) 		if (ops_references_rec(ops, rec))
22968c4f3c3fSSteven Rostedt (Red Hat) 		    cnt++;
22978c4f3c3fSSteven Rostedt (Red Hat) 	}
22988c4f3c3fSSteven Rostedt (Red Hat) 
22998c4f3c3fSSteven Rostedt (Red Hat) 	return cnt;
2300f7bc8b61SSteven Rostedt }
2301f7bc8b61SSteven Rostedt 
23021dc43cf0SJiri Slaby static int ftrace_update_code(struct module *mod, struct ftrace_page *new_pgs)
23033d083395SSteven Rostedt {
230485ae32aeSSteven Rostedt 	struct ftrace_page *pg;
2305e94142a6SLai Jiangshan 	struct dyn_ftrace *p;
2306f22f9a89SAbhishek Sagar 	cycle_t start, stop;
23071dc43cf0SJiri Slaby 	unsigned long update_cnt = 0;
2308f7bc8b61SSteven Rostedt 	unsigned long ref = 0;
23098c4f3c3fSSteven Rostedt (Red Hat) 	bool test = false;
231085ae32aeSSteven Rostedt 	int i;
2311f7bc8b61SSteven Rostedt 
2312f7bc8b61SSteven Rostedt 	/*
2313f7bc8b61SSteven Rostedt 	 * When adding a module, we need to check if tracers are
2314f7bc8b61SSteven Rostedt 	 * currently enabled and if they are set to trace all functions.
2315f7bc8b61SSteven Rostedt 	 * If they are, we need to enable the module functions as well
2316f7bc8b61SSteven Rostedt 	 * as update the reference counts for those function records.
2317f7bc8b61SSteven Rostedt 	 */
2318f7bc8b61SSteven Rostedt 	if (mod) {
2319f7bc8b61SSteven Rostedt 		struct ftrace_ops *ops;
2320f7bc8b61SSteven Rostedt 
2321f7bc8b61SSteven Rostedt 		for (ops = ftrace_ops_list;
2322f7bc8b61SSteven Rostedt 		     ops != &ftrace_list_end; ops = ops->next) {
23238c4f3c3fSSteven Rostedt (Red Hat) 			if (ops->flags & FTRACE_OPS_FL_ENABLED) {
23248c4f3c3fSSteven Rostedt (Red Hat) 				if (ops_traces_mod(ops))
2325f7bc8b61SSteven Rostedt 					ref++;
23268c4f3c3fSSteven Rostedt (Red Hat) 				else
23278c4f3c3fSSteven Rostedt (Red Hat) 					test = true;
23288c4f3c3fSSteven Rostedt (Red Hat) 			}
2329f7bc8b61SSteven Rostedt 		}
2330f7bc8b61SSteven Rostedt 	}
23313d083395SSteven Rostedt 
2332750ed1a4SIngo Molnar 	start = ftrace_now(raw_smp_processor_id());
23333d083395SSteven Rostedt 
23341dc43cf0SJiri Slaby 	for (pg = new_pgs; pg; pg = pg->next) {
2335f22f9a89SAbhishek Sagar 
233685ae32aeSSteven Rostedt 		for (i = 0; i < pg->index; i++) {
23378c4f3c3fSSteven Rostedt (Red Hat) 			int cnt = ref;
23388c4f3c3fSSteven Rostedt (Red Hat) 
233908f5ac90SSteven Rostedt 			/* If something went wrong, bail without enabling anything */
234008f5ac90SSteven Rostedt 			if (unlikely(ftrace_disabled))
234108f5ac90SSteven Rostedt 				return -1;
23423d083395SSteven Rostedt 
234385ae32aeSSteven Rostedt 			p = &pg->records[i];
23448c4f3c3fSSteven Rostedt (Red Hat) 			if (test)
23458c4f3c3fSSteven Rostedt (Red Hat) 				cnt += referenced_filters(p);
23468c4f3c3fSSteven Rostedt (Red Hat) 			p->flags = cnt;
2347f22f9a89SAbhishek Sagar 
23485cb084bbSJiri Olsa 			/*
234925985edcSLucas De Marchi 			 * Do the initial record conversion from mcount jump
23505cb084bbSJiri Olsa 			 * to the NOP instructions.
23515cb084bbSJiri Olsa 			 */
235232082309SSteven Rostedt 			if (!ftrace_code_disable(mod, p))
2353d2c8c3eaSSteven Rostedt 				break;
23545cb084bbSJiri Olsa 
23551dc43cf0SJiri Slaby 			update_cnt++;
23565cb084bbSJiri Olsa 
23575cb084bbSJiri Olsa 			/*
23585cb084bbSJiri Olsa 			 * If the tracing is enabled, go ahead and enable the record.
23595cb084bbSJiri Olsa 			 *
23605cb084bbSJiri Olsa 			 * The reason not to enable the record immediatelly is the
23615cb084bbSJiri Olsa 			 * inherent check of ftrace_make_nop/ftrace_make_call for
23625cb084bbSJiri Olsa 			 * correct previous instructions.  Making first the NOP
23635cb084bbSJiri Olsa 			 * conversion puts the module to the correct state, thus
23645cb084bbSJiri Olsa 			 * passing the ftrace_make_call check.
23655cb084bbSJiri Olsa 			 */
23668c4f3c3fSSteven Rostedt (Red Hat) 			if (ftrace_start_up && cnt) {
23675cb084bbSJiri Olsa 				int failed = __ftrace_replace_code(p, 1);
236832082309SSteven Rostedt 				if (failed)
23695cb084bbSJiri Olsa 					ftrace_bug(failed, p->ip);
23705cb084bbSJiri Olsa 			}
23715cb084bbSJiri Olsa 		}
237285ae32aeSSteven Rostedt 	}
237385ae32aeSSteven Rostedt 
2374750ed1a4SIngo Molnar 	stop = ftrace_now(raw_smp_processor_id());
23753d083395SSteven Rostedt 	ftrace_update_time = stop - start;
23761dc43cf0SJiri Slaby 	ftrace_update_tot_cnt += update_cnt;
23773d083395SSteven Rostedt 
237816444a8aSArnaldo Carvalho de Melo 	return 0;
237916444a8aSArnaldo Carvalho de Melo }
238016444a8aSArnaldo Carvalho de Melo 
2381a7900875SSteven Rostedt static int ftrace_allocate_records(struct ftrace_page *pg, int count)
23823c1720f0SSteven Rostedt {
2383a7900875SSteven Rostedt 	int order;
23843c1720f0SSteven Rostedt 	int cnt;
23853c1720f0SSteven Rostedt 
2386a7900875SSteven Rostedt 	if (WARN_ON(!count))
2387a7900875SSteven Rostedt 		return -EINVAL;
2388a7900875SSteven Rostedt 
2389a7900875SSteven Rostedt 	order = get_count_order(DIV_ROUND_UP(count, ENTRIES_PER_PAGE));
23903c1720f0SSteven Rostedt 
23913c1720f0SSteven Rostedt 	/*
2392a7900875SSteven Rostedt 	 * We want to fill as much as possible. No more than a page
2393a7900875SSteven Rostedt 	 * may be empty.
23943c1720f0SSteven Rostedt 	 */
2395a7900875SSteven Rostedt 	while ((PAGE_SIZE << order) / ENTRY_SIZE >= count + ENTRIES_PER_PAGE)
2396a7900875SSteven Rostedt 		order--;
23973c1720f0SSteven Rostedt 
2398a7900875SSteven Rostedt  again:
2399a7900875SSteven Rostedt 	pg->records = (void *)__get_free_pages(GFP_KERNEL | __GFP_ZERO, order);
2400a7900875SSteven Rostedt 
2401a7900875SSteven Rostedt 	if (!pg->records) {
2402a7900875SSteven Rostedt 		/* if we can't allocate this size, try something smaller */
2403a7900875SSteven Rostedt 		if (!order)
2404a7900875SSteven Rostedt 			return -ENOMEM;
2405a7900875SSteven Rostedt 		order >>= 1;
2406a7900875SSteven Rostedt 		goto again;
2407a7900875SSteven Rostedt 	}
2408a7900875SSteven Rostedt 
2409a7900875SSteven Rostedt 	cnt = (PAGE_SIZE << order) / ENTRY_SIZE;
2410a7900875SSteven Rostedt 	pg->size = cnt;
2411a7900875SSteven Rostedt 
2412a7900875SSteven Rostedt 	if (cnt > count)
2413a7900875SSteven Rostedt 		cnt = count;
2414a7900875SSteven Rostedt 
2415a7900875SSteven Rostedt 	return cnt;
2416a7900875SSteven Rostedt }
2417a7900875SSteven Rostedt 
2418a7900875SSteven Rostedt static struct ftrace_page *
2419a7900875SSteven Rostedt ftrace_allocate_pages(unsigned long num_to_init)
2420a7900875SSteven Rostedt {
2421a7900875SSteven Rostedt 	struct ftrace_page *start_pg;
2422a7900875SSteven Rostedt 	struct ftrace_page *pg;
2423a7900875SSteven Rostedt 	int order;
2424a7900875SSteven Rostedt 	int cnt;
2425a7900875SSteven Rostedt 
2426a7900875SSteven Rostedt 	if (!num_to_init)
2427a7900875SSteven Rostedt 		return 0;
2428a7900875SSteven Rostedt 
2429a7900875SSteven Rostedt 	start_pg = pg = kzalloc(sizeof(*pg), GFP_KERNEL);
2430a7900875SSteven Rostedt 	if (!pg)
2431a7900875SSteven Rostedt 		return NULL;
2432a7900875SSteven Rostedt 
2433a7900875SSteven Rostedt 	/*
2434a7900875SSteven Rostedt 	 * Try to allocate as much as possible in one continues
2435a7900875SSteven Rostedt 	 * location that fills in all of the space. We want to
2436a7900875SSteven Rostedt 	 * waste as little space as possible.
2437a7900875SSteven Rostedt 	 */
2438a7900875SSteven Rostedt 	for (;;) {
2439a7900875SSteven Rostedt 		cnt = ftrace_allocate_records(pg, num_to_init);
2440a7900875SSteven Rostedt 		if (cnt < 0)
2441a7900875SSteven Rostedt 			goto free_pages;
2442a7900875SSteven Rostedt 
2443a7900875SSteven Rostedt 		num_to_init -= cnt;
2444a7900875SSteven Rostedt 		if (!num_to_init)
2445a7900875SSteven Rostedt 			break;
2446a7900875SSteven Rostedt 
2447a7900875SSteven Rostedt 		pg->next = kzalloc(sizeof(*pg), GFP_KERNEL);
2448a7900875SSteven Rostedt 		if (!pg->next)
2449a7900875SSteven Rostedt 			goto free_pages;
2450a7900875SSteven Rostedt 
2451a7900875SSteven Rostedt 		pg = pg->next;
2452a7900875SSteven Rostedt 	}
2453a7900875SSteven Rostedt 
2454a7900875SSteven Rostedt 	return start_pg;
2455a7900875SSteven Rostedt 
2456a7900875SSteven Rostedt  free_pages:
2457a7900875SSteven Rostedt 	while (start_pg) {
2458a7900875SSteven Rostedt 		order = get_count_order(pg->size / ENTRIES_PER_PAGE);
2459a7900875SSteven Rostedt 		free_pages((unsigned long)pg->records, order);
2460a7900875SSteven Rostedt 		start_pg = pg->next;
2461a7900875SSteven Rostedt 		kfree(pg);
2462a7900875SSteven Rostedt 		pg = start_pg;
2463a7900875SSteven Rostedt 	}
2464a7900875SSteven Rostedt 	pr_info("ftrace: FAILED to allocate memory for functions\n");
2465a7900875SSteven Rostedt 	return NULL;
2466a7900875SSteven Rostedt }
2467a7900875SSteven Rostedt 
24685072c59fSSteven Rostedt #define FTRACE_BUFF_MAX (KSYM_SYMBOL_LEN+4) /* room for wildcards */
24695072c59fSSteven Rostedt 
24705072c59fSSteven Rostedt struct ftrace_iterator {
247198c4fd04SSteven Rostedt 	loff_t				pos;
24722bccfffdSSteven Rostedt 	loff_t				func_pos;
24735072c59fSSteven Rostedt 	struct ftrace_page		*pg;
24744aeb6967SSteven Rostedt 	struct dyn_ftrace		*func;
24754aeb6967SSteven Rostedt 	struct ftrace_func_probe	*probe;
24764aeb6967SSteven Rostedt 	struct trace_parser		parser;
24771cf41dd7SSteven Rostedt 	struct ftrace_hash		*hash;
247833dc9b12SSteven Rostedt 	struct ftrace_ops		*ops;
24798fc0c701SSteven Rostedt 	int				hidx;
2480431aa3fbSSteven Rostedt 	int				idx;
24815072c59fSSteven Rostedt 	unsigned			flags;
24825072c59fSSteven Rostedt };
24835072c59fSSteven Rostedt 
2484e309b41dSIngo Molnar static void *
24854aeb6967SSteven Rostedt t_hash_next(struct seq_file *m, loff_t *pos)
24868fc0c701SSteven Rostedt {
24878fc0c701SSteven Rostedt 	struct ftrace_iterator *iter = m->private;
24884aeb6967SSteven Rostedt 	struct hlist_node *hnd = NULL;
24898fc0c701SSteven Rostedt 	struct hlist_head *hhd;
24908fc0c701SSteven Rostedt 
24918fc0c701SSteven Rostedt 	(*pos)++;
249298c4fd04SSteven Rostedt 	iter->pos = *pos;
24938fc0c701SSteven Rostedt 
24944aeb6967SSteven Rostedt 	if (iter->probe)
24954aeb6967SSteven Rostedt 		hnd = &iter->probe->node;
24968fc0c701SSteven Rostedt  retry:
24978fc0c701SSteven Rostedt 	if (iter->hidx >= FTRACE_FUNC_HASHSIZE)
24988fc0c701SSteven Rostedt 		return NULL;
24998fc0c701SSteven Rostedt 
25008fc0c701SSteven Rostedt 	hhd = &ftrace_func_hash[iter->hidx];
25018fc0c701SSteven Rostedt 
25028fc0c701SSteven Rostedt 	if (hlist_empty(hhd)) {
25038fc0c701SSteven Rostedt 		iter->hidx++;
25048fc0c701SSteven Rostedt 		hnd = NULL;
25058fc0c701SSteven Rostedt 		goto retry;
25068fc0c701SSteven Rostedt 	}
25078fc0c701SSteven Rostedt 
25088fc0c701SSteven Rostedt 	if (!hnd)
25098fc0c701SSteven Rostedt 		hnd = hhd->first;
25108fc0c701SSteven Rostedt 	else {
25118fc0c701SSteven Rostedt 		hnd = hnd->next;
25128fc0c701SSteven Rostedt 		if (!hnd) {
25138fc0c701SSteven Rostedt 			iter->hidx++;
25148fc0c701SSteven Rostedt 			goto retry;
25158fc0c701SSteven Rostedt 		}
25168fc0c701SSteven Rostedt 	}
25178fc0c701SSteven Rostedt 
25184aeb6967SSteven Rostedt 	if (WARN_ON_ONCE(!hnd))
25194aeb6967SSteven Rostedt 		return NULL;
25204aeb6967SSteven Rostedt 
25214aeb6967SSteven Rostedt 	iter->probe = hlist_entry(hnd, struct ftrace_func_probe, node);
25224aeb6967SSteven Rostedt 
25234aeb6967SSteven Rostedt 	return iter;
25248fc0c701SSteven Rostedt }
25258fc0c701SSteven Rostedt 
25268fc0c701SSteven Rostedt static void *t_hash_start(struct seq_file *m, loff_t *pos)
25278fc0c701SSteven Rostedt {
25288fc0c701SSteven Rostedt 	struct ftrace_iterator *iter = m->private;
25298fc0c701SSteven Rostedt 	void *p = NULL;
2530d82d6244SLi Zefan 	loff_t l;
2531d82d6244SLi Zefan 
253269a3083cSSteven Rostedt 	if (!(iter->flags & FTRACE_ITER_DO_HASH))
253369a3083cSSteven Rostedt 		return NULL;
253469a3083cSSteven Rostedt 
25352bccfffdSSteven Rostedt 	if (iter->func_pos > *pos)
25362bccfffdSSteven Rostedt 		return NULL;
25378fc0c701SSteven Rostedt 
2538d82d6244SLi Zefan 	iter->hidx = 0;
25392bccfffdSSteven Rostedt 	for (l = 0; l <= (*pos - iter->func_pos); ) {
25404aeb6967SSteven Rostedt 		p = t_hash_next(m, &l);
2541d82d6244SLi Zefan 		if (!p)
2542d82d6244SLi Zefan 			break;
2543d82d6244SLi Zefan 	}
25444aeb6967SSteven Rostedt 	if (!p)
25454aeb6967SSteven Rostedt 		return NULL;
25464aeb6967SSteven Rostedt 
254798c4fd04SSteven Rostedt 	/* Only set this if we have an item */
254898c4fd04SSteven Rostedt 	iter->flags |= FTRACE_ITER_HASH;
254998c4fd04SSteven Rostedt 
25504aeb6967SSteven Rostedt 	return iter;
25518fc0c701SSteven Rostedt }
25528fc0c701SSteven Rostedt 
25534aeb6967SSteven Rostedt static int
25544aeb6967SSteven Rostedt t_hash_show(struct seq_file *m, struct ftrace_iterator *iter)
25558fc0c701SSteven Rostedt {
2556b6887d79SSteven Rostedt 	struct ftrace_func_probe *rec;
25578fc0c701SSteven Rostedt 
25584aeb6967SSteven Rostedt 	rec = iter->probe;
25594aeb6967SSteven Rostedt 	if (WARN_ON_ONCE(!rec))
25604aeb6967SSteven Rostedt 		return -EIO;
25618fc0c701SSteven Rostedt 
2562809dcf29SSteven Rostedt 	if (rec->ops->print)
2563809dcf29SSteven Rostedt 		return rec->ops->print(m, rec->ip, rec->ops, rec->data);
2564809dcf29SSteven Rostedt 
2565b375a11aSSteven Rostedt 	seq_printf(m, "%ps:%ps", (void *)rec->ip, (void *)rec->ops->func);
25668fc0c701SSteven Rostedt 
25678fc0c701SSteven Rostedt 	if (rec->data)
25688fc0c701SSteven Rostedt 		seq_printf(m, ":%p", rec->data);
25698fc0c701SSteven Rostedt 	seq_putc(m, '\n');
25708fc0c701SSteven Rostedt 
25718fc0c701SSteven Rostedt 	return 0;
25728fc0c701SSteven Rostedt }
25738fc0c701SSteven Rostedt 
25748fc0c701SSteven Rostedt static void *
25755072c59fSSteven Rostedt t_next(struct seq_file *m, void *v, loff_t *pos)
25765072c59fSSteven Rostedt {
25775072c59fSSteven Rostedt 	struct ftrace_iterator *iter = m->private;
2578fc13cb0cSSteven Rostedt 	struct ftrace_ops *ops = iter->ops;
25795072c59fSSteven Rostedt 	struct dyn_ftrace *rec = NULL;
25805072c59fSSteven Rostedt 
258145a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
258245a4a237SSteven Rostedt 		return NULL;
258345a4a237SSteven Rostedt 
25848fc0c701SSteven Rostedt 	if (iter->flags & FTRACE_ITER_HASH)
25854aeb6967SSteven Rostedt 		return t_hash_next(m, pos);
25868fc0c701SSteven Rostedt 
25875072c59fSSteven Rostedt 	(*pos)++;
25881106b699SJiri Olsa 	iter->pos = iter->func_pos = *pos;
25895072c59fSSteven Rostedt 
25900c75a3edSSteven Rostedt 	if (iter->flags & FTRACE_ITER_PRINTALL)
259157c072c7SSteven Rostedt 		return t_hash_start(m, pos);
25920c75a3edSSteven Rostedt 
25935072c59fSSteven Rostedt  retry:
25945072c59fSSteven Rostedt 	if (iter->idx >= iter->pg->index) {
25955072c59fSSteven Rostedt 		if (iter->pg->next) {
25965072c59fSSteven Rostedt 			iter->pg = iter->pg->next;
25975072c59fSSteven Rostedt 			iter->idx = 0;
25985072c59fSSteven Rostedt 			goto retry;
25995072c59fSSteven Rostedt 		}
26005072c59fSSteven Rostedt 	} else {
26015072c59fSSteven Rostedt 		rec = &iter->pg->records[iter->idx++];
260232082309SSteven Rostedt 		if (((iter->flags & FTRACE_ITER_FILTER) &&
2603f45948e8SSteven Rostedt 		     !(ftrace_lookup_ip(ops->filter_hash, rec->ip))) ||
26040183fb1cSSteven Rostedt 
260541c52c0dSSteven Rostedt 		    ((iter->flags & FTRACE_ITER_NOTRACE) &&
2606647bcd03SSteven Rostedt 		     !ftrace_lookup_ip(ops->notrace_hash, rec->ip)) ||
2607647bcd03SSteven Rostedt 
2608647bcd03SSteven Rostedt 		    ((iter->flags & FTRACE_ITER_ENABLED) &&
260923ea9c4dSSteven Rostedt (Red Hat) 		     !(rec->flags & FTRACE_FL_ENABLED))) {
2610647bcd03SSteven Rostedt 
26115072c59fSSteven Rostedt 			rec = NULL;
26125072c59fSSteven Rostedt 			goto retry;
26135072c59fSSteven Rostedt 		}
26145072c59fSSteven Rostedt 	}
26155072c59fSSteven Rostedt 
26164aeb6967SSteven Rostedt 	if (!rec)
261757c072c7SSteven Rostedt 		return t_hash_start(m, pos);
26184aeb6967SSteven Rostedt 
26194aeb6967SSteven Rostedt 	iter->func = rec;
26204aeb6967SSteven Rostedt 
26214aeb6967SSteven Rostedt 	return iter;
26225072c59fSSteven Rostedt }
26235072c59fSSteven Rostedt 
262498c4fd04SSteven Rostedt static void reset_iter_read(struct ftrace_iterator *iter)
262598c4fd04SSteven Rostedt {
262698c4fd04SSteven Rostedt 	iter->pos = 0;
262798c4fd04SSteven Rostedt 	iter->func_pos = 0;
262870f77b3fSDan Carpenter 	iter->flags &= ~(FTRACE_ITER_PRINTALL | FTRACE_ITER_HASH);
26295072c59fSSteven Rostedt }
26305072c59fSSteven Rostedt 
26315072c59fSSteven Rostedt static void *t_start(struct seq_file *m, loff_t *pos)
26325072c59fSSteven Rostedt {
26335072c59fSSteven Rostedt 	struct ftrace_iterator *iter = m->private;
2634fc13cb0cSSteven Rostedt 	struct ftrace_ops *ops = iter->ops;
26355072c59fSSteven Rostedt 	void *p = NULL;
2636694ce0a5SLi Zefan 	loff_t l;
26375072c59fSSteven Rostedt 
26388fc0c701SSteven Rostedt 	mutex_lock(&ftrace_lock);
263945a4a237SSteven Rostedt 
264045a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
264145a4a237SSteven Rostedt 		return NULL;
264245a4a237SSteven Rostedt 
26430c75a3edSSteven Rostedt 	/*
264498c4fd04SSteven Rostedt 	 * If an lseek was done, then reset and start from beginning.
264598c4fd04SSteven Rostedt 	 */
264698c4fd04SSteven Rostedt 	if (*pos < iter->pos)
264798c4fd04SSteven Rostedt 		reset_iter_read(iter);
264898c4fd04SSteven Rostedt 
264998c4fd04SSteven Rostedt 	/*
26500c75a3edSSteven Rostedt 	 * For set_ftrace_filter reading, if we have the filter
26510c75a3edSSteven Rostedt 	 * off, we can short cut and just print out that all
26520c75a3edSSteven Rostedt 	 * functions are enabled.
26530c75a3edSSteven Rostedt 	 */
265406a51d93SSteven Rostedt 	if (iter->flags & FTRACE_ITER_FILTER &&
265506a51d93SSteven Rostedt 	    ftrace_hash_empty(ops->filter_hash)) {
26560c75a3edSSteven Rostedt 		if (*pos > 0)
26578fc0c701SSteven Rostedt 			return t_hash_start(m, pos);
26580c75a3edSSteven Rostedt 		iter->flags |= FTRACE_ITER_PRINTALL;
2659df091625SChris Wright 		/* reset in case of seek/pread */
2660df091625SChris Wright 		iter->flags &= ~FTRACE_ITER_HASH;
26610c75a3edSSteven Rostedt 		return iter;
26620c75a3edSSteven Rostedt 	}
26630c75a3edSSteven Rostedt 
26648fc0c701SSteven Rostedt 	if (iter->flags & FTRACE_ITER_HASH)
26658fc0c701SSteven Rostedt 		return t_hash_start(m, pos);
26668fc0c701SSteven Rostedt 
266798c4fd04SSteven Rostedt 	/*
266898c4fd04SSteven Rostedt 	 * Unfortunately, we need to restart at ftrace_pages_start
266998c4fd04SSteven Rostedt 	 * every time we let go of the ftrace_mutex. This is because
267098c4fd04SSteven Rostedt 	 * those pointers can change without the lock.
267198c4fd04SSteven Rostedt 	 */
2672694ce0a5SLi Zefan 	iter->pg = ftrace_pages_start;
2673694ce0a5SLi Zefan 	iter->idx = 0;
2674694ce0a5SLi Zefan 	for (l = 0; l <= *pos; ) {
2675694ce0a5SLi Zefan 		p = t_next(m, p, &l);
2676694ce0a5SLi Zefan 		if (!p)
2677694ce0a5SLi Zefan 			break;
267850cdaf08SLiming Wang 	}
26795821e1b7Swalimis 
268069a3083cSSteven Rostedt 	if (!p)
26818fc0c701SSteven Rostedt 		return t_hash_start(m, pos);
26828fc0c701SSteven Rostedt 
26834aeb6967SSteven Rostedt 	return iter;
26845072c59fSSteven Rostedt }
26855072c59fSSteven Rostedt 
26865072c59fSSteven Rostedt static void t_stop(struct seq_file *m, void *p)
26875072c59fSSteven Rostedt {
26888fc0c701SSteven Rostedt 	mutex_unlock(&ftrace_lock);
26895072c59fSSteven Rostedt }
26905072c59fSSteven Rostedt 
26915072c59fSSteven Rostedt static int t_show(struct seq_file *m, void *v)
26925072c59fSSteven Rostedt {
26930c75a3edSSteven Rostedt 	struct ftrace_iterator *iter = m->private;
26944aeb6967SSteven Rostedt 	struct dyn_ftrace *rec;
26955072c59fSSteven Rostedt 
26968fc0c701SSteven Rostedt 	if (iter->flags & FTRACE_ITER_HASH)
26974aeb6967SSteven Rostedt 		return t_hash_show(m, iter);
26988fc0c701SSteven Rostedt 
26990c75a3edSSteven Rostedt 	if (iter->flags & FTRACE_ITER_PRINTALL) {
27000c75a3edSSteven Rostedt 		seq_printf(m, "#### all functions enabled ####\n");
27010c75a3edSSteven Rostedt 		return 0;
27020c75a3edSSteven Rostedt 	}
27030c75a3edSSteven Rostedt 
27044aeb6967SSteven Rostedt 	rec = iter->func;
27054aeb6967SSteven Rostedt 
27065072c59fSSteven Rostedt 	if (!rec)
27075072c59fSSteven Rostedt 		return 0;
27085072c59fSSteven Rostedt 
2709647bcd03SSteven Rostedt 	seq_printf(m, "%ps", (void *)rec->ip);
2710647bcd03SSteven Rostedt 	if (iter->flags & FTRACE_ITER_ENABLED)
271108f6fba5SSteven Rostedt 		seq_printf(m, " (%ld)%s",
271208f6fba5SSteven Rostedt 			   rec->flags & ~FTRACE_FL_MASK,
271308f6fba5SSteven Rostedt 			   rec->flags & FTRACE_FL_REGS ? " R" : "");
2714647bcd03SSteven Rostedt 	seq_printf(m, "\n");
27155072c59fSSteven Rostedt 
27165072c59fSSteven Rostedt 	return 0;
27175072c59fSSteven Rostedt }
27185072c59fSSteven Rostedt 
271988e9d34cSJames Morris static const struct seq_operations show_ftrace_seq_ops = {
27205072c59fSSteven Rostedt 	.start = t_start,
27215072c59fSSteven Rostedt 	.next = t_next,
27225072c59fSSteven Rostedt 	.stop = t_stop,
27235072c59fSSteven Rostedt 	.show = t_show,
27245072c59fSSteven Rostedt };
27255072c59fSSteven Rostedt 
2726e309b41dSIngo Molnar static int
27275072c59fSSteven Rostedt ftrace_avail_open(struct inode *inode, struct file *file)
27285072c59fSSteven Rostedt {
27295072c59fSSteven Rostedt 	struct ftrace_iterator *iter;
27305072c59fSSteven Rostedt 
27314eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
27324eebcc81SSteven Rostedt 		return -ENODEV;
27334eebcc81SSteven Rostedt 
273450e18b94SJiri Olsa 	iter = __seq_open_private(file, &show_ftrace_seq_ops, sizeof(*iter));
273550e18b94SJiri Olsa 	if (iter) {
27365072c59fSSteven Rostedt 		iter->pg = ftrace_pages_start;
2737fc13cb0cSSteven Rostedt 		iter->ops = &global_ops;
27384bf39a94SIngo Molnar 	}
27395072c59fSSteven Rostedt 
274050e18b94SJiri Olsa 	return iter ? 0 : -ENOMEM;
27415072c59fSSteven Rostedt }
27425072c59fSSteven Rostedt 
2743647bcd03SSteven Rostedt static int
2744647bcd03SSteven Rostedt ftrace_enabled_open(struct inode *inode, struct file *file)
2745647bcd03SSteven Rostedt {
2746647bcd03SSteven Rostedt 	struct ftrace_iterator *iter;
2747647bcd03SSteven Rostedt 
2748647bcd03SSteven Rostedt 	if (unlikely(ftrace_disabled))
2749647bcd03SSteven Rostedt 		return -ENODEV;
2750647bcd03SSteven Rostedt 
275150e18b94SJiri Olsa 	iter = __seq_open_private(file, &show_ftrace_seq_ops, sizeof(*iter));
275250e18b94SJiri Olsa 	if (iter) {
2753647bcd03SSteven Rostedt 		iter->pg = ftrace_pages_start;
2754647bcd03SSteven Rostedt 		iter->flags = FTRACE_ITER_ENABLED;
2755fc13cb0cSSteven Rostedt 		iter->ops = &global_ops;
2756647bcd03SSteven Rostedt 	}
2757647bcd03SSteven Rostedt 
275850e18b94SJiri Olsa 	return iter ? 0 : -ENOMEM;
2759647bcd03SSteven Rostedt }
2760647bcd03SSteven Rostedt 
27611cf41dd7SSteven Rostedt static void ftrace_filter_reset(struct ftrace_hash *hash)
27625072c59fSSteven Rostedt {
276352baf119SSteven Rostedt 	mutex_lock(&ftrace_lock);
27641cf41dd7SSteven Rostedt 	ftrace_hash_clear(hash);
276552baf119SSteven Rostedt 	mutex_unlock(&ftrace_lock);
27665072c59fSSteven Rostedt }
27675072c59fSSteven Rostedt 
2768fc13cb0cSSteven Rostedt /**
2769fc13cb0cSSteven Rostedt  * ftrace_regex_open - initialize function tracer filter files
2770fc13cb0cSSteven Rostedt  * @ops: The ftrace_ops that hold the hash filters
2771fc13cb0cSSteven Rostedt  * @flag: The type of filter to process
2772fc13cb0cSSteven Rostedt  * @inode: The inode, usually passed in to your open routine
2773fc13cb0cSSteven Rostedt  * @file: The file, usually passed in to your open routine
2774fc13cb0cSSteven Rostedt  *
2775fc13cb0cSSteven Rostedt  * ftrace_regex_open() initializes the filter files for the
2776fc13cb0cSSteven Rostedt  * @ops. Depending on @flag it may process the filter hash or
2777fc13cb0cSSteven Rostedt  * the notrace hash of @ops. With this called from the open
2778fc13cb0cSSteven Rostedt  * routine, you can use ftrace_filter_write() for the write
2779fc13cb0cSSteven Rostedt  * routine if @flag has FTRACE_ITER_FILTER set, or
2780fc13cb0cSSteven Rostedt  * ftrace_notrace_write() if @flag has FTRACE_ITER_NOTRACE set.
2781098c879eSSteven Rostedt (Red Hat)  * tracing_lseek() should be used as the lseek routine, and
2782fc13cb0cSSteven Rostedt  * release must call ftrace_regex_release().
2783fc13cb0cSSteven Rostedt  */
2784fc13cb0cSSteven Rostedt int
2785f45948e8SSteven Rostedt ftrace_regex_open(struct ftrace_ops *ops, int flag,
27861cf41dd7SSteven Rostedt 		  struct inode *inode, struct file *file)
27875072c59fSSteven Rostedt {
27885072c59fSSteven Rostedt 	struct ftrace_iterator *iter;
2789f45948e8SSteven Rostedt 	struct ftrace_hash *hash;
27905072c59fSSteven Rostedt 	int ret = 0;
27915072c59fSSteven Rostedt 
2792f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
2793f04f24fbSMasami Hiramatsu 
27944eebcc81SSteven Rostedt 	if (unlikely(ftrace_disabled))
27954eebcc81SSteven Rostedt 		return -ENODEV;
27964eebcc81SSteven Rostedt 
27975072c59fSSteven Rostedt 	iter = kzalloc(sizeof(*iter), GFP_KERNEL);
27985072c59fSSteven Rostedt 	if (!iter)
27995072c59fSSteven Rostedt 		return -ENOMEM;
28005072c59fSSteven Rostedt 
2801689fd8b6S[email protected] 	if (trace_parser_get_init(&iter->parser, FTRACE_BUFF_MAX)) {
2802689fd8b6S[email protected] 		kfree(iter);
2803689fd8b6S[email protected] 		return -ENOMEM;
2804689fd8b6S[email protected] 	}
2805689fd8b6S[email protected] 
28063f2367baSMasami Hiramatsu 	iter->ops = ops;
28073f2367baSMasami Hiramatsu 	iter->flags = flag;
28083f2367baSMasami Hiramatsu 
28093f2367baSMasami Hiramatsu 	mutex_lock(&ops->regex_lock);
28103f2367baSMasami Hiramatsu 
2811f45948e8SSteven Rostedt 	if (flag & FTRACE_ITER_NOTRACE)
2812f45948e8SSteven Rostedt 		hash = ops->notrace_hash;
2813f45948e8SSteven Rostedt 	else
2814f45948e8SSteven Rostedt 		hash = ops->filter_hash;
2815f45948e8SSteven Rostedt 
281633dc9b12SSteven Rostedt 	if (file->f_mode & FMODE_WRITE) {
281733dc9b12SSteven Rostedt 		iter->hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, hash);
281833dc9b12SSteven Rostedt 		if (!iter->hash) {
281933dc9b12SSteven Rostedt 			trace_parser_put(&iter->parser);
282033dc9b12SSteven Rostedt 			kfree(iter);
28213f2367baSMasami Hiramatsu 			ret = -ENOMEM;
28223f2367baSMasami Hiramatsu 			goto out_unlock;
282333dc9b12SSteven Rostedt 		}
282433dc9b12SSteven Rostedt 	}
28251cf41dd7SSteven Rostedt 
28265072c59fSSteven Rostedt 	if ((file->f_mode & FMODE_WRITE) &&
28278650ae32SSteven Rostedt 	    (file->f_flags & O_TRUNC))
282833dc9b12SSteven Rostedt 		ftrace_filter_reset(iter->hash);
28295072c59fSSteven Rostedt 
28305072c59fSSteven Rostedt 	if (file->f_mode & FMODE_READ) {
28315072c59fSSteven Rostedt 		iter->pg = ftrace_pages_start;
28325072c59fSSteven Rostedt 
28335072c59fSSteven Rostedt 		ret = seq_open(file, &show_ftrace_seq_ops);
28345072c59fSSteven Rostedt 		if (!ret) {
28355072c59fSSteven Rostedt 			struct seq_file *m = file->private_data;
28365072c59fSSteven Rostedt 			m->private = iter;
283779fe249cSLi Zefan 		} else {
283833dc9b12SSteven Rostedt 			/* Failed */
283933dc9b12SSteven Rostedt 			free_ftrace_hash(iter->hash);
284079fe249cSLi Zefan 			trace_parser_put(&iter->parser);
28415072c59fSSteven Rostedt 			kfree(iter);
284279fe249cSLi Zefan 		}
28435072c59fSSteven Rostedt 	} else
28445072c59fSSteven Rostedt 		file->private_data = iter;
28453f2367baSMasami Hiramatsu 
28463f2367baSMasami Hiramatsu  out_unlock:
2847f04f24fbSMasami Hiramatsu 	mutex_unlock(&ops->regex_lock);
28485072c59fSSteven Rostedt 
28495072c59fSSteven Rostedt 	return ret;
28505072c59fSSteven Rostedt }
28515072c59fSSteven Rostedt 
285241c52c0dSSteven Rostedt static int
285341c52c0dSSteven Rostedt ftrace_filter_open(struct inode *inode, struct file *file)
285441c52c0dSSteven Rostedt {
2855e3b3e2e8SSteven Rostedt (Red Hat) 	struct ftrace_ops *ops = inode->i_private;
2856e3b3e2e8SSteven Rostedt (Red Hat) 
2857e3b3e2e8SSteven Rostedt (Red Hat) 	return ftrace_regex_open(ops,
285869a3083cSSteven Rostedt 			FTRACE_ITER_FILTER | FTRACE_ITER_DO_HASH,
28591cf41dd7SSteven Rostedt 			inode, file);
286041c52c0dSSteven Rostedt }
286141c52c0dSSteven Rostedt 
286241c52c0dSSteven Rostedt static int
286341c52c0dSSteven Rostedt ftrace_notrace_open(struct inode *inode, struct file *file)
286441c52c0dSSteven Rostedt {
2865e3b3e2e8SSteven Rostedt (Red Hat) 	struct ftrace_ops *ops = inode->i_private;
2866e3b3e2e8SSteven Rostedt (Red Hat) 
2867e3b3e2e8SSteven Rostedt (Red Hat) 	return ftrace_regex_open(ops, FTRACE_ITER_NOTRACE,
28681cf41dd7SSteven Rostedt 				 inode, file);
286941c52c0dSSteven Rostedt }
287041c52c0dSSteven Rostedt 
287164e7c440SSteven Rostedt static int ftrace_match(char *str, char *regex, int len, int type)
28729f4801e3SSteven Rostedt {
28739f4801e3SSteven Rostedt 	int matched = 0;
2874751e9983SLi Zefan 	int slen;
28759f4801e3SSteven Rostedt 
28769f4801e3SSteven Rostedt 	switch (type) {
28779f4801e3SSteven Rostedt 	case MATCH_FULL:
28789f4801e3SSteven Rostedt 		if (strcmp(str, regex) == 0)
28799f4801e3SSteven Rostedt 			matched = 1;
28809f4801e3SSteven Rostedt 		break;
28819f4801e3SSteven Rostedt 	case MATCH_FRONT_ONLY:
28829f4801e3SSteven Rostedt 		if (strncmp(str, regex, len) == 0)
28839f4801e3SSteven Rostedt 			matched = 1;
28849f4801e3SSteven Rostedt 		break;
28859f4801e3SSteven Rostedt 	case MATCH_MIDDLE_ONLY:
28869f4801e3SSteven Rostedt 		if (strstr(str, regex))
28879f4801e3SSteven Rostedt 			matched = 1;
28889f4801e3SSteven Rostedt 		break;
28899f4801e3SSteven Rostedt 	case MATCH_END_ONLY:
2890751e9983SLi Zefan 		slen = strlen(str);
2891751e9983SLi Zefan 		if (slen >= len && memcmp(str + slen - len, regex, len) == 0)
28929f4801e3SSteven Rostedt 			matched = 1;
28939f4801e3SSteven Rostedt 		break;
28949f4801e3SSteven Rostedt 	}
28959f4801e3SSteven Rostedt 
28969f4801e3SSteven Rostedt 	return matched;
28979f4801e3SSteven Rostedt }
28989f4801e3SSteven Rostedt 
2899b448c4e3SSteven Rostedt static int
29001cf41dd7SSteven Rostedt enter_record(struct ftrace_hash *hash, struct dyn_ftrace *rec, int not)
290164e7c440SSteven Rostedt {
2902b448c4e3SSteven Rostedt 	struct ftrace_func_entry *entry;
2903b448c4e3SSteven Rostedt 	int ret = 0;
2904b448c4e3SSteven Rostedt 
29051cf41dd7SSteven Rostedt 	entry = ftrace_lookup_ip(hash, rec->ip);
2906b448c4e3SSteven Rostedt 	if (not) {
2907b448c4e3SSteven Rostedt 		/* Do nothing if it doesn't exist */
2908b448c4e3SSteven Rostedt 		if (!entry)
2909b448c4e3SSteven Rostedt 			return 0;
2910b448c4e3SSteven Rostedt 
291133dc9b12SSteven Rostedt 		free_hash_entry(hash, entry);
2912b448c4e3SSteven Rostedt 	} else {
2913b448c4e3SSteven Rostedt 		/* Do nothing if it exists */
2914b448c4e3SSteven Rostedt 		if (entry)
2915b448c4e3SSteven Rostedt 			return 0;
2916b448c4e3SSteven Rostedt 
2917b448c4e3SSteven Rostedt 		ret = add_hash_entry(hash, rec->ip);
2918b448c4e3SSteven Rostedt 	}
2919b448c4e3SSteven Rostedt 	return ret;
2920996e87beSSteven Rostedt }
2921996e87beSSteven Rostedt 
292264e7c440SSteven Rostedt static int
2923b9df92d2SSteven Rostedt ftrace_match_record(struct dyn_ftrace *rec, char *mod,
2924b9df92d2SSteven Rostedt 		    char *regex, int len, int type)
292564e7c440SSteven Rostedt {
292664e7c440SSteven Rostedt 	char str[KSYM_SYMBOL_LEN];
2927b9df92d2SSteven Rostedt 	char *modname;
292864e7c440SSteven Rostedt 
2929b9df92d2SSteven Rostedt 	kallsyms_lookup(rec->ip, NULL, NULL, &modname, str);
2930b9df92d2SSteven Rostedt 
2931b9df92d2SSteven Rostedt 	if (mod) {
2932b9df92d2SSteven Rostedt 		/* module lookup requires matching the module */
2933b9df92d2SSteven Rostedt 		if (!modname || strcmp(modname, mod))
2934b9df92d2SSteven Rostedt 			return 0;
2935b9df92d2SSteven Rostedt 
2936b9df92d2SSteven Rostedt 		/* blank search means to match all funcs in the mod */
2937b9df92d2SSteven Rostedt 		if (!len)
2938b9df92d2SSteven Rostedt 			return 1;
2939b9df92d2SSteven Rostedt 	}
2940b9df92d2SSteven Rostedt 
294164e7c440SSteven Rostedt 	return ftrace_match(str, regex, len, type);
294264e7c440SSteven Rostedt }
294364e7c440SSteven Rostedt 
29441cf41dd7SSteven Rostedt static int
29451cf41dd7SSteven Rostedt match_records(struct ftrace_hash *hash, char *buff,
29461cf41dd7SSteven Rostedt 	      int len, char *mod, int not)
29479f4801e3SSteven Rostedt {
2948b9df92d2SSteven Rostedt 	unsigned search_len = 0;
29499f4801e3SSteven Rostedt 	struct ftrace_page *pg;
29509f4801e3SSteven Rostedt 	struct dyn_ftrace *rec;
2951b9df92d2SSteven Rostedt 	int type = MATCH_FULL;
2952b9df92d2SSteven Rostedt 	char *search = buff;
2953311d16daSLi Zefan 	int found = 0;
2954b448c4e3SSteven Rostedt 	int ret;
29559f4801e3SSteven Rostedt 
2956b9df92d2SSteven Rostedt 	if (len) {
29573f6fe06dSFrederic Weisbecker 		type = filter_parse_regex(buff, len, &search, &not);
29589f4801e3SSteven Rostedt 		search_len = strlen(search);
2959b9df92d2SSteven Rostedt 	}
2960b9df92d2SSteven Rostedt 
296152baf119SSteven Rostedt 	mutex_lock(&ftrace_lock);
2962b9df92d2SSteven Rostedt 
2963b9df92d2SSteven Rostedt 	if (unlikely(ftrace_disabled))
2964b9df92d2SSteven Rostedt 		goto out_unlock;
2965b9df92d2SSteven Rostedt 
2966265c831cSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
2967b9df92d2SSteven Rostedt 		if (ftrace_match_record(rec, mod, search, search_len, type)) {
29681cf41dd7SSteven Rostedt 			ret = enter_record(hash, rec, not);
2969b448c4e3SSteven Rostedt 			if (ret < 0) {
2970b448c4e3SSteven Rostedt 				found = ret;
2971b448c4e3SSteven Rostedt 				goto out_unlock;
2972b448c4e3SSteven Rostedt 			}
2973311d16daSLi Zefan 			found = 1;
29745072c59fSSteven Rostedt 		}
2975265c831cSSteven Rostedt 	} while_for_each_ftrace_rec();
2976b9df92d2SSteven Rostedt  out_unlock:
297752baf119SSteven Rostedt 	mutex_unlock(&ftrace_lock);
2978311d16daSLi Zefan 
2979311d16daSLi Zefan 	return found;
29805072c59fSSteven Rostedt }
29815072c59fSSteven Rostedt 
298264e7c440SSteven Rostedt static int
29831cf41dd7SSteven Rostedt ftrace_match_records(struct ftrace_hash *hash, char *buff, int len)
298464e7c440SSteven Rostedt {
29851cf41dd7SSteven Rostedt 	return match_records(hash, buff, len, NULL, 0);
298664e7c440SSteven Rostedt }
298764e7c440SSteven Rostedt 
29881cf41dd7SSteven Rostedt static int
29891cf41dd7SSteven Rostedt ftrace_match_module_records(struct ftrace_hash *hash, char *buff, char *mod)
299064e7c440SSteven Rostedt {
299164e7c440SSteven Rostedt 	int not = 0;
29926a24a244SSteven Rostedt 
299364e7c440SSteven Rostedt 	/* blank or '*' mean the same */
299464e7c440SSteven Rostedt 	if (strcmp(buff, "*") == 0)
299564e7c440SSteven Rostedt 		buff[0] = 0;
299664e7c440SSteven Rostedt 
299764e7c440SSteven Rostedt 	/* handle the case of 'dont filter this module' */
299864e7c440SSteven Rostedt 	if (strcmp(buff, "!") == 0 || strcmp(buff, "!*") == 0) {
299964e7c440SSteven Rostedt 		buff[0] = 0;
300064e7c440SSteven Rostedt 		not = 1;
300164e7c440SSteven Rostedt 	}
300264e7c440SSteven Rostedt 
30031cf41dd7SSteven Rostedt 	return match_records(hash, buff, strlen(buff), mod, not);
300464e7c440SSteven Rostedt }
300564e7c440SSteven Rostedt 
3006f6180773SSteven Rostedt /*
3007f6180773SSteven Rostedt  * We register the module command as a template to show others how
3008f6180773SSteven Rostedt  * to register the a command as well.
3009f6180773SSteven Rostedt  */
3010f6180773SSteven Rostedt 
3011f6180773SSteven Rostedt static int
301243dd61c9SSteven Rostedt ftrace_mod_callback(struct ftrace_hash *hash,
301343dd61c9SSteven Rostedt 		    char *func, char *cmd, char *param, int enable)
3014f6180773SSteven Rostedt {
3015f6180773SSteven Rostedt 	char *mod;
3016b448c4e3SSteven Rostedt 	int ret = -EINVAL;
3017f6180773SSteven Rostedt 
3018f6180773SSteven Rostedt 	/*
3019f6180773SSteven Rostedt 	 * cmd == 'mod' because we only registered this func
3020f6180773SSteven Rostedt 	 * for the 'mod' ftrace_func_command.
3021f6180773SSteven Rostedt 	 * But if you register one func with multiple commands,
3022f6180773SSteven Rostedt 	 * you can tell which command was used by the cmd
3023f6180773SSteven Rostedt 	 * parameter.
3024f6180773SSteven Rostedt 	 */
3025f6180773SSteven Rostedt 
3026f6180773SSteven Rostedt 	/* we must have a module name */
3027f6180773SSteven Rostedt 	if (!param)
3028b448c4e3SSteven Rostedt 		return ret;
3029f6180773SSteven Rostedt 
3030f6180773SSteven Rostedt 	mod = strsep(&param, ":");
3031f6180773SSteven Rostedt 	if (!strlen(mod))
3032b448c4e3SSteven Rostedt 		return ret;
3033f6180773SSteven Rostedt 
30341cf41dd7SSteven Rostedt 	ret = ftrace_match_module_records(hash, func, mod);
3035b448c4e3SSteven Rostedt 	if (!ret)
3036b448c4e3SSteven Rostedt 		ret = -EINVAL;
3037b448c4e3SSteven Rostedt 	if (ret < 0)
3038b448c4e3SSteven Rostedt 		return ret;
3039b448c4e3SSteven Rostedt 
3040f6180773SSteven Rostedt 	return 0;
3041f6180773SSteven Rostedt }
3042f6180773SSteven Rostedt 
3043f6180773SSteven Rostedt static struct ftrace_func_command ftrace_mod_cmd = {
3044f6180773SSteven Rostedt 	.name			= "mod",
3045f6180773SSteven Rostedt 	.func			= ftrace_mod_callback,
3046f6180773SSteven Rostedt };
3047f6180773SSteven Rostedt 
3048f6180773SSteven Rostedt static int __init ftrace_mod_cmd_init(void)
3049f6180773SSteven Rostedt {
3050f6180773SSteven Rostedt 	return register_ftrace_command(&ftrace_mod_cmd);
3051f6180773SSteven Rostedt }
30526f415672SSteven Rostedt core_initcall(ftrace_mod_cmd_init);
3053f6180773SSteven Rostedt 
30542f5f6ad9SSteven Rostedt static void function_trace_probe_call(unsigned long ip, unsigned long parent_ip,
3055a1e2e31dSSteven Rostedt 				      struct ftrace_ops *op, struct pt_regs *pt_regs)
305659df055fSSteven Rostedt {
3057b6887d79SSteven Rostedt 	struct ftrace_func_probe *entry;
305859df055fSSteven Rostedt 	struct hlist_head *hhd;
305959df055fSSteven Rostedt 	unsigned long key;
306059df055fSSteven Rostedt 
306159df055fSSteven Rostedt 	key = hash_long(ip, FTRACE_HASH_BITS);
306259df055fSSteven Rostedt 
306359df055fSSteven Rostedt 	hhd = &ftrace_func_hash[key];
306459df055fSSteven Rostedt 
306559df055fSSteven Rostedt 	if (hlist_empty(hhd))
306659df055fSSteven Rostedt 		return;
306759df055fSSteven Rostedt 
306859df055fSSteven Rostedt 	/*
306959df055fSSteven Rostedt 	 * Disable preemption for these calls to prevent a RCU grace
307059df055fSSteven Rostedt 	 * period. This syncs the hash iteration and freeing of items
307159df055fSSteven Rostedt 	 * on the hash. rcu_read_lock is too dangerous here.
307259df055fSSteven Rostedt 	 */
30735168ae50SSteven Rostedt 	preempt_disable_notrace();
30741bb539caSSteven Rostedt 	hlist_for_each_entry_rcu_notrace(entry, hhd, node) {
307559df055fSSteven Rostedt 		if (entry->ip == ip)
307659df055fSSteven Rostedt 			entry->ops->func(ip, parent_ip, &entry->data);
307759df055fSSteven Rostedt 	}
30785168ae50SSteven Rostedt 	preempt_enable_notrace();
307959df055fSSteven Rostedt }
308059df055fSSteven Rostedt 
3081b6887d79SSteven Rostedt static struct ftrace_ops trace_probe_ops __read_mostly =
308259df055fSSteven Rostedt {
3083b6887d79SSteven Rostedt 	.func		= function_trace_probe_call,
3084f04f24fbSMasami Hiramatsu 	.flags		= FTRACE_OPS_FL_INITIALIZED,
3085f04f24fbSMasami Hiramatsu 	INIT_REGEX_LOCK(trace_probe_ops)
308659df055fSSteven Rostedt };
308759df055fSSteven Rostedt 
3088b6887d79SSteven Rostedt static int ftrace_probe_registered;
308959df055fSSteven Rostedt 
3090b6887d79SSteven Rostedt static void __enable_ftrace_function_probe(void)
309159df055fSSteven Rostedt {
3092b848914cSSteven Rostedt 	int ret;
309359df055fSSteven Rostedt 	int i;
309459df055fSSteven Rostedt 
309519dd603eSSteven Rostedt (Red Hat) 	if (ftrace_probe_registered) {
309619dd603eSSteven Rostedt (Red Hat) 		/* still need to update the function call sites */
309719dd603eSSteven Rostedt (Red Hat) 		if (ftrace_enabled)
309819dd603eSSteven Rostedt (Red Hat) 			ftrace_run_update_code(FTRACE_UPDATE_CALLS);
309959df055fSSteven Rostedt 		return;
310019dd603eSSteven Rostedt (Red Hat) 	}
310159df055fSSteven Rostedt 
310259df055fSSteven Rostedt 	for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) {
310359df055fSSteven Rostedt 		struct hlist_head *hhd = &ftrace_func_hash[i];
310459df055fSSteven Rostedt 		if (hhd->first)
310559df055fSSteven Rostedt 			break;
310659df055fSSteven Rostedt 	}
310759df055fSSteven Rostedt 	/* Nothing registered? */
310859df055fSSteven Rostedt 	if (i == FTRACE_FUNC_HASHSIZE)
310959df055fSSteven Rostedt 		return;
311059df055fSSteven Rostedt 
3111a1cd6173SSteven Rostedt 	ret = ftrace_startup(&trace_probe_ops, 0);
3112b848914cSSteven Rostedt 
3113b6887d79SSteven Rostedt 	ftrace_probe_registered = 1;
311459df055fSSteven Rostedt }
311559df055fSSteven Rostedt 
3116b6887d79SSteven Rostedt static void __disable_ftrace_function_probe(void)
311759df055fSSteven Rostedt {
311859df055fSSteven Rostedt 	int i;
311959df055fSSteven Rostedt 
3120b6887d79SSteven Rostedt 	if (!ftrace_probe_registered)
312159df055fSSteven Rostedt 		return;
312259df055fSSteven Rostedt 
312359df055fSSteven Rostedt 	for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) {
312459df055fSSteven Rostedt 		struct hlist_head *hhd = &ftrace_func_hash[i];
312559df055fSSteven Rostedt 		if (hhd->first)
312659df055fSSteven Rostedt 			return;
312759df055fSSteven Rostedt 	}
312859df055fSSteven Rostedt 
312959df055fSSteven Rostedt 	/* no more funcs left */
3130b848914cSSteven Rostedt 	ftrace_shutdown(&trace_probe_ops, 0);
3131b848914cSSteven Rostedt 
3132b6887d79SSteven Rostedt 	ftrace_probe_registered = 0;
313359df055fSSteven Rostedt }
313459df055fSSteven Rostedt 
313559df055fSSteven Rostedt 
31367818b388SSteven Rostedt (Red Hat) static void ftrace_free_entry(struct ftrace_func_probe *entry)
313759df055fSSteven Rostedt {
313859df055fSSteven Rostedt 	if (entry->ops->free)
3139e67efb93SSteven Rostedt (Red Hat) 		entry->ops->free(entry->ops, entry->ip, &entry->data);
314059df055fSSteven Rostedt 	kfree(entry);
314159df055fSSteven Rostedt }
314259df055fSSteven Rostedt 
314359df055fSSteven Rostedt int
3144b6887d79SSteven Rostedt register_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops,
314559df055fSSteven Rostedt 			      void *data)
314659df055fSSteven Rostedt {
3147b6887d79SSteven Rostedt 	struct ftrace_func_probe *entry;
3148e1df4cb6SSteven Rostedt (Red Hat) 	struct ftrace_hash **orig_hash = &trace_probe_ops.filter_hash;
3149e1df4cb6SSteven Rostedt (Red Hat) 	struct ftrace_hash *hash;
315059df055fSSteven Rostedt 	struct ftrace_page *pg;
315159df055fSSteven Rostedt 	struct dyn_ftrace *rec;
315259df055fSSteven Rostedt 	int type, len, not;
31536a24a244SSteven Rostedt 	unsigned long key;
315459df055fSSteven Rostedt 	int count = 0;
315559df055fSSteven Rostedt 	char *search;
3156e1df4cb6SSteven Rostedt (Red Hat) 	int ret;
315759df055fSSteven Rostedt 
31583f6fe06dSFrederic Weisbecker 	type = filter_parse_regex(glob, strlen(glob), &search, &not);
315959df055fSSteven Rostedt 	len = strlen(search);
316059df055fSSteven Rostedt 
3161b6887d79SSteven Rostedt 	/* we do not support '!' for function probes */
316259df055fSSteven Rostedt 	if (WARN_ON(not))
316359df055fSSteven Rostedt 		return -EINVAL;
316459df055fSSteven Rostedt 
31653f2367baSMasami Hiramatsu 	mutex_lock(&trace_probe_ops.regex_lock);
316659df055fSSteven Rostedt 
3167e1df4cb6SSteven Rostedt (Red Hat) 	hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, *orig_hash);
3168e1df4cb6SSteven Rostedt (Red Hat) 	if (!hash) {
3169e1df4cb6SSteven Rostedt (Red Hat) 		count = -ENOMEM;
31705ae0bf59SSteven Rostedt (Red Hat) 		goto out;
3171e1df4cb6SSteven Rostedt (Red Hat) 	}
3172e1df4cb6SSteven Rostedt (Red Hat) 
3173e1df4cb6SSteven Rostedt (Red Hat) 	if (unlikely(ftrace_disabled)) {
3174e1df4cb6SSteven Rostedt (Red Hat) 		count = -ENODEV;
31755ae0bf59SSteven Rostedt (Red Hat) 		goto out;
3176e1df4cb6SSteven Rostedt (Red Hat) 	}
317745a4a237SSteven Rostedt 
31785ae0bf59SSteven Rostedt (Red Hat) 	mutex_lock(&ftrace_lock);
31795ae0bf59SSteven Rostedt (Red Hat) 
318059df055fSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
318159df055fSSteven Rostedt 
3182b9df92d2SSteven Rostedt 		if (!ftrace_match_record(rec, NULL, search, len, type))
318359df055fSSteven Rostedt 			continue;
318459df055fSSteven Rostedt 
318559df055fSSteven Rostedt 		entry = kmalloc(sizeof(*entry), GFP_KERNEL);
318659df055fSSteven Rostedt 		if (!entry) {
3187b6887d79SSteven Rostedt 			/* If we did not process any, then return error */
318859df055fSSteven Rostedt 			if (!count)
318959df055fSSteven Rostedt 				count = -ENOMEM;
319059df055fSSteven Rostedt 			goto out_unlock;
319159df055fSSteven Rostedt 		}
319259df055fSSteven Rostedt 
319359df055fSSteven Rostedt 		count++;
319459df055fSSteven Rostedt 
319559df055fSSteven Rostedt 		entry->data = data;
319659df055fSSteven Rostedt 
319759df055fSSteven Rostedt 		/*
319859df055fSSteven Rostedt 		 * The caller might want to do something special
319959df055fSSteven Rostedt 		 * for each function we find. We call the callback
320059df055fSSteven Rostedt 		 * to give the caller an opportunity to do so.
320159df055fSSteven Rostedt 		 */
3202e67efb93SSteven Rostedt (Red Hat) 		if (ops->init) {
3203e67efb93SSteven Rostedt (Red Hat) 			if (ops->init(ops, rec->ip, &entry->data) < 0) {
320459df055fSSteven Rostedt 				/* caller does not like this func */
320559df055fSSteven Rostedt 				kfree(entry);
320659df055fSSteven Rostedt 				continue;
320759df055fSSteven Rostedt 			}
320859df055fSSteven Rostedt 		}
320959df055fSSteven Rostedt 
3210e1df4cb6SSteven Rostedt (Red Hat) 		ret = enter_record(hash, rec, 0);
3211e1df4cb6SSteven Rostedt (Red Hat) 		if (ret < 0) {
3212e1df4cb6SSteven Rostedt (Red Hat) 			kfree(entry);
3213e1df4cb6SSteven Rostedt (Red Hat) 			count = ret;
3214e1df4cb6SSteven Rostedt (Red Hat) 			goto out_unlock;
3215e1df4cb6SSteven Rostedt (Red Hat) 		}
3216e1df4cb6SSteven Rostedt (Red Hat) 
321759df055fSSteven Rostedt 		entry->ops = ops;
321859df055fSSteven Rostedt 		entry->ip = rec->ip;
321959df055fSSteven Rostedt 
322059df055fSSteven Rostedt 		key = hash_long(entry->ip, FTRACE_HASH_BITS);
322159df055fSSteven Rostedt 		hlist_add_head_rcu(&entry->node, &ftrace_func_hash[key]);
322259df055fSSteven Rostedt 
322359df055fSSteven Rostedt 	} while_for_each_ftrace_rec();
3224e1df4cb6SSteven Rostedt (Red Hat) 
3225e1df4cb6SSteven Rostedt (Red Hat) 	ret = ftrace_hash_move(&trace_probe_ops, 1, orig_hash, hash);
3226e1df4cb6SSteven Rostedt (Red Hat) 	if (ret < 0)
3227e1df4cb6SSteven Rostedt (Red Hat) 		count = ret;
3228e1df4cb6SSteven Rostedt (Red Hat) 
3229b6887d79SSteven Rostedt 	__enable_ftrace_function_probe();
323059df055fSSteven Rostedt 
323159df055fSSteven Rostedt  out_unlock:
32325ae0bf59SSteven Rostedt (Red Hat) 	mutex_unlock(&ftrace_lock);
32335ae0bf59SSteven Rostedt (Red Hat)  out:
32343f2367baSMasami Hiramatsu 	mutex_unlock(&trace_probe_ops.regex_lock);
3235e1df4cb6SSteven Rostedt (Red Hat) 	free_ftrace_hash(hash);
323659df055fSSteven Rostedt 
323759df055fSSteven Rostedt 	return count;
323859df055fSSteven Rostedt }
323959df055fSSteven Rostedt 
324059df055fSSteven Rostedt enum {
3241b6887d79SSteven Rostedt 	PROBE_TEST_FUNC		= 1,
3242b6887d79SSteven Rostedt 	PROBE_TEST_DATA		= 2
324359df055fSSteven Rostedt };
324459df055fSSteven Rostedt 
324559df055fSSteven Rostedt static void
3246b6887d79SSteven Rostedt __unregister_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops,
324759df055fSSteven Rostedt 				  void *data, int flags)
324859df055fSSteven Rostedt {
3249e1df4cb6SSteven Rostedt (Red Hat) 	struct ftrace_func_entry *rec_entry;
3250b6887d79SSteven Rostedt 	struct ftrace_func_probe *entry;
32517818b388SSteven Rostedt (Red Hat) 	struct ftrace_func_probe *p;
3252e1df4cb6SSteven Rostedt (Red Hat) 	struct ftrace_hash **orig_hash = &trace_probe_ops.filter_hash;
32537818b388SSteven Rostedt (Red Hat) 	struct list_head free_list;
3254e1df4cb6SSteven Rostedt (Red Hat) 	struct ftrace_hash *hash;
3255b67bfe0dSSasha Levin 	struct hlist_node *tmp;
325659df055fSSteven Rostedt 	char str[KSYM_SYMBOL_LEN];
325759df055fSSteven Rostedt 	int type = MATCH_FULL;
325859df055fSSteven Rostedt 	int i, len = 0;
325959df055fSSteven Rostedt 	char *search;
326059df055fSSteven Rostedt 
3261b36461daSAtsushi Tsuji 	if (glob && (strcmp(glob, "*") == 0 || !strlen(glob)))
326259df055fSSteven Rostedt 		glob = NULL;
3263b36461daSAtsushi Tsuji 	else if (glob) {
326459df055fSSteven Rostedt 		int not;
326559df055fSSteven Rostedt 
32663f6fe06dSFrederic Weisbecker 		type = filter_parse_regex(glob, strlen(glob), &search, &not);
326759df055fSSteven Rostedt 		len = strlen(search);
326859df055fSSteven Rostedt 
3269b6887d79SSteven Rostedt 		/* we do not support '!' for function probes */
327059df055fSSteven Rostedt 		if (WARN_ON(not))
327159df055fSSteven Rostedt 			return;
327259df055fSSteven Rostedt 	}
327359df055fSSteven Rostedt 
32743f2367baSMasami Hiramatsu 	mutex_lock(&trace_probe_ops.regex_lock);
3275e1df4cb6SSteven Rostedt (Red Hat) 
3276e1df4cb6SSteven Rostedt (Red Hat) 	hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, *orig_hash);
3277e1df4cb6SSteven Rostedt (Red Hat) 	if (!hash)
3278e1df4cb6SSteven Rostedt (Red Hat) 		/* Hmm, should report this somehow */
3279e1df4cb6SSteven Rostedt (Red Hat) 		goto out_unlock;
3280e1df4cb6SSteven Rostedt (Red Hat) 
32817818b388SSteven Rostedt (Red Hat) 	INIT_LIST_HEAD(&free_list);
32827818b388SSteven Rostedt (Red Hat) 
328359df055fSSteven Rostedt 	for (i = 0; i < FTRACE_FUNC_HASHSIZE; i++) {
328459df055fSSteven Rostedt 		struct hlist_head *hhd = &ftrace_func_hash[i];
328559df055fSSteven Rostedt 
3286b67bfe0dSSasha Levin 		hlist_for_each_entry_safe(entry, tmp, hhd, node) {
328759df055fSSteven Rostedt 
328859df055fSSteven Rostedt 			/* break up if statements for readability */
3289b6887d79SSteven Rostedt 			if ((flags & PROBE_TEST_FUNC) && entry->ops != ops)
329059df055fSSteven Rostedt 				continue;
329159df055fSSteven Rostedt 
3292b6887d79SSteven Rostedt 			if ((flags & PROBE_TEST_DATA) && entry->data != data)
329359df055fSSteven Rostedt 				continue;
329459df055fSSteven Rostedt 
329559df055fSSteven Rostedt 			/* do this last, since it is the most expensive */
329659df055fSSteven Rostedt 			if (glob) {
329759df055fSSteven Rostedt 				kallsyms_lookup(entry->ip, NULL, NULL,
329859df055fSSteven Rostedt 						NULL, str);
329959df055fSSteven Rostedt 				if (!ftrace_match(str, glob, len, type))
330059df055fSSteven Rostedt 					continue;
330159df055fSSteven Rostedt 			}
330259df055fSSteven Rostedt 
3303e1df4cb6SSteven Rostedt (Red Hat) 			rec_entry = ftrace_lookup_ip(hash, entry->ip);
3304e1df4cb6SSteven Rostedt (Red Hat) 			/* It is possible more than one entry had this ip */
3305e1df4cb6SSteven Rostedt (Red Hat) 			if (rec_entry)
3306e1df4cb6SSteven Rostedt (Red Hat) 				free_hash_entry(hash, rec_entry);
3307e1df4cb6SSteven Rostedt (Red Hat) 
3308740466bcSSteven Rostedt (Red Hat) 			hlist_del_rcu(&entry->node);
33097818b388SSteven Rostedt (Red Hat) 			list_add(&entry->free_list, &free_list);
331059df055fSSteven Rostedt 		}
331159df055fSSteven Rostedt 	}
33123f2367baSMasami Hiramatsu 	mutex_lock(&ftrace_lock);
3313b6887d79SSteven Rostedt 	__disable_ftrace_function_probe();
3314e1df4cb6SSteven Rostedt (Red Hat) 	/*
3315e1df4cb6SSteven Rostedt (Red Hat) 	 * Remove after the disable is called. Otherwise, if the last
3316e1df4cb6SSteven Rostedt (Red Hat) 	 * probe is removed, a null hash means *all enabled*.
3317e1df4cb6SSteven Rostedt (Red Hat) 	 */
3318e1df4cb6SSteven Rostedt (Red Hat) 	ftrace_hash_move(&trace_probe_ops, 1, orig_hash, hash);
33197818b388SSteven Rostedt (Red Hat) 	synchronize_sched();
33207818b388SSteven Rostedt (Red Hat) 	list_for_each_entry_safe(entry, p, &free_list, free_list) {
33217818b388SSteven Rostedt (Red Hat) 		list_del(&entry->free_list);
33227818b388SSteven Rostedt (Red Hat) 		ftrace_free_entry(entry);
33237818b388SSteven Rostedt (Red Hat) 	}
33243f2367baSMasami Hiramatsu 	mutex_unlock(&ftrace_lock);
33257818b388SSteven Rostedt (Red Hat) 
3326e1df4cb6SSteven Rostedt (Red Hat)  out_unlock:
33273f2367baSMasami Hiramatsu 	mutex_unlock(&trace_probe_ops.regex_lock);
3328e1df4cb6SSteven Rostedt (Red Hat) 	free_ftrace_hash(hash);
332959df055fSSteven Rostedt }
333059df055fSSteven Rostedt 
333159df055fSSteven Rostedt void
3332b6887d79SSteven Rostedt unregister_ftrace_function_probe(char *glob, struct ftrace_probe_ops *ops,
333359df055fSSteven Rostedt 				void *data)
333459df055fSSteven Rostedt {
3335b6887d79SSteven Rostedt 	__unregister_ftrace_function_probe(glob, ops, data,
3336b6887d79SSteven Rostedt 					  PROBE_TEST_FUNC | PROBE_TEST_DATA);
333759df055fSSteven Rostedt }
333859df055fSSteven Rostedt 
333959df055fSSteven Rostedt void
3340b6887d79SSteven Rostedt unregister_ftrace_function_probe_func(char *glob, struct ftrace_probe_ops *ops)
334159df055fSSteven Rostedt {
3342b6887d79SSteven Rostedt 	__unregister_ftrace_function_probe(glob, ops, NULL, PROBE_TEST_FUNC);
334359df055fSSteven Rostedt }
334459df055fSSteven Rostedt 
3345b6887d79SSteven Rostedt void unregister_ftrace_function_probe_all(char *glob)
334659df055fSSteven Rostedt {
3347b6887d79SSteven Rostedt 	__unregister_ftrace_function_probe(glob, NULL, NULL, 0);
334859df055fSSteven Rostedt }
334959df055fSSteven Rostedt 
3350f6180773SSteven Rostedt static LIST_HEAD(ftrace_commands);
3351f6180773SSteven Rostedt static DEFINE_MUTEX(ftrace_cmd_mutex);
3352f6180773SSteven Rostedt 
335338de93abSTom Zanussi /*
335438de93abSTom Zanussi  * Currently we only register ftrace commands from __init, so mark this
335538de93abSTom Zanussi  * __init too.
335638de93abSTom Zanussi  */
335738de93abSTom Zanussi __init int register_ftrace_command(struct ftrace_func_command *cmd)
3358f6180773SSteven Rostedt {
3359f6180773SSteven Rostedt 	struct ftrace_func_command *p;
3360f6180773SSteven Rostedt 	int ret = 0;
3361f6180773SSteven Rostedt 
3362f6180773SSteven Rostedt 	mutex_lock(&ftrace_cmd_mutex);
3363f6180773SSteven Rostedt 	list_for_each_entry(p, &ftrace_commands, list) {
3364f6180773SSteven Rostedt 		if (strcmp(cmd->name, p->name) == 0) {
3365f6180773SSteven Rostedt 			ret = -EBUSY;
3366f6180773SSteven Rostedt 			goto out_unlock;
3367f6180773SSteven Rostedt 		}
3368f6180773SSteven Rostedt 	}
3369f6180773SSteven Rostedt 	list_add(&cmd->list, &ftrace_commands);
3370f6180773SSteven Rostedt  out_unlock:
3371f6180773SSteven Rostedt 	mutex_unlock(&ftrace_cmd_mutex);
3372f6180773SSteven Rostedt 
3373f6180773SSteven Rostedt 	return ret;
3374f6180773SSteven Rostedt }
3375f6180773SSteven Rostedt 
337638de93abSTom Zanussi /*
337738de93abSTom Zanussi  * Currently we only unregister ftrace commands from __init, so mark
337838de93abSTom Zanussi  * this __init too.
337938de93abSTom Zanussi  */
338038de93abSTom Zanussi __init int unregister_ftrace_command(struct ftrace_func_command *cmd)
3381f6180773SSteven Rostedt {
3382f6180773SSteven Rostedt 	struct ftrace_func_command *p, *n;
3383f6180773SSteven Rostedt 	int ret = -ENODEV;
3384f6180773SSteven Rostedt 
3385f6180773SSteven Rostedt 	mutex_lock(&ftrace_cmd_mutex);
3386f6180773SSteven Rostedt 	list_for_each_entry_safe(p, n, &ftrace_commands, list) {
3387f6180773SSteven Rostedt 		if (strcmp(cmd->name, p->name) == 0) {
3388f6180773SSteven Rostedt 			ret = 0;
3389f6180773SSteven Rostedt 			list_del_init(&p->list);
3390f6180773SSteven Rostedt 			goto out_unlock;
3391f6180773SSteven Rostedt 		}
3392f6180773SSteven Rostedt 	}
3393f6180773SSteven Rostedt  out_unlock:
3394f6180773SSteven Rostedt 	mutex_unlock(&ftrace_cmd_mutex);
3395f6180773SSteven Rostedt 
3396f6180773SSteven Rostedt 	return ret;
3397f6180773SSteven Rostedt }
3398f6180773SSteven Rostedt 
339933dc9b12SSteven Rostedt static int ftrace_process_regex(struct ftrace_hash *hash,
340033dc9b12SSteven Rostedt 				char *buff, int len, int enable)
340164e7c440SSteven Rostedt {
3402f6180773SSteven Rostedt 	char *func, *command, *next = buff;
34036a24a244SSteven Rostedt 	struct ftrace_func_command *p;
34040aff1c0cSGuoWen Li 	int ret = -EINVAL;
340564e7c440SSteven Rostedt 
340664e7c440SSteven Rostedt 	func = strsep(&next, ":");
340764e7c440SSteven Rostedt 
340864e7c440SSteven Rostedt 	if (!next) {
34091cf41dd7SSteven Rostedt 		ret = ftrace_match_records(hash, func, len);
3410b448c4e3SSteven Rostedt 		if (!ret)
3411b448c4e3SSteven Rostedt 			ret = -EINVAL;
3412b448c4e3SSteven Rostedt 		if (ret < 0)
3413311d16daSLi Zefan 			return ret;
3414b448c4e3SSteven Rostedt 		return 0;
341564e7c440SSteven Rostedt 	}
341664e7c440SSteven Rostedt 
3417f6180773SSteven Rostedt 	/* command found */
341864e7c440SSteven Rostedt 
341964e7c440SSteven Rostedt 	command = strsep(&next, ":");
342064e7c440SSteven Rostedt 
3421f6180773SSteven Rostedt 	mutex_lock(&ftrace_cmd_mutex);
3422f6180773SSteven Rostedt 	list_for_each_entry(p, &ftrace_commands, list) {
3423f6180773SSteven Rostedt 		if (strcmp(p->name, command) == 0) {
342443dd61c9SSteven Rostedt 			ret = p->func(hash, func, command, next, enable);
3425f6180773SSteven Rostedt 			goto out_unlock;
342664e7c440SSteven Rostedt 		}
3427f6180773SSteven Rostedt 	}
3428f6180773SSteven Rostedt  out_unlock:
3429f6180773SSteven Rostedt 	mutex_unlock(&ftrace_cmd_mutex);
343064e7c440SSteven Rostedt 
3431f6180773SSteven Rostedt 	return ret;
343264e7c440SSteven Rostedt }
343364e7c440SSteven Rostedt 
3434e309b41dSIngo Molnar static ssize_t
343541c52c0dSSteven Rostedt ftrace_regex_write(struct file *file, const char __user *ubuf,
343641c52c0dSSteven Rostedt 		   size_t cnt, loff_t *ppos, int enable)
34375072c59fSSteven Rostedt {
34385072c59fSSteven Rostedt 	struct ftrace_iterator *iter;
3439689fd8b6S[email protected] 	struct trace_parser *parser;
3440689fd8b6S[email protected] 	ssize_t ret, read;
34415072c59fSSteven Rostedt 
34424ba7978eSLi Zefan 	if (!cnt)
34435072c59fSSteven Rostedt 		return 0;
34445072c59fSSteven Rostedt 
34455072c59fSSteven Rostedt 	if (file->f_mode & FMODE_READ) {
34465072c59fSSteven Rostedt 		struct seq_file *m = file->private_data;
34475072c59fSSteven Rostedt 		iter = m->private;
34485072c59fSSteven Rostedt 	} else
34495072c59fSSteven Rostedt 		iter = file->private_data;
34505072c59fSSteven Rostedt 
3451f04f24fbSMasami Hiramatsu 	if (unlikely(ftrace_disabled))
34523f2367baSMasami Hiramatsu 		return -ENODEV;
34533f2367baSMasami Hiramatsu 
34543f2367baSMasami Hiramatsu 	/* iter->hash is a local copy, so we don't need regex_lock */
3455f04f24fbSMasami Hiramatsu 
3456689fd8b6S[email protected] 	parser = &iter->parser;
3457689fd8b6S[email protected] 	read = trace_get_user(parser, ubuf, cnt, ppos);
34585072c59fSSteven Rostedt 
34594ba7978eSLi Zefan 	if (read >= 0 && trace_parser_loaded(parser) &&
3460689fd8b6S[email protected] 	    !trace_parser_cont(parser)) {
346133dc9b12SSteven Rostedt 		ret = ftrace_process_regex(iter->hash, parser->buffer,
3462689fd8b6S[email protected] 					   parser->idx, enable);
3463313254a9SLi Zefan 		trace_parser_clear(parser);
34647c088b51SSteven Rostedt (Red Hat) 		if (ret < 0)
34653f2367baSMasami Hiramatsu 			goto out;
34665072c59fSSteven Rostedt 	}
34675072c59fSSteven Rostedt 
34685072c59fSSteven Rostedt 	ret = read;
34693f2367baSMasami Hiramatsu  out:
34705072c59fSSteven Rostedt 	return ret;
34715072c59fSSteven Rostedt }
34725072c59fSSteven Rostedt 
3473fc13cb0cSSteven Rostedt ssize_t
347441c52c0dSSteven Rostedt ftrace_filter_write(struct file *file, const char __user *ubuf,
347541c52c0dSSteven Rostedt 		    size_t cnt, loff_t *ppos)
347641c52c0dSSteven Rostedt {
347741c52c0dSSteven Rostedt 	return ftrace_regex_write(file, ubuf, cnt, ppos, 1);
347841c52c0dSSteven Rostedt }
347941c52c0dSSteven Rostedt 
3480fc13cb0cSSteven Rostedt ssize_t
348141c52c0dSSteven Rostedt ftrace_notrace_write(struct file *file, const char __user *ubuf,
348241c52c0dSSteven Rostedt 		     size_t cnt, loff_t *ppos)
348341c52c0dSSteven Rostedt {
348441c52c0dSSteven Rostedt 	return ftrace_regex_write(file, ubuf, cnt, ppos, 0);
348541c52c0dSSteven Rostedt }
348641c52c0dSSteven Rostedt 
348733dc9b12SSteven Rostedt static int
3488647664eaSMasami Hiramatsu ftrace_match_addr(struct ftrace_hash *hash, unsigned long ip, int remove)
3489647664eaSMasami Hiramatsu {
3490647664eaSMasami Hiramatsu 	struct ftrace_func_entry *entry;
3491647664eaSMasami Hiramatsu 
3492647664eaSMasami Hiramatsu 	if (!ftrace_location(ip))
3493647664eaSMasami Hiramatsu 		return -EINVAL;
3494647664eaSMasami Hiramatsu 
3495647664eaSMasami Hiramatsu 	if (remove) {
3496647664eaSMasami Hiramatsu 		entry = ftrace_lookup_ip(hash, ip);
3497647664eaSMasami Hiramatsu 		if (!entry)
3498647664eaSMasami Hiramatsu 			return -ENOENT;
3499647664eaSMasami Hiramatsu 		free_hash_entry(hash, entry);
3500647664eaSMasami Hiramatsu 		return 0;
3501647664eaSMasami Hiramatsu 	}
3502647664eaSMasami Hiramatsu 
3503647664eaSMasami Hiramatsu 	return add_hash_entry(hash, ip);
3504647664eaSMasami Hiramatsu }
3505647664eaSMasami Hiramatsu 
35061c80c432SSteven Rostedt (Red Hat) static void ftrace_ops_update_code(struct ftrace_ops *ops)
35071c80c432SSteven Rostedt (Red Hat) {
35081c80c432SSteven Rostedt (Red Hat) 	if (ops->flags & FTRACE_OPS_FL_ENABLED && ftrace_enabled)
35091c80c432SSteven Rostedt (Red Hat) 		ftrace_run_update_code(FTRACE_UPDATE_CALLS);
35101c80c432SSteven Rostedt (Red Hat) }
35111c80c432SSteven Rostedt (Red Hat) 
3512647664eaSMasami Hiramatsu static int
3513647664eaSMasami Hiramatsu ftrace_set_hash(struct ftrace_ops *ops, unsigned char *buf, int len,
3514647664eaSMasami Hiramatsu 		unsigned long ip, int remove, int reset, int enable)
351541c52c0dSSteven Rostedt {
351633dc9b12SSteven Rostedt 	struct ftrace_hash **orig_hash;
3517f45948e8SSteven Rostedt 	struct ftrace_hash *hash;
351833dc9b12SSteven Rostedt 	int ret;
3519f45948e8SSteven Rostedt 
3520936e074bSSteven Rostedt 	/* All global ops uses the global ops filters */
3521936e074bSSteven Rostedt 	if (ops->flags & FTRACE_OPS_FL_GLOBAL)
3522936e074bSSteven Rostedt 		ops = &global_ops;
3523936e074bSSteven Rostedt 
352441c52c0dSSteven Rostedt 	if (unlikely(ftrace_disabled))
352533dc9b12SSteven Rostedt 		return -ENODEV;
352641c52c0dSSteven Rostedt 
35273f2367baSMasami Hiramatsu 	mutex_lock(&ops->regex_lock);
35283f2367baSMasami Hiramatsu 
3529f45948e8SSteven Rostedt 	if (enable)
353033dc9b12SSteven Rostedt 		orig_hash = &ops->filter_hash;
3531f45948e8SSteven Rostedt 	else
353233dc9b12SSteven Rostedt 		orig_hash = &ops->notrace_hash;
353333dc9b12SSteven Rostedt 
353433dc9b12SSteven Rostedt 	hash = alloc_and_copy_ftrace_hash(FTRACE_HASH_DEFAULT_BITS, *orig_hash);
35353f2367baSMasami Hiramatsu 	if (!hash) {
35363f2367baSMasami Hiramatsu 		ret = -ENOMEM;
35373f2367baSMasami Hiramatsu 		goto out_regex_unlock;
35383f2367baSMasami Hiramatsu 	}
3539f45948e8SSteven Rostedt 
354041c52c0dSSteven Rostedt 	if (reset)
35411cf41dd7SSteven Rostedt 		ftrace_filter_reset(hash);
3542ac483c44SJiri Olsa 	if (buf && !ftrace_match_records(hash, buf, len)) {
3543ac483c44SJiri Olsa 		ret = -EINVAL;
3544ac483c44SJiri Olsa 		goto out_regex_unlock;
3545ac483c44SJiri Olsa 	}
3546647664eaSMasami Hiramatsu 	if (ip) {
3547647664eaSMasami Hiramatsu 		ret = ftrace_match_addr(hash, ip, remove);
3548647664eaSMasami Hiramatsu 		if (ret < 0)
3549647664eaSMasami Hiramatsu 			goto out_regex_unlock;
3550647664eaSMasami Hiramatsu 	}
355133dc9b12SSteven Rostedt 
355233dc9b12SSteven Rostedt 	mutex_lock(&ftrace_lock);
355341fb61c2SSteven Rostedt 	ret = ftrace_hash_move(ops, enable, orig_hash, hash);
35541c80c432SSteven Rostedt (Red Hat) 	if (!ret)
35551c80c432SSteven Rostedt (Red Hat) 		ftrace_ops_update_code(ops);
3556072126f4SSteven Rostedt 
355733dc9b12SSteven Rostedt 	mutex_unlock(&ftrace_lock);
355833dc9b12SSteven Rostedt 
3559ac483c44SJiri Olsa  out_regex_unlock:
3560f04f24fbSMasami Hiramatsu 	mutex_unlock(&ops->regex_lock);
356133dc9b12SSteven Rostedt 
356233dc9b12SSteven Rostedt 	free_ftrace_hash(hash);
356333dc9b12SSteven Rostedt 	return ret;
356441c52c0dSSteven Rostedt }
356541c52c0dSSteven Rostedt 
3566647664eaSMasami Hiramatsu static int
3567647664eaSMasami Hiramatsu ftrace_set_addr(struct ftrace_ops *ops, unsigned long ip, int remove,
3568647664eaSMasami Hiramatsu 		int reset, int enable)
3569647664eaSMasami Hiramatsu {
3570647664eaSMasami Hiramatsu 	return ftrace_set_hash(ops, 0, 0, ip, remove, reset, enable);
3571647664eaSMasami Hiramatsu }
3572647664eaSMasami Hiramatsu 
3573647664eaSMasami Hiramatsu /**
3574647664eaSMasami Hiramatsu  * ftrace_set_filter_ip - set a function to filter on in ftrace by address
3575647664eaSMasami Hiramatsu  * @ops - the ops to set the filter with
3576647664eaSMasami Hiramatsu  * @ip - the address to add to or remove from the filter.
3577647664eaSMasami Hiramatsu  * @remove - non zero to remove the ip from the filter
3578647664eaSMasami Hiramatsu  * @reset - non zero to reset all filters before applying this filter.
3579647664eaSMasami Hiramatsu  *
3580647664eaSMasami Hiramatsu  * Filters denote which functions should be enabled when tracing is enabled
3581647664eaSMasami Hiramatsu  * If @ip is NULL, it failes to update filter.
3582647664eaSMasami Hiramatsu  */
3583647664eaSMasami Hiramatsu int ftrace_set_filter_ip(struct ftrace_ops *ops, unsigned long ip,
3584647664eaSMasami Hiramatsu 			 int remove, int reset)
3585647664eaSMasami Hiramatsu {
3586f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
3587647664eaSMasami Hiramatsu 	return ftrace_set_addr(ops, ip, remove, reset, 1);
3588647664eaSMasami Hiramatsu }
3589647664eaSMasami Hiramatsu EXPORT_SYMBOL_GPL(ftrace_set_filter_ip);
3590647664eaSMasami Hiramatsu 
3591647664eaSMasami Hiramatsu static int
3592647664eaSMasami Hiramatsu ftrace_set_regex(struct ftrace_ops *ops, unsigned char *buf, int len,
3593647664eaSMasami Hiramatsu 		 int reset, int enable)
3594647664eaSMasami Hiramatsu {
3595647664eaSMasami Hiramatsu 	return ftrace_set_hash(ops, buf, len, 0, 0, reset, enable);
3596647664eaSMasami Hiramatsu }
3597647664eaSMasami Hiramatsu 
359877a2b37dSSteven Rostedt /**
359977a2b37dSSteven Rostedt  * ftrace_set_filter - set a function to filter on in ftrace
3600936e074bSSteven Rostedt  * @ops - the ops to set the filter with
360177a2b37dSSteven Rostedt  * @buf - the string that holds the function filter text.
360277a2b37dSSteven Rostedt  * @len - the length of the string.
360377a2b37dSSteven Rostedt  * @reset - non zero to reset all filters before applying this filter.
360477a2b37dSSteven Rostedt  *
360577a2b37dSSteven Rostedt  * Filters denote which functions should be enabled when tracing is enabled.
360677a2b37dSSteven Rostedt  * If @buf is NULL and reset is set, all functions will be enabled for tracing.
360777a2b37dSSteven Rostedt  */
3608ac483c44SJiri Olsa int ftrace_set_filter(struct ftrace_ops *ops, unsigned char *buf,
3609936e074bSSteven Rostedt 		       int len, int reset)
361077a2b37dSSteven Rostedt {
3611f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
3612ac483c44SJiri Olsa 	return ftrace_set_regex(ops, buf, len, reset, 1);
361341c52c0dSSteven Rostedt }
3614936e074bSSteven Rostedt EXPORT_SYMBOL_GPL(ftrace_set_filter);
36154eebcc81SSteven Rostedt 
361641c52c0dSSteven Rostedt /**
361741c52c0dSSteven Rostedt  * ftrace_set_notrace - set a function to not trace in ftrace
3618936e074bSSteven Rostedt  * @ops - the ops to set the notrace filter with
361941c52c0dSSteven Rostedt  * @buf - the string that holds the function notrace text.
362041c52c0dSSteven Rostedt  * @len - the length of the string.
362141c52c0dSSteven Rostedt  * @reset - non zero to reset all filters before applying this filter.
362241c52c0dSSteven Rostedt  *
362341c52c0dSSteven Rostedt  * Notrace Filters denote which functions should not be enabled when tracing
362441c52c0dSSteven Rostedt  * is enabled. If @buf is NULL and reset is set, all functions will be enabled
362541c52c0dSSteven Rostedt  * for tracing.
362641c52c0dSSteven Rostedt  */
3627ac483c44SJiri Olsa int ftrace_set_notrace(struct ftrace_ops *ops, unsigned char *buf,
3628936e074bSSteven Rostedt 			int len, int reset)
3629936e074bSSteven Rostedt {
3630f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
3631ac483c44SJiri Olsa 	return ftrace_set_regex(ops, buf, len, reset, 0);
3632936e074bSSteven Rostedt }
3633936e074bSSteven Rostedt EXPORT_SYMBOL_GPL(ftrace_set_notrace);
3634936e074bSSteven Rostedt /**
3635936e074bSSteven Rostedt  * ftrace_set_filter - set a function to filter on in ftrace
3636936e074bSSteven Rostedt  * @ops - the ops to set the filter with
3637936e074bSSteven Rostedt  * @buf - the string that holds the function filter text.
3638936e074bSSteven Rostedt  * @len - the length of the string.
3639936e074bSSteven Rostedt  * @reset - non zero to reset all filters before applying this filter.
3640936e074bSSteven Rostedt  *
3641936e074bSSteven Rostedt  * Filters denote which functions should be enabled when tracing is enabled.
3642936e074bSSteven Rostedt  * If @buf is NULL and reset is set, all functions will be enabled for tracing.
3643936e074bSSteven Rostedt  */
3644936e074bSSteven Rostedt void ftrace_set_global_filter(unsigned char *buf, int len, int reset)
3645936e074bSSteven Rostedt {
3646936e074bSSteven Rostedt 	ftrace_set_regex(&global_ops, buf, len, reset, 1);
3647936e074bSSteven Rostedt }
3648936e074bSSteven Rostedt EXPORT_SYMBOL_GPL(ftrace_set_global_filter);
3649936e074bSSteven Rostedt 
3650936e074bSSteven Rostedt /**
3651936e074bSSteven Rostedt  * ftrace_set_notrace - set a function to not trace in ftrace
3652936e074bSSteven Rostedt  * @ops - the ops to set the notrace filter with
3653936e074bSSteven Rostedt  * @buf - the string that holds the function notrace text.
3654936e074bSSteven Rostedt  * @len - the length of the string.
3655936e074bSSteven Rostedt  * @reset - non zero to reset all filters before applying this filter.
3656936e074bSSteven Rostedt  *
3657936e074bSSteven Rostedt  * Notrace Filters denote which functions should not be enabled when tracing
3658936e074bSSteven Rostedt  * is enabled. If @buf is NULL and reset is set, all functions will be enabled
3659936e074bSSteven Rostedt  * for tracing.
3660936e074bSSteven Rostedt  */
3661936e074bSSteven Rostedt void ftrace_set_global_notrace(unsigned char *buf, int len, int reset)
366241c52c0dSSteven Rostedt {
3663f45948e8SSteven Rostedt 	ftrace_set_regex(&global_ops, buf, len, reset, 0);
366477a2b37dSSteven Rostedt }
3665936e074bSSteven Rostedt EXPORT_SYMBOL_GPL(ftrace_set_global_notrace);
366677a2b37dSSteven Rostedt 
36672af15d6aSSteven Rostedt /*
36682af15d6aSSteven Rostedt  * command line interface to allow users to set filters on boot up.
36692af15d6aSSteven Rostedt  */
36702af15d6aSSteven Rostedt #define FTRACE_FILTER_SIZE		COMMAND_LINE_SIZE
36712af15d6aSSteven Rostedt static char ftrace_notrace_buf[FTRACE_FILTER_SIZE] __initdata;
36722af15d6aSSteven Rostedt static char ftrace_filter_buf[FTRACE_FILTER_SIZE] __initdata;
36732af15d6aSSteven Rostedt 
3674f1ed7c74SSteven Rostedt (Red Hat) /* Used by function selftest to not test if filter is set */
3675f1ed7c74SSteven Rostedt (Red Hat) bool ftrace_filter_param __initdata;
3676f1ed7c74SSteven Rostedt (Red Hat) 
36772af15d6aSSteven Rostedt static int __init set_ftrace_notrace(char *str)
36782af15d6aSSteven Rostedt {
3679f1ed7c74SSteven Rostedt (Red Hat) 	ftrace_filter_param = true;
368075761cc1SChen Gang 	strlcpy(ftrace_notrace_buf, str, FTRACE_FILTER_SIZE);
36812af15d6aSSteven Rostedt 	return 1;
36822af15d6aSSteven Rostedt }
36832af15d6aSSteven Rostedt __setup("ftrace_notrace=", set_ftrace_notrace);
36842af15d6aSSteven Rostedt 
36852af15d6aSSteven Rostedt static int __init set_ftrace_filter(char *str)
36862af15d6aSSteven Rostedt {
3687f1ed7c74SSteven Rostedt (Red Hat) 	ftrace_filter_param = true;
368875761cc1SChen Gang 	strlcpy(ftrace_filter_buf, str, FTRACE_FILTER_SIZE);
36892af15d6aSSteven Rostedt 	return 1;
36902af15d6aSSteven Rostedt }
36912af15d6aSSteven Rostedt __setup("ftrace_filter=", set_ftrace_filter);
36922af15d6aSSteven Rostedt 
3693369bc18fSStefan Assmann #ifdef CONFIG_FUNCTION_GRAPH_TRACER
3694f6060f46SLai Jiangshan static char ftrace_graph_buf[FTRACE_FILTER_SIZE] __initdata;
3695faf982a6SNamhyung Kim static int ftrace_set_func(unsigned long *array, int *idx, int size, char *buffer);
3696801c29fdSSteven Rostedt 
3697369bc18fSStefan Assmann static int __init set_graph_function(char *str)
3698369bc18fSStefan Assmann {
369906f43d66SFrederic Weisbecker 	strlcpy(ftrace_graph_buf, str, FTRACE_FILTER_SIZE);
3700369bc18fSStefan Assmann 	return 1;
3701369bc18fSStefan Assmann }
3702369bc18fSStefan Assmann __setup("ftrace_graph_filter=", set_graph_function);
3703369bc18fSStefan Assmann 
3704369bc18fSStefan Assmann static void __init set_ftrace_early_graph(char *buf)
3705369bc18fSStefan Assmann {
3706369bc18fSStefan Assmann 	int ret;
3707369bc18fSStefan Assmann 	char *func;
3708369bc18fSStefan Assmann 
3709369bc18fSStefan Assmann 	while (buf) {
3710369bc18fSStefan Assmann 		func = strsep(&buf, ",");
3711369bc18fSStefan Assmann 		/* we allow only one expression at a time */
3712369bc18fSStefan Assmann 		ret = ftrace_set_func(ftrace_graph_funcs, &ftrace_graph_count,
3713faf982a6SNamhyung Kim 				      FTRACE_GRAPH_MAX_FUNCS, func);
3714369bc18fSStefan Assmann 		if (ret)
3715369bc18fSStefan Assmann 			printk(KERN_DEBUG "ftrace: function %s not "
3716369bc18fSStefan Assmann 					  "traceable\n", func);
3717369bc18fSStefan Assmann 	}
3718369bc18fSStefan Assmann }
3719369bc18fSStefan Assmann #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
3720369bc18fSStefan Assmann 
37212a85a37fSSteven Rostedt void __init
37222a85a37fSSteven Rostedt ftrace_set_early_filter(struct ftrace_ops *ops, char *buf, int enable)
37232af15d6aSSteven Rostedt {
37242af15d6aSSteven Rostedt 	char *func;
37252af15d6aSSteven Rostedt 
3726f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
3727f04f24fbSMasami Hiramatsu 
37282af15d6aSSteven Rostedt 	while (buf) {
37292af15d6aSSteven Rostedt 		func = strsep(&buf, ",");
3730f45948e8SSteven Rostedt 		ftrace_set_regex(ops, func, strlen(func), 0, enable);
37312af15d6aSSteven Rostedt 	}
37322af15d6aSSteven Rostedt }
37332af15d6aSSteven Rostedt 
37342af15d6aSSteven Rostedt static void __init set_ftrace_early_filters(void)
37352af15d6aSSteven Rostedt {
37362af15d6aSSteven Rostedt 	if (ftrace_filter_buf[0])
37372a85a37fSSteven Rostedt 		ftrace_set_early_filter(&global_ops, ftrace_filter_buf, 1);
37382af15d6aSSteven Rostedt 	if (ftrace_notrace_buf[0])
37392a85a37fSSteven Rostedt 		ftrace_set_early_filter(&global_ops, ftrace_notrace_buf, 0);
3740369bc18fSStefan Assmann #ifdef CONFIG_FUNCTION_GRAPH_TRACER
3741369bc18fSStefan Assmann 	if (ftrace_graph_buf[0])
3742369bc18fSStefan Assmann 		set_ftrace_early_graph(ftrace_graph_buf);
3743369bc18fSStefan Assmann #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
37442af15d6aSSteven Rostedt }
37452af15d6aSSteven Rostedt 
3746fc13cb0cSSteven Rostedt int ftrace_regex_release(struct inode *inode, struct file *file)
37475072c59fSSteven Rostedt {
37485072c59fSSteven Rostedt 	struct seq_file *m = (struct seq_file *)file->private_data;
37495072c59fSSteven Rostedt 	struct ftrace_iterator *iter;
375033dc9b12SSteven Rostedt 	struct ftrace_hash **orig_hash;
3751689fd8b6S[email protected] 	struct trace_parser *parser;
3752ed926f9bSSteven Rostedt 	int filter_hash;
375333dc9b12SSteven Rostedt 	int ret;
37545072c59fSSteven Rostedt 
37555072c59fSSteven Rostedt 	if (file->f_mode & FMODE_READ) {
37565072c59fSSteven Rostedt 		iter = m->private;
37575072c59fSSteven Rostedt 		seq_release(inode, file);
37585072c59fSSteven Rostedt 	} else
37595072c59fSSteven Rostedt 		iter = file->private_data;
37605072c59fSSteven Rostedt 
3761689fd8b6S[email protected] 	parser = &iter->parser;
3762689fd8b6S[email protected] 	if (trace_parser_loaded(parser)) {
3763689fd8b6S[email protected] 		parser->buffer[parser->idx] = 0;
37641cf41dd7SSteven Rostedt 		ftrace_match_records(iter->hash, parser->buffer, parser->idx);
37655072c59fSSteven Rostedt 	}
37665072c59fSSteven Rostedt 
3767058e297dSSteven Rostedt 	trace_parser_put(parser);
3768058e297dSSteven Rostedt 
37693f2367baSMasami Hiramatsu 	mutex_lock(&iter->ops->regex_lock);
37703f2367baSMasami Hiramatsu 
3771058e297dSSteven Rostedt 	if (file->f_mode & FMODE_WRITE) {
3772ed926f9bSSteven Rostedt 		filter_hash = !!(iter->flags & FTRACE_ITER_FILTER);
3773ed926f9bSSteven Rostedt 
3774ed926f9bSSteven Rostedt 		if (filter_hash)
377533dc9b12SSteven Rostedt 			orig_hash = &iter->ops->filter_hash;
3776ed926f9bSSteven Rostedt 		else
3777ed926f9bSSteven Rostedt 			orig_hash = &iter->ops->notrace_hash;
377833dc9b12SSteven Rostedt 
3779e6ea44e9SSteven Rostedt 		mutex_lock(&ftrace_lock);
378041fb61c2SSteven Rostedt 		ret = ftrace_hash_move(iter->ops, filter_hash,
378141fb61c2SSteven Rostedt 				       orig_hash, iter->hash);
37821c80c432SSteven Rostedt (Red Hat) 		if (!ret)
37831c80c432SSteven Rostedt (Red Hat) 			ftrace_ops_update_code(iter->ops);
378441fb61c2SSteven Rostedt 
3785e6ea44e9SSteven Rostedt 		mutex_unlock(&ftrace_lock);
3786058e297dSSteven Rostedt 	}
37873f2367baSMasami Hiramatsu 
37883f2367baSMasami Hiramatsu 	mutex_unlock(&iter->ops->regex_lock);
378933dc9b12SSteven Rostedt 	free_ftrace_hash(iter->hash);
379033dc9b12SSteven Rostedt 	kfree(iter);
3791689fd8b6S[email protected] 
37925072c59fSSteven Rostedt 	return 0;
37935072c59fSSteven Rostedt }
37945072c59fSSteven Rostedt 
37955e2336a0SSteven Rostedt static const struct file_operations ftrace_avail_fops = {
37965072c59fSSteven Rostedt 	.open = ftrace_avail_open,
37975072c59fSSteven Rostedt 	.read = seq_read,
37985072c59fSSteven Rostedt 	.llseek = seq_lseek,
37993be04b47SLi Zefan 	.release = seq_release_private,
38005072c59fSSteven Rostedt };
38015072c59fSSteven Rostedt 
3802647bcd03SSteven Rostedt static const struct file_operations ftrace_enabled_fops = {
3803647bcd03SSteven Rostedt 	.open = ftrace_enabled_open,
3804647bcd03SSteven Rostedt 	.read = seq_read,
3805647bcd03SSteven Rostedt 	.llseek = seq_lseek,
3806647bcd03SSteven Rostedt 	.release = seq_release_private,
3807647bcd03SSteven Rostedt };
3808647bcd03SSteven Rostedt 
38095e2336a0SSteven Rostedt static const struct file_operations ftrace_filter_fops = {
38105072c59fSSteven Rostedt 	.open = ftrace_filter_open,
3811850a80cfSLai Jiangshan 	.read = seq_read,
38125072c59fSSteven Rostedt 	.write = ftrace_filter_write,
3813098c879eSSteven Rostedt (Red Hat) 	.llseek = tracing_lseek,
38141cf41dd7SSteven Rostedt 	.release = ftrace_regex_release,
38155072c59fSSteven Rostedt };
38165072c59fSSteven Rostedt 
38175e2336a0SSteven Rostedt static const struct file_operations ftrace_notrace_fops = {
381841c52c0dSSteven Rostedt 	.open = ftrace_notrace_open,
3819850a80cfSLai Jiangshan 	.read = seq_read,
382041c52c0dSSteven Rostedt 	.write = ftrace_notrace_write,
3821098c879eSSteven Rostedt (Red Hat) 	.llseek = tracing_lseek,
38221cf41dd7SSteven Rostedt 	.release = ftrace_regex_release,
382341c52c0dSSteven Rostedt };
382441c52c0dSSteven Rostedt 
3825ea4e2bc4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
3826ea4e2bc4SSteven Rostedt 
3827ea4e2bc4SSteven Rostedt static DEFINE_MUTEX(graph_lock);
3828ea4e2bc4SSteven Rostedt 
3829ea4e2bc4SSteven Rostedt int ftrace_graph_count;
383029ad23b0SNamhyung Kim int ftrace_graph_notrace_count;
3831ea4e2bc4SSteven Rostedt unsigned long ftrace_graph_funcs[FTRACE_GRAPH_MAX_FUNCS] __read_mostly;
383229ad23b0SNamhyung Kim unsigned long ftrace_graph_notrace_funcs[FTRACE_GRAPH_MAX_FUNCS] __read_mostly;
3833ea4e2bc4SSteven Rostedt 
3834faf982a6SNamhyung Kim struct ftrace_graph_data {
3835faf982a6SNamhyung Kim 	unsigned long *table;
3836faf982a6SNamhyung Kim 	size_t size;
3837faf982a6SNamhyung Kim 	int *count;
3838faf982a6SNamhyung Kim 	const struct seq_operations *seq_ops;
3839faf982a6SNamhyung Kim };
3840faf982a6SNamhyung Kim 
3841ea4e2bc4SSteven Rostedt static void *
384285951842SLi Zefan __g_next(struct seq_file *m, loff_t *pos)
3843ea4e2bc4SSteven Rostedt {
3844faf982a6SNamhyung Kim 	struct ftrace_graph_data *fgd = m->private;
3845faf982a6SNamhyung Kim 
3846faf982a6SNamhyung Kim 	if (*pos >= *fgd->count)
3847ea4e2bc4SSteven Rostedt 		return NULL;
3848faf982a6SNamhyung Kim 	return &fgd->table[*pos];
384985951842SLi Zefan }
3850ea4e2bc4SSteven Rostedt 
385185951842SLi Zefan static void *
385285951842SLi Zefan g_next(struct seq_file *m, void *v, loff_t *pos)
385385951842SLi Zefan {
385485951842SLi Zefan 	(*pos)++;
385585951842SLi Zefan 	return __g_next(m, pos);
3856ea4e2bc4SSteven Rostedt }
3857ea4e2bc4SSteven Rostedt 
3858ea4e2bc4SSteven Rostedt static void *g_start(struct seq_file *m, loff_t *pos)
3859ea4e2bc4SSteven Rostedt {
3860faf982a6SNamhyung Kim 	struct ftrace_graph_data *fgd = m->private;
3861faf982a6SNamhyung Kim 
3862ea4e2bc4SSteven Rostedt 	mutex_lock(&graph_lock);
3863ea4e2bc4SSteven Rostedt 
3864f9349a8fSFrederic Weisbecker 	/* Nothing, tell g_show to print all functions are enabled */
3865faf982a6SNamhyung Kim 	if (!*fgd->count && !*pos)
3866f9349a8fSFrederic Weisbecker 		return (void *)1;
3867f9349a8fSFrederic Weisbecker 
386885951842SLi Zefan 	return __g_next(m, pos);
3869ea4e2bc4SSteven Rostedt }
3870ea4e2bc4SSteven Rostedt 
3871ea4e2bc4SSteven Rostedt static void g_stop(struct seq_file *m, void *p)
3872ea4e2bc4SSteven Rostedt {
3873ea4e2bc4SSteven Rostedt 	mutex_unlock(&graph_lock);
3874ea4e2bc4SSteven Rostedt }
3875ea4e2bc4SSteven Rostedt 
3876ea4e2bc4SSteven Rostedt static int g_show(struct seq_file *m, void *v)
3877ea4e2bc4SSteven Rostedt {
3878ea4e2bc4SSteven Rostedt 	unsigned long *ptr = v;
3879ea4e2bc4SSteven Rostedt 
3880ea4e2bc4SSteven Rostedt 	if (!ptr)
3881ea4e2bc4SSteven Rostedt 		return 0;
3882ea4e2bc4SSteven Rostedt 
3883f9349a8fSFrederic Weisbecker 	if (ptr == (unsigned long *)1) {
3884f9349a8fSFrederic Weisbecker 		seq_printf(m, "#### all functions enabled ####\n");
3885f9349a8fSFrederic Weisbecker 		return 0;
3886f9349a8fSFrederic Weisbecker 	}
3887f9349a8fSFrederic Weisbecker 
3888b375a11aSSteven Rostedt 	seq_printf(m, "%ps\n", (void *)*ptr);
3889ea4e2bc4SSteven Rostedt 
3890ea4e2bc4SSteven Rostedt 	return 0;
3891ea4e2bc4SSteven Rostedt }
3892ea4e2bc4SSteven Rostedt 
389388e9d34cSJames Morris static const struct seq_operations ftrace_graph_seq_ops = {
3894ea4e2bc4SSteven Rostedt 	.start = g_start,
3895ea4e2bc4SSteven Rostedt 	.next = g_next,
3896ea4e2bc4SSteven Rostedt 	.stop = g_stop,
3897ea4e2bc4SSteven Rostedt 	.show = g_show,
3898ea4e2bc4SSteven Rostedt };
3899ea4e2bc4SSteven Rostedt 
3900ea4e2bc4SSteven Rostedt static int
3901faf982a6SNamhyung Kim __ftrace_graph_open(struct inode *inode, struct file *file,
3902faf982a6SNamhyung Kim 		    struct ftrace_graph_data *fgd)
3903ea4e2bc4SSteven Rostedt {
3904ea4e2bc4SSteven Rostedt 	int ret = 0;
3905ea4e2bc4SSteven Rostedt 
3906ea4e2bc4SSteven Rostedt 	mutex_lock(&graph_lock);
3907ea4e2bc4SSteven Rostedt 	if ((file->f_mode & FMODE_WRITE) &&
39088650ae32SSteven Rostedt 	    (file->f_flags & O_TRUNC)) {
3909faf982a6SNamhyung Kim 		*fgd->count = 0;
3910faf982a6SNamhyung Kim 		memset(fgd->table, 0, fgd->size * sizeof(*fgd->table));
3911ea4e2bc4SSteven Rostedt 	}
3912ea4e2bc4SSteven Rostedt 	mutex_unlock(&graph_lock);
3913ea4e2bc4SSteven Rostedt 
3914faf982a6SNamhyung Kim 	if (file->f_mode & FMODE_READ) {
3915faf982a6SNamhyung Kim 		ret = seq_open(file, fgd->seq_ops);
3916faf982a6SNamhyung Kim 		if (!ret) {
3917faf982a6SNamhyung Kim 			struct seq_file *m = file->private_data;
3918faf982a6SNamhyung Kim 			m->private = fgd;
3919faf982a6SNamhyung Kim 		}
3920faf982a6SNamhyung Kim 	} else
3921faf982a6SNamhyung Kim 		file->private_data = fgd;
3922a4ec5e0cSLi Zefan 
3923ea4e2bc4SSteven Rostedt 	return ret;
3924ea4e2bc4SSteven Rostedt }
3925ea4e2bc4SSteven Rostedt 
3926ea4e2bc4SSteven Rostedt static int
3927faf982a6SNamhyung Kim ftrace_graph_open(struct inode *inode, struct file *file)
3928faf982a6SNamhyung Kim {
3929faf982a6SNamhyung Kim 	struct ftrace_graph_data *fgd;
3930faf982a6SNamhyung Kim 
3931faf982a6SNamhyung Kim 	if (unlikely(ftrace_disabled))
3932faf982a6SNamhyung Kim 		return -ENODEV;
3933faf982a6SNamhyung Kim 
3934faf982a6SNamhyung Kim 	fgd = kmalloc(sizeof(*fgd), GFP_KERNEL);
3935faf982a6SNamhyung Kim 	if (fgd == NULL)
3936faf982a6SNamhyung Kim 		return -ENOMEM;
3937faf982a6SNamhyung Kim 
3938faf982a6SNamhyung Kim 	fgd->table = ftrace_graph_funcs;
3939faf982a6SNamhyung Kim 	fgd->size = FTRACE_GRAPH_MAX_FUNCS;
3940faf982a6SNamhyung Kim 	fgd->count = &ftrace_graph_count;
3941faf982a6SNamhyung Kim 	fgd->seq_ops = &ftrace_graph_seq_ops;
3942faf982a6SNamhyung Kim 
3943faf982a6SNamhyung Kim 	return __ftrace_graph_open(inode, file, fgd);
3944faf982a6SNamhyung Kim }
3945faf982a6SNamhyung Kim 
3946faf982a6SNamhyung Kim static int
394729ad23b0SNamhyung Kim ftrace_graph_notrace_open(struct inode *inode, struct file *file)
394829ad23b0SNamhyung Kim {
394929ad23b0SNamhyung Kim 	struct ftrace_graph_data *fgd;
395029ad23b0SNamhyung Kim 
395129ad23b0SNamhyung Kim 	if (unlikely(ftrace_disabled))
395229ad23b0SNamhyung Kim 		return -ENODEV;
395329ad23b0SNamhyung Kim 
395429ad23b0SNamhyung Kim 	fgd = kmalloc(sizeof(*fgd), GFP_KERNEL);
395529ad23b0SNamhyung Kim 	if (fgd == NULL)
395629ad23b0SNamhyung Kim 		return -ENOMEM;
395729ad23b0SNamhyung Kim 
395829ad23b0SNamhyung Kim 	fgd->table = ftrace_graph_notrace_funcs;
395929ad23b0SNamhyung Kim 	fgd->size = FTRACE_GRAPH_MAX_FUNCS;
396029ad23b0SNamhyung Kim 	fgd->count = &ftrace_graph_notrace_count;
396129ad23b0SNamhyung Kim 	fgd->seq_ops = &ftrace_graph_seq_ops;
396229ad23b0SNamhyung Kim 
396329ad23b0SNamhyung Kim 	return __ftrace_graph_open(inode, file, fgd);
396429ad23b0SNamhyung Kim }
396529ad23b0SNamhyung Kim 
396629ad23b0SNamhyung Kim static int
396787827111SLi Zefan ftrace_graph_release(struct inode *inode, struct file *file)
396887827111SLi Zefan {
3969faf982a6SNamhyung Kim 	if (file->f_mode & FMODE_READ) {
3970faf982a6SNamhyung Kim 		struct seq_file *m = file->private_data;
3971faf982a6SNamhyung Kim 
3972faf982a6SNamhyung Kim 		kfree(m->private);
397387827111SLi Zefan 		seq_release(inode, file);
3974faf982a6SNamhyung Kim 	} else {
3975faf982a6SNamhyung Kim 		kfree(file->private_data);
3976faf982a6SNamhyung Kim 	}
3977faf982a6SNamhyung Kim 
397887827111SLi Zefan 	return 0;
397987827111SLi Zefan }
398087827111SLi Zefan 
398187827111SLi Zefan static int
3982faf982a6SNamhyung Kim ftrace_set_func(unsigned long *array, int *idx, int size, char *buffer)
3983ea4e2bc4SSteven Rostedt {
3984ea4e2bc4SSteven Rostedt 	struct dyn_ftrace *rec;
3985ea4e2bc4SSteven Rostedt 	struct ftrace_page *pg;
3986f9349a8fSFrederic Weisbecker 	int search_len;
3987c7c6b1feSLi Zefan 	int fail = 1;
3988f9349a8fSFrederic Weisbecker 	int type, not;
3989f9349a8fSFrederic Weisbecker 	char *search;
3990f9349a8fSFrederic Weisbecker 	bool exists;
3991f9349a8fSFrederic Weisbecker 	int i;
3992ea4e2bc4SSteven Rostedt 
3993f9349a8fSFrederic Weisbecker 	/* decode regex */
39943f6fe06dSFrederic Weisbecker 	type = filter_parse_regex(buffer, strlen(buffer), &search, &not);
3995faf982a6SNamhyung Kim 	if (!not && *idx >= size)
3996c7c6b1feSLi Zefan 		return -EBUSY;
3997f9349a8fSFrederic Weisbecker 
3998f9349a8fSFrederic Weisbecker 	search_len = strlen(search);
3999f9349a8fSFrederic Weisbecker 
400052baf119SSteven Rostedt 	mutex_lock(&ftrace_lock);
400145a4a237SSteven Rostedt 
400245a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled)) {
400345a4a237SSteven Rostedt 		mutex_unlock(&ftrace_lock);
400445a4a237SSteven Rostedt 		return -ENODEV;
400545a4a237SSteven Rostedt 	}
400645a4a237SSteven Rostedt 
4007265c831cSSteven Rostedt 	do_for_each_ftrace_rec(pg, rec) {
4008ea4e2bc4SSteven Rostedt 
4009b9df92d2SSteven Rostedt 		if (ftrace_match_record(rec, NULL, search, search_len, type)) {
4010c7c6b1feSLi Zefan 			/* if it is in the array */
4011f9349a8fSFrederic Weisbecker 			exists = false;
4012c7c6b1feSLi Zefan 			for (i = 0; i < *idx; i++) {
4013f9349a8fSFrederic Weisbecker 				if (array[i] == rec->ip) {
4014f9349a8fSFrederic Weisbecker 					exists = true;
4015faec2ec5SLiming Wang 					break;
4016faec2ec5SLiming Wang 				}
4017c7c6b1feSLi Zefan 			}
4018c7c6b1feSLi Zefan 
4019c7c6b1feSLi Zefan 			if (!not) {
4020c7c6b1feSLi Zefan 				fail = 0;
4021c7c6b1feSLi Zefan 				if (!exists) {
4022f9349a8fSFrederic Weisbecker 					array[(*idx)++] = rec->ip;
4023faf982a6SNamhyung Kim 					if (*idx >= size)
4024c7c6b1feSLi Zefan 						goto out;
4025c7c6b1feSLi Zefan 				}
4026c7c6b1feSLi Zefan 			} else {
4027c7c6b1feSLi Zefan 				if (exists) {
4028c7c6b1feSLi Zefan 					array[i] = array[--(*idx)];
4029c7c6b1feSLi Zefan 					array[*idx] = 0;
4030c7c6b1feSLi Zefan 					fail = 0;
4031c7c6b1feSLi Zefan 				}
4032c7c6b1feSLi Zefan 			}
4033f9349a8fSFrederic Weisbecker 		}
4034265c831cSSteven Rostedt 	} while_for_each_ftrace_rec();
4035c7c6b1feSLi Zefan out:
403652baf119SSteven Rostedt 	mutex_unlock(&ftrace_lock);
4037ea4e2bc4SSteven Rostedt 
4038c7c6b1feSLi Zefan 	if (fail)
4039c7c6b1feSLi Zefan 		return -EINVAL;
4040c7c6b1feSLi Zefan 
4041c7c6b1feSLi Zefan 	return 0;
4042ea4e2bc4SSteven Rostedt }
4043ea4e2bc4SSteven Rostedt 
4044ea4e2bc4SSteven Rostedt static ssize_t
4045ea4e2bc4SSteven Rostedt ftrace_graph_write(struct file *file, const char __user *ubuf,
4046ea4e2bc4SSteven Rostedt 		   size_t cnt, loff_t *ppos)
4047ea4e2bc4SSteven Rostedt {
4048689fd8b6S[email protected] 	struct trace_parser parser;
40496a10108bSNamhyung Kim 	ssize_t read, ret = 0;
4050faf982a6SNamhyung Kim 	struct ftrace_graph_data *fgd = file->private_data;
4051ea4e2bc4SSteven Rostedt 
4052c7c6b1feSLi Zefan 	if (!cnt)
4053ea4e2bc4SSteven Rostedt 		return 0;
4054ea4e2bc4SSteven Rostedt 
40556a10108bSNamhyung Kim 	if (trace_parser_get_init(&parser, FTRACE_BUFF_MAX))
40566a10108bSNamhyung Kim 		return -ENOMEM;
4057ea4e2bc4SSteven Rostedt 
4058689fd8b6S[email protected] 	read = trace_get_user(&parser, ubuf, cnt, ppos);
4059689fd8b6S[email protected] 
40604ba7978eSLi Zefan 	if (read >= 0 && trace_parser_loaded((&parser))) {
4061689fd8b6S[email protected] 		parser.buffer[parser.idx] = 0;
4062ea4e2bc4SSteven Rostedt 
40636a10108bSNamhyung Kim 		mutex_lock(&graph_lock);
40646a10108bSNamhyung Kim 
4065f9349a8fSFrederic Weisbecker 		/* we allow only one expression at a time */
4066faf982a6SNamhyung Kim 		ret = ftrace_set_func(fgd->table, fgd->count, fgd->size,
4067689fd8b6S[email protected] 				      parser.buffer);
40686a10108bSNamhyung Kim 
40696a10108bSNamhyung Kim 		mutex_unlock(&graph_lock);
4070689fd8b6S[email protected] 	}
4071ea4e2bc4SSteven Rostedt 
40726a10108bSNamhyung Kim 	if (!ret)
4073ea4e2bc4SSteven Rostedt 		ret = read;
40741eb90f13SLi Zefan 
4075689fd8b6S[email protected] 	trace_parser_put(&parser);
4076ea4e2bc4SSteven Rostedt 
4077ea4e2bc4SSteven Rostedt 	return ret;
4078ea4e2bc4SSteven Rostedt }
4079ea4e2bc4SSteven Rostedt 
4080ea4e2bc4SSteven Rostedt static const struct file_operations ftrace_graph_fops = {
4081ea4e2bc4SSteven Rostedt 	.open		= ftrace_graph_open,
4082850a80cfSLai Jiangshan 	.read		= seq_read,
4083ea4e2bc4SSteven Rostedt 	.write		= ftrace_graph_write,
4084098c879eSSteven Rostedt (Red Hat) 	.llseek		= tracing_lseek,
408587827111SLi Zefan 	.release	= ftrace_graph_release,
4086ea4e2bc4SSteven Rostedt };
408729ad23b0SNamhyung Kim 
408829ad23b0SNamhyung Kim static const struct file_operations ftrace_graph_notrace_fops = {
408929ad23b0SNamhyung Kim 	.open		= ftrace_graph_notrace_open,
409029ad23b0SNamhyung Kim 	.read		= seq_read,
409129ad23b0SNamhyung Kim 	.write		= ftrace_graph_write,
4092098c879eSSteven Rostedt (Red Hat) 	.llseek		= tracing_lseek,
409329ad23b0SNamhyung Kim 	.release	= ftrace_graph_release,
409429ad23b0SNamhyung Kim };
4095ea4e2bc4SSteven Rostedt #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
4096ea4e2bc4SSteven Rostedt 
4097591dffdaSSteven Rostedt (Red Hat) void ftrace_create_filter_files(struct ftrace_ops *ops,
4098591dffdaSSteven Rostedt (Red Hat) 				struct dentry *parent)
4099591dffdaSSteven Rostedt (Red Hat) {
4100591dffdaSSteven Rostedt (Red Hat) 
4101591dffdaSSteven Rostedt (Red Hat) 	trace_create_file("set_ftrace_filter", 0644, parent,
4102591dffdaSSteven Rostedt (Red Hat) 			  ops, &ftrace_filter_fops);
4103591dffdaSSteven Rostedt (Red Hat) 
4104591dffdaSSteven Rostedt (Red Hat) 	trace_create_file("set_ftrace_notrace", 0644, parent,
4105591dffdaSSteven Rostedt (Red Hat) 			  ops, &ftrace_notrace_fops);
4106591dffdaSSteven Rostedt (Red Hat) }
4107591dffdaSSteven Rostedt (Red Hat) 
4108591dffdaSSteven Rostedt (Red Hat) /*
4109591dffdaSSteven Rostedt (Red Hat)  * The name "destroy_filter_files" is really a misnomer. Although
4110591dffdaSSteven Rostedt (Red Hat)  * in the future, it may actualy delete the files, but this is
4111591dffdaSSteven Rostedt (Red Hat)  * really intended to make sure the ops passed in are disabled
4112591dffdaSSteven Rostedt (Red Hat)  * and that when this function returns, the caller is free to
4113591dffdaSSteven Rostedt (Red Hat)  * free the ops.
4114591dffdaSSteven Rostedt (Red Hat)  *
4115591dffdaSSteven Rostedt (Red Hat)  * The "destroy" name is only to match the "create" name that this
4116591dffdaSSteven Rostedt (Red Hat)  * should be paired with.
4117591dffdaSSteven Rostedt (Red Hat)  */
4118591dffdaSSteven Rostedt (Red Hat) void ftrace_destroy_filter_files(struct ftrace_ops *ops)
4119591dffdaSSteven Rostedt (Red Hat) {
4120591dffdaSSteven Rostedt (Red Hat) 	mutex_lock(&ftrace_lock);
4121591dffdaSSteven Rostedt (Red Hat) 	if (ops->flags & FTRACE_OPS_FL_ENABLED)
4122591dffdaSSteven Rostedt (Red Hat) 		ftrace_shutdown(ops, 0);
4123591dffdaSSteven Rostedt (Red Hat) 	ops->flags |= FTRACE_OPS_FL_DELETED;
4124591dffdaSSteven Rostedt (Red Hat) 	mutex_unlock(&ftrace_lock);
4125591dffdaSSteven Rostedt (Red Hat) }
4126591dffdaSSteven Rostedt (Red Hat) 
4127df4fc315SSteven Rostedt static __init int ftrace_init_dyn_debugfs(struct dentry *d_tracer)
41285072c59fSSteven Rostedt {
41295072c59fSSteven Rostedt 
41305452af66SFrederic Weisbecker 	trace_create_file("available_filter_functions", 0444,
41315072c59fSSteven Rostedt 			d_tracer, NULL, &ftrace_avail_fops);
41325072c59fSSteven Rostedt 
4133647bcd03SSteven Rostedt 	trace_create_file("enabled_functions", 0444,
4134647bcd03SSteven Rostedt 			d_tracer, NULL, &ftrace_enabled_fops);
4135647bcd03SSteven Rostedt 
4136591dffdaSSteven Rostedt (Red Hat) 	ftrace_create_filter_files(&global_ops, d_tracer);
4137ad90c0e3SSteven Rostedt 
4138ea4e2bc4SSteven Rostedt #ifdef CONFIG_FUNCTION_GRAPH_TRACER
41395452af66SFrederic Weisbecker 	trace_create_file("set_graph_function", 0444, d_tracer,
4140ea4e2bc4SSteven Rostedt 				    NULL,
4141ea4e2bc4SSteven Rostedt 				    &ftrace_graph_fops);
414229ad23b0SNamhyung Kim 	trace_create_file("set_graph_notrace", 0444, d_tracer,
414329ad23b0SNamhyung Kim 				    NULL,
414429ad23b0SNamhyung Kim 				    &ftrace_graph_notrace_fops);
4145ea4e2bc4SSteven Rostedt #endif /* CONFIG_FUNCTION_GRAPH_TRACER */
4146ea4e2bc4SSteven Rostedt 
41475072c59fSSteven Rostedt 	return 0;
41485072c59fSSteven Rostedt }
41495072c59fSSteven Rostedt 
41509fd49328SSteven Rostedt static int ftrace_cmp_ips(const void *a, const void *b)
415168950619SSteven Rostedt {
41529fd49328SSteven Rostedt 	const unsigned long *ipa = a;
41539fd49328SSteven Rostedt 	const unsigned long *ipb = b;
415468950619SSteven Rostedt 
41559fd49328SSteven Rostedt 	if (*ipa > *ipb)
41569fd49328SSteven Rostedt 		return 1;
41579fd49328SSteven Rostedt 	if (*ipa < *ipb)
41589fd49328SSteven Rostedt 		return -1;
41599fd49328SSteven Rostedt 	return 0;
41609fd49328SSteven Rostedt }
41619fd49328SSteven Rostedt 
41629fd49328SSteven Rostedt static void ftrace_swap_ips(void *a, void *b, int size)
41639fd49328SSteven Rostedt {
41649fd49328SSteven Rostedt 	unsigned long *ipa = a;
41659fd49328SSteven Rostedt 	unsigned long *ipb = b;
41669fd49328SSteven Rostedt 	unsigned long t;
41679fd49328SSteven Rostedt 
41689fd49328SSteven Rostedt 	t = *ipa;
41699fd49328SSteven Rostedt 	*ipa = *ipb;
41709fd49328SSteven Rostedt 	*ipb = t;
417168950619SSteven Rostedt }
417268950619SSteven Rostedt 
41735cb084bbSJiri Olsa static int ftrace_process_locs(struct module *mod,
417431e88909SSteven Rostedt 			       unsigned long *start,
417568bf21aaSSteven Rostedt 			       unsigned long *end)
417668bf21aaSSteven Rostedt {
4177706c81f8SSteven Rostedt 	struct ftrace_page *start_pg;
4178a7900875SSteven Rostedt 	struct ftrace_page *pg;
4179706c81f8SSteven Rostedt 	struct dyn_ftrace *rec;
4180a7900875SSteven Rostedt 	unsigned long count;
418168bf21aaSSteven Rostedt 	unsigned long *p;
418268bf21aaSSteven Rostedt 	unsigned long addr;
41834376cac6SSteven Rostedt 	unsigned long flags = 0; /* Shut up gcc */
4184a7900875SSteven Rostedt 	int ret = -ENOMEM;
4185a7900875SSteven Rostedt 
4186a7900875SSteven Rostedt 	count = end - start;
4187a7900875SSteven Rostedt 
4188a7900875SSteven Rostedt 	if (!count)
4189a7900875SSteven Rostedt 		return 0;
4190a7900875SSteven Rostedt 
41919fd49328SSteven Rostedt 	sort(start, count, sizeof(*start),
41929fd49328SSteven Rostedt 	     ftrace_cmp_ips, ftrace_swap_ips);
41939fd49328SSteven Rostedt 
4194706c81f8SSteven Rostedt 	start_pg = ftrace_allocate_pages(count);
4195706c81f8SSteven Rostedt 	if (!start_pg)
4196a7900875SSteven Rostedt 		return -ENOMEM;
419768bf21aaSSteven Rostedt 
4198e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
4199a7900875SSteven Rostedt 
420032082309SSteven Rostedt 	/*
420132082309SSteven Rostedt 	 * Core and each module needs their own pages, as
420232082309SSteven Rostedt 	 * modules will free them when they are removed.
420332082309SSteven Rostedt 	 * Force a new page to be allocated for modules.
420432082309SSteven Rostedt 	 */
4205a7900875SSteven Rostedt 	if (!mod) {
4206a7900875SSteven Rostedt 		WARN_ON(ftrace_pages || ftrace_pages_start);
4207a7900875SSteven Rostedt 		/* First initialization */
4208706c81f8SSteven Rostedt 		ftrace_pages = ftrace_pages_start = start_pg;
4209a7900875SSteven Rostedt 	} else {
421032082309SSteven Rostedt 		if (!ftrace_pages)
4211a7900875SSteven Rostedt 			goto out;
421232082309SSteven Rostedt 
4213a7900875SSteven Rostedt 		if (WARN_ON(ftrace_pages->next)) {
4214a7900875SSteven Rostedt 			/* Hmm, we have free pages? */
4215a7900875SSteven Rostedt 			while (ftrace_pages->next)
421632082309SSteven Rostedt 				ftrace_pages = ftrace_pages->next;
421732082309SSteven Rostedt 		}
4218a7900875SSteven Rostedt 
4219706c81f8SSteven Rostedt 		ftrace_pages->next = start_pg;
422032082309SSteven Rostedt 	}
422132082309SSteven Rostedt 
422268bf21aaSSteven Rostedt 	p = start;
4223706c81f8SSteven Rostedt 	pg = start_pg;
422468bf21aaSSteven Rostedt 	while (p < end) {
422568bf21aaSSteven Rostedt 		addr = ftrace_call_adjust(*p++);
422620e5227eSSteven Rostedt 		/*
422720e5227eSSteven Rostedt 		 * Some architecture linkers will pad between
422820e5227eSSteven Rostedt 		 * the different mcount_loc sections of different
422920e5227eSSteven Rostedt 		 * object files to satisfy alignments.
423020e5227eSSteven Rostedt 		 * Skip any NULL pointers.
423120e5227eSSteven Rostedt 		 */
423220e5227eSSteven Rostedt 		if (!addr)
423320e5227eSSteven Rostedt 			continue;
4234706c81f8SSteven Rostedt 
4235706c81f8SSteven Rostedt 		if (pg->index == pg->size) {
4236706c81f8SSteven Rostedt 			/* We should have allocated enough */
4237706c81f8SSteven Rostedt 			if (WARN_ON(!pg->next))
4238a7900875SSteven Rostedt 				break;
4239706c81f8SSteven Rostedt 			pg = pg->next;
424068bf21aaSSteven Rostedt 		}
424168bf21aaSSteven Rostedt 
4242706c81f8SSteven Rostedt 		rec = &pg->records[pg->index++];
4243706c81f8SSteven Rostedt 		rec->ip = addr;
4244706c81f8SSteven Rostedt 	}
4245706c81f8SSteven Rostedt 
4246706c81f8SSteven Rostedt 	/* We should have used all pages */
4247706c81f8SSteven Rostedt 	WARN_ON(pg->next);
4248706c81f8SSteven Rostedt 
4249706c81f8SSteven Rostedt 	/* Assign the last page to ftrace_pages */
4250706c81f8SSteven Rostedt 	ftrace_pages = pg;
4251706c81f8SSteven Rostedt 
4252a4f18ed1SSteven Rostedt 	/*
42534376cac6SSteven Rostedt 	 * We only need to disable interrupts on start up
42544376cac6SSteven Rostedt 	 * because we are modifying code that an interrupt
42554376cac6SSteven Rostedt 	 * may execute, and the modification is not atomic.
42564376cac6SSteven Rostedt 	 * But for modules, nothing runs the code we modify
42574376cac6SSteven Rostedt 	 * until we are finished with it, and there's no
42584376cac6SSteven Rostedt 	 * reason to cause large interrupt latencies while we do it.
4259a4f18ed1SSteven Rostedt 	 */
42604376cac6SSteven Rostedt 	if (!mod)
4261a4f18ed1SSteven Rostedt 		local_irq_save(flags);
42621dc43cf0SJiri Slaby 	ftrace_update_code(mod, start_pg);
42634376cac6SSteven Rostedt 	if (!mod)
4264a4f18ed1SSteven Rostedt 		local_irq_restore(flags);
4265a7900875SSteven Rostedt 	ret = 0;
4266a7900875SSteven Rostedt  out:
4267e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
426868bf21aaSSteven Rostedt 
4269a7900875SSteven Rostedt 	return ret;
427068bf21aaSSteven Rostedt }
427168bf21aaSSteven Rostedt 
427293eb677dSSteven Rostedt #ifdef CONFIG_MODULES
427332082309SSteven Rostedt 
427432082309SSteven Rostedt #define next_to_ftrace_page(p) container_of(p, struct ftrace_page, next)
427532082309SSteven Rostedt 
4276e7247a15S[email protected] void ftrace_release_mod(struct module *mod)
427793eb677dSSteven Rostedt {
427893eb677dSSteven Rostedt 	struct dyn_ftrace *rec;
427932082309SSteven Rostedt 	struct ftrace_page **last_pg;
428093eb677dSSteven Rostedt 	struct ftrace_page *pg;
4281a7900875SSteven Rostedt 	int order;
428293eb677dSSteven Rostedt 
428393eb677dSSteven Rostedt 	mutex_lock(&ftrace_lock);
428445a4a237SSteven Rostedt 
428545a4a237SSteven Rostedt 	if (ftrace_disabled)
428645a4a237SSteven Rostedt 		goto out_unlock;
428745a4a237SSteven Rostedt 
428832082309SSteven Rostedt 	/*
428932082309SSteven Rostedt 	 * Each module has its own ftrace_pages, remove
429032082309SSteven Rostedt 	 * them from the list.
429132082309SSteven Rostedt 	 */
429232082309SSteven Rostedt 	last_pg = &ftrace_pages_start;
429332082309SSteven Rostedt 	for (pg = ftrace_pages_start; pg; pg = *last_pg) {
429432082309SSteven Rostedt 		rec = &pg->records[0];
4295e7247a15S[email protected] 		if (within_module_core(rec->ip, mod)) {
429693eb677dSSteven Rostedt 			/*
429732082309SSteven Rostedt 			 * As core pages are first, the first
429832082309SSteven Rostedt 			 * page should never be a module page.
429993eb677dSSteven Rostedt 			 */
430032082309SSteven Rostedt 			if (WARN_ON(pg == ftrace_pages_start))
430132082309SSteven Rostedt 				goto out_unlock;
430232082309SSteven Rostedt 
430332082309SSteven Rostedt 			/* Check if we are deleting the last page */
430432082309SSteven Rostedt 			if (pg == ftrace_pages)
430532082309SSteven Rostedt 				ftrace_pages = next_to_ftrace_page(last_pg);
430632082309SSteven Rostedt 
430732082309SSteven Rostedt 			*last_pg = pg->next;
4308a7900875SSteven Rostedt 			order = get_count_order(pg->size / ENTRIES_PER_PAGE);
4309a7900875SSteven Rostedt 			free_pages((unsigned long)pg->records, order);
4310a7900875SSteven Rostedt 			kfree(pg);
431132082309SSteven Rostedt 		} else
431232082309SSteven Rostedt 			last_pg = &pg->next;
431393eb677dSSteven Rostedt 	}
431445a4a237SSteven Rostedt  out_unlock:
431593eb677dSSteven Rostedt 	mutex_unlock(&ftrace_lock);
431693eb677dSSteven Rostedt }
431793eb677dSSteven Rostedt 
431893eb677dSSteven Rostedt static void ftrace_init_module(struct module *mod,
431931e88909SSteven Rostedt 			       unsigned long *start, unsigned long *end)
432090d595feSSteven Rostedt {
432100fd61aeSSteven Rostedt 	if (ftrace_disabled || start == end)
4322fed1939cSSteven Rostedt 		return;
43235cb084bbSJiri Olsa 	ftrace_process_locs(mod, start, end);
432490d595feSSteven Rostedt }
432590d595feSSteven Rostedt 
43268c189ea6SSteven Rostedt (Red Hat) static int ftrace_module_notify_enter(struct notifier_block *self,
432793eb677dSSteven Rostedt 				      unsigned long val, void *data)
432893eb677dSSteven Rostedt {
432993eb677dSSteven Rostedt 	struct module *mod = data;
433093eb677dSSteven Rostedt 
43318c189ea6SSteven Rostedt (Red Hat) 	if (val == MODULE_STATE_COMING)
433293eb677dSSteven Rostedt 		ftrace_init_module(mod, mod->ftrace_callsites,
433393eb677dSSteven Rostedt 				   mod->ftrace_callsites +
433493eb677dSSteven Rostedt 				   mod->num_ftrace_callsites);
43358c189ea6SSteven Rostedt (Red Hat) 	return 0;
433693eb677dSSteven Rostedt }
433793eb677dSSteven Rostedt 
43388c189ea6SSteven Rostedt (Red Hat) static int ftrace_module_notify_exit(struct notifier_block *self,
43398c189ea6SSteven Rostedt (Red Hat) 				     unsigned long val, void *data)
43408c189ea6SSteven Rostedt (Red Hat) {
43418c189ea6SSteven Rostedt (Red Hat) 	struct module *mod = data;
43428c189ea6SSteven Rostedt (Red Hat) 
43438c189ea6SSteven Rostedt (Red Hat) 	if (val == MODULE_STATE_GOING)
43448c189ea6SSteven Rostedt (Red Hat) 		ftrace_release_mod(mod);
43458c189ea6SSteven Rostedt (Red Hat) 
434693eb677dSSteven Rostedt 	return 0;
434793eb677dSSteven Rostedt }
434893eb677dSSteven Rostedt #else
43498c189ea6SSteven Rostedt (Red Hat) static int ftrace_module_notify_enter(struct notifier_block *self,
43508c189ea6SSteven Rostedt (Red Hat) 				      unsigned long val, void *data)
43518c189ea6SSteven Rostedt (Red Hat) {
43528c189ea6SSteven Rostedt (Red Hat) 	return 0;
43538c189ea6SSteven Rostedt (Red Hat) }
43548c189ea6SSteven Rostedt (Red Hat) static int ftrace_module_notify_exit(struct notifier_block *self,
435593eb677dSSteven Rostedt 				     unsigned long val, void *data)
435693eb677dSSteven Rostedt {
435793eb677dSSteven Rostedt 	return 0;
435893eb677dSSteven Rostedt }
435993eb677dSSteven Rostedt #endif /* CONFIG_MODULES */
436093eb677dSSteven Rostedt 
43618c189ea6SSteven Rostedt (Red Hat) struct notifier_block ftrace_module_enter_nb = {
43628c189ea6SSteven Rostedt (Red Hat) 	.notifier_call = ftrace_module_notify_enter,
4363c1bf08acSSteven Rostedt 	.priority = INT_MAX,	/* Run before anything that can use kprobes */
436493eb677dSSteven Rostedt };
436593eb677dSSteven Rostedt 
43668c189ea6SSteven Rostedt (Red Hat) struct notifier_block ftrace_module_exit_nb = {
43678c189ea6SSteven Rostedt (Red Hat) 	.notifier_call = ftrace_module_notify_exit,
43688c189ea6SSteven Rostedt (Red Hat) 	.priority = INT_MIN,	/* Run after anything that can remove kprobes */
43698c189ea6SSteven Rostedt (Red Hat) };
43708c189ea6SSteven Rostedt (Red Hat) 
437168bf21aaSSteven Rostedt void __init ftrace_init(void)
437268bf21aaSSteven Rostedt {
43731dc43cf0SJiri Slaby 	extern unsigned long __start_mcount_loc[];
43741dc43cf0SJiri Slaby 	extern unsigned long __stop_mcount_loc[];
4375*3a36cb11SJiri Slaby 	unsigned long count, flags;
437668bf21aaSSteven Rostedt 	int ret;
437768bf21aaSSteven Rostedt 
437868bf21aaSSteven Rostedt 	local_irq_save(flags);
4379*3a36cb11SJiri Slaby 	ret = ftrace_dyn_arch_init();
438068bf21aaSSteven Rostedt 	local_irq_restore(flags);
4381af64a7cbSJiri Slaby 	if (ret)
438268bf21aaSSteven Rostedt 		goto failed;
438368bf21aaSSteven Rostedt 
438468bf21aaSSteven Rostedt 	count = __stop_mcount_loc - __start_mcount_loc;
4385c867ccd8SJiri Slaby 	if (!count) {
4386c867ccd8SJiri Slaby 		pr_info("ftrace: No functions to be traced?\n");
438768bf21aaSSteven Rostedt 		goto failed;
4388c867ccd8SJiri Slaby 	}
4389c867ccd8SJiri Slaby 
4390c867ccd8SJiri Slaby 	pr_info("ftrace: allocating %ld entries in %ld pages\n",
4391c867ccd8SJiri Slaby 		count, count / ENTRIES_PER_PAGE + 1);
439268bf21aaSSteven Rostedt 
439368bf21aaSSteven Rostedt 	last_ftrace_enabled = ftrace_enabled = 1;
439468bf21aaSSteven Rostedt 
43955cb084bbSJiri Olsa 	ret = ftrace_process_locs(NULL,
439631e88909SSteven Rostedt 				  __start_mcount_loc,
439768bf21aaSSteven Rostedt 				  __stop_mcount_loc);
439868bf21aaSSteven Rostedt 
43998c189ea6SSteven Rostedt (Red Hat) 	ret = register_module_notifier(&ftrace_module_enter_nb);
440024ed0c4bSMing Lei 	if (ret)
44018c189ea6SSteven Rostedt (Red Hat) 		pr_warning("Failed to register trace ftrace module enter notifier\n");
44028c189ea6SSteven Rostedt (Red Hat) 
44038c189ea6SSteven Rostedt (Red Hat) 	ret = register_module_notifier(&ftrace_module_exit_nb);
44048c189ea6SSteven Rostedt (Red Hat) 	if (ret)
44058c189ea6SSteven Rostedt (Red Hat) 		pr_warning("Failed to register trace ftrace module exit notifier\n");
440693eb677dSSteven Rostedt 
44072af15d6aSSteven Rostedt 	set_ftrace_early_filters();
44082af15d6aSSteven Rostedt 
440968bf21aaSSteven Rostedt 	return;
441068bf21aaSSteven Rostedt  failed:
441168bf21aaSSteven Rostedt 	ftrace_disabled = 1;
441268bf21aaSSteven Rostedt }
441368bf21aaSSteven Rostedt 
44143d083395SSteven Rostedt #else
44150b6e4d56SFrederic Weisbecker 
44162b499381SSteven Rostedt static struct ftrace_ops global_ops = {
4417bd69c30bSSteven Rostedt 	.func			= ftrace_stub,
4418f04f24fbSMasami Hiramatsu 	.flags			= FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_INITIALIZED,
4419f04f24fbSMasami Hiramatsu 	INIT_REGEX_LOCK(global_ops)
4420bd69c30bSSteven Rostedt };
4421bd69c30bSSteven Rostedt 
44220b6e4d56SFrederic Weisbecker static int __init ftrace_nodyn_init(void)
44230b6e4d56SFrederic Weisbecker {
44240b6e4d56SFrederic Weisbecker 	ftrace_enabled = 1;
44250b6e4d56SFrederic Weisbecker 	return 0;
44260b6e4d56SFrederic Weisbecker }
44276f415672SSteven Rostedt core_initcall(ftrace_nodyn_init);
44280b6e4d56SFrederic Weisbecker 
4429df4fc315SSteven Rostedt static inline int ftrace_init_dyn_debugfs(struct dentry *d_tracer) { return 0; }
4430df4fc315SSteven Rostedt static inline void ftrace_startup_enable(int command) { }
44315a45cfe1SSteven Rostedt /* Keep as macros so we do not need to define the commands */
44323b6cfdb1SSteven Rostedt # define ftrace_startup(ops, command)					\
44333b6cfdb1SSteven Rostedt 	({								\
44348a56d776SSteven Rostedt (Red Hat) 		int ___ret = __register_ftrace_function(ops);		\
44358a56d776SSteven Rostedt (Red Hat) 		if (!___ret)						\
44363b6cfdb1SSteven Rostedt 			(ops)->flags |= FTRACE_OPS_FL_ENABLED;		\
44378a56d776SSteven Rostedt (Red Hat) 		___ret;							\
44383b6cfdb1SSteven Rostedt 	})
44391fcc1553SSteven Rostedt (Red Hat) # define ftrace_shutdown(ops, command)					\
44401fcc1553SSteven Rostedt (Red Hat) 	({								\
44411fcc1553SSteven Rostedt (Red Hat) 		int ___ret = __unregister_ftrace_function(ops);		\
44421fcc1553SSteven Rostedt (Red Hat) 		if (!___ret)						\
44431fcc1553SSteven Rostedt (Red Hat) 			(ops)->flags &= ~FTRACE_OPS_FL_ENABLED;		\
44441fcc1553SSteven Rostedt (Red Hat) 		___ret;							\
44451fcc1553SSteven Rostedt (Red Hat) 	})
44468a56d776SSteven Rostedt (Red Hat) 
4447b0fc494fSSteven Rostedt # define ftrace_startup_sysctl()	do { } while (0)
4448b0fc494fSSteven Rostedt # define ftrace_shutdown_sysctl()	do { } while (0)
4449b848914cSSteven Rostedt 
4450b848914cSSteven Rostedt static inline int
4451195a8afcSSteven Rostedt (Red Hat) ftrace_ops_test(struct ftrace_ops *ops, unsigned long ip, void *regs)
4452b848914cSSteven Rostedt {
4453b848914cSSteven Rostedt 	return 1;
4454b848914cSSteven Rostedt }
4455b848914cSSteven Rostedt 
44563d083395SSteven Rostedt #endif /* CONFIG_DYNAMIC_FTRACE */
44573d083395SSteven Rostedt 
4458b848914cSSteven Rostedt static void
44592f5f6ad9SSteven Rostedt ftrace_ops_control_func(unsigned long ip, unsigned long parent_ip,
4460a1e2e31dSSteven Rostedt 			struct ftrace_ops *op, struct pt_regs *regs)
4461e248491aSJiri Olsa {
4462e248491aSJiri Olsa 	if (unlikely(trace_recursion_test(TRACE_CONTROL_BIT)))
4463e248491aSJiri Olsa 		return;
4464e248491aSJiri Olsa 
4465e248491aSJiri Olsa 	/*
4466e248491aSJiri Olsa 	 * Some of the ops may be dynamically allocated,
4467e248491aSJiri Olsa 	 * they must be freed after a synchronize_sched().
4468e248491aSJiri Olsa 	 */
4469e248491aSJiri Olsa 	preempt_disable_notrace();
4470e248491aSJiri Olsa 	trace_recursion_set(TRACE_CONTROL_BIT);
4471b5aa3a47SSteven Rostedt (Red Hat) 
4472b5aa3a47SSteven Rostedt (Red Hat) 	/*
4473b5aa3a47SSteven Rostedt (Red Hat) 	 * Control funcs (perf) uses RCU. Only trace if
4474b5aa3a47SSteven Rostedt (Red Hat) 	 * RCU is currently active.
4475b5aa3a47SSteven Rostedt (Red Hat) 	 */
4476b5aa3a47SSteven Rostedt (Red Hat) 	if (!rcu_is_watching())
4477b5aa3a47SSteven Rostedt (Red Hat) 		goto out;
4478b5aa3a47SSteven Rostedt (Red Hat) 
44790a016409SSteven Rostedt 	do_for_each_ftrace_op(op, ftrace_control_list) {
4480395b97a3SSteven Rostedt (Red Hat) 		if (!(op->flags & FTRACE_OPS_FL_STUB) &&
4481395b97a3SSteven Rostedt (Red Hat) 		    !ftrace_function_local_disabled(op) &&
4482195a8afcSSteven Rostedt (Red Hat) 		    ftrace_ops_test(op, ip, regs))
4483a1e2e31dSSteven Rostedt 			op->func(ip, parent_ip, op, regs);
44840a016409SSteven Rostedt 	} while_for_each_ftrace_op(op);
4485b5aa3a47SSteven Rostedt (Red Hat)  out:
4486e248491aSJiri Olsa 	trace_recursion_clear(TRACE_CONTROL_BIT);
4487e248491aSJiri Olsa 	preempt_enable_notrace();
4488e248491aSJiri Olsa }
4489e248491aSJiri Olsa 
4490e248491aSJiri Olsa static struct ftrace_ops control_ops = {
4491e248491aSJiri Olsa 	.func	= ftrace_ops_control_func,
4492f04f24fbSMasami Hiramatsu 	.flags	= FTRACE_OPS_FL_RECURSION_SAFE | FTRACE_OPS_FL_INITIALIZED,
4493f04f24fbSMasami Hiramatsu 	INIT_REGEX_LOCK(control_ops)
4494e248491aSJiri Olsa };
4495e248491aSJiri Olsa 
44962f5f6ad9SSteven Rostedt static inline void
44972f5f6ad9SSteven Rostedt __ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
4498a1e2e31dSSteven Rostedt 		       struct ftrace_ops *ignored, struct pt_regs *regs)
4499b848914cSSteven Rostedt {
4500cdbe61bfSSteven Rostedt 	struct ftrace_ops *op;
4501edc15cafSSteven Rostedt 	int bit;
4502b848914cSSteven Rostedt 
4503ccf3672dSSteven Rostedt 	if (function_trace_stop)
4504ccf3672dSSteven Rostedt 		return;
4505ccf3672dSSteven Rostedt 
4506edc15cafSSteven Rostedt 	bit = trace_test_and_set_recursion(TRACE_LIST_START, TRACE_LIST_MAX);
4507edc15cafSSteven Rostedt 	if (bit < 0)
4508b1cff0adSSteven Rostedt 		return;
4509b1cff0adSSteven Rostedt 
4510cdbe61bfSSteven Rostedt 	/*
4511cdbe61bfSSteven Rostedt 	 * Some of the ops may be dynamically allocated,
4512cdbe61bfSSteven Rostedt 	 * they must be freed after a synchronize_sched().
4513cdbe61bfSSteven Rostedt 	 */
4514cdbe61bfSSteven Rostedt 	preempt_disable_notrace();
45150a016409SSteven Rostedt 	do_for_each_ftrace_op(op, ftrace_ops_list) {
4516195a8afcSSteven Rostedt (Red Hat) 		if (ftrace_ops_test(op, ip, regs))
4517a1e2e31dSSteven Rostedt 			op->func(ip, parent_ip, op, regs);
45180a016409SSteven Rostedt 	} while_for_each_ftrace_op(op);
4519cdbe61bfSSteven Rostedt 	preempt_enable_notrace();
4520edc15cafSSteven Rostedt 	trace_clear_recursion(bit);
4521b848914cSSteven Rostedt }
4522b848914cSSteven Rostedt 
45232f5f6ad9SSteven Rostedt /*
45242f5f6ad9SSteven Rostedt  * Some archs only support passing ip and parent_ip. Even though
45252f5f6ad9SSteven Rostedt  * the list function ignores the op parameter, we do not want any
45262f5f6ad9SSteven Rostedt  * C side effects, where a function is called without the caller
45272f5f6ad9SSteven Rostedt  * sending a third parameter.
4528a1e2e31dSSteven Rostedt  * Archs are to support both the regs and ftrace_ops at the same time.
4529a1e2e31dSSteven Rostedt  * If they support ftrace_ops, it is assumed they support regs.
4530a1e2e31dSSteven Rostedt  * If call backs want to use regs, they must either check for regs
453106aeaaeaSMasami Hiramatsu  * being NULL, or CONFIG_DYNAMIC_FTRACE_WITH_REGS.
453206aeaaeaSMasami Hiramatsu  * Note, CONFIG_DYNAMIC_FTRACE_WITH_REGS expects a full regs to be saved.
4533a1e2e31dSSteven Rostedt  * An architecture can pass partial regs with ftrace_ops and still
4534a1e2e31dSSteven Rostedt  * set the ARCH_SUPPORT_FTARCE_OPS.
45352f5f6ad9SSteven Rostedt  */
45362f5f6ad9SSteven Rostedt #if ARCH_SUPPORTS_FTRACE_OPS
45372f5f6ad9SSteven Rostedt static void ftrace_ops_list_func(unsigned long ip, unsigned long parent_ip,
4538a1e2e31dSSteven Rostedt 				 struct ftrace_ops *op, struct pt_regs *regs)
45392f5f6ad9SSteven Rostedt {
4540a1e2e31dSSteven Rostedt 	__ftrace_ops_list_func(ip, parent_ip, NULL, regs);
45412f5f6ad9SSteven Rostedt }
45422f5f6ad9SSteven Rostedt #else
45432f5f6ad9SSteven Rostedt static void ftrace_ops_no_ops(unsigned long ip, unsigned long parent_ip)
45442f5f6ad9SSteven Rostedt {
4545a1e2e31dSSteven Rostedt 	__ftrace_ops_list_func(ip, parent_ip, NULL, NULL);
45462f5f6ad9SSteven Rostedt }
45472f5f6ad9SSteven Rostedt #endif
45482f5f6ad9SSteven Rostedt 
4549e32d8956SSteven Rostedt static void clear_ftrace_swapper(void)
4550e32d8956SSteven Rostedt {
4551e32d8956SSteven Rostedt 	struct task_struct *p;
4552e32d8956SSteven Rostedt 	int cpu;
4553e32d8956SSteven Rostedt 
4554e32d8956SSteven Rostedt 	get_online_cpus();
4555e32d8956SSteven Rostedt 	for_each_online_cpu(cpu) {
4556e32d8956SSteven Rostedt 		p = idle_task(cpu);
4557e32d8956SSteven Rostedt 		clear_tsk_trace_trace(p);
4558e32d8956SSteven Rostedt 	}
4559e32d8956SSteven Rostedt 	put_online_cpus();
4560e32d8956SSteven Rostedt }
4561e32d8956SSteven Rostedt 
4562e32d8956SSteven Rostedt static void set_ftrace_swapper(void)
4563e32d8956SSteven Rostedt {
4564e32d8956SSteven Rostedt 	struct task_struct *p;
4565e32d8956SSteven Rostedt 	int cpu;
4566e32d8956SSteven Rostedt 
4567e32d8956SSteven Rostedt 	get_online_cpus();
4568e32d8956SSteven Rostedt 	for_each_online_cpu(cpu) {
4569e32d8956SSteven Rostedt 		p = idle_task(cpu);
4570e32d8956SSteven Rostedt 		set_tsk_trace_trace(p);
4571e32d8956SSteven Rostedt 	}
4572e32d8956SSteven Rostedt 	put_online_cpus();
4573e32d8956SSteven Rostedt }
4574e32d8956SSteven Rostedt 
4575e32d8956SSteven Rostedt static void clear_ftrace_pid(struct pid *pid)
4576978f3a45SSteven Rostedt {
4577978f3a45SSteven Rostedt 	struct task_struct *p;
4578978f3a45SSteven Rostedt 
4579229c4ef8SOleg Nesterov 	rcu_read_lock();
4580e32d8956SSteven Rostedt 	do_each_pid_task(pid, PIDTYPE_PID, p) {
4581978f3a45SSteven Rostedt 		clear_tsk_trace_trace(p);
4582e32d8956SSteven Rostedt 	} while_each_pid_task(pid, PIDTYPE_PID, p);
4583229c4ef8SOleg Nesterov 	rcu_read_unlock();
4584229c4ef8SOleg Nesterov 
4585e32d8956SSteven Rostedt 	put_pid(pid);
4586978f3a45SSteven Rostedt }
4587978f3a45SSteven Rostedt 
4588e32d8956SSteven Rostedt static void set_ftrace_pid(struct pid *pid)
4589978f3a45SSteven Rostedt {
4590978f3a45SSteven Rostedt 	struct task_struct *p;
4591978f3a45SSteven Rostedt 
4592229c4ef8SOleg Nesterov 	rcu_read_lock();
4593978f3a45SSteven Rostedt 	do_each_pid_task(pid, PIDTYPE_PID, p) {
4594978f3a45SSteven Rostedt 		set_tsk_trace_trace(p);
4595978f3a45SSteven Rostedt 	} while_each_pid_task(pid, PIDTYPE_PID, p);
4596229c4ef8SOleg Nesterov 	rcu_read_unlock();
4597978f3a45SSteven Rostedt }
4598978f3a45SSteven Rostedt 
4599756d17eeS[email protected] static void clear_ftrace_pid_task(struct pid *pid)
4600e32d8956SSteven Rostedt {
4601756d17eeS[email protected] 	if (pid == ftrace_swapper_pid)
4602e32d8956SSteven Rostedt 		clear_ftrace_swapper();
4603e32d8956SSteven Rostedt 	else
4604756d17eeS[email protected] 		clear_ftrace_pid(pid);
4605e32d8956SSteven Rostedt }
4606e32d8956SSteven Rostedt 
4607e32d8956SSteven Rostedt static void set_ftrace_pid_task(struct pid *pid)
4608e32d8956SSteven Rostedt {
4609e32d8956SSteven Rostedt 	if (pid == ftrace_swapper_pid)
4610e32d8956SSteven Rostedt 		set_ftrace_swapper();
4611e32d8956SSteven Rostedt 	else
4612e32d8956SSteven Rostedt 		set_ftrace_pid(pid);
4613e32d8956SSteven Rostedt }
4614e32d8956SSteven Rostedt 
4615756d17eeS[email protected] static int ftrace_pid_add(int p)
4616756d17eeS[email protected] {
4617756d17eeS[email protected] 	struct pid *pid;
4618756d17eeS[email protected] 	struct ftrace_pid *fpid;
4619756d17eeS[email protected] 	int ret = -EINVAL;
4620756d17eeS[email protected] 
4621756d17eeS[email protected] 	mutex_lock(&ftrace_lock);
4622756d17eeS[email protected] 
4623756d17eeS[email protected] 	if (!p)
4624756d17eeS[email protected] 		pid = ftrace_swapper_pid;
4625756d17eeS[email protected] 	else
4626756d17eeS[email protected] 		pid = find_get_pid(p);
4627756d17eeS[email protected] 
4628756d17eeS[email protected] 	if (!pid)
4629756d17eeS[email protected] 		goto out;
4630756d17eeS[email protected] 
4631756d17eeS[email protected] 	ret = 0;
4632756d17eeS[email protected] 
4633756d17eeS[email protected] 	list_for_each_entry(fpid, &ftrace_pids, list)
4634756d17eeS[email protected] 		if (fpid->pid == pid)
4635756d17eeS[email protected] 			goto out_put;
4636756d17eeS[email protected] 
4637756d17eeS[email protected] 	ret = -ENOMEM;
4638756d17eeS[email protected] 
4639756d17eeS[email protected] 	fpid = kmalloc(sizeof(*fpid), GFP_KERNEL);
4640756d17eeS[email protected] 	if (!fpid)
4641756d17eeS[email protected] 		goto out_put;
4642756d17eeS[email protected] 
4643756d17eeS[email protected] 	list_add(&fpid->list, &ftrace_pids);
4644756d17eeS[email protected] 	fpid->pid = pid;
4645756d17eeS[email protected] 
4646756d17eeS[email protected] 	set_ftrace_pid_task(pid);
4647756d17eeS[email protected] 
4648756d17eeS[email protected] 	ftrace_update_pid_func();
4649756d17eeS[email protected] 	ftrace_startup_enable(0);
4650756d17eeS[email protected] 
4651756d17eeS[email protected] 	mutex_unlock(&ftrace_lock);
4652756d17eeS[email protected] 	return 0;
4653756d17eeS[email protected] 
4654756d17eeS[email protected] out_put:
4655756d17eeS[email protected] 	if (pid != ftrace_swapper_pid)
4656756d17eeS[email protected] 		put_pid(pid);
4657756d17eeS[email protected] 
4658756d17eeS[email protected] out:
4659756d17eeS[email protected] 	mutex_unlock(&ftrace_lock);
4660756d17eeS[email protected] 	return ret;
4661756d17eeS[email protected] }
4662756d17eeS[email protected] 
4663756d17eeS[email protected] static void ftrace_pid_reset(void)
4664756d17eeS[email protected] {
4665756d17eeS[email protected] 	struct ftrace_pid *fpid, *safe;
4666756d17eeS[email protected] 
4667756d17eeS[email protected] 	mutex_lock(&ftrace_lock);
4668756d17eeS[email protected] 	list_for_each_entry_safe(fpid, safe, &ftrace_pids, list) {
4669756d17eeS[email protected] 		struct pid *pid = fpid->pid;
4670756d17eeS[email protected] 
4671756d17eeS[email protected] 		clear_ftrace_pid_task(pid);
4672756d17eeS[email protected] 
4673756d17eeS[email protected] 		list_del(&fpid->list);
4674756d17eeS[email protected] 		kfree(fpid);
4675756d17eeS[email protected] 	}
4676756d17eeS[email protected] 
4677756d17eeS[email protected] 	ftrace_update_pid_func();
4678756d17eeS[email protected] 	ftrace_startup_enable(0);
4679756d17eeS[email protected] 
4680756d17eeS[email protected] 	mutex_unlock(&ftrace_lock);
4681756d17eeS[email protected] }
4682756d17eeS[email protected] 
4683756d17eeS[email protected] static void *fpid_start(struct seq_file *m, loff_t *pos)
4684756d17eeS[email protected] {
4685756d17eeS[email protected] 	mutex_lock(&ftrace_lock);
4686756d17eeS[email protected] 
4687756d17eeS[email protected] 	if (list_empty(&ftrace_pids) && (!*pos))
4688756d17eeS[email protected] 		return (void *) 1;
4689756d17eeS[email protected] 
4690756d17eeS[email protected] 	return seq_list_start(&ftrace_pids, *pos);
4691756d17eeS[email protected] }
4692756d17eeS[email protected] 
4693756d17eeS[email protected] static void *fpid_next(struct seq_file *m, void *v, loff_t *pos)
4694756d17eeS[email protected] {
4695756d17eeS[email protected] 	if (v == (void *)1)
4696756d17eeS[email protected] 		return NULL;
4697756d17eeS[email protected] 
4698756d17eeS[email protected] 	return seq_list_next(v, &ftrace_pids, pos);
4699756d17eeS[email protected] }
4700756d17eeS[email protected] 
4701756d17eeS[email protected] static void fpid_stop(struct seq_file *m, void *p)
4702756d17eeS[email protected] {
4703756d17eeS[email protected] 	mutex_unlock(&ftrace_lock);
4704756d17eeS[email protected] }
4705756d17eeS[email protected] 
4706756d17eeS[email protected] static int fpid_show(struct seq_file *m, void *v)
4707756d17eeS[email protected] {
4708756d17eeS[email protected] 	const struct ftrace_pid *fpid = list_entry(v, struct ftrace_pid, list);
4709756d17eeS[email protected] 
4710756d17eeS[email protected] 	if (v == (void *)1) {
4711756d17eeS[email protected] 		seq_printf(m, "no pid\n");
4712756d17eeS[email protected] 		return 0;
4713756d17eeS[email protected] 	}
4714756d17eeS[email protected] 
4715756d17eeS[email protected] 	if (fpid->pid == ftrace_swapper_pid)
4716756d17eeS[email protected] 		seq_printf(m, "swapper tasks\n");
4717756d17eeS[email protected] 	else
4718756d17eeS[email protected] 		seq_printf(m, "%u\n", pid_vnr(fpid->pid));
4719756d17eeS[email protected] 
4720756d17eeS[email protected] 	return 0;
4721756d17eeS[email protected] }
4722756d17eeS[email protected] 
4723756d17eeS[email protected] static const struct seq_operations ftrace_pid_sops = {
4724756d17eeS[email protected] 	.start = fpid_start,
4725756d17eeS[email protected] 	.next = fpid_next,
4726756d17eeS[email protected] 	.stop = fpid_stop,
4727756d17eeS[email protected] 	.show = fpid_show,
4728756d17eeS[email protected] };
4729756d17eeS[email protected] 
4730756d17eeS[email protected] static int
4731756d17eeS[email protected] ftrace_pid_open(struct inode *inode, struct file *file)
4732756d17eeS[email protected] {
4733756d17eeS[email protected] 	int ret = 0;
4734756d17eeS[email protected] 
4735756d17eeS[email protected] 	if ((file->f_mode & FMODE_WRITE) &&
4736756d17eeS[email protected] 	    (file->f_flags & O_TRUNC))
4737756d17eeS[email protected] 		ftrace_pid_reset();
4738756d17eeS[email protected] 
4739756d17eeS[email protected] 	if (file->f_mode & FMODE_READ)
4740756d17eeS[email protected] 		ret = seq_open(file, &ftrace_pid_sops);
4741756d17eeS[email protected] 
4742756d17eeS[email protected] 	return ret;
4743756d17eeS[email protected] }
4744756d17eeS[email protected] 
4745df4fc315SSteven Rostedt static ssize_t
4746df4fc315SSteven Rostedt ftrace_pid_write(struct file *filp, const char __user *ubuf,
4747df4fc315SSteven Rostedt 		   size_t cnt, loff_t *ppos)
4748df4fc315SSteven Rostedt {
4749457dc928SIngo Molnar 	char buf[64], *tmp;
4750df4fc315SSteven Rostedt 	long val;
4751df4fc315SSteven Rostedt 	int ret;
4752df4fc315SSteven Rostedt 
4753df4fc315SSteven Rostedt 	if (cnt >= sizeof(buf))
4754df4fc315SSteven Rostedt 		return -EINVAL;
4755df4fc315SSteven Rostedt 
4756df4fc315SSteven Rostedt 	if (copy_from_user(&buf, ubuf, cnt))
4757df4fc315SSteven Rostedt 		return -EFAULT;
4758df4fc315SSteven Rostedt 
4759df4fc315SSteven Rostedt 	buf[cnt] = 0;
4760df4fc315SSteven Rostedt 
4761756d17eeS[email protected] 	/*
4762756d17eeS[email protected] 	 * Allow "echo > set_ftrace_pid" or "echo -n '' > set_ftrace_pid"
4763756d17eeS[email protected] 	 * to clean the filter quietly.
4764756d17eeS[email protected] 	 */
4765457dc928SIngo Molnar 	tmp = strstrip(buf);
4766457dc928SIngo Molnar 	if (strlen(tmp) == 0)
4767756d17eeS[email protected] 		return 1;
4768756d17eeS[email protected] 
4769bcd83ea6SDaniel Walter 	ret = kstrtol(tmp, 10, &val);
4770df4fc315SSteven Rostedt 	if (ret < 0)
4771df4fc315SSteven Rostedt 		return ret;
4772df4fc315SSteven Rostedt 
4773756d17eeS[email protected] 	ret = ftrace_pid_add(val);
4774978f3a45SSteven Rostedt 
4775756d17eeS[email protected] 	return ret ? ret : cnt;
4776e32d8956SSteven Rostedt }
4777978f3a45SSteven Rostedt 
4778756d17eeS[email protected] static int
4779756d17eeS[email protected] ftrace_pid_release(struct inode *inode, struct file *file)
4780756d17eeS[email protected] {
4781756d17eeS[email protected] 	if (file->f_mode & FMODE_READ)
4782756d17eeS[email protected] 		seq_release(inode, file);
4783978f3a45SSteven Rostedt 
4784756d17eeS[email protected] 	return 0;
4785df4fc315SSteven Rostedt }
4786df4fc315SSteven Rostedt 
47875e2336a0SSteven Rostedt static const struct file_operations ftrace_pid_fops = {
4788756d17eeS[email protected] 	.open		= ftrace_pid_open,
4789df4fc315SSteven Rostedt 	.write		= ftrace_pid_write,
4790756d17eeS[email protected] 	.read		= seq_read,
4791098c879eSSteven Rostedt (Red Hat) 	.llseek		= tracing_lseek,
4792756d17eeS[email protected] 	.release	= ftrace_pid_release,
4793df4fc315SSteven Rostedt };
4794df4fc315SSteven Rostedt 
4795df4fc315SSteven Rostedt static __init int ftrace_init_debugfs(void)
4796df4fc315SSteven Rostedt {
4797df4fc315SSteven Rostedt 	struct dentry *d_tracer;
4798df4fc315SSteven Rostedt 
4799df4fc315SSteven Rostedt 	d_tracer = tracing_init_dentry();
4800df4fc315SSteven Rostedt 	if (!d_tracer)
4801df4fc315SSteven Rostedt 		return 0;
4802df4fc315SSteven Rostedt 
4803df4fc315SSteven Rostedt 	ftrace_init_dyn_debugfs(d_tracer);
4804df4fc315SSteven Rostedt 
48055452af66SFrederic Weisbecker 	trace_create_file("set_ftrace_pid", 0644, d_tracer,
4806df4fc315SSteven Rostedt 			    NULL, &ftrace_pid_fops);
4807493762fcSSteven Rostedt 
4808493762fcSSteven Rostedt 	ftrace_profile_debugfs(d_tracer);
4809493762fcSSteven Rostedt 
4810df4fc315SSteven Rostedt 	return 0;
4811df4fc315SSteven Rostedt }
4812df4fc315SSteven Rostedt fs_initcall(ftrace_init_debugfs);
4813df4fc315SSteven Rostedt 
48143d083395SSteven Rostedt /**
481581adbdc0SSteven Rostedt  * ftrace_kill - kill ftrace
4816a2bb6a3dSSteven Rostedt  *
4817a2bb6a3dSSteven Rostedt  * This function should be used by panic code. It stops ftrace
4818a2bb6a3dSSteven Rostedt  * but in a not so nice way. If you need to simply kill ftrace
4819a2bb6a3dSSteven Rostedt  * from a non-atomic section, use ftrace_kill.
4820a2bb6a3dSSteven Rostedt  */
482181adbdc0SSteven Rostedt void ftrace_kill(void)
4822a2bb6a3dSSteven Rostedt {
4823a2bb6a3dSSteven Rostedt 	ftrace_disabled = 1;
4824a2bb6a3dSSteven Rostedt 	ftrace_enabled = 0;
4825a2bb6a3dSSteven Rostedt 	clear_ftrace_function();
4826a2bb6a3dSSteven Rostedt }
4827a2bb6a3dSSteven Rostedt 
4828a2bb6a3dSSteven Rostedt /**
4829e0a413f6SSteven Rostedt  * Test if ftrace is dead or not.
4830e0a413f6SSteven Rostedt  */
4831e0a413f6SSteven Rostedt int ftrace_is_dead(void)
4832e0a413f6SSteven Rostedt {
4833e0a413f6SSteven Rostedt 	return ftrace_disabled;
4834e0a413f6SSteven Rostedt }
4835e0a413f6SSteven Rostedt 
4836e0a413f6SSteven Rostedt /**
48373d083395SSteven Rostedt  * register_ftrace_function - register a function for profiling
48383d083395SSteven Rostedt  * @ops - ops structure that holds the function for profiling.
48393d083395SSteven Rostedt  *
48403d083395SSteven Rostedt  * Register a function to be called by all functions in the
48413d083395SSteven Rostedt  * kernel.
48423d083395SSteven Rostedt  *
48433d083395SSteven Rostedt  * Note: @ops->func and all the functions it calls must be labeled
48443d083395SSteven Rostedt  *       with "notrace", otherwise it will go into a
48453d083395SSteven Rostedt  *       recursive loop.
48463d083395SSteven Rostedt  */
48473d083395SSteven Rostedt int register_ftrace_function(struct ftrace_ops *ops)
48483d083395SSteven Rostedt {
484945a4a237SSteven Rostedt 	int ret = -1;
48504eebcc81SSteven Rostedt 
4851f04f24fbSMasami Hiramatsu 	ftrace_ops_init(ops);
4852f04f24fbSMasami Hiramatsu 
4853e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
4854e7d3737eSFrederic Weisbecker 
4855a1cd6173SSteven Rostedt 	ret = ftrace_startup(ops, 0);
4856b848914cSSteven Rostedt 
4857e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
48588d240dd8SBorislav Petkov 
4859b0fc494fSSteven Rostedt 	return ret;
48603d083395SSteven Rostedt }
4861cdbe61bfSSteven Rostedt EXPORT_SYMBOL_GPL(register_ftrace_function);
48623d083395SSteven Rostedt 
48633d083395SSteven Rostedt /**
486432632920SUwe Kleine-Koenig  * unregister_ftrace_function - unregister a function for profiling.
48653d083395SSteven Rostedt  * @ops - ops structure that holds the function to unregister
48663d083395SSteven Rostedt  *
48673d083395SSteven Rostedt  * Unregister a function that was added to be called by ftrace profiling.
48683d083395SSteven Rostedt  */
48693d083395SSteven Rostedt int unregister_ftrace_function(struct ftrace_ops *ops)
48703d083395SSteven Rostedt {
48713d083395SSteven Rostedt 	int ret;
48723d083395SSteven Rostedt 
4873e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
48748a56d776SSteven Rostedt (Red Hat) 	ret = ftrace_shutdown(ops, 0);
4875e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
4876b0fc494fSSteven Rostedt 
4877b0fc494fSSteven Rostedt 	return ret;
4878b0fc494fSSteven Rostedt }
4879cdbe61bfSSteven Rostedt EXPORT_SYMBOL_GPL(unregister_ftrace_function);
4880b0fc494fSSteven Rostedt 
4881e309b41dSIngo Molnar int
4882b0fc494fSSteven Rostedt ftrace_enable_sysctl(struct ctl_table *table, int write,
48838d65af78SAlexey Dobriyan 		     void __user *buffer, size_t *lenp,
4884b0fc494fSSteven Rostedt 		     loff_t *ppos)
4885b0fc494fSSteven Rostedt {
488645a4a237SSteven Rostedt 	int ret = -ENODEV;
48874eebcc81SSteven Rostedt 
4888e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
4889b0fc494fSSteven Rostedt 
489045a4a237SSteven Rostedt 	if (unlikely(ftrace_disabled))
489145a4a237SSteven Rostedt 		goto out;
489245a4a237SSteven Rostedt 
48938d65af78SAlexey Dobriyan 	ret = proc_dointvec(table, write, buffer, lenp, ppos);
4894b0fc494fSSteven Rostedt 
4895a32c7765SLi Zefan 	if (ret || !write || (last_ftrace_enabled == !!ftrace_enabled))
4896b0fc494fSSteven Rostedt 		goto out;
4897b0fc494fSSteven Rostedt 
4898a32c7765SLi Zefan 	last_ftrace_enabled = !!ftrace_enabled;
4899b0fc494fSSteven Rostedt 
4900b0fc494fSSteven Rostedt 	if (ftrace_enabled) {
4901b0fc494fSSteven Rostedt 
4902b0fc494fSSteven Rostedt 		ftrace_startup_sysctl();
4903b0fc494fSSteven Rostedt 
4904b0fc494fSSteven Rostedt 		/* we are starting ftrace again */
49055000c418SJan Kiszka 		if (ftrace_ops_list != &ftrace_list_end)
49065000c418SJan Kiszka 			update_ftrace_function();
4907b0fc494fSSteven Rostedt 
4908b0fc494fSSteven Rostedt 	} else {
4909b0fc494fSSteven Rostedt 		/* stopping ftrace calls (just send to ftrace_stub) */
4910b0fc494fSSteven Rostedt 		ftrace_trace_function = ftrace_stub;
4911b0fc494fSSteven Rostedt 
4912b0fc494fSSteven Rostedt 		ftrace_shutdown_sysctl();
4913b0fc494fSSteven Rostedt 	}
4914b0fc494fSSteven Rostedt 
4915b0fc494fSSteven Rostedt  out:
4916e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
49173d083395SSteven Rostedt 	return ret;
491816444a8aSArnaldo Carvalho de Melo }
4919f17845e5SIngo Molnar 
4920fb52607aSFrederic Weisbecker #ifdef CONFIG_FUNCTION_GRAPH_TRACER
4921e7d3737eSFrederic Weisbecker 
4922597af815SSteven Rostedt static int ftrace_graph_active;
49234a2b8ddaSFrederic Weisbecker static struct notifier_block ftrace_suspend_notifier;
4924f201ae23SFrederic Weisbecker 
4925e49dc19cSSteven Rostedt int ftrace_graph_entry_stub(struct ftrace_graph_ent *trace)
4926e49dc19cSSteven Rostedt {
4927e49dc19cSSteven Rostedt 	return 0;
4928e49dc19cSSteven Rostedt }
4929e49dc19cSSteven Rostedt 
4930287b6e68SFrederic Weisbecker /* The callbacks that hook a function */
4931287b6e68SFrederic Weisbecker trace_func_graph_ret_t ftrace_graph_return =
4932287b6e68SFrederic Weisbecker 			(trace_func_graph_ret_t)ftrace_stub;
4933e49dc19cSSteven Rostedt trace_func_graph_ent_t ftrace_graph_entry = ftrace_graph_entry_stub;
493423a8e844SSteven Rostedt (Red Hat) static trace_func_graph_ent_t __ftrace_graph_entry = ftrace_graph_entry_stub;
4935f201ae23SFrederic Weisbecker 
4936f201ae23SFrederic Weisbecker /* Try to assign a return stack array on FTRACE_RETSTACK_ALLOC_SIZE tasks. */
4937f201ae23SFrederic Weisbecker static int alloc_retstack_tasklist(struct ftrace_ret_stack **ret_stack_list)
4938f201ae23SFrederic Weisbecker {
4939f201ae23SFrederic Weisbecker 	int i;
4940f201ae23SFrederic Weisbecker 	int ret = 0;
4941f201ae23SFrederic Weisbecker 	unsigned long flags;
4942f201ae23SFrederic Weisbecker 	int start = 0, end = FTRACE_RETSTACK_ALLOC_SIZE;
4943f201ae23SFrederic Weisbecker 	struct task_struct *g, *t;
4944f201ae23SFrederic Weisbecker 
4945f201ae23SFrederic Weisbecker 	for (i = 0; i < FTRACE_RETSTACK_ALLOC_SIZE; i++) {
4946f201ae23SFrederic Weisbecker 		ret_stack_list[i] = kmalloc(FTRACE_RETFUNC_DEPTH
4947f201ae23SFrederic Weisbecker 					* sizeof(struct ftrace_ret_stack),
4948f201ae23SFrederic Weisbecker 					GFP_KERNEL);
4949f201ae23SFrederic Weisbecker 		if (!ret_stack_list[i]) {
4950f201ae23SFrederic Weisbecker 			start = 0;
4951f201ae23SFrederic Weisbecker 			end = i;
4952f201ae23SFrederic Weisbecker 			ret = -ENOMEM;
4953f201ae23SFrederic Weisbecker 			goto free;
4954f201ae23SFrederic Weisbecker 		}
4955f201ae23SFrederic Weisbecker 	}
4956f201ae23SFrederic Weisbecker 
4957f201ae23SFrederic Weisbecker 	read_lock_irqsave(&tasklist_lock, flags);
4958f201ae23SFrederic Weisbecker 	do_each_thread(g, t) {
4959f201ae23SFrederic Weisbecker 		if (start == end) {
4960f201ae23SFrederic Weisbecker 			ret = -EAGAIN;
4961f201ae23SFrederic Weisbecker 			goto unlock;
4962f201ae23SFrederic Weisbecker 		}
4963f201ae23SFrederic Weisbecker 
4964f201ae23SFrederic Weisbecker 		if (t->ret_stack == NULL) {
4965380c4b14SFrederic Weisbecker 			atomic_set(&t->tracing_graph_pause, 0);
4966f201ae23SFrederic Weisbecker 			atomic_set(&t->trace_overrun, 0);
496726c01624SSteven Rostedt 			t->curr_ret_stack = -1;
496826c01624SSteven Rostedt 			/* Make sure the tasks see the -1 first: */
496926c01624SSteven Rostedt 			smp_wmb();
497026c01624SSteven Rostedt 			t->ret_stack = ret_stack_list[start++];
4971f201ae23SFrederic Weisbecker 		}
4972f201ae23SFrederic Weisbecker 	} while_each_thread(g, t);
4973f201ae23SFrederic Weisbecker 
4974f201ae23SFrederic Weisbecker unlock:
4975f201ae23SFrederic Weisbecker 	read_unlock_irqrestore(&tasklist_lock, flags);
4976f201ae23SFrederic Weisbecker free:
4977f201ae23SFrederic Weisbecker 	for (i = start; i < end; i++)
4978f201ae23SFrederic Weisbecker 		kfree(ret_stack_list[i]);
4979f201ae23SFrederic Weisbecker 	return ret;
4980f201ae23SFrederic Weisbecker }
4981f201ae23SFrederic Weisbecker 
49828aef2d28SSteven Rostedt static void
498338516ab5SSteven Rostedt ftrace_graph_probe_sched_switch(void *ignore,
498438516ab5SSteven Rostedt 			struct task_struct *prev, struct task_struct *next)
49858aef2d28SSteven Rostedt {
49868aef2d28SSteven Rostedt 	unsigned long long timestamp;
49878aef2d28SSteven Rostedt 	int index;
49888aef2d28SSteven Rostedt 
4989be6f164aSSteven Rostedt 	/*
4990be6f164aSSteven Rostedt 	 * Does the user want to count the time a function was asleep.
4991be6f164aSSteven Rostedt 	 * If so, do not update the time stamps.
4992be6f164aSSteven Rostedt 	 */
4993be6f164aSSteven Rostedt 	if (trace_flags & TRACE_ITER_SLEEP_TIME)
4994be6f164aSSteven Rostedt 		return;
4995be6f164aSSteven Rostedt 
49968aef2d28SSteven Rostedt 	timestamp = trace_clock_local();
49978aef2d28SSteven Rostedt 
49988aef2d28SSteven Rostedt 	prev->ftrace_timestamp = timestamp;
49998aef2d28SSteven Rostedt 
50008aef2d28SSteven Rostedt 	/* only process tasks that we timestamped */
50018aef2d28SSteven Rostedt 	if (!next->ftrace_timestamp)
50028aef2d28SSteven Rostedt 		return;
50038aef2d28SSteven Rostedt 
50048aef2d28SSteven Rostedt 	/*
50058aef2d28SSteven Rostedt 	 * Update all the counters in next to make up for the
50068aef2d28SSteven Rostedt 	 * time next was sleeping.
50078aef2d28SSteven Rostedt 	 */
50088aef2d28SSteven Rostedt 	timestamp -= next->ftrace_timestamp;
50098aef2d28SSteven Rostedt 
50108aef2d28SSteven Rostedt 	for (index = next->curr_ret_stack; index >= 0; index--)
50118aef2d28SSteven Rostedt 		next->ret_stack[index].calltime += timestamp;
50128aef2d28SSteven Rostedt }
50138aef2d28SSteven Rostedt 
5014f201ae23SFrederic Weisbecker /* Allocate a return stack for each task */
5015fb52607aSFrederic Weisbecker static int start_graph_tracing(void)
5016f201ae23SFrederic Weisbecker {
5017f201ae23SFrederic Weisbecker 	struct ftrace_ret_stack **ret_stack_list;
50185b058bcdSFrederic Weisbecker 	int ret, cpu;
5019f201ae23SFrederic Weisbecker 
5020f201ae23SFrederic Weisbecker 	ret_stack_list = kmalloc(FTRACE_RETSTACK_ALLOC_SIZE *
5021f201ae23SFrederic Weisbecker 				sizeof(struct ftrace_ret_stack *),
5022f201ae23SFrederic Weisbecker 				GFP_KERNEL);
5023f201ae23SFrederic Weisbecker 
5024f201ae23SFrederic Weisbecker 	if (!ret_stack_list)
5025f201ae23SFrederic Weisbecker 		return -ENOMEM;
5026f201ae23SFrederic Weisbecker 
50275b058bcdSFrederic Weisbecker 	/* The cpu_boot init_task->ret_stack will never be freed */
5028179c498aSSteven Rostedt 	for_each_online_cpu(cpu) {
5029179c498aSSteven Rostedt 		if (!idle_task(cpu)->ret_stack)
5030868baf07SSteven Rostedt 			ftrace_graph_init_idle_task(idle_task(cpu), cpu);
5031179c498aSSteven Rostedt 	}
50325b058bcdSFrederic Weisbecker 
5033f201ae23SFrederic Weisbecker 	do {
5034f201ae23SFrederic Weisbecker 		ret = alloc_retstack_tasklist(ret_stack_list);
5035f201ae23SFrederic Weisbecker 	} while (ret == -EAGAIN);
5036f201ae23SFrederic Weisbecker 
50378aef2d28SSteven Rostedt 	if (!ret) {
503838516ab5SSteven Rostedt 		ret = register_trace_sched_switch(ftrace_graph_probe_sched_switch, NULL);
50398aef2d28SSteven Rostedt 		if (ret)
50408aef2d28SSteven Rostedt 			pr_info("ftrace_graph: Couldn't activate tracepoint"
50418aef2d28SSteven Rostedt 				" probe to kernel_sched_switch\n");
50428aef2d28SSteven Rostedt 	}
50438aef2d28SSteven Rostedt 
5044f201ae23SFrederic Weisbecker 	kfree(ret_stack_list);
5045f201ae23SFrederic Weisbecker 	return ret;
5046f201ae23SFrederic Weisbecker }
5047f201ae23SFrederic Weisbecker 
50484a2b8ddaSFrederic Weisbecker /*
50494a2b8ddaSFrederic Weisbecker  * Hibernation protection.
50504a2b8ddaSFrederic Weisbecker  * The state of the current task is too much unstable during
50514a2b8ddaSFrederic Weisbecker  * suspend/restore to disk. We want to protect against that.
50524a2b8ddaSFrederic Weisbecker  */
50534a2b8ddaSFrederic Weisbecker static int
50544a2b8ddaSFrederic Weisbecker ftrace_suspend_notifier_call(struct notifier_block *bl, unsigned long state,
50554a2b8ddaSFrederic Weisbecker 							void *unused)
50564a2b8ddaSFrederic Weisbecker {
50574a2b8ddaSFrederic Weisbecker 	switch (state) {
50584a2b8ddaSFrederic Weisbecker 	case PM_HIBERNATION_PREPARE:
50594a2b8ddaSFrederic Weisbecker 		pause_graph_tracing();
50604a2b8ddaSFrederic Weisbecker 		break;
50614a2b8ddaSFrederic Weisbecker 
50624a2b8ddaSFrederic Weisbecker 	case PM_POST_HIBERNATION:
50634a2b8ddaSFrederic Weisbecker 		unpause_graph_tracing();
50644a2b8ddaSFrederic Weisbecker 		break;
50654a2b8ddaSFrederic Weisbecker 	}
50664a2b8ddaSFrederic Weisbecker 	return NOTIFY_DONE;
50674a2b8ddaSFrederic Weisbecker }
50684a2b8ddaSFrederic Weisbecker 
50698a56d776SSteven Rostedt (Red Hat) /* Just a place holder for function graph */
50708a56d776SSteven Rostedt (Red Hat) static struct ftrace_ops fgraph_ops __read_mostly = {
50718a56d776SSteven Rostedt (Red Hat) 	.func		= ftrace_stub,
50728a56d776SSteven Rostedt (Red Hat) 	.flags		= FTRACE_OPS_FL_STUB | FTRACE_OPS_FL_GLOBAL |
50738a56d776SSteven Rostedt (Red Hat) 				FTRACE_OPS_FL_RECURSION_SAFE,
50748a56d776SSteven Rostedt (Red Hat) };
50758a56d776SSteven Rostedt (Red Hat) 
507623a8e844SSteven Rostedt (Red Hat) static int ftrace_graph_entry_test(struct ftrace_graph_ent *trace)
507723a8e844SSteven Rostedt (Red Hat) {
507823a8e844SSteven Rostedt (Red Hat) 	if (!ftrace_ops_test(&global_ops, trace->func, NULL))
507923a8e844SSteven Rostedt (Red Hat) 		return 0;
508023a8e844SSteven Rostedt (Red Hat) 	return __ftrace_graph_entry(trace);
508123a8e844SSteven Rostedt (Red Hat) }
508223a8e844SSteven Rostedt (Red Hat) 
508323a8e844SSteven Rostedt (Red Hat) /*
508423a8e844SSteven Rostedt (Red Hat)  * The function graph tracer should only trace the functions defined
508523a8e844SSteven Rostedt (Red Hat)  * by set_ftrace_filter and set_ftrace_notrace. If another function
508623a8e844SSteven Rostedt (Red Hat)  * tracer ops is registered, the graph tracer requires testing the
508723a8e844SSteven Rostedt (Red Hat)  * function against the global ops, and not just trace any function
508823a8e844SSteven Rostedt (Red Hat)  * that any ftrace_ops registered.
508923a8e844SSteven Rostedt (Red Hat)  */
509023a8e844SSteven Rostedt (Red Hat) static void update_function_graph_func(void)
509123a8e844SSteven Rostedt (Red Hat) {
509223a8e844SSteven Rostedt (Red Hat) 	if (ftrace_ops_list == &ftrace_list_end ||
509323a8e844SSteven Rostedt (Red Hat) 	    (ftrace_ops_list == &global_ops &&
509423a8e844SSteven Rostedt (Red Hat) 	     global_ops.next == &ftrace_list_end))
509523a8e844SSteven Rostedt (Red Hat) 		ftrace_graph_entry = __ftrace_graph_entry;
509623a8e844SSteven Rostedt (Red Hat) 	else
509723a8e844SSteven Rostedt (Red Hat) 		ftrace_graph_entry = ftrace_graph_entry_test;
509823a8e844SSteven Rostedt (Red Hat) }
509923a8e844SSteven Rostedt (Red Hat) 
5100287b6e68SFrederic Weisbecker int register_ftrace_graph(trace_func_graph_ret_t retfunc,
5101287b6e68SFrederic Weisbecker 			trace_func_graph_ent_t entryfunc)
510215e6cb36SFrederic Weisbecker {
5103e7d3737eSFrederic Weisbecker 	int ret = 0;
5104e7d3737eSFrederic Weisbecker 
5105e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
5106e7d3737eSFrederic Weisbecker 
510705ce5818SSteven Rostedt 	/* we currently allow only one tracer registered at a time */
5108597af815SSteven Rostedt 	if (ftrace_graph_active) {
510905ce5818SSteven Rostedt 		ret = -EBUSY;
511005ce5818SSteven Rostedt 		goto out;
511105ce5818SSteven Rostedt 	}
511205ce5818SSteven Rostedt 
51134a2b8ddaSFrederic Weisbecker 	ftrace_suspend_notifier.notifier_call = ftrace_suspend_notifier_call;
51144a2b8ddaSFrederic Weisbecker 	register_pm_notifier(&ftrace_suspend_notifier);
51154a2b8ddaSFrederic Weisbecker 
5116597af815SSteven Rostedt 	ftrace_graph_active++;
5117fb52607aSFrederic Weisbecker 	ret = start_graph_tracing();
5118f201ae23SFrederic Weisbecker 	if (ret) {
5119597af815SSteven Rostedt 		ftrace_graph_active--;
5120f201ae23SFrederic Weisbecker 		goto out;
5121f201ae23SFrederic Weisbecker 	}
5122e53a6319SSteven Rostedt 
5123287b6e68SFrederic Weisbecker 	ftrace_graph_return = retfunc;
512423a8e844SSteven Rostedt (Red Hat) 
512523a8e844SSteven Rostedt (Red Hat) 	/*
512623a8e844SSteven Rostedt (Red Hat) 	 * Update the indirect function to the entryfunc, and the
512723a8e844SSteven Rostedt (Red Hat) 	 * function that gets called to the entry_test first. Then
512823a8e844SSteven Rostedt (Red Hat) 	 * call the update fgraph entry function to determine if
512923a8e844SSteven Rostedt (Red Hat) 	 * the entryfunc should be called directly or not.
513023a8e844SSteven Rostedt (Red Hat) 	 */
513123a8e844SSteven Rostedt (Red Hat) 	__ftrace_graph_entry = entryfunc;
513223a8e844SSteven Rostedt (Red Hat) 	ftrace_graph_entry = ftrace_graph_entry_test;
513323a8e844SSteven Rostedt (Red Hat) 	update_function_graph_func();
5134e53a6319SSteven Rostedt 
51358a56d776SSteven Rostedt (Red Hat) 	ret = ftrace_startup(&fgraph_ops, FTRACE_START_FUNC_RET);
5136e7d3737eSFrederic Weisbecker 
5137e7d3737eSFrederic Weisbecker out:
5138e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
5139e7d3737eSFrederic Weisbecker 	return ret;
514015e6cb36SFrederic Weisbecker }
514115e6cb36SFrederic Weisbecker 
5142fb52607aSFrederic Weisbecker void unregister_ftrace_graph(void)
514315e6cb36SFrederic Weisbecker {
5144e6ea44e9SSteven Rostedt 	mutex_lock(&ftrace_lock);
5145e7d3737eSFrederic Weisbecker 
5146597af815SSteven Rostedt 	if (unlikely(!ftrace_graph_active))
51472aad1b76SSteven Rostedt 		goto out;
51482aad1b76SSteven Rostedt 
5149597af815SSteven Rostedt 	ftrace_graph_active--;
5150287b6e68SFrederic Weisbecker 	ftrace_graph_return = (trace_func_graph_ret_t)ftrace_stub;
5151e49dc19cSSteven Rostedt 	ftrace_graph_entry = ftrace_graph_entry_stub;
515223a8e844SSteven Rostedt (Red Hat) 	__ftrace_graph_entry = ftrace_graph_entry_stub;
51538a56d776SSteven Rostedt (Red Hat) 	ftrace_shutdown(&fgraph_ops, FTRACE_STOP_FUNC_RET);
51544a2b8ddaSFrederic Weisbecker 	unregister_pm_notifier(&ftrace_suspend_notifier);
515538516ab5SSteven Rostedt 	unregister_trace_sched_switch(ftrace_graph_probe_sched_switch, NULL);
5156e7d3737eSFrederic Weisbecker 
51572aad1b76SSteven Rostedt  out:
5158e6ea44e9SSteven Rostedt 	mutex_unlock(&ftrace_lock);
515915e6cb36SFrederic Weisbecker }
5160f201ae23SFrederic Weisbecker 
5161868baf07SSteven Rostedt static DEFINE_PER_CPU(struct ftrace_ret_stack *, idle_ret_stack);
5162868baf07SSteven Rostedt 
5163868baf07SSteven Rostedt static void
5164868baf07SSteven Rostedt graph_init_task(struct task_struct *t, struct ftrace_ret_stack *ret_stack)
5165868baf07SSteven Rostedt {
5166868baf07SSteven Rostedt 	atomic_set(&t->tracing_graph_pause, 0);
5167868baf07SSteven Rostedt 	atomic_set(&t->trace_overrun, 0);
5168868baf07SSteven Rostedt 	t->ftrace_timestamp = 0;
516925985edcSLucas De Marchi 	/* make curr_ret_stack visible before we add the ret_stack */
5170868baf07SSteven Rostedt 	smp_wmb();
5171868baf07SSteven Rostedt 	t->ret_stack = ret_stack;
5172868baf07SSteven Rostedt }
5173868baf07SSteven Rostedt 
5174868baf07SSteven Rostedt /*
5175868baf07SSteven Rostedt  * Allocate a return stack for the idle task. May be the first
5176868baf07SSteven Rostedt  * time through, or it may be done by CPU hotplug online.
5177868baf07SSteven Rostedt  */
5178868baf07SSteven Rostedt void ftrace_graph_init_idle_task(struct task_struct *t, int cpu)
5179868baf07SSteven Rostedt {
5180868baf07SSteven Rostedt 	t->curr_ret_stack = -1;
5181868baf07SSteven Rostedt 	/*
5182868baf07SSteven Rostedt 	 * The idle task has no parent, it either has its own
5183868baf07SSteven Rostedt 	 * stack or no stack at all.
5184868baf07SSteven Rostedt 	 */
5185868baf07SSteven Rostedt 	if (t->ret_stack)
5186868baf07SSteven Rostedt 		WARN_ON(t->ret_stack != per_cpu(idle_ret_stack, cpu));
5187868baf07SSteven Rostedt 
5188868baf07SSteven Rostedt 	if (ftrace_graph_active) {
5189868baf07SSteven Rostedt 		struct ftrace_ret_stack *ret_stack;
5190868baf07SSteven Rostedt 
5191868baf07SSteven Rostedt 		ret_stack = per_cpu(idle_ret_stack, cpu);
5192868baf07SSteven Rostedt 		if (!ret_stack) {
5193868baf07SSteven Rostedt 			ret_stack = kmalloc(FTRACE_RETFUNC_DEPTH
5194868baf07SSteven Rostedt 					    * sizeof(struct ftrace_ret_stack),
5195868baf07SSteven Rostedt 					    GFP_KERNEL);
5196868baf07SSteven Rostedt 			if (!ret_stack)
5197868baf07SSteven Rostedt 				return;
5198868baf07SSteven Rostedt 			per_cpu(idle_ret_stack, cpu) = ret_stack;
5199868baf07SSteven Rostedt 		}
5200868baf07SSteven Rostedt 		graph_init_task(t, ret_stack);
5201868baf07SSteven Rostedt 	}
5202868baf07SSteven Rostedt }
5203868baf07SSteven Rostedt 
5204f201ae23SFrederic Weisbecker /* Allocate a return stack for newly created task */
5205fb52607aSFrederic Weisbecker void ftrace_graph_init_task(struct task_struct *t)
5206f201ae23SFrederic Weisbecker {
520784047e36SSteven Rostedt 	/* Make sure we do not use the parent ret_stack */
520884047e36SSteven Rostedt 	t->ret_stack = NULL;
5209ea14eb71SSteven Rostedt 	t->curr_ret_stack = -1;
521084047e36SSteven Rostedt 
5211597af815SSteven Rostedt 	if (ftrace_graph_active) {
521282310a32SSteven Rostedt 		struct ftrace_ret_stack *ret_stack;
521382310a32SSteven Rostedt 
521482310a32SSteven Rostedt 		ret_stack = kmalloc(FTRACE_RETFUNC_DEPTH
5215f201ae23SFrederic Weisbecker 				* sizeof(struct ftrace_ret_stack),
5216f201ae23SFrederic Weisbecker 				GFP_KERNEL);
521782310a32SSteven Rostedt 		if (!ret_stack)
5218f201ae23SFrederic Weisbecker 			return;
5219868baf07SSteven Rostedt 		graph_init_task(t, ret_stack);
522084047e36SSteven Rostedt 	}
5221f201ae23SFrederic Weisbecker }
5222f201ae23SFrederic Weisbecker 
5223fb52607aSFrederic Weisbecker void ftrace_graph_exit_task(struct task_struct *t)
5224f201ae23SFrederic Weisbecker {
5225eae849caSFrederic Weisbecker 	struct ftrace_ret_stack	*ret_stack = t->ret_stack;
5226eae849caSFrederic Weisbecker 
5227f201ae23SFrederic Weisbecker 	t->ret_stack = NULL;
5228eae849caSFrederic Weisbecker 	/* NULL must become visible to IRQs before we free it: */
5229eae849caSFrederic Weisbecker 	barrier();
5230eae849caSFrederic Weisbecker 
5231eae849caSFrederic Weisbecker 	kfree(ret_stack);
5232f201ae23SFrederic Weisbecker }
523314a866c5SSteven Rostedt 
523414a866c5SSteven Rostedt void ftrace_graph_stop(void)
523514a866c5SSteven Rostedt {
523614a866c5SSteven Rostedt 	ftrace_stop();
523714a866c5SSteven Rostedt }
523815e6cb36SFrederic Weisbecker #endif
5239