X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Ftrigger.c;h=8a88c6d91ffb84d1e50a293dbca334b1bb4d0552;hp=34d4208076a22eec5d9e6d003ec880e887f08d40;hb=bf8ac2eaec39ba99aaac4de946bbaf3dca4f5539;hpb=8dbb86b8a981917165030d3a7fbbc2cb932888ed

diff --git a/src/common/trigger.c b/src/common/trigger.c
index 34d420807..8a88c6d91 100644
--- a/src/common/trigger.c
+++ b/src/common/trigger.c
@@ -9,6 +9,7 @@
 #include <common/credentials.h>
 #include <common/dynamic-array.h>
 #include <common/error.h>
+#include <common/mi-lttng.h>
 #include <common/optional.h>
 #include <common/payload-view.h>
 #include <common/payload.h>
@@ -19,6 +20,7 @@
 #include <lttng/condition/event-rule-matches-internal.h>
 #include <lttng/condition/event-rule-matches.h>
 #include <lttng/domain.h>
+#include <lttng/error-query-internal.h>
 #include <lttng/event-expr-internal.h>
 #include <lttng/event-rule/event-rule-internal.h>
 #include <lttng/trigger/trigger-internal.h>
@@ -346,7 +348,12 @@ LTTNG_HIDDEN
 bool lttng_trigger_is_equal(
 		const struct lttng_trigger *a, const struct lttng_trigger *b)
 {
-	if (strcmp(a->name, b->name) != 0) {
+	if (!!a->name != !!b->name) {
+		/* Both must be either anonymous or named. */
+		return false;
+	}
+
+	if (a->name && strcmp(a->name, b->name) != 0) {
 		return false;
 	}
 
@@ -366,22 +373,24 @@ bool lttng_trigger_is_equal(
 	return true;
 }
 
+LTTNG_HIDDEN
 enum lttng_trigger_status lttng_trigger_set_name(struct lttng_trigger *trigger,
 		const char* name)
 {
 	char *name_copy = NULL;
 	enum lttng_trigger_status status = LTTNG_TRIGGER_STATUS_OK;
 
-	if (!trigger || !name ||
-			strlen(name) == 0) {
+	if (!trigger) {
 		status = LTTNG_TRIGGER_STATUS_INVALID;
 		goto end;
 	}
 
-	name_copy = strdup(name);
-	if (!name_copy) {
-		status = LTTNG_TRIGGER_STATUS_ERROR;
-		goto end;
+	if (name) {
+		name_copy = strdup(name);
+		if (!name_copy) {
+			status = LTTNG_TRIGGER_STATUS_ERROR;
+			goto end;
+		}
 	}
 
 	free(trigger->name);
@@ -851,11 +860,35 @@ struct lttng_trigger *lttng_trigger_copy(const struct lttng_trigger *trigger)
 {
 	int ret;
 	struct lttng_payload copy_buffer;
+	struct lttng_condition *condition_copy = NULL;
+	struct lttng_action *action_copy = NULL;
 	struct lttng_trigger *copy = NULL;
+	enum lttng_trigger_status trigger_status;
+	const char *trigger_name;
+	uid_t trigger_owner_uid;
 
 	lttng_payload_init(&copy_buffer);
 
-	ret = lttng_trigger_serialize(trigger, &copy_buffer);
+	ret = lttng_condition_serialize(trigger->condition, &copy_buffer);
+	if (ret < 0) {
+		goto end;
+	}
+
+	{
+		struct lttng_payload_view view =
+				lttng_payload_view_from_payload(
+						&copy_buffer, 0, -1);
+
+		ret = lttng_condition_create_from_payload(
+				&view, &condition_copy);
+		if (ret < 0) {
+			goto end;
+		}
+	}
+
+	lttng_payload_clear(&copy_buffer);
+
+	ret = lttng_action_serialize(trigger->action, &copy_buffer);
 	if (ret < 0) {
 		goto end;
 	}
@@ -864,15 +897,59 @@ struct lttng_trigger *lttng_trigger_copy(const struct lttng_trigger *trigger)
 		struct lttng_payload_view view =
 				lttng_payload_view_from_payload(
 						&copy_buffer, 0, -1);
-		ret = lttng_trigger_create_from_payload(
-				&view, &copy);
+
+		ret = lttng_action_create_from_payload(
+				&view, &action_copy);
 		if (ret < 0) {
-			copy = NULL;
 			goto end;
 		}
 	}
 
+	copy = lttng_trigger_create(condition_copy, action_copy);
+	if (!copy) {
+		ERR("Failed to allocate trigger during trigger copy");
+		goto end;
+	}
+
+	trigger_status = lttng_trigger_get_name(trigger, &trigger_name);
+	switch (trigger_status) {
+	case LTTNG_TRIGGER_STATUS_OK:
+		trigger_status = lttng_trigger_set_name(copy, trigger_name);
+		if (trigger_status != LTTNG_TRIGGER_STATUS_OK) {
+			ERR("Failed to set name of new trigger during copy");
+			goto error_cleanup_trigger;
+		}
+		break;
+	case LTTNG_TRIGGER_STATUS_UNSET:
+		break;
+	default:
+		ERR("Failed to get name of original trigger during copy");
+		goto error_cleanup_trigger;
+	}
+
+	trigger_status = lttng_trigger_get_owner_uid(
+			trigger, &trigger_owner_uid);
+	switch (trigger_status) {
+	case LTTNG_TRIGGER_STATUS_OK:
+		LTTNG_OPTIONAL_SET(&copy->creds.uid, trigger_owner_uid);
+		break;
+	case LTTNG_TRIGGER_STATUS_UNSET:
+		break;
+	default:
+		ERR("Failed to get owner uid of original trigger during copy");
+		goto error_cleanup_trigger;
+	}
+
+	copy->tracer_token = trigger->tracer_token;
+	copy->registered = trigger->registered;
+	goto end;
+
+error_cleanup_trigger:
+	lttng_trigger_destroy(copy);
+	copy = NULL;
 end:
+	lttng_condition_put(condition_copy);
+	lttng_action_put(action_copy);
 	lttng_payload_reset(&copy_buffer);
 	return copy;
 }
@@ -943,3 +1020,201 @@ void lttng_trigger_unlock(struct lttng_trigger *trigger)
 {
 	pthread_mutex_unlock(&trigger->lock);
 }
+
+LTTNG_HIDDEN
+enum lttng_error_code lttng_trigger_mi_serialize(const struct lttng_trigger *trigger,
+		struct mi_writer *writer,
+		const struct mi_lttng_error_query_callbacks
+				*error_query_callbacks)
+{
+	int ret;
+	enum lttng_error_code ret_code;
+	enum lttng_trigger_status trigger_status;
+	const struct lttng_condition *condition = NULL;
+	const struct lttng_action *action = NULL;
+	struct lttng_dynamic_array action_path_indexes;
+	uid_t owner_uid;
+
+	assert(trigger);
+	assert(writer);
+
+	lttng_dynamic_array_init(&action_path_indexes, sizeof(uint64_t), NULL);
+
+	/* Open trigger element. */
+	ret = mi_lttng_writer_open_element(writer, mi_lttng_element_trigger);
+	if (ret) {
+		goto mi_error;
+	}
+
+	trigger_status = lttng_trigger_get_owner_uid(trigger, &owner_uid);
+	assert(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+	/* Name. */
+	ret = mi_lttng_writer_write_element_string(
+			writer, config_element_name, trigger->name);
+	if (ret) {
+		goto mi_error;
+	}
+
+	/* Owner uid. */
+	ret = mi_lttng_writer_write_element_signed_int(writer,
+			mi_lttng_element_trigger_owner_uid,
+			(int64_t) owner_uid);
+	if (ret) {
+		goto mi_error;
+	}
+
+	/* Condition. */
+	condition = lttng_trigger_get_const_condition(trigger);
+	assert(condition);
+	ret_code = lttng_condition_mi_serialize(
+			trigger, condition, writer, error_query_callbacks);
+	if (ret_code != LTTNG_OK) {
+		goto end;
+	}
+
+	/* Action. */
+	action = lttng_trigger_get_const_action(trigger);
+	assert(action);
+	ret_code = lttng_action_mi_serialize(trigger, action, writer,
+			error_query_callbacks, &action_path_indexes);
+	if (ret_code != LTTNG_OK) {
+		goto end;
+	}
+
+	if (error_query_callbacks && error_query_callbacks->trigger_cb) {
+		struct lttng_error_query_results *results = NULL;
+
+		ret_code = error_query_callbacks->trigger_cb(trigger, &results);
+		if (ret_code != LTTNG_OK) {
+			goto end;
+		}
+
+		ret_code = lttng_error_query_results_mi_serialize(
+				results, writer);
+		lttng_error_query_results_destroy(results);
+		if (ret_code != LTTNG_OK) {
+			goto end;
+		}
+	}
+
+	/* Close trigger element. */
+	ret = mi_lttng_writer_close_element(writer);
+	if (ret) {
+		goto mi_error;
+	}
+
+	ret_code = LTTNG_OK;
+	goto end;
+
+mi_error:
+	ret_code = LTTNG_ERR_MI_IO_FAIL;
+end:
+	lttng_dynamic_array_reset(&action_path_indexes);
+	return ret_code;
+}
+
+/* Used by qsort, which expects the semantics of strcmp(). */
+static int compare_triggers_by_name(const void *a, const void *b)
+{
+	const struct lttng_trigger *trigger_a =
+			*((const struct lttng_trigger **) a);
+	const struct lttng_trigger *trigger_b =
+			*((const struct lttng_trigger **) b);
+	const char *name_a, *name_b;
+	enum lttng_trigger_status trigger_status;
+
+	/* Anonymous triggers are not reachable here. */
+	trigger_status = lttng_trigger_get_name(trigger_a, &name_a);
+	assert(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+	trigger_status = lttng_trigger_get_name(trigger_b, &name_b);
+	assert(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+	return strcmp(name_a, name_b);
+}
+
+LTTNG_HIDDEN
+enum lttng_error_code lttng_triggers_mi_serialize(const struct lttng_triggers *triggers,
+		struct mi_writer *writer,
+		const struct mi_lttng_error_query_callbacks
+				*error_query_callbacks)
+{
+	int ret;
+	enum lttng_error_code ret_code;
+	enum lttng_trigger_status status;
+	unsigned int count, i;
+	struct lttng_dynamic_pointer_array sorted_triggers;
+
+	assert(triggers);
+	assert(writer);
+
+	/*
+	 * Sort trigger by name to ensure an order at the MI level and ignore
+	 * any anonymous trigger present.
+	 */
+	lttng_dynamic_pointer_array_init(&sorted_triggers, NULL);
+
+	status = lttng_triggers_get_count(triggers, &count);
+	assert(status == LTTNG_TRIGGER_STATUS_OK);
+
+	for (i = 0; i < count; i++) {
+		int add_ret;
+		const char *unused_name;
+		const struct lttng_trigger *trigger =
+				lttng_triggers_get_at_index(triggers, i);
+
+		status = lttng_trigger_get_name(trigger, &unused_name);
+		switch (status) {
+		case LTTNG_TRIGGER_STATUS_OK:
+			break;
+		case LTTNG_TRIGGER_STATUS_UNSET:
+			/* Don't list anonymous triggers. */
+			continue;
+		default:
+			abort();
+		}
+
+		add_ret = lttng_dynamic_pointer_array_add_pointer(
+				&sorted_triggers, (void *) trigger);
+
+		if (add_ret) {
+			ERR("Failed to lttng_trigger to sorting array.");
+			ret_code = LTTNG_ERR_NOMEM;
+			goto error;
+		}
+	}
+
+	qsort(sorted_triggers.array.buffer.data, count,
+			sizeof(struct lttng_trigger *),
+			compare_triggers_by_name);
+
+	/* Open triggers element. */
+	ret = mi_lttng_writer_open_element(writer, mi_lttng_element_triggers);
+	if (ret) {
+		ret_code = LTTNG_ERR_MI_IO_FAIL;
+		goto error;
+	}
+
+	for (i = 0; i < lttng_dynamic_pointer_array_get_count(&sorted_triggers); i++) {
+		const struct lttng_trigger *trigger =
+				(const struct lttng_trigger *)
+				lttng_dynamic_pointer_array_get_pointer(
+						&sorted_triggers, i);
+
+		lttng_trigger_mi_serialize(trigger, writer, error_query_callbacks);
+	}
+
+	/* Close triggers element. */
+	ret = mi_lttng_writer_close_element(writer);
+	if (ret) {
+		ret_code = LTTNG_ERR_MI_IO_FAIL;
+		goto error;
+	}
+
+	ret_code = LTTNG_OK;
+
+error:
+	lttng_dynamic_pointer_array_reset(&sorted_triggers);
+	return ret_code;
+}