X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=lib%2Fgraph%2Fconnection.c;h=a9a5f9db667b2a86b0e8e8e117ab1aefe518f4a1;hb=fa054faf3a18fd8003510c32718c1fd4fbf3dd46;hp=bfc9c04b8f0b0831fd03a5a6cc47648a6713afa0;hpb=dc77b5215ef41c67486cf32bce85bfde0b415952;p=babeltrace.git

diff --git a/lib/graph/connection.c b/lib/graph/connection.c
index bfc9c04b..a9a5f9db 100644
--- a/lib/graph/connection.c
+++ b/lib/graph/connection.c
@@ -140,7 +140,8 @@ struct bt_port *bt_connection_get_downstream_port(
 
 struct bt_notification_iterator *
 bt_private_connection_create_notification_iterator(
-		struct bt_private_connection *private_connection)
+		struct bt_private_connection *private_connection,
+		const enum bt_notification_type *notification_types)
 {
 	enum bt_notification_iterator_status ret_iterator;
 	enum bt_component_class_type upstream_comp_class_type;
@@ -150,11 +151,19 @@ bt_private_connection_create_notification_iterator(
 	struct bt_component_class *upstream_comp_class = NULL;
 	struct bt_connection *connection = NULL;
 	bt_component_class_notification_iterator_init_method init_method = NULL;
+	static const enum bt_notification_type all_notif_types[] = {
+		BT_NOTIFICATION_TYPE_ALL,
+		BT_NOTIFICATION_TYPE_SENTINEL,
+	};
 
 	if (!private_connection) {
 		goto error;
 	}
 
+	if (!notification_types) {
+		notification_types = all_notif_types;
+	}
+
 	connection = bt_connection_from_private(private_connection);
 
 	if (!connection->upstream_port || !connection->downstream_port) {
@@ -180,7 +189,7 @@ bt_private_connection_create_notification_iterator(
 	}
 
 	iterator = bt_notification_iterator_create(upstream_component,
-		upstream_port);
+		upstream_port, notification_types);
 	if (!iterator) {
 		goto error;
 	}