00b6b17e74a749089c1885b3506d4ad7af44f9d9
[firefly-linux-kernel-4.4.55.git] / tools / perf / util / symbol.c
1 #include <dirent.h>
2 #include <errno.h>
3 #include <stdlib.h>
4 #include <stdio.h>
5 #include <string.h>
6 #include <sys/types.h>
7 #include <sys/stat.h>
8 #include <sys/param.h>
9 #include <fcntl.h>
10 #include <unistd.h>
11 #include <inttypes.h>
12 #include "build-id.h"
13 #include "util.h"
14 #include "debug.h"
15 #include "machine.h"
16 #include "symbol.h"
17 #include "strlist.h"
18 #include "intlist.h"
19 #include "header.h"
20
21 #include <elf.h>
22 #include <limits.h>
23 #include <symbol/kallsyms.h>
24 #include <sys/utsname.h>
25
26 static int dso__load_kernel_sym(struct dso *dso, struct map *map,
27                                 symbol_filter_t filter);
28 static int dso__load_guest_kernel_sym(struct dso *dso, struct map *map,
29                         symbol_filter_t filter);
30 int vmlinux_path__nr_entries;
31 char **vmlinux_path;
32
33 struct symbol_conf symbol_conf = {
34         .use_modules            = true,
35         .try_vmlinux_path       = true,
36         .annotate_src           = true,
37         .demangle               = true,
38         .demangle_kernel        = false,
39         .cumulate_callchain     = true,
40         .show_hist_headers      = true,
41         .symfs                  = "",
42 };
43
44 static enum dso_binary_type binary_type_symtab[] = {
45         DSO_BINARY_TYPE__KALLSYMS,
46         DSO_BINARY_TYPE__GUEST_KALLSYMS,
47         DSO_BINARY_TYPE__JAVA_JIT,
48         DSO_BINARY_TYPE__DEBUGLINK,
49         DSO_BINARY_TYPE__BUILD_ID_CACHE,
50         DSO_BINARY_TYPE__FEDORA_DEBUGINFO,
51         DSO_BINARY_TYPE__UBUNTU_DEBUGINFO,
52         DSO_BINARY_TYPE__BUILDID_DEBUGINFO,
53         DSO_BINARY_TYPE__SYSTEM_PATH_DSO,
54         DSO_BINARY_TYPE__GUEST_KMODULE,
55         DSO_BINARY_TYPE__GUEST_KMODULE_COMP,
56         DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE,
57         DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE_COMP,
58         DSO_BINARY_TYPE__OPENEMBEDDED_DEBUGINFO,
59         DSO_BINARY_TYPE__NOT_FOUND,
60 };
61
62 #define DSO_BINARY_TYPE__SYMTAB_CNT ARRAY_SIZE(binary_type_symtab)
63
64 bool symbol_type__is_a(char symbol_type, enum map_type map_type)
65 {
66         symbol_type = toupper(symbol_type);
67
68         switch (map_type) {
69         case MAP__FUNCTION:
70                 return symbol_type == 'T' || symbol_type == 'W';
71         case MAP__VARIABLE:
72                 return symbol_type == 'D';
73         default:
74                 return false;
75         }
76 }
77
78 static int prefix_underscores_count(const char *str)
79 {
80         const char *tail = str;
81
82         while (*tail == '_')
83                 tail++;
84
85         return tail - str;
86 }
87
88 int __weak arch__choose_best_symbol(struct symbol *syma,
89                                     struct symbol *symb __maybe_unused)
90 {
91         /* Avoid "SyS" kernel syscall aliases */
92         if (strlen(syma->name) >= 3 && !strncmp(syma->name, "SyS", 3))
93                 return SYMBOL_B;
94         if (strlen(syma->name) >= 10 && !strncmp(syma->name, "compat_SyS", 10))
95                 return SYMBOL_B;
96
97         return SYMBOL_A;
98 }
99
100 static int choose_best_symbol(struct symbol *syma, struct symbol *symb)
101 {
102         s64 a;
103         s64 b;
104         size_t na, nb;
105
106         /* Prefer a symbol with non zero length */
107         a = syma->end - syma->start;
108         b = symb->end - symb->start;
109         if ((b == 0) && (a > 0))
110                 return SYMBOL_A;
111         else if ((a == 0) && (b > 0))
112                 return SYMBOL_B;
113
114         /* Prefer a non weak symbol over a weak one */
115         a = syma->binding == STB_WEAK;
116         b = symb->binding == STB_WEAK;
117         if (b && !a)
118                 return SYMBOL_A;
119         if (a && !b)
120                 return SYMBOL_B;
121
122         /* Prefer a global symbol over a non global one */
123         a = syma->binding == STB_GLOBAL;
124         b = symb->binding == STB_GLOBAL;
125         if (a && !b)
126                 return SYMBOL_A;
127         if (b && !a)
128                 return SYMBOL_B;
129
130         /* Prefer a symbol with less underscores */
131         a = prefix_underscores_count(syma->name);
132         b = prefix_underscores_count(symb->name);
133         if (b > a)
134                 return SYMBOL_A;
135         else if (a > b)
136                 return SYMBOL_B;
137
138         /* Choose the symbol with the longest name */
139         na = strlen(syma->name);
140         nb = strlen(symb->name);
141         if (na > nb)
142                 return SYMBOL_A;
143         else if (na < nb)
144                 return SYMBOL_B;
145
146         return arch__choose_best_symbol(syma, symb);
147 }
148
149 void symbols__fixup_duplicate(struct rb_root *symbols)
150 {
151         struct rb_node *nd;
152         struct symbol *curr, *next;
153
154         nd = rb_first(symbols);
155
156         while (nd) {
157                 curr = rb_entry(nd, struct symbol, rb_node);
158 again:
159                 nd = rb_next(&curr->rb_node);
160                 next = rb_entry(nd, struct symbol, rb_node);
161
162                 if (!nd)
163                         break;
164
165                 if (curr->start != next->start)
166                         continue;
167
168                 if (choose_best_symbol(curr, next) == SYMBOL_A) {
169                         rb_erase(&next->rb_node, symbols);
170                         symbol__delete(next);
171                         goto again;
172                 } else {
173                         nd = rb_next(&curr->rb_node);
174                         rb_erase(&curr->rb_node, symbols);
175                         symbol__delete(curr);
176                 }
177         }
178 }
179
180 void symbols__fixup_end(struct rb_root *symbols)
181 {
182         struct rb_node *nd, *prevnd = rb_first(symbols);
183         struct symbol *curr, *prev;
184
185         if (prevnd == NULL)
186                 return;
187
188         curr = rb_entry(prevnd, struct symbol, rb_node);
189
190         for (nd = rb_next(prevnd); nd; nd = rb_next(nd)) {
191                 prev = curr;
192                 curr = rb_entry(nd, struct symbol, rb_node);
193
194                 if (prev->end == prev->start && prev->end != curr->start)
195                         prev->end = curr->start;
196         }
197
198         /* Last entry */
199         if (curr->end == curr->start)
200                 curr->end = roundup(curr->start, 4096);
201 }
202
203 void __map_groups__fixup_end(struct map_groups *mg, enum map_type type)
204 {
205         struct rb_root *maps = &mg->maps[type];
206         struct map *next, *curr;
207
208         curr = maps__first(maps);
209         if (curr == NULL)
210                 return;
211
212         for (next = map__next(curr); next; next = map__next(curr)) {
213                 curr->end = next->start;
214                 curr = next;
215         }
216
217         /*
218          * We still haven't the actual symbols, so guess the
219          * last map final address.
220          */
221         curr->end = ~0ULL;
222 }
223
224 struct symbol *symbol__new(u64 start, u64 len, u8 binding, const char *name)
225 {
226         size_t namelen = strlen(name) + 1;
227         struct symbol *sym = calloc(1, (symbol_conf.priv_size +
228                                         sizeof(*sym) + namelen));
229         if (sym == NULL)
230                 return NULL;
231
232         if (symbol_conf.priv_size)
233                 sym = ((void *)sym) + symbol_conf.priv_size;
234
235         sym->start   = start;
236         sym->end     = len ? start + len : start;
237         sym->binding = binding;
238         sym->namelen = namelen - 1;
239
240         pr_debug4("%s: %s %#" PRIx64 "-%#" PRIx64 "\n",
241                   __func__, name, start, sym->end);
242         memcpy(sym->name, name, namelen);
243
244         return sym;
245 }
246
247 void symbol__delete(struct symbol *sym)
248 {
249         free(((void *)sym) - symbol_conf.priv_size);
250 }
251
252 size_t symbol__fprintf(struct symbol *sym, FILE *fp)
253 {
254         return fprintf(fp, " %" PRIx64 "-%" PRIx64 " %c %s\n",
255                        sym->start, sym->end,
256                        sym->binding == STB_GLOBAL ? 'g' :
257                        sym->binding == STB_LOCAL  ? 'l' : 'w',
258                        sym->name);
259 }
260
261 size_t symbol__fprintf_symname_offs(const struct symbol *sym,
262                                     const struct addr_location *al, FILE *fp)
263 {
264         unsigned long offset;
265         size_t length;
266
267         if (sym && sym->name) {
268                 length = fprintf(fp, "%s", sym->name);
269                 if (al) {
270                         if (al->addr < sym->end)
271                                 offset = al->addr - sym->start;
272                         else
273                                 offset = al->addr - al->map->start - sym->start;
274                         length += fprintf(fp, "+0x%lx", offset);
275                 }
276                 return length;
277         } else
278                 return fprintf(fp, "[unknown]");
279 }
280
281 size_t symbol__fprintf_symname(const struct symbol *sym, FILE *fp)
282 {
283         return symbol__fprintf_symname_offs(sym, NULL, fp);
284 }
285
286 void symbols__delete(struct rb_root *symbols)
287 {
288         struct symbol *pos;
289         struct rb_node *next = rb_first(symbols);
290
291         while (next) {
292                 pos = rb_entry(next, struct symbol, rb_node);
293                 next = rb_next(&pos->rb_node);
294                 rb_erase(&pos->rb_node, symbols);
295                 symbol__delete(pos);
296         }
297 }
298
299 void symbols__insert(struct rb_root *symbols, struct symbol *sym)
300 {
301         struct rb_node **p = &symbols->rb_node;
302         struct rb_node *parent = NULL;
303         const u64 ip = sym->start;
304         struct symbol *s;
305
306         while (*p != NULL) {
307                 parent = *p;
308                 s = rb_entry(parent, struct symbol, rb_node);
309                 if (ip < s->start)
310                         p = &(*p)->rb_left;
311                 else
312                         p = &(*p)->rb_right;
313         }
314         rb_link_node(&sym->rb_node, parent, p);
315         rb_insert_color(&sym->rb_node, symbols);
316 }
317
318 static struct symbol *symbols__find(struct rb_root *symbols, u64 ip)
319 {
320         struct rb_node *n;
321
322         if (symbols == NULL)
323                 return NULL;
324
325         n = symbols->rb_node;
326
327         while (n) {
328                 struct symbol *s = rb_entry(n, struct symbol, rb_node);
329
330                 if (ip < s->start)
331                         n = n->rb_left;
332                 else if (ip >= s->end)
333                         n = n->rb_right;
334                 else
335                         return s;
336         }
337
338         return NULL;
339 }
340
341 static struct symbol *symbols__first(struct rb_root *symbols)
342 {
343         struct rb_node *n = rb_first(symbols);
344
345         if (n)
346                 return rb_entry(n, struct symbol, rb_node);
347
348         return NULL;
349 }
350
351 static struct symbol *symbols__next(struct symbol *sym)
352 {
353         struct rb_node *n = rb_next(&sym->rb_node);
354
355         if (n)
356                 return rb_entry(n, struct symbol, rb_node);
357
358         return NULL;
359 }
360
361 struct symbol_name_rb_node {
362         struct rb_node  rb_node;
363         struct symbol   sym;
364 };
365
366 static void symbols__insert_by_name(struct rb_root *symbols, struct symbol *sym)
367 {
368         struct rb_node **p = &symbols->rb_node;
369         struct rb_node *parent = NULL;
370         struct symbol_name_rb_node *symn, *s;
371
372         symn = container_of(sym, struct symbol_name_rb_node, sym);
373
374         while (*p != NULL) {
375                 parent = *p;
376                 s = rb_entry(parent, struct symbol_name_rb_node, rb_node);
377                 if (strcmp(sym->name, s->sym.name) < 0)
378                         p = &(*p)->rb_left;
379                 else
380                         p = &(*p)->rb_right;
381         }
382         rb_link_node(&symn->rb_node, parent, p);
383         rb_insert_color(&symn->rb_node, symbols);
384 }
385
386 static void symbols__sort_by_name(struct rb_root *symbols,
387                                   struct rb_root *source)
388 {
389         struct rb_node *nd;
390
391         for (nd = rb_first(source); nd; nd = rb_next(nd)) {
392                 struct symbol *pos = rb_entry(nd, struct symbol, rb_node);
393                 symbols__insert_by_name(symbols, pos);
394         }
395 }
396
397 static struct symbol *symbols__find_by_name(struct rb_root *symbols,
398                                             const char *name)
399 {
400         struct rb_node *n;
401         struct symbol_name_rb_node *s;
402
403         if (symbols == NULL)
404                 return NULL;
405
406         n = symbols->rb_node;
407
408         while (n) {
409                 int cmp;
410
411                 s = rb_entry(n, struct symbol_name_rb_node, rb_node);
412                 cmp = arch__compare_symbol_names(name, s->sym.name);
413
414                 if (cmp < 0)
415                         n = n->rb_left;
416                 else if (cmp > 0)
417                         n = n->rb_right;
418                 else
419                         break;
420         }
421
422         if (n == NULL)
423                 return NULL;
424
425         /* return first symbol that has same name (if any) */
426         for (n = rb_prev(n); n; n = rb_prev(n)) {
427                 struct symbol_name_rb_node *tmp;
428
429                 tmp = rb_entry(n, struct symbol_name_rb_node, rb_node);
430                 if (arch__compare_symbol_names(tmp->sym.name, s->sym.name))
431                         break;
432
433                 s = tmp;
434         }
435
436         return &s->sym;
437 }
438
439 struct symbol *dso__find_symbol(struct dso *dso,
440                                 enum map_type type, u64 addr)
441 {
442         return symbols__find(&dso->symbols[type], addr);
443 }
444
445 struct symbol *dso__first_symbol(struct dso *dso, enum map_type type)
446 {
447         return symbols__first(&dso->symbols[type]);
448 }
449
450 struct symbol *dso__next_symbol(struct symbol *sym)
451 {
452         return symbols__next(sym);
453 }
454
455 struct symbol *symbol__next_by_name(struct symbol *sym)
456 {
457         struct symbol_name_rb_node *s = container_of(sym, struct symbol_name_rb_node, sym);
458         struct rb_node *n = rb_next(&s->rb_node);
459
460         return n ? &rb_entry(n, struct symbol_name_rb_node, rb_node)->sym : NULL;
461 }
462
463  /*
464   * Teturns first symbol that matched with @name.
465   */
466 struct symbol *dso__find_symbol_by_name(struct dso *dso, enum map_type type,
467                                         const char *name)
468 {
469         return symbols__find_by_name(&dso->symbol_names[type], name);
470 }
471
472 void dso__sort_by_name(struct dso *dso, enum map_type type)
473 {
474         dso__set_sorted_by_name(dso, type);
475         return symbols__sort_by_name(&dso->symbol_names[type],
476                                      &dso->symbols[type]);
477 }
478
479 size_t dso__fprintf_symbols_by_name(struct dso *dso,
480                                     enum map_type type, FILE *fp)
481 {
482         size_t ret = 0;
483         struct rb_node *nd;
484         struct symbol_name_rb_node *pos;
485
486         for (nd = rb_first(&dso->symbol_names[type]); nd; nd = rb_next(nd)) {
487                 pos = rb_entry(nd, struct symbol_name_rb_node, rb_node);
488                 fprintf(fp, "%s\n", pos->sym.name);
489         }
490
491         return ret;
492 }
493
494 int modules__parse(const char *filename, void *arg,
495                    int (*process_module)(void *arg, const char *name,
496                                          u64 start))
497 {
498         char *line = NULL;
499         size_t n;
500         FILE *file;
501         int err = 0;
502
503         file = fopen(filename, "r");
504         if (file == NULL)
505                 return -1;
506
507         while (1) {
508                 char name[PATH_MAX];
509                 u64 start;
510                 char *sep;
511                 ssize_t line_len;
512
513                 line_len = getline(&line, &n, file);
514                 if (line_len < 0) {
515                         if (feof(file))
516                                 break;
517                         err = -1;
518                         goto out;
519                 }
520
521                 if (!line) {
522                         err = -1;
523                         goto out;
524                 }
525
526                 line[--line_len] = '\0'; /* \n */
527
528                 sep = strrchr(line, 'x');
529                 if (sep == NULL)
530                         continue;
531
532                 hex2u64(sep + 1, &start);
533
534                 sep = strchr(line, ' ');
535                 if (sep == NULL)
536                         continue;
537
538                 *sep = '\0';
539
540                 scnprintf(name, sizeof(name), "[%s]", line);
541
542                 err = process_module(arg, name, start);
543                 if (err)
544                         break;
545         }
546 out:
547         free(line);
548         fclose(file);
549         return err;
550 }
551
552 struct process_kallsyms_args {
553         struct map *map;
554         struct dso *dso;
555 };
556
557 /*
558  * These are symbols in the kernel image, so make sure that
559  * sym is from a kernel DSO.
560  */
561 bool symbol__is_idle(struct symbol *sym)
562 {
563         const char * const idle_symbols[] = {
564                 "cpu_idle",
565                 "cpu_startup_entry",
566                 "intel_idle",
567                 "default_idle",
568                 "native_safe_halt",
569                 "enter_idle",
570                 "exit_idle",
571                 "mwait_idle",
572                 "mwait_idle_with_hints",
573                 "poll_idle",
574                 "ppc64_runlatch_off",
575                 "pseries_dedicated_idle_sleep",
576                 NULL
577         };
578
579         int i;
580
581         if (!sym)
582                 return false;
583
584         for (i = 0; idle_symbols[i]; i++) {
585                 if (!strcmp(idle_symbols[i], sym->name))
586                         return true;
587         }
588
589         return false;
590 }
591
592 static int map__process_kallsym_symbol(void *arg, const char *name,
593                                        char type, u64 start)
594 {
595         struct symbol *sym;
596         struct process_kallsyms_args *a = arg;
597         struct rb_root *root = &a->dso->symbols[a->map->type];
598
599         if (!symbol_type__is_a(type, a->map->type))
600                 return 0;
601
602         /*
603          * module symbols are not sorted so we add all
604          * symbols, setting length to 0, and rely on
605          * symbols__fixup_end() to fix it up.
606          */
607         sym = symbol__new(start, 0, kallsyms2elf_type(type), name);
608         if (sym == NULL)
609                 return -ENOMEM;
610         /*
611          * We will pass the symbols to the filter later, in
612          * map__split_kallsyms, when we have split the maps per module
613          */
614         symbols__insert(root, sym);
615
616         return 0;
617 }
618
619 /*
620  * Loads the function entries in /proc/kallsyms into kernel_map->dso,
621  * so that we can in the next step set the symbol ->end address and then
622  * call kernel_maps__split_kallsyms.
623  */
624 static int dso__load_all_kallsyms(struct dso *dso, const char *filename,
625                                   struct map *map)
626 {
627         struct process_kallsyms_args args = { .map = map, .dso = dso, };
628         return kallsyms__parse(filename, &args, map__process_kallsym_symbol);
629 }
630
631 static int dso__split_kallsyms_for_kcore(struct dso *dso, struct map *map,
632                                          symbol_filter_t filter)
633 {
634         struct map_groups *kmaps = map__kmaps(map);
635         struct map *curr_map;
636         struct symbol *pos;
637         int count = 0, moved = 0;
638         struct rb_root *root = &dso->symbols[map->type];
639         struct rb_node *next = rb_first(root);
640
641         if (!kmaps)
642                 return -1;
643
644         while (next) {
645                 char *module;
646
647                 pos = rb_entry(next, struct symbol, rb_node);
648                 next = rb_next(&pos->rb_node);
649
650                 module = strchr(pos->name, '\t');
651                 if (module)
652                         *module = '\0';
653
654                 curr_map = map_groups__find(kmaps, map->type, pos->start);
655
656                 if (!curr_map || (filter && filter(curr_map, pos))) {
657                         rb_erase(&pos->rb_node, root);
658                         symbol__delete(pos);
659                 } else {
660                         pos->start -= curr_map->start - curr_map->pgoff;
661                         if (pos->end)
662                                 pos->end -= curr_map->start - curr_map->pgoff;
663                         if (curr_map != map) {
664                                 rb_erase(&pos->rb_node, root);
665                                 symbols__insert(
666                                         &curr_map->dso->symbols[curr_map->type],
667                                         pos);
668                                 ++moved;
669                         } else {
670                                 ++count;
671                         }
672                 }
673         }
674
675         /* Symbols have been adjusted */
676         dso->adjust_symbols = 1;
677
678         return count + moved;
679 }
680
681 /*
682  * Split the symbols into maps, making sure there are no overlaps, i.e. the
683  * kernel range is broken in several maps, named [kernel].N, as we don't have
684  * the original ELF section names vmlinux have.
685  */
686 static int dso__split_kallsyms(struct dso *dso, struct map *map, u64 delta,
687                                symbol_filter_t filter)
688 {
689         struct map_groups *kmaps = map__kmaps(map);
690         struct machine *machine;
691         struct map *curr_map = map;
692         struct symbol *pos;
693         int count = 0, moved = 0;
694         struct rb_root *root = &dso->symbols[map->type];
695         struct rb_node *next = rb_first(root);
696         int kernel_range = 0;
697
698         if (!kmaps)
699                 return -1;
700
701         machine = kmaps->machine;
702
703         while (next) {
704                 char *module;
705
706                 pos = rb_entry(next, struct symbol, rb_node);
707                 next = rb_next(&pos->rb_node);
708
709                 module = strchr(pos->name, '\t');
710                 if (module) {
711                         if (!symbol_conf.use_modules)
712                                 goto discard_symbol;
713
714                         *module++ = '\0';
715
716                         if (strcmp(curr_map->dso->short_name, module)) {
717                                 if (curr_map != map &&
718                                     dso->kernel == DSO_TYPE_GUEST_KERNEL &&
719                                     machine__is_default_guest(machine)) {
720                                         /*
721                                          * We assume all symbols of a module are
722                                          * continuous in * kallsyms, so curr_map
723                                          * points to a module and all its
724                                          * symbols are in its kmap. Mark it as
725                                          * loaded.
726                                          */
727                                         dso__set_loaded(curr_map->dso,
728                                                         curr_map->type);
729                                 }
730
731                                 curr_map = map_groups__find_by_name(kmaps,
732                                                         map->type, module);
733                                 if (curr_map == NULL) {
734                                         pr_debug("%s/proc/{kallsyms,modules} "
735                                                  "inconsistency while looking "
736                                                  "for \"%s\" module!\n",
737                                                  machine->root_dir, module);
738                                         curr_map = map;
739                                         goto discard_symbol;
740                                 }
741
742                                 if (curr_map->dso->loaded &&
743                                     !machine__is_default_guest(machine))
744                                         goto discard_symbol;
745                         }
746                         /*
747                          * So that we look just like we get from .ko files,
748                          * i.e. not prelinked, relative to map->start.
749                          */
750                         pos->start = curr_map->map_ip(curr_map, pos->start);
751                         pos->end   = curr_map->map_ip(curr_map, pos->end);
752                 } else if (curr_map != map) {
753                         char dso_name[PATH_MAX];
754                         struct dso *ndso;
755
756                         if (delta) {
757                                 /* Kernel was relocated at boot time */
758                                 pos->start -= delta;
759                                 pos->end -= delta;
760                         }
761
762                         if (count == 0) {
763                                 curr_map = map;
764                                 goto filter_symbol;
765                         }
766
767                         if (dso->kernel == DSO_TYPE_GUEST_KERNEL)
768                                 snprintf(dso_name, sizeof(dso_name),
769                                         "[guest.kernel].%d",
770                                         kernel_range++);
771                         else
772                                 snprintf(dso_name, sizeof(dso_name),
773                                         "[kernel].%d",
774                                         kernel_range++);
775
776                         ndso = dso__new(dso_name);
777                         if (ndso == NULL)
778                                 return -1;
779
780                         ndso->kernel = dso->kernel;
781
782                         curr_map = map__new2(pos->start, ndso, map->type);
783                         if (curr_map == NULL) {
784                                 dso__delete(ndso);
785                                 return -1;
786                         }
787
788                         curr_map->map_ip = curr_map->unmap_ip = identity__map_ip;
789                         map_groups__insert(kmaps, curr_map);
790                         ++kernel_range;
791                 } else if (delta) {
792                         /* Kernel was relocated at boot time */
793                         pos->start -= delta;
794                         pos->end -= delta;
795                 }
796 filter_symbol:
797                 if (filter && filter(curr_map, pos)) {
798 discard_symbol:         rb_erase(&pos->rb_node, root);
799                         symbol__delete(pos);
800                 } else {
801                         if (curr_map != map) {
802                                 rb_erase(&pos->rb_node, root);
803                                 symbols__insert(&curr_map->dso->symbols[curr_map->type], pos);
804                                 ++moved;
805                         } else
806                                 ++count;
807                 }
808         }
809
810         if (curr_map != map &&
811             dso->kernel == DSO_TYPE_GUEST_KERNEL &&
812             machine__is_default_guest(kmaps->machine)) {
813                 dso__set_loaded(curr_map->dso, curr_map->type);
814         }
815
816         return count + moved;
817 }
818
819 bool symbol__restricted_filename(const char *filename,
820                                  const char *restricted_filename)
821 {
822         bool restricted = false;
823
824         if (symbol_conf.kptr_restrict) {
825                 char *r = realpath(filename, NULL);
826
827                 if (r != NULL) {
828                         restricted = strcmp(r, restricted_filename) == 0;
829                         free(r);
830                         return restricted;
831                 }
832         }
833
834         return restricted;
835 }
836
837 struct module_info {
838         struct rb_node rb_node;
839         char *name;
840         u64 start;
841 };
842
843 static void add_module(struct module_info *mi, struct rb_root *modules)
844 {
845         struct rb_node **p = &modules->rb_node;
846         struct rb_node *parent = NULL;
847         struct module_info *m;
848
849         while (*p != NULL) {
850                 parent = *p;
851                 m = rb_entry(parent, struct module_info, rb_node);
852                 if (strcmp(mi->name, m->name) < 0)
853                         p = &(*p)->rb_left;
854                 else
855                         p = &(*p)->rb_right;
856         }
857         rb_link_node(&mi->rb_node, parent, p);
858         rb_insert_color(&mi->rb_node, modules);
859 }
860
861 static void delete_modules(struct rb_root *modules)
862 {
863         struct module_info *mi;
864         struct rb_node *next = rb_first(modules);
865
866         while (next) {
867                 mi = rb_entry(next, struct module_info, rb_node);
868                 next = rb_next(&mi->rb_node);
869                 rb_erase(&mi->rb_node, modules);
870                 zfree(&mi->name);
871                 free(mi);
872         }
873 }
874
875 static struct module_info *find_module(const char *name,
876                                        struct rb_root *modules)
877 {
878         struct rb_node *n = modules->rb_node;
879
880         while (n) {
881                 struct module_info *m;
882                 int cmp;
883
884                 m = rb_entry(n, struct module_info, rb_node);
885                 cmp = strcmp(name, m->name);
886                 if (cmp < 0)
887                         n = n->rb_left;
888                 else if (cmp > 0)
889                         n = n->rb_right;
890                 else
891                         return m;
892         }
893
894         return NULL;
895 }
896
897 static int __read_proc_modules(void *arg, const char *name, u64 start)
898 {
899         struct rb_root *modules = arg;
900         struct module_info *mi;
901
902         mi = zalloc(sizeof(struct module_info));
903         if (!mi)
904                 return -ENOMEM;
905
906         mi->name = strdup(name);
907         mi->start = start;
908
909         if (!mi->name) {
910                 free(mi);
911                 return -ENOMEM;
912         }
913
914         add_module(mi, modules);
915
916         return 0;
917 }
918
919 static int read_proc_modules(const char *filename, struct rb_root *modules)
920 {
921         if (symbol__restricted_filename(filename, "/proc/modules"))
922                 return -1;
923
924         if (modules__parse(filename, modules, __read_proc_modules)) {
925                 delete_modules(modules);
926                 return -1;
927         }
928
929         return 0;
930 }
931
932 int compare_proc_modules(const char *from, const char *to)
933 {
934         struct rb_root from_modules = RB_ROOT;
935         struct rb_root to_modules = RB_ROOT;
936         struct rb_node *from_node, *to_node;
937         struct module_info *from_m, *to_m;
938         int ret = -1;
939
940         if (read_proc_modules(from, &from_modules))
941                 return -1;
942
943         if (read_proc_modules(to, &to_modules))
944                 goto out_delete_from;
945
946         from_node = rb_first(&from_modules);
947         to_node = rb_first(&to_modules);
948         while (from_node) {
949                 if (!to_node)
950                         break;
951
952                 from_m = rb_entry(from_node, struct module_info, rb_node);
953                 to_m = rb_entry(to_node, struct module_info, rb_node);
954
955                 if (from_m->start != to_m->start ||
956                     strcmp(from_m->name, to_m->name))
957                         break;
958
959                 from_node = rb_next(from_node);
960                 to_node = rb_next(to_node);
961         }
962
963         if (!from_node && !to_node)
964                 ret = 0;
965
966         delete_modules(&to_modules);
967 out_delete_from:
968         delete_modules(&from_modules);
969
970         return ret;
971 }
972
973 static int do_validate_kcore_modules(const char *filename, struct map *map,
974                                   struct map_groups *kmaps)
975 {
976         struct rb_root modules = RB_ROOT;
977         struct map *old_map;
978         int err;
979
980         err = read_proc_modules(filename, &modules);
981         if (err)
982                 return err;
983
984         old_map = map_groups__first(kmaps, map->type);
985         while (old_map) {
986                 struct map *next = map_groups__next(old_map);
987                 struct module_info *mi;
988
989                 if (old_map == map || old_map->start == map->start) {
990                         /* The kernel map */
991                         old_map = next;
992                         continue;
993                 }
994
995                 /* Module must be in memory at the same address */
996                 mi = find_module(old_map->dso->short_name, &modules);
997                 if (!mi || mi->start != old_map->start) {
998                         err = -EINVAL;
999                         goto out;
1000                 }
1001
1002                 old_map = next;
1003         }
1004 out:
1005         delete_modules(&modules);
1006         return err;
1007 }
1008
1009 /*
1010  * If kallsyms is referenced by name then we look for filename in the same
1011  * directory.
1012  */
1013 static bool filename_from_kallsyms_filename(char *filename,
1014                                             const char *base_name,
1015                                             const char *kallsyms_filename)
1016 {
1017         char *name;
1018
1019         strcpy(filename, kallsyms_filename);
1020         name = strrchr(filename, '/');
1021         if (!name)
1022                 return false;
1023
1024         name += 1;
1025
1026         if (!strcmp(name, "kallsyms")) {
1027                 strcpy(name, base_name);
1028                 return true;
1029         }
1030
1031         return false;
1032 }
1033
1034 static int validate_kcore_modules(const char *kallsyms_filename,
1035                                   struct map *map)
1036 {
1037         struct map_groups *kmaps = map__kmaps(map);
1038         char modules_filename[PATH_MAX];
1039
1040         if (!kmaps)
1041                 return -EINVAL;
1042
1043         if (!filename_from_kallsyms_filename(modules_filename, "modules",
1044                                              kallsyms_filename))
1045                 return -EINVAL;
1046
1047         if (do_validate_kcore_modules(modules_filename, map, kmaps))
1048                 return -EINVAL;
1049
1050         return 0;
1051 }
1052
1053 static int validate_kcore_addresses(const char *kallsyms_filename,
1054                                     struct map *map)
1055 {
1056         struct kmap *kmap = map__kmap(map);
1057
1058         if (!kmap)
1059                 return -EINVAL;
1060
1061         if (kmap->ref_reloc_sym && kmap->ref_reloc_sym->name) {
1062                 u64 start;
1063
1064                 start = kallsyms__get_function_start(kallsyms_filename,
1065                                                      kmap->ref_reloc_sym->name);
1066                 if (start != kmap->ref_reloc_sym->addr)
1067                         return -EINVAL;
1068         }
1069
1070         return validate_kcore_modules(kallsyms_filename, map);
1071 }
1072
1073 struct kcore_mapfn_data {
1074         struct dso *dso;
1075         enum map_type type;
1076         struct list_head maps;
1077 };
1078
1079 static int kcore_mapfn(u64 start, u64 len, u64 pgoff, void *data)
1080 {
1081         struct kcore_mapfn_data *md = data;
1082         struct map *map;
1083
1084         map = map__new2(start, md->dso, md->type);
1085         if (map == NULL)
1086                 return -ENOMEM;
1087
1088         map->end = map->start + len;
1089         map->pgoff = pgoff;
1090
1091         list_add(&map->node, &md->maps);
1092
1093         return 0;
1094 }
1095
1096 static int dso__load_kcore(struct dso *dso, struct map *map,
1097                            const char *kallsyms_filename)
1098 {
1099         struct map_groups *kmaps = map__kmaps(map);
1100         struct machine *machine;
1101         struct kcore_mapfn_data md;
1102         struct map *old_map, *new_map, *replacement_map = NULL;
1103         bool is_64_bit;
1104         int err, fd;
1105         char kcore_filename[PATH_MAX];
1106         struct symbol *sym;
1107
1108         if (!kmaps)
1109                 return -EINVAL;
1110
1111         machine = kmaps->machine;
1112
1113         /* This function requires that the map is the kernel map */
1114         if (map != machine->vmlinux_maps[map->type])
1115                 return -EINVAL;
1116
1117         if (!filename_from_kallsyms_filename(kcore_filename, "kcore",
1118                                              kallsyms_filename))
1119                 return -EINVAL;
1120
1121         /* Modules and kernel must be present at their original addresses */
1122         if (validate_kcore_addresses(kallsyms_filename, map))
1123                 return -EINVAL;
1124
1125         md.dso = dso;
1126         md.type = map->type;
1127         INIT_LIST_HEAD(&md.maps);
1128
1129         fd = open(kcore_filename, O_RDONLY);
1130         if (fd < 0)
1131                 return -EINVAL;
1132
1133         /* Read new maps into temporary lists */
1134         err = file__read_maps(fd, md.type == MAP__FUNCTION, kcore_mapfn, &md,
1135                               &is_64_bit);
1136         if (err)
1137                 goto out_err;
1138         dso->is_64_bit = is_64_bit;
1139
1140         if (list_empty(&md.maps)) {
1141                 err = -EINVAL;
1142                 goto out_err;
1143         }
1144
1145         /* Remove old maps */
1146         old_map = map_groups__first(kmaps, map->type);
1147         while (old_map) {
1148                 struct map *next = map_groups__next(old_map);
1149
1150                 if (old_map != map)
1151                         map_groups__remove(kmaps, old_map);
1152                 old_map = next;
1153         }
1154
1155         /* Find the kernel map using the first symbol */
1156         sym = dso__first_symbol(dso, map->type);
1157         list_for_each_entry(new_map, &md.maps, node) {
1158                 if (sym && sym->start >= new_map->start &&
1159                     sym->start < new_map->end) {
1160                         replacement_map = new_map;
1161                         break;
1162                 }
1163         }
1164
1165         if (!replacement_map)
1166                 replacement_map = list_entry(md.maps.next, struct map, node);
1167
1168         /* Add new maps */
1169         while (!list_empty(&md.maps)) {
1170                 new_map = list_entry(md.maps.next, struct map, node);
1171                 list_del(&new_map->node);
1172                 if (new_map == replacement_map) {
1173                         map->start      = new_map->start;
1174                         map->end        = new_map->end;
1175                         map->pgoff      = new_map->pgoff;
1176                         map->map_ip     = new_map->map_ip;
1177                         map->unmap_ip   = new_map->unmap_ip;
1178                         map__delete(new_map);
1179                         /* Ensure maps are correctly ordered */
1180                         map_groups__remove(kmaps, map);
1181                         map_groups__insert(kmaps, map);
1182                 } else {
1183                         map_groups__insert(kmaps, new_map);
1184                 }
1185         }
1186
1187         /*
1188          * Set the data type and long name so that kcore can be read via
1189          * dso__data_read_addr().
1190          */
1191         if (dso->kernel == DSO_TYPE_GUEST_KERNEL)
1192                 dso->binary_type = DSO_BINARY_TYPE__GUEST_KCORE;
1193         else
1194                 dso->binary_type = DSO_BINARY_TYPE__KCORE;
1195         dso__set_long_name(dso, strdup(kcore_filename), true);
1196
1197         close(fd);
1198
1199         if (map->type == MAP__FUNCTION)
1200                 pr_debug("Using %s for kernel object code\n", kcore_filename);
1201         else
1202                 pr_debug("Using %s for kernel data\n", kcore_filename);
1203
1204         return 0;
1205
1206 out_err:
1207         while (!list_empty(&md.maps)) {
1208                 map = list_entry(md.maps.next, struct map, node);
1209                 list_del(&map->node);
1210                 map__delete(map);
1211         }
1212         close(fd);
1213         return -EINVAL;
1214 }
1215
1216 /*
1217  * If the kernel is relocated at boot time, kallsyms won't match.  Compute the
1218  * delta based on the relocation reference symbol.
1219  */
1220 static int kallsyms__delta(struct map *map, const char *filename, u64 *delta)
1221 {
1222         struct kmap *kmap = map__kmap(map);
1223         u64 addr;
1224
1225         if (!kmap)
1226                 return -1;
1227
1228         if (!kmap->ref_reloc_sym || !kmap->ref_reloc_sym->name)
1229                 return 0;
1230
1231         addr = kallsyms__get_function_start(filename,
1232                                             kmap->ref_reloc_sym->name);
1233         if (!addr)
1234                 return -1;
1235
1236         *delta = addr - kmap->ref_reloc_sym->addr;
1237         return 0;
1238 }
1239
1240 int dso__load_kallsyms(struct dso *dso, const char *filename,
1241                        struct map *map, symbol_filter_t filter)
1242 {
1243         u64 delta = 0;
1244
1245         if (symbol__restricted_filename(filename, "/proc/kallsyms"))
1246                 return -1;
1247
1248         if (dso__load_all_kallsyms(dso, filename, map) < 0)
1249                 return -1;
1250
1251         if (kallsyms__delta(map, filename, &delta))
1252                 return -1;
1253
1254         symbols__fixup_duplicate(&dso->symbols[map->type]);
1255         symbols__fixup_end(&dso->symbols[map->type]);
1256
1257         if (dso->kernel == DSO_TYPE_GUEST_KERNEL)
1258                 dso->symtab_type = DSO_BINARY_TYPE__GUEST_KALLSYMS;
1259         else
1260                 dso->symtab_type = DSO_BINARY_TYPE__KALLSYMS;
1261
1262         if (!dso__load_kcore(dso, map, filename))
1263                 return dso__split_kallsyms_for_kcore(dso, map, filter);
1264         else
1265                 return dso__split_kallsyms(dso, map, delta, filter);
1266 }
1267
1268 static int dso__load_perf_map(struct dso *dso, struct map *map,
1269                               symbol_filter_t filter)
1270 {
1271         char *line = NULL;
1272         size_t n;
1273         FILE *file;
1274         int nr_syms = 0;
1275
1276         file = fopen(dso->long_name, "r");
1277         if (file == NULL)
1278                 goto out_failure;
1279
1280         while (!feof(file)) {
1281                 u64 start, size;
1282                 struct symbol *sym;
1283                 int line_len, len;
1284
1285                 line_len = getline(&line, &n, file);
1286                 if (line_len < 0)
1287                         break;
1288
1289                 if (!line)
1290                         goto out_failure;
1291
1292                 line[--line_len] = '\0'; /* \n */
1293
1294                 len = hex2u64(line, &start);
1295
1296                 len++;
1297                 if (len + 2 >= line_len)
1298                         continue;
1299
1300                 len += hex2u64(line + len, &size);
1301
1302                 len++;
1303                 if (len + 2 >= line_len)
1304                         continue;
1305
1306                 sym = symbol__new(start, size, STB_GLOBAL, line + len);
1307
1308                 if (sym == NULL)
1309                         goto out_delete_line;
1310
1311                 if (filter && filter(map, sym))
1312                         symbol__delete(sym);
1313                 else {
1314                         symbols__insert(&dso->symbols[map->type], sym);
1315                         nr_syms++;
1316                 }
1317         }
1318
1319         free(line);
1320         fclose(file);
1321
1322         return nr_syms;
1323
1324 out_delete_line:
1325         free(line);
1326 out_failure:
1327         return -1;
1328 }
1329
1330 static bool dso__is_compatible_symtab_type(struct dso *dso, bool kmod,
1331                                            enum dso_binary_type type)
1332 {
1333         switch (type) {
1334         case DSO_BINARY_TYPE__JAVA_JIT:
1335         case DSO_BINARY_TYPE__DEBUGLINK:
1336         case DSO_BINARY_TYPE__SYSTEM_PATH_DSO:
1337         case DSO_BINARY_TYPE__FEDORA_DEBUGINFO:
1338         case DSO_BINARY_TYPE__UBUNTU_DEBUGINFO:
1339         case DSO_BINARY_TYPE__BUILDID_DEBUGINFO:
1340         case DSO_BINARY_TYPE__OPENEMBEDDED_DEBUGINFO:
1341                 return !kmod && dso->kernel == DSO_TYPE_USER;
1342
1343         case DSO_BINARY_TYPE__KALLSYMS:
1344         case DSO_BINARY_TYPE__VMLINUX:
1345         case DSO_BINARY_TYPE__KCORE:
1346                 return dso->kernel == DSO_TYPE_KERNEL;
1347
1348         case DSO_BINARY_TYPE__GUEST_KALLSYMS:
1349         case DSO_BINARY_TYPE__GUEST_VMLINUX:
1350         case DSO_BINARY_TYPE__GUEST_KCORE:
1351                 return dso->kernel == DSO_TYPE_GUEST_KERNEL;
1352
1353         case DSO_BINARY_TYPE__GUEST_KMODULE:
1354         case DSO_BINARY_TYPE__GUEST_KMODULE_COMP:
1355         case DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE:
1356         case DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE_COMP:
1357                 /*
1358                  * kernel modules know their symtab type - it's set when
1359                  * creating a module dso in machine__new_module().
1360                  */
1361                 return kmod && dso->symtab_type == type;
1362
1363         case DSO_BINARY_TYPE__BUILD_ID_CACHE:
1364                 return true;
1365
1366         case DSO_BINARY_TYPE__NOT_FOUND:
1367         default:
1368                 return false;
1369         }
1370 }
1371
1372 int dso__load(struct dso *dso, struct map *map, symbol_filter_t filter)
1373 {
1374         char *name;
1375         int ret = -1;
1376         u_int i;
1377         struct machine *machine;
1378         char *root_dir = (char *) "";
1379         int ss_pos = 0;
1380         struct symsrc ss_[2];
1381         struct symsrc *syms_ss = NULL, *runtime_ss = NULL;
1382         bool kmod;
1383
1384         pthread_mutex_lock(&dso->lock);
1385
1386         /* check again under the dso->lock */
1387         if (dso__loaded(dso, map->type)) {
1388                 ret = 1;
1389                 goto out;
1390         }
1391
1392         if (dso->kernel) {
1393                 if (dso->kernel == DSO_TYPE_KERNEL)
1394                         ret = dso__load_kernel_sym(dso, map, filter);
1395                 else if (dso->kernel == DSO_TYPE_GUEST_KERNEL)
1396                         ret = dso__load_guest_kernel_sym(dso, map, filter);
1397
1398                 goto out;
1399         }
1400
1401         if (map->groups && map->groups->machine)
1402                 machine = map->groups->machine;
1403         else
1404                 machine = NULL;
1405
1406         dso->adjust_symbols = 0;
1407
1408         if (strncmp(dso->name, "/tmp/perf-", 10) == 0) {
1409                 struct stat st;
1410
1411                 if (lstat(dso->name, &st) < 0)
1412                         goto out;
1413
1414                 if (st.st_uid && (st.st_uid != geteuid())) {
1415                         pr_warning("File %s not owned by current user or root, "
1416                                 "ignoring it.\n", dso->name);
1417                         goto out;
1418                 }
1419
1420                 ret = dso__load_perf_map(dso, map, filter);
1421                 dso->symtab_type = ret > 0 ? DSO_BINARY_TYPE__JAVA_JIT :
1422                                              DSO_BINARY_TYPE__NOT_FOUND;
1423                 goto out;
1424         }
1425
1426         if (machine)
1427                 root_dir = machine->root_dir;
1428
1429         name = malloc(PATH_MAX);
1430         if (!name)
1431                 goto out;
1432
1433         kmod = dso->symtab_type == DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE ||
1434                 dso->symtab_type == DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE_COMP ||
1435                 dso->symtab_type == DSO_BINARY_TYPE__GUEST_KMODULE ||
1436                 dso->symtab_type == DSO_BINARY_TYPE__GUEST_KMODULE_COMP;
1437
1438         /*
1439          * Iterate over candidate debug images.
1440          * Keep track of "interesting" ones (those which have a symtab, dynsym,
1441          * and/or opd section) for processing.
1442          */
1443         for (i = 0; i < DSO_BINARY_TYPE__SYMTAB_CNT; i++) {
1444                 struct symsrc *ss = &ss_[ss_pos];
1445                 bool next_slot = false;
1446
1447                 enum dso_binary_type symtab_type = binary_type_symtab[i];
1448
1449                 if (!dso__is_compatible_symtab_type(dso, kmod, symtab_type))
1450                         continue;
1451
1452                 if (dso__read_binary_type_filename(dso, symtab_type,
1453                                                    root_dir, name, PATH_MAX))
1454                         continue;
1455
1456                 /* Name is now the name of the next image to try */
1457                 if (symsrc__init(ss, dso, name, symtab_type) < 0)
1458                         continue;
1459
1460                 if (!syms_ss && symsrc__has_symtab(ss)) {
1461                         syms_ss = ss;
1462                         next_slot = true;
1463                         if (!dso->symsrc_filename)
1464                                 dso->symsrc_filename = strdup(name);
1465                 }
1466
1467                 if (!runtime_ss && symsrc__possibly_runtime(ss)) {
1468                         runtime_ss = ss;
1469                         next_slot = true;
1470                 }
1471
1472                 if (next_slot) {
1473                         ss_pos++;
1474
1475                         if (syms_ss && runtime_ss)
1476                                 break;
1477                 } else {
1478                         symsrc__destroy(ss);
1479                 }
1480
1481         }
1482
1483         if (!runtime_ss && !syms_ss)
1484                 goto out_free;
1485
1486         if (runtime_ss && !syms_ss) {
1487                 syms_ss = runtime_ss;
1488         }
1489
1490         /* We'll have to hope for the best */
1491         if (!runtime_ss && syms_ss)
1492                 runtime_ss = syms_ss;
1493
1494         if (syms_ss)
1495                 ret = dso__load_sym(dso, map, syms_ss, runtime_ss, filter, kmod);
1496         else
1497                 ret = -1;
1498
1499         if (ret > 0) {
1500                 int nr_plt;
1501
1502                 nr_plt = dso__synthesize_plt_symbols(dso, runtime_ss, map, filter);
1503                 if (nr_plt > 0)
1504                         ret += nr_plt;
1505         }
1506
1507         for (; ss_pos > 0; ss_pos--)
1508                 symsrc__destroy(&ss_[ss_pos - 1]);
1509 out_free:
1510         free(name);
1511         if (ret < 0 && strstr(dso->name, " (deleted)") != NULL)
1512                 ret = 0;
1513 out:
1514         dso__set_loaded(dso, map->type);
1515         pthread_mutex_unlock(&dso->lock);
1516
1517         return ret;
1518 }
1519
1520 struct map *map_groups__find_by_name(struct map_groups *mg,
1521                                      enum map_type type, const char *name)
1522 {
1523         struct rb_root *maps = &mg->maps[type];
1524         struct map *map;
1525
1526         for (map = maps__first(maps); map; map = map__next(map)) {
1527                 if (map->dso && strcmp(map->dso->short_name, name) == 0)
1528                         return map;
1529         }
1530
1531         return NULL;
1532 }
1533
1534 int dso__load_vmlinux(struct dso *dso, struct map *map,
1535                       const char *vmlinux, bool vmlinux_allocated,
1536                       symbol_filter_t filter)
1537 {
1538         int err = -1;
1539         struct symsrc ss;
1540         char symfs_vmlinux[PATH_MAX];
1541         enum dso_binary_type symtab_type;
1542
1543         if (vmlinux[0] == '/')
1544                 snprintf(symfs_vmlinux, sizeof(symfs_vmlinux), "%s", vmlinux);
1545         else
1546                 symbol__join_symfs(symfs_vmlinux, vmlinux);
1547
1548         if (dso->kernel == DSO_TYPE_GUEST_KERNEL)
1549                 symtab_type = DSO_BINARY_TYPE__GUEST_VMLINUX;
1550         else
1551                 symtab_type = DSO_BINARY_TYPE__VMLINUX;
1552
1553         if (symsrc__init(&ss, dso, symfs_vmlinux, symtab_type))
1554                 return -1;
1555
1556         err = dso__load_sym(dso, map, &ss, &ss, filter, 0);
1557         symsrc__destroy(&ss);
1558
1559         if (err > 0) {
1560                 if (dso->kernel == DSO_TYPE_GUEST_KERNEL)
1561                         dso->binary_type = DSO_BINARY_TYPE__GUEST_VMLINUX;
1562                 else
1563                         dso->binary_type = DSO_BINARY_TYPE__VMLINUX;
1564                 dso__set_long_name(dso, vmlinux, vmlinux_allocated);
1565                 dso__set_loaded(dso, map->type);
1566                 pr_debug("Using %s for symbols\n", symfs_vmlinux);
1567         }
1568
1569         return err;
1570 }
1571
1572 int dso__load_vmlinux_path(struct dso *dso, struct map *map,
1573                            symbol_filter_t filter)
1574 {
1575         int i, err = 0;
1576         char *filename = NULL;
1577
1578         if (!symbol_conf.ignore_vmlinux_buildid)
1579                 filename = dso__build_id_filename(dso, NULL, 0);
1580         if (filename != NULL) {
1581                 err = dso__load_vmlinux(dso, map, filename, true, filter);
1582                 if (err > 0)
1583                         goto out;
1584                 free(filename);
1585         }
1586
1587         pr_debug("Looking at the vmlinux_path (%d entries long)\n",
1588                  vmlinux_path__nr_entries + 1);
1589
1590         for (i = 0; i < vmlinux_path__nr_entries; ++i) {
1591                 err = dso__load_vmlinux(dso, map, vmlinux_path[i], false, filter);
1592                 if (err > 0)
1593                         break;
1594         }
1595 out:
1596         return err;
1597 }
1598
1599 static int find_matching_kcore(struct map *map, char *dir, size_t dir_sz)
1600 {
1601         char kallsyms_filename[PATH_MAX];
1602         struct dirent *dent;
1603         int ret = -1;
1604         DIR *d;
1605
1606         d = opendir(dir);
1607         if (!d)
1608                 return -1;
1609
1610         while (1) {
1611                 dent = readdir(d);
1612                 if (!dent)
1613                         break;
1614                 if (dent->d_type != DT_DIR)
1615                         continue;
1616                 scnprintf(kallsyms_filename, sizeof(kallsyms_filename),
1617                           "%s/%s/kallsyms", dir, dent->d_name);
1618                 if (!validate_kcore_addresses(kallsyms_filename, map)) {
1619                         strlcpy(dir, kallsyms_filename, dir_sz);
1620                         ret = 0;
1621                         break;
1622                 }
1623         }
1624
1625         closedir(d);
1626
1627         return ret;
1628 }
1629
1630 static char *dso__find_kallsyms(struct dso *dso, struct map *map)
1631 {
1632         u8 host_build_id[BUILD_ID_SIZE];
1633         char sbuild_id[BUILD_ID_SIZE * 2 + 1];
1634         bool is_host = false;
1635         char path[PATH_MAX];
1636
1637         if (!dso->has_build_id) {
1638                 /*
1639                  * Last resort, if we don't have a build-id and couldn't find
1640                  * any vmlinux file, try the running kernel kallsyms table.
1641                  */
1642                 goto proc_kallsyms;
1643         }
1644
1645         if (sysfs__read_build_id("/sys/kernel/notes", host_build_id,
1646                                  sizeof(host_build_id)) == 0)
1647                 is_host = dso__build_id_equal(dso, host_build_id);
1648
1649         build_id__sprintf(dso->build_id, sizeof(dso->build_id), sbuild_id);
1650
1651         scnprintf(path, sizeof(path), "%s/[kernel.kcore]/%s", buildid_dir,
1652                   sbuild_id);
1653
1654         /* Use /proc/kallsyms if possible */
1655         if (is_host) {
1656                 DIR *d;
1657                 int fd;
1658
1659                 /* If no cached kcore go with /proc/kallsyms */
1660                 d = opendir(path);
1661                 if (!d)
1662                         goto proc_kallsyms;
1663                 closedir(d);
1664
1665                 /*
1666                  * Do not check the build-id cache, until we know we cannot use
1667                  * /proc/kcore.
1668                  */
1669                 fd = open("/proc/kcore", O_RDONLY);
1670                 if (fd != -1) {
1671                         close(fd);
1672                         /* If module maps match go with /proc/kallsyms */
1673                         if (!validate_kcore_addresses("/proc/kallsyms", map))
1674                                 goto proc_kallsyms;
1675                 }
1676
1677                 /* Find kallsyms in build-id cache with kcore */
1678                 if (!find_matching_kcore(map, path, sizeof(path)))
1679                         return strdup(path);
1680
1681                 goto proc_kallsyms;
1682         }
1683
1684         /* Find kallsyms in build-id cache with kcore */
1685         if (!find_matching_kcore(map, path, sizeof(path)))
1686                 return strdup(path);
1687
1688         scnprintf(path, sizeof(path), "%s/[kernel.kallsyms]/%s",
1689                   buildid_dir, sbuild_id);
1690
1691         if (access(path, F_OK)) {
1692                 pr_err("No kallsyms or vmlinux with build-id %s was found\n",
1693                        sbuild_id);
1694                 return NULL;
1695         }
1696
1697         return strdup(path);
1698
1699 proc_kallsyms:
1700         return strdup("/proc/kallsyms");
1701 }
1702
1703 static int dso__load_kernel_sym(struct dso *dso, struct map *map,
1704                                 symbol_filter_t filter)
1705 {
1706         int err;
1707         const char *kallsyms_filename = NULL;
1708         char *kallsyms_allocated_filename = NULL;
1709         /*
1710          * Step 1: if the user specified a kallsyms or vmlinux filename, use
1711          * it and only it, reporting errors to the user if it cannot be used.
1712          *
1713          * For instance, try to analyse an ARM perf.data file _without_ a
1714          * build-id, or if the user specifies the wrong path to the right
1715          * vmlinux file, obviously we can't fallback to another vmlinux (a
1716          * x86_86 one, on the machine where analysis is being performed, say),
1717          * or worse, /proc/kallsyms.
1718          *
1719          * If the specified file _has_ a build-id and there is a build-id
1720          * section in the perf.data file, we will still do the expected
1721          * validation in dso__load_vmlinux and will bail out if they don't
1722          * match.
1723          */
1724         if (symbol_conf.kallsyms_name != NULL) {
1725                 kallsyms_filename = symbol_conf.kallsyms_name;
1726                 goto do_kallsyms;
1727         }
1728
1729         if (!symbol_conf.ignore_vmlinux && symbol_conf.vmlinux_name != NULL) {
1730                 return dso__load_vmlinux(dso, map, symbol_conf.vmlinux_name,
1731                                          false, filter);
1732         }
1733
1734         if (!symbol_conf.ignore_vmlinux && vmlinux_path != NULL) {
1735                 err = dso__load_vmlinux_path(dso, map, filter);
1736                 if (err > 0)
1737                         return err;
1738         }
1739
1740         /* do not try local files if a symfs was given */
1741         if (symbol_conf.symfs[0] != 0)
1742                 return -1;
1743
1744         kallsyms_allocated_filename = dso__find_kallsyms(dso, map);
1745         if (!kallsyms_allocated_filename)
1746                 return -1;
1747
1748         kallsyms_filename = kallsyms_allocated_filename;
1749
1750 do_kallsyms:
1751         err = dso__load_kallsyms(dso, kallsyms_filename, map, filter);
1752         if (err > 0)
1753                 pr_debug("Using %s for symbols\n", kallsyms_filename);
1754         free(kallsyms_allocated_filename);
1755
1756         if (err > 0 && !dso__is_kcore(dso)) {
1757                 dso->binary_type = DSO_BINARY_TYPE__KALLSYMS;
1758                 dso__set_long_name(dso, "[kernel.kallsyms]", false);
1759                 map__fixup_start(map);
1760                 map__fixup_end(map);
1761         }
1762
1763         return err;
1764 }
1765
1766 static int dso__load_guest_kernel_sym(struct dso *dso, struct map *map,
1767                                       symbol_filter_t filter)
1768 {
1769         int err;
1770         const char *kallsyms_filename = NULL;
1771         struct machine *machine;
1772         char path[PATH_MAX];
1773
1774         if (!map->groups) {
1775                 pr_debug("Guest kernel map hasn't the point to groups\n");
1776                 return -1;
1777         }
1778         machine = map->groups->machine;
1779
1780         if (machine__is_default_guest(machine)) {
1781                 /*
1782                  * if the user specified a vmlinux filename, use it and only
1783                  * it, reporting errors to the user if it cannot be used.
1784                  * Or use file guest_kallsyms inputted by user on commandline
1785                  */
1786                 if (symbol_conf.default_guest_vmlinux_name != NULL) {
1787                         err = dso__load_vmlinux(dso, map,
1788                                                 symbol_conf.default_guest_vmlinux_name,
1789                                                 false, filter);
1790                         return err;
1791                 }
1792
1793                 kallsyms_filename = symbol_conf.default_guest_kallsyms;
1794                 if (!kallsyms_filename)
1795                         return -1;
1796         } else {
1797                 sprintf(path, "%s/proc/kallsyms", machine->root_dir);
1798                 kallsyms_filename = path;
1799         }
1800
1801         err = dso__load_kallsyms(dso, kallsyms_filename, map, filter);
1802         if (err > 0)
1803                 pr_debug("Using %s for symbols\n", kallsyms_filename);
1804         if (err > 0 && !dso__is_kcore(dso)) {
1805                 dso->binary_type = DSO_BINARY_TYPE__GUEST_KALLSYMS;
1806                 machine__mmap_name(machine, path, sizeof(path));
1807                 dso__set_long_name(dso, strdup(path), true);
1808                 map__fixup_start(map);
1809                 map__fixup_end(map);
1810         }
1811
1812         return err;
1813 }
1814
1815 static void vmlinux_path__exit(void)
1816 {
1817         while (--vmlinux_path__nr_entries >= 0)
1818                 zfree(&vmlinux_path[vmlinux_path__nr_entries]);
1819         vmlinux_path__nr_entries = 0;
1820
1821         zfree(&vmlinux_path);
1822 }
1823
1824 static int vmlinux_path__init(struct perf_session_env *env)
1825 {
1826         struct utsname uts;
1827         char bf[PATH_MAX];
1828         char *kernel_version;
1829
1830         vmlinux_path = malloc(sizeof(char *) * 6);
1831         if (vmlinux_path == NULL)
1832                 return -1;
1833
1834         vmlinux_path[vmlinux_path__nr_entries] = strdup("vmlinux");
1835         if (vmlinux_path[vmlinux_path__nr_entries] == NULL)
1836                 goto out_fail;
1837         ++vmlinux_path__nr_entries;
1838         vmlinux_path[vmlinux_path__nr_entries] = strdup("/boot/vmlinux");
1839         if (vmlinux_path[vmlinux_path__nr_entries] == NULL)
1840                 goto out_fail;
1841         ++vmlinux_path__nr_entries;
1842
1843         /* only try kernel version if no symfs was given */
1844         if (symbol_conf.symfs[0] != 0)
1845                 return 0;
1846
1847         if (env) {
1848                 kernel_version = env->os_release;
1849         } else {
1850                 if (uname(&uts) < 0)
1851                         goto out_fail;
1852
1853                 kernel_version = uts.release;
1854         }
1855
1856         snprintf(bf, sizeof(bf), "/boot/vmlinux-%s", kernel_version);
1857         vmlinux_path[vmlinux_path__nr_entries] = strdup(bf);
1858         if (vmlinux_path[vmlinux_path__nr_entries] == NULL)
1859                 goto out_fail;
1860         ++vmlinux_path__nr_entries;
1861         snprintf(bf, sizeof(bf), "/usr/lib/debug/boot/vmlinux-%s",
1862                  kernel_version);
1863         vmlinux_path[vmlinux_path__nr_entries] = strdup(bf);
1864         if (vmlinux_path[vmlinux_path__nr_entries] == NULL)
1865                 goto out_fail;
1866         ++vmlinux_path__nr_entries;
1867         snprintf(bf, sizeof(bf), "/lib/modules/%s/build/vmlinux", kernel_version);
1868         vmlinux_path[vmlinux_path__nr_entries] = strdup(bf);
1869         if (vmlinux_path[vmlinux_path__nr_entries] == NULL)
1870                 goto out_fail;
1871         ++vmlinux_path__nr_entries;
1872         snprintf(bf, sizeof(bf), "/usr/lib/debug/lib/modules/%s/vmlinux",
1873                  kernel_version);
1874         vmlinux_path[vmlinux_path__nr_entries] = strdup(bf);
1875         if (vmlinux_path[vmlinux_path__nr_entries] == NULL)
1876                 goto out_fail;
1877         ++vmlinux_path__nr_entries;
1878
1879         return 0;
1880
1881 out_fail:
1882         vmlinux_path__exit();
1883         return -1;
1884 }
1885
1886 int setup_list(struct strlist **list, const char *list_str,
1887                       const char *list_name)
1888 {
1889         if (list_str == NULL)
1890                 return 0;
1891
1892         *list = strlist__new(true, list_str);
1893         if (!*list) {
1894                 pr_err("problems parsing %s list\n", list_name);
1895                 return -1;
1896         }
1897         return 0;
1898 }
1899
1900 int setup_intlist(struct intlist **list, const char *list_str,
1901                   const char *list_name)
1902 {
1903         if (list_str == NULL)
1904                 return 0;
1905
1906         *list = intlist__new(list_str);
1907         if (!*list) {
1908                 pr_err("problems parsing %s list\n", list_name);
1909                 return -1;
1910         }
1911         return 0;
1912 }
1913
1914 static bool symbol__read_kptr_restrict(void)
1915 {
1916         bool value = false;
1917
1918         if (geteuid() != 0) {
1919                 FILE *fp = fopen("/proc/sys/kernel/kptr_restrict", "r");
1920                 if (fp != NULL) {
1921                         char line[8];
1922
1923                         if (fgets(line, sizeof(line), fp) != NULL)
1924                                 value = atoi(line) != 0;
1925
1926                         fclose(fp);
1927                 }
1928         }
1929
1930         return value;
1931 }
1932
1933 int symbol__init(struct perf_session_env *env)
1934 {
1935         const char *symfs;
1936
1937         if (symbol_conf.initialized)
1938                 return 0;
1939
1940         symbol_conf.priv_size = PERF_ALIGN(symbol_conf.priv_size, sizeof(u64));
1941
1942         symbol__elf_init();
1943
1944         if (symbol_conf.sort_by_name)
1945                 symbol_conf.priv_size += (sizeof(struct symbol_name_rb_node) -
1946                                           sizeof(struct symbol));
1947
1948         if (symbol_conf.try_vmlinux_path && vmlinux_path__init(env) < 0)
1949                 return -1;
1950
1951         if (symbol_conf.field_sep && *symbol_conf.field_sep == '.') {
1952                 pr_err("'.' is the only non valid --field-separator argument\n");
1953                 return -1;
1954         }
1955
1956         if (setup_list(&symbol_conf.dso_list,
1957                        symbol_conf.dso_list_str, "dso") < 0)
1958                 return -1;
1959
1960         if (setup_list(&symbol_conf.comm_list,
1961                        symbol_conf.comm_list_str, "comm") < 0)
1962                 goto out_free_dso_list;
1963
1964         if (setup_intlist(&symbol_conf.pid_list,
1965                        symbol_conf.pid_list_str, "pid") < 0)
1966                 goto out_free_comm_list;
1967
1968         if (setup_intlist(&symbol_conf.tid_list,
1969                        symbol_conf.tid_list_str, "tid") < 0)
1970                 goto out_free_pid_list;
1971
1972         if (setup_list(&symbol_conf.sym_list,
1973                        symbol_conf.sym_list_str, "symbol") < 0)
1974                 goto out_free_tid_list;
1975
1976         /*
1977          * A path to symbols of "/" is identical to ""
1978          * reset here for simplicity.
1979          */
1980         symfs = realpath(symbol_conf.symfs, NULL);
1981         if (symfs == NULL)
1982                 symfs = symbol_conf.symfs;
1983         if (strcmp(symfs, "/") == 0)
1984                 symbol_conf.symfs = "";
1985         if (symfs != symbol_conf.symfs)
1986                 free((void *)symfs);
1987
1988         symbol_conf.kptr_restrict = symbol__read_kptr_restrict();
1989
1990         symbol_conf.initialized = true;
1991         return 0;
1992
1993 out_free_tid_list:
1994         intlist__delete(symbol_conf.tid_list);
1995 out_free_pid_list:
1996         intlist__delete(symbol_conf.pid_list);
1997 out_free_comm_list:
1998         strlist__delete(symbol_conf.comm_list);
1999 out_free_dso_list:
2000         strlist__delete(symbol_conf.dso_list);
2001         return -1;
2002 }
2003
2004 void symbol__exit(void)
2005 {
2006         if (!symbol_conf.initialized)
2007                 return;
2008         strlist__delete(symbol_conf.sym_list);
2009         strlist__delete(symbol_conf.dso_list);
2010         strlist__delete(symbol_conf.comm_list);
2011         intlist__delete(symbol_conf.tid_list);
2012         intlist__delete(symbol_conf.pid_list);
2013         vmlinux_path__exit();
2014         symbol_conf.sym_list = symbol_conf.dso_list = symbol_conf.comm_list = NULL;
2015         symbol_conf.initialized = false;
2016 }