Make possible to check memory allocations inside chromium

base/allocator/allocator_shim.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 "base/allocator/allocator_shim.h"
 
 #include <config.h>
 #include "base/allocator/allocator_extension_thunks.h"
+#include "base/process/memory.h"
 #include "base/profiler/alternate_timer.h"
 #include "base/sysinfo.h"
 #include "jemalloc.h"
 
 // This shim make it possible to use different allocators via an environment
 // variable set before running the program. This may reduce the
 // amount of inlining that we get with malloc/free/etc.
 
 // TODO(mbelshe): Ensure that all calls to tcmalloc have the proper call depth
 // from the "user code" so that debugging tools (HeapChecker) can work.
@@ skipping 354 matching lines @@
   }
 }
 
 #endif  // WIN32
 
 #include "generic_allocators.cc"
 
 }  // extern C
 
 namespace base {
+
+bool UncheckedMalloc(size_t size, void** result) {
+  switch (::allocator) {
+    case JEMALLOC:
+      *result = je_malloc(size);
+      break;
+    case WINHEAP:
+    case WINLFH:
+      *result = win_heap_malloc(size);
+      break;
+    case TCMALLOC:
+      *result = do_malloc(size);
+      break;
+  }
+
+  return *result;
+}
+
 namespace allocator {
 
 void SetupSubprocessAllocator() {
   size_t primary_length = 0;
   getenv_s(&primary_length, NULL, 0, primary_name);
 
   size_t secondary_length = 0;
   char buffer[20];
   getenv_s(&secondary_length, buffer, sizeof(buffer), secondary_name);
   DCHECK_GT(sizeof(buffer), secondary_length);
@@ skipping 20 matching lines @@
 void TCMallocDoFreeForTest(void* ptr) {
   do_free(ptr);
 }
 
 size_t ExcludeSpaceForMarkForTest(size_t size) {
   return ExcludeSpaceForMark(size);
 }
 
 }  // namespace allocator.
 }  // namespace base.