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