blob: a351189b65106f5456d5072bc38f972e05c15797 [file] [log] [blame]
Kostya Serebryany7ac41482012-05-10 13:48:04 +00001//===-- 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
22using namespace __tsan; // NOLINT
23
24struct sigset_t {
25 u64 val[kSigCount / 8 / sizeof(u64)];
26};
27
28struct ucontext_t {
29 u64 opaque[1024];
30};
31
32extern "C" int pthread_attr_init(void *attr);
33extern "C" int pthread_attr_destroy(void *attr);
34extern "C" int pthread_attr_getdetachstate(void *attr, int *v);
35extern "C" int pthread_attr_setstacksize(void *attr, uptr stacksize);
36extern "C" int pthread_attr_getstacksize(void *attr, uptr *stacksize);
37extern "C" int pthread_key_create(unsigned *key, void (*destructor)(void* v));
38extern "C" int pthread_setspecific(unsigned key, const void *v);
39extern "C" int pthread_mutexattr_gettype(void *a, int *type);
40extern "C" int pthread_yield();
41extern "C" int pthread_sigmask(int how, const sigset_t *set, sigset_t *oldset);
42extern "C" int sigfillset(sigset_t *set);
43extern "C" void *pthread_self();
44extern "C" int getcontext(ucontext_t *ucp);
45extern "C" void _exit(int status);
46extern "C" int __cxa_atexit(void (*func)(void *arg), void *arg, void *dso);
47extern "C" int *__errno_location();
48extern "C" int usleep(unsigned usec);
49const int PTHREAD_MUTEX_RECURSIVE = 1;
50const int PTHREAD_MUTEX_RECURSIVE_NP = 1;
51const int kPthreadAttrSize = 56;
52const int EINVAL = 22;
53const int EBUSY = 16;
54const int EPOLL_CTL_ADD = 1;
Dmitry Vyukovd91d0672012-05-21 08:26:51 +000055const int SIGILL = 4;
56const int SIGABRT = 6;
57const int SIGFPE = 8;
58const int SIGSEGV = 11;
59const int SIGPIPE = 13;
60const int SIGBUS = 7;
Kostya Serebryany7ac41482012-05-10 13:48:04 +000061void *const MAP_FAILED = (void*)-1;
62const int PTHREAD_BARRIER_SERIAL_THREAD = -1;
63const int MAP_FIXED = 0x10;
64typedef long long_t; // NOLINT
65
66typedef void (*sighandler_t)(int sig);
67
68union pthread_attr_t {
69 char size[kPthreadAttrSize];
70 void *align;
71};
72
73struct 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
83const sighandler_t SIG_DFL = (sighandler_t)0;
84const sighandler_t SIG_IGN = (sighandler_t)1;
Dmitry Vyukovd91d0672012-05-21 08:26:51 +000085const sighandler_t SIG_ERR = (sighandler_t)-1;
Kostya Serebryany7ac41482012-05-10 13:48:04 +000086const int SA_SIGINFO = 4;
87const int SIG_SETMASK = 2;
88
89static sigaction_t sigactions[kSigCount];
90
91static unsigned g_thread_finalize_key;
92
93static void process_pending_signals(ThreadState *thr);
94
95class 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
146class 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
193static AtExitContext *atexit_ctx;
194
195static 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 Vyukovb7b6b1c2012-05-17 15:00:27 +0000205 if (status)
206 _exit(status);
Kostya Serebryany7ac41482012-05-10 13:48:04 +0000207}
208
209TSAN_INTERCEPTOR(int, atexit, void (*f)()) {
210 SCOPED_TSAN_INTERCEPTOR(atexit, f);
211 return atexit_ctx->atexit(thr, pc, f);
212 return 0;
213}
214
215static uptr fd2addr(int fd) {
216 (void)fd;
217 static u64 addr;
218 return (uptr)&addr;
219}
220
221static uptr epollfd2addr(int fd) {
222 (void)fd;
223 static u64 addr;
224 return (uptr)&addr;
225}
226
227static uptr file2addr(char *path) {
228 (void)path;
229 static u64 addr;
230 return (uptr)&addr;
231}
232
233static uptr dir2addr(char *path) {
234 (void)path;
235 static u64 addr;
236 return (uptr)&addr;
237}
238
239TSAN_INTERCEPTOR(void*, malloc, uptr size) {
240 SCOPED_INTERCEPTOR_RAW(malloc, size);
241 return user_alloc(thr, pc, size);
242}
243
244TSAN_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
251TSAN_INTERCEPTOR(void*, realloc, void *p, uptr size) {
252 SCOPED_INTERCEPTOR_RAW(realloc, p, size);
253 return user_realloc(thr, pc, p, size);
254}
255
256TSAN_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
263TSAN_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
270TSAN_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
277TSAN_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
283TSAN_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
290TSAN_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
302TSAN_INTERCEPTOR(int, strncmp, const char *s1, const char *s2, uptr n) {
303 SCOPED_TSAN_INTERCEPTOR(strncmp, s1, s2, n);
304 uptr len = 0;
Dmitry Vyukov7d386342012-05-18 08:53:16 +0000305 for (; len < n && s1[len] && s2[len]; len++) {
Kostya Serebryany7ac41482012-05-10 13:48:04 +0000306 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
314TSAN_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
322TSAN_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
328TSAN_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
335TSAN_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
348TSAN_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
356TSAN_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
364TSAN_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
370TSAN_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
378TSAN_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
386TSAN_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
396static 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
410TSAN_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
422TSAN_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
434TSAN_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)
443TSAN_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)
449TSAN_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)
455TSAN_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)
461TSAN_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*)
471TSAN_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)
479TSAN_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*)
487TSAN_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)
495TSAN_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
502TSAN_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
507TSAN_INTERCEPTOR(void*, valloc, uptr sz) {
508 SCOPED_TSAN_INTERCEPTOR(valloc, sz);
509 return user_alloc_aligned(thr, pc, sz, kPageSize);
510}
511
512TSAN_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
518TSAN_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.
525TSAN_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
536TSAN_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
542static 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
558struct ThreadParam {
559 void* (*callback)(void *arg);
560 void *param;
561 atomic_uintptr_t tid;
562};
563
564extern "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
590TSAN_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
625TSAN_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
635TSAN_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
645TSAN_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
661TSAN_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
670TSAN_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
679TSAN_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
688TSAN_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
697TSAN_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
704TSAN_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
713TSAN_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
722TSAN_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
731TSAN_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
740TSAN_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
747TSAN_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
756TSAN_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
765TSAN_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
774TSAN_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
783TSAN_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
792TSAN_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
801TSAN_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
810TSAN_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
819TSAN_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
826TSAN_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
832TSAN_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
838TSAN_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
844TSAN_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
850TSAN_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
858TSAN_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
866TSAN_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
873TSAN_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
880TSAN_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
892TSAN_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
917TSAN_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
923TSAN_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
929TSAN_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
938TSAN_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
947TSAN_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
956TSAN_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
963TSAN_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
972TSAN_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
981TSAN_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
990TSAN_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
999TSAN_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
1008TSAN_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
1017TSAN_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
1024TSAN_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
1031TSAN_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
1038TSAN_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
1045TSAN_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
1052TSAN_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
1059TSAN_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
1066TSAN_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
1075TSAN_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
1084TSAN_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
1091TSAN_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
1098TSAN_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
1104TSAN_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
1110TSAN_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
1116TSAN_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
1123TSAN_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
1130TSAN_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
1139TSAN_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 Vyukovd91d0672012-05-21 08:26:51 +00001148static void ALWAYS_INLINE rtl_generic_sighandler(bool sigact, int sig,
1149 my_siginfo_t *info, void *ctx) {
Kostya Serebryany7ac41482012-05-10 13:48:04 +00001150 ThreadState *thr = cur_thread();
Dmitry Vyukovd91d0672012-05-21 08:26:51 +00001151 // 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 Serebryany7ac41482012-05-10 13:48:04 +00001169 SignalDesc *signal = &thr->pending_signals[sig];
1170 if (signal->armed == false) {
1171 signal->armed = true;
Dmitry Vyukovd91d0672012-05-21 08:26:51 +00001172 signal->sigaction = sigact;
1173 if (info)
1174 signal->siginfo = *info;
Kostya Serebryany7ac41482012-05-10 13:48:04 +00001175 thr->pending_signal_count++;
1176 }
1177}
1178
Dmitry Vyukovd91d0672012-05-21 08:26:51 +00001179static void rtl_sighandler(int sig) {
1180 rtl_generic_sighandler(false, sig, 0, 0);
1181}
1182
Kostya Serebryany7ac41482012-05-10 13:48:04 +00001183static void rtl_sigaction(int sig, my_siginfo_t *info, void *ctx) {
Dmitry Vyukovd91d0672012-05-21 08:26:51 +00001184 rtl_generic_sighandler(true, sig, info, ctx);
Kostya Serebryany7ac41482012-05-10 13:48:04 +00001185}
1186
1187TSAN_INTERCEPTOR(int, sigaction, int sig, sigaction_t *act, sigaction_t *old) {
1188 SCOPED_TSAN_INTERCEPTOR(sigaction, sig, act, old);
Dmitry Vyukovd91d0672012-05-21 08:26:51 +00001189 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 Serebryany7ac41482012-05-10 13:48:04 +00001196 if (newact.sa_flags & SA_SIGINFO)
1197 newact.sa_sigaction = rtl_sigaction;
1198 else
1199 newact.sa_handler = rtl_sighandler;
Dmitry Vyukovd91d0672012-05-21 08:26:51 +00001200 }
1201 int res = REAL(sigaction)(sig, &newact, 0);
1202 return res;
1203}
1204
1205TSAN_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
1217TSAN_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
1227TSAN_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
1241TSAN_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 Serebryany7ac41482012-05-10 13:48:04 +00001251 }
1252 return res;
1253}
1254
1255static 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 Vyukovd91d0672012-05-21 08:26:51 +00001259 CHECK_EQ(thr->in_signal_handler, false);
Kostya Serebryany7ac41482012-05-10 13:48:04 +00001260 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 Vyukovd91d0672012-05-21 08:26:51 +00001272 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 Serebryany7ac41482012-05-10 13:48:04 +00001279 }
1280 }
1281 pthread_sigmask(SIG_SETMASK, &oldset, 0);
1282 CHECK_EQ(thr->in_signal_handler, true);
1283 thr->in_signal_handler = false;
1284}
1285
1286namespace __tsan {
1287
1288// Used until we obtain real efficient functions.
1289static 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
1295static 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
1301void 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 Vyukovd91d0672012-05-21 08:26:51 +00001424 TSAN_INTERCEPT(signal);
1425 TSAN_INTERCEPT(raise);
1426 TSAN_INTERCEPT(kill);
1427 TSAN_INTERCEPT(pthread_kill);
Kostya Serebryany7ac41482012-05-10 13:48:04 +00001428
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
1443void internal_memset(void *ptr, int c, uptr size) {
1444 REAL(memset)(ptr, c, size);
1445}
1446
1447void internal_memcpy(void *dst, const void *src, uptr size) {
1448 REAL(memcpy)(dst, src, size);
1449}
1450
1451int internal_memcmp(const void *s1, const void *s2, uptr size) {
1452 return REAL(memcmp)(s1, s2, size);
1453}
1454
1455int internal_strcmp(const char *s1, const char *s2) {
1456 return REAL(strcmp)(s1, s2);
1457}
1458
1459int internal_strncmp(const char *s1, const char *s2, uptr size) {
1460 return REAL(strncmp)(s1, s2, size);
1461}
1462
1463void internal_strcpy(char *s1, const char *s2) {
1464 REAL(strcpy)(s1, s2); // NOLINT
1465}
1466
1467uptr internal_strlen(const char *s) {
1468 return REAL(strlen)(s);
1469}
1470
1471char* 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
1479const char *internal_strstr(const char *where, const char *what) {
1480 return REAL(strstr)(where, what);
1481}
1482
1483const char *internal_strchr(const char *where, char what) {
1484 return (const char*)REAL(strchr)((void*)where, what);
1485}
1486
1487} // namespace __tsan