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
is_android_lib(const char * filename)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
replace_android_lib(const char * filename,char * newfilename)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
map__init(struct map * map,u64 start,u64 end,u64 pgoff,struct dso * dso)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
map__new(struct machine * machine,u64 start,u64 len,u64 pgoff,struct dso_id * id,u32 prot,u32 flags,struct build_id * bid,char * filename,struct thread * thread)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 */
map__new2(u64 start,struct dso * dso)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
__map__is_kernel(const struct map * map)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
__map__is_extra_kernel_map(const struct map * map)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
__map__is_bpf_prog(const struct map * map)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
__map__is_bpf_image(const struct map * map)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
__map__is_ool(const struct map * map)274 bool __map__is_ool(const struct map *map)
275 {
276 return map->dso && map->dso->binary_type == DSO_BINARY_TYPE__OOL;
277 }
278
map__has_symbols(const struct map * map)279 bool map__has_symbols(const struct map *map)
280 {
281 return dso__has_symbols(map->dso);
282 }
283
map__exit(struct map * map)284 static void map__exit(struct map *map)
285 {
286 BUG_ON(refcount_read(&map->refcnt) != 0);
287 dso__zput(map->dso);
288 }
289
map__delete(struct map * map)290 void map__delete(struct map *map)
291 {
292 map__exit(map);
293 free(map);
294 }
295
map__put(struct map * map)296 void map__put(struct map *map)
297 {
298 if (map && refcount_dec_and_test(&map->refcnt))
299 map__delete(map);
300 }
301
map__fixup_start(struct map * map)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
map__fixup_end(struct map * map)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
map__load(struct map * map)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
map__find_symbol(struct map * map,u64 addr)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
map__find_symbol_by_name(struct map * map,const char * name)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
map__clone(struct map * from)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
map__fprintf(struct map * map,FILE * fp)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
map__fprintf_dsoname(struct map * map,FILE * fp)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
map__srcline(struct map * map,u64 addr,struct symbol * sym)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
map__fprintf_srcline(struct map * map,u64 addr,const char * prefix,FILE * fp)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
srccode_state_free(struct srccode_state * state)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 */
map__rip_2objdump(struct map * map,u64 rip)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 */
map__objdump_2mem(struct map * map,u64 ip)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
maps__init(struct maps * maps,struct machine * machine)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
__maps__free_maps_by_name(struct maps * maps)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
maps__insert(struct maps * maps,struct map * map)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
__maps__remove(struct maps * maps,struct map * map)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
maps__remove(struct maps * maps,struct map * map)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
__maps__purge(struct maps * maps)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
maps__exit(struct maps * maps)616 void maps__exit(struct maps *maps)
617 {
618 down_write(&maps->lock);
619 __maps__purge(maps);
620 up_write(&maps->lock);
621 }
622
maps__empty(struct maps * maps)623 bool maps__empty(struct maps *maps)
624 {
625 return !maps__first(maps);
626 }
627
maps__new(struct machine * machine)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
maps__delete(struct maps * maps)638 void maps__delete(struct maps *maps)
639 {
640 maps__exit(maps);
641 unwind__finish_access(maps);
642 free(maps);
643 }
644
maps__put(struct maps * maps)645 void maps__put(struct maps *maps)
646 {
647 if (maps && refcount_dec_and_test(&maps->refcnt))
648 maps__delete(maps);
649 }
650
maps__find_symbol(struct maps * maps,u64 addr,struct map ** mapp)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
map__contains_symbol(struct map * map,struct symbol * sym)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
maps__find_symbol_by_name(struct maps * maps,const char * name,struct map ** mapp)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
maps__find_ams(struct maps * maps,struct addr_map_symbol * ams)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
maps__fprintf(struct maps * maps,FILE * fp)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
maps__fixup_overlappings(struct maps * maps,struct map * map,FILE * fp)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 */
maps__clone(struct thread * thread,struct maps * parent)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
__maps__insert(struct maps * maps,struct map * map)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
maps__find(struct maps * maps,u64 ip)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
maps__first(struct maps * maps)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
__map__next(struct map * map)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
map__next(struct map * map)933 struct map *map__next(struct map *map)
934 {
935 return map ? __map__next(map) : NULL;
936 }
937
__map__kmap(struct map * map)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
map__kmap(struct map * map)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
map__kmaps(struct map * map)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 }
964