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