X-Git-Url: http://git.efficios.com/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-sessiond%2Fnotification-thread-commands.c;fp=src%2Fbin%2Flttng-sessiond%2Fnotification-thread-commands.c;h=3069c57d78ed4607342bb237d4a358dd244da572;hp=dc03e6a2df2fd9e7eab8634b5ece6ebb557c91de;hb=2463b7879c00298daa79744cdaae82ac061a4ed8;hpb=3a4595c2469472dee1656cde5f8882c2123efd3c

diff --git a/src/bin/lttng-sessiond/notification-thread-commands.c b/src/bin/lttng-sessiond/notification-thread-commands.c
index dc03e6a2d..3069c57d7 100644
--- a/src/bin/lttng-sessiond/notification-thread-commands.c
+++ b/src/bin/lttng-sessiond/notification-thread-commands.c
@@ -268,6 +268,112 @@ end:
 	return ret_code;
 }
 
+enum lttng_error_code notification_thread_command_add_application(
+		struct notification_thread_handle *handle,
+		int fd,
+		enum lttng_domain_type domain)
+{
+	int ret;
+	enum lttng_error_code ret_code;
+	struct notification_thread_command cmd = {};
+
+	assert(!(fd < 0));
+
+	init_notification_thread_command(&cmd);
+
+	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_APPLICATION;
+	cmd.parameters.application.read_side_trigger_event_application_pipe = fd;
+	cmd.parameters.application.domain = domain;
+
+	ret = run_command_wait(handle, &cmd);
+	if (ret) {
+		ret_code = LTTNG_ERR_UNK;
+		goto end;
+	}
+	ret_code = cmd.reply_code;
+end:
+	return ret_code;
+}
+
+enum lttng_error_code notification_thread_command_remove_application(
+		struct notification_thread_handle *handle,
+		int fd)
+{
+	int ret;
+	enum lttng_error_code ret_code;
+	struct notification_thread_command cmd = {};
+
+	init_notification_thread_command(&cmd);
+
+	cmd.type = NOTIFICATION_COMMAND_TYPE_REMOVE_APPLICATION;
+	cmd.parameters.application.read_side_trigger_event_application_pipe = fd;
+
+	ret = run_command_wait(handle, &cmd);
+	if (ret) {
+		ret_code = LTTNG_ERR_UNK;
+		goto end;
+	}
+	ret_code = cmd.reply_code;
+end:
+	return ret_code;
+}
+
+enum lttng_error_code notification_thread_command_get_tokens(
+		struct notification_thread_handle *handle,
+		struct lttng_triggers **tokens_triggers)
+{
+	int ret;
+	enum lttng_error_code ret_code;
+	struct notification_thread_command cmd = {};
+
+	assert(handle);
+	assert(tokens_triggers);
+
+	init_notification_thread_command(&cmd);
+
+	cmd.type = NOTIFICATION_COMMAND_TYPE_GET_TOKENS;
+
+	ret = run_command_wait(handle, &cmd);
+	if (ret) {
+		ret_code = LTTNG_ERR_UNK;
+		goto end;
+	}
+	ret_code = cmd.reply_code;
+	*tokens_triggers = cmd.reply.get_tokens.triggers;
+
+end:
+	return ret_code;
+}
+
+enum lttng_error_code notification_thread_command_list_triggers(
+		struct notification_thread_handle *handle,
+		uid_t uid,
+		struct lttng_triggers **triggers)
+{
+	int ret;
+	enum lttng_error_code ret_code;
+	struct notification_thread_command cmd = {};
+
+	assert(handle);
+	assert(triggers);
+
+	init_notification_thread_command(&cmd);
+
+	cmd.type = NOTIFICATION_COMMAND_TYPE_LIST_TRIGGERS;
+	cmd.parameters.list_triggers.uid = uid;
+
+	ret = run_command_wait(handle, &cmd);
+	if (ret) {
+		ret_code = LTTNG_ERR_UNK;
+		goto end;
+	}
+	ret_code = cmd.reply_code;
+	*triggers = cmd.reply.list_triggers.triggers;
+
+end:
+	return ret_code;
+}
+
 void notification_thread_command_quit(
 		struct notification_thread_handle *handle)
 {
@@ -295,3 +401,52 @@ int notification_thread_client_communication_update(
 	cmd.parameters.client_communication_update.status = transmission_status;
 	return run_command_no_wait(handle, &cmd);
 }
+
+/*
+ * Takes ownership of the payload if present.
+ */
+LTTNG_HIDDEN
+struct lttng_trigger_notification *lttng_trigger_notification_create(
+		uint64_t id,
+		enum lttng_domain_type domain,
+		char *payload,
+		size_t payload_size)
+{
+	struct lttng_trigger_notification *notification = NULL;
+
+	assert(domain != LTTNG_DOMAIN_NONE);
+
+	if (payload) {
+		assert(payload_size > 0);
+	} else {
+		assert(payload_size == 0);
+	}
+
+	notification = zmalloc(sizeof(struct lttng_trigger_notification));
+	if (notification == NULL) {
+		ERR("[notification-thread] Error allocating notification ");
+		goto end;
+	}
+
+	notification->id = id;
+	notification->type = domain;
+	notification->capture_buffer = payload;
+	notification->capture_buf_size = payload_size;
+
+end:
+	return notification;
+}
+
+LTTNG_HIDDEN
+void lttng_trigger_notification_destroy(
+		struct lttng_trigger_notification *notification)
+{
+	if (!notification) {
+		return;
+	}
+
+	if(notification->capture_buffer) {
+		free(notification->capture_buffer);
+	}
+	free(notification);
+}