Commit | Line | Data |
---|---|---|
54d01ffb DG |
1 | /* |
2 | * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca> | |
3 | * | |
d14d33bf AM |
4 | * This program is free software; you can redistribute it and/or modify |
5 | * it under the terms of the GNU General Public License, version 2 only, | |
6 | * as published by the Free Software Foundation. | |
54d01ffb | 7 | * |
d14d33bf AM |
8 | * This program is distributed in the hope that it will be useful, |
9 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
10 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
11 | * GNU General Public License for more details. | |
54d01ffb | 12 | * |
d14d33bf AM |
13 | * You should have received a copy of the GNU General Public License along |
14 | * with this program; if not, write to the Free Software Foundation, Inc., | |
15 | * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | |
54d01ffb DG |
16 | */ |
17 | ||
be040666 | 18 | #define _GNU_SOURCE |
6c1c0768 | 19 | #define _LGPL_SOURCE |
d87bfb32 | 20 | #include <errno.h> |
54d01ffb | 21 | #include <urcu/list.h> |
2bdd86d4 | 22 | #include <string.h> |
54d01ffb DG |
23 | |
24 | #include <lttng/lttng.h> | |
db758600 | 25 | #include <common/error.h> |
10a8a223 | 26 | #include <common/sessiond-comm/sessiond-comm.h> |
54d01ffb DG |
27 | |
28 | #include "channel.h" | |
29 | #include "event.h" | |
4771f025 | 30 | #include "kernel.h" |
be6a6276 | 31 | #include "lttng-sessiond.h" |
9df8df5e | 32 | #include "ust-ctl.h" |
edb67388 DG |
33 | #include "ust-app.h" |
34 | #include "trace-kernel.h" | |
35 | #include "trace-ust.h" | |
54d01ffb | 36 | |
025faf73 DG |
37 | /* |
38 | * Add unique UST event based on the event name, filter bytecode and loglevel. | |
39 | */ | |
18eace3b DG |
40 | static void add_unique_ust_event(struct lttng_ht *ht, |
41 | struct ltt_ust_event *event) | |
42 | { | |
43 | struct cds_lfht_node *node_ptr; | |
44 | struct ltt_ust_ht_key key; | |
45 | ||
46 | assert(ht); | |
47 | assert(ht->ht); | |
48 | assert(event); | |
49 | ||
50 | key.name = event->attr.name; | |
51 | key.filter = (struct lttng_filter_bytecode *) event->filter; | |
52 | key.loglevel = event->attr.loglevel; | |
4031e53e | 53 | key.exclusion = event->exclusion; |
18eace3b DG |
54 | |
55 | node_ptr = cds_lfht_add_unique(ht->ht, | |
56 | ht->hash_fct(event->node.key, lttng_ht_seed), | |
57 | trace_ust_ht_match_event, &key, &event->node.node); | |
58 | assert(node_ptr == &event->node.node); | |
59 | } | |
60 | ||
54d01ffb | 61 | /* |
7a3d1328 | 62 | * Disable kernel tracepoint event for a channel from the kernel session. |
54d01ffb | 63 | */ |
d0ae4ea8 | 64 | int event_kernel_disable_event(struct ltt_kernel_channel *kchan, |
d3a56674 | 65 | char *event_name) |
54d01ffb DG |
66 | { |
67 | int ret; | |
68 | struct ltt_kernel_event *kevent; | |
69 | ||
0525e9ae DG |
70 | assert(kchan); |
71 | ||
d0ae4ea8 MD |
72 | kevent = trace_kernel_get_event_by_name(event_name, kchan, |
73 | LTTNG_EVENT_ALL); | |
54d01ffb | 74 | if (kevent == NULL) { |
f73fabfd | 75 | ret = LTTNG_ERR_NO_EVENT; |
54d01ffb DG |
76 | goto error; |
77 | } | |
78 | ||
79 | ret = kernel_disable_event(kevent); | |
80 | if (ret < 0) { | |
f73fabfd | 81 | ret = LTTNG_ERR_KERN_DISABLE_FAIL; |
54d01ffb DG |
82 | goto error; |
83 | } | |
84 | ||
85 | DBG("Kernel event %s disable for channel %s.", | |
86 | kevent->event->name, kchan->channel->name); | |
87 | ||
f73fabfd | 88 | ret = LTTNG_OK; |
54d01ffb DG |
89 | |
90 | error: | |
91 | return ret; | |
92 | } | |
93 | ||
94 | /* | |
7a3d1328 | 95 | * Disable kernel tracepoint events for a channel from the kernel session. |
54d01ffb | 96 | */ |
d0ae4ea8 MD |
97 | int event_kernel_disable_event_type(struct ltt_kernel_channel *kchan, |
98 | enum lttng_event_type type) | |
54d01ffb DG |
99 | { |
100 | int ret; | |
101 | struct ltt_kernel_event *kevent; | |
102 | ||
0525e9ae DG |
103 | assert(kchan); |
104 | ||
54d01ffb DG |
105 | /* For each event in the kernel session */ |
106 | cds_list_for_each_entry(kevent, &kchan->events_list.head, list) { | |
d0ae4ea8 MD |
107 | if (type != LTTNG_EVENT_ALL && kevent->type != type) |
108 | continue; | |
54d01ffb DG |
109 | ret = kernel_disable_event(kevent); |
110 | if (ret < 0) { | |
111 | /* We continue disabling the rest */ | |
112 | continue; | |
113 | } | |
114 | } | |
f73fabfd | 115 | ret = LTTNG_OK; |
7a3d1328 MD |
116 | return ret; |
117 | } | |
118 | ||
7a3d1328 MD |
119 | /* |
120 | * Disable all kernel event for a channel from the kernel session. | |
121 | */ | |
d0ae4ea8 | 122 | int event_kernel_disable_event_all(struct ltt_kernel_channel *kchan) |
7a3d1328 | 123 | { |
d0ae4ea8 | 124 | return event_kernel_disable_event_type(kchan, LTTNG_EVENT_ALL); |
54d01ffb DG |
125 | } |
126 | ||
127 | /* | |
7a3d1328 | 128 | * Enable kernel tracepoint event for a channel from the kernel session. |
49d21f93 | 129 | * We own filter_expression and filter. |
54d01ffb | 130 | */ |
d0ae4ea8 | 131 | int event_kernel_enable_event(struct ltt_kernel_channel *kchan, |
00a62084 MD |
132 | struct lttng_event *event, char *filter_expression, |
133 | struct lttng_filter_bytecode *filter) | |
54d01ffb DG |
134 | { |
135 | int ret; | |
136 | struct ltt_kernel_event *kevent; | |
137 | ||
0525e9ae DG |
138 | assert(kchan); |
139 | assert(event); | |
140 | ||
00a62084 MD |
141 | kevent = trace_kernel_find_event(event->name, kchan, |
142 | event->type, filter); | |
54d01ffb | 143 | if (kevent == NULL) { |
00a62084 MD |
144 | ret = kernel_create_event(event, kchan, |
145 | filter_expression, filter); | |
a969e101 MD |
146 | /* We have passed ownership */ |
147 | filter_expression = NULL; | |
148 | filter = NULL; | |
54d01ffb | 149 | if (ret < 0) { |
bd29c13d DG |
150 | switch (-ret) { |
151 | case EEXIST: | |
f73fabfd | 152 | ret = LTTNG_ERR_KERN_EVENT_EXIST; |
bd29c13d DG |
153 | break; |
154 | case ENOSYS: | |
f73fabfd | 155 | ret = LTTNG_ERR_KERN_EVENT_ENOSYS; |
bd29c13d DG |
156 | break; |
157 | default: | |
f73fabfd | 158 | ret = LTTNG_ERR_KERN_ENABLE_FAIL; |
bd29c13d | 159 | break; |
d87bfb32 | 160 | } |
7a3d1328 | 161 | goto end; |
54d01ffb DG |
162 | } |
163 | } else if (kevent->enabled == 0) { | |
164 | ret = kernel_enable_event(kevent); | |
165 | if (ret < 0) { | |
f73fabfd | 166 | ret = LTTNG_ERR_KERN_ENABLE_FAIL; |
7a3d1328 | 167 | goto end; |
54d01ffb | 168 | } |
42224349 DG |
169 | } else { |
170 | /* At this point, the event is considered enabled */ | |
f73fabfd | 171 | ret = LTTNG_ERR_KERN_EVENT_EXIST; |
42224349 | 172 | goto end; |
54d01ffb | 173 | } |
42224349 | 174 | |
f73fabfd | 175 | ret = LTTNG_OK; |
7a3d1328 | 176 | end: |
a969e101 MD |
177 | free(filter_expression); |
178 | free(filter); | |
54d01ffb DG |
179 | return ret; |
180 | } | |
181 | ||
7f79d3a1 DG |
182 | /* |
183 | * ============================ | |
184 | * UST : The Ultimate Frontier! | |
185 | * ============================ | |
186 | */ | |
187 | ||
76d45b40 DG |
188 | /* |
189 | * Enable all UST tracepoints for a channel from a UST session. | |
190 | */ | |
7972aab2 | 191 | int event_ust_enable_all_tracepoints(struct ltt_ust_session *usess, |
6b453b5e JG |
192 | struct ltt_ust_channel *uchan, |
193 | char *filter_expression, | |
194 | struct lttng_filter_bytecode *filter) | |
76d45b40 | 195 | { |
6775595e | 196 | int ret, i, size; |
bec39940 | 197 | struct lttng_ht_iter iter; |
76d45b40 | 198 | struct ltt_ust_event *uevent = NULL; |
7f79d3a1 | 199 | struct lttng_event *events = NULL; |
76d45b40 | 200 | |
0525e9ae DG |
201 | assert(usess); |
202 | assert(uchan); | |
203 | ||
025faf73 DG |
204 | rcu_read_lock(); |
205 | ||
7972aab2 DG |
206 | /* Enable existing events */ |
207 | cds_lfht_for_each_entry(uchan->events->ht, &iter.iter, uevent, | |
208 | node.node) { | |
209 | if (uevent->enabled == 0) { | |
210 | ret = ust_app_enable_event_glb(usess, uchan, uevent); | |
211 | if (ret < 0) { | |
76d45b40 DG |
212 | continue; |
213 | } | |
7972aab2 DG |
214 | uevent->enabled = 1; |
215 | } | |
216 | } | |
76d45b40 | 217 | |
7972aab2 DG |
218 | /* Get all UST available events */ |
219 | size = ust_app_list_events(&events); | |
220 | if (size < 0) { | |
221 | ret = LTTNG_ERR_UST_LIST_FAIL; | |
222 | goto error; | |
223 | } | |
76d45b40 | 224 | |
7972aab2 DG |
225 | for (i = 0; i < size; i++) { |
226 | /* | |
227 | * Check if event exist and if so, continue since it was enable | |
228 | * previously. | |
229 | */ | |
230 | uevent = trace_ust_find_event(uchan->events, events[i].name, filter, | |
10646003 | 231 | events[i].loglevel, NULL); |
7972aab2 | 232 | if (uevent != NULL) { |
76d45b40 DG |
233 | ret = ust_app_enable_event_pid(usess, uchan, uevent, |
234 | events[i].pid); | |
235 | if (ret < 0) { | |
7972aab2 | 236 | if (ret != -LTTNG_UST_ERR_EXIST) { |
f73fabfd | 237 | ret = LTTNG_ERR_UST_ENABLE_FAIL; |
7972aab2 | 238 | goto error; |
76d45b40 | 239 | } |
76d45b40 | 240 | } |
7972aab2 DG |
241 | continue; |
242 | } | |
76d45b40 | 243 | |
7972aab2 | 244 | /* Create ust event */ |
6b453b5e | 245 | uevent = trace_ust_create_event(&events[i], filter_expression, |
9732459c | 246 | filter, NULL, false); |
7972aab2 DG |
247 | if (uevent == NULL) { |
248 | ret = LTTNG_ERR_FATAL; | |
249 | goto error_destroy; | |
76d45b40 DG |
250 | } |
251 | ||
7972aab2 DG |
252 | /* Create event for the specific PID */ |
253 | ret = ust_app_enable_event_pid(usess, uchan, uevent, | |
254 | events[i].pid); | |
255 | if (ret < 0) { | |
256 | if (ret == -LTTNG_UST_ERR_EXIST) { | |
257 | ret = LTTNG_ERR_UST_EVENT_EXIST; | |
258 | goto error; | |
259 | } else { | |
260 | ret = LTTNG_ERR_UST_ENABLE_FAIL; | |
261 | goto error_destroy; | |
262 | } | |
263 | } | |
264 | ||
265 | uevent->enabled = 1; | |
266 | /* Add ltt ust event to channel */ | |
267 | rcu_read_lock(); | |
268 | add_unique_ust_event(uchan->events, uevent); | |
269 | rcu_read_unlock(); | |
76d45b40 | 270 | } |
7972aab2 | 271 | free(events); |
76d45b40 | 272 | |
025faf73 | 273 | rcu_read_unlock(); |
f73fabfd | 274 | return LTTNG_OK; |
76d45b40 | 275 | |
7f79d3a1 | 276 | error_destroy: |
76d45b40 | 277 | trace_ust_destroy_event(uevent); |
7f79d3a1 DG |
278 | |
279 | error: | |
280 | free(events); | |
025faf73 | 281 | rcu_read_unlock(); |
76d45b40 DG |
282 | return ret; |
283 | } | |
284 | ||
2bdd86d4 MD |
285 | /* |
286 | * Enable UST tracepoint event for a channel from a UST session. | |
49d21f93 | 287 | * We own filter_expression, filter, and exclusion. |
2bdd86d4 | 288 | */ |
7972aab2 | 289 | int event_ust_enable_tracepoint(struct ltt_ust_session *usess, |
025faf73 | 290 | struct ltt_ust_channel *uchan, struct lttng_event *event, |
6b453b5e | 291 | char *filter_expression, |
f1613f52 | 292 | struct lttng_filter_bytecode *filter, |
9732459c JG |
293 | struct lttng_event_exclusion *exclusion, |
294 | bool internal_event) | |
2bdd86d4 | 295 | { |
f73fabfd | 296 | int ret = LTTNG_OK, to_create = 0; |
edb67388 DG |
297 | struct ltt_ust_event *uevent; |
298 | ||
0525e9ae DG |
299 | assert(usess); |
300 | assert(uchan); | |
301 | assert(event); | |
302 | ||
18eace3b DG |
303 | rcu_read_lock(); |
304 | ||
025faf73 | 305 | uevent = trace_ust_find_event(uchan->events, event->name, filter, |
10646003 | 306 | event->loglevel, exclusion); |
edb67388 | 307 | if (uevent == NULL) { |
6b453b5e | 308 | uevent = trace_ust_create_event(event, filter_expression, |
9732459c | 309 | filter, exclusion, internal_event); |
49d21f93 MD |
310 | /* We have passed ownership */ |
311 | filter_expression = NULL; | |
312 | filter = NULL; | |
313 | exclusion = NULL; | |
edb67388 | 314 | if (uevent == NULL) { |
95a82664 | 315 | ret = LTTNG_ERR_UST_ENABLE_FAIL; |
edb67388 DG |
316 | goto error; |
317 | } | |
025faf73 | 318 | |
fc34caaa | 319 | /* Valid to set it after the goto error since uevent is still NULL */ |
edb67388 DG |
320 | to_create = 1; |
321 | } | |
2bdd86d4 | 322 | |
7f79d3a1 DG |
323 | if (uevent->enabled) { |
324 | /* It's already enabled so everything is OK */ | |
5bcdda4f | 325 | ret = LTTNG_ERR_UST_EVENT_ENABLED; |
7f79d3a1 DG |
326 | goto end; |
327 | } | |
328 | ||
fc34caaa DG |
329 | uevent->enabled = 1; |
330 | ||
7972aab2 DG |
331 | if (to_create) { |
332 | /* Create event on all UST registered apps for session */ | |
333 | ret = ust_app_create_event_glb(usess, uchan, uevent); | |
334 | } else { | |
335 | /* Enable event on all UST registered apps for session */ | |
336 | ret = ust_app_enable_event_glb(usess, uchan, uevent); | |
337 | } | |
48842b30 | 338 | |
7972aab2 DG |
339 | if (ret < 0) { |
340 | if (ret == -LTTNG_UST_ERR_EXIST) { | |
341 | ret = LTTNG_ERR_UST_EVENT_EXIST; | |
342 | goto end; | |
343 | } else { | |
344 | ret = LTTNG_ERR_UST_ENABLE_FAIL; | |
345 | goto error; | |
edb67388 | 346 | } |
2bdd86d4 | 347 | } |
48842b30 | 348 | |
7f79d3a1 | 349 | if (to_create) { |
fc34caaa | 350 | /* Add ltt ust event to channel */ |
18eace3b | 351 | add_unique_ust_event(uchan->events, uevent); |
7f79d3a1 | 352 | } |
edb67388 | 353 | |
7f79d3a1 DG |
354 | DBG("Event UST %s %s in channel %s", uevent->attr.name, |
355 | to_create ? "created" : "enabled", uchan->name); | |
356 | ||
f73fabfd | 357 | ret = LTTNG_OK; |
fc34caaa | 358 | |
fb89d070 | 359 | end: |
18eace3b | 360 | rcu_read_unlock(); |
49d21f93 MD |
361 | free(filter_expression); |
362 | free(filter); | |
363 | free(exclusion); | |
fc34caaa | 364 | return ret; |
edb67388 DG |
365 | |
366 | error: | |
fc34caaa DG |
367 | /* |
368 | * Only destroy event on creation time (not enabling time) because if the | |
369 | * event is found in the channel (to_create == 0), it means that at some | |
370 | * point the enable_event worked and it's thus valid to keep it alive. | |
371 | * Destroying it also implies that we also destroy it's shadow copy to sync | |
372 | * everyone up. | |
373 | */ | |
374 | if (to_create) { | |
375 | /* In this code path, the uevent was not added to the hash table */ | |
376 | trace_ust_destroy_event(uevent); | |
377 | } | |
18eace3b | 378 | rcu_read_unlock(); |
49d21f93 MD |
379 | free(filter_expression); |
380 | free(filter); | |
381 | free(exclusion); | |
2bdd86d4 MD |
382 | return ret; |
383 | } | |
384 | ||
7f79d3a1 DG |
385 | /* |
386 | * Disable UST tracepoint of a channel from a UST session. | |
387 | */ | |
7972aab2 | 388 | int event_ust_disable_tracepoint(struct ltt_ust_session *usess, |
7f79d3a1 | 389 | struct ltt_ust_channel *uchan, char *event_name) |
2bdd86d4 MD |
390 | { |
391 | int ret; | |
7f79d3a1 | 392 | struct ltt_ust_event *uevent; |
18eace3b DG |
393 | struct lttng_ht_node_str *node; |
394 | struct lttng_ht_iter iter; | |
18eace3b | 395 | struct lttng_ht *ht; |
2bdd86d4 | 396 | |
0525e9ae DG |
397 | assert(usess); |
398 | assert(uchan); | |
399 | assert(event_name); | |
400 | ||
18eace3b DG |
401 | ht = uchan->events; |
402 | ||
18eace3b | 403 | rcu_read_lock(); |
025faf73 DG |
404 | |
405 | /* | |
406 | * We use a custom lookup since we need the iterator for the next_duplicate | |
407 | * call in the do while loop below. | |
408 | */ | |
409 | cds_lfht_lookup(ht->ht, ht->hash_fct((void *) event_name, lttng_ht_seed), | |
410 | trace_ust_ht_match_event_by_name, event_name, &iter.iter); | |
18eace3b DG |
411 | node = lttng_ht_iter_get_node_str(&iter); |
412 | if (node == NULL) { | |
413 | DBG2("Trace UST event NOT found by name %s", event_name); | |
f73fabfd | 414 | ret = LTTNG_ERR_UST_EVENT_NOT_FOUND; |
7f79d3a1 | 415 | goto error; |
2bdd86d4 | 416 | } |
7f79d3a1 | 417 | |
18eace3b DG |
418 | do { |
419 | uevent = caa_container_of(node, struct ltt_ust_event, node); | |
025faf73 DG |
420 | assert(uevent); |
421 | ||
18eace3b DG |
422 | if (uevent->enabled == 0) { |
423 | /* It's already disabled so everything is OK */ | |
a1dcaf0f | 424 | goto next; |
7f79d3a1 | 425 | } |
18eace3b | 426 | |
7972aab2 DG |
427 | ret = ust_app_disable_event_glb(usess, uchan, uevent); |
428 | if (ret < 0 && ret != -LTTNG_UST_ERR_EXIST) { | |
429 | ret = LTTNG_ERR_UST_DISABLE_FAIL; | |
18eace3b DG |
430 | goto error; |
431 | } | |
18eace3b DG |
432 | uevent->enabled = 0; |
433 | ||
025faf73 DG |
434 | DBG2("Event UST %s disabled in channel %s", uevent->attr.name, |
435 | uchan->name); | |
436 | ||
a1dcaf0f | 437 | next: |
18eace3b DG |
438 | /* Get next duplicate event by name. */ |
439 | cds_lfht_next_duplicate(ht->ht, trace_ust_ht_match_event_by_name, | |
440 | event_name, &iter.iter); | |
441 | node = lttng_ht_iter_get_node_str(&iter); | |
442 | } while (node); | |
7f79d3a1 | 443 | |
f73fabfd | 444 | ret = LTTNG_OK; |
7f79d3a1 | 445 | |
7f79d3a1 | 446 | error: |
18eace3b | 447 | rcu_read_unlock(); |
7f79d3a1 DG |
448 | return ret; |
449 | } | |
450 | ||
451 | /* | |
452 | * Disable all UST tracepoints for a channel from a UST session. | |
453 | */ | |
7972aab2 | 454 | int event_ust_disable_all_tracepoints(struct ltt_ust_session *usess, |
7f79d3a1 DG |
455 | struct ltt_ust_channel *uchan) |
456 | { | |
6775595e | 457 | int ret, i, size; |
bec39940 | 458 | struct lttng_ht_iter iter; |
7f79d3a1 DG |
459 | struct ltt_ust_event *uevent = NULL; |
460 | struct lttng_event *events = NULL; | |
461 | ||
0525e9ae DG |
462 | assert(usess); |
463 | assert(uchan); | |
464 | ||
025faf73 DG |
465 | rcu_read_lock(); |
466 | ||
7972aab2 DG |
467 | /* Disabling existing events */ |
468 | cds_lfht_for_each_entry(uchan->events->ht, &iter.iter, uevent, | |
469 | node.node) { | |
470 | if (uevent->enabled == 1) { | |
471 | ret = event_ust_disable_tracepoint(usess, uchan, | |
472 | uevent->attr.name); | |
473 | if (ret < 0) { | |
7f79d3a1 DG |
474 | continue; |
475 | } | |
476 | } | |
7f79d3a1 | 477 | } |
7972aab2 DG |
478 | |
479 | /* Get all UST available events */ | |
480 | size = ust_app_list_events(&events); | |
481 | if (size < 0) { | |
482 | ret = LTTNG_ERR_UST_LIST_FAIL; | |
7f79d3a1 DG |
483 | goto error; |
484 | } | |
485 | ||
7972aab2 DG |
486 | for (i = 0; i < size; i++) { |
487 | ret = event_ust_disable_tracepoint(usess, uchan, | |
488 | events[i].name); | |
489 | if (ret != LTTNG_OK) { | |
490 | /* Continue to disable the rest... */ | |
491 | continue; | |
492 | } | |
493 | } | |
494 | free(events); | |
495 | ||
025faf73 | 496 | rcu_read_unlock(); |
f73fabfd | 497 | return LTTNG_OK; |
7f79d3a1 DG |
498 | |
499 | error: | |
500 | free(events); | |
025faf73 | 501 | rcu_read_unlock(); |
2bdd86d4 MD |
502 | return ret; |
503 | } | |
f20baf8e DG |
504 | |
505 | /* | |
022d91ba | 506 | * Enable all agent event for a given UST session. |
f20baf8e DG |
507 | * |
508 | * Return LTTNG_OK on success or else a LTTNG_ERR* code. | |
509 | */ | |
022d91ba | 510 | int event_agent_enable_all(struct ltt_ust_session *usess, |
fefd409b DG |
511 | struct agent *agt, struct lttng_event *event, |
512 | struct lttng_filter_bytecode *filter) | |
f20baf8e DG |
513 | { |
514 | int ret; | |
fefd409b | 515 | struct agent_event *aevent; |
f20baf8e DG |
516 | struct lttng_ht_iter iter; |
517 | ||
518 | assert(usess); | |
519 | ||
022d91ba | 520 | DBG("Event agent enabling ALL events for session %" PRIu64, usess->id); |
f20baf8e | 521 | |
022d91ba | 522 | /* Enable event on agent application through TCP socket. */ |
fefd409b | 523 | ret = event_agent_enable(usess, agt, event, filter); |
f20baf8e DG |
524 | if (ret != LTTNG_OK) { |
525 | goto error; | |
526 | } | |
527 | ||
528 | /* Flag every event that they are now enabled. */ | |
529 | rcu_read_lock(); | |
fefd409b | 530 | cds_lfht_for_each_entry(agt->events->ht, &iter.iter, aevent, |
f20baf8e | 531 | node.node) { |
fefd409b | 532 | aevent->enabled = 1; |
f20baf8e DG |
533 | } |
534 | rcu_read_unlock(); | |
535 | ||
536 | ret = LTTNG_OK; | |
537 | ||
538 | error: | |
539 | return ret; | |
540 | } | |
541 | ||
542 | /* | |
022d91ba | 543 | * Enable a single agent event for a given UST session. |
f20baf8e DG |
544 | * |
545 | * Return LTTNG_OK on success or else a LTTNG_ERR* code. | |
546 | */ | |
fefd409b DG |
547 | int event_agent_enable(struct ltt_ust_session *usess, |
548 | struct agent *agt, struct lttng_event *event, | |
be6a6276 | 549 | struct lttng_filter_bytecode *filter) |
f20baf8e DG |
550 | { |
551 | int ret, created = 0; | |
fefd409b | 552 | struct agent_event *aevent; |
f20baf8e DG |
553 | |
554 | assert(usess); | |
555 | assert(event); | |
fefd409b | 556 | assert(agt); |
f20baf8e | 557 | |
022d91ba | 558 | DBG("Event agent enabling %s for session %" PRIu64 " with loglevel type %d " |
b2064f54 DG |
559 | "and loglevel %d", event->name, usess->id, event->loglevel_type, |
560 | event->loglevel); | |
f20baf8e | 561 | |
fefd409b DG |
562 | aevent = agent_find_event(event->name, event->loglevel, agt); |
563 | if (!aevent) { | |
564 | aevent = agent_create_event(event->name, filter); | |
565 | if (!aevent) { | |
f20baf8e DG |
566 | ret = LTTNG_ERR_NOMEM; |
567 | goto error; | |
568 | } | |
fefd409b DG |
569 | aevent->loglevel = event->loglevel; |
570 | aevent->loglevel_type = event->loglevel_type; | |
f20baf8e DG |
571 | created = 1; |
572 | } | |
573 | ||
574 | /* Already enabled? */ | |
fefd409b | 575 | if (aevent->enabled) { |
f20baf8e DG |
576 | goto end; |
577 | } | |
578 | ||
fefd409b | 579 | ret = agent_enable_event(aevent, agt->domain); |
f20baf8e DG |
580 | if (ret != LTTNG_OK) { |
581 | goto error; | |
582 | } | |
583 | ||
584 | /* If the event was created prior to the enable, add it to the domain. */ | |
585 | if (created) { | |
fefd409b | 586 | agent_add_event(aevent, agt); |
f20baf8e DG |
587 | } |
588 | ||
589 | end: | |
590 | return LTTNG_OK; | |
591 | ||
592 | error: | |
593 | if (created) { | |
fefd409b | 594 | agent_destroy_event(aevent); |
f20baf8e DG |
595 | } |
596 | return ret; | |
597 | } | |
598 | ||
da6c3a50 DG |
599 | /* |
600 | * Return the agent default event name to use by testing if the process is root | |
601 | * or not. Return NULL on error. | |
602 | */ | |
603 | const char *event_get_default_agent_ust_name(enum lttng_domain_type domain) | |
604 | { | |
605 | const char *default_event_name = NULL; | |
606 | ||
0e115563 DG |
607 | switch (domain) { |
608 | case LTTNG_DOMAIN_LOG4J: | |
da6c3a50 | 609 | if (is_root) { |
0e115563 | 610 | default_event_name = DEFAULT_SYS_LOG4J_EVENT_NAME; |
da6c3a50 | 611 | } else { |
0e115563 | 612 | default_event_name = DEFAULT_USER_LOG4J_EVENT_NAME; |
da6c3a50 | 613 | } |
0e115563 DG |
614 | break; |
615 | case LTTNG_DOMAIN_JUL: | |
da6c3a50 | 616 | if (is_root) { |
0e115563 | 617 | default_event_name = DEFAULT_SYS_JUL_EVENT_NAME; |
da6c3a50 | 618 | } else { |
0e115563 | 619 | default_event_name = DEFAULT_USER_JUL_EVENT_NAME; |
da6c3a50 | 620 | } |
0e115563 DG |
621 | break; |
622 | case LTTNG_DOMAIN_PYTHON: | |
b1a92ef5 | 623 | default_event_name = DEFAULT_PYTHON_EVENT_NAME; |
0e115563 DG |
624 | break; |
625 | default: | |
da6c3a50 DG |
626 | assert(0); |
627 | } | |
628 | ||
629 | return default_event_name; | |
630 | } | |
631 | ||
f20baf8e | 632 | /* |
022d91ba | 633 | * Disable a single agent event for a given UST session. |
f20baf8e DG |
634 | * |
635 | * Return LTTNG_OK on success or else a LTTNG_ERR* code. | |
636 | */ | |
fefd409b DG |
637 | int event_agent_disable(struct ltt_ust_session *usess, struct agent *agt, |
638 | char *event_name) | |
f20baf8e DG |
639 | { |
640 | int ret; | |
fefd409b | 641 | struct agent_event *aevent; |
be6a6276 DG |
642 | struct ltt_ust_event *uevent = NULL; |
643 | struct ltt_ust_channel *uchan = NULL; | |
71aecbf8 | 644 | const char *ust_event_name, *ust_channel_name; |
f20baf8e | 645 | |
fefd409b | 646 | assert(agt); |
f20baf8e DG |
647 | assert(usess); |
648 | assert(event_name); | |
649 | ||
022d91ba | 650 | DBG("Event agent disabling %s for session %" PRIu64, event_name, usess->id); |
f20baf8e | 651 | |
fefd409b DG |
652 | aevent = agent_find_event_by_name(event_name, agt); |
653 | if (!aevent) { | |
f20baf8e DG |
654 | ret = LTTNG_ERR_UST_EVENT_NOT_FOUND; |
655 | goto error; | |
656 | } | |
657 | ||
658 | /* Already disabled? */ | |
fefd409b | 659 | if (!aevent->enabled) { |
f20baf8e DG |
660 | goto end; |
661 | } | |
662 | ||
71aecbf8 DG |
663 | if (agt->domain == LTTNG_DOMAIN_JUL) { |
664 | ust_channel_name = DEFAULT_JUL_CHANNEL_NAME; | |
665 | } else if (agt->domain == LTTNG_DOMAIN_LOG4J) { | |
666 | ust_channel_name = DEFAULT_LOG4J_CHANNEL_NAME; | |
0e115563 DG |
667 | } else if (agt->domain == LTTNG_DOMAIN_PYTHON) { |
668 | ust_channel_name = DEFAULT_PYTHON_CHANNEL_NAME; | |
71aecbf8 DG |
669 | } else { |
670 | ret = LTTNG_ERR_INVALID; | |
671 | goto error; | |
672 | } | |
673 | ||
be6a6276 DG |
674 | /* |
675 | * Disable it on the UST side. First get the channel reference then find | |
676 | * the event and finally disable it. | |
677 | */ | |
678 | uchan = trace_ust_find_channel_by_name(usess->domain_global.channels, | |
71aecbf8 | 679 | (char *) ust_channel_name); |
be6a6276 DG |
680 | if (!uchan) { |
681 | ret = LTTNG_ERR_UST_CHAN_NOT_FOUND; | |
682 | goto error; | |
683 | } | |
684 | ||
da6c3a50 DG |
685 | ust_event_name = event_get_default_agent_ust_name(agt->domain); |
686 | if (!ust_event_name) { | |
687 | ret = LTTNG_ERR_FATAL; | |
688 | goto error; | |
be6a6276 DG |
689 | } |
690 | ||
691 | /* | |
022d91ba | 692 | * The loglevel is hardcoded with 0 here since the agent ust event is set |
be6a6276 | 693 | * with the loglevel type to ALL thus the loglevel stays 0. The event's |
022d91ba | 694 | * filter is the one handling the loglevel for agent. |
be6a6276 | 695 | */ |
da6c3a50 | 696 | uevent = trace_ust_find_event(uchan->events, (char *) ust_event_name, |
fefd409b | 697 | aevent->filter, 0, NULL); |
022d91ba | 698 | /* If the agent event exists, it must be available on the UST side. */ |
be6a6276 DG |
699 | assert(uevent); |
700 | ||
701 | ret = ust_app_disable_event_glb(usess, uchan, uevent); | |
702 | if (ret < 0 && ret != -LTTNG_UST_ERR_EXIST) { | |
703 | ret = LTTNG_ERR_UST_DISABLE_FAIL; | |
704 | goto error; | |
705 | } | |
706 | ||
0dcfcf94 DG |
707 | /* |
708 | * Flag event that it's disabled so the shadow copy on the ust app side | |
709 | * will disable it if an application shows up. | |
710 | */ | |
711 | uevent->enabled = 0; | |
712 | ||
fefd409b | 713 | ret = agent_disable_event(aevent, agt->domain); |
f20baf8e DG |
714 | if (ret != LTTNG_OK) { |
715 | goto error; | |
716 | } | |
717 | ||
718 | end: | |
719 | return LTTNG_OK; | |
720 | ||
721 | error: | |
722 | return ret; | |
723 | } | |
724 | /* | |
022d91ba | 725 | * Disable all agent event for a given UST session. |
f20baf8e DG |
726 | * |
727 | * Return LTTNG_OK on success or else a LTTNG_ERR* code. | |
728 | */ | |
fefd409b DG |
729 | int event_agent_disable_all(struct ltt_ust_session *usess, |
730 | struct agent *agt) | |
f20baf8e | 731 | { |
0dcfcf94 | 732 | int ret; |
fefd409b | 733 | struct agent_event *aevent; |
f20baf8e DG |
734 | struct lttng_ht_iter iter; |
735 | ||
fefd409b | 736 | assert(agt); |
f20baf8e DG |
737 | assert(usess); |
738 | ||
0dcfcf94 DG |
739 | /* |
740 | * Disable event on agent application. Continue to disable all other events | |
741 | * if the * event is not found. | |
742 | */ | |
fefd409b | 743 | ret = event_agent_disable(usess, agt, "*"); |
0dcfcf94 DG |
744 | if (ret != LTTNG_OK && ret != LTTNG_ERR_UST_EVENT_NOT_FOUND) { |
745 | goto error; | |
f20baf8e DG |
746 | } |
747 | ||
748 | /* Flag every event that they are now enabled. */ | |
749 | rcu_read_lock(); | |
fefd409b | 750 | cds_lfht_for_each_entry(agt->events->ht, &iter.iter, aevent, |
f20baf8e | 751 | node.node) { |
0dcfcf94 DG |
752 | if (!aevent->enabled) { |
753 | continue; | |
754 | } | |
755 | ||
756 | ret = event_agent_disable(usess, agt, aevent->name); | |
757 | if (ret != LTTNG_OK) { | |
758 | rcu_read_unlock(); | |
759 | goto error; | |
f20baf8e | 760 | } |
f20baf8e DG |
761 | } |
762 | rcu_read_unlock(); | |
763 | ||
764 | ret = LTTNG_OK; | |
765 | ||
766 | error: | |
767 | return ret; | |
768 | } |