X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=plugins%2Ftext%2Fpretty%2Fpretty.c;h=e8f89d3a38ee2d2901b3427a14507c1df197c776;hb=73d5c1adb1411e16c9c613c38a4c74a29ee608ae;hp=9c994095871e556d149edf0fef49bdbe50482b00;hpb=5280f7427ab8be3f2c229ce001e987d8a292e1ea;p=babeltrace.git

diff --git a/plugins/text/pretty/pretty.c b/plugins/text/pretty/pretty.c
index 9c994095..e8f89d3a 100644
--- a/plugins/text/pretty/pretty.c
+++ b/plugins/text/pretty/pretty.c
@@ -167,6 +167,7 @@ void pretty_port_connected(
 		struct bt_private_port *self_port,
 		struct bt_port *other_port)
 {
+	enum bt_connection_status conn_status;
 	struct bt_private_connection *connection;
 	struct pretty_component *pretty;
 	static const enum bt_notification_type notif_types[] = {
@@ -179,11 +180,9 @@ void pretty_port_connected(
 	assert(!pretty->input_iterator);
 	connection = bt_private_port_get_private_connection(self_port);
 	assert(connection);
-	pretty->input_iterator =
-		bt_private_connection_create_notification_iterator(connection,
-			notif_types);
-
-	if (!pretty->input_iterator) {
+	conn_status = bt_private_connection_create_notification_iterator(
+		connection, notif_types, &pretty->input_iterator);
+	if (conn_status != BT_CONNECTION_STATUS_OK) {
 		pretty->error = true;
 	}
 
@@ -734,22 +733,18 @@ enum bt_component_status pretty_init(
 {
 	enum bt_component_status ret;
 	struct pretty_component *pretty = create_pretty();
-	void *priv_port;
 
 	if (!pretty) {
 		ret = BT_COMPONENT_STATUS_NOMEM;
 		goto end;
 	}
 
-	priv_port = bt_private_component_sink_add_input_private_port(component,
-		"in", NULL);
-	if (!priv_port) {
-		ret = BT_COMPONENT_STATUS_NOMEM;
+	ret = bt_private_component_sink_add_input_private_port(component,
+		"in", NULL, NULL);
+	if (ret != BT_COMPONENT_STATUS_OK) {
 		goto end;
 	}
 
-	bt_put(priv_port);
-
 	pretty->out = stdout;
 	pretty->err = stderr;