[babeltrace.git] / lib / graph / connection.c

/*
 * connection.c
 *
 * Babeltrace Connection
 *
 * Copyright 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#define BT_LOG_TAG "CONNECTION"
#include <babeltrace/lib-logging-internal.h>

#include <babeltrace/graph/notification-iterator-internal.h>
#include <babeltrace/graph/component-internal.h>
#include <babeltrace/graph/component-source-internal.h>
#include <babeltrace/graph/component-filter-internal.h>
#include <babeltrace/graph/connection-internal.h>
#include <babeltrace/graph/private-connection.h>
#include <babeltrace/graph/graph-internal.h>
#include <babeltrace/graph/port-internal.h>
#include <babeltrace/object-internal.h>
#include <babeltrace/compiler-internal.h>
#include <stdlib.h>
#include <glib.h>

static
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);

	/*
	 * Make sure that each notification iterator which was created
	 * for this connection is finalized before we destroy it. Once a
	 * notification iterator is finalized, all its method return
	 * NULL or the BT_NOTIFICATION_ITERATOR_STATUS_CANCELED status.
	 *
	 * Because connections are destroyed before components within a
	 * graph, this ensures that notification iterators are always
	 * finalized before their upstream component.
	 */
	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);
	}

	/*
	 * No bt_put on ports as a connection only holds _weak_ references
	 * to them.
	 */
	g_free(connection);
}

static
void bt_connection_try_remove_from_graph(struct bt_connection *connection)
{
	void *graph = bt_object_borrow_parent(&connection->base);

	if (connection->base.ref_count.count > 0 ||
			connection->downstream_port ||
			connection->upstream_port ||
			connection->iterators->len > 0) {
		return;
	}

	/*
	 * At this point we know that:
	 *
	 * 1. The connection is ended (ports were disconnected).
	 * 2. All the notification iterators that this connection
	 *    created, if any, are finalized.
	 * 3. The connection's reference count is 0, so only the
	 *    parent (graph) owns this connection after this call.
	 *
	 * In other words, no other object than the graph knows this
	 * connection.
	 *
	 * It is safe to remove the connection from the graph, therefore
	 * destroying it.
	 */
	BT_LOGD("Removing self from graph's connections: "
		"graph-addr=%p, conn-addr=%p", graph, connection);
	bt_graph_remove_connection(graph, connection);
}

static
void bt_connection_parent_is_owner(struct bt_object *obj)
{
	struct bt_connection *connection = container_of(obj,
			struct bt_connection, base);

	bt_connection_try_remove_from_graph(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_port,
		struct bt_port *downstream_port)
{
	struct bt_connection *connection = NULL;

	if (bt_port_get_type(upstream_port) != BT_PORT_TYPE_OUTPUT) {
		BT_LOGW_STR("Invalid parameter: upstream port is not an output port.");
		goto end;
	}
	if (bt_port_get_type(downstream_port) != BT_PORT_TYPE_INPUT) {
		BT_LOGW_STR("Invalid parameter: downstream port is not an input port.");
		goto end;
	}

	BT_LOGD("Creating connection: "
		"graph-addr=%p, upstream-port-addr=%p, uptream-port-name=\"%s\", "
		"downstream-port-addr=%p, downstream-port-name=\"%s\"",
		graph, upstream_port, bt_port_get_name(upstream_port),
		downstream_port, bt_port_get_name(downstream_port));
	connection = g_new0(struct bt_connection, 1);
	if (!connection) {
		BT_LOGE_STR("Failed to allocate one connection.");
		goto end;
	}

	bt_object_init(connection, bt_connection_destroy);
	bt_object_set_parent_is_owner_listener(connection,
		bt_connection_parent_is_owner);
	connection->iterators = g_ptr_array_new();
	if (!connection->iterators) {
		BT_LOGE_STR("Failed to allocate a GPtrArray.");
		BT_PUT(connection);
		goto end;
	}

	/* Weak references are taken, see comment in header. */
	connection->upstream_port = upstream_port;
	connection->downstream_port = downstream_port;
	BT_LOGD_STR("Setting upstream port's connection.");
	bt_port_set_connection(upstream_port, connection);
	BT_LOGD_STR("Setting downstream port's connection.");
	bt_port_set_connection(downstream_port, connection);
	bt_object_set_parent(connection, &graph->base);
	BT_LOGD("Created connection: "
		"graph-addr=%p, upstream-port-addr=%p, uptream-port-name=\"%s\", "
		"downstream-port-addr=%p, downstream-port-name=\"%s\", "
		"conn-addr=%p",
		graph, upstream_port, bt_port_get_name(upstream_port),
		downstream_port, bt_port_get_name(downstream_port),
		connection);

end:
	return connection;
}

BT_HIDDEN
void bt_connection_disconnect_ports(struct bt_connection *conn)
{
	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);
	size_t i;

	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) {
		/* bt_component_port_disconnected() logs details */
		bt_component_port_disconnected(downstream_comp,
			downstream_port);
	}

	if (upstream_comp) {
		/* bt_component_port_disconnected() logs details */
		bt_component_port_disconnected(upstream_comp, upstream_port);
	}

	assert(graph);
	/* bt_graph_notify_ports_disconnected() logs details */
	bt_graph_notify_ports_disconnected(graph, upstream_comp,
		downstream_comp, upstream_port, downstream_port);
	bt_put(downstream_comp);
	bt_put(upstream_comp);

	/*
	 * Because this connection is ended, finalize (cancel) each
	 * notification iterator created from it.
	 */
	for (i = 0; i < conn->iterators->len; i++) {
		struct bt_notification_iterator *iterator =
			g_ptr_array_index(conn->iterators, i);

		BT_LOGD("Finalizing notification iterator created by this ended connection: "
			"conn-addr=%p, iter-addr=%p", conn, 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_set_size(conn->iterators, 0);
	bt_connection_try_remove_from_graph(conn);
}

struct bt_port *bt_connection_get_upstream_port(
		struct bt_connection *connection)
{
	return connection ? bt_get(connection->upstream_port) : NULL;
}

struct bt_port *bt_connection_get_downstream_port(
		struct bt_connection *connection)
{
	return connection ? bt_get(connection->downstream_port) : NULL;
}

struct bt_notification_iterator *
bt_private_connection_create_notification_iterator(
		struct bt_private_connection *private_connection,
		const enum bt_notification_type *notification_types)
{
	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_component_class *upstream_comp_class = NULL;
	struct bt_connection *connection = NULL;
	bt_component_class_notification_iterator_init_method init_method = NULL;
	static const enum bt_notification_type all_notif_types[] = {
		BT_NOTIFICATION_TYPE_ALL,
		BT_NOTIFICATION_TYPE_SENTINEL,
	};

	if (!private_connection) {
		BT_LOGW_STR("Invalid parameter: private connection is NULL.");
		goto error;
	}

	if (!notification_types) {
		BT_LOGD_STR("No notification types: subscribing to all notifications.");
		notification_types = all_notif_types;
	}

	connection = bt_connection_from_private(private_connection);
	if (!connection->upstream_port || !connection->downstream_port) {
		BT_LOGW("Invalid parameter: connection is ended: "
			"conn-addr=%p", connection);
		goto error;
	}

	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;
	BT_LOGD("Creating notification iterator from connection: "
		"conn-addr=%p, upstream-port-addr=%p, "
		"upstream-port-name=\"%s\", upstream-comp-addr=%p, "
		"upstream-comp-name=\"%s\"",
		connection, connection->upstream_port,
		bt_port_get_name(connection->upstream_port),
		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) {
		BT_LOGW("Cannot create notification iterator from connection.");
		goto error;
	}

	switch (upstream_comp_class_type) {
	case BT_COMPONENT_CLASS_TYPE_SOURCE:
	{
		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:
	{
		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:
		/* Unreachable. */
		abort();
	}

	if (init_method) {
		enum bt_notification_iterator_status status;

		BT_LOGD("Calling user's initialization method: iter-addr=%p",
			iterator);
		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_LOGW_STR("Initialization method failed.");
			goto error;
		}
	}

	g_ptr_array_add(connection->iterators, iterator);
	BT_LOGD("Created notification iterator from connection: "
		"conn-addr=%p, upstream-port-addr=%p, "
		"upstream-port-name=\"%s\", upstream-comp-addr=%p, "
		"upstream-comp-name=\"%s\", iter-addr=%p",
		connection, connection->upstream_port,
		bt_port_get_name(connection->upstream_port),
		upstream_component, bt_component_get_name(upstream_component),
		iterator);
	goto end;

error:
	BT_PUT(iterator);

end:
	bt_put(upstream_component);
	return iterator;
}

BT_HIDDEN
void bt_connection_remove_iterator(struct bt_connection *conn,
		struct bt_notification_iterator *iterator)
{
	g_ptr_array_remove(conn->iterators, iterator);
	BT_LOGV("Removed notification iterator from connection: "
		"conn-addr=%p, iter-addr=%p", conn, iterator);
	bt_connection_try_remove_from_graph(conn);
}
Commit	Line	Data
784cdc68	1	/*
7d55361f	2	* connection.c
784cdc68 JG	3	*
	4	* Babeltrace Connection
	5	*
	6	* Copyright 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
	7	*
	8	* Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
	9	*
	10	* Permission is hereby granted, free of charge, to any person obtaining a copy
	11	* of this software and associated documentation files (the "Software"), to deal
	12	* in the Software without restriction, including without limitation the rights
	13	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	14	* copies of the Software, and to permit persons to whom the Software is
	15	* furnished to do so, subject to the following conditions:
	16	*
	17	* The above copyright notice and this permission notice shall be included in
	18	* all copies or substantial portions of the Software.
	19	*
	20	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	21	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	22	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
	23	* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	24	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	25	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
	26	* SOFTWARE.
	27	*/
	28
a36bfb16 PP	29	#define BT_LOG_TAG "CONNECTION"
	30	#include <babeltrace/lib-logging-internal.h>
	31
b2e0c907 PP	32	#include <babeltrace/graph/notification-iterator-internal.h>
	33	#include <babeltrace/graph/component-internal.h>
	34	#include <babeltrace/graph/component-source-internal.h>
	35	#include <babeltrace/graph/component-filter-internal.h>
	36	#include <babeltrace/graph/connection-internal.h>
	37	#include <babeltrace/graph/private-connection.h>
	38	#include <babeltrace/graph/graph-internal.h>
	39	#include <babeltrace/graph/port-internal.h>
784cdc68	40	#include <babeltrace/object-internal.h>
3d9990ac	41	#include <babeltrace/compiler-internal.h>
0fbb9a9f	42	#include <stdlib.h>
784cdc68 JG	43	#include <glib.h>
	44
	45	static
	46	void bt_connection_destroy(struct bt_object *obj)
	47	{
	48	struct bt_connection *connection = container_of(obj,
	49	struct bt_connection, base);
bd14d768 PP	50	size_t i;
bd14d768 PP	51
a36bfb16 PP	52	BT_LOGD("Destroying connection: addr=%p", connection);
a36bfb16 PP	53
bd14d768 PP	54	/*
	55	* Make sure that each notification iterator which was created
	56	* for this connection is finalized before we destroy it. Once a
	57	* notification iterator is finalized, all its method return
	58	* NULL or the BT_NOTIFICATION_ITERATOR_STATUS_CANCELED status.
	59	*
	60	* Because connections are destroyed before components within a
	61	* graph, this ensures that notification iterators are always
	62	* finalized before their upstream component.
	63	*/
	64	if (connection->iterators) {
	65	for (i = 0; i < connection->iterators->len; i++) {
	66	struct bt_notification_iterator *iterator =
	67	g_ptr_array_index(connection->iterators, i);
	68
a36bfb16 PP	69	BT_LOGD("Finalizing notification iterator created by this connection: "
a36bfb16 PP	70	"iter-addr=%p", iterator);
bd14d768 PP	71	bt_notification_iterator_finalize(iterator);
	72
	73	/*
	74	* Make sure this iterator does not try to
	75	* remove itself from this connection's
	76	* iterators on destruction because this
	77	* connection won't exist anymore.
	78	*/
	79	bt_notification_iterator_set_connection(iterator,
	80	NULL);
	81	}
	82
	83	g_ptr_array_free(connection->iterators, TRUE);
	84	}
784cdc68 JG	85
	86	/*
	87	* No bt_put on ports as a connection only holds _weak_ references
	88	* to them.
	89	*/
	90	g_free(connection);
	91	}
	92
f167d3c0 PP	93	static
	94	void bt_connection_try_remove_from_graph(struct bt_connection *connection)
	95	{
	96	void *graph = bt_object_borrow_parent(&connection->base);
	97
	98	if (connection->base.ref_count.count > 0 \|\|
	99	connection->downstream_port \|\|
	100	connection->upstream_port \|\|
	101	connection->iterators->len > 0) {
	102	return;
	103	}
	104
	105	/*
	106	* At this point we know that:
	107	*
a36bfb16	108	* 1. The connection is ended (ports were disconnected).
f167d3c0 PP	109	* 2. All the notification iterators that this connection
	110	* created, if any, are finalized.
	111	* 3. The connection's reference count is 0, so only the
	112	* parent (graph) owns this connection after this call.
	113	*
	114	* In other words, no other object than the graph knows this
	115	* connection.
	116	*
	117	* It is safe to remove the connection from the graph, therefore
	118	* destroying it.
	119	*/
a36bfb16 PP	120	BT_LOGD("Removing self from graph's connections: "
a36bfb16 PP	121	"graph-addr=%p, conn-addr=%p", graph, connection);
f167d3c0 PP	122	bt_graph_remove_connection(graph, connection);
	123	}
	124
	125	static
	126	void bt_connection_parent_is_owner(struct bt_object *obj)
	127	{
	128	struct bt_connection *connection = container_of(obj,
	129	struct bt_connection, base);
	130
	131	bt_connection_try_remove_from_graph(connection);
	132	}
	133
890882ef PP	134	struct bt_connection *bt_connection_from_private_connection(
	135	struct bt_private_connection *private_connection)
	136	{
	137	return bt_get(bt_connection_from_private(private_connection));
	138	}
	139
784cdc68 JG	140	BT_HIDDEN
	141	struct bt_connection *bt_connection_create(
	142	struct bt_graph *graph,
72b913fb PP	143	struct bt_port *upstream_port,
72b913fb PP	144	struct bt_port *downstream_port)
784cdc68 JG	145	{
	146	struct bt_connection *connection = NULL;
	147
72b913fb	148	if (bt_port_get_type(upstream_port) != BT_PORT_TYPE_OUTPUT) {
a36bfb16	149	BT_LOGW_STR("Invalid parameter: upstream port is not an output port.");
784cdc68 JG	150	goto end;
784cdc68 JG	151	}
72b913fb	152	if (bt_port_get_type(downstream_port) != BT_PORT_TYPE_INPUT) {
a36bfb16	153	BT_LOGW_STR("Invalid parameter: downstream port is not an input port.");
784cdc68 JG	154	goto end;
	155	}
	156
a36bfb16 PP	157	BT_LOGD("Creating connection: "
	158	"graph-addr=%p, upstream-port-addr=%p, uptream-port-name=\"%s\", "
	159	"downstream-port-addr=%p, downstream-port-name=\"%s\"",
	160	graph, upstream_port, bt_port_get_name(upstream_port),
	161	downstream_port, bt_port_get_name(downstream_port));
784cdc68 JG	162	connection = g_new0(struct bt_connection, 1);
784cdc68 JG	163	if (!connection) {
a36bfb16	164	BT_LOGE_STR("Failed to allocate one connection.");
784cdc68 JG	165	goto end;
	166	}
	167
	168	bt_object_init(connection, bt_connection_destroy);
f167d3c0 PP	169	bt_object_set_parent_is_owner_listener(connection,
f167d3c0 PP	170	bt_connection_parent_is_owner);
bd14d768 PP	171	connection->iterators = g_ptr_array_new();
bd14d768 PP	172	if (!connection->iterators) {
a36bfb16	173	BT_LOGE_STR("Failed to allocate a GPtrArray.");
bd14d768 PP	174	BT_PUT(connection);
	175	goto end;
	176	}
	177
784cdc68	178	/* Weak references are taken, see comment in header. */
72b913fb PP	179	connection->upstream_port = upstream_port;
72b913fb PP	180	connection->downstream_port = downstream_port;
a36bfb16	181	BT_LOGD_STR("Setting upstream port's connection.");
72b913fb	182	bt_port_set_connection(upstream_port, connection);
a36bfb16	183	BT_LOGD_STR("Setting downstream port's connection.");
72b913fb	184	bt_port_set_connection(downstream_port, connection);
784cdc68	185	bt_object_set_parent(connection, &graph->base);
a36bfb16 PP	186	BT_LOGD("Created connection: "
	187	"graph-addr=%p, upstream-port-addr=%p, uptream-port-name=\"%s\", "
	188	"downstream-port-addr=%p, downstream-port-name=\"%s\", "
	189	"conn-addr=%p",
	190	graph, upstream_port, bt_port_get_name(upstream_port),
	191	downstream_port, bt_port_get_name(downstream_port),
	192	connection);
	193
784cdc68 JG	194	end:
	195	return connection;
	196	}
	197
72b913fb	198	BT_HIDDEN
2038affb	199	void bt_connection_disconnect_ports(struct bt_connection *conn)
72b913fb PP	200	{
	201	struct bt_component *downstream_comp = NULL;
	202	struct bt_component *upstream_comp = NULL;
	203	struct bt_port *downstream_port = conn->downstream_port;
	204	struct bt_port *upstream_port = conn->upstream_port;
f167d3c0 PP	205	struct bt_graph graph = (void ) bt_object_borrow_parent(conn);
f167d3c0 PP	206	size_t i;
72b913fb PP	207
	208	if (downstream_port) {
	209	downstream_comp = bt_port_get_component(downstream_port);
	210	bt_port_set_connection(downstream_port, NULL);
	211	conn->downstream_port = NULL;
	212	}
	213
	214	if (upstream_port) {
	215	upstream_comp = bt_port_get_component(upstream_port);
	216	bt_port_set_connection(upstream_port, NULL);
	217	conn->upstream_port = NULL;
	218	}
	219
2038affb	220	if (downstream_comp) {
a36bfb16	221	/* bt_component_port_disconnected() logs details */
72b913fb PP	222	bt_component_port_disconnected(downstream_comp,
	223	downstream_port);
	224	}
	225
2038affb	226	if (upstream_comp) {
a36bfb16	227	/* bt_component_port_disconnected() logs details */
72b913fb PP	228	bt_component_port_disconnected(upstream_comp, upstream_port);
	229	}
	230
f345f8bb	231	assert(graph);
a36bfb16	232	/* bt_graph_notify_ports_disconnected() logs details */
f345f8bb PP	233	bt_graph_notify_ports_disconnected(graph, upstream_comp,
f345f8bb PP	234	downstream_comp, upstream_port, downstream_port);
72b913fb PP	235	bt_put(downstream_comp);
72b913fb PP	236	bt_put(upstream_comp);
f167d3c0 PP	237
f167d3c0 PP	238	/*
a36bfb16	239	* Because this connection is ended, finalize (cancel) each
f167d3c0 PP	240	* notification iterator created from it.
	241	*/
	242	for (i = 0; i < conn->iterators->len; i++) {
	243	struct bt_notification_iterator *iterator =
	244	g_ptr_array_index(conn->iterators, i);
	245
a36bfb16 PP	246	BT_LOGD("Finalizing notification iterator created by this ended connection: "
a36bfb16 PP	247	"conn-addr=%p, iter-addr=%p", conn, iterator);
f167d3c0 PP	248	bt_notification_iterator_finalize(iterator);
	249
	250	/*
	251	* Make sure this iterator does not try to remove itself
	252	* from this connection's iterators on destruction
	253	* because this connection won't exist anymore.
	254	*/
	255	bt_notification_iterator_set_connection(iterator,
	256	NULL);
	257	}
	258
	259	g_ptr_array_set_size(conn->iterators, 0);
	260	bt_connection_try_remove_from_graph(conn);
72b913fb PP	261	}
	262
	263	struct bt_port *bt_connection_get_upstream_port(
784cdc68 JG	264	struct bt_connection *connection)
784cdc68 JG	265	{
72b913fb	266	return connection ? bt_get(connection->upstream_port) : NULL;
784cdc68 JG	267	}
784cdc68 JG	268
72b913fb	269	struct bt_port *bt_connection_get_downstream_port(
784cdc68 JG	270	struct bt_connection *connection)
784cdc68 JG	271	{
72b913fb	272	return connection ? bt_get(connection->downstream_port) : NULL;
784cdc68 JG	273	}
	274
	275	struct bt_notification_iterator *
890882ef	276	bt_private_connection_create_notification_iterator(
fa054faf PP	277	struct bt_private_connection *private_connection,
fa054faf PP	278	const enum bt_notification_type *notification_types)
784cdc68	279	{
890882ef PP	280	enum bt_component_class_type upstream_comp_class_type;
	281	struct bt_notification_iterator *iterator = NULL;
	282	struct bt_port *upstream_port = NULL;
784cdc68	283	struct bt_component *upstream_component = NULL;
890882ef PP	284	struct bt_component_class *upstream_comp_class = NULL;
	285	struct bt_connection *connection = NULL;
	286	bt_component_class_notification_iterator_init_method init_method = NULL;
fa054faf PP	287	static const enum bt_notification_type all_notif_types[] = {
	288	BT_NOTIFICATION_TYPE_ALL,
	289	BT_NOTIFICATION_TYPE_SENTINEL,
	290	};
784cdc68	291
890882ef	292	if (!private_connection) {
a36bfb16	293	BT_LOGW_STR("Invalid parameter: private connection is NULL.");
890882ef	294	goto error;
784cdc68 JG	295	}
784cdc68 JG	296
fa054faf	297	if (!notification_types) {
a36bfb16	298	BT_LOGD_STR("No notification types: subscribing to all notifications.");
fa054faf PP	299	notification_types = all_notif_types;
	300	}
	301
890882ef	302	connection = bt_connection_from_private(private_connection);
72b913fb	303	if (!connection->upstream_port \|\| !connection->downstream_port) {
a36bfb16 PP	304	BT_LOGW("Invalid parameter: connection is ended: "
a36bfb16 PP	305	"conn-addr=%p", connection);
890882ef	306	goto error;
72b913fb PP	307	}
72b913fb PP	308
890882ef PP	309	upstream_port = connection->upstream_port;
	310	assert(upstream_port);
	311	upstream_component = bt_port_get_component(upstream_port);
784cdc68	312	assert(upstream_component);
890882ef	313	upstream_comp_class = upstream_component->class;
a36bfb16 PP	314	BT_LOGD("Creating notification iterator from connection: "
	315	"conn-addr=%p, upstream-port-addr=%p, "
	316	"upstream-port-name=\"%s\", upstream-comp-addr=%p, "
	317	"upstream-comp-name=\"%s\"",
	318	connection, connection->upstream_port,
	319	bt_port_get_name(connection->upstream_port),
	320	upstream_component, bt_component_get_name(upstream_component));
890882ef PP	321	upstream_comp_class_type =
	322	bt_component_get_class_type(upstream_component);
	323	if (upstream_comp_class_type != BT_COMPONENT_CLASS_TYPE_SOURCE &&
	324	upstream_comp_class_type != BT_COMPONENT_CLASS_TYPE_FILTER) {
	325	/* Unsupported operation. */
a36bfb16 PP	326	BT_LOGW("Upstream component's class is not a source or filter component class: "
	327	"comp-class-type=%s",
	328	bt_component_class_type_string(upstream_comp_class_type));
890882ef PP	329	goto error;
890882ef PP	330	}
784cdc68	331
3230ee6b	332	iterator = bt_notification_iterator_create(upstream_component,
bd14d768	333	upstream_port, notification_types, connection);
890882ef	334	if (!iterator) {
a36bfb16	335	BT_LOGW("Cannot create notification iterator from connection.");
890882ef PP	336	goto error;
	337	}
	338
	339	switch (upstream_comp_class_type) {
784cdc68	340	case BT_COMPONENT_CLASS_TYPE_SOURCE:
890882ef PP	341	{
	342	struct bt_component_class_source *source_class =
	343	container_of(upstream_comp_class,
	344	struct bt_component_class_source, parent);
	345	init_method = source_class->methods.iterator.init;
784cdc68	346	break;
890882ef	347	}
784cdc68	348	case BT_COMPONENT_CLASS_TYPE_FILTER:
890882ef PP	349	{
	350	struct bt_component_class_filter *filter_class =
	351	container_of(upstream_comp_class,
	352	struct bt_component_class_filter, parent);
	353	init_method = filter_class->methods.iterator.init;
784cdc68	354	break;
890882ef	355	}
784cdc68	356	default:
890882ef	357	/* Unreachable. */
0fbb9a9f	358	abort();
890882ef PP	359	}
	360
	361	if (init_method) {
a36bfb16 PP	362	enum bt_notification_iterator_status status;
	363
	364	BT_LOGD("Calling user's initialization method: iter-addr=%p",
	365	iterator);
	366	status = init_method(
91457551 PP	367	bt_private_notification_iterator_from_notification_iterator(iterator),
91457551 PP	368	bt_private_port_from_port(upstream_port));
a36bfb16 PP	369	BT_LOGD("User method returned: status=%s",
a36bfb16 PP	370	bt_notification_iterator_status_string(status));
890882ef	371	if (status < 0) {
a36bfb16	372	BT_LOGW_STR("Initialization method failed.");
890882ef PP	373	goto error;
	374	}
	375	}
	376
bd14d768	377	g_ptr_array_add(connection->iterators, iterator);
a36bfb16 PP	378	BT_LOGD("Created notification iterator from connection: "
	379	"conn-addr=%p, upstream-port-addr=%p, "
	380	"upstream-port-name=\"%s\", upstream-comp-addr=%p, "
	381	"upstream-comp-name=\"%s\", iter-addr=%p",
	382	connection, connection->upstream_port,
	383	bt_port_get_name(connection->upstream_port),
	384	upstream_component, bt_component_get_name(upstream_component),
	385	iterator);
890882ef PP	386	goto end;
	387
	388	error:
	389	BT_PUT(iterator);
	390
784cdc68 JG	391	end:
784cdc68 JG	392	bt_put(upstream_component);
890882ef	393	return iterator;
784cdc68	394	}
bd14d768 PP	395
	396	BT_HIDDEN
	397	void bt_connection_remove_iterator(struct bt_connection *conn,
	398	struct bt_notification_iterator *iterator)
	399	{
	400	g_ptr_array_remove(conn->iterators, iterator);
a36bfb16 PP	401	BT_LOGV("Removed notification iterator from connection: "
a36bfb16 PP	402	"conn-addr=%p, iter-addr=%p", conn, iterator);
f167d3c0	403	bt_connection_try_remove_from_graph(conn);
bd14d768	404	}