src/utilities.h

   1 // Copyright (c) 2008, Google Inc.
   2 // All rights reserved.
   3 //
   4 // Redistribution and use in source and binary forms, with or without
   5 // modification, are permitted provided that the following conditions are
   6 // met:
   7 //
   8 //     * Redistributions of source code must retain the above copyright
   9 // notice, this list of conditions and the following disclaimer.
  10 //     * Redistributions in binary form must reproduce the above
  11 // copyright notice, this list of conditions and the following disclaimer
  12 // in the documentation and/or other materials provided with the
  13 // distribution.
  14 //     * Neither the name of Google Inc. nor the names of its
  15 // contributors may be used to endorse or promote products derived from
  16 // this software without specific prior written permission.
  17 //
  18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29 //
  30 // Author: Shinichiro Hamaji
  31 //
  32 // Define utilties for glog internal usage.
  33
  34 #ifndef UTILITIES_H__
  35 #define UTILITIES_H__
  36
  37 #if defined(WIN32) || defined(_WIN32) || defined(__WIN32__)
  38 # define OS_WINDOWS
  39 #elif defined(__CYGWIN__) || defined(__CYGWIN32__)
  40 # define OS_CYGWIN
  41 #elif defined(linux) || defined(__linux) || defined(__linux__)
  42 # ifndef OS_LINUX
  43 #  define OS_LINUX
  44 # endif
  45 #elif defined(macintosh) || defined(__APPLE__) || defined(__APPLE_CC__)
  46 # define OS_MACOSX
  47 #elif defined(__FreeBSD__)
  48 # define OS_FREEBSD
  49 #elif defined(__NetBSD__)
  50 # define OS_NETBSD
  51 #elif defined(__OpenBSD__)
  52 # define OS_OPENBSD
  53 #else
  54 // TODO(hamaji): Add other platforms.
  55 #endif
  56
  57 // printf macros for size_t, in the style of inttypes.h
  58 #ifdef _LP64
  59 #define __PRIS_PREFIX "z"
  60 #else
  61 #define __PRIS_PREFIX
  62 #endif
  63
  64 // Use these macros after a % in a printf format string
  65 // to get correct 32/64 bit behavior, like this:
  66 // size_t size = records.size();
  67 // printf("%"PRIuS"\n", size);
  68
  69 #define PRIdS __PRIS_PREFIX "d"
  70 #define PRIxS __PRIS_PREFIX "x"
  71 #define PRIuS __PRIS_PREFIX "u"
  72 #define PRIXS __PRIS_PREFIX "X"
  73 #define PRIoS __PRIS_PREFIX "o"
  74
  75 #include "base/mutex.h"  // This must go first so we get _XOPEN_SOURCE
  76
  77 #include <string>
  78
  79 #if defined(OS_WINDOWS)
  80 # include "port.h"
  81 #endif
  82
  83 #include "config.h"
  84 #include "glog/logging.h"
  85
  86 // There are three different ways we can try to get the stack trace:
  87 //
  88 // 1) The libunwind library.  This is still in development, and as a
  89 //    separate library adds a new dependency, but doesn't need a frame
  90 //    pointer.  It also doesn't call malloc.
  91 //
  92 // 2) Our hand-coded stack-unwinder.  This depends on a certain stack
  93 //    layout, which is used by gcc (and those systems using a
  94 //    gcc-compatible ABI) on x86 systems, at least since gcc 2.95.
  95 //    It uses the frame pointer to do its work.
  96 //
  97 // 3) The gdb unwinder -- also the one used by the c++ exception code.
  98 //    It's obviously well-tested, but has a fatal flaw: it can call
  99 //    malloc() from the unwinder.  This is a problem because we're
 100 //    trying to use the unwinder to instrument malloc().
 101 //
 102 // Note: if you add a new implementation here, make sure it works
 103 // correctly when GetStackTrace() is called with max_depth == 0.
 104 // Some code may do that.
 105
 106 #if defined(HAVE_LIB_UNWIND)
 107 # define STACKTRACE_H "stacktrace_libunwind-inl.h"
 108 #elif !defined(NO_FRAME_POINTER)
 109 # if defined(__i386__) && __GNUC__ >= 2
 110 #  define STACKTRACE_H "stacktrace_x86-inl.h"
 111 # elif defined(__x86_64__) && __GNUC__ >= 2 && HAVE_UNWIND_H
 112 #  define STACKTRACE_H "stacktrace_x86_64-inl.h"
 113 # elif (defined(__ppc__) || defined(__PPC__)) && __GNUC__ >= 2
 114 #  define STACKTRACE_H "stacktrace_powerpc-inl.h"
 115 # endif
 116 #endif
 117
 118 #if !defined(STACKTRACE_H) && defined(HAVE_EXECINFO_H)
 119 # define STACKTRACE_H "stacktrace_generic-inl.h"
 120 #endif
 121
 122 #if defined(STACKTRACE_H)
 123 # define HAVE_STACKTRACE
 124 #endif
 125
 126 // defined by gcc
 127 #if defined(__ELF__) && defined(OS_LINUX)
 128 # define HAVE_SYMBOLIZE
 129 #elif defined(OS_MACOSX) && defined(HAVE_DLADDR)
 130 // Use dladdr to symbolize.
 131 # define HAVE_SYMBOLIZE
 132 #endif
 133
 134 #ifndef ARRAYSIZE
 135 // There is a better way, but this is good enough for our purpose.
 136 # define ARRAYSIZE(a) (sizeof(a) / sizeof(*(a)))
 137 #endif
 138
 139 _START_GOOGLE_NAMESPACE_
 140
 141 namespace glog_internal_namespace_ {
 142
 143 #ifdef HAVE___ATTRIBUTE__
 144 # define ATTRIBUTE_NOINLINE __attribute__ ((noinline))
 145 # define HAVE_ATTRIBUTE_NOINLINE
 146 #else
 147 # define ATTRIBUTE_NOINLINE
 148 #endif
 149
 150 const char* ProgramInvocationShortName();
 151
 152 bool IsGoogleLoggingInitialized();
 153
 154 bool is_default_thread();
 155
 156 int64 CycleClock_Now();
 157
 158 int64 UsecToCycles(int64 usec);
 159
 160 typedef double WallTime;
 161 WallTime WallTime_Now();
 162
 163 int32 GetMainThreadPid();
 164 bool PidHasChanged();
 165
 166 pid_t GetTID();
 167
 168 const std::string& MyUserName();
 169
 170 // Get the part of filepath after the last path separator.
 171 // (Doesn't modify filepath, contrary to basename() in libgen.h.)
 172 const char* const_basename(const char* filepath);
 173
 174 // Wrapper of __sync_val_compare_and_swap. If the GCC extension isn't
 175 // defined, we try the CPU specific logics (we only support x86 and
 176 // x86_64 for now) first, then use a naive implementation, which has a
 177 // race condition.
 178 template<typename T>
 179 inline T sync_val_compare_and_swap(T* ptr, T oldval, T newval) {
 180 #if defined(HAVE___SYNC_VAL_COMPARE_AND_SWAP)
 181   return __sync_val_compare_and_swap(ptr, oldval, newval);
 182 #elif defined(__GNUC__) && (defined(__i386__) || defined(__x86_64__))
 183   T ret;
 184   __asm__ __volatile__("lock; cmpxchg %1, (%2);"
 185                        :"=a"(ret)
 186                         // GCC may produces %sil or %dil for
 187                         // constraint "r", but some of apple's gas
 188                         // dosn't know the 8 bit registers.
 189                         // We use "q" to avoid these registers.
 190                        :"q"(newval), "q"(ptr), "a"(oldval)
 191                        :"memory", "cc");
 192   return ret;
 193 #else
 194   T ret = *ptr;
 195   if (ret == oldval) {
 196     *ptr = newval;
 197   }
 198   return ret;
 199 #endif
 200 }
 201
 202 void DumpStackTraceToString(std::string* stacktrace);
 203
 204 struct CrashReason {
 205   CrashReason() : filename(0), line_number(0), message(0), depth(0) {}
 206
 207   const char* filename;
 208   int line_number;
 209   const char* message;
 210
 211   // We'll also store a bit of stack trace context at the time of crash as
 212   // it may not be available later on.
 213   void* stack[32];
 214   int depth;
 215 };
 216
 217 void SetCrashReason(const CrashReason* r);
 218
 219 void InitGoogleLoggingUtilities(const char* argv0);
 220 void ShutdownGoogleLoggingUtilities();
 221
 222 }  // namespace glog_internal_namespace_
 223
 224 _END_GOOGLE_NAMESPACE_
 225
 226 using namespace GOOGLE_NAMESPACE::glog_internal_namespace_;
 227
 228 #endif  // UTILITIES_H__