Merge branch 'for-linus' of git://git.kernel.org/pub/scm/linux/kernel/git/mason/linux...
[pandora-kernel.git] / tools / perf / util / machine.c
1 #include "callchain.h"
2 #include "debug.h"
3 #include "event.h"
4 #include "evsel.h"
5 #include "hist.h"
6 #include "machine.h"
7 #include "map.h"
8 #include "sort.h"
9 #include "strlist.h"
10 #include "thread.h"
11 #include <stdbool.h>
12 #include <symbol/kallsyms.h>
13 #include "unwind.h"
14
15 int machine__init(struct machine *machine, const char *root_dir, pid_t pid)
16 {
17         map_groups__init(&machine->kmaps);
18         RB_CLEAR_NODE(&machine->rb_node);
19         INIT_LIST_HEAD(&machine->user_dsos);
20         INIT_LIST_HEAD(&machine->kernel_dsos);
21
22         machine->threads = RB_ROOT;
23         INIT_LIST_HEAD(&machine->dead_threads);
24         machine->last_match = NULL;
25
26         machine->kmaps.machine = machine;
27         machine->pid = pid;
28
29         machine->symbol_filter = NULL;
30         machine->id_hdr_size = 0;
31
32         machine->root_dir = strdup(root_dir);
33         if (machine->root_dir == NULL)
34                 return -ENOMEM;
35
36         if (pid != HOST_KERNEL_ID) {
37                 struct thread *thread = machine__findnew_thread(machine, 0,
38                                                                 pid);
39                 char comm[64];
40
41                 if (thread == NULL)
42                         return -ENOMEM;
43
44                 snprintf(comm, sizeof(comm), "[guest/%d]", pid);
45                 thread__set_comm(thread, comm, 0);
46         }
47
48         return 0;
49 }
50
51 struct machine *machine__new_host(void)
52 {
53         struct machine *machine = malloc(sizeof(*machine));
54
55         if (machine != NULL) {
56                 machine__init(machine, "", HOST_KERNEL_ID);
57
58                 if (machine__create_kernel_maps(machine) < 0)
59                         goto out_delete;
60         }
61
62         return machine;
63 out_delete:
64         free(machine);
65         return NULL;
66 }
67
68 static void dsos__delete(struct list_head *dsos)
69 {
70         struct dso *pos, *n;
71
72         list_for_each_entry_safe(pos, n, dsos, node) {
73                 list_del(&pos->node);
74                 dso__delete(pos);
75         }
76 }
77
78 void machine__delete_dead_threads(struct machine *machine)
79 {
80         struct thread *n, *t;
81
82         list_for_each_entry_safe(t, n, &machine->dead_threads, node) {
83                 list_del(&t->node);
84                 thread__delete(t);
85         }
86 }
87
88 void machine__delete_threads(struct machine *machine)
89 {
90         struct rb_node *nd = rb_first(&machine->threads);
91
92         while (nd) {
93                 struct thread *t = rb_entry(nd, struct thread, rb_node);
94
95                 rb_erase(&t->rb_node, &machine->threads);
96                 nd = rb_next(nd);
97                 thread__delete(t);
98         }
99 }
100
101 void machine__exit(struct machine *machine)
102 {
103         map_groups__exit(&machine->kmaps);
104         dsos__delete(&machine->user_dsos);
105         dsos__delete(&machine->kernel_dsos);
106         zfree(&machine->root_dir);
107 }
108
109 void machine__delete(struct machine *machine)
110 {
111         machine__exit(machine);
112         free(machine);
113 }
114
115 void machines__init(struct machines *machines)
116 {
117         machine__init(&machines->host, "", HOST_KERNEL_ID);
118         machines->guests = RB_ROOT;
119         machines->symbol_filter = NULL;
120 }
121
122 void machines__exit(struct machines *machines)
123 {
124         machine__exit(&machines->host);
125         /* XXX exit guest */
126 }
127
128 struct machine *machines__add(struct machines *machines, pid_t pid,
129                               const char *root_dir)
130 {
131         struct rb_node **p = &machines->guests.rb_node;
132         struct rb_node *parent = NULL;
133         struct machine *pos, *machine = malloc(sizeof(*machine));
134
135         if (machine == NULL)
136                 return NULL;
137
138         if (machine__init(machine, root_dir, pid) != 0) {
139                 free(machine);
140                 return NULL;
141         }
142
143         machine->symbol_filter = machines->symbol_filter;
144
145         while (*p != NULL) {
146                 parent = *p;
147                 pos = rb_entry(parent, struct machine, rb_node);
148                 if (pid < pos->pid)
149                         p = &(*p)->rb_left;
150                 else
151                         p = &(*p)->rb_right;
152         }
153
154         rb_link_node(&machine->rb_node, parent, p);
155         rb_insert_color(&machine->rb_node, &machines->guests);
156
157         return machine;
158 }
159
160 void machines__set_symbol_filter(struct machines *machines,
161                                  symbol_filter_t symbol_filter)
162 {
163         struct rb_node *nd;
164
165         machines->symbol_filter = symbol_filter;
166         machines->host.symbol_filter = symbol_filter;
167
168         for (nd = rb_first(&machines->guests); nd; nd = rb_next(nd)) {
169                 struct machine *machine = rb_entry(nd, struct machine, rb_node);
170
171                 machine->symbol_filter = symbol_filter;
172         }
173 }
174
175 struct machine *machines__find(struct machines *machines, pid_t pid)
176 {
177         struct rb_node **p = &machines->guests.rb_node;
178         struct rb_node *parent = NULL;
179         struct machine *machine;
180         struct machine *default_machine = NULL;
181
182         if (pid == HOST_KERNEL_ID)
183                 return &machines->host;
184
185         while (*p != NULL) {
186                 parent = *p;
187                 machine = rb_entry(parent, struct machine, rb_node);
188                 if (pid < machine->pid)
189                         p = &(*p)->rb_left;
190                 else if (pid > machine->pid)
191                         p = &(*p)->rb_right;
192                 else
193                         return machine;
194                 if (!machine->pid)
195                         default_machine = machine;
196         }
197
198         return default_machine;
199 }
200
201 struct machine *machines__findnew(struct machines *machines, pid_t pid)
202 {
203         char path[PATH_MAX];
204         const char *root_dir = "";
205         struct machine *machine = machines__find(machines, pid);
206
207         if (machine && (machine->pid == pid))
208                 goto out;
209
210         if ((pid != HOST_KERNEL_ID) &&
211             (pid != DEFAULT_GUEST_KERNEL_ID) &&
212             (symbol_conf.guestmount)) {
213                 sprintf(path, "%s/%d", symbol_conf.guestmount, pid);
214                 if (access(path, R_OK)) {
215                         static struct strlist *seen;
216
217                         if (!seen)
218                                 seen = strlist__new(true, NULL);
219
220                         if (!strlist__has_entry(seen, path)) {
221                                 pr_err("Can't access file %s\n", path);
222                                 strlist__add(seen, path);
223                         }
224                         machine = NULL;
225                         goto out;
226                 }
227                 root_dir = path;
228         }
229
230         machine = machines__add(machines, pid, root_dir);
231 out:
232         return machine;
233 }
234
235 void machines__process_guests(struct machines *machines,
236                               machine__process_t process, void *data)
237 {
238         struct rb_node *nd;
239
240         for (nd = rb_first(&machines->guests); nd; nd = rb_next(nd)) {
241                 struct machine *pos = rb_entry(nd, struct machine, rb_node);
242                 process(pos, data);
243         }
244 }
245
246 char *machine__mmap_name(struct machine *machine, char *bf, size_t size)
247 {
248         if (machine__is_host(machine))
249                 snprintf(bf, size, "[%s]", "kernel.kallsyms");
250         else if (machine__is_default_guest(machine))
251                 snprintf(bf, size, "[%s]", "guest.kernel.kallsyms");
252         else {
253                 snprintf(bf, size, "[%s.%d]", "guest.kernel.kallsyms",
254                          machine->pid);
255         }
256
257         return bf;
258 }
259
260 void machines__set_id_hdr_size(struct machines *machines, u16 id_hdr_size)
261 {
262         struct rb_node *node;
263         struct machine *machine;
264
265         machines->host.id_hdr_size = id_hdr_size;
266
267         for (node = rb_first(&machines->guests); node; node = rb_next(node)) {
268                 machine = rb_entry(node, struct machine, rb_node);
269                 machine->id_hdr_size = id_hdr_size;
270         }
271
272         return;
273 }
274
275 static struct thread *__machine__findnew_thread(struct machine *machine,
276                                                 pid_t pid, pid_t tid,
277                                                 bool create)
278 {
279         struct rb_node **p = &machine->threads.rb_node;
280         struct rb_node *parent = NULL;
281         struct thread *th;
282
283         /*
284          * Front-end cache - TID lookups come in blocks,
285          * so most of the time we dont have to look up
286          * the full rbtree:
287          */
288         if (machine->last_match && machine->last_match->tid == tid) {
289                 if (pid && pid != machine->last_match->pid_)
290                         machine->last_match->pid_ = pid;
291                 return machine->last_match;
292         }
293
294         while (*p != NULL) {
295                 parent = *p;
296                 th = rb_entry(parent, struct thread, rb_node);
297
298                 if (th->tid == tid) {
299                         machine->last_match = th;
300                         if (pid && pid != th->pid_)
301                                 th->pid_ = pid;
302                         return th;
303                 }
304
305                 if (tid < th->tid)
306                         p = &(*p)->rb_left;
307                 else
308                         p = &(*p)->rb_right;
309         }
310
311         if (!create)
312                 return NULL;
313
314         th = thread__new(pid, tid);
315         if (th != NULL) {
316                 rb_link_node(&th->rb_node, parent, p);
317                 rb_insert_color(&th->rb_node, &machine->threads);
318                 machine->last_match = th;
319         }
320
321         return th;
322 }
323
324 struct thread *machine__findnew_thread(struct machine *machine, pid_t pid,
325                                        pid_t tid)
326 {
327         return __machine__findnew_thread(machine, pid, tid, true);
328 }
329
330 struct thread *machine__find_thread(struct machine *machine, pid_t pid,
331                                     pid_t tid)
332 {
333         return __machine__findnew_thread(machine, pid, tid, false);
334 }
335
336 int machine__process_comm_event(struct machine *machine, union perf_event *event,
337                                 struct perf_sample *sample)
338 {
339         struct thread *thread = machine__findnew_thread(machine,
340                                                         event->comm.pid,
341                                                         event->comm.tid);
342
343         if (dump_trace)
344                 perf_event__fprintf_comm(event, stdout);
345
346         if (thread == NULL || thread__set_comm(thread, event->comm.comm, sample->time)) {
347                 dump_printf("problem processing PERF_RECORD_COMM, skipping event.\n");
348                 return -1;
349         }
350
351         return 0;
352 }
353
354 int machine__process_lost_event(struct machine *machine __maybe_unused,
355                                 union perf_event *event, struct perf_sample *sample __maybe_unused)
356 {
357         dump_printf(": id:%" PRIu64 ": lost:%" PRIu64 "\n",
358                     event->lost.id, event->lost.lost);
359         return 0;
360 }
361
362 struct map *machine__new_module(struct machine *machine, u64 start,
363                                 const char *filename)
364 {
365         struct map *map;
366         struct dso *dso = __dsos__findnew(&machine->kernel_dsos, filename);
367
368         if (dso == NULL)
369                 return NULL;
370
371         map = map__new2(start, dso, MAP__FUNCTION);
372         if (map == NULL)
373                 return NULL;
374
375         if (machine__is_host(machine))
376                 dso->symtab_type = DSO_BINARY_TYPE__SYSTEM_PATH_KMODULE;
377         else
378                 dso->symtab_type = DSO_BINARY_TYPE__GUEST_KMODULE;
379         map_groups__insert(&machine->kmaps, map);
380         return map;
381 }
382
383 size_t machines__fprintf_dsos(struct machines *machines, FILE *fp)
384 {
385         struct rb_node *nd;
386         size_t ret = __dsos__fprintf(&machines->host.kernel_dsos, fp) +
387                      __dsos__fprintf(&machines->host.user_dsos, fp);
388
389         for (nd = rb_first(&machines->guests); nd; nd = rb_next(nd)) {
390                 struct machine *pos = rb_entry(nd, struct machine, rb_node);
391                 ret += __dsos__fprintf(&pos->kernel_dsos, fp);
392                 ret += __dsos__fprintf(&pos->user_dsos, fp);
393         }
394
395         return ret;
396 }
397
398 size_t machine__fprintf_dsos_buildid(struct machine *machine, FILE *fp,
399                                      bool (skip)(struct dso *dso, int parm), int parm)
400 {
401         return __dsos__fprintf_buildid(&machine->kernel_dsos, fp, skip, parm) +
402                __dsos__fprintf_buildid(&machine->user_dsos, fp, skip, parm);
403 }
404
405 size_t machines__fprintf_dsos_buildid(struct machines *machines, FILE *fp,
406                                      bool (skip)(struct dso *dso, int parm), int parm)
407 {
408         struct rb_node *nd;
409         size_t ret = machine__fprintf_dsos_buildid(&machines->host, fp, skip, parm);
410
411         for (nd = rb_first(&machines->guests); nd; nd = rb_next(nd)) {
412                 struct machine *pos = rb_entry(nd, struct machine, rb_node);
413                 ret += machine__fprintf_dsos_buildid(pos, fp, skip, parm);
414         }
415         return ret;
416 }
417
418 size_t machine__fprintf_vmlinux_path(struct machine *machine, FILE *fp)
419 {
420         int i;
421         size_t printed = 0;
422         struct dso *kdso = machine->vmlinux_maps[MAP__FUNCTION]->dso;
423
424         if (kdso->has_build_id) {
425                 char filename[PATH_MAX];
426                 if (dso__build_id_filename(kdso, filename, sizeof(filename)))
427                         printed += fprintf(fp, "[0] %s\n", filename);
428         }
429
430         for (i = 0; i < vmlinux_path__nr_entries; ++i)
431                 printed += fprintf(fp, "[%d] %s\n",
432                                    i + kdso->has_build_id, vmlinux_path[i]);
433
434         return printed;
435 }
436
437 size_t machine__fprintf(struct machine *machine, FILE *fp)
438 {
439         size_t ret = 0;
440         struct rb_node *nd;
441
442         for (nd = rb_first(&machine->threads); nd; nd = rb_next(nd)) {
443                 struct thread *pos = rb_entry(nd, struct thread, rb_node);
444
445                 ret += thread__fprintf(pos, fp);
446         }
447
448         return ret;
449 }
450
451 static struct dso *machine__get_kernel(struct machine *machine)
452 {
453         const char *vmlinux_name = NULL;
454         struct dso *kernel;
455
456         if (machine__is_host(machine)) {
457                 vmlinux_name = symbol_conf.vmlinux_name;
458                 if (!vmlinux_name)
459                         vmlinux_name = "[kernel.kallsyms]";
460
461                 kernel = dso__kernel_findnew(machine, vmlinux_name,
462                                              "[kernel]",
463                                              DSO_TYPE_KERNEL);
464         } else {
465                 char bf[PATH_MAX];
466
467                 if (machine__is_default_guest(machine))
468                         vmlinux_name = symbol_conf.default_guest_vmlinux_name;
469                 if (!vmlinux_name)
470                         vmlinux_name = machine__mmap_name(machine, bf,
471                                                           sizeof(bf));
472
473                 kernel = dso__kernel_findnew(machine, vmlinux_name,
474                                              "[guest.kernel]",
475                                              DSO_TYPE_GUEST_KERNEL);
476         }
477
478         if (kernel != NULL && (!kernel->has_build_id))
479                 dso__read_running_kernel_build_id(kernel, machine);
480
481         return kernel;
482 }
483
484 struct process_args {
485         u64 start;
486 };
487
488 static int symbol__in_kernel(void *arg, const char *name,
489                              char type __maybe_unused, u64 start)
490 {
491         struct process_args *args = arg;
492
493         if (strchr(name, '['))
494                 return 0;
495
496         args->start = start;
497         return 1;
498 }
499
500 static void machine__get_kallsyms_filename(struct machine *machine, char *buf,
501                                            size_t bufsz)
502 {
503         if (machine__is_default_guest(machine))
504                 scnprintf(buf, bufsz, "%s", symbol_conf.default_guest_kallsyms);
505         else
506                 scnprintf(buf, bufsz, "%s/proc/kallsyms", machine->root_dir);
507 }
508
509 /* Figure out the start address of kernel map from /proc/kallsyms */
510 static u64 machine__get_kernel_start_addr(struct machine *machine)
511 {
512         char filename[PATH_MAX];
513         struct process_args args;
514
515         machine__get_kallsyms_filename(machine, filename, PATH_MAX);
516
517         if (symbol__restricted_filename(filename, "/proc/kallsyms"))
518                 return 0;
519
520         if (kallsyms__parse(filename, &args, symbol__in_kernel) <= 0)
521                 return 0;
522
523         return args.start;
524 }
525
526 int __machine__create_kernel_maps(struct machine *machine, struct dso *kernel)
527 {
528         enum map_type type;
529         u64 start = machine__get_kernel_start_addr(machine);
530
531         for (type = 0; type < MAP__NR_TYPES; ++type) {
532                 struct kmap *kmap;
533
534                 machine->vmlinux_maps[type] = map__new2(start, kernel, type);
535                 if (machine->vmlinux_maps[type] == NULL)
536                         return -1;
537
538                 machine->vmlinux_maps[type]->map_ip =
539                         machine->vmlinux_maps[type]->unmap_ip =
540                                 identity__map_ip;
541                 kmap = map__kmap(machine->vmlinux_maps[type]);
542                 kmap->kmaps = &machine->kmaps;
543                 map_groups__insert(&machine->kmaps,
544                                    machine->vmlinux_maps[type]);
545         }
546
547         return 0;
548 }
549
550 void machine__destroy_kernel_maps(struct machine *machine)
551 {
552         enum map_type type;
553
554         for (type = 0; type < MAP__NR_TYPES; ++type) {
555                 struct kmap *kmap;
556
557                 if (machine->vmlinux_maps[type] == NULL)
558                         continue;
559
560                 kmap = map__kmap(machine->vmlinux_maps[type]);
561                 map_groups__remove(&machine->kmaps,
562                                    machine->vmlinux_maps[type]);
563                 if (kmap->ref_reloc_sym) {
564                         /*
565                          * ref_reloc_sym is shared among all maps, so free just
566                          * on one of them.
567                          */
568                         if (type == MAP__FUNCTION) {
569                                 zfree((char **)&kmap->ref_reloc_sym->name);
570                                 zfree(&kmap->ref_reloc_sym);
571                         } else
572                                 kmap->ref_reloc_sym = NULL;
573                 }
574
575                 map__delete(machine->vmlinux_maps[type]);
576                 machine->vmlinux_maps[type] = NULL;
577         }
578 }
579
580 int machines__create_guest_kernel_maps(struct machines *machines)
581 {
582         int ret = 0;
583         struct dirent **namelist = NULL;
584         int i, items = 0;
585         char path[PATH_MAX];
586         pid_t pid;
587         char *endp;
588
589         if (symbol_conf.default_guest_vmlinux_name ||
590             symbol_conf.default_guest_modules ||
591             symbol_conf.default_guest_kallsyms) {
592                 machines__create_kernel_maps(machines, DEFAULT_GUEST_KERNEL_ID);
593         }
594
595         if (symbol_conf.guestmount) {
596                 items = scandir(symbol_conf.guestmount, &namelist, NULL, NULL);
597                 if (items <= 0)
598                         return -ENOENT;
599                 for (i = 0; i < items; i++) {
600                         if (!isdigit(namelist[i]->d_name[0])) {
601                                 /* Filter out . and .. */
602                                 continue;
603                         }
604                         pid = (pid_t)strtol(namelist[i]->d_name, &endp, 10);
605                         if ((*endp != '\0') ||
606                             (endp == namelist[i]->d_name) ||
607                             (errno == ERANGE)) {
608                                 pr_debug("invalid directory (%s). Skipping.\n",
609                                          namelist[i]->d_name);
610                                 continue;
611                         }
612                         sprintf(path, "%s/%s/proc/kallsyms",
613                                 symbol_conf.guestmount,
614                                 namelist[i]->d_name);
615                         ret = access(path, R_OK);
616                         if (ret) {
617                                 pr_debug("Can't access file %s\n", path);
618                                 goto failure;
619                         }
620                         machines__create_kernel_maps(machines, pid);
621                 }
622 failure:
623                 free(namelist);
624         }
625
626         return ret;
627 }
628
629 void machines__destroy_kernel_maps(struct machines *machines)
630 {
631         struct rb_node *next = rb_first(&machines->guests);
632
633         machine__destroy_kernel_maps(&machines->host);
634
635         while (next) {
636                 struct machine *pos = rb_entry(next, struct machine, rb_node);
637
638                 next = rb_next(&pos->rb_node);
639                 rb_erase(&pos->rb_node, &machines->guests);
640                 machine__delete(pos);
641         }
642 }
643
644 int machines__create_kernel_maps(struct machines *machines, pid_t pid)
645 {
646         struct machine *machine = machines__findnew(machines, pid);
647
648         if (machine == NULL)
649                 return -1;
650
651         return machine__create_kernel_maps(machine);
652 }
653
654 int machine__load_kallsyms(struct machine *machine, const char *filename,
655                            enum map_type type, symbol_filter_t filter)
656 {
657         struct map *map = machine->vmlinux_maps[type];
658         int ret = dso__load_kallsyms(map->dso, filename, map, filter);
659
660         if (ret > 0) {
661                 dso__set_loaded(map->dso, type);
662                 /*
663                  * Since /proc/kallsyms will have multiple sessions for the
664                  * kernel, with modules between them, fixup the end of all
665                  * sections.
666                  */
667                 __map_groups__fixup_end(&machine->kmaps, type);
668         }
669
670         return ret;
671 }
672
673 int machine__load_vmlinux_path(struct machine *machine, enum map_type type,
674                                symbol_filter_t filter)
675 {
676         struct map *map = machine->vmlinux_maps[type];
677         int ret = dso__load_vmlinux_path(map->dso, map, filter);
678
679         if (ret > 0)
680                 dso__set_loaded(map->dso, type);
681
682         return ret;
683 }
684
685 static void map_groups__fixup_end(struct map_groups *mg)
686 {
687         int i;
688         for (i = 0; i < MAP__NR_TYPES; ++i)
689                 __map_groups__fixup_end(mg, i);
690 }
691
692 static char *get_kernel_version(const char *root_dir)
693 {
694         char version[PATH_MAX];
695         FILE *file;
696         char *name, *tmp;
697         const char *prefix = "Linux version ";
698
699         sprintf(version, "%s/proc/version", root_dir);
700         file = fopen(version, "r");
701         if (!file)
702                 return NULL;
703
704         version[0] = '\0';
705         tmp = fgets(version, sizeof(version), file);
706         fclose(file);
707
708         name = strstr(version, prefix);
709         if (!name)
710                 return NULL;
711         name += strlen(prefix);
712         tmp = strchr(name, ' ');
713         if (tmp)
714                 *tmp = '\0';
715
716         return strdup(name);
717 }
718
719 static int map_groups__set_modules_path_dir(struct map_groups *mg,
720                                 const char *dir_name)
721 {
722         struct dirent *dent;
723         DIR *dir = opendir(dir_name);
724         int ret = 0;
725
726         if (!dir) {
727                 pr_debug("%s: cannot open %s dir\n", __func__, dir_name);
728                 return -1;
729         }
730
731         while ((dent = readdir(dir)) != NULL) {
732                 char path[PATH_MAX];
733                 struct stat st;
734
735                 /*sshfs might return bad dent->d_type, so we have to stat*/
736                 snprintf(path, sizeof(path), "%s/%s", dir_name, dent->d_name);
737                 if (stat(path, &st))
738                         continue;
739
740                 if (S_ISDIR(st.st_mode)) {
741                         if (!strcmp(dent->d_name, ".") ||
742                             !strcmp(dent->d_name, ".."))
743                                 continue;
744
745                         ret = map_groups__set_modules_path_dir(mg, path);
746                         if (ret < 0)
747                                 goto out;
748                 } else {
749                         char *dot = strrchr(dent->d_name, '.'),
750                              dso_name[PATH_MAX];
751                         struct map *map;
752                         char *long_name;
753
754                         if (dot == NULL || strcmp(dot, ".ko"))
755                                 continue;
756                         snprintf(dso_name, sizeof(dso_name), "[%.*s]",
757                                  (int)(dot - dent->d_name), dent->d_name);
758
759                         strxfrchar(dso_name, '-', '_');
760                         map = map_groups__find_by_name(mg, MAP__FUNCTION,
761                                                        dso_name);
762                         if (map == NULL)
763                                 continue;
764
765                         long_name = strdup(path);
766                         if (long_name == NULL) {
767                                 ret = -1;
768                                 goto out;
769                         }
770                         dso__set_long_name(map->dso, long_name, true);
771                         dso__kernel_module_get_build_id(map->dso, "");
772                 }
773         }
774
775 out:
776         closedir(dir);
777         return ret;
778 }
779
780 static int machine__set_modules_path(struct machine *machine)
781 {
782         char *version;
783         char modules_path[PATH_MAX];
784
785         version = get_kernel_version(machine->root_dir);
786         if (!version)
787                 return -1;
788
789         snprintf(modules_path, sizeof(modules_path), "%s/lib/modules/%s/kernel",
790                  machine->root_dir, version);
791         free(version);
792
793         return map_groups__set_modules_path_dir(&machine->kmaps, modules_path);
794 }
795
796 static int machine__create_module(void *arg, const char *name, u64 start)
797 {
798         struct machine *machine = arg;
799         struct map *map;
800
801         map = machine__new_module(machine, start, name);
802         if (map == NULL)
803                 return -1;
804
805         dso__kernel_module_get_build_id(map->dso, machine->root_dir);
806
807         return 0;
808 }
809
810 static int machine__create_modules(struct machine *machine)
811 {
812         const char *modules;
813         char path[PATH_MAX];
814
815         if (machine__is_default_guest(machine)) {
816                 modules = symbol_conf.default_guest_modules;
817         } else {
818                 snprintf(path, PATH_MAX, "%s/proc/modules", machine->root_dir);
819                 modules = path;
820         }
821
822         if (symbol__restricted_filename(modules, "/proc/modules"))
823                 return -1;
824
825         if (modules__parse(modules, machine, machine__create_module))
826                 return -1;
827
828         if (!machine__set_modules_path(machine))
829                 return 0;
830
831         pr_debug("Problems setting modules path maps, continuing anyway...\n");
832
833         return 0;
834 }
835
836 const char *ref_reloc_sym_names[] = {"_text", "_stext", NULL};
837
838 int machine__create_kernel_maps(struct machine *machine)
839 {
840         struct dso *kernel = machine__get_kernel(machine);
841         char filename[PATH_MAX];
842         const char *name;
843         u64 addr = 0;
844         int i;
845
846         machine__get_kallsyms_filename(machine, filename, PATH_MAX);
847
848         for (i = 0; (name = ref_reloc_sym_names[i]) != NULL; i++) {
849                 addr = kallsyms__get_function_start(filename, name);
850                 if (addr)
851                         break;
852         }
853         if (!addr)
854                 return -1;
855
856         if (kernel == NULL ||
857             __machine__create_kernel_maps(machine, kernel) < 0)
858                 return -1;
859
860         if (symbol_conf.use_modules && machine__create_modules(machine) < 0) {
861                 if (machine__is_host(machine))
862                         pr_debug("Problems creating module maps, "
863                                  "continuing anyway...\n");
864                 else
865                         pr_debug("Problems creating module maps for guest %d, "
866                                  "continuing anyway...\n", machine->pid);
867         }
868
869         /*
870          * Now that we have all the maps created, just set the ->end of them:
871          */
872         map_groups__fixup_end(&machine->kmaps);
873
874         if (maps__set_kallsyms_ref_reloc_sym(machine->vmlinux_maps, name,
875                                              addr)) {
876                 machine__destroy_kernel_maps(machine);
877                 return -1;
878         }
879
880         return 0;
881 }
882
883 static void machine__set_kernel_mmap_len(struct machine *machine,
884                                          union perf_event *event)
885 {
886         int i;
887
888         for (i = 0; i < MAP__NR_TYPES; i++) {
889                 machine->vmlinux_maps[i]->start = event->mmap.start;
890                 machine->vmlinux_maps[i]->end   = (event->mmap.start +
891                                                    event->mmap.len);
892                 /*
893                  * Be a bit paranoid here, some perf.data file came with
894                  * a zero sized synthesized MMAP event for the kernel.
895                  */
896                 if (machine->vmlinux_maps[i]->end == 0)
897                         machine->vmlinux_maps[i]->end = ~0ULL;
898         }
899 }
900
901 static bool machine__uses_kcore(struct machine *machine)
902 {
903         struct dso *dso;
904
905         list_for_each_entry(dso, &machine->kernel_dsos, node) {
906                 if (dso__is_kcore(dso))
907                         return true;
908         }
909
910         return false;
911 }
912
913 static int machine__process_kernel_mmap_event(struct machine *machine,
914                                               union perf_event *event)
915 {
916         struct map *map;
917         char kmmap_prefix[PATH_MAX];
918         enum dso_kernel_type kernel_type;
919         bool is_kernel_mmap;
920
921         /* If we have maps from kcore then we do not need or want any others */
922         if (machine__uses_kcore(machine))
923                 return 0;
924
925         machine__mmap_name(machine, kmmap_prefix, sizeof(kmmap_prefix));
926         if (machine__is_host(machine))
927                 kernel_type = DSO_TYPE_KERNEL;
928         else
929                 kernel_type = DSO_TYPE_GUEST_KERNEL;
930
931         is_kernel_mmap = memcmp(event->mmap.filename,
932                                 kmmap_prefix,
933                                 strlen(kmmap_prefix) - 1) == 0;
934         if (event->mmap.filename[0] == '/' ||
935             (!is_kernel_mmap && event->mmap.filename[0] == '[')) {
936
937                 char short_module_name[1024];
938                 char *name, *dot;
939
940                 if (event->mmap.filename[0] == '/') {
941                         name = strrchr(event->mmap.filename, '/');
942                         if (name == NULL)
943                                 goto out_problem;
944
945                         ++name; /* skip / */
946                         dot = strrchr(name, '.');
947                         if (dot == NULL)
948                                 goto out_problem;
949                         snprintf(short_module_name, sizeof(short_module_name),
950                                         "[%.*s]", (int)(dot - name), name);
951                         strxfrchar(short_module_name, '-', '_');
952                 } else
953                         strcpy(short_module_name, event->mmap.filename);
954
955                 map = machine__new_module(machine, event->mmap.start,
956                                           event->mmap.filename);
957                 if (map == NULL)
958                         goto out_problem;
959
960                 name = strdup(short_module_name);
961                 if (name == NULL)
962                         goto out_problem;
963
964                 dso__set_short_name(map->dso, name, true);
965                 map->end = map->start + event->mmap.len;
966         } else if (is_kernel_mmap) {
967                 const char *symbol_name = (event->mmap.filename +
968                                 strlen(kmmap_prefix));
969                 /*
970                  * Should be there already, from the build-id table in
971                  * the header.
972                  */
973                 struct dso *kernel = __dsos__findnew(&machine->kernel_dsos,
974                                                      kmmap_prefix);
975                 if (kernel == NULL)
976                         goto out_problem;
977
978                 kernel->kernel = kernel_type;
979                 if (__machine__create_kernel_maps(machine, kernel) < 0)
980                         goto out_problem;
981
982                 machine__set_kernel_mmap_len(machine, event);
983
984                 /*
985                  * Avoid using a zero address (kptr_restrict) for the ref reloc
986                  * symbol. Effectively having zero here means that at record
987                  * time /proc/sys/kernel/kptr_restrict was non zero.
988                  */
989                 if (event->mmap.pgoff != 0) {
990                         maps__set_kallsyms_ref_reloc_sym(machine->vmlinux_maps,
991                                                          symbol_name,
992                                                          event->mmap.pgoff);
993                 }
994
995                 if (machine__is_default_guest(machine)) {
996                         /*
997                          * preload dso of guest kernel and modules
998                          */
999                         dso__load(kernel, machine->vmlinux_maps[MAP__FUNCTION],
1000                                   NULL);
1001                 }
1002         }
1003         return 0;
1004 out_problem:
1005         return -1;
1006 }
1007
1008 int machine__process_mmap2_event(struct machine *machine,
1009                                  union perf_event *event,
1010                                  struct perf_sample *sample __maybe_unused)
1011 {
1012         u8 cpumode = event->header.misc & PERF_RECORD_MISC_CPUMODE_MASK;
1013         struct thread *thread;
1014         struct map *map;
1015         enum map_type type;
1016         int ret = 0;
1017
1018         if (dump_trace)
1019                 perf_event__fprintf_mmap2(event, stdout);
1020
1021         if (cpumode == PERF_RECORD_MISC_GUEST_KERNEL ||
1022             cpumode == PERF_RECORD_MISC_KERNEL) {
1023                 ret = machine__process_kernel_mmap_event(machine, event);
1024                 if (ret < 0)
1025                         goto out_problem;
1026                 return 0;
1027         }
1028
1029         thread = machine__findnew_thread(machine, event->mmap2.pid,
1030                                         event->mmap2.tid);
1031         if (thread == NULL)
1032                 goto out_problem;
1033
1034         if (event->header.misc & PERF_RECORD_MISC_MMAP_DATA)
1035                 type = MAP__VARIABLE;
1036         else
1037                 type = MAP__FUNCTION;
1038
1039         map = map__new(&machine->user_dsos, event->mmap2.start,
1040                         event->mmap2.len, event->mmap2.pgoff,
1041                         event->mmap2.pid, event->mmap2.maj,
1042                         event->mmap2.min, event->mmap2.ino,
1043                         event->mmap2.ino_generation,
1044                         event->mmap2.filename, type);
1045
1046         if (map == NULL)
1047                 goto out_problem;
1048
1049         thread__insert_map(thread, map);
1050         return 0;
1051
1052 out_problem:
1053         dump_printf("problem processing PERF_RECORD_MMAP2, skipping event.\n");
1054         return 0;
1055 }
1056
1057 int machine__process_mmap_event(struct machine *machine, union perf_event *event,
1058                                 struct perf_sample *sample __maybe_unused)
1059 {
1060         u8 cpumode = event->header.misc & PERF_RECORD_MISC_CPUMODE_MASK;
1061         struct thread *thread;
1062         struct map *map;
1063         enum map_type type;
1064         int ret = 0;
1065
1066         if (dump_trace)
1067                 perf_event__fprintf_mmap(event, stdout);
1068
1069         if (cpumode == PERF_RECORD_MISC_GUEST_KERNEL ||
1070             cpumode == PERF_RECORD_MISC_KERNEL) {
1071                 ret = machine__process_kernel_mmap_event(machine, event);
1072                 if (ret < 0)
1073                         goto out_problem;
1074                 return 0;
1075         }
1076
1077         thread = machine__findnew_thread(machine, event->mmap.pid,
1078                                          event->mmap.tid);
1079         if (thread == NULL)
1080                 goto out_problem;
1081
1082         if (event->header.misc & PERF_RECORD_MISC_MMAP_DATA)
1083                 type = MAP__VARIABLE;
1084         else
1085                 type = MAP__FUNCTION;
1086
1087         map = map__new(&machine->user_dsos, event->mmap.start,
1088                         event->mmap.len, event->mmap.pgoff,
1089                         event->mmap.pid, 0, 0, 0, 0,
1090                         event->mmap.filename,
1091                         type);
1092
1093         if (map == NULL)
1094                 goto out_problem;
1095
1096         thread__insert_map(thread, map);
1097         return 0;
1098
1099 out_problem:
1100         dump_printf("problem processing PERF_RECORD_MMAP, skipping event.\n");
1101         return 0;
1102 }
1103
1104 static void machine__remove_thread(struct machine *machine, struct thread *th)
1105 {
1106         machine->last_match = NULL;
1107         rb_erase(&th->rb_node, &machine->threads);
1108         /*
1109          * We may have references to this thread, for instance in some hist_entry
1110          * instances, so just move them to a separate list.
1111          */
1112         list_add_tail(&th->node, &machine->dead_threads);
1113 }
1114
1115 int machine__process_fork_event(struct machine *machine, union perf_event *event,
1116                                 struct perf_sample *sample)
1117 {
1118         struct thread *thread = machine__find_thread(machine,
1119                                                      event->fork.pid,
1120                                                      event->fork.tid);
1121         struct thread *parent = machine__findnew_thread(machine,
1122                                                         event->fork.ppid,
1123                                                         event->fork.ptid);
1124
1125         /* if a thread currently exists for the thread id remove it */
1126         if (thread != NULL)
1127                 machine__remove_thread(machine, thread);
1128
1129         thread = machine__findnew_thread(machine, event->fork.pid,
1130                                          event->fork.tid);
1131         if (dump_trace)
1132                 perf_event__fprintf_task(event, stdout);
1133
1134         if (thread == NULL || parent == NULL ||
1135             thread__fork(thread, parent, sample->time) < 0) {
1136                 dump_printf("problem processing PERF_RECORD_FORK, skipping event.\n");
1137                 return -1;
1138         }
1139
1140         return 0;
1141 }
1142
1143 int machine__process_exit_event(struct machine *machine, union perf_event *event,
1144                                 struct perf_sample *sample __maybe_unused)
1145 {
1146         struct thread *thread = machine__find_thread(machine,
1147                                                      event->fork.pid,
1148                                                      event->fork.tid);
1149
1150         if (dump_trace)
1151                 perf_event__fprintf_task(event, stdout);
1152
1153         if (thread != NULL)
1154                 thread__exited(thread);
1155
1156         return 0;
1157 }
1158
1159 int machine__process_event(struct machine *machine, union perf_event *event,
1160                            struct perf_sample *sample)
1161 {
1162         int ret;
1163
1164         switch (event->header.type) {
1165         case PERF_RECORD_COMM:
1166                 ret = machine__process_comm_event(machine, event, sample); break;
1167         case PERF_RECORD_MMAP:
1168                 ret = machine__process_mmap_event(machine, event, sample); break;
1169         case PERF_RECORD_MMAP2:
1170                 ret = machine__process_mmap2_event(machine, event, sample); break;
1171         case PERF_RECORD_FORK:
1172                 ret = machine__process_fork_event(machine, event, sample); break;
1173         case PERF_RECORD_EXIT:
1174                 ret = machine__process_exit_event(machine, event, sample); break;
1175         case PERF_RECORD_LOST:
1176                 ret = machine__process_lost_event(machine, event, sample); break;
1177         default:
1178                 ret = -1;
1179                 break;
1180         }
1181
1182         return ret;
1183 }
1184
1185 static bool symbol__match_regex(struct symbol *sym, regex_t *regex)
1186 {
1187         if (sym->name && !regexec(regex, sym->name, 0, NULL, 0))
1188                 return 1;
1189         return 0;
1190 }
1191
1192 static void ip__resolve_ams(struct machine *machine, struct thread *thread,
1193                             struct addr_map_symbol *ams,
1194                             u64 ip)
1195 {
1196         struct addr_location al;
1197
1198         memset(&al, 0, sizeof(al));
1199         /*
1200          * We cannot use the header.misc hint to determine whether a
1201          * branch stack address is user, kernel, guest, hypervisor.
1202          * Branches may straddle the kernel/user/hypervisor boundaries.
1203          * Thus, we have to try consecutively until we find a match
1204          * or else, the symbol is unknown
1205          */
1206         thread__find_cpumode_addr_location(thread, machine, MAP__FUNCTION, ip, &al);
1207
1208         ams->addr = ip;
1209         ams->al_addr = al.addr;
1210         ams->sym = al.sym;
1211         ams->map = al.map;
1212 }
1213
1214 static void ip__resolve_data(struct machine *machine, struct thread *thread,
1215                              u8 m, struct addr_map_symbol *ams, u64 addr)
1216 {
1217         struct addr_location al;
1218
1219         memset(&al, 0, sizeof(al));
1220
1221         thread__find_addr_location(thread, machine, m, MAP__VARIABLE, addr,
1222                                    &al);
1223         ams->addr = addr;
1224         ams->al_addr = al.addr;
1225         ams->sym = al.sym;
1226         ams->map = al.map;
1227 }
1228
1229 struct mem_info *sample__resolve_mem(struct perf_sample *sample,
1230                                      struct addr_location *al)
1231 {
1232         struct mem_info *mi = zalloc(sizeof(*mi));
1233
1234         if (!mi)
1235                 return NULL;
1236
1237         ip__resolve_ams(al->machine, al->thread, &mi->iaddr, sample->ip);
1238         ip__resolve_data(al->machine, al->thread, al->cpumode,
1239                          &mi->daddr, sample->addr);
1240         mi->data_src.val = sample->data_src;
1241
1242         return mi;
1243 }
1244
1245 struct branch_info *sample__resolve_bstack(struct perf_sample *sample,
1246                                            struct addr_location *al)
1247 {
1248         unsigned int i;
1249         const struct branch_stack *bs = sample->branch_stack;
1250         struct branch_info *bi = calloc(bs->nr, sizeof(struct branch_info));
1251
1252         if (!bi)
1253                 return NULL;
1254
1255         for (i = 0; i < bs->nr; i++) {
1256                 ip__resolve_ams(al->machine, al->thread, &bi[i].to, bs->entries[i].to);
1257                 ip__resolve_ams(al->machine, al->thread, &bi[i].from, bs->entries[i].from);
1258                 bi[i].flags = bs->entries[i].flags;
1259         }
1260         return bi;
1261 }
1262
1263 static int machine__resolve_callchain_sample(struct machine *machine,
1264                                              struct thread *thread,
1265                                              struct ip_callchain *chain,
1266                                              struct symbol **parent,
1267                                              struct addr_location *root_al,
1268                                              int max_stack)
1269 {
1270         u8 cpumode = PERF_RECORD_MISC_USER;
1271         int chain_nr = min(max_stack, (int)chain->nr);
1272         int i;
1273         int err;
1274
1275         callchain_cursor_reset(&callchain_cursor);
1276
1277         if (chain->nr > PERF_MAX_STACK_DEPTH) {
1278                 pr_warning("corrupted callchain. skipping...\n");
1279                 return 0;
1280         }
1281
1282         for (i = 0; i < chain_nr; i++) {
1283                 u64 ip;
1284                 struct addr_location al;
1285
1286                 if (callchain_param.order == ORDER_CALLEE)
1287                         ip = chain->ips[i];
1288                 else
1289                         ip = chain->ips[chain->nr - i - 1];
1290
1291                 if (ip >= PERF_CONTEXT_MAX) {
1292                         switch (ip) {
1293                         case PERF_CONTEXT_HV:
1294                                 cpumode = PERF_RECORD_MISC_HYPERVISOR;
1295                                 break;
1296                         case PERF_CONTEXT_KERNEL:
1297                                 cpumode = PERF_RECORD_MISC_KERNEL;
1298                                 break;
1299                         case PERF_CONTEXT_USER:
1300                                 cpumode = PERF_RECORD_MISC_USER;
1301                                 break;
1302                         default:
1303                                 pr_debug("invalid callchain context: "
1304                                          "%"PRId64"\n", (s64) ip);
1305                                 /*
1306                                  * It seems the callchain is corrupted.
1307                                  * Discard all.
1308                                  */
1309                                 callchain_cursor_reset(&callchain_cursor);
1310                                 return 0;
1311                         }
1312                         continue;
1313                 }
1314
1315                 al.filtered = 0;
1316                 thread__find_addr_location(thread, machine, cpumode,
1317                                            MAP__FUNCTION, ip, &al);
1318                 if (al.sym != NULL) {
1319                         if (sort__has_parent && !*parent &&
1320                             symbol__match_regex(al.sym, &parent_regex))
1321                                 *parent = al.sym;
1322                         else if (have_ignore_callees && root_al &&
1323                           symbol__match_regex(al.sym, &ignore_callees_regex)) {
1324                                 /* Treat this symbol as the root,
1325                                    forgetting its callees. */
1326                                 *root_al = al;
1327                                 callchain_cursor_reset(&callchain_cursor);
1328                         }
1329                 }
1330
1331                 err = callchain_cursor_append(&callchain_cursor,
1332                                               ip, al.map, al.sym);
1333                 if (err)
1334                         return err;
1335         }
1336
1337         return 0;
1338 }
1339
1340 static int unwind_entry(struct unwind_entry *entry, void *arg)
1341 {
1342         struct callchain_cursor *cursor = arg;
1343         return callchain_cursor_append(cursor, entry->ip,
1344                                        entry->map, entry->sym);
1345 }
1346
1347 int machine__resolve_callchain(struct machine *machine,
1348                                struct perf_evsel *evsel,
1349                                struct thread *thread,
1350                                struct perf_sample *sample,
1351                                struct symbol **parent,
1352                                struct addr_location *root_al,
1353                                int max_stack)
1354 {
1355         int ret;
1356
1357         ret = machine__resolve_callchain_sample(machine, thread,
1358                                                 sample->callchain, parent,
1359                                                 root_al, max_stack);
1360         if (ret)
1361                 return ret;
1362
1363         /* Can we do dwarf post unwind? */
1364         if (!((evsel->attr.sample_type & PERF_SAMPLE_REGS_USER) &&
1365               (evsel->attr.sample_type & PERF_SAMPLE_STACK_USER)))
1366                 return 0;
1367
1368         /* Bail out if nothing was captured. */
1369         if ((!sample->user_regs.regs) ||
1370             (!sample->user_stack.size))
1371                 return 0;
1372
1373         return unwind__get_entries(unwind_entry, &callchain_cursor, machine,
1374                                    thread, sample, max_stack);
1375
1376 }
1377
1378 int machine__for_each_thread(struct machine *machine,
1379                              int (*fn)(struct thread *thread, void *p),
1380                              void *priv)
1381 {
1382         struct rb_node *nd;
1383         struct thread *thread;
1384         int rc = 0;
1385
1386         for (nd = rb_first(&machine->threads); nd; nd = rb_next(nd)) {
1387                 thread = rb_entry(nd, struct thread, rb_node);
1388                 rc = fn(thread, priv);
1389                 if (rc != 0)
1390                         return rc;
1391         }
1392
1393         list_for_each_entry(thread, &machine->dead_threads, node) {
1394                 rc = fn(thread, priv);
1395                 if (rc != 0)
1396                         return rc;
1397         }
1398         return rc;
1399 }
1400
1401 int __machine__synthesize_threads(struct machine *machine, struct perf_tool *tool,
1402                                   struct target *target, struct thread_map *threads,
1403                                   perf_event__handler_t process, bool data_mmap)
1404 {
1405         if (target__has_task(target))
1406                 return perf_event__synthesize_thread_map(tool, threads, process, machine, data_mmap);
1407         else if (target__has_cpu(target))
1408                 return perf_event__synthesize_threads(tool, process, machine, data_mmap);
1409         /* command specified */
1410         return 0;
1411 }