Cleanup: extract function to borrow hashlist bucket
[deliverable/lttng-ust.git] / liblttng-ust / lttng-events.c
1 /*
2 * lttng-events.c
3 *
4 * Holds LTTng per-session event registry.
5 *
6 * Copyright (C) 2010-2012 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
7 *
8 * This library is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; only
11 * version 2.1 of the License.
12 *
13 * This library is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
17 *
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with this library; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21 */
22
23 #define _LGPL_SOURCE
24 #include <stdio.h>
25 #include <urcu/list.h>
26 #include <urcu/hlist.h>
27 #include <pthread.h>
28 #include <errno.h>
29 #include <sys/shm.h>
30 #include <sys/ipc.h>
31 #include <stdint.h>
32 #include <stddef.h>
33 #include <inttypes.h>
34 #include <time.h>
35 #include <stdbool.h>
36 #include <lttng/ust-endian.h>
37 #include "clock.h"
38
39 #include <urcu-bp.h>
40 #include <urcu/compiler.h>
41 #include <urcu/uatomic.h>
42 #include <urcu/arch.h>
43
44 #include <lttng/tracepoint.h>
45 #include <lttng/ust-events.h>
46
47 #include <usterr-signal-safe.h>
48 #include <helper.h>
49 #include <lttng/ust-ctl.h>
50 #include <ust-comm.h>
51 #include <lttng/ust-dynamic-type.h>
52 #include <lttng/ust-context-provider.h>
53 #include "error.h"
54 #include "compat.h"
55 #include "lttng-ust-uuid.h"
56
57 #include "tracepoint-internal.h"
58 #include "string-utils.h"
59 #include "lttng-tracer.h"
60 #include "lttng-tracer-core.h"
61 #include "lttng-ust-statedump.h"
62 #include "ust-events-internal.h"
63 #include "wait.h"
64 #include "../libringbuffer/shm.h"
65 #include "jhash.h"
66
67 /*
68 * All operations within this file are called by the communication
69 * thread, under ust_lock protection.
70 */
71
72 static CDS_LIST_HEAD(sessions);
73
74 struct cds_list_head *_lttng_get_sessions(void)
75 {
76 return &sessions;
77 }
78
79 static void _lttng_event_destroy(struct lttng_event *event);
80 static void _lttng_enum_destroy(struct lttng_enum *_enum);
81
82 static
83 void lttng_session_lazy_sync_event_enablers(struct lttng_session *session);
84 static
85 void lttng_session_sync_event_enablers(struct lttng_session *session);
86 static
87 void lttng_enabler_destroy(struct lttng_enabler *enabler);
88
89 /*
90 * Called with ust lock held.
91 */
92 int lttng_session_active(void)
93 {
94 struct lttng_session *iter;
95
96 cds_list_for_each_entry(iter, &sessions, node) {
97 if (iter->active)
98 return 1;
99 }
100 return 0;
101 }
102
103 static
104 int lttng_loglevel_match(int loglevel,
105 unsigned int has_loglevel,
106 enum lttng_ust_loglevel_type req_type,
107 int req_loglevel)
108 {
109 if (!has_loglevel)
110 loglevel = TRACE_DEFAULT;
111 switch (req_type) {
112 case LTTNG_UST_LOGLEVEL_RANGE:
113 if (loglevel <= req_loglevel
114 || (req_loglevel == -1 && loglevel <= TRACE_DEBUG))
115 return 1;
116 else
117 return 0;
118 case LTTNG_UST_LOGLEVEL_SINGLE:
119 if (loglevel == req_loglevel
120 || (req_loglevel == -1 && loglevel <= TRACE_DEBUG))
121 return 1;
122 else
123 return 0;
124 case LTTNG_UST_LOGLEVEL_ALL:
125 default:
126 if (loglevel <= TRACE_DEBUG)
127 return 1;
128 else
129 return 0;
130 }
131 }
132
133 void synchronize_trace(void)
134 {
135 synchronize_rcu();
136 }
137
138 struct lttng_session *lttng_session_create(void)
139 {
140 struct lttng_session *session;
141 int i;
142
143 session = zmalloc(sizeof(struct lttng_session));
144 if (!session)
145 return NULL;
146 if (lttng_session_context_init(&session->ctx)) {
147 free(session);
148 return NULL;
149 }
150 CDS_INIT_LIST_HEAD(&session->chan_head);
151 CDS_INIT_LIST_HEAD(&session->events_head);
152 CDS_INIT_LIST_HEAD(&session->enums_head);
153 CDS_INIT_LIST_HEAD(&session->enablers_head);
154 for (i = 0; i < LTTNG_UST_EVENT_HT_SIZE; i++)
155 CDS_INIT_HLIST_HEAD(&session->events_ht.table[i]);
156 for (i = 0; i < LTTNG_UST_ENUM_HT_SIZE; i++)
157 CDS_INIT_HLIST_HEAD(&session->enums_ht.table[i]);
158 cds_list_add(&session->node, &sessions);
159 return session;
160 }
161
162 /*
163 * Only used internally at session destruction.
164 */
165 static
166 void _lttng_channel_unmap(struct lttng_channel *lttng_chan)
167 {
168 struct channel *chan;
169 struct lttng_ust_shm_handle *handle;
170
171 cds_list_del(&lttng_chan->node);
172 lttng_destroy_context(lttng_chan->ctx);
173 chan = lttng_chan->chan;
174 handle = lttng_chan->handle;
175 /*
176 * note: lttng_chan is private data contained within handle. It
177 * will be freed along with the handle.
178 */
179 channel_destroy(chan, handle, 0);
180 }
181
182 static
183 void register_event(struct lttng_event *event)
184 {
185 int ret;
186 const struct lttng_event_desc *desc;
187
188 assert(event->registered == 0);
189 desc = event->desc;
190 ret = __tracepoint_probe_register_queue_release(desc->name,
191 desc->probe_callback,
192 event, desc->signature);
193 WARN_ON_ONCE(ret);
194 if (!ret)
195 event->registered = 1;
196 }
197
198 static
199 void unregister_event(struct lttng_event *event)
200 {
201 int ret;
202 const struct lttng_event_desc *desc;
203
204 assert(event->registered == 1);
205 desc = event->desc;
206 ret = __tracepoint_probe_unregister_queue_release(desc->name,
207 desc->probe_callback,
208 event);
209 WARN_ON_ONCE(ret);
210 if (!ret)
211 event->registered = 0;
212 }
213
214 /*
215 * Only used internally at session destruction.
216 */
217 static
218 void _lttng_event_unregister(struct lttng_event *event)
219 {
220 if (event->registered)
221 unregister_event(event);
222 }
223
224 void lttng_session_destroy(struct lttng_session *session)
225 {
226 struct lttng_channel *chan, *tmpchan;
227 struct lttng_event *event, *tmpevent;
228 struct lttng_enum *_enum, *tmp_enum;
229 struct lttng_event_enabler *event_enabler, *event_tmpenabler;
230
231 CMM_ACCESS_ONCE(session->active) = 0;
232 cds_list_for_each_entry(event, &session->events_head, node) {
233 _lttng_event_unregister(event);
234 }
235 synchronize_trace(); /* Wait for in-flight events to complete */
236 __tracepoint_probe_prune_release_queue();
237 cds_list_for_each_entry_safe(event_enabler, event_tmpenabler,
238 &session->enablers_head, node)
239 lttng_event_enabler_destroy(event_enabler);
240 cds_list_for_each_entry_safe(event, tmpevent,
241 &session->events_head, node)
242 _lttng_event_destroy(event);
243 cds_list_for_each_entry_safe(_enum, tmp_enum,
244 &session->enums_head, node)
245 _lttng_enum_destroy(_enum);
246 cds_list_for_each_entry_safe(chan, tmpchan, &session->chan_head, node)
247 _lttng_channel_unmap(chan);
248 cds_list_del(&session->node);
249 lttng_destroy_context(session->ctx);
250 free(session);
251 }
252
253 static
254 void lttng_enabler_destroy(struct lttng_enabler *enabler)
255 {
256 struct lttng_ust_filter_bytecode_node *filter_node, *tmp_filter_node;
257 struct lttng_ust_excluder_node *excluder_node, *tmp_excluder_node;
258
259 if (!enabler) {
260 return;
261 }
262
263 /* Destroy filter bytecode */
264 cds_list_for_each_entry_safe(filter_node, tmp_filter_node,
265 &enabler->filter_bytecode_head, node) {
266 free(filter_node);
267 }
268
269 /* Destroy excluders */
270 cds_list_for_each_entry_safe(excluder_node, tmp_excluder_node,
271 &enabler->excluder_head, node) {
272 free(excluder_node);
273 }
274 }
275
276 static
277 int lttng_enum_create(const struct lttng_enum_desc *desc,
278 struct lttng_session *session)
279 {
280 const char *enum_name = desc->name;
281 struct lttng_enum *_enum;
282 struct cds_hlist_head *head;
283 int ret = 0;
284 size_t name_len = strlen(enum_name);
285 uint32_t hash;
286 int notify_socket;
287
288 /* Check if this enum is already registered for this session. */
289 hash = jhash(enum_name, name_len, 0);
290 head = &session->enums_ht.table[hash & (LTTNG_UST_ENUM_HT_SIZE - 1)];
291
292 _enum = lttng_ust_enum_get_from_desc(session, desc);
293 if (_enum) {
294 ret = -EEXIST;
295 goto exist;
296 }
297
298 notify_socket = lttng_get_notify_socket(session->owner);
299 if (notify_socket < 0) {
300 ret = notify_socket;
301 goto socket_error;
302 }
303
304 _enum = zmalloc(sizeof(*_enum));
305 if (!_enum) {
306 ret = -ENOMEM;
307 goto cache_error;
308 }
309 _enum->session = session;
310 _enum->desc = desc;
311
312 ret = ustcomm_register_enum(notify_socket,
313 session->objd,
314 enum_name,
315 desc->nr_entries,
316 desc->entries,
317 &_enum->id);
318 if (ret < 0) {
319 DBG("Error (%d) registering enumeration to sessiond", ret);
320 goto sessiond_register_error;
321 }
322 cds_list_add(&_enum->node, &session->enums_head);
323 cds_hlist_add_head(&_enum->hlist, head);
324 return 0;
325
326 sessiond_register_error:
327 free(_enum);
328 cache_error:
329 socket_error:
330 exist:
331 return ret;
332 }
333
334 static
335 int lttng_create_enum_check(const struct lttng_type *type,
336 struct lttng_session *session)
337 {
338 switch (type->atype) {
339 case atype_enum:
340 {
341 const struct lttng_enum_desc *enum_desc;
342 int ret;
343
344 enum_desc = type->u.legacy.basic.enumeration.desc;
345 ret = lttng_enum_create(enum_desc, session);
346 if (ret && ret != -EEXIST) {
347 DBG("Unable to create enum error: (%d)", ret);
348 return ret;
349 }
350 break;
351 }
352 case atype_enum_nestable:
353 {
354 const struct lttng_enum_desc *enum_desc;
355 int ret;
356
357 enum_desc = type->u.enum_nestable.desc;
358 ret = lttng_enum_create(enum_desc, session);
359 if (ret && ret != -EEXIST) {
360 DBG("Unable to create enum error: (%d)", ret);
361 return ret;
362 }
363 break;
364 }
365 case atype_dynamic:
366 {
367 const struct lttng_event_field *tag_field_generic;
368 const struct lttng_enum_desc *enum_desc;
369 int ret;
370
371 tag_field_generic = lttng_ust_dynamic_type_tag_field();
372 enum_desc = tag_field_generic->type.u.enum_nestable.desc;
373 ret = lttng_enum_create(enum_desc, session);
374 if (ret && ret != -EEXIST) {
375 DBG("Unable to create enum error: (%d)", ret);
376 return ret;
377 }
378 break;
379 }
380 default:
381 /* TODO: nested types when they become supported. */
382 break;
383 }
384 return 0;
385 }
386
387 static
388 int lttng_create_all_event_enums(size_t nr_fields,
389 const struct lttng_event_field *event_fields,
390 struct lttng_session *session)
391 {
392 size_t i;
393 int ret;
394
395 /* For each field, ensure enum is part of the session. */
396 for (i = 0; i < nr_fields; i++) {
397 const struct lttng_type *type = &event_fields[i].type;
398
399 ret = lttng_create_enum_check(type, session);
400 if (ret)
401 return ret;
402 }
403 return 0;
404 }
405
406 static
407 int lttng_create_all_ctx_enums(size_t nr_fields,
408 const struct lttng_ctx_field *ctx_fields,
409 struct lttng_session *session)
410 {
411 size_t i;
412 int ret;
413
414 /* For each field, ensure enum is part of the session. */
415 for (i = 0; i < nr_fields; i++) {
416 const struct lttng_type *type = &ctx_fields[i].event_field.type;
417
418 ret = lttng_create_enum_check(type, session);
419 if (ret)
420 return ret;
421 }
422 return 0;
423 }
424
425 /*
426 * Ensure that a state-dump will be performed for this session at the end
427 * of the current handle_message().
428 */
429 int lttng_session_statedump(struct lttng_session *session)
430 {
431 session->statedump_pending = 1;
432 lttng_ust_sockinfo_session_enabled(session->owner);
433 return 0;
434 }
435
436 int lttng_session_enable(struct lttng_session *session)
437 {
438 int ret = 0;
439 struct lttng_channel *chan;
440 int notify_socket;
441
442 if (session->active) {
443 ret = -EBUSY;
444 goto end;
445 }
446
447 notify_socket = lttng_get_notify_socket(session->owner);
448 if (notify_socket < 0)
449 return notify_socket;
450
451 /* Set transient enabler state to "enabled" */
452 session->tstate = 1;
453
454 /* We need to sync enablers with session before activation. */
455 lttng_session_sync_event_enablers(session);
456
457 /*
458 * Snapshot the number of events per channel to know the type of header
459 * we need to use.
460 */
461 cds_list_for_each_entry(chan, &session->chan_head, node) {
462 const struct lttng_ctx *ctx;
463 const struct lttng_ctx_field *fields = NULL;
464 size_t nr_fields = 0;
465 uint32_t chan_id;
466
467 /* don't change it if session stop/restart */
468 if (chan->header_type)
469 continue;
470 ctx = chan->ctx;
471 if (ctx) {
472 nr_fields = ctx->nr_fields;
473 fields = ctx->fields;
474 ret = lttng_create_all_ctx_enums(nr_fields, fields,
475 session);
476 if (ret < 0) {
477 DBG("Error (%d) adding enum to session", ret);
478 return ret;
479 }
480 }
481 ret = ustcomm_register_channel(notify_socket,
482 session,
483 session->objd,
484 chan->objd,
485 nr_fields,
486 fields,
487 &chan_id,
488 &chan->header_type);
489 if (ret) {
490 DBG("Error (%d) registering channel to sessiond", ret);
491 return ret;
492 }
493 if (chan_id != chan->id) {
494 DBG("Error: channel registration id (%u) does not match id assigned at creation (%u)",
495 chan_id, chan->id);
496 return -EINVAL;
497 }
498 }
499
500 /* Set atomically the state to "active" */
501 CMM_ACCESS_ONCE(session->active) = 1;
502 CMM_ACCESS_ONCE(session->been_active) = 1;
503
504 ret = lttng_session_statedump(session);
505 if (ret)
506 return ret;
507 end:
508 return ret;
509 }
510
511 int lttng_session_disable(struct lttng_session *session)
512 {
513 int ret = 0;
514
515 if (!session->active) {
516 ret = -EBUSY;
517 goto end;
518 }
519 /* Set atomically the state to "inactive" */
520 CMM_ACCESS_ONCE(session->active) = 0;
521
522 /* Set transient enabler state to "disabled" */
523 session->tstate = 0;
524 lttng_session_sync_event_enablers(session);
525 end:
526 return ret;
527 }
528
529 int lttng_channel_enable(struct lttng_channel *channel)
530 {
531 int ret = 0;
532
533 if (channel->enabled) {
534 ret = -EBUSY;
535 goto end;
536 }
537 /* Set transient enabler state to "enabled" */
538 channel->tstate = 1;
539 lttng_session_sync_event_enablers(channel->session);
540 /* Set atomically the state to "enabled" */
541 CMM_ACCESS_ONCE(channel->enabled) = 1;
542 end:
543 return ret;
544 }
545
546 int lttng_channel_disable(struct lttng_channel *channel)
547 {
548 int ret = 0;
549
550 if (!channel->enabled) {
551 ret = -EBUSY;
552 goto end;
553 }
554 /* Set atomically the state to "disabled" */
555 CMM_ACCESS_ONCE(channel->enabled) = 0;
556 /* Set transient enabler state to "enabled" */
557 channel->tstate = 0;
558 lttng_session_sync_event_enablers(channel->session);
559 end:
560 return ret;
561 }
562
563 static inline
564 struct cds_hlist_head *borrow_hash_table_bucket(
565 struct cds_hlist_head *hash_table,
566 unsigned int hash_table_size,
567 const struct lttng_event_desc *desc)
568 {
569 const char *event_name;
570 size_t name_len;
571 uint32_t hash;
572
573 event_name = desc->name;
574 name_len = strlen(event_name);
575
576 hash = jhash(event_name, name_len, 0);
577 return &hash_table[hash & (hash_table_size - 1)];
578 }
579
580 /*
581 * Supports event creation while tracing session is active.
582 */
583 static
584 int lttng_event_create(const struct lttng_event_desc *desc,
585 struct lttng_channel *chan)
586 {
587 struct lttng_event *event;
588 struct lttng_session *session = chan->session;
589 struct cds_hlist_head *head;
590 int ret = 0;
591 int notify_socket, loglevel;
592 const char *uri;
593
594 head = borrow_hash_table_bucket(chan->session->events_ht.table,
595 LTTNG_UST_EVENT_HT_SIZE, desc);
596
597 notify_socket = lttng_get_notify_socket(session->owner);
598 if (notify_socket < 0) {
599 ret = notify_socket;
600 goto socket_error;
601 }
602
603 ret = lttng_create_all_event_enums(desc->nr_fields, desc->fields,
604 session);
605 if (ret < 0) {
606 DBG("Error (%d) adding enum to session", ret);
607 goto create_enum_error;
608 }
609
610 /*
611 * Check if loglevel match. Refuse to connect event if not.
612 */
613 event = zmalloc(sizeof(struct lttng_event));
614 if (!event) {
615 ret = -ENOMEM;
616 goto cache_error;
617 }
618 event->chan = chan;
619
620 /* Event will be enabled by enabler sync. */
621 event->enabled = 0;
622 event->registered = 0;
623 CDS_INIT_LIST_HEAD(&event->bytecode_runtime_head);
624 CDS_INIT_LIST_HEAD(&event->enablers_ref_head);
625 event->desc = desc;
626
627 if (desc->loglevel)
628 loglevel = *(*event->desc->loglevel);
629 else
630 loglevel = TRACE_DEFAULT;
631 if (desc->u.ext.model_emf_uri)
632 uri = *(desc->u.ext.model_emf_uri);
633 else
634 uri = NULL;
635
636 /* Fetch event ID from sessiond */
637 ret = ustcomm_register_event(notify_socket,
638 session,
639 session->objd,
640 chan->objd,
641 desc->name,
642 loglevel,
643 desc->signature,
644 desc->nr_fields,
645 desc->fields,
646 uri,
647 &event->id);
648 if (ret < 0) {
649 DBG("Error (%d) registering event to sessiond", ret);
650 goto sessiond_register_error;
651 }
652
653 cds_list_add(&event->node, &chan->session->events_head);
654 cds_hlist_add_head(&event->hlist, head);
655 return 0;
656
657 sessiond_register_error:
658 free(event);
659 cache_error:
660 create_enum_error:
661 socket_error:
662 return ret;
663 }
664
665 static
666 int lttng_desc_match_star_glob_enabler(const struct lttng_event_desc *desc,
667 struct lttng_enabler *enabler)
668 {
669 int loglevel = 0;
670 unsigned int has_loglevel = 0;
671
672 assert(enabler->format_type == LTTNG_ENABLER_FORMAT_STAR_GLOB);
673 if (!strutils_star_glob_match(enabler->event_param.name, SIZE_MAX,
674 desc->name, SIZE_MAX))
675 return 0;
676 if (desc->loglevel) {
677 loglevel = *(*desc->loglevel);
678 has_loglevel = 1;
679 }
680 if (!lttng_loglevel_match(loglevel,
681 has_loglevel,
682 enabler->event_param.loglevel_type,
683 enabler->event_param.loglevel))
684 return 0;
685 return 1;
686 }
687
688 static
689 int lttng_desc_match_event_enabler(const struct lttng_event_desc *desc,
690 struct lttng_enabler *enabler)
691 {
692 int loglevel = 0;
693 unsigned int has_loglevel = 0;
694
695 assert(enabler->format_type == LTTNG_ENABLER_FORMAT_EVENT);
696 if (strcmp(desc->name, enabler->event_param.name))
697 return 0;
698 if (desc->loglevel) {
699 loglevel = *(*desc->loglevel);
700 has_loglevel = 1;
701 }
702 if (!lttng_loglevel_match(loglevel,
703 has_loglevel,
704 enabler->event_param.loglevel_type,
705 enabler->event_param.loglevel))
706 return 0;
707 return 1;
708 }
709
710 static
711 int lttng_desc_match_enabler(const struct lttng_event_desc *desc,
712 struct lttng_enabler *enabler)
713 {
714 switch (enabler->format_type) {
715 case LTTNG_ENABLER_FORMAT_STAR_GLOB:
716 {
717 struct lttng_ust_excluder_node *excluder;
718
719 if (!lttng_desc_match_star_glob_enabler(desc, enabler)) {
720 return 0;
721 }
722
723 /*
724 * If the matching event matches with an excluder,
725 * return 'does not match'
726 */
727 cds_list_for_each_entry(excluder, &enabler->excluder_head, node) {
728 int count;
729
730 for (count = 0; count < excluder->excluder.count; count++) {
731 int len;
732 char *excluder_name;
733
734 excluder_name = (char *) (excluder->excluder.names)
735 + count * LTTNG_UST_SYM_NAME_LEN;
736 len = strnlen(excluder_name, LTTNG_UST_SYM_NAME_LEN);
737 if (len > 0 && strutils_star_glob_match(excluder_name, len, desc->name, SIZE_MAX))
738 return 0;
739 }
740 }
741 return 1;
742 }
743 case LTTNG_ENABLER_FORMAT_EVENT:
744 return lttng_desc_match_event_enabler(desc, enabler);
745 default:
746 return -EINVAL;
747 }
748 }
749
750 static
751 int lttng_event_enabler_match_event(struct lttng_event_enabler *event_enabler,
752 struct lttng_event *event)
753 {
754 if (lttng_desc_match_enabler(event->desc,
755 lttng_event_enabler_as_enabler(event_enabler))
756 && event->chan == event_enabler->chan)
757 return 1;
758 else
759 return 0;
760 }
761
762 static
763 struct lttng_enabler_ref *lttng_enabler_ref(
764 struct cds_list_head *enabler_ref_list,
765 struct lttng_enabler *enabler)
766 {
767 struct lttng_enabler_ref *enabler_ref;
768
769 cds_list_for_each_entry(enabler_ref, enabler_ref_list, node) {
770 if (enabler_ref->ref == enabler)
771 return enabler_ref;
772 }
773 return NULL;
774 }
775
776 /*
777 * Create struct lttng_event if it is missing and present in the list of
778 * tracepoint probes.
779 */
780 static
781 void lttng_create_event_if_missing(struct lttng_event_enabler *event_enabler)
782 {
783 struct lttng_session *session = event_enabler->chan->session;
784 struct lttng_probe_desc *probe_desc;
785 const struct lttng_event_desc *desc;
786 struct lttng_event *event;
787 int i;
788 struct cds_list_head *probe_list;
789
790 probe_list = lttng_get_probe_list_head();
791 /*
792 * For each probe event, if we find that a probe event matches
793 * our enabler, create an associated lttng_event if not
794 * already present.
795 */
796 cds_list_for_each_entry(probe_desc, probe_list, head) {
797 for (i = 0; i < probe_desc->nr_events; i++) {
798 int ret;
799 bool found = false;
800 struct cds_hlist_head *head;
801 struct cds_hlist_node *node;
802
803 desc = probe_desc->event_desc[i];
804 if (!lttng_desc_match_enabler(desc,
805 lttng_event_enabler_as_enabler(event_enabler)))
806 continue;
807
808 head = borrow_hash_table_bucket(
809 session->events_ht.table,
810 LTTNG_UST_EVENT_HT_SIZE, desc);
811
812 cds_hlist_for_each_entry(event, node, head, hlist) {
813 if (event->desc == desc
814 && event->chan == event_enabler->chan) {
815 found = true;
816 break;
817 }
818 }
819 if (found)
820 continue;
821
822 /*
823 * We need to create an event for this
824 * event probe.
825 */
826 ret = lttng_event_create(probe_desc->event_desc[i],
827 event_enabler->chan);
828 if (ret) {
829 DBG("Unable to create event %s, error %d\n",
830 probe_desc->event_desc[i]->name, ret);
831 }
832 }
833 }
834 }
835
836 /*
837 * Iterate over all the UST sessions to unregister and destroy all probes from
838 * the probe provider descriptor received as argument. Must me called with the
839 * ust_lock held.
840 */
841 void lttng_probe_provider_unregister_events(struct lttng_probe_desc *provider_desc)
842 {
843 struct cds_hlist_node *node, *tmp_node;
844 struct cds_list_head *sessionsp;
845 struct lttng_session *session;
846 struct cds_hlist_head *head;
847 struct lttng_event *event;
848 unsigned int i, j;
849
850 /* Get handle on list of sessions. */
851 sessionsp = _lttng_get_sessions();
852
853 /*
854 * Iterate over all events in the probe provider descriptions and sessions
855 * to queue the unregistration of the events.
856 */
857 for (i = 0; i < provider_desc->nr_events; i++) {
858 const struct lttng_event_desc *event_desc;
859
860 event_desc = provider_desc->event_desc[i];
861
862 /* Iterate over all session to find the current event description. */
863 cds_list_for_each_entry(session, sessionsp, node) {
864 /*
865 * Get the list of events in the hashtable bucket and iterate to
866 * find the event matching this descriptor.
867 */
868 head = borrow_hash_table_bucket(
869 session->events_ht.table,
870 LTTNG_UST_EVENT_HT_SIZE, event_desc);
871
872 cds_hlist_for_each_entry(event, node, head, hlist) {
873 if (event_desc == event->desc) {
874 /* Queue the unregistration of this event. */
875 _lttng_event_unregister(event);
876 break;
877 }
878 }
879 }
880 }
881
882 /* Wait for grace period. */
883 synchronize_trace();
884 /* Prune the unregistration queue. */
885 __tracepoint_probe_prune_release_queue();
886
887 /*
888 * It is now safe to destroy the events and remove them from the event list
889 * and hashtables.
890 */
891 for (i = 0; i < provider_desc->nr_events; i++) {
892 const struct lttng_event_desc *event_desc;
893
894 event_desc = provider_desc->event_desc[i];
895
896 /* Iterate over all sessions to find the current event description. */
897 cds_list_for_each_entry(session, sessionsp, node) {
898 /*
899 * Get the list of events in the hashtable bucket and iterate to
900 * find the event matching this descriptor.
901 */
902 head = borrow_hash_table_bucket(
903 session->events_ht.table,
904 LTTNG_UST_EVENT_HT_SIZE, event_desc);
905
906 cds_hlist_for_each_entry_safe(event, node, tmp_node, head, hlist) {
907 if (event_desc == event->desc) {
908 /* Destroy enums of the current event. */
909 for (j = 0; j < event->desc->nr_fields; j++) {
910 const struct lttng_enum_desc *enum_desc;
911 const struct lttng_event_field *field;
912 struct lttng_enum *curr_enum;
913
914 field = &(event->desc->fields[j]);
915 switch (field->type.atype) {
916 case atype_enum:
917 enum_desc = field->type.u.legacy.basic.enumeration.desc;
918 break;
919 case atype_enum_nestable:
920 enum_desc = field->type.u.enum_nestable.desc;
921 break;
922 default:
923 continue;
924 }
925 curr_enum = lttng_ust_enum_get_from_desc(session, enum_desc);
926 if (curr_enum) {
927 _lttng_enum_destroy(curr_enum);
928 }
929 }
930
931 /* Destroy event. */
932 _lttng_event_destroy(event);
933 break;
934 }
935 }
936 }
937 }
938 }
939
940 /*
941 * Create events associated with an enabler (if not already present),
942 * and add backward reference from the event to the enabler.
943 */
944 static
945 int lttng_event_enabler_ref_events(struct lttng_event_enabler *event_enabler)
946 {
947 struct lttng_session *session = event_enabler->chan->session;
948 struct lttng_event *event;
949
950 if (!lttng_event_enabler_as_enabler(event_enabler)->enabled)
951 goto end;
952
953 /* First ensure that probe events are created for this enabler. */
954 lttng_create_event_if_missing(event_enabler);
955
956 /* For each event matching enabler in session event list. */
957 cds_list_for_each_entry(event, &session->events_head, node) {
958 struct lttng_enabler_ref *enabler_ref;
959
960 if (!lttng_event_enabler_match_event(event_enabler, event))
961 continue;
962
963 enabler_ref = lttng_enabler_ref(&event->enablers_ref_head,
964 lttng_event_enabler_as_enabler(event_enabler));
965 if (!enabler_ref) {
966 /*
967 * If no backward ref, create it.
968 * Add backward ref from event to enabler.
969 */
970 enabler_ref = zmalloc(sizeof(*enabler_ref));
971 if (!enabler_ref)
972 return -ENOMEM;
973 enabler_ref->ref = lttng_event_enabler_as_enabler(
974 event_enabler);
975 cds_list_add(&enabler_ref->node,
976 &event->enablers_ref_head);
977 }
978
979 /*
980 * Link filter bytecodes if not linked yet.
981 */
982 lttng_enabler_link_bytecode(event->desc,
983 &session->ctx,
984 &event->bytecode_runtime_head,
985 lttng_event_enabler_as_enabler(event_enabler));
986
987 /* TODO: merge event context. */
988 }
989 end:
990 return 0;
991 }
992
993 /*
994 * Called at library load: connect the probe on all enablers matching
995 * this event.
996 * Called with session mutex held.
997 */
998 int lttng_fix_pending_events(void)
999 {
1000 struct lttng_session *session;
1001
1002 cds_list_for_each_entry(session, &sessions, node) {
1003 lttng_session_lazy_sync_event_enablers(session);
1004 }
1005 return 0;
1006 }
1007
1008 /*
1009 * For each session of the owner thread, execute pending statedump.
1010 * Only dump state for the sessions owned by the caller thread, because
1011 * we don't keep ust_lock across the entire iteration.
1012 */
1013 void lttng_handle_pending_statedump(void *owner)
1014 {
1015 struct lttng_session *session;
1016
1017 /* Execute state dump */
1018 do_lttng_ust_statedump(owner);
1019
1020 /* Clear pending state dump */
1021 if (ust_lock()) {
1022 goto end;
1023 }
1024 cds_list_for_each_entry(session, &sessions, node) {
1025 if (session->owner != owner)
1026 continue;
1027 if (!session->statedump_pending)
1028 continue;
1029 session->statedump_pending = 0;
1030 }
1031 end:
1032 ust_unlock();
1033 return;
1034 }
1035
1036 /*
1037 * Only used internally at session destruction.
1038 */
1039 static
1040 void _lttng_event_destroy(struct lttng_event *event)
1041 {
1042 struct lttng_enabler_ref *enabler_ref, *tmp_enabler_ref;
1043
1044 /* Remove from event list. */
1045 cds_list_del(&event->node);
1046 /* Remove from event hash table. */
1047 cds_hlist_del(&event->hlist);
1048
1049 lttng_destroy_context(event->ctx);
1050 lttng_free_event_filter_runtime(event);
1051 /* Free event enabler refs */
1052 cds_list_for_each_entry_safe(enabler_ref, tmp_enabler_ref,
1053 &event->enablers_ref_head, node)
1054 free(enabler_ref);
1055 free(event);
1056 }
1057
1058 static
1059 void _lttng_enum_destroy(struct lttng_enum *_enum)
1060 {
1061 cds_list_del(&_enum->node);
1062 cds_hlist_del(&_enum->hlist);
1063 free(_enum);
1064 }
1065
1066 void lttng_ust_events_exit(void)
1067 {
1068 struct lttng_session *session, *tmpsession;
1069
1070 cds_list_for_each_entry_safe(session, tmpsession, &sessions, node)
1071 lttng_session_destroy(session);
1072 }
1073
1074 /*
1075 * Enabler management.
1076 */
1077 struct lttng_event_enabler *lttng_event_enabler_create(
1078 enum lttng_enabler_format_type format_type,
1079 struct lttng_ust_event *event_param,
1080 struct lttng_channel *chan)
1081 {
1082 struct lttng_event_enabler *event_enabler;
1083
1084 event_enabler = zmalloc(sizeof(*event_enabler));
1085 if (!event_enabler)
1086 return NULL;
1087 event_enabler->base.format_type = format_type;
1088 CDS_INIT_LIST_HEAD(&event_enabler->base.filter_bytecode_head);
1089 CDS_INIT_LIST_HEAD(&event_enabler->base.excluder_head);
1090 memcpy(&event_enabler->base.event_param, event_param,
1091 sizeof(event_enabler->base.event_param));
1092 event_enabler->chan = chan;
1093 /* ctx left NULL */
1094 event_enabler->base.enabled = 0;
1095 cds_list_add(&event_enabler->node, &event_enabler->chan->session->enablers_head);
1096 lttng_session_lazy_sync_event_enablers(event_enabler->chan->session);
1097
1098 return event_enabler;
1099 }
1100
1101 int lttng_event_enabler_enable(struct lttng_event_enabler *event_enabler)
1102 {
1103 lttng_event_enabler_as_enabler(event_enabler)->enabled = 1;
1104 lttng_session_lazy_sync_event_enablers(event_enabler->chan->session);
1105
1106 return 0;
1107 }
1108
1109 int lttng_event_enabler_disable(struct lttng_event_enabler *event_enabler)
1110 {
1111 lttng_event_enabler_as_enabler(event_enabler)->enabled = 0;
1112 lttng_session_lazy_sync_event_enablers(event_enabler->chan->session);
1113
1114 return 0;
1115 }
1116
1117 static
1118 void _lttng_enabler_attach_bytecode(struct lttng_enabler *enabler,
1119 struct lttng_ust_filter_bytecode_node *bytecode)
1120 {
1121 bytecode->enabler = enabler;
1122 cds_list_add_tail(&bytecode->node, &enabler->filter_bytecode_head);
1123 }
1124
1125 int lttng_event_enabler_attach_bytecode(struct lttng_event_enabler *event_enabler,
1126 struct lttng_ust_filter_bytecode_node *bytecode)
1127 {
1128 _lttng_enabler_attach_bytecode(
1129 lttng_event_enabler_as_enabler(event_enabler), bytecode);
1130
1131 lttng_session_lazy_sync_event_enablers(event_enabler->chan->session);
1132 return 0;
1133 }
1134
1135 static
1136 void _lttng_enabler_attach_exclusion(struct lttng_enabler *enabler,
1137 struct lttng_ust_excluder_node *excluder)
1138 {
1139 excluder->enabler = enabler;
1140 cds_list_add_tail(&excluder->node, &enabler->excluder_head);
1141 }
1142
1143 int lttng_event_enabler_attach_exclusion(struct lttng_event_enabler *event_enabler,
1144 struct lttng_ust_excluder_node *excluder)
1145 {
1146 _lttng_enabler_attach_exclusion(
1147 lttng_event_enabler_as_enabler(event_enabler), excluder);
1148
1149 lttng_session_lazy_sync_event_enablers(event_enabler->chan->session);
1150 return 0;
1151 }
1152
1153 int lttng_attach_context(struct lttng_ust_context *context_param,
1154 union ust_args *uargs,
1155 struct lttng_ctx **ctx, struct lttng_session *session)
1156 {
1157 /*
1158 * We cannot attach a context after trace has been started for a
1159 * session because the metadata does not allow expressing this
1160 * information outside of the original channel scope.
1161 */
1162 if (session->been_active)
1163 return -EPERM;
1164
1165 switch (context_param->ctx) {
1166 case LTTNG_UST_CONTEXT_PTHREAD_ID:
1167 return lttng_add_pthread_id_to_ctx(ctx);
1168 case LTTNG_UST_CONTEXT_PERF_THREAD_COUNTER:
1169 {
1170 struct lttng_ust_perf_counter_ctx *perf_ctx_param;
1171
1172 perf_ctx_param = &context_param->u.perf_counter;
1173 return lttng_add_perf_counter_to_ctx(
1174 perf_ctx_param->type,
1175 perf_ctx_param->config,
1176 perf_ctx_param->name,
1177 ctx);
1178 }
1179 case LTTNG_UST_CONTEXT_VTID:
1180 return lttng_add_vtid_to_ctx(ctx);
1181 case LTTNG_UST_CONTEXT_VPID:
1182 return lttng_add_vpid_to_ctx(ctx);
1183 case LTTNG_UST_CONTEXT_PROCNAME:
1184 return lttng_add_procname_to_ctx(ctx);
1185 case LTTNG_UST_CONTEXT_IP:
1186 return lttng_add_ip_to_ctx(ctx);
1187 case LTTNG_UST_CONTEXT_CPU_ID:
1188 return lttng_add_cpu_id_to_ctx(ctx);
1189 case LTTNG_UST_CONTEXT_APP_CONTEXT:
1190 return lttng_ust_add_app_context_to_ctx_rcu(uargs->app_context.ctxname,
1191 ctx);
1192 case LTTNG_UST_CONTEXT_CGROUP_NS:
1193 return lttng_add_cgroup_ns_to_ctx(ctx);
1194 case LTTNG_UST_CONTEXT_IPC_NS:
1195 return lttng_add_ipc_ns_to_ctx(ctx);
1196 case LTTNG_UST_CONTEXT_MNT_NS:
1197 return lttng_add_mnt_ns_to_ctx(ctx);
1198 case LTTNG_UST_CONTEXT_NET_NS:
1199 return lttng_add_net_ns_to_ctx(ctx);
1200 case LTTNG_UST_CONTEXT_PID_NS:
1201 return lttng_add_pid_ns_to_ctx(ctx);
1202 case LTTNG_UST_CONTEXT_TIME_NS:
1203 return lttng_add_time_ns_to_ctx(ctx);
1204 case LTTNG_UST_CONTEXT_USER_NS:
1205 return lttng_add_user_ns_to_ctx(ctx);
1206 case LTTNG_UST_CONTEXT_UTS_NS:
1207 return lttng_add_uts_ns_to_ctx(ctx);
1208 case LTTNG_UST_CONTEXT_VUID:
1209 return lttng_add_vuid_to_ctx(ctx);
1210 case LTTNG_UST_CONTEXT_VEUID:
1211 return lttng_add_veuid_to_ctx(ctx);
1212 case LTTNG_UST_CONTEXT_VSUID:
1213 return lttng_add_vsuid_to_ctx(ctx);
1214 case LTTNG_UST_CONTEXT_VGID:
1215 return lttng_add_vgid_to_ctx(ctx);
1216 case LTTNG_UST_CONTEXT_VEGID:
1217 return lttng_add_vegid_to_ctx(ctx);
1218 case LTTNG_UST_CONTEXT_VSGID:
1219 return lttng_add_vsgid_to_ctx(ctx);
1220 default:
1221 return -EINVAL;
1222 }
1223 }
1224
1225 int lttng_event_enabler_attach_context(struct lttng_event_enabler *enabler,
1226 struct lttng_ust_context *context_param)
1227 {
1228 return -ENOSYS;
1229 }
1230
1231 void lttng_event_enabler_destroy(struct lttng_event_enabler *event_enabler)
1232 {
1233 if (!event_enabler) {
1234 return;
1235 }
1236 cds_list_del(&event_enabler->node);
1237
1238 lttng_enabler_destroy(lttng_event_enabler_as_enabler(event_enabler));
1239
1240 lttng_destroy_context(event_enabler->ctx);
1241 free(event_enabler);
1242 }
1243
1244 /*
1245 * lttng_session_sync_event_enablers should be called just before starting a
1246 * session.
1247 */
1248 static
1249 void lttng_session_sync_event_enablers(struct lttng_session *session)
1250 {
1251 struct lttng_event_enabler *event_enabler;
1252 struct lttng_event *event;
1253
1254 cds_list_for_each_entry(event_enabler, &session->enablers_head, node)
1255 lttng_event_enabler_ref_events(event_enabler);
1256 /*
1257 * For each event, if at least one of its enablers is enabled,
1258 * and its channel and session transient states are enabled, we
1259 * enable the event, else we disable it.
1260 */
1261 cds_list_for_each_entry(event, &session->events_head, node) {
1262 struct lttng_enabler_ref *enabler_ref;
1263 struct lttng_bytecode_runtime *runtime;
1264 int enabled = 0, has_enablers_without_bytecode = 0;
1265
1266 /* Enable events */
1267 cds_list_for_each_entry(enabler_ref,
1268 &event->enablers_ref_head, node) {
1269 if (enabler_ref->ref->enabled) {
1270 enabled = 1;
1271 break;
1272 }
1273 }
1274 /*
1275 * Enabled state is based on union of enablers, with
1276 * intesection of session and channel transient enable
1277 * states.
1278 */
1279 enabled = enabled && session->tstate && event->chan->tstate;
1280
1281 CMM_STORE_SHARED(event->enabled, enabled);
1282 /*
1283 * Sync tracepoint registration with event enabled
1284 * state.
1285 */
1286 if (enabled) {
1287 if (!event->registered)
1288 register_event(event);
1289 } else {
1290 if (event->registered)
1291 unregister_event(event);
1292 }
1293
1294 /* Check if has enablers without bytecode enabled */
1295 cds_list_for_each_entry(enabler_ref,
1296 &event->enablers_ref_head, node) {
1297 if (enabler_ref->ref->enabled
1298 && cds_list_empty(&enabler_ref->ref->filter_bytecode_head)) {
1299 has_enablers_without_bytecode = 1;
1300 break;
1301 }
1302 }
1303 event->has_enablers_without_bytecode =
1304 has_enablers_without_bytecode;
1305
1306 /* Enable filters */
1307 cds_list_for_each_entry(runtime,
1308 &event->bytecode_runtime_head, node) {
1309 lttng_filter_sync_state(runtime);
1310 }
1311 }
1312 __tracepoint_probe_prune_release_queue();
1313 }
1314
1315 /*
1316 * Apply enablers to session events, adding events to session if need
1317 * be. It is required after each modification applied to an active
1318 * session, and right before session "start".
1319 * "lazy" sync means we only sync if required.
1320 */
1321 static
1322 void lttng_session_lazy_sync_event_enablers(struct lttng_session *session)
1323 {
1324 /* We can skip if session is not active */
1325 if (!session->active)
1326 return;
1327 lttng_session_sync_event_enablers(session);
1328 }
1329
1330 /*
1331 * Update all sessions with the given app context.
1332 * Called with ust lock held.
1333 * This is invoked when an application context gets loaded/unloaded. It
1334 * ensures the context callbacks are in sync with the application
1335 * context (either app context callbacks, or dummy callbacks).
1336 */
1337 void lttng_ust_context_set_session_provider(const char *name,
1338 size_t (*get_size)(struct lttng_ctx_field *field, size_t offset),
1339 void (*record)(struct lttng_ctx_field *field,
1340 struct lttng_ust_lib_ring_buffer_ctx *ctx,
1341 struct lttng_channel *chan),
1342 void (*get_value)(struct lttng_ctx_field *field,
1343 struct lttng_ctx_value *value))
1344 {
1345 struct lttng_session *session;
1346
1347 cds_list_for_each_entry(session, &sessions, node) {
1348 struct lttng_channel *chan;
1349 struct lttng_event *event;
1350 int ret;
1351
1352 ret = lttng_ust_context_set_provider_rcu(&session->ctx,
1353 name, get_size, record, get_value);
1354 if (ret)
1355 abort();
1356 cds_list_for_each_entry(chan, &session->chan_head, node) {
1357 ret = lttng_ust_context_set_provider_rcu(&chan->ctx,
1358 name, get_size, record, get_value);
1359 if (ret)
1360 abort();
1361 }
1362 cds_list_for_each_entry(event, &session->events_head, node) {
1363 ret = lttng_ust_context_set_provider_rcu(&event->ctx,
1364 name, get_size, record, get_value);
1365 if (ret)
1366 abort();
1367 }
1368 }
1369 }
This page took 0.098422 seconds and 5 git commands to generate.