Fix: don't chain RCU free
[lttng-tools.git] / src / bin / lttng-sessiond / trace-ust.c
CommitLineData
97ee3a89
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.
97ee3a89 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.
97ee3a89 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.
97ee3a89
DG
16 */
17
18#define _GNU_SOURCE
6c1c0768 19#define _LGPL_SOURCE
97ee3a89
DG
20#include <stdio.h>
21#include <stdlib.h>
22#include <string.h>
23#include <unistd.h>
d9bf3ca4 24#include <inttypes.h>
97ee3a89 25
990570ed
DG
26#include <common/common.h>
27#include <common/defaults.h>
97ee3a89 28
7972aab2 29#include "buffer-registry.h"
97ee3a89 30#include "trace-ust.h"
0b2dc8df 31#include "utils.h"
a9ad0c8f 32#include "ust-app.h"
7c1d2758 33#include "agent.h"
97ee3a89 34
025faf73
DG
35/*
36 * Match function for the events hash table lookup.
37 *
38 * Matches by name only. Used by the disable command.
39 */
18eace3b
DG
40int trace_ust_ht_match_event_by_name(struct cds_lfht_node *node,
41 const void *_key)
42{
43 struct ltt_ust_event *event;
44 const char *name;
45
46 assert(node);
47 assert(_key);
48
49 event = caa_container_of(node, struct ltt_ust_event, node.node);
50 name = _key;
51
52 /* Event name */
53 if (strncmp(event->attr.name, name, sizeof(event->attr.name)) != 0) {
54 goto no_match;
55 }
56
025faf73 57 /* Match */
18eace3b
DG
58 return 1;
59
60no_match:
61 return 0;
62}
63
025faf73
DG
64/*
65 * Match function for the hash table lookup.
66 *
67 * It matches an ust event based on three attributes which are the event name,
68 * the filter bytecode and the loglevel.
69 */
18eace3b
DG
70int trace_ust_ht_match_event(struct cds_lfht_node *node, const void *_key)
71{
72 struct ltt_ust_event *event;
73 const struct ltt_ust_ht_key *key;
74
75 assert(node);
76 assert(_key);
77
78 event = caa_container_of(node, struct ltt_ust_event, node.node);
79 key = _key;
80
f19e9f8b 81 /* Match the 4 elements of the key: name, filter, loglevel, exclusions. */
18eace3b
DG
82
83 /* Event name */
84 if (strncmp(event->attr.name, key->name, sizeof(event->attr.name)) != 0) {
18eace3b
DG
85 goto no_match;
86 }
87
88 /* Event loglevel. */
89 if (event->attr.loglevel != key->loglevel) {
025faf73
DG
90 if (event->attr.loglevel_type == LTTNG_UST_LOGLEVEL_ALL
91 && key->loglevel == 0 && event->attr.loglevel == -1) {
18eace3b 92 /*
025faf73
DG
93 * Match is accepted. This is because on event creation, the
94 * loglevel is set to -1 if the event loglevel type is ALL so 0 and
95 * -1 are accepted for this loglevel type since 0 is the one set by
96 * the API when receiving an enable event.
18eace3b
DG
97 */
98 } else {
18eace3b
DG
99 goto no_match;
100 }
101 }
102
103 /* Only one of the filters is NULL, fail. */
104 if ((key->filter && !event->filter) || (!key->filter && event->filter)) {
18eace3b
DG
105 goto no_match;
106 }
107
025faf73
DG
108 if (key->filter && event->filter) {
109 /* Both filters exists, check length followed by the bytecode. */
110 if (event->filter->len != key->filter->len ||
111 memcmp(event->filter->data, key->filter->data,
112 event->filter->len) != 0) {
113 goto no_match;
114 }
18eace3b
DG
115 }
116
f19e9f8b
JI
117 /* If only one of the exclusions is NULL, fail. */
118 if ((key->exclusion && !event->exclusion) || (!key->exclusion && event->exclusion)) {
119 goto no_match;
120 }
121
122 if (key->exclusion && event->exclusion) {
123 /* Both exclusions exist; check count followed by names. */
124 if (event->exclusion->count != key->exclusion->count ||
125 memcmp(event->exclusion->names, key->exclusion->names,
126 event->exclusion->count * LTTNG_SYMBOL_NAME_LEN) != 0) {
127 goto no_match;
128 }
129 }
025faf73
DG
130 /* Match. */
131 return 1;
18eace3b
DG
132
133no_match:
134 return 0;
18eace3b
DG
135}
136
0177d773 137/*
2223c96f
DG
138 * Find the channel in the hashtable and return channel pointer. RCU read side
139 * lock MUST be acquired before calling this.
0177d773 140 */
bec39940 141struct ltt_ust_channel *trace_ust_find_channel_by_name(struct lttng_ht *ht,
f6a9efaa 142 char *name)
0177d773 143{
bec39940
DG
144 struct lttng_ht_node_str *node;
145 struct lttng_ht_iter iter;
0177d773 146
85076754
MD
147 /*
148 * If we receive an empty string for channel name, it means the
149 * default channel name is requested.
150 */
151 if (name[0] == '\0')
152 name = DEFAULT_CHANNEL_NAME;
153
bec39940
DG
154 lttng_ht_lookup(ht, (void *)name, &iter);
155 node = lttng_ht_iter_get_node_str(&iter);
f6a9efaa 156 if (node == NULL) {
44d3bd01
DG
157 goto error;
158 }
159
f6a9efaa 160 DBG2("Trace UST channel %s found by name", name);
0177d773 161
f6a9efaa 162 return caa_container_of(node, struct ltt_ust_channel, node);
97ee3a89
DG
163
164error:
f6a9efaa 165 DBG2("Trace UST channel %s not found by name", name);
97ee3a89
DG
166 return NULL;
167}
168
169/*
2223c96f
DG
170 * Find the event in the hashtable and return event pointer. RCU read side lock
171 * MUST be acquired before calling this.
97ee3a89 172 */
18eace3b 173struct ltt_ust_event *trace_ust_find_event(struct lttng_ht *ht,
10646003
JI
174 char *name, struct lttng_filter_bytecode *filter, int loglevel,
175 struct lttng_event_exclusion *exclusion)
97ee3a89 176{
bec39940
DG
177 struct lttng_ht_node_str *node;
178 struct lttng_ht_iter iter;
18eace3b 179 struct ltt_ust_ht_key key;
97ee3a89 180
18eace3b
DG
181 assert(name);
182 assert(ht);
183
184 key.name = name;
185 key.filter = filter;
186 key.loglevel = loglevel;
10646003 187 key.exclusion = exclusion;
18eace3b 188
18eace3b
DG
189 cds_lfht_lookup(ht->ht, ht->hash_fct((void *) name, lttng_ht_seed),
190 trace_ust_ht_match_event, &key, &iter.iter);
bec39940 191 node = lttng_ht_iter_get_node_str(&iter);
f6a9efaa 192 if (node == NULL) {
97ee3a89
DG
193 goto error;
194 }
195
18eace3b 196 DBG2("Trace UST event %s found", key.name);
f6a9efaa
DG
197
198 return caa_container_of(node, struct ltt_ust_event, node);
97ee3a89
DG
199
200error:
18eace3b 201 DBG2("Trace UST event %s NOT found", key.name);
97ee3a89
DG
202 return NULL;
203}
204
fefd409b
DG
205/*
206 * Lookup an agent in the session agents hash table by domain type and return
207 * the object if found else NULL.
4da703ad
JG
208 *
209 * RCU read side lock must be acquired before calling and only released
210 * once the agent is no longer in scope or being used.
fefd409b
DG
211 */
212struct agent *trace_ust_find_agent(struct ltt_ust_session *session,
213 enum lttng_domain_type domain_type)
214{
215 struct agent *agt = NULL;
216 struct lttng_ht_node_u64 *node;
217 struct lttng_ht_iter iter;
218 uint64_t key;
219
220 assert(session);
221
222 DBG3("Trace ust agent lookup for domain %d", domain_type);
223
224 key = domain_type;
225
226 lttng_ht_lookup(session->agents, &key, &iter);
227 node = lttng_ht_iter_get_node_u64(&iter);
228 if (!node) {
229 goto end;
230 }
231 agt = caa_container_of(node, struct agent, node);
232
233end:
234 return agt;
235}
236
97ee3a89
DG
237/*
238 * Allocate and initialize a ust session data structure.
239 *
240 * Return pointer to structure or NULL.
241 */
d9bf3ca4 242struct ltt_ust_session *trace_ust_create_session(uint64_t session_id)
97ee3a89
DG
243{
244 struct ltt_ust_session *lus;
245
246 /* Allocate a new ltt ust session */
ba7f0ae5 247 lus = zmalloc(sizeof(struct ltt_ust_session));
97ee3a89 248 if (lus == NULL) {
ba7f0ae5 249 PERROR("create ust session zmalloc");
97ee3a89
DG
250 goto error;
251 }
252
253 /* Init data structure */
a991f516 254 lus->id = session_id;
14fb1ebe 255 lus->active = 0;
97ee3a89 256
84ad93e8
DG
257 /* Set default metadata channel attribute. */
258 lus->metadata_attr.overwrite = DEFAULT_CHANNEL_OVERWRITE;
259 lus->metadata_attr.subbuf_size = default_get_metadata_subbuf_size();
260 lus->metadata_attr.num_subbuf = DEFAULT_METADATA_SUBBUF_NUM;
261 lus->metadata_attr.switch_timer_interval = DEFAULT_METADATA_SWITCH_TIMER;
262 lus->metadata_attr.read_timer_interval = DEFAULT_METADATA_READ_TIMER;
263 lus->metadata_attr.output = LTTNG_UST_MMAP;
264
7972aab2
DG
265 /*
266 * Default buffer type. This can be changed through an enable channel
267 * requesting a different type. Note that this can only be changed once
268 * during the session lifetime which is at the first enable channel and
269 * only before start. The flag buffer_type_changed indicates the status.
270 */
8692d4e5 271 lus->buffer_type = LTTNG_BUFFER_PER_UID;
7972aab2
DG
272 /* Once set to 1, the buffer_type is immutable for the session. */
273 lus->buffer_type_changed = 0;
274 /* Init it in case it get used after allocation. */
275 CDS_INIT_LIST_HEAD(&lus->buffer_reg_uid_list);
f6a9efaa
DG
276
277 /* Alloc UST global domain channels' HT */
bec39940 278 lus->domain_global.channels = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
fefd409b
DG
279 /* Alloc agent hash table. */
280 lus->agents = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
44d3bd01 281
00e2e675
DG
282 lus->consumer = consumer_create_output(CONSUMER_DST_LOCAL);
283 if (lus->consumer == NULL) {
09a90bcd 284 goto error_consumer;
00e2e675
DG
285 }
286
44d3bd01
DG
287 DBG2("UST trace session create successful");
288
97ee3a89
DG
289 return lus;
290
09a90bcd 291error_consumer:
0b2dc8df 292 ht_cleanup_push(lus->domain_global.channels);
fefd409b 293 ht_cleanup_push(lus->agents);
44844c29 294 free(lus);
97ee3a89
DG
295error:
296 return NULL;
297}
298
299/*
300 * Allocate and initialize a ust channel data structure.
301 *
302 * Return pointer to structure or NULL.
303 */
51755dc8
JG
304struct ltt_ust_channel *trace_ust_create_channel(struct lttng_channel *chan,
305 enum lttng_domain_type domain)
97ee3a89 306{
97ee3a89
DG
307 struct ltt_ust_channel *luc;
308
0525e9ae 309 assert(chan);
0525e9ae 310
37357452 311 luc = zmalloc(sizeof(struct ltt_ust_channel));
97ee3a89 312 if (luc == NULL) {
df0f840b 313 PERROR("ltt_ust_channel zmalloc");
97ee3a89
DG
314 goto error;
315 }
316
51755dc8
JG
317 luc->domain = domain;
318
44d3bd01 319 /* Copy UST channel attributes */
48842b30
DG
320 luc->attr.overwrite = chan->attr.overwrite;
321 luc->attr.subbuf_size = chan->attr.subbuf_size;
322 luc->attr.num_subbuf = chan->attr.num_subbuf;
323 luc->attr.switch_timer_interval = chan->attr.switch_timer_interval;
324 luc->attr.read_timer_interval = chan->attr.read_timer_interval;
6775595e 325 luc->attr.output = (enum lttng_ust_output) chan->attr.output;
44d3bd01
DG
326
327 /* Translate to UST output enum */
328 switch (luc->attr.output) {
329 default:
330 luc->attr.output = LTTNG_UST_MMAP;
331 break;
97ee3a89 332 }
97ee3a89 333
85076754
MD
334 /*
335 * If we receive an empty string for channel name, it means the
336 * default channel name is requested.
337 */
338 if (chan->name[0] == '\0') {
339 strncpy(luc->name, DEFAULT_CHANNEL_NAME, sizeof(luc->name));
340 } else {
341 /* Copy channel name */
342 strncpy(luc->name, chan->name, sizeof(luc->name));
343 }
97ee3a89
DG
344 luc->name[LTTNG_UST_SYM_NAME_LEN - 1] = '\0';
345
f6a9efaa 346 /* Init node */
bec39940 347 lttng_ht_node_init_str(&luc->node, luc->name);
31746f93
DG
348 CDS_INIT_LIST_HEAD(&luc->ctx_list);
349
f6a9efaa 350 /* Alloc hash tables */
bec39940
DG
351 luc->events = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
352 luc->ctx = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
f6a9efaa 353
1624d5b7
JD
354 /* On-disk circular buffer parameters */
355 luc->tracefile_size = chan->attr.tracefile_size;
356 luc->tracefile_count = chan->attr.tracefile_count;
357
f6a9efaa
DG
358 DBG2("Trace UST channel %s created", luc->name);
359
97ee3a89 360error:
7972aab2 361 return luc;
97ee3a89
DG
362}
363
364/*
365 * Allocate and initialize a ust event. Set name and event type.
49d21f93 366 * We own filter_expression, filter, and exclusion.
97ee3a89
DG
367 *
368 * Return pointer to structure or NULL.
369 */
025faf73 370struct ltt_ust_event *trace_ust_create_event(struct lttng_event *ev,
6b453b5e 371 char *filter_expression,
561c6897 372 struct lttng_filter_bytecode *filter,
88f06f15
JG
373 struct lttng_event_exclusion *exclusion,
374 bool internal_event)
97ee3a89
DG
375{
376 struct ltt_ust_event *lue;
97ee3a89 377
0525e9ae
DG
378 assert(ev);
379
37357452 380 lue = zmalloc(sizeof(struct ltt_ust_event));
44d3bd01 381 if (lue == NULL) {
ba7f0ae5 382 PERROR("ust event zmalloc");
97ee3a89
DG
383 goto error;
384 }
385
88f06f15
JG
386 lue->internal = internal_event;
387
97ee3a89
DG
388 switch (ev->type) {
389 case LTTNG_EVENT_PROBE:
44d3bd01 390 lue->attr.instrumentation = LTTNG_UST_PROBE;
97ee3a89
DG
391 break;
392 case LTTNG_EVENT_FUNCTION:
44d3bd01 393 lue->attr.instrumentation = LTTNG_UST_FUNCTION;
97ee3a89
DG
394 break;
395 case LTTNG_EVENT_FUNCTION_ENTRY:
44d3bd01 396 lue->attr.instrumentation = LTTNG_UST_FUNCTION;
97ee3a89
DG
397 break;
398 case LTTNG_EVENT_TRACEPOINT:
44d3bd01 399 lue->attr.instrumentation = LTTNG_UST_TRACEPOINT;
97ee3a89
DG
400 break;
401 default:
402 ERR("Unknown ust instrumentation type (%d)", ev->type);
44844c29 403 goto error_free_event;
97ee3a89
DG
404 }
405
406 /* Copy event name */
44d3bd01
DG
407 strncpy(lue->attr.name, ev->name, LTTNG_UST_SYM_NAME_LEN);
408 lue->attr.name[LTTNG_UST_SYM_NAME_LEN - 1] = '\0';
97ee3a89 409
0cda4f28 410 switch (ev->loglevel_type) {
8005f29a
MD
411 case LTTNG_EVENT_LOGLEVEL_ALL:
412 lue->attr.loglevel_type = LTTNG_UST_LOGLEVEL_ALL;
22e25b71 413 lue->attr.loglevel = -1; /* Force to -1 */
0cda4f28 414 break;
8005f29a
MD
415 case LTTNG_EVENT_LOGLEVEL_RANGE:
416 lue->attr.loglevel_type = LTTNG_UST_LOGLEVEL_RANGE;
22e25b71 417 lue->attr.loglevel = ev->loglevel;
8005f29a
MD
418 break;
419 case LTTNG_EVENT_LOGLEVEL_SINGLE:
420 lue->attr.loglevel_type = LTTNG_UST_LOGLEVEL_SINGLE;
22e25b71 421 lue->attr.loglevel = ev->loglevel;
0cda4f28
MD
422 break;
423 default:
4431494b 424 ERR("Unknown ust loglevel type (%d)", ev->loglevel_type);
0cda4f28
MD
425 goto error_free_event;
426 }
427
025faf73 428 /* Same layout. */
6b453b5e 429 lue->filter_expression = filter_expression;
51755dc8
JG
430 lue->filter = filter;
431 lue->exclusion = exclusion;
0cda4f28 432
f6a9efaa 433 /* Init node */
bec39940 434 lttng_ht_node_init_str(&lue->node, lue->attr.name);
97ee3a89 435
300b8fd5
MD
436 DBG2("Trace UST event %s, loglevel (%d,%d) created",
437 lue->attr.name, lue->attr.loglevel_type,
438 lue->attr.loglevel);
284d8f55 439
97ee3a89
DG
440 return lue;
441
44844c29
MD
442error_free_event:
443 free(lue);
97ee3a89 444error:
49d21f93
MD
445 free(filter_expression);
446 free(filter);
447 free(exclusion);
97ee3a89
DG
448 return NULL;
449}
450
aa3514e9
MD
451static
452int trace_ust_context_type_event_to_ust(enum lttng_event_context_type type)
55cc08a6 453{
aa3514e9 454 int utype;
0525e9ae 455
aa3514e9 456 switch (type) {
9197c5c4
MD
457 case LTTNG_EVENT_CONTEXT_VTID:
458 utype = LTTNG_UST_CONTEXT_VTID;
459 break;
460 case LTTNG_EVENT_CONTEXT_VPID:
461 utype = LTTNG_UST_CONTEXT_VPID;
462 break;
463 case LTTNG_EVENT_CONTEXT_PTHREAD_ID:
464 utype = LTTNG_UST_CONTEXT_PTHREAD_ID;
465 break;
466 case LTTNG_EVENT_CONTEXT_PROCNAME:
467 utype = LTTNG_UST_CONTEXT_PROCNAME;
468 break;
7c612c2e
WP
469 case LTTNG_EVENT_CONTEXT_IP:
470 utype = LTTNG_UST_CONTEXT_IP;
471 break;
aa3514e9 472 case LTTNG_EVENT_CONTEXT_PERF_THREAD_COUNTER:
354e561b
MD
473 if (!ustctl_has_perf_counters()) {
474 utype = -1;
475 WARN("Perf counters not implemented in UST");
476 } else {
477 utype = LTTNG_UST_CONTEXT_PERF_THREAD_COUNTER;
478 }
aa3514e9 479 break;
9197c5c4
MD
480 default:
481 ERR("Invalid UST context");
aa3514e9
MD
482 utype = -1;
483 break;
484 }
485 return utype;
486}
487
488/*
489 * Return 1 if contexts match, 0 otherwise.
490 */
491int trace_ust_match_context(struct ltt_ust_context *uctx,
492 struct lttng_event_context *ctx)
493{
494 int utype;
495
496 utype = trace_ust_context_type_event_to_ust(ctx->ctx);
497 if (utype < 0) {
498 return 0;
499 }
500 if (uctx->ctx.ctx != utype) {
501 return 0;
502 }
503 switch (utype) {
504 case LTTNG_UST_CONTEXT_PERF_THREAD_COUNTER:
505 if (uctx->ctx.u.perf_counter.type
506 != ctx->u.perf_counter.type) {
507 return 0;
508 }
509 if (uctx->ctx.u.perf_counter.config
510 != ctx->u.perf_counter.config) {
511 return 0;
512 }
513 if (strncmp(uctx->ctx.u.perf_counter.name,
514 ctx->u.perf_counter.name,
515 LTTNG_UST_SYM_NAME_LEN)) {
516 return 0;
517 }
518 break;
519 default:
520 break;
521
522 }
523 return 1;
524}
525
526/*
527 * Allocate and initialize an UST context.
528 *
529 * Return pointer to structure or NULL.
530 */
531struct ltt_ust_context *trace_ust_create_context(
532 struct lttng_event_context *ctx)
533{
534 struct ltt_ust_context *uctx;
535 int utype;
536
537 assert(ctx);
538
539 utype = trace_ust_context_type_event_to_ust(ctx->ctx);
540 if (utype < 0) {
9197c5c4
MD
541 return NULL;
542 }
55cc08a6
DG
543
544 uctx = zmalloc(sizeof(struct ltt_ust_context));
545 if (uctx == NULL) {
546 PERROR("zmalloc ltt_ust_context");
547 goto error;
548 }
549
aa3514e9
MD
550 uctx->ctx.ctx = (enum lttng_ust_context_type) utype;
551 switch (utype) {
552 case LTTNG_UST_CONTEXT_PERF_THREAD_COUNTER:
553 uctx->ctx.u.perf_counter.type = ctx->u.perf_counter.type;
554 uctx->ctx.u.perf_counter.config = ctx->u.perf_counter.config;
555 strncpy(uctx->ctx.u.perf_counter.name, ctx->u.perf_counter.name,
556 LTTNG_UST_SYM_NAME_LEN);
557 uctx->ctx.u.perf_counter.name[LTTNG_UST_SYM_NAME_LEN - 1] = '\0';
558 break;
559 default:
560 break;
561 }
bec39940 562 lttng_ht_node_init_ulong(&uctx->node, (unsigned long) uctx->ctx.ctx);
55cc08a6
DG
563
564 return uctx;
565
566error:
567 return NULL;
568}
569
a9ad0c8f
MD
570static
571void destroy_pid_tracker_node_rcu(struct rcu_head *head)
572{
573 struct ust_pid_tracker_node *tracker_node =
574 caa_container_of(head, struct ust_pid_tracker_node, node.head);
575 free(tracker_node);
576}
577
578static
579void destroy_pid_tracker_node(struct ust_pid_tracker_node *tracker_node)
580{
581
582 call_rcu(&tracker_node->node.head, destroy_pid_tracker_node_rcu);
583}
584
585static
586int init_pid_tracker(struct ust_pid_tracker *pid_tracker)
587{
588 int ret = 0;
589
590 pid_tracker->ht = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
591 if (!pid_tracker->ht) {
592 ret = -1;
593 goto end;
594 }
595
596end:
597 return ret;
598}
599
600/*
601 * Teardown pid tracker content, but don't free pid_tracker object.
602 */
603static
604void fini_pid_tracker(struct ust_pid_tracker *pid_tracker)
605{
606 struct ust_pid_tracker_node *tracker_node;
607 struct lttng_ht_iter iter;
608
609 if (!pid_tracker->ht) {
610 return;
611 }
612 rcu_read_lock();
613 cds_lfht_for_each_entry(pid_tracker->ht->ht,
614 &iter.iter, tracker_node, node.node) {
615 int ret = lttng_ht_del(pid_tracker->ht, &iter);
616
617 assert(!ret);
618 destroy_pid_tracker_node(tracker_node);
619 }
620 rcu_read_unlock();
621 ht_cleanup_push(pid_tracker->ht);
622 pid_tracker->ht = NULL;
623}
624
625static
626struct ust_pid_tracker_node *pid_tracker_lookup(
627 struct ust_pid_tracker *pid_tracker, int pid,
628 struct lttng_ht_iter *iter)
629{
630 unsigned long _pid = (unsigned long) pid;
631 struct lttng_ht_node_ulong *node;
632
633 lttng_ht_lookup(pid_tracker->ht, (void *) _pid, iter);
634 node = lttng_ht_iter_get_node_ulong(iter);
635 if (node) {
636 return caa_container_of(node, struct ust_pid_tracker_node,
637 node);
638 } else {
639 return NULL;
640 }
641}
642
643static
644int pid_tracker_add_pid(struct ust_pid_tracker *pid_tracker, int pid)
645{
646 int retval = LTTNG_OK;
647 struct ust_pid_tracker_node *tracker_node;
648 struct lttng_ht_iter iter;
649
650 if (pid < 0) {
651 retval = LTTNG_ERR_INVALID;
652 goto end;
653 }
654 tracker_node = pid_tracker_lookup(pid_tracker, pid, &iter);
655 if (tracker_node) {
656 /* Already exists. */
b93a4a13 657 retval = LTTNG_ERR_PID_TRACKED;
a9ad0c8f
MD
658 goto end;
659 }
660 tracker_node = zmalloc(sizeof(*tracker_node));
661 if (!tracker_node) {
662 retval = LTTNG_ERR_NOMEM;
663 goto end;
664 }
665 lttng_ht_node_init_ulong(&tracker_node->node, (unsigned long) pid);
666 lttng_ht_add_unique_ulong(pid_tracker->ht, &tracker_node->node);
667end:
668 return retval;
669}
670
671static
672int pid_tracker_del_pid(struct ust_pid_tracker *pid_tracker, int pid)
673{
674 int retval = LTTNG_OK, ret;
675 struct ust_pid_tracker_node *tracker_node;
676 struct lttng_ht_iter iter;
677
678 if (pid < 0) {
679 retval = LTTNG_ERR_INVALID;
680 goto end;
681 }
682 tracker_node = pid_tracker_lookup(pid_tracker, pid, &iter);
683 if (!tracker_node) {
684 /* Not found */
b93a4a13 685 retval = LTTNG_ERR_PID_NOT_TRACKED;
a9ad0c8f
MD
686 goto end;
687 }
688 ret = lttng_ht_del(pid_tracker->ht, &iter);
689 assert(!ret);
690
691 destroy_pid_tracker_node(tracker_node);
692end:
693 return retval;
694}
695
696/*
697 * The session lock is held when calling this function.
698 */
699int trace_ust_pid_tracker_lookup(struct ltt_ust_session *session, int pid)
700{
701 struct lttng_ht_iter iter;
702
703 if (!session->pid_tracker.ht) {
704 return 1;
705 }
706 if (pid_tracker_lookup(&session->pid_tracker, pid, &iter)) {
707 return 1;
708 }
709 return 0;
710}
711
712/*
713 * Called with the session lock held.
714 */
715int trace_ust_track_pid(struct ltt_ust_session *session, int pid)
716{
717 int retval = LTTNG_OK;
718
719 if (pid == -1) {
720 /* Track all pids: destroy tracker if exists. */
721 if (session->pid_tracker.ht) {
722 fini_pid_tracker(&session->pid_tracker);
723 /* Ensure all apps have session. */
724 ust_app_global_update_all(session);
725 }
726 } else {
727 int ret;
728
729 if (!session->pid_tracker.ht) {
730 /* Create tracker. */
731 if (init_pid_tracker(&session->pid_tracker)) {
732 ERR("Error initializing PID tracker");
733 retval = LTTNG_ERR_NOMEM;
734 goto end;
735 }
736 ret = pid_tracker_add_pid(&session->pid_tracker, pid);
737 if (ret != LTTNG_OK) {
738 retval = ret;
739 fini_pid_tracker(&session->pid_tracker);
740 goto end;
741 }
742 /* Remove all apps from session except pid. */
743 ust_app_global_update_all(session);
744 } else {
745 struct ust_app *app;
746
747 ret = pid_tracker_add_pid(&session->pid_tracker, pid);
748 if (ret != LTTNG_OK) {
749 retval = ret;
750 goto end;
751 }
752 /* Add session to application */
753 app = ust_app_find_by_pid(pid);
754 if (app) {
755 ust_app_global_update(session, app);
756 }
757 }
758 }
759end:
760 return retval;
761}
762
763/*
764 * Called with the session lock held.
765 */
766int trace_ust_untrack_pid(struct ltt_ust_session *session, int pid)
767{
768 int retval = LTTNG_OK;
769
770 if (pid == -1) {
771 /* Create empty tracker, replace old tracker. */
772 struct ust_pid_tracker tmp_tracker;
773
774 tmp_tracker = session->pid_tracker;
775 if (init_pid_tracker(&session->pid_tracker)) {
776 ERR("Error initializing PID tracker");
777 retval = LTTNG_ERR_NOMEM;
778 /* Rollback operation. */
779 session->pid_tracker = tmp_tracker;
780 goto end;
781 }
782 fini_pid_tracker(&tmp_tracker);
783
784 /* Remove session from all applications */
785 ust_app_global_update_all(session);
786 } else {
787 int ret;
788 struct ust_app *app;
789
790 if (!session->pid_tracker.ht) {
791 retval = LTTNG_ERR_INVALID;
792 goto end;
793 }
794 /* Remove PID from tracker */
795 ret = pid_tracker_del_pid(&session->pid_tracker, pid);
796 if (ret != LTTNG_OK) {
797 retval = ret;
798 goto end;
799 }
800 /* Remove session from application. */
801 app = ust_app_find_by_pid(pid);
802 if (app) {
803 ust_app_global_update(session, app);
804 }
805 }
806end:
807 return retval;
808}
809
a5dfbb9d
MD
810/*
811 * Called with session lock held.
812 */
813ssize_t trace_ust_list_tracker_pids(struct ltt_ust_session *session,
814 int32_t **_pids)
815{
816 struct ust_pid_tracker_node *tracker_node;
817 struct lttng_ht_iter iter;
818 unsigned long count, i = 0;
819 long approx[2];
820 int32_t *pids;
821 int ret = 0;
822
823 if (!session->pid_tracker.ht) {
824 /* Tracker disabled. Set first entry to -1. */
825 pids = zmalloc(sizeof(*pids));
826 if (!pids) {
827 ret = -1;
828 goto end;
829 }
830 pids[0] = -1;
831 *_pids = pids;
832 return 1;
833 }
834
835 rcu_read_lock();
836 cds_lfht_count_nodes(session->pid_tracker.ht->ht,
837 &approx[0], &count, &approx[1]);
838 pids = zmalloc(sizeof(*pids) * count);
839 if (!pids) {
840 ret = -1;
841 goto end;
842 }
843 cds_lfht_for_each_entry(session->pid_tracker.ht->ht,
844 &iter.iter, tracker_node, node.node) {
845 pids[i++] = tracker_node->node.key;
846 }
847 *_pids = pids;
848 ret = count;
849end:
850 rcu_read_unlock();
851 return ret;
852}
853
f6a9efaa
DG
854/*
855 * RCU safe free context structure.
856 */
857static void destroy_context_rcu(struct rcu_head *head)
858{
bec39940
DG
859 struct lttng_ht_node_ulong *node =
860 caa_container_of(head, struct lttng_ht_node_ulong, head);
f6a9efaa
DG
861 struct ltt_ust_context *ctx =
862 caa_container_of(node, struct ltt_ust_context, node);
863
864 free(ctx);
865}
866
867/*
868 * Cleanup UST context hash table.
869 */
7bd39047 870static void destroy_contexts(struct lttng_ht *ht)
f6a9efaa
DG
871{
872 int ret;
bec39940
DG
873 struct lttng_ht_node_ulong *node;
874 struct lttng_ht_iter iter;
31746f93 875 struct ltt_ust_context *ctx;
f6a9efaa 876
0525e9ae
DG
877 assert(ht);
878
36b588ed 879 rcu_read_lock();
bec39940 880 cds_lfht_for_each_entry(ht->ht, &iter.iter, node, node) {
31746f93
DG
881 /* Remove from ordered list. */
882 ctx = caa_container_of(node, struct ltt_ust_context, node);
883 cds_list_del(&ctx->list);
884 /* Remove from channel's hash table. */
bec39940 885 ret = lttng_ht_del(ht, &iter);
f6a9efaa
DG
886 if (!ret) {
887 call_rcu(&node->head, destroy_context_rcu);
888 }
f6a9efaa 889 }
36b588ed 890 rcu_read_unlock();
ed52805d 891
0b2dc8df 892 ht_cleanup_push(ht);
f6a9efaa
DG
893}
894
97ee3a89
DG
895/*
896 * Cleanup ust event structure.
897 */
898void trace_ust_destroy_event(struct ltt_ust_event *event)
899{
0525e9ae
DG
900 assert(event);
901
f6a9efaa 902 DBG2("Trace destroy UST event %s", event->attr.name);
6b453b5e 903 free(event->filter_expression);
53a80697 904 free(event->filter);
40024f8a 905 free(event->exclusion);
97ee3a89
DG
906 free(event);
907}
908
f6a9efaa
DG
909/*
910 * URCU intermediate call to complete destroy event.
911 */
912static void destroy_event_rcu(struct rcu_head *head)
913{
bec39940
DG
914 struct lttng_ht_node_str *node =
915 caa_container_of(head, struct lttng_ht_node_str, head);
f6a9efaa
DG
916 struct ltt_ust_event *event =
917 caa_container_of(node, struct ltt_ust_event, node);
918
919 trace_ust_destroy_event(event);
920}
921
284d8f55
DG
922/*
923 * Cleanup UST events hashtable.
924 */
7bd39047 925static void destroy_events(struct lttng_ht *events)
ed52805d
DG
926{
927 int ret;
bec39940
DG
928 struct lttng_ht_node_str *node;
929 struct lttng_ht_iter iter;
ed52805d 930
0525e9ae
DG
931 assert(events);
932
36b588ed 933 rcu_read_lock();
bec39940
DG
934 cds_lfht_for_each_entry(events->ht, &iter.iter, node, node) {
935 ret = lttng_ht_del(events, &iter);
7bd39047
DG
936 assert(!ret);
937 call_rcu(&node->head, destroy_event_rcu);
ed52805d 938 }
36b588ed 939 rcu_read_unlock();
ed52805d 940
0b2dc8df 941 ht_cleanup_push(events);
ed52805d
DG
942}
943
97ee3a89
DG
944/*
945 * Cleanup ust channel structure.
36b588ed
MD
946 *
947 * Should _NOT_ be called with RCU read lock held.
97ee3a89 948 */
36b588ed 949static void _trace_ust_destroy_channel(struct ltt_ust_channel *channel)
97ee3a89 950{
0525e9ae
DG
951 assert(channel);
952
f6a9efaa 953 DBG2("Trace destroy UST channel %s", channel->name);
97ee3a89 954
97ee3a89 955 free(channel);
f6a9efaa
DG
956}
957
958/*
959 * URCU intermediate call to complete destroy channel.
960 */
961static void destroy_channel_rcu(struct rcu_head *head)
962{
bec39940
DG
963 struct lttng_ht_node_str *node =
964 caa_container_of(head, struct lttng_ht_node_str, head);
f6a9efaa
DG
965 struct ltt_ust_channel *channel =
966 caa_container_of(node, struct ltt_ust_channel, node);
967
36b588ed
MD
968 _trace_ust_destroy_channel(channel);
969}
970
971void trace_ust_destroy_channel(struct ltt_ust_channel *channel)
972{
627cbbe7
MD
973 /* Destroying all events of the channel */
974 destroy_events(channel->events);
975 /* Destroying all context of the channel */
976 destroy_contexts(channel->ctx);
977
36b588ed 978 call_rcu(&channel->node.head, destroy_channel_rcu);
97ee3a89
DG
979}
980
d5979e4a
DG
981/*
982 * Remove an UST channel from a channel HT.
983 */
984void trace_ust_delete_channel(struct lttng_ht *ht,
985 struct ltt_ust_channel *channel)
986{
987 int ret;
988 struct lttng_ht_iter iter;
989
990 assert(ht);
991 assert(channel);
992
993 iter.iter.node = &channel->node.node;
994 ret = lttng_ht_del(ht, &iter);
995 assert(!ret);
996}
997
97ee3a89 998/*
f6a9efaa 999 * Iterate over a hash table containing channels and cleanup safely.
97ee3a89 1000 */
bec39940 1001static void destroy_channels(struct lttng_ht *channels)
f6a9efaa 1002{
bec39940
DG
1003 struct lttng_ht_node_str *node;
1004 struct lttng_ht_iter iter;
f6a9efaa 1005
0525e9ae
DG
1006 assert(channels);
1007
24d1723f 1008 rcu_read_lock();
bec39940 1009 cds_lfht_for_each_entry(channels->ht, &iter.iter, node, node) {
627cbbe7
MD
1010 struct ltt_ust_channel *chan =
1011 caa_container_of(node, struct ltt_ust_channel, node);
1012
1013 trace_ust_delete_channel(channels, chan);
1014 trace_ust_destroy_channel(chan);
f6a9efaa 1015 }
36b588ed 1016 rcu_read_unlock();
ed52805d 1017
0b2dc8df 1018 ht_cleanup_push(channels);
f6a9efaa
DG
1019}
1020
f6a9efaa
DG
1021/*
1022 * Cleanup UST global domain.
1023 */
1024static void destroy_domain_global(struct ltt_ust_domain_global *dom)
1025{
0525e9ae
DG
1026 assert(dom);
1027
f6a9efaa
DG
1028 destroy_channels(dom->channels);
1029}
97ee3a89 1030
f6a9efaa
DG
1031/*
1032 * Cleanup ust session structure
36b588ed
MD
1033 *
1034 * Should *NOT* be called with RCU read-side lock held.
f6a9efaa
DG
1035 */
1036void trace_ust_destroy_session(struct ltt_ust_session *session)
1037{
fefd409b 1038 struct agent *agt;
7972aab2 1039 struct buffer_reg_uid *reg, *sreg;
fefd409b 1040 struct lttng_ht_iter iter;
7972aab2 1041
0525e9ae 1042 assert(session);
97ee3a89 1043
d9bf3ca4 1044 DBG2("Trace UST destroy session %" PRIu64, session->id);
f6a9efaa 1045
f6a9efaa
DG
1046 /* Cleaning up UST domain */
1047 destroy_domain_global(&session->domain_global);
fefd409b 1048
0c0fcd77 1049 rcu_read_lock();
fefd409b 1050 cds_lfht_for_each_entry(session->agents->ht, &iter.iter, agt, node.node) {
d0edf546
JG
1051 int ret = lttng_ht_del(session->agents, &iter);
1052
1053 assert(!ret);
fefd409b
DG
1054 agent_destroy(agt);
1055 }
0c0fcd77 1056 rcu_read_unlock();
7972aab2 1057
8ada2175
MD
1058 ht_cleanup_push(session->agents);
1059
7972aab2
DG
1060 /* Cleanup UID buffer registry object(s). */
1061 cds_list_for_each_entry_safe(reg, sreg, &session->buffer_reg_uid_list,
1062 lnode) {
1063 cds_list_del(&reg->lnode);
1064 buffer_reg_uid_remove(reg);
1065 buffer_reg_uid_destroy(reg, session->consumer);
1066 }
44d3bd01 1067
6addfa37 1068 consumer_output_put(session->consumer);
3f8e211f 1069
a9ad0c8f
MD
1070 fini_pid_tracker(&session->pid_tracker);
1071
97ee3a89
DG
1072 free(session);
1073}
This page took 0.153929 seconds and 4 git commands to generate.