Type profiler by intercepting 'new' and 'delete' expressions.

content/app/content_main_runner.cc

 // Copyright (c) 2012 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 "content/public/app/content_main_runner.h"
 
 #include <stdlib.h>
 
 #include "base/allocator/allocator_extension.h"
 #include "base/at_exit.h"
@@ skipping 31 matching lines @@
 #include "ipc/ipc_switches.h"
 #include "media/base/media.h"
 #include "sandbox/win/src/sandbox_types.h"
 #include "ui/base/ui_base_switches.h"
 #include "ui/base/ui_base_paths.h"
 #include "ui/base/win/dpi.h"
 #include "webkit/user_agent/user_agent.h"
 
 #if defined(USE_TCMALLOC)
 #include "third_party/tcmalloc/chromium/src/gperftools/malloc_extension.h"
+#if defined(TYPE_PROFILING)
+#include "base/allocator/type_profiler.h"
+#include "base/allocator/type_profiler_tcmalloc.h"
+#endif
 #endif
 
 #if defined(OS_WIN)
 #include <cstring>
 #include <atlbase.h>
 #include <atlapp.h>
 #include <malloc.h>
 #elif defined(OS_MACOSX)
 #include "base/mac/scoped_nsautorelease_pool.h"
 #include "base/mach_ipc_mac.h"
@@ skipping 424 matching lines @@
 
     RegisterInvalidParamHandler();
     _Module.Init(NULL, static_cast<HINSTANCE>(instance));
 
     sandbox_info_ = *sandbox_info;
 #else  // !OS_WIN
   virtual int Initialize(int argc,
                          const char** argv,
                          ContentMainDelegate* delegate) OVERRIDE {
 
-    // NOTE(willchan): One might ask why this call is done here rather than in
-    // process_util_linux.cc with the definition of
+    // NOTE(willchan): One might ask why these TCMalloc-related calls are done
+    // here rather than in process_util_linux.cc with the definition of
     // EnableTerminationOnOutOfMemory().  That's because base shouldn't have a
     // dependency on TCMalloc.  Really, we ought to have our allocator shim code
     // implement this EnableTerminationOnOutOfMemory() function.  Whateverz.
     // This works for now.
 #if !defined(OS_MACOSX) && defined(USE_TCMALLOC)
+
+#if defined(TYPE_PROFILING)
+    base::type_profiler::InterceptFunctions::SetFunctions(
+        base::type_profiler::NewInterceptForTCMalloc,
+        base::type_profiler::DeleteInterceptForTCMalloc);
+#endif
+
     // For tcmalloc, we need to tell it to behave like new.
     tc_set_new_mode(1);
 
     // On windows, we've already set these thunks up in _heap_init()
     base::allocator::SetGetStatsFunction(GetStatsThunk);
     base::allocator::SetReleaseFreeMemoryFunction(ReleaseFreeMemoryThunk);
 
     // Provide optional hook for monitoring allocation quantities on a
     // per-thread basis.  Only set the hook if the environment indicates this
     // needs to be enabled.
@@ skipping 252 matching lines @@
 
   DISALLOW_COPY_AND_ASSIGN(ContentMainRunnerImpl);
 };
 
 // static
 ContentMainRunner* ContentMainRunner::Create() {
   return new ContentMainRunnerImpl();
 }
 
 }  // namespace content