A deeper heap profile dumper in third_party/tcmalloc/chromium.

third_party/tcmalloc/chromium/src/deep-heap-profile.cc

Index: third_party/tcmalloc/chromium/src/deep-heap-profile.cc
diff --git a/third_party/tcmalloc/chromium/src/deep-heap-profile.cc b/third_party/tcmalloc/chromium/src/deep-heap-profile.cc
new file mode 100644
index 0000000000000000000000000000000000000000..25a933b7ad92322f33449b4f0bb7ca5d8692db2b
--- /dev/null
+++ b/third_party/tcmalloc/chromium/src/deep-heap-profile.cc
@@ -0,0 +1,471 @@
+// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include <config.h>
+
+#include "deep-heap-profile.h"
+
+#ifdef DEEP_HEAP_PROFILE
+#include <fcntl.h>
+#include <sys/types.h>
+#include <sys/stat.h>

[Alexander Potapenko, 2011/12/20 13:57:23]

Please mind the #include order, see
http://google-styleguide.googlecode.com/svn/trunk/cppguide.xml#Names_and_Orde...

[Dai Mikurube (NOT FULLTIME), 2011/12/21 09:13:49]

Done.

+#ifdef HAVE_UNISTD_H

[Alexander Potapenko, 2011/12/20 13:57:23]

What if you don't have unistd.h?

[Dai Mikurube (NOT FULLTIME), 2011/12/21 09:13:49]

It's required for getpid().  I though about the case of no unistd.h, but other
code calls getpid() with no consideration.

I guess they assume that getpid() is defined in another way.  For example,
getpid() is defined in windows/port.h which is included in config.h.

+#include <unistd.h>
+#endif
+
+#include "base/sysinfo.h"
+#include "base/cycleclock.h"
+
+static const int kProfilerBufferSize = 1 << 20;
+static const int kHashTableSize = 179999;  // The same as heap-profile-table.cc
+
+static const int PAGE_SIZE = 4096;
+static const int PAGEMAP_BYTES = 8;
+
+// header of the dumped heap profile
+static const char kProfileHeader[] = "Deep Memory Profile\n";
+static const char kGlobalStatsHeader[] = "GLOBAL_STATS:\n";
+static const char kStacktraceHeader[] = "STACKTRACES:\n";
+static const char kProcSelfMapsHeader[] = "\nMAPPED_LIBRARIES:\n";
+
+DeepHeapProfile::DeepHeapProfile(HeapProfileTable* heap_profile,
+                                 const char* prefix)
+    : heap_profile_(heap_profile),
+      pagemap_fd_(-1),
+      most_recent_pid_(-1),
+      stats_(),
+      page_map_(NULL),
+      dump_count_(0),
+      filename_prefix_(NULL),
+      profiler_buffer_(NULL),
+      bucket_id_(0) {
+  page_map_ = new(heap_profile_->alloc_(sizeof(PageStateMap)))
+      PageStateMap(heap_profile_->alloc_, heap_profile_->dealloc_);
+
+  deep_bucket_map_ = new(heap_profile_->alloc_(sizeof(DeepBucketMap)))
+      DeepBucketMap(heap_profile_->alloc_, heap_profile_->dealloc_);
+
+  // Copy filename prefix
+  RAW_DCHECK(filename_prefix_ == NULL, "");
+  const int prefix_length = strlen(prefix);
+  filename_prefix_ =
+      reinterpret_cast<char*>(heap_profile_->alloc_(prefix_length + 1));
+  memcpy(filename_prefix_, prefix, prefix_length);
+  filename_prefix_[prefix_length] = '\0';
+
+  profiler_buffer_ =
+      reinterpret_cast<char*>(heap_profile_->alloc_(kProfilerBufferSize));
+}
+
+DeepHeapProfile::~DeepHeapProfile() {
+  heap_profile_->dealloc_(profiler_buffer_);
+  heap_profile_->dealloc_(filename_prefix_);
+  deep_bucket_map_->~DeepBucketMap();
+  heap_profile_->dealloc_(deep_bucket_map_);
+  page_map_->~PageStateMap();
+  heap_profile_->dealloc_(page_map_);
+}
+
+int DeepHeapProfile::FillOrderedProfile(char buf[], int size) {
+  int64 start_time = CycleClock::Now();
+  ++dump_count_;
+
+  // Re-open files in /proc/pid/ if the process is newly forked one.
+  if (most_recent_pid_ != getpid()) {
+    most_recent_pid_ = getpid();
+    OpenProcPagemap();
+
+    // Write maps into a .maps file with using the global buffer.
+    WriteMapsToFile(profiler_buffer_, kProfilerBufferSize);
+  }
+
+  // Reset committed sizes of buckets.
+  ResetCommittedSize(heap_profile_->malloc_table_);
+  ResetCommittedSize(heap_profile_->mmap_table_);
+
+  GetGlobalStats();
+  uint64 anonymous_committed = stats_.anonymous.committed_bytes;
+
+  // Note: Don't allocate any memory from here.
+
+  // Record committed sizes.
+  RecordAllAllocs();
+
+  // Check if committed bytes changed during RecordAllAllocs.
+  GetGlobalStats();
+  uint64 committed_difference =
+      stats_.anonymous.committed_bytes - anonymous_committed;
+  if (committed_difference != 0)
+    RAW_LOG(0, "Difference in committed size: %"PRId64"", committed_difference);
+
+  HeapProfileTable::Stats stats;
+  memset(&stats, 0, sizeof(stats));
+
+  // Start filling buf with the ordered profile.
+  int bucket_length = snprintf(buf, size, kProfileHeader);
+  if (bucket_length < 0 || bucket_length >= size) return 0;
+
+  // Fill buf with the global stats.
+  bucket_length +=
+      snprintf(buf + bucket_length, size - bucket_length, kGlobalStatsHeader);
+  bucket_length = UnparseGlobalStats(buf, bucket_length, size);
+
+  // Fill buf with the header for buckets.
+  bucket_length +=
+      snprintf(buf + bucket_length, size - bucket_length, kStacktraceHeader);
+  bucket_length += snprintf(buf + bucket_length, size - bucket_length,
+                            "%10s %10s\n", "virtual", "committed");
+
+  // Fill buf with stack trace buckets.
+  bucket_length = FillBucketTable(heap_profile_->malloc_table_,
+                                  buf, size, bucket_length, &stats);
+  bucket_length = FillBucketTable(heap_profile_->mmap_table_,
+                                  buf, size, bucket_length, &stats);
+
+  RAW_DCHECK(bucket_length < size, "");
+
+  // Note: Don't allocate any memory until here.
+
+  // Write the bucket listing into a .bucket file.
+  WriteBucketsToBucketFile();
+
+  int64 dt = CycleClock::Now() - start_time;
+  double dtf = dt / CyclesPerSecond();
+  RAW_LOG(0, "Time spent on DeepProfiler: %.3f sec\n", dtf);
+
+  return bucket_length;
+}
+
+DeepHeapProfile::DeepBucket*
+DeepHeapProfile::GetDeepBucket(Bucket* bucket) {
+  DeepBucket* found = deep_bucket_map_->FindMutable(bucket);
+  if (found == NULL) {
+    DeepBucket created;
+    created.bucket = bucket;
+    created.committed_size = 0;
+    created.id = (bucket_id_++);
+    created.is_logged = false;
+    deep_bucket_map_->Insert(bucket, created);
+    return deep_bucket_map_->FindMutable(bucket);
+  } else {
+    return found;
+  }
+}
+
+void DeepHeapProfile::ResetCommittedSize(Bucket** table) {
+  for (int i = 0; i < kHashTableSize; i++) {
+    for (Bucket* b = table[i]; b != 0; b = b->next) {
+      DeepBucket* db = GetDeepBucket(b);
+      db->committed_size = 0;
+    }
+  }
+}
+
+int DeepHeapProfile::FillBucketTable(Bucket** table,
+                                     char buf[], int size, int bucket_length,
+                                     HeapProfileTable::Stats* stats) {
+  for (int i = 0; i < kHashTableSize; i++) {
+    for (Bucket* b = table[i]; b != 0; b = b->next) {
+      if (b->alloc_size - b->free_size == 0)
+        continue;  // Skip empty buckets
+      const DeepBucket& db = *GetDeepBucket(b);
+      bucket_length = UnparseBucket(db, buf, bucket_length, size, "", stats);
+    }
+  }
+  return bucket_length;
+}
+
+// This function need to be called after each fork
+void DeepHeapProfile::OpenProcPagemap() {
+  char filename[100];
+  sprintf(filename, "/proc/%d/pagemap", getpid());
+  pagemap_fd_ = open(filename, O_RDONLY);
+  RAW_DCHECK(pagemap_fd_ != -1, "Failed to open /proc/self/pagemap");
+}
+
+bool DeepHeapProfile::SeekProcPagemap(uint64 address) {
+  uint64 index = (address / PAGE_SIZE) * PAGEMAP_BYTES;
+  uint64 o = lseek64(pagemap_fd_, index, SEEK_SET);
+  RAW_DCHECK(o == index, "");
+  return true;
+}
+
+bool DeepHeapProfile::ReadProcPagemap(PageState* state) {
+  static const uint64 U64_1 = 1;
+  static const uint64 PFN_FILTER = (U64_1 << 55) - U64_1;
+  static const uint64 PAGE_PRESENT = U64_1 << 63;
+  static const uint64 PAGE_SWAP = U64_1 << 62;
+  static const uint64 PAGE_RESERVED = U64_1 << 61;
+  static const uint64 FLAG_NOPAGE = U64_1 << 20;
+  static const uint64 FLAG_KSM = U64_1 << 21;
+  static const uint64 FLAG_MMAP = U64_1 << 11;
+
+  uint64 pagemap_value;
+  int result = read(pagemap_fd_, &pagemap_value, PAGEMAP_BYTES);
+  if (result != PAGEMAP_BYTES)
+    return false;
+
+  // Check if the page is committed
+  state->is_committed = (pagemap_value & (PAGE_PRESENT | PAGE_SWAP));
+
+  state->is_present = (pagemap_value & PAGE_PRESENT);
+  state->is_swapped = (pagemap_value & PAGE_SWAP);
+  state->is_shared = false;
+
+  return true;
+}
+
+uint64 DeepHeapProfile::GetCommittedSize(uint64 address, uint64 size) {
+  uint64 page_address = (address / PAGE_SIZE) * PAGE_SIZE;
+  uint64 committed_size = 0;
+
+  SeekProcPagemap(address);
+
+  // Check every pages on which the allocation reside
+  while (page_address < address + size) {
+    // Read corresponding physical page
+    PageState state;
+    if (ReadProcPagemap(&state) == false) {
+      // We can't read the last region (e.g vsyscall)
+      RAW_LOG(0, "pagemap read failed @ %#llx %"PRId64" bytes", address, size);
+      return 0;
+    }
+
+    if (state.is_committed) {
+      // Calculate the size of the allocation part in this page
+      uint64 bytes = PAGE_SIZE;
+      if (page_address < address)
+        bytes -= address - page_address;
+      if (address + size < page_address + PAGE_SIZE)
+        bytes -= PAGE_SIZE - (address + size - page_address);
+
+      committed_size += bytes;
+    }
+    page_address += PAGE_SIZE;
+  }
+
+  return committed_size;
+}
+
+void DeepHeapProfile::InitRegionStats(RegionStats* stats) {
+  stats->virtual_bytes = 0;
+  stats->committed_bytes = 0;
+}
+
+void DeepHeapProfile::RecordRegionStats(uint64 start,
+                                        uint64 end,
+                                        RegionStats* stats) {
+  stats->virtual_bytes += end - start;
+  stats->committed_bytes += GetCommittedSize(start, end - start);
+}
+
+void DeepHeapProfile::GetGlobalStats() {
+  ProcMapsIterator::Buffer iterator_buffer;
+  ProcMapsIterator it(0, &iterator_buffer);
+  uint64 start, end, offset;
+  int64 inode;
+  char *flags, *filename;
+
+  InitRegionStats(&(stats_.total));
+  InitRegionStats(&(stats_.file_mapped));
+  InitRegionStats(&(stats_.anonymous));
+  InitRegionStats(&(stats_.other));
+
+  while (it.Next(&start, &end, &flags, &offset, &inode, &filename)) {
+    if (strcmp("[vsyscall]", filename) == 0)
+      continue;  // pagemap read fails in this region
+
+    int64 committed_bytes = stats_.total.committed_bytes;
+    RecordRegionStats(start, end, &(stats_.total));
+    committed_bytes = stats_.total.committed_bytes - committed_bytes;
+
+    if (filename[0] == '/') {
+      RecordRegionStats(start, end, &(stats_.file_mapped));
+    } else if (filename[0] == '\0' ||
+               filename[0] == '\n' ||
+               filename[0] == EOF) {
+      RecordRegionStats(start, end, &(stats_.anonymous));
+    } else {
+      RecordRegionStats(start, end, &(stats_.other));
+    }
+  }
+}
+
+void DeepHeapProfile::RecordAlloc(const void* pointer,
+                                  AllocValue* v,
+                                  DeepHeapProfile* deep_profile) {
+  uint64 alloc_address = (uint64) pointer;
+  uint64 committed = deep_profile->GetCommittedSize(alloc_address, v->bytes);
+
+  (deep_profile->GetDeepBucket(v->bucket()))->committed_size += committed;
+  deep_profile->stats_.record_tcmalloc.virtual_bytes += v->bytes;
+  deep_profile->stats_.record_tcmalloc.committed_bytes += committed;
+}
+
+void DeepHeapProfile::RecordMMap(const void* pointer,
+                                 AllocValue* v,
+                                 DeepHeapProfile* deep_profile) {
+  uint64 alloc_address = (uint64) pointer;
+  uint64 committed = deep_profile->GetCommittedSize(alloc_address, v->bytes);
+
+  (deep_profile->GetDeepBucket(v->bucket()))->committed_size += committed;
+  deep_profile->stats_.record_mmap.virtual_bytes += v->bytes;
+  deep_profile->stats_.record_mmap.committed_bytes += committed;
+}
+
+void DeepHeapProfile::RecordAllAllocs() {
+  stats_.record_mmap.virtual_bytes = 0;
+  stats_.record_mmap.committed_bytes = 0;
+  stats_.record_tcmalloc.virtual_bytes = 0;
+  stats_.record_tcmalloc.committed_bytes = 0;
+
+  // Tcmalloc allocs
+  heap_profile_->allocation_->Iterate(RecordAlloc, this);
+
+  // Mmap allocs
+  heap_profile_->mmap_allocation_->Iterate(RecordMMap, this);
+}
+
+void DeepHeapProfile::WriteMapsToFile(char buf[], int size) {
+  char file_name[100];
+  snprintf(file_name, sizeof(file_name),
+           "%s.%05d.maps", filename_prefix_, getpid());
+
+  RawFD maps_fd = RawOpenForWriting(file_name);
+  RAW_DCHECK(maps_fd != kIllegalRawFD, "");
+
+  int map_length;
+  bool dummy;   // "wrote_all" -- did /proc/self/maps fit in its entirety?

[Alexander Potapenko, 2011/12/20 13:57:23]

s/dummy/wrote_all. You even won't need the comment.

[Dai Mikurube (NOT FULLTIME), 2011/12/21 09:13:49]

Done.

+  map_length =
+      tcmalloc::FillProcSelfMaps(profiler_buffer_, kProfilerBufferSize, &dummy);

[Alexander Potapenko, 2011/12/20 13:57:23]

Please check for wrote_all.

[Dai Mikurube (NOT FULLTIME), 2011/12/21 09:13:49]

Done.

+  RAW_DCHECK(map_length <= kProfilerBufferSize, "");
+  RawWrite(maps_fd, profiler_buffer_, map_length);
+  RawClose(maps_fd);
+}
+
+int DeepHeapProfile::FillBucketForBucketFile(const DeepBucket* deep_bucket,
+                                             char buf[], int bufsize) {
+  const Bucket* bucket = deep_bucket->bucket;
+  int buflen = 0;
+  buflen += snprintf(buf + buflen, bufsize - buflen, "%05d",
+                     deep_bucket->id);
+  for (int d = 0; d < bucket->depth; d++) {
+    buflen += snprintf(buf + buflen, bufsize - buflen, " 0x%08" PRIxPTR,
+                       reinterpret_cast<uintptr_t>(bucket->stack[d]));
+  }
+  buflen += snprintf(buf + buflen, bufsize - buflen, "\n");
+  return buflen;
+}
+
+void DeepHeapProfile::WriteBucketsTableToBucketFile(Bucket** table,
+                                                    RawFD bucket_fd) {
+  // We will use the global buffer here

[Alexander Potapenko, 2011/12/20 13:57:23]

Please put periods at the end of single-line comments.

[Dai Mikurube (NOT FULLTIME), 2011/12/21 09:13:49]

Done.

+  char* buf = profiler_buffer_;
+  int size = kProfilerBufferSize;
+  int buflen = 0;
+
+  for (int i = 0; i < kHashTableSize; i++) {
+    for (Bucket* b = table[i]; b != 0; b = b->next) {
+      DeepBucket* db = GetDeepBucket(b);
+      if (db->is_logged) continue;  // Skip the bucket if it is already logged
+      if (b->alloc_size - b->free_size <= 64) continue;  // Skip small buckets
+
+      buflen += FillBucketForBucketFile(db, buf + buflen, size - buflen);
+      db->is_logged = true;
+
+      // Write to file if buffer 80% full
+      if (buflen > size * 0.8) {
+        RawWrite(bucket_fd, buf, buflen);
+        buflen = 0;
+      }
+    }
+  }
+
+  RawWrite(bucket_fd, buf, buflen);
+}
+
+void DeepHeapProfile::WriteBucketsToBucketFile() {
+  char file_name[100];
+  snprintf(file_name, sizeof(file_name), "%s.%05d.%04d.buckets",
+           filename_prefix_, getpid(), dump_count_);
+  RawFD bucket_fd = RawOpenForWriting(file_name);
+  RAW_DCHECK(bucket_fd != kIllegalRawFD, "");
+
+  WriteBucketsTableToBucketFile(heap_profile_->malloc_table_, bucket_fd);
+  WriteBucketsTableToBucketFile(heap_profile_->mmap_table_, bucket_fd);
+
+  RawClose(bucket_fd);
+}
+
+int DeepHeapProfile::UnparseBucket(const DeepBucket& deep_bucket,
+                                   char* buf, int buflen, int bufsize,
+                                   const char* extra,
+                                   Stats* profile_stats) {
+  const Bucket& bucket = *deep_bucket.bucket;
+  if (profile_stats != NULL) {
+    profile_stats->allocs += bucket.allocs;
+    profile_stats->alloc_size += bucket.alloc_size;
+    profile_stats->frees += bucket.frees;
+    profile_stats->free_size += bucket.free_size;
+  }
+
+  int printed = snprintf(buf + buflen, bufsize - buflen,
+      "%10"PRId64" %10"PRId64" %6d %6d @%s %d\n",
+       bucket.alloc_size - bucket.free_size, deep_bucket.committed_size,
+       bucket.allocs, bucket.frees, extra, deep_bucket.id);
+  // If it looks like the snprintf failed, ignore the fact we printed anything
+  if (printed < 0 || printed >= bufsize - buflen) return buflen;
+  buflen += printed;
+
+  return buflen;
+}
+
+int DeepHeapProfile::UnparseRegionStats(const RegionStats* stats,
+                                        const char* name,
+                                        char* buf,
+                                        int buflen,
+                                        int bufsize) {
+  int printed = snprintf(buf + buflen, bufsize - buflen,
+                         "%15s %10"PRId64" %10"PRId64"\n",
+                         name,
+                         stats->virtual_bytes,
+                         stats->committed_bytes);
+
+  return buflen + printed;
+}
+
+int DeepHeapProfile::UnparseGlobalStats(char* buf, int buflen, int bufsize) {
+  buflen += snprintf(buf + buflen, bufsize - buflen,
+                     "%15s %10s %10s\n",
+                     "", "virtual", "committed");
+
+  buflen = UnparseRegionStats(
+      &(stats_.total), "total", buf, buflen, bufsize);
+  buflen = UnparseRegionStats(
+      &(stats_.file_mapped), "file mapped", buf, buflen, bufsize);
+  buflen = UnparseRegionStats(
+      &(stats_.anonymous), "anonymous", buf, buflen, bufsize);
+  buflen = UnparseRegionStats(
+      &(stats_.other), "other", buf, buflen, bufsize);
+  buflen = UnparseRegionStats(
+      &(stats_.record_mmap), "mmap", buf, buflen, bufsize);
+  buflen = UnparseRegionStats(
+      &(stats_.record_tcmalloc), "tcmalloc", buf, buflen, bufsize);
+  return buflen;
+}
+#else  // DEEP_HEAP_PROFILE
+
+DeepHeapProfile::DeepHeapProfile(HeapProfileTable* heap_profile,
+                                 const char* prefix)
+    : heap_profile_(heap_profile) {
+}
+
+DeepHeapProfile::~DeepHeapProfile() {
+}
+
+int DeepHeapProfile::FillOrderedProfile(char buf[], int size) {
+  return heap_profile_->FillOrderedProfile(buf, size);
+}
+
+#endif  // DEEP_HEAP_PROFILE