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