1 #include <pthread.h>
2 #include <stdlib.h>
3 #include <stdio.h>
4 #include <unistd.h>
5 #include <dlfcn.h>
6 #include <stddef.h>
7 #include <sched.h>
8 #include <stdarg.h>
9
10 #ifdef __APPLE__
11 #include <mach/mach_time.h>
12 #endif
13
14 // TSan-invisible barrier.
15 // Tests use it to establish necessary execution order in a way that does not
16 // interfere with tsan (does not establish synchronization between threads).
17 typedef unsigned long long invisible_barrier_t;
18
19 #ifdef __cplusplus
20 extern "C" {
21 #endif
22 void __tsan_testonly_barrier_init(invisible_barrier_t *barrier,
23 unsigned count);
24 void __tsan_testonly_barrier_wait(invisible_barrier_t *barrier);
25 #ifdef __cplusplus
26 }
27 #endif
28
barrier_init(invisible_barrier_t * barrier,unsigned count)29 static inline void barrier_init(invisible_barrier_t *barrier, unsigned count) {
30 __tsan_testonly_barrier_init(barrier, count);
31 }
32
barrier_wait(invisible_barrier_t * barrier)33 static inline void barrier_wait(invisible_barrier_t *barrier) {
34 __tsan_testonly_barrier_wait(barrier);
35 }
36
37 // Default instance of the barrier, but a test can declare more manually.
38 invisible_barrier_t barrier;
39
print_address(const char * str,int n,...)40 void print_address(const char *str, int n, ...) {
41 fprintf(stderr, "%s", str);
42 va_list ap;
43 va_start(ap, n);
44 while (n--) {
45 void *p = va_arg(ap, void *);
46 #if defined(__x86_64__) || defined(__aarch64__) || defined(__powerpc64__)
47 // On FreeBSD, the %p conversion specifier works as 0x%x and thus does not
48 // match to the format used in the diagnotic message.
49 fprintf(stderr, "0x%012lx ", (unsigned long) p);
50 #elif defined(__mips64)
51 fprintf(stderr, "0x%010lx ", (unsigned long) p);
52 #endif
53 }
54 fprintf(stderr, "\n");
55 }
56
57 #ifdef __APPLE__
monotonic_clock_ns()58 unsigned long long monotonic_clock_ns() {
59 static mach_timebase_info_data_t timebase_info;
60 if (timebase_info.denom == 0) mach_timebase_info(&timebase_info);
61 return (mach_absolute_time() * timebase_info.numer) / timebase_info.denom;
62 }
63 #else
monotonic_clock_ns()64 unsigned long long monotonic_clock_ns() {
65 struct timespec t;
66 clock_gettime(CLOCK_MONOTONIC, &t);
67 return (unsigned long long)t.tv_sec * 1000000000ull + t.tv_nsec;
68 }
69 #endif
70