common trace-chunk: introduce lttng_trace_chunk_get_name_overridden
[lttng-tools.git] / src / common / ust-consumer / ust-consumer.c
CommitLineData
3bd1e081
MD
1/*
2 * Copyright (C) 2011 - Julien Desfossez <julien.desfossez@polymtl.ca>
3 * Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
b3530820 4 * Copyright (C) 2017 - Jérémie Galarneau <jeremie.galarneau@efficios.com>
3bd1e081 5 *
d14d33bf
AM
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License, version 2 only,
8 * as published by the Free Software Foundation.
3bd1e081
MD
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
d14d33bf
AM
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
3bd1e081
MD
18 */
19
6c1c0768 20#define _LGPL_SOURCE
3bd1e081 21#include <assert.h>
f02e1e8a 22#include <lttng/ust-ctl.h>
3bd1e081
MD
23#include <poll.h>
24#include <pthread.h>
25#include <stdlib.h>
26#include <string.h>
27#include <sys/mman.h>
28#include <sys/socket.h>
dbb5dfe6 29#include <sys/stat.h>
3bd1e081 30#include <sys/types.h>
77c7c900 31#include <inttypes.h>
3bd1e081 32#include <unistd.h>
ffe60014 33#include <urcu/list.h>
331744e3 34#include <signal.h>
02d02e31 35#include <stdbool.h>
0857097f 36
51a9e1c7 37#include <bin/lttng-consumerd/health-consumerd.h>
990570ed 38#include <common/common.h>
10a8a223 39#include <common/sessiond-comm/sessiond-comm.h>
00e2e675 40#include <common/relayd/relayd.h>
dbb5dfe6 41#include <common/compat/fcntl.h>
f263b7fd 42#include <common/compat/endian.h>
c8fea79c
JR
43#include <common/consumer/consumer-metadata-cache.h>
44#include <common/consumer/consumer-stream.h>
45#include <common/consumer/consumer-timer.h>
fe4477ee 46#include <common/utils.h>
309167d2 47#include <common/index/index.h>
10a8a223
DG
48
49#include "ust-consumer.h"
3bd1e081 50
45863397 51#define INT_MAX_STR_LEN 12 /* includes \0 */
4628484a 52
3bd1e081
MD
53extern struct lttng_consumer_global_data consumer_data;
54extern int consumer_poll_timeout;
3bd1e081
MD
55
56/*
ffe60014
DG
57 * Free channel object and all streams associated with it. This MUST be used
58 * only and only if the channel has _NEVER_ been added to the global channel
59 * hash table.
3bd1e081 60 */
ffe60014 61static void destroy_channel(struct lttng_consumer_channel *channel)
3bd1e081 62{
ffe60014
DG
63 struct lttng_consumer_stream *stream, *stmp;
64
65 assert(channel);
66
67 DBG("UST consumer cleaning stream list");
68
69 cds_list_for_each_entry_safe(stream, stmp, &channel->streams.head,
70 send_node) {
9ce5646a
MD
71
72 health_code_update();
73
ffe60014
DG
74 cds_list_del(&stream->send_node);
75 ustctl_destroy_stream(stream->ustream);
d2956687 76 lttng_trace_chunk_put(stream->trace_chunk);
ffe60014
DG
77 free(stream);
78 }
79
80 /*
81 * If a channel is available meaning that was created before the streams
82 * were, delete it.
83 */
84 if (channel->uchan) {
85 lttng_ustconsumer_del_channel(channel);
b83e03c4 86 lttng_ustconsumer_free_channel(channel);
ffe60014
DG
87 }
88 free(channel);
89}
3bd1e081
MD
90
91/*
ffe60014 92 * Add channel to internal consumer state.
3bd1e081 93 *
ffe60014 94 * Returns 0 on success or else a negative value.
3bd1e081 95 */
ffe60014
DG
96static int add_channel(struct lttng_consumer_channel *channel,
97 struct lttng_consumer_local_data *ctx)
3bd1e081
MD
98{
99 int ret = 0;
100
ffe60014
DG
101 assert(channel);
102 assert(ctx);
103
104 if (ctx->on_recv_channel != NULL) {
105 ret = ctx->on_recv_channel(channel);
106 if (ret == 0) {
d8ef542d 107 ret = consumer_add_channel(channel, ctx);
ffe60014
DG
108 } else if (ret < 0) {
109 /* Most likely an ENOMEM. */
110 lttng_consumer_send_error(ctx, LTTCOMM_CONSUMERD_OUTFD_ERROR);
111 goto error;
112 }
113 } else {
d8ef542d 114 ret = consumer_add_channel(channel, ctx);
3bd1e081
MD
115 }
116
d88aee68 117 DBG("UST consumer channel added (key: %" PRIu64 ")", channel->key);
ffe60014
DG
118
119error:
3bd1e081
MD
120 return ret;
121}
122
123/*
ffe60014
DG
124 * Allocate and return a consumer channel object.
125 */
126static struct lttng_consumer_channel *allocate_channel(uint64_t session_id,
d2956687 127 const uint64_t *chunk_id, const char *pathname, const char *name,
da009f2c 128 uint64_t relayd_id, uint64_t key, enum lttng_event_output output,
2bba9e53 129 uint64_t tracefile_size, uint64_t tracefile_count,
ecc48a90 130 uint64_t session_id_per_pid, unsigned int monitor,
d7ba1388 131 unsigned int live_timer_interval,
3d071855 132 const char *root_shm_path, const char *shm_path)
ffe60014
DG
133{
134 assert(pathname);
135 assert(name);
136
d2956687
JG
137 return consumer_allocate_channel(key, session_id, chunk_id, pathname,
138 name, relayd_id, output, tracefile_size,
d7ba1388 139 tracefile_count, session_id_per_pid, monitor,
3d071855 140 live_timer_interval, root_shm_path, shm_path);
ffe60014
DG
141}
142
143/*
144 * Allocate and return a consumer stream object. If _alloc_ret is not NULL, the
145 * error value if applicable is set in it else it is kept untouched.
3bd1e081 146 *
ffe60014 147 * Return NULL on error else the newly allocated stream object.
3bd1e081 148 */
ffe60014
DG
149static struct lttng_consumer_stream *allocate_stream(int cpu, int key,
150 struct lttng_consumer_channel *channel,
d2956687 151 struct lttng_consumer_local_data *ctx, int *_alloc_ret)
ffe60014
DG
152{
153 int alloc_ret;
154 struct lttng_consumer_stream *stream = NULL;
155
156 assert(channel);
157 assert(ctx);
158
159 stream = consumer_allocate_stream(channel->key,
160 key,
ffe60014 161 channel->name,
ffe60014
DG
162 channel->relayd_id,
163 channel->session_id,
d2956687 164 channel->trace_chunk,
ffe60014
DG
165 cpu,
166 &alloc_ret,
4891ece8 167 channel->type,
d2956687 168 channel->monitor);
ffe60014
DG
169 if (stream == NULL) {
170 switch (alloc_ret) {
171 case -ENOENT:
172 /*
173 * We could not find the channel. Can happen if cpu hotplug
174 * happens while tearing down.
175 */
176 DBG3("Could not find channel");
177 break;
178 case -ENOMEM:
179 case -EINVAL:
180 default:
181 lttng_consumer_send_error(ctx, LTTCOMM_CONSUMERD_OUTFD_ERROR);
182 break;
183 }
184 goto error;
185 }
186
d9a2e16e 187 consumer_stream_update_channel_attributes(stream, channel);
ffe60014
DG
188 stream->chan = channel;
189
190error:
191 if (_alloc_ret) {
192 *_alloc_ret = alloc_ret;
193 }
194 return stream;
195}
196
197/*
198 * Send the given stream pointer to the corresponding thread.
199 *
200 * Returns 0 on success else a negative value.
201 */
202static int send_stream_to_thread(struct lttng_consumer_stream *stream,
203 struct lttng_consumer_local_data *ctx)
204{
dae10966
DG
205 int ret;
206 struct lttng_pipe *stream_pipe;
ffe60014
DG
207
208 /* Get the right pipe where the stream will be sent. */
209 if (stream->metadata_flag) {
66d583dc 210 consumer_add_metadata_stream(stream);
dae10966 211 stream_pipe = ctx->consumer_metadata_pipe;
ffe60014 212 } else {
66d583dc 213 consumer_add_data_stream(stream);
dae10966 214 stream_pipe = ctx->consumer_data_pipe;
ffe60014
DG
215 }
216
5ab66908
MD
217 /*
218 * From this point on, the stream's ownership has been moved away from
a8086cf4
JR
219 * the channel and it becomes globally visible. Hence, remove it from
220 * the local stream list to prevent the stream from being both local and
221 * global.
5ab66908
MD
222 */
223 stream->globally_visible = 1;
a8086cf4 224 cds_list_del(&stream->send_node);
5ab66908 225
dae10966 226 ret = lttng_pipe_write(stream_pipe, &stream, sizeof(stream));
ffe60014 227 if (ret < 0) {
dae10966
DG
228 ERR("Consumer write %s stream to pipe %d",
229 stream->metadata_flag ? "metadata" : "data",
230 lttng_pipe_get_writefd(stream_pipe));
5ab66908
MD
231 if (stream->metadata_flag) {
232 consumer_del_stream_for_metadata(stream);
233 } else {
234 consumer_del_stream_for_data(stream);
235 }
a8086cf4 236 goto error;
ffe60014 237 }
a8086cf4 238
5ab66908 239error:
ffe60014
DG
240 return ret;
241}
242
4628484a
MD
243static
244int get_stream_shm_path(char *stream_shm_path, const char *shm_path, int cpu)
245{
45863397 246 char cpu_nr[INT_MAX_STR_LEN]; /* int max len */
4628484a
MD
247 int ret;
248
249 strncpy(stream_shm_path, shm_path, PATH_MAX);
250 stream_shm_path[PATH_MAX - 1] = '\0';
45863397 251 ret = snprintf(cpu_nr, INT_MAX_STR_LEN, "%i", cpu);
67f8cb8d
MD
252 if (ret < 0) {
253 PERROR("snprintf");
4628484a
MD
254 goto end;
255 }
256 strncat(stream_shm_path, cpu_nr,
257 PATH_MAX - strlen(stream_shm_path) - 1);
258 ret = 0;
259end:
260 return ret;
261}
262
d88aee68
DG
263/*
264 * Create streams for the given channel using liblttng-ust-ctl.
d2956687 265 * The channel lock must be acquired by the caller.
d88aee68
DG
266 *
267 * Return 0 on success else a negative value.
268 */
ffe60014 269static int create_ust_streams(struct lttng_consumer_channel *channel,
d2956687 270 struct lttng_consumer_local_data *ctx)
ffe60014
DG
271{
272 int ret, cpu = 0;
273 struct ustctl_consumer_stream *ustream;
274 struct lttng_consumer_stream *stream;
d2956687 275 pthread_mutex_t *current_stream_lock = NULL;
ffe60014
DG
276
277 assert(channel);
278 assert(ctx);
279
280 /*
281 * While a stream is available from ustctl. When NULL is returned, we've
282 * reached the end of the possible stream for the channel.
283 */
284 while ((ustream = ustctl_create_stream(channel->uchan, cpu))) {
285 int wait_fd;
04ef1097 286 int ust_metadata_pipe[2];
ffe60014 287
9ce5646a
MD
288 health_code_update();
289
04ef1097
MD
290 if (channel->type == CONSUMER_CHANNEL_TYPE_METADATA && channel->monitor) {
291 ret = utils_create_pipe_cloexec_nonblock(ust_metadata_pipe);
292 if (ret < 0) {
293 ERR("Create ust metadata poll pipe");
294 goto error;
295 }
296 wait_fd = ust_metadata_pipe[0];
297 } else {
298 wait_fd = ustctl_stream_get_wait_fd(ustream);
299 }
ffe60014
DG
300
301 /* Allocate consumer stream object. */
d2956687 302 stream = allocate_stream(cpu, wait_fd, channel, ctx, &ret);
ffe60014
DG
303 if (!stream) {
304 goto error_alloc;
305 }
306 stream->ustream = ustream;
307 /*
308 * Store it so we can save multiple function calls afterwards since
309 * this value is used heavily in the stream threads. This is UST
310 * specific so this is why it's done after allocation.
311 */
312 stream->wait_fd = wait_fd;
313
b31398bb
DG
314 /*
315 * Increment channel refcount since the channel reference has now been
316 * assigned in the allocation process above.
317 */
10a50311
JD
318 if (stream->chan->monitor) {
319 uatomic_inc(&stream->chan->refcount);
320 }
b31398bb 321
d2956687
JG
322 pthread_mutex_lock(&stream->lock);
323 current_stream_lock = &stream->lock;
ffe60014
DG
324 /*
325 * Order is important this is why a list is used. On error, the caller
326 * should clean this list.
327 */
328 cds_list_add_tail(&stream->send_node, &channel->streams.head);
329
330 ret = ustctl_get_max_subbuf_size(stream->ustream,
331 &stream->max_sb_size);
332 if (ret < 0) {
333 ERR("ustctl_get_max_subbuf_size failed for stream %s",
334 stream->name);
335 goto error;
336 }
337
338 /* Do actions once stream has been received. */
339 if (ctx->on_recv_stream) {
340 ret = ctx->on_recv_stream(stream);
341 if (ret < 0) {
342 goto error;
343 }
344 }
345
d88aee68 346 DBG("UST consumer add stream %s (key: %" PRIu64 ") with relayd id %" PRIu64,
ffe60014
DG
347 stream->name, stream->key, stream->relayd_stream_id);
348
349 /* Set next CPU stream. */
350 channel->streams.count = ++cpu;
d88aee68
DG
351
352 /* Keep stream reference when creating metadata. */
353 if (channel->type == CONSUMER_CHANNEL_TYPE_METADATA) {
354 channel->metadata_stream = stream;
8de4f941
JG
355 if (channel->monitor) {
356 /* Set metadata poll pipe if we created one */
357 memcpy(stream->ust_metadata_poll_pipe,
358 ust_metadata_pipe,
359 sizeof(ust_metadata_pipe));
360 }
d88aee68 361 }
d2956687
JG
362 pthread_mutex_unlock(&stream->lock);
363 current_stream_lock = NULL;
ffe60014
DG
364 }
365
366 return 0;
367
368error:
369error_alloc:
d2956687
JG
370 if (current_stream_lock) {
371 pthread_mutex_unlock(current_stream_lock);
372 }
ffe60014
DG
373 return ret;
374}
375
4628484a
MD
376/*
377 * create_posix_shm is never called concurrently within a process.
378 */
379static
380int create_posix_shm(void)
381{
382 char tmp_name[NAME_MAX];
383 int shmfd, ret;
384
385 ret = snprintf(tmp_name, NAME_MAX, "/ust-shm-consumer-%d", getpid());
386 if (ret < 0) {
387 PERROR("snprintf");
388 return -1;
389 }
390 /*
391 * Allocate shm, and immediately unlink its shm oject, keeping
392 * only the file descriptor as a reference to the object.
393 * We specifically do _not_ use the / at the beginning of the
394 * pathname so that some OS implementations can keep it local to
395 * the process (POSIX leaves this implementation-defined).
396 */
397 shmfd = shm_open(tmp_name, O_CREAT | O_EXCL | O_RDWR, 0700);
398 if (shmfd < 0) {
399 PERROR("shm_open");
400 goto error_shm_open;
401 }
402 ret = shm_unlink(tmp_name);
403 if (ret < 0 && errno != ENOENT) {
404 PERROR("shm_unlink");
405 goto error_shm_release;
406 }
407 return shmfd;
408
409error_shm_release:
410 ret = close(shmfd);
411 if (ret) {
412 PERROR("close");
413 }
414error_shm_open:
415 return -1;
416}
417
d2956687
JG
418static int open_ust_stream_fd(struct lttng_consumer_channel *channel, int cpu,
419 const struct lttng_credentials *session_credentials)
4628484a
MD
420{
421 char shm_path[PATH_MAX];
422 int ret;
423
424 if (!channel->shm_path[0]) {
425 return create_posix_shm();
426 }
427 ret = get_stream_shm_path(shm_path, channel->shm_path, cpu);
428 if (ret) {
429 goto error_shm_path;
430 }
431 return run_as_open(shm_path,
432 O_RDWR | O_CREAT | O_EXCL, S_IRUSR | S_IWUSR,
d2956687 433 session_credentials->uid, session_credentials->gid);
4628484a
MD
434
435error_shm_path:
436 return -1;
437}
438
ffe60014
DG
439/*
440 * Create an UST channel with the given attributes and send it to the session
441 * daemon using the ust ctl API.
442 *
443 * Return 0 on success or else a negative value.
444 */
4628484a
MD
445static int create_ust_channel(struct lttng_consumer_channel *channel,
446 struct ustctl_consumer_channel_attr *attr,
447 struct ustctl_consumer_channel **ust_chanp)
ffe60014 448{
4628484a
MD
449 int ret, nr_stream_fds, i, j;
450 int *stream_fds;
451 struct ustctl_consumer_channel *ust_channel;
ffe60014 452
4628484a 453 assert(channel);
ffe60014 454 assert(attr);
4628484a 455 assert(ust_chanp);
d2956687 456 assert(channel->buffer_credentials.is_set);
ffe60014
DG
457
458 DBG3("Creating channel to ustctl with attr: [overwrite: %d, "
459 "subbuf_size: %" PRIu64 ", num_subbuf: %" PRIu64 ", "
460 "switch_timer_interval: %u, read_timer_interval: %u, "
461 "output: %d, type: %d", attr->overwrite, attr->subbuf_size,
462 attr->num_subbuf, attr->switch_timer_interval,
463 attr->read_timer_interval, attr->output, attr->type);
464
4628484a
MD
465 if (channel->type == CONSUMER_CHANNEL_TYPE_METADATA)
466 nr_stream_fds = 1;
467 else
468 nr_stream_fds = ustctl_get_nr_stream_per_channel();
469 stream_fds = zmalloc(nr_stream_fds * sizeof(*stream_fds));
470 if (!stream_fds) {
471 ret = -1;
472 goto error_alloc;
473 }
474 for (i = 0; i < nr_stream_fds; i++) {
d2956687
JG
475 stream_fds[i] = open_ust_stream_fd(channel, i,
476 &channel->buffer_credentials.value);
4628484a
MD
477 if (stream_fds[i] < 0) {
478 ret = -1;
479 goto error_open;
480 }
481 }
482 ust_channel = ustctl_create_channel(attr, stream_fds, nr_stream_fds);
483 if (!ust_channel) {
ffe60014
DG
484 ret = -1;
485 goto error_create;
486 }
4628484a
MD
487 channel->nr_stream_fds = nr_stream_fds;
488 channel->stream_fds = stream_fds;
489 *ust_chanp = ust_channel;
ffe60014
DG
490
491 return 0;
492
493error_create:
4628484a
MD
494error_open:
495 for (j = i - 1; j >= 0; j--) {
496 int closeret;
497
498 closeret = close(stream_fds[j]);
499 if (closeret) {
500 PERROR("close");
501 }
502 if (channel->shm_path[0]) {
503 char shm_path[PATH_MAX];
504
505 closeret = get_stream_shm_path(shm_path,
506 channel->shm_path, j);
507 if (closeret) {
508 ERR("Cannot get stream shm path");
509 }
510 closeret = run_as_unlink(shm_path,
d2956687
JG
511 channel->buffer_credentials.value.uid,
512 channel->buffer_credentials.value.gid);
4628484a 513 if (closeret) {
4628484a
MD
514 PERROR("unlink %s", shm_path);
515 }
516 }
517 }
518 /* Try to rmdir all directories under shm_path root. */
519 if (channel->root_shm_path[0]) {
602766ec 520 (void) run_as_rmdir_recursive(channel->root_shm_path,
d2956687 521 channel->buffer_credentials.value.uid,
f75c5439
MD
522 channel->buffer_credentials.value.gid,
523 LTTNG_DIRECTORY_HANDLE_SKIP_NON_EMPTY_FLAG);
4628484a
MD
524 }
525 free(stream_fds);
526error_alloc:
ffe60014
DG
527 return ret;
528}
529
d88aee68
DG
530/*
531 * Send a single given stream to the session daemon using the sock.
532 *
533 * Return 0 on success else a negative value.
534 */
ffe60014
DG
535static int send_sessiond_stream(int sock, struct lttng_consumer_stream *stream)
536{
537 int ret;
538
539 assert(stream);
540 assert(sock >= 0);
541
3eb914c0 542 DBG("UST consumer sending stream %" PRIu64 " to sessiond", stream->key);
ffe60014
DG
543
544 /* Send stream to session daemon. */
545 ret = ustctl_send_stream_to_sessiond(sock, stream->ustream);
546 if (ret < 0) {
547 goto error;
548 }
549
ffe60014
DG
550error:
551 return ret;
552}
553
554/*
a3a86f35 555 * Send channel to sessiond and relayd if applicable.
ffe60014 556 *
d88aee68 557 * Return 0 on success or else a negative value.
ffe60014 558 */
a3a86f35 559static int send_channel_to_sessiond_and_relayd(int sock,
ffe60014
DG
560 struct lttng_consumer_channel *channel,
561 struct lttng_consumer_local_data *ctx, int *relayd_error)
562{
0c759fc9 563 int ret, ret_code = LTTCOMM_CONSUMERD_SUCCESS;
ffe60014 564 struct lttng_consumer_stream *stream;
a4baae1b 565 uint64_t net_seq_idx = -1ULL;
ffe60014
DG
566
567 assert(channel);
568 assert(ctx);
569 assert(sock >= 0);
570
571 DBG("UST consumer sending channel %s to sessiond", channel->name);
572
62285ea4
DG
573 if (channel->relayd_id != (uint64_t) -1ULL) {
574 cds_list_for_each_entry(stream, &channel->streams.head, send_node) {
9ce5646a
MD
575
576 health_code_update();
577
62285ea4 578 /* Try to send the stream to the relayd if one is available. */
a3a86f35
JG
579 DBG("Sending stream %" PRIu64 " of channel \"%s\" to relayd",
580 stream->key, channel->name);
62285ea4
DG
581 ret = consumer_send_relayd_stream(stream, stream->chan->pathname);
582 if (ret < 0) {
583 /*
584 * Flag that the relayd was the problem here probably due to a
585 * communicaton error on the socket.
586 */
587 if (relayd_error) {
588 *relayd_error = 1;
589 }
725d28b2 590 ret_code = LTTCOMM_CONSUMERD_RELAYD_FAIL;
ffe60014 591 }
a4baae1b
JD
592 if (net_seq_idx == -1ULL) {
593 net_seq_idx = stream->net_seq_idx;
594 }
595 }
f2a444f1 596 }
ffe60014 597
f2a444f1
DG
598 /* Inform sessiond that we are about to send channel and streams. */
599 ret = consumer_send_status_msg(sock, ret_code);
0c759fc9 600 if (ret < 0 || ret_code != LTTCOMM_CONSUMERD_SUCCESS) {
f2a444f1
DG
601 /*
602 * Either the session daemon is not responding or the relayd died so we
603 * stop now.
604 */
605 goto error;
606 }
607
608 /* Send channel to sessiond. */
609 ret = ustctl_send_channel_to_sessiond(sock, channel->uchan);
610 if (ret < 0) {
611 goto error;
612 }
613
614 ret = ustctl_channel_close_wakeup_fd(channel->uchan);
615 if (ret < 0) {
616 goto error;
617 }
618
619 /* The channel was sent successfully to the sessiond at this point. */
620 cds_list_for_each_entry(stream, &channel->streams.head, send_node) {
9ce5646a
MD
621
622 health_code_update();
623
ffe60014
DG
624 /* Send stream to session daemon. */
625 ret = send_sessiond_stream(sock, stream);
626 if (ret < 0) {
627 goto error;
628 }
629 }
630
631 /* Tell sessiond there is no more stream. */
632 ret = ustctl_send_stream_to_sessiond(sock, NULL);
633 if (ret < 0) {
634 goto error;
635 }
636
637 DBG("UST consumer NULL stream sent to sessiond");
638
639 return 0;
640
641error:
0c759fc9 642 if (ret_code != LTTCOMM_CONSUMERD_SUCCESS) {
f2a444f1
DG
643 ret = -1;
644 }
ffe60014
DG
645 return ret;
646}
647
648/*
649 * Creates a channel and streams and add the channel it to the channel internal
650 * state. The created stream must ONLY be sent once the GET_CHANNEL command is
651 * received.
652 *
653 * Return 0 on success or else, a negative value is returned and the channel
654 * MUST be destroyed by consumer_del_channel().
655 */
75cfe9e6 656static int ask_channel(struct lttng_consumer_local_data *ctx,
ffe60014 657 struct lttng_consumer_channel *channel,
d2956687 658 struct ustctl_consumer_channel_attr *attr)
3bd1e081
MD
659{
660 int ret;
661
ffe60014
DG
662 assert(ctx);
663 assert(channel);
664 assert(attr);
665
666 /*
667 * This value is still used by the kernel consumer since for the kernel,
668 * the stream ownership is not IN the consumer so we need to have the
669 * number of left stream that needs to be initialized so we can know when
670 * to delete the channel (see consumer.c).
671 *
672 * As for the user space tracer now, the consumer creates and sends the
673 * stream to the session daemon which only sends them to the application
674 * once every stream of a channel is received making this value useless
675 * because we they will be added to the poll thread before the application
676 * receives them. This ensures that a stream can not hang up during
677 * initilization of a channel.
678 */
679 channel->nb_init_stream_left = 0;
680
681 /* The reply msg status is handled in the following call. */
4628484a 682 ret = create_ust_channel(channel, attr, &channel->uchan);
ffe60014 683 if (ret < 0) {
10a50311 684 goto end;
3bd1e081
MD
685 }
686
d8ef542d
MD
687 channel->wait_fd = ustctl_channel_get_wait_fd(channel->uchan);
688
10a50311
JD
689 /*
690 * For the snapshots (no monitor), we create the metadata streams
691 * on demand, not during the channel creation.
692 */
693 if (channel->type == CONSUMER_CHANNEL_TYPE_METADATA && !channel->monitor) {
694 ret = 0;
695 goto end;
696 }
697
ffe60014 698 /* Open all streams for this channel. */
d2956687
JG
699 pthread_mutex_lock(&channel->lock);
700 ret = create_ust_streams(channel, ctx);
701 pthread_mutex_unlock(&channel->lock);
ffe60014 702 if (ret < 0) {
10a50311 703 goto end;
ffe60014
DG
704 }
705
10a50311 706end:
3bd1e081
MD
707 return ret;
708}
709
d88aee68
DG
710/*
711 * Send all stream of a channel to the right thread handling it.
712 *
713 * On error, return a negative value else 0 on success.
714 */
715static int send_streams_to_thread(struct lttng_consumer_channel *channel,
716 struct lttng_consumer_local_data *ctx)
717{
718 int ret = 0;
719 struct lttng_consumer_stream *stream, *stmp;
720
721 assert(channel);
722 assert(ctx);
723
724 /* Send streams to the corresponding thread. */
725 cds_list_for_each_entry_safe(stream, stmp, &channel->streams.head,
726 send_node) {
9ce5646a
MD
727
728 health_code_update();
729
d88aee68
DG
730 /* Sending the stream to the thread. */
731 ret = send_stream_to_thread(stream, ctx);
732 if (ret < 0) {
733 /*
734 * If we are unable to send the stream to the thread, there is
735 * a big problem so just stop everything.
736 */
737 goto error;
738 }
d88aee68
DG
739 }
740
741error:
742 return ret;
743}
744
7972aab2
DG
745/*
746 * Flush channel's streams using the given key to retrieve the channel.
747 *
748 * Return 0 on success else an LTTng error code.
749 */
750static int flush_channel(uint64_t chan_key)
751{
752 int ret = 0;
753 struct lttng_consumer_channel *channel;
754 struct lttng_consumer_stream *stream;
755 struct lttng_ht *ht;
756 struct lttng_ht_iter iter;
757
8fd623e0 758 DBG("UST consumer flush channel key %" PRIu64, chan_key);
7972aab2 759
a500c257 760 rcu_read_lock();
7972aab2
DG
761 channel = consumer_find_channel(chan_key);
762 if (!channel) {
8fd623e0 763 ERR("UST consumer flush channel %" PRIu64 " not found", chan_key);
7972aab2
DG
764 ret = LTTNG_ERR_UST_CHAN_NOT_FOUND;
765 goto error;
766 }
767
768 ht = consumer_data.stream_per_chan_id_ht;
769
770 /* For each stream of the channel id, flush it. */
7972aab2
DG
771 cds_lfht_for_each_entry_duplicate(ht->ht,
772 ht->hash_fct(&channel->key, lttng_ht_seed), ht->match_fct,
773 &channel->key, &iter.iter, stream, node_channel_id.node) {
9ce5646a
MD
774
775 health_code_update();
776
0dd01979 777 pthread_mutex_lock(&stream->lock);
5cfcab67
JR
778
779 /*
780 * Protect against concurrent teardown of a stream.
781 */
782 if (cds_lfht_is_node_deleted(&stream->node.node)) {
783 goto next;
784 }
785
0dd01979
MD
786 if (!stream->quiescent) {
787 ustctl_flush_buffer(stream->ustream, 0);
788 stream->quiescent = true;
789 }
5cfcab67 790next:
0dd01979
MD
791 pthread_mutex_unlock(&stream->lock);
792 }
793error:
794 rcu_read_unlock();
795 return ret;
796}
797
798/*
799 * Clear quiescent state from channel's streams using the given key to
800 * retrieve the channel.
801 *
802 * Return 0 on success else an LTTng error code.
803 */
804static int clear_quiescent_channel(uint64_t chan_key)
805{
806 int ret = 0;
807 struct lttng_consumer_channel *channel;
808 struct lttng_consumer_stream *stream;
809 struct lttng_ht *ht;
810 struct lttng_ht_iter iter;
811
812 DBG("UST consumer clear quiescent channel key %" PRIu64, chan_key);
813
814 rcu_read_lock();
815 channel = consumer_find_channel(chan_key);
816 if (!channel) {
817 ERR("UST consumer clear quiescent channel %" PRIu64 " not found", chan_key);
818 ret = LTTNG_ERR_UST_CHAN_NOT_FOUND;
819 goto error;
820 }
821
822 ht = consumer_data.stream_per_chan_id_ht;
823
824 /* For each stream of the channel id, clear quiescent state. */
825 cds_lfht_for_each_entry_duplicate(ht->ht,
826 ht->hash_fct(&channel->key, lttng_ht_seed), ht->match_fct,
827 &channel->key, &iter.iter, stream, node_channel_id.node) {
828
829 health_code_update();
830
831 pthread_mutex_lock(&stream->lock);
832 stream->quiescent = false;
833 pthread_mutex_unlock(&stream->lock);
7972aab2 834 }
7972aab2 835error:
a500c257 836 rcu_read_unlock();
7972aab2
DG
837 return ret;
838}
839
d88aee68
DG
840/*
841 * Close metadata stream wakeup_fd using the given key to retrieve the channel.
842 *
843 * Return 0 on success else an LTTng error code.
844 */
845static int close_metadata(uint64_t chan_key)
846{
ea88ca2a 847 int ret = 0;
d88aee68 848 struct lttng_consumer_channel *channel;
f65a74be 849 unsigned int channel_monitor;
d88aee68 850
8fd623e0 851 DBG("UST consumer close metadata key %" PRIu64, chan_key);
d88aee68
DG
852
853 channel = consumer_find_channel(chan_key);
854 if (!channel) {
84cc9aa0
DG
855 /*
856 * This is possible if the metadata thread has issue a delete because
857 * the endpoint point of the stream hung up. There is no way the
858 * session daemon can know about it thus use a DBG instead of an actual
859 * error.
860 */
861 DBG("UST consumer close metadata %" PRIu64 " not found", chan_key);
d88aee68
DG
862 ret = LTTNG_ERR_UST_CHAN_NOT_FOUND;
863 goto error;
864 }
865
ea88ca2a 866 pthread_mutex_lock(&consumer_data.lock);
a9838785 867 pthread_mutex_lock(&channel->lock);
f65a74be 868 channel_monitor = channel->monitor;
73811ecc
DG
869 if (cds_lfht_is_node_deleted(&channel->node.node)) {
870 goto error_unlock;
871 }
872
6d574024 873 lttng_ustconsumer_close_metadata(channel);
f65a74be
JG
874 pthread_mutex_unlock(&channel->lock);
875 pthread_mutex_unlock(&consumer_data.lock);
d88aee68 876
f65a74be
JG
877 /*
878 * The ownership of a metadata channel depends on the type of
879 * session to which it belongs. In effect, the monitor flag is checked
880 * to determine if this metadata channel is in "snapshot" mode or not.
881 *
882 * In the non-snapshot case, the metadata channel is created along with
883 * a single stream which will remain present until the metadata channel
884 * is destroyed (on the destruction of its session). In this case, the
885 * metadata stream in "monitored" by the metadata poll thread and holds
886 * the ownership of its channel.
887 *
888 * Closing the metadata will cause the metadata stream's "metadata poll
889 * pipe" to be closed. Closing this pipe will wake-up the metadata poll
890 * thread which will teardown the metadata stream which, in return,
891 * deletes the metadata channel.
892 *
893 * In the snapshot case, the metadata stream is created and destroyed
894 * on every snapshot record. Since the channel doesn't have an owner
895 * other than the session daemon, it is safe to destroy it immediately
896 * on reception of the CLOSE_METADATA command.
897 */
898 if (!channel_monitor) {
899 /*
900 * The channel and consumer_data locks must be
901 * released before this call since consumer_del_channel
902 * re-acquires the channel and consumer_data locks to teardown
903 * the channel and queue its reclamation by the "call_rcu"
904 * worker thread.
905 */
906 consumer_del_channel(channel);
907 }
908
909 return ret;
ea88ca2a 910error_unlock:
a9838785 911 pthread_mutex_unlock(&channel->lock);
ea88ca2a 912 pthread_mutex_unlock(&consumer_data.lock);
d88aee68
DG
913error:
914 return ret;
915}
916
917/*
918 * RCU read side lock MUST be acquired before calling this function.
919 *
920 * Return 0 on success else an LTTng error code.
921 */
922static int setup_metadata(struct lttng_consumer_local_data *ctx, uint64_t key)
923{
924 int ret;
925 struct lttng_consumer_channel *metadata;
926
8fd623e0 927 DBG("UST consumer setup metadata key %" PRIu64, key);
d88aee68
DG
928
929 metadata = consumer_find_channel(key);
930 if (!metadata) {
931 ERR("UST consumer push metadata %" PRIu64 " not found", key);
932 ret = LTTNG_ERR_UST_CHAN_NOT_FOUND;
10a50311
JD
933 goto end;
934 }
935
936 /*
937 * In no monitor mode, the metadata channel has no stream(s) so skip the
938 * ownership transfer to the metadata thread.
939 */
940 if (!metadata->monitor) {
941 DBG("Metadata channel in no monitor");
942 ret = 0;
943 goto end;
d88aee68
DG
944 }
945
946 /*
947 * Send metadata stream to relayd if one available. Availability is
948 * known if the stream is still in the list of the channel.
949 */
950 if (cds_list_empty(&metadata->streams.head)) {
951 ERR("Metadata channel key %" PRIu64 ", no stream available.", key);
952 ret = LTTCOMM_CONSUMERD_ERROR_METADATA;
f5a0c9cf 953 goto error_no_stream;
d88aee68
DG
954 }
955
956 /* Send metadata stream to relayd if needed. */
62285ea4
DG
957 if (metadata->metadata_stream->net_seq_idx != (uint64_t) -1ULL) {
958 ret = consumer_send_relayd_stream(metadata->metadata_stream,
959 metadata->pathname);
960 if (ret < 0) {
961 ret = LTTCOMM_CONSUMERD_ERROR_METADATA;
962 goto error;
963 }
601262d6
JD
964 ret = consumer_send_relayd_streams_sent(
965 metadata->metadata_stream->net_seq_idx);
966 if (ret < 0) {
967 ret = LTTCOMM_CONSUMERD_RELAYD_FAIL;
968 goto error;
969 }
d88aee68
DG
970 }
971
a8086cf4
JR
972 /*
973 * Ownership of metadata stream is passed along. Freeing is handled by
974 * the callee.
975 */
d88aee68
DG
976 ret = send_streams_to_thread(metadata, ctx);
977 if (ret < 0) {
978 /*
979 * If we are unable to send the stream to the thread, there is
980 * a big problem so just stop everything.
981 */
982 ret = LTTCOMM_CONSUMERD_FATAL;
a8086cf4 983 goto send_streams_error;
d88aee68
DG
984 }
985 /* List MUST be empty after or else it could be reused. */
986 assert(cds_list_empty(&metadata->streams.head));
987
10a50311
JD
988 ret = 0;
989 goto end;
d88aee68
DG
990
991error:
f2a444f1
DG
992 /*
993 * Delete metadata channel on error. At this point, the metadata stream can
994 * NOT be monitored by the metadata thread thus having the guarantee that
995 * the stream is still in the local stream list of the channel. This call
996 * will make sure to clean that list.
997 */
f5a0c9cf 998 consumer_stream_destroy(metadata->metadata_stream, NULL);
212d67a2
DG
999 cds_list_del(&metadata->metadata_stream->send_node);
1000 metadata->metadata_stream = NULL;
a8086cf4 1001send_streams_error:
f5a0c9cf 1002error_no_stream:
10a50311
JD
1003end:
1004 return ret;
1005}
1006
1007/*
1008 * Snapshot the whole metadata.
d2956687 1009 * RCU read-side lock must be held by the caller.
10a50311
JD
1010 *
1011 * Returns 0 on success, < 0 on error
1012 */
3eb928aa
MD
1013static int snapshot_metadata(struct lttng_consumer_channel *metadata_channel,
1014 uint64_t key, char *path, uint64_t relayd_id,
d2956687 1015 struct lttng_consumer_local_data *ctx)
10a50311
JD
1016{
1017 int ret = 0;
10a50311
JD
1018 struct lttng_consumer_stream *metadata_stream;
1019
1020 assert(path);
1021 assert(ctx);
1022
1023 DBG("UST consumer snapshot metadata with key %" PRIu64 " at path %s",
1024 key, path);
1025
1026 rcu_read_lock();
1027
10a50311
JD
1028 assert(!metadata_channel->monitor);
1029
9ce5646a
MD
1030 health_code_update();
1031
10a50311
JD
1032 /*
1033 * Ask the sessiond if we have new metadata waiting and update the
1034 * consumer metadata cache.
1035 */
94d49140 1036 ret = lttng_ustconsumer_request_metadata(ctx, metadata_channel, 0, 1);
10a50311
JD
1037 if (ret < 0) {
1038 goto error;
1039 }
1040
9ce5646a
MD
1041 health_code_update();
1042
10a50311
JD
1043 /*
1044 * The metadata stream is NOT created in no monitor mode when the channel
1045 * is created on a sessiond ask channel command.
1046 */
d2956687 1047 ret = create_ust_streams(metadata_channel, ctx);
10a50311
JD
1048 if (ret < 0) {
1049 goto error;
1050 }
1051
1052 metadata_stream = metadata_channel->metadata_stream;
1053 assert(metadata_stream);
1054
d2956687 1055 pthread_mutex_lock(&metadata_stream->lock);
10a50311
JD
1056 if (relayd_id != (uint64_t) -1ULL) {
1057 metadata_stream->net_seq_idx = relayd_id;
1058 ret = consumer_send_relayd_stream(metadata_stream, path);
10a50311 1059 } else {
d2956687
JG
1060 ret = consumer_stream_create_output_files(metadata_stream,
1061 false);
1062 }
1063 pthread_mutex_unlock(&metadata_stream->lock);
1064 if (ret < 0) {
1065 goto error_stream;
10a50311
JD
1066 }
1067
04ef1097 1068 do {
9ce5646a
MD
1069 health_code_update();
1070
10a50311
JD
1071 ret = lttng_consumer_read_subbuffer(metadata_stream, ctx);
1072 if (ret < 0) {
94d49140 1073 goto error_stream;
10a50311 1074 }
04ef1097 1075 } while (ret > 0);
10a50311 1076
10a50311
JD
1077error_stream:
1078 /*
1079 * Clean up the stream completly because the next snapshot will use a new
1080 * metadata stream.
1081 */
10a50311 1082 consumer_stream_destroy(metadata_stream, NULL);
212d67a2 1083 cds_list_del(&metadata_stream->send_node);
10a50311
JD
1084 metadata_channel->metadata_stream = NULL;
1085
1086error:
1087 rcu_read_unlock();
1088 return ret;
1089}
1090
1091/*
1092 * Take a snapshot of all the stream of a channel.
d2956687 1093 * RCU read-side lock and the channel lock must be held by the caller.
10a50311
JD
1094 *
1095 * Returns 0 on success, < 0 on error
1096 */
3eb928aa
MD
1097static int snapshot_channel(struct lttng_consumer_channel *channel,
1098 uint64_t key, char *path, uint64_t relayd_id,
e098433c
JG
1099 uint64_t nb_packets_per_stream,
1100 struct lttng_consumer_local_data *ctx)
10a50311
JD
1101{
1102 int ret;
1103 unsigned use_relayd = 0;
1104 unsigned long consumed_pos, produced_pos;
10a50311
JD
1105 struct lttng_consumer_stream *stream;
1106
1107 assert(path);
1108 assert(ctx);
1109
1110 rcu_read_lock();
1111
1112 if (relayd_id != (uint64_t) -1ULL) {
1113 use_relayd = 1;
1114 }
1115
10a50311 1116 assert(!channel->monitor);
6a00837f 1117 DBG("UST consumer snapshot channel %" PRIu64, key);
10a50311
JD
1118
1119 cds_list_for_each_entry(stream, &channel->streams.head, send_node) {
9ce5646a
MD
1120 health_code_update();
1121
10a50311
JD
1122 /* Lock stream because we are about to change its state. */
1123 pthread_mutex_lock(&stream->lock);
d2956687
JG
1124 assert(channel->trace_chunk);
1125 if (!lttng_trace_chunk_get(channel->trace_chunk)) {
1126 /*
1127 * Can't happen barring an internal error as the channel
1128 * holds a reference to the trace chunk.
1129 */
1130 ERR("Failed to acquire reference to channel's trace chunk");
1131 ret = -1;
1132 goto error_unlock;
1133 }
1134 assert(!stream->trace_chunk);
1135 stream->trace_chunk = channel->trace_chunk;
1136
10a50311
JD
1137 stream->net_seq_idx = relayd_id;
1138
1139 if (use_relayd) {
1140 ret = consumer_send_relayd_stream(stream, path);
1141 if (ret < 0) {
1142 goto error_unlock;
1143 }
1144 } else {
d2956687
JG
1145 ret = consumer_stream_create_output_files(stream,
1146 false);
10a50311
JD
1147 if (ret < 0) {
1148 goto error_unlock;
1149 }
d2956687
JG
1150 DBG("UST consumer snapshot stream (%" PRIu64 ")",
1151 stream->key);
10a50311
JD
1152 }
1153
d4d80f77
MD
1154 /*
1155 * If tracing is active, we want to perform a "full" buffer flush.
1156 * Else, if quiescent, it has already been done by the prior stop.
1157 */
1158 if (!stream->quiescent) {
1159 ustctl_flush_buffer(stream->ustream, 0);
1160 }
10a50311
JD
1161
1162 ret = lttng_ustconsumer_take_snapshot(stream);
1163 if (ret < 0) {
1164 ERR("Taking UST snapshot");
1165 goto error_unlock;
1166 }
1167
1168 ret = lttng_ustconsumer_get_produced_snapshot(stream, &produced_pos);
1169 if (ret < 0) {
1170 ERR("Produced UST snapshot position");
1171 goto error_unlock;
1172 }
1173
1174 ret = lttng_ustconsumer_get_consumed_snapshot(stream, &consumed_pos);
1175 if (ret < 0) {
1176 ERR("Consumerd UST snapshot position");
1177 goto error_unlock;
1178 }
1179
5c786ded
JD
1180 /*
1181 * The original value is sent back if max stream size is larger than
d07ceecd 1182 * the possible size of the snapshot. Also, we assume that the session
5c786ded
JD
1183 * daemon should never send a maximum stream size that is lower than
1184 * subbuffer size.
1185 */
d07ceecd
MD
1186 consumed_pos = consumer_get_consume_start_pos(consumed_pos,
1187 produced_pos, nb_packets_per_stream,
1188 stream->max_sb_size);
5c786ded 1189
9377d830 1190 while ((long) (consumed_pos - produced_pos) < 0) {
10a50311
JD
1191 ssize_t read_len;
1192 unsigned long len, padded_len;
1193
9ce5646a
MD
1194 health_code_update();
1195
10a50311
JD
1196 DBG("UST consumer taking snapshot at pos %lu", consumed_pos);
1197
1198 ret = ustctl_get_subbuf(stream->ustream, &consumed_pos);
1199 if (ret < 0) {
1200 if (ret != -EAGAIN) {
1201 PERROR("ustctl_get_subbuf snapshot");
1202 goto error_close_stream;
1203 }
1204 DBG("UST consumer get subbuf failed. Skipping it.");
1205 consumed_pos += stream->max_sb_size;
ddc93ee4 1206 stream->chan->lost_packets++;
10a50311
JD
1207 continue;
1208 }
1209
1210 ret = ustctl_get_subbuf_size(stream->ustream, &len);
1211 if (ret < 0) {
1212 ERR("Snapshot ustctl_get_subbuf_size");
1213 goto error_put_subbuf;
1214 }
1215
1216 ret = ustctl_get_padded_subbuf_size(stream->ustream, &padded_len);
1217 if (ret < 0) {
1218 ERR("Snapshot ustctl_get_padded_subbuf_size");
1219 goto error_put_subbuf;
1220 }
1221
1222 read_len = lttng_consumer_on_read_subbuffer_mmap(ctx, stream, len,
309167d2 1223 padded_len - len, NULL);
10a50311
JD
1224 if (use_relayd) {
1225 if (read_len != len) {
56591bac 1226 ret = -EPERM;
10a50311
JD
1227 goto error_put_subbuf;
1228 }
1229 } else {
1230 if (read_len != padded_len) {
56591bac 1231 ret = -EPERM;
10a50311
JD
1232 goto error_put_subbuf;
1233 }
1234 }
1235
1236 ret = ustctl_put_subbuf(stream->ustream);
1237 if (ret < 0) {
1238 ERR("Snapshot ustctl_put_subbuf");
1239 goto error_close_stream;
1240 }
1241 consumed_pos += stream->max_sb_size;
1242 }
1243
1244 /* Simply close the stream so we can use it on the next snapshot. */
1245 consumer_stream_close(stream);
1246 pthread_mutex_unlock(&stream->lock);
1247 }
1248
1249 rcu_read_unlock();
1250 return 0;
1251
1252error_put_subbuf:
1253 if (ustctl_put_subbuf(stream->ustream) < 0) {
1254 ERR("Snapshot ustctl_put_subbuf");
1255 }
1256error_close_stream:
1257 consumer_stream_close(stream);
1258error_unlock:
1259 pthread_mutex_unlock(&stream->lock);
10a50311 1260 rcu_read_unlock();
d88aee68
DG
1261 return ret;
1262}
1263
331744e3 1264/*
c585821b
MD
1265 * Receive the metadata updates from the sessiond. Supports receiving
1266 * overlapping metadata, but is needs to always belong to a contiguous
1267 * range starting from 0.
1268 * Be careful about the locks held when calling this function: it needs
1269 * the metadata cache flush to concurrently progress in order to
1270 * complete.
331744e3
JD
1271 */
1272int lttng_ustconsumer_recv_metadata(int sock, uint64_t key, uint64_t offset,
93ec662e
JD
1273 uint64_t len, uint64_t version,
1274 struct lttng_consumer_channel *channel, int timer, int wait)
331744e3 1275{
0c759fc9 1276 int ret, ret_code = LTTCOMM_CONSUMERD_SUCCESS;
331744e3
JD
1277 char *metadata_str;
1278
8fd623e0 1279 DBG("UST consumer push metadata key %" PRIu64 " of len %" PRIu64, key, len);
331744e3
JD
1280
1281 metadata_str = zmalloc(len * sizeof(char));
1282 if (!metadata_str) {
1283 PERROR("zmalloc metadata string");
1284 ret_code = LTTCOMM_CONSUMERD_ENOMEM;
1285 goto end;
1286 }
1287
9ce5646a
MD
1288 health_code_update();
1289
331744e3
JD
1290 /* Receive metadata string. */
1291 ret = lttcomm_recv_unix_sock(sock, metadata_str, len);
1292 if (ret < 0) {
1293 /* Session daemon is dead so return gracefully. */
1294 ret_code = ret;
1295 goto end_free;
1296 }
1297
9ce5646a
MD
1298 health_code_update();
1299
331744e3 1300 pthread_mutex_lock(&channel->metadata_cache->lock);
93ec662e
JD
1301 ret = consumer_metadata_cache_write(channel, offset, len, version,
1302 metadata_str);
331744e3
JD
1303 if (ret < 0) {
1304 /* Unable to handle metadata. Notify session daemon. */
1305 ret_code = LTTCOMM_CONSUMERD_ERROR_METADATA;
a32bd775
DG
1306 /*
1307 * Skip metadata flush on write error since the offset and len might
1308 * not have been updated which could create an infinite loop below when
1309 * waiting for the metadata cache to be flushed.
1310 */
1311 pthread_mutex_unlock(&channel->metadata_cache->lock);
a32bd775 1312 goto end_free;
331744e3
JD
1313 }
1314 pthread_mutex_unlock(&channel->metadata_cache->lock);
1315
94d49140
JD
1316 if (!wait) {
1317 goto end_free;
1318 }
5e41ebe1 1319 while (consumer_metadata_cache_flushed(channel, offset + len, timer)) {
331744e3 1320 DBG("Waiting for metadata to be flushed");
9ce5646a
MD
1321
1322 health_code_update();
1323
331744e3
JD
1324 usleep(DEFAULT_METADATA_AVAILABILITY_WAIT_TIME);
1325 }
1326
1327end_free:
1328 free(metadata_str);
1329end:
1330 return ret_code;
1331}
1332
4cbc1a04
DG
1333/*
1334 * Receive command from session daemon and process it.
1335 *
1336 * Return 1 on success else a negative value or 0.
1337 */
3bd1e081
MD
1338int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
1339 int sock, struct pollfd *consumer_sockpoll)
1340{
1341 ssize_t ret;
0c759fc9 1342 enum lttcomm_return_code ret_code = LTTCOMM_CONSUMERD_SUCCESS;
3bd1e081 1343 struct lttcomm_consumer_msg msg;
ffe60014 1344 struct lttng_consumer_channel *channel = NULL;
3bd1e081 1345
9ce5646a
MD
1346 health_code_update();
1347
3bd1e081
MD
1348 ret = lttcomm_recv_unix_sock(sock, &msg, sizeof(msg));
1349 if (ret != sizeof(msg)) {
173af62f
DG
1350 DBG("Consumer received unexpected message size %zd (expects %zu)",
1351 ret, sizeof(msg));
3be74084
DG
1352 /*
1353 * The ret value might 0 meaning an orderly shutdown but this is ok
1354 * since the caller handles this.
1355 */
489f70e9 1356 if (ret > 0) {
c6857fcf 1357 lttng_consumer_send_error(ctx, LTTCOMM_CONSUMERD_ERROR_RECV_CMD);
489f70e9
MD
1358 ret = -1;
1359 }
3bd1e081
MD
1360 return ret;
1361 }
9ce5646a
MD
1362
1363 health_code_update();
1364
84382d49
MD
1365 /* deprecated */
1366 assert(msg.cmd_type != LTTNG_CONSUMER_STOP);
3bd1e081 1367
9ce5646a
MD
1368 health_code_update();
1369
3f8e211f 1370 /* relayd needs RCU read-side lock */
b0b335c8
MD
1371 rcu_read_lock();
1372
3bd1e081 1373 switch (msg.cmd_type) {
00e2e675
DG
1374 case LTTNG_CONSUMER_ADD_RELAYD_SOCKET:
1375 {
f50f23d9 1376 /* Session daemon status message are handled in the following call. */
2527bf85 1377 consumer_add_relayd_socket(msg.u.relayd_sock.net_index,
7735ef9e 1378 msg.u.relayd_sock.type, ctx, sock, consumer_sockpoll,
d3e2ba59
JD
1379 &msg.u.relayd_sock.sock, msg.u.relayd_sock.session_id,
1380 msg.u.relayd_sock.relayd_session_id);
00e2e675
DG
1381 goto end_nosignal;
1382 }
173af62f
DG
1383 case LTTNG_CONSUMER_DESTROY_RELAYD:
1384 {
a6ba4fe1 1385 uint64_t index = msg.u.destroy_relayd.net_seq_idx;
173af62f
DG
1386 struct consumer_relayd_sock_pair *relayd;
1387
a6ba4fe1 1388 DBG("UST consumer destroying relayd %" PRIu64, index);
173af62f
DG
1389
1390 /* Get relayd reference if exists. */
a6ba4fe1 1391 relayd = consumer_find_relayd(index);
173af62f 1392 if (relayd == NULL) {
3448e266 1393 DBG("Unable to find relayd %" PRIu64, index);
e462382a 1394 ret_code = LTTCOMM_CONSUMERD_RELAYD_FAIL;
173af62f
DG
1395 }
1396
a6ba4fe1
DG
1397 /*
1398 * Each relayd socket pair has a refcount of stream attached to it
1399 * which tells if the relayd is still active or not depending on the
1400 * refcount value.
1401 *
1402 * This will set the destroy flag of the relayd object and destroy it
1403 * if the refcount reaches zero when called.
1404 *
1405 * The destroy can happen either here or when a stream fd hangs up.
1406 */
f50f23d9
DG
1407 if (relayd) {
1408 consumer_flag_relayd_for_destroy(relayd);
1409 }
1410
d88aee68 1411 goto end_msg_sessiond;
173af62f 1412 }
3bd1e081
MD
1413 case LTTNG_CONSUMER_UPDATE_STREAM:
1414 {
3f8e211f 1415 rcu_read_unlock();
7ad0a0cb 1416 return -ENOSYS;
3bd1e081 1417 }
6d805429 1418 case LTTNG_CONSUMER_DATA_PENDING:
53632229 1419 {
3be74084 1420 int ret, is_data_pending;
6d805429 1421 uint64_t id = msg.u.data_pending.session_id;
ca22feea 1422
6d805429 1423 DBG("UST consumer data pending command for id %" PRIu64, id);
ca22feea 1424
3be74084 1425 is_data_pending = consumer_data_pending(id);
ca22feea
DG
1426
1427 /* Send back returned value to session daemon */
3be74084
DG
1428 ret = lttcomm_send_unix_sock(sock, &is_data_pending,
1429 sizeof(is_data_pending));
ca22feea 1430 if (ret < 0) {
3be74084 1431 DBG("Error when sending the data pending ret code: %d", ret);
489f70e9 1432 goto error_fatal;
ca22feea 1433 }
f50f23d9
DG
1434
1435 /*
1436 * No need to send back a status message since the data pending
1437 * returned value is the response.
1438 */
ca22feea 1439 break;
53632229 1440 }
ffe60014
DG
1441 case LTTNG_CONSUMER_ASK_CHANNEL_CREATION:
1442 {
1443 int ret;
1444 struct ustctl_consumer_channel_attr attr;
d2956687
JG
1445 const uint64_t chunk_id = msg.u.ask_channel.chunk_id.value;
1446 const struct lttng_credentials buffer_credentials = {
1447 .uid = msg.u.ask_channel.buffer_credentials.uid,
1448 .gid = msg.u.ask_channel.buffer_credentials.gid,
1449 };
ffe60014
DG
1450
1451 /* Create a plain object and reserve a channel key. */
1452 channel = allocate_channel(msg.u.ask_channel.session_id,
d2956687
JG
1453 msg.u.ask_channel.chunk_id.is_set ?
1454 &chunk_id : NULL,
1455 msg.u.ask_channel.pathname,
1456 msg.u.ask_channel.name,
1457 msg.u.ask_channel.relayd_id,
1458 msg.u.ask_channel.key,
1624d5b7
JD
1459 (enum lttng_event_output) msg.u.ask_channel.output,
1460 msg.u.ask_channel.tracefile_size,
2bba9e53 1461 msg.u.ask_channel.tracefile_count,
1950109e 1462 msg.u.ask_channel.session_id_per_pid,
ecc48a90 1463 msg.u.ask_channel.monitor,
d7ba1388 1464 msg.u.ask_channel.live_timer_interval,
3d071855 1465 msg.u.ask_channel.root_shm_path,
d7ba1388 1466 msg.u.ask_channel.shm_path);
ffe60014
DG
1467 if (!channel) {
1468 goto end_channel_error;
1469 }
1470
d2956687
JG
1471 LTTNG_OPTIONAL_SET(&channel->buffer_credentials,
1472 buffer_credentials);
1473
567eb353
DG
1474 /*
1475 * Assign UST application UID to the channel. This value is ignored for
1476 * per PID buffers. This is specific to UST thus setting this after the
1477 * allocation.
1478 */
1479 channel->ust_app_uid = msg.u.ask_channel.ust_app_uid;
1480
ffe60014
DG
1481 /* Build channel attributes from received message. */
1482 attr.subbuf_size = msg.u.ask_channel.subbuf_size;
1483 attr.num_subbuf = msg.u.ask_channel.num_subbuf;
1484 attr.overwrite = msg.u.ask_channel.overwrite;
1485 attr.switch_timer_interval = msg.u.ask_channel.switch_timer_interval;
1486 attr.read_timer_interval = msg.u.ask_channel.read_timer_interval;
7972aab2 1487 attr.chan_id = msg.u.ask_channel.chan_id;
ffe60014 1488 memcpy(attr.uuid, msg.u.ask_channel.uuid, sizeof(attr.uuid));
491d1539 1489 attr.blocking_timeout= msg.u.ask_channel.blocking_timeout;
ffe60014 1490
0c759fc9
DG
1491 /* Match channel buffer type to the UST abi. */
1492 switch (msg.u.ask_channel.output) {
1493 case LTTNG_EVENT_MMAP:
1494 default:
1495 attr.output = LTTNG_UST_MMAP;
1496 break;
1497 }
1498
ffe60014
DG
1499 /* Translate and save channel type. */
1500 switch (msg.u.ask_channel.type) {
1501 case LTTNG_UST_CHAN_PER_CPU:
1502 channel->type = CONSUMER_CHANNEL_TYPE_DATA;
1503 attr.type = LTTNG_UST_CHAN_PER_CPU;
8633d6e3
MD
1504 /*
1505 * Set refcount to 1 for owner. Below, we will
1506 * pass ownership to the
1507 * consumer_thread_channel_poll() thread.
1508 */
1509 channel->refcount = 1;
ffe60014
DG
1510 break;
1511 case LTTNG_UST_CHAN_METADATA:
1512 channel->type = CONSUMER_CHANNEL_TYPE_METADATA;
1513 attr.type = LTTNG_UST_CHAN_METADATA;
1514 break;
1515 default:
1516 assert(0);
1517 goto error_fatal;
1518 };
1519
9ce5646a
MD
1520 health_code_update();
1521
d2956687 1522 ret = ask_channel(ctx, channel, &attr);
ffe60014
DG
1523 if (ret < 0) {
1524 goto end_channel_error;
1525 }
1526
fc643247
MD
1527 if (msg.u.ask_channel.type == LTTNG_UST_CHAN_METADATA) {
1528 ret = consumer_metadata_cache_allocate(channel);
1529 if (ret < 0) {
1530 ERR("Allocating metadata cache");
1531 goto end_channel_error;
1532 }
1533 consumer_timer_switch_start(channel, attr.switch_timer_interval);
1534 attr.switch_timer_interval = 0;
94d49140 1535 } else {
e9404c27
JG
1536 int monitor_start_ret;
1537
94d49140
JD
1538 consumer_timer_live_start(channel,
1539 msg.u.ask_channel.live_timer_interval);
e9404c27
JG
1540 monitor_start_ret = consumer_timer_monitor_start(
1541 channel,
1542 msg.u.ask_channel.monitor_timer_interval);
1543 if (monitor_start_ret < 0) {
1544 ERR("Starting channel monitoring timer failed");
1545 goto end_channel_error;
1546 }
fc643247
MD
1547 }
1548
9ce5646a
MD
1549 health_code_update();
1550
ffe60014
DG
1551 /*
1552 * Add the channel to the internal state AFTER all streams were created
1553 * and successfully sent to session daemon. This way, all streams must
1554 * be ready before this channel is visible to the threads.
fc643247
MD
1555 * If add_channel succeeds, ownership of the channel is
1556 * passed to consumer_thread_channel_poll().
ffe60014
DG
1557 */
1558 ret = add_channel(channel, ctx);
1559 if (ret < 0) {
ea88ca2a
MD
1560 if (msg.u.ask_channel.type == LTTNG_UST_CHAN_METADATA) {
1561 if (channel->switch_timer_enabled == 1) {
1562 consumer_timer_switch_stop(channel);
1563 }
1564 consumer_metadata_cache_destroy(channel);
1565 }
d3e2ba59
JD
1566 if (channel->live_timer_enabled == 1) {
1567 consumer_timer_live_stop(channel);
1568 }
e9404c27
JG
1569 if (channel->monitor_timer_enabled == 1) {
1570 consumer_timer_monitor_stop(channel);
1571 }
ffe60014
DG
1572 goto end_channel_error;
1573 }
1574
9ce5646a
MD
1575 health_code_update();
1576
ffe60014
DG
1577 /*
1578 * Channel and streams are now created. Inform the session daemon that
1579 * everything went well and should wait to receive the channel and
1580 * streams with ustctl API.
1581 */
1582 ret = consumer_send_status_channel(sock, channel);
1583 if (ret < 0) {
1584 /*
489f70e9 1585 * There is probably a problem on the socket.
ffe60014 1586 */
489f70e9 1587 goto error_fatal;
ffe60014
DG
1588 }
1589
1590 break;
1591 }
1592 case LTTNG_CONSUMER_GET_CHANNEL:
1593 {
1594 int ret, relayd_err = 0;
d88aee68 1595 uint64_t key = msg.u.get_channel.key;
ffe60014 1596 struct lttng_consumer_channel *channel;
ffe60014
DG
1597
1598 channel = consumer_find_channel(key);
1599 if (!channel) {
8fd623e0 1600 ERR("UST consumer get channel key %" PRIu64 " not found", key);
e462382a 1601 ret_code = LTTCOMM_CONSUMERD_CHAN_NOT_FOUND;
f3a1fc7b 1602 goto end_get_channel;
ffe60014
DG
1603 }
1604
9ce5646a
MD
1605 health_code_update();
1606
a3a86f35
JG
1607 /* Send the channel to sessiond (and relayd, if applicable). */
1608 ret = send_channel_to_sessiond_and_relayd(sock, channel, ctx,
1609 &relayd_err);
ffe60014
DG
1610 if (ret < 0) {
1611 if (relayd_err) {
1612 /*
1613 * We were unable to send to the relayd the stream so avoid
1614 * sending back a fatal error to the thread since this is OK
f2a444f1
DG
1615 * and the consumer can continue its work. The above call
1616 * has sent the error status message to the sessiond.
ffe60014 1617 */
f3a1fc7b 1618 goto end_get_channel_nosignal;
ffe60014
DG
1619 }
1620 /*
1621 * The communicaton was broken hence there is a bad state between
1622 * the consumer and sessiond so stop everything.
1623 */
f3a1fc7b 1624 goto error_get_channel_fatal;
ffe60014
DG
1625 }
1626
9ce5646a
MD
1627 health_code_update();
1628
10a50311
JD
1629 /*
1630 * In no monitor mode, the streams ownership is kept inside the channel
1631 * so don't send them to the data thread.
1632 */
1633 if (!channel->monitor) {
f3a1fc7b 1634 goto end_get_channel;
10a50311
JD
1635 }
1636
d88aee68
DG
1637 ret = send_streams_to_thread(channel, ctx);
1638 if (ret < 0) {
1639 /*
1640 * If we are unable to send the stream to the thread, there is
1641 * a big problem so just stop everything.
1642 */
f3a1fc7b 1643 goto error_get_channel_fatal;
ffe60014 1644 }
ffe60014
DG
1645 /* List MUST be empty after or else it could be reused. */
1646 assert(cds_list_empty(&channel->streams.head));
f3a1fc7b 1647end_get_channel:
d88aee68 1648 goto end_msg_sessiond;
f3a1fc7b
JG
1649error_get_channel_fatal:
1650 goto error_fatal;
1651end_get_channel_nosignal:
1652 goto end_nosignal;
d88aee68
DG
1653 }
1654 case LTTNG_CONSUMER_DESTROY_CHANNEL:
1655 {
1656 uint64_t key = msg.u.destroy_channel.key;
d88aee68 1657
a0cbdd2e
MD
1658 /*
1659 * Only called if streams have not been sent to stream
1660 * manager thread. However, channel has been sent to
1661 * channel manager thread.
1662 */
1663 notify_thread_del_channel(ctx, key);
d88aee68 1664 goto end_msg_sessiond;
ffe60014 1665 }
d88aee68
DG
1666 case LTTNG_CONSUMER_CLOSE_METADATA:
1667 {
1668 int ret;
1669
1670 ret = close_metadata(msg.u.close_metadata.key);
1671 if (ret != 0) {
1672 ret_code = ret;
1673 }
1674
1675 goto end_msg_sessiond;
1676 }
7972aab2
DG
1677 case LTTNG_CONSUMER_FLUSH_CHANNEL:
1678 {
1679 int ret;
1680
1681 ret = flush_channel(msg.u.flush_channel.key);
1682 if (ret != 0) {
1683 ret_code = ret;
1684 }
1685
1686 goto end_msg_sessiond;
1687 }
0dd01979
MD
1688 case LTTNG_CONSUMER_CLEAR_QUIESCENT_CHANNEL:
1689 {
1690 int ret;
1691
1692 ret = clear_quiescent_channel(
1693 msg.u.clear_quiescent_channel.key);
1694 if (ret != 0) {
1695 ret_code = ret;
1696 }
1697
1698 goto end_msg_sessiond;
1699 }
d88aee68 1700 case LTTNG_CONSUMER_PUSH_METADATA:
ffe60014
DG
1701 {
1702 int ret;
d88aee68 1703 uint64_t len = msg.u.push_metadata.len;
d88aee68 1704 uint64_t key = msg.u.push_metadata.key;
331744e3 1705 uint64_t offset = msg.u.push_metadata.target_offset;
93ec662e 1706 uint64_t version = msg.u.push_metadata.version;
ffe60014
DG
1707 struct lttng_consumer_channel *channel;
1708
8fd623e0
DG
1709 DBG("UST consumer push metadata key %" PRIu64 " of len %" PRIu64, key,
1710 len);
ffe60014
DG
1711
1712 channel = consumer_find_channel(key);
1713 if (!channel) {
000baf6a
DG
1714 /*
1715 * This is possible if the metadata creation on the consumer side
1716 * is in flight vis-a-vis a concurrent push metadata from the
1717 * session daemon. Simply return that the channel failed and the
1718 * session daemon will handle that message correctly considering
1719 * that this race is acceptable thus the DBG() statement here.
1720 */
1721 DBG("UST consumer push metadata %" PRIu64 " not found", key);
1722 ret_code = LTTCOMM_CONSUMERD_CHANNEL_FAIL;
a8ffe244 1723 goto end_push_metadata_msg_sessiond;
d88aee68
DG
1724 }
1725
9ce5646a
MD
1726 health_code_update();
1727
c585821b
MD
1728 if (!len) {
1729 /*
1730 * There is nothing to receive. We have simply
1731 * checked whether the channel can be found.
1732 */
1733 ret_code = LTTCOMM_CONSUMERD_SUCCESS;
a8ffe244 1734 goto end_push_metadata_msg_sessiond;
c585821b
MD
1735 }
1736
d88aee68 1737 /* Tell session daemon we are ready to receive the metadata. */
0c759fc9 1738 ret = consumer_send_status_msg(sock, LTTCOMM_CONSUMERD_SUCCESS);
ffe60014
DG
1739 if (ret < 0) {
1740 /* Somehow, the session daemon is not responding anymore. */
a8ffe244 1741 goto error_push_metadata_fatal;
d88aee68
DG
1742 }
1743
9ce5646a
MD
1744 health_code_update();
1745
d88aee68 1746 /* Wait for more data. */
9ce5646a
MD
1747 health_poll_entry();
1748 ret = lttng_consumer_poll_socket(consumer_sockpoll);
1749 health_poll_exit();
84382d49 1750 if (ret) {
a8ffe244 1751 goto error_push_metadata_fatal;
d88aee68
DG
1752 }
1753
9ce5646a
MD
1754 health_code_update();
1755
331744e3 1756 ret = lttng_ustconsumer_recv_metadata(sock, key, offset,
93ec662e 1757 len, version, channel, 0, 1);
d88aee68 1758 if (ret < 0) {
331744e3 1759 /* error receiving from sessiond */
a8ffe244 1760 goto error_push_metadata_fatal;
331744e3
JD
1761 } else {
1762 ret_code = ret;
a8ffe244 1763 goto end_push_metadata_msg_sessiond;
d88aee68 1764 }
a8ffe244
JG
1765end_push_metadata_msg_sessiond:
1766 goto end_msg_sessiond;
1767error_push_metadata_fatal:
1768 goto error_fatal;
d88aee68
DG
1769 }
1770 case LTTNG_CONSUMER_SETUP_METADATA:
1771 {
1772 int ret;
1773
1774 ret = setup_metadata(ctx, msg.u.setup_metadata.key);
1775 if (ret) {
1776 ret_code = ret;
1777 }
1778 goto end_msg_sessiond;
ffe60014 1779 }
6dc3064a
DG
1780 case LTTNG_CONSUMER_SNAPSHOT_CHANNEL:
1781 {
3eb928aa
MD
1782 struct lttng_consumer_channel *channel;
1783 uint64_t key = msg.u.snapshot_channel.key;
1784
1785 channel = consumer_find_channel(key);
1786 if (!channel) {
1787 DBG("UST snapshot channel not found for key %" PRIu64, key);
1788 ret_code = LTTCOMM_CONSUMERD_CHAN_NOT_FOUND;
10a50311 1789 } else {
3eb928aa
MD
1790 if (msg.u.snapshot_channel.metadata) {
1791 ret = snapshot_metadata(channel, key,
1792 msg.u.snapshot_channel.pathname,
1793 msg.u.snapshot_channel.relayd_id,
d2956687 1794 ctx);
3eb928aa
MD
1795 if (ret < 0) {
1796 ERR("Snapshot metadata failed");
1797 ret_code = LTTCOMM_CONSUMERD_SNAPSHOT_FAILED;
1798 }
1799 } else {
1800 ret = snapshot_channel(channel, key,
1801 msg.u.snapshot_channel.pathname,
1802 msg.u.snapshot_channel.relayd_id,
1803 msg.u.snapshot_channel.nb_packets_per_stream,
1804 ctx);
1805 if (ret < 0) {
1806 ERR("Snapshot channel failed");
1807 ret_code = LTTCOMM_CONSUMERD_SNAPSHOT_FAILED;
1808 }
10a50311
JD
1809 }
1810 }
9ce5646a 1811 health_code_update();
6dc3064a
DG
1812 ret = consumer_send_status_msg(sock, ret_code);
1813 if (ret < 0) {
1814 /* Somehow, the session daemon is not responding anymore. */
1815 goto end_nosignal;
1816 }
9ce5646a 1817 health_code_update();
6dc3064a
DG
1818 break;
1819 }
fb83fe64
JD
1820 case LTTNG_CONSUMER_DISCARDED_EVENTS:
1821 {
beb59458
MJ
1822 int ret = 0;
1823 uint64_t discarded_events;
fb83fe64
JD
1824 struct lttng_ht_iter iter;
1825 struct lttng_ht *ht;
1826 struct lttng_consumer_stream *stream;
1827 uint64_t id = msg.u.discarded_events.session_id;
1828 uint64_t key = msg.u.discarded_events.channel_key;
1829
1830 DBG("UST consumer discarded events command for session id %"
1831 PRIu64, id);
1832 rcu_read_lock();
1833 pthread_mutex_lock(&consumer_data.lock);
1834
1835 ht = consumer_data.stream_list_ht;
1836
1837 /*
1838 * We only need a reference to the channel, but they are not
1839 * directly indexed, so we just use the first matching stream
1840 * to extract the information we need, we default to 0 if not
1841 * found (no events are dropped if the channel is not yet in
1842 * use).
1843 */
beb59458 1844 discarded_events = 0;
fb83fe64
JD
1845 cds_lfht_for_each_entry_duplicate(ht->ht,
1846 ht->hash_fct(&id, lttng_ht_seed),
1847 ht->match_fct, &id,
1848 &iter.iter, stream, node_session_id.node) {
1849 if (stream->chan->key == key) {
beb59458 1850 discarded_events = stream->chan->discarded_events;
fb83fe64
JD
1851 break;
1852 }
1853 }
1854 pthread_mutex_unlock(&consumer_data.lock);
1855 rcu_read_unlock();
1856
1857 DBG("UST consumer discarded events command for session id %"
1858 PRIu64 ", channel key %" PRIu64, id, key);
1859
1860 health_code_update();
1861
1862 /* Send back returned value to session daemon */
beb59458 1863 ret = lttcomm_send_unix_sock(sock, &discarded_events, sizeof(discarded_events));
fb83fe64
JD
1864 if (ret < 0) {
1865 PERROR("send discarded events");
1866 goto error_fatal;
1867 }
1868
1869 break;
1870 }
1871 case LTTNG_CONSUMER_LOST_PACKETS:
1872 {
9a06e8d4
JG
1873 int ret;
1874 uint64_t lost_packets;
fb83fe64
JD
1875 struct lttng_ht_iter iter;
1876 struct lttng_ht *ht;
1877 struct lttng_consumer_stream *stream;
1878 uint64_t id = msg.u.lost_packets.session_id;
1879 uint64_t key = msg.u.lost_packets.channel_key;
1880
1881 DBG("UST consumer lost packets command for session id %"
1882 PRIu64, id);
1883 rcu_read_lock();
1884 pthread_mutex_lock(&consumer_data.lock);
1885
1886 ht = consumer_data.stream_list_ht;
1887
1888 /*
1889 * We only need a reference to the channel, but they are not
1890 * directly indexed, so we just use the first matching stream
1891 * to extract the information we need, we default to 0 if not
1892 * found (no packets lost if the channel is not yet in use).
1893 */
9a06e8d4 1894 lost_packets = 0;
fb83fe64
JD
1895 cds_lfht_for_each_entry_duplicate(ht->ht,
1896 ht->hash_fct(&id, lttng_ht_seed),
1897 ht->match_fct, &id,
1898 &iter.iter, stream, node_session_id.node) {
1899 if (stream->chan->key == key) {
9a06e8d4 1900 lost_packets = stream->chan->lost_packets;
fb83fe64
JD
1901 break;
1902 }
1903 }
1904 pthread_mutex_unlock(&consumer_data.lock);
1905 rcu_read_unlock();
1906
1907 DBG("UST consumer lost packets command for session id %"
1908 PRIu64 ", channel key %" PRIu64, id, key);
1909
1910 health_code_update();
1911
1912 /* Send back returned value to session daemon */
9a06e8d4
JG
1913 ret = lttcomm_send_unix_sock(sock, &lost_packets,
1914 sizeof(lost_packets));
fb83fe64
JD
1915 if (ret < 0) {
1916 PERROR("send lost packets");
1917 goto error_fatal;
1918 }
1919
1920 break;
1921 }
b3530820
JG
1922 case LTTNG_CONSUMER_SET_CHANNEL_MONITOR_PIPE:
1923 {
1924 int channel_monitor_pipe;
1925
1926 ret_code = LTTCOMM_CONSUMERD_SUCCESS;
1927 /* Successfully received the command's type. */
1928 ret = consumer_send_status_msg(sock, ret_code);
1929 if (ret < 0) {
1930 goto error_fatal;
1931 }
1932
1933 ret = lttcomm_recv_fds_unix_sock(sock, &channel_monitor_pipe,
1934 1);
1935 if (ret != sizeof(channel_monitor_pipe)) {
1936 ERR("Failed to receive channel monitor pipe");
1937 goto error_fatal;
1938 }
1939
1940 DBG("Received channel monitor pipe (%d)", channel_monitor_pipe);
1941 ret = consumer_timer_thread_set_channel_monitor_pipe(
1942 channel_monitor_pipe);
1943 if (!ret) {
1944 int flags;
1945
1946 ret_code = LTTCOMM_CONSUMERD_SUCCESS;
1947 /* Set the pipe as non-blocking. */
1948 ret = fcntl(channel_monitor_pipe, F_GETFL, 0);
1949 if (ret == -1) {
1950 PERROR("fcntl get flags of the channel monitoring pipe");
1951 goto error_fatal;
1952 }
1953 flags = ret;
1954
1955 ret = fcntl(channel_monitor_pipe, F_SETFL,
1956 flags | O_NONBLOCK);
1957 if (ret == -1) {
1958 PERROR("fcntl set O_NONBLOCK flag of the channel monitoring pipe");
1959 goto error_fatal;
1960 }
1961 DBG("Channel monitor pipe set as non-blocking");
1962 } else {
1963 ret_code = LTTCOMM_CONSUMERD_ALREADY_SET;
1964 }
1965 goto end_msg_sessiond;
1966 }
b99a8d42
JD
1967 case LTTNG_CONSUMER_ROTATE_CHANNEL:
1968 {
92b7a7f8
MD
1969 struct lttng_consumer_channel *channel;
1970 uint64_t key = msg.u.rotate_channel.key;
b99a8d42 1971
92b7a7f8
MD
1972 channel = consumer_find_channel(key);
1973 if (!channel) {
1974 DBG("Channel %" PRIu64 " not found", key);
1975 ret_code = LTTCOMM_CONSUMERD_CHAN_NOT_FOUND;
1976 } else {
1977 /*
1978 * Sample the rotate position of all the streams in
1979 * this channel.
1980 */
1981 ret = lttng_consumer_rotate_channel(channel, key,
92b7a7f8
MD
1982 msg.u.rotate_channel.relayd_id,
1983 msg.u.rotate_channel.metadata,
92b7a7f8
MD
1984 ctx);
1985 if (ret < 0) {
1986 ERR("Rotate channel failed");
1987 ret_code = LTTCOMM_CONSUMERD_ROTATION_FAIL;
1988 }
b99a8d42 1989
92b7a7f8
MD
1990 health_code_update();
1991 }
b99a8d42
JD
1992 ret = consumer_send_status_msg(sock, ret_code);
1993 if (ret < 0) {
1994 /* Somehow, the session daemon is not responding anymore. */
41c7a76d 1995 goto end_rotate_channel_nosignal;
b99a8d42
JD
1996 }
1997
1998 /*
1999 * Rotate the streams that are ready right now.
2000 * FIXME: this is a second consecutive iteration over the
2001 * streams in a channel, there is probably a better way to
2002 * handle this, but it needs to be after the
2003 * consumer_send_status_msg() call.
2004 */
92b7a7f8
MD
2005 if (channel) {
2006 ret = lttng_consumer_rotate_ready_streams(
2007 channel, key, ctx);
2008 if (ret < 0) {
2009 ERR("Rotate channel failed");
2010 }
b99a8d42
JD
2011 }
2012 break;
41c7a76d
JG
2013end_rotate_channel_nosignal:
2014 goto end_nosignal;
b99a8d42 2015 }
d2956687 2016 case LTTNG_CONSUMER_INIT:
00fb02ac 2017 {
d2956687
JG
2018 ret_code = lttng_consumer_init_command(ctx,
2019 msg.u.init.sessiond_uuid);
d88744a4 2020 health_code_update();
d88744a4
JD
2021 ret = consumer_send_status_msg(sock, ret_code);
2022 if (ret < 0) {
2023 /* Somehow, the session daemon is not responding anymore. */
2024 goto end_nosignal;
2025 }
2026 break;
2027 }
d2956687 2028 case LTTNG_CONSUMER_CREATE_TRACE_CHUNK:
d88744a4 2029 {
d2956687 2030 const struct lttng_credentials credentials = {
e5add6d0
JG
2031 .uid = msg.u.create_trace_chunk.credentials.value.uid,
2032 .gid = msg.u.create_trace_chunk.credentials.value.gid,
d2956687
JG
2033 };
2034 const bool is_local_trace =
2035 !msg.u.create_trace_chunk.relayd_id.is_set;
2036 const uint64_t relayd_id =
2037 msg.u.create_trace_chunk.relayd_id.value;
2038 const char *chunk_override_name =
2039 *msg.u.create_trace_chunk.override_name ?
2040 msg.u.create_trace_chunk.override_name :
2041 NULL;
cbf53d23 2042 struct lttng_directory_handle *chunk_directory_handle = NULL;
d88744a4 2043
d2956687
JG
2044 /*
2045 * The session daemon will only provide a chunk directory file
2046 * descriptor for local traces.
2047 */
2048 if (is_local_trace) {
2049 int chunk_dirfd;
19990ed5 2050
d2956687
JG
2051 /* Acnowledge the reception of the command. */
2052 ret = consumer_send_status_msg(sock,
2053 LTTCOMM_CONSUMERD_SUCCESS);
2054 if (ret < 0) {
2055 /* Somehow, the session daemon is not responding anymore. */
2056 goto end_nosignal;
2057 }
92816cc3 2058
5da88b0f
MD
2059 /*
2060 * Receive trace chunk domain dirfd.
2061 */
d2956687
JG
2062 ret = lttcomm_recv_fds_unix_sock(sock, &chunk_dirfd, 1);
2063 if (ret != sizeof(chunk_dirfd)) {
5da88b0f 2064 ERR("Failed to receive trace chunk domain directory file descriptor");
d2956687
JG
2065 goto error_fatal;
2066 }
92816cc3 2067
5da88b0f 2068 DBG("Received trace chunk domain directory fd (%d)",
d2956687 2069 chunk_dirfd);
cbf53d23 2070 chunk_directory_handle = lttng_directory_handle_create_from_dirfd(
d2956687 2071 chunk_dirfd);
cbf53d23 2072 if (!chunk_directory_handle) {
5da88b0f 2073 ERR("Failed to initialize chunk domain directory handle from directory file descriptor");
d2956687
JG
2074 if (close(chunk_dirfd)) {
2075 PERROR("Failed to close chunk directory file descriptor");
2076 }
2077 goto error_fatal;
2078 }
92816cc3
JG
2079 }
2080
d2956687
JG
2081 ret_code = lttng_consumer_create_trace_chunk(
2082 !is_local_trace ? &relayd_id : NULL,
2083 msg.u.create_trace_chunk.session_id,
2084 msg.u.create_trace_chunk.chunk_id,
e5add6d0
JG
2085 (time_t) msg.u.create_trace_chunk
2086 .creation_timestamp,
d2956687 2087 chunk_override_name,
e5add6d0
JG
2088 msg.u.create_trace_chunk.credentials.is_set ?
2089 &credentials :
2090 NULL,
cbf53d23
JG
2091 chunk_directory_handle);
2092 lttng_directory_handle_put(chunk_directory_handle);
d2956687 2093 goto end_msg_sessiond;
00fb02ac 2094 }
d2956687 2095 case LTTNG_CONSUMER_CLOSE_TRACE_CHUNK:
a1ae2ea5 2096 {
bbc4768c
JG
2097 enum lttng_trace_chunk_command_type close_command =
2098 msg.u.close_trace_chunk.close_command.value;
d2956687
JG
2099 const uint64_t relayd_id =
2100 msg.u.close_trace_chunk.relayd_id.value;
ecd1a12f
MD
2101 struct lttcomm_consumer_close_trace_chunk_reply reply;
2102 char closed_trace_chunk_path[LTTNG_PATH_MAX];
2103 int ret;
d2956687
JG
2104
2105 ret_code = lttng_consumer_close_trace_chunk(
2106 msg.u.close_trace_chunk.relayd_id.is_set ?
bbc4768c
JG
2107 &relayd_id :
2108 NULL,
d2956687
JG
2109 msg.u.close_trace_chunk.session_id,
2110 msg.u.close_trace_chunk.chunk_id,
bbc4768c
JG
2111 (time_t) msg.u.close_trace_chunk.close_timestamp,
2112 msg.u.close_trace_chunk.close_command.is_set ?
2113 &close_command :
ecd1a12f
MD
2114 NULL, closed_trace_chunk_path);
2115 reply.ret_code = ret_code;
2116 reply.path_length = strlen(closed_trace_chunk_path) + 1;
2117 ret = lttcomm_send_unix_sock(sock, &reply, sizeof(reply));
2118 if (ret != sizeof(reply)) {
2119 goto error_fatal;
2120 }
2121 ret = lttcomm_send_unix_sock(sock, closed_trace_chunk_path,
2122 reply.path_length);
2123 if (ret != reply.path_length) {
2124 goto error_fatal;
2125 }
2126 goto end_nosignal;
a1ae2ea5 2127 }
d2956687 2128 case LTTNG_CONSUMER_TRACE_CHUNK_EXISTS:
3654ed19 2129 {
d2956687
JG
2130 const uint64_t relayd_id =
2131 msg.u.trace_chunk_exists.relayd_id.value;
2132
2133 ret_code = lttng_consumer_trace_chunk_exists(
2134 msg.u.trace_chunk_exists.relayd_id.is_set ?
2135 &relayd_id : NULL,
2136 msg.u.trace_chunk_exists.session_id,
2137 msg.u.trace_chunk_exists.chunk_id);
2138 goto end_msg_sessiond;
3654ed19 2139 }
3bd1e081
MD
2140 default:
2141 break;
2142 }
3f8e211f 2143
3bd1e081 2144end_nosignal:
4cbc1a04
DG
2145 /*
2146 * Return 1 to indicate success since the 0 value can be a socket
2147 * shutdown during the recv() or send() call.
2148 */
f3a1fc7b
JG
2149 ret = 1;
2150 goto end;
ffe60014
DG
2151
2152end_msg_sessiond:
2153 /*
2154 * The returned value here is not useful since either way we'll return 1 to
2155 * the caller because the session daemon socket management is done
2156 * elsewhere. Returning a negative code or 0 will shutdown the consumer.
2157 */
489f70e9
MD
2158 ret = consumer_send_status_msg(sock, ret_code);
2159 if (ret < 0) {
2160 goto error_fatal;
2161 }
f3a1fc7b
JG
2162 ret = 1;
2163 goto end;
9ce5646a 2164
ffe60014
DG
2165end_channel_error:
2166 if (channel) {
2167 /*
2168 * Free channel here since no one has a reference to it. We don't
2169 * free after that because a stream can store this pointer.
2170 */
2171 destroy_channel(channel);
2172 }
2173 /* We have to send a status channel message indicating an error. */
2174 ret = consumer_send_status_channel(sock, NULL);
2175 if (ret < 0) {
2176 /* Stop everything if session daemon can not be notified. */
2177 goto error_fatal;
2178 }
f3a1fc7b
JG
2179 ret = 1;
2180 goto end;
9ce5646a 2181
ffe60014 2182error_fatal:
ffe60014 2183 /* This will issue a consumer stop. */
f3a1fc7b
JG
2184 ret = -1;
2185 goto end;
2186
2187end:
2188 rcu_read_unlock();
2189 health_code_update();
2190 return ret;
3bd1e081
MD
2191}
2192
ffe60014
DG
2193/*
2194 * Wrapper over the mmap() read offset from ust-ctl library. Since this can be
2195 * compiled out, we isolate it in this library.
2196 */
2197int lttng_ustctl_get_mmap_read_offset(struct lttng_consumer_stream *stream,
2198 unsigned long *off)
3bd1e081 2199{
ffe60014
DG
2200 assert(stream);
2201 assert(stream->ustream);
b5c5fc29 2202
ffe60014 2203 return ustctl_get_mmap_read_offset(stream->ustream, off);
3bd1e081
MD
2204}
2205
ffe60014
DG
2206/*
2207 * Wrapper over the mmap() read offset from ust-ctl library. Since this can be
2208 * compiled out, we isolate it in this library.
2209 */
2210void *lttng_ustctl_get_mmap_base(struct lttng_consumer_stream *stream)
d056b477 2211{
ffe60014
DG
2212 assert(stream);
2213 assert(stream->ustream);
2214
2215 return ustctl_get_mmap_base(stream->ustream);
d056b477
MD
2216}
2217
fc6d7a51
JD
2218void lttng_ustctl_flush_buffer(struct lttng_consumer_stream *stream,
2219 int producer_active)
2220{
2221 assert(stream);
2222 assert(stream->ustream);
2223
2224 ustctl_flush_buffer(stream->ustream, producer_active);
2225}
2226
ffe60014 2227/*
e9404c27 2228 * Take a snapshot for a specific stream.
ffe60014
DG
2229 *
2230 * Returns 0 on success, < 0 on error
2231 */
2232int lttng_ustconsumer_take_snapshot(struct lttng_consumer_stream *stream)
3bd1e081 2233{
ffe60014
DG
2234 assert(stream);
2235 assert(stream->ustream);
2236
2237 return ustctl_snapshot(stream->ustream);
3bd1e081
MD
2238}
2239
e9404c27
JG
2240/*
2241 * Sample consumed and produced positions for a specific stream.
2242 *
2243 * Returns 0 on success, < 0 on error.
2244 */
2245int lttng_ustconsumer_sample_snapshot_positions(
2246 struct lttng_consumer_stream *stream)
2247{
2248 assert(stream);
2249 assert(stream->ustream);
2250
2251 return ustctl_snapshot_sample_positions(stream->ustream);
2252}
2253
ffe60014
DG
2254/*
2255 * Get the produced position
2256 *
2257 * Returns 0 on success, < 0 on error
2258 */
2259int lttng_ustconsumer_get_produced_snapshot(
2260 struct lttng_consumer_stream *stream, unsigned long *pos)
3bd1e081 2261{
ffe60014
DG
2262 assert(stream);
2263 assert(stream->ustream);
2264 assert(pos);
7a57cf92 2265
ffe60014
DG
2266 return ustctl_snapshot_get_produced(stream->ustream, pos);
2267}
7a57cf92 2268
10a50311
JD
2269/*
2270 * Get the consumed position
2271 *
2272 * Returns 0 on success, < 0 on error
2273 */
2274int lttng_ustconsumer_get_consumed_snapshot(
2275 struct lttng_consumer_stream *stream, unsigned long *pos)
2276{
2277 assert(stream);
2278 assert(stream->ustream);
2279 assert(pos);
2280
2281 return ustctl_snapshot_get_consumed(stream->ustream, pos);
2282}
2283
84a182ce
DG
2284void lttng_ustconsumer_flush_buffer(struct lttng_consumer_stream *stream,
2285 int producer)
2286{
2287 assert(stream);
2288 assert(stream->ustream);
2289
2290 ustctl_flush_buffer(stream->ustream, producer);
2291}
2292
214f70e0
JR
2293void lttng_ustconsumer_clear_buffer(struct lttng_consumer_stream *stream)
2294{
2295 assert(stream);
2296 assert(stream->ustream);
2297
2298 ustctl_clear_buffer(stream->ustream);
2299}
2300
84a182ce
DG
2301int lttng_ustconsumer_get_current_timestamp(
2302 struct lttng_consumer_stream *stream, uint64_t *ts)
2303{
2304 assert(stream);
2305 assert(stream->ustream);
2306 assert(ts);
2307
2308 return ustctl_get_current_timestamp(stream->ustream, ts);
2309}
2310
fb83fe64
JD
2311int lttng_ustconsumer_get_sequence_number(
2312 struct lttng_consumer_stream *stream, uint64_t *seq)
2313{
2314 assert(stream);
2315 assert(stream->ustream);
2316 assert(seq);
2317
2318 return ustctl_get_sequence_number(stream->ustream, seq);
2319}
2320
ffe60014 2321/*
0dd01979 2322 * Called when the stream signals the consumer that it has hung up.
ffe60014
DG
2323 */
2324void lttng_ustconsumer_on_stream_hangup(struct lttng_consumer_stream *stream)
2325{
2326 assert(stream);
2327 assert(stream->ustream);
2c1dd183 2328
0dd01979
MD
2329 pthread_mutex_lock(&stream->lock);
2330 if (!stream->quiescent) {
2331 ustctl_flush_buffer(stream->ustream, 0);
2332 stream->quiescent = true;
2333 }
2334 pthread_mutex_unlock(&stream->lock);
ffe60014
DG
2335 stream->hangup_flush_done = 1;
2336}
ee77a7b0 2337
ffe60014
DG
2338void lttng_ustconsumer_del_channel(struct lttng_consumer_channel *chan)
2339{
4628484a
MD
2340 int i;
2341
ffe60014
DG
2342 assert(chan);
2343 assert(chan->uchan);
d2956687 2344 assert(chan->buffer_credentials.is_set);
e316aad5 2345
ea88ca2a
MD
2346 if (chan->switch_timer_enabled == 1) {
2347 consumer_timer_switch_stop(chan);
2348 }
4628484a
MD
2349 for (i = 0; i < chan->nr_stream_fds; i++) {
2350 int ret;
2351
2352 ret = close(chan->stream_fds[i]);
2353 if (ret) {
2354 PERROR("close");
2355 }
2356 if (chan->shm_path[0]) {
2357 char shm_path[PATH_MAX];
2358
2359 ret = get_stream_shm_path(shm_path, chan->shm_path, i);
2360 if (ret) {
2361 ERR("Cannot get stream shm path");
2362 }
d2956687
JG
2363 ret = run_as_unlink(shm_path,
2364 chan->buffer_credentials.value.uid,
2365 chan->buffer_credentials.value.gid);
4628484a 2366 if (ret) {
4628484a
MD
2367 PERROR("unlink %s", shm_path);
2368 }
2369 }
2370 }
3bd1e081
MD
2371}
2372
b83e03c4
MD
2373void lttng_ustconsumer_free_channel(struct lttng_consumer_channel *chan)
2374{
2375 assert(chan);
2376 assert(chan->uchan);
d2956687 2377 assert(chan->buffer_credentials.is_set);
b83e03c4
MD
2378
2379 consumer_metadata_cache_destroy(chan);
2380 ustctl_destroy_channel(chan->uchan);
ea853771
JR
2381 /* Try to rmdir all directories under shm_path root. */
2382 if (chan->root_shm_path[0]) {
602766ec 2383 (void) run_as_rmdir_recursive(chan->root_shm_path,
d2956687 2384 chan->buffer_credentials.value.uid,
f75c5439
MD
2385 chan->buffer_credentials.value.gid,
2386 LTTNG_DIRECTORY_HANDLE_SKIP_NON_EMPTY_FLAG);
ea853771 2387 }
b83e03c4
MD
2388 free(chan->stream_fds);
2389}
2390
3bd1e081
MD
2391void lttng_ustconsumer_del_stream(struct lttng_consumer_stream *stream)
2392{
ffe60014
DG
2393 assert(stream);
2394 assert(stream->ustream);
d41f73b7 2395
ea88ca2a
MD
2396 if (stream->chan->switch_timer_enabled == 1) {
2397 consumer_timer_switch_stop(stream->chan);
2398 }
ffe60014
DG
2399 ustctl_destroy_stream(stream->ustream);
2400}
d41f73b7 2401
6d574024
DG
2402int lttng_ustconsumer_get_wakeup_fd(struct lttng_consumer_stream *stream)
2403{
2404 assert(stream);
2405 assert(stream->ustream);
2406
2407 return ustctl_stream_get_wakeup_fd(stream->ustream);
2408}
2409
2410int lttng_ustconsumer_close_wakeup_fd(struct lttng_consumer_stream *stream)
2411{
2412 assert(stream);
2413 assert(stream->ustream);
2414
2415 return ustctl_stream_close_wakeup_fd(stream->ustream);
2416}
2417
309167d2
JD
2418/*
2419 * Populate index values of a UST stream. Values are set in big endian order.
2420 *
2421 * Return 0 on success or else a negative value.
2422 */
50adc264 2423static int get_index_values(struct ctf_packet_index *index,
309167d2
JD
2424 struct ustctl_consumer_stream *ustream)
2425{
2426 int ret;
85d1db9f
JG
2427 uint64_t packet_size, content_size, timestamp_begin, timestamp_end,
2428 events_discarded, stream_id, stream_instance_id,
2429 packet_seq_num;
309167d2 2430
85d1db9f 2431 ret = ustctl_get_timestamp_begin(ustream, &timestamp_begin);
309167d2
JD
2432 if (ret < 0) {
2433 PERROR("ustctl_get_timestamp_begin");
2434 goto error;
2435 }
309167d2 2436
85d1db9f 2437 ret = ustctl_get_timestamp_end(ustream, &timestamp_end);
309167d2
JD
2438 if (ret < 0) {
2439 PERROR("ustctl_get_timestamp_end");
2440 goto error;
2441 }
309167d2 2442
85d1db9f 2443 ret = ustctl_get_events_discarded(ustream, &events_discarded);
309167d2
JD
2444 if (ret < 0) {
2445 PERROR("ustctl_get_events_discarded");
2446 goto error;
2447 }
309167d2 2448
85d1db9f 2449 ret = ustctl_get_content_size(ustream, &content_size);
309167d2
JD
2450 if (ret < 0) {
2451 PERROR("ustctl_get_content_size");
2452 goto error;
2453 }
309167d2 2454
85d1db9f 2455 ret = ustctl_get_packet_size(ustream, &packet_size);
309167d2
JD
2456 if (ret < 0) {
2457 PERROR("ustctl_get_packet_size");
2458 goto error;
2459 }
309167d2 2460
85d1db9f 2461 ret = ustctl_get_stream_id(ustream, &stream_id);
309167d2
JD
2462 if (ret < 0) {
2463 PERROR("ustctl_get_stream_id");
2464 goto error;
2465 }
309167d2 2466
85d1db9f 2467 ret = ustctl_get_instance_id(ustream, &stream_instance_id);
234cd636
JD
2468 if (ret < 0) {
2469 PERROR("ustctl_get_instance_id");
2470 goto error;
2471 }
234cd636 2472
85d1db9f 2473 ret = ustctl_get_sequence_number(ustream, &packet_seq_num);
234cd636
JD
2474 if (ret < 0) {
2475 PERROR("ustctl_get_sequence_number");
2476 goto error;
2477 }
85d1db9f
JG
2478
2479 *index = (typeof(*index)) {
2480 .offset = index->offset,
2481 .packet_size = htobe64(packet_size),
2482 .content_size = htobe64(content_size),
2483 .timestamp_begin = htobe64(timestamp_begin),
2484 .timestamp_end = htobe64(timestamp_end),
2485 .events_discarded = htobe64(events_discarded),
2486 .stream_id = htobe64(stream_id),
2487 .stream_instance_id = htobe64(stream_instance_id),
2488 .packet_seq_num = htobe64(packet_seq_num),
2489 };
234cd636 2490
309167d2
JD
2491error:
2492 return ret;
2493}
2494
93ec662e
JD
2495static
2496void metadata_stream_reset_cache(struct lttng_consumer_stream *stream,
2497 struct consumer_metadata_cache *cache)
2498{
2499 DBG("Metadata stream update to version %" PRIu64,
2500 cache->version);
2501 stream->ust_metadata_pushed = 0;
2502 stream->metadata_version = cache->version;
2503 stream->reset_metadata_flag = 1;
2504}
2505
2506/*
2507 * Check if the version of the metadata stream and metadata cache match.
2508 * If the cache got updated, reset the metadata stream.
2509 * The stream lock and metadata cache lock MUST be held.
2510 * Return 0 on success, a negative value on error.
2511 */
2512static
2513int metadata_stream_check_version(struct lttng_consumer_stream *stream)
2514{
2515 int ret = 0;
2516 struct consumer_metadata_cache *cache = stream->chan->metadata_cache;
2517
2518 if (cache->version == stream->metadata_version) {
2519 goto end;
2520 }
2521 metadata_stream_reset_cache(stream, cache);
2522
2523end:
2524 return ret;
2525}
2526
94d49140
JD
2527/*
2528 * Write up to one packet from the metadata cache to the channel.
2529 *
2530 * Returns the number of bytes pushed in the cache, or a negative value
2531 * on error.
2532 */
2533static
2534int commit_one_metadata_packet(struct lttng_consumer_stream *stream)
2535{
2536 ssize_t write_len;
2537 int ret;
2538
2539 pthread_mutex_lock(&stream->chan->metadata_cache->lock);
93ec662e
JD
2540 ret = metadata_stream_check_version(stream);
2541 if (ret < 0) {
2542 goto end;
2543 }
c585821b 2544 if (stream->chan->metadata_cache->max_offset
94d49140
JD
2545 == stream->ust_metadata_pushed) {
2546 ret = 0;
2547 goto end;
2548 }
2549
2550 write_len = ustctl_write_one_packet_to_channel(stream->chan->uchan,
2551 &stream->chan->metadata_cache->data[stream->ust_metadata_pushed],
c585821b 2552 stream->chan->metadata_cache->max_offset
94d49140
JD
2553 - stream->ust_metadata_pushed);
2554 assert(write_len != 0);
2555 if (write_len < 0) {
2556 ERR("Writing one metadata packet");
2557 ret = -1;
2558 goto end;
2559 }
2560 stream->ust_metadata_pushed += write_len;
2561
c585821b 2562 assert(stream->chan->metadata_cache->max_offset >=
94d49140
JD
2563 stream->ust_metadata_pushed);
2564 ret = write_len;
2565
0d88e046
JG
2566 /*
2567 * Switch packet (but don't open the next one) on every commit of
2568 * a metadata packet. Since the subbuffer is fully filled (with padding,
2569 * if needed), the stream is "quiescent" after this commit.
2570 */
2571 ustctl_flush_buffer(stream->ustream, 1);
2572 stream->quiescent = true;
94d49140
JD
2573end:
2574 pthread_mutex_unlock(&stream->chan->metadata_cache->lock);
2575 return ret;
2576}
2577
309167d2 2578
94d49140
JD
2579/*
2580 * Sync metadata meaning request them to the session daemon and snapshot to the
2581 * metadata thread can consumer them.
2582 *
c585821b
MD
2583 * Metadata stream lock is held here, but we need to release it when
2584 * interacting with sessiond, else we cause a deadlock with live
2585 * awaiting on metadata to be pushed out.
94d49140 2586 *
cdb72e4e
JG
2587 * The RCU read side lock must be held by the caller.
2588 *
94d49140
JD
2589 * Return 0 if new metadatda is available, EAGAIN if the metadata stream
2590 * is empty or a negative value on error.
2591 */
2592int lttng_ustconsumer_sync_metadata(struct lttng_consumer_local_data *ctx,
cdb72e4e 2593 struct lttng_consumer_stream *metadata_stream)
94d49140
JD
2594{
2595 int ret;
2596 int retry = 0;
cdb72e4e 2597 struct lttng_consumer_channel *metadata_channel;
94d49140
JD
2598
2599 assert(ctx);
cdb72e4e 2600 assert(metadata_stream);
94d49140 2601
cdb72e4e
JG
2602 metadata_channel = metadata_stream->chan;
2603 pthread_mutex_unlock(&metadata_stream->lock);
94d49140
JD
2604 /*
2605 * Request metadata from the sessiond, but don't wait for the flush
2606 * because we locked the metadata thread.
2607 */
cdb72e4e
JG
2608 ret = lttng_ustconsumer_request_metadata(ctx, metadata_channel, 0, 0);
2609 pthread_mutex_lock(&metadata_stream->lock);
94d49140
JD
2610 if (ret < 0) {
2611 goto end;
2612 }
2613
cdb72e4e
JG
2614 /*
2615 * The metadata stream and channel can be deleted while the
2616 * metadata stream lock was released. The streamed is checked
2617 * for deletion before we use it further.
2618 *
2619 * Note that it is safe to access a logically-deleted stream since its
2620 * existence is still guaranteed by the RCU read side lock. However,
2621 * it should no longer be used. The close/deletion of the metadata
2622 * channel and stream already guarantees that all metadata has been
2623 * consumed. Therefore, there is nothing left to do in this function.
2624 */
2625 if (consumer_stream_is_deleted(metadata_stream)) {
2626 DBG("Metadata stream %" PRIu64 " was deleted during the metadata synchronization",
2627 metadata_stream->key);
2628 ret = 0;
2629 goto end;
2630 }
2631
2632 ret = commit_one_metadata_packet(metadata_stream);
94d49140
JD
2633 if (ret <= 0) {
2634 goto end;
2635 } else if (ret > 0) {
2636 retry = 1;
2637 }
2638
cdb72e4e 2639 ret = ustctl_snapshot(metadata_stream->ustream);
94d49140
JD
2640 if (ret < 0) {
2641 if (errno != EAGAIN) {
2642 ERR("Sync metadata, taking UST snapshot");
2643 goto end;
2644 }
2645 DBG("No new metadata when syncing them.");
2646 /* No new metadata, exit. */
2647 ret = ENODATA;
2648 goto end;
2649 }
2650
2651 /*
2652 * After this flush, we still need to extract metadata.
2653 */
2654 if (retry) {
2655 ret = EAGAIN;
2656 }
2657
2658end:
2659 return ret;
2660}
2661
02b3d176
DG
2662/*
2663 * Return 0 on success else a negative value.
2664 */
2665static int notify_if_more_data(struct lttng_consumer_stream *stream,
2666 struct lttng_consumer_local_data *ctx)
2667{
2668 int ret;
2669 struct ustctl_consumer_stream *ustream;
2670
2671 assert(stream);
2672 assert(ctx);
2673
2674 ustream = stream->ustream;
2675
2676 /*
2677 * First, we are going to check if there is a new subbuffer available
2678 * before reading the stream wait_fd.
2679 */
2680 /* Get the next subbuffer */
2681 ret = ustctl_get_next_subbuf(ustream);
2682 if (ret) {
2683 /* No more data found, flag the stream. */
2684 stream->has_data = 0;
2685 ret = 0;
2686 goto end;
2687 }
2688
5420e5db 2689 ret = ustctl_put_subbuf(ustream);
02b3d176
DG
2690 assert(!ret);
2691
2692 /* This stream still has data. Flag it and wake up the data thread. */
2693 stream->has_data = 1;
2694
2695 if (stream->monitor && !stream->hangup_flush_done && !ctx->has_wakeup) {
2696 ssize_t writelen;
2697
2698 writelen = lttng_pipe_write(ctx->consumer_wakeup_pipe, "!", 1);
2699 if (writelen < 0 && errno != EAGAIN && errno != EWOULDBLOCK) {
2700 ret = writelen;
2701 goto end;
2702 }
2703
2704 /* The wake up pipe has been notified. */
2705 ctx->has_wakeup = 1;
2706 }
2707 ret = 0;
2708
2709end:
2710 return ret;
2711}
2712
fb83fe64
JD
2713static
2714int update_stream_stats(struct lttng_consumer_stream *stream)
2715{
2716 int ret;
2717 uint64_t seq, discarded;
2718
2719 ret = ustctl_get_sequence_number(stream->ustream, &seq);
2720 if (ret < 0) {
2721 PERROR("ustctl_get_sequence_number");
2722 goto end;
2723 }
2724 /*
2725 * Start the sequence when we extract the first packet in case we don't
2726 * start at 0 (for example if a consumer is not connected to the
2727 * session immediately after the beginning).
2728 */
2729 if (stream->last_sequence_number == -1ULL) {
2730 stream->last_sequence_number = seq;
2731 } else if (seq > stream->last_sequence_number) {
2732 stream->chan->lost_packets += seq -
2733 stream->last_sequence_number - 1;
2734 } else {
2735 /* seq <= last_sequence_number */
2736 ERR("Sequence number inconsistent : prev = %" PRIu64
2737 ", current = %" PRIu64,
2738 stream->last_sequence_number, seq);
2739 ret = -1;
2740 goto end;
2741 }
2742 stream->last_sequence_number = seq;
2743
2744 ret = ustctl_get_events_discarded(stream->ustream, &discarded);
2745 if (ret < 0) {
2746 PERROR("kernctl_get_events_discarded");
2747 goto end;
2748 }
2749 if (discarded < stream->last_discarded_events) {
2750 /*
83f4233d
MJ
2751 * Overflow has occurred. We assume only one wrap-around
2752 * has occurred.
fb83fe64
JD
2753 */
2754 stream->chan->discarded_events +=
2755 (1ULL << (CAA_BITS_PER_LONG - 1)) -
2756 stream->last_discarded_events + discarded;
2757 } else {
2758 stream->chan->discarded_events += discarded -
2759 stream->last_discarded_events;
2760 }
2761 stream->last_discarded_events = discarded;
2762 ret = 0;
2763
2764end:
2765 return ret;
2766}
2767
94d49140
JD
2768/*
2769 * Read subbuffer from the given stream.
2770 *
d2956687 2771 * Stream and channel locks MUST be acquired by the caller.
94d49140
JD
2772 *
2773 * Return 0 on success else a negative value.
2774 */
d41f73b7 2775int lttng_ustconsumer_read_subbuffer(struct lttng_consumer_stream *stream,
d2956687 2776 struct lttng_consumer_local_data *ctx)
d41f73b7 2777{
1d4dfdef 2778 unsigned long len, subbuf_size, padding;
02d02e31 2779 int err, write_index = 1, rotation_ret;
d41f73b7 2780 long ret = 0;
ffe60014 2781 struct ustctl_consumer_stream *ustream;
50adc264 2782 struct ctf_packet_index index;
ffe60014
DG
2783
2784 assert(stream);
2785 assert(stream->ustream);
2786 assert(ctx);
d41f73b7 2787
3eb914c0 2788 DBG("In UST read_subbuffer (wait_fd: %d, name: %s)", stream->wait_fd,
ffe60014
DG
2789 stream->name);
2790
2791 /* Ease our life for what's next. */
2792 ustream = stream->ustream;
d41f73b7 2793
6cd525e8 2794 /*
02b3d176
DG
2795 * We can consume the 1 byte written into the wait_fd by UST. Don't trigger
2796 * error if we cannot read this one byte (read returns 0), or if the error
2797 * is EAGAIN or EWOULDBLOCK.
2798 *
2799 * This is only done when the stream is monitored by a thread, before the
2800 * flush is done after a hangup and if the stream is not flagged with data
2801 * since there might be nothing to consume in the wait fd but still have
2802 * data available flagged by the consumer wake up pipe.
6cd525e8 2803 */
02b3d176
DG
2804 if (stream->monitor && !stream->hangup_flush_done && !stream->has_data) {
2805 char dummy;
c617c0c6
MD
2806 ssize_t readlen;
2807
6cd525e8
MD
2808 readlen = lttng_read(stream->wait_fd, &dummy, 1);
2809 if (readlen < 0 && errno != EAGAIN && errno != EWOULDBLOCK) {
effcf122 2810 ret = readlen;
02d02e31
JD
2811 goto error;
2812 }
2813 }
2814
2815 /*
2816 * If the stream was flagged to be ready for rotation before we extract the
2817 * next packet, rotate it now.
2818 */
2819 if (stream->rotate_ready) {
2820 DBG("Rotate stream before extracting data");
d2956687 2821 rotation_ret = lttng_consumer_rotate_stream(ctx, stream);
02d02e31
JD
2822 if (rotation_ret < 0) {
2823 ERR("Stream rotation error");
2824 ret = -1;
2825 goto error;
effcf122 2826 }
d41f73b7
MD
2827 }
2828
04ef1097 2829retry:
d41f73b7 2830 /* Get the next subbuffer */
ffe60014 2831 err = ustctl_get_next_subbuf(ustream);
d41f73b7 2832 if (err != 0) {
04ef1097
MD
2833 /*
2834 * Populate metadata info if the existing info has
2835 * already been read.
2836 */
2837 if (stream->metadata_flag) {
94d49140
JD
2838 ret = commit_one_metadata_packet(stream);
2839 if (ret <= 0) {
02d02e31 2840 goto error;
04ef1097 2841 }
04ef1097
MD
2842 goto retry;
2843 }
2844
1d4dfdef 2845 ret = err; /* ustctl_get_next_subbuf returns negative, caller expect positive. */
d41f73b7
MD
2846 /*
2847 * This is a debug message even for single-threaded consumer,
2848 * because poll() have more relaxed criterions than get subbuf,
2849 * so get_subbuf may fail for short race windows where poll()
2850 * would issue wakeups.
2851 */
2852 DBG("Reserving sub buffer failed (everything is normal, "
ffe60014 2853 "it is due to concurrency) [ret: %d]", err);
02d02e31 2854 goto error;
d41f73b7 2855 }
ffe60014 2856 assert(stream->chan->output == CONSUMER_CHANNEL_MMAP);
309167d2 2857
1c20f0e2 2858 if (!stream->metadata_flag) {
309167d2
JD
2859 index.offset = htobe64(stream->out_fd_offset);
2860 ret = get_index_values(&index, ustream);
2861 if (ret < 0) {
7b87473d
MD
2862 err = ustctl_put_subbuf(ustream);
2863 assert(err == 0);
02d02e31 2864 goto error;
309167d2 2865 }
fb83fe64
JD
2866
2867 /* Update the stream's sequence and discarded events count. */
2868 ret = update_stream_stats(stream);
2869 if (ret < 0) {
2870 PERROR("kernctl_get_events_discarded");
7b87473d
MD
2871 err = ustctl_put_subbuf(ustream);
2872 assert(err == 0);
02d02e31 2873 goto error;
fb83fe64 2874 }
1c20f0e2
JD
2875 } else {
2876 write_index = 0;
309167d2
JD
2877 }
2878
1d4dfdef 2879 /* Get the full padded subbuffer size */
ffe60014 2880 err = ustctl_get_padded_subbuf_size(ustream, &len);
effcf122 2881 assert(err == 0);
1d4dfdef
DG
2882
2883 /* Get subbuffer data size (without padding) */
ffe60014 2884 err = ustctl_get_subbuf_size(ustream, &subbuf_size);
1d4dfdef
DG
2885 assert(err == 0);
2886
2887 /* Make sure we don't get a subbuffer size bigger than the padded */
2888 assert(len >= subbuf_size);
2889
2890 padding = len - subbuf_size;
02d02e31 2891
d41f73b7 2892 /* write the subbuffer to the tracefile */
309167d2 2893 ret = lttng_consumer_on_read_subbuffer_mmap(ctx, stream, subbuf_size, padding, &index);
91dfef6e
DG
2894 /*
2895 * The mmap operation should write subbuf_size amount of data when network
2896 * streaming or the full padding (len) size when we are _not_ streaming.
2897 */
d88aee68
DG
2898 if ((ret != subbuf_size && stream->net_seq_idx != (uint64_t) -1ULL) ||
2899 (ret != len && stream->net_seq_idx == (uint64_t) -1ULL)) {
d41f73b7 2900 /*
91dfef6e 2901 * Display the error but continue processing to try to release the
c5c45efa
DG
2902 * subbuffer. This is a DBG statement since any unexpected kill or
2903 * signal, the application gets unregistered, relayd gets closed or
2904 * anything that affects the buffer lifetime will trigger this error.
2905 * So, for the sake of the user, don't print this error since it can
2906 * happen and it is OK with the code flow.
d41f73b7 2907 */
c5c45efa 2908 DBG("Error writing to tracefile "
8fd623e0 2909 "(ret: %ld != len: %lu != subbuf_size: %lu)",
91dfef6e 2910 ret, len, subbuf_size);
309167d2 2911 write_index = 0;
d41f73b7 2912 }
ffe60014 2913 err = ustctl_put_next_subbuf(ustream);
effcf122 2914 assert(err == 0);
331744e3 2915
02b3d176
DG
2916 /*
2917 * This will consumer the byte on the wait_fd if and only if there is not
2918 * next subbuffer to be acquired.
2919 */
2920 if (!stream->metadata_flag) {
2921 ret = notify_if_more_data(stream, ctx);
2922 if (ret < 0) {
02d02e31 2923 goto error;
02b3d176
DG
2924 }
2925 }
2926
309167d2 2927 /* Write index if needed. */
1c20f0e2 2928 if (!write_index) {
02d02e31 2929 goto rotate;
1c20f0e2
JD
2930 }
2931
94d49140
JD
2932 if (stream->chan->live_timer_interval && !stream->metadata_flag) {
2933 /*
2934 * In live, block until all the metadata is sent.
2935 */
c585821b
MD
2936 pthread_mutex_lock(&stream->metadata_timer_lock);
2937 assert(!stream->missed_metadata_flush);
2938 stream->waiting_on_metadata = true;
2939 pthread_mutex_unlock(&stream->metadata_timer_lock);
2940
94d49140 2941 err = consumer_stream_sync_metadata(ctx, stream->session_id);
c585821b
MD
2942
2943 pthread_mutex_lock(&stream->metadata_timer_lock);
2944 stream->waiting_on_metadata = false;
2945 if (stream->missed_metadata_flush) {
2946 stream->missed_metadata_flush = false;
2947 pthread_mutex_unlock(&stream->metadata_timer_lock);
2948 (void) consumer_flush_ust_index(stream);
2949 } else {
2950 pthread_mutex_unlock(&stream->metadata_timer_lock);
2951 }
2952
94d49140 2953 if (err < 0) {
02d02e31 2954 goto error;
94d49140
JD
2955 }
2956 }
2957
1c20f0e2
JD
2958 assert(!stream->metadata_flag);
2959 err = consumer_stream_write_index(stream, &index);
2960 if (err < 0) {
02d02e31 2961 goto error;
309167d2
JD
2962 }
2963
02d02e31
JD
2964rotate:
2965 /*
2966 * After extracting the packet, we check if the stream is now ready to be
2967 * rotated and perform the action immediately.
2968 */
2969 rotation_ret = lttng_consumer_stream_is_rotate_ready(stream);
2970 if (rotation_ret == 1) {
d2956687 2971 rotation_ret = lttng_consumer_rotate_stream(ctx, stream);
02d02e31
JD
2972 if (rotation_ret < 0) {
2973 ERR("Stream rotation error");
2974 ret = -1;
2975 goto error;
2976 }
2977 } else if (rotation_ret < 0) {
2978 ERR("Checking if stream is ready to rotate");
2979 ret = -1;
2980 goto error;
2981 }
2982error:
d41f73b7
MD
2983 return ret;
2984}
2985
ffe60014
DG
2986/*
2987 * Called when a stream is created.
fe4477ee
JD
2988 *
2989 * Return 0 on success or else a negative value.
ffe60014 2990 */
d41f73b7
MD
2991int lttng_ustconsumer_on_recv_stream(struct lttng_consumer_stream *stream)
2992{
fe4477ee
JD
2993 int ret;
2994
10a50311
JD
2995 assert(stream);
2996
d2956687
JG
2997 /*
2998 * Don't create anything if this is set for streaming or if there is
2999 * no current trace chunk on the parent channel.
3000 */
3001 if (stream->net_seq_idx == (uint64_t) -1ULL && stream->chan->monitor &&
3002 stream->chan->trace_chunk) {
3003 ret = consumer_stream_create_output_files(stream, true);
3004 if (ret) {
fe4477ee
JD
3005 goto error;
3006 }
fe4477ee
JD
3007 }
3008 ret = 0;
3009
3010error:
3011 return ret;
d41f73b7 3012}
ca22feea
DG
3013
3014/*
3015 * Check if data is still being extracted from the buffers for a specific
4e9a4686
DG
3016 * stream. Consumer data lock MUST be acquired before calling this function
3017 * and the stream lock.
ca22feea 3018 *
6d805429 3019 * Return 1 if the traced data are still getting read else 0 meaning that the
ca22feea
DG
3020 * data is available for trace viewer reading.
3021 */
6d805429 3022int lttng_ustconsumer_data_pending(struct lttng_consumer_stream *stream)
ca22feea
DG
3023{
3024 int ret;
3025
3026 assert(stream);
ffe60014 3027 assert(stream->ustream);
ca22feea 3028
6d805429 3029 DBG("UST consumer checking data pending");
c8f59ee5 3030
ca6b395f
MD
3031 if (stream->endpoint_status != CONSUMER_ENDPOINT_ACTIVE) {
3032 ret = 0;
3033 goto end;
3034 }
3035
04ef1097 3036 if (stream->chan->type == CONSUMER_CHANNEL_TYPE_METADATA) {
e6ee4eab
DG
3037 uint64_t contiguous, pushed;
3038
3039 /* Ease our life a bit. */
c585821b 3040 contiguous = stream->chan->metadata_cache->max_offset;
e6ee4eab
DG
3041 pushed = stream->ust_metadata_pushed;
3042
04ef1097
MD
3043 /*
3044 * We can simply check whether all contiguously available data
3045 * has been pushed to the ring buffer, since the push operation
3046 * is performed within get_next_subbuf(), and because both
3047 * get_next_subbuf() and put_next_subbuf() are issued atomically
3048 * thanks to the stream lock within
3049 * lttng_ustconsumer_read_subbuffer(). This basically means that
3050 * whetnever ust_metadata_pushed is incremented, the associated
3051 * metadata has been consumed from the metadata stream.
3052 */
3053 DBG("UST consumer metadata pending check: contiguous %" PRIu64 " vs pushed %" PRIu64,
e6ee4eab 3054 contiguous, pushed);
aa01b94c 3055 assert(((int64_t) (contiguous - pushed)) >= 0);
e6ee4eab 3056 if ((contiguous != pushed) ||
6acdf328 3057 (((int64_t) contiguous - pushed) > 0 || contiguous == 0)) {
04ef1097
MD
3058 ret = 1; /* Data is pending */
3059 goto end;
3060 }
3061 } else {
3062 ret = ustctl_get_next_subbuf(stream->ustream);
3063 if (ret == 0) {
3064 /*
3065 * There is still data so let's put back this
3066 * subbuffer.
3067 */
3068 ret = ustctl_put_subbuf(stream->ustream);
3069 assert(ret == 0);
3070 ret = 1; /* Data is pending */
3071 goto end;
3072 }
ca22feea
DG
3073 }
3074
6d805429
DG
3075 /* Data is NOT pending so ready to be read. */
3076 ret = 0;
ca22feea 3077
6efae65e
DG
3078end:
3079 return ret;
ca22feea 3080}
d88aee68 3081
6d574024
DG
3082/*
3083 * Stop a given metadata channel timer if enabled and close the wait fd which
3084 * is the poll pipe of the metadata stream.
3085 *
d2c82a5a 3086 * This MUST be called with the metadata channel lock acquired.
6d574024
DG
3087 */
3088void lttng_ustconsumer_close_metadata(struct lttng_consumer_channel *metadata)
3089{
3090 int ret;
3091
3092 assert(metadata);
3093 assert(metadata->type == CONSUMER_CHANNEL_TYPE_METADATA);
3094
3095 DBG("Closing metadata channel key %" PRIu64, metadata->key);
3096
3097 if (metadata->switch_timer_enabled == 1) {
3098 consumer_timer_switch_stop(metadata);
3099 }
3100
3101 if (!metadata->metadata_stream) {
3102 goto end;
3103 }
3104
3105 /*
3106 * Closing write side so the thread monitoring the stream wakes up if any
3107 * and clean the metadata stream.
3108 */
3109 if (metadata->metadata_stream->ust_metadata_poll_pipe[1] >= 0) {
3110 ret = close(metadata->metadata_stream->ust_metadata_poll_pipe[1]);
3111 if (ret < 0) {
3112 PERROR("closing metadata pipe write side");
3113 }
3114 metadata->metadata_stream->ust_metadata_poll_pipe[1] = -1;
3115 }
3116
3117end:
3118 return;
3119}
3120
d88aee68
DG
3121/*
3122 * Close every metadata stream wait fd of the metadata hash table. This
3123 * function MUST be used very carefully so not to run into a race between the
3124 * metadata thread handling streams and this function closing their wait fd.
3125 *
3126 * For UST, this is used when the session daemon hangs up. Its the metadata
3127 * producer so calling this is safe because we are assured that no state change
3128 * can occur in the metadata thread for the streams in the hash table.
3129 */
6d574024 3130void lttng_ustconsumer_close_all_metadata(struct lttng_ht *metadata_ht)
d88aee68 3131{
d88aee68
DG
3132 struct lttng_ht_iter iter;
3133 struct lttng_consumer_stream *stream;
3134
3135 assert(metadata_ht);
3136 assert(metadata_ht->ht);
3137
3138 DBG("UST consumer closing all metadata streams");
3139
3140 rcu_read_lock();
3141 cds_lfht_for_each_entry(metadata_ht->ht, &iter.iter, stream,
3142 node.node) {
9ce5646a
MD
3143
3144 health_code_update();
3145
be2b50c7 3146 pthread_mutex_lock(&stream->chan->lock);
6d574024 3147 lttng_ustconsumer_close_metadata(stream->chan);
be2b50c7
DG
3148 pthread_mutex_unlock(&stream->chan->lock);
3149
d88aee68
DG
3150 }
3151 rcu_read_unlock();
3152}
d8ef542d
MD
3153
3154void lttng_ustconsumer_close_stream_wakeup(struct lttng_consumer_stream *stream)
3155{
3156 int ret;
3157
3158 ret = ustctl_stream_close_wakeup_fd(stream->ustream);
3159 if (ret < 0) {
3160 ERR("Unable to close wakeup fd");
3161 }
3162}
331744e3 3163
f666ae70
MD
3164/*
3165 * Please refer to consumer-timer.c before adding any lock within this
3166 * function or any of its callees. Timers have a very strict locking
3167 * semantic with respect to teardown. Failure to respect this semantic
3168 * introduces deadlocks.
c585821b
MD
3169 *
3170 * DON'T hold the metadata lock when calling this function, else this
3171 * can cause deadlock involving consumer awaiting for metadata to be
3172 * pushed out due to concurrent interaction with the session daemon.
f666ae70 3173 */
331744e3 3174int lttng_ustconsumer_request_metadata(struct lttng_consumer_local_data *ctx,
94d49140 3175 struct lttng_consumer_channel *channel, int timer, int wait)
331744e3
JD
3176{
3177 struct lttcomm_metadata_request_msg request;
3178 struct lttcomm_consumer_msg msg;
0c759fc9 3179 enum lttcomm_return_code ret_code = LTTCOMM_CONSUMERD_SUCCESS;
93ec662e 3180 uint64_t len, key, offset, version;
331744e3
JD
3181 int ret;
3182
3183 assert(channel);
3184 assert(channel->metadata_cache);
3185
53efb85a
MD
3186 memset(&request, 0, sizeof(request));
3187
331744e3
JD
3188 /* send the metadata request to sessiond */
3189 switch (consumer_data.type) {
3190 case LTTNG_CONSUMER64_UST:
3191 request.bits_per_long = 64;
3192 break;
3193 case LTTNG_CONSUMER32_UST:
3194 request.bits_per_long = 32;
3195 break;
3196 default:
3197 request.bits_per_long = 0;
3198 break;
3199 }
3200
3201 request.session_id = channel->session_id;
1950109e 3202 request.session_id_per_pid = channel->session_id_per_pid;
567eb353
DG
3203 /*
3204 * Request the application UID here so the metadata of that application can
3205 * be sent back. The channel UID corresponds to the user UID of the session
3206 * used for the rights on the stream file(s).
3207 */
3208 request.uid = channel->ust_app_uid;
331744e3 3209 request.key = channel->key;
567eb353 3210
1950109e 3211 DBG("Sending metadata request to sessiond, session id %" PRIu64
3e25d926 3212 ", per-pid %" PRIu64 ", app UID %u and channel key %" PRIu64,
567eb353
DG
3213 request.session_id, request.session_id_per_pid, request.uid,
3214 request.key);
331744e3 3215
75d83e50 3216 pthread_mutex_lock(&ctx->metadata_socket_lock);
9ce5646a
MD
3217
3218 health_code_update();
3219
331744e3
JD
3220 ret = lttcomm_send_unix_sock(ctx->consumer_metadata_socket, &request,
3221 sizeof(request));
3222 if (ret < 0) {
3223 ERR("Asking metadata to sessiond");
3224 goto end;
3225 }
3226
9ce5646a
MD
3227 health_code_update();
3228
331744e3
JD
3229 /* Receive the metadata from sessiond */
3230 ret = lttcomm_recv_unix_sock(ctx->consumer_metadata_socket, &msg,
3231 sizeof(msg));
3232 if (ret != sizeof(msg)) {
8fd623e0 3233 DBG("Consumer received unexpected message size %d (expects %zu)",
331744e3
JD
3234 ret, sizeof(msg));
3235 lttng_consumer_send_error(ctx, LTTCOMM_CONSUMERD_ERROR_RECV_CMD);
3236 /*
3237 * The ret value might 0 meaning an orderly shutdown but this is ok
3238 * since the caller handles this.
3239 */
3240 goto end;
3241 }
3242
9ce5646a
MD
3243 health_code_update();
3244
331744e3
JD
3245 if (msg.cmd_type == LTTNG_ERR_UND) {
3246 /* No registry found */
3247 (void) consumer_send_status_msg(ctx->consumer_metadata_socket,
3248 ret_code);
3249 ret = 0;
3250 goto end;
3251 } else if (msg.cmd_type != LTTNG_CONSUMER_PUSH_METADATA) {
3252 ERR("Unexpected cmd_type received %d", msg.cmd_type);
3253 ret = -1;
3254 goto end;
3255 }
3256
3257 len = msg.u.push_metadata.len;
3258 key = msg.u.push_metadata.key;
3259 offset = msg.u.push_metadata.target_offset;
93ec662e 3260 version = msg.u.push_metadata.version;
331744e3
JD
3261
3262 assert(key == channel->key);
3263 if (len == 0) {
3264 DBG("No new metadata to receive for key %" PRIu64, key);
3265 }
3266
9ce5646a
MD
3267 health_code_update();
3268
331744e3
JD
3269 /* Tell session daemon we are ready to receive the metadata. */
3270 ret = consumer_send_status_msg(ctx->consumer_metadata_socket,
0c759fc9 3271 LTTCOMM_CONSUMERD_SUCCESS);
331744e3
JD
3272 if (ret < 0 || len == 0) {
3273 /*
3274 * Somehow, the session daemon is not responding anymore or there is
3275 * nothing to receive.
3276 */
3277 goto end;
3278 }
3279
9ce5646a
MD
3280 health_code_update();
3281
1eb682be 3282 ret = lttng_ustconsumer_recv_metadata(ctx->consumer_metadata_socket,
93ec662e 3283 key, offset, len, version, channel, timer, wait);
1eb682be 3284 if (ret >= 0) {
f2a444f1
DG
3285 /*
3286 * Only send the status msg if the sessiond is alive meaning a positive
3287 * ret code.
3288 */
1eb682be 3289 (void) consumer_send_status_msg(ctx->consumer_metadata_socket, ret);
f2a444f1 3290 }
331744e3
JD
3291 ret = 0;
3292
3293end:
9ce5646a
MD
3294 health_code_update();
3295
75d83e50 3296 pthread_mutex_unlock(&ctx->metadata_socket_lock);
331744e3
JD
3297 return ret;
3298}
70190e1c
DG
3299
3300/*
3301 * Return the ustctl call for the get stream id.
3302 */
3303int lttng_ustconsumer_get_stream_id(struct lttng_consumer_stream *stream,
3304 uint64_t *stream_id)
3305{
3306 assert(stream);
3307 assert(stream_id);
3308
3309 return ustctl_get_stream_id(stream->ustream, stream_id);
3310}
This page took 0.265 seconds and 5 git commands to generate.