llvm-project

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((void *)p, 0); break;
16
    case 1: __sanitizer_unaligned_store32((void *)p, 0); break;
17
    case 2: __sanitizer_unaligned_store64((void *)p, 0); break;
18
    default: exit(1);
19
    }
20
  } else {
21
    switch (sz) {
22
    case 0: __sanitizer_unaligned_load16((void *)p); break;
23
    case 1: __sanitizer_unaligned_load32((void *)p); break;
24
    case 2: __sanitizer_unaligned_load64((void *)p); break;
25
    default: exit(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(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, sz1, true);
44
  } else {
45
    p += off2;
46
    if (rw) {
47
      *p = 42;
48
    } else {
49
       if (*p == 42)
50
         printf("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(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, (char*)obj);
105
          obj += 2;
106
        }
107
      }
108
    }
109
  }
110
}
111

112
void *Thread(void *p) {
113
  (void)p;
114
  barrier_wait(&barrier);
115
  Test(false);
116
  return 0;
117
}
118

119
int main() {
120
  barrier_init(&barrier, 2);
121
  pthread_t th;
122
  pthread_create(&th, 0, Thread, 0);
123
  Test(true);
124
  barrier_wait(&barrier);
125
  pthread_join(th, 0);
126
}
127

128
// CHECK: WARNING: ThreadSanitizer: data race
129
// CHECK: ThreadSanitizer: reported 224 warnings
130