linux/tools/perf/util/data-convert-bt.c
<<
>>
Prefs
   1/*
   2 * CTF writing support via babeltrace.
   3 *
   4 * Copyright (C) 2014, Jiri Olsa <jolsa@redhat.com>
   5 * Copyright (C) 2014, Sebastian Andrzej Siewior <bigeasy@linutronix.de>
   6 *
   7 * Released under the GPL v2. (and only v2, not any later version)
   8 */
   9
  10#include <linux/compiler.h>
  11#include <babeltrace/ctf-writer/writer.h>
  12#include <babeltrace/ctf-writer/clock.h>
  13#include <babeltrace/ctf-writer/stream.h>
  14#include <babeltrace/ctf-writer/event.h>
  15#include <babeltrace/ctf-writer/event-types.h>
  16#include <babeltrace/ctf-writer/event-fields.h>
  17#include <babeltrace/ctf-ir/utils.h>
  18#include <babeltrace/ctf/events.h>
  19#include <traceevent/event-parse.h>
  20#include "asm/bug.h"
  21#include "data-convert-bt.h"
  22#include "session.h"
  23#include "util.h"
  24#include "debug.h"
  25#include "tool.h"
  26#include "evlist.h"
  27#include "evsel.h"
  28#include "machine.h"
  29#include "config.h"
  30
  31#define pr_N(n, fmt, ...) \
  32        eprintf(n, debug_data_convert, fmt, ##__VA_ARGS__)
  33
  34#define pr(fmt, ...)  pr_N(1, pr_fmt(fmt), ##__VA_ARGS__)
  35#define pr2(fmt, ...) pr_N(2, pr_fmt(fmt), ##__VA_ARGS__)
  36
  37#define pr_time2(t, fmt, ...) pr_time_N(2, debug_data_convert, t, pr_fmt(fmt), ##__VA_ARGS__)
  38
  39struct evsel_priv {
  40        struct bt_ctf_event_class *event_class;
  41};
  42
  43#define MAX_CPUS        4096
  44
  45struct ctf_stream {
  46        struct bt_ctf_stream *stream;
  47        int cpu;
  48        u32 count;
  49};
  50
  51struct ctf_writer {
  52        /* writer primitives */
  53        struct bt_ctf_writer             *writer;
  54        struct ctf_stream               **stream;
  55        int                               stream_cnt;
  56        struct bt_ctf_stream_class       *stream_class;
  57        struct bt_ctf_clock              *clock;
  58
  59        /* data types */
  60        union {
  61                struct {
  62                        struct bt_ctf_field_type        *s64;
  63                        struct bt_ctf_field_type        *u64;
  64                        struct bt_ctf_field_type        *s32;
  65                        struct bt_ctf_field_type        *u32;
  66                        struct bt_ctf_field_type        *string;
  67                        struct bt_ctf_field_type        *u32_hex;
  68                        struct bt_ctf_field_type        *u64_hex;
  69                };
  70                struct bt_ctf_field_type *array[6];
  71        } data;
  72        struct bt_ctf_event_class       *comm_class;
  73        struct bt_ctf_event_class       *exit_class;
  74        struct bt_ctf_event_class       *fork_class;
  75};
  76
  77struct convert {
  78        struct perf_tool        tool;
  79        struct ctf_writer       writer;
  80
  81        u64                     events_size;
  82        u64                     events_count;
  83        u64                     non_sample_count;
  84
  85        /* Ordered events configured queue size. */
  86        u64                     queue_size;
  87};
  88
  89static int value_set(struct bt_ctf_field_type *type,
  90                     struct bt_ctf_event *event,
  91                     const char *name, u64 val)
  92{
  93        struct bt_ctf_field *field;
  94        bool sign = bt_ctf_field_type_integer_get_signed(type);
  95        int ret;
  96
  97        field = bt_ctf_field_create(type);
  98        if (!field) {
  99                pr_err("failed to create a field %s\n", name);
 100                return -1;
 101        }
 102
 103        if (sign) {
 104                ret = bt_ctf_field_signed_integer_set_value(field, val);
 105                if (ret) {
 106                        pr_err("failed to set field value %s\n", name);
 107                        goto err;
 108                }
 109        } else {
 110                ret = bt_ctf_field_unsigned_integer_set_value(field, val);
 111                if (ret) {
 112                        pr_err("failed to set field value %s\n", name);
 113                        goto err;
 114                }
 115        }
 116
 117        ret = bt_ctf_event_set_payload(event, name, field);
 118        if (ret) {
 119                pr_err("failed to set payload %s\n", name);
 120                goto err;
 121        }
 122
 123        pr2("  SET [%s = %" PRIu64 "]\n", name, val);
 124
 125err:
 126        bt_ctf_field_put(field);
 127        return ret;
 128}
 129
 130#define __FUNC_VALUE_SET(_name, _val_type)                              \
 131static __maybe_unused int value_set_##_name(struct ctf_writer *cw,      \
 132                             struct bt_ctf_event *event,                \
 133                             const char *name,                          \
 134                             _val_type val)                             \
 135{                                                                       \
 136        struct bt_ctf_field_type *type = cw->data._name;                \
 137        return value_set(type, event, name, (u64) val);                 \
 138}
 139
 140#define FUNC_VALUE_SET(_name) __FUNC_VALUE_SET(_name, _name)
 141
 142FUNC_VALUE_SET(s32)
 143FUNC_VALUE_SET(u32)
 144FUNC_VALUE_SET(s64)
 145FUNC_VALUE_SET(u64)
 146__FUNC_VALUE_SET(u64_hex, u64)
 147
 148static int string_set_value(struct bt_ctf_field *field, const char *string);
 149static __maybe_unused int
 150value_set_string(struct ctf_writer *cw, struct bt_ctf_event *event,
 151                 const char *name, const char *string)
 152{
 153        struct bt_ctf_field_type *type = cw->data.string;
 154        struct bt_ctf_field *field;
 155        int ret = 0;
 156
 157        field = bt_ctf_field_create(type);
 158        if (!field) {
 159                pr_err("failed to create a field %s\n", name);
 160                return -1;
 161        }
 162
 163        ret = string_set_value(field, string);
 164        if (ret) {
 165                pr_err("failed to set value %s\n", name);
 166                goto err_put_field;
 167        }
 168
 169        ret = bt_ctf_event_set_payload(event, name, field);
 170        if (ret)
 171                pr_err("failed to set payload %s\n", name);
 172
 173err_put_field:
 174        bt_ctf_field_put(field);
 175        return ret;
 176}
 177
 178static struct bt_ctf_field_type*
 179get_tracepoint_field_type(struct ctf_writer *cw, struct format_field *field)
 180{
 181        unsigned long flags = field->flags;
 182
 183        if (flags & FIELD_IS_STRING)
 184                return cw->data.string;
 185
 186        if (!(flags & FIELD_IS_SIGNED)) {
 187                /* unsigned long are mostly pointers */
 188                if (flags & FIELD_IS_LONG || flags & FIELD_IS_POINTER)
 189                        return cw->data.u64_hex;
 190        }
 191
 192        if (flags & FIELD_IS_SIGNED) {
 193                if (field->size == 8)
 194                        return cw->data.s64;
 195                else
 196                        return cw->data.s32;
 197        }
 198
 199        if (field->size == 8)
 200                return cw->data.u64;
 201        else
 202                return cw->data.u32;
 203}
 204
 205static unsigned long long adjust_signedness(unsigned long long value_int, int size)
 206{
 207        unsigned long long value_mask;
 208
 209        /*
 210         * value_mask = (1 << (size * 8 - 1)) - 1.
 211         * Directly set value_mask for code readers.
 212         */
 213        switch (size) {
 214        case 1:
 215                value_mask = 0x7fULL;
 216                break;
 217        case 2:
 218                value_mask = 0x7fffULL;
 219                break;
 220        case 4:
 221                value_mask = 0x7fffffffULL;
 222                break;
 223        case 8:
 224                /*
 225                 * For 64 bit value, return it self. There is no need
 226                 * to fill high bit.
 227                 */
 228                /* Fall through */
 229        default:
 230                /* BUG! */
 231                return value_int;
 232        }
 233
 234        /* If it is a positive value, don't adjust. */
 235        if ((value_int & (~0ULL - value_mask)) == 0)
 236                return value_int;
 237
 238        /* Fill upper part of value_int with 1 to make it a negative long long. */
 239        return (value_int & value_mask) | ~value_mask;
 240}
 241
 242static int string_set_value(struct bt_ctf_field *field, const char *string)
 243{
 244        char *buffer = NULL;
 245        size_t len = strlen(string), i, p;
 246        int err;
 247
 248        for (i = p = 0; i < len; i++, p++) {
 249                if (isprint(string[i])) {
 250                        if (!buffer)
 251                                continue;
 252                        buffer[p] = string[i];
 253                } else {
 254                        char numstr[5];
 255
 256                        snprintf(numstr, sizeof(numstr), "\\x%02x",
 257                                 (unsigned int)(string[i]) & 0xff);
 258
 259                        if (!buffer) {
 260                                buffer = zalloc(i + (len - i) * 4 + 2);
 261                                if (!buffer) {
 262                                        pr_err("failed to set unprintable string '%s'\n", string);
 263                                        return bt_ctf_field_string_set_value(field, "UNPRINTABLE-STRING");
 264                                }
 265                                if (i > 0)
 266                                        strncpy(buffer, string, i);
 267                        }
 268                        strncat(buffer + p, numstr, 4);
 269                        p += 3;
 270                }
 271        }
 272
 273        if (!buffer)
 274                return bt_ctf_field_string_set_value(field, string);
 275        err = bt_ctf_field_string_set_value(field, buffer);
 276        free(buffer);
 277        return err;
 278}
 279
 280static int add_tracepoint_field_value(struct ctf_writer *cw,
 281                                      struct bt_ctf_event_class *event_class,
 282                                      struct bt_ctf_event *event,
 283                                      struct perf_sample *sample,
 284                                      struct format_field *fmtf)
 285{
 286        struct bt_ctf_field_type *type;
 287        struct bt_ctf_field *array_field;
 288        struct bt_ctf_field *field;
 289        const char *name = fmtf->name;
 290        void *data = sample->raw_data;
 291        unsigned long flags = fmtf->flags;
 292        unsigned int n_items;
 293        unsigned int i;
 294        unsigned int offset;
 295        unsigned int len;
 296        int ret;
 297
 298        name = fmtf->alias;
 299        offset = fmtf->offset;
 300        len = fmtf->size;
 301        if (flags & FIELD_IS_STRING)
 302                flags &= ~FIELD_IS_ARRAY;
 303
 304        if (flags & FIELD_IS_DYNAMIC) {
 305                unsigned long long tmp_val;
 306
 307                tmp_val = pevent_read_number(fmtf->event->pevent,
 308                                data + offset, len);
 309                offset = tmp_val;
 310                len = offset >> 16;
 311                offset &= 0xffff;
 312        }
 313
 314        if (flags & FIELD_IS_ARRAY) {
 315
 316                type = bt_ctf_event_class_get_field_by_name(
 317                                event_class, name);
 318                array_field = bt_ctf_field_create(type);
 319                bt_ctf_field_type_put(type);
 320                if (!array_field) {
 321                        pr_err("Failed to create array type %s\n", name);
 322                        return -1;
 323                }
 324
 325                len = fmtf->size / fmtf->arraylen;
 326                n_items = fmtf->arraylen;
 327        } else {
 328                n_items = 1;
 329                array_field = NULL;
 330        }
 331
 332        type = get_tracepoint_field_type(cw, fmtf);
 333
 334        for (i = 0; i < n_items; i++) {
 335                if (flags & FIELD_IS_ARRAY)
 336                        field = bt_ctf_field_array_get_field(array_field, i);
 337                else
 338                        field = bt_ctf_field_create(type);
 339
 340                if (!field) {
 341                        pr_err("failed to create a field %s\n", name);
 342                        return -1;
 343                }
 344
 345                if (flags & FIELD_IS_STRING)
 346                        ret = string_set_value(field, data + offset + i * len);
 347                else {
 348                        unsigned long long value_int;
 349
 350                        value_int = pevent_read_number(
 351                                        fmtf->event->pevent,
 352                                        data + offset + i * len, len);
 353
 354                        if (!(flags & FIELD_IS_SIGNED))
 355                                ret = bt_ctf_field_unsigned_integer_set_value(
 356                                                field, value_int);
 357                        else
 358                                ret = bt_ctf_field_signed_integer_set_value(
 359                                                field, adjust_signedness(value_int, len));
 360                }
 361
 362                if (ret) {
 363                        pr_err("failed to set file value %s\n", name);
 364                        goto err_put_field;
 365                }
 366                if (!(flags & FIELD_IS_ARRAY)) {
 367                        ret = bt_ctf_event_set_payload(event, name, field);
 368                        if (ret) {
 369                                pr_err("failed to set payload %s\n", name);
 370                                goto err_put_field;
 371                        }
 372                }
 373                bt_ctf_field_put(field);
 374        }
 375        if (flags & FIELD_IS_ARRAY) {
 376                ret = bt_ctf_event_set_payload(event, name, array_field);
 377                if (ret) {
 378                        pr_err("Failed add payload array %s\n", name);
 379                        return -1;
 380                }
 381                bt_ctf_field_put(array_field);
 382        }
 383        return 0;
 384
 385err_put_field:
 386        bt_ctf_field_put(field);
 387        return -1;
 388}
 389
 390static int add_tracepoint_fields_values(struct ctf_writer *cw,
 391                                        struct bt_ctf_event_class *event_class,
 392                                        struct bt_ctf_event *event,
 393                                        struct format_field *fields,
 394                                        struct perf_sample *sample)
 395{
 396        struct format_field *field;
 397        int ret;
 398
 399        for (field = fields; field; field = field->next) {
 400                ret = add_tracepoint_field_value(cw, event_class, event, sample,
 401                                field);
 402                if (ret)
 403                        return -1;
 404        }
 405        return 0;
 406}
 407
 408static int add_tracepoint_values(struct ctf_writer *cw,
 409                                 struct bt_ctf_event_class *event_class,
 410                                 struct bt_ctf_event *event,
 411                                 struct perf_evsel *evsel,
 412                                 struct perf_sample *sample)
 413{
 414        struct format_field *common_fields = evsel->tp_format->format.common_fields;
 415        struct format_field *fields        = evsel->tp_format->format.fields;
 416        int ret;
 417
 418        ret = add_tracepoint_fields_values(cw, event_class, event,
 419                                           common_fields, sample);
 420        if (!ret)
 421                ret = add_tracepoint_fields_values(cw, event_class, event,
 422                                                   fields, sample);
 423
 424        return ret;
 425}
 426
 427static int
 428add_bpf_output_values(struct bt_ctf_event_class *event_class,
 429                      struct bt_ctf_event *event,
 430                      struct perf_sample *sample)
 431{
 432        struct bt_ctf_field_type *len_type, *seq_type;
 433        struct bt_ctf_field *len_field, *seq_field;
 434        unsigned int raw_size = sample->raw_size;
 435        unsigned int nr_elements = raw_size / sizeof(u32);
 436        unsigned int i;
 437        int ret;
 438
 439        if (nr_elements * sizeof(u32) != raw_size)
 440                pr_warning("Incorrect raw_size (%u) in bpf output event, skip %lu bytes\n",
 441                           raw_size, nr_elements * sizeof(u32) - raw_size);
 442
 443        len_type = bt_ctf_event_class_get_field_by_name(event_class, "raw_len");
 444        len_field = bt_ctf_field_create(len_type);
 445        if (!len_field) {
 446                pr_err("failed to create 'raw_len' for bpf output event\n");
 447                ret = -1;
 448                goto put_len_type;
 449        }
 450
 451        ret = bt_ctf_field_unsigned_integer_set_value(len_field, nr_elements);
 452        if (ret) {
 453                pr_err("failed to set field value for raw_len\n");
 454                goto put_len_field;
 455        }
 456        ret = bt_ctf_event_set_payload(event, "raw_len", len_field);
 457        if (ret) {
 458                pr_err("failed to set payload to raw_len\n");
 459                goto put_len_field;
 460        }
 461
 462        seq_type = bt_ctf_event_class_get_field_by_name(event_class, "raw_data");
 463        seq_field = bt_ctf_field_create(seq_type);
 464        if (!seq_field) {
 465                pr_err("failed to create 'raw_data' for bpf output event\n");
 466                ret = -1;
 467                goto put_seq_type;
 468        }
 469
 470        ret = bt_ctf_field_sequence_set_length(seq_field, len_field);
 471        if (ret) {
 472                pr_err("failed to set length of 'raw_data'\n");
 473                goto put_seq_field;
 474        }
 475
 476        for (i = 0; i < nr_elements; i++) {
 477                struct bt_ctf_field *elem_field =
 478                        bt_ctf_field_sequence_get_field(seq_field, i);
 479
 480                ret = bt_ctf_field_unsigned_integer_set_value(elem_field,
 481                                ((u32 *)(sample->raw_data))[i]);
 482
 483                bt_ctf_field_put(elem_field);
 484                if (ret) {
 485                        pr_err("failed to set raw_data[%d]\n", i);
 486                        goto put_seq_field;
 487                }
 488        }
 489
 490        ret = bt_ctf_event_set_payload(event, "raw_data", seq_field);
 491        if (ret)
 492                pr_err("failed to set payload for raw_data\n");
 493
 494put_seq_field:
 495        bt_ctf_field_put(seq_field);
 496put_seq_type:
 497        bt_ctf_field_type_put(seq_type);
 498put_len_field:
 499        bt_ctf_field_put(len_field);
 500put_len_type:
 501        bt_ctf_field_type_put(len_type);
 502        return ret;
 503}
 504
 505static int add_generic_values(struct ctf_writer *cw,
 506                              struct bt_ctf_event *event,
 507                              struct perf_evsel *evsel,
 508                              struct perf_sample *sample)
 509{
 510        u64 type = evsel->attr.sample_type;
 511        int ret;
 512
 513        /*
 514         * missing:
 515         *   PERF_SAMPLE_TIME         - not needed as we have it in
 516         *                              ctf event header
 517         *   PERF_SAMPLE_READ         - TODO
 518         *   PERF_SAMPLE_CALLCHAIN    - TODO
 519         *   PERF_SAMPLE_RAW          - tracepoint fields are handled separately
 520         *   PERF_SAMPLE_BRANCH_STACK - TODO
 521         *   PERF_SAMPLE_REGS_USER    - TODO
 522         *   PERF_SAMPLE_STACK_USER   - TODO
 523         */
 524
 525        if (type & PERF_SAMPLE_IP) {
 526                ret = value_set_u64_hex(cw, event, "perf_ip", sample->ip);
 527                if (ret)
 528                        return -1;
 529        }
 530
 531        if (type & PERF_SAMPLE_TID) {
 532                ret = value_set_s32(cw, event, "perf_tid", sample->tid);
 533                if (ret)
 534                        return -1;
 535
 536                ret = value_set_s32(cw, event, "perf_pid", sample->pid);
 537                if (ret)
 538                        return -1;
 539        }
 540
 541        if ((type & PERF_SAMPLE_ID) ||
 542            (type & PERF_SAMPLE_IDENTIFIER)) {
 543                ret = value_set_u64(cw, event, "perf_id", sample->id);
 544                if (ret)
 545                        return -1;
 546        }
 547
 548        if (type & PERF_SAMPLE_STREAM_ID) {
 549                ret = value_set_u64(cw, event, "perf_stream_id", sample->stream_id);
 550                if (ret)
 551                        return -1;
 552        }
 553
 554        if (type & PERF_SAMPLE_PERIOD) {
 555                ret = value_set_u64(cw, event, "perf_period", sample->period);
 556                if (ret)
 557                        return -1;
 558        }
 559
 560        if (type & PERF_SAMPLE_WEIGHT) {
 561                ret = value_set_u64(cw, event, "perf_weight", sample->weight);
 562                if (ret)
 563                        return -1;
 564        }
 565
 566        if (type & PERF_SAMPLE_DATA_SRC) {
 567                ret = value_set_u64(cw, event, "perf_data_src",
 568                                sample->data_src);
 569                if (ret)
 570                        return -1;
 571        }
 572
 573        if (type & PERF_SAMPLE_TRANSACTION) {
 574                ret = value_set_u64(cw, event, "perf_transaction",
 575                                sample->transaction);
 576                if (ret)
 577                        return -1;
 578        }
 579
 580        return 0;
 581}
 582
 583static int ctf_stream__flush(struct ctf_stream *cs)
 584{
 585        int err = 0;
 586
 587        if (cs) {
 588                err = bt_ctf_stream_flush(cs->stream);
 589                if (err)
 590                        pr_err("CTF stream %d flush failed\n", cs->cpu);
 591
 592                pr("Flush stream for cpu %d (%u samples)\n",
 593                   cs->cpu, cs->count);
 594
 595                cs->count = 0;
 596        }
 597
 598        return err;
 599}
 600
 601static struct ctf_stream *ctf_stream__create(struct ctf_writer *cw, int cpu)
 602{
 603        struct ctf_stream *cs;
 604        struct bt_ctf_field *pkt_ctx   = NULL;
 605        struct bt_ctf_field *cpu_field = NULL;
 606        struct bt_ctf_stream *stream   = NULL;
 607        int ret;
 608
 609        cs = zalloc(sizeof(*cs));
 610        if (!cs) {
 611                pr_err("Failed to allocate ctf stream\n");
 612                return NULL;
 613        }
 614
 615        stream = bt_ctf_writer_create_stream(cw->writer, cw->stream_class);
 616        if (!stream) {
 617                pr_err("Failed to create CTF stream\n");
 618                goto out;
 619        }
 620
 621        pkt_ctx = bt_ctf_stream_get_packet_context(stream);
 622        if (!pkt_ctx) {
 623                pr_err("Failed to obtain packet context\n");
 624                goto out;
 625        }
 626
 627        cpu_field = bt_ctf_field_structure_get_field(pkt_ctx, "cpu_id");
 628        bt_ctf_field_put(pkt_ctx);
 629        if (!cpu_field) {
 630                pr_err("Failed to obtain cpu field\n");
 631                goto out;
 632        }
 633
 634        ret = bt_ctf_field_unsigned_integer_set_value(cpu_field, (u32) cpu);
 635        if (ret) {
 636                pr_err("Failed to update CPU number\n");
 637                goto out;
 638        }
 639
 640        bt_ctf_field_put(cpu_field);
 641
 642        cs->cpu    = cpu;
 643        cs->stream = stream;
 644        return cs;
 645
 646out:
 647        if (cpu_field)
 648                bt_ctf_field_put(cpu_field);
 649        if (stream)
 650                bt_ctf_stream_put(stream);
 651
 652        free(cs);
 653        return NULL;
 654}
 655
 656static void ctf_stream__delete(struct ctf_stream *cs)
 657{
 658        if (cs) {
 659                bt_ctf_stream_put(cs->stream);
 660                free(cs);
 661        }
 662}
 663
 664static struct ctf_stream *ctf_stream(struct ctf_writer *cw, int cpu)
 665{
 666        struct ctf_stream *cs = cw->stream[cpu];
 667
 668        if (!cs) {
 669                cs = ctf_stream__create(cw, cpu);
 670                cw->stream[cpu] = cs;
 671        }
 672
 673        return cs;
 674}
 675
 676static int get_sample_cpu(struct ctf_writer *cw, struct perf_sample *sample,
 677                          struct perf_evsel *evsel)
 678{
 679        int cpu = 0;
 680
 681        if (evsel->attr.sample_type & PERF_SAMPLE_CPU)
 682                cpu = sample->cpu;
 683
 684        if (cpu > cw->stream_cnt) {
 685                pr_err("Event was recorded for CPU %d, limit is at %d.\n",
 686                        cpu, cw->stream_cnt);
 687                cpu = 0;
 688        }
 689
 690        return cpu;
 691}
 692
 693#define STREAM_FLUSH_COUNT 100000
 694
 695/*
 696 * Currently we have no other way to determine the
 697 * time for the stream flush other than keep track
 698 * of the number of events and check it against
 699 * threshold.
 700 */
 701static bool is_flush_needed(struct ctf_stream *cs)
 702{
 703        return cs->count >= STREAM_FLUSH_COUNT;
 704}
 705
 706static int process_sample_event(struct perf_tool *tool,
 707                                union perf_event *_event,
 708                                struct perf_sample *sample,
 709                                struct perf_evsel *evsel,
 710                                struct machine *machine __maybe_unused)
 711{
 712        struct convert *c = container_of(tool, struct convert, tool);
 713        struct evsel_priv *priv = evsel->priv;
 714        struct ctf_writer *cw = &c->writer;
 715        struct ctf_stream *cs;
 716        struct bt_ctf_event_class *event_class;
 717        struct bt_ctf_event *event;
 718        int ret;
 719
 720        if (WARN_ONCE(!priv, "Failed to setup all events.\n"))
 721                return 0;
 722
 723        event_class = priv->event_class;
 724
 725        /* update stats */
 726        c->events_count++;
 727        c->events_size += _event->header.size;
 728
 729        pr_time2(sample->time, "sample %" PRIu64 "\n", c->events_count);
 730
 731        event = bt_ctf_event_create(event_class);
 732        if (!event) {
 733                pr_err("Failed to create an CTF event\n");
 734                return -1;
 735        }
 736
 737        bt_ctf_clock_set_time(cw->clock, sample->time);
 738
 739        ret = add_generic_values(cw, event, evsel, sample);
 740        if (ret)
 741                return -1;
 742
 743        if (evsel->attr.type == PERF_TYPE_TRACEPOINT) {
 744                ret = add_tracepoint_values(cw, event_class, event,
 745                                            evsel, sample);
 746                if (ret)
 747                        return -1;
 748        }
 749
 750        if (perf_evsel__is_bpf_output(evsel)) {
 751                ret = add_bpf_output_values(event_class, event, sample);
 752                if (ret)
 753                        return -1;
 754        }
 755
 756        cs = ctf_stream(cw, get_sample_cpu(cw, sample, evsel));
 757        if (cs) {
 758                if (is_flush_needed(cs))
 759                        ctf_stream__flush(cs);
 760
 761                cs->count++;
 762                bt_ctf_stream_append_event(cs->stream, event);
 763        }
 764
 765        bt_ctf_event_put(event);
 766        return cs ? 0 : -1;
 767}
 768
 769#define __NON_SAMPLE_SET_FIELD(_name, _type, _field)    \
 770do {                                                    \
 771        ret = value_set_##_type(cw, event, #_field, _event->_name._field);\
 772        if (ret)                                        \
 773                return -1;                              \
 774} while(0)
 775
 776#define __FUNC_PROCESS_NON_SAMPLE(_name, body)  \
 777static int process_##_name##_event(struct perf_tool *tool,      \
 778                                   union perf_event *_event,    \
 779                                   struct perf_sample *sample,  \
 780                                   struct machine *machine)     \
 781{                                                               \
 782        struct convert *c = container_of(tool, struct convert, tool);\
 783        struct ctf_writer *cw = &c->writer;                     \
 784        struct bt_ctf_event_class *event_class = cw->_name##_class;\
 785        struct bt_ctf_event *event;                             \
 786        struct ctf_stream *cs;                                  \
 787        int ret;                                                \
 788                                                                \
 789        c->non_sample_count++;                                  \
 790        c->events_size += _event->header.size;                  \
 791        event = bt_ctf_event_create(event_class);               \
 792        if (!event) {                                           \
 793                pr_err("Failed to create an CTF event\n");      \
 794                return -1;                                      \
 795        }                                                       \
 796                                                                \
 797        bt_ctf_clock_set_time(cw->clock, sample->time);         \
 798        body                                                    \
 799        cs = ctf_stream(cw, 0);                                 \
 800        if (cs) {                                               \
 801                if (is_flush_needed(cs))                        \
 802                        ctf_stream__flush(cs);                  \
 803                                                                \
 804                cs->count++;                                    \
 805                bt_ctf_stream_append_event(cs->stream, event);  \
 806        }                                                       \
 807        bt_ctf_event_put(event);                                \
 808                                                                \
 809        return perf_event__process_##_name(tool, _event, sample, machine);\
 810}
 811
 812__FUNC_PROCESS_NON_SAMPLE(comm,
 813        __NON_SAMPLE_SET_FIELD(comm, u32, pid);
 814        __NON_SAMPLE_SET_FIELD(comm, u32, tid);
 815        __NON_SAMPLE_SET_FIELD(comm, string, comm);
 816)
 817__FUNC_PROCESS_NON_SAMPLE(fork,
 818        __NON_SAMPLE_SET_FIELD(fork, u32, pid);
 819        __NON_SAMPLE_SET_FIELD(fork, u32, ppid);
 820        __NON_SAMPLE_SET_FIELD(fork, u32, tid);
 821        __NON_SAMPLE_SET_FIELD(fork, u32, ptid);
 822        __NON_SAMPLE_SET_FIELD(fork, u64, time);
 823)
 824
 825__FUNC_PROCESS_NON_SAMPLE(exit,
 826        __NON_SAMPLE_SET_FIELD(fork, u32, pid);
 827        __NON_SAMPLE_SET_FIELD(fork, u32, ppid);
 828        __NON_SAMPLE_SET_FIELD(fork, u32, tid);
 829        __NON_SAMPLE_SET_FIELD(fork, u32, ptid);
 830        __NON_SAMPLE_SET_FIELD(fork, u64, time);
 831)
 832#undef __NON_SAMPLE_SET_FIELD
 833#undef __FUNC_PROCESS_NON_SAMPLE
 834
 835/* If dup < 0, add a prefix. Else, add _dupl_X suffix. */
 836static char *change_name(char *name, char *orig_name, int dup)
 837{
 838        char *new_name = NULL;
 839        size_t len;
 840
 841        if (!name)
 842                name = orig_name;
 843
 844        if (dup >= 10)
 845                goto out;
 846        /*
 847         * Add '_' prefix to potential keywork.  According to
 848         * Mathieu Desnoyers (https://lkml.org/lkml/2015/1/23/652),
 849         * futher CTF spec updating may require us to use '$'.
 850         */
 851        if (dup < 0)
 852                len = strlen(name) + sizeof("_");
 853        else
 854                len = strlen(orig_name) + sizeof("_dupl_X");
 855
 856        new_name = malloc(len);
 857        if (!new_name)
 858                goto out;
 859
 860        if (dup < 0)
 861                snprintf(new_name, len, "_%s", name);
 862        else
 863                snprintf(new_name, len, "%s_dupl_%d", orig_name, dup);
 864
 865out:
 866        if (name != orig_name)
 867                free(name);
 868        return new_name;
 869}
 870
 871static int event_class_add_field(struct bt_ctf_event_class *event_class,
 872                struct bt_ctf_field_type *type,
 873                struct format_field *field)
 874{
 875        struct bt_ctf_field_type *t = NULL;
 876        char *name;
 877        int dup = 1;
 878        int ret;
 879
 880        /* alias was already assigned */
 881        if (field->alias != field->name)
 882                return bt_ctf_event_class_add_field(event_class, type,
 883                                (char *)field->alias);
 884
 885        name = field->name;
 886
 887        /* If 'name' is a keywork, add prefix. */
 888        if (bt_ctf_validate_identifier(name))
 889                name = change_name(name, field->name, -1);
 890
 891        if (!name) {
 892                pr_err("Failed to fix invalid identifier.");
 893                return -1;
 894        }
 895        while ((t = bt_ctf_event_class_get_field_by_name(event_class, name))) {
 896                bt_ctf_field_type_put(t);
 897                name = change_name(name, field->name, dup++);
 898                if (!name) {
 899                        pr_err("Failed to create dup name for '%s'\n", field->name);
 900                        return -1;
 901                }
 902        }
 903
 904        ret = bt_ctf_event_class_add_field(event_class, type, name);
 905        if (!ret)
 906                field->alias = name;
 907
 908        return ret;
 909}
 910
 911static int add_tracepoint_fields_types(struct ctf_writer *cw,
 912                                       struct format_field *fields,
 913                                       struct bt_ctf_event_class *event_class)
 914{
 915        struct format_field *field;
 916        int ret;
 917
 918        for (field = fields; field; field = field->next) {
 919                struct bt_ctf_field_type *type;
 920                unsigned long flags = field->flags;
 921
 922                pr2("  field '%s'\n", field->name);
 923
 924                type = get_tracepoint_field_type(cw, field);
 925                if (!type)
 926                        return -1;
 927
 928                /*
 929                 * A string is an array of chars. For this we use the string
 930                 * type and don't care that it is an array. What we don't
 931                 * support is an array of strings.
 932                 */
 933                if (flags & FIELD_IS_STRING)
 934                        flags &= ~FIELD_IS_ARRAY;
 935
 936                if (flags & FIELD_IS_ARRAY)
 937                        type = bt_ctf_field_type_array_create(type, field->arraylen);
 938
 939                ret = event_class_add_field(event_class, type, field);
 940
 941                if (flags & FIELD_IS_ARRAY)
 942                        bt_ctf_field_type_put(type);
 943
 944                if (ret) {
 945                        pr_err("Failed to add field '%s': %d\n",
 946                                        field->name, ret);
 947                        return -1;
 948                }
 949        }
 950
 951        return 0;
 952}
 953
 954static int add_tracepoint_types(struct ctf_writer *cw,
 955                                struct perf_evsel *evsel,
 956                                struct bt_ctf_event_class *class)
 957{
 958        struct format_field *common_fields = evsel->tp_format->format.common_fields;
 959        struct format_field *fields        = evsel->tp_format->format.fields;
 960        int ret;
 961
 962        ret = add_tracepoint_fields_types(cw, common_fields, class);
 963        if (!ret)
 964                ret = add_tracepoint_fields_types(cw, fields, class);
 965
 966        return ret;
 967}
 968
 969static int add_bpf_output_types(struct ctf_writer *cw,
 970                                struct bt_ctf_event_class *class)
 971{
 972        struct bt_ctf_field_type *len_type = cw->data.u32;
 973        struct bt_ctf_field_type *seq_base_type = cw->data.u32_hex;
 974        struct bt_ctf_field_type *seq_type;
 975        int ret;
 976
 977        ret = bt_ctf_event_class_add_field(class, len_type, "raw_len");
 978        if (ret)
 979                return ret;
 980
 981        seq_type = bt_ctf_field_type_sequence_create(seq_base_type, "raw_len");
 982        if (!seq_type)
 983                return -1;
 984
 985        return bt_ctf_event_class_add_field(class, seq_type, "raw_data");
 986}
 987
 988static int add_generic_types(struct ctf_writer *cw, struct perf_evsel *evsel,
 989                             struct bt_ctf_event_class *event_class)
 990{
 991        u64 type = evsel->attr.sample_type;
 992
 993        /*
 994         * missing:
 995         *   PERF_SAMPLE_TIME         - not needed as we have it in
 996         *                              ctf event header
 997         *   PERF_SAMPLE_READ         - TODO
 998         *   PERF_SAMPLE_CALLCHAIN    - TODO
 999         *   PERF_SAMPLE_RAW          - tracepoint fields and BPF output
1000         *                              are handled separately
1001         *   PERF_SAMPLE_BRANCH_STACK - TODO
1002         *   PERF_SAMPLE_REGS_USER    - TODO
1003         *   PERF_SAMPLE_STACK_USER   - TODO
1004         */
1005
1006#define ADD_FIELD(cl, t, n)                                             \
1007        do {                                                            \
1008                pr2("  field '%s'\n", n);                               \
1009                if (bt_ctf_event_class_add_field(cl, t, n)) {           \
1010                        pr_err("Failed to add field '%s';\n", n);       \
1011                        return -1;                                      \
1012                }                                                       \
1013        } while (0)
1014
1015        if (type & PERF_SAMPLE_IP)
1016                ADD_FIELD(event_class, cw->data.u64_hex, "perf_ip");
1017
1018        if (type & PERF_SAMPLE_TID) {
1019                ADD_FIELD(event_class, cw->data.s32, "perf_tid");
1020                ADD_FIELD(event_class, cw->data.s32, "perf_pid");
1021        }
1022
1023        if ((type & PERF_SAMPLE_ID) ||
1024            (type & PERF_SAMPLE_IDENTIFIER))
1025                ADD_FIELD(event_class, cw->data.u64, "perf_id");
1026
1027        if (type & PERF_SAMPLE_STREAM_ID)
1028                ADD_FIELD(event_class, cw->data.u64, "perf_stream_id");
1029
1030        if (type & PERF_SAMPLE_PERIOD)
1031                ADD_FIELD(event_class, cw->data.u64, "perf_period");
1032
1033        if (type & PERF_SAMPLE_WEIGHT)
1034                ADD_FIELD(event_class, cw->data.u64, "perf_weight");
1035
1036        if (type & PERF_SAMPLE_DATA_SRC)
1037                ADD_FIELD(event_class, cw->data.u64, "perf_data_src");
1038
1039        if (type & PERF_SAMPLE_TRANSACTION)
1040                ADD_FIELD(event_class, cw->data.u64, "perf_transaction");
1041
1042#undef ADD_FIELD
1043        return 0;
1044}
1045
1046static int add_event(struct ctf_writer *cw, struct perf_evsel *evsel)
1047{
1048        struct bt_ctf_event_class *event_class;
1049        struct evsel_priv *priv;
1050        const char *name = perf_evsel__name(evsel);
1051        int ret;
1052
1053        pr("Adding event '%s' (type %d)\n", name, evsel->attr.type);
1054
1055        event_class = bt_ctf_event_class_create(name);
1056        if (!event_class)
1057                return -1;
1058
1059        ret = add_generic_types(cw, evsel, event_class);
1060        if (ret)
1061                goto err;
1062
1063        if (evsel->attr.type == PERF_TYPE_TRACEPOINT) {
1064                ret = add_tracepoint_types(cw, evsel, event_class);
1065                if (ret)
1066                        goto err;
1067        }
1068
1069        if (perf_evsel__is_bpf_output(evsel)) {
1070                ret = add_bpf_output_types(cw, event_class);
1071                if (ret)
1072                        goto err;
1073        }
1074
1075        ret = bt_ctf_stream_class_add_event_class(cw->stream_class, event_class);
1076        if (ret) {
1077                pr("Failed to add event class into stream.\n");
1078                goto err;
1079        }
1080
1081        priv = malloc(sizeof(*priv));
1082        if (!priv)
1083                goto err;
1084
1085        priv->event_class = event_class;
1086        evsel->priv       = priv;
1087        return 0;
1088
1089err:
1090        bt_ctf_event_class_put(event_class);
1091        pr_err("Failed to add event '%s'.\n", name);
1092        return -1;
1093}
1094
1095static int setup_events(struct ctf_writer *cw, struct perf_session *session)
1096{
1097        struct perf_evlist *evlist = session->evlist;
1098        struct perf_evsel *evsel;
1099        int ret;
1100
1101        evlist__for_each_entry(evlist, evsel) {
1102                ret = add_event(cw, evsel);
1103                if (ret)
1104                        return ret;
1105        }
1106        return 0;
1107}
1108
1109#define __NON_SAMPLE_ADD_FIELD(t, n)                                            \
1110        do {                                                    \
1111                pr2("  field '%s'\n", #n);                      \
1112                if (bt_ctf_event_class_add_field(event_class, cw->data.t, #n)) {\
1113                        pr_err("Failed to add field '%s';\n", #n);\
1114                        return -1;                              \
1115                }                                               \
1116        } while(0)
1117
1118#define __FUNC_ADD_NON_SAMPLE_EVENT_CLASS(_name, body)          \
1119static int add_##_name##_event(struct ctf_writer *cw)           \
1120{                                                               \
1121        struct bt_ctf_event_class *event_class;                 \
1122        int ret;                                                \
1123                                                                \
1124        pr("Adding "#_name" event\n");                          \
1125        event_class = bt_ctf_event_class_create("perf_" #_name);\
1126        if (!event_class)                                       \
1127                return -1;                                      \
1128        body                                                    \
1129                                                                \
1130        ret = bt_ctf_stream_class_add_event_class(cw->stream_class, event_class);\
1131        if (ret) {                                              \
1132                pr("Failed to add event class '"#_name"' into stream.\n");\
1133                return ret;                                     \
1134        }                                                       \
1135                                                                \
1136        cw->_name##_class = event_class;                        \
1137        bt_ctf_event_class_put(event_class);                    \
1138        return 0;                                               \
1139}
1140
1141__FUNC_ADD_NON_SAMPLE_EVENT_CLASS(comm,
1142        __NON_SAMPLE_ADD_FIELD(u32, pid);
1143        __NON_SAMPLE_ADD_FIELD(u32, tid);
1144        __NON_SAMPLE_ADD_FIELD(string, comm);
1145)
1146
1147__FUNC_ADD_NON_SAMPLE_EVENT_CLASS(fork,
1148        __NON_SAMPLE_ADD_FIELD(u32, pid);
1149        __NON_SAMPLE_ADD_FIELD(u32, ppid);
1150        __NON_SAMPLE_ADD_FIELD(u32, tid);
1151        __NON_SAMPLE_ADD_FIELD(u32, ptid);
1152        __NON_SAMPLE_ADD_FIELD(u64, time);
1153)
1154
1155__FUNC_ADD_NON_SAMPLE_EVENT_CLASS(exit,
1156        __NON_SAMPLE_ADD_FIELD(u32, pid);
1157        __NON_SAMPLE_ADD_FIELD(u32, ppid);
1158        __NON_SAMPLE_ADD_FIELD(u32, tid);
1159        __NON_SAMPLE_ADD_FIELD(u32, ptid);
1160        __NON_SAMPLE_ADD_FIELD(u64, time);
1161)
1162
1163#undef __NON_SAMPLE_ADD_FIELD
1164#undef __FUNC_ADD_NON_SAMPLE_EVENT_CLASS
1165
1166static int setup_non_sample_events(struct ctf_writer *cw,
1167                                   struct perf_session *session __maybe_unused)
1168{
1169        int ret;
1170
1171        ret = add_comm_event(cw);
1172        if (ret)
1173                return ret;
1174        ret = add_exit_event(cw);
1175        if (ret)
1176                return ret;
1177        ret = add_fork_event(cw);
1178        if (ret)
1179                return ret;
1180        return 0;
1181}
1182
1183static void cleanup_events(struct perf_session *session)
1184{
1185        struct perf_evlist *evlist = session->evlist;
1186        struct perf_evsel *evsel;
1187
1188        evlist__for_each_entry(evlist, evsel) {
1189                struct evsel_priv *priv;
1190
1191                priv = evsel->priv;
1192                bt_ctf_event_class_put(priv->event_class);
1193                zfree(&evsel->priv);
1194        }
1195
1196        perf_evlist__delete(evlist);
1197        session->evlist = NULL;
1198}
1199
1200static int setup_streams(struct ctf_writer *cw, struct perf_session *session)
1201{
1202        struct ctf_stream **stream;
1203        struct perf_header *ph = &session->header;
1204        int ncpus;
1205
1206        /*
1207         * Try to get the number of cpus used in the data file,
1208         * if not present fallback to the MAX_CPUS.
1209         */
1210        ncpus = ph->env.nr_cpus_avail ?: MAX_CPUS;
1211
1212        stream = zalloc(sizeof(*stream) * ncpus);
1213        if (!stream) {
1214                pr_err("Failed to allocate streams.\n");
1215                return -ENOMEM;
1216        }
1217
1218        cw->stream     = stream;
1219        cw->stream_cnt = ncpus;
1220        return 0;
1221}
1222
1223static void free_streams(struct ctf_writer *cw)
1224{
1225        int cpu;
1226
1227        for (cpu = 0; cpu < cw->stream_cnt; cpu++)
1228                ctf_stream__delete(cw->stream[cpu]);
1229
1230        free(cw->stream);
1231}
1232
1233static int ctf_writer__setup_env(struct ctf_writer *cw,
1234                                 struct perf_session *session)
1235{
1236        struct perf_header *header = &session->header;
1237        struct bt_ctf_writer *writer = cw->writer;
1238
1239#define ADD(__n, __v)                                                   \
1240do {                                                                    \
1241        if (bt_ctf_writer_add_environment_field(writer, __n, __v))      \
1242                return -1;                                              \
1243} while (0)
1244
1245        ADD("host",    header->env.hostname);
1246        ADD("sysname", "Linux");
1247        ADD("release", header->env.os_release);
1248        ADD("version", header->env.version);
1249        ADD("machine", header->env.arch);
1250        ADD("domain", "kernel");
1251        ADD("tracer_name", "perf");
1252
1253#undef ADD
1254        return 0;
1255}
1256
1257static int ctf_writer__setup_clock(struct ctf_writer *cw)
1258{
1259        struct bt_ctf_clock *clock = cw->clock;
1260
1261        bt_ctf_clock_set_description(clock, "perf clock");
1262
1263#define SET(__n, __v)                           \
1264do {                                            \
1265        if (bt_ctf_clock_set_##__n(clock, __v)) \
1266                return -1;                      \
1267} while (0)
1268
1269        SET(frequency,   1000000000);
1270        SET(offset_s,    0);
1271        SET(offset,      0);
1272        SET(precision,   10);
1273        SET(is_absolute, 0);
1274
1275#undef SET
1276        return 0;
1277}
1278
1279static struct bt_ctf_field_type *create_int_type(int size, bool sign, bool hex)
1280{
1281        struct bt_ctf_field_type *type;
1282
1283        type = bt_ctf_field_type_integer_create(size);
1284        if (!type)
1285                return NULL;
1286
1287        if (sign &&
1288            bt_ctf_field_type_integer_set_signed(type, 1))
1289                goto err;
1290
1291        if (hex &&
1292            bt_ctf_field_type_integer_set_base(type, BT_CTF_INTEGER_BASE_HEXADECIMAL))
1293                goto err;
1294
1295#if __BYTE_ORDER == __BIG_ENDIAN
1296        bt_ctf_field_type_set_byte_order(type, BT_CTF_BYTE_ORDER_BIG_ENDIAN);
1297#else
1298        bt_ctf_field_type_set_byte_order(type, BT_CTF_BYTE_ORDER_LITTLE_ENDIAN);
1299#endif
1300
1301        pr2("Created type: INTEGER %d-bit %ssigned %s\n",
1302            size, sign ? "un" : "", hex ? "hex" : "");
1303        return type;
1304
1305err:
1306        bt_ctf_field_type_put(type);
1307        return NULL;
1308}
1309
1310static void ctf_writer__cleanup_data(struct ctf_writer *cw)
1311{
1312        unsigned int i;
1313
1314        for (i = 0; i < ARRAY_SIZE(cw->data.array); i++)
1315                bt_ctf_field_type_put(cw->data.array[i]);
1316}
1317
1318static int ctf_writer__init_data(struct ctf_writer *cw)
1319{
1320#define CREATE_INT_TYPE(type, size, sign, hex)          \
1321do {                                                    \
1322        (type) = create_int_type(size, sign, hex);      \
1323        if (!(type))                                    \
1324                goto err;                               \
1325} while (0)
1326
1327        CREATE_INT_TYPE(cw->data.s64, 64, true,  false);
1328        CREATE_INT_TYPE(cw->data.u64, 64, false, false);
1329        CREATE_INT_TYPE(cw->data.s32, 32, true,  false);
1330        CREATE_INT_TYPE(cw->data.u32, 32, false, false);
1331        CREATE_INT_TYPE(cw->data.u32_hex, 32, false, true);
1332        CREATE_INT_TYPE(cw->data.u64_hex, 64, false, true);
1333
1334        cw->data.string  = bt_ctf_field_type_string_create();
1335        if (cw->data.string)
1336                return 0;
1337
1338err:
1339        ctf_writer__cleanup_data(cw);
1340        pr_err("Failed to create data types.\n");
1341        return -1;
1342}
1343
1344static void ctf_writer__cleanup(struct ctf_writer *cw)
1345{
1346        ctf_writer__cleanup_data(cw);
1347
1348        bt_ctf_clock_put(cw->clock);
1349        free_streams(cw);
1350        bt_ctf_stream_class_put(cw->stream_class);
1351        bt_ctf_writer_put(cw->writer);
1352
1353        /* and NULL all the pointers */
1354        memset(cw, 0, sizeof(*cw));
1355}
1356
1357static int ctf_writer__init(struct ctf_writer *cw, const char *path)
1358{
1359        struct bt_ctf_writer            *writer;
1360        struct bt_ctf_stream_class      *stream_class;
1361        struct bt_ctf_clock             *clock;
1362        struct bt_ctf_field_type        *pkt_ctx_type;
1363        int                             ret;
1364
1365        /* CTF writer */
1366        writer = bt_ctf_writer_create(path);
1367        if (!writer)
1368                goto err;
1369
1370        cw->writer = writer;
1371
1372        /* CTF clock */
1373        clock = bt_ctf_clock_create("perf_clock");
1374        if (!clock) {
1375                pr("Failed to create CTF clock.\n");
1376                goto err_cleanup;
1377        }
1378
1379        cw->clock = clock;
1380
1381        if (ctf_writer__setup_clock(cw)) {
1382                pr("Failed to setup CTF clock.\n");
1383                goto err_cleanup;
1384        }
1385
1386        /* CTF stream class */
1387        stream_class = bt_ctf_stream_class_create("perf_stream");
1388        if (!stream_class) {
1389                pr("Failed to create CTF stream class.\n");
1390                goto err_cleanup;
1391        }
1392
1393        cw->stream_class = stream_class;
1394
1395        /* CTF clock stream setup */
1396        if (bt_ctf_stream_class_set_clock(stream_class, clock)) {
1397                pr("Failed to assign CTF clock to stream class.\n");
1398                goto err_cleanup;
1399        }
1400
1401        if (ctf_writer__init_data(cw))
1402                goto err_cleanup;
1403
1404        /* Add cpu_id for packet context */
1405        pkt_ctx_type = bt_ctf_stream_class_get_packet_context_type(stream_class);
1406        if (!pkt_ctx_type)
1407                goto err_cleanup;
1408
1409        ret = bt_ctf_field_type_structure_add_field(pkt_ctx_type, cw->data.u32, "cpu_id");
1410        bt_ctf_field_type_put(pkt_ctx_type);
1411        if (ret)
1412                goto err_cleanup;
1413
1414        /* CTF clock writer setup */
1415        if (bt_ctf_writer_add_clock(writer, clock)) {
1416                pr("Failed to assign CTF clock to writer.\n");
1417                goto err_cleanup;
1418        }
1419
1420        return 0;
1421
1422err_cleanup:
1423        ctf_writer__cleanup(cw);
1424err:
1425        pr_err("Failed to setup CTF writer.\n");
1426        return -1;
1427}
1428
1429static int ctf_writer__flush_streams(struct ctf_writer *cw)
1430{
1431        int cpu, ret = 0;
1432
1433        for (cpu = 0; cpu < cw->stream_cnt && !ret; cpu++)
1434                ret = ctf_stream__flush(cw->stream[cpu]);
1435
1436        return ret;
1437}
1438
1439static int convert__config(const char *var, const char *value, void *cb)
1440{
1441        struct convert *c = cb;
1442
1443        if (!strcmp(var, "convert.queue-size")) {
1444                c->queue_size = perf_config_u64(var, value);
1445                return 0;
1446        }
1447
1448        return 0;
1449}
1450
1451int bt_convert__perf2ctf(const char *input, const char *path,
1452                         struct perf_data_convert_opts *opts)
1453{
1454        struct perf_session *session;
1455        struct perf_data_file file = {
1456                .path = input,
1457                .mode = PERF_DATA_MODE_READ,
1458                .force = opts->force,
1459        };
1460        struct convert c = {
1461                .tool = {
1462                        .sample          = process_sample_event,
1463                        .mmap            = perf_event__process_mmap,
1464                        .mmap2           = perf_event__process_mmap2,
1465                        .comm            = perf_event__process_comm,
1466                        .exit            = perf_event__process_exit,
1467                        .fork            = perf_event__process_fork,
1468                        .lost            = perf_event__process_lost,
1469                        .tracing_data    = perf_event__process_tracing_data,
1470                        .build_id        = perf_event__process_build_id,
1471                        .ordered_events  = true,
1472                        .ordering_requires_timestamps = true,
1473                },
1474        };
1475        struct ctf_writer *cw = &c.writer;
1476        int err = -1;
1477
1478        if (opts->all) {
1479                c.tool.comm = process_comm_event;
1480                c.tool.exit = process_exit_event;
1481                c.tool.fork = process_fork_event;
1482        }
1483
1484        perf_config(convert__config, &c);
1485
1486        /* CTF writer */
1487        if (ctf_writer__init(cw, path))
1488                return -1;
1489
1490        /* perf.data session */
1491        session = perf_session__new(&file, 0, &c.tool);
1492        if (!session)
1493                goto free_writer;
1494
1495        if (c.queue_size) {
1496                ordered_events__set_alloc_size(&session->ordered_events,
1497                                               c.queue_size);
1498        }
1499
1500        /* CTF writer env/clock setup  */
1501        if (ctf_writer__setup_env(cw, session))
1502                goto free_session;
1503
1504        /* CTF events setup */
1505        if (setup_events(cw, session))
1506                goto free_session;
1507
1508        if (opts->all && setup_non_sample_events(cw, session))
1509                goto free_session;
1510
1511        if (setup_streams(cw, session))
1512                goto free_session;
1513
1514        err = perf_session__process_events(session);
1515        if (!err)
1516                err = ctf_writer__flush_streams(cw);
1517        else
1518                pr_err("Error during conversion.\n");
1519
1520        fprintf(stderr,
1521                "[ perf data convert: Converted '%s' into CTF data '%s' ]\n",
1522                file.path, path);
1523
1524        fprintf(stderr,
1525                "[ perf data convert: Converted and wrote %.3f MB (%" PRIu64 " samples",
1526                (double) c.events_size / 1024.0 / 1024.0,
1527                c.events_count);
1528
1529        if (!c.non_sample_count)
1530                fprintf(stderr, ") ]\n");
1531        else
1532                fprintf(stderr, ", %" PRIu64 " non-samples) ]\n", c.non_sample_count);
1533
1534        cleanup_events(session);
1535        perf_session__delete(session);
1536        ctf_writer__cleanup(cw);
1537
1538        return err;
1539
1540free_session:
1541        perf_session__delete(session);
1542free_writer:
1543        ctf_writer__cleanup(cw);
1544        pr_err("Error during conversion setup.\n");
1545        return err;
1546}
1547