GNU Linux-libre 4.14.266-gnu1
[releases.git] / tools / perf / util / map.c
1 // SPDX-License-Identifier: GPL-2.0
2 #include "symbol.h"
3 #include <assert.h>
4 #include <errno.h>
5 #include <inttypes.h>
6 #include <limits.h>
7 #include <stdlib.h>
8 #include <string.h>
9 #include <stdio.h>
10 #include <unistd.h>
11 #include <uapi/linux/mman.h> /* To get things like MAP_HUGETLB even on older libc headers */
12 #include "map.h"
13 #include "thread.h"
14 #include "vdso.h"
15 #include "build-id.h"
16 #include "util.h"
17 #include "debug.h"
18 #include "machine.h"
19 #include <linux/string.h>
20 #include "srcline.h"
21 #include "namespaces.h"
22 #include "unwind.h"
23
24 static void __maps__insert(struct maps *maps, struct map *map);
25
26 const char *map_type__name[MAP__NR_TYPES] = {
27         [MAP__FUNCTION] = "Functions",
28         [MAP__VARIABLE] = "Variables",
29 };
30
31 static inline int is_anon_memory(const char *filename, u32 flags)
32 {
33         return flags & MAP_HUGETLB ||
34                !strcmp(filename, "//anon") ||
35                !strncmp(filename, "/dev/zero", sizeof("/dev/zero") - 1) ||
36                !strncmp(filename, "/anon_hugepage", sizeof("/anon_hugepage") - 1);
37 }
38
39 static inline int is_no_dso_memory(const char *filename)
40 {
41         return !strncmp(filename, "[stack", 6) ||
42                !strncmp(filename, "/SYSV",5)   ||
43                !strcmp(filename, "[heap]");
44 }
45
46 static inline int is_android_lib(const char *filename)
47 {
48         return !strncmp(filename, "/data/app-lib", 13) ||
49                !strncmp(filename, "/system/lib", 11);
50 }
51
52 static inline bool replace_android_lib(const char *filename, char *newfilename)
53 {
54         const char *libname;
55         char *app_abi;
56         size_t app_abi_length, new_length;
57         size_t lib_length = 0;
58
59         libname  = strrchr(filename, '/');
60         if (libname)
61                 lib_length = strlen(libname);
62
63         app_abi = getenv("APP_ABI");
64         if (!app_abi)
65                 return false;
66
67         app_abi_length = strlen(app_abi);
68
69         if (!strncmp(filename, "/data/app-lib", 13)) {
70                 char *apk_path;
71
72                 if (!app_abi_length)
73                         return false;
74
75                 new_length = 7 + app_abi_length + lib_length;
76
77                 apk_path = getenv("APK_PATH");
78                 if (apk_path) {
79                         new_length += strlen(apk_path) + 1;
80                         if (new_length > PATH_MAX)
81                                 return false;
82                         snprintf(newfilename, new_length,
83                                  "%s/libs/%s/%s", apk_path, app_abi, libname);
84                 } else {
85                         if (new_length > PATH_MAX)
86                                 return false;
87                         snprintf(newfilename, new_length,
88                                  "libs/%s/%s", app_abi, libname);
89                 }
90                 return true;
91         }
92
93         if (!strncmp(filename, "/system/lib/", 12)) {
94                 char *ndk, *app;
95                 const char *arch;
96                 int ndk_length, app_length;
97
98                 ndk = getenv("NDK_ROOT");
99                 app = getenv("APP_PLATFORM");
100
101                 if (!(ndk && app))
102                         return false;
103
104                 ndk_length = strlen(ndk);
105                 app_length = strlen(app);
106
107                 if (!(ndk_length && app_length && app_abi_length))
108                         return false;
109
110                 arch = !strncmp(app_abi, "arm", 3) ? "arm" :
111                        !strncmp(app_abi, "mips", 4) ? "mips" :
112                        !strncmp(app_abi, "x86", 3) ? "x86" : NULL;
113
114                 if (!arch)
115                         return false;
116
117                 new_length = 27 + ndk_length +
118                              app_length + lib_length
119                            + strlen(arch);
120
121                 if (new_length > PATH_MAX)
122                         return false;
123                 snprintf(newfilename, new_length,
124                         "%.*s/platforms/%.*s/arch-%s/usr/lib/%s",
125                         ndk_length, ndk, app_length, app, arch, libname);
126
127                 return true;
128         }
129         return false;
130 }
131
132 void map__init(struct map *map, enum map_type type,
133                u64 start, u64 end, u64 pgoff, struct dso *dso)
134 {
135         map->type     = type;
136         map->start    = start;
137         map->end      = end;
138         map->pgoff    = pgoff;
139         map->reloc    = 0;
140         map->dso      = dso__get(dso);
141         map->map_ip   = map__map_ip;
142         map->unmap_ip = map__unmap_ip;
143         RB_CLEAR_NODE(&map->rb_node);
144         map->groups   = NULL;
145         map->erange_warned = false;
146         refcount_set(&map->refcnt, 1);
147 }
148
149 struct map *map__new(struct machine *machine, u64 start, u64 len,
150                      u64 pgoff, u32 d_maj, u32 d_min, u64 ino,
151                      u64 ino_gen, u32 prot, u32 flags, char *filename,
152                      enum map_type type, struct thread *thread)
153 {
154         struct map *map = malloc(sizeof(*map));
155         struct nsinfo *nsi = NULL;
156         struct nsinfo *nnsi;
157
158         if (map != NULL) {
159                 char newfilename[PATH_MAX];
160                 struct dso *dso;
161                 int anon, no_dso, vdso, android;
162
163                 android = is_android_lib(filename);
164                 anon = is_anon_memory(filename, flags);
165                 vdso = is_vdso_map(filename);
166                 no_dso = is_no_dso_memory(filename);
167
168                 map->maj = d_maj;
169                 map->min = d_min;
170                 map->ino = ino;
171                 map->ino_generation = ino_gen;
172                 map->prot = prot;
173                 map->flags = flags;
174                 nsi = nsinfo__get(thread->nsinfo);
175
176                 if ((anon || no_dso) && nsi && type == MAP__FUNCTION) {
177                         snprintf(newfilename, sizeof(newfilename),
178                                  "/tmp/perf-%d.map", nsi->pid);
179                         filename = newfilename;
180                 }
181
182                 if (android) {
183                         if (replace_android_lib(filename, newfilename))
184                                 filename = newfilename;
185                 }
186
187                 if (vdso) {
188                         /* The vdso maps are always on the host and not the
189                          * container.  Ensure that we don't use setns to look
190                          * them up.
191                          */
192                         nnsi = nsinfo__copy(nsi);
193                         if (nnsi) {
194                                 nsinfo__put(nsi);
195                                 nnsi->need_setns = false;
196                                 nsi = nnsi;
197                         }
198                         pgoff = 0;
199                         dso = machine__findnew_vdso(machine, thread);
200                 } else
201                         dso = machine__findnew_dso(machine, filename);
202
203                 if (dso == NULL)
204                         goto out_delete;
205
206                 map__init(map, type, start, start + len, pgoff, dso);
207
208                 if (anon || no_dso) {
209                         map->map_ip = map->unmap_ip = identity__map_ip;
210
211                         /*
212                          * Set memory without DSO as loaded. All map__find_*
213                          * functions still return NULL, and we avoid the
214                          * unnecessary map__load warning.
215                          */
216                         if (type != MAP__FUNCTION)
217                                 dso__set_loaded(dso, map->type);
218                 }
219                 dso->nsinfo = nsi;
220                 dso__put(dso);
221         }
222         return map;
223 out_delete:
224         nsinfo__put(nsi);
225         free(map);
226         return NULL;
227 }
228
229 /*
230  * Constructor variant for modules (where we know from /proc/modules where
231  * they are loaded) and for vmlinux, where only after we load all the
232  * symbols we'll know where it starts and ends.
233  */
234 struct map *map__new2(u64 start, struct dso *dso, enum map_type type)
235 {
236         struct map *map = calloc(1, (sizeof(*map) +
237                                      (dso->kernel ? sizeof(struct kmap) : 0)));
238         if (map != NULL) {
239                 /*
240                  * ->end will be filled after we load all the symbols
241                  */
242                 map__init(map, type, start, 0, 0, dso);
243         }
244
245         return map;
246 }
247
248 /*
249  * Use this and __map__is_kmodule() for map instances that are in
250  * machine->kmaps, and thus have map->groups->machine all properly set, to
251  * disambiguate between the kernel and modules.
252  *
253  * When the need arises, introduce map__is_{kernel,kmodule)() that
254  * checks (map->groups != NULL && map->groups->machine != NULL &&
255  * map->dso->kernel) before calling __map__is_{kernel,kmodule}())
256  */
257 bool __map__is_kernel(const struct map *map)
258 {
259         return __machine__kernel_map(map->groups->machine, map->type) == map;
260 }
261
262 static void map__exit(struct map *map)
263 {
264         BUG_ON(!RB_EMPTY_NODE(&map->rb_node));
265         dso__zput(map->dso);
266 }
267
268 void map__delete(struct map *map)
269 {
270         map__exit(map);
271         free(map);
272 }
273
274 void map__put(struct map *map)
275 {
276         if (map && refcount_dec_and_test(&map->refcnt))
277                 map__delete(map);
278 }
279
280 void map__fixup_start(struct map *map)
281 {
282         struct rb_root *symbols = &map->dso->symbols[map->type];
283         struct rb_node *nd = rb_first(symbols);
284         if (nd != NULL) {
285                 struct symbol *sym = rb_entry(nd, struct symbol, rb_node);
286                 map->start = sym->start;
287         }
288 }
289
290 void map__fixup_end(struct map *map)
291 {
292         struct rb_root *symbols = &map->dso->symbols[map->type];
293         struct rb_node *nd = rb_last(symbols);
294         if (nd != NULL) {
295                 struct symbol *sym = rb_entry(nd, struct symbol, rb_node);
296                 map->end = sym->end;
297         }
298 }
299
300 #define DSO__DELETED "(deleted)"
301
302 int map__load(struct map *map)
303 {
304         const char *name = map->dso->long_name;
305         int nr;
306
307         if (dso__loaded(map->dso, map->type))
308                 return 0;
309
310         nr = dso__load(map->dso, map);
311         if (nr < 0) {
312                 if (map->dso->has_build_id) {
313                         char sbuild_id[SBUILD_ID_SIZE];
314
315                         build_id__sprintf(map->dso->build_id,
316                                           sizeof(map->dso->build_id),
317                                           sbuild_id);
318                         pr_warning("%s with build id %s not found",
319                                    name, sbuild_id);
320                 } else
321                         pr_warning("Failed to open %s", name);
322
323                 pr_warning(", continuing without symbols\n");
324                 return -1;
325         } else if (nr == 0) {
326 #ifdef HAVE_LIBELF_SUPPORT
327                 const size_t len = strlen(name);
328                 const size_t real_len = len - sizeof(DSO__DELETED);
329
330                 if (len > sizeof(DSO__DELETED) &&
331                     strcmp(name + real_len + 1, DSO__DELETED) == 0) {
332                         pr_warning("%.*s was updated (is prelink enabled?). "
333                                 "Restart the long running apps that use it!\n",
334                                    (int)real_len, name);
335                 } else {
336                         pr_warning("no symbols found in %s, maybe install "
337                                    "a debug package?\n", name);
338                 }
339 #endif
340                 return -1;
341         }
342
343         return 0;
344 }
345
346 struct symbol *map__find_symbol(struct map *map, u64 addr)
347 {
348         if (map__load(map) < 0)
349                 return NULL;
350
351         return dso__find_symbol(map->dso, map->type, addr);
352 }
353
354 struct symbol *map__find_symbol_by_name(struct map *map, const char *name)
355 {
356         if (map__load(map) < 0)
357                 return NULL;
358
359         if (!dso__sorted_by_name(map->dso, map->type))
360                 dso__sort_by_name(map->dso, map->type);
361
362         return dso__find_symbol_by_name(map->dso, map->type, name);
363 }
364
365 struct map *map__clone(struct map *from)
366 {
367         struct map *map = memdup(from, sizeof(*map));
368
369         if (map != NULL) {
370                 refcount_set(&map->refcnt, 1);
371                 RB_CLEAR_NODE(&map->rb_node);
372                 dso__get(map->dso);
373                 map->groups = NULL;
374         }
375
376         return map;
377 }
378
379 int map__overlap(struct map *l, struct map *r)
380 {
381         if (l->start > r->start) {
382                 struct map *t = l;
383                 l = r;
384                 r = t;
385         }
386
387         if (l->end > r->start)
388                 return 1;
389
390         return 0;
391 }
392
393 size_t map__fprintf(struct map *map, FILE *fp)
394 {
395         return fprintf(fp, " %" PRIx64 "-%" PRIx64 " %" PRIx64 " %s\n",
396                        map->start, map->end, map->pgoff, map->dso->name);
397 }
398
399 size_t map__fprintf_dsoname(struct map *map, FILE *fp)
400 {
401         const char *dsoname = "[unknown]";
402
403         if (map && map->dso) {
404                 if (symbol_conf.show_kernel_path && map->dso->long_name)
405                         dsoname = map->dso->long_name;
406                 else
407                         dsoname = map->dso->name;
408         }
409
410         return fprintf(fp, "%s", dsoname);
411 }
412
413 int map__fprintf_srcline(struct map *map, u64 addr, const char *prefix,
414                          FILE *fp)
415 {
416         char *srcline;
417         int ret = 0;
418
419         if (map && map->dso) {
420                 srcline = get_srcline(map->dso,
421                                       map__rip_2objdump(map, addr), NULL,
422                                       true, true);
423                 if (srcline != SRCLINE_UNKNOWN)
424                         ret = fprintf(fp, "%s%s", prefix, srcline);
425                 free_srcline(srcline);
426         }
427         return ret;
428 }
429
430 /**
431  * map__rip_2objdump - convert symbol start address to objdump address.
432  * @map: memory map
433  * @rip: symbol start address
434  *
435  * objdump wants/reports absolute IPs for ET_EXEC, and RIPs for ET_DYN.
436  * map->dso->adjust_symbols==1 for ET_EXEC-like cases except ET_REL which is
437  * relative to section start.
438  *
439  * Return: Address suitable for passing to "objdump --start-address="
440  */
441 u64 map__rip_2objdump(struct map *map, u64 rip)
442 {
443         if (!map->dso->adjust_symbols)
444                 return rip;
445
446         if (map->dso->rel)
447                 return rip - map->pgoff;
448
449         /*
450          * kernel modules also have DSO_TYPE_USER in dso->kernel,
451          * but all kernel modules are ET_REL, so won't get here.
452          */
453         if (map->dso->kernel == DSO_TYPE_USER)
454                 return rip + map->dso->text_offset;
455
456         return map->unmap_ip(map, rip) - map->reloc;
457 }
458
459 /**
460  * map__objdump_2mem - convert objdump address to a memory address.
461  * @map: memory map
462  * @ip: objdump address
463  *
464  * Closely related to map__rip_2objdump(), this function takes an address from
465  * objdump and converts it to a memory address.  Note this assumes that @map
466  * contains the address.  To be sure the result is valid, check it forwards
467  * e.g. map__rip_2objdump(map->map_ip(map, map__objdump_2mem(map, ip))) == ip
468  *
469  * Return: Memory address.
470  */
471 u64 map__objdump_2mem(struct map *map, u64 ip)
472 {
473         if (!map->dso->adjust_symbols)
474                 return map->unmap_ip(map, ip);
475
476         if (map->dso->rel)
477                 return map->unmap_ip(map, ip + map->pgoff);
478
479         /*
480          * kernel modules also have DSO_TYPE_USER in dso->kernel,
481          * but all kernel modules are ET_REL, so won't get here.
482          */
483         if (map->dso->kernel == DSO_TYPE_USER)
484                 return map->unmap_ip(map, ip - map->dso->text_offset);
485
486         return ip + map->reloc;
487 }
488
489 static void maps__init(struct maps *maps)
490 {
491         maps->entries = RB_ROOT;
492         pthread_rwlock_init(&maps->lock, NULL);
493 }
494
495 void map_groups__init(struct map_groups *mg, struct machine *machine)
496 {
497         int i;
498         for (i = 0; i < MAP__NR_TYPES; ++i) {
499                 maps__init(&mg->maps[i]);
500         }
501         mg->machine = machine;
502         refcount_set(&mg->refcnt, 1);
503 }
504
505 static void __maps__purge(struct maps *maps)
506 {
507         struct rb_root *root = &maps->entries;
508         struct rb_node *next = rb_first(root);
509
510         while (next) {
511                 struct map *pos = rb_entry(next, struct map, rb_node);
512
513                 next = rb_next(&pos->rb_node);
514                 rb_erase_init(&pos->rb_node, root);
515                 map__put(pos);
516         }
517 }
518
519 static void maps__exit(struct maps *maps)
520 {
521         pthread_rwlock_wrlock(&maps->lock);
522         __maps__purge(maps);
523         pthread_rwlock_unlock(&maps->lock);
524 }
525
526 void map_groups__exit(struct map_groups *mg)
527 {
528         int i;
529
530         for (i = 0; i < MAP__NR_TYPES; ++i)
531                 maps__exit(&mg->maps[i]);
532 }
533
534 bool map_groups__empty(struct map_groups *mg)
535 {
536         int i;
537
538         for (i = 0; i < MAP__NR_TYPES; ++i) {
539                 if (maps__first(&mg->maps[i]))
540                         return false;
541         }
542
543         return true;
544 }
545
546 struct map_groups *map_groups__new(struct machine *machine)
547 {
548         struct map_groups *mg = malloc(sizeof(*mg));
549
550         if (mg != NULL)
551                 map_groups__init(mg, machine);
552
553         return mg;
554 }
555
556 void map_groups__delete(struct map_groups *mg)
557 {
558         map_groups__exit(mg);
559         free(mg);
560 }
561
562 void map_groups__put(struct map_groups *mg)
563 {
564         if (mg && refcount_dec_and_test(&mg->refcnt))
565                 map_groups__delete(mg);
566 }
567
568 struct symbol *map_groups__find_symbol(struct map_groups *mg,
569                                        enum map_type type, u64 addr,
570                                        struct map **mapp)
571 {
572         struct map *map = map_groups__find(mg, type, addr);
573
574         /* Ensure map is loaded before using map->map_ip */
575         if (map != NULL && map__load(map) >= 0) {
576                 if (mapp != NULL)
577                         *mapp = map;
578                 return map__find_symbol(map, map->map_ip(map, addr));
579         }
580
581         return NULL;
582 }
583
584 struct symbol *maps__find_symbol_by_name(struct maps *maps, const char *name,
585                                          struct map **mapp)
586 {
587         struct symbol *sym;
588         struct rb_node *nd;
589
590         pthread_rwlock_rdlock(&maps->lock);
591
592         for (nd = rb_first(&maps->entries); nd; nd = rb_next(nd)) {
593                 struct map *pos = rb_entry(nd, struct map, rb_node);
594
595                 sym = map__find_symbol_by_name(pos, name);
596
597                 if (sym == NULL)
598                         continue;
599                 if (mapp != NULL)
600                         *mapp = pos;
601                 goto out;
602         }
603
604         sym = NULL;
605 out:
606         pthread_rwlock_unlock(&maps->lock);
607         return sym;
608 }
609
610 struct symbol *map_groups__find_symbol_by_name(struct map_groups *mg,
611                                                enum map_type type,
612                                                const char *name,
613                                                struct map **mapp)
614 {
615         struct symbol *sym = maps__find_symbol_by_name(&mg->maps[type], name, mapp);
616
617         return sym;
618 }
619
620 int map_groups__find_ams(struct addr_map_symbol *ams)
621 {
622         if (ams->addr < ams->map->start || ams->addr >= ams->map->end) {
623                 if (ams->map->groups == NULL)
624                         return -1;
625                 ams->map = map_groups__find(ams->map->groups, ams->map->type,
626                                             ams->addr);
627                 if (ams->map == NULL)
628                         return -1;
629         }
630
631         ams->al_addr = ams->map->map_ip(ams->map, ams->addr);
632         ams->sym = map__find_symbol(ams->map, ams->al_addr);
633
634         return ams->sym ? 0 : -1;
635 }
636
637 static size_t maps__fprintf(struct maps *maps, FILE *fp)
638 {
639         size_t printed = 0;
640         struct rb_node *nd;
641
642         pthread_rwlock_rdlock(&maps->lock);
643
644         for (nd = rb_first(&maps->entries); nd; nd = rb_next(nd)) {
645                 struct map *pos = rb_entry(nd, struct map, rb_node);
646                 printed += fprintf(fp, "Map:");
647                 printed += map__fprintf(pos, fp);
648                 if (verbose > 2) {
649                         printed += dso__fprintf(pos->dso, pos->type, fp);
650                         printed += fprintf(fp, "--\n");
651                 }
652         }
653
654         pthread_rwlock_unlock(&maps->lock);
655
656         return printed;
657 }
658
659 size_t __map_groups__fprintf_maps(struct map_groups *mg, enum map_type type,
660                                   FILE *fp)
661 {
662         size_t printed = fprintf(fp, "%s:\n", map_type__name[type]);
663         return printed += maps__fprintf(&mg->maps[type], fp);
664 }
665
666 size_t map_groups__fprintf(struct map_groups *mg, FILE *fp)
667 {
668         size_t printed = 0, i;
669         for (i = 0; i < MAP__NR_TYPES; ++i)
670                 printed += __map_groups__fprintf_maps(mg, i, fp);
671         return printed;
672 }
673
674 static void __map_groups__insert(struct map_groups *mg, struct map *map)
675 {
676         __maps__insert(&mg->maps[map->type], map);
677         map->groups = mg;
678 }
679
680 static int maps__fixup_overlappings(struct maps *maps, struct map *map, FILE *fp)
681 {
682         struct rb_root *root;
683         struct rb_node *next;
684         int err = 0;
685
686         pthread_rwlock_wrlock(&maps->lock);
687
688         root = &maps->entries;
689         next = rb_first(root);
690
691         while (next) {
692                 struct map *pos = rb_entry(next, struct map, rb_node);
693                 next = rb_next(&pos->rb_node);
694
695                 if (!map__overlap(pos, map))
696                         continue;
697
698                 if (verbose >= 2) {
699
700                         if (use_browser) {
701                                 pr_warning("overlapping maps in %s "
702                                            "(disable tui for more info)\n",
703                                            map->dso->name);
704                         } else {
705                                 fputs("overlapping maps:\n", fp);
706                                 map__fprintf(map, fp);
707                                 map__fprintf(pos, fp);
708                         }
709                 }
710
711                 rb_erase_init(&pos->rb_node, root);
712                 /*
713                  * Now check if we need to create new maps for areas not
714                  * overlapped by the new map:
715                  */
716                 if (map->start > pos->start) {
717                         struct map *before = map__clone(pos);
718
719                         if (before == NULL) {
720                                 err = -ENOMEM;
721                                 goto put_map;
722                         }
723
724                         before->end = map->start;
725                         __map_groups__insert(pos->groups, before);
726                         if (verbose >= 2 && !use_browser)
727                                 map__fprintf(before, fp);
728                         map__put(before);
729                 }
730
731                 if (map->end < pos->end) {
732                         struct map *after = map__clone(pos);
733
734                         if (after == NULL) {
735                                 err = -ENOMEM;
736                                 goto put_map;
737                         }
738
739                         after->start = map->end;
740                         after->pgoff += map->end - pos->start;
741                         assert(pos->map_ip(pos, map->end) == after->map_ip(after, map->end));
742                         __map_groups__insert(pos->groups, after);
743                         if (verbose >= 2 && !use_browser)
744                                 map__fprintf(after, fp);
745                         map__put(after);
746                 }
747 put_map:
748                 map__put(pos);
749
750                 if (err)
751                         goto out;
752         }
753
754         err = 0;
755 out:
756         pthread_rwlock_unlock(&maps->lock);
757         return err;
758 }
759
760 int map_groups__fixup_overlappings(struct map_groups *mg, struct map *map,
761                                    FILE *fp)
762 {
763         return maps__fixup_overlappings(&mg->maps[map->type], map, fp);
764 }
765
766 /*
767  * XXX This should not really _copy_ te maps, but refcount them.
768  */
769 int map_groups__clone(struct thread *thread,
770                       struct map_groups *parent, enum map_type type)
771 {
772         struct map_groups *mg = thread->mg;
773         int err = -ENOMEM;
774         struct map *map;
775         struct maps *maps = &parent->maps[type];
776
777         pthread_rwlock_rdlock(&maps->lock);
778
779         for (map = maps__first(maps); map; map = map__next(map)) {
780                 struct map *new = map__clone(map);
781                 if (new == NULL)
782                         goto out_unlock;
783
784                 err = unwind__prepare_access(thread, new, NULL);
785                 if (err)
786                         goto out_unlock;
787
788                 map_groups__insert(mg, new);
789                 map__put(new);
790         }
791
792         err = 0;
793 out_unlock:
794         pthread_rwlock_unlock(&maps->lock);
795         return err;
796 }
797
798 static void __maps__insert(struct maps *maps, struct map *map)
799 {
800         struct rb_node **p = &maps->entries.rb_node;
801         struct rb_node *parent = NULL;
802         const u64 ip = map->start;
803         struct map *m;
804
805         while (*p != NULL) {
806                 parent = *p;
807                 m = rb_entry(parent, struct map, rb_node);
808                 if (ip < m->start)
809                         p = &(*p)->rb_left;
810                 else
811                         p = &(*p)->rb_right;
812         }
813
814         rb_link_node(&map->rb_node, parent, p);
815         rb_insert_color(&map->rb_node, &maps->entries);
816         map__get(map);
817 }
818
819 void maps__insert(struct maps *maps, struct map *map)
820 {
821         pthread_rwlock_wrlock(&maps->lock);
822         __maps__insert(maps, map);
823         pthread_rwlock_unlock(&maps->lock);
824 }
825
826 static void __maps__remove(struct maps *maps, struct map *map)
827 {
828         rb_erase_init(&map->rb_node, &maps->entries);
829         map__put(map);
830 }
831
832 void maps__remove(struct maps *maps, struct map *map)
833 {
834         pthread_rwlock_wrlock(&maps->lock);
835         __maps__remove(maps, map);
836         pthread_rwlock_unlock(&maps->lock);
837 }
838
839 struct map *maps__find(struct maps *maps, u64 ip)
840 {
841         struct rb_node **p, *parent = NULL;
842         struct map *m;
843
844         pthread_rwlock_rdlock(&maps->lock);
845
846         p = &maps->entries.rb_node;
847         while (*p != NULL) {
848                 parent = *p;
849                 m = rb_entry(parent, struct map, rb_node);
850                 if (ip < m->start)
851                         p = &(*p)->rb_left;
852                 else if (ip >= m->end)
853                         p = &(*p)->rb_right;
854                 else
855                         goto out;
856         }
857
858         m = NULL;
859 out:
860         pthread_rwlock_unlock(&maps->lock);
861         return m;
862 }
863
864 struct map *maps__first(struct maps *maps)
865 {
866         struct rb_node *first = rb_first(&maps->entries);
867
868         if (first)
869                 return rb_entry(first, struct map, rb_node);
870         return NULL;
871 }
872
873 struct map *map__next(struct map *map)
874 {
875         struct rb_node *next = rb_next(&map->rb_node);
876
877         if (next)
878                 return rb_entry(next, struct map, rb_node);
879         return NULL;
880 }
881
882 struct kmap *map__kmap(struct map *map)
883 {
884         if (!map->dso || !map->dso->kernel) {
885                 pr_err("Internal error: map__kmap with a non-kernel map\n");
886                 return NULL;
887         }
888         return (struct kmap *)(map + 1);
889 }
890
891 struct map_groups *map__kmaps(struct map *map)
892 {
893         struct kmap *kmap = map__kmap(map);
894
895         if (!kmap || !kmap->kmaps) {
896                 pr_err("Internal error: map__kmaps with a non-kernel map\n");
897                 return NULL;
898         }
899         return kmap->kmaps;
900 }