X-Git-Url: https://git.liburcu.org/?a=blobdiff_plain;f=src%2Fcommon%2Fust-consumer%2Fust-consumer.c;h=44a8240b055e55dd798e98874f642f1efc72c414;hb=4a2eb0cab793f67a94d7d9d10b534057804def95;hp=bc0f585385814f1ba25be632f2a21fbd04232de9;hpb=fe4477ee14abb348ce9e167f8b4c09312d67de36;p=lttng-tools.git

diff --git a/src/common/ust-consumer/ust-consumer.c b/src/common/ust-consumer/ust-consumer.c
index bc0f58538..44a8240b0 100644
--- a/src/common/ust-consumer/ust-consumer.c
+++ b/src/common/ust-consumer/ust-consumer.c
@@ -223,7 +223,9 @@ static int send_stream_to_relayd(struct lttng_consumer_stream *stream)
 		pthread_mutex_lock(&relayd->ctrl_sock_mutex);
 		/* Add stream on the relayd */
 		ret = relayd_add_stream(&relayd->control_sock, stream->name,
-				stream->chan->pathname, &stream->relayd_stream_id);
+				stream->chan->pathname, &stream->relayd_stream_id,
+				stream->chan->tracefile_size,
+				stream->chan->tracefile_count);
 		pthread_mutex_unlock(&relayd->ctrl_sock_mutex);
 		if (ret < 0) {
 			goto error;
@@ -554,7 +556,7 @@ int lttng_ustconsumer_push_metadata(struct lttng_consumer_channel *metadata,
 	ret = ustctl_write_metadata_to_channel(metadata->uchan,
 			metadata_str + target_offset, len);
 	if (ret < 0) {
-		ERR("ustctl write metadata fail with ret %d, len %ld", ret, len);
+		ERR("ustctl write metadata fail with ret %d, len %" PRIu64, ret, len);
 		goto error;
 	}
 
@@ -577,11 +579,11 @@ static int flush_channel(uint64_t chan_key)
 	struct lttng_ht *ht;
 	struct lttng_ht_iter iter;
 
-	DBG("UST consumer flush channel key %lu", chan_key);
+	DBG("UST consumer flush channel key %" PRIu64, chan_key);
 
 	channel = consumer_find_channel(chan_key);
 	if (!channel) {
-		ERR("UST consumer flush channel %lu not found", chan_key);
+		ERR("UST consumer flush channel %" PRIu64 " not found", chan_key);
 		ret = LTTNG_ERR_UST_CHAN_NOT_FOUND;
 		goto error;
 	}
@@ -611,11 +613,11 @@ static int close_metadata(uint64_t chan_key)
 	int ret;
 	struct lttng_consumer_channel *channel;
 
-	DBG("UST consumer close metadata key %lu", chan_key);
+	DBG("UST consumer close metadata key %" PRIu64, chan_key);
 
 	channel = consumer_find_channel(chan_key);
 	if (!channel) {
-		ERR("UST consumer close metadata %lu not found", chan_key);
+		ERR("UST consumer close metadata %" PRIu64 " not found", chan_key);
 		ret = LTTNG_ERR_UST_CHAN_NOT_FOUND;
 		goto error;
 	}
@@ -646,7 +648,7 @@ static int setup_metadata(struct lttng_consumer_local_data *ctx, uint64_t key)
 	int ret;
 	struct lttng_consumer_channel *metadata;
 
-	DBG("UST consumer setup metadata key %lu", key);
+	DBG("UST consumer setup metadata key %" PRIu64, key);
 
 	metadata = consumer_find_channel(key);
 	if (!metadata) {
@@ -699,7 +701,7 @@ int lttng_ustconsumer_recv_metadata(int sock, uint64_t key, uint64_t offset,
 	int ret, ret_code = LTTNG_OK;
 	char *metadata_str;
 
-	DBG("UST consumer push metadata key %lu of len %lu", key, len);
+	DBG("UST consumer push metadata key %" PRIu64 " of len %" PRIu64, key, len);
 
 	metadata_str = zmalloc(len * sizeof(char));
 	if (!metadata_str) {
@@ -857,6 +859,13 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 			goto end_channel_error;
 		}
 
+		/*
+		 * Set refcount to 1 for owner. Below, we will pass
+		 * ownership to the consumer_thread_channel_poll()
+		 * thread.
+		 */
+		channel->refcount = 1;
+
 		/* Build channel attributes from received message. */
 		attr.subbuf_size = msg.u.ask_channel.subbuf_size;
 		attr.num_subbuf = msg.u.ask_channel.num_subbuf;
@@ -896,10 +905,22 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 			goto end_channel_error;
 		}
 
+		if (msg.u.ask_channel.type == LTTNG_UST_CHAN_METADATA) {
+			ret = consumer_metadata_cache_allocate(channel);
+			if (ret < 0) {
+				ERR("Allocating metadata cache");
+				goto end_channel_error;
+			}
+			consumer_timer_switch_start(channel, attr.switch_timer_interval);
+			attr.switch_timer_interval = 0;
+		}
+
 		/*
 		 * Add the channel to the internal state AFTER all streams were created
 		 * and successfully sent to session daemon. This way, all streams must
 		 * be ready before this channel is visible to the threads.
+		 * If add_channel succeeds, ownership of the channel is
+		 * passed to consumer_thread_channel_poll().
 		 */
 		ret = add_channel(channel, ctx);
 		if (ret < 0) {
@@ -921,16 +942,6 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 			goto end_nosignal;
 		}
 
-		if (msg.u.ask_channel.type == LTTNG_UST_CHAN_METADATA) {
-			ret = consumer_metadata_cache_allocate(channel);
-			if (ret < 0) {
-				ERR("Allocating metadata cache");
-				goto end_channel_error;
-			}
-			consumer_timer_switch_start(channel, attr.switch_timer_interval);
-			attr.switch_timer_interval = 0;
-		}
-
 		break;
 	}
 	case LTTNG_CONSUMER_GET_CHANNEL:
@@ -941,7 +952,7 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 
 		channel = consumer_find_channel(key);
 		if (!channel) {
-			ERR("UST consumer get channel key %lu not found", key);
+			ERR("UST consumer get channel key %" PRIu64 " not found", key);
 			ret_code = LTTNG_ERR_UST_CHAN_NOT_FOUND;
 			goto end_msg_sessiond;
 		}
@@ -992,7 +1003,7 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 
 		channel = consumer_find_channel(key);
 		if (!channel) {
-			ERR("UST consumer get channel key %lu not found", key);
+			ERR("UST consumer get channel key %" PRIu64 " not found", key);
 			ret_code = LTTNG_ERR_UST_CHAN_NOT_FOUND;
 			goto end_msg_sessiond;
 		}
@@ -1031,12 +1042,14 @@ int lttng_ustconsumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 		uint64_t offset = msg.u.push_metadata.target_offset;
 		struct lttng_consumer_channel *channel;
 
-		DBG("UST consumer push metadata key %lu of len %lu", key, len);
+		DBG("UST consumer push metadata key %" PRIu64 " of len %" PRIu64, key,
+				len);
 
 		channel = consumer_find_channel(key);
 		if (!channel) {
-			ERR("UST consumer push metadata %lu not found", key);
+			ERR("UST consumer push metadata %" PRIu64 " not found", key);
 			ret_code = LTTNG_ERR_UST_CHAN_NOT_FOUND;
+			goto end_msg_sessiond;
 		}
 
 		/* Tell session daemon we are ready to receive the metadata. */
@@ -1272,7 +1285,7 @@ int lttng_ustconsumer_read_subbuffer(struct lttng_consumer_stream *stream,
 		 * happen and it is OK with the code flow.
 		 */
 		DBG("Error writing to tracefile "
-				"(ret: %zd != len: %lu != subbuf_size: %lu)",
+				"(ret: %ld != len: %lu != subbuf_size: %lu)",
 				ret, len, subbuf_size);
 	}
 	err = ustctl_put_next_subbuf(ustream);
@@ -1431,7 +1444,7 @@ int lttng_ustconsumer_request_metadata(struct lttng_consumer_local_data *ctx,
 	ret = lttcomm_recv_unix_sock(ctx->consumer_metadata_socket, &msg,
 			sizeof(msg));
 	if (ret != sizeof(msg)) {
-		DBG("Consumer received unexpected message size %d (expects %lu)",
+		DBG("Consumer received unexpected message size %d (expects %zu)",
 			ret, sizeof(msg));
 		lttng_consumer_send_error(ctx, LTTCOMM_CONSUMERD_ERROR_RECV_CMD);
 		/*