]> Pileus Git - ~andy/linux/blobdiff - tools/perf/util/header.c
perf tools: Bring linear set of section headers for features
[~andy/linux] / tools / perf / util / header.c
index 9aae360c0f284762e04e39e6ab8b144d577597fc..ebed4f44ed36f8dff5641a0655956dcdc495088e 100644 (file)
@@ -2,11 +2,15 @@
 #include <unistd.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <linux/list.h>
 
 #include "util.h"
 #include "header.h"
 #include "../perf.h"
 #include "trace-event.h"
+#include "symbol.h"
+#include "data_map.h"
+#include "debug.h"
 
 /*
  * Create new perf.data header attribute:
@@ -48,25 +52,17 @@ void perf_header_attr__add_id(struct perf_header_attr *self, u64 id)
  */
 struct perf_header *perf_header__new(void)
 {
-       struct perf_header *self = malloc(sizeof(*self));
+       struct perf_header *self = calloc(sizeof(*self), 1);
 
        if (!self)
                die("nomem");
 
-       self->frozen = 0;
-
-       self->attrs = 0;
        self->size = 1;
        self->attr = malloc(sizeof(void *));
 
        if (!self->attr)
                die("nomem");
 
-       self->data_offset = 0;
-       self->data_size = 0;
-       self->trace_info_offset = 0;
-       self->trace_info_size = 0;
-
        return self;
 }
 
@@ -101,7 +97,7 @@ static struct perf_trace_event_type *events;
 void perf_header__push_event(u64 id, const char *name)
 {
        if (strlen(name) > MAX_EVENT_NAME)
-               printf("Event %s will be truncated\n", name);
+               pr_warning("Event %s will be truncated\n", name);
 
        if (!events) {
                events = malloc(sizeof(struct perf_trace_event_type));
@@ -149,14 +145,17 @@ struct perf_file_header {
        struct perf_file_section        attrs;
        struct perf_file_section        data;
        struct perf_file_section        event_types;
-       struct perf_file_section        trace_info;
+       DECLARE_BITMAP(adds_features, HEADER_FEAT_BITS);
 };
 
-static int trace_info;
+void perf_header__set_feat(struct perf_header *self, int feat)
+{
+       set_bit(feat, self->adds_features);
+}
 
-void perf_header__set_trace_info(void)
+bool perf_header__has_feat(const struct perf_header *self, int feat)
 {
-       trace_info = 1;
+       return test_bit(feat, self->adds_features);
 }
 
 static void do_write(int fd, void *buf, size_t size)
@@ -172,7 +171,75 @@ static void do_write(int fd, void *buf, size_t size)
        }
 }
 
-void perf_header__write(struct perf_header *self, int fd)
+static void write_buildid_table(int fd, struct list_head *id_head)
+{
+       struct build_id_list *iter, *next;
+
+       list_for_each_entry_safe(iter, next, id_head, list) {
+               struct build_id_event *b = &iter->event;
+
+               do_write(fd, b, sizeof(*b));
+               do_write(fd, (void *)iter->dso_name, iter->len);
+               list_del(&iter->list);
+               free(iter);
+       }
+}
+
+static void
+perf_header__adds_write(struct perf_header *self, int fd)
+{
+       LIST_HEAD(id_list);
+       int nr_sections;
+       struct perf_file_section *feat_sec;
+       int sec_size;
+       u64 sec_start;
+       int idx = 0;
+
+       if (fetch_build_id_table(&id_list))
+               perf_header__set_feat(self, HEADER_BUILD_ID);
+
+       nr_sections = bitmap_weight(self->adds_features, HEADER_FEAT_BITS);
+       if (!nr_sections)
+               return;
+
+       feat_sec = calloc(sizeof(*feat_sec), nr_sections);
+       if (!feat_sec)
+               die("No memory");
+
+       sec_size = sizeof(*feat_sec) * nr_sections;
+
+       sec_start = self->data_offset + self->data_size;
+       lseek(fd, sec_start + sec_size, SEEK_SET);
+
+       if (perf_header__has_feat(self, HEADER_TRACE_INFO)) {
+               struct perf_file_section *trace_sec;
+
+               trace_sec = &feat_sec[idx++];
+
+               /* Write trace info */
+               trace_sec->offset = lseek(fd, 0, SEEK_CUR);
+               read_tracing_data(fd, attrs, nr_counters);
+               trace_sec->size = lseek(fd, 0, SEEK_CUR) - trace_sec->offset;
+       }
+
+
+       if (perf_header__has_feat(self, HEADER_BUILD_ID)) {
+               struct perf_file_section *buildid_sec;
+
+               buildid_sec = &feat_sec[idx++];
+
+               /* Write build-ids */
+               buildid_sec->offset = lseek(fd, 0, SEEK_CUR);
+               write_buildid_table(fd, &id_list);
+               buildid_sec->size = lseek(fd, 0, SEEK_CUR) - buildid_sec->offset;
+       }
+
+       lseek(fd, sec_start, SEEK_SET);
+       do_write(fd, feat_sec, sec_size);
+       free(feat_sec);
+}
+
+void perf_header__write(struct perf_header *self, int fd, bool at_exit)
 {
        struct perf_file_header f_header;
        struct perf_file_attr   f_attr;
@@ -210,26 +277,11 @@ void perf_header__write(struct perf_header *self, int fd)
        if (events)
                do_write(fd, events, self->event_size);
 
-       if (trace_info) {
-               static int trace_info_written;
-
-               /*
-                * Write it only once
-                */
-               if (!trace_info_written) {
-                       self->trace_info_offset = lseek(fd, 0, SEEK_CUR);
-                       read_tracing_data(fd, attrs, nr_counters);
-                       self->trace_info_size = lseek(fd, 0, SEEK_CUR) -
-                                               self->trace_info_offset;
-                       trace_info_written = 1;
-               } else {
-                       lseek(fd, self->trace_info_offset +
-                               self->trace_info_size, SEEK_SET);
-               }
-       }
-
        self->data_offset = lseek(fd, 0, SEEK_CUR);
 
+       if (at_exit)
+               perf_header__adds_write(self, fd);
+
        f_header = (struct perf_file_header){
                .magic     = PERF_MAGIC,
                .size      = sizeof(f_header),
@@ -246,12 +298,10 @@ void perf_header__write(struct perf_header *self, int fd)
                        .offset = self->event_offset,
                        .size   = self->event_size,
                },
-               .trace_info = {
-                       .offset = self->trace_info_offset,
-                       .size = self->trace_info_size,
-               },
        };
 
+       memcpy(&f_header.adds_features, &self->adds_features, sizeof(self->adds_features));
+
        lseek(fd, 0, SEEK_SET);
        do_write(fd, &f_header, sizeof(f_header));
        lseek(fd, self->data_offset + self->data_size, SEEK_SET);
@@ -274,6 +324,48 @@ static void do_read(int fd, void *buf, size_t size)
        }
 }
 
+static void perf_header__adds_read(struct perf_header *self, int fd)
+{
+       struct perf_file_section *feat_sec;
+       int nr_sections;
+       int sec_size;
+       int idx = 0;
+
+
+       nr_sections = bitmap_weight(self->adds_features, HEADER_FEAT_BITS);
+       if (!nr_sections)
+               return;
+
+       feat_sec = calloc(sizeof(*feat_sec), nr_sections);
+       if (!feat_sec)
+               die("No memory");
+
+       sec_size = sizeof(*feat_sec) * nr_sections;
+
+       lseek(fd, self->data_offset + self->data_size, SEEK_SET);
+
+       do_read(fd, feat_sec, sec_size);
+
+       if (perf_header__has_feat(self, HEADER_TRACE_INFO)) {
+               struct perf_file_section *trace_sec;
+
+               trace_sec = &feat_sec[idx++];
+               lseek(fd, trace_sec->offset, SEEK_SET);
+               trace_report(fd);
+       }
+
+       if (perf_header__has_feat(self, HEADER_BUILD_ID)) {
+               struct perf_file_section *buildid_sec;
+
+               buildid_sec = &feat_sec[idx++];
+               lseek(fd, buildid_sec->offset, SEEK_SET);
+               if (perf_header__read_build_ids(fd, buildid_sec->size))
+                       pr_debug("failed to read buildids, continuing...\n");
+       }
+
+       free(feat_sec);
+};
+
 struct perf_header *perf_header__read(int fd)
 {
        struct perf_header      *self = perf_header__new();
@@ -292,8 +384,8 @@ struct perf_header *perf_header__read(int fd)
 
        if (f_header.size != sizeof(f_header)) {
                /* Support the previous format */
-               if (f_header.size == offsetof(typeof(f_header), trace_info))
-                       f_header.trace_info.size = 0;
+               if (f_header.size == offsetof(typeof(f_header), adds_features))
+                       bitmap_zero(f_header.adds_features, HEADER_FEAT_BITS);
                else
                        die("incompatible file format");
        }
@@ -330,13 +422,7 @@ struct perf_header *perf_header__read(int fd)
                event_count =  f_header.event_types.size / sizeof(struct perf_trace_event_type);
        }
 
-       self->trace_info_offset = f_header.trace_info.offset;
-       self->trace_info_size = f_header.trace_info.size;
-
-       if (self->trace_info_size) {
-               lseek(fd, self->trace_info_offset, SEEK_SET);
-               trace_report(fd);
-       }
+       memcpy(&self->adds_features, &f_header.adds_features, sizeof(f_header.adds_features));
 
        self->event_offset = f_header.event_types.offset;
        self->event_size   = f_header.event_types.size;
@@ -344,6 +430,8 @@ struct perf_header *perf_header__read(int fd)
        self->data_offset = f_header.data.offset;
        self->data_size   = f_header.data.size;
 
+       perf_header__adds_read(self, fd);
+
        lseek(fd, self->data_offset, SEEK_SET);
 
        self->frozen = 1;