Remove tcmalloc when not being used. Restore shim on Windows.

base/allocator/allocator_shim_win.cc

Index: base/allocator/allocator_shim_win.cc
diff --git a/base/allocator/allocator_shim_win.cc b/base/allocator/allocator_shim_win.cc
new file mode 100644
index 0000000000000000000000000000000000000000..aaf0f4864593230af21948d52ebcb4bc647b2356
--- /dev/null
+++ b/base/allocator/allocator_shim_win.cc
@@ -0,0 +1,250 @@
+// 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 <malloc.h>
+#include <new.h>
+#include <windows.h>
+
+#include <limits>
+
+//#include "base/allocator/allocator_extension_thunks.h"

[scottmg, 2015/01/09 23:32:48]

remove these?

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

remove commented includes.

[Will Harris, 2015/01/11 06:30:39]

Done.

[Will Harris, 2015/01/11 06:30:39]

Done.

+#include "base/basictypes.h"
+//#include "base/profiler/alternate_timer.h"

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

ditto

[Will Harris, 2015/01/11 06:30:39]

Done.

+
+// This shim make it possible to perform additional checks on allocations
+// before passing them to the Heap functions.
+
+// new_mode behaves similarly to MSVC's _set_new_mode.
+// If flag is 0 (default), calls to malloc will behave normally.
+// If flag is 1, calls to malloc will behave like calls to new,
+// and the std_new_handler will be invoked on failure.
+// Can be set by calling _set_new_mode().
+static int new_mode = 0;
+
+// This is a simple allocator based on the windows heap.
+extern "C" {

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

I don't think we need extern here, since these guys are not exposed.

[Will Harris, 2015/01/11 06:30:39]

Done.

+
+const size_t kWindowsPageSize = 4096;
+size_t kMaxWindowsAllocation =
+    std::numeric_limits<int>::max() - kWindowsPageSize;

[scottmg, 2015/01/09 23:32:48]

less confusing to delete this and leave the initialization only in win_heap_init

[Will Harris, 2015/01/11 06:30:39]

Done.

+static HANDLE win_heap;
+
+bool win_heap_init() {

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

add comment, along the lines
// VS2013 crt uses the process heap as its heap, so we do the same here.

[Will Harris, 2015/01/11 06:30:39]

Done.

+  win_heap = GetProcessHeap();
+  if (win_heap == NULL)
+    return false;
+
+  ULONG enable_lfh = 2;
+  // NOTE: Setting LFH may fail.  Vista already has it enabled.
+  //       And under the debugger, it won't use LFH.  So we
+  //       ignore any errors.
+  HeapSetInformation(win_heap, HeapCompatibilityInformation, &enable_lfh,
+                     sizeof(enable_lfh));
+
+  // In Debug, _mt_init() performs an allocation before _cinit is called to
+  // initalize C data, so manually initialize kMaxWindowsAllocation here.
+  kMaxWindowsAllocation = std::numeric_limits<int>::max() - kWindowsPageSize;

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

I think if define kMaxWindowAllocation without ...max() and instead as just
kMaxWindowAllocation = number

Then it can be const and we don't need line 47

of course we need ifdef() for different numbers for 32 and 64.

[Will Harris, 2015/01/11 06:30:39]

Done.  The allocation limit is the same on both 32 and 64.

+  return true;
+}
+
+void* win_heap_malloc(size_t size) {
+  if (size < kMaxWindowsAllocation)
+    return HeapAlloc(win_heap, 0, size);
+  return NULL;
+}
+
+void win_heap_free(void* size) {
+  HeapFree(win_heap, 0, size);
+}
+
+void* win_heap_realloc(void* ptr, size_t size) {
+  if (!ptr)
+    return win_heap_malloc(size);
+  if (!size) {
+    win_heap_free(ptr);
+    return NULL;
+  }
+  if (size < kMaxWindowsAllocation)
+    return HeapReAlloc(win_heap, 0, ptr, size);
+  return NULL;
+}
+
+size_t win_heap_msize(void* ptr) {
+  return HeapSize(win_heap, 0, ptr);
+}
+
+void* win_heap_memalign(size_t alignment, size_t size) {
+  // Reserve enough space to ensure we can align and set aligned_ptr[-1] to the
+  // original allocation for use with win_heap_memalign_free() later.
+  size_t allocation_size = size + (alignment - 1) + sizeof(void*);
+
+  // Check for overflow.  Alignment and size are checked in allocator_shim.
+  if (size >= allocation_size || alignment >= allocation_size) {
+    return NULL;
+  }
+
+  // Since we're directly calling the allocator function, before OOM handling,
+  // we need to NULL check to ensure the allocation succeeded.
+  void* ptr = win_heap_malloc(allocation_size);
+  if (!ptr)
+    return ptr;
+
+  char* aligned_ptr = static_cast<char*>(ptr) + sizeof(void*);
+  aligned_ptr +=
+      alignment - reinterpret_cast<uintptr_t>(aligned_ptr) & (alignment - 1);
+
+  reinterpret_cast<void**>(aligned_ptr)[-1] = ptr;
+  return aligned_ptr;
+}
+
+void win_heap_memalign_free(void* ptr) {
+  if (ptr)
+    win_heap_free(static_cast<void**>(ptr)[-1]);
+}
+
+}  // extern "C"
+
+// Call the new handler, if one has been set.
+// Returns true on successfully calling the handler, false otherwise.
+inline bool call_new_handler(bool nothrow, size_t size) {
+  // Get the current new handler.
+  _PNH nh = _query_new_handler();
+#if (defined(__GNUC__) && !defined(__EXCEPTIONS)) || \

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

kill _GNUC_

[Will Harris, 2015/01/11 06:30:39]

Done.

+    (defined(_HAS_EXCEPTIONS) && !_HAS_EXCEPTIONS)
+  if (!nh)
+    return false;
+  // Since exceptions are disabled, we don't really know if new_handler
+  // failed.  Assume it will abort if it fails.
+  return nh(size);
+#else
+  // If no new_handler is established, the allocation failed.
+  if (!nh) {
+    if (nothrow)
+      return false;
+    throw std::bad_alloc();
+  }
+  // Otherwise, try the new_handler.  If it returns, retry the
+  // allocation.  If it throws std::bad_alloc, fail the allocation.
+  // if it throws something else, don't interfere.
+  try {
+    return nh(size);
+  } catch (const std::bad_alloc&) {
+    if (!nothrow)
+      throw;
+    return true;
+  }
+#endif  // (defined(__GNUC__) && !defined(__EXCEPTIONS)) || (defined(_HAS_EXCEPTIONS) && !_HAS_EXCEPTIONS)

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

instead of this monkeing, we should just #error if __EXCEPTIONS or the other one
are defined

+  return false;
+}
+
+extern "C" {
+void* malloc(size_t size) {
+  void* ptr;
+  for (;;) {
+    ptr = win_heap_malloc(size);
+    if (ptr)
+      return ptr;
+
+    if (!new_mode || !call_new_handler(true, size))
+      break;
+  }
+  return ptr;
+}
+
+void free(void* p) {
+  win_heap_free(p);
+  return;
+}
+
+void* realloc(void* ptr, size_t size) {
+  // Webkit is brittle for allocators that return NULL for malloc(0).  The
+  // realloc(0, 0) code path does not guarantee a non-NULL return, so be sure
+  // to call malloc for this case.
+  if (!ptr)
+    return malloc(size);
+
+  void* new_ptr;
+  for (;;) {
+    new_ptr = win_heap_realloc(ptr, size);
+
+    // Subtle warning:  NULL return does not alwas indicate out-of-memory.  If
+    // the requested new size is zero, realloc should free the ptr and return
+    // NULL.
+    if (new_ptr || !size)
+      return new_ptr;
+    if (!new_mode || !call_new_handler(true, size))
+      break;
+  }
+  return new_ptr;
+}
+
+// TODO(mbelshe): Implement this for other allocators.

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

I think this can be deleted.

[Will Harris, 2015/01/11 06:30:39]

Done.

+void malloc_stats(void) {
+  // No stats.
+  return;
+}
+
+#ifdef WIN32

[scottmg, 2015/01/09 23:32:48]

remove this #if since we're in a _win.cc.

[Will Harris, 2015/01/11 06:30:39]

Done.

+
+extern "C" size_t _msize(void* p) {
+  return win_heap_msize(p);
+}
+
+extern "C" intptr_t _get_heap_handle() {
+  return reinterpret_cast<intptr_t>(win_heap);
+}
+
+static bool get_allocator_waste_size_thunk(size_t* size) {
+  // TODO(alexeif): Implement for allocators other than tcmalloc.
+  return false;
+}
+
+// The CRT heap initialization stub.
+extern "C" int _heap_init() {
+  return win_heap_init() ? 1 : 0;
+}
+
+// The CRT heap cleanup stub.
+extern "C" void _heap_term() {}
+
+// We set this to 1 because part of the CRT uses a check of _crtheap != 0
+// to test whether the CRT has been initialized.  Once we've ripped out
+// the allocators from libcmt, we need to provide this definition so that
+// the rest of the CRT is still usable.
+extern "C" void* _crtheap = reinterpret_cast<void*>(1);
+
+// Provide support for aligned memory through Windows only _aligned_malloc().
+void* _aligned_malloc(size_t size, size_t alignment) {
+  // _aligned_malloc guarantees parameter validation, so do so here.  These
+  // checks are somewhat stricter than _aligned_malloc() since we're effectively
+  // using memalign() under the hood.
+  if (size == 0U || (alignment & (alignment - 1)) != 0U ||
+      (alignment % sizeof(void*)) != 0U)
+    return NULL;
+
+  void* ptr;
+  for (;;) {
+    ptr = win_heap_memalign(alignment, size);
+
+    if (ptr) {
+      return ptr;
+    }
+
+    if (!new_mode || !call_new_handler(true, size))
+      break;
+  }
+  return ptr;
+}
+
+void _aligned_free(void* p) {
+  // Pointers allocated with win_heap_memalign() MUST be freed via
+  // win_heap_memalign_free() since the aligned pointer is not the real one.
+  win_heap_memalign_free(p);
+}
+
+#endif  // WIN32

[cpu_(ooo_6.6-7.5), 2015/01/09 23:46:56]

kill this

[Will Harris, 2015/01/11 06:30:39]

Done.

+
+#include "generic_allocators.cc"
+
+}  // extern C