tap-driver.sh: flush stdout after each test result
[babeltrace.git] / plugins / ctf / common / msg-iter / msg-iter.c
index 907b299a7247fb3c62b4dc150c079f58f5c4030d..0671d9c3344bfb4c275f4552cbc51d0a69414a7d 100644 (file)
 #include <stdio.h>
 #include <stddef.h>
 #include <stdbool.h>
-#include <babeltrace/assert-internal.h>
+#include <babeltrace2/assert-internal.h>
 #include <string.h>
-#include <babeltrace/babeltrace.h>
-#include <babeltrace/common-internal.h>
+#include <babeltrace2/babeltrace.h>
+#include <babeltrace2/common-internal.h>
 #include <glib.h>
 #include <stdlib.h>
 
@@ -80,8 +80,13 @@ enum state {
        STATE_DSCOPE_STREAM_PACKET_CONTEXT_BEGIN,
        STATE_DSCOPE_STREAM_PACKET_CONTEXT_CONTINUE,
        STATE_AFTER_STREAM_PACKET_CONTEXT,
+       STATE_CHECK_EMIT_MSG_STREAM_BEGINNING,
        STATE_EMIT_MSG_STREAM_BEGINNING,
        STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING,
+       STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS,
+       STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS,
+       STATE_EMIT_MSG_DISCARDED_EVENTS,
+       STATE_EMIT_MSG_DISCARDED_PACKETS,
        STATE_EMIT_MSG_PACKET_BEGINNING,
        STATE_DSCOPE_EVENT_HEADER_BEGIN,
        STATE_DSCOPE_EVENT_HEADER_CONTINUE,
@@ -96,11 +101,19 @@ enum state {
        STATE_SKIP_PACKET_PADDING,
        STATE_EMIT_MSG_PACKET_END_MULTI,
        STATE_EMIT_MSG_PACKET_END_SINGLE,
+       STATE_CHECK_EMIT_MSG_STREAM_ACTIVITY_END,
        STATE_EMIT_MSG_STREAM_ACTIVITY_END,
        STATE_EMIT_MSG_STREAM_END,
        STATE_DONE,
 };
 
+struct end_of_packet_snapshots {
+       uint64_t discarded_events;
+       uint64_t packets;
+       uint64_t beginning_clock;
+       uint64_t end_clock;
+};
+
 /* CTF message iterator */
 struct bt_msg_iter {
        /* Visit stack */
@@ -115,11 +128,12 @@ struct bt_msg_iter {
         */
        bool emit_stream_begin_msg;
 
-       /*
-        * True to emit stream end and stream activity end messages.
-        */
+       /* True to emit stream end and stream activity end messages */
        bool emit_stream_end_msg;
 
+       /* True to set the stream */
+       bool set_stream;
+
        /*
         * Current dynamic scope field pointer.
         *
@@ -222,13 +236,11 @@ struct bt_msg_iter {
        /* Default clock's current value */
        uint64_t default_clock_snapshot;
 
-       /* End of packet snapshots */
-       struct {
-               uint64_t discarded_events;
-               uint64_t packets;
-               uint64_t beginning_clock;
-               uint64_t end_clock;
-       } snapshots;
+       /* End of current packet snapshots */
+       struct end_of_packet_snapshots snapshots;
+
+       /* End of previous packet snapshots */
+       struct end_of_packet_snapshots prev_packet_snapshots;
 
        /* Stored values (for sequence lengths, variant tags) */
        GArray *stored_values;
@@ -258,6 +270,10 @@ const char *state_string(enum state state)
                return "STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING";
        case STATE_EMIT_MSG_PACKET_BEGINNING:
                return "STATE_EMIT_MSG_PACKET_BEGINNING";
+       case STATE_EMIT_MSG_DISCARDED_EVENTS:
+               return "STATE_EMIT_MSG_DISCARDED_EVENTS";
+       case STATE_EMIT_MSG_DISCARDED_PACKETS:
+               return "STATE_EMIT_MSG_DISCARDED_PACKETS";
        case STATE_DSCOPE_EVENT_HEADER_BEGIN:
                return "STATE_DSCOPE_EVENT_HEADER_BEGIN";
        case STATE_DSCOPE_EVENT_HEADER_CONTINUE:
@@ -673,7 +689,7 @@ enum bt_msg_iter_status read_packet_header_begin_state(
                break;
        case BT_MSG_ITER_STATUS_EOF:
                ret = BT_MSG_ITER_STATUS_OK;
-               notit->state = STATE_EMIT_MSG_STREAM_ACTIVITY_END;
+               notit->state = STATE_CHECK_EMIT_MSG_STREAM_ACTIVITY_END;
                goto end;
        default:
                goto end;
@@ -953,14 +969,8 @@ enum bt_msg_iter_status set_current_packet_content_sizes(
 
        if (notit->cur_exp_packet_total_size == -1) {
                if (notit->cur_exp_packet_content_size != -1) {
-                       BT_LOGW("Content size is set, but packet size is not: "
-                               "notit-addr=%p, packet-context-field-addr=%p, "
-                               "packet-size=%" PRId64 ", content-size=%" PRId64,
-                               notit, notit->dscopes.stream_packet_context,
-                               notit->cur_exp_packet_total_size,
-                               notit->cur_exp_packet_content_size);
-                       status = BT_MSG_ITER_STATUS_ERROR;
-                       goto end;
+                       notit->cur_exp_packet_total_size =
+                               notit->cur_exp_packet_content_size;
                }
        } else {
                if (notit->cur_exp_packet_content_size == -1) {
@@ -969,6 +979,11 @@ enum bt_msg_iter_status set_current_packet_content_sizes(
                }
        }
 
+       BT_ASSERT((notit->cur_exp_packet_total_size >= 0 &&
+               notit->cur_exp_packet_content_size >= 0) ||
+               (notit->cur_exp_packet_total_size < 0 &&
+               notit->cur_exp_packet_content_size < 0));
+
        if (notit->cur_exp_packet_content_size >
                        notit->cur_exp_packet_total_size) {
                BT_LOGW("Invalid packet or content size: "
@@ -1007,9 +1022,9 @@ enum bt_msg_iter_status after_packet_context_state(struct bt_msg_iter *notit)
                 * least one packet beginning message, so the initial
                 * stream beginning message was also emitted.
                 */
-               notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
+               notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS;
        } else {
-               notit->state = STATE_EMIT_MSG_STREAM_BEGINNING;
+               notit->state = STATE_CHECK_EMIT_MSG_STREAM_BEGINNING;
        }
 
 end:
@@ -1167,8 +1182,17 @@ enum bt_msg_iter_status set_current_event_message(
                notit->meta.ec->name->str,
                notit->packet);
        BT_ASSERT(notit->msg_iter);
-       msg = bt_message_event_create(notit->msg_iter,
-               notit->meta.ec->ir_ec, notit->packet);
+       BT_ASSERT(notit->meta.sc);
+
+       if (bt_stream_class_borrow_default_clock_class(notit->meta.sc->ir_sc)) {
+               msg = bt_message_event_create_with_default_clock_snapshot(
+                       notit->msg_iter, notit->meta.ec->ir_ec,
+                       notit->packet, notit->default_clock_snapshot);
+       } else {
+               msg = bt_message_event_create(notit->msg_iter,
+                       notit->meta.ec->ir_ec, notit->packet);
+       }
+
        if (!msg) {
                BT_LOGE("Cannot create event message: "
                        "notit-addr=%p, ec-addr=%p, ec-name=\"%s\", "
@@ -1416,6 +1440,127 @@ end:
        return status;
 }
 
+static
+enum bt_msg_iter_status check_emit_msg_stream_beginning_state(
+               struct bt_msg_iter *notit)
+{
+       enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
+
+       if (notit->set_stream) {
+               status = set_current_stream(notit);
+               if (status != BT_MSG_ITER_STATUS_OK) {
+                       goto end;
+               }
+       }
+
+       if (notit->emit_stream_begin_msg) {
+               notit->state = STATE_EMIT_MSG_STREAM_BEGINNING;
+       } else {
+               /* Stream's first packet */
+               notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS;
+       }
+
+end:
+       return status;
+}
+
+static
+enum bt_msg_iter_status check_emit_msg_discarded_events(
+               struct bt_msg_iter *notit)
+{
+       notit->state = STATE_EMIT_MSG_DISCARDED_EVENTS;
+
+       if (!notit->meta.sc->has_discarded_events) {
+               notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS;
+               goto end;
+       }
+
+       if (notit->prev_packet_snapshots.discarded_events == UINT64_C(-1)) {
+               if (notit->snapshots.discarded_events == 0 ||
+                               notit->snapshots.discarded_events == UINT64_C(-1)) {
+                       /*
+                        * Stream's first packet with no discarded
+                        * events or no information about discarded
+                        * events: do not emit.
+                        */
+                       notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS;
+               }
+       } else {
+               /*
+                * If the previous packet has a value for this counter,
+                * then this counter is defined for the whole stream.
+                */
+               BT_ASSERT(notit->snapshots.discarded_events != UINT64_C(-1));
+
+               if (notit->snapshots.discarded_events -
+                               notit->prev_packet_snapshots.discarded_events == 0) {
+                       /*
+                        * No discarded events since previous packet: do
+                        * not emit.
+                        */
+                       notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS;
+               }
+       }
+
+end:
+       return BT_MSG_ITER_STATUS_OK;
+}
+
+static
+enum bt_msg_iter_status check_emit_msg_discarded_packets(
+               struct bt_msg_iter *notit)
+{
+       notit->state = STATE_EMIT_MSG_DISCARDED_PACKETS;
+
+       if (!notit->meta.sc->has_discarded_packets) {
+               notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
+               goto end;
+       }
+
+       if (notit->prev_packet_snapshots.packets == UINT64_C(-1)) {
+               /*
+                * Stream's first packet or no information about
+                * discarded packets: do not emit. In other words, if
+                * this is the first packet and its sequence number is
+                * not 0, do not consider that packets were previously
+                * lost: we might be reading a partial stream (LTTng
+                * snapshot for example).
+                */
+               notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
+       } else {
+               /*
+                * If the previous packet has a value for this counter,
+                * then this counter is defined for the whole stream.
+                */
+               BT_ASSERT(notit->snapshots.packets != UINT64_C(-1));
+
+               if (notit->snapshots.packets -
+                               notit->prev_packet_snapshots.packets <= 1) {
+                       /*
+                        * No discarded packets since previous packet:
+                        * do not emit.
+                        */
+                       notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
+               }
+       }
+
+end:
+       return BT_MSG_ITER_STATUS_OK;
+}
+
+static
+enum bt_msg_iter_status check_emit_msg_stream_activity_end(
+               struct bt_msg_iter *notit)
+{
+       if (notit->emit_stream_end_msg) {
+               notit->state = STATE_EMIT_MSG_STREAM_ACTIVITY_END;
+       } else {
+               notit->state = STATE_DONE;
+       }
+
+       return BT_MSG_ITER_STATUS_OK;
+}
+
 static inline
 enum bt_msg_iter_status handle_state(struct bt_msg_iter *notit)
 {
@@ -1448,10 +1593,25 @@ enum bt_msg_iter_status handle_state(struct bt_msg_iter *notit)
        case STATE_AFTER_STREAM_PACKET_CONTEXT:
                status = after_packet_context_state(notit);
                break;
+       case STATE_CHECK_EMIT_MSG_STREAM_BEGINNING:
+               status = check_emit_msg_stream_beginning_state(notit);
+               break;
        case STATE_EMIT_MSG_STREAM_BEGINNING:
                notit->state = STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING;
                break;
        case STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING:
+               notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS;
+               break;
+       case STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS:
+               status = check_emit_msg_discarded_events(notit);
+               break;
+       case STATE_EMIT_MSG_DISCARDED_EVENTS:
+               notit->state = STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS;
+               break;
+       case STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS:
+               status = check_emit_msg_discarded_packets(notit);
+               break;
+       case STATE_EMIT_MSG_DISCARDED_PACKETS:
                notit->state = STATE_EMIT_MSG_PACKET_BEGINNING;
                break;
        case STATE_EMIT_MSG_PACKET_BEGINNING:
@@ -1494,7 +1654,10 @@ enum bt_msg_iter_status handle_state(struct bt_msg_iter *notit)
                notit->state = STATE_SKIP_PACKET_PADDING;
                break;
        case STATE_EMIT_MSG_PACKET_END_SINGLE:
-               notit->state = STATE_EMIT_MSG_STREAM_ACTIVITY_END;
+               notit->state = STATE_CHECK_EMIT_MSG_STREAM_ACTIVITY_END;
+               break;
+       case STATE_CHECK_EMIT_MSG_STREAM_ACTIVITY_END:
+               status = check_emit_msg_stream_activity_end(notit);
                break;
        case STATE_EMIT_MSG_STREAM_ACTIVITY_END:
                notit->state = STATE_EMIT_MSG_STREAM_END;
@@ -1517,11 +1680,8 @@ enum bt_msg_iter_status handle_state(struct bt_msg_iter *notit)
        return status;
 }
 
-/**
- * Resets the internal state of a CTF message iterator.
- */
 BT_HIDDEN
-void bt_msg_iter_reset(struct bt_msg_iter *notit)
+void bt_msg_iter_reset_for_next_stream_file(struct bt_msg_iter *notit)
 {
        BT_ASSERT(notit);
        BT_LOGD("Resetting message iterator: addr=%p", notit);
@@ -1548,11 +1708,28 @@ void bt_msg_iter_reset(struct bt_msg_iter *notit)
        notit->cur_exp_packet_content_size = -1;
        notit->cur_exp_packet_total_size = -1;
        notit->cur_packet_offset = -1;
-       notit->cur_stream_class_id = -1;
        notit->cur_event_class_id = -1;
+       notit->snapshots.beginning_clock = UINT64_C(-1);
+       notit->snapshots.end_clock = UINT64_C(-1);
+}
+
+/**
+ * Resets the internal state of a CTF message iterator.
+ */
+BT_HIDDEN
+void bt_msg_iter_reset(struct bt_msg_iter *notit)
+{
+       bt_msg_iter_reset_for_next_stream_file(notit);
+       notit->cur_stream_class_id = -1;
        notit->cur_data_stream_id = -1;
        notit->emit_stream_begin_msg = true;
        notit->emit_stream_end_msg = true;
+       notit->snapshots.discarded_events = UINT64_C(-1);
+       notit->snapshots.packets = UINT64_C(-1);
+       notit->prev_packet_snapshots.discarded_events = UINT64_C(-1);
+       notit->prev_packet_snapshots.packets = UINT64_C(-1);
+       notit->prev_packet_snapshots.beginning_clock = UINT64_C(-1);
+       notit->prev_packet_snapshots.end_clock = UINT64_C(-1);
 }
 
 static
@@ -1609,6 +1786,7 @@ int bt_msg_iter_switch_packet(struct bt_msg_iter *notit)
        notit->cur_stream_class_id = -1;
        notit->cur_event_class_id = -1;
        notit->cur_data_stream_id = -1;
+       notit->prev_packet_snapshots = notit->snapshots;
        notit->snapshots.discarded_events = UINT64_C(-1);
        notit->snapshots.packets = UINT64_C(-1);
        notit->snapshots.beginning_clock = UINT64_C(-1);
@@ -2122,11 +2300,23 @@ int64_t bfcr_get_sequence_length_cb(struct ctf_field_class *fc, void *data)
                seq_fc->stored_length_index);
        seq_field = stack_top(notit->stack)->base;
        BT_ASSERT(seq_field);
-       ret = bt_field_dynamic_array_set_length(seq_field, (uint64_t) length);
-       if (ret) {
-               BT_LOGE("Cannot set dynamic array field's length field: "
-                       "notit-addr=%p, field-addr=%p, "
-                       "length=%" PRIu64, notit, seq_field, length);
+
+       /*
+        * bfcr_get_sequence_length_cb() also gets called back for a
+        * text sequence, but the destination field is a string field.
+        * Only set the field's sequence length if the destination field
+        * is a sequence field.
+        */
+       if (!seq_fc->base.is_text) {
+               BT_ASSERT(bt_field_get_class_type(seq_field) ==
+                       BT_FIELD_CLASS_TYPE_DYNAMIC_ARRAY);
+               ret = bt_field_dynamic_array_set_length(seq_field,
+                       (uint64_t) length);
+               if (ret) {
+                       BT_LOGE("Cannot set dynamic array field's length field: "
+                               "notit-addr=%p, field-addr=%p, "
+                               "length=%" PRIu64, notit, seq_field, length);
+               }
        }
 
        return length;
@@ -2211,20 +2401,6 @@ end:
        return ret_fc;
 }
 
-static
-void set_event_default_clock_snapshot(struct bt_msg_iter *notit)
-{
-       bt_event *event = bt_message_event_borrow_event(notit->event_msg);
-       bt_stream_class *sc = notit->meta.sc->ir_sc;
-
-       BT_ASSERT(event);
-
-       if (bt_stream_class_borrow_default_clock_class(sc)) {
-               bt_event_set_default_clock_snapshot(event,
-                       notit->default_clock_snapshot);
-       }
-}
-
 static
 void create_msg_stream_beginning(struct bt_msg_iter *notit,
                bt_message **message)
@@ -2333,30 +2509,6 @@ void create_msg_packet_beginning(struct bt_msg_iter *notit,
        sc = notit->meta.sc->ir_sc;
        BT_ASSERT(sc);
 
-       if (bt_stream_class_packets_have_discarded_event_counter_snapshot(sc)) {
-               BT_ASSERT(notit->snapshots.discarded_events != UINT64_C(-1));
-               bt_packet_set_discarded_event_counter_snapshot(
-                       notit->packet, notit->snapshots.discarded_events);
-       }
-
-       if (bt_stream_class_packets_have_packet_counter_snapshot(sc)) {
-               BT_ASSERT(notit->snapshots.packets != UINT64_C(-1));
-               bt_packet_set_packet_counter_snapshot(
-                       notit->packet, notit->snapshots.packets);
-       }
-
-       if (bt_stream_class_packets_have_default_beginning_clock_snapshot(sc)) {
-               BT_ASSERT(notit->snapshots.beginning_clock != UINT64_C(-1));
-               bt_packet_set_default_beginning_clock_snapshot(
-                       notit->packet, notit->snapshots.beginning_clock);
-       }
-
-       if (bt_stream_class_packets_have_default_end_clock_snapshot(sc)) {
-               BT_ASSERT(notit->snapshots.end_clock != UINT64_C(-1));
-               bt_packet_set_default_end_clock_snapshot(
-                       notit->packet, notit->snapshots.end_clock);
-       }
-
        if (notit->packet_context_field) {
                ret = bt_packet_move_context_field(
                        notit->packet, notit->packet_context_field);
@@ -2377,8 +2529,17 @@ void create_msg_packet_beginning(struct bt_msg_iter *notit,
        }
 
        BT_ASSERT(notit->msg_iter);
-       msg = bt_message_packet_beginning_create(notit->msg_iter,
-               notit->packet);
+
+       if (notit->meta.sc->packets_have_ts_begin) {
+               BT_ASSERT(notit->snapshots.beginning_clock != UINT64_C(-1));
+               msg = bt_message_packet_beginning_create_with_default_clock_snapshot(
+                       notit->msg_iter, notit->packet,
+                       notit->snapshots.beginning_clock);
+       } else {
+               msg = bt_message_packet_beginning_create(notit->msg_iter,
+                       notit->packet);
+       }
+
        if (!msg) {
                BT_LOGE("Cannot create packet beginning message: "
                        "notit-addr=%p, packet-addr=%p",
@@ -2407,8 +2568,17 @@ void create_msg_packet_end(struct bt_msg_iter *notit, bt_message **message)
        }
 
        BT_ASSERT(notit->msg_iter);
-       msg = bt_message_packet_end_create(notit->msg_iter,
-               notit->packet);
+
+       if (notit->meta.sc->packets_have_ts_end) {
+               BT_ASSERT(notit->snapshots.end_clock != UINT64_C(-1));
+               msg = bt_message_packet_end_create_with_default_clock_snapshot(
+                       notit->msg_iter, notit->packet,
+                       notit->snapshots.end_clock);
+       } else {
+               msg = bt_message_packet_end_create(notit->msg_iter,
+                       notit->packet);
+       }
+
        if (!msg) {
                BT_LOGE("Cannot create packet end message: "
                        "notit-addr=%p, packet-addr=%p",
@@ -2421,6 +2591,95 @@ void create_msg_packet_end(struct bt_msg_iter *notit, bt_message **message)
        *message = msg;
 }
 
+static
+void create_msg_discarded_events(struct bt_msg_iter *notit,
+               bt_message **message)
+{
+       bt_message *msg;
+       uint64_t beginning_raw_value = UINT64_C(-1);
+       uint64_t end_raw_value = UINT64_C(-1);
+
+       BT_ASSERT(notit->msg_iter);
+       BT_ASSERT(notit->stream);
+       BT_ASSERT(notit->meta.sc->has_discarded_events);
+
+       if (notit->meta.sc->discarded_events_have_default_cs) {
+               if (notit->prev_packet_snapshots.discarded_events == UINT64_C(-1)) {
+                       /*
+                        * We discarded events, but before (and possibly
+                        * including) the current packet: use this packet's time
+                        * range, and do not have a specific count.
+                        */
+                       beginning_raw_value = notit->snapshots.beginning_clock;
+                       end_raw_value = notit->snapshots.end_clock;
+               } else {
+                       beginning_raw_value = notit->prev_packet_snapshots.end_clock;
+                       end_raw_value = notit->snapshots.end_clock;
+               }
+
+               BT_ASSERT(beginning_raw_value != UINT64_C(-1));
+               BT_ASSERT(end_raw_value != UINT64_C(-1));
+               msg = bt_message_discarded_events_create_with_default_clock_snapshots(
+                       notit->msg_iter, notit->stream, beginning_raw_value,
+                       end_raw_value);
+       } else {
+               msg = bt_message_discarded_events_create(notit->msg_iter,
+                       notit->stream);
+       }
+
+       if (!msg) {
+               BT_LOGE("Cannot create discarded events message: "
+                       "notit-addr=%p, stream-addr=%p",
+                       notit, notit->stream);
+               return;
+       }
+
+       if (notit->prev_packet_snapshots.discarded_events != UINT64_C(-1)) {
+               bt_message_discarded_events_set_count(msg,
+                       notit->snapshots.discarded_events -
+                       notit->prev_packet_snapshots.discarded_events);
+       }
+
+       *message = msg;
+}
+
+static
+void create_msg_discarded_packets(struct bt_msg_iter *notit,
+               bt_message **message)
+{
+       bt_message *msg;
+
+       BT_ASSERT(notit->msg_iter);
+       BT_ASSERT(notit->stream);
+       BT_ASSERT(notit->meta.sc->has_discarded_packets);
+       BT_ASSERT(notit->prev_packet_snapshots.packets !=
+               UINT64_C(-1));
+
+       if (notit->meta.sc->discarded_packets_have_default_cs) {
+               BT_ASSERT(notit->prev_packet_snapshots.end_clock != UINT64_C(-1));
+               BT_ASSERT(notit->snapshots.beginning_clock != UINT64_C(-1));
+               msg = bt_message_discarded_packets_create_with_default_clock_snapshots(
+                       notit->msg_iter, notit->stream,
+                       notit->prev_packet_snapshots.end_clock,
+                       notit->snapshots.beginning_clock);
+       } else {
+               msg = bt_message_discarded_packets_create(notit->msg_iter,
+                       notit->stream);
+       }
+
+       if (!msg) {
+               BT_LOGE("Cannot create discarded packets message: "
+                       "notit-addr=%p, stream-addr=%p",
+                       notit, notit->stream);
+               return;
+       }
+
+       bt_message_discarded_packets_set_count(msg,
+               notit->snapshots.packets -
+                       notit->prev_packet_snapshots.packets - 1);
+       *message = msg;
+}
+
 BT_HIDDEN
 struct bt_msg_iter *bt_msg_iter_create(struct ctf_trace_class *tc,
                size_t max_request_sz,
@@ -2524,6 +2783,7 @@ enum bt_msg_iter_status bt_msg_iter_get_next_message(
        BT_ASSERT(notit);
        BT_ASSERT(message);
        notit->msg_iter = msg_iter;
+       notit->set_stream = true;
        BT_LOGV("Getting next message: notit-addr=%p", notit);
 
        while (true) {
@@ -2540,9 +2800,26 @@ enum bt_msg_iter_status bt_msg_iter_get_next_message(
                switch (notit->state) {
                case STATE_EMIT_MSG_EVENT:
                        BT_ASSERT(notit->event_msg);
-                       set_event_default_clock_snapshot(notit);
                        *message = notit->event_msg;
                        notit->event_msg = NULL;
+                       goto end;
+               case STATE_EMIT_MSG_DISCARDED_EVENTS:
+                       /* create_msg_discared_events() logs errors */
+                       create_msg_discarded_events(notit, message);
+
+                       if (!*message) {
+                               status = BT_MSG_ITER_STATUS_ERROR;
+                       }
+
+                       goto end;
+               case STATE_EMIT_MSG_DISCARDED_PACKETS:
+                       /* create_msg_discared_packets() logs errors */
+                       create_msg_discarded_packets(notit, message);
+
+                       if (!*message) {
+                               status = BT_MSG_ITER_STATUS_ERROR;
+                       }
+
                        goto end;
                case STATE_EMIT_MSG_PACKET_BEGINNING:
                        /* create_msg_packet_beginning() logs errors */
@@ -2564,62 +2841,41 @@ enum bt_msg_iter_status bt_msg_iter_get_next_message(
 
                        goto end;
                case STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING:
-                       if (notit->emit_stream_begin_msg) {
-                               /* create_msg_stream_activity_beginning() logs errors */
-                               create_msg_stream_activity_beginning(notit, message);
+                       /* create_msg_stream_activity_beginning() logs errors */
+                       create_msg_stream_activity_beginning(notit, message);
 
-                               if (!*message) {
-                                       status = BT_MSG_ITER_STATUS_ERROR;
-                               }
-
-                               goto end;
+                       if (!*message) {
+                               status = BT_MSG_ITER_STATUS_ERROR;
                        }
 
-                       break;
+                       goto end;
                case STATE_EMIT_MSG_STREAM_ACTIVITY_END:
-                       if (notit->emit_stream_end_msg) {
-                               /* create_msg_stream_activity_end() logs errors */
-                               create_msg_stream_activity_end(notit, message);
+                       /* create_msg_stream_activity_end() logs errors */
+                       create_msg_stream_activity_end(notit, message);
 
-                               if (!*message) {
-                                       status = BT_MSG_ITER_STATUS_ERROR;
-                               }
-
-                               goto end;
+                       if (!*message) {
+                               status = BT_MSG_ITER_STATUS_ERROR;
                        }
 
-                       break;
+                       goto end;
                case STATE_EMIT_MSG_STREAM_BEGINNING:
-                       status = set_current_stream(notit);
-                       if (status != BT_MSG_ITER_STATUS_OK) {
-                               goto end;
-                       }
-
-                       if (notit->emit_stream_begin_msg) {
-                               /* create_msg_stream_beginning() logs errors */
-                               create_msg_stream_beginning(notit, message);
+                       /* create_msg_stream_beginning() logs errors */
+                       create_msg_stream_beginning(notit, message);
 
-                               if (!*message) {
-                                       status = BT_MSG_ITER_STATUS_ERROR;
-                               }
-
-                               goto end;
+                       if (!*message) {
+                               status = BT_MSG_ITER_STATUS_ERROR;
                        }
 
-                       break;
+                       goto end;
                case STATE_EMIT_MSG_STREAM_END:
-                       if (notit->emit_stream_end_msg) {
-                               /* create_msg_stream_end() logs errors */
-                               create_msg_stream_end(notit, message);
+                       /* create_msg_stream_end() logs errors */
+                       create_msg_stream_end(notit, message);
 
-                               if (!*message) {
-                                       status = BT_MSG_ITER_STATUS_ERROR;
-                               }
-
-                               goto end;
+                       if (!*message) {
+                               status = BT_MSG_ITER_STATUS_ERROR;
                        }
 
-                       break;
+                       goto end;
                case STATE_DONE:
                        status = BT_MSG_ITER_STATUS_EOF;
                        goto end;
@@ -2641,6 +2897,7 @@ enum bt_msg_iter_status read_packet_header_context_fields(
        enum bt_msg_iter_status status = BT_MSG_ITER_STATUS_OK;
 
        BT_ASSERT(notit);
+       notit->set_stream = false;
 
        if (notit->state == STATE_EMIT_MSG_PACKET_BEGINNING) {
                /* We're already there */
@@ -2672,8 +2929,13 @@ enum bt_msg_iter_status read_packet_header_context_fields(
                case STATE_DSCOPE_STREAM_PACKET_CONTEXT_BEGIN:
                case STATE_DSCOPE_STREAM_PACKET_CONTEXT_CONTINUE:
                case STATE_AFTER_STREAM_PACKET_CONTEXT:
+               case STATE_CHECK_EMIT_MSG_STREAM_BEGINNING:
                case STATE_EMIT_MSG_STREAM_BEGINNING:
                case STATE_EMIT_MSG_STREAM_ACTIVITY_BEGINNING:
+               case STATE_CHECK_EMIT_MSG_DISCARDED_EVENTS:
+               case STATE_EMIT_MSG_DISCARDED_EVENTS:
+               case STATE_CHECK_EMIT_MSG_DISCARDED_PACKETS:
+               case STATE_EMIT_MSG_DISCARDED_PACKETS:
                        /* Non-emitting state: continue */
                        break;
                default:
@@ -2691,7 +2953,6 @@ end:
        ret = set_current_packet_content_sizes(notit);
        if (ret) {
                status = BT_MSG_ITER_STATUS_ERROR;
-               goto end;
        }
 
        return status;
@@ -2759,7 +3020,6 @@ enum bt_msg_iter_status bt_msg_iter_get_packet_properties(
 
        props->exp_packet_total_size = notit->cur_exp_packet_total_size;
        props->exp_packet_content_size = notit->cur_exp_packet_content_size;
-       BT_ASSERT(props->stream_class_id >= 0);
        props->stream_class_id = (uint64_t) notit->cur_stream_class_id;
        props->data_stream_id = notit->cur_data_stream_id;
        props->snapshots.discarded_events = notit->snapshots.discarded_events;
This page took 0.031304 seconds and 4 git commands to generate.