Implement a ScopedThreadHeapUsage class to allow profiling per-thread heap usage.

base/allocator/allocator_shim_default_dispatch_to_tcmalloc.cc

 // Copyright 2016 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 "base/allocator/allocator_shim.h"
 #include "base/allocator/allocator_shim_internals.h"
 #include "third_party/tcmalloc/chromium/src/config.h"
 #include "third_party/tcmalloc/chromium/src/gperftools/tcmalloc.h"
 
 namespace {
@@ skipping 13 matching lines @@
 }
 
 void* TCRealloc(const AllocatorDispatch*, void* address, size_t size) {
   return tc_realloc(address, size);
 }
 
 void TCFree(const AllocatorDispatch*, void* address) {
   tc_free(address);
 }
 
+size_t TCGetSizeEstimate(const AllocatorDispatch*, void* address) {
+  return tc_malloc_size(address);
+}
+
 }  // namespace
 
 const AllocatorDispatch AllocatorDispatch::default_dispatch = {
-    &TCMalloc,   /* alloc_function */
-    &TCCalloc,   /* alloc_zero_initialized_function */
-    &TCMemalign, /* alloc_aligned_function */
-    &TCRealloc,  /* realloc_function */
-    &TCFree,     /* free_function */
-    nullptr,     /* next */
+    &TCMalloc,          /* alloc_function */
+    &TCCalloc,          /* alloc_zero_initialized_function */
+    &TCMemalign,        /* alloc_aligned_function */
+    &TCRealloc,         /* realloc_function */
+    &TCFree,            /* free_function */
+    &TCGetSizeEstimate, /* get_size_estimate_function */
+    nullptr,            /* next */
 };
 
 // In the case of tcmalloc we have also to route the diagnostic symbols,
 // which are not part of the unified shim layer, to tcmalloc for consistency.
 
 extern "C" {
 
 SHIM_ALWAYS_EXPORT void malloc_stats(void) __THROW {
   return tc_malloc_stats();
 }
@@ skipping 14 matching lines @@
 
 #if defined(__ANDROID__)
 SHIM_ALWAYS_EXPORT size_t malloc_usable_size(const void* address) __THROW {
 #else
 SHIM_ALWAYS_EXPORT size_t malloc_usable_size(void* address) __THROW {
 #endif
   return tc_malloc_size(address);
 }
 
 }  // extern "C"