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

base/trace_event/malloc_dump_provider.cc

 // Copyright 2015 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/trace_event/malloc_dump_provider.h"
 
 #include <stddef.h>
 
 #include "base/allocator/allocator_extension.h"
 #include "base/allocator/allocator_shim.h"
@@ skipping 59 matching lines @@
   return ptr;
 }
 
 void HookFree(const AllocatorDispatch* self, void* address) {
   if (address)
     MallocDumpProvider::GetInstance()->RemoveAllocation(address);
   const AllocatorDispatch* const next = self->next;
   next->free_function(next, address);
 }
 
+size_t HookGetSizeEstimate(const AllocatorDispatch* self, void* address) {
+  const AllocatorDispatch* const next = self->next;
+  return next->get_size_estimate_function(next, address);
+}
+
 AllocatorDispatch g_allocator_hooks = {
-    &HookAlloc,         /* alloc_function */
-    &HookZeroInitAlloc, /* alloc_zero_initialized_function */
-    &HookllocAligned,   /* alloc_aligned_function */
-    &HookRealloc,       /* realloc_function */
-    &HookFree,          /* free_function */
-    nullptr,            /* next */
+    &HookAlloc,           /* alloc_function */
+    &HookZeroInitAlloc,   /* alloc_zero_initialized_function */
+    &HookllocAligned,     /* alloc_aligned_function */
+    &HookRealloc,         /* realloc_function */
+    &HookFree,            /* free_function */
+    &HookGetSizeEstimate, /* get_size_estimate_function */
+    nullptr,              /* next */
 };
 #endif  // BUILDFLAG(USE_EXPERIMENTAL_ALLOCATOR_SHIM)
 
 #if defined(OS_WIN)
 // A structure containing some information about a given heap.
 struct WinHeapInfo {
   HANDLE heap_id;
   size_t committed_size;
   size_t uncommitted_size;
   size_t allocated_size;
@@ skipping 254 matching lines @@
       tid_dumping_heap_ == PlatformThread::CurrentId())
     return;
   AutoLock lock(allocation_register_lock_);
   if (!allocation_register_)
     return;
   allocation_register_->Remove(address);
 }
 
 }  // namespace trace_event
 }  // namespace base