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

/*
 * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
 * Copyright 2015 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 "COMP"
#include <babeltrace/lib-logging-internal.h>

#include <babeltrace/assert-internal.h>
#include <babeltrace/assert-pre-internal.h>
#include <babeltrace/graph/self-component.h>
#include <babeltrace/graph/component-const.h>
#include <babeltrace/graph/component-source-const.h>
#include <babeltrace/graph/component-filter-const.h>
#include <babeltrace/graph/component-sink-const.h>
#include <babeltrace/graph/component-internal.h>
#include <babeltrace/graph/component-class-internal.h>
#include <babeltrace/graph/component-source-internal.h>
#include <babeltrace/graph/component-filter-internal.h>
#include <babeltrace/graph/component-sink-internal.h>
#include <babeltrace/graph/connection-internal.h>
#include <babeltrace/graph/graph-internal.h>
#include <babeltrace/graph/message-iterator-internal.h>
#include <babeltrace/graph/port-internal.h>
#include <babeltrace/babeltrace-internal.h>
#include <babeltrace/compiler-internal.h>
#include <babeltrace/types.h>
#include <babeltrace/value.h>
#include <babeltrace/value-internal.h>
#include <stdint.h>
#include <inttypes.h>

static
struct bt_component * (* const component_create_funcs[])(
		const struct bt_component_class *) = {
	[BT_COMPONENT_CLASS_TYPE_SOURCE] = bt_component_source_create,
	[BT_COMPONENT_CLASS_TYPE_SINK] = bt_component_sink_create,
	[BT_COMPONENT_CLASS_TYPE_FILTER] = bt_component_filter_create,
};

static
void (*component_destroy_funcs[])(struct bt_component *) = {
	[BT_COMPONENT_CLASS_TYPE_SOURCE] = bt_component_source_destroy,
	[BT_COMPONENT_CLASS_TYPE_SINK] = bt_component_sink_destroy,
	[BT_COMPONENT_CLASS_TYPE_FILTER] = bt_component_filter_destroy,
};

static
void finalize_component(struct bt_component *comp)
{
	typedef void (*method_t)(void *);

	method_t method = NULL;

	BT_ASSERT(comp);

	switch (comp->class->type) {
	case BT_COMPONENT_CLASS_TYPE_SOURCE:
	{
		struct bt_component_class_source *src_cc = (void *) comp->class;

		method = (method_t) src_cc->methods.finalize;
		break;
	}
	case BT_COMPONENT_CLASS_TYPE_FILTER:
	{
		struct bt_component_class_filter *flt_cc = (void *) comp->class;

		method = (method_t) flt_cc->methods.finalize;
		break;
	}
	case BT_COMPONENT_CLASS_TYPE_SINK:
	{
		struct bt_component_class_sink *sink_cc = (void *) comp->class;

		method = (method_t) sink_cc->methods.finalize;
		break;
	}
	default:
		abort();
	}

	if (method) {
		BT_LIB_LOGD("Calling user's finalization method: "
			"%![comp-]+c", comp);
		method(comp);
	}
}

static
void destroy_component(struct bt_object *obj)
{
	struct bt_component *component = NULL;
	int i;

	if (!obj) {
		return;
	}

	/*
	 * The component's reference count is 0 if we're here. Increment
	 * it to avoid a double-destroy (possibly infinitely recursive).
	 * This could happen for example if the component's finalization
	 * function does bt_object_get_ref() (or anything that causes
	 * bt_object_get_ref() to be called) on itself (ref. count goes
	 * from 0 to 1), and then bt_object_put_ref(): the reference
	 * count would go from 1 to 0 again and this function would be
	 * called again.
	 */
	obj->ref_count++;
	component = container_of(obj, struct bt_component, base);
	BT_LIB_LOGD("Destroying component: %![comp-]+c, %![graph-]+g",
		component, bt_component_borrow_graph(component));

	/* Call destroy listeners in reverse registration order */
	BT_LOGD_STR("Calling destroy listeners.");

	for (i = component->destroy_listeners->len - 1; i >= 0; i--) {
		struct bt_component_destroy_listener *listener =
			&g_array_index(component->destroy_listeners,
				struct bt_component_destroy_listener, i);

		listener->func(component, listener->data);
	}

	/*
	 * User data is destroyed first, followed by the concrete
	 * component instance. Do not finalize if the component's user
	 * initialization method failed in the first place.
	 */
	if (component->initialized) {
		finalize_component(component);
	}

	if (component->destroy) {
		BT_LOGD_STR("Destroying type-specific data.");
		component->destroy(component);
	}

	if (component->input_ports) {
		BT_LOGD_STR("Destroying input ports.");
		g_ptr_array_free(component->input_ports, TRUE);
		component->input_ports = NULL;
	}

	if (component->output_ports) {
		BT_LOGD_STR("Destroying output ports.");
		g_ptr_array_free(component->output_ports, TRUE);
		component->output_ports = NULL;
	}

	if (component->destroy_listeners) {
		g_array_free(component->destroy_listeners, TRUE);
		component->destroy_listeners = NULL;
	}

	if (component->name) {
		g_string_free(component->name, TRUE);
		component->name = NULL;
	}

	BT_LOGD_STR("Putting component class.");
	BT_OBJECT_PUT_REF_AND_RESET(component->class);
	g_free(component);
}

enum bt_component_class_type bt_component_get_class_type(
		const struct bt_component *component)
{
	BT_ASSERT_PRE_NON_NULL(component, "Component");
	return component->class->type;
}

static
struct bt_port *add_port(
		struct bt_component *component, GPtrArray *ports,
		enum bt_port_type port_type, const char *name, void *user_data)
{
	struct bt_port *new_port = NULL;
	struct bt_graph *graph = NULL;

	BT_ASSERT_PRE_NON_NULL(component, "Component");
	BT_ASSERT_PRE_NON_NULL(name, "Name");
	BT_ASSERT_PRE(strlen(name) > 0, "Name is empty");
	graph = bt_component_borrow_graph(component);
	BT_ASSERT_PRE(graph && !bt_graph_is_canceled(graph),
		"Component's graph is canceled: %![comp-]+c, %![graph-]+g",
		component, graph);

	// TODO: Validate that the name is not already used.

	BT_LIB_LOGD("Adding port to component: %![comp-]+c, "
		"port-type=%s, port-name=\"%s\"", component,
		bt_port_type_string(port_type), name);

	new_port = bt_port_create(component, port_type, name, user_data);
	if (!new_port) {
		BT_LOGE_STR("Cannot create port object.");
		goto end;
	}

	/*
	 * No name clash, add the port.
	 * The component is now the port's parent; it should _not_
	 * hold a reference to the port since the port's lifetime
	 * is now protected by the component's own lifetime.
	 */
	g_ptr_array_add(ports, new_port);

	/*
	 * Notify the graph's creator that a new port was added.
	 */
	bt_object_get_ref(bt_component_borrow_graph(component));
	graph = bt_component_borrow_graph(component);
	if (graph) {
		bt_graph_notify_port_added(graph, new_port);
		BT_OBJECT_PUT_REF_AND_RESET(graph);
	}

	BT_LIB_LOGD("Created and added port to component: "
		"%![comp-]+c, %![port-]+p", component, new_port);

end:
	return new_port;
}

BT_HIDDEN
uint64_t bt_component_get_input_port_count(const struct bt_component *comp)
{
	BT_ASSERT_PRE_NON_NULL(comp, "Component");
	return (uint64_t) comp->input_ports->len;
}

BT_HIDDEN
uint64_t bt_component_get_output_port_count(const struct bt_component *comp)
{
	BT_ASSERT_PRE_NON_NULL(comp, "Component");
	return (uint64_t) comp->output_ports->len;
}

BT_HIDDEN
int bt_component_create(struct bt_component_class *component_class,
		const char *name, struct bt_component **user_component)
{
	int ret = 0;
	struct bt_component *component = NULL;
	enum bt_component_class_type type;

	BT_ASSERT(user_component);
	BT_ASSERT(component_class);
	BT_ASSERT(name);

	type = bt_component_class_get_type(component_class);
	BT_LIB_LOGD("Creating empty component from component class: %![cc-]+C, "
		"comp-name=\"%s\"", component_class, name);
	component = component_create_funcs[type](component_class);
	if (!component) {
		BT_LOGE_STR("Cannot create specific component object.");
		ret = -1;
		goto end;
	}

	bt_object_init_shared_with_parent(&component->base,
		destroy_component);
	component->class = component_class;
	bt_object_get_no_null_check(component->class);
	component->destroy = component_destroy_funcs[type];
	component->name = g_string_new(name);
	if (!component->name) {
		BT_LOGE_STR("Failed to allocate one GString.");
		ret = -1;
		goto end;
	}

	component->input_ports = g_ptr_array_new_with_free_func(
		(GDestroyNotify) bt_object_try_spec_release);
	if (!component->input_ports) {
		BT_LOGE_STR("Failed to allocate one GPtrArray.");
		ret = -1;
		goto end;
	}

	component->output_ports = g_ptr_array_new_with_free_func(
		(GDestroyNotify) bt_object_try_spec_release);
	if (!component->output_ports) {
		BT_LOGE_STR("Failed to allocate one GPtrArray.");
		ret = -1;
		goto end;
	}

	component->destroy_listeners = g_array_new(FALSE, TRUE,
		sizeof(struct bt_component_destroy_listener));
	if (!component->destroy_listeners) {
		BT_LOGE_STR("Failed to allocate one GArray.");
		ret = -1;
		goto end;
	}

	BT_LIB_LOGD("Created empty component from component class: "
		"%![cc-]+C, %![comp-]+c", component_class, component);
	BT_OBJECT_MOVE_REF(*user_component, component);

end:
	bt_object_put_ref(component);
	return ret;
}

const char *bt_component_get_name(const struct bt_component *component)
{
	BT_ASSERT_PRE_NON_NULL(component, "Component");
	return component->name->str;
}

struct bt_component_class *bt_component_borrow_class(
		const struct bt_component *component)
{
	BT_ASSERT_PRE_NON_NULL(component, "Component");
	return component->class;
}

void *bt_self_component_get_data(const struct bt_self_component *self_comp)
{
	struct bt_component *component = (void *) self_comp;

	BT_ASSERT_PRE_NON_NULL(component, "Component");
	return component->user_data;
}

void bt_self_component_set_data(struct bt_self_component *self_comp,
		void *data)
{
	struct bt_component *component = (void *) self_comp;

	BT_ASSERT_PRE_NON_NULL(component, "Component");
	component->user_data = data;
	BT_LIB_LOGV("Set component's user data: %!+c", component);
}

BT_HIDDEN
void bt_component_set_graph(struct bt_component *component,
		struct bt_graph *graph)
{
	bt_object_set_parent(&component->base,
		graph ? &graph->base : NULL);
}

bt_bool bt_component_graph_is_canceled(const struct bt_component *component)
{
	return bt_graph_is_canceled(
		(void *) bt_object_borrow_parent(&component->base));
}

static
struct bt_port *borrow_port_by_name(GPtrArray *ports,
		const char *name)
{
	uint64_t i;
	struct bt_port *ret_port = NULL;

	BT_ASSERT(name);

	for (i = 0; i < ports->len; i++) {
		struct bt_port *port = g_ptr_array_index(ports, i);

		if (!strcmp(name, port->name->str)) {
			ret_port = port;
			break;
		}
	}

	return ret_port;
}

BT_HIDDEN
struct bt_port_input *bt_component_borrow_input_port_by_name(
		struct bt_component *comp, const char *name)
{
	BT_ASSERT(comp);
	return (void *) borrow_port_by_name(comp->input_ports, name);
}

BT_HIDDEN
struct bt_port_output *bt_component_borrow_output_port_by_name(
		struct bt_component *comp, const char *name)
{
	BT_ASSERT_PRE_NON_NULL(comp, "Component");
	return (void *)
		borrow_port_by_name(comp->output_ports, name);
}

static
struct bt_port *borrow_port_by_index(GPtrArray *ports, uint64_t index)
{
	BT_ASSERT(index < ports->len);
	return g_ptr_array_index(ports, index);
}

BT_HIDDEN
struct bt_port_input *bt_component_borrow_input_port_by_index(
		struct bt_component *comp, uint64_t index)
{
	BT_ASSERT_PRE_NON_NULL(comp, "Component");
	BT_ASSERT_PRE_VALID_INDEX(index, comp->input_ports->len);
	return (void *)
		borrow_port_by_index(comp->input_ports, index);
}

BT_HIDDEN
struct bt_port_output *bt_component_borrow_output_port_by_index(
		struct bt_component *comp, uint64_t index)
{
	BT_ASSERT_PRE_NON_NULL(comp, "Component");
	BT_ASSERT_PRE_VALID_INDEX(index, comp->output_ports->len);
	return (void *)
		borrow_port_by_index(comp->output_ports, index);
}

BT_HIDDEN
struct bt_port_input *bt_component_add_input_port(
		struct bt_component *component, const char *name,
		void *user_data)
{
	/* add_port() logs details */
	return (void *)
		add_port(component, component->input_ports,
			BT_PORT_TYPE_INPUT, name, user_data);
}

BT_HIDDEN
struct bt_port_output *bt_component_add_output_port(
		struct bt_component *component, const char *name,
		void *user_data)
{
	/* add_port() logs details */
	return (void *)
		add_port(component, component->output_ports,
			BT_PORT_TYPE_OUTPUT, name, user_data);
}

static
void remove_port_by_index(struct bt_component *component,
		GPtrArray *ports, uint64_t index)
{
	struct bt_port *port;
	struct bt_graph *graph;

	BT_ASSERT(ports);
	BT_ASSERT(index < ports->len);
	port = g_ptr_array_index(ports, index);
	BT_LIB_LOGD("Removing port from component: %![comp-]+c, %![port-]+p",
		component, port);

	/* Disconnect both ports of this port's connection, if any */
	if (port->connection) {
		bt_connection_end(port->connection, true);
	}

	/*
	 * The port's current reference count can be 0 at this point,
	 * which means its parent (component) keeps it alive. We are
	 * about to remove the port from its parent's container (with
	 * the g_ptr_array_remove_index() call below), which in this
	 * case would destroy it. This is not good because we still
	 * need the port for the bt_graph_notify_port_removed() call
	 * below (in which its component is `NULL` as expected because
	 * of the bt_object_set_parent() call below).
	 *
	 * To avoid a destroyed port during the message callback,
	 * get a reference now, and put it (destroying the port if its
	 * reference count is 0 at this point) after notifying the
	 * graph's user.
	 */
	bt_object_get_no_null_check(&port->base);

	/*
	 * Remove from parent's array of ports (weak refs). This never
	 * destroys the port object because its reference count is at
	 * least 1 thanks to the bt_object_get_no_null_check() call
	 * above.
	 */
	g_ptr_array_remove_index(ports, index);

	/* Detach port from its component parent */
	bt_object_set_parent(&port->base, NULL);

	/*
	 * Notify the graph's creator that a port is removed.
	 */
	graph = bt_component_borrow_graph(component);
	if (graph) {
		bt_graph_notify_port_removed(graph, component, port);
	}

	BT_LIB_LOGD("Removed port from component: %![comp-]+c, %![port-]+p",
		component, port);

	/*
	 * Put the local reference. If this port's reference count was 0
	 * when entering this function, it is 1 now, so it is destroyed
	 * immediately.
	 */
	bt_object_put_no_null_check(&port->base);
}

BT_HIDDEN
void bt_component_remove_port(struct bt_component *component,
		struct bt_port *port)
{
	uint64_t i;
	GPtrArray *ports = NULL;

	BT_ASSERT(component);
	BT_ASSERT(port);

	switch (port->type) {
	case BT_PORT_TYPE_INPUT:
		ports = component->input_ports;
		break;
	case BT_PORT_TYPE_OUTPUT:
		ports = component->output_ports;
		break;
	default:
		abort();
	}

	BT_ASSERT(ports);

	for (i = 0; i < ports->len; i++) {
		struct bt_port *cur_port = g_ptr_array_index(ports, i);

		if (cur_port == port) {
			remove_port_by_index(component,
				ports, i);
			goto end;
		}
	}

	BT_LIB_LOGW("Port to remove from component was not found: "
		"%![comp-]+c, %![port-]+p", component, port);

end:
	return;
}

BT_HIDDEN
enum bt_self_component_status bt_component_accept_port_connection(
		struct bt_component *comp, struct bt_port *self_port,
		struct bt_port *other_port)
{
	typedef enum bt_self_component_status (*method_t)(
		void *, void *, const void *);

	enum bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK;
	method_t method = NULL;

	BT_ASSERT(comp);
	BT_ASSERT(self_port);
	BT_ASSERT(other_port);

	switch (comp->class->type) {
	case BT_COMPONENT_CLASS_TYPE_SOURCE:
	{
		struct bt_component_class_source *src_cc = (void *) comp->class;

		switch (self_port->type) {
		case BT_PORT_TYPE_OUTPUT:
			method = (method_t) src_cc->methods.accept_output_port_connection;
			break;
		default:
			abort();
		}

		break;
	}
	case BT_COMPONENT_CLASS_TYPE_FILTER:
	{
		struct bt_component_class_filter *flt_cc = (void *) comp->class;

		switch (self_port->type) {
		case BT_PORT_TYPE_INPUT:
			method = (method_t) flt_cc->methods.accept_input_port_connection;
			break;
		case BT_PORT_TYPE_OUTPUT:
			method = (method_t) flt_cc->methods.accept_output_port_connection;
			break;
		default:
			abort();
		}

		break;
	}
	case BT_COMPONENT_CLASS_TYPE_SINK:
	{
		struct bt_component_class_sink *sink_cc = (void *) comp->class;

		switch (self_port->type) {
		case BT_PORT_TYPE_INPUT:
			method = (method_t) sink_cc->methods.accept_input_port_connection;
			break;
		default:
			abort();
		}

		break;
	}
	default:
		abort();
	}

	if (method) {
		BT_LIB_LOGD("Calling user's \"accept port connection\" method: "
			"%![comp-]+c, %![self-port-]+p, %![other-port-]+p",
			comp, self_port, other_port);
		status = method(comp, self_port, (void *) other_port);
		BT_LOGD("User method returned: status=%s",
			bt_self_component_status_string(status));
	}

	return status;
}

BT_HIDDEN
enum bt_self_component_status bt_component_port_connected(
		struct bt_component *comp, struct bt_port *self_port,
		struct bt_port *other_port)
{
	typedef enum bt_self_component_status (*method_t)(
		void *, void *, const void *);

	enum bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK;
	method_t method = NULL;

	BT_ASSERT(comp);
	BT_ASSERT(self_port);
	BT_ASSERT(other_port);

	switch (comp->class->type) {
	case BT_COMPONENT_CLASS_TYPE_SOURCE:
	{
		struct bt_component_class_source *src_cc = (void *) comp->class;

		switch (self_port->type) {
		case BT_PORT_TYPE_OUTPUT:
			method = (method_t) src_cc->methods.output_port_connected;
			break;
		default:
			abort();
		}

		break;
	}
	case BT_COMPONENT_CLASS_TYPE_FILTER:
	{
		struct bt_component_class_filter *flt_cc = (void *) comp->class;

		switch (self_port->type) {
		case BT_PORT_TYPE_INPUT:
			method = (method_t) flt_cc->methods.input_port_connected;
			break;
		case BT_PORT_TYPE_OUTPUT:
			method = (method_t) flt_cc->methods.output_port_connected;
			break;
		default:
			abort();
		}

		break;
	}
	case BT_COMPONENT_CLASS_TYPE_SINK:
	{
		struct bt_component_class_sink *sink_cc = (void *) comp->class;

		switch (self_port->type) {
		case BT_PORT_TYPE_INPUT:
			method = (method_t) sink_cc->methods.input_port_connected;
			break;
		default:
			abort();
		}

		break;
	}
	default:
		abort();
	}

	if (method) {
		BT_LIB_LOGD("Calling user's \"port connected\" method: "
			"%![comp-]+c, %![self-port-]+p, %![other-port-]+p",
			comp, self_port, other_port);
		status = method(comp, self_port, (void *) other_port);
		BT_LOGD("User method returned: status=%s",
			bt_self_component_status_string(status));
	}

	return status;
}

BT_HIDDEN
void bt_component_port_disconnected(struct bt_component *comp,
		struct bt_port *port)
{
	typedef void (*method_t)(void *, void *);

	method_t method = NULL;

	BT_ASSERT(comp);
	BT_ASSERT(port);

	switch (comp->class->type) {
	case BT_COMPONENT_CLASS_TYPE_SOURCE:
	{
		struct bt_component_class_source *src_cc = (void *) comp->class;

		switch (port->type) {
		case BT_PORT_TYPE_OUTPUT:
			method = (method_t) src_cc->methods.output_port_disconnected;
			break;
		default:
			abort();
		}

		break;
	}
	case BT_COMPONENT_CLASS_TYPE_FILTER:
	{
		struct bt_component_class_filter *flt_cc = (void *) comp->class;

		switch (port->type) {
		case BT_PORT_TYPE_INPUT:
			method = (method_t) flt_cc->methods.input_port_disconnected;
			break;
		case BT_PORT_TYPE_OUTPUT:
			method = (method_t) flt_cc->methods.output_port_disconnected;
			break;
		default:
			abort();
		}

		break;
	}
	case BT_COMPONENT_CLASS_TYPE_SINK:
	{
		struct bt_component_class_sink *sink_cc = (void *) comp->class;

		switch (port->type) {
		case BT_PORT_TYPE_INPUT:
			method = (method_t) sink_cc->methods.input_port_disconnected;
			break;
		default:
			abort();
		}

		break;
	}
	default:
		abort();
	}

	if (method) {
		BT_LIB_LOGD("Calling user's \"port disconnected\" method: "
			"%![comp-]+c, %![port-]+p", comp, port);
		method(comp, port);
	}
}

BT_HIDDEN
void bt_component_add_destroy_listener(struct bt_component *component,
		bt_component_destroy_listener_func func, void *data)
{
	struct bt_component_destroy_listener listener;

	BT_ASSERT(component);
	BT_ASSERT(func);
	listener.func = func;
	listener.data = data;
	g_array_append_val(component->destroy_listeners, listener);
	BT_LIB_LOGV("Added destroy listener: %![comp-]+c, "
		"func-addr=%p, data-addr=%p",
		component, func, data);
}

BT_HIDDEN
void bt_component_remove_destroy_listener(struct bt_component *component,
		bt_component_destroy_listener_func func, void *data)
{
	uint64_t i;

	BT_ASSERT(component);
	BT_ASSERT(func);

	for (i = 0; i < component->destroy_listeners->len; i++) {
		struct bt_component_destroy_listener *listener =
			&g_array_index(component->destroy_listeners,
				struct bt_component_destroy_listener, i);

		if (listener->func == func && listener->data == data) {
			g_array_remove_index(component->destroy_listeners, i);
			i--;
			BT_LIB_LOGV("Removed destroy listener: %![comp-]+c, "
				"func-addr=%p, data-addr=%p",
				component, func, data);
		}
	}
}

void bt_component_get_ref(const struct bt_component *component)
{
	bt_object_get_ref(component);
}

void bt_component_put_ref(const struct bt_component *component)
{
	bt_object_put_ref(component);
}