You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.3 KiB
50 lines
1.3 KiB
4 months ago
|
#ifndef TSAN_INTERCEPTORS_H
|
||
|
|
#define TSAN_INTERCEPTORS_H
|
||
|
|
|
||
|
|
#include "sanitizer_common/sanitizer_stacktrace.h"
|
||
|
|
#include "tsan_rtl.h"
|
||
|
|
|
||
|
|
namespace __tsan {
|
||
|
|
|
||
|
|
class ScopedInterceptor {
|
||
|
|
public:
|
||
|
|
ScopedInterceptor(ThreadState *thr, const char *fname, uptr pc);
|
||
|
|
~ScopedInterceptor();
|
||
|
|
void UserCallbackStart();
|
||
|
|
void UserCallbackEnd();
|
||
|
|
private:
|
||
|
|
ThreadState *const thr_;
|
||
|
|
const uptr pc_;
|
||
|
|
bool in_ignored_lib_;
|
||
|
|
};
|
||
|
|
|
||
|
|
} // namespace __tsan
|
||
|
|
|
||
|
|
#define SCOPED_INTERCEPTOR_RAW(func, ...) \
|
||
|
|
ThreadState *thr = cur_thread(); \
|
||
|
|
const uptr caller_pc = GET_CALLER_PC(); \
|
||
|
|
ScopedInterceptor si(thr, #func, caller_pc); \
|
||
|
|
const uptr pc = StackTrace::GetCurrentPc(); \
|
||
|
|
(void)pc; \
|
||
|
|
/**/
|
||
|
|
|
||
|
|
#define SCOPED_TSAN_INTERCEPTOR(func, ...) \
|
||
|
|
SCOPED_INTERCEPTOR_RAW(func, __VA_ARGS__); \
|
||
|
|
if (REAL(func) == 0) { \
|
||
|
|
Report("FATAL: ThreadSanitizer: failed to intercept %s\n", #func); \
|
||
|
|
Die(); \
|
||
|
|
} \
|
||
|
|
if (!thr->is_inited || thr->ignore_interceptors || thr->in_ignored_lib) \
|
||
|
|
return REAL(func)(__VA_ARGS__); \
|
||
|
|
/**/
|
||
|
|
|
||
|
|
#define SCOPED_TSAN_INTERCEPTOR_USER_CALLBACK_START() \
|
||
|
|
si.UserCallbackStart();
|
||
|
|
|
||
|
|
#define SCOPED_TSAN_INTERCEPTOR_USER_CALLBACK_END() \
|
||
|
|
si.UserCallbackEnd();
|
||
|
|
|
||
|
|
#define TSAN_INTERCEPTOR(ret, func, ...) INTERCEPTOR(ret, func, __VA_ARGS__)
|
||
|
|
|
||
|
|
#endif // TSAN_INTERCEPTORS_H
|