X-Git-Url: https://git.liburcu.org/?a=blobdiff_plain;f=src%2Fbin%2Flttng%2Futils.hpp;h=262775d1d6bf0513e75f83c1ff0a939c68266b7d;hb=42a11b8f4c8d98b33fa1eadcdcda96253e651f0b;hp=64605575e94c33da7c3fdb0f2e4d52bf67a817b0;hpb=28f23191dcbf047429d51950a337a57d7a3f866a;p=lttng-tools.git

diff --git a/src/bin/lttng/utils.hpp b/src/bin/lttng/utils.hpp
index 64605575e..262775d1d 100644
--- a/src/bin/lttng/utils.hpp
+++ b/src/bin/lttng/utils.hpp
@@ -10,9 +10,12 @@
 
 #include <common/argpar/argpar.h>
 #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 +25,149 @@ extern pid_t sessiond_pid;
 
 struct cmd_struct;
 
+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;
+};
+
+/*
+ * We don't use a std::vector here because it would make a copy of the C array.
+ */
+class session_list {
+	template <typename ContainerType, typename DereferenceReturnType>
+	class _iterator
+		: public std::iterator<std::random_access_iterator_tag, std::size_t> {
+	public:
+		explicit _iterator(ContainerType& list, std::size_t k) :
+			_list(list), _index(k)
+		{
+		}
+
+		_iterator& operator++() noexcept
+		{
+			++_index;
+			return *this;
+		}
+
+		_iterator& operator--() noexcept
+		{
+			--_index;
+			return *this;
+		}
+
+		_iterator& operator++(int) noexcept
+		{
+			_index++;
+			return *this;
+		}
+
+		_iterator& operator--(int) noexcept
+		{
+			_index--;
+			return *this;
+		}
+
+		bool operator==(_iterator other) const noexcept
+		{
+			return _index == other._index;
+		}
+
+		bool operator!=(_iterator other) const noexcept
+		{
+			return !(*this == other);
+		}
+
+		DereferenceReturnType& operator*() const noexcept
+		{
+			return _list[_index];
+		}
+
+	private:
+		ContainerType& _list;
+		std::size_t _index;
+	};
+
+	using iterator = _iterator<session_list, lttng_session>;
+	using const_iterator = _iterator<const session_list, const lttng_session>;
+
+public:
+	session_list() : _sessions_count(0), _sessions(nullptr)
+	{
+	}
+
+	session_list(session_list&& original, std::size_t new_count)
+	{
+		_sessions_count = new_count;
+		_sessions = std::move(original._sessions);
+	}
+
+	session_list(struct lttng_session *raw_sessions, std::size_t raw_sessions_count)
+	{
+		_sessions_count = raw_sessions_count;
+		_sessions.reset(raw_sessions);
+	}
+
+	iterator begin() noexcept
+	{
+		return iterator(*this, 0);
+	}
+
+	iterator end() noexcept
+	{
+		return iterator(*this, _sessions_count);
+	}
+
+	const_iterator begin() const noexcept
+	{
+		return const_iterator(*this, 0);
+	}
+
+	const_iterator end() const noexcept
+	{
+		return const_iterator(*this, _sessions_count);
+	}
+
+	std::size_t size() const noexcept
+	{
+		return _sessions_count;
+	}
+
+	void resize(std::size_t new_size) noexcept
+	{
+		_sessions_count = new_size;
+	}
+
+	lttng_session& operator[](std::size_t index)
+	{
+		LTTNG_ASSERT(index < _sessions_count);
+		return _sessions.get()[index];
+	}
+
+	const lttng_session& operator[](std::size_t index) const
+	{
+		LTTNG_ASSERT(index < _sessions_count);
+		return _sessions.get()[index];
+	}
+
+private:
+	std::size_t _sessions_count;
+	std::unique_ptr<lttng_session,
+			lttng::memory::create_deleter_class<lttng_session, lttng::free>::deleter>
+		_sessions;
+};
+
 char *get_session_name(void);
 char *get_session_name_quiet(void);
 void list_commands(struct cmd_struct *commands, FILE *ofp);
@@ -62,4 +208,6 @@ int print_trace_archive_location(const struct lttng_trace_archive_location *loca
 int validate_exclusion_list(const char *event_name,
 			    const struct lttng_dynamic_pointer_array *exclusions);
 
+session_list list_sessions(const struct session_spec& spec);
+
 #endif /* _LTTNG_UTILS_H */