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