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