]> granicus.if.org Git - strace/blobdiff - unwind.c
unwind: move unwind_tcb_init invocation to after_successful_attach
[strace] / unwind.c
index da256c9e0c688f86873f851f183bf82c1afb181b..586f74500b8bbae053a39d76bfcc7a9ee6c86a67 100644 (file)
--- a/unwind.c
+++ b/unwind.c
@@ -1,5 +1,6 @@
 /*
  * Copyright (c) 2013 Luca Clementi <luca.clementi@gmail.com>
+ * Copyright (c) 2013-2018 The strace developers.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  */
 
 #include "defs.h"
-#include <limits.h>
-#include <libunwind-ptrace.h>
-
-#ifdef _LARGEFILE64_SOURCE
-# ifdef HAVE_FOPEN64
-#  define fopen_for_input fopen64
-# else
-#  define fopen_for_input fopen
+#include "mmap_cache.h"
+#include "unwind.h"
+
+#ifdef USE_DEMANGLE
+# if defined HAVE_DEMANGLE_H
+#  include <demangle.h>
+# elif defined HAVE_LIBIBERTY_DEMANGLE_H
+#  include <libiberty/demangle.h>
 # endif
-#else
-# define fopen_for_input fopen
 #endif
 
-#define DPRINTF(F, A, ...) if (debug_flag) fprintf(stderr, " [unwind(" A ")] " F "\n", __VA_ARGS__)
-
-/*
- * Кeep a sorted array of cache entries,
- * so that we can binary search through it.
- */
-struct mmap_cache_t {
-       /**
-        * example entry:
-        * 7fabbb09b000-7fabbb09f000 r--p 00179000 fc:00 1180246 /lib/libc-2.11.1.so
-        *
-        * start_addr  is 0x7fabbb09b000
-        * end_addr    is 0x7fabbb09f000
-        * mmap_offset is 0x179000
-        * binary_filename is "/lib/libc-2.11.1.so"
-        */
-       unsigned long start_addr;
-       unsigned long end_addr;
-       unsigned long mmap_offset;
-       char *binary_filename;
-       bool deleted;
-};
-
-/*
- * Type used in stacktrace walker
- */
-typedef void (*call_action_fn)(void *data,
-                              const char *binary_filename,
-                              const char *symbol_name,
-                              unw_word_t function_offset,
-                              unsigned long true_offset);
-typedef void (*error_action_fn)(void *data,
-                               const char *error,
-                               unsigned long true_offset);
-
 /*
  * Type used in stacktrace capturing
  */
 struct call_t {
-       struct call_tnext;
-       char *output_line;
+       struct call_t *next;
+       char *output_line;
 };
 
-struct queue_t {
-       struct call_t *tail;
-       struct call_t *head;
+struct unwind_queue_t {
+       struct call_t *tail;
+       struct call_t *head;
 };
 
-static void queue_print(struct queue_t *queue);
-static void delete_mmap_cache(struct tcb *tcp, const char *caller);
+static void queue_print(struct unwind_queue_t *queue);
 
-static unw_addr_space_t libunwind_as;
-static unsigned int mmap_cache_generation;
+static const char asprintf_error_str[] = "???";
 
 void
 unwind_init(void)
 {
-       libunwind_as = unw_create_addr_space(&_UPT_accessors, 0);
-       if (!libunwind_as)
-               error_msg_and_die("failed to create address space for stack tracing");
-       unw_set_caching_policy(libunwind_as, UNW_CACHE_GLOBAL);
+       if (unwinder.init)
+               unwinder.init();
+       mmap_cache_enable();
 }
 
 void
 unwind_tcb_init(struct tcb *tcp)
 {
-       tcp->libunwind_ui = _UPT_create(tcp->pid);
-       if (!tcp->libunwind_ui)
-               die_out_of_memory();
-
-       tcp->queue = malloc(sizeof(*tcp->queue));
-       if (!tcp->queue)
-               die_out_of_memory();
-       tcp->queue->head = NULL;
-       tcp->queue->tail = NULL;
-}
-
-void
-unwind_tcb_fin(struct tcb *tcp)
-{
-       queue_print(tcp->queue);
-       free(tcp->queue);
-       tcp->queue = NULL;
-
-       delete_mmap_cache(tcp, __FUNCTION__);
-
-       _UPT_destroy(tcp->libunwind_ui);
-       tcp->libunwind_ui = NULL;
-}
-
-/*
- * caching of /proc/ID/maps for each process to speed up stack tracing
- *
- * The cache must be refreshed after some syscall: mmap, mprotect, munmap, execve
- */
-static void
-build_mmap_cache(struct tcb* tcp)
-{
-       unsigned long start_addr, end_addr, mmap_offset;
-       char filename[sizeof ("/proc/0123456789/maps")];
-       char buffer[PATH_MAX + 80];
-       char binary_path[PATH_MAX];
-       struct mmap_cache_t *cur_entry, *prev_entry;
-       /* start with a small dynamically-allocated array and then expand it */
-       size_t cur_array_size = 10;
-       struct mmap_cache_t *cache_head;
-       FILE *fp;
-
-       const char *deleted = " (deleted)";
-       size_t blen;
-       size_t dlen;
-
-       unw_flush_cache (libunwind_as, 0, 0);
-
-       sprintf(filename, "/proc/%d/maps", tcp->pid);
-       fp = fopen_for_input(filename, "r");
-       if (!fp) {
-               perror_msg("fopen: %s", filename);
+       if (tcp->unwind_queue)
                return;
-       }
-
-       cache_head = calloc(cur_array_size, sizeof(*cache_head));
-       if (!cache_head)
-               die_out_of_memory();
-
-       while (fgets(buffer, sizeof(buffer), fp) != NULL) {
-               binary_path[0] = '\0'; // 'reset' it just to be paranoid
-
-               sscanf(buffer, "%lx-%lx %*c%*c%*c%*c %lx %*x:%*x %*d %[^\n]",
-                      &start_addr, &end_addr, &mmap_offset, binary_path);
-
-               /* ignore special 'fake files' like "[vdso]", "[heap]", "[stack]", */
-               if (binary_path[0] == '[') {
-                       continue;
-               }
-
-               if (binary_path[0] == '\0') {
-                       continue;
-               }
-
-               if (end_addr < start_addr)
-                       perror_msg_and_die("%s: unrecognized maps file format",
-                                          filename);
-
-               cur_entry = &cache_head[tcp->mmap_cache_size];
-               cur_entry->start_addr = start_addr;
-               cur_entry->end_addr = end_addr;
-               cur_entry->mmap_offset = mmap_offset;
-               cur_entry->binary_filename = strdup(binary_path);
-
-               dlen = strlen(deleted);
-               blen = strlen(binary_path);
-               if (blen >= dlen && strcmp(binary_path + blen - dlen, deleted) == 0)
-                       cur_entry->deleted = true;
-               else
-                       cur_entry->deleted = false;
-
-               /*
-                * sanity check to make sure that we're storing
-                * non-overlapping regions in ascending order
-                */
-               if (tcp->mmap_cache_size > 0) {
-                       prev_entry = &cache_head[tcp->mmap_cache_size - 1];
-                       if (prev_entry->start_addr >= cur_entry->start_addr)
-                               perror_msg_and_die("Overlaying memory region in %s",
-                                                  filename);
-                       if (prev_entry->end_addr > cur_entry->start_addr)
-                               perror_msg_and_die("Overlaying memory region in %s",
-                                                  filename);
-               }
-               tcp->mmap_cache_size++;
-
-               /* resize doubling its size */
-               if (tcp->mmap_cache_size >= cur_array_size) {
-                       cur_array_size *= 2;
-                       cache_head = realloc(cache_head, cur_array_size * sizeof(*cache_head));
-                       if (!cache_head)
-                               die_out_of_memory();
-               }
-       }
-       fclose(fp);
-       tcp->mmap_cache = cache_head;
-       tcp->mmap_cache_generation = mmap_cache_generation;
-
-       DPRINTF("tgen=%u, ggen=%u, tcp=%p, cache=%p",
-               "cache-build",
-               tcp->mmap_cache_generation,
-               mmap_cache_generation,
-               tcp, tcp->mmap_cache);
-}
-
-/* deleting the cache */
-static void
-delete_mmap_cache(struct tcb *tcp, const char *caller)
-{
-       unsigned int i;
-
-       DPRINTF("tgen=%u, ggen=%u, tcp=%p, cache=%p, caller=%s",
-               "cache-delete",
-               tcp->mmap_cache_generation,
-               mmap_cache_generation,
-               tcp, tcp->mmap_cache, caller);
-
-       for (i = 0; i < tcp->mmap_cache_size; i++) {
-               free(tcp->mmap_cache[i].binary_filename);
-               tcp->mmap_cache[i].binary_filename = NULL;
-       }
-       free(tcp->mmap_cache);
-       tcp->mmap_cache = NULL;
-       tcp->mmap_cache_size = 0;
-}
 
-static bool
-rebuild_cache_if_invalid(struct tcb *tcp, const char *caller)
-{
-       if ((tcp->mmap_cache_generation != mmap_cache_generation)
-           && tcp->mmap_cache)
-               delete_mmap_cache(tcp, caller);
-
-       if (!tcp->mmap_cache)
-               build_mmap_cache(tcp);
+       tcp->unwind_queue = xmalloc(sizeof(*tcp->unwind_queue));
+       tcp->unwind_queue->head = NULL;
+       tcp->unwind_queue->tail = NULL;
 
-       if (!tcp->mmap_cache || !tcp->mmap_cache_size)
-               return false;
-       else
-               return true;
+       tcp->unwind_ctx = unwinder.tcb_init(tcp);
 }
 
 void
-unwind_cache_invalidate(struct tcb* tcp)
+unwind_tcb_fin(struct tcb *tcp)
 {
-#if SUPPORTED_PERSONALITIES > 1
-       if (tcp->currpers != DEFAULT_PERSONALITY) {
-               /* disable strack trace */
+       if (!tcp->unwind_queue)
                return;
-       }
-#endif
-       mmap_cache_generation++;
-       DPRINTF("tgen=%u, ggen=%u, tcp=%p, cache=%p", "increment",
-               tcp->mmap_cache_generation,
-               mmap_cache_generation,
-               tcp,
-               tcp->mmap_cache);
-}
 
-/*
- * walking the stack
- */
-static void
-stacktrace_walk(struct tcb *tcp,
-               call_action_fn call_action,
-               error_action_fn error_action,
-               void *data)
-{
-       unw_word_t ip;
-       unw_cursor_t cursor;
-       unw_word_t function_offset;
-       int stack_depth = 0, ret_val;
-       /* these are used for the binary search through the mmap_chace */
-       int lower, upper, mid;
-       size_t symbol_name_size = 40;
-       char * symbol_name;
-       struct mmap_cache_t* cur_mmap_cache;
-       unsigned long true_offset;
-       bool berror_expected = false;
-
-       if (!tcp->mmap_cache)
-               error_msg_and_die("bug: mmap_cache is NULL");
-       if (tcp->mmap_cache_size == 0)
-               error_msg_and_die("bug: mmap_cache is empty");
-
-       symbol_name = malloc(symbol_name_size);
-       if (!symbol_name)
-               die_out_of_memory();
-
-       if (unw_init_remote(&cursor, libunwind_as, tcp->libunwind_ui) < 0)
-               perror_msg_and_die("Can't initiate libunwind");
-
-       do {
-               /* looping on the stack frame */
-               if (unw_get_reg(&cursor, UNW_REG_IP, &ip) < 0) {
-                       perror_msg("Can't walk the stack of process %d", tcp->pid);
-                       break;
-               }
-
-               lower = 0;
-               upper = (int) tcp->mmap_cache_size - 1;
-
-               while (lower <= upper) {
-                       /* find the mmap_cache and print the stack frame */
-                       mid = (upper + lower) / 2;
-                       cur_mmap_cache = &tcp->mmap_cache[mid];
-
-                       if (ip >= cur_mmap_cache->start_addr &&
-                           ip < cur_mmap_cache->end_addr) {
-                               for (;;) {
-                                       symbol_name[0] = '\0';
-                                       ret_val = unw_get_proc_name(&cursor, symbol_name,
-                                               symbol_name_size, &function_offset);
-                                       if (ret_val != -UNW_ENOMEM)
-                                               break;
-                                       symbol_name_size *= 2;
-                                       symbol_name = realloc(symbol_name, symbol_name_size);
-                                       if (!symbol_name)
-                                               die_out_of_memory();
-                               }
-
-                               if (cur_mmap_cache->deleted)
-                                       berror_expected = true;
-
-                               true_offset = ip - cur_mmap_cache->start_addr +
-                                       cur_mmap_cache->mmap_offset;
-                               if (symbol_name[0]) {
-                                       call_action(data,
-                                                   cur_mmap_cache->binary_filename,
-                                                   symbol_name,
-                                                   function_offset,
-                                                   true_offset);
-                               } else {
-                                       call_action(data,
-                                                   cur_mmap_cache->binary_filename,
-                                                   symbol_name,
-                                                   0,
-                                                   true_offset);
-                               }
-                               break; /* stack frame printed */
-                       }
-                       else if (mid == 0) {
-                               /*
-                                * there is a bug in libunwind >= 1.0
-                                * after a set_tid_address syscall
-                                * unw_get_reg returns IP == 0
-                                */
-                               if(ip)
-                                       error_action(data,
-                                                    "backtracing_error", 0);
-                               goto ret;
-                       }
-                       else if (ip < cur_mmap_cache->start_addr)
-                               upper = mid - 1;
-                       else
-                               lower = mid + 1;
-
-               }
-               if (lower > upper) {
-                       error_action(data,
-                                    berror_expected
-                                    ?"expected_backtracing_error"
-                                    :"unexpected_backtracing_error",
-                                    ip);
-                       goto ret;
-               }
-
-               ret_val = unw_step(&cursor);
-
-               if (++stack_depth > 255) {
-                       error_action(data,
-                                    "too many stack frames", 0);
-                       break;
-               }
-       } while (ret_val > 0);
-ret:
-       free(symbol_name);
+       queue_print(tcp->unwind_queue);
+       free(tcp->unwind_queue);
+       tcp->unwind_queue = NULL;
+
+       unwinder.tcb_fin(tcp);
+       tcp->unwind_ctx = NULL;
 }
 
 /*
@@ -415,10 +101,10 @@ ret:
  * /lib64/libc.so.6(__libc_start_main+0xed) [0x7fa2f8a5976d]
  * ./a.out() [0x400569]
  */
-#define STACK_ENTRY_SYMBOL_FMT                 \
+#define STACK_ENTRY_SYMBOL_FMT(SYM)            \
        " > %s(%s+0x%lx) [0x%lx]\n",            \
        binary_filename,                        \
-       symbol_name,                            \
+       (SYM),                                  \
        (unsigned long) function_offset,        \
        true_offset
 #define STACK_ENTRY_NOSYMBOL_FMT               \
@@ -435,15 +121,28 @@ static void
 print_call_cb(void *dummy,
              const char *binary_filename,
              const char *symbol_name,
-             unw_word_t function_offset,
+             unwind_function_offset_t function_offset,
              unsigned long true_offset)
 {
-       if (symbol_name)
-               tprintf(STACK_ENTRY_SYMBOL_FMT);
+       if (symbol_name && (symbol_name[0] != '\0')) {
+#ifdef USE_DEMANGLE
+               char *demangled_name =
+                       cplus_demangle(symbol_name,
+                                      DMGL_AUTO | DMGL_PARAMS);
+#endif
+               tprintf(STACK_ENTRY_SYMBOL_FMT(
+#ifdef USE_DEMANGLE
+                                              demangled_name ? demangled_name :
+#endif
+                                              symbol_name));
+#ifdef USE_DEMANGLE
+               free(demangled_name);
+#endif
+       }
        else if (binary_filename)
                tprintf(STACK_ENTRY_NOSYMBOL_FMT);
        else
-               tprintf(STACK_ENTRY_BUG_FMT, __FUNCTION__);
+               tprintf(STACK_ENTRY_BUG_FMT, __func__);
 
        line_ended();
 }
@@ -464,47 +163,60 @@ print_error_cb(void *dummy,
 static char *
 sprint_call_or_error(const char *binary_filename,
                     const char *symbol_name,
-                    unw_word_t function_offset,
+                    unwind_function_offset_t function_offset,
                     unsigned long true_offset,
                     const char *error)
 {
-       char *output_line = NULL;
-       int n;
-
-       if (symbol_name)
-               n = asprintf(&output_line, STACK_ENTRY_SYMBOL_FMT);
-       else if (binary_filename)
-               n = asprintf(&output_line, STACK_ENTRY_NOSYMBOL_FMT);
-       else if (error)
-               n = true_offset
-                       ? asprintf(&output_line, STACK_ENTRY_ERROR_WITH_OFFSET_FMT)
-                       : asprintf(&output_line, STACK_ENTRY_ERROR_FMT);
-       else
-               n = asprintf(&output_line, STACK_ENTRY_BUG_FMT, __FUNCTION__);
-
-       if (n < 0)
-               error_msg_and_die("error in asprintf");
-
-       return output_line;
+       char *output_line = NULL;
+       int n;
+
+       if (symbol_name) {
+#ifdef USE_DEMANGLE
+               char *demangled_name =
+                       cplus_demangle(symbol_name,
+                                      DMGL_AUTO | DMGL_PARAMS);
+#endif
+               n = asprintf(&output_line,
+                            STACK_ENTRY_SYMBOL_FMT(
+#ifdef USE_DEMANGLE
+                                                   demangled_name ? demangled_name :
+#endif
+                                                   symbol_name));
+#ifdef USE_DEMANGLE
+               free(demangled_name);
+#endif
+       }
+       else if (binary_filename)
+               n = asprintf(&output_line, STACK_ENTRY_NOSYMBOL_FMT);
+       else if (error)
+               n = true_offset
+                       ? asprintf(&output_line, STACK_ENTRY_ERROR_WITH_OFFSET_FMT)
+                       : asprintf(&output_line, STACK_ENTRY_ERROR_FMT);
+       else
+               n = asprintf(&output_line, STACK_ENTRY_BUG_FMT, __func__);
+
+       if (n < 0) {
+               perror_func_msg("asprintf");
+               output_line = (char *) asprintf_error_str;
+       }
+
+       return output_line;
 }
 
 /*
  * queue manipulators
  */
 static void
-queue_put(struct queue_t *queue,
+queue_put(struct unwind_queue_t *queue,
          const char *binary_filename,
          const char *symbol_name,
-         unw_word_t function_offset,
+         unwind_function_offset_t function_offset,
          unsigned long true_offset,
          const char *error)
 {
        struct call_t *call;
 
-       call = malloc(sizeof(*call));
-       if (!call)
-               die_out_of_memory();
-
+       call = xmalloc(sizeof(*call));
        call->output_line = sprint_call_or_error(binary_filename,
                                                 symbol_name,
                                                 function_offset,
@@ -525,7 +237,7 @@ static void
 queue_put_call(void *queue,
               const char *binary_filename,
               const char *symbol_name,
-              unw_word_t function_offset,
+              unwind_function_offset_t function_offset,
               unsigned long true_offset)
 {
        queue_put(queue,
@@ -545,7 +257,7 @@ queue_put_error(void *queue,
 }
 
 static void
-queue_print(struct queue_t *queue)
+queue_print(struct unwind_queue_t *queue)
 {
        struct call_t *call, *tmp;
 
@@ -559,7 +271,9 @@ queue_print(struct queue_t *queue)
                tprints(tmp->output_line);
                line_ended();
 
-               free(tmp->output_line);
+               if (tmp->output_line != asprintf_error_str)
+                       free(tmp->output_line);
+
                tmp->output_line = NULL;
                tmp->next = NULL;
                free(tmp);
@@ -570,42 +284,60 @@ queue_print(struct queue_t *queue)
  * printing stack
  */
 void
-unwind_print_stacktrace(struct tcb* tcp)
+unwind_tcb_print(struct tcb *tcp)
 {
 #if SUPPORTED_PERSONALITIES > 1
        if (tcp->currpers != DEFAULT_PERSONALITY) {
-               /* disable strack trace */
+               /* disable stack trace */
                return;
        }
 #endif
-       if (tcp->queue->head) {
-              DPRINTF("tcp=%p, queue=%p", "queueprint", tcp, tcp->queue->head);
-              queue_print(tcp->queue);
-       }
-       else if (rebuild_cache_if_invalid(tcp, __FUNCTION__)) {
-               DPRINTF("tcp=%p, queue=%p", "stackprint", tcp, tcp->queue->head);
-               stacktrace_walk(tcp, print_call_cb, print_error_cb, NULL);
-       }
+       if (tcp->unwind_queue->head) {
+               debug_func_msg("head: tcp=%p, queue=%p",
+                              tcp, tcp->unwind_queue->head);
+               queue_print(tcp->unwind_queue);
+       } else switch (mmap_cache_rebuild_if_invalid(tcp, __func__)) {
+               case MMAP_CACHE_REBUILD_RENEWED:
+                       unwinder.tcb_flush_cache(tcp);
+                       ATTRIBUTE_FALLTHROUGH;
+               case MMAP_CACHE_REBUILD_READY:
+                       debug_func_msg("walk: tcp=%p, queue=%p",
+                                      tcp, tcp->unwind_queue->head);
+                       unwinder.tcb_walk(tcp, print_call_cb, print_error_cb, NULL);
+                       break;
+               default:
+                       /* Do nothing */
+                       ;
+       }
 }
 
 /*
  * capturing stack
  */
 void
-unwind_capture_stacktrace(struct tcb *tcp)
+unwind_tcb_capture(struct tcb *tcp)
 {
 #if SUPPORTED_PERSONALITIES > 1
        if (tcp->currpers != DEFAULT_PERSONALITY) {
-               /* disable strack trace */
+               /* disable stack trace */
                return;
        }
 #endif
-       if (tcp->queue->head)
+       if (tcp->unwind_queue->head)
                error_msg_and_die("bug: unprinted entries in queue");
 
-       if (rebuild_cache_if_invalid(tcp, __FUNCTION__)) {
-               stacktrace_walk(tcp, queue_put_call, queue_put_error,
-                               tcp->queue);
-               DPRINTF("tcp=%p, queue=%p", "captured", tcp, tcp->queue->head);
+       switch (mmap_cache_rebuild_if_invalid(tcp, __func__)) {
+       case MMAP_CACHE_REBUILD_RENEWED:
+               unwinder.tcb_flush_cache(tcp);
+               ATTRIBUTE_FALLTHROUGH;
+       case MMAP_CACHE_REBUILD_READY:
+               unwinder.tcb_walk(tcp, queue_put_call, queue_put_error,
+                                  tcp->unwind_queue);
+               debug_func_msg("tcp=%p, queue=%p",
+                              tcp, tcp->unwind_queue->head);
+               break;
+       default:
+               /* Do nothing */
+               ;
        }
 }