1 // RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
2 // UNSUPPORTED: darwin
3 #include <pthread.h>
4 #include <stdio.h>
5 #include <stdlib.h>
6 #include <signal.h>
7 #include <sys/types.h>
8 #include <sys/time.h>
9 #include <unistd.h>
10 #include <errno.h>
11 
12 // Test synchronization in signal handled within IgnoreSync region.
13 
14 extern "C" void AnnotateIgnoreSyncBegin(const char *f, int l);
15 extern "C" void AnnotateIgnoreSyncEnd(const char *f, int l);
16 
17 const int kSignalCount = 500;
18 
19 __thread int process_signals;
20 int signals_handled;
21 int done;
22 int ready[kSignalCount];
23 long long data[kSignalCount];
24 
handler(int sig)25 static void handler(int sig) {
26   if (!__atomic_load_n(&process_signals, __ATOMIC_RELAXED))
27     return;
28   int pos = signals_handled++;
29   if (pos >= kSignalCount)
30     return;
31   data[pos] = pos;
32   __atomic_store_n(&ready[pos], 1, __ATOMIC_RELEASE);
33 }
34 
thr(void * p)35 static void* thr(void *p) {
36   AnnotateIgnoreSyncBegin(__FILE__, __LINE__);
37   __atomic_store_n(&process_signals, 1, __ATOMIC_RELAXED);
38   while (!__atomic_load_n(&done, __ATOMIC_RELAXED)) {
39   }
40   AnnotateIgnoreSyncEnd(__FILE__, __LINE__);
41   return 0;
42 }
43 
main()44 int main() {
45   struct sigaction act = {};
46   act.sa_handler = handler;
47   if (sigaction(SIGPROF, &act, 0)) {
48     perror("sigaction");
49     exit(1);
50   }
51   itimerval t;
52   t.it_value.tv_sec = 0;
53   t.it_value.tv_usec = 10;
54   t.it_interval = t.it_value;
55   if (setitimer(ITIMER_PROF, &t, 0)) {
56     perror("setitimer");
57     exit(1);
58   }
59 
60   pthread_t th;
61   pthread_create(&th, 0, thr, 0);
62   for (int pos = 0; pos < kSignalCount; pos++) {
63     while (__atomic_load_n(&ready[pos], __ATOMIC_ACQUIRE) == 0) {
64     }
65     if (data[pos] != pos) {
66       printf("at pos %d, expect %d, got %lld\n", pos, pos, data[pos]);
67       exit(1);
68     }
69   }
70   __atomic_store_n(&done, 1, __ATOMIC_RELAXED);
71   pthread_join(th, 0);
72   fprintf(stderr, "DONE\n");
73   return 0;
74 }
75 
76 // CHECK-NOT: WARNING: ThreadSanitizer:
77 // CHECK: DONE
78