third_party/tcmalloc/chromium/src/windows/override_functions.cc

   1 // Copyright (c) 2007, Google Inc.
   2 // All rights reserved.
   3 //
   4 // Redistribution and use in source and binary forms, with or without
   5 // modification, are permitted provided that the following conditions are
   6 // met:
   7 //
   8 //     * Redistributions of source code must retain the above copyright
   9 // notice, this list of conditions and the following disclaimer.
  10 //     * Redistributions in binary form must reproduce the above
  11 // copyright notice, this list of conditions and the following disclaimer
  12 // in the documentation and/or other materials provided with the
  13 // distribution.
  14 //     * Neither the name of Google Inc. nor the names of its
  15 // contributors may be used to endorse or promote products derived from
  16 // this software without specific prior written permission.
  17 //
  18 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29 //
  30 // ---
  31 // Author: Mike Belshe
  32 //
  33 // To link tcmalloc into a EXE or DLL statically without using the patching
  34 // facility, we can take a stock libcmt and remove all the allocator functions.
  35 // When we relink the EXE/DLL with the modified libcmt and tcmalloc, a few
  36 // functions are missing.  This file contains the additional overrides which
  37 // are required in the VS2005 libcmt in order to link the modified libcmt.
  38 //
  39 // See also
  40 // http://groups.google.com/group/google-perftools/browse_thread/thread/41cd3710af85e57b
  41
  42 #include <config.h>
  43
  44 #ifndef _WIN32
  45 # error You should only be including this file in a windows environment!
  46 #endif
  47
  48 #ifndef WIN32_OVERRIDE_ALLOCATORS
  49 # error This file is intended for use when overriding allocators
  50 #endif
  51
  52 #include "tcmalloc.cc"
  53
  54 extern "C" void* _recalloc(void* p, size_t n, size_t size) {
  55   void* result = realloc(p, n * size);
  56   memset(result, 0, n * size);
  57   return result;
  58 }
  59
  60 extern "C" void* _calloc_impl(size_t n, size_t size) {
  61   return calloc(n, size);
  62 }
  63
  64 extern "C" size_t _msize(void* p) {
  65   return MallocExtension::instance()->GetAllocatedSize(p);
  66 }
  67
  68 extern "C" intptr_t _get_heap_handle() {
  69   return 0;
  70 }
  71
  72 // The CRT heap initialization stub.
  73 extern "C" int _heap_init() {
  74   // We intentionally leak this object.  It lasts for the process
  75   // lifetime.  Trying to teardown at _heap_term() is so late that
  76   // you can't do anything useful anyway.
  77   new TCMallocGuard();
  78   return 1;
  79 }
  80
  81 // The CRT heap cleanup stub.
  82 extern "C" void _heap_term() {
  83 }
  84
  85 extern "C" int _set_new_mode(int flag) {
  86   return tc_set_new_mode(flag);
  87 }
  88
  89 #ifndef NDEBUG
  90 #undef malloc
  91 #undef free
  92 #undef calloc
  93 int _CrtDbgReport(int, const char*, int, const char*, const char*, ...) {
  94   return 0;
  95 }
  96
  97 int _CrtDbgReportW(int, const wchar_t*, int, const wchar_t*, const wchar_t*, ...) {
  98   return 0;
  99 }
 100
 101 int _CrtSetReportMode(int, int) {
 102   return 0;
 103 }
 104
 105 extern "C" void* _malloc_dbg(size_t size, int , const char*, int) {
 106   return malloc(size);
 107 }
 108
 109 extern "C" void _free_dbg(void* ptr, int) {
 110   free(ptr);
 111 }
 112
 113 extern "C" void* _calloc_dbg(size_t n, size_t size, int, const char*, int) {
 114   return calloc(n, size);
 115 }
 116 #endif  // NDEBUG
 117
 118 // We set this to 1 because part of the CRT uses a check of _crtheap != 0
 119 // to test whether the CRT has been initialized.  Once we've ripped out
 120 // the allocators from libcmt, we need to provide this definition so that
 121 // the rest of the CRT is still usable.
 122 extern "C" void* _crtheap = reinterpret_cast<void*>(1);