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