End connection on destruction

[babeltrace.git] / include / babeltrace / graph / connection-internal.h

diff --git a/include/babeltrace/graph/connection-internal.h b/include/babeltrace/graph/connection-internal.h
index 13e7d9ee65ad95aa0460819131a9e0dfb3b83347..6044f0a5934dd4c0a2fdf81d6d884b5a5d595fa6 100644 (file)
--- a/include/babeltrace/graph/connection-internal.h
+++ b/include/babeltrace/graph/connection-internal.h
@@ -31,6 +31,7 @@
 #include <babeltrace/graph/notification-iterator.h>
 #include <babeltrace/graph/private-connection.h>
 #include <babeltrace/object-internal.h>
+#include <stdbool.h>
 
 struct bt_graph;
 
@@ -77,12 +78,19 @@ struct bt_connection *bt_connection_create(struct bt_graph *graph,
                struct bt_port *downstream_port);
 
 BT_HIDDEN
-void bt_connection_disconnect_ports(struct bt_connection *conn);
+void bt_connection_end(struct bt_connection *conn, bool try_remove_from_graph);
 
 BT_HIDDEN
 void bt_connection_remove_iterator(struct bt_connection *conn,
                struct bt_notification_iterator *iterator);
 
+static inline
+struct bt_graph *bt_connection_borrow_graph(struct bt_connection *conn)
+{
+       assert(conn);
+       return (void *) conn->base.parent;
+}
+
 static inline
 enum bt_connection_status
 bt_connection_status_from_notification_iterator_status(