Index: base/allocator/allocator_shim_override_ucrt_symbols_win.h |
diff --git a/base/allocator/allocator_shim_override_ucrt_symbols_win.h b/base/allocator/allocator_shim_override_ucrt_symbols_win.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..2aa872f1b3fbb1c5d4d7d120f96b5383f2d8b54d |
--- /dev/null |
+++ b/base/allocator/allocator_shim_override_ucrt_symbols_win.h |
@@ -0,0 +1,74 @@ |
+// Copyright 2016 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. |
+ |
+// This header defines symbols to override the same functions in the Visual C++ |
+// CRT implementation. |
+ |
+#ifdef BASE_ALLOCATOR_ALLOCATOR_SHIM_OVERRIDE_UCRT_SYMBOLS_WIN_H_ |
+#error This header is meant to be included only once by allocator_shim.cc |
+#endif |
+#define BASE_ALLOCATOR_ALLOCATOR_SHIM_OVERRIDE_UCRT_SYMBOLS_WIN_H_ |
+ |
+#include <malloc.h> |
+ |
+extern "C" { |
+ |
+void* (*malloc_unchecked)(size_t) = &base::allocator::UncheckedAlloc; |
+ |
+namespace { |
+ |
+int win_new_mode = 0; |
+ |
+} // namespace |
+ |
+// This function 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. |
+// Returns the previous mode. |
+// |
+// Replaces _set_new_mode in ucrt\heap\new_mode.cpp |
+int _set_new_mode(int flag) { |
+ // The MS CRT calls this function early on in startup, so this serves as a low |
+ // overhead proof that the allocator shim is in place for this process. |
+ base::allocator::g_is_win_shim_layer_initialized = true; |
+ int old_mode = win_new_mode; |
+ win_new_mode = flag; |
+ |
+ base::allocator::SetCallNewHandlerOnMallocFailure(win_new_mode != 0); |
+ |
+ return old_mode; |
+} |
+ |
+// Replaces _query_new_mode in ucrt\heap\new_mode.cpp |
+int _query_new_mode() { |
+ return win_new_mode; |
+} |
+ |
+// These symbols override the CRT's implementation of the same functions. |
+__declspec(restrict) void* malloc(size_t size) { |
+ return ShimMalloc(size); |
+} |
+ |
+void free(void* ptr) { |
+ ShimFree(ptr); |
+} |
+ |
+__declspec(restrict) void* realloc(void* ptr, size_t size) { |
+ return ShimRealloc(ptr, size); |
+} |
+ |
+__declspec(restrict) void* calloc(size_t n, size_t size) { |
+ return ShimCalloc(n, size); |
+} |
+ |
+// The default dispatch translation unit has to define also the following |
+// symbols (unless they are ultimately routed to the system symbols): |
+// void malloc_stats(void); |
+// int mallopt(int, int); |
+// struct mallinfo mallinfo(void); |
+// size_t malloc_size(void*); |
+// size_t malloc_usable_size(const void*); |
+ |
+} // extern "C" |