[babeltrace.git] / src / 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 "LIB/COMPONENT"
#include "lib/logging.h"

#include "common/common.h"
#include "common/assert.h"
#include "lib/assert-pre.h"
#include "lib/assert-post.h"
#include <babeltrace2/graph/self-component.h>
#include <babeltrace2/graph/component-const.h>
#include <babeltrace2/graph/component-source-const.h>
#include <babeltrace2/graph/component-filter-const.h>
#include <babeltrace2/graph/component-sink-const.h>
#include <babeltrace2/graph/graph-const.h>
#include "common/macros.h"
#include "compat/compiler.h"
#include <babeltrace2/types.h>
#include <babeltrace2/value.h>
#include "lib/value.h"
#include <stdint.h>
#include <inttypes.h>

#include "component.h"
#include "component-class.h"
#include "component-source.h"
#include "component-filter.h"
#include "component-sink.h"
#include "connection.h"
#include "graph.h"
#include "message/iterator.h"
#include "port.h"
#include "lib/func-status.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_LOGI("Calling user's component 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_LOGI("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_DEV_NON_NULL(component, "Component");
        return component->class->type;
}

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

        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->config_state == BT_GRAPH_CONFIGURATION_STATE_CONFIGURING,
                "Component's graph is already configured: "
                "%![comp-]+c, %![graph-]+g", component, graph);

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

        BT_LIB_LOGI("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_LIB_LOGE_APPEND_CAUSE("Cannot create port object.");
                status = BT_FUNC_STATUS_MEMORY_ERROR;
                goto error;
        }

        /*
         * 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.
         */
        graph = bt_component_borrow_graph(component);
        if (graph) {
                enum bt_graph_listener_func_status listener_status;

                listener_status = bt_graph_notify_port_added(graph, new_port);
                if (listener_status != BT_FUNC_STATUS_OK) {
                        bt_graph_make_faulty(graph);
                        status = (int) listener_status;
                        goto error;
                }
        }

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

        *port = new_port;
        status = BT_FUNC_STATUS_OK;

        goto end;
error:
        /*
         * We need to release the reference that we would otherwise have
         * returned to the caller.
         */
        BT_PORT_PUT_REF_AND_RESET(new_port);

end:
        return status;
}

BT_HIDDEN
uint64_t bt_component_get_input_port_count(const struct bt_component *comp)
{
        BT_ASSERT_PRE_DEV_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_DEV_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, bt_logging_level log_level,
                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_LOGI("Creating empty component from component class: %![cc-]+C, "
                "comp-name=\"%s\", log-level=%s", component_class, name,
                bt_common_logging_level_string(log_level));
        component = component_create_funcs[type](component_class);
        if (!component) {
                BT_LIB_LOGE_APPEND_CAUSE(
                        "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_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GString.");
                ret = -1;
                goto end;
        }

        component->log_level = log_level;
        component->input_ports = g_ptr_array_new_with_free_func(
                (GDestroyNotify) bt_object_try_spec_release);
        if (!component->input_ports) {
                BT_LIB_LOGE_APPEND_CAUSE("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_LIB_LOGE_APPEND_CAUSE("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_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GArray.");
                ret = -1;
                goto end;
        }

        BT_LIB_LOGI("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_DEV_NON_NULL(component, "Component");
        return component->name->str;
}

const struct bt_component_class *bt_component_borrow_class_const(
                const struct bt_component *component)
{
        BT_ASSERT_PRE_DEV_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_DEV_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_DEV_NON_NULL(component, "Component");
        component->user_data = data;
        BT_LIB_LOGD("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);
}

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) == 0) {
                        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_DEV_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_DEV_NON_NULL(comp, "Component");
        BT_ASSERT_PRE_DEV_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_DEV_NON_NULL(comp, "Component");
        BT_ASSERT_PRE_DEV_VALID_INDEX(index, comp->output_ports->len);
        return (void *)
                borrow_port_by_index(comp->output_ports, index);
}

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

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

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

        enum bt_component_class_port_connected_method_status status =
                BT_FUNC_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 = (int) method(comp, self_port, (void *) other_port);
                BT_LOGD("User method returned: status=%s",
                        bt_common_func_status_string(status));
                BT_ASSERT_POST(status == BT_FUNC_STATUS_OK ||
                        status == BT_FUNC_STATUS_ERROR ||
                        status == BT_FUNC_STATUS_MEMORY_ERROR,
                        "Unexpected returned component status: status=%s",
                        bt_common_func_status_string(status));
        }

        return status;
}

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_LOGD("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_LOGD("Removed destroy listener: %![comp-]+c, "
                                "func-addr=%p, data-addr=%p",
                                component, func, data);
                }
        }
}

bt_logging_level bt_component_get_logging_level(
                const struct bt_component *component)
{
        BT_ASSERT_PRE_DEV_NON_NULL(component, "Component");
        return component->log_level;
}

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