[lttng-tools.git] / src / common / actions / rotate-session.cpp

/*
 * Copyright (C) 2019 Simon Marchi <simon.marchi@efficios.com>
 *
 * SPDX-License-Identifier: LGPL-2.1-only
 *
 */

#include <common/error.hpp>
#include <common/macros.hpp>
#include <common/mi-lttng.hpp>
#include <lttng/action/action-internal.hpp>
#include <lttng/action/rate-policy-internal.hpp>
#include <lttng/action/rate-policy.h>
#include <lttng/action/rotate-session-internal.hpp>
#include <lttng/action/rotate-session.h>

#define IS_ROTATE_SESSION_ACTION(action) \
	(lttng_action_get_type(action) == LTTNG_ACTION_TYPE_ROTATE_SESSION)

struct lttng_action_rotate_session {
	struct lttng_action parent;

	/* Owned by this. */
	char *session_name;
	struct lttng_rate_policy *policy;
};

struct lttng_action_rotate_session_comm {
	/* Includes the trailing \0. */
	uint32_t session_name_len;

	/*
	 * Variable data:
	 *
	 *  - session name (null terminated)
	 *  - policy
	 */
	char data[];
} LTTNG_PACKED;

static const struct lttng_rate_policy *
lttng_action_rotate_session_internal_get_rate_policy(
		const struct lttng_action *action);

static struct lttng_action_rotate_session *action_rotate_session_from_action(
		struct lttng_action *action)
{
	LTTNG_ASSERT(action);

	return container_of(action, struct lttng_action_rotate_session, parent);
}

static const struct lttng_action_rotate_session *
action_rotate_session_from_action_const(const struct lttng_action *action)
{
	LTTNG_ASSERT(action);

	return container_of(action, struct lttng_action_rotate_session, parent);
}

static bool lttng_action_rotate_session_validate(struct lttng_action *action)
{
	bool valid;
	struct lttng_action_rotate_session *action_rotate_session;

	if (!action) {
		valid = false;
		goto end;
	}

	action_rotate_session = action_rotate_session_from_action(action);

	/* A non-empty session name is mandatory. */
	if (!action_rotate_session->session_name ||
			strlen(action_rotate_session->session_name) == 0) {
		valid = false;
		goto end;
	}

	valid = true;
end:
	return valid;
}

static bool lttng_action_rotate_session_is_equal(
		const struct lttng_action *_a, const struct lttng_action *_b)
{
	bool is_equal = false;
	const struct lttng_action_rotate_session *a, *b;

	a = action_rotate_session_from_action_const(_a);
	b = action_rotate_session_from_action_const(_b);

	/* Action is not valid if this is not true. */
	LTTNG_ASSERT(a->session_name);
	LTTNG_ASSERT(b->session_name);
	if (strcmp(a->session_name, b->session_name)) {
		goto end;
	}

	is_equal = lttng_rate_policy_is_equal(a->policy, b->policy);
end:
	return is_equal;
}
static int lttng_action_rotate_session_serialize(
		struct lttng_action *action, struct lttng_payload *payload)
{
	struct lttng_action_rotate_session *action_rotate_session;
	struct lttng_action_rotate_session_comm comm;
	size_t session_name_len;
	int ret;

	LTTNG_ASSERT(action);
	LTTNG_ASSERT(payload);

	action_rotate_session = action_rotate_session_from_action(action);

	LTTNG_ASSERT(action_rotate_session->session_name);

	DBG("Serializing rotate session action: session-name: %s",
			action_rotate_session->session_name);

	session_name_len = strlen(action_rotate_session->session_name) + 1;
	comm.session_name_len = session_name_len;

	ret = lttng_dynamic_buffer_append(
			&payload->buffer, &comm, sizeof(comm));
	if (ret) {
		ret = -1;
		goto end;
	}

	ret = lttng_dynamic_buffer_append(&payload->buffer,
			action_rotate_session->session_name, session_name_len);
	if (ret) {
		ret = -1;
		goto end;
	}

	ret = lttng_rate_policy_serialize(
			action_rotate_session->policy, payload);
	if (ret) {
		ret = -1;
		goto end;
	}
end:
	return ret;
}

static void lttng_action_rotate_session_destroy(struct lttng_action *action)
{
	struct lttng_action_rotate_session *action_rotate_session;

	if (!action) {
		goto end;
	}

	action_rotate_session = action_rotate_session_from_action(action);

	lttng_rate_policy_destroy(action_rotate_session->policy);
	free(action_rotate_session->session_name);
	free(action_rotate_session);

end:
	return;
}

ssize_t lttng_action_rotate_session_create_from_payload(
		struct lttng_payload_view *view,
		struct lttng_action **p_action)
{
	ssize_t consumed_len, ret;
	const struct lttng_action_rotate_session_comm *comm;
	const char *session_name;
	struct lttng_action *action;
	enum lttng_action_status status;
	struct lttng_rate_policy *policy = NULL;

	action = lttng_action_rotate_session_create();
	if (!action) {
		consumed_len = -1;
		goto end;
	}

	comm = (typeof(comm)) view->buffer.data;
	session_name = (const char *) &comm->data;

	if (!lttng_buffer_view_contains_string(
			&view->buffer, session_name, comm->session_name_len)) {
		consumed_len = -1;
		goto end;
	}
	consumed_len = sizeof(*comm) + comm->session_name_len;

	/* Rate policy. */
	{
		struct lttng_payload_view policy_view =
				lttng_payload_view_from_view(
						view, consumed_len, -1);
		ret = lttng_rate_policy_create_from_payload(
				&policy_view, &policy);
		if (ret < 0) {
			consumed_len = -1;
			goto end;
		}
		consumed_len += ret;
	}

	status = lttng_action_rotate_session_set_session_name(
			action, session_name);
	if (status != LTTNG_ACTION_STATUS_OK) {
		consumed_len = -1;
		goto end;
	}

	LTTNG_ASSERT(policy);
	status = lttng_action_rotate_session_set_rate_policy(action, policy);
	if (status != LTTNG_ACTION_STATUS_OK) {
		consumed_len = -1;
		goto end;
	}

	*p_action = action;
	action = NULL;

end:
	lttng_rate_policy_destroy(policy);
	lttng_action_rotate_session_destroy(action);

	return consumed_len;
}

static enum lttng_error_code lttng_action_rotate_session_mi_serialize(
		const struct lttng_action *action, struct mi_writer *writer)
{
	int ret;
	enum lttng_error_code ret_code;
	enum lttng_action_status status;
	const char *session_name = NULL;
	const struct lttng_rate_policy *policy = NULL;

	LTTNG_ASSERT(action);
	LTTNG_ASSERT(IS_ROTATE_SESSION_ACTION(action));

	status = lttng_action_rotate_session_get_session_name(
			action, &session_name);
	LTTNG_ASSERT(status == LTTNG_ACTION_STATUS_OK);
	LTTNG_ASSERT(session_name != NULL);

	status = lttng_action_notify_get_rate_policy(action, &policy);
	LTTNG_ASSERT(status == LTTNG_ACTION_STATUS_OK);
	LTTNG_ASSERT(policy != NULL);

	/* Open action rotate session element. */
	ret = mi_lttng_writer_open_element(
			writer, mi_lttng_element_action_rotate_session);
	if (ret) {
		goto mi_error;
	}

	/* Session name. */
	ret = mi_lttng_writer_write_element_string(
			writer, mi_lttng_element_session_name, session_name);
	if (ret) {
		goto mi_error;
	}

	/* Rate policy. */
	ret_code = lttng_rate_policy_mi_serialize(policy, writer);
	if (ret_code != LTTNG_OK) {
		goto end;
	}

	/* Close action rotate session 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:
	return ret_code;
}

struct lttng_action *lttng_action_rotate_session_create(void)
{
	struct lttng_action_rotate_session *action_rotate = NULL;
	struct lttng_rate_policy *policy = NULL;
	enum lttng_action_status status;

	/* Create a every N = 1 rate policy. */
	policy = lttng_rate_policy_every_n_create(1);
	if (!policy) {
		goto end;
	}

	action_rotate = zmalloc<lttng_action_rotate_session>();
	if (!action_rotate) {
		goto end;
	}

	lttng_action_init(&action_rotate->parent,
			LTTNG_ACTION_TYPE_ROTATE_SESSION,
			lttng_action_rotate_session_validate,
			lttng_action_rotate_session_serialize,
			lttng_action_rotate_session_is_equal,
			lttng_action_rotate_session_destroy,
			lttng_action_rotate_session_internal_get_rate_policy,
			lttng_action_generic_add_error_query_results,
			lttng_action_rotate_session_mi_serialize);

	status = lttng_action_rotate_session_set_rate_policy(
			&action_rotate->parent, policy);
	if (status != LTTNG_ACTION_STATUS_OK) {
		lttng_action_destroy(&action_rotate->parent);
		action_rotate = NULL;
		goto end;
	}

end:
	lttng_rate_policy_destroy(policy);
	return action_rotate ? &action_rotate->parent : nullptr;
}

enum lttng_action_status lttng_action_rotate_session_set_session_name(
		struct lttng_action *action, const char *session_name)
{
	struct lttng_action_rotate_session *action_rotate_session;
	enum lttng_action_status status;

	if (!action || !IS_ROTATE_SESSION_ACTION(action) || !session_name ||
			strlen(session_name) == 0) {
		status = LTTNG_ACTION_STATUS_INVALID;
		goto end;
	}

	action_rotate_session = action_rotate_session_from_action(action);

	free(action_rotate_session->session_name);

	action_rotate_session->session_name = strdup(session_name);
	if (!action_rotate_session->session_name) {
		status = LTTNG_ACTION_STATUS_ERROR;
		goto end;
	}

	status = LTTNG_ACTION_STATUS_OK;
end:
	return status;
}

enum lttng_action_status lttng_action_rotate_session_get_session_name(
		const struct lttng_action *action, const char **session_name)
{
	const struct lttng_action_rotate_session *action_rotate_session;
	enum lttng_action_status status;

	if (!action || !IS_ROTATE_SESSION_ACTION(action) || !session_name) {
		status = LTTNG_ACTION_STATUS_INVALID;
		goto end;
	}

	action_rotate_session = action_rotate_session_from_action_const(action);

	*session_name = action_rotate_session->session_name;

	status = LTTNG_ACTION_STATUS_OK;
end:
	return status;
}

enum lttng_action_status lttng_action_rotate_session_set_rate_policy(
		struct lttng_action *action,
		const struct lttng_rate_policy *policy)
{
	enum lttng_action_status status;
	struct lttng_action_rotate_session *rotate_session_action;
	struct lttng_rate_policy *copy = NULL;

	if (!action || !policy || !IS_ROTATE_SESSION_ACTION(action)) {
		status = LTTNG_ACTION_STATUS_INVALID;
		goto end;
	}

	copy = lttng_rate_policy_copy(policy);
	if (!copy) {
		status = LTTNG_ACTION_STATUS_ERROR;
		goto end;
	}

	rotate_session_action = action_rotate_session_from_action(action);

	/* Free the previous rate policy .*/
	lttng_rate_policy_destroy(rotate_session_action->policy);

	/* Assign the policy. */
	rotate_session_action->policy = copy;
	status = LTTNG_ACTION_STATUS_OK;
	copy = NULL;

end:
	lttng_rate_policy_destroy(copy);
	return status;
}

enum lttng_action_status lttng_action_rotate_session_get_rate_policy(
		const struct lttng_action *action,
		const struct lttng_rate_policy **policy)
{
	enum lttng_action_status status;
	const struct lttng_action_rotate_session *rotate_session_action;

	if (!action || !policy || !IS_ROTATE_SESSION_ACTION(action)) {
		status = LTTNG_ACTION_STATUS_INVALID;
		goto end;
	}

	rotate_session_action = action_rotate_session_from_action_const(action);

	*policy = rotate_session_action->policy;
	status = LTTNG_ACTION_STATUS_OK;
end:
	return status;
}

static const struct lttng_rate_policy *
lttng_action_rotate_session_internal_get_rate_policy(
		const struct lttng_action *action)
{
	const struct lttng_action_rotate_session *_action;
	_action = action_rotate_session_from_action_const(action);

	return _action->policy;
}
Commit	Line	Data
bfb2ec6a SM	1	/*
	2	* Copyright (C) 2019 Simon Marchi <simon.marchi@efficios.com>
	3	*
	4	* SPDX-License-Identifier: LGPL-2.1-only
	5	*
	6	*/
	7
c9e313bc SM	8	#include <common/error.hpp>
	9	#include <common/macros.hpp>
	10	#include <common/mi-lttng.hpp>
	11	#include <lttng/action/action-internal.hpp>
	12	#include <lttng/action/rate-policy-internal.hpp>
7f4d5b07	13	#include <lttng/action/rate-policy.h>
c9e313bc	14	#include <lttng/action/rotate-session-internal.hpp>
bfb2ec6a SM	15	#include <lttng/action/rotate-session.h>
	16
	17	#define IS_ROTATE_SESSION_ACTION(action) \
17182cfd	18	(lttng_action_get_type(action) == LTTNG_ACTION_TYPE_ROTATE_SESSION)
bfb2ec6a SM	19
	20	struct lttng_action_rotate_session {
	21	struct lttng_action parent;
	22
	23	/* Owned by this. */
	24	char *session_name;
7f4d5b07	25	struct lttng_rate_policy *policy;
bfb2ec6a SM	26	};
	27
	28	struct lttng_action_rotate_session_comm {
	29	/* Includes the trailing \0. */
	30	uint32_t session_name_len;
	31
	32	/*
	33	* Variable data:
	34	*
	35	* - session name (null terminated)
c2c58434	36	* - policy
bfb2ec6a SM	37	*/
	38	char data[];
	39	} LTTNG_PACKED;
	40
7f4d5b07 JR	41	static const struct lttng_rate_policy *
7f4d5b07 JR	42	lttng_action_rotate_session_internal_get_rate_policy(
2d57482c JR	43	const struct lttng_action *action);
2d57482c JR	44
bfb2ec6a SM	45	static struct lttng_action_rotate_session *action_rotate_session_from_action(
	46	struct lttng_action *action)
	47	{
a0377dfe	48	LTTNG_ASSERT(action);
bfb2ec6a SM	49
	50	return container_of(action, struct lttng_action_rotate_session, parent);
	51	}
	52
	53	static const struct lttng_action_rotate_session *
	54	action_rotate_session_from_action_const(const struct lttng_action *action)
	55	{
a0377dfe	56	LTTNG_ASSERT(action);
bfb2ec6a SM	57
	58	return container_of(action, struct lttng_action_rotate_session, parent);
	59	}
	60
	61	static bool lttng_action_rotate_session_validate(struct lttng_action *action)
	62	{
	63	bool valid;
	64	struct lttng_action_rotate_session *action_rotate_session;
	65
	66	if (!action) {
	67	valid = false;
	68	goto end;
	69	}
	70
	71	action_rotate_session = action_rotate_session_from_action(action);
	72
	73	/* A non-empty session name is mandatory. */
	74	if (!action_rotate_session->session_name \|\|
	75	strlen(action_rotate_session->session_name) == 0) {
	76	valid = false;
	77	goto end;
	78	}
	79
	80	valid = true;
	81	end:
	82	return valid;
	83	}
	84
	85	static bool lttng_action_rotate_session_is_equal(
	86	const struct lttng_action _a, const struct lttng_action _b)
	87	{
	88	bool is_equal = false;
	89	const struct lttng_action_rotate_session a, b;
	90
	91	a = action_rotate_session_from_action_const(_a);
	92	b = action_rotate_session_from_action_const(_b);
	93
	94	/* Action is not valid if this is not true. */
a0377dfe FD	95	LTTNG_ASSERT(a->session_name);
a0377dfe FD	96	LTTNG_ASSERT(b->session_name);
bfb2ec6a SM	97	if (strcmp(a->session_name, b->session_name)) {
	98	goto end;
	99	}
	100
7f4d5b07	101	is_equal = lttng_rate_policy_is_equal(a->policy, b->policy);
bfb2ec6a SM	102	end:
	103	return is_equal;
	104	}
	105	static int lttng_action_rotate_session_serialize(
c0a66c84	106	struct lttng_action action, struct lttng_payload payload)
bfb2ec6a SM	107	{
	108	struct lttng_action_rotate_session *action_rotate_session;
	109	struct lttng_action_rotate_session_comm comm;
	110	size_t session_name_len;
	111	int ret;
	112
a0377dfe FD	113	LTTNG_ASSERT(action);
a0377dfe FD	114	LTTNG_ASSERT(payload);
bfb2ec6a SM	115
	116	action_rotate_session = action_rotate_session_from_action(action);
	117
a0377dfe	118	LTTNG_ASSERT(action_rotate_session->session_name);
bfb2ec6a SM	119
	120	DBG("Serializing rotate session action: session-name: %s",
	121	action_rotate_session->session_name);
	122
	123	session_name_len = strlen(action_rotate_session->session_name) + 1;
	124	comm.session_name_len = session_name_len;
	125
c0a66c84 JG	126	ret = lttng_dynamic_buffer_append(
c0a66c84 JG	127	&payload->buffer, &comm, sizeof(comm));
bfb2ec6a SM	128	if (ret) {
	129	ret = -1;
	130	goto end;
	131	}
	132
c0a66c84	133	ret = lttng_dynamic_buffer_append(&payload->buffer,
bfb2ec6a SM	134	action_rotate_session->session_name, session_name_len);
	135	if (ret) {
	136	ret = -1;
	137	goto end;
	138	}
	139
7f4d5b07	140	ret = lttng_rate_policy_serialize(
c2c58434 JR	141	action_rotate_session->policy, payload);
	142	if (ret) {
	143	ret = -1;
	144	goto end;
	145	}
bfb2ec6a SM	146	end:
	147	return ret;
	148	}
	149
	150	static void lttng_action_rotate_session_destroy(struct lttng_action *action)
	151	{
	152	struct lttng_action_rotate_session *action_rotate_session;
	153
	154	if (!action) {
	155	goto end;
	156	}
	157
	158	action_rotate_session = action_rotate_session_from_action(action);
	159
7f4d5b07	160	lttng_rate_policy_destroy(action_rotate_session->policy);
bfb2ec6a SM	161	free(action_rotate_session->session_name);
	162	free(action_rotate_session);
	163
	164	end:
	165	return;
	166	}
	167
c0a66c84 JG	168	ssize_t lttng_action_rotate_session_create_from_payload(
c0a66c84 JG	169	struct lttng_payload_view *view,
bfb2ec6a SM	170	struct lttng_action **p_action)
bfb2ec6a SM	171	{
c2c58434	172	ssize_t consumed_len, ret;
bfb2ec6a SM	173	const struct lttng_action_rotate_session_comm *comm;
	174	const char *session_name;
	175	struct lttng_action *action;
	176	enum lttng_action_status status;
7f4d5b07	177	struct lttng_rate_policy *policy = NULL;
bfb2ec6a SM	178
	179	action = lttng_action_rotate_session_create();
	180	if (!action) {
	181	consumed_len = -1;
	182	goto end;
	183	}
	184
c0a66c84	185	comm = (typeof(comm)) view->buffer.data;
bfb2ec6a SM	186	session_name = (const char *) &comm->data;
	187
	188	if (!lttng_buffer_view_contains_string(
c0a66c84	189	&view->buffer, session_name, comm->session_name_len)) {
bfb2ec6a SM	190	consumed_len = -1;
	191	goto end;
	192	}
c2c58434 JR	193	consumed_len = sizeof(*comm) + comm->session_name_len;
c2c58434 JR	194
7f4d5b07	195	/* Rate policy. */
c2c58434 JR	196	{
	197	struct lttng_payload_view policy_view =
	198	lttng_payload_view_from_view(
	199	view, consumed_len, -1);
7f4d5b07	200	ret = lttng_rate_policy_create_from_payload(
c2c58434 JR	201	&policy_view, &policy);
	202	if (ret < 0) {
	203	consumed_len = -1;
	204	goto end;
	205	}
	206	consumed_len += ret;
	207	}
bfb2ec6a SM	208
	209	status = lttng_action_rotate_session_set_session_name(
	210	action, session_name);
	211	if (status != LTTNG_ACTION_STATUS_OK) {
	212	consumed_len = -1;
	213	goto end;
	214	}
	215
a0377dfe	216	LTTNG_ASSERT(policy);
7f4d5b07	217	status = lttng_action_rotate_session_set_rate_policy(action, policy);
c2c58434 JR	218	if (status != LTTNG_ACTION_STATUS_OK) {
	219	consumed_len = -1;
	220	goto end;
	221	}
	222
bfb2ec6a SM	223	*p_action = action;
	224	action = NULL;
	225
	226	end:
7f4d5b07	227	lttng_rate_policy_destroy(policy);
bfb2ec6a SM	228	lttng_action_rotate_session_destroy(action);
	229
	230	return consumed_len;
	231	}
	232
6a751b95 JR	233	static enum lttng_error_code lttng_action_rotate_session_mi_serialize(
	234	const struct lttng_action action, struct mi_writer writer)
	235	{
	236	int ret;
	237	enum lttng_error_code ret_code;
	238	enum lttng_action_status status;
	239	const char *session_name = NULL;
	240	const struct lttng_rate_policy *policy = NULL;
	241
a0377dfe FD	242	LTTNG_ASSERT(action);
a0377dfe FD	243	LTTNG_ASSERT(IS_ROTATE_SESSION_ACTION(action));
6a751b95 JR	244
	245	status = lttng_action_rotate_session_get_session_name(
	246	action, &session_name);
a0377dfe FD	247	LTTNG_ASSERT(status == LTTNG_ACTION_STATUS_OK);
a0377dfe FD	248	LTTNG_ASSERT(session_name != NULL);
6a751b95 JR	249
6a751b95 JR	250	status = lttng_action_notify_get_rate_policy(action, &policy);
a0377dfe FD	251	LTTNG_ASSERT(status == LTTNG_ACTION_STATUS_OK);
a0377dfe FD	252	LTTNG_ASSERT(policy != NULL);
6a751b95 JR	253
	254	/* Open action rotate session element. */
	255	ret = mi_lttng_writer_open_element(
	256	writer, mi_lttng_element_action_rotate_session);
	257	if (ret) {
	258	goto mi_error;
	259	}
	260
	261	/* Session name. */
	262	ret = mi_lttng_writer_write_element_string(
	263	writer, mi_lttng_element_session_name, session_name);
	264	if (ret) {
	265	goto mi_error;
	266	}
	267
	268	/* Rate policy. */
	269	ret_code = lttng_rate_policy_mi_serialize(policy, writer);
	270	if (ret_code != LTTNG_OK) {
	271	goto end;
	272	}
	273
	274	/* Close action rotate session element. */
	275	ret = mi_lttng_writer_close_element(writer);
	276	if (ret) {
	277	goto mi_error;
	278	}
	279
	280	ret_code = LTTNG_OK;
	281	goto end;
	282
	283	mi_error:
	284	ret_code = LTTNG_ERR_MI_IO_FAIL;
	285	end:
	286	return ret_code;
	287	}
	288
bfb2ec6a SM	289	struct lttng_action *lttng_action_rotate_session_create(void)
bfb2ec6a SM	290	{
64803277	291	struct lttng_action_rotate_session *action_rotate = NULL;
7f4d5b07	292	struct lttng_rate_policy *policy = NULL;
c2c58434 JR	293	enum lttng_action_status status;
c2c58434 JR	294
7f4d5b07 JR	295	/* Create a every N = 1 rate policy. */
7f4d5b07 JR	296	policy = lttng_rate_policy_every_n_create(1);
c2c58434 JR	297	if (!policy) {
	298	goto end;
	299	}
bfb2ec6a	300
64803277 SM	301	action_rotate = zmalloc<lttng_action_rotate_session>();
64803277 SM	302	if (!action_rotate) {
bfb2ec6a SM	303	goto end;
	304	}
	305
64803277 SM	306	lttng_action_init(&action_rotate->parent,
64803277 SM	307	LTTNG_ACTION_TYPE_ROTATE_SESSION,
bfb2ec6a SM	308	lttng_action_rotate_session_validate,
	309	lttng_action_rotate_session_serialize,
	310	lttng_action_rotate_session_is_equal,
2d57482c	311	lttng_action_rotate_session_destroy,
588c4b0d	312	lttng_action_rotate_session_internal_get_rate_policy,
6a751b95 JR	313	lttng_action_generic_add_error_query_results,
6a751b95 JR	314	lttng_action_rotate_session_mi_serialize);
bfb2ec6a	315
64803277 SM	316	status = lttng_action_rotate_session_set_rate_policy(
64803277 SM	317	&action_rotate->parent, policy);
c2c58434	318	if (status != LTTNG_ACTION_STATUS_OK) {
64803277 SM	319	lttng_action_destroy(&action_rotate->parent);
64803277 SM	320	action_rotate = NULL;
c2c58434 JR	321	goto end;
	322	}
	323
bfb2ec6a	324	end:
7f4d5b07	325	lttng_rate_policy_destroy(policy);
64803277	326	return action_rotate ? &action_rotate->parent : nullptr;
bfb2ec6a SM	327	}
	328
	329	enum lttng_action_status lttng_action_rotate_session_set_session_name(
	330	struct lttng_action action, const char session_name)
	331	{
	332	struct lttng_action_rotate_session *action_rotate_session;
	333	enum lttng_action_status status;
	334
	335	if (!action \|\| !IS_ROTATE_SESSION_ACTION(action) \|\| !session_name \|\|
	336	strlen(session_name) == 0) {
	337	status = LTTNG_ACTION_STATUS_INVALID;
	338	goto end;
	339	}
	340
	341	action_rotate_session = action_rotate_session_from_action(action);
	342
	343	free(action_rotate_session->session_name);
	344
	345	action_rotate_session->session_name = strdup(session_name);
	346	if (!action_rotate_session->session_name) {
	347	status = LTTNG_ACTION_STATUS_ERROR;
	348	goto end;
	349	}
	350
	351	status = LTTNG_ACTION_STATUS_OK;
	352	end:
	353	return status;
	354	}
	355
	356	enum lttng_action_status lttng_action_rotate_session_get_session_name(
	357	const struct lttng_action action, const char *session_name)
	358	{
	359	const struct lttng_action_rotate_session *action_rotate_session;
	360	enum lttng_action_status status;
	361
	362	if (!action \|\| !IS_ROTATE_SESSION_ACTION(action) \|\| !session_name) {
	363	status = LTTNG_ACTION_STATUS_INVALID;
	364	goto end;
	365	}
	366
	367	action_rotate_session = action_rotate_session_from_action_const(action);
	368
	369	*session_name = action_rotate_session->session_name;
	370
	371	status = LTTNG_ACTION_STATUS_OK;
	372	end:
	373	return status;
	374	}
c2c58434	375
7f4d5b07	376	enum lttng_action_status lttng_action_rotate_session_set_rate_policy(
c2c58434	377	struct lttng_action *action,
7f4d5b07	378	const struct lttng_rate_policy *policy)
c2c58434 JR	379	{
	380	enum lttng_action_status status;
	381	struct lttng_action_rotate_session *rotate_session_action;
7f4d5b07	382	struct lttng_rate_policy *copy = NULL;
c2c58434 JR	383
	384	if (!action \|\| !policy \|\| !IS_ROTATE_SESSION_ACTION(action)) {
	385	status = LTTNG_ACTION_STATUS_INVALID;
	386	goto end;
	387	}
	388
7f4d5b07	389	copy = lttng_rate_policy_copy(policy);
c2c58434 JR	390	if (!copy) {
	391	status = LTTNG_ACTION_STATUS_ERROR;
	392	goto end;
	393	}
	394
	395	rotate_session_action = action_rotate_session_from_action(action);
	396
7f4d5b07 JR	397	/* Free the previous rate policy .*/
7f4d5b07 JR	398	lttng_rate_policy_destroy(rotate_session_action->policy);
c2c58434 JR	399
	400	/* Assign the policy. */
	401	rotate_session_action->policy = copy;
	402	status = LTTNG_ACTION_STATUS_OK;
	403	copy = NULL;
	404
	405	end:
7f4d5b07	406	lttng_rate_policy_destroy(copy);
c2c58434 JR	407	return status;
	408	}
	409
7f4d5b07	410	enum lttng_action_status lttng_action_rotate_session_get_rate_policy(
c2c58434	411	const struct lttng_action *action,
7f4d5b07	412	const struct lttng_rate_policy **policy)
c2c58434 JR	413	{
	414	enum lttng_action_status status;
	415	const struct lttng_action_rotate_session *rotate_session_action;
	416
	417	if (!action \|\| !policy \|\| !IS_ROTATE_SESSION_ACTION(action)) {
	418	status = LTTNG_ACTION_STATUS_INVALID;
	419	goto end;
	420	}
	421
	422	rotate_session_action = action_rotate_session_from_action_const(action);
	423
	424	*policy = rotate_session_action->policy;
	425	status = LTTNG_ACTION_STATUS_OK;
	426	end:
	427	return status;
	428	}
2d57482c	429
7f4d5b07 JR	430	static const struct lttng_rate_policy *
7f4d5b07 JR	431	lttng_action_rotate_session_internal_get_rate_policy(
2d57482c JR	432	const struct lttng_action *action)
	433	{
	434	const struct lttng_action_rotate_session *_action;
	435	_action = action_rotate_session_from_action_const(action);
	436
	437	return _action->policy;
	438	}