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

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

#include <lttng/trigger/trigger.h>
#include <lttng/lttng-error.h>
#include "notification-thread.h"
#include "notification-thread-commands.h"
#include <common/error.h>
#include <unistd.h>
#include <stdint.h>
#include <inttypes.h>

static
void init_notification_thread_command(struct notification_thread_command *cmd)
{
	memset(cmd, 0, sizeof(*cmd));
	CDS_INIT_LIST_HEAD(&cmd->cmd_list_node);
	lttng_waiter_init(&cmd->reply_waiter);
}

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

	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(lttng_pipe_get_writefd(handle->cmd_queue.event_pipe),
			&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);

	lttng_waiter_wait(&cmd->reply_waiter);
	return 0;
error_unlock_queue:
	pthread_mutex_unlock(&handle->cmd_queue.lock);
	return -1;
}

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

	init_notification_thread_command(&cmd);

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

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_UNREGISTER_TRIGGER;
	cmd.parameters.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_channel(
		struct notification_thread_handle *handle,
		char *session_name, uid_t uid, gid_t gid,
		char *channel_name, uint64_t key,
		enum lttng_domain_type domain, uint64_t capacity)
{
	int ret;
	enum lttng_error_code ret_code;
	struct notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_CHANNEL;
	cmd.parameters.add_channel.session.name = session_name;
	cmd.parameters.add_channel.session.uid = uid;
	cmd.parameters.add_channel.session.gid = gid;
	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;
	struct 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,
		const char *session_name, uid_t uid, gid_t gid,
		uint64_t trace_archive_chunk_id)
{
	int ret;
	enum lttng_error_code ret_code;
	struct notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_SESSION_ROTATION_ONGOING;
	cmd.parameters.session_rotation.session_name = session_name;
	cmd.parameters.session_rotation.uid = uid;
	cmd.parameters.session_rotation.gid = gid;
	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,
		const char *session_name, uid_t uid, gid_t gid,
		uint64_t trace_archive_chunk_id,
		struct lttng_trace_archive_location *location)
{
	int ret;
	enum lttng_error_code ret_code;
	struct notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_SESSION_ROTATION_COMPLETED;
	cmd.parameters.session_rotation.session_name = session_name;
	cmd.parameters.session_rotation.uid = uid;
	cmd.parameters.session_rotation.gid = gid;
	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;
}

void notification_thread_command_quit(
		struct notification_thread_handle *handle)
{
	int ret;
	struct notification_thread_command cmd;

	init_notification_thread_command(&cmd);

	cmd.type = NOTIFICATION_COMMAND_TYPE_QUIT;
	ret = run_command_wait(handle, &cmd);
	assert(!ret && cmd.reply_code == LTTNG_OK);
}
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	*/
	7
	8	#include <lttng/trigger/trigger.h>
	9	#include <lttng/lttng-error.h>
	10	#include "notification-thread.h"
	11	#include "notification-thread-commands.h"
	12	#include <common/error.h>
ab0ee2ca JG	13	#include <unistd.h>
	14	#include <stdint.h>
	15	#include <inttypes.h>
	16
	17	static
	18	void init_notification_thread_command(struct notification_thread_command *cmd)
	19	{
	20	memset(cmd, 0, sizeof(*cmd));
	21	CDS_INIT_LIST_HEAD(&cmd->cmd_list_node);
8ada111f	22	lttng_waiter_init(&cmd->reply_waiter);
ab0ee2ca JG	23	}
	24
	25	static
	26	int run_command_wait(struct notification_thread_handle *handle,
	27	struct notification_thread_command *cmd)
	28	{
	29	int ret;
	30	uint64_t notification_counter = 1;
	31
ab0ee2ca JG	32	pthread_mutex_lock(&handle->cmd_queue.lock);
	33	/* Add to queue. */
	34	cds_list_add_tail(&cmd->cmd_list_node,
	35	&handle->cmd_queue.list);
	36	/* Wake-up thread. */
58d3fed5	37	ret = lttng_write(lttng_pipe_get_writefd(handle->cmd_queue.event_pipe),
ab0ee2ca	38	&notification_counter, sizeof(notification_counter));
58d3fed5	39	if (ret != sizeof(notification_counter)) {
ab0ee2ca JG	40	PERROR("write to notification thread's queue event fd");
	41	/*
	42	* Remove the command from the list so the notification
	43	* thread does not process it.
	44	*/
	45	cds_list_del(&cmd->cmd_list_node);
	46	goto error_unlock_queue;
	47	}
	48	pthread_mutex_unlock(&handle->cmd_queue.lock);
	49
8ada111f	50	lttng_waiter_wait(&cmd->reply_waiter);
ab0ee2ca JG	51	return 0;
	52	error_unlock_queue:
	53	pthread_mutex_unlock(&handle->cmd_queue.lock);
	54	return -1;
	55	}
	56
	57	enum lttng_error_code notification_thread_command_register_trigger(
	58	struct notification_thread_handle *handle,
	59	struct lttng_trigger *trigger)
	60	{
	61	int ret;
	62	enum lttng_error_code ret_code;
	63	struct notification_thread_command cmd;
	64
	65	init_notification_thread_command(&cmd);
	66
	67	cmd.type = NOTIFICATION_COMMAND_TYPE_REGISTER_TRIGGER;
	68	cmd.parameters.trigger = trigger;
	69
	70	ret = run_command_wait(handle, &cmd);
	71	if (ret) {
	72	ret_code = LTTNG_ERR_UNK;
	73	goto end;
	74	}
	75	ret_code = cmd.reply_code;
	76	end:
	77	return ret_code;
	78	}
	79
	80	enum lttng_error_code notification_thread_command_unregister_trigger(
	81	struct notification_thread_handle *handle,
	82	struct lttng_trigger *trigger)
	83	{
	84	int ret;
	85	enum lttng_error_code ret_code;
	86	struct notification_thread_command cmd;
	87
	88	init_notification_thread_command(&cmd);
	89
	90	cmd.type = NOTIFICATION_COMMAND_TYPE_UNREGISTER_TRIGGER;
	91	cmd.parameters.trigger = trigger;
	92
	93	ret = run_command_wait(handle, &cmd);
	94	if (ret) {
	95	ret_code = LTTNG_ERR_UNK;
	96	goto end;
	97	}
	98	ret_code = cmd.reply_code;
	99	end:
	100	return ret_code;
	101	}
	102
	103	enum lttng_error_code notification_thread_command_add_channel(
	104	struct notification_thread_handle *handle,
	105	char *session_name, uid_t uid, gid_t gid,
	106	char *channel_name, uint64_t key,
	107	enum lttng_domain_type domain, uint64_t capacity)
	108	{
	109	int ret;
	110	enum lttng_error_code ret_code;
	111	struct notification_thread_command cmd;
	112
	113	init_notification_thread_command(&cmd);
	114
115	cmd.type = NOTIFICATION_COMMAND_TYPE_ADD_CHANNEL;
8abe313a JG	116	cmd.parameters.add_channel.session.name = session_name;
	117	cmd.parameters.add_channel.session.uid = uid;
	118	cmd.parameters.add_channel.session.gid = gid;
	119	cmd.parameters.add_channel.channel.name = channel_name;
	120	cmd.parameters.add_channel.channel.key = key;
	121	cmd.parameters.add_channel.channel.domain = domain;
	122	cmd.parameters.add_channel.channel.capacity = capacity;
ab0ee2ca JG	123
	124	ret = run_command_wait(handle, &cmd);
	125	if (ret) {
	126	ret_code = LTTNG_ERR_UNK;
	127	goto end;
	128	}
	129	ret_code = cmd.reply_code;
	130	end:
	131	return ret_code;
	132	}
	133
	134	enum lttng_error_code notification_thread_command_remove_channel(
	135	struct notification_thread_handle *handle,
	136	uint64_t key, enum lttng_domain_type domain)
	137	{
	138	int ret;
	139	enum lttng_error_code ret_code;
	140	struct notification_thread_command cmd;
	141
	142	init_notification_thread_command(&cmd);
	143
	144	cmd.type = NOTIFICATION_COMMAND_TYPE_REMOVE_CHANNEL;
	145	cmd.parameters.remove_channel.key = key;
	146	cmd.parameters.remove_channel.domain = domain;
	147
	148	ret = run_command_wait(handle, &cmd);
	149	if (ret) {
	150	ret_code = LTTNG_ERR_UNK;
	151	goto end;
	152	}
	153	ret_code = cmd.reply_code;
	154	end:
	155	return ret_code;
	156	}
	157
731c1b12 JG	158	enum lttng_error_code notification_thread_command_session_rotation_ongoing(
	159	struct notification_thread_handle *handle,
	160	const char *session_name, uid_t uid, gid_t gid,
	161	uint64_t trace_archive_chunk_id)
	162	{
	163	int ret;
	164	enum lttng_error_code ret_code;
	165	struct notification_thread_command cmd;
	166
	167	init_notification_thread_command(&cmd);
	168
	169	cmd.type = NOTIFICATION_COMMAND_TYPE_SESSION_ROTATION_ONGOING;
	170	cmd.parameters.session_rotation.session_name = session_name;
	171	cmd.parameters.session_rotation.uid = uid;
	172	cmd.parameters.session_rotation.gid = gid;
	173	cmd.parameters.session_rotation.trace_archive_chunk_id =
	174	trace_archive_chunk_id;
	175
	176	ret = run_command_wait(handle, &cmd);
	177	if (ret) {
	178	ret_code = LTTNG_ERR_UNK;
	179	goto end;
	180	}
	181	ret_code = cmd.reply_code;
	182	end:
	183	return ret_code;
	184	}
	185
	186	enum lttng_error_code notification_thread_command_session_rotation_completed(
	187	struct notification_thread_handle *handle,
	188	const char *session_name, uid_t uid, gid_t gid,
	189	uint64_t trace_archive_chunk_id,
	190	struct lttng_trace_archive_location *location)
	191	{
	192	int ret;
	193	enum lttng_error_code ret_code;
	194	struct notification_thread_command cmd;
	195
	196	init_notification_thread_command(&cmd);
	197
	198	cmd.type = NOTIFICATION_COMMAND_TYPE_SESSION_ROTATION_COMPLETED;
	199	cmd.parameters.session_rotation.session_name = session_name;
	200	cmd.parameters.session_rotation.uid = uid;
	201	cmd.parameters.session_rotation.gid = gid;
	202	cmd.parameters.session_rotation.trace_archive_chunk_id =
	203	trace_archive_chunk_id;
	204	cmd.parameters.session_rotation.location = location;
	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
ab0ee2ca JG	216	void notification_thread_command_quit(
	217	struct notification_thread_handle *handle)
	218	{
	219	int ret;
	220	struct notification_thread_command cmd;
	221
	222	init_notification_thread_command(&cmd);
	223
	224	cmd.type = NOTIFICATION_COMMAND_TYPE_QUIT;
	225	ret = run_command_wait(handle, &cmd);
	226	assert(!ret && cmd.reply_code == LTTNG_OK);
	227	}