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

/*
 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
 * Copyright (C) 2012 - Mathieu Desnoyers <mathieu.desnoyers@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 <unistd.h>
#include <urcu/list.h>

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

#include "filter.h"
#include "kernel.h"
#include "ust-app.h"
#include "trace-ust.h"

/*
 * Add UST context to event.
 */
static int add_ufilter_to_event(struct ltt_ust_session *usess, int domain,
		struct ltt_ust_channel *uchan, struct ltt_ust_event *uevent,
		struct lttng_filter_bytecode *bytecode)
{
	int ret;

	if (uevent->filter) {
		ret = -EEXIST;
		goto error;
	}
	/* Same layout. */
	uevent->filter = (struct lttng_ust_filter_bytecode *) bytecode;
	uevent->filter->seqnum = usess->filter_seq_num;

	switch (domain) {
	case LTTNG_DOMAIN_UST:
		ret = ust_app_set_filter_event_glb(usess, uchan, uevent,
						bytecode);
		if (ret < 0) {
			goto error;
		}
		usess->filter_seq_num++;
		break;
	default:
		ret = -ENOSYS;
		goto error;
	}

	DBG("Filter UST added to event %s",uevent->attr.name);

	return 0;

error:
	free(bytecode);
	return ret;
}

/*
 * Add UST context to tracer.
 */
int filter_ust_set(struct ltt_ust_session *usess, int domain,
		struct lttng_filter_bytecode *bytecode, struct lttng_event *event,
		char *channel_name)
{
	int ret = LTTNG_OK, have_event = 0;
	struct lttng_ht_iter iter;
	struct lttng_ht *chan_ht;
	struct ltt_ust_channel *uchan = NULL;
	struct ltt_ust_event *uevent = NULL;

	/*
	 * Define which channel's hashtable to use from the domain or quit if
	 * unknown domain.
	 */
	switch (domain) {
	case LTTNG_DOMAIN_UST:
		chan_ht = usess->domain_global.channels;
		break;
#if 0
	case LTTNG_DOMAIN_UST_EXEC_NAME:
	case LTTNG_DOMAIN_UST_PID:
	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
#endif
	default:
		ret = LTTNG_ERR_UND;
		goto error;
	}

	/* Do we have a valid event. */
	if (event && event->name[0] != '\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 = LTTNG_ERR_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 = LTTNG_ERR_UST_EVENT_NOT_FOUND;
			goto error;
		}
	}

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

	if (uchan && uevent) {				/* Add filter to event in channel */
		ret = add_ufilter_to_event(usess, domain, uchan, uevent,
					bytecode);
	} else if (uchan && !have_event) {	/* Add filter to channel */
		ERR("Cannot add filter to channel");
		ret = LTTNG_ERR_FATAL;	/* not supported. */
		goto error;
	} else if (!uchan && have_event) {	/* Add filter to event */
		/* Add context to event without having the channel name */
		cds_lfht_for_each_entry(chan_ht->ht, &iter.iter, uchan, node.node) {
			uevent = trace_ust_find_event_by_name(uchan->events, event->name);
			if (uevent != NULL) {
				ret = add_ufilter_to_event(usess, domain, uchan, uevent, bytecode);
				/*
				 * 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, we stop.
				 */
				goto end;
			}
		}
		ret = LTTNG_ERR_UST_EVENT_NOT_FOUND;
		goto error;
	} else if (!uchan && !have_event) {	/* Add filter all events, all channels */
		ERR("Cannot add filter to channel");
		ret = LTTNG_ERR_FATAL;	/* not supported. */
		goto error;
	}

end:
	/* Must handle both local internal error and UST code. */
	switch (ret) {
	case -EEXIST:
	case -LTTNG_UST_ERR_EXIST:
		ret = LTTNG_ERR_FILTER_EXIST;
		break;
	case -ENOMEM:
		ret = LTTNG_ERR_FATAL;
		break;
	case -EINVAL:
	case -LTTNG_UST_ERR_INVAL:
		ret = LTTNG_ERR_FILTER_INVAL;
		break;
	case -ENOSYS:
	case -LTTNG_UST_ERR_NOSYS:
		ret = LTTNG_ERR_UNKNOWN_DOMAIN;
		break;
	default:
		ret = LTTNG_OK;
		break;
	}

error:
	return ret;
}
Commit	Line	Data
53a80697 MD	1	/*
	2	* Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
	3	* Copyright (C) 2012 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
	4	*
	5	* This program is free software; you can redistribute it and/or modify
	6	* it under the terms of the GNU General Public License, version 2 only,
	7	* as published by the Free Software Foundation.
	8	*
	9	* This program is distributed in the hope that it will be useful,
	10	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	11	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	12	* GNU General Public License for more details.
	13	*
	14	* You should have received a copy of the GNU General Public License along
	15	* with this program; if not, write to the Free Software Foundation, Inc.,
	16	* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
	17	*/
	18
	19	#define _GNU_SOURCE
	20	#include <stdio.h>
	21	#include <stdlib.h>
	22	#include <string.h>
	23	#include <unistd.h>
	24	#include <urcu/list.h>
	25
	26	#include <common/error.h>
	27	#include <common/sessiond-comm/sessiond-comm.h>
	28
	29	#include "filter.h"
	30	#include "kernel.h"
	31	#include "ust-app.h"
	32	#include "trace-ust.h"
	33
	34	/*
	35	* Add UST context to event.
	36	*/
	37	static int add_ufilter_to_event(struct ltt_ust_session *usess, int domain,
	38	struct ltt_ust_channel uchan, struct ltt_ust_event uevent,
	39	struct lttng_filter_bytecode *bytecode)
	40	{
	41	int ret;
	42
	43	if (uevent->filter) {
	44	ret = -EEXIST;
	45	goto error;
	46	}
	47	/* Same layout. */
	48	uevent->filter = (struct lttng_ust_filter_bytecode *) bytecode;
f3f0db50	49	uevent->filter->seqnum = usess->filter_seq_num;
53a80697 MD	50
	51	switch (domain) {
	52	case LTTNG_DOMAIN_UST:
	53	ret = ust_app_set_filter_event_glb(usess, uchan, uevent,
	54	bytecode);
	55	if (ret < 0) {
	56	goto error;
	57	}
f3f0db50	58	usess->filter_seq_num++;
53a80697 MD	59	break;
	60	default:
	61	ret = -ENOSYS;
	62	goto error;
	63	}
	64
	65	DBG("Filter UST added to event %s",uevent->attr.name);
	66
	67	return 0;
	68
	69	error:
	70	free(bytecode);
	71	return ret;
	72	}
	73
	74	/*
	75	* Add UST context to tracer.
	76	*/
	77	int filter_ust_set(struct ltt_ust_session *usess, int domain,
178191b3	78	struct lttng_filter_bytecode bytecode, struct lttng_event event,
53a80697 MD	79	char *channel_name)
53a80697 MD	80	{
f73fabfd	81	int ret = LTTNG_OK, have_event = 0;
53a80697 MD	82	struct lttng_ht_iter iter;
	83	struct lttng_ht *chan_ht;
	84	struct ltt_ust_channel *uchan = NULL;
	85	struct ltt_ust_event *uevent = NULL;
	86
	87	/*
	88	* Define which channel's hashtable to use from the domain or quit if
	89	* unknown domain.
	90	*/
	91	switch (domain) {
	92	case LTTNG_DOMAIN_UST:
	93	chan_ht = usess->domain_global.channels;
	94	break;
	95	#if 0
	96	case LTTNG_DOMAIN_UST_EXEC_NAME:
	97	case LTTNG_DOMAIN_UST_PID:
	98	case LTTNG_DOMAIN_UST_PID_FOLLOW_CHILDREN:
	99	#endif
	100	default:
f73fabfd	101	ret = LTTNG_ERR_UND;
53a80697 MD	102	goto error;
	103	}
	104
178191b3 DG	105	/* Do we have a valid event. */
178191b3 DG	106	if (event && event->name[0] != '\0') {
53a80697 MD	107	have_event = 1;
	108	}
	109
	110	/* Get UST channel if defined */
	111	if (strlen(channel_name) != 0) {
	112	uchan = trace_ust_find_channel_by_name(chan_ht, channel_name);
	113	if (uchan == NULL) {
f73fabfd	114	ret = LTTNG_ERR_UST_CHAN_NOT_FOUND;
53a80697 MD	115	goto error;
	116	}
	117	}
	118
	119	/* If UST channel specified and event name, get UST event ref */
	120	if (uchan && have_event) {
178191b3	121	uevent = trace_ust_find_event_by_name(uchan->events, event->name);
53a80697	122	if (uevent == NULL) {
f73fabfd	123	ret = LTTNG_ERR_UST_EVENT_NOT_FOUND;
53a80697 MD	124	goto error;
	125	}
	126	}
	127
	128	/* At this point, we have 4 possibilities */
	129
	130	if (uchan && uevent) { /* Add filter to event in channel */
	131	ret = add_ufilter_to_event(usess, domain, uchan, uevent,
	132	bytecode);
	133	} else if (uchan && !have_event) { /* Add filter to channel */
	134	ERR("Cannot add filter to channel");
f73fabfd	135	ret = LTTNG_ERR_FATAL; /* not supported. */
53a80697 MD	136	goto error;
	137	} else if (!uchan && have_event) { /* Add filter to event */
	138	/* Add context to event without having the channel name */
	139	cds_lfht_for_each_entry(chan_ht->ht, &iter.iter, uchan, node.node) {
178191b3	140	uevent = trace_ust_find_event_by_name(uchan->events, event->name);
53a80697 MD	141	if (uevent != NULL) {
	142	ret = add_ufilter_to_event(usess, domain, uchan, uevent, bytecode);
	143	/*
	144	* LTTng UST does not allowed the same event to be registered
	145	* multiple time in different or the same channel. So, if we
	146	* found our event, we stop.
	147	*/
	148	goto end;
	149	}
	150	}
f73fabfd	151	ret = LTTNG_ERR_UST_EVENT_NOT_FOUND;
53a80697 MD	152	goto error;
	153	} else if (!uchan && !have_event) { /* Add filter all events, all channels */
	154	ERR("Cannot add filter to channel");
f73fabfd	155	ret = LTTNG_ERR_FATAL; /* not supported. */
53a80697 MD	156	goto error;
	157	}
	158
	159	end:
49c336c1	160	/* Must handle both local internal error and UST code. */
53a80697 MD	161	switch (ret) {
53a80697 MD	162	case -EEXIST:
49c336c1	163	case -LTTNG_UST_ERR_EXIST:
f73fabfd	164	ret = LTTNG_ERR_FILTER_EXIST;
53a80697 MD	165	break;
53a80697 MD	166	case -ENOMEM:
f73fabfd	167	ret = LTTNG_ERR_FATAL;
53a80697 MD	168	break;
53a80697 MD	169	case -EINVAL:
49c336c1	170	case -LTTNG_UST_ERR_INVAL:
f73fabfd	171	ret = LTTNG_ERR_FILTER_INVAL;
53a80697 MD	172	break;
53a80697 MD	173	case -ENOSYS:
49c336c1	174	case -LTTNG_UST_ERR_NOSYS:
f73fabfd	175	ret = LTTNG_ERR_UNKNOWN_DOMAIN;
53a80697 MD	176	break;
53a80697 MD	177	default:
f73fabfd	178	ret = LTTNG_OK;
53a80697 MD	179	break;
	180	}
	181
	182	error:
	183	return ret;
	184	}