Add `-internal` suffix to all internal header files

[babeltrace.git] / lib / component / iterator.c

diff --git a/lib/component/iterator.c b/lib/component/iterator.c
index e2ffd2f017c098bfe7bc378b0a5b127a728570ee..29562955dd72e4baa3b8e455df24cefb8098fc04 100644 (file)
--- a/lib/component/iterator.c
+++ b/lib/component/iterator.c
@@ -26,13 +26,14 @@
  * SOFTWARE.
  */
 
-#include <babeltrace/compiler.h>
+#include <babeltrace/compiler-internal.h>
 #include <babeltrace/ref.h>
-#include <babeltrace/component/component.h>
-#include <babeltrace/component/component-source-internal.h>
-#include <babeltrace/component/component-class-internal.h>
-#include <babeltrace/component/notification/iterator.h>
-#include <babeltrace/component/notification/iterator-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-iterator.h>
+#include <babeltrace/graph/notification-iterator-internal.h>
+#include <babeltrace/graph/notification-internal.h>
 
 static
 void bt_notification_iterator_destroy(struct bt_object *obj)
@@ -226,6 +227,7 @@ bt_notification_iterator_next(struct bt_notification_iterator *iterator)
 
                BT_MOVE(iterator->current_notification,
                        next_return.notification);
+               bt_notification_freeze(iterator->current_notification);
        }
 
 end:
@@ -238,6 +240,15 @@ struct bt_component *bt_notification_iterator_get_component(
        return bt_get(iterator->component);
 }
 
+struct bt_private_component *
+bt_private_notification_iterator_get_private_component(
+               struct bt_private_notification_iterator *private_iterator)
+{
+       return bt_private_component_from_component(
+               bt_notification_iterator_get_component(
+                       bt_notification_iterator_from_private(private_iterator)));
+}
+
 enum bt_notification_iterator_status bt_notification_iterator_seek_time(
                struct bt_notification_iterator *iterator,
                enum bt_notification_iterator_seek_origin seek_origin,