mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
284 lines
8.7 KiB
284 lines
8.7 KiB
/* |
|
* |
|
* Copyright 2015, Google Inc. |
|
* All rights reserved. |
|
* |
|
* Redistribution and use in source and binary forms, with or without |
|
* modification, are permitted provided that the following conditions are |
|
* met: |
|
* |
|
* * Redistributions of source code must retain the above copyright |
|
* notice, this list of conditions and the following disclaimer. |
|
* * Redistributions in binary form must reproduce the above |
|
* copyright notice, this list of conditions and the following disclaimer |
|
* in the documentation and/or other materials provided with the |
|
* distribution. |
|
* * Neither the name of Google Inc. nor the names of its |
|
* contributors may be used to endorse or promote products derived from |
|
* this software without specific prior written permission. |
|
* |
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
* |
|
*/ |
|
|
|
#include "test/core/util/test_config.h" |
|
|
|
#include <grpc/support/log.h> |
|
#include <signal.h> |
|
#include <stdbool.h> |
|
#include <stdio.h> |
|
#include <stdlib.h> |
|
#include "src/core/lib/support/string.h" |
|
|
|
double g_fixture_slowdown_factor = 1.0; |
|
|
|
#if GPR_GETPID_IN_UNISTD_H |
|
#include <unistd.h> |
|
static unsigned seed(void) { return (unsigned)getpid(); } |
|
#endif |
|
|
|
#if GPR_GETPID_IN_PROCESS_H |
|
#include <process.h> |
|
static unsigned seed(void) { return (unsigned)_getpid(); } |
|
#endif |
|
|
|
#if GPR_WINDOWS_CRASH_HANDLER |
|
#include <windows.h> |
|
|
|
#include <tchar.h> |
|
|
|
// disable warning 4091 - dbghelp.h is broken for msvc2015 |
|
#pragma warning(disable : 4091) |
|
#define DBGHELP_TRANSLATE_TCHAR |
|
#include <dbghelp.h> |
|
|
|
#ifdef _MSC_VER |
|
#pragma comment(lib, "dbghelp.lib") |
|
#endif |
|
|
|
static void print_current_stack() { |
|
typedef USHORT(WINAPI * CaptureStackBackTraceType)( |
|
__in ULONG, __in ULONG, __out PVOID *, __out_opt PULONG); |
|
CaptureStackBackTraceType func = (CaptureStackBackTraceType)(GetProcAddress( |
|
LoadLibrary(_T("kernel32.dll")), "RtlCaptureStackBackTrace")); |
|
|
|
if (func == NULL) return; // WOE 29.SEP.2010 |
|
|
|
// Quote from Microsoft Documentation: |
|
// ## Windows Server 2003 and Windows XP: |
|
// ## The sum of the FramesToSkip and FramesToCapture parameters must be less |
|
// than 63. |
|
#define MAX_CALLERS 62 |
|
|
|
void *callers_stack[MAX_CALLERS]; |
|
unsigned short frames; |
|
SYMBOL_INFOW *symbol; |
|
HANDLE process; |
|
process = GetCurrentProcess(); |
|
SymInitialize(process, NULL, TRUE); |
|
frames = (func)(0, MAX_CALLERS, callers_stack, NULL); |
|
symbol = |
|
(SYMBOL_INFOW *)calloc(sizeof(SYMBOL_INFOW) + 256 * sizeof(wchar_t), 1); |
|
symbol->MaxNameLen = 255; |
|
symbol->SizeOfStruct = sizeof(SYMBOL_INFOW); |
|
|
|
const unsigned short MAX_CALLERS_SHOWN = 32; |
|
frames = frames < MAX_CALLERS_SHOWN ? frames : MAX_CALLERS_SHOWN; |
|
for (unsigned int i = 0; i < frames; i++) { |
|
SymFromAddrW(process, (DWORD64)(callers_stack[i]), 0, symbol); |
|
fwprintf(stderr, L"*** %d: %016I64X %ls - %016I64X\n", i, |
|
(DWORD64)callers_stack[i], symbol->Name, (DWORD64)symbol->Address); |
|
fflush(stderr); |
|
} |
|
|
|
free(symbol); |
|
} |
|
|
|
static void print_stack_from_context(CONTEXT c) { |
|
STACKFRAME s; // in/out stackframe |
|
memset(&s, 0, sizeof(s)); |
|
DWORD imageType; |
|
#ifdef _M_IX86 |
|
// normally, call ImageNtHeader() and use machine info from PE header |
|
imageType = IMAGE_FILE_MACHINE_I386; |
|
s.AddrPC.Offset = c.Eip; |
|
s.AddrPC.Mode = AddrModeFlat; |
|
s.AddrFrame.Offset = c.Ebp; |
|
s.AddrFrame.Mode = AddrModeFlat; |
|
s.AddrStack.Offset = c.Esp; |
|
s.AddrStack.Mode = AddrModeFlat; |
|
#elif _M_X64 |
|
imageType = IMAGE_FILE_MACHINE_AMD64; |
|
s.AddrPC.Offset = c.Rip; |
|
s.AddrPC.Mode = AddrModeFlat; |
|
s.AddrFrame.Offset = c.Rsp; |
|
s.AddrFrame.Mode = AddrModeFlat; |
|
s.AddrStack.Offset = c.Rsp; |
|
s.AddrStack.Mode = AddrModeFlat; |
|
#elif _M_IA64 |
|
imageType = IMAGE_FILE_MACHINE_IA64; |
|
s.AddrPC.Offset = c.StIIP; |
|
s.AddrPC.Mode = AddrModeFlat; |
|
s.AddrFrame.Offset = c.IntSp; |
|
s.AddrFrame.Mode = AddrModeFlat; |
|
s.AddrBStore.Offset = c.RsBSP; |
|
s.AddrBStore.Mode = AddrModeFlat; |
|
s.AddrStack.Offset = c.IntSp; |
|
s.AddrStack.Mode = AddrModeFlat; |
|
#else |
|
#error "Platform not supported!" |
|
#endif |
|
|
|
HANDLE process = GetCurrentProcess(); |
|
HANDLE thread = GetCurrentThread(); |
|
|
|
SYMBOL_INFOW *symbol = |
|
(SYMBOL_INFOW *)calloc(sizeof(SYMBOL_INFOW) + 256 * sizeof(wchar_t), 1); |
|
symbol->MaxNameLen = 255; |
|
symbol->SizeOfStruct = sizeof(SYMBOL_INFOW); |
|
|
|
while (StackWalk(imageType, process, thread, &s, &c, 0, |
|
SymFunctionTableAccess, SymGetModuleBase, 0)) { |
|
BOOL has_symbol = |
|
SymFromAddrW(process, (DWORD64)(s.AddrPC.Offset), 0, symbol); |
|
fwprintf( |
|
stderr, L"*** %016I64X %ls - %016I64X\n", (DWORD64)(s.AddrPC.Offset), |
|
has_symbol ? symbol->Name : L"<<no symbol>>", (DWORD64)symbol->Address); |
|
fflush(stderr); |
|
} |
|
|
|
free(symbol); |
|
} |
|
|
|
static LONG crash_handler(struct _EXCEPTION_POINTERS *ex_info) { |
|
fprintf(stderr, "Exception handler called, dumping information\n"); |
|
bool try_to_print_stack = true; |
|
PEXCEPTION_RECORD exrec = ex_info->ExceptionRecord; |
|
while (exrec) { |
|
DWORD code = exrec->ExceptionCode; |
|
DWORD flgs = exrec->ExceptionFlags; |
|
PVOID addr = exrec->ExceptionAddress; |
|
if (code == EXCEPTION_STACK_OVERFLOW) try_to_print_stack = false; |
|
fprintf(stderr, "code: %x - flags: %d - address: %p\n", code, flgs, addr); |
|
exrec = exrec->ExceptionRecord; |
|
} |
|
if (try_to_print_stack) { |
|
print_stack_from_context(*ex_info->ContextRecord); |
|
} |
|
if (IsDebuggerPresent()) { |
|
__debugbreak(); |
|
} else { |
|
_exit(1); |
|
} |
|
return EXCEPTION_EXECUTE_HANDLER; |
|
} |
|
|
|
static void abort_handler(int sig) { |
|
fprintf(stderr, "Abort handler called.\n"); |
|
print_current_stack(NULL); |
|
if (IsDebuggerPresent()) { |
|
__debugbreak(); |
|
} else { |
|
_exit(1); |
|
} |
|
} |
|
|
|
static void install_crash_handler() { |
|
if (!SymInitialize(GetCurrentProcess(), NULL, TRUE)) { |
|
fprintf(stderr, "SymInitialize failed: %d\n", GetLastError()); |
|
} |
|
SetUnhandledExceptionFilter((LPTOP_LEVEL_EXCEPTION_FILTER)crash_handler); |
|
_set_abort_behavior(0, _WRITE_ABORT_MSG); |
|
_set_abort_behavior(0, _CALL_REPORTFAULT); |
|
signal(SIGABRT, abort_handler); |
|
} |
|
#elif GPR_POSIX_CRASH_HANDLER |
|
#include <errno.h> |
|
#include <execinfo.h> |
|
#include <grpc/support/useful.h> |
|
#include <stdio.h> |
|
#include <string.h> |
|
|
|
static char g_alt_stack[GPR_MAX(MINSIGSTKSZ, 65536)]; |
|
|
|
#define MAX_FRAMES 32 |
|
|
|
/* signal safe output */ |
|
static void output_string(const char *string) { |
|
size_t len = strlen(string); |
|
ssize_t r; |
|
|
|
do { |
|
r = write(STDERR_FILENO, string, len); |
|
} while (r == -1 && errno == EINTR); |
|
} |
|
|
|
static void output_num(long num) { |
|
char buf[GPR_LTOA_MIN_BUFSIZE]; |
|
gpr_ltoa(num, buf); |
|
output_string(buf); |
|
} |
|
|
|
static void crash_handler(int signum, siginfo_t *info, void *data) { |
|
void *addrlist[MAX_FRAMES + 1]; |
|
int addrlen; |
|
|
|
output_string("\n\n\n*******************************\nCaught signal "); |
|
output_num(signum); |
|
output_string("\n"); |
|
|
|
addrlen = backtrace(addrlist, GPR_ARRAY_SIZE(addrlist)); |
|
|
|
if (addrlen == 0) { |
|
output_string(" no backtrace\n"); |
|
} else { |
|
backtrace_symbols_fd(addrlist, addrlen, STDERR_FILENO); |
|
} |
|
|
|
/* try to get a core dump for SIGTERM */ |
|
if (signum == SIGTERM) signum = SIGQUIT; |
|
raise(signum); |
|
} |
|
|
|
static void install_crash_handler() { |
|
stack_t ss; |
|
struct sigaction sa; |
|
|
|
memset(&ss, 0, sizeof(ss)); |
|
memset(&sa, 0, sizeof(sa)); |
|
ss.ss_size = sizeof(g_alt_stack); |
|
ss.ss_sp = g_alt_stack; |
|
GPR_ASSERT(sigaltstack(&ss, NULL) == 0); |
|
sa.sa_flags = (int)(SA_SIGINFO | SA_ONSTACK | SA_RESETHAND); |
|
sa.sa_sigaction = crash_handler; |
|
GPR_ASSERT(sigaction(SIGILL, &sa, NULL) == 0); |
|
GPR_ASSERT(sigaction(SIGABRT, &sa, NULL) == 0); |
|
GPR_ASSERT(sigaction(SIGBUS, &sa, NULL) == 0); |
|
GPR_ASSERT(sigaction(SIGSEGV, &sa, NULL) == 0); |
|
GPR_ASSERT(sigaction(SIGTERM, &sa, NULL) == 0); |
|
GPR_ASSERT(sigaction(SIGQUIT, &sa, NULL) == 0); |
|
} |
|
#else |
|
static void install_crash_handler() {} |
|
#endif |
|
|
|
void grpc_test_init(int argc, char **argv) { |
|
install_crash_handler(); |
|
gpr_log(GPR_DEBUG, "test slowdown: machine=%f build=%f total=%f", |
|
(double)GRPC_TEST_SLOWDOWN_MACHINE_FACTOR, |
|
(double)GRPC_TEST_SLOWDOWN_BUILD_FACTOR, |
|
(double)GRPC_TEST_SLOWDOWN_FACTOR); |
|
/* seed rng with pid, so we don't end up with the same random numbers as a |
|
concurrently running test binary */ |
|
srand(seed()); |
|
}
|
|
|