Implement ctf fs iterator functions

[babeltrace.git] / lib / plugin-system / iterator.c

diff --git a/lib/plugin-system/iterator.c b/lib/plugin-system/iterator.c
index 45f73bdd3c3be6df976df747d43fe22c3b6081cf..06fea4aa85824f8294f7f36bf626bff956299d85 100644 (file)
--- a/lib/plugin-system/iterator.c
+++ b/lib/plugin-system/iterator.c
@@ -42,7 +42,10 @@ void bt_notification_iterator_destroy(struct bt_object *obj)
        iterator = container_of(obj, struct bt_notification_iterator,
                        base);
        assert(iterator->user_destroy || !iterator->user_data);
-       iterator->user_destroy(iterator);
+       if (iterator->user_destroy) {
+               iterator->user_destroy(iterator);
+       }
+       BT_PUT(iterator->component);
        g_free(iterator);
 }
 
@@ -62,6 +65,7 @@ struct bt_notification_iterator *bt_notification_iterator_create(
                goto end;
        }
 
+       iterator->component = bt_get(component);
        bt_object_init(iterator, bt_notification_iterator_destroy);
 end:
        return iterator;
@@ -156,3 +160,10 @@ bt_notification_iterator_set_private_data(
 end:
        return ret;
 }
+
+struct bt_component *bt_notification_iterator_get_component(
+               struct bt_notification_iterator *iterator)
+{
+       return bt_get(iterator->component);
+}
+