• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * event tracer
3  *
4  * Copyright (C) 2008 Red Hat Inc, Steven Rostedt <srostedt@redhat.com>
5  *
6  *  - Added format output of fields of the trace point.
7  *    This was based off of work by Tom Zanussi <tzanussi@gmail.com>.
8  *
9  */
10 
11 #define pr_fmt(fmt) fmt
12 
13 #include <linux/workqueue.h>
14 #include <linux/spinlock.h>
15 #include <linux/kthread.h>
16 #include <linux/tracefs.h>
17 #include <linux/uaccess.h>
18 #include <linux/module.h>
19 #include <linux/ctype.h>
20 #include <linux/sort.h>
21 #include <linux/slab.h>
22 #include <linux/delay.h>
23 
24 #include <trace/events/sched.h>
25 
26 #include <asm/setup.h>
27 
28 #include "trace_output.h"
29 
30 #undef TRACE_SYSTEM
31 #define TRACE_SYSTEM "TRACE_SYSTEM"
32 
33 DEFINE_MUTEX(event_mutex);
34 
35 LIST_HEAD(ftrace_events);
36 static LIST_HEAD(ftrace_generic_fields);
37 static LIST_HEAD(ftrace_common_fields);
38 
39 #define GFP_TRACE (GFP_KERNEL | __GFP_ZERO)
40 
41 static struct kmem_cache *field_cachep;
42 static struct kmem_cache *file_cachep;
43 
system_refcount(struct event_subsystem * system)44 static inline int system_refcount(struct event_subsystem *system)
45 {
46 	return system->ref_count;
47 }
48 
system_refcount_inc(struct event_subsystem * system)49 static int system_refcount_inc(struct event_subsystem *system)
50 {
51 	return system->ref_count++;
52 }
53 
system_refcount_dec(struct event_subsystem * system)54 static int system_refcount_dec(struct event_subsystem *system)
55 {
56 	return --system->ref_count;
57 }
58 
59 /* Double loops, do not use break, only goto's work */
60 #define do_for_each_event_file(tr, file)			\
61 	list_for_each_entry(tr, &ftrace_trace_arrays, list) {	\
62 		list_for_each_entry(file, &tr->events, list)
63 
64 #define do_for_each_event_file_safe(tr, file)			\
65 	list_for_each_entry(tr, &ftrace_trace_arrays, list) {	\
66 		struct trace_event_file *___n;				\
67 		list_for_each_entry_safe(file, ___n, &tr->events, list)
68 
69 #define while_for_each_event_file()		\
70 	}
71 
72 static struct list_head *
trace_get_fields(struct trace_event_call * event_call)73 trace_get_fields(struct trace_event_call *event_call)
74 {
75 	if (!event_call->class->get_fields)
76 		return &event_call->class->fields;
77 	return event_call->class->get_fields(event_call);
78 }
79 
80 static struct ftrace_event_field *
__find_event_field(struct list_head * head,char * name)81 __find_event_field(struct list_head *head, char *name)
82 {
83 	struct ftrace_event_field *field;
84 
85 	list_for_each_entry(field, head, link) {
86 		if (!strcmp(field->name, name))
87 			return field;
88 	}
89 
90 	return NULL;
91 }
92 
93 struct ftrace_event_field *
trace_find_event_field(struct trace_event_call * call,char * name)94 trace_find_event_field(struct trace_event_call *call, char *name)
95 {
96 	struct ftrace_event_field *field;
97 	struct list_head *head;
98 
99 	head = trace_get_fields(call);
100 	field = __find_event_field(head, name);
101 	if (field)
102 		return field;
103 
104 	field = __find_event_field(&ftrace_generic_fields, name);
105 	if (field)
106 		return field;
107 
108 	return __find_event_field(&ftrace_common_fields, name);
109 }
110 
__trace_define_field(struct list_head * head,const char * type,const char * name,int offset,int size,int is_signed,int filter_type)111 static int __trace_define_field(struct list_head *head, const char *type,
112 				const char *name, int offset, int size,
113 				int is_signed, int filter_type)
114 {
115 	struct ftrace_event_field *field;
116 
117 	field = kmem_cache_alloc(field_cachep, GFP_TRACE);
118 	if (!field)
119 		return -ENOMEM;
120 
121 	field->name = name;
122 	field->type = type;
123 
124 	if (filter_type == FILTER_OTHER)
125 		field->filter_type = filter_assign_type(type);
126 	else
127 		field->filter_type = filter_type;
128 
129 	field->offset = offset;
130 	field->size = size;
131 	field->is_signed = is_signed;
132 
133 	list_add(&field->link, head);
134 
135 	return 0;
136 }
137 
trace_define_field(struct trace_event_call * call,const char * type,const char * name,int offset,int size,int is_signed,int filter_type)138 int trace_define_field(struct trace_event_call *call, const char *type,
139 		       const char *name, int offset, int size, int is_signed,
140 		       int filter_type)
141 {
142 	struct list_head *head;
143 
144 	if (WARN_ON(!call->class))
145 		return 0;
146 
147 	head = trace_get_fields(call);
148 	return __trace_define_field(head, type, name, offset, size,
149 				    is_signed, filter_type);
150 }
151 EXPORT_SYMBOL_GPL(trace_define_field);
152 
153 #define __generic_field(type, item, filter_type)			\
154 	ret = __trace_define_field(&ftrace_generic_fields, #type,	\
155 				   #item, 0, 0, is_signed_type(type),	\
156 				   filter_type);			\
157 	if (ret)							\
158 		return ret;
159 
160 #define __common_field(type, item)					\
161 	ret = __trace_define_field(&ftrace_common_fields, #type,	\
162 				   "common_" #item,			\
163 				   offsetof(typeof(ent), item),		\
164 				   sizeof(ent.item),			\
165 				   is_signed_type(type), FILTER_OTHER);	\
166 	if (ret)							\
167 		return ret;
168 
trace_define_generic_fields(void)169 static int trace_define_generic_fields(void)
170 {
171 	int ret;
172 
173 	__generic_field(int, CPU, FILTER_CPU);
174 	__generic_field(int, cpu, FILTER_CPU);
175 	__generic_field(char *, COMM, FILTER_COMM);
176 	__generic_field(char *, comm, FILTER_COMM);
177 
178 	return ret;
179 }
180 
trace_define_common_fields(void)181 static int trace_define_common_fields(void)
182 {
183 	int ret;
184 	struct trace_entry ent;
185 
186 	__common_field(unsigned short, type);
187 	__common_field(unsigned char, flags);
188 	__common_field(unsigned char, preempt_count);
189 	__common_field(int, pid);
190 
191 	return ret;
192 }
193 
trace_destroy_fields(struct trace_event_call * call)194 static void trace_destroy_fields(struct trace_event_call *call)
195 {
196 	struct ftrace_event_field *field, *next;
197 	struct list_head *head;
198 
199 	head = trace_get_fields(call);
200 	list_for_each_entry_safe(field, next, head, link) {
201 		list_del(&field->link);
202 		kmem_cache_free(field_cachep, field);
203 	}
204 }
205 
206 /*
207  * run-time version of trace_event_get_offsets_<call>() that returns the last
208  * accessible offset of trace fields excluding __dynamic_array bytes
209  */
trace_event_get_offsets(struct trace_event_call * call)210 int trace_event_get_offsets(struct trace_event_call *call)
211 {
212 	struct ftrace_event_field *tail;
213 	struct list_head *head;
214 
215 	head = trace_get_fields(call);
216 	/*
217 	 * head->next points to the last field with the largest offset,
218 	 * since it was added last by trace_define_field()
219 	 */
220 	tail = list_first_entry(head, struct ftrace_event_field, link);
221 	return tail->offset + tail->size;
222 }
223 
trace_event_raw_init(struct trace_event_call * call)224 int trace_event_raw_init(struct trace_event_call *call)
225 {
226 	int id;
227 
228 	id = register_trace_event(&call->event);
229 	if (!id)
230 		return -ENODEV;
231 
232 	return 0;
233 }
234 EXPORT_SYMBOL_GPL(trace_event_raw_init);
235 
trace_event_ignore_this_pid(struct trace_event_file * trace_file)236 bool trace_event_ignore_this_pid(struct trace_event_file *trace_file)
237 {
238 	struct trace_array *tr = trace_file->tr;
239 	struct trace_array_cpu *data;
240 	struct trace_pid_list *pid_list;
241 
242 	pid_list = rcu_dereference_sched(tr->filtered_pids);
243 	if (!pid_list)
244 		return false;
245 
246 	data = this_cpu_ptr(tr->trace_buffer.data);
247 
248 	return data->ignore_pid;
249 }
250 EXPORT_SYMBOL_GPL(trace_event_ignore_this_pid);
251 
trace_event_buffer_reserve(struct trace_event_buffer * fbuffer,struct trace_event_file * trace_file,unsigned long len)252 void *trace_event_buffer_reserve(struct trace_event_buffer *fbuffer,
253 				 struct trace_event_file *trace_file,
254 				 unsigned long len)
255 {
256 	struct trace_event_call *event_call = trace_file->event_call;
257 
258 	if ((trace_file->flags & EVENT_FILE_FL_PID_FILTER) &&
259 	    trace_event_ignore_this_pid(trace_file))
260 		return NULL;
261 
262 	local_save_flags(fbuffer->flags);
263 	fbuffer->pc = preempt_count();
264 	/*
265 	 * If CONFIG_PREEMPT is enabled, then the tracepoint itself disables
266 	 * preemption (adding one to the preempt_count). Since we are
267 	 * interested in the preempt_count at the time the tracepoint was
268 	 * hit, we need to subtract one to offset the increment.
269 	 */
270 	if (IS_ENABLED(CONFIG_PREEMPT))
271 		fbuffer->pc--;
272 	fbuffer->trace_file = trace_file;
273 
274 	fbuffer->event =
275 		trace_event_buffer_lock_reserve(&fbuffer->buffer, trace_file,
276 						event_call->event.type, len,
277 						fbuffer->flags, fbuffer->pc);
278 	if (!fbuffer->event)
279 		return NULL;
280 
281 	fbuffer->entry = ring_buffer_event_data(fbuffer->event);
282 	return fbuffer->entry;
283 }
284 EXPORT_SYMBOL_GPL(trace_event_buffer_reserve);
285 
trace_event_reg(struct trace_event_call * call,enum trace_reg type,void * data)286 int trace_event_reg(struct trace_event_call *call,
287 		    enum trace_reg type, void *data)
288 {
289 	struct trace_event_file *file = data;
290 
291 	WARN_ON(!(call->flags & TRACE_EVENT_FL_TRACEPOINT));
292 	switch (type) {
293 	case TRACE_REG_REGISTER:
294 		return tracepoint_probe_register(call->tp,
295 						 call->class->probe,
296 						 file);
297 	case TRACE_REG_UNREGISTER:
298 		tracepoint_probe_unregister(call->tp,
299 					    call->class->probe,
300 					    file);
301 		return 0;
302 
303 #ifdef CONFIG_PERF_EVENTS
304 	case TRACE_REG_PERF_REGISTER:
305 		return tracepoint_probe_register(call->tp,
306 						 call->class->perf_probe,
307 						 call);
308 	case TRACE_REG_PERF_UNREGISTER:
309 		tracepoint_probe_unregister(call->tp,
310 					    call->class->perf_probe,
311 					    call);
312 		return 0;
313 	case TRACE_REG_PERF_OPEN:
314 	case TRACE_REG_PERF_CLOSE:
315 	case TRACE_REG_PERF_ADD:
316 	case TRACE_REG_PERF_DEL:
317 		return 0;
318 #endif
319 	}
320 	return 0;
321 }
322 EXPORT_SYMBOL_GPL(trace_event_reg);
323 
trace_event_enable_cmd_record(bool enable)324 void trace_event_enable_cmd_record(bool enable)
325 {
326 	struct trace_event_file *file;
327 	struct trace_array *tr;
328 
329 	lockdep_assert_held(&event_mutex);
330 
331 	do_for_each_event_file(tr, file) {
332 
333 		if (!(file->flags & EVENT_FILE_FL_ENABLED))
334 			continue;
335 
336 		if (enable) {
337 			tracing_start_cmdline_record();
338 			set_bit(EVENT_FILE_FL_RECORDED_CMD_BIT, &file->flags);
339 		} else {
340 			tracing_stop_cmdline_record();
341 			clear_bit(EVENT_FILE_FL_RECORDED_CMD_BIT, &file->flags);
342 		}
343 	} while_for_each_event_file();
344 }
345 
trace_event_enable_tgid_record(bool enable)346 void trace_event_enable_tgid_record(bool enable)
347 {
348 	struct trace_event_file *file;
349 	struct trace_array *tr;
350 
351 	lockdep_assert_held(&event_mutex);
352 
353 	do_for_each_event_file(tr, file) {
354 		if (!(file->flags & EVENT_FILE_FL_ENABLED))
355 			continue;
356 
357 		if (enable) {
358 			tracing_start_tgid_record();
359 			set_bit(EVENT_FILE_FL_RECORDED_TGID_BIT, &file->flags);
360 		} else {
361 			tracing_stop_tgid_record();
362 			clear_bit(EVENT_FILE_FL_RECORDED_TGID_BIT,
363 				  &file->flags);
364 		}
365 	} while_for_each_event_file();
366 }
367 
__ftrace_event_enable_disable(struct trace_event_file * file,int enable,int soft_disable)368 static int __ftrace_event_enable_disable(struct trace_event_file *file,
369 					 int enable, int soft_disable)
370 {
371 	struct trace_event_call *call = file->event_call;
372 	struct trace_array *tr = file->tr;
373 	unsigned long file_flags = file->flags;
374 	int ret = 0;
375 	int disable;
376 
377 	switch (enable) {
378 	case 0:
379 		/*
380 		 * When soft_disable is set and enable is cleared, the sm_ref
381 		 * reference counter is decremented. If it reaches 0, we want
382 		 * to clear the SOFT_DISABLED flag but leave the event in the
383 		 * state that it was. That is, if the event was enabled and
384 		 * SOFT_DISABLED isn't set, then do nothing. But if SOFT_DISABLED
385 		 * is set we do not want the event to be enabled before we
386 		 * clear the bit.
387 		 *
388 		 * When soft_disable is not set but the SOFT_MODE flag is,
389 		 * we do nothing. Do not disable the tracepoint, otherwise
390 		 * "soft enable"s (clearing the SOFT_DISABLED bit) wont work.
391 		 */
392 		if (soft_disable) {
393 			if (atomic_dec_return(&file->sm_ref) > 0)
394 				break;
395 			disable = file->flags & EVENT_FILE_FL_SOFT_DISABLED;
396 			clear_bit(EVENT_FILE_FL_SOFT_MODE_BIT, &file->flags);
397 		} else
398 			disable = !(file->flags & EVENT_FILE_FL_SOFT_MODE);
399 
400 		if (disable && (file->flags & EVENT_FILE_FL_ENABLED)) {
401 			clear_bit(EVENT_FILE_FL_ENABLED_BIT, &file->flags);
402 			if (file->flags & EVENT_FILE_FL_RECORDED_CMD) {
403 				tracing_stop_cmdline_record();
404 				clear_bit(EVENT_FILE_FL_RECORDED_CMD_BIT, &file->flags);
405 			}
406 
407 			if (file->flags & EVENT_FILE_FL_RECORDED_TGID) {
408 				tracing_stop_tgid_record();
409 				clear_bit(EVENT_FILE_FL_RECORDED_TGID_BIT, &file->flags);
410 			}
411 
412 			call->class->reg(call, TRACE_REG_UNREGISTER, file);
413 		}
414 		/* If in SOFT_MODE, just set the SOFT_DISABLE_BIT, else clear it */
415 		if (file->flags & EVENT_FILE_FL_SOFT_MODE)
416 			set_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &file->flags);
417 		else
418 			clear_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &file->flags);
419 		break;
420 	case 1:
421 		/*
422 		 * When soft_disable is set and enable is set, we want to
423 		 * register the tracepoint for the event, but leave the event
424 		 * as is. That means, if the event was already enabled, we do
425 		 * nothing (but set SOFT_MODE). If the event is disabled, we
426 		 * set SOFT_DISABLED before enabling the event tracepoint, so
427 		 * it still seems to be disabled.
428 		 */
429 		if (!soft_disable)
430 			clear_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &file->flags);
431 		else {
432 			if (atomic_inc_return(&file->sm_ref) > 1)
433 				break;
434 			set_bit(EVENT_FILE_FL_SOFT_MODE_BIT, &file->flags);
435 		}
436 
437 		if (!(file->flags & EVENT_FILE_FL_ENABLED)) {
438 			bool cmd = false, tgid = false;
439 
440 			/* Keep the event disabled, when going to SOFT_MODE. */
441 			if (soft_disable)
442 				set_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &file->flags);
443 
444 			if (tr->trace_flags & TRACE_ITER_RECORD_CMD) {
445 				cmd = true;
446 				tracing_start_cmdline_record();
447 				set_bit(EVENT_FILE_FL_RECORDED_CMD_BIT, &file->flags);
448 			}
449 
450 			if (tr->trace_flags & TRACE_ITER_RECORD_TGID) {
451 				tgid = true;
452 				tracing_start_tgid_record();
453 				set_bit(EVENT_FILE_FL_RECORDED_TGID_BIT, &file->flags);
454 			}
455 
456 			ret = call->class->reg(call, TRACE_REG_REGISTER, file);
457 			if (ret) {
458 				if (cmd)
459 					tracing_stop_cmdline_record();
460 				if (tgid)
461 					tracing_stop_tgid_record();
462 				pr_info("event trace: Could not enable event "
463 					"%s\n", trace_event_name(call));
464 				break;
465 			}
466 			set_bit(EVENT_FILE_FL_ENABLED_BIT, &file->flags);
467 
468 			/* WAS_ENABLED gets set but never cleared. */
469 			set_bit(EVENT_FILE_FL_WAS_ENABLED_BIT, &file->flags);
470 		}
471 		break;
472 	}
473 
474 	/* Enable or disable use of trace_buffered_event */
475 	if ((file_flags & EVENT_FILE_FL_SOFT_DISABLED) !=
476 	    (file->flags & EVENT_FILE_FL_SOFT_DISABLED)) {
477 		if (file->flags & EVENT_FILE_FL_SOFT_DISABLED)
478 			trace_buffered_event_enable();
479 		else
480 			trace_buffered_event_disable();
481 	}
482 
483 	return ret;
484 }
485 
trace_event_enable_disable(struct trace_event_file * file,int enable,int soft_disable)486 int trace_event_enable_disable(struct trace_event_file *file,
487 			       int enable, int soft_disable)
488 {
489 	return __ftrace_event_enable_disable(file, enable, soft_disable);
490 }
491 
ftrace_event_enable_disable(struct trace_event_file * file,int enable)492 static int ftrace_event_enable_disable(struct trace_event_file *file,
493 				       int enable)
494 {
495 	return __ftrace_event_enable_disable(file, enable, 0);
496 }
497 
ftrace_clear_events(struct trace_array * tr)498 static void ftrace_clear_events(struct trace_array *tr)
499 {
500 	struct trace_event_file *file;
501 
502 	mutex_lock(&event_mutex);
503 	list_for_each_entry(file, &tr->events, list) {
504 		ftrace_event_enable_disable(file, 0);
505 	}
506 	mutex_unlock(&event_mutex);
507 }
508 
509 static void
event_filter_pid_sched_process_exit(void * data,struct task_struct * task)510 event_filter_pid_sched_process_exit(void *data, struct task_struct *task)
511 {
512 	struct trace_pid_list *pid_list;
513 	struct trace_array *tr = data;
514 
515 	pid_list = rcu_dereference_sched(tr->filtered_pids);
516 	trace_filter_add_remove_task(pid_list, NULL, task);
517 }
518 
519 static void
event_filter_pid_sched_process_fork(void * data,struct task_struct * self,struct task_struct * task)520 event_filter_pid_sched_process_fork(void *data,
521 				    struct task_struct *self,
522 				    struct task_struct *task)
523 {
524 	struct trace_pid_list *pid_list;
525 	struct trace_array *tr = data;
526 
527 	pid_list = rcu_dereference_sched(tr->filtered_pids);
528 	trace_filter_add_remove_task(pid_list, self, task);
529 }
530 
trace_event_follow_fork(struct trace_array * tr,bool enable)531 void trace_event_follow_fork(struct trace_array *tr, bool enable)
532 {
533 	if (enable) {
534 		register_trace_prio_sched_process_fork(event_filter_pid_sched_process_fork,
535 						       tr, INT_MIN);
536 		register_trace_prio_sched_process_exit(event_filter_pid_sched_process_exit,
537 						       tr, INT_MAX);
538 	} else {
539 		unregister_trace_sched_process_fork(event_filter_pid_sched_process_fork,
540 						    tr);
541 		unregister_trace_sched_process_exit(event_filter_pid_sched_process_exit,
542 						    tr);
543 	}
544 }
545 
546 static void
event_filter_pid_sched_switch_probe_pre(void * data,bool preempt,struct task_struct * prev,struct task_struct * next)547 event_filter_pid_sched_switch_probe_pre(void *data, bool preempt,
548 		    struct task_struct *prev, struct task_struct *next)
549 {
550 	struct trace_array *tr = data;
551 	struct trace_pid_list *pid_list;
552 
553 	pid_list = rcu_dereference_sched(tr->filtered_pids);
554 
555 	this_cpu_write(tr->trace_buffer.data->ignore_pid,
556 		       trace_ignore_this_task(pid_list, prev) &&
557 		       trace_ignore_this_task(pid_list, next));
558 }
559 
560 static void
event_filter_pid_sched_switch_probe_post(void * data,bool preempt,struct task_struct * prev,struct task_struct * next)561 event_filter_pid_sched_switch_probe_post(void *data, bool preempt,
562 		    struct task_struct *prev, struct task_struct *next)
563 {
564 	struct trace_array *tr = data;
565 	struct trace_pid_list *pid_list;
566 
567 	pid_list = rcu_dereference_sched(tr->filtered_pids);
568 
569 	this_cpu_write(tr->trace_buffer.data->ignore_pid,
570 		       trace_ignore_this_task(pid_list, next));
571 }
572 
573 static void
event_filter_pid_sched_wakeup_probe_pre(void * data,struct task_struct * task)574 event_filter_pid_sched_wakeup_probe_pre(void *data, struct task_struct *task)
575 {
576 	struct trace_array *tr = data;
577 	struct trace_pid_list *pid_list;
578 
579 	/* Nothing to do if we are already tracing */
580 	if (!this_cpu_read(tr->trace_buffer.data->ignore_pid))
581 		return;
582 
583 	pid_list = rcu_dereference_sched(tr->filtered_pids);
584 
585 	this_cpu_write(tr->trace_buffer.data->ignore_pid,
586 		       trace_ignore_this_task(pid_list, task));
587 }
588 
589 static void
event_filter_pid_sched_wakeup_probe_post(void * data,struct task_struct * task)590 event_filter_pid_sched_wakeup_probe_post(void *data, struct task_struct *task)
591 {
592 	struct trace_array *tr = data;
593 	struct trace_pid_list *pid_list;
594 
595 	/* Nothing to do if we are not tracing */
596 	if (this_cpu_read(tr->trace_buffer.data->ignore_pid))
597 		return;
598 
599 	pid_list = rcu_dereference_sched(tr->filtered_pids);
600 
601 	/* Set tracing if current is enabled */
602 	this_cpu_write(tr->trace_buffer.data->ignore_pid,
603 		       trace_ignore_this_task(pid_list, current));
604 }
605 
__ftrace_clear_event_pids(struct trace_array * tr)606 static void __ftrace_clear_event_pids(struct trace_array *tr)
607 {
608 	struct trace_pid_list *pid_list;
609 	struct trace_event_file *file;
610 	int cpu;
611 
612 	pid_list = rcu_dereference_protected(tr->filtered_pids,
613 					     lockdep_is_held(&event_mutex));
614 	if (!pid_list)
615 		return;
616 
617 	unregister_trace_sched_switch(event_filter_pid_sched_switch_probe_pre, tr);
618 	unregister_trace_sched_switch(event_filter_pid_sched_switch_probe_post, tr);
619 
620 	unregister_trace_sched_wakeup(event_filter_pid_sched_wakeup_probe_pre, tr);
621 	unregister_trace_sched_wakeup(event_filter_pid_sched_wakeup_probe_post, tr);
622 
623 	unregister_trace_sched_wakeup_new(event_filter_pid_sched_wakeup_probe_pre, tr);
624 	unregister_trace_sched_wakeup_new(event_filter_pid_sched_wakeup_probe_post, tr);
625 
626 	unregister_trace_sched_waking(event_filter_pid_sched_wakeup_probe_pre, tr);
627 	unregister_trace_sched_waking(event_filter_pid_sched_wakeup_probe_post, tr);
628 
629 	list_for_each_entry(file, &tr->events, list) {
630 		clear_bit(EVENT_FILE_FL_PID_FILTER_BIT, &file->flags);
631 	}
632 
633 	for_each_possible_cpu(cpu)
634 		per_cpu_ptr(tr->trace_buffer.data, cpu)->ignore_pid = false;
635 
636 	rcu_assign_pointer(tr->filtered_pids, NULL);
637 
638 	/* Wait till all users are no longer using pid filtering */
639 	synchronize_sched();
640 
641 	trace_free_pid_list(pid_list);
642 }
643 
ftrace_clear_event_pids(struct trace_array * tr)644 static void ftrace_clear_event_pids(struct trace_array *tr)
645 {
646 	mutex_lock(&event_mutex);
647 	__ftrace_clear_event_pids(tr);
648 	mutex_unlock(&event_mutex);
649 }
650 
__put_system(struct event_subsystem * system)651 static void __put_system(struct event_subsystem *system)
652 {
653 	struct event_filter *filter = system->filter;
654 
655 	WARN_ON_ONCE(system_refcount(system) == 0);
656 	if (system_refcount_dec(system))
657 		return;
658 
659 	list_del(&system->list);
660 
661 	if (filter) {
662 		kfree(filter->filter_string);
663 		kfree(filter);
664 	}
665 	kfree_const(system->name);
666 	kfree(system);
667 }
668 
__get_system(struct event_subsystem * system)669 static void __get_system(struct event_subsystem *system)
670 {
671 	WARN_ON_ONCE(system_refcount(system) == 0);
672 	system_refcount_inc(system);
673 }
674 
__get_system_dir(struct trace_subsystem_dir * dir)675 static void __get_system_dir(struct trace_subsystem_dir *dir)
676 {
677 	WARN_ON_ONCE(dir->ref_count == 0);
678 	dir->ref_count++;
679 	__get_system(dir->subsystem);
680 }
681 
__put_system_dir(struct trace_subsystem_dir * dir)682 static void __put_system_dir(struct trace_subsystem_dir *dir)
683 {
684 	WARN_ON_ONCE(dir->ref_count == 0);
685 	/* If the subsystem is about to be freed, the dir must be too */
686 	WARN_ON_ONCE(system_refcount(dir->subsystem) == 1 && dir->ref_count != 1);
687 
688 	__put_system(dir->subsystem);
689 	if (!--dir->ref_count)
690 		kfree(dir);
691 }
692 
put_system(struct trace_subsystem_dir * dir)693 static void put_system(struct trace_subsystem_dir *dir)
694 {
695 	mutex_lock(&event_mutex);
696 	__put_system_dir(dir);
697 	mutex_unlock(&event_mutex);
698 }
699 
remove_subsystem(struct trace_subsystem_dir * dir)700 static void remove_subsystem(struct trace_subsystem_dir *dir)
701 {
702 	if (!dir)
703 		return;
704 
705 	if (!--dir->nr_events) {
706 		tracefs_remove_recursive(dir->entry);
707 		list_del(&dir->list);
708 		__put_system_dir(dir);
709 	}
710 }
711 
remove_event_file_dir(struct trace_event_file * file)712 static void remove_event_file_dir(struct trace_event_file *file)
713 {
714 	struct dentry *dir = file->dir;
715 	struct dentry *child;
716 
717 	if (dir) {
718 		spin_lock(&dir->d_lock);	/* probably unneeded */
719 		list_for_each_entry(child, &dir->d_subdirs, d_child) {
720 			if (d_really_is_positive(child))	/* probably unneeded */
721 				d_inode(child)->i_private = NULL;
722 		}
723 		spin_unlock(&dir->d_lock);
724 
725 		tracefs_remove_recursive(dir);
726 	}
727 
728 	list_del(&file->list);
729 	remove_subsystem(file->system);
730 	free_event_filter(file->filter);
731 	kmem_cache_free(file_cachep, file);
732 }
733 
734 /*
735  * __ftrace_set_clr_event(NULL, NULL, NULL, set) will set/unset all events.
736  */
737 static int
__ftrace_set_clr_event_nolock(struct trace_array * tr,const char * match,const char * sub,const char * event,int set)738 __ftrace_set_clr_event_nolock(struct trace_array *tr, const char *match,
739 			      const char *sub, const char *event, int set)
740 {
741 	struct trace_event_file *file;
742 	struct trace_event_call *call;
743 	const char *name;
744 	int ret = -EINVAL;
745 	int eret = 0;
746 
747 	list_for_each_entry(file, &tr->events, list) {
748 
749 		call = file->event_call;
750 		name = trace_event_name(call);
751 
752 		if (!name || !call->class || !call->class->reg)
753 			continue;
754 
755 		if (call->flags & TRACE_EVENT_FL_IGNORE_ENABLE)
756 			continue;
757 
758 		if (match &&
759 		    strcmp(match, name) != 0 &&
760 		    strcmp(match, call->class->system) != 0)
761 			continue;
762 
763 		if (sub && strcmp(sub, call->class->system) != 0)
764 			continue;
765 
766 		if (event && strcmp(event, name) != 0)
767 			continue;
768 
769 		ret = ftrace_event_enable_disable(file, set);
770 
771 		/*
772 		 * Save the first error and return that. Some events
773 		 * may still have been enabled, but let the user
774 		 * know that something went wrong.
775 		 */
776 		if (ret && !eret)
777 			eret = ret;
778 
779 		ret = eret;
780 	}
781 
782 	return ret;
783 }
784 
__ftrace_set_clr_event(struct trace_array * tr,const char * match,const char * sub,const char * event,int set)785 static int __ftrace_set_clr_event(struct trace_array *tr, const char *match,
786 				  const char *sub, const char *event, int set)
787 {
788 	int ret;
789 
790 	mutex_lock(&event_mutex);
791 	ret = __ftrace_set_clr_event_nolock(tr, match, sub, event, set);
792 	mutex_unlock(&event_mutex);
793 
794 	return ret;
795 }
796 
ftrace_set_clr_event(struct trace_array * tr,char * buf,int set)797 static int ftrace_set_clr_event(struct trace_array *tr, char *buf, int set)
798 {
799 	char *event = NULL, *sub = NULL, *match;
800 	int ret;
801 
802 	/*
803 	 * The buf format can be <subsystem>:<event-name>
804 	 *  *:<event-name> means any event by that name.
805 	 *  :<event-name> is the same.
806 	 *
807 	 *  <subsystem>:* means all events in that subsystem
808 	 *  <subsystem>: means the same.
809 	 *
810 	 *  <name> (no ':') means all events in a subsystem with
811 	 *  the name <name> or any event that matches <name>
812 	 */
813 
814 	match = strsep(&buf, ":");
815 	if (buf) {
816 		sub = match;
817 		event = buf;
818 		match = NULL;
819 
820 		if (!strlen(sub) || strcmp(sub, "*") == 0)
821 			sub = NULL;
822 		if (!strlen(event) || strcmp(event, "*") == 0)
823 			event = NULL;
824 	}
825 
826 	ret = __ftrace_set_clr_event(tr, match, sub, event, set);
827 
828 	/* Put back the colon to allow this to be called again */
829 	if (buf)
830 		*(buf - 1) = ':';
831 
832 	return ret;
833 }
834 
835 /**
836  * trace_set_clr_event - enable or disable an event
837  * @system: system name to match (NULL for any system)
838  * @event: event name to match (NULL for all events, within system)
839  * @set: 1 to enable, 0 to disable
840  *
841  * This is a way for other parts of the kernel to enable or disable
842  * event recording.
843  *
844  * Returns 0 on success, -EINVAL if the parameters do not match any
845  * registered events.
846  */
trace_set_clr_event(const char * system,const char * event,int set)847 int trace_set_clr_event(const char *system, const char *event, int set)
848 {
849 	struct trace_array *tr = top_trace_array();
850 
851 	if (!tr)
852 		return -ENODEV;
853 
854 	return __ftrace_set_clr_event(tr, NULL, system, event, set);
855 }
856 EXPORT_SYMBOL_GPL(trace_set_clr_event);
857 
858 /* 128 should be much more than enough */
859 #define EVENT_BUF_SIZE		127
860 
861 static ssize_t
ftrace_event_write(struct file * file,const char __user * ubuf,size_t cnt,loff_t * ppos)862 ftrace_event_write(struct file *file, const char __user *ubuf,
863 		   size_t cnt, loff_t *ppos)
864 {
865 	struct trace_parser parser;
866 	struct seq_file *m = file->private_data;
867 	struct trace_array *tr = m->private;
868 	ssize_t read, ret;
869 
870 	if (!cnt)
871 		return 0;
872 
873 	ret = tracing_update_buffers();
874 	if (ret < 0)
875 		return ret;
876 
877 	if (trace_parser_get_init(&parser, EVENT_BUF_SIZE + 1))
878 		return -ENOMEM;
879 
880 	read = trace_get_user(&parser, ubuf, cnt, ppos);
881 
882 	if (read >= 0 && trace_parser_loaded((&parser))) {
883 		int set = 1;
884 
885 		if (*parser.buffer == '!')
886 			set = 0;
887 
888 		parser.buffer[parser.idx] = 0;
889 
890 		ret = ftrace_set_clr_event(tr, parser.buffer + !set, set);
891 		if (ret)
892 			goto out_put;
893 	}
894 
895 	ret = read;
896 
897  out_put:
898 	trace_parser_put(&parser);
899 
900 	return ret;
901 }
902 
903 static void *
t_next(struct seq_file * m,void * v,loff_t * pos)904 t_next(struct seq_file *m, void *v, loff_t *pos)
905 {
906 	struct trace_event_file *file = v;
907 	struct trace_event_call *call;
908 	struct trace_array *tr = m->private;
909 
910 	(*pos)++;
911 
912 	list_for_each_entry_continue(file, &tr->events, list) {
913 		call = file->event_call;
914 		/*
915 		 * The ftrace subsystem is for showing formats only.
916 		 * They can not be enabled or disabled via the event files.
917 		 */
918 		if (call->class && call->class->reg &&
919 		    !(call->flags & TRACE_EVENT_FL_IGNORE_ENABLE))
920 			return file;
921 	}
922 
923 	return NULL;
924 }
925 
t_start(struct seq_file * m,loff_t * pos)926 static void *t_start(struct seq_file *m, loff_t *pos)
927 {
928 	struct trace_event_file *file;
929 	struct trace_array *tr = m->private;
930 	loff_t l;
931 
932 	mutex_lock(&event_mutex);
933 
934 	file = list_entry(&tr->events, struct trace_event_file, list);
935 	for (l = 0; l <= *pos; ) {
936 		file = t_next(m, file, &l);
937 		if (!file)
938 			break;
939 	}
940 	return file;
941 }
942 
943 static void *
s_next(struct seq_file * m,void * v,loff_t * pos)944 s_next(struct seq_file *m, void *v, loff_t *pos)
945 {
946 	struct trace_event_file *file = v;
947 	struct trace_array *tr = m->private;
948 
949 	(*pos)++;
950 
951 	list_for_each_entry_continue(file, &tr->events, list) {
952 		if (file->flags & EVENT_FILE_FL_ENABLED)
953 			return file;
954 	}
955 
956 	return NULL;
957 }
958 
s_start(struct seq_file * m,loff_t * pos)959 static void *s_start(struct seq_file *m, loff_t *pos)
960 {
961 	struct trace_event_file *file;
962 	struct trace_array *tr = m->private;
963 	loff_t l;
964 
965 	mutex_lock(&event_mutex);
966 
967 	file = list_entry(&tr->events, struct trace_event_file, list);
968 	for (l = 0; l <= *pos; ) {
969 		file = s_next(m, file, &l);
970 		if (!file)
971 			break;
972 	}
973 	return file;
974 }
975 
t_show(struct seq_file * m,void * v)976 static int t_show(struct seq_file *m, void *v)
977 {
978 	struct trace_event_file *file = v;
979 	struct trace_event_call *call = file->event_call;
980 
981 	if (strcmp(call->class->system, TRACE_SYSTEM) != 0)
982 		seq_printf(m, "%s:", call->class->system);
983 	seq_printf(m, "%s\n", trace_event_name(call));
984 
985 	return 0;
986 }
987 
t_stop(struct seq_file * m,void * p)988 static void t_stop(struct seq_file *m, void *p)
989 {
990 	mutex_unlock(&event_mutex);
991 }
992 
993 static void *
p_next(struct seq_file * m,void * v,loff_t * pos)994 p_next(struct seq_file *m, void *v, loff_t *pos)
995 {
996 	struct trace_array *tr = m->private;
997 	struct trace_pid_list *pid_list = rcu_dereference_sched(tr->filtered_pids);
998 
999 	return trace_pid_next(pid_list, v, pos);
1000 }
1001 
p_start(struct seq_file * m,loff_t * pos)1002 static void *p_start(struct seq_file *m, loff_t *pos)
1003 	__acquires(RCU)
1004 {
1005 	struct trace_pid_list *pid_list;
1006 	struct trace_array *tr = m->private;
1007 
1008 	/*
1009 	 * Grab the mutex, to keep calls to p_next() having the same
1010 	 * tr->filtered_pids as p_start() has.
1011 	 * If we just passed the tr->filtered_pids around, then RCU would
1012 	 * have been enough, but doing that makes things more complex.
1013 	 */
1014 	mutex_lock(&event_mutex);
1015 	rcu_read_lock_sched();
1016 
1017 	pid_list = rcu_dereference_sched(tr->filtered_pids);
1018 
1019 	if (!pid_list)
1020 		return NULL;
1021 
1022 	return trace_pid_start(pid_list, pos);
1023 }
1024 
p_stop(struct seq_file * m,void * p)1025 static void p_stop(struct seq_file *m, void *p)
1026 	__releases(RCU)
1027 {
1028 	rcu_read_unlock_sched();
1029 	mutex_unlock(&event_mutex);
1030 }
1031 
1032 static ssize_t
event_enable_read(struct file * filp,char __user * ubuf,size_t cnt,loff_t * ppos)1033 event_enable_read(struct file *filp, char __user *ubuf, size_t cnt,
1034 		  loff_t *ppos)
1035 {
1036 	struct trace_event_file *file;
1037 	unsigned long flags;
1038 	char buf[4] = "0";
1039 
1040 	mutex_lock(&event_mutex);
1041 	file = event_file_data(filp);
1042 	if (likely(file))
1043 		flags = file->flags;
1044 	mutex_unlock(&event_mutex);
1045 
1046 	if (!file)
1047 		return -ENODEV;
1048 
1049 	if (flags & EVENT_FILE_FL_ENABLED &&
1050 	    !(flags & EVENT_FILE_FL_SOFT_DISABLED))
1051 		strcpy(buf, "1");
1052 
1053 	if (flags & EVENT_FILE_FL_SOFT_DISABLED ||
1054 	    flags & EVENT_FILE_FL_SOFT_MODE)
1055 		strcat(buf, "*");
1056 
1057 	strcat(buf, "\n");
1058 
1059 	return simple_read_from_buffer(ubuf, cnt, ppos, buf, strlen(buf));
1060 }
1061 
1062 static ssize_t
event_enable_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos)1063 event_enable_write(struct file *filp, const char __user *ubuf, size_t cnt,
1064 		   loff_t *ppos)
1065 {
1066 	struct trace_event_file *file;
1067 	unsigned long val;
1068 	int ret;
1069 
1070 	ret = kstrtoul_from_user(ubuf, cnt, 10, &val);
1071 	if (ret)
1072 		return ret;
1073 
1074 	ret = tracing_update_buffers();
1075 	if (ret < 0)
1076 		return ret;
1077 
1078 	switch (val) {
1079 	case 0:
1080 	case 1:
1081 		ret = -ENODEV;
1082 		mutex_lock(&event_mutex);
1083 		file = event_file_data(filp);
1084 		if (likely(file))
1085 			ret = ftrace_event_enable_disable(file, val);
1086 		mutex_unlock(&event_mutex);
1087 		break;
1088 
1089 	default:
1090 		return -EINVAL;
1091 	}
1092 
1093 	*ppos += cnt;
1094 
1095 	return ret ? ret : cnt;
1096 }
1097 
1098 static ssize_t
system_enable_read(struct file * filp,char __user * ubuf,size_t cnt,loff_t * ppos)1099 system_enable_read(struct file *filp, char __user *ubuf, size_t cnt,
1100 		   loff_t *ppos)
1101 {
1102 	const char set_to_char[4] = { '?', '0', '1', 'X' };
1103 	struct trace_subsystem_dir *dir = filp->private_data;
1104 	struct event_subsystem *system = dir->subsystem;
1105 	struct trace_event_call *call;
1106 	struct trace_event_file *file;
1107 	struct trace_array *tr = dir->tr;
1108 	char buf[2];
1109 	int set = 0;
1110 	int ret;
1111 
1112 	mutex_lock(&event_mutex);
1113 	list_for_each_entry(file, &tr->events, list) {
1114 		call = file->event_call;
1115 		if (!trace_event_name(call) || !call->class || !call->class->reg)
1116 			continue;
1117 
1118 		if (system && strcmp(call->class->system, system->name) != 0)
1119 			continue;
1120 
1121 		/*
1122 		 * We need to find out if all the events are set
1123 		 * or if all events or cleared, or if we have
1124 		 * a mixture.
1125 		 */
1126 		set |= (1 << !!(file->flags & EVENT_FILE_FL_ENABLED));
1127 
1128 		/*
1129 		 * If we have a mixture, no need to look further.
1130 		 */
1131 		if (set == 3)
1132 			break;
1133 	}
1134 	mutex_unlock(&event_mutex);
1135 
1136 	buf[0] = set_to_char[set];
1137 	buf[1] = '\n';
1138 
1139 	ret = simple_read_from_buffer(ubuf, cnt, ppos, buf, 2);
1140 
1141 	return ret;
1142 }
1143 
1144 static ssize_t
system_enable_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos)1145 system_enable_write(struct file *filp, const char __user *ubuf, size_t cnt,
1146 		    loff_t *ppos)
1147 {
1148 	struct trace_subsystem_dir *dir = filp->private_data;
1149 	struct event_subsystem *system = dir->subsystem;
1150 	const char *name = NULL;
1151 	unsigned long val;
1152 	ssize_t ret;
1153 
1154 	ret = kstrtoul_from_user(ubuf, cnt, 10, &val);
1155 	if (ret)
1156 		return ret;
1157 
1158 	ret = tracing_update_buffers();
1159 	if (ret < 0)
1160 		return ret;
1161 
1162 	if (val != 0 && val != 1)
1163 		return -EINVAL;
1164 
1165 	/*
1166 	 * Opening of "enable" adds a ref count to system,
1167 	 * so the name is safe to use.
1168 	 */
1169 	if (system)
1170 		name = system->name;
1171 
1172 	ret = __ftrace_set_clr_event(dir->tr, NULL, name, NULL, val);
1173 	if (ret)
1174 		goto out;
1175 
1176 	ret = cnt;
1177 
1178 out:
1179 	*ppos += cnt;
1180 
1181 	return ret;
1182 }
1183 
1184 enum {
1185 	FORMAT_HEADER		= 1,
1186 	FORMAT_FIELD_SEPERATOR	= 2,
1187 	FORMAT_PRINTFMT		= 3,
1188 };
1189 
f_next(struct seq_file * m,void * v,loff_t * pos)1190 static void *f_next(struct seq_file *m, void *v, loff_t *pos)
1191 {
1192 	struct trace_event_call *call = event_file_data(m->private);
1193 	struct list_head *common_head = &ftrace_common_fields;
1194 	struct list_head *head = trace_get_fields(call);
1195 	struct list_head *node = v;
1196 
1197 	(*pos)++;
1198 
1199 	switch ((unsigned long)v) {
1200 	case FORMAT_HEADER:
1201 		node = common_head;
1202 		break;
1203 
1204 	case FORMAT_FIELD_SEPERATOR:
1205 		node = head;
1206 		break;
1207 
1208 	case FORMAT_PRINTFMT:
1209 		/* all done */
1210 		return NULL;
1211 	}
1212 
1213 	node = node->prev;
1214 	if (node == common_head)
1215 		return (void *)FORMAT_FIELD_SEPERATOR;
1216 	else if (node == head)
1217 		return (void *)FORMAT_PRINTFMT;
1218 	else
1219 		return node;
1220 }
1221 
f_show(struct seq_file * m,void * v)1222 static int f_show(struct seq_file *m, void *v)
1223 {
1224 	struct trace_event_call *call = event_file_data(m->private);
1225 	struct ftrace_event_field *field;
1226 	const char *array_descriptor;
1227 
1228 	switch ((unsigned long)v) {
1229 	case FORMAT_HEADER:
1230 		seq_printf(m, "name: %s\n", trace_event_name(call));
1231 		seq_printf(m, "ID: %d\n", call->event.type);
1232 		seq_puts(m, "format:\n");
1233 		return 0;
1234 
1235 	case FORMAT_FIELD_SEPERATOR:
1236 		seq_putc(m, '\n');
1237 		return 0;
1238 
1239 	case FORMAT_PRINTFMT:
1240 		seq_printf(m, "\nprint fmt: %s\n",
1241 			   call->print_fmt);
1242 		return 0;
1243 	}
1244 
1245 	field = list_entry(v, struct ftrace_event_field, link);
1246 	/*
1247 	 * Smartly shows the array type(except dynamic array).
1248 	 * Normal:
1249 	 *	field:TYPE VAR
1250 	 * If TYPE := TYPE[LEN], it is shown:
1251 	 *	field:TYPE VAR[LEN]
1252 	 */
1253 	array_descriptor = strchr(field->type, '[');
1254 
1255 	if (!strncmp(field->type, "__data_loc", 10))
1256 		array_descriptor = NULL;
1257 
1258 	if (!array_descriptor)
1259 		seq_printf(m, "\tfield:%s %s;\toffset:%u;\tsize:%u;\tsigned:%d;\n",
1260 			   field->type, field->name, field->offset,
1261 			   field->size, !!field->is_signed);
1262 	else
1263 		seq_printf(m, "\tfield:%.*s %s%s;\toffset:%u;\tsize:%u;\tsigned:%d;\n",
1264 			   (int)(array_descriptor - field->type),
1265 			   field->type, field->name,
1266 			   array_descriptor, field->offset,
1267 			   field->size, !!field->is_signed);
1268 
1269 	return 0;
1270 }
1271 
f_start(struct seq_file * m,loff_t * pos)1272 static void *f_start(struct seq_file *m, loff_t *pos)
1273 {
1274 	void *p = (void *)FORMAT_HEADER;
1275 	loff_t l = 0;
1276 
1277 	/* ->stop() is called even if ->start() fails */
1278 	mutex_lock(&event_mutex);
1279 	if (!event_file_data(m->private))
1280 		return ERR_PTR(-ENODEV);
1281 
1282 	while (l < *pos && p)
1283 		p = f_next(m, p, &l);
1284 
1285 	return p;
1286 }
1287 
f_stop(struct seq_file * m,void * p)1288 static void f_stop(struct seq_file *m, void *p)
1289 {
1290 	mutex_unlock(&event_mutex);
1291 }
1292 
1293 static const struct seq_operations trace_format_seq_ops = {
1294 	.start		= f_start,
1295 	.next		= f_next,
1296 	.stop		= f_stop,
1297 	.show		= f_show,
1298 };
1299 
trace_format_open(struct inode * inode,struct file * file)1300 static int trace_format_open(struct inode *inode, struct file *file)
1301 {
1302 	struct seq_file *m;
1303 	int ret;
1304 
1305 	ret = seq_open(file, &trace_format_seq_ops);
1306 	if (ret < 0)
1307 		return ret;
1308 
1309 	m = file->private_data;
1310 	m->private = file;
1311 
1312 	return 0;
1313 }
1314 
1315 static ssize_t
event_id_read(struct file * filp,char __user * ubuf,size_t cnt,loff_t * ppos)1316 event_id_read(struct file *filp, char __user *ubuf, size_t cnt, loff_t *ppos)
1317 {
1318 	int id = (long)event_file_data(filp);
1319 	char buf[32];
1320 	int len;
1321 
1322 	if (unlikely(!id))
1323 		return -ENODEV;
1324 
1325 	len = sprintf(buf, "%d\n", id);
1326 
1327 	return simple_read_from_buffer(ubuf, cnt, ppos, buf, len);
1328 }
1329 
1330 static ssize_t
event_filter_read(struct file * filp,char __user * ubuf,size_t cnt,loff_t * ppos)1331 event_filter_read(struct file *filp, char __user *ubuf, size_t cnt,
1332 		  loff_t *ppos)
1333 {
1334 	struct trace_event_file *file;
1335 	struct trace_seq *s;
1336 	int r = -ENODEV;
1337 
1338 	if (*ppos)
1339 		return 0;
1340 
1341 	s = kmalloc(sizeof(*s), GFP_KERNEL);
1342 
1343 	if (!s)
1344 		return -ENOMEM;
1345 
1346 	trace_seq_init(s);
1347 
1348 	mutex_lock(&event_mutex);
1349 	file = event_file_data(filp);
1350 	if (file)
1351 		print_event_filter(file, s);
1352 	mutex_unlock(&event_mutex);
1353 
1354 	if (file)
1355 		r = simple_read_from_buffer(ubuf, cnt, ppos,
1356 					    s->buffer, trace_seq_used(s));
1357 
1358 	kfree(s);
1359 
1360 	return r;
1361 }
1362 
1363 static ssize_t
event_filter_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos)1364 event_filter_write(struct file *filp, const char __user *ubuf, size_t cnt,
1365 		   loff_t *ppos)
1366 {
1367 	struct trace_event_file *file;
1368 	char *buf;
1369 	int err = -ENODEV;
1370 
1371 	if (cnt >= PAGE_SIZE)
1372 		return -EINVAL;
1373 
1374 	buf = memdup_user_nul(ubuf, cnt);
1375 	if (IS_ERR(buf))
1376 		return PTR_ERR(buf);
1377 
1378 	mutex_lock(&event_mutex);
1379 	file = event_file_data(filp);
1380 	if (file)
1381 		err = apply_event_filter(file, buf);
1382 	mutex_unlock(&event_mutex);
1383 
1384 	kfree(buf);
1385 	if (err < 0)
1386 		return err;
1387 
1388 	*ppos += cnt;
1389 
1390 	return cnt;
1391 }
1392 
1393 static LIST_HEAD(event_subsystems);
1394 
subsystem_open(struct inode * inode,struct file * filp)1395 static int subsystem_open(struct inode *inode, struct file *filp)
1396 {
1397 	struct event_subsystem *system = NULL;
1398 	struct trace_subsystem_dir *dir = NULL; /* Initialize for gcc */
1399 	struct trace_array *tr;
1400 	int ret;
1401 
1402 	if (tracing_is_disabled())
1403 		return -ENODEV;
1404 
1405 	/* Make sure the system still exists */
1406 	mutex_lock(&trace_types_lock);
1407 	mutex_lock(&event_mutex);
1408 	list_for_each_entry(tr, &ftrace_trace_arrays, list) {
1409 		list_for_each_entry(dir, &tr->systems, list) {
1410 			if (dir == inode->i_private) {
1411 				/* Don't open systems with no events */
1412 				if (dir->nr_events) {
1413 					__get_system_dir(dir);
1414 					system = dir->subsystem;
1415 				}
1416 				goto exit_loop;
1417 			}
1418 		}
1419 	}
1420  exit_loop:
1421 	mutex_unlock(&event_mutex);
1422 	mutex_unlock(&trace_types_lock);
1423 
1424 	if (!system)
1425 		return -ENODEV;
1426 
1427 	/* Some versions of gcc think dir can be uninitialized here */
1428 	WARN_ON(!dir);
1429 
1430 	/* Still need to increment the ref count of the system */
1431 	if (trace_array_get(tr) < 0) {
1432 		put_system(dir);
1433 		return -ENODEV;
1434 	}
1435 
1436 	ret = tracing_open_generic(inode, filp);
1437 	if (ret < 0) {
1438 		trace_array_put(tr);
1439 		put_system(dir);
1440 	}
1441 
1442 	return ret;
1443 }
1444 
system_tr_open(struct inode * inode,struct file * filp)1445 static int system_tr_open(struct inode *inode, struct file *filp)
1446 {
1447 	struct trace_subsystem_dir *dir;
1448 	struct trace_array *tr = inode->i_private;
1449 	int ret;
1450 
1451 	if (tracing_is_disabled())
1452 		return -ENODEV;
1453 
1454 	if (trace_array_get(tr) < 0)
1455 		return -ENODEV;
1456 
1457 	/* Make a temporary dir that has no system but points to tr */
1458 	dir = kzalloc(sizeof(*dir), GFP_KERNEL);
1459 	if (!dir) {
1460 		trace_array_put(tr);
1461 		return -ENOMEM;
1462 	}
1463 
1464 	dir->tr = tr;
1465 
1466 	ret = tracing_open_generic(inode, filp);
1467 	if (ret < 0) {
1468 		trace_array_put(tr);
1469 		kfree(dir);
1470 		return ret;
1471 	}
1472 
1473 	filp->private_data = dir;
1474 
1475 	return 0;
1476 }
1477 
subsystem_release(struct inode * inode,struct file * file)1478 static int subsystem_release(struct inode *inode, struct file *file)
1479 {
1480 	struct trace_subsystem_dir *dir = file->private_data;
1481 
1482 	trace_array_put(dir->tr);
1483 
1484 	/*
1485 	 * If dir->subsystem is NULL, then this is a temporary
1486 	 * descriptor that was made for a trace_array to enable
1487 	 * all subsystems.
1488 	 */
1489 	if (dir->subsystem)
1490 		put_system(dir);
1491 	else
1492 		kfree(dir);
1493 
1494 	return 0;
1495 }
1496 
1497 static ssize_t
subsystem_filter_read(struct file * filp,char __user * ubuf,size_t cnt,loff_t * ppos)1498 subsystem_filter_read(struct file *filp, char __user *ubuf, size_t cnt,
1499 		      loff_t *ppos)
1500 {
1501 	struct trace_subsystem_dir *dir = filp->private_data;
1502 	struct event_subsystem *system = dir->subsystem;
1503 	struct trace_seq *s;
1504 	int r;
1505 
1506 	if (*ppos)
1507 		return 0;
1508 
1509 	s = kmalloc(sizeof(*s), GFP_KERNEL);
1510 	if (!s)
1511 		return -ENOMEM;
1512 
1513 	trace_seq_init(s);
1514 
1515 	print_subsystem_event_filter(system, s);
1516 	r = simple_read_from_buffer(ubuf, cnt, ppos,
1517 				    s->buffer, trace_seq_used(s));
1518 
1519 	kfree(s);
1520 
1521 	return r;
1522 }
1523 
1524 static ssize_t
subsystem_filter_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos)1525 subsystem_filter_write(struct file *filp, const char __user *ubuf, size_t cnt,
1526 		       loff_t *ppos)
1527 {
1528 	struct trace_subsystem_dir *dir = filp->private_data;
1529 	char *buf;
1530 	int err;
1531 
1532 	if (cnt >= PAGE_SIZE)
1533 		return -EINVAL;
1534 
1535 	buf = memdup_user_nul(ubuf, cnt);
1536 	if (IS_ERR(buf))
1537 		return PTR_ERR(buf);
1538 
1539 	err = apply_subsystem_event_filter(dir, buf);
1540 	kfree(buf);
1541 	if (err < 0)
1542 		return err;
1543 
1544 	*ppos += cnt;
1545 
1546 	return cnt;
1547 }
1548 
1549 static ssize_t
show_header(struct file * filp,char __user * ubuf,size_t cnt,loff_t * ppos)1550 show_header(struct file *filp, char __user *ubuf, size_t cnt, loff_t *ppos)
1551 {
1552 	int (*func)(struct trace_seq *s) = filp->private_data;
1553 	struct trace_seq *s;
1554 	int r;
1555 
1556 	if (*ppos)
1557 		return 0;
1558 
1559 	s = kmalloc(sizeof(*s), GFP_KERNEL);
1560 	if (!s)
1561 		return -ENOMEM;
1562 
1563 	trace_seq_init(s);
1564 
1565 	func(s);
1566 	r = simple_read_from_buffer(ubuf, cnt, ppos,
1567 				    s->buffer, trace_seq_used(s));
1568 
1569 	kfree(s);
1570 
1571 	return r;
1572 }
1573 
ignore_task_cpu(void * data)1574 static void ignore_task_cpu(void *data)
1575 {
1576 	struct trace_array *tr = data;
1577 	struct trace_pid_list *pid_list;
1578 
1579 	/*
1580 	 * This function is called by on_each_cpu() while the
1581 	 * event_mutex is held.
1582 	 */
1583 	pid_list = rcu_dereference_protected(tr->filtered_pids,
1584 					     mutex_is_locked(&event_mutex));
1585 
1586 	this_cpu_write(tr->trace_buffer.data->ignore_pid,
1587 		       trace_ignore_this_task(pid_list, current));
1588 }
1589 
1590 static ssize_t
ftrace_event_pid_write(struct file * filp,const char __user * ubuf,size_t cnt,loff_t * ppos)1591 ftrace_event_pid_write(struct file *filp, const char __user *ubuf,
1592 		       size_t cnt, loff_t *ppos)
1593 {
1594 	struct seq_file *m = filp->private_data;
1595 	struct trace_array *tr = m->private;
1596 	struct trace_pid_list *filtered_pids = NULL;
1597 	struct trace_pid_list *pid_list;
1598 	struct trace_event_file *file;
1599 	ssize_t ret;
1600 
1601 	if (!cnt)
1602 		return 0;
1603 
1604 	ret = tracing_update_buffers();
1605 	if (ret < 0)
1606 		return ret;
1607 
1608 	mutex_lock(&event_mutex);
1609 
1610 	filtered_pids = rcu_dereference_protected(tr->filtered_pids,
1611 					     lockdep_is_held(&event_mutex));
1612 
1613 	ret = trace_pid_write(filtered_pids, &pid_list, ubuf, cnt);
1614 	if (ret < 0)
1615 		goto out;
1616 
1617 	rcu_assign_pointer(tr->filtered_pids, pid_list);
1618 
1619 	list_for_each_entry(file, &tr->events, list) {
1620 		set_bit(EVENT_FILE_FL_PID_FILTER_BIT, &file->flags);
1621 	}
1622 
1623 	if (filtered_pids) {
1624 		synchronize_sched();
1625 		trace_free_pid_list(filtered_pids);
1626 	} else if (pid_list) {
1627 		/*
1628 		 * Register a probe that is called before all other probes
1629 		 * to set ignore_pid if next or prev do not match.
1630 		 * Register a probe this is called after all other probes
1631 		 * to only keep ignore_pid set if next pid matches.
1632 		 */
1633 		register_trace_prio_sched_switch(event_filter_pid_sched_switch_probe_pre,
1634 						 tr, INT_MAX);
1635 		register_trace_prio_sched_switch(event_filter_pid_sched_switch_probe_post,
1636 						 tr, 0);
1637 
1638 		register_trace_prio_sched_wakeup(event_filter_pid_sched_wakeup_probe_pre,
1639 						 tr, INT_MAX);
1640 		register_trace_prio_sched_wakeup(event_filter_pid_sched_wakeup_probe_post,
1641 						 tr, 0);
1642 
1643 		register_trace_prio_sched_wakeup_new(event_filter_pid_sched_wakeup_probe_pre,
1644 						     tr, INT_MAX);
1645 		register_trace_prio_sched_wakeup_new(event_filter_pid_sched_wakeup_probe_post,
1646 						     tr, 0);
1647 
1648 		register_trace_prio_sched_waking(event_filter_pid_sched_wakeup_probe_pre,
1649 						 tr, INT_MAX);
1650 		register_trace_prio_sched_waking(event_filter_pid_sched_wakeup_probe_post,
1651 						 tr, 0);
1652 	}
1653 
1654 	/*
1655 	 * Ignoring of pids is done at task switch. But we have to
1656 	 * check for those tasks that are currently running.
1657 	 * Always do this in case a pid was appended or removed.
1658 	 */
1659 	on_each_cpu(ignore_task_cpu, tr, 1);
1660 
1661  out:
1662 	mutex_unlock(&event_mutex);
1663 
1664 	if (ret > 0)
1665 		*ppos += ret;
1666 
1667 	return ret;
1668 }
1669 
1670 static int ftrace_event_avail_open(struct inode *inode, struct file *file);
1671 static int ftrace_event_set_open(struct inode *inode, struct file *file);
1672 static int ftrace_event_set_pid_open(struct inode *inode, struct file *file);
1673 static int ftrace_event_release(struct inode *inode, struct file *file);
1674 
1675 static const struct seq_operations show_event_seq_ops = {
1676 	.start = t_start,
1677 	.next = t_next,
1678 	.show = t_show,
1679 	.stop = t_stop,
1680 };
1681 
1682 static const struct seq_operations show_set_event_seq_ops = {
1683 	.start = s_start,
1684 	.next = s_next,
1685 	.show = t_show,
1686 	.stop = t_stop,
1687 };
1688 
1689 static const struct seq_operations show_set_pid_seq_ops = {
1690 	.start = p_start,
1691 	.next = p_next,
1692 	.show = trace_pid_show,
1693 	.stop = p_stop,
1694 };
1695 
1696 static const struct file_operations ftrace_avail_fops = {
1697 	.open = ftrace_event_avail_open,
1698 	.read = seq_read,
1699 	.llseek = seq_lseek,
1700 	.release = seq_release,
1701 };
1702 
1703 static const struct file_operations ftrace_set_event_fops = {
1704 	.open = ftrace_event_set_open,
1705 	.read = seq_read,
1706 	.write = ftrace_event_write,
1707 	.llseek = seq_lseek,
1708 	.release = ftrace_event_release,
1709 };
1710 
1711 static const struct file_operations ftrace_set_event_pid_fops = {
1712 	.open = ftrace_event_set_pid_open,
1713 	.read = seq_read,
1714 	.write = ftrace_event_pid_write,
1715 	.llseek = seq_lseek,
1716 	.release = ftrace_event_release,
1717 };
1718 
1719 static const struct file_operations ftrace_enable_fops = {
1720 	.open = tracing_open_generic,
1721 	.read = event_enable_read,
1722 	.write = event_enable_write,
1723 	.llseek = default_llseek,
1724 };
1725 
1726 static const struct file_operations ftrace_event_format_fops = {
1727 	.open = trace_format_open,
1728 	.read = seq_read,
1729 	.llseek = seq_lseek,
1730 	.release = seq_release,
1731 };
1732 
1733 static const struct file_operations ftrace_event_id_fops = {
1734 	.read = event_id_read,
1735 	.llseek = default_llseek,
1736 };
1737 
1738 static const struct file_operations ftrace_event_filter_fops = {
1739 	.open = tracing_open_generic,
1740 	.read = event_filter_read,
1741 	.write = event_filter_write,
1742 	.llseek = default_llseek,
1743 };
1744 
1745 static const struct file_operations ftrace_subsystem_filter_fops = {
1746 	.open = subsystem_open,
1747 	.read = subsystem_filter_read,
1748 	.write = subsystem_filter_write,
1749 	.llseek = default_llseek,
1750 	.release = subsystem_release,
1751 };
1752 
1753 static const struct file_operations ftrace_system_enable_fops = {
1754 	.open = subsystem_open,
1755 	.read = system_enable_read,
1756 	.write = system_enable_write,
1757 	.llseek = default_llseek,
1758 	.release = subsystem_release,
1759 };
1760 
1761 static const struct file_operations ftrace_tr_enable_fops = {
1762 	.open = system_tr_open,
1763 	.read = system_enable_read,
1764 	.write = system_enable_write,
1765 	.llseek = default_llseek,
1766 	.release = subsystem_release,
1767 };
1768 
1769 static const struct file_operations ftrace_show_header_fops = {
1770 	.open = tracing_open_generic,
1771 	.read = show_header,
1772 	.llseek = default_llseek,
1773 };
1774 
1775 static int
ftrace_event_open(struct inode * inode,struct file * file,const struct seq_operations * seq_ops)1776 ftrace_event_open(struct inode *inode, struct file *file,
1777 		  const struct seq_operations *seq_ops)
1778 {
1779 	struct seq_file *m;
1780 	int ret;
1781 
1782 	ret = seq_open(file, seq_ops);
1783 	if (ret < 0)
1784 		return ret;
1785 	m = file->private_data;
1786 	/* copy tr over to seq ops */
1787 	m->private = inode->i_private;
1788 
1789 	return ret;
1790 }
1791 
ftrace_event_release(struct inode * inode,struct file * file)1792 static int ftrace_event_release(struct inode *inode, struct file *file)
1793 {
1794 	struct trace_array *tr = inode->i_private;
1795 
1796 	trace_array_put(tr);
1797 
1798 	return seq_release(inode, file);
1799 }
1800 
1801 static int
ftrace_event_avail_open(struct inode * inode,struct file * file)1802 ftrace_event_avail_open(struct inode *inode, struct file *file)
1803 {
1804 	const struct seq_operations *seq_ops = &show_event_seq_ops;
1805 
1806 	return ftrace_event_open(inode, file, seq_ops);
1807 }
1808 
1809 static int
ftrace_event_set_open(struct inode * inode,struct file * file)1810 ftrace_event_set_open(struct inode *inode, struct file *file)
1811 {
1812 	const struct seq_operations *seq_ops = &show_set_event_seq_ops;
1813 	struct trace_array *tr = inode->i_private;
1814 	int ret;
1815 
1816 	if (trace_array_get(tr) < 0)
1817 		return -ENODEV;
1818 
1819 	if ((file->f_mode & FMODE_WRITE) &&
1820 	    (file->f_flags & O_TRUNC))
1821 		ftrace_clear_events(tr);
1822 
1823 	ret = ftrace_event_open(inode, file, seq_ops);
1824 	if (ret < 0)
1825 		trace_array_put(tr);
1826 	return ret;
1827 }
1828 
1829 static int
ftrace_event_set_pid_open(struct inode * inode,struct file * file)1830 ftrace_event_set_pid_open(struct inode *inode, struct file *file)
1831 {
1832 	const struct seq_operations *seq_ops = &show_set_pid_seq_ops;
1833 	struct trace_array *tr = inode->i_private;
1834 	int ret;
1835 
1836 	if (trace_array_get(tr) < 0)
1837 		return -ENODEV;
1838 
1839 	if ((file->f_mode & FMODE_WRITE) &&
1840 	    (file->f_flags & O_TRUNC))
1841 		ftrace_clear_event_pids(tr);
1842 
1843 	ret = ftrace_event_open(inode, file, seq_ops);
1844 	if (ret < 0)
1845 		trace_array_put(tr);
1846 	return ret;
1847 }
1848 
1849 static struct event_subsystem *
create_new_subsystem(const char * name)1850 create_new_subsystem(const char *name)
1851 {
1852 	struct event_subsystem *system;
1853 
1854 	/* need to create new entry */
1855 	system = kmalloc(sizeof(*system), GFP_KERNEL);
1856 	if (!system)
1857 		return NULL;
1858 
1859 	system->ref_count = 1;
1860 
1861 	/* Only allocate if dynamic (kprobes and modules) */
1862 	system->name = kstrdup_const(name, GFP_KERNEL);
1863 	if (!system->name)
1864 		goto out_free;
1865 
1866 	system->filter = NULL;
1867 
1868 	system->filter = kzalloc(sizeof(struct event_filter), GFP_KERNEL);
1869 	if (!system->filter)
1870 		goto out_free;
1871 
1872 	list_add(&system->list, &event_subsystems);
1873 
1874 	return system;
1875 
1876  out_free:
1877 	kfree_const(system->name);
1878 	kfree(system);
1879 	return NULL;
1880 }
1881 
1882 static struct dentry *
event_subsystem_dir(struct trace_array * tr,const char * name,struct trace_event_file * file,struct dentry * parent)1883 event_subsystem_dir(struct trace_array *tr, const char *name,
1884 		    struct trace_event_file *file, struct dentry *parent)
1885 {
1886 	struct trace_subsystem_dir *dir;
1887 	struct event_subsystem *system;
1888 	struct dentry *entry;
1889 
1890 	/* First see if we did not already create this dir */
1891 	list_for_each_entry(dir, &tr->systems, list) {
1892 		system = dir->subsystem;
1893 		if (strcmp(system->name, name) == 0) {
1894 			dir->nr_events++;
1895 			file->system = dir;
1896 			return dir->entry;
1897 		}
1898 	}
1899 
1900 	/* Now see if the system itself exists. */
1901 	list_for_each_entry(system, &event_subsystems, list) {
1902 		if (strcmp(system->name, name) == 0)
1903 			break;
1904 	}
1905 	/* Reset system variable when not found */
1906 	if (&system->list == &event_subsystems)
1907 		system = NULL;
1908 
1909 	dir = kmalloc(sizeof(*dir), GFP_KERNEL);
1910 	if (!dir)
1911 		goto out_fail;
1912 
1913 	if (!system) {
1914 		system = create_new_subsystem(name);
1915 		if (!system)
1916 			goto out_free;
1917 	} else
1918 		__get_system(system);
1919 
1920 	dir->entry = tracefs_create_dir(name, parent);
1921 	if (!dir->entry) {
1922 		pr_warn("Failed to create system directory %s\n", name);
1923 		__put_system(system);
1924 		goto out_free;
1925 	}
1926 
1927 	dir->tr = tr;
1928 	dir->ref_count = 1;
1929 	dir->nr_events = 1;
1930 	dir->subsystem = system;
1931 	file->system = dir;
1932 
1933 	entry = tracefs_create_file("filter", 0644, dir->entry, dir,
1934 				    &ftrace_subsystem_filter_fops);
1935 	if (!entry) {
1936 		kfree(system->filter);
1937 		system->filter = NULL;
1938 		pr_warn("Could not create tracefs '%s/filter' entry\n", name);
1939 	}
1940 
1941 	trace_create_file("enable", 0644, dir->entry, dir,
1942 			  &ftrace_system_enable_fops);
1943 
1944 	list_add(&dir->list, &tr->systems);
1945 
1946 	return dir->entry;
1947 
1948  out_free:
1949 	kfree(dir);
1950  out_fail:
1951 	/* Only print this message if failed on memory allocation */
1952 	if (!dir || !system)
1953 		pr_warn("No memory to create event subsystem %s\n", name);
1954 	return NULL;
1955 }
1956 
1957 static int
event_create_dir(struct dentry * parent,struct trace_event_file * file)1958 event_create_dir(struct dentry *parent, struct trace_event_file *file)
1959 {
1960 	struct trace_event_call *call = file->event_call;
1961 	struct trace_array *tr = file->tr;
1962 	struct list_head *head;
1963 	struct dentry *d_events;
1964 	const char *name;
1965 	int ret;
1966 
1967 	/*
1968 	 * If the trace point header did not define TRACE_SYSTEM
1969 	 * then the system would be called "TRACE_SYSTEM".
1970 	 */
1971 	if (strcmp(call->class->system, TRACE_SYSTEM) != 0) {
1972 		d_events = event_subsystem_dir(tr, call->class->system, file, parent);
1973 		if (!d_events)
1974 			return -ENOMEM;
1975 	} else
1976 		d_events = parent;
1977 
1978 	name = trace_event_name(call);
1979 	file->dir = tracefs_create_dir(name, d_events);
1980 	if (!file->dir) {
1981 		pr_warn("Could not create tracefs '%s' directory\n", name);
1982 		return -1;
1983 	}
1984 
1985 	if (call->class->reg && !(call->flags & TRACE_EVENT_FL_IGNORE_ENABLE))
1986 		trace_create_file("enable", 0644, file->dir, file,
1987 				  &ftrace_enable_fops);
1988 
1989 #ifdef CONFIG_PERF_EVENTS
1990 	if (call->event.type && call->class->reg)
1991 		trace_create_file("id", 0444, file->dir,
1992 				  (void *)(long)call->event.type,
1993 				  &ftrace_event_id_fops);
1994 #endif
1995 
1996 	/*
1997 	 * Other events may have the same class. Only update
1998 	 * the fields if they are not already defined.
1999 	 */
2000 	head = trace_get_fields(call);
2001 	if (list_empty(head)) {
2002 		ret = call->class->define_fields(call);
2003 		if (ret < 0) {
2004 			pr_warn("Could not initialize trace point events/%s\n",
2005 				name);
2006 			return -1;
2007 		}
2008 	}
2009 	trace_create_file("filter", 0644, file->dir, file,
2010 			  &ftrace_event_filter_fops);
2011 
2012 	/*
2013 	 * Only event directories that can be enabled should have
2014 	 * triggers.
2015 	 */
2016 	if (!(call->flags & TRACE_EVENT_FL_IGNORE_ENABLE))
2017 		trace_create_file("trigger", 0644, file->dir, file,
2018 				  &event_trigger_fops);
2019 
2020 #ifdef CONFIG_HIST_TRIGGERS
2021 	trace_create_file("hist", 0444, file->dir, file,
2022 			  &event_hist_fops);
2023 #endif
2024 	trace_create_file("format", 0444, file->dir, call,
2025 			  &ftrace_event_format_fops);
2026 
2027 	return 0;
2028 }
2029 
remove_event_from_tracers(struct trace_event_call * call)2030 static void remove_event_from_tracers(struct trace_event_call *call)
2031 {
2032 	struct trace_event_file *file;
2033 	struct trace_array *tr;
2034 
2035 	do_for_each_event_file_safe(tr, file) {
2036 		if (file->event_call != call)
2037 			continue;
2038 
2039 		remove_event_file_dir(file);
2040 		/*
2041 		 * The do_for_each_event_file_safe() is
2042 		 * a double loop. After finding the call for this
2043 		 * trace_array, we use break to jump to the next
2044 		 * trace_array.
2045 		 */
2046 		break;
2047 	} while_for_each_event_file();
2048 }
2049 
event_remove(struct trace_event_call * call)2050 static void event_remove(struct trace_event_call *call)
2051 {
2052 	struct trace_array *tr;
2053 	struct trace_event_file *file;
2054 
2055 	do_for_each_event_file(tr, file) {
2056 		if (file->event_call != call)
2057 			continue;
2058 
2059 		if (file->flags & EVENT_FILE_FL_WAS_ENABLED)
2060 			tr->clear_trace = true;
2061 
2062 		ftrace_event_enable_disable(file, 0);
2063 		/*
2064 		 * The do_for_each_event_file() is
2065 		 * a double loop. After finding the call for this
2066 		 * trace_array, we use break to jump to the next
2067 		 * trace_array.
2068 		 */
2069 		break;
2070 	} while_for_each_event_file();
2071 
2072 	if (call->event.funcs)
2073 		__unregister_trace_event(&call->event);
2074 	remove_event_from_tracers(call);
2075 	list_del(&call->list);
2076 }
2077 
event_init(struct trace_event_call * call)2078 static int event_init(struct trace_event_call *call)
2079 {
2080 	int ret = 0;
2081 	const char *name;
2082 
2083 	name = trace_event_name(call);
2084 	if (WARN_ON(!name))
2085 		return -EINVAL;
2086 
2087 	if (call->class->raw_init) {
2088 		ret = call->class->raw_init(call);
2089 		if (ret < 0 && ret != -ENOSYS)
2090 			pr_warn("Could not initialize trace events/%s\n", name);
2091 	}
2092 
2093 	return ret;
2094 }
2095 
2096 static int
__register_event(struct trace_event_call * call,struct module * mod)2097 __register_event(struct trace_event_call *call, struct module *mod)
2098 {
2099 	int ret;
2100 
2101 	ret = event_init(call);
2102 	if (ret < 0)
2103 		return ret;
2104 
2105 	list_add(&call->list, &ftrace_events);
2106 	call->mod = mod;
2107 
2108 	return 0;
2109 }
2110 
eval_replace(char * ptr,struct trace_eval_map * map,int len)2111 static char *eval_replace(char *ptr, struct trace_eval_map *map, int len)
2112 {
2113 	int rlen;
2114 	int elen;
2115 
2116 	/* Find the length of the eval value as a string */
2117 	elen = snprintf(ptr, 0, "%ld", map->eval_value);
2118 	/* Make sure there's enough room to replace the string with the value */
2119 	if (len < elen)
2120 		return NULL;
2121 
2122 	snprintf(ptr, elen + 1, "%ld", map->eval_value);
2123 
2124 	/* Get the rest of the string of ptr */
2125 	rlen = strlen(ptr + len);
2126 	memmove(ptr + elen, ptr + len, rlen);
2127 	/* Make sure we end the new string */
2128 	ptr[elen + rlen] = 0;
2129 
2130 	return ptr + elen;
2131 }
2132 
update_event_printk(struct trace_event_call * call,struct trace_eval_map * map)2133 static void update_event_printk(struct trace_event_call *call,
2134 				struct trace_eval_map *map)
2135 {
2136 	char *ptr;
2137 	int quote = 0;
2138 	int len = strlen(map->eval_string);
2139 
2140 	for (ptr = call->print_fmt; *ptr; ptr++) {
2141 		if (*ptr == '\\') {
2142 			ptr++;
2143 			/* paranoid */
2144 			if (!*ptr)
2145 				break;
2146 			continue;
2147 		}
2148 		if (*ptr == '"') {
2149 			quote ^= 1;
2150 			continue;
2151 		}
2152 		if (quote)
2153 			continue;
2154 		if (isdigit(*ptr)) {
2155 			/* skip numbers */
2156 			do {
2157 				ptr++;
2158 				/* Check for alpha chars like ULL */
2159 			} while (isalnum(*ptr));
2160 			if (!*ptr)
2161 				break;
2162 			/*
2163 			 * A number must have some kind of delimiter after
2164 			 * it, and we can ignore that too.
2165 			 */
2166 			continue;
2167 		}
2168 		if (isalpha(*ptr) || *ptr == '_') {
2169 			if (strncmp(map->eval_string, ptr, len) == 0 &&
2170 			    !isalnum(ptr[len]) && ptr[len] != '_') {
2171 				ptr = eval_replace(ptr, map, len);
2172 				/* enum/sizeof string smaller than value */
2173 				if (WARN_ON_ONCE(!ptr))
2174 					return;
2175 				/*
2176 				 * No need to decrement here, as eval_replace()
2177 				 * returns the pointer to the character passed
2178 				 * the eval, and two evals can not be placed
2179 				 * back to back without something in between.
2180 				 * We can skip that something in between.
2181 				 */
2182 				continue;
2183 			}
2184 		skip_more:
2185 			do {
2186 				ptr++;
2187 			} while (isalnum(*ptr) || *ptr == '_');
2188 			if (!*ptr)
2189 				break;
2190 			/*
2191 			 * If what comes after this variable is a '.' or
2192 			 * '->' then we can continue to ignore that string.
2193 			 */
2194 			if (*ptr == '.' || (ptr[0] == '-' && ptr[1] == '>')) {
2195 				ptr += *ptr == '.' ? 1 : 2;
2196 				if (!*ptr)
2197 					break;
2198 				goto skip_more;
2199 			}
2200 			/*
2201 			 * Once again, we can skip the delimiter that came
2202 			 * after the string.
2203 			 */
2204 			continue;
2205 		}
2206 	}
2207 }
2208 
trace_event_eval_update(struct trace_eval_map ** map,int len)2209 void trace_event_eval_update(struct trace_eval_map **map, int len)
2210 {
2211 	struct trace_event_call *call, *p;
2212 	const char *last_system = NULL;
2213 	bool first = false;
2214 	int last_i;
2215 	int i;
2216 
2217 	down_write(&trace_event_sem);
2218 	list_for_each_entry_safe(call, p, &ftrace_events, list) {
2219 		/* events are usually grouped together with systems */
2220 		if (!last_system || call->class->system != last_system) {
2221 			first = true;
2222 			last_i = 0;
2223 			last_system = call->class->system;
2224 		}
2225 
2226 		/*
2227 		 * Since calls are grouped by systems, the likelyhood that the
2228 		 * next call in the iteration belongs to the same system as the
2229 		 * previous call is high. As an optimization, we skip seaching
2230 		 * for a map[] that matches the call's system if the last call
2231 		 * was from the same system. That's what last_i is for. If the
2232 		 * call has the same system as the previous call, then last_i
2233 		 * will be the index of the first map[] that has a matching
2234 		 * system.
2235 		 */
2236 		for (i = last_i; i < len; i++) {
2237 			if (call->class->system == map[i]->system) {
2238 				/* Save the first system if need be */
2239 				if (first) {
2240 					last_i = i;
2241 					first = false;
2242 				}
2243 				update_event_printk(call, map[i]);
2244 			}
2245 		}
2246 	}
2247 	up_write(&trace_event_sem);
2248 }
2249 
2250 static struct trace_event_file *
trace_create_new_event(struct trace_event_call * call,struct trace_array * tr)2251 trace_create_new_event(struct trace_event_call *call,
2252 		       struct trace_array *tr)
2253 {
2254 	struct trace_event_file *file;
2255 
2256 	file = kmem_cache_alloc(file_cachep, GFP_TRACE);
2257 	if (!file)
2258 		return NULL;
2259 
2260 	file->event_call = call;
2261 	file->tr = tr;
2262 	atomic_set(&file->sm_ref, 0);
2263 	atomic_set(&file->tm_ref, 0);
2264 	INIT_LIST_HEAD(&file->triggers);
2265 	list_add(&file->list, &tr->events);
2266 
2267 	return file;
2268 }
2269 
2270 /* Add an event to a trace directory */
2271 static int
__trace_add_new_event(struct trace_event_call * call,struct trace_array * tr)2272 __trace_add_new_event(struct trace_event_call *call, struct trace_array *tr)
2273 {
2274 	struct trace_event_file *file;
2275 
2276 	file = trace_create_new_event(call, tr);
2277 	if (!file)
2278 		return -ENOMEM;
2279 
2280 	return event_create_dir(tr->event_dir, file);
2281 }
2282 
2283 /*
2284  * Just create a decriptor for early init. A descriptor is required
2285  * for enabling events at boot. We want to enable events before
2286  * the filesystem is initialized.
2287  */
2288 static __init int
__trace_early_add_new_event(struct trace_event_call * call,struct trace_array * tr)2289 __trace_early_add_new_event(struct trace_event_call *call,
2290 			    struct trace_array *tr)
2291 {
2292 	struct trace_event_file *file;
2293 
2294 	file = trace_create_new_event(call, tr);
2295 	if (!file)
2296 		return -ENOMEM;
2297 
2298 	return 0;
2299 }
2300 
2301 struct ftrace_module_file_ops;
2302 static void __add_event_to_tracers(struct trace_event_call *call);
2303 
2304 /* Add an additional event_call dynamically */
trace_add_event_call(struct trace_event_call * call)2305 int trace_add_event_call(struct trace_event_call *call)
2306 {
2307 	int ret;
2308 	mutex_lock(&trace_types_lock);
2309 	mutex_lock(&event_mutex);
2310 
2311 	ret = __register_event(call, NULL);
2312 	if (ret >= 0)
2313 		__add_event_to_tracers(call);
2314 
2315 	mutex_unlock(&event_mutex);
2316 	mutex_unlock(&trace_types_lock);
2317 	return ret;
2318 }
2319 
2320 /*
2321  * Must be called under locking of trace_types_lock, event_mutex and
2322  * trace_event_sem.
2323  */
__trace_remove_event_call(struct trace_event_call * call)2324 static void __trace_remove_event_call(struct trace_event_call *call)
2325 {
2326 	event_remove(call);
2327 	trace_destroy_fields(call);
2328 	free_event_filter(call->filter);
2329 	call->filter = NULL;
2330 }
2331 
probe_remove_event_call(struct trace_event_call * call)2332 static int probe_remove_event_call(struct trace_event_call *call)
2333 {
2334 	struct trace_array *tr;
2335 	struct trace_event_file *file;
2336 
2337 #ifdef CONFIG_PERF_EVENTS
2338 	if (call->perf_refcount)
2339 		return -EBUSY;
2340 #endif
2341 	do_for_each_event_file(tr, file) {
2342 		if (file->event_call != call)
2343 			continue;
2344 		/*
2345 		 * We can't rely on ftrace_event_enable_disable(enable => 0)
2346 		 * we are going to do, EVENT_FILE_FL_SOFT_MODE can suppress
2347 		 * TRACE_REG_UNREGISTER.
2348 		 */
2349 		if (file->flags & EVENT_FILE_FL_ENABLED)
2350 			return -EBUSY;
2351 		/*
2352 		 * The do_for_each_event_file_safe() is
2353 		 * a double loop. After finding the call for this
2354 		 * trace_array, we use break to jump to the next
2355 		 * trace_array.
2356 		 */
2357 		break;
2358 	} while_for_each_event_file();
2359 
2360 	__trace_remove_event_call(call);
2361 
2362 	return 0;
2363 }
2364 
2365 /* Remove an event_call */
trace_remove_event_call(struct trace_event_call * call)2366 int trace_remove_event_call(struct trace_event_call *call)
2367 {
2368 	int ret;
2369 
2370 	mutex_lock(&trace_types_lock);
2371 	mutex_lock(&event_mutex);
2372 	down_write(&trace_event_sem);
2373 	ret = probe_remove_event_call(call);
2374 	up_write(&trace_event_sem);
2375 	mutex_unlock(&event_mutex);
2376 	mutex_unlock(&trace_types_lock);
2377 
2378 	return ret;
2379 }
2380 
2381 #define for_each_event(event, start, end)			\
2382 	for (event = start;					\
2383 	     (unsigned long)event < (unsigned long)end;		\
2384 	     event++)
2385 
2386 #ifdef CONFIG_MODULES
2387 
trace_module_add_events(struct module * mod)2388 static void trace_module_add_events(struct module *mod)
2389 {
2390 	struct trace_event_call **call, **start, **end;
2391 
2392 	if (!mod->num_trace_events)
2393 		return;
2394 
2395 	/* Don't add infrastructure for mods without tracepoints */
2396 	if (trace_module_has_bad_taint(mod)) {
2397 		pr_err("%s: module has bad taint, not creating trace events\n",
2398 		       mod->name);
2399 		return;
2400 	}
2401 
2402 	start = mod->trace_events;
2403 	end = mod->trace_events + mod->num_trace_events;
2404 
2405 	for_each_event(call, start, end) {
2406 		__register_event(*call, mod);
2407 		__add_event_to_tracers(*call);
2408 	}
2409 }
2410 
trace_module_remove_events(struct module * mod)2411 static void trace_module_remove_events(struct module *mod)
2412 {
2413 	struct trace_event_call *call, *p;
2414 
2415 	down_write(&trace_event_sem);
2416 	list_for_each_entry_safe(call, p, &ftrace_events, list) {
2417 		if (call->mod == mod)
2418 			__trace_remove_event_call(call);
2419 	}
2420 	up_write(&trace_event_sem);
2421 
2422 	/*
2423 	 * It is safest to reset the ring buffer if the module being unloaded
2424 	 * registered any events that were used. The only worry is if
2425 	 * a new module gets loaded, and takes on the same id as the events
2426 	 * of this module. When printing out the buffer, traced events left
2427 	 * over from this module may be passed to the new module events and
2428 	 * unexpected results may occur.
2429 	 */
2430 	tracing_reset_all_online_cpus();
2431 }
2432 
trace_module_notify(struct notifier_block * self,unsigned long val,void * data)2433 static int trace_module_notify(struct notifier_block *self,
2434 			       unsigned long val, void *data)
2435 {
2436 	struct module *mod = data;
2437 
2438 	mutex_lock(&trace_types_lock);
2439 	mutex_lock(&event_mutex);
2440 	switch (val) {
2441 	case MODULE_STATE_COMING:
2442 		trace_module_add_events(mod);
2443 		break;
2444 	case MODULE_STATE_GOING:
2445 		trace_module_remove_events(mod);
2446 		break;
2447 	}
2448 	mutex_unlock(&event_mutex);
2449 	mutex_unlock(&trace_types_lock);
2450 
2451 	return 0;
2452 }
2453 
2454 static struct notifier_block trace_module_nb = {
2455 	.notifier_call = trace_module_notify,
2456 	.priority = 1, /* higher than trace.c module notify */
2457 };
2458 #endif /* CONFIG_MODULES */
2459 
2460 /* Create a new event directory structure for a trace directory. */
2461 static void
__trace_add_event_dirs(struct trace_array * tr)2462 __trace_add_event_dirs(struct trace_array *tr)
2463 {
2464 	struct trace_event_call *call;
2465 	int ret;
2466 
2467 	list_for_each_entry(call, &ftrace_events, list) {
2468 		ret = __trace_add_new_event(call, tr);
2469 		if (ret < 0)
2470 			pr_warn("Could not create directory for event %s\n",
2471 				trace_event_name(call));
2472 	}
2473 }
2474 
2475 struct trace_event_file *
find_event_file(struct trace_array * tr,const char * system,const char * event)2476 find_event_file(struct trace_array *tr, const char *system,  const char *event)
2477 {
2478 	struct trace_event_file *file;
2479 	struct trace_event_call *call;
2480 	const char *name;
2481 
2482 	list_for_each_entry(file, &tr->events, list) {
2483 
2484 		call = file->event_call;
2485 		name = trace_event_name(call);
2486 
2487 		if (!name || !call->class || !call->class->reg)
2488 			continue;
2489 
2490 		if (call->flags & TRACE_EVENT_FL_IGNORE_ENABLE)
2491 			continue;
2492 
2493 		if (strcmp(event, name) == 0 &&
2494 		    strcmp(system, call->class->system) == 0)
2495 			return file;
2496 	}
2497 	return NULL;
2498 }
2499 
2500 #ifdef CONFIG_DYNAMIC_FTRACE
2501 
2502 /* Avoid typos */
2503 #define ENABLE_EVENT_STR	"enable_event"
2504 #define DISABLE_EVENT_STR	"disable_event"
2505 
2506 struct event_probe_data {
2507 	struct trace_event_file	*file;
2508 	unsigned long			count;
2509 	int				ref;
2510 	bool				enable;
2511 };
2512 
update_event_probe(struct event_probe_data * data)2513 static void update_event_probe(struct event_probe_data *data)
2514 {
2515 	if (data->enable)
2516 		clear_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &data->file->flags);
2517 	else
2518 		set_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &data->file->flags);
2519 }
2520 
2521 static void
event_enable_probe(unsigned long ip,unsigned long parent_ip,struct trace_array * tr,struct ftrace_probe_ops * ops,void * data)2522 event_enable_probe(unsigned long ip, unsigned long parent_ip,
2523 		   struct trace_array *tr, struct ftrace_probe_ops *ops,
2524 		   void *data)
2525 {
2526 	struct ftrace_func_mapper *mapper = data;
2527 	struct event_probe_data *edata;
2528 	void **pdata;
2529 
2530 	pdata = ftrace_func_mapper_find_ip(mapper, ip);
2531 	if (!pdata || !*pdata)
2532 		return;
2533 
2534 	edata = *pdata;
2535 	update_event_probe(edata);
2536 }
2537 
2538 static void
event_enable_count_probe(unsigned long ip,unsigned long parent_ip,struct trace_array * tr,struct ftrace_probe_ops * ops,void * data)2539 event_enable_count_probe(unsigned long ip, unsigned long parent_ip,
2540 			 struct trace_array *tr, struct ftrace_probe_ops *ops,
2541 			 void *data)
2542 {
2543 	struct ftrace_func_mapper *mapper = data;
2544 	struct event_probe_data *edata;
2545 	void **pdata;
2546 
2547 	pdata = ftrace_func_mapper_find_ip(mapper, ip);
2548 	if (!pdata || !*pdata)
2549 		return;
2550 
2551 	edata = *pdata;
2552 
2553 	if (!edata->count)
2554 		return;
2555 
2556 	/* Skip if the event is in a state we want to switch to */
2557 	if (edata->enable == !(edata->file->flags & EVENT_FILE_FL_SOFT_DISABLED))
2558 		return;
2559 
2560 	if (edata->count != -1)
2561 		(edata->count)--;
2562 
2563 	update_event_probe(edata);
2564 }
2565 
2566 static int
event_enable_print(struct seq_file * m,unsigned long ip,struct ftrace_probe_ops * ops,void * data)2567 event_enable_print(struct seq_file *m, unsigned long ip,
2568 		   struct ftrace_probe_ops *ops, void *data)
2569 {
2570 	struct ftrace_func_mapper *mapper = data;
2571 	struct event_probe_data *edata;
2572 	void **pdata;
2573 
2574 	pdata = ftrace_func_mapper_find_ip(mapper, ip);
2575 
2576 	if (WARN_ON_ONCE(!pdata || !*pdata))
2577 		return 0;
2578 
2579 	edata = *pdata;
2580 
2581 	seq_printf(m, "%ps:", (void *)ip);
2582 
2583 	seq_printf(m, "%s:%s:%s",
2584 		   edata->enable ? ENABLE_EVENT_STR : DISABLE_EVENT_STR,
2585 		   edata->file->event_call->class->system,
2586 		   trace_event_name(edata->file->event_call));
2587 
2588 	if (edata->count == -1)
2589 		seq_puts(m, ":unlimited\n");
2590 	else
2591 		seq_printf(m, ":count=%ld\n", edata->count);
2592 
2593 	return 0;
2594 }
2595 
2596 static int
event_enable_init(struct ftrace_probe_ops * ops,struct trace_array * tr,unsigned long ip,void * init_data,void ** data)2597 event_enable_init(struct ftrace_probe_ops *ops, struct trace_array *tr,
2598 		  unsigned long ip, void *init_data, void **data)
2599 {
2600 	struct ftrace_func_mapper *mapper = *data;
2601 	struct event_probe_data *edata = init_data;
2602 	int ret;
2603 
2604 	if (!mapper) {
2605 		mapper = allocate_ftrace_func_mapper();
2606 		if (!mapper)
2607 			return -ENODEV;
2608 		*data = mapper;
2609 	}
2610 
2611 	ret = ftrace_func_mapper_add_ip(mapper, ip, edata);
2612 	if (ret < 0)
2613 		return ret;
2614 
2615 	edata->ref++;
2616 
2617 	return 0;
2618 }
2619 
free_probe_data(void * data)2620 static int free_probe_data(void *data)
2621 {
2622 	struct event_probe_data *edata = data;
2623 
2624 	edata->ref--;
2625 	if (!edata->ref) {
2626 		/* Remove the SOFT_MODE flag */
2627 		__ftrace_event_enable_disable(edata->file, 0, 1);
2628 		module_put(edata->file->event_call->mod);
2629 		kfree(edata);
2630 	}
2631 	return 0;
2632 }
2633 
2634 static void
event_enable_free(struct ftrace_probe_ops * ops,struct trace_array * tr,unsigned long ip,void * data)2635 event_enable_free(struct ftrace_probe_ops *ops, struct trace_array *tr,
2636 		  unsigned long ip, void *data)
2637 {
2638 	struct ftrace_func_mapper *mapper = data;
2639 	struct event_probe_data *edata;
2640 
2641 	if (!ip) {
2642 		if (!mapper)
2643 			return;
2644 		free_ftrace_func_mapper(mapper, free_probe_data);
2645 		return;
2646 	}
2647 
2648 	edata = ftrace_func_mapper_remove_ip(mapper, ip);
2649 
2650 	if (WARN_ON_ONCE(!edata))
2651 		return;
2652 
2653 	if (WARN_ON_ONCE(edata->ref <= 0))
2654 		return;
2655 
2656 	free_probe_data(edata);
2657 }
2658 
2659 static struct ftrace_probe_ops event_enable_probe_ops = {
2660 	.func			= event_enable_probe,
2661 	.print			= event_enable_print,
2662 	.init			= event_enable_init,
2663 	.free			= event_enable_free,
2664 };
2665 
2666 static struct ftrace_probe_ops event_enable_count_probe_ops = {
2667 	.func			= event_enable_count_probe,
2668 	.print			= event_enable_print,
2669 	.init			= event_enable_init,
2670 	.free			= event_enable_free,
2671 };
2672 
2673 static struct ftrace_probe_ops event_disable_probe_ops = {
2674 	.func			= event_enable_probe,
2675 	.print			= event_enable_print,
2676 	.init			= event_enable_init,
2677 	.free			= event_enable_free,
2678 };
2679 
2680 static struct ftrace_probe_ops event_disable_count_probe_ops = {
2681 	.func			= event_enable_count_probe,
2682 	.print			= event_enable_print,
2683 	.init			= event_enable_init,
2684 	.free			= event_enable_free,
2685 };
2686 
2687 static int
event_enable_func(struct trace_array * tr,struct ftrace_hash * hash,char * glob,char * cmd,char * param,int enabled)2688 event_enable_func(struct trace_array *tr, struct ftrace_hash *hash,
2689 		  char *glob, char *cmd, char *param, int enabled)
2690 {
2691 	struct trace_event_file *file;
2692 	struct ftrace_probe_ops *ops;
2693 	struct event_probe_data *data;
2694 	const char *system;
2695 	const char *event;
2696 	char *number;
2697 	bool enable;
2698 	int ret;
2699 
2700 	if (!tr)
2701 		return -ENODEV;
2702 
2703 	/* hash funcs only work with set_ftrace_filter */
2704 	if (!enabled || !param)
2705 		return -EINVAL;
2706 
2707 	system = strsep(&param, ":");
2708 	if (!param)
2709 		return -EINVAL;
2710 
2711 	event = strsep(&param, ":");
2712 
2713 	mutex_lock(&event_mutex);
2714 
2715 	ret = -EINVAL;
2716 	file = find_event_file(tr, system, event);
2717 	if (!file)
2718 		goto out;
2719 
2720 	enable = strcmp(cmd, ENABLE_EVENT_STR) == 0;
2721 
2722 	if (enable)
2723 		ops = param ? &event_enable_count_probe_ops : &event_enable_probe_ops;
2724 	else
2725 		ops = param ? &event_disable_count_probe_ops : &event_disable_probe_ops;
2726 
2727 	if (glob[0] == '!') {
2728 		ret = unregister_ftrace_function_probe_func(glob+1, tr, ops);
2729 		goto out;
2730 	}
2731 
2732 	ret = -ENOMEM;
2733 
2734 	data = kzalloc(sizeof(*data), GFP_KERNEL);
2735 	if (!data)
2736 		goto out;
2737 
2738 	data->enable = enable;
2739 	data->count = -1;
2740 	data->file = file;
2741 
2742 	if (!param)
2743 		goto out_reg;
2744 
2745 	number = strsep(&param, ":");
2746 
2747 	ret = -EINVAL;
2748 	if (!strlen(number))
2749 		goto out_free;
2750 
2751 	/*
2752 	 * We use the callback data field (which is a pointer)
2753 	 * as our counter.
2754 	 */
2755 	ret = kstrtoul(number, 0, &data->count);
2756 	if (ret)
2757 		goto out_free;
2758 
2759  out_reg:
2760 	/* Don't let event modules unload while probe registered */
2761 	ret = try_module_get(file->event_call->mod);
2762 	if (!ret) {
2763 		ret = -EBUSY;
2764 		goto out_free;
2765 	}
2766 
2767 	ret = __ftrace_event_enable_disable(file, 1, 1);
2768 	if (ret < 0)
2769 		goto out_put;
2770 
2771 	ret = register_ftrace_function_probe(glob, tr, ops, data);
2772 	/*
2773 	 * The above returns on success the # of functions enabled,
2774 	 * but if it didn't find any functions it returns zero.
2775 	 * Consider no functions a failure too.
2776 	 */
2777 	if (!ret) {
2778 		ret = -ENOENT;
2779 		goto out_disable;
2780 	} else if (ret < 0)
2781 		goto out_disable;
2782 	/* Just return zero, not the number of enabled functions */
2783 	ret = 0;
2784  out:
2785 	mutex_unlock(&event_mutex);
2786 	return ret;
2787 
2788  out_disable:
2789 	__ftrace_event_enable_disable(file, 0, 1);
2790  out_put:
2791 	module_put(file->event_call->mod);
2792  out_free:
2793 	kfree(data);
2794 	goto out;
2795 }
2796 
2797 static struct ftrace_func_command event_enable_cmd = {
2798 	.name			= ENABLE_EVENT_STR,
2799 	.func			= event_enable_func,
2800 };
2801 
2802 static struct ftrace_func_command event_disable_cmd = {
2803 	.name			= DISABLE_EVENT_STR,
2804 	.func			= event_enable_func,
2805 };
2806 
register_event_cmds(void)2807 static __init int register_event_cmds(void)
2808 {
2809 	int ret;
2810 
2811 	ret = register_ftrace_command(&event_enable_cmd);
2812 	if (WARN_ON(ret < 0))
2813 		return ret;
2814 	ret = register_ftrace_command(&event_disable_cmd);
2815 	if (WARN_ON(ret < 0))
2816 		unregister_ftrace_command(&event_enable_cmd);
2817 	return ret;
2818 }
2819 #else
register_event_cmds(void)2820 static inline int register_event_cmds(void) { return 0; }
2821 #endif /* CONFIG_DYNAMIC_FTRACE */
2822 
2823 /*
2824  * The top level array has already had its trace_event_file
2825  * descriptors created in order to allow for early events to
2826  * be recorded. This function is called after the tracefs has been
2827  * initialized, and we now have to create the files associated
2828  * to the events.
2829  */
2830 static __init void
__trace_early_add_event_dirs(struct trace_array * tr)2831 __trace_early_add_event_dirs(struct trace_array *tr)
2832 {
2833 	struct trace_event_file *file;
2834 	int ret;
2835 
2836 
2837 	list_for_each_entry(file, &tr->events, list) {
2838 		ret = event_create_dir(tr->event_dir, file);
2839 		if (ret < 0)
2840 			pr_warn("Could not create directory for event %s\n",
2841 				trace_event_name(file->event_call));
2842 	}
2843 }
2844 
2845 /*
2846  * For early boot up, the top trace array requires to have
2847  * a list of events that can be enabled. This must be done before
2848  * the filesystem is set up in order to allow events to be traced
2849  * early.
2850  */
2851 static __init void
__trace_early_add_events(struct trace_array * tr)2852 __trace_early_add_events(struct trace_array *tr)
2853 {
2854 	struct trace_event_call *call;
2855 	int ret;
2856 
2857 	list_for_each_entry(call, &ftrace_events, list) {
2858 		/* Early boot up should not have any modules loaded */
2859 		if (WARN_ON_ONCE(call->mod))
2860 			continue;
2861 
2862 		ret = __trace_early_add_new_event(call, tr);
2863 		if (ret < 0)
2864 			pr_warn("Could not create early event %s\n",
2865 				trace_event_name(call));
2866 	}
2867 }
2868 
2869 /* Remove the event directory structure for a trace directory. */
2870 static void
__trace_remove_event_dirs(struct trace_array * tr)2871 __trace_remove_event_dirs(struct trace_array *tr)
2872 {
2873 	struct trace_event_file *file, *next;
2874 
2875 	list_for_each_entry_safe(file, next, &tr->events, list)
2876 		remove_event_file_dir(file);
2877 }
2878 
__add_event_to_tracers(struct trace_event_call * call)2879 static void __add_event_to_tracers(struct trace_event_call *call)
2880 {
2881 	struct trace_array *tr;
2882 
2883 	list_for_each_entry(tr, &ftrace_trace_arrays, list)
2884 		__trace_add_new_event(call, tr);
2885 }
2886 
2887 extern struct trace_event_call *__start_ftrace_events[];
2888 extern struct trace_event_call *__stop_ftrace_events[];
2889 
2890 static char bootup_event_buf[COMMAND_LINE_SIZE] __initdata;
2891 
setup_trace_event(char * str)2892 static __init int setup_trace_event(char *str)
2893 {
2894 	strlcpy(bootup_event_buf, str, COMMAND_LINE_SIZE);
2895 	ring_buffer_expanded = true;
2896 	tracing_selftest_disabled = true;
2897 
2898 	return 1;
2899 }
2900 __setup("trace_event=", setup_trace_event);
2901 
2902 /* Expects to have event_mutex held when called */
2903 static int
create_event_toplevel_files(struct dentry * parent,struct trace_array * tr)2904 create_event_toplevel_files(struct dentry *parent, struct trace_array *tr)
2905 {
2906 	struct dentry *d_events;
2907 	struct dentry *entry;
2908 
2909 	entry = tracefs_create_file("set_event", 0644, parent,
2910 				    tr, &ftrace_set_event_fops);
2911 	if (!entry) {
2912 		pr_warn("Could not create tracefs 'set_event' entry\n");
2913 		return -ENOMEM;
2914 	}
2915 
2916 	d_events = tracefs_create_dir("events", parent);
2917 	if (!d_events) {
2918 		pr_warn("Could not create tracefs 'events' directory\n");
2919 		return -ENOMEM;
2920 	}
2921 
2922 	entry = trace_create_file("enable", 0644, d_events,
2923 				  tr, &ftrace_tr_enable_fops);
2924 	if (!entry) {
2925 		pr_warn("Could not create tracefs 'enable' entry\n");
2926 		return -ENOMEM;
2927 	}
2928 
2929 	/* There are not as crucial, just warn if they are not created */
2930 
2931 	entry = tracefs_create_file("set_event_pid", 0644, parent,
2932 				    tr, &ftrace_set_event_pid_fops);
2933 	if (!entry)
2934 		pr_warn("Could not create tracefs 'set_event_pid' entry\n");
2935 
2936 	/* ring buffer internal formats */
2937 	entry = trace_create_file("header_page", 0444, d_events,
2938 				  ring_buffer_print_page_header,
2939 				  &ftrace_show_header_fops);
2940 	if (!entry)
2941 		pr_warn("Could not create tracefs 'header_page' entry\n");
2942 
2943 	entry = trace_create_file("header_event", 0444, d_events,
2944 				  ring_buffer_print_entry_header,
2945 				  &ftrace_show_header_fops);
2946 	if (!entry)
2947 		pr_warn("Could not create tracefs 'header_event' entry\n");
2948 
2949 	tr->event_dir = d_events;
2950 
2951 	return 0;
2952 }
2953 
2954 /**
2955  * event_trace_add_tracer - add a instance of a trace_array to events
2956  * @parent: The parent dentry to place the files/directories for events in
2957  * @tr: The trace array associated with these events
2958  *
2959  * When a new instance is created, it needs to set up its events
2960  * directory, as well as other files associated with events. It also
2961  * creates the event hierachry in the @parent/events directory.
2962  *
2963  * Returns 0 on success.
2964  */
event_trace_add_tracer(struct dentry * parent,struct trace_array * tr)2965 int event_trace_add_tracer(struct dentry *parent, struct trace_array *tr)
2966 {
2967 	int ret;
2968 
2969 	mutex_lock(&event_mutex);
2970 
2971 	ret = create_event_toplevel_files(parent, tr);
2972 	if (ret)
2973 		goto out_unlock;
2974 
2975 	down_write(&trace_event_sem);
2976 	__trace_add_event_dirs(tr);
2977 	up_write(&trace_event_sem);
2978 
2979  out_unlock:
2980 	mutex_unlock(&event_mutex);
2981 
2982 	return ret;
2983 }
2984 
2985 /*
2986  * The top trace array already had its file descriptors created.
2987  * Now the files themselves need to be created.
2988  */
2989 static __init int
early_event_add_tracer(struct dentry * parent,struct trace_array * tr)2990 early_event_add_tracer(struct dentry *parent, struct trace_array *tr)
2991 {
2992 	int ret;
2993 
2994 	mutex_lock(&event_mutex);
2995 
2996 	ret = create_event_toplevel_files(parent, tr);
2997 	if (ret)
2998 		goto out_unlock;
2999 
3000 	down_write(&trace_event_sem);
3001 	__trace_early_add_event_dirs(tr);
3002 	up_write(&trace_event_sem);
3003 
3004  out_unlock:
3005 	mutex_unlock(&event_mutex);
3006 
3007 	return ret;
3008 }
3009 
event_trace_del_tracer(struct trace_array * tr)3010 int event_trace_del_tracer(struct trace_array *tr)
3011 {
3012 	mutex_lock(&event_mutex);
3013 
3014 	/* Disable any event triggers and associated soft-disabled events */
3015 	clear_event_triggers(tr);
3016 
3017 	/* Clear the pid list */
3018 	__ftrace_clear_event_pids(tr);
3019 
3020 	/* Disable any running events */
3021 	__ftrace_set_clr_event_nolock(tr, NULL, NULL, NULL, 0);
3022 
3023 	/* Access to events are within rcu_read_lock_sched() */
3024 	synchronize_sched();
3025 
3026 	down_write(&trace_event_sem);
3027 	__trace_remove_event_dirs(tr);
3028 	tracefs_remove_recursive(tr->event_dir);
3029 	up_write(&trace_event_sem);
3030 
3031 	tr->event_dir = NULL;
3032 
3033 	mutex_unlock(&event_mutex);
3034 
3035 	return 0;
3036 }
3037 
event_trace_memsetup(void)3038 static __init int event_trace_memsetup(void)
3039 {
3040 	field_cachep = KMEM_CACHE(ftrace_event_field, SLAB_PANIC);
3041 	file_cachep = KMEM_CACHE(trace_event_file, SLAB_PANIC);
3042 	return 0;
3043 }
3044 
3045 static __init void
early_enable_events(struct trace_array * tr,bool disable_first)3046 early_enable_events(struct trace_array *tr, bool disable_first)
3047 {
3048 	char *buf = bootup_event_buf;
3049 	char *token;
3050 	int ret;
3051 
3052 	while (true) {
3053 		token = strsep(&buf, ",");
3054 
3055 		if (!token)
3056 			break;
3057 
3058 		if (*token) {
3059 			/* Restarting syscalls requires that we stop them first */
3060 			if (disable_first)
3061 				ftrace_set_clr_event(tr, token, 0);
3062 
3063 			ret = ftrace_set_clr_event(tr, token, 1);
3064 			if (ret)
3065 				pr_warn("Failed to enable trace event: %s\n", token);
3066 		}
3067 
3068 		/* Put back the comma to allow this to be called again */
3069 		if (buf)
3070 			*(buf - 1) = ',';
3071 	}
3072 }
3073 
event_trace_enable(void)3074 static __init int event_trace_enable(void)
3075 {
3076 	struct trace_array *tr = top_trace_array();
3077 	struct trace_event_call **iter, *call;
3078 	int ret;
3079 
3080 	if (!tr)
3081 		return -ENODEV;
3082 
3083 	for_each_event(iter, __start_ftrace_events, __stop_ftrace_events) {
3084 
3085 		call = *iter;
3086 		ret = event_init(call);
3087 		if (!ret)
3088 			list_add(&call->list, &ftrace_events);
3089 	}
3090 
3091 	/*
3092 	 * We need the top trace array to have a working set of trace
3093 	 * points at early init, before the debug files and directories
3094 	 * are created. Create the file entries now, and attach them
3095 	 * to the actual file dentries later.
3096 	 */
3097 	__trace_early_add_events(tr);
3098 
3099 	early_enable_events(tr, false);
3100 
3101 	trace_printk_start_comm();
3102 
3103 	register_event_cmds();
3104 
3105 	register_trigger_cmds();
3106 
3107 	return 0;
3108 }
3109 
3110 /*
3111  * event_trace_enable() is called from trace_event_init() first to
3112  * initialize events and perhaps start any events that are on the
3113  * command line. Unfortunately, there are some events that will not
3114  * start this early, like the system call tracepoints that need
3115  * to set the TIF_SYSCALL_TRACEPOINT flag of pid 1. But event_trace_enable()
3116  * is called before pid 1 starts, and this flag is never set, making
3117  * the syscall tracepoint never get reached, but the event is enabled
3118  * regardless (and not doing anything).
3119  */
event_trace_enable_again(void)3120 static __init int event_trace_enable_again(void)
3121 {
3122 	struct trace_array *tr;
3123 
3124 	tr = top_trace_array();
3125 	if (!tr)
3126 		return -ENODEV;
3127 
3128 	early_enable_events(tr, true);
3129 
3130 	return 0;
3131 }
3132 
3133 early_initcall(event_trace_enable_again);
3134 
event_trace_init(void)3135 static __init int event_trace_init(void)
3136 {
3137 	struct trace_array *tr;
3138 	struct dentry *d_tracer;
3139 	struct dentry *entry;
3140 	int ret;
3141 
3142 	tr = top_trace_array();
3143 	if (!tr)
3144 		return -ENODEV;
3145 
3146 	d_tracer = tracing_init_dentry();
3147 	if (IS_ERR(d_tracer))
3148 		return 0;
3149 
3150 	entry = tracefs_create_file("available_events", 0444, d_tracer,
3151 				    tr, &ftrace_avail_fops);
3152 	if (!entry)
3153 		pr_warn("Could not create tracefs 'available_events' entry\n");
3154 
3155 	if (trace_define_generic_fields())
3156 		pr_warn("tracing: Failed to allocated generic fields");
3157 
3158 	if (trace_define_common_fields())
3159 		pr_warn("tracing: Failed to allocate common fields");
3160 
3161 	ret = early_event_add_tracer(d_tracer, tr);
3162 	if (ret)
3163 		return ret;
3164 
3165 #ifdef CONFIG_MODULES
3166 	ret = register_module_notifier(&trace_module_nb);
3167 	if (ret)
3168 		pr_warn("Failed to register trace events module notifier\n");
3169 #endif
3170 	return 0;
3171 }
3172 
trace_event_init(void)3173 void __init trace_event_init(void)
3174 {
3175 	event_trace_memsetup();
3176 	init_ftrace_syscalls();
3177 	event_trace_enable();
3178 }
3179 
3180 fs_initcall(event_trace_init);
3181 
3182 #ifdef CONFIG_FTRACE_STARTUP_TEST
3183 
3184 static DEFINE_SPINLOCK(test_spinlock);
3185 static DEFINE_SPINLOCK(test_spinlock_irq);
3186 static DEFINE_MUTEX(test_mutex);
3187 
test_work(struct work_struct * dummy)3188 static __init void test_work(struct work_struct *dummy)
3189 {
3190 	spin_lock(&test_spinlock);
3191 	spin_lock_irq(&test_spinlock_irq);
3192 	udelay(1);
3193 	spin_unlock_irq(&test_spinlock_irq);
3194 	spin_unlock(&test_spinlock);
3195 
3196 	mutex_lock(&test_mutex);
3197 	msleep(1);
3198 	mutex_unlock(&test_mutex);
3199 }
3200 
event_test_thread(void * unused)3201 static __init int event_test_thread(void *unused)
3202 {
3203 	void *test_malloc;
3204 
3205 	test_malloc = kmalloc(1234, GFP_KERNEL);
3206 	if (!test_malloc)
3207 		pr_info("failed to kmalloc\n");
3208 
3209 	schedule_on_each_cpu(test_work);
3210 
3211 	kfree(test_malloc);
3212 
3213 	set_current_state(TASK_INTERRUPTIBLE);
3214 	while (!kthread_should_stop()) {
3215 		schedule();
3216 		set_current_state(TASK_INTERRUPTIBLE);
3217 	}
3218 	__set_current_state(TASK_RUNNING);
3219 
3220 	return 0;
3221 }
3222 
3223 /*
3224  * Do various things that may trigger events.
3225  */
event_test_stuff(void)3226 static __init void event_test_stuff(void)
3227 {
3228 	struct task_struct *test_thread;
3229 
3230 	test_thread = kthread_run(event_test_thread, NULL, "test-events");
3231 	msleep(1);
3232 	kthread_stop(test_thread);
3233 }
3234 
3235 /*
3236  * For every trace event defined, we will test each trace point separately,
3237  * and then by groups, and finally all trace points.
3238  */
event_trace_self_tests(void)3239 static __init void event_trace_self_tests(void)
3240 {
3241 	struct trace_subsystem_dir *dir;
3242 	struct trace_event_file *file;
3243 	struct trace_event_call *call;
3244 	struct event_subsystem *system;
3245 	struct trace_array *tr;
3246 	int ret;
3247 
3248 	tr = top_trace_array();
3249 	if (!tr)
3250 		return;
3251 
3252 	pr_info("Running tests on trace events:\n");
3253 
3254 	list_for_each_entry(file, &tr->events, list) {
3255 
3256 		call = file->event_call;
3257 
3258 		/* Only test those that have a probe */
3259 		if (!call->class || !call->class->probe)
3260 			continue;
3261 
3262 /*
3263  * Testing syscall events here is pretty useless, but
3264  * we still do it if configured. But this is time consuming.
3265  * What we really need is a user thread to perform the
3266  * syscalls as we test.
3267  */
3268 #ifndef CONFIG_EVENT_TRACE_TEST_SYSCALLS
3269 		if (call->class->system &&
3270 		    strcmp(call->class->system, "syscalls") == 0)
3271 			continue;
3272 #endif
3273 
3274 		pr_info("Testing event %s: ", trace_event_name(call));
3275 
3276 		/*
3277 		 * If an event is already enabled, someone is using
3278 		 * it and the self test should not be on.
3279 		 */
3280 		if (file->flags & EVENT_FILE_FL_ENABLED) {
3281 			pr_warn("Enabled event during self test!\n");
3282 			WARN_ON_ONCE(1);
3283 			continue;
3284 		}
3285 
3286 		ftrace_event_enable_disable(file, 1);
3287 		event_test_stuff();
3288 		ftrace_event_enable_disable(file, 0);
3289 
3290 		pr_cont("OK\n");
3291 	}
3292 
3293 	/* Now test at the sub system level */
3294 
3295 	pr_info("Running tests on trace event systems:\n");
3296 
3297 	list_for_each_entry(dir, &tr->systems, list) {
3298 
3299 		system = dir->subsystem;
3300 
3301 		/* the ftrace system is special, skip it */
3302 		if (strcmp(system->name, "ftrace") == 0)
3303 			continue;
3304 
3305 		pr_info("Testing event system %s: ", system->name);
3306 
3307 		ret = __ftrace_set_clr_event(tr, NULL, system->name, NULL, 1);
3308 		if (WARN_ON_ONCE(ret)) {
3309 			pr_warn("error enabling system %s\n",
3310 				system->name);
3311 			continue;
3312 		}
3313 
3314 		event_test_stuff();
3315 
3316 		ret = __ftrace_set_clr_event(tr, NULL, system->name, NULL, 0);
3317 		if (WARN_ON_ONCE(ret)) {
3318 			pr_warn("error disabling system %s\n",
3319 				system->name);
3320 			continue;
3321 		}
3322 
3323 		pr_cont("OK\n");
3324 	}
3325 
3326 	/* Test with all events enabled */
3327 
3328 	pr_info("Running tests on all trace events:\n");
3329 	pr_info("Testing all events: ");
3330 
3331 	ret = __ftrace_set_clr_event(tr, NULL, NULL, NULL, 1);
3332 	if (WARN_ON_ONCE(ret)) {
3333 		pr_warn("error enabling all events\n");
3334 		return;
3335 	}
3336 
3337 	event_test_stuff();
3338 
3339 	/* reset sysname */
3340 	ret = __ftrace_set_clr_event(tr, NULL, NULL, NULL, 0);
3341 	if (WARN_ON_ONCE(ret)) {
3342 		pr_warn("error disabling all events\n");
3343 		return;
3344 	}
3345 
3346 	pr_cont("OK\n");
3347 }
3348 
3349 #ifdef CONFIG_FUNCTION_TRACER
3350 
3351 static DEFINE_PER_CPU(atomic_t, ftrace_test_event_disable);
3352 
3353 static struct trace_event_file event_trace_file __initdata;
3354 
3355 static void __init
function_test_events_call(unsigned long ip,unsigned long parent_ip,struct ftrace_ops * op,struct pt_regs * pt_regs)3356 function_test_events_call(unsigned long ip, unsigned long parent_ip,
3357 			  struct ftrace_ops *op, struct pt_regs *pt_regs)
3358 {
3359 	struct ring_buffer_event *event;
3360 	struct ring_buffer *buffer;
3361 	struct ftrace_entry *entry;
3362 	unsigned long flags;
3363 	long disabled;
3364 	int cpu;
3365 	int pc;
3366 
3367 	pc = preempt_count();
3368 	preempt_disable_notrace();
3369 	cpu = raw_smp_processor_id();
3370 	disabled = atomic_inc_return(&per_cpu(ftrace_test_event_disable, cpu));
3371 
3372 	if (disabled != 1)
3373 		goto out;
3374 
3375 	local_save_flags(flags);
3376 
3377 	event = trace_event_buffer_lock_reserve(&buffer, &event_trace_file,
3378 						TRACE_FN, sizeof(*entry),
3379 						flags, pc);
3380 	if (!event)
3381 		goto out;
3382 	entry	= ring_buffer_event_data(event);
3383 	entry->ip			= ip;
3384 	entry->parent_ip		= parent_ip;
3385 
3386 	event_trigger_unlock_commit(&event_trace_file, buffer, event,
3387 				    entry, flags, pc);
3388  out:
3389 	atomic_dec(&per_cpu(ftrace_test_event_disable, cpu));
3390 	preempt_enable_notrace();
3391 }
3392 
3393 static struct ftrace_ops trace_ops __initdata  =
3394 {
3395 	.func = function_test_events_call,
3396 	.flags = FTRACE_OPS_FL_RECURSION_SAFE,
3397 };
3398 
event_trace_self_test_with_function(void)3399 static __init void event_trace_self_test_with_function(void)
3400 {
3401 	int ret;
3402 
3403 	event_trace_file.tr = top_trace_array();
3404 	if (WARN_ON(!event_trace_file.tr))
3405 		return;
3406 
3407 	ret = register_ftrace_function(&trace_ops);
3408 	if (WARN_ON(ret < 0)) {
3409 		pr_info("Failed to enable function tracer for event tests\n");
3410 		return;
3411 	}
3412 	pr_info("Running tests again, along with the function tracer\n");
3413 	event_trace_self_tests();
3414 	unregister_ftrace_function(&trace_ops);
3415 }
3416 #else
event_trace_self_test_with_function(void)3417 static __init void event_trace_self_test_with_function(void)
3418 {
3419 }
3420 #endif
3421 
event_trace_self_tests_init(void)3422 static __init int event_trace_self_tests_init(void)
3423 {
3424 	if (!tracing_selftest_disabled) {
3425 		event_trace_self_tests();
3426 		event_trace_self_test_with_function();
3427 	}
3428 
3429 	return 0;
3430 }
3431 
3432 late_initcall(event_trace_self_tests_init);
3433 
3434 #endif
3435