ctf.fs source: use last path parameter's element as base for trace names
[babeltrace.git] / plugins / ctf / fs-src / fs.c
index 6fc53e3c30da2f47f917e09c735defdbdd03b2eb..0e199e92e773fbc165af731c6c2e9b28a532e9db 100644 (file)
  * SOFTWARE.
  */
 
+#include <babeltrace/common-internal.h>
 #include <babeltrace/ctf-ir/packet.h>
 #include <babeltrace/ctf-ir/clock-class.h>
+#include <babeltrace/ctf-ir/stream.h>
+#include <babeltrace/ctf-ir/fields.h>
 #include <babeltrace/graph/private-port.h>
 #include <babeltrace/graph/private-component.h>
 #include <babeltrace/graph/private-component-source.h>
 #include <plugins-common.h>
 #include <glib.h>
 #include <assert.h>
+#include <inttypes.h>
 #include <stdbool.h>
 #include <unistd.h>
 #include "fs.h"
 #include "metadata.h"
-#include "data-stream.h"
+#include "data-stream-file.h"
 #include "file.h"
 #include "../common/metadata/decoder.h"
 
 BT_HIDDEN
 bool ctf_fs_debug;
 
+static
+int notif_iter_data_set_current_ds_file(struct ctf_fs_notif_iter_data *notif_iter_data)
+{
+       struct ctf_fs_ds_file_info *ds_file_info;
+       int ret = 0;
+
+       assert(notif_iter_data->ds_file_info_index <
+               notif_iter_data->ds_file_group->ds_file_infos->len);
+       ds_file_info = g_ptr_array_index(
+               notif_iter_data->ds_file_group->ds_file_infos,
+               notif_iter_data->ds_file_info_index);
+
+       ctf_fs_ds_file_destroy(notif_iter_data->ds_file);
+       notif_iter_data->ds_file = ctf_fs_ds_file_create(
+               notif_iter_data->ds_file_group->ctf_fs_trace,
+               notif_iter_data->ds_file_group->stream,
+               ds_file_info->path->str, true);
+       if (!notif_iter_data->ds_file) {
+               ret = -1;
+       }
+
+       return ret;
+}
+
+static
+void ctf_fs_notif_iter_data_destroy(
+               struct ctf_fs_notif_iter_data *notif_iter_data)
+{
+       if (!notif_iter_data) {
+               return;
+       }
+
+       ctf_fs_ds_file_destroy(notif_iter_data->ds_file);
+       g_free(notif_iter_data);
+}
+
 struct bt_notification_iterator_next_return ctf_fs_iterator_next(
                struct bt_private_notification_iterator *iterator)
 {
-       struct ctf_fs_stream *fs_stream =
+       struct bt_notification_iterator_next_return next_ret;
+       struct ctf_fs_notif_iter_data *notif_iter_data =
                bt_private_notification_iterator_get_user_data(iterator);
+       int ret;
 
-       return ctf_fs_stream_next(fs_stream);
+       assert(notif_iter_data->ds_file);
+       next_ret = ctf_fs_ds_file_next(notif_iter_data->ds_file);
+       if (next_ret.status == BT_NOTIFICATION_ITERATOR_STATUS_END) {
+               assert(!next_ret.notification);
+               notif_iter_data->ds_file_info_index++;
+
+               if (notif_iter_data->ds_file_info_index ==
+                               notif_iter_data->ds_file_group->ds_file_infos->len) {
+                       /*
+                        * No more stream files to read: we reached the
+                        * real end.
+                        */
+                       goto end;
+               }
+
+               /*
+                * Open and start reading the next stream file within
+                * our stream file group.
+                */
+               ret = notif_iter_data_set_current_ds_file(notif_iter_data);
+               if (ret) {
+                       next_ret.status = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
+                       goto end;
+               }
+
+               next_ret = ctf_fs_ds_file_next(notif_iter_data->ds_file);
+
+               /*
+                * We should not get BT_NOTIFICATION_ITERATOR_STATUS_END
+                * with a brand new stream file because empty stream
+                * files are not even part of stream file groups, which
+                * means we're sure to get at least one pair of "packet
+                * begin" and "packet end" notifications in the case of
+                * a single, empty packet.
+                */
+               assert(next_ret.status != BT_NOTIFICATION_ITERATOR_STATUS_END);
+       }
+
+end:
+       return next_ret;
 }
 
 void ctf_fs_iterator_finalize(struct bt_private_notification_iterator *it)
 {
-       void *ctf_fs_stream =
+       void *notif_iter_data =
                bt_private_notification_iterator_get_user_data(it);
 
-       ctf_fs_stream_destroy(ctf_fs_stream);
+       ctf_fs_notif_iter_data_destroy(notif_iter_data);
 }
 
 enum bt_notification_iterator_status ctf_fs_iterator_init(
                struct bt_private_notification_iterator *it,
                struct bt_private_port *port)
 {
-       struct ctf_fs_stream *ctf_fs_stream = NULL;
        struct ctf_fs_port_data *port_data;
+       struct ctf_fs_notif_iter_data *notif_iter_data = NULL;
        enum bt_notification_iterator_status ret =
                BT_NOTIFICATION_ITERATOR_STATUS_OK;
+       int iret;
 
        port_data = bt_private_port_get_user_data(port);
        if (!port_data) {
@@ -86,18 +168,25 @@ enum bt_notification_iterator_status ctf_fs_iterator_init(
                goto error;
        }
 
-       ctf_fs_stream = ctf_fs_stream_create(port_data->ctf_fs_trace,
-               port_data->path->str);
-       if (!ctf_fs_stream) {
+       notif_iter_data = g_new0(struct ctf_fs_notif_iter_data, 1);
+       if (!notif_iter_data) {
+               ret = BT_NOTIFICATION_ITERATOR_STATUS_NOMEM;
                goto error;
        }
 
-       ret = bt_private_notification_iterator_set_user_data(it, ctf_fs_stream);
+       notif_iter_data->ds_file_group = port_data->ds_file_group;
+       iret = notif_iter_data_set_current_ds_file(notif_iter_data);
+       if (iret) {
+               ret = BT_NOTIFICATION_ITERATOR_STATUS_ERROR;
+               goto error;
+       }
+
+       ret = bt_private_notification_iterator_set_user_data(it, notif_iter_data);
        if (ret) {
                goto error;
        }
 
-       ctf_fs_stream = NULL;
+       notif_iter_data = NULL;
        goto end;
 
 error:
@@ -108,7 +197,7 @@ error:
        }
 
 end:
-       ctf_fs_stream_destroy(ctf_fs_stream);
+       ctf_fs_notif_iter_data_destroy(notif_iter_data);
        return ret;
 }
 
@@ -139,6 +228,10 @@ void ctf_fs_trace_destroy(void *data)
                return;
        }
 
+       if (ctf_fs_trace->ds_file_groups) {
+               g_ptr_array_free(ctf_fs_trace->ds_file_groups, TRUE);
+       }
+
        if (ctf_fs_trace->path) {
                g_string_free(ctf_fs_trace->path, TRUE);
        }
@@ -171,30 +264,34 @@ void port_data_destroy(void *data) {
                return;
        }
 
-       if (port_data->path) {
-               g_string_free(port_data->path, TRUE);
-       }
-
        g_free(port_data);
 }
 
 static
 int create_one_port_for_trace(struct ctf_fs_trace *ctf_fs_trace,
-               const char *stream_path)
+               struct ctf_fs_ds_file_group *ds_file_group)
 {
        int ret = 0;
        struct bt_private_port *port = NULL;
        struct ctf_fs_port_data *port_data = NULL;
        GString *port_name = NULL;
        struct ctf_fs_component *ctf_fs = ctf_fs_trace->ctf_fs;
+       struct ctf_fs_ds_file_info *ds_file_info =
+               g_ptr_array_index(ds_file_group->ds_file_infos, 0);
 
        port_name = g_string_new(NULL);
        if (!port_name) {
                goto error;
        }
 
-       /* Assign the name for the new output port */
-       g_string_printf(port_name, "%s", stream_path);
+       /*
+        * Assign the name for the new output port. If there's more than
+        * one stream file in the stream file group, the first
+        * (earliest) stream file's path is used.
+        */
+       assert(ds_file_group->ds_file_infos->len > 0);
+       ds_file_info = g_ptr_array_index(ds_file_group->ds_file_infos, 0);
+       g_string_assign(port_name, ds_file_info->path->str);
        PDBG("Creating one port named `%s`\n", port_name->str);
 
        /* Create output port for this file */
@@ -203,12 +300,7 @@ int create_one_port_for_trace(struct ctf_fs_trace *ctf_fs_trace,
                goto error;
        }
 
-       port_data->ctf_fs_trace = ctf_fs_trace;
-       port_data->path = g_string_new(stream_path);
-       if (!port_data->path) {
-               goto error;
-       }
-
+       port_data->ds_file_group = ds_file_group;
        port = bt_private_component_source_add_output_private_port(
                ctf_fs->priv_comp, port_name->str, port_data);
        if (!port) {
@@ -234,15 +326,434 @@ end:
 
 static
 int create_ports_for_trace(struct ctf_fs_trace *ctf_fs_trace)
+{
+       int ret = 0;
+       struct ctf_fs_component *ctf_fs = ctf_fs_trace->ctf_fs;
+       size_t i;
+
+       /* Create one output port for each stream file group */
+       for (i = 0; i < ctf_fs_trace->ds_file_groups->len; i++) {
+               struct ctf_fs_ds_file_group *ds_file_group =
+                       g_ptr_array_index(ctf_fs_trace->ds_file_groups, i);
+
+               ret = create_one_port_for_trace(ctf_fs_trace, ds_file_group);
+               if (ret) {
+                       PERR("Cannot create output port.\n");
+                       goto end;
+               }
+       }
+
+end:
+       return ret;
+}
+
+uint64_t get_packet_header_stream_instance_id(struct ctf_fs_trace *ctf_fs_trace,
+               struct bt_ctf_field *packet_header_field)
+{
+       struct bt_ctf_field *stream_instance_id_field = NULL;
+       uint64_t stream_instance_id = -1ULL;
+       int ret;
+
+       if (!packet_header_field) {
+               goto end;
+       }
+
+       stream_instance_id_field = bt_ctf_field_structure_get_field_by_name(
+               packet_header_field, "stream_instance_id");
+       if (!stream_instance_id_field) {
+               goto end;
+       }
+
+       ret = bt_ctf_field_unsigned_integer_get_value(stream_instance_id_field,
+               &stream_instance_id);
+       if (ret) {
+               stream_instance_id = -1ULL;
+               goto end;
+       }
+
+end:
+       bt_put(stream_instance_id_field);
+       return stream_instance_id;
+}
+
+struct bt_ctf_stream_class *stream_class_from_packet_header(
+               struct ctf_fs_trace *ctf_fs_trace,
+               struct bt_ctf_field *packet_header_field)
+{
+       struct bt_ctf_field *stream_id_field = NULL;
+       struct bt_ctf_stream_class *stream_class = NULL;
+       uint64_t stream_id = -1ULL;
+       int ret;
+
+       if (!packet_header_field) {
+               goto single_stream_class;
+       }
+
+       stream_id_field = bt_ctf_field_structure_get_field_by_name(
+               packet_header_field, "stream_id");
+       if (!stream_id_field) {
+               goto end;
+       }
+
+       ret = bt_ctf_field_unsigned_integer_get_value(stream_id_field,
+               &stream_id);
+       if (ret) {
+               stream_id = -1ULL;
+       }
+
+       if (stream_id == -1ULL) {
+single_stream_class:
+               /* Single stream class */
+               if (bt_ctf_trace_get_stream_class_count(
+                               ctf_fs_trace->metadata->trace) == 0) {
+                       goto end;
+               }
+
+               stream_class = bt_ctf_trace_get_stream_class_by_index(
+                       ctf_fs_trace->metadata->trace, 0);
+       } else {
+               stream_class = bt_ctf_trace_get_stream_class_by_id(
+                       ctf_fs_trace->metadata->trace, stream_id);
+       }
+
+end:
+       bt_put(stream_id_field);
+       return stream_class;
+}
+
+uint64_t get_packet_context_timestamp_begin_ns(
+               struct ctf_fs_trace *ctf_fs_trace,
+               struct bt_ctf_field *packet_context_field)
+{
+       int ret;
+       struct bt_ctf_field *timestamp_begin_field = NULL;
+       struct bt_ctf_field_type *timestamp_begin_ft = NULL;
+       uint64_t timestamp_begin_raw_value = -1ULL;
+       uint64_t timestamp_begin_ns = -1ULL;
+       int64_t timestamp_begin_ns_signed;
+       struct bt_ctf_clock_class *timestamp_begin_clock_class = NULL;
+       struct bt_ctf_clock_value *clock_value = NULL;
+
+       if (!packet_context_field) {
+               goto end;
+       }
+
+       timestamp_begin_field = bt_ctf_field_structure_get_field_by_name(
+               packet_context_field, "timestamp_begin");
+       if (!timestamp_begin_field) {
+               goto end;
+       }
+
+       timestamp_begin_ft = bt_ctf_field_get_type(timestamp_begin_field);
+       assert(timestamp_begin_ft);
+       timestamp_begin_clock_class =
+               bt_ctf_field_type_integer_get_mapped_clock_class(timestamp_begin_ft);
+       if (!timestamp_begin_clock_class) {
+               goto end;
+       }
+
+       ret = bt_ctf_field_unsigned_integer_get_value(timestamp_begin_field,
+               &timestamp_begin_raw_value);
+       if (ret) {
+               goto end;
+       }
+
+       clock_value = bt_ctf_clock_value_create(timestamp_begin_clock_class,
+               timestamp_begin_raw_value);
+       if (!clock_value) {
+               goto end;
+       }
+
+       ret = bt_ctf_clock_value_get_value_ns_from_epoch(clock_value,
+               &timestamp_begin_ns_signed);
+       if (ret) {
+               goto end;
+       }
+
+       timestamp_begin_ns = (uint64_t) timestamp_begin_ns_signed;
+
+end:
+       bt_put(timestamp_begin_field);
+       bt_put(timestamp_begin_ft);
+       bt_put(timestamp_begin_clock_class);
+       bt_put(clock_value);
+       return timestamp_begin_ns;
+}
+
+static
+void ctf_fs_ds_file_info_destroy(struct ctf_fs_ds_file_info *ds_file_info)
+{
+       if (!ds_file_info) {
+               return;
+       }
+
+       if (ds_file_info->path) {
+               g_string_free(ds_file_info->path, TRUE);
+       }
+
+       g_free(ds_file_info);
+}
+
+static
+struct ctf_fs_ds_file_info *ctf_fs_ds_file_info_create(const char *path,
+               uint64_t begin_ns)
+{
+       struct ctf_fs_ds_file_info *ds_file_info;
+
+       ds_file_info = g_new0(struct ctf_fs_ds_file_info, 1);
+       if (!ds_file_info) {
+               goto end;
+       }
+
+       ds_file_info->path = g_string_new(path);
+       if (!ds_file_info->path) {
+               ctf_fs_ds_file_info_destroy(ds_file_info);
+               ds_file_info = NULL;
+               goto end;
+       }
+
+       ds_file_info->begin_ns = begin_ns;
+
+end:
+       return ds_file_info;
+}
+
+static
+void ctf_fs_ds_file_group_destroy(struct ctf_fs_ds_file_group *ds_file_group)
+{
+       if (!ds_file_group) {
+               return;
+       }
+
+       if (ds_file_group->ds_file_infos) {
+               g_ptr_array_free(ds_file_group->ds_file_infos, TRUE);
+       }
+
+       bt_put(ds_file_group->stream);
+       g_free(ds_file_group);
+}
+
+static
+struct ctf_fs_ds_file_group *ctf_fs_ds_file_group_create(
+               struct ctf_fs_trace *ctf_fs_trace,
+               struct bt_ctf_stream_class *stream_class,
+               uint64_t stream_instance_id)
+{
+       struct ctf_fs_ds_file_group *ds_file_group;
+
+       assert(stream_class);
+       ds_file_group = g_new0(struct ctf_fs_ds_file_group, 1);
+       if (!ds_file_group) {
+               goto error;
+       }
+
+       ds_file_group->ds_file_infos = g_ptr_array_new_with_free_func(
+               (GDestroyNotify) ctf_fs_ds_file_info_destroy);
+       if (!ds_file_group->ds_file_infos) {
+               goto error;
+       }
+
+       if (stream_instance_id == -1ULL) {
+               ds_file_group->stream = bt_ctf_stream_create(
+                       stream_class, NULL);
+       } else {
+               ds_file_group->stream = bt_ctf_stream_create_with_id(
+                       stream_class, NULL, stream_instance_id);
+       }
+
+       if (!ds_file_group->stream) {
+               goto error;
+       }
+
+       ds_file_group->ctf_fs_trace = ctf_fs_trace;
+
+       goto end;
+
+error:
+       ctf_fs_ds_file_group_destroy(ds_file_group);
+       ds_file_group = NULL;
+
+end:
+       return ds_file_group;
+}
+
+static
+int ctf_fs_ds_file_group_add_ds_file_info(
+               struct ctf_fs_ds_file_group *ds_file_group,
+               const char *path, uint64_t begin_ns)
+{
+       struct ctf_fs_ds_file_info *ds_file_info;
+       gint i = 0;
+       int ret = 0;
+
+       ds_file_info = ctf_fs_ds_file_info_create(path, begin_ns);
+       if (!ds_file_info) {
+               goto error;
+       }
+
+       /* Find a spot to insert this one */
+       for (i = 0; i < ds_file_group->ds_file_infos->len; i++) {
+               struct ctf_fs_ds_file_info *other_ds_file_info =
+                       g_ptr_array_index(ds_file_group->ds_file_infos, i);
+
+               if (begin_ns < other_ds_file_info->begin_ns) {
+                       break;
+               }
+       }
+
+       if (i == ds_file_group->ds_file_infos->len) {
+               /* Append instead */
+               i = -1;
+       }
+
+       g_ptr_array_insert(ds_file_group->ds_file_infos, i, ds_file_info);
+       ds_file_info = NULL;
+       goto end;
+
+error:
+       ctf_fs_ds_file_info_destroy(ds_file_info);
+       ret = -1;
+
+end:
+       return ret;
+}
+
+static
+int add_ds_file_to_ds_file_group(struct ctf_fs_trace *ctf_fs_trace,
+               const char *path)
+{
+       struct bt_ctf_field *packet_header_field = NULL;
+       struct bt_ctf_field *packet_context_field = NULL;
+       struct bt_ctf_stream_class *stream_class = NULL;
+       struct ctf_fs_component *ctf_fs = ctf_fs_trace->ctf_fs;
+       uint64_t stream_instance_id = -1ULL;
+       uint64_t begin_ns = -1ULL;
+       struct ctf_fs_ds_file_group *ds_file_group = NULL;
+       bool add_group = false;
+       int ret;
+       size_t i;
+
+       ret = ctf_fs_ds_file_get_packet_header_context_fields(
+               ctf_fs_trace, path, &packet_header_field,
+               &packet_context_field);
+       if (ret) {
+               PERR("Cannot get stream file's first packet's header and context fields (`%s`).\n",
+                       path);
+               goto error;
+       }
+
+       stream_instance_id = get_packet_header_stream_instance_id(ctf_fs_trace,
+               packet_header_field);
+       begin_ns = get_packet_context_timestamp_begin_ns(ctf_fs_trace,
+               packet_context_field);
+       stream_class = stream_class_from_packet_header(ctf_fs_trace,
+               packet_header_field);
+       if (!stream_class) {
+               goto error;
+       }
+
+       if (begin_ns == -1ULL) {
+               /*
+                * No beggining timestamp to sort the stream files
+                * within a stream file group, so consider that this
+                * file must be the only one within its group.
+                */
+               stream_instance_id = -1ULL;
+       }
+
+       if (stream_instance_id == -1ULL) {
+               /*
+                * No stream instance ID or no beginning timestamp:
+                * create a unique stream file group for this stream
+                * file because, even if there's a stream instance ID,
+                * there's no timestamp to order the file within its
+                * group.
+                */
+
+               ds_file_group = ctf_fs_ds_file_group_create(ctf_fs_trace,
+                       stream_class, stream_instance_id);
+               if (!ds_file_group) {
+                       goto error;
+               }
+
+               ret = ctf_fs_ds_file_group_add_ds_file_info(ds_file_group,
+                               path, begin_ns);
+               if (ret) {
+                       goto error;
+               }
+
+               add_group = true;
+               goto end;
+       }
+
+       assert(stream_instance_id != -1ULL);
+       assert(begin_ns != -1ULL);
+
+       /* Find an existing stream file group with this ID */
+       for (i = 0; i < ctf_fs_trace->ds_file_groups->len; i++) {
+               int64_t id;
+               struct bt_ctf_stream_class *cand_stream_class;
+
+               ds_file_group = g_ptr_array_index(
+                       ctf_fs_trace->ds_file_groups, i);
+               id = bt_ctf_stream_get_id(ds_file_group->stream);
+               cand_stream_class = bt_ctf_stream_get_class(
+                       ds_file_group->stream);
+
+               assert(cand_stream_class);
+               bt_put(cand_stream_class);
+
+               if (cand_stream_class == stream_class &&
+                               (uint64_t) id == stream_instance_id) {
+                       break;
+               }
+
+               ds_file_group = NULL;
+       }
+
+       if (!ds_file_group) {
+               ds_file_group = ctf_fs_ds_file_group_create(ctf_fs_trace,
+                       stream_class, stream_instance_id);
+               if (!ds_file_group) {
+                       goto error;
+               }
+
+               add_group = true;
+       }
+
+       ret = ctf_fs_ds_file_group_add_ds_file_info(ds_file_group,
+                       path, begin_ns);
+       if (ret) {
+               goto error;
+       }
+
+       goto end;
+
+error:
+       ctf_fs_ds_file_group_destroy(ds_file_group);
+       ret = -1;
+
+end:
+       if (add_group && ds_file_group) {
+               g_ptr_array_add(ctf_fs_trace->ds_file_groups, ds_file_group);
+       }
+
+       bt_put(packet_header_field);
+       bt_put(packet_context_field);
+       bt_put(stream_class);
+       return ret;
+}
+
+static
+int create_ds_file_groups(struct ctf_fs_trace *ctf_fs_trace)
 {
        int ret = 0;
        const char *basename;
        GError *error = NULL;
        GDir *dir = NULL;
-       struct ctf_fs_file *file = NULL;
        struct ctf_fs_component *ctf_fs = ctf_fs_trace->ctf_fs;
 
-       /* Create one output port for each stream file */
+       /* Check each file in the path directory, except specific ones */
        dir = g_dir_open(ctf_fs_trace->path->str, 0, &error);
        if (!dir) {
                PERR("Cannot open directory `%s`: %s (code %d)\n",
@@ -252,6 +763,8 @@ int create_ports_for_trace(struct ctf_fs_trace *ctf_fs_trace)
        }
 
        while ((basename = g_dir_read_name(dir))) {
+               struct ctf_fs_file *file;
+
                if (!strcmp(basename, CTF_FS_METADATA_FILENAME)) {
                        /* Ignore the metadata stream. */
                        PDBG("Ignoring metadata file `%s/%s`\n",
@@ -294,19 +807,19 @@ int create_ports_for_trace(struct ctf_fs_trace *ctf_fs_trace)
                        /* Skip empty stream. */
                        PDBG("Ignoring empty file `%s`\n", file->path->str);
                        ctf_fs_file_destroy(file);
-                       file = NULL;
                        continue;
                }
 
-               ret = create_one_port_for_trace(ctf_fs_trace, file->path->str);
+               ret = add_ds_file_to_ds_file_group(ctf_fs_trace,
+                       file->path->str);
                if (ret) {
-                       PERR("Cannot create output port for file `%s`\n",
+                       PDBG("Cannot add stream file `%s` to stream file group\n",
                                file->path->str);
+                       ctf_fs_file_destroy(file);
                        goto error;
                }
 
                ctf_fs_file_destroy(file);
-               file = NULL;
        }
 
        goto end;
@@ -324,7 +837,6 @@ end:
                g_error_free(error);
        }
 
-       ctf_fs_file_destroy(file);
        return ret;
 }
 
@@ -393,11 +905,22 @@ struct ctf_fs_trace *ctf_fs_trace_create(struct ctf_fs_component *ctf_fs,
                goto error;
        }
 
+       ctf_fs_trace->ds_file_groups = g_ptr_array_new_with_free_func(
+               (GDestroyNotify) ctf_fs_ds_file_group_destroy);
+       if (!ctf_fs_trace->ds_file_groups) {
+               goto error;
+       }
+
        ret = ctf_fs_metadata_set_trace(ctf_fs_trace);
        if (ret) {
                goto error;
        }
 
+       ret = create_ds_file_groups(ctf_fs_trace);
+       if (ret) {
+               goto error;
+       }
+
        ret = create_cc_prio_map(ctf_fs_trace);
        if (ret) {
                goto error;
@@ -408,6 +931,13 @@ struct ctf_fs_trace *ctf_fs_trace_create(struct ctf_fs_component *ctf_fs,
                goto error;
        }
 
+       /*
+        * create_ds_file_groups() created all the streams that this
+        * trace needs. There won't be any more. Therefore it is safe to
+        * make this trace static.
+        */
+       (void) bt_ctf_trace_set_is_static(ctf_fs_trace->metadata->trace);
+
        goto end;
 
 error:
@@ -444,39 +974,31 @@ static
 int add_trace_path(struct ctf_fs_component *ctf_fs, GList **trace_paths,
                const char *path)
 {
-       GString *path_str = g_string_new(NULL);
+       GString *norm_path = NULL;
        int ret = 0;
-       char *rp = NULL;
 
-       if (!path_str) {
+       norm_path = bt_common_normalize_path(path, NULL);
+       if (!norm_path) {
+               PERR("Failed to normalize path `%s`.\n", path);
                ret = -1;
                goto end;
        }
 
-       /*
-        * Find the real path so that we don't have relative components
-        * in the trace name. This also squashes consecutive slashes and
-        * removes any slash at the end.
-        */
-       rp = realpath(path, NULL);
-       if (!rp) {
-               PERR("realpath() failed: %s (%d)\n", strerror(errno), errno);
-               ret = -1;
-               goto end;
-       }
-
-       if (strcmp(rp, "/") == 0) {
+       if (strcmp(norm_path->str, "/") == 0) {
                PERR("Opening a trace in `/` is not supported.\n");
                ret = -1;
                goto end;
        }
 
-       g_string_assign(path_str, rp);
-       *trace_paths = g_list_prepend(*trace_paths, path_str);
+       *trace_paths = g_list_prepend(*trace_paths, norm_path);
        assert(*trace_paths);
+       norm_path = NULL;
 
 end:
-       free(rp);
+       if (norm_path) {
+               g_string_free(norm_path, TRUE);
+       }
+
        return ret;
 }
 
@@ -497,8 +1019,8 @@ int find_ctf_traces(struct ctf_fs_component *ctf_fs,
 
        if (ret) {
                /*
-                * Do not even recurse: a CTF trace cannot contain
-                * another CTF trace.
+                * Stop recursion: a CTF trace cannot contain another
+                * CTF trace.
                 */
                ret = add_trace_path(ctf_fs, trace_paths, start_path);
                goto end;
@@ -553,68 +1075,52 @@ end:
 }
 
 static
-GList *create_trace_names(GList *trace_paths) {
+GList *create_trace_names(GList *trace_paths, const char *base_path) {
        GList *trace_names = NULL;
-       size_t chars_to_strip = 0;
-       size_t at = 0;
        GList *node;
-       bool done = false;
+       const char *last_sep;
+       size_t base_dist;
 
        /*
-        * Find the number of characters to strip from the beginning,
-        * that is, the longest prefix until a common slash (also
-        * stripped).
+        * At this point we know that all the trace paths are
+        * normalized, and so is the base path. This means that
+        * they are absolute and they don't end with a separator.
+        * We can simply find the location of the last separator
+        * in the base path, which gives us the name of the actual
+        * directory to look into, and use this location as the
+        * start of each trace name within each trace path.
+        *
+        * For example:
+        *
+        *     Base path: /home/user/my-traces/some-trace
+        *     Trace paths:
+        *       - /home/user/my-traces/some-trace/host1/trace1
+        *       - /home/user/my-traces/some-trace/host1/trace2
+        *       - /home/user/my-traces/some-trace/host2/trace
+        *       - /home/user/my-traces/some-trace/other-trace
+        *
+        * In this case the trace names are:
+        *
+        *       - some-trace/host1/trace1
+        *       - some-trace/host1/trace2
+        *       - some-trace/host2/trace
+        *       - some-trace/other-trace
         */
-       while (true) {
-               gchar common_ch = '\0';
-
-               for (node = trace_paths; node; node = g_list_next(node)) {
-                       GString *gstr = node->data;
-                       gchar this_ch = gstr->str[at];
-
-                       if (this_ch == '\0') {
-                               done = true;
-                               break;
-                       }
-
-                       if (common_ch == '\0') {
-                               /*
-                                * Establish the expected common
-                                * character at this position.
-                                */
-                               common_ch = this_ch;
-                               continue;
-                       }
+       last_sep = strrchr(base_path, G_DIR_SEPARATOR);
 
-                       if (this_ch != common_ch) {
-                               done = true;
-                               break;
-                       }
-               }
-
-               if (done) {
-                       break;
-               }
+       /* We know there's at least one separator */
+       assert(last_sep);
 
-               if (common_ch == '/') {
-                       /*
-                        * Common character is a slash: safe to include
-                        * this slash in the number of characters to
-                        * strip because the paths are guaranteed not to
-                        * end with slash.
-                        */
-                       chars_to_strip = at + 1;
-               }
-
-               at++;
-       }
+       /* Distance to base */
+       base_dist = last_sep - base_path + 1;
 
        /* Create the trace names */
        for (node = trace_paths; node; node = g_list_next(node)) {
                GString *trace_name = g_string_new(NULL);
                GString *trace_path = node->data;
 
-               g_string_assign(trace_name, &trace_path->str[chars_to_strip]);
+               assert(trace_name);
+               g_string_assign(trace_name, &trace_path->str[base_dist]);
                trace_names = g_list_append(trace_names, trace_name);
        }
 
@@ -627,12 +1133,20 @@ int create_ctf_fs_traces(struct ctf_fs_component *ctf_fs,
 {
        struct ctf_fs_trace *ctf_fs_trace = NULL;
        int ret = 0;
+       GString *norm_path = NULL;
        GList *trace_paths = NULL;
        GList *trace_names = NULL;
        GList *tp_node;
        GList *tn_node;
 
-       ret = find_ctf_traces(ctf_fs, &trace_paths, path_param);
+       norm_path = bt_common_normalize_path(path_param, NULL);
+       if (!norm_path) {
+               PERR("Failed to normalize path: `%s`.\n",
+                       path_param);
+               goto error;
+       }
+
+       ret = find_ctf_traces(ctf_fs, &trace_paths, norm_path->str);
        if (ret) {
                goto error;
        }
@@ -643,7 +1157,7 @@ int create_ctf_fs_traces(struct ctf_fs_component *ctf_fs,
                goto error;
        }
 
-       trace_names = create_trace_names(trace_paths);
+       trace_names = create_trace_names(trace_paths, norm_path->str);
        if (!trace_names) {
                PERR("Cannot create trace names from trace paths.\n");
                goto error;
@@ -662,10 +1176,11 @@ int create_ctf_fs_traces(struct ctf_fs_component *ctf_fs,
                                trace_path->str);
                        goto error;
                }
+
+               g_ptr_array_add(ctf_fs->traces, ctf_fs_trace);
+               ctf_fs_trace = NULL;
        }
 
-       g_ptr_array_add(ctf_fs->traces, ctf_fs_trace);
-       ctf_fs_trace = NULL;
        goto end;
 
 error:
@@ -693,6 +1208,10 @@ end:
                g_list_free(trace_names);
        }
 
+       if (norm_path) {
+               g_string_free(norm_path, TRUE);
+       }
+
        return ret;
 }
 
This page took 0.035834 seconds and 4 git commands to generate.