1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3 * Copyright (C) 2009, Steven Rostedt <srostedt@redhat.com>
4 */
5 #include <dirent.h>
6 #include <stdio.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <stdarg.h>
10 #include <sys/types.h>
11 #include <sys/stat.h>
12 #include <sys/wait.h>
13 #include <sys/mman.h>
14 #include <fcntl.h>
15 #include <unistd.h>
16 #include <errno.h>
17
18 #include "trace-event.h"
19 #include "debug.h"
20
21 static int input_fd;
22
23 static ssize_t trace_data_size;
24 static bool repipe;
25
__do_read(int fd,void * buf,int size)26 static int __do_read(int fd, void *buf, int size)
27 {
28 int rsize = size;
29
30 while (size) {
31 int ret = read(fd, buf, size);
32
33 if (ret <= 0)
34 return -1;
35
36 if (repipe) {
37 int retw = write(STDOUT_FILENO, buf, ret);
38
39 if (retw <= 0 || retw != ret) {
40 pr_debug("repiping input file");
41 return -1;
42 }
43 }
44
45 size -= ret;
46 buf += ret;
47 }
48
49 return rsize;
50 }
51
do_read(void * data,int size)52 static int do_read(void *data, int size)
53 {
54 int r;
55
56 r = __do_read(input_fd, data, size);
57 if (r <= 0) {
58 pr_debug("reading input file (size expected=%d received=%d)",
59 size, r);
60 return -1;
61 }
62
63 trace_data_size += r;
64
65 return r;
66 }
67
68 /* If it fails, the next read will report it */
skip(int size)69 static void skip(int size)
70 {
71 char buf[BUFSIZ];
72 int r;
73
74 while (size) {
75 r = size > BUFSIZ ? BUFSIZ : size;
76 do_read(buf, r);
77 size -= r;
78 }
79 }
80
read4(struct tep_handle * pevent)81 static unsigned int read4(struct tep_handle *pevent)
82 {
83 unsigned int data;
84
85 if (do_read(&data, 4) < 0)
86 return 0;
87 return tep_read_number(pevent, &data, 4);
88 }
89
read8(struct tep_handle * pevent)90 static unsigned long long read8(struct tep_handle *pevent)
91 {
92 unsigned long long data;
93
94 if (do_read(&data, 8) < 0)
95 return 0;
96 return tep_read_number(pevent, &data, 8);
97 }
98
read_string(void)99 static char *read_string(void)
100 {
101 char buf[BUFSIZ];
102 char *str = NULL;
103 int size = 0;
104 off_t r;
105 char c;
106
107 for (;;) {
108 r = read(input_fd, &c, 1);
109 if (r < 0) {
110 pr_debug("reading input file");
111 goto out;
112 }
113
114 if (!r) {
115 pr_debug("no data");
116 goto out;
117 }
118
119 if (repipe) {
120 int retw = write(STDOUT_FILENO, &c, 1);
121
122 if (retw <= 0 || retw != r) {
123 pr_debug("repiping input file string");
124 goto out;
125 }
126 }
127
128 buf[size++] = c;
129
130 if (!c)
131 break;
132 }
133
134 trace_data_size += size;
135
136 str = malloc(size);
137 if (str)
138 memcpy(str, buf, size);
139 out:
140 return str;
141 }
142
read_proc_kallsyms(struct tep_handle * pevent)143 static int read_proc_kallsyms(struct tep_handle *pevent)
144 {
145 unsigned int size;
146
147 size = read4(pevent);
148 if (!size)
149 return 0;
150 /*
151 * Just skip it, now that we configure libtraceevent to use the
152 * tools/perf/ symbol resolver.
153 *
154 * We need to skip it so that we can continue parsing old perf.data
155 * files, that contains this /proc/kallsyms payload.
156 *
157 * Newer perf.data files will have just the 4-bytes zeros "kallsyms
158 * payload", so that older tools can continue reading it and interpret
159 * it as "no kallsyms payload is present".
160 */
161 lseek(input_fd, size, SEEK_CUR);
162 trace_data_size += size;
163 return 0;
164 }
165
read_ftrace_printk(struct tep_handle * pevent)166 static int read_ftrace_printk(struct tep_handle *pevent)
167 {
168 unsigned int size;
169 char *buf;
170
171 /* it can have 0 size */
172 size = read4(pevent);
173 if (!size)
174 return 0;
175
176 buf = malloc(size + 1);
177 if (buf == NULL)
178 return -1;
179
180 if (do_read(buf, size) < 0) {
181 free(buf);
182 return -1;
183 }
184
185 buf[size] = '\0';
186
187 parse_ftrace_printk(pevent, buf, size);
188
189 free(buf);
190 return 0;
191 }
192
read_header_files(struct tep_handle * pevent)193 static int read_header_files(struct tep_handle *pevent)
194 {
195 unsigned long long size;
196 char *header_page;
197 char buf[BUFSIZ];
198 int ret = 0;
199
200 if (do_read(buf, 12) < 0)
201 return -1;
202
203 if (memcmp(buf, "header_page", 12) != 0) {
204 pr_debug("did not read header page");
205 return -1;
206 }
207
208 size = read8(pevent);
209
210 header_page = malloc(size);
211 if (header_page == NULL)
212 return -1;
213
214 if (do_read(header_page, size) < 0) {
215 pr_debug("did not read header page");
216 free(header_page);
217 return -1;
218 }
219
220 if (!tep_parse_header_page(pevent, header_page, size,
221 tep_get_long_size(pevent))) {
222 /*
223 * The commit field in the page is of type long,
224 * use that instead, since it represents the kernel.
225 */
226 tep_set_long_size(pevent, tep_get_header_page_size(pevent));
227 }
228 free(header_page);
229
230 if (do_read(buf, 13) < 0)
231 return -1;
232
233 if (memcmp(buf, "header_event", 13) != 0) {
234 pr_debug("did not read header event");
235 return -1;
236 }
237
238 size = read8(pevent);
239 skip(size);
240
241 return ret;
242 }
243
read_ftrace_file(struct tep_handle * pevent,unsigned long long size)244 static int read_ftrace_file(struct tep_handle *pevent, unsigned long long size)
245 {
246 int ret;
247 char *buf;
248
249 buf = malloc(size);
250 if (buf == NULL) {
251 pr_debug("memory allocation failure\n");
252 return -1;
253 }
254
255 ret = do_read(buf, size);
256 if (ret < 0) {
257 pr_debug("error reading ftrace file.\n");
258 goto out;
259 }
260
261 ret = parse_ftrace_file(pevent, buf, size);
262 if (ret < 0)
263 pr_debug("error parsing ftrace file.\n");
264 out:
265 free(buf);
266 return ret;
267 }
268
read_event_file(struct tep_handle * pevent,char * sys,unsigned long long size)269 static int read_event_file(struct tep_handle *pevent, char *sys,
270 unsigned long long size)
271 {
272 int ret;
273 char *buf;
274
275 buf = malloc(size);
276 if (buf == NULL) {
277 pr_debug("memory allocation failure\n");
278 return -1;
279 }
280
281 ret = do_read(buf, size);
282 if (ret < 0)
283 goto out;
284
285 ret = parse_event_file(pevent, buf, size, sys);
286 if (ret < 0)
287 pr_debug("error parsing event file.\n");
288 out:
289 free(buf);
290 return ret;
291 }
292
read_ftrace_files(struct tep_handle * pevent)293 static int read_ftrace_files(struct tep_handle *pevent)
294 {
295 unsigned long long size;
296 int count;
297 int i;
298 int ret;
299
300 count = read4(pevent);
301
302 for (i = 0; i < count; i++) {
303 size = read8(pevent);
304 ret = read_ftrace_file(pevent, size);
305 if (ret)
306 return ret;
307 }
308 return 0;
309 }
310
read_event_files(struct tep_handle * pevent)311 static int read_event_files(struct tep_handle *pevent)
312 {
313 unsigned long long size;
314 char *sys;
315 int systems;
316 int count;
317 int i,x;
318 int ret;
319
320 systems = read4(pevent);
321
322 for (i = 0; i < systems; i++) {
323 sys = read_string();
324 if (sys == NULL)
325 return -1;
326
327 count = read4(pevent);
328
329 for (x=0; x < count; x++) {
330 size = read8(pevent);
331 ret = read_event_file(pevent, sys, size);
332 if (ret) {
333 free(sys);
334 return ret;
335 }
336 }
337 free(sys);
338 }
339 return 0;
340 }
341
read_saved_cmdline(struct tep_handle * pevent)342 static int read_saved_cmdline(struct tep_handle *pevent)
343 {
344 unsigned long long size;
345 char *buf;
346 int ret;
347
348 /* it can have 0 size */
349 size = read8(pevent);
350 if (!size)
351 return 0;
352
353 buf = malloc(size + 1);
354 if (buf == NULL) {
355 pr_debug("memory allocation failure\n");
356 return -1;
357 }
358
359 ret = do_read(buf, size);
360 if (ret < 0) {
361 pr_debug("error reading saved cmdlines\n");
362 goto out;
363 }
364
365 parse_saved_cmdline(pevent, buf, size);
366 ret = 0;
367 out:
368 free(buf);
369 return ret;
370 }
371
trace_report(int fd,struct trace_event * tevent,bool __repipe)372 ssize_t trace_report(int fd, struct trace_event *tevent, bool __repipe)
373 {
374 char buf[BUFSIZ];
375 char test[] = { 23, 8, 68 };
376 char *version;
377 int show_version = 0;
378 int show_funcs = 0;
379 int show_printk = 0;
380 ssize_t size = -1;
381 int file_bigendian;
382 int host_bigendian;
383 int file_long_size;
384 int file_page_size;
385 struct tep_handle *pevent = NULL;
386 int err;
387
388 repipe = __repipe;
389 input_fd = fd;
390
391 if (do_read(buf, 3) < 0)
392 return -1;
393 if (memcmp(buf, test, 3) != 0) {
394 pr_debug("no trace data in the file");
395 return -1;
396 }
397
398 if (do_read(buf, 7) < 0)
399 return -1;
400 if (memcmp(buf, "tracing", 7) != 0) {
401 pr_debug("not a trace file (missing 'tracing' tag)");
402 return -1;
403 }
404
405 version = read_string();
406 if (version == NULL)
407 return -1;
408 if (show_version)
409 printf("version = %s\n", version);
410
411 if (do_read(buf, 1) < 0) {
412 free(version);
413 return -1;
414 }
415 file_bigendian = buf[0];
416 host_bigendian = bigendian();
417
418 if (trace_event__init(tevent)) {
419 pr_debug("trace_event__init failed");
420 goto out;
421 }
422
423 pevent = tevent->pevent;
424
425 tep_set_flag(pevent, TEP_NSEC_OUTPUT);
426 tep_set_file_bigendian(pevent, file_bigendian);
427 tep_set_local_bigendian(pevent, host_bigendian);
428
429 if (do_read(buf, 1) < 0)
430 goto out;
431 file_long_size = buf[0];
432
433 file_page_size = read4(pevent);
434 if (!file_page_size)
435 goto out;
436
437 tep_set_long_size(pevent, file_long_size);
438 tep_set_page_size(pevent, file_page_size);
439
440 err = read_header_files(pevent);
441 if (err)
442 goto out;
443 err = read_ftrace_files(pevent);
444 if (err)
445 goto out;
446 err = read_event_files(pevent);
447 if (err)
448 goto out;
449 err = read_proc_kallsyms(pevent);
450 if (err)
451 goto out;
452 err = read_ftrace_printk(pevent);
453 if (err)
454 goto out;
455 if (atof(version) >= 0.6) {
456 err = read_saved_cmdline(pevent);
457 if (err)
458 goto out;
459 }
460
461 size = trace_data_size;
462 repipe = false;
463
464 if (show_funcs) {
465 tep_print_funcs(pevent);
466 } else if (show_printk) {
467 tep_print_printk(pevent);
468 }
469
470 pevent = NULL;
471
472 out:
473 if (pevent)
474 trace_event__cleanup(tevent);
475 free(version);
476 return size;
477 }
478