2 * Copyright (c) 2010 The FreeBSD Foundation
3 * Copyright (c) 2008 John Birrell (jb@freebsd.org)
6 * Portions of this software were developed by Rui Paulo under sponsorship
7 * from the FreeBSD Foundation.
9 * Redistribution and use in source and binary forms, with or without
10 * modification, are permitted provided that the following conditions
12 * 1. Redistributions of source code must retain the above copyright
13 * notice, this list of conditions and the following disclaimer.
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
18 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33 #include <sys/types.h>
49 extern char *__cxa_demangle(const char *, char *, size_t *, int *);
51 static void proc_rdl2prmap(rd_loadobj_t *, prmap_t *);
54 demangle(const char *symbol, char *buf, size_t len)
62 dembuf = __cxa_demangle(symbol, dembuf, &demlen, NULL);
65 strlcpy(buf, dembuf, len);
70 strlcpy(buf, symbol, len);
74 proc_rdl2prmap(rd_loadobj_t *rdl, prmap_t *map)
76 map->pr_vaddr = rdl->rdl_saddr;
77 map->pr_size = rdl->rdl_eaddr - rdl->rdl_saddr;
78 map->pr_offset = rdl->rdl_offset;
80 if (rdl->rdl_prot & RD_RDL_R)
81 map->pr_mflags |= MA_READ;
82 if (rdl->rdl_prot & RD_RDL_W)
83 map->pr_mflags |= MA_WRITE;
84 if (rdl->rdl_prot & RD_RDL_X)
85 map->pr_mflags |= MA_EXEC;
86 strlcpy(map->pr_mapname, rdl->rdl_path,
87 sizeof(map->pr_mapname));
91 proc_objname(struct proc_handle *p, uintptr_t addr, char *objname,
97 for (i = 0; i < p->nobjs; i++) {
99 if (addr >= rdl->rdl_saddr && addr <= rdl->rdl_eaddr) {
100 strlcpy(objname, rdl->rdl_path, objnamesz);
108 proc_obj2map(struct proc_handle *p, const char *objname)
113 char path[MAXPATHLEN];
115 for (i = 0; i < p->nobjs; i++) {
117 basename_r(rdl->rdl_path, path);
118 if (strcmp(path, objname) == 0) {
119 if ((map = malloc(sizeof(*map))) == NULL)
121 proc_rdl2prmap(rdl, map);
129 proc_iter_objs(struct proc_handle *p, proc_map_f *func, void *cd)
134 char path[MAXPATHLEN];
135 char last[MAXPATHLEN];
139 memset(last, 0, sizeof(last));
140 for (i = 0; i < p->nobjs; i++) {
142 proc_rdl2prmap(rdl, &map);
143 basename_r(rdl->rdl_path, path);
145 * We shouldn't call the callback twice with the same object.
146 * To do that we are assuming the fact that if there are
147 * repeated object names (i.e. different mappings for the
148 * same object) they occur next to each other.
150 if (strcmp(path, last) == 0)
152 (*func)(cd, &map, path);
153 strlcpy(last, path, sizeof(last));
160 proc_addr2map(struct proc_handle *p, uintptr_t addr)
166 struct kinfo_vmentry *kves, *kve;
169 * If we don't have a cache of listed objects, we need to query
173 if ((kves = kinfo_getvmmap(p->pid, &cnt)) == NULL)
175 for (i = 0; i < (size_t)cnt; i++) {
177 if (kve->kve_type == KVME_TYPE_VNODE)
179 if (addr >= kve->kve_start && addr <= kve->kve_end) {
180 if ((map = malloc(sizeof(*map))) == NULL) {
184 map->pr_vaddr = kve->kve_start;
185 map->pr_size = kve->kve_end - kve->kve_start;
186 map->pr_offset = kve->kve_offset;
188 if (kve->kve_protection & KVME_PROT_READ)
189 map->pr_mflags |= MA_READ;
190 if (kve->kve_protection & KVME_PROT_WRITE)
191 map->pr_mflags |= MA_WRITE;
192 if (kve->kve_protection & KVME_PROT_EXEC)
193 map->pr_mflags |= MA_EXEC;
194 if (kve->kve_flags & KVME_FLAG_COW)
195 map->pr_mflags |= MA_COW;
196 if (kve->kve_flags & KVME_FLAG_NEEDS_COPY)
197 map->pr_mflags |= MA_NEEDS_COPY;
198 if (kve->kve_flags & KVME_FLAG_NOCOREDUMP)
199 map->pr_mflags |= MA_NOCOREDUMP;
200 strlcpy(map->pr_mapname, kves[lastvn].kve_path,
201 sizeof(map->pr_mapname));
210 for (i = 0; i < p->nobjs; i++) {
212 if (addr >= rdl->rdl_saddr && addr <= rdl->rdl_eaddr) {
213 if ((map = malloc(sizeof(*map))) == NULL)
215 proc_rdl2prmap(rdl, map);
223 proc_addr2sym(struct proc_handle *p, uintptr_t addr, char *name,
224 size_t namesz, GElf_Sym *symcopy)
227 Elf_Scn *scn, *dynsymscn = NULL, *symtabscn = NULL;
237 unsigned long symtabstridx = 0, dynsymstridx = 0;
239 if ((map = proc_addr2map(p, addr)) == NULL)
241 if (!map->pr_mapname || (fd = open(map->pr_mapname, O_RDONLY, 0)) < 0) {
242 warn("ERROR: open %s failed", map->pr_mapname);
245 if ((e = elf_begin(fd, ELF_C_READ, NULL)) == NULL) {
246 warn("ERROR: elf_begin() failed");
249 if (gelf_getehdr(e, &ehdr) == NULL) {
250 warn("ERROR: gelf_getehdr() failed");
254 * Find the index of the STRTAB and SYMTAB sections to locate
258 while ((scn = elf_nextscn(e, scn)) != NULL) {
259 gelf_getshdr(scn, &shdr);
260 switch (shdr.sh_type) {
263 symtabstridx = shdr.sh_link;
267 dynsymstridx = shdr.sh_link;
274 * Iterate over the Dynamic Symbols table to find the symbol.
275 * Then look up the string name in STRTAB (.dynstr)
277 if ((data = elf_getdata(dynsymscn, NULL)) == NULL) {
278 DPRINTF("ERROR: elf_getdata() failed");
282 while (gelf_getsym(data, i++, &sym) != NULL) {
284 * Calculate the address mapped to the virtual memory
287 rsym = map->pr_vaddr + sym.st_value;
288 if (addr >= rsym && addr <= (rsym + sym.st_size)) {
289 s = elf_strptr(e, dynsymstridx, sym.st_name);
291 if (s[0] == '_' && s[1] == 'Z' && s[2])
292 demangle(s, name, namesz);
294 strlcpy(name, s, namesz);
295 memcpy(symcopy, &sym, sizeof(sym));
297 * DTrace expects the st_value to contain
298 * only the address relative to the start of
301 symcopy->st_value = rsym;
308 * Iterate over the Symbols Table to find the symbol.
309 * Then look up the string name in STRTAB (.dynstr)
311 if (symtabscn == NULL)
313 if ((data = elf_getdata(symtabscn, NULL)) == NULL) {
314 DPRINTF("ERROR: elf_getdata() failed");
318 while (gelf_getsym(data, i++, &sym) != NULL) {
320 * Calculate the address mapped to the virtual memory
323 if (ehdr.e_type != ET_EXEC)
324 rsym = map->pr_vaddr + sym.st_value;
327 if (addr >= rsym && addr <= (rsym + sym.st_size)) {
328 s = elf_strptr(e, symtabstridx, sym.st_name);
330 if (s[0] == '_' && s[1] == 'Z' && s[2])
331 demangle(s, name, namesz);
333 strlcpy(name, s, namesz);
334 memcpy(symcopy, &sym, sizeof(sym));
336 * DTrace expects the st_value to contain
337 * only the address relative to the start of
340 symcopy->st_value = rsym;
357 proc_name2map(struct proc_handle *p, const char *name)
362 char tmppath[MAXPATHLEN];
363 struct kinfo_vmentry *kves, *kve;
367 * If we haven't iterated over the list of loaded objects,
368 * librtld_db isn't yet initialized and it's very likely
369 * that librtld_db called us. We need to do the heavy
370 * lifting here to find the symbol librtld_db is looking for.
373 if ((kves = kinfo_getvmmap(proc_getpid(p), &cnt)) == NULL)
375 for (i = 0; i < (size_t)cnt; i++) {
377 basename_r(kve->kve_path, tmppath);
378 if (strcmp(tmppath, name) == 0) {
379 map = proc_addr2map(p, kve->kve_start);
387 if (name == NULL || strcmp(name, "a.out") == 0) {
388 map = proc_addr2map(p, p->rdobjs[0].rdl_saddr);
391 for (i = 0; i < p->nobjs; i++) {
393 basename_r(rdl->rdl_path, tmppath);
394 if (strcmp(tmppath, name) == 0) {
395 if ((map = malloc(sizeof(*map))) == NULL)
397 proc_rdl2prmap(rdl, map);
406 proc_name2sym(struct proc_handle *p, const char *object, const char *symbol,
410 Elf_Scn *scn, *dynsymscn = NULL, *symtabscn = NULL;
419 unsigned long symtabstridx = 0, dynsymstridx = 0;
421 if ((map = proc_name2map(p, object)) == NULL) {
422 DPRINTF("ERROR: couldn't find object %s", object);
425 if ((fd = open(map->pr_mapname, O_RDONLY, 0)) < 0) {
426 DPRINTF("ERROR: open %s failed", map->pr_mapname);
429 if ((e = elf_begin(fd, ELF_C_READ, NULL)) == NULL) {
430 warn("ERROR: elf_begin() failed");
433 if (gelf_getehdr(e, &ehdr) == NULL) {
434 warn("ERROR: gelf_getehdr() failed");
438 * Find the index of the STRTAB and SYMTAB sections to locate
442 while ((scn = elf_nextscn(e, scn)) != NULL) {
443 gelf_getshdr(scn, &shdr);
444 switch (shdr.sh_type) {
447 symtabstridx = shdr.sh_link;
451 dynsymstridx = shdr.sh_link;
458 * Iterate over the Dynamic Symbols table to find the symbol.
459 * Then look up the string name in STRTAB (.dynstr)
461 if ((data = elf_getdata(dynsymscn, NULL))) {
462 DPRINTF("ERROR: elf_getdata() failed");
464 while (gelf_getsym(data, i++, &sym) != NULL) {
465 s = elf_strptr(e, dynsymstridx, sym.st_name);
466 if (s && strcmp(s, symbol) == 0) {
467 memcpy(symcopy, &sym, sizeof(sym));
468 symcopy->st_value = map->pr_vaddr + sym.st_value;
475 * Iterate over the Symbols Table to find the symbol.
476 * Then look up the string name in STRTAB (.dynstr)
478 if (symtabscn == NULL)
480 if ((data = elf_getdata(symtabscn, NULL))) {
482 while (gelf_getsym(data, i++, &sym) != NULL) {
483 s = elf_strptr(e, symtabstridx, sym.st_name);
484 if (s && strcmp(s, symbol) == 0) {
485 memcpy(symcopy, &sym, sizeof(sym));
504 proc_iter_symbyaddr(struct proc_handle *p, const char *object, int which,
505 int mask, proc_sym_f *func, void *cd)
510 Elf_Scn *scn, *foundscn = NULL;
514 unsigned long stridx = -1;
518 if ((map = proc_name2map(p, object)) == NULL)
520 if ((fd = open(map->pr_mapname, O_RDONLY)) < 0) {
521 warn("ERROR: open %s failed", map->pr_mapname);
524 if ((e = elf_begin(fd, ELF_C_READ, NULL)) == NULL) {
525 warn("ERROR: elf_begin() failed");
529 * Find the section we are looking for.
532 while ((scn = elf_nextscn(e, scn)) != NULL) {
533 gelf_getshdr(scn, &shdr);
534 if (which == PR_SYMTAB &&
535 shdr.sh_type == SHT_SYMTAB) {
538 } else if (which == PR_DYNSYM &&
539 shdr.sh_type == SHT_DYNSYM) {
546 stridx = shdr.sh_link;
547 if ((data = elf_getdata(foundscn, NULL)) == NULL) {
548 DPRINTF("ERROR: elf_getdata() failed");
552 while (gelf_getsym(data, i++, &sym) != NULL) {
553 if (GELF_ST_BIND(sym.st_info) == STB_LOCAL &&
554 (mask & BIND_LOCAL) == 0)
556 if (GELF_ST_BIND(sym.st_info) == STB_GLOBAL &&
557 (mask & BIND_GLOBAL) == 0)
559 if (GELF_ST_BIND(sym.st_info) == STB_WEAK &&
560 (mask & BIND_WEAK) == 0)
562 if (GELF_ST_TYPE(sym.st_info) == STT_NOTYPE &&
563 (mask & TYPE_NOTYPE) == 0)
565 if (GELF_ST_TYPE(sym.st_info) == STT_OBJECT &&
566 (mask & TYPE_OBJECT) == 0)
568 if (GELF_ST_TYPE(sym.st_info) == STT_FUNC &&
569 (mask & TYPE_FUNC) == 0)
571 if (GELF_ST_TYPE(sym.st_info) == STT_SECTION &&
572 (mask & TYPE_SECTION) == 0)
574 if (GELF_ST_TYPE(sym.st_info) == STT_FILE &&
575 (mask & TYPE_FILE) == 0)
577 s = elf_strptr(e, stridx, sym.st_name);
578 sym.st_value += map->pr_vaddr;
579 (*func)(cd, &sym, s);