Loading...
1// SPDX-License-Identifier: GPL-2.0
2/*
3 * trace_events_trigger - trace event triggers
4 *
5 * Copyright (C) 2013 Tom Zanussi <tom.zanussi@linux.intel.com>
6 */
7
8#include <linux/security.h>
9#include <linux/module.h>
10#include <linux/ctype.h>
11#include <linux/mutex.h>
12#include <linux/slab.h>
13#include <linux/rculist.h>
14
15#include "trace.h"
16
17static LIST_HEAD(trigger_commands);
18static DEFINE_MUTEX(trigger_cmd_mutex);
19
20void trigger_data_free(struct event_trigger_data *data)
21{
22 if (data->cmd_ops->set_filter)
23 data->cmd_ops->set_filter(NULL, data, NULL);
24
25 /* make sure current triggers exit before free */
26 tracepoint_synchronize_unregister();
27
28 kfree(data);
29}
30
31/**
32 * event_triggers_call - Call triggers associated with a trace event
33 * @file: The trace_event_file associated with the event
34 * @rec: The trace entry for the event, NULL for unconditional invocation
35 *
36 * For each trigger associated with an event, invoke the trigger
37 * function registered with the associated trigger command. If rec is
38 * non-NULL, it means that the trigger requires further processing and
39 * shouldn't be unconditionally invoked. If rec is non-NULL and the
40 * trigger has a filter associated with it, rec will checked against
41 * the filter and if the record matches the trigger will be invoked.
42 * If the trigger is a 'post_trigger', meaning it shouldn't be invoked
43 * in any case until the current event is written, the trigger
44 * function isn't invoked but the bit associated with the deferred
45 * trigger is set in the return value.
46 *
47 * Returns an enum event_trigger_type value containing a set bit for
48 * any trigger that should be deferred, ETT_NONE if nothing to defer.
49 *
50 * Called from tracepoint handlers (with rcu_read_lock_sched() held).
51 *
52 * Return: an enum event_trigger_type value containing a set bit for
53 * any trigger that should be deferred, ETT_NONE if nothing to defer.
54 */
55enum event_trigger_type
56event_triggers_call(struct trace_event_file *file, void *rec,
57 struct ring_buffer_event *event)
58{
59 struct event_trigger_data *data;
60 enum event_trigger_type tt = ETT_NONE;
61 struct event_filter *filter;
62
63 if (list_empty(&file->triggers))
64 return tt;
65
66 list_for_each_entry_rcu(data, &file->triggers, list) {
67 if (data->paused)
68 continue;
69 if (!rec) {
70 data->ops->func(data, rec, event);
71 continue;
72 }
73 filter = rcu_dereference_sched(data->filter);
74 if (filter && !filter_match_preds(filter, rec))
75 continue;
76 if (event_command_post_trigger(data->cmd_ops)) {
77 tt |= data->cmd_ops->trigger_type;
78 continue;
79 }
80 data->ops->func(data, rec, event);
81 }
82 return tt;
83}
84EXPORT_SYMBOL_GPL(event_triggers_call);
85
86/**
87 * event_triggers_post_call - Call 'post_triggers' for a trace event
88 * @file: The trace_event_file associated with the event
89 * @tt: enum event_trigger_type containing a set bit for each trigger to invoke
90 *
91 * For each trigger associated with an event, invoke the trigger
92 * function registered with the associated trigger command, if the
93 * corresponding bit is set in the tt enum passed into this function.
94 * See @event_triggers_call for details on how those bits are set.
95 *
96 * Called from tracepoint handlers (with rcu_read_lock_sched() held).
97 */
98void
99event_triggers_post_call(struct trace_event_file *file,
100 enum event_trigger_type tt)
101{
102 struct event_trigger_data *data;
103
104 list_for_each_entry_rcu(data, &file->triggers, list) {
105 if (data->paused)
106 continue;
107 if (data->cmd_ops->trigger_type & tt)
108 data->ops->func(data, NULL, NULL);
109 }
110}
111EXPORT_SYMBOL_GPL(event_triggers_post_call);
112
113#define SHOW_AVAILABLE_TRIGGERS (void *)(1UL)
114
115static void *trigger_next(struct seq_file *m, void *t, loff_t *pos)
116{
117 struct trace_event_file *event_file = event_file_data(m->private);
118
119 if (t == SHOW_AVAILABLE_TRIGGERS)
120 return NULL;
121
122 return seq_list_next(t, &event_file->triggers, pos);
123}
124
125static void *trigger_start(struct seq_file *m, loff_t *pos)
126{
127 struct trace_event_file *event_file;
128
129 /* ->stop() is called even if ->start() fails */
130 mutex_lock(&event_mutex);
131 event_file = event_file_data(m->private);
132 if (unlikely(!event_file))
133 return ERR_PTR(-ENODEV);
134
135 if (list_empty(&event_file->triggers))
136 return *pos == 0 ? SHOW_AVAILABLE_TRIGGERS : NULL;
137
138 return seq_list_start(&event_file->triggers, *pos);
139}
140
141static void trigger_stop(struct seq_file *m, void *t)
142{
143 mutex_unlock(&event_mutex);
144}
145
146static int trigger_show(struct seq_file *m, void *v)
147{
148 struct event_trigger_data *data;
149 struct event_command *p;
150
151 if (v == SHOW_AVAILABLE_TRIGGERS) {
152 seq_puts(m, "# Available triggers:\n");
153 seq_putc(m, '#');
154 mutex_lock(&trigger_cmd_mutex);
155 list_for_each_entry_reverse(p, &trigger_commands, list)
156 seq_printf(m, " %s", p->name);
157 seq_putc(m, '\n');
158 mutex_unlock(&trigger_cmd_mutex);
159 return 0;
160 }
161
162 data = list_entry(v, struct event_trigger_data, list);
163 data->ops->print(m, data->ops, data);
164
165 return 0;
166}
167
168static const struct seq_operations event_triggers_seq_ops = {
169 .start = trigger_start,
170 .next = trigger_next,
171 .stop = trigger_stop,
172 .show = trigger_show,
173};
174
175static int event_trigger_regex_open(struct inode *inode, struct file *file)
176{
177 int ret;
178
179 ret = security_locked_down(LOCKDOWN_TRACEFS);
180 if (ret)
181 return ret;
182
183 mutex_lock(&event_mutex);
184
185 if (unlikely(!event_file_data(file))) {
186 mutex_unlock(&event_mutex);
187 return -ENODEV;
188 }
189
190 if ((file->f_mode & FMODE_WRITE) &&
191 (file->f_flags & O_TRUNC)) {
192 struct trace_event_file *event_file;
193 struct event_command *p;
194
195 event_file = event_file_data(file);
196
197 list_for_each_entry(p, &trigger_commands, list) {
198 if (p->unreg_all)
199 p->unreg_all(event_file);
200 }
201 }
202
203 if (file->f_mode & FMODE_READ) {
204 ret = seq_open(file, &event_triggers_seq_ops);
205 if (!ret) {
206 struct seq_file *m = file->private_data;
207 m->private = file;
208 }
209 }
210
211 mutex_unlock(&event_mutex);
212
213 return ret;
214}
215
216static int trigger_process_regex(struct trace_event_file *file, char *buff)
217{
218 char *command, *next = buff;
219 struct event_command *p;
220 int ret = -EINVAL;
221
222 command = strsep(&next, ": \t");
223 command = (command[0] != '!') ? command : command + 1;
224
225 mutex_lock(&trigger_cmd_mutex);
226 list_for_each_entry(p, &trigger_commands, list) {
227 if (strcmp(p->name, command) == 0) {
228 ret = p->func(p, file, buff, command, next);
229 goto out_unlock;
230 }
231 }
232 out_unlock:
233 mutex_unlock(&trigger_cmd_mutex);
234
235 return ret;
236}
237
238static ssize_t event_trigger_regex_write(struct file *file,
239 const char __user *ubuf,
240 size_t cnt, loff_t *ppos)
241{
242 struct trace_event_file *event_file;
243 ssize_t ret;
244 char *buf;
245
246 if (!cnt)
247 return 0;
248
249 if (cnt >= PAGE_SIZE)
250 return -EINVAL;
251
252 buf = memdup_user_nul(ubuf, cnt);
253 if (IS_ERR(buf))
254 return PTR_ERR(buf);
255
256 strim(buf);
257
258 mutex_lock(&event_mutex);
259 event_file = event_file_data(file);
260 if (unlikely(!event_file)) {
261 mutex_unlock(&event_mutex);
262 kfree(buf);
263 return -ENODEV;
264 }
265 ret = trigger_process_regex(event_file, buf);
266 mutex_unlock(&event_mutex);
267
268 kfree(buf);
269 if (ret < 0)
270 goto out;
271
272 *ppos += cnt;
273 ret = cnt;
274 out:
275 return ret;
276}
277
278static int event_trigger_regex_release(struct inode *inode, struct file *file)
279{
280 mutex_lock(&event_mutex);
281
282 if (file->f_mode & FMODE_READ)
283 seq_release(inode, file);
284
285 mutex_unlock(&event_mutex);
286
287 return 0;
288}
289
290static ssize_t
291event_trigger_write(struct file *filp, const char __user *ubuf,
292 size_t cnt, loff_t *ppos)
293{
294 return event_trigger_regex_write(filp, ubuf, cnt, ppos);
295}
296
297static int
298event_trigger_open(struct inode *inode, struct file *filp)
299{
300 /* Checks for tracefs lockdown */
301 return event_trigger_regex_open(inode, filp);
302}
303
304static int
305event_trigger_release(struct inode *inode, struct file *file)
306{
307 return event_trigger_regex_release(inode, file);
308}
309
310const struct file_operations event_trigger_fops = {
311 .open = event_trigger_open,
312 .read = seq_read,
313 .write = event_trigger_write,
314 .llseek = tracing_lseek,
315 .release = event_trigger_release,
316};
317
318/*
319 * Currently we only register event commands from __init, so mark this
320 * __init too.
321 */
322__init int register_event_command(struct event_command *cmd)
323{
324 struct event_command *p;
325 int ret = 0;
326
327 mutex_lock(&trigger_cmd_mutex);
328 list_for_each_entry(p, &trigger_commands, list) {
329 if (strcmp(cmd->name, p->name) == 0) {
330 ret = -EBUSY;
331 goto out_unlock;
332 }
333 }
334 list_add(&cmd->list, &trigger_commands);
335 out_unlock:
336 mutex_unlock(&trigger_cmd_mutex);
337
338 return ret;
339}
340
341/*
342 * Currently we only unregister event commands from __init, so mark
343 * this __init too.
344 */
345__init int unregister_event_command(struct event_command *cmd)
346{
347 struct event_command *p, *n;
348 int ret = -ENODEV;
349
350 mutex_lock(&trigger_cmd_mutex);
351 list_for_each_entry_safe(p, n, &trigger_commands, list) {
352 if (strcmp(cmd->name, p->name) == 0) {
353 ret = 0;
354 list_del_init(&p->list);
355 goto out_unlock;
356 }
357 }
358 out_unlock:
359 mutex_unlock(&trigger_cmd_mutex);
360
361 return ret;
362}
363
364/**
365 * event_trigger_print - Generic event_trigger_ops @print implementation
366 * @name: The name of the event trigger
367 * @m: The seq_file being printed to
368 * @data: Trigger-specific data
369 * @filter_str: filter_str to print, if present
370 *
371 * Common implementation for event triggers to print themselves.
372 *
373 * Usually wrapped by a function that simply sets the @name of the
374 * trigger command and then invokes this.
375 *
376 * Return: 0 on success, errno otherwise
377 */
378static int
379event_trigger_print(const char *name, struct seq_file *m,
380 void *data, char *filter_str)
381{
382 long count = (long)data;
383
384 seq_puts(m, name);
385
386 if (count == -1)
387 seq_puts(m, ":unlimited");
388 else
389 seq_printf(m, ":count=%ld", count);
390
391 if (filter_str)
392 seq_printf(m, " if %s\n", filter_str);
393 else
394 seq_putc(m, '\n');
395
396 return 0;
397}
398
399/**
400 * event_trigger_init - Generic event_trigger_ops @init implementation
401 * @ops: The trigger ops associated with the trigger
402 * @data: Trigger-specific data
403 *
404 * Common implementation of event trigger initialization.
405 *
406 * Usually used directly as the @init method in event trigger
407 * implementations.
408 *
409 * Return: 0 on success, errno otherwise
410 */
411int event_trigger_init(struct event_trigger_ops *ops,
412 struct event_trigger_data *data)
413{
414 data->ref++;
415 return 0;
416}
417
418/**
419 * event_trigger_free - Generic event_trigger_ops @free implementation
420 * @ops: The trigger ops associated with the trigger
421 * @data: Trigger-specific data
422 *
423 * Common implementation of event trigger de-initialization.
424 *
425 * Usually used directly as the @free method in event trigger
426 * implementations.
427 */
428static void
429event_trigger_free(struct event_trigger_ops *ops,
430 struct event_trigger_data *data)
431{
432 if (WARN_ON_ONCE(data->ref <= 0))
433 return;
434
435 data->ref--;
436 if (!data->ref)
437 trigger_data_free(data);
438}
439
440int trace_event_trigger_enable_disable(struct trace_event_file *file,
441 int trigger_enable)
442{
443 int ret = 0;
444
445 if (trigger_enable) {
446 if (atomic_inc_return(&file->tm_ref) > 1)
447 return ret;
448 set_bit(EVENT_FILE_FL_TRIGGER_MODE_BIT, &file->flags);
449 ret = trace_event_enable_disable(file, 1, 1);
450 } else {
451 if (atomic_dec_return(&file->tm_ref) > 0)
452 return ret;
453 clear_bit(EVENT_FILE_FL_TRIGGER_MODE_BIT, &file->flags);
454 ret = trace_event_enable_disable(file, 0, 1);
455 }
456
457 return ret;
458}
459
460/**
461 * clear_event_triggers - Clear all triggers associated with a trace array
462 * @tr: The trace array to clear
463 *
464 * For each trigger, the triggering event has its tm_ref decremented
465 * via trace_event_trigger_enable_disable(), and any associated event
466 * (in the case of enable/disable_event triggers) will have its sm_ref
467 * decremented via free()->trace_event_enable_disable(). That
468 * combination effectively reverses the soft-mode/trigger state added
469 * by trigger registration.
470 *
471 * Must be called with event_mutex held.
472 */
473void
474clear_event_triggers(struct trace_array *tr)
475{
476 struct trace_event_file *file;
477
478 list_for_each_entry(file, &tr->events, list) {
479 struct event_trigger_data *data, *n;
480 list_for_each_entry_safe(data, n, &file->triggers, list) {
481 trace_event_trigger_enable_disable(file, 0);
482 list_del_rcu(&data->list);
483 if (data->ops->free)
484 data->ops->free(data->ops, data);
485 }
486 }
487}
488
489/**
490 * update_cond_flag - Set or reset the TRIGGER_COND bit
491 * @file: The trace_event_file associated with the event
492 *
493 * If an event has triggers and any of those triggers has a filter or
494 * a post_trigger, trigger invocation needs to be deferred until after
495 * the current event has logged its data, and the event should have
496 * its TRIGGER_COND bit set, otherwise the TRIGGER_COND bit should be
497 * cleared.
498 */
499void update_cond_flag(struct trace_event_file *file)
500{
501 struct event_trigger_data *data;
502 bool set_cond = false;
503
504 list_for_each_entry_rcu(data, &file->triggers, list) {
505 if (data->filter || event_command_post_trigger(data->cmd_ops) ||
506 event_command_needs_rec(data->cmd_ops)) {
507 set_cond = true;
508 break;
509 }
510 }
511
512 if (set_cond)
513 set_bit(EVENT_FILE_FL_TRIGGER_COND_BIT, &file->flags);
514 else
515 clear_bit(EVENT_FILE_FL_TRIGGER_COND_BIT, &file->flags);
516}
517
518/**
519 * register_trigger - Generic event_command @reg implementation
520 * @glob: The raw string used to register the trigger
521 * @ops: The trigger ops associated with the trigger
522 * @data: Trigger-specific data to associate with the trigger
523 * @file: The trace_event_file associated with the event
524 *
525 * Common implementation for event trigger registration.
526 *
527 * Usually used directly as the @reg method in event command
528 * implementations.
529 *
530 * Return: 0 on success, errno otherwise
531 */
532static int register_trigger(char *glob, struct event_trigger_ops *ops,
533 struct event_trigger_data *data,
534 struct trace_event_file *file)
535{
536 struct event_trigger_data *test;
537 int ret = 0;
538
539 list_for_each_entry_rcu(test, &file->triggers, list) {
540 if (test->cmd_ops->trigger_type == data->cmd_ops->trigger_type) {
541 ret = -EEXIST;
542 goto out;
543 }
544 }
545
546 if (data->ops->init) {
547 ret = data->ops->init(data->ops, data);
548 if (ret < 0)
549 goto out;
550 }
551
552 list_add_rcu(&data->list, &file->triggers);
553 ret++;
554
555 update_cond_flag(file);
556 if (trace_event_trigger_enable_disable(file, 1) < 0) {
557 list_del_rcu(&data->list);
558 update_cond_flag(file);
559 ret--;
560 }
561out:
562 return ret;
563}
564
565/**
566 * unregister_trigger - Generic event_command @unreg implementation
567 * @glob: The raw string used to register the trigger
568 * @ops: The trigger ops associated with the trigger
569 * @test: Trigger-specific data used to find the trigger to remove
570 * @file: The trace_event_file associated with the event
571 *
572 * Common implementation for event trigger unregistration.
573 *
574 * Usually used directly as the @unreg method in event command
575 * implementations.
576 */
577static void unregister_trigger(char *glob, struct event_trigger_ops *ops,
578 struct event_trigger_data *test,
579 struct trace_event_file *file)
580{
581 struct event_trigger_data *data;
582 bool unregistered = false;
583
584 list_for_each_entry_rcu(data, &file->triggers, list) {
585 if (data->cmd_ops->trigger_type == test->cmd_ops->trigger_type) {
586 unregistered = true;
587 list_del_rcu(&data->list);
588 trace_event_trigger_enable_disable(file, 0);
589 update_cond_flag(file);
590 break;
591 }
592 }
593
594 if (unregistered && data->ops->free)
595 data->ops->free(data->ops, data);
596}
597
598/**
599 * event_trigger_callback - Generic event_command @func implementation
600 * @cmd_ops: The command ops, used for trigger registration
601 * @file: The trace_event_file associated with the event
602 * @glob: The raw string used to register the trigger
603 * @cmd: The cmd portion of the string used to register the trigger
604 * @param: The params portion of the string used to register the trigger
605 *
606 * Common implementation for event command parsing and trigger
607 * instantiation.
608 *
609 * Usually used directly as the @func method in event command
610 * implementations.
611 *
612 * Return: 0 on success, errno otherwise
613 */
614static int
615event_trigger_callback(struct event_command *cmd_ops,
616 struct trace_event_file *file,
617 char *glob, char *cmd, char *param)
618{
619 struct event_trigger_data *trigger_data;
620 struct event_trigger_ops *trigger_ops;
621 char *trigger = NULL;
622 char *number;
623 int ret;
624
625 /* separate the trigger from the filter (t:n [if filter]) */
626 if (param && isdigit(param[0]))
627 trigger = strsep(¶m, " \t");
628
629 trigger_ops = cmd_ops->get_trigger_ops(cmd, trigger);
630
631 ret = -ENOMEM;
632 trigger_data = kzalloc(sizeof(*trigger_data), GFP_KERNEL);
633 if (!trigger_data)
634 goto out;
635
636 trigger_data->count = -1;
637 trigger_data->ops = trigger_ops;
638 trigger_data->cmd_ops = cmd_ops;
639 trigger_data->private_data = file;
640 INIT_LIST_HEAD(&trigger_data->list);
641 INIT_LIST_HEAD(&trigger_data->named_list);
642
643 if (glob[0] == '!') {
644 cmd_ops->unreg(glob+1, trigger_ops, trigger_data, file);
645 kfree(trigger_data);
646 ret = 0;
647 goto out;
648 }
649
650 if (trigger) {
651 number = strsep(&trigger, ":");
652
653 ret = -EINVAL;
654 if (!strlen(number))
655 goto out_free;
656
657 /*
658 * We use the callback data field (which is a pointer)
659 * as our counter.
660 */
661 ret = kstrtoul(number, 0, &trigger_data->count);
662 if (ret)
663 goto out_free;
664 }
665
666 if (!param) /* if param is non-empty, it's supposed to be a filter */
667 goto out_reg;
668
669 if (!cmd_ops->set_filter)
670 goto out_reg;
671
672 ret = cmd_ops->set_filter(param, trigger_data, file);
673 if (ret < 0)
674 goto out_free;
675
676 out_reg:
677 /* Up the trigger_data count to make sure reg doesn't free it on failure */
678 event_trigger_init(trigger_ops, trigger_data);
679 ret = cmd_ops->reg(glob, trigger_ops, trigger_data, file);
680 /*
681 * The above returns on success the # of functions enabled,
682 * but if it didn't find any functions it returns zero.
683 * Consider no functions a failure too.
684 */
685 if (!ret) {
686 cmd_ops->unreg(glob, trigger_ops, trigger_data, file);
687 ret = -ENOENT;
688 } else if (ret > 0)
689 ret = 0;
690
691 /* Down the counter of trigger_data or free it if not used anymore */
692 event_trigger_free(trigger_ops, trigger_data);
693 out:
694 return ret;
695
696 out_free:
697 if (cmd_ops->set_filter)
698 cmd_ops->set_filter(NULL, trigger_data, NULL);
699 kfree(trigger_data);
700 goto out;
701}
702
703/**
704 * set_trigger_filter - Generic event_command @set_filter implementation
705 * @filter_str: The filter string for the trigger, NULL to remove filter
706 * @trigger_data: Trigger-specific data
707 * @file: The trace_event_file associated with the event
708 *
709 * Common implementation for event command filter parsing and filter
710 * instantiation.
711 *
712 * Usually used directly as the @set_filter method in event command
713 * implementations.
714 *
715 * Also used to remove a filter (if filter_str = NULL).
716 *
717 * Return: 0 on success, errno otherwise
718 */
719int set_trigger_filter(char *filter_str,
720 struct event_trigger_data *trigger_data,
721 struct trace_event_file *file)
722{
723 struct event_trigger_data *data = trigger_data;
724 struct event_filter *filter = NULL, *tmp;
725 int ret = -EINVAL;
726 char *s;
727
728 if (!filter_str) /* clear the current filter */
729 goto assign;
730
731 s = strsep(&filter_str, " \t");
732
733 if (!strlen(s) || strcmp(s, "if") != 0)
734 goto out;
735
736 if (!filter_str)
737 goto out;
738
739 /* The filter is for the 'trigger' event, not the triggered event */
740 ret = create_event_filter(file->tr, file->event_call,
741 filter_str, false, &filter);
742 /*
743 * If create_event_filter() fails, filter still needs to be freed.
744 * Which the calling code will do with data->filter.
745 */
746 assign:
747 tmp = rcu_access_pointer(data->filter);
748
749 rcu_assign_pointer(data->filter, filter);
750
751 if (tmp) {
752 /* Make sure the call is done with the filter */
753 tracepoint_synchronize_unregister();
754 free_event_filter(tmp);
755 }
756
757 kfree(data->filter_str);
758 data->filter_str = NULL;
759
760 if (filter_str) {
761 data->filter_str = kstrdup(filter_str, GFP_KERNEL);
762 if (!data->filter_str) {
763 free_event_filter(rcu_access_pointer(data->filter));
764 data->filter = NULL;
765 ret = -ENOMEM;
766 }
767 }
768 out:
769 return ret;
770}
771
772static LIST_HEAD(named_triggers);
773
774/**
775 * find_named_trigger - Find the common named trigger associated with @name
776 * @name: The name of the set of named triggers to find the common data for
777 *
778 * Named triggers are sets of triggers that share a common set of
779 * trigger data. The first named trigger registered with a given name
780 * owns the common trigger data that the others subsequently
781 * registered with the same name will reference. This function
782 * returns the common trigger data associated with that first
783 * registered instance.
784 *
785 * Return: the common trigger data for the given named trigger on
786 * success, NULL otherwise.
787 */
788struct event_trigger_data *find_named_trigger(const char *name)
789{
790 struct event_trigger_data *data;
791
792 if (!name)
793 return NULL;
794
795 list_for_each_entry(data, &named_triggers, named_list) {
796 if (data->named_data)
797 continue;
798 if (strcmp(data->name, name) == 0)
799 return data;
800 }
801
802 return NULL;
803}
804
805/**
806 * is_named_trigger - determine if a given trigger is a named trigger
807 * @test: The trigger data to test
808 *
809 * Return: true if 'test' is a named trigger, false otherwise.
810 */
811bool is_named_trigger(struct event_trigger_data *test)
812{
813 struct event_trigger_data *data;
814
815 list_for_each_entry(data, &named_triggers, named_list) {
816 if (test == data)
817 return true;
818 }
819
820 return false;
821}
822
823/**
824 * save_named_trigger - save the trigger in the named trigger list
825 * @name: The name of the named trigger set
826 * @data: The trigger data to save
827 *
828 * Return: 0 if successful, negative error otherwise.
829 */
830int save_named_trigger(const char *name, struct event_trigger_data *data)
831{
832 data->name = kstrdup(name, GFP_KERNEL);
833 if (!data->name)
834 return -ENOMEM;
835
836 list_add(&data->named_list, &named_triggers);
837
838 return 0;
839}
840
841/**
842 * del_named_trigger - delete a trigger from the named trigger list
843 * @data: The trigger data to delete
844 */
845void del_named_trigger(struct event_trigger_data *data)
846{
847 kfree(data->name);
848 data->name = NULL;
849
850 list_del(&data->named_list);
851}
852
853static void __pause_named_trigger(struct event_trigger_data *data, bool pause)
854{
855 struct event_trigger_data *test;
856
857 list_for_each_entry(test, &named_triggers, named_list) {
858 if (strcmp(test->name, data->name) == 0) {
859 if (pause) {
860 test->paused_tmp = test->paused;
861 test->paused = true;
862 } else {
863 test->paused = test->paused_tmp;
864 }
865 }
866 }
867}
868
869/**
870 * pause_named_trigger - Pause all named triggers with the same name
871 * @data: The trigger data of a named trigger to pause
872 *
873 * Pauses a named trigger along with all other triggers having the
874 * same name. Because named triggers share a common set of data,
875 * pausing only one is meaningless, so pausing one named trigger needs
876 * to pause all triggers with the same name.
877 */
878void pause_named_trigger(struct event_trigger_data *data)
879{
880 __pause_named_trigger(data, true);
881}
882
883/**
884 * unpause_named_trigger - Un-pause all named triggers with the same name
885 * @data: The trigger data of a named trigger to unpause
886 *
887 * Un-pauses a named trigger along with all other triggers having the
888 * same name. Because named triggers share a common set of data,
889 * unpausing only one is meaningless, so unpausing one named trigger
890 * needs to unpause all triggers with the same name.
891 */
892void unpause_named_trigger(struct event_trigger_data *data)
893{
894 __pause_named_trigger(data, false);
895}
896
897/**
898 * set_named_trigger_data - Associate common named trigger data
899 * @data: The trigger data of a named trigger to unpause
900 *
901 * Named triggers are sets of triggers that share a common set of
902 * trigger data. The first named trigger registered with a given name
903 * owns the common trigger data that the others subsequently
904 * registered with the same name will reference. This function
905 * associates the common trigger data from the first trigger with the
906 * given trigger.
907 */
908void set_named_trigger_data(struct event_trigger_data *data,
909 struct event_trigger_data *named_data)
910{
911 data->named_data = named_data;
912}
913
914struct event_trigger_data *
915get_named_trigger_data(struct event_trigger_data *data)
916{
917 return data->named_data;
918}
919
920static void
921traceon_trigger(struct event_trigger_data *data, void *rec,
922 struct ring_buffer_event *event)
923{
924 if (tracing_is_on())
925 return;
926
927 tracing_on();
928}
929
930static void
931traceon_count_trigger(struct event_trigger_data *data, void *rec,
932 struct ring_buffer_event *event)
933{
934 if (tracing_is_on())
935 return;
936
937 if (!data->count)
938 return;
939
940 if (data->count != -1)
941 (data->count)--;
942
943 tracing_on();
944}
945
946static void
947traceoff_trigger(struct event_trigger_data *data, void *rec,
948 struct ring_buffer_event *event)
949{
950 if (!tracing_is_on())
951 return;
952
953 tracing_off();
954}
955
956static void
957traceoff_count_trigger(struct event_trigger_data *data, void *rec,
958 struct ring_buffer_event *event)
959{
960 if (!tracing_is_on())
961 return;
962
963 if (!data->count)
964 return;
965
966 if (data->count != -1)
967 (data->count)--;
968
969 tracing_off();
970}
971
972static int
973traceon_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
974 struct event_trigger_data *data)
975{
976 return event_trigger_print("traceon", m, (void *)data->count,
977 data->filter_str);
978}
979
980static int
981traceoff_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
982 struct event_trigger_data *data)
983{
984 return event_trigger_print("traceoff", m, (void *)data->count,
985 data->filter_str);
986}
987
988static struct event_trigger_ops traceon_trigger_ops = {
989 .func = traceon_trigger,
990 .print = traceon_trigger_print,
991 .init = event_trigger_init,
992 .free = event_trigger_free,
993};
994
995static struct event_trigger_ops traceon_count_trigger_ops = {
996 .func = traceon_count_trigger,
997 .print = traceon_trigger_print,
998 .init = event_trigger_init,
999 .free = event_trigger_free,
1000};
1001
1002static struct event_trigger_ops traceoff_trigger_ops = {
1003 .func = traceoff_trigger,
1004 .print = traceoff_trigger_print,
1005 .init = event_trigger_init,
1006 .free = event_trigger_free,
1007};
1008
1009static struct event_trigger_ops traceoff_count_trigger_ops = {
1010 .func = traceoff_count_trigger,
1011 .print = traceoff_trigger_print,
1012 .init = event_trigger_init,
1013 .free = event_trigger_free,
1014};
1015
1016static struct event_trigger_ops *
1017onoff_get_trigger_ops(char *cmd, char *param)
1018{
1019 struct event_trigger_ops *ops;
1020
1021 /* we register both traceon and traceoff to this callback */
1022 if (strcmp(cmd, "traceon") == 0)
1023 ops = param ? &traceon_count_trigger_ops :
1024 &traceon_trigger_ops;
1025 else
1026 ops = param ? &traceoff_count_trigger_ops :
1027 &traceoff_trigger_ops;
1028
1029 return ops;
1030}
1031
1032static struct event_command trigger_traceon_cmd = {
1033 .name = "traceon",
1034 .trigger_type = ETT_TRACE_ONOFF,
1035 .func = event_trigger_callback,
1036 .reg = register_trigger,
1037 .unreg = unregister_trigger,
1038 .get_trigger_ops = onoff_get_trigger_ops,
1039 .set_filter = set_trigger_filter,
1040};
1041
1042static struct event_command trigger_traceoff_cmd = {
1043 .name = "traceoff",
1044 .trigger_type = ETT_TRACE_ONOFF,
1045 .flags = EVENT_CMD_FL_POST_TRIGGER,
1046 .func = event_trigger_callback,
1047 .reg = register_trigger,
1048 .unreg = unregister_trigger,
1049 .get_trigger_ops = onoff_get_trigger_ops,
1050 .set_filter = set_trigger_filter,
1051};
1052
1053#ifdef CONFIG_TRACER_SNAPSHOT
1054static void
1055snapshot_trigger(struct event_trigger_data *data, void *rec,
1056 struct ring_buffer_event *event)
1057{
1058 struct trace_event_file *file = data->private_data;
1059
1060 if (file)
1061 tracing_snapshot_instance(file->tr);
1062 else
1063 tracing_snapshot();
1064}
1065
1066static void
1067snapshot_count_trigger(struct event_trigger_data *data, void *rec,
1068 struct ring_buffer_event *event)
1069{
1070 if (!data->count)
1071 return;
1072
1073 if (data->count != -1)
1074 (data->count)--;
1075
1076 snapshot_trigger(data, rec, event);
1077}
1078
1079static int
1080register_snapshot_trigger(char *glob, struct event_trigger_ops *ops,
1081 struct event_trigger_data *data,
1082 struct trace_event_file *file)
1083{
1084 int ret = register_trigger(glob, ops, data, file);
1085
1086 if (ret > 0 && tracing_alloc_snapshot_instance(file->tr) != 0) {
1087 unregister_trigger(glob, ops, data, file);
1088 ret = 0;
1089 }
1090
1091 return ret;
1092}
1093
1094static int
1095snapshot_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
1096 struct event_trigger_data *data)
1097{
1098 return event_trigger_print("snapshot", m, (void *)data->count,
1099 data->filter_str);
1100}
1101
1102static struct event_trigger_ops snapshot_trigger_ops = {
1103 .func = snapshot_trigger,
1104 .print = snapshot_trigger_print,
1105 .init = event_trigger_init,
1106 .free = event_trigger_free,
1107};
1108
1109static struct event_trigger_ops snapshot_count_trigger_ops = {
1110 .func = snapshot_count_trigger,
1111 .print = snapshot_trigger_print,
1112 .init = event_trigger_init,
1113 .free = event_trigger_free,
1114};
1115
1116static struct event_trigger_ops *
1117snapshot_get_trigger_ops(char *cmd, char *param)
1118{
1119 return param ? &snapshot_count_trigger_ops : &snapshot_trigger_ops;
1120}
1121
1122static struct event_command trigger_snapshot_cmd = {
1123 .name = "snapshot",
1124 .trigger_type = ETT_SNAPSHOT,
1125 .func = event_trigger_callback,
1126 .reg = register_snapshot_trigger,
1127 .unreg = unregister_trigger,
1128 .get_trigger_ops = snapshot_get_trigger_ops,
1129 .set_filter = set_trigger_filter,
1130};
1131
1132static __init int register_trigger_snapshot_cmd(void)
1133{
1134 int ret;
1135
1136 ret = register_event_command(&trigger_snapshot_cmd);
1137 WARN_ON(ret < 0);
1138
1139 return ret;
1140}
1141#else
1142static __init int register_trigger_snapshot_cmd(void) { return 0; }
1143#endif /* CONFIG_TRACER_SNAPSHOT */
1144
1145#ifdef CONFIG_STACKTRACE
1146#ifdef CONFIG_UNWINDER_ORC
1147/* Skip 2:
1148 * event_triggers_post_call()
1149 * trace_event_raw_event_xxx()
1150 */
1151# define STACK_SKIP 2
1152#else
1153/*
1154 * Skip 4:
1155 * stacktrace_trigger()
1156 * event_triggers_post_call()
1157 * trace_event_buffer_commit()
1158 * trace_event_raw_event_xxx()
1159 */
1160#define STACK_SKIP 4
1161#endif
1162
1163static void
1164stacktrace_trigger(struct event_trigger_data *data, void *rec,
1165 struct ring_buffer_event *event)
1166{
1167 trace_dump_stack(STACK_SKIP);
1168}
1169
1170static void
1171stacktrace_count_trigger(struct event_trigger_data *data, void *rec,
1172 struct ring_buffer_event *event)
1173{
1174 if (!data->count)
1175 return;
1176
1177 if (data->count != -1)
1178 (data->count)--;
1179
1180 stacktrace_trigger(data, rec, event);
1181}
1182
1183static int
1184stacktrace_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
1185 struct event_trigger_data *data)
1186{
1187 return event_trigger_print("stacktrace", m, (void *)data->count,
1188 data->filter_str);
1189}
1190
1191static struct event_trigger_ops stacktrace_trigger_ops = {
1192 .func = stacktrace_trigger,
1193 .print = stacktrace_trigger_print,
1194 .init = event_trigger_init,
1195 .free = event_trigger_free,
1196};
1197
1198static struct event_trigger_ops stacktrace_count_trigger_ops = {
1199 .func = stacktrace_count_trigger,
1200 .print = stacktrace_trigger_print,
1201 .init = event_trigger_init,
1202 .free = event_trigger_free,
1203};
1204
1205static struct event_trigger_ops *
1206stacktrace_get_trigger_ops(char *cmd, char *param)
1207{
1208 return param ? &stacktrace_count_trigger_ops : &stacktrace_trigger_ops;
1209}
1210
1211static struct event_command trigger_stacktrace_cmd = {
1212 .name = "stacktrace",
1213 .trigger_type = ETT_STACKTRACE,
1214 .flags = EVENT_CMD_FL_POST_TRIGGER,
1215 .func = event_trigger_callback,
1216 .reg = register_trigger,
1217 .unreg = unregister_trigger,
1218 .get_trigger_ops = stacktrace_get_trigger_ops,
1219 .set_filter = set_trigger_filter,
1220};
1221
1222static __init int register_trigger_stacktrace_cmd(void)
1223{
1224 int ret;
1225
1226 ret = register_event_command(&trigger_stacktrace_cmd);
1227 WARN_ON(ret < 0);
1228
1229 return ret;
1230}
1231#else
1232static __init int register_trigger_stacktrace_cmd(void) { return 0; }
1233#endif /* CONFIG_STACKTRACE */
1234
1235static __init void unregister_trigger_traceon_traceoff_cmds(void)
1236{
1237 unregister_event_command(&trigger_traceon_cmd);
1238 unregister_event_command(&trigger_traceoff_cmd);
1239}
1240
1241static void
1242event_enable_trigger(struct event_trigger_data *data, void *rec,
1243 struct ring_buffer_event *event)
1244{
1245 struct enable_trigger_data *enable_data = data->private_data;
1246
1247 if (enable_data->enable)
1248 clear_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &enable_data->file->flags);
1249 else
1250 set_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &enable_data->file->flags);
1251}
1252
1253static void
1254event_enable_count_trigger(struct event_trigger_data *data, void *rec,
1255 struct ring_buffer_event *event)
1256{
1257 struct enable_trigger_data *enable_data = data->private_data;
1258
1259 if (!data->count)
1260 return;
1261
1262 /* Skip if the event is in a state we want to switch to */
1263 if (enable_data->enable == !(enable_data->file->flags & EVENT_FILE_FL_SOFT_DISABLED))
1264 return;
1265
1266 if (data->count != -1)
1267 (data->count)--;
1268
1269 event_enable_trigger(data, rec, event);
1270}
1271
1272int event_enable_trigger_print(struct seq_file *m,
1273 struct event_trigger_ops *ops,
1274 struct event_trigger_data *data)
1275{
1276 struct enable_trigger_data *enable_data = data->private_data;
1277
1278 seq_printf(m, "%s:%s:%s",
1279 enable_data->hist ?
1280 (enable_data->enable ? ENABLE_HIST_STR : DISABLE_HIST_STR) :
1281 (enable_data->enable ? ENABLE_EVENT_STR : DISABLE_EVENT_STR),
1282 enable_data->file->event_call->class->system,
1283 trace_event_name(enable_data->file->event_call));
1284
1285 if (data->count == -1)
1286 seq_puts(m, ":unlimited");
1287 else
1288 seq_printf(m, ":count=%ld", data->count);
1289
1290 if (data->filter_str)
1291 seq_printf(m, " if %s\n", data->filter_str);
1292 else
1293 seq_putc(m, '\n');
1294
1295 return 0;
1296}
1297
1298void event_enable_trigger_free(struct event_trigger_ops *ops,
1299 struct event_trigger_data *data)
1300{
1301 struct enable_trigger_data *enable_data = data->private_data;
1302
1303 if (WARN_ON_ONCE(data->ref <= 0))
1304 return;
1305
1306 data->ref--;
1307 if (!data->ref) {
1308 /* Remove the SOFT_MODE flag */
1309 trace_event_enable_disable(enable_data->file, 0, 1);
1310 module_put(enable_data->file->event_call->mod);
1311 trigger_data_free(data);
1312 kfree(enable_data);
1313 }
1314}
1315
1316static struct event_trigger_ops event_enable_trigger_ops = {
1317 .func = event_enable_trigger,
1318 .print = event_enable_trigger_print,
1319 .init = event_trigger_init,
1320 .free = event_enable_trigger_free,
1321};
1322
1323static struct event_trigger_ops event_enable_count_trigger_ops = {
1324 .func = event_enable_count_trigger,
1325 .print = event_enable_trigger_print,
1326 .init = event_trigger_init,
1327 .free = event_enable_trigger_free,
1328};
1329
1330static struct event_trigger_ops event_disable_trigger_ops = {
1331 .func = event_enable_trigger,
1332 .print = event_enable_trigger_print,
1333 .init = event_trigger_init,
1334 .free = event_enable_trigger_free,
1335};
1336
1337static struct event_trigger_ops event_disable_count_trigger_ops = {
1338 .func = event_enable_count_trigger,
1339 .print = event_enable_trigger_print,
1340 .init = event_trigger_init,
1341 .free = event_enable_trigger_free,
1342};
1343
1344int event_enable_trigger_func(struct event_command *cmd_ops,
1345 struct trace_event_file *file,
1346 char *glob, char *cmd, char *param)
1347{
1348 struct trace_event_file *event_enable_file;
1349 struct enable_trigger_data *enable_data;
1350 struct event_trigger_data *trigger_data;
1351 struct event_trigger_ops *trigger_ops;
1352 struct trace_array *tr = file->tr;
1353 const char *system;
1354 const char *event;
1355 bool hist = false;
1356 char *trigger;
1357 char *number;
1358 bool enable;
1359 int ret;
1360
1361 if (!param)
1362 return -EINVAL;
1363
1364 /* separate the trigger from the filter (s:e:n [if filter]) */
1365 trigger = strsep(¶m, " \t");
1366 if (!trigger)
1367 return -EINVAL;
1368
1369 system = strsep(&trigger, ":");
1370 if (!trigger)
1371 return -EINVAL;
1372
1373 event = strsep(&trigger, ":");
1374
1375 ret = -EINVAL;
1376 event_enable_file = find_event_file(tr, system, event);
1377 if (!event_enable_file)
1378 goto out;
1379
1380#ifdef CONFIG_HIST_TRIGGERS
1381 hist = ((strcmp(cmd, ENABLE_HIST_STR) == 0) ||
1382 (strcmp(cmd, DISABLE_HIST_STR) == 0));
1383
1384 enable = ((strcmp(cmd, ENABLE_EVENT_STR) == 0) ||
1385 (strcmp(cmd, ENABLE_HIST_STR) == 0));
1386#else
1387 enable = strcmp(cmd, ENABLE_EVENT_STR) == 0;
1388#endif
1389 trigger_ops = cmd_ops->get_trigger_ops(cmd, trigger);
1390
1391 ret = -ENOMEM;
1392 trigger_data = kzalloc(sizeof(*trigger_data), GFP_KERNEL);
1393 if (!trigger_data)
1394 goto out;
1395
1396 enable_data = kzalloc(sizeof(*enable_data), GFP_KERNEL);
1397 if (!enable_data) {
1398 kfree(trigger_data);
1399 goto out;
1400 }
1401
1402 trigger_data->count = -1;
1403 trigger_data->ops = trigger_ops;
1404 trigger_data->cmd_ops = cmd_ops;
1405 INIT_LIST_HEAD(&trigger_data->list);
1406 RCU_INIT_POINTER(trigger_data->filter, NULL);
1407
1408 enable_data->hist = hist;
1409 enable_data->enable = enable;
1410 enable_data->file = event_enable_file;
1411 trigger_data->private_data = enable_data;
1412
1413 if (glob[0] == '!') {
1414 cmd_ops->unreg(glob+1, trigger_ops, trigger_data, file);
1415 kfree(trigger_data);
1416 kfree(enable_data);
1417 ret = 0;
1418 goto out;
1419 }
1420
1421 /* Up the trigger_data count to make sure nothing frees it on failure */
1422 event_trigger_init(trigger_ops, trigger_data);
1423
1424 if (trigger) {
1425 number = strsep(&trigger, ":");
1426
1427 ret = -EINVAL;
1428 if (!strlen(number))
1429 goto out_free;
1430
1431 /*
1432 * We use the callback data field (which is a pointer)
1433 * as our counter.
1434 */
1435 ret = kstrtoul(number, 0, &trigger_data->count);
1436 if (ret)
1437 goto out_free;
1438 }
1439
1440 if (!param) /* if param is non-empty, it's supposed to be a filter */
1441 goto out_reg;
1442
1443 if (!cmd_ops->set_filter)
1444 goto out_reg;
1445
1446 ret = cmd_ops->set_filter(param, trigger_data, file);
1447 if (ret < 0)
1448 goto out_free;
1449
1450 out_reg:
1451 /* Don't let event modules unload while probe registered */
1452 ret = try_module_get(event_enable_file->event_call->mod);
1453 if (!ret) {
1454 ret = -EBUSY;
1455 goto out_free;
1456 }
1457
1458 ret = trace_event_enable_disable(event_enable_file, 1, 1);
1459 if (ret < 0)
1460 goto out_put;
1461 ret = cmd_ops->reg(glob, trigger_ops, trigger_data, file);
1462 /*
1463 * The above returns on success the # of functions enabled,
1464 * but if it didn't find any functions it returns zero.
1465 * Consider no functions a failure too.
1466 */
1467 if (!ret) {
1468 ret = -ENOENT;
1469 goto out_disable;
1470 } else if (ret < 0)
1471 goto out_disable;
1472 /* Just return zero, not the number of enabled functions */
1473 ret = 0;
1474 event_trigger_free(trigger_ops, trigger_data);
1475 out:
1476 return ret;
1477
1478 out_disable:
1479 trace_event_enable_disable(event_enable_file, 0, 1);
1480 out_put:
1481 module_put(event_enable_file->event_call->mod);
1482 out_free:
1483 if (cmd_ops->set_filter)
1484 cmd_ops->set_filter(NULL, trigger_data, NULL);
1485 event_trigger_free(trigger_ops, trigger_data);
1486 kfree(enable_data);
1487 goto out;
1488}
1489
1490int event_enable_register_trigger(char *glob,
1491 struct event_trigger_ops *ops,
1492 struct event_trigger_data *data,
1493 struct trace_event_file *file)
1494{
1495 struct enable_trigger_data *enable_data = data->private_data;
1496 struct enable_trigger_data *test_enable_data;
1497 struct event_trigger_data *test;
1498 int ret = 0;
1499
1500 list_for_each_entry_rcu(test, &file->triggers, list) {
1501 test_enable_data = test->private_data;
1502 if (test_enable_data &&
1503 (test->cmd_ops->trigger_type ==
1504 data->cmd_ops->trigger_type) &&
1505 (test_enable_data->file == enable_data->file)) {
1506 ret = -EEXIST;
1507 goto out;
1508 }
1509 }
1510
1511 if (data->ops->init) {
1512 ret = data->ops->init(data->ops, data);
1513 if (ret < 0)
1514 goto out;
1515 }
1516
1517 list_add_rcu(&data->list, &file->triggers);
1518 ret++;
1519
1520 update_cond_flag(file);
1521 if (trace_event_trigger_enable_disable(file, 1) < 0) {
1522 list_del_rcu(&data->list);
1523 update_cond_flag(file);
1524 ret--;
1525 }
1526out:
1527 return ret;
1528}
1529
1530void event_enable_unregister_trigger(char *glob,
1531 struct event_trigger_ops *ops,
1532 struct event_trigger_data *test,
1533 struct trace_event_file *file)
1534{
1535 struct enable_trigger_data *test_enable_data = test->private_data;
1536 struct enable_trigger_data *enable_data;
1537 struct event_trigger_data *data;
1538 bool unregistered = false;
1539
1540 list_for_each_entry_rcu(data, &file->triggers, list) {
1541 enable_data = data->private_data;
1542 if (enable_data &&
1543 (data->cmd_ops->trigger_type ==
1544 test->cmd_ops->trigger_type) &&
1545 (enable_data->file == test_enable_data->file)) {
1546 unregistered = true;
1547 list_del_rcu(&data->list);
1548 trace_event_trigger_enable_disable(file, 0);
1549 update_cond_flag(file);
1550 break;
1551 }
1552 }
1553
1554 if (unregistered && data->ops->free)
1555 data->ops->free(data->ops, data);
1556}
1557
1558static struct event_trigger_ops *
1559event_enable_get_trigger_ops(char *cmd, char *param)
1560{
1561 struct event_trigger_ops *ops;
1562 bool enable;
1563
1564#ifdef CONFIG_HIST_TRIGGERS
1565 enable = ((strcmp(cmd, ENABLE_EVENT_STR) == 0) ||
1566 (strcmp(cmd, ENABLE_HIST_STR) == 0));
1567#else
1568 enable = strcmp(cmd, ENABLE_EVENT_STR) == 0;
1569#endif
1570 if (enable)
1571 ops = param ? &event_enable_count_trigger_ops :
1572 &event_enable_trigger_ops;
1573 else
1574 ops = param ? &event_disable_count_trigger_ops :
1575 &event_disable_trigger_ops;
1576
1577 return ops;
1578}
1579
1580static struct event_command trigger_enable_cmd = {
1581 .name = ENABLE_EVENT_STR,
1582 .trigger_type = ETT_EVENT_ENABLE,
1583 .func = event_enable_trigger_func,
1584 .reg = event_enable_register_trigger,
1585 .unreg = event_enable_unregister_trigger,
1586 .get_trigger_ops = event_enable_get_trigger_ops,
1587 .set_filter = set_trigger_filter,
1588};
1589
1590static struct event_command trigger_disable_cmd = {
1591 .name = DISABLE_EVENT_STR,
1592 .trigger_type = ETT_EVENT_ENABLE,
1593 .func = event_enable_trigger_func,
1594 .reg = event_enable_register_trigger,
1595 .unreg = event_enable_unregister_trigger,
1596 .get_trigger_ops = event_enable_get_trigger_ops,
1597 .set_filter = set_trigger_filter,
1598};
1599
1600static __init void unregister_trigger_enable_disable_cmds(void)
1601{
1602 unregister_event_command(&trigger_enable_cmd);
1603 unregister_event_command(&trigger_disable_cmd);
1604}
1605
1606static __init int register_trigger_enable_disable_cmds(void)
1607{
1608 int ret;
1609
1610 ret = register_event_command(&trigger_enable_cmd);
1611 if (WARN_ON(ret < 0))
1612 return ret;
1613 ret = register_event_command(&trigger_disable_cmd);
1614 if (WARN_ON(ret < 0))
1615 unregister_trigger_enable_disable_cmds();
1616
1617 return ret;
1618}
1619
1620static __init int register_trigger_traceon_traceoff_cmds(void)
1621{
1622 int ret;
1623
1624 ret = register_event_command(&trigger_traceon_cmd);
1625 if (WARN_ON(ret < 0))
1626 return ret;
1627 ret = register_event_command(&trigger_traceoff_cmd);
1628 if (WARN_ON(ret < 0))
1629 unregister_trigger_traceon_traceoff_cmds();
1630
1631 return ret;
1632}
1633
1634__init int register_trigger_cmds(void)
1635{
1636 register_trigger_traceon_traceoff_cmds();
1637 register_trigger_snapshot_cmd();
1638 register_trigger_stacktrace_cmd();
1639 register_trigger_enable_disable_cmds();
1640 register_trigger_hist_enable_disable_cmds();
1641 register_trigger_hist_cmd();
1642
1643 return 0;
1644}
1/*
2 * trace_events_trigger - trace event triggers
3 *
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation; either version 2 of the License, or
7 * (at your option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with this program; if not, write to the Free Software
16 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17 *
18 * Copyright (C) 2013 Tom Zanussi <tom.zanussi@linux.intel.com>
19 */
20
21#include <linux/module.h>
22#include <linux/ctype.h>
23#include <linux/mutex.h>
24#include <linux/slab.h>
25
26#include "trace.h"
27
28static LIST_HEAD(trigger_commands);
29static DEFINE_MUTEX(trigger_cmd_mutex);
30
31void trigger_data_free(struct event_trigger_data *data)
32{
33 if (data->cmd_ops->set_filter)
34 data->cmd_ops->set_filter(NULL, data, NULL);
35
36 synchronize_sched(); /* make sure current triggers exit before free */
37 kfree(data);
38}
39
40/**
41 * event_triggers_call - Call triggers associated with a trace event
42 * @file: The trace_event_file associated with the event
43 * @rec: The trace entry for the event, NULL for unconditional invocation
44 *
45 * For each trigger associated with an event, invoke the trigger
46 * function registered with the associated trigger command. If rec is
47 * non-NULL, it means that the trigger requires further processing and
48 * shouldn't be unconditionally invoked. If rec is non-NULL and the
49 * trigger has a filter associated with it, rec will checked against
50 * the filter and if the record matches the trigger will be invoked.
51 * If the trigger is a 'post_trigger', meaning it shouldn't be invoked
52 * in any case until the current event is written, the trigger
53 * function isn't invoked but the bit associated with the deferred
54 * trigger is set in the return value.
55 *
56 * Returns an enum event_trigger_type value containing a set bit for
57 * any trigger that should be deferred, ETT_NONE if nothing to defer.
58 *
59 * Called from tracepoint handlers (with rcu_read_lock_sched() held).
60 *
61 * Return: an enum event_trigger_type value containing a set bit for
62 * any trigger that should be deferred, ETT_NONE if nothing to defer.
63 */
64enum event_trigger_type
65event_triggers_call(struct trace_event_file *file, void *rec)
66{
67 struct event_trigger_data *data;
68 enum event_trigger_type tt = ETT_NONE;
69 struct event_filter *filter;
70
71 if (list_empty(&file->triggers))
72 return tt;
73
74 list_for_each_entry_rcu(data, &file->triggers, list) {
75 if (data->paused)
76 continue;
77 if (!rec) {
78 data->ops->func(data, rec);
79 continue;
80 }
81 filter = rcu_dereference_sched(data->filter);
82 if (filter && !filter_match_preds(filter, rec))
83 continue;
84 if (event_command_post_trigger(data->cmd_ops)) {
85 tt |= data->cmd_ops->trigger_type;
86 continue;
87 }
88 data->ops->func(data, rec);
89 }
90 return tt;
91}
92EXPORT_SYMBOL_GPL(event_triggers_call);
93
94/**
95 * event_triggers_post_call - Call 'post_triggers' for a trace event
96 * @file: The trace_event_file associated with the event
97 * @tt: enum event_trigger_type containing a set bit for each trigger to invoke
98 * @rec: The trace entry for the event
99 *
100 * For each trigger associated with an event, invoke the trigger
101 * function registered with the associated trigger command, if the
102 * corresponding bit is set in the tt enum passed into this function.
103 * See @event_triggers_call for details on how those bits are set.
104 *
105 * Called from tracepoint handlers (with rcu_read_lock_sched() held).
106 */
107void
108event_triggers_post_call(struct trace_event_file *file,
109 enum event_trigger_type tt,
110 void *rec)
111{
112 struct event_trigger_data *data;
113
114 list_for_each_entry_rcu(data, &file->triggers, list) {
115 if (data->paused)
116 continue;
117 if (data->cmd_ops->trigger_type & tt)
118 data->ops->func(data, rec);
119 }
120}
121EXPORT_SYMBOL_GPL(event_triggers_post_call);
122
123#define SHOW_AVAILABLE_TRIGGERS (void *)(1UL)
124
125static void *trigger_next(struct seq_file *m, void *t, loff_t *pos)
126{
127 struct trace_event_file *event_file = event_file_data(m->private);
128
129 if (t == SHOW_AVAILABLE_TRIGGERS)
130 return NULL;
131
132 return seq_list_next(t, &event_file->triggers, pos);
133}
134
135static void *trigger_start(struct seq_file *m, loff_t *pos)
136{
137 struct trace_event_file *event_file;
138
139 /* ->stop() is called even if ->start() fails */
140 mutex_lock(&event_mutex);
141 event_file = event_file_data(m->private);
142 if (unlikely(!event_file))
143 return ERR_PTR(-ENODEV);
144
145 if (list_empty(&event_file->triggers))
146 return *pos == 0 ? SHOW_AVAILABLE_TRIGGERS : NULL;
147
148 return seq_list_start(&event_file->triggers, *pos);
149}
150
151static void trigger_stop(struct seq_file *m, void *t)
152{
153 mutex_unlock(&event_mutex);
154}
155
156static int trigger_show(struct seq_file *m, void *v)
157{
158 struct event_trigger_data *data;
159 struct event_command *p;
160
161 if (v == SHOW_AVAILABLE_TRIGGERS) {
162 seq_puts(m, "# Available triggers:\n");
163 seq_putc(m, '#');
164 mutex_lock(&trigger_cmd_mutex);
165 list_for_each_entry_reverse(p, &trigger_commands, list)
166 seq_printf(m, " %s", p->name);
167 seq_putc(m, '\n');
168 mutex_unlock(&trigger_cmd_mutex);
169 return 0;
170 }
171
172 data = list_entry(v, struct event_trigger_data, list);
173 data->ops->print(m, data->ops, data);
174
175 return 0;
176}
177
178static const struct seq_operations event_triggers_seq_ops = {
179 .start = trigger_start,
180 .next = trigger_next,
181 .stop = trigger_stop,
182 .show = trigger_show,
183};
184
185static int event_trigger_regex_open(struct inode *inode, struct file *file)
186{
187 int ret = 0;
188
189 mutex_lock(&event_mutex);
190
191 if (unlikely(!event_file_data(file))) {
192 mutex_unlock(&event_mutex);
193 return -ENODEV;
194 }
195
196 if ((file->f_mode & FMODE_WRITE) &&
197 (file->f_flags & O_TRUNC)) {
198 struct trace_event_file *event_file;
199 struct event_command *p;
200
201 event_file = event_file_data(file);
202
203 list_for_each_entry(p, &trigger_commands, list) {
204 if (p->unreg_all)
205 p->unreg_all(event_file);
206 }
207 }
208
209 if (file->f_mode & FMODE_READ) {
210 ret = seq_open(file, &event_triggers_seq_ops);
211 if (!ret) {
212 struct seq_file *m = file->private_data;
213 m->private = file;
214 }
215 }
216
217 mutex_unlock(&event_mutex);
218
219 return ret;
220}
221
222static int trigger_process_regex(struct trace_event_file *file, char *buff)
223{
224 char *command, *next = buff;
225 struct event_command *p;
226 int ret = -EINVAL;
227
228 command = strsep(&next, ": \t");
229 command = (command[0] != '!') ? command : command + 1;
230
231 mutex_lock(&trigger_cmd_mutex);
232 list_for_each_entry(p, &trigger_commands, list) {
233 if (strcmp(p->name, command) == 0) {
234 ret = p->func(p, file, buff, command, next);
235 goto out_unlock;
236 }
237 }
238 out_unlock:
239 mutex_unlock(&trigger_cmd_mutex);
240
241 return ret;
242}
243
244static ssize_t event_trigger_regex_write(struct file *file,
245 const char __user *ubuf,
246 size_t cnt, loff_t *ppos)
247{
248 struct trace_event_file *event_file;
249 ssize_t ret;
250 char *buf;
251
252 if (!cnt)
253 return 0;
254
255 if (cnt >= PAGE_SIZE)
256 return -EINVAL;
257
258 buf = memdup_user_nul(ubuf, cnt);
259 if (IS_ERR(buf))
260 return PTR_ERR(buf);
261
262 strim(buf);
263
264 mutex_lock(&event_mutex);
265 event_file = event_file_data(file);
266 if (unlikely(!event_file)) {
267 mutex_unlock(&event_mutex);
268 kfree(buf);
269 return -ENODEV;
270 }
271 ret = trigger_process_regex(event_file, buf);
272 mutex_unlock(&event_mutex);
273
274 kfree(buf);
275 if (ret < 0)
276 goto out;
277
278 *ppos += cnt;
279 ret = cnt;
280 out:
281 return ret;
282}
283
284static int event_trigger_regex_release(struct inode *inode, struct file *file)
285{
286 mutex_lock(&event_mutex);
287
288 if (file->f_mode & FMODE_READ)
289 seq_release(inode, file);
290
291 mutex_unlock(&event_mutex);
292
293 return 0;
294}
295
296static ssize_t
297event_trigger_write(struct file *filp, const char __user *ubuf,
298 size_t cnt, loff_t *ppos)
299{
300 return event_trigger_regex_write(filp, ubuf, cnt, ppos);
301}
302
303static int
304event_trigger_open(struct inode *inode, struct file *filp)
305{
306 return event_trigger_regex_open(inode, filp);
307}
308
309static int
310event_trigger_release(struct inode *inode, struct file *file)
311{
312 return event_trigger_regex_release(inode, file);
313}
314
315const struct file_operations event_trigger_fops = {
316 .open = event_trigger_open,
317 .read = seq_read,
318 .write = event_trigger_write,
319 .llseek = tracing_lseek,
320 .release = event_trigger_release,
321};
322
323/*
324 * Currently we only register event commands from __init, so mark this
325 * __init too.
326 */
327__init int register_event_command(struct event_command *cmd)
328{
329 struct event_command *p;
330 int ret = 0;
331
332 mutex_lock(&trigger_cmd_mutex);
333 list_for_each_entry(p, &trigger_commands, list) {
334 if (strcmp(cmd->name, p->name) == 0) {
335 ret = -EBUSY;
336 goto out_unlock;
337 }
338 }
339 list_add(&cmd->list, &trigger_commands);
340 out_unlock:
341 mutex_unlock(&trigger_cmd_mutex);
342
343 return ret;
344}
345
346/*
347 * Currently we only unregister event commands from __init, so mark
348 * this __init too.
349 */
350static __init int unregister_event_command(struct event_command *cmd)
351{
352 struct event_command *p, *n;
353 int ret = -ENODEV;
354
355 mutex_lock(&trigger_cmd_mutex);
356 list_for_each_entry_safe(p, n, &trigger_commands, list) {
357 if (strcmp(cmd->name, p->name) == 0) {
358 ret = 0;
359 list_del_init(&p->list);
360 goto out_unlock;
361 }
362 }
363 out_unlock:
364 mutex_unlock(&trigger_cmd_mutex);
365
366 return ret;
367}
368
369/**
370 * event_trigger_print - Generic event_trigger_ops @print implementation
371 * @name: The name of the event trigger
372 * @m: The seq_file being printed to
373 * @data: Trigger-specific data
374 * @filter_str: filter_str to print, if present
375 *
376 * Common implementation for event triggers to print themselves.
377 *
378 * Usually wrapped by a function that simply sets the @name of the
379 * trigger command and then invokes this.
380 *
381 * Return: 0 on success, errno otherwise
382 */
383static int
384event_trigger_print(const char *name, struct seq_file *m,
385 void *data, char *filter_str)
386{
387 long count = (long)data;
388
389 seq_puts(m, name);
390
391 if (count == -1)
392 seq_puts(m, ":unlimited");
393 else
394 seq_printf(m, ":count=%ld", count);
395
396 if (filter_str)
397 seq_printf(m, " if %s\n", filter_str);
398 else
399 seq_putc(m, '\n');
400
401 return 0;
402}
403
404/**
405 * event_trigger_init - Generic event_trigger_ops @init implementation
406 * @ops: The trigger ops associated with the trigger
407 * @data: Trigger-specific data
408 *
409 * Common implementation of event trigger initialization.
410 *
411 * Usually used directly as the @init method in event trigger
412 * implementations.
413 *
414 * Return: 0 on success, errno otherwise
415 */
416int event_trigger_init(struct event_trigger_ops *ops,
417 struct event_trigger_data *data)
418{
419 data->ref++;
420 return 0;
421}
422
423/**
424 * event_trigger_free - Generic event_trigger_ops @free implementation
425 * @ops: The trigger ops associated with the trigger
426 * @data: Trigger-specific data
427 *
428 * Common implementation of event trigger de-initialization.
429 *
430 * Usually used directly as the @free method in event trigger
431 * implementations.
432 */
433static void
434event_trigger_free(struct event_trigger_ops *ops,
435 struct event_trigger_data *data)
436{
437 if (WARN_ON_ONCE(data->ref <= 0))
438 return;
439
440 data->ref--;
441 if (!data->ref)
442 trigger_data_free(data);
443}
444
445int trace_event_trigger_enable_disable(struct trace_event_file *file,
446 int trigger_enable)
447{
448 int ret = 0;
449
450 if (trigger_enable) {
451 if (atomic_inc_return(&file->tm_ref) > 1)
452 return ret;
453 set_bit(EVENT_FILE_FL_TRIGGER_MODE_BIT, &file->flags);
454 ret = trace_event_enable_disable(file, 1, 1);
455 } else {
456 if (atomic_dec_return(&file->tm_ref) > 0)
457 return ret;
458 clear_bit(EVENT_FILE_FL_TRIGGER_MODE_BIT, &file->flags);
459 ret = trace_event_enable_disable(file, 0, 1);
460 }
461
462 return ret;
463}
464
465/**
466 * clear_event_triggers - Clear all triggers associated with a trace array
467 * @tr: The trace array to clear
468 *
469 * For each trigger, the triggering event has its tm_ref decremented
470 * via trace_event_trigger_enable_disable(), and any associated event
471 * (in the case of enable/disable_event triggers) will have its sm_ref
472 * decremented via free()->trace_event_enable_disable(). That
473 * combination effectively reverses the soft-mode/trigger state added
474 * by trigger registration.
475 *
476 * Must be called with event_mutex held.
477 */
478void
479clear_event_triggers(struct trace_array *tr)
480{
481 struct trace_event_file *file;
482
483 list_for_each_entry(file, &tr->events, list) {
484 struct event_trigger_data *data;
485 list_for_each_entry_rcu(data, &file->triggers, list) {
486 trace_event_trigger_enable_disable(file, 0);
487 if (data->ops->free)
488 data->ops->free(data->ops, data);
489 }
490 }
491}
492
493/**
494 * update_cond_flag - Set or reset the TRIGGER_COND bit
495 * @file: The trace_event_file associated with the event
496 *
497 * If an event has triggers and any of those triggers has a filter or
498 * a post_trigger, trigger invocation needs to be deferred until after
499 * the current event has logged its data, and the event should have
500 * its TRIGGER_COND bit set, otherwise the TRIGGER_COND bit should be
501 * cleared.
502 */
503void update_cond_flag(struct trace_event_file *file)
504{
505 struct event_trigger_data *data;
506 bool set_cond = false;
507
508 list_for_each_entry_rcu(data, &file->triggers, list) {
509 if (data->filter || event_command_post_trigger(data->cmd_ops) ||
510 event_command_needs_rec(data->cmd_ops)) {
511 set_cond = true;
512 break;
513 }
514 }
515
516 if (set_cond)
517 set_bit(EVENT_FILE_FL_TRIGGER_COND_BIT, &file->flags);
518 else
519 clear_bit(EVENT_FILE_FL_TRIGGER_COND_BIT, &file->flags);
520}
521
522/**
523 * register_trigger - Generic event_command @reg implementation
524 * @glob: The raw string used to register the trigger
525 * @ops: The trigger ops associated with the trigger
526 * @data: Trigger-specific data to associate with the trigger
527 * @file: The trace_event_file associated with the event
528 *
529 * Common implementation for event trigger registration.
530 *
531 * Usually used directly as the @reg method in event command
532 * implementations.
533 *
534 * Return: 0 on success, errno otherwise
535 */
536static int register_trigger(char *glob, struct event_trigger_ops *ops,
537 struct event_trigger_data *data,
538 struct trace_event_file *file)
539{
540 struct event_trigger_data *test;
541 int ret = 0;
542
543 list_for_each_entry_rcu(test, &file->triggers, list) {
544 if (test->cmd_ops->trigger_type == data->cmd_ops->trigger_type) {
545 ret = -EEXIST;
546 goto out;
547 }
548 }
549
550 if (data->ops->init) {
551 ret = data->ops->init(data->ops, data);
552 if (ret < 0)
553 goto out;
554 }
555
556 list_add_rcu(&data->list, &file->triggers);
557 ret++;
558
559 update_cond_flag(file);
560 if (trace_event_trigger_enable_disable(file, 1) < 0) {
561 list_del_rcu(&data->list);
562 update_cond_flag(file);
563 ret--;
564 }
565out:
566 return ret;
567}
568
569/**
570 * unregister_trigger - Generic event_command @unreg implementation
571 * @glob: The raw string used to register the trigger
572 * @ops: The trigger ops associated with the trigger
573 * @test: Trigger-specific data used to find the trigger to remove
574 * @file: The trace_event_file associated with the event
575 *
576 * Common implementation for event trigger unregistration.
577 *
578 * Usually used directly as the @unreg method in event command
579 * implementations.
580 */
581void unregister_trigger(char *glob, struct event_trigger_ops *ops,
582 struct event_trigger_data *test,
583 struct trace_event_file *file)
584{
585 struct event_trigger_data *data;
586 bool unregistered = false;
587
588 list_for_each_entry_rcu(data, &file->triggers, list) {
589 if (data->cmd_ops->trigger_type == test->cmd_ops->trigger_type) {
590 unregistered = true;
591 list_del_rcu(&data->list);
592 trace_event_trigger_enable_disable(file, 0);
593 update_cond_flag(file);
594 break;
595 }
596 }
597
598 if (unregistered && data->ops->free)
599 data->ops->free(data->ops, data);
600}
601
602/**
603 * event_trigger_callback - Generic event_command @func implementation
604 * @cmd_ops: The command ops, used for trigger registration
605 * @file: The trace_event_file associated with the event
606 * @glob: The raw string used to register the trigger
607 * @cmd: The cmd portion of the string used to register the trigger
608 * @param: The params portion of the string used to register the trigger
609 *
610 * Common implementation for event command parsing and trigger
611 * instantiation.
612 *
613 * Usually used directly as the @func method in event command
614 * implementations.
615 *
616 * Return: 0 on success, errno otherwise
617 */
618static int
619event_trigger_callback(struct event_command *cmd_ops,
620 struct trace_event_file *file,
621 char *glob, char *cmd, char *param)
622{
623 struct event_trigger_data *trigger_data;
624 struct event_trigger_ops *trigger_ops;
625 char *trigger = NULL;
626 char *number;
627 int ret;
628
629 /* separate the trigger from the filter (t:n [if filter]) */
630 if (param && isdigit(param[0]))
631 trigger = strsep(¶m, " \t");
632
633 trigger_ops = cmd_ops->get_trigger_ops(cmd, trigger);
634
635 ret = -ENOMEM;
636 trigger_data = kzalloc(sizeof(*trigger_data), GFP_KERNEL);
637 if (!trigger_data)
638 goto out;
639
640 trigger_data->count = -1;
641 trigger_data->ops = trigger_ops;
642 trigger_data->cmd_ops = cmd_ops;
643 INIT_LIST_HEAD(&trigger_data->list);
644
645 if (glob[0] == '!') {
646 cmd_ops->unreg(glob+1, trigger_ops, trigger_data, file);
647 kfree(trigger_data);
648 ret = 0;
649 goto out;
650 }
651
652 if (trigger) {
653 number = strsep(&trigger, ":");
654
655 ret = -EINVAL;
656 if (!strlen(number))
657 goto out_free;
658
659 /*
660 * We use the callback data field (which is a pointer)
661 * as our counter.
662 */
663 ret = kstrtoul(number, 0, &trigger_data->count);
664 if (ret)
665 goto out_free;
666 }
667
668 if (!param) /* if param is non-empty, it's supposed to be a filter */
669 goto out_reg;
670
671 if (!cmd_ops->set_filter)
672 goto out_reg;
673
674 ret = cmd_ops->set_filter(param, trigger_data, file);
675 if (ret < 0)
676 goto out_free;
677
678 out_reg:
679 ret = cmd_ops->reg(glob, trigger_ops, trigger_data, file);
680 /*
681 * The above returns on success the # of functions enabled,
682 * but if it didn't find any functions it returns zero.
683 * Consider no functions a failure too.
684 */
685 if (!ret) {
686 ret = -ENOENT;
687 goto out_free;
688 } else if (ret < 0)
689 goto out_free;
690 ret = 0;
691 out:
692 return ret;
693
694 out_free:
695 if (cmd_ops->set_filter)
696 cmd_ops->set_filter(NULL, trigger_data, NULL);
697 kfree(trigger_data);
698 goto out;
699}
700
701/**
702 * set_trigger_filter - Generic event_command @set_filter implementation
703 * @filter_str: The filter string for the trigger, NULL to remove filter
704 * @trigger_data: Trigger-specific data
705 * @file: The trace_event_file associated with the event
706 *
707 * Common implementation for event command filter parsing and filter
708 * instantiation.
709 *
710 * Usually used directly as the @set_filter method in event command
711 * implementations.
712 *
713 * Also used to remove a filter (if filter_str = NULL).
714 *
715 * Return: 0 on success, errno otherwise
716 */
717int set_trigger_filter(char *filter_str,
718 struct event_trigger_data *trigger_data,
719 struct trace_event_file *file)
720{
721 struct event_trigger_data *data = trigger_data;
722 struct event_filter *filter = NULL, *tmp;
723 int ret = -EINVAL;
724 char *s;
725
726 if (!filter_str) /* clear the current filter */
727 goto assign;
728
729 s = strsep(&filter_str, " \t");
730
731 if (!strlen(s) || strcmp(s, "if") != 0)
732 goto out;
733
734 if (!filter_str)
735 goto out;
736
737 /* The filter is for the 'trigger' event, not the triggered event */
738 ret = create_event_filter(file->event_call, filter_str, false, &filter);
739 if (ret)
740 goto out;
741 assign:
742 tmp = rcu_access_pointer(data->filter);
743
744 rcu_assign_pointer(data->filter, filter);
745
746 if (tmp) {
747 /* Make sure the call is done with the filter */
748 synchronize_sched();
749 free_event_filter(tmp);
750 }
751
752 kfree(data->filter_str);
753 data->filter_str = NULL;
754
755 if (filter_str) {
756 data->filter_str = kstrdup(filter_str, GFP_KERNEL);
757 if (!data->filter_str) {
758 free_event_filter(rcu_access_pointer(data->filter));
759 data->filter = NULL;
760 ret = -ENOMEM;
761 }
762 }
763 out:
764 return ret;
765}
766
767static void
768traceon_trigger(struct event_trigger_data *data, void *rec)
769{
770 if (tracing_is_on())
771 return;
772
773 tracing_on();
774}
775
776static void
777traceon_count_trigger(struct event_trigger_data *data, void *rec)
778{
779 if (tracing_is_on())
780 return;
781
782 if (!data->count)
783 return;
784
785 if (data->count != -1)
786 (data->count)--;
787
788 tracing_on();
789}
790
791static void
792traceoff_trigger(struct event_trigger_data *data, void *rec)
793{
794 if (!tracing_is_on())
795 return;
796
797 tracing_off();
798}
799
800static void
801traceoff_count_trigger(struct event_trigger_data *data, void *rec)
802{
803 if (!tracing_is_on())
804 return;
805
806 if (!data->count)
807 return;
808
809 if (data->count != -1)
810 (data->count)--;
811
812 tracing_off();
813}
814
815static int
816traceon_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
817 struct event_trigger_data *data)
818{
819 return event_trigger_print("traceon", m, (void *)data->count,
820 data->filter_str);
821}
822
823static int
824traceoff_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
825 struct event_trigger_data *data)
826{
827 return event_trigger_print("traceoff", m, (void *)data->count,
828 data->filter_str);
829}
830
831static struct event_trigger_ops traceon_trigger_ops = {
832 .func = traceon_trigger,
833 .print = traceon_trigger_print,
834 .init = event_trigger_init,
835 .free = event_trigger_free,
836};
837
838static struct event_trigger_ops traceon_count_trigger_ops = {
839 .func = traceon_count_trigger,
840 .print = traceon_trigger_print,
841 .init = event_trigger_init,
842 .free = event_trigger_free,
843};
844
845static struct event_trigger_ops traceoff_trigger_ops = {
846 .func = traceoff_trigger,
847 .print = traceoff_trigger_print,
848 .init = event_trigger_init,
849 .free = event_trigger_free,
850};
851
852static struct event_trigger_ops traceoff_count_trigger_ops = {
853 .func = traceoff_count_trigger,
854 .print = traceoff_trigger_print,
855 .init = event_trigger_init,
856 .free = event_trigger_free,
857};
858
859static struct event_trigger_ops *
860onoff_get_trigger_ops(char *cmd, char *param)
861{
862 struct event_trigger_ops *ops;
863
864 /* we register both traceon and traceoff to this callback */
865 if (strcmp(cmd, "traceon") == 0)
866 ops = param ? &traceon_count_trigger_ops :
867 &traceon_trigger_ops;
868 else
869 ops = param ? &traceoff_count_trigger_ops :
870 &traceoff_trigger_ops;
871
872 return ops;
873}
874
875static struct event_command trigger_traceon_cmd = {
876 .name = "traceon",
877 .trigger_type = ETT_TRACE_ONOFF,
878 .func = event_trigger_callback,
879 .reg = register_trigger,
880 .unreg = unregister_trigger,
881 .get_trigger_ops = onoff_get_trigger_ops,
882 .set_filter = set_trigger_filter,
883};
884
885static struct event_command trigger_traceoff_cmd = {
886 .name = "traceoff",
887 .trigger_type = ETT_TRACE_ONOFF,
888 .func = event_trigger_callback,
889 .reg = register_trigger,
890 .unreg = unregister_trigger,
891 .get_trigger_ops = onoff_get_trigger_ops,
892 .set_filter = set_trigger_filter,
893};
894
895#ifdef CONFIG_TRACER_SNAPSHOT
896static void
897snapshot_trigger(struct event_trigger_data *data, void *rec)
898{
899 tracing_snapshot();
900}
901
902static void
903snapshot_count_trigger(struct event_trigger_data *data, void *rec)
904{
905 if (!data->count)
906 return;
907
908 if (data->count != -1)
909 (data->count)--;
910
911 snapshot_trigger(data, rec);
912}
913
914static int
915register_snapshot_trigger(char *glob, struct event_trigger_ops *ops,
916 struct event_trigger_data *data,
917 struct trace_event_file *file)
918{
919 int ret = register_trigger(glob, ops, data, file);
920
921 if (ret > 0 && tracing_alloc_snapshot() != 0) {
922 unregister_trigger(glob, ops, data, file);
923 ret = 0;
924 }
925
926 return ret;
927}
928
929static int
930snapshot_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
931 struct event_trigger_data *data)
932{
933 return event_trigger_print("snapshot", m, (void *)data->count,
934 data->filter_str);
935}
936
937static struct event_trigger_ops snapshot_trigger_ops = {
938 .func = snapshot_trigger,
939 .print = snapshot_trigger_print,
940 .init = event_trigger_init,
941 .free = event_trigger_free,
942};
943
944static struct event_trigger_ops snapshot_count_trigger_ops = {
945 .func = snapshot_count_trigger,
946 .print = snapshot_trigger_print,
947 .init = event_trigger_init,
948 .free = event_trigger_free,
949};
950
951static struct event_trigger_ops *
952snapshot_get_trigger_ops(char *cmd, char *param)
953{
954 return param ? &snapshot_count_trigger_ops : &snapshot_trigger_ops;
955}
956
957static struct event_command trigger_snapshot_cmd = {
958 .name = "snapshot",
959 .trigger_type = ETT_SNAPSHOT,
960 .func = event_trigger_callback,
961 .reg = register_snapshot_trigger,
962 .unreg = unregister_trigger,
963 .get_trigger_ops = snapshot_get_trigger_ops,
964 .set_filter = set_trigger_filter,
965};
966
967static __init int register_trigger_snapshot_cmd(void)
968{
969 int ret;
970
971 ret = register_event_command(&trigger_snapshot_cmd);
972 WARN_ON(ret < 0);
973
974 return ret;
975}
976#else
977static __init int register_trigger_snapshot_cmd(void) { return 0; }
978#endif /* CONFIG_TRACER_SNAPSHOT */
979
980#ifdef CONFIG_STACKTRACE
981/*
982 * Skip 3:
983 * stacktrace_trigger()
984 * event_triggers_post_call()
985 * trace_event_raw_event_xxx()
986 */
987#define STACK_SKIP 3
988
989static void
990stacktrace_trigger(struct event_trigger_data *data, void *rec)
991{
992 trace_dump_stack(STACK_SKIP);
993}
994
995static void
996stacktrace_count_trigger(struct event_trigger_data *data, void *rec)
997{
998 if (!data->count)
999 return;
1000
1001 if (data->count != -1)
1002 (data->count)--;
1003
1004 stacktrace_trigger(data, rec);
1005}
1006
1007static int
1008stacktrace_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
1009 struct event_trigger_data *data)
1010{
1011 return event_trigger_print("stacktrace", m, (void *)data->count,
1012 data->filter_str);
1013}
1014
1015static struct event_trigger_ops stacktrace_trigger_ops = {
1016 .func = stacktrace_trigger,
1017 .print = stacktrace_trigger_print,
1018 .init = event_trigger_init,
1019 .free = event_trigger_free,
1020};
1021
1022static struct event_trigger_ops stacktrace_count_trigger_ops = {
1023 .func = stacktrace_count_trigger,
1024 .print = stacktrace_trigger_print,
1025 .init = event_trigger_init,
1026 .free = event_trigger_free,
1027};
1028
1029static struct event_trigger_ops *
1030stacktrace_get_trigger_ops(char *cmd, char *param)
1031{
1032 return param ? &stacktrace_count_trigger_ops : &stacktrace_trigger_ops;
1033}
1034
1035static struct event_command trigger_stacktrace_cmd = {
1036 .name = "stacktrace",
1037 .trigger_type = ETT_STACKTRACE,
1038 .flags = EVENT_CMD_FL_POST_TRIGGER,
1039 .func = event_trigger_callback,
1040 .reg = register_trigger,
1041 .unreg = unregister_trigger,
1042 .get_trigger_ops = stacktrace_get_trigger_ops,
1043 .set_filter = set_trigger_filter,
1044};
1045
1046static __init int register_trigger_stacktrace_cmd(void)
1047{
1048 int ret;
1049
1050 ret = register_event_command(&trigger_stacktrace_cmd);
1051 WARN_ON(ret < 0);
1052
1053 return ret;
1054}
1055#else
1056static __init int register_trigger_stacktrace_cmd(void) { return 0; }
1057#endif /* CONFIG_STACKTRACE */
1058
1059static __init void unregister_trigger_traceon_traceoff_cmds(void)
1060{
1061 unregister_event_command(&trigger_traceon_cmd);
1062 unregister_event_command(&trigger_traceoff_cmd);
1063}
1064
1065/* Avoid typos */
1066#define ENABLE_EVENT_STR "enable_event"
1067#define DISABLE_EVENT_STR "disable_event"
1068
1069struct enable_trigger_data {
1070 struct trace_event_file *file;
1071 bool enable;
1072};
1073
1074static void
1075event_enable_trigger(struct event_trigger_data *data, void *rec)
1076{
1077 struct enable_trigger_data *enable_data = data->private_data;
1078
1079 if (enable_data->enable)
1080 clear_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &enable_data->file->flags);
1081 else
1082 set_bit(EVENT_FILE_FL_SOFT_DISABLED_BIT, &enable_data->file->flags);
1083}
1084
1085static void
1086event_enable_count_trigger(struct event_trigger_data *data, void *rec)
1087{
1088 struct enable_trigger_data *enable_data = data->private_data;
1089
1090 if (!data->count)
1091 return;
1092
1093 /* Skip if the event is in a state we want to switch to */
1094 if (enable_data->enable == !(enable_data->file->flags & EVENT_FILE_FL_SOFT_DISABLED))
1095 return;
1096
1097 if (data->count != -1)
1098 (data->count)--;
1099
1100 event_enable_trigger(data, rec);
1101}
1102
1103static int
1104event_enable_trigger_print(struct seq_file *m, struct event_trigger_ops *ops,
1105 struct event_trigger_data *data)
1106{
1107 struct enable_trigger_data *enable_data = data->private_data;
1108
1109 seq_printf(m, "%s:%s:%s",
1110 enable_data->enable ? ENABLE_EVENT_STR : DISABLE_EVENT_STR,
1111 enable_data->file->event_call->class->system,
1112 trace_event_name(enable_data->file->event_call));
1113
1114 if (data->count == -1)
1115 seq_puts(m, ":unlimited");
1116 else
1117 seq_printf(m, ":count=%ld", data->count);
1118
1119 if (data->filter_str)
1120 seq_printf(m, " if %s\n", data->filter_str);
1121 else
1122 seq_putc(m, '\n');
1123
1124 return 0;
1125}
1126
1127static void
1128event_enable_trigger_free(struct event_trigger_ops *ops,
1129 struct event_trigger_data *data)
1130{
1131 struct enable_trigger_data *enable_data = data->private_data;
1132
1133 if (WARN_ON_ONCE(data->ref <= 0))
1134 return;
1135
1136 data->ref--;
1137 if (!data->ref) {
1138 /* Remove the SOFT_MODE flag */
1139 trace_event_enable_disable(enable_data->file, 0, 1);
1140 module_put(enable_data->file->event_call->mod);
1141 trigger_data_free(data);
1142 kfree(enable_data);
1143 }
1144}
1145
1146static struct event_trigger_ops event_enable_trigger_ops = {
1147 .func = event_enable_trigger,
1148 .print = event_enable_trigger_print,
1149 .init = event_trigger_init,
1150 .free = event_enable_trigger_free,
1151};
1152
1153static struct event_trigger_ops event_enable_count_trigger_ops = {
1154 .func = event_enable_count_trigger,
1155 .print = event_enable_trigger_print,
1156 .init = event_trigger_init,
1157 .free = event_enable_trigger_free,
1158};
1159
1160static struct event_trigger_ops event_disable_trigger_ops = {
1161 .func = event_enable_trigger,
1162 .print = event_enable_trigger_print,
1163 .init = event_trigger_init,
1164 .free = event_enable_trigger_free,
1165};
1166
1167static struct event_trigger_ops event_disable_count_trigger_ops = {
1168 .func = event_enable_count_trigger,
1169 .print = event_enable_trigger_print,
1170 .init = event_trigger_init,
1171 .free = event_enable_trigger_free,
1172};
1173
1174static int
1175event_enable_trigger_func(struct event_command *cmd_ops,
1176 struct trace_event_file *file,
1177 char *glob, char *cmd, char *param)
1178{
1179 struct trace_event_file *event_enable_file;
1180 struct enable_trigger_data *enable_data;
1181 struct event_trigger_data *trigger_data;
1182 struct event_trigger_ops *trigger_ops;
1183 struct trace_array *tr = file->tr;
1184 const char *system;
1185 const char *event;
1186 char *trigger;
1187 char *number;
1188 bool enable;
1189 int ret;
1190
1191 if (!param)
1192 return -EINVAL;
1193
1194 /* separate the trigger from the filter (s:e:n [if filter]) */
1195 trigger = strsep(¶m, " \t");
1196 if (!trigger)
1197 return -EINVAL;
1198
1199 system = strsep(&trigger, ":");
1200 if (!trigger)
1201 return -EINVAL;
1202
1203 event = strsep(&trigger, ":");
1204
1205 ret = -EINVAL;
1206 event_enable_file = find_event_file(tr, system, event);
1207 if (!event_enable_file)
1208 goto out;
1209
1210 enable = strcmp(cmd, ENABLE_EVENT_STR) == 0;
1211
1212 trigger_ops = cmd_ops->get_trigger_ops(cmd, trigger);
1213
1214 ret = -ENOMEM;
1215 trigger_data = kzalloc(sizeof(*trigger_data), GFP_KERNEL);
1216 if (!trigger_data)
1217 goto out;
1218
1219 enable_data = kzalloc(sizeof(*enable_data), GFP_KERNEL);
1220 if (!enable_data) {
1221 kfree(trigger_data);
1222 goto out;
1223 }
1224
1225 trigger_data->count = -1;
1226 trigger_data->ops = trigger_ops;
1227 trigger_data->cmd_ops = cmd_ops;
1228 INIT_LIST_HEAD(&trigger_data->list);
1229 RCU_INIT_POINTER(trigger_data->filter, NULL);
1230
1231 enable_data->enable = enable;
1232 enable_data->file = event_enable_file;
1233 trigger_data->private_data = enable_data;
1234
1235 if (glob[0] == '!') {
1236 cmd_ops->unreg(glob+1, trigger_ops, trigger_data, file);
1237 kfree(trigger_data);
1238 kfree(enable_data);
1239 ret = 0;
1240 goto out;
1241 }
1242
1243 if (trigger) {
1244 number = strsep(&trigger, ":");
1245
1246 ret = -EINVAL;
1247 if (!strlen(number))
1248 goto out_free;
1249
1250 /*
1251 * We use the callback data field (which is a pointer)
1252 * as our counter.
1253 */
1254 ret = kstrtoul(number, 0, &trigger_data->count);
1255 if (ret)
1256 goto out_free;
1257 }
1258
1259 if (!param) /* if param is non-empty, it's supposed to be a filter */
1260 goto out_reg;
1261
1262 if (!cmd_ops->set_filter)
1263 goto out_reg;
1264
1265 ret = cmd_ops->set_filter(param, trigger_data, file);
1266 if (ret < 0)
1267 goto out_free;
1268
1269 out_reg:
1270 /* Don't let event modules unload while probe registered */
1271 ret = try_module_get(event_enable_file->event_call->mod);
1272 if (!ret) {
1273 ret = -EBUSY;
1274 goto out_free;
1275 }
1276
1277 ret = trace_event_enable_disable(event_enable_file, 1, 1);
1278 if (ret < 0)
1279 goto out_put;
1280 ret = cmd_ops->reg(glob, trigger_ops, trigger_data, file);
1281 /*
1282 * The above returns on success the # of functions enabled,
1283 * but if it didn't find any functions it returns zero.
1284 * Consider no functions a failure too.
1285 */
1286 if (!ret) {
1287 ret = -ENOENT;
1288 goto out_disable;
1289 } else if (ret < 0)
1290 goto out_disable;
1291 /* Just return zero, not the number of enabled functions */
1292 ret = 0;
1293 out:
1294 return ret;
1295
1296 out_disable:
1297 trace_event_enable_disable(event_enable_file, 0, 1);
1298 out_put:
1299 module_put(event_enable_file->event_call->mod);
1300 out_free:
1301 if (cmd_ops->set_filter)
1302 cmd_ops->set_filter(NULL, trigger_data, NULL);
1303 kfree(trigger_data);
1304 kfree(enable_data);
1305 goto out;
1306}
1307
1308static int event_enable_register_trigger(char *glob,
1309 struct event_trigger_ops *ops,
1310 struct event_trigger_data *data,
1311 struct trace_event_file *file)
1312{
1313 struct enable_trigger_data *enable_data = data->private_data;
1314 struct enable_trigger_data *test_enable_data;
1315 struct event_trigger_data *test;
1316 int ret = 0;
1317
1318 list_for_each_entry_rcu(test, &file->triggers, list) {
1319 test_enable_data = test->private_data;
1320 if (test_enable_data &&
1321 (test_enable_data->file == enable_data->file)) {
1322 ret = -EEXIST;
1323 goto out;
1324 }
1325 }
1326
1327 if (data->ops->init) {
1328 ret = data->ops->init(data->ops, data);
1329 if (ret < 0)
1330 goto out;
1331 }
1332
1333 list_add_rcu(&data->list, &file->triggers);
1334 ret++;
1335
1336 update_cond_flag(file);
1337 if (trace_event_trigger_enable_disable(file, 1) < 0) {
1338 list_del_rcu(&data->list);
1339 update_cond_flag(file);
1340 ret--;
1341 }
1342out:
1343 return ret;
1344}
1345
1346static void event_enable_unregister_trigger(char *glob,
1347 struct event_trigger_ops *ops,
1348 struct event_trigger_data *test,
1349 struct trace_event_file *file)
1350{
1351 struct enable_trigger_data *test_enable_data = test->private_data;
1352 struct enable_trigger_data *enable_data;
1353 struct event_trigger_data *data;
1354 bool unregistered = false;
1355
1356 list_for_each_entry_rcu(data, &file->triggers, list) {
1357 enable_data = data->private_data;
1358 if (enable_data &&
1359 (enable_data->file == test_enable_data->file)) {
1360 unregistered = true;
1361 list_del_rcu(&data->list);
1362 trace_event_trigger_enable_disable(file, 0);
1363 update_cond_flag(file);
1364 break;
1365 }
1366 }
1367
1368 if (unregistered && data->ops->free)
1369 data->ops->free(data->ops, data);
1370}
1371
1372static struct event_trigger_ops *
1373event_enable_get_trigger_ops(char *cmd, char *param)
1374{
1375 struct event_trigger_ops *ops;
1376 bool enable;
1377
1378 enable = strcmp(cmd, ENABLE_EVENT_STR) == 0;
1379
1380 if (enable)
1381 ops = param ? &event_enable_count_trigger_ops :
1382 &event_enable_trigger_ops;
1383 else
1384 ops = param ? &event_disable_count_trigger_ops :
1385 &event_disable_trigger_ops;
1386
1387 return ops;
1388}
1389
1390static struct event_command trigger_enable_cmd = {
1391 .name = ENABLE_EVENT_STR,
1392 .trigger_type = ETT_EVENT_ENABLE,
1393 .func = event_enable_trigger_func,
1394 .reg = event_enable_register_trigger,
1395 .unreg = event_enable_unregister_trigger,
1396 .get_trigger_ops = event_enable_get_trigger_ops,
1397 .set_filter = set_trigger_filter,
1398};
1399
1400static struct event_command trigger_disable_cmd = {
1401 .name = DISABLE_EVENT_STR,
1402 .trigger_type = ETT_EVENT_ENABLE,
1403 .func = event_enable_trigger_func,
1404 .reg = event_enable_register_trigger,
1405 .unreg = event_enable_unregister_trigger,
1406 .get_trigger_ops = event_enable_get_trigger_ops,
1407 .set_filter = set_trigger_filter,
1408};
1409
1410static __init void unregister_trigger_enable_disable_cmds(void)
1411{
1412 unregister_event_command(&trigger_enable_cmd);
1413 unregister_event_command(&trigger_disable_cmd);
1414}
1415
1416static __init int register_trigger_enable_disable_cmds(void)
1417{
1418 int ret;
1419
1420 ret = register_event_command(&trigger_enable_cmd);
1421 if (WARN_ON(ret < 0))
1422 return ret;
1423 ret = register_event_command(&trigger_disable_cmd);
1424 if (WARN_ON(ret < 0))
1425 unregister_trigger_enable_disable_cmds();
1426
1427 return ret;
1428}
1429
1430static __init int register_trigger_traceon_traceoff_cmds(void)
1431{
1432 int ret;
1433
1434 ret = register_event_command(&trigger_traceon_cmd);
1435 if (WARN_ON(ret < 0))
1436 return ret;
1437 ret = register_event_command(&trigger_traceoff_cmd);
1438 if (WARN_ON(ret < 0))
1439 unregister_trigger_traceon_traceoff_cmds();
1440
1441 return ret;
1442}
1443
1444__init int register_trigger_cmds(void)
1445{
1446 register_trigger_traceon_traceoff_cmds();
1447 register_trigger_snapshot_cmd();
1448 register_trigger_stacktrace_cmd();
1449 register_trigger_enable_disable_cmds();
1450
1451 return 0;
1452}