contrib/compiler-rt/lib/asan/asan_memory_profile.cc

   1 //===-- asan_memory_profile.cc.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 a part of AddressSanitizer, an address sanity checker.
  11 //
  12 // This file implements __sanitizer_print_memory_profile.
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "sanitizer_common/sanitizer_common.h"
  16 #include "sanitizer_common/sanitizer_stackdepot.h"
  17 #include "sanitizer_common/sanitizer_stacktrace.h"
  18 #include "sanitizer_common/sanitizer_stoptheworld.h"
  19 #include "lsan/lsan_common.h"
  20 #include "asan/asan_allocator.h"
  21
  22 #if CAN_SANITIZE_LEAKS
  23
  24 namespace __asan {
  25
  26 struct AllocationSite {
  27   u32 id;
  28   uptr total_size;
  29   uptr count;
  30 };
  31
  32 class HeapProfile {
  33  public:
  34   HeapProfile() : allocations_(1024) {}
  35   void Insert(u32 id, uptr size) {
  36     total_allocated_ += size;
  37     total_count_++;
  38     // Linear lookup will be good enough for most cases (although not all).
  39     for (uptr i = 0; i < allocations_.size(); i++) {
  40       if (allocations_[i].id == id) {
  41         allocations_[i].total_size += size;
  42         allocations_[i].count++;
  43         return;
  44       }
  45     }
  46     allocations_.push_back({id, size, 1});
  47   }
  48
  49   void Print(uptr top_percent) {
  50     InternalSort(&allocations_, allocations_.size(),
  51                  [](const AllocationSite &a, const AllocationSite &b) {
  52                    return a.total_size > b.total_size;
  53                  });
  54     CHECK(total_allocated_);
  55     uptr total_shown = 0;
  56     Printf("Live Heap Allocations: %zd bytes from %zd allocations; "
  57            "showing top %zd%%\n", total_allocated_, total_count_, top_percent);
  58     for (uptr i = 0; i < allocations_.size(); i++) {
  59       auto &a = allocations_[i];
  60       Printf("%zd byte(s) (%zd%%) in %zd allocation(s)\n", a.total_size,
  61              a.total_size * 100 / total_allocated_, a.count);
  62       StackDepotGet(a.id).Print();
  63       total_shown += a.total_size;
  64       if (total_shown * 100 / total_allocated_ > top_percent)
  65         break;
  66     }
  67   }
  68
  69  private:
  70   uptr total_allocated_ = 0;
  71   uptr total_count_ = 0;
  72   InternalMmapVector<AllocationSite> allocations_;
  73 };
  74
  75 static void ChunkCallback(uptr chunk, void *arg) {
  76   HeapProfile *hp = reinterpret_cast<HeapProfile*>(arg);
  77   AsanChunkView cv = FindHeapChunkByAddress(chunk);
  78   if (!cv.IsAllocated()) return;
  79   u32 id = cv.GetAllocStackId();
  80   if (!id) return;
  81   hp->Insert(id, cv.UsedSize());
  82 }
  83
  84 static void MemoryProfileCB(const SuspendedThreadsList &suspended_threads_list,
  85                             void *argument) {
  86   HeapProfile hp;
  87   __lsan::ForEachChunk(ChunkCallback, &hp);
  88   hp.Print(reinterpret_cast<uptr>(argument));
  89 }
  90
  91 }  // namespace __asan
  92
  93 extern "C" {
  94 SANITIZER_INTERFACE_ATTRIBUTE
  95 void __sanitizer_print_memory_profile(uptr top_percent) {
  96   __sanitizer::StopTheWorld(__asan::MemoryProfileCB, (void*)top_percent);
  97 }
  98 }  // extern "C"
  99
 100 #endif  // CAN_SANITIZE_LEAKS