lib/sanitizer_common/sanitizer_stacktrace.cc

   1 //===-- sanitizer_stacktrace.cc -------------------------------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file is shared between AddressSanitizer and ThreadSanitizer
  11 // run-time libraries.
  12 //===----------------------------------------------------------------------===//
  13
  14 #include "sanitizer_common.h"
  15 #include "sanitizer_procmaps.h"
  16 #include "sanitizer_stacktrace.h"
  17 #include "sanitizer_symbolizer.h"
  18
  19 namespace __sanitizer {
  20 static const char *StripPathPrefix(const char *filepath,
  21                                    const char *strip_file_prefix) {
  22   if (filepath == internal_strstr(filepath, strip_file_prefix))
  23     return filepath + internal_strlen(strip_file_prefix);
  24   return filepath;
  25 }
  26
  27 // ----------------------- StackTrace ----------------------------- {{{1
  28 uptr StackTrace::GetPreviousInstructionPc(uptr pc) {
  29 #ifdef __arm__
  30   // Cancel Thumb bit.
  31   pc = pc & (~1);
  32 #endif
  33 #if defined(__powerpc__) || defined(__powerpc64__)
  34   // PCs are always 4 byte aligned.
  35   return pc - 4;
  36 #elif defined(__sparc__)
  37   return pc - 8;
  38 #else
  39   return pc - 1;
  40 #endif
  41 }
  42
  43 static void PrintStackFramePrefix(uptr frame_num, uptr pc) {
  44   Printf("    #%zu 0x%zx", frame_num, pc);
  45 }
  46
  47 static void PrintSourceLocation(const char *file, int line, int column,
  48                                 const char *strip_file_prefix) {
  49   CHECK(file);
  50   Printf(" %s", StripPathPrefix(file, strip_file_prefix));
  51   if (line > 0) {
  52     Printf(":%d", line);
  53     if (column > 0)
  54       Printf(":%d", column);
  55   }
  56 }
  57
  58 static void PrintModuleAndOffset(const char *module, uptr offset,
  59                                  const char *strip_file_prefix) {
  60   Printf(" (%s+0x%zx)", StripPathPrefix(module, strip_file_prefix), offset);
  61 }
  62
  63 void StackTrace::PrintStack(const uptr *addr, uptr size,
  64                             bool symbolize, const char *strip_file_prefix,
  65                             SymbolizeCallback symbolize_callback ) {
  66   MemoryMappingLayout proc_maps;
  67   InternalScopedBuffer<char> buff(GetPageSizeCached() * 2);
  68   InternalScopedBuffer<AddressInfo> addr_frames(64);
  69   uptr frame_num = 0;
  70   for (uptr i = 0; i < size && addr[i]; i++) {
  71     // PCs in stack traces are actually the return addresses, that is,
  72     // addresses of the next instructions after the call.
  73     uptr pc = GetPreviousInstructionPc(addr[i]);
  74     uptr addr_frames_num = 0;  // The number of stack frames for current
  75                                // instruction address.
  76     if (symbolize_callback) {
  77       if (symbolize_callback((void*)pc, buff.data(), buff.size())) {
  78         addr_frames_num = 1;
  79         PrintStackFramePrefix(frame_num, pc);
  80         // We can't know anything about the string returned by external
  81         // symbolizer, but if it starts with filename, try to strip path prefix
  82         // from it.
  83         Printf(" %s\n", StripPathPrefix(buff.data(), strip_file_prefix));
  84         frame_num++;
  85       }
  86     }
  87     if (symbolize && addr_frames_num == 0) {
  88       // Use our own (online) symbolizer, if necessary.
  89       addr_frames_num = SymbolizeCode(pc, addr_frames.data(),
  90                                       addr_frames.size());
  91       for (uptr j = 0; j < addr_frames_num; j++) {
  92         AddressInfo &info = addr_frames[j];
  93         PrintStackFramePrefix(frame_num, pc);
  94         if (info.function) {
  95           Printf(" in %s", info.function);
  96         }
  97         if (info.file) {
  98           PrintSourceLocation(info.file, info.line, info.column,
  99                               strip_file_prefix);
 100         } else if (info.module) {
 101           PrintModuleAndOffset(info.module, info.module_offset,
 102                                strip_file_prefix);
 103         }
 104         Printf("\n");
 105         info.Clear();
 106         frame_num++;
 107       }
 108     }
 109     if (addr_frames_num == 0) {
 110       // If online symbolization failed, try to output at least module and
 111       // offset for instruction.
 112       PrintStackFramePrefix(frame_num, pc);
 113       uptr offset;
 114       if (proc_maps.GetObjectNameAndOffset(pc, &offset,
 115                                            buff.data(), buff.size())) {
 116         PrintModuleAndOffset(buff.data(), offset, strip_file_prefix);
 117       }
 118       Printf("\n");
 119       frame_num++;
 120     }
 121   }
 122 }
 123
 124 uptr StackTrace::GetCurrentPc() {
 125   return GET_CALLER_PC();
 126 }
 127
 128 void StackTrace::FastUnwindStack(uptr pc, uptr bp,
 129                                  uptr stack_top, uptr stack_bottom) {
 130   CHECK(size == 0 && trace[0] == pc);
 131   size = 1;
 132   uhwptr *frame = (uhwptr *)bp;
 133   uhwptr *prev_frame = frame;
 134   while (frame >= prev_frame &&
 135          frame < (uhwptr *)stack_top - 2 &&
 136          frame > (uhwptr *)stack_bottom &&
 137          size < max_size) {
 138     uhwptr pc1 = frame[1];
 139     if (pc1 != pc) {
 140       trace[size++] = (uptr) pc1;
 141     }
 142     prev_frame = frame;
 143     frame = (uhwptr *)frame[0];
 144   }
 145 }
 146
 147 void StackTrace::PopStackFrames(uptr count) {
 148   CHECK(size >= count);
 149   size -= count;
 150   for (uptr i = 0; i < size; i++) {
 151     trace[i] = trace[i + count];
 152   }
 153 }
 154
 155 // On 32-bits we don't compress stack traces.
 156 // On 64-bits we compress stack traces: if a given pc differes slightly from
 157 // the previous one, we record a 31-bit offset instead of the full pc.
 158 SANITIZER_INTERFACE_ATTRIBUTE
 159 uptr StackTrace::CompressStack(StackTrace *stack, u32 *compressed, uptr size) {
 160 #if SANITIZER_WORDSIZE == 32
 161   // Don't compress, just copy.
 162   uptr res = 0;
 163   for (uptr i = 0; i < stack->size && i < size; i++) {
 164     compressed[i] = stack->trace[i];
 165     res++;
 166   }
 167   if (stack->size < size)
 168     compressed[stack->size] = 0;
 169 #else  // 64 bits, compress.
 170   uptr prev_pc = 0;
 171   const uptr kMaxOffset = (1ULL << 30) - 1;
 172   uptr c_index = 0;
 173   uptr res = 0;
 174   for (uptr i = 0, n = stack->size; i < n; i++) {
 175     uptr pc = stack->trace[i];
 176     if (!pc) break;
 177     if ((s64)pc < 0) break;
 178     // Printf("C pc[%zu] %zx\n", i, pc);
 179     if (prev_pc - pc < kMaxOffset || pc - prev_pc < kMaxOffset) {
 180       uptr offset = (s64)(pc - prev_pc);
 181       offset |= (1U << 31);
 182       if (c_index >= size) break;
 183       // Printf("C co[%zu] offset %zx\n", i, offset);
 184       compressed[c_index++] = offset;
 185     } else {
 186       uptr hi = pc >> 32;
 187       uptr lo = (pc << 32) >> 32;
 188       CHECK_EQ((hi & (1 << 31)), 0);
 189       if (c_index + 1 >= size) break;
 190       // Printf("C co[%zu] hi/lo: %zx %zx\n", c_index, hi, lo);
 191       compressed[c_index++] = hi;
 192       compressed[c_index++] = lo;
 193     }
 194     res++;
 195     prev_pc = pc;
 196   }
 197   if (c_index < size)
 198     compressed[c_index] = 0;
 199   if (c_index + 1 < size)
 200     compressed[c_index + 1] = 0;
 201 #endif  // SANITIZER_WORDSIZE
 202
 203   // debug-only code
 204 #if 0
 205   StackTrace check_stack;
 206   UncompressStack(&check_stack, compressed, size);
 207   if (res < check_stack.size) {
 208     Printf("res %zu check_stack.size %zu; c_size %zu\n", res,
 209            check_stack.size, size);
 210   }
 211   // |res| may be greater than check_stack.size, because
 212   // UncompressStack(CompressStack(stack)) eliminates the 0x0 frames.
 213   CHECK(res >= check_stack.size);
 214   CHECK_EQ(0, REAL(memcmp)(check_stack.trace, stack->trace,
 215                           check_stack.size * sizeof(uptr)));
 216 #endif
 217
 218   return res;
 219 }
 220
 221 SANITIZER_INTERFACE_ATTRIBUTE
 222 void StackTrace::UncompressStack(StackTrace *stack,
 223                                  u32 *compressed, uptr size) {
 224 #if SANITIZER_WORDSIZE == 32
 225   // Don't uncompress, just copy.
 226   stack->size = 0;
 227   for (uptr i = 0; i < size && i < kStackTraceMax; i++) {
 228     if (!compressed[i]) break;
 229     stack->size++;
 230     stack->trace[i] = compressed[i];
 231   }
 232 #else  // 64 bits, uncompress
 233   uptr prev_pc = 0;
 234   stack->size = 0;
 235   for (uptr i = 0; i < size && stack->size < kStackTraceMax; i++) {
 236     u32 x = compressed[i];
 237     uptr pc = 0;
 238     if (x & (1U << 31)) {
 239       // Printf("U co[%zu] offset: %x\n", i, x);
 240       // this is an offset
 241       s32 offset = x;
 242       offset = (offset << 1) >> 1;  // remove the 31-byte and sign-extend.
 243       pc = prev_pc + offset;
 244       CHECK(pc);
 245     } else {
 246       // CHECK(i + 1 < size);
 247       if (i + 1 >= size) break;
 248       uptr hi = x;
 249       uptr lo = compressed[i+1];
 250       // Printf("U co[%zu] hi/lo: %zx %zx\n", i, hi, lo);
 251       i++;
 252       pc = (hi << 32) | lo;
 253       if (!pc) break;
 254     }
 255     // Printf("U pc[%zu] %zx\n", stack->size, pc);
 256     stack->trace[stack->size++] = pc;
 257     prev_pc = pc;
 258   }
 259 #endif  // SANITIZER_WORDSIZE
 260 }
 261
 262 }  // namespace __sanitizer