Dump type statistics of malloc'ed objects.

third_party/tcmalloc/chromium/src/heap-profiler.cc

 // Copyright (c) 2005, Google Inc.
 // All rights reserved.
 // 
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are
 // met:
 // 
 //     * Redistributions of source code must retain the above copyright
 // notice, this list of conditions and the following disclaimer.
 //     * Redistributions in binary form must reproduce the above
@@ skipping 107 matching lines @@
 DEFINE_bool(mmap_profile,
             EnvToBool("HEAP_PROFILE_MMAP", false),
             "If heap-profiling is on, also profile mmap, mremap, and sbrk)");
 DEFINE_bool(only_mmap_profile,
             EnvToBool("HEAP_PROFILE_ONLY_MMAP", false),
             "If heap-profiling is on, only profile mmap, mremap, and sbrk; "
             "do not profile malloc/new/etc");
 DEFINE_bool(deep_heap_profile,
             EnvToBool("DEEP_HEAP_PROFILE", false),
             "If heap-profiling is on, profile deeper (only on Linux)");
+#if defined(TYPE_PROFILING)
+DEFINE_bool(heap_profile_type_statistics,
+            EnvToBool("HEAP_PROFILE_TYPE_STATISTICS", false),
+            "If heap-profiling is on, dump type statistics.");
+#endif  // defined(TYPE_PROFILING)
 
 
 //----------------------------------------------------------------------
 // Locking
 //----------------------------------------------------------------------
 
 // A pthread_mutex has way too much lock contention to be used here.
 //
 // I would like to use Mutex, but it can call malloc(),
 // which can cause us to fall into an infinite recursion.
@@ skipping 160 matching lines @@
   if (global_profiler_buffer == NULL) {
     global_profiler_buffer =
         reinterpret_cast<char*>(ProfilerMalloc(kProfileBufferSize));
   }
 
   char* profile = DoGetHeapProfileLocked(global_profiler_buffer,
                                          kProfileBufferSize);
   RawWrite(fd, profile, strlen(profile));
   RawClose(fd);
 
+#if defined(TYPE_PROFILING)
+  if (FLAGS_heap_profile_type_statistics) {
+    snprintf(file_name, sizeof(file_name), "%s.%05d.%04d.type",
+             filename_prefix, getpid(), dump_count);
+    RAW_VLOG(0, "Dumping type statistics to %s", file_name);
+    heap_profile->DumpTypeStatistics(file_name);
+  }
+#endif  // defined(TYPE_PROFILING)
+
   dumping = false;
 }
 
 //----------------------------------------------------------------------
 // Profile collection
 //----------------------------------------------------------------------
 
 // Dump a profile after either an allocation or deallocation, if
 // the memory use has changed enough since the last dump.
 static void MaybeDumpProfileLocked() {
@@ skipping 334 matching lines @@
 
 // class used for finalization -- dumps the heap-profile at program exit
 struct HeapProfileEndWriter {
   ~HeapProfileEndWriter() { HeapProfilerDump("Exiting"); }
 };
 
 // We want to make sure tcmalloc is up and running before starting the profiler
 static const TCMallocGuard tcmalloc_initializer;
 REGISTER_MODULE_INITIALIZER(heapprofiler, HeapProfilerInit());
 static HeapProfileEndWriter heap_profile_end_writer;