.gitignore: add some missing files
[babeltrace.git] / src / plugins / ctf / lttng-live / lttng-live.hpp
index 87c80e533d999b2fd60c0a0c3a8a48f676ddf374..384cc424194944e219c6e87374dbe75df79d6260 100644 (file)
@@ -8,20 +8,46 @@
  * BabelTrace - LTTng-live client Component
  */
 
-#ifndef BABELTRACE_PLUGIN_CTF_LTTNG_LIVE_H
-#define BABELTRACE_PLUGIN_CTF_LTTNG_LIVE_H
+#ifndef BABELTRACE_PLUGINS_CTF_LTTNG_LIVE_LTTNG_LIVE_HPP
+#define BABELTRACE_PLUGINS_CTF_LTTNG_LIVE_LTTNG_LIVE_HPP
 
 #include <glib.h>
 #include <stdint.h>
 
 #include <babeltrace2/babeltrace.h>
 
+#include "cpp-common/bt2/message.hpp"
 #include "cpp-common/vendor/fmt/format.h" /* IWYU pragma: keep */
 
-#include "../common/src/metadata/tsdl/decoder.hpp"
-#include "../common/src/msg-iter/msg-iter.hpp"
+#include "../common/src/metadata/metadata-stream-parser-utils.hpp"
+#include "../common/src/msg-iter.hpp"
 #include "viewer-connection.hpp"
 
+/*
+ * bt_common_lttng_live_url_parts is defined in common code, and is also used
+ * by C code, so it can't be C++-ified yet.  Use this separate deleter object
+ * in the mean time.
+ */
+struct bt_common_lttng_live_url_parts_deleter
+{
+    explicit bt_common_lttng_live_url_parts_deleter(bt_common_lttng_live_url_parts& obj) noexcept :
+        _mObj {&obj}
+    {
+    }
+
+    bt_common_lttng_live_url_parts_deleter(const bt_common_lttng_live_url_parts_deleter&) = delete;
+    bt_common_lttng_live_url_parts&
+    operator=(const bt_common_lttng_live_url_parts_deleter&) = delete;
+
+    ~bt_common_lttng_live_url_parts_deleter()
+    {
+        bt_common_destroy_lttng_live_url_parts(_mObj);
+    }
+
+private:
+    bt_common_lttng_live_url_parts *_mObj;
+};
+
 enum lttng_live_stream_state
 {
     /* This stream won't have data until some known time in the future. */
@@ -67,15 +93,18 @@ inline const char *format_as(const lttng_live_stream_state state) noexcept
 /* Iterator over a live stream. */
 struct lttng_live_stream_iterator
 {
+    using UP = std::unique_ptr<lttng_live_stream_iterator>;
+
     explicit lttng_live_stream_iterator(const bt2c::Logger& parentLogger) :
         logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/STREAM-ITER"}
     {
     }
 
+    ~lttng_live_stream_iterator();
+
     bt2c::Logger logger;
 
-    /* Owned by this. */
-    bt_stream *stream = nullptr;
+    bt2::Stream::Shared stream;
 
     /* Weak reference. */
     struct lttng_live_trace *trace = nullptr;
@@ -84,7 +113,7 @@ struct lttng_live_stream_iterator
      * Since only a single iterator per viewer connection, we have
      * only a single message iterator per stream.
      */
-    struct ctf_msg_iter *msg_iter = nullptr;
+    bt2s::optional<ctf::src::MsgIter> msg_iter;
 
     uint64_t viewer_stream_id = 0;
 
@@ -94,13 +123,6 @@ struct lttng_live_stream_iterator
         uint64_t value = 0;
     } ctf_stream_class_id;
 
-    /* base offset in current index. */
-    uint64_t base_offset = 0;
-    /* len to read in current index. */
-    uint64_t len = 0;
-    /* offset in current index. */
-    uint64_t offset = 0;
-
     /*
      * Clock Snapshot value of the last message iterator inactivity message
      * sent downstream.
@@ -119,38 +141,69 @@ struct lttng_live_stream_iterator
 
     enum lttng_live_stream_state state = LTTNG_LIVE_STREAM_QUIESCENT;
 
-    /*
-     * The current message produced by this live stream iterator. Owned by
-     * this.
-     */
-    const bt_message *current_msg = nullptr;
+    /* The current message produced by this live stream iterator. */
+    bt2::ConstMessage::Shared current_msg;
 
     /* Timestamp in nanoseconds of the current message (current_msg). */
     int64_t current_msg_ts_ns = 0;
 
-    /* Owned by this. */
-    uint8_t *buf = nullptr;
-    size_t buflen = 0;
-
-    /* Owned by this. */
-    GString *name = nullptr;
+    std::string name;
 
     bool has_stream_hung_up = false;
+
+    struct CurPktInfo
+    {
+        bt2c::DataLen offsetInRelay;
+        bt2c::DataLen len;
+    };
+
+    bt2s::optional<CurPktInfo> curPktInfo;
 };
 
 struct lttng_live_metadata
 {
-    explicit lttng_live_metadata(const bt2c::Logger& parentLogger) :
-        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/METADATA"}
+    using UP = std::unique_ptr<lttng_live_metadata>;
+
+    explicit lttng_live_metadata(const bt2::SelfComponent selfComp,
+                                 const bt2c::Logger& parentLogger) :
+        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/METADATA"},
+        _mSelfComp {selfComp}
+
     {
     }
 
+    const ctf::src::TraceCls *traceCls() const
+    {
+        return _mMetadataStreamParser->traceCls();
+    }
+
+    const bt2s::optional<bt2c::Uuid>& metadataStreamUuid() const noexcept
+    {
+        return _mMetadataStreamParser->metadataStreamUuid();
+    }
+
+    void parseSection(const bt2c::ConstBytes data)
+    {
+        if (!_mMetadataStreamParser) {
+            _mMetadataStreamParser =
+                ctf::src::createMetadataStreamParser(data, _mSelfComp, {}, logger);
+        }
+
+        _mMetadataStreamParser->parseSection(data);
+    }
+
+    bt2::SelfComponent selfComp() const noexcept
+    {
+        return _mSelfComp;
+    }
+
     bt2c::Logger logger;
 
     uint64_t stream_id = 0;
 
-    /* Weak reference. */
-    ctf_metadata_decoder_up decoder;
+private:
+    bt2::SelfComponent _mSelfComp;
+    ctf::src::MetadataStreamParser::UP _mMetadataStreamParser;
 };
 
 enum lttng_live_metadata_stream_state
@@ -177,6 +230,8 @@ enum lttng_live_metadata_stream_state
 
 struct lttng_live_trace
 {
+    using UP = std::unique_ptr<lttng_live_trace>;
+
     explicit lttng_live_trace(const bt2c::Logger& parentLogger) :
         logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/TRACE"}
     {
@@ -190,19 +245,13 @@ struct lttng_live_trace
     /* ctf trace ID within the session. */
     uint64_t id = 0;
 
-    /* Owned by this. */
-    bt_trace *trace = nullptr;
-
-    /* Weak reference. */
-    bt_trace_class *trace_class = nullptr;
+    bt2::Trace::Shared trace;
 
-    struct lttng_live_metadata *metadata = nullptr;
+    lttng_live_metadata::UP metadata;
 
-    const bt_clock_class *clock_class = nullptr;
+    bt2::OptionalBorrowedObject<bt2::ConstClockClass> clock_class;
 
-    /* Array of pointers to struct lttng_live_stream_iterator. */
-    /* Owned by this. */
-    GPtrArray *stream_iterators = nullptr;
+    std::vector<lttng_live_stream_iterator::UP> stream_iterators;
 
     enum lttng_live_metadata_stream_state metadata_stream_state =
         LTTNG_LIVE_METADATA_STREAM_STATE_NEEDED;
@@ -210,28 +259,31 @@ struct lttng_live_trace
 
 struct lttng_live_session
 {
-    explicit lttng_live_session(const bt2c::Logger& parentLogger) :
-        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/SESSION"}
+    using UP = std::unique_ptr<lttng_live_session>;
+
+    explicit lttng_live_session(const bt2c::Logger& parentLogger,
+                                const bt2::SelfComponent selfCompParam) :
+        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/SESSION"},
+        selfComp {selfCompParam}
     {
     }
 
+    ~lttng_live_session();
+
     bt2c::Logger logger;
 
-    bt_self_component *self_comp = nullptr;
+    bt2::SelfComponent selfComp;
 
     /* Weak reference. */
     struct lttng_live_msg_iter *lttng_live_msg_iter = nullptr;
 
-    /* Owned by this. */
-    GString *hostname = nullptr;
+    std::string hostname;
 
-    /* Owned by this. */
-    GString *session_name = nullptr;
+    std::string session_name;
 
     uint64_t id = 0;
 
-    /* Array of pointers to struct lttng_live_trace. */
-    GPtrArray *traces = nullptr;
+    std::vector<lttng_live_trace::UP> traces;
 
     bool attached = false;
     bool new_streams_needed = false;
@@ -251,19 +303,22 @@ enum session_not_found_action
  */
 struct lttng_live_component
 {
-    explicit lttng_live_component(bt2c::Logger loggerParam) noexcept :
-        logger {std::move(loggerParam)}
+    using UP = std::unique_ptr<lttng_live_component>;
+
+    explicit lttng_live_component(bt2c::Logger loggerParam,
+                                  const bt2::SelfComponent selfCompParam) noexcept :
+        logger {std::move(loggerParam)},
+        selfComp {selfCompParam}
     {
     }
 
     bt2c::Logger logger;
 
-    /* Weak reference. */
-    bt_self_component *self_comp = nullptr;
+    bt2::SelfComponent selfComp;
 
     struct
     {
-        GString *url = nullptr;
+        std::string url;
         enum session_not_found_action sess_not_found_act = SESSION_NOT_FOUND_ACTION_CONTINUE;
     } params;
 
@@ -278,26 +333,30 @@ struct lttng_live_component
 
 struct lttng_live_msg_iter
 {
-    explicit lttng_live_msg_iter(const bt2c::Logger& parentLogger) :
-        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/MSG-ITER"}
+    using UP = std::unique_ptr<lttng_live_msg_iter>;
+
+    explicit lttng_live_msg_iter(const bt2c::Logger& parentLogger,
+                                 const bt2::SelfComponent selfCompParam,
+                                 const bt2::SelfMessageIterator selfMsgIter) :
+        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/MSG-ITER"},
+        selfComp {selfCompParam}, selfMsgIter {selfMsgIter}
     {
     }
 
+    ~lttng_live_msg_iter();
+
     bt2c::Logger logger;
 
-    bt_self_component *self_comp = nullptr;
+    bt2::SelfComponent selfComp;
 
     /* Weak reference. */
     struct lttng_live_component *lttng_live_comp = nullptr;
 
-    /* Weak reference. */
-    bt_self_message_iterator *self_msg_iter = nullptr;
+    bt2::SelfMessageIterator selfMsgIter;
 
-    /* Owned by this. */
-    struct live_viewer_connection *viewer_connection = nullptr;
+    live_viewer_connection::UP viewer_connection;
 
-    /* Array of pointers to struct lttng_live_session. */
-    GPtrArray *sessions = nullptr;
+    std::vector<lttng_live_session::UP> sessions;
 
     /* Number of live stream iterator this message iterator has.*/
     uint64_t active_stream_iter = 0;
@@ -373,7 +432,7 @@ bt_component_class_query_method_status lttng_live_query(bt_self_component_class_
 void lttng_live_component_finalize(bt_self_component_source *component);
 
 bt_message_iterator_class_next_method_status
-lttng_live_msg_iter_next(bt_self_message_iterator *iterator, bt_message_array_const msgs,
+lttng_live_msg_iter_next(bt_self_message_iterator *self_msg_it, bt_message_array_const msgs,
                          uint64_t capacity, uint64_t *count);
 
 bt_message_iterator_class_initialize_method_status
@@ -383,14 +442,12 @@ lttng_live_msg_iter_init(bt_self_message_iterator *self_msg_it,
 
 void lttng_live_msg_iter_finalize(bt_self_message_iterator *it);
 
-enum lttng_live_viewer_status lttng_live_session_attach(struct lttng_live_session *session,
-                                                        bt_self_message_iterator *self_msg_iter);
+enum lttng_live_viewer_status lttng_live_session_attach(struct lttng_live_session *session);
 
 enum lttng_live_viewer_status lttng_live_session_detach(struct lttng_live_session *session);
 
 enum lttng_live_iterator_status
-lttng_live_session_get_new_streams(struct lttng_live_session *session,
-                                   bt_self_message_iterator *self_msg_iter);
+lttng_live_session_get_new_streams(struct lttng_live_session *session);
 
 struct lttng_live_trace *
 lttng_live_session_borrow_or_create_trace_by_id(struct lttng_live_session *session,
@@ -407,20 +464,18 @@ int lttng_live_add_session(struct lttng_live_msg_iter *lttng_live_msg_iter, uint
  * written to the file.
  */
 enum lttng_live_get_one_metadata_status
-lttng_live_get_one_metadata_packet(struct lttng_live_trace *trace, std::vector<char>& buf);
+lttng_live_get_one_metadata_packet(struct lttng_live_trace *trace, std::vector<uint8_t>& buf);
 
 enum lttng_live_iterator_status
 lttng_live_get_next_index(struct lttng_live_msg_iter *lttng_live_msg_iter,
                           struct lttng_live_stream_iterator *stream, struct packet_index *index);
 
-enum ctf_msg_iter_medium_status
-lttng_live_get_stream_bytes(struct lttng_live_msg_iter *lttng_live_msg_iter,
-                            struct lttng_live_stream_iterator *stream, uint8_t *buf,
-                            uint64_t offset, uint64_t req_len, uint64_t *recv_len);
-
 bool lttng_live_graph_is_canceled(struct lttng_live_msg_iter *msg_iter);
 
 void lttng_live_stream_iterator_set_state(struct lttng_live_stream_iterator *stream_iter,
                                           enum lttng_live_stream_state new_state);
 
-#endif /* BABELTRACE_PLUGIN_CTF_LTTNG_LIVE_H */
+void lttng_live_stream_iterator_set_stream_class(lttng_live_stream_iterator *streamIter,
+                                                 uint64_t ctfStreamClsId);
+
+#endif /* BABELTRACE_PLUGINS_CTF_LTTNG_LIVE_LTTNG_LIVE_HPP */
This page took 0.040535 seconds and 4 git commands to generate.