| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 // The allocator shim is only enabled in Release Static builds. | |
| 6 // This #if is needed as gyp can't have different compile | |
| 7 // targets between Debug and Release. | |
| 8 // TODO(wfh): Remove this once gyp is dead. | |
| 9 // TODO(siggi): Remove this file once the generic shim sticks. | |
| 10 #if defined(ALLOCATOR_SHIM) | |
| 11 | |
| 12 #include <limits.h> | |
| 13 #include <malloc.h> | |
| 14 #include <new.h> | |
| 15 #include <windows.h> | |
| 16 #include <stddef.h> | |
| 17 | |
| 18 #include "base/allocator/allocator_shim_win.h" | |
| 19 #include "base/allocator/winheap_stubs_win.h" | |
| 20 | |
| 21 // This shim make it possible to perform additional checks on allocations | |
| 22 // before passing them to the Heap functions. | |
| 23 | |
| 24 // Override heap functions to perform additional checks: | |
| 25 // 1. Enforcing the maximum size that can be allocated to 2Gb. | |
| 26 // 2. Calling new_handler if malloc fails | |
| 27 | |
| 28 // See definitions of original functions in ucrt\corecrt_malloc.h in SDK | |
| 29 // include directory. | |
| 30 | |
| 31 namespace { | |
| 32 | |
| 33 int new_mode = 0; | |
| 34 | |
| 35 } // namespace | |
| 36 | |
| 37 extern "C" { | |
| 38 | |
| 39 // Symbol to allow weak linkage to WinHeapMalloc from allocator_impl_win.cc. | |
| 40 void* (*malloc_unchecked)(size_t) = &base::allocator::WinHeapMalloc; | |
| 41 | |
| 42 // This function behaves similarly to MSVC's _set_new_mode. | |
| 43 // If flag is 0 (default), calls to malloc will behave normally. | |
| 44 // If flag is 1, calls to malloc will behave like calls to new, | |
| 45 // and the std_new_handler will be invoked on failure. | |
| 46 // Returns the previous mode. | |
| 47 // | |
| 48 // Replaces _set_new_mode in ucrt\heap\new_mode.cpp | |
| 49 int _set_new_mode(int flag) { | |
| 50 // The MS CRT calls this function early on in startup, so this serves as a low | |
| 51 // overhead proof that the allocator shim is in place for this process. | |
| 52 base::allocator::g_is_win_shim_layer_initialized = true; | |
| 53 int old_mode = new_mode; | |
| 54 new_mode = flag; | |
| 55 | |
| 56 return old_mode; | |
| 57 } | |
| 58 | |
| 59 // Replaces _query_new_mode in ucrt\heap\new_mode.cpp | |
| 60 int _query_new_mode() { | |
| 61 return new_mode; | |
| 62 } | |
| 63 | |
| 64 // Replaces malloc in ucrt\heap\malloc.cpp | |
| 65 __declspec(restrict) void* malloc(size_t size) { | |
| 66 void* ptr; | |
| 67 for (;;) { | |
| 68 ptr = base::allocator::WinHeapMalloc(size); | |
| 69 if (ptr) | |
| 70 return ptr; | |
| 71 | |
| 72 if (!new_mode || base::allocator::WinCallNewHandler(size)) | |
| 73 break; | |
| 74 } | |
| 75 return ptr; | |
| 76 } | |
| 77 | |
| 78 // Replaces free in ucrt\heap\free.cpp | |
| 79 void free(void* p) { | |
| 80 base::allocator::WinHeapFree(p); | |
| 81 return; | |
| 82 } | |
| 83 | |
| 84 // Replaces realloc in ucrt\heap\realloc.cpp | |
| 85 __declspec(restrict) void* realloc(void* ptr, size_t size) { | |
| 86 // Webkit is brittle for allocators that return NULL for malloc(0). The | |
| 87 // realloc(0, 0) code path does not guarantee a non-NULL return, so be sure | |
| 88 // to call malloc for this case. | |
| 89 if (!ptr) | |
| 90 return malloc(size); | |
| 91 | |
| 92 void* new_ptr; | |
| 93 for (;;) { | |
| 94 new_ptr = base::allocator::WinHeapRealloc(ptr, size); | |
| 95 | |
| 96 // Subtle warning: NULL return does not alwas indicate out-of-memory. If | |
| 97 // the requested new size is zero, realloc should free the ptr and return | |
| 98 // NULL. | |
| 99 if (new_ptr || !size) | |
| 100 return new_ptr; | |
| 101 if (!new_mode || !base::allocator::WinCallNewHandler(size)) | |
| 102 break; | |
| 103 } | |
| 104 return new_ptr; | |
| 105 } | |
| 106 | |
| 107 // Replaces calloc in ucrt\heap\calloc.cpp | |
| 108 __declspec(restrict) void* calloc(size_t n, size_t elem_size) { | |
| 109 // Overflow check. | |
| 110 const size_t size = n * elem_size; | |
| 111 if (elem_size != 0 && size / elem_size != n) | |
| 112 return nullptr; | |
| 113 | |
| 114 void* result = malloc(size); | |
| 115 if (result) { | |
| 116 memset(result, 0, size); | |
| 117 } | |
| 118 return result; | |
| 119 } | |
| 120 | |
| 121 } // extern C | |
| 122 | |
| 123 #endif // defined(ALLOCATOR_SHIM) | |
| OLD | NEW |