src.ctf.fs: manage lifetime of open file with FileUP

[babeltrace.git] / src / plugins / ctf / fs-src / fs.hpp

/*
 * SPDX-License-Identifier: MIT
 *
 * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 * Copyright 2016 Philippe Proulx <pproulx@efficios.com>
 *
 * BabelTrace - CTF on File System Component
 */

#ifndef BABELTRACE_PLUGIN_CTF_FS_H
#define BABELTRACE_PLUGIN_CTF_FS_H

#include <glib.h>

#include <babeltrace2/babeltrace.h>

#include "cpp-common/bt2c/glib-up.hpp"
#include "cpp-common/bt2c/logging.hpp"

#include "metadata.hpp"

extern bool ctf_fs_debug;

struct ctf_fs_file
{
    explicit ctf_fs_file(const bt2c::Logger& parentLogger) :
        logger {parentLogger, "PLUGIN/SRC.CTF.FS/FILE"}
    {
    }

    bt2c::Logger logger;

    /* Owned by this */
    GString *path = nullptr;

    /* Owned by this */
    FILE *fp = nullptr;

    off_t size = 0;
};

struct ctf_fs_metadata
{
    /* Owned by this */
    struct ctf_metadata_decoder *decoder = nullptr;

    /* Owned by this */
    bt_trace_class *trace_class = nullptr;

    /* Weak (owned by `decoder` above) */
    struct ctf_trace_class *tc = nullptr;

    /* Owned by this */
    char *text = nullptr;

    int bo = 0;
};

struct ctf_fs_component_deleter
{
    void operator()(struct ctf_fs_component *);
};

struct ctf_fs_component
{
    using UP = std::unique_ptr<ctf_fs_component, ctf_fs_component_deleter>;

    explicit ctf_fs_component(const bt2c::Logger& parentLogger) :
        logger {parentLogger, "PLUGIN/SRC.CTF.FS/COMP"}
    {
    }

    bt2c::Logger logger;

    /* Array of struct ctf_fs_port_data *, owned by this */
    GPtrArray *port_data = nullptr;

    /* Owned by this */
    struct ctf_fs_trace *trace = nullptr;

    struct ctf_fs_metadata_config metadata_config;
};

struct ctf_fs_trace
{
    explicit ctf_fs_trace(const bt2c::Logger& parentLogger) :
        logger {parentLogger, "PLUGIN/SRC.CTF.FS/TRACE"}
    {
    }

    bt2c::Logger logger;

    /* Owned by this */
    struct ctf_fs_metadata *metadata = nullptr;

    /* Owned by this */
    bt_trace *trace = nullptr;

    /* Array of struct ctf_fs_ds_file_group *, owned by this */
    GPtrArray *ds_file_groups = nullptr;

    /* Owned by this */
    GString *path = nullptr;

    /* Next automatic stream ID when not provided by packet header */
    uint64_t next_stream_id = 0;
};

struct ctf_fs_ds_index_entry
{
    /* Weak, belongs to ctf_fs_ds_file_info. */
    const char *path = nullptr;

    /* Position, in bytes, of the packet from the beginning of the file. */
    uint64_t offset = 0;

    /* Size of the packet, in bytes. */
    uint64_t packet_size = 0;

    /*
     * Extracted from the packet context, relative to the respective fields'
     * mapped clock classes (in cycles).
     */
    uint64_t timestamp_begin = 0, timestamp_end = 0;

    /*
     * Converted from the packet context, relative to the trace's EPOCH
     * (in ns since EPOCH).
     */
    int64_t timestamp_begin_ns = 0, timestamp_end_ns = 0;

    /*
     * Packet sequence number, or UINT64_MAX if not present in the index.
     */
    uint64_t packet_seq_num = 0;
};

struct ctf_fs_ds_index
{
    /* Array of pointer to struct ctf_fs_ds_index_entry. */
    GPtrArray *entries = nullptr;
};

struct ctf_fs_ds_file_group
{
    /*
     * Array of struct ctf_fs_ds_file_info, owned by this.
     *
     * This is an _ordered_ array of data stream file infos which
     * belong to this group (a single stream instance).
     *
     * You can call ctf_fs_ds_file_create() with one of those paths
     * and the trace IR stream below.
     */
    GPtrArray *ds_file_infos = nullptr;

    /* Owned by this */
    struct ctf_stream_class *sc = nullptr;

    /* Owned by this */
    bt_stream *stream = nullptr;

    /* Stream (instance) ID; -1ULL means none */
    uint64_t stream_id = 0;

    /* Weak, belongs to component */
    struct ctf_fs_trace *ctf_fs_trace = nullptr;

    /*
     * Owned by this.
     */
    struct ctf_fs_ds_index *index = nullptr;
};

struct ctf_fs_port_data
{
    /* Weak, belongs to ctf_fs_trace */
    struct ctf_fs_ds_file_group *ds_file_group = nullptr;

    /* Weak */
    struct ctf_fs_component *ctf_fs = nullptr;
};

struct ctf_fs_msg_iter_data
{
    explicit ctf_fs_msg_iter_data(bt_self_message_iterator *selfMsgIter) :
        self_msg_iter {selfMsgIter}, logger {bt2::SelfMessageIterator {self_msg_iter},
                                             "PLUGIN/SRC.CTF.FS/MSG-ITER"}
    {
    }

    /* Weak */
    bt_self_message_iterator *self_msg_iter = nullptr;

    bt2c::Logger logger;

    /* Weak, belongs to ctf_fs_trace */
    struct ctf_fs_ds_file_group *ds_file_group = nullptr;

    /* Owned by this */
    struct ctf_msg_iter *msg_iter = nullptr;

    /*
     * Saved error.  If we hit an error in the _next method, but have some
     * messages ready to return, we save the error here and return it on
     * the next _next call.
     */
    bt_message_iterator_class_next_method_status next_saved_status =
        BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK;
    const struct bt_error *next_saved_error = nullptr;

    struct ctf_fs_ds_group_medops_data *msg_iter_medops_data = nullptr;
};

bt_component_class_initialize_method_status
ctf_fs_init(bt_self_component_source *source, bt_self_component_source_configuration *config,
            const bt_value *params, void *init_method_data);

void ctf_fs_finalize(bt_self_component_source *component);

bt_component_class_query_method_status ctf_fs_query(bt_self_component_class_source *comp_class,
                                                    bt_private_query_executor *priv_query_exec,
                                                    const char *object, const bt_value *params,
                                                    void *method_data, const bt_value **result);

bt_message_iterator_class_initialize_method_status
ctf_fs_iterator_init(bt_self_message_iterator *self_msg_iter,
                     bt_self_message_iterator_configuration *config,
                     bt_self_component_port_output *self_port);

void ctf_fs_iterator_finalize(bt_self_message_iterator *it);

bt_message_iterator_class_next_method_status
ctf_fs_iterator_next(bt_self_message_iterator *iterator, bt_message_array_const msgs,
                     uint64_t capacity, uint64_t *count);

bt_message_iterator_class_seek_beginning_method_status
ctf_fs_iterator_seek_beginning(bt_self_message_iterator *message_iterator);

/* Create and initialize a new, empty ctf_fs_component. */

ctf_fs_component::UP ctf_fs_component_create(const bt2c::Logger& parentLogger);

/*
 * Create one `struct ctf_fs_trace` from one trace, or multiple traces sharing
 * the same UUID.
 *
 * `paths_value` must be an array of strings,
 *
 * The created `struct ctf_fs_trace` is assigned to `ctf_fs->trace`.
 *
 * `self_comp` and `self_comp_class` are used for logging, only one of them
 * should be set.
 */

int ctf_fs_component_create_ctf_fs_trace(struct ctf_fs_component *ctf_fs,
                                         const bt_value *paths_value,
                                         const bt_value *trace_name_value,
                                         bt_self_component *selfComp);

/* Free `ctf_fs` and everything it owns. */

void ctf_fs_destroy(struct ctf_fs_component *ctf_fs);

/*
 * Read and validate parameters taken by the src.ctf.fs plugin.
 *
 *  - The mandatory `paths` parameter is returned in `*paths`.
 *  - The optional `clock-class-offset-s` and `clock-class-offset-ns`, if
 *    present, are recorded in the `ctf_fs` structure.
 *  - The optional `trace-name` parameter is returned in `*trace_name` if
 *    present, else `*trace_name` is set to NULL.
 *
 * `self_comp` and `self_comp_class` are used for logging, only one of them
 * should be set.
 *
 * Return true on success, false if any parameter didn't pass validation.
 */

bool read_src_fs_parameters(const bt_value *params, const bt_value **paths,
                            const bt_value **trace_name, struct ctf_fs_component *ctf_fs);

/*
 * Generate the port name to be used for a given data stream file group.
 */

bt2c::GCharUP ctf_fs_make_port_name(struct ctf_fs_ds_file_group *ds_file_group);

#endif /* BABELTRACE_PLUGIN_CTF_FS_H */