lib/graph/connection.c

   1 /*
   2  * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
   3  * Copyright 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a copy
   6  * of this software and associated documentation files (the "Software"), to deal
   7  * in the Software without restriction, including without limitation the rights
   8  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   9  * copies of the Software, and to permit persons to whom the Software is
  10  * furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in
  13  * all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21  * SOFTWARE.
  22  */
  23
  24 #define BT_LOG_TAG "CONNECTION"
  25 #include <babeltrace/lib-logging-internal.h>
  26
  27 #include <babeltrace/assert-internal.h>
  28 #include <babeltrace/assert-pre-internal.h>
  29 #include <babeltrace/graph/message-iterator-internal.h>
  30 #include <babeltrace/graph/component-internal.h>
  31 #include <babeltrace/graph/connection-internal.h>
  32 #include <babeltrace/graph/connection-const.h>
  33 #include <babeltrace/graph/graph-internal.h>
  34 #include <babeltrace/graph/port-internal.h>
  35 #include <babeltrace/object-internal.h>
  36 #include <babeltrace/compiler-internal.h>
  37 #include <stdlib.h>
  38 #include <glib.h>
  39
  40 static
  41 void destroy_connection(struct bt_object *obj)
  42 {
  43         struct bt_connection *connection = container_of(obj,
  44                         struct bt_connection, base);
  45
  46         BT_LIB_LOGD("Destroying connection: %!+x", connection);
  47
  48         /*
  49          * Make sure that each message iterator which was created
  50          * for this connection is finalized before we destroy it. Once a
  51          * message iterator is finalized, all its method return
  52          * NULL or the BT_MESSAGE_ITERATOR_STATUS_CANCELED status.
  53          *
  54          * Because connections are destroyed before components within a
  55          * graph, this ensures that message iterators are always
  56          * finalized before their upstream component.
  57          *
  58          * Ending the connection does exactly this. We pass `false` to
  59          * bt_connection_end() here to avoid removing this connection
  60          * from the graph: if we're here, we're already in the graph's
  61          * destructor.
  62          */
  63         bt_connection_end(connection, false);
  64         g_ptr_array_free(connection->iterators, TRUE);
  65         connection->iterators = NULL;
  66
  67         /*
  68          * No bt_object_put_ref on ports as a connection only holds _weak_
  69          * references to them.
  70          */
  71         g_free(connection);
  72 }
  73
  74 static
  75 void try_remove_connection_from_graph(struct bt_connection *connection)
  76 {
  77         void *graph = (void *) bt_object_borrow_parent(&connection->base);
  78
  79         if (connection->base.ref_count > 0 ||
  80                         connection->downstream_port ||
  81                         connection->upstream_port ||
  82                         connection->iterators->len > 0) {
  83                 return;
  84         }
  85
  86         /*
  87          * At this point we know that:
  88          *
  89          * 1. The connection is ended (ports were disconnected).
  90          * 2. All the message iterators that this connection
  91          *    created, if any, are finalized.
  92          * 3. The connection's reference count is 0, so only the
  93          *    parent (graph) owns this connection after this call.
  94          *
  95          * In other words, no other object than the graph knows this
  96          * connection.
  97          *
  98          * It is safe to remove the connection from the graph, therefore
  99          * destroying it.
 100          */
 101         BT_LIB_LOGD("Removing self from graph's connections: "
 102                 "%![graph-]+g, %![conn-]+x", graph, connection);
 103         bt_graph_remove_connection(graph, connection);
 104 }
 105
 106 static
 107 void parent_is_owner(struct bt_object *obj)
 108 {
 109         struct bt_connection *connection = container_of(obj,
 110                         struct bt_connection, base);
 111
 112         try_remove_connection_from_graph(connection);
 113 }
 114
 115 BT_HIDDEN
 116 struct bt_connection *bt_connection_create(struct bt_graph *graph,
 117                 struct bt_port *upstream_port,
 118                 struct bt_port *downstream_port)
 119 {
 120         struct bt_connection *connection = NULL;
 121
 122         BT_LIB_LOGD("Creating connection: "
 123                 "%![graph-]+g, %![up-port-]+p, %![down-port-]+p",
 124                 graph, upstream_port, downstream_port);
 125         connection = g_new0(struct bt_connection, 1);
 126         if (!connection) {
 127                 BT_LOGE_STR("Failed to allocate one connection.");
 128                 goto end;
 129         }
 130
 131         bt_object_init_shared_with_parent(&connection->base,
 132                 destroy_connection);
 133         bt_object_set_parent_is_owner_listener_func(&connection->base,
 134                 parent_is_owner);
 135         connection->iterators = g_ptr_array_new();
 136         if (!connection->iterators) {
 137                 BT_LOGE_STR("Failed to allocate a GPtrArray.");
 138                 BT_OBJECT_PUT_REF_AND_RESET(connection);
 139                 goto end;
 140         }
 141
 142         /* Weak references are taken, see comment in header. */
 143         connection->upstream_port = upstream_port;
 144         connection->downstream_port = downstream_port;
 145         BT_LIB_LOGD("Setting upstream port's connection: %!+p", upstream_port);
 146         bt_port_set_connection(upstream_port, connection);
 147         BT_LIB_LOGD("Setting downstream port's connection: %!+p",
 148                 downstream_port);
 149         bt_port_set_connection(downstream_port, connection);
 150         bt_object_set_parent(&connection->base, &graph->base);
 151         BT_LIB_LOGD("Created connection: %!+x", connection);
 152
 153 end:
 154         return connection;
 155 }
 156
 157 BT_HIDDEN
 158 void bt_connection_end(struct bt_connection *conn, bool try_remove_from_graph)
 159 {
 160         struct bt_component *downstream_comp = NULL;
 161         struct bt_component *upstream_comp = NULL;
 162         struct bt_port *downstream_port = conn->downstream_port;
 163         struct bt_port *upstream_port = conn->upstream_port;
 164         struct bt_graph *graph = bt_connection_borrow_graph(conn);
 165         size_t i;
 166
 167         BT_LIB_LOGD("Ending connection: %!+x, try-remove-from-graph=%d",
 168                 conn, try_remove_from_graph);
 169
 170         /*
 171          * Any of the following message callback functions could
 172          * remove one of the connection's ports from its component. To
 173          * make sure that at least logging in called functions works
 174          * with existing objects, get a local reference on both ports.
 175          */
 176         bt_object_get_ref(downstream_port);
 177         bt_object_get_ref(upstream_port);
 178
 179         if (downstream_port) {
 180                 BT_LIB_LOGD("Disconnecting connection's downstream port: %!+p",
 181                         downstream_port);
 182                 downstream_comp = bt_port_borrow_component_inline(
 183                         downstream_port);
 184                 bt_port_set_connection(downstream_port, NULL);
 185                 conn->downstream_port = NULL;
 186         }
 187
 188         if (upstream_port) {
 189                 BT_LIB_LOGD("Disconnecting connection's upstream port: %!+p",
 190                         upstream_port);
 191                 upstream_comp = bt_port_borrow_component_inline(
 192                         upstream_port);
 193                 bt_port_set_connection(upstream_port, NULL);
 194                 conn->upstream_port = NULL;
 195         }
 196
 197         if (downstream_comp && conn->msgied_downstream_port_connected &&
 198                         !conn->msgied_downstream_port_disconnected) {
 199                 /* bt_component_port_disconnected() logs details */
 200                 bt_component_port_disconnected(downstream_comp,
 201                         downstream_port);
 202                 conn->msgied_downstream_port_disconnected = true;
 203         }
 204
 205         if (upstream_comp && conn->msgied_upstream_port_connected &&
 206                         !conn->msgied_upstream_port_disconnected) {
 207                 /* bt_component_port_disconnected() logs details */
 208                 bt_component_port_disconnected(upstream_comp, upstream_port);
 209                 conn->msgied_upstream_port_disconnected = true;
 210         }
 211
 212         BT_ASSERT(graph);
 213
 214         if (conn->msgied_graph_ports_connected &&
 215                         !conn->msgied_graph_ports_disconnected) {
 216                 /* bt_graph_notify_ports_disconnected() logs details */
 217                 bt_graph_notify_ports_disconnected(graph, upstream_comp,
 218                         downstream_comp, upstream_port, downstream_port);
 219                 conn->msgied_graph_ports_disconnected = true;
 220         }
 221
 222         /*
 223          * It is safe to put the local port references now that we don't
 224          * need them anymore. This could indeed destroy them.
 225          */
 226         bt_object_put_ref(downstream_port);
 227         bt_object_put_ref(upstream_port);
 228
 229         /*
 230          * Because this connection is ended, finalize (cancel) each
 231          * message iterator created from it.
 232          */
 233         for (i = 0; i < conn->iterators->len; i++) {
 234                 struct bt_self_component_port_input_message_iterator *iterator =
 235                         g_ptr_array_index(conn->iterators, i);
 236
 237                 BT_LIB_LOGD("Finalizing message iterator created by "
 238                         "this ended connection: %![iter-]+i", iterator);
 239                 bt_self_component_port_input_message_iterator_finalize(
 240                         iterator);
 241
 242                 /*
 243                  * Make sure this iterator does not try to remove itself
 244                  * from this connection's iterators on destruction
 245                  * because this connection won't exist anymore.
 246                  */
 247                 bt_self_component_port_input_message_iterator_set_connection(
 248                         iterator, NULL);
 249         }
 250
 251         g_ptr_array_set_size(conn->iterators, 0);
 252
 253         if (try_remove_from_graph) {
 254                 try_remove_connection_from_graph(conn);
 255         }
 256 }
 257
 258 const struct bt_port_output *bt_connection_borrow_upstream_port_const(
 259                 const struct bt_connection *connection)
 260 {
 261         BT_ASSERT_PRE_NON_NULL(connection, "Connection");
 262         return (void *) connection->upstream_port;
 263 }
 264
 265 const struct bt_port_input *bt_connection_borrow_downstream_port_const(
 266                 const struct bt_connection *connection)
 267 {
 268         BT_ASSERT_PRE_NON_NULL(connection, "Connection");
 269         return (void *) connection->downstream_port;
 270 }
 271
 272 BT_HIDDEN
 273 void bt_connection_remove_iterator(struct bt_connection *conn,
 274                 struct bt_self_component_port_input_message_iterator *iterator)
 275 {
 276         g_ptr_array_remove(conn->iterators, iterator);
 277         BT_LIB_LOGV("Removed message iterator from connection: "
 278                 "%![conn-]+x, %![iter-]+i", conn, iterator);
 279         try_remove_connection_from_graph(conn);
 280 }
 281
 282 bt_bool bt_connection_is_ended(const struct bt_connection *connection)
 283 {
 284         return !connection->downstream_port && !connection->upstream_port;
 285 }
 286
 287 void bt_connection_get_ref(const struct bt_connection *connection)
 288 {
 289         bt_object_get_ref(connection);
 290 }
 291
 292 void bt_connection_put_ref(const struct bt_connection *connection)
 293 {
 294         bt_object_put_ref(connection);
 295 }