End connection on destruction
[babeltrace.git] / lib / graph / connection.c
index a13bce3e55e2d3114aaf374891897eace082eb59..ef4fefede918ff625a4349a7e165797b51b7804e 100644 (file)
@@ -47,7 +47,6 @@ void bt_connection_destroy(struct bt_object *obj)
 {
        struct bt_connection *connection = container_of(obj,
                        struct bt_connection, base);
-       size_t i;
 
        BT_LOGD("Destroying connection: addr=%p", connection);
 
@@ -60,32 +59,18 @@ void bt_connection_destroy(struct bt_object *obj)
         * Because connections are destroyed before components within a
         * graph, this ensures that notification iterators are always
         * finalized before their upstream component.
+        *
+        * Ending the connection does exactly this. We pass `false` to
+        * bt_connection_end() here to avoid removing this connection
+        * from the graph: if we're here, we're already in the graph's
+        * destructor.
         */
-       if (connection->iterators) {
-               for (i = 0; i < connection->iterators->len; i++) {
-                       struct bt_notification_iterator *iterator =
-                               g_ptr_array_index(connection->iterators, i);
-
-                       BT_LOGD("Finalizing notification iterator created by this connection: "
-                               "iter-addr=%p", iterator);
-                       bt_notification_iterator_finalize(iterator);
-
-                       /*
-                        * Make sure this iterator does not try to
-                        * remove itself from this connection's
-                        * iterators on destruction because this
-                        * connection won't exist anymore.
-                        */
-                       bt_notification_iterator_set_connection(iterator,
-                               NULL);
-               }
-
-               g_ptr_array_free(connection->iterators, TRUE);
-       }
+       bt_connection_end(connection, false);
+       g_ptr_array_free(connection->iterators, TRUE);
 
        /*
-        * No bt_put on ports as a connection only holds _weak_ references
-        * to them.
+        * No bt_put on ports as a connection only holds _weak_
+        * references to them.
         */
        g_free(connection);
 }
@@ -196,22 +181,32 @@ end:
 }
 
 BT_HIDDEN
-void bt_connection_disconnect_ports(struct bt_connection *conn)
+void bt_connection_end(struct bt_connection *conn,
+               bool try_remove_from_graph)
 {
        struct bt_component *downstream_comp = NULL;
        struct bt_component *upstream_comp = NULL;
        struct bt_port *downstream_port = conn->downstream_port;
        struct bt_port *upstream_port = conn->upstream_port;
-       struct bt_graph *graph = (void *) bt_object_borrow_parent(conn);
+       struct bt_graph *graph = bt_connection_borrow_graph(conn);
        size_t i;
 
+       BT_LOGD("Ending connection: conn-addr=%p, try-remove-from-graph=%d",
+               conn, try_remove_from_graph);
+
        if (downstream_port) {
+               BT_LOGD("Disconnecting connection's downstream port: "
+                       "port-addr=%p, port-name=\"%s\"",
+                       downstream_port, bt_port_get_name(downstream_port));
                downstream_comp = bt_port_get_component(downstream_port);
                bt_port_set_connection(downstream_port, NULL);
                conn->downstream_port = NULL;
        }
 
        if (upstream_port) {
+               BT_LOGD("Disconnecting connection's upstream port: "
+                       "port-addr=%p, port-name=\"%s\"",
+                       upstream_port, bt_port_get_name(upstream_port));
                upstream_comp = bt_port_get_component(upstream_port);
                bt_port_set_connection(upstream_port, NULL);
                conn->upstream_port = NULL;
@@ -257,7 +252,10 @@ void bt_connection_disconnect_ports(struct bt_connection *conn)
        }
 
        g_ptr_array_set_size(conn->iterators, 0);
-       bt_connection_try_remove_from_graph(conn);
+
+       if (try_remove_from_graph) {
+               bt_connection_try_remove_from_graph(conn);
+       }
 }
 
 struct bt_port *bt_connection_get_upstream_port(
@@ -272,10 +270,11 @@ struct bt_port *bt_connection_get_downstream_port(
        return connection ? bt_get(connection->downstream_port) : NULL;
 }
 
-struct bt_notification_iterator *
+enum bt_connection_status
 bt_private_connection_create_notification_iterator(
                struct bt_private_connection *private_connection,
-               const enum bt_notification_type *notification_types)
+               const enum bt_notification_type *notification_types,
+               struct bt_notification_iterator **user_iterator)
 {
        enum bt_component_class_type upstream_comp_class_type;
        struct bt_notification_iterator *iterator = NULL;
@@ -284,6 +283,7 @@ 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;
+       enum bt_connection_status status;
        static const enum bt_notification_type all_notif_types[] = {
                BT_NOTIFICATION_TYPE_ALL,
                BT_NOTIFICATION_TYPE_SENTINEL,
@@ -291,19 +291,43 @@ bt_private_connection_create_notification_iterator(
 
        if (!private_connection) {
                BT_LOGW_STR("Invalid parameter: private connection is NULL.");
-               goto error;
+               status = BT_CONNECTION_STATUS_INVALID;
+               goto end;
        }
 
-       if (!notification_types) {
-               BT_LOGD_STR("No notification types: subscribing to all notifications.");
-               notification_types = all_notif_types;
+       if (!user_iterator) {
+               BT_LOGW_STR("Invalid parameter: notification iterator pointer is NULL.");
+               status = BT_CONNECTION_STATUS_INVALID;
+               goto end;
        }
 
        connection = bt_connection_from_private(private_connection);
-       if (!connection->upstream_port || !connection->downstream_port) {
+
+       if (bt_graph_is_canceled(bt_connection_borrow_graph(connection))) {
+               BT_LOGW("Cannot create notification iterator from connection: "
+                       "connection's graph is canceled: "
+                       "conn-addr=%p, upstream-port-addr=%p, "
+                       "upstream-port-name=\"%s\", upstream-comp-addr=%p, "
+                       "upstream-comp-name=\"%s\", graph-addr=%p",
+                       connection, connection->upstream_port,
+                       bt_port_get_name(connection->upstream_port),
+                       upstream_component,
+                       bt_component_get_name(upstream_component),
+                       bt_connection_borrow_graph(connection));
+               status = BT_CONNECTION_STATUS_GRAPH_IS_CANCELED;
+               goto end;
+       }
+
+       if (bt_connection_is_ended(connection)) {
                BT_LOGW("Invalid parameter: connection is ended: "
                        "conn-addr=%p", connection);
-               goto error;
+               status = BT_CONNECTION_STATUS_IS_ENDED;
+               goto end;
+       }
+
+       if (!notification_types) {
+               BT_LOGD_STR("No notification types: subscribing to all notifications.");
+               notification_types = all_notif_types;
        }
 
        upstream_port = connection->upstream_port;
@@ -320,20 +344,13 @@ bt_private_connection_create_notification_iterator(
                upstream_component, bt_component_get_name(upstream_component));
        upstream_comp_class_type =
                bt_component_get_class_type(upstream_component);
-       if (upstream_comp_class_type != BT_COMPONENT_CLASS_TYPE_SOURCE &&
-                       upstream_comp_class_type != BT_COMPONENT_CLASS_TYPE_FILTER) {
-               /* Unsupported operation. */
-               BT_LOGW("Upstream component's class is not a source or filter component class: "
-                       "comp-class-type=%s",
-                       bt_component_class_type_string(upstream_comp_class_type));
-               goto error;
-       }
-
-       iterator = bt_notification_iterator_create(upstream_component,
-               upstream_port, notification_types, connection);
-       if (!iterator) {
+       assert(upstream_comp_class_type == BT_COMPONENT_CLASS_TYPE_SOURCE ||
+                       upstream_comp_class_type == BT_COMPONENT_CLASS_TYPE_FILTER);
+       status = bt_notification_iterator_create(upstream_component,
+               upstream_port, notification_types, connection, &iterator);
+       if (status != BT_CONNECTION_STATUS_OK) {
                BT_LOGW("Cannot create notification iterator from connection.");
-               goto error;
+               goto end;
        }
 
        switch (upstream_comp_class_type) {
@@ -355,22 +372,26 @@ bt_private_connection_create_notification_iterator(
        }
        default:
                /* Unreachable. */
+               BT_LOGF("Unknown component class type: type=%d",
+                       upstream_comp_class_type);
                abort();
        }
 
        if (init_method) {
-               enum bt_notification_iterator_status status;
+               enum bt_notification_iterator_status iter_status;
 
                BT_LOGD("Calling user's initialization method: iter-addr=%p",
                        iterator);
-               status = init_method(
+               iter_status = init_method(
                        bt_private_notification_iterator_from_notification_iterator(iterator),
                        bt_private_port_from_port(upstream_port));
                BT_LOGD("User method returned: status=%s",
-                       bt_notification_iterator_status_string(status));
-               if (status < 0) {
+                       bt_notification_iterator_status_string(iter_status));
+               if (iter_status != BT_NOTIFICATION_ITERATOR_STATUS_OK) {
                        BT_LOGW_STR("Initialization method failed.");
-                       goto error;
+                       status = bt_connection_status_from_notification_iterator_status(
+                               iter_status);
+                       goto end;
                }
        }
 
@@ -383,14 +404,15 @@ bt_private_connection_create_notification_iterator(
                bt_port_get_name(connection->upstream_port),
                upstream_component, bt_component_get_name(upstream_component),
                iterator);
-       goto end;
 
-error:
-       BT_PUT(iterator);
+       /* Move reference to user */
+       *user_iterator = iterator;
+       iterator = NULL;
 
 end:
        bt_put(upstream_component);
-       return iterator;
+       bt_put(iterator);
+       return status;
 }
 
 BT_HIDDEN
This page took 0.026527 seconds and 4 git commands to generate.