src.ctf.fs: remove ctf_fs_trace_destroy
[babeltrace.git] / src / plugins / ctf / fs-src / fs.cpp
CommitLineData
7a278c8e 1/*
0235b0db 2 * SPDX-License-Identifier: MIT
7a278c8e 3 *
1a9f7075 4 * Copyright 2015-2017 Philippe Proulx <pproulx@efficios.com>
f3bc2010 5 * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
7a278c8e 6 *
0235b0db 7 * Babeltrace CTF file system Reader Component
7a278c8e
JG
8 */
9
c802cacb
SM
10#include <glib.h>
11#include <inttypes.h>
c802cacb
SM
12
13#include <babeltrace2/babeltrace.h>
14
c802cacb 15#include "common/assert.h"
578e048b 16#include "common/common.h"
6162e6b7 17#include "common/uuid.h"
945312a2 18#include "cpp-common/bt2s/make-unique.hpp"
c802cacb
SM
19
20#include "plugins/common/param-validation/param-validation.h"
21
5656cea5
PP
22#include "../common/src/metadata/tsdl/ctf-meta-configure-ir-trace.hpp"
23#include "../common/src/msg-iter/msg-iter.hpp"
c802cacb
SM
24#include "data-stream-file.hpp"
25#include "file.hpp"
26#include "fs.hpp"
27#include "metadata.hpp"
087cd0f5 28#include "query.hpp"
e7a4393b 29
4164020e
SM
30struct tracer_info
31{
32 const char *name;
33 int64_t major;
34 int64_t minor;
35 int64_t patch;
626cc488
FD
36};
37
4164020e
SM
38static bt_message_iterator_class_next_method_status
39ctf_fs_iterator_next_one(struct ctf_fs_msg_iter_data *msg_iter_data, const bt_message **out_msg)
ea0b4b9e 40{
4164020e 41 bt_message_iterator_class_next_method_status status;
537fddc0
SM
42 const auto msg_iter_status =
43 ctf_msg_iter_get_next_message(msg_iter_data->msg_iter.get(), out_msg);
4164020e
SM
44
45 switch (msg_iter_status) {
46 case CTF_MSG_ITER_STATUS_OK:
47 /* Cool, message has been written to *out_msg. */
48 status = BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK;
49 break;
50
51 case CTF_MSG_ITER_STATUS_EOF:
52 status = BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_END;
53 break;
54
55 case CTF_MSG_ITER_STATUS_AGAIN:
56 /*
57 * Should not make it this far as this is
58 * medium-specific; there is nothing for the user to do
59 * and it should have been handled upstream.
60 */
61 bt_common_abort();
62
63 case CTF_MSG_ITER_STATUS_ERROR:
0f5c5d5c
SM
64 BT_CPPLOGE_APPEND_CAUSE_SPEC(msg_iter_data->logger,
65 "Failed to get next message from CTF message iterator.");
4164020e
SM
66 status = BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_ERROR;
67 break;
68
69 case CTF_MSG_ITER_STATUS_MEMORY_ERROR:
0f5c5d5c
SM
70 BT_CPPLOGE_APPEND_CAUSE_SPEC(msg_iter_data->logger,
71 "Failed to get next message from CTF message iterator.");
4164020e
SM
72 status = BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_MEMORY_ERROR;
73 break;
74
75 default:
76 bt_common_abort();
77 }
78
79 return status;
d4393e08
PP
80}
81
4164020e
SM
82bt_message_iterator_class_next_method_status
83ctf_fs_iterator_next(bt_self_message_iterator *iterator, bt_message_array_const msgs,
84 uint64_t capacity, uint64_t *count)
d4393e08 85{
1e690349
SM
86 try {
87 bt_message_iterator_class_next_method_status status;
88 struct ctf_fs_msg_iter_data *msg_iter_data =
89 (struct ctf_fs_msg_iter_data *) bt_self_message_iterator_get_data(iterator);
90 uint64_t i = 0;
4164020e 91
1e690349
SM
92 if (G_UNLIKELY(msg_iter_data->next_saved_error)) {
93 /*
4164020e
SM
94 * Last time we were called, we hit an error but had some
95 * messages to deliver, so we stashed the error here. Return
96 * it now.
97 */
1e690349
SM
98 BT_CURRENT_THREAD_MOVE_ERROR_AND_RESET(msg_iter_data->next_saved_error);
99 status = msg_iter_data->next_saved_status;
100 goto end;
4164020e 101 }
4164020e 102
1e690349
SM
103 do {
104 status = ctf_fs_iterator_next_one(msg_iter_data, &msgs[i]);
105 if (status == BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK) {
106 i++;
107 }
108 } while (i < capacity && status == BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK);
109
110 if (i > 0) {
111 /*
4164020e
SM
112 * Even if ctf_fs_iterator_next_one() returned something
113 * else than BT_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_OK, we
114 * accumulated message objects in the output
115 * message array, so we need to return
116 * BT_MESSAGE_ITERATOR_NEXT_METHOD_STATUS_OK so that they are
e7401568 117 * transferred to downstream. This other status occurs
4164020e
SM
118 * again the next time muxer_msg_iter_do_next() is
119 * called, possibly without any accumulated
120 * message, in which case we'll return it.
121 */
1e690349
SM
122 if (status < 0) {
123 /*
4164020e
SM
124 * Save this error for the next _next call. Assume that
125 * this component always appends error causes when
126 * returning an error status code, which will cause the
127 * current thread error to be non-NULL.
128 */
1e690349
SM
129 msg_iter_data->next_saved_error = bt_current_thread_take_error();
130 BT_ASSERT(msg_iter_data->next_saved_error);
131 msg_iter_data->next_saved_status = status;
132 }
4164020e 133
1e690349
SM
134 *count = i;
135 status = BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_OK;
136 }
d4393e08 137
cbca1c06 138end:
1e690349
SM
139 return status;
140 return status;
141 } catch (const std::bad_alloc&) {
142 return BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_MEMORY_ERROR;
143 } catch (const bt2::Error&) {
144 return BT_MESSAGE_ITERATOR_CLASS_NEXT_METHOD_STATUS_ERROR;
145 }
ea0b4b9e 146}
bfd20a42 147
a3f0c7db 148bt_message_iterator_class_seek_beginning_method_status
d24d5663 149ctf_fs_iterator_seek_beginning(bt_self_message_iterator *it)
6a9bb5e9 150{
1e690349
SM
151 try {
152 struct ctf_fs_msg_iter_data *msg_iter_data =
153 (struct ctf_fs_msg_iter_data *) bt_self_message_iterator_get_data(it);
6a9bb5e9 154
1e690349 155 BT_ASSERT(msg_iter_data);
6a9bb5e9 156
537fddc0 157 ctf_msg_iter_reset(msg_iter_data->msg_iter.get());
3cf88182 158 ctf_fs_ds_group_medops_data_reset(msg_iter_data->msg_iter_medops_data.get());
f6e68e70 159
1e690349
SM
160 return BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_OK;
161 } catch (const std::bad_alloc&) {
162 return BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_MEMORY_ERROR;
163 } catch (const bt2::Error&) {
164 return BT_MESSAGE_ITERATOR_CLASS_SEEK_BEGINNING_METHOD_STATUS_ERROR;
165 }
6a9bb5e9
PP
166}
167
d6e69534 168void ctf_fs_iterator_finalize(bt_self_message_iterator *it)
760051fa 169{
adf452d1
SM
170 ctf_fs_msg_iter_data::UP {
171 (static_cast<ctf_fs_msg_iter_data *>(bt_self_message_iterator_get_data(it)))};
760051fa
JG
172}
173
4164020e
SM
174static bt_message_iterator_class_initialize_method_status
175ctf_msg_iter_medium_status_to_msg_iter_initialize_status(enum ctf_msg_iter_medium_status status)
1b7b1ef9 176{
4164020e
SM
177 switch (status) {
178 case CTF_MSG_ITER_MEDIUM_STATUS_EOF:
179 case CTF_MSG_ITER_MEDIUM_STATUS_AGAIN:
180 case CTF_MSG_ITER_MEDIUM_STATUS_ERROR:
181 return BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
182 case CTF_MSG_ITER_MEDIUM_STATUS_MEMORY_ERROR:
183 return BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR;
184 case CTF_MSG_ITER_MEDIUM_STATUS_OK:
185 return BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK;
186 }
187
188 bt_common_abort();
1b7b1ef9
SM
189}
190
4164020e
SM
191bt_message_iterator_class_initialize_method_status
192ctf_fs_iterator_init(bt_self_message_iterator *self_msg_iter,
193 bt_self_message_iterator_configuration *config,
194 bt_self_component_port_output *self_port)
4c1456f0 195{
1e690349
SM
196 try {
197 struct ctf_fs_port_data *port_data;
198 bt_message_iterator_class_initialize_method_status status;
199 enum ctf_msg_iter_medium_status medium_status;
200
201 port_data = (struct ctf_fs_port_data *) bt_self_component_port_get_data(
202 bt_self_component_port_output_as_self_component_port(self_port));
203 BT_ASSERT(port_data);
204
adf452d1 205 auto msg_iter_data = bt2s::make_unique<ctf_fs_msg_iter_data>(self_msg_iter);
1e690349
SM
206 msg_iter_data->ds_file_group = port_data->ds_file_group;
207
208 medium_status = ctf_fs_ds_group_medops_data_create(msg_iter_data->ds_file_group,
209 self_msg_iter, msg_iter_data->logger,
3cf88182 210 msg_iter_data->msg_iter_medops_data);
1e690349
SM
211 BT_ASSERT(medium_status == CTF_MSG_ITER_MEDIUM_STATUS_OK ||
212 medium_status == CTF_MSG_ITER_MEDIUM_STATUS_ERROR ||
213 medium_status == CTF_MSG_ITER_MEDIUM_STATUS_MEMORY_ERROR);
214 if (medium_status != CTF_MSG_ITER_MEDIUM_STATUS_OK) {
215 BT_CPPLOGE_APPEND_CAUSE_SPEC(msg_iter_data->logger,
216 "Failed to create ctf_fs_ds_group_medops");
217 status = ctf_msg_iter_medium_status_to_msg_iter_initialize_status(medium_status);
218 goto error;
219 }
4164020e 220
537fddc0
SM
221 msg_iter_data->msg_iter = ctf_msg_iter_create(
222 msg_iter_data->ds_file_group->ctf_fs_trace->metadata->tc,
223 bt_common_get_page_size(static_cast<int>(msg_iter_data->logger.level())) * 8,
224 ctf_fs_ds_group_medops, msg_iter_data->msg_iter_medops_data.get(), self_msg_iter,
225 msg_iter_data->logger);
1e690349
SM
226 if (!msg_iter_data->msg_iter) {
227 BT_CPPLOGE_APPEND_CAUSE_SPEC(msg_iter_data->logger,
228 "Cannot create a CTF message iterator.");
229 status = BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR;
230 goto error;
231 }
4164020e 232
1e690349 233 /*
4164020e
SM
234 * This iterator can seek forward if its stream class has a default
235 * clock class.
236 */
1e690349
SM
237 if (msg_iter_data->ds_file_group->sc->default_clock_class) {
238 bt_self_message_iterator_configuration_set_can_seek_forward(config, true);
239 }
4164020e 240
adf452d1 241 bt_self_message_iterator_set_data(self_msg_iter, msg_iter_data.release());
4164020e 242
1e690349
SM
243 status = BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_OK;
244 goto end;
5b29e799 245
4f1f88a6 246error:
1e690349 247 bt_self_message_iterator_set_data(self_msg_iter, NULL);
4f1f88a6 248
760051fa 249end:
1e690349
SM
250 return status;
251 } catch (const std::bad_alloc&) {
252 return BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR;
253 } catch (const bt2::Error&) {
254 return BT_MESSAGE_ITERATOR_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
255 }
760051fa
JG
256}
257
f340a3e8 258ctf_fs_component::UP ctf_fs_component_create(const bt2c::Logger& parentLogger)
a4792757 259{
57ec0a35 260 return bt2s::make_unique<ctf_fs_component>(parentLogger);
f280892e
SM
261}
262
263void ctf_fs_finalize(bt_self_component_source *component)
264{
57ec0a35
SM
265 ctf_fs_component::UP {static_cast<ctf_fs_component *>(
266 bt_self_component_get_data(bt_self_component_source_as_self_component(component)))};
5b29e799
JG
267}
268
49b956cc 269bt2c::GCharUP ctf_fs_make_port_name(struct ctf_fs_ds_file_group *ds_file_group)
547eacf1 270{
4164020e
SM
271 GString *name = g_string_new(NULL);
272
273 /*
274 * The unique port name is generated by concatenating unique identifiers
275 * for:
276 *
277 * - the trace
278 * - the stream class
279 * - the stream
280 */
281
282 /* For the trace, use the uuid if present, else the path. */
283 if (ds_file_group->ctf_fs_trace->metadata->tc->is_uuid_set) {
284 char uuid_str[BT_UUID_STR_LEN + 1];
285
286 bt_uuid_to_str(ds_file_group->ctf_fs_trace->metadata->tc->uuid, uuid_str);
287 g_string_assign(name, uuid_str);
288 } else {
cee8a466 289 g_string_assign(name, ds_file_group->ctf_fs_trace->path.c_str());
4164020e
SM
290 }
291
292 /*
293 * For the stream class, use the id if present. We can omit this field
294 * otherwise, as there will only be a single stream class.
295 */
296 if (ds_file_group->sc->id != UINT64_C(-1)) {
297 g_string_append_printf(name, " | %" PRIu64, ds_file_group->sc->id);
298 }
299
300 /* For the stream, use the id if present, else, use the path. */
301 if (ds_file_group->stream_id != UINT64_C(-1)) {
302 g_string_append_printf(name, " | %" PRIu64, ds_file_group->stream_id);
303 } else {
f3d74124
SM
304 BT_ASSERT(ds_file_group->ds_file_infos.size() == 1);
305 const auto& ds_file_info = *ds_file_group->ds_file_infos[0];
306 g_string_append_printf(name, " | %s", ds_file_info.path.c_str());
4164020e
SM
307 }
308
49b956cc 309 return bt2c::GCharUP {g_string_free(name, FALSE)};
547eacf1
PP
310}
311
ce11b8c4 312static int create_one_port_for_trace(struct ctf_fs_component *ctf_fs,
4164020e
SM
313 struct ctf_fs_ds_file_group *ds_file_group,
314 bt_self_component_source *self_comp_src)
5b29e799 315{
4164020e 316 int ret = 0;
945312a2 317 ctf_fs_port_data::UP port_data;
4164020e 318
49b956cc 319 bt2c::GCharUP port_name = ctf_fs_make_port_name(ds_file_group);
4164020e
SM
320 if (!port_name) {
321 goto error;
322 }
323
49b956cc 324 BT_CPPLOGI_SPEC(ctf_fs->logger, "Creating one port named `{}`", port_name.get());
4164020e
SM
325
326 /* Create output port for this file */
945312a2 327 port_data = bt2s::make_unique<ctf_fs_port_data>();
4164020e
SM
328 port_data->ctf_fs = ctf_fs;
329 port_data->ds_file_group = ds_file_group;
945312a2
SM
330 ret = bt_self_component_source_add_output_port(self_comp_src, port_name.get(), port_data.get(),
331 NULL);
4164020e
SM
332 if (ret) {
333 goto error;
334 }
335
945312a2 336 ctf_fs->port_data.emplace_back(std::move(port_data));
4164020e 337 goto end;
4f1f88a6
PP
338
339error:
4164020e 340 ret = -1;
4f1f88a6
PP
341
342end:
4164020e 343 return ret;
5b29e799
JG
344}
345
4164020e
SM
346static int create_ports_for_trace(struct ctf_fs_component *ctf_fs,
347 struct ctf_fs_trace *ctf_fs_trace,
348 bt_self_component_source *self_comp_src)
94cf822e 349{
4164020e 350 int ret = 0;
4164020e
SM
351
352 /* Create one output port for each stream file group */
cdf7de78
SM
353 for (const auto& ds_file_group : ctf_fs_trace->ds_file_groups) {
354 ret = create_one_port_for_trace(ctf_fs, ds_file_group.get(), self_comp_src);
4164020e 355 if (ret) {
0f5c5d5c 356 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger, "Cannot create output port.");
4164020e
SM
357 goto end;
358 }
359 }
94cf822e
PP
360
361end:
4164020e 362 return ret;
94cf822e
PP
363}
364
41a65f30
SM
365/*
366 * Insert ds_file_info in ds_file_group's list of ds_file_infos at the right
367 * place to keep it sorted.
368 */
369
4164020e 370static void ds_file_group_insert_ds_file_info_sorted(struct ctf_fs_ds_file_group *ds_file_group,
f3d74124 371 ctf_fs_ds_file_info::UP ds_file_info)
41a65f30 372{
4164020e 373 /* Find the spot where to insert this ds_file_info. */
f3d74124 374 auto it = ds_file_group->ds_file_infos.begin();
41a65f30 375
f3d74124
SM
376 for (; it != ds_file_group->ds_file_infos.end(); ++it) {
377 const ctf_fs_ds_file_info& other_ds_file_info = **it;
378
379 if (ds_file_info->begin_ns < other_ds_file_info.begin_ns) {
4164020e
SM
380 break;
381 }
382 }
41a65f30 383
f3d74124 384 ds_file_group->ds_file_infos.insert(it, std::move(ds_file_info));
41a65f30
SM
385}
386
4164020e
SM
387static bool ds_index_entries_equal(const struct ctf_fs_ds_index_entry *left,
388 const struct ctf_fs_ds_index_entry *right)
1505f33a 389{
ef7d7ac2 390 if (left->packetSize != right->packetSize) {
4164020e
SM
391 return false;
392 }
1505f33a 393
4164020e
SM
394 if (left->timestamp_begin != right->timestamp_begin) {
395 return false;
396 }
1505f33a 397
4164020e
SM
398 if (left->timestamp_end != right->timestamp_end) {
399 return false;
400 }
1505f33a 401
4164020e
SM
402 if (left->packet_seq_num != right->packet_seq_num) {
403 return false;
404 }
1505f33a 405
4164020e 406 return true;
1505f33a
SM
407}
408
409/*
410 * Insert `entry` into `index`, without duplication.
411 *
412 * The entry is inserted only if there isn't an identical entry already.
413 *
414 * In any case, the ownership of `entry` is transferred to this function. So if
415 * the entry is not inserted, it is freed.
416 */
417
4164020e 418static void ds_index_insert_ds_index_entry_sorted(struct ctf_fs_ds_index *index,
2fb7af12 419 ctf_fs_ds_index_entry::UP entry)
7ed5243a 420{
4164020e 421 /* Find the spot where to insert this index entry. */
2fb7af12
SM
422 auto otherEntry = index->entries.begin();
423 for (; otherEntry != index->entries.end(); ++otherEntry) {
424 if (entry->timestamp_begin_ns <= (*otherEntry)->timestamp_begin_ns) {
4164020e
SM
425 break;
426 }
427 }
428
429 /*
430 * Insert the entry only if a duplicate doesn't already exist.
431 *
432 * There can be duplicate packets if reading multiple overlapping
433 * snapshots of the same trace. We then want the index to contain
434 * a reference to only one copy of that packet.
435 */
2fb7af12
SM
436 if (otherEntry == index->entries.end() ||
437 !ds_index_entries_equal(entry.get(), otherEntry->get())) {
438 index->entries.insert(otherEntry, std::move(entry));
4164020e 439 }
ce75de14
SM
440}
441
fe2f9cda 442static void merge_ctf_fs_ds_indexes(struct ctf_fs_ds_index *dest, ctf_fs_ds_index::UP src)
ce75de14 443{
2fb7af12
SM
444 for (auto& entry : src->entries) {
445 ds_index_insert_ds_index_entry_sorted(dest, std::move(entry));
4164020e 446 }
7ed5243a
FD
447}
448
4164020e 449static int add_ds_file_to_ds_file_group(struct ctf_fs_trace *ctf_fs_trace, const char *path)
94cf822e 450{
4164020e
SM
451 int64_t stream_instance_id = -1;
452 int64_t begin_ns = -1;
453 struct ctf_fs_ds_file_group *ds_file_group = NULL;
fe2e19c4 454 ctf_fs_ds_file_group::UP new_ds_file_group;
4164020e 455 int ret;
f3d74124 456 ctf_fs_ds_file_info::UP ds_file_info;
fe2f9cda 457 ctf_fs_ds_index::UP index;
4420a5b6 458 ctf_msg_iter_up msg_iter;
4164020e
SM
459 struct ctf_stream_class *sc = NULL;
460 struct ctf_msg_iter_packet_properties props;
4164020e
SM
461
462 /*
463 * Create a temporary ds_file to read some properties about the data
464 * stream file.
465 */
884feb7c
SM
466 const auto ds_file =
467 ctf_fs_ds_file_create(ctf_fs_trace, bt2::Stream::Shared {}, path, ctf_fs_trace->logger);
4164020e
SM
468 if (!ds_file) {
469 goto error;
470 }
471
472 /* Create a temporary iterator to read the ds_file. */
0f5c5d5c 473 msg_iter = ctf_msg_iter_create(
4420a5b6
SM
474 ctf_fs_trace->metadata->tc,
475 bt_common_get_page_size(static_cast<int>(ctf_fs_trace->logger.level())) * 8,
476 ctf_fs_ds_file_medops, ds_file.get(), nullptr, ctf_fs_trace->logger);
4164020e 477 if (!msg_iter) {
0f5c5d5c 478 BT_CPPLOGE_STR_SPEC(ctf_fs_trace->logger, "Cannot create a CTF message iterator.");
4164020e
SM
479 goto error;
480 }
481
4420a5b6 482 ctf_msg_iter_set_dry_run(msg_iter.get(), true);
4164020e 483
4420a5b6 484 ret = ctf_msg_iter_get_packet_properties(msg_iter.get(), &props);
4164020e 485 if (ret) {
0f5c5d5c
SM
486 BT_CPPLOGE_APPEND_CAUSE_SPEC(
487 ctf_fs_trace->logger,
488 "Cannot get stream file's first packet's header and context fields (`{}`).", path);
4164020e
SM
489 goto error;
490 }
491
492 sc = ctf_trace_class_borrow_stream_class_by_id(ds_file->metadata->tc, props.stream_class_id);
493 BT_ASSERT(sc);
494 stream_instance_id = props.data_stream_id;
495
496 if (props.snapshots.beginning_clock != UINT64_C(-1)) {
497 BT_ASSERT(sc->default_clock_class);
498 ret = bt_util_clock_cycles_to_ns_from_origin(
499 props.snapshots.beginning_clock, sc->default_clock_class->frequency,
500 sc->default_clock_class->offset_seconds, sc->default_clock_class->offset_cycles,
501 &begin_ns);
502 if (ret) {
0f5c5d5c
SM
503 BT_CPPLOGE_APPEND_CAUSE_SPEC(
504 ctf_fs_trace->logger,
505 "Cannot convert clock cycles to nanoseconds from origin (`{}`).", path);
4164020e
SM
506 goto error;
507 }
508 }
509
f3d74124 510 ds_file_info = ctf_fs_ds_file_info_create(path, begin_ns);
4164020e
SM
511 if (!ds_file_info) {
512 goto error;
513 }
514
4420a5b6 515 index = ctf_fs_ds_file_build_index(ds_file.get(), ds_file_info.get(), msg_iter.get());
4164020e 516 if (!index) {
0f5c5d5c 517 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs_trace->logger, "Failed to index CTF stream file \'{}\'",
a39d9817 518 ds_file->file->path);
4164020e
SM
519 goto error;
520 }
521
522 if (begin_ns == -1) {
523 /*
524 * No beginning timestamp to sort the stream files
525 * within a stream file group, so consider that this
526 * file must be the only one within its group.
527 */
528 stream_instance_id = -1;
529 }
530
531 if (stream_instance_id == -1) {
532 /*
533 * No stream instance ID or no beginning timestamp:
534 * create a unique stream file group for this stream
535 * file because, even if there's a stream instance ID,
536 * there's no timestamp to order the file within its
537 * group.
538 */
fe2f9cda
SM
539 new_ds_file_group =
540 ctf_fs_ds_file_group_create(ctf_fs_trace, sc, UINT64_C(-1), std::move(index));
4164020e 541
fe2e19c4 542 if (!new_ds_file_group) {
4164020e
SM
543 goto error;
544 }
545
f3d74124 546 ds_file_group_insert_ds_file_info_sorted(new_ds_file_group.get(), std::move(ds_file_info));
cdf7de78 547 ctf_fs_trace->ds_file_groups.emplace_back(std::move(new_ds_file_group));
4164020e
SM
548 goto end;
549 }
550
551 BT_ASSERT(stream_instance_id != -1);
552 BT_ASSERT(begin_ns != -1);
553
554 /* Find an existing stream file group with this ID */
cdf7de78
SM
555 for (const auto& candidate : ctf_fs_trace->ds_file_groups) {
556 if (candidate->sc == sc && candidate->stream_id == stream_instance_id) {
557 ds_file_group = candidate.get();
4164020e
SM
558 break;
559 }
4164020e
SM
560 }
561
562 if (!ds_file_group) {
fe2e19c4 563 new_ds_file_group =
fe2f9cda 564 ctf_fs_ds_file_group_create(ctf_fs_trace, sc, stream_instance_id, std::move(index));
fe2e19c4 565 if (!new_ds_file_group) {
4164020e
SM
566 goto error;
567 }
568
fe2e19c4 569 ds_file_group = new_ds_file_group.get();
cdf7de78 570 ctf_fs_trace->ds_file_groups.emplace_back(std::move(new_ds_file_group));
4164020e 571 } else {
fe2f9cda 572 merge_ctf_fs_ds_indexes(ds_file_group->index.get(), std::move(index));
4164020e
SM
573 }
574
f3d74124 575 ds_file_group_insert_ds_file_info_sorted(ds_file_group, std::move(ds_file_info));
4164020e
SM
576
577 goto end;
94cf822e
PP
578
579error:
4164020e 580 ret = -1;
94cf822e
PP
581
582end:
4164020e 583 return ret;
94cf822e
PP
584}
585
4164020e 586static int create_ds_file_groups(struct ctf_fs_trace *ctf_fs_trace)
e7a4393b 587{
4164020e
SM
588 int ret = 0;
589 const char *basename;
590 GError *error = NULL;
591 GDir *dir = NULL;
4164020e
SM
592
593 /* Check each file in the path directory, except specific ones */
cee8a466 594 dir = g_dir_open(ctf_fs_trace->path.c_str(), 0, &error);
4164020e 595 if (!dir) {
0f5c5d5c 596 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs_trace->logger,
cee8a466
SM
597 "Cannot open directory `{}`: {} (code {})", ctf_fs_trace->path,
598 error->message, error->code);
4164020e
SM
599 goto error;
600 }
601
602 while ((basename = g_dir_read_name(dir))) {
4164020e
SM
603 if (strcmp(basename, CTF_FS_METADATA_FILENAME) == 0) {
604 /* Ignore the metadata stream. */
0f5c5d5c
SM
605 BT_CPPLOGI_SPEC(ctf_fs_trace->logger,
606 "Ignoring metadata file `{}" G_DIR_SEPARATOR_S "{}`",
cee8a466 607 ctf_fs_trace->path, basename);
4164020e
SM
608 continue;
609 }
610
611 if (basename[0] == '.') {
0f5c5d5c 612 BT_CPPLOGI_SPEC(ctf_fs_trace->logger,
cee8a466
SM
613 "Ignoring hidden file `{}" G_DIR_SEPARATOR_S "{}`", ctf_fs_trace->path,
614 basename);
4164020e
SM
615 continue;
616 }
617
618 /* Create the file. */
2b6f09e0 619 const auto file = ctf_fs_file_create(ctf_fs_trace->logger);
4164020e 620 if (!file) {
0f5c5d5c
SM
621 BT_CPPLOGE_APPEND_CAUSE_SPEC(
622 ctf_fs_trace->logger,
623 "Cannot create stream file object for file `{}" G_DIR_SEPARATOR_S "{}`",
cee8a466 624 ctf_fs_trace->path, basename);
4164020e
SM
625 goto error;
626 }
627
628 /* Create full path string. */
cee8a466 629 file->path = fmt::format("{}" G_DIR_SEPARATOR_S "{}", ctf_fs_trace->path, basename);
a39d9817
SM
630
631 if (!g_file_test(file->path.c_str(), G_FILE_TEST_IS_REGULAR)) {
632 BT_CPPLOGI_SPEC(ctf_fs_trace->logger, "Ignoring non-regular file `{}`", file->path);
4164020e
SM
633 continue;
634 }
635
2b6f09e0 636 ret = ctf_fs_file_open(file.get(), "rb");
4164020e 637 if (ret) {
0f5c5d5c 638 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs_trace->logger, "Cannot open stream file `{}`",
a39d9817 639 file->path);
4164020e
SM
640 goto error;
641 }
642
643 if (file->size == 0) {
644 /* Skip empty stream. */
a39d9817 645 BT_CPPLOGI_SPEC(ctf_fs_trace->logger, "Ignoring empty file `{}`", file->path);
4164020e
SM
646 continue;
647 }
648
a39d9817 649 ret = add_ds_file_to_ds_file_group(ctf_fs_trace, file->path.c_str());
4164020e 650 if (ret) {
0f5c5d5c
SM
651 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs_trace->logger,
652 "Cannot add stream file `{}` to stream file group",
a39d9817 653 file->path);
4164020e
SM
654 goto error;
655 }
4164020e
SM
656 }
657
658 goto end;
4f1f88a6 659
e7a4393b 660error:
4164020e 661 ret = -1;
4f1f88a6 662
e7a4393b 663end:
4164020e
SM
664 if (dir) {
665 g_dir_close(dir);
666 dir = NULL;
667 }
4f1f88a6 668
4164020e
SM
669 if (error) {
670 g_error_free(error);
671 }
5b29e799 672
4164020e 673 return ret;
5b29e799
JG
674}
675
0f5c5d5c 676static int set_trace_name(bt_trace *trace, const char *name_suffix, const bt2c::Logger& logger)
862ca4ed 677{
4164020e
SM
678 int ret = 0;
679 const bt_value *val;
680 GString *name;
681
682 name = g_string_new(NULL);
683 if (!name) {
0f5c5d5c 684 BT_CPPLOGE_STR_SPEC(logger, "Failed to allocate a GString.");
4164020e
SM
685 ret = -1;
686 goto end;
687 }
688
689 /*
690 * Check if we have a trace environment string value named `hostname`.
691 * If so, use it as the trace name's prefix.
692 */
693 val = bt_trace_borrow_environment_entry_value_by_name_const(trace, "hostname");
694 if (val && bt_value_is_string(val)) {
695 g_string_append(name, bt_value_string_get(val));
696
697 if (name_suffix) {
698 g_string_append_c(name, G_DIR_SEPARATOR);
699 }
700 }
701
702 if (name_suffix) {
703 g_string_append(name, name_suffix);
704 }
705
706 ret = bt_trace_set_name(trace, name->str);
707 if (ret) {
708 goto end;
709 }
710
711 goto end;
862ca4ed
PP
712
713end:
4164020e
SM
714 if (name) {
715 g_string_free(name, TRUE);
716 }
862ca4ed 717
4164020e 718 return ret;
862ca4ed
PP
719}
720
2ca337f1
SM
721static ctf_fs_trace::UP ctf_fs_trace_create(const char *path, const char *name,
722 const ctf::src::ClkClsCfg& clkClsCfg,
723 bt_self_component *selfComp,
724 const bt2c::Logger& parentLogger)
1a9f7075 725{
4164020e 726 int ret;
2ca337f1 727 ctf_fs_trace::UP ctf_fs_trace {new struct ctf_fs_trace(parentLogger)};
4164020e 728
cee8a466 729 ctf_fs_trace->path = path;
2dba3a29 730 ctf_fs_trace->metadata = bt2s::make_unique<ctf_fs_metadata>();
4164020e 731
2ca337f1 732 ret = ctf_fs_metadata_set_trace_class(selfComp, ctf_fs_trace.get(), clkClsCfg);
4164020e
SM
733 if (ret) {
734 goto error;
735 }
736
737 if (ctf_fs_trace->metadata->trace_class) {
ded1f3d6 738 bt_trace *trace = bt_trace_create(ctf_fs_trace->metadata->trace_class->libObjPtr());
e44859b1 739 if (!trace) {
4164020e
SM
740 goto error;
741 }
e44859b1
SM
742
743 ctf_fs_trace->trace = bt2::Trace::Shared::createWithoutRef(trace);
4164020e
SM
744 }
745
746 if (ctf_fs_trace->trace) {
e44859b1
SM
747 ret = ctf_trace_class_configure_ir_trace(ctf_fs_trace->metadata->tc,
748 ctf_fs_trace->trace->libObjPtr());
4164020e
SM
749 if (ret) {
750 goto error;
751 }
752
e44859b1 753 ret = set_trace_name(ctf_fs_trace->trace->libObjPtr(), name, ctf_fs_trace->logger);
4164020e
SM
754 if (ret) {
755 goto error;
756 }
757 }
758
2ca337f1 759 ret = create_ds_file_groups(ctf_fs_trace.get());
4164020e
SM
760 if (ret) {
761 goto error;
762 }
763
764 goto end;
1a9f7075
PP
765
766error:
2ca337f1 767 ctf_fs_trace.reset();
44c440bc 768
1a9f7075 769end:
4164020e 770 return ctf_fs_trace;
1a9f7075
PP
771}
772
4164020e 773static int path_is_ctf_trace(const char *path)
1a9f7075 774{
4164020e
SM
775 GString *metadata_path = g_string_new(NULL);
776 int ret = 0;
1a9f7075 777
4164020e
SM
778 if (!metadata_path) {
779 ret = -1;
780 goto end;
781 }
1a9f7075 782
4164020e 783 g_string_printf(metadata_path, "%s" G_DIR_SEPARATOR_S "%s", path, CTF_FS_METADATA_FILENAME);
1a9f7075 784
4164020e
SM
785 if (g_file_test(metadata_path->str, G_FILE_TEST_IS_REGULAR)) {
786 ret = 1;
787 goto end;
788 }
1a9f7075
PP
789
790end:
4164020e
SM
791 g_string_free(metadata_path, TRUE);
792 return ret;
1a9f7075
PP
793}
794
a0cd55ad 795/* Helper for ctf_fs_component_create_ctf_fs_trace, to handle a single path. */
f280892e 796
4164020e
SM
797static int ctf_fs_component_create_ctf_fs_trace_one_path(struct ctf_fs_component *ctf_fs,
798 const char *path_param,
a4c955d9
SM
799 const char *trace_name,
800 std::vector<ctf_fs_trace::UP>& traces,
0f5c5d5c 801 bt_self_component *selfComp)
1a9f7075 802{
2ca337f1 803 ctf_fs_trace::UP ctf_fs_trace;
4164020e
SM
804 int ret;
805 GString *norm_path;
4164020e
SM
806
807 norm_path = bt_common_normalize_path(path_param, NULL);
808 if (!norm_path) {
0f5c5d5c 809 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger, "Failed to normalize path: `{}`.", path_param);
4164020e
SM
810 goto error;
811 }
812
813 ret = path_is_ctf_trace(norm_path->str);
814 if (ret < 0) {
0f5c5d5c
SM
815 BT_CPPLOGE_APPEND_CAUSE_SPEC(
816 ctf_fs->logger, "Failed to check if path is a CTF trace: path={}", norm_path->str);
4164020e
SM
817 goto error;
818 } else if (ret == 0) {
0f5c5d5c
SM
819 BT_CPPLOGE_APPEND_CAUSE_SPEC(
820 ctf_fs->logger, "Path is not a CTF trace (does not contain a metadata file): `{}`.",
821 norm_path->str);
4164020e
SM
822 goto error;
823 }
824
825 // FIXME: Remove or ifdef for __MINGW32__
826 if (strcmp(norm_path->str, "/") == 0) {
0f5c5d5c 827 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger, "Opening a trace in `/` is not supported.");
4164020e
SM
828 ret = -1;
829 goto end;
830 }
831
c942e7a2
SM
832 ctf_fs_trace = ctf_fs_trace_create(norm_path->str, trace_name, ctf_fs->clkClsCfg, selfComp,
833 ctf_fs->logger);
4164020e 834 if (!ctf_fs_trace) {
0f5c5d5c
SM
835 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger, "Cannot create trace for `{}`.",
836 norm_path->str);
4164020e
SM
837 goto error;
838 }
839
a4c955d9 840 traces.emplace_back(std::move(ctf_fs_trace));
4164020e
SM
841
842 ret = 0;
843 goto end;
1a9f7075
PP
844
845error:
4164020e 846 ret = -1;
1a9f7075
PP
847
848end:
4164020e
SM
849 if (norm_path) {
850 g_string_free(norm_path, TRUE);
851 }
4bd72b60 852
4164020e 853 return ret;
1a9f7075
PP
854}
855
41a65f30
SM
856/*
857 * Count the number of stream and event classes defined by this trace's metadata.
858 *
859 * This is used to determine which metadata is the "latest", out of multiple
860 * traces sharing the same UUID. It is assumed that amongst all these metadatas,
861 * a bigger metadata is a superset of a smaller metadata. Therefore, it is
862 * enough to just count the classes.
863 */
864
4164020e 865static unsigned int metadata_count_stream_and_event_classes(struct ctf_fs_trace *trace)
41a65f30 866{
4164020e
SM
867 unsigned int num = trace->metadata->tc->stream_classes->len;
868 guint i;
41a65f30 869
4164020e
SM
870 for (i = 0; i < trace->metadata->tc->stream_classes->len; i++) {
871 struct ctf_stream_class *sc =
872 (struct ctf_stream_class *) trace->metadata->tc->stream_classes->pdata[i];
873 num += sc->event_classes->len;
874 }
41a65f30 875
4164020e 876 return num;
41a65f30
SM
877}
878
879/*
880 * Merge the src ds_file_group into dest. This consists of merging their
881 * ds_file_infos, making sure to keep the result sorted.
882 */
883
4164020e 884static void merge_ctf_fs_ds_file_groups(struct ctf_fs_ds_file_group *dest,
cdf7de78 885 ctf_fs_ds_file_group::UP src)
41a65f30 886{
f3d74124
SM
887 for (auto& ds_file_info : src->ds_file_infos) {
888 ds_file_group_insert_ds_file_info_sorted(dest, std::move(ds_file_info));
4164020e 889 }
41a65f30 890
4164020e 891 /* Merge both indexes. */
fe2f9cda 892 merge_ctf_fs_ds_indexes(dest->index.get(), std::move(src->index));
7ed5243a 893}
a4c955d9 894
41a65f30
SM
895/* Merge src_trace's data stream file groups into dest_trace's. */
896
4164020e 897static int merge_matching_ctf_fs_ds_file_groups(struct ctf_fs_trace *dest_trace,
a4c955d9 898 ctf_fs_trace::UP src_trace)
41a65f30 899{
cdf7de78
SM
900 std::vector<ctf_fs_ds_file_group::UP>& dest = dest_trace->ds_file_groups;
901 std::vector<ctf_fs_ds_file_group::UP>& src = src_trace->ds_file_groups;
4164020e
SM
902 int ret = 0;
903
904 /*
905 * Save the initial length of dest: we only want to check against the
906 * original elements in the inner loop.
907 */
cdf7de78 908 size_t dest_len = dest.size();
4164020e 909
cdf7de78 910 for (auto& src_group : src) {
4164020e
SM
911 struct ctf_fs_ds_file_group *dest_group = NULL;
912
913 /* A stream instance without ID can't match a stream in the other trace. */
914 if (src_group->stream_id != -1) {
4164020e 915 /* Let's search for a matching ds_file_group in the destination. */
cdf7de78
SM
916 for (size_t d_i = 0; d_i < dest_len; ++d_i) {
917 ctf_fs_ds_file_group *candidate_dest = dest[d_i].get();
4164020e
SM
918
919 /* Can't match a stream instance without ID. */
920 if (candidate_dest->stream_id == -1) {
921 continue;
922 }
923
924 /*
925 * If the two groups have the same stream instance id
926 * and belong to the same stream class (stream instance
927 * ids are per-stream class), they represent the same
928 * stream instance.
929 */
930 if (candidate_dest->stream_id != src_group->stream_id ||
931 candidate_dest->sc->id != src_group->sc->id) {
932 continue;
933 }
934
935 dest_group = candidate_dest;
936 break;
937 }
938 }
939
940 /*
941 * Didn't find a friend in dest to merge our src_group into?
942 * Create a new empty one. This can happen if a stream was
943 * active in the source trace chunk but not in the destination
944 * trace chunk.
945 */
946 if (!dest_group) {
947 struct ctf_stream_class *sc;
4164020e
SM
948
949 sc = ctf_trace_class_borrow_stream_class_by_id(dest_trace->metadata->tc,
950 src_group->sc->id);
951 BT_ASSERT(sc);
952
2fb7af12 953 auto index = ctf_fs_ds_index_create();
4164020e
SM
954 if (!index) {
955 ret = -1;
956 goto end;
957 }
958
fe2e19c4 959 auto new_dest_group =
fe2f9cda 960 ctf_fs_ds_file_group_create(dest_trace, sc, src_group->stream_id, std::move(index));
fe2e19c4 961
fe2e19c4 962 if (!new_dest_group) {
4164020e
SM
963 ret = -1;
964 goto end;
965 }
966
fe2e19c4 967 dest_group = new_dest_group.get();
cdf7de78 968 dest_trace->ds_file_groups.emplace_back(std::move(new_dest_group));
4164020e
SM
969 }
970
971 BT_ASSERT(dest_group);
cdf7de78 972 merge_ctf_fs_ds_file_groups(dest_group, std::move(src_group));
4164020e 973 }
54ef52bd
FD
974
975end:
4164020e 976 return ret;
41a65f30
SM
977}
978
979/*
980 * Collapse the given traces, which must all share the same UUID, in a single
981 * one.
982 *
983 * The trace with the most expansive metadata is chosen and all other traces
a4c955d9
SM
984 * are merged into that one. On return, the elements of `traces` are nullptr
985 * and the merged trace is placed in `out_trace`.
41a65f30
SM
986 */
987
a4c955d9 988static int merge_ctf_fs_traces(std::vector<ctf_fs_trace::UP> traces, ctf_fs_trace::UP& out_trace)
41a65f30 989{
4164020e
SM
990 unsigned int winner_count;
991 struct ctf_fs_trace *winner;
992 guint i, winner_i;
993 int ret = 0;
994
a4c955d9 995 BT_ASSERT(traces.size() >= 2);
4164020e 996
a4c955d9
SM
997 winner_count = metadata_count_stream_and_event_classes(traces[0].get());
998 winner = traces[0].get();
4164020e
SM
999 winner_i = 0;
1000
1001 /* Find the trace with the largest metadata. */
a4c955d9
SM
1002 for (i = 1; i < traces.size(); i++) {
1003 ctf_fs_trace *candidate = traces[i].get();
4164020e
SM
1004 unsigned int candidate_count;
1005
4164020e
SM
1006 /* A bit of sanity check. */
1007 BT_ASSERT(bt_uuid_compare(winner->metadata->tc->uuid, candidate->metadata->tc->uuid) == 0);
1008
1009 candidate_count = metadata_count_stream_and_event_classes(candidate);
1010
1011 if (candidate_count > winner_count) {
1012 winner_count = candidate_count;
1013 winner = candidate;
1014 winner_i = i;
1015 }
1016 }
1017
1018 /* Merge all the other traces in the winning trace. */
a4c955d9 1019 for (ctf_fs_trace::UP& trace : traces) {
4164020e 1020 /* Don't merge the winner into itself. */
a4c955d9 1021 if (trace.get() == winner) {
4164020e
SM
1022 continue;
1023 }
1024
1025 /* Merge trace's data stream file groups into winner's. */
a4c955d9 1026 ret = merge_matching_ctf_fs_ds_file_groups(winner, std::move(trace));
4164020e
SM
1027 if (ret) {
1028 goto end;
1029 }
1030 }
1031
1032 /*
1033 * Move the winner out of the array, into `*out_trace`.
1034 */
a4c955d9 1035 out_trace = std::move(traces[winner_i]);
54ef52bd
FD
1036
1037end:
4164020e 1038 return ret;
41a65f30
SM
1039}
1040
4164020e
SM
1041enum target_event
1042{
1043 FIRST_EVENT,
1044 LAST_EVENT,
1719bf64
FD
1045};
1046
4164020e
SM
1047static int decode_clock_snapshot_after_event(struct ctf_fs_trace *ctf_fs_trace,
1048 struct ctf_clock_class *default_cc,
1049 struct ctf_fs_ds_index_entry *index_entry,
1050 enum target_event target_event, uint64_t *cs,
1051 int64_t *ts_ns)
1719bf64 1052{
4164020e 1053 enum ctf_msg_iter_status iter_status = CTF_MSG_ITER_STATUS_OK;
6d4acae3 1054 ctf_msg_iter_up msg_iter;
4164020e
SM
1055 int ret = 0;
1056
1057 BT_ASSERT(ctf_fs_trace);
1058 BT_ASSERT(index_entry);
1059 BT_ASSERT(index_entry->path);
1060
21c7fd8b
SM
1061 const auto ds_file = ctf_fs_ds_file_create(ctf_fs_trace, bt2::Stream::Shared {},
1062 index_entry->path, ctf_fs_trace->logger);
4164020e 1063 if (!ds_file) {
0f5c5d5c 1064 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs_trace->logger, "Failed to create a ctf_fs_ds_file");
4164020e
SM
1065 ret = -1;
1066 goto end;
1067 }
1068
1069 BT_ASSERT(ctf_fs_trace->metadata);
1070 BT_ASSERT(ctf_fs_trace->metadata->tc);
1071
0f5c5d5c 1072 msg_iter = ctf_msg_iter_create(
6d4acae3
SM
1073 ctf_fs_trace->metadata->tc,
1074 bt_common_get_page_size(static_cast<int>(ctf_fs_trace->logger.level())) * 8,
1075 ctf_fs_ds_file_medops, ds_file.get(), NULL, ctf_fs_trace->logger);
4164020e
SM
1076 if (!msg_iter) {
1077 /* ctf_msg_iter_create() logs errors. */
1078 ret = -1;
1079 goto end;
1080 }
1081
1082 /*
1083 * Turn on dry run mode to prevent the creation and usage of Babeltrace
1084 * library objects (bt_field, bt_message_*, etc.).
1085 */
6d4acae3 1086 ctf_msg_iter_set_dry_run(msg_iter.get(), true);
4164020e
SM
1087
1088 /* Seek to the beginning of the target packet. */
6d4acae3 1089 iter_status = ctf_msg_iter_seek(msg_iter.get(), index_entry->offset.bytes());
4164020e
SM
1090 if (iter_status) {
1091 /* ctf_msg_iter_seek() logs errors. */
1092 ret = -1;
1093 goto end;
1094 }
1095
1096 switch (target_event) {
1097 case FIRST_EVENT:
1098 /*
1099 * Start to decode the packet until we reach the end of
1100 * the first event. To extract the first event's clock
1101 * snapshot.
1102 */
6d4acae3 1103 iter_status = ctf_msg_iter_curr_packet_first_event_clock_snapshot(msg_iter.get(), cs);
4164020e
SM
1104 break;
1105 case LAST_EVENT:
1106 /* Decode the packet to extract the last event's clock snapshot. */
6d4acae3 1107 iter_status = ctf_msg_iter_curr_packet_last_event_clock_snapshot(msg_iter.get(), cs);
4164020e
SM
1108 break;
1109 default:
1110 bt_common_abort();
1111 }
1112 if (iter_status) {
1113 ret = -1;
1114 goto end;
1115 }
1116
1117 /* Convert clock snapshot to timestamp. */
1118 ret = bt_util_clock_cycles_to_ns_from_origin(
1119 *cs, default_cc->frequency, default_cc->offset_seconds, default_cc->offset_cycles, ts_ns);
1120 if (ret) {
0f5c5d5c
SM
1121 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs_trace->logger,
1122 "Failed to convert clock snapshot to timestamp");
4164020e
SM
1123 goto end;
1124 }
1719bf64
FD
1125
1126end:
4164020e 1127 return ret;
1719bf64
FD
1128}
1129
4164020e
SM
1130static int decode_packet_first_event_timestamp(struct ctf_fs_trace *ctf_fs_trace,
1131 struct ctf_clock_class *default_cc,
1132 struct ctf_fs_ds_index_entry *index_entry,
1133 uint64_t *cs, int64_t *ts_ns)
c43092a5 1134{
4164020e
SM
1135 return decode_clock_snapshot_after_event(ctf_fs_trace, default_cc, index_entry, FIRST_EVENT, cs,
1136 ts_ns);
c43092a5
FD
1137}
1138
4164020e
SM
1139static int decode_packet_last_event_timestamp(struct ctf_fs_trace *ctf_fs_trace,
1140 struct ctf_clock_class *default_cc,
1141 struct ctf_fs_ds_index_entry *index_entry,
1142 uint64_t *cs, int64_t *ts_ns)
1719bf64 1143{
4164020e
SM
1144 return decode_clock_snapshot_after_event(ctf_fs_trace, default_cc, index_entry, LAST_EVENT, cs,
1145 ts_ns);
1719bf64
FD
1146}
1147
1148/*
1149 * Fix up packet index entries for lttng's "event-after-packet" bug.
1150 * Some buggy lttng tracer versions may emit events with a timestamp that is
1151 * larger (after) than the timestamp_end of the their packets.
1152 *
1153 * To fix up this erroneous data we do the following:
1154 * 1. If it's not the stream file's last packet: set the packet index entry's
1155 * end time to the next packet's beginning time.
1156 * 2. If it's the stream file's last packet, set the packet index entry's end
1157 * time to the packet's last event's time, if any, or to the packet's
1158 * beginning time otherwise.
1159 *
1160 * Known buggy tracer versions:
1161 * - before lttng-ust 2.11.0
1162 * - before lttng-module 2.11.0
1163 * - before lttng-module 2.10.10
1164 * - before lttng-module 2.9.13
1165 */
4164020e 1166static int fix_index_lttng_event_after_packet_bug(struct ctf_fs_trace *trace)
1719bf64 1167{
4164020e 1168 int ret = 0;
4164020e 1169
cdf7de78 1170 for (const auto& ds_file_group : trace->ds_file_groups) {
4164020e 1171 struct ctf_clock_class *default_cc;
4164020e 1172
4164020e 1173 BT_ASSERT(ds_file_group);
fe2f9cda 1174 const auto index = ds_file_group->index.get();
4164020e
SM
1175
1176 BT_ASSERT(index);
2fb7af12 1177 BT_ASSERT(!index->entries.empty());
4164020e
SM
1178
1179 /*
1180 * Iterate over all entries but the last one. The last one is
1181 * fixed differently after.
1182 */
2fb7af12
SM
1183 for (size_t entry_i = 0; entry_i < index->entries.size() - 1; ++entry_i) {
1184 ctf_fs_ds_index_entry *curr_entry = index->entries[entry_i].get();
1185 ctf_fs_ds_index_entry *next_entry = index->entries[entry_i + 1].get();
4164020e
SM
1186
1187 /*
1188 * 1. Set the current index entry `end` timestamp to
1189 * the next index entry `begin` timestamp.
1190 */
1191 curr_entry->timestamp_end = next_entry->timestamp_begin;
1192 curr_entry->timestamp_end_ns = next_entry->timestamp_begin_ns;
1193 }
1194
1195 /*
1196 * 2. Fix the last entry by decoding the last event of the last
1197 * packet.
1198 */
2fb7af12 1199 const auto last_entry = index->entries.back().get();
4164020e
SM
1200 BT_ASSERT(last_entry);
1201
1202 BT_ASSERT(ds_file_group->sc->default_clock_class);
1203 default_cc = ds_file_group->sc->default_clock_class;
1204
1205 /*
1206 * Decode packet to read the timestamp of the last event of the
1207 * entry.
1208 */
1209 ret = decode_packet_last_event_timestamp(trace, default_cc, last_entry,
1210 &last_entry->timestamp_end,
1211 &last_entry->timestamp_end_ns);
1212 if (ret) {
0f5c5d5c
SM
1213 BT_CPPLOGE_APPEND_CAUSE_SPEC(
1214 trace->logger,
4164020e
SM
1215 "Failed to decode stream's last packet to get its last event's clock snapshot.");
1216 goto end;
1217 }
1218 }
1719bf64
FD
1219
1220end:
4164020e 1221 return ret;
1719bf64
FD
1222}
1223
c43092a5
FD
1224/*
1225 * Fix up packet index entries for barectf's "event-before-packet" bug.
1226 * Some buggy barectf tracer versions may emit events with a timestamp that is
1227 * less than the timestamp_begin of the their packets.
1228 *
1229 * To fix up this erroneous data we do the following:
1230 * 1. Starting at the second index entry, set the timestamp_begin of the
1231 * current entry to the timestamp of the first event of the packet.
1232 * 2. Set the previous entry's timestamp_end to the timestamp_begin of the
1233 * current packet.
1234 *
1235 * Known buggy tracer versions:
1236 * - before barectf 2.3.1
1237 */
4164020e 1238static int fix_index_barectf_event_before_packet_bug(struct ctf_fs_trace *trace)
c43092a5 1239{
4164020e 1240 int ret = 0;
4164020e 1241
cdf7de78 1242 for (const auto& ds_file_group : trace->ds_file_groups) {
4164020e 1243 struct ctf_clock_class *default_cc;
fe2f9cda 1244 const auto index = ds_file_group->index.get();
4164020e
SM
1245
1246 BT_ASSERT(index);
2fb7af12 1247 BT_ASSERT(!index->entries.empty());
4164020e
SM
1248
1249 BT_ASSERT(ds_file_group->sc->default_clock_class);
1250 default_cc = ds_file_group->sc->default_clock_class;
1251
1252 /*
1253 * 1. Iterate over the index, starting from the second entry
1254 * (index = 1).
1255 */
2fb7af12
SM
1256 for (size_t entry_i = 1; entry_i < index->entries.size(); ++entry_i) {
1257 ctf_fs_ds_index_entry *prev_entry = index->entries[entry_i - 1].get();
1258 ctf_fs_ds_index_entry *curr_entry = index->entries[entry_i].get();
4164020e
SM
1259 /*
1260 * 2. Set the current entry `begin` timestamp to the
1261 * timestamp of the first event of the current packet.
1262 */
1263 ret = decode_packet_first_event_timestamp(trace, default_cc, curr_entry,
1264 &curr_entry->timestamp_begin,
1265 &curr_entry->timestamp_begin_ns);
1266 if (ret) {
0f5c5d5c
SM
1267 BT_CPPLOGE_APPEND_CAUSE_SPEC(trace->logger,
1268 "Failed to decode first event's clock snapshot");
4164020e
SM
1269 goto end;
1270 }
1271
1272 /*
1273 * 3. Set the previous entry `end` timestamp to the
1274 * timestamp of the first event of the current packet.
1275 */
1276 prev_entry->timestamp_end = curr_entry->timestamp_begin;
1277 prev_entry->timestamp_end_ns = curr_entry->timestamp_begin_ns;
1278 }
1279 }
c43092a5 1280end:
4164020e 1281 return ret;
c43092a5
FD
1282}
1283
aada78b5
FD
1284/*
1285 * When using the lttng-crash feature it's likely that the last packets of each
1286 * stream have their timestamp_end set to zero. This is caused by the fact that
1287 * the tracer crashed and was not able to properly close the packets.
1288 *
1289 * To fix up this erroneous data we do the following:
1290 * For each index entry, if the entry's timestamp_end is 0 and the
1291 * timestamp_begin is not 0:
1292 * - If it's the stream file's last packet: set the packet index entry's end
1293 * time to the packet's last event's time, if any, or to the packet's
1294 * beginning time otherwise.
1295 * - If it's not the stream file's last packet: set the packet index
1296 * entry's end time to the next packet's beginning time.
1297 *
1298 * Affected versions:
1299 * - All current and future lttng-ust and lttng-modules versions.
1300 */
4164020e 1301static int fix_index_lttng_crash_quirk(struct ctf_fs_trace *trace)
aada78b5 1302{
4164020e 1303 int ret = 0;
4164020e 1304
cdf7de78 1305 for (const auto& ds_file_group : trace->ds_file_groups) {
4164020e 1306 struct ctf_clock_class *default_cc;
4164020e 1307
4164020e 1308 BT_ASSERT(ds_file_group);
fe2f9cda 1309 const auto index = ds_file_group->index.get();
4164020e
SM
1310
1311 BT_ASSERT(ds_file_group->sc->default_clock_class);
1312 default_cc = ds_file_group->sc->default_clock_class;
1313
1314 BT_ASSERT(index);
2fb7af12 1315 BT_ASSERT(!index->entries.empty());
4164020e 1316
2fb7af12 1317 const auto last_entry = index->entries.back().get();
4164020e
SM
1318 BT_ASSERT(last_entry);
1319
1320 /* 1. Fix the last entry first. */
1321 if (last_entry->timestamp_end == 0 && last_entry->timestamp_begin != 0) {
1322 /*
1323 * Decode packet to read the timestamp of the
1324 * last event of the stream file.
1325 */
1326 ret = decode_packet_last_event_timestamp(trace, default_cc, last_entry,
1327 &last_entry->timestamp_end,
1328 &last_entry->timestamp_end_ns);
1329 if (ret) {
0f5c5d5c
SM
1330 BT_CPPLOGE_APPEND_CAUSE_SPEC(trace->logger,
1331 "Failed to decode last event's clock snapshot");
4164020e
SM
1332 goto end;
1333 }
1334 }
1335
1336 /* Iterate over all entries but the last one. */
2fb7af12
SM
1337 for (size_t entry_idx = 0; entry_idx < index->entries.size() - 1; ++entry_idx) {
1338 ctf_fs_ds_index_entry *curr_entry = index->entries[entry_idx].get();
1339 ctf_fs_ds_index_entry *next_entry = index->entries[entry_idx + 1].get();
4164020e
SM
1340
1341 if (curr_entry->timestamp_end == 0 && curr_entry->timestamp_begin != 0) {
1342 /*
1343 * 2. Set the current index entry `end` timestamp to
1344 * the next index entry `begin` timestamp.
1345 */
1346 curr_entry->timestamp_end = next_entry->timestamp_begin;
1347 curr_entry->timestamp_end_ns = next_entry->timestamp_begin_ns;
1348 }
1349 }
1350 }
aada78b5
FD
1351
1352end:
4164020e 1353 return ret;
aada78b5
FD
1354}
1355
626cc488
FD
1356/*
1357 * Extract the tracer information necessary to compare versions.
1358 * Returns 0 on success, and -1 if the extraction is not successful because the
1359 * necessary fields are absents in the trace metadata.
1360 */
4164020e 1361static int extract_tracer_info(struct ctf_fs_trace *trace, struct tracer_info *current_tracer_info)
626cc488 1362{
4164020e
SM
1363 int ret = 0;
1364 struct ctf_trace_class_env_entry *entry;
1365
1366 /* Clear the current_tracer_info struct */
1367 memset(current_tracer_info, 0, sizeof(*current_tracer_info));
1368
1369 /*
1370 * To compare 2 tracer versions, at least the tracer name and it's
1371 * major version are needed. If one of these is missing, consider it an
1372 * extraction failure.
1373 */
1374 entry = ctf_trace_class_borrow_env_entry_by_name(trace->metadata->tc, "tracer_name");
1375 if (!entry || entry->type != CTF_TRACE_CLASS_ENV_ENTRY_TYPE_STR) {
1376 goto missing_bare_minimum;
1377 }
1378
1379 /* Set tracer name. */
1380 current_tracer_info->name = entry->value.str->str;
1381
1382 entry = ctf_trace_class_borrow_env_entry_by_name(trace->metadata->tc, "tracer_major");
1383 if (!entry || entry->type != CTF_TRACE_CLASS_ENV_ENTRY_TYPE_INT) {
1384 goto missing_bare_minimum;
1385 }
1386
1387 /* Set major version number. */
1388 current_tracer_info->major = entry->value.i;
1389
1390 entry = ctf_trace_class_borrow_env_entry_by_name(trace->metadata->tc, "tracer_minor");
1391 if (!entry || entry->type != CTF_TRACE_CLASS_ENV_ENTRY_TYPE_INT) {
1392 goto end;
1393 }
1394
1395 /* Set minor version number. */
1396 current_tracer_info->minor = entry->value.i;
1397
1398 entry = ctf_trace_class_borrow_env_entry_by_name(trace->metadata->tc, "tracer_patch");
1399 if (!entry) {
1400 /*
1401 * If `tracer_patch` doesn't exist `tracer_patchlevel` might.
1402 * For example, `lttng-modules` uses entry name
1403 * `tracer_patchlevel`.
1404 */
1405 entry = ctf_trace_class_borrow_env_entry_by_name(trace->metadata->tc, "tracer_patchlevel");
1406 }
1407
1408 if (!entry || entry->type != CTF_TRACE_CLASS_ENV_ENTRY_TYPE_INT) {
1409 goto end;
1410 }
1411
1412 /* Set patch version number. */
1413 current_tracer_info->patch = entry->value.i;
1414
1415 goto end;
626cc488
FD
1416
1417missing_bare_minimum:
4164020e 1418 ret = -1;
626cc488 1419end:
4164020e 1420 return ret;
626cc488
FD
1421}
1422
4164020e 1423static bool is_tracer_affected_by_lttng_event_after_packet_bug(struct tracer_info *curr_tracer_info)
1719bf64 1424{
4164020e
SM
1425 bool is_affected = false;
1426
1427 if (strcmp(curr_tracer_info->name, "lttng-ust") == 0) {
1428 if (curr_tracer_info->major < 2) {
1429 is_affected = true;
1430 } else if (curr_tracer_info->major == 2) {
1431 /* fixed in lttng-ust 2.11.0 */
1432 if (curr_tracer_info->minor < 11) {
1433 is_affected = true;
1434 }
1435 }
1436 } else if (strcmp(curr_tracer_info->name, "lttng-modules") == 0) {
1437 if (curr_tracer_info->major < 2) {
1438 is_affected = true;
1439 } else if (curr_tracer_info->major == 2) {
1440 /* fixed in lttng-modules 2.11.0 */
1441 if (curr_tracer_info->minor == 10) {
1442 /* fixed in lttng-modules 2.10.10 */
1443 if (curr_tracer_info->patch < 10) {
1444 is_affected = true;
1445 }
1446 } else if (curr_tracer_info->minor == 9) {
1447 /* fixed in lttng-modules 2.9.13 */
1448 if (curr_tracer_info->patch < 13) {
1449 is_affected = true;
1450 }
1451 } else if (curr_tracer_info->minor < 9) {
1452 is_affected = true;
1453 }
1454 }
1455 }
1456
1457 return is_affected;
1719bf64
FD
1458}
1459
4164020e
SM
1460static bool
1461is_tracer_affected_by_barectf_event_before_packet_bug(struct tracer_info *curr_tracer_info)
c43092a5 1462{
4164020e
SM
1463 bool is_affected = false;
1464
1465 if (strcmp(curr_tracer_info->name, "barectf") == 0) {
1466 if (curr_tracer_info->major < 2) {
1467 is_affected = true;
1468 } else if (curr_tracer_info->major == 2) {
1469 if (curr_tracer_info->minor < 3) {
1470 is_affected = true;
1471 } else if (curr_tracer_info->minor == 3) {
1472 /* fixed in barectf 2.3.1 */
1473 if (curr_tracer_info->patch < 1) {
1474 is_affected = true;
1475 }
1476 }
1477 }
1478 }
1479
1480 return is_affected;
c43092a5
FD
1481}
1482
4164020e 1483static bool is_tracer_affected_by_lttng_crash_quirk(struct tracer_info *curr_tracer_info)
aada78b5 1484{
4164020e 1485 bool is_affected = false;
aada78b5 1486
4164020e
SM
1487 /* All LTTng tracer may be affected by this lttng crash quirk. */
1488 if (strcmp(curr_tracer_info->name, "lttng-ust") == 0) {
1489 is_affected = true;
1490 } else if (strcmp(curr_tracer_info->name, "lttng-modules") == 0) {
1491 is_affected = true;
1492 }
aada78b5 1493
4164020e 1494 return is_affected;
aada78b5
FD
1495}
1496
1719bf64
FD
1497/*
1498 * Looks for trace produced by known buggy tracers and fix up the index
1499 * produced earlier.
1500 */
0f5c5d5c 1501static int fix_packet_index_tracer_bugs(ctf_fs_trace *trace)
1719bf64 1502{
4164020e
SM
1503 int ret = 0;
1504 struct tracer_info current_tracer_info;
4164020e 1505
0f5c5d5c 1506 ret = extract_tracer_info(trace, &current_tracer_info);
4164020e
SM
1507 if (ret) {
1508 /*
1509 * A trace may not have all the necessary environment
1510 * entries to do the tracer version comparison.
1511 * At least, the tracer name and major version number
1512 * are needed. Failing to extract these entries is not
1513 * an error.
1514 */
1515 ret = 0;
0f5c5d5c
SM
1516 BT_CPPLOGI_STR_SPEC(
1517 trace->logger,
1518 "Cannot extract tracer information necessary to compare with buggy versions.");
4164020e 1519 goto end;
4164020e
SM
1520 }
1521
1522 /* Check if the trace may be affected by old tracer bugs. */
1523 if (is_tracer_affected_by_lttng_event_after_packet_bug(&current_tracer_info)) {
0f5c5d5c
SM
1524 BT_CPPLOGI_STR_SPEC(
1525 trace->logger,
1526 "Trace may be affected by LTTng tracer packet timestamp bug. Fixing up.");
1527 ret = fix_index_lttng_event_after_packet_bug(trace);
4164020e 1528 if (ret) {
0f5c5d5c
SM
1529 BT_CPPLOGE_APPEND_CAUSE_SPEC(trace->logger,
1530 "Failed to fix LTTng event-after-packet bug.");
4164020e
SM
1531 goto end;
1532 }
0f5c5d5c 1533 trace->metadata->tc->quirks.lttng_event_after_packet = true;
4164020e
SM
1534 }
1535
1536 if (is_tracer_affected_by_barectf_event_before_packet_bug(&current_tracer_info)) {
0f5c5d5c
SM
1537 BT_CPPLOGI_STR_SPEC(
1538 trace->logger,
1539 "Trace may be affected by barectf tracer packet timestamp bug. Fixing up.");
1540 ret = fix_index_barectf_event_before_packet_bug(trace);
4164020e 1541 if (ret) {
0f5c5d5c
SM
1542 BT_CPPLOGE_APPEND_CAUSE_SPEC(trace->logger,
1543 "Failed to fix barectf event-before-packet bug.");
4164020e
SM
1544 goto end;
1545 }
0f5c5d5c 1546 trace->metadata->tc->quirks.barectf_event_before_packet = true;
4164020e
SM
1547 }
1548
1549 if (is_tracer_affected_by_lttng_crash_quirk(&current_tracer_info)) {
0f5c5d5c 1550 ret = fix_index_lttng_crash_quirk(trace);
4164020e 1551 if (ret) {
0f5c5d5c
SM
1552 BT_CPPLOGE_APPEND_CAUSE_SPEC(trace->logger,
1553 "Failed to fix lttng-crash timestamp quirks.");
4164020e
SM
1554 goto end;
1555 }
0f5c5d5c 1556 trace->metadata->tc->quirks.lttng_crash = true;
4164020e 1557 }
a0cd55ad 1558
1719bf64 1559end:
4164020e 1560 return ret;
1719bf64
FD
1561}
1562
cdf7de78
SM
1563static bool compare_ds_file_groups_by_first_path(const ctf_fs_ds_file_group::UP& ds_file_group_a,
1564 const ctf_fs_ds_file_group::UP& ds_file_group_b)
e9b3611f 1565{
f3d74124
SM
1566 BT_ASSERT(!ds_file_group_a->ds_file_infos.empty());
1567 BT_ASSERT(!ds_file_group_b->ds_file_infos.empty());
087cd0f5 1568
f3d74124
SM
1569 const auto& first_ds_file_info_a = *ds_file_group_a->ds_file_infos[0];
1570 const auto& first_ds_file_info_b = *ds_file_group_b->ds_file_infos[0];
087cd0f5 1571
f3d74124 1572 return first_ds_file_info_a.path < first_ds_file_info_b.path;
e9b3611f
PP
1573}
1574
4164020e 1575static gint compare_strings(gconstpointer p_a, gconstpointer p_b)
7b69723d 1576{
4164020e
SM
1577 const char *a = *((const char **) p_a);
1578 const char *b = *((const char **) p_b);
7b69723d 1579
4164020e 1580 return strcmp(a, b);
7b69723d
SM
1581}
1582
4164020e
SM
1583int ctf_fs_component_create_ctf_fs_trace(struct ctf_fs_component *ctf_fs,
1584 const bt_value *paths_value,
1585 const bt_value *trace_name_value,
0f5c5d5c 1586 bt_self_component *selfComp)
f280892e 1587{
4164020e
SM
1588 int ret = 0;
1589 uint64_t i;
4164020e 1590 GPtrArray *paths = NULL;
a4c955d9 1591 std::vector<ctf_fs_trace::UP> traces;
4164020e
SM
1592 const char *trace_name;
1593
1594 BT_ASSERT(bt_value_get_type(paths_value) == BT_VALUE_TYPE_ARRAY);
1595 BT_ASSERT(!bt_value_array_is_empty(paths_value));
1596
4164020e
SM
1597 paths = g_ptr_array_new_with_free_func(g_free);
1598 if (!paths) {
0f5c5d5c 1599 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger, "Failed to allocate a GPtrArray.");
4164020e
SM
1600 goto error;
1601 }
1602
1603 trace_name = trace_name_value ? bt_value_string_get(trace_name_value) : NULL;
1604
1605 /*
1606 * Create a sorted array of the paths, to make the execution of this
1607 * component deterministic.
1608 */
1609 for (i = 0; i < bt_value_array_get_length(paths_value); i++) {
1610 const bt_value *path_value = bt_value_array_borrow_element_by_index_const(paths_value, i);
1611 const char *input = bt_value_string_get(path_value);
1612 gchar *input_copy;
1613
1614 input_copy = g_strdup(input);
1615 if (!input_copy) {
0f5c5d5c 1616 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger, "Failed to copy a string.");
4164020e
SM
1617 goto error;
1618 }
1619
1620 g_ptr_array_add(paths, input_copy);
1621 }
1622
1623 g_ptr_array_sort(paths, compare_strings);
1624
1625 /* Create a separate ctf_fs_trace object for each path. */
1626 for (i = 0; i < paths->len; i++) {
1627 const char *path = (const char *) g_ptr_array_index(paths, i);
1628
1629 ret = ctf_fs_component_create_ctf_fs_trace_one_path(ctf_fs, path, trace_name, traces,
0f5c5d5c 1630 selfComp);
4164020e
SM
1631 if (ret) {
1632 goto end;
1633 }
1634 }
1635
a4c955d9
SM
1636 if (traces.size() > 1) {
1637 ctf_fs_trace *first_trace = traces[0].get();
4164020e 1638 const uint8_t *first_trace_uuid = first_trace->metadata->tc->uuid;
4164020e
SM
1639
1640 /*
1641 * We have more than one trace, they must all share the same
1642 * UUID, verify that.
1643 */
a4c955d9
SM
1644 for (i = 0; i < traces.size(); i++) {
1645 ctf_fs_trace *this_trace = traces[i].get();
4164020e
SM
1646 const uint8_t *this_trace_uuid = this_trace->metadata->tc->uuid;
1647
1648 if (!this_trace->metadata->tc->is_uuid_set) {
0f5c5d5c
SM
1649 BT_CPPLOGE_APPEND_CAUSE_SPEC(
1650 ctf_fs->logger,
1651 "Multiple traces given, but a trace does not have a UUID: path={}",
cee8a466 1652 this_trace->path);
4164020e
SM
1653 goto error;
1654 }
1655
1656 if (bt_uuid_compare(first_trace_uuid, this_trace_uuid) != 0) {
1657 char first_trace_uuid_str[BT_UUID_STR_LEN + 1];
1658 char this_trace_uuid_str[BT_UUID_STR_LEN + 1];
1659
1660 bt_uuid_to_str(first_trace_uuid, first_trace_uuid_str);
1661 bt_uuid_to_str(this_trace_uuid, this_trace_uuid_str);
1662
0f5c5d5c
SM
1663 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger,
1664 "Multiple traces given, but UUIDs don't match: "
1665 "first-trace-uuid={}, first-trace-path={}, "
1666 "trace-uuid={}, trace-path={}",
cee8a466
SM
1667 first_trace_uuid_str, first_trace->path,
1668 this_trace_uuid_str, this_trace->path);
4164020e
SM
1669 goto error;
1670 }
1671 }
1672
a4c955d9 1673 ret = merge_ctf_fs_traces(std::move(traces), ctf_fs->trace);
4164020e 1674 if (ret) {
0f5c5d5c
SM
1675 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger,
1676 "Failed to merge traces with the same UUID.");
4164020e
SM
1677 goto error;
1678 }
4164020e
SM
1679 } else {
1680 /* Just one trace, it may or may not have a UUID, both are fine. */
a4c955d9 1681 ctf_fs->trace = std::move(traces[0]);
4164020e
SM
1682 }
1683
7df773f2 1684 ret = fix_packet_index_tracer_bugs(ctf_fs->trace.get());
4164020e 1685 if (ret) {
0f5c5d5c 1686 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger, "Failed to fix packet index tracer bugs.");
4164020e
SM
1687 }
1688
1689 /*
1690 * Sort data stream file groups by first data stream file info
1691 * path to get a deterministic order. This order influences the
1692 * order of the output ports. It also influences the order of
1693 * the automatic stream IDs if the trace's packet headers do not
1694 * contain a `stream_instance_id` field, in which case the data
1695 * stream file to stream ID association is always the same,
1696 * whatever the build and the system.
1697 *
1698 * Having a deterministic order here can help debugging and
1699 * testing.
1700 */
cdf7de78
SM
1701 std::sort(ctf_fs->trace->ds_file_groups.begin(), ctf_fs->trace->ds_file_groups.end(),
1702 compare_ds_file_groups_by_first_path);
4164020e 1703 goto end;
a0cd55ad 1704error:
4164020e 1705 ret = -1;
a0cd55ad 1706
f280892e 1707end:
4164020e
SM
1708 if (paths) {
1709 g_ptr_array_free(paths, TRUE);
1710 }
7b69723d 1711
4164020e 1712 return ret;
f280892e
SM
1713}
1714
4164020e 1715static GString *get_stream_instance_unique_name(struct ctf_fs_ds_file_group *ds_file_group)
a38d7650 1716{
4164020e
SM
1717 GString *name;
1718 struct ctf_fs_ds_file_info *ds_file_info;
1719
1720 name = g_string_new(NULL);
1721 if (!name) {
1722 goto end;
1723 }
1724
1725 /*
1726 * If there's more than one stream file in the stream file
1727 * group, the first (earliest) stream file's path is used as
1728 * the stream's unique name.
1729 */
f3d74124
SM
1730 BT_ASSERT(!ds_file_group->ds_file_infos.empty());
1731 ds_file_info = ds_file_group->ds_file_infos[0].get();
4d199954 1732 g_string_assign(name, ds_file_info->path.c_str());
a38d7650
SM
1733
1734end:
4164020e 1735 return name;
a38d7650
SM
1736}
1737
f280892e
SM
1738/* Create the IR stream objects for ctf_fs_trace. */
1739
4164020e 1740static int create_streams_for_trace(struct ctf_fs_trace *ctf_fs_trace)
f280892e 1741{
4164020e
SM
1742 int ret;
1743 GString *name = NULL;
4164020e 1744
cdf7de78
SM
1745 for (const auto& ds_file_group : ctf_fs_trace->ds_file_groups) {
1746 name = get_stream_instance_unique_name(ds_file_group.get());
4164020e
SM
1747
1748 if (!name) {
1749 goto error;
1750 }
1751
f0940b01
SM
1752 BT_ASSERT(ds_file_group->sc->ir_sc);
1753 BT_ASSERT(ctf_fs_trace->trace);
1754
be215bcd
SM
1755 bt_stream *stream;
1756
f0940b01
SM
1757 if (ds_file_group->stream_id == UINT64_C(-1)) {
1758 /* No stream ID: use 0 */
e44859b1
SM
1759 stream =
1760 bt_stream_create_with_id(ds_file_group->sc->ir_sc, ctf_fs_trace->trace->libObjPtr(),
1761 ctf_fs_trace->next_stream_id);
f0940b01 1762 ctf_fs_trace->next_stream_id++;
4164020e 1763 } else {
f0940b01 1764 /* Specific stream ID */
e44859b1
SM
1765 stream =
1766 bt_stream_create_with_id(ds_file_group->sc->ir_sc, ctf_fs_trace->trace->libObjPtr(),
1767 (uint64_t) ds_file_group->stream_id);
4164020e
SM
1768 }
1769
be215bcd 1770 if (!stream) {
0f5c5d5c
SM
1771 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs_trace->logger,
1772 "Cannot create stream for DS file group: "
1773 "addr={}, stream-name=\"{}\"",
1774 fmt::ptr(ds_file_group), name->str);
4164020e
SM
1775 goto error;
1776 }
1777
be215bcd
SM
1778 ds_file_group->stream = bt2::Stream::Shared::createWithoutRef(stream);
1779
1780 ret = bt_stream_set_name(ds_file_group->stream->libObjPtr(), name->str);
4164020e 1781 if (ret) {
0f5c5d5c
SM
1782 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs_trace->logger,
1783 "Cannot set stream's name: "
1784 "addr={}, stream-name=\"{}\"",
be215bcd 1785 fmt::ptr(ds_file_group->stream->libObjPtr()), name->str);
4164020e
SM
1786 goto error;
1787 }
1788
1789 g_string_free(name, TRUE);
1790 name = NULL;
1791 }
1792
1793 ret = 0;
1794 goto end;
f280892e
SM
1795
1796error:
4164020e 1797 ret = -1;
f280892e
SM
1798
1799end:
1800
4164020e
SM
1801 if (name) {
1802 g_string_free(name, TRUE);
1803 }
1804 return ret;
f280892e
SM
1805}
1806
88730e42
SM
1807static const bt_param_validation_value_descr inputs_elem_descr =
1808 bt_param_validation_value_descr::makeString();
087cd0f5
SM
1809
1810static bt_param_validation_map_value_entry_descr fs_params_entries_descr[] = {
88730e42
SM
1811 {"inputs", BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_MANDATORY,
1812 bt_param_validation_value_descr::makeArray(1, BT_PARAM_VALIDATION_INFINITE,
1813 inputs_elem_descr)},
1814 {"trace-name", BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_OPTIONAL,
1815 bt_param_validation_value_descr::makeString()},
1816 {"clock-class-offset-s", BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_OPTIONAL,
1817 bt_param_validation_value_descr::makeSignedInteger()},
1818 {"clock-class-offset-ns", BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_OPTIONAL,
1819 bt_param_validation_value_descr::makeSignedInteger()},
1820 {"force-clock-class-origin-unix-epoch", BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_OPTIONAL,
1821 bt_param_validation_value_descr::makeBool()},
4164020e
SM
1822 BT_PARAM_VALIDATION_MAP_VALUE_ENTRY_END};
1823
1824bool read_src_fs_parameters(const bt_value *params, const bt_value **inputs,
0f5c5d5c 1825 const bt_value **trace_name, struct ctf_fs_component *ctf_fs)
4164020e
SM
1826{
1827 bool ret;
1828 const bt_value *value;
4164020e
SM
1829 enum bt_param_validation_status validate_value_status;
1830 gchar *error = NULL;
1831
1832 validate_value_status = bt_param_validation_validate(params, fs_params_entries_descr, &error);
1833 if (validate_value_status != BT_PARAM_VALIDATION_STATUS_OK) {
0f5c5d5c 1834 BT_CPPLOGE_APPEND_CAUSE_SPEC(ctf_fs->logger, "{}", error);
4164020e
SM
1835 ret = false;
1836 goto end;
1837 }
1838
1839 /* inputs parameter */
1840 *inputs = bt_value_map_borrow_entry_value_const(params, "inputs");
1841
1842 /* clock-class-offset-s parameter */
1843 value = bt_value_map_borrow_entry_value_const(params, "clock-class-offset-s");
1844 if (value) {
c942e7a2 1845 ctf_fs->clkClsCfg.offsetSec = bt_value_integer_signed_get(value);
4164020e
SM
1846 }
1847
1848 /* clock-class-offset-ns parameter */
1849 value = bt_value_map_borrow_entry_value_const(params, "clock-class-offset-ns");
1850 if (value) {
c942e7a2 1851 ctf_fs->clkClsCfg.offsetNanoSec = bt_value_integer_signed_get(value);
4164020e
SM
1852 }
1853
1854 /* force-clock-class-origin-unix-epoch parameter */
1855 value = bt_value_map_borrow_entry_value_const(params, "force-clock-class-origin-unix-epoch");
1856 if (value) {
c942e7a2 1857 ctf_fs->clkClsCfg.forceOriginIsUnixEpoch = bt_value_bool_get(value);
4164020e
SM
1858 }
1859
1860 /* trace-name parameter */
1861 *trace_name = bt_value_map_borrow_entry_value_const(params, "trace-name");
1862
1863 ret = true;
d907165c
SM
1864
1865end:
4164020e
SM
1866 g_free(error);
1867 return ret;
d907165c
SM
1868}
1869
f340a3e8
SM
1870static ctf_fs_component::UP ctf_fs_create(const bt_value *params,
1871 bt_self_component_source *self_comp_src)
56a1cced 1872{
4164020e
SM
1873 const bt_value *inputs_value;
1874 const bt_value *trace_name_value;
1875 bt_self_component *self_comp = bt_self_component_source_as_self_component(self_comp_src);
56a1cced 1876
f340a3e8 1877 ctf_fs_component::UP ctf_fs = ctf_fs_component_create(
0f5c5d5c 1878 bt2c::Logger {bt2::SelfSourceComponent {self_comp_src}, "PLUGIN/SRC.CTF.FS/COMP"});
4164020e 1879 if (!ctf_fs) {
f340a3e8 1880 return nullptr;
4164020e 1881 }
f280892e 1882
f340a3e8
SM
1883 if (!read_src_fs_parameters(params, &inputs_value, &trace_name_value, ctf_fs.get())) {
1884 return nullptr;
4164020e 1885 }
56a1cced 1886
f340a3e8
SM
1887 if (ctf_fs_component_create_ctf_fs_trace(ctf_fs.get(), inputs_value, trace_name_value,
1888 self_comp)) {
1889 return nullptr;
4164020e 1890 }
4f1f88a6 1891
7df773f2 1892 if (create_streams_for_trace(ctf_fs->trace.get())) {
f340a3e8 1893 return nullptr;
4164020e 1894 }
f280892e 1895
7df773f2 1896 if (create_ports_for_trace(ctf_fs.get(), ctf_fs->trace.get(), self_comp_src)) {
f340a3e8 1897 return nullptr;
4164020e 1898 }
4f1f88a6 1899
4164020e 1900 return ctf_fs;
56a1cced
JG
1901}
1902
50b9f4b5
SM
1903bt_component_class_initialize_method_status ctf_fs_init(bt_self_component_source *self_comp_src,
1904 bt_self_component_source_configuration *,
1905 const bt_value *params, void *)
ea0b4b9e 1906{
1e690349 1907 try {
1e690349
SM
1908 bt_component_class_initialize_method_status ret =
1909 BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_OK;
1910
f340a3e8 1911 ctf_fs_component::UP ctf_fs = ctf_fs_create(params, self_comp_src);
1e690349
SM
1912 if (!ctf_fs) {
1913 ret = BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
1914 }
ea0b4b9e 1915
f340a3e8
SM
1916 bt_self_component_set_data(bt_self_component_source_as_self_component(self_comp_src),
1917 ctf_fs.release());
1e690349
SM
1918 return ret;
1919 } catch (const std::bad_alloc&) {
1920 return BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_MEMORY_ERROR;
1921 } catch (const bt2::Error&) {
1922 return BT_COMPONENT_CLASS_INITIALIZE_METHOD_STATUS_ERROR;
4164020e 1923 }
ea0b4b9e 1924}
33f93973 1925
0f5c5d5c 1926bt_component_class_query_method_status ctf_fs_query(bt_self_component_class_source *comp_class_src,
4164020e
SM
1927 bt_private_query_executor *priv_query_exec,
1928 const char *object, const bt_value *params,
1929 __attribute__((unused)) void *method_data,
1930 const bt_value **result)
33f93973 1931{
1e690349 1932 try {
1e690349
SM
1933 bt2c::Logger logger {bt2::SelfComponentClass {comp_class_src},
1934 bt2::PrivateQueryExecutor {priv_query_exec},
1935 "PLUGIN/SRC.CTF.FS/QUERY"};
c02af779
SM
1936 bt2::ConstMapValue paramsObj(params);
1937 bt2::Value::Shared resultObj;
1e690349
SM
1938
1939 if (strcmp(object, "metadata-info") == 0) {
c02af779 1940 resultObj = metadata_info_query(paramsObj, logger);
1e690349 1941 } else if (strcmp(object, "babeltrace.trace-infos") == 0) {
c02af779 1942 resultObj = trace_infos_query(paramsObj, logger);
1e690349 1943 } else if (!strcmp(object, "babeltrace.support-info")) {
c02af779 1944 resultObj = support_info_query(paramsObj, logger);
1e690349
SM
1945 } else {
1946 BT_CPPLOGE_SPEC(logger, "Unknown query object `{}`", object);
c02af779 1947 return BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_UNKNOWN_OBJECT;
1e690349 1948 }
c02af779
SM
1949
1950 *result = resultObj.release().libObjPtr();
1951
1952 return BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_OK;
1e690349
SM
1953 } catch (const std::bad_alloc&) {
1954 return BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_MEMORY_ERROR;
1955 } catch (const bt2::Error&) {
1956 return BT_COMPONENT_CLASS_QUERY_METHOD_STATUS_ERROR;
1957 }
33f93973 1958}
This page took 0.230305 seconds and 4 git commands to generate.