Fix: channel errors on local stress-tests
[lttng-tools.git] / src / bin / lttng-sessiond / ust-app.c
CommitLineData
91d76f53
DG
1/*
2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
bdf64013 3 * Copyright (C) 2016 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
91d76f53 4 *
d14d33bf
AM
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License, version 2 only,
7 * as published by the Free Software Foundation.
91d76f53
DG
8 *
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
d14d33bf
AM
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
91d76f53
DG
17 */
18
6c1c0768 19#define _LGPL_SOURCE
91d76f53 20#include <errno.h>
7972aab2 21#include <inttypes.h>
91d76f53
DG
22#include <pthread.h>
23#include <stdio.h>
24#include <stdlib.h>
099e26bd 25#include <string.h>
aba8e916
DG
26#include <sys/stat.h>
27#include <sys/types.h>
099e26bd 28#include <unistd.h>
0df502fd 29#include <urcu/compiler.h>
fb54cdbf 30#include <lttng/ust-error.h>
331744e3 31#include <signal.h>
bec39940 32
990570ed 33#include <common/common.h>
86acf0da 34#include <common/sessiond-comm/sessiond-comm.h>
1e307fab 35
7972aab2 36#include "buffer-registry.h"
86acf0da 37#include "fd-limit.h"
8782cc74 38#include "health-sessiond.h"
56fff090 39#include "ust-app.h"
48842b30 40#include "ust-consumer.h"
d80a6244 41#include "ust-ctl.h"
0b2dc8df 42#include "utils.h"
fb83fe64 43#include "session.h"
e9404c27
JG
44#include "lttng-sessiond.h"
45#include "notification-thread-commands.h"
d80a6244 46
c4b88406
MD
47static
48int ust_app_flush_app_session(struct ust_app *app, struct ust_app_session *ua_sess);
49
d9bf3ca4
MD
50/* Next available channel key. Access under next_channel_key_lock. */
51static uint64_t _next_channel_key;
52static pthread_mutex_t next_channel_key_lock = PTHREAD_MUTEX_INITIALIZER;
53
54/* Next available session ID. Access under next_session_id_lock. */
55static uint64_t _next_session_id;
56static pthread_mutex_t next_session_id_lock = PTHREAD_MUTEX_INITIALIZER;
ffe60014
DG
57
58/*
d9bf3ca4 59 * Return the incremented value of next_channel_key.
ffe60014 60 */
d9bf3ca4 61static uint64_t get_next_channel_key(void)
ffe60014 62{
d9bf3ca4
MD
63 uint64_t ret;
64
65 pthread_mutex_lock(&next_channel_key_lock);
66 ret = ++_next_channel_key;
67 pthread_mutex_unlock(&next_channel_key_lock);
68 return ret;
ffe60014
DG
69}
70
71/*
7972aab2 72 * Return the atomically incremented value of next_session_id.
ffe60014 73 */
d9bf3ca4 74static uint64_t get_next_session_id(void)
ffe60014 75{
d9bf3ca4
MD
76 uint64_t ret;
77
78 pthread_mutex_lock(&next_session_id_lock);
79 ret = ++_next_session_id;
80 pthread_mutex_unlock(&next_session_id_lock);
81 return ret;
ffe60014
DG
82}
83
d65d2de8
DG
84static void copy_channel_attr_to_ustctl(
85 struct ustctl_consumer_channel_attr *attr,
86 struct lttng_ust_channel_attr *uattr)
87{
88 /* Copy event attributes since the layout is different. */
89 attr->subbuf_size = uattr->subbuf_size;
90 attr->num_subbuf = uattr->num_subbuf;
91 attr->overwrite = uattr->overwrite;
92 attr->switch_timer_interval = uattr->switch_timer_interval;
93 attr->read_timer_interval = uattr->read_timer_interval;
94 attr->output = uattr->output;
f515d318 95 attr->blocking_timeout = uattr->u.s.blocking_timeout;
d65d2de8
DG
96}
97
025faf73
DG
98/*
99 * Match function for the hash table lookup.
100 *
101 * It matches an ust app event based on three attributes which are the event
102 * name, the filter bytecode and the loglevel.
103 */
18eace3b
DG
104static int ht_match_ust_app_event(struct cds_lfht_node *node, const void *_key)
105{
106 struct ust_app_event *event;
107 const struct ust_app_ht_key *key;
2106efa0 108 int ev_loglevel_value;
18eace3b
DG
109
110 assert(node);
111 assert(_key);
112
113 event = caa_container_of(node, struct ust_app_event, node.node);
114 key = _key;
2106efa0 115 ev_loglevel_value = event->attr.loglevel;
18eace3b 116
1af53eb5 117 /* Match the 4 elements of the key: name, filter, loglevel, exclusions */
18eace3b
DG
118
119 /* Event name */
120 if (strncmp(event->attr.name, key->name, sizeof(event->attr.name)) != 0) {
121 goto no_match;
122 }
123
124 /* Event loglevel. */
2106efa0 125 if (ev_loglevel_value != key->loglevel_type) {
025faf73 126 if (event->attr.loglevel_type == LTTNG_UST_LOGLEVEL_ALL
2106efa0
PP
127 && key->loglevel_type == 0 &&
128 ev_loglevel_value == -1) {
025faf73
DG
129 /*
130 * Match is accepted. This is because on event creation, the
131 * loglevel is set to -1 if the event loglevel type is ALL so 0 and
132 * -1 are accepted for this loglevel type since 0 is the one set by
133 * the API when receiving an enable event.
134 */
135 } else {
136 goto no_match;
137 }
18eace3b
DG
138 }
139
140 /* One of the filters is NULL, fail. */
141 if ((key->filter && !event->filter) || (!key->filter && event->filter)) {
142 goto no_match;
143 }
144
025faf73
DG
145 if (key->filter && event->filter) {
146 /* Both filters exists, check length followed by the bytecode. */
147 if (event->filter->len != key->filter->len ||
148 memcmp(event->filter->data, key->filter->data,
149 event->filter->len) != 0) {
150 goto no_match;
151 }
18eace3b
DG
152 }
153
1af53eb5
JI
154 /* One of the exclusions is NULL, fail. */
155 if ((key->exclusion && !event->exclusion) || (!key->exclusion && event->exclusion)) {
156 goto no_match;
157 }
158
159 if (key->exclusion && event->exclusion) {
160 /* Both exclusions exists, check count followed by the names. */
161 if (event->exclusion->count != key->exclusion->count ||
162 memcmp(event->exclusion->names, key->exclusion->names,
163 event->exclusion->count * LTTNG_UST_SYM_NAME_LEN) != 0) {
164 goto no_match;
165 }
166 }
167
168
025faf73 169 /* Match. */
18eace3b
DG
170 return 1;
171
172no_match:
173 return 0;
18eace3b
DG
174}
175
025faf73
DG
176/*
177 * Unique add of an ust app event in the given ht. This uses the custom
178 * ht_match_ust_app_event match function and the event name as hash.
179 */
d0b96690 180static void add_unique_ust_app_event(struct ust_app_channel *ua_chan,
18eace3b
DG
181 struct ust_app_event *event)
182{
183 struct cds_lfht_node *node_ptr;
184 struct ust_app_ht_key key;
d0b96690 185 struct lttng_ht *ht;
18eace3b 186
d0b96690
DG
187 assert(ua_chan);
188 assert(ua_chan->events);
18eace3b
DG
189 assert(event);
190
d0b96690 191 ht = ua_chan->events;
18eace3b
DG
192 key.name = event->attr.name;
193 key.filter = event->filter;
2106efa0 194 key.loglevel_type = event->attr.loglevel;
91c89f23 195 key.exclusion = event->exclusion;
18eace3b
DG
196
197 node_ptr = cds_lfht_add_unique(ht->ht,
198 ht->hash_fct(event->node.key, lttng_ht_seed),
199 ht_match_ust_app_event, &key, &event->node.node);
200 assert(node_ptr == &event->node.node);
201}
202
d88aee68
DG
203/*
204 * Close the notify socket from the given RCU head object. This MUST be called
205 * through a call_rcu().
206 */
207static void close_notify_sock_rcu(struct rcu_head *head)
208{
209 int ret;
210 struct ust_app_notify_sock_obj *obj =
211 caa_container_of(head, struct ust_app_notify_sock_obj, head);
212
213 /* Must have a valid fd here. */
214 assert(obj->fd >= 0);
215
216 ret = close(obj->fd);
217 if (ret) {
218 ERR("close notify sock %d RCU", obj->fd);
219 }
220 lttng_fd_put(LTTNG_FD_APPS, 1);
221
222 free(obj);
223}
224
7972aab2
DG
225/*
226 * Return the session registry according to the buffer type of the given
227 * session.
228 *
229 * A registry per UID object MUST exists before calling this function or else
230 * it assert() if not found. RCU read side lock must be acquired.
231 */
232static struct ust_registry_session *get_session_registry(
233 struct ust_app_session *ua_sess)
234{
235 struct ust_registry_session *registry = NULL;
236
237 assert(ua_sess);
238
239 switch (ua_sess->buffer_type) {
240 case LTTNG_BUFFER_PER_PID:
241 {
242 struct buffer_reg_pid *reg_pid = buffer_reg_pid_find(ua_sess->id);
243 if (!reg_pid) {
244 goto error;
245 }
246 registry = reg_pid->registry->reg.ust;
247 break;
248 }
249 case LTTNG_BUFFER_PER_UID:
250 {
251 struct buffer_reg_uid *reg_uid = buffer_reg_uid_find(
252 ua_sess->tracing_id, ua_sess->bits_per_long, ua_sess->uid);
253 if (!reg_uid) {
254 goto error;
255 }
256 registry = reg_uid->registry->reg.ust;
257 break;
258 }
259 default:
260 assert(0);
261 };
262
263error:
264 return registry;
265}
266
55cc08a6
DG
267/*
268 * Delete ust context safely. RCU read lock must be held before calling
269 * this function.
270 */
271static
fb45065e
MD
272void delete_ust_app_ctx(int sock, struct ust_app_ctx *ua_ctx,
273 struct ust_app *app)
55cc08a6 274{
ffe60014
DG
275 int ret;
276
277 assert(ua_ctx);
278
55cc08a6 279 if (ua_ctx->obj) {
fb45065e 280 pthread_mutex_lock(&app->sock_lock);
ffe60014 281 ret = ustctl_release_object(sock, ua_ctx->obj);
fb45065e 282 pthread_mutex_unlock(&app->sock_lock);
d0b96690
DG
283 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
284 ERR("UST app sock %d release ctx obj handle %d failed with ret %d",
285 sock, ua_ctx->obj->handle, ret);
ffe60014 286 }
55cc08a6
DG
287 free(ua_ctx->obj);
288 }
289 free(ua_ctx);
290}
291
d80a6244
DG
292/*
293 * Delete ust app event safely. RCU read lock must be held before calling
294 * this function.
295 */
8b366481 296static
fb45065e
MD
297void delete_ust_app_event(int sock, struct ust_app_event *ua_event,
298 struct ust_app *app)
d80a6244 299{
ffe60014
DG
300 int ret;
301
302 assert(ua_event);
303
53a80697 304 free(ua_event->filter);
951f0b71
JI
305 if (ua_event->exclusion != NULL)
306 free(ua_event->exclusion);
edb67388 307 if (ua_event->obj != NULL) {
fb45065e 308 pthread_mutex_lock(&app->sock_lock);
ffe60014 309 ret = ustctl_release_object(sock, ua_event->obj);
fb45065e 310 pthread_mutex_unlock(&app->sock_lock);
ffe60014
DG
311 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
312 ERR("UST app sock %d release event obj failed with ret %d",
313 sock, ret);
314 }
edb67388
DG
315 free(ua_event->obj);
316 }
d80a6244
DG
317 free(ua_event);
318}
319
320/*
7972aab2
DG
321 * Release ust data object of the given stream.
322 *
323 * Return 0 on success or else a negative value.
d80a6244 324 */
fb45065e
MD
325static int release_ust_app_stream(int sock, struct ust_app_stream *stream,
326 struct ust_app *app)
d80a6244 327{
7972aab2 328 int ret = 0;
ffe60014
DG
329
330 assert(stream);
331
8b366481 332 if (stream->obj) {
fb45065e 333 pthread_mutex_lock(&app->sock_lock);
ffe60014 334 ret = ustctl_release_object(sock, stream->obj);
fb45065e 335 pthread_mutex_unlock(&app->sock_lock);
ffe60014
DG
336 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
337 ERR("UST app sock %d release stream obj failed with ret %d",
338 sock, ret);
339 }
4063050c 340 lttng_fd_put(LTTNG_FD_APPS, 2);
8b366481
DG
341 free(stream->obj);
342 }
7972aab2
DG
343
344 return ret;
345}
346
347/*
348 * Delete ust app stream safely. RCU read lock must be held before calling
349 * this function.
350 */
351static
fb45065e
MD
352void delete_ust_app_stream(int sock, struct ust_app_stream *stream,
353 struct ust_app *app)
7972aab2
DG
354{
355 assert(stream);
356
fb45065e 357 (void) release_ust_app_stream(sock, stream, app);
84cd17c6 358 free(stream);
d80a6244
DG
359}
360
36b588ed
MD
361/*
362 * We need to execute ht_destroy outside of RCU read-side critical
0b2dc8df
MD
363 * section and outside of call_rcu thread, so we postpone its execution
364 * using ht_cleanup_push. It is simpler than to change the semantic of
365 * the many callers of delete_ust_app_session().
36b588ed
MD
366 */
367static
368void delete_ust_app_channel_rcu(struct rcu_head *head)
369{
370 struct ust_app_channel *ua_chan =
371 caa_container_of(head, struct ust_app_channel, rcu_head);
372
0b2dc8df
MD
373 ht_cleanup_push(ua_chan->ctx);
374 ht_cleanup_push(ua_chan->events);
36b588ed
MD
375 free(ua_chan);
376}
377
fb83fe64
JD
378/*
379 * Extract the lost packet or discarded events counter when the channel is
380 * being deleted and store the value in the parent channel so we can
381 * access it from lttng list and at stop/destroy.
82cac6d2
JG
382 *
383 * The session list lock must be held by the caller.
fb83fe64
JD
384 */
385static
386void save_per_pid_lost_discarded_counters(struct ust_app_channel *ua_chan)
387{
388 uint64_t discarded = 0, lost = 0;
389 struct ltt_session *session;
390 struct ltt_ust_channel *uchan;
391
392 if (ua_chan->attr.type != LTTNG_UST_CHAN_PER_CPU) {
393 return;
394 }
395
396 rcu_read_lock();
397 session = session_find_by_id(ua_chan->session->tracing_id);
d68ec974
JG
398 if (!session || !session->ust_session) {
399 /*
400 * Not finding the session is not an error because there are
401 * multiple ways the channels can be torn down.
402 *
403 * 1) The session daemon can initiate the destruction of the
404 * ust app session after receiving a destroy command or
405 * during its shutdown/teardown.
406 * 2) The application, since we are in per-pid tracing, is
407 * unregistering and tearing down its ust app session.
408 *
409 * Both paths are protected by the session list lock which
410 * ensures that the accounting of lost packets and discarded
411 * events is done exactly once. The session is then unpublished
412 * from the session list, resulting in this condition.
413 */
fb83fe64
JD
414 goto end;
415 }
416
417 if (ua_chan->attr.overwrite) {
418 consumer_get_lost_packets(ua_chan->session->tracing_id,
419 ua_chan->key, session->ust_session->consumer,
420 &lost);
421 } else {
422 consumer_get_discarded_events(ua_chan->session->tracing_id,
423 ua_chan->key, session->ust_session->consumer,
424 &discarded);
425 }
426 uchan = trace_ust_find_channel_by_name(
427 session->ust_session->domain_global.channels,
428 ua_chan->name);
429 if (!uchan) {
430 ERR("Missing UST channel to store discarded counters");
431 goto end;
432 }
433
434 uchan->per_pid_closed_app_discarded += discarded;
435 uchan->per_pid_closed_app_lost += lost;
436
437end:
438 rcu_read_unlock();
439}
440
d80a6244
DG
441/*
442 * Delete ust app channel safely. RCU read lock must be held before calling
443 * this function.
82cac6d2
JG
444 *
445 * The session list lock must be held by the caller.
d80a6244 446 */
8b366481 447static
d0b96690
DG
448void delete_ust_app_channel(int sock, struct ust_app_channel *ua_chan,
449 struct ust_app *app)
d80a6244
DG
450{
451 int ret;
bec39940 452 struct lttng_ht_iter iter;
d80a6244 453 struct ust_app_event *ua_event;
55cc08a6 454 struct ust_app_ctx *ua_ctx;
030a66fa 455 struct ust_app_stream *stream, *stmp;
7972aab2 456 struct ust_registry_session *registry;
d80a6244 457
ffe60014
DG
458 assert(ua_chan);
459
460 DBG3("UST app deleting channel %s", ua_chan->name);
461
55cc08a6 462 /* Wipe stream */
d80a6244 463 cds_list_for_each_entry_safe(stream, stmp, &ua_chan->streams.head, list) {
84cd17c6 464 cds_list_del(&stream->list);
fb45065e 465 delete_ust_app_stream(sock, stream, app);
d80a6244
DG
466 }
467
55cc08a6 468 /* Wipe context */
bec39940 469 cds_lfht_for_each_entry(ua_chan->ctx->ht, &iter.iter, ua_ctx, node.node) {
31746f93 470 cds_list_del(&ua_ctx->list);
bec39940 471 ret = lttng_ht_del(ua_chan->ctx, &iter);
55cc08a6 472 assert(!ret);
fb45065e 473 delete_ust_app_ctx(sock, ua_ctx, app);
55cc08a6 474 }
d80a6244 475
55cc08a6 476 /* Wipe events */
bec39940
DG
477 cds_lfht_for_each_entry(ua_chan->events->ht, &iter.iter, ua_event,
478 node.node) {
479 ret = lttng_ht_del(ua_chan->events, &iter);
525b0740 480 assert(!ret);
fb45065e 481 delete_ust_app_event(sock, ua_event, app);
d80a6244 482 }
edb67388 483
c8335706
MD
484 if (ua_chan->session->buffer_type == LTTNG_BUFFER_PER_PID) {
485 /* Wipe and free registry from session registry. */
486 registry = get_session_registry(ua_chan->session);
487 if (registry) {
e9404c27 488 ust_registry_channel_del_free(registry, ua_chan->key,
2ac146ae
MD
489 sock >= 0);
490 }
491 if (sock >= 0) {
492 save_per_pid_lost_discarded_counters(ua_chan);
c8335706 493 }
7972aab2 494 }
d0b96690 495
edb67388 496 if (ua_chan->obj != NULL) {
d0b96690
DG
497 /* Remove channel from application UST object descriptor. */
498 iter.iter.node = &ua_chan->ust_objd_node.node;
c6e62271
DG
499 ret = lttng_ht_del(app->ust_objd, &iter);
500 assert(!ret);
fb45065e 501 pthread_mutex_lock(&app->sock_lock);
ffe60014 502 ret = ustctl_release_object(sock, ua_chan->obj);
fb45065e 503 pthread_mutex_unlock(&app->sock_lock);
ffe60014
DG
504 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
505 ERR("UST app sock %d release channel obj failed with ret %d",
506 sock, ret);
507 }
7972aab2 508 lttng_fd_put(LTTNG_FD_APPS, 1);
edb67388
DG
509 free(ua_chan->obj);
510 }
36b588ed 511 call_rcu(&ua_chan->rcu_head, delete_ust_app_channel_rcu);
d80a6244
DG
512}
513
fb45065e
MD
514int ust_app_register_done(struct ust_app *app)
515{
516 int ret;
517
518 pthread_mutex_lock(&app->sock_lock);
519 ret = ustctl_register_done(app->sock);
520 pthread_mutex_unlock(&app->sock_lock);
521 return ret;
522}
523
524int ust_app_release_object(struct ust_app *app, struct lttng_ust_object_data *data)
525{
526 int ret, sock;
527
528 if (app) {
529 pthread_mutex_lock(&app->sock_lock);
530 sock = app->sock;
531 } else {
532 sock = -1;
533 }
534 ret = ustctl_release_object(sock, data);
535 if (app) {
536 pthread_mutex_unlock(&app->sock_lock);
537 }
538 return ret;
539}
540
331744e3 541/*
1b532a60
DG
542 * Push metadata to consumer socket.
543 *
dc2bbdae
MD
544 * RCU read-side lock must be held to guarantee existance of socket.
545 * Must be called with the ust app session lock held.
546 * Must be called with the registry lock held.
331744e3
JD
547 *
548 * On success, return the len of metadata pushed or else a negative value.
2c57e06d
MD
549 * Returning a -EPIPE return value means we could not send the metadata,
550 * but it can be caused by recoverable errors (e.g. the application has
551 * terminated concurrently).
331744e3
JD
552 */
553ssize_t ust_app_push_metadata(struct ust_registry_session *registry,
554 struct consumer_socket *socket, int send_zero_data)
555{
556 int ret;
557 char *metadata_str = NULL;
c585821b 558 size_t len, offset, new_metadata_len_sent;
331744e3 559 ssize_t ret_val;
93ec662e 560 uint64_t metadata_key, metadata_version;
331744e3
JD
561
562 assert(registry);
563 assert(socket);
1b532a60 564
c585821b
MD
565 metadata_key = registry->metadata_key;
566
ce34fcd0 567 /*
dc2bbdae
MD
568 * Means that no metadata was assigned to the session. This can
569 * happens if no start has been done previously.
ce34fcd0 570 */
c585821b 571 if (!metadata_key) {
ce34fcd0
MD
572 return 0;
573 }
574
331744e3
JD
575 offset = registry->metadata_len_sent;
576 len = registry->metadata_len - registry->metadata_len_sent;
c585821b 577 new_metadata_len_sent = registry->metadata_len;
93ec662e 578 metadata_version = registry->metadata_version;
331744e3
JD
579 if (len == 0) {
580 DBG3("No metadata to push for metadata key %" PRIu64,
581 registry->metadata_key);
582 ret_val = len;
583 if (send_zero_data) {
584 DBG("No metadata to push");
585 goto push_data;
586 }
587 goto end;
588 }
589
590 /* Allocate only what we have to send. */
591 metadata_str = zmalloc(len);
592 if (!metadata_str) {
593 PERROR("zmalloc ust app metadata string");
594 ret_val = -ENOMEM;
595 goto error;
596 }
c585821b 597 /* Copy what we haven't sent out. */
331744e3 598 memcpy(metadata_str, registry->metadata + offset, len);
331744e3
JD
599
600push_data:
c585821b
MD
601 pthread_mutex_unlock(&registry->lock);
602 /*
603 * We need to unlock the registry while we push metadata to
604 * break a circular dependency between the consumerd metadata
605 * lock and the sessiond registry lock. Indeed, pushing metadata
606 * to the consumerd awaits that it gets pushed all the way to
607 * relayd, but doing so requires grabbing the metadata lock. If
608 * a concurrent metadata request is being performed by
609 * consumerd, this can try to grab the registry lock on the
610 * sessiond while holding the metadata lock on the consumer
611 * daemon. Those push and pull schemes are performed on two
612 * different bidirectionnal communication sockets.
613 */
614 ret = consumer_push_metadata(socket, metadata_key,
93ec662e 615 metadata_str, len, offset, metadata_version);
c585821b 616 pthread_mutex_lock(&registry->lock);
331744e3 617 if (ret < 0) {
000baf6a 618 /*
dc2bbdae
MD
619 * There is an acceptable race here between the registry
620 * metadata key assignment and the creation on the
621 * consumer. The session daemon can concurrently push
622 * metadata for this registry while being created on the
623 * consumer since the metadata key of the registry is
624 * assigned *before* it is setup to avoid the consumer
625 * to ask for metadata that could possibly be not found
626 * in the session daemon.
000baf6a 627 *
dc2bbdae
MD
628 * The metadata will get pushed either by the session
629 * being stopped or the consumer requesting metadata if
630 * that race is triggered.
000baf6a
DG
631 */
632 if (ret == -LTTCOMM_CONSUMERD_CHANNEL_FAIL) {
633 ret = 0;
c585821b
MD
634 } else {
635 ERR("Error pushing metadata to consumer");
000baf6a 636 }
331744e3
JD
637 ret_val = ret;
638 goto error_push;
c585821b
MD
639 } else {
640 /*
641 * Metadata may have been concurrently pushed, since
642 * we're not holding the registry lock while pushing to
643 * consumer. This is handled by the fact that we send
644 * the metadata content, size, and the offset at which
645 * that metadata belongs. This may arrive out of order
646 * on the consumer side, and the consumer is able to
647 * deal with overlapping fragments. The consumer
648 * supports overlapping fragments, which must be
649 * contiguous starting from offset 0. We keep the
650 * largest metadata_len_sent value of the concurrent
651 * send.
652 */
653 registry->metadata_len_sent =
654 max_t(size_t, registry->metadata_len_sent,
655 new_metadata_len_sent);
331744e3 656 }
331744e3
JD
657 free(metadata_str);
658 return len;
659
660end:
661error:
ce34fcd0
MD
662 if (ret_val) {
663 /*
dc2bbdae
MD
664 * On error, flag the registry that the metadata is
665 * closed. We were unable to push anything and this
666 * means that either the consumer is not responding or
667 * the metadata cache has been destroyed on the
668 * consumer.
ce34fcd0
MD
669 */
670 registry->metadata_closed = 1;
671 }
331744e3
JD
672error_push:
673 free(metadata_str);
674 return ret_val;
675}
676
d88aee68 677/*
ce34fcd0 678 * For a given application and session, push metadata to consumer.
331744e3
JD
679 * Either sock or consumer is required : if sock is NULL, the default
680 * socket to send the metadata is retrieved from consumer, if sock
681 * is not NULL we use it to send the metadata.
ce34fcd0 682 * RCU read-side lock must be held while calling this function,
dc2bbdae
MD
683 * therefore ensuring existance of registry. It also ensures existance
684 * of socket throughout this function.
d88aee68
DG
685 *
686 * Return 0 on success else a negative error.
2c57e06d
MD
687 * Returning a -EPIPE return value means we could not send the metadata,
688 * but it can be caused by recoverable errors (e.g. the application has
689 * terminated concurrently).
d88aee68 690 */
7972aab2
DG
691static int push_metadata(struct ust_registry_session *registry,
692 struct consumer_output *consumer)
d88aee68 693{
331744e3
JD
694 int ret_val;
695 ssize_t ret;
d88aee68
DG
696 struct consumer_socket *socket;
697
7972aab2
DG
698 assert(registry);
699 assert(consumer);
700
ce34fcd0 701 pthread_mutex_lock(&registry->lock);
ce34fcd0 702 if (registry->metadata_closed) {
dc2bbdae
MD
703 ret_val = -EPIPE;
704 goto error;
d88aee68
DG
705 }
706
d88aee68 707 /* Get consumer socket to use to push the metadata.*/
7972aab2
DG
708 socket = consumer_find_socket_by_bitness(registry->bits_per_long,
709 consumer);
d88aee68 710 if (!socket) {
331744e3 711 ret_val = -1;
ce34fcd0 712 goto error;
d88aee68
DG
713 }
714
331744e3 715 ret = ust_app_push_metadata(registry, socket, 0);
d88aee68 716 if (ret < 0) {
331744e3 717 ret_val = ret;
ce34fcd0 718 goto error;
d88aee68 719 }
dc2bbdae 720 pthread_mutex_unlock(&registry->lock);
d88aee68
DG
721 return 0;
722
ce34fcd0 723error:
dc2bbdae 724 pthread_mutex_unlock(&registry->lock);
331744e3 725 return ret_val;
d88aee68
DG
726}
727
728/*
729 * Send to the consumer a close metadata command for the given session. Once
730 * done, the metadata channel is deleted and the session metadata pointer is
dc2bbdae 731 * nullified. The session lock MUST be held unless the application is
d88aee68
DG
732 * in the destroy path.
733 *
734 * Return 0 on success else a negative value.
735 */
7972aab2
DG
736static int close_metadata(struct ust_registry_session *registry,
737 struct consumer_output *consumer)
d88aee68
DG
738{
739 int ret;
740 struct consumer_socket *socket;
741
7972aab2
DG
742 assert(registry);
743 assert(consumer);
d88aee68 744
7972aab2
DG
745 rcu_read_lock();
746
ce34fcd0
MD
747 pthread_mutex_lock(&registry->lock);
748
7972aab2 749 if (!registry->metadata_key || registry->metadata_closed) {
d88aee68 750 ret = 0;
1b532a60 751 goto end;
d88aee68
DG
752 }
753
d88aee68 754 /* Get consumer socket to use to push the metadata.*/
7972aab2
DG
755 socket = consumer_find_socket_by_bitness(registry->bits_per_long,
756 consumer);
d88aee68
DG
757 if (!socket) {
758 ret = -1;
7972aab2 759 goto error;
d88aee68
DG
760 }
761
7972aab2 762 ret = consumer_close_metadata(socket, registry->metadata_key);
d88aee68 763 if (ret < 0) {
7972aab2 764 goto error;
d88aee68
DG
765 }
766
d88aee68 767error:
1b532a60
DG
768 /*
769 * Metadata closed. Even on error this means that the consumer is not
770 * responding or not found so either way a second close should NOT be emit
771 * for this registry.
772 */
773 registry->metadata_closed = 1;
774end:
ce34fcd0 775 pthread_mutex_unlock(&registry->lock);
7972aab2 776 rcu_read_unlock();
d88aee68
DG
777 return ret;
778}
779
36b588ed
MD
780/*
781 * We need to execute ht_destroy outside of RCU read-side critical
0b2dc8df
MD
782 * section and outside of call_rcu thread, so we postpone its execution
783 * using ht_cleanup_push. It is simpler than to change the semantic of
784 * the many callers of delete_ust_app_session().
36b588ed
MD
785 */
786static
787void delete_ust_app_session_rcu(struct rcu_head *head)
788{
789 struct ust_app_session *ua_sess =
790 caa_container_of(head, struct ust_app_session, rcu_head);
791
0b2dc8df 792 ht_cleanup_push(ua_sess->channels);
36b588ed
MD
793 free(ua_sess);
794}
795
d80a6244
DG
796/*
797 * Delete ust app session safely. RCU read lock must be held before calling
798 * this function.
82cac6d2
JG
799 *
800 * The session list lock must be held by the caller.
d80a6244 801 */
8b366481 802static
d0b96690
DG
803void delete_ust_app_session(int sock, struct ust_app_session *ua_sess,
804 struct ust_app *app)
d80a6244
DG
805{
806 int ret;
bec39940 807 struct lttng_ht_iter iter;
d80a6244 808 struct ust_app_channel *ua_chan;
7972aab2 809 struct ust_registry_session *registry;
d80a6244 810
d88aee68
DG
811 assert(ua_sess);
812
1b532a60
DG
813 pthread_mutex_lock(&ua_sess->lock);
814
b161602a
MD
815 assert(!ua_sess->deleted);
816 ua_sess->deleted = true;
817
7972aab2 818 registry = get_session_registry(ua_sess);
5e98072f 819 /* Registry can be null on error path during initialization. */
ce34fcd0 820 if (registry) {
d88aee68 821 /* Push metadata for application before freeing the application. */
7972aab2 822 (void) push_metadata(registry, ua_sess->consumer);
d88aee68 823
7972aab2
DG
824 /*
825 * Don't ask to close metadata for global per UID buffers. Close
1b532a60
DG
826 * metadata only on destroy trace session in this case. Also, the
827 * previous push metadata could have flag the metadata registry to
828 * close so don't send a close command if closed.
7972aab2 829 */
ce34fcd0 830 if (ua_sess->buffer_type != LTTNG_BUFFER_PER_UID) {
7972aab2
DG
831 /* And ask to close it for this session registry. */
832 (void) close_metadata(registry, ua_sess->consumer);
833 }
d80a6244
DG
834 }
835
bec39940
DG
836 cds_lfht_for_each_entry(ua_sess->channels->ht, &iter.iter, ua_chan,
837 node.node) {
838 ret = lttng_ht_del(ua_sess->channels, &iter);
525b0740 839 assert(!ret);
d0b96690 840 delete_ust_app_channel(sock, ua_chan, app);
d80a6244 841 }
d80a6244 842
7972aab2
DG
843 /* In case of per PID, the registry is kept in the session. */
844 if (ua_sess->buffer_type == LTTNG_BUFFER_PER_PID) {
845 struct buffer_reg_pid *reg_pid = buffer_reg_pid_find(ua_sess->id);
846 if (reg_pid) {
5e98072f
JR
847 /*
848 * Registry can be null on error path during
849 * initialization.
850 */
7972aab2
DG
851 buffer_reg_pid_remove(reg_pid);
852 buffer_reg_pid_destroy(reg_pid);
853 }
854 }
d0b96690 855
aee6bafd 856 if (ua_sess->handle != -1) {
fb45065e 857 pthread_mutex_lock(&app->sock_lock);
ffe60014 858 ret = ustctl_release_handle(sock, ua_sess->handle);
fb45065e 859 pthread_mutex_unlock(&app->sock_lock);
ffe60014
DG
860 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
861 ERR("UST app sock %d release session handle failed with ret %d",
862 sock, ret);
863 }
10b56aef
MD
864 /* Remove session from application UST object descriptor. */
865 iter.iter.node = &ua_sess->ust_objd_node.node;
866 ret = lttng_ht_del(app->ust_sessions_objd, &iter);
867 assert(!ret);
aee6bafd 868 }
10b56aef 869
1b532a60
DG
870 pthread_mutex_unlock(&ua_sess->lock);
871
6addfa37
MD
872 consumer_output_put(ua_sess->consumer);
873
36b588ed 874 call_rcu(&ua_sess->rcu_head, delete_ust_app_session_rcu);
d80a6244 875}
91d76f53
DG
876
877/*
284d8f55
DG
878 * Delete a traceable application structure from the global list. Never call
879 * this function outside of a call_rcu call.
36b588ed
MD
880 *
881 * RCU read side lock should _NOT_ be held when calling this function.
91d76f53 882 */
8b366481
DG
883static
884void delete_ust_app(struct ust_app *app)
91d76f53 885{
8b366481 886 int ret, sock;
d42f20df 887 struct ust_app_session *ua_sess, *tmp_ua_sess;
44d3bd01 888
82cac6d2
JG
889 /*
890 * The session list lock must be held during this function to guarantee
891 * the existence of ua_sess.
892 */
893 session_lock_list();
d80a6244 894 /* Delete ust app sessions info */
852d0037
DG
895 sock = app->sock;
896 app->sock = -1;
d80a6244 897
8b366481 898 /* Wipe sessions */
d42f20df
DG
899 cds_list_for_each_entry_safe(ua_sess, tmp_ua_sess, &app->teardown_head,
900 teardown_node) {
901 /* Free every object in the session and the session. */
36b588ed 902 rcu_read_lock();
d0b96690 903 delete_ust_app_session(sock, ua_sess, app);
36b588ed 904 rcu_read_unlock();
d80a6244 905 }
36b588ed 906
0b2dc8df 907 ht_cleanup_push(app->sessions);
10b56aef 908 ht_cleanup_push(app->ust_sessions_objd);
0b2dc8df 909 ht_cleanup_push(app->ust_objd);
d80a6244 910
6414a713 911 /*
852d0037
DG
912 * Wait until we have deleted the application from the sock hash table
913 * before closing this socket, otherwise an application could re-use the
914 * socket ID and race with the teardown, using the same hash table entry.
915 *
916 * It's OK to leave the close in call_rcu. We want it to stay unique for
917 * all RCU readers that could run concurrently with unregister app,
918 * therefore we _need_ to only close that socket after a grace period. So
919 * it should stay in this RCU callback.
920 *
921 * This close() is a very important step of the synchronization model so
922 * every modification to this function must be carefully reviewed.
6414a713 923 */
799e2c4f
MD
924 ret = close(sock);
925 if (ret) {
926 PERROR("close");
927 }
4063050c 928 lttng_fd_put(LTTNG_FD_APPS, 1);
d80a6244 929
852d0037 930 DBG2("UST app pid %d deleted", app->pid);
284d8f55 931 free(app);
82cac6d2 932 session_unlock_list();
099e26bd
DG
933}
934
935/*
f6a9efaa 936 * URCU intermediate call to delete an UST app.
099e26bd 937 */
8b366481
DG
938static
939void delete_ust_app_rcu(struct rcu_head *head)
099e26bd 940{
bec39940
DG
941 struct lttng_ht_node_ulong *node =
942 caa_container_of(head, struct lttng_ht_node_ulong, head);
f6a9efaa 943 struct ust_app *app =
852d0037 944 caa_container_of(node, struct ust_app, pid_n);
f6a9efaa 945
852d0037 946 DBG3("Call RCU deleting app PID %d", app->pid);
f6a9efaa 947 delete_ust_app(app);
099e26bd
DG
948}
949
ffe60014
DG
950/*
951 * Delete the session from the application ht and delete the data structure by
952 * freeing every object inside and releasing them.
82cac6d2
JG
953 *
954 * The session list lock must be held by the caller.
ffe60014 955 */
d0b96690 956static void destroy_app_session(struct ust_app *app,
ffe60014
DG
957 struct ust_app_session *ua_sess)
958{
959 int ret;
960 struct lttng_ht_iter iter;
961
962 assert(app);
963 assert(ua_sess);
964
965 iter.iter.node = &ua_sess->node.node;
966 ret = lttng_ht_del(app->sessions, &iter);
967 if (ret) {
968 /* Already scheduled for teardown. */
969 goto end;
970 }
971
972 /* Once deleted, free the data structure. */
d0b96690 973 delete_ust_app_session(app->sock, ua_sess, app);
ffe60014
DG
974
975end:
976 return;
977}
978
8b366481
DG
979/*
980 * Alloc new UST app session.
981 */
982static
d0b96690 983struct ust_app_session *alloc_ust_app_session(struct ust_app *app)
8b366481
DG
984{
985 struct ust_app_session *ua_sess;
986
987 /* Init most of the default value by allocating and zeroing */
988 ua_sess = zmalloc(sizeof(struct ust_app_session));
989 if (ua_sess == NULL) {
990 PERROR("malloc");
ffe60014 991 goto error_free;
8b366481
DG
992 }
993
994 ua_sess->handle = -1;
bec39940 995 ua_sess->channels = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
ad7a9107 996 ua_sess->metadata_attr.type = LTTNG_UST_CHAN_METADATA;
84ad93e8 997 pthread_mutex_init(&ua_sess->lock, NULL);
ad7a9107 998
8b366481
DG
999 return ua_sess;
1000
ffe60014 1001error_free:
8b366481
DG
1002 return NULL;
1003}
1004
1005/*
1006 * Alloc new UST app channel.
1007 */
1008static
1009struct ust_app_channel *alloc_ust_app_channel(char *name,
d0b96690 1010 struct ust_app_session *ua_sess,
ffe60014 1011 struct lttng_ust_channel_attr *attr)
8b366481
DG
1012{
1013 struct ust_app_channel *ua_chan;
1014
1015 /* Init most of the default value by allocating and zeroing */
1016 ua_chan = zmalloc(sizeof(struct ust_app_channel));
1017 if (ua_chan == NULL) {
1018 PERROR("malloc");
1019 goto error;
1020 }
1021
1022 /* Setup channel name */
1023 strncpy(ua_chan->name, name, sizeof(ua_chan->name));
1024 ua_chan->name[sizeof(ua_chan->name) - 1] = '\0';
1025
1026 ua_chan->enabled = 1;
1027 ua_chan->handle = -1;
45893984 1028 ua_chan->session = ua_sess;
ffe60014 1029 ua_chan->key = get_next_channel_key();
bec39940
DG
1030 ua_chan->ctx = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
1031 ua_chan->events = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
1032 lttng_ht_node_init_str(&ua_chan->node, ua_chan->name);
8b366481
DG
1033
1034 CDS_INIT_LIST_HEAD(&ua_chan->streams.head);
31746f93 1035 CDS_INIT_LIST_HEAD(&ua_chan->ctx_list);
8b366481
DG
1036
1037 /* Copy attributes */
1038 if (attr) {
ffe60014 1039 /* Translate from lttng_ust_channel to ustctl_consumer_channel_attr. */
2fe6e7f5
DG
1040 ua_chan->attr.subbuf_size = attr->subbuf_size;
1041 ua_chan->attr.num_subbuf = attr->num_subbuf;
1042 ua_chan->attr.overwrite = attr->overwrite;
1043 ua_chan->attr.switch_timer_interval = attr->switch_timer_interval;
1044 ua_chan->attr.read_timer_interval = attr->read_timer_interval;
1045 ua_chan->attr.output = attr->output;
f515d318 1046 ua_chan->attr.blocking_timeout = attr->u.s.blocking_timeout;
8b366481 1047 }
ffe60014
DG
1048 /* By default, the channel is a per cpu channel. */
1049 ua_chan->attr.type = LTTNG_UST_CHAN_PER_CPU;
8b366481
DG
1050
1051 DBG3("UST app channel %s allocated", ua_chan->name);
1052
1053 return ua_chan;
1054
1055error:
1056 return NULL;
1057}
1058
37f1c236
DG
1059/*
1060 * Allocate and initialize a UST app stream.
1061 *
1062 * Return newly allocated stream pointer or NULL on error.
1063 */
ffe60014 1064struct ust_app_stream *ust_app_alloc_stream(void)
37f1c236
DG
1065{
1066 struct ust_app_stream *stream = NULL;
1067
1068 stream = zmalloc(sizeof(*stream));
1069 if (stream == NULL) {
1070 PERROR("zmalloc ust app stream");
1071 goto error;
1072 }
1073
1074 /* Zero could be a valid value for a handle so flag it to -1. */
1075 stream->handle = -1;
1076
1077error:
1078 return stream;
1079}
1080
8b366481
DG
1081/*
1082 * Alloc new UST app event.
1083 */
1084static
1085struct ust_app_event *alloc_ust_app_event(char *name,
1086 struct lttng_ust_event *attr)
1087{
1088 struct ust_app_event *ua_event;
1089
1090 /* Init most of the default value by allocating and zeroing */
1091 ua_event = zmalloc(sizeof(struct ust_app_event));
1092 if (ua_event == NULL) {
1093 PERROR("malloc");
1094 goto error;
1095 }
1096
1097 ua_event->enabled = 1;
1098 strncpy(ua_event->name, name, sizeof(ua_event->name));
1099 ua_event->name[sizeof(ua_event->name) - 1] = '\0';
bec39940 1100 lttng_ht_node_init_str(&ua_event->node, ua_event->name);
8b366481
DG
1101
1102 /* Copy attributes */
1103 if (attr) {
1104 memcpy(&ua_event->attr, attr, sizeof(ua_event->attr));
1105 }
1106
1107 DBG3("UST app event %s allocated", ua_event->name);
1108
1109 return ua_event;
1110
1111error:
1112 return NULL;
1113}
1114
1115/*
1116 * Alloc new UST app context.
1117 */
1118static
bdf64013 1119struct ust_app_ctx *alloc_ust_app_ctx(struct lttng_ust_context_attr *uctx)
8b366481
DG
1120{
1121 struct ust_app_ctx *ua_ctx;
1122
1123 ua_ctx = zmalloc(sizeof(struct ust_app_ctx));
1124 if (ua_ctx == NULL) {
1125 goto error;
1126 }
1127
31746f93
DG
1128 CDS_INIT_LIST_HEAD(&ua_ctx->list);
1129
8b366481
DG
1130 if (uctx) {
1131 memcpy(&ua_ctx->ctx, uctx, sizeof(ua_ctx->ctx));
bdf64013
JG
1132 if (uctx->ctx == LTTNG_UST_CONTEXT_APP_CONTEXT) {
1133 char *provider_name = NULL, *ctx_name = NULL;
1134
1135 provider_name = strdup(uctx->u.app_ctx.provider_name);
1136 ctx_name = strdup(uctx->u.app_ctx.ctx_name);
1137 if (!provider_name || !ctx_name) {
1138 free(provider_name);
1139 free(ctx_name);
1140 goto error;
1141 }
1142
1143 ua_ctx->ctx.u.app_ctx.provider_name = provider_name;
1144 ua_ctx->ctx.u.app_ctx.ctx_name = ctx_name;
1145 }
8b366481
DG
1146 }
1147
1148 DBG3("UST app context %d allocated", ua_ctx->ctx.ctx);
8b366481 1149 return ua_ctx;
bdf64013
JG
1150error:
1151 free(ua_ctx);
1152 return NULL;
8b366481
DG
1153}
1154
025faf73
DG
1155/*
1156 * Allocate a filter and copy the given original filter.
1157 *
1158 * Return allocated filter or NULL on error.
1159 */
51755dc8
JG
1160static struct lttng_filter_bytecode *copy_filter_bytecode(
1161 struct lttng_filter_bytecode *orig_f)
025faf73 1162{
51755dc8 1163 struct lttng_filter_bytecode *filter = NULL;
025faf73
DG
1164
1165 /* Copy filter bytecode */
1166 filter = zmalloc(sizeof(*filter) + orig_f->len);
1167 if (!filter) {
51755dc8 1168 PERROR("zmalloc alloc filter bytecode");
025faf73
DG
1169 goto error;
1170 }
1171
1172 memcpy(filter, orig_f, sizeof(*filter) + orig_f->len);
1173
1174error:
1175 return filter;
1176}
1177
51755dc8
JG
1178/*
1179 * Create a liblttng-ust filter bytecode from given bytecode.
1180 *
1181 * Return allocated filter or NULL on error.
1182 */
1183static struct lttng_ust_filter_bytecode *create_ust_bytecode_from_bytecode(
1184 struct lttng_filter_bytecode *orig_f)
1185{
1186 struct lttng_ust_filter_bytecode *filter = NULL;
1187
1188 /* Copy filter bytecode */
1189 filter = zmalloc(sizeof(*filter) + orig_f->len);
1190 if (!filter) {
1191 PERROR("zmalloc alloc ust filter bytecode");
1192 goto error;
1193 }
1194
1195 assert(sizeof(struct lttng_filter_bytecode) ==
1196 sizeof(struct lttng_ust_filter_bytecode));
1197 memcpy(filter, orig_f, sizeof(*filter) + orig_f->len);
1198error:
1199 return filter;
1200}
1201
099e26bd 1202/*
421cb601
DG
1203 * Find an ust_app using the sock and return it. RCU read side lock must be
1204 * held before calling this helper function.
099e26bd 1205 */
f20baf8e 1206struct ust_app *ust_app_find_by_sock(int sock)
099e26bd 1207{
bec39940 1208 struct lttng_ht_node_ulong *node;
bec39940 1209 struct lttng_ht_iter iter;
f6a9efaa 1210
852d0037 1211 lttng_ht_lookup(ust_app_ht_by_sock, (void *)((unsigned long) sock), &iter);
bec39940 1212 node = lttng_ht_iter_get_node_ulong(&iter);
f6a9efaa
DG
1213 if (node == NULL) {
1214 DBG2("UST app find by sock %d not found", sock);
f6a9efaa
DG
1215 goto error;
1216 }
852d0037
DG
1217
1218 return caa_container_of(node, struct ust_app, sock_n);
f6a9efaa
DG
1219
1220error:
1221 return NULL;
099e26bd
DG
1222}
1223
d0b96690
DG
1224/*
1225 * Find an ust_app using the notify sock and return it. RCU read side lock must
1226 * be held before calling this helper function.
1227 */
1228static struct ust_app *find_app_by_notify_sock(int sock)
1229{
1230 struct lttng_ht_node_ulong *node;
1231 struct lttng_ht_iter iter;
1232
1233 lttng_ht_lookup(ust_app_ht_by_notify_sock, (void *)((unsigned long) sock),
1234 &iter);
1235 node = lttng_ht_iter_get_node_ulong(&iter);
1236 if (node == NULL) {
1237 DBG2("UST app find by notify sock %d not found", sock);
1238 goto error;
1239 }
1240
1241 return caa_container_of(node, struct ust_app, notify_sock_n);
1242
1243error:
1244 return NULL;
1245}
1246
025faf73
DG
1247/*
1248 * Lookup for an ust app event based on event name, filter bytecode and the
1249 * event loglevel.
1250 *
1251 * Return an ust_app_event object or NULL on error.
1252 */
18eace3b 1253static struct ust_app_event *find_ust_app_event(struct lttng_ht *ht,
2106efa0
PP
1254 char *name, struct lttng_filter_bytecode *filter,
1255 int loglevel_value,
39c5a3a7 1256 const struct lttng_event_exclusion *exclusion)
18eace3b
DG
1257{
1258 struct lttng_ht_iter iter;
1259 struct lttng_ht_node_str *node;
1260 struct ust_app_event *event = NULL;
1261 struct ust_app_ht_key key;
18eace3b
DG
1262
1263 assert(name);
1264 assert(ht);
1265
1266 /* Setup key for event lookup. */
1267 key.name = name;
1268 key.filter = filter;
2106efa0 1269 key.loglevel_type = loglevel_value;
39c5a3a7 1270 /* lttng_event_exclusion and lttng_ust_event_exclusion structures are similar */
51755dc8 1271 key.exclusion = exclusion;
18eace3b 1272
025faf73
DG
1273 /* Lookup using the event name as hash and a custom match fct. */
1274 cds_lfht_lookup(ht->ht, ht->hash_fct((void *) name, lttng_ht_seed),
1275 ht_match_ust_app_event, &key, &iter.iter);
18eace3b
DG
1276 node = lttng_ht_iter_get_node_str(&iter);
1277 if (node == NULL) {
1278 goto end;
1279 }
1280
1281 event = caa_container_of(node, struct ust_app_event, node);
1282
1283end:
18eace3b
DG
1284 return event;
1285}
1286
55cc08a6
DG
1287/*
1288 * Create the channel context on the tracer.
d0b96690
DG
1289 *
1290 * Called with UST app session lock held.
55cc08a6
DG
1291 */
1292static
1293int create_ust_channel_context(struct ust_app_channel *ua_chan,
1294 struct ust_app_ctx *ua_ctx, struct ust_app *app)
1295{
1296 int ret;
1297
840cb59c 1298 health_code_update();
86acf0da 1299
fb45065e 1300 pthread_mutex_lock(&app->sock_lock);
852d0037 1301 ret = ustctl_add_context(app->sock, &ua_ctx->ctx,
55cc08a6 1302 ua_chan->obj, &ua_ctx->obj);
fb45065e 1303 pthread_mutex_unlock(&app->sock_lock);
55cc08a6 1304 if (ret < 0) {
ffe60014
DG
1305 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1306 ERR("UST app create channel context failed for app (pid: %d) "
1307 "with ret %d", app->pid, ret);
1308 } else {
3757b385
DG
1309 /*
1310 * This is normal behavior, an application can die during the
1311 * creation process. Don't report an error so the execution can
1312 * continue normally.
1313 */
1314 ret = 0;
ffe60014
DG
1315 DBG3("UST app disable event failed. Application is dead.");
1316 }
55cc08a6
DG
1317 goto error;
1318 }
1319
1320 ua_ctx->handle = ua_ctx->obj->handle;
1321
d0b96690
DG
1322 DBG2("UST app context handle %d created successfully for channel %s",
1323 ua_ctx->handle, ua_chan->name);
55cc08a6
DG
1324
1325error:
840cb59c 1326 health_code_update();
55cc08a6
DG
1327 return ret;
1328}
1329
53a80697
MD
1330/*
1331 * Set the filter on the tracer.
1332 */
1333static
1334int set_ust_event_filter(struct ust_app_event *ua_event,
1335 struct ust_app *app)
1336{
1337 int ret;
51755dc8 1338 struct lttng_ust_filter_bytecode *ust_bytecode = NULL;
53a80697 1339
840cb59c 1340 health_code_update();
86acf0da 1341
53a80697 1342 if (!ua_event->filter) {
86acf0da
DG
1343 ret = 0;
1344 goto error;
53a80697
MD
1345 }
1346
51755dc8
JG
1347 ust_bytecode = create_ust_bytecode_from_bytecode(ua_event->filter);
1348 if (!ust_bytecode) {
1349 ret = -LTTNG_ERR_NOMEM;
1350 goto error;
1351 }
fb45065e 1352 pthread_mutex_lock(&app->sock_lock);
51755dc8 1353 ret = ustctl_set_filter(app->sock, ust_bytecode,
53a80697 1354 ua_event->obj);
fb45065e 1355 pthread_mutex_unlock(&app->sock_lock);
53a80697 1356 if (ret < 0) {
ffe60014
DG
1357 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1358 ERR("UST app event %s filter failed for app (pid: %d) "
1359 "with ret %d", ua_event->attr.name, app->pid, ret);
1360 } else {
3757b385
DG
1361 /*
1362 * This is normal behavior, an application can die during the
1363 * creation process. Don't report an error so the execution can
1364 * continue normally.
1365 */
1366 ret = 0;
ffe60014
DG
1367 DBG3("UST app filter event failed. Application is dead.");
1368 }
53a80697
MD
1369 goto error;
1370 }
1371
1372 DBG2("UST filter set successfully for event %s", ua_event->name);
1373
1374error:
840cb59c 1375 health_code_update();
51755dc8 1376 free(ust_bytecode);
53a80697
MD
1377 return ret;
1378}
1379
51755dc8
JG
1380static
1381struct lttng_ust_event_exclusion *create_ust_exclusion_from_exclusion(
1382 struct lttng_event_exclusion *exclusion)
1383{
1384 struct lttng_ust_event_exclusion *ust_exclusion = NULL;
1385 size_t exclusion_alloc_size = sizeof(struct lttng_ust_event_exclusion) +
1386 LTTNG_UST_SYM_NAME_LEN * exclusion->count;
1387
1388 ust_exclusion = zmalloc(exclusion_alloc_size);
1389 if (!ust_exclusion) {
1390 PERROR("malloc");
1391 goto end;
1392 }
1393
1394 assert(sizeof(struct lttng_event_exclusion) ==
1395 sizeof(struct lttng_ust_event_exclusion));
1396 memcpy(ust_exclusion, exclusion, exclusion_alloc_size);
1397end:
1398 return ust_exclusion;
1399}
1400
7cc9a73c
JI
1401/*
1402 * Set event exclusions on the tracer.
1403 */
1404static
1405int set_ust_event_exclusion(struct ust_app_event *ua_event,
1406 struct ust_app *app)
1407{
1408 int ret;
51755dc8 1409 struct lttng_ust_event_exclusion *ust_exclusion = NULL;
7cc9a73c
JI
1410
1411 health_code_update();
1412
1413 if (!ua_event->exclusion || !ua_event->exclusion->count) {
1414 ret = 0;
1415 goto error;
1416 }
1417
51755dc8
JG
1418 ust_exclusion = create_ust_exclusion_from_exclusion(
1419 ua_event->exclusion);
1420 if (!ust_exclusion) {
1421 ret = -LTTNG_ERR_NOMEM;
1422 goto error;
1423 }
fb45065e 1424 pthread_mutex_lock(&app->sock_lock);
51755dc8 1425 ret = ustctl_set_exclusion(app->sock, ust_exclusion, ua_event->obj);
fb45065e 1426 pthread_mutex_unlock(&app->sock_lock);
7cc9a73c
JI
1427 if (ret < 0) {
1428 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1429 ERR("UST app event %s exclusions failed for app (pid: %d) "
1430 "with ret %d", ua_event->attr.name, app->pid, ret);
1431 } else {
1432 /*
1433 * This is normal behavior, an application can die during the
1434 * creation process. Don't report an error so the execution can
1435 * continue normally.
1436 */
1437 ret = 0;
1438 DBG3("UST app event exclusion failed. Application is dead.");
1439 }
1440 goto error;
1441 }
1442
1443 DBG2("UST exclusion set successfully for event %s", ua_event->name);
1444
1445error:
1446 health_code_update();
51755dc8 1447 free(ust_exclusion);
7cc9a73c
JI
1448 return ret;
1449}
1450
9730260e
DG
1451/*
1452 * Disable the specified event on to UST tracer for the UST session.
1453 */
1454static int disable_ust_event(struct ust_app *app,
1455 struct ust_app_session *ua_sess, struct ust_app_event *ua_event)
1456{
1457 int ret;
1458
840cb59c 1459 health_code_update();
86acf0da 1460
fb45065e 1461 pthread_mutex_lock(&app->sock_lock);
852d0037 1462 ret = ustctl_disable(app->sock, ua_event->obj);
fb45065e 1463 pthread_mutex_unlock(&app->sock_lock);
9730260e 1464 if (ret < 0) {
ffe60014
DG
1465 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1466 ERR("UST app event %s disable failed for app (pid: %d) "
1467 "and session handle %d with ret %d",
1468 ua_event->attr.name, app->pid, ua_sess->handle, ret);
1469 } else {
3757b385
DG
1470 /*
1471 * This is normal behavior, an application can die during the
1472 * creation process. Don't report an error so the execution can
1473 * continue normally.
1474 */
1475 ret = 0;
ffe60014
DG
1476 DBG3("UST app disable event failed. Application is dead.");
1477 }
9730260e
DG
1478 goto error;
1479 }
1480
1481 DBG2("UST app event %s disabled successfully for app (pid: %d)",
852d0037 1482 ua_event->attr.name, app->pid);
9730260e
DG
1483
1484error:
840cb59c 1485 health_code_update();
9730260e
DG
1486 return ret;
1487}
1488
78f0bacd
DG
1489/*
1490 * Disable the specified channel on to UST tracer for the UST session.
1491 */
1492static int disable_ust_channel(struct ust_app *app,
1493 struct ust_app_session *ua_sess, struct ust_app_channel *ua_chan)
1494{
1495 int ret;
1496
840cb59c 1497 health_code_update();
86acf0da 1498
fb45065e 1499 pthread_mutex_lock(&app->sock_lock);
852d0037 1500 ret = ustctl_disable(app->sock, ua_chan->obj);
fb45065e 1501 pthread_mutex_unlock(&app->sock_lock);
78f0bacd 1502 if (ret < 0) {
ffe60014
DG
1503 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1504 ERR("UST app channel %s disable failed for app (pid: %d) "
1505 "and session handle %d with ret %d",
1506 ua_chan->name, app->pid, ua_sess->handle, ret);
1507 } else {
3757b385
DG
1508 /*
1509 * This is normal behavior, an application can die during the
1510 * creation process. Don't report an error so the execution can
1511 * continue normally.
1512 */
1513 ret = 0;
ffe60014
DG
1514 DBG3("UST app disable channel failed. Application is dead.");
1515 }
78f0bacd
DG
1516 goto error;
1517 }
1518
78f0bacd 1519 DBG2("UST app channel %s disabled successfully for app (pid: %d)",
852d0037 1520 ua_chan->name, app->pid);
78f0bacd
DG
1521
1522error:
840cb59c 1523 health_code_update();
78f0bacd
DG
1524 return ret;
1525}
1526
1527/*
1528 * Enable the specified channel on to UST tracer for the UST session.
1529 */
1530static int enable_ust_channel(struct ust_app *app,
1531 struct ust_app_session *ua_sess, struct ust_app_channel *ua_chan)
1532{
1533 int ret;
1534
840cb59c 1535 health_code_update();
86acf0da 1536
fb45065e 1537 pthread_mutex_lock(&app->sock_lock);
852d0037 1538 ret = ustctl_enable(app->sock, ua_chan->obj);
fb45065e 1539 pthread_mutex_unlock(&app->sock_lock);
78f0bacd 1540 if (ret < 0) {
ffe60014
DG
1541 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1542 ERR("UST app channel %s enable failed for app (pid: %d) "
1543 "and session handle %d with ret %d",
1544 ua_chan->name, app->pid, ua_sess->handle, ret);
1545 } else {
3757b385
DG
1546 /*
1547 * This is normal behavior, an application can die during the
1548 * creation process. Don't report an error so the execution can
1549 * continue normally.
1550 */
1551 ret = 0;
ffe60014
DG
1552 DBG3("UST app enable channel failed. Application is dead.");
1553 }
78f0bacd
DG
1554 goto error;
1555 }
1556
1557 ua_chan->enabled = 1;
1558
1559 DBG2("UST app channel %s enabled successfully for app (pid: %d)",
852d0037 1560 ua_chan->name, app->pid);
78f0bacd
DG
1561
1562error:
840cb59c 1563 health_code_update();
78f0bacd
DG
1564 return ret;
1565}
1566
edb67388
DG
1567/*
1568 * Enable the specified event on to UST tracer for the UST session.
1569 */
1570static int enable_ust_event(struct ust_app *app,
1571 struct ust_app_session *ua_sess, struct ust_app_event *ua_event)
1572{
1573 int ret;
1574
840cb59c 1575 health_code_update();
86acf0da 1576
fb45065e 1577 pthread_mutex_lock(&app->sock_lock);
852d0037 1578 ret = ustctl_enable(app->sock, ua_event->obj);
fb45065e 1579 pthread_mutex_unlock(&app->sock_lock);
edb67388 1580 if (ret < 0) {
ffe60014
DG
1581 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1582 ERR("UST app event %s enable failed for app (pid: %d) "
1583 "and session handle %d with ret %d",
1584 ua_event->attr.name, app->pid, ua_sess->handle, ret);
1585 } else {
3757b385
DG
1586 /*
1587 * This is normal behavior, an application can die during the
1588 * creation process. Don't report an error so the execution can
1589 * continue normally.
1590 */
1591 ret = 0;
ffe60014
DG
1592 DBG3("UST app enable event failed. Application is dead.");
1593 }
edb67388
DG
1594 goto error;
1595 }
1596
1597 DBG2("UST app event %s enabled successfully for app (pid: %d)",
852d0037 1598 ua_event->attr.name, app->pid);
edb67388
DG
1599
1600error:
840cb59c 1601 health_code_update();
edb67388
DG
1602 return ret;
1603}
1604
099e26bd 1605/*
7972aab2 1606 * Send channel and stream buffer to application.
4f3ab6ee 1607 *
ffe60014 1608 * Return 0 on success. On error, a negative value is returned.
4f3ab6ee 1609 */
7972aab2
DG
1610static int send_channel_pid_to_ust(struct ust_app *app,
1611 struct ust_app_session *ua_sess, struct ust_app_channel *ua_chan)
4f3ab6ee
DG
1612{
1613 int ret;
ffe60014 1614 struct ust_app_stream *stream, *stmp;
4f3ab6ee
DG
1615
1616 assert(app);
ffe60014 1617 assert(ua_sess);
4f3ab6ee 1618 assert(ua_chan);
4f3ab6ee 1619
840cb59c 1620 health_code_update();
4f3ab6ee 1621
7972aab2
DG
1622 DBG("UST app sending channel %s to UST app sock %d", ua_chan->name,
1623 app->sock);
86acf0da 1624
ffe60014
DG
1625 /* Send channel to the application. */
1626 ret = ust_consumer_send_channel_to_ust(app, ua_sess, ua_chan);
a7169585
MD
1627 if (ret == -EPIPE || ret == -LTTNG_UST_ERR_EXITING) {
1628 ret = -ENOTCONN; /* Caused by app exiting. */
1629 goto error;
1630 } else if (ret < 0) {
b551a063
DG
1631 goto error;
1632 }
1633
d88aee68
DG
1634 health_code_update();
1635
ffe60014
DG
1636 /* Send all streams to application. */
1637 cds_list_for_each_entry_safe(stream, stmp, &ua_chan->streams.head, list) {
1638 ret = ust_consumer_send_stream_to_ust(app, ua_chan, stream);
a7169585
MD
1639 if (ret == -EPIPE || ret == -LTTNG_UST_ERR_EXITING) {
1640 ret = -ENOTCONN; /* Caused by app exiting. */
1641 goto error;
1642 } else if (ret < 0) {
ffe60014
DG
1643 goto error;
1644 }
1645 /* We don't need the stream anymore once sent to the tracer. */
1646 cds_list_del(&stream->list);
fb45065e 1647 delete_ust_app_stream(-1, stream, app);
ffe60014 1648 }
ffe60014
DG
1649 /* Flag the channel that it is sent to the application. */
1650 ua_chan->is_sent = 1;
ffe60014 1651
b551a063 1652error:
840cb59c 1653 health_code_update();
b551a063
DG
1654 return ret;
1655}
1656
91d76f53 1657/*
5b4a0ec0 1658 * Create the specified event onto the UST tracer for a UST session.
d0b96690
DG
1659 *
1660 * Should be called with session mutex held.
91d76f53 1661 */
edb67388
DG
1662static
1663int create_ust_event(struct ust_app *app, struct ust_app_session *ua_sess,
1664 struct ust_app_channel *ua_chan, struct ust_app_event *ua_event)
91d76f53 1665{
5b4a0ec0 1666 int ret = 0;
284d8f55 1667
840cb59c 1668 health_code_update();
86acf0da 1669
5b4a0ec0 1670 /* Create UST event on tracer */
fb45065e 1671 pthread_mutex_lock(&app->sock_lock);
852d0037 1672 ret = ustctl_create_event(app->sock, &ua_event->attr, ua_chan->obj,
5b4a0ec0 1673 &ua_event->obj);
fb45065e 1674 pthread_mutex_unlock(&app->sock_lock);
5b4a0ec0 1675 if (ret < 0) {
ffe60014
DG
1676 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1677 ERR("Error ustctl create event %s for app pid: %d with ret %d",
1678 ua_event->attr.name, app->pid, ret);
1679 } else {
3757b385
DG
1680 /*
1681 * This is normal behavior, an application can die during the
1682 * creation process. Don't report an error so the execution can
1683 * continue normally.
1684 */
1685 ret = 0;
ffe60014
DG
1686 DBG3("UST app create event failed. Application is dead.");
1687 }
5b4a0ec0 1688 goto error;
91d76f53 1689 }
f6a9efaa 1690
5b4a0ec0 1691 ua_event->handle = ua_event->obj->handle;
284d8f55 1692
5b4a0ec0 1693 DBG2("UST app event %s created successfully for pid:%d",
852d0037 1694 ua_event->attr.name, app->pid);
f6a9efaa 1695
840cb59c 1696 health_code_update();
86acf0da 1697
025faf73
DG
1698 /* Set filter if one is present. */
1699 if (ua_event->filter) {
1700 ret = set_ust_event_filter(ua_event, app);
1701 if (ret < 0) {
1702 goto error;
1703 }
1704 }
1705
7cc9a73c
JI
1706 /* Set exclusions for the event */
1707 if (ua_event->exclusion) {
1708 ret = set_ust_event_exclusion(ua_event, app);
1709 if (ret < 0) {
1710 goto error;
1711 }
1712 }
1713
8535a6d9 1714 /* If event not enabled, disable it on the tracer */
40113787
MD
1715 if (ua_event->enabled) {
1716 /*
1717 * We now need to explicitly enable the event, since it
1718 * is now disabled at creation.
1719 */
1720 ret = enable_ust_event(app, ua_sess, ua_event);
1721 if (ret < 0) {
1722 /*
1723 * If we hit an EPERM, something is wrong with our enable call. If
1724 * we get an EEXIST, there is a problem on the tracer side since we
1725 * just created it.
1726 */
1727 switch (ret) {
1728 case -LTTNG_UST_ERR_PERM:
1729 /* Code flow problem */
1730 assert(0);
1731 case -LTTNG_UST_ERR_EXIST:
1732 /* It's OK for our use case. */
1733 ret = 0;
1734 break;
1735 default:
1736 break;
1737 }
1738 goto error;
1739 }
8535a6d9
DG
1740 }
1741
5b4a0ec0 1742error:
840cb59c 1743 health_code_update();
5b4a0ec0 1744 return ret;
91d76f53 1745}
48842b30 1746
5b4a0ec0
DG
1747/*
1748 * Copy data between an UST app event and a LTT event.
1749 */
421cb601 1750static void shadow_copy_event(struct ust_app_event *ua_event,
48842b30
DG
1751 struct ltt_ust_event *uevent)
1752{
b4ffad32
JI
1753 size_t exclusion_alloc_size;
1754
48842b30
DG
1755 strncpy(ua_event->name, uevent->attr.name, sizeof(ua_event->name));
1756 ua_event->name[sizeof(ua_event->name) - 1] = '\0';
1757
fc34caaa
DG
1758 ua_event->enabled = uevent->enabled;
1759
5b4a0ec0
DG
1760 /* Copy event attributes */
1761 memcpy(&ua_event->attr, &uevent->attr, sizeof(ua_event->attr));
1762
53a80697
MD
1763 /* Copy filter bytecode */
1764 if (uevent->filter) {
51755dc8 1765 ua_event->filter = copy_filter_bytecode(uevent->filter);
025faf73 1766 /* Filter might be NULL here in case of ENONEM. */
53a80697 1767 }
b4ffad32
JI
1768
1769 /* Copy exclusion data */
1770 if (uevent->exclusion) {
51755dc8 1771 exclusion_alloc_size = sizeof(struct lttng_event_exclusion) +
b4ffad32
JI
1772 LTTNG_UST_SYM_NAME_LEN * uevent->exclusion->count;
1773 ua_event->exclusion = zmalloc(exclusion_alloc_size);
5f8df26c
JI
1774 if (ua_event->exclusion == NULL) {
1775 PERROR("malloc");
1776 } else {
1777 memcpy(ua_event->exclusion, uevent->exclusion,
1778 exclusion_alloc_size);
b4ffad32
JI
1779 }
1780 }
48842b30
DG
1781}
1782
5b4a0ec0
DG
1783/*
1784 * Copy data between an UST app channel and a LTT channel.
1785 */
421cb601 1786static void shadow_copy_channel(struct ust_app_channel *ua_chan,
48842b30
DG
1787 struct ltt_ust_channel *uchan)
1788{
bec39940 1789 struct lttng_ht_iter iter;
48842b30 1790 struct ltt_ust_event *uevent;
55cc08a6 1791 struct ltt_ust_context *uctx;
48842b30
DG
1792 struct ust_app_event *ua_event;
1793
fc34caaa 1794 DBG2("UST app shadow copy of channel %s started", ua_chan->name);
48842b30
DG
1795
1796 strncpy(ua_chan->name, uchan->name, sizeof(ua_chan->name));
1797 ua_chan->name[sizeof(ua_chan->name) - 1] = '\0';
ffe60014 1798
1624d5b7
JD
1799 ua_chan->tracefile_size = uchan->tracefile_size;
1800 ua_chan->tracefile_count = uchan->tracefile_count;
1801
ffe60014
DG
1802 /* Copy event attributes since the layout is different. */
1803 ua_chan->attr.subbuf_size = uchan->attr.subbuf_size;
1804 ua_chan->attr.num_subbuf = uchan->attr.num_subbuf;
1805 ua_chan->attr.overwrite = uchan->attr.overwrite;
1806 ua_chan->attr.switch_timer_interval = uchan->attr.switch_timer_interval;
1807 ua_chan->attr.read_timer_interval = uchan->attr.read_timer_interval;
e9404c27 1808 ua_chan->monitor_timer_interval = uchan->monitor_timer_interval;
ffe60014 1809 ua_chan->attr.output = uchan->attr.output;
f515d318
MD
1810 ua_chan->attr.blocking_timeout = uchan->attr.u.s.blocking_timeout;
1811
ffe60014
DG
1812 /*
1813 * Note that the attribute channel type is not set since the channel on the
1814 * tracing registry side does not have this information.
1815 */
48842b30 1816
fc34caaa 1817 ua_chan->enabled = uchan->enabled;
7972aab2 1818 ua_chan->tracing_channel_id = uchan->id;
fc34caaa 1819
31746f93 1820 cds_list_for_each_entry(uctx, &uchan->ctx_list, list) {
bdf64013
JG
1821 struct ust_app_ctx *ua_ctx = alloc_ust_app_ctx(&uctx->ctx);
1822
55cc08a6
DG
1823 if (ua_ctx == NULL) {
1824 continue;
1825 }
bec39940
DG
1826 lttng_ht_node_init_ulong(&ua_ctx->node,
1827 (unsigned long) ua_ctx->ctx.ctx);
aa3514e9 1828 lttng_ht_add_ulong(ua_chan->ctx, &ua_ctx->node);
31746f93 1829 cds_list_add_tail(&ua_ctx->list, &ua_chan->ctx_list);
55cc08a6 1830 }
48842b30 1831
421cb601 1832 /* Copy all events from ltt ust channel to ust app channel */
bec39940 1833 cds_lfht_for_each_entry(uchan->events->ht, &iter.iter, uevent, node.node) {
18eace3b 1834 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
39c5a3a7 1835 uevent->filter, uevent->attr.loglevel, uevent->exclusion);
18eace3b 1836 if (ua_event == NULL) {
421cb601 1837 DBG2("UST event %s not found on shadow copy channel",
48842b30 1838 uevent->attr.name);
284d8f55 1839 ua_event = alloc_ust_app_event(uevent->attr.name, &uevent->attr);
48842b30 1840 if (ua_event == NULL) {
5b4a0ec0 1841 continue;
48842b30 1842 }
421cb601 1843 shadow_copy_event(ua_event, uevent);
d0b96690 1844 add_unique_ust_app_event(ua_chan, ua_event);
48842b30 1845 }
48842b30
DG
1846 }
1847
fc34caaa 1848 DBG3("UST app shadow copy of channel %s done", ua_chan->name);
48842b30
DG
1849}
1850
5b4a0ec0
DG
1851/*
1852 * Copy data between a UST app session and a regular LTT session.
1853 */
421cb601 1854static void shadow_copy_session(struct ust_app_session *ua_sess,
bec39940 1855 struct ltt_ust_session *usess, struct ust_app *app)
48842b30 1856{
bec39940
DG
1857 struct lttng_ht_node_str *ua_chan_node;
1858 struct lttng_ht_iter iter;
48842b30
DG
1859 struct ltt_ust_channel *uchan;
1860 struct ust_app_channel *ua_chan;
477d7741
MD
1861 time_t rawtime;
1862 struct tm *timeinfo;
1863 char datetime[16];
1864 int ret;
d7ba1388 1865 char tmp_shm_path[PATH_MAX];
477d7741
MD
1866
1867 /* Get date and time for unique app path */
1868 time(&rawtime);
1869 timeinfo = localtime(&rawtime);
1870 strftime(datetime, sizeof(datetime), "%Y%m%d-%H%M%S", timeinfo);
48842b30 1871
421cb601 1872 DBG2("Shadow copy of session handle %d", ua_sess->handle);
48842b30 1873
7972aab2
DG
1874 ua_sess->tracing_id = usess->id;
1875 ua_sess->id = get_next_session_id();
1876 ua_sess->uid = app->uid;
1877 ua_sess->gid = app->gid;
1878 ua_sess->euid = usess->uid;
1879 ua_sess->egid = usess->gid;
1880 ua_sess->buffer_type = usess->buffer_type;
1881 ua_sess->bits_per_long = app->bits_per_long;
6addfa37 1882
7972aab2 1883 /* There is only one consumer object per session possible. */
6addfa37 1884 consumer_output_get(usess->consumer);
7972aab2 1885 ua_sess->consumer = usess->consumer;
6addfa37 1886
2bba9e53 1887 ua_sess->output_traces = usess->output_traces;
ecc48a90 1888 ua_sess->live_timer_interval = usess->live_timer_interval;
84ad93e8
DG
1889 copy_channel_attr_to_ustctl(&ua_sess->metadata_attr,
1890 &usess->metadata_attr);
7972aab2
DG
1891
1892 switch (ua_sess->buffer_type) {
1893 case LTTNG_BUFFER_PER_PID:
1894 ret = snprintf(ua_sess->path, sizeof(ua_sess->path),
dec56f6c 1895 DEFAULT_UST_TRACE_PID_PATH "/%s-%d-%s", app->name, app->pid,
7972aab2
DG
1896 datetime);
1897 break;
1898 case LTTNG_BUFFER_PER_UID:
1899 ret = snprintf(ua_sess->path, sizeof(ua_sess->path),
1900 DEFAULT_UST_TRACE_UID_PATH, ua_sess->uid, app->bits_per_long);
1901 break;
1902 default:
1903 assert(0);
1904 goto error;
1905 }
477d7741
MD
1906 if (ret < 0) {
1907 PERROR("asprintf UST shadow copy session");
477d7741 1908 assert(0);
7972aab2 1909 goto error;
477d7741
MD
1910 }
1911
3d071855
MD
1912 strncpy(ua_sess->root_shm_path, usess->root_shm_path,
1913 sizeof(ua_sess->root_shm_path));
1914 ua_sess->root_shm_path[sizeof(ua_sess->root_shm_path) - 1] = '\0';
d7ba1388
MD
1915 strncpy(ua_sess->shm_path, usess->shm_path,
1916 sizeof(ua_sess->shm_path));
1917 ua_sess->shm_path[sizeof(ua_sess->shm_path) - 1] = '\0';
1918 if (ua_sess->shm_path[0]) {
1919 switch (ua_sess->buffer_type) {
1920 case LTTNG_BUFFER_PER_PID:
1921 ret = snprintf(tmp_shm_path, sizeof(tmp_shm_path),
1922 DEFAULT_UST_TRACE_PID_PATH "/%s-%d-%s",
1923 app->name, app->pid, datetime);
1924 break;
1925 case LTTNG_BUFFER_PER_UID:
1926 ret = snprintf(tmp_shm_path, sizeof(tmp_shm_path),
1927 DEFAULT_UST_TRACE_UID_PATH,
1928 app->uid, app->bits_per_long);
1929 break;
1930 default:
1931 assert(0);
1932 goto error;
1933 }
1934 if (ret < 0) {
1935 PERROR("sprintf UST shadow copy session");
1936 assert(0);
1937 goto error;
1938 }
1939 strncat(ua_sess->shm_path, tmp_shm_path,
1940 sizeof(ua_sess->shm_path) - strlen(ua_sess->shm_path) - 1);
1941 ua_sess->shm_path[sizeof(ua_sess->shm_path) - 1] = '\0';
1942 }
1943
48842b30 1944 /* Iterate over all channels in global domain. */
bec39940
DG
1945 cds_lfht_for_each_entry(usess->domain_global.channels->ht, &iter.iter,
1946 uchan, node.node) {
1947 struct lttng_ht_iter uiter;
ba767faf 1948
bec39940
DG
1949 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
1950 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
5b4a0ec0 1951 if (ua_chan_node != NULL) {
fc34caaa 1952 /* Session exist. Contiuing. */
5b4a0ec0
DG
1953 continue;
1954 }
421cb601 1955
5b4a0ec0
DG
1956 DBG2("Channel %s not found on shadow session copy, creating it",
1957 uchan->name);
fb83fe64
JD
1958 ua_chan = alloc_ust_app_channel(uchan->name, ua_sess,
1959 &uchan->attr);
5b4a0ec0 1960 if (ua_chan == NULL) {
fc34caaa 1961 /* malloc failed FIXME: Might want to do handle ENOMEM .. */
5b4a0ec0 1962 continue;
48842b30 1963 }
5b4a0ec0 1964 shadow_copy_channel(ua_chan, uchan);
ffe60014
DG
1965 /*
1966 * The concept of metadata channel does not exist on the tracing
1967 * registry side of the session daemon so this can only be a per CPU
1968 * channel and not metadata.
1969 */
1970 ua_chan->attr.type = LTTNG_UST_CHAN_PER_CPU;
1971
bec39940 1972 lttng_ht_add_unique_str(ua_sess->channels, &ua_chan->node);
48842b30 1973 }
6addfa37 1974 return;
7972aab2
DG
1975
1976error:
6addfa37 1977 consumer_output_put(ua_sess->consumer);
48842b30
DG
1978}
1979
78f0bacd
DG
1980/*
1981 * Lookup sesison wrapper.
1982 */
84cd17c6
MD
1983static
1984void __lookup_session_by_app(struct ltt_ust_session *usess,
bec39940 1985 struct ust_app *app, struct lttng_ht_iter *iter)
84cd17c6
MD
1986{
1987 /* Get right UST app session from app */
d9bf3ca4 1988 lttng_ht_lookup(app->sessions, &usess->id, iter);
84cd17c6
MD
1989}
1990
421cb601
DG
1991/*
1992 * Return ust app session from the app session hashtable using the UST session
a991f516 1993 * id.
421cb601 1994 */
48842b30
DG
1995static struct ust_app_session *lookup_session_by_app(
1996 struct ltt_ust_session *usess, struct ust_app *app)
1997{
bec39940 1998 struct lttng_ht_iter iter;
d9bf3ca4 1999 struct lttng_ht_node_u64 *node;
48842b30 2000
84cd17c6 2001 __lookup_session_by_app(usess, app, &iter);
d9bf3ca4 2002 node = lttng_ht_iter_get_node_u64(&iter);
48842b30
DG
2003 if (node == NULL) {
2004 goto error;
2005 }
2006
2007 return caa_container_of(node, struct ust_app_session, node);
2008
2009error:
2010 return NULL;
2011}
2012
7972aab2
DG
2013/*
2014 * Setup buffer registry per PID for the given session and application. If none
2015 * is found, a new one is created, added to the global registry and
2016 * initialized. If regp is valid, it's set with the newly created object.
2017 *
2018 * Return 0 on success or else a negative value.
2019 */
2020static int setup_buffer_reg_pid(struct ust_app_session *ua_sess,
2021 struct ust_app *app, struct buffer_reg_pid **regp)
2022{
2023 int ret = 0;
2024 struct buffer_reg_pid *reg_pid;
2025
2026 assert(ua_sess);
2027 assert(app);
2028
2029 rcu_read_lock();
2030
2031 reg_pid = buffer_reg_pid_find(ua_sess->id);
2032 if (!reg_pid) {
2033 /*
2034 * This is the create channel path meaning that if there is NO
2035 * registry available, we have to create one for this session.
2036 */
d7ba1388 2037 ret = buffer_reg_pid_create(ua_sess->id, &reg_pid,
3d071855 2038 ua_sess->root_shm_path, ua_sess->shm_path);
7972aab2
DG
2039 if (ret < 0) {
2040 goto error;
2041 }
7972aab2
DG
2042 } else {
2043 goto end;
2044 }
2045
2046 /* Initialize registry. */
2047 ret = ust_registry_session_init(&reg_pid->registry->reg.ust, app,
2048 app->bits_per_long, app->uint8_t_alignment,
2049 app->uint16_t_alignment, app->uint32_t_alignment,
af6142cf
MD
2050 app->uint64_t_alignment, app->long_alignment,
2051 app->byte_order, app->version.major,
3d071855
MD
2052 app->version.minor, reg_pid->root_shm_path,
2053 reg_pid->shm_path,
d7ba1388 2054 ua_sess->euid, ua_sess->egid);
7972aab2 2055 if (ret < 0) {
286c991a
MD
2056 /*
2057 * reg_pid->registry->reg.ust is NULL upon error, so we need to
2058 * destroy the buffer registry, because it is always expected
2059 * that if the buffer registry can be found, its ust registry is
2060 * non-NULL.
2061 */
2062 buffer_reg_pid_destroy(reg_pid);
7972aab2
DG
2063 goto error;
2064 }
2065
286c991a
MD
2066 buffer_reg_pid_add(reg_pid);
2067
7972aab2
DG
2068 DBG3("UST app buffer registry per PID created successfully");
2069
2070end:
2071 if (regp) {
2072 *regp = reg_pid;
2073 }
2074error:
2075 rcu_read_unlock();
2076 return ret;
2077}
2078
2079/*
2080 * Setup buffer registry per UID for the given session and application. If none
2081 * is found, a new one is created, added to the global registry and
2082 * initialized. If regp is valid, it's set with the newly created object.
2083 *
2084 * Return 0 on success or else a negative value.
2085 */
2086static int setup_buffer_reg_uid(struct ltt_ust_session *usess,
d7ba1388 2087 struct ust_app_session *ua_sess,
7972aab2
DG
2088 struct ust_app *app, struct buffer_reg_uid **regp)
2089{
2090 int ret = 0;
2091 struct buffer_reg_uid *reg_uid;
2092
2093 assert(usess);
2094 assert(app);
2095
2096 rcu_read_lock();
2097
2098 reg_uid = buffer_reg_uid_find(usess->id, app->bits_per_long, app->uid);
2099 if (!reg_uid) {
2100 /*
2101 * This is the create channel path meaning that if there is NO
2102 * registry available, we have to create one for this session.
2103 */
2104 ret = buffer_reg_uid_create(usess->id, app->bits_per_long, app->uid,
3d071855
MD
2105 LTTNG_DOMAIN_UST, &reg_uid,
2106 ua_sess->root_shm_path, ua_sess->shm_path);
7972aab2
DG
2107 if (ret < 0) {
2108 goto error;
2109 }
7972aab2
DG
2110 } else {
2111 goto end;
2112 }
2113
2114 /* Initialize registry. */
af6142cf 2115 ret = ust_registry_session_init(&reg_uid->registry->reg.ust, NULL,
7972aab2
DG
2116 app->bits_per_long, app->uint8_t_alignment,
2117 app->uint16_t_alignment, app->uint32_t_alignment,
af6142cf
MD
2118 app->uint64_t_alignment, app->long_alignment,
2119 app->byte_order, app->version.major,
3d071855
MD
2120 app->version.minor, reg_uid->root_shm_path,
2121 reg_uid->shm_path, usess->uid, usess->gid);
7972aab2 2122 if (ret < 0) {
286c991a
MD
2123 /*
2124 * reg_uid->registry->reg.ust is NULL upon error, so we need to
2125 * destroy the buffer registry, because it is always expected
2126 * that if the buffer registry can be found, its ust registry is
2127 * non-NULL.
2128 */
2129 buffer_reg_uid_destroy(reg_uid, NULL);
7972aab2
DG
2130 goto error;
2131 }
2132 /* Add node to teardown list of the session. */
2133 cds_list_add(&reg_uid->lnode, &usess->buffer_reg_uid_list);
2134
286c991a 2135 buffer_reg_uid_add(reg_uid);
7972aab2 2136
286c991a 2137 DBG3("UST app buffer registry per UID created successfully");
7972aab2
DG
2138end:
2139 if (regp) {
2140 *regp = reg_uid;
2141 }
2142error:
2143 rcu_read_unlock();
2144 return ret;
2145}
2146
421cb601 2147/*
3d8ca23b 2148 * Create a session on the tracer side for the given app.
421cb601 2149 *
3d8ca23b
DG
2150 * On success, ua_sess_ptr is populated with the session pointer or else left
2151 * untouched. If the session was created, is_created is set to 1. On error,
2152 * it's left untouched. Note that ua_sess_ptr is mandatory but is_created can
2153 * be NULL.
2154 *
2155 * Returns 0 on success or else a negative code which is either -ENOMEM or
2156 * -ENOTCONN which is the default code if the ustctl_create_session fails.
421cb601 2157 */
3d8ca23b
DG
2158static int create_ust_app_session(struct ltt_ust_session *usess,
2159 struct ust_app *app, struct ust_app_session **ua_sess_ptr,
2160 int *is_created)
421cb601 2161{
3d8ca23b 2162 int ret, created = 0;
421cb601
DG
2163 struct ust_app_session *ua_sess;
2164
3d8ca23b
DG
2165 assert(usess);
2166 assert(app);
2167 assert(ua_sess_ptr);
2168
840cb59c 2169 health_code_update();
86acf0da 2170
421cb601
DG
2171 ua_sess = lookup_session_by_app(usess, app);
2172 if (ua_sess == NULL) {
d9bf3ca4 2173 DBG2("UST app pid: %d session id %" PRIu64 " not found, creating it",
852d0037 2174 app->pid, usess->id);
d0b96690 2175 ua_sess = alloc_ust_app_session(app);
421cb601
DG
2176 if (ua_sess == NULL) {
2177 /* Only malloc can failed so something is really wrong */
3d8ca23b
DG
2178 ret = -ENOMEM;
2179 goto error;
421cb601 2180 }
477d7741 2181 shadow_copy_session(ua_sess, usess, app);
3d8ca23b 2182 created = 1;
421cb601
DG
2183 }
2184
7972aab2
DG
2185 switch (usess->buffer_type) {
2186 case LTTNG_BUFFER_PER_PID:
2187 /* Init local registry. */
2188 ret = setup_buffer_reg_pid(ua_sess, app, NULL);
421cb601 2189 if (ret < 0) {
e64207cf 2190 delete_ust_app_session(-1, ua_sess, app);
7972aab2
DG
2191 goto error;
2192 }
2193 break;
2194 case LTTNG_BUFFER_PER_UID:
2195 /* Look for a global registry. If none exists, create one. */
d7ba1388 2196 ret = setup_buffer_reg_uid(usess, ua_sess, app, NULL);
7972aab2 2197 if (ret < 0) {
e64207cf 2198 delete_ust_app_session(-1, ua_sess, app);
7972aab2
DG
2199 goto error;
2200 }
2201 break;
2202 default:
2203 assert(0);
2204 ret = -EINVAL;
2205 goto error;
2206 }
2207
2208 health_code_update();
2209
2210 if (ua_sess->handle == -1) {
fb45065e 2211 pthread_mutex_lock(&app->sock_lock);
7972aab2 2212 ret = ustctl_create_session(app->sock);
fb45065e 2213 pthread_mutex_unlock(&app->sock_lock);
7972aab2
DG
2214 if (ret < 0) {
2215 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
2216 ERR("Creating session for app pid %d with ret %d",
ffe60014
DG
2217 app->pid, ret);
2218 } else {
2219 DBG("UST app creating session failed. Application is dead");
3757b385
DG
2220 /*
2221 * This is normal behavior, an application can die during the
2222 * creation process. Don't report an error so the execution can
2223 * continue normally. This will get flagged ENOTCONN and the
2224 * caller will handle it.
2225 */
2226 ret = 0;
ffe60014 2227 }
d0b96690 2228 delete_ust_app_session(-1, ua_sess, app);
3d8ca23b
DG
2229 if (ret != -ENOMEM) {
2230 /*
2231 * Tracer is probably gone or got an internal error so let's
2232 * behave like it will soon unregister or not usable.
2233 */
2234 ret = -ENOTCONN;
2235 }
2236 goto error;
421cb601
DG
2237 }
2238
7972aab2
DG
2239 ua_sess->handle = ret;
2240
2241 /* Add ust app session to app's HT */
d9bf3ca4
MD
2242 lttng_ht_node_init_u64(&ua_sess->node,
2243 ua_sess->tracing_id);
2244 lttng_ht_add_unique_u64(app->sessions, &ua_sess->node);
10b56aef
MD
2245 lttng_ht_node_init_ulong(&ua_sess->ust_objd_node, ua_sess->handle);
2246 lttng_ht_add_unique_ulong(app->ust_sessions_objd,
2247 &ua_sess->ust_objd_node);
7972aab2
DG
2248
2249 DBG2("UST app session created successfully with handle %d", ret);
2250 }
2251
2252 *ua_sess_ptr = ua_sess;
2253 if (is_created) {
2254 *is_created = created;
2255 }
2256
2257 /* Everything went well. */
2258 ret = 0;
2259
2260error:
2261 health_code_update();
2262 return ret;
2263}
2264
6a6b2068
JG
2265/*
2266 * Match function for a hash table lookup of ust_app_ctx.
2267 *
2268 * It matches an ust app context based on the context type and, in the case
2269 * of perf counters, their name.
2270 */
2271static int ht_match_ust_app_ctx(struct cds_lfht_node *node, const void *_key)
2272{
2273 struct ust_app_ctx *ctx;
bdf64013 2274 const struct lttng_ust_context_attr *key;
6a6b2068
JG
2275
2276 assert(node);
2277 assert(_key);
2278
2279 ctx = caa_container_of(node, struct ust_app_ctx, node.node);
2280 key = _key;
2281
2282 /* Context type */
2283 if (ctx->ctx.ctx != key->ctx) {
2284 goto no_match;
2285 }
2286
bdf64013
JG
2287 switch(key->ctx) {
2288 case LTTNG_UST_CONTEXT_PERF_THREAD_COUNTER:
6a6b2068 2289 if (strncmp(key->u.perf_counter.name,
bdf64013
JG
2290 ctx->ctx.u.perf_counter.name,
2291 sizeof(key->u.perf_counter.name))) {
2292 goto no_match;
2293 }
2294 break;
2295 case LTTNG_UST_CONTEXT_APP_CONTEXT:
2296 if (strcmp(key->u.app_ctx.provider_name,
2297 ctx->ctx.u.app_ctx.provider_name) ||
2298 strcmp(key->u.app_ctx.ctx_name,
2299 ctx->ctx.u.app_ctx.ctx_name)) {
6a6b2068
JG
2300 goto no_match;
2301 }
bdf64013
JG
2302 break;
2303 default:
2304 break;
6a6b2068
JG
2305 }
2306
2307 /* Match. */
2308 return 1;
2309
2310no_match:
2311 return 0;
2312}
2313
2314/*
2315 * Lookup for an ust app context from an lttng_ust_context.
2316 *
be184a0f 2317 * Must be called while holding RCU read side lock.
6a6b2068
JG
2318 * Return an ust_app_ctx object or NULL on error.
2319 */
2320static
2321struct ust_app_ctx *find_ust_app_context(struct lttng_ht *ht,
bdf64013 2322 struct lttng_ust_context_attr *uctx)
6a6b2068
JG
2323{
2324 struct lttng_ht_iter iter;
2325 struct lttng_ht_node_ulong *node;
2326 struct ust_app_ctx *app_ctx = NULL;
2327
2328 assert(uctx);
2329 assert(ht);
2330
2331 /* Lookup using the lttng_ust_context_type and a custom match fct. */
2332 cds_lfht_lookup(ht->ht, ht->hash_fct((void *) uctx->ctx, lttng_ht_seed),
2333 ht_match_ust_app_ctx, uctx, &iter.iter);
2334 node = lttng_ht_iter_get_node_ulong(&iter);
2335 if (!node) {
2336 goto end;
2337 }
2338
2339 app_ctx = caa_container_of(node, struct ust_app_ctx, node);
2340
2341end:
2342 return app_ctx;
2343}
2344
7972aab2
DG
2345/*
2346 * Create a context for the channel on the tracer.
2347 *
2348 * Called with UST app session lock held and a RCU read side lock.
2349 */
2350static
2351int create_ust_app_channel_context(struct ust_app_session *ua_sess,
bdf64013
JG
2352 struct ust_app_channel *ua_chan,
2353 struct lttng_ust_context_attr *uctx,
7972aab2
DG
2354 struct ust_app *app)
2355{
2356 int ret = 0;
7972aab2
DG
2357 struct ust_app_ctx *ua_ctx;
2358
2359 DBG2("UST app adding context to channel %s", ua_chan->name);
2360
6a6b2068
JG
2361 ua_ctx = find_ust_app_context(ua_chan->ctx, uctx);
2362 if (ua_ctx) {
7972aab2
DG
2363 ret = -EEXIST;
2364 goto error;
2365 }
2366
2367 ua_ctx = alloc_ust_app_ctx(uctx);
2368 if (ua_ctx == NULL) {
2369 /* malloc failed */
2370 ret = -1;
2371 goto error;
2372 }
2373
2374 lttng_ht_node_init_ulong(&ua_ctx->node, (unsigned long) ua_ctx->ctx.ctx);
aa3514e9 2375 lttng_ht_add_ulong(ua_chan->ctx, &ua_ctx->node);
31746f93 2376 cds_list_add_tail(&ua_ctx->list, &ua_chan->ctx_list);
7972aab2
DG
2377
2378 ret = create_ust_channel_context(ua_chan, ua_ctx, app);
2379 if (ret < 0) {
2380 goto error;
2381 }
2382
2383error:
2384 return ret;
2385}
2386
2387/*
2388 * Enable on the tracer side a ust app event for the session and channel.
2389 *
2390 * Called with UST app session lock held.
2391 */
2392static
2393int enable_ust_app_event(struct ust_app_session *ua_sess,
2394 struct ust_app_event *ua_event, struct ust_app *app)
2395{
2396 int ret;
2397
2398 ret = enable_ust_event(app, ua_sess, ua_event);
2399 if (ret < 0) {
2400 goto error;
2401 }
2402
2403 ua_event->enabled = 1;
2404
2405error:
2406 return ret;
2407}
2408
2409/*
2410 * Disable on the tracer side a ust app event for the session and channel.
2411 */
2412static int disable_ust_app_event(struct ust_app_session *ua_sess,
2413 struct ust_app_event *ua_event, struct ust_app *app)
2414{
2415 int ret;
2416
2417 ret = disable_ust_event(app, ua_sess, ua_event);
2418 if (ret < 0) {
2419 goto error;
2420 }
2421
2422 ua_event->enabled = 0;
2423
2424error:
2425 return ret;
2426}
2427
2428/*
2429 * Lookup ust app channel for session and disable it on the tracer side.
2430 */
2431static
2432int disable_ust_app_channel(struct ust_app_session *ua_sess,
2433 struct ust_app_channel *ua_chan, struct ust_app *app)
2434{
2435 int ret;
2436
2437 ret = disable_ust_channel(app, ua_sess, ua_chan);
2438 if (ret < 0) {
2439 goto error;
2440 }
2441
2442 ua_chan->enabled = 0;
2443
2444error:
2445 return ret;
2446}
2447
2448/*
2449 * Lookup ust app channel for session and enable it on the tracer side. This
2450 * MUST be called with a RCU read side lock acquired.
2451 */
2452static int enable_ust_app_channel(struct ust_app_session *ua_sess,
2453 struct ltt_ust_channel *uchan, struct ust_app *app)
2454{
2455 int ret = 0;
2456 struct lttng_ht_iter iter;
2457 struct lttng_ht_node_str *ua_chan_node;
2458 struct ust_app_channel *ua_chan;
2459
2460 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &iter);
2461 ua_chan_node = lttng_ht_iter_get_node_str(&iter);
2462 if (ua_chan_node == NULL) {
d9bf3ca4 2463 DBG2("Unable to find channel %s in ust session id %" PRIu64,
7972aab2
DG
2464 uchan->name, ua_sess->tracing_id);
2465 goto error;
2466 }
2467
2468 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
2469
2470 ret = enable_ust_channel(app, ua_sess, ua_chan);
2471 if (ret < 0) {
2472 goto error;
2473 }
2474
2475error:
2476 return ret;
2477}
2478
2479/*
2480 * Ask the consumer to create a channel and get it if successful.
2481 *
5e98072f
JR
2482 * Called with UST app session lock held.
2483 *
7972aab2
DG
2484 * Return 0 on success or else a negative value.
2485 */
2486static int do_consumer_create_channel(struct ltt_ust_session *usess,
2487 struct ust_app_session *ua_sess, struct ust_app_channel *ua_chan,
2488 int bitness, struct ust_registry_session *registry)
2489{
2490 int ret;
2491 unsigned int nb_fd = 0;
2492 struct consumer_socket *socket;
2493
2494 assert(usess);
2495 assert(ua_sess);
2496 assert(ua_chan);
2497 assert(registry);
2498
2499 rcu_read_lock();
2500 health_code_update();
2501
2502 /* Get the right consumer socket for the application. */
2503 socket = consumer_find_socket_by_bitness(bitness, usess->consumer);
2504 if (!socket) {
2505 ret = -EINVAL;
2506 goto error;
2507 }
2508
2509 health_code_update();
2510
2511 /* Need one fd for the channel. */
2512 ret = lttng_fd_get(LTTNG_FD_APPS, 1);
2513 if (ret < 0) {
2514 ERR("Exhausted number of available FD upon create channel");
2515 goto error;
2516 }
2517
2518 /*
2519 * Ask consumer to create channel. The consumer will return the number of
2520 * stream we have to expect.
2521 */
2522 ret = ust_consumer_ask_channel(ua_sess, ua_chan, usess->consumer, socket,
2523 registry);
2524 if (ret < 0) {
2525 goto error_ask;
2526 }
2527
2528 /*
2529 * Compute the number of fd needed before receiving them. It must be 2 per
2530 * stream (2 being the default value here).
2531 */
2532 nb_fd = DEFAULT_UST_STREAM_FD_NUM * ua_chan->expected_stream_count;
2533
2534 /* Reserve the amount of file descriptor we need. */
2535 ret = lttng_fd_get(LTTNG_FD_APPS, nb_fd);
2536 if (ret < 0) {
2537 ERR("Exhausted number of available FD upon create channel");
2538 goto error_fd_get_stream;
2539 }
2540
2541 health_code_update();
2542
2543 /*
2544 * Now get the channel from the consumer. This call wil populate the stream
2545 * list of that channel and set the ust objects.
2546 */
d9078d0c
DG
2547 if (usess->consumer->enabled) {
2548 ret = ust_consumer_get_channel(socket, ua_chan);
2549 if (ret < 0) {
2550 goto error_destroy;
2551 }
7972aab2
DG
2552 }
2553
2554 rcu_read_unlock();
2555 return 0;
2556
2557error_destroy:
2558 lttng_fd_put(LTTNG_FD_APPS, nb_fd);
2559error_fd_get_stream:
2560 /*
2561 * Initiate a destroy channel on the consumer since we had an error
2562 * handling it on our side. The return value is of no importance since we
2563 * already have a ret value set by the previous error that we need to
2564 * return.
2565 */
2566 (void) ust_consumer_destroy_channel(socket, ua_chan);
2567error_ask:
2568 lttng_fd_put(LTTNG_FD_APPS, 1);
2569error:
2570 health_code_update();
2571 rcu_read_unlock();
2572 return ret;
2573}
2574
2575/*
2576 * Duplicate the ust data object of the ust app stream and save it in the
2577 * buffer registry stream.
2578 *
2579 * Return 0 on success or else a negative value.
2580 */
2581static int duplicate_stream_object(struct buffer_reg_stream *reg_stream,
2582 struct ust_app_stream *stream)
2583{
2584 int ret;
2585
2586 assert(reg_stream);
2587 assert(stream);
2588
2589 /* Reserve the amount of file descriptor we need. */
2590 ret = lttng_fd_get(LTTNG_FD_APPS, 2);
2591 if (ret < 0) {
2592 ERR("Exhausted number of available FD upon duplicate stream");
2593 goto error;
2594 }
2595
2596 /* Duplicate object for stream once the original is in the registry. */
2597 ret = ustctl_duplicate_ust_object_data(&stream->obj,
2598 reg_stream->obj.ust);
2599 if (ret < 0) {
2600 ERR("Duplicate stream obj from %p to %p failed with ret %d",
2601 reg_stream->obj.ust, stream->obj, ret);
2602 lttng_fd_put(LTTNG_FD_APPS, 2);
2603 goto error;
2604 }
2605 stream->handle = stream->obj->handle;
2606
2607error:
2608 return ret;
2609}
2610
2611/*
2612 * Duplicate the ust data object of the ust app. channel and save it in the
2613 * buffer registry channel.
2614 *
2615 * Return 0 on success or else a negative value.
2616 */
2617static int duplicate_channel_object(struct buffer_reg_channel *reg_chan,
2618 struct ust_app_channel *ua_chan)
2619{
2620 int ret;
2621
2622 assert(reg_chan);
2623 assert(ua_chan);
2624
2625 /* Need two fds for the channel. */
2626 ret = lttng_fd_get(LTTNG_FD_APPS, 1);
2627 if (ret < 0) {
2628 ERR("Exhausted number of available FD upon duplicate channel");
2629 goto error_fd_get;
2630 }
2631
2632 /* Duplicate object for stream once the original is in the registry. */
2633 ret = ustctl_duplicate_ust_object_data(&ua_chan->obj, reg_chan->obj.ust);
2634 if (ret < 0) {
2635 ERR("Duplicate channel obj from %p to %p failed with ret: %d",
2636 reg_chan->obj.ust, ua_chan->obj, ret);
2637 goto error;
2638 }
2639 ua_chan->handle = ua_chan->obj->handle;
2640
2641 return 0;
2642
2643error:
2644 lttng_fd_put(LTTNG_FD_APPS, 1);
2645error_fd_get:
2646 return ret;
2647}
2648
2649/*
2650 * For a given channel buffer registry, setup all streams of the given ust
2651 * application channel.
2652 *
2653 * Return 0 on success or else a negative value.
2654 */
2655static int setup_buffer_reg_streams(struct buffer_reg_channel *reg_chan,
fb45065e
MD
2656 struct ust_app_channel *ua_chan,
2657 struct ust_app *app)
7972aab2
DG
2658{
2659 int ret = 0;
2660 struct ust_app_stream *stream, *stmp;
2661
2662 assert(reg_chan);
2663 assert(ua_chan);
2664
2665 DBG2("UST app setup buffer registry stream");
2666
2667 /* Send all streams to application. */
2668 cds_list_for_each_entry_safe(stream, stmp, &ua_chan->streams.head, list) {
2669 struct buffer_reg_stream *reg_stream;
2670
2671 ret = buffer_reg_stream_create(&reg_stream);
2672 if (ret < 0) {
2673 goto error;
2674 }
2675
2676 /*
2677 * Keep original pointer and nullify it in the stream so the delete
2678 * stream call does not release the object.
2679 */
2680 reg_stream->obj.ust = stream->obj;
2681 stream->obj = NULL;
2682 buffer_reg_stream_add(reg_stream, reg_chan);
421cb601 2683
7972aab2
DG
2684 /* We don't need the streams anymore. */
2685 cds_list_del(&stream->list);
fb45065e 2686 delete_ust_app_stream(-1, stream, app);
7972aab2 2687 }
421cb601 2688
7972aab2
DG
2689error:
2690 return ret;
2691}
2692
2693/*
2694 * Create a buffer registry channel for the given session registry and
2695 * application channel object. If regp pointer is valid, it's set with the
2696 * created object. Important, the created object is NOT added to the session
2697 * registry hash table.
2698 *
2699 * Return 0 on success else a negative value.
2700 */
2701static int create_buffer_reg_channel(struct buffer_reg_session *reg_sess,
2702 struct ust_app_channel *ua_chan, struct buffer_reg_channel **regp)
2703{
2704 int ret;
2705 struct buffer_reg_channel *reg_chan = NULL;
2706
2707 assert(reg_sess);
2708 assert(ua_chan);
2709
2710 DBG2("UST app creating buffer registry channel for %s", ua_chan->name);
2711
2712 /* Create buffer registry channel. */
2713 ret = buffer_reg_channel_create(ua_chan->tracing_channel_id, &reg_chan);
2714 if (ret < 0) {
2715 goto error_create;
421cb601 2716 }
7972aab2
DG
2717 assert(reg_chan);
2718 reg_chan->consumer_key = ua_chan->key;
8c924c7b 2719 reg_chan->subbuf_size = ua_chan->attr.subbuf_size;
d07ceecd 2720 reg_chan->num_subbuf = ua_chan->attr.num_subbuf;
421cb601 2721
7972aab2
DG
2722 /* Create and add a channel registry to session. */
2723 ret = ust_registry_channel_add(reg_sess->reg.ust,
2724 ua_chan->tracing_channel_id);
2725 if (ret < 0) {
2726 goto error;
d88aee68 2727 }
7972aab2 2728 buffer_reg_channel_add(reg_sess, reg_chan);
d88aee68 2729
7972aab2
DG
2730 if (regp) {
2731 *regp = reg_chan;
3d8ca23b 2732 }
d88aee68 2733
7972aab2 2734 return 0;
3d8ca23b
DG
2735
2736error:
7972aab2
DG
2737 /* Safe because the registry channel object was not added to any HT. */
2738 buffer_reg_channel_destroy(reg_chan, LTTNG_DOMAIN_UST);
2739error_create:
3d8ca23b 2740 return ret;
421cb601
DG
2741}
2742
55cc08a6 2743/*
7972aab2
DG
2744 * Setup buffer registry channel for the given session registry and application
2745 * channel object. If regp pointer is valid, it's set with the created object.
d0b96690 2746 *
7972aab2 2747 * Return 0 on success else a negative value.
55cc08a6 2748 */
7972aab2 2749static int setup_buffer_reg_channel(struct buffer_reg_session *reg_sess,
fb45065e
MD
2750 struct ust_app_channel *ua_chan, struct buffer_reg_channel *reg_chan,
2751 struct ust_app *app)
55cc08a6 2752{
7972aab2 2753 int ret;
55cc08a6 2754
7972aab2
DG
2755 assert(reg_sess);
2756 assert(reg_chan);
2757 assert(ua_chan);
2758 assert(ua_chan->obj);
55cc08a6 2759
7972aab2 2760 DBG2("UST app setup buffer registry channel for %s", ua_chan->name);
55cc08a6 2761
7972aab2 2762 /* Setup all streams for the registry. */
fb45065e 2763 ret = setup_buffer_reg_streams(reg_chan, ua_chan, app);
7972aab2 2764 if (ret < 0) {
55cc08a6
DG
2765 goto error;
2766 }
2767
7972aab2
DG
2768 reg_chan->obj.ust = ua_chan->obj;
2769 ua_chan->obj = NULL;
55cc08a6 2770
7972aab2 2771 return 0;
55cc08a6
DG
2772
2773error:
7972aab2
DG
2774 buffer_reg_channel_remove(reg_sess, reg_chan);
2775 buffer_reg_channel_destroy(reg_chan, LTTNG_DOMAIN_UST);
55cc08a6
DG
2776 return ret;
2777}
2778
edb67388 2779/*
7972aab2 2780 * Send buffer registry channel to the application.
d0b96690 2781 *
7972aab2 2782 * Return 0 on success else a negative value.
edb67388 2783 */
7972aab2
DG
2784static int send_channel_uid_to_ust(struct buffer_reg_channel *reg_chan,
2785 struct ust_app *app, struct ust_app_session *ua_sess,
2786 struct ust_app_channel *ua_chan)
edb67388
DG
2787{
2788 int ret;
7972aab2 2789 struct buffer_reg_stream *reg_stream;
edb67388 2790
7972aab2
DG
2791 assert(reg_chan);
2792 assert(app);
2793 assert(ua_sess);
2794 assert(ua_chan);
2795
2796 DBG("UST app sending buffer registry channel to ust sock %d", app->sock);
2797
2798 ret = duplicate_channel_object(reg_chan, ua_chan);
edb67388
DG
2799 if (ret < 0) {
2800 goto error;
2801 }
2802
7972aab2
DG
2803 /* Send channel to the application. */
2804 ret = ust_consumer_send_channel_to_ust(app, ua_sess, ua_chan);
a7169585
MD
2805 if (ret == -EPIPE || ret == -LTTNG_UST_ERR_EXITING) {
2806 ret = -ENOTCONN; /* Caused by app exiting. */
2807 goto error;
2808 } else if (ret < 0) {
7972aab2
DG
2809 goto error;
2810 }
2811
2812 health_code_update();
2813
2814 /* Send all streams to application. */
2815 pthread_mutex_lock(&reg_chan->stream_list_lock);
2816 cds_list_for_each_entry(reg_stream, &reg_chan->streams, lnode) {
2817 struct ust_app_stream stream;
2818
2819 ret = duplicate_stream_object(reg_stream, &stream);
2820 if (ret < 0) {
2821 goto error_stream_unlock;
2822 }
2823
2824 ret = ust_consumer_send_stream_to_ust(app, ua_chan, &stream);
2825 if (ret < 0) {
fb45065e 2826 (void) release_ust_app_stream(-1, &stream, app);
a7169585
MD
2827 if (ret == -EPIPE || ret == -LTTNG_UST_ERR_EXITING) {
2828 ret = -ENOTCONN; /* Caused by app exiting. */
a7169585 2829 }
7972aab2
DG
2830 goto error_stream_unlock;
2831 }
edb67388 2832
7972aab2
DG
2833 /*
2834 * The return value is not important here. This function will output an
2835 * error if needed.
2836 */
fb45065e 2837 (void) release_ust_app_stream(-1, &stream, app);
7972aab2
DG
2838 }
2839 ua_chan->is_sent = 1;
2840
2841error_stream_unlock:
2842 pthread_mutex_unlock(&reg_chan->stream_list_lock);
edb67388
DG
2843error:
2844 return ret;
2845}
2846
9730260e 2847/*
7972aab2
DG
2848 * Create and send to the application the created buffers with per UID buffers.
2849 *
2850 * Return 0 on success else a negative value.
9730260e 2851 */
7972aab2
DG
2852static int create_channel_per_uid(struct ust_app *app,
2853 struct ltt_ust_session *usess, struct ust_app_session *ua_sess,
2854 struct ust_app_channel *ua_chan)
9730260e
DG
2855{
2856 int ret;
7972aab2
DG
2857 struct buffer_reg_uid *reg_uid;
2858 struct buffer_reg_channel *reg_chan;
e9404c27 2859 bool created = false;
9730260e 2860
7972aab2
DG
2861 assert(app);
2862 assert(usess);
2863 assert(ua_sess);
2864 assert(ua_chan);
2865
2866 DBG("UST app creating channel %s with per UID buffers", ua_chan->name);
2867
2868 reg_uid = buffer_reg_uid_find(usess->id, app->bits_per_long, app->uid);
2869 /*
2870 * The session creation handles the creation of this global registry
2871 * object. If none can be find, there is a code flow problem or a
2872 * teardown race.
2873 */
2874 assert(reg_uid);
2875
2876 reg_chan = buffer_reg_channel_find(ua_chan->tracing_channel_id,
2877 reg_uid);
2878 if (!reg_chan) {
2879 /* Create the buffer registry channel object. */
2880 ret = create_buffer_reg_channel(reg_uid->registry, ua_chan, &reg_chan);
2881 if (ret < 0) {
f14256d6
MD
2882 ERR("Error creating the UST channel \"%s\" registry instance",
2883 ua_chan->name);
7972aab2
DG
2884 goto error;
2885 }
2886 assert(reg_chan);
2887
2888 /*
2889 * Create the buffers on the consumer side. This call populates the
2890 * ust app channel object with all streams and data object.
2891 */
2892 ret = do_consumer_create_channel(usess, ua_sess, ua_chan,
2893 app->bits_per_long, reg_uid->registry->reg.ust);
2894 if (ret < 0) {
f14256d6
MD
2895 ERR("Error creating UST channel \"%s\" on the consumer daemon",
2896 ua_chan->name);
2897
07d2ae95
DG
2898 /*
2899 * Let's remove the previously created buffer registry channel so
2900 * it's not visible anymore in the session registry.
2901 */
2902 ust_registry_channel_del_free(reg_uid->registry->reg.ust,
e9404c27 2903 ua_chan->tracing_channel_id, false);
07d2ae95
DG
2904 buffer_reg_channel_remove(reg_uid->registry, reg_chan);
2905 buffer_reg_channel_destroy(reg_chan, LTTNG_DOMAIN_UST);
7972aab2
DG
2906 goto error;
2907 }
2908
2909 /*
2910 * Setup the streams and add it to the session registry.
2911 */
fb45065e
MD
2912 ret = setup_buffer_reg_channel(reg_uid->registry,
2913 ua_chan, reg_chan, app);
7972aab2 2914 if (ret < 0) {
f14256d6
MD
2915 ERR("Error setting up UST channel \"%s\"",
2916 ua_chan->name);
7972aab2
DG
2917 goto error;
2918 }
e9404c27 2919 created = true;
7972aab2
DG
2920 }
2921
e9404c27
JG
2922 if (created) {
2923 enum lttng_error_code cmd_ret;
2924 struct ltt_session *session;
2925 uint64_t chan_reg_key;
2926 struct ust_registry_channel *chan_reg;
2927
2928 rcu_read_lock();
2929 chan_reg_key = ua_chan->tracing_channel_id;
2930
2931 pthread_mutex_lock(&reg_uid->registry->reg.ust->lock);
2932 chan_reg = ust_registry_channel_find(reg_uid->registry->reg.ust,
2933 chan_reg_key);
2934 assert(chan_reg);
2935 chan_reg->consumer_key = ua_chan->key;
2936 chan_reg = NULL;
2937 pthread_mutex_unlock(&reg_uid->registry->reg.ust->lock);
2938
2939 session = session_find_by_id(ua_sess->tracing_id);
2940 assert(session);
2941
2942 cmd_ret = notification_thread_command_add_channel(
2943 notification_thread_handle, session->name,
2944 ua_sess->euid, ua_sess->egid,
2945 ua_chan->name,
2946 ua_chan->key,
2947 LTTNG_DOMAIN_UST,
2948 ua_chan->attr.subbuf_size * ua_chan->attr.num_subbuf);
2949 rcu_read_unlock();
2950 if (cmd_ret != LTTNG_OK) {
2951 ret = - (int) cmd_ret;
2952 ERR("Failed to add channel to notification thread");
2953 goto error;
2954 }
2955 }
2956
8a06890a
JG
2957 /* Send buffers to the application. */
2958 ret = send_channel_uid_to_ust(reg_chan, app, ua_sess, ua_chan);
2959 if (ret < 0) {
2960 if (ret != -ENOTCONN) {
2961 ERR("Error sending channel to application");
2962 }
2963 goto error;
2964 }
2965
9730260e
DG
2966error:
2967 return ret;
2968}
2969
78f0bacd 2970/*
7972aab2
DG
2971 * Create and send to the application the created buffers with per PID buffers.
2972 *
5e98072f
JR
2973 * Called with UST app session lock held.
2974 *
7972aab2 2975 * Return 0 on success else a negative value.
78f0bacd 2976 */
7972aab2
DG
2977static int create_channel_per_pid(struct ust_app *app,
2978 struct ltt_ust_session *usess, struct ust_app_session *ua_sess,
2979 struct ust_app_channel *ua_chan)
78f0bacd 2980{
8535a6d9 2981 int ret;
7972aab2 2982 struct ust_registry_session *registry;
e9404c27
JG
2983 enum lttng_error_code cmd_ret;
2984 struct ltt_session *session;
2985 uint64_t chan_reg_key;
2986 struct ust_registry_channel *chan_reg;
78f0bacd 2987
7972aab2
DG
2988 assert(app);
2989 assert(usess);
2990 assert(ua_sess);
2991 assert(ua_chan);
2992
2993 DBG("UST app creating channel %s with per PID buffers", ua_chan->name);
2994
2995 rcu_read_lock();
2996
2997 registry = get_session_registry(ua_sess);
5e98072f 2998 /* The UST app session lock is held, registry shall not be null. */
7972aab2
DG
2999 assert(registry);
3000
3001 /* Create and add a new channel registry to session. */
3002 ret = ust_registry_channel_add(registry, ua_chan->key);
78f0bacd 3003 if (ret < 0) {
f14256d6
MD
3004 ERR("Error creating the UST channel \"%s\" registry instance",
3005 ua_chan->name);
78f0bacd
DG
3006 goto error;
3007 }
3008
7972aab2
DG
3009 /* Create and get channel on the consumer side. */
3010 ret = do_consumer_create_channel(usess, ua_sess, ua_chan,
3011 app->bits_per_long, registry);
3012 if (ret < 0) {
f14256d6
MD
3013 ERR("Error creating UST channel \"%s\" on the consumer daemon",
3014 ua_chan->name);
7972aab2
DG
3015 goto error;
3016 }
3017
3018 ret = send_channel_pid_to_ust(app, ua_sess, ua_chan);
3019 if (ret < 0) {
a7169585
MD
3020 if (ret != -ENOTCONN) {
3021 ERR("Error sending channel to application");
3022 }
7972aab2
DG
3023 goto error;
3024 }
8535a6d9 3025
e9404c27
JG
3026 session = session_find_by_id(ua_sess->tracing_id);
3027 assert(session);
3028
3029 chan_reg_key = ua_chan->key;
3030 pthread_mutex_lock(&registry->lock);
3031 chan_reg = ust_registry_channel_find(registry, chan_reg_key);
3032 assert(chan_reg);
3033 chan_reg->consumer_key = ua_chan->key;
3034 pthread_mutex_unlock(&registry->lock);
3035
3036 cmd_ret = notification_thread_command_add_channel(
3037 notification_thread_handle, session->name,
3038 ua_sess->euid, ua_sess->egid,
3039 ua_chan->name,
3040 ua_chan->key,
3041 LTTNG_DOMAIN_UST,
3042 ua_chan->attr.subbuf_size * ua_chan->attr.num_subbuf);
3043 if (cmd_ret != LTTNG_OK) {
3044 ret = - (int) cmd_ret;
3045 ERR("Failed to add channel to notification thread");
3046 goto error;
3047 }
3048
78f0bacd 3049error:
7972aab2 3050 rcu_read_unlock();
78f0bacd
DG
3051 return ret;
3052}
3053
3054/*
7972aab2
DG
3055 * From an already allocated ust app channel, create the channel buffers if
3056 * need and send it to the application. This MUST be called with a RCU read
3057 * side lock acquired.
3058 *
5e98072f
JR
3059 * Called with UST app session lock held.
3060 *
a7169585
MD
3061 * Return 0 on success or else a negative value. Returns -ENOTCONN if
3062 * the application exited concurrently.
78f0bacd 3063 */
7972aab2
DG
3064static int do_create_channel(struct ust_app *app,
3065 struct ltt_ust_session *usess, struct ust_app_session *ua_sess,
3066 struct ust_app_channel *ua_chan)
78f0bacd 3067{
7972aab2 3068 int ret;
78f0bacd 3069
7972aab2
DG
3070 assert(app);
3071 assert(usess);
3072 assert(ua_sess);
3073 assert(ua_chan);
3074
3075 /* Handle buffer type before sending the channel to the application. */
3076 switch (usess->buffer_type) {
3077 case LTTNG_BUFFER_PER_UID:
3078 {
3079 ret = create_channel_per_uid(app, usess, ua_sess, ua_chan);
3080 if (ret < 0) {
3081 goto error;
3082 }
3083 break;
3084 }
3085 case LTTNG_BUFFER_PER_PID:
3086 {
3087 ret = create_channel_per_pid(app, usess, ua_sess, ua_chan);
3088 if (ret < 0) {
3089 goto error;
3090 }
3091 break;
3092 }
3093 default:
3094 assert(0);
3095 ret = -EINVAL;
78f0bacd
DG
3096 goto error;
3097 }
3098
7972aab2
DG
3099 /* Initialize ust objd object using the received handle and add it. */
3100 lttng_ht_node_init_ulong(&ua_chan->ust_objd_node, ua_chan->handle);
3101 lttng_ht_add_unique_ulong(app->ust_objd, &ua_chan->ust_objd_node);
78f0bacd 3102
7972aab2
DG
3103 /* If channel is not enabled, disable it on the tracer */
3104 if (!ua_chan->enabled) {
3105 ret = disable_ust_channel(app, ua_sess, ua_chan);
3106 if (ret < 0) {
3107 goto error;
3108 }
78f0bacd
DG
3109 }
3110
3111error:
3112 return ret;
3113}
3114
284d8f55 3115/*
4d710ac2
DG
3116 * Create UST app channel and create it on the tracer. Set ua_chanp of the
3117 * newly created channel if not NULL.
d0b96690 3118 *
36b588ed 3119 * Called with UST app session lock and RCU read-side lock held.
7972aab2 3120 *
a7169585
MD
3121 * Return 0 on success or else a negative value. Returns -ENOTCONN if
3122 * the application exited concurrently.
284d8f55 3123 */
4d710ac2
DG
3124static int create_ust_app_channel(struct ust_app_session *ua_sess,
3125 struct ltt_ust_channel *uchan, struct ust_app *app,
7972aab2 3126 enum lttng_ust_chan_type type, struct ltt_ust_session *usess,
4d710ac2 3127 struct ust_app_channel **ua_chanp)
5b4a0ec0
DG
3128{
3129 int ret = 0;
bec39940
DG
3130 struct lttng_ht_iter iter;
3131 struct lttng_ht_node_str *ua_chan_node;
5b4a0ec0
DG
3132 struct ust_app_channel *ua_chan;
3133
3134 /* Lookup channel in the ust app session */
bec39940
DG
3135 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &iter);
3136 ua_chan_node = lttng_ht_iter_get_node_str(&iter);
fc34caaa 3137 if (ua_chan_node != NULL) {
5b4a0ec0 3138 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
fc34caaa 3139 goto end;
5b4a0ec0
DG
3140 }
3141
d0b96690 3142 ua_chan = alloc_ust_app_channel(uchan->name, ua_sess, &uchan->attr);
fc34caaa
DG
3143 if (ua_chan == NULL) {
3144 /* Only malloc can fail here */
4d710ac2 3145 ret = -ENOMEM;
094d1690 3146 goto error_alloc;
fc34caaa
DG
3147 }
3148 shadow_copy_channel(ua_chan, uchan);
3149
ffe60014
DG
3150 /* Set channel type. */
3151 ua_chan->attr.type = type;
3152
7972aab2 3153 ret = do_create_channel(app, usess, ua_sess, ua_chan);
5b4a0ec0
DG
3154 if (ret < 0) {
3155 goto error;
3156 }
3157
fc34caaa 3158 DBG2("UST app create channel %s for PID %d completed", ua_chan->name,
852d0037 3159 app->pid);
fc34caaa 3160
d0b96690
DG
3161 /* Only add the channel if successful on the tracer side. */
3162 lttng_ht_add_unique_str(ua_sess->channels, &ua_chan->node);
fc34caaa 3163end:
4d710ac2
DG
3164 if (ua_chanp) {
3165 *ua_chanp = ua_chan;
3166 }
3167
3168 /* Everything went well. */
3169 return 0;
5b4a0ec0
DG
3170
3171error:
d0b96690 3172 delete_ust_app_channel(ua_chan->is_sent ? app->sock : -1, ua_chan, app);
094d1690 3173error_alloc:
4d710ac2 3174 return ret;
5b4a0ec0
DG
3175}
3176
3177/*
3178 * Create UST app event and create it on the tracer side.
d0b96690
DG
3179 *
3180 * Called with ust app session mutex held.
5b4a0ec0 3181 */
edb67388
DG
3182static
3183int create_ust_app_event(struct ust_app_session *ua_sess,
3184 struct ust_app_channel *ua_chan, struct ltt_ust_event *uevent,
3185 struct ust_app *app)
284d8f55 3186{
edb67388 3187 int ret = 0;
5b4a0ec0 3188 struct ust_app_event *ua_event;
284d8f55 3189
5b4a0ec0 3190 /* Get event node */
18eace3b 3191 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
39c5a3a7 3192 uevent->filter, uevent->attr.loglevel, uevent->exclusion);
18eace3b 3193 if (ua_event != NULL) {
fc34caaa 3194 ret = -EEXIST;
edb67388
DG
3195 goto end;
3196 }
5b4a0ec0 3197
edb67388
DG
3198 /* Does not exist so create one */
3199 ua_event = alloc_ust_app_event(uevent->attr.name, &uevent->attr);
3200 if (ua_event == NULL) {
3201 /* Only malloc can failed so something is really wrong */
3202 ret = -ENOMEM;
fc34caaa 3203 goto end;
5b4a0ec0 3204 }
edb67388 3205 shadow_copy_event(ua_event, uevent);
5b4a0ec0 3206
edb67388 3207 /* Create it on the tracer side */
5b4a0ec0 3208 ret = create_ust_event(app, ua_sess, ua_chan, ua_event);
284d8f55 3209 if (ret < 0) {
fc34caaa 3210 /* Not found previously means that it does not exist on the tracer */
76f66f63 3211 assert(ret != -LTTNG_UST_ERR_EXIST);
284d8f55
DG
3212 goto error;
3213 }
3214
d0b96690 3215 add_unique_ust_app_event(ua_chan, ua_event);
284d8f55 3216
fc34caaa 3217 DBG2("UST app create event %s for PID %d completed", ua_event->name,
852d0037 3218 app->pid);
7f79d3a1 3219
edb67388 3220end:
fc34caaa
DG
3221 return ret;
3222
5b4a0ec0 3223error:
fc34caaa 3224 /* Valid. Calling here is already in a read side lock */
fb45065e 3225 delete_ust_app_event(-1, ua_event, app);
edb67388 3226 return ret;
5b4a0ec0
DG
3227}
3228
3229/*
3230 * Create UST metadata and open it on the tracer side.
d0b96690 3231 *
7972aab2 3232 * Called with UST app session lock held and RCU read side lock.
5b4a0ec0
DG
3233 */
3234static int create_ust_app_metadata(struct ust_app_session *ua_sess,
ad7a9107 3235 struct ust_app *app, struct consumer_output *consumer)
5b4a0ec0
DG
3236{
3237 int ret = 0;
ffe60014 3238 struct ust_app_channel *metadata;
d88aee68 3239 struct consumer_socket *socket;
7972aab2 3240 struct ust_registry_session *registry;
5b4a0ec0 3241
ffe60014
DG
3242 assert(ua_sess);
3243 assert(app);
d88aee68 3244 assert(consumer);
5b4a0ec0 3245
7972aab2 3246 registry = get_session_registry(ua_sess);
5e98072f 3247 /* The UST app session is held registry shall not be null. */
7972aab2
DG
3248 assert(registry);
3249
ce34fcd0
MD
3250 pthread_mutex_lock(&registry->lock);
3251
1b532a60
DG
3252 /* Metadata already exists for this registry or it was closed previously */
3253 if (registry->metadata_key || registry->metadata_closed) {
7972aab2
DG
3254 ret = 0;
3255 goto error;
5b4a0ec0
DG
3256 }
3257
ffe60014 3258 /* Allocate UST metadata */
d0b96690 3259 metadata = alloc_ust_app_channel(DEFAULT_METADATA_NAME, ua_sess, NULL);
ffe60014
DG
3260 if (!metadata) {
3261 /* malloc() failed */
3262 ret = -ENOMEM;
3263 goto error;
3264 }
5b4a0ec0 3265
ad7a9107 3266 memcpy(&metadata->attr, &ua_sess->metadata_attr, sizeof(metadata->attr));
5b4a0ec0 3267
7972aab2
DG
3268 /* Need one fd for the channel. */
3269 ret = lttng_fd_get(LTTNG_FD_APPS, 1);
3270 if (ret < 0) {
3271 ERR("Exhausted number of available FD upon create metadata");
3272 goto error;
3273 }
3274
4dc3dfc5
DG
3275 /* Get the right consumer socket for the application. */
3276 socket = consumer_find_socket_by_bitness(app->bits_per_long, consumer);
3277 if (!socket) {
3278 ret = -EINVAL;
3279 goto error_consumer;
3280 }
3281
331744e3
JD
3282 /*
3283 * Keep metadata key so we can identify it on the consumer side. Assign it
3284 * to the registry *before* we ask the consumer so we avoid the race of the
3285 * consumer requesting the metadata and the ask_channel call on our side
3286 * did not returned yet.
3287 */
3288 registry->metadata_key = metadata->key;
3289
d88aee68
DG
3290 /*
3291 * Ask the metadata channel creation to the consumer. The metadata object
3292 * will be created by the consumer and kept their. However, the stream is
3293 * never added or monitored until we do a first push metadata to the
3294 * consumer.
3295 */
7972aab2
DG
3296 ret = ust_consumer_ask_channel(ua_sess, metadata, consumer, socket,
3297 registry);
d88aee68 3298 if (ret < 0) {
f2a444f1
DG
3299 /* Nullify the metadata key so we don't try to close it later on. */
3300 registry->metadata_key = 0;
d88aee68
DG
3301 goto error_consumer;
3302 }
3303
3304 /*
3305 * The setup command will make the metadata stream be sent to the relayd,
3306 * if applicable, and the thread managing the metadatas. This is important
3307 * because after this point, if an error occurs, the only way the stream
3308 * can be deleted is to be monitored in the consumer.
3309 */
7972aab2 3310 ret = consumer_setup_metadata(socket, metadata->key);
ffe60014 3311 if (ret < 0) {
f2a444f1
DG
3312 /* Nullify the metadata key so we don't try to close it later on. */
3313 registry->metadata_key = 0;
d88aee68 3314 goto error_consumer;
5b4a0ec0
DG
3315 }
3316
7972aab2
DG
3317 DBG2("UST metadata with key %" PRIu64 " created for app pid %d",
3318 metadata->key, app->pid);
5b4a0ec0 3319
d88aee68 3320error_consumer:
b80f0b6c 3321 lttng_fd_put(LTTNG_FD_APPS, 1);
d88aee68 3322 delete_ust_app_channel(-1, metadata, app);
5b4a0ec0 3323error:
ce34fcd0 3324 pthread_mutex_unlock(&registry->lock);
ffe60014 3325 return ret;
5b4a0ec0
DG
3326}
3327
5b4a0ec0 3328/*
d88aee68
DG
3329 * Return ust app pointer or NULL if not found. RCU read side lock MUST be
3330 * acquired before calling this function.
5b4a0ec0
DG
3331 */
3332struct ust_app *ust_app_find_by_pid(pid_t pid)
3333{
d88aee68 3334 struct ust_app *app = NULL;
bec39940
DG
3335 struct lttng_ht_node_ulong *node;
3336 struct lttng_ht_iter iter;
5b4a0ec0 3337
bec39940
DG
3338 lttng_ht_lookup(ust_app_ht, (void *)((unsigned long) pid), &iter);
3339 node = lttng_ht_iter_get_node_ulong(&iter);
5b4a0ec0
DG
3340 if (node == NULL) {
3341 DBG2("UST app no found with pid %d", pid);
3342 goto error;
3343 }
5b4a0ec0
DG
3344
3345 DBG2("Found UST app by pid %d", pid);
3346
d88aee68 3347 app = caa_container_of(node, struct ust_app, pid_n);
5b4a0ec0
DG
3348
3349error:
d88aee68 3350 return app;
5b4a0ec0
DG
3351}
3352
d88aee68
DG
3353/*
3354 * Allocate and init an UST app object using the registration information and
3355 * the command socket. This is called when the command socket connects to the
3356 * session daemon.
3357 *
3358 * The object is returned on success or else NULL.
3359 */
d0b96690 3360struct ust_app *ust_app_create(struct ust_register_msg *msg, int sock)
5b4a0ec0 3361{
d0b96690
DG
3362 struct ust_app *lta = NULL;
3363
3364 assert(msg);
3365 assert(sock >= 0);
3366
3367 DBG3("UST app creating application for socket %d", sock);
5b4a0ec0 3368
173af62f
DG
3369 if ((msg->bits_per_long == 64 &&
3370 (uatomic_read(&ust_consumerd64_fd) == -EINVAL))
3371 || (msg->bits_per_long == 32 &&
3372 (uatomic_read(&ust_consumerd32_fd) == -EINVAL))) {
f943b0fb 3373 ERR("Registration failed: application \"%s\" (pid: %d) has "
d0b96690
DG
3374 "%d-bit long, but no consumerd for this size is available.\n",
3375 msg->name, msg->pid, msg->bits_per_long);
3376 goto error;
3f2c5fcc 3377 }
d0b96690 3378
5b4a0ec0
DG
3379 lta = zmalloc(sizeof(struct ust_app));
3380 if (lta == NULL) {
3381 PERROR("malloc");
d0b96690 3382 goto error;
5b4a0ec0
DG
3383 }
3384
3385 lta->ppid = msg->ppid;
3386 lta->uid = msg->uid;
3387 lta->gid = msg->gid;
d0b96690 3388
7753dea8 3389 lta->bits_per_long = msg->bits_per_long;
d0b96690
DG
3390 lta->uint8_t_alignment = msg->uint8_t_alignment;
3391 lta->uint16_t_alignment = msg->uint16_t_alignment;
3392 lta->uint32_t_alignment = msg->uint32_t_alignment;
3393 lta->uint64_t_alignment = msg->uint64_t_alignment;
3394 lta->long_alignment = msg->long_alignment;
3395 lta->byte_order = msg->byte_order;
3396
5b4a0ec0
DG
3397 lta->v_major = msg->major;
3398 lta->v_minor = msg->minor;
d9bf3ca4 3399 lta->sessions = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
d0b96690 3400 lta->ust_objd = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
10b56aef 3401 lta->ust_sessions_objd = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
d0b96690 3402 lta->notify_sock = -1;
d88aee68
DG
3403
3404 /* Copy name and make sure it's NULL terminated. */
3405 strncpy(lta->name, msg->name, sizeof(lta->name));
3406 lta->name[UST_APP_PROCNAME_LEN] = '\0';
3407
3408 /*
3409 * Before this can be called, when receiving the registration information,
3410 * the application compatibility is checked. So, at this point, the
3411 * application can work with this session daemon.
3412 */
d0b96690 3413 lta->compatible = 1;
5b4a0ec0 3414
852d0037 3415 lta->pid = msg->pid;
d0b96690 3416 lttng_ht_node_init_ulong(&lta->pid_n, (unsigned long) lta->pid);
852d0037 3417 lta->sock = sock;
fb45065e 3418 pthread_mutex_init(&lta->sock_lock, NULL);
d0b96690 3419 lttng_ht_node_init_ulong(&lta->sock_n, (unsigned long) lta->sock);
5b4a0ec0 3420
d42f20df 3421 CDS_INIT_LIST_HEAD(&lta->teardown_head);
d0b96690
DG
3422error:
3423 return lta;
3424}
3425
d88aee68
DG
3426/*
3427 * For a given application object, add it to every hash table.
3428 */
d0b96690
DG
3429void ust_app_add(struct ust_app *app)
3430{
3431 assert(app);
3432 assert(app->notify_sock >= 0);
3433
5b4a0ec0 3434 rcu_read_lock();
852d0037
DG
3435
3436 /*
3437 * On a re-registration, we want to kick out the previous registration of
3438 * that pid
3439 */
d0b96690 3440 lttng_ht_add_replace_ulong(ust_app_ht, &app->pid_n);
852d0037
DG
3441
3442 /*
3443 * The socket _should_ be unique until _we_ call close. So, a add_unique
3444 * for the ust_app_ht_by_sock is used which asserts fail if the entry was
3445 * already in the table.
3446 */
d0b96690 3447 lttng_ht_add_unique_ulong(ust_app_ht_by_sock, &app->sock_n);
852d0037 3448
d0b96690
DG
3449 /* Add application to the notify socket hash table. */
3450 lttng_ht_node_init_ulong(&app->notify_sock_n, app->notify_sock);
3451 lttng_ht_add_unique_ulong(ust_app_ht_by_notify_sock, &app->notify_sock_n);
5b4a0ec0 3452
d0b96690 3453 DBG("App registered with pid:%d ppid:%d uid:%d gid:%d sock:%d name:%s "
d88aee68
DG
3454 "notify_sock:%d (version %d.%d)", app->pid, app->ppid, app->uid,
3455 app->gid, app->sock, app->name, app->notify_sock, app->v_major,
3456 app->v_minor);
5b4a0ec0 3457
d0b96690
DG
3458 rcu_read_unlock();
3459}
3460
d88aee68
DG
3461/*
3462 * Set the application version into the object.
3463 *
3464 * Return 0 on success else a negative value either an errno code or a
3465 * LTTng-UST error code.
3466 */
d0b96690
DG
3467int ust_app_version(struct ust_app *app)
3468{
d88aee68
DG
3469 int ret;
3470
d0b96690 3471 assert(app);
d88aee68 3472
fb45065e 3473 pthread_mutex_lock(&app->sock_lock);
d88aee68 3474 ret = ustctl_tracer_version(app->sock, &app->version);
fb45065e 3475 pthread_mutex_unlock(&app->sock_lock);
d88aee68
DG
3476 if (ret < 0) {
3477 if (ret != -LTTNG_UST_ERR_EXITING && ret != -EPIPE) {
5368d366 3478 ERR("UST app %d version failed with ret %d", app->sock, ret);
d88aee68 3479 } else {
5368d366 3480 DBG3("UST app %d version failed. Application is dead", app->sock);
d88aee68
DG
3481 }
3482 }
3483
3484 return ret;
5b4a0ec0
DG
3485}
3486
3487/*
3488 * Unregister app by removing it from the global traceable app list and freeing
3489 * the data struct.
3490 *
3491 * The socket is already closed at this point so no close to sock.
3492 */
3493void ust_app_unregister(int sock)
3494{
3495 struct ust_app *lta;
bec39940 3496 struct lttng_ht_node_ulong *node;
c4b88406 3497 struct lttng_ht_iter ust_app_sock_iter;
bec39940 3498 struct lttng_ht_iter iter;
d42f20df 3499 struct ust_app_session *ua_sess;
525b0740 3500 int ret;
5b4a0ec0
DG
3501
3502 rcu_read_lock();
886459c6 3503
5b4a0ec0 3504 /* Get the node reference for a call_rcu */
c4b88406
MD
3505 lttng_ht_lookup(ust_app_ht_by_sock, (void *)((unsigned long) sock), &ust_app_sock_iter);
3506 node = lttng_ht_iter_get_node_ulong(&ust_app_sock_iter);
d0b96690 3507 assert(node);
284d8f55 3508
852d0037 3509 lta = caa_container_of(node, struct ust_app, sock_n);
852d0037
DG
3510 DBG("PID %d unregistering with sock %d", lta->pid, sock);
3511
d88aee68 3512 /*
ce34fcd0
MD
3513 * For per-PID buffers, perform "push metadata" and flush all
3514 * application streams before removing app from hash tables,
3515 * ensuring proper behavior of data_pending check.
c4b88406 3516 * Remove sessions so they are not visible during deletion.
d88aee68 3517 */
d42f20df
DG
3518 cds_lfht_for_each_entry(lta->sessions->ht, &iter.iter, ua_sess,
3519 node.node) {
7972aab2
DG
3520 struct ust_registry_session *registry;
3521
d42f20df
DG
3522 ret = lttng_ht_del(lta->sessions, &iter);
3523 if (ret) {
3524 /* The session was already removed so scheduled for teardown. */
3525 continue;
3526 }
3527
ce34fcd0
MD
3528 if (ua_sess->buffer_type == LTTNG_BUFFER_PER_PID) {
3529 (void) ust_app_flush_app_session(lta, ua_sess);
3530 }
c4b88406 3531
d42f20df
DG
3532 /*
3533 * Add session to list for teardown. This is safe since at this point we
3534 * are the only one using this list.
3535 */
d88aee68
DG
3536 pthread_mutex_lock(&ua_sess->lock);
3537
b161602a
MD
3538 if (ua_sess->deleted) {
3539 pthread_mutex_unlock(&ua_sess->lock);
3540 continue;
3541 }
3542
d88aee68
DG
3543 /*
3544 * Normally, this is done in the delete session process which is
3545 * executed in the call rcu below. However, upon registration we can't
3546 * afford to wait for the grace period before pushing data or else the
3547 * data pending feature can race between the unregistration and stop
3548 * command where the data pending command is sent *before* the grace
3549 * period ended.
3550 *
3551 * The close metadata below nullifies the metadata pointer in the
3552 * session so the delete session will NOT push/close a second time.
3553 */
7972aab2 3554 registry = get_session_registry(ua_sess);
ce34fcd0 3555 if (registry) {
7972aab2
DG
3556 /* Push metadata for application before freeing the application. */
3557 (void) push_metadata(registry, ua_sess->consumer);
3558
3559 /*
3560 * Don't ask to close metadata for global per UID buffers. Close
1b532a60
DG
3561 * metadata only on destroy trace session in this case. Also, the
3562 * previous push metadata could have flag the metadata registry to
3563 * close so don't send a close command if closed.
7972aab2 3564 */
ce34fcd0 3565 if (ua_sess->buffer_type != LTTNG_BUFFER_PER_UID) {
7972aab2
DG
3566 /* And ask to close it for this session registry. */
3567 (void) close_metadata(registry, ua_sess->consumer);
3568 }
3569 }
d42f20df 3570 cds_list_add(&ua_sess->teardown_node, &lta->teardown_head);
c4b88406 3571
d88aee68 3572 pthread_mutex_unlock(&ua_sess->lock);
d42f20df
DG
3573 }
3574
c4b88406
MD
3575 /* Remove application from PID hash table */
3576 ret = lttng_ht_del(ust_app_ht_by_sock, &ust_app_sock_iter);
3577 assert(!ret);
3578
3579 /*
3580 * Remove application from notify hash table. The thread handling the
3581 * notify socket could have deleted the node so ignore on error because
3582 * either way it's valid. The close of that socket is handled by the other
3583 * thread.
3584 */
3585 iter.iter.node = &lta->notify_sock_n.node;
3586 (void) lttng_ht_del(ust_app_ht_by_notify_sock, &iter);
3587
3588 /*
3589 * Ignore return value since the node might have been removed before by an
3590 * add replace during app registration because the PID can be reassigned by
3591 * the OS.
3592 */
3593 iter.iter.node = &lta->pid_n.node;
3594 ret = lttng_ht_del(ust_app_ht, &iter);
3595 if (ret) {
3596 DBG3("Unregister app by PID %d failed. This can happen on pid reuse",
3597 lta->pid);
3598 }
3599
852d0037
DG
3600 /* Free memory */
3601 call_rcu(&lta->pid_n.head, delete_ust_app_rcu);
3602
5b4a0ec0
DG
3603 rcu_read_unlock();
3604 return;
284d8f55
DG
3605}
3606
5b4a0ec0
DG
3607/*
3608 * Fill events array with all events name of all registered apps.
3609 */
3610int ust_app_list_events(struct lttng_event **events)
421cb601 3611{
5b4a0ec0
DG
3612 int ret, handle;
3613 size_t nbmem, count = 0;
bec39940 3614 struct lttng_ht_iter iter;
5b4a0ec0 3615 struct ust_app *app;
c617c0c6 3616 struct lttng_event *tmp_event;
421cb601 3617
5b4a0ec0 3618 nbmem = UST_APP_EVENT_LIST_SIZE;
c617c0c6
MD
3619 tmp_event = zmalloc(nbmem * sizeof(struct lttng_event));
3620 if (tmp_event == NULL) {
5b4a0ec0
DG
3621 PERROR("zmalloc ust app events");
3622 ret = -ENOMEM;
421cb601
DG
3623 goto error;
3624 }
3625
5b4a0ec0 3626 rcu_read_lock();
421cb601 3627
852d0037 3628 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
90eaa0d2 3629 struct lttng_ust_tracepoint_iter uiter;
ac3bd9c0 3630
840cb59c 3631 health_code_update();
86acf0da 3632
e0c7ec2b
DG
3633 if (!app->compatible) {
3634 /*
3635 * TODO: In time, we should notice the caller of this error by
3636 * telling him that this is a version error.
3637 */
3638 continue;
3639 }
fb45065e 3640 pthread_mutex_lock(&app->sock_lock);
852d0037 3641 handle = ustctl_tracepoint_list(app->sock);
5b4a0ec0 3642 if (handle < 0) {
ffe60014
DG
3643 if (handle != -EPIPE && handle != -LTTNG_UST_ERR_EXITING) {
3644 ERR("UST app list events getting handle failed for app pid %d",
3645 app->pid);
3646 }
fb45065e 3647 pthread_mutex_unlock(&app->sock_lock);
5b4a0ec0
DG
3648 continue;
3649 }
421cb601 3650
852d0037 3651 while ((ret = ustctl_tracepoint_list_get(app->sock, handle,
fb54cdbf 3652 &uiter)) != -LTTNG_UST_ERR_NOENT) {
ffe60014
DG
3653 /* Handle ustctl error. */
3654 if (ret < 0) {
fb45065e
MD
3655 int release_ret;
3656
a2ba1ab0 3657 if (ret != -LTTNG_UST_ERR_EXITING && ret != -EPIPE) {
ffe60014
DG
3658 ERR("UST app tp list get failed for app %d with ret %d",
3659 app->sock, ret);
3660 } else {
3661 DBG3("UST app tp list get failed. Application is dead");
3757b385
DG
3662 /*
3663 * This is normal behavior, an application can die during the
3664 * creation process. Don't report an error so the execution can
3665 * continue normally. Continue normal execution.
3666 */
3667 break;
ffe60014 3668 }
98f595d4 3669 free(tmp_event);
fb45065e 3670 release_ret = ustctl_release_handle(app->sock, handle);
68313703
JG
3671 if (release_ret < 0 &&
3672 release_ret != -LTTNG_UST_ERR_EXITING &&
3673 release_ret != -EPIPE) {
fb45065e
MD
3674 ERR("Error releasing app handle for app %d with ret %d", app->sock, release_ret);
3675 }
3676 pthread_mutex_unlock(&app->sock_lock);
ffe60014
DG
3677 goto rcu_error;
3678 }
3679
840cb59c 3680 health_code_update();
815564d8 3681 if (count >= nbmem) {
d7b3776f 3682 /* In case the realloc fails, we free the memory */
53efb85a
MD
3683 struct lttng_event *new_tmp_event;
3684 size_t new_nbmem;
3685
3686 new_nbmem = nbmem << 1;
3687 DBG2("Reallocating event list from %zu to %zu entries",
3688 nbmem, new_nbmem);
3689 new_tmp_event = realloc(tmp_event,
3690 new_nbmem * sizeof(struct lttng_event));
3691 if (new_tmp_event == NULL) {
fb45065e
MD
3692 int release_ret;
3693
5b4a0ec0 3694 PERROR("realloc ust app events");
c617c0c6 3695 free(tmp_event);
5b4a0ec0 3696 ret = -ENOMEM;
fb45065e 3697 release_ret = ustctl_release_handle(app->sock, handle);
68313703
JG
3698 if (release_ret < 0 &&
3699 release_ret != -LTTNG_UST_ERR_EXITING &&
3700 release_ret != -EPIPE) {
fb45065e
MD
3701 ERR("Error releasing app handle for app %d with ret %d", app->sock, release_ret);
3702 }
3703 pthread_mutex_unlock(&app->sock_lock);
5b4a0ec0
DG
3704 goto rcu_error;
3705 }
53efb85a
MD
3706 /* Zero the new memory */
3707 memset(new_tmp_event + nbmem, 0,
3708 (new_nbmem - nbmem) * sizeof(struct lttng_event));
3709 nbmem = new_nbmem;
3710 tmp_event = new_tmp_event;
5b4a0ec0 3711 }
c617c0c6
MD
3712 memcpy(tmp_event[count].name, uiter.name, LTTNG_UST_SYM_NAME_LEN);
3713 tmp_event[count].loglevel = uiter.loglevel;
3714 tmp_event[count].type = (enum lttng_event_type) LTTNG_UST_TRACEPOINT;
3715 tmp_event[count].pid = app->pid;
3716 tmp_event[count].enabled = -1;
5b4a0ec0 3717 count++;
421cb601 3718 }
fb45065e
MD
3719 ret = ustctl_release_handle(app->sock, handle);
3720 pthread_mutex_unlock(&app->sock_lock);
68313703 3721 if (ret < 0 && ret != -LTTNG_UST_ERR_EXITING && ret != -EPIPE) {
fb45065e
MD
3722 ERR("Error releasing app handle for app %d with ret %d", app->sock, ret);
3723 }
421cb601
DG
3724 }
3725
5b4a0ec0 3726 ret = count;
c617c0c6 3727 *events = tmp_event;
421cb601 3728
5b4a0ec0 3729 DBG2("UST app list events done (%zu events)", count);
421cb601 3730
5b4a0ec0
DG
3731rcu_error:
3732 rcu_read_unlock();
421cb601 3733error:
840cb59c 3734 health_code_update();
5b4a0ec0 3735 return ret;
421cb601
DG
3736}
3737
f37d259d
MD
3738/*
3739 * Fill events array with all events name of all registered apps.
3740 */
3741int ust_app_list_event_fields(struct lttng_event_field **fields)
3742{
3743 int ret, handle;
3744 size_t nbmem, count = 0;
3745 struct lttng_ht_iter iter;
3746 struct ust_app *app;
c617c0c6 3747 struct lttng_event_field *tmp_event;
f37d259d
MD
3748
3749 nbmem = UST_APP_EVENT_LIST_SIZE;
c617c0c6
MD
3750 tmp_event = zmalloc(nbmem * sizeof(struct lttng_event_field));
3751 if (tmp_event == NULL) {
f37d259d
MD
3752 PERROR("zmalloc ust app event fields");
3753 ret = -ENOMEM;
3754 goto error;
3755 }
3756
3757 rcu_read_lock();
3758
3759 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
3760 struct lttng_ust_field_iter uiter;
3761
840cb59c 3762 health_code_update();
86acf0da 3763
f37d259d
MD
3764 if (!app->compatible) {
3765 /*
3766 * TODO: In time, we should notice the caller of this error by
3767 * telling him that this is a version error.
3768 */
3769 continue;
3770 }
fb45065e 3771 pthread_mutex_lock(&app->sock_lock);
f37d259d
MD
3772 handle = ustctl_tracepoint_field_list(app->sock);
3773 if (handle < 0) {
ffe60014
DG
3774 if (handle != -EPIPE && handle != -LTTNG_UST_ERR_EXITING) {
3775 ERR("UST app list field getting handle failed for app pid %d",
3776 app->pid);
3777 }
fb45065e 3778 pthread_mutex_unlock(&app->sock_lock);
f37d259d
MD
3779 continue;
3780 }
3781
3782 while ((ret = ustctl_tracepoint_field_list_get(app->sock, handle,
fb54cdbf 3783 &uiter)) != -LTTNG_UST_ERR_NOENT) {
ffe60014
DG
3784 /* Handle ustctl error. */
3785 if (ret < 0) {
fb45065e
MD
3786 int release_ret;
3787
a2ba1ab0 3788 if (ret != -LTTNG_UST_ERR_EXITING && ret != -EPIPE) {
ffe60014
DG
3789 ERR("UST app tp list field failed for app %d with ret %d",
3790 app->sock, ret);
3791 } else {
3792 DBG3("UST app tp list field failed. Application is dead");
3757b385
DG
3793 /*
3794 * This is normal behavior, an application can die during the
3795 * creation process. Don't report an error so the execution can
98f595d4 3796 * continue normally. Reset list and count for next app.
3757b385
DG
3797 */
3798 break;
ffe60014 3799 }
98f595d4 3800 free(tmp_event);
fb45065e
MD
3801 release_ret = ustctl_release_handle(app->sock, handle);
3802 pthread_mutex_unlock(&app->sock_lock);
68313703
JG
3803 if (release_ret < 0 &&
3804 release_ret != -LTTNG_UST_ERR_EXITING &&
3805 release_ret != -EPIPE) {
fb45065e
MD
3806 ERR("Error releasing app handle for app %d with ret %d", app->sock, release_ret);
3807 }
ffe60014
DG
3808 goto rcu_error;
3809 }
3810
840cb59c 3811 health_code_update();
f37d259d 3812 if (count >= nbmem) {
d7b3776f 3813 /* In case the realloc fails, we free the memory */
53efb85a
MD
3814 struct lttng_event_field *new_tmp_event;
3815 size_t new_nbmem;
3816
3817 new_nbmem = nbmem << 1;
3818 DBG2("Reallocating event field list from %zu to %zu entries",
3819 nbmem, new_nbmem);
3820 new_tmp_event = realloc(tmp_event,
3821 new_nbmem * sizeof(struct lttng_event_field));
3822 if (new_tmp_event == NULL) {
fb45065e
MD
3823 int release_ret;
3824
f37d259d 3825 PERROR("realloc ust app event fields");
c617c0c6 3826 free(tmp_event);
f37d259d 3827 ret = -ENOMEM;
fb45065e
MD
3828 release_ret = ustctl_release_handle(app->sock, handle);
3829 pthread_mutex_unlock(&app->sock_lock);
68313703
JG
3830 if (release_ret &&
3831 release_ret != -LTTNG_UST_ERR_EXITING &&
3832 release_ret != -EPIPE) {
fb45065e
MD
3833 ERR("Error releasing app handle for app %d with ret %d", app->sock, release_ret);
3834 }
f37d259d
MD
3835 goto rcu_error;
3836 }
53efb85a
MD
3837 /* Zero the new memory */
3838 memset(new_tmp_event + nbmem, 0,
3839 (new_nbmem - nbmem) * sizeof(struct lttng_event_field));
3840 nbmem = new_nbmem;
3841 tmp_event = new_tmp_event;
f37d259d 3842 }
f37d259d 3843
c617c0c6 3844 memcpy(tmp_event[count].field_name, uiter.field_name, LTTNG_UST_SYM_NAME_LEN);
2e84128e
DG
3845 /* Mapping between these enums matches 1 to 1. */
3846 tmp_event[count].type = (enum lttng_event_field_type) uiter.type;
c617c0c6 3847 tmp_event[count].nowrite = uiter.nowrite;
f37d259d 3848
c617c0c6
MD
3849 memcpy(tmp_event[count].event.name, uiter.event_name, LTTNG_UST_SYM_NAME_LEN);
3850 tmp_event[count].event.loglevel = uiter.loglevel;
2e84128e 3851 tmp_event[count].event.type = LTTNG_EVENT_TRACEPOINT;
c617c0c6
MD
3852 tmp_event[count].event.pid = app->pid;
3853 tmp_event[count].event.enabled = -1;
f37d259d
MD
3854 count++;
3855 }
fb45065e
MD
3856 ret = ustctl_release_handle(app->sock, handle);
3857 pthread_mutex_unlock(&app->sock_lock);
68313703
JG
3858 if (ret < 0 &&
3859 ret != -LTTNG_UST_ERR_EXITING &&
3860 ret != -EPIPE) {
fb45065e
MD
3861 ERR("Error releasing app handle for app %d with ret %d", app->sock, ret);
3862 }
f37d259d
MD
3863 }
3864
3865 ret = count;
c617c0c6 3866 *fields = tmp_event;
f37d259d
MD
3867
3868 DBG2("UST app list event fields done (%zu events)", count);
3869
3870rcu_error:
3871 rcu_read_unlock();
3872error:
840cb59c 3873 health_code_update();
f37d259d
MD
3874 return ret;
3875}
3876
5b4a0ec0
DG
3877/*
3878 * Free and clean all traceable apps of the global list.
36b588ed
MD
3879 *
3880 * Should _NOT_ be called with RCU read-side lock held.
5b4a0ec0
DG
3881 */
3882void ust_app_clean_list(void)
421cb601 3883{
5b4a0ec0 3884 int ret;
659ed79f 3885 struct ust_app *app;
bec39940 3886 struct lttng_ht_iter iter;
421cb601 3887
5b4a0ec0 3888 DBG2("UST app cleaning registered apps hash table");
421cb601 3889
5b4a0ec0 3890 rcu_read_lock();
421cb601 3891
f1b711c4
MD
3892 if (ust_app_ht) {
3893 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
3894 ret = lttng_ht_del(ust_app_ht, &iter);
3895 assert(!ret);
3896 call_rcu(&app->pid_n.head, delete_ust_app_rcu);
3897 }
421cb601
DG
3898 }
3899
852d0037 3900 /* Cleanup socket hash table */
f1b711c4
MD
3901 if (ust_app_ht_by_sock) {
3902 cds_lfht_for_each_entry(ust_app_ht_by_sock->ht, &iter.iter, app,
3903 sock_n.node) {
3904 ret = lttng_ht_del(ust_app_ht_by_sock, &iter);
3905 assert(!ret);
3906 }
bec39940 3907 }
852d0037 3908
d88aee68 3909 /* Cleanup notify socket hash table */
f1b711c4
MD
3910 if (ust_app_ht_by_notify_sock) {
3911 cds_lfht_for_each_entry(ust_app_ht_by_notify_sock->ht, &iter.iter, app,
3912 notify_sock_n.node) {
3913 ret = lttng_ht_del(ust_app_ht_by_notify_sock, &iter);
3914 assert(!ret);
3915 }
d88aee68 3916 }
36b588ed 3917 rcu_read_unlock();
d88aee68 3918
bec39940 3919 /* Destroy is done only when the ht is empty */
f1b711c4
MD
3920 if (ust_app_ht) {
3921 ht_cleanup_push(ust_app_ht);
3922 }
3923 if (ust_app_ht_by_sock) {
3924 ht_cleanup_push(ust_app_ht_by_sock);
3925 }
3926 if (ust_app_ht_by_notify_sock) {
3927 ht_cleanup_push(ust_app_ht_by_notify_sock);
3928 }
5b4a0ec0
DG
3929}
3930
3931/*
3932 * Init UST app hash table.
3933 */
57703f6e 3934int ust_app_ht_alloc(void)
5b4a0ec0 3935{
bec39940 3936 ust_app_ht = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
57703f6e
MD
3937 if (!ust_app_ht) {
3938 return -1;
3939 }
852d0037 3940 ust_app_ht_by_sock = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
57703f6e
MD
3941 if (!ust_app_ht_by_sock) {
3942 return -1;
3943 }
d0b96690 3944 ust_app_ht_by_notify_sock = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
57703f6e
MD
3945 if (!ust_app_ht_by_notify_sock) {
3946 return -1;
3947 }
3948 return 0;
421cb601
DG
3949}
3950
78f0bacd
DG
3951/*
3952 * For a specific UST session, disable the channel for all registered apps.
3953 */
35a9059d 3954int ust_app_disable_channel_glb(struct ltt_ust_session *usess,
78f0bacd
DG
3955 struct ltt_ust_channel *uchan)
3956{
3957 int ret = 0;
bec39940
DG
3958 struct lttng_ht_iter iter;
3959 struct lttng_ht_node_str *ua_chan_node;
78f0bacd
DG
3960 struct ust_app *app;
3961 struct ust_app_session *ua_sess;
8535a6d9 3962 struct ust_app_channel *ua_chan;
78f0bacd
DG
3963
3964 if (usess == NULL || uchan == NULL) {
3965 ERR("Disabling UST global channel with NULL values");
3966 ret = -1;
3967 goto error;
3968 }
3969
d9bf3ca4 3970 DBG2("UST app disabling channel %s from global domain for session id %" PRIu64,
a991f516 3971 uchan->name, usess->id);
78f0bacd
DG
3972
3973 rcu_read_lock();
3974
3975 /* For every registered applications */
852d0037 3976 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
bec39940 3977 struct lttng_ht_iter uiter;
e0c7ec2b
DG
3978 if (!app->compatible) {
3979 /*
3980 * TODO: In time, we should notice the caller of this error by
3981 * telling him that this is a version error.
3982 */
3983 continue;
3984 }
78f0bacd
DG
3985 ua_sess = lookup_session_by_app(usess, app);
3986 if (ua_sess == NULL) {
3987 continue;
3988 }
3989
8535a6d9 3990 /* Get channel */
bec39940
DG
3991 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
3992 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
8535a6d9
DG
3993 /* If the session if found for the app, the channel must be there */
3994 assert(ua_chan_node);
3995
3996 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
3997 /* The channel must not be already disabled */
3998 assert(ua_chan->enabled == 1);
3999
4000 /* Disable channel onto application */
4001 ret = disable_ust_app_channel(ua_sess, ua_chan, app);
78f0bacd
DG
4002 if (ret < 0) {
4003 /* XXX: We might want to report this error at some point... */
4004 continue;
4005 }
4006 }
4007
4008 rcu_read_unlock();
4009
4010error:
4011 return ret;
4012}
4013
4014/*
4015 * For a specific UST session, enable the channel for all registered apps.
4016 */
35a9059d 4017int ust_app_enable_channel_glb(struct ltt_ust_session *usess,
78f0bacd
DG
4018 struct ltt_ust_channel *uchan)
4019{
4020 int ret = 0;
bec39940 4021 struct lttng_ht_iter iter;
78f0bacd
DG
4022 struct ust_app *app;
4023 struct ust_app_session *ua_sess;
4024
4025 if (usess == NULL || uchan == NULL) {
4026 ERR("Adding UST global channel to NULL values");
4027 ret = -1;
4028 goto error;
4029 }
4030
d9bf3ca4 4031 DBG2("UST app enabling channel %s to global domain for session id %" PRIu64,
a991f516 4032 uchan->name, usess->id);
78f0bacd
DG
4033
4034 rcu_read_lock();
4035
4036 /* For every registered applications */
852d0037 4037 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
4038 if (!app->compatible) {
4039 /*
4040 * TODO: In time, we should notice the caller of this error by
4041 * telling him that this is a version error.
4042 */
4043 continue;
4044 }
78f0bacd
DG
4045 ua_sess = lookup_session_by_app(usess, app);
4046 if (ua_sess == NULL) {
4047 continue;
4048 }
4049
4050 /* Enable channel onto application */
4051 ret = enable_ust_app_channel(ua_sess, uchan, app);
4052 if (ret < 0) {
4053 /* XXX: We might want to report this error at some point... */
4054 continue;
4055 }
4056 }
4057
4058 rcu_read_unlock();
4059
4060error:
4061 return ret;
4062}
4063
b0a40d28
DG
4064/*
4065 * Disable an event in a channel and for a specific session.
4066 */
35a9059d
DG
4067int ust_app_disable_event_glb(struct ltt_ust_session *usess,
4068 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent)
b0a40d28
DG
4069{
4070 int ret = 0;
bec39940 4071 struct lttng_ht_iter iter, uiter;
700c5a9d 4072 struct lttng_ht_node_str *ua_chan_node;
b0a40d28
DG
4073 struct ust_app *app;
4074 struct ust_app_session *ua_sess;
4075 struct ust_app_channel *ua_chan;
4076 struct ust_app_event *ua_event;
4077
4078 DBG("UST app disabling event %s for all apps in channel "
d9bf3ca4
MD
4079 "%s for session id %" PRIu64,
4080 uevent->attr.name, uchan->name, usess->id);
b0a40d28
DG
4081
4082 rcu_read_lock();
4083
4084 /* For all registered applications */
852d0037 4085 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
4086 if (!app->compatible) {
4087 /*
4088 * TODO: In time, we should notice the caller of this error by
4089 * telling him that this is a version error.
4090 */
4091 continue;
4092 }
b0a40d28
DG
4093 ua_sess = lookup_session_by_app(usess, app);
4094 if (ua_sess == NULL) {
4095 /* Next app */
4096 continue;
4097 }
4098
4099 /* Lookup channel in the ust app session */
bec39940
DG
4100 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
4101 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
b0a40d28 4102 if (ua_chan_node == NULL) {
d9bf3ca4 4103 DBG2("Channel %s not found in session id %" PRIu64 " for app pid %d."
852d0037 4104 "Skipping", uchan->name, usess->id, app->pid);
b0a40d28
DG
4105 continue;
4106 }
4107 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
4108
700c5a9d
JR
4109 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
4110 uevent->filter, uevent->attr.loglevel,
4111 uevent->exclusion);
4112 if (ua_event == NULL) {
b0a40d28 4113 DBG2("Event %s not found in channel %s for app pid %d."
852d0037 4114 "Skipping", uevent->attr.name, uchan->name, app->pid);
b0a40d28
DG
4115 continue;
4116 }
b0a40d28 4117
7f79d3a1 4118 ret = disable_ust_app_event(ua_sess, ua_event, app);
b0a40d28
DG
4119 if (ret < 0) {
4120 /* XXX: Report error someday... */
4121 continue;
4122 }
4123 }
4124
4125 rcu_read_unlock();
4126
4127 return ret;
4128}
4129
421cb601 4130/*
5b4a0ec0 4131 * For a specific UST session, create the channel for all registered apps.
421cb601 4132 */
35a9059d 4133int ust_app_create_channel_glb(struct ltt_ust_session *usess,
48842b30
DG
4134 struct ltt_ust_channel *uchan)
4135{
3d8ca23b 4136 int ret = 0, created;
bec39940 4137 struct lttng_ht_iter iter;
48842b30 4138 struct ust_app *app;
3d8ca23b 4139 struct ust_app_session *ua_sess = NULL;
48842b30 4140
fc34caaa
DG
4141 /* Very wrong code flow */
4142 assert(usess);
4143 assert(uchan);
421cb601 4144
d9bf3ca4 4145 DBG2("UST app adding channel %s to UST domain for session id %" PRIu64,
a991f516 4146 uchan->name, usess->id);
48842b30
DG
4147
4148 rcu_read_lock();
421cb601 4149
5b4a0ec0 4150 /* For every registered applications */
852d0037 4151 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
4152 if (!app->compatible) {
4153 /*
4154 * TODO: In time, we should notice the caller of this error by
4155 * telling him that this is a version error.
4156 */
4157 continue;
4158 }
a9ad0c8f
MD
4159 if (!trace_ust_pid_tracker_lookup(usess, app->pid)) {
4160 /* Skip. */
4161 continue;
4162 }
4163
edb67388
DG
4164 /*
4165 * Create session on the tracer side and add it to app session HT. Note
4166 * that if session exist, it will simply return a pointer to the ust
4167 * app session.
4168 */
3d8ca23b
DG
4169 ret = create_ust_app_session(usess, app, &ua_sess, &created);
4170 if (ret < 0) {
4171 switch (ret) {
4172 case -ENOTCONN:
4173 /*
4174 * The application's socket is not valid. Either a bad socket
4175 * or a timeout on it. We can't inform the caller that for a
4176 * specific app, the session failed so lets continue here.
4177 */
a7169585 4178 ret = 0; /* Not an error. */
3d8ca23b
DG
4179 continue;
4180 case -ENOMEM:
4181 default:
4182 goto error_rcu_unlock;
4183 }
48842b30 4184 }
3d8ca23b 4185 assert(ua_sess);
48842b30 4186
d0b96690 4187 pthread_mutex_lock(&ua_sess->lock);
b161602a
MD
4188
4189 if (ua_sess->deleted) {
4190 pthread_mutex_unlock(&ua_sess->lock);
4191 continue;
4192 }
4193
d65d2de8
DG
4194 if (!strncmp(uchan->name, DEFAULT_METADATA_NAME,
4195 sizeof(uchan->name))) {
ad7a9107
DG
4196 copy_channel_attr_to_ustctl(&ua_sess->metadata_attr, &uchan->attr);
4197 ret = 0;
d65d2de8
DG
4198 } else {
4199 /* Create channel onto application. We don't need the chan ref. */
4200 ret = create_ust_app_channel(ua_sess, uchan, app,
4201 LTTNG_UST_CHAN_PER_CPU, usess, NULL);
4202 }
d0b96690 4203 pthread_mutex_unlock(&ua_sess->lock);
3d8ca23b 4204 if (ret < 0) {
3d8ca23b
DG
4205 /* Cleanup the created session if it's the case. */
4206 if (created) {
d0b96690 4207 destroy_app_session(app, ua_sess);
3d8ca23b 4208 }
a7169585
MD
4209 switch (ret) {
4210 case -ENOTCONN:
4211 /*
4212 * The application's socket is not valid. Either a bad socket
4213 * or a timeout on it. We can't inform the caller that for a
4214 * specific app, the session failed so lets continue here.
4215 */
4216 ret = 0; /* Not an error. */
4217 continue;
4218 case -ENOMEM:
4219 default:
4220 goto error_rcu_unlock;
4221 }
48842b30 4222 }
48842b30 4223 }
5b4a0ec0 4224
95e047ff 4225error_rcu_unlock:
48842b30 4226 rcu_read_unlock();
3c14c33f 4227 return ret;
48842b30
DG
4228}
4229
5b4a0ec0 4230/*
edb67388 4231 * Enable event for a specific session and channel on the tracer.
5b4a0ec0 4232 */
35a9059d 4233int ust_app_enable_event_glb(struct ltt_ust_session *usess,
48842b30
DG
4234 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent)
4235{
4236 int ret = 0;
bec39940 4237 struct lttng_ht_iter iter, uiter;
18eace3b 4238 struct lttng_ht_node_str *ua_chan_node;
48842b30
DG
4239 struct ust_app *app;
4240 struct ust_app_session *ua_sess;
4241 struct ust_app_channel *ua_chan;
4242 struct ust_app_event *ua_event;
48842b30 4243
d9bf3ca4 4244 DBG("UST app enabling event %s for all apps for session id %" PRIu64,
a991f516 4245 uevent->attr.name, usess->id);
48842b30 4246
edb67388
DG
4247 /*
4248 * NOTE: At this point, this function is called only if the session and
4249 * channel passed are already created for all apps. and enabled on the
4250 * tracer also.
4251 */
4252
48842b30 4253 rcu_read_lock();
421cb601
DG
4254
4255 /* For all registered applications */
852d0037 4256 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
4257 if (!app->compatible) {
4258 /*
4259 * TODO: In time, we should notice the caller of this error by
4260 * telling him that this is a version error.
4261 */
4262 continue;
4263 }
edb67388 4264 ua_sess = lookup_session_by_app(usess, app);
c4a1715b
DG
4265 if (!ua_sess) {
4266 /* The application has problem or is probably dead. */
4267 continue;
4268 }
ba767faf 4269
d0b96690
DG
4270 pthread_mutex_lock(&ua_sess->lock);
4271
b161602a
MD
4272 if (ua_sess->deleted) {
4273 pthread_mutex_unlock(&ua_sess->lock);
4274 continue;
4275 }
4276
edb67388 4277 /* Lookup channel in the ust app session */
bec39940
DG
4278 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
4279 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
a7169585
MD
4280 /*
4281 * It is possible that the channel cannot be found is
4282 * the channel/event creation occurs concurrently with
4283 * an application exit.
4284 */
4285 if (!ua_chan_node) {
4286 pthread_mutex_unlock(&ua_sess->lock);
4287 continue;
4288 }
edb67388
DG
4289
4290 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
4291
18eace3b
DG
4292 /* Get event node */
4293 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
39c5a3a7 4294 uevent->filter, uevent->attr.loglevel, uevent->exclusion);
18eace3b 4295 if (ua_event == NULL) {
7f79d3a1 4296 DBG3("UST app enable event %s not found for app PID %d."
852d0037 4297 "Skipping app", uevent->attr.name, app->pid);
d0b96690 4298 goto next_app;
35a9059d 4299 }
35a9059d
DG
4300
4301 ret = enable_ust_app_event(ua_sess, ua_event, app);
4302 if (ret < 0) {
d0b96690 4303 pthread_mutex_unlock(&ua_sess->lock);
7f79d3a1 4304 goto error;
48842b30 4305 }
d0b96690
DG
4306 next_app:
4307 pthread_mutex_unlock(&ua_sess->lock);
edb67388
DG
4308 }
4309
7f79d3a1 4310error:
edb67388 4311 rcu_read_unlock();
edb67388
DG
4312 return ret;
4313}
4314
4315/*
4316 * For a specific existing UST session and UST channel, creates the event for
4317 * all registered apps.
4318 */
35a9059d 4319int ust_app_create_event_glb(struct ltt_ust_session *usess,
edb67388
DG
4320 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent)
4321{
4322 int ret = 0;
bec39940
DG
4323 struct lttng_ht_iter iter, uiter;
4324 struct lttng_ht_node_str *ua_chan_node;
edb67388
DG
4325 struct ust_app *app;
4326 struct ust_app_session *ua_sess;
4327 struct ust_app_channel *ua_chan;
4328
d9bf3ca4 4329 DBG("UST app creating event %s for all apps for session id %" PRIu64,
a991f516 4330 uevent->attr.name, usess->id);
edb67388 4331
edb67388
DG
4332 rcu_read_lock();
4333
4334 /* For all registered applications */
852d0037 4335 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
4336 if (!app->compatible) {
4337 /*
4338 * TODO: In time, we should notice the caller of this error by
4339 * telling him that this is a version error.
4340 */
4341 continue;
4342 }
edb67388 4343 ua_sess = lookup_session_by_app(usess, app);
c4a1715b
DG
4344 if (!ua_sess) {
4345 /* The application has problem or is probably dead. */
4346 continue;
4347 }
48842b30 4348
d0b96690 4349 pthread_mutex_lock(&ua_sess->lock);
b161602a
MD
4350
4351 if (ua_sess->deleted) {
4352 pthread_mutex_unlock(&ua_sess->lock);
4353 continue;
4354 }
4355
48842b30 4356 /* Lookup channel in the ust app session */
bec39940
DG
4357 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
4358 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
edb67388
DG
4359 /* If the channel is not found, there is a code flow error */
4360 assert(ua_chan_node);
4361
48842b30
DG
4362 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
4363
edb67388 4364 ret = create_ust_app_event(ua_sess, ua_chan, uevent, app);
d0b96690 4365 pthread_mutex_unlock(&ua_sess->lock);
edb67388 4366 if (ret < 0) {
49c336c1 4367 if (ret != -LTTNG_UST_ERR_EXIST) {
fc34caaa
DG
4368 /* Possible value at this point: -ENOMEM. If so, we stop! */
4369 break;
4370 }
4371 DBG2("UST app event %s already exist on app PID %d",
852d0037 4372 uevent->attr.name, app->pid);
5b4a0ec0 4373 continue;
48842b30 4374 }
48842b30 4375 }
5b4a0ec0 4376
48842b30
DG
4377 rcu_read_unlock();
4378
4379 return ret;
4380}
4381
5b4a0ec0
DG
4382/*
4383 * Start tracing for a specific UST session and app.
5e98072f
JR
4384 *
4385 * Called with UST app session lock held.
4386 *
5b4a0ec0 4387 */
b34cbebf 4388static
421cb601 4389int ust_app_start_trace(struct ltt_ust_session *usess, struct ust_app *app)
48842b30
DG
4390{
4391 int ret = 0;
48842b30 4392 struct ust_app_session *ua_sess;
48842b30 4393
852d0037 4394 DBG("Starting tracing for ust app pid %d", app->pid);
5cf5d0e7 4395
509cbaf8
MD
4396 rcu_read_lock();
4397
e0c7ec2b
DG
4398 if (!app->compatible) {
4399 goto end;
4400 }
4401
421cb601
DG
4402 ua_sess = lookup_session_by_app(usess, app);
4403 if (ua_sess == NULL) {
d42f20df
DG
4404 /* The session is in teardown process. Ignore and continue. */
4405 goto end;
421cb601 4406 }
48842b30 4407
d0b96690
DG
4408 pthread_mutex_lock(&ua_sess->lock);
4409
b161602a
MD
4410 if (ua_sess->deleted) {
4411 pthread_mutex_unlock(&ua_sess->lock);
4412 goto end;
4413 }
4414
aea829b3
DG
4415 /* Upon restart, we skip the setup, already done */
4416 if (ua_sess->started) {
8be98f9a 4417 goto skip_setup;
aea829b3 4418 }
8be98f9a 4419
a4b92340
DG
4420 /* Create directories if consumer is LOCAL and has a path defined. */
4421 if (usess->consumer->type == CONSUMER_DST_LOCAL &&
4422 strlen(usess->consumer->dst.trace_path) > 0) {
4423 ret = run_as_mkdir_recursive(usess->consumer->dst.trace_path,
7972aab2 4424 S_IRWXU | S_IRWXG, ua_sess->euid, ua_sess->egid);
a4b92340 4425 if (ret < 0) {
df5b86c8 4426 if (errno != EEXIST) {
a4b92340 4427 ERR("Trace directory creation error");
d0b96690 4428 goto error_unlock;
421cb601 4429 }
173af62f 4430 }
7753dea8 4431 }
aea829b3 4432
d65d2de8
DG
4433 /*
4434 * Create the metadata for the application. This returns gracefully if a
4435 * metadata was already set for the session.
4436 */
ad7a9107 4437 ret = create_ust_app_metadata(ua_sess, app, usess->consumer);
421cb601 4438 if (ret < 0) {
d0b96690 4439 goto error_unlock;
421cb601 4440 }
48842b30 4441
840cb59c 4442 health_code_update();
86acf0da 4443
8be98f9a 4444skip_setup:
421cb601 4445 /* This start the UST tracing */
fb45065e 4446 pthread_mutex_lock(&app->sock_lock);
852d0037 4447 ret = ustctl_start_session(app->sock, ua_sess->handle);
fb45065e 4448 pthread_mutex_unlock(&app->sock_lock);
421cb601 4449 if (ret < 0) {
ffe60014
DG
4450 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4451 ERR("Error starting tracing for app pid: %d (ret: %d)",
4452 app->pid, ret);
4453 } else {
4454 DBG("UST app start session failed. Application is dead.");
3757b385
DG
4455 /*
4456 * This is normal behavior, an application can die during the
4457 * creation process. Don't report an error so the execution can
4458 * continue normally.
4459 */
4460 pthread_mutex_unlock(&ua_sess->lock);
4461 goto end;
ffe60014 4462 }
d0b96690 4463 goto error_unlock;
421cb601 4464 }
5b4a0ec0 4465
55c3953d
DG
4466 /* Indicate that the session has been started once */
4467 ua_sess->started = 1;
4468
d0b96690
DG
4469 pthread_mutex_unlock(&ua_sess->lock);
4470
840cb59c 4471 health_code_update();
86acf0da 4472
421cb601 4473 /* Quiescent wait after starting trace */
fb45065e 4474 pthread_mutex_lock(&app->sock_lock);
ffe60014 4475 ret = ustctl_wait_quiescent(app->sock);
fb45065e 4476 pthread_mutex_unlock(&app->sock_lock);
ffe60014
DG
4477 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4478 ERR("UST app wait quiescent failed for app pid %d ret %d",
4479 app->pid, ret);
4480 }
48842b30 4481
e0c7ec2b
DG
4482end:
4483 rcu_read_unlock();
840cb59c 4484 health_code_update();
421cb601 4485 return 0;
48842b30 4486
d0b96690
DG
4487error_unlock:
4488 pthread_mutex_unlock(&ua_sess->lock);
509cbaf8 4489 rcu_read_unlock();
840cb59c 4490 health_code_update();
421cb601
DG
4491 return -1;
4492}
48842b30 4493
8be98f9a
MD
4494/*
4495 * Stop tracing for a specific UST session and app.
4496 */
b34cbebf 4497static
8be98f9a
MD
4498int ust_app_stop_trace(struct ltt_ust_session *usess, struct ust_app *app)
4499{
4500 int ret = 0;
4501 struct ust_app_session *ua_sess;
7972aab2 4502 struct ust_registry_session *registry;
8be98f9a 4503
852d0037 4504 DBG("Stopping tracing for ust app pid %d", app->pid);
8be98f9a
MD
4505
4506 rcu_read_lock();
4507
e0c7ec2b 4508 if (!app->compatible) {
d88aee68 4509 goto end_no_session;
e0c7ec2b
DG
4510 }
4511
8be98f9a
MD
4512 ua_sess = lookup_session_by_app(usess, app);
4513 if (ua_sess == NULL) {
d88aee68 4514 goto end_no_session;
8be98f9a
MD
4515 }
4516
d88aee68
DG
4517 pthread_mutex_lock(&ua_sess->lock);
4518
b161602a
MD
4519 if (ua_sess->deleted) {
4520 pthread_mutex_unlock(&ua_sess->lock);
4521 goto end_no_session;
4522 }
4523
9bc07046
DG
4524 /*
4525 * If started = 0, it means that stop trace has been called for a session
c45536e1
DG
4526 * that was never started. It's possible since we can have a fail start
4527 * from either the application manager thread or the command thread. Simply
4528 * indicate that this is a stop error.
9bc07046 4529 */
f9dfc3d9 4530 if (!ua_sess->started) {
c45536e1
DG
4531 goto error_rcu_unlock;
4532 }
7db205b5 4533
840cb59c 4534 health_code_update();
86acf0da 4535
9d6c7d3f 4536 /* This inhibits UST tracing */
fb45065e 4537 pthread_mutex_lock(&app->sock_lock);
852d0037 4538 ret = ustctl_stop_session(app->sock, ua_sess->handle);
fb45065e 4539 pthread_mutex_unlock(&app->sock_lock);
9d6c7d3f 4540 if (ret < 0) {
ffe60014
DG
4541 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4542 ERR("Error stopping tracing for app pid: %d (ret: %d)",
4543 app->pid, ret);
4544 } else {
4545 DBG("UST app stop session failed. Application is dead.");
3757b385
DG
4546 /*
4547 * This is normal behavior, an application can die during the
4548 * creation process. Don't report an error so the execution can
4549 * continue normally.
4550 */
4551 goto end_unlock;
ffe60014 4552 }
9d6c7d3f
DG
4553 goto error_rcu_unlock;
4554 }
4555
840cb59c 4556 health_code_update();
86acf0da 4557
9d6c7d3f 4558 /* Quiescent wait after stopping trace */
fb45065e 4559 pthread_mutex_lock(&app->sock_lock);
ffe60014 4560 ret = ustctl_wait_quiescent(app->sock);
fb45065e 4561 pthread_mutex_unlock(&app->sock_lock);
ffe60014
DG
4562 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4563 ERR("UST app wait quiescent failed for app pid %d ret %d",
4564 app->pid, ret);
4565 }
9d6c7d3f 4566
840cb59c 4567 health_code_update();
86acf0da 4568
b34cbebf 4569 registry = get_session_registry(ua_sess);
5e98072f
JR
4570
4571 /* The UST app session is held registry shall not be null. */
b34cbebf 4572 assert(registry);
1b532a60 4573
ce34fcd0
MD
4574 /* Push metadata for application before freeing the application. */
4575 (void) push_metadata(registry, ua_sess->consumer);
b34cbebf 4576
3757b385 4577end_unlock:
b34cbebf
MD
4578 pthread_mutex_unlock(&ua_sess->lock);
4579end_no_session:
4580 rcu_read_unlock();
4581 health_code_update();
4582 return 0;
4583
4584error_rcu_unlock:
4585 pthread_mutex_unlock(&ua_sess->lock);
4586 rcu_read_unlock();
4587 health_code_update();
4588 return -1;
4589}
4590
b34cbebf 4591static
c4b88406
MD
4592int ust_app_flush_app_session(struct ust_app *app,
4593 struct ust_app_session *ua_sess)
b34cbebf 4594{
c4b88406 4595 int ret, retval = 0;
b34cbebf 4596 struct lttng_ht_iter iter;
b34cbebf 4597 struct ust_app_channel *ua_chan;
c4b88406 4598 struct consumer_socket *socket;
b34cbebf 4599
c4b88406 4600 DBG("Flushing app session buffers for ust app pid %d", app->pid);
b34cbebf
MD
4601
4602 rcu_read_lock();
4603
4604 if (!app->compatible) {
c4b88406 4605 goto end_not_compatible;
b34cbebf
MD
4606 }
4607
4608 pthread_mutex_lock(&ua_sess->lock);
4609
b161602a
MD
4610 if (ua_sess->deleted) {
4611 goto end_deleted;
4612 }
4613
b34cbebf
MD
4614 health_code_update();
4615
9d6c7d3f 4616 /* Flushing buffers */
c4b88406
MD
4617 socket = consumer_find_socket_by_bitness(app->bits_per_long,
4618 ua_sess->consumer);
ce34fcd0
MD
4619
4620 /* Flush buffers and push metadata. */
4621 switch (ua_sess->buffer_type) {
4622 case LTTNG_BUFFER_PER_PID:
4623 cds_lfht_for_each_entry(ua_sess->channels->ht, &iter.iter, ua_chan,
4624 node.node) {
4625 health_code_update();
ce34fcd0
MD
4626 ret = consumer_flush_channel(socket, ua_chan->key);
4627 if (ret) {
4628 ERR("Error flushing consumer channel");
4629 retval = -1;
4630 continue;
4631 }
8be98f9a 4632 }
ce34fcd0
MD
4633 break;
4634 case LTTNG_BUFFER_PER_UID:
4635 default:
4636 assert(0);
4637 break;
8be98f9a 4638 }
8be98f9a 4639
840cb59c 4640 health_code_update();
86acf0da 4641
b161602a 4642end_deleted:
d88aee68 4643 pthread_mutex_unlock(&ua_sess->lock);
ce34fcd0 4644
c4b88406
MD
4645end_not_compatible:
4646 rcu_read_unlock();
4647 health_code_update();
4648 return retval;
4649}
4650
4651/*
ce34fcd0
MD
4652 * Flush buffers for all applications for a specific UST session.
4653 * Called with UST session lock held.
c4b88406
MD
4654 */
4655static
ce34fcd0 4656int ust_app_flush_session(struct ltt_ust_session *usess)
c4b88406
MD
4657
4658{
99b1411c 4659 int ret = 0;
c4b88406 4660
ce34fcd0 4661 DBG("Flushing session buffers for all ust apps");
c4b88406
MD
4662
4663 rcu_read_lock();
4664
ce34fcd0
MD
4665 /* Flush buffers and push metadata. */
4666 switch (usess->buffer_type) {
4667 case LTTNG_BUFFER_PER_UID:
4668 {
4669 struct buffer_reg_uid *reg;
4670 struct lttng_ht_iter iter;
4671
4672 /* Flush all per UID buffers associated to that session. */
4673 cds_list_for_each_entry(reg, &usess->buffer_reg_uid_list, lnode) {
4674 struct ust_registry_session *ust_session_reg;
4675 struct buffer_reg_channel *reg_chan;
4676 struct consumer_socket *socket;
4677
4678 /* Get consumer socket to use to push the metadata.*/
4679 socket = consumer_find_socket_by_bitness(reg->bits_per_long,
4680 usess->consumer);
4681 if (!socket) {
4682 /* Ignore request if no consumer is found for the session. */
4683 continue;
4684 }
4685
4686 cds_lfht_for_each_entry(reg->registry->channels->ht, &iter.iter,
4687 reg_chan, node.node) {
4688 /*
4689 * The following call will print error values so the return
4690 * code is of little importance because whatever happens, we
4691 * have to try them all.
4692 */
4693 (void) consumer_flush_channel(socket, reg_chan->consumer_key);
4694 }
4695
4696 ust_session_reg = reg->registry->reg.ust;
4697 /* Push metadata. */
4698 (void) push_metadata(ust_session_reg, usess->consumer);
4699 }
ce34fcd0
MD
4700 break;
4701 }
4702 case LTTNG_BUFFER_PER_PID:
4703 {
4704 struct ust_app_session *ua_sess;
4705 struct lttng_ht_iter iter;
4706 struct ust_app *app;
4707
4708 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
4709 ua_sess = lookup_session_by_app(usess, app);
4710 if (ua_sess == NULL) {
4711 continue;
4712 }
4713 (void) ust_app_flush_app_session(app, ua_sess);
4714 }
4715 break;
4716 }
4717 default:
99b1411c 4718 ret = -1;
ce34fcd0
MD
4719 assert(0);
4720 break;
c4b88406 4721 }
c4b88406 4722
7db205b5 4723 rcu_read_unlock();
840cb59c 4724 health_code_update();
c4b88406 4725 return ret;
8be98f9a
MD
4726}
4727
0dd01979
MD
4728static
4729int ust_app_clear_quiescent_app_session(struct ust_app *app,
4730 struct ust_app_session *ua_sess)
4731{
4732 int ret = 0;
4733 struct lttng_ht_iter iter;
4734 struct ust_app_channel *ua_chan;
4735 struct consumer_socket *socket;
4736
4737 DBG("Clearing stream quiescent state for ust app pid %d", app->pid);
4738
4739 rcu_read_lock();
4740
4741 if (!app->compatible) {
4742 goto end_not_compatible;
4743 }
4744
4745 pthread_mutex_lock(&ua_sess->lock);
4746
4747 if (ua_sess->deleted) {
4748 goto end_unlock;
4749 }
4750
4751 health_code_update();
4752
4753 socket = consumer_find_socket_by_bitness(app->bits_per_long,
4754 ua_sess->consumer);
4755 if (!socket) {
4756 ERR("Failed to find consumer (%" PRIu32 ") socket",
4757 app->bits_per_long);
4758 ret = -1;
4759 goto end_unlock;
4760 }
4761
4762 /* Clear quiescent state. */
4763 switch (ua_sess->buffer_type) {
4764 case LTTNG_BUFFER_PER_PID:
4765 cds_lfht_for_each_entry(ua_sess->channels->ht, &iter.iter,
4766 ua_chan, node.node) {
4767 health_code_update();
4768 ret = consumer_clear_quiescent_channel(socket,
4769 ua_chan->key);
4770 if (ret) {
4771 ERR("Error clearing quiescent state for consumer channel");
4772 ret = -1;
4773 continue;
4774 }
4775 }
4776 break;
4777 case LTTNG_BUFFER_PER_UID:
4778 default:
4779 assert(0);
4780 ret = -1;
4781 break;
4782 }
4783
4784 health_code_update();
4785
4786end_unlock:
4787 pthread_mutex_unlock(&ua_sess->lock);
4788
4789end_not_compatible:
4790 rcu_read_unlock();
4791 health_code_update();
4792 return ret;
4793}
4794
4795/*
4796 * Clear quiescent state in each stream for all applications for a
4797 * specific UST session.
4798 * Called with UST session lock held.
4799 */
4800static
4801int ust_app_clear_quiescent_session(struct ltt_ust_session *usess)
4802
4803{
4804 int ret = 0;
4805
4806 DBG("Clearing stream quiescent state for all ust apps");
4807
4808 rcu_read_lock();
4809
4810 switch (usess->buffer_type) {
4811 case LTTNG_BUFFER_PER_UID:
4812 {
4813 struct lttng_ht_iter iter;
4814 struct buffer_reg_uid *reg;
4815
4816 /*
4817 * Clear quiescent for all per UID buffers associated to
4818 * that session.
4819 */
4820 cds_list_for_each_entry(reg, &usess->buffer_reg_uid_list, lnode) {
4821 struct consumer_socket *socket;
4822 struct buffer_reg_channel *reg_chan;
4823
4824 /* Get associated consumer socket.*/
4825 socket = consumer_find_socket_by_bitness(
4826 reg->bits_per_long, usess->consumer);
4827 if (!socket) {
4828 /*
4829 * Ignore request if no consumer is found for
4830 * the session.
4831 */
4832 continue;
4833 }
4834
4835 cds_lfht_for_each_entry(reg->registry->channels->ht,
4836 &iter.iter, reg_chan, node.node) {
4837 /*
4838 * The following call will print error values so
4839 * the return code is of little importance
4840 * because whatever happens, we have to try them
4841 * all.
4842 */
4843 (void) consumer_clear_quiescent_channel(socket,
4844 reg_chan->consumer_key);
4845 }
4846 }
4847 break;
4848 }
4849 case LTTNG_BUFFER_PER_PID:
4850 {
4851 struct ust_app_session *ua_sess;
4852 struct lttng_ht_iter iter;
4853 struct ust_app *app;
4854
4855 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app,
4856 pid_n.node) {
4857 ua_sess = lookup_session_by_app(usess, app);
4858 if (ua_sess == NULL) {
4859 continue;
4860 }
4861 (void) ust_app_clear_quiescent_app_session(app,
4862 ua_sess);
4863 }
4864 break;
4865 }
4866 default:
4867 ret = -1;
4868 assert(0);
4869 break;
4870 }
4871
4872 rcu_read_unlock();
4873 health_code_update();
4874 return ret;
4875}
4876
84cd17c6
MD
4877/*
4878 * Destroy a specific UST session in apps.
4879 */
3353de95 4880static int destroy_trace(struct ltt_ust_session *usess, struct ust_app *app)
84cd17c6 4881{
ffe60014 4882 int ret;
84cd17c6 4883 struct ust_app_session *ua_sess;
bec39940 4884 struct lttng_ht_iter iter;
d9bf3ca4 4885 struct lttng_ht_node_u64 *node;
84cd17c6 4886
852d0037 4887 DBG("Destroy tracing for ust app pid %d", app->pid);
84cd17c6
MD
4888
4889 rcu_read_lock();
4890
e0c7ec2b
DG
4891 if (!app->compatible) {
4892 goto end;
4893 }
4894
84cd17c6 4895 __lookup_session_by_app(usess, app, &iter);
d9bf3ca4 4896 node = lttng_ht_iter_get_node_u64(&iter);
84cd17c6 4897 if (node == NULL) {
d42f20df
DG
4898 /* Session is being or is deleted. */
4899 goto end;
84cd17c6
MD
4900 }
4901 ua_sess = caa_container_of(node, struct ust_app_session, node);
c4a1715b 4902
840cb59c 4903 health_code_update();
d0b96690 4904 destroy_app_session(app, ua_sess);
84cd17c6 4905
840cb59c 4906 health_code_update();
7db205b5 4907
84cd17c6 4908 /* Quiescent wait after stopping trace */
fb45065e 4909 pthread_mutex_lock(&app->sock_lock);
ffe60014 4910 ret = ustctl_wait_quiescent(app->sock);
fb45065e 4911 pthread_mutex_unlock(&app->sock_lock);
ffe60014
DG
4912 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4913 ERR("UST app wait quiescent failed for app pid %d ret %d",
4914 app->pid, ret);
4915 }
e0c7ec2b
DG
4916end:
4917 rcu_read_unlock();
840cb59c 4918 health_code_update();
84cd17c6 4919 return 0;
84cd17c6
MD
4920}
4921
5b4a0ec0
DG
4922/*
4923 * Start tracing for the UST session.
4924 */
421cb601
DG
4925int ust_app_start_trace_all(struct ltt_ust_session *usess)
4926{
4927 int ret = 0;
bec39940 4928 struct lttng_ht_iter iter;
421cb601 4929 struct ust_app *app;
48842b30 4930
421cb601
DG
4931 DBG("Starting all UST traces");
4932
4933 rcu_read_lock();
421cb601 4934
0dd01979
MD
4935 /*
4936 * In a start-stop-start use-case, we need to clear the quiescent state
4937 * of each channel set by the prior stop command, thus ensuring that a
4938 * following stop or destroy is sure to grab a timestamp_end near those
4939 * operations, even if the packet is empty.
4940 */
4941 (void) ust_app_clear_quiescent_session(usess);
4942
852d0037 4943 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
421cb601 4944 ret = ust_app_start_trace(usess, app);
48842b30 4945 if (ret < 0) {
5b4a0ec0
DG
4946 /* Continue to next apps even on error */
4947 continue;
48842b30 4948 }
48842b30 4949 }
5b4a0ec0 4950
48842b30
DG
4951 rcu_read_unlock();
4952
4953 return 0;
4954}
487cf67c 4955
8be98f9a
MD
4956/*
4957 * Start tracing for the UST session.
ce34fcd0 4958 * Called with UST session lock held.
8be98f9a
MD
4959 */
4960int ust_app_stop_trace_all(struct ltt_ust_session *usess)
4961{
4962 int ret = 0;
bec39940 4963 struct lttng_ht_iter iter;
8be98f9a
MD
4964 struct ust_app *app;
4965
4966 DBG("Stopping all UST traces");
4967
4968 rcu_read_lock();
4969
b34cbebf
MD
4970 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
4971 ret = ust_app_stop_trace(usess, app);
4972 if (ret < 0) {
4973 /* Continue to next apps even on error */
4974 continue;
4975 }
4976 }
4977
ce34fcd0 4978 (void) ust_app_flush_session(usess);
8be98f9a
MD
4979
4980 rcu_read_unlock();
4981
4982 return 0;
4983}
4984
84cd17c6
MD
4985/*
4986 * Destroy app UST session.
4987 */
4988int ust_app_destroy_trace_all(struct ltt_ust_session *usess)
4989{
4990 int ret = 0;
bec39940 4991 struct lttng_ht_iter iter;
84cd17c6
MD
4992 struct ust_app *app;
4993
4994 DBG("Destroy all UST traces");
4995
4996 rcu_read_lock();
4997
852d0037 4998 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
3353de95 4999 ret = destroy_trace(usess, app);
84cd17c6
MD
5000 if (ret < 0) {
5001 /* Continue to next apps even on error */
5002 continue;
5003 }
5004 }
5005
5006 rcu_read_unlock();
5007
5008 return 0;
5009}
5010
a9ad0c8f
MD
5011static
5012void ust_app_global_create(struct ltt_ust_session *usess, struct ust_app *app)
487cf67c 5013{
55c54cce 5014 int ret = 0;
31746f93 5015 struct lttng_ht_iter iter, uiter;
3d8ca23b 5016 struct ust_app_session *ua_sess = NULL;
487cf67c
DG
5017 struct ust_app_channel *ua_chan;
5018 struct ust_app_event *ua_event;
727d5404 5019 struct ust_app_ctx *ua_ctx;
a9ad0c8f 5020 int is_created = 0;
1f3580c7 5021
a9ad0c8f 5022 ret = create_ust_app_session(usess, app, &ua_sess, &is_created);
3d8ca23b
DG
5023 if (ret < 0) {
5024 /* Tracer is probably gone or ENOMEM. */
487cf67c
DG
5025 goto error;
5026 }
a9ad0c8f
MD
5027 if (!is_created) {
5028 /* App session already created. */
5029 goto end;
5030 }
3d8ca23b 5031 assert(ua_sess);
487cf67c 5032
d0b96690
DG
5033 pthread_mutex_lock(&ua_sess->lock);
5034
b161602a
MD
5035 if (ua_sess->deleted) {
5036 pthread_mutex_unlock(&ua_sess->lock);
5037 goto end;
5038 }
5039
284d8f55 5040 /*
d0b96690 5041 * We can iterate safely here over all UST app session since the create ust
284d8f55
DG
5042 * app session above made a shadow copy of the UST global domain from the
5043 * ltt ust session.
5044 */
bec39940
DG
5045 cds_lfht_for_each_entry(ua_sess->channels->ht, &iter.iter, ua_chan,
5046 node.node) {
ad7a9107 5047 ret = do_create_channel(app, usess, ua_sess, ua_chan);
a7169585 5048 if (ret < 0 && ret != -ENOTCONN) {
ad7a9107 5049 /*
a7169585
MD
5050 * Stop everything. On error, the application
5051 * failed, no more file descriptor are available
5052 * or ENOMEM so stopping here is the only thing
5053 * we can do for now. The only exception is
5054 * -ENOTCONN, which indicates that the application
5055 * has exit.
ad7a9107
DG
5056 */
5057 goto error_unlock;
487cf67c
DG
5058 }
5059
31746f93
DG
5060 /*
5061 * Add context using the list so they are enabled in the same order the
5062 * user added them.
5063 */
5064 cds_list_for_each_entry(ua_ctx, &ua_chan->ctx_list, list) {
727d5404
DG
5065 ret = create_ust_channel_context(ua_chan, ua_ctx, app);
5066 if (ret < 0) {
d0b96690 5067 goto error_unlock;
727d5404
DG
5068 }
5069 }
5070
5071
284d8f55 5072 /* For each events */
bec39940
DG
5073 cds_lfht_for_each_entry(ua_chan->events->ht, &uiter.iter, ua_event,
5074 node.node) {
284d8f55
DG
5075 ret = create_ust_event(app, ua_sess, ua_chan, ua_event);
5076 if (ret < 0) {
d0b96690 5077 goto error_unlock;
487cf67c 5078 }
36dc12cc 5079 }
487cf67c
DG
5080 }
5081
d0b96690
DG
5082 pthread_mutex_unlock(&ua_sess->lock);
5083
14fb1ebe 5084 if (usess->active) {
421cb601 5085 ret = ust_app_start_trace(usess, app);
36dc12cc 5086 if (ret < 0) {
36dc12cc
DG
5087 goto error;
5088 }
5089
852d0037 5090 DBG2("UST trace started for app pid %d", app->pid);
36dc12cc 5091 }
a9ad0c8f 5092end:
ffe60014 5093 /* Everything went well at this point. */
ffe60014
DG
5094 return;
5095
d0b96690
DG
5096error_unlock:
5097 pthread_mutex_unlock(&ua_sess->lock);
487cf67c 5098error:
ffe60014 5099 if (ua_sess) {
d0b96690 5100 destroy_app_session(app, ua_sess);
ffe60014 5101 }
487cf67c
DG
5102 return;
5103}
55cc08a6 5104
a9ad0c8f
MD
5105static
5106void ust_app_global_destroy(struct ltt_ust_session *usess, struct ust_app *app)
5107{
5108 struct ust_app_session *ua_sess;
5109
5110 ua_sess = lookup_session_by_app(usess, app);
5111 if (ua_sess == NULL) {
5112 return;
5113 }
5114 destroy_app_session(app, ua_sess);
5115}
5116
5117/*
5118 * Add channels/events from UST global domain to registered apps at sock.
5119 *
5120 * Called with session lock held.
5121 * Called with RCU read-side lock held.
5122 */
5123void ust_app_global_update(struct ltt_ust_session *usess, struct ust_app *app)
5124{
5125 assert(usess);
5126
5127 DBG2("UST app global update for app sock %d for session id %" PRIu64,
5128 app->sock, usess->id);
5129
5130 if (!app->compatible) {
5131 return;
5132 }
5133
5134 if (trace_ust_pid_tracker_lookup(usess, app->pid)) {
5135 ust_app_global_create(usess, app);
5136 } else {
5137 ust_app_global_destroy(usess, app);
5138 }
5139}
5140
5141/*
5142 * Called with session lock held.
5143 */
5144void ust_app_global_update_all(struct ltt_ust_session *usess)
5145{
5146 struct lttng_ht_iter iter;
5147 struct ust_app *app;
5148
5149 rcu_read_lock();
5150 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
5151 ust_app_global_update(usess, app);
5152 }
5153 rcu_read_unlock();
5154}
5155
55cc08a6
DG
5156/*
5157 * Add context to a specific channel for global UST domain.
5158 */
5159int ust_app_add_ctx_channel_glb(struct ltt_ust_session *usess,
5160 struct ltt_ust_channel *uchan, struct ltt_ust_context *uctx)
5161{
5162 int ret = 0;
bec39940
DG
5163 struct lttng_ht_node_str *ua_chan_node;
5164 struct lttng_ht_iter iter, uiter;
55cc08a6
DG
5165 struct ust_app_channel *ua_chan = NULL;
5166 struct ust_app_session *ua_sess;
5167 struct ust_app *app;
5168
5169 rcu_read_lock();
5170
852d0037 5171 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
5172 if (!app->compatible) {
5173 /*
5174 * TODO: In time, we should notice the caller of this error by
5175 * telling him that this is a version error.
5176 */
5177 continue;
5178 }
55cc08a6
DG
5179 ua_sess = lookup_session_by_app(usess, app);
5180 if (ua_sess == NULL) {
5181 continue;
5182 }
5183
d0b96690 5184 pthread_mutex_lock(&ua_sess->lock);
b161602a
MD
5185
5186 if (ua_sess->deleted) {
5187 pthread_mutex_unlock(&ua_sess->lock);
5188 continue;
5189 }
5190
55cc08a6 5191 /* Lookup channel in the ust app session */
bec39940
DG
5192 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
5193 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
55cc08a6 5194 if (ua_chan_node == NULL) {
d0b96690 5195 goto next_app;
55cc08a6
DG
5196 }
5197 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel,
5198 node);
55cc08a6
DG
5199 ret = create_ust_app_channel_context(ua_sess, ua_chan, &uctx->ctx, app);
5200 if (ret < 0) {
d0b96690 5201 goto next_app;
55cc08a6 5202 }
d0b96690
DG
5203 next_app:
5204 pthread_mutex_unlock(&ua_sess->lock);
55cc08a6
DG
5205 }
5206
55cc08a6
DG
5207 rcu_read_unlock();
5208 return ret;
5209}
5210
76d45b40
DG
5211/*
5212 * Enable event for a channel from a UST session for a specific PID.
5213 */
5214int ust_app_enable_event_pid(struct ltt_ust_session *usess,
5215 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent, pid_t pid)
5216{
5217 int ret = 0;
bec39940 5218 struct lttng_ht_iter iter;
18eace3b 5219 struct lttng_ht_node_str *ua_chan_node;
76d45b40
DG
5220 struct ust_app *app;
5221 struct ust_app_session *ua_sess;
5222 struct ust_app_channel *ua_chan;
5223 struct ust_app_event *ua_event;
5224
5225 DBG("UST app enabling event %s for PID %d", uevent->attr.name, pid);
5226
5227 rcu_read_lock();
5228
5229 app = ust_app_find_by_pid(pid);
5230 if (app == NULL) {
5231 ERR("UST app enable event per PID %d not found", pid);
5232 ret = -1;
d0b96690 5233 goto end;
76d45b40
DG
5234 }
5235
e0c7ec2b
DG
5236 if (!app->compatible) {
5237 ret = 0;
d0b96690 5238 goto end;
e0c7ec2b
DG
5239 }
5240
76d45b40 5241 ua_sess = lookup_session_by_app(usess, app);
c4a1715b
DG
5242 if (!ua_sess) {
5243 /* The application has problem or is probably dead. */
d0b96690
DG
5244 ret = 0;
5245 goto end;
c4a1715b 5246 }
76d45b40 5247
d0b96690 5248 pthread_mutex_lock(&ua_sess->lock);
b161602a
MD
5249
5250 if (ua_sess->deleted) {
5251 ret = 0;
5252 goto end_unlock;
5253 }
5254
76d45b40 5255 /* Lookup channel in the ust app session */
bec39940
DG
5256 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &iter);
5257 ua_chan_node = lttng_ht_iter_get_node_str(&iter);
76d45b40
DG
5258 /* If the channel is not found, there is a code flow error */
5259 assert(ua_chan_node);
5260
5261 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
5262
18eace3b 5263 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
39c5a3a7 5264 uevent->filter, uevent->attr.loglevel, uevent->exclusion);
18eace3b 5265 if (ua_event == NULL) {
76d45b40
DG
5266 ret = create_ust_app_event(ua_sess, ua_chan, uevent, app);
5267 if (ret < 0) {
d0b96690 5268 goto end_unlock;
76d45b40
DG
5269 }
5270 } else {
76d45b40
DG
5271 ret = enable_ust_app_event(ua_sess, ua_event, app);
5272 if (ret < 0) {
d0b96690 5273 goto end_unlock;
76d45b40
DG
5274 }
5275 }
5276
d0b96690
DG
5277end_unlock:
5278 pthread_mutex_unlock(&ua_sess->lock);
5279end:
76d45b40
DG
5280 rcu_read_unlock();
5281 return ret;
5282}
7f79d3a1 5283
d0b96690
DG
5284/*
5285 * Receive registration and populate the given msg structure.
5286 *
5287 * On success return 0 else a negative value returned by the ustctl call.
5288 */
5289int ust_app_recv_registration(int sock, struct ust_register_msg *msg)
5290{
5291 int ret;
5292 uint32_t pid, ppid, uid, gid;
5293
5294 assert(msg);
5295
5296 ret = ustctl_recv_reg_msg(sock, &msg->type, &msg->major, &msg->minor,
5297 &pid, &ppid, &uid, &gid,
5298 &msg->bits_per_long,
5299 &msg->uint8_t_alignment,
5300 &msg->uint16_t_alignment,
5301 &msg->uint32_t_alignment,
5302 &msg->uint64_t_alignment,
5303 &msg->long_alignment,
5304 &msg->byte_order,
5305 msg->name);
5306 if (ret < 0) {
5307 switch (-ret) {
5308 case EPIPE:
5309 case ECONNRESET:
5310 case LTTNG_UST_ERR_EXITING:
5311 DBG3("UST app recv reg message failed. Application died");
5312 break;
5313 case LTTNG_UST_ERR_UNSUP_MAJOR:
5314 ERR("UST app recv reg unsupported version %d.%d. Supporting %d.%d",
5315 msg->major, msg->minor, LTTNG_UST_ABI_MAJOR_VERSION,
5316 LTTNG_UST_ABI_MINOR_VERSION);
5317 break;
5318 default:
5319 ERR("UST app recv reg message failed with ret %d", ret);
5320 break;
5321 }
5322 goto error;
5323 }
5324 msg->pid = (pid_t) pid;
5325 msg->ppid = (pid_t) ppid;
5326 msg->uid = (uid_t) uid;
5327 msg->gid = (gid_t) gid;
5328
5329error:
5330 return ret;
5331}
5332
10b56aef
MD
5333/*
5334 * Return a ust app session object using the application object and the
5335 * session object descriptor has a key. If not found, NULL is returned.
5336 * A RCU read side lock MUST be acquired when calling this function.
5337*/
5338static struct ust_app_session *find_session_by_objd(struct ust_app *app,
5339 int objd)
5340{
5341 struct lttng_ht_node_ulong *node;
5342 struct lttng_ht_iter iter;
5343 struct ust_app_session *ua_sess = NULL;
5344
5345 assert(app);
5346
5347 lttng_ht_lookup(app->ust_sessions_objd, (void *)((unsigned long) objd), &iter);
5348 node = lttng_ht_iter_get_node_ulong(&iter);
5349 if (node == NULL) {
5350 DBG2("UST app session find by objd %d not found", objd);
5351 goto error;
5352 }
5353
5354 ua_sess = caa_container_of(node, struct ust_app_session, ust_objd_node);
5355
5356error:
5357 return ua_sess;
5358}
5359
d88aee68
DG
5360/*
5361 * Return a ust app channel object using the application object and the channel
5362 * object descriptor has a key. If not found, NULL is returned. A RCU read side
5363 * lock MUST be acquired before calling this function.
5364 */
d0b96690
DG
5365static struct ust_app_channel *find_channel_by_objd(struct ust_app *app,
5366 int objd)
5367{
5368 struct lttng_ht_node_ulong *node;
5369 struct lttng_ht_iter iter;
5370 struct ust_app_channel *ua_chan = NULL;
5371
5372 assert(app);
5373
5374 lttng_ht_lookup(app->ust_objd, (void *)((unsigned long) objd), &iter);
5375 node = lttng_ht_iter_get_node_ulong(&iter);
5376 if (node == NULL) {
5377 DBG2("UST app channel find by objd %d not found", objd);
5378 goto error;
5379 }
5380
5381 ua_chan = caa_container_of(node, struct ust_app_channel, ust_objd_node);
5382
5383error:
5384 return ua_chan;
5385}
5386
d88aee68
DG
5387/*
5388 * Reply to a register channel notification from an application on the notify
5389 * socket. The channel metadata is also created.
5390 *
5391 * The session UST registry lock is acquired in this function.
5392 *
5393 * On success 0 is returned else a negative value.
5394 */
d0b96690
DG
5395static int reply_ust_register_channel(int sock, int sobjd, int cobjd,
5396 size_t nr_fields, struct ustctl_field *fields)
5397{
5398 int ret, ret_code = 0;
bd6e2c0b 5399 uint32_t chan_id;
7972aab2 5400 uint64_t chan_reg_key;
d0b96690
DG
5401 enum ustctl_channel_header type;
5402 struct ust_app *app;
5403 struct ust_app_channel *ua_chan;
5404 struct ust_app_session *ua_sess;
7972aab2 5405 struct ust_registry_session *registry;
45893984 5406 struct ust_registry_channel *chan_reg;
d0b96690
DG
5407
5408 rcu_read_lock();
5409
5410 /* Lookup application. If not found, there is a code flow error. */
5411 app = find_app_by_notify_sock(sock);
d88aee68 5412 if (!app) {
5e98072f 5413 DBG("Application socket %d is being torn down. Abort event notify",
d88aee68
DG
5414 sock);
5415 ret = 0;
5416 goto error_rcu_unlock;
5417 }
d0b96690 5418
4950b860 5419 /* Lookup channel by UST object descriptor. */
d0b96690 5420 ua_chan = find_channel_by_objd(app, cobjd);
4950b860 5421 if (!ua_chan) {
5e98072f 5422 DBG("Application channel is being torn down. Abort event notify");
4950b860
MD
5423 ret = 0;
5424 goto error_rcu_unlock;
5425 }
5426
d0b96690
DG
5427 assert(ua_chan->session);
5428 ua_sess = ua_chan->session;
d0b96690 5429
7972aab2
DG
5430 /* Get right session registry depending on the session buffer type. */
5431 registry = get_session_registry(ua_sess);
5e98072f
JR
5432 if (!registry) {
5433 DBG("Application session is being torn down. Abort event notify");
5434 ret = 0;
5435 goto error_rcu_unlock;
5436 };
45893984 5437
7972aab2
DG
5438 /* Depending on the buffer type, a different channel key is used. */
5439 if (ua_sess->buffer_type == LTTNG_BUFFER_PER_UID) {
5440 chan_reg_key = ua_chan->tracing_channel_id;
d0b96690 5441 } else {
7972aab2 5442 chan_reg_key = ua_chan->key;
d0b96690
DG
5443 }
5444
7972aab2
DG
5445 pthread_mutex_lock(&registry->lock);
5446
5447 chan_reg = ust_registry_channel_find(registry, chan_reg_key);
5448 assert(chan_reg);
5449
5450 if (!chan_reg->register_done) {
bd6e2c0b
MD
5451 /*
5452 * TODO: eventually use the registry event count for
5453 * this channel to better guess header type for per-pid
5454 * buffers.
5455 */
5456 type = USTCTL_CHANNEL_HEADER_LARGE;
7972aab2
DG
5457 chan_reg->nr_ctx_fields = nr_fields;
5458 chan_reg->ctx_fields = fields;
5e98072f 5459 fields = NULL;
7972aab2 5460 chan_reg->header_type = type;
d0b96690 5461 } else {
7972aab2
DG
5462 /* Get current already assigned values. */
5463 type = chan_reg->header_type;
d0b96690 5464 }
7972aab2
DG
5465 /* Channel id is set during the object creation. */
5466 chan_id = chan_reg->chan_id;
d0b96690
DG
5467
5468 /* Append to metadata */
7972aab2
DG
5469 if (!chan_reg->metadata_dumped) {
5470 ret_code = ust_metadata_channel_statedump(registry, chan_reg);
d0b96690
DG
5471 if (ret_code) {
5472 ERR("Error appending channel metadata (errno = %d)", ret_code);
5473 goto reply;
5474 }
5475 }
5476
5477reply:
7972aab2
DG
5478 DBG3("UST app replying to register channel key %" PRIu64
5479 " with id %u, type: %d, ret: %d", chan_reg_key, chan_id, type,
5480 ret_code);
d0b96690
DG
5481
5482 ret = ustctl_reply_register_channel(sock, chan_id, type, ret_code);
5483 if (ret < 0) {
5484 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
5485 ERR("UST app reply channel failed with ret %d", ret);
5486 } else {
5487 DBG3("UST app reply channel failed. Application died");
5488 }
5489 goto error;
5490 }
5491
7972aab2
DG
5492 /* This channel registry registration is completed. */
5493 chan_reg->register_done = 1;
5494
d0b96690 5495error:
7972aab2 5496 pthread_mutex_unlock(&registry->lock);
d88aee68 5497error_rcu_unlock:
d0b96690 5498 rcu_read_unlock();
5e98072f 5499 free(fields);
d0b96690
DG
5500 return ret;
5501}
5502
d88aee68
DG
5503/*
5504 * Add event to the UST channel registry. When the event is added to the
5505 * registry, the metadata is also created. Once done, this replies to the
5506 * application with the appropriate error code.
5507 *
5508 * The session UST registry lock is acquired in the function.
5509 *
5510 * On success 0 is returned else a negative value.
5511 */
d0b96690 5512static int add_event_ust_registry(int sock, int sobjd, int cobjd, char *name,
2106efa0
PP
5513 char *sig, size_t nr_fields, struct ustctl_field *fields,
5514 int loglevel_value, char *model_emf_uri)
d0b96690
DG
5515{
5516 int ret, ret_code;
5517 uint32_t event_id = 0;
7972aab2 5518 uint64_t chan_reg_key;
d0b96690
DG
5519 struct ust_app *app;
5520 struct ust_app_channel *ua_chan;
5521 struct ust_app_session *ua_sess;
7972aab2 5522 struct ust_registry_session *registry;
d0b96690
DG
5523
5524 rcu_read_lock();
5525
5526 /* Lookup application. If not found, there is a code flow error. */
5527 app = find_app_by_notify_sock(sock);
d88aee68 5528 if (!app) {
5e98072f 5529 DBG("Application socket %d is being torn down. Abort event notify",
d88aee68
DG
5530 sock);
5531 ret = 0;
5532 goto error_rcu_unlock;
5533 }
d0b96690 5534
4950b860 5535 /* Lookup channel by UST object descriptor. */
d0b96690 5536 ua_chan = find_channel_by_objd(app, cobjd);
4950b860 5537 if (!ua_chan) {
5e98072f 5538 DBG("Application channel is being torn down. Abort event notify");
4950b860
MD
5539 ret = 0;
5540 goto error_rcu_unlock;
5541 }
5542
d0b96690
DG
5543 assert(ua_chan->session);
5544 ua_sess = ua_chan->session;
5545
7972aab2 5546 registry = get_session_registry(ua_sess);
5e98072f
JR
5547 if (!registry) {
5548 DBG("Application session is being torn down. Abort event notify");
5549 ret = 0;
5550 goto error_rcu_unlock;
5551 }
7972aab2
DG
5552
5553 if (ua_sess->buffer_type == LTTNG_BUFFER_PER_UID) {
5554 chan_reg_key = ua_chan->tracing_channel_id;
5555 } else {
5556 chan_reg_key = ua_chan->key;
5557 }
5558
5559 pthread_mutex_lock(&registry->lock);
d0b96690 5560
d5d629b5
DG
5561 /*
5562 * From this point on, this call acquires the ownership of the sig, fields
5563 * and model_emf_uri meaning any free are done inside it if needed. These
5564 * three variables MUST NOT be read/write after this.
5565 */
7972aab2 5566 ret_code = ust_registry_create_event(registry, chan_reg_key,
2106efa0
PP
5567 sobjd, cobjd, name, sig, nr_fields, fields,
5568 loglevel_value, model_emf_uri, ua_sess->buffer_type,
5569 &event_id, app);
5e98072f
JR
5570 sig = NULL;
5571 fields = NULL;
5572 model_emf_uri = NULL;
d0b96690
DG
5573
5574 /*
5575 * The return value is returned to ustctl so in case of an error, the
5576 * application can be notified. In case of an error, it's important not to
5577 * return a negative error or else the application will get closed.
5578 */
5579 ret = ustctl_reply_register_event(sock, event_id, ret_code);
5580 if (ret < 0) {
5581 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
5582 ERR("UST app reply event failed with ret %d", ret);
5583 } else {
5584 DBG3("UST app reply event failed. Application died");
5585 }
5586 /*
5587 * No need to wipe the create event since the application socket will
5588 * get close on error hence cleaning up everything by itself.
5589 */
5590 goto error;
5591 }
5592
7972aab2
DG
5593 DBG3("UST registry event %s with id %" PRId32 " added successfully",
5594 name, event_id);
d88aee68 5595
d0b96690 5596error:
7972aab2 5597 pthread_mutex_unlock(&registry->lock);
d88aee68 5598error_rcu_unlock:
d0b96690 5599 rcu_read_unlock();
5e98072f
JR
5600 free(sig);
5601 free(fields);
5602 free(model_emf_uri);
d0b96690
DG
5603 return ret;
5604}
5605
10b56aef
MD
5606/*
5607 * Add enum to the UST session registry. Once done, this replies to the
5608 * application with the appropriate error code.
5609 *
5610 * The session UST registry lock is acquired within this function.
5611 *
5612 * On success 0 is returned else a negative value.
5613 */
5614static int add_enum_ust_registry(int sock, int sobjd, char *name,
5615 struct ustctl_enum_entry *entries, size_t nr_entries)
5616{
5617 int ret = 0, ret_code;
5618 struct ust_app *app;
5619 struct ust_app_session *ua_sess;
5620 struct ust_registry_session *registry;
5621 uint64_t enum_id = -1ULL;
5622
5623 rcu_read_lock();
5624
5625 /* Lookup application. If not found, there is a code flow error. */
5626 app = find_app_by_notify_sock(sock);
5627 if (!app) {
5628 /* Return an error since this is not an error */
5629 DBG("Application socket %d is being torn down. Aborting enum registration",
5630 sock);
5631 free(entries);
5632 goto error_rcu_unlock;
5633 }
5634
5635 /* Lookup session by UST object descriptor. */
5636 ua_sess = find_session_by_objd(app, sobjd);
5637 if (!ua_sess) {
5638 /* Return an error since this is not an error */
5e98072f 5639 DBG("Application session is being torn down (session not found). Aborting enum registration.");
10b56aef
MD
5640 free(entries);
5641 goto error_rcu_unlock;
5642 }
5643
5644 registry = get_session_registry(ua_sess);
5e98072f
JR
5645 if (!registry) {
5646 DBG("Application session is being torn down (registry not found). Aborting enum registration.");
5647 free(entries);
5648 goto error_rcu_unlock;
5649 }
10b56aef
MD
5650
5651 pthread_mutex_lock(&registry->lock);
5652
5653 /*
5654 * From this point on, the callee acquires the ownership of
5655 * entries. The variable entries MUST NOT be read/written after
5656 * call.
5657 */
5658 ret_code = ust_registry_create_or_find_enum(registry, sobjd, name,
5659 entries, nr_entries, &enum_id);
5660 entries = NULL;
5661
5662 /*
5663 * The return value is returned to ustctl so in case of an error, the
5664 * application can be notified. In case of an error, it's important not to
5665 * return a negative error or else the application will get closed.
5666 */
5667 ret = ustctl_reply_register_enum(sock, enum_id, ret_code);
5668 if (ret < 0) {
5669 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
5670 ERR("UST app reply enum failed with ret %d", ret);
5671 } else {
5672 DBG3("UST app reply enum failed. Application died");
5673 }
5674 /*
5675 * No need to wipe the create enum since the application socket will
5676 * get close on error hence cleaning up everything by itself.
5677 */
5678 goto error;
5679 }
5680
5681 DBG3("UST registry enum %s added successfully or already found", name);
5682
5683error:
5684 pthread_mutex_unlock(&registry->lock);
5685error_rcu_unlock:
5686 rcu_read_unlock();
5687 return ret;
5688}
5689
d88aee68
DG
5690/*
5691 * Handle application notification through the given notify socket.
5692 *
5693 * Return 0 on success or else a negative value.
5694 */
d0b96690
DG
5695int ust_app_recv_notify(int sock)
5696{
5697 int ret;
5698 enum ustctl_notify_cmd cmd;
5699
5700 DBG3("UST app receiving notify from sock %d", sock);
5701
5702 ret = ustctl_recv_notify(sock, &cmd);
5703 if (ret < 0) {
5704 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
5705 ERR("UST app recv notify failed with ret %d", ret);
5706 } else {
5707 DBG3("UST app recv notify failed. Application died");
5708 }
5709 goto error;
5710 }
5711
5712 switch (cmd) {
5713 case USTCTL_NOTIFY_CMD_EVENT:
5714 {
2106efa0 5715 int sobjd, cobjd, loglevel_value;
d0b96690
DG
5716 char name[LTTNG_UST_SYM_NAME_LEN], *sig, *model_emf_uri;
5717 size_t nr_fields;
5718 struct ustctl_field *fields;
5719
5720 DBG2("UST app ustctl register event received");
5721
2106efa0
PP
5722 ret = ustctl_recv_register_event(sock, &sobjd, &cobjd, name,
5723 &loglevel_value, &sig, &nr_fields, &fields,
5724 &model_emf_uri);
d0b96690
DG
5725 if (ret < 0) {
5726 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
5727 ERR("UST app recv event failed with ret %d", ret);
5728 } else {
5729 DBG3("UST app recv event failed. Application died");
5730 }
5731 goto error;
5732 }
5733
d5d629b5
DG
5734 /*
5735 * Add event to the UST registry coming from the notify socket. This
5736 * call will free if needed the sig, fields and model_emf_uri. This
5737 * code path loses the ownsership of these variables and transfer them
5738 * to the this function.
5739 */
d0b96690 5740 ret = add_event_ust_registry(sock, sobjd, cobjd, name, sig, nr_fields,
2106efa0 5741 fields, loglevel_value, model_emf_uri);
d0b96690
DG
5742 if (ret < 0) {
5743 goto error;
5744 }
5745
5746 break;
5747 }
5748 case USTCTL_NOTIFY_CMD_CHANNEL:
5749 {
5750 int sobjd, cobjd;
5751 size_t nr_fields;
5752 struct ustctl_field *fields;
5753
5754 DBG2("UST app ustctl register channel received");
5755
5756 ret = ustctl_recv_register_channel(sock, &sobjd, &cobjd, &nr_fields,
5757 &fields);
5758 if (ret < 0) {
5759 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
5760 ERR("UST app recv channel failed with ret %d", ret);
5761 } else {
5762 DBG3("UST app recv channel failed. Application died");
5763 }
5764 goto error;
5765 }
5766
d5d629b5
DG
5767 /*
5768 * The fields ownership are transfered to this function call meaning
5769 * that if needed it will be freed. After this, it's invalid to access
5770 * fields or clean it up.
5771 */
d0b96690
DG
5772 ret = reply_ust_register_channel(sock, sobjd, cobjd, nr_fields,
5773 fields);
5774 if (ret < 0) {
5775 goto error;
5776 }
5777
5778 break;
5779 }
10b56aef
MD
5780 case USTCTL_NOTIFY_CMD_ENUM:
5781 {
5782 int sobjd;
5783 char name[LTTNG_UST_SYM_NAME_LEN];
5784 size_t nr_entries;
5785 struct ustctl_enum_entry *entries;
5786
5787 DBG2("UST app ustctl register enum received");
5788
5789 ret = ustctl_recv_register_enum(sock, &sobjd, name,
5790 &entries, &nr_entries);
5791 if (ret < 0) {
5792 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
5793 ERR("UST app recv enum failed with ret %d", ret);
5794 } else {
5795 DBG3("UST app recv enum failed. Application died");
5796 }
5797 goto error;
5798 }
5799
5800 /* Callee assumes ownership of entries */
5801 ret = add_enum_ust_registry(sock, sobjd, name,
5802 entries, nr_entries);
5803 if (ret < 0) {
5804 goto error;
5805 }
5806
5807 break;
5808 }
d0b96690
DG
5809 default:
5810 /* Should NEVER happen. */
5811 assert(0);
5812 }
5813
5814error:
5815 return ret;
5816}
d88aee68
DG
5817
5818/*
5819 * Once the notify socket hangs up, this is called. First, it tries to find the
5820 * corresponding application. On failure, the call_rcu to close the socket is
5821 * executed. If an application is found, it tries to delete it from the notify
5822 * socket hash table. Whathever the result, it proceeds to the call_rcu.
5823 *
5824 * Note that an object needs to be allocated here so on ENOMEM failure, the
5825 * call RCU is not done but the rest of the cleanup is.
5826 */
5827void ust_app_notify_sock_unregister(int sock)
5828{
5829 int err_enomem = 0;
5830 struct lttng_ht_iter iter;
5831 struct ust_app *app;
5832 struct ust_app_notify_sock_obj *obj;
5833
5834 assert(sock >= 0);
5835
5836 rcu_read_lock();
5837
5838 obj = zmalloc(sizeof(*obj));
5839 if (!obj) {
5840 /*
5841 * An ENOMEM is kind of uncool. If this strikes we continue the
5842 * procedure but the call_rcu will not be called. In this case, we
5843 * accept the fd leak rather than possibly creating an unsynchronized
5844 * state between threads.
5845 *
5846 * TODO: The notify object should be created once the notify socket is
5847 * registered and stored independantely from the ust app object. The
5848 * tricky part is to synchronize the teardown of the application and
5849 * this notify object. Let's keep that in mind so we can avoid this
5850 * kind of shenanigans with ENOMEM in the teardown path.
5851 */
5852 err_enomem = 1;
5853 } else {
5854 obj->fd = sock;
5855 }
5856
5857 DBG("UST app notify socket unregister %d", sock);
5858
5859 /*
5860 * Lookup application by notify socket. If this fails, this means that the
5861 * hash table delete has already been done by the application
5862 * unregistration process so we can safely close the notify socket in a
5863 * call RCU.
5864 */
5865 app = find_app_by_notify_sock(sock);
5866 if (!app) {
5867 goto close_socket;
5868 }
5869
5870 iter.iter.node = &app->notify_sock_n.node;
5871
5872 /*
5873 * Whatever happens here either we fail or succeed, in both cases we have
5874 * to close the socket after a grace period to continue to the call RCU
5875 * here. If the deletion is successful, the application is not visible
5876 * anymore by other threads and is it fails it means that it was already
5877 * deleted from the hash table so either way we just have to close the
5878 * socket.
5879 */
5880 (void) lttng_ht_del(ust_app_ht_by_notify_sock, &iter);
5881
5882close_socket:
5883 rcu_read_unlock();
5884
5885 /*
5886 * Close socket after a grace period to avoid for the socket to be reused
5887 * before the application object is freed creating potential race between
5888 * threads trying to add unique in the global hash table.
5889 */
5890 if (!err_enomem) {
5891 call_rcu(&obj->head, close_notify_sock_rcu);
5892 }
5893}
f45e313d
DG
5894
5895/*
5896 * Destroy a ust app data structure and free its memory.
5897 */
5898void ust_app_destroy(struct ust_app *app)
5899{
5900 if (!app) {
5901 return;
5902 }
5903
5904 call_rcu(&app->pid_n.head, delete_ust_app_rcu);
5905}
6dc3064a
DG
5906
5907/*
5908 * Take a snapshot for a given UST session. The snapshot is sent to the given
5909 * output.
5910 *
5911 * Return 0 on success or else a negative value.
5912 */
5913int ust_app_snapshot_record(struct ltt_ust_session *usess,
d07ceecd
MD
5914 struct snapshot_output *output, int wait,
5915 uint64_t nb_packets_per_stream)
6dc3064a
DG
5916{
5917 int ret = 0;
5918 struct lttng_ht_iter iter;
5919 struct ust_app *app;
af706bb7 5920 char pathname[PATH_MAX];
6dc3064a
DG
5921
5922 assert(usess);
5923 assert(output);
5924
5925 rcu_read_lock();
5926
8c924c7b
MD
5927 switch (usess->buffer_type) {
5928 case LTTNG_BUFFER_PER_UID:
5929 {
5930 struct buffer_reg_uid *reg;
6dc3064a 5931
8c924c7b
MD
5932 cds_list_for_each_entry(reg, &usess->buffer_reg_uid_list, lnode) {
5933 struct buffer_reg_channel *reg_chan;
5934 struct consumer_socket *socket;
6dc3064a 5935
a848acfd
JR
5936 if (!reg->registry->reg.ust->metadata_key) {
5937 /* Skip since no metadata is present */
5938 continue;
5939 }
5940
8c924c7b
MD
5941 /* Get consumer socket to use to push the metadata.*/
5942 socket = consumer_find_socket_by_bitness(reg->bits_per_long,
5943 usess->consumer);
5944 if (!socket) {
5945 ret = -EINVAL;
5946 goto error;
5947 }
6dc3064a 5948
8c924c7b
MD
5949 memset(pathname, 0, sizeof(pathname));
5950 ret = snprintf(pathname, sizeof(pathname),
5951 DEFAULT_UST_TRACE_DIR "/" DEFAULT_UST_TRACE_UID_PATH,
5952 reg->uid, reg->bits_per_long);
5953 if (ret < 0) {
5954 PERROR("snprintf snapshot path");
5955 goto error;
5956 }
5957
5958 /* Add the UST default trace dir to path. */
5959 cds_lfht_for_each_entry(reg->registry->channels->ht, &iter.iter,
5960 reg_chan, node.node) {
68808f4e
DG
5961 ret = consumer_snapshot_channel(socket, reg_chan->consumer_key,
5962 output, 0, usess->uid, usess->gid, pathname, wait,
d07ceecd 5963 nb_packets_per_stream);
8c924c7b
MD
5964 if (ret < 0) {
5965 goto error;
5966 }
5967 }
68808f4e
DG
5968 ret = consumer_snapshot_channel(socket,
5969 reg->registry->reg.ust->metadata_key, output, 1,
d07ceecd 5970 usess->uid, usess->gid, pathname, wait, 0);
8c924c7b
MD
5971 if (ret < 0) {
5972 goto error;
5973 }
af706bb7 5974 }
8c924c7b
MD
5975 break;
5976 }
5977 case LTTNG_BUFFER_PER_PID:
5978 {
5979 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
5980 struct consumer_socket *socket;
5981 struct lttng_ht_iter chan_iter;
5982 struct ust_app_channel *ua_chan;
5983 struct ust_app_session *ua_sess;
5984 struct ust_registry_session *registry;
5985
5986 ua_sess = lookup_session_by_app(usess, app);
5987 if (!ua_sess) {
5988 /* Session not associated with this app. */
5989 continue;
5990 }
af706bb7 5991
8c924c7b
MD
5992 /* Get the right consumer socket for the application. */
5993 socket = consumer_find_socket_by_bitness(app->bits_per_long,
5994 output->consumer);
5995 if (!socket) {
5c786ded 5996 ret = -EINVAL;
5c786ded
JD
5997 goto error;
5998 }
5999
8c924c7b
MD
6000 /* Add the UST default trace dir to path. */
6001 memset(pathname, 0, sizeof(pathname));
6002 ret = snprintf(pathname, sizeof(pathname), DEFAULT_UST_TRACE_DIR "/%s",
6003 ua_sess->path);
6dc3064a 6004 if (ret < 0) {
8c924c7b 6005 PERROR("snprintf snapshot path");
6dc3064a
DG
6006 goto error;
6007 }
6dc3064a 6008
8c924c7b
MD
6009 cds_lfht_for_each_entry(ua_sess->channels->ht, &chan_iter.iter,
6010 ua_chan, node.node) {
68808f4e
DG
6011 ret = consumer_snapshot_channel(socket, ua_chan->key, output,
6012 0, ua_sess->euid, ua_sess->egid, pathname, wait,
d07ceecd 6013 nb_packets_per_stream);
8c924c7b
MD
6014 if (ret < 0) {
6015 goto error;
6016 }
6017 }
6018
6019 registry = get_session_registry(ua_sess);
5e98072f
JR
6020 if (!registry) {
6021 DBG("Application session is being torn down. Abort snapshot record.");
6022 ret = -1;
6023 goto error;
6024 }
8c924c7b 6025 ret = consumer_snapshot_channel(socket, registry->metadata_key, output,
d07ceecd 6026 1, ua_sess->euid, ua_sess->egid, pathname, wait, 0);
8c924c7b
MD
6027 if (ret < 0) {
6028 goto error;
6029 }
6030 }
6031 break;
6032 }
6033 default:
6034 assert(0);
6035 break;
6dc3064a
DG
6036 }
6037
6038error:
6039 rcu_read_unlock();
6040 return ret;
6041}
5c786ded
JD
6042
6043/*
d07ceecd 6044 * Return the size taken by one more packet per stream.
5c786ded 6045 */
d07ceecd
MD
6046uint64_t ust_app_get_size_one_more_packet_per_stream(struct ltt_ust_session *usess,
6047 uint64_t cur_nr_packets)
5c786ded 6048{
d07ceecd 6049 uint64_t tot_size = 0;
5c786ded
JD
6050 struct ust_app *app;
6051 struct lttng_ht_iter iter;
6052
6053 assert(usess);
6054
6055 switch (usess->buffer_type) {
6056 case LTTNG_BUFFER_PER_UID:
6057 {
6058 struct buffer_reg_uid *reg;
6059
6060 cds_list_for_each_entry(reg, &usess->buffer_reg_uid_list, lnode) {
6061 struct buffer_reg_channel *reg_chan;
6062
b7064eaa 6063 rcu_read_lock();
5c786ded
JD
6064 cds_lfht_for_each_entry(reg->registry->channels->ht, &iter.iter,
6065 reg_chan, node.node) {
d07ceecd
MD
6066 if (cur_nr_packets >= reg_chan->num_subbuf) {
6067 /*
6068 * Don't take channel into account if we
6069 * already grab all its packets.
6070 */
6071 continue;
6072 }
6073 tot_size += reg_chan->subbuf_size * reg_chan->stream_count;
5c786ded 6074 }
b7064eaa 6075 rcu_read_unlock();
5c786ded
JD
6076 }
6077 break;
6078 }
6079 case LTTNG_BUFFER_PER_PID:
6080 {
6081 rcu_read_lock();
6082 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
6083 struct ust_app_channel *ua_chan;
6084 struct ust_app_session *ua_sess;
6085 struct lttng_ht_iter chan_iter;
6086
6087 ua_sess = lookup_session_by_app(usess, app);
6088 if (!ua_sess) {
6089 /* Session not associated with this app. */
6090 continue;
6091 }
6092
6093 cds_lfht_for_each_entry(ua_sess->channels->ht, &chan_iter.iter,
6094 ua_chan, node.node) {
d07ceecd
MD
6095 if (cur_nr_packets >= ua_chan->attr.num_subbuf) {
6096 /*
6097 * Don't take channel into account if we
6098 * already grab all its packets.
6099 */
6100 continue;
6101 }
6102 tot_size += ua_chan->attr.subbuf_size * ua_chan->streams.count;
5c786ded
JD
6103 }
6104 }
6105 rcu_read_unlock();
6106 break;
6107 }
6108 default:
6109 assert(0);
6110 break;
6111 }
6112
d07ceecd 6113 return tot_size;
5c786ded 6114}
fb83fe64
JD
6115
6116int ust_app_uid_get_channel_runtime_stats(uint64_t ust_session_id,
6117 struct cds_list_head *buffer_reg_uid_list,
6118 struct consumer_output *consumer, uint64_t uchan_id,
6119 int overwrite, uint64_t *discarded, uint64_t *lost)
6120{
6121 int ret;
6122 uint64_t consumer_chan_key;
6123
b1234e71
MD
6124 *discarded = 0;
6125 *lost = 0;
6126
fb83fe64
JD
6127 ret = buffer_reg_uid_consumer_channel_key(
6128 buffer_reg_uid_list, ust_session_id,
6129 uchan_id, &consumer_chan_key);
6130 if (ret < 0) {
b1234e71
MD
6131 /* Not found */
6132 ret = 0;
fb83fe64
JD
6133 goto end;
6134 }
6135
6136 if (overwrite) {
6137 ret = consumer_get_lost_packets(ust_session_id,
6138 consumer_chan_key, consumer, lost);
6139 } else {
6140 ret = consumer_get_discarded_events(ust_session_id,
6141 consumer_chan_key, consumer, discarded);
6142 }
6143
6144end:
6145 return ret;
6146}
6147
6148int ust_app_pid_get_channel_runtime_stats(struct ltt_ust_session *usess,
6149 struct ltt_ust_channel *uchan,
6150 struct consumer_output *consumer, int overwrite,
6151 uint64_t *discarded, uint64_t *lost)
6152{
6153 int ret = 0;
6154 struct lttng_ht_iter iter;
6155 struct lttng_ht_node_str *ua_chan_node;
6156 struct ust_app *app;
6157 struct ust_app_session *ua_sess;
6158 struct ust_app_channel *ua_chan;
6159
b1234e71
MD
6160 *discarded = 0;
6161 *lost = 0;
6162
fb83fe64
JD
6163 rcu_read_lock();
6164 /*
b1234e71
MD
6165 * Iterate over every registered applications. Sum counters for
6166 * all applications containing requested session and channel.
fb83fe64
JD
6167 */
6168 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
6169 struct lttng_ht_iter uiter;
6170
6171 ua_sess = lookup_session_by_app(usess, app);
6172 if (ua_sess == NULL) {
6173 continue;
6174 }
6175
6176 /* Get channel */
ee022399 6177 lttng_ht_lookup(ua_sess->channels, (void *) uchan->name, &uiter);
fb83fe64
JD
6178 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
6179 /* If the session is found for the app, the channel must be there */
6180 assert(ua_chan_node);
6181
6182 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
6183
6184 if (overwrite) {
b1234e71
MD
6185 uint64_t _lost;
6186
fb83fe64 6187 ret = consumer_get_lost_packets(usess->id, ua_chan->key,
b1234e71
MD
6188 consumer, &_lost);
6189 if (ret < 0) {
6190 break;
6191 }
6192 (*lost) += _lost;
fb83fe64 6193 } else {
b1234e71
MD
6194 uint64_t _discarded;
6195
fb83fe64 6196 ret = consumer_get_discarded_events(usess->id,
b1234e71
MD
6197 ua_chan->key, consumer, &_discarded);
6198 if (ret < 0) {
6199 break;
6200 }
6201 (*discarded) += _discarded;
fb83fe64 6202 }
fb83fe64
JD
6203 }
6204
fb83fe64
JD
6205 rcu_read_unlock();
6206 return ret;
6207}
c2561365
JD
6208
6209static
6210int ust_app_regenerate_statedump(struct ltt_ust_session *usess,
6211 struct ust_app *app)
6212{
6213 int ret = 0;
6214 struct ust_app_session *ua_sess;
6215
6216 DBG("Regenerating the metadata for ust app pid %d", app->pid);
6217
6218 rcu_read_lock();
6219
6220 ua_sess = lookup_session_by_app(usess, app);
6221 if (ua_sess == NULL) {
6222 /* The session is in teardown process. Ignore and continue. */
6223 goto end;
6224 }
6225
6226 pthread_mutex_lock(&ua_sess->lock);
6227
6228 if (ua_sess->deleted) {
6229 goto end_unlock;
6230 }
6231
6232 pthread_mutex_lock(&app->sock_lock);
6233 ret = ustctl_regenerate_statedump(app->sock, ua_sess->handle);
6234 pthread_mutex_unlock(&app->sock_lock);
6235
6236end_unlock:
6237 pthread_mutex_unlock(&ua_sess->lock);
6238
6239end:
6240 rcu_read_unlock();
6241 health_code_update();
6242 return ret;
6243}
6244
6245/*
6246 * Regenerate the statedump for each app in the session.
6247 */
6248int ust_app_regenerate_statedump_all(struct ltt_ust_session *usess)
6249{
6250 int ret = 0;
6251 struct lttng_ht_iter iter;
6252 struct ust_app *app;
6253
6254 DBG("Regenerating the metadata for all UST apps");
6255
6256 rcu_read_lock();
6257
6258 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
6259 if (!app->compatible) {
6260 continue;
6261 }
6262
6263 ret = ust_app_regenerate_statedump(usess, app);
6264 if (ret < 0) {
6265 /* Continue to the next app even on error */
6266 continue;
6267 }
6268 }
6269
6270 rcu_read_unlock();
6271
6272 return 0;
6273}
This page took 0.450373 seconds and 5 git commands to generate.