[lttng-tools.git] / src / bin / lttng-sessiond / kernel-consumer.c

/*
 * Copyright (C) 2012 - David Goulet <dgoulet@efficios.com>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License, version 2 only, as
 * published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 *
 * You should have received a copy of the GNU General Public License along with
 * this program; if not, write to the Free Software Foundation, Inc., 51
 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */

#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>

#include <common/common.h>
#include <common/defaults.h>

#include "consumer.h"
#include "health-sessiond.h"
#include "kernel-consumer.h"

static char *create_channel_path(struct consumer_output *consumer,
		uid_t uid, gid_t gid)
{
	int ret;
	char tmp_path[PATH_MAX];
	char *pathname = NULL;

	assert(consumer);

	/* Get the right path name destination */
	if (consumer->type == CONSUMER_DST_LOCAL) {
		/* Set application path to the destination path */
		ret = snprintf(tmp_path, sizeof(tmp_path), "%s%s",
				consumer->dst.trace_path, consumer->subdir);
		if (ret < 0) {
			PERROR("snprintf kernel channel path");
			goto error;
		}
		pathname = strndup(tmp_path, sizeof(tmp_path));

		/* Create directory */
		ret = run_as_mkdir_recursive(pathname, S_IRWXU | S_IRWXG, uid, gid);
		if (ret < 0) {
			if (ret != -EEXIST) {
				ERR("Trace directory creation error");
				goto error;
			}
		}
		DBG3("Kernel local consumer tracefile path: %s", pathname);
	} else {
		ret = snprintf(tmp_path, sizeof(tmp_path), "%s", consumer->subdir);
		if (ret < 0) {
			PERROR("snprintf kernel metadata path");
			goto error;
		}
		pathname = strndup(tmp_path, sizeof(tmp_path));
		DBG3("Kernel network consumer subdir path: %s", pathname);
	}

	return pathname;

error:
	free(pathname);
	return NULL;
}

/*
 * Sending a single channel to the consumer with command ADD_CHANNEL.
 */
int kernel_consumer_add_channel(struct consumer_socket *sock,
		struct ltt_kernel_channel *channel, struct ltt_kernel_session *session,
		unsigned int monitor)
{
	int ret;
	char *pathname;
	struct lttcomm_consumer_msg lkm;
	struct consumer_output *consumer;

	/* Safety net */
	assert(channel);
	assert(session);
	assert(session->consumer);

	consumer = session->consumer;

	DBG("Kernel consumer adding channel %s to kernel consumer",
			channel->channel->name);

	if (monitor) {
		pathname = create_channel_path(consumer, session->uid, session->gid);
		if (!pathname) {
			ret = -1;
			goto error;
		}
	} else {
		/* Empty path. */
		pathname = "";
	}

	/* Prep channel message structure */
	consumer_init_channel_comm_msg(&lkm,
			LTTNG_CONSUMER_ADD_CHANNEL,
			channel->fd,
			session->id,
			pathname,
			session->uid,
			session->gid,
			consumer->net_seq_index,
			channel->channel->name,
			channel->stream_count,
			channel->channel->attr.output,
			CONSUMER_CHANNEL_TYPE_DATA,
			channel->channel->attr.tracefile_size,
			channel->channel->attr.tracefile_count,
			monitor,
			channel->channel->attr.live_timer_interval);

	health_code_update();

	ret = consumer_send_channel(sock, &lkm);
	if (ret < 0) {
		goto error;
	}

	health_code_update();

error:
	return ret;
}

/*
 * Sending metadata to the consumer with command ADD_CHANNEL and ADD_STREAM.
 */
int kernel_consumer_add_metadata(struct consumer_socket *sock,
		struct ltt_kernel_session *session, unsigned int monitor)
{
	int ret;
	char *pathname;
	struct lttcomm_consumer_msg lkm;
	struct consumer_output *consumer;

	/* Safety net */
	assert(session);
	assert(session->consumer);
	assert(sock);

	DBG("Sending metadata %d to kernel consumer", session->metadata_stream_fd);

	/* Get consumer output pointer */
	consumer = session->consumer;

	if (monitor) {
		pathname = create_channel_path(consumer, session->uid, session->gid);
		if (!pathname) {
			ret = -1;
			goto error;
		}
	} else {
		/* Empty path. */
		pathname = "";
	}

	/* Prep channel message structure */
	consumer_init_channel_comm_msg(&lkm,
			LTTNG_CONSUMER_ADD_CHANNEL,
			session->metadata->fd,
			session->id,
			pathname,
			session->uid,
			session->gid,
			consumer->net_seq_index,
			DEFAULT_METADATA_NAME,
			1,
			DEFAULT_KERNEL_CHANNEL_OUTPUT,
			CONSUMER_CHANNEL_TYPE_METADATA,
			0, 0,
			monitor, 0);

	health_code_update();

	ret = consumer_send_channel(sock, &lkm);
	if (ret < 0) {
		goto error;
	}

	health_code_update();

	/* Prep stream message structure */
	consumer_init_stream_comm_msg(&lkm,
			LTTNG_CONSUMER_ADD_STREAM,
			session->metadata->fd,
			session->metadata_stream_fd,
			0); /* CPU: 0 for metadata. */

	health_code_update();

	/* Send stream and file descriptor */
	ret = consumer_send_stream(sock, consumer, &lkm,
			&session->metadata_stream_fd, 1);
	if (ret < 0) {
		goto error;
	}

	health_code_update();

error:
	return ret;
}

/*
 * Sending a single stream to the consumer with command ADD_STREAM.
 */
int kernel_consumer_add_stream(struct consumer_socket *sock,
		struct ltt_kernel_channel *channel, struct ltt_kernel_stream *stream,
		struct ltt_kernel_session *session, unsigned int monitor)
{
	int ret;
	struct lttcomm_consumer_msg lkm;
	struct consumer_output *consumer;

	assert(channel);
	assert(stream);
	assert(session);
	assert(session->consumer);
	assert(sock);

	DBG("Sending stream %d of channel %s to kernel consumer",
			stream->fd, channel->channel->name);

	/* Get consumer output pointer */
	consumer = session->consumer;

	/* Prep stream consumer message */
	consumer_init_stream_comm_msg(&lkm,
			LTTNG_CONSUMER_ADD_STREAM,
			channel->fd,
			stream->fd,
			stream->cpu);

	health_code_update();

	/* Send stream and file descriptor */
	ret = consumer_send_stream(sock, consumer, &lkm, &stream->fd, 1);
	if (ret < 0) {
		goto error;
	}

	health_code_update();

error:
	return ret;
}

/*
 * Sending the notification that all streams were sent with STREAMS_SENT.
 */
int kernel_consumer_streams_sent(struct consumer_socket *sock,
		struct ltt_kernel_session *session, uint64_t channel_key)
{
	int ret;
	struct lttcomm_consumer_msg lkm;
	struct consumer_output *consumer;

	assert(sock);
	assert(session);

	DBG("Sending streams_sent");
	/* Get consumer output pointer */
	consumer = session->consumer;

	/* Prep stream consumer message */
	consumer_init_streams_sent_comm_msg(&lkm,
			LTTNG_CONSUMER_STREAMS_SENT,
			channel_key, consumer->net_seq_index);

	health_code_update();

	/* Send stream and file descriptor */
	ret = consumer_send_msg(sock, &lkm);
	if (ret < 0) {
		goto error;
	}

error:
	return ret;
}

/*
 * Send all stream fds of kernel channel to the consumer.
 */
int kernel_consumer_send_channel_stream(struct consumer_socket *sock,
		struct ltt_kernel_channel *channel, struct ltt_kernel_session *session,
		unsigned int monitor)
{
	int ret;
	struct ltt_kernel_stream *stream;

	/* Safety net */
	assert(channel);
	assert(session);
	assert(session->consumer);
	assert(sock);

	/* Bail out if consumer is disabled */
	if (!session->consumer->enabled) {
		ret = LTTNG_OK;
		goto error;
	}

	DBG("Sending streams of channel %s to kernel consumer",
			channel->channel->name);

	ret = kernel_consumer_add_channel(sock, channel, session, monitor);
	if (ret < 0) {
		goto error;
	}

	/* Send streams */
	cds_list_for_each_entry(stream, &channel->stream_list.head, list) {
		if (!stream->fd) {
			continue;
		}

		/* Add stream on the kernel consumer side. */
		ret = kernel_consumer_add_stream(sock, channel, stream, session,
				monitor);
		if (ret < 0) {
			goto error;
		}
	}

error:
	return ret;
}

/*
 * Send all stream fds of the kernel session to the consumer.
 */
int kernel_consumer_send_session(struct consumer_socket *sock,
		struct ltt_kernel_session *session)
{
	int ret, monitor = 0;
	struct ltt_kernel_channel *chan;

	/* Safety net */
	assert(session);
	assert(session->consumer);
	assert(sock);

	/* Bail out if consumer is disabled */
	if (!session->consumer->enabled) {
		ret = LTTNG_OK;
		goto error;
	}

	/* Don't monitor the streams on the consumer if in flight recorder. */
	if (session->output_traces) {
		monitor = 1;
	}

	DBG("Sending session stream to kernel consumer");

	if (session->metadata_stream_fd >= 0) {
		ret = kernel_consumer_add_metadata(sock, session, monitor);
		if (ret < 0) {
			goto error;
		}

		/* Flag that at least the metadata has been sent to the consumer. */
		session->consumer_fds_sent = 1;
	}

	/* Send channel and streams of it */
	cds_list_for_each_entry(chan, &session->channel_list.head, list) {
		ret = kernel_consumer_send_channel_stream(sock, chan, session,
				monitor);
		if (ret < 0) {
			goto error;
		}
		if (monitor) {
			/*
			 * Inform the relay that all the streams for the
			 * channel were sent.
			 */
			ret = kernel_consumer_streams_sent(sock, session, chan->fd);
			if (ret < 0) {
				goto error;
			}
		}
	}

	DBG("Kernel consumer FDs of metadata and channel streams sent");

	return 0;

error:
	return ret;
}

int kernel_consumer_destroy_channel(struct consumer_socket *socket,
		struct ltt_kernel_channel *channel)
{
	int ret;
	struct lttcomm_consumer_msg msg;

	assert(channel);
	assert(socket);

	DBG("Sending kernel consumer destroy channel key %d", channel->fd);

	msg.cmd_type = LTTNG_CONSUMER_DESTROY_CHANNEL;
	msg.u.destroy_channel.key = channel->fd;

	pthread_mutex_lock(socket->lock);
	health_code_update();

	ret = consumer_send_msg(socket, &msg);
	if (ret < 0) {
		goto error;
	}

error:
	health_code_update();
	pthread_mutex_unlock(socket->lock);
	return ret;
}

int kernel_consumer_destroy_metadata(struct consumer_socket *socket,
		struct ltt_kernel_metadata *metadata)
{
	int ret;
	struct lttcomm_consumer_msg msg;

	assert(metadata);
	assert(socket);

	DBG("Sending kernel consumer destroy channel key %d", metadata->fd);

	msg.cmd_type = LTTNG_CONSUMER_DESTROY_CHANNEL;
	msg.u.destroy_channel.key = metadata->fd;

	pthread_mutex_lock(socket->lock);
	health_code_update();

	ret = consumer_send_msg(socket, &msg);
	if (ret < 0) {
		goto error;
	}

error:
	health_code_update();
	pthread_mutex_unlock(socket->lock);
	return ret;
}
Commit	Line	Data
f1e16794 DG	1	/*
	2	* Copyright (C) 2012 - David Goulet <dgoulet@efficios.com>
	3	*
	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.
	7	*
	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
	11	* more details.
	12	*
	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.
	16	*/
	17
	18	#define _GNU_SOURCE
	19	#include <stdio.h>
	20	#include <stdlib.h>
	21	#include <string.h>
	22	#include <sys/stat.h>
	23	#include <unistd.h>
	24
	25	#include <common/common.h>
	26	#include <common/defaults.h>
f1e16794	27
00e2e675	28	#include "consumer.h"
8782cc74	29	#include "health-sessiond.h"
f1e16794 DG	30	#include "kernel-consumer.h"
f1e16794 DG	31
2bba9e53 DG	32	static char create_channel_path(struct consumer_output consumer,
2bba9e53 DG	33	uid_t uid, gid_t gid)
00e2e675 DG	34	{
00e2e675 DG	35	int ret;
ffe60014	36	char tmp_path[PATH_MAX];
2bba9e53	37	char *pathname = NULL;
00e2e675	38
2bba9e53	39	assert(consumer);
00e2e675	40
ffe60014 DG	41	/* Get the right path name destination */
	42	if (consumer->type == CONSUMER_DST_LOCAL) {
	43	/* Set application path to the destination path */
dec56f6c	44	ret = snprintf(tmp_path, sizeof(tmp_path), "%s%s",
ffe60014 DG	45	consumer->dst.trace_path, consumer->subdir);
ffe60014 DG	46	if (ret < 0) {
2bba9e53	47	PERROR("snprintf kernel channel path");
ffe60014 DG	48	goto error;
ffe60014 DG	49	}
2bba9e53	50	pathname = strndup(tmp_path, sizeof(tmp_path));
ffe60014 DG	51
ffe60014 DG	52	/* Create directory */
2bba9e53	53	ret = run_as_mkdir_recursive(pathname, S_IRWXU \| S_IRWXG, uid, gid);
ffe60014 DG	54	if (ret < 0) {
	55	if (ret != -EEXIST) {
	56	ERR("Trace directory creation error");
	57	goto error;
	58	}
	59	}
	60	DBG3("Kernel local consumer tracefile path: %s", pathname);
	61	} else {
	62	ret = snprintf(tmp_path, sizeof(tmp_path), "%s", consumer->subdir);
	63	if (ret < 0) {
2bba9e53	64	PERROR("snprintf kernel metadata path");
ffe60014 DG	65	goto error;
ffe60014 DG	66	}
2bba9e53	67	pathname = strndup(tmp_path, sizeof(tmp_path));
ffe60014 DG	68	DBG3("Kernel network consumer subdir path: %s", pathname);
	69	}
	70
2bba9e53 DG	71	return pathname;
	72
	73	error:
	74	free(pathname);
	75	return NULL;
	76	}
	77
	78	/*
	79	* Sending a single channel to the consumer with command ADD_CHANNEL.
	80	*/
	81	int kernel_consumer_add_channel(struct consumer_socket *sock,
	82	struct ltt_kernel_channel channel, struct ltt_kernel_session session,
	83	unsigned int monitor)
	84	{
	85	int ret;
	86	char *pathname;
	87	struct lttcomm_consumer_msg lkm;
	88	struct consumer_output *consumer;
	89
	90	/* Safety net */
	91	assert(channel);
	92	assert(session);
	93	assert(session->consumer);
	94
	95	consumer = session->consumer;
	96
	97	DBG("Kernel consumer adding channel %s to kernel consumer",
	98	channel->channel->name);
	99
	100	if (monitor) {
	101	pathname = create_channel_path(consumer, session->uid, session->gid);
	102	if (!pathname) {
	103	ret = -1;
	104	goto error;
	105	}
	106	} else {
	107	/* Empty path. */
	108	pathname = "";
	109	}
	110
00e2e675 DG	111	/* Prep channel message structure */
	112	consumer_init_channel_comm_msg(&lkm,
	113	LTTNG_CONSUMER_ADD_CHANNEL,
	114	channel->fd,
ffe60014 DG	115	session->id,
	116	pathname,
	117	session->uid,
	118	session->gid,
	119	consumer->net_seq_index,
c30aaa51	120	channel->channel->name,
ffe60014 DG	121	channel->stream_count,
ffe60014 DG	122	channel->channel->attr.output,
1624d5b7 JD	123	CONSUMER_CHANNEL_TYPE_DATA,
1624d5b7 JD	124	channel->channel->attr.tracefile_size,
2bba9e53	125	channel->channel->attr.tracefile_count,
ecc48a90 JD	126	monitor,
ecc48a90 JD	127	channel->channel->attr.live_timer_interval);
00e2e675	128
840cb59c	129	health_code_update();
ca03de58	130
00e2e675 DG	131	ret = consumer_send_channel(sock, &lkm);
	132	if (ret < 0) {
	133	goto error;
	134	}
	135
840cb59c	136	health_code_update();
ca03de58	137
00e2e675 DG	138	error:
	139	return ret;
	140	}
	141
	142	/*
	143	* Sending metadata to the consumer with command ADD_CHANNEL and ADD_STREAM.
	144	*/
f50f23d9	145	int kernel_consumer_add_metadata(struct consumer_socket *sock,
2bba9e53	146	struct ltt_kernel_session *session, unsigned int monitor)
00e2e675 DG	147	{
00e2e675 DG	148	int ret;
2bba9e53	149	char *pathname;
00e2e675	150	struct lttcomm_consumer_msg lkm;
a7d9a3e7	151	struct consumer_output *consumer;
00e2e675 DG	152
	153	/* Safety net */
	154	assert(session);
	155	assert(session->consumer);
f50f23d9	156	assert(sock);
00e2e675 DG	157
	158	DBG("Sending metadata %d to kernel consumer", session->metadata_stream_fd);
	159
	160	/* Get consumer output pointer */
a7d9a3e7	161	consumer = session->consumer;
00e2e675	162
2bba9e53 DG	163	if (monitor) {
	164	pathname = create_channel_path(consumer, session->uid, session->gid);
	165	if (!pathname) {
	166	ret = -1;
a7d9a3e7 DG	167	goto error;
a7d9a3e7 DG	168	}
00e2e675	169	} else {
2bba9e53 DG	170	/* Empty path. */
2bba9e53 DG	171	pathname = "";
00e2e675 DG	172	}
	173
	174	/* Prep channel message structure */
	175	consumer_init_channel_comm_msg(&lkm,
	176	LTTNG_CONSUMER_ADD_CHANNEL,
	177	session->metadata->fd,
ffe60014 DG	178	session->id,
	179	pathname,
	180	session->uid,
	181	session->gid,
	182	consumer->net_seq_index,
30079b6b	183	DEFAULT_METADATA_NAME,
ffe60014 DG	184	1,
ffe60014 DG	185	DEFAULT_KERNEL_CHANNEL_OUTPUT,
1624d5b7	186	CONSUMER_CHANNEL_TYPE_METADATA,
2bba9e53	187	0, 0,
ecc48a90	188	monitor, 0);
00e2e675	189
840cb59c	190	health_code_update();
ca03de58	191
00e2e675 DG	192	ret = consumer_send_channel(sock, &lkm);
	193	if (ret < 0) {
	194	goto error;
	195	}
	196
840cb59c	197	health_code_update();
ca03de58	198
00e2e675 DG	199	/* Prep stream message structure */
	200	consumer_init_stream_comm_msg(&lkm,
	201	LTTNG_CONSUMER_ADD_STREAM,
	202	session->metadata->fd,
	203	session->metadata_stream_fd,
1624d5b7	204	0); /* CPU: 0 for metadata. */
00e2e675	205
840cb59c	206	health_code_update();
ca03de58	207
00e2e675	208	/* Send stream and file descriptor */
a7d9a3e7	209	ret = consumer_send_stream(sock, consumer, &lkm,
00e2e675 DG	210	&session->metadata_stream_fd, 1);
	211	if (ret < 0) {
	212	goto error;
	213	}
	214
840cb59c	215	health_code_update();
ca03de58	216
00e2e675 DG	217	error:
	218	return ret;
	219	}
	220
	221	/*
	222	* Sending a single stream to the consumer with command ADD_STREAM.
	223	*/
f50f23d9 DG	224	int kernel_consumer_add_stream(struct consumer_socket *sock,
f50f23d9 DG	225	struct ltt_kernel_channel channel, struct ltt_kernel_stream stream,
2bba9e53	226	struct ltt_kernel_session *session, unsigned int monitor)
00e2e675 DG	227	{
00e2e675 DG	228	int ret;
00e2e675	229	struct lttcomm_consumer_msg lkm;
a7d9a3e7	230	struct consumer_output *consumer;
00e2e675 DG	231
	232	assert(channel);
	233	assert(stream);
	234	assert(session);
	235	assert(session->consumer);
f50f23d9	236	assert(sock);
00e2e675 DG	237
	238	DBG("Sending stream %d of channel %s to kernel consumer",
	239	stream->fd, channel->channel->name);
	240
	241	/* Get consumer output pointer */
a7d9a3e7	242	consumer = session->consumer;
00e2e675	243
00e2e675	244	/* Prep stream consumer message */
ffe60014 DG	245	consumer_init_stream_comm_msg(&lkm,
ffe60014 DG	246	LTTNG_CONSUMER_ADD_STREAM,
00e2e675 DG	247	channel->fd,
00e2e675 DG	248	stream->fd,
ffe60014	249	stream->cpu);
00e2e675	250
840cb59c	251	health_code_update();
ca03de58	252
00e2e675	253	/* Send stream and file descriptor */
a7d9a3e7	254	ret = consumer_send_stream(sock, consumer, &lkm, &stream->fd, 1);
00e2e675 DG	255	if (ret < 0) {
	256	goto error;
	257	}
	258
840cb59c	259	health_code_update();
ca03de58	260
00e2e675 DG	261	error:
	262	return ret;
	263	}
	264
a4baae1b JD	265	/*
	266	* Sending the notification that all streams were sent with STREAMS_SENT.
	267	*/
	268	int kernel_consumer_streams_sent(struct consumer_socket *sock,
	269	struct ltt_kernel_session *session, uint64_t channel_key)
	270	{
	271	int ret;
	272	struct lttcomm_consumer_msg lkm;
	273	struct consumer_output *consumer;
	274
	275	assert(sock);
	276	assert(session);
	277
	278	DBG("Sending streams_sent");
	279	/* Get consumer output pointer */
	280	consumer = session->consumer;
	281
	282	/* Prep stream consumer message */
	283	consumer_init_streams_sent_comm_msg(&lkm,
	284	LTTNG_CONSUMER_STREAMS_SENT,
	285	channel_key, consumer->net_seq_index);
	286
	287	health_code_update();
	288
	289	/* Send stream and file descriptor */
	290	ret = consumer_send_msg(sock, &lkm);
	291	if (ret < 0) {
	292	goto error;
	293	}
	294
	295	error:
	296	return ret;
	297	}
	298
f1e16794 DG	299	/*
	300	* Send all stream fds of kernel channel to the consumer.
	301	*/
f50f23d9	302	int kernel_consumer_send_channel_stream(struct consumer_socket *sock,
2bba9e53 DG	303	struct ltt_kernel_channel channel, struct ltt_kernel_session session,
2bba9e53 DG	304	unsigned int monitor)
f1e16794	305	{
00e2e675	306	int ret;
f1e16794	307	struct ltt_kernel_stream *stream;
00e2e675 DG	308
	309	/* Safety net */
	310	assert(channel);
	311	assert(session);
	312	assert(session->consumer);
f50f23d9	313	assert(sock);
00e2e675 DG	314
	315	/* Bail out if consumer is disabled */
	316	if (!session->consumer->enabled) {
f73fabfd	317	ret = LTTNG_OK;
00e2e675 DG	318	goto error;
00e2e675 DG	319	}
f1e16794 DG	320
	321	DBG("Sending streams of channel %s to kernel consumer",
	322	channel->channel->name);
	323
2bba9e53	324	ret = kernel_consumer_add_channel(sock, channel, session, monitor);
f1e16794	325	if (ret < 0) {
f1e16794 DG	326	goto error;
	327	}
	328
	329	/* Send streams */
	330	cds_list_for_each_entry(stream, &channel->stream_list.head, list) {
	331	if (!stream->fd) {
	332	continue;
	333	}
00e2e675 DG	334
00e2e675 DG	335	/* Add stream on the kernel consumer side. */
2bba9e53 DG	336	ret = kernel_consumer_add_stream(sock, channel, stream, session,
2bba9e53 DG	337	monitor);
f1e16794	338	if (ret < 0) {
f1e16794 DG	339	goto error;
	340	}
	341	}
	342
f1e16794 DG	343	error:
	344	return ret;
	345	}
	346
	347	/*
	348	* Send all stream fds of the kernel session to the consumer.
	349	*/
f50f23d9 DG	350	int kernel_consumer_send_session(struct consumer_socket *sock,
f50f23d9 DG	351	struct ltt_kernel_session *session)
f1e16794	352	{
2bba9e53	353	int ret, monitor = 0;
f1e16794	354	struct ltt_kernel_channel *chan;
f1e16794	355
00e2e675 DG	356	/* Safety net */
	357	assert(session);
	358	assert(session->consumer);
f50f23d9	359	assert(sock);
f1e16794	360
00e2e675 DG	361	/* Bail out if consumer is disabled */
00e2e675 DG	362	if (!session->consumer->enabled) {
f73fabfd	363	ret = LTTNG_OK;
00e2e675	364	goto error;
f1e16794 DG	365	}
f1e16794 DG	366
2bba9e53 DG	367	/* Don't monitor the streams on the consumer if in flight recorder. */
	368	if (session->output_traces) {
	369	monitor = 1;
	370	}
	371
00e2e675 DG	372	DBG("Sending session stream to kernel consumer");
00e2e675 DG	373
f1e16794	374	if (session->metadata_stream_fd >= 0) {
2bba9e53	375	ret = kernel_consumer_add_metadata(sock, session, monitor);
f1e16794	376	if (ret < 0) {
f1e16794 DG	377	goto error;
	378	}
	379
00e2e675 DG	380	/* Flag that at least the metadata has been sent to the consumer. */
00e2e675 DG	381	session->consumer_fds_sent = 1;
f1e16794 DG	382	}
f1e16794 DG	383
00e2e675	384	/* Send channel and streams of it */
f1e16794	385	cds_list_for_each_entry(chan, &session->channel_list.head, list) {
2bba9e53 DG	386	ret = kernel_consumer_send_channel_stream(sock, chan, session,
2bba9e53 DG	387	monitor);
f1e16794 DG	388	if (ret < 0) {
	389	goto error;
	390	}
601262d6 JD	391	if (monitor) {
	392	/*
	393	* Inform the relay that all the streams for the
	394	* channel were sent.
	395	*/
	396	ret = kernel_consumer_streams_sent(sock, session, chan->fd);
	397	if (ret < 0) {
	398	goto error;
	399	}
	400	}
f1e16794 DG	401	}
f1e16794 DG	402
00e2e675	403	DBG("Kernel consumer FDs of metadata and channel streams sent");
f1e16794 DG	404
	405	return 0;
	406
	407	error:
	408	return ret;
	409	}
07b86b52 JD	410
	411	int kernel_consumer_destroy_channel(struct consumer_socket *socket,
	412	struct ltt_kernel_channel *channel)
	413	{
	414	int ret;
	415	struct lttcomm_consumer_msg msg;
	416
	417	assert(channel);
	418	assert(socket);
07b86b52 JD	419
	420	DBG("Sending kernel consumer destroy channel key %d", channel->fd);
	421
	422	msg.cmd_type = LTTNG_CONSUMER_DESTROY_CHANNEL;
	423	msg.u.destroy_channel.key = channel->fd;
	424
	425	pthread_mutex_lock(socket->lock);
	426	health_code_update();
	427
	428	ret = consumer_send_msg(socket, &msg);
	429	if (ret < 0) {
	430	goto error;
	431	}
	432
	433	error:
	434	health_code_update();
	435	pthread_mutex_unlock(socket->lock);
	436	return ret;
	437	}
	438
	439	int kernel_consumer_destroy_metadata(struct consumer_socket *socket,
	440	struct ltt_kernel_metadata *metadata)
	441	{
	442	int ret;
	443	struct lttcomm_consumer_msg msg;
	444
	445	assert(metadata);
	446	assert(socket);
07b86b52 JD	447
	448	DBG("Sending kernel consumer destroy channel key %d", metadata->fd);
	449
	450	msg.cmd_type = LTTNG_CONSUMER_DESTROY_CHANNEL;
	451	msg.u.destroy_channel.key = metadata->fd;
	452
	453	pthread_mutex_lock(socket->lock);
	454	health_code_update();
	455
	456	ret = consumer_send_msg(socket, &msg);
	457	if (ret < 0) {
	458	goto error;
	459	}
	460
	461	error:
	462	health_code_update();
	463	pthread_mutex_unlock(socket->lock);
	464	return ret;
	465	}