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