DO NOT COMMIT: Heap profiler for Skia (sk_malloc)

skia/ext/skia_memory_dump_provider.h

 // 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.
 
 #ifndef SKIA_EXT_SKIA_MEMORY_DUMP_PROVIDER_H_
 #define SKIA_EXT_SKIA_MEMORY_DUMP_PROVIDER_H_
 
 #include "base/macros.h"
 #include "base/memory/singleton.h"
+#include "base/trace_event/heap_profiler_allocation_register.h"
 #include "base/trace_event/memory_dump_provider.h"
 #include "third_party/skia/include/core/SkTypes.h"
 
 namespace skia {
 
+class SkiaAllocHooks {
+ public:
+  typedef void AllocationHook(void* address, size_t size,
+                              const char* type_name);
+  typedef void FreeHook(void* address);
+  static void SetAllocationHook(AllocationHook* hook) {
+    allocation_hook_ = hook;
+  }
+  static void SetFreeHook(FreeHook* hook) {
+    free_hook_ = hook;
+  }
+  static void AllocationHookIfEnabled(void* address, size_t size,
+                                      const char* type_name) {
+    AllocationHook* allocation_hook = allocation_hook_;
+    if (UNLIKELY(allocation_hook != nullptr)) {
+      allocation_hook(address, size, type_name);
+    }
+  }
+  static void FreeHookIfEnabled(void* address) {
+    FreeHook* free_hook = free_hook_;
+    if (UNLIKELY(free_hook != nullptr)) {
+      free_hook(address);
+    }
+  }
+
+ private:
+  static AllocationHook* allocation_hook_;
+  static FreeHook* free_hook_;
+};
+
 class SK_API SkiaMemoryDumpProvider
     : public base::trace_event::MemoryDumpProvider {
  public:
   static SkiaMemoryDumpProvider* GetInstance();
 
   // base::trace_event::MemoryDumpProvider implementation:
   bool OnMemoryDump(
       const base::trace_event::MemoryDumpArgs& args,
       base::trace_event::ProcessMemoryDump* process_memory_dump) override;
+  void OnHeapProfilingEnabled(bool enabled) override;
+
+  void Insert(void* address, size_t size, const char* type_name);
+  void Remove(void* address);
 
  private:
   friend struct base::DefaultSingletonTraits<SkiaMemoryDumpProvider>;
 
   SkiaMemoryDumpProvider();
   ~SkiaMemoryDumpProvider() override;
 
+  base::Lock lock_;
+  scoped_ptr<base::trace_event::AllocationRegister> allocation_register_;
+  bool is_heap_profiling_enabled_;
+
   DISALLOW_COPY_AND_ASSIGN(SkiaMemoryDumpProvider);
 };
 
 }  // namespace skia
 
 #endif  // SKIA_EXT_SKIA_MEMORY_DUMP_PROVIDER_H_