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