1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * Copyright (c) 2013 Google, Inc
4 */
5
6 /* Decode and dump U-Boot profiling information */
7
8 #include <assert.h>
9 #include <ctype.h>
10 #include <limits.h>
11 #include <regex.h>
12 #include <stdarg.h>
13 #include <stdio.h>
14 #include <stdlib.h>
15 #include <string.h>
16 #include <unistd.h>
17 #include <sys/param.h>
18 #include <sys/types.h>
19
20 #include <compiler.h>
21 #include <trace.h>
22
23 #define MAX_LINE_LEN 500
24
25 enum {
26 FUNCF_TRACE = 1 << 0, /* Include this function in trace */
27 };
28
29 struct func_info {
30 unsigned long offset;
31 const char *name;
32 unsigned long code_size;
33 unsigned long call_count;
34 unsigned flags;
35 /* the section this function is in */
36 struct objsection_info *objsection;
37 };
38
39 enum trace_line_type {
40 TRACE_LINE_INCLUDE,
41 TRACE_LINE_EXCLUDE,
42 };
43
44 struct trace_configline_info {
45 struct trace_configline_info *next;
46 enum trace_line_type type;
47 const char *name; /* identifier name / wildcard */
48 regex_t regex; /* Regex to use if name starts with / */
49 };
50
51 /* The contents of the trace config file */
52 struct trace_configline_info *trace_config_head;
53
54 struct func_info *func_list;
55 int func_count;
56 struct trace_call *call_list;
57 int call_count;
58 int verbose; /* Verbosity level 0=none, 1=warn, 2=notice, 3=info, 4=debug */
59 unsigned long text_offset; /* text address of first function */
60
61 static void outf(int level, const char *fmt, ...)
62 __attribute__ ((format (__printf__, 2, 3)));
63 #define error(fmt, b...) outf(0, fmt, ##b)
64 #define warn(fmt, b...) outf(1, fmt, ##b)
65 #define notice(fmt, b...) outf(2, fmt, ##b)
66 #define info(fmt, b...) outf(3, fmt, ##b)
67 #define debug(fmt, b...) outf(4, fmt, ##b)
68
69
outf(int level,const char * fmt,...)70 static void outf(int level, const char *fmt, ...)
71 {
72 if (verbose >= level) {
73 va_list args;
74
75 va_start(args, fmt);
76 vfprintf(stderr, fmt, args);
77 va_end(args);
78 }
79 }
80
usage(void)81 static void usage(void)
82 {
83 fprintf(stderr,
84 "Usage: proftool -cds -v3 <cmd> <profdata>\n"
85 "\n"
86 "Commands\n"
87 " dump-ftrace\t\tDump out textual data in ftrace format\n"
88 "\n"
89 "Options:\n"
90 " -m <map>\tSpecify Systen.map file\n"
91 " -t <trace>\tSpecific trace data file (from U-Boot)\n"
92 " -v <0-4>\tSpecify verbosity\n");
93 exit(EXIT_FAILURE);
94 }
95
h_cmp_offset(const void * v1,const void * v2)96 static int h_cmp_offset(const void *v1, const void *v2)
97 {
98 const struct func_info *f1 = v1, *f2 = v2;
99
100 return (f1->offset / FUNC_SITE_SIZE) - (f2->offset / FUNC_SITE_SIZE);
101 }
102
read_system_map(FILE * fin)103 static int read_system_map(FILE *fin)
104 {
105 unsigned long offset, start = 0;
106 struct func_info *func;
107 char buff[MAX_LINE_LEN];
108 char symtype;
109 char symname[MAX_LINE_LEN + 1];
110 int linenum;
111 int alloced;
112
113 for (linenum = 1, alloced = func_count = 0;; linenum++) {
114 int fields = 0;
115
116 if (fgets(buff, sizeof(buff), fin))
117 fields = sscanf(buff, "%lx %c %100s\n", &offset,
118 &symtype, symname);
119 if (fields == 2) {
120 continue;
121 } else if (feof(fin)) {
122 break;
123 } else if (fields < 2) {
124 error("Map file line %d: invalid format\n", linenum);
125 return 1;
126 }
127
128 /* Must be a text symbol */
129 symtype = tolower(symtype);
130 if (symtype != 't' && symtype != 'w')
131 continue;
132
133 if (func_count == alloced) {
134 alloced += 256;
135 func_list = realloc(func_list,
136 sizeof(struct func_info) * alloced);
137 assert(func_list);
138 }
139 if (!func_count)
140 start = offset;
141
142 func = &func_list[func_count++];
143 memset(func, '\0', sizeof(*func));
144 func->offset = offset - start;
145 func->name = strdup(symname);
146 func->flags = FUNCF_TRACE; /* trace by default */
147
148 /* Update previous function's code size */
149 if (func_count > 1)
150 func[-1].code_size = func->offset - func[-1].offset;
151 }
152 notice("%d functions found in map file\n", func_count);
153 text_offset = start;
154 return 0;
155 }
156
read_data(FILE * fin,void * buff,int size)157 static int read_data(FILE *fin, void *buff, int size)
158 {
159 int err;
160
161 err = fread(buff, 1, size, fin);
162 if (!err)
163 return 1;
164 if (err != size) {
165 error("Cannot read profile file at pos %ld\n", ftell(fin));
166 return -1;
167 }
168 return 0;
169 }
170
find_func_by_offset(uint32_t offset)171 static struct func_info *find_func_by_offset(uint32_t offset)
172 {
173 struct func_info key, *found;
174
175 key.offset = offset;
176 found = bsearch(&key, func_list, func_count, sizeof(struct func_info),
177 h_cmp_offset);
178
179 return found;
180 }
181
182 /* This finds the function which contains the given offset */
find_caller_by_offset(uint32_t offset)183 static struct func_info *find_caller_by_offset(uint32_t offset)
184 {
185 int low; /* least function that could be a match */
186 int high; /* greated function that could be a match */
187 struct func_info key;
188
189 low = 0;
190 high = func_count - 1;
191 key.offset = offset;
192 while (high > low + 1) {
193 int mid = (low + high) / 2;
194 int result;
195
196 result = h_cmp_offset(&key, &func_list[mid]);
197 if (result > 0)
198 low = mid;
199 else if (result < 0)
200 high = mid;
201 else
202 return &func_list[mid];
203 }
204
205 return low >= 0 ? &func_list[low] : NULL;
206 }
207
read_calls(FILE * fin,size_t count)208 static int read_calls(FILE *fin, size_t count)
209 {
210 struct trace_call *call_data;
211 int i;
212
213 notice("call count: %zu\n", count);
214 call_list = (struct trace_call *)calloc(count, sizeof(*call_data));
215 if (!call_list) {
216 error("Cannot allocate call_list\n");
217 return -1;
218 }
219 call_count = count;
220
221 call_data = call_list;
222 for (i = 0; i < count; i++, call_data++) {
223 if (read_data(fin, call_data, sizeof(*call_data)))
224 return 1;
225 }
226 return 0;
227 }
228
read_profile(FILE * fin,int * not_found)229 static int read_profile(FILE *fin, int *not_found)
230 {
231 struct trace_output_hdr hdr;
232
233 *not_found = 0;
234 while (!feof(fin)) {
235 int err;
236
237 err = read_data(fin, &hdr, sizeof(hdr));
238 if (err == 1)
239 break; /* EOF */
240 else if (err)
241 return 1;
242
243 switch (hdr.type) {
244 case TRACE_CHUNK_FUNCS:
245 /* Ignored at present */
246 break;
247
248 case TRACE_CHUNK_CALLS:
249 if (read_calls(fin, hdr.rec_count))
250 return 1;
251 break;
252 }
253 }
254 return 0;
255 }
256
read_map_file(const char * fname)257 static int read_map_file(const char *fname)
258 {
259 FILE *fmap;
260 int err = 0;
261
262 fmap = fopen(fname, "r");
263 if (!fmap) {
264 error("Cannot open map file '%s'\n", fname);
265 return 1;
266 }
267 if (fmap) {
268 err = read_system_map(fmap);
269 fclose(fmap);
270 }
271 return err;
272 }
273
read_profile_file(const char * fname)274 static int read_profile_file(const char *fname)
275 {
276 int not_found = INT_MAX;
277 FILE *fprof;
278 int err;
279
280 fprof = fopen(fname, "rb");
281 if (!fprof) {
282 error("Cannot open profile data file '%s'\n",
283 fname);
284 return 1;
285 } else {
286 err = read_profile(fprof, ¬_found);
287 fclose(fprof);
288 if (err)
289 return err;
290
291 if (not_found) {
292 warn("%d profile functions could not be found in the map file - are you sure that your profile data and map file correspond?\n",
293 not_found);
294 return 1;
295 }
296 }
297 return 0;
298 }
299
regex_report_error(regex_t * regex,int err,const char * op,const char * name)300 static int regex_report_error(regex_t *regex, int err, const char *op,
301 const char *name)
302 {
303 char buf[200];
304
305 regerror(err, regex, buf, sizeof(buf));
306 error("Regex error '%s' in %s '%s'\n", buf, op, name);
307 return -1;
308 }
309
check_trace_config_line(struct trace_configline_info * item)310 static void check_trace_config_line(struct trace_configline_info *item)
311 {
312 struct func_info *func, *end;
313 int err;
314
315 debug("Checking trace config line '%s'\n", item->name);
316 for (func = func_list, end = func + func_count; func < end; func++) {
317 err = regexec(&item->regex, func->name, 0, NULL, 0);
318 debug(" - regex '%s', string '%s': %d\n", item->name,
319 func->name, err);
320 if (err == REG_NOMATCH)
321 continue;
322
323 if (err) {
324 regex_report_error(&item->regex, err, "match",
325 item->name);
326 break;
327 }
328
329 /* It matches, so perform the action */
330 switch (item->type) {
331 case TRACE_LINE_INCLUDE:
332 info(" include %s at %lx\n", func->name,
333 text_offset + func->offset);
334 func->flags |= FUNCF_TRACE;
335 break;
336
337 case TRACE_LINE_EXCLUDE:
338 info(" exclude %s at %lx\n", func->name,
339 text_offset + func->offset);
340 func->flags &= ~FUNCF_TRACE;
341 break;
342 }
343 }
344 }
345
check_trace_config(void)346 static void check_trace_config(void)
347 {
348 struct trace_configline_info *line;
349
350 for (line = trace_config_head; line; line = line->next)
351 check_trace_config_line(line);
352 }
353
354 /**
355 * Check the functions to see if they each have an objsection. If not, then
356 * the linker must have eliminated them.
357 */
check_functions(void)358 static void check_functions(void)
359 {
360 struct func_info *func, *end;
361 unsigned long removed_code_size = 0;
362 int not_found = 0;
363
364 /* Look for missing functions */
365 for (func = func_list, end = func + func_count; func < end; func++) {
366 if (!func->objsection) {
367 removed_code_size += func->code_size;
368 not_found++;
369 }
370 }
371
372 /* Figure out what functions we want to trace */
373 check_trace_config();
374
375 warn("%d functions removed by linker, %ld code size\n",
376 not_found, removed_code_size);
377 }
378
read_trace_config(FILE * fin)379 static int read_trace_config(FILE *fin)
380 {
381 char buff[200];
382 int linenum = 0;
383 struct trace_configline_info **tailp = &trace_config_head;
384
385 while (fgets(buff, sizeof(buff), fin)) {
386 int len = strlen(buff);
387 struct trace_configline_info *line;
388 char *saveptr;
389 char *s, *tok;
390 int err;
391
392 linenum++;
393 if (len && buff[len - 1] == '\n')
394 buff[len - 1] = '\0';
395
396 /* skip blank lines and comments */
397 for (s = buff; *s == ' ' || *s == '\t'; s++)
398 ;
399 if (!*s || *s == '#')
400 continue;
401
402 line = (struct trace_configline_info *)calloc(1,
403 sizeof(*line));
404 if (!line) {
405 error("Cannot allocate config line\n");
406 return -1;
407 }
408
409 tok = strtok_r(s, " \t", &saveptr);
410 if (!tok) {
411 error("Invalid trace config data on line %d\n",
412 linenum);
413 return -1;
414 }
415 if (0 == strcmp(tok, "include-func")) {
416 line->type = TRACE_LINE_INCLUDE;
417 } else if (0 == strcmp(tok, "exclude-func")) {
418 line->type = TRACE_LINE_EXCLUDE;
419 } else {
420 error("Unknown command in trace config data line %d\n",
421 linenum);
422 return -1;
423 }
424
425 tok = strtok_r(NULL, " \t", &saveptr);
426 if (!tok) {
427 error("Missing pattern in trace config data line %d\n",
428 linenum);
429 return -1;
430 }
431
432 err = regcomp(&line->regex, tok, REG_NOSUB);
433 if (err) {
434 int r = regex_report_error(&line->regex, err,
435 "compile", tok);
436 free(line);
437 return r;
438 }
439
440 /* link this new one to the end of the list */
441 line->name = strdup(tok);
442 line->next = NULL;
443 *tailp = line;
444 tailp = &line->next;
445 }
446
447 if (!feof(fin)) {
448 error("Cannot read from trace config file at position %ld\n",
449 ftell(fin));
450 return -1;
451 }
452 return 0;
453 }
454
read_trace_config_file(const char * fname)455 static int read_trace_config_file(const char *fname)
456 {
457 FILE *fin;
458 int err;
459
460 fin = fopen(fname, "r");
461 if (!fin) {
462 error("Cannot open trace_config file '%s'\n", fname);
463 return -1;
464 }
465 err = read_trace_config(fin);
466 fclose(fin);
467 return err;
468 }
469
out_func(ulong func_offset,int is_caller,const char * suffix)470 static void out_func(ulong func_offset, int is_caller, const char *suffix)
471 {
472 struct func_info *func;
473
474 func = (is_caller ? find_caller_by_offset : find_func_by_offset)
475 (func_offset);
476
477 if (func)
478 printf("%s%s", func->name, suffix);
479 else
480 printf("%lx%s", func_offset, suffix);
481 }
482
483 /*
484 * # tracer: function
485 * #
486 * # TASK-PID CPU# TIMESTAMP FUNCTION
487 * # | | | | |
488 * # bash-4251 [01] 10152.583854: path_put <-path_walk
489 * # bash-4251 [01] 10152.583855: dput <-path_put
490 * # bash-4251 [01] 10152.583855: _atomic_dec_and_lock <-dput
491 */
make_ftrace(void)492 static int make_ftrace(void)
493 {
494 struct trace_call *call;
495 int missing_count = 0, skip_count = 0;
496 int i;
497
498 printf("# tracer: ftrace\n"
499 "#\n"
500 "# TASK-PID CPU# TIMESTAMP FUNCTION\n"
501 "# | | | | |\n");
502 for (i = 0, call = call_list; i < call_count; i++, call++) {
503 struct func_info *func = find_func_by_offset(call->func);
504 ulong time = call->flags & FUNCF_TIMESTAMP_MASK;
505
506 if (TRACE_CALL_TYPE(call) != FUNCF_ENTRY &&
507 TRACE_CALL_TYPE(call) != FUNCF_EXIT)
508 continue;
509 if (!func) {
510 warn("Cannot find function at %lx\n",
511 text_offset + call->func);
512 missing_count++;
513 continue;
514 }
515
516 if (!(func->flags & FUNCF_TRACE)) {
517 debug("Funcion '%s' is excluded from trace\n",
518 func->name);
519 skip_count++;
520 continue;
521 }
522
523 printf("%16s-%-5d [01] %lu.%06lu: ", "uboot", 1,
524 time / 1000000, time % 1000000);
525
526 out_func(call->func, 0, " <- ");
527 out_func(call->caller, 1, "\n");
528 }
529 info("ftrace: %d functions not found, %d excluded\n", missing_count,
530 skip_count);
531
532 return 0;
533 }
534
prof_tool(int argc,char * const argv[],const char * prof_fname,const char * map_fname,const char * trace_config_fname)535 static int prof_tool(int argc, char *const argv[],
536 const char *prof_fname, const char *map_fname,
537 const char *trace_config_fname)
538 {
539 int err = 0;
540
541 if (read_map_file(map_fname))
542 return -1;
543 if (prof_fname && read_profile_file(prof_fname))
544 return -1;
545 if (trace_config_fname && read_trace_config_file(trace_config_fname))
546 return -1;
547
548 check_functions();
549
550 for (; argc; argc--, argv++) {
551 const char *cmd = *argv;
552
553 if (0 == strcmp(cmd, "dump-ftrace"))
554 err = make_ftrace();
555 else
556 warn("Unknown command '%s'\n", cmd);
557 }
558
559 return err;
560 }
561
main(int argc,char * argv[])562 int main(int argc, char *argv[])
563 {
564 const char *map_fname = "System.map";
565 const char *prof_fname = NULL;
566 const char *trace_config_fname = NULL;
567 int opt;
568
569 verbose = 2;
570 while ((opt = getopt(argc, argv, "m:p:t:v:")) != -1) {
571 switch (opt) {
572 case 'm':
573 map_fname = optarg;
574 break;
575
576 case 'p':
577 prof_fname = optarg;
578 break;
579
580 case 't':
581 trace_config_fname = optarg;
582 break;
583
584 case 'v':
585 verbose = atoi(optarg);
586 break;
587
588 default:
589 usage();
590 }
591 }
592 argc -= optind; argv += optind;
593 if (argc < 1)
594 usage();
595
596 debug("Debug enabled\n");
597 return prof_tool(argc, argv, prof_fname, map_fname,
598 trace_config_fname);
599 }
600