X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=lib%2Fplugin-system%2Fiterator.c;h=06fea4aa85824f8294f7f36bf626bff956299d85;hb=413bc2c417a706e7e2cc331077421e89e36daea0;hp=6a40ba70c176c27a0b72b33110dfbf9df8a005c2;hpb=56a1ccedd4a58b6c87d1bbd94e22094ad5ac1a98;p=babeltrace.git

diff --git a/lib/plugin-system/iterator.c b/lib/plugin-system/iterator.c
index 6a40ba70..06fea4aa 100644
--- a/lib/plugin-system/iterator.c
+++ b/lib/plugin-system/iterator.c
@@ -45,6 +45,7 @@ void bt_notification_iterator_destroy(struct bt_object *obj)
 	if (iterator->user_destroy) {
 		iterator->user_destroy(iterator);
 	}
+	BT_PUT(iterator->component);
 	g_free(iterator);
 }
 
@@ -64,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;
@@ -158,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);
+}
+