diff options
Diffstat (limited to 'rts')
-rw-r--r-- | rts/Libdw.c | 334 | ||||
-rw-r--r-- | rts/Libdw.h | 58 | ||||
-rw-r--r-- | rts/ghc.mk | 8 | ||||
-rw-r--r-- | rts/package.conf.in | 4 |
4 files changed, 404 insertions, 0 deletions
diff --git a/rts/Libdw.c b/rts/Libdw.c new file mode 100644 index 0000000000..86f07c3c5e --- /dev/null +++ b/rts/Libdw.c @@ -0,0 +1,334 @@ +/* --------------------------------------------------------------------------- + * + * (c) The GHC Team, 2014-2015 + * + * Producing DWARF-based stacktraces with libdw. + * + * --------------------------------------------------------------------------*/ + +#ifdef USE_LIBDW + +#include <elfutils/libdwfl.h> +#include <dwarf.h> +#include <unistd.h> + +#include "Rts.h" +#include "Libdw.h" +#include "RtsUtils.h" + +static BacktraceChunk *backtrace_alloc_chunk(BacktraceChunk *next) { + BacktraceChunk *chunk = stgMallocBytes(sizeof(BacktraceChunk), + "backtrace_alloc_chunk"); + chunk->n_frames = 0; + chunk->next = next; + return chunk; +} + +// Allocate a Backtrace +static Backtrace *backtrace_alloc(void) { + Backtrace *bt = stgMallocBytes(sizeof(Backtrace), "backtrace_alloc"); + bt->n_frames = 0; + bt->last = backtrace_alloc_chunk(NULL); + return bt; +} + +static void backtrace_push(Backtrace *bt, StgPtr pc) { + // Is this chunk full? + if (bt->last->n_frames == BACKTRACE_CHUNK_SZ) + bt->last = backtrace_alloc_chunk(bt->last); + + // Push the PC + bt->last->frames[bt->last->n_frames] = pc; + bt->last->n_frames++; + bt->n_frames++; +} + +void backtrace_free(Backtrace *bt) { + if (bt == NULL) + return; + BacktraceChunk *chunk = bt->last; + while (chunk != NULL) { + BacktraceChunk *next = chunk->next; + stgFree(chunk); + chunk = next; + } + stgFree(bt); +} + +struct LibDwSession_ { + Dwfl *dwfl; + Backtrace *cur_bt; // The current backtrace we are collecting (if any) +}; + +typedef struct LibDwSession_ LibDwSession; + +static const Dwfl_Thread_Callbacks thread_cbs; + +void libdw_free(LibDwSession *session) { + if (session == NULL) + return; + dwfl_end(session->dwfl); + stgFree(session); +} + +// Create a libdw session with DWARF information for all loaded modules +LibDwSession *libdw_init() { + LibDwSession *session = stgCallocBytes(1, sizeof(LibDwSession), + "libdw_init"); + // Initialize ELF library + if (elf_version(EV_CURRENT) == EV_NONE) { + sysErrorBelch("libelf version too old!"); + return NULL; + } + + // Initialize a libdwfl session + static char *debuginfo_path; + static const Dwfl_Callbacks proc_callbacks = + { + .find_debuginfo = dwfl_standard_find_debuginfo, + .debuginfo_path = &debuginfo_path, + .find_elf = dwfl_linux_proc_find_elf, + }; + session->dwfl = dwfl_begin (&proc_callbacks); + if (session->dwfl == NULL) { + sysErrorBelch("dwfl_begin failed: %s", dwfl_errmsg(dwfl_errno())); + free(session); + return NULL; + } + + // Report the loaded modules + int ret = dwfl_linux_proc_report(session->dwfl, getpid()); + if (ret < 0) { + sysErrorBelch("dwfl_linux_proc_report failed: %s", + dwfl_errmsg(dwfl_errno())); + goto fail; + } + if (dwfl_report_end (session->dwfl, NULL, NULL) != 0) { + sysErrorBelch("dwfl_report_end failed: %s", dwfl_errmsg(dwfl_errno())); + goto fail; + } + + pid_t pid = getpid(); + if (! dwfl_attach_state(session->dwfl, NULL, pid, &thread_cbs, NULL)) { + sysErrorBelch("dwfl_attach_state failed: %s", + dwfl_errmsg(dwfl_errno())); + goto fail; + } + + return session; + + fail: + dwfl_end(session->dwfl); + free(session); + return NULL; +} + +int libdw_lookup_location(LibDwSession *session, Location *frame, + StgPtr pc) { + // Find the module containing PC + Dwfl_Module *mod = dwfl_addrmodule(session->dwfl, (Dwarf_Addr) pc); + if (mod == NULL) + return 1; + dwfl_module_info(mod, NULL, NULL, NULL, NULL, NULL, + &frame->object_file, NULL); + + // Find function name + frame->function = dwfl_module_addrname(mod, (Dwarf_Addr) pc); + + // Try looking up source location + Dwfl_Line *line = dwfl_module_getsrc(mod, (Dwarf_Addr) pc); + if (line != NULL) { + Dwarf_Addr addr; + int lineno, colno; + /* libdwfl owns the source_file buffer, don't free it */ + frame->source_file = dwfl_lineinfo(line, &addr, &lineno, + &colno, NULL, NULL); + frame->lineno = lineno; + frame->colno = colno; + } + + if (line == NULL || frame->source_file == NULL) { + frame->source_file = NULL; + frame->lineno = 0; + frame->colno = 0; + } + return 0; +} + +int foreach_frame_outwards(Backtrace *bt, + int (*cb)(StgPtr, void*), + void *user_data) +{ + int n_chunks = bt->n_frames / BACKTRACE_CHUNK_SZ; + if (bt->n_frames % BACKTRACE_CHUNK_SZ != 0) + n_chunks++; + + BacktraceChunk **chunks = + stgMallocBytes(n_chunks * sizeof(BacktraceChunk *), + "foreach_frame_outwards"); + + // First build a list of chunks, ending with the inner-most chunk + int chunk_idx; + chunks[0] = bt->last; + for (chunk_idx = 1; chunk_idx < n_chunks; chunk_idx++) { + chunks[chunk_idx] = chunks[chunk_idx-1]->next; + } + + // Now iterate back through the frames + int res = 0; + for (chunk_idx = n_chunks-1; chunk_idx >= 0 && res == 0; chunk_idx--) { + unsigned int i; + BacktraceChunk *chunk = chunks[chunk_idx]; + for (i = 0; i < chunk->n_frames; i++) { + res = cb(chunk->frames[i], user_data); + if (res != 0) break; + } + } + free(chunks); + return res; +} + +struct PrintData { + LibDwSession *session; + FILE *file; +}; + +static int print_frame(StgPtr pc, void *cbdata) +{ + struct PrintData *pd = (struct PrintData *) cbdata; + Location loc; + libdw_lookup_location(pd->session, &loc, pc); + fprintf(pd->file, " %24p %s ", + (void*) pc, loc.function); + if (loc.source_file) + fprintf(pd->file, "(%s:%d.%d)\n", + loc.source_file, loc.lineno, loc.colno); + else + fprintf(pd->file, "(%s)\n", loc.object_file); + return 0; +} + +void libdw_print_backtrace(LibDwSession *session, FILE *file, Backtrace *bt) { + if (bt == NULL) { + fprintf(file, "Warning: tried to print failed backtrace\n"); + return; + } + + struct PrintData pd = { session, file }; + foreach_frame_outwards(bt, print_frame, &pd); +} + +// Remember that we are traversing from the inner-most to the outer-most frame +static int frame_cb(Dwfl_Frame *frame, void *arg) { + LibDwSession *session = arg; + Dwarf_Addr pc; + bool is_activation; + if (! dwfl_frame_pc(frame, &pc, &is_activation)) { + // failed to find PC + backtrace_push(session->cur_bt, 0x0); + } else { + if (is_activation) + pc -= 1; // TODO: is this right? + backtrace_push(session->cur_bt, (StgPtr) pc); + } + + if ((void *) pc == &stg_stop_thread_info) + return DWARF_CB_ABORT; + else + return DWARF_CB_OK; +} + +Backtrace *libdw_get_backtrace(LibDwSession *session) { + if (session->cur_bt != NULL) { + sysErrorBelch("Already collecting backtrace. Uh oh."); + return NULL; + } + + Backtrace *bt = backtrace_alloc(); + session->cur_bt = bt; + + int pid = getpid(); + int ret = dwfl_getthread_frames(session->dwfl, pid, frame_cb, session); + if (ret == -1) + sysErrorBelch("Failed to get stack frames of current process: %s", + dwfl_errmsg(dwfl_errno())); + + session->cur_bt = NULL; + return bt; +} + +static pid_t next_thread(Dwfl *dwfl, void *arg, void **thread_argp) { + /* there is only the current thread */ + if (*thread_argp != NULL) + return 0; + + *thread_argp = arg; + return dwfl_pid(dwfl); +} + +static bool memory_read(Dwfl *dwfl STG_UNUSED, Dwarf_Addr addr, + Dwarf_Word *result, void *arg STG_UNUSED) { + *result = *(Dwarf_Word *) addr; + return true; +} + +static bool set_initial_registers(Dwfl_Thread *thread, void *arg); + +#ifdef x86_64_HOST_ARCH +static bool set_initial_registers(Dwfl_Thread *thread, + void *arg STG_UNUSED) { + Dwarf_Word regs[17]; + __asm__ ("movq %%rax, 0x00(%0)\n\t" + "movq %%rdx, 0x08(%0)\n\t" + "movq %%rcx, 0x10(%0)\n\t" + "movq %%rbx, 0x18(%0)\n\t" + "movq %%rsi, 0x20(%0)\n\t" + "movq %%rdi, 0x28(%0)\n\t" + "movq %%rbp, 0x30(%0)\n\t" + "movq %%rsp, 0x38(%0)\n\t" + "movq %%r8, 0x40(%0)\n\t" + "movq %%r9, 0x48(%0)\n\t" + "movq %%r10, 0x50(%0)\n\t" + "movq %%r11, 0x58(%0)\n\t" + "movq %%r12, 0x60(%0)\n\t" + "movq %%r13, 0x68(%0)\n\t" + "movq %%r14, 0x70(%0)\n\t" + "movq %%r15, 0x78(%0)\n\t" + "lea 0(%%rip), %%rax\n\t" + "movq %%rax, 0x80(%0)\n\t" + : /* no output */ + :"r" (®s[0]) /* input */ + :"%rax" /* clobbered */ + ); + return dwfl_thread_state_registers(thread, 0, 17, regs); +} +#endif +#ifdef i386_HOST_ARCH +static bool set_initial_registers(Dwfl_Thread *thread, + void *arg STG_UNUSED) { + Dwarf_Word regs[9]; + __asm__ ("movl %%eax, 0x00(%0)\n\t" + "movl %%ecx, 0x04(%0)\n\t" + "movl %%edx, 0x08(%0)\n\t" + "movl %%ebx, 0x0c(%0)\n\t" + "movl %%esp, 0x10(%0)\n\t" + "movl %%ebp, 0x14(%0)\n\t" + "movl %%esp, 0x18(%0)\n\t" + "movl %%edi, 0x1c(%0)\n\t" + "lea 0(%%eip), %%eax\n\t" + "movl %%eax, 0x20(%0)\n\t" + : /* no output */ + :"r" (®s[0]) /* input */ + :"%eax" /* clobbered */ + ); + return dwfl_thread_state_registers(thread, 0, 9, regs); +} +#endif + +static const Dwfl_Thread_Callbacks thread_cbs = { + .next_thread = next_thread, + .memory_read = memory_read, + .set_initial_registers = set_initial_registers, +}; + +#endif /* USE_LIBDW */ diff --git a/rts/Libdw.h b/rts/Libdw.h new file mode 100644 index 0000000000..8c5f1cadb8 --- /dev/null +++ b/rts/Libdw.h @@ -0,0 +1,58 @@ +/* ----------------------------------------------------------------------------- + * + * (c) The GHC Team, 2014-2015 + * + * Producing stacktraces with DWARF unwinding using libdw.. + * + * Do not #include this file directly: #include "Rts.h" instead. + * + * To understand the structure of the RTS headers, see the wiki: + * http://ghc.haskell.org/trac/ghc/wiki/Commentary/SourceTree/Includes + * + * -------------------------------------------------------------------------- */ + +#ifndef LIBDW_H +#define LIBDW_H + +#include "BeginPrivate.h" + +#ifdef USE_LIBDW + +struct LibDwSession_; +typedef struct LibDwSession_ LibDwSession; + +/* Begin a libdw session. A session is tied to a particular capability */ +LibDwSession *libdw_init(void); + +/* Free a session */ +void libdw_free(LibDwSession *session); + +/* Request a backtrace of the current stack state */ +Backtrace *libdw_get_backtrace(LibDwSession *session); + +/* Lookup Location information for the given address. + * Returns 0 if successful, 1 if address could not be found. */ +int libdw_lookup_location(LibDwSession *session, Location *loc, StgPtr pc); + +/* Pretty-print a backtrace to std*/ +void libdw_print_backtrace(LibDwSession *session, FILE *file, Backtrace *bt); + +// Traverse backtrace in order of outer-most to inner-most frame +#define FOREACH_FRAME_INWARDS(pc, bt) \ + BacktraceChunk *_chunk; \ + unsigned int _frame_idx; \ + for (_chunk = &bt->frames; _chunk != NULL; _chunk = _chunk->next) \ + for (_frame_idx=0; \ + pc = _chunk->frames[_frame_idx], _frame_idx < _chunk->n_frames; \ + _frame_idx++) + +// Traverse a backtrace in order of inner-most to outer-most frame +int foreach_frame_outwards(Backtrace *bt, + int (*cb)(StgPtr, void*), + void *user_data); + +#endif /* USE_LIBDW */ + +#include "EndPrivate.h" + +#endif /* LIBDW_H */ diff --git a/rts/ghc.mk b/rts/ghc.mk index 84f1b761b9..4b7f28ad89 100644 --- a/rts/ghc.mk +++ b/rts/ghc.mk @@ -489,6 +489,14 @@ rts_PACKAGE_CPP_OPTS += '-DFFI_LIB="C$(LIBFFI_NAME)"' endif +#----------------------------------------------------------------------------- +# Add support for reading DWARF debugging information, if available + +ifeq "$(GhcRtsWithLibdw)" "YES" +rts_CC_OPTS += -DUSE_LIBDW +rts_PACKAGE_CPP_OPTS += -DUSE_LIBDW +endif + # ----------------------------------------------------------------------------- # dependencies diff --git a/rts/package.conf.in b/rts/package.conf.in index 0a096583c5..97cc4f98fa 100644 --- a/rts/package.conf.in +++ b/rts/package.conf.in @@ -56,6 +56,10 @@ extra-libraries: #if USE_PAPI , "papi" #endif +#ifdef USE_LIBDW + , "elf" + , "dw" /* for backtraces */ +#endif #ifdef INSTALLING include-dirs: INCLUDE_DIR PAPI_INCLUDE_DIR FFI_INCLUDE_DIR |