Fix: don't fail on push metadata if no channel
[lttng-tools.git] / src / bin / lttng-sessiond / ust-app.c
CommitLineData
91d76f53
DG
1/*
2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
3 *
d14d33bf
AM
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License, version 2 only,
6 * as published by the Free Software Foundation.
91d76f53
DG
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
d14d33bf
AM
13 * You should have received a copy of the GNU General Public License along
14 * with this program; if not, write to the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
91d76f53
DG
16 */
17
18#define _GNU_SOURCE
19#include <errno.h>
7972aab2 20#include <inttypes.h>
91d76f53
DG
21#include <pthread.h>
22#include <stdio.h>
23#include <stdlib.h>
099e26bd 24#include <string.h>
aba8e916
DG
25#include <sys/stat.h>
26#include <sys/types.h>
099e26bd 27#include <unistd.h>
0df502fd 28#include <urcu/compiler.h>
fb54cdbf 29#include <lttng/ust-error.h>
331744e3 30#include <signal.h>
bec39940 31
990570ed 32#include <common/common.h>
86acf0da 33#include <common/sessiond-comm/sessiond-comm.h>
1e307fab 34
7972aab2 35#include "buffer-registry.h"
86acf0da
DG
36#include "fd-limit.h"
37#include "health.h"
56fff090 38#include "ust-app.h"
48842b30 39#include "ust-consumer.h"
d80a6244 40#include "ust-ctl.h"
0b2dc8df 41#include "utils.h"
d80a6244 42
77c5447b
MD
43/* Next available channel key. Access under next_channel_key_lock. */
44static uint64_t _next_channel_key;
45static pthread_mutex_t next_channel_key_lock = PTHREAD_MUTEX_INITIALIZER;
46
47/* Next available session ID. Access under next_session_id_lock. */
48static uint64_t _next_session_id;
49static pthread_mutex_t next_session_id_lock = PTHREAD_MUTEX_INITIALIZER;
ffe60014
DG
50
51/*
77c5447b 52 * Return the incremented value of next_channel_key.
ffe60014 53 */
77c5447b 54static uint64_t get_next_channel_key(void)
ffe60014 55{
77c5447b
MD
56 uint64_t ret;
57
58 pthread_mutex_lock(&next_channel_key_lock);
59 ret = ++_next_channel_key;
60 pthread_mutex_unlock(&next_channel_key_lock);
61 return ret;
ffe60014
DG
62}
63
64/*
7972aab2 65 * Return the atomically incremented value of next_session_id.
ffe60014 66 */
77c5447b 67static uint64_t get_next_session_id(void)
ffe60014 68{
77c5447b
MD
69 uint64_t ret;
70
71 pthread_mutex_lock(&next_session_id_lock);
72 ret = ++_next_session_id;
73 pthread_mutex_unlock(&next_session_id_lock);
74 return ret;
ffe60014
DG
75}
76
d65d2de8
DG
77static void copy_channel_attr_to_ustctl(
78 struct ustctl_consumer_channel_attr *attr,
79 struct lttng_ust_channel_attr *uattr)
80{
81 /* Copy event attributes since the layout is different. */
82 attr->subbuf_size = uattr->subbuf_size;
83 attr->num_subbuf = uattr->num_subbuf;
84 attr->overwrite = uattr->overwrite;
85 attr->switch_timer_interval = uattr->switch_timer_interval;
86 attr->read_timer_interval = uattr->read_timer_interval;
87 attr->output = uattr->output;
88}
89
025faf73
DG
90/*
91 * Match function for the hash table lookup.
92 *
93 * It matches an ust app event based on three attributes which are the event
94 * name, the filter bytecode and the loglevel.
95 */
18eace3b
DG
96static int ht_match_ust_app_event(struct cds_lfht_node *node, const void *_key)
97{
98 struct ust_app_event *event;
99 const struct ust_app_ht_key *key;
100
101 assert(node);
102 assert(_key);
103
104 event = caa_container_of(node, struct ust_app_event, node.node);
105 key = _key;
106
107 /* Match the 3 elements of the key: name, filter and loglevel. */
108
109 /* Event name */
110 if (strncmp(event->attr.name, key->name, sizeof(event->attr.name)) != 0) {
111 goto no_match;
112 }
113
114 /* Event loglevel. */
115 if (event->attr.loglevel != key->loglevel) {
025faf73
DG
116 if (event->attr.loglevel_type == LTTNG_UST_LOGLEVEL_ALL
117 && key->loglevel == 0 && event->attr.loglevel == -1) {
118 /*
119 * Match is accepted. This is because on event creation, the
120 * loglevel is set to -1 if the event loglevel type is ALL so 0 and
121 * -1 are accepted for this loglevel type since 0 is the one set by
122 * the API when receiving an enable event.
123 */
124 } else {
125 goto no_match;
126 }
18eace3b
DG
127 }
128
129 /* One of the filters is NULL, fail. */
130 if ((key->filter && !event->filter) || (!key->filter && event->filter)) {
131 goto no_match;
132 }
133
025faf73
DG
134 if (key->filter && event->filter) {
135 /* Both filters exists, check length followed by the bytecode. */
136 if (event->filter->len != key->filter->len ||
137 memcmp(event->filter->data, key->filter->data,
138 event->filter->len) != 0) {
139 goto no_match;
140 }
18eace3b
DG
141 }
142
025faf73 143 /* Match. */
18eace3b
DG
144 return 1;
145
146no_match:
147 return 0;
18eace3b
DG
148}
149
025faf73
DG
150/*
151 * Unique add of an ust app event in the given ht. This uses the custom
152 * ht_match_ust_app_event match function and the event name as hash.
153 */
d0b96690 154static void add_unique_ust_app_event(struct ust_app_channel *ua_chan,
18eace3b
DG
155 struct ust_app_event *event)
156{
157 struct cds_lfht_node *node_ptr;
158 struct ust_app_ht_key key;
d0b96690 159 struct lttng_ht *ht;
18eace3b 160
d0b96690
DG
161 assert(ua_chan);
162 assert(ua_chan->events);
18eace3b
DG
163 assert(event);
164
d0b96690 165 ht = ua_chan->events;
18eace3b
DG
166 key.name = event->attr.name;
167 key.filter = event->filter;
168 key.loglevel = event->attr.loglevel;
169
170 node_ptr = cds_lfht_add_unique(ht->ht,
171 ht->hash_fct(event->node.key, lttng_ht_seed),
172 ht_match_ust_app_event, &key, &event->node.node);
173 assert(node_ptr == &event->node.node);
174}
175
d88aee68
DG
176/*
177 * Close the notify socket from the given RCU head object. This MUST be called
178 * through a call_rcu().
179 */
180static void close_notify_sock_rcu(struct rcu_head *head)
181{
182 int ret;
183 struct ust_app_notify_sock_obj *obj =
184 caa_container_of(head, struct ust_app_notify_sock_obj, head);
185
186 /* Must have a valid fd here. */
187 assert(obj->fd >= 0);
188
189 ret = close(obj->fd);
190 if (ret) {
191 ERR("close notify sock %d RCU", obj->fd);
192 }
193 lttng_fd_put(LTTNG_FD_APPS, 1);
194
195 free(obj);
196}
197
7972aab2
DG
198/*
199 * Return the session registry according to the buffer type of the given
200 * session.
201 *
202 * A registry per UID object MUST exists before calling this function or else
203 * it assert() if not found. RCU read side lock must be acquired.
204 */
205static struct ust_registry_session *get_session_registry(
206 struct ust_app_session *ua_sess)
207{
208 struct ust_registry_session *registry = NULL;
209
210 assert(ua_sess);
211
212 switch (ua_sess->buffer_type) {
213 case LTTNG_BUFFER_PER_PID:
214 {
215 struct buffer_reg_pid *reg_pid = buffer_reg_pid_find(ua_sess->id);
216 if (!reg_pid) {
217 goto error;
218 }
219 registry = reg_pid->registry->reg.ust;
220 break;
221 }
222 case LTTNG_BUFFER_PER_UID:
223 {
224 struct buffer_reg_uid *reg_uid = buffer_reg_uid_find(
225 ua_sess->tracing_id, ua_sess->bits_per_long, ua_sess->uid);
226 if (!reg_uid) {
227 goto error;
228 }
229 registry = reg_uid->registry->reg.ust;
230 break;
231 }
232 default:
233 assert(0);
234 };
235
236error:
237 return registry;
238}
239
55cc08a6
DG
240/*
241 * Delete ust context safely. RCU read lock must be held before calling
242 * this function.
243 */
244static
245void delete_ust_app_ctx(int sock, struct ust_app_ctx *ua_ctx)
246{
ffe60014
DG
247 int ret;
248
249 assert(ua_ctx);
250
55cc08a6 251 if (ua_ctx->obj) {
ffe60014 252 ret = ustctl_release_object(sock, ua_ctx->obj);
d0b96690
DG
253 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
254 ERR("UST app sock %d release ctx obj handle %d failed with ret %d",
255 sock, ua_ctx->obj->handle, ret);
ffe60014 256 }
55cc08a6
DG
257 free(ua_ctx->obj);
258 }
259 free(ua_ctx);
260}
261
d80a6244
DG
262/*
263 * Delete ust app event safely. RCU read lock must be held before calling
264 * this function.
265 */
8b366481
DG
266static
267void delete_ust_app_event(int sock, struct ust_app_event *ua_event)
d80a6244 268{
ffe60014
DG
269 int ret;
270
271 assert(ua_event);
272
53a80697 273 free(ua_event->filter);
d80a6244 274
edb67388 275 if (ua_event->obj != NULL) {
ffe60014
DG
276 ret = ustctl_release_object(sock, ua_event->obj);
277 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
278 ERR("UST app sock %d release event obj failed with ret %d",
279 sock, ret);
280 }
edb67388
DG
281 free(ua_event->obj);
282 }
d80a6244
DG
283 free(ua_event);
284}
285
286/*
7972aab2
DG
287 * Release ust data object of the given stream.
288 *
289 * Return 0 on success or else a negative value.
d80a6244 290 */
7972aab2 291static int release_ust_app_stream(int sock, struct ust_app_stream *stream)
d80a6244 292{
7972aab2 293 int ret = 0;
ffe60014
DG
294
295 assert(stream);
296
8b366481 297 if (stream->obj) {
ffe60014
DG
298 ret = ustctl_release_object(sock, stream->obj);
299 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
300 ERR("UST app sock %d release stream obj failed with ret %d",
301 sock, ret);
302 }
4063050c 303 lttng_fd_put(LTTNG_FD_APPS, 2);
8b366481
DG
304 free(stream->obj);
305 }
7972aab2
DG
306
307 return ret;
308}
309
310/*
311 * Delete ust app stream safely. RCU read lock must be held before calling
312 * this function.
313 */
314static
315void delete_ust_app_stream(int sock, struct ust_app_stream *stream)
316{
317 assert(stream);
318
319 (void) release_ust_app_stream(sock, stream);
84cd17c6 320 free(stream);
d80a6244
DG
321}
322
36b588ed
MD
323/*
324 * We need to execute ht_destroy outside of RCU read-side critical
0b2dc8df
MD
325 * section and outside of call_rcu thread, so we postpone its execution
326 * using ht_cleanup_push. It is simpler than to change the semantic of
327 * the many callers of delete_ust_app_session().
36b588ed
MD
328 */
329static
330void delete_ust_app_channel_rcu(struct rcu_head *head)
331{
332 struct ust_app_channel *ua_chan =
333 caa_container_of(head, struct ust_app_channel, rcu_head);
334
0b2dc8df
MD
335 ht_cleanup_push(ua_chan->ctx);
336 ht_cleanup_push(ua_chan->events);
36b588ed
MD
337 free(ua_chan);
338}
339
d80a6244
DG
340/*
341 * Delete ust app channel safely. RCU read lock must be held before calling
342 * this function.
343 */
8b366481 344static
d0b96690
DG
345void delete_ust_app_channel(int sock, struct ust_app_channel *ua_chan,
346 struct ust_app *app)
d80a6244
DG
347{
348 int ret;
bec39940 349 struct lttng_ht_iter iter;
d80a6244 350 struct ust_app_event *ua_event;
55cc08a6 351 struct ust_app_ctx *ua_ctx;
030a66fa 352 struct ust_app_stream *stream, *stmp;
7972aab2 353 struct ust_registry_session *registry;
d80a6244 354
ffe60014
DG
355 assert(ua_chan);
356
357 DBG3("UST app deleting channel %s", ua_chan->name);
358
55cc08a6 359 /* Wipe stream */
d80a6244 360 cds_list_for_each_entry_safe(stream, stmp, &ua_chan->streams.head, list) {
84cd17c6 361 cds_list_del(&stream->list);
d80a6244
DG
362 delete_ust_app_stream(sock, stream);
363 }
364
55cc08a6 365 /* Wipe context */
bec39940 366 cds_lfht_for_each_entry(ua_chan->ctx->ht, &iter.iter, ua_ctx, node.node) {
4623416f 367 cds_list_del(&ua_ctx->list);
bec39940 368 ret = lttng_ht_del(ua_chan->ctx, &iter);
55cc08a6
DG
369 assert(!ret);
370 delete_ust_app_ctx(sock, ua_ctx);
371 }
d80a6244 372
55cc08a6 373 /* Wipe events */
bec39940
DG
374 cds_lfht_for_each_entry(ua_chan->events->ht, &iter.iter, ua_event,
375 node.node) {
376 ret = lttng_ht_del(ua_chan->events, &iter);
525b0740 377 assert(!ret);
d80a6244
DG
378 delete_ust_app_event(sock, ua_event);
379 }
edb67388 380
5894f2c8
MD
381 if (ua_chan->session->buffer_type == LTTNG_BUFFER_PER_PID) {
382 /* Wipe and free registry from session registry. */
383 registry = get_session_registry(ua_chan->session);
384 if (registry) {
385 ust_registry_channel_del_free(registry, ua_chan->key);
386 }
7972aab2 387 }
d0b96690 388
edb67388 389 if (ua_chan->obj != NULL) {
d0b96690
DG
390 /* Remove channel from application UST object descriptor. */
391 iter.iter.node = &ua_chan->ust_objd_node.node;
392 lttng_ht_del(app->ust_objd, &iter);
ffe60014
DG
393 ret = ustctl_release_object(sock, ua_chan->obj);
394 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
395 ERR("UST app sock %d release channel obj failed with ret %d",
396 sock, ret);
397 }
7972aab2 398 lttng_fd_put(LTTNG_FD_APPS, 1);
edb67388
DG
399 free(ua_chan->obj);
400 }
36b588ed 401 call_rcu(&ua_chan->rcu_head, delete_ust_app_channel_rcu);
d80a6244
DG
402}
403
331744e3 404/*
1b532a60
DG
405 * Push metadata to consumer socket.
406 *
407 * The socket lock MUST be acquired.
408 * The ust app session lock MUST be acquired.
331744e3
JD
409 *
410 * On success, return the len of metadata pushed or else a negative value.
411 */
412ssize_t ust_app_push_metadata(struct ust_registry_session *registry,
413 struct consumer_socket *socket, int send_zero_data)
414{
415 int ret;
416 char *metadata_str = NULL;
417 size_t len, offset;
418 ssize_t ret_val;
419
420 assert(registry);
421 assert(socket);
1b532a60
DG
422
423 /*
424 * On a push metadata error either the consumer is dead or the metadata
425 * channel has been destroyed because its endpoint might have died (e.g:
426 * relayd). If so, the metadata closed flag is set to 1 so we deny pushing
427 * metadata again which is not valid anymore on the consumer side.
428 *
429 * The ust app session mutex locked allows us to make this check without
430 * the registry lock.
431 */
432 if (registry->metadata_closed) {
433 return -EPIPE;
434 }
331744e3
JD
435
436 pthread_mutex_lock(&registry->lock);
437
438 offset = registry->metadata_len_sent;
439 len = registry->metadata_len - registry->metadata_len_sent;
440 if (len == 0) {
441 DBG3("No metadata to push for metadata key %" PRIu64,
442 registry->metadata_key);
443 ret_val = len;
444 if (send_zero_data) {
445 DBG("No metadata to push");
446 goto push_data;
447 }
448 goto end;
449 }
450
451 /* Allocate only what we have to send. */
452 metadata_str = zmalloc(len);
453 if (!metadata_str) {
454 PERROR("zmalloc ust app metadata string");
455 ret_val = -ENOMEM;
456 goto error;
457 }
458 /* Copy what we haven't send out. */
459 memcpy(metadata_str, registry->metadata + offset, len);
460 registry->metadata_len_sent += len;
461
462push_data:
463 pthread_mutex_unlock(&registry->lock);
464 ret = consumer_push_metadata(socket, registry->metadata_key,
465 metadata_str, len, offset);
466 if (ret < 0) {
890d27f4
DG
467 /*
468 * There is an acceptable race here between the registry metadata key
469 * assignment and the creation on the consumer. The session daemon can
470 * concurrently push metadata for this registry while being created on
471 * the consumer since the metadata key of the registry is assigned
472 * *before* it is setup to avoid the consumer to ask for metadata that
473 * could possibly be not found in the session daemon.
474 *
475 * The metadata will get pushed either by the session being stopped or
476 * the consumer requesting metadata if that race is triggered.
477 */
478 if (ret == -LTTCOMM_CONSUMERD_CHANNEL_FAIL) {
479 ret = 0;
480 }
331744e3
JD
481 ret_val = ret;
482 goto error_push;
483 }
484
485 free(metadata_str);
486 return len;
487
488end:
489error:
490 pthread_mutex_unlock(&registry->lock);
491error_push:
492 free(metadata_str);
493 return ret_val;
494}
495
d88aee68
DG
496/*
497 * For a given application and session, push metadata to consumer. The session
498 * lock MUST be acquired here before calling this.
331744e3
JD
499 * Either sock or consumer is required : if sock is NULL, the default
500 * socket to send the metadata is retrieved from consumer, if sock
501 * is not NULL we use it to send the metadata.
d88aee68
DG
502 *
503 * Return 0 on success else a negative error.
504 */
7972aab2
DG
505static int push_metadata(struct ust_registry_session *registry,
506 struct consumer_output *consumer)
d88aee68 507{
331744e3
JD
508 int ret_val;
509 ssize_t ret;
d88aee68
DG
510 struct consumer_socket *socket;
511
7972aab2
DG
512 assert(registry);
513 assert(consumer);
514
515 rcu_read_lock();
d88aee68 516
7972aab2
DG
517 /*
518 * Means that no metadata was assigned to the session. This can happens if
519 * no start has been done previously.
520 */
521 if (!registry->metadata_key) {
331744e3 522 ret_val = 0;
1b532a60 523 goto end_rcu_unlock;
d88aee68
DG
524 }
525
d88aee68 526 /* Get consumer socket to use to push the metadata.*/
7972aab2
DG
527 socket = consumer_find_socket_by_bitness(registry->bits_per_long,
528 consumer);
d88aee68 529 if (!socket) {
331744e3 530 ret_val = -1;
d88aee68
DG
531 goto error_rcu_unlock;
532 }
533
534 /*
535 * TODO: Currently, we hold the socket lock around sampling of the next
536 * metadata segment to ensure we send metadata over the consumer socket in
537 * the correct order. This makes the registry lock nest inside the socket
538 * lock.
539 *
540 * Please note that this is a temporary measure: we should move this lock
541 * back into ust_consumer_push_metadata() when the consumer gets the
542 * ability to reorder the metadata it receives.
543 */
544 pthread_mutex_lock(socket->lock);
331744e3
JD
545 ret = ust_app_push_metadata(registry, socket, 0);
546 pthread_mutex_unlock(socket->lock);
d88aee68 547 if (ret < 0) {
331744e3 548 ret_val = ret;
d88aee68
DG
549 goto error_rcu_unlock;
550 }
551
d88aee68 552 rcu_read_unlock();
d88aee68
DG
553 return 0;
554
d88aee68 555error_rcu_unlock:
1b532a60
DG
556 /*
557 * On error, flag the registry that the metadata is closed. We were unable
558 * to push anything and this means that either the consumer is not
559 * responding or the metadata cache has been destroyed on the consumer.
560 */
561 registry->metadata_closed = 1;
562end_rcu_unlock:
d88aee68 563 rcu_read_unlock();
331744e3 564 return ret_val;
d88aee68
DG
565}
566
567/*
568 * Send to the consumer a close metadata command for the given session. Once
569 * done, the metadata channel is deleted and the session metadata pointer is
570 * nullified. The session lock MUST be acquired here unless the application is
571 * in the destroy path.
572 *
573 * Return 0 on success else a negative value.
574 */
7972aab2
DG
575static int close_metadata(struct ust_registry_session *registry,
576 struct consumer_output *consumer)
d88aee68
DG
577{
578 int ret;
579 struct consumer_socket *socket;
580
7972aab2
DG
581 assert(registry);
582 assert(consumer);
d88aee68 583
7972aab2
DG
584 rcu_read_lock();
585
586 if (!registry->metadata_key || registry->metadata_closed) {
d88aee68 587 ret = 0;
1b532a60 588 goto end;
d88aee68
DG
589 }
590
d88aee68 591 /* Get consumer socket to use to push the metadata.*/
7972aab2
DG
592 socket = consumer_find_socket_by_bitness(registry->bits_per_long,
593 consumer);
d88aee68
DG
594 if (!socket) {
595 ret = -1;
7972aab2 596 goto error;
d88aee68
DG
597 }
598
7972aab2 599 ret = consumer_close_metadata(socket, registry->metadata_key);
d88aee68 600 if (ret < 0) {
7972aab2 601 goto error;
d88aee68
DG
602 }
603
d88aee68 604error:
1b532a60
DG
605 /*
606 * Metadata closed. Even on error this means that the consumer is not
607 * responding or not found so either way a second close should NOT be emit
608 * for this registry.
609 */
610 registry->metadata_closed = 1;
611end:
7972aab2 612 rcu_read_unlock();
d88aee68
DG
613 return ret;
614}
615
36b588ed
MD
616/*
617 * We need to execute ht_destroy outside of RCU read-side critical
0b2dc8df
MD
618 * section and outside of call_rcu thread, so we postpone its execution
619 * using ht_cleanup_push. It is simpler than to change the semantic of
620 * the many callers of delete_ust_app_session().
36b588ed
MD
621 */
622static
623void delete_ust_app_session_rcu(struct rcu_head *head)
624{
625 struct ust_app_session *ua_sess =
626 caa_container_of(head, struct ust_app_session, rcu_head);
627
0b2dc8df 628 ht_cleanup_push(ua_sess->channels);
36b588ed
MD
629 free(ua_sess);
630}
631
d80a6244
DG
632/*
633 * Delete ust app session safely. RCU read lock must be held before calling
634 * this function.
635 */
8b366481 636static
d0b96690
DG
637void delete_ust_app_session(int sock, struct ust_app_session *ua_sess,
638 struct ust_app *app)
d80a6244
DG
639{
640 int ret;
bec39940 641 struct lttng_ht_iter iter;
d80a6244 642 struct ust_app_channel *ua_chan;
7972aab2 643 struct ust_registry_session *registry;
d80a6244 644
d88aee68
DG
645 assert(ua_sess);
646
1b532a60
DG
647 pthread_mutex_lock(&ua_sess->lock);
648
7972aab2 649 registry = get_session_registry(ua_sess);
1b532a60 650 if (registry && !registry->metadata_closed) {
d88aee68 651 /* Push metadata for application before freeing the application. */
7972aab2 652 (void) push_metadata(registry, ua_sess->consumer);
d88aee68 653
7972aab2
DG
654 /*
655 * Don't ask to close metadata for global per UID buffers. Close
1b532a60
DG
656 * metadata only on destroy trace session in this case. Also, the
657 * previous push metadata could have flag the metadata registry to
658 * close so don't send a close command if closed.
7972aab2 659 */
1b532a60
DG
660 if (ua_sess->buffer_type != LTTNG_BUFFER_PER_UID &&
661 !registry->metadata_closed) {
7972aab2
DG
662 /* And ask to close it for this session registry. */
663 (void) close_metadata(registry, ua_sess->consumer);
664 }
d80a6244
DG
665 }
666
bec39940
DG
667 cds_lfht_for_each_entry(ua_sess->channels->ht, &iter.iter, ua_chan,
668 node.node) {
669 ret = lttng_ht_del(ua_sess->channels, &iter);
525b0740 670 assert(!ret);
d0b96690 671 delete_ust_app_channel(sock, ua_chan, app);
d80a6244 672 }
d80a6244 673
7972aab2
DG
674 /* In case of per PID, the registry is kept in the session. */
675 if (ua_sess->buffer_type == LTTNG_BUFFER_PER_PID) {
676 struct buffer_reg_pid *reg_pid = buffer_reg_pid_find(ua_sess->id);
677 if (reg_pid) {
678 buffer_reg_pid_remove(reg_pid);
679 buffer_reg_pid_destroy(reg_pid);
680 }
681 }
d0b96690 682
aee6bafd 683 if (ua_sess->handle != -1) {
ffe60014
DG
684 ret = ustctl_release_handle(sock, ua_sess->handle);
685 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
686 ERR("UST app sock %d release session handle failed with ret %d",
687 sock, ret);
688 }
aee6bafd 689 }
1b532a60
DG
690 pthread_mutex_unlock(&ua_sess->lock);
691
36b588ed 692 call_rcu(&ua_sess->rcu_head, delete_ust_app_session_rcu);
d80a6244 693}
91d76f53
DG
694
695/*
284d8f55
DG
696 * Delete a traceable application structure from the global list. Never call
697 * this function outside of a call_rcu call.
36b588ed
MD
698 *
699 * RCU read side lock should _NOT_ be held when calling this function.
91d76f53 700 */
8b366481
DG
701static
702void delete_ust_app(struct ust_app *app)
91d76f53 703{
8b366481 704 int ret, sock;
d42f20df 705 struct ust_app_session *ua_sess, *tmp_ua_sess;
44d3bd01 706
d80a6244 707 /* Delete ust app sessions info */
852d0037
DG
708 sock = app->sock;
709 app->sock = -1;
d80a6244 710
8b366481 711 /* Wipe sessions */
d42f20df
DG
712 cds_list_for_each_entry_safe(ua_sess, tmp_ua_sess, &app->teardown_head,
713 teardown_node) {
714 /* Free every object in the session and the session. */
36b588ed 715 rcu_read_lock();
d0b96690 716 delete_ust_app_session(sock, ua_sess, app);
36b588ed 717 rcu_read_unlock();
d80a6244 718 }
36b588ed 719
0b2dc8df
MD
720 ht_cleanup_push(app->sessions);
721 ht_cleanup_push(app->ust_objd);
d80a6244 722
6414a713 723 /*
852d0037
DG
724 * Wait until we have deleted the application from the sock hash table
725 * before closing this socket, otherwise an application could re-use the
726 * socket ID and race with the teardown, using the same hash table entry.
727 *
728 * It's OK to leave the close in call_rcu. We want it to stay unique for
729 * all RCU readers that could run concurrently with unregister app,
730 * therefore we _need_ to only close that socket after a grace period. So
731 * it should stay in this RCU callback.
732 *
733 * This close() is a very important step of the synchronization model so
734 * every modification to this function must be carefully reviewed.
6414a713 735 */
799e2c4f
MD
736 ret = close(sock);
737 if (ret) {
738 PERROR("close");
739 }
4063050c 740 lttng_fd_put(LTTNG_FD_APPS, 1);
d80a6244 741
852d0037 742 DBG2("UST app pid %d deleted", app->pid);
284d8f55 743 free(app);
099e26bd
DG
744}
745
746/*
f6a9efaa 747 * URCU intermediate call to delete an UST app.
099e26bd 748 */
8b366481
DG
749static
750void delete_ust_app_rcu(struct rcu_head *head)
099e26bd 751{
bec39940
DG
752 struct lttng_ht_node_ulong *node =
753 caa_container_of(head, struct lttng_ht_node_ulong, head);
f6a9efaa 754 struct ust_app *app =
852d0037 755 caa_container_of(node, struct ust_app, pid_n);
f6a9efaa 756
852d0037 757 DBG3("Call RCU deleting app PID %d", app->pid);
f6a9efaa 758 delete_ust_app(app);
099e26bd
DG
759}
760
ffe60014
DG
761/*
762 * Delete the session from the application ht and delete the data structure by
763 * freeing every object inside and releasing them.
764 */
d0b96690 765static void destroy_app_session(struct ust_app *app,
ffe60014
DG
766 struct ust_app_session *ua_sess)
767{
768 int ret;
769 struct lttng_ht_iter iter;
770
771 assert(app);
772 assert(ua_sess);
773
774 iter.iter.node = &ua_sess->node.node;
775 ret = lttng_ht_del(app->sessions, &iter);
776 if (ret) {
777 /* Already scheduled for teardown. */
778 goto end;
779 }
780
781 /* Once deleted, free the data structure. */
d0b96690 782 delete_ust_app_session(app->sock, ua_sess, app);
ffe60014
DG
783
784end:
785 return;
786}
787
8b366481
DG
788/*
789 * Alloc new UST app session.
790 */
791static
d0b96690 792struct ust_app_session *alloc_ust_app_session(struct ust_app *app)
8b366481
DG
793{
794 struct ust_app_session *ua_sess;
795
796 /* Init most of the default value by allocating and zeroing */
797 ua_sess = zmalloc(sizeof(struct ust_app_session));
798 if (ua_sess == NULL) {
799 PERROR("malloc");
ffe60014 800 goto error_free;
8b366481
DG
801 }
802
803 ua_sess->handle = -1;
bec39940 804 ua_sess->channels = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
d0b96690 805 pthread_mutex_init(&ua_sess->lock, NULL);
ffe60014 806
8b366481
DG
807 return ua_sess;
808
ffe60014 809error_free:
8b366481
DG
810 return NULL;
811}
812
813/*
814 * Alloc new UST app channel.
815 */
816static
817struct ust_app_channel *alloc_ust_app_channel(char *name,
d0b96690 818 struct ust_app_session *ua_sess,
ffe60014 819 struct lttng_ust_channel_attr *attr)
8b366481
DG
820{
821 struct ust_app_channel *ua_chan;
822
823 /* Init most of the default value by allocating and zeroing */
824 ua_chan = zmalloc(sizeof(struct ust_app_channel));
825 if (ua_chan == NULL) {
826 PERROR("malloc");
827 goto error;
828 }
829
830 /* Setup channel name */
831 strncpy(ua_chan->name, name, sizeof(ua_chan->name));
832 ua_chan->name[sizeof(ua_chan->name) - 1] = '\0';
833
834 ua_chan->enabled = 1;
835 ua_chan->handle = -1;
45893984 836 ua_chan->session = ua_sess;
ffe60014 837 ua_chan->key = get_next_channel_key();
bec39940
DG
838 ua_chan->ctx = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
839 ua_chan->events = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
840 lttng_ht_node_init_str(&ua_chan->node, ua_chan->name);
8b366481
DG
841
842 CDS_INIT_LIST_HEAD(&ua_chan->streams.head);
4623416f 843 CDS_INIT_LIST_HEAD(&ua_chan->ctx_list);
8b366481
DG
844
845 /* Copy attributes */
846 if (attr) {
ffe60014 847 /* Translate from lttng_ust_channel to ustctl_consumer_channel_attr. */
2fe6e7f5
DG
848 ua_chan->attr.subbuf_size = attr->subbuf_size;
849 ua_chan->attr.num_subbuf = attr->num_subbuf;
850 ua_chan->attr.overwrite = attr->overwrite;
851 ua_chan->attr.switch_timer_interval = attr->switch_timer_interval;
852 ua_chan->attr.read_timer_interval = attr->read_timer_interval;
853 ua_chan->attr.output = attr->output;
8b366481 854 }
ffe60014
DG
855 /* By default, the channel is a per cpu channel. */
856 ua_chan->attr.type = LTTNG_UST_CHAN_PER_CPU;
8b366481
DG
857
858 DBG3("UST app channel %s allocated", ua_chan->name);
859
860 return ua_chan;
861
862error:
863 return NULL;
864}
865
37f1c236
DG
866/*
867 * Allocate and initialize a UST app stream.
868 *
869 * Return newly allocated stream pointer or NULL on error.
870 */
ffe60014 871struct ust_app_stream *ust_app_alloc_stream(void)
37f1c236
DG
872{
873 struct ust_app_stream *stream = NULL;
874
875 stream = zmalloc(sizeof(*stream));
876 if (stream == NULL) {
877 PERROR("zmalloc ust app stream");
878 goto error;
879 }
880
881 /* Zero could be a valid value for a handle so flag it to -1. */
882 stream->handle = -1;
883
884error:
885 return stream;
886}
887
8b366481
DG
888/*
889 * Alloc new UST app event.
890 */
891static
892struct ust_app_event *alloc_ust_app_event(char *name,
893 struct lttng_ust_event *attr)
894{
895 struct ust_app_event *ua_event;
896
897 /* Init most of the default value by allocating and zeroing */
898 ua_event = zmalloc(sizeof(struct ust_app_event));
899 if (ua_event == NULL) {
900 PERROR("malloc");
901 goto error;
902 }
903
904 ua_event->enabled = 1;
905 strncpy(ua_event->name, name, sizeof(ua_event->name));
906 ua_event->name[sizeof(ua_event->name) - 1] = '\0';
bec39940 907 lttng_ht_node_init_str(&ua_event->node, ua_event->name);
8b366481
DG
908
909 /* Copy attributes */
910 if (attr) {
911 memcpy(&ua_event->attr, attr, sizeof(ua_event->attr));
912 }
913
914 DBG3("UST app event %s allocated", ua_event->name);
915
916 return ua_event;
917
918error:
919 return NULL;
920}
921
922/*
923 * Alloc new UST app context.
924 */
925static
926struct ust_app_ctx *alloc_ust_app_ctx(struct lttng_ust_context *uctx)
927{
928 struct ust_app_ctx *ua_ctx;
929
930 ua_ctx = zmalloc(sizeof(struct ust_app_ctx));
931 if (ua_ctx == NULL) {
932 goto error;
933 }
934
4623416f
DG
935 CDS_INIT_LIST_HEAD(&ua_ctx->list);
936
8b366481
DG
937 if (uctx) {
938 memcpy(&ua_ctx->ctx, uctx, sizeof(ua_ctx->ctx));
939 }
940
941 DBG3("UST app context %d allocated", ua_ctx->ctx.ctx);
942
943error:
944 return ua_ctx;
945}
946
025faf73
DG
947/*
948 * Allocate a filter and copy the given original filter.
949 *
950 * Return allocated filter or NULL on error.
951 */
952static struct lttng_ust_filter_bytecode *alloc_copy_ust_app_filter(
953 struct lttng_ust_filter_bytecode *orig_f)
954{
955 struct lttng_ust_filter_bytecode *filter = NULL;
956
957 /* Copy filter bytecode */
958 filter = zmalloc(sizeof(*filter) + orig_f->len);
959 if (!filter) {
960 PERROR("zmalloc alloc ust app filter");
961 goto error;
962 }
963
964 memcpy(filter, orig_f, sizeof(*filter) + orig_f->len);
965
966error:
967 return filter;
968}
969
099e26bd 970/*
421cb601
DG
971 * Find an ust_app using the sock and return it. RCU read side lock must be
972 * held before calling this helper function.
099e26bd 973 */
8b366481
DG
974static
975struct ust_app *find_app_by_sock(int sock)
099e26bd 976{
bec39940 977 struct lttng_ht_node_ulong *node;
bec39940 978 struct lttng_ht_iter iter;
f6a9efaa 979
852d0037 980 lttng_ht_lookup(ust_app_ht_by_sock, (void *)((unsigned long) sock), &iter);
bec39940 981 node = lttng_ht_iter_get_node_ulong(&iter);
f6a9efaa
DG
982 if (node == NULL) {
983 DBG2("UST app find by sock %d not found", sock);
f6a9efaa
DG
984 goto error;
985 }
852d0037
DG
986
987 return caa_container_of(node, struct ust_app, sock_n);
f6a9efaa
DG
988
989error:
990 return NULL;
099e26bd
DG
991}
992
d0b96690
DG
993/*
994 * Find an ust_app using the notify sock and return it. RCU read side lock must
995 * be held before calling this helper function.
996 */
997static struct ust_app *find_app_by_notify_sock(int sock)
998{
999 struct lttng_ht_node_ulong *node;
1000 struct lttng_ht_iter iter;
1001
1002 lttng_ht_lookup(ust_app_ht_by_notify_sock, (void *)((unsigned long) sock),
1003 &iter);
1004 node = lttng_ht_iter_get_node_ulong(&iter);
1005 if (node == NULL) {
1006 DBG2("UST app find by notify sock %d not found", sock);
1007 goto error;
1008 }
1009
1010 return caa_container_of(node, struct ust_app, notify_sock_n);
1011
1012error:
1013 return NULL;
1014}
1015
025faf73
DG
1016/*
1017 * Lookup for an ust app event based on event name, filter bytecode and the
1018 * event loglevel.
1019 *
1020 * Return an ust_app_event object or NULL on error.
1021 */
18eace3b
DG
1022static struct ust_app_event *find_ust_app_event(struct lttng_ht *ht,
1023 char *name, struct lttng_ust_filter_bytecode *filter, int loglevel)
1024{
1025 struct lttng_ht_iter iter;
1026 struct lttng_ht_node_str *node;
1027 struct ust_app_event *event = NULL;
1028 struct ust_app_ht_key key;
18eace3b
DG
1029
1030 assert(name);
1031 assert(ht);
1032
1033 /* Setup key for event lookup. */
1034 key.name = name;
1035 key.filter = filter;
1036 key.loglevel = loglevel;
1037
025faf73
DG
1038 /* Lookup using the event name as hash and a custom match fct. */
1039 cds_lfht_lookup(ht->ht, ht->hash_fct((void *) name, lttng_ht_seed),
1040 ht_match_ust_app_event, &key, &iter.iter);
18eace3b
DG
1041 node = lttng_ht_iter_get_node_str(&iter);
1042 if (node == NULL) {
1043 goto end;
1044 }
1045
1046 event = caa_container_of(node, struct ust_app_event, node);
1047
1048end:
18eace3b
DG
1049 return event;
1050}
1051
55cc08a6
DG
1052/*
1053 * Create the channel context on the tracer.
d0b96690
DG
1054 *
1055 * Called with UST app session lock held.
55cc08a6
DG
1056 */
1057static
1058int create_ust_channel_context(struct ust_app_channel *ua_chan,
1059 struct ust_app_ctx *ua_ctx, struct ust_app *app)
1060{
1061 int ret;
1062
840cb59c 1063 health_code_update();
86acf0da 1064
852d0037 1065 ret = ustctl_add_context(app->sock, &ua_ctx->ctx,
55cc08a6
DG
1066 ua_chan->obj, &ua_ctx->obj);
1067 if (ret < 0) {
ffe60014
DG
1068 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1069 ERR("UST app create channel context failed for app (pid: %d) "
1070 "with ret %d", app->pid, ret);
1071 } else {
47cb9d4a
DG
1072 /*
1073 * This is normal behavior, an application can die during the
1074 * creation process. Don't report an error so the execution can
1075 * continue normally.
1076 */
1077 ret = 0;
ffe60014
DG
1078 DBG3("UST app disable event failed. Application is dead.");
1079 }
55cc08a6
DG
1080 goto error;
1081 }
1082
1083 ua_ctx->handle = ua_ctx->obj->handle;
1084
d0b96690
DG
1085 DBG2("UST app context handle %d created successfully for channel %s",
1086 ua_ctx->handle, ua_chan->name);
55cc08a6
DG
1087
1088error:
840cb59c 1089 health_code_update();
55cc08a6
DG
1090 return ret;
1091}
1092
53a80697
MD
1093/*
1094 * Set the filter on the tracer.
1095 */
1096static
1097int set_ust_event_filter(struct ust_app_event *ua_event,
1098 struct ust_app *app)
1099{
1100 int ret;
1101
840cb59c 1102 health_code_update();
86acf0da 1103
53a80697 1104 if (!ua_event->filter) {
86acf0da
DG
1105 ret = 0;
1106 goto error;
53a80697
MD
1107 }
1108
1109 ret = ustctl_set_filter(app->sock, ua_event->filter,
1110 ua_event->obj);
1111 if (ret < 0) {
ffe60014
DG
1112 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1113 ERR("UST app event %s filter failed for app (pid: %d) "
1114 "with ret %d", ua_event->attr.name, app->pid, ret);
1115 } else {
47cb9d4a
DG
1116 /*
1117 * This is normal behavior, an application can die during the
1118 * creation process. Don't report an error so the execution can
1119 * continue normally.
1120 */
1121 ret = 0;
ffe60014
DG
1122 DBG3("UST app filter event failed. Application is dead.");
1123 }
53a80697
MD
1124 goto error;
1125 }
1126
1127 DBG2("UST filter set successfully for event %s", ua_event->name);
1128
1129error:
840cb59c 1130 health_code_update();
53a80697
MD
1131 return ret;
1132}
1133
9730260e
DG
1134/*
1135 * Disable the specified event on to UST tracer for the UST session.
1136 */
1137static int disable_ust_event(struct ust_app *app,
1138 struct ust_app_session *ua_sess, struct ust_app_event *ua_event)
1139{
1140 int ret;
1141
840cb59c 1142 health_code_update();
86acf0da 1143
852d0037 1144 ret = ustctl_disable(app->sock, ua_event->obj);
9730260e 1145 if (ret < 0) {
ffe60014
DG
1146 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1147 ERR("UST app event %s disable failed for app (pid: %d) "
1148 "and session handle %d with ret %d",
1149 ua_event->attr.name, app->pid, ua_sess->handle, ret);
1150 } else {
47cb9d4a
DG
1151 /*
1152 * This is normal behavior, an application can die during the
1153 * creation process. Don't report an error so the execution can
1154 * continue normally.
1155 */
1156 ret = 0;
ffe60014
DG
1157 DBG3("UST app disable event failed. Application is dead.");
1158 }
9730260e
DG
1159 goto error;
1160 }
1161
1162 DBG2("UST app event %s disabled successfully for app (pid: %d)",
852d0037 1163 ua_event->attr.name, app->pid);
9730260e
DG
1164
1165error:
840cb59c 1166 health_code_update();
9730260e
DG
1167 return ret;
1168}
1169
78f0bacd
DG
1170/*
1171 * Disable the specified channel on to UST tracer for the UST session.
1172 */
1173static int disable_ust_channel(struct ust_app *app,
1174 struct ust_app_session *ua_sess, struct ust_app_channel *ua_chan)
1175{
1176 int ret;
1177
840cb59c 1178 health_code_update();
86acf0da 1179
852d0037 1180 ret = ustctl_disable(app->sock, ua_chan->obj);
78f0bacd 1181 if (ret < 0) {
ffe60014
DG
1182 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1183 ERR("UST app channel %s disable failed for app (pid: %d) "
1184 "and session handle %d with ret %d",
1185 ua_chan->name, app->pid, ua_sess->handle, ret);
1186 } else {
47cb9d4a
DG
1187 /*
1188 * This is normal behavior, an application can die during the
1189 * creation process. Don't report an error so the execution can
1190 * continue normally.
1191 */
1192 ret = 0;
ffe60014
DG
1193 DBG3("UST app disable channel failed. Application is dead.");
1194 }
78f0bacd
DG
1195 goto error;
1196 }
1197
78f0bacd 1198 DBG2("UST app channel %s disabled successfully for app (pid: %d)",
852d0037 1199 ua_chan->name, app->pid);
78f0bacd
DG
1200
1201error:
840cb59c 1202 health_code_update();
78f0bacd
DG
1203 return ret;
1204}
1205
1206/*
1207 * Enable the specified channel on to UST tracer for the UST session.
1208 */
1209static int enable_ust_channel(struct ust_app *app,
1210 struct ust_app_session *ua_sess, struct ust_app_channel *ua_chan)
1211{
1212 int ret;
1213
840cb59c 1214 health_code_update();
86acf0da 1215
852d0037 1216 ret = ustctl_enable(app->sock, ua_chan->obj);
78f0bacd 1217 if (ret < 0) {
ffe60014
DG
1218 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1219 ERR("UST app channel %s enable failed for app (pid: %d) "
1220 "and session handle %d with ret %d",
1221 ua_chan->name, app->pid, ua_sess->handle, ret);
1222 } else {
47cb9d4a
DG
1223 /*
1224 * This is normal behavior, an application can die during the
1225 * creation process. Don't report an error so the execution can
1226 * continue normally.
1227 */
1228 ret = 0;
ffe60014
DG
1229 DBG3("UST app enable channel failed. Application is dead.");
1230 }
78f0bacd
DG
1231 goto error;
1232 }
1233
1234 ua_chan->enabled = 1;
1235
1236 DBG2("UST app channel %s enabled successfully for app (pid: %d)",
852d0037 1237 ua_chan->name, app->pid);
78f0bacd
DG
1238
1239error:
840cb59c 1240 health_code_update();
78f0bacd
DG
1241 return ret;
1242}
1243
edb67388
DG
1244/*
1245 * Enable the specified event on to UST tracer for the UST session.
1246 */
1247static int enable_ust_event(struct ust_app *app,
1248 struct ust_app_session *ua_sess, struct ust_app_event *ua_event)
1249{
1250 int ret;
1251
840cb59c 1252 health_code_update();
86acf0da 1253
852d0037 1254 ret = ustctl_enable(app->sock, ua_event->obj);
edb67388 1255 if (ret < 0) {
ffe60014
DG
1256 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1257 ERR("UST app event %s enable failed for app (pid: %d) "
1258 "and session handle %d with ret %d",
1259 ua_event->attr.name, app->pid, ua_sess->handle, ret);
1260 } else {
47cb9d4a
DG
1261 /*
1262 * This is normal behavior, an application can die during the
1263 * creation process. Don't report an error so the execution can
1264 * continue normally.
1265 */
1266 ret = 0;
ffe60014
DG
1267 DBG3("UST app enable event failed. Application is dead.");
1268 }
edb67388
DG
1269 goto error;
1270 }
1271
1272 DBG2("UST app event %s enabled successfully for app (pid: %d)",
852d0037 1273 ua_event->attr.name, app->pid);
edb67388
DG
1274
1275error:
840cb59c 1276 health_code_update();
edb67388
DG
1277 return ret;
1278}
1279
099e26bd 1280/*
7972aab2 1281 * Send channel and stream buffer to application.
4f3ab6ee 1282 *
ffe60014 1283 * Return 0 on success. On error, a negative value is returned.
4f3ab6ee 1284 */
7972aab2
DG
1285static int send_channel_pid_to_ust(struct ust_app *app,
1286 struct ust_app_session *ua_sess, struct ust_app_channel *ua_chan)
4f3ab6ee
DG
1287{
1288 int ret;
ffe60014 1289 struct ust_app_stream *stream, *stmp;
4f3ab6ee
DG
1290
1291 assert(app);
ffe60014 1292 assert(ua_sess);
4f3ab6ee 1293 assert(ua_chan);
4f3ab6ee 1294
840cb59c 1295 health_code_update();
4f3ab6ee 1296
7972aab2
DG
1297 DBG("UST app sending channel %s to UST app sock %d", ua_chan->name,
1298 app->sock);
86acf0da 1299
ffe60014
DG
1300 /* Send channel to the application. */
1301 ret = ust_consumer_send_channel_to_ust(app, ua_sess, ua_chan);
5b4a0ec0 1302 if (ret < 0) {
b551a063
DG
1303 goto error;
1304 }
1305
d88aee68
DG
1306 health_code_update();
1307
ffe60014
DG
1308 /* Send all streams to application. */
1309 cds_list_for_each_entry_safe(stream, stmp, &ua_chan->streams.head, list) {
1310 ret = ust_consumer_send_stream_to_ust(app, ua_chan, stream);
1311 if (ret < 0) {
1312 goto error;
1313 }
1314 /* We don't need the stream anymore once sent to the tracer. */
1315 cds_list_del(&stream->list);
1316 delete_ust_app_stream(-1, stream);
1317 }
ffe60014
DG
1318 /* Flag the channel that it is sent to the application. */
1319 ua_chan->is_sent = 1;
ffe60014 1320
b551a063 1321error:
840cb59c 1322 health_code_update();
b551a063
DG
1323 return ret;
1324}
1325
91d76f53 1326/*
5b4a0ec0 1327 * Create the specified event onto the UST tracer for a UST session.
d0b96690
DG
1328 *
1329 * Should be called with session mutex held.
91d76f53 1330 */
edb67388
DG
1331static
1332int create_ust_event(struct ust_app *app, struct ust_app_session *ua_sess,
1333 struct ust_app_channel *ua_chan, struct ust_app_event *ua_event)
91d76f53 1334{
5b4a0ec0 1335 int ret = 0;
284d8f55 1336
840cb59c 1337 health_code_update();
86acf0da 1338
5b4a0ec0 1339 /* Create UST event on tracer */
852d0037 1340 ret = ustctl_create_event(app->sock, &ua_event->attr, ua_chan->obj,
5b4a0ec0
DG
1341 &ua_event->obj);
1342 if (ret < 0) {
ffe60014
DG
1343 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1344 ERR("Error ustctl create event %s for app pid: %d with ret %d",
1345 ua_event->attr.name, app->pid, ret);
1346 } else {
47cb9d4a
DG
1347 /*
1348 * This is normal behavior, an application can die during the
1349 * creation process. Don't report an error so the execution can
1350 * continue normally.
1351 */
1352 ret = 0;
ffe60014
DG
1353 DBG3("UST app create event failed. Application is dead.");
1354 }
5b4a0ec0 1355 goto error;
91d76f53 1356 }
f6a9efaa 1357
5b4a0ec0 1358 ua_event->handle = ua_event->obj->handle;
284d8f55 1359
5b4a0ec0 1360 DBG2("UST app event %s created successfully for pid:%d",
852d0037 1361 ua_event->attr.name, app->pid);
f6a9efaa 1362
840cb59c 1363 health_code_update();
86acf0da 1364
025faf73
DG
1365 /* Set filter if one is present. */
1366 if (ua_event->filter) {
1367 ret = set_ust_event_filter(ua_event, app);
1368 if (ret < 0) {
1369 goto error;
1370 }
1371 }
1372
8535a6d9 1373 /* If event not enabled, disable it on the tracer */
fc34caaa 1374 if (ua_event->enabled == 0) {
8535a6d9
DG
1375 ret = disable_ust_event(app, ua_sess, ua_event);
1376 if (ret < 0) {
fc34caaa
DG
1377 /*
1378 * If we hit an EPERM, something is wrong with our disable call. If
1379 * we get an EEXIST, there is a problem on the tracer side since we
1380 * just created it.
1381 */
1382 switch (ret) {
49c336c1 1383 case -LTTNG_UST_ERR_PERM:
fc34caaa
DG
1384 /* Code flow problem */
1385 assert(0);
49c336c1 1386 case -LTTNG_UST_ERR_EXIST:
fc34caaa
DG
1387 /* It's OK for our use case. */
1388 ret = 0;
1389 break;
1390 default:
1391 break;
1392 }
8535a6d9
DG
1393 goto error;
1394 }
1395 }
1396
5b4a0ec0 1397error:
840cb59c 1398 health_code_update();
5b4a0ec0 1399 return ret;
91d76f53 1400}
48842b30 1401
5b4a0ec0
DG
1402/*
1403 * Copy data between an UST app event and a LTT event.
1404 */
421cb601 1405static void shadow_copy_event(struct ust_app_event *ua_event,
48842b30
DG
1406 struct ltt_ust_event *uevent)
1407{
1408 strncpy(ua_event->name, uevent->attr.name, sizeof(ua_event->name));
1409 ua_event->name[sizeof(ua_event->name) - 1] = '\0';
1410
fc34caaa
DG
1411 ua_event->enabled = uevent->enabled;
1412
5b4a0ec0
DG
1413 /* Copy event attributes */
1414 memcpy(&ua_event->attr, &uevent->attr, sizeof(ua_event->attr));
1415
53a80697
MD
1416 /* Copy filter bytecode */
1417 if (uevent->filter) {
025faf73
DG
1418 ua_event->filter = alloc_copy_ust_app_filter(uevent->filter);
1419 /* Filter might be NULL here in case of ENONEM. */
53a80697 1420 }
48842b30
DG
1421}
1422
5b4a0ec0
DG
1423/*
1424 * Copy data between an UST app channel and a LTT channel.
1425 */
421cb601 1426static void shadow_copy_channel(struct ust_app_channel *ua_chan,
48842b30
DG
1427 struct ltt_ust_channel *uchan)
1428{
bec39940 1429 struct lttng_ht_iter iter;
48842b30 1430 struct ltt_ust_event *uevent;
55cc08a6 1431 struct ltt_ust_context *uctx;
48842b30 1432 struct ust_app_event *ua_event;
55cc08a6 1433 struct ust_app_ctx *ua_ctx;
48842b30 1434
fc34caaa 1435 DBG2("UST app shadow copy of channel %s started", ua_chan->name);
48842b30
DG
1436
1437 strncpy(ua_chan->name, uchan->name, sizeof(ua_chan->name));
1438 ua_chan->name[sizeof(ua_chan->name) - 1] = '\0';
ffe60014 1439
1624d5b7
JD
1440 ua_chan->tracefile_size = uchan->tracefile_size;
1441 ua_chan->tracefile_count = uchan->tracefile_count;
1442
ffe60014
DG
1443 /* Copy event attributes since the layout is different. */
1444 ua_chan->attr.subbuf_size = uchan->attr.subbuf_size;
1445 ua_chan->attr.num_subbuf = uchan->attr.num_subbuf;
1446 ua_chan->attr.overwrite = uchan->attr.overwrite;
1447 ua_chan->attr.switch_timer_interval = uchan->attr.switch_timer_interval;
1448 ua_chan->attr.read_timer_interval = uchan->attr.read_timer_interval;
1449 ua_chan->attr.output = uchan->attr.output;
1450 /*
1451 * Note that the attribute channel type is not set since the channel on the
1452 * tracing registry side does not have this information.
1453 */
48842b30 1454
fc34caaa 1455 ua_chan->enabled = uchan->enabled;
7972aab2 1456 ua_chan->tracing_channel_id = uchan->id;
fc34caaa 1457
4623416f 1458 cds_list_for_each_entry(uctx, &uchan->ctx_list, list) {
55cc08a6
DG
1459 ua_ctx = alloc_ust_app_ctx(&uctx->ctx);
1460 if (ua_ctx == NULL) {
1461 continue;
1462 }
bec39940
DG
1463 lttng_ht_node_init_ulong(&ua_ctx->node,
1464 (unsigned long) ua_ctx->ctx.ctx);
1465 lttng_ht_add_unique_ulong(ua_chan->ctx, &ua_ctx->node);
4623416f 1466 cds_list_add_tail(&ua_ctx->list, &ua_chan->ctx_list);
55cc08a6 1467 }
48842b30 1468
421cb601 1469 /* Copy all events from ltt ust channel to ust app channel */
bec39940 1470 cds_lfht_for_each_entry(uchan->events->ht, &iter.iter, uevent, node.node) {
18eace3b
DG
1471 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
1472 uevent->filter, uevent->attr.loglevel);
1473 if (ua_event == NULL) {
421cb601 1474 DBG2("UST event %s not found on shadow copy channel",
48842b30 1475 uevent->attr.name);
284d8f55 1476 ua_event = alloc_ust_app_event(uevent->attr.name, &uevent->attr);
48842b30 1477 if (ua_event == NULL) {
5b4a0ec0 1478 continue;
48842b30 1479 }
421cb601 1480 shadow_copy_event(ua_event, uevent);
d0b96690 1481 add_unique_ust_app_event(ua_chan, ua_event);
48842b30 1482 }
48842b30
DG
1483 }
1484
fc34caaa 1485 DBG3("UST app shadow copy of channel %s done", ua_chan->name);
48842b30
DG
1486}
1487
5b4a0ec0
DG
1488/*
1489 * Copy data between a UST app session and a regular LTT session.
1490 */
421cb601 1491static void shadow_copy_session(struct ust_app_session *ua_sess,
bec39940 1492 struct ltt_ust_session *usess, struct ust_app *app)
48842b30 1493{
bec39940
DG
1494 struct lttng_ht_node_str *ua_chan_node;
1495 struct lttng_ht_iter iter;
48842b30
DG
1496 struct ltt_ust_channel *uchan;
1497 struct ust_app_channel *ua_chan;
477d7741
MD
1498 time_t rawtime;
1499 struct tm *timeinfo;
1500 char datetime[16];
1501 int ret;
1502
1503 /* Get date and time for unique app path */
1504 time(&rawtime);
1505 timeinfo = localtime(&rawtime);
1506 strftime(datetime, sizeof(datetime), "%Y%m%d-%H%M%S", timeinfo);
48842b30 1507
421cb601 1508 DBG2("Shadow copy of session handle %d", ua_sess->handle);
48842b30 1509
7972aab2
DG
1510 ua_sess->tracing_id = usess->id;
1511 ua_sess->id = get_next_session_id();
1512 ua_sess->uid = app->uid;
1513 ua_sess->gid = app->gid;
1514 ua_sess->euid = usess->uid;
1515 ua_sess->egid = usess->gid;
1516 ua_sess->buffer_type = usess->buffer_type;
1517 ua_sess->bits_per_long = app->bits_per_long;
1518 /* There is only one consumer object per session possible. */
1519 ua_sess->consumer = usess->consumer;
1520
1521 switch (ua_sess->buffer_type) {
1522 case LTTNG_BUFFER_PER_PID:
1523 ret = snprintf(ua_sess->path, sizeof(ua_sess->path),
dec56f6c 1524 DEFAULT_UST_TRACE_PID_PATH "/%s-%d-%s", app->name, app->pid,
7972aab2
DG
1525 datetime);
1526 break;
1527 case LTTNG_BUFFER_PER_UID:
1528 ret = snprintf(ua_sess->path, sizeof(ua_sess->path),
1529 DEFAULT_UST_TRACE_UID_PATH, ua_sess->uid, app->bits_per_long);
1530 break;
1531 default:
1532 assert(0);
1533 goto error;
1534 }
477d7741
MD
1535 if (ret < 0) {
1536 PERROR("asprintf UST shadow copy session");
477d7741 1537 assert(0);
7972aab2 1538 goto error;
477d7741
MD
1539 }
1540
48842b30 1541 /* Iterate over all channels in global domain. */
bec39940
DG
1542 cds_lfht_for_each_entry(usess->domain_global.channels->ht, &iter.iter,
1543 uchan, node.node) {
1544 struct lttng_ht_iter uiter;
ba767faf 1545
bec39940
DG
1546 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
1547 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
5b4a0ec0 1548 if (ua_chan_node != NULL) {
fc34caaa 1549 /* Session exist. Contiuing. */
5b4a0ec0
DG
1550 continue;
1551 }
421cb601 1552
5b4a0ec0
DG
1553 DBG2("Channel %s not found on shadow session copy, creating it",
1554 uchan->name);
d0b96690 1555 ua_chan = alloc_ust_app_channel(uchan->name, ua_sess, &uchan->attr);
5b4a0ec0 1556 if (ua_chan == NULL) {
fc34caaa 1557 /* malloc failed FIXME: Might want to do handle ENOMEM .. */
5b4a0ec0 1558 continue;
48842b30 1559 }
5b4a0ec0 1560 shadow_copy_channel(ua_chan, uchan);
ffe60014
DG
1561 /*
1562 * The concept of metadata channel does not exist on the tracing
1563 * registry side of the session daemon so this can only be a per CPU
1564 * channel and not metadata.
1565 */
1566 ua_chan->attr.type = LTTNG_UST_CHAN_PER_CPU;
1567
bec39940 1568 lttng_ht_add_unique_str(ua_sess->channels, &ua_chan->node);
48842b30 1569 }
7972aab2
DG
1570
1571error:
1572 return;
48842b30
DG
1573}
1574
78f0bacd
DG
1575/*
1576 * Lookup sesison wrapper.
1577 */
84cd17c6
MD
1578static
1579void __lookup_session_by_app(struct ltt_ust_session *usess,
bec39940 1580 struct ust_app *app, struct lttng_ht_iter *iter)
84cd17c6
MD
1581{
1582 /* Get right UST app session from app */
77c5447b 1583 lttng_ht_lookup(app->sessions, &usess->id, iter);
84cd17c6
MD
1584}
1585
421cb601
DG
1586/*
1587 * Return ust app session from the app session hashtable using the UST session
a991f516 1588 * id.
421cb601 1589 */
48842b30
DG
1590static struct ust_app_session *lookup_session_by_app(
1591 struct ltt_ust_session *usess, struct ust_app *app)
1592{
bec39940 1593 struct lttng_ht_iter iter;
77c5447b 1594 struct lttng_ht_node_u64 *node;
48842b30 1595
84cd17c6 1596 __lookup_session_by_app(usess, app, &iter);
77c5447b 1597 node = lttng_ht_iter_get_node_u64(&iter);
48842b30
DG
1598 if (node == NULL) {
1599 goto error;
1600 }
1601
1602 return caa_container_of(node, struct ust_app_session, node);
1603
1604error:
1605 return NULL;
1606}
1607
7972aab2
DG
1608/*
1609 * Setup buffer registry per PID for the given session and application. If none
1610 * is found, a new one is created, added to the global registry and
1611 * initialized. If regp is valid, it's set with the newly created object.
1612 *
1613 * Return 0 on success or else a negative value.
1614 */
1615static int setup_buffer_reg_pid(struct ust_app_session *ua_sess,
1616 struct ust_app *app, struct buffer_reg_pid **regp)
1617{
1618 int ret = 0;
1619 struct buffer_reg_pid *reg_pid;
1620
1621 assert(ua_sess);
1622 assert(app);
1623
1624 rcu_read_lock();
1625
1626 reg_pid = buffer_reg_pid_find(ua_sess->id);
1627 if (!reg_pid) {
1628 /*
1629 * This is the create channel path meaning that if there is NO
1630 * registry available, we have to create one for this session.
1631 */
1632 ret = buffer_reg_pid_create(ua_sess->id, &reg_pid);
1633 if (ret < 0) {
1634 goto error;
1635 }
1636 buffer_reg_pid_add(reg_pid);
1637 } else {
1638 goto end;
1639 }
1640
1641 /* Initialize registry. */
1642 ret = ust_registry_session_init(&reg_pid->registry->reg.ust, app,
1643 app->bits_per_long, app->uint8_t_alignment,
1644 app->uint16_t_alignment, app->uint32_t_alignment,
af6142cf
MD
1645 app->uint64_t_alignment, app->long_alignment,
1646 app->byte_order, app->version.major,
1647 app->version.minor);
7972aab2
DG
1648 if (ret < 0) {
1649 goto error;
1650 }
1651
1652 DBG3("UST app buffer registry per PID created successfully");
1653
1654end:
1655 if (regp) {
1656 *regp = reg_pid;
1657 }
1658error:
1659 rcu_read_unlock();
1660 return ret;
1661}
1662
1663/*
1664 * Setup buffer registry per UID for the given session and application. If none
1665 * is found, a new one is created, added to the global registry and
1666 * initialized. If regp is valid, it's set with the newly created object.
1667 *
1668 * Return 0 on success or else a negative value.
1669 */
1670static int setup_buffer_reg_uid(struct ltt_ust_session *usess,
1671 struct ust_app *app, struct buffer_reg_uid **regp)
1672{
1673 int ret = 0;
1674 struct buffer_reg_uid *reg_uid;
1675
1676 assert(usess);
1677 assert(app);
1678
1679 rcu_read_lock();
1680
1681 reg_uid = buffer_reg_uid_find(usess->id, app->bits_per_long, app->uid);
1682 if (!reg_uid) {
1683 /*
1684 * This is the create channel path meaning that if there is NO
1685 * registry available, we have to create one for this session.
1686 */
1687 ret = buffer_reg_uid_create(usess->id, app->bits_per_long, app->uid,
1688 LTTNG_DOMAIN_UST, &reg_uid);
1689 if (ret < 0) {
1690 goto error;
1691 }
1692 buffer_reg_uid_add(reg_uid);
1693 } else {
1694 goto end;
1695 }
1696
1697 /* Initialize registry. */
af6142cf 1698 ret = ust_registry_session_init(&reg_uid->registry->reg.ust, NULL,
7972aab2
DG
1699 app->bits_per_long, app->uint8_t_alignment,
1700 app->uint16_t_alignment, app->uint32_t_alignment,
af6142cf
MD
1701 app->uint64_t_alignment, app->long_alignment,
1702 app->byte_order, app->version.major,
1703 app->version.minor);
7972aab2
DG
1704 if (ret < 0) {
1705 goto error;
1706 }
1707 /* Add node to teardown list of the session. */
1708 cds_list_add(&reg_uid->lnode, &usess->buffer_reg_uid_list);
1709
1710 DBG3("UST app buffer registry per UID created successfully");
1711
1712end:
1713 if (regp) {
1714 *regp = reg_uid;
1715 }
1716error:
1717 rcu_read_unlock();
1718 return ret;
1719}
1720
421cb601 1721/*
3d8ca23b 1722 * Create a session on the tracer side for the given app.
421cb601 1723 *
3d8ca23b
DG
1724 * On success, ua_sess_ptr is populated with the session pointer or else left
1725 * untouched. If the session was created, is_created is set to 1. On error,
1726 * it's left untouched. Note that ua_sess_ptr is mandatory but is_created can
1727 * be NULL.
1728 *
1729 * Returns 0 on success or else a negative code which is either -ENOMEM or
1730 * -ENOTCONN which is the default code if the ustctl_create_session fails.
421cb601 1731 */
3d8ca23b
DG
1732static int create_ust_app_session(struct ltt_ust_session *usess,
1733 struct ust_app *app, struct ust_app_session **ua_sess_ptr,
1734 int *is_created)
421cb601 1735{
3d8ca23b 1736 int ret, created = 0;
421cb601
DG
1737 struct ust_app_session *ua_sess;
1738
3d8ca23b
DG
1739 assert(usess);
1740 assert(app);
1741 assert(ua_sess_ptr);
1742
840cb59c 1743 health_code_update();
86acf0da 1744
421cb601
DG
1745 ua_sess = lookup_session_by_app(usess, app);
1746 if (ua_sess == NULL) {
77c5447b 1747 DBG2("UST app pid: %d session id %" PRIu64 " not found, creating it",
852d0037 1748 app->pid, usess->id);
d0b96690 1749 ua_sess = alloc_ust_app_session(app);
421cb601
DG
1750 if (ua_sess == NULL) {
1751 /* Only malloc can failed so something is really wrong */
3d8ca23b
DG
1752 ret = -ENOMEM;
1753 goto error;
421cb601 1754 }
477d7741 1755 shadow_copy_session(ua_sess, usess, app);
3d8ca23b 1756 created = 1;
421cb601
DG
1757 }
1758
7972aab2
DG
1759 switch (usess->buffer_type) {
1760 case LTTNG_BUFFER_PER_PID:
1761 /* Init local registry. */
1762 ret = setup_buffer_reg_pid(ua_sess, app, NULL);
421cb601 1763 if (ret < 0) {
7972aab2
DG
1764 goto error;
1765 }
1766 break;
1767 case LTTNG_BUFFER_PER_UID:
1768 /* Look for a global registry. If none exists, create one. */
1769 ret = setup_buffer_reg_uid(usess, app, NULL);
1770 if (ret < 0) {
1771 goto error;
1772 }
1773 break;
1774 default:
1775 assert(0);
1776 ret = -EINVAL;
1777 goto error;
1778 }
1779
1780 health_code_update();
1781
1782 if (ua_sess->handle == -1) {
1783 ret = ustctl_create_session(app->sock);
1784 if (ret < 0) {
1785 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
1786 ERR("Creating session for app pid %d with ret %d",
ffe60014
DG
1787 app->pid, ret);
1788 } else {
1789 DBG("UST app creating session failed. Application is dead");
47cb9d4a
DG
1790 /*
1791 * This is normal behavior, an application can die during the
1792 * creation process. Don't report an error so the execution can
1793 * continue normally. This will get flagged ENOTCONN and the
1794 * caller will handle it.
1795 */
1796 ret = 0;
ffe60014 1797 }
d0b96690 1798 delete_ust_app_session(-1, ua_sess, app);
3d8ca23b
DG
1799 if (ret != -ENOMEM) {
1800 /*
1801 * Tracer is probably gone or got an internal error so let's
1802 * behave like it will soon unregister or not usable.
1803 */
1804 ret = -ENOTCONN;
1805 }
1806 goto error;
421cb601
DG
1807 }
1808
7972aab2
DG
1809 ua_sess->handle = ret;
1810
1811 /* Add ust app session to app's HT */
77c5447b
MD
1812 lttng_ht_node_init_u64(&ua_sess->node,
1813 ua_sess->tracing_id);
1814 lttng_ht_add_unique_u64(app->sessions, &ua_sess->node);
7972aab2
DG
1815
1816 DBG2("UST app session created successfully with handle %d", ret);
1817 }
1818
1819 *ua_sess_ptr = ua_sess;
1820 if (is_created) {
1821 *is_created = created;
1822 }
1823
1824 /* Everything went well. */
1825 ret = 0;
1826
1827error:
1828 health_code_update();
1829 return ret;
1830}
1831
1832/*
1833 * Create a context for the channel on the tracer.
1834 *
1835 * Called with UST app session lock held and a RCU read side lock.
1836 */
1837static
1838int create_ust_app_channel_context(struct ust_app_session *ua_sess,
1839 struct ust_app_channel *ua_chan, struct lttng_ust_context *uctx,
1840 struct ust_app *app)
1841{
1842 int ret = 0;
1843 struct lttng_ht_iter iter;
1844 struct lttng_ht_node_ulong *node;
1845 struct ust_app_ctx *ua_ctx;
1846
1847 DBG2("UST app adding context to channel %s", ua_chan->name);
1848
1849 lttng_ht_lookup(ua_chan->ctx, (void *)((unsigned long)uctx->ctx), &iter);
1850 node = lttng_ht_iter_get_node_ulong(&iter);
1851 if (node != NULL) {
1852 ret = -EEXIST;
1853 goto error;
1854 }
1855
1856 ua_ctx = alloc_ust_app_ctx(uctx);
1857 if (ua_ctx == NULL) {
1858 /* malloc failed */
1859 ret = -1;
1860 goto error;
1861 }
1862
1863 lttng_ht_node_init_ulong(&ua_ctx->node, (unsigned long) ua_ctx->ctx.ctx);
1864 lttng_ht_add_unique_ulong(ua_chan->ctx, &ua_ctx->node);
4623416f 1865 cds_list_add_tail(&ua_ctx->list, &ua_chan->ctx_list);
7972aab2
DG
1866
1867 ret = create_ust_channel_context(ua_chan, ua_ctx, app);
1868 if (ret < 0) {
1869 goto error;
1870 }
1871
1872error:
1873 return ret;
1874}
1875
1876/*
1877 * Enable on the tracer side a ust app event for the session and channel.
1878 *
1879 * Called with UST app session lock held.
1880 */
1881static
1882int enable_ust_app_event(struct ust_app_session *ua_sess,
1883 struct ust_app_event *ua_event, struct ust_app *app)
1884{
1885 int ret;
1886
1887 ret = enable_ust_event(app, ua_sess, ua_event);
1888 if (ret < 0) {
1889 goto error;
1890 }
1891
1892 ua_event->enabled = 1;
1893
1894error:
1895 return ret;
1896}
1897
1898/*
1899 * Disable on the tracer side a ust app event for the session and channel.
1900 */
1901static int disable_ust_app_event(struct ust_app_session *ua_sess,
1902 struct ust_app_event *ua_event, struct ust_app *app)
1903{
1904 int ret;
1905
1906 ret = disable_ust_event(app, ua_sess, ua_event);
1907 if (ret < 0) {
1908 goto error;
1909 }
1910
1911 ua_event->enabled = 0;
1912
1913error:
1914 return ret;
1915}
1916
1917/*
1918 * Lookup ust app channel for session and disable it on the tracer side.
1919 */
1920static
1921int disable_ust_app_channel(struct ust_app_session *ua_sess,
1922 struct ust_app_channel *ua_chan, struct ust_app *app)
1923{
1924 int ret;
1925
1926 ret = disable_ust_channel(app, ua_sess, ua_chan);
1927 if (ret < 0) {
1928 goto error;
1929 }
1930
1931 ua_chan->enabled = 0;
1932
1933error:
1934 return ret;
1935}
1936
1937/*
1938 * Lookup ust app channel for session and enable it on the tracer side. This
1939 * MUST be called with a RCU read side lock acquired.
1940 */
1941static int enable_ust_app_channel(struct ust_app_session *ua_sess,
1942 struct ltt_ust_channel *uchan, struct ust_app *app)
1943{
1944 int ret = 0;
1945 struct lttng_ht_iter iter;
1946 struct lttng_ht_node_str *ua_chan_node;
1947 struct ust_app_channel *ua_chan;
1948
1949 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &iter);
1950 ua_chan_node = lttng_ht_iter_get_node_str(&iter);
1951 if (ua_chan_node == NULL) {
77c5447b 1952 DBG2("Unable to find channel %s in ust session id %" PRIu64,
7972aab2
DG
1953 uchan->name, ua_sess->tracing_id);
1954 goto error;
1955 }
1956
1957 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
1958
1959 ret = enable_ust_channel(app, ua_sess, ua_chan);
1960 if (ret < 0) {
1961 goto error;
1962 }
1963
1964error:
1965 return ret;
1966}
1967
1968/*
1969 * Ask the consumer to create a channel and get it if successful.
1970 *
1971 * Return 0 on success or else a negative value.
1972 */
1973static int do_consumer_create_channel(struct ltt_ust_session *usess,
1974 struct ust_app_session *ua_sess, struct ust_app_channel *ua_chan,
1975 int bitness, struct ust_registry_session *registry)
1976{
1977 int ret;
1978 unsigned int nb_fd = 0;
1979 struct consumer_socket *socket;
1980
1981 assert(usess);
1982 assert(ua_sess);
1983 assert(ua_chan);
1984 assert(registry);
1985
1986 rcu_read_lock();
1987 health_code_update();
1988
1989 /* Get the right consumer socket for the application. */
1990 socket = consumer_find_socket_by_bitness(bitness, usess->consumer);
1991 if (!socket) {
1992 ret = -EINVAL;
1993 goto error;
1994 }
1995
1996 health_code_update();
1997
1998 /* Need one fd for the channel. */
1999 ret = lttng_fd_get(LTTNG_FD_APPS, 1);
2000 if (ret < 0) {
2001 ERR("Exhausted number of available FD upon create channel");
2002 goto error;
2003 }
2004
2005 /*
2006 * Ask consumer to create channel. The consumer will return the number of
2007 * stream we have to expect.
2008 */
2009 ret = ust_consumer_ask_channel(ua_sess, ua_chan, usess->consumer, socket,
2010 registry);
2011 if (ret < 0) {
2012 goto error_ask;
2013 }
2014
2015 /*
2016 * Compute the number of fd needed before receiving them. It must be 2 per
2017 * stream (2 being the default value here).
2018 */
2019 nb_fd = DEFAULT_UST_STREAM_FD_NUM * ua_chan->expected_stream_count;
2020
2021 /* Reserve the amount of file descriptor we need. */
2022 ret = lttng_fd_get(LTTNG_FD_APPS, nb_fd);
2023 if (ret < 0) {
2024 ERR("Exhausted number of available FD upon create channel");
2025 goto error_fd_get_stream;
2026 }
2027
2028 health_code_update();
2029
2030 /*
2031 * Now get the channel from the consumer. This call wil populate the stream
2032 * list of that channel and set the ust objects.
2033 */
d9078d0c
DG
2034 if (usess->consumer->enabled) {
2035 ret = ust_consumer_get_channel(socket, ua_chan);
2036 if (ret < 0) {
2037 goto error_destroy;
2038 }
7972aab2
DG
2039 }
2040
2041 rcu_read_unlock();
2042 return 0;
2043
2044error_destroy:
2045 lttng_fd_put(LTTNG_FD_APPS, nb_fd);
2046error_fd_get_stream:
2047 /*
2048 * Initiate a destroy channel on the consumer since we had an error
2049 * handling it on our side. The return value is of no importance since we
2050 * already have a ret value set by the previous error that we need to
2051 * return.
2052 */
2053 (void) ust_consumer_destroy_channel(socket, ua_chan);
2054error_ask:
2055 lttng_fd_put(LTTNG_FD_APPS, 1);
2056error:
2057 health_code_update();
2058 rcu_read_unlock();
2059 return ret;
2060}
2061
2062/*
2063 * Duplicate the ust data object of the ust app stream and save it in the
2064 * buffer registry stream.
2065 *
2066 * Return 0 on success or else a negative value.
2067 */
2068static int duplicate_stream_object(struct buffer_reg_stream *reg_stream,
2069 struct ust_app_stream *stream)
2070{
2071 int ret;
2072
2073 assert(reg_stream);
2074 assert(stream);
2075
2076 /* Reserve the amount of file descriptor we need. */
2077 ret = lttng_fd_get(LTTNG_FD_APPS, 2);
2078 if (ret < 0) {
2079 ERR("Exhausted number of available FD upon duplicate stream");
2080 goto error;
2081 }
2082
2083 /* Duplicate object for stream once the original is in the registry. */
2084 ret = ustctl_duplicate_ust_object_data(&stream->obj,
2085 reg_stream->obj.ust);
2086 if (ret < 0) {
2087 ERR("Duplicate stream obj from %p to %p failed with ret %d",
2088 reg_stream->obj.ust, stream->obj, ret);
2089 lttng_fd_put(LTTNG_FD_APPS, 2);
2090 goto error;
2091 }
2092 stream->handle = stream->obj->handle;
2093
2094error:
2095 return ret;
2096}
2097
2098/*
2099 * Duplicate the ust data object of the ust app. channel and save it in the
2100 * buffer registry channel.
2101 *
2102 * Return 0 on success or else a negative value.
2103 */
2104static int duplicate_channel_object(struct buffer_reg_channel *reg_chan,
2105 struct ust_app_channel *ua_chan)
2106{
2107 int ret;
2108
2109 assert(reg_chan);
2110 assert(ua_chan);
2111
2112 /* Need two fds for the channel. */
2113 ret = lttng_fd_get(LTTNG_FD_APPS, 1);
2114 if (ret < 0) {
2115 ERR("Exhausted number of available FD upon duplicate channel");
2116 goto error_fd_get;
2117 }
2118
2119 /* Duplicate object for stream once the original is in the registry. */
2120 ret = ustctl_duplicate_ust_object_data(&ua_chan->obj, reg_chan->obj.ust);
2121 if (ret < 0) {
2122 ERR("Duplicate channel obj from %p to %p failed with ret: %d",
2123 reg_chan->obj.ust, ua_chan->obj, ret);
2124 goto error;
2125 }
2126 ua_chan->handle = ua_chan->obj->handle;
2127
2128 return 0;
2129
2130error:
2131 lttng_fd_put(LTTNG_FD_APPS, 1);
2132error_fd_get:
2133 return ret;
2134}
2135
2136/*
2137 * For a given channel buffer registry, setup all streams of the given ust
2138 * application channel.
2139 *
2140 * Return 0 on success or else a negative value.
2141 */
2142static int setup_buffer_reg_streams(struct buffer_reg_channel *reg_chan,
2143 struct ust_app_channel *ua_chan)
2144{
2145 int ret = 0;
2146 struct ust_app_stream *stream, *stmp;
2147
2148 assert(reg_chan);
2149 assert(ua_chan);
2150
2151 DBG2("UST app setup buffer registry stream");
2152
2153 /* Send all streams to application. */
2154 cds_list_for_each_entry_safe(stream, stmp, &ua_chan->streams.head, list) {
2155 struct buffer_reg_stream *reg_stream;
2156
2157 ret = buffer_reg_stream_create(&reg_stream);
2158 if (ret < 0) {
2159 goto error;
2160 }
2161
2162 /*
2163 * Keep original pointer and nullify it in the stream so the delete
2164 * stream call does not release the object.
2165 */
2166 reg_stream->obj.ust = stream->obj;
2167 stream->obj = NULL;
2168 buffer_reg_stream_add(reg_stream, reg_chan);
421cb601 2169
7972aab2
DG
2170 /* We don't need the streams anymore. */
2171 cds_list_del(&stream->list);
2172 delete_ust_app_stream(-1, stream);
2173 }
421cb601 2174
7972aab2
DG
2175error:
2176 return ret;
2177}
2178
2179/*
2180 * Create a buffer registry channel for the given session registry and
2181 * application channel object. If regp pointer is valid, it's set with the
2182 * created object. Important, the created object is NOT added to the session
2183 * registry hash table.
2184 *
2185 * Return 0 on success else a negative value.
2186 */
2187static int create_buffer_reg_channel(struct buffer_reg_session *reg_sess,
2188 struct ust_app_channel *ua_chan, struct buffer_reg_channel **regp)
2189{
2190 int ret;
2191 struct buffer_reg_channel *reg_chan = NULL;
2192
2193 assert(reg_sess);
2194 assert(ua_chan);
2195
2196 DBG2("UST app creating buffer registry channel for %s", ua_chan->name);
2197
2198 /* Create buffer registry channel. */
2199 ret = buffer_reg_channel_create(ua_chan->tracing_channel_id, &reg_chan);
2200 if (ret < 0) {
2201 goto error_create;
421cb601 2202 }
7972aab2
DG
2203 assert(reg_chan);
2204 reg_chan->consumer_key = ua_chan->key;
421cb601 2205
7972aab2
DG
2206 /* Create and add a channel registry to session. */
2207 ret = ust_registry_channel_add(reg_sess->reg.ust,
2208 ua_chan->tracing_channel_id);
2209 if (ret < 0) {
2210 goto error;
d88aee68 2211 }
7972aab2 2212 buffer_reg_channel_add(reg_sess, reg_chan);
d88aee68 2213
7972aab2
DG
2214 if (regp) {
2215 *regp = reg_chan;
3d8ca23b 2216 }
d88aee68 2217
7972aab2 2218 return 0;
3d8ca23b
DG
2219
2220error:
7972aab2
DG
2221 /* Safe because the registry channel object was not added to any HT. */
2222 buffer_reg_channel_destroy(reg_chan, LTTNG_DOMAIN_UST);
2223error_create:
3d8ca23b 2224 return ret;
421cb601
DG
2225}
2226
55cc08a6 2227/*
7972aab2
DG
2228 * Setup buffer registry channel for the given session registry and application
2229 * channel object. If regp pointer is valid, it's set with the created object.
d0b96690 2230 *
7972aab2 2231 * Return 0 on success else a negative value.
55cc08a6 2232 */
7972aab2
DG
2233static int setup_buffer_reg_channel(struct buffer_reg_session *reg_sess,
2234 struct ust_app_channel *ua_chan, struct buffer_reg_channel *reg_chan)
55cc08a6 2235{
7972aab2 2236 int ret;
55cc08a6 2237
7972aab2
DG
2238 assert(reg_sess);
2239 assert(reg_chan);
2240 assert(ua_chan);
2241 assert(ua_chan->obj);
55cc08a6 2242
7972aab2 2243 DBG2("UST app setup buffer registry channel for %s", ua_chan->name);
55cc08a6 2244
7972aab2
DG
2245 /* Setup all streams for the registry. */
2246 ret = setup_buffer_reg_streams(reg_chan, ua_chan);
2247 if (ret < 0) {
55cc08a6
DG
2248 goto error;
2249 }
2250
7972aab2
DG
2251 reg_chan->obj.ust = ua_chan->obj;
2252 ua_chan->obj = NULL;
55cc08a6 2253
7972aab2 2254 return 0;
55cc08a6
DG
2255
2256error:
7972aab2
DG
2257 buffer_reg_channel_remove(reg_sess, reg_chan);
2258 buffer_reg_channel_destroy(reg_chan, LTTNG_DOMAIN_UST);
55cc08a6
DG
2259 return ret;
2260}
2261
edb67388 2262/*
7972aab2 2263 * Send buffer registry channel to the application.
d0b96690 2264 *
7972aab2 2265 * Return 0 on success else a negative value.
edb67388 2266 */
7972aab2
DG
2267static int send_channel_uid_to_ust(struct buffer_reg_channel *reg_chan,
2268 struct ust_app *app, struct ust_app_session *ua_sess,
2269 struct ust_app_channel *ua_chan)
edb67388
DG
2270{
2271 int ret;
7972aab2 2272 struct buffer_reg_stream *reg_stream;
edb67388 2273
7972aab2
DG
2274 assert(reg_chan);
2275 assert(app);
2276 assert(ua_sess);
2277 assert(ua_chan);
2278
2279 DBG("UST app sending buffer registry channel to ust sock %d", app->sock);
2280
2281 ret = duplicate_channel_object(reg_chan, ua_chan);
edb67388
DG
2282 if (ret < 0) {
2283 goto error;
2284 }
2285
7972aab2
DG
2286 /* Send channel to the application. */
2287 ret = ust_consumer_send_channel_to_ust(app, ua_sess, ua_chan);
2288 if (ret < 0) {
2289 goto error;
2290 }
2291
2292 health_code_update();
2293
2294 /* Send all streams to application. */
2295 pthread_mutex_lock(&reg_chan->stream_list_lock);
2296 cds_list_for_each_entry(reg_stream, &reg_chan->streams, lnode) {
2297 struct ust_app_stream stream;
2298
2299 ret = duplicate_stream_object(reg_stream, &stream);
2300 if (ret < 0) {
2301 goto error_stream_unlock;
2302 }
2303
2304 ret = ust_consumer_send_stream_to_ust(app, ua_chan, &stream);
2305 if (ret < 0) {
8c067051 2306 (void) release_ust_app_stream(-1, &stream);
7972aab2
DG
2307 goto error_stream_unlock;
2308 }
edb67388 2309
7972aab2
DG
2310 /*
2311 * The return value is not important here. This function will output an
2312 * error if needed.
2313 */
2314 (void) release_ust_app_stream(-1, &stream);
2315 }
2316 ua_chan->is_sent = 1;
2317
2318error_stream_unlock:
2319 pthread_mutex_unlock(&reg_chan->stream_list_lock);
edb67388
DG
2320error:
2321 return ret;
2322}
2323
9730260e 2324/*
7972aab2
DG
2325 * Create and send to the application the created buffers with per UID buffers.
2326 *
2327 * Return 0 on success else a negative value.
9730260e 2328 */
7972aab2
DG
2329static int create_channel_per_uid(struct ust_app *app,
2330 struct ltt_ust_session *usess, struct ust_app_session *ua_sess,
2331 struct ust_app_channel *ua_chan)
9730260e
DG
2332{
2333 int ret;
7972aab2
DG
2334 struct buffer_reg_uid *reg_uid;
2335 struct buffer_reg_channel *reg_chan;
9730260e 2336
7972aab2
DG
2337 assert(app);
2338 assert(usess);
2339 assert(ua_sess);
2340 assert(ua_chan);
2341
2342 DBG("UST app creating channel %s with per UID buffers", ua_chan->name);
2343
2344 reg_uid = buffer_reg_uid_find(usess->id, app->bits_per_long, app->uid);
2345 /*
2346 * The session creation handles the creation of this global registry
2347 * object. If none can be find, there is a code flow problem or a
2348 * teardown race.
2349 */
2350 assert(reg_uid);
2351
2352 reg_chan = buffer_reg_channel_find(ua_chan->tracing_channel_id,
2353 reg_uid);
2354 if (!reg_chan) {
2355 /* Create the buffer registry channel object. */
2356 ret = create_buffer_reg_channel(reg_uid->registry, ua_chan, &reg_chan);
2357 if (ret < 0) {
2358 goto error;
2359 }
2360 assert(reg_chan);
2361
2362 /*
2363 * Create the buffers on the consumer side. This call populates the
2364 * ust app channel object with all streams and data object.
2365 */
2366 ret = do_consumer_create_channel(usess, ua_sess, ua_chan,
2367 app->bits_per_long, reg_uid->registry->reg.ust);
2368 if (ret < 0) {
07d2ae95
DG
2369 /*
2370 * Let's remove the previously created buffer registry channel so
2371 * it's not visible anymore in the session registry.
2372 */
2373 ust_registry_channel_del_free(reg_uid->registry->reg.ust,
2374 ua_chan->tracing_channel_id);
2375 buffer_reg_channel_remove(reg_uid->registry, reg_chan);
2376 buffer_reg_channel_destroy(reg_chan, LTTNG_DOMAIN_UST);
7972aab2
DG
2377 goto error;
2378 }
2379
2380 /*
2381 * Setup the streams and add it to the session registry.
2382 */
2383 ret = setup_buffer_reg_channel(reg_uid->registry, ua_chan, reg_chan);
2384 if (ret < 0) {
2385 goto error;
2386 }
2387
2388 }
2389
2390 /* Send buffers to the application. */
2391 ret = send_channel_uid_to_ust(reg_chan, app, ua_sess, ua_chan);
9730260e
DG
2392 if (ret < 0) {
2393 goto error;
2394 }
2395
9730260e
DG
2396error:
2397 return ret;
2398}
2399
78f0bacd 2400/*
7972aab2
DG
2401 * Create and send to the application the created buffers with per PID buffers.
2402 *
2403 * Return 0 on success else a negative value.
78f0bacd 2404 */
7972aab2
DG
2405static int create_channel_per_pid(struct ust_app *app,
2406 struct ltt_ust_session *usess, struct ust_app_session *ua_sess,
2407 struct ust_app_channel *ua_chan)
78f0bacd 2408{
8535a6d9 2409 int ret;
7972aab2 2410 struct ust_registry_session *registry;
78f0bacd 2411
7972aab2
DG
2412 assert(app);
2413 assert(usess);
2414 assert(ua_sess);
2415 assert(ua_chan);
2416
2417 DBG("UST app creating channel %s with per PID buffers", ua_chan->name);
2418
2419 rcu_read_lock();
2420
2421 registry = get_session_registry(ua_sess);
2422 assert(registry);
2423
2424 /* Create and add a new channel registry to session. */
2425 ret = ust_registry_channel_add(registry, ua_chan->key);
78f0bacd
DG
2426 if (ret < 0) {
2427 goto error;
2428 }
2429
7972aab2
DG
2430 /* Create and get channel on the consumer side. */
2431 ret = do_consumer_create_channel(usess, ua_sess, ua_chan,
2432 app->bits_per_long, registry);
2433 if (ret < 0) {
2434 goto error;
2435 }
2436
2437 ret = send_channel_pid_to_ust(app, ua_sess, ua_chan);
2438 if (ret < 0) {
2439 goto error;
2440 }
8535a6d9 2441
78f0bacd 2442error:
7972aab2 2443 rcu_read_unlock();
78f0bacd
DG
2444 return ret;
2445}
2446
2447/*
7972aab2
DG
2448 * From an already allocated ust app channel, create the channel buffers if
2449 * need and send it to the application. This MUST be called with a RCU read
2450 * side lock acquired.
2451 *
2452 * Return 0 on success or else a negative value.
78f0bacd 2453 */
7972aab2
DG
2454static int do_create_channel(struct ust_app *app,
2455 struct ltt_ust_session *usess, struct ust_app_session *ua_sess,
2456 struct ust_app_channel *ua_chan)
78f0bacd 2457{
7972aab2 2458 int ret;
78f0bacd 2459
7972aab2
DG
2460 assert(app);
2461 assert(usess);
2462 assert(ua_sess);
2463 assert(ua_chan);
2464
2465 /* Handle buffer type before sending the channel to the application. */
2466 switch (usess->buffer_type) {
2467 case LTTNG_BUFFER_PER_UID:
2468 {
2469 ret = create_channel_per_uid(app, usess, ua_sess, ua_chan);
2470 if (ret < 0) {
2471 goto error;
2472 }
2473 break;
2474 }
2475 case LTTNG_BUFFER_PER_PID:
2476 {
2477 ret = create_channel_per_pid(app, usess, ua_sess, ua_chan);
2478 if (ret < 0) {
2479 goto error;
2480 }
2481 break;
2482 }
2483 default:
2484 assert(0);
2485 ret = -EINVAL;
78f0bacd
DG
2486 goto error;
2487 }
2488
7972aab2
DG
2489 /* Initialize ust objd object using the received handle and add it. */
2490 lttng_ht_node_init_ulong(&ua_chan->ust_objd_node, ua_chan->handle);
2491 lttng_ht_add_unique_ulong(app->ust_objd, &ua_chan->ust_objd_node);
78f0bacd 2492
7972aab2
DG
2493 /* If channel is not enabled, disable it on the tracer */
2494 if (!ua_chan->enabled) {
2495 ret = disable_ust_channel(app, ua_sess, ua_chan);
2496 if (ret < 0) {
2497 goto error;
2498 }
78f0bacd
DG
2499 }
2500
2501error:
2502 return ret;
2503}
2504
284d8f55 2505/*
4d710ac2
DG
2506 * Create UST app channel and create it on the tracer. Set ua_chanp of the
2507 * newly created channel if not NULL.
d0b96690 2508 *
36b588ed 2509 * Called with UST app session lock and RCU read-side lock held.
7972aab2
DG
2510 *
2511 * Return 0 on success or else a negative value.
284d8f55 2512 */
4d710ac2
DG
2513static int create_ust_app_channel(struct ust_app_session *ua_sess,
2514 struct ltt_ust_channel *uchan, struct ust_app *app,
7972aab2 2515 enum lttng_ust_chan_type type, struct ltt_ust_session *usess,
4d710ac2 2516 struct ust_app_channel **ua_chanp)
5b4a0ec0
DG
2517{
2518 int ret = 0;
bec39940
DG
2519 struct lttng_ht_iter iter;
2520 struct lttng_ht_node_str *ua_chan_node;
5b4a0ec0
DG
2521 struct ust_app_channel *ua_chan;
2522
2523 /* Lookup channel in the ust app session */
bec39940
DG
2524 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &iter);
2525 ua_chan_node = lttng_ht_iter_get_node_str(&iter);
fc34caaa 2526 if (ua_chan_node != NULL) {
5b4a0ec0 2527 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
fc34caaa 2528 goto end;
5b4a0ec0
DG
2529 }
2530
d0b96690 2531 ua_chan = alloc_ust_app_channel(uchan->name, ua_sess, &uchan->attr);
fc34caaa
DG
2532 if (ua_chan == NULL) {
2533 /* Only malloc can fail here */
4d710ac2 2534 ret = -ENOMEM;
094d1690 2535 goto error_alloc;
fc34caaa
DG
2536 }
2537 shadow_copy_channel(ua_chan, uchan);
2538
ffe60014
DG
2539 /* Set channel type. */
2540 ua_chan->attr.type = type;
2541
7972aab2 2542 ret = do_create_channel(app, usess, ua_sess, ua_chan);
5b4a0ec0
DG
2543 if (ret < 0) {
2544 goto error;
2545 }
2546
fc34caaa 2547 DBG2("UST app create channel %s for PID %d completed", ua_chan->name,
852d0037 2548 app->pid);
fc34caaa 2549
d0b96690
DG
2550 /* Only add the channel if successful on the tracer side. */
2551 lttng_ht_add_unique_str(ua_sess->channels, &ua_chan->node);
2552
fc34caaa 2553end:
4d710ac2
DG
2554 if (ua_chanp) {
2555 *ua_chanp = ua_chan;
2556 }
2557
2558 /* Everything went well. */
2559 return 0;
5b4a0ec0
DG
2560
2561error:
d0b96690 2562 delete_ust_app_channel(ua_chan->is_sent ? app->sock : -1, ua_chan, app);
094d1690 2563error_alloc:
4d710ac2 2564 return ret;
5b4a0ec0
DG
2565}
2566
2567/*
2568 * Create UST app event and create it on the tracer side.
d0b96690
DG
2569 *
2570 * Called with ust app session mutex held.
5b4a0ec0 2571 */
edb67388
DG
2572static
2573int create_ust_app_event(struct ust_app_session *ua_sess,
2574 struct ust_app_channel *ua_chan, struct ltt_ust_event *uevent,
2575 struct ust_app *app)
284d8f55 2576{
edb67388 2577 int ret = 0;
5b4a0ec0 2578 struct ust_app_event *ua_event;
284d8f55 2579
5b4a0ec0 2580 /* Get event node */
18eace3b
DG
2581 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
2582 uevent->filter, uevent->attr.loglevel);
2583 if (ua_event != NULL) {
fc34caaa 2584 ret = -EEXIST;
edb67388
DG
2585 goto end;
2586 }
5b4a0ec0 2587
edb67388
DG
2588 /* Does not exist so create one */
2589 ua_event = alloc_ust_app_event(uevent->attr.name, &uevent->attr);
2590 if (ua_event == NULL) {
2591 /* Only malloc can failed so something is really wrong */
2592 ret = -ENOMEM;
fc34caaa 2593 goto end;
5b4a0ec0 2594 }
edb67388 2595 shadow_copy_event(ua_event, uevent);
5b4a0ec0 2596
edb67388 2597 /* Create it on the tracer side */
5b4a0ec0 2598 ret = create_ust_event(app, ua_sess, ua_chan, ua_event);
284d8f55 2599 if (ret < 0) {
fc34caaa 2600 /* Not found previously means that it does not exist on the tracer */
76f66f63 2601 assert(ret != -LTTNG_UST_ERR_EXIST);
284d8f55
DG
2602 goto error;
2603 }
2604
d0b96690 2605 add_unique_ust_app_event(ua_chan, ua_event);
284d8f55 2606
fc34caaa 2607 DBG2("UST app create event %s for PID %d completed", ua_event->name,
852d0037 2608 app->pid);
7f79d3a1 2609
edb67388 2610end:
fc34caaa
DG
2611 return ret;
2612
5b4a0ec0 2613error:
fc34caaa
DG
2614 /* Valid. Calling here is already in a read side lock */
2615 delete_ust_app_event(-1, ua_event);
edb67388 2616 return ret;
5b4a0ec0
DG
2617}
2618
2619/*
2620 * Create UST metadata and open it on the tracer side.
d0b96690 2621 *
7972aab2 2622 * Called with UST app session lock held and RCU read side lock.
5b4a0ec0
DG
2623 */
2624static int create_ust_app_metadata(struct ust_app_session *ua_sess,
d65d2de8
DG
2625 struct ust_app *app, struct consumer_output *consumer,
2626 struct ustctl_consumer_channel_attr *attr)
5b4a0ec0
DG
2627{
2628 int ret = 0;
ffe60014 2629 struct ust_app_channel *metadata;
d88aee68 2630 struct consumer_socket *socket;
7972aab2 2631 struct ust_registry_session *registry;
5b4a0ec0 2632
ffe60014
DG
2633 assert(ua_sess);
2634 assert(app);
d88aee68 2635 assert(consumer);
5b4a0ec0 2636
7972aab2
DG
2637 registry = get_session_registry(ua_sess);
2638 assert(registry);
2639
1b532a60
DG
2640 /* Metadata already exists for this registry or it was closed previously */
2641 if (registry->metadata_key || registry->metadata_closed) {
7972aab2
DG
2642 ret = 0;
2643 goto error;
5b4a0ec0
DG
2644 }
2645
ffe60014 2646 /* Allocate UST metadata */
d0b96690 2647 metadata = alloc_ust_app_channel(DEFAULT_METADATA_NAME, ua_sess, NULL);
ffe60014
DG
2648 if (!metadata) {
2649 /* malloc() failed */
2650 ret = -ENOMEM;
2651 goto error;
2652 }
5b4a0ec0 2653
d65d2de8
DG
2654 if (!attr) {
2655 /* Set default attributes for metadata. */
2656 metadata->attr.overwrite = DEFAULT_CHANNEL_OVERWRITE;
2657 metadata->attr.subbuf_size = default_get_metadata_subbuf_size();
2658 metadata->attr.num_subbuf = DEFAULT_METADATA_SUBBUF_NUM;
0a9c6494
DG
2659 metadata->attr.switch_timer_interval = DEFAULT_METADATA_SWITCH_TIMER;
2660 metadata->attr.read_timer_interval = DEFAULT_METADATA_READ_TIMER;
d65d2de8
DG
2661 metadata->attr.output = LTTNG_UST_MMAP;
2662 metadata->attr.type = LTTNG_UST_CHAN_METADATA;
2663 } else {
2664 memcpy(&metadata->attr, attr, sizeof(metadata->attr));
2665 metadata->attr.output = LTTNG_UST_MMAP;
2666 metadata->attr.type = LTTNG_UST_CHAN_METADATA;
2667 }
5b4a0ec0 2668
7972aab2
DG
2669 /* Need one fd for the channel. */
2670 ret = lttng_fd_get(LTTNG_FD_APPS, 1);
2671 if (ret < 0) {
2672 ERR("Exhausted number of available FD upon create metadata");
2673 goto error;
2674 }
2675
4dc3dfc5
DG
2676 /* Get the right consumer socket for the application. */
2677 socket = consumer_find_socket_by_bitness(app->bits_per_long, consumer);
2678 if (!socket) {
2679 ret = -EINVAL;
2680 goto error_consumer;
2681 }
2682
331744e3
JD
2683 /*
2684 * Keep metadata key so we can identify it on the consumer side. Assign it
2685 * to the registry *before* we ask the consumer so we avoid the race of the
2686 * consumer requesting the metadata and the ask_channel call on our side
2687 * did not returned yet.
2688 */
2689 registry->metadata_key = metadata->key;
2690
d88aee68
DG
2691 /*
2692 * Ask the metadata channel creation to the consumer. The metadata object
2693 * will be created by the consumer and kept their. However, the stream is
2694 * never added or monitored until we do a first push metadata to the
2695 * consumer.
2696 */
7972aab2
DG
2697 ret = ust_consumer_ask_channel(ua_sess, metadata, consumer, socket,
2698 registry);
d88aee68 2699 if (ret < 0) {
f2a444f1
DG
2700 /* Nullify the metadata key so we don't try to close it later on. */
2701 registry->metadata_key = 0;
d88aee68
DG
2702 goto error_consumer;
2703 }
2704
2705 /*
2706 * The setup command will make the metadata stream be sent to the relayd,
2707 * if applicable, and the thread managing the metadatas. This is important
2708 * because after this point, if an error occurs, the only way the stream
2709 * can be deleted is to be monitored in the consumer.
2710 */
7972aab2 2711 ret = consumer_setup_metadata(socket, metadata->key);
ffe60014 2712 if (ret < 0) {
f2a444f1
DG
2713 /* Nullify the metadata key so we don't try to close it later on. */
2714 registry->metadata_key = 0;
d88aee68 2715 goto error_consumer;
5b4a0ec0
DG
2716 }
2717
7972aab2
DG
2718 DBG2("UST metadata with key %" PRIu64 " created for app pid %d",
2719 metadata->key, app->pid);
5b4a0ec0 2720
d88aee68 2721error_consumer:
b80f0b6c 2722 lttng_fd_put(LTTNG_FD_APPS, 1);
d88aee68 2723 delete_ust_app_channel(-1, metadata, app);
5b4a0ec0 2724error:
ffe60014 2725 return ret;
5b4a0ec0
DG
2726}
2727
2728/*
2729 * Return pointer to traceable apps list.
2730 */
bec39940 2731struct lttng_ht *ust_app_get_ht(void)
5b4a0ec0
DG
2732{
2733 return ust_app_ht;
2734}
2735
2736/*
d88aee68
DG
2737 * Return ust app pointer or NULL if not found. RCU read side lock MUST be
2738 * acquired before calling this function.
5b4a0ec0
DG
2739 */
2740struct ust_app *ust_app_find_by_pid(pid_t pid)
2741{
d88aee68 2742 struct ust_app *app = NULL;
bec39940
DG
2743 struct lttng_ht_node_ulong *node;
2744 struct lttng_ht_iter iter;
5b4a0ec0 2745
bec39940
DG
2746 lttng_ht_lookup(ust_app_ht, (void *)((unsigned long) pid), &iter);
2747 node = lttng_ht_iter_get_node_ulong(&iter);
5b4a0ec0
DG
2748 if (node == NULL) {
2749 DBG2("UST app no found with pid %d", pid);
2750 goto error;
2751 }
5b4a0ec0
DG
2752
2753 DBG2("Found UST app by pid %d", pid);
2754
d88aee68 2755 app = caa_container_of(node, struct ust_app, pid_n);
5b4a0ec0
DG
2756
2757error:
d88aee68 2758 return app;
5b4a0ec0
DG
2759}
2760
d88aee68
DG
2761/*
2762 * Allocate and init an UST app object using the registration information and
2763 * the command socket. This is called when the command socket connects to the
2764 * session daemon.
2765 *
2766 * The object is returned on success or else NULL.
2767 */
d0b96690 2768struct ust_app *ust_app_create(struct ust_register_msg *msg, int sock)
5b4a0ec0 2769{
d0b96690
DG
2770 struct ust_app *lta = NULL;
2771
2772 assert(msg);
2773 assert(sock >= 0);
2774
2775 DBG3("UST app creating application for socket %d", sock);
5b4a0ec0 2776
173af62f
DG
2777 if ((msg->bits_per_long == 64 &&
2778 (uatomic_read(&ust_consumerd64_fd) == -EINVAL))
2779 || (msg->bits_per_long == 32 &&
2780 (uatomic_read(&ust_consumerd32_fd) == -EINVAL))) {
f943b0fb 2781 ERR("Registration failed: application \"%s\" (pid: %d) has "
d0b96690
DG
2782 "%d-bit long, but no consumerd for this size is available.\n",
2783 msg->name, msg->pid, msg->bits_per_long);
2784 goto error;
3f2c5fcc 2785 }
d0b96690 2786
5b4a0ec0
DG
2787 lta = zmalloc(sizeof(struct ust_app));
2788 if (lta == NULL) {
2789 PERROR("malloc");
d0b96690 2790 goto error;
5b4a0ec0
DG
2791 }
2792
2793 lta->ppid = msg->ppid;
2794 lta->uid = msg->uid;
2795 lta->gid = msg->gid;
d0b96690 2796
7753dea8 2797 lta->bits_per_long = msg->bits_per_long;
d0b96690
DG
2798 lta->uint8_t_alignment = msg->uint8_t_alignment;
2799 lta->uint16_t_alignment = msg->uint16_t_alignment;
2800 lta->uint32_t_alignment = msg->uint32_t_alignment;
2801 lta->uint64_t_alignment = msg->uint64_t_alignment;
2802 lta->long_alignment = msg->long_alignment;
2803 lta->byte_order = msg->byte_order;
2804
5b4a0ec0
DG
2805 lta->v_major = msg->major;
2806 lta->v_minor = msg->minor;
77c5447b 2807 lta->sessions = lttng_ht_new(0, LTTNG_HT_TYPE_U64);
d0b96690
DG
2808 lta->ust_objd = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
2809 lta->notify_sock = -1;
d88aee68
DG
2810
2811 /* Copy name and make sure it's NULL terminated. */
2812 strncpy(lta->name, msg->name, sizeof(lta->name));
2813 lta->name[UST_APP_PROCNAME_LEN] = '\0';
2814
2815 /*
2816 * Before this can be called, when receiving the registration information,
2817 * the application compatibility is checked. So, at this point, the
2818 * application can work with this session daemon.
2819 */
d0b96690 2820 lta->compatible = 1;
5b4a0ec0 2821
852d0037 2822 lta->pid = msg->pid;
d0b96690 2823 lttng_ht_node_init_ulong(&lta->pid_n, (unsigned long) lta->pid);
852d0037 2824 lta->sock = sock;
d0b96690 2825 lttng_ht_node_init_ulong(&lta->sock_n, (unsigned long) lta->sock);
5b4a0ec0 2826
d42f20df
DG
2827 CDS_INIT_LIST_HEAD(&lta->teardown_head);
2828
d0b96690
DG
2829error:
2830 return lta;
2831}
2832
d88aee68
DG
2833/*
2834 * For a given application object, add it to every hash table.
2835 */
d0b96690
DG
2836void ust_app_add(struct ust_app *app)
2837{
2838 assert(app);
2839 assert(app->notify_sock >= 0);
2840
5b4a0ec0 2841 rcu_read_lock();
852d0037
DG
2842
2843 /*
2844 * On a re-registration, we want to kick out the previous registration of
2845 * that pid
2846 */
d0b96690 2847 lttng_ht_add_replace_ulong(ust_app_ht, &app->pid_n);
852d0037
DG
2848
2849 /*
2850 * The socket _should_ be unique until _we_ call close. So, a add_unique
2851 * for the ust_app_ht_by_sock is used which asserts fail if the entry was
2852 * already in the table.
2853 */
d0b96690 2854 lttng_ht_add_unique_ulong(ust_app_ht_by_sock, &app->sock_n);
852d0037 2855
d0b96690
DG
2856 /* Add application to the notify socket hash table. */
2857 lttng_ht_node_init_ulong(&app->notify_sock_n, app->notify_sock);
2858 lttng_ht_add_unique_ulong(ust_app_ht_by_notify_sock, &app->notify_sock_n);
5b4a0ec0 2859
d0b96690 2860 DBG("App registered with pid:%d ppid:%d uid:%d gid:%d sock:%d name:%s "
d88aee68
DG
2861 "notify_sock:%d (version %d.%d)", app->pid, app->ppid, app->uid,
2862 app->gid, app->sock, app->name, app->notify_sock, app->v_major,
2863 app->v_minor);
5b4a0ec0 2864
d0b96690
DG
2865 rcu_read_unlock();
2866}
2867
d88aee68
DG
2868/*
2869 * Set the application version into the object.
2870 *
2871 * Return 0 on success else a negative value either an errno code or a
2872 * LTTng-UST error code.
2873 */
d0b96690
DG
2874int ust_app_version(struct ust_app *app)
2875{
d88aee68
DG
2876 int ret;
2877
d0b96690 2878 assert(app);
d88aee68
DG
2879
2880 ret = ustctl_tracer_version(app->sock, &app->version);
2881 if (ret < 0) {
2882 if (ret != -LTTNG_UST_ERR_EXITING && ret != -EPIPE) {
2883 ERR("UST app %d verson failed with ret %d", app->sock, ret);
2884 } else {
2885 DBG3("UST app %d verion failed. Application is dead", app->sock);
2886 }
2887 }
2888
2889 return ret;
5b4a0ec0
DG
2890}
2891
2892/*
2893 * Unregister app by removing it from the global traceable app list and freeing
2894 * the data struct.
2895 *
2896 * The socket is already closed at this point so no close to sock.
2897 */
2898void ust_app_unregister(int sock)
2899{
2900 struct ust_app *lta;
bec39940
DG
2901 struct lttng_ht_node_ulong *node;
2902 struct lttng_ht_iter iter;
d42f20df 2903 struct ust_app_session *ua_sess;
525b0740 2904 int ret;
5b4a0ec0
DG
2905
2906 rcu_read_lock();
886459c6 2907
5b4a0ec0 2908 /* Get the node reference for a call_rcu */
852d0037 2909 lttng_ht_lookup(ust_app_ht_by_sock, (void *)((unsigned long) sock), &iter);
bec39940 2910 node = lttng_ht_iter_get_node_ulong(&iter);
d0b96690 2911 assert(node);
284d8f55 2912
852d0037 2913 lta = caa_container_of(node, struct ust_app, sock_n);
852d0037
DG
2914 DBG("PID %d unregistering with sock %d", lta->pid, sock);
2915
886459c6 2916 /* Remove application from PID hash table */
852d0037
DG
2917 ret = lttng_ht_del(ust_app_ht_by_sock, &iter);
2918 assert(!ret);
2919
d88aee68
DG
2920 /*
2921 * Remove application from notify hash table. The thread handling the
2922 * notify socket could have deleted the node so ignore on error because
2923 * either way it's valid. The close of that socket is handled by the other
2924 * thread.
2925 */
d0b96690 2926 iter.iter.node = &lta->notify_sock_n.node;
d88aee68 2927 (void) lttng_ht_del(ust_app_ht_by_notify_sock, &iter);
852d0037 2928
5b98a774
DG
2929 /*
2930 * Ignore return value since the node might have been removed before by an
2931 * add replace during app registration because the PID can be reassigned by
2932 * the OS.
2933 */
d0b96690 2934 iter.iter.node = &lta->pid_n.node;
bec39940 2935 ret = lttng_ht_del(ust_app_ht, &iter);
5b98a774
DG
2936 if (ret) {
2937 DBG3("Unregister app by PID %d failed. This can happen on pid reuse",
2938 lta->pid);
2939 }
852d0037 2940
d42f20df
DG
2941 /* Remove sessions so they are not visible during deletion.*/
2942 cds_lfht_for_each_entry(lta->sessions->ht, &iter.iter, ua_sess,
2943 node.node) {
7972aab2
DG
2944 struct ust_registry_session *registry;
2945
d42f20df
DG
2946 ret = lttng_ht_del(lta->sessions, &iter);
2947 if (ret) {
2948 /* The session was already removed so scheduled for teardown. */
2949 continue;
2950 }
2951
2952 /*
2953 * Add session to list for teardown. This is safe since at this point we
2954 * are the only one using this list.
2955 */
d88aee68
DG
2956 pthread_mutex_lock(&ua_sess->lock);
2957
2958 /*
2959 * Normally, this is done in the delete session process which is
2960 * executed in the call rcu below. However, upon registration we can't
2961 * afford to wait for the grace period before pushing data or else the
2962 * data pending feature can race between the unregistration and stop
2963 * command where the data pending command is sent *before* the grace
2964 * period ended.
2965 *
2966 * The close metadata below nullifies the metadata pointer in the
2967 * session so the delete session will NOT push/close a second time.
2968 */
7972aab2 2969 registry = get_session_registry(ua_sess);
1b532a60 2970 if (registry && !registry->metadata_closed) {
7972aab2
DG
2971 /* Push metadata for application before freeing the application. */
2972 (void) push_metadata(registry, ua_sess->consumer);
2973
2974 /*
2975 * Don't ask to close metadata for global per UID buffers. Close
1b532a60
DG
2976 * metadata only on destroy trace session in this case. Also, the
2977 * previous push metadata could have flag the metadata registry to
2978 * close so don't send a close command if closed.
7972aab2 2979 */
1b532a60
DG
2980 if (ua_sess->buffer_type != LTTNG_BUFFER_PER_UID &&
2981 !registry->metadata_closed) {
7972aab2
DG
2982 /* And ask to close it for this session registry. */
2983 (void) close_metadata(registry, ua_sess->consumer);
2984 }
2985 }
d88aee68 2986
d42f20df 2987 cds_list_add(&ua_sess->teardown_node, &lta->teardown_head);
d88aee68 2988 pthread_mutex_unlock(&ua_sess->lock);
d42f20df
DG
2989 }
2990
852d0037
DG
2991 /* Free memory */
2992 call_rcu(&lta->pid_n.head, delete_ust_app_rcu);
2993
5b4a0ec0
DG
2994 rcu_read_unlock();
2995 return;
284d8f55
DG
2996}
2997
2998/*
5b4a0ec0 2999 * Return traceable_app_count
284d8f55 3000 */
5b4a0ec0 3001unsigned long ust_app_list_count(void)
284d8f55 3002{
5b4a0ec0 3003 unsigned long count;
284d8f55 3004
5b4a0ec0 3005 rcu_read_lock();
bec39940 3006 count = lttng_ht_get_count(ust_app_ht);
5b4a0ec0 3007 rcu_read_unlock();
284d8f55 3008
5b4a0ec0 3009 return count;
284d8f55
DG
3010}
3011
5b4a0ec0
DG
3012/*
3013 * Fill events array with all events name of all registered apps.
3014 */
3015int ust_app_list_events(struct lttng_event **events)
421cb601 3016{
5b4a0ec0
DG
3017 int ret, handle;
3018 size_t nbmem, count = 0;
bec39940 3019 struct lttng_ht_iter iter;
5b4a0ec0 3020 struct ust_app *app;
c617c0c6 3021 struct lttng_event *tmp_event;
421cb601 3022
5b4a0ec0 3023 nbmem = UST_APP_EVENT_LIST_SIZE;
c617c0c6
MD
3024 tmp_event = zmalloc(nbmem * sizeof(struct lttng_event));
3025 if (tmp_event == NULL) {
5b4a0ec0
DG
3026 PERROR("zmalloc ust app events");
3027 ret = -ENOMEM;
421cb601
DG
3028 goto error;
3029 }
3030
5b4a0ec0 3031 rcu_read_lock();
421cb601 3032
852d0037 3033 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
90eaa0d2 3034 struct lttng_ust_tracepoint_iter uiter;
ac3bd9c0 3035
840cb59c 3036 health_code_update();
86acf0da 3037
e0c7ec2b
DG
3038 if (!app->compatible) {
3039 /*
3040 * TODO: In time, we should notice the caller of this error by
3041 * telling him that this is a version error.
3042 */
3043 continue;
3044 }
852d0037 3045 handle = ustctl_tracepoint_list(app->sock);
5b4a0ec0 3046 if (handle < 0) {
ffe60014
DG
3047 if (handle != -EPIPE && handle != -LTTNG_UST_ERR_EXITING) {
3048 ERR("UST app list events getting handle failed for app pid %d",
3049 app->pid);
3050 }
5b4a0ec0
DG
3051 continue;
3052 }
421cb601 3053
852d0037 3054 while ((ret = ustctl_tracepoint_list_get(app->sock, handle,
fb54cdbf 3055 &uiter)) != -LTTNG_UST_ERR_NOENT) {
ffe60014
DG
3056 /* Handle ustctl error. */
3057 if (ret < 0) {
3058 free(tmp_event);
3059 if (ret != -LTTNG_UST_ERR_EXITING || ret != -EPIPE) {
3060 ERR("UST app tp list get failed for app %d with ret %d",
3061 app->sock, ret);
3062 } else {
3063 DBG3("UST app tp list get failed. Application is dead");
47cb9d4a
DG
3064 /*
3065 * This is normal behavior, an application can die during the
3066 * creation process. Don't report an error so the execution can
3067 * continue normally. Continue normal execution.
3068 */
3069 break;
ffe60014
DG
3070 }
3071 goto rcu_error;
3072 }
3073
840cb59c 3074 health_code_update();
815564d8 3075 if (count >= nbmem) {
d7b3776f 3076 /* In case the realloc fails, we free the memory */
c617c0c6
MD
3077 void *ptr;
3078
815564d8
MD
3079 DBG2("Reallocating event list from %zu to %zu entries", nbmem,
3080 2 * nbmem);
3081 nbmem *= 2;
c617c0c6
MD
3082 ptr = realloc(tmp_event, nbmem * sizeof(struct lttng_event));
3083 if (ptr == NULL) {
5b4a0ec0 3084 PERROR("realloc ust app events");
c617c0c6 3085 free(tmp_event);
5b4a0ec0
DG
3086 ret = -ENOMEM;
3087 goto rcu_error;
3088 }
c617c0c6 3089 tmp_event = ptr;
5b4a0ec0 3090 }
c617c0c6
MD
3091 memcpy(tmp_event[count].name, uiter.name, LTTNG_UST_SYM_NAME_LEN);
3092 tmp_event[count].loglevel = uiter.loglevel;
3093 tmp_event[count].type = (enum lttng_event_type) LTTNG_UST_TRACEPOINT;
3094 tmp_event[count].pid = app->pid;
3095 tmp_event[count].enabled = -1;
5b4a0ec0 3096 count++;
421cb601 3097 }
421cb601
DG
3098 }
3099
5b4a0ec0 3100 ret = count;
c617c0c6 3101 *events = tmp_event;
421cb601 3102
5b4a0ec0 3103 DBG2("UST app list events done (%zu events)", count);
421cb601 3104
5b4a0ec0
DG
3105rcu_error:
3106 rcu_read_unlock();
421cb601 3107error:
840cb59c 3108 health_code_update();
5b4a0ec0 3109 return ret;
421cb601
DG
3110}
3111
f37d259d
MD
3112/*
3113 * Fill events array with all events name of all registered apps.
3114 */
3115int ust_app_list_event_fields(struct lttng_event_field **fields)
3116{
3117 int ret, handle;
3118 size_t nbmem, count = 0;
3119 struct lttng_ht_iter iter;
3120 struct ust_app *app;
c617c0c6 3121 struct lttng_event_field *tmp_event;
f37d259d
MD
3122
3123 nbmem = UST_APP_EVENT_LIST_SIZE;
c617c0c6
MD
3124 tmp_event = zmalloc(nbmem * sizeof(struct lttng_event_field));
3125 if (tmp_event == NULL) {
f37d259d
MD
3126 PERROR("zmalloc ust app event fields");
3127 ret = -ENOMEM;
3128 goto error;
3129 }
3130
3131 rcu_read_lock();
3132
3133 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
3134 struct lttng_ust_field_iter uiter;
3135
840cb59c 3136 health_code_update();
86acf0da 3137
f37d259d
MD
3138 if (!app->compatible) {
3139 /*
3140 * TODO: In time, we should notice the caller of this error by
3141 * telling him that this is a version error.
3142 */
3143 continue;
3144 }
3145 handle = ustctl_tracepoint_field_list(app->sock);
3146 if (handle < 0) {
ffe60014
DG
3147 if (handle != -EPIPE && handle != -LTTNG_UST_ERR_EXITING) {
3148 ERR("UST app list field getting handle failed for app pid %d",
3149 app->pid);
3150 }
f37d259d
MD
3151 continue;
3152 }
3153
3154 while ((ret = ustctl_tracepoint_field_list_get(app->sock, handle,
fb54cdbf 3155 &uiter)) != -LTTNG_UST_ERR_NOENT) {
ffe60014
DG
3156 /* Handle ustctl error. */
3157 if (ret < 0) {
3158 free(tmp_event);
3159 if (ret != -LTTNG_UST_ERR_EXITING || ret != -EPIPE) {
3160 ERR("UST app tp list field failed for app %d with ret %d",
3161 app->sock, ret);
3162 } else {
3163 DBG3("UST app tp list field failed. Application is dead");
47cb9d4a
DG
3164 /*
3165 * This is normal behavior, an application can die during the
3166 * creation process. Don't report an error so the execution can
3167 * continue normally.
3168 */
3169 break;
ffe60014
DG
3170 }
3171 goto rcu_error;
3172 }
3173
840cb59c 3174 health_code_update();
f37d259d 3175 if (count >= nbmem) {
d7b3776f 3176 /* In case the realloc fails, we free the memory */
c617c0c6
MD
3177 void *ptr;
3178
f37d259d
MD
3179 DBG2("Reallocating event field list from %zu to %zu entries", nbmem,
3180 2 * nbmem);
3181 nbmem *= 2;
c617c0c6
MD
3182 ptr = realloc(tmp_event, nbmem * sizeof(struct lttng_event_field));
3183 if (ptr == NULL) {
f37d259d 3184 PERROR("realloc ust app event fields");
c617c0c6 3185 free(tmp_event);
f37d259d
MD
3186 ret = -ENOMEM;
3187 goto rcu_error;
3188 }
c617c0c6 3189 tmp_event = ptr;
f37d259d 3190 }
f37d259d 3191
c617c0c6 3192 memcpy(tmp_event[count].field_name, uiter.field_name, LTTNG_UST_SYM_NAME_LEN);
13d9b823
DG
3193 /* Mapping between these enums matches 1 to 1. */
3194 tmp_event[count].type = (enum lttng_event_field_type) uiter.type;
c617c0c6 3195 tmp_event[count].nowrite = uiter.nowrite;
f37d259d 3196
c617c0c6
MD
3197 memcpy(tmp_event[count].event.name, uiter.event_name, LTTNG_UST_SYM_NAME_LEN);
3198 tmp_event[count].event.loglevel = uiter.loglevel;
13d9b823 3199 tmp_event[count].event.type = LTTNG_EVENT_TRACEPOINT;
c617c0c6
MD
3200 tmp_event[count].event.pid = app->pid;
3201 tmp_event[count].event.enabled = -1;
f37d259d
MD
3202 count++;
3203 }
3204 }
3205
3206 ret = count;
c617c0c6 3207 *fields = tmp_event;
f37d259d
MD
3208
3209 DBG2("UST app list event fields done (%zu events)", count);
3210
3211rcu_error:
3212 rcu_read_unlock();
3213error:
840cb59c 3214 health_code_update();
f37d259d
MD
3215 return ret;
3216}
3217
5b4a0ec0
DG
3218/*
3219 * Free and clean all traceable apps of the global list.
36b588ed
MD
3220 *
3221 * Should _NOT_ be called with RCU read-side lock held.
5b4a0ec0
DG
3222 */
3223void ust_app_clean_list(void)
421cb601 3224{
5b4a0ec0 3225 int ret;
659ed79f 3226 struct ust_app *app;
bec39940 3227 struct lttng_ht_iter iter;
421cb601 3228
5b4a0ec0 3229 DBG2("UST app cleaning registered apps hash table");
421cb601 3230
5b4a0ec0 3231 rcu_read_lock();
421cb601 3232
659ed79f 3233 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
bec39940 3234 ret = lttng_ht_del(ust_app_ht, &iter);
525b0740 3235 assert(!ret);
659ed79f 3236 call_rcu(&app->pid_n.head, delete_ust_app_rcu);
421cb601
DG
3237 }
3238
852d0037 3239 /* Cleanup socket hash table */
659ed79f
DG
3240 cds_lfht_for_each_entry(ust_app_ht_by_sock->ht, &iter.iter, app,
3241 sock_n.node) {
852d0037 3242 ret = lttng_ht_del(ust_app_ht_by_sock, &iter);
bec39940
DG
3243 assert(!ret);
3244 }
852d0037 3245
d88aee68
DG
3246 /* Cleanup notify socket hash table */
3247 cds_lfht_for_each_entry(ust_app_ht_by_notify_sock->ht, &iter.iter, app,
3248 notify_sock_n.node) {
3249 ret = lttng_ht_del(ust_app_ht_by_notify_sock, &iter);
3250 assert(!ret);
3251 }
36b588ed 3252 rcu_read_unlock();
d88aee68 3253
bec39940 3254 /* Destroy is done only when the ht is empty */
0b2dc8df
MD
3255 ht_cleanup_push(ust_app_ht);
3256 ht_cleanup_push(ust_app_ht_by_sock);
3257 ht_cleanup_push(ust_app_ht_by_notify_sock);
5b4a0ec0
DG
3258}
3259
3260/*
3261 * Init UST app hash table.
3262 */
3263void ust_app_ht_alloc(void)
3264{
bec39940 3265 ust_app_ht = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
852d0037 3266 ust_app_ht_by_sock = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
d0b96690 3267 ust_app_ht_by_notify_sock = lttng_ht_new(0, LTTNG_HT_TYPE_ULONG);
421cb601
DG
3268}
3269
78f0bacd
DG
3270/*
3271 * For a specific UST session, disable the channel for all registered apps.
3272 */
35a9059d 3273int ust_app_disable_channel_glb(struct ltt_ust_session *usess,
78f0bacd
DG
3274 struct ltt_ust_channel *uchan)
3275{
3276 int ret = 0;
bec39940
DG
3277 struct lttng_ht_iter iter;
3278 struct lttng_ht_node_str *ua_chan_node;
78f0bacd
DG
3279 struct ust_app *app;
3280 struct ust_app_session *ua_sess;
8535a6d9 3281 struct ust_app_channel *ua_chan;
78f0bacd
DG
3282
3283 if (usess == NULL || uchan == NULL) {
3284 ERR("Disabling UST global channel with NULL values");
3285 ret = -1;
3286 goto error;
3287 }
3288
77c5447b 3289 DBG2("UST app disabling channel %s from global domain for session id %" PRIu64,
a991f516 3290 uchan->name, usess->id);
78f0bacd
DG
3291
3292 rcu_read_lock();
3293
3294 /* For every registered applications */
852d0037 3295 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
bec39940 3296 struct lttng_ht_iter uiter;
e0c7ec2b
DG
3297 if (!app->compatible) {
3298 /*
3299 * TODO: In time, we should notice the caller of this error by
3300 * telling him that this is a version error.
3301 */
3302 continue;
3303 }
78f0bacd
DG
3304 ua_sess = lookup_session_by_app(usess, app);
3305 if (ua_sess == NULL) {
3306 continue;
3307 }
3308
8535a6d9 3309 /* Get channel */
bec39940
DG
3310 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
3311 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
8535a6d9
DG
3312 /* If the session if found for the app, the channel must be there */
3313 assert(ua_chan_node);
3314
3315 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
3316 /* The channel must not be already disabled */
3317 assert(ua_chan->enabled == 1);
3318
3319 /* Disable channel onto application */
3320 ret = disable_ust_app_channel(ua_sess, ua_chan, app);
78f0bacd
DG
3321 if (ret < 0) {
3322 /* XXX: We might want to report this error at some point... */
3323 continue;
3324 }
3325 }
3326
3327 rcu_read_unlock();
3328
3329error:
3330 return ret;
3331}
3332
3333/*
3334 * For a specific UST session, enable the channel for all registered apps.
3335 */
35a9059d 3336int ust_app_enable_channel_glb(struct ltt_ust_session *usess,
78f0bacd
DG
3337 struct ltt_ust_channel *uchan)
3338{
3339 int ret = 0;
bec39940 3340 struct lttng_ht_iter iter;
78f0bacd
DG
3341 struct ust_app *app;
3342 struct ust_app_session *ua_sess;
3343
3344 if (usess == NULL || uchan == NULL) {
3345 ERR("Adding UST global channel to NULL values");
3346 ret = -1;
3347 goto error;
3348 }
3349
77c5447b 3350 DBG2("UST app enabling channel %s to global domain for session id %" PRIu64,
a991f516 3351 uchan->name, usess->id);
78f0bacd
DG
3352
3353 rcu_read_lock();
3354
3355 /* For every registered applications */
852d0037 3356 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
3357 if (!app->compatible) {
3358 /*
3359 * TODO: In time, we should notice the caller of this error by
3360 * telling him that this is a version error.
3361 */
3362 continue;
3363 }
78f0bacd
DG
3364 ua_sess = lookup_session_by_app(usess, app);
3365 if (ua_sess == NULL) {
3366 continue;
3367 }
3368
3369 /* Enable channel onto application */
3370 ret = enable_ust_app_channel(ua_sess, uchan, app);
3371 if (ret < 0) {
3372 /* XXX: We might want to report this error at some point... */
3373 continue;
3374 }
3375 }
3376
3377 rcu_read_unlock();
3378
3379error:
3380 return ret;
3381}
3382
b0a40d28
DG
3383/*
3384 * Disable an event in a channel and for a specific session.
3385 */
35a9059d
DG
3386int ust_app_disable_event_glb(struct ltt_ust_session *usess,
3387 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent)
b0a40d28
DG
3388{
3389 int ret = 0;
bec39940
DG
3390 struct lttng_ht_iter iter, uiter;
3391 struct lttng_ht_node_str *ua_chan_node, *ua_event_node;
b0a40d28
DG
3392 struct ust_app *app;
3393 struct ust_app_session *ua_sess;
3394 struct ust_app_channel *ua_chan;
3395 struct ust_app_event *ua_event;
3396
3397 DBG("UST app disabling event %s for all apps in channel "
77c5447b
MD
3398 "%s for session id %" PRIu64,
3399 uevent->attr.name, uchan->name, usess->id);
b0a40d28
DG
3400
3401 rcu_read_lock();
3402
3403 /* For all registered applications */
852d0037 3404 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
3405 if (!app->compatible) {
3406 /*
3407 * TODO: In time, we should notice the caller of this error by
3408 * telling him that this is a version error.
3409 */
3410 continue;
3411 }
b0a40d28
DG
3412 ua_sess = lookup_session_by_app(usess, app);
3413 if (ua_sess == NULL) {
3414 /* Next app */
3415 continue;
3416 }
3417
3418 /* Lookup channel in the ust app session */
bec39940
DG
3419 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
3420 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
b0a40d28 3421 if (ua_chan_node == NULL) {
77c5447b 3422 DBG2("Channel %s not found in session id %" PRIu64 " for app pid %d."
852d0037 3423 "Skipping", uchan->name, usess->id, app->pid);
b0a40d28
DG
3424 continue;
3425 }
3426 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
3427
bec39940
DG
3428 lttng_ht_lookup(ua_chan->events, (void *)uevent->attr.name, &uiter);
3429 ua_event_node = lttng_ht_iter_get_node_str(&uiter);
b0a40d28
DG
3430 if (ua_event_node == NULL) {
3431 DBG2("Event %s not found in channel %s for app pid %d."
852d0037 3432 "Skipping", uevent->attr.name, uchan->name, app->pid);
b0a40d28
DG
3433 continue;
3434 }
3435 ua_event = caa_container_of(ua_event_node, struct ust_app_event, node);
3436
7f79d3a1 3437 ret = disable_ust_app_event(ua_sess, ua_event, app);
b0a40d28
DG
3438 if (ret < 0) {
3439 /* XXX: Report error someday... */
3440 continue;
3441 }
3442 }
3443
3444 rcu_read_unlock();
3445
3446 return ret;
3447}
3448
9730260e 3449/*
edb67388 3450 * For a specific UST session and UST channel, the event for all
9730260e
DG
3451 * registered apps.
3452 */
35a9059d 3453int ust_app_disable_all_event_glb(struct ltt_ust_session *usess,
9730260e
DG
3454 struct ltt_ust_channel *uchan)
3455{
3456 int ret = 0;
bec39940
DG
3457 struct lttng_ht_iter iter, uiter;
3458 struct lttng_ht_node_str *ua_chan_node;
9730260e
DG
3459 struct ust_app *app;
3460 struct ust_app_session *ua_sess;
3461 struct ust_app_channel *ua_chan;
3462 struct ust_app_event *ua_event;
3463
3464 DBG("UST app disabling all event for all apps in channel "
77c5447b 3465 "%s for session id %" PRIu64, uchan->name, usess->id);
9730260e
DG
3466
3467 rcu_read_lock();
3468
3469 /* For all registered applications */
852d0037 3470 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
3471 if (!app->compatible) {
3472 /*
3473 * TODO: In time, we should notice the caller of this error by
3474 * telling him that this is a version error.
3475 */
3476 continue;
3477 }
9730260e 3478 ua_sess = lookup_session_by_app(usess, app);
c4a1715b
DG
3479 if (!ua_sess) {
3480 /* The application has problem or is probably dead. */
3481 continue;
3482 }
9730260e
DG
3483
3484 /* Lookup channel in the ust app session */
bec39940
DG
3485 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
3486 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
edb67388
DG
3487 /* If the channel is not found, there is a code flow error */
3488 assert(ua_chan_node);
3489
9730260e
DG
3490 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
3491
3492 /* Disable each events of channel */
bec39940
DG
3493 cds_lfht_for_each_entry(ua_chan->events->ht, &uiter.iter, ua_event,
3494 node.node) {
7f79d3a1 3495 ret = disable_ust_app_event(ua_sess, ua_event, app);
9730260e
DG
3496 if (ret < 0) {
3497 /* XXX: Report error someday... */
3498 continue;
3499 }
3500 }
3501 }
3502
3503 rcu_read_unlock();
3504
3505 return ret;
3506}
3507
421cb601 3508/*
5b4a0ec0 3509 * For a specific UST session, create the channel for all registered apps.
421cb601 3510 */
35a9059d 3511int ust_app_create_channel_glb(struct ltt_ust_session *usess,
48842b30
DG
3512 struct ltt_ust_channel *uchan)
3513{
3d8ca23b 3514 int ret = 0, created;
bec39940 3515 struct lttng_ht_iter iter;
48842b30 3516 struct ust_app *app;
3d8ca23b 3517 struct ust_app_session *ua_sess = NULL;
48842b30 3518
fc34caaa
DG
3519 /* Very wrong code flow */
3520 assert(usess);
3521 assert(uchan);
421cb601 3522
77c5447b 3523 DBG2("UST app adding channel %s to UST domain for session id %" PRIu64,
a991f516 3524 uchan->name, usess->id);
48842b30
DG
3525
3526 rcu_read_lock();
421cb601 3527
5b4a0ec0 3528 /* For every registered applications */
852d0037 3529 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
3530 if (!app->compatible) {
3531 /*
3532 * TODO: In time, we should notice the caller of this error by
3533 * telling him that this is a version error.
3534 */
3535 continue;
3536 }
edb67388
DG
3537 /*
3538 * Create session on the tracer side and add it to app session HT. Note
3539 * that if session exist, it will simply return a pointer to the ust
3540 * app session.
3541 */
3d8ca23b
DG
3542 ret = create_ust_app_session(usess, app, &ua_sess, &created);
3543 if (ret < 0) {
3544 switch (ret) {
3545 case -ENOTCONN:
3546 /*
3547 * The application's socket is not valid. Either a bad socket
3548 * or a timeout on it. We can't inform the caller that for a
3549 * specific app, the session failed so lets continue here.
3550 */
3551 continue;
3552 case -ENOMEM:
3553 default:
3554 goto error_rcu_unlock;
3555 }
48842b30 3556 }
3d8ca23b 3557 assert(ua_sess);
48842b30 3558
d0b96690 3559 pthread_mutex_lock(&ua_sess->lock);
d65d2de8
DG
3560 if (!strncmp(uchan->name, DEFAULT_METADATA_NAME,
3561 sizeof(uchan->name))) {
3562 struct ustctl_consumer_channel_attr attr;
3563 copy_channel_attr_to_ustctl(&attr, &uchan->attr);
3564 ret = create_ust_app_metadata(ua_sess, app, usess->consumer,
3565 &attr);
3566 } else {
3567 /* Create channel onto application. We don't need the chan ref. */
3568 ret = create_ust_app_channel(ua_sess, uchan, app,
3569 LTTNG_UST_CHAN_PER_CPU, usess, NULL);
3570 }
d0b96690 3571 pthread_mutex_unlock(&ua_sess->lock);
3d8ca23b
DG
3572 if (ret < 0) {
3573 if (ret == -ENOMEM) {
3574 /* No more memory is a fatal error. Stop right now. */
3575 goto error_rcu_unlock;
3576 }
3577 /* Cleanup the created session if it's the case. */
3578 if (created) {
d0b96690 3579 destroy_app_session(app, ua_sess);
3d8ca23b 3580 }
48842b30 3581 }
48842b30 3582 }
5b4a0ec0 3583
95e047ff 3584error_rcu_unlock:
48842b30 3585 rcu_read_unlock();
3c14c33f 3586 return ret;
48842b30
DG
3587}
3588
5b4a0ec0 3589/*
edb67388 3590 * Enable event for a specific session and channel on the tracer.
5b4a0ec0 3591 */
35a9059d 3592int ust_app_enable_event_glb(struct ltt_ust_session *usess,
48842b30
DG
3593 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent)
3594{
3595 int ret = 0;
bec39940 3596 struct lttng_ht_iter iter, uiter;
18eace3b 3597 struct lttng_ht_node_str *ua_chan_node;
48842b30
DG
3598 struct ust_app *app;
3599 struct ust_app_session *ua_sess;
3600 struct ust_app_channel *ua_chan;
3601 struct ust_app_event *ua_event;
48842b30 3602
77c5447b 3603 DBG("UST app enabling event %s for all apps for session id %" PRIu64,
a991f516 3604 uevent->attr.name, usess->id);
48842b30 3605
edb67388
DG
3606 /*
3607 * NOTE: At this point, this function is called only if the session and
3608 * channel passed are already created for all apps. and enabled on the
3609 * tracer also.
3610 */
3611
48842b30 3612 rcu_read_lock();
421cb601
DG
3613
3614 /* For all registered applications */
852d0037 3615 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
3616 if (!app->compatible) {
3617 /*
3618 * TODO: In time, we should notice the caller of this error by
3619 * telling him that this is a version error.
3620 */
3621 continue;
3622 }
edb67388 3623 ua_sess = lookup_session_by_app(usess, app);
c4a1715b
DG
3624 if (!ua_sess) {
3625 /* The application has problem or is probably dead. */
3626 continue;
3627 }
ba767faf 3628
d0b96690
DG
3629 pthread_mutex_lock(&ua_sess->lock);
3630
edb67388 3631 /* Lookup channel in the ust app session */
bec39940
DG
3632 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
3633 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
edb67388
DG
3634 /* If the channel is not found, there is a code flow error */
3635 assert(ua_chan_node);
3636
3637 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
3638
18eace3b
DG
3639 /* Get event node */
3640 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
3641 uevent->filter, uevent->attr.loglevel);
3642 if (ua_event == NULL) {
7f79d3a1 3643 DBG3("UST app enable event %s not found for app PID %d."
852d0037 3644 "Skipping app", uevent->attr.name, app->pid);
d0b96690 3645 goto next_app;
35a9059d 3646 }
35a9059d
DG
3647
3648 ret = enable_ust_app_event(ua_sess, ua_event, app);
3649 if (ret < 0) {
d0b96690 3650 pthread_mutex_unlock(&ua_sess->lock);
7f79d3a1 3651 goto error;
48842b30 3652 }
d0b96690
DG
3653 next_app:
3654 pthread_mutex_unlock(&ua_sess->lock);
edb67388
DG
3655 }
3656
7f79d3a1 3657error:
edb67388 3658 rcu_read_unlock();
edb67388
DG
3659 return ret;
3660}
3661
3662/*
3663 * For a specific existing UST session and UST channel, creates the event for
3664 * all registered apps.
3665 */
35a9059d 3666int ust_app_create_event_glb(struct ltt_ust_session *usess,
edb67388
DG
3667 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent)
3668{
3669 int ret = 0;
bec39940
DG
3670 struct lttng_ht_iter iter, uiter;
3671 struct lttng_ht_node_str *ua_chan_node;
edb67388
DG
3672 struct ust_app *app;
3673 struct ust_app_session *ua_sess;
3674 struct ust_app_channel *ua_chan;
3675
77c5447b 3676 DBG("UST app creating event %s for all apps for session id %" PRIu64,
a991f516 3677 uevent->attr.name, usess->id);
edb67388 3678
edb67388
DG
3679 rcu_read_lock();
3680
3681 /* For all registered applications */
852d0037 3682 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
3683 if (!app->compatible) {
3684 /*
3685 * TODO: In time, we should notice the caller of this error by
3686 * telling him that this is a version error.
3687 */
3688 continue;
3689 }
edb67388 3690 ua_sess = lookup_session_by_app(usess, app);
c4a1715b
DG
3691 if (!ua_sess) {
3692 /* The application has problem or is probably dead. */
3693 continue;
3694 }
48842b30 3695
d0b96690 3696 pthread_mutex_lock(&ua_sess->lock);
48842b30 3697 /* Lookup channel in the ust app session */
bec39940
DG
3698 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
3699 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
edb67388
DG
3700 /* If the channel is not found, there is a code flow error */
3701 assert(ua_chan_node);
3702
48842b30
DG
3703 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
3704
edb67388 3705 ret = create_ust_app_event(ua_sess, ua_chan, uevent, app);
d0b96690 3706 pthread_mutex_unlock(&ua_sess->lock);
edb67388 3707 if (ret < 0) {
49c336c1 3708 if (ret != -LTTNG_UST_ERR_EXIST) {
fc34caaa
DG
3709 /* Possible value at this point: -ENOMEM. If so, we stop! */
3710 break;
3711 }
3712 DBG2("UST app event %s already exist on app PID %d",
852d0037 3713 uevent->attr.name, app->pid);
5b4a0ec0 3714 continue;
48842b30 3715 }
48842b30 3716 }
5b4a0ec0 3717
48842b30
DG
3718 rcu_read_unlock();
3719
3720 return ret;
3721}
3722
5b4a0ec0
DG
3723/*
3724 * Start tracing for a specific UST session and app.
3725 */
b34cbebf 3726static
421cb601 3727int ust_app_start_trace(struct ltt_ust_session *usess, struct ust_app *app)
48842b30
DG
3728{
3729 int ret = 0;
48842b30 3730 struct ust_app_session *ua_sess;
48842b30 3731
852d0037 3732 DBG("Starting tracing for ust app pid %d", app->pid);
5cf5d0e7 3733
509cbaf8
MD
3734 rcu_read_lock();
3735
e0c7ec2b
DG
3736 if (!app->compatible) {
3737 goto end;
3738 }
3739
421cb601
DG
3740 ua_sess = lookup_session_by_app(usess, app);
3741 if (ua_sess == NULL) {
d42f20df
DG
3742 /* The session is in teardown process. Ignore and continue. */
3743 goto end;
421cb601 3744 }
48842b30 3745
d0b96690
DG
3746 pthread_mutex_lock(&ua_sess->lock);
3747
aea829b3
DG
3748 /* Upon restart, we skip the setup, already done */
3749 if (ua_sess->started) {
8be98f9a 3750 goto skip_setup;
aea829b3 3751 }
8be98f9a 3752
a4b92340
DG
3753 /* Create directories if consumer is LOCAL and has a path defined. */
3754 if (usess->consumer->type == CONSUMER_DST_LOCAL &&
3755 strlen(usess->consumer->dst.trace_path) > 0) {
3756 ret = run_as_mkdir_recursive(usess->consumer->dst.trace_path,
7972aab2 3757 S_IRWXU | S_IRWXG, ua_sess->euid, ua_sess->egid);
a4b92340
DG
3758 if (ret < 0) {
3759 if (ret != -EEXIST) {
3760 ERR("Trace directory creation error");
d0b96690 3761 goto error_unlock;
421cb601 3762 }
173af62f 3763 }
7753dea8 3764 }
aea829b3 3765
d65d2de8
DG
3766 /*
3767 * Create the metadata for the application. This returns gracefully if a
3768 * metadata was already set for the session.
3769 */
3770 ret = create_ust_app_metadata(ua_sess, app, usess->consumer, NULL);
421cb601 3771 if (ret < 0) {
d0b96690 3772 goto error_unlock;
421cb601 3773 }
48842b30 3774
840cb59c 3775 health_code_update();
86acf0da 3776
8be98f9a 3777skip_setup:
421cb601 3778 /* This start the UST tracing */
852d0037 3779 ret = ustctl_start_session(app->sock, ua_sess->handle);
421cb601 3780 if (ret < 0) {
ffe60014
DG
3781 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
3782 ERR("Error starting tracing for app pid: %d (ret: %d)",
3783 app->pid, ret);
3784 } else {
3785 DBG("UST app start session failed. Application is dead.");
47cb9d4a
DG
3786 /*
3787 * This is normal behavior, an application can die during the
3788 * creation process. Don't report an error so the execution can
3789 * continue normally.
3790 */
3791 pthread_mutex_unlock(&ua_sess->lock);
3792 goto end;
ffe60014 3793 }
d0b96690 3794 goto error_unlock;
421cb601 3795 }
5b4a0ec0 3796
55c3953d
DG
3797 /* Indicate that the session has been started once */
3798 ua_sess->started = 1;
3799
d0b96690
DG
3800 pthread_mutex_unlock(&ua_sess->lock);
3801
840cb59c 3802 health_code_update();
86acf0da 3803
421cb601 3804 /* Quiescent wait after starting trace */
ffe60014
DG
3805 ret = ustctl_wait_quiescent(app->sock);
3806 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
3807 ERR("UST app wait quiescent failed for app pid %d ret %d",
3808 app->pid, ret);
3809 }
48842b30 3810
e0c7ec2b
DG
3811end:
3812 rcu_read_unlock();
840cb59c 3813 health_code_update();
421cb601 3814 return 0;
48842b30 3815
d0b96690
DG
3816error_unlock:
3817 pthread_mutex_unlock(&ua_sess->lock);
509cbaf8 3818 rcu_read_unlock();
840cb59c 3819 health_code_update();
421cb601
DG
3820 return -1;
3821}
48842b30 3822
8be98f9a
MD
3823/*
3824 * Stop tracing for a specific UST session and app.
3825 */
b34cbebf 3826static
8be98f9a
MD
3827int ust_app_stop_trace(struct ltt_ust_session *usess, struct ust_app *app)
3828{
3829 int ret = 0;
3830 struct ust_app_session *ua_sess;
7972aab2 3831 struct ust_registry_session *registry;
8be98f9a 3832
852d0037 3833 DBG("Stopping tracing for ust app pid %d", app->pid);
8be98f9a
MD
3834
3835 rcu_read_lock();
3836
e0c7ec2b 3837 if (!app->compatible) {
d88aee68 3838 goto end_no_session;
e0c7ec2b
DG
3839 }
3840
8be98f9a
MD
3841 ua_sess = lookup_session_by_app(usess, app);
3842 if (ua_sess == NULL) {
d88aee68 3843 goto end_no_session;
8be98f9a
MD
3844 }
3845
d88aee68
DG
3846 pthread_mutex_lock(&ua_sess->lock);
3847
9bc07046
DG
3848 /*
3849 * If started = 0, it means that stop trace has been called for a session
c45536e1
DG
3850 * that was never started. It's possible since we can have a fail start
3851 * from either the application manager thread or the command thread. Simply
3852 * indicate that this is a stop error.
9bc07046 3853 */
f9dfc3d9 3854 if (!ua_sess->started) {
c45536e1
DG
3855 goto error_rcu_unlock;
3856 }
7db205b5 3857
840cb59c 3858 health_code_update();
86acf0da 3859
9d6c7d3f 3860 /* This inhibits UST tracing */
852d0037 3861 ret = ustctl_stop_session(app->sock, ua_sess->handle);
9d6c7d3f 3862 if (ret < 0) {
ffe60014
DG
3863 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
3864 ERR("Error stopping tracing for app pid: %d (ret: %d)",
3865 app->pid, ret);
3866 } else {
3867 DBG("UST app stop session failed. Application is dead.");
47cb9d4a
DG
3868 /*
3869 * This is normal behavior, an application can die during the
3870 * creation process. Don't report an error so the execution can
3871 * continue normally.
3872 */
3873 goto end_unlock;
ffe60014 3874 }
9d6c7d3f
DG
3875 goto error_rcu_unlock;
3876 }
3877
840cb59c 3878 health_code_update();
86acf0da 3879
9d6c7d3f 3880 /* Quiescent wait after stopping trace */
ffe60014
DG
3881 ret = ustctl_wait_quiescent(app->sock);
3882 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
3883 ERR("UST app wait quiescent failed for app pid %d ret %d",
3884 app->pid, ret);
3885 }
9d6c7d3f 3886
840cb59c 3887 health_code_update();
86acf0da 3888
b34cbebf
MD
3889 registry = get_session_registry(ua_sess);
3890 assert(registry);
1b532a60
DG
3891
3892 if (!registry->metadata_closed) {
3893 /* Push metadata for application before freeing the application. */
3894 (void) push_metadata(registry, ua_sess->consumer);
3895 }
b34cbebf 3896
47cb9d4a 3897end_unlock:
b34cbebf
MD
3898 pthread_mutex_unlock(&ua_sess->lock);
3899end_no_session:
3900 rcu_read_unlock();
3901 health_code_update();
3902 return 0;
3903
3904error_rcu_unlock:
3905 pthread_mutex_unlock(&ua_sess->lock);
3906 rcu_read_unlock();
3907 health_code_update();
3908 return -1;
3909}
3910
3911/*
3912 * Flush buffers for a specific UST session and app.
3913 */
3914static
3915int ust_app_flush_trace(struct ltt_ust_session *usess, struct ust_app *app)
3916{
3917 int ret = 0;
3918 struct lttng_ht_iter iter;
3919 struct ust_app_session *ua_sess;
3920 struct ust_app_channel *ua_chan;
3921
3922 DBG("Flushing buffers for ust app pid %d", app->pid);
3923
3924 rcu_read_lock();
3925
3926 if (!app->compatible) {
3927 goto end_no_session;
3928 }
3929
3930 ua_sess = lookup_session_by_app(usess, app);
3931 if (ua_sess == NULL) {
3932 goto end_no_session;
3933 }
3934
3935 pthread_mutex_lock(&ua_sess->lock);
3936
3937 health_code_update();
3938
9d6c7d3f 3939 /* Flushing buffers */
bec39940
DG
3940 cds_lfht_for_each_entry(ua_sess->channels->ht, &iter.iter, ua_chan,
3941 node.node) {
840cb59c 3942 health_code_update();
ffe60014 3943 assert(ua_chan->is_sent);
852d0037 3944 ret = ustctl_sock_flush_buffer(app->sock, ua_chan->obj);
8be98f9a 3945 if (ret < 0) {
ffe60014
DG
3946 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
3947 ERR("UST app PID %d channel %s flush failed with ret %d",
3948 app->pid, ua_chan->name, ret);
3949 } else {
3950 DBG3("UST app failed to flush %s. Application is dead.",
3951 ua_chan->name);
47cb9d4a
DG
3952 /*
3953 * This is normal behavior, an application can die during the
3954 * creation process. Don't report an error so the execution can
3955 * continue normally.
3956 */
ffe60014 3957 }
6d3686da
DG
3958 /* Continuing flushing all buffers */
3959 continue;
8be98f9a
MD
3960 }
3961 }
8be98f9a 3962
840cb59c 3963 health_code_update();
86acf0da 3964
d88aee68
DG
3965 pthread_mutex_unlock(&ua_sess->lock);
3966end_no_session:
7db205b5 3967 rcu_read_unlock();
840cb59c 3968 health_code_update();
8be98f9a 3969 return 0;
8be98f9a
MD
3970}
3971
84cd17c6
MD
3972/*
3973 * Destroy a specific UST session in apps.
3974 */
3353de95 3975static int destroy_trace(struct ltt_ust_session *usess, struct ust_app *app)
84cd17c6 3976{
ffe60014 3977 int ret;
84cd17c6 3978 struct ust_app_session *ua_sess;
bec39940 3979 struct lttng_ht_iter iter;
77c5447b 3980 struct lttng_ht_node_u64 *node;
84cd17c6 3981
852d0037 3982 DBG("Destroy tracing for ust app pid %d", app->pid);
84cd17c6
MD
3983
3984 rcu_read_lock();
3985
e0c7ec2b
DG
3986 if (!app->compatible) {
3987 goto end;
3988 }
3989
84cd17c6 3990 __lookup_session_by_app(usess, app, &iter);
77c5447b 3991 node = lttng_ht_iter_get_node_u64(&iter);
84cd17c6 3992 if (node == NULL) {
d42f20df
DG
3993 /* Session is being or is deleted. */
3994 goto end;
84cd17c6
MD
3995 }
3996 ua_sess = caa_container_of(node, struct ust_app_session, node);
c4a1715b 3997
840cb59c 3998 health_code_update();
d0b96690 3999 destroy_app_session(app, ua_sess);
84cd17c6 4000
840cb59c 4001 health_code_update();
7db205b5 4002
84cd17c6 4003 /* Quiescent wait after stopping trace */
ffe60014
DG
4004 ret = ustctl_wait_quiescent(app->sock);
4005 if (ret < 0 && ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4006 ERR("UST app wait quiescent failed for app pid %d ret %d",
4007 app->pid, ret);
4008 }
e0c7ec2b
DG
4009end:
4010 rcu_read_unlock();
840cb59c 4011 health_code_update();
84cd17c6 4012 return 0;
84cd17c6
MD
4013}
4014
5b4a0ec0
DG
4015/*
4016 * Start tracing for the UST session.
4017 */
421cb601
DG
4018int ust_app_start_trace_all(struct ltt_ust_session *usess)
4019{
4020 int ret = 0;
bec39940 4021 struct lttng_ht_iter iter;
421cb601 4022 struct ust_app *app;
48842b30 4023
421cb601
DG
4024 DBG("Starting all UST traces");
4025
4026 rcu_read_lock();
421cb601 4027
852d0037 4028 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
421cb601 4029 ret = ust_app_start_trace(usess, app);
48842b30 4030 if (ret < 0) {
5b4a0ec0
DG
4031 /* Continue to next apps even on error */
4032 continue;
48842b30 4033 }
48842b30 4034 }
5b4a0ec0 4035
48842b30
DG
4036 rcu_read_unlock();
4037
4038 return 0;
4039}
487cf67c 4040
8be98f9a
MD
4041/*
4042 * Start tracing for the UST session.
4043 */
4044int ust_app_stop_trace_all(struct ltt_ust_session *usess)
4045{
4046 int ret = 0;
bec39940 4047 struct lttng_ht_iter iter;
8be98f9a
MD
4048 struct ust_app *app;
4049
4050 DBG("Stopping all UST traces");
4051
4052 rcu_read_lock();
4053
b34cbebf
MD
4054 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
4055 ret = ust_app_stop_trace(usess, app);
4056 if (ret < 0) {
4057 /* Continue to next apps even on error */
4058 continue;
4059 }
4060 }
4061
5e674ece 4062 /* Flush buffers and push metadata (for UID buffers). */
b34cbebf
MD
4063 switch (usess->buffer_type) {
4064 case LTTNG_BUFFER_PER_UID:
4065 {
7972aab2 4066 struct buffer_reg_uid *reg;
b34cbebf
MD
4067
4068 /* Flush all per UID buffers associated to that session. */
7972aab2 4069 cds_list_for_each_entry(reg, &usess->buffer_reg_uid_list, lnode) {
5e674ece 4070 struct ust_registry_session *ust_session_reg;
7972aab2
DG
4071 struct buffer_reg_channel *reg_chan;
4072 struct consumer_socket *socket;
4073
4074 /* Get consumer socket to use to push the metadata.*/
4075 socket = consumer_find_socket_by_bitness(reg->bits_per_long,
4076 usess->consumer);
4077 if (!socket) {
4078 /* Ignore request if no consumer is found for the session. */
4079 continue;
4080 }
4081
4082 cds_lfht_for_each_entry(reg->registry->channels->ht, &iter.iter,
4083 reg_chan, node.node) {
4084 /*
4085 * The following call will print error values so the return
4086 * code is of little importance because whatever happens, we
4087 * have to try them all.
4088 */
4089 (void) consumer_flush_channel(socket, reg_chan->consumer_key);
4090 }
5e674ece
MD
4091
4092 ust_session_reg = reg->registry->reg.ust;
4093 if (!ust_session_reg->metadata_closed) {
4094 /* Push metadata. */
4095 (void) push_metadata(ust_session_reg, usess->consumer);
4096 }
7972aab2 4097 }
5e674ece 4098
b34cbebf 4099 break;
7972aab2 4100 }
b34cbebf
MD
4101 case LTTNG_BUFFER_PER_PID:
4102 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
4103 ret = ust_app_flush_trace(usess, app);
4104 if (ret < 0) {
4105 /* Continue to next apps even on error */
4106 continue;
4107 }
8be98f9a 4108 }
b34cbebf
MD
4109 break;
4110 default:
4111 assert(0);
4112 break;
8be98f9a
MD
4113 }
4114
4115 rcu_read_unlock();
4116
4117 return 0;
4118}
4119
84cd17c6
MD
4120/*
4121 * Destroy app UST session.
4122 */
4123int ust_app_destroy_trace_all(struct ltt_ust_session *usess)
4124{
4125 int ret = 0;
bec39940 4126 struct lttng_ht_iter iter;
84cd17c6
MD
4127 struct ust_app *app;
4128
4129 DBG("Destroy all UST traces");
4130
4131 rcu_read_lock();
4132
852d0037 4133 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
3353de95 4134 ret = destroy_trace(usess, app);
84cd17c6
MD
4135 if (ret < 0) {
4136 /* Continue to next apps even on error */
4137 continue;
4138 }
4139 }
4140
4141 rcu_read_unlock();
4142
4143 return 0;
4144}
4145
5b4a0ec0
DG
4146/*
4147 * Add channels/events from UST global domain to registered apps at sock.
4148 */
487cf67c
DG
4149void ust_app_global_update(struct ltt_ust_session *usess, int sock)
4150{
55c54cce 4151 int ret = 0;
4623416f 4152 struct lttng_ht_iter iter, uiter;
487cf67c 4153 struct ust_app *app;
3d8ca23b 4154 struct ust_app_session *ua_sess = NULL;
487cf67c
DG
4155 struct ust_app_channel *ua_chan;
4156 struct ust_app_event *ua_event;
727d5404 4157 struct ust_app_ctx *ua_ctx;
1f3580c7 4158
95e047ff 4159 assert(usess);
ffe60014 4160 assert(sock >= 0);
1f3580c7 4161
77c5447b 4162 DBG2("UST app global update for app sock %d for session id %" PRIu64, sock,
a991f516 4163 usess->id);
487cf67c 4164
284d8f55
DG
4165 rcu_read_lock();
4166
487cf67c
DG
4167 app = find_app_by_sock(sock);
4168 if (app == NULL) {
d88aee68
DG
4169 /*
4170 * Application can be unregistered before so this is possible hence
4171 * simply stopping the update.
4172 */
4173 DBG3("UST app update failed to find app sock %d", sock);
487cf67c
DG
4174 goto error;
4175 }
4176
e0c7ec2b
DG
4177 if (!app->compatible) {
4178 goto error;
4179 }
4180
3d8ca23b
DG
4181 ret = create_ust_app_session(usess, app, &ua_sess, NULL);
4182 if (ret < 0) {
4183 /* Tracer is probably gone or ENOMEM. */
487cf67c
DG
4184 goto error;
4185 }
3d8ca23b 4186 assert(ua_sess);
487cf67c 4187
d0b96690
DG
4188 pthread_mutex_lock(&ua_sess->lock);
4189
284d8f55 4190 /*
d0b96690 4191 * We can iterate safely here over all UST app session since the create ust
284d8f55
DG
4192 * app session above made a shadow copy of the UST global domain from the
4193 * ltt ust session.
4194 */
bec39940
DG
4195 cds_lfht_for_each_entry(ua_sess->channels->ht, &iter.iter, ua_chan,
4196 node.node) {
d65d2de8
DG
4197 /*
4198 * For a metadata channel, handle it differently.
4199 */
4200 if (!strncmp(ua_chan->name, DEFAULT_METADATA_NAME,
4201 sizeof(ua_chan->name))) {
4202 ret = create_ust_app_metadata(ua_sess, app, usess->consumer,
4203 &ua_chan->attr);
c7a339aa
DG
4204 if (ret < 0) {
4205 goto error_unlock;
4206 }
d65d2de8
DG
4207 /* Remove it from the hash table and continue!. */
4208 ret = lttng_ht_del(ua_sess->channels, &iter);
4209 assert(!ret);
4210 delete_ust_app_channel(-1, ua_chan, app);
4211 continue;
4212 } else {
4213 ret = do_create_channel(app, usess, ua_sess, ua_chan);
c7a339aa
DG
4214 if (ret < 0) {
4215 /*
4216 * Stop everything. On error, the application failed, no more
4217 * file descriptor are available or ENOMEM so stopping here is
4218 * the only thing we can do for now.
4219 */
4220 goto error_unlock;
4221 }
487cf67c
DG
4222 }
4223
4623416f
DG
4224 /*
4225 * Add context using the list so they are enabled in the same order the
4226 * user added them.
4227 */
4228 cds_list_for_each_entry(ua_ctx, &ua_chan->ctx_list, list) {
727d5404
DG
4229 ret = create_ust_channel_context(ua_chan, ua_ctx, app);
4230 if (ret < 0) {
d0b96690 4231 goto error_unlock;
727d5404
DG
4232 }
4233 }
4234
4235
284d8f55 4236 /* For each events */
bec39940
DG
4237 cds_lfht_for_each_entry(ua_chan->events->ht, &uiter.iter, ua_event,
4238 node.node) {
284d8f55
DG
4239 ret = create_ust_event(app, ua_sess, ua_chan, ua_event);
4240 if (ret < 0) {
d0b96690 4241 goto error_unlock;
487cf67c 4242 }
36dc12cc 4243 }
487cf67c
DG
4244 }
4245
d0b96690
DG
4246 pthread_mutex_unlock(&ua_sess->lock);
4247
36dc12cc 4248 if (usess->start_trace) {
421cb601 4249 ret = ust_app_start_trace(usess, app);
36dc12cc 4250 if (ret < 0) {
36dc12cc
DG
4251 goto error;
4252 }
4253
852d0037 4254 DBG2("UST trace started for app pid %d", app->pid);
36dc12cc
DG
4255 }
4256
ffe60014
DG
4257 /* Everything went well at this point. */
4258 rcu_read_unlock();
4259 return;
4260
d0b96690
DG
4261error_unlock:
4262 pthread_mutex_unlock(&ua_sess->lock);
487cf67c 4263error:
ffe60014 4264 if (ua_sess) {
d0b96690 4265 destroy_app_session(app, ua_sess);
ffe60014 4266 }
487cf67c
DG
4267 rcu_read_unlock();
4268 return;
4269}
55cc08a6
DG
4270
4271/*
4272 * Add context to a specific channel for global UST domain.
4273 */
4274int ust_app_add_ctx_channel_glb(struct ltt_ust_session *usess,
4275 struct ltt_ust_channel *uchan, struct ltt_ust_context *uctx)
4276{
4277 int ret = 0;
bec39940
DG
4278 struct lttng_ht_node_str *ua_chan_node;
4279 struct lttng_ht_iter iter, uiter;
55cc08a6
DG
4280 struct ust_app_channel *ua_chan = NULL;
4281 struct ust_app_session *ua_sess;
4282 struct ust_app *app;
4283
4284 rcu_read_lock();
4285
852d0037 4286 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
e0c7ec2b
DG
4287 if (!app->compatible) {
4288 /*
4289 * TODO: In time, we should notice the caller of this error by
4290 * telling him that this is a version error.
4291 */
4292 continue;
4293 }
55cc08a6
DG
4294 ua_sess = lookup_session_by_app(usess, app);
4295 if (ua_sess == NULL) {
4296 continue;
4297 }
4298
d0b96690 4299 pthread_mutex_lock(&ua_sess->lock);
55cc08a6 4300 /* Lookup channel in the ust app session */
bec39940
DG
4301 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &uiter);
4302 ua_chan_node = lttng_ht_iter_get_node_str(&uiter);
55cc08a6 4303 if (ua_chan_node == NULL) {
d0b96690 4304 goto next_app;
55cc08a6
DG
4305 }
4306 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel,
4307 node);
55cc08a6
DG
4308 ret = create_ust_app_channel_context(ua_sess, ua_chan, &uctx->ctx, app);
4309 if (ret < 0) {
d0b96690 4310 goto next_app;
55cc08a6 4311 }
d0b96690
DG
4312 next_app:
4313 pthread_mutex_unlock(&ua_sess->lock);
55cc08a6
DG
4314 }
4315
55cc08a6
DG
4316 rcu_read_unlock();
4317 return ret;
4318}
4319
76d45b40
DG
4320/*
4321 * Enable event for a channel from a UST session for a specific PID.
4322 */
4323int ust_app_enable_event_pid(struct ltt_ust_session *usess,
4324 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent, pid_t pid)
4325{
4326 int ret = 0;
bec39940 4327 struct lttng_ht_iter iter;
18eace3b 4328 struct lttng_ht_node_str *ua_chan_node;
76d45b40
DG
4329 struct ust_app *app;
4330 struct ust_app_session *ua_sess;
4331 struct ust_app_channel *ua_chan;
4332 struct ust_app_event *ua_event;
4333
4334 DBG("UST app enabling event %s for PID %d", uevent->attr.name, pid);
4335
4336 rcu_read_lock();
4337
4338 app = ust_app_find_by_pid(pid);
4339 if (app == NULL) {
4340 ERR("UST app enable event per PID %d not found", pid);
4341 ret = -1;
d0b96690 4342 goto end;
76d45b40
DG
4343 }
4344
e0c7ec2b
DG
4345 if (!app->compatible) {
4346 ret = 0;
d0b96690 4347 goto end;
e0c7ec2b
DG
4348 }
4349
76d45b40 4350 ua_sess = lookup_session_by_app(usess, app);
c4a1715b
DG
4351 if (!ua_sess) {
4352 /* The application has problem or is probably dead. */
d0b96690
DG
4353 ret = 0;
4354 goto end;
c4a1715b 4355 }
76d45b40 4356
d0b96690 4357 pthread_mutex_lock(&ua_sess->lock);
76d45b40 4358 /* Lookup channel in the ust app session */
bec39940
DG
4359 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &iter);
4360 ua_chan_node = lttng_ht_iter_get_node_str(&iter);
76d45b40
DG
4361 /* If the channel is not found, there is a code flow error */
4362 assert(ua_chan_node);
4363
4364 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
4365
18eace3b
DG
4366 ua_event = find_ust_app_event(ua_chan->events, uevent->attr.name,
4367 uevent->filter, uevent->attr.loglevel);
4368 if (ua_event == NULL) {
76d45b40
DG
4369 ret = create_ust_app_event(ua_sess, ua_chan, uevent, app);
4370 if (ret < 0) {
d0b96690 4371 goto end_unlock;
76d45b40
DG
4372 }
4373 } else {
76d45b40
DG
4374 ret = enable_ust_app_event(ua_sess, ua_event, app);
4375 if (ret < 0) {
d0b96690 4376 goto end_unlock;
76d45b40
DG
4377 }
4378 }
4379
d0b96690
DG
4380end_unlock:
4381 pthread_mutex_unlock(&ua_sess->lock);
4382end:
76d45b40
DG
4383 rcu_read_unlock();
4384 return ret;
4385}
7f79d3a1
DG
4386
4387/*
4388 * Disable event for a channel from a UST session for a specific PID.
4389 */
4390int ust_app_disable_event_pid(struct ltt_ust_session *usess,
4391 struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent, pid_t pid)
4392{
4393 int ret = 0;
bec39940
DG
4394 struct lttng_ht_iter iter;
4395 struct lttng_ht_node_str *ua_chan_node, *ua_event_node;
7f79d3a1
DG
4396 struct ust_app *app;
4397 struct ust_app_session *ua_sess;
4398 struct ust_app_channel *ua_chan;
4399 struct ust_app_event *ua_event;
4400
4401 DBG("UST app disabling event %s for PID %d", uevent->attr.name, pid);
4402
4403 rcu_read_lock();
4404
4405 app = ust_app_find_by_pid(pid);
4406 if (app == NULL) {
4407 ERR("UST app disable event per PID %d not found", pid);
4408 ret = -1;
4409 goto error;
4410 }
4411
e0c7ec2b
DG
4412 if (!app->compatible) {
4413 ret = 0;
4414 goto error;
4415 }
4416
7f79d3a1 4417 ua_sess = lookup_session_by_app(usess, app);
c4a1715b
DG
4418 if (!ua_sess) {
4419 /* The application has problem or is probably dead. */
4420 goto error;
4421 }
7f79d3a1
DG
4422
4423 /* Lookup channel in the ust app session */
bec39940
DG
4424 lttng_ht_lookup(ua_sess->channels, (void *)uchan->name, &iter);
4425 ua_chan_node = lttng_ht_iter_get_node_str(&iter);
7f79d3a1
DG
4426 if (ua_chan_node == NULL) {
4427 /* Channel does not exist, skip disabling */
4428 goto error;
4429 }
4430 ua_chan = caa_container_of(ua_chan_node, struct ust_app_channel, node);
4431
bec39940
DG
4432 lttng_ht_lookup(ua_chan->events, (void *)uevent->attr.name, &iter);
4433 ua_event_node = lttng_ht_iter_get_node_str(&iter);
7f79d3a1
DG
4434 if (ua_event_node == NULL) {
4435 /* Event does not exist, skip disabling */
4436 goto error;
4437 }
4438 ua_event = caa_container_of(ua_event_node, struct ust_app_event, node);
4439
4440 ret = disable_ust_app_event(ua_sess, ua_event, app);
4441 if (ret < 0) {
4442 goto error;
4443 }
4444
4445error:
4446 rcu_read_unlock();
4447 return ret;
4448}
e0c7ec2b 4449
4466912f
DG
4450/*
4451 * Calibrate registered applications.
4452 */
4453int ust_app_calibrate_glb(struct lttng_ust_calibrate *calibrate)
4454{
4455 int ret = 0;
4456 struct lttng_ht_iter iter;
4457 struct ust_app *app;
4458
4459 rcu_read_lock();
4460
852d0037 4461 cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
4466912f
DG
4462 if (!app->compatible) {
4463 /*
4464 * TODO: In time, we should notice the caller of this error by
4465 * telling him that this is a version error.
4466 */
4467 continue;
4468 }
4469
840cb59c 4470 health_code_update();
86acf0da 4471
852d0037 4472 ret = ustctl_calibrate(app->sock, calibrate);
4466912f
DG
4473 if (ret < 0) {
4474 switch (ret) {
4475 case -ENOSYS:
4476 /* Means that it's not implemented on the tracer side. */
4477 ret = 0;
4478 break;
4479 default:
4466912f 4480 DBG2("Calibrate app PID %d returned with error %d",
852d0037 4481 app->pid, ret);
4466912f
DG
4482 break;
4483 }
4484 }
4485 }
4486
4487 DBG("UST app global domain calibration finished");
4488
4489 rcu_read_unlock();
4490
840cb59c 4491 health_code_update();
86acf0da 4492
4466912f
DG
4493 return ret;
4494}
d0b96690
DG
4495
4496/*
4497 * Receive registration and populate the given msg structure.
4498 *
4499 * On success return 0 else a negative value returned by the ustctl call.
4500 */
4501int ust_app_recv_registration(int sock, struct ust_register_msg *msg)
4502{
4503 int ret;
4504 uint32_t pid, ppid, uid, gid;
4505
4506 assert(msg);
4507
4508 ret = ustctl_recv_reg_msg(sock, &msg->type, &msg->major, &msg->minor,
4509 &pid, &ppid, &uid, &gid,
4510 &msg->bits_per_long,
4511 &msg->uint8_t_alignment,
4512 &msg->uint16_t_alignment,
4513 &msg->uint32_t_alignment,
4514 &msg->uint64_t_alignment,
4515 &msg->long_alignment,
4516 &msg->byte_order,
4517 msg->name);
4518 if (ret < 0) {
4519 switch (-ret) {
4520 case EPIPE:
4521 case ECONNRESET:
4522 case LTTNG_UST_ERR_EXITING:
4523 DBG3("UST app recv reg message failed. Application died");
4524 break;
4525 case LTTNG_UST_ERR_UNSUP_MAJOR:
4526 ERR("UST app recv reg unsupported version %d.%d. Supporting %d.%d",
4527 msg->major, msg->minor, LTTNG_UST_ABI_MAJOR_VERSION,
4528 LTTNG_UST_ABI_MINOR_VERSION);
4529 break;
4530 default:
4531 ERR("UST app recv reg message failed with ret %d", ret);
4532 break;
4533 }
4534 goto error;
4535 }
4536 msg->pid = (pid_t) pid;
4537 msg->ppid = (pid_t) ppid;
4538 msg->uid = (uid_t) uid;
4539 msg->gid = (gid_t) gid;
4540
4541error:
4542 return ret;
4543}
4544
d88aee68
DG
4545/*
4546 * Return a ust app channel object using the application object and the channel
4547 * object descriptor has a key. If not found, NULL is returned. A RCU read side
4548 * lock MUST be acquired before calling this function.
4549 */
d0b96690
DG
4550static struct ust_app_channel *find_channel_by_objd(struct ust_app *app,
4551 int objd)
4552{
4553 struct lttng_ht_node_ulong *node;
4554 struct lttng_ht_iter iter;
4555 struct ust_app_channel *ua_chan = NULL;
4556
4557 assert(app);
4558
4559 lttng_ht_lookup(app->ust_objd, (void *)((unsigned long) objd), &iter);
4560 node = lttng_ht_iter_get_node_ulong(&iter);
4561 if (node == NULL) {
4562 DBG2("UST app channel find by objd %d not found", objd);
4563 goto error;
4564 }
4565
4566 ua_chan = caa_container_of(node, struct ust_app_channel, ust_objd_node);
4567
4568error:
4569 return ua_chan;
4570}
4571
d88aee68
DG
4572/*
4573 * Reply to a register channel notification from an application on the notify
4574 * socket. The channel metadata is also created.
4575 *
4576 * The session UST registry lock is acquired in this function.
4577 *
4578 * On success 0 is returned else a negative value.
4579 */
d0b96690
DG
4580static int reply_ust_register_channel(int sock, int sobjd, int cobjd,
4581 size_t nr_fields, struct ustctl_field *fields)
4582{
4583 int ret, ret_code = 0;
4584 uint32_t chan_id, reg_count;
7972aab2 4585 uint64_t chan_reg_key;
d0b96690
DG
4586 enum ustctl_channel_header type;
4587 struct ust_app *app;
4588 struct ust_app_channel *ua_chan;
4589 struct ust_app_session *ua_sess;
7972aab2 4590 struct ust_registry_session *registry;
45893984 4591 struct ust_registry_channel *chan_reg;
d0b96690
DG
4592
4593 rcu_read_lock();
4594
4595 /* Lookup application. If not found, there is a code flow error. */
4596 app = find_app_by_notify_sock(sock);
d88aee68
DG
4597 if (!app) {
4598 DBG("Application socket %d is being teardown. Abort event notify",
4599 sock);
4600 ret = 0;
d5d629b5 4601 free(fields);
d88aee68
DG
4602 goto error_rcu_unlock;
4603 }
d0b96690 4604
4950b860 4605 /* Lookup channel by UST object descriptor. */
d0b96690 4606 ua_chan = find_channel_by_objd(app, cobjd);
4950b860
MD
4607 if (!ua_chan) {
4608 DBG("Application channel is being teardown. Abort event notify");
4609 ret = 0;
d5d629b5 4610 free(fields);
4950b860
MD
4611 goto error_rcu_unlock;
4612 }
4613
d0b96690
DG
4614 assert(ua_chan->session);
4615 ua_sess = ua_chan->session;
d0b96690 4616
7972aab2
DG
4617 /* Get right session registry depending on the session buffer type. */
4618 registry = get_session_registry(ua_sess);
4619 assert(registry);
45893984 4620
7972aab2
DG
4621 /* Depending on the buffer type, a different channel key is used. */
4622 if (ua_sess->buffer_type == LTTNG_BUFFER_PER_UID) {
4623 chan_reg_key = ua_chan->tracing_channel_id;
d0b96690 4624 } else {
7972aab2 4625 chan_reg_key = ua_chan->key;
d0b96690
DG
4626 }
4627
7972aab2
DG
4628 pthread_mutex_lock(&registry->lock);
4629
4630 chan_reg = ust_registry_channel_find(registry, chan_reg_key);
4631 assert(chan_reg);
4632
4633 if (!chan_reg->register_done) {
4634 reg_count = ust_registry_get_event_count(chan_reg);
4635 if (reg_count < 31) {
4636 type = USTCTL_CHANNEL_HEADER_COMPACT;
4637 } else {
4638 type = USTCTL_CHANNEL_HEADER_LARGE;
4639 }
4640
4641 chan_reg->nr_ctx_fields = nr_fields;
4642 chan_reg->ctx_fields = fields;
4643 chan_reg->header_type = type;
d0b96690 4644 } else {
7972aab2
DG
4645 /* Get current already assigned values. */
4646 type = chan_reg->header_type;
d5d629b5
DG
4647 free(fields);
4648 /* Set to NULL so the error path does not do a double free. */
4649 fields = NULL;
d0b96690 4650 }
7972aab2
DG
4651 /* Channel id is set during the object creation. */
4652 chan_id = chan_reg->chan_id;
d0b96690
DG
4653
4654 /* Append to metadata */
7972aab2
DG
4655 if (!chan_reg->metadata_dumped) {
4656 ret_code = ust_metadata_channel_statedump(registry, chan_reg);
d0b96690
DG
4657 if (ret_code) {
4658 ERR("Error appending channel metadata (errno = %d)", ret_code);
4659 goto reply;
4660 }
4661 }
4662
4663reply:
7972aab2
DG
4664 DBG3("UST app replying to register channel key %" PRIu64
4665 " with id %u, type: %d, ret: %d", chan_reg_key, chan_id, type,
4666 ret_code);
d0b96690
DG
4667
4668 ret = ustctl_reply_register_channel(sock, chan_id, type, ret_code);
4669 if (ret < 0) {
4670 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4671 ERR("UST app reply channel failed with ret %d", ret);
4672 } else {
4673 DBG3("UST app reply channel failed. Application died");
4674 }
4675 goto error;
4676 }
4677
7972aab2
DG
4678 /* This channel registry registration is completed. */
4679 chan_reg->register_done = 1;
4680
d0b96690 4681error:
7972aab2 4682 pthread_mutex_unlock(&registry->lock);
d88aee68 4683error_rcu_unlock:
d0b96690 4684 rcu_read_unlock();
d5d629b5
DG
4685 if (ret) {
4686 free(fields);
4687 }
d0b96690
DG
4688 return ret;
4689}
4690
d88aee68
DG
4691/*
4692 * Add event to the UST channel registry. When the event is added to the
4693 * registry, the metadata is also created. Once done, this replies to the
4694 * application with the appropriate error code.
4695 *
4696 * The session UST registry lock is acquired in the function.
4697 *
4698 * On success 0 is returned else a negative value.
4699 */
d0b96690
DG
4700static int add_event_ust_registry(int sock, int sobjd, int cobjd, char *name,
4701 char *sig, size_t nr_fields, struct ustctl_field *fields, int loglevel,
4702 char *model_emf_uri)
4703{
4704 int ret, ret_code;
4705 uint32_t event_id = 0;
7972aab2 4706 uint64_t chan_reg_key;
d0b96690
DG
4707 struct ust_app *app;
4708 struct ust_app_channel *ua_chan;
4709 struct ust_app_session *ua_sess;
7972aab2 4710 struct ust_registry_session *registry;
d0b96690
DG
4711
4712 rcu_read_lock();
4713
4714 /* Lookup application. If not found, there is a code flow error. */
4715 app = find_app_by_notify_sock(sock);
d88aee68
DG
4716 if (!app) {
4717 DBG("Application socket %d is being teardown. Abort event notify",
4718 sock);
4719 ret = 0;
d5d629b5
DG
4720 free(sig);
4721 free(fields);
4722 free(model_emf_uri);
d88aee68
DG
4723 goto error_rcu_unlock;
4724 }
d0b96690 4725
4950b860 4726 /* Lookup channel by UST object descriptor. */
d0b96690 4727 ua_chan = find_channel_by_objd(app, cobjd);
4950b860
MD
4728 if (!ua_chan) {
4729 DBG("Application channel is being teardown. Abort event notify");
4730 ret = 0;
d5d629b5
DG
4731 free(sig);
4732 free(fields);
4733 free(model_emf_uri);
4950b860
MD
4734 goto error_rcu_unlock;
4735 }
4736
d0b96690
DG
4737 assert(ua_chan->session);
4738 ua_sess = ua_chan->session;
4739
7972aab2
DG
4740 registry = get_session_registry(ua_sess);
4741 assert(registry);
4742
4743 if (ua_sess->buffer_type == LTTNG_BUFFER_PER_UID) {
4744 chan_reg_key = ua_chan->tracing_channel_id;
4745 } else {
4746 chan_reg_key = ua_chan->key;
4747 }
4748
4749 pthread_mutex_lock(&registry->lock);
d0b96690 4750
d5d629b5
DG
4751 /*
4752 * From this point on, this call acquires the ownership of the sig, fields
4753 * and model_emf_uri meaning any free are done inside it if needed. These
4754 * three variables MUST NOT be read/write after this.
4755 */
7972aab2 4756 ret_code = ust_registry_create_event(registry, chan_reg_key,
45893984 4757 sobjd, cobjd, name, sig, nr_fields, fields, loglevel,
9664f59b
MD
4758 model_emf_uri, ua_sess->buffer_type, &event_id,
4759 app);
d0b96690
DG
4760
4761 /*
4762 * The return value is returned to ustctl so in case of an error, the
4763 * application can be notified. In case of an error, it's important not to
4764 * return a negative error or else the application will get closed.
4765 */
4766 ret = ustctl_reply_register_event(sock, event_id, ret_code);
4767 if (ret < 0) {
4768 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4769 ERR("UST app reply event failed with ret %d", ret);
4770 } else {
4771 DBG3("UST app reply event failed. Application died");
4772 }
4773 /*
4774 * No need to wipe the create event since the application socket will
4775 * get close on error hence cleaning up everything by itself.
4776 */
4777 goto error;
4778 }
4779
7972aab2
DG
4780 DBG3("UST registry event %s with id %" PRId32 " added successfully",
4781 name, event_id);
d88aee68 4782
d0b96690 4783error:
7972aab2 4784 pthread_mutex_unlock(&registry->lock);
d88aee68 4785error_rcu_unlock:
d0b96690
DG
4786 rcu_read_unlock();
4787 return ret;
4788}
4789
d88aee68
DG
4790/*
4791 * Handle application notification through the given notify socket.
4792 *
4793 * Return 0 on success or else a negative value.
4794 */
d0b96690
DG
4795int ust_app_recv_notify(int sock)
4796{
4797 int ret;
4798 enum ustctl_notify_cmd cmd;
4799
4800 DBG3("UST app receiving notify from sock %d", sock);
4801
4802 ret = ustctl_recv_notify(sock, &cmd);
4803 if (ret < 0) {
4804 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4805 ERR("UST app recv notify failed with ret %d", ret);
4806 } else {
4807 DBG3("UST app recv notify failed. Application died");
4808 }
4809 goto error;
4810 }
4811
4812 switch (cmd) {
4813 case USTCTL_NOTIFY_CMD_EVENT:
4814 {
4815 int sobjd, cobjd, loglevel;
4816 char name[LTTNG_UST_SYM_NAME_LEN], *sig, *model_emf_uri;
4817 size_t nr_fields;
4818 struct ustctl_field *fields;
4819
4820 DBG2("UST app ustctl register event received");
4821
4822 ret = ustctl_recv_register_event(sock, &sobjd, &cobjd, name, &loglevel,
4823 &sig, &nr_fields, &fields, &model_emf_uri);
4824 if (ret < 0) {
4825 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4826 ERR("UST app recv event failed with ret %d", ret);
4827 } else {
4828 DBG3("UST app recv event failed. Application died");
4829 }
4830 goto error;
4831 }
4832
d5d629b5
DG
4833 /*
4834 * Add event to the UST registry coming from the notify socket. This
4835 * call will free if needed the sig, fields and model_emf_uri. This
4836 * code path loses the ownsership of these variables and transfer them
4837 * to the this function.
4838 */
d0b96690
DG
4839 ret = add_event_ust_registry(sock, sobjd, cobjd, name, sig, nr_fields,
4840 fields, loglevel, model_emf_uri);
4841 if (ret < 0) {
4842 goto error;
4843 }
4844
4845 break;
4846 }
4847 case USTCTL_NOTIFY_CMD_CHANNEL:
4848 {
4849 int sobjd, cobjd;
4850 size_t nr_fields;
4851 struct ustctl_field *fields;
4852
4853 DBG2("UST app ustctl register channel received");
4854
4855 ret = ustctl_recv_register_channel(sock, &sobjd, &cobjd, &nr_fields,
4856 &fields);
4857 if (ret < 0) {
4858 if (ret != -EPIPE && ret != -LTTNG_UST_ERR_EXITING) {
4859 ERR("UST app recv channel failed with ret %d", ret);
4860 } else {
4861 DBG3("UST app recv channel failed. Application died");
4862 }
4863 goto error;
4864 }
4865
d5d629b5
DG
4866 /*
4867 * The fields ownership are transfered to this function call meaning
4868 * that if needed it will be freed. After this, it's invalid to access
4869 * fields or clean it up.
4870 */
d0b96690
DG
4871 ret = reply_ust_register_channel(sock, sobjd, cobjd, nr_fields,
4872 fields);
4873 if (ret < 0) {
4874 goto error;
4875 }
4876
4877 break;
4878 }
4879 default:
4880 /* Should NEVER happen. */
4881 assert(0);
4882 }
4883
4884error:
4885 return ret;
4886}
d88aee68
DG
4887
4888/*
4889 * Once the notify socket hangs up, this is called. First, it tries to find the
4890 * corresponding application. On failure, the call_rcu to close the socket is
4891 * executed. If an application is found, it tries to delete it from the notify
4892 * socket hash table. Whathever the result, it proceeds to the call_rcu.
4893 *
4894 * Note that an object needs to be allocated here so on ENOMEM failure, the
4895 * call RCU is not done but the rest of the cleanup is.
4896 */
4897void ust_app_notify_sock_unregister(int sock)
4898{
4899 int err_enomem = 0;
4900 struct lttng_ht_iter iter;
4901 struct ust_app *app;
4902 struct ust_app_notify_sock_obj *obj;
4903
4904 assert(sock >= 0);
4905
4906 rcu_read_lock();
4907
4908 obj = zmalloc(sizeof(*obj));
4909 if (!obj) {
4910 /*
4911 * An ENOMEM is kind of uncool. If this strikes we continue the
4912 * procedure but the call_rcu will not be called. In this case, we
4913 * accept the fd leak rather than possibly creating an unsynchronized
4914 * state between threads.
4915 *
4916 * TODO: The notify object should be created once the notify socket is
4917 * registered and stored independantely from the ust app object. The
4918 * tricky part is to synchronize the teardown of the application and
4919 * this notify object. Let's keep that in mind so we can avoid this
4920 * kind of shenanigans with ENOMEM in the teardown path.
4921 */
4922 err_enomem = 1;
4923 } else {
4924 obj->fd = sock;
4925 }
4926
4927 DBG("UST app notify socket unregister %d", sock);
4928
4929 /*
4930 * Lookup application by notify socket. If this fails, this means that the
4931 * hash table delete has already been done by the application
4932 * unregistration process so we can safely close the notify socket in a
4933 * call RCU.
4934 */
4935 app = find_app_by_notify_sock(sock);
4936 if (!app) {
4937 goto close_socket;
4938 }
4939
4940 iter.iter.node = &app->notify_sock_n.node;
4941
4942 /*
4943 * Whatever happens here either we fail or succeed, in both cases we have
4944 * to close the socket after a grace period to continue to the call RCU
4945 * here. If the deletion is successful, the application is not visible
4946 * anymore by other threads and is it fails it means that it was already
4947 * deleted from the hash table so either way we just have to close the
4948 * socket.
4949 */
4950 (void) lttng_ht_del(ust_app_ht_by_notify_sock, &iter);
4951
4952close_socket:
4953 rcu_read_unlock();
4954
4955 /*
4956 * Close socket after a grace period to avoid for the socket to be reused
4957 * before the application object is freed creating potential race between
4958 * threads trying to add unique in the global hash table.
4959 */
4960 if (!err_enomem) {
4961 call_rcu(&obj->head, close_notify_sock_rcu);
4962 }
4963}
f45e313d
DG
4964
4965/*
4966 * Destroy a ust app data structure and free its memory.
4967 */
4968void ust_app_destroy(struct ust_app *app)
4969{
4970 if (!app) {
4971 return;
4972 }
4973
4974 call_rcu(&app->pid_n.head, delete_ust_app_rcu);
4975}
This page took 0.305173 seconds and 4 git commands to generate.