Commit | Line | Data |
---|---|---|
7c6a1c65 PZ |
1 | #include <sys/types.h> |
2 | #include <unistd.h> | |
3 | #include <stdio.h> | |
4 | #include <stdlib.h> | |
8671dab9 | 5 | #include <linux/list.h> |
7c6a1c65 PZ |
6 | |
7 | #include "util.h" | |
8 | #include "header.h" | |
03456a15 FW |
9 | #include "../perf.h" |
10 | #include "trace-event.h" | |
8671dab9 | 11 | #include "symbol.h" |
4778d2e4 FW |
12 | #include "data_map.h" |
13 | #include "debug.h" | |
7c6a1c65 PZ |
14 | |
15 | /* | |
ec60a3fe | 16 | * Create new perf.data header attribute: |
7c6a1c65 | 17 | */ |
cdd6c482 | 18 | struct perf_header_attr *perf_header_attr__new(struct perf_event_attr *attr) |
7c6a1c65 PZ |
19 | { |
20 | struct perf_header_attr *self = malloc(sizeof(*self)); | |
21 | ||
dc79c0fc ACM |
22 | if (self != NULL) { |
23 | self->attr = *attr; | |
24 | self->ids = 0; | |
25 | self->size = 1; | |
26 | self->id = malloc(sizeof(u64)); | |
27 | if (self->id == NULL) { | |
28 | free(self); | |
29 | self = NULL; | |
30 | } | |
31 | } | |
7c6a1c65 PZ |
32 | |
33 | return self; | |
34 | } | |
35 | ||
11deb1f9 ACM |
36 | void perf_header_attr__delete(struct perf_header_attr *self) |
37 | { | |
38 | free(self->id); | |
39 | free(self); | |
40 | } | |
41 | ||
58754121 | 42 | int perf_header_attr__add_id(struct perf_header_attr *self, u64 id) |
7c6a1c65 PZ |
43 | { |
44 | int pos = self->ids; | |
45 | ||
46 | self->ids++; | |
47 | if (self->ids > self->size) { | |
58754121 ACM |
48 | int nsize = self->size * 2; |
49 | u64 *nid = realloc(self->id, nsize * sizeof(u64)); | |
50 | ||
51 | if (nid == NULL) | |
52 | return -1; | |
53 | ||
54 | self->size = nsize; | |
55 | self->id = nid; | |
7c6a1c65 PZ |
56 | } |
57 | self->id[pos] = id; | |
58754121 | 58 | return 0; |
7c6a1c65 PZ |
59 | } |
60 | ||
61 | /* | |
ec60a3fe | 62 | * Create new perf.data header: |
7c6a1c65 | 63 | */ |
7c6a1c65 PZ |
64 | struct perf_header *perf_header__new(void) |
65 | { | |
2ba08250 | 66 | struct perf_header *self = calloc(sizeof(*self), 1); |
7c6a1c65 | 67 | |
a9a70bbc ACM |
68 | if (self != NULL) { |
69 | self->size = 1; | |
70 | self->attr = malloc(sizeof(void *)); | |
7c6a1c65 | 71 | |
a9a70bbc ACM |
72 | if (self->attr == NULL) { |
73 | free(self); | |
74 | self = NULL; | |
75 | } | |
76 | } | |
7c6a1c65 | 77 | |
7c6a1c65 PZ |
78 | return self; |
79 | } | |
80 | ||
11deb1f9 ACM |
81 | int perf_header__add_attr(struct perf_header *self, |
82 | struct perf_header_attr *attr) | |
7c6a1c65 PZ |
83 | { |
84 | int pos = self->attrs; | |
85 | ||
86 | if (self->frozen) | |
11deb1f9 | 87 | return -1; |
7c6a1c65 PZ |
88 | |
89 | self->attrs++; | |
90 | if (self->attrs > self->size) { | |
11deb1f9 ACM |
91 | int nsize = self->size * 2; |
92 | struct perf_header_attr **nattr; | |
93 | ||
94 | nattr = realloc(self->attr, nsize * sizeof(void *)); | |
95 | if (nattr == NULL) | |
96 | return -1; | |
97 | ||
98 | self->size = nsize; | |
99 | self->attr = nattr; | |
7c6a1c65 PZ |
100 | } |
101 | self->attr[pos] = attr; | |
11deb1f9 | 102 | return 0; |
7c6a1c65 PZ |
103 | } |
104 | ||
ec60a3fe AV |
105 | #define MAX_EVENT_NAME 64 |
106 | ||
8755a8f2 AV |
107 | struct perf_trace_event_type { |
108 | u64 event_id; | |
ec60a3fe | 109 | char name[MAX_EVENT_NAME]; |
8755a8f2 AV |
110 | }; |
111 | ||
112 | static int event_count; | |
113 | static struct perf_trace_event_type *events; | |
114 | ||
115 | void perf_header__push_event(u64 id, const char *name) | |
116 | { | |
ec60a3fe | 117 | if (strlen(name) > MAX_EVENT_NAME) |
6beba7ad | 118 | pr_warning("Event %s will be truncated\n", name); |
8755a8f2 AV |
119 | |
120 | if (!events) { | |
121 | events = malloc(sizeof(struct perf_trace_event_type)); | |
122 | if (!events) | |
123 | die("nomem"); | |
124 | } else { | |
125 | events = realloc(events, (event_count + 1) * sizeof(struct perf_trace_event_type)); | |
126 | if (!events) | |
127 | die("nomem"); | |
128 | } | |
129 | memset(&events[event_count], 0, sizeof(struct perf_trace_event_type)); | |
130 | events[event_count].event_id = id; | |
ec60a3fe | 131 | strncpy(events[event_count].name, name, MAX_EVENT_NAME - 1); |
8755a8f2 AV |
132 | event_count++; |
133 | } | |
134 | ||
135 | char *perf_header__find_event(u64 id) | |
136 | { | |
137 | int i; | |
138 | for (i = 0 ; i < event_count; i++) { | |
139 | if (events[i].event_id == id) | |
140 | return events[i].name; | |
141 | } | |
142 | return NULL; | |
143 | } | |
144 | ||
7c6a1c65 PZ |
145 | static const char *__perf_magic = "PERFFILE"; |
146 | ||
147 | #define PERF_MAGIC (*(u64 *)__perf_magic) | |
148 | ||
7c6a1c65 | 149 | struct perf_file_attr { |
cdd6c482 | 150 | struct perf_event_attr attr; |
7c6a1c65 PZ |
151 | struct perf_file_section ids; |
152 | }; | |
153 | ||
8d06367f ACM |
154 | void perf_header__set_feat(struct perf_header *self, int feat) |
155 | { | |
156 | set_bit(feat, self->adds_features); | |
157 | } | |
158 | ||
159 | bool perf_header__has_feat(const struct perf_header *self, int feat) | |
160 | { | |
161 | return test_bit(feat, self->adds_features); | |
162 | } | |
163 | ||
3726cc75 | 164 | static int do_write(int fd, const void *buf, size_t size) |
7c6a1c65 PZ |
165 | { |
166 | while (size) { | |
167 | int ret = write(fd, buf, size); | |
168 | ||
169 | if (ret < 0) | |
3726cc75 | 170 | return -1; |
7c6a1c65 PZ |
171 | |
172 | size -= ret; | |
173 | buf += ret; | |
174 | } | |
3726cc75 ACM |
175 | |
176 | return 0; | |
7c6a1c65 PZ |
177 | } |
178 | ||
3726cc75 | 179 | static int write_buildid_table(int fd, struct list_head *id_head) |
8671dab9 | 180 | { |
57f395a7 FW |
181 | struct build_id_list *iter, *next; |
182 | ||
183 | list_for_each_entry_safe(iter, next, id_head, list) { | |
184 | struct build_id_event *b = &iter->event; | |
8671dab9 | 185 | |
3726cc75 ACM |
186 | if (do_write(fd, b, sizeof(*b)) < 0 || |
187 | do_write(fd, iter->dso_name, iter->len) < 0) | |
188 | return -1; | |
57f395a7 FW |
189 | list_del(&iter->list); |
190 | free(iter); | |
191 | } | |
3726cc75 ACM |
192 | |
193 | return 0; | |
8671dab9 FW |
194 | } |
195 | ||
196 | static void | |
9e827dd0 | 197 | perf_header__adds_write(struct perf_header *self, int fd) |
2ba08250 | 198 | { |
9e827dd0 FW |
199 | LIST_HEAD(id_list); |
200 | int nr_sections; | |
201 | struct perf_file_section *feat_sec; | |
202 | int sec_size; | |
203 | u64 sec_start; | |
204 | int idx = 0; | |
205 | ||
206 | if (fetch_build_id_table(&id_list)) | |
207 | perf_header__set_feat(self, HEADER_BUILD_ID); | |
208 | ||
209 | nr_sections = bitmap_weight(self->adds_features, HEADER_FEAT_BITS); | |
210 | if (!nr_sections) | |
211 | return; | |
212 | ||
213 | feat_sec = calloc(sizeof(*feat_sec), nr_sections); | |
214 | if (!feat_sec) | |
215 | die("No memory"); | |
216 | ||
217 | sec_size = sizeof(*feat_sec) * nr_sections; | |
218 | ||
219 | sec_start = self->data_offset + self->data_size; | |
220 | lseek(fd, sec_start + sec_size, SEEK_SET); | |
2ba08250 | 221 | |
3e13ab2d | 222 | if (perf_header__has_feat(self, HEADER_TRACE_INFO)) { |
9e827dd0 FW |
223 | struct perf_file_section *trace_sec; |
224 | ||
225 | trace_sec = &feat_sec[idx++]; | |
226 | ||
2ba08250 | 227 | /* Write trace info */ |
9e827dd0 | 228 | trace_sec->offset = lseek(fd, 0, SEEK_CUR); |
2ba08250 | 229 | read_tracing_data(fd, attrs, nr_counters); |
9e827dd0 | 230 | trace_sec->size = lseek(fd, 0, SEEK_CUR) - trace_sec->offset; |
2ba08250 | 231 | } |
8671dab9 | 232 | |
57f395a7 | 233 | |
9e827dd0 FW |
234 | if (perf_header__has_feat(self, HEADER_BUILD_ID)) { |
235 | struct perf_file_section *buildid_sec; | |
236 | ||
237 | buildid_sec = &feat_sec[idx++]; | |
238 | ||
239 | /* Write build-ids */ | |
240 | buildid_sec->offset = lseek(fd, 0, SEEK_CUR); | |
3726cc75 ACM |
241 | if (write_buildid_table(fd, &id_list) < 0) |
242 | die("failed to write buildid table"); | |
9e827dd0 | 243 | buildid_sec->size = lseek(fd, 0, SEEK_CUR) - buildid_sec->offset; |
8671dab9 | 244 | } |
9e827dd0 FW |
245 | |
246 | lseek(fd, sec_start, SEEK_SET); | |
3726cc75 ACM |
247 | if (do_write(fd, feat_sec, sec_size) < 0) |
248 | die("failed to write feature section"); | |
9e827dd0 FW |
249 | free(feat_sec); |
250 | } | |
2ba08250 | 251 | |
8671dab9 | 252 | void perf_header__write(struct perf_header *self, int fd, bool at_exit) |
7c6a1c65 PZ |
253 | { |
254 | struct perf_file_header f_header; | |
255 | struct perf_file_attr f_attr; | |
256 | struct perf_header_attr *attr; | |
257 | int i; | |
258 | ||
259 | lseek(fd, sizeof(f_header), SEEK_SET); | |
260 | ||
261 | ||
262 | for (i = 0; i < self->attrs; i++) { | |
263 | attr = self->attr[i]; | |
264 | ||
265 | attr->id_offset = lseek(fd, 0, SEEK_CUR); | |
3726cc75 ACM |
266 | if (do_write(fd, attr->id, attr->ids * sizeof(u64)) < 0) |
267 | die("failed to write perf header"); | |
7c6a1c65 PZ |
268 | } |
269 | ||
270 | ||
271 | self->attr_offset = lseek(fd, 0, SEEK_CUR); | |
272 | ||
273 | for (i = 0; i < self->attrs; i++) { | |
274 | attr = self->attr[i]; | |
275 | ||
276 | f_attr = (struct perf_file_attr){ | |
277 | .attr = attr->attr, | |
278 | .ids = { | |
279 | .offset = attr->id_offset, | |
280 | .size = attr->ids * sizeof(u64), | |
281 | } | |
282 | }; | |
3726cc75 ACM |
283 | if (do_write(fd, &f_attr, sizeof(f_attr)) < 0) |
284 | die("failed to write perf header attribute"); | |
7c6a1c65 PZ |
285 | } |
286 | ||
8755a8f2 AV |
287 | self->event_offset = lseek(fd, 0, SEEK_CUR); |
288 | self->event_size = event_count * sizeof(struct perf_trace_event_type); | |
289 | if (events) | |
3726cc75 ACM |
290 | if (do_write(fd, events, self->event_size) < 0) |
291 | die("failed to write perf header events"); | |
8755a8f2 | 292 | |
7c6a1c65 PZ |
293 | self->data_offset = lseek(fd, 0, SEEK_CUR); |
294 | ||
9e827dd0 FW |
295 | if (at_exit) |
296 | perf_header__adds_write(self, fd); | |
297 | ||
7c6a1c65 PZ |
298 | f_header = (struct perf_file_header){ |
299 | .magic = PERF_MAGIC, | |
300 | .size = sizeof(f_header), | |
301 | .attr_size = sizeof(f_attr), | |
302 | .attrs = { | |
303 | .offset = self->attr_offset, | |
304 | .size = self->attrs * sizeof(f_attr), | |
305 | }, | |
306 | .data = { | |
307 | .offset = self->data_offset, | |
308 | .size = self->data_size, | |
309 | }, | |
8755a8f2 AV |
310 | .event_types = { |
311 | .offset = self->event_offset, | |
312 | .size = self->event_size, | |
313 | }, | |
7c6a1c65 PZ |
314 | }; |
315 | ||
db9f11e3 | 316 | memcpy(&f_header.adds_features, &self->adds_features, sizeof(self->adds_features)); |
2ba08250 | 317 | |
7c6a1c65 | 318 | lseek(fd, 0, SEEK_SET); |
3726cc75 ACM |
319 | if (do_write(fd, &f_header, sizeof(f_header)) < 0) |
320 | die("failed to write perf header"); | |
7c6a1c65 PZ |
321 | lseek(fd, self->data_offset + self->data_size, SEEK_SET); |
322 | ||
323 | self->frozen = 1; | |
324 | } | |
325 | ||
326 | static void do_read(int fd, void *buf, size_t size) | |
327 | { | |
328 | while (size) { | |
329 | int ret = read(fd, buf, size); | |
330 | ||
331 | if (ret < 0) | |
332 | die("failed to read"); | |
7eac7e9e PH |
333 | if (ret == 0) |
334 | die("failed to read: missing data"); | |
7c6a1c65 PZ |
335 | |
336 | size -= ret; | |
337 | buf += ret; | |
338 | } | |
339 | } | |
340 | ||
37562eac ACM |
341 | int perf_header__process_sections(struct perf_header *self, int fd, |
342 | int (*process)(struct perf_file_section *self, | |
343 | int feat, int fd)) | |
2ba08250 | 344 | { |
9e827dd0 FW |
345 | struct perf_file_section *feat_sec; |
346 | int nr_sections; | |
347 | int sec_size; | |
348 | int idx = 0; | |
37562eac | 349 | int err = 0, feat = 1; |
9e827dd0 FW |
350 | |
351 | nr_sections = bitmap_weight(self->adds_features, HEADER_FEAT_BITS); | |
352 | if (!nr_sections) | |
37562eac | 353 | return 0; |
9e827dd0 FW |
354 | |
355 | feat_sec = calloc(sizeof(*feat_sec), nr_sections); | |
356 | if (!feat_sec) | |
37562eac | 357 | return -1; |
9e827dd0 FW |
358 | |
359 | sec_size = sizeof(*feat_sec) * nr_sections; | |
360 | ||
361 | lseek(fd, self->data_offset + self->data_size, SEEK_SET); | |
362 | ||
363 | do_read(fd, feat_sec, sec_size); | |
364 | ||
37562eac ACM |
365 | while (idx < nr_sections && feat < HEADER_LAST_FEATURE) { |
366 | if (perf_header__has_feat(self, feat)) { | |
367 | struct perf_file_section *sec = &feat_sec[idx++]; | |
2ba08250 | 368 | |
37562eac ACM |
369 | err = process(sec, feat, fd); |
370 | if (err < 0) | |
371 | break; | |
372 | } | |
373 | ++feat; | |
2ba08250 | 374 | } |
4778d2e4 | 375 | |
37562eac ACM |
376 | free(feat_sec); |
377 | return err; | |
378 | }; | |
4778d2e4 | 379 | |
37562eac ACM |
380 | int perf_file_header__read(struct perf_file_header *self, |
381 | struct perf_header *ph, int fd) | |
382 | { | |
383 | lseek(fd, 0, SEEK_SET); | |
384 | do_read(fd, self, sizeof(*self)); | |
385 | ||
386 | if (self->magic != PERF_MAGIC || | |
387 | self->attr_size != sizeof(struct perf_file_attr)) | |
388 | return -1; | |
389 | ||
390 | if (self->size != sizeof(*self)) { | |
391 | /* Support the previous format */ | |
392 | if (self->size == offsetof(typeof(*self), adds_features)) | |
393 | bitmap_zero(self->adds_features, HEADER_FEAT_BITS); | |
394 | else | |
395 | return -1; | |
4778d2e4 | 396 | } |
9e827dd0 | 397 | |
37562eac ACM |
398 | memcpy(&ph->adds_features, &self->adds_features, |
399 | sizeof(self->adds_features)); | |
400 | ||
401 | ph->event_offset = self->event_types.offset; | |
402 | ph->event_size = self->event_types.size; | |
403 | ph->data_offset = self->data.offset; | |
404 | ph->data_size = self->data.size; | |
405 | return 0; | |
406 | } | |
407 | ||
408 | static int perf_file_section__process(struct perf_file_section *self, | |
409 | int feat, int fd) | |
410 | { | |
411 | if (lseek(fd, self->offset, SEEK_SET) < 0) { | |
412 | pr_debug("Failed to lseek to %Ld offset for feature %d, " | |
413 | "continuing...\n", self->offset, feat); | |
414 | return 0; | |
415 | } | |
416 | ||
417 | switch (feat) { | |
418 | case HEADER_TRACE_INFO: | |
419 | trace_report(fd); | |
420 | break; | |
421 | ||
422 | case HEADER_BUILD_ID: | |
423 | if (perf_header__read_build_ids(fd, self->offset, self->size)) | |
424 | pr_debug("Failed to read buildids, continuing...\n"); | |
425 | break; | |
426 | default: | |
427 | pr_debug("unknown feature %d, continuing...\n", feat); | |
428 | } | |
429 | ||
430 | return 0; | |
431 | } | |
2ba08250 | 432 | |
7c6a1c65 PZ |
433 | struct perf_header *perf_header__read(int fd) |
434 | { | |
435 | struct perf_header *self = perf_header__new(); | |
436 | struct perf_file_header f_header; | |
437 | struct perf_file_attr f_attr; | |
438 | u64 f_id; | |
7c6a1c65 PZ |
439 | int nr_attrs, nr_ids, i, j; |
440 | ||
a9a70bbc ACM |
441 | if (self == NULL) |
442 | die("nomem"); | |
443 | ||
37562eac | 444 | if (perf_file_header__read(&f_header, self, fd) < 0) |
7c6a1c65 PZ |
445 | die("incompatible file format"); |
446 | ||
447 | nr_attrs = f_header.attrs.size / sizeof(f_attr); | |
448 | lseek(fd, f_header.attrs.offset, SEEK_SET); | |
449 | ||
450 | for (i = 0; i < nr_attrs; i++) { | |
451 | struct perf_header_attr *attr; | |
1c222bce | 452 | off_t tmp; |
7c6a1c65 PZ |
453 | |
454 | do_read(fd, &f_attr, sizeof(f_attr)); | |
1c222bce | 455 | tmp = lseek(fd, 0, SEEK_CUR); |
7c6a1c65 PZ |
456 | |
457 | attr = perf_header_attr__new(&f_attr.attr); | |
dc79c0fc ACM |
458 | if (attr == NULL) |
459 | die("nomem"); | |
7c6a1c65 PZ |
460 | |
461 | nr_ids = f_attr.ids.size / sizeof(u64); | |
462 | lseek(fd, f_attr.ids.offset, SEEK_SET); | |
463 | ||
464 | for (j = 0; j < nr_ids; j++) { | |
465 | do_read(fd, &f_id, sizeof(f_id)); | |
466 | ||
58754121 ACM |
467 | if (perf_header_attr__add_id(attr, f_id) < 0) |
468 | die("nomem"); | |
7c6a1c65 | 469 | } |
11deb1f9 ACM |
470 | if (perf_header__add_attr(self, attr) < 0) |
471 | die("nomem"); | |
472 | ||
7c6a1c65 PZ |
473 | lseek(fd, tmp, SEEK_SET); |
474 | } | |
475 | ||
8755a8f2 AV |
476 | if (f_header.event_types.size) { |
477 | lseek(fd, f_header.event_types.offset, SEEK_SET); | |
478 | events = malloc(f_header.event_types.size); | |
479 | if (!events) | |
480 | die("nomem"); | |
481 | do_read(fd, events, f_header.event_types.size); | |
482 | event_count = f_header.event_types.size / sizeof(struct perf_trace_event_type); | |
483 | } | |
03456a15 | 484 | |
37562eac | 485 | perf_header__process_sections(self, fd, perf_file_section__process); |
4778d2e4 | 486 | |
2e01d179 | 487 | lseek(fd, self->data_offset, SEEK_SET); |
7c6a1c65 PZ |
488 | |
489 | self->frozen = 1; | |
490 | ||
491 | return self; | |
492 | } | |
0d3a5c88 FW |
493 | |
494 | u64 perf_header__sample_type(struct perf_header *header) | |
495 | { | |
496 | u64 type = 0; | |
497 | int i; | |
498 | ||
499 | for (i = 0; i < header->attrs; i++) { | |
500 | struct perf_header_attr *attr = header->attr[i]; | |
501 | ||
502 | if (!type) | |
503 | type = attr->attr.sample_type; | |
504 | else if (type != attr->attr.sample_type) | |
505 | die("non matching sample_type"); | |
506 | } | |
507 | ||
508 | return type; | |
509 | } | |
510 | ||
cdd6c482 | 511 | struct perf_event_attr * |
0d3a5c88 FW |
512 | perf_header__find_attr(u64 id, struct perf_header *header) |
513 | { | |
514 | int i; | |
515 | ||
516 | for (i = 0; i < header->attrs; i++) { | |
517 | struct perf_header_attr *attr = header->attr[i]; | |
518 | int j; | |
519 | ||
520 | for (j = 0; j < attr->ids; j++) { | |
521 | if (attr->id[j] == id) | |
522 | return &attr->attr; | |
523 | } | |
524 | } | |
525 | ||
526 | return NULL; | |
527 | } |