4 * Babeltrace library to create a copy of a CTF trace
6 * Copyright 2017 Julien Desfossez <jdesfossez@efficios.com>
8 * Author: Julien Desfossez <jdesfossez@efficios.com>
10 * Permission is hereby granted, free of charge, to any person obtaining a copy
11 * of this software and associated documentation files (the "Software"), to deal
12 * in the Software without restriction, including without limitation the rights
13 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
14 * copies of the Software, and to permit persons to whom the Software is
15 * furnished to do so, subject to the following conditions:
17 * The above copyright notice and this permission notice shall be included in
18 * all copies or substantial portions of the Software.
20 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
21 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
23 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
24 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
25 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
29 #include <babeltrace/ctf-ir/event.h>
30 #include <babeltrace/ctf-ir/packet.h>
31 #include <babeltrace/ctf-ir/event-class.h>
32 #include <babeltrace/ctf-ir/stream.h>
33 #include <babeltrace/ctf-ir/stream-class.h>
34 #include <babeltrace/ctf-ir/clock-class.h>
35 #include <babeltrace/ctf-ir/fields.h>
36 #include <babeltrace/ctf-writer/stream.h>
39 #include "ctfcopytrace.h"
40 #include "clock-fields.h"
43 struct bt_ctf_clock_class
*ctf_copy_clock_class(FILE *err
,
44 struct bt_ctf_clock_class
*clock_class
)
46 int64_t offset
, offset_s
;
49 const char *name
, *description
;
50 struct bt_ctf_clock_class
*writer_clock_class
= NULL
;
52 assert(err
&& clock_class
);
54 name
= bt_ctf_clock_class_get_name(clock_class
);
56 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
61 writer_clock_class
= bt_ctf_clock_class_create(name
);
62 if (!writer_clock_class
) {
63 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
68 description
= bt_ctf_clock_class_get_description(clock_class
);
70 int_ret
= bt_ctf_clock_class_set_description(writer_clock_class
,
73 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
79 u64_ret
= bt_ctf_clock_class_get_frequency(clock_class
);
80 if (u64_ret
== -1ULL) {
81 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
85 int_ret
= bt_ctf_clock_class_set_frequency(writer_clock_class
, u64_ret
);
87 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
92 u64_ret
= bt_ctf_clock_class_get_precision(clock_class
);
93 if (u64_ret
== -1ULL) {
94 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
98 int_ret
= bt_ctf_clock_class_set_precision(writer_clock_class
,
101 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
106 int_ret
= bt_ctf_clock_class_get_offset_s(clock_class
, &offset_s
);
108 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
113 int_ret
= bt_ctf_clock_class_set_offset_s(writer_clock_class
, offset_s
);
115 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
120 int_ret
= bt_ctf_clock_class_get_offset_cycles(clock_class
, &offset
);
122 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
127 int_ret
= bt_ctf_clock_class_set_offset_cycles(writer_clock_class
, offset
);
129 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
134 int_ret
= bt_ctf_clock_class_is_absolute(clock_class
);
136 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
141 int_ret
= bt_ctf_clock_class_set_is_absolute(writer_clock_class
, int_ret
);
143 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
151 BT_PUT(writer_clock_class
);
153 return writer_clock_class
;
157 enum bt_component_status
ctf_copy_clock_classes(FILE *err
,
158 struct bt_ctf_trace
*writer_trace
,
159 struct bt_ctf_stream_class
*writer_stream_class
,
160 struct bt_ctf_trace
*trace
)
162 enum bt_component_status ret
;
163 int int_ret
, clock_class_count
, i
;
165 clock_class_count
= bt_ctf_trace_get_clock_class_count(trace
);
167 for (i
= 0; i
< clock_class_count
; i
++) {
168 struct bt_ctf_clock_class
*writer_clock_class
;
169 struct bt_ctf_clock_class
*clock_class
=
170 bt_ctf_trace_get_clock_class_by_index(trace
, i
);
173 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
175 ret
= BT_COMPONENT_STATUS_ERROR
;
179 writer_clock_class
= ctf_copy_clock_class(err
, clock_class
);
181 if (!writer_clock_class
) {
182 fprintf(err
, "Failed to copy clock class");
183 ret
= BT_COMPONENT_STATUS_ERROR
;
187 int_ret
= bt_ctf_trace_add_clock_class(writer_trace
, writer_clock_class
);
189 BT_PUT(writer_clock_class
);
190 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
192 ret
= BT_COMPONENT_STATUS_ERROR
;
197 * Ownership transferred to the trace.
199 bt_put(writer_clock_class
);
202 ret
= BT_COMPONENT_STATUS_OK
;
209 struct bt_ctf_event_class
*ctf_copy_event_class(FILE *err
,
210 struct bt_ctf_event_class
*event_class
)
212 struct bt_ctf_event_class
*writer_event_class
= NULL
;
213 struct bt_ctf_field_type
*context
, *payload_type
;
217 enum bt_ctf_event_class_log_level log_level
;
220 name
= bt_ctf_event_class_get_name(event_class
);
222 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
227 writer_event_class
= bt_ctf_event_class_create(name
);
228 if (!writer_event_class
) {
229 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
234 id
= bt_ctf_event_class_get_id(event_class
);
236 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
241 ret
= bt_ctf_event_class_set_id(writer_event_class
, id
);
243 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
248 log_level
= bt_ctf_event_class_get_log_level(event_class
);
250 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
255 ret
= bt_ctf_event_class_set_log_level(writer_event_class
, log_level
);
257 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
262 emf_uri
= bt_ctf_event_class_get_emf_uri(event_class
);
264 ret
= bt_ctf_event_class_set_emf_uri(writer_event_class
,
267 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
273 payload_type
= bt_ctf_event_class_get_payload_type(event_class
);
275 ret
= bt_ctf_event_class_set_payload_type(writer_event_class
,
278 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
282 BT_PUT(payload_type
);
285 context
= bt_ctf_event_class_get_context_type(event_class
);
287 ret
= bt_ctf_event_class_set_context_type(
288 writer_event_class
, context
);
291 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
300 BT_PUT(writer_event_class
);
302 return writer_event_class
;
306 enum bt_component_status
ctf_copy_event_classes(FILE *err
,
307 struct bt_ctf_stream_class
*stream_class
,
308 struct bt_ctf_stream_class
*writer_stream_class
)
310 enum bt_component_status ret
= BT_COMPONENT_STATUS_OK
;
311 struct bt_ctf_event_class
*event_class
= NULL
, *writer_event_class
= NULL
;
314 count
= bt_ctf_stream_class_get_event_class_count(stream_class
);
316 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
321 for (i
= 0; i
< count
; i
++) {
324 event_class
= bt_ctf_stream_class_get_event_class_by_index(
327 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
329 ret
= BT_COMPONENT_STATUS_ERROR
;
332 if (i
< bt_ctf_stream_class_get_event_class_count(writer_stream_class
)) {
333 writer_event_class
= bt_ctf_stream_class_get_event_class_by_index(
334 writer_stream_class
, i
);
335 if (writer_event_class
) {
337 * If the writer_event_class already exists,
338 * just skip it. It can be used to resync the
339 * event_classes after a trace has become
342 BT_PUT(writer_event_class
);
348 writer_event_class
= ctf_copy_event_class(err
, event_class
);
349 if (!writer_event_class
) {
350 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
352 ret
= BT_COMPONENT_STATUS_ERROR
;
356 int_ret
= bt_ctf_stream_class_add_event_class(writer_stream_class
,
359 fprintf(err
, "[error] Failed to add event class\n");
360 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
362 ret
= BT_COMPONENT_STATUS_ERROR
;
365 BT_PUT(writer_event_class
);
373 bt_put(writer_event_class
);
379 struct bt_ctf_stream_class
*ctf_copy_stream_class(FILE *err
,
380 struct bt_ctf_stream_class
*stream_class
,
381 struct bt_ctf_trace
*writer_trace
,
384 struct bt_ctf_field_type
*type
= NULL
;
385 struct bt_ctf_stream_class
*writer_stream_class
= NULL
;
387 const char *name
= bt_ctf_stream_class_get_name(stream_class
);
389 writer_stream_class
= bt_ctf_stream_class_create_empty(name
);
390 if (!writer_stream_class
) {
391 fprintf(err
, "[error] %s in %s:%d\n",
392 __func__
, __FILE__
, __LINE__
);
396 type
= bt_ctf_stream_class_get_packet_context_type(stream_class
);
398 ret_int
= bt_ctf_stream_class_set_packet_context_type(
399 writer_stream_class
, type
);
401 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
408 type
= bt_ctf_stream_class_get_event_header_type(stream_class
);
410 ret_int
= bt_ctf_trace_get_clock_class_count(writer_trace
);
412 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
416 if (override_ts64
&& ret_int
> 0) {
417 struct bt_ctf_field_type
*new_event_header_type
;
419 new_event_header_type
= override_header_type(err
, type
,
421 if (!new_event_header_type
) {
422 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
426 ret_int
= bt_ctf_stream_class_set_event_header_type(
427 writer_stream_class
, new_event_header_type
);
428 BT_PUT(new_event_header_type
);
430 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
435 ret_int
= bt_ctf_stream_class_set_event_header_type(
436 writer_stream_class
, type
);
438 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
446 type
= bt_ctf_stream_class_get_event_context_type(stream_class
);
448 ret_int
= bt_ctf_stream_class_set_event_context_type(
449 writer_stream_class
, type
);
451 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
461 BT_PUT(writer_stream_class
);
464 return writer_stream_class
;
468 int ctf_stream_copy_packet_header(FILE *err
, struct bt_ctf_packet
*packet
,
469 struct bt_ctf_stream
*writer_stream
)
471 struct bt_ctf_field
*packet_header
= NULL
, *writer_packet_header
= NULL
;
474 packet_header
= bt_ctf_packet_get_header(packet
);
475 if (!packet_header
) {
479 writer_packet_header
= bt_ctf_field_copy(packet_header
);
480 if (!writer_packet_header
) {
481 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
486 ret
= bt_ctf_stream_set_packet_header(writer_stream
,
487 writer_packet_header
);
489 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
499 bt_put(writer_packet_header
);
500 bt_put(packet_header
);
505 int ctf_packet_copy_header(FILE *err
, struct bt_ctf_packet
*packet
,
506 struct bt_ctf_packet
*writer_packet
)
508 struct bt_ctf_field
*packet_header
= NULL
, *writer_packet_header
= NULL
;
511 packet_header
= bt_ctf_packet_get_header(packet
);
512 if (!packet_header
) {
516 writer_packet_header
= bt_ctf_field_copy(packet_header
);
517 if (!writer_packet_header
) {
518 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
523 ret
= bt_ctf_packet_set_header(writer_packet
, writer_packet_header
);
525 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
535 bt_put(packet_header
);
536 bt_put(writer_packet_header
);
541 int ctf_stream_copy_packet_context(FILE *err
, struct bt_ctf_packet
*packet
,
542 struct bt_ctf_stream
*writer_stream
)
544 struct bt_ctf_field
*packet_context
= NULL
, *writer_packet_context
= NULL
;
547 packet_context
= bt_ctf_packet_get_context(packet
);
548 if (!packet_context
) {
552 writer_packet_context
= bt_ctf_field_copy(packet_context
);
553 if (!writer_packet_context
) {
554 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
559 ret
= bt_ctf_stream_set_packet_context(writer_stream
,
560 writer_packet_context
);
562 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
572 bt_put(packet_context
);
573 bt_put(writer_packet_context
);
578 int ctf_packet_copy_context(FILE *err
, struct bt_ctf_packet
*packet
,
579 struct bt_ctf_stream
*writer_stream
,
580 struct bt_ctf_packet
*writer_packet
)
582 struct bt_ctf_field
*packet_context
= NULL
, *writer_packet_context
= NULL
;
585 packet_context
= bt_ctf_packet_get_context(packet
);
586 if (!packet_context
) {
590 writer_packet_context
= bt_ctf_field_copy(packet_context
);
591 if (!writer_packet_context
) {
592 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
597 ret
= bt_ctf_packet_set_context(writer_packet
, writer_packet_context
);
599 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
609 bt_put(writer_packet_context
);
610 bt_put(packet_context
);
615 int ctf_copy_event_header(FILE *err
, struct bt_ctf_event
*event
,
616 struct bt_ctf_event_class
*writer_event_class
,
617 struct bt_ctf_event
*writer_event
,
618 struct bt_ctf_field
*event_header
)
620 struct bt_ctf_clock_class
*clock_class
= NULL
, *writer_clock_class
= NULL
;
621 struct bt_ctf_clock_value
*clock_value
= NULL
, *writer_clock_value
= NULL
;
624 struct bt_ctf_field
*writer_event_header
= NULL
;
627 clock_class
= event_get_clock_class(err
, event
);
629 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
634 clock_value
= bt_ctf_event_get_clock_value(event
, clock_class
);
637 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
642 ret
= bt_ctf_clock_value_get_value(clock_value
, &value
);
645 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
650 writer_clock_class
= event_get_clock_class(err
, writer_event
);
651 if (!writer_clock_class
) {
652 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
657 writer_clock_value
= bt_ctf_clock_value_create(writer_clock_class
, value
);
658 BT_PUT(writer_clock_class
);
659 if (!writer_clock_value
) {
660 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
665 ret
= bt_ctf_event_set_clock_value(writer_event
, writer_clock_value
);
666 BT_PUT(writer_clock_value
);
668 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
673 writer_event_header
= bt_ctf_field_copy(event_header
);
674 if (!writer_event_header
) {
675 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
680 ret
= bt_ctf_event_set_header(writer_event
, writer_event_header
);
681 BT_PUT(writer_event_header
);
683 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
699 struct bt_ctf_trace
*event_class_get_trace(FILE *err
,
700 struct bt_ctf_event_class
*event_class
)
702 struct bt_ctf_trace
*trace
= NULL
;
703 struct bt_ctf_stream_class
*stream_class
= NULL
;
705 stream_class
= bt_ctf_event_class_get_stream_class(event_class
);
707 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
712 trace
= bt_ctf_stream_class_get_trace(stream_class
);
714 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
724 bt_put(stream_class
);
729 struct bt_ctf_event
*ctf_copy_event(FILE *err
, struct bt_ctf_event
*event
,
730 struct bt_ctf_event_class
*writer_event_class
,
733 struct bt_ctf_event
*writer_event
= NULL
;
734 struct bt_ctf_field
*field
= NULL
, *copy_field
= NULL
;
735 struct bt_ctf_trace
*writer_trace
= NULL
;
738 writer_event
= bt_ctf_event_create(writer_event_class
);
740 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
745 writer_trace
= event_class_get_trace(err
, writer_event_class
);
747 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
752 field
= bt_ctf_event_get_header(event
);
755 * If override_ts64, we override all integer fields mapped to a
756 * clock to a uint64_t field type, otherwise, we just copy it as
759 ret
= bt_ctf_trace_get_clock_class_count(writer_trace
);
761 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
765 if (override_ts64
&& ret
> 0) {
766 copy_field
= bt_ctf_event_get_header(writer_event
);
768 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
773 ret
= copy_override_field(err
, event
, writer_event
, field
,
776 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
782 ret
= ctf_copy_event_header(err
, event
, writer_event_class
,
783 writer_event
, field
);
785 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
793 /* Optional field, so it can fail silently. */
794 field
= bt_ctf_event_get_stream_event_context(event
);
796 copy_field
= bt_ctf_field_copy(field
);
798 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
802 ret
= bt_ctf_event_set_stream_event_context(writer_event
,
805 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
813 /* Optional field, so it can fail silently. */
814 field
= bt_ctf_event_get_event_context(event
);
816 copy_field
= bt_ctf_field_copy(field
);
818 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
822 ret
= bt_ctf_event_set_event_context(writer_event
, copy_field
);
824 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
832 field
= bt_ctf_event_get_event_payload(event
);
834 copy_field
= bt_ctf_field_copy(field
);
836 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
840 ret
= bt_ctf_event_set_event_payload(writer_event
, copy_field
);
842 fprintf(err
, "[error] %s in %s:%d\n", __func__
,
853 BT_PUT(writer_event
);
857 bt_put(writer_trace
);
862 enum bt_component_status
ctf_copy_trace(FILE *err
, struct bt_ctf_trace
*trace
,
863 struct bt_ctf_trace
*writer_trace
)
865 enum bt_component_status ret
= BT_COMPONENT_STATUS_OK
;
866 int field_count
, i
, int_ret
;
867 struct bt_ctf_field_type
*header_type
= NULL
;
868 enum bt_ctf_byte_order order
;
869 const char *trace_name
;
871 field_count
= bt_ctf_trace_get_environment_field_count(trace
);
872 for (i
= 0; i
< field_count
; i
++) {
875 struct bt_value
*value
= NULL
;
877 name
= bt_ctf_trace_get_environment_field_name_by_index(
880 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
882 ret
= BT_COMPONENT_STATUS_ERROR
;
885 value
= bt_ctf_trace_get_environment_field_value_by_index(
888 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
890 ret
= BT_COMPONENT_STATUS_ERROR
;
894 ret_int
= bt_ctf_trace_set_environment_field(writer_trace
,
898 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
,
900 fprintf(err
, "[error] Unable to set environment field %s\n",
902 ret
= BT_COMPONENT_STATUS_ERROR
;
907 order
= bt_ctf_trace_get_native_byte_order(trace
);
908 if (order
== BT_CTF_BYTE_ORDER_UNKNOWN
) {
909 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
, __LINE__
);
910 ret
= BT_COMPONENT_STATUS_ERROR
;
915 * Only explicitly set the writer trace's native byte order if
916 * the original trace has a specific one. Otherwise leave what
917 * the CTF writer object chooses, which is the machine's native
920 if (order
!= BT_CTF_BYTE_ORDER_UNSPECIFIED
) {
921 ret
= bt_ctf_trace_set_native_byte_order(writer_trace
, order
);
923 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
, __LINE__
);
924 ret
= BT_COMPONENT_STATUS_ERROR
;
929 header_type
= bt_ctf_trace_get_packet_header_type(trace
);
931 int_ret
= bt_ctf_trace_set_packet_header_type(writer_trace
, header_type
);
934 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
, __LINE__
);
935 ret
= BT_COMPONENT_STATUS_ERROR
;
940 trace_name
= bt_ctf_trace_get_name(trace
);
942 int_ret
= bt_ctf_trace_set_name(writer_trace
, trace_name
);
944 fprintf(err
, "[error] %s in %s:%d\n", __func__
, __FILE__
, __LINE__
);
945 ret
= BT_COMPONENT_STATUS_ERROR
;