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