X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-sessiond%2Fnotification-thread-events.c;h=6e507642a224be3affae1540bd762bc2a138849a;hp=285bcf951aafe3a5bb2801777e7a13eb06dc5ec7;hb=dd877ea65a8db6b9abcffec9162695059ced90f7;hpb=505b2d90aa87592186ecc2a119cf67fb3f90d168

diff --git a/src/bin/lttng-sessiond/notification-thread-events.c b/src/bin/lttng-sessiond/notification-thread-events.c
index 285bcf951..6e507642a 100644
--- a/src/bin/lttng-sessiond/notification-thread-events.c
+++ b/src/bin/lttng-sessiond/notification-thread-events.c
@@ -2720,13 +2720,61 @@ int handle_notification_thread_trigger_unregister_all(
 	return error_occurred ? -1 : 0;
 }
 
+static
+int client_handle_transmission_status(
+		struct notification_client *client,
+		enum client_transmission_status transmission_status,
+		struct notification_thread_state *state)
+{
+	int ret = 0;
+
+	switch (transmission_status) {
+	case CLIENT_TRANSMISSION_STATUS_COMPLETE:
+		ret = lttng_poll_mod(&state->events, client->socket,
+				CLIENT_POLL_MASK_IN);
+		if (ret) {
+			goto end;
+		}
+
+		client->communication.outbound.queued_command_reply = false;
+		client->communication.outbound.dropped_notification = false;
+		break;
+	case CLIENT_TRANSMISSION_STATUS_QUEUED:
+		/*
+		 * We want to be notified whenever there is buffer space
+		 * available to send the rest of the payload.
+		 */
+		ret = lttng_poll_mod(&state->events, client->socket,
+				CLIENT_POLL_MASK_IN_OUT);
+		if (ret) {
+			goto end;
+		}
+		break;
+	case CLIENT_TRANSMISSION_STATUS_FAIL:
+		ret = notification_thread_client_disconnect(client, state);
+		if (ret) {
+			goto end;
+		}
+		break;
+	case CLIENT_TRANSMISSION_STATUS_ERROR:
+		ret = -1;
+		goto end;
+	default:
+		abort();
+	}
+end:
+	return ret;
+}
+
 /* Client lock must be acquired by caller. */
 static
-int client_flush_outgoing_queue(struct notification_client *client,
+enum client_transmission_status client_flush_outgoing_queue(
+		struct notification_client *client,
 		struct notification_thread_state *state)
 {
 	ssize_t ret;
 	size_t to_send_count;
+	enum client_transmission_status status;
 
 	ASSERT_LOCKED(client->lock);
 
@@ -2751,41 +2799,29 @@ int client_flush_outgoing_queue(struct notification_client *client,
 				&client->communication.outbound.buffer,
 				to_send_count);
 		if (ret) {
+			status = CLIENT_TRANSMISSION_STATUS_ERROR;
 			goto error;
 		}
-
-		/*
-		 * We want to be notified whenever there is buffer space
-		 * available to send the rest of the payload.
-		 */
-		ret = lttng_poll_mod(&state->events, client->socket,
-				CLIENT_POLL_MASK_IN_OUT);
-		if (ret) {
-			goto error;
-		}
+		status = CLIENT_TRANSMISSION_STATUS_QUEUED;
 	} else if (ret < 0) {
 		/* Generic error, disconnect the client. */
-		ERR("[notification-thread] Failed to send flush outgoing queue, disconnecting client (socket fd = %i)",
+		ERR("[notification-thread] Failed to flush outgoing queue, disconnecting client (socket fd = %i)",
 				client->socket);
-		ret = notification_thread_client_disconnect(client, state);
-		if (ret) {
-			goto error;
-		}
+		status = CLIENT_TRANSMISSION_STATUS_FAIL;
 	} else {
 		/* No error and flushed the queue completely. */
 		ret = lttng_dynamic_buffer_set_size(
 				&client->communication.outbound.buffer, 0);
 		if (ret) {
+			status = CLIENT_TRANSMISSION_STATUS_ERROR;
 			goto error;
 		}
-		ret = lttng_poll_mod(&state->events, client->socket,
-				CLIENT_POLL_MASK_IN);
-		if (ret) {
-			goto error;
-		}
+		status = CLIENT_TRANSMISSION_STATUS_COMPLETE;
+	}
 
-		client->communication.outbound.queued_command_reply = false;
-		client->communication.outbound.dropped_notification = false;
+	ret = client_handle_transmission_status(client, status, state);
+	if (ret) {
+		goto error;
 	}
 
 	return 0;