[babeltrace.git] / src / lib / graph / component-class.c

/*
 * SPDX-License-Identifier: MIT
 *
 * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
 * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 */

#define BT_LOG_TAG "LIB/COMPONENT-CLASS"
#include "lib/logging.h"

#include "common/assert.h"
#include "lib/assert-cond.h"
#include "compat/compiler.h"
#include <babeltrace2/graph/component-class.h>
#include <babeltrace2/types.h>
#include <glib.h>

#include "component-class.h"
#include "lib/func-status.h"
#include "lib/graph/message-iterator-class.h"

#define BT_ASSERT_PRE_DEV_COMP_CLS_HOT(_cc)				\
	BT_ASSERT_PRE_DEV_HOT("component-class",			\
		((const struct bt_component_class *) (_cc)), \
		"Component class", ": %!+C", (_cc))

static
void destroy_component_class(struct bt_object *obj)
{
	struct bt_component_class *class;
	int i;

	BT_ASSERT(obj);
	class = container_of(obj, struct bt_component_class, base);

	BT_LIB_LOGI("Destroying component class: %!+C", class);

	/* Call destroy listeners in reverse registration order */
	for (i = class->destroy_listeners->len - 1; i >= 0; i--) {
		struct bt_component_class_destroy_listener *listener =
			&g_array_index(class->destroy_listeners,
				struct bt_component_class_destroy_listener,
				i);

		BT_LOGD("Calling destroy listener: func-addr=%p, data-addr=%p",
			listener->func, listener->data);
		listener->func(class, listener->data);
	}

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

	if (class->description) {
		g_string_free(class->description, TRUE);
		class->description = NULL;
	}

	if (class->help) {
		g_string_free(class->help, TRUE);
		class->help = NULL;
	}

	if (class->plugin_name) {
		g_string_free(class->plugin_name, TRUE);
		class->plugin_name = NULL;
	}

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

	if (bt_component_class_has_message_iterator_class(class)) {
		struct bt_component_class_with_iterator_class *class_with_iter_class =
			container_of(class, struct bt_component_class_with_iterator_class, parent);

		BT_ASSERT(class_with_iter_class->msg_iter_cls);
		bt_message_iterator_class_put_ref(class_with_iter_class->msg_iter_cls);
		class_with_iter_class->msg_iter_cls = NULL;
	}

	g_free(class);
}

static
int bt_component_class_init(struct bt_component_class *class,
		enum bt_component_class_type type, const char *name)
{
	int ret = 0;

	bt_object_init_shared(&class->base, destroy_component_class);
	class->type = type;
	class->name = g_string_new(name);
	if (!class->name) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
		goto error;
	}

	class->description = g_string_new(NULL);
	if (!class->description) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
		goto error;
	}

	class->help = g_string_new(NULL);
	if (!class->help) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
		goto error;
	}

	class->plugin_name = g_string_new(NULL);
	if (!class->plugin_name) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
		goto error;
	}

	class->destroy_listeners = g_array_new(FALSE, TRUE,
		sizeof(struct bt_component_class_destroy_listener));
	if (!class->destroy_listeners) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GArray.");
		goto error;
	}

	goto end;

error:
	BT_OBJECT_PUT_REF_AND_RESET(class);
	ret = -1;

end:
	return ret;
}

static
int bt_component_class_with_iterator_class_init(
		struct bt_component_class_with_iterator_class *class,
		enum bt_component_class_type type, const char *name,
		struct bt_message_iterator_class *message_iterator_class)
{
	int ret;

	ret = bt_component_class_init(&class->parent, type, name);
	if (ret != 0) {
		goto end;
	}

	class->msg_iter_cls = message_iterator_class;
	bt_message_iterator_class_get_ref(class->msg_iter_cls);
	bt_message_iterator_class_freeze(class->msg_iter_cls);

end:
	return ret;
}

struct bt_component_class_source *bt_component_class_source_create(
		const char *name,
		struct bt_message_iterator_class *message_iterator_class)
{
	struct bt_component_class_source *source_class = NULL;
	int ret;

	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_NAME_NON_NULL(name);
	BT_ASSERT_PRE_MSG_ITER_CLS_NON_NULL(message_iterator_class);
	BT_LIB_LOGI("Creating source component class: "
		"name=\"%s\", %![msg-iter-cls-]+I",
		name, message_iterator_class);
	source_class = g_new0(struct bt_component_class_source, 1);
	if (!source_class) {
		BT_LIB_LOGE_APPEND_CAUSE(
			"Failed to allocate one source component class.");
		goto end;
	}

	/* bt_component_class_init() logs errors */
	ret = bt_component_class_with_iterator_class_init(&source_class->parent,
		BT_COMPONENT_CLASS_TYPE_SOURCE, name, message_iterator_class);
	if (ret) {
		/*
		 * If bt_component_class_init() fails, the component
		 * class is put, therefore its memory is already
		 * freed.
		 */
		source_class = NULL;
		goto end;
	}

	BT_LIB_LOGI("Created source component class: %!+C", source_class);

end:
	return (void *) source_class;
}

struct bt_component_class_filter *bt_component_class_filter_create(
		const char *name,
		struct bt_message_iterator_class *message_iterator_class)
{
	struct bt_component_class_filter *filter_class = NULL;
	int ret;

	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_NAME_NON_NULL(name);
	BT_ASSERT_PRE_MSG_ITER_CLS_NON_NULL(message_iterator_class);
	BT_LIB_LOGI("Creating filter component class: "
		"name=\"%s\", %![msg-iter-cls-]+I",
		name, message_iterator_class);
	filter_class = g_new0(struct bt_component_class_filter, 1);
	if (!filter_class) {
		BT_LIB_LOGE_APPEND_CAUSE(
			"Failed to allocate one filter component class.");
		goto end;
	}

	/* bt_component_class_init() logs errors */
	ret = bt_component_class_with_iterator_class_init(&filter_class->parent,
		BT_COMPONENT_CLASS_TYPE_FILTER, name, message_iterator_class);
	if (ret) {
		/*
		 * If bt_component_class_init() fails, the component
		 * class is put, therefore its memory is already
		 * freed.
		 */
		filter_class = NULL;
		goto end;
	}

	BT_LIB_LOGI("Created filter component class: %!+C", filter_class);

end:
	return (void *) filter_class;
}

struct bt_component_class_sink *bt_component_class_sink_create(
		const char *name, bt_component_class_sink_consume_method method)
{
	struct bt_component_class_sink *sink_class = NULL;
	int ret;

	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_NAME_NON_NULL(name);
	BT_ASSERT_PRE_NON_NULL("consume-method", method, "Consume next method");
	BT_LOGI("Creating sink component class: "
		"name=\"%s\", consume-method-addr=%p",
		name, method);
	sink_class = g_new0(struct bt_component_class_sink, 1);
	if (!sink_class) {
		BT_LIB_LOGE_APPEND_CAUSE(
			"Failed to allocate one sink component class.");
		goto end;
	}

	/* bt_component_class_init() logs errors */
	ret = bt_component_class_init(&sink_class->parent,
		BT_COMPONENT_CLASS_TYPE_SINK, name);
	if (ret) {
		/*
		 * If bt_component_class_init() fails, the component
		 * class is put, therefore its memory is already
		 * freed.
		 */
		sink_class = NULL;
		goto end;
	}

	sink_class->methods.consume = method;
	BT_LIB_LOGI("Created sink component class: %!+C", sink_class);

end:
	return (void *) sink_class;
}

enum bt_component_class_set_method_status
bt_component_class_source_set_get_supported_mip_versions_method(
		struct bt_component_class_source *comp_cls,
		bt_component_class_source_get_supported_mip_versions_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.get_supported_mip_versions = method;
	BT_LIB_LOGD("Set source component class's \"get supported MIP versions\" method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_filter_set_get_supported_mip_versions_method(
		struct bt_component_class_filter *comp_cls,
		bt_component_class_filter_get_supported_mip_versions_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.get_supported_mip_versions = method;
	BT_LIB_LOGD("Set filter component class's \"get supported MIP versions\" method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_sink_set_get_supported_mip_versions_method(
		struct bt_component_class_sink *comp_cls,
		bt_component_class_sink_get_supported_mip_versions_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.get_supported_mip_versions = method;
	BT_LIB_LOGD("Set sink component class's \"get supported MIP versions\" method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_source_set_initialize_method(
		struct bt_component_class_source *comp_cls,
		bt_component_class_source_initialize_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.init = method;
	BT_LIB_LOGD("Set source component class's initialization method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_filter_set_initialize_method(
		struct bt_component_class_filter *comp_cls,
		bt_component_class_filter_initialize_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.init = method;
	BT_LIB_LOGD("Set filter component class's initialization method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_sink_set_initialize_method(
		struct bt_component_class_sink *comp_cls,
		bt_component_class_sink_initialize_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.init = method;
	BT_LIB_LOGD("Set sink component class's initialization method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_source_set_finalize_method(
		struct bt_component_class_source *comp_cls,
		bt_component_class_source_finalize_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.finalize = method;
	BT_LIB_LOGD("Set source component class's finalization method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_filter_set_finalize_method(
		struct bt_component_class_filter *comp_cls,
		bt_component_class_filter_finalize_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.finalize = method;
	BT_LIB_LOGD("Set filter component class's finalization method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_sink_set_finalize_method(
		struct bt_component_class_sink *comp_cls,
		bt_component_class_sink_finalize_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.finalize = method;
	BT_LIB_LOGD("Set sink component class's finalization method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_source_set_query_method(
		struct bt_component_class_source *comp_cls,
		bt_component_class_source_query_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.query = method;
	BT_LIB_LOGD("Set source component class's query method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_filter_set_query_method(
		struct bt_component_class_filter *comp_cls,
		bt_component_class_filter_query_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.query = method;
	BT_LIB_LOGD("Set filter component class's query method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_sink_set_query_method(
		struct bt_component_class_sink *comp_cls,
		bt_component_class_sink_query_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.query = method;
	BT_LIB_LOGD("Set sink component class's query method: "
		"%!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_filter_set_input_port_connected_method(
		struct bt_component_class_filter *comp_cls,
		bt_component_class_filter_input_port_connected_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.input_port_connected = method;
	BT_LIB_LOGD("Set filter component class's \"input port connected\" method"
		": %!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_sink_set_input_port_connected_method(
		struct bt_component_class_sink *comp_cls,
		bt_component_class_sink_input_port_connected_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.input_port_connected = method;
	BT_LIB_LOGD("Set sink component class's \"input port connected\" method"
		": %!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_source_set_output_port_connected_method(
		struct bt_component_class_source *comp_cls,
		bt_component_class_source_output_port_connected_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.output_port_connected = method;
	BT_LIB_LOGD("Set source component class's \"output port connected\" method"
		": %!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_filter_set_output_port_connected_method(
		struct bt_component_class_filter *comp_cls,
		bt_component_class_filter_output_port_connected_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.output_port_connected = method;
	BT_LIB_LOGD("Set filter component class's \"output port connected\" method"
		": %!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_method_status
bt_component_class_sink_set_graph_is_configured_method(
		struct bt_component_class_sink *comp_cls,
		bt_component_class_sink_graph_is_configured_method method)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_METHOD_NON_NULL(method);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	comp_cls->methods.graph_is_configured = method;
	BT_LIB_LOGD("Set sink component class's \"graph is configured\" method"
		": %!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_description_status
bt_component_class_set_description(
		struct bt_component_class *comp_cls,
		const char *description)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_DESCR_NON_NULL(description);
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	g_string_assign(comp_cls->description, description);
	BT_LIB_LOGD("Set component class's description: "
		"addr=%p, name=\"%s\", type=%s",
		comp_cls,
		bt_component_class_get_name(comp_cls),
		bt_common_component_class_type_string(comp_cls->type));
	return BT_FUNC_STATUS_OK;
}

enum bt_component_class_set_help_status bt_component_class_set_help(
		struct bt_component_class *comp_cls,
		const char *help)
{
	BT_ASSERT_PRE_NO_ERROR();
	BT_ASSERT_PRE_COMP_CLS_NON_NULL(comp_cls);
	BT_ASSERT_PRE_NON_NULL("help-text", help, "Help text");
	BT_ASSERT_PRE_DEV_COMP_CLS_HOT(comp_cls);
	g_string_assign(comp_cls->help, help);
	BT_LIB_LOGD("Set component class's help text: %!+C", comp_cls);
	return BT_FUNC_STATUS_OK;
}

const char *bt_component_class_get_name(const struct bt_component_class *comp_cls)
{
	BT_ASSERT_PRE_DEV_COMP_CLS_NON_NULL(comp_cls);
	return comp_cls->name->str;
}

enum bt_component_class_type bt_component_class_get_type(
		const struct bt_component_class *comp_cls)
{
	BT_ASSERT_PRE_DEV_COMP_CLS_NON_NULL(comp_cls);
	return comp_cls->type;
}

const char *bt_component_class_get_description(
		const struct bt_component_class *comp_cls)
{
	BT_ASSERT_PRE_DEV_COMP_CLS_NON_NULL(comp_cls);
	return comp_cls->description &&
		comp_cls->description->str[0] != '\0' ?
		comp_cls->description->str : NULL;
}

const char *bt_component_class_get_help(
		const struct bt_component_class *comp_cls)
{
	BT_ASSERT_PRE_DEV_COMP_CLS_NON_NULL(comp_cls);
	return comp_cls->help &&
		comp_cls->help->str[0] != '\0' ? comp_cls->help->str : NULL;
}

BT_HIDDEN
void bt_component_class_add_destroy_listener(
		struct bt_component_class *comp_cls,
		bt_component_class_destroy_listener_func func, void *data)
{
	struct bt_component_class_destroy_listener listener;

	BT_ASSERT(comp_cls);
	BT_ASSERT(func);
	listener.func = func;
	listener.data = data;
	g_array_append_val(comp_cls->destroy_listeners, listener);
	BT_LIB_LOGD("Added destroy listener to component class: "
		"%![cc-]+C, listener-func-addr=%p", comp_cls, func);
}

BT_HIDDEN
void _bt_component_class_freeze(const struct bt_component_class *comp_cls)
{
	BT_ASSERT(comp_cls);
	BT_LIB_LOGD("Freezing component class: %!+C", comp_cls);
	((struct bt_component_class *) comp_cls)->frozen = true;
}

void bt_component_class_get_ref(
		const struct bt_component_class *component_class)
{
	bt_object_get_ref(component_class);
}

void bt_component_class_put_ref(
		const struct bt_component_class *component_class)
{
	bt_object_put_ref(component_class);
}

void bt_component_class_source_get_ref(
		const struct bt_component_class_source *component_class_source)
{
	bt_object_get_ref(component_class_source);
}

void bt_component_class_source_put_ref(
		const struct bt_component_class_source *component_class_source)
{
	bt_object_put_ref(component_class_source);
}

void bt_component_class_filter_get_ref(
		const struct bt_component_class_filter *component_class_filter)
{
	bt_object_get_ref(component_class_filter);
}

void bt_component_class_filter_put_ref(
		const struct bt_component_class_filter *component_class_filter)
{
	bt_object_put_ref(component_class_filter);
}

void bt_component_class_sink_get_ref(
		const struct bt_component_class_sink *component_class_sink)
{
	bt_object_get_ref(component_class_sink);
}

void bt_component_class_sink_put_ref(
		const struct bt_component_class_sink *component_class_sink)
{
	bt_object_put_ref(component_class_sink);
}