X-Git-Url: http://git.lttng.org/?a=blobdiff_plain;f=src%2Fbin%2Flttng-sessiond%2Fnotification-thread-events.c;h=8f250dfdd715f38da9020b746c4cfbb30321f6b0;hb=4b01971f9df6293ace728642f15f5262691cec66;hp=b0657a4c5675dde9d0d86d1478afbeaedf26646a;hpb=8ada111f6d3ab40d1c33cf1a7b2546de9a47d1d5;p=lttng-tools.git

diff --git a/src/bin/lttng-sessiond/notification-thread-events.c b/src/bin/lttng-sessiond/notification-thread-events.c
index b0657a4c5..8f250dfdd 100644
--- a/src/bin/lttng-sessiond/notification-thread-events.c
+++ b/src/bin/lttng-sessiond/notification-thread-events.c
@@ -97,6 +97,11 @@ struct notification_client {
 	struct cds_lfht_node client_socket_ht_node;
 	struct {
 		struct {
+			/*
+			 * During the reception of a message, the reception
+			 * buffers' "size" is set to contain the current
+			 * message's complete payload.
+			 */
 			struct lttng_dynamic_buffer buffer;
 			/* Bytes left to receive for the current message. */
 			size_t bytes_to_receive;
@@ -112,6 +117,7 @@ struct notification_client {
 			 * from the client.
 			 */
 			bool creds_received;
+			/* Only used during credentials reception. */
 			lttng_sock_cred creds;
 		} inbound;
 		struct {
@@ -1153,7 +1159,7 @@ end:
 }
 
 static
-void client_reset_inbound_state(struct notification_client *client)
+int client_reset_inbound_state(struct notification_client *client)
 {
 	int ret;
 
@@ -1168,6 +1174,10 @@ void client_reset_inbound_state(struct notification_client *client)
 	client->communication.inbound.receive_creds = false;
 	LTTNG_SOCK_SET_UID_CRED(&client->communication.inbound.creds, -1);
 	LTTNG_SOCK_SET_GID_CRED(&client->communication.inbound.creds, -1);
+	ret = lttng_dynamic_buffer_set_size(
+			&client->communication.inbound.buffer,
+			client->communication.inbound.bytes_to_receive);
+	return ret;
 }
 
 int handle_notification_thread_client_connect(
@@ -1187,7 +1197,12 @@ int handle_notification_thread_client_connect(
 	CDS_INIT_LIST_HEAD(&client->condition_list);
 	lttng_dynamic_buffer_init(&client->communication.inbound.buffer);
 	lttng_dynamic_buffer_init(&client->communication.outbound.buffer);
-	client_reset_inbound_state(client);
+	ret = client_reset_inbound_state(client);
+	if (ret) {
+		ERR("[notification-thread] Failed to reset client communication's inbound state");
+	        ret = 0;
+		goto error;
+	}
 
 	ret = lttcomm_accept_unix_sock(state->notification_channel_socket);
 	if (ret < 0) {
@@ -1222,7 +1237,6 @@ int handle_notification_thread_client_connect(
 	DBG("[notification-thread] Added new notification channel client socket (%i) to poll set",
 			client->socket);
 
-	/* Add to ht. */
 	rcu_read_lock();
 	cds_lfht_add(state->client_socket_ht,
 			hash_client_socket(client->socket),
@@ -1482,7 +1496,7 @@ int client_dispatch_message(struct notification_client *client,
 			client->communication.inbound.receive_creds = true;
 		}
 		ret = lttng_dynamic_buffer_set_size(
-				&client->communication.inbound.buffer, 0);
+				&client->communication.inbound.buffer, msg->size);
 		if (ret) {
 			goto end;
 		}
@@ -1510,7 +1524,7 @@ int client_dispatch_message(struct notification_client *client,
 		handshake_client =
 				(struct lttng_notification_channel_command_handshake *)
 					client->communication.inbound.buffer.data;
-				client->major = handshake_client->major;
+		client->major = handshake_client->major;
 		client->minor = handshake_client->minor;
 		if (!client->communication.inbound.creds_received) {
 			ERR("[notification-thread] No credentials received from client");
@@ -1549,7 +1563,11 @@ int client_dispatch_message(struct notification_client *client,
 		}
 
 		/* Set reception state to receive the next message header. */
-		client_reset_inbound_state(client);
+		ret = client_reset_inbound_state(client);
+		if (ret) {
+			ERR("[notification-thread] Failed to reset client communication's inbound state");
+			goto end;
+		}
 		client->validated = true;
 		break;
 	}
@@ -1596,7 +1614,11 @@ int client_dispatch_message(struct notification_client *client,
 		}
 
 		/* Set reception state to receive the next message header. */
-		client_reset_inbound_state(client);
+		ret = client_reset_inbound_state(client);
+		if (ret) {
+			ERR("[notification-thread] Failed to reset client communication's inbound state");
+			goto end;
+		}
 		break;
 	}
 	default:
@@ -1610,7 +1632,7 @@ end:
 int handle_notification_thread_client_in(
 		struct notification_thread_state *state, int socket)
 {
-	int ret;
+	int ret = 0;
 	struct notification_client *client;
 	ssize_t recv_ret;
 	size_t offset;
@@ -1622,14 +1644,8 @@ int handle_notification_thread_client_in(
 		goto end;
 	}
 
-	offset = client->communication.inbound.buffer.size;
-	ret = lttng_dynamic_buffer_set_size(
-			&client->communication.inbound.buffer,
-			client->communication.inbound.bytes_to_receive);
-	if (ret) {
-		goto end;
-	}
-
+	offset = client->communication.inbound.buffer.size -
+			client->communication.inbound.bytes_to_receive;
 	if (client->communication.inbound.receive_creds) {
 		recv_ret = lttcomm_recv_creds_unix_sock(socket,
 				client->communication.inbound.buffer.data + offset,
@@ -1649,14 +1665,6 @@ int handle_notification_thread_client_in(
 	}
 
 	client->communication.inbound.bytes_to_receive -= recv_ret;
-	ret = lttng_dynamic_buffer_set_size(
-			&client->communication.inbound.buffer,
-			client->communication.inbound.buffer.size -
-			client->communication.inbound.bytes_to_receive);
-	if (ret) {
-		goto end;
-	}
-
 	if (client->communication.inbound.bytes_to_receive == 0) {
 		ret = client_dispatch_message(client, state);
 		if (ret) {