X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-sessiond%2Fust-app.c;h=38aeaab60a9399796fbce28080686af087279803;hp=0db0eb17382077d34fd95e0790acba624047ad98;hb=5c408ad8ef08a226c018702aca969536f36ac4e5;hpb=d086f507d02078aed618ab291a0bc4a634958fa3

diff --git a/src/bin/lttng-sessiond/ust-app.c b/src/bin/lttng-sessiond/ust-app.c
index 0db0eb173..38aeaab60 100644
--- a/src/bin/lttng-sessiond/ust-app.c
+++ b/src/bin/lttng-sessiond/ust-app.c
@@ -43,6 +43,7 @@
 #include "session.h"
 #include "lttng-sessiond.h"
 #include "notification-thread-commands.h"
+#include "rotate.h"
 
 static
 int ust_app_flush_app_session(struct ust_app *app, struct ust_app_session *ua_sess);
@@ -6288,3 +6289,195 @@ int ust_app_regenerate_statedump_all(struct ltt_ust_session *usess)
 
 	return 0;
 }
+
+/*
+ * Rotate all the channels of a session.
+ *
+ * Return 0 on success or else a negative value.
+ */
+int ust_app_rotate_session(struct ltt_session *session, bool *ust_active)
+{
+	int ret = 0;
+	struct lttng_ht_iter iter;
+	struct ust_app *app;
+	struct ltt_ust_session *usess = session->ust_session;
+	char pathname[LTTNG_PATH_MAX];
+
+	assert(usess);
+
+	rcu_read_lock();
+
+	switch (usess->buffer_type) {
+	case LTTNG_BUFFER_PER_UID:
+	{
+		struct buffer_reg_uid *reg;
+
+		cds_list_for_each_entry(reg, &usess->buffer_reg_uid_list, lnode) {
+			struct buffer_reg_channel *reg_chan;
+			struct consumer_socket *socket;
+
+			/* Get consumer socket to use to push the metadata.*/
+			socket = consumer_find_socket_by_bitness(reg->bits_per_long,
+					usess->consumer);
+			if (!socket) {
+				ret = -EINVAL;
+				goto error;
+			}
+
+			/*
+			 * Account the metadata channel first to make sure the
+			 * number of channels waiting for a rotation cannot
+			 * reach 0 before we complete the iteration over all
+			 * the channels.
+			 */
+			ret = rotate_add_channel_pending(
+					reg->registry->reg.ust->metadata_key,
+					LTTNG_DOMAIN_UST, session);
+			if (ret < 0) {
+				ret = LTTNG_ERR_KERN_CONSUMER_FAIL;
+				goto error;
+			}
+
+			ret = snprintf(pathname, sizeof(pathname),
+					DEFAULT_UST_TRACE_DIR "/" DEFAULT_UST_TRACE_UID_PATH,
+					reg->uid, reg->bits_per_long);
+			if (ret < 0 || ret == sizeof(pathname)) {
+				PERROR("Failed to format rotation path");
+				goto error;
+			}
+
+			/* Rotate the data channels. */
+			cds_lfht_for_each_entry(reg->registry->channels->ht, &iter.iter,
+					reg_chan, node.node) {
+				ret = rotate_add_channel_pending(
+						reg_chan->consumer_key,
+						LTTNG_DOMAIN_UST, session);
+				if (ret < 0) {
+					ret = LTTNG_ERR_KERN_CONSUMER_FAIL;
+					goto error;
+				}
+				ret = consumer_rotate_channel(socket,
+						reg_chan->consumer_key,
+						usess->uid, usess->gid,
+						usess->consumer, pathname,
+						/* is_metadata_channel */ false,
+						session->rotate_count,
+						&session->rotate_pending_relay);
+				if (ret < 0) {
+					goto error;
+				}
+			}
+
+			(void) push_metadata(reg->registry->reg.ust, usess->consumer);
+
+			ret = consumer_rotate_channel(socket,
+					reg->registry->reg.ust->metadata_key,
+					usess->uid, usess->gid,
+					usess->consumer, pathname,
+					/* is_metadata_channel */ true,
+					session->rotate_count,
+					&session->rotate_pending_relay);
+			if (ret < 0) {
+				goto error;
+			}
+			*ust_active = true;
+		}
+		break;
+	}
+	case LTTNG_BUFFER_PER_PID:
+	{
+		cds_lfht_for_each_entry(ust_app_ht->ht, &iter.iter, app, pid_n.node) {
+			struct consumer_socket *socket;
+			struct lttng_ht_iter chan_iter;
+			struct ust_app_channel *ua_chan;
+			struct ust_app_session *ua_sess;
+			struct ust_registry_session *registry;
+
+			ua_sess = lookup_session_by_app(usess, app);
+			if (!ua_sess) {
+				/* Session not associated with this app. */
+				continue;
+			}
+			ret = snprintf(pathname, sizeof(pathname),
+					DEFAULT_UST_TRACE_DIR "/%s",
+					ua_sess->path);
+			if (ret < 0 || ret == sizeof(pathname)) {
+				PERROR("Failed to format rotation path");
+				goto error;
+			}
+
+			/* Get the right consumer socket for the application. */
+			socket = consumer_find_socket_by_bitness(app->bits_per_long,
+					usess->consumer);
+			if (!socket) {
+				ret = -EINVAL;
+				goto error;
+			}
+
+			registry = get_session_registry(ua_sess);
+			if (!registry) {
+				DBG("Application session is being torn down. Abort snapshot record.");
+				ret = -1;
+				goto error;
+			}
+
+			/*
+			 * Account the metadata channel first to make sure the
+			 * number of channels waiting for a rotation cannot
+			 * reach 0 before we complete the iteration over all
+			 * the channels.
+			 */
+			ret = rotate_add_channel_pending(registry->metadata_key,
+					LTTNG_DOMAIN_UST, session);
+			if (ret < 0) {
+				ret = LTTNG_ERR_KERN_CONSUMER_FAIL;
+				goto error;
+			}
+
+			/* Rotate the data channels. */
+			cds_lfht_for_each_entry(ua_sess->channels->ht, &chan_iter.iter,
+					ua_chan, node.node) {
+				ret = rotate_add_channel_pending(
+						ua_chan->key, LTTNG_DOMAIN_UST,
+						session);
+				if (ret < 0) {
+					ret = LTTNG_ERR_KERN_CONSUMER_FAIL;
+					goto error;
+				}
+				ret = consumer_rotate_channel(socket, ua_chan->key,
+						ua_sess->euid, ua_sess->egid,
+						ua_sess->consumer, pathname,
+						/* is_metadata_channel */ false,
+						session->rotate_count,
+						&session->rotate_pending_relay);
+				if (ret < 0) {
+					goto error;
+				}
+			}
+
+			/* Rotate the metadata channel. */
+			(void) push_metadata(registry, usess->consumer);
+			ret = consumer_rotate_channel(socket, registry->metadata_key,
+					ua_sess->euid, ua_sess->egid,
+					ua_sess->consumer, pathname,
+					/* is_metadata_channel */ true,
+					session->rotate_count,
+					&session->rotate_pending_relay);
+			if (ret < 0) {
+				goto error;
+			}
+			*ust_active = true;
+		}
+		break;
+	}
+	default:
+		assert(0);
+		break;
+	}
+
+	ret = LTTNG_OK;
+
+error:
+	rcu_read_unlock();
+	return ret;
+}