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