X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=src%2Fbin%2Flttng%2Futils.hpp;h=000509da5f4ad2dacfec76c8525dbd5c02650fd5;hb=dcd24bbf7dbc74e3584d1d0d52715e749023c452;hp=64605575e94c33da7c3fdb0f2e4d52bf67a817b0;hpb=28f23191dcbf047429d51950a337a57d7a3f866a;p=lttng-tools.git

diff --git a/src/bin/lttng/utils.hpp b/src/bin/lttng/utils.hpp
index 64605575e..000509da5 100644
--- a/src/bin/lttng/utils.hpp
+++ b/src/bin/lttng/utils.hpp
@@ -9,10 +9,14 @@
 #define _LTTNG_UTILS_H
 
 #include <common/argpar/argpar.h>
+#include <common/container-wrapper.hpp>
 #include <common/dynamic-array.hpp>
+#include <common/make-unique-wrapper.hpp>
 
 #include <lttng/lttng.h>
 
+#include <iterator>
+#include <memory>
 #include <popt.h>
 
 extern char *opt_relayd_path;
@@ -22,6 +26,119 @@ extern pid_t sessiond_pid;
 
 struct cmd_struct;
 
+namespace lttng {
+namespace cli {
+
+struct session_spec {
+	enum class type {
+		NAME,
+		GLOB_PATTERN,
+		ALL,
+	};
+
+	explicit session_spec(type spec_type, const char *name_or_pattern = nullptr) noexcept :
+		type_(spec_type), value(name_or_pattern)
+	{
+	}
+
+	/* Disambiguate type enum from the member for buggy g++ versions. */
+	type type_;
+	const char *value;
+};
+
+class session_list;
+
+namespace details {
+class session_storage {
+public:
+	session_storage(lttng_session *raw_sessions, std::size_t sessions_count) :
+		_array(raw_sessions), _count(sessions_count)
+	{
+	}
+
+	session_storage(session_storage&& original) :
+		_array(std::move(original._array)), _count(original._count)
+	{
+	}
+
+	session_storage(session_storage&& original, std::size_t new_count) :
+		_array(std::move(original._array)), _count(new_count)
+	{
+	}
+
+	std::unique_ptr<lttng_session,
+			lttng::memory::create_deleter_class<lttng_session, lttng::free>::deleter>
+		_array = nullptr;
+	std::size_t _count = 0;
+};
+
+class session_list_operations {
+public:
+	static lttng_session& get(const lttng::cli::details::session_storage& storage,
+				  std::size_t index) noexcept
+	{
+		return storage._array.get()[index];
+	}
+
+	static std::size_t size(const lttng::cli::details::session_storage& storage)
+	{
+		return storage._count;
+	}
+};
+} /* namespace details */
+
+/*
+ * We don't use a std::vector here because it would make a copy of the C array.
+ */
+class session_list
+	: public lttng::utils::random_access_container_wrapper<details::session_storage,
+							       lttng_session&,
+							       details::session_list_operations> {
+public:
+	friend details::session_list_operations;
+
+	session_list() :
+		lttng::utils::random_access_container_wrapper<details::session_storage,
+							      lttng_session&,
+							      details::session_list_operations>(
+			{ nullptr, 0 })
+	{
+	}
+
+	session_list(session_list&& original) :
+		lttng::utils::random_access_container_wrapper<details::session_storage,
+							      lttng_session&,
+							      details::session_list_operations>(
+			std::move(original._container))
+	{
+	}
+
+	session_list(session_list&& original, std::size_t new_count) :
+		lttng::utils::random_access_container_wrapper<details::session_storage,
+							      lttng_session&,
+							      details::session_list_operations>(
+			{ std::move(original._container), new_count })
+	{
+	}
+
+	session_list(lttng_session *raw_sessions, std::size_t raw_sessions_count) :
+		lttng::utils::random_access_container_wrapper<details::session_storage,
+							      lttng_session&,
+							      details::session_list_operations>(
+			{ raw_sessions, raw_sessions_count })
+	{
+	}
+
+	void resize(std::size_t new_size) noexcept
+	{
+		_container._count = new_size;
+	}
+};
+
+lttng::cli::session_list list_sessions(const struct session_spec& spec);
+} /* namespace cli */
+} /* namespace lttng */
+
 char *get_session_name(void);
 char *get_session_name_quiet(void);
 void list_commands(struct cmd_struct *commands, FILE *ofp);