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