1#ifndef TSAN_INTERCEPTORS_H
2#define TSAN_INTERCEPTORS_H
3
4#include "sanitizer_common/sanitizer_stacktrace.h"
5#include "tsan_rtl.h"
6
7namespace __tsan {
8
9class 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->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#if SANITIZER_FREEBSD
50#define __libc_free __free
51#define __libc_malloc __malloc
52#endif
53
54extern "C" void __libc_free(void *ptr);
55extern "C" void *__libc_malloc(uptr size);
56
57#endif  // TSAN_INTERCEPTORS_H
58