From: Francis Deslauriers <francis.deslauriers@efficios.com>
Date: Thu, 18 Nov 2021 17:38:11 +0000 (-0500)
Subject: relayd: add `lttng_viewer_command_str()`
X-Git-Url: https://git.lttng.org./?a=commitdiff_plain;h=7201079fc6b433affa4e6d5f0cd3cc8a08b549ac;p=lttng-tools.git

relayd: add `lttng_viewer_command_str()`

Signed-off-by: Francis Deslauriers <francis.deslauriers@efficios.com>
Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>
Change-Id: Icb7be4901727654c6339ed47344320140fdaf9fb
---

diff --git a/src/bin/lttng-relayd/live.cpp b/src/bin/lttng-relayd/live.cpp
index 46faea031..e3d27b1ef 100644
--- a/src/bin/lttng-relayd/live.cpp
+++ b/src/bin/lttng-relayd/live.cpp
@@ -89,6 +89,33 @@ static uint64_t last_relay_viewer_session_id;
 static pthread_mutex_t last_relay_viewer_session_id_lock =
 		PTHREAD_MUTEX_INITIALIZER;
 
+static
+const char *lttng_viewer_command_str(lttng_viewer_command cmd)
+{
+	switch (cmd) {
+	case LTTNG_VIEWER_CONNECT:
+		return "CONNECT";
+	case LTTNG_VIEWER_LIST_SESSIONS:
+		return "LIST_SESSIONS";
+	case LTTNG_VIEWER_ATTACH_SESSION:
+		return "ATTACH_SESSION";
+	case LTTNG_VIEWER_GET_NEXT_INDEX:
+		return "GET_NEXT_INDEX";
+	case LTTNG_VIEWER_GET_PACKET:
+		return "GET_PACKET";
+	case LTTNG_VIEWER_GET_METADATA:
+		return "GET_METADATA";
+	case LTTNG_VIEWER_GET_NEW_STREAMS:
+		return "GET_NEW_STREAMS";
+	case LTTNG_VIEWER_CREATE_SESSION:
+		return "CREATE_SESSION";
+	case LTTNG_VIEWER_DETACH_SESSION:
+		return "DETACH_SESSION";
+	default:
+		abort();
+	}
+}
+
 /*
  * Cleanup the daemon
  */
@@ -937,8 +964,6 @@ int viewer_connect(struct relay_connection *conn)
 
 	health_code_update();
 
-	DBG("Viewer is establishing a connection to the relayd.");
-
 	ret = recv_request(conn->sock, &msg, sizeof(msg));
 	if (ret < 0) {
 		goto end;
@@ -1025,8 +1050,6 @@ int viewer_list_sessions(struct relay_connection *conn)
 	uint32_t buf_count = SESSION_BUF_DEFAULT_COUNT;
 	uint32_t count = 0;
 
-	DBG("List sessions received");
-
 	send_session_buf = (lttng_viewer_session *) zmalloc(SESSION_BUF_DEFAULT_COUNT * sizeof(*send_session_buf));
 	if (!send_session_buf) {
 		return -1;
@@ -1131,8 +1154,6 @@ int viewer_get_new_streams(struct relay_connection *conn)
 
 	LTTNG_ASSERT(conn);
 
-	DBG("Get new streams received");
-
 	health_code_update();
 
 	/* Receive the request from the connected client. */
@@ -1609,8 +1630,6 @@ int viewer_get_next_index(struct relay_connection *conn)
 
 	LTTNG_ASSERT(conn);
 
-	DBG("Viewer get next index");
-
 	memset(&viewer_index, 0, sizeof(viewer_index));
 	health_code_update();
 
@@ -1915,8 +1934,6 @@ int viewer_get_packet(struct relay_connection *conn)
 	ssize_t read_len;
 	uint64_t stream_id;
 
-	DBG2("Relay get data packet");
-
 	health_code_update();
 
 	ret = recv_request(conn->sock, &get_packet_info,
@@ -2029,8 +2046,6 @@ int viewer_get_metadata(struct relay_connection *conn)
 
 	LTTNG_ASSERT(conn);
 
-	DBG("Relay get metadata");
-
 	health_code_update();
 
 	ret = recv_request(conn->sock, &request, sizeof(request));
@@ -2290,8 +2305,6 @@ int viewer_create_session(struct relay_connection *conn)
 	int ret;
 	struct lttng_viewer_create_session_response resp;
 
-	DBG("Viewer create session received");
-
 	memset(&resp, 0, sizeof(resp));
 	resp.status = htobe32(LTTNG_VIEWER_CREATE_SESSION_OK);
 	conn->viewer_session = viewer_session_create();
@@ -2328,8 +2341,6 @@ int viewer_detach_session(struct relay_connection *conn)
 	struct relay_session *session = NULL;
 	uint64_t viewer_session_to_close;
 
-	DBG("Viewer detach session received");
-
 	LTTNG_ASSERT(conn);
 
 	health_code_update();
@@ -2408,21 +2419,24 @@ int process_control(struct lttng_viewer_cmd *recv_hdr,
 		struct relay_connection *conn)
 {
 	int ret = 0;
-	uint32_t msg_value;
-
-	msg_value = be32toh(recv_hdr->cmd);
+	lttng_viewer_command cmd =
+			(lttng_viewer_command) be32toh(recv_hdr->cmd);
 
 	/*
-	 * Make sure we've done the version check before any command other then a
-	 * new client connection.
+	 * Make sure we've done the version check before any command other then
+	 * a new client connection.
 	 */
-	if (msg_value != LTTNG_VIEWER_CONNECT && !conn->version_check_done) {
-		ERR("Viewer conn value %" PRIu32 " before version check", msg_value);
+	if (cmd != LTTNG_VIEWER_CONNECT && !conn->version_check_done) {
+		ERR("Viewer on connection %d requested %s command before version check",
+			conn->sock->fd, lttng_viewer_command_str(cmd));
 		ret = -1;
 		goto end;
 	}
 
-	switch (msg_value) {
+	DBG("Processing %s viewer command from connection %d",
+			lttng_viewer_command_str(cmd), conn->sock->fd);
+
+	switch (cmd) {
 	case LTTNG_VIEWER_CONNECT:
 		ret = viewer_connect(conn);
 		break;