X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=src%2Fbin%2Flttng%2Futils.cpp;h=e799e2f906e5e6fed14311dfc704b5472a9d775c;hb=4d4c8b8e68081f91347458b7e072dcfcac32385b;hp=90ca90fdeb3a4892424e62827d01e920d7951ef5;hpb=cd9adb8b829564212158943a0d279bb35322ab30;p=lttng-tools.git

diff --git a/src/bin/lttng/utils.cpp b/src/bin/lttng/utils.cpp
index 90ca90fde..e799e2f90 100644
--- a/src/bin/lttng/utils.cpp
+++ b/src/bin/lttng/utils.cpp
@@ -12,11 +12,15 @@
 
 #include <common/defaults.hpp>
 #include <common/error.hpp>
+#include <common/exception.hpp>
+#include <common/make-unique-wrapper.hpp>
 #include <common/utils.hpp>
 
 #include <arpa/inet.h>
 #include <ctype.h>
+#include <fnmatch.h>
 #include <inttypes.h>
+#include <iostream>
 #include <limits.h>
 #include <netinet/in.h>
 #include <signal.h>
@@ -659,3 +663,131 @@ end:
 	}
 	return ret;
 }
+
+namespace {
+template <typename FilterFunctionType>
+lttng::cli::session_list get_sessions(const FilterFunctionType& filter,
+				      bool return_first_match_only = false)
+{
+	lttng::cli::session_list list = []() {
+		int list_ret;
+		struct lttng_session *psessions;
+
+		list_ret = lttng_list_sessions(&psessions);
+
+		if (list_ret < 0) {
+			LTTNG_THROW_CTL("Failed to list sessions",
+					static_cast<lttng_error_code>(list_ret));
+		}
+
+		return lttng::cli::session_list(psessions, list_ret);
+	}();
+
+	std::size_t write_to = 0;
+	for (std::size_t read_from = 0; read_from < list.size(); ++read_from) {
+		if (!filter(list[read_from])) {
+			continue;
+		}
+
+		if (read_from != write_to) {
+			list[write_to] = list[read_from];
+		}
+
+		++write_to;
+
+		if (return_first_match_only) {
+			return lttng::cli::session_list(std::move(list), 1);
+		}
+	}
+
+	list.resize(write_to);
+
+	return list;
+}
+} /* namespace */
+
+lttng::cli::session_list lttng::cli::list_sessions(const struct session_spec& spec)
+{
+	switch (spec.type_) {
+	case lttng::cli::session_spec::type::NAME:
+		if (spec.value == nullptr) {
+			const auto configured_name =
+				lttng::make_unique_wrapper<char, lttng::free>(get_session_name());
+
+			if (configured_name) {
+				const struct lttng::cli::session_spec new_spec(
+					lttng::cli::session_spec::type::NAME,
+					configured_name.get());
+
+				return list_sessions(new_spec);
+			}
+
+			return lttng::cli::session_list();
+		}
+
+		return get_sessions(
+			[&spec](const lttng_session& session) {
+				return strcmp(session.name, spec.value) == 0;
+			},
+			true);
+	case lttng::cli::session_spec::type::GLOB_PATTERN:
+		return get_sessions([&spec](const lttng_session& session) {
+			return fnmatch(spec.value, session.name, 0) == 0;
+		});
+	case lttng::cli::session_spec::type::ALL:
+		return get_sessions([](const lttng_session&) { return true; });
+	}
+
+	return lttng::cli::session_list();
+}
+
+void print_kernel_tracer_status_error()
+{
+	if (lttng_opt_mi) {
+		return;
+	}
+
+	enum lttng_kernel_tracer_status kernel_tracer_status;
+	const auto ret = lttng_get_kernel_tracer_status(&kernel_tracer_status);
+
+	if (ret < 0) {
+		ERR("Failed to get kernel tracer status: %s", lttng_strerror(ret));
+	} else {
+		switch (kernel_tracer_status) {
+		case LTTNG_KERNEL_TRACER_STATUS_INITIALIZED:
+			return;
+		case LTTNG_KERNEL_TRACER_STATUS_ERR_MODULES_UNKNOWN:
+			std::cerr << "\tKernel module loading failed" << std::endl;
+			break;
+		case LTTNG_KERNEL_TRACER_STATUS_ERR_MODULES_MISSING:
+			std::cerr << "\tMissing one or more required kernel modules" << std::endl;
+			break;
+		case LTTNG_KERNEL_TRACER_STATUS_ERR_MODULES_SIGNATURE:
+			std::cerr
+				<< "\tKernel module signature error prevented loading of one or more required kernel modules"
+				<< std::endl;
+			break;
+		case LTTNG_KERNEL_TRACER_STATUS_ERR_NEED_ROOT:
+			std::cerr << "\tlttng-sessiond isn't running as root" << std::endl;
+			break;
+		case LTTNG_KERNEL_TRACER_STATUS_ERR_NOTIFIER:
+			std::cerr << "\tFailed to setup notifiers" << std::endl;
+			break;
+		case LTTNG_KERNEL_TRACER_STATUS_ERR_OPEN_PROC_LTTNG:
+			std::cerr << "\tlttng-sessiond failed to open /proc/lttng" << std::endl;
+			break;
+		case LTTNG_KERNEL_TRACER_STATUS_ERR_VERSION_MISMATCH:
+			std::cerr
+				<< "\tVersion mismatch between kernel tracer and kernel tracer ABI"
+				<< std::endl;
+			break;
+		default:
+			std::cerr << lttng::format("\t\tUnknown kernel tracer status (%d)",
+						   static_cast<int>(kernel_tracer_status))
+				  << std::endl;
+			break;
+		}
+
+		std::cerr << "\tConsult lttng-sessiond logs for more information" << std::endl;
+	}
+}