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

/*
 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation; only version 2 of the License.
 *
 * 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., 59 Temple
 * Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#define _GNU_SOURCE
#include <string.h>
#include <unistd.h>

#include <common/common.h>
#include <common/defaults.h>
#include <common/sessiond-comm/sessiond-comm.h>

#include "channel.h"
#include "kernel.h"
#include "ust-ctl.h"
#include "utils.h"
#include "ust-app.h"

/*
 * Return allocated channel attributes.
 */
struct lttng_channel *channel_new_default_attr(int dom)
{
	struct lttng_channel *chan;

	chan = zmalloc(sizeof(struct lttng_channel));
	if (chan == NULL) {
		PERROR("zmalloc channel init");
		goto error_alloc;
	}

	if (snprintf(chan->name, sizeof(chan->name), "%s",
				DEFAULT_CHANNEL_NAME) < 0) {
		PERROR("snprintf default channel name");
		goto error;
	}

	chan->attr.overwrite = DEFAULT_CHANNEL_OVERWRITE;
	chan->attr.switch_timer_interval = DEFAULT_CHANNEL_SWITCH_TIMER;
	chan->attr.read_timer_interval = DEFAULT_CHANNEL_READ_TIMER;

	switch (dom) {
	case LTTNG_DOMAIN_KERNEL:
		chan->attr.subbuf_size = DEFAULT_KERNEL_CHANNEL_SUBBUF_SIZE;
		chan->attr.num_subbuf = DEFAULT_KERNEL_CHANNEL_SUBBUF_NUM;
		chan->attr.output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
		break;
	case LTTNG_DOMAIN_UST:
#if 0
	case LTTNG_DOMAIN_UST_PID:
	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	case LTTNG_DOMAIN_UST_EXEC_NAME:
#endif
		chan->attr.subbuf_size = DEFAULT_UST_CHANNEL_SUBBUF_SIZE;
		chan->attr.num_subbuf = DEFAULT_UST_CHANNEL_SUBBUF_NUM;
		chan->attr.output = DEFAULT_UST_CHANNEL_OUTPUT;
		break;
	default:
		goto error;	/* Not implemented */
	}

	return chan;

error:
	free(chan);
error_alloc:
	return NULL;
}

/*
 * Disable kernel channel of the kernel session.
 */
int channel_kernel_disable(struct ltt_kernel_session *ksession,
		char *channel_name)
{
	int ret;
	struct ltt_kernel_channel *kchan;

	kchan = trace_kernel_get_channel_by_name(channel_name, ksession);
	if (kchan == NULL) {
		ret = LTTCOMM_KERN_CHAN_NOT_FOUND;
		goto error;
	} else if (kchan->enabled == 1) {
		ret = kernel_disable_channel(kchan);
		if (ret < 0 && ret != -EEXIST) {
			ret = LTTCOMM_KERN_CHAN_DISABLE_FAIL;
			goto error;
		}
	}

	ret = LTTCOMM_OK;

error:
	return ret;
}

/*
 * Enable kernel channel of the kernel session.
 */
int channel_kernel_enable(struct ltt_kernel_session *ksession,
		struct ltt_kernel_channel *kchan)
{
	int ret;

	if (kchan->enabled == 0) {
		ret = kernel_enable_channel(kchan);
		if (ret < 0) {
			ret = LTTCOMM_KERN_CHAN_ENABLE_FAIL;
			goto error;
		}
	}

	ret = LTTCOMM_OK;

error:
	return ret;
}

/*
 * Create kernel channel of the kernel session and notify kernel thread.
 */
int channel_kernel_create(struct ltt_kernel_session *ksession,
		struct lttng_channel *attr, int kernel_pipe)
{
	int ret;
	struct lttng_channel *defattr = NULL;

	/* Creating channel attributes if needed */
	if (attr == NULL) {
		defattr = channel_new_default_attr(LTTNG_DOMAIN_KERNEL);
		if (defattr == NULL) {
			ret = LTTCOMM_FATAL;
			goto error;
		}
		attr = defattr;
	}

	/* Channel not found, creating it */
	ret = kernel_create_channel(ksession, attr, ksession->trace_path);
	if (ret < 0) {
		ret = LTTCOMM_KERN_CHAN_FAIL;
		goto error;
	}

	/* Notify kernel thread that there is a new channel */
	ret = notify_thread_pipe(kernel_pipe);
	if (ret < 0) {
		ret = LTTCOMM_FATAL;
		goto error;
	}

	ret = LTTCOMM_OK;
error:
	free(defattr);
	return ret;
}

/*
 * Enable UST channel for session and domain.
 */
int channel_ust_enable(struct ltt_ust_session *usess, int domain,
		struct ltt_ust_channel *uchan)
{
	int ret = LTTCOMM_OK;

	/* If already enabled, everything is OK */
	if (uchan->enabled) {
		DBG3("Channel %s already enabled. Skipping", uchan->name);
		goto end;
	}

	switch (domain) {
	case LTTNG_DOMAIN_UST:
		DBG2("Channel %s being enabled in UST global domain", uchan->name);
		/* Enable channel for global domain */
		ret = ust_app_enable_channel_glb(usess, uchan);
		break;
#if 0
	case LTTNG_DOMAIN_UST_PID:
	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	case LTTNG_DOMAIN_UST_EXEC_NAME:
#endif
	default:
		ret = LTTCOMM_UND;
		goto error;
	}

	if (ret < 0) {
		if (ret != -EEXIST) {
			ret = LTTCOMM_UST_CHAN_ENABLE_FAIL;
			goto error;
		} else {
			ret = LTTCOMM_OK;
		}
	}

	uchan->enabled = 1;
	DBG2("Channel %s enabled successfully", uchan->name);

end:
error:
	return ret;
}

/*
 * Create UST channel for session and domain.
 */
int channel_ust_create(struct ltt_ust_session *usess, int domain,
		struct lttng_channel *attr)
{
	int ret = LTTCOMM_OK;
	struct ltt_ust_channel *uchan = NULL;
	struct lttng_channel *defattr = NULL;

	/* Creating channel attributes if needed */
	if (attr == NULL) {
		defattr = channel_new_default_attr(domain);
		if (defattr == NULL) {
			ret = LTTCOMM_FATAL;
			goto error;
		}
		attr = defattr;
	}

	/* Create UST channel */
	uchan = trace_ust_create_channel(attr, usess->pathname);
	if (uchan == NULL) {
		ret = LTTCOMM_FATAL;
		goto error;
	}
	uchan->enabled = 1;

	switch (domain) {
	case LTTNG_DOMAIN_UST:
		DBG2("Channel %s being created in UST global domain", uchan->name);

		/* Enable channel for global domain */
		ret = ust_app_create_channel_glb(usess, uchan);
		break;
#if 0
	case LTTNG_DOMAIN_UST_PID:
	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	case LTTNG_DOMAIN_UST_EXEC_NAME:
#endif
	default:
		ret = LTTCOMM_UND;
		goto error_free_chan;
	}

	if (ret < 0 && ret != -EEXIST) {
		ret = LTTCOMM_UST_CHAN_ENABLE_FAIL;
		goto error_free_chan;
	}

	/* Adding the channel to the channel hash table. */
	rcu_read_lock();
	lttng_ht_add_unique_str(usess->domain_global.channels, &uchan->node);
	rcu_read_unlock();

	DBG2("Channel %s created successfully", uchan->name);

	free(defattr);
	return LTTCOMM_OK;

error_free_chan:
	/*
	 * No need to remove the channel from the hash table because at this point
	 * it was not added hence the direct call and no call_rcu().
	 */
	trace_ust_destroy_channel(uchan);
error:
	free(defattr);
	return ret;
}

/*
 * Disable UST channel for session and domain.
 */
int channel_ust_disable(struct ltt_ust_session *usess, int domain,
		struct ltt_ust_channel *uchan)
{
	int ret = LTTCOMM_OK;

	/* Already disabled */
	if (uchan->enabled == 0) {
		DBG2("Channel UST %s already disabled", uchan->name);
		goto end;
	}

	/* Get the right channel's hashtable */
	switch (domain) {
	case LTTNG_DOMAIN_UST:
		DBG2("Channel %s being disabled in UST global domain", uchan->name);
		/* Disable channel for global domain */
		ret = ust_app_disable_channel_glb(usess, uchan);
		break;
#if 0
	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	case LTTNG_DOMAIN_UST_EXEC_NAME:
	case LTTNG_DOMAIN_UST_PID:
#endif
	default:
		ret = LTTCOMM_UND;
		goto error;
	}

	if (ret < 0 && ret != -EEXIST) {
		ret = LTTCOMM_UST_DISABLE_FAIL;
		goto error;
	}

	uchan->enabled = 0;

	DBG2("Channel %s disabled successfully", uchan->name);

	return LTTCOMM_OK;

end:
error:
	return ret;
}
Commit	Line	Data
54d01ffb DG	1	/*
	2	* Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
	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 as published by the Free
	6	* Software Foundation; only version 2 of the License.
	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., 59 Temple
	15	* Place - Suite 330, Boston, MA 02111-1307, USA.
	16	*/
	17
7885e399	18	#define _GNU_SOURCE
56fff090	19	#include <string.h>
54d01ffb DG	20	#include <unistd.h>
54d01ffb DG	21
990570ed DG	22	#include <common/common.h>
990570ed DG	23	#include <common/defaults.h>
db758600	24	#include <common/sessiond-comm/sessiond-comm.h>
54d01ffb DG	25
54d01ffb DG	26	#include "channel.h"
4771f025	27	#include "kernel.h"
44d3bd01	28	#include "ust-ctl.h"
54d01ffb	29	#include "utils.h"
7885e399	30	#include "ust-app.h"
54d01ffb DG	31
	32	/*
	33	* Return allocated channel attributes.
	34	*/
f6cd6b0f	35	struct lttng_channel *channel_new_default_attr(int dom)
54d01ffb DG	36	{
	37	struct lttng_channel *chan;
	38
	39	chan = zmalloc(sizeof(struct lttng_channel));
	40	if (chan == NULL) {
7885e399	41	PERROR("zmalloc channel init");
54d01ffb DG	42	goto error_alloc;
	43	}
	44
44d3bd01 DG	45	if (snprintf(chan->name, sizeof(chan->name), "%s",
44d3bd01 DG	46	DEFAULT_CHANNEL_NAME) < 0) {
7885e399	47	PERROR("snprintf default channel name");
54d01ffb DG	48	goto error;
	49	}
	50
	51	chan->attr.overwrite = DEFAULT_CHANNEL_OVERWRITE;
	52	chan->attr.switch_timer_interval = DEFAULT_CHANNEL_SWITCH_TIMER;
	53	chan->attr.read_timer_interval = DEFAULT_CHANNEL_READ_TIMER;
	54
	55	switch (dom) {
1b1c65fa MD	56	case LTTNG_DOMAIN_KERNEL:
	57	chan->attr.subbuf_size = DEFAULT_KERNEL_CHANNEL_SUBBUF_SIZE;
	58	chan->attr.num_subbuf = DEFAULT_KERNEL_CHANNEL_SUBBUF_NUM;
	59	chan->attr.output = DEFAULT_KERNEL_CHANNEL_OUTPUT;
	60	break;
	61	case LTTNG_DOMAIN_UST:
d78d6610	62	#if 0
1b1c65fa	63	case LTTNG_DOMAIN_UST_PID:
d78d6610 DG	64	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	65	case LTTNG_DOMAIN_UST_EXEC_NAME:
	66	#endif
1b1c65fa MD	67	chan->attr.subbuf_size = DEFAULT_UST_CHANNEL_SUBBUF_SIZE;
	68	chan->attr.num_subbuf = DEFAULT_UST_CHANNEL_SUBBUF_NUM;
	69	chan->attr.output = DEFAULT_UST_CHANNEL_OUTPUT;
	70	break;
	71	default:
	72	goto error; /* Not implemented */
54d01ffb DG	73	}
	74
	75	return chan;
	76
	77	error:
	78	free(chan);
	79	error_alloc:
	80	return NULL;
	81	}
	82
	83	/*
	84	* Disable kernel channel of the kernel session.
	85	*/
	86	int channel_kernel_disable(struct ltt_kernel_session *ksession,
	87	char *channel_name)
	88	{
	89	int ret;
	90	struct ltt_kernel_channel *kchan;
	91
	92	kchan = trace_kernel_get_channel_by_name(channel_name, ksession);
	93	if (kchan == NULL) {
	94	ret = LTTCOMM_KERN_CHAN_NOT_FOUND;
	95	goto error;
	96	} else if (kchan->enabled == 1) {
	97	ret = kernel_disable_channel(kchan);
7885e399 DG	98	if (ret < 0 && ret != -EEXIST) {
7885e399 DG	99	ret = LTTCOMM_KERN_CHAN_DISABLE_FAIL;
54d01ffb DG	100	goto error;
	101	}
	102	}
	103
	104	ret = LTTCOMM_OK;
	105
	106	error:
	107	return ret;
	108	}
	109
	110	/*
	111	* Enable kernel channel of the kernel session.
	112	*/
	113	int channel_kernel_enable(struct ltt_kernel_session *ksession,
	114	struct ltt_kernel_channel *kchan)
	115	{
	116	int ret;
	117
	118	if (kchan->enabled == 0) {
	119	ret = kernel_enable_channel(kchan);
	120	if (ret < 0) {
	121	ret = LTTCOMM_KERN_CHAN_ENABLE_FAIL;
	122	goto error;
	123	}
	124	}
	125
	126	ret = LTTCOMM_OK;
	127
	128	error:
	129	return ret;
	130	}
	131
	132	/*
	133	* Create kernel channel of the kernel session and notify kernel thread.
	134	*/
	135	int channel_kernel_create(struct ltt_kernel_session *ksession,
ff4d74e6	136	struct lttng_channel *attr, int kernel_pipe)
54d01ffb DG	137	{
54d01ffb DG	138	int ret;
ff4d74e6	139	struct lttng_channel *defattr = NULL;
54d01ffb DG	140
	141	/* Creating channel attributes if needed */
	142	if (attr == NULL) {
ff4d74e6 MD	143	defattr = channel_new_default_attr(LTTNG_DOMAIN_KERNEL);
ff4d74e6 MD	144	if (defattr == NULL) {
54d01ffb DG	145	ret = LTTCOMM_FATAL;
	146	goto error;
	147	}
ff4d74e6	148	attr = defattr;
54d01ffb DG	149	}
	150
	151	/* Channel not found, creating it */
	152	ret = kernel_create_channel(ksession, attr, ksession->trace_path);
	153	if (ret < 0) {
	154	ret = LTTCOMM_KERN_CHAN_FAIL;
	155	goto error;
	156	}
	157
	158	/* Notify kernel thread that there is a new channel */
	159	ret = notify_thread_pipe(kernel_pipe);
	160	if (ret < 0) {
	161	ret = LTTCOMM_FATAL;
	162	goto error;
	163	}
	164
	165	ret = LTTCOMM_OK;
54d01ffb	166	error:
ff4d74e6	167	free(defattr);
54d01ffb DG	168	return ret;
54d01ffb DG	169	}
7885e399 DG	170
	171	/*
	172	* Enable UST channel for session and domain.
	173	*/
	174	int channel_ust_enable(struct ltt_ust_session *usess, int domain,
	175	struct ltt_ust_channel *uchan)
	176	{
	177	int ret = LTTCOMM_OK;
	178
	179	/* If already enabled, everything is OK */
	180	if (uchan->enabled) {
	181	DBG3("Channel %s already enabled. Skipping", uchan->name);
	182	goto end;
	183	}
	184
	185	switch (domain) {
	186	case LTTNG_DOMAIN_UST:
	187	DBG2("Channel %s being enabled in UST global domain", uchan->name);
	188	/* Enable channel for global domain */
	189	ret = ust_app_enable_channel_glb(usess, uchan);
	190	break;
d78d6610	191	#if 0
7885e399 DG	192	case LTTNG_DOMAIN_UST_PID:
	193	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	194	case LTTNG_DOMAIN_UST_EXEC_NAME:
d78d6610 DG	195	#endif
d78d6610 DG	196	default:
1ab1ea0b	197	ret = LTTCOMM_UND;
7885e399 DG	198	goto error;
	199	}
	200
	201	if (ret < 0) {
	202	if (ret != -EEXIST) {
	203	ret = LTTCOMM_UST_CHAN_ENABLE_FAIL;
	204	goto error;
	205	} else {
	206	ret = LTTCOMM_OK;
	207	}
	208	}
	209
	210	uchan->enabled = 1;
	211	DBG2("Channel %s enabled successfully", uchan->name);
	212
	213	end:
	214	error:
	215	return ret;
	216	}
	217
	218	/*
	219	* Create UST channel for session and domain.
	220	*/
	221	int channel_ust_create(struct ltt_ust_session *usess, int domain,
	222	struct lttng_channel *attr)
	223	{
	224	int ret = LTTCOMM_OK;
7885e399 DG	225	struct ltt_ust_channel *uchan = NULL;
	226	struct lttng_channel *defattr = NULL;
	227
	228	/* Creating channel attributes if needed */
	229	if (attr == NULL) {
	230	defattr = channel_new_default_attr(domain);
	231	if (defattr == NULL) {
	232	ret = LTTCOMM_FATAL;
	233	goto error;
	234	}
	235	attr = defattr;
	236	}
	237
	238	/* Create UST channel */
	239	uchan = trace_ust_create_channel(attr, usess->pathname);
	240	if (uchan == NULL) {
	241	ret = LTTCOMM_FATAL;
	242	goto error;
	243	}
58f3ca76	244	uchan->enabled = 1;
7885e399 DG	245
	246	switch (domain) {
	247	case LTTNG_DOMAIN_UST:
	248	DBG2("Channel %s being created in UST global domain", uchan->name);
58f3ca76	249
7885e399 DG	250	/* Enable channel for global domain */
	251	ret = ust_app_create_channel_glb(usess, uchan);
	252	break;
d78d6610	253	#if 0
7885e399 DG	254	case LTTNG_DOMAIN_UST_PID:
	255	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	256	case LTTNG_DOMAIN_UST_EXEC_NAME:
d78d6610	257	#endif
7885e399	258	default:
1ab1ea0b	259	ret = LTTCOMM_UND;
7885e399 DG	260	goto error_free_chan;
	261	}
	262
	263	if (ret < 0 && ret != -EEXIST) {
	264	ret = LTTCOMM_UST_CHAN_ENABLE_FAIL;
	265	goto error_free_chan;
	266	}
	267
fc34caaa DG	268	/* Adding the channel to the channel hash table. */
	269	rcu_read_lock();
	270	lttng_ht_add_unique_str(usess->domain_global.channels, &uchan->node);
	271	rcu_read_unlock();
	272
7885e399 DG	273	DBG2("Channel %s created successfully", uchan->name);
	274
	275	free(defattr);
	276	return LTTCOMM_OK;
	277
	278	error_free_chan:
24d1723f DG	279	/*
	280	* No need to remove the channel from the hash table because at this point
	281	* it was not added hence the direct call and no call_rcu().
	282	*/
7885e399 DG	283	trace_ust_destroy_channel(uchan);
	284	error:
	285	free(defattr);
	286	return ret;
	287	}
	288
	289	/*
	290	* Disable UST channel for session and domain.
	291	*/
	292	int channel_ust_disable(struct ltt_ust_session *usess, int domain,
	293	struct ltt_ust_channel *uchan)
	294	{
	295	int ret = LTTCOMM_OK;
7885e399 DG	296
	297	/* Already disabled */
	298	if (uchan->enabled == 0) {
	299	DBG2("Channel UST %s already disabled", uchan->name);
	300	goto end;
	301	}
	302
	303	/* Get the right channel's hashtable */
	304	switch (domain) {
	305	case LTTNG_DOMAIN_UST:
	306	DBG2("Channel %s being disabled in UST global domain", uchan->name);
7885e399 DG	307	/* Disable channel for global domain */
	308	ret = ust_app_disable_channel_glb(usess, uchan);
	309	break;
d78d6610	310	#if 0
7885e399 DG	311	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	312	case LTTNG_DOMAIN_UST_EXEC_NAME:
	313	case LTTNG_DOMAIN_UST_PID:
d78d6610 DG	314	#endif
d78d6610 DG	315	default:
1ab1ea0b	316	ret = LTTCOMM_UND;
7885e399 DG	317	goto error;
	318	}
	319
	320	if (ret < 0 && ret != -EEXIST) {
	321	ret = LTTCOMM_UST_DISABLE_FAIL;
	322	goto error;
	323	}
	324
	325	uchan->enabled = 0;
	326
	327	DBG2("Channel %s disabled successfully", uchan->name);
	328
	329	return LTTCOMM_OK;
	330
	331	end:
	332	error:
	333	return ret;
	334	}