Add microbenchmark covering hooked new/delete.

PiperOrigin-RevId: 770235127
Change-Id: I2c2063336fb80dfa097355295d0cbbbf61f43e0b

Author: ckennelly

tcmalloc/testing/BUILD

@@ -93,9 +93,11 @@ create_tcmalloc_benchmark_suite(
     ],
     deps = [
         "//tcmalloc:malloc_extension",
+        "//tcmalloc:malloc_hook",
         "//tcmalloc/internal:declarations",
         "@com_github_google_benchmark//:benchmark",
         "@com_google_absl//absl/base:core_headers",
+        "@com_google_absl//absl/log:check",
         "@com_google_absl//absl/random",
         "@com_google_absl//absl/synchronization",
         "@com_google_absl//absl/time",

tcmalloc/testing/tcmalloc_benchmark.cc

@@ -26,6 +26,7 @@
 #include <vector>
 
 #include "absl/base/attributes.h"
+#include "absl/log/check.h"
 #include "absl/random/random.h"
 #include "absl/synchronization/notification.h"
 #include "absl/time/clock.h"
@@ -34,6 +35,8 @@
 #include "tcmalloc/common.h"
 #include "tcmalloc/internal/declarations.h"
 #include "tcmalloc/malloc_extension.h"
+#include "tcmalloc/malloc_hook.h"
+#include "tcmalloc/malloc_hook_invoke.h"
 
 extern "C" ABSL_ATTRIBUTE_WEAK void MallocExtension_Internal_GetStats(
     std::string* ret);
@@ -70,13 +73,49 @@ BENCHMARK_TEMPLATE(BM_new_delete_fixed, 4096);
 static void BM_new_sized_delete(benchmark::State& state) {
   const int arg = state.range(0);
 
+  CHECK_EQ(tcmalloc_internal::new_hooks_.size(), 0);
+  CHECK_EQ(tcmalloc_internal::delete_hooks_.size(), 0);
   for (auto s : state) {
     void* ptr = ::operator new(arg);
     ::operator delete(ptr, arg);
   }
 }
 BENCHMARK(BM_new_sized_delete)->Range(1, 1 << 20);
 
+static void BM_hooked_new_sized_delete(benchmark::State& state) {
+  const int arg = state.range(0);
+
+  auto new_hook = [](const MallocHook::NewInfo& info) {
+    benchmark::DoNotOptimize(info.ptr);
+  };
+  CHECK(MallocHook::AddNewHook(new_hook));
+
+  for (auto s : state) {
+    void* ptr = ::operator new(arg);
+    ::operator delete(ptr, arg);
+  }
+
+  CHECK(MallocHook::RemoveNewHook(new_hook));
+}
+BENCHMARK(BM_hooked_new_sized_delete)->Range(1, 1 << 20);
+
+static void BM_new_hooked_sized_delete(benchmark::State& state) {
+  const int arg = state.range(0);
+
+  auto delete_hook = [](const MallocHook::DeleteInfo& info) {
+    benchmark::DoNotOptimize(info.ptr);
+  };
+  CHECK(MallocHook::AddDeleteHook(delete_hook));
+
+  for (auto s : state) {
+    void* ptr = ::operator new(arg);
+    ::operator delete(ptr, arg);
+  }
+
+  CHECK(MallocHook::RemoveDeleteHook(delete_hook));
+}
+BENCHMARK(BM_new_hooked_sized_delete)->Range(1, 1 << 20);
+
 static void BM_size_returning_new_delete(benchmark::State& state) {
   const int arg = state.range(0);