| // Copyright 2008 Google Inc. All Rights Reserved. |
| // Author: hamaji@google.com (Shinichiro Hamaji) |
| |
| #include "utilities.h" |
| |
| #include <signal.h> |
| #ifdef HAVE_SYS_TIME_H |
| # include <sys/time.h> |
| #endif |
| #include <time.h> |
| #if defined(HAVE_SYSCALL_H) |
| #include <syscall.h> // for syscall() |
| #elif defined(HAVE_SYS_SYSCALL_H) |
| #include <sys/syscall.h> // for syscall() |
| #endif |
| |
| #include "base/googleinit.h" |
| |
| using std::string; |
| |
| _START_GOOGLE_NAMESPACE_ |
| |
| static const char* g_program_invocation_short_name = NULL; |
| static pthread_t g_main_thread_id; |
| |
| _END_GOOGLE_NAMESPACE_ |
| |
| // The following APIs are all internal. |
| #ifdef HAVE_STACKTRACE |
| |
| #include "stacktrace.h" |
| #include "symbolize.h" |
| #include "base/commandlineflags.h" |
| |
| DEFINE_bool(symbolize_stacktrace, true, |
| "Symbolize the stack trace in the tombstone"); |
| |
| _START_GOOGLE_NAMESPACE_ |
| |
| typedef void DebugWriter(const char*, void*); |
| |
| // The %p field width for printf() functions is two characters per byte. |
| // For some environments, add two extra bytes for the leading "0x". |
| static const int kPrintfPointerFieldWidth = 2 + 2 * sizeof(void*); |
| |
| static void DebugWriteToStderr(const char* data, void *unused) { |
| // This one is signal-safe. |
| write(STDERR_FILENO, data, strlen(data)); |
| } |
| |
| void DebugWriteToString(const char* data, void *arg) { |
| reinterpret_cast<string*>(arg)->append(data); |
| } |
| |
| // Print a program counter and its symbol name. |
| static void DumpPCAndSymbol(DebugWriter *writerfn, void *arg, void *pc, |
| const char * const prefix) { |
| char tmp[1024]; |
| const char *symbol = "(unknown)"; |
| // Symbolizes the previous address of pc because pc may be in the |
| // next function. The overrun happens when the function ends with |
| // a call to a function annotated noreturn (e.g. CHECK). |
| if (Symbolize(reinterpret_cast<char *>(pc) - 1, tmp, sizeof(tmp))) { |
| symbol = tmp; |
| } |
| char buf[1024]; |
| snprintf(buf, sizeof(buf), "%s@ %*p %s\n", |
| prefix, kPrintfPointerFieldWidth, pc, symbol); |
| writerfn(buf, arg); |
| } |
| |
| // Print a program counter and the corresponding stack frame size. |
| static void DumpPCAndFrameSize(DebugWriter *writerfn, void *arg, void *pc, |
| int framesize, const char * const prefix) { |
| char buf[100]; |
| if (framesize <= 0) { |
| snprintf(buf, sizeof(buf), "%s@ %*p (unknown)\n", |
| prefix, kPrintfPointerFieldWidth, pc); |
| } else { |
| snprintf(buf, sizeof(buf), "%s@ %*p %9d\n", |
| prefix, kPrintfPointerFieldWidth, pc, framesize); |
| } |
| writerfn(buf, arg); |
| } |
| |
| static void DumpPC(DebugWriter *writerfn, void *arg, void *pc, |
| const char * const prefix) { |
| char buf[100]; |
| snprintf(buf, sizeof(buf), "%s@ %*p\n", |
| prefix, kPrintfPointerFieldWidth, pc); |
| writerfn(buf, arg); |
| } |
| |
| // Dump current stack trace as directed by writerfn |
| static void DumpStackTrace(int skip_count, DebugWriter *writerfn, void *arg) { |
| // Print stack trace |
| void* stack[32]; |
| int depth = GetStackTrace(stack, ARRAYSIZE(stack), skip_count+1); |
| for (int i = 0; i < depth; i++) { |
| #if defined(HAVE_SYMBOLIZE) |
| if (FLAGS_symbolize_stacktrace) { |
| DumpPCAndSymbol(writerfn, arg, stack[i], " "); |
| } else { |
| DumpPC(writerfn, arg, stack[i], " "); |
| } |
| #else |
| DumpPC(writerfn, arg, stack[i], " "); |
| #endif |
| } |
| } |
| |
| static void DumpStackTraceAndExit() { |
| DumpStackTrace(1, DebugWriteToStderr, NULL); |
| |
| // Set the default signal handler for SIGABRT, to avoid invoking our |
| // own signal handler installed by InstallFailedSignalHandler(). |
| struct sigaction sig_action = {}; // Zero-clear. |
| sigemptyset(&sig_action.sa_mask); |
| sig_action.sa_handler = SIG_DFL; |
| sigaction(SIGABRT, &sig_action, NULL); |
| |
| abort(); |
| } |
| |
| _END_GOOGLE_NAMESPACE_ |
| |
| #endif // HAVE_STACKTRACE |
| |
| _START_GOOGLE_NAMESPACE_ |
| |
| namespace glog_internal_namespace_ { |
| |
| const char* ProgramInvocationShortName() { |
| if (g_program_invocation_short_name != NULL) { |
| return g_program_invocation_short_name; |
| } else { |
| // TODO(hamaji): Use /proc/self/cmdline and so? |
| return "UNKNOWN"; |
| } |
| } |
| |
| bool IsGoogleLoggingInitialized() { |
| return g_program_invocation_short_name != NULL; |
| } |
| |
| bool is_default_thread() { |
| if (g_program_invocation_short_name == NULL) { |
| // InitGoogleLogging() not yet called, so unlikely to be in a different |
| // thread |
| return true; |
| } else { |
| return pthread_equal(pthread_self(), g_main_thread_id); |
| } |
| } |
| |
| int64 CycleClock_Now() { |
| // TODO(hamaji): temporary impementation - it might be too slow. |
| #ifdef OS_WINDOWS |
| SYSTEMTIME now; |
| GetSystemTime(&now); |
| return (static_cast<int64>(now.wSecond) * 1000000 + |
| static_cast<int64>(now.wMilliseconds) * 1000); |
| #else |
| struct timeval tv; |
| gettimeofday(&tv, NULL); |
| return static_cast<int64>(tv.tv_sec) * 1000000 + tv.tv_usec; |
| #endif |
| } |
| |
| int64 UsecToCycles(int64 usec) { |
| return usec; |
| } |
| |
| WallTime WallTime_Now() { |
| // Now, cycle clock is retuning microseconds since the epoch. |
| return CycleClock_Now() * 0.000001; |
| } |
| |
| static int32 g_main_thread_pid = getpid(); |
| int32 GetMainThreadPid() { |
| return g_main_thread_pid; |
| } |
| |
| pid_t GetTID() { |
| // On Linux and FreeBSD, we try to use gettid(). |
| #if defined OS_LINUX || defined OS_FREEBSD || defined OS_MACOSX |
| #ifndef __NR_gettid |
| #ifdef OS_MACOSX |
| #define __NR_gettid SYS_gettid |
| #elif ! defined __i386__ |
| #error "Must define __NR_gettid for non-x86 platforms" |
| #else |
| #define __NR_gettid 224 |
| #endif |
| #endif |
| static bool lacks_gettid = false; |
| if (!lacks_gettid) { |
| pid_t tid = syscall(__NR_gettid); |
| if (tid != -1) { |
| return tid; |
| } |
| // Technically, this variable has to be volatile, but there is a small |
| // performance penalty in accessing volatile variables and there should |
| // not be any serious adverse effect if a thread does not immediately see |
| // the value change to "true". |
| lacks_gettid = true; |
| } |
| #endif // OS_LINUX || OS_FREEBSD |
| |
| // If gettid() could not be used, we use one of the following. |
| #if defined OS_LINUX |
| return getpid(); // Linux: getpid returns thread ID when gettid is absent |
| #elif defined OS_WINDOWS || defined OS_CYGWIN |
| return GetCurrentThreadId(); |
| #else |
| // If none of the techniques above worked, we use pthread_self(). |
| return (pid_t)pthread_self(); |
| #endif |
| } |
| |
| const char* const_basename(const char* filepath) { |
| const char* base = strrchr(filepath, '/'); |
| #ifdef OS_WINDOWS // Look for either path separator in Windows |
| if (!base) |
| base = strrchr(filepath, '\\'); |
| #endif |
| return base ? (base+1) : filepath; |
| } |
| |
| static string g_my_user_name; |
| const string& MyUserName() { |
| return g_my_user_name; |
| } |
| static void MyUserNameInitializer() { |
| // TODO(hamaji): Probably this is not portable. |
| const char* user = getenv("USER"); |
| if (user != NULL) { |
| g_my_user_name = user; |
| } else { |
| g_my_user_name = "invalid-user"; |
| } |
| } |
| REGISTER_MODULE_INITIALIZER(utilities, MyUserNameInitializer()); |
| |
| #ifdef HAVE_STACKTRACE |
| void DumpStackTraceToString(string* stacktrace) { |
| DumpStackTrace(1, DebugWriteToString, stacktrace); |
| } |
| #endif |
| |
| // We use an atomic operation to prevent problems with calling CrashReason |
| // from inside the Mutex implementation (potentially through RAW_CHECK). |
| static const CrashReason* g_reason = 0; |
| |
| void SetCrashReason(const CrashReason* r) { |
| sync_val_compare_and_swap(&g_reason, |
| reinterpret_cast<const CrashReason*>(0), |
| r); |
| } |
| |
| } // namespace glog_internal_namespace_ |
| |
| void InitGoogleLogging(const char* argv0) { |
| const char* slash = strrchr(argv0, '/'); |
| #ifdef OS_WINDOWS |
| if (!slash) slash = strrchr(argv0, '\\'); |
| #endif |
| g_program_invocation_short_name = slash ? slash + 1 : argv0; |
| g_main_thread_id = pthread_self(); |
| |
| #ifdef HAVE_STACKTRACE |
| InstallFailureFunction(&DumpStackTraceAndExit); |
| #endif |
| } |
| |
| _END_GOOGLE_NAMESPACE_ |
| |
| // Make an implementation of stacktrace compiled. |
| #ifdef STACKTRACE_H |
| # include STACKTRACE_H |
| #endif |