Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1 | //===-- tsan_interceptors_linux.cc ------------------------------*- C++ -*-===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
| 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
| 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This file is a part of ThreadSanitizer (TSan), a race detector. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | #include "interception/interception.h" |
| 15 | #include "tsan_rtl.h" |
| 16 | #include "tsan_interface.h" |
| 17 | #include "tsan_atomic.h" |
| 18 | #include "tsan_platform.h" |
| 19 | #include "tsan_mman.h" |
| 20 | #include "tsan_placement_new.h" |
| 21 | |
| 22 | using namespace __tsan; // NOLINT |
| 23 | |
| 24 | struct sigset_t { |
| 25 | u64 val[kSigCount / 8 / sizeof(u64)]; |
| 26 | }; |
| 27 | |
| 28 | struct ucontext_t { |
| 29 | u64 opaque[1024]; |
| 30 | }; |
| 31 | |
| 32 | extern "C" int pthread_attr_init(void *attr); |
| 33 | extern "C" int pthread_attr_destroy(void *attr); |
| 34 | extern "C" int pthread_attr_getdetachstate(void *attr, int *v); |
| 35 | extern "C" int pthread_attr_setstacksize(void *attr, uptr stacksize); |
| 36 | extern "C" int pthread_attr_getstacksize(void *attr, uptr *stacksize); |
| 37 | extern "C" int pthread_key_create(unsigned *key, void (*destructor)(void* v)); |
| 38 | extern "C" int pthread_setspecific(unsigned key, const void *v); |
| 39 | extern "C" int pthread_mutexattr_gettype(void *a, int *type); |
| 40 | extern "C" int pthread_yield(); |
| 41 | extern "C" int pthread_sigmask(int how, const sigset_t *set, sigset_t *oldset); |
| 42 | extern "C" int sigfillset(sigset_t *set); |
| 43 | extern "C" void *pthread_self(); |
| 44 | extern "C" int getcontext(ucontext_t *ucp); |
| 45 | extern "C" void _exit(int status); |
| 46 | extern "C" int __cxa_atexit(void (*func)(void *arg), void *arg, void *dso); |
| 47 | extern "C" int *__errno_location(); |
| 48 | extern "C" int usleep(unsigned usec); |
| 49 | const int PTHREAD_MUTEX_RECURSIVE = 1; |
| 50 | const int PTHREAD_MUTEX_RECURSIVE_NP = 1; |
| 51 | const int kPthreadAttrSize = 56; |
| 52 | const int EINVAL = 22; |
| 53 | const int EBUSY = 16; |
| 54 | const int EPOLL_CTL_ADD = 1; |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 55 | const int SIGILL = 4; |
| 56 | const int SIGABRT = 6; |
| 57 | const int SIGFPE = 8; |
| 58 | const int SIGSEGV = 11; |
| 59 | const int SIGPIPE = 13; |
| 60 | const int SIGBUS = 7; |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 61 | void *const MAP_FAILED = (void*)-1; |
| 62 | const int PTHREAD_BARRIER_SERIAL_THREAD = -1; |
| 63 | const int MAP_FIXED = 0x10; |
| 64 | typedef long long_t; // NOLINT |
| 65 | |
| 66 | typedef void (*sighandler_t)(int sig); |
| 67 | |
| 68 | union pthread_attr_t { |
| 69 | char size[kPthreadAttrSize]; |
| 70 | void *align; |
| 71 | }; |
| 72 | |
| 73 | struct sigaction_t { |
| 74 | union { |
| 75 | sighandler_t sa_handler; |
| 76 | void (*sa_sigaction)(int sig, my_siginfo_t *siginfo, void *uctx); |
| 77 | }; |
| 78 | sigset_t sa_mask; |
| 79 | int sa_flags; |
| 80 | void (*sa_restorer)(); |
| 81 | }; |
| 82 | |
| 83 | const sighandler_t SIG_DFL = (sighandler_t)0; |
| 84 | const sighandler_t SIG_IGN = (sighandler_t)1; |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 85 | const sighandler_t SIG_ERR = (sighandler_t)-1; |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 86 | const int SA_SIGINFO = 4; |
| 87 | const int SIG_SETMASK = 2; |
| 88 | |
| 89 | static sigaction_t sigactions[kSigCount]; |
| 90 | |
| 91 | static unsigned g_thread_finalize_key; |
| 92 | |
| 93 | static void process_pending_signals(ThreadState *thr); |
| 94 | |
| 95 | class ScopedInterceptor { |
| 96 | public: |
| 97 | ScopedInterceptor(ThreadState *thr, const char *fname, uptr pc) |
| 98 | : thr_(thr) |
| 99 | , in_rtl_(thr->in_rtl) { |
| 100 | if (thr_->in_rtl == 0) { |
| 101 | Initialize(thr); |
| 102 | FuncEntry(thr, pc); |
| 103 | thr_->in_rtl++; |
| 104 | DPrintf("#%d: intercept %s()\n", thr_->tid, fname); |
| 105 | } else { |
| 106 | thr_->in_rtl++; |
| 107 | } |
| 108 | } |
| 109 | |
| 110 | ~ScopedInterceptor() { |
| 111 | thr_->in_rtl--; |
| 112 | if (thr_->in_rtl == 0) { |
| 113 | FuncExit(thr_); |
| 114 | process_pending_signals(thr_); |
| 115 | } |
| 116 | CHECK_EQ(in_rtl_, thr_->in_rtl); |
| 117 | } |
| 118 | |
| 119 | private: |
| 120 | ThreadState *const thr_; |
| 121 | const int in_rtl_; |
| 122 | }; |
| 123 | |
| 124 | #define SCOPED_INTERCEPTOR_RAW(func, ...) \ |
| 125 | ThreadState *thr = cur_thread(); \ |
| 126 | StatInc(thr, StatInterceptor); \ |
| 127 | StatInc(thr, StatInt_##func); \ |
| 128 | ScopedInterceptor si(thr, #func, \ |
| 129 | (__tsan::uptr)__builtin_return_address(0)); \ |
| 130 | const uptr pc = (uptr)&func; \ |
| 131 | (void)pc; \ |
| 132 | /**/ |
| 133 | |
| 134 | #define SCOPED_TSAN_INTERCEPTOR(func, ...) \ |
| 135 | SCOPED_INTERCEPTOR_RAW(func, __VA_ARGS__); \ |
| 136 | if (thr->in_rtl > 1) \ |
| 137 | return REAL(func)(__VA_ARGS__); \ |
| 138 | /**/ |
| 139 | |
| 140 | #define TSAN_INTERCEPTOR(ret, func, ...) INTERCEPTOR(ret, func, __VA_ARGS__) |
| 141 | #define TSAN_INTERCEPT(func) \ |
| 142 | if (!INTERCEPT_FUNCTION(func)) \ |
| 143 | Printf("ThreadSanitizer: failed to intercept '" #func "' function\n"); \ |
| 144 | /**/ |
| 145 | |
| 146 | class AtExitContext { |
| 147 | public: |
| 148 | AtExitContext() |
| 149 | : mtx_(MutexTypeAtExit, StatMtxAtExit) |
| 150 | , pos_() { |
| 151 | } |
| 152 | |
| 153 | typedef void(*atexit_t)(); |
| 154 | |
| 155 | int atexit(ThreadState *thr, uptr pc, atexit_t f) { |
| 156 | Lock l(&mtx_); |
| 157 | if (pos_ == kMaxAtExit) |
| 158 | return 1; |
| 159 | Release(thr, pc, (uptr)this); |
| 160 | stack_[pos_] = f; |
| 161 | pos_++; |
| 162 | return 0; |
| 163 | } |
| 164 | |
| 165 | void exit(ThreadState *thr, uptr pc) { |
| 166 | CHECK_EQ(thr->in_rtl, 0); |
| 167 | for (;;) { |
| 168 | atexit_t f = 0; |
| 169 | { |
| 170 | Lock l(&mtx_); |
| 171 | if (pos_) { |
| 172 | pos_--; |
| 173 | f = stack_[pos_]; |
| 174 | ScopedInRtl in_rtl; |
| 175 | Acquire(thr, pc, (uptr)this); |
| 176 | } |
| 177 | } |
| 178 | if (f == 0) |
| 179 | break; |
| 180 | DPrintf("#%d: executing atexit func %p\n", thr->tid, f); |
| 181 | CHECK_EQ(thr->in_rtl, 0); |
| 182 | f(); |
| 183 | } |
| 184 | } |
| 185 | |
| 186 | private: |
| 187 | static const int kMaxAtExit = 128; |
| 188 | Mutex mtx_; |
| 189 | atexit_t stack_[kMaxAtExit]; |
| 190 | int pos_; |
| 191 | }; |
| 192 | |
| 193 | static AtExitContext *atexit_ctx; |
| 194 | |
| 195 | static void finalize(void *arg) { |
| 196 | ThreadState * thr = cur_thread(); |
| 197 | uptr pc = 0; |
| 198 | atexit_ctx->exit(thr, pc); |
| 199 | { |
| 200 | ScopedInRtl in_rtl; |
| 201 | DestroyAndFree(atexit_ctx); |
| 202 | usleep(flags()->atexit_sleep_ms * 1000); |
| 203 | } |
| 204 | int status = Finalize(cur_thread()); |
Dmitry Vyukov | b7b6b1c | 2012-05-17 15:00:27 +0000 | [diff] [blame] | 205 | if (status) |
| 206 | _exit(status); |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 207 | } |
| 208 | |
| 209 | TSAN_INTERCEPTOR(int, atexit, void (*f)()) { |
| 210 | SCOPED_TSAN_INTERCEPTOR(atexit, f); |
| 211 | return atexit_ctx->atexit(thr, pc, f); |
| 212 | return 0; |
| 213 | } |
| 214 | |
| 215 | static uptr fd2addr(int fd) { |
| 216 | (void)fd; |
| 217 | static u64 addr; |
| 218 | return (uptr)&addr; |
| 219 | } |
| 220 | |
| 221 | static uptr epollfd2addr(int fd) { |
| 222 | (void)fd; |
| 223 | static u64 addr; |
| 224 | return (uptr)&addr; |
| 225 | } |
| 226 | |
| 227 | static uptr file2addr(char *path) { |
| 228 | (void)path; |
| 229 | static u64 addr; |
| 230 | return (uptr)&addr; |
| 231 | } |
| 232 | |
| 233 | static uptr dir2addr(char *path) { |
| 234 | (void)path; |
| 235 | static u64 addr; |
| 236 | return (uptr)&addr; |
| 237 | } |
| 238 | |
| 239 | TSAN_INTERCEPTOR(void*, malloc, uptr size) { |
| 240 | SCOPED_INTERCEPTOR_RAW(malloc, size); |
| 241 | return user_alloc(thr, pc, size); |
| 242 | } |
| 243 | |
| 244 | TSAN_INTERCEPTOR(void*, calloc, uptr size, uptr n) { |
| 245 | SCOPED_INTERCEPTOR_RAW(calloc, size, n); |
| 246 | void *p = user_alloc(thr, pc, n * size); |
| 247 | internal_memset(p, 0, n * size); |
| 248 | return p; |
| 249 | } |
| 250 | |
| 251 | TSAN_INTERCEPTOR(void*, realloc, void *p, uptr size) { |
| 252 | SCOPED_INTERCEPTOR_RAW(realloc, p, size); |
| 253 | return user_realloc(thr, pc, p, size); |
| 254 | } |
| 255 | |
| 256 | TSAN_INTERCEPTOR(void, free, void *p) { |
| 257 | if (p == 0) |
| 258 | return; |
| 259 | SCOPED_INTERCEPTOR_RAW(free, p); |
| 260 | user_free(thr, pc, p); |
| 261 | } |
| 262 | |
| 263 | TSAN_INTERCEPTOR(void, cfree, void *p) { |
| 264 | if (p == 0) |
| 265 | return; |
| 266 | SCOPED_INTERCEPTOR_RAW(cfree, p); |
| 267 | user_free(thr, pc, p); |
| 268 | } |
| 269 | |
| 270 | TSAN_INTERCEPTOR(uptr, strlen, const void *s) { |
| 271 | SCOPED_TSAN_INTERCEPTOR(strlen, s); |
| 272 | uptr len = REAL(strlen)(s); |
| 273 | MemoryAccessRange(thr, pc, (uptr)s, len + 1, false); |
| 274 | return len; |
| 275 | } |
| 276 | |
| 277 | TSAN_INTERCEPTOR(void*, memset, void *dst, int v, uptr size) { |
| 278 | SCOPED_TSAN_INTERCEPTOR(memset, dst, v, size); |
| 279 | MemoryAccessRange(thr, pc, (uptr)dst, size, true); |
| 280 | return REAL(memset)(dst, v, size); |
| 281 | } |
| 282 | |
| 283 | TSAN_INTERCEPTOR(void*, memcpy, void *dst, const void *src, uptr size) { |
| 284 | SCOPED_TSAN_INTERCEPTOR(memcpy, dst, src, size); |
| 285 | MemoryAccessRange(thr, pc, (uptr)dst, size, true); |
| 286 | MemoryAccessRange(thr, pc, (uptr)src, size, false); |
| 287 | return REAL(memcpy)(dst, src, size); |
| 288 | } |
| 289 | |
| 290 | TSAN_INTERCEPTOR(int, strcmp, const char *s1, const char *s2) { |
| 291 | SCOPED_TSAN_INTERCEPTOR(strcmp, s1, s2); |
| 292 | uptr len = 0; |
| 293 | for (; s1[len] && s2[len]; len++) { |
| 294 | if (s1[len] != s2[len]) |
| 295 | break; |
| 296 | } |
| 297 | MemoryAccessRange(thr, pc, (uptr)s1, len + 1, false); |
| 298 | MemoryAccessRange(thr, pc, (uptr)s2, len + 1, false); |
| 299 | return s1[len] - s2[len]; |
| 300 | } |
| 301 | |
| 302 | TSAN_INTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr n) { |
| 303 | SCOPED_TSAN_INTERCEPTOR(strncmp, s1, s2, n); |
| 304 | uptr len = 0; |
Dmitry Vyukov | 7d38634 | 2012-05-18 08:53:16 +0000 | [diff] [blame] | 305 | for (; len < n && s1[len] && s2[len]; len++) { |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 306 | if (s1[len] != s2[len]) |
| 307 | break; |
| 308 | } |
| 309 | MemoryAccessRange(thr, pc, (uptr)s1, len < n ? len + 1 : n, false); |
| 310 | MemoryAccessRange(thr, pc, (uptr)s2, len < n ? len + 1 : n, false); |
| 311 | return len == n ? 0 : s1[len] - s2[len]; |
| 312 | } |
| 313 | |
| 314 | TSAN_INTERCEPTOR(void*, memchr, void *s, int c, uptr n) { |
| 315 | SCOPED_TSAN_INTERCEPTOR(memchr, s, c, n); |
| 316 | void *res = REAL(memchr)(s, c, n); |
| 317 | uptr len = res ? (char*)res - (char*)s + 1 : n; |
| 318 | MemoryAccessRange(thr, pc, (uptr)s, len, false); |
| 319 | return res; |
| 320 | } |
| 321 | |
| 322 | TSAN_INTERCEPTOR(void*, memrchr, char *s, int c, uptr n) { |
| 323 | SCOPED_TSAN_INTERCEPTOR(memrchr, s, c, n); |
| 324 | MemoryAccessRange(thr, pc, (uptr)s, n, false); |
| 325 | return REAL(memrchr)(s, c, n); |
| 326 | } |
| 327 | |
| 328 | TSAN_INTERCEPTOR(void*, memmove, void *dst, void *src, uptr n) { |
| 329 | SCOPED_TSAN_INTERCEPTOR(memmove, dst, src, n); |
| 330 | MemoryAccessRange(thr, pc, (uptr)dst, n, true); |
| 331 | MemoryAccessRange(thr, pc, (uptr)src, n, false); |
| 332 | return REAL(memmove)(dst, src, n); |
| 333 | } |
| 334 | |
| 335 | TSAN_INTERCEPTOR(int, memcmp, const void *s1, const void *s2, uptr n) { |
| 336 | SCOPED_TSAN_INTERCEPTOR(memcmp, s1, s2, n); |
| 337 | int res = 0; |
| 338 | uptr len = 0; |
| 339 | for (; len < n; len++) { |
| 340 | if ((res = ((unsigned char*)s1)[len] - ((unsigned char*)s2)[len])) |
| 341 | break; |
| 342 | } |
| 343 | MemoryAccessRange(thr, pc, (uptr)s1, len < n ? len + 1 : n, false); |
| 344 | MemoryAccessRange(thr, pc, (uptr)s2, len < n ? len + 1 : n, false); |
| 345 | return res; |
| 346 | } |
| 347 | |
| 348 | TSAN_INTERCEPTOR(void*, strchr, void *s, int c) { |
| 349 | SCOPED_TSAN_INTERCEPTOR(strchr, s, c); |
| 350 | void *res = REAL(strchr)(s, c); |
| 351 | uptr len = res ? (char*)res - (char*)s + 1 : REAL(strlen)(s) + 1; |
| 352 | MemoryAccessRange(thr, pc, (uptr)s, len, false); |
| 353 | return res; |
| 354 | } |
| 355 | |
| 356 | TSAN_INTERCEPTOR(void*, strchrnul, void *s, int c) { |
| 357 | SCOPED_TSAN_INTERCEPTOR(strchrnul, s, c); |
| 358 | void *res = REAL(strchrnul)(s, c); |
| 359 | uptr len = (char*)res - (char*)s + 1; |
| 360 | MemoryAccessRange(thr, pc, (uptr)s, len, false); |
| 361 | return res; |
| 362 | } |
| 363 | |
| 364 | TSAN_INTERCEPTOR(void*, strrchr, void *s, int c) { |
| 365 | SCOPED_TSAN_INTERCEPTOR(strrchr, s, c); |
| 366 | MemoryAccessRange(thr, pc, (uptr)s, REAL(strlen)(s) + 1, false); |
| 367 | return REAL(strrchr)(s, c); |
| 368 | } |
| 369 | |
| 370 | TSAN_INTERCEPTOR(void*, strcpy, void *dst, const void *src) { // NOLINT |
| 371 | SCOPED_TSAN_INTERCEPTOR(strcpy, dst, src); // NOLINT |
| 372 | uptr srclen = REAL(strlen)(src); |
| 373 | MemoryAccessRange(thr, pc, (uptr)dst, srclen + 1, true); |
| 374 | MemoryAccessRange(thr, pc, (uptr)src, srclen + 1, false); |
| 375 | return REAL(strcpy)(dst, src); // NOLINT |
| 376 | } |
| 377 | |
| 378 | TSAN_INTERCEPTOR(void*, strncpy, void *dst, void *src, uptr n) { |
| 379 | SCOPED_TSAN_INTERCEPTOR(strncpy, dst, src, n); |
| 380 | uptr srclen = REAL(strlen)(src); |
| 381 | MemoryAccessRange(thr, pc, (uptr)dst, n, true); |
| 382 | MemoryAccessRange(thr, pc, (uptr)src, min(srclen + 1, n), false); |
| 383 | return REAL(strncpy)(dst, src, n); |
| 384 | } |
| 385 | |
| 386 | TSAN_INTERCEPTOR(const char*, strstr, const char *s1, const char *s2) { |
| 387 | SCOPED_TSAN_INTERCEPTOR(strstr, s1, s2); |
| 388 | const char *res = REAL(strstr)(s1, s2); |
| 389 | uptr len1 = REAL(strlen)(s1); |
| 390 | uptr len2 = REAL(strlen)(s2); |
| 391 | MemoryAccessRange(thr, pc, (uptr)s1, len1 + 1, false); |
| 392 | MemoryAccessRange(thr, pc, (uptr)s2, len2 + 1, false); |
| 393 | return res; |
| 394 | } |
| 395 | |
| 396 | static bool fix_mmap_addr(void **addr, long_t sz, int flags) { |
| 397 | if (*addr) { |
| 398 | if (!IsAppMem((uptr)*addr) || !IsAppMem((uptr)*addr + sz - 1)) { |
| 399 | if (flags & MAP_FIXED) { |
| 400 | *__errno_location() = EINVAL; |
| 401 | return false; |
| 402 | } else { |
| 403 | *addr = 0; |
| 404 | } |
| 405 | } |
| 406 | } |
| 407 | return true; |
| 408 | } |
| 409 | |
| 410 | TSAN_INTERCEPTOR(void*, mmap, void *addr, long_t sz, int prot, |
| 411 | int flags, int fd, unsigned off) { |
| 412 | SCOPED_TSAN_INTERCEPTOR(mmap, addr, sz, prot, flags, fd, off); |
| 413 | if (!fix_mmap_addr(&addr, sz, flags)) |
| 414 | return MAP_FAILED; |
| 415 | void *res = REAL(mmap)(addr, sz, prot, flags, fd, off); |
| 416 | if (res != MAP_FAILED) { |
| 417 | MemoryResetRange(thr, pc, (uptr)res, sz); |
| 418 | } |
| 419 | return res; |
| 420 | } |
| 421 | |
| 422 | TSAN_INTERCEPTOR(void*, mmap64, void *addr, long_t sz, int prot, |
| 423 | int flags, int fd, u64 off) { |
| 424 | SCOPED_TSAN_INTERCEPTOR(mmap64, addr, sz, prot, flags, fd, off); |
| 425 | if (!fix_mmap_addr(&addr, sz, flags)) |
| 426 | return MAP_FAILED; |
| 427 | void *res = REAL(mmap64)(addr, sz, prot, flags, fd, off); |
| 428 | if (res != MAP_FAILED) { |
| 429 | MemoryResetRange(thr, pc, (uptr)res, sz); |
| 430 | } |
| 431 | return res; |
| 432 | } |
| 433 | |
| 434 | TSAN_INTERCEPTOR(int, munmap, void *addr, long_t sz) { |
| 435 | SCOPED_TSAN_INTERCEPTOR(munmap, addr, sz); |
| 436 | int res = REAL(munmap)(addr, sz); |
| 437 | return res; |
| 438 | } |
| 439 | |
| 440 | #ifdef __LP64__ |
| 441 | |
| 442 | // void *operator new(size_t) |
| 443 | TSAN_INTERCEPTOR(void*, _Znwm, uptr sz) { |
| 444 | SCOPED_TSAN_INTERCEPTOR(_Znwm, sz); |
| 445 | return user_alloc(thr, pc, sz); |
| 446 | } |
| 447 | |
| 448 | // void *operator new(size_t, nothrow_t) |
| 449 | TSAN_INTERCEPTOR(void*, _ZnwmRKSt9nothrow_t, uptr sz) { |
| 450 | SCOPED_TSAN_INTERCEPTOR(_ZnwmRKSt9nothrow_t, sz); |
| 451 | return user_alloc(thr, pc, sz); |
| 452 | } |
| 453 | |
| 454 | // void *operator new[](size_t) |
| 455 | TSAN_INTERCEPTOR(void*, _Znam, uptr sz) { |
| 456 | SCOPED_TSAN_INTERCEPTOR(_Znam, sz); |
| 457 | return user_alloc(thr, pc, sz); |
| 458 | } |
| 459 | |
| 460 | // void *operator new[](size_t, nothrow_t) |
| 461 | TSAN_INTERCEPTOR(void*, _ZnamRKSt9nothrow_t, uptr sz) { |
| 462 | SCOPED_TSAN_INTERCEPTOR(_ZnamRKSt9nothrow_t, sz); |
| 463 | return user_alloc(thr, pc, sz); |
| 464 | } |
| 465 | |
| 466 | #else |
| 467 | #error "Not implemented" |
| 468 | #endif |
| 469 | |
| 470 | // void operator delete(void*) |
| 471 | TSAN_INTERCEPTOR(void, _ZdlPv, void *p) { |
| 472 | if (p == 0) |
| 473 | return; |
| 474 | SCOPED_TSAN_INTERCEPTOR(_ZdlPv, p); |
| 475 | user_free(thr, pc, p); |
| 476 | } |
| 477 | |
| 478 | // void operator delete(void*, nothrow_t) |
| 479 | TSAN_INTERCEPTOR(void, _ZdlPvRKSt9nothrow_t, void *p) { |
| 480 | if (p == 0) |
| 481 | return; |
| 482 | SCOPED_TSAN_INTERCEPTOR(_ZdlPvRKSt9nothrow_t, p); |
| 483 | user_free(thr, pc, p); |
| 484 | } |
| 485 | |
| 486 | // void operator delete[](void*) |
| 487 | TSAN_INTERCEPTOR(void, _ZdaPv, void *p) { |
| 488 | if (p == 0) |
| 489 | return; |
| 490 | SCOPED_TSAN_INTERCEPTOR(_ZdaPv, p); |
| 491 | user_free(thr, pc, p); |
| 492 | } |
| 493 | |
| 494 | // void operator delete[](void*, nothrow_t) |
| 495 | TSAN_INTERCEPTOR(void, _ZdaPvRKSt9nothrow_t, void *p) { |
| 496 | if (p == 0) |
| 497 | return; |
| 498 | SCOPED_TSAN_INTERCEPTOR(_ZdaPvRKSt9nothrow_t, p); |
| 499 | user_free(thr, pc, p); |
| 500 | } |
| 501 | |
| 502 | TSAN_INTERCEPTOR(void*, memalign, uptr align, uptr sz) { |
| 503 | SCOPED_TSAN_INTERCEPTOR(memalign, align, sz); |
| 504 | return user_alloc_aligned(thr, pc, sz, align); |
| 505 | } |
| 506 | |
| 507 | TSAN_INTERCEPTOR(void*, valloc, uptr sz) { |
| 508 | SCOPED_TSAN_INTERCEPTOR(valloc, sz); |
| 509 | return user_alloc_aligned(thr, pc, sz, kPageSize); |
| 510 | } |
| 511 | |
| 512 | TSAN_INTERCEPTOR(void*, pvalloc, uptr sz) { |
| 513 | SCOPED_TSAN_INTERCEPTOR(pvalloc, sz); |
| 514 | sz = RoundUp(sz, kPageSize); |
| 515 | return user_alloc_aligned(thr, pc, sz, kPageSize); |
| 516 | } |
| 517 | |
| 518 | TSAN_INTERCEPTOR(int, posix_memalign, void **memptr, uptr align, uptr sz) { |
| 519 | SCOPED_TSAN_INTERCEPTOR(posix_memalign, memptr, align, sz); |
| 520 | *memptr = user_alloc_aligned(thr, pc, sz, align); |
| 521 | return 0; |
| 522 | } |
| 523 | |
| 524 | // Used in thread-safe function static initialization. |
| 525 | TSAN_INTERCEPTOR(int, __cxa_guard_acquire, char *m) { |
| 526 | SCOPED_TSAN_INTERCEPTOR(__cxa_guard_acquire, m); |
| 527 | int res = REAL(__cxa_guard_acquire)(m); |
| 528 | if (res) { |
| 529 | // This thread does the init. |
| 530 | } else { |
| 531 | Acquire(thr, pc, (uptr)m); |
| 532 | } |
| 533 | return res; |
| 534 | } |
| 535 | |
| 536 | TSAN_INTERCEPTOR(void, __cxa_guard_release, char *m) { |
| 537 | SCOPED_TSAN_INTERCEPTOR(__cxa_guard_release, m); |
| 538 | Release(thr, pc, (uptr)m); |
| 539 | REAL(__cxa_guard_release)(m); |
| 540 | } |
| 541 | |
| 542 | static void thread_finalize(void *v) { |
| 543 | uptr iter = (uptr)v; |
| 544 | if (iter > 1) { |
| 545 | if (pthread_setspecific(g_thread_finalize_key, (void*)(iter - 1))) { |
| 546 | Printf("ThreadSanitizer: failed to set thread key\n"); |
| 547 | Die(); |
| 548 | } |
| 549 | return; |
| 550 | } |
| 551 | { |
| 552 | ScopedInRtl in_rtl; |
| 553 | ThreadFinish(cur_thread()); |
| 554 | } |
| 555 | } |
| 556 | |
| 557 | |
| 558 | struct ThreadParam { |
| 559 | void* (*callback)(void *arg); |
| 560 | void *param; |
| 561 | atomic_uintptr_t tid; |
| 562 | }; |
| 563 | |
| 564 | extern "C" void *__tsan_thread_start_func(void *arg) { |
| 565 | ThreadParam *p = (ThreadParam*)arg; |
| 566 | void* (*callback)(void *arg) = p->callback; |
| 567 | void *param = p->param; |
| 568 | int tid = 0; |
| 569 | { |
| 570 | ThreadState *thr = cur_thread(); |
| 571 | ScopedInRtl in_rtl; |
| 572 | if (pthread_setspecific(g_thread_finalize_key, (void*)4)) { |
| 573 | Printf("ThreadSanitizer: failed to set thread key\n"); |
| 574 | Die(); |
| 575 | } |
| 576 | while ((tid = atomic_load(&p->tid, memory_order_acquire)) == 0) |
| 577 | pthread_yield(); |
| 578 | atomic_store(&p->tid, 0, memory_order_release); |
| 579 | ThreadStart(thr, tid); |
| 580 | CHECK_EQ(thr->in_rtl, 1); |
| 581 | } |
| 582 | void *res = callback(param); |
| 583 | // Prevent the callback from being tail called, |
| 584 | // it mixes up stack traces. |
| 585 | volatile int foo = 42; |
| 586 | foo++; |
| 587 | return res; |
| 588 | } |
| 589 | |
| 590 | TSAN_INTERCEPTOR(int, pthread_create, |
| 591 | void *th, void *attr, void *(*callback)(void*), void * param) { |
| 592 | SCOPED_TSAN_INTERCEPTOR(pthread_create, th, attr, callback, param); |
| 593 | pthread_attr_t myattr; |
| 594 | if (attr == 0) { |
| 595 | pthread_attr_init(&myattr); |
| 596 | attr = &myattr; |
| 597 | } |
| 598 | int detached = 0; |
| 599 | pthread_attr_getdetachstate(attr, &detached); |
| 600 | uptr stacksize = 0; |
| 601 | pthread_attr_getstacksize(attr, &stacksize); |
| 602 | // We place the huge ThreadState object into TLS, account for that. |
| 603 | const uptr minstacksize = GetTlsSize() + 128*1024; |
| 604 | if (stacksize < minstacksize) { |
| 605 | DPrintf("ThreadSanitizer: stacksize %lu->%lu\n", stacksize, minstacksize); |
| 606 | pthread_attr_setstacksize(attr, minstacksize); |
| 607 | } |
| 608 | ThreadParam p; |
| 609 | p.callback = callback; |
| 610 | p.param = param; |
| 611 | atomic_store(&p.tid, 0, memory_order_relaxed); |
| 612 | int res = REAL(pthread_create)(th, attr, __tsan_thread_start_func, &p); |
| 613 | if (res == 0) { |
| 614 | int tid = ThreadCreate(cur_thread(), pc, *(uptr*)th, detached); |
| 615 | CHECK_NE(tid, 0); |
| 616 | atomic_store(&p.tid, tid, memory_order_release); |
| 617 | while (atomic_load(&p.tid, memory_order_acquire) != 0) |
| 618 | pthread_yield(); |
| 619 | } |
| 620 | if (attr == &myattr) |
| 621 | pthread_attr_destroy(&myattr); |
| 622 | return res; |
| 623 | } |
| 624 | |
| 625 | TSAN_INTERCEPTOR(int, pthread_join, void *th, void **ret) { |
| 626 | SCOPED_TSAN_INTERCEPTOR(pthread_join, th, ret); |
| 627 | int tid = ThreadTid(thr, pc, (uptr)th); |
| 628 | int res = REAL(pthread_join)(th, ret); |
| 629 | if (res == 0) { |
| 630 | ThreadJoin(cur_thread(), pc, tid); |
| 631 | } |
| 632 | return res; |
| 633 | } |
| 634 | |
| 635 | TSAN_INTERCEPTOR(int, pthread_detach, void *th) { |
| 636 | SCOPED_TSAN_INTERCEPTOR(pthread_detach, th); |
| 637 | int tid = ThreadTid(thr, pc, (uptr)th); |
| 638 | int res = REAL(pthread_detach)(th); |
| 639 | if (res == 0) { |
| 640 | ThreadDetach(cur_thread(), pc, tid); |
| 641 | } |
| 642 | return res; |
| 643 | } |
| 644 | |
| 645 | TSAN_INTERCEPTOR(int, pthread_mutex_init, void *m, void *a) { |
| 646 | SCOPED_TSAN_INTERCEPTOR(pthread_mutex_init, m, a); |
| 647 | int res = REAL(pthread_mutex_init)(m, a); |
| 648 | if (res == 0) { |
| 649 | bool recursive = false; |
| 650 | if (a) { |
| 651 | int type = 0; |
| 652 | if (pthread_mutexattr_gettype(a, &type) == 0) |
| 653 | recursive = (type == PTHREAD_MUTEX_RECURSIVE |
| 654 | || type == PTHREAD_MUTEX_RECURSIVE_NP); |
| 655 | } |
| 656 | MutexCreate(cur_thread(), pc, (uptr)m, false, recursive); |
| 657 | } |
| 658 | return res; |
| 659 | } |
| 660 | |
| 661 | TSAN_INTERCEPTOR(int, pthread_mutex_destroy, void *m) { |
| 662 | SCOPED_TSAN_INTERCEPTOR(pthread_mutex_destroy, m); |
| 663 | int res = REAL(pthread_mutex_destroy)(m); |
| 664 | if (res == 0 || res == EBUSY) { |
| 665 | MutexDestroy(cur_thread(), pc, (uptr)m); |
| 666 | } |
| 667 | return res; |
| 668 | } |
| 669 | |
| 670 | TSAN_INTERCEPTOR(int, pthread_mutex_lock, void *m) { |
| 671 | SCOPED_TSAN_INTERCEPTOR(pthread_mutex_lock, m); |
| 672 | int res = REAL(pthread_mutex_lock)(m); |
| 673 | if (res == 0) { |
| 674 | MutexLock(cur_thread(), pc, (uptr)m); |
| 675 | } |
| 676 | return res; |
| 677 | } |
| 678 | |
| 679 | TSAN_INTERCEPTOR(int, pthread_mutex_trylock, void *m) { |
| 680 | SCOPED_TSAN_INTERCEPTOR(pthread_mutex_trylock, m); |
| 681 | int res = REAL(pthread_mutex_trylock)(m); |
| 682 | if (res == 0) { |
| 683 | MutexLock(cur_thread(), pc, (uptr)m); |
| 684 | } |
| 685 | return res; |
| 686 | } |
| 687 | |
| 688 | TSAN_INTERCEPTOR(int, pthread_mutex_timedlock, void *m, void *abstime) { |
| 689 | SCOPED_TSAN_INTERCEPTOR(pthread_mutex_timedlock, m, abstime); |
| 690 | int res = REAL(pthread_mutex_timedlock)(m, abstime); |
| 691 | if (res == 0) { |
| 692 | MutexLock(cur_thread(), pc, (uptr)m); |
| 693 | } |
| 694 | return res; |
| 695 | } |
| 696 | |
| 697 | TSAN_INTERCEPTOR(int, pthread_mutex_unlock, void *m) { |
| 698 | SCOPED_TSAN_INTERCEPTOR(pthread_mutex_unlock, m); |
| 699 | MutexUnlock(cur_thread(), pc, (uptr)m); |
| 700 | int res = REAL(pthread_mutex_unlock)(m); |
| 701 | return res; |
| 702 | } |
| 703 | |
| 704 | TSAN_INTERCEPTOR(int, pthread_spin_init, void *m, int pshared) { |
| 705 | SCOPED_TSAN_INTERCEPTOR(pthread_spin_init, m, pshared); |
| 706 | int res = REAL(pthread_spin_init)(m, pshared); |
| 707 | if (res == 0) { |
| 708 | MutexCreate(cur_thread(), pc, (uptr)m, false, false); |
| 709 | } |
| 710 | return res; |
| 711 | } |
| 712 | |
| 713 | TSAN_INTERCEPTOR(int, pthread_spin_destroy, void *m) { |
| 714 | SCOPED_TSAN_INTERCEPTOR(pthread_spin_destroy, m); |
| 715 | int res = REAL(pthread_spin_destroy)(m); |
| 716 | if (res == 0) { |
| 717 | MutexDestroy(cur_thread(), pc, (uptr)m); |
| 718 | } |
| 719 | return res; |
| 720 | } |
| 721 | |
| 722 | TSAN_INTERCEPTOR(int, pthread_spin_lock, void *m) { |
| 723 | SCOPED_TSAN_INTERCEPTOR(pthread_spin_lock, m); |
| 724 | int res = REAL(pthread_spin_lock)(m); |
| 725 | if (res == 0) { |
| 726 | MutexLock(cur_thread(), pc, (uptr)m); |
| 727 | } |
| 728 | return res; |
| 729 | } |
| 730 | |
| 731 | TSAN_INTERCEPTOR(int, pthread_spin_trylock, void *m) { |
| 732 | SCOPED_TSAN_INTERCEPTOR(pthread_spin_trylock, m); |
| 733 | int res = REAL(pthread_spin_trylock)(m); |
| 734 | if (res == 0) { |
| 735 | MutexLock(cur_thread(), pc, (uptr)m); |
| 736 | } |
| 737 | return res; |
| 738 | } |
| 739 | |
| 740 | TSAN_INTERCEPTOR(int, pthread_spin_unlock, void *m) { |
| 741 | SCOPED_TSAN_INTERCEPTOR(pthread_spin_unlock, m); |
| 742 | MutexUnlock(cur_thread(), pc, (uptr)m); |
| 743 | int res = REAL(pthread_spin_unlock)(m); |
| 744 | return res; |
| 745 | } |
| 746 | |
| 747 | TSAN_INTERCEPTOR(int, pthread_rwlock_init, void *m, void *a) { |
| 748 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_init, m, a); |
| 749 | int res = REAL(pthread_rwlock_init)(m, a); |
| 750 | if (res == 0) { |
| 751 | MutexCreate(cur_thread(), pc, (uptr)m, true, false); |
| 752 | } |
| 753 | return res; |
| 754 | } |
| 755 | |
| 756 | TSAN_INTERCEPTOR(int, pthread_rwlock_destroy, void *m) { |
| 757 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_destroy, m); |
| 758 | int res = REAL(pthread_rwlock_destroy)(m); |
| 759 | if (res == 0) { |
| 760 | MutexDestroy(cur_thread(), pc, (uptr)m); |
| 761 | } |
| 762 | return res; |
| 763 | } |
| 764 | |
| 765 | TSAN_INTERCEPTOR(int, pthread_rwlock_rdlock, void *m) { |
| 766 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_rdlock, m); |
| 767 | int res = REAL(pthread_rwlock_rdlock)(m); |
| 768 | if (res == 0) { |
| 769 | MutexReadLock(cur_thread(), pc, (uptr)m); |
| 770 | } |
| 771 | return res; |
| 772 | } |
| 773 | |
| 774 | TSAN_INTERCEPTOR(int, pthread_rwlock_tryrdlock, void *m) { |
| 775 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_tryrdlock, m); |
| 776 | int res = REAL(pthread_rwlock_tryrdlock)(m); |
| 777 | if (res == 0) { |
| 778 | MutexReadLock(cur_thread(), pc, (uptr)m); |
| 779 | } |
| 780 | return res; |
| 781 | } |
| 782 | |
| 783 | TSAN_INTERCEPTOR(int, pthread_rwlock_timedrdlock, void *m, void *abstime) { |
| 784 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_timedrdlock, m, abstime); |
| 785 | int res = REAL(pthread_rwlock_timedrdlock)(m, abstime); |
| 786 | if (res == 0) { |
| 787 | MutexReadLock(cur_thread(), pc, (uptr)m); |
| 788 | } |
| 789 | return res; |
| 790 | } |
| 791 | |
| 792 | TSAN_INTERCEPTOR(int, pthread_rwlock_wrlock, void *m) { |
| 793 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_wrlock, m); |
| 794 | int res = REAL(pthread_rwlock_wrlock)(m); |
| 795 | if (res == 0) { |
| 796 | MutexLock(cur_thread(), pc, (uptr)m); |
| 797 | } |
| 798 | return res; |
| 799 | } |
| 800 | |
| 801 | TSAN_INTERCEPTOR(int, pthread_rwlock_trywrlock, void *m) { |
| 802 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_trywrlock, m); |
| 803 | int res = REAL(pthread_rwlock_trywrlock)(m); |
| 804 | if (res == 0) { |
| 805 | MutexLock(cur_thread(), pc, (uptr)m); |
| 806 | } |
| 807 | return res; |
| 808 | } |
| 809 | |
| 810 | TSAN_INTERCEPTOR(int, pthread_rwlock_timedwrlock, void *m, void *abstime) { |
| 811 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_timedwrlock, m, abstime); |
| 812 | int res = REAL(pthread_rwlock_timedwrlock)(m, abstime); |
| 813 | if (res == 0) { |
| 814 | MutexLock(cur_thread(), pc, (uptr)m); |
| 815 | } |
| 816 | return res; |
| 817 | } |
| 818 | |
| 819 | TSAN_INTERCEPTOR(int, pthread_rwlock_unlock, void *m) { |
| 820 | SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_unlock, m); |
| 821 | MutexReadOrWriteUnlock(cur_thread(), pc, (uptr)m); |
| 822 | int res = REAL(pthread_rwlock_unlock)(m); |
| 823 | return res; |
| 824 | } |
| 825 | |
| 826 | TSAN_INTERCEPTOR(int, pthread_cond_init, void *c, void *a) { |
| 827 | SCOPED_TSAN_INTERCEPTOR(pthread_cond_init, c, a); |
| 828 | int res = REAL(pthread_cond_init)(c, a); |
| 829 | return res; |
| 830 | } |
| 831 | |
| 832 | TSAN_INTERCEPTOR(int, pthread_cond_destroy, void *c) { |
| 833 | SCOPED_TSAN_INTERCEPTOR(pthread_cond_destroy, c); |
| 834 | int res = REAL(pthread_cond_destroy)(c); |
| 835 | return res; |
| 836 | } |
| 837 | |
| 838 | TSAN_INTERCEPTOR(int, pthread_cond_signal, void *c) { |
| 839 | SCOPED_TSAN_INTERCEPTOR(pthread_cond_signal, c); |
| 840 | int res = REAL(pthread_cond_signal)(c); |
| 841 | return res; |
| 842 | } |
| 843 | |
| 844 | TSAN_INTERCEPTOR(int, pthread_cond_broadcast, void *c) { |
| 845 | SCOPED_TSAN_INTERCEPTOR(pthread_cond_broadcast, c); |
| 846 | int res = REAL(pthread_cond_broadcast)(c); |
| 847 | return res; |
| 848 | } |
| 849 | |
| 850 | TSAN_INTERCEPTOR(int, pthread_cond_wait, void *c, void *m) { |
| 851 | SCOPED_TSAN_INTERCEPTOR(pthread_cond_wait, c, m); |
| 852 | MutexUnlock(cur_thread(), pc, (uptr)m); |
| 853 | int res = REAL(pthread_cond_wait)(c, m); |
| 854 | MutexLock(cur_thread(), pc, (uptr)m); |
| 855 | return res; |
| 856 | } |
| 857 | |
| 858 | TSAN_INTERCEPTOR(int, pthread_cond_timedwait, void *c, void *m, void *abstime) { |
| 859 | SCOPED_TSAN_INTERCEPTOR(pthread_cond_timedwait, c, m, abstime); |
| 860 | MutexUnlock(cur_thread(), pc, (uptr)m); |
| 861 | int res = REAL(pthread_cond_timedwait)(c, m, abstime); |
| 862 | MutexLock(cur_thread(), pc, (uptr)m); |
| 863 | return res; |
| 864 | } |
| 865 | |
| 866 | TSAN_INTERCEPTOR(int, pthread_barrier_init, void *b, void *a, unsigned count) { |
| 867 | SCOPED_TSAN_INTERCEPTOR(pthread_barrier_init, b, a, count); |
| 868 | MemoryWrite1Byte(thr, pc, (uptr)b); |
| 869 | int res = REAL(pthread_barrier_init)(b, a, count); |
| 870 | return res; |
| 871 | } |
| 872 | |
| 873 | TSAN_INTERCEPTOR(int, pthread_barrier_destroy, void *b) { |
| 874 | SCOPED_TSAN_INTERCEPTOR(pthread_barrier_destroy, b); |
| 875 | MemoryWrite1Byte(thr, pc, (uptr)b); |
| 876 | int res = REAL(pthread_barrier_destroy)(b); |
| 877 | return res; |
| 878 | } |
| 879 | |
| 880 | TSAN_INTERCEPTOR(int, pthread_barrier_wait, void *b) { |
| 881 | SCOPED_TSAN_INTERCEPTOR(pthread_barrier_wait, b); |
| 882 | Release(cur_thread(), pc, (uptr)b); |
| 883 | MemoryRead1Byte(thr, pc, (uptr)b); |
| 884 | int res = REAL(pthread_barrier_wait)(b); |
| 885 | MemoryRead1Byte(thr, pc, (uptr)b); |
| 886 | if (res == 0 || res == PTHREAD_BARRIER_SERIAL_THREAD) { |
| 887 | Acquire(cur_thread(), pc, (uptr)b); |
| 888 | } |
| 889 | return res; |
| 890 | } |
| 891 | |
| 892 | TSAN_INTERCEPTOR(int, pthread_once, void *o, void (*f)()) { |
| 893 | SCOPED_TSAN_INTERCEPTOR(pthread_once, o, f); |
| 894 | if (o == 0 || f == 0) |
| 895 | return EINVAL; |
| 896 | atomic_uint32_t *a = static_cast<atomic_uint32_t*>(o); |
| 897 | u32 v = atomic_load(a, memory_order_acquire); |
| 898 | if (v == 0 && atomic_compare_exchange_strong(a, &v, 1, |
| 899 | memory_order_relaxed)) { |
| 900 | const int old_in_rtl = thr->in_rtl; |
| 901 | thr->in_rtl = 0; |
| 902 | (*f)(); |
| 903 | CHECK_EQ(thr->in_rtl, 0); |
| 904 | thr->in_rtl = old_in_rtl; |
| 905 | Release(cur_thread(), pc, (uptr)o); |
| 906 | atomic_store(a, 2, memory_order_release); |
| 907 | } else { |
| 908 | while (v != 2) { |
| 909 | pthread_yield(); |
| 910 | v = atomic_load(a, memory_order_acquire); |
| 911 | } |
| 912 | Acquire(cur_thread(), pc, (uptr)o); |
| 913 | } |
| 914 | return 0; |
| 915 | } |
| 916 | |
| 917 | TSAN_INTERCEPTOR(int, sem_init, void *s, int pshared, unsigned value) { |
| 918 | SCOPED_TSAN_INTERCEPTOR(sem_init, s, pshared, value); |
| 919 | int res = REAL(sem_init)(s, pshared, value); |
| 920 | return res; |
| 921 | } |
| 922 | |
| 923 | TSAN_INTERCEPTOR(int, sem_destroy, void *s) { |
| 924 | SCOPED_TSAN_INTERCEPTOR(sem_destroy, s); |
| 925 | int res = REAL(sem_destroy)(s); |
| 926 | return res; |
| 927 | } |
| 928 | |
| 929 | TSAN_INTERCEPTOR(int, sem_wait, void *s) { |
| 930 | SCOPED_TSAN_INTERCEPTOR(sem_wait, s); |
| 931 | int res = REAL(sem_wait)(s); |
| 932 | if (res == 0) { |
| 933 | Acquire(cur_thread(), pc, (uptr)s); |
| 934 | } |
| 935 | return res; |
| 936 | } |
| 937 | |
| 938 | TSAN_INTERCEPTOR(int, sem_trywait, void *s) { |
| 939 | SCOPED_TSAN_INTERCEPTOR(sem_trywait, s); |
| 940 | int res = REAL(sem_trywait)(s); |
| 941 | if (res == 0) { |
| 942 | Acquire(cur_thread(), pc, (uptr)s); |
| 943 | } |
| 944 | return res; |
| 945 | } |
| 946 | |
| 947 | TSAN_INTERCEPTOR(int, sem_timedwait, void *s, void *abstime) { |
| 948 | SCOPED_TSAN_INTERCEPTOR(sem_timedwait, s, abstime); |
| 949 | int res = REAL(sem_timedwait)(s, abstime); |
| 950 | if (res == 0) { |
| 951 | Acquire(cur_thread(), pc, (uptr)s); |
| 952 | } |
| 953 | return res; |
| 954 | } |
| 955 | |
| 956 | TSAN_INTERCEPTOR(int, sem_post, void *s) { |
| 957 | SCOPED_TSAN_INTERCEPTOR(sem_post, s); |
| 958 | Release(cur_thread(), pc, (uptr)s); |
| 959 | int res = REAL(sem_post)(s); |
| 960 | return res; |
| 961 | } |
| 962 | |
| 963 | TSAN_INTERCEPTOR(int, sem_getvalue, void *s, int *sval) { |
| 964 | SCOPED_TSAN_INTERCEPTOR(sem_getvalue, s, sval); |
| 965 | int res = REAL(sem_getvalue)(s, sval); |
| 966 | if (res == 0) { |
| 967 | Acquire(cur_thread(), pc, (uptr)s); |
| 968 | } |
| 969 | return res; |
| 970 | } |
| 971 | |
| 972 | TSAN_INTERCEPTOR(long_t, read, int fd, void *buf, long_t sz) { |
| 973 | SCOPED_TSAN_INTERCEPTOR(read, fd, buf, sz); |
| 974 | int res = REAL(read)(fd, buf, sz); |
| 975 | if (res >= 0) { |
| 976 | Acquire(cur_thread(), pc, fd2addr(fd)); |
| 977 | } |
| 978 | return res; |
| 979 | } |
| 980 | |
| 981 | TSAN_INTERCEPTOR(long_t, pread, int fd, void *buf, long_t sz, unsigned off) { |
| 982 | SCOPED_TSAN_INTERCEPTOR(pread, fd, buf, sz, off); |
| 983 | int res = REAL(pread)(fd, buf, sz, off); |
| 984 | if (res >= 0) { |
| 985 | Acquire(cur_thread(), pc, fd2addr(fd)); |
| 986 | } |
| 987 | return res; |
| 988 | } |
| 989 | |
| 990 | TSAN_INTERCEPTOR(long_t, pread64, int fd, void *buf, long_t sz, u64 off) { |
| 991 | SCOPED_TSAN_INTERCEPTOR(pread64, fd, buf, sz, off); |
| 992 | int res = REAL(pread64)(fd, buf, sz, off); |
| 993 | if (res >= 0) { |
| 994 | Acquire(cur_thread(), pc, fd2addr(fd)); |
| 995 | } |
| 996 | return res; |
| 997 | } |
| 998 | |
| 999 | TSAN_INTERCEPTOR(long_t, readv, int fd, void *vec, int cnt) { |
| 1000 | SCOPED_TSAN_INTERCEPTOR(readv, fd, vec, cnt); |
| 1001 | int res = REAL(readv)(fd, vec, cnt); |
| 1002 | if (res >= 0) { |
| 1003 | Acquire(cur_thread(), pc, fd2addr(fd)); |
| 1004 | } |
| 1005 | return res; |
| 1006 | } |
| 1007 | |
| 1008 | TSAN_INTERCEPTOR(long_t, preadv64, int fd, void *vec, int cnt, u64 off) { |
| 1009 | SCOPED_TSAN_INTERCEPTOR(preadv64, fd, vec, cnt, off); |
| 1010 | int res = REAL(preadv64)(fd, vec, cnt, off); |
| 1011 | if (res >= 0) { |
| 1012 | Acquire(cur_thread(), pc, fd2addr(fd)); |
| 1013 | } |
| 1014 | return res; |
| 1015 | } |
| 1016 | |
| 1017 | TSAN_INTERCEPTOR(long_t, write, int fd, void *buf, long_t sz) { |
| 1018 | SCOPED_TSAN_INTERCEPTOR(write, fd, buf, sz); |
| 1019 | Release(cur_thread(), pc, fd2addr(fd)); |
| 1020 | int res = REAL(write)(fd, buf, sz); |
| 1021 | return res; |
| 1022 | } |
| 1023 | |
| 1024 | TSAN_INTERCEPTOR(long_t, pwrite, int fd, void *buf, long_t sz, unsigned off) { |
| 1025 | SCOPED_TSAN_INTERCEPTOR(pwrite, fd, buf, sz, off); |
| 1026 | Release(cur_thread(), pc, fd2addr(fd)); |
| 1027 | int res = REAL(pwrite)(fd, buf, sz, off); |
| 1028 | return res; |
| 1029 | } |
| 1030 | |
| 1031 | TSAN_INTERCEPTOR(long_t, pwrite64, int fd, void *buf, long_t sz, unsigned off) { |
| 1032 | SCOPED_TSAN_INTERCEPTOR(pwrite64, fd, buf, sz, off); |
| 1033 | Release(cur_thread(), pc, fd2addr(fd)); |
| 1034 | int res = REAL(pwrite64)(fd, buf, sz, off); |
| 1035 | return res; |
| 1036 | } |
| 1037 | |
| 1038 | TSAN_INTERCEPTOR(long_t, writev, int fd, void *vec, int cnt) { |
| 1039 | SCOPED_TSAN_INTERCEPTOR(writev, fd, vec, cnt); |
| 1040 | Release(cur_thread(), pc, fd2addr(fd)); |
| 1041 | int res = REAL(writev)(fd, vec, cnt); |
| 1042 | return res; |
| 1043 | } |
| 1044 | |
| 1045 | TSAN_INTERCEPTOR(long_t, pwritev64, int fd, void *vec, int cnt, u64 off) { |
| 1046 | SCOPED_TSAN_INTERCEPTOR(pwritev64, fd, vec, cnt, off); |
| 1047 | Release(cur_thread(), pc, fd2addr(fd)); |
| 1048 | int res = REAL(pwritev64)(fd, vec, cnt, off); |
| 1049 | return res; |
| 1050 | } |
| 1051 | |
| 1052 | TSAN_INTERCEPTOR(long_t, send, int fd, void *buf, long_t len, int flags) { |
| 1053 | SCOPED_TSAN_INTERCEPTOR(send, fd, buf, len, flags); |
| 1054 | Release(cur_thread(), pc, fd2addr(fd)); |
| 1055 | int res = REAL(send)(fd, buf, len, flags); |
| 1056 | return res; |
| 1057 | } |
| 1058 | |
| 1059 | TSAN_INTERCEPTOR(long_t, sendmsg, int fd, void *msg, int flags) { |
| 1060 | SCOPED_TSAN_INTERCEPTOR(sendmsg, fd, msg, flags); |
| 1061 | Release(cur_thread(), pc, fd2addr(fd)); |
| 1062 | int res = REAL(sendmsg)(fd, msg, flags); |
| 1063 | return res; |
| 1064 | } |
| 1065 | |
| 1066 | TSAN_INTERCEPTOR(long_t, recv, int fd, void *buf, long_t len, int flags) { |
| 1067 | SCOPED_TSAN_INTERCEPTOR(recv, fd, buf, len, flags); |
| 1068 | int res = REAL(recv)(fd, buf, len, flags); |
| 1069 | if (res >= 0) { |
| 1070 | Acquire(cur_thread(), pc, fd2addr(fd)); |
| 1071 | } |
| 1072 | return res; |
| 1073 | } |
| 1074 | |
| 1075 | TSAN_INTERCEPTOR(long_t, recvmsg, int fd, void *msg, int flags) { |
| 1076 | SCOPED_TSAN_INTERCEPTOR(recvmsg, fd, msg, flags); |
| 1077 | int res = REAL(recvmsg)(fd, msg, flags); |
| 1078 | if (res >= 0) { |
| 1079 | Acquire(cur_thread(), pc, fd2addr(fd)); |
| 1080 | } |
| 1081 | return res; |
| 1082 | } |
| 1083 | |
| 1084 | TSAN_INTERCEPTOR(int, unlink, char *path) { |
| 1085 | SCOPED_TSAN_INTERCEPTOR(unlink, path); |
| 1086 | Release(cur_thread(), pc, file2addr(path)); |
| 1087 | int res = REAL(unlink)(path); |
| 1088 | return res; |
| 1089 | } |
| 1090 | |
| 1091 | TSAN_INTERCEPTOR(void*, fopen, char *path, char *mode) { |
| 1092 | SCOPED_TSAN_INTERCEPTOR(fopen, path, mode); |
| 1093 | void *res = REAL(fopen)(path, mode); |
| 1094 | Acquire(cur_thread(), pc, file2addr(path)); |
| 1095 | return res; |
| 1096 | } |
| 1097 | |
| 1098 | TSAN_INTERCEPTOR(uptr, fread, void *ptr, uptr size, uptr nmemb, void *f) { |
| 1099 | SCOPED_TSAN_INTERCEPTOR(fread, ptr, size, nmemb, f); |
| 1100 | MemoryAccessRange(thr, pc, (uptr)ptr, size * nmemb, true); |
| 1101 | return REAL(fread)(ptr, size, nmemb, f); |
| 1102 | } |
| 1103 | |
| 1104 | TSAN_INTERCEPTOR(uptr, fwrite, const void *p, uptr size, uptr nmemb, void *f) { |
| 1105 | SCOPED_TSAN_INTERCEPTOR(fwrite, p, size, nmemb, f); |
| 1106 | MemoryAccessRange(thr, pc, (uptr)p, size * nmemb, false); |
| 1107 | return REAL(fwrite)(p, size, nmemb, f); |
| 1108 | } |
| 1109 | |
| 1110 | TSAN_INTERCEPTOR(int, puts, const char *s) { |
| 1111 | SCOPED_TSAN_INTERCEPTOR(puts, s); |
| 1112 | MemoryAccessRange(thr, pc, (uptr)s, REAL(strlen)(s), false); |
| 1113 | return REAL(puts)(s); |
| 1114 | } |
| 1115 | |
| 1116 | TSAN_INTERCEPTOR(int, rmdir, char *path) { |
| 1117 | SCOPED_TSAN_INTERCEPTOR(rmdir, path); |
| 1118 | Release(cur_thread(), pc, dir2addr(path)); |
| 1119 | int res = REAL(rmdir)(path); |
| 1120 | return res; |
| 1121 | } |
| 1122 | |
| 1123 | TSAN_INTERCEPTOR(void*, opendir, char *path) { |
| 1124 | SCOPED_TSAN_INTERCEPTOR(opendir, path); |
| 1125 | void *res = REAL(opendir)(path); |
| 1126 | Acquire(cur_thread(), pc, dir2addr(path)); |
| 1127 | return res; |
| 1128 | } |
| 1129 | |
| 1130 | TSAN_INTERCEPTOR(int, epoll_ctl, int epfd, int op, int fd, void *ev) { |
| 1131 | SCOPED_TSAN_INTERCEPTOR(epoll_ctl, epfd, op, fd, ev); |
| 1132 | if (op == EPOLL_CTL_ADD) { |
| 1133 | Release(cur_thread(), pc, epollfd2addr(epfd)); |
| 1134 | } |
| 1135 | int res = REAL(epoll_ctl)(epfd, op, fd, ev); |
| 1136 | return res; |
| 1137 | } |
| 1138 | |
| 1139 | TSAN_INTERCEPTOR(int, epoll_wait, int epfd, void *ev, int cnt, int timeout) { |
| 1140 | SCOPED_TSAN_INTERCEPTOR(epoll_wait, epfd, ev, cnt, timeout); |
| 1141 | int res = REAL(epoll_wait)(epfd, ev, cnt, timeout); |
| 1142 | if (res > 0) { |
| 1143 | Acquire(cur_thread(), pc, epollfd2addr(epfd)); |
| 1144 | } |
| 1145 | return res; |
| 1146 | } |
| 1147 | |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1148 | static void ALWAYS_INLINE rtl_generic_sighandler(bool sigact, int sig, |
| 1149 | my_siginfo_t *info, void *ctx) { |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1150 | ThreadState *thr = cur_thread(); |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1151 | // Don't mess with synchronous signals. |
| 1152 | if (sig == SIGSEGV || sig == SIGBUS || sig == SIGILL || sig == SIGABRT || |
| 1153 | sig == SIGFPE || sig == SIGPIPE || sig == thr->int_signal_send) { |
| 1154 | CHECK(thr->in_rtl == 0 || thr->in_rtl == 1); |
| 1155 | int in_rtl = thr->in_rtl; |
| 1156 | thr->in_rtl = 0; |
| 1157 | CHECK_EQ(thr->in_signal_handler, false); |
| 1158 | thr->in_signal_handler = true; |
| 1159 | if (sigact) |
| 1160 | sigactions[sig].sa_sigaction(sig, info, ctx); |
| 1161 | else |
| 1162 | sigactions[sig].sa_handler(sig); |
| 1163 | CHECK_EQ(thr->in_signal_handler, true); |
| 1164 | thr->in_signal_handler = false; |
| 1165 | thr->in_rtl = in_rtl; |
| 1166 | return; |
| 1167 | } |
| 1168 | |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1169 | SignalDesc *signal = &thr->pending_signals[sig]; |
| 1170 | if (signal->armed == false) { |
| 1171 | signal->armed = true; |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1172 | signal->sigaction = sigact; |
| 1173 | if (info) |
| 1174 | signal->siginfo = *info; |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1175 | thr->pending_signal_count++; |
| 1176 | } |
| 1177 | } |
| 1178 | |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1179 | static void rtl_sighandler(int sig) { |
| 1180 | rtl_generic_sighandler(false, sig, 0, 0); |
| 1181 | } |
| 1182 | |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1183 | static void rtl_sigaction(int sig, my_siginfo_t *info, void *ctx) { |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1184 | rtl_generic_sighandler(true, sig, info, ctx); |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1185 | } |
| 1186 | |
| 1187 | TSAN_INTERCEPTOR(int, sigaction, int sig, sigaction_t *act, sigaction_t *old) { |
| 1188 | SCOPED_TSAN_INTERCEPTOR(sigaction, sig, act, old); |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1189 | if (old) |
| 1190 | *old = sigactions[sig]; |
| 1191 | if (act == 0) |
| 1192 | return 0; |
| 1193 | sigactions[sig] = *act; |
| 1194 | sigaction_t newact = *act; |
| 1195 | if (act->sa_handler != SIG_IGN && act->sa_handler != SIG_DFL) { |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1196 | if (newact.sa_flags & SA_SIGINFO) |
| 1197 | newact.sa_sigaction = rtl_sigaction; |
| 1198 | else |
| 1199 | newact.sa_handler = rtl_sighandler; |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1200 | } |
| 1201 | int res = REAL(sigaction)(sig, &newact, 0); |
| 1202 | return res; |
| 1203 | } |
| 1204 | |
| 1205 | TSAN_INTERCEPTOR(sighandler_t, signal, int sig, sighandler_t h) { |
| 1206 | sigaction_t act = {}; |
| 1207 | act.sa_handler = h; |
| 1208 | internal_memset(&act.sa_mask, -1, sizeof(act.sa_mask)); |
| 1209 | act.sa_flags = 0; |
| 1210 | sigaction_t old = {}; |
| 1211 | int res = sigaction(sig, &act, &old); |
| 1212 | if (res) |
| 1213 | return SIG_ERR; |
| 1214 | return old.sa_handler; |
| 1215 | } |
| 1216 | |
| 1217 | TSAN_INTERCEPTOR(int, raise, int sig) { |
| 1218 | SCOPED_TSAN_INTERCEPTOR(raise, sig); |
| 1219 | CHECK_EQ(thr->int_signal_send, 0); |
| 1220 | thr->int_signal_send = sig; |
| 1221 | int res = REAL(raise)(sig); |
| 1222 | CHECK_EQ(thr->int_signal_send, sig); |
| 1223 | thr->int_signal_send = 0; |
| 1224 | return res; |
| 1225 | } |
| 1226 | |
| 1227 | TSAN_INTERCEPTOR(int, kill, int pid, int sig) { |
| 1228 | SCOPED_TSAN_INTERCEPTOR(kill, pid, sig); |
| 1229 | if (pid == GetPid()) { |
| 1230 | CHECK_EQ(thr->int_signal_send, 0); |
| 1231 | thr->int_signal_send = sig; |
| 1232 | } |
| 1233 | int res = REAL(kill)(pid, sig); |
| 1234 | if (pid == GetPid()) { |
| 1235 | CHECK_EQ(thr->int_signal_send, sig); |
| 1236 | thr->int_signal_send = 0; |
| 1237 | } |
| 1238 | return res; |
| 1239 | } |
| 1240 | |
| 1241 | TSAN_INTERCEPTOR(int, pthread_kill, void *tid, int sig) { |
| 1242 | SCOPED_TSAN_INTERCEPTOR(pthread_kill, tid, sig); |
| 1243 | if (tid == pthread_self()) { |
| 1244 | CHECK_EQ(thr->int_signal_send, 0); |
| 1245 | thr->int_signal_send = sig; |
| 1246 | } |
| 1247 | int res = REAL(pthread_kill)(tid, sig); |
| 1248 | if (tid == pthread_self()) { |
| 1249 | CHECK_EQ(thr->int_signal_send, sig); |
| 1250 | thr->int_signal_send = 0; |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1251 | } |
| 1252 | return res; |
| 1253 | } |
| 1254 | |
| 1255 | static void process_pending_signals(ThreadState *thr) { |
| 1256 | CHECK_EQ(thr->in_rtl, 0); |
| 1257 | if (thr->pending_signal_count == 0 || thr->in_signal_handler) |
| 1258 | return; |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1259 | CHECK_EQ(thr->in_signal_handler, false); |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1260 | thr->in_signal_handler = true; |
| 1261 | thr->pending_signal_count = 0; |
| 1262 | // These are too big for stack. |
| 1263 | static THREADLOCAL ucontext_t uctx; |
| 1264 | static THREADLOCAL sigset_t emptyset, oldset; |
| 1265 | getcontext(&uctx); |
| 1266 | sigfillset(&emptyset); |
| 1267 | pthread_sigmask(SIG_SETMASK, &emptyset, &oldset); |
| 1268 | for (int sig = 0; sig < kSigCount; sig++) { |
| 1269 | SignalDesc *signal = &thr->pending_signals[sig]; |
| 1270 | if (signal->armed) { |
| 1271 | signal->armed = false; |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1272 | if (sigactions[sig].sa_handler != SIG_DFL |
| 1273 | && sigactions[sig].sa_handler != SIG_IGN) { |
| 1274 | if (signal->sigaction) |
| 1275 | sigactions[sig].sa_sigaction(sig, &signal->siginfo, &uctx); |
| 1276 | else |
| 1277 | sigactions[sig].sa_handler(sig); |
| 1278 | } |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1279 | } |
| 1280 | } |
| 1281 | pthread_sigmask(SIG_SETMASK, &oldset, 0); |
| 1282 | CHECK_EQ(thr->in_signal_handler, true); |
| 1283 | thr->in_signal_handler = false; |
| 1284 | } |
| 1285 | |
| 1286 | namespace __tsan { |
| 1287 | |
| 1288 | // Used until we obtain real efficient functions. |
| 1289 | static void* poormans_memset(void *dst, int v, uptr size) { |
| 1290 | for (uptr i = 0; i < size; i++) |
| 1291 | ((char*)dst)[i] = (char)v; |
| 1292 | return dst; |
| 1293 | } |
| 1294 | |
| 1295 | static void* poormans_memcpy(void *dst, const void *src, uptr size) { |
| 1296 | for (uptr i = 0; i < size; i++) |
| 1297 | ((char*)dst)[i] = ((char*)src)[i]; |
| 1298 | return dst; |
| 1299 | } |
| 1300 | |
| 1301 | void InitializeInterceptors() { |
| 1302 | CHECK_GT(cur_thread()->in_rtl, 0); |
| 1303 | |
| 1304 | // We need to setup it early, because functions like dlsym() can call it. |
| 1305 | REAL(memset) = poormans_memset; |
| 1306 | REAL(memcpy) = poormans_memcpy; |
| 1307 | |
| 1308 | TSAN_INTERCEPT(malloc); |
| 1309 | TSAN_INTERCEPT(calloc); |
| 1310 | TSAN_INTERCEPT(realloc); |
| 1311 | TSAN_INTERCEPT(free); |
| 1312 | TSAN_INTERCEPT(cfree); |
| 1313 | TSAN_INTERCEPT(mmap); |
| 1314 | TSAN_INTERCEPT(mmap64); |
| 1315 | TSAN_INTERCEPT(munmap); |
| 1316 | TSAN_INTERCEPT(memalign); |
| 1317 | TSAN_INTERCEPT(valloc); |
| 1318 | TSAN_INTERCEPT(pvalloc); |
| 1319 | TSAN_INTERCEPT(posix_memalign); |
| 1320 | |
| 1321 | TSAN_INTERCEPT(_Znwm); |
| 1322 | TSAN_INTERCEPT(_ZnwmRKSt9nothrow_t); |
| 1323 | TSAN_INTERCEPT(_Znam); |
| 1324 | TSAN_INTERCEPT(_ZnamRKSt9nothrow_t); |
| 1325 | TSAN_INTERCEPT(_ZdlPv); |
| 1326 | TSAN_INTERCEPT(_ZdlPvRKSt9nothrow_t); |
| 1327 | TSAN_INTERCEPT(_ZdaPv); |
| 1328 | TSAN_INTERCEPT(_ZdaPvRKSt9nothrow_t); |
| 1329 | |
| 1330 | TSAN_INTERCEPT(strlen); |
| 1331 | TSAN_INTERCEPT(memset); |
| 1332 | TSAN_INTERCEPT(memcpy); |
| 1333 | TSAN_INTERCEPT(strcmp); |
| 1334 | TSAN_INTERCEPT(memchr); |
| 1335 | TSAN_INTERCEPT(memrchr); |
| 1336 | TSAN_INTERCEPT(memmove); |
| 1337 | TSAN_INTERCEPT(memcmp); |
| 1338 | TSAN_INTERCEPT(strchr); |
| 1339 | TSAN_INTERCEPT(strchrnul); |
| 1340 | TSAN_INTERCEPT(strrchr); |
| 1341 | TSAN_INTERCEPT(strncmp); |
| 1342 | TSAN_INTERCEPT(strcpy); // NOLINT |
| 1343 | TSAN_INTERCEPT(strncpy); |
| 1344 | TSAN_INTERCEPT(strstr); |
| 1345 | |
| 1346 | TSAN_INTERCEPT(__cxa_guard_acquire); |
| 1347 | TSAN_INTERCEPT(__cxa_guard_release); |
| 1348 | |
| 1349 | TSAN_INTERCEPT(pthread_create); |
| 1350 | TSAN_INTERCEPT(pthread_join); |
| 1351 | TSAN_INTERCEPT(pthread_detach); |
| 1352 | |
| 1353 | TSAN_INTERCEPT(pthread_mutex_init); |
| 1354 | TSAN_INTERCEPT(pthread_mutex_destroy); |
| 1355 | TSAN_INTERCEPT(pthread_mutex_lock); |
| 1356 | TSAN_INTERCEPT(pthread_mutex_trylock); |
| 1357 | TSAN_INTERCEPT(pthread_mutex_timedlock); |
| 1358 | TSAN_INTERCEPT(pthread_mutex_unlock); |
| 1359 | |
| 1360 | TSAN_INTERCEPT(pthread_spin_init); |
| 1361 | TSAN_INTERCEPT(pthread_spin_destroy); |
| 1362 | TSAN_INTERCEPT(pthread_spin_lock); |
| 1363 | TSAN_INTERCEPT(pthread_spin_trylock); |
| 1364 | TSAN_INTERCEPT(pthread_spin_unlock); |
| 1365 | |
| 1366 | TSAN_INTERCEPT(pthread_rwlock_init); |
| 1367 | TSAN_INTERCEPT(pthread_rwlock_destroy); |
| 1368 | TSAN_INTERCEPT(pthread_rwlock_rdlock); |
| 1369 | TSAN_INTERCEPT(pthread_rwlock_tryrdlock); |
| 1370 | TSAN_INTERCEPT(pthread_rwlock_timedrdlock); |
| 1371 | TSAN_INTERCEPT(pthread_rwlock_wrlock); |
| 1372 | TSAN_INTERCEPT(pthread_rwlock_trywrlock); |
| 1373 | TSAN_INTERCEPT(pthread_rwlock_timedwrlock); |
| 1374 | TSAN_INTERCEPT(pthread_rwlock_unlock); |
| 1375 | |
| 1376 | TSAN_INTERCEPT(pthread_cond_init); |
| 1377 | TSAN_INTERCEPT(pthread_cond_destroy); |
| 1378 | TSAN_INTERCEPT(pthread_cond_signal); |
| 1379 | TSAN_INTERCEPT(pthread_cond_broadcast); |
| 1380 | TSAN_INTERCEPT(pthread_cond_wait); |
| 1381 | TSAN_INTERCEPT(pthread_cond_timedwait); |
| 1382 | |
| 1383 | TSAN_INTERCEPT(pthread_barrier_init); |
| 1384 | TSAN_INTERCEPT(pthread_barrier_destroy); |
| 1385 | TSAN_INTERCEPT(pthread_barrier_wait); |
| 1386 | |
| 1387 | TSAN_INTERCEPT(pthread_once); |
| 1388 | |
| 1389 | TSAN_INTERCEPT(sem_init); |
| 1390 | TSAN_INTERCEPT(sem_destroy); |
| 1391 | TSAN_INTERCEPT(sem_wait); |
| 1392 | TSAN_INTERCEPT(sem_trywait); |
| 1393 | TSAN_INTERCEPT(sem_timedwait); |
| 1394 | TSAN_INTERCEPT(sem_post); |
| 1395 | TSAN_INTERCEPT(sem_getvalue); |
| 1396 | |
| 1397 | TSAN_INTERCEPT(read); |
| 1398 | TSAN_INTERCEPT(pread); |
| 1399 | TSAN_INTERCEPT(pread64); |
| 1400 | TSAN_INTERCEPT(readv); |
| 1401 | TSAN_INTERCEPT(preadv64); |
| 1402 | TSAN_INTERCEPT(write); |
| 1403 | TSAN_INTERCEPT(pwrite); |
| 1404 | TSAN_INTERCEPT(pwrite64); |
| 1405 | TSAN_INTERCEPT(writev); |
| 1406 | TSAN_INTERCEPT(pwritev64); |
| 1407 | TSAN_INTERCEPT(send); |
| 1408 | TSAN_INTERCEPT(sendmsg); |
| 1409 | TSAN_INTERCEPT(recv); |
| 1410 | TSAN_INTERCEPT(recvmsg); |
| 1411 | |
| 1412 | TSAN_INTERCEPT(unlink); |
| 1413 | TSAN_INTERCEPT(fopen); |
| 1414 | TSAN_INTERCEPT(fread); |
| 1415 | TSAN_INTERCEPT(fwrite); |
| 1416 | TSAN_INTERCEPT(puts); |
| 1417 | TSAN_INTERCEPT(rmdir); |
| 1418 | TSAN_INTERCEPT(opendir); |
| 1419 | |
| 1420 | TSAN_INTERCEPT(epoll_ctl); |
| 1421 | TSAN_INTERCEPT(epoll_wait); |
| 1422 | |
| 1423 | TSAN_INTERCEPT(sigaction); |
Dmitry Vyukov | d91d067 | 2012-05-21 08:26:51 +0000 | [diff] [blame^] | 1424 | TSAN_INTERCEPT(signal); |
| 1425 | TSAN_INTERCEPT(raise); |
| 1426 | TSAN_INTERCEPT(kill); |
| 1427 | TSAN_INTERCEPT(pthread_kill); |
Kostya Serebryany | 7ac4148 | 2012-05-10 13:48:04 +0000 | [diff] [blame] | 1428 | |
| 1429 | atexit_ctx = new(internal_alloc(MBlockAtExit, sizeof(AtExitContext))) |
| 1430 | AtExitContext(); |
| 1431 | |
| 1432 | if (__cxa_atexit(&finalize, 0, 0)) { |
| 1433 | Printf("ThreadSanitizer: failed to setup atexit callback\n"); |
| 1434 | Die(); |
| 1435 | } |
| 1436 | |
| 1437 | if (pthread_key_create(&g_thread_finalize_key, &thread_finalize)) { |
| 1438 | Printf("ThreadSanitizer: failed to create thread key\n"); |
| 1439 | Die(); |
| 1440 | } |
| 1441 | } |
| 1442 | |
| 1443 | void internal_memset(void *ptr, int c, uptr size) { |
| 1444 | REAL(memset)(ptr, c, size); |
| 1445 | } |
| 1446 | |
| 1447 | void internal_memcpy(void *dst, const void *src, uptr size) { |
| 1448 | REAL(memcpy)(dst, src, size); |
| 1449 | } |
| 1450 | |
| 1451 | int internal_memcmp(const void *s1, const void *s2, uptr size) { |
| 1452 | return REAL(memcmp)(s1, s2, size); |
| 1453 | } |
| 1454 | |
| 1455 | int internal_strcmp(const char *s1, const char *s2) { |
| 1456 | return REAL(strcmp)(s1, s2); |
| 1457 | } |
| 1458 | |
| 1459 | int internal_strncmp(const char *s1, const char *s2, uptr size) { |
| 1460 | return REAL(strncmp)(s1, s2, size); |
| 1461 | } |
| 1462 | |
| 1463 | void internal_strcpy(char *s1, const char *s2) { |
| 1464 | REAL(strcpy)(s1, s2); // NOLINT |
| 1465 | } |
| 1466 | |
| 1467 | uptr internal_strlen(const char *s) { |
| 1468 | return REAL(strlen)(s); |
| 1469 | } |
| 1470 | |
| 1471 | char* internal_strdup(const char *s) { |
| 1472 | uptr len = internal_strlen(s); |
| 1473 | char *s2 = (char*)internal_alloc(MBlockString, len + 1); |
| 1474 | internal_memcpy(s2, s, len); |
| 1475 | s2[len] = 0; |
| 1476 | return s2; |
| 1477 | } |
| 1478 | |
| 1479 | const char *internal_strstr(const char *where, const char *what) { |
| 1480 | return REAL(strstr)(where, what); |
| 1481 | } |
| 1482 | |
| 1483 | const char *internal_strchr(const char *where, char what) { |
| 1484 | return (const char*)REAL(strchr)((void*)where, what); |
| 1485 | } |
| 1486 | |
| 1487 | } // namespace __tsan |