libsanitizer/asan/asan_new_delete.cc

   1 //===-- asan_interceptors.cc ----------------------------------------------===//
   2 //
   3 // This file is distributed under the University of Illinois Open Source
   4 // License. See LICENSE.TXT for details.
   5 //
   6 //===----------------------------------------------------------------------===//
   7 //
   8 // This file is a part of AddressSanitizer, an address sanity checker.
   9 //
  10 // Interceptors for operators new and delete.
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "asan_allocator.h"
  14 #include "asan_internal.h"
  15 #include "asan_stack.h"
  16
  17 #include "sanitizer_common/sanitizer_interception.h"
  18
  19 #include <stddef.h>
  20
  21 using namespace __asan;  // NOLINT
  22
  23 // This code has issues on OSX.
  24 // See https://code.google.com/p/address-sanitizer/issues/detail?id=131.
  25
  26 // Fake std::nothrow_t to avoid including <new>.
  27 namespace std {
  28 struct nothrow_t {};
  29 }  // namespace std
  30
  31 #define OPERATOR_NEW_BODY(type) \
  32   GET_STACK_TRACE_MALLOC;\
  33   return asan_memalign(0, size, &stack, type);
  34
  35 // On OS X it's not enough to just provide our own 'operator new' and
  36 // 'operator delete' implementations, because they're going to be in the
  37 // runtime dylib, and the main executable will depend on both the runtime
  38 // dylib and libstdc++, each of those'll have its implementation of new and
  39 // delete.
  40 // To make sure that C++ allocation/deallocation operators are overridden on
  41 // OS X we need to intercept them using their mangled names.
  42 #if !SANITIZER_MAC
  43 // FreeBSD prior v9.2 have wrong definition of 'size_t'.
  44 // http://svnweb.freebsd.org/base?view=revision&revision=232261
  45 #if SANITIZER_FREEBSD && SANITIZER_WORDSIZE == 32
  46 #include <sys/param.h>
  47 #if __FreeBSD_version <= 902001  // v9.2
  48 #define size_t unsigned
  49 #endif  // __FreeBSD_version
  50 #endif  // SANITIZER_FREEBSD && SANITIZER_WORDSIZE == 32
  51
  52 INTERCEPTOR_ATTRIBUTE
  53 void *operator new(size_t size) { OPERATOR_NEW_BODY(FROM_NEW); }
  54 INTERCEPTOR_ATTRIBUTE
  55 void *operator new[](size_t size) { OPERATOR_NEW_BODY(FROM_NEW_BR); }
  56 INTERCEPTOR_ATTRIBUTE
  57 void *operator new(size_t size, std::nothrow_t const&)
  58 { OPERATOR_NEW_BODY(FROM_NEW); }
  59 INTERCEPTOR_ATTRIBUTE
  60 void *operator new[](size_t size, std::nothrow_t const&)
  61 { OPERATOR_NEW_BODY(FROM_NEW_BR); }
  62
  63 #else  // SANITIZER_MAC
  64 INTERCEPTOR(void *, _Znwm, size_t size) {
  65   OPERATOR_NEW_BODY(FROM_NEW);
  66 }
  67 INTERCEPTOR(void *, _Znam, size_t size) {
  68   OPERATOR_NEW_BODY(FROM_NEW_BR);
  69 }
  70 INTERCEPTOR(void *, _ZnwmRKSt9nothrow_t, size_t size, std::nothrow_t const&) {
  71   OPERATOR_NEW_BODY(FROM_NEW);
  72 }
  73 INTERCEPTOR(void *, _ZnamRKSt9nothrow_t, size_t size, std::nothrow_t const&) {
  74   OPERATOR_NEW_BODY(FROM_NEW_BR);
  75 }
  76 #endif
  77
  78 #define OPERATOR_DELETE_BODY(type) \
  79   GET_STACK_TRACE_FREE;\
  80   asan_free(ptr, &stack, type);
  81
  82 #if !SANITIZER_MAC
  83 INTERCEPTOR_ATTRIBUTE
  84 void operator delete(void *ptr) throw() {
  85   OPERATOR_DELETE_BODY(FROM_NEW);
  86 }
  87 INTERCEPTOR_ATTRIBUTE
  88 void operator delete[](void *ptr) throw() {
  89   OPERATOR_DELETE_BODY(FROM_NEW_BR);
  90 }
  91 INTERCEPTOR_ATTRIBUTE
  92 void operator delete(void *ptr, std::nothrow_t const&) {
  93   OPERATOR_DELETE_BODY(FROM_NEW);
  94 }
  95 INTERCEPTOR_ATTRIBUTE
  96 void operator delete[](void *ptr, std::nothrow_t const&) {
  97   OPERATOR_DELETE_BODY(FROM_NEW_BR);
  98 }
  99
 100 #else  // SANITIZER_MAC
 101 INTERCEPTOR(void, _ZdlPv, void *ptr) {
 102   OPERATOR_DELETE_BODY(FROM_NEW);
 103 }
 104 INTERCEPTOR(void, _ZdaPv, void *ptr) {
 105   OPERATOR_DELETE_BODY(FROM_NEW_BR);
 106 }
 107 INTERCEPTOR(void, _ZdlPvRKSt9nothrow_t, void *ptr, std::nothrow_t const&) {
 108   OPERATOR_DELETE_BODY(FROM_NEW);
 109 }
 110 INTERCEPTOR(void, _ZdaPvRKSt9nothrow_t, void *ptr, std::nothrow_t const&) {
 111   OPERATOR_DELETE_BODY(FROM_NEW_BR);
 112 }
 113 #endif