[lttng-tools.git] / lttng-sessiond / context.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 <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <urcu/list.h>

#include <lttng-sessiond-comm.h>
#include <lttngerr.h>

#include "context.h"
#include "hashtable.h"
#include "kernel.h"
#include "ust-app.h"

/*
 * Add kernel context to an event of a specific channel.
 */
static int add_kctx_to_event(struct lttng_kernel_context *kctx,
		struct ltt_kernel_channel *kchan, char *event_name)
{
	int ret, found = 0;
	struct ltt_kernel_event *kevent;

	DBG("Add kernel context to event %s", event_name);

	kevent = trace_kernel_get_event_by_name(event_name, kchan);
	if (kevent != NULL) {
		ret = kernel_add_event_context(kevent, kctx);
		if (ret < 0) {
			goto error;
		}
		found = 1;
	}

	ret = found;

error:
	return ret;
}

/*
 * Add kernel context to all channel.
 *
 * If event_name is specified, add context to event instead.
 */
static int add_kctx_all_channels(struct ltt_kernel_session *ksession,
		struct lttng_kernel_context *kctx, char *event_name)
{
	int ret, no_event = 0, found = 0;
	struct ltt_kernel_channel *kchan;

	if (strlen(event_name) == 0) {
		no_event = 1;
	}

	DBG("Adding kernel context to all channels (event: %s)", event_name);

	/* Go over all channels */
	cds_list_for_each_entry(kchan, &ksession->channel_list.head, list) {
		if (no_event) {
			ret = kernel_add_channel_context(kchan, kctx);
			if (ret < 0) {
				ret = LTTCOMM_KERN_CONTEXT_FAIL;
				goto error;
			}
		} else {
			ret = add_kctx_to_event(kctx, kchan, event_name);
			if (ret < 0) {
				ret = LTTCOMM_KERN_CONTEXT_FAIL;
				goto error;
			} else if (ret == 1) {
				/* Event found and context added */
				found = 1;
				break;
			}
		}
	}

	if (!found && !no_event) {
		ret = LTTCOMM_NO_EVENT;
		goto error;
	}

	ret = LTTCOMM_OK;

error:
	return ret;
}

/*
 * Add kernel context to a specific channel.
 *
 * If event_name is specified, add context to that event.
 */
static int add_kctx_to_channel(struct lttng_kernel_context *kctx,
		struct ltt_kernel_channel *kchan, char *event_name)
{
	int ret, no_event = 0, found = 0;

	if (strlen(event_name) == 0) {
		no_event = 1;
	}

	DBG("Add kernel context to channel '%s', event '%s'",
			kchan->channel->name, event_name);

	if (no_event) {
		ret = kernel_add_channel_context(kchan, kctx);
		if (ret < 0) {
			ret = LTTCOMM_KERN_CONTEXT_FAIL;
			goto error;
		}
	} else {
		ret = add_kctx_to_event(kctx, kchan, event_name);
		if (ret < 0) {
			ret = LTTCOMM_KERN_CONTEXT_FAIL;
			goto error;
		} else if (ret == 1) {
			/* Event found and context added */
			found = 1;
		}
	}

	if (!found && !no_event) {
		ret = LTTCOMM_NO_EVENT;
		goto error;
	}

	ret = LTTCOMM_OK;

error:
	return ret;
}

/*
 * Add kernel context to tracer.
 */
int context_kernel_add(struct ltt_kernel_session *ksession,
		struct lttng_event_context *ctx, char *event_name,
		char *channel_name)
{
	int ret;
	struct ltt_kernel_channel *kchan;
	struct lttng_kernel_context kctx;

	/* Setup kernel context structure */
	kctx.ctx = ctx->ctx;
	kctx.u.perf_counter.type = ctx->u.perf_counter.type;
	kctx.u.perf_counter.config = ctx->u.perf_counter.config;
	strncpy(kctx.u.perf_counter.name, ctx->u.perf_counter.name,
			LTTNG_SYMBOL_NAME_LEN);
	kctx.u.perf_counter.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0';

	if (strlen(channel_name) == 0) {
		ret = add_kctx_all_channels(ksession, &kctx, event_name);
		if (ret != LTTCOMM_OK) {
			goto error;
		}
	} else {
		/* Get kernel channel */
		kchan = trace_kernel_get_channel_by_name(channel_name, ksession);
		if (kchan == NULL) {
			ret = LTTCOMM_KERN_CHAN_NOT_FOUND;
			goto error;
		}

		ret = add_kctx_to_channel(&kctx, kchan, event_name);
		if (ret != LTTCOMM_OK) {
			goto error;
		}
	}

	ret = LTTCOMM_OK;

error:
	return ret;
}

/*
 * Add UST context to tracer.
 */
int context_ust_add(struct ltt_ust_session *usess, int domain,
		struct lttng_event_context *ctx, char *event_name,
		char *channel_name)
{
	int ret = LTTCOMM_OK, have_event = 0;
	struct cds_lfht_iter iter, uiter;
	struct cds_lfht *chan_ht;
	struct ltt_ust_channel *uchan = NULL;
	struct ltt_ust_event *uevent = NULL;
	struct ltt_ust_context *uctx;

	switch (domain) {
	case LTTNG_DOMAIN_UST:
		chan_ht = usess->domain_global.channels;
		break;
	case LTTNG_DOMAIN_UST_EXEC_NAME:
	case LTTNG_DOMAIN_UST_PID:
	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	default:
		ret = LTTCOMM_NOT_IMPLEMENTED;
		goto error;
	}

	if (strlen(event_name) != 0) {
		have_event = 1;
	}

	/* Get UST channel if defined */
	if (strlen(channel_name) != 0) {
		uchan = trace_ust_find_channel_by_name(chan_ht, channel_name);
		if (uchan == NULL) {
			ret = LTTCOMM_UST_CHAN_NOT_FOUND;
			goto error;
		}
	}

	/* If UST channel specified and event name, get UST event ref */
	if (uchan && have_event) {
		uevent = trace_ust_find_event_by_name(uchan->events, event_name);
		if (uevent == NULL) {
			ret = LTTCOMM_UST_EVENT_NOT_FOUND;
			goto error;
		}
	}

	/* Create ltt UST context */
	uctx = trace_ust_create_context(ctx);
	if (uctx == NULL) {
		ret = LTTCOMM_FATAL;
		goto error;
	}

	/* At this point, we have 4 possibilities */

	if (uchan && uevent) {			/* Add ctx to event in channel */
		ret = ust_app_add_ctx_event_glb(usess, uchan, uevent, uctx);
	} else if (uchan && !have_event) {	/* Add ctx to channel */
		ret = ust_app_add_ctx_channel_glb(usess, uchan, uctx);
	} else if (!uchan && have_event) {	/* Add ctx to event */
		cds_lfht_for_each_entry(chan_ht, &iter, uchan, node) {
			uevent = trace_ust_find_event_by_name(uchan->events, event_name);
			if (uevent != NULL) {
				ret = ust_app_add_ctx_event_glb(usess, uchan, uevent, uctx);
				/*
				 * LTTng UST does not allowed the same event to be registered
				 * multiple time in different or the same channel. So, if we
				 * found our event in the first place, no need to continue.
				 */
				goto end;
			}
		}
		ret = LTTCOMM_UST_EVENT_NOT_FOUND;
		goto error_free_ctx;
	} else if (!uchan && !have_event) {	/* Add ctx to all events, all channels */
		cds_lfht_for_each_entry(chan_ht, &iter, uchan, node) {
			ret = ust_app_add_ctx_channel_glb(usess, uchan, uctx);
			if (ret < 0) {
				/* Add failed so uctx was not added. We can keep it. */
				continue;
			}
			cds_lfht_for_each_entry(uchan->events, &uiter, uevent, node) {
				ret = ust_app_add_ctx_event_glb(usess, uchan, uevent, uctx);
				if (ret < 0) {
					/* Add failed so uctx was not added. We can keep it. */
					continue;
				}
				/* Create ltt UST context */
				uctx = trace_ust_create_context(ctx);
				if (uctx == NULL) {
					ret = LTTCOMM_FATAL;
					goto error;
				}
			}
			/* Create ltt UST context */
			uctx = trace_ust_create_context(ctx);
			if (uctx == NULL) {
				ret = LTTCOMM_FATAL;
				goto error;
			}
		}
	}

end:
	switch (ret) {
	case -EEXIST:
		ret = LTTCOMM_UST_CONTEXT_EXIST;
		goto error_free_ctx;
	case -ENOMEM:
		ret = LTTCOMM_FATAL;
		goto error_free_ctx;
	}

	return LTTCOMM_OK;

error_free_ctx:
	free(uctx);
error:
	return ret;
}
Commit	Line	Data
b579acd9 DG	1	/*
	2	* Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
	3	*
1e307fab DG	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.
b579acd9	7	*
1e307fab DG	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.
b579acd9	12	*
1e307fab DG	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.
b579acd9 DG	16	*/
	17
	18	#define _GNU_SOURCE
	19	#include <stdio.h>
	20	#include <stdlib.h>
	21	#include <string.h>
	22	#include <unistd.h>
54d01ffb	23	#include <urcu/list.h>
1e307fab DG	24
1e307fab DG	25	#include <lttng-sessiond-comm.h>
54d01ffb	26	#include <lttngerr.h>
b579acd9	27
b579acd9	28	#include "context.h"
48842b30	29	#include "hashtable.h"
4771f025	30	#include "kernel.h"
55cc08a6	31	#include "ust-app.h"
b579acd9 DG	32
	33	/*
	34	* Add kernel context to an event of a specific channel.
	35	*/
	36	static int add_kctx_to_event(struct lttng_kernel_context *kctx,
	37	struct ltt_kernel_channel kchan, char event_name)
	38	{
	39	int ret, found = 0;
	40	struct ltt_kernel_event *kevent;
	41
	42	DBG("Add kernel context to event %s", event_name);
	43
62499ad6	44	kevent = trace_kernel_get_event_by_name(event_name, kchan);
b579acd9 DG	45	if (kevent != NULL) {
	46	ret = kernel_add_event_context(kevent, kctx);
	47	if (ret < 0) {
	48	goto error;
	49	}
	50	found = 1;
	51	}
	52
	53	ret = found;
	54
	55	error:
	56	return ret;
	57	}
	58
	59	/*
	60	* Add kernel context to all channel.
	61	*
	62	* If event_name is specified, add context to event instead.
	63	*/
	64	static int add_kctx_all_channels(struct ltt_kernel_session *ksession,
	65	struct lttng_kernel_context kctx, char event_name)
	66	{
	67	int ret, no_event = 0, found = 0;
	68	struct ltt_kernel_channel *kchan;
	69
	70	if (strlen(event_name) == 0) {
	71	no_event = 1;
	72	}
	73
	74	DBG("Adding kernel context to all channels (event: %s)", event_name);
	75
	76	/* Go over all channels */
	77	cds_list_for_each_entry(kchan, &ksession->channel_list.head, list) {
	78	if (no_event) {
	79	ret = kernel_add_channel_context(kchan, kctx);
	80	if (ret < 0) {
	81	ret = LTTCOMM_KERN_CONTEXT_FAIL;
	82	goto error;
	83	}
	84	} else {
	85	ret = add_kctx_to_event(kctx, kchan, event_name);
	86	if (ret < 0) {
	87	ret = LTTCOMM_KERN_CONTEXT_FAIL;
	88	goto error;
	89	} else if (ret == 1) {
	90	/* Event found and context added */
	91	found = 1;
	92	break;
	93	}
	94	}
	95	}
	96
	97	if (!found && !no_event) {
	98	ret = LTTCOMM_NO_EVENT;
	99	goto error;
	100	}
	101
	102	ret = LTTCOMM_OK;
	103
	104	error:
	105	return ret;
	106	}
	107
	108	/*
109	* Add kernel context to a specific channel.
110	*
111	* If event_name is specified, add context to that event.
112	*/
113	static int add_kctx_to_channel(struct lttng_kernel_context *kctx,
114	struct ltt_kernel_channel kchan, char event_name)
115	{
116	int ret, no_event = 0, found = 0;
117
118	if (strlen(event_name) == 0) {
119	no_event = 1;
120	}
121
122	DBG("Add kernel context to channel '%s', event '%s'",
123	kchan->channel->name, event_name);
124
125	if (no_event) {
126	ret = kernel_add_channel_context(kchan, kctx);
127	if (ret < 0) {
128	ret = LTTCOMM_KERN_CONTEXT_FAIL;
129	goto error;
130	}
131	} else {
132	ret = add_kctx_to_event(kctx, kchan, event_name);
133	if (ret < 0) {
134	ret = LTTCOMM_KERN_CONTEXT_FAIL;
135	goto error;
136	} else if (ret == 1) {
137	/* Event found and context added */
138	found = 1;
139	}
140	}
141
142	if (!found && !no_event) {
143	ret = LTTCOMM_NO_EVENT;
144	goto error;
145	}
146
147	ret = LTTCOMM_OK;
148
149	error:
150	return ret;
151	}
152
153	/*
154	* Add kernel context to tracer.
155	*/
54d01ffb DG	156	int context_kernel_add(struct ltt_kernel_session *ksession,
54d01ffb DG	157	struct lttng_event_context ctx, char event_name,
b579acd9 DG	158	char *channel_name)
	159	{
	160	int ret;
	161	struct ltt_kernel_channel *kchan;
54d01ffb DG	162	struct lttng_kernel_context kctx;
	163
	164	/* Setup kernel context structure */
	165	kctx.ctx = ctx->ctx;
	166	kctx.u.perf_counter.type = ctx->u.perf_counter.type;
	167	kctx.u.perf_counter.config = ctx->u.perf_counter.config;
	168	strncpy(kctx.u.perf_counter.name, ctx->u.perf_counter.name,
	169	LTTNG_SYMBOL_NAME_LEN);
	170	kctx.u.perf_counter.name[LTTNG_SYMBOL_NAME_LEN - 1] = '\0';
b579acd9 DG	171
b579acd9 DG	172	if (strlen(channel_name) == 0) {
54d01ffb	173	ret = add_kctx_all_channels(ksession, &kctx, event_name);
b579acd9 DG	174	if (ret != LTTCOMM_OK) {
	175	goto error;
	176	}
	177	} else {
	178	/* Get kernel channel */
62499ad6	179	kchan = trace_kernel_get_channel_by_name(channel_name, ksession);
b579acd9 DG	180	if (kchan == NULL) {
	181	ret = LTTCOMM_KERN_CHAN_NOT_FOUND;
	182	goto error;
	183	}
	184
54d01ffb	185	ret = add_kctx_to_channel(&kctx, kchan, event_name);
b579acd9 DG	186	if (ret != LTTCOMM_OK) {
	187	goto error;
	188	}
	189	}
	190
	191	ret = LTTCOMM_OK;
	192
	193	error:
	194	return ret;
	195	}
2bdd86d4 MD	196
2bdd86d4 MD	197	/*
55cc08a6	198	* Add UST context to tracer.
2bdd86d4	199	*/
55cc08a6 DG	200	int context_ust_add(struct ltt_ust_session *usess, int domain,
	201	struct lttng_event_context ctx, char event_name,
	202	char *channel_name)
2bdd86d4	203	{
55cc08a6 DG	204	int ret = LTTCOMM_OK, have_event = 0;
	205	struct cds_lfht_iter iter, uiter;
	206	struct cds_lfht *chan_ht;
	207	struct ltt_ust_channel *uchan = NULL;
	208	struct ltt_ust_event *uevent = NULL;
	209	struct ltt_ust_context *uctx;
2bdd86d4	210
55cc08a6 DG	211	switch (domain) {
	212	case LTTNG_DOMAIN_UST:
	213	chan_ht = usess->domain_global.channels;
	214	break;
	215	case LTTNG_DOMAIN_UST_EXEC_NAME:
	216	case LTTNG_DOMAIN_UST_PID:
	217	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	218	default:
	219	ret = LTTCOMM_NOT_IMPLEMENTED;
2bdd86d4 MD	220	goto error;
2bdd86d4 MD	221	}
2bdd86d4	222
55cc08a6 DG	223	if (strlen(event_name) != 0) {
55cc08a6 DG	224	have_event = 1;
2bdd86d4 MD	225	}
2bdd86d4 MD	226
55cc08a6 DG	227	/* Get UST channel if defined */
	228	if (strlen(channel_name) != 0) {
	229	uchan = trace_ust_find_channel_by_name(chan_ht, channel_name);
	230	if (uchan == NULL) {
	231	ret = LTTCOMM_UST_CHAN_NOT_FOUND;
2bdd86d4 MD	232	goto error;
2bdd86d4 MD	233	}
55cc08a6 DG	234	}
	235
	236	/* If UST channel specified and event name, get UST event ref */
	237	if (uchan && have_event) {
	238	uevent = trace_ust_find_event_by_name(uchan->events, event_name);
	239	if (uevent == NULL) {
	240	ret = LTTCOMM_UST_EVENT_NOT_FOUND;
2bdd86d4	241	goto error;
2bdd86d4 MD	242	}
	243	}
	244
55cc08a6 DG	245	/* Create ltt UST context */
	246	uctx = trace_ust_create_context(ctx);
	247	if (uctx == NULL) {
	248	ret = LTTCOMM_FATAL;
2bdd86d4 MD	249	goto error;
	250	}
	251
55cc08a6 DG	252	/* At this point, we have 4 possibilities */
	253
	254	if (uchan && uevent) { /* Add ctx to event in channel */
	255	ret = ust_app_add_ctx_event_glb(usess, uchan, uevent, uctx);
	256	} else if (uchan && !have_event) { /* Add ctx to channel */
	257	ret = ust_app_add_ctx_channel_glb(usess, uchan, uctx);
	258	} else if (!uchan && have_event) { /* Add ctx to event */
	259	cds_lfht_for_each_entry(chan_ht, &iter, uchan, node) {
	260	uevent = trace_ust_find_event_by_name(uchan->events, event_name);
	261	if (uevent != NULL) {
	262	ret = ust_app_add_ctx_event_glb(usess, uchan, uevent, uctx);
	263	/*
	264	* LTTng UST does not allowed the same event to be registered
	265	* multiple time in different or the same channel. So, if we
	266	* found our event in the first place, no need to continue.
	267	*/
	268	goto end;
	269	}
2bdd86d4	270	}
55cc08a6 DG	271	ret = LTTCOMM_UST_EVENT_NOT_FOUND;
	272	goto error_free_ctx;
	273	} else if (!uchan && !have_event) { /* Add ctx to all events, all channels */
	274	cds_lfht_for_each_entry(chan_ht, &iter, uchan, node) {
	275	ret = ust_app_add_ctx_channel_glb(usess, uchan, uctx);
	276	if (ret < 0) {
	277	/* Add failed so uctx was not added. We can keep it. */
	278	continue;
	279	}
	280	cds_lfht_for_each_entry(uchan->events, &uiter, uevent, node) {
	281	ret = ust_app_add_ctx_event_glb(usess, uchan, uevent, uctx);
	282	if (ret < 0) {
	283	/* Add failed so uctx was not added. We can keep it. */
	284	continue;
	285	}
	286	/* Create ltt UST context */
	287	uctx = trace_ust_create_context(ctx);
	288	if (uctx == NULL) {
	289	ret = LTTCOMM_FATAL;
	290	goto error;
	291	}
	292	}
	293	/* Create ltt UST context */
	294	uctx = trace_ust_create_context(ctx);
	295	if (uctx == NULL) {
	296	ret = LTTCOMM_FATAL;
	297	goto error;
	298	}
2bdd86d4	299	}
55cc08a6	300	}
2bdd86d4	301
55cc08a6 DG	302	end:
	303	switch (ret) {
	304	case -EEXIST:
	305	ret = LTTCOMM_UST_CONTEXT_EXIST;
	306	goto error_free_ctx;
	307	case -ENOMEM:
	308	ret = LTTCOMM_FATAL;
	309	goto error_free_ctx;
2bdd86d4 MD	310	}
2bdd86d4 MD	311
55cc08a6	312	return LTTCOMM_OK;
2bdd86d4	313
55cc08a6 DG	314	error_free_ctx:
55cc08a6 DG	315	free(uctx);
2bdd86d4 MD	316	error:
	317	return ret;
	318	}