mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-26 16:30:56 +00:00
06d3487b3f
Summary: Some scripts (like regression_build_test.sh) redirect stdio to a tmp file and delete it on exit. This would miss the stack trace output on segfault. Output to stderr would hopefully show us the stack trace in the continuous build output. Test Plan: ./signal_test, make check Reviewers: dhruba Reviewed By: dhruba CC: leveldb Differential Revision: https://reviews.facebook.net/D10485
98 lines
2.2 KiB
C++
98 lines
2.2 KiB
C++
#include "util/stack_trace.h"
|
|
|
|
#ifdef OS_LINUX
|
|
|
|
#include <execinfo.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
namespace leveldb {
|
|
|
|
static const char* GetExecutableName()
|
|
{
|
|
static char name[1024];
|
|
|
|
char link[1024];
|
|
snprintf(link, sizeof(link), "/proc/%d/exe", getpid());
|
|
auto read = readlink(link, name, sizeof(name));
|
|
if (-1 == read) {
|
|
return nullptr;
|
|
} else {
|
|
name[read] = 0;
|
|
return name;
|
|
}
|
|
}
|
|
|
|
static void StackTraceHandler(int sig) {
|
|
// reset to default handler
|
|
signal(sig, SIG_DFL);
|
|
|
|
fprintf(stderr, "Received signal %d (%s)\n", sig, strsignal(sig));
|
|
|
|
const int kMaxFrames = 100;
|
|
void *frames[kMaxFrames];
|
|
|
|
auto num_frames = backtrace(frames, kMaxFrames);
|
|
auto symbols = backtrace_symbols(frames, num_frames);
|
|
|
|
auto executable = GetExecutableName();
|
|
|
|
const int kSkip = 2; // skip the top two signal handler related frames
|
|
|
|
for (int i = kSkip; i < num_frames; ++i)
|
|
{
|
|
fprintf(stderr, "#%-2d %p ", i - kSkip, frames[i]);
|
|
if (symbols) {
|
|
fprintf(stderr, "%s ", symbols[i]);
|
|
}
|
|
if (executable) {
|
|
// out source to addr2line, for the address translation
|
|
const int kLineMax = 256;
|
|
char cmd[kLineMax];
|
|
sprintf(cmd,"addr2line %p -e %s 2>&1", frames[i] , executable);
|
|
auto f = popen(cmd, "r");
|
|
if (f) {
|
|
char line[kLineMax];
|
|
while (fgets(line, sizeof(line), f)) {
|
|
fprintf(stderr, "%s", line);
|
|
}
|
|
pclose(f);
|
|
} else {
|
|
fprintf(stderr, "\n");
|
|
}
|
|
} else {
|
|
fprintf(stderr, "\n");
|
|
}
|
|
}
|
|
|
|
// re-signal to default handler (so we still get core dump if needed...)
|
|
raise(sig);
|
|
}
|
|
|
|
void InstallStackTraceHandler() {
|
|
// just use the plain old signal as it's simple and sufficient
|
|
// for this use case
|
|
signal(SIGILL, StackTraceHandler);
|
|
signal(SIGSEGV, StackTraceHandler);
|
|
signal(SIGBUS, StackTraceHandler);
|
|
signal(SIGABRT, StackTraceHandler);
|
|
|
|
printf("Installed stack trace handler for SIGILL SIGSEGV SIGBUS SIGABRT\n");
|
|
|
|
}
|
|
|
|
} // namespace leveldb
|
|
|
|
#else // no-op for non-linux system for now
|
|
|
|
namespace leveldb {
|
|
|
|
void InstallStackTraceHandler() {}
|
|
|
|
}
|
|
|
|
#endif // OS_LINUX
|