| 1 | // RUN: %clangxx_tsan -O1 %s -o %t && %deflake %run %t | FileCheck %s |
| 2 | #include "test.h" |
| 3 | #include <stdint.h> |
| 4 | |
| 5 | #define NOINLINE __attribute__((noinline)) |
| 6 | |
| 7 | volatile uint64_t objs[8*2*(2 + 4 + 8)][2]; |
| 8 | |
| 9 | // All this mess is to generate unique stack for each race, |
| 10 | // otherwise tsan will suppress similar stacks. |
| 11 | |
| 12 | static NOINLINE void access(volatile void *p, int sz, int rw) { |
| 13 | if (rw) { |
| 14 | switch (sz) { |
| 15 | case 0: __sanitizer_unaligned_store16(p: (void *)p, x: 0); break; |
| 16 | case 1: __sanitizer_unaligned_store32(p: (void *)p, x: 0); break; |
| 17 | case 2: __sanitizer_unaligned_store64(p: (void *)p, x: 0); break; |
| 18 | default: exit(status: 1); |
| 19 | } |
| 20 | } else { |
| 21 | switch (sz) { |
| 22 | case 0: __sanitizer_unaligned_load16(p: (void *)p); break; |
| 23 | case 1: __sanitizer_unaligned_load32(p: (void *)p); break; |
| 24 | case 2: __sanitizer_unaligned_load64(p: (void *)p); break; |
| 25 | default: exit(status: 1); |
| 26 | } |
| 27 | } |
| 28 | } |
| 29 | |
| 30 | static int accesssize(int sz) { |
| 31 | switch (sz) { |
| 32 | case 0: return 2; |
| 33 | case 1: return 4; |
| 34 | case 2: return 8; |
| 35 | } |
| 36 | exit(status: 1); |
| 37 | } |
| 38 | |
| 39 | template<int off, int off2> |
| 40 | static NOINLINE void access3(bool main, int sz1, bool rw, volatile char *p) { |
| 41 | p += off; |
| 42 | if (main) { |
| 43 | access(p, sz: sz1, rw: true); |
| 44 | } else { |
| 45 | p += off2; |
| 46 | if (rw) { |
| 47 | *p = 42; |
| 48 | } else { |
| 49 | if (*p == 42) |
| 50 | printf(format: "bingo!\n" ); |
| 51 | } |
| 52 | } |
| 53 | } |
| 54 | |
| 55 | template<int off> |
| 56 | static NOINLINE void |
| 57 | access2(bool main, int sz1, int off2, bool rw, volatile char *obj) { |
| 58 | if (off2 == 0) |
| 59 | access3<off, 0>(main, sz1, rw, obj); |
| 60 | else if (off2 == 1) |
| 61 | access3<off, 1>(main, sz1, rw, obj); |
| 62 | else if (off2 == 2) |
| 63 | access3<off, 2>(main, sz1, rw, obj); |
| 64 | else if (off2 == 3) |
| 65 | access3<off, 3>(main, sz1, rw, obj); |
| 66 | else if (off2 == 4) |
| 67 | access3<off, 4>(main, sz1, rw, obj); |
| 68 | else if (off2 == 5) |
| 69 | access3<off, 5>(main, sz1, rw, obj); |
| 70 | else if (off2 == 6) |
| 71 | access3<off, 6>(main, sz1, rw, obj); |
| 72 | else if (off2 == 7) |
| 73 | access3<off, 7>(main, sz1, rw, obj); |
| 74 | } |
| 75 | |
| 76 | static NOINLINE void |
| 77 | access1(bool main, int off, int sz1, int off2, bool rw, char *obj) { |
| 78 | if (off == 0) |
| 79 | access2<0>(main, sz1, off2, rw, obj); |
| 80 | else if (off == 1) |
| 81 | access2<1>(main, sz1, off2, rw, obj); |
| 82 | else if (off == 2) |
| 83 | access2<2>(main, sz1, off2, rw, obj); |
| 84 | else if (off == 3) |
| 85 | access2<3>(main, sz1, off2, rw, obj); |
| 86 | else if (off == 4) |
| 87 | access2<4>(main, sz1, off2, rw, obj); |
| 88 | else if (off == 5) |
| 89 | access2<5>(main, sz1, off2, rw, obj); |
| 90 | else if (off == 6) |
| 91 | access2<6>(main, sz1, off2, rw, obj); |
| 92 | else if (off == 7) |
| 93 | access2<7>(main, sz1, off2, rw, obj); |
| 94 | } |
| 95 | |
| 96 | NOINLINE void Test(bool main) { |
| 97 | volatile uint64_t *obj = objs[0]; |
| 98 | for (int off = 0; off < 8; off++) { |
| 99 | for (int sz1 = 0; sz1 < 3; sz1++) { |
| 100 | for (int off2 = 0; off2 < accesssize(sz: sz1); off2++) { |
| 101 | for (int rw = 0; rw < 2; rw++) { |
| 102 | // printf("thr=%d off=%d sz1=%d off2=%d rw=%d p=%p\n", |
| 103 | // main, off, sz1, off2, rw, obj); |
| 104 | access1(main, off, sz1, off2, rw, obj: (char*)obj); |
| 105 | obj += 2; |
| 106 | } |
| 107 | } |
| 108 | } |
| 109 | } |
| 110 | } |
| 111 | |
| 112 | void *Thread(void *p) { |
| 113 | (void)p; |
| 114 | barrier_wait(barrier: &barrier); |
| 115 | Test(main: false); |
| 116 | return 0; |
| 117 | } |
| 118 | |
| 119 | int main() { |
| 120 | barrier_init(barrier: &barrier, count: 2); |
| 121 | pthread_t th; |
| 122 | pthread_create(newthread: &th, attr: 0, start_routine: Thread, arg: 0); |
| 123 | Test(main: true); |
| 124 | barrier_wait(barrier: &barrier); |
| 125 | pthread_join(th: th, thread_return: 0); |
| 126 | } |
| 127 | |
| 128 | // CHECK: WARNING: ThreadSanitizer: data race |
| 129 | // CHECK: ThreadSanitizer: reported 224 warnings |
| 130 | |