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