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