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);
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);
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&);
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&);
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);
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);
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&);
operator new(__sanitizer::uptr size,std::align_val_t align,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&);
operator new[](__sanitizer::uptr size,std::align_val_t align,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;
operator delete(void * ptr)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;
operator delete[](void * ptr)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&);
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&);
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;
operator delete(void * ptr,__sanitizer::uptr size)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;
operator delete[](void * ptr,__sanitizer::uptr size)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;
operator delete(void * ptr,std::align_val_t align)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;
operator delete[](void * ptr,std::align_val_t align)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&);
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&);
operator delete[](void * ptr,std::align_val_t align,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;
operator delete(void * ptr,__sanitizer::uptr size,std::align_val_t align)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;
operator delete[](void * ptr,__sanitizer::uptr size,std::align_val_t align)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