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

diff --git a/src/common/trigger.c b/src/common/trigger.c
index 34d420807..76082a085 100644
--- a/src/common/trigger.c
+++ b/src/common/trigger.c
@@ -5,10 +5,10 @@
  *
  */
 
-#include <assert.h>
 #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,12 +19,12 @@
 #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>
 #include <pthread.h>
 
-LTTNG_HIDDEN
 bool lttng_trigger_validate(const struct lttng_trigger *trigger)
 {
 	bool valid;
@@ -117,8 +117,8 @@ static void trigger_destroy_ref(struct urcu_ref *ref)
 	struct lttng_condition *condition =
 			lttng_trigger_get_condition(trigger);
 
-	assert(action);
-	assert(condition);
+	LTTNG_ASSERT(action);
+	LTTNG_ASSERT(condition);
 
 	/* Release ownership. */
 	lttng_action_put(action);
@@ -135,7 +135,6 @@ void lttng_trigger_destroy(struct lttng_trigger *trigger)
 	lttng_trigger_put(trigger);
 }
 
-LTTNG_HIDDEN
 ssize_t lttng_trigger_create_from_payload(
 		struct lttng_payload_view *src_view,
 		struct lttng_trigger **_trigger)
@@ -286,7 +285,6 @@ end:
  * Both elements are stored contiguously, see their "*_comm" structure
  * for the detailed format.
  */
-LTTNG_HIDDEN
 int lttng_trigger_serialize(const struct lttng_trigger *trigger,
 		struct lttng_payload *payload)
 {
@@ -297,7 +295,7 @@ int lttng_trigger_serialize(const struct lttng_trigger *trigger,
 	const struct lttng_credentials *creds = NULL;
 
 	creds = lttng_trigger_get_credentials(trigger);
-	assert(creds);
+	LTTNG_ASSERT(creds);
 
 	trigger_comm.uid = LTTNG_OPTIONAL_GET(creds->uid);
 
@@ -342,11 +340,15 @@ end:
 	return ret;
 }
 
-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;
 	}
 
@@ -363,25 +365,41 @@ bool lttng_trigger_is_equal(
 		return false;
 	}
 
+	if (a->is_hidden != b->is_hidden) {
+		return false;
+	}
+
 	return true;
 }
 
+bool lttng_trigger_is_hidden(const struct lttng_trigger *trigger)
+{
+	return trigger->is_hidden;
+}
+
+void lttng_trigger_set_hidden(struct lttng_trigger *trigger)
+{
+	LTTNG_ASSERT(!trigger->is_hidden);
+	trigger->is_hidden = true;
+}
+
 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);
@@ -411,7 +429,6 @@ end:
 	return status;
 }
 
-LTTNG_HIDDEN
 int lttng_trigger_assign_name(struct lttng_trigger *dst,
 		const struct lttng_trigger *src)
 {
@@ -428,30 +445,27 @@ end:
 	return ret;
 }
 
-LTTNG_HIDDEN
 void lttng_trigger_set_tracer_token(struct lttng_trigger *trigger,
 		uint64_t token)
 {
-	assert(trigger);
+	LTTNG_ASSERT(trigger);
 	LTTNG_OPTIONAL_SET(&trigger->tracer_token, token);
 }
 
-LTTNG_HIDDEN
 uint64_t lttng_trigger_get_tracer_token(const struct lttng_trigger *trigger)
 {
-	assert(trigger);
+	LTTNG_ASSERT(trigger);
 
 	return LTTNG_OPTIONAL_GET(trigger->tracer_token);
 }
 
-LTTNG_HIDDEN
 int lttng_trigger_generate_name(struct lttng_trigger *trigger,
 		uint64_t unique_id)
 {
 	int ret = 0;
 	char *generated_name = NULL;
 
-	ret = asprintf(&generated_name, "T%" PRIu64 "", unique_id);
+	ret = asprintf(&generated_name, "trigger%" PRIu64 "", unique_id);
 	if (ret < 0) {
 		ERR("Failed to generate trigger name");
 		ret = -1;
@@ -465,13 +479,11 @@ end:
 	return ret;
 }
 
-LTTNG_HIDDEN
 void lttng_trigger_get(struct lttng_trigger *trigger)
 {
 	urcu_ref_get(&trigger->ref);
 }
 
-LTTNG_HIDDEN
 void lttng_trigger_put(struct lttng_trigger *trigger)
 {
 	if (!trigger) {
@@ -488,7 +500,6 @@ static void delete_trigger_array_element(void *ptr)
 	lttng_trigger_put(trigger);
 }
 
-LTTNG_HIDDEN
 struct lttng_triggers *lttng_triggers_create(void)
 {
 	struct lttng_triggers *triggers = NULL;
@@ -504,13 +515,12 @@ end:
 	return triggers;
 }
 
-LTTNG_HIDDEN
 struct lttng_trigger *lttng_triggers_borrow_mutable_at_index(
 		const struct lttng_triggers *triggers, unsigned int index)
 {
 	struct lttng_trigger *trigger = NULL;
 
-	assert(triggers);
+	LTTNG_ASSERT(triggers);
 	if (index >= lttng_dynamic_pointer_array_get_count(&triggers->array)) {
 		goto end;
 	}
@@ -522,14 +532,13 @@ end:
 	return trigger;
 }
 
-LTTNG_HIDDEN
 int lttng_triggers_add(
 		struct lttng_triggers *triggers, struct lttng_trigger *trigger)
 {
 	int ret;
 
-	assert(triggers);
-	assert(trigger);
+	LTTNG_ASSERT(triggers);
+	LTTNG_ASSERT(trigger);
 
 	lttng_trigger_get(trigger);
 
@@ -541,6 +550,39 @@ int lttng_triggers_add(
 	return ret;
 }
 
+int lttng_triggers_remove_hidden_triggers(struct lttng_triggers *triggers)
+{
+	int ret;
+	unsigned int trigger_count, i = 0;
+	enum lttng_trigger_status trigger_status;
+
+	LTTNG_ASSERT(triggers);
+
+	trigger_status = lttng_triggers_get_count(triggers, &trigger_count);
+	LTTNG_ASSERT(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+	while (i < trigger_count) {
+		const struct lttng_trigger *trigger =
+				lttng_triggers_get_at_index(triggers, i);
+
+		if (lttng_trigger_is_hidden(trigger)) {
+			ret = lttng_dynamic_pointer_array_remove_pointer(
+					&triggers->array, i);
+			if (ret) {
+				goto end;
+			}
+
+			trigger_count--;
+		} else {
+			i++;
+		}
+	}
+
+	ret = 0;
+end:
+	return ret;
+}
+
 const struct lttng_trigger *lttng_triggers_get_at_index(
 		const struct lttng_triggers *triggers, unsigned int index)
 {
@@ -603,7 +645,7 @@ int lttng_triggers_serialize(const struct lttng_triggers *triggers,
 		const struct lttng_trigger *trigger =
 				lttng_triggers_get_at_index(triggers, i);
 
-		assert(trigger);
+		LTTNG_ASSERT(trigger);
 
 		ret = lttng_trigger_serialize(trigger, payload);
 		if (ret) {
@@ -618,7 +660,6 @@ end:
 	return ret;
 }
 
-LTTNG_HIDDEN
 ssize_t lttng_triggers_create_from_payload(
 		struct lttng_payload_view *src_view,
 		struct lttng_triggers **triggers)
@@ -685,18 +726,16 @@ error:
 	return ret;
 }
 
-LTTNG_HIDDEN
 const struct lttng_credentials *lttng_trigger_get_credentials(
 		const struct lttng_trigger *trigger)
 {
 	return &trigger->creds;
 }
 
-LTTNG_HIDDEN
 void lttng_trigger_set_credentials(struct lttng_trigger *trigger,
 		const struct lttng_credentials *creds)
 {
-	assert(creds);
+	LTTNG_ASSERT(creds);
 	trigger->creds = *creds;
 }
 
@@ -750,7 +789,6 @@ end:
 	return ret;
 }
 
-LTTNG_HIDDEN
 enum lttng_domain_type lttng_trigger_get_underlying_domain_type_restriction(
 		const struct lttng_trigger *trigger)
 {
@@ -759,8 +797,8 @@ enum lttng_domain_type lttng_trigger_get_underlying_domain_type_restriction(
 	enum lttng_condition_status c_status;
 	enum lttng_condition_type c_type;
 
-	assert(trigger);
-	assert(trigger->condition);
+	LTTNG_ASSERT(trigger);
+	LTTNG_ASSERT(trigger->condition);
 
 	c_type = lttng_condition_get_type(trigger->condition);
 	assert (c_type != LTTNG_CONDITION_TYPE_UNKNOWN);
@@ -776,7 +814,7 @@ enum lttng_domain_type lttng_trigger_get_underlying_domain_type_restriction(
 		/* Return the domain of the event rule. */
 		c_status = lttng_condition_event_rule_matches_get_rule(
 				trigger->condition, &event_rule);
-		assert(c_status == LTTNG_CONDITION_STATUS_OK);
+		LTTNG_ASSERT(c_status == LTTNG_CONDITION_STATUS_OK);
 		type = lttng_event_rule_get_domain_type(event_rule);
 		break;
 	case LTTNG_CONDITION_TYPE_BUFFER_USAGE_HIGH:
@@ -784,7 +822,7 @@ enum lttng_domain_type lttng_trigger_get_underlying_domain_type_restriction(
 		/* Return the domain of the channel being monitored. */
 		c_status = lttng_condition_buffer_usage_get_domain_type(
 				trigger->condition, &type);
-		assert(c_status == LTTNG_CONDITION_STATUS_OK);
+		LTTNG_ASSERT(c_status == LTTNG_CONDITION_STATUS_OK);
 		break;
 	default:
 		abort();
@@ -797,7 +835,6 @@ enum lttng_domain_type lttng_trigger_get_underlying_domain_type_restriction(
  * Generate bytecode related to the trigger.
  * On success LTTNG_OK. On error, returns lttng_error code.
  */
-LTTNG_HIDDEN
 enum lttng_error_code lttng_trigger_generate_bytecode(
 		struct lttng_trigger *trigger,
 		const struct lttng_credentials *creds)
@@ -819,7 +856,7 @@ enum lttng_error_code lttng_trigger_generate_bytecode(
 				lttng_condition_event_rule_matches_borrow_rule_mutable(
 						condition, &event_rule);
 
-		assert(condition_status == LTTNG_CONDITION_STATUS_OK);
+		LTTNG_ASSERT(condition_status == LTTNG_CONDITION_STATUS_OK);
 
 		/* Generate the filter bytecode. */
 		ret = lttng_event_rule_generate_filter_bytecode(
@@ -846,16 +883,20 @@ end:
 	return ret;
 }
 
-LTTNG_HIDDEN
 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;
 	}
@@ -864,20 +905,83 @@ 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_condition_create_from_payload(
+				&view, &condition_copy);
 		if (ret < 0) {
-			copy = NULL;
 			goto end;
 		}
 	}
 
+	lttng_payload_clear(&copy_buffer);
+
+	ret = lttng_action_serialize(trigger->action, &copy_buffer);
+	if (ret < 0) {
+		goto end;
+	}
+
+	{
+		struct lttng_payload_view view =
+				lttng_payload_view_from_payload(
+						&copy_buffer, 0, -1);
+
+		ret = lttng_action_create_from_payload(
+				&view, &action_copy);
+		if (ret < 0) {
+			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;
+	copy->is_hidden = trigger->is_hidden;
+	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;
 }
 
-LTTNG_HIDDEN
 bool lttng_trigger_needs_tracer_notifier(const struct lttng_trigger *trigger)
 {
 	bool needs_tracer_notifier = false;
@@ -902,7 +1006,6 @@ end:
 	return needs_tracer_notifier;
 }
 
-LTTNG_HIDDEN
 void lttng_trigger_set_as_registered(struct lttng_trigger *trigger)
 {
 	pthread_mutex_lock(&trigger->lock);
@@ -910,7 +1013,6 @@ void lttng_trigger_set_as_registered(struct lttng_trigger *trigger)
 	pthread_mutex_unlock(&trigger->lock);
 }
 
-LTTNG_HIDDEN
 void lttng_trigger_set_as_unregistered(struct lttng_trigger *trigger)
 {
 	pthread_mutex_lock(&trigger->lock);
@@ -925,21 +1027,214 @@ void lttng_trigger_set_as_unregistered(struct lttng_trigger *trigger)
  * the trigger lock for the duration of the manipulation using
  * `lttng_trigger_lock` and `lttng_trigger_unlock`.
  */
-LTTNG_HIDDEN
 bool lttng_trigger_is_registered(struct lttng_trigger *trigger)
 {
 	ASSERT_LOCKED(trigger->lock);
 	return trigger->registered;
 }
 
-LTTNG_HIDDEN
 void lttng_trigger_lock(struct lttng_trigger *trigger)
 {
 	pthread_mutex_lock(&trigger->lock);
 }
 
-LTTNG_HIDDEN
 void lttng_trigger_unlock(struct lttng_trigger *trigger)
 {
 	pthread_mutex_unlock(&trigger->lock);
 }
+
+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;
+
+	LTTNG_ASSERT(trigger);
+	LTTNG_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);
+	LTTNG_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);
+	LTTNG_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);
+	LTTNG_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);
+	LTTNG_ASSERT(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+	trigger_status = lttng_trigger_get_name(trigger_b, &name_b);
+	LTTNG_ASSERT(trigger_status == LTTNG_TRIGGER_STATUS_OK);
+
+	return strcmp(name_a, name_b);
+}
+
+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;
+
+	LTTNG_ASSERT(triggers);
+	LTTNG_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);
+	LTTNG_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;
+}