X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng%2Futils.c;h=912cbd5775b08bd97485c3dd3768a8ed1a6a8a71;hp=c518405cae2944c6fd3e6e997f024eed307ad0ff;hb=e223f308e0f0899d4c6d129922cad25d3b718efc;hpb=0e1155633b00241b8b1e5a9bb683cf91b28f5eeb

diff --git a/src/bin/lttng/utils.c b/src/bin/lttng/utils.c
index c518405ca..912cbd577 100644
--- a/src/bin/lttng/utils.c
+++ b/src/bin/lttng/utils.c
@@ -15,7 +15,7 @@
  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
-#define _GNU_SOURCE
+#define _LGPL_SOURCE
 #include <assert.h>
 #include <stdlib.h>
 #include <ctype.h>
@@ -25,9 +25,12 @@
 #include <signal.h>
 #include <netinet/in.h>
 #include <arpa/inet.h>
+#include <inttypes.h>
+#include <unistd.h>
 
 #include <common/error.h>
 #include <common/utils.h>
+#include <common/defaults.h>
 
 #include "conf.h"
 #include "utils.h"
@@ -39,13 +42,8 @@ static const char *str_jul = "JUL";
 static const char *str_log4j = "LOG4J";
 static const char *str_python = "Python";
 
-/*
- *  get_session_name
- *
- *  Return allocated string with the session name found in the config
- *  directory.
- */
-char *get_session_name(void)
+static
+char *_get_session_name(int quiet)
 {
 	char *path, *session_name = NULL;
 
@@ -56,7 +54,8 @@ char *get_session_name(void)
 	}
 
 	/* Get session name from config */
-	session_name = config_read_session_name(path);
+	session_name = quiet ? config_read_session_name_quiet(path) :
+		config_read_session_name(path);
 	if (session_name == NULL) {
 		goto error;
 	}
@@ -69,6 +68,28 @@ error:
 	return NULL;
 }
 
+/*
+ *  get_session_name
+ *
+ *  Return allocated string with the session name found in the config
+ *  directory.
+ */
+char *get_session_name(void)
+{
+	return _get_session_name(0);
+}
+
+/*
+ *  get_session_name_quiet (no warnings/errors emitted)
+ *
+ *  Return allocated string with the session name found in the config
+ *  directory.
+ */
+char *get_session_name_quiet(void)
+{
+	return _get_session_name(1);
+}
+
 /*
  *  list_commands
  *
@@ -320,14 +341,14 @@ int spawn_relayd(const char *pathname, int port)
 		if (errno == ENOENT) {
 			ERR("No relayd found. Use --relayd-path.");
 		} else {
-			perror("execlp");
+			PERROR("execlp");
 		}
 		kill(getppid(), SIGTERM);	/* wake parent */
 		exit(EXIT_FAILURE);
 	} else if (pid > 0) {
 		goto end;
 	} else {
-		perror("fork");
+		PERROR("fork");
 		ret = -1;
 		goto end;
 	}
@@ -348,7 +369,7 @@ int check_relayd(void)
 
 	fd = socket(AF_INET, SOCK_STREAM, 0);
 	if (fd < 0) {
-		perror("socket check relayd");
+		PERROR("socket check relayd");
 		ret = -1;
 		goto error_socket;
 	}
@@ -357,7 +378,7 @@ int check_relayd(void)
 	sin.sin_port = htons(DEFAULT_NETWORK_VIEWER_PORT);
 	ret = inet_pton(sin.sin_family, "127.0.0.1", &sin.sin_addr);
 	if (ret < 1) {
-		perror("inet_pton check relayd");
+		PERROR("inet_pton check relayd");
 		ret = -1;
 		goto error;
 	}
@@ -377,8 +398,97 @@ int check_relayd(void)
 
 error:
 	if (close(fd) < 0) {
-		perror("close relayd fd");
+		PERROR("close relayd fd");
 	}
 error_socket:
 	return ret;
 }
+
+int print_missing_or_multiple_domains(unsigned int sum)
+{
+	int ret = 0;
+
+	if (sum == 0) {
+		ERR("Please specify a domain (-k/-u/-j).");
+		ret = -1;
+	} else if (sum > 1) {
+		ERR("Multiple domains specified.");
+		ret = -1;
+	}
+
+	return ret;
+}
+
+/*
+ * Get the discarded events and lost packet counts.
+ */
+void print_session_stats(const char *session_name)
+{
+	int count, nb_domains, domain_idx, channel_idx;
+	struct lttng_domain *domains;
+	struct lttng_channel *channels;
+	uint64_t discarded_total = 0, lost_total = 0;
+
+	nb_domains = lttng_list_domains(session_name, &domains);
+	if (nb_domains < 0) {
+		goto end;
+	}
+	for (domain_idx = 0; domain_idx < nb_domains; domain_idx++) {
+		struct lttng_handle *handle = lttng_create_handle(session_name,
+				&domains[domain_idx]);
+
+		if (!handle) {
+			ERR("Failed to create session handle while printing session stats.");
+			goto end;
+		}
+
+		count = lttng_list_channels(handle, &channels);
+		for (channel_idx = 0; channel_idx < count; channel_idx++) {
+			int ret;
+			uint64_t discarded = 0, lost = 0;
+			struct lttng_channel *channel = &channels[channel_idx];
+
+			ret = lttng_channel_get_discarded_event_count(channel,
+					&discarded);
+			if (ret) {
+				ERR("Failed to retrieve discarded event count from channel %s",
+						channel->name);
+			}
+
+			ret = lttng_channel_get_lost_packet_count(channel,
+					&lost);
+			if (ret) {
+				ERR("Failed to retrieve lost packet count from channel %s",
+						channel->name);
+			}
+
+			discarded_total += discarded;
+			lost_total += lost;
+		}
+		lttng_destroy_handle(handle);
+	}
+	if (discarded_total > 0) {
+		MSG("[warning] %" PRIu64 " events discarded, please refer to "
+				"the documentation on channel configuration.",
+				discarded_total);
+	}
+	if (lost_total > 0) {
+		MSG("[warning] %" PRIu64 " packets lost, please refer to "
+				"the documentation on channel configuration.",
+				lost_total);
+	}
+
+end:
+	return;
+}
+
+int show_cmd_man_page(const char *cmd_name)
+{
+	int ret;
+	char page_name[32];
+
+	ret = sprintf(page_name, "lttng-%s", cmd_name);
+	assert(ret > 0 && ret < 32);
+
+	return utils_show_man_page(1, page_name);
+}