2 * Copyright (C) 2012 - David Goulet <dgoulet@efficios.com>
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License, version 2 only, as
6 * published by the Free Software Foundation.
8 * This program is distributed in the hope that it will be useful, but WITHOUT
9 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
13 * You should have received a copy of the GNU General Public License along with
14 * this program; if not, write to the Free Software Foundation, Inc., 51
15 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
20 #include <urcu/list.h>
21 #include <urcu/uatomic.h>
23 #include <common/defaults.h>
24 #include <common/common.h>
25 #include <common/sessiond-comm/sessiond-comm.h>
26 #include <common/relayd/relayd.h>
32 #include "kernel-consumer.h"
33 #include "lttng-sessiond.h"
39 * Used to keep a unique index for each relayd socket created where this value
40 * is associated with streams on the consumer so it can match the right relayd
43 * This value should be incremented atomically for safety purposes and future
44 * possible concurrent access.
46 static unsigned int relayd_net_seq_idx
;
49 * Create a session path used by list_lttng_sessions for the case that the
50 * session consumer is on the network.
52 static int build_network_session_path(char *dst
, size_t size
,
53 struct ltt_session
*session
)
55 int ret
, kdata_port
, udata_port
;
56 struct lttng_uri
*kuri
= NULL
, *uuri
= NULL
, *uri
= NULL
;
57 char tmp_uurl
[PATH_MAX
], tmp_urls
[PATH_MAX
];
62 memset(tmp_urls
, 0, sizeof(tmp_urls
));
63 memset(tmp_uurl
, 0, sizeof(tmp_uurl
));
65 kdata_port
= udata_port
= DEFAULT_NETWORK_DATA_PORT
;
67 if (session
->kernel_session
&& session
->kernel_session
->consumer
) {
68 kuri
= &session
->kernel_session
->consumer
->dst
.net
.control
;
69 kdata_port
= session
->kernel_session
->consumer
->dst
.net
.data
.port
;
72 if (session
->ust_session
&& session
->ust_session
->consumer
) {
73 uuri
= &session
->ust_session
->consumer
->dst
.net
.control
;
74 udata_port
= session
->ust_session
->consumer
->dst
.net
.data
.port
;
77 if (uuri
== NULL
&& kuri
== NULL
) {
78 uri
= &session
->consumer
->dst
.net
.control
;
79 kdata_port
= session
->consumer
->dst
.net
.data
.port
;
80 } else if (kuri
&& uuri
) {
81 ret
= uri_compare(kuri
, uuri
);
85 /* Build uuri URL string */
86 ret
= uri_to_str_url(uuri
, tmp_uurl
, sizeof(tmp_uurl
));
93 } else if (kuri
&& uuri
== NULL
) {
95 } else if (uuri
&& kuri
== NULL
) {
99 ret
= uri_to_str_url(uri
, tmp_urls
, sizeof(tmp_urls
));
105 * Do we have a UST url set. If yes, this means we have both kernel and UST
108 if (strlen(tmp_uurl
) > 0) {
109 ret
= snprintf(dst
, size
, "[K]: %s [data: %d] -- [U]: %s [data: %d]",
110 tmp_urls
, kdata_port
, tmp_uurl
, udata_port
);
116 /* No kernel URI, use the UST port. */
119 ret
= snprintf(dst
, size
, "%s [data: %d]", tmp_urls
, dport
);
127 * Fill lttng_channel array of all channels.
129 static void list_lttng_channels(int domain
, struct ltt_session
*session
,
130 struct lttng_channel
*channels
)
133 struct ltt_kernel_channel
*kchan
;
135 DBG("Listing channels for session %s", session
->name
);
138 case LTTNG_DOMAIN_KERNEL
:
139 /* Kernel channels */
140 if (session
->kernel_session
!= NULL
) {
141 cds_list_for_each_entry(kchan
,
142 &session
->kernel_session
->channel_list
.head
, list
) {
143 /* Copy lttng_channel struct to array */
144 memcpy(&channels
[i
], kchan
->channel
, sizeof(struct lttng_channel
));
145 channels
[i
].enabled
= kchan
->enabled
;
150 case LTTNG_DOMAIN_UST
:
152 struct lttng_ht_iter iter
;
153 struct ltt_ust_channel
*uchan
;
156 cds_lfht_for_each_entry(session
->ust_session
->domain_global
.channels
->ht
,
157 &iter
.iter
, uchan
, node
.node
) {
158 strncpy(channels
[i
].name
, uchan
->name
, LTTNG_SYMBOL_NAME_LEN
);
159 channels
[i
].attr
.overwrite
= uchan
->attr
.overwrite
;
160 channels
[i
].attr
.subbuf_size
= uchan
->attr
.subbuf_size
;
161 channels
[i
].attr
.num_subbuf
= uchan
->attr
.num_subbuf
;
162 channels
[i
].attr
.switch_timer_interval
=
163 uchan
->attr
.switch_timer_interval
;
164 channels
[i
].attr
.read_timer_interval
=
165 uchan
->attr
.read_timer_interval
;
166 channels
[i
].enabled
= uchan
->enabled
;
167 switch (uchan
->attr
.output
) {
170 channels
[i
].attr
.output
= LTTNG_EVENT_MMAP
;
184 * Create a list of ust global domain events.
186 static int list_lttng_ust_global_events(char *channel_name
,
187 struct ltt_ust_domain_global
*ust_global
, struct lttng_event
**events
)
190 unsigned int nb_event
= 0;
191 struct lttng_ht_iter iter
;
192 struct lttng_ht_node_str
*node
;
193 struct ltt_ust_channel
*uchan
;
194 struct ltt_ust_event
*uevent
;
195 struct lttng_event
*tmp
;
197 DBG("Listing UST global events for channel %s", channel_name
);
201 lttng_ht_lookup(ust_global
->channels
, (void *)channel_name
, &iter
);
202 node
= lttng_ht_iter_get_node_str(&iter
);
204 ret
= LTTNG_ERR_UST_CHAN_NOT_FOUND
;
208 uchan
= caa_container_of(&node
->node
, struct ltt_ust_channel
, node
.node
);
210 nb_event
+= lttng_ht_get_count(uchan
->events
);
217 DBG3("Listing UST global %d events", nb_event
);
219 tmp
= zmalloc(nb_event
* sizeof(struct lttng_event
));
221 ret
= LTTNG_ERR_FATAL
;
225 cds_lfht_for_each_entry(uchan
->events
->ht
, &iter
.iter
, uevent
, node
.node
) {
226 strncpy(tmp
[i
].name
, uevent
->attr
.name
, LTTNG_SYMBOL_NAME_LEN
);
227 tmp
[i
].name
[LTTNG_SYMBOL_NAME_LEN
- 1] = '\0';
228 tmp
[i
].enabled
= uevent
->enabled
;
230 switch (uevent
->attr
.instrumentation
) {
231 case LTTNG_UST_TRACEPOINT
:
232 tmp
[i
].type
= LTTNG_EVENT_TRACEPOINT
;
234 case LTTNG_UST_PROBE
:
235 tmp
[i
].type
= LTTNG_EVENT_PROBE
;
237 case LTTNG_UST_FUNCTION
:
238 tmp
[i
].type
= LTTNG_EVENT_FUNCTION
;
242 tmp
[i
].loglevel
= uevent
->attr
.loglevel
;
243 switch (uevent
->attr
.loglevel_type
) {
244 case LTTNG_UST_LOGLEVEL_ALL
:
245 tmp
[i
].loglevel_type
= LTTNG_EVENT_LOGLEVEL_ALL
;
247 case LTTNG_UST_LOGLEVEL_RANGE
:
248 tmp
[i
].loglevel_type
= LTTNG_EVENT_LOGLEVEL_RANGE
;
250 case LTTNG_UST_LOGLEVEL_SINGLE
:
251 tmp
[i
].loglevel_type
= LTTNG_EVENT_LOGLEVEL_SINGLE
;
254 if (uevent
->filter
) {
269 * Fill lttng_event array of all kernel events in the channel.
271 static int list_lttng_kernel_events(char *channel_name
,
272 struct ltt_kernel_session
*kernel_session
, struct lttng_event
**events
)
275 unsigned int nb_event
;
276 struct ltt_kernel_event
*event
;
277 struct ltt_kernel_channel
*kchan
;
279 kchan
= trace_kernel_get_channel_by_name(channel_name
, kernel_session
);
281 ret
= LTTNG_ERR_KERN_CHAN_NOT_FOUND
;
285 nb_event
= kchan
->event_count
;
287 DBG("Listing events for channel %s", kchan
->channel
->name
);
293 *events
= zmalloc(nb_event
* sizeof(struct lttng_event
));
294 if (*events
== NULL
) {
295 ret
= LTTNG_ERR_FATAL
;
299 /* Kernel channels */
300 cds_list_for_each_entry(event
, &kchan
->events_list
.head
, list
) {
301 strncpy((*events
)[i
].name
, event
->event
->name
, LTTNG_SYMBOL_NAME_LEN
);
302 (*events
)[i
].name
[LTTNG_SYMBOL_NAME_LEN
- 1] = '\0';
303 (*events
)[i
].enabled
= event
->enabled
;
305 switch (event
->event
->instrumentation
) {
306 case LTTNG_KERNEL_TRACEPOINT
:
307 (*events
)[i
].type
= LTTNG_EVENT_TRACEPOINT
;
309 case LTTNG_KERNEL_KPROBE
:
310 case LTTNG_KERNEL_KRETPROBE
:
311 (*events
)[i
].type
= LTTNG_EVENT_PROBE
;
312 memcpy(&(*events
)[i
].attr
.probe
, &event
->event
->u
.kprobe
,
313 sizeof(struct lttng_kernel_kprobe
));
315 case LTTNG_KERNEL_FUNCTION
:
316 (*events
)[i
].type
= LTTNG_EVENT_FUNCTION
;
317 memcpy(&((*events
)[i
].attr
.ftrace
), &event
->event
->u
.ftrace
,
318 sizeof(struct lttng_kernel_function
));
320 case LTTNG_KERNEL_NOOP
:
321 (*events
)[i
].type
= LTTNG_EVENT_NOOP
;
323 case LTTNG_KERNEL_SYSCALL
:
324 (*events
)[i
].type
= LTTNG_EVENT_SYSCALL
;
326 case LTTNG_KERNEL_ALL
:
337 /* Negate the error code to differentiate the size from an error */
342 * Add URI so the consumer output object. Set the correct path depending on the
343 * domain adding the default trace directory.
345 static int add_uri_to_consumer(struct consumer_output
*consumer
,
346 struct lttng_uri
*uri
, int domain
, const char *session_name
)
349 const char *default_trace_dir
;
353 if (consumer
== NULL
) {
354 DBG("No consumer detected. Don't add URI. Stopping.");
355 ret
= LTTNG_ERR_NO_CONSUMER
;
360 case LTTNG_DOMAIN_KERNEL
:
361 default_trace_dir
= DEFAULT_KERNEL_TRACE_DIR
;
363 case LTTNG_DOMAIN_UST
:
364 default_trace_dir
= DEFAULT_UST_TRACE_DIR
;
368 * This case is possible is we try to add the URI to the global tracing
369 * session consumer object which in this case there is no subdir.
371 default_trace_dir
= "";
374 switch (uri
->dtype
) {
377 DBG2("Setting network URI to consumer");
379 /* Set URI into consumer output object */
380 ret
= consumer_set_network_uri(consumer
, uri
);
382 ret
= LTTNG_ERR_FATAL
;
384 } else if (ret
== 1) {
386 * URI was the same in the consumer so we do not append the subdir
387 * again so to not duplicate output dir.
392 if (uri
->stype
== LTTNG_STREAM_CONTROL
&& strlen(uri
->subdir
) == 0) {
393 ret
= consumer_set_subdir(consumer
, session_name
);
395 ret
= LTTNG_ERR_FATAL
;
400 if (uri
->stype
== LTTNG_STREAM_CONTROL
) {
401 /* On a new subdir, reappend the default trace dir. */
402 strncat(consumer
->subdir
, default_trace_dir
,
403 sizeof(consumer
->subdir
) - strlen(consumer
->subdir
) - 1);
404 DBG3("Append domain trace name to subdir %s", consumer
->subdir
);
409 DBG2("Setting trace directory path from URI to %s", uri
->dst
.path
);
410 memset(consumer
->dst
.trace_path
, 0,
411 sizeof(consumer
->dst
.trace_path
));
412 strncpy(consumer
->dst
.trace_path
, uri
->dst
.path
,
413 sizeof(consumer
->dst
.trace_path
));
414 /* Append default trace dir */
415 strncat(consumer
->dst
.trace_path
, default_trace_dir
,
416 sizeof(consumer
->dst
.trace_path
) -
417 strlen(consumer
->dst
.trace_path
) - 1);
418 /* Flag consumer as local. */
419 consumer
->type
= CONSUMER_DST_LOCAL
;
428 * Init tracing by creating trace directory and sending fds kernel consumer.
430 static int init_kernel_tracing(struct ltt_kernel_session
*session
)
433 struct lttng_ht_iter iter
;
434 struct consumer_socket
*socket
;
440 if (session
->consumer_fds_sent
== 0 && session
->consumer
!= NULL
) {
441 cds_lfht_for_each_entry(session
->consumer
->socks
->ht
, &iter
.iter
,
443 /* Code flow error */
444 assert(socket
->fd
>= 0);
446 pthread_mutex_lock(socket
->lock
);
447 ret
= kernel_consumer_send_session(socket
, session
);
448 pthread_mutex_unlock(socket
->lock
);
450 ret
= LTTNG_ERR_KERN_CONSUMER_FAIL
;
462 * Create a socket to the relayd using the URI.
464 * On success, the relayd_sock pointer is set to the created socket.
465 * Else, it's stays untouched and a lttcomm error code is returned.
467 static int create_connect_relayd(struct consumer_output
*output
,
468 const char *session_name
, struct lttng_uri
*uri
,
469 struct lttcomm_sock
**relayd_sock
)
472 struct lttcomm_sock
*sock
;
474 /* Create socket object from URI */
475 sock
= lttcomm_alloc_sock_from_uri(uri
);
477 ret
= LTTNG_ERR_FATAL
;
481 ret
= lttcomm_create_sock(sock
);
483 ret
= LTTNG_ERR_FATAL
;
487 /* Connect to relayd so we can proceed with a session creation. */
488 ret
= relayd_connect(sock
);
490 ERR("Unable to reach lttng-relayd");
491 ret
= LTTNG_ERR_RELAYD_CONNECT_FAIL
;
495 /* Create socket for control stream. */
496 if (uri
->stype
== LTTNG_STREAM_CONTROL
) {
497 DBG3("Creating relayd stream socket from URI");
499 /* Check relayd version */
500 ret
= relayd_version_check(sock
, RELAYD_VERSION_COMM_MAJOR
,
501 RELAYD_VERSION_COMM_MINOR
);
503 ret
= LTTNG_ERR_RELAYD_VERSION_FAIL
;
506 } else if (uri
->stype
== LTTNG_STREAM_DATA
) {
507 DBG3("Creating relayd data socket from URI");
509 /* Command is not valid */
510 ERR("Relayd invalid stream type: %d", uri
->stype
);
511 ret
= LTTNG_ERR_INVALID
;
521 (void) relayd_close(sock
);
525 lttcomm_destroy_sock(sock
);
532 * Connect to the relayd using URI and send the socket to the right consumer.
534 static int send_consumer_relayd_socket(int domain
, struct ltt_session
*session
,
535 struct lttng_uri
*relayd_uri
, struct consumer_output
*consumer
,
536 struct consumer_socket
*consumer_sock
)
539 struct lttcomm_sock
*sock
= NULL
;
541 /* Set the network sequence index if not set. */
542 if (consumer
->net_seq_index
== -1) {
544 * Increment net_seq_idx because we are about to transfer the
545 * new relayd socket to the consumer.
547 uatomic_inc(&relayd_net_seq_idx
);
548 /* Assign unique key so the consumer can match streams */
549 uatomic_set(&consumer
->net_seq_index
,
550 uatomic_read(&relayd_net_seq_idx
));
553 /* Connect to relayd and make version check if uri is the control. */
554 ret
= create_connect_relayd(consumer
, session
->name
, relayd_uri
, &sock
);
555 if (ret
!= LTTNG_OK
) {
559 /* If the control socket is connected, network session is ready */
560 if (relayd_uri
->stype
== LTTNG_STREAM_CONTROL
) {
561 session
->net_handle
= 1;
564 /* Send relayd socket to consumer. */
565 ret
= consumer_send_relayd_socket(consumer_sock
, sock
,
566 consumer
, relayd_uri
->stype
, session
->id
);
568 ret
= LTTNG_ERR_ENABLE_CONSUMER_FAIL
;
572 /* Flag that the corresponding socket was sent. */
573 if (relayd_uri
->stype
== LTTNG_STREAM_CONTROL
) {
574 consumer
->dst
.net
.control_sock_sent
= 1;
575 } else if (relayd_uri
->stype
== LTTNG_STREAM_DATA
) {
576 consumer
->dst
.net
.data_sock_sent
= 1;
582 * Close socket which was dup on the consumer side. The session daemon does
583 * NOT keep track of the relayd socket(s) once transfer to the consumer.
588 (void) relayd_close(sock
);
589 lttcomm_destroy_sock(sock
);
596 * Send both relayd sockets to a specific consumer and domain. This is a
597 * helper function to facilitate sending the information to the consumer for a
600 static int send_consumer_relayd_sockets(int domain
,
601 struct ltt_session
*session
, struct consumer_output
*consumer
,
602 struct consumer_socket
*sock
)
609 /* Sending control relayd socket. */
610 if (!consumer
->dst
.net
.control_sock_sent
) {
611 ret
= send_consumer_relayd_socket(domain
, session
,
612 &consumer
->dst
.net
.control
, consumer
, sock
);
613 if (ret
!= LTTNG_OK
) {
618 /* Sending data relayd socket. */
619 if (!consumer
->dst
.net
.data_sock_sent
) {
620 ret
= send_consumer_relayd_socket(domain
, session
,
621 &consumer
->dst
.net
.data
, consumer
, sock
);
622 if (ret
!= LTTNG_OK
) {
632 * Setup relayd connections for a tracing session. First creates the socket to
633 * the relayd and send them to the right domain consumer. Consumer type MUST be
636 static int setup_relayd(struct ltt_session
*session
)
639 struct ltt_ust_session
*usess
;
640 struct ltt_kernel_session
*ksess
;
641 struct consumer_socket
*socket
;
642 struct lttng_ht_iter iter
;
646 usess
= session
->ust_session
;
647 ksess
= session
->kernel_session
;
649 DBG2("Setting relayd for session %s", session
->name
);
653 if (usess
&& usess
->consumer
&& usess
->consumer
->type
== CONSUMER_DST_NET
654 && usess
->consumer
->enabled
) {
655 /* For each consumer socket, send relayd sockets */
656 cds_lfht_for_each_entry(usess
->consumer
->socks
->ht
, &iter
.iter
,
658 /* Code flow error */
659 assert(socket
->fd
>= 0);
661 pthread_mutex_lock(socket
->lock
);
662 ret
= send_consumer_relayd_sockets(LTTNG_DOMAIN_UST
, session
,
663 usess
->consumer
, socket
);
664 pthread_mutex_unlock(socket
->lock
);
665 if (ret
!= LTTNG_OK
) {
671 if (ksess
&& ksess
->consumer
&& ksess
->consumer
->type
== CONSUMER_DST_NET
672 && ksess
->consumer
->enabled
) {
673 cds_lfht_for_each_entry(ksess
->consumer
->socks
->ht
, &iter
.iter
,
675 /* Code flow error */
676 assert(socket
->fd
>= 0);
678 pthread_mutex_lock(socket
->lock
);
679 ret
= send_consumer_relayd_sockets(LTTNG_DOMAIN_KERNEL
, session
,
680 ksess
->consumer
, socket
);
681 pthread_mutex_unlock(socket
->lock
);
682 if (ret
!= LTTNG_OK
) {
694 * Start a kernel session by opening all necessary streams.
696 static int start_kernel_session(struct ltt_kernel_session
*ksess
, int wpipe
)
699 struct ltt_kernel_channel
*kchan
;
701 /* Open kernel metadata */
702 if (ksess
->metadata
== NULL
) {
703 ret
= kernel_open_metadata(ksess
);
705 ret
= LTTNG_ERR_KERN_META_FAIL
;
710 /* Open kernel metadata stream */
711 if (ksess
->metadata_stream_fd
< 0) {
712 ret
= kernel_open_metadata_stream(ksess
);
714 ERR("Kernel create metadata stream failed");
715 ret
= LTTNG_ERR_KERN_STREAM_FAIL
;
720 /* For each channel */
721 cds_list_for_each_entry(kchan
, &ksess
->channel_list
.head
, list
) {
722 if (kchan
->stream_count
== 0) {
723 ret
= kernel_open_channel_stream(kchan
);
725 ret
= LTTNG_ERR_KERN_STREAM_FAIL
;
728 /* Update the stream global counter */
729 ksess
->stream_count_global
+= ret
;
733 /* Setup kernel consumer socket and send fds to it */
734 ret
= init_kernel_tracing(ksess
);
736 ret
= LTTNG_ERR_KERN_START_FAIL
;
740 /* This start the kernel tracing */
741 ret
= kernel_start_session(ksess
);
743 ret
= LTTNG_ERR_KERN_START_FAIL
;
747 /* Quiescent wait after starting trace */
748 kernel_wait_quiescent(kernel_tracer_fd
);
759 * Command LTTNG_DISABLE_CHANNEL processed by the client thread.
761 int cmd_disable_channel(struct ltt_session
*session
, int domain
,
765 struct ltt_ust_session
*usess
;
767 usess
= session
->ust_session
;
770 case LTTNG_DOMAIN_KERNEL
:
772 ret
= channel_kernel_disable(session
->kernel_session
,
774 if (ret
!= LTTNG_OK
) {
778 kernel_wait_quiescent(kernel_tracer_fd
);
781 case LTTNG_DOMAIN_UST
:
783 struct ltt_ust_channel
*uchan
;
784 struct lttng_ht
*chan_ht
;
786 chan_ht
= usess
->domain_global
.channels
;
788 uchan
= trace_ust_find_channel_by_name(chan_ht
, channel_name
);
790 ret
= LTTNG_ERR_UST_CHAN_NOT_FOUND
;
794 ret
= channel_ust_disable(usess
, domain
, uchan
);
795 if (ret
!= LTTNG_OK
) {
801 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
802 case LTTNG_DOMAIN_UST_EXEC_NAME
:
803 case LTTNG_DOMAIN_UST_PID
:
806 ret
= LTTNG_ERR_UNKNOWN_DOMAIN
;
817 * Command LTTNG_ENABLE_CHANNEL processed by the client thread.
819 * The wpipe arguments is used as a notifier for the kernel thread.
821 int cmd_enable_channel(struct ltt_session
*session
,
822 int domain
, struct lttng_channel
*attr
, int wpipe
)
825 struct ltt_ust_session
*usess
= session
->ust_session
;
826 struct lttng_ht
*chan_ht
;
831 DBG("Enabling channel %s for session %s", attr
->name
, session
->name
);
834 case LTTNG_DOMAIN_KERNEL
:
836 struct ltt_kernel_channel
*kchan
;
838 kchan
= trace_kernel_get_channel_by_name(attr
->name
,
839 session
->kernel_session
);
841 ret
= channel_kernel_create(session
->kernel_session
, attr
, wpipe
);
843 ret
= channel_kernel_enable(session
->kernel_session
, kchan
);
846 if (ret
!= LTTNG_OK
) {
850 kernel_wait_quiescent(kernel_tracer_fd
);
853 * If the session was previously started, start as well this newly
854 * created kernel session so the events/channels enabled *after* the
855 * start actually work.
857 if (session
->started
&& !session
->kernel_session
->started
) {
858 ret
= start_kernel_session(session
->kernel_session
, wpipe
);
859 if (ret
!= LTTNG_OK
) {
865 case LTTNG_DOMAIN_UST
:
867 struct ltt_ust_channel
*uchan
;
869 chan_ht
= usess
->domain_global
.channels
;
871 uchan
= trace_ust_find_channel_by_name(chan_ht
, attr
->name
);
873 ret
= channel_ust_create(usess
, domain
, attr
);
875 ret
= channel_ust_enable(usess
, domain
, uchan
);
878 /* Start the UST session if the session was already started. */
879 if (session
->started
&& !usess
->start_trace
) {
880 ret
= ust_app_start_trace_all(usess
);
882 ret
= LTTNG_ERR_UST_START_FAIL
;
886 usess
->start_trace
= 1;
891 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
892 case LTTNG_DOMAIN_UST_EXEC_NAME
:
893 case LTTNG_DOMAIN_UST_PID
:
896 ret
= LTTNG_ERR_UNKNOWN_DOMAIN
;
906 * Command LTTNG_DISABLE_EVENT processed by the client thread.
908 int cmd_disable_event(struct ltt_session
*session
, int domain
,
909 char *channel_name
, char *event_name
)
914 case LTTNG_DOMAIN_KERNEL
:
916 struct ltt_kernel_channel
*kchan
;
917 struct ltt_kernel_session
*ksess
;
919 ksess
= session
->kernel_session
;
921 kchan
= trace_kernel_get_channel_by_name(channel_name
, ksess
);
923 ret
= LTTNG_ERR_KERN_CHAN_NOT_FOUND
;
927 ret
= event_kernel_disable_tracepoint(kchan
, event_name
);
928 if (ret
!= LTTNG_OK
) {
932 kernel_wait_quiescent(kernel_tracer_fd
);
935 case LTTNG_DOMAIN_UST
:
937 struct ltt_ust_channel
*uchan
;
938 struct ltt_ust_session
*usess
;
940 usess
= session
->ust_session
;
942 uchan
= trace_ust_find_channel_by_name(usess
->domain_global
.channels
,
945 ret
= LTTNG_ERR_UST_CHAN_NOT_FOUND
;
949 ret
= event_ust_disable_tracepoint(usess
, domain
, uchan
, event_name
);
950 if (ret
!= LTTNG_OK
) {
954 DBG3("Disable UST event %s in channel %s completed", event_name
,
959 case LTTNG_DOMAIN_UST_EXEC_NAME
:
960 case LTTNG_DOMAIN_UST_PID
:
961 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
975 * Command LTTNG_DISABLE_ALL_EVENT processed by the client thread.
977 int cmd_disable_event_all(struct ltt_session
*session
, int domain
,
983 case LTTNG_DOMAIN_KERNEL
:
985 struct ltt_kernel_session
*ksess
;
986 struct ltt_kernel_channel
*kchan
;
988 ksess
= session
->kernel_session
;
990 kchan
= trace_kernel_get_channel_by_name(channel_name
, ksess
);
992 ret
= LTTNG_ERR_KERN_CHAN_NOT_FOUND
;
996 ret
= event_kernel_disable_all(kchan
);
997 if (ret
!= LTTNG_OK
) {
1001 kernel_wait_quiescent(kernel_tracer_fd
);
1004 case LTTNG_DOMAIN_UST
:
1006 struct ltt_ust_session
*usess
;
1007 struct ltt_ust_channel
*uchan
;
1009 usess
= session
->ust_session
;
1011 uchan
= trace_ust_find_channel_by_name(usess
->domain_global
.channels
,
1013 if (uchan
== NULL
) {
1014 ret
= LTTNG_ERR_UST_CHAN_NOT_FOUND
;
1018 ret
= event_ust_disable_all_tracepoints(usess
, domain
, uchan
);
1023 DBG3("Disable all UST events in channel %s completed", channel_name
);
1028 case LTTNG_DOMAIN_UST_EXEC_NAME
:
1029 case LTTNG_DOMAIN_UST_PID
:
1030 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
1033 ret
= LTTNG_ERR_UND
;
1044 * Command LTTNG_ADD_CONTEXT processed by the client thread.
1046 int cmd_add_context(struct ltt_session
*session
, int domain
,
1047 char *channel_name
, struct lttng_event_context
*ctx
, int kwpipe
)
1052 case LTTNG_DOMAIN_KERNEL
:
1053 assert(session
->kernel_session
);
1055 if (session
->kernel_session
->channel_count
== 0) {
1056 /* Create default channel */
1057 ret
= channel_kernel_create(session
->kernel_session
, NULL
, kwpipe
);
1058 if (ret
!= LTTNG_OK
) {
1063 /* Add kernel context to kernel tracer */
1064 ret
= context_kernel_add(session
->kernel_session
, ctx
, channel_name
);
1065 if (ret
!= LTTNG_OK
) {
1069 case LTTNG_DOMAIN_UST
:
1071 struct ltt_ust_session
*usess
= session
->ust_session
;
1074 unsigned int chan_count
=
1075 lttng_ht_get_count(usess
->domain_global
.channels
);
1076 if (chan_count
== 0) {
1077 struct lttng_channel
*attr
;
1078 /* Create default channel */
1079 attr
= channel_new_default_attr(domain
);
1081 ret
= LTTNG_ERR_FATAL
;
1085 ret
= channel_ust_create(usess
, domain
, attr
);
1086 if (ret
!= LTTNG_OK
) {
1093 ret
= context_ust_add(usess
, domain
, ctx
, channel_name
);
1094 if (ret
!= LTTNG_OK
) {
1100 case LTTNG_DOMAIN_UST_EXEC_NAME
:
1101 case LTTNG_DOMAIN_UST_PID
:
1102 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
1105 ret
= LTTNG_ERR_UND
;
1116 * Command LTTNG_ENABLE_EVENT processed by the client thread.
1118 int cmd_enable_event(struct ltt_session
*session
, int domain
,
1119 char *channel_name
, struct lttng_event
*event
,
1120 struct lttng_filter_bytecode
*filter
, int wpipe
)
1122 int ret
, channel_created
= 0;
1123 struct lttng_channel
*attr
;
1127 assert(channel_name
);
1130 case LTTNG_DOMAIN_KERNEL
:
1132 struct ltt_kernel_channel
*kchan
;
1134 kchan
= trace_kernel_get_channel_by_name(channel_name
,
1135 session
->kernel_session
);
1136 if (kchan
== NULL
) {
1137 attr
= channel_new_default_attr(domain
);
1139 ret
= LTTNG_ERR_FATAL
;
1142 strncpy(attr
->name
, channel_name
, sizeof(attr
->name
));
1144 ret
= cmd_enable_channel(session
, domain
, attr
, wpipe
);
1145 if (ret
!= LTTNG_OK
) {
1151 channel_created
= 1;
1154 /* Get the newly created kernel channel pointer */
1155 kchan
= trace_kernel_get_channel_by_name(channel_name
,
1156 session
->kernel_session
);
1157 if (kchan
== NULL
) {
1158 /* This sould not happen... */
1159 ret
= LTTNG_ERR_FATAL
;
1163 ret
= event_kernel_enable_tracepoint(kchan
, event
);
1164 if (ret
!= LTTNG_OK
) {
1165 if (channel_created
) {
1166 /* Let's not leak a useless channel. */
1167 kernel_destroy_channel(kchan
);
1172 kernel_wait_quiescent(kernel_tracer_fd
);
1175 case LTTNG_DOMAIN_UST
:
1177 struct ltt_ust_channel
*uchan
;
1178 struct ltt_ust_session
*usess
= session
->ust_session
;
1182 /* Get channel from global UST domain */
1183 uchan
= trace_ust_find_channel_by_name(usess
->domain_global
.channels
,
1185 if (uchan
== NULL
) {
1186 /* Create default channel */
1187 attr
= channel_new_default_attr(domain
);
1189 ret
= LTTNG_ERR_FATAL
;
1192 strncpy(attr
->name
, channel_name
, sizeof(attr
->name
));
1194 ret
= cmd_enable_channel(session
, domain
, attr
, wpipe
);
1195 if (ret
!= LTTNG_OK
) {
1201 /* Get the newly created channel reference back */
1202 uchan
= trace_ust_find_channel_by_name(
1203 usess
->domain_global
.channels
, channel_name
);
1207 /* At this point, the session and channel exist on the tracer */
1208 ret
= event_ust_enable_tracepoint(usess
, domain
, uchan
, event
, filter
);
1209 if (ret
!= LTTNG_OK
) {
1215 case LTTNG_DOMAIN_UST_EXEC_NAME
:
1216 case LTTNG_DOMAIN_UST_PID
:
1217 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
1220 ret
= LTTNG_ERR_UND
;
1231 * Command LTTNG_ENABLE_ALL_EVENT processed by the client thread.
1233 int cmd_enable_event_all(struct ltt_session
*session
, int domain
,
1234 char *channel_name
, int event_type
,
1235 struct lttng_filter_bytecode
*filter
, int wpipe
)
1238 struct lttng_channel
*attr
;
1241 assert(channel_name
);
1244 case LTTNG_DOMAIN_KERNEL
:
1246 struct ltt_kernel_channel
*kchan
;
1248 assert(session
->kernel_session
);
1250 kchan
= trace_kernel_get_channel_by_name(channel_name
,
1251 session
->kernel_session
);
1252 if (kchan
== NULL
) {
1253 /* Create default channel */
1254 attr
= channel_new_default_attr(domain
);
1256 ret
= LTTNG_ERR_FATAL
;
1259 strncpy(attr
->name
, channel_name
, sizeof(attr
->name
));
1261 ret
= cmd_enable_channel(session
, domain
, attr
, wpipe
);
1262 if (ret
!= LTTNG_OK
) {
1268 /* Get the newly created kernel channel pointer */
1269 kchan
= trace_kernel_get_channel_by_name(channel_name
,
1270 session
->kernel_session
);
1274 switch (event_type
) {
1275 case LTTNG_EVENT_SYSCALL
:
1276 ret
= event_kernel_enable_all_syscalls(kchan
, kernel_tracer_fd
);
1278 case LTTNG_EVENT_TRACEPOINT
:
1280 * This call enables all LTTNG_KERNEL_TRACEPOINTS and
1281 * events already registered to the channel.
1283 ret
= event_kernel_enable_all_tracepoints(kchan
, kernel_tracer_fd
);
1285 case LTTNG_EVENT_ALL
:
1286 /* Enable syscalls and tracepoints */
1287 ret
= event_kernel_enable_all(kchan
, kernel_tracer_fd
);
1290 ret
= LTTNG_ERR_KERN_ENABLE_FAIL
;
1294 /* Manage return value */
1295 if (ret
!= LTTNG_OK
) {
1297 * On error, cmd_enable_channel call will take care of destroying
1298 * the created channel if it was needed.
1303 kernel_wait_quiescent(kernel_tracer_fd
);
1306 case LTTNG_DOMAIN_UST
:
1308 struct ltt_ust_channel
*uchan
;
1309 struct ltt_ust_session
*usess
= session
->ust_session
;
1313 /* Get channel from global UST domain */
1314 uchan
= trace_ust_find_channel_by_name(usess
->domain_global
.channels
,
1316 if (uchan
== NULL
) {
1317 /* Create default channel */
1318 attr
= channel_new_default_attr(domain
);
1320 ret
= LTTNG_ERR_FATAL
;
1323 strncpy(attr
->name
, channel_name
, sizeof(attr
->name
));
1325 ret
= cmd_enable_channel(session
, domain
, attr
, wpipe
);
1326 if (ret
!= LTTNG_OK
) {
1332 /* Get the newly created channel reference back */
1333 uchan
= trace_ust_find_channel_by_name(
1334 usess
->domain_global
.channels
, channel_name
);
1338 /* At this point, the session and channel exist on the tracer */
1340 switch (event_type
) {
1341 case LTTNG_EVENT_ALL
:
1342 case LTTNG_EVENT_TRACEPOINT
:
1343 ret
= event_ust_enable_all_tracepoints(usess
, domain
, uchan
,
1345 if (ret
!= LTTNG_OK
) {
1350 ret
= LTTNG_ERR_UST_ENABLE_FAIL
;
1354 /* Manage return value */
1355 if (ret
!= LTTNG_OK
) {
1362 case LTTNG_DOMAIN_UST_EXEC_NAME
:
1363 case LTTNG_DOMAIN_UST_PID
:
1364 case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN
:
1367 ret
= LTTNG_ERR_UND
;
1379 * Command LTTNG_LIST_TRACEPOINTS processed by the client thread.
1381 ssize_t
cmd_list_tracepoints(int domain
, struct lttng_event
**events
)
1384 ssize_t nb_events
= 0;
1387 case LTTNG_DOMAIN_KERNEL
:
1388 nb_events
= kernel_list_events(kernel_tracer_fd
, events
);
1389 if (nb_events
< 0) {
1390 ret
= LTTNG_ERR_KERN_LIST_FAIL
;
1394 case LTTNG_DOMAIN_UST
:
1395 nb_events
= ust_app_list_events(events
);
1396 if (nb_events
< 0) {
1397 ret
= LTTNG_ERR_UST_LIST_FAIL
;
1402 ret
= LTTNG_ERR_UND
;
1409 /* Return negative value to differentiate return code */
1414 * Command LTTNG_LIST_TRACEPOINT_FIELDS processed by the client thread.
1416 ssize_t
cmd_list_tracepoint_fields(int domain
,
1417 struct lttng_event_field
**fields
)
1420 ssize_t nb_fields
= 0;
1423 case LTTNG_DOMAIN_UST
:
1424 nb_fields
= ust_app_list_event_fields(fields
);
1425 if (nb_fields
< 0) {
1426 ret
= LTTNG_ERR_UST_LIST_FAIL
;
1430 case LTTNG_DOMAIN_KERNEL
:
1431 default: /* fall-through */
1432 ret
= LTTNG_ERR_UND
;
1439 /* Return negative value to differentiate return code */
1444 * Command LTTNG_START_TRACE processed by the client thread.
1446 int cmd_start_trace(struct ltt_session
*session
)
1449 struct ltt_kernel_session
*ksession
;
1450 struct ltt_ust_session
*usess
;
1454 /* Ease our life a bit ;) */
1455 ksession
= session
->kernel_session
;
1456 usess
= session
->ust_session
;
1458 if (session
->enabled
) {
1459 /* Already started. */
1460 ret
= LTTNG_ERR_TRACE_ALREADY_STARTED
;
1464 session
->enabled
= 1;
1466 ret
= setup_relayd(session
);
1467 if (ret
!= LTTNG_OK
) {
1468 ERR("Error setting up relayd for session %s", session
->name
);
1472 /* Kernel tracing */
1473 if (ksession
!= NULL
) {
1474 ret
= start_kernel_session(ksession
, kernel_tracer_fd
);
1475 if (ret
!= LTTNG_OK
) {
1480 /* Flag session that trace should start automatically */
1482 usess
->start_trace
= 1;
1484 ret
= ust_app_start_trace_all(usess
);
1486 ret
= LTTNG_ERR_UST_START_FAIL
;
1491 session
->started
= 1;
1500 * Command LTTNG_STOP_TRACE processed by the client thread.
1502 int cmd_stop_trace(struct ltt_session
*session
)
1505 struct ltt_kernel_channel
*kchan
;
1506 struct ltt_kernel_session
*ksession
;
1507 struct ltt_ust_session
*usess
;
1512 ksession
= session
->kernel_session
;
1513 usess
= session
->ust_session
;
1515 if (!session
->enabled
) {
1516 ret
= LTTNG_ERR_TRACE_ALREADY_STOPPED
;
1520 session
->enabled
= 0;
1524 DBG("Stop kernel tracing");
1526 /* Flush metadata if exist */
1527 if (ksession
->metadata_stream_fd
>= 0) {
1528 ret
= kernel_metadata_flush_buffer(ksession
->metadata_stream_fd
);
1530 ERR("Kernel metadata flush failed");
1534 /* Flush all buffers before stopping */
1535 cds_list_for_each_entry(kchan
, &ksession
->channel_list
.head
, list
) {
1536 ret
= kernel_flush_buffer(kchan
);
1538 ERR("Kernel flush buffer error");
1542 ret
= kernel_stop_session(ksession
);
1544 ret
= LTTNG_ERR_KERN_STOP_FAIL
;
1548 kernel_wait_quiescent(kernel_tracer_fd
);
1550 ksession
->started
= 0;
1554 usess
->start_trace
= 0;
1556 ret
= ust_app_stop_trace_all(usess
);
1558 ret
= LTTNG_ERR_UST_STOP_FAIL
;
1563 session
->started
= 0;
1572 * Command LTTNG_SET_CONSUMER_URI processed by the client thread.
1574 int cmd_set_consumer_uri(int domain
, struct ltt_session
*session
,
1575 size_t nb_uri
, struct lttng_uri
*uris
)
1578 struct ltt_kernel_session
*ksess
= session
->kernel_session
;
1579 struct ltt_ust_session
*usess
= session
->ust_session
;
1580 struct consumer_output
*consumer
= NULL
;
1586 /* Can't enable consumer after session started. */
1587 if (session
->enabled
) {
1588 ret
= LTTNG_ERR_TRACE_ALREADY_STARTED
;
1593 * This case switch makes sure the domain session has a temporary consumer
1594 * so the URL can be set.
1598 /* Code flow error. A session MUST always have a consumer object */
1599 assert(session
->consumer
);
1601 * The URL will be added to the tracing session consumer instead of a
1602 * specific domain consumer.
1604 consumer
= session
->consumer
;
1606 case LTTNG_DOMAIN_KERNEL
:
1607 /* Code flow error if we don't have a kernel session here. */
1610 /* Create consumer output if none exists */
1611 consumer
= ksess
->tmp_consumer
;
1612 if (consumer
== NULL
) {
1613 consumer
= consumer_copy_output(ksess
->consumer
);
1614 if (consumer
== NULL
) {
1615 ret
= LTTNG_ERR_FATAL
;
1618 /* Trash the consumer subdir, we are about to set a new one. */
1619 memset(consumer
->subdir
, 0, sizeof(consumer
->subdir
));
1620 ksess
->tmp_consumer
= consumer
;
1624 case LTTNG_DOMAIN_UST
:
1625 /* Code flow error if we don't have a kernel session here. */
1628 /* Create consumer output if none exists */
1629 consumer
= usess
->tmp_consumer
;
1630 if (consumer
== NULL
) {
1631 consumer
= consumer_copy_output(usess
->consumer
);
1632 if (consumer
== NULL
) {
1633 ret
= LTTNG_ERR_FATAL
;
1636 /* Trash the consumer subdir, we are about to set a new one. */
1637 memset(consumer
->subdir
, 0, sizeof(consumer
->subdir
));
1638 usess
->tmp_consumer
= consumer
;
1644 for (i
= 0; i
< nb_uri
; i
++) {
1645 struct consumer_socket
*socket
;
1646 struct lttng_ht_iter iter
;
1648 ret
= add_uri_to_consumer(consumer
, &uris
[i
], domain
, session
->name
);
1654 * Don't send relayd socket if URI is NOT remote or if the relayd
1655 * socket for the session was already sent.
1657 if (uris
[i
].dtype
== LTTNG_DST_PATH
||
1658 (uris
[i
].stype
== LTTNG_STREAM_CONTROL
&&
1659 consumer
->dst
.net
.control_sock_sent
) ||
1660 (uris
[i
].stype
== LTTNG_STREAM_DATA
&&
1661 consumer
->dst
.net
.data_sock_sent
)) {
1665 /* Try to send relayd URI to the consumer if exist. */
1667 cds_lfht_for_each_entry(consumer
->socks
->ht
, &iter
.iter
,
1668 socket
, node
.node
) {
1670 /* A socket in the HT should never have a negative fd */
1671 assert(socket
->fd
>= 0);
1673 pthread_mutex_lock(socket
->lock
);
1674 ret
= send_consumer_relayd_socket(domain
, session
, &uris
[i
],
1676 pthread_mutex_unlock(socket
->lock
);
1677 if (ret
!= LTTNG_OK
) {
1693 * Command LTTNG_CREATE_SESSION processed by the client thread.
1695 int cmd_create_session_uri(char *name
, struct lttng_uri
*uris
,
1696 size_t nb_uri
, lttng_sock_cred
*creds
)
1700 struct ltt_session
*session
;
1705 * Verify if the session already exist
1707 * XXX: There is no need for the session lock list here since the caller
1708 * (process_client_msg) is holding it. We might want to change that so a
1709 * single command does not lock the entire session list.
1711 session
= session_find_by_name(name
);
1712 if (session
!= NULL
) {
1713 ret
= LTTNG_ERR_EXIST_SESS
;
1717 /* Create tracing session in the registry */
1718 ret
= session_create(name
, path
, LTTNG_SOCK_GET_UID_CRED(creds
),
1719 LTTNG_SOCK_GET_GID_CRED(creds
));
1720 if (ret
!= LTTNG_OK
) {
1725 * Get the newly created session pointer back
1727 * XXX: There is no need for the session lock list here since the caller
1728 * (process_client_msg) is holding it. We might want to change that so a
1729 * single command does not lock the entire session list.
1731 session
= session_find_by_name(name
);
1734 /* Create default consumer output for the session not yet created. */
1735 session
->consumer
= consumer_create_output(CONSUMER_DST_LOCAL
);
1736 if (session
->consumer
== NULL
) {
1737 ret
= LTTNG_ERR_FATAL
;
1738 goto consumer_error
;
1742 * This means that the lttng_create_session call was called with the _path_
1743 * argument set to NULL.
1747 * At this point, we'll skip the consumer URI setup and create a
1748 * session with a NULL path which will flag the session to NOT spawn a
1751 DBG("Create session %s with NO uri, skipping consumer setup", name
);
1755 session
->start_consumer
= 1;
1757 ret
= cmd_set_consumer_uri(0, session
, nb_uri
, uris
);
1758 if (ret
!= LTTNG_OK
) {
1759 goto consumer_error
;
1762 session
->consumer
->enabled
= 1;
1768 session_destroy(session
);
1775 * Command LTTNG_DESTROY_SESSION processed by the client thread.
1777 int cmd_destroy_session(struct ltt_session
*session
, int wpipe
)
1780 struct ltt_ust_session
*usess
;
1781 struct ltt_kernel_session
*ksess
;
1786 usess
= session
->ust_session
;
1787 ksess
= session
->kernel_session
;
1789 /* Clean kernel session teardown */
1790 kernel_destroy_session(ksess
);
1792 /* UST session teardown */
1794 /* Close any relayd session */
1795 consumer_output_send_destroy_relayd(usess
->consumer
);
1797 /* Destroy every UST application related to this session. */
1798 ret
= ust_app_destroy_trace_all(usess
);
1800 ERR("Error in ust_app_destroy_trace_all");
1803 /* Clean up the rest. */
1804 trace_ust_destroy_session(usess
);
1808 * Must notify the kernel thread here to update it's poll set in order to
1809 * remove the channel(s)' fd just destroyed.
1811 ret
= notify_thread_pipe(wpipe
);
1813 PERROR("write kernel poll pipe");
1816 ret
= session_destroy(session
);
1822 * Command LTTNG_CALIBRATE processed by the client thread.
1824 int cmd_calibrate(int domain
, struct lttng_calibrate
*calibrate
)
1829 case LTTNG_DOMAIN_KERNEL
:
1831 struct lttng_kernel_calibrate kcalibrate
;
1833 kcalibrate
.type
= calibrate
->type
;
1834 ret
= kernel_calibrate(kernel_tracer_fd
, &kcalibrate
);
1836 ret
= LTTNG_ERR_KERN_ENABLE_FAIL
;
1841 case LTTNG_DOMAIN_UST
:
1843 struct lttng_ust_calibrate ucalibrate
;
1845 ucalibrate
.type
= calibrate
->type
;
1846 ret
= ust_app_calibrate_glb(&ucalibrate
);
1848 ret
= LTTNG_ERR_UST_CALIBRATE_FAIL
;
1854 ret
= LTTNG_ERR_UND
;
1865 * Command LTTNG_REGISTER_CONSUMER processed by the client thread.
1867 int cmd_register_consumer(struct ltt_session
*session
, int domain
,
1868 const char *sock_path
, struct consumer_data
*cdata
)
1871 struct consumer_socket
*socket
;
1878 case LTTNG_DOMAIN_KERNEL
:
1880 struct ltt_kernel_session
*ksess
= session
->kernel_session
;
1884 /* Can't register a consumer if there is already one */
1885 if (ksess
->consumer_fds_sent
!= 0) {
1886 ret
= LTTNG_ERR_KERN_CONSUMER_FAIL
;
1890 sock
= lttcomm_connect_unix_sock(sock_path
);
1892 ret
= LTTNG_ERR_CONNECT_FAIL
;
1896 socket
= consumer_allocate_socket(sock
);
1897 if (socket
== NULL
) {
1900 PERROR("close register consumer");
1902 ret
= LTTNG_ERR_FATAL
;
1906 socket
->lock
= zmalloc(sizeof(pthread_mutex_t
));
1907 if (socket
->lock
== NULL
) {
1908 PERROR("zmalloc pthread mutex");
1909 ret
= LTTNG_ERR_FATAL
;
1912 pthread_mutex_init(socket
->lock
, NULL
);
1913 socket
->registered
= 1;
1916 consumer_add_socket(socket
, ksess
->consumer
);
1919 pthread_mutex_lock(&cdata
->pid_mutex
);
1921 pthread_mutex_unlock(&cdata
->pid_mutex
);
1926 /* TODO: Userspace tracing */
1927 ret
= LTTNG_ERR_UND
;
1938 * Command LTTNG_LIST_DOMAINS processed by the client thread.
1940 ssize_t
cmd_list_domains(struct ltt_session
*session
,
1941 struct lttng_domain
**domains
)
1946 if (session
->kernel_session
!= NULL
) {
1947 DBG3("Listing domains found kernel domain");
1951 if (session
->ust_session
!= NULL
) {
1952 DBG3("Listing domains found UST global domain");
1956 *domains
= zmalloc(nb_dom
* sizeof(struct lttng_domain
));
1957 if (*domains
== NULL
) {
1958 ret
= LTTNG_ERR_FATAL
;
1962 if (session
->kernel_session
!= NULL
) {
1963 (*domains
)[index
].type
= LTTNG_DOMAIN_KERNEL
;
1967 if (session
->ust_session
!= NULL
) {
1968 (*domains
)[index
].type
= LTTNG_DOMAIN_UST
;
1975 /* Return negative value to differentiate return code */
1981 * Command LTTNG_LIST_CHANNELS processed by the client thread.
1983 ssize_t
cmd_list_channels(int domain
, struct ltt_session
*session
,
1984 struct lttng_channel
**channels
)
1987 ssize_t nb_chan
= 0;
1990 case LTTNG_DOMAIN_KERNEL
:
1991 if (session
->kernel_session
!= NULL
) {
1992 nb_chan
= session
->kernel_session
->channel_count
;
1994 DBG3("Number of kernel channels %zd", nb_chan
);
1996 ret
= LTTNG_ERR_KERN_CHAN_NOT_FOUND
;
1999 case LTTNG_DOMAIN_UST
:
2000 if (session
->ust_session
!= NULL
) {
2001 nb_chan
= lttng_ht_get_count(
2002 session
->ust_session
->domain_global
.channels
);
2004 DBG3("Number of UST global channels %zd", nb_chan
);
2006 ret
= LTTNG_ERR_UST_CHAN_NOT_FOUND
;
2011 ret
= LTTNG_ERR_UND
;
2016 *channels
= zmalloc(nb_chan
* sizeof(struct lttng_channel
));
2017 if (*channels
== NULL
) {
2018 ret
= LTTNG_ERR_FATAL
;
2022 list_lttng_channels(domain
, session
, *channels
);
2025 /* Ret value was set in the domain switch case */
2032 /* Return negative value to differentiate return code */
2037 * Command LTTNG_LIST_EVENTS processed by the client thread.
2039 ssize_t
cmd_list_events(int domain
, struct ltt_session
*session
,
2040 char *channel_name
, struct lttng_event
**events
)
2043 ssize_t nb_event
= 0;
2046 case LTTNG_DOMAIN_KERNEL
:
2047 if (session
->kernel_session
!= NULL
) {
2048 nb_event
= list_lttng_kernel_events(channel_name
,
2049 session
->kernel_session
, events
);
2052 case LTTNG_DOMAIN_UST
:
2054 if (session
->ust_session
!= NULL
) {
2055 nb_event
= list_lttng_ust_global_events(channel_name
,
2056 &session
->ust_session
->domain_global
, events
);
2061 ret
= LTTNG_ERR_UND
;
2068 /* Return negative value to differentiate return code */
2073 * Using the session list, filled a lttng_session array to send back to the
2074 * client for session listing.
2076 * The session list lock MUST be acquired before calling this function. Use
2077 * session_lock_list() and session_unlock_list().
2079 void cmd_list_lttng_sessions(struct lttng_session
*sessions
, uid_t uid
,
2084 struct ltt_session
*session
;
2085 struct ltt_session_list
*list
= session_get_list();
2087 DBG("Getting all available session for UID %d GID %d",
2090 * Iterate over session list and append data after the control struct in
2093 cds_list_for_each_entry(session
, &list
->head
, list
) {
2095 * Only list the sessions the user can control.
2097 if (!session_access_ok(session
, uid
, gid
)) {
2101 struct ltt_kernel_session
*ksess
= session
->kernel_session
;
2102 struct ltt_ust_session
*usess
= session
->ust_session
;
2104 if (session
->consumer
->type
== CONSUMER_DST_NET
||
2105 (ksess
&& ksess
->consumer
->type
== CONSUMER_DST_NET
) ||
2106 (usess
&& usess
->consumer
->type
== CONSUMER_DST_NET
)) {
2107 ret
= build_network_session_path(sessions
[i
].path
,
2108 sizeof(session
[i
].path
), session
);
2110 ret
= snprintf(sessions
[i
].path
, sizeof(session
[i
].path
), "%s",
2111 session
->consumer
->dst
.trace_path
);
2114 PERROR("snprintf session path");
2118 strncpy(sessions
[i
].name
, session
->name
, NAME_MAX
);
2119 sessions
[i
].name
[NAME_MAX
- 1] = '\0';
2120 sessions
[i
].enabled
= session
->enabled
;
2126 * Command LTTNG_DISABLE_CONSUMER processed by the client thread.
2128 int cmd_disable_consumer(int domain
, struct ltt_session
*session
)
2131 struct ltt_kernel_session
*ksess
= session
->kernel_session
;
2132 struct ltt_ust_session
*usess
= session
->ust_session
;
2133 struct consumer_output
*consumer
;
2137 if (session
->enabled
) {
2138 /* Can't disable consumer on an already started session */
2139 ret
= LTTNG_ERR_TRACE_ALREADY_STARTED
;
2143 if (!session
->start_consumer
) {
2144 ret
= LTTNG_ERR_NO_CONSUMER
;
2150 DBG("Disable tracing session %s consumer", session
->name
);
2151 consumer
= session
->consumer
;
2153 case LTTNG_DOMAIN_KERNEL
:
2154 /* Code flow error if we don't have a kernel session here. */
2157 DBG("Disabling kernel consumer");
2158 consumer
= ksess
->consumer
;
2161 case LTTNG_DOMAIN_UST
:
2162 /* Code flow error if we don't have a UST session here. */
2165 DBG("Disabling UST consumer");
2166 consumer
= usess
->consumer
;
2170 ret
= LTTNG_ERR_UNKNOWN_DOMAIN
;
2175 consumer
->enabled
= 0;
2176 /* Success at this point */
2179 ret
= LTTNG_ERR_NO_CONSUMER
;
2187 * Command LTTNG_ENABLE_CONSUMER processed by the client thread.
2189 int cmd_enable_consumer(int domain
, struct ltt_session
*session
)
2192 struct ltt_kernel_session
*ksess
= session
->kernel_session
;
2193 struct ltt_ust_session
*usess
= session
->ust_session
;
2194 struct consumer_output
*consumer
= NULL
;
2198 /* Can't enable consumer after session started. */
2199 if (session
->enabled
) {
2200 ret
= LTTNG_ERR_TRACE_ALREADY_STARTED
;
2206 assert(session
->consumer
);
2207 consumer
= session
->consumer
;
2209 case LTTNG_DOMAIN_KERNEL
:
2210 /* Code flow error if we don't have a kernel session here. */
2214 * Check if we have already sent fds to the consumer. In that case,
2215 * the enable-consumer command can't be used because a start trace
2216 * had previously occured.
2218 if (ksess
->consumer_fds_sent
) {
2219 ret
= LTTNG_ERR_ENABLE_CONSUMER_FAIL
;
2223 consumer
= ksess
->tmp_consumer
;
2224 if (consumer
== NULL
) {
2226 /* No temp. consumer output exists. Using the current one. */
2227 DBG3("No temporary consumer. Using default");
2228 consumer
= ksess
->consumer
;
2232 switch (consumer
->type
) {
2233 case CONSUMER_DST_LOCAL
:
2234 DBG2("Consumer output is local. Creating directory(ies)");
2236 /* Create directory(ies) */
2237 ret
= run_as_mkdir_recursive(consumer
->dst
.trace_path
,
2238 S_IRWXU
| S_IRWXG
, session
->uid
, session
->gid
);
2240 if (ret
!= -EEXIST
) {
2241 ERR("Trace directory creation error");
2242 ret
= LTTNG_ERR_FATAL
;
2247 case CONSUMER_DST_NET
:
2248 DBG2("Consumer output is network. Validating URIs");
2249 /* Validate if we have both control and data path set. */
2250 if (!consumer
->dst
.net
.control_isset
) {
2251 ret
= LTTNG_ERR_URL_CTRL_MISS
;
2255 if (!consumer
->dst
.net
.data_isset
) {
2256 ret
= LTTNG_ERR_URL_DATA_MISS
;
2260 /* Check established network session state */
2261 if (session
->net_handle
== 0) {
2262 ret
= LTTNG_ERR_ENABLE_CONSUMER_FAIL
;
2263 ERR("Session network handle is not set on enable-consumer");
2272 * This is race free for now since the session lock is acquired before
2273 * ending up in this function. No other threads can access this kernel
2274 * session without this lock hence freeing the consumer output object
2278 /* Destroy current consumer. We are about to replace it */
2279 consumer_destroy_output(ksess
->consumer
);
2281 ksess
->consumer
= consumer
;
2282 ksess
->tmp_consumer
= NULL
;
2285 case LTTNG_DOMAIN_UST
:
2286 /* Code flow error if we don't have a UST session here. */
2290 * Check if we have already sent fds to the consumer. In that case,
2291 * the enable-consumer command can't be used because a start trace
2292 * had previously occured.
2294 if (usess
->start_trace
) {
2295 ret
= LTTNG_ERR_ENABLE_CONSUMER_FAIL
;
2299 consumer
= usess
->tmp_consumer
;
2300 if (consumer
== NULL
) {
2302 /* No temp. consumer output exists. Using the current one. */
2303 DBG3("No temporary consumer. Using default");
2304 consumer
= usess
->consumer
;
2308 switch (consumer
->type
) {
2309 case CONSUMER_DST_LOCAL
:
2310 DBG2("Consumer output is local. Creating directory(ies)");
2312 /* Create directory(ies) */
2313 ret
= run_as_mkdir_recursive(consumer
->dst
.trace_path
,
2314 S_IRWXU
| S_IRWXG
, session
->uid
, session
->gid
);
2316 if (ret
!= -EEXIST
) {
2317 ERR("Trace directory creation error");
2318 ret
= LTTNG_ERR_FATAL
;
2323 case CONSUMER_DST_NET
:
2324 DBG2("Consumer output is network. Validating URIs");
2325 /* Validate if we have both control and data path set. */
2326 if (!consumer
->dst
.net
.control_isset
) {
2327 ret
= LTTNG_ERR_URL_CTRL_MISS
;
2331 if (!consumer
->dst
.net
.data_isset
) {
2332 ret
= LTTNG_ERR_URL_DATA_MISS
;
2336 /* Check established network session state */
2337 if (session
->net_handle
== 0) {
2338 ret
= LTTNG_ERR_ENABLE_CONSUMER_FAIL
;
2339 DBG2("Session network handle is not set on enable-consumer");
2343 if (consumer
->net_seq_index
== -1) {
2344 ret
= LTTNG_ERR_ENABLE_CONSUMER_FAIL
;
2345 DBG2("Network index is not set on the consumer");
2354 * This is race free for now since the session lock is acquired before
2355 * ending up in this function. No other threads can access this kernel
2356 * session without this lock hence freeing the consumer output object
2360 /* Destroy current consumer. We are about to replace it */
2361 consumer_destroy_output(usess
->consumer
);
2363 usess
->consumer
= consumer
;
2364 usess
->tmp_consumer
= NULL
;
2369 session
->start_consumer
= 1;
2373 consumer
->enabled
= 1;
2374 /* Success at this point */
2377 /* Should not really happend... */
2378 ret
= LTTNG_ERR_NO_CONSUMER
;
2386 * Command LTTNG_DATA_PENDING returning 0 if the data is NOT pending meaning
2387 * ready for trace analysis (or anykind of reader) or else 1 for pending data.
2389 int cmd_data_pending(struct ltt_session
*session
)
2392 struct ltt_kernel_session
*ksess
= session
->kernel_session
;
2393 struct ltt_ust_session
*usess
= session
->ust_session
;
2397 /* Session MUST be stopped to ask for data availability. */
2398 if (session
->enabled
) {
2399 ret
= LTTNG_ERR_SESSION_STARTED
;
2403 if (ksess
&& ksess
->consumer
) {
2404 ret
= consumer_is_data_pending(ksess
->id
, ksess
->consumer
);
2406 /* Data is still being extracted for the kernel. */
2411 if (usess
&& usess
->consumer
) {
2412 ret
= consumer_is_data_pending(usess
->id
, usess
->consumer
);
2414 /* Data is still being extracted for the kernel. */
2419 /* Data is ready to be read by a viewer */
2427 * Init command subsystem.
2432 * Set network sequence index to 1 for streams to match a relayd socket on
2433 * the consumer side.
2435 uatomic_set(&relayd_net_seq_idx
, 1);
2437 DBG("Command subsystem initialized");