copytrace: packet_context is optional
[babeltrace.git] / plugins / libctfcopytrace / ctfcopytrace.c
CommitLineData
91b73004
JD
1/*
2 * copytrace.c
3 *
4 * Babeltrace library to create a copy of a CTF trace
5 *
6 * Copyright 2017 Julien Desfossez <jdesfossez@efficios.com>
7 *
8 * Author: Julien Desfossez <jdesfossez@efficios.com>
9 *
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:
16 *
17 * The above copyright notice and this permission notice shall be included in
18 * all copies or substantial portions of the Software.
19 *
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
26 * SOFTWARE.
27 */
28
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>
37#include <assert.h>
38
39#include "ctfcopytrace.h"
b2f1f465 40#include "clock-fields.h"
91b73004 41
9ac68eb1 42BT_HIDDEN
91b73004
JD
43struct bt_ctf_clock_class *ctf_copy_clock_class(FILE *err,
44 struct bt_ctf_clock_class *clock_class)
45{
46 int64_t offset, offset_s;
47 int int_ret;
48 uint64_t u64_ret;
49 const char *name, *description;
50 struct bt_ctf_clock_class *writer_clock_class = NULL;
51
52 assert(err && clock_class);
53
54 name = bt_ctf_clock_class_get_name(clock_class);
55 if (!name) {
56 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
57 __LINE__);
58 goto end;
59 }
60
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__,
64 __LINE__);
65 goto end;
66 }
67
68 description = bt_ctf_clock_class_get_description(clock_class);
16cd0c80
JD
69 if (description) {
70 int_ret = bt_ctf_clock_class_set_description(writer_clock_class,
71 description);
72 if (int_ret != 0) {
73 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
74 __LINE__);
75 goto end_destroy;
76 }
91b73004
JD
77 }
78
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__,
82 __LINE__);
83 goto end_destroy;
84 }
85 int_ret = bt_ctf_clock_class_set_frequency(writer_clock_class, u64_ret);
86 if (int_ret != 0) {
87 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
88 __LINE__);
89 goto end_destroy;
90 }
91
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__,
95 __LINE__);
96 goto end_destroy;
97 }
98 int_ret = bt_ctf_clock_class_set_precision(writer_clock_class,
99 u64_ret);
100 if (int_ret != 0) {
101 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
102 __LINE__);
103 goto end_destroy;
104 }
105
106 int_ret = bt_ctf_clock_class_get_offset_s(clock_class, &offset_s);
107 if (int_ret != 0) {
108 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
109 __LINE__);
110 goto end_destroy;
111 }
112
113 int_ret = bt_ctf_clock_class_set_offset_s(writer_clock_class, offset_s);
114 if (int_ret != 0) {
115 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
116 __LINE__);
117 goto end_destroy;
118 }
119
120 int_ret = bt_ctf_clock_class_get_offset_cycles(clock_class, &offset);
121 if (int_ret != 0) {
122 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
123 __LINE__);
124 goto end_destroy;
125 }
126
127 int_ret = bt_ctf_clock_class_set_offset_cycles(writer_clock_class, offset);
128 if (int_ret != 0) {
129 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
130 __LINE__);
131 goto end_destroy;
132 }
133
acd6aeb1 134 int_ret = bt_ctf_clock_class_is_absolute(clock_class);
91b73004
JD
135 if (int_ret == -1) {
136 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
137 __LINE__);
138 goto end_destroy;
139 }
140
141 int_ret = bt_ctf_clock_class_set_is_absolute(writer_clock_class, int_ret);
142 if (int_ret != 0) {
143 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
144 __LINE__);
145 goto end_destroy;
146 }
147
148 goto end;
149
150end_destroy:
151 BT_PUT(writer_clock_class);
152end:
153 return writer_clock_class;
154}
155
9ac68eb1 156BT_HIDDEN
91b73004
JD
157enum 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)
161{
162 enum bt_component_status ret;
91b73004
JD
163 int int_ret, clock_class_count, i;
164
165 clock_class_count = bt_ctf_trace_get_clock_class_count(trace);
166
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 =
9ac68eb1 170 bt_ctf_trace_get_clock_class_by_index(trace, i);
91b73004
JD
171
172 if (!clock_class) {
173 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
174 __LINE__);
175 ret = BT_COMPONENT_STATUS_ERROR;
176 goto end;
177 }
178
179 writer_clock_class = ctf_copy_clock_class(err, clock_class);
180 bt_put(clock_class);
181 if (!writer_clock_class) {
182 fprintf(err, "Failed to copy clock class");
183 ret = BT_COMPONENT_STATUS_ERROR;
184 goto end;
185 }
186
187 int_ret = bt_ctf_trace_add_clock_class(writer_trace, writer_clock_class);
188 if (int_ret != 0) {
189 BT_PUT(writer_clock_class);
190 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
191 __LINE__);
192 ret = BT_COMPONENT_STATUS_ERROR;
193 goto end;
194 }
195
196 /*
b2f1f465 197 * Ownership transferred to the trace.
91b73004
JD
198 */
199 bt_put(writer_clock_class);
200 }
201
202 ret = BT_COMPONENT_STATUS_OK;
203
204end:
205 return ret;
206}
207
9ac68eb1 208BT_HIDDEN
91b73004
JD
209struct bt_ctf_event_class *ctf_copy_event_class(FILE *err,
210 struct bt_ctf_event_class *event_class)
211{
212 struct bt_ctf_event_class *writer_event_class = NULL;
f87fb9b4 213 struct bt_ctf_field_type *context, *payload_type;
91b73004 214 const char *name;
cf76ce92
PP
215 int ret;
216 int64_t id;
217 enum bt_ctf_event_class_log_level log_level;
218 const char *emf_uri;
91b73004
JD
219
220 name = bt_ctf_event_class_get_name(event_class);
221 if (!name) {
222 fprintf(err, "[error] %s in %s:%d\n", __func__,
223 __FILE__, __LINE__);
224 goto end;
225 }
226
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__,
230 __FILE__, __LINE__);
231 goto end;
232 }
233
cf76ce92
PP
234 id = bt_ctf_event_class_get_id(event_class);
235 if (id < 0) {
236 fprintf(err, "[error] %s in %s:%d\n", __func__,
237 __FILE__, __LINE__);
238 goto error;
239 }
91b73004 240
cf76ce92
PP
241 ret = bt_ctf_event_class_set_id(writer_event_class, id);
242 if (ret) {
243 fprintf(err, "[error] %s in %s:%d\n", __func__,
244 __FILE__, __LINE__);
245 goto error;
246 }
91b73004 247
cf76ce92
PP
248 log_level = bt_ctf_event_class_get_log_level(event_class);
249 if (log_level < 0) {
250 fprintf(err, "[error] %s in %s:%d\n", __func__,
251 __FILE__, __LINE__);
252 goto error;
253 }
254
255 ret = bt_ctf_event_class_set_log_level(writer_event_class, log_level);
256 if (ret) {
257 fprintf(err, "[error] %s in %s:%d\n", __func__,
258 __FILE__, __LINE__);
259 goto error;
260 }
261
262 emf_uri = bt_ctf_event_class_get_emf_uri(event_class);
263 if (emf_uri) {
264 ret = bt_ctf_event_class_set_emf_uri(writer_event_class,
265 emf_uri);
266 if (ret) {
91b73004
JD
267 fprintf(err, "[error] %s in %s:%d\n", __func__,
268 __FILE__, __LINE__);
9ae49d3d 269 goto error;
91b73004
JD
270 }
271 }
272
f87fb9b4
JD
273 payload_type = bt_ctf_event_class_get_payload_type(event_class);
274 if (payload_type) {
275 ret = bt_ctf_event_class_set_payload_type(writer_event_class,
276 payload_type);
91b73004 277 if (ret < 0) {
f87fb9b4
JD
278 fprintf(err, "[error] %s in %s:%d\n", __func__,
279 __FILE__, __LINE__);
9ae49d3d 280 goto error;
91b73004 281 }
f87fb9b4 282 BT_PUT(payload_type);
91b73004
JD
283 }
284
279c77d0
JD
285 context = bt_ctf_event_class_get_context_type(event_class);
286 if (context) {
287 ret = bt_ctf_event_class_set_context_type(
288 writer_event_class, context);
289 BT_PUT(context);
290 if (ret < 0) {
291 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
292 __LINE__);
293 goto error;
294 }
295 }
296
9ae49d3d
JD
297 goto end;
298
299error:
300 BT_PUT(writer_event_class);
91b73004
JD
301end:
302 return writer_event_class;
303}
304
9ac68eb1 305BT_HIDDEN
91b73004
JD
306enum 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)
309{
310 enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
9ae49d3d 311 struct bt_ctf_event_class *event_class = NULL, *writer_event_class = NULL;
91b73004
JD
312 int count, i;
313
314 count = bt_ctf_stream_class_get_event_class_count(stream_class);
315 if (count < 0) {
316 fprintf(err, "[error] %s in %s:%d\n", __func__,
317 __FILE__, __LINE__);
318 goto end;
319 }
320
321 for (i = 0; i < count; i++) {
91b73004
JD
322 int int_ret;
323
9ac68eb1 324 event_class = bt_ctf_stream_class_get_event_class_by_index(
91b73004
JD
325 stream_class, i);
326 if (!event_class) {
327 fprintf(err, "[error] %s in %s:%d\n", __func__,
328 __FILE__, __LINE__);
329 ret = BT_COMPONENT_STATUS_ERROR;
9ae49d3d 330 goto error;
91b73004 331 }
cb0a5cf8
JD
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) {
336 /*
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
340 * static.
341 */
342 BT_PUT(writer_event_class);
343 BT_PUT(event_class);
344 continue;
345 }
346 }
347
91b73004
JD
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__,
351 __FILE__, __LINE__);
352 ret = BT_COMPONENT_STATUS_ERROR;
9ae49d3d 353 goto error;
91b73004
JD
354 }
355
91b73004
JD
356 int_ret = bt_ctf_stream_class_add_event_class(writer_stream_class,
357 writer_event_class);
358 if (int_ret < 0) {
359 fprintf(err, "[error] Failed to add event class\n");
360 fprintf(err, "[error] %s in %s:%d\n", __func__,
361 __FILE__, __LINE__);
362 ret = BT_COMPONENT_STATUS_ERROR;
9ae49d3d 363 goto error;
91b73004 364 }
9ae49d3d
JD
365 BT_PUT(writer_event_class);
366 BT_PUT(event_class);
91b73004
JD
367 }
368
9ae49d3d
JD
369 goto end;
370
371error:
372 bt_put(event_class);
373 bt_put(writer_event_class);
91b73004
JD
374end:
375 return ret;
376}
377
9ac68eb1 378BT_HIDDEN
91b73004 379struct bt_ctf_stream_class *ctf_copy_stream_class(FILE *err,
b2f1f465
JD
380 struct bt_ctf_stream_class *stream_class,
381 struct bt_ctf_trace *writer_trace,
382 bool override_ts64)
91b73004 383{
9ae49d3d
JD
384 struct bt_ctf_field_type *type = NULL;
385 struct bt_ctf_stream_class *writer_stream_class = NULL;
91b73004
JD
386 int ret_int;
387 const char *name = bt_ctf_stream_class_get_name(stream_class);
388
389 if (strlen(name) == 0) {
390 name = NULL;
391 }
392
e49da5dd 393 writer_stream_class = bt_ctf_stream_class_create_empty(name);
91b73004
JD
394 if (!writer_stream_class) {
395 fprintf(err, "[error] %s in %s:%d\n",
396 __func__, __FILE__, __LINE__);
397 goto end;
398 }
399
400 type = bt_ctf_stream_class_get_packet_context_type(stream_class);
cfb74f3d
JD
401 if (type) {
402 ret_int = bt_ctf_stream_class_set_packet_context_type(
403 writer_stream_class, type);
404 if (ret_int < 0) {
405 fprintf(err, "[error] %s in %s:%d\n", __func__,
406 __FILE__, __LINE__);
407 goto error;
408 }
409 BT_PUT(type);
91b73004
JD
410 }
411
412 type = bt_ctf_stream_class_get_event_header_type(stream_class);
413 if (!type) {
414 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
415 __LINE__);
416 goto error;
417 }
418
b2f1f465 419 if (override_ts64) {
9ae49d3d
JD
420 struct bt_ctf_field_type *new_event_header_type;
421
b2f1f465
JD
422 new_event_header_type = override_header_type(err, type,
423 writer_trace);
b2f1f465
JD
424 if (!new_event_header_type) {
425 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
426 __LINE__);
427 goto error;
428 }
0f29db56
JD
429 ret_int = bt_ctf_stream_class_set_event_header_type(
430 writer_stream_class, new_event_header_type);
9ae49d3d 431 BT_PUT(new_event_header_type);
0f29db56
JD
432 if (ret_int < 0) {
433 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
434 __LINE__);
435 goto error;
436 }
b2f1f465 437 } else {
0f29db56
JD
438 ret_int = bt_ctf_stream_class_set_event_header_type(
439 writer_stream_class, type);
0f29db56
JD
440 if (ret_int < 0) {
441 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
442 __LINE__);
443 goto error;
444 }
91b73004 445 }
9ae49d3d 446 BT_PUT(type);
91b73004
JD
447
448 type = bt_ctf_stream_class_get_event_context_type(stream_class);
449 if (type) {
450 ret_int = bt_ctf_stream_class_set_event_context_type(
451 writer_stream_class, type);
91b73004
JD
452 if (ret_int < 0) {
453 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
454 __LINE__);
455 goto error;
456 }
457 }
9ae49d3d 458 BT_PUT(type);
91b73004
JD
459
460 goto end;
461
462error:
463 BT_PUT(writer_stream_class);
464end:
9ae49d3d 465 bt_put(type);
91b73004
JD
466 return writer_stream_class;
467}
468
9ac68eb1 469BT_HIDDEN
91b73004
JD
470enum bt_component_status ctf_copy_packet_context_field(FILE *err,
471 struct bt_ctf_field *field, const char *field_name,
472 struct bt_ctf_field *writer_packet_context,
473 struct bt_ctf_field_type *writer_packet_context_type)
474{
475 enum bt_component_status ret;
9ae49d3d
JD
476 struct bt_ctf_field *writer_field = NULL;
477 struct bt_ctf_field_type *field_type = NULL;
91b73004
JD
478 int int_ret;
479 uint64_t value;
480
481 field_type = bt_ctf_field_get_type(field);
482 if (!field_type) {
483 ret = BT_COMPONENT_STATUS_ERROR;
484 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
485 __LINE__);
486 goto end;
487 }
488 /*
489 * Only support for integers for now.
490 */
1487a16a 491 if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) {
91b73004 492 fprintf(err, "[error] Unsupported packet context field type\n");
91b73004 493 ret = BT_COMPONENT_STATUS_ERROR;
9ae49d3d 494 goto error;
91b73004 495 }
9ae49d3d 496 BT_PUT(field_type);
91b73004 497
9ac68eb1
PP
498 writer_field = bt_ctf_field_structure_get_field_by_name(
499 writer_packet_context, field_name);
91b73004
JD
500 if (!writer_field) {
501 ret = BT_COMPONENT_STATUS_ERROR;
502 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
503 __LINE__);
504 goto end;
505 }
506
91b73004
JD
507 int_ret = bt_ctf_field_unsigned_integer_get_value(field, &value);
508 if (int_ret < 0) {
509 fprintf(err, "[error] Wrong packet_context field type\n");
510 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
511 __LINE__);
512 ret = BT_COMPONENT_STATUS_ERROR;
9ae49d3d 513 goto end;
91b73004
JD
514 }
515
516 int_ret = bt_ctf_field_unsigned_integer_set_value(writer_field, value);
517 if (int_ret < 0) {
518 ret = BT_COMPONENT_STATUS_ERROR;
519 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
520 __LINE__);
9ae49d3d 521 goto end;
91b73004
JD
522 }
523
524 ret = BT_COMPONENT_STATUS_OK;
525
9ae49d3d
JD
526 goto end;
527
528error:
529 bt_put(field_type);
91b73004 530end:
9ae49d3d 531 bt_put(writer_field);
91b73004
JD
532 return ret;
533}
534
9ac68eb1 535BT_HIDDEN
b2f1f465 536struct bt_ctf_field *ctf_copy_packet_context(FILE *err,
91b73004 537 struct bt_ctf_packet *packet,
79697a58 538 struct bt_ctf_stream *writer_stream)
91b73004
JD
539{
540 enum bt_component_status ret;
9ae49d3d
JD
541 struct bt_ctf_field *packet_context = NULL, *writer_packet_context = NULL;
542 struct bt_ctf_field_type *struct_type = NULL, *writer_packet_context_type = NULL;
543 struct bt_ctf_stream_class *writer_stream_class = NULL;
f6f999a3 544 struct bt_ctf_field *field = NULL;
9571c709 545 struct bt_ctf_field_type *field_type = NULL;
b2f1f465 546 int nr_fields, i;
91b73004
JD
547
548 packet_context = bt_ctf_packet_get_context(packet);
549 if (!packet_context) {
60ef553b 550 goto end;
91b73004
JD
551 }
552
553 writer_stream_class = bt_ctf_stream_get_class(writer_stream);
554 if (!writer_stream_class) {
91b73004
JD
555 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
556 __LINE__);
9ae49d3d 557 goto error;
91b73004
JD
558 }
559
560 writer_packet_context_type = bt_ctf_stream_class_get_packet_context_type(
561 writer_stream_class);
9ae49d3d 562 BT_PUT(writer_stream_class);
91b73004 563 if (!writer_packet_context_type) {
91b73004
JD
564 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
565 __LINE__);
9ae49d3d 566 goto error;
91b73004
JD
567 }
568
569 struct_type = bt_ctf_field_get_type(packet_context);
570 if (!struct_type) {
91b73004
JD
571 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
572 __LINE__);
9ae49d3d 573 goto error;
91b73004
JD
574 }
575
576 writer_packet_context = bt_ctf_field_create(writer_packet_context_type);
577 if (!writer_packet_context) {
91b73004
JD
578 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
579 __LINE__);
9ae49d3d 580 goto error;
91b73004
JD
581 }
582
583 nr_fields = bt_ctf_field_type_structure_get_field_count(struct_type);
584 for (i = 0; i < nr_fields; i++) {
91b73004
JD
585 const char *field_name;
586
0f29db56 587 field = bt_ctf_field_structure_get_field_by_index(
91b73004
JD
588 packet_context, i);
589 if (!field) {
91b73004
JD
590 fprintf(err, "[error] %s in %s:%d\n", __func__,
591 __FILE__, __LINE__);
9ae49d3d 592 goto error;
91b73004 593 }
9ac68eb1 594 if (bt_ctf_field_type_structure_get_field_by_index(struct_type,
91b73004 595 &field_name, &field_type, i) < 0) {
91b73004
JD
596 fprintf(err, "[error] %s in %s:%d\n", __func__,
597 __FILE__, __LINE__);
9ae49d3d 598 goto error;
91b73004 599 }
91b73004 600
1487a16a 601 if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) {
91b73004 602 fprintf(err, "[error] Unexpected packet context field type\n");
9ae49d3d 603 goto error;
91b73004
JD
604 }
605
606 ret = ctf_copy_packet_context_field(err, field, field_name,
607 writer_packet_context, writer_packet_context_type);
9ae49d3d
JD
608 BT_PUT(field_type);
609 BT_PUT(field);
91b73004
JD
610 if (ret != BT_COMPONENT_STATUS_OK) {
611 fprintf(err, "[error] %s in %s:%d\n", __func__,
612 __FILE__, __LINE__);
9ae49d3d 613 goto error;
91b73004
JD
614 }
615 }
616
9ae49d3d
JD
617 goto end;
618
619error:
620 BT_PUT(writer_packet_context);
621end:
622 bt_put(field);
623 bt_put(field_type);
91b73004 624 bt_put(struct_type);
91b73004 625 bt_put(writer_packet_context_type);
91b73004 626 bt_put(writer_stream_class);
91b73004 627 bt_put(packet_context);
b2f1f465
JD
628 return writer_packet_context;
629}
630
9ac68eb1 631BT_HIDDEN
0f29db56 632int ctf_copy_event_header(FILE *err, struct bt_ctf_event *event,
b2f1f465
JD
633 struct bt_ctf_event_class *writer_event_class,
634 struct bt_ctf_event *writer_event,
635 struct bt_ctf_field *event_header)
636{
9ae49d3d
JD
637 struct bt_ctf_clock_class *clock_class = NULL, *writer_clock_class = NULL;
638 struct bt_ctf_clock_value *clock_value = NULL, *writer_clock_value = NULL;
b2f1f465
JD
639
640 int ret;
0f29db56
JD
641 struct bt_ctf_field *writer_event_header = NULL;
642 uint64_t value;
b2f1f465 643
0f29db56
JD
644 clock_class = event_get_clock_class(err, event);
645 if (!clock_class) {
646 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
647 __LINE__);
648 goto error;
649 }
650
651 clock_value = bt_ctf_event_get_clock_value(event, clock_class);
9ae49d3d 652 BT_PUT(clock_class);
0f29db56
JD
653 if (!clock_value) {
654 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
655 __LINE__);
656 goto error;
657 }
658
659 ret = bt_ctf_clock_value_get_value(clock_value, &value);
9ae49d3d 660 BT_PUT(clock_value);
0f29db56
JD
661 if (ret) {
662 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
663 __LINE__);
664 goto error;
b2f1f465
JD
665 }
666
667 writer_clock_class = event_get_clock_class(err, writer_event);
668 if (!writer_clock_class) {
669 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
670 __LINE__);
671 goto error;
672 }
673
0f29db56 674 writer_clock_value = bt_ctf_clock_value_create(writer_clock_class, value);
9ae49d3d 675 BT_PUT(writer_clock_class);
0f29db56 676 if (!writer_clock_value) {
b2f1f465
JD
677 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
678 __LINE__);
9ae49d3d 679 goto error;
b2f1f465
JD
680 }
681
0f29db56 682 ret = bt_ctf_event_set_clock_value(writer_event, writer_clock_value);
9ae49d3d 683 BT_PUT(writer_clock_value);
b2f1f465
JD
684 if (ret) {
685 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
686 __LINE__);
687 goto error;
688 }
689
0f29db56
JD
690 writer_event_header = bt_ctf_field_copy(event_header);
691 if (!writer_event_header) {
692 fprintf(err, "[error] %s in %s:%d\n", __func__,
693 __FILE__, __LINE__);
0f29db56
JD
694 goto end;
695 }
696
b2f1f465 697 ret = bt_ctf_event_set_header(writer_event, writer_event_header);
9ae49d3d 698 BT_PUT(writer_event_header);
b2f1f465
JD
699 if (ret < 0) {
700 fprintf(err, "[error] %s in %s:%d\n", __func__,
701 __FILE__, __LINE__);
702 goto error;
703 }
b2f1f465
JD
704
705 ret = 0;
706
707 goto end;
708
709error:
b2f1f465 710 ret = -1;
91b73004
JD
711end:
712 return ret;
713}
714
9ac68eb1 715BT_HIDDEN
91b73004 716struct bt_ctf_event *ctf_copy_event(FILE *err, struct bt_ctf_event *event,
b2f1f465
JD
717 struct bt_ctf_event_class *writer_event_class,
718 bool override_ts64)
91b73004 719{
9ae49d3d
JD
720 struct bt_ctf_event *writer_event = NULL;
721 struct bt_ctf_field *field = NULL, *copy_field = NULL;
91b73004
JD
722 int ret;
723
724 writer_event = bt_ctf_event_create(writer_event_class);
725 if (!writer_event) {
726 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
727 __LINE__);
728 goto end;
729 }
730
731 field = bt_ctf_event_get_header(event);
732 if (!field) {
91b73004
JD
733 fprintf(err, "[error] %s in %s:%d\n", __func__,
734 __FILE__, __LINE__);
9ae49d3d 735 goto error;
91b73004
JD
736 }
737
b2f1f465
JD
738 /*
739 * If override_ts64, we override all integer fields mapped to a clock
740 * to a uint64_t field type, otherwise, we just copy it as is.
741 */
742 if (override_ts64) {
743 copy_field = bt_ctf_event_get_header(writer_event);
744 if (!copy_field) {
91b73004
JD
745 fprintf(err, "[error] %s in %s:%d\n", __func__,
746 __FILE__, __LINE__);
9ae49d3d 747 goto error;
b2f1f465
JD
748 }
749
750 ret = copy_override_field(err, event, writer_event, field,
751 copy_field);
b2f1f465 752 if (ret) {
b2f1f465
JD
753 fprintf(err, "[error] %s in %s:%d\n", __func__,
754 __FILE__, __LINE__);
9ae49d3d 755 goto error;
b2f1f465 756 }
9ae49d3d 757 BT_PUT(copy_field);
b2f1f465 758 } else {
0f29db56 759 ret = ctf_copy_event_header(err, event, writer_event_class,
b2f1f465
JD
760 writer_event, field);
761 if (ret) {
b2f1f465
JD
762 fprintf(err, "[error] %s in %s:%d\n", __func__,
763 __FILE__, __LINE__);
9ae49d3d 764 goto error;
91b73004 765 }
91b73004 766 }
9ae49d3d 767 BT_PUT(field);
91b73004
JD
768
769 /* Optional field, so it can fail silently. */
770 field = bt_ctf_event_get_stream_event_context(event);
961ec227
JD
771 if (field) {
772 copy_field = bt_ctf_field_copy(field);
773 if (!copy_field) {
774 fprintf(err, "[error] %s in %s:%d\n", __func__,
775 __FILE__, __LINE__);
776 goto error;
777 }
91b73004
JD
778 ret = bt_ctf_event_set_stream_event_context(writer_event,
779 copy_field);
91b73004
JD
780 if (ret < 0) {
781 fprintf(err, "[error] %s in %s:%d\n", __func__,
782 __FILE__, __LINE__);
783 goto error;
784 }
961ec227
JD
785 BT_PUT(field);
786 BT_PUT(copy_field);
91b73004
JD
787 }
788
789 /* Optional field, so it can fail silently. */
790 field = bt_ctf_event_get_event_context(event);
961ec227
JD
791 if (field) {
792 copy_field = bt_ctf_field_copy(field);
793 if (!copy_field) {
794 fprintf(err, "[error] %s in %s:%d\n", __func__,
795 __FILE__, __LINE__);
796 goto error;
797 }
91b73004 798 ret = bt_ctf_event_set_event_context(writer_event, copy_field);
91b73004
JD
799 if (ret < 0) {
800 fprintf(err, "[error] %s in %s:%d\n", __func__,
801 __FILE__, __LINE__);
802 goto error;
803 }
961ec227
JD
804 BT_PUT(field);
805 BT_PUT(copy_field);
91b73004
JD
806 }
807
9ac68eb1 808 field = bt_ctf_event_get_event_payload(event);
961ec227
JD
809 if (field) {
810 copy_field = bt_ctf_field_copy(field);
811 if (!copy_field) {
812 fprintf(err, "[error] %s in %s:%d\n", __func__,
813 __FILE__, __LINE__);
814 goto error;
815 }
9ac68eb1 816 ret = bt_ctf_event_set_event_payload(writer_event, copy_field);
91b73004
JD
817 if (ret < 0) {
818 fprintf(err, "[error] %s in %s:%d\n", __func__,
819 __FILE__, __LINE__);
820 goto error;
821 }
961ec227
JD
822 BT_PUT(field);
823 BT_PUT(copy_field);
91b73004 824 }
9ae49d3d 825
91b73004
JD
826 goto end;
827
828error:
829 BT_PUT(writer_event);
830end:
9ae49d3d
JD
831 bt_put(field);
832 bt_put(copy_field);
91b73004
JD
833 return writer_event;
834}
835
9ac68eb1 836BT_HIDDEN
91b73004
JD
837enum bt_component_status ctf_copy_trace(FILE *err, struct bt_ctf_trace *trace,
838 struct bt_ctf_trace *writer_trace)
839{
840 enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
841 int field_count, i, int_ret;
9ae49d3d 842 struct bt_ctf_field_type *header_type = NULL;
6468dbd6 843 enum bt_ctf_byte_order order;
0ac862b4 844 const char *trace_name;
91b73004
JD
845
846 field_count = bt_ctf_trace_get_environment_field_count(trace);
847 for (i = 0; i < field_count; i++) {
848 int ret_int;
849 const char *name;
9ae49d3d 850 struct bt_value *value = NULL;
91b73004 851
9ac68eb1
PP
852 name = bt_ctf_trace_get_environment_field_name_by_index(
853 trace, i);
91b73004
JD
854 if (!name) {
855 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
856 __LINE__);
857 ret = BT_COMPONENT_STATUS_ERROR;
9ae49d3d 858 goto end;
91b73004 859 }
9ac68eb1
PP
860 value = bt_ctf_trace_get_environment_field_value_by_index(
861 trace, i);
91b73004
JD
862 if (!value) {
863 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
864 __LINE__);
865 ret = BT_COMPONENT_STATUS_ERROR;
9ae49d3d 866 goto end;
91b73004
JD
867 }
868
869 ret_int = bt_ctf_trace_set_environment_field(writer_trace,
870 name, value);
9ae49d3d 871 BT_PUT(value);
91b73004
JD
872 if (ret_int < 0) {
873 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
874 __LINE__);
875 fprintf(err, "[error] Unable to set environment field %s\n",
876 name);
877 ret = BT_COMPONENT_STATUS_ERROR;
9ae49d3d 878 goto end;
91b73004
JD
879 }
880 }
881
6468dbd6
JD
882 order = bt_ctf_trace_get_native_byte_order(trace);
883 if (order == BT_CTF_BYTE_ORDER_UNKNOWN) {
884 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, __LINE__);
885 ret = BT_COMPONENT_STATUS_ERROR;
886 goto end;
887 }
888
d41cff38
PP
889 /*
890 * Only explicitly set the writer trace's native byte order if
891 * the original trace has a specific one. Otherwise leave what
892 * the CTF writer object chooses, which is the machine's native
893 * byte order.
894 */
2f4b93cd 895 if (order != BT_CTF_BYTE_ORDER_UNSPECIFIED) {
d41cff38
PP
896 ret = bt_ctf_trace_set_native_byte_order(writer_trace, order);
897 if (ret) {
898 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, __LINE__);
899 ret = BT_COMPONENT_STATUS_ERROR;
900 goto end;
901 }
6468dbd6
JD
902 }
903
91b73004 904 header_type = bt_ctf_trace_get_packet_header_type(writer_trace);
944eed39
JD
905 if (header_type) {
906 int_ret = bt_ctf_trace_set_packet_header_type(writer_trace, header_type);
907 BT_PUT(header_type);
908 if (int_ret < 0) {
909 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, __LINE__);
910 ret = BT_COMPONENT_STATUS_ERROR;
911 goto end;
912 }
91b73004
JD
913 }
914
0ac862b4
JD
915 trace_name = bt_ctf_trace_get_name(trace);
916 if (trace_name) {
917 int_ret = bt_ctf_trace_set_name(writer_trace, trace_name);
918 if (int_ret < 0) {
919 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__, __LINE__);
920 ret = BT_COMPONENT_STATUS_ERROR;
921 goto end;
922 }
923 }
924
9ae49d3d 925end:
91b73004
JD
926 return ret;
927}
This page took 0.068988 seconds and 4 git commands to generate.