Home | History | Annotate | Download | only in rtl
      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