1 #ifndef TSAN_INTERCEPTORS_H 2 #define TSAN_INTERCEPTORS_H 3 4 #include "sanitizer_common/sanitizer_stacktrace.h" 5 #include "tsan_rtl.h" 6 7 namespace __tsan { 8 9 class ScopedInterceptor { 10 public: 11 ScopedInterceptor(ThreadState *thr, const char *fname, uptr pc); 12 ~ScopedInterceptor(); 13 void UserCallbackStart(); 14 void UserCallbackEnd(); 15 private: 16 ThreadState *const thr_; 17 const uptr pc_; 18 bool in_ignored_lib_; 19 }; 20 21 } // namespace __tsan 22 23 #define SCOPED_INTERCEPTOR_RAW(func, ...) \ 24 ThreadState *thr = cur_thread(); \ 25 const uptr caller_pc = GET_CALLER_PC(); \ 26 ScopedInterceptor si(thr, #func, caller_pc); \ 27 const uptr pc = StackTrace::GetCurrentPc(); \ 28 (void)pc; \ 29 /**/ 30 31 #define SCOPED_TSAN_INTERCEPTOR(func, ...) \ 32 SCOPED_INTERCEPTOR_RAW(func, __VA_ARGS__); \ 33 if (REAL(func) == 0) { \ 34 Report("FATAL: ThreadSanitizer: failed to intercept %s\n", #func); \ 35 Die(); \ 36 } \ 37 if (!thr->is_inited || thr->ignore_interceptors || thr->in_ignored_lib) \ 38 return REAL(func)(__VA_ARGS__); \ 39 /**/ 40 41 #define SCOPED_TSAN_INTERCEPTOR_USER_CALLBACK_START() \ 42 si.UserCallbackStart(); 43 44 #define SCOPED_TSAN_INTERCEPTOR_USER_CALLBACK_END() \ 45 si.UserCallbackEnd(); 46 47 #define TSAN_INTERCEPTOR(ret, func, ...) INTERCEPTOR(ret, func, __VA_ARGS__) 48 49 #endif // TSAN_INTERCEPTORS_H 50