123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178 |
- // 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.
- #ifndef BASE_ALLOCATOR_ALLOCATOR_SHIM_OVERRIDE_UCRT_SYMBOLS_WIN_H_
- #define BASE_ALLOCATOR_ALLOCATOR_SHIM_OVERRIDE_UCRT_SYMBOLS_WIN_H_
- #include <malloc.h>
- #include <windows.h>
- #include "base/allocator/allocator_shim_internals.h"
- // Even though most C++ allocation operators can be left alone since the
- // interception works at a lower level, these ones should be
- // overridden. Otherwise they redirect to malloc(), which is configured to crash
- // with an OOM in failure cases, such as allocation requests that are too large.
- SHIM_ALWAYS_EXPORT void* operator new(size_t size,
- const std::nothrow_t&) noexcept {
- return ShimCppNewNoThrow(size);
- }
- SHIM_ALWAYS_EXPORT void* operator new[](size_t size,
- const std::nothrow_t&) noexcept {
- return ShimCppNewNoThrow(size);
- }
- 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, nullptr);
- }
- void free(void* ptr) {
- ShimFree(ptr, nullptr);
- }
- __declspec(restrict) void* realloc(void* ptr, size_t size) {
- return ShimRealloc(ptr, size, nullptr);
- }
- __declspec(restrict) void* calloc(size_t n, size_t size) {
- return ShimCalloc(n, size, nullptr);
- }
- // _msize() is the Windows equivalent of malloc_size().
- size_t _msize(void* memblock) {
- return ShimGetSizeEstimate(memblock, nullptr);
- }
- __declspec(restrict) void* _aligned_malloc(size_t size, size_t alignment) {
- return ShimAlignedMalloc(size, alignment, nullptr);
- }
- __declspec(restrict) void* _aligned_realloc(void* address,
- size_t size,
- size_t alignment) {
- return ShimAlignedRealloc(address, size, alignment, nullptr);
- }
- void _aligned_free(void* address) {
- ShimAlignedFree(address, nullptr);
- }
- // _recalloc_base is called by CRT internally.
- __declspec(restrict) void* _recalloc_base(void* block,
- size_t count,
- size_t size) {
- const size_t old_block_size = (block != nullptr) ? _msize(block) : 0;
- base::CheckedNumeric<size_t> new_block_size_checked = count;
- new_block_size_checked *= size;
- const size_t new_block_size = new_block_size_checked.ValueOrDie();
- void* const new_block = realloc(block, new_block_size);
- if (new_block != nullptr && old_block_size < new_block_size) {
- memset(static_cast<char*>(new_block) + old_block_size, 0,
- new_block_size - old_block_size);
- }
- return new_block;
- }
- __declspec(restrict) void* _malloc_base(size_t size) {
- return malloc(size);
- }
- __declspec(restrict) void* _calloc_base(size_t n, size_t size) {
- return calloc(n, size);
- }
- void _free_base(void* block) {
- free(block);
- }
- __declspec(restrict) void* _recalloc(void* block, size_t count, size_t size) {
- return _recalloc_base(block, count, size);
- }
- // The following uncommon _aligned_* routines are not used in Chromium and have
- // been shimmed to immediately crash to ensure that implementations are added if
- // uses are introduced.
- __declspec(restrict) void* _aligned_recalloc(void* address,
- size_t num,
- size_t size,
- size_t alignment) {
- CHECK(false) << "This routine has not been implemented";
- __builtin_unreachable();
- }
- size_t _aligned_msize(void* address, size_t alignment, size_t offset) {
- CHECK(false) << "This routine has not been implemented";
- __builtin_unreachable();
- }
- __declspec(restrict) void* _aligned_offset_malloc(size_t size,
- size_t alignment,
- size_t offset) {
- CHECK(false) << "This routine has not been implemented";
- __builtin_unreachable();
- }
- __declspec(restrict) void* _aligned_offset_realloc(void* address,
- size_t size,
- size_t alignment,
- size_t offset) {
- CHECK(false) << "This routine has not been implemented";
- __builtin_unreachable();
- }
- __declspec(restrict) void* _aligned_offset_recalloc(void* address,
- size_t num,
- size_t size,
- size_t alignment,
- size_t offset) {
- CHECK(false) << "This routine has not been implemented";
- __builtin_unreachable();
- }
- } // extern "C"
- #endif // BASE_ALLOCATOR_ALLOCATOR_SHIM_OVERRIDE_UCRT_SYMBOLS_WIN_H_
|