1 //===-- tsan_new_delete.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 ThreadSanitizer (TSan), a race detector. 9 // 10 // Interceptors for operators new and delete. 11 //===----------------------------------------------------------------------===// 12 #include "interception/interception.h" 13 #include "sanitizer_common/sanitizer_allocator.h" 14 #include "sanitizer_common/sanitizer_allocator_report.h" 15 #include "sanitizer_common/sanitizer_internal_defs.h" 16 #include "tsan_interceptors.h" 17 #include "tsan_rtl.h" 18 19 using namespace __tsan; // NOLINT 20 21 namespace std { 22 struct nothrow_t {}; 23 enum class align_val_t: __sanitizer::uptr {}; 24 } // namespace std 25 26 DECLARE_REAL(void *, malloc, uptr size) 27 DECLARE_REAL(void, free, void *ptr) 28 29 // TODO(alekseys): throw std::bad_alloc instead of dying on OOM. 30 #define OPERATOR_NEW_BODY(mangled_name, nothrow) \ 31 if (cur_thread()->in_symbolizer) \ 32 return InternalAlloc(size); \ 33 void *p = 0; \ 34 { \ 35 SCOPED_INTERCEPTOR_RAW(mangled_name, size); \ 36 p = user_alloc(thr, pc, size); \ 37 if (!nothrow && UNLIKELY(!p)) { \ 38 GET_STACK_TRACE_FATAL(thr, pc); \ 39 ReportOutOfMemory(size, &stack); \ 40 } \ 41 } \ 42 invoke_malloc_hook(p, size); \ 43 return p; 44 45 #define OPERATOR_NEW_BODY_ALIGN(mangled_name, nothrow) \ 46 if (cur_thread()->in_symbolizer) \ 47 return InternalAlloc(size, nullptr, (uptr)align); \ 48 void *p = 0; \ 49 { \ 50 SCOPED_INTERCEPTOR_RAW(mangled_name, size); \ 51 p = user_memalign(thr, pc, (uptr)align, size); \ 52 if (!nothrow && UNLIKELY(!p)) { \ 53 GET_STACK_TRACE_FATAL(thr, pc); \ 54 ReportOutOfMemory(size, &stack); \ 55 } \ 56 } \ 57 invoke_malloc_hook(p, size); \ 58 return p; 59 60 SANITIZER_INTERFACE_ATTRIBUTE 61 void *operator new(__sanitizer::uptr size); 62 void *operator new(__sanitizer::uptr size) { 63 OPERATOR_NEW_BODY(_Znwm, false /*nothrow*/); 64 } 65 66 SANITIZER_INTERFACE_ATTRIBUTE 67 void *operator new[](__sanitizer::uptr size); 68 void *operator new[](__sanitizer::uptr size) { 69 OPERATOR_NEW_BODY(_Znam, false /*nothrow*/); 70 } 71 72 SANITIZER_INTERFACE_ATTRIBUTE 73 void *operator new(__sanitizer::uptr size, std::nothrow_t const&); 74 void *operator new(__sanitizer::uptr size, std::nothrow_t const&) { 75 OPERATOR_NEW_BODY(_ZnwmRKSt9nothrow_t, true /*nothrow*/); 76 } 77 78 SANITIZER_INTERFACE_ATTRIBUTE 79 void *operator new[](__sanitizer::uptr size, std::nothrow_t const&); 80 void *operator new[](__sanitizer::uptr size, std::nothrow_t const&) { 81 OPERATOR_NEW_BODY(_ZnamRKSt9nothrow_t, true /*nothrow*/); 82 } 83 84 SANITIZER_INTERFACE_ATTRIBUTE 85 void *operator new(__sanitizer::uptr size, std::align_val_t align); 86 void *operator new(__sanitizer::uptr size, std::align_val_t align) { 87 OPERATOR_NEW_BODY_ALIGN(_ZnwmSt11align_val_t, false /*nothrow*/); 88 } 89 90 SANITIZER_INTERFACE_ATTRIBUTE 91 void *operator new[](__sanitizer::uptr size, std::align_val_t align); 92 void *operator new[](__sanitizer::uptr size, std::align_val_t align) { 93 OPERATOR_NEW_BODY_ALIGN(_ZnamSt11align_val_t, false /*nothrow*/); 94 } 95 96 SANITIZER_INTERFACE_ATTRIBUTE 97 void *operator new(__sanitizer::uptr size, std::align_val_t align, 98 std::nothrow_t const&); 99 void *operator new(__sanitizer::uptr size, std::align_val_t align, 100 std::nothrow_t const&) { 101 OPERATOR_NEW_BODY_ALIGN(_ZnwmSt11align_val_tRKSt9nothrow_t, 102 true /*nothrow*/); 103 } 104 105 SANITIZER_INTERFACE_ATTRIBUTE 106 void *operator new[](__sanitizer::uptr size, std::align_val_t align, 107 std::nothrow_t const&); 108 void *operator new[](__sanitizer::uptr size, std::align_val_t align, 109 std::nothrow_t const&) { 110 OPERATOR_NEW_BODY_ALIGN(_ZnamSt11align_val_tRKSt9nothrow_t, 111 true /*nothrow*/); 112 } 113 114 #define OPERATOR_DELETE_BODY(mangled_name) \ 115 if (ptr == 0) return; \ 116 if (cur_thread()->in_symbolizer) \ 117 return InternalFree(ptr); \ 118 invoke_free_hook(ptr); \ 119 SCOPED_INTERCEPTOR_RAW(mangled_name, ptr); \ 120 user_free(thr, pc, ptr); 121 122 SANITIZER_INTERFACE_ATTRIBUTE 123 void operator delete(void *ptr) NOEXCEPT; 124 void operator delete(void *ptr) NOEXCEPT { 125 OPERATOR_DELETE_BODY(_ZdlPv); 126 } 127 128 SANITIZER_INTERFACE_ATTRIBUTE 129 void operator delete[](void *ptr) NOEXCEPT; 130 void operator delete[](void *ptr) NOEXCEPT { 131 OPERATOR_DELETE_BODY(_ZdaPv); 132 } 133 134 SANITIZER_INTERFACE_ATTRIBUTE 135 void operator delete(void *ptr, std::nothrow_t const&); 136 void operator delete(void *ptr, std::nothrow_t const&) { 137 OPERATOR_DELETE_BODY(_ZdlPvRKSt9nothrow_t); 138 } 139 140 SANITIZER_INTERFACE_ATTRIBUTE 141 void operator delete[](void *ptr, std::nothrow_t const&); 142 void operator delete[](void *ptr, std::nothrow_t const&) { 143 OPERATOR_DELETE_BODY(_ZdaPvRKSt9nothrow_t); 144 } 145 146 SANITIZER_INTERFACE_ATTRIBUTE 147 void operator delete(void *ptr, __sanitizer::uptr size) NOEXCEPT; 148 void operator delete(void *ptr, __sanitizer::uptr size) NOEXCEPT { 149 OPERATOR_DELETE_BODY(_ZdlPvm); 150 } 151 152 SANITIZER_INTERFACE_ATTRIBUTE 153 void operator delete[](void *ptr, __sanitizer::uptr size) NOEXCEPT; 154 void operator delete[](void *ptr, __sanitizer::uptr size) NOEXCEPT { 155 OPERATOR_DELETE_BODY(_ZdaPvm); 156 } 157 158 SANITIZER_INTERFACE_ATTRIBUTE 159 void operator delete(void *ptr, std::align_val_t align) NOEXCEPT; 160 void operator delete(void *ptr, std::align_val_t align) NOEXCEPT { 161 OPERATOR_DELETE_BODY(_ZdlPvSt11align_val_t); 162 } 163 164 SANITIZER_INTERFACE_ATTRIBUTE 165 void operator delete[](void *ptr, std::align_val_t align) NOEXCEPT; 166 void operator delete[](void *ptr, std::align_val_t align) NOEXCEPT { 167 OPERATOR_DELETE_BODY(_ZdaPvSt11align_val_t); 168 } 169 170 SANITIZER_INTERFACE_ATTRIBUTE 171 void operator delete(void *ptr, std::align_val_t align, std::nothrow_t const&); 172 void operator delete(void *ptr, std::align_val_t align, std::nothrow_t const&) { 173 OPERATOR_DELETE_BODY(_ZdlPvSt11align_val_tRKSt9nothrow_t); 174 } 175 176 SANITIZER_INTERFACE_ATTRIBUTE 177 void operator delete[](void *ptr, std::align_val_t align, 178 std::nothrow_t const&); 179 void operator delete[](void *ptr, std::align_val_t align, 180 std::nothrow_t const&) { 181 OPERATOR_DELETE_BODY(_ZdaPvSt11align_val_tRKSt9nothrow_t); 182 } 183 184 SANITIZER_INTERFACE_ATTRIBUTE 185 void operator delete(void *ptr, __sanitizer::uptr size, 186 std::align_val_t align) NOEXCEPT; 187 void operator delete(void *ptr, __sanitizer::uptr size, 188 std::align_val_t align) NOEXCEPT { 189 OPERATOR_DELETE_BODY(_ZdlPvmSt11align_val_t); 190 } 191 192 SANITIZER_INTERFACE_ATTRIBUTE 193 void operator delete[](void *ptr, __sanitizer::uptr size, 194 std::align_val_t align) NOEXCEPT; 195 void operator delete[](void *ptr, __sanitizer::uptr size, 196 std::align_val_t align) NOEXCEPT { 197 OPERATOR_DELETE_BODY(_ZdaPvmSt11align_val_t); 198 } 199