- assert(stream);
- stream_end_notif = bt_notification_stream_end_create(stream);
- if (!stream_end_notif) {
- BT_LOGE_STR("Cannot create stream end notification.");
- goto error;
- }
-
- add_action_push_notif(iterator, stream_end_notif);
- BT_LOGV("Added \"push stream end notification\" action: "
- "stream-addr=%p, stream-name=\"%s\"",
- stream, bt_ctf_stream_get_name(stream));
- goto end;
-
-error:
- ret = -1;
-
-end:
- bt_put(stream_end_notif);
- return ret;
-}
-
-static
-int add_action_push_notif_packet_begin(
- struct bt_notification_iterator *iterator,
- struct bt_ctf_packet *packet)
-{
- int ret = 0;
- struct bt_notification *packet_begin_notif = NULL;
-
- if (!is_subscribed_to_notification_type(iterator,
- BT_NOTIFICATION_TYPE_PACKET_BEGIN)) {
- BT_LOGV("Not adding \"push packet beginning notification\" action: "
- "notification iterator is not subscribed: addr=%p",
- iterator);
- goto end;
- }
-
- assert(packet);
- packet_begin_notif = bt_notification_packet_begin_create(packet);
- if (!packet_begin_notif) {
- BT_LOGE_STR("Cannot create packet beginning notification.");
- goto error;
- }
-
- add_action_push_notif(iterator, packet_begin_notif);
- BT_LOGV("Added \"push packet beginning notification\" action: "
- "packet-addr=%p", packet);
- goto end;
-
-error:
- ret = -1;
-
-end:
- bt_put(packet_begin_notif);
- return ret;
-}
-
-static
-int add_action_push_notif_packet_end(
- struct bt_notification_iterator *iterator,
- struct bt_ctf_packet *packet)
-{
- int ret = 0;
- struct bt_notification *packet_end_notif = NULL;
-
- if (!is_subscribed_to_notification_type(iterator,
- BT_NOTIFICATION_TYPE_PACKET_END)) {
- BT_LOGV("Not adding \"push packet end notification\" action: "
- "notification iterator is not subscribed: addr=%p",
- iterator);
- goto end;
- }
-
- assert(packet);
- packet_end_notif = bt_notification_packet_end_create(packet);
- if (!packet_end_notif) {
- BT_LOGE_STR("Cannot create packet end notification.");
- goto error;
- }
-
- add_action_push_notif(iterator, packet_end_notif);
- BT_LOGV("Added \"push packet end notification\" action: "
- "packet-addr=%p", packet);
- goto end;
-
-error:
- ret = -1;
-
-end:
- bt_put(packet_end_notif);
- return ret;
-}
-
-static
-void add_action_set_stream_state_is_ended(
- struct bt_notification_iterator *iterator,
- struct stream_state *stream_state)
-{
- struct action action = {
- .type = ACTION_TYPE_SET_STREAM_STATE_IS_ENDED,
- .payload.set_stream_state_is_ended = {
- .stream_state = stream_state,
- },
- };
-
- assert(stream_state);
- add_action(iterator, &action);
- BT_LOGV("Added \"set stream state's ended\" action: "
- "stream-state-addr=%p", stream_state);
-}
-
-static
-void add_action_set_stream_state_cur_packet(
- struct bt_notification_iterator *iterator,
- struct stream_state *stream_state,
- struct bt_ctf_packet *packet)
-{
- struct action action = {
- .type = ACTION_TYPE_SET_STREAM_STATE_CUR_PACKET,
- .payload.set_stream_state_cur_packet = {
- .stream_state = stream_state,
- .packet = bt_get(packet),
- },
- };
-
- assert(stream_state);
- add_action(iterator, &action);
- BT_LOGV("Added \"set stream state's current packet\" action: "
- "stream-state-addr=%p, packet-addr=%p",
- stream_state, packet);
-}
-
-static
-void add_action_update_stream_state_discarded_elements(
- struct bt_notification_iterator *iterator,
- enum action_type type,
- struct stream_state *stream_state,
- struct bt_ctf_clock_value *cur_begin,
- uint64_t cur_count)
-{
- struct action action = {
- .type = type,
- .payload.update_stream_state_discarded_elements = {
- .stream_state = stream_state,
- .cur_begin = bt_get(cur_begin),
- .cur_count = cur_count,
- },
- };
-
- assert(stream_state);
- assert(type == ACTION_TYPE_UPDATE_STREAM_STATE_DISCARDED_PACKETS ||
- type == ACTION_TYPE_UPDATE_STREAM_STATE_DISCARDED_EVENTS);
- add_action(iterator, &action);
- if (type == ACTION_TYPE_UPDATE_STREAM_STATE_DISCARDED_PACKETS) {
- BT_LOGV("Added \"update stream state's discarded packets\" action: "
- "stream-state-addr=%p, cur-begin-addr=%p, cur-count=%" PRIu64,
- stream_state, cur_begin, cur_count);
- } else if (type == ACTION_TYPE_UPDATE_STREAM_STATE_DISCARDED_EVENTS) {
- BT_LOGV("Added \"update stream state's discarded events\" action: "
- "stream-state-addr=%p, cur-begin-addr=%p, cur-count=%" PRIu64,
- stream_state, cur_begin, cur_count);
- }
-}
-
-static
-int ensure_stream_state_exists(struct bt_notification_iterator *iterator,
- struct bt_notification *stream_begin_notif,
- struct bt_ctf_stream *notif_stream,
- struct stream_state **stream_state)
-{
- int ret = 0;
-
- if (!notif_stream) {
- /*
- * The notification does not reference any stream: no
- * need to get or create a stream state.
- */
- goto end;
- }
-
- *stream_state = g_hash_table_lookup(iterator->stream_states,
- notif_stream);
- if (!*stream_state) {
- /*
- * This iterator did not bump into this stream yet:
- * create a stream state and a "stream begin"
- * notification.
- */
- struct action action = {
- .type = ACTION_TYPE_ADD_STREAM_STATE,
- .payload.add_stream_state = {
- .stream = bt_get(notif_stream),
- .stream_state = NULL,
- },
- };
-
- *stream_state = create_stream_state(notif_stream);
- if (!stream_state) {
- BT_LOGE_STR("Cannot create stream state.");
- goto error;
- }
-
- action.payload.add_stream_state.stream_state =
- *stream_state;
- add_action(iterator, &action);
-
- if (stream_begin_notif) {
- add_action_push_notif(iterator, stream_begin_notif);
- } else {
- ret = add_action_push_notif_stream_begin(iterator,
- notif_stream);
- if (ret) {
- BT_LOGE_STR("Cannot add \"push stream beginning notification\" action.");
- goto error;
- }
- }
- }
-
- goto end;
-
-error:
- destroy_stream_state(*stream_state);
- ret = -1;
-
-end:
- return ret;
-}
-
-static
-struct bt_ctf_field *get_struct_field_uint(struct bt_ctf_field *struct_field,
- const char *field_name)
-{
- struct bt_ctf_field *field = NULL;
- struct bt_ctf_field_type *ft = NULL;
-
- field = bt_ctf_field_structure_get_field_by_name(struct_field,
- field_name);
- if (!field) {
- BT_LOGV_STR("`%s` field does not exist.");
- goto end;
- }
-
- if (!bt_ctf_field_is_integer(field)) {
- BT_LOGV("Skipping `%s` field because its type is not an integer field type: "
- "field-addr=%p, ft-addr=%p, ft-id=%s", field_name,
- field, ft, bt_ctf_field_type_id_string(
- bt_ctf_field_type_get_type_id(ft)));
- BT_PUT(field);
- goto end;
- }
-
- ft = bt_ctf_field_get_type(field);
- assert(ft);
-
- if (bt_ctf_field_type_integer_is_signed(ft)) {
- BT_LOGV("Skipping `%s` integer field because its type is signed: "
- "field-addr=%p, ft-addr=%p", field_name, field, ft);
- BT_PUT(field);
- goto end;
- }
-
-end:
- bt_put(ft);
- return field;
-}
-
-static
-uint64_t get_packet_context_events_discarded(struct bt_ctf_packet *packet)
-{
- struct bt_ctf_field *packet_context = NULL;
- struct bt_ctf_field *field = NULL;
- uint64_t retval = -1ULL;
- int ret;
-
- packet_context = bt_ctf_packet_get_context(packet);
- if (!packet_context) {
- goto end;
- }
-
- field = get_struct_field_uint(packet_context, "events_discarded");
- if (!field) {
- BT_LOGV("`events_discarded` field does not exist in packet's context field: "
- "packet-addr=%p, packet-context-field-addr=%p",
- packet, packet_context);
- goto end;
- }
-
- assert(bt_ctf_field_is_integer(field));
- ret = bt_ctf_field_unsigned_integer_get_value(field, &retval);
- if (ret) {
- BT_LOGV("Cannot get raw value of packet's context field's `events_discarded` integer field: "
- "packet-addr=%p, field-addr=%p",
- packet, field);
- retval = -1ULL;
- goto end;
- }
-
-end:
- bt_put(packet_context);
- bt_put(field);
- return retval;
-}
-
-static
-uint64_t get_packet_context_packet_seq_num(struct bt_ctf_packet *packet)
-{
- struct bt_ctf_field *packet_context = NULL;
- struct bt_ctf_field *field = NULL;
- uint64_t retval = -1ULL;
- int ret;
-
- packet_context = bt_ctf_packet_get_context(packet);
- if (!packet_context) {
- goto end;
- }
-
- field = get_struct_field_uint(packet_context, "packet_seq_num");
- if (!field) {
- BT_LOGV("`packet_seq_num` field does not exist in packet's context field: "
- "packet-addr=%p, packet-context-field-addr=%p",
- packet, packet_context);
- goto end;
- }
-
- assert(bt_ctf_field_is_integer(field));
- ret = bt_ctf_field_unsigned_integer_get_value(field, &retval);
- if (ret) {
- BT_LOGV("Cannot get raw value of packet's context field's `packet_seq_num` integer field: "
- "packet-addr=%p, field-addr=%p",
- packet, field);
- retval = -1ULL;
- goto end;
- }
-
-end:
- bt_put(packet_context);
- bt_put(field);
- return retval;
-}
-
-static
-int handle_discarded_packets(struct bt_notification_iterator *iterator,
- struct bt_ctf_packet *packet,
- struct bt_ctf_clock_value *ts_begin,
- struct bt_ctf_clock_value *ts_end,
- struct stream_state *stream_state)
-{
- struct bt_notification *notif = NULL;
- uint64_t diff;
- uint64_t prev_count, next_count;
- int ret = 0;
-
- next_count = get_packet_context_packet_seq_num(packet);
- if (next_count == -1ULL) {
- /*
- * Stream does not have seqnum field, skip discarded
- * packets feature.
- */
- goto end;
- }
- prev_count = stream_state->discarded_packets_state.cur_count;
-
- if (prev_count != -1ULL) {
- if (next_count < prev_count) {
- BT_LOGW("Current value of packet's context field's `packet_seq_num` field is lesser than the previous value for the same stream: "
- "not updating the stream state's current value: "
- "packet-addr=%p, prev-count=%" PRIu64 ", "
- "cur-count=%" PRIu64,
- packet, prev_count, next_count);
- goto end;
- }
- if (next_count == prev_count) {
- BT_LOGW("Current value of packet's context field's `packet_seq_num` field is equal to the previous value for the same stream: "
- "not updating the stream state's current value: "
- "packet-addr=%p, prev-count=%" PRIu64 ", "
- "cur-count=%" PRIu64,
- packet, prev_count, next_count);
- goto end;
- }
-
- diff = next_count - prev_count;
- if (diff > 1) {
- /*
- * Add a discarded packets notification. The packets
- * are considered to be lost between the state's last time
- * and the current packet's beginning time.
- * The counter is expected to monotonically increase of
- * 1 for each packet. Therefore, the number of missing
- * packets is 'diff - 1'.
- */
- notif = bt_notification_discarded_elements_create(
- BT_NOTIFICATION_TYPE_DISCARDED_PACKETS,
- stream_state->stream,
- stream_state->discarded_packets_state.cur_begin,
- ts_begin, diff - 1);
- if (!notif) {
- BT_LOGE_STR("Cannot create discarded packets notification.");
- ret = -1;
- goto end;
- }
-
- add_action_push_notif(iterator, notif);
- }
- }
-
- add_action_update_stream_state_discarded_elements(iterator,
- ACTION_TYPE_UPDATE_STREAM_STATE_DISCARDED_PACKETS,
- stream_state, ts_end, next_count);
-
-end:
- bt_put(notif);
- return ret;
-}
-
-static
-int handle_discarded_events(struct bt_notification_iterator *iterator,
- struct bt_ctf_packet *packet,
- struct bt_ctf_clock_value *ts_begin,
- struct bt_ctf_clock_value *ts_end,
- struct stream_state *stream_state)
-{
- struct bt_notification *notif = NULL;
- uint64_t diff;
- uint64_t next_count;
- int ret = 0;
-
- next_count = get_packet_context_events_discarded(packet);
- if (next_count == -1ULL) {
- next_count = stream_state->discarded_events_state.cur_count;
- goto update_state;
- }
-
- if (next_count < stream_state->discarded_events_state.cur_count) {
- BT_LOGW("Current value of packet's context field's `events_discarded` field is lesser than the previous value for the same stream: "
- "not updating the stream state's current value: "
- "packet-addr=%p, prev-count=%" PRIu64 ", "
- "cur-count=%" PRIu64,
- packet, stream_state->discarded_events_state.cur_count,
- next_count);
- goto end;
- }
-
- diff = next_count - stream_state->discarded_events_state.cur_count;
- if (diff > 0) {
- /*
- * Add a discarded events notification. The events are
- * considered to be lost betweem the state's last time
- * and the current packet's end time.
- */
- notif = bt_notification_discarded_elements_create(
- BT_NOTIFICATION_TYPE_DISCARDED_EVENTS,
- stream_state->stream,
- stream_state->discarded_events_state.cur_begin,
- ts_end, diff);
- if (!notif) {
- BT_LOGE_STR("Cannot create discarded events notification.");
- ret = -1;
- goto end;
- }
-
- add_action_push_notif(iterator, notif);
- }
-
-update_state:
- add_action_update_stream_state_discarded_elements(iterator,
- ACTION_TYPE_UPDATE_STREAM_STATE_DISCARDED_EVENTS,
- stream_state, ts_end, next_count);
-
-end:
- bt_put(notif);
- return ret;
-}
-
-static
-int get_field_clock_value(struct bt_ctf_field *root_field,
- const char *field_name,
- struct bt_ctf_clock_value **user_clock_val)
-{
- struct bt_ctf_field *field;
- struct bt_ctf_field_type *ft = NULL;
- struct bt_ctf_clock_class *clock_class = NULL;
- struct bt_ctf_clock_value *clock_value = NULL;
- uint64_t val;
- int ret = 0;
-
- field = get_struct_field_uint(root_field, field_name);
- if (!field) {
- /* Not an error: skip this */
- goto end;
- }
-
- ft = bt_ctf_field_get_type(field);
- assert(ft);
- clock_class = bt_ctf_field_type_integer_get_mapped_clock_class(ft);
- if (!clock_class) {
- BT_LOGW("Integer field type has no mapped clock class but it's expected to have one: "
- "ft-addr=%p", ft);
- ret = -1;
- goto end;
- }
-
- ret = bt_ctf_field_unsigned_integer_get_value(field, &val);
- if (ret) {
- BT_LOGW("Cannot get integer field's raw value: "
- "field-addr=%p", field);
- ret = -1;
- goto end;
- }
-
- clock_value = bt_ctf_clock_value_create(clock_class, val);
- if (!clock_value) {
- BT_LOGE_STR("Cannot create clock value from clock class.");
- ret = -1;
- goto end;
- }
-
- /* Move clock value to user */
- *user_clock_val = clock_value;
- clock_value = NULL;
-
-end:
- bt_put(field);
- bt_put(ft);
- bt_put(clock_class);
- bt_put(clock_value);
- return ret;
-}
-
-static
-int get_ts_begin_ts_end_from_packet(struct bt_ctf_packet *packet,
- struct bt_ctf_clock_value **user_ts_begin,
- struct bt_ctf_clock_value **user_ts_end)
-{
- struct bt_ctf_field *packet_context = NULL;
- struct bt_ctf_clock_value *ts_begin = NULL;
- struct bt_ctf_clock_value *ts_end = NULL;
- int ret = 0;
-
- packet_context = bt_ctf_packet_get_context(packet);
- if (!packet_context) {
- goto end;
- }
-
- ret = get_field_clock_value(packet_context, "timestamp_begin",
- &ts_begin);
- if (ret) {
- BT_LOGW("Cannot create clock value for packet context's `timestamp_begin` field: "
- "packet-addr=%p, packet-context-field-addr=%p",
- packet, packet_context);
- goto end;
- }
-
- ret = get_field_clock_value(packet_context, "timestamp_end",
- &ts_end);
- if (ret) {
- BT_LOGW("Cannot create clock value for packet context's `timestamp_begin` field: "
- "packet-addr=%p, packet-context-field-addr=%p",
- packet, packet_context);
- goto end;
- }
-
- /* Move clock values to user */
- *user_ts_begin = ts_begin;
- ts_begin = NULL;
- *user_ts_end = ts_end;
- ts_end = NULL;
-
-end:
- bt_put(packet_context);
- bt_put(ts_begin);
- bt_put(ts_end);
- return ret;
-}
-
-static
-int handle_discarded_elements(struct bt_notification_iterator *iterator,
- struct bt_ctf_packet *packet, struct stream_state *stream_state)
-{
- struct bt_ctf_clock_value *ts_begin = NULL;
- struct bt_ctf_clock_value *ts_end = NULL;
- int ret;
-
- ret = get_ts_begin_ts_end_from_packet(packet, &ts_begin, &ts_end);
- if (ret) {
- BT_LOGW("Cannot get packet's beginning or end clock values: "
- "packet-addr=%p, ret=%d", packet, ret);
- ret = -1;
- goto end;
- }
-
- ret = handle_discarded_packets(iterator, packet, ts_begin, ts_end,
- stream_state);
- if (ret) {
- BT_LOGW("Cannot handle discarded packets for packet: "
- "packet-addr=%p, ret=%d", packet, ret);
- ret = -1;
- goto end;
- }
-
- ret = handle_discarded_events(iterator, packet, ts_begin, ts_end,
- stream_state);
- if (ret) {
- BT_LOGW("Cannot handle discarded events for packet: "
- "packet-addr=%p, ret=%d", packet, ret);
- ret = -1;
- goto end;
- }
-
-end:
- bt_put(ts_begin);
- bt_put(ts_end);
- return ret;