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