12c633af4SJohn Birrell /*- 28eb20f36SRui Paulo * Copyright (c) 2010 The FreeBSD Foundation 32c633af4SJohn Birrell * Copyright (c) 2008 John Birrell (jb@freebsd.org) 42c633af4SJohn Birrell * All rights reserved. 52c633af4SJohn Birrell * 68eb20f36SRui Paulo * Portions of this software were developed by Rui Paulo under sponsorship 78eb20f36SRui Paulo * from the FreeBSD Foundation. 88eb20f36SRui Paulo * 92c633af4SJohn Birrell * Redistribution and use in source and binary forms, with or without 102c633af4SJohn Birrell * modification, are permitted provided that the following conditions 112c633af4SJohn Birrell * are met: 122c633af4SJohn Birrell * 1. Redistributions of source code must retain the above copyright 132c633af4SJohn Birrell * notice, this list of conditions and the following disclaimer. 142c633af4SJohn Birrell * 2. Redistributions in binary form must reproduce the above copyright 152c633af4SJohn Birrell * notice, this list of conditions and the following disclaimer in the 162c633af4SJohn Birrell * documentation and/or other materials provided with the distribution. 172c633af4SJohn Birrell * 182c633af4SJohn Birrell * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 192c633af4SJohn Birrell * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 202c633af4SJohn Birrell * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 212c633af4SJohn Birrell * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 222c633af4SJohn Birrell * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 232c633af4SJohn Birrell * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 242c633af4SJohn Birrell * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 252c633af4SJohn Birrell * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 262c633af4SJohn Birrell * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 272c633af4SJohn Birrell * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 282c633af4SJohn Birrell * SUCH DAMAGE. 292c633af4SJohn Birrell */ 302c633af4SJohn Birrell 31540cc663SMark Johnston #include <sys/cdefs.h> 32540cc663SMark Johnston __FBSDID("$FreeBSD$"); 33540cc663SMark Johnston 348eb20f36SRui Paulo #include <sys/types.h> 3541da933cSMark Johnston #ifndef NO_CTF 3641da933cSMark Johnston #include <sys/ctf.h> 3741da933cSMark Johnston #include <sys/ctf_api.h> 3841da933cSMark Johnston #endif 398eb20f36SRui Paulo #include <sys/user.h> 408eb20f36SRui Paulo 418eb20f36SRui Paulo #include <assert.h> 428eb20f36SRui Paulo #include <err.h> 438eb20f36SRui Paulo #include <fcntl.h> 44540cc663SMark Johnston #include <libgen.h> 45540cc663SMark Johnston #include <stdio.h> 46540cc663SMark Johnston #include <stdlib.h> 478eb20f36SRui Paulo #include <string.h> 488eb20f36SRui Paulo #include <unistd.h> 4941da933cSMark Johnston #ifndef NO_CTF 5041da933cSMark Johnston #include <libctf.h> 5141da933cSMark Johnston #endif 528eb20f36SRui Paulo #include <libutil.h> 538eb20f36SRui Paulo 54*76c6531cSMark Johnston #include "crc32.h" 558eb20f36SRui Paulo #include "_libproc.h" 568eb20f36SRui Paulo 57*76c6531cSMark Johnston #define PATH_DEBUG_DIR "/usr/lib/debug" 58*76c6531cSMark Johnston 5941da933cSMark Johnston #ifdef NO_CTF 6041da933cSMark Johnston typedef struct ctf_file ctf_file_t; 6141da933cSMark Johnston #endif 6241da933cSMark Johnston 633d12a343SDimitry Andric #ifndef NO_CXA_DEMANGLE 64cd906041SRui Paulo extern char *__cxa_demangle(const char *, char *, size_t *, int *); 653d12a343SDimitry Andric #endif /* NO_CXA_DEMANGLE */ 66cd906041SRui Paulo 678eb20f36SRui Paulo static void proc_rdl2prmap(rd_loadobj_t *, prmap_t *); 688eb20f36SRui Paulo 69*76c6531cSMark Johnston static int 70*76c6531cSMark Johnston crc32_file(int fd, uint32_t *crc) 71*76c6531cSMark Johnston { 72*76c6531cSMark Johnston uint8_t buf[PAGE_SIZE], *p; 73*76c6531cSMark Johnston size_t n; 74*76c6531cSMark Johnston 75*76c6531cSMark Johnston *crc = ~0; 76*76c6531cSMark Johnston while ((n = read(fd, buf, sizeof(buf))) > 0) { 77*76c6531cSMark Johnston p = &buf[0]; 78*76c6531cSMark Johnston while (n-- > 0) 79*76c6531cSMark Johnston *crc = crc32_tab[(*crc ^ *p++) & 0xff] ^ (*crc >> 8); 80*76c6531cSMark Johnston } 81*76c6531cSMark Johnston *crc = ~*crc; 82*76c6531cSMark Johnston return (n); 83*76c6531cSMark Johnston } 84*76c6531cSMark Johnston 858eb20f36SRui Paulo static void 86404087ccSRui Paulo demangle(const char *symbol, char *buf, size_t len) 87404087ccSRui Paulo { 883d12a343SDimitry Andric #ifndef NO_CXA_DEMANGLE 89404087ccSRui Paulo char *dembuf; 90404087ccSRui Paulo 913d12a343SDimitry Andric if (symbol[0] == '_' && symbol[1] == 'Z' && symbol[2]) { 9296ec3cddSDimitry Andric dembuf = __cxa_demangle(symbol, NULL, NULL, NULL); 93404087ccSRui Paulo if (!dembuf) 94404087ccSRui Paulo goto fail; 95404087ccSRui Paulo strlcpy(buf, dembuf, len); 96404087ccSRui Paulo free(dembuf); 97404087ccSRui Paulo return; 9896ec3cddSDimitry Andric } 99404087ccSRui Paulo fail: 1003d12a343SDimitry Andric #endif /* NO_CXA_DEMANGLE */ 101404087ccSRui Paulo strlcpy(buf, symbol, len); 102404087ccSRui Paulo } 103404087ccSRui Paulo 104eda94c07SRui Paulo static int 105*76c6531cSMark Johnston open_debug_file(char *path, const char *debugfile, uint32_t crc) 106eda94c07SRui Paulo { 107*76c6531cSMark Johnston size_t n; 108*76c6531cSMark Johnston uint32_t compcrc; 109eda94c07SRui Paulo int fd; 110eda94c07SRui Paulo 111*76c6531cSMark Johnston fd = -1; 112*76c6531cSMark Johnston if ((n = strlcat(path, "/", PATH_MAX)) >= PATH_MAX) 113eda94c07SRui Paulo return (fd); 114*76c6531cSMark Johnston if (strlcat(path, debugfile, PATH_MAX) >= PATH_MAX) 115*76c6531cSMark Johnston goto out; 116*76c6531cSMark Johnston if ((fd = open(path, O_RDONLY | O_CLOEXEC)) < 0) 117*76c6531cSMark Johnston goto out; 118*76c6531cSMark Johnston if (crc32_file(fd, &compcrc) != 0 || crc != compcrc) { 119*76c6531cSMark Johnston DPRINTFX("ERROR: CRC32 mismatch for %s", path); 120*76c6531cSMark Johnston (void)close(fd); 121*76c6531cSMark Johnston fd = -1; 122*76c6531cSMark Johnston } 123*76c6531cSMark Johnston out: 124*76c6531cSMark Johnston path[n] = '\0'; 125*76c6531cSMark Johnston return (fd); 126*76c6531cSMark Johnston } 127*76c6531cSMark Johnston 128*76c6531cSMark Johnston /* 129*76c6531cSMark Johnston * Obtain an ELF descriptor for the specified mapped object. If a GNU debuglink 130*76c6531cSMark Johnston * section is present, a descriptor for the corresponding debug file is 131*76c6531cSMark Johnston * returned. 132*76c6531cSMark Johnston */ 133*76c6531cSMark Johnston static int 134*76c6531cSMark Johnston open_object(prmap_t *map, Elf **elfp, int *fdp) 135*76c6531cSMark Johnston { 136*76c6531cSMark Johnston char path[PATH_MAX]; 137*76c6531cSMark Johnston GElf_Shdr shdr; 138*76c6531cSMark Johnston Elf *e, *e2; 139*76c6531cSMark Johnston Elf_Data *data; 140*76c6531cSMark Johnston Elf_Scn *scn; 141*76c6531cSMark Johnston const char *debugfile, *scnname; 142*76c6531cSMark Johnston size_t ndx; 143*76c6531cSMark Johnston uint32_t crc; 144*76c6531cSMark Johnston int fd, fd2; 145*76c6531cSMark Johnston 146*76c6531cSMark Johnston if ((fd = open(map->pr_mapname, O_RDONLY | O_CLOEXEC)) < 0) { 147*76c6531cSMark Johnston DPRINTF("ERROR: open %s failed", map->pr_mapname); 148*76c6531cSMark Johnston return (-1); 149*76c6531cSMark Johnston } 150*76c6531cSMark Johnston if ((e = elf_begin(fd, ELF_C_READ, NULL)) == NULL) { 151*76c6531cSMark Johnston DPRINTFX("ERROR: elf_begin() failed: %s", elf_errmsg(-1)); 152*76c6531cSMark Johnston goto err; 153*76c6531cSMark Johnston } 154*76c6531cSMark Johnston 155*76c6531cSMark Johnston scn = NULL; 156*76c6531cSMark Johnston while ((scn = elf_nextscn(e, scn)) != NULL) { 157*76c6531cSMark Johnston if (gelf_getshdr(scn, &shdr) != &shdr) { 158*76c6531cSMark Johnston DPRINTFX("ERROR: gelf_getshdr failed: %s", 159*76c6531cSMark Johnston elf_errmsg(-1)); 160*76c6531cSMark Johnston goto err; 161*76c6531cSMark Johnston } 162*76c6531cSMark Johnston if (shdr.sh_type != SHT_PROGBITS) 163*76c6531cSMark Johnston continue; 164*76c6531cSMark Johnston if (elf_getshdrstrndx(e, &ndx) != 0) { 165*76c6531cSMark Johnston DPRINTFX("ERROR: elf_getshdrstrndx failed: %s", 166*76c6531cSMark Johnston elf_errmsg(-1)); 167*76c6531cSMark Johnston goto err; 168*76c6531cSMark Johnston } 169*76c6531cSMark Johnston if ((scnname = elf_strptr(e, ndx, shdr.sh_name)) == NULL) 170*76c6531cSMark Johnston continue; 171*76c6531cSMark Johnston 172*76c6531cSMark Johnston if (strcmp(scnname, ".gnu_debuglink") == 0) 173*76c6531cSMark Johnston break; 174*76c6531cSMark Johnston } 175*76c6531cSMark Johnston if (scn == NULL) 176*76c6531cSMark Johnston goto internal; 177*76c6531cSMark Johnston 178*76c6531cSMark Johnston if ((data = elf_getdata(scn, NULL)) == NULL) { 179*76c6531cSMark Johnston DPRINTFX("ERROR: elf_getdata failed: %s", elf_errmsg(-1)); 180*76c6531cSMark Johnston goto err; 181*76c6531cSMark Johnston } 182*76c6531cSMark Johnston 183*76c6531cSMark Johnston /* 184*76c6531cSMark Johnston * The data contains a null-terminated file name followed by a 4-byte 185*76c6531cSMark Johnston * CRC. 186*76c6531cSMark Johnston */ 187*76c6531cSMark Johnston if (data->d_size < sizeof(crc) + 1) { 188*76c6531cSMark Johnston DPRINTFX("ERROR: debuglink section is too small (%zd bytes)", 189*76c6531cSMark Johnston data->d_size); 190*76c6531cSMark Johnston goto internal; 191*76c6531cSMark Johnston } 192*76c6531cSMark Johnston if (strnlen(data->d_buf, data->d_size) >= data->d_size - sizeof(crc)) { 193*76c6531cSMark Johnston DPRINTFX("ERROR: no null-terminator in gnu_debuglink section"); 194*76c6531cSMark Johnston goto internal; 195*76c6531cSMark Johnston } 196*76c6531cSMark Johnston 197*76c6531cSMark Johnston debugfile = data->d_buf; 198*76c6531cSMark Johnston memcpy(&crc, (char *)data->d_buf + data->d_size - sizeof(crc), 199*76c6531cSMark Johnston sizeof(crc)); 200*76c6531cSMark Johnston 201*76c6531cSMark Johnston /* 202*76c6531cSMark Johnston * Search for the debug file using the algorithm described in the gdb 203*76c6531cSMark Johnston * documentation: 204*76c6531cSMark Johnston * - look in the directory containing the object, 205*76c6531cSMark Johnston * - look in the subdirectory ".debug" of the directory containing the 206*76c6531cSMark Johnston * object, 207*76c6531cSMark Johnston * - look in the global debug directories (currently /usr/lib/debug). 208*76c6531cSMark Johnston */ 209*76c6531cSMark Johnston (void)strlcpy(path, map->pr_mapname, sizeof(path)); 210*76c6531cSMark Johnston (void)dirname(path); 211*76c6531cSMark Johnston 212*76c6531cSMark Johnston if ((fd2 = open_debug_file(path, debugfile, crc)) >= 0) 213*76c6531cSMark Johnston goto external; 214*76c6531cSMark Johnston 215*76c6531cSMark Johnston if (strlcat(path, "/.debug", sizeof(path)) < sizeof(path) && 216*76c6531cSMark Johnston (fd2 = open_debug_file(path, debugfile, crc)) >= 0) 217*76c6531cSMark Johnston goto external; 218*76c6531cSMark Johnston 219*76c6531cSMark Johnston (void)snprintf(path, sizeof(path), PATH_DEBUG_DIR); 220*76c6531cSMark Johnston if (strlcat(path, map->pr_mapname, sizeof(path)) < sizeof(path)) { 221*76c6531cSMark Johnston (void)dirname(path); 222*76c6531cSMark Johnston if ((fd2 = open_debug_file(path, debugfile, crc)) >= 0) 223*76c6531cSMark Johnston goto external; 224*76c6531cSMark Johnston } 225*76c6531cSMark Johnston 226*76c6531cSMark Johnston internal: 227*76c6531cSMark Johnston /* We didn't find a debug file, just return the object's descriptor. */ 228*76c6531cSMark Johnston *elfp = e; 229*76c6531cSMark Johnston *fdp = fd; 230*76c6531cSMark Johnston return (0); 231*76c6531cSMark Johnston 232*76c6531cSMark Johnston external: 233*76c6531cSMark Johnston if ((e2 = elf_begin(fd2, ELF_C_READ, NULL)) == NULL) { 234*76c6531cSMark Johnston DPRINTFX("ERROR: elf_begin failed: %s", elf_errmsg(-1)); 235*76c6531cSMark Johnston (void)close(fd2); 236*76c6531cSMark Johnston goto err; 237*76c6531cSMark Johnston } 238*76c6531cSMark Johnston (void)elf_end(e); 239*76c6531cSMark Johnston (void)close(fd); 240*76c6531cSMark Johnston *elfp = e2; 241*76c6531cSMark Johnston *fdp = fd2; 242*76c6531cSMark Johnston return (0); 243*76c6531cSMark Johnston 244*76c6531cSMark Johnston err: 245*76c6531cSMark Johnston if (e != NULL) 246*76c6531cSMark Johnston (void)elf_end(e); 247*76c6531cSMark Johnston (void)close(fd); 248*76c6531cSMark Johnston return (-1); 249eda94c07SRui Paulo } 250eda94c07SRui Paulo 251404087ccSRui Paulo static void 2528eb20f36SRui Paulo proc_rdl2prmap(rd_loadobj_t *rdl, prmap_t *map) 2538eb20f36SRui Paulo { 2548eb20f36SRui Paulo map->pr_vaddr = rdl->rdl_saddr; 2558eb20f36SRui Paulo map->pr_size = rdl->rdl_eaddr - rdl->rdl_saddr; 2568eb20f36SRui Paulo map->pr_offset = rdl->rdl_offset; 2578eb20f36SRui Paulo map->pr_mflags = 0; 2588eb20f36SRui Paulo if (rdl->rdl_prot & RD_RDL_R) 2598eb20f36SRui Paulo map->pr_mflags |= MA_READ; 2608eb20f36SRui Paulo if (rdl->rdl_prot & RD_RDL_W) 2618eb20f36SRui Paulo map->pr_mflags |= MA_WRITE; 2628eb20f36SRui Paulo if (rdl->rdl_prot & RD_RDL_X) 2638eb20f36SRui Paulo map->pr_mflags |= MA_EXEC; 2648eb20f36SRui Paulo strlcpy(map->pr_mapname, rdl->rdl_path, 2658eb20f36SRui Paulo sizeof(map->pr_mapname)); 2668eb20f36SRui Paulo } 2672c633af4SJohn Birrell 2682c633af4SJohn Birrell char * 2692c633af4SJohn Birrell proc_objname(struct proc_handle *p, uintptr_t addr, char *objname, 2702c633af4SJohn Birrell size_t objnamesz) 2712c633af4SJohn Birrell { 2728eb20f36SRui Paulo size_t i; 2738eb20f36SRui Paulo rd_loadobj_t *rdl; 2748eb20f36SRui Paulo 2758eb20f36SRui Paulo for (i = 0; i < p->nobjs; i++) { 2768eb20f36SRui Paulo rdl = &p->rdobjs[i]; 2771e6b3858SMark Johnston if (addr >= rdl->rdl_saddr && addr < rdl->rdl_eaddr) { 2788eb20f36SRui Paulo strlcpy(objname, rdl->rdl_path, objnamesz); 2798eb20f36SRui Paulo return (objname); 2808eb20f36SRui Paulo } 2818eb20f36SRui Paulo } 2822c633af4SJohn Birrell return (NULL); 2832c633af4SJohn Birrell } 2842c633af4SJohn Birrell 2858eb20f36SRui Paulo prmap_t * 2868eb20f36SRui Paulo proc_obj2map(struct proc_handle *p, const char *objname) 2878eb20f36SRui Paulo { 2888eb20f36SRui Paulo size_t i; 2898eb20f36SRui Paulo prmap_t *map; 2908eb20f36SRui Paulo rd_loadobj_t *rdl; 2918eb20f36SRui Paulo char path[MAXPATHLEN]; 2928eb20f36SRui Paulo 293acc0eea6SMark Johnston rdl = NULL; 2948eb20f36SRui Paulo for (i = 0; i < p->nobjs; i++) { 295acc0eea6SMark Johnston basename_r(p->rdobjs[i].rdl_path, path); 2968eb20f36SRui Paulo if (strcmp(path, objname) == 0) { 297acc0eea6SMark Johnston rdl = &p->rdobjs[i]; 298acc0eea6SMark Johnston break; 299acc0eea6SMark Johnston } 300acc0eea6SMark Johnston } 30124a08d30SMark Johnston if (rdl == NULL) { 30224a08d30SMark Johnston if (strcmp(objname, "a.out") == 0 && p->rdexec != NULL) 303acc0eea6SMark Johnston rdl = p->rdexec; 304acc0eea6SMark Johnston else 305acc0eea6SMark Johnston return (NULL); 30624a08d30SMark Johnston } 307acc0eea6SMark Johnston 3088eb20f36SRui Paulo if ((map = malloc(sizeof(*map))) == NULL) 3098eb20f36SRui Paulo return (NULL); 3108eb20f36SRui Paulo proc_rdl2prmap(rdl, map); 3118eb20f36SRui Paulo return (map); 3128eb20f36SRui Paulo } 3138eb20f36SRui Paulo 3148eb20f36SRui Paulo int 3158eb20f36SRui Paulo proc_iter_objs(struct proc_handle *p, proc_map_f *func, void *cd) 3168eb20f36SRui Paulo { 3178eb20f36SRui Paulo size_t i; 3188eb20f36SRui Paulo rd_loadobj_t *rdl; 3198eb20f36SRui Paulo prmap_t map; 3208eb20f36SRui Paulo char path[MAXPATHLEN]; 3214c74b245SRui Paulo char last[MAXPATHLEN]; 322c7fa6f0bSMark Johnston int error; 3238eb20f36SRui Paulo 3248eb20f36SRui Paulo if (p->nobjs == 0) 3258eb20f36SRui Paulo return (-1); 326c7fa6f0bSMark Johnston 327c7fa6f0bSMark Johnston error = 0; 3284c74b245SRui Paulo memset(last, 0, sizeof(last)); 3298eb20f36SRui Paulo for (i = 0; i < p->nobjs; i++) { 3308eb20f36SRui Paulo rdl = &p->rdobjs[i]; 3318eb20f36SRui Paulo proc_rdl2prmap(rdl, &map); 3328eb20f36SRui Paulo basename_r(rdl->rdl_path, path); 3334c74b245SRui Paulo /* 3344c74b245SRui Paulo * We shouldn't call the callback twice with the same object. 3354c74b245SRui Paulo * To do that we are assuming the fact that if there are 3364c74b245SRui Paulo * repeated object names (i.e. different mappings for the 3374c74b245SRui Paulo * same object) they occur next to each other. 3384c74b245SRui Paulo */ 3394c74b245SRui Paulo if (strcmp(path, last) == 0) 3404c74b245SRui Paulo continue; 341c7fa6f0bSMark Johnston if ((error = (*func)(cd, &map, path)) != 0) 342c7fa6f0bSMark Johnston break; 3434c74b245SRui Paulo strlcpy(last, path, sizeof(last)); 3448eb20f36SRui Paulo } 345c7fa6f0bSMark Johnston return (error); 3468eb20f36SRui Paulo } 3478eb20f36SRui Paulo 3488eb20f36SRui Paulo prmap_t * 3492c633af4SJohn Birrell proc_addr2map(struct proc_handle *p, uintptr_t addr) 3502c633af4SJohn Birrell { 3518eb20f36SRui Paulo size_t i; 3528eb20f36SRui Paulo int cnt, lastvn = 0; 3538eb20f36SRui Paulo prmap_t *map; 3548eb20f36SRui Paulo rd_loadobj_t *rdl; 3558eb20f36SRui Paulo struct kinfo_vmentry *kves, *kve; 3568eb20f36SRui Paulo 3578eb20f36SRui Paulo /* 3588eb20f36SRui Paulo * If we don't have a cache of listed objects, we need to query 3598eb20f36SRui Paulo * it ourselves. 3608eb20f36SRui Paulo */ 3618eb20f36SRui Paulo if (p->nobjs == 0) { 362b1bb30e5SMark Johnston if ((kves = kinfo_getvmmap(proc_getpid(p), &cnt)) == NULL) 3638eb20f36SRui Paulo return (NULL); 3648eb20f36SRui Paulo for (i = 0; i < (size_t)cnt; i++) { 3658eb20f36SRui Paulo kve = kves + i; 3668eb20f36SRui Paulo if (kve->kve_type == KVME_TYPE_VNODE) 3678eb20f36SRui Paulo lastvn = i; 3681e6b3858SMark Johnston if (addr >= kve->kve_start && addr < kve->kve_end) { 3698eb20f36SRui Paulo if ((map = malloc(sizeof(*map))) == NULL) { 3708eb20f36SRui Paulo free(kves); 3718eb20f36SRui Paulo return (NULL); 3728eb20f36SRui Paulo } 3738eb20f36SRui Paulo map->pr_vaddr = kve->kve_start; 3748eb20f36SRui Paulo map->pr_size = kve->kve_end - kve->kve_start; 3758eb20f36SRui Paulo map->pr_offset = kve->kve_offset; 3768eb20f36SRui Paulo map->pr_mflags = 0; 3778eb20f36SRui Paulo if (kve->kve_protection & KVME_PROT_READ) 3788eb20f36SRui Paulo map->pr_mflags |= MA_READ; 3798eb20f36SRui Paulo if (kve->kve_protection & KVME_PROT_WRITE) 3808eb20f36SRui Paulo map->pr_mflags |= MA_WRITE; 3818eb20f36SRui Paulo if (kve->kve_protection & KVME_PROT_EXEC) 3828eb20f36SRui Paulo map->pr_mflags |= MA_EXEC; 3838eb20f36SRui Paulo if (kve->kve_flags & KVME_FLAG_COW) 3848eb20f36SRui Paulo map->pr_mflags |= MA_COW; 3858eb20f36SRui Paulo if (kve->kve_flags & KVME_FLAG_NEEDS_COPY) 3868eb20f36SRui Paulo map->pr_mflags |= MA_NEEDS_COPY; 3878eb20f36SRui Paulo if (kve->kve_flags & KVME_FLAG_NOCOREDUMP) 3888eb20f36SRui Paulo map->pr_mflags |= MA_NOCOREDUMP; 3898eb20f36SRui Paulo strlcpy(map->pr_mapname, kves[lastvn].kve_path, 3908eb20f36SRui Paulo sizeof(map->pr_mapname)); 3918eb20f36SRui Paulo free(kves); 3928eb20f36SRui Paulo return (map); 3938eb20f36SRui Paulo } 3948eb20f36SRui Paulo } 3958eb20f36SRui Paulo free(kves); 3968eb20f36SRui Paulo return (NULL); 3978eb20f36SRui Paulo } 3988eb20f36SRui Paulo 3998eb20f36SRui Paulo for (i = 0; i < p->nobjs; i++) { 4008eb20f36SRui Paulo rdl = &p->rdobjs[i]; 4011e6b3858SMark Johnston if (addr >= rdl->rdl_saddr && addr < rdl->rdl_eaddr) { 4028eb20f36SRui Paulo if ((map = malloc(sizeof(*map))) == NULL) 4038eb20f36SRui Paulo return (NULL); 4048eb20f36SRui Paulo proc_rdl2prmap(rdl, map); 4058eb20f36SRui Paulo return (map); 4068eb20f36SRui Paulo } 4078eb20f36SRui Paulo } 4082c633af4SJohn Birrell return (NULL); 4092c633af4SJohn Birrell } 4102c633af4SJohn Birrell 411540cc663SMark Johnston /* 412540cc663SMark Johnston * Look up the symbol at addr, returning a copy of the symbol and its name. 413540cc663SMark Johnston */ 414540cc663SMark Johnston static int 415540cc663SMark Johnston lookup_addr(Elf *e, Elf_Scn *scn, u_long stridx, uintptr_t off, uintptr_t addr, 416540cc663SMark Johnston const char **name, GElf_Sym *symcopy) 417540cc663SMark Johnston { 418540cc663SMark Johnston GElf_Sym sym; 419540cc663SMark Johnston Elf_Data *data; 420540cc663SMark Johnston const char *s; 421540cc663SMark Johnston uint64_t rsym; 422540cc663SMark Johnston int i; 423540cc663SMark Johnston 424540cc663SMark Johnston if ((data = elf_getdata(scn, NULL)) == NULL) { 425540cc663SMark Johnston DPRINTFX("ERROR: elf_getdata() failed: %s", elf_errmsg(-1)); 426540cc663SMark Johnston return (1); 427540cc663SMark Johnston } 428540cc663SMark Johnston for (i = 0; gelf_getsym(data, i, &sym) != NULL; i++) { 429540cc663SMark Johnston rsym = off + sym.st_value; 430540cc663SMark Johnston if (addr >= rsym && addr < rsym + sym.st_size) { 431540cc663SMark Johnston s = elf_strptr(e, stridx, sym.st_name); 432540cc663SMark Johnston if (s != NULL) { 433540cc663SMark Johnston *name = s; 434540cc663SMark Johnston memcpy(symcopy, &sym, sizeof(*symcopy)); 435540cc663SMark Johnston /* 436540cc663SMark Johnston * DTrace expects the st_value to contain 437540cc663SMark Johnston * only the address relative to the start of 438540cc663SMark Johnston * the function. 439540cc663SMark Johnston */ 440540cc663SMark Johnston symcopy->st_value = rsym; 441540cc663SMark Johnston return (0); 442540cc663SMark Johnston } 443540cc663SMark Johnston } 444540cc663SMark Johnston } 445540cc663SMark Johnston return (1); 446540cc663SMark Johnston } 447540cc663SMark Johnston 4482c633af4SJohn Birrell int 4492c633af4SJohn Birrell proc_addr2sym(struct proc_handle *p, uintptr_t addr, char *name, 4508eb20f36SRui Paulo size_t namesz, GElf_Sym *symcopy) 4512c633af4SJohn Birrell { 452540cc663SMark Johnston GElf_Ehdr ehdr; 453540cc663SMark Johnston GElf_Shdr shdr; 4548eb20f36SRui Paulo Elf *e; 4558eb20f36SRui Paulo Elf_Scn *scn, *dynsymscn = NULL, *symtabscn = NULL; 4568eb20f36SRui Paulo prmap_t *map; 457540cc663SMark Johnston const char *s; 458540cc663SMark Johnston uintptr_t off; 459540cc663SMark Johnston u_long symtabstridx = 0, dynsymstridx = 0; 460540cc663SMark Johnston int fd, error = -1; 4618eb20f36SRui Paulo 462*76c6531cSMark Johnston if ((map = proc_addr2map(p, addr)) == NULL) { 463*76c6531cSMark Johnston DPRINTFX("ERROR: proc_addr2map failed to resolve 0x%jx", addr); 4648eb20f36SRui Paulo return (-1); 4658eb20f36SRui Paulo } 466*76c6531cSMark Johnston if (open_object(map, &e, &fd) != 0) { 467*76c6531cSMark Johnston DPRINTFX("ERROR: failed to open object %s", map->pr_mapname); 468*76c6531cSMark Johnston return (-1); 4698eb20f36SRui Paulo } 4708eb20f36SRui Paulo if (gelf_getehdr(e, &ehdr) == NULL) { 47130e81f7eSMark Johnston DPRINTFX("ERROR: gelf_getehdr() failed: %s", elf_errmsg(-1)); 472*76c6531cSMark Johnston goto err; 4738eb20f36SRui Paulo } 474540cc663SMark Johnston 4758eb20f36SRui Paulo /* 4768eb20f36SRui Paulo * Find the index of the STRTAB and SYMTAB sections to locate 4778eb20f36SRui Paulo * symbol names. 4788eb20f36SRui Paulo */ 4798eb20f36SRui Paulo scn = NULL; 4808eb20f36SRui Paulo while ((scn = elf_nextscn(e, scn)) != NULL) { 4818eb20f36SRui Paulo gelf_getshdr(scn, &shdr); 4828eb20f36SRui Paulo switch (shdr.sh_type) { 4838eb20f36SRui Paulo case SHT_SYMTAB: 4848eb20f36SRui Paulo symtabscn = scn; 4858eb20f36SRui Paulo symtabstridx = shdr.sh_link; 4868eb20f36SRui Paulo break; 4878eb20f36SRui Paulo case SHT_DYNSYM: 4888eb20f36SRui Paulo dynsymscn = scn; 4898eb20f36SRui Paulo dynsymstridx = shdr.sh_link; 4908eb20f36SRui Paulo break; 4918eb20f36SRui Paulo } 4928eb20f36SRui Paulo } 493540cc663SMark Johnston 494540cc663SMark Johnston off = ehdr.e_type == ET_EXEC ? 0 : map->pr_vaddr; 495540cc663SMark Johnston 4968eb20f36SRui Paulo /* 497540cc663SMark Johnston * First look up the symbol in the dynsymtab, and fall back to the 498540cc663SMark Johnston * symtab if the lookup fails. 4998eb20f36SRui Paulo */ 500540cc663SMark Johnston error = lookup_addr(e, dynsymscn, dynsymstridx, off, addr, &s, symcopy); 501540cc663SMark Johnston if (error == 0) 5028eb20f36SRui Paulo goto out; 503540cc663SMark Johnston 504540cc663SMark Johnston error = lookup_addr(e, symtabscn, symtabstridx, off, addr, &s, symcopy); 505cd9c9939SMark Johnston if (error != 0) 506*76c6531cSMark Johnston goto err; 507540cc663SMark Johnston 5088eb20f36SRui Paulo out: 509540cc663SMark Johnston demangle(s, name, namesz); 510*76c6531cSMark Johnston err: 511*76c6531cSMark Johnston (void)elf_end(e); 512*76c6531cSMark Johnston (void)close(fd); 5138eb20f36SRui Paulo free(map); 5148eb20f36SRui Paulo return (error); 5152c633af4SJohn Birrell } 5162c633af4SJohn Birrell 5178eb20f36SRui Paulo prmap_t * 5182c633af4SJohn Birrell proc_name2map(struct proc_handle *p, const char *name) 5192c633af4SJohn Birrell { 5208eb20f36SRui Paulo size_t i; 5218eb20f36SRui Paulo int cnt; 522540cc663SMark Johnston prmap_t *map = NULL; 5238eb20f36SRui Paulo char tmppath[MAXPATHLEN]; 5248eb20f36SRui Paulo struct kinfo_vmentry *kves, *kve; 5258eb20f36SRui Paulo rd_loadobj_t *rdl; 5268eb20f36SRui Paulo 5278eb20f36SRui Paulo /* 5288eb20f36SRui Paulo * If we haven't iterated over the list of loaded objects, 5298eb20f36SRui Paulo * librtld_db isn't yet initialized and it's very likely 5308eb20f36SRui Paulo * that librtld_db called us. We need to do the heavy 5318eb20f36SRui Paulo * lifting here to find the symbol librtld_db is looking for. 5328eb20f36SRui Paulo */ 5338eb20f36SRui Paulo if (p->nobjs == 0) { 5348eb20f36SRui Paulo if ((kves = kinfo_getvmmap(proc_getpid(p), &cnt)) == NULL) 5358eb20f36SRui Paulo return (NULL); 5368eb20f36SRui Paulo for (i = 0; i < (size_t)cnt; i++) { 5378eb20f36SRui Paulo kve = kves + i; 5388eb20f36SRui Paulo basename_r(kve->kve_path, tmppath); 5398eb20f36SRui Paulo if (strcmp(tmppath, name) == 0) { 5408eb20f36SRui Paulo map = proc_addr2map(p, kve->kve_start); 541540cc663SMark Johnston break; 5428eb20f36SRui Paulo } 5438eb20f36SRui Paulo } 5448eb20f36SRui Paulo free(kves); 545540cc663SMark Johnston } else 5468eb20f36SRui Paulo for (i = 0; i < p->nobjs; i++) { 5478eb20f36SRui Paulo rdl = &p->rdobjs[i]; 5488eb20f36SRui Paulo basename_r(rdl->rdl_path, tmppath); 5498eb20f36SRui Paulo if (strcmp(tmppath, name) == 0) { 5508eb20f36SRui Paulo if ((map = malloc(sizeof(*map))) == NULL) 5518eb20f36SRui Paulo return (NULL); 5528eb20f36SRui Paulo proc_rdl2prmap(rdl, map); 553540cc663SMark Johnston break; 5548eb20f36SRui Paulo } 5558eb20f36SRui Paulo } 5568eb20f36SRui Paulo 557540cc663SMark Johnston if (map == NULL && strcmp(name, "a.out") == 0 && p->rdexec != NULL) 558540cc663SMark Johnston map = proc_addr2map(p, p->rdexec->rdl_saddr); 559540cc663SMark Johnston 560540cc663SMark Johnston return (map); 561540cc663SMark Johnston } 562540cc663SMark Johnston 563540cc663SMark Johnston /* 564540cc663SMark Johnston * Look up the symbol with the given name and return a copy of it. 565540cc663SMark Johnston */ 566540cc663SMark Johnston static int 567540cc663SMark Johnston lookup_name(Elf *e, Elf_Scn *scn, u_long stridx, const char *symbol, 56841da933cSMark Johnston GElf_Sym *symcopy, prsyminfo_t *si) 569540cc663SMark Johnston { 570540cc663SMark Johnston GElf_Sym sym; 571540cc663SMark Johnston Elf_Data *data; 572540cc663SMark Johnston char *s; 573540cc663SMark Johnston int i; 574540cc663SMark Johnston 575540cc663SMark Johnston if ((data = elf_getdata(scn, NULL)) == NULL) { 576540cc663SMark Johnston DPRINTFX("ERROR: elf_getdata() failed: %s", elf_errmsg(-1)); 577540cc663SMark Johnston return (1); 578540cc663SMark Johnston } 579540cc663SMark Johnston for (i = 0; gelf_getsym(data, i, &sym) != NULL; i++) { 580540cc663SMark Johnston s = elf_strptr(e, stridx, sym.st_name); 581540cc663SMark Johnston if (s != NULL && strcmp(s, symbol) == 0) { 582540cc663SMark Johnston memcpy(symcopy, &sym, sizeof(*symcopy)); 58341da933cSMark Johnston if (si != NULL) 58441da933cSMark Johnston si->prs_id = i; 585540cc663SMark Johnston return (0); 586540cc663SMark Johnston } 587540cc663SMark Johnston } 588540cc663SMark Johnston return (1); 5892c633af4SJohn Birrell } 5902c633af4SJohn Birrell 5912c633af4SJohn Birrell int 5922c633af4SJohn Birrell proc_name2sym(struct proc_handle *p, const char *object, const char *symbol, 59341da933cSMark Johnston GElf_Sym *symcopy, prsyminfo_t *si) 5942c633af4SJohn Birrell { 5958eb20f36SRui Paulo Elf *e; 5968eb20f36SRui Paulo Elf_Scn *scn, *dynsymscn = NULL, *symtabscn = NULL; 5978eb20f36SRui Paulo GElf_Shdr shdr; 5988eb20f36SRui Paulo GElf_Ehdr ehdr; 5998eb20f36SRui Paulo prmap_t *map; 600540cc663SMark Johnston uintptr_t off; 601540cc663SMark Johnston u_long symtabstridx = 0, dynsymstridx = 0; 602540cc663SMark Johnston int fd, error = -1; 6038eb20f36SRui Paulo 6048eb20f36SRui Paulo if ((map = proc_name2map(p, object)) == NULL) { 605*76c6531cSMark Johnston DPRINTFX("ERROR: proc_name2map failed to resolve %s", object); 606*76c6531cSMark Johnston return (-1); 6078eb20f36SRui Paulo } 608*76c6531cSMark Johnston if (open_object(map, &e, &fd) != 0) { 609*76c6531cSMark Johnston DPRINTFX("ERROR: failed to open object %s", map->pr_mapname); 610*76c6531cSMark Johnston return (-1); 6118eb20f36SRui Paulo } 6128eb20f36SRui Paulo if (gelf_getehdr(e, &ehdr) == NULL) { 61330e81f7eSMark Johnston DPRINTFX("ERROR: gelf_getehdr() failed: %s", elf_errmsg(-1)); 614*76c6531cSMark Johnston goto err; 6158eb20f36SRui Paulo } 616*76c6531cSMark Johnston 6178eb20f36SRui Paulo /* 6188eb20f36SRui Paulo * Find the index of the STRTAB and SYMTAB sections to locate 6198eb20f36SRui Paulo * symbol names. 6208eb20f36SRui Paulo */ 6218eb20f36SRui Paulo scn = NULL; 6228eb20f36SRui Paulo while ((scn = elf_nextscn(e, scn)) != NULL) { 6238eb20f36SRui Paulo gelf_getshdr(scn, &shdr); 6248eb20f36SRui Paulo switch (shdr.sh_type) { 6258eb20f36SRui Paulo case SHT_SYMTAB: 6268eb20f36SRui Paulo symtabscn = scn; 6278eb20f36SRui Paulo symtabstridx = shdr.sh_link; 6288eb20f36SRui Paulo break; 6298eb20f36SRui Paulo case SHT_DYNSYM: 6308eb20f36SRui Paulo dynsymscn = scn; 6318eb20f36SRui Paulo dynsymstridx = shdr.sh_link; 6328eb20f36SRui Paulo break; 6338eb20f36SRui Paulo } 6348eb20f36SRui Paulo } 635540cc663SMark Johnston 6368eb20f36SRui Paulo /* 637540cc663SMark Johnston * First look up the symbol in the dynsymtab, and fall back to the 638540cc663SMark Johnston * symtab if the lookup fails. 6398eb20f36SRui Paulo */ 64041da933cSMark Johnston error = lookup_name(e, dynsymscn, dynsymstridx, symbol, symcopy, si); 641540cc663SMark Johnston if (error == 0) 6428eb20f36SRui Paulo goto out; 643540cc663SMark Johnston 64441da933cSMark Johnston error = lookup_name(e, symtabscn, symtabstridx, symbol, symcopy, si); 645540cc663SMark Johnston if (error == 0) 6468eb20f36SRui Paulo goto out; 647540cc663SMark Johnston 6488eb20f36SRui Paulo out: 649540cc663SMark Johnston off = ehdr.e_type == ET_EXEC ? 0 : map->pr_vaddr; 650540cc663SMark Johnston symcopy->st_value += off; 651540cc663SMark Johnston 652*76c6531cSMark Johnston err: 653*76c6531cSMark Johnston (void)elf_end(e); 654*76c6531cSMark Johnston (void)close(fd); 6558eb20f36SRui Paulo free(map); 6568eb20f36SRui Paulo return (error); 6578eb20f36SRui Paulo } 6588eb20f36SRui Paulo 65941da933cSMark Johnston ctf_file_t * 66041da933cSMark Johnston proc_name2ctf(struct proc_handle *p, const char *name) 66141da933cSMark Johnston { 66241da933cSMark Johnston #ifndef NO_CTF 6633f1cbdbeSMark Johnston ctf_file_t *ctf; 66441da933cSMark Johnston prmap_t *map; 66541da933cSMark Johnston int error; 66641da933cSMark Johnston 667ec045490SMark Johnston if ((map = proc_name2map(p, name)) == NULL) 66841da933cSMark Johnston return (NULL); 66941da933cSMark Johnston 6703f1cbdbeSMark Johnston ctf = ctf_open(map->pr_mapname, &error); 6713f1cbdbeSMark Johnston free(map); 6723f1cbdbeSMark Johnston return (ctf); 67341da933cSMark Johnston #else 67441da933cSMark Johnston (void)p; 67541da933cSMark Johnston (void)name; 67641da933cSMark Johnston return (NULL); 67741da933cSMark Johnston #endif 67841da933cSMark Johnston } 67941da933cSMark Johnston 6808eb20f36SRui Paulo int 6818eb20f36SRui Paulo proc_iter_symbyaddr(struct proc_handle *p, const char *object, int which, 6828eb20f36SRui Paulo int mask, proc_sym_f *func, void *cd) 6838eb20f36SRui Paulo { 6848eb20f36SRui Paulo Elf *e; 6858eb20f36SRui Paulo int i, fd; 6868eb20f36SRui Paulo prmap_t *map; 687*76c6531cSMark Johnston Elf_Scn *scn, *foundscn; 6888eb20f36SRui Paulo Elf_Data *data; 689fbce8e16SMark Johnston GElf_Ehdr ehdr; 6908eb20f36SRui Paulo GElf_Shdr shdr; 6918eb20f36SRui Paulo GElf_Sym sym; 6928eb20f36SRui Paulo unsigned long stridx = -1; 6938eb20f36SRui Paulo char *s; 6948eb20f36SRui Paulo int error = -1; 6958eb20f36SRui Paulo 696*76c6531cSMark Johnston if ((map = proc_name2map(p, object)) == NULL) { 697*76c6531cSMark Johnston DPRINTFX("ERROR: proc_name2map failed to resolve %s", object); 6988eb20f36SRui Paulo return (-1); 6998eb20f36SRui Paulo } 700*76c6531cSMark Johnston if (open_object(map, &e, &fd) != 0) { 701*76c6531cSMark Johnston DPRINTFX("ERROR: failed to open object %s", map->pr_mapname); 702*76c6531cSMark Johnston return (-1); 7038eb20f36SRui Paulo } 704fbce8e16SMark Johnston if (gelf_getehdr(e, &ehdr) == NULL) { 705fbce8e16SMark Johnston DPRINTFX("ERROR: gelf_getehdr() failed: %s", elf_errmsg(-1)); 706*76c6531cSMark Johnston goto err; 707fbce8e16SMark Johnston } 7088eb20f36SRui Paulo /* 7098eb20f36SRui Paulo * Find the section we are looking for. 7108eb20f36SRui Paulo */ 711*76c6531cSMark Johnston foundscn = scn = NULL; 7128eb20f36SRui Paulo while ((scn = elf_nextscn(e, scn)) != NULL) { 7138eb20f36SRui Paulo gelf_getshdr(scn, &shdr); 7148eb20f36SRui Paulo if (which == PR_SYMTAB && 7158eb20f36SRui Paulo shdr.sh_type == SHT_SYMTAB) { 7168eb20f36SRui Paulo foundscn = scn; 7178eb20f36SRui Paulo break; 7188eb20f36SRui Paulo } else if (which == PR_DYNSYM && 7198eb20f36SRui Paulo shdr.sh_type == SHT_DYNSYM) { 7208eb20f36SRui Paulo foundscn = scn; 7218eb20f36SRui Paulo break; 7228eb20f36SRui Paulo } 7238eb20f36SRui Paulo } 7248eb20f36SRui Paulo if (!foundscn) 7258eb20f36SRui Paulo return (-1); 7268eb20f36SRui Paulo stridx = shdr.sh_link; 7278eb20f36SRui Paulo if ((data = elf_getdata(foundscn, NULL)) == NULL) { 72830e81f7eSMark Johnston DPRINTFX("ERROR: elf_getdata() failed: %s", elf_errmsg(-1)); 729*76c6531cSMark Johnston goto err; 7308eb20f36SRui Paulo } 731540cc663SMark Johnston for (i = 0; gelf_getsym(data, i, &sym) != NULL; i++) { 7328eb20f36SRui Paulo if (GELF_ST_BIND(sym.st_info) == STB_LOCAL && 7338eb20f36SRui Paulo (mask & BIND_LOCAL) == 0) 7348eb20f36SRui Paulo continue; 7358eb20f36SRui Paulo if (GELF_ST_BIND(sym.st_info) == STB_GLOBAL && 7368eb20f36SRui Paulo (mask & BIND_GLOBAL) == 0) 7378eb20f36SRui Paulo continue; 7388eb20f36SRui Paulo if (GELF_ST_BIND(sym.st_info) == STB_WEAK && 7398eb20f36SRui Paulo (mask & BIND_WEAK) == 0) 7408eb20f36SRui Paulo continue; 7418eb20f36SRui Paulo if (GELF_ST_TYPE(sym.st_info) == STT_NOTYPE && 7428eb20f36SRui Paulo (mask & TYPE_NOTYPE) == 0) 7438eb20f36SRui Paulo continue; 7448eb20f36SRui Paulo if (GELF_ST_TYPE(sym.st_info) == STT_OBJECT && 7458eb20f36SRui Paulo (mask & TYPE_OBJECT) == 0) 7468eb20f36SRui Paulo continue; 7478eb20f36SRui Paulo if (GELF_ST_TYPE(sym.st_info) == STT_FUNC && 7488eb20f36SRui Paulo (mask & TYPE_FUNC) == 0) 7498eb20f36SRui Paulo continue; 7508eb20f36SRui Paulo if (GELF_ST_TYPE(sym.st_info) == STT_SECTION && 7518eb20f36SRui Paulo (mask & TYPE_SECTION) == 0) 7528eb20f36SRui Paulo continue; 7538eb20f36SRui Paulo if (GELF_ST_TYPE(sym.st_info) == STT_FILE && 7548eb20f36SRui Paulo (mask & TYPE_FILE) == 0) 7558eb20f36SRui Paulo continue; 7568eb20f36SRui Paulo s = elf_strptr(e, stridx, sym.st_name); 757fbce8e16SMark Johnston if (ehdr.e_type != ET_EXEC) 7588eb20f36SRui Paulo sym.st_value += map->pr_vaddr; 759c7fa6f0bSMark Johnston if ((error = (*func)(cd, &sym, s)) != 0) 760*76c6531cSMark Johnston goto err; 7618eb20f36SRui Paulo } 7628eb20f36SRui Paulo error = 0; 763*76c6531cSMark Johnston err: 7648eb20f36SRui Paulo elf_end(e); 7658eb20f36SRui Paulo close(fd); 7668eb20f36SRui Paulo free(map); 7678eb20f36SRui Paulo return (error); 7682c633af4SJohn Birrell } 769