139 lines
3.3 KiB
C++
139 lines
3.3 KiB
C++
//
|
|
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2021
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
#include "td/utils/port/stacktrace.h"
|
|
|
|
#include "td/utils/port/signals.h"
|
|
|
|
#if __GLIBC__
|
|
#include <execinfo.h>
|
|
#endif
|
|
|
|
#if TD_LINUX || TD_FREEBSD
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <unistd.h>
|
|
|
|
#if TD_LINUX
|
|
#include <sys/prctl.h>
|
|
#endif
|
|
#endif
|
|
|
|
namespace td {
|
|
|
|
namespace {
|
|
|
|
void print_backtrace(void) {
|
|
void *buffer[128];
|
|
#if TD_PORT_WINDOWS
|
|
USHORT nptrs = CaptureStackBackTrace(0, 128, buffer, nullptr);
|
|
#elif __GLIBC__
|
|
int nptrs = backtrace(buffer, 128);
|
|
#else
|
|
return;
|
|
#endif
|
|
|
|
signal_safe_write("------- Stack Backtrace -------\n", false);
|
|
#if TD_PORT_WINDOWS
|
|
for (USHORT i = 0; i < nptrs; i++) {
|
|
signal_safe_write_pointer(buffer[i], false);
|
|
}
|
|
#elif __GLIBC__
|
|
backtrace_symbols_fd(buffer, nptrs, 2);
|
|
#endif
|
|
signal_safe_write("-------------------------------\n", false);
|
|
}
|
|
|
|
void print_backtrace_gdb(void) {
|
|
#if TD_LINUX || TD_FREEBSD
|
|
char pid_buf[30];
|
|
char *pid_buf_begin = pid_buf + sizeof(pid_buf);
|
|
pid_t pid = getpid();
|
|
*--pid_buf_begin = '\0';
|
|
do {
|
|
*--pid_buf_begin = static_cast<char>(pid % 10 + '0');
|
|
pid /= 10;
|
|
} while (pid > 0);
|
|
|
|
char name_buf[512];
|
|
ssize_t res = readlink("/proc/self/exe", name_buf, 511); // TODO works only under Linux
|
|
if (res >= 0) {
|
|
name_buf[res] = 0;
|
|
|
|
#if TD_LINUX
|
|
#if defined(PR_SET_DUMPABLE)
|
|
if (prctl(PR_SET_DUMPABLE, 1, 0, 0, 0) < 0) {
|
|
signal_safe_write("Can't set dumpable\n");
|
|
return;
|
|
}
|
|
#endif
|
|
#if defined(PR_SET_PTRACER)
|
|
// We can't use event fd because we are in a signal handler
|
|
int fds[2];
|
|
bool need_set_ptracer = true;
|
|
if (pipe(fds) < 0) {
|
|
need_set_ptracer = false;
|
|
signal_safe_write("Can't create a pipe\n");
|
|
}
|
|
#endif
|
|
#endif
|
|
|
|
int child_pid = fork();
|
|
if (child_pid < 0) {
|
|
signal_safe_write("Can't fork() to run gdb\n");
|
|
return;
|
|
}
|
|
if (!child_pid) {
|
|
#if TD_LINUX && defined(PR_SET_PTRACER)
|
|
if (need_set_ptracer) {
|
|
char c;
|
|
if (read(fds[0], &c, 1) < 0) {
|
|
signal_safe_write("Failed to read from pipe\n");
|
|
}
|
|
}
|
|
#endif
|
|
dup2(2, 1); // redirect output to stderr
|
|
execlp("gdb", "gdb", "--batch", "-n", "-ex", "thread", "-ex", "thread apply all bt full", name_buf, pid_buf_begin,
|
|
nullptr);
|
|
return;
|
|
} else {
|
|
#if TD_LINUX && defined(PR_SET_PTRACER)
|
|
if (need_set_ptracer) {
|
|
if (prctl(PR_SET_PTRACER, child_pid, 0, 0, 0) < 0) {
|
|
signal_safe_write("Can't set ptracer\n");
|
|
}
|
|
if (write(fds[1], "a", 1) != 1) {
|
|
signal_safe_write("Can't write to pipe\n");
|
|
}
|
|
}
|
|
#endif
|
|
waitpid(child_pid, nullptr, 0);
|
|
}
|
|
} else {
|
|
signal_safe_write("Can't get name of executable file to pass to gdb\n");
|
|
}
|
|
#endif
|
|
}
|
|
|
|
} // namespace
|
|
|
|
void Stacktrace::print_to_stderr(const PrintOptions &options) {
|
|
print_backtrace();
|
|
if (options.use_gdb) {
|
|
print_backtrace_gdb();
|
|
}
|
|
}
|
|
|
|
void Stacktrace::init() {
|
|
#if __GLIBC__
|
|
// backtrace needs to be called once to ensure that next calls are async-signal-safe
|
|
void *buffer[1];
|
|
backtrace(buffer, 1);
|
|
#endif
|
|
}
|
|
|
|
} // namespace td
|