X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fcommon%2Factions%2Frotate-session.cpp;fp=src%2Fcommon%2Factions%2Frotate-session.cpp;h=985036acc84bc26818444b45b63c30add3126716;hp=0000000000000000000000000000000000000000;hb=a6bc4ca9d659caf016ef932fcd944029737ac57c;hpb=97535efaa975ca52bf02c2d5e76351bfd2e3defa

diff --git a/src/common/actions/rotate-session.cpp b/src/common/actions/rotate-session.cpp
new file mode 100644
index 000000000..985036acc
--- /dev/null
+++ b/src/common/actions/rotate-session.cpp
@@ -0,0 +1,436 @@
+/*
+ * Copyright (C) 2019 Simon Marchi <simon.marchi@efficios.com>
+ *
+ * SPDX-License-Identifier: LGPL-2.1-only
+ *
+ */
+
+#include <common/error.h>
+#include <common/macros.h>
+#include <common/mi-lttng.h>
+#include <lttng/action/action-internal.h>
+#include <lttng/action/rate-policy-internal.h>
+#include <lttng/action/rate-policy.h>
+#include <lttng/action/rotate-session-internal.h>
+#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 *action = 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 = (lttng_action *) zmalloc(sizeof(struct lttng_action_rotate_session));
+	if (!action) {
+		goto end;
+	}
+
+	lttng_action_init(action, 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, policy);
+	if (status != LTTNG_ACTION_STATUS_OK) {
+		free(action);
+		action = NULL;
+		goto end;
+	}
+
+end:
+	lttng_rate_policy_destroy(policy);
+	return action;
+}
+
+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;
+}