Commit | Line | Data |
---|---|---|
02119ee5 MD |
1 | /* |
2 | * ltt-probes.c | |
3 | * | |
4 | * Copyright 2010 (c) - Mathieu Desnoyers <mathieu.desnoyers@efficios.com> | |
5 | * | |
6 | * Holds LTTng probes registry. | |
7 | */ | |
8 | ||
9 | #include <linux/module.h> | |
10 | #include <linux/list.h> | |
11 | #include <linux/mutex.h> | |
271b6681 | 12 | #include <linux/seq_file.h> |
02119ee5 | 13 | |
85a9ca7f | 14 | #include "ltt-events.h" |
02119ee5 MD |
15 | |
16 | static LIST_HEAD(probe_list); | |
17 | static DEFINE_MUTEX(probe_mutex); | |
02119ee5 | 18 | |
85a9ca7f MD |
19 | static |
20 | const struct lttng_event_desc *find_event(const char *name) | |
02119ee5 | 21 | { |
85a9ca7f MD |
22 | struct lttng_probe_desc *probe_desc; |
23 | int i; | |
02119ee5 | 24 | |
85a9ca7f MD |
25 | list_for_each_entry(probe_desc, &probe_list, head) { |
26 | for (i = 0; i < probe_desc->nr_events; i++) { | |
27 | if (!strcmp(probe_desc->event_desc[i].name, name)) | |
28 | return &probe_desc->event_desc[i]; | |
29 | } | |
02119ee5 MD |
30 | } |
31 | return NULL; | |
32 | } | |
33 | ||
85a9ca7f | 34 | int ltt_probe_register(struct lttng_probe_desc *desc) |
02119ee5 | 35 | { |
02119ee5 | 36 | int ret = 0; |
85a9ca7f | 37 | int i; |
02119ee5 MD |
38 | |
39 | mutex_lock(&probe_mutex); | |
85a9ca7f MD |
40 | /* |
41 | * TODO: This is O(N^2). Turn into a hash table when probe registration | |
42 | * overhead becomes an issue. | |
43 | */ | |
44 | for (i = 0; i < desc->nr_events; i++) { | |
45 | if (find_event(desc->event_desc[i].name)) { | |
46 | ret = -EEXIST; | |
47 | goto end; | |
48 | } | |
02119ee5 | 49 | } |
85a9ca7f | 50 | list_add(&desc->head, &probe_list); |
02119ee5 MD |
51 | end: |
52 | mutex_unlock(&probe_mutex); | |
53 | return ret; | |
54 | } | |
55 | EXPORT_SYMBOL_GPL(ltt_probe_register); | |
56 | ||
85a9ca7f | 57 | void ltt_probe_unregister(struct lttng_probe_desc *desc) |
02119ee5 | 58 | { |
02119ee5 | 59 | mutex_lock(&probe_mutex); |
85a9ca7f | 60 | list_del(&desc->head); |
02119ee5 | 61 | mutex_unlock(&probe_mutex); |
02119ee5 MD |
62 | } |
63 | EXPORT_SYMBOL_GPL(ltt_probe_unregister); | |
64 | ||
85a9ca7f | 65 | const struct lttng_event_desc *ltt_event_get(const char *name) |
02119ee5 | 66 | { |
85a9ca7f | 67 | const struct lttng_event_desc *event; |
02119ee5 MD |
68 | int ret; |
69 | ||
70 | mutex_lock(&probe_mutex); | |
85a9ca7f | 71 | event = find_event(name); |
c099397a | 72 | mutex_unlock(&probe_mutex); |
85a9ca7f | 73 | if (!event) |
c099397a | 74 | return NULL; |
dc7f600a | 75 | ret = try_module_get(event->owner); |
02119ee5 | 76 | WARN_ON_ONCE(!ret); |
85a9ca7f | 77 | return event; |
02119ee5 | 78 | } |
85a9ca7f | 79 | EXPORT_SYMBOL_GPL(ltt_event_get); |
02119ee5 | 80 | |
85a9ca7f | 81 | void ltt_event_put(const struct lttng_event_desc *event) |
02119ee5 | 82 | { |
dc7f600a | 83 | module_put(event->owner); |
02119ee5 | 84 | } |
85a9ca7f | 85 | EXPORT_SYMBOL_GPL(ltt_event_put); |
271b6681 MD |
86 | |
87 | static | |
88 | void *tp_list_start(struct seq_file *m, loff_t *pos) | |
89 | { | |
90 | struct lttng_probe_desc *probe_desc; | |
91 | int iter = 0, i; | |
92 | ||
cbd4ecb2 | 93 | mutex_lock(&probe_mutex); |
271b6681 MD |
94 | list_for_each_entry(probe_desc, &probe_list, head) { |
95 | for (i = 0; i < probe_desc->nr_events; i++) { | |
96 | if (iter++ >= *pos) | |
97 | return (void *) &probe_desc->event_desc[i]; | |
98 | } | |
99 | } | |
100 | /* End of list */ | |
101 | return NULL; | |
102 | } | |
103 | ||
104 | static | |
105 | void *tp_list_next(struct seq_file *m, void *p, loff_t *ppos) | |
106 | { | |
107 | struct lttng_probe_desc *probe_desc; | |
108 | int iter = 0, i; | |
109 | ||
110 | (*ppos)++; | |
111 | list_for_each_entry(probe_desc, &probe_list, head) { | |
112 | for (i = 0; i < probe_desc->nr_events; i++) { | |
113 | if (iter++ >= *ppos) | |
114 | return (void *) &probe_desc->event_desc[i]; | |
115 | } | |
116 | } | |
117 | /* End of list */ | |
118 | return NULL; | |
119 | } | |
120 | ||
121 | static | |
122 | void tp_list_stop(struct seq_file *m, void *p) | |
123 | { | |
cbd4ecb2 | 124 | mutex_unlock(&probe_mutex); |
271b6681 MD |
125 | } |
126 | ||
127 | static | |
128 | int tp_list_show(struct seq_file *m, void *p) | |
129 | { | |
130 | const struct lttng_event_desc *probe_desc = p; | |
131 | ||
132 | seq_printf(m, "event { name = %s; };\n", | |
133 | probe_desc->name); | |
134 | return 0; | |
135 | } | |
136 | ||
137 | static | |
138 | const struct seq_operations lttng_tracepoint_list_seq_ops = { | |
139 | .start = tp_list_start, | |
140 | .next = tp_list_next, | |
141 | .stop = tp_list_stop, | |
142 | .show = tp_list_show, | |
143 | }; | |
144 | ||
145 | static | |
146 | int lttng_tracepoint_list_open(struct inode *inode, struct file *file) | |
147 | { | |
148 | return seq_open(file, <tng_tracepoint_list_seq_ops); | |
149 | } | |
150 | ||
151 | const struct file_operations lttng_tracepoint_list_fops = { | |
152 | .open = lttng_tracepoint_list_open, | |
153 | .read = seq_read, | |
154 | .llseek = seq_lseek, | |
a51729c7 | 155 | .release = seq_release, |
271b6681 | 156 | }; |