From: Jérémie Galarneau <jeremie.galarneau@efficios.com>
Date: Sun, 11 Dec 2016 08:59:43 +0000 (-0500)
Subject: Clean-up notification iterator creation function
X-Git-Url: http://drtracing.org/?a=commitdiff_plain;h=16b7b0235963f6fc1a9d12ff8688e9698f32f147;p=deliverable%2Fbabeltrace.git

Clean-up notification iterator creation function

Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>
---

diff --git a/lib/plugin-system/iterator.c b/lib/plugin-system/iterator.c
index 71393d126..b0b5d38b6 100644
--- a/lib/plugin-system/iterator.c
+++ b/lib/plugin-system/iterator.c
@@ -53,10 +53,19 @@ BT_HIDDEN
 struct bt_notification_iterator *bt_notification_iterator_create(
 		struct bt_component *component)
 {
+	enum bt_component_type type;
 	struct bt_notification_iterator *iterator = NULL;
 
-	if (!component ||
-		bt_component_get_type(component) != BT_COMPONENT_TYPE_SOURCE) {
+	if (!component) {
+		goto end;
+	}
+
+	type = bt_component_get_type(component);
+	switch (type) {
+	case BT_COMPONENT_TYPE_SOURCE:
+	case BT_COMPONENT_TYPE_FILTER:
+		break;
+	default:
 		goto end;
 	}