2 * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
3 * Copyright 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
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:
12 * The above copyright notice and this permission notice shall be included in
13 * all copies or substantial portions of the Software.
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
24 #define BT_LOG_TAG "CONNECTION"
25 #include <babeltrace/lib-logging-internal.h>
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>
41 void destroy_connection(struct bt_object
*obj
)
43 struct bt_connection
*connection
= container_of(obj
,
44 struct bt_connection
, base
);
46 BT_LIB_LOGD("Destroying connection: %!+x", connection
);
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.
54 * Because connections are destroyed before components within a
55 * graph, this ensures that message iterators are always
56 * finalized before their upstream component.
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
63 bt_connection_end(connection
, false);
64 g_ptr_array_free(connection
->iterators
, TRUE
);
65 connection
->iterators
= NULL
;
68 * No bt_object_put_ref on ports as a connection only holds _weak_
75 void try_remove_connection_from_graph(struct bt_connection
*connection
)
77 void *graph
= (void *) bt_object_borrow_parent(&connection
->base
);
79 if (connection
->base
.ref_count
> 0 ||
80 connection
->downstream_port
||
81 connection
->upstream_port
||
82 connection
->iterators
->len
> 0) {
87 * At this point we know that:
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.
95 * In other words, no other object than the graph knows this
98 * It is safe to remove the connection from the graph, therefore
101 BT_LIB_LOGD("Removing self from graph's connections: "
102 "%![graph-]+g, %![conn-]+x", graph
, connection
);
103 bt_graph_remove_connection(graph
, connection
);
107 void parent_is_owner(struct bt_object
*obj
)
109 struct bt_connection
*connection
= container_of(obj
,
110 struct bt_connection
, base
);
112 try_remove_connection_from_graph(connection
);
116 struct bt_connection
*bt_connection_create(struct bt_graph
*graph
,
117 struct bt_port
*upstream_port
,
118 struct bt_port
*downstream_port
)
120 struct bt_connection
*connection
= NULL
;
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);
127 BT_LOGE_STR("Failed to allocate one connection.");
131 bt_object_init_shared_with_parent(&connection
->base
,
133 bt_object_set_parent_is_owner_listener_func(&connection
->base
,
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
);
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",
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
);
158 void bt_connection_end(struct bt_connection
*conn
, bool try_remove_from_graph
)
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
);
167 BT_LIB_LOGD("Ending connection: %!+x, try-remove-from-graph=%d",
168 conn
, try_remove_from_graph
);
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.
176 bt_object_get_ref(downstream_port
);
177 bt_object_get_ref(upstream_port
);
179 if (downstream_port
) {
180 BT_LIB_LOGD("Disconnecting connection's downstream port: %!+p",
182 downstream_comp
= bt_port_borrow_component_inline(
184 bt_port_set_connection(downstream_port
, NULL
);
185 conn
->downstream_port
= NULL
;
189 BT_LIB_LOGD("Disconnecting connection's upstream port: %!+p",
191 upstream_comp
= bt_port_borrow_component_inline(
193 bt_port_set_connection(upstream_port
, NULL
);
194 conn
->upstream_port
= NULL
;
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
,
202 conn
->msgied_downstream_port_disconnected
= true;
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;
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;
223 * It is safe to put the local port references now that we don't
224 * need them anymore. This could indeed destroy them.
226 bt_object_put_ref(downstream_port
);
227 bt_object_put_ref(upstream_port
);
230 * Because this connection is ended, finalize (cancel) each
231 * message iterator created from it.
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
);
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(
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.
247 bt_self_component_port_input_message_iterator_set_connection(
251 g_ptr_array_set_size(conn
->iterators
, 0);
253 if (try_remove_from_graph
) {
254 try_remove_connection_from_graph(conn
);
258 const struct bt_port_output
*bt_connection_borrow_upstream_port_const(
259 const struct bt_connection
*connection
)
261 BT_ASSERT_PRE_NON_NULL(connection
, "Connection");
262 return (void *) connection
->upstream_port
;
265 const struct bt_port_input
*bt_connection_borrow_downstream_port_const(
266 const struct bt_connection
*connection
)
268 BT_ASSERT_PRE_NON_NULL(connection
, "Connection");
269 return (void *) connection
->downstream_port
;
273 void bt_connection_remove_iterator(struct bt_connection
*conn
,
274 struct bt_self_component_port_input_message_iterator
*iterator
)
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
);
282 bt_bool
bt_connection_is_ended(const struct bt_connection
*connection
)
284 return !connection
->downstream_port
&& !connection
->upstream_port
;
287 void bt_connection_get_ref(const struct bt_connection
*connection
)
289 bt_object_get_ref(connection
);
292 void bt_connection_put_ref(const struct bt_connection
*connection
)
294 bt_object_put_ref(connection
);