* SOFTWARE.
*/
+#include <babeltrace/component/notification/iterator-internal.h>
+#include <babeltrace/component/component-internal.h>
+#include <babeltrace/component/component-source-internal.h>
+#include <babeltrace/component/component-filter-internal.h>
#include <babeltrace/component/connection-internal.h>
+#include <babeltrace/component/private-connection.h>
#include <babeltrace/component/graph-internal.h>
#include <babeltrace/component/port-internal.h>
-#include <babeltrace/component/component-source-internal.h>
-#include <babeltrace/component/component-filter-internal.h>
#include <babeltrace/object-internal.h>
#include <babeltrace/compiler.h>
#include <glib.h>
g_free(connection);
}
+struct bt_connection *bt_connection_from_private_connection(
+ struct bt_private_connection *private_connection)
+{
+ return bt_get(bt_connection_from_private(private_connection));
+}
+
BT_HIDDEN
struct bt_connection *bt_connection_create(
struct bt_graph *graph,
- struct bt_port *upstream, struct bt_port *downstream)
+ struct bt_port *upstream_port,
+ struct bt_port *downstream_port)
{
struct bt_connection *connection = NULL;
- if (bt_port_get_type(upstream) != BT_PORT_TYPE_OUTPUT) {
+ if (bt_port_get_type(upstream_port) != BT_PORT_TYPE_OUTPUT) {
goto end;
}
- if (bt_port_get_type(downstream) != BT_PORT_TYPE_INPUT) {
+ if (bt_port_get_type(downstream_port) != BT_PORT_TYPE_INPUT) {
goto end;
}
bt_object_init(connection, bt_connection_destroy);
/* Weak references are taken, see comment in header. */
- connection->output_port = upstream;
- connection->input_port = downstream;
- bt_port_add_connection(upstream, connection);
- bt_port_add_connection(downstream, connection);
+ connection->upstream_port = upstream_port;
+ connection->downstream_port = downstream_port;
+ bt_port_set_connection(upstream_port, connection);
+ bt_port_set_connection(downstream_port, connection);
bt_object_set_parent(connection, &graph->base);
end:
return connection;
}
-struct bt_port *bt_connection_get_input_port(
+BT_HIDDEN
+void bt_connection_disconnect_ports(struct bt_connection *conn,
+ struct bt_component *acting_comp)
+{
+ 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;
+
+ if (downstream_port) {
+ downstream_comp = bt_port_get_component(downstream_port);
+ bt_port_set_connection(downstream_port, NULL);
+ conn->downstream_port = NULL;
+ }
+
+ if (upstream_port) {
+ upstream_comp = bt_port_get_component(upstream_port);
+ bt_port_set_connection(upstream_port, NULL);
+ conn->upstream_port = NULL;
+ }
+
+ if (downstream_comp && downstream_comp != acting_comp) {
+ bt_component_port_disconnected(downstream_comp,
+ downstream_port);
+ }
+
+ if (upstream_comp && upstream_comp != acting_comp) {
+ bt_component_port_disconnected(upstream_comp, upstream_port);
+ }
+
+ if (upstream_comp) {
+ struct bt_graph *graph = bt_component_get_graph(upstream_comp);
+
+ assert(graph);
+ bt_graph_notify_port_disconnected(graph, upstream_comp,
+ upstream_port);
+ bt_put(graph);
+ }
+
+ if (downstream_comp) {
+ struct bt_graph *graph =
+ bt_component_get_graph(downstream_comp);
+
+ assert(graph);
+ bt_graph_notify_port_disconnected(graph, downstream_comp,
+ downstream_port);
+ bt_put(graph);
+ }
+
+ bt_put(downstream_comp);
+ bt_put(upstream_comp);
+}
+
+struct bt_port *bt_connection_get_upstream_port(
struct bt_connection *connection)
{
- return connection ? connection->input_port : NULL;
+ return connection ? bt_get(connection->upstream_port) : NULL;
}
-struct bt_port *bt_connection_get_output_port(
+struct bt_port *bt_connection_get_downstream_port(
struct bt_connection *connection)
{
- return connection ? connection->output_port : NULL;
+ return connection ? bt_get(connection->downstream_port) : NULL;
}
struct bt_notification_iterator *
-bt_connection_create_notification_iterator(struct bt_connection *connection)
+bt_private_connection_create_notification_iterator(
+ struct bt_private_connection *private_connection)
{
+ enum bt_notification_iterator_status ret_iterator;
+ enum bt_component_class_type upstream_comp_class_type;
+ struct bt_notification_iterator *iterator = NULL;
+ struct bt_port *upstream_port = NULL;
struct bt_component *upstream_component = NULL;
- struct bt_notification_iterator *it = NULL;
+ struct bt_component_class *upstream_comp_class = NULL;
+ struct bt_connection *connection = NULL;
+ bt_component_class_notification_iterator_init_method init_method = NULL;
- if (!connection) {
- goto end;
+ if (!private_connection) {
+ goto error;
+ }
+
+ connection = bt_connection_from_private(private_connection);
+
+ if (!connection->upstream_port || !connection->downstream_port) {
+ goto error;
}
- upstream_component = bt_port_get_component(connection->output_port);
+ upstream_port = connection->upstream_port;
+ assert(upstream_port);
+ upstream_component = bt_port_get_component(upstream_port);
assert(upstream_component);
+ upstream_comp_class = upstream_component->class;
+
+ if (!upstream_component) {
+ goto error;
+ }
+
+ 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. */
+ goto error;
+ }
- switch (bt_component_get_class_type(upstream_component)) {
+ iterator = bt_notification_iterator_create(upstream_component);
+ if (!iterator) {
+ goto error;
+ }
+
+ switch (upstream_comp_class_type) {
case BT_COMPONENT_CLASS_TYPE_SOURCE:
- it = bt_component_source_create_notification_iterator(
- upstream_component);
+ {
+ struct bt_component_class_source *source_class =
+ container_of(upstream_comp_class,
+ struct bt_component_class_source, parent);
+ init_method = source_class->methods.iterator.init;
break;
+ }
case BT_COMPONENT_CLASS_TYPE_FILTER:
- it = bt_component_filter_create_notification_iterator(
- upstream_component);
+ {
+ struct bt_component_class_filter *filter_class =
+ container_of(upstream_comp_class,
+ struct bt_component_class_filter, parent);
+ init_method = filter_class->methods.iterator.init;
break;
+ }
default:
- goto end;
+ /* Unreachable. */
+ assert(0);
+ }
+
+ if (init_method) {
+ enum bt_notification_iterator_status status = init_method(
+ bt_private_component_from_component(upstream_component),
+ bt_private_port_from_port(upstream_port),
+ bt_private_notification_iterator_from_notification_iterator(iterator));
+ if (status < 0) {
+ goto error;
+ }
+ }
+
+ ret_iterator = bt_notification_iterator_validate(iterator);
+ if (ret_iterator != BT_NOTIFICATION_ITERATOR_STATUS_OK) {
+ goto error;
}
+
+ goto end;
+
+error:
+ BT_PUT(iterator);
+
end:
bt_put(upstream_component);
- return it;
+ return iterator;
}