tsan_interceptors.cc revision 917f62eaa5c2839155f703dee114a767ff81cca5
1//===-- tsan_interceptors.cc ----------------------------------------------===// 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// FIXME: move as many interceptors as possible into 13// sanitizer_common/sanitizer_common_interceptors.inc 14//===----------------------------------------------------------------------===// 15 16#include "sanitizer_common/sanitizer_atomic.h" 17#include "sanitizer_common/sanitizer_libc.h" 18#include "sanitizer_common/sanitizer_linux.h" 19#include "sanitizer_common/sanitizer_platform_limits_posix.h" 20#include "sanitizer_common/sanitizer_placement_new.h" 21#include "sanitizer_common/sanitizer_stacktrace.h" 22#include "interception/interception.h" 23#include "tsan_interface.h" 24#include "tsan_platform.h" 25#include "tsan_rtl.h" 26#include "tsan_mman.h" 27#include "tsan_fd.h" 28 29using namespace __tsan; // NOLINT 30 31const int kSigCount = 64; 32 33struct my_siginfo_t { 34 // The size is determined by looking at sizeof of real siginfo_t on linux. 35 u64 opaque[128 / sizeof(u64)]; 36}; 37 38struct sigset_t { 39 // The size is determined by looking at sizeof of real sigset_t on linux. 40 u64 val[128 / sizeof(u64)]; 41}; 42 43struct ucontext_t { 44 // The size is determined by looking at sizeof of real ucontext_t on linux. 45 u64 opaque[936 / sizeof(u64) + 1]; 46}; 47 48extern "C" int pthread_attr_init(void *attr); 49extern "C" int pthread_attr_destroy(void *attr); 50extern "C" int pthread_attr_getdetachstate(void *attr, int *v); 51extern "C" int pthread_attr_setstacksize(void *attr, uptr stacksize); 52extern "C" int pthread_attr_getstacksize(void *attr, uptr *stacksize); 53extern "C" int pthread_key_create(unsigned *key, void (*destructor)(void* v)); 54extern "C" int pthread_setspecific(unsigned key, const void *v); 55extern "C" int pthread_mutexattr_gettype(void *a, int *type); 56extern "C" int pthread_yield(); 57extern "C" int pthread_sigmask(int how, const sigset_t *set, sigset_t *oldset); 58extern "C" int sigfillset(sigset_t *set); 59extern "C" void *pthread_self(); 60extern "C" void _exit(int status); 61extern "C" int *__errno_location(); 62extern "C" int fileno_unlocked(void *stream); 63extern "C" void *__libc_malloc(uptr size); 64extern "C" void *__libc_calloc(uptr size, uptr n); 65extern "C" void *__libc_realloc(void *ptr, uptr size); 66extern "C" void __libc_free(void *ptr); 67extern "C" int mallopt(int param, int value); 68const int PTHREAD_MUTEX_RECURSIVE = 1; 69const int PTHREAD_MUTEX_RECURSIVE_NP = 1; 70const int kPthreadAttrSize = 56; 71const int EINVAL = 22; 72const int EBUSY = 16; 73const int EPOLL_CTL_ADD = 1; 74const int SIGILL = 4; 75const int SIGABRT = 6; 76const int SIGFPE = 8; 77const int SIGSEGV = 11; 78const int SIGPIPE = 13; 79const int SIGBUS = 7; 80void *const MAP_FAILED = (void*)-1; 81const int PTHREAD_BARRIER_SERIAL_THREAD = -1; 82const int MAP_FIXED = 0x10; 83typedef long long_t; // NOLINT 84 85// From /usr/include/unistd.h 86# define F_ULOCK 0 /* Unlock a previously locked region. */ 87# define F_LOCK 1 /* Lock a region for exclusive use. */ 88# define F_TLOCK 2 /* Test and lock a region for exclusive use. */ 89# define F_TEST 3 /* Test a region for other processes locks. */ 90 91typedef void (*sighandler_t)(int sig); 92 93#define errno (*__errno_location()) 94 95struct sigaction_t { 96 union { 97 sighandler_t sa_handler; 98 void (*sa_sigaction)(int sig, my_siginfo_t *siginfo, void *uctx); 99 }; 100 sigset_t sa_mask; 101 int sa_flags; 102 void (*sa_restorer)(); 103}; 104 105const sighandler_t SIG_DFL = (sighandler_t)0; 106const sighandler_t SIG_IGN = (sighandler_t)1; 107const sighandler_t SIG_ERR = (sighandler_t)-1; 108const int SA_SIGINFO = 4; 109const int SIG_SETMASK = 2; 110 111namespace std { 112struct nothrow_t {}; 113} // namespace std 114 115static sigaction_t sigactions[kSigCount]; 116 117namespace __tsan { 118struct SignalDesc { 119 bool armed; 120 bool sigaction; 121 my_siginfo_t siginfo; 122 ucontext_t ctx; 123}; 124 125struct SignalContext { 126 int in_blocking_func; 127 int int_signal_send; 128 int pending_signal_count; 129 SignalDesc pending_signals[kSigCount]; 130}; 131} // namespace __tsan 132 133static SignalContext *SigCtx(ThreadState *thr) { 134 SignalContext *ctx = (SignalContext*)thr->signal_ctx; 135 if (ctx == 0 && thr->is_alive) { 136 ScopedInRtl in_rtl; 137 ctx = (SignalContext*)MmapOrDie(sizeof(*ctx), "SignalContext"); 138 MemoryResetRange(thr, (uptr)&SigCtx, (uptr)ctx, sizeof(*ctx)); 139 thr->signal_ctx = ctx; 140 } 141 return ctx; 142} 143 144static unsigned g_thread_finalize_key; 145 146class ScopedInterceptor { 147 public: 148 ScopedInterceptor(ThreadState *thr, const char *fname, uptr pc); 149 ~ScopedInterceptor(); 150 private: 151 ThreadState *const thr_; 152 const int in_rtl_; 153}; 154 155ScopedInterceptor::ScopedInterceptor(ThreadState *thr, const char *fname, 156 uptr pc) 157 : thr_(thr) 158 , in_rtl_(thr->in_rtl) { 159 if (thr_->in_rtl == 0) { 160 Initialize(thr); 161 FuncEntry(thr, pc); 162 thr_->in_rtl++; 163 DPrintf("#%d: intercept %s()\n", thr_->tid, fname); 164 } else { 165 thr_->in_rtl++; 166 } 167} 168 169ScopedInterceptor::~ScopedInterceptor() { 170 thr_->in_rtl--; 171 if (thr_->in_rtl == 0) { 172 FuncExit(thr_); 173 ProcessPendingSignals(thr_); 174 } 175 CHECK_EQ(in_rtl_, thr_->in_rtl); 176} 177 178#define SCOPED_INTERCEPTOR_RAW(func, ...) \ 179 ThreadState *thr = cur_thread(); \ 180 StatInc(thr, StatInterceptor); \ 181 StatInc(thr, StatInt_##func); \ 182 const uptr caller_pc = GET_CALLER_PC(); \ 183 ScopedInterceptor si(thr, #func, caller_pc); \ 184 const uptr pc = __sanitizer::StackTrace::GetPreviousInstructionPc( \ 185 __sanitizer::StackTrace::GetCurrentPc()); \ 186 (void)pc; \ 187/**/ 188 189#define SCOPED_TSAN_INTERCEPTOR(func, ...) \ 190 SCOPED_INTERCEPTOR_RAW(func, __VA_ARGS__); \ 191 if (REAL(func) == 0) { \ 192 Printf("FATAL: ThreadSanitizer: failed to intercept %s\n", #func); \ 193 Die(); \ 194 } \ 195 if (thr->in_rtl > 1) \ 196 return REAL(func)(__VA_ARGS__); \ 197/**/ 198 199#define TSAN_INTERCEPTOR(ret, func, ...) INTERCEPTOR(ret, func, __VA_ARGS__) 200#define TSAN_INTERCEPT(func) INTERCEPT_FUNCTION(func) 201 202#define BLOCK_REAL(name) (BlockingCall(thr), REAL(name)) 203 204struct BlockingCall { 205 explicit BlockingCall(ThreadState *thr) 206 : ctx(SigCtx(thr)) { 207 ctx->in_blocking_func++; 208 } 209 210 ~BlockingCall() { 211 ctx->in_blocking_func--; 212 } 213 214 SignalContext *ctx; 215}; 216 217TSAN_INTERCEPTOR(unsigned, sleep, unsigned sec) { 218 SCOPED_TSAN_INTERCEPTOR(sleep, sec); 219 unsigned res = BLOCK_REAL(sleep)(sec); 220 AfterSleep(thr, pc); 221 return res; 222} 223 224TSAN_INTERCEPTOR(int, usleep, long_t usec) { 225 SCOPED_TSAN_INTERCEPTOR(usleep, usec); 226 int res = BLOCK_REAL(usleep)(usec); 227 AfterSleep(thr, pc); 228 return res; 229} 230 231TSAN_INTERCEPTOR(int, nanosleep, void *req, void *rem) { 232 SCOPED_TSAN_INTERCEPTOR(nanosleep, req, rem); 233 int res = BLOCK_REAL(nanosleep)(req, rem); 234 AfterSleep(thr, pc); 235 return res; 236} 237 238class AtExitContext { 239 public: 240 AtExitContext() 241 : mtx_(MutexTypeAtExit, StatMtxAtExit) 242 , pos_() { 243 } 244 245 typedef void(*atexit_t)(); 246 247 int atexit(ThreadState *thr, uptr pc, bool is_on_exit, 248 atexit_t f, void *arg) { 249 Lock l(&mtx_); 250 if (pos_ == kMaxAtExit) 251 return 1; 252 Release(thr, pc, (uptr)this); 253 stack_[pos_] = f; 254 args_[pos_] = arg; 255 is_on_exits_[pos_] = is_on_exit; 256 pos_++; 257 return 0; 258 } 259 260 void exit(ThreadState *thr, uptr pc) { 261 CHECK_EQ(thr->in_rtl, 0); 262 for (;;) { 263 atexit_t f = 0; 264 void *arg = 0; 265 bool is_on_exit = false; 266 { 267 Lock l(&mtx_); 268 if (pos_) { 269 pos_--; 270 f = stack_[pos_]; 271 arg = args_[pos_]; 272 is_on_exit = is_on_exits_[pos_]; 273 ScopedInRtl in_rtl; 274 Acquire(thr, pc, (uptr)this); 275 } 276 } 277 if (f == 0) 278 break; 279 DPrintf("#%d: executing atexit func %p\n", thr->tid, f); 280 CHECK_EQ(thr->in_rtl, 0); 281 if (is_on_exit) 282 ((void(*)(int status, void *arg))f)(0, arg); 283 else 284 ((void(*)(void *arg, void *dso))f)(arg, 0); 285 } 286 } 287 288 private: 289 static const int kMaxAtExit = 128; 290 Mutex mtx_; 291 atexit_t stack_[kMaxAtExit]; 292 void *args_[kMaxAtExit]; 293 bool is_on_exits_[kMaxAtExit]; 294 int pos_; 295}; 296 297static AtExitContext *atexit_ctx; 298 299TSAN_INTERCEPTOR(int, atexit, void (*f)()) { 300 if (cur_thread()->in_symbolizer) 301 return 0; 302 SCOPED_TSAN_INTERCEPTOR(atexit, f); 303 return atexit_ctx->atexit(thr, pc, false, (void(*)())f, 0); 304} 305 306TSAN_INTERCEPTOR(int, on_exit, void(*f)(int, void*), void *arg) { 307 if (cur_thread()->in_symbolizer) 308 return 0; 309 SCOPED_TSAN_INTERCEPTOR(on_exit, f, arg); 310 return atexit_ctx->atexit(thr, pc, true, (void(*)())f, arg); 311} 312 313TSAN_INTERCEPTOR(int, __cxa_atexit, void (*f)(void *a), void *arg, void *dso) { 314 if (cur_thread()->in_symbolizer) 315 return 0; 316 SCOPED_TSAN_INTERCEPTOR(__cxa_atexit, f, arg, dso); 317 if (dso) 318 return REAL(__cxa_atexit)(f, arg, dso); 319 return atexit_ctx->atexit(thr, pc, false, (void(*)())f, arg); 320} 321 322// Cleanup old bufs. 323static void JmpBufGarbageCollect(ThreadState *thr, uptr sp) { 324 for (uptr i = 0; i < thr->jmp_bufs.Size(); i++) { 325 JmpBuf *buf = &thr->jmp_bufs[i]; 326 if (buf->sp <= sp) { 327 uptr sz = thr->jmp_bufs.Size(); 328 thr->jmp_bufs[i] = thr->jmp_bufs[sz - 1]; 329 thr->jmp_bufs.PopBack(); 330 i--; 331 } 332 } 333} 334 335static void SetJmp(ThreadState *thr, uptr sp, uptr mangled_sp) { 336 if (thr->shadow_stack_pos == 0) // called from libc guts during bootstrap 337 return; 338 // Cleanup old bufs. 339 JmpBufGarbageCollect(thr, sp); 340 // Remember the buf. 341 JmpBuf *buf = thr->jmp_bufs.PushBack(); 342 buf->sp = sp; 343 buf->mangled_sp = mangled_sp; 344 buf->shadow_stack_pos = thr->shadow_stack_pos; 345} 346 347static void LongJmp(ThreadState *thr, uptr *env) { 348 uptr mangled_sp = env[6]; 349 // Find the saved buf by mangled_sp. 350 for (uptr i = 0; i < thr->jmp_bufs.Size(); i++) { 351 JmpBuf *buf = &thr->jmp_bufs[i]; 352 if (buf->mangled_sp == mangled_sp) { 353 CHECK_GE(thr->shadow_stack_pos, buf->shadow_stack_pos); 354 // Unwind the stack. 355 while (thr->shadow_stack_pos > buf->shadow_stack_pos) 356 FuncExit(thr); 357 JmpBufGarbageCollect(thr, buf->sp - 1); // do not collect buf->sp 358 return; 359 } 360 } 361 Printf("ThreadSanitizer: can't find longjmp buf\n"); 362 CHECK(0); 363} 364 365extern "C" void __tsan_setjmp(uptr sp, uptr mangled_sp) { 366 ScopedInRtl in_rtl; 367 SetJmp(cur_thread(), sp, mangled_sp); 368} 369 370// Not called. Merely to satisfy TSAN_INTERCEPT(). 371extern "C" int __interceptor_setjmp(void *env) { 372 CHECK(0); 373 return 0; 374} 375 376extern "C" int __interceptor__setjmp(void *env) { 377 CHECK(0); 378 return 0; 379} 380 381extern "C" int __interceptor_sigsetjmp(void *env) { 382 CHECK(0); 383 return 0; 384} 385 386extern "C" int __interceptor___sigsetjmp(void *env) { 387 CHECK(0); 388 return 0; 389} 390 391extern "C" int setjmp(void *env); 392extern "C" int _setjmp(void *env); 393extern "C" int sigsetjmp(void *env); 394extern "C" int __sigsetjmp(void *env); 395DEFINE_REAL(int, setjmp, void *env) 396DEFINE_REAL(int, _setjmp, void *env) 397DEFINE_REAL(int, sigsetjmp, void *env) 398DEFINE_REAL(int, __sigsetjmp, void *env) 399 400TSAN_INTERCEPTOR(void, longjmp, uptr *env, int val) { 401 { 402 SCOPED_TSAN_INTERCEPTOR(longjmp, env, val); 403 } 404 LongJmp(cur_thread(), env); 405 REAL(longjmp)(env, val); 406} 407 408TSAN_INTERCEPTOR(void, siglongjmp, uptr *env, int val) { 409 { 410 SCOPED_TSAN_INTERCEPTOR(siglongjmp, env, val); 411 } 412 LongJmp(cur_thread(), env); 413 REAL(siglongjmp)(env, val); 414} 415 416TSAN_INTERCEPTOR(void*, malloc, uptr size) { 417 if (cur_thread()->in_symbolizer) 418 return __libc_malloc(size); 419 void *p = 0; 420 { 421 SCOPED_INTERCEPTOR_RAW(malloc, size); 422 p = user_alloc(thr, pc, size); 423 } 424 invoke_malloc_hook(p, size); 425 return p; 426} 427 428TSAN_INTERCEPTOR(void*, __libc_memalign, uptr align, uptr sz) { 429 SCOPED_TSAN_INTERCEPTOR(__libc_memalign, align, sz); 430 return user_alloc(thr, pc, sz, align); 431} 432 433TSAN_INTERCEPTOR(void*, calloc, uptr size, uptr n) { 434 if (cur_thread()->in_symbolizer) 435 return __libc_calloc(size, n); 436 if (__sanitizer::CallocShouldReturnNullDueToOverflow(size, n)) return 0; 437 void *p = 0; 438 { 439 SCOPED_INTERCEPTOR_RAW(calloc, size, n); 440 p = user_alloc(thr, pc, n * size); 441 if (p) 442 internal_memset(p, 0, n * size); 443 } 444 invoke_malloc_hook(p, n * size); 445 return p; 446} 447 448TSAN_INTERCEPTOR(void*, realloc, void *p, uptr size) { 449 if (cur_thread()->in_symbolizer) 450 return __libc_realloc(p, size); 451 if (p) 452 invoke_free_hook(p); 453 { 454 SCOPED_INTERCEPTOR_RAW(realloc, p, size); 455 p = user_realloc(thr, pc, p, size); 456 } 457 invoke_malloc_hook(p, size); 458 return p; 459} 460 461TSAN_INTERCEPTOR(void, free, void *p) { 462 if (p == 0) 463 return; 464 if (cur_thread()->in_symbolizer) 465 return __libc_free(p); 466 invoke_free_hook(p); 467 SCOPED_INTERCEPTOR_RAW(free, p); 468 user_free(thr, pc, p); 469} 470 471TSAN_INTERCEPTOR(void, cfree, void *p) { 472 if (p == 0) 473 return; 474 if (cur_thread()->in_symbolizer) 475 return __libc_free(p); 476 invoke_free_hook(p); 477 SCOPED_INTERCEPTOR_RAW(cfree, p); 478 user_free(thr, pc, p); 479} 480 481TSAN_INTERCEPTOR(uptr, malloc_usable_size, void *p) { 482 SCOPED_INTERCEPTOR_RAW(malloc_usable_size, p); 483 return user_alloc_usable_size(thr, pc, p); 484} 485 486#define OPERATOR_NEW_BODY(mangled_name) \ 487 if (cur_thread()->in_symbolizer) \ 488 return __libc_malloc(size); \ 489 void *p = 0; \ 490 { \ 491 SCOPED_INTERCEPTOR_RAW(mangled_name, size); \ 492 p = user_alloc(thr, pc, size); \ 493 } \ 494 invoke_malloc_hook(p, size); \ 495 return p; 496 497void *operator new(__sanitizer::uptr size) { 498 OPERATOR_NEW_BODY(_Znwm); 499} 500void *operator new[](__sanitizer::uptr size) { 501 OPERATOR_NEW_BODY(_Znam); 502} 503void *operator new(__sanitizer::uptr size, std::nothrow_t const&) { 504 OPERATOR_NEW_BODY(_ZnwmRKSt9nothrow_t); 505} 506void *operator new[](__sanitizer::uptr size, std::nothrow_t const&) { 507 OPERATOR_NEW_BODY(_ZnamRKSt9nothrow_t); 508} 509 510#define OPERATOR_DELETE_BODY(mangled_name) \ 511 if (ptr == 0) return; \ 512 if (cur_thread()->in_symbolizer) \ 513 return __libc_free(ptr); \ 514 invoke_free_hook(ptr); \ 515 SCOPED_INTERCEPTOR_RAW(mangled_name, ptr); \ 516 user_free(thr, pc, ptr); 517 518void operator delete(void *ptr) { 519 OPERATOR_DELETE_BODY(_ZdlPv); 520} 521void operator delete[](void *ptr) { 522 OPERATOR_DELETE_BODY(_ZdlPvRKSt9nothrow_t); 523} 524void operator delete(void *ptr, std::nothrow_t const&) { 525 OPERATOR_DELETE_BODY(_ZdaPv); 526} 527void operator delete[](void *ptr, std::nothrow_t const&) { 528 OPERATOR_DELETE_BODY(_ZdaPvRKSt9nothrow_t); 529} 530 531TSAN_INTERCEPTOR(uptr, strlen, const char *s) { 532 SCOPED_TSAN_INTERCEPTOR(strlen, s); 533 uptr len = internal_strlen(s); 534 MemoryAccessRange(thr, pc, (uptr)s, len + 1, false); 535 return len; 536} 537 538TSAN_INTERCEPTOR(void*, memset, void *dst, int v, uptr size) { 539 SCOPED_TSAN_INTERCEPTOR(memset, dst, v, size); 540 MemoryAccessRange(thr, pc, (uptr)dst, size, true); 541 return internal_memset(dst, v, size); 542} 543 544TSAN_INTERCEPTOR(void*, memcpy, void *dst, const void *src, uptr size) { 545 SCOPED_TSAN_INTERCEPTOR(memcpy, dst, src, size); 546 MemoryAccessRange(thr, pc, (uptr)dst, size, true); 547 MemoryAccessRange(thr, pc, (uptr)src, size, false); 548 return internal_memcpy(dst, src, size); 549} 550 551TSAN_INTERCEPTOR(int, memcmp, const void *s1, const void *s2, uptr n) { 552 SCOPED_TSAN_INTERCEPTOR(memcmp, s1, s2, n); 553 int res = 0; 554 uptr len = 0; 555 for (; len < n; len++) { 556 if ((res = ((unsigned char*)s1)[len] - ((unsigned char*)s2)[len])) 557 break; 558 } 559 MemoryAccessRange(thr, pc, (uptr)s1, len < n ? len + 1 : n, false); 560 MemoryAccessRange(thr, pc, (uptr)s2, len < n ? len + 1 : n, false); 561 return res; 562} 563 564TSAN_INTERCEPTOR(int, strcmp, const char *s1, const char *s2) { 565 SCOPED_TSAN_INTERCEPTOR(strcmp, s1, s2); 566 uptr len = 0; 567 for (; s1[len] && s2[len]; len++) { 568 if (s1[len] != s2[len]) 569 break; 570 } 571 MemoryAccessRange(thr, pc, (uptr)s1, len + 1, false); 572 MemoryAccessRange(thr, pc, (uptr)s2, len + 1, false); 573 return s1[len] - s2[len]; 574} 575 576TSAN_INTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr n) { 577 SCOPED_TSAN_INTERCEPTOR(strncmp, s1, s2, n); 578 uptr len = 0; 579 for (; len < n && s1[len] && s2[len]; len++) { 580 if (s1[len] != s2[len]) 581 break; 582 } 583 MemoryAccessRange(thr, pc, (uptr)s1, len < n ? len + 1 : n, false); 584 MemoryAccessRange(thr, pc, (uptr)s2, len < n ? len + 1 : n, false); 585 return len == n ? 0 : s1[len] - s2[len]; 586} 587 588TSAN_INTERCEPTOR(void*, memchr, void *s, int c, uptr n) { 589 SCOPED_TSAN_INTERCEPTOR(memchr, s, c, n); 590 void *res = REAL(memchr)(s, c, n); 591 uptr len = res ? (char*)res - (char*)s + 1 : n; 592 MemoryAccessRange(thr, pc, (uptr)s, len, false); 593 return res; 594} 595 596TSAN_INTERCEPTOR(void*, memrchr, char *s, int c, uptr n) { 597 SCOPED_TSAN_INTERCEPTOR(memrchr, s, c, n); 598 MemoryAccessRange(thr, pc, (uptr)s, n, false); 599 return REAL(memrchr)(s, c, n); 600} 601 602TSAN_INTERCEPTOR(void*, memmove, void *dst, void *src, uptr n) { 603 SCOPED_TSAN_INTERCEPTOR(memmove, dst, src, n); 604 MemoryAccessRange(thr, pc, (uptr)dst, n, true); 605 MemoryAccessRange(thr, pc, (uptr)src, n, false); 606 return REAL(memmove)(dst, src, n); 607} 608 609TSAN_INTERCEPTOR(char*, strchr, char *s, int c) { 610 SCOPED_TSAN_INTERCEPTOR(strchr, s, c); 611 char *res = REAL(strchr)(s, c); 612 uptr len = res ? (char*)res - (char*)s + 1 : internal_strlen(s) + 1; 613 MemoryAccessRange(thr, pc, (uptr)s, len, false); 614 return res; 615} 616 617TSAN_INTERCEPTOR(char*, strchrnul, char *s, int c) { 618 SCOPED_TSAN_INTERCEPTOR(strchrnul, s, c); 619 char *res = REAL(strchrnul)(s, c); 620 uptr len = (char*)res - (char*)s + 1; 621 MemoryAccessRange(thr, pc, (uptr)s, len, false); 622 return res; 623} 624 625TSAN_INTERCEPTOR(char*, strrchr, char *s, int c) { 626 SCOPED_TSAN_INTERCEPTOR(strrchr, s, c); 627 MemoryAccessRange(thr, pc, (uptr)s, internal_strlen(s) + 1, false); 628 return REAL(strrchr)(s, c); 629} 630 631TSAN_INTERCEPTOR(char*, strcpy, char *dst, const char *src) { // NOLINT 632 SCOPED_TSAN_INTERCEPTOR(strcpy, dst, src); // NOLINT 633 uptr srclen = internal_strlen(src); 634 MemoryAccessRange(thr, pc, (uptr)dst, srclen + 1, true); 635 MemoryAccessRange(thr, pc, (uptr)src, srclen + 1, false); 636 return REAL(strcpy)(dst, src); // NOLINT 637} 638 639TSAN_INTERCEPTOR(char*, strncpy, char *dst, char *src, uptr n) { 640 SCOPED_TSAN_INTERCEPTOR(strncpy, dst, src, n); 641 uptr srclen = internal_strnlen(src, n); 642 MemoryAccessRange(thr, pc, (uptr)dst, n, true); 643 MemoryAccessRange(thr, pc, (uptr)src, min(srclen + 1, n), false); 644 return REAL(strncpy)(dst, src, n); 645} 646 647TSAN_INTERCEPTOR(const char*, strstr, const char *s1, const char *s2) { 648 SCOPED_TSAN_INTERCEPTOR(strstr, s1, s2); 649 const char *res = REAL(strstr)(s1, s2); 650 uptr len1 = internal_strlen(s1); 651 uptr len2 = internal_strlen(s2); 652 MemoryAccessRange(thr, pc, (uptr)s1, len1 + 1, false); 653 MemoryAccessRange(thr, pc, (uptr)s2, len2 + 1, false); 654 return res; 655} 656 657static bool fix_mmap_addr(void **addr, long_t sz, int flags) { 658 if (*addr) { 659 if (!IsAppMem((uptr)*addr) || !IsAppMem((uptr)*addr + sz - 1)) { 660 if (flags & MAP_FIXED) { 661 errno = EINVAL; 662 return false; 663 } else { 664 *addr = 0; 665 } 666 } 667 } 668 return true; 669} 670 671TSAN_INTERCEPTOR(void*, mmap, void *addr, long_t sz, int prot, 672 int flags, int fd, unsigned off) { 673 SCOPED_TSAN_INTERCEPTOR(mmap, addr, sz, prot, flags, fd, off); 674 if (!fix_mmap_addr(&addr, sz, flags)) 675 return MAP_FAILED; 676 void *res = REAL(mmap)(addr, sz, prot, flags, fd, off); 677 if (res != MAP_FAILED) { 678 if (fd > 0) 679 FdAccess(thr, pc, fd); 680 MemoryRangeImitateWrite(thr, pc, (uptr)res, sz); 681 } 682 return res; 683} 684 685TSAN_INTERCEPTOR(void*, mmap64, void *addr, long_t sz, int prot, 686 int flags, int fd, u64 off) { 687 SCOPED_TSAN_INTERCEPTOR(mmap64, addr, sz, prot, flags, fd, off); 688 if (!fix_mmap_addr(&addr, sz, flags)) 689 return MAP_FAILED; 690 void *res = REAL(mmap64)(addr, sz, prot, flags, fd, off); 691 if (res != MAP_FAILED) { 692 if (fd > 0) 693 FdAccess(thr, pc, fd); 694 MemoryRangeImitateWrite(thr, pc, (uptr)res, sz); 695 } 696 return res; 697} 698 699TSAN_INTERCEPTOR(int, munmap, void *addr, long_t sz) { 700 SCOPED_TSAN_INTERCEPTOR(munmap, addr, sz); 701 DontNeedShadowFor((uptr)addr, sz); 702 int res = REAL(munmap)(addr, sz); 703 return res; 704} 705 706TSAN_INTERCEPTOR(void*, memalign, uptr align, uptr sz) { 707 SCOPED_TSAN_INTERCEPTOR(memalign, align, sz); 708 return user_alloc(thr, pc, sz, align); 709} 710 711TSAN_INTERCEPTOR(void*, valloc, uptr sz) { 712 SCOPED_TSAN_INTERCEPTOR(valloc, sz); 713 return user_alloc(thr, pc, sz, GetPageSizeCached()); 714} 715 716TSAN_INTERCEPTOR(void*, pvalloc, uptr sz) { 717 SCOPED_TSAN_INTERCEPTOR(pvalloc, sz); 718 sz = RoundUp(sz, GetPageSizeCached()); 719 return user_alloc(thr, pc, sz, GetPageSizeCached()); 720} 721 722TSAN_INTERCEPTOR(int, posix_memalign, void **memptr, uptr align, uptr sz) { 723 SCOPED_TSAN_INTERCEPTOR(posix_memalign, memptr, align, sz); 724 *memptr = user_alloc(thr, pc, sz, align); 725 return 0; 726} 727 728// Used in thread-safe function static initialization. 729extern "C" int INTERFACE_ATTRIBUTE __cxa_guard_acquire(atomic_uint32_t *g) { 730 SCOPED_INTERCEPTOR_RAW(__cxa_guard_acquire, g); 731 for (;;) { 732 u32 cmp = atomic_load(g, memory_order_acquire); 733 if (cmp == 0) { 734 if (atomic_compare_exchange_strong(g, &cmp, 1<<16, memory_order_relaxed)) 735 return 1; 736 } else if (cmp == 1) { 737 Acquire(thr, pc, (uptr)g); 738 return 0; 739 } else { 740 internal_sched_yield(); 741 } 742 } 743} 744 745extern "C" void INTERFACE_ATTRIBUTE __cxa_guard_release(atomic_uint32_t *g) { 746 SCOPED_INTERCEPTOR_RAW(__cxa_guard_release, g); 747 Release(thr, pc, (uptr)g); 748 atomic_store(g, 1, memory_order_release); 749} 750 751extern "C" void INTERFACE_ATTRIBUTE __cxa_guard_abort(atomic_uint32_t *g) { 752 SCOPED_INTERCEPTOR_RAW(__cxa_guard_abort, g); 753 atomic_store(g, 0, memory_order_relaxed); 754} 755 756static void thread_finalize(void *v) { 757 uptr iter = (uptr)v; 758 if (iter > 1) { 759 if (pthread_setspecific(g_thread_finalize_key, (void*)(iter - 1))) { 760 Printf("ThreadSanitizer: failed to set thread key\n"); 761 Die(); 762 } 763 return; 764 } 765 { 766 ScopedInRtl in_rtl; 767 ThreadState *thr = cur_thread(); 768 ThreadFinish(thr); 769 SignalContext *sctx = thr->signal_ctx; 770 if (sctx) { 771 thr->signal_ctx = 0; 772 UnmapOrDie(sctx, sizeof(*sctx)); 773 } 774 } 775} 776 777 778struct ThreadParam { 779 void* (*callback)(void *arg); 780 void *param; 781 atomic_uintptr_t tid; 782}; 783 784extern "C" void *__tsan_thread_start_func(void *arg) { 785 ThreadParam *p = (ThreadParam*)arg; 786 void* (*callback)(void *arg) = p->callback; 787 void *param = p->param; 788 int tid = 0; 789 { 790 ThreadState *thr = cur_thread(); 791 ScopedInRtl in_rtl; 792 if (pthread_setspecific(g_thread_finalize_key, (void*)4)) { 793 Printf("ThreadSanitizer: failed to set thread key\n"); 794 Die(); 795 } 796 while ((tid = atomic_load(&p->tid, memory_order_acquire)) == 0) 797 pthread_yield(); 798 atomic_store(&p->tid, 0, memory_order_release); 799 ThreadStart(thr, tid, GetTid()); 800 CHECK_EQ(thr->in_rtl, 1); 801 } 802 void *res = callback(param); 803 // Prevent the callback from being tail called, 804 // it mixes up stack traces. 805 volatile int foo = 42; 806 foo++; 807 return res; 808} 809 810TSAN_INTERCEPTOR(int, pthread_create, 811 void *th, void *attr, void *(*callback)(void*), void * param) { 812 SCOPED_TSAN_INTERCEPTOR(pthread_create, th, attr, callback, param); 813 __sanitizer_pthread_attr_t myattr; 814 if (attr == 0) { 815 pthread_attr_init(&myattr); 816 attr = &myattr; 817 } 818 int detached = 0; 819 pthread_attr_getdetachstate(attr, &detached); 820 821#if defined(TSAN_DEBUG_OUTPUT) 822 int verbosity = (TSAN_DEBUG_OUTPUT); 823#else 824 int verbosity = 0; 825#endif 826 AdjustStackSizeLinux(attr, verbosity); 827 828 ThreadParam p; 829 p.callback = callback; 830 p.param = param; 831 atomic_store(&p.tid, 0, memory_order_relaxed); 832 int res = REAL(pthread_create)(th, attr, __tsan_thread_start_func, &p); 833 if (res == 0) { 834 int tid = ThreadCreate(thr, pc, *(uptr*)th, detached); 835 CHECK_NE(tid, 0); 836 atomic_store(&p.tid, tid, memory_order_release); 837 while (atomic_load(&p.tid, memory_order_acquire) != 0) 838 pthread_yield(); 839 } 840 if (attr == &myattr) 841 pthread_attr_destroy(&myattr); 842 return res; 843} 844 845TSAN_INTERCEPTOR(int, pthread_join, void *th, void **ret) { 846 SCOPED_TSAN_INTERCEPTOR(pthread_join, th, ret); 847 int tid = ThreadTid(thr, pc, (uptr)th); 848 int res = BLOCK_REAL(pthread_join)(th, ret); 849 if (res == 0) { 850 ThreadJoin(thr, pc, tid); 851 } 852 return res; 853} 854 855TSAN_INTERCEPTOR(int, pthread_detach, void *th) { 856 SCOPED_TSAN_INTERCEPTOR(pthread_detach, th); 857 int tid = ThreadTid(thr, pc, (uptr)th); 858 int res = REAL(pthread_detach)(th); 859 if (res == 0) { 860 ThreadDetach(thr, pc, tid); 861 } 862 return res; 863} 864 865TSAN_INTERCEPTOR(int, pthread_mutex_init, void *m, void *a) { 866 SCOPED_TSAN_INTERCEPTOR(pthread_mutex_init, m, a); 867 int res = REAL(pthread_mutex_init)(m, a); 868 if (res == 0) { 869 bool recursive = false; 870 if (a) { 871 int type = 0; 872 if (pthread_mutexattr_gettype(a, &type) == 0) 873 recursive = (type == PTHREAD_MUTEX_RECURSIVE 874 || type == PTHREAD_MUTEX_RECURSIVE_NP); 875 } 876 MutexCreate(thr, pc, (uptr)m, false, recursive, false); 877 } 878 return res; 879} 880 881TSAN_INTERCEPTOR(int, pthread_mutex_destroy, void *m) { 882 SCOPED_TSAN_INTERCEPTOR(pthread_mutex_destroy, m); 883 int res = REAL(pthread_mutex_destroy)(m); 884 if (res == 0 || res == EBUSY) { 885 MutexDestroy(thr, pc, (uptr)m); 886 } 887 return res; 888} 889 890TSAN_INTERCEPTOR(int, pthread_mutex_lock, void *m) { 891 SCOPED_TSAN_INTERCEPTOR(pthread_mutex_lock, m); 892 int res = REAL(pthread_mutex_lock)(m); 893 if (res == 0) { 894 MutexLock(thr, pc, (uptr)m); 895 } 896 return res; 897} 898 899TSAN_INTERCEPTOR(int, pthread_mutex_trylock, void *m) { 900 SCOPED_TSAN_INTERCEPTOR(pthread_mutex_trylock, m); 901 int res = REAL(pthread_mutex_trylock)(m); 902 if (res == 0) { 903 MutexLock(thr, pc, (uptr)m); 904 } 905 return res; 906} 907 908TSAN_INTERCEPTOR(int, pthread_mutex_timedlock, void *m, void *abstime) { 909 SCOPED_TSAN_INTERCEPTOR(pthread_mutex_timedlock, m, abstime); 910 int res = REAL(pthread_mutex_timedlock)(m, abstime); 911 if (res == 0) { 912 MutexLock(thr, pc, (uptr)m); 913 } 914 return res; 915} 916 917TSAN_INTERCEPTOR(int, pthread_mutex_unlock, void *m) { 918 SCOPED_TSAN_INTERCEPTOR(pthread_mutex_unlock, m); 919 MutexUnlock(thr, pc, (uptr)m); 920 int res = REAL(pthread_mutex_unlock)(m); 921 return res; 922} 923 924TSAN_INTERCEPTOR(int, pthread_spin_init, void *m, int pshared) { 925 SCOPED_TSAN_INTERCEPTOR(pthread_spin_init, m, pshared); 926 int res = REAL(pthread_spin_init)(m, pshared); 927 if (res == 0) { 928 MutexCreate(thr, pc, (uptr)m, false, false, false); 929 } 930 return res; 931} 932 933TSAN_INTERCEPTOR(int, pthread_spin_destroy, void *m) { 934 SCOPED_TSAN_INTERCEPTOR(pthread_spin_destroy, m); 935 int res = REAL(pthread_spin_destroy)(m); 936 if (res == 0) { 937 MutexDestroy(thr, pc, (uptr)m); 938 } 939 return res; 940} 941 942TSAN_INTERCEPTOR(int, pthread_spin_lock, void *m) { 943 SCOPED_TSAN_INTERCEPTOR(pthread_spin_lock, m); 944 int res = REAL(pthread_spin_lock)(m); 945 if (res == 0) { 946 MutexLock(thr, pc, (uptr)m); 947 } 948 return res; 949} 950 951TSAN_INTERCEPTOR(int, pthread_spin_trylock, void *m) { 952 SCOPED_TSAN_INTERCEPTOR(pthread_spin_trylock, m); 953 int res = REAL(pthread_spin_trylock)(m); 954 if (res == 0) { 955 MutexLock(thr, pc, (uptr)m); 956 } 957 return res; 958} 959 960TSAN_INTERCEPTOR(int, pthread_spin_unlock, void *m) { 961 SCOPED_TSAN_INTERCEPTOR(pthread_spin_unlock, m); 962 MutexUnlock(thr, pc, (uptr)m); 963 int res = REAL(pthread_spin_unlock)(m); 964 return res; 965} 966 967TSAN_INTERCEPTOR(int, pthread_rwlock_init, void *m, void *a) { 968 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_init, m, a); 969 int res = REAL(pthread_rwlock_init)(m, a); 970 if (res == 0) { 971 MutexCreate(thr, pc, (uptr)m, true, false, false); 972 } 973 return res; 974} 975 976TSAN_INTERCEPTOR(int, pthread_rwlock_destroy, void *m) { 977 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_destroy, m); 978 int res = REAL(pthread_rwlock_destroy)(m); 979 if (res == 0) { 980 MutexDestroy(thr, pc, (uptr)m); 981 } 982 return res; 983} 984 985TSAN_INTERCEPTOR(int, pthread_rwlock_rdlock, void *m) { 986 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_rdlock, m); 987 int res = REAL(pthread_rwlock_rdlock)(m); 988 if (res == 0) { 989 MutexReadLock(thr, pc, (uptr)m); 990 } 991 return res; 992} 993 994TSAN_INTERCEPTOR(int, pthread_rwlock_tryrdlock, void *m) { 995 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_tryrdlock, m); 996 int res = REAL(pthread_rwlock_tryrdlock)(m); 997 if (res == 0) { 998 MutexReadLock(thr, pc, (uptr)m); 999 } 1000 return res; 1001} 1002 1003TSAN_INTERCEPTOR(int, pthread_rwlock_timedrdlock, void *m, void *abstime) { 1004 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_timedrdlock, m, abstime); 1005 int res = REAL(pthread_rwlock_timedrdlock)(m, abstime); 1006 if (res == 0) { 1007 MutexReadLock(thr, pc, (uptr)m); 1008 } 1009 return res; 1010} 1011 1012TSAN_INTERCEPTOR(int, pthread_rwlock_wrlock, void *m) { 1013 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_wrlock, m); 1014 int res = REAL(pthread_rwlock_wrlock)(m); 1015 if (res == 0) { 1016 MutexLock(thr, pc, (uptr)m); 1017 } 1018 return res; 1019} 1020 1021TSAN_INTERCEPTOR(int, pthread_rwlock_trywrlock, void *m) { 1022 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_trywrlock, m); 1023 int res = REAL(pthread_rwlock_trywrlock)(m); 1024 if (res == 0) { 1025 MutexLock(thr, pc, (uptr)m); 1026 } 1027 return res; 1028} 1029 1030TSAN_INTERCEPTOR(int, pthread_rwlock_timedwrlock, void *m, void *abstime) { 1031 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_timedwrlock, m, abstime); 1032 int res = REAL(pthread_rwlock_timedwrlock)(m, abstime); 1033 if (res == 0) { 1034 MutexLock(thr, pc, (uptr)m); 1035 } 1036 return res; 1037} 1038 1039TSAN_INTERCEPTOR(int, pthread_rwlock_unlock, void *m) { 1040 SCOPED_TSAN_INTERCEPTOR(pthread_rwlock_unlock, m); 1041 MutexReadOrWriteUnlock(thr, pc, (uptr)m); 1042 int res = REAL(pthread_rwlock_unlock)(m); 1043 return res; 1044} 1045 1046// libpthread.so contains several versions of pthread_cond_init symbol. 1047// When we just dlsym() it, we get the wrong (old) version. 1048/* 1049TSAN_INTERCEPTOR(int, pthread_cond_init, void *c, void *a) { 1050 SCOPED_TSAN_INTERCEPTOR(pthread_cond_init, c, a); 1051 int res = REAL(pthread_cond_init)(c, a); 1052 return res; 1053} 1054*/ 1055 1056TSAN_INTERCEPTOR(int, pthread_cond_destroy, void *c) { 1057 SCOPED_TSAN_INTERCEPTOR(pthread_cond_destroy, c); 1058 int res = REAL(pthread_cond_destroy)(c); 1059 return res; 1060} 1061 1062TSAN_INTERCEPTOR(int, pthread_cond_signal, void *c) { 1063 SCOPED_TSAN_INTERCEPTOR(pthread_cond_signal, c); 1064 int res = REAL(pthread_cond_signal)(c); 1065 return res; 1066} 1067 1068TSAN_INTERCEPTOR(int, pthread_cond_broadcast, void *c) { 1069 SCOPED_TSAN_INTERCEPTOR(pthread_cond_broadcast, c); 1070 int res = REAL(pthread_cond_broadcast)(c); 1071 return res; 1072} 1073 1074TSAN_INTERCEPTOR(int, pthread_cond_wait, void *c, void *m) { 1075 SCOPED_TSAN_INTERCEPTOR(pthread_cond_wait, c, m); 1076 MutexUnlock(thr, pc, (uptr)m); 1077 int res = REAL(pthread_cond_wait)(c, m); 1078 MutexLock(thr, pc, (uptr)m); 1079 return res; 1080} 1081 1082TSAN_INTERCEPTOR(int, pthread_cond_timedwait, void *c, void *m, void *abstime) { 1083 SCOPED_TSAN_INTERCEPTOR(pthread_cond_timedwait, c, m, abstime); 1084 MutexUnlock(thr, pc, (uptr)m); 1085 int res = REAL(pthread_cond_timedwait)(c, m, abstime); 1086 MutexLock(thr, pc, (uptr)m); 1087 return res; 1088} 1089 1090TSAN_INTERCEPTOR(int, pthread_barrier_init, void *b, void *a, unsigned count) { 1091 SCOPED_TSAN_INTERCEPTOR(pthread_barrier_init, b, a, count); 1092 MemoryWrite(thr, pc, (uptr)b, kSizeLog1); 1093 int res = REAL(pthread_barrier_init)(b, a, count); 1094 return res; 1095} 1096 1097TSAN_INTERCEPTOR(int, pthread_barrier_destroy, void *b) { 1098 SCOPED_TSAN_INTERCEPTOR(pthread_barrier_destroy, b); 1099 MemoryWrite(thr, pc, (uptr)b, kSizeLog1); 1100 int res = REAL(pthread_barrier_destroy)(b); 1101 return res; 1102} 1103 1104TSAN_INTERCEPTOR(int, pthread_barrier_wait, void *b) { 1105 SCOPED_TSAN_INTERCEPTOR(pthread_barrier_wait, b); 1106 Release(thr, pc, (uptr)b); 1107 MemoryRead(thr, pc, (uptr)b, kSizeLog1); 1108 int res = REAL(pthread_barrier_wait)(b); 1109 MemoryRead(thr, pc, (uptr)b, kSizeLog1); 1110 if (res == 0 || res == PTHREAD_BARRIER_SERIAL_THREAD) { 1111 Acquire(thr, pc, (uptr)b); 1112 } 1113 return res; 1114} 1115 1116TSAN_INTERCEPTOR(int, pthread_once, void *o, void (*f)()) { 1117 SCOPED_TSAN_INTERCEPTOR(pthread_once, o, f); 1118 if (o == 0 || f == 0) 1119 return EINVAL; 1120 atomic_uint32_t *a = static_cast<atomic_uint32_t*>(o); 1121 u32 v = atomic_load(a, memory_order_acquire); 1122 if (v == 0 && atomic_compare_exchange_strong(a, &v, 1, 1123 memory_order_relaxed)) { 1124 const int old_in_rtl = thr->in_rtl; 1125 thr->in_rtl = 0; 1126 (*f)(); 1127 CHECK_EQ(thr->in_rtl, 0); 1128 thr->in_rtl = old_in_rtl; 1129 Release(thr, pc, (uptr)o); 1130 atomic_store(a, 2, memory_order_release); 1131 } else { 1132 while (v != 2) { 1133 pthread_yield(); 1134 v = atomic_load(a, memory_order_acquire); 1135 } 1136 Acquire(thr, pc, (uptr)o); 1137 } 1138 return 0; 1139} 1140 1141TSAN_INTERCEPTOR(int, sem_init, void *s, int pshared, unsigned value) { 1142 SCOPED_TSAN_INTERCEPTOR(sem_init, s, pshared, value); 1143 int res = REAL(sem_init)(s, pshared, value); 1144 return res; 1145} 1146 1147TSAN_INTERCEPTOR(int, sem_destroy, void *s) { 1148 SCOPED_TSAN_INTERCEPTOR(sem_destroy, s); 1149 int res = REAL(sem_destroy)(s); 1150 return res; 1151} 1152 1153TSAN_INTERCEPTOR(int, sem_wait, void *s) { 1154 SCOPED_TSAN_INTERCEPTOR(sem_wait, s); 1155 int res = BLOCK_REAL(sem_wait)(s); 1156 if (res == 0) { 1157 Acquire(thr, pc, (uptr)s); 1158 } 1159 return res; 1160} 1161 1162TSAN_INTERCEPTOR(int, sem_trywait, void *s) { 1163 SCOPED_TSAN_INTERCEPTOR(sem_trywait, s); 1164 int res = BLOCK_REAL(sem_trywait)(s); 1165 if (res == 0) { 1166 Acquire(thr, pc, (uptr)s); 1167 } 1168 return res; 1169} 1170 1171TSAN_INTERCEPTOR(int, sem_timedwait, void *s, void *abstime) { 1172 SCOPED_TSAN_INTERCEPTOR(sem_timedwait, s, abstime); 1173 int res = BLOCK_REAL(sem_timedwait)(s, abstime); 1174 if (res == 0) { 1175 Acquire(thr, pc, (uptr)s); 1176 } 1177 return res; 1178} 1179 1180TSAN_INTERCEPTOR(int, sem_post, void *s) { 1181 SCOPED_TSAN_INTERCEPTOR(sem_post, s); 1182 Release(thr, pc, (uptr)s); 1183 int res = REAL(sem_post)(s); 1184 return res; 1185} 1186 1187TSAN_INTERCEPTOR(int, sem_getvalue, void *s, int *sval) { 1188 SCOPED_TSAN_INTERCEPTOR(sem_getvalue, s, sval); 1189 int res = REAL(sem_getvalue)(s, sval); 1190 if (res == 0) { 1191 Acquire(thr, pc, (uptr)s); 1192 } 1193 return res; 1194} 1195 1196TSAN_INTERCEPTOR(int, __xstat, int version, const char *path, void *buf) { 1197 SCOPED_TSAN_INTERCEPTOR(__xstat, version, path, buf); 1198 return REAL(__xstat)(version, path, buf); 1199} 1200 1201TSAN_INTERCEPTOR(int, stat, const char *path, void *buf) { 1202 SCOPED_TSAN_INTERCEPTOR(__xstat, 0, path, buf); 1203 return REAL(__xstat)(0, path, buf); 1204} 1205 1206TSAN_INTERCEPTOR(int, __xstat64, int version, const char *path, void *buf) { 1207 SCOPED_TSAN_INTERCEPTOR(__xstat64, version, path, buf); 1208 return REAL(__xstat64)(version, path, buf); 1209} 1210 1211TSAN_INTERCEPTOR(int, stat64, const char *path, void *buf) { 1212 SCOPED_TSAN_INTERCEPTOR(__xstat64, 0, path, buf); 1213 return REAL(__xstat64)(0, path, buf); 1214} 1215 1216TSAN_INTERCEPTOR(int, __lxstat, int version, const char *path, void *buf) { 1217 SCOPED_TSAN_INTERCEPTOR(__lxstat, version, path, buf); 1218 return REAL(__lxstat)(version, path, buf); 1219} 1220 1221TSAN_INTERCEPTOR(int, lstat, const char *path, void *buf) { 1222 SCOPED_TSAN_INTERCEPTOR(__lxstat, 0, path, buf); 1223 return REAL(__lxstat)(0, path, buf); 1224} 1225 1226TSAN_INTERCEPTOR(int, __lxstat64, int version, const char *path, void *buf) { 1227 SCOPED_TSAN_INTERCEPTOR(__lxstat64, version, path, buf); 1228 return REAL(__lxstat64)(version, path, buf); 1229} 1230 1231TSAN_INTERCEPTOR(int, lstat64, const char *path, void *buf) { 1232 SCOPED_TSAN_INTERCEPTOR(__lxstat64, 0, path, buf); 1233 return REAL(__lxstat64)(0, path, buf); 1234} 1235 1236TSAN_INTERCEPTOR(int, __fxstat, int version, int fd, void *buf) { 1237 SCOPED_TSAN_INTERCEPTOR(__fxstat, version, fd, buf); 1238 if (fd > 0) 1239 FdAccess(thr, pc, fd); 1240 return REAL(__fxstat)(version, fd, buf); 1241} 1242 1243TSAN_INTERCEPTOR(int, fstat, int fd, void *buf) { 1244 SCOPED_TSAN_INTERCEPTOR(__fxstat, 0, fd, buf); 1245 if (fd > 0) 1246 FdAccess(thr, pc, fd); 1247 return REAL(__fxstat)(0, fd, buf); 1248} 1249 1250TSAN_INTERCEPTOR(int, __fxstat64, int version, int fd, void *buf) { 1251 SCOPED_TSAN_INTERCEPTOR(__fxstat64, version, fd, buf); 1252 if (fd > 0) 1253 FdAccess(thr, pc, fd); 1254 return REAL(__fxstat64)(version, fd, buf); 1255} 1256 1257TSAN_INTERCEPTOR(int, fstat64, int fd, void *buf) { 1258 SCOPED_TSAN_INTERCEPTOR(__fxstat64, 0, fd, buf); 1259 if (fd > 0) 1260 FdAccess(thr, pc, fd); 1261 return REAL(__fxstat64)(0, fd, buf); 1262} 1263 1264TSAN_INTERCEPTOR(int, open, const char *name, int flags, int mode) { 1265 SCOPED_TSAN_INTERCEPTOR(open, name, flags, mode); 1266 int fd = REAL(open)(name, flags, mode); 1267 if (fd >= 0) 1268 FdFileCreate(thr, pc, fd); 1269 return fd; 1270} 1271 1272TSAN_INTERCEPTOR(int, open64, const char *name, int flags, int mode) { 1273 SCOPED_TSAN_INTERCEPTOR(open64, name, flags, mode); 1274 int fd = REAL(open64)(name, flags, mode); 1275 if (fd >= 0) 1276 FdFileCreate(thr, pc, fd); 1277 return fd; 1278} 1279 1280TSAN_INTERCEPTOR(int, creat, const char *name, int mode) { 1281 SCOPED_TSAN_INTERCEPTOR(creat, name, mode); 1282 int fd = REAL(creat)(name, mode); 1283 if (fd >= 0) 1284 FdFileCreate(thr, pc, fd); 1285 return fd; 1286} 1287 1288TSAN_INTERCEPTOR(int, creat64, const char *name, int mode) { 1289 SCOPED_TSAN_INTERCEPTOR(creat64, name, mode); 1290 int fd = REAL(creat64)(name, mode); 1291 if (fd >= 0) 1292 FdFileCreate(thr, pc, fd); 1293 return fd; 1294} 1295 1296TSAN_INTERCEPTOR(int, dup, int oldfd) { 1297 SCOPED_TSAN_INTERCEPTOR(dup, oldfd); 1298 int newfd = REAL(dup)(oldfd); 1299 if (oldfd >= 0 && newfd >= 0 && newfd != oldfd) 1300 FdDup(thr, pc, oldfd, newfd); 1301 return newfd; 1302} 1303 1304TSAN_INTERCEPTOR(int, dup2, int oldfd, int newfd) { 1305 SCOPED_TSAN_INTERCEPTOR(dup2, oldfd, newfd); 1306 int newfd2 = REAL(dup2)(oldfd, newfd); 1307 if (oldfd >= 0 && newfd2 >= 0 && newfd2 != oldfd) 1308 FdDup(thr, pc, oldfd, newfd2); 1309 return newfd2; 1310} 1311 1312TSAN_INTERCEPTOR(int, dup3, int oldfd, int newfd, int flags) { 1313 SCOPED_TSAN_INTERCEPTOR(dup3, oldfd, newfd, flags); 1314 int newfd2 = REAL(dup3)(oldfd, newfd, flags); 1315 if (oldfd >= 0 && newfd2 >= 0 && newfd2 != oldfd) 1316 FdDup(thr, pc, oldfd, newfd2); 1317 return newfd2; 1318} 1319 1320TSAN_INTERCEPTOR(int, eventfd, unsigned initval, int flags) { 1321 SCOPED_TSAN_INTERCEPTOR(eventfd, initval, flags); 1322 int fd = REAL(eventfd)(initval, flags); 1323 if (fd >= 0) 1324 FdEventCreate(thr, pc, fd); 1325 return fd; 1326} 1327 1328TSAN_INTERCEPTOR(int, signalfd, int fd, void *mask, int flags) { 1329 SCOPED_TSAN_INTERCEPTOR(signalfd, fd, mask, flags); 1330 if (fd >= 0) 1331 FdClose(thr, pc, fd); 1332 fd = REAL(signalfd)(fd, mask, flags); 1333 if (fd >= 0) 1334 FdSignalCreate(thr, pc, fd); 1335 return fd; 1336} 1337 1338TSAN_INTERCEPTOR(int, inotify_init, int fake) { 1339 SCOPED_TSAN_INTERCEPTOR(inotify_init, fake); 1340 int fd = REAL(inotify_init)(fake); 1341 if (fd >= 0) 1342 FdInotifyCreate(thr, pc, fd); 1343 return fd; 1344} 1345 1346TSAN_INTERCEPTOR(int, inotify_init1, int flags) { 1347 SCOPED_TSAN_INTERCEPTOR(inotify_init1, flags); 1348 int fd = REAL(inotify_init1)(flags); 1349 if (fd >= 0) 1350 FdInotifyCreate(thr, pc, fd); 1351 return fd; 1352} 1353 1354TSAN_INTERCEPTOR(int, socket, int domain, int type, int protocol) { 1355 SCOPED_TSAN_INTERCEPTOR(socket, domain, type, protocol); 1356 int fd = REAL(socket)(domain, type, protocol); 1357 if (fd >= 0) 1358 FdSocketCreate(thr, pc, fd); 1359 return fd; 1360} 1361 1362TSAN_INTERCEPTOR(int, socketpair, int domain, int type, int protocol, int *fd) { 1363 SCOPED_TSAN_INTERCEPTOR(socketpair, domain, type, protocol, fd); 1364 int res = REAL(socketpair)(domain, type, protocol, fd); 1365 if (res == 0 && fd[0] >= 0 && fd[1] >= 0) 1366 FdPipeCreate(thr, pc, fd[0], fd[1]); 1367 return res; 1368} 1369 1370TSAN_INTERCEPTOR(int, connect, int fd, void *addr, unsigned addrlen) { 1371 SCOPED_TSAN_INTERCEPTOR(connect, fd, addr, addrlen); 1372 FdSocketConnecting(thr, pc, fd); 1373 int res = REAL(connect)(fd, addr, addrlen); 1374 if (res == 0 && fd >= 0) 1375 FdSocketConnect(thr, pc, fd); 1376 return res; 1377} 1378 1379TSAN_INTERCEPTOR(int, bind, int fd, void *addr, unsigned addrlen) { 1380 SCOPED_TSAN_INTERCEPTOR(bind, fd, addr, addrlen); 1381 int res = REAL(bind)(fd, addr, addrlen); 1382 if (fd > 0 && res == 0) 1383 FdAccess(thr, pc, fd); 1384 return res; 1385} 1386 1387TSAN_INTERCEPTOR(int, listen, int fd, int backlog) { 1388 SCOPED_TSAN_INTERCEPTOR(listen, fd, backlog); 1389 int res = REAL(listen)(fd, backlog); 1390 if (fd > 0 && res == 0) 1391 FdAccess(thr, pc, fd); 1392 return res; 1393} 1394 1395TSAN_INTERCEPTOR(int, epoll_create, int size) { 1396 SCOPED_TSAN_INTERCEPTOR(epoll_create, size); 1397 int fd = REAL(epoll_create)(size); 1398 if (fd >= 0) 1399 FdPollCreate(thr, pc, fd); 1400 return fd; 1401} 1402 1403TSAN_INTERCEPTOR(int, epoll_create1, int flags) { 1404 SCOPED_TSAN_INTERCEPTOR(epoll_create1, flags); 1405 int fd = REAL(epoll_create1)(flags); 1406 if (fd >= 0) 1407 FdPollCreate(thr, pc, fd); 1408 return fd; 1409} 1410 1411TSAN_INTERCEPTOR(int, close, int fd) { 1412 SCOPED_TSAN_INTERCEPTOR(close, fd); 1413 if (fd >= 0) 1414 FdClose(thr, pc, fd); 1415 return REAL(close)(fd); 1416} 1417 1418TSAN_INTERCEPTOR(int, __close, int fd) { 1419 SCOPED_TSAN_INTERCEPTOR(__close, fd); 1420 if (fd >= 0) 1421 FdClose(thr, pc, fd); 1422 return REAL(__close)(fd); 1423} 1424 1425// glibc guts 1426TSAN_INTERCEPTOR(void, __res_iclose, void *state, bool free_addr) { 1427 SCOPED_TSAN_INTERCEPTOR(__res_iclose, state, free_addr); 1428 int fds[64]; 1429 int cnt = ExtractResolvFDs(state, fds, ARRAY_SIZE(fds)); 1430 for (int i = 0; i < cnt; i++) { 1431 if (fds[i] > 0) 1432 FdClose(thr, pc, fds[i]); 1433 } 1434 REAL(__res_iclose)(state, free_addr); 1435} 1436 1437TSAN_INTERCEPTOR(int, pipe, int *pipefd) { 1438 SCOPED_TSAN_INTERCEPTOR(pipe, pipefd); 1439 int res = REAL(pipe)(pipefd); 1440 if (res == 0 && pipefd[0] >= 0 && pipefd[1] >= 0) 1441 FdPipeCreate(thr, pc, pipefd[0], pipefd[1]); 1442 return res; 1443} 1444 1445TSAN_INTERCEPTOR(int, pipe2, int *pipefd, int flags) { 1446 SCOPED_TSAN_INTERCEPTOR(pipe2, pipefd, flags); 1447 int res = REAL(pipe2)(pipefd, flags); 1448 if (res == 0 && pipefd[0] >= 0 && pipefd[1] >= 0) 1449 FdPipeCreate(thr, pc, pipefd[0], pipefd[1]); 1450 return res; 1451} 1452 1453TSAN_INTERCEPTOR(long_t, readv, int fd, void *vec, int cnt) { 1454 SCOPED_TSAN_INTERCEPTOR(readv, fd, vec, cnt); 1455 int res = REAL(readv)(fd, vec, cnt); 1456 if (res >= 0 && fd >= 0) { 1457 FdAcquire(thr, pc, fd); 1458 } 1459 return res; 1460} 1461 1462TSAN_INTERCEPTOR(long_t, preadv64, int fd, void *vec, int cnt, u64 off) { 1463 SCOPED_TSAN_INTERCEPTOR(preadv64, fd, vec, cnt, off); 1464 int res = REAL(preadv64)(fd, vec, cnt, off); 1465 if (res >= 0 && fd >= 0) { 1466 FdAcquire(thr, pc, fd); 1467 } 1468 return res; 1469} 1470 1471TSAN_INTERCEPTOR(long_t, writev, int fd, void *vec, int cnt) { 1472 SCOPED_TSAN_INTERCEPTOR(writev, fd, vec, cnt); 1473 if (fd >= 0) 1474 FdRelease(thr, pc, fd); 1475 int res = REAL(writev)(fd, vec, cnt); 1476 return res; 1477} 1478 1479TSAN_INTERCEPTOR(long_t, pwritev64, int fd, void *vec, int cnt, u64 off) { 1480 SCOPED_TSAN_INTERCEPTOR(pwritev64, fd, vec, cnt, off); 1481 if (fd >= 0) 1482 FdRelease(thr, pc, fd); 1483 int res = REAL(pwritev64)(fd, vec, cnt, off); 1484 return res; 1485} 1486 1487TSAN_INTERCEPTOR(long_t, send, int fd, void *buf, long_t len, int flags) { 1488 SCOPED_TSAN_INTERCEPTOR(send, fd, buf, len, flags); 1489 if (fd >= 0) 1490 FdRelease(thr, pc, fd); 1491 int res = REAL(send)(fd, buf, len, flags); 1492 return res; 1493} 1494 1495TSAN_INTERCEPTOR(long_t, sendmsg, int fd, void *msg, int flags) { 1496 SCOPED_TSAN_INTERCEPTOR(sendmsg, fd, msg, flags); 1497 if (fd >= 0) 1498 FdRelease(thr, pc, fd); 1499 int res = REAL(sendmsg)(fd, msg, flags); 1500 return res; 1501} 1502 1503TSAN_INTERCEPTOR(long_t, recv, int fd, void *buf, long_t len, int flags) { 1504 SCOPED_TSAN_INTERCEPTOR(recv, fd, buf, len, flags); 1505 int res = REAL(recv)(fd, buf, len, flags); 1506 if (res >= 0 && fd >= 0) { 1507 FdAcquire(thr, pc, fd); 1508 } 1509 return res; 1510} 1511 1512TSAN_INTERCEPTOR(int, unlink, char *path) { 1513 SCOPED_TSAN_INTERCEPTOR(unlink, path); 1514 Release(thr, pc, File2addr(path)); 1515 int res = REAL(unlink)(path); 1516 return res; 1517} 1518 1519TSAN_INTERCEPTOR(void*, fopen, char *path, char *mode) { 1520 SCOPED_TSAN_INTERCEPTOR(fopen, path, mode); 1521 void *res = REAL(fopen)(path, mode); 1522 Acquire(thr, pc, File2addr(path)); 1523 if (res) { 1524 int fd = fileno_unlocked(res); 1525 if (fd >= 0) 1526 FdFileCreate(thr, pc, fd); 1527 } 1528 return res; 1529} 1530 1531TSAN_INTERCEPTOR(void*, freopen, char *path, char *mode, void *stream) { 1532 SCOPED_TSAN_INTERCEPTOR(freopen, path, mode, stream); 1533 if (stream) { 1534 int fd = fileno_unlocked(stream); 1535 if (fd >= 0) 1536 FdClose(thr, pc, fd); 1537 } 1538 void *res = REAL(freopen)(path, mode, stream); 1539 Acquire(thr, pc, File2addr(path)); 1540 if (res) { 1541 int fd = fileno_unlocked(res); 1542 if (fd >= 0) 1543 FdFileCreate(thr, pc, fd); 1544 } 1545 return res; 1546} 1547 1548TSAN_INTERCEPTOR(int, fclose, void *stream) { 1549 { 1550 SCOPED_TSAN_INTERCEPTOR(fclose, stream); 1551 if (stream) { 1552 int fd = fileno_unlocked(stream); 1553 if (fd >= 0) 1554 FdClose(thr, pc, fd); 1555 } 1556 } 1557 return REAL(fclose)(stream); 1558} 1559 1560TSAN_INTERCEPTOR(uptr, fread, void *ptr, uptr size, uptr nmemb, void *f) { 1561 { 1562 SCOPED_TSAN_INTERCEPTOR(fread, ptr, size, nmemb, f); 1563 MemoryAccessRange(thr, pc, (uptr)ptr, size * nmemb, true); 1564 } 1565 return REAL(fread)(ptr, size, nmemb, f); 1566} 1567 1568TSAN_INTERCEPTOR(uptr, fwrite, const void *p, uptr size, uptr nmemb, void *f) { 1569 { 1570 SCOPED_TSAN_INTERCEPTOR(fwrite, p, size, nmemb, f); 1571 MemoryAccessRange(thr, pc, (uptr)p, size * nmemb, false); 1572 } 1573 return REAL(fwrite)(p, size, nmemb, f); 1574} 1575 1576TSAN_INTERCEPTOR(int, fflush, void *stream) { 1577 SCOPED_TSAN_INTERCEPTOR(fflush, stream); 1578 return REAL(fflush)(stream); 1579} 1580 1581TSAN_INTERCEPTOR(void, abort, int fake) { 1582 SCOPED_TSAN_INTERCEPTOR(abort, fake); 1583 REAL(fflush)(0); 1584 REAL(abort)(fake); 1585} 1586 1587TSAN_INTERCEPTOR(int, puts, const char *s) { 1588 SCOPED_TSAN_INTERCEPTOR(puts, s); 1589 MemoryAccessRange(thr, pc, (uptr)s, internal_strlen(s), false); 1590 return REAL(puts)(s); 1591} 1592 1593TSAN_INTERCEPTOR(int, rmdir, char *path) { 1594 SCOPED_TSAN_INTERCEPTOR(rmdir, path); 1595 Release(thr, pc, Dir2addr(path)); 1596 int res = REAL(rmdir)(path); 1597 return res; 1598} 1599 1600TSAN_INTERCEPTOR(void*, opendir, char *path) { 1601 SCOPED_TSAN_INTERCEPTOR(opendir, path); 1602 void *res = REAL(opendir)(path); 1603 if (res != 0) 1604 Acquire(thr, pc, Dir2addr(path)); 1605 return res; 1606} 1607 1608TSAN_INTERCEPTOR(int, epoll_ctl, int epfd, int op, int fd, void *ev) { 1609 SCOPED_TSAN_INTERCEPTOR(epoll_ctl, epfd, op, fd, ev); 1610 if (op == EPOLL_CTL_ADD && epfd >= 0) { 1611 FdRelease(thr, pc, epfd); 1612 } 1613 int res = REAL(epoll_ctl)(epfd, op, fd, ev); 1614 if (fd >= 0) 1615 FdAccess(thr, pc, fd); 1616 return res; 1617} 1618 1619TSAN_INTERCEPTOR(int, epoll_wait, int epfd, void *ev, int cnt, int timeout) { 1620 SCOPED_TSAN_INTERCEPTOR(epoll_wait, epfd, ev, cnt, timeout); 1621 int res = BLOCK_REAL(epoll_wait)(epfd, ev, cnt, timeout); 1622 if (res > 0 && epfd >= 0) { 1623 FdAcquire(thr, pc, epfd); 1624 } 1625 return res; 1626} 1627 1628TSAN_INTERCEPTOR(int, poll, void *fds, long_t nfds, int timeout) { 1629 SCOPED_TSAN_INTERCEPTOR(poll, fds, nfds, timeout); 1630 int res = BLOCK_REAL(poll)(fds, nfds, timeout); 1631 return res; 1632} 1633 1634void ALWAYS_INLINE rtl_generic_sighandler(bool sigact, int sig, 1635 my_siginfo_t *info, void *ctx) { 1636 ThreadState *thr = cur_thread(); 1637 SignalContext *sctx = SigCtx(thr); 1638 // Don't mess with synchronous signals. 1639 if (sig == SIGSEGV || sig == SIGBUS || sig == SIGILL || 1640 sig == SIGABRT || sig == SIGFPE || sig == SIGPIPE || 1641 // If we are sending signal to ourselves, we must process it now. 1642 (sctx && sig == sctx->int_signal_send) || 1643 // If we are in blocking function, we can safely process it now 1644 // (but check if we are in a recursive interceptor, 1645 // i.e. pthread_join()->munmap()). 1646 (sctx && sctx->in_blocking_func == 1 && thr->in_rtl == 1)) { 1647 int in_rtl = thr->in_rtl; 1648 thr->in_rtl = 0; 1649 CHECK_EQ(thr->in_signal_handler, false); 1650 thr->in_signal_handler = true; 1651 if (sigact) 1652 sigactions[sig].sa_sigaction(sig, info, ctx); 1653 else 1654 sigactions[sig].sa_handler(sig); 1655 CHECK_EQ(thr->in_signal_handler, true); 1656 thr->in_signal_handler = false; 1657 thr->in_rtl = in_rtl; 1658 return; 1659 } 1660 1661 if (sctx == 0) 1662 return; 1663 SignalDesc *signal = &sctx->pending_signals[sig]; 1664 if (signal->armed == false) { 1665 signal->armed = true; 1666 signal->sigaction = sigact; 1667 if (info) 1668 internal_memcpy(&signal->siginfo, info, sizeof(*info)); 1669 if (ctx) 1670 internal_memcpy(&signal->ctx, ctx, sizeof(signal->ctx)); 1671 sctx->pending_signal_count++; 1672 } 1673} 1674 1675static void rtl_sighandler(int sig) { 1676 rtl_generic_sighandler(false, sig, 0, 0); 1677} 1678 1679static void rtl_sigaction(int sig, my_siginfo_t *info, void *ctx) { 1680 rtl_generic_sighandler(true, sig, info, ctx); 1681} 1682 1683TSAN_INTERCEPTOR(int, sigaction, int sig, sigaction_t *act, sigaction_t *old) { 1684 SCOPED_TSAN_INTERCEPTOR(sigaction, sig, act, old); 1685 if (old) 1686 internal_memcpy(old, &sigactions[sig], sizeof(*old)); 1687 if (act == 0) 1688 return 0; 1689 internal_memcpy(&sigactions[sig], act, sizeof(*act)); 1690 sigaction_t newact; 1691 internal_memcpy(&newact, act, sizeof(newact)); 1692 sigfillset(&newact.sa_mask); 1693 if (act->sa_handler != SIG_IGN && act->sa_handler != SIG_DFL) { 1694 if (newact.sa_flags & SA_SIGINFO) 1695 newact.sa_sigaction = rtl_sigaction; 1696 else 1697 newact.sa_handler = rtl_sighandler; 1698 } 1699 int res = REAL(sigaction)(sig, &newact, 0); 1700 return res; 1701} 1702 1703TSAN_INTERCEPTOR(sighandler_t, signal, int sig, sighandler_t h) { 1704 sigaction_t act; 1705 act.sa_handler = h; 1706 REAL(memset)(&act.sa_mask, -1, sizeof(act.sa_mask)); 1707 act.sa_flags = 0; 1708 sigaction_t old; 1709 int res = sigaction(sig, &act, &old); 1710 if (res) 1711 return SIG_ERR; 1712 return old.sa_handler; 1713} 1714 1715TSAN_INTERCEPTOR(int, raise, int sig) { 1716 SCOPED_TSAN_INTERCEPTOR(raise, sig); 1717 SignalContext *sctx = SigCtx(thr); 1718 CHECK_NE(sctx, 0); 1719 int prev = sctx->int_signal_send; 1720 sctx->int_signal_send = sig; 1721 int res = REAL(raise)(sig); 1722 CHECK_EQ(sctx->int_signal_send, sig); 1723 sctx->int_signal_send = prev; 1724 return res; 1725} 1726 1727TSAN_INTERCEPTOR(int, kill, int pid, int sig) { 1728 SCOPED_TSAN_INTERCEPTOR(kill, pid, sig); 1729 SignalContext *sctx = SigCtx(thr); 1730 CHECK_NE(sctx, 0); 1731 int prev = sctx->int_signal_send; 1732 if (pid == (int)internal_getpid()) { 1733 sctx->int_signal_send = sig; 1734 } 1735 int res = REAL(kill)(pid, sig); 1736 if (pid == (int)internal_getpid()) { 1737 CHECK_EQ(sctx->int_signal_send, sig); 1738 sctx->int_signal_send = prev; 1739 } 1740 return res; 1741} 1742 1743TSAN_INTERCEPTOR(int, pthread_kill, void *tid, int sig) { 1744 SCOPED_TSAN_INTERCEPTOR(pthread_kill, tid, sig); 1745 SignalContext *sctx = SigCtx(thr); 1746 CHECK_NE(sctx, 0); 1747 int prev = sctx->int_signal_send; 1748 if (tid == pthread_self()) { 1749 sctx->int_signal_send = sig; 1750 } 1751 int res = REAL(pthread_kill)(tid, sig); 1752 if (tid == pthread_self()) { 1753 CHECK_EQ(sctx->int_signal_send, sig); 1754 sctx->int_signal_send = prev; 1755 } 1756 return res; 1757} 1758 1759TSAN_INTERCEPTOR(int, gettimeofday, void *tv, void *tz) { 1760 SCOPED_TSAN_INTERCEPTOR(gettimeofday, tv, tz); 1761 // It's intercepted merely to process pending signals. 1762 return REAL(gettimeofday)(tv, tz); 1763} 1764 1765// Linux kernel has a bug that leads to kernel deadlock if a process 1766// maps TBs of memory and then calls mlock(). 1767static void MlockIsUnsupported() { 1768 static atomic_uint8_t printed; 1769 if (atomic_exchange(&printed, 1, memory_order_relaxed)) 1770 return; 1771 Printf("INFO: ThreadSanitizer ignores mlock/mlockall/munlock/munlockall\n"); 1772} 1773 1774TSAN_INTERCEPTOR(int, mlock, const void *addr, uptr len) { 1775 MlockIsUnsupported(); 1776 return 0; 1777} 1778 1779TSAN_INTERCEPTOR(int, munlock, const void *addr, uptr len) { 1780 MlockIsUnsupported(); 1781 return 0; 1782} 1783 1784TSAN_INTERCEPTOR(int, mlockall, int flags) { 1785 MlockIsUnsupported(); 1786 return 0; 1787} 1788 1789TSAN_INTERCEPTOR(int, munlockall, void) { 1790 MlockIsUnsupported(); 1791 return 0; 1792} 1793 1794TSAN_INTERCEPTOR(int, fork, int fake) { 1795 SCOPED_TSAN_INTERCEPTOR(fork, fake); 1796 // It's intercepted merely to process pending signals. 1797 int pid = REAL(fork)(fake); 1798 if (pid == 0) { 1799 // child 1800 FdOnFork(thr, pc); 1801 } else if (pid > 0) { 1802 // parent 1803 } 1804 return pid; 1805} 1806 1807struct TsanInterceptorContext { 1808 ThreadState *thr; 1809 const uptr caller_pc; 1810 const uptr pc; 1811}; 1812 1813#include "sanitizer_common/sanitizer_platform_interceptors.h" 1814// Causes interceptor recursion (getpwuid_r() calls fopen()) 1815#undef SANITIZER_INTERCEPT_GETPWNAM_AND_FRIENDS 1816#undef SANITIZER_INTERCEPT_GETPWNAM_R_AND_FRIENDS 1817// Causes interceptor recursion (getaddrinfo() and fopen()) 1818#undef SANITIZER_INTERCEPT_GETADDRINFO 1819// Causes interceptor recursion (glob64() calls lstat64()) 1820#undef SANITIZER_INTERCEPT_GLOB 1821 1822#define COMMON_INTERCEPTOR_WRITE_RANGE(ctx, ptr, size) \ 1823 MemoryAccessRange(((TsanInterceptorContext *) ctx)->thr, \ 1824 ((TsanInterceptorContext *) ctx)->pc, (uptr) ptr, size, \ 1825 true) 1826#define COMMON_INTERCEPTOR_READ_RANGE(ctx, ptr, size) \ 1827 MemoryAccessRange(((TsanInterceptorContext *) ctx)->thr, \ 1828 ((TsanInterceptorContext *) ctx)->pc, (uptr) ptr, size, \ 1829 false) 1830#define COMMON_INTERCEPTOR_ENTER(ctx, func, ...) \ 1831 SCOPED_TSAN_INTERCEPTOR(func, __VA_ARGS__); \ 1832 TsanInterceptorContext _ctx = {thr, caller_pc, pc}; \ 1833 ctx = (void *)&_ctx; \ 1834 (void) ctx; 1835#define COMMON_INTERCEPTOR_FD_ACQUIRE(ctx, fd) \ 1836 FdAcquire(((TsanInterceptorContext *) ctx)->thr, pc, fd) 1837#define COMMON_INTERCEPTOR_FD_RELEASE(ctx, fd) \ 1838 FdRelease(((TsanInterceptorContext *) ctx)->thr, pc, fd) 1839#define COMMON_INTERCEPTOR_FD_SOCKET_ACCEPT(ctx, fd, newfd) \ 1840 FdSocketAccept(((TsanInterceptorContext *) ctx)->thr, pc, fd, newfd) 1841#define COMMON_INTERCEPTOR_SET_THREAD_NAME(ctx, name) \ 1842 ThreadSetName(((TsanInterceptorContext *) ctx)->thr, name) 1843#include "sanitizer_common/sanitizer_common_interceptors.inc" 1844 1845// FIXME: Implement these with MemoryAccessRange(). 1846#define COMMON_SYSCALL_PRE_READ_RANGE(p, s) 1847#define COMMON_SYSCALL_PRE_WRITE_RANGE(p, s) 1848#define COMMON_SYSCALL_POST_READ_RANGE(p, s) 1849#define COMMON_SYSCALL_POST_WRITE_RANGE(p, s) 1850#include "sanitizer_common/sanitizer_common_syscalls.inc" 1851 1852namespace __tsan { 1853 1854void ProcessPendingSignals(ThreadState *thr) { 1855 CHECK_EQ(thr->in_rtl, 0); 1856 SignalContext *sctx = SigCtx(thr); 1857 if (sctx == 0 || sctx->pending_signal_count == 0 || thr->in_signal_handler) 1858 return; 1859 Context *ctx = CTX(); 1860 thr->in_signal_handler = true; 1861 sctx->pending_signal_count = 0; 1862 // These are too big for stack. 1863 static THREADLOCAL sigset_t emptyset, oldset; 1864 sigfillset(&emptyset); 1865 pthread_sigmask(SIG_SETMASK, &emptyset, &oldset); 1866 for (int sig = 0; sig < kSigCount; sig++) { 1867 SignalDesc *signal = &sctx->pending_signals[sig]; 1868 if (signal->armed) { 1869 signal->armed = false; 1870 if (sigactions[sig].sa_handler != SIG_DFL 1871 && sigactions[sig].sa_handler != SIG_IGN) { 1872 // Insure that the handler does not spoil errno. 1873 const int saved_errno = errno; 1874 errno = 0; 1875 if (signal->sigaction) 1876 sigactions[sig].sa_sigaction(sig, &signal->siginfo, &signal->ctx); 1877 else 1878 sigactions[sig].sa_handler(sig); 1879 if (flags()->report_bugs && errno != 0) { 1880 ScopedInRtl in_rtl; 1881 __tsan::StackTrace stack; 1882 uptr pc = signal->sigaction ? 1883 (uptr)sigactions[sig].sa_sigaction : 1884 (uptr)sigactions[sig].sa_handler; 1885 stack.Init(&pc, 1); 1886 ThreadRegistryLock l(ctx->thread_registry); 1887 ScopedReport rep(ReportTypeErrnoInSignal); 1888 if (!IsFiredSuppression(ctx, rep, stack)) { 1889 rep.AddStack(&stack); 1890 OutputReport(ctx, rep, rep.GetReport()->stacks[0]); 1891 } 1892 } 1893 errno = saved_errno; 1894 } 1895 } 1896 } 1897 pthread_sigmask(SIG_SETMASK, &oldset, 0); 1898 CHECK_EQ(thr->in_signal_handler, true); 1899 thr->in_signal_handler = false; 1900} 1901 1902static void finalize(void *arg) { 1903 ThreadState * thr = cur_thread(); 1904 uptr pc = 0; 1905 atexit_ctx->exit(thr, pc); 1906 int status = Finalize(cur_thread()); 1907 REAL(fflush)(0); 1908 if (status) 1909 _exit(status); 1910} 1911 1912static void unreachable() { 1913 Printf("FATAL: ThreadSanitizer: unreachable called\n"); 1914 Die(); 1915} 1916 1917void InitializeInterceptors() { 1918 CHECK_GT(cur_thread()->in_rtl, 0); 1919 1920 // We need to setup it early, because functions like dlsym() can call it. 1921 REAL(memset) = internal_memset; 1922 REAL(memcpy) = internal_memcpy; 1923 REAL(memcmp) = internal_memcmp; 1924 1925 // Instruct libc malloc to consume less memory. 1926 mallopt(1, 0); // M_MXFAST 1927 mallopt(-3, 32*1024); // M_MMAP_THRESHOLD 1928 1929 SANITIZER_COMMON_INTERCEPTORS_INIT; 1930 1931 TSAN_INTERCEPT(setjmp); 1932 TSAN_INTERCEPT(_setjmp); 1933 TSAN_INTERCEPT(sigsetjmp); 1934 TSAN_INTERCEPT(__sigsetjmp); 1935 TSAN_INTERCEPT(longjmp); 1936 TSAN_INTERCEPT(siglongjmp); 1937 1938 TSAN_INTERCEPT(malloc); 1939 TSAN_INTERCEPT(__libc_memalign); 1940 TSAN_INTERCEPT(calloc); 1941 TSAN_INTERCEPT(realloc); 1942 TSAN_INTERCEPT(free); 1943 TSAN_INTERCEPT(cfree); 1944 TSAN_INTERCEPT(mmap); 1945 TSAN_INTERCEPT(mmap64); 1946 TSAN_INTERCEPT(munmap); 1947 TSAN_INTERCEPT(memalign); 1948 TSAN_INTERCEPT(valloc); 1949 TSAN_INTERCEPT(pvalloc); 1950 TSAN_INTERCEPT(posix_memalign); 1951 1952 TSAN_INTERCEPT(strlen); 1953 TSAN_INTERCEPT(memset); 1954 TSAN_INTERCEPT(memcpy); 1955 TSAN_INTERCEPT(strcmp); 1956 TSAN_INTERCEPT(memchr); 1957 TSAN_INTERCEPT(memrchr); 1958 TSAN_INTERCEPT(memmove); 1959 TSAN_INTERCEPT(memcmp); 1960 TSAN_INTERCEPT(strchr); 1961 TSAN_INTERCEPT(strchrnul); 1962 TSAN_INTERCEPT(strrchr); 1963 TSAN_INTERCEPT(strncmp); 1964 TSAN_INTERCEPT(strcpy); // NOLINT 1965 TSAN_INTERCEPT(strncpy); 1966 TSAN_INTERCEPT(strstr); 1967 1968 TSAN_INTERCEPT(pthread_create); 1969 TSAN_INTERCEPT(pthread_join); 1970 TSAN_INTERCEPT(pthread_detach); 1971 1972 TSAN_INTERCEPT(pthread_mutex_init); 1973 TSAN_INTERCEPT(pthread_mutex_destroy); 1974 TSAN_INTERCEPT(pthread_mutex_lock); 1975 TSAN_INTERCEPT(pthread_mutex_trylock); 1976 TSAN_INTERCEPT(pthread_mutex_timedlock); 1977 TSAN_INTERCEPT(pthread_mutex_unlock); 1978 1979 TSAN_INTERCEPT(pthread_spin_init); 1980 TSAN_INTERCEPT(pthread_spin_destroy); 1981 TSAN_INTERCEPT(pthread_spin_lock); 1982 TSAN_INTERCEPT(pthread_spin_trylock); 1983 TSAN_INTERCEPT(pthread_spin_unlock); 1984 1985 TSAN_INTERCEPT(pthread_rwlock_init); 1986 TSAN_INTERCEPT(pthread_rwlock_destroy); 1987 TSAN_INTERCEPT(pthread_rwlock_rdlock); 1988 TSAN_INTERCEPT(pthread_rwlock_tryrdlock); 1989 TSAN_INTERCEPT(pthread_rwlock_timedrdlock); 1990 TSAN_INTERCEPT(pthread_rwlock_wrlock); 1991 TSAN_INTERCEPT(pthread_rwlock_trywrlock); 1992 TSAN_INTERCEPT(pthread_rwlock_timedwrlock); 1993 TSAN_INTERCEPT(pthread_rwlock_unlock); 1994 1995 // TSAN_INTERCEPT(pthread_cond_init); 1996 TSAN_INTERCEPT(pthread_cond_destroy); 1997 TSAN_INTERCEPT(pthread_cond_signal); 1998 TSAN_INTERCEPT(pthread_cond_broadcast); 1999 TSAN_INTERCEPT(pthread_cond_wait); 2000 TSAN_INTERCEPT(pthread_cond_timedwait); 2001 2002 TSAN_INTERCEPT(pthread_barrier_init); 2003 TSAN_INTERCEPT(pthread_barrier_destroy); 2004 TSAN_INTERCEPT(pthread_barrier_wait); 2005 2006 TSAN_INTERCEPT(pthread_once); 2007 2008 TSAN_INTERCEPT(sem_init); 2009 TSAN_INTERCEPT(sem_destroy); 2010 TSAN_INTERCEPT(sem_wait); 2011 TSAN_INTERCEPT(sem_trywait); 2012 TSAN_INTERCEPT(sem_timedwait); 2013 TSAN_INTERCEPT(sem_post); 2014 TSAN_INTERCEPT(sem_getvalue); 2015 2016 TSAN_INTERCEPT(stat); 2017 TSAN_INTERCEPT(__xstat); 2018 TSAN_INTERCEPT(stat64); 2019 TSAN_INTERCEPT(__xstat64); 2020 TSAN_INTERCEPT(lstat); 2021 TSAN_INTERCEPT(__lxstat); 2022 TSAN_INTERCEPT(lstat64); 2023 TSAN_INTERCEPT(__lxstat64); 2024 TSAN_INTERCEPT(fstat); 2025 TSAN_INTERCEPT(__fxstat); 2026 TSAN_INTERCEPT(fstat64); 2027 TSAN_INTERCEPT(__fxstat64); 2028 TSAN_INTERCEPT(open); 2029 TSAN_INTERCEPT(open64); 2030 TSAN_INTERCEPT(creat); 2031 TSAN_INTERCEPT(creat64); 2032 TSAN_INTERCEPT(dup); 2033 TSAN_INTERCEPT(dup2); 2034 TSAN_INTERCEPT(dup3); 2035 TSAN_INTERCEPT(eventfd); 2036 TSAN_INTERCEPT(signalfd); 2037 TSAN_INTERCEPT(inotify_init); 2038 TSAN_INTERCEPT(inotify_init1); 2039 TSAN_INTERCEPT(socket); 2040 TSAN_INTERCEPT(socketpair); 2041 TSAN_INTERCEPT(connect); 2042 TSAN_INTERCEPT(bind); 2043 TSAN_INTERCEPT(listen); 2044 TSAN_INTERCEPT(epoll_create); 2045 TSAN_INTERCEPT(epoll_create1); 2046 TSAN_INTERCEPT(close); 2047 TSAN_INTERCEPT(__close); 2048 TSAN_INTERCEPT(__res_iclose); 2049 TSAN_INTERCEPT(pipe); 2050 TSAN_INTERCEPT(pipe2); 2051 2052 TSAN_INTERCEPT(readv); 2053 TSAN_INTERCEPT(preadv64); 2054 TSAN_INTERCEPT(writev); 2055 TSAN_INTERCEPT(pwritev64); 2056 TSAN_INTERCEPT(send); 2057 TSAN_INTERCEPT(sendmsg); 2058 TSAN_INTERCEPT(recv); 2059 2060 TSAN_INTERCEPT(unlink); 2061 TSAN_INTERCEPT(fopen); 2062 TSAN_INTERCEPT(freopen); 2063 TSAN_INTERCEPT(fclose); 2064 TSAN_INTERCEPT(fread); 2065 TSAN_INTERCEPT(fwrite); 2066 TSAN_INTERCEPT(fflush); 2067 TSAN_INTERCEPT(abort); 2068 TSAN_INTERCEPT(puts); 2069 TSAN_INTERCEPT(rmdir); 2070 TSAN_INTERCEPT(opendir); 2071 2072 TSAN_INTERCEPT(epoll_ctl); 2073 TSAN_INTERCEPT(epoll_wait); 2074 TSAN_INTERCEPT(poll); 2075 2076 TSAN_INTERCEPT(sigaction); 2077 TSAN_INTERCEPT(signal); 2078 TSAN_INTERCEPT(raise); 2079 TSAN_INTERCEPT(kill); 2080 TSAN_INTERCEPT(pthread_kill); 2081 TSAN_INTERCEPT(sleep); 2082 TSAN_INTERCEPT(usleep); 2083 TSAN_INTERCEPT(nanosleep); 2084 TSAN_INTERCEPT(gettimeofday); 2085 2086 TSAN_INTERCEPT(mlock); 2087 TSAN_INTERCEPT(munlock); 2088 TSAN_INTERCEPT(mlockall); 2089 TSAN_INTERCEPT(munlockall); 2090 2091 TSAN_INTERCEPT(fork); 2092 TSAN_INTERCEPT(on_exit); 2093 TSAN_INTERCEPT(__cxa_atexit); 2094 2095 // Need to setup it, because interceptors check that the function is resolved. 2096 // But atexit is emitted directly into the module, so can't be resolved. 2097 REAL(atexit) = (int(*)(void(*)()))unreachable; 2098 atexit_ctx = new(internal_alloc(MBlockAtExit, sizeof(AtExitContext))) 2099 AtExitContext(); 2100 2101 if (REAL(__cxa_atexit)(&finalize, 0, 0)) { 2102 Printf("ThreadSanitizer: failed to setup atexit callback\n"); 2103 Die(); 2104 } 2105 2106 if (pthread_key_create(&g_thread_finalize_key, &thread_finalize)) { 2107 Printf("ThreadSanitizer: failed to create thread key\n"); 2108 Die(); 2109 } 2110 2111 FdInit(); 2112} 2113 2114void internal_start_thread(void(*func)(void *arg), void *arg) { 2115 void *th; 2116 REAL(pthread_create)(&th, 0, (void*(*)(void *arg))func, arg); 2117 REAL(pthread_detach)(th); 2118} 2119 2120} // namespace __tsan 2121