lib/libexecinfo/backtrace.c

   1 /*      $NetBSD: backtrace.c,v 1.3 2013/08/29 14:58:56 christos Exp $   */
   2
   3 /*-
   4  * Copyright (c) 2012 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * This code is derived from software contributed to The NetBSD Foundation
   8  * by Christos Zoulas.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  *
  19  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  29  * POSSIBILITY OF SUCH DAMAGE.
  30  */
  31 #include <sys/cdefs.h>
  32 #include <sys/param.h>
  33 #include <assert.h>
  34 #define _WITH_DPRINTF
  35 #include <stdio.h>
  36 #include <string.h>
  37 #include <stdlib.h>
  38 #include <stdarg.h>
  39 #include <stdint.h>
  40 #include <stddef.h>
  41 #include <unistd.h>
  42 #include <fcntl.h>
  43 #include <dlfcn.h>
  44 #include <elf.h>
  45
  46 #include "execinfo.h"
  47 #include "symtab.h"
  48
  49 #ifdef __linux__
  50 #define SELF    "/proc/self/exe"
  51 #else
  52 #include <sys/sysctl.h>
  53 #define SELF    "/proc/curproc/file"
  54 #endif
  55
  56 static int
  57 open_self(int flags)
  58 {
  59         const char *pathname = SELF;
  60 #ifdef KERN_PROC_PATHNAME
  61         static const int name[] = {
  62                 CTL_KERN, KERN_PROC, -1, KERN_PROC_PATHNAME,
  63         };
  64         char path[MAXPATHLEN];
  65         size_t len;
  66
  67         len = sizeof(path);
  68         if (sysctl(name, 4, path, &len, NULL, 0) != -1)
  69                 pathname = path;
  70 #endif
  71         return open(pathname, flags);
  72 }
  73
  74
  75 static int __printflike(4, 5)
  76 rasprintf(char **buf, size_t *bufsiz, size_t offs, const char *fmt, ...)
  77 {
  78         for (;;) {
  79                 size_t nbufsiz;
  80                 char *nbuf;
  81
  82                 if (*buf && offs < *bufsiz) {
  83                         va_list ap;
  84                         int len;
  85
  86                         va_start(ap, fmt);
  87                         len = vsnprintf(*buf + offs, *bufsiz - offs, fmt, ap);
  88                         va_end(ap);
  89
  90                         if (len < 0 || (size_t)len + 1 < *bufsiz - offs)
  91                                 return len;
  92                         nbufsiz = MAX(*bufsiz + 512, (size_t)len + 1);
  93                 } else
  94                         nbufsiz = MAX(offs, *bufsiz) + 512;
  95
  96                 nbuf = realloc(*buf, nbufsiz);
  97                 if (nbuf == NULL)
  98                         return -1;
  99                 *buf = nbuf;
 100                 *bufsiz = nbufsiz;
 101         }
 102 }
 103
 104 /*
 105  * format specifiers:
 106  *      %a      = address
 107  *      %n      = symbol_name
 108  *      %d      = symbol_address - address
 109  *      %D      = if symbol_address == address "" else +%d
 110  *      %f      = filename
 111  */
 112 static ssize_t
 113 format_string(char **buf, size_t *bufsiz, size_t offs, const char *fmt,
 114     Dl_info *dli, const void *addr)
 115 {
 116         ptrdiff_t diff = (const char *)addr - (const char *)dli->dli_saddr;
 117         size_t o = offs;
 118         int len;
 119
 120         for (; *fmt; fmt++) {
 121                 if (*fmt != '%')
 122                         goto printone;
 123                 switch (*++fmt) {
 124                 case 'a':
 125                         len = rasprintf(buf, bufsiz, o, "%p", addr);
 126                         break;
 127                 case 'n':
 128                         len = rasprintf(buf, bufsiz, o, "%s", dli->dli_sname);
 129                         break;
 130                 case 'D':
 131                         if (diff)
 132                                 len = rasprintf(buf, bufsiz, o, "+0x%tx", diff);
 133                         else
 134                                 len = 0;
 135                         break;
 136                 case 'd':
 137                         len = rasprintf(buf, bufsiz, o, "0x%tx", diff);
 138                         break;
 139                 case 'f':
 140                         len = rasprintf(buf, bufsiz, o, "%s", dli->dli_fname);
 141                         break;
 142                 default:
 143                 printone:
 144                         len = rasprintf(buf, bufsiz, o, "%c", *fmt);
 145                         break;
 146                 }
 147                 if (len == -1)
 148                         return -1;
 149                 o += len;
 150         }
 151         return o - offs;
 152 }
 153
 154 static ssize_t
 155 format_address(symtab_t *st, char **buf, size_t *bufsiz, size_t offs,
 156     const char *fmt, const void *addr)
 157 {
 158         Dl_info dli;
 159
 160         memset(&dli, 0, sizeof(dli));
 161         (void)dladdr(addr, &dli);
 162         if (st)
 163                 symtab_find(st, addr, &dli);
 164
 165         if (dli.dli_sname == NULL)
 166                 dli.dli_sname = "???";
 167         if (dli.dli_fname == NULL)
 168                 dli.dli_fname = "???";
 169         if (dli.dli_saddr == NULL)
 170                 dli.dli_saddr = (void *)(intptr_t)addr;
 171
 172         return format_string(buf, bufsiz, offs, fmt, &dli, addr);
 173 }
 174
 175 char **
 176 backtrace_symbols_fmt(void *const *trace, size_t len, const char *fmt)
 177 {
 178
 179         static const size_t slen = sizeof(char *) + 64; /* estimate */
 180         char *ptr;
 181         symtab_t *st;
 182         int fd;
 183
 184         if ((fd = open_self(O_RDONLY)) != -1)
 185                 st = symtab_create(fd, -1, STT_FUNC);
 186         else
 187                 st = NULL;
 188
 189         if ((ptr = calloc(len, slen)) == NULL)
 190                 goto out;
 191
 192         size_t psize = len * slen;
 193         size_t offs = len * sizeof(char *);
 194
 195         /* We store only offsets in the first pass because of realloc */
 196         for (size_t i = 0; i < len; i++) {
 197                 ssize_t x;
 198                 ((char **)(void *)ptr)[i] = (void *)offs;
 199                 x = format_address(st, &ptr, &psize, offs, fmt, trace[i]);
 200                 if (x == -1) {
 201                         free(ptr);
 202                         ptr = NULL;
 203                         goto out;
 204                 }
 205                 offs += x;
 206                 ptr[offs++] = '\0';
 207                 assert(offs < psize);
 208         }
 209
 210         /* Change offsets to pointers */
 211         for (size_t j = 0; j < len; j++)
 212                 ((char **)(void *)ptr)[j] += (intptr_t)ptr;
 213
 214 out:
 215         symtab_destroy(st);
 216         if (fd != -1)
 217                 (void)close(fd);
 218
 219         return (void *)ptr;
 220 }
 221
 222 int
 223 backtrace_symbols_fd_fmt(void *const *trace, size_t len, int fd,
 224     const char *fmt)
 225 {
 226         char **s = backtrace_symbols_fmt(trace, len, fmt);
 227         if (s == NULL)
 228                 return -1;
 229         for (size_t i = 0; i < len; i++)
 230                 if (dprintf(fd, "%s\n", s[i]) < 0)
 231                         break;
 232         free(s);
 233         return 0;
 234 }
 235
 236 static const char fmt[] = "%a <%n%D> at %f";
 237
 238 char **
 239 backtrace_symbols(void *const *trace, size_t len)
 240 {
 241         return backtrace_symbols_fmt(trace, len, fmt);
 242 }
 243
 244 int
 245 backtrace_symbols_fd(void *const *trace, size_t len, int fd)
 246 {
 247         return backtrace_symbols_fd_fmt(trace, len, fd, fmt);
 248 }