[lttng-tools.git] / src / bin / lttng-sessiond / notification-thread-commands.cpp

/*
 * Copyright (C) 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * SPDX-License-Identifier: GPL-2.0-only
 *
 */

#include "notification-thread-commands.hpp"
#include "notification-thread.hpp"

#include <common/error.hpp>

#include <lttng/lttng-error.h>
#include <lttng/trigger/trigger.h>

#include <inttypes.h>
#include <stdint.h>
#include <unistd.h>

static void init_notification_thread_command(struct notification_thread_command *cmd)
{
	CDS_INIT_LIST_HEAD(&cmd->cmd_list_node);
}

static int run_command_wait(struct notification_thread_handle *handle,
			    struct notification_thread_command *cmd)
{
	int ret;
	uint64_t notification_counter = 1;

	lttng::synchro::waiter command_completion_waiter;
	cmd->command_completed_waker.emplace(command_completion_waiter.get_waker());

	pthread_mutex_lock(&handle->cmd_queue.lock);
	/* Add to queue. */
	cds_list_add_tail(&cmd->cmd_list_node, &handle->cmd_queue.list);
	/* Wake-up thread. */
	ret = lttng_write(
		handle->cmd_queue.event_fd, &notification_counter, sizeof(notification_counter));
	if (ret != sizeof(notification_counter)) {
		PERROR("write to notification thread's queue event fd");
		/*
		 * Remove the command from the list so the notification
		 * thread does not process it.
		 */
		cds_list_del(&cmd->cmd_list_node);
		goto error_unlock_queue;
	}
	pthread_mutex_unlock(&handle->cmd_queue.lock);

	command_completion_waiter.wait();
	;
	return 0;
error_unlock_queue:
	pthread_mutex_unlock(&handle->cmd_queue.lock);
	return -1;
}

static struct notification_thread_command *
notification_thread_command_copy(const struct notification_thread_command *original_cmd)
{
	struct notification_thread_command *new_cmd;

	try {
		new_cmd = new notification_thread_command;
	} catch (const std::bad_alloc& e) {
		ERR("Failed to allocate notification_thread_command: %s", e.what());
		return nullptr;
	}

	*new_cmd = *original_cmd;
	init_notification_thread_command(new_cmd);
	return new_cmd;
}

static int run_command_no_wait(struct notification_thread_handle *handle,
			       const struct notification_thread_command *in_cmd)
{
	int ret;
	uint64_t notification_counter = 1;
	struct notification_thread_command *new_cmd = notification_thread_command_copy(in_cmd);

	if (!new_cmd) {
		goto error;
	}
	new_cmd->is_async = true;

	pthread_mutex_lock(&handle->cmd_queue.lock);
	/* Add to queue. */
	cds_list_add_tail(&new_cmd->cmd_list_node, &handle->cmd_queue.list);
	/* Wake-up thread. */
	ret = lttng_write(
		handle->cmd_queue.event_fd, &notification_counter, sizeof(notification_counter));
	if (ret != sizeof(notification_counter)) {
		PERROR("write to notification thread's queue event fd");
		/*
		 * Remove the command from the list so the notification
		 * thread does not process it.
		 */
		cds_list_del(&new_cmd->cmd_list_node);
		goto error_unlock_queue;
	}

	pthread_mutex_unlock(&handle->cmd_queue.lock);
	return 0;
error_unlock_queue:

	delete new_cmd;
	pthread_mutex_unlock(&handle->cmd_queue.lock);
error:
	return -1;
}

enum lttng_error_code
notification_thread_command_register_trigger(struct notification_thread_handle *handle,
					     struct lttng_trigger *trigger,
					     bool is_trigger_anonymous)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	LTTNG_ASSERT(trigger);
	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_REGISTER_TRIGGER;
	lttng_trigger_get(trigger);
	cmd.parameters.register_trigger.trigger = trigger;
	cmd.parameters.register_trigger.is_trigger_anonymous = is_trigger_anonymous;

	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_unregister_trigger(struct notification_thread_handle *handle,
					       const struct lttng_trigger *trigger)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_UNREGISTER_TRIGGER;
	cmd.parameters.unregister_trigger.trigger = trigger;

	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_add_session(struct notification_thread_handle *handle,
					uint64_t session_id,
					const char *session_name,
					uid_t session_uid,
					gid_t session_gid)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_SESSION;
	cmd.parameters.add_session.session_id = session_id;
	cmd.parameters.add_session.session_name = session_name;
	cmd.parameters.add_session.session_uid = session_uid;
	cmd.parameters.add_session.session_gid = session_gid;

	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_session(struct notification_thread_handle *handle,
					   uint64_t session_id)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_REMOVE_SESSION;
	cmd.parameters.remove_session.session_id = session_id;

	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_add_channel(struct notification_thread_handle *handle,
					uint64_t session_id,
					char *channel_name,
					uint64_t key,
					enum lttng_domain_type domain,
					uint64_t capacity)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_CHANNEL;
	cmd.parameters.add_channel.session.id = session_id;
	cmd.parameters.add_channel.channel.name = channel_name;
	cmd.parameters.add_channel.channel.key = key;
	cmd.parameters.add_channel.channel.domain = domain;
	cmd.parameters.add_channel.channel.capacity = capacity;

	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_channel(
	struct notification_thread_handle *handle, uint64_t key, enum lttng_domain_type domain)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_REMOVE_CHANNEL;
	cmd.parameters.remove_channel.key = key;
	cmd.parameters.remove_channel.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_session_rotation_ongoing(struct notification_thread_handle *handle,
						     uint64_t session_id,
						     uint64_t trace_archive_chunk_id)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_SESSION_ROTATION_ONGOING;
	cmd.parameters.session_rotation.session_id = session_id;
	cmd.parameters.session_rotation.trace_archive_chunk_id = trace_archive_chunk_id;

	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_session_rotation_completed(
	struct notification_thread_handle *handle,
	uint64_t session_id,
	uint64_t trace_archive_chunk_id,
	struct lttng_trace_archive_location *location)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_SESSION_ROTATION_COMPLETED;
	cmd.parameters.session_rotation.session_id = session_id;
	cmd.parameters.session_rotation.trace_archive_chunk_id = trace_archive_chunk_id;
	cmd.parameters.session_rotation.location = location;

	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_add_tracer_event_source(struct notification_thread_handle *handle,
						    int tracer_event_source_fd,
						    enum lttng_domain_type domain)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	LTTNG_ASSERT(tracer_event_source_fd >= 0);

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_TRACER_EVENT_SOURCE;
	cmd.parameters.tracer_event_source.tracer_event_source_fd = tracer_event_source_fd;
	cmd.parameters.tracer_event_source.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_tracer_event_source(struct notification_thread_handle *handle,
						       int tracer_event_source_fd)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_REMOVE_TRACER_EVENT_SOURCE;
	cmd.parameters.tracer_event_source.tracer_event_source_fd = tracer_event_source_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_list_triggers(
	struct notification_thread_handle *handle, uid_t uid, struct lttng_triggers **triggers)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	LTTNG_ASSERT(handle);
	LTTNG_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)
{
	int ret;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_QUIT;
	ret = run_command_wait(handle, &cmd);
	LTTNG_ASSERT(!ret && cmd.reply_code == LTTNG_OK);
}

int notification_thread_client_communication_update(
	struct notification_thread_handle *handle,
	notification_client_id id,
	enum client_transmission_status transmission_status)
{
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_CLIENT_COMMUNICATION_UPDATE;
	cmd.parameters.client_communication_update.id = id;
	cmd.parameters.client_communication_update.status = transmission_status;
	return run_command_no_wait(handle, &cmd);
}

enum lttng_error_code
notification_thread_command_get_trigger(struct notification_thread_handle *handle,
					const struct lttng_trigger *trigger,
					struct lttng_trigger **real_trigger)
{
	int ret;
	enum lttng_error_code ret_code;
	notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_GET_TRIGGER;
	cmd.parameters.get_trigger.trigger = trigger;
	ret = run_command_wait(handle, &cmd);
	if (ret) {
		ret_code = LTTNG_ERR_UNK;
		goto end;
	}

	ret_code = cmd.reply_code;
	*real_trigger = cmd.reply.get_trigger.trigger;

end:
	return ret_code;
}

/*
 * Takes ownership of the payload if present.
 */
struct lttng_event_notifier_notification *lttng_event_notifier_notification_create(
	uint64_t tracer_token, enum lttng_domain_type domain, char *payload, size_t payload_size)
{
	struct lttng_event_notifier_notification *notification = nullptr;

	LTTNG_ASSERT(domain != LTTNG_DOMAIN_NONE);
	LTTNG_ASSERT((payload && payload_size) || (!payload && !payload_size));

	notification = zmalloc<lttng_event_notifier_notification>();
	if (notification == nullptr) {
		ERR("Error allocating notification");
		goto end;
	}

	notification->tracer_token = tracer_token;
	notification->type = domain;
	notification->capture_buffer = payload;
	notification->capture_buf_size = payload_size;

end:
	return notification;
}

void lttng_event_notifier_notification_destroy(
	struct lttng_event_notifier_notification *notification)
{
	if (!notification) {
		return;
	}

	free(notification->capture_buffer);
	free(notification);
}
Commit	Line	Data
ab0ee2ca	1	/*
ab5be9fa	2	* Copyright (C) 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
ab0ee2ca	3	*
ab5be9fa	4	* SPDX-License-Identifier: GPL-2.0-only
ab0ee2ca	5	*
ab0ee2ca JG	6	*/
ab0ee2ca JG	7
c9e313bc	8	#include "notification-thread-commands.hpp"
28ab034a JG	9	#include "notification-thread.hpp"
28ab034a JG	10
c9e313bc	11	#include <common/error.hpp>
28ab034a JG	12
	13	#include <lttng/lttng-error.h>
	14	#include <lttng/trigger/trigger.h>
	15
ab0ee2ca	16	#include <inttypes.h>
28ab034a JG	17	#include <stdint.h>
28ab034a JG	18	#include <unistd.h>
ab0ee2ca	19
28ab034a	20	static void init_notification_thread_command(struct notification_thread_command *cmd)
ab0ee2ca	21	{
ab0ee2ca JG	22	CDS_INIT_LIST_HEAD(&cmd->cmd_list_node);
	23	}
	24
28ab034a JG	25	static int run_command_wait(struct notification_thread_handle *handle,
28ab034a JG	26	struct notification_thread_command *cmd)
ab0ee2ca JG	27	{
	28	int ret;
	29	uint64_t notification_counter = 1;
	30
32670d71 JG	31	lttng::synchro::waiter command_completion_waiter;
	32	cmd->command_completed_waker.emplace(command_completion_waiter.get_waker());
	33
ab0ee2ca JG	34	pthread_mutex_lock(&handle->cmd_queue.lock);
ab0ee2ca JG	35	/* Add to queue. */
28ab034a	36	cds_list_add_tail(&cmd->cmd_list_node, &handle->cmd_queue.list);
ab0ee2ca	37	/* Wake-up thread. */
28ab034a JG	38	ret = lttng_write(
28ab034a JG	39	handle->cmd_queue.event_fd, &notification_counter, sizeof(notification_counter));
58d3fed5	40	if (ret != sizeof(notification_counter)) {
ab0ee2ca JG	41	PERROR("write to notification thread's queue event fd");
	42	/*
	43	* Remove the command from the list so the notification
	44	* thread does not process it.
	45	*/
	46	cds_list_del(&cmd->cmd_list_node);
	47	goto error_unlock_queue;
	48	}
	49	pthread_mutex_unlock(&handle->cmd_queue.lock);
	50
11927a78 JG	51	command_completion_waiter.wait();
11927a78 JG	52	;
ab0ee2ca JG	53	return 0;
	54	error_unlock_queue:
	55	pthread_mutex_unlock(&handle->cmd_queue.lock);
	56	return -1;
	57	}
	58
28ab034a JG	59	static struct notification_thread_command *
28ab034a JG	60	notification_thread_command_copy(const struct notification_thread_command *original_cmd)
0ab399e0 JG	61	{
	62	struct notification_thread_command *new_cmd;
	63
32670d71 JG	64	try {
32670d71 JG	65	new_cmd = new notification_thread_command;
11927a78	66	} catch (const std::bad_alloc& e) {
32670d71 JG	67	ERR("Failed to allocate notification_thread_command: %s", e.what());
32670d71 JG	68	return nullptr;
0ab399e0 JG	69	}
	70
	71	new_cmd = original_cmd;
	72	init_notification_thread_command(new_cmd);
0ab399e0 JG	73	return new_cmd;
	74	}
	75
28ab034a JG	76	static int run_command_no_wait(struct notification_thread_handle *handle,
28ab034a JG	77	const struct notification_thread_command *in_cmd)
0ab399e0 JG	78	{
	79	int ret;
	80	uint64_t notification_counter = 1;
28ab034a	81	struct notification_thread_command *new_cmd = notification_thread_command_copy(in_cmd);
0ab399e0 JG	82
	83	if (!new_cmd) {
	84	goto error;
	85	}
	86	new_cmd->is_async = true;
	87
	88	pthread_mutex_lock(&handle->cmd_queue.lock);
	89	/* Add to queue. */
28ab034a	90	cds_list_add_tail(&new_cmd->cmd_list_node, &handle->cmd_queue.list);
0ab399e0	91	/* Wake-up thread. */
28ab034a JG	92	ret = lttng_write(
28ab034a JG	93	handle->cmd_queue.event_fd, &notification_counter, sizeof(notification_counter));
0ab399e0 JG	94	if (ret != sizeof(notification_counter)) {
	95	PERROR("write to notification thread's queue event fd");
	96	/*
	97	* Remove the command from the list so the notification
	98	* thread does not process it.
	99	*/
	100	cds_list_del(&new_cmd->cmd_list_node);
	101	goto error_unlock_queue;
	102	}
32670d71	103
0ab399e0 JG	104	pthread_mutex_unlock(&handle->cmd_queue.lock);
	105	return 0;
	106	error_unlock_queue:
32670d71 JG	107
32670d71 JG	108	delete new_cmd;
0ab399e0 JG	109	pthread_mutex_unlock(&handle->cmd_queue.lock);
	110	error:
	111	return -1;
	112	}
	113
28ab034a JG	114	enum lttng_error_code
	115	notification_thread_command_register_trigger(struct notification_thread_handle *handle,
	116	struct lttng_trigger *trigger,
	117	bool is_trigger_anonymous)
ab0ee2ca JG	118	{
	119	int ret;
	120	enum lttng_error_code ret_code;
32670d71	121	notification_thread_command cmd;
ab0ee2ca	122
a0377dfe	123	LTTNG_ASSERT(trigger);
ab0ee2ca JG	124	init_notification_thread_command(&cmd);
	125
	126	cmd.type = NOTIFICATION_COMMAND_TYPE_REGISTER_TRIGGER;
242388e4	127	lttng_trigger_get(trigger);
ac16173e	128	cmd.parameters.register_trigger.trigger = trigger;
28ab034a	129	cmd.parameters.register_trigger.is_trigger_anonymous = is_trigger_anonymous;
ab0ee2ca JG	130
	131	ret = run_command_wait(handle, &cmd);
	132	if (ret) {
	133	ret_code = LTTNG_ERR_UNK;
	134	goto end;
	135	}
	136	ret_code = cmd.reply_code;
	137	end:
	138	return ret_code;
	139	}
	140
28ab034a JG	141	enum lttng_error_code
	142	notification_thread_command_unregister_trigger(struct notification_thread_handle *handle,
	143	const struct lttng_trigger *trigger)
ab0ee2ca JG	144	{
	145	int ret;
	146	enum lttng_error_code ret_code;
32670d71	147	notification_thread_command cmd;
ab0ee2ca JG	148
	149	init_notification_thread_command(&cmd);
	150
	151	cmd.type = NOTIFICATION_COMMAND_TYPE_UNREGISTER_TRIGGER;
ac16173e	152	cmd.parameters.unregister_trigger.trigger = trigger;
ab0ee2ca JG	153
	154	ret = run_command_wait(handle, &cmd);
	155	if (ret) {
	156	ret_code = LTTNG_ERR_UNK;
	157	goto end;
	158	}
	159	ret_code = cmd.reply_code;
	160	end:
	161	return ret_code;
	162	}
	163
28ab034a JG	164	enum lttng_error_code
	165	notification_thread_command_add_session(struct notification_thread_handle *handle,
	166	uint64_t session_id,
	167	const char *session_name,
	168	uid_t session_uid,
	169	gid_t session_gid)
139a8d25 JG	170	{
	171	int ret;
	172	enum lttng_error_code ret_code;
32670d71	173	notification_thread_command cmd;
139a8d25 JG	174
	175	init_notification_thread_command(&cmd);
	176
	177	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_SESSION;
	178	cmd.parameters.add_session.session_id = session_id;
	179	cmd.parameters.add_session.session_name = session_name;
	180	cmd.parameters.add_session.session_uid = session_uid;
	181	cmd.parameters.add_session.session_gid = session_gid;
	182
	183	ret = run_command_wait(handle, &cmd);
	184	if (ret) {
	185	ret_code = LTTNG_ERR_UNK;
	186	goto end;
	187	}
	188	ret_code = cmd.reply_code;
	189	end:
	190	return ret_code;
	191	}
	192
28ab034a JG	193	enum lttng_error_code
	194	notification_thread_command_remove_session(struct notification_thread_handle *handle,
	195	uint64_t session_id)
139a8d25 JG	196	{
	197	int ret;
	198	enum lttng_error_code ret_code;
32670d71	199	notification_thread_command cmd;
139a8d25 JG	200
	201	init_notification_thread_command(&cmd);
	202
	203	cmd.type = NOTIFICATION_COMMAND_TYPE_REMOVE_SESSION;
	204	cmd.parameters.remove_session.session_id = session_id;
	205
	206	ret = run_command_wait(handle, &cmd);
	207	if (ret) {
	208	ret_code = LTTNG_ERR_UNK;
	209	goto end;
	210	}
	211	ret_code = cmd.reply_code;
	212	end:
	213	return ret_code;
	214	}
	215
28ab034a JG	216	enum lttng_error_code
	217	notification_thread_command_add_channel(struct notification_thread_handle *handle,
	218	uint64_t session_id,
	219	char *channel_name,
	220	uint64_t key,
	221	enum lttng_domain_type domain,
	222	uint64_t capacity)
ab0ee2ca JG	223	{
	224	int ret;
	225	enum lttng_error_code ret_code;
32670d71	226	notification_thread_command cmd;
ab0ee2ca JG	227
	228	init_notification_thread_command(&cmd);
	229
	230	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_CHANNEL;
139a8d25	231	cmd.parameters.add_channel.session.id = session_id;
8abe313a JG	232	cmd.parameters.add_channel.channel.name = channel_name;
	233	cmd.parameters.add_channel.channel.key = key;
	234	cmd.parameters.add_channel.channel.domain = domain;
	235	cmd.parameters.add_channel.channel.capacity = capacity;
ab0ee2ca JG	236
	237	ret = run_command_wait(handle, &cmd);
	238	if (ret) {
	239	ret_code = LTTNG_ERR_UNK;
	240	goto end;
	241	}
	242	ret_code = cmd.reply_code;
	243	end:
	244	return ret_code;
	245	}
	246
	247	enum lttng_error_code notification_thread_command_remove_channel(
28ab034a	248	struct notification_thread_handle *handle, uint64_t key, enum lttng_domain_type domain)
ab0ee2ca JG	249	{
	250	int ret;
	251	enum lttng_error_code ret_code;
32670d71	252	notification_thread_command cmd;
ab0ee2ca JG	253
	254	init_notification_thread_command(&cmd);
	255
	256	cmd.type = NOTIFICATION_COMMAND_TYPE_REMOVE_CHANNEL;
	257	cmd.parameters.remove_channel.key = key;
	258	cmd.parameters.remove_channel.domain = domain;
	259
	260	ret = run_command_wait(handle, &cmd);
	261	if (ret) {
	262	ret_code = LTTNG_ERR_UNK;
	263	goto end;
	264	}
	265	ret_code = cmd.reply_code;
	266	end:
	267	return ret_code;
	268	}
	269
28ab034a JG	270	enum lttng_error_code
	271	notification_thread_command_session_rotation_ongoing(struct notification_thread_handle *handle,
	272	uint64_t session_id,
	273	uint64_t trace_archive_chunk_id)
731c1b12 JG	274	{
	275	int ret;
	276	enum lttng_error_code ret_code;
32670d71	277	notification_thread_command cmd;
731c1b12 JG	278
	279	init_notification_thread_command(&cmd);
	280
	281	cmd.type = NOTIFICATION_COMMAND_TYPE_SESSION_ROTATION_ONGOING;
139a8d25	282	cmd.parameters.session_rotation.session_id = session_id;
28ab034a	283	cmd.parameters.session_rotation.trace_archive_chunk_id = trace_archive_chunk_id;
731c1b12 JG	284
	285	ret = run_command_wait(handle, &cmd);
	286	if (ret) {
	287	ret_code = LTTNG_ERR_UNK;
	288	goto end;
	289	}
	290	ret_code = cmd.reply_code;
	291	end:
	292	return ret_code;
	293	}
	294
	295	enum lttng_error_code notification_thread_command_session_rotation_completed(
28ab034a JG	296	struct notification_thread_handle *handle,
	297	uint64_t session_id,
	298	uint64_t trace_archive_chunk_id,
	299	struct lttng_trace_archive_location *location)
731c1b12 JG	300	{
	301	int ret;
	302	enum lttng_error_code ret_code;
32670d71	303	notification_thread_command cmd;
731c1b12 JG	304
	305	init_notification_thread_command(&cmd);
	306
	307	cmd.type = NOTIFICATION_COMMAND_TYPE_SESSION_ROTATION_COMPLETED;
139a8d25	308	cmd.parameters.session_rotation.session_id = session_id;
28ab034a	309	cmd.parameters.session_rotation.trace_archive_chunk_id = trace_archive_chunk_id;
731c1b12 JG	310	cmd.parameters.session_rotation.location = location;
	311
	312	ret = run_command_wait(handle, &cmd);
	313	if (ret) {
	314	ret_code = LTTNG_ERR_UNK;
	315	goto end;
	316	}
	317	ret_code = cmd.reply_code;
	318	end:
	319	return ret_code;
	320	}
	321
28ab034a JG	322	enum lttng_error_code
	323	notification_thread_command_add_tracer_event_source(struct notification_thread_handle *handle,
	324	int tracer_event_source_fd,
	325	enum lttng_domain_type domain)
d02d7404 JR	326	{
	327	int ret;
	328	enum lttng_error_code ret_code;
32670d71	329	notification_thread_command cmd;
d02d7404	330
a0377dfe	331	LTTNG_ASSERT(tracer_event_source_fd >= 0);
d02d7404 JR	332
	333	init_notification_thread_command(&cmd);
	334
	335	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_TRACER_EVENT_SOURCE;
28ab034a	336	cmd.parameters.tracer_event_source.tracer_event_source_fd = tracer_event_source_fd;
d02d7404 JR	337	cmd.parameters.tracer_event_source.domain = domain;
	338
	339	ret = run_command_wait(handle, &cmd);
	340	if (ret) {
	341	ret_code = LTTNG_ERR_UNK;
	342	goto end;
	343	}
	344
	345	ret_code = cmd.reply_code;
	346	end:
	347	return ret_code;
	348	}
	349
28ab034a JG	350	enum lttng_error_code
	351	notification_thread_command_remove_tracer_event_source(struct notification_thread_handle *handle,
	352	int tracer_event_source_fd)
d02d7404 JR	353	{
	354	int ret;
	355	enum lttng_error_code ret_code;
32670d71	356	notification_thread_command cmd;
d02d7404 JR	357
	358	init_notification_thread_command(&cmd);
	359
	360	cmd.type = NOTIFICATION_COMMAND_TYPE_REMOVE_TRACER_EVENT_SOURCE;
28ab034a	361	cmd.parameters.tracer_event_source.tracer_event_source_fd = tracer_event_source_fd;
d02d7404 JR	362
	363	ret = run_command_wait(handle, &cmd);
	364	if (ret) {
	365	ret_code = LTTNG_ERR_UNK;
	366	goto end;
	367	}
	368
	369	ret_code = cmd.reply_code;
	370	end:
	371	return ret_code;
	372	}
	373
fbc9f37d	374	enum lttng_error_code notification_thread_command_list_triggers(
28ab034a	375	struct notification_thread_handle handle, uid_t uid, struct lttng_triggers *triggers)
fbc9f37d JR	376	{
	377	int ret;
	378	enum lttng_error_code ret_code;
32670d71	379	notification_thread_command cmd;
fbc9f37d	380
a0377dfe FD	381	LTTNG_ASSERT(handle);
a0377dfe FD	382	LTTNG_ASSERT(triggers);
fbc9f37d JR	383
	384	init_notification_thread_command(&cmd);
	385
	386	cmd.type = NOTIFICATION_COMMAND_TYPE_LIST_TRIGGERS;
	387	cmd.parameters.list_triggers.uid = uid;
	388
	389	ret = run_command_wait(handle, &cmd);
	390	if (ret) {
	391	ret_code = LTTNG_ERR_UNK;
	392	goto end;
	393	}
	394
	395	ret_code = cmd.reply_code;
	396	*triggers = cmd.reply.list_triggers.triggers;
	397
	398	end:
	399	return ret_code;
	400	}
	401
28ab034a	402	void notification_thread_command_quit(struct notification_thread_handle *handle)
ab0ee2ca JG	403	{
ab0ee2ca JG	404	int ret;
32670d71	405	notification_thread_command cmd;
ab0ee2ca JG	406
	407	init_notification_thread_command(&cmd);
	408
	409	cmd.type = NOTIFICATION_COMMAND_TYPE_QUIT;
	410	ret = run_command_wait(handle, &cmd);
a0377dfe	411	LTTNG_ASSERT(!ret && cmd.reply_code == LTTNG_OK);
ab0ee2ca	412	}
f2b3ef9f JG	413
f2b3ef9f JG	414	int notification_thread_client_communication_update(
28ab034a JG	415	struct notification_thread_handle *handle,
	416	notification_client_id id,
	417	enum client_transmission_status transmission_status)
f2b3ef9f	418	{
32670d71	419	notification_thread_command cmd;
f2b3ef9f JG	420
	421	init_notification_thread_command(&cmd);
	422
	423	cmd.type = NOTIFICATION_COMMAND_TYPE_CLIENT_COMMUNICATION_UPDATE;
	424	cmd.parameters.client_communication_update.id = id;
	425	cmd.parameters.client_communication_update.status = transmission_status;
	426	return run_command_no_wait(handle, &cmd);
	427	}
b9a8d78f	428
28ab034a JG	429	enum lttng_error_code
	430	notification_thread_command_get_trigger(struct notification_thread_handle *handle,
	431	const struct lttng_trigger *trigger,
	432	struct lttng_trigger **real_trigger)
8790759c FD	433	{
	434	int ret;
	435	enum lttng_error_code ret_code;
32670d71	436	notification_thread_command cmd;
8790759c FD	437
	438	init_notification_thread_command(&cmd);
	439
	440	cmd.type = NOTIFICATION_COMMAND_TYPE_GET_TRIGGER;
	441	cmd.parameters.get_trigger.trigger = trigger;
	442	ret = run_command_wait(handle, &cmd);
	443	if (ret) {
	444	ret_code = LTTNG_ERR_UNK;
	445	goto end;
	446	}
	447
	448	ret_code = cmd.reply_code;
	449	*real_trigger = cmd.reply.get_trigger.trigger;
	450
	451	end:
	452	return ret_code;
	453	}
	454
82b3cbf4 JR	455	/*
	456	* Takes ownership of the payload if present.
	457	*/
82b3cbf4	458	struct lttng_event_notifier_notification *lttng_event_notifier_notification_create(
28ab034a	459	uint64_t tracer_token, enum lttng_domain_type domain, char *payload, size_t payload_size)
b9a8d78f	460	{
cd9adb8b	461	struct lttng_event_notifier_notification *notification = nullptr;
b9a8d78f	462
a0377dfe FD	463	LTTNG_ASSERT(domain != LTTNG_DOMAIN_NONE);
a0377dfe FD	464	LTTNG_ASSERT((payload && payload_size) \|\| (!payload && !payload_size));
b9a8d78f	465
64803277	466	notification = zmalloc<lttng_event_notifier_notification>();
cd9adb8b	467	if (notification == nullptr) {
bd0514a5	468	ERR("Error allocating notification");
b9a8d78f FD	469	goto end;
	470	}
	471
	472	notification->tracer_token = tracer_token;
	473	notification->type = domain;
82b3cbf4 JR	474	notification->capture_buffer = payload;
82b3cbf4 JR	475	notification->capture_buf_size = payload_size;
b9a8d78f FD	476
	477	end:
	478	return notification;
	479	}
	480
b9a8d78f	481	void lttng_event_notifier_notification_destroy(
28ab034a	482	struct lttng_event_notifier_notification *notification)
b9a8d78f FD	483	{
	484	if (!notification) {
	485	return;
	486	}
	487
82b3cbf4	488	free(notification->capture_buffer);
b9a8d78f FD	489	free(notification);
b9a8d78f FD	490	}