Replace all assert(false) and assert(0) with abort()
[babeltrace.git] / lib / graph / iterator.c
index 389eec528c022d6c1ff32b7e1b7ccde3a07e0a3f..1a295cb3c4f1865cf92dc59267c4ffa1eeb987b1 100644 (file)
  * SOFTWARE.
  */
 
+#define BT_LOG_TAG "NOTIF-ITER"
+#include <babeltrace/lib-logging-internal.h>
+
 #include <babeltrace/compiler-internal.h>
 #include <babeltrace/ref.h>
 #include <babeltrace/ctf-ir/event-internal.h>
 #include <babeltrace/ctf-ir/packet-internal.h>
 #include <babeltrace/ctf-ir/stream-internal.h>
+#include <babeltrace/graph/connection-internal.h>
 #include <babeltrace/graph/component.h>
 #include <babeltrace/graph/component-source-internal.h>
 #include <babeltrace/graph/component-class-internal.h>
+#include <babeltrace/graph/notification.h>
 #include <babeltrace/graph/notification-iterator.h>
 #include <babeltrace/graph/notification-iterator-internal.h>
 #include <babeltrace/graph/notification-internal.h>
 #include <babeltrace/graph/notification-stream.h>
 #include <babeltrace/graph/notification-stream-internal.h>
 #include <babeltrace/graph/port.h>
+#include <babeltrace/types.h>
+#include <stdint.h>
+#include <stdlib.h>
 
 struct stream_state {
        struct bt_ctf_stream *stream; /* owned by this */
        struct bt_ctf_packet *cur_packet; /* owned by this */
-       bool is_ended;
+       bt_bool is_ended;
 };
 
 enum action_type {
@@ -108,7 +116,10 @@ void destroy_stream_state(struct stream_state *stream_state)
                return;
        }
 
+       BT_LOGV("Destroying stream state: stream-state-addr=%p", stream_state);
+       BT_LOGV_STR("Putting stream state's current packet.");
        bt_put(stream_state->cur_packet);
+       BT_LOGV_STR("Putting stream state's stream.");
        bt_put(stream_state->stream);
        g_free(stream_state);
 }
@@ -139,7 +150,7 @@ void destroy_action(struct action *action)
        case ACTION_TYPE_SET_STREAM_STATE_IS_ENDED:
                break;
        default:
-               assert(false);
+               abort();
        }
 }
 
@@ -165,15 +176,40 @@ void clear_actions(struct bt_notification_iterator *iterator)
        g_array_set_size(iterator->actions, 0);
 }
 
+static inline
+const char *action_type_string(enum action_type type)
+{
+       switch (type) {
+       case ACTION_TYPE_PUSH_NOTIF:
+               return "ACTION_TYPE_PUSH_NOTIF";
+       case ACTION_TYPE_MAP_PORT_TO_COMP_IN_STREAM:
+               return "ACTION_TYPE_MAP_PORT_TO_COMP_IN_STREAM";
+       case ACTION_TYPE_ADD_STREAM_STATE:
+               return "ACTION_TYPE_ADD_STREAM_STATE";
+       case ACTION_TYPE_SET_STREAM_STATE_IS_ENDED:
+               return "ACTION_TYPE_SET_STREAM_STATE_IS_ENDED";
+       case ACTION_TYPE_SET_STREAM_STATE_CUR_PACKET:
+               return "ACTION_TYPE_SET_STREAM_STATE_CUR_PACKET";
+       default:
+               return "(unknown)";
+       }
+}
+
 static
 void apply_actions(struct bt_notification_iterator *iterator)
 {
        size_t i;
 
+       BT_LOGV("Applying notification's iterator current actions: "
+               "count=%u", iterator->actions->len);
+
        for (i = 0; i < iterator->actions->len; i++) {
                struct action *action = &g_array_index(iterator->actions,
                        struct action, i);
 
+               BT_LOGV("Applying action: index=%zu, type=%s",
+                       i, action_type_string(action->type));
+
                switch (action->type) {
                case ACTION_TYPE_PUSH_NOTIF:
                        /* Move notification to queue */
@@ -217,7 +253,7 @@ void apply_actions(struct bt_notification_iterator *iterator)
                        bt_ctf_stream_add_destroy_listener(
                                action->payload.set_stream_state_is_ended.stream_state->stream,
                                stream_destroy_listener, iterator);
-                       action->payload.set_stream_state_is_ended.stream_state->is_ended = true;
+                       action->payload.set_stream_state_is_ended.stream_state->is_ended = BT_TRUE;
                        BT_PUT(action->payload.set_stream_state_is_ended.stream_state->stream);
                        break;
                case ACTION_TYPE_SET_STREAM_STATE_CUR_PACKET:
@@ -226,7 +262,7 @@ void apply_actions(struct bt_notification_iterator *iterator)
                                action->payload.set_stream_state_cur_packet.packet);
                        break;
                default:
-                       assert(false);
+                       abort();
                }
        }
 
@@ -239,6 +275,7 @@ struct stream_state *create_stream_state(struct bt_ctf_stream *stream)
        struct stream_state *stream_state = g_new0(struct stream_state, 1);
 
        if (!stream_state) {
+               BT_LOGE_STR("Failed to allocate one stream state.");
                goto end;
        }
 
@@ -251,6 +288,9 @@ struct stream_state *create_stream_state(struct bt_ctf_stream *stream)
         * We put this reference when the stream is marked as ended.
         */
        stream_state->stream = bt_get(stream);
+       BT_LOGV("Created stream state: stream-addr=%p, stream-name=\"%s\", "
+               "stream-state-addr=%p",
+               stream, bt_ctf_stream_get_name(stream), stream_state);
 
 end:
        return stream_state;
@@ -260,48 +300,30 @@ static
 void bt_notification_iterator_destroy(struct bt_object *obj)
 {
        struct bt_notification_iterator *iterator;
-       struct bt_component_class *comp_class;
 
        assert(obj);
-       iterator = container_of(obj, struct bt_notification_iterator,
-                       base);
-       assert(iterator->upstream_component);
-       comp_class = iterator->upstream_component->class;
-
-       /* Call user-defined destroy method */
-       switch (comp_class->type) {
-       case BT_COMPONENT_CLASS_TYPE_SOURCE:
-       {
-               struct bt_component_class_source *source_class;
-
-               source_class = container_of(comp_class, struct bt_component_class_source, parent);
-
-               if (source_class->methods.iterator.finalize) {
-                       source_class->methods.iterator.finalize(
-                               bt_private_notification_iterator_from_notification_iterator(iterator));
-               }
-               break;
-       }
-       case BT_COMPONENT_CLASS_TYPE_FILTER:
-       {
-               struct bt_component_class_filter *filter_class;
-
-               filter_class = container_of(comp_class, struct bt_component_class_filter, parent);
 
-               if (filter_class->methods.iterator.finalize) {
-                       filter_class->methods.iterator.finalize(
-                               bt_private_notification_iterator_from_notification_iterator(iterator));
-               }
-               break;
-       }
-       default:
-               /* Unreachable */
-               assert(0);
-       }
+       /*
+        * The notification iterator's reference count is 0 if we're
+        * here. Increment it to avoid a double-destroy (possibly
+        * infinitely recursive). This could happen for example if the
+        * notification iterator's finalization function does bt_get()
+        * (or anything that causes bt_get() to be called) on itself
+        * (ref. count goes from 0 to 1), and then bt_put(): the
+        * reference count would go from 1 to 0 again and this function
+        * would be called again.
+        */
+       obj->ref_count.count++;
+       iterator = container_of(obj, struct bt_notification_iterator, base);
+       BT_LOGD("Destroying notification iterator object: addr=%p",
+               iterator);
+       bt_notification_iterator_finalize(iterator);
 
        if (iterator->queue) {
                struct bt_notification *notif;
 
+               BT_LOGD("Putting notifications in queue.");
+
                while ((notif = g_queue_pop_tail(iterator->queue))) {
                        bt_put(notif);
                }
@@ -323,6 +345,8 @@ void bt_notification_iterator_destroy(struct bt_object *obj)
 
                while (g_hash_table_iter_next(&ht_iter, &stream_gptr, &stream_state_gptr)) {
                        assert(stream_gptr);
+
+                       BT_LOGD_STR("Removing stream's destroy listener for notification iterator.");
                        bt_ctf_stream_remove_destroy_listener(
                                (void *) stream_gptr, stream_destroy_listener,
                                iterator);
@@ -335,58 +359,227 @@ void bt_notification_iterator_destroy(struct bt_object *obj)
                g_array_free(iterator->actions, TRUE);
        }
 
+       if (iterator->connection) {
+               /*
+                * Remove ourself from the originating connection so
+                * that it does not try to finalize a dangling pointer
+                * later.
+                */
+               bt_connection_remove_iterator(iterator->connection, iterator);
+       }
+
+       BT_LOGD_STR("Putting current notification.");
        bt_put(iterator->current_notification);
-       bt_put(iterator->upstream_component);
-       bt_put(iterator->upstream_port);
        g_free(iterator);
 }
 
+BT_HIDDEN
+void bt_notification_iterator_finalize(
+               struct bt_notification_iterator *iterator)
+{
+       struct bt_component_class *comp_class = NULL;
+       bt_component_class_notification_iterator_finalize_method
+               finalize_method = NULL;
+
+       assert(iterator);
+
+       switch (iterator->state) {
+       case BT_NOTIFICATION_ITERATOR_STATE_FINALIZED:
+       case BT_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED:
+               /* Already finalized */
+               BT_LOGD("Not finalizing notification iterator: already finalized: "
+                       "addr=%p", iterator);
+               return;
+       default:
+               break;
+       }
+
+       BT_LOGD("Finalizing notification iterator: addr=%p", iterator);
+
+       if (iterator->state == BT_NOTIFICATION_ITERATOR_STATE_ENDED) {
+               BT_LOGD("Updating notification iterator's state: "
+                       "new-state=BT_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED");
+               iterator->state = BT_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED;
+       } else {
+               BT_LOGD("Updating notification iterator's state: "
+                       "new-state=BT_NOTIFICATION_ITERATOR_STATE_FINALIZED");
+               iterator->state = BT_NOTIFICATION_ITERATOR_STATE_FINALIZED;
+       }
+
+       assert(iterator->upstream_component);
+       comp_class = iterator->upstream_component->class;
+
+       /* Call user-defined destroy method */
+       switch (comp_class->type) {
+       case BT_COMPONENT_CLASS_TYPE_SOURCE:
+       {
+               struct bt_component_class_source *source_class;
+
+               source_class = container_of(comp_class, struct bt_component_class_source, parent);
+               finalize_method = source_class->methods.iterator.finalize;
+               break;
+       }
+       case BT_COMPONENT_CLASS_TYPE_FILTER:
+       {
+               struct bt_component_class_filter *filter_class;
+
+               filter_class = container_of(comp_class, struct bt_component_class_filter, parent);
+               finalize_method = filter_class->methods.iterator.finalize;
+               break;
+       }
+       default:
+               /* Unreachable */
+               abort();
+       }
+
+       if (finalize_method) {
+               BT_LOGD("Calling user's finalization method: addr=%p",
+                       iterator);
+               finalize_method(
+                       bt_private_notification_iterator_from_notification_iterator(iterator));
+       }
+
+       iterator->upstream_component = NULL;
+       iterator->upstream_port = NULL;
+       BT_LOGD("Finalized notification iterator: addr=%p", iterator);
+}
+
+BT_HIDDEN
+void bt_notification_iterator_set_connection(
+               struct bt_notification_iterator *iterator,
+               struct bt_connection *connection)
+{
+       assert(iterator);
+       iterator->connection = connection;
+       BT_LOGV("Set notification iterator's connection: "
+               "iter-addr=%p, conn-addr=%p", iterator, connection);
+}
+
+static
+int create_subscription_mask_from_notification_types(
+               struct bt_notification_iterator *iterator,
+               const enum bt_notification_type *notif_types)
+{
+       const enum bt_notification_type *notif_type;
+       int ret = 0;
+
+       assert(notif_types);
+       iterator->subscription_mask = 0;
+
+       for (notif_type = notif_types;
+                       *notif_type != BT_NOTIFICATION_TYPE_SENTINEL;
+                       notif_type++) {
+               switch (*notif_type) {
+               case BT_NOTIFICATION_TYPE_ALL:
+                       iterator->subscription_mask |=
+                               BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_EVENT |
+                               BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_INACTIVITY |
+                               BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_BEGIN |
+                               BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_END |
+                               BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_BEGIN |
+                               BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_END;
+                       break;
+               case BT_NOTIFICATION_TYPE_EVENT:
+                       iterator->subscription_mask |= BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_EVENT;
+                       break;
+               case BT_NOTIFICATION_TYPE_INACTIVITY:
+                       iterator->subscription_mask |= BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_INACTIVITY;
+                       break;
+               case BT_NOTIFICATION_TYPE_STREAM_BEGIN:
+                       iterator->subscription_mask |= BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_BEGIN;
+                       break;
+               case BT_NOTIFICATION_TYPE_STREAM_END:
+                       iterator->subscription_mask |= BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_END;
+                       break;
+               case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
+                       iterator->subscription_mask |= BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_BEGIN;
+                       break;
+               case BT_NOTIFICATION_TYPE_PACKET_END:
+                       iterator->subscription_mask |= BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_END;
+                       break;
+               default:
+                       ret = -1;
+                       goto end;
+               }
+
+               BT_LOGV("Added notification type to subscription mask: "
+                       "type=%s, mask=%x",
+                       bt_notification_type_string(*notif_type),
+                       iterator->subscription_mask);
+       }
+
+end:
+       return ret;
+}
+
 BT_HIDDEN
 struct bt_notification_iterator *bt_notification_iterator_create(
                struct bt_component *upstream_comp,
-               struct bt_port *upstream_port)
+               struct bt_port *upstream_port,
+               const enum bt_notification_type *notification_types,
+               struct bt_connection *connection)
 {
        enum bt_component_class_type type;
        struct bt_notification_iterator *iterator = NULL;
 
        assert(upstream_comp);
        assert(upstream_port);
+       assert(notification_types);
        assert(bt_port_is_connected(upstream_port));
-
+       BT_LOGD("Creating notification iterator: "
+               "upstream-comp-addr=%p, upstream-comp-name=\"%s\", "
+               "upstream-port-addr=%p, upstream-port-name=\"%s\", "
+               "conn-addr=%p",
+               upstream_comp, bt_component_get_name(upstream_comp),
+               upstream_port, bt_port_get_name(upstream_port),
+               connection);
        type = bt_component_get_class_type(upstream_comp);
-       switch (type) {
-       case BT_COMPONENT_CLASS_TYPE_SOURCE:
-       case BT_COMPONENT_CLASS_TYPE_FILTER:
-               break;
-       default:
-               goto error;
-       }
-
+       assert(type == BT_COMPONENT_CLASS_TYPE_SOURCE ||
+               type == BT_COMPONENT_CLASS_TYPE_FILTER);
        iterator = g_new0(struct bt_notification_iterator, 1);
        if (!iterator) {
+               BT_LOGE_STR("Failed to allocate one notification iterator.");
                goto error;
        }
 
        bt_object_init(iterator, bt_notification_iterator_destroy);
 
+       if (create_subscription_mask_from_notification_types(iterator,
+                       notification_types)) {
+               BT_LOGW_STR("Cannot create subscription mask from notification types.");
+               goto error;
+       }
+
        iterator->stream_states = g_hash_table_new_full(g_direct_hash,
                g_direct_equal, NULL, (GDestroyNotify) destroy_stream_state);
        if (!iterator->stream_states) {
+               BT_LOGE_STR("Failed to allocate a GHashTable.");
                goto error;
        }
 
        iterator->queue = g_queue_new();
        if (!iterator->queue) {
+               BT_LOGE_STR("Failed to allocate a GQueue.");
                goto error;
        }
 
        iterator->actions = g_array_new(FALSE, FALSE, sizeof(struct action));
        if (!iterator->actions) {
+               BT_LOGE_STR("Failed to allocate a GArray.");
                goto error;
        }
 
-       iterator->upstream_component = bt_get(upstream_comp);
-       iterator->upstream_port = bt_get(upstream_port);
+       iterator->upstream_component = upstream_comp;
+       iterator->upstream_port = upstream_port;
+       iterator->connection = connection;
+       iterator->state = BT_NOTIFICATION_ITERATOR_STATE_ACTIVE;
+       BT_LOGD("Created notification iterator: "
+               "upstream-comp-addr=%p, upstream-comp-name=\"%s\", "
+               "upstream-port-addr=%p, upstream-port-name=\"%s\", "
+               "conn-addr=%p, iter-addr=%p",
+               upstream_comp, bt_component_get_name(upstream_comp),
+               upstream_port, bt_port_get_name(upstream_port),
+               connection, iterator);
        goto end;
 
 error:
@@ -396,21 +589,6 @@ end:
        return iterator;
 }
 
-BT_HIDDEN
-enum bt_notification_iterator_status bt_notification_iterator_validate(
-               struct bt_notification_iterator *iterator)
-{
-       enum bt_notification_iterator_status ret =
-                       BT_NOTIFICATION_ITERATOR_STATUS_OK;
-
-       if (!iterator) {
-               ret = BT_NOTIFICATION_ITERATOR_STATUS_INVALID;
-               goto end;
-       }
-end:
-       return ret;
-}
-
 void *bt_private_notification_iterator_get_user_data(
                struct bt_private_notification_iterator *private_iterator)
 {
@@ -431,11 +609,15 @@ bt_private_notification_iterator_set_user_data(
                bt_notification_iterator_from_private(private_iterator);
 
        if (!iterator) {
+               BT_LOGW_STR("Invalid parameter: notification iterator is NULL.");
                ret = BT_NOTIFICATION_ITERATOR_STATUS_INVALID;
                goto end;
        }
 
        iterator->user_data = data;
+       BT_LOGV("Set notification iterator's user data: "
+               "iter-addr=%p, user-data-addr=%p", iterator, data);
+
 end:
        return ret;
 }
@@ -446,6 +628,7 @@ struct bt_notification *bt_notification_iterator_get_notification(
        struct bt_notification *notification = NULL;
 
        if (!iterator) {
+               BT_LOGW_STR("Invalid parameter: notification iterator is NULL.");
                goto end;
        }
 
@@ -456,12 +639,45 @@ end:
 }
 
 static
-bool validate_notification(struct bt_notification_iterator *iterator,
+enum bt_notification_iterator_notif_type
+bt_notification_iterator_notif_type_from_notif_type(
+               enum bt_notification_type notif_type)
+{
+       enum bt_notification_iterator_notif_type iter_notif_type;
+
+       switch (notif_type) {
+       case BT_NOTIFICATION_TYPE_EVENT:
+               iter_notif_type = BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_EVENT;
+               break;
+       case BT_NOTIFICATION_TYPE_INACTIVITY:
+               iter_notif_type = BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_INACTIVITY;
+               break;
+       case BT_NOTIFICATION_TYPE_STREAM_BEGIN:
+               iter_notif_type = BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_BEGIN;
+               break;
+       case BT_NOTIFICATION_TYPE_STREAM_END:
+               iter_notif_type = BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_STREAM_END;
+               break;
+       case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
+               iter_notif_type = BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_BEGIN;
+               break;
+       case BT_NOTIFICATION_TYPE_PACKET_END:
+               iter_notif_type = BT_NOTIFICATION_ITERATOR_NOTIF_TYPE_PACKET_END;
+               break;
+       default:
+               abort();
+       }
+
+       return iter_notif_type;
+}
+
+static
+bt_bool validate_notification(struct bt_notification_iterator *iterator,
                struct bt_notification *notif,
                struct bt_ctf_stream *notif_stream,
                struct bt_ctf_packet *notif_packet)
 {
-       bool is_valid = true;
+       bt_bool is_valid = BT_TRUE;
        struct stream_state *stream_state;
        struct bt_port *stream_comp_cur_port;
 
@@ -495,7 +711,19 @@ bool validate_notification(struct bt_notification_iterator *iterator,
                         * bad: the API guarantees that it can never
                         * happen.
                         */
-                       is_valid = false;
+                       BT_LOGW("Two different ports of the same component are emitting notifications which refer to the same stream: "
+                               "stream-addr=%p, stream-name=\"%s\", "
+                               "stream-comp-cur-port-addr=%p, "
+                               "stream-comp-cur-port-name=%p, "
+                               "iter-upstream-port-addr=%p, "
+                               "iter-upstream-port-name=%s",
+                               notif_stream,
+                               bt_ctf_stream_get_name(notif_stream),
+                               stream_comp_cur_port,
+                               bt_port_get_name(stream_comp_cur_port),
+                               iterator->upstream_port,
+                               bt_port_get_name(iterator->upstream_port));
+                       is_valid = BT_FALSE;
                        goto end;
                }
 
@@ -504,6 +732,12 @@ bool validate_notification(struct bt_notification_iterator *iterator,
        stream_state = g_hash_table_lookup(iterator->stream_states,
                notif_stream);
        if (stream_state) {
+               BT_LOGV("Stream state already exists: "
+                       "stream-addr=%p, stream-name=\"%s\", "
+                       "stream-state-addr=%p",
+                       notif_stream,
+                       bt_ctf_stream_get_name(notif_stream), stream_state);
+
                if (stream_state->is_ended) {
                        /*
                         * There's a new notification which has a
@@ -513,7 +747,11 @@ bool validate_notification(struct bt_notification_iterator *iterator,
                         * bad: the API guarantees that it can never
                         * happen.
                         */
-                       is_valid = false;
+                       BT_LOGW("Stream is already ended: "
+                               "stream-addr=%p, stream-name=\"%s\"",
+                               notif_stream,
+                               bt_ctf_stream_get_name(notif_stream));
+                       is_valid = BT_FALSE;
                        goto end;
                }
 
@@ -524,12 +762,22 @@ bool validate_notification(struct bt_notification_iterator *iterator,
                         * we already returned a "stream begin"
                         * notification: this is an invalid duplicate.
                         */
-                       is_valid = false;
+                       BT_LOGW("Duplicate stream beginning notification: "
+                               "stream-addr=%p, stream-name=\"%s\"",
+                               notif_stream,
+                               bt_ctf_stream_get_name(notif_stream));
+                       is_valid = BT_FALSE;
                        goto end;
                case BT_NOTIFICATION_TYPE_PACKET_BEGIN:
                        if (notif_packet == stream_state->cur_packet) {
                                /* Duplicate "packet begin" notification */
-                               is_valid = false;
+                               BT_LOGW("Duplicate stream beginning notification: "
+                                       "stream-addr=%p, stream-name=\"%s\", "
+                                       "packet-addr=%p",
+                                       notif_stream,
+                                       bt_ctf_stream_get_name(notif_stream),
+                                       notif_packet);
+                               is_valid = BT_FALSE;
                                goto end;
                        }
                        break;
@@ -542,19 +790,34 @@ end:
        return is_valid;
 }
 
+static
+bt_bool is_subscribed_to_notification_type(struct bt_notification_iterator *iterator,
+               enum bt_notification_type notif_type)
+{
+       uint32_t iter_notif_type =
+               (uint32_t) bt_notification_iterator_notif_type_from_notif_type(
+                       notif_type);
+
+       return (iter_notif_type & iterator->subscription_mask) ? BT_TRUE : BT_FALSE;
+}
+
 static
 void add_action_push_notif(struct bt_notification_iterator *iterator,
                struct bt_notification *notif)
 {
        struct action action = {
                .type = ACTION_TYPE_PUSH_NOTIF,
-               .payload.push_notif = {
-                       .notif = bt_get(notif),
-               },
        };
 
        assert(notif);
+
+       if (!is_subscribed_to_notification_type(iterator, notif->type)) {
+               return;
+       }
+
+       action.payload.push_notif.notif = bt_get(notif);
        add_action(iterator, &action);
+       BT_LOGV("Added \"push notification\" action: notif-addr=%p", notif);
 }
 
 static
@@ -565,13 +828,25 @@ int add_action_push_notif_stream_begin(
        int ret = 0;
        struct bt_notification *stream_begin_notif = NULL;
 
+       if (!is_subscribed_to_notification_type(iterator,
+                       BT_NOTIFICATION_TYPE_STREAM_BEGIN)) {
+               BT_LOGV("Not adding \"push stream beginning notification\" action: "
+                       "notification iterator is not subscribed: addr=%p",
+                       iterator);
+               goto end;
+       }
+
        assert(stream);
        stream_begin_notif = bt_notification_stream_begin_create(stream);
        if (!stream_begin_notif) {
+               BT_LOGE_STR("Cannot create stream beginning notification.");
                goto error;
        }
 
        add_action_push_notif(iterator, stream_begin_notif);
+       BT_LOGV("Added \"push stream beginning notification\" action: "
+               "stream-addr=%p, stream-name=\"%s\"",
+               stream, bt_ctf_stream_get_name(stream));
        goto end;
 
 error:
@@ -590,13 +865,25 @@ int add_action_push_notif_stream_end(
        int ret = 0;
        struct bt_notification *stream_end_notif = NULL;
 
+       if (!is_subscribed_to_notification_type(iterator,
+                       BT_NOTIFICATION_TYPE_STREAM_END)) {
+               BT_LOGV("Not adding \"push stream end notification\" action: "
+                       "notification iterator is not subscribed: addr=%p",
+                       iterator);
+               goto end;
+       }
+
        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:
@@ -615,13 +902,24 @@ int add_action_push_notif_packet_begin(
        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:
@@ -640,13 +938,24 @@ int add_action_push_notif_packet_end(
        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:
@@ -671,6 +980,8 @@ void add_action_set_stream_state_is_ended(
 
        assert(stream_state);
        add_action(iterator, &action);
+       BT_LOGV("Added \"set stream state's ended\" action: "
+               "stream-state-addr=%p", stream_state);
 }
 
 static
@@ -689,6 +1000,9 @@ void add_action_set_stream_state_cur_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
@@ -725,6 +1039,7 @@ int ensure_stream_state_exists(struct bt_notification_iterator *iterator,
 
                *stream_state = create_stream_state(notif_stream);
                if (!stream_state) {
+                       BT_LOGE_STR("Cannot create stream state.");
                        goto error;
                }
 
@@ -738,6 +1053,7 @@ int ensure_stream_state_exists(struct bt_notification_iterator *iterator,
                        ret = add_action_push_notif_stream_begin(iterator,
                                notif_stream);
                        if (ret) {
+                               BT_LOGE_STR("Cannot add \"push stream beginning notification\" action.");
                                goto error;
                        }
                }
@@ -765,11 +1081,16 @@ int handle_packet_switch(struct bt_notification_iterator *iterator,
                goto end;
        }
 
+       BT_LOGV("Handling packet switch: "
+               "cur-packet-addr=%p, new-packet-addr=%p",
+               stream_state->cur_packet, new_packet);
+
        if (stream_state->cur_packet) {
                /* End of the current packet */
                ret = add_action_push_notif_packet_end(iterator,
                        stream_state->cur_packet);
                if (ret) {
+                       BT_LOGE_STR("Cannot add \"push packet end notification\" action.");
                        goto error;
                }
        }
@@ -781,6 +1102,7 @@ int handle_packet_switch(struct bt_notification_iterator *iterator,
                ret = add_action_push_notif_packet_begin(iterator,
                        new_packet);
                if (ret) {
+                       BT_LOGE_STR("Cannot add \"push packet beginning notification\" action.");
                        goto error;
                }
        }
@@ -810,6 +1132,7 @@ int handle_notif_stream_begin(
        ret = ensure_stream_state_exists(iterator, notif, notif_stream,
                &stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot ensure that stream state exists.");
                goto error;
        }
 
@@ -836,11 +1159,13 @@ int handle_notif_stream_end(
        ret = ensure_stream_state_exists(iterator, NULL, notif_stream,
                &stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot ensure that stream state exists.");
                goto error;
        }
 
        ret = handle_packet_switch(iterator, NULL, NULL, stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot handle packet switch.");
                goto error;
        }
 
@@ -870,11 +1195,13 @@ int handle_notif_packet_begin(
        ret = ensure_stream_state_exists(iterator, NULL, notif_stream,
                &stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot ensure that stream state exists.");
                goto error;
        }
 
        ret = handle_packet_switch(iterator, notif, notif_packet, stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot handle packet switch.");
                goto error;
        }
 
@@ -902,11 +1229,13 @@ int handle_notif_packet_end(
        ret = ensure_stream_state_exists(iterator, NULL, notif_stream,
                &stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot ensure that stream state exists.");
                goto error;
        }
 
        ret = handle_packet_switch(iterator, NULL, notif_packet, stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot handle packet switch.");
                goto error;
        }
 
@@ -937,11 +1266,13 @@ int handle_notif_event(
        ret = ensure_stream_state_exists(iterator, NULL, notif_stream,
                &stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot ensure that stream state exists.");
                goto error;
        }
 
        ret = handle_packet_switch(iterator, NULL, notif_packet, stream_state);
        if (ret) {
+               BT_LOGE_STR("Cannot handle packet switch.");
                goto error;
        }
 
@@ -967,6 +1298,12 @@ int enqueue_notification_and_automatic(
 
        assert(notif);
 
+       BT_LOGV("Enqueuing user notification and automatic notifications: "
+               "iter-addr=%p, notif-addr=%p", iterator, notif);
+
+       // TODO: Skip most of this if the iterator is only subscribed
+       //       to event/inactivity notifications.
+
        /* Get the stream and packet referred by the notification */
        switch (notif->type) {
        case BT_NOTIFICATION_TYPE_EVENT:
@@ -995,7 +1332,7 @@ int enqueue_notification_and_automatic(
                break;
        case BT_NOTIFICATION_TYPE_INACTIVITY:
                /* Always valid */
-               break;
+               goto handle_notif;
        default:
                /*
                 * Invalid type of notification. Only the notification
@@ -1015,14 +1352,17 @@ int enqueue_notification_and_automatic(
                 * The notification has no reference to a stream: it
                 * cannot cause the creation of automatic notifications.
                 */
+               BT_LOGV_STR("Notification has no reference to any stream: skipping automatic notification generation.");
                goto end;
        }
 
        if (!validate_notification(iterator, notif, notif_stream,
                        notif_packet)) {
+               BT_LOGW_STR("Invalid notification.");
                goto error;
        }
 
+handle_notif:
        switch (notif->type) {
        case BT_NOTIFICATION_TYPE_EVENT:
                ret = handle_notif_event(iterator, notif, notif_stream,
@@ -1042,15 +1382,21 @@ int enqueue_notification_and_automatic(
                ret = handle_notif_packet_end(iterator, notif, notif_stream,
                        notif_packet);
                break;
+       case BT_NOTIFICATION_TYPE_INACTIVITY:
+               add_action_push_notif(iterator, notif);
+               break;
        default:
                break;
        }
 
        if (ret) {
+               BT_LOGW_STR("Failed to handle notification for automatic notification generation.");
                goto error;
        }
 
        apply_actions(iterator);
+       BT_LOGV("Enqueued user notification and automatic notifications: "
+               "iter-addr=%p, notif-addr=%p", iterator, notif);
        goto end;
 
 error:
@@ -1067,6 +1413,8 @@ int handle_end(struct bt_notification_iterator *iterator)
        gpointer stream_gptr, stream_state_gptr;
        int ret = 0;
 
+       BT_LOGV("Handling end of iteration: addr=%p", iterator);
+
        /*
         * Emit a "stream end" notification for each non-ended stream
         * known by this iterator and mark them as ended.
@@ -1085,11 +1433,13 @@ int handle_end(struct bt_notification_iterator *iterator)
 
                ret = handle_packet_switch(iterator, NULL, NULL, stream_state);
                if (ret) {
+                       BT_LOGE_STR("Cannot handle packet switch.");
                        goto error;
                }
 
                ret = add_action_push_notif_stream_end(iterator, stream_gptr);
                if (ret) {
+                       BT_LOGE_STR("Cannot add \"push stream end notification\" action.");
                        goto error;
                }
 
@@ -1097,6 +1447,7 @@ int handle_end(struct bt_notification_iterator *iterator)
        }
 
        apply_actions(iterator);
+       BT_LOGV("Handled end of iteration: addr=%p", iterator);
        goto end;
 
 error:
@@ -1122,15 +1473,27 @@ enum bt_notification_iterator_status ensure_queue_has_notifications(
        int ret;
 
        assert(iterator);
+       BT_LOGD("Ensuring that notification iterator's queue has at least one notification: "
+               "iter-addr=%p, queue-size=%u",
+               iterator, iterator->queue->length);
 
        if (iterator->queue->length > 0) {
                /* We already have enough */
+               BT_LOGD_STR("Queue already has at least one notification.");
                goto end;
        }
 
-       if (iterator->is_ended) {
+       switch (iterator->state) {
+       case BT_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED:
+               BT_LOGD_STR("Notification iterator's \"next\" called, but it is finalized.");
+               status = BT_NOTIFICATION_ITERATOR_STATUS_CANCELED;
+               goto end;
+       case BT_NOTIFICATION_ITERATOR_STATE_ENDED:
+               BT_LOGD_STR("Notification iterator is ended.");
                status = BT_NOTIFICATION_ITERATOR_STATUS_END;
                goto end;
+       default:
+               break;
        }
 
        assert(iterator->upstream_component);
@@ -1159,61 +1522,82 @@ enum bt_notification_iterator_status ensure_queue_has_notifications(
                break;
        }
        default:
-               assert(false);
-               break;
+               abort();
        }
 
        /*
         * Call the user's "next" method to get the next notification
-        * and status, skipping the forwarded automatic notifications
-        * if any.
+        * and status.
         */
        assert(next_method);
-       next_return = next_method(priv_iterator);
-       if (next_return.status < 0) {
-               status = next_return.status;
-               goto end;
-       }
 
-       switch (next_return.status) {
-       case BT_NOTIFICATION_ITERATOR_STATUS_END:
-               ret = handle_end(iterator);
-               if (ret) {
-                       status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
+       while (iterator->queue->length == 0) {
+               BT_LOGD_STR("Calling user's \"next\" method.");
+               next_return = next_method(priv_iterator);
+               BT_LOGD("User method returned: status=%s",
+                       bt_notification_iterator_status_string(next_return.status));
+               if (next_return.status < 0) {
+                       BT_LOGW_STR("User method failed.");
+                       status = next_return.status;
                        goto end;
                }
 
-               if (iterator->queue->length == 0) {
-                       status = BT_NOTIFICATION_ITERATOR_STATUS_END;
-               }
+               switch (next_return.status) {
+               case BT_NOTIFICATION_ITERATOR_STATUS_END:
+                       ret = handle_end(iterator);
+                       if (ret) {
+                               BT_LOGW_STR("Cannot handle end of iteration.");
+                               status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
+                               goto end;
+                       }
 
-               iterator->is_ended = true;
-               break;
-       case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
-               status = BT_NOTIFICATION_ITERATOR_STATUS_AGAIN;
-               break;
-       case BT_NOTIFICATION_ITERATOR_STATUS_OK:
-               if (!next_return.notification) {
-                       status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
-                       goto end;
-               }
+                       if (iterator->state == BT_NOTIFICATION_ITERATOR_STATE_FINALIZED) {
+                               iterator->state =
+                                       BT_NOTIFICATION_ITERATOR_STATE_FINALIZED_AND_ENDED;
 
-               /*
-                * We know the notification is valid. Before we push it
-                * to the head of the queue, push the appropriate
-                * automatic notifications if any.
-                */
-               ret = enqueue_notification_and_automatic(iterator,
-                       next_return.notification);
-               BT_PUT(next_return.notification);
-               if (ret) {
-                       status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
+                               if (iterator->queue->length == 0) {
+                                       status = BT_NOTIFICATION_ITERATOR_STATUS_CANCELED;
+                               }
+                       } else {
+                               iterator->state =
+                                       BT_NOTIFICATION_ITERATOR_STATE_ENDED;
+
+                               if (iterator->queue->length == 0) {
+                                       status = BT_NOTIFICATION_ITERATOR_STATUS_END;
+                               }
+                       }
+
+                       BT_LOGD("Set new status: status=%s",
+                               bt_notification_iterator_status_string(status));
                        goto end;
+               case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
+                       status = BT_NOTIFICATION_ITERATOR_STATUS_AGAIN;
+                       goto end;
+               case BT_NOTIFICATION_ITERATOR_STATUS_OK:
+                       if (!next_return.notification) {
+                               BT_LOGW_STR("User method returned BT_NOTIFICATION_ITERATOR_STATUS_OK, but notification is NULL.");
+                               status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
+                               goto end;
+                       }
+
+                       /*
+                        * We know the notification is valid. Before we
+                        * push it to the head of the queue, push the
+                        * appropriate automatic notifications if any.
+                        */
+                       ret = enqueue_notification_and_automatic(iterator,
+                               next_return.notification);
+                       BT_PUT(next_return.notification);
+                       if (ret) {
+                               BT_LOGW("Cannot enqueue notification and automatic notifications.");
+                               status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
+                               goto end;
+                       }
+                       break;
+               default:
+                       /* Unknown non-error status */
+                       abort();
                }
-               break;
-       default:
-               /* Unknown non-error status */
-               assert(false);
        }
 
 end:
@@ -1226,16 +1610,20 @@ bt_notification_iterator_next(struct bt_notification_iterator *iterator)
        enum bt_notification_iterator_status status;
 
        if (!iterator) {
+               BT_LOGW_STR("Invalid parameter: notification iterator is NULL.");
                status = BT_NOTIFICATION_ITERATOR_STATUS_INVALID;
                goto end;
        }
 
+       BT_LOGD("Notification iterator's \"next\": iter-addr=%p", iterator);
+
        /*
         * Make sure that the iterator's queue contains at least one
         * notification.
         */
        status = ensure_queue_has_notifications(iterator);
        if (status != BT_NOTIFICATION_ITERATOR_STATUS_OK) {
+               /* Not an error */
                goto end;
        }
 
This page took 0.035262 seconds and 4 git commands to generate.