1#ifndef __TSAN_TEST_H__
2#define __TSAN_TEST_H__
3
4#include <pthread.h>
5#include <stdlib.h>
6#include <stdio.h>
7#include <unistd.h>
8#include <dlfcn.h>
9#include <stddef.h>
10#include <sched.h>
11#include <stdarg.h>
12#include "sanitizer_common/print_address.h"
13
14#include <sanitizer/tsan_interface.h>
15
16#ifdef __APPLE__
17#include <mach/mach_time.h>
18#endif
19
20#ifndef TSAN_VECTORIZE
21# define TSAN_VECTORIZE __SSE4_2__
22#endif
23
24#if TSAN_VECTORIZE
25# include <emmintrin.h>
26# include <smmintrin.h>
27#else
28struct __m128i {
29 unsigned long long x[2];
30};
31#endif
32
33// TSan-invisible barrier.
34// Tests use it to establish necessary execution order in a way that does not
35// interfere with tsan (does not establish synchronization between threads).
36typedef unsigned invisible_barrier_t;
37
38#ifdef __cplusplus
39extern "C" {
40#endif
41void __tsan_testonly_barrier_init(invisible_barrier_t *barrier,
42 unsigned count);
43void __tsan_testonly_barrier_wait(invisible_barrier_t *barrier);
44unsigned long __tsan_testonly_shadow_stack_current_size();
45#ifdef __cplusplus
46}
47#endif
48
49static inline void barrier_init(invisible_barrier_t *barrier, unsigned count) {
50 __tsan_testonly_barrier_init(barrier, count);
51}
52
53static inline void barrier_wait(invisible_barrier_t *barrier) {
54 __tsan_testonly_barrier_wait(barrier);
55}
56
57// Default instance of the barrier, but a test can declare more manually.
58invisible_barrier_t barrier;
59
60#ifdef __APPLE__
61unsigned long long monotonic_clock_ns() {
62 static mach_timebase_info_data_t timebase_info;
63 if (timebase_info.denom == 0) mach_timebase_info(&timebase_info);
64 return (mach_absolute_time() * timebase_info.numer) / timebase_info.denom;
65}
66#else
67unsigned long long monotonic_clock_ns() {
68 struct timespec t;
69 clock_gettime(CLOCK_MONOTONIC, tp: &t);
70 return (unsigned long long)t.tv_sec * 1000000000ull + t.tv_nsec;
71}
72#endif
73
74//The const kPCInc must be in sync with StackTrace::GetPreviousInstructionPc
75#if defined(__s390__) || defined(__i386__) || defined(__x86_64__)
76const int kPCInc = 1;
77#elif defined(__sparc__) || defined(__mips__)
78const int kPCInc = 8;
79#else
80const int kPCInc = 4;
81#endif
82
83#ifdef __cplusplus
84extern "C" {
85#endif
86
87void AnnotateThreadName(const char *f, int l, const char *name);
88
89void AnnotateRWLockCreate(const char *f, int l, void *m);
90void AnnotateRWLockCreateStatic(const char *f, int l, void *m);
91void AnnotateRWLockDestroy(const char *f, int l, void *m);
92void AnnotateRWLockAcquired(const char *f, int l, void *m, long is_w);
93void AnnotateRWLockReleased(const char *f, int l, void *m, long is_w);
94
95void AnnotateIgnoreReadsBegin(const char *f, int l);
96void AnnotateIgnoreReadsEnd(const char *f, int l);
97void AnnotateIgnoreWritesBegin(const char *f, int l);
98void AnnotateIgnoreWritesEnd(const char *f, int l);
99
100void AnnotateIgnoreSyncBegin(const char *f, int l);
101void AnnotateIgnoreSyncEnd(const char *f, int l);
102
103void AnnotateHappensBefore(const char *f, int l, void *addr);
104void AnnotateHappensAfter(const char *f, int l, void *addr);
105
106void AnnotateBenignRaceSized(const char *f, int l, const volatile void *mem,
107 unsigned int size, const char *desc);
108void WTFAnnotateBenignRaceSized(const char *f, int l, const volatile void *mem,
109 unsigned int size, const char *desc);
110
111#ifdef __cplusplus
112}
113#endif
114
115#define ANNOTATE_RWLOCK_CREATE(m) \
116 AnnotateRWLockCreate(__FILE__, __LINE__, m)
117#define ANNOTATE_RWLOCK_CREATE_STATIC(m) \
118 AnnotateRWLockCreateStatic(__FILE__, __LINE__, m)
119#define ANNOTATE_RWLOCK_DESTROY(m) \
120 AnnotateRWLockDestroy(__FILE__, __LINE__, m)
121#define ANNOTATE_RWLOCK_ACQUIRED(m, is_w) \
122 AnnotateRWLockAcquired(__FILE__, __LINE__, m, is_w)
123#define ANNOTATE_RWLOCK_RELEASED(m, is_w) \
124 AnnotateRWLockReleased(__FILE__, __LINE__, m, is_w)
125#define ANNOTATE_HAPPENS_BEFORE(addr) \
126 AnnotateHappensBefore(__FILE__, __LINE__, (void *)(addr))
127#define ANNOTATE_HAPPENS_AFTER(addr) \
128 AnnotateHappensAfter(__FILE__, __LINE__, (void *)(addr))
129#define ANNOTATE_BENIGN_RACE(var) \
130 AnnotateBenignRaceSized(__FILE__, __LINE__, &(var), sizeof(var), #var)
131#define WTF_ANNOTATE_BENIGN_RACE(var) \
132 WTFAnnotateBenignRaceSized(__FILE__, __LINE__, &(var), sizeof(var), #var)
133
134#ifdef __APPLE__
135#define ASM_SYMBOL(symbol) "_" #symbol
136#else
137#define ASM_SYMBOL(symbol) #symbol
138#endif
139
140#endif // __TSAN_TEST_H__
141

source code of compiler-rt/test/tsan/test.h