Fix: source.ctf.lttng-live: muxing failure on clear
[babeltrace.git] / src / plugins / ctf / lttng-live / lttng-live.c
index d812e818f24c2177672400946ab5f7e852f9328c..04464c10785f9fa61875030056ee64e9e3310f75 100644 (file)
@@ -740,7 +740,7 @@ int live_get_msg_ts_ns(struct lttng_live_stream_iterator *stream_iter,
                        msg);
                break;
        case BT_MESSAGE_TYPE_MESSAGE_ITERATOR_INACTIVITY:
-               clock_snapshot = bt_message_message_iterator_inactivity_borrow_default_clock_snapshot_const(
+               clock_snapshot = bt_message_message_iterator_inactivity_borrow_clock_snapshot_const(
                        msg);
                break;
        default:
@@ -873,6 +873,10 @@ enum lttng_live_iterator_status lttng_live_iterator_close_stream(
                        "Error getting the next message from CTF message iterator");
                live_status = LTTNG_LIVE_ITERATOR_STATUS_ERROR;
                goto end;
+       } else if (status == CTF_MSG_ITER_STATUS_EOF) {
+               BT_COMP_LOGI("Reached the end of the live stream iterator.");
+               live_status = LTTNG_LIVE_ITERATOR_STATUS_END;
+               goto end;
        }
 
        BT_ASSERT(status == CTF_MSG_ITER_STATUS_OK);
@@ -997,6 +1001,77 @@ end:
        return live_status;
 }
 
+static
+bool is_discarded_packet_or_event_message(const bt_message *msg)
+{
+       const enum bt_message_type msg_type = bt_message_get_type(msg);
+
+       return msg_type == BT_MESSAGE_TYPE_DISCARDED_EVENTS ||
+                       msg_type == BT_MESSAGE_TYPE_DISCARDED_PACKETS;
+}
+
+static
+enum lttng_live_iterator_status adjust_discarded_packets_message(
+               bt_self_message_iterator *iter,
+               const bt_stream *stream,
+               const bt_message *msg_in, bt_message **msg_out,
+               uint64_t new_begin_ts)
+{
+       enum lttng_live_iterator_status status = LTTNG_LIVE_ITERATOR_STATUS_OK;
+       enum bt_property_availability availability;
+       const bt_clock_snapshot *clock_snapshot;
+       uint64_t end_ts;
+       uint64_t count;
+
+       clock_snapshot = bt_message_discarded_packets_borrow_end_default_clock_snapshot_const(msg_in);
+       end_ts = bt_clock_snapshot_get_value(clock_snapshot);
+
+       availability = bt_message_discarded_packets_get_count(msg_in, &count);
+       BT_ASSERT_DBG(availability == BT_PROPERTY_AVAILABILITY_AVAILABLE);
+
+       *msg_out = bt_message_discarded_packets_create_with_default_clock_snapshots(
+               iter, stream, new_begin_ts, end_ts);
+       if (!*msg_out) {
+               status = LTTNG_LIVE_ITERATOR_STATUS_NOMEM;
+               goto end;
+       }
+
+       bt_message_discarded_packets_set_count(*msg_out, count);
+end:
+       return status;
+}
+
+static
+enum lttng_live_iterator_status adjust_discarded_events_message(
+               bt_self_message_iterator *iter,
+               const bt_stream *stream,
+               const bt_message *msg_in, bt_message **msg_out,
+               uint64_t new_begin_ts)
+{
+       enum lttng_live_iterator_status status = LTTNG_LIVE_ITERATOR_STATUS_OK;
+       enum bt_property_availability availability;
+       const bt_clock_snapshot *clock_snapshot;
+       uint64_t end_ts;
+       uint64_t count;
+
+       clock_snapshot = bt_message_discarded_events_borrow_end_default_clock_snapshot_const(msg_in);
+       end_ts = bt_clock_snapshot_get_value(clock_snapshot);
+
+       availability = bt_message_discarded_events_get_count(msg_in, &count);
+       BT_ASSERT_DBG(availability == BT_PROPERTY_AVAILABILITY_AVAILABLE);
+
+       *msg_out = bt_message_discarded_events_create_with_default_clock_snapshots(
+               iter, stream, new_begin_ts, end_ts);
+       if (!*msg_out) {
+               status = LTTNG_LIVE_ITERATOR_STATUS_NOMEM;
+               goto end;
+       }
+
+       bt_message_discarded_events_set_count(*msg_out, count);
+end:
+       return status;
+}
+
 static
 enum lttng_live_iterator_status next_stream_iterator_for_trace(
                struct lttng_live_msg_iter *lttng_live_msg_iter,
@@ -1061,13 +1136,66 @@ enum lttng_live_iterator_status next_stream_iterator_for_trace(
 
                        /*
                         * Check if the message of the current live stream
-                        * iterator occured at the exact same time or after the
+                        * iterator occurred at the exact same time or after the
                         * last message returned by this component's message
                         * iterator. If not, we return an error.
                         */
                        if (curr_msg_ts_ns >= lttng_live_msg_iter->last_msg_ts_ns) {
                                stream_iter->current_msg = msg;
                                stream_iter->current_msg_ts_ns = curr_msg_ts_ns;
+                       } else if (stream_iter->last_inactivity_ts > curr_msg_ts_ns &&
+                                       is_discarded_packet_or_event_message(msg)) {
+                               /*
+                                * The CTF message iterator emits Discarded
+                                * Packets and Events with synthesized begin and
+                                * end timestamps from the bounds of the last
+                                * known packet and the newly decoded packet
+                                * header.
+                                *
+                                * The CTF message iterator is not aware of
+                                * stream inactivity beacons. Hence, we have
+                                * to adjust the begin timestamp of those types
+                                * of messages if a stream signaled its
+                                * inactivity up until _after_ the last known
+                                * packet's begin timestamp.
+                                *
+                                * Otherwise, the monotonicity guarantee would
+                                * not be preserved.
+                                */
+                               const enum bt_message_type msg_type =
+                                       bt_message_get_type(msg);
+                               enum lttng_live_iterator_status adjust_status =
+                                       LTTNG_LIVE_ITERATOR_STATUS_OK;
+                               bt_message *adjusted_message;
+
+                               switch (msg_type) {
+                               case BT_MESSAGE_TYPE_DISCARDED_EVENTS:
+                                       adjust_status = adjust_discarded_events_message(
+                                               lttng_live_msg_iter->self_msg_iter,
+                                               stream_iter->stream,
+                                               msg, &adjusted_message,
+                                               stream_iter->last_inactivity_ts);
+                                       break;
+                               case BT_MESSAGE_TYPE_DISCARDED_PACKETS:
+                                       adjust_status = adjust_discarded_packets_message(
+                                               lttng_live_msg_iter->self_msg_iter,
+                                               stream_iter->stream,
+                                               msg, &adjusted_message,
+                                               stream_iter->last_inactivity_ts);
+                                       break;
+                               default:
+                                       bt_common_abort();
+                               }
+
+                               if (adjust_status != LTTNG_LIVE_ITERATOR_STATUS_OK) {
+                                       stream_iter_status = adjust_status;
+                                       goto end;
+                               }
+
+                               BT_ASSERT_DBG(adjusted_message);
+                               stream_iter->current_msg = adjusted_message;
+                               stream_iter->current_msg_ts_ns =
+                                       stream_iter->last_inactivity_ts;
                        } else {
                                /*
                                 * We received a message in the past. To ensure
This page took 0.025429 seconds and 4 git commands to generate.