plugins/libctfcopytrace/ctfcopytrace.c

   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"
  40 #include "clock-fields.h"
  41
  42 BT_HIDDEN
  43 struct 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);
  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                 }
  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
 134         int_ret = bt_ctf_clock_class_is_absolute(clock_class);
 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
 150 end_destroy:
 151         BT_PUT(writer_clock_class);
 152 end:
 153         return writer_clock_class;
 154 }
 155
 156 BT_HIDDEN
 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)
 161 {
 162         enum bt_component_status ret;
 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 =
 170                         bt_ctf_trace_get_clock_class_by_index(trace, i);
 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                 /*
 197                  * Ownership transferred to the trace.
 198                  */
 199                 bt_put(writer_clock_class);
 200         }
 201
 202         ret = BT_COMPONENT_STATUS_OK;
 203
 204 end:
 205         return ret;
 206 }
 207
 208 BT_HIDDEN
 209 struct 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;
 213         struct bt_ctf_field_type *context, *payload_type;
 214         const char *name;
 215         int ret;
 216         int64_t id;
 217         enum bt_ctf_event_class_log_level log_level;
 218         const char *emf_uri;
 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
 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         }
 240
 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         }
 247
 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) {
 267                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 268                                         __FILE__, __LINE__);
 269                         goto error;
 270                 }
 271         }
 272
 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);
 277                 if (ret < 0) {
 278                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 279                                         __FILE__, __LINE__);
 280                         goto error;
 281                 }
 282                 BT_PUT(payload_type);
 283         }
 284
 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
 297         goto end;
 298
 299 error:
 300         BT_PUT(writer_event_class);
 301 end:
 302         return writer_event_class;
 303 }
 304
 305 BT_HIDDEN
 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)
 309 {
 310         enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
 311         struct bt_ctf_event_class *event_class = NULL, *writer_event_class = NULL;
 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++) {
 322                 int int_ret;
 323
 324                 event_class = bt_ctf_stream_class_get_event_class_by_index(
 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;
 330                         goto error;
 331                 }
 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
 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;
 353                         goto error;
 354                 }
 355
 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;
 363                         goto error;
 364                 }
 365                 BT_PUT(writer_event_class);
 366                 BT_PUT(event_class);
 367         }
 368
 369         goto end;
 370
 371 error:
 372         bt_put(event_class);
 373         bt_put(writer_event_class);
 374 end:
 375         return ret;
 376 }
 377
 378 BT_HIDDEN
 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,
 382                 bool override_ts64)
 383 {
 384         struct bt_ctf_field_type *type = NULL;
 385         struct bt_ctf_stream_class *writer_stream_class = NULL;
 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
 393         writer_stream_class = bt_ctf_stream_class_create_empty(name);
 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);
 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);
 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
 419         if (override_ts64) {
 420                 struct bt_ctf_field_type *new_event_header_type;
 421
 422                 new_event_header_type = override_header_type(err, type,
 423                                 writer_trace);
 424                 if (!new_event_header_type) {
 425                         fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 426                                         __LINE__);
 427                         goto error;
 428                 }
 429                 ret_int = bt_ctf_stream_class_set_event_header_type(
 430                                 writer_stream_class, new_event_header_type);
 431                 BT_PUT(new_event_header_type);
 432                 if (ret_int < 0) {
 433                         fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 434                                         __LINE__);
 435                         goto error;
 436                 }
 437         } else {
 438                 ret_int = bt_ctf_stream_class_set_event_header_type(
 439                                 writer_stream_class, type);
 440                 if (ret_int < 0) {
 441                         fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 442                                         __LINE__);
 443                         goto error;
 444                 }
 445         }
 446         BT_PUT(type);
 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);
 452                 if (ret_int < 0) {
 453                         fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 454                                         __LINE__);
 455                         goto error;
 456                 }
 457         }
 458         BT_PUT(type);
 459
 460         goto end;
 461
 462 error:
 463         BT_PUT(writer_stream_class);
 464 end:
 465         bt_put(type);
 466         return writer_stream_class;
 467 }
 468
 469 BT_HIDDEN
 470 enum 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;
 476         struct bt_ctf_field *writer_field = NULL;
 477         struct bt_ctf_field_type *field_type = NULL;
 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          */
 491         if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) {
 492                 fprintf(err, "[error] Unsupported packet context field type\n");
 493                 ret = BT_COMPONENT_STATUS_ERROR;
 494                 goto error;
 495         }
 496         BT_PUT(field_type);
 497
 498         writer_field = bt_ctf_field_structure_get_field_by_name(
 499                 writer_packet_context, field_name);
 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
 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;
 513                 goto end;
 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__);
 521                 goto end;
 522         }
 523
 524         ret = BT_COMPONENT_STATUS_OK;
 525
 526         goto end;
 527
 528 error:
 529         bt_put(field_type);
 530 end:
 531         bt_put(writer_field);
 532         return ret;
 533 }
 534
 535 BT_HIDDEN
 536 struct bt_ctf_field *ctf_copy_packet_context(FILE *err,
 537                 struct bt_ctf_packet *packet,
 538                 struct bt_ctf_stream *writer_stream)
 539 {
 540         enum bt_component_status ret;
 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;
 544         struct bt_ctf_field *field = NULL;
 545         struct bt_ctf_field_type *field_type = NULL;
 546         int nr_fields, i;
 547
 548         packet_context = bt_ctf_packet_get_context(packet);
 549         if (!packet_context) {
 550                 goto end;
 551         }
 552
 553         writer_stream_class = bt_ctf_stream_get_class(writer_stream);
 554         if (!writer_stream_class) {
 555                 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 556                                 __LINE__);
 557                 goto error;
 558         }
 559
 560         writer_packet_context_type = bt_ctf_stream_class_get_packet_context_type(
 561                         writer_stream_class);
 562         BT_PUT(writer_stream_class);
 563         if (!writer_packet_context_type) {
 564                 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 565                                 __LINE__);
 566                 goto error;
 567         }
 568
 569         struct_type = bt_ctf_field_get_type(packet_context);
 570         if (!struct_type) {
 571                 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 572                                 __LINE__);
 573                 goto error;
 574         }
 575
 576         writer_packet_context = bt_ctf_field_create(writer_packet_context_type);
 577         if (!writer_packet_context) {
 578                 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 579                                 __LINE__);
 580                 goto error;
 581         }
 582
 583         nr_fields = bt_ctf_field_type_structure_get_field_count(struct_type);
 584         for (i = 0; i < nr_fields; i++) {
 585                 const char *field_name;
 586
 587                 field = bt_ctf_field_structure_get_field_by_index(
 588                                 packet_context, i);
 589                 if (!field) {
 590                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 591                                         __FILE__, __LINE__);
 592                         goto error;
 593                 }
 594                 if (bt_ctf_field_type_structure_get_field_by_index(struct_type,
 595                                         &field_name, &field_type, i) < 0) {
 596                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 597                                         __FILE__, __LINE__);
 598                         goto error;
 599                 }
 600
 601                 if (bt_ctf_field_type_get_type_id(field_type) != BT_CTF_FIELD_TYPE_ID_INTEGER) {
 602                         fprintf(err, "[error] Unexpected packet context field type\n");
 603                         goto error;
 604                 }
 605
 606                 ret = ctf_copy_packet_context_field(err, field, field_name,
 607                                 writer_packet_context, writer_packet_context_type);
 608                 BT_PUT(field_type);
 609                 BT_PUT(field);
 610                 if (ret != BT_COMPONENT_STATUS_OK) {
 611                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 612                                         __FILE__, __LINE__);
 613                         goto error;
 614                 }
 615         }
 616
 617         goto end;
 618
 619 error:
 620         BT_PUT(writer_packet_context);
 621 end:
 622         bt_put(field);
 623         bt_put(field_type);
 624         bt_put(struct_type);
 625         bt_put(writer_packet_context_type);
 626         bt_put(writer_stream_class);
 627         bt_put(packet_context);
 628         return writer_packet_context;
 629 }
 630
 631 BT_HIDDEN
 632 int ctf_copy_event_header(FILE *err, struct bt_ctf_event *event,
 633                 struct bt_ctf_event_class *writer_event_class,
 634                 struct bt_ctf_event *writer_event,
 635                 struct bt_ctf_field *event_header)
 636 {
 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;
 639
 640         int ret;
 641         struct bt_ctf_field *writer_event_header = NULL;
 642         uint64_t value;
 643
 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);
 652         BT_PUT(clock_class);
 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);
 660         BT_PUT(clock_value);
 661         if (ret) {
 662                 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 663                                 __LINE__);
 664                 goto error;
 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
 674         writer_clock_value = bt_ctf_clock_value_create(writer_clock_class, value);
 675         BT_PUT(writer_clock_class);
 676         if (!writer_clock_value) {
 677                 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 678                                 __LINE__);
 679                 goto error;
 680         }
 681
 682         ret = bt_ctf_event_set_clock_value(writer_event, writer_clock_value);
 683         BT_PUT(writer_clock_value);
 684         if (ret) {
 685                 fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 686                                 __LINE__);
 687                 goto error;
 688         }
 689
 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__);
 694                 goto end;
 695         }
 696
 697         ret = bt_ctf_event_set_header(writer_event, writer_event_header);
 698         BT_PUT(writer_event_header);
 699         if (ret < 0) {
 700                 fprintf(err, "[error] %s in %s:%d\n", __func__,
 701                                 __FILE__, __LINE__);
 702                 goto error;
 703         }
 704
 705         ret = 0;
 706
 707         goto end;
 708
 709 error:
 710         ret = -1;
 711 end:
 712         return ret;
 713 }
 714
 715 BT_HIDDEN
 716 struct bt_ctf_event *ctf_copy_event(FILE *err, struct bt_ctf_event *event,
 717                 struct bt_ctf_event_class *writer_event_class,
 718                 bool override_ts64)
 719 {
 720         struct bt_ctf_event *writer_event = NULL;
 721         struct bt_ctf_field *field = NULL, *copy_field = NULL;
 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) {
 733                 fprintf(err, "[error] %s in %s:%d\n", __func__,
 734                                 __FILE__, __LINE__);
 735                 goto error;
 736         }
 737
 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) {
 745                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 746                                         __FILE__, __LINE__);
 747                         goto error;
 748                 }
 749
 750                 ret = copy_override_field(err, event, writer_event, field,
 751                                 copy_field);
 752                 if (ret) {
 753                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 754                                         __FILE__, __LINE__);
 755                         goto error;
 756                 }
 757                 BT_PUT(copy_field);
 758         } else {
 759                 ret = ctf_copy_event_header(err, event, writer_event_class,
 760                                 writer_event, field);
 761                 if (ret) {
 762                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 763                                         __FILE__, __LINE__);
 764                         goto error;
 765                 }
 766         }
 767         BT_PUT(field);
 768
 769         /* Optional field, so it can fail silently. */
 770         field = bt_ctf_event_get_stream_event_context(event);
 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                 }
 778                 ret = bt_ctf_event_set_stream_event_context(writer_event,
 779                                 copy_field);
 780                 if (ret < 0) {
 781                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 782                                         __FILE__, __LINE__);
 783                         goto error;
 784                 }
 785                 BT_PUT(field);
 786                 BT_PUT(copy_field);
 787         }
 788
 789         /* Optional field, so it can fail silently. */
 790         field = bt_ctf_event_get_event_context(event);
 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                 }
 798                 ret = bt_ctf_event_set_event_context(writer_event, copy_field);
 799                 if (ret < 0) {
 800                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 801                                         __FILE__, __LINE__);
 802                         goto error;
 803                 }
 804                 BT_PUT(field);
 805                 BT_PUT(copy_field);
 806         }
 807
 808         field = bt_ctf_event_get_event_payload(event);
 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                 }
 816                 ret = bt_ctf_event_set_event_payload(writer_event, copy_field);
 817                 if (ret < 0) {
 818                         fprintf(err, "[error] %s in %s:%d\n", __func__,
 819                                         __FILE__, __LINE__);
 820                         goto error;
 821                 }
 822                 BT_PUT(field);
 823                 BT_PUT(copy_field);
 824         }
 825
 826         goto end;
 827
 828 error:
 829         BT_PUT(writer_event);
 830 end:
 831         bt_put(field);
 832         bt_put(copy_field);
 833         return writer_event;
 834 }
 835
 836 BT_HIDDEN
 837 enum 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;
 842         struct bt_ctf_field_type *header_type = NULL;
 843         enum bt_ctf_byte_order order;
 844         const char *trace_name;
 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;
 850                 struct bt_value *value = NULL;
 851
 852                 name = bt_ctf_trace_get_environment_field_name_by_index(
 853                         trace, i);
 854                 if (!name) {
 855                         fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 856                                         __LINE__);
 857                         ret = BT_COMPONENT_STATUS_ERROR;
 858                         goto end;
 859                 }
 860                 value = bt_ctf_trace_get_environment_field_value_by_index(
 861                         trace, i);
 862                 if (!value) {
 863                         fprintf(err, "[error] %s in %s:%d\n", __func__, __FILE__,
 864                                         __LINE__);
 865                         ret = BT_COMPONENT_STATUS_ERROR;
 866                         goto end;
 867                 }
 868
 869                 ret_int = bt_ctf_trace_set_environment_field(writer_trace,
 870                                 name, value);
 871                 BT_PUT(value);
 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;
 878                         goto end;
 879                 }
 880         }
 881
 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
 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          */
 895         if (order != BT_CTF_BYTE_ORDER_UNSPECIFIED) {
 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                 }
 902         }
 903
 904         header_type = bt_ctf_trace_get_packet_header_type(writer_trace);
 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                 }
 913         }
 914
 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
 925 end:
 926         return ret;
 927 }