X-Git-Url: http://git.efficios.com/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng%2Fcommands%2Fremove_trigger.c;fp=src%2Fbin%2Flttng%2Fcommands%2Fremove_trigger.c;h=e7ef293185ff1370c973848fe4f8e3025fc06d08;hp=0000000000000000000000000000000000000000;hb=2463b7879c00298daa79744cdaae82ac061a4ed8;hpb=3a4595c2469472dee1656cde5f8882c2123efd3c

diff --git a/src/bin/lttng/commands/remove_trigger.c b/src/bin/lttng/commands/remove_trigger.c
new file mode 100644
index 000000000..e7ef29318
--- /dev/null
+++ b/src/bin/lttng/commands/remove_trigger.c
@@ -0,0 +1,191 @@
+#include <stdio.h>
+
+#include "../command.h"
+
+#include "common/argpar/argpar.h"
+
+#ifdef LTTNG_EMBED_HELP
+static const char help_msg[] =
+#include <lttng-remove-trigger.1.h>
+;
+#endif
+
+enum {
+	OPT_HELP,
+	OPT_LIST_OPTIONS,
+	OPT_USER_ID,
+};
+
+static const
+struct argpar_opt_descr remove_trigger_options[] = {
+	{ OPT_HELP, 'h', "help", false },
+	{ OPT_LIST_OPTIONS, '\0', "list-options", false },
+	{ OPT_USER_ID, '\0', "user-id", true },
+	ARGPAR_OPT_DESCR_SENTINEL,
+};
+
+static
+bool assign_string(char **dest, const char *src, const char *opt_name)
+{
+	bool ret;
+
+	if (*dest) {
+		ERR(
+			"Duplicate %s given.", opt_name);
+		goto error;
+	}
+
+	*dest = strdup(src);
+	if (!*dest) {
+		ERR("Failed to allocate %s string.", opt_name);
+		goto error;
+	}
+
+	ret = true;
+	goto end;
+
+error:
+	ret = false;
+
+end:
+	return ret;
+}
+
+int cmd_remove_trigger(int argc, const char **argv)
+{
+	int ret;
+	struct argpar_parse_ret argpar_parse_ret = { 0 };
+	const char *id = NULL;
+	int i;
+	struct lttng_triggers *triggers = NULL;
+	unsigned int triggers_count;
+	enum lttng_trigger_status trigger_status;
+	const struct lttng_trigger *trigger_to_remove = NULL;
+	char *user_id = NULL;
+	long long uid;
+
+	argpar_parse_ret = argpar_parse(argc - 1, argv + 1,
+		remove_trigger_options, true);
+	if (!argpar_parse_ret.items) {
+		ERR("%s", argpar_parse_ret.error);
+		goto error;
+	}
+
+	for (i = 0; i < argpar_parse_ret.items->n_items; i++) {
+		struct argpar_item *item = argpar_parse_ret.items->items[i];
+
+		if (item->type == ARGPAR_ITEM_TYPE_OPT) {
+			struct argpar_item_opt *item_opt =
+				(struct argpar_item_opt *) item;
+
+			switch (item_opt->descr->id) {
+			case OPT_HELP:
+				SHOW_HELP();
+				ret = 0;
+				goto end;
+
+			case OPT_LIST_OPTIONS:
+				list_cmd_options_argpar(stdout,
+					remove_trigger_options);
+				ret = 0;
+				goto end;
+
+			case OPT_USER_ID:
+			{
+				if (!assign_string(&user_id, item_opt->arg,
+						    "--user-id")) {
+					goto error;
+				}
+				break;
+			}
+
+			default:
+				abort();
+			}
+
+		} else {
+			struct argpar_item_non_opt *item_non_opt =
+				(struct argpar_item_non_opt *) item;
+
+			if (id) {
+				ERR("Unexpected argument: %s", item_non_opt->arg);
+				goto error;
+			}
+
+			id = item_non_opt->arg;
+		}
+	}
+
+	if (!id) {
+		ERR("Missing `id` argument.");
+		goto error;
+	}
+
+	if (user_id) {
+		char *end;
+
+		errno = 0;
+		uid = strtol(user_id, &end, 10);
+		if (end == user_id || *end != '\0' || errno != 0) {
+			ERR("Failed to parse `%s` as an integer.", user_id);
+		}
+
+	} else {
+		uid = geteuid();
+	}
+
+	ret = lttng_list_triggers(&triggers);
+	if (ret != LTTNG_OK) {
+		ERR("Failed to get the list of triggers.");
+		goto error;
+	}
+
+	trigger_status = lttng_triggers_get_count(triggers, &triggers_count);
+	assert(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+	for (i = 0; i < triggers_count; i++) {
+		const struct lttng_trigger *trigger;
+		const char *trigger_name;
+		uid_t trigger_uid;
+
+		trigger = lttng_triggers_get_at_index(triggers, i);
+		trigger_status = lttng_trigger_get_name(trigger, &trigger_name);
+		assert(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+		trigger_status = lttng_trigger_get_user_identity(
+				trigger, &trigger_uid);
+		assert(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+		if (trigger_uid == uid &&
+				strcmp(trigger_name, id) == 0) {
+			trigger_to_remove = trigger;
+			break;
+		}
+	}
+
+	if (!trigger_to_remove) {
+		ERR("Couldn't find trigger with id `%s`.", id);
+		goto error;
+	}
+
+	ret = lttng_unregister_trigger(trigger_to_remove);
+	if (ret != 0) {
+		ERR("Failed to unregister trigger `%s`.", id);
+		goto error;
+	}
+
+	MSG("Removed trigger `%s`.", id);
+
+	ret = 0;
+	goto end;
+
+error:
+	ret = 1;
+
+end:
+	argpar_parse_ret_fini(&argpar_parse_ret);
+	lttng_triggers_destroy(triggers);
+	free(user_id);
+
+	return ret;
+}