1// RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
2// Regression test for https://github.com/golang/go/issues/39186
3
4// pthread barriers are not available on OS X
5// UNSUPPORTED: darwin
6
7#include "java.h"
8#include <string.h>
9
10struct Heap {
11 uint64_t data;
12 uint64_t ready;
13 uint64_t finalized;
14 uint64_t wg;
15 pthread_barrier_t barrier_finalizer;
16 pthread_barrier_t barrier_ballast;
17};
18
19void *Thread1(void *p) {
20 Heap* heap = (Heap*)p;
21 pthread_barrier_wait(barrier: &heap->barrier_finalizer);
22 __tsan_java_finalize();
23 __atomic_fetch_add(&heap->wg, 1, __ATOMIC_RELEASE);
24 __atomic_store_n(&heap->finalized, 1, __ATOMIC_RELAXED);
25 return 0;
26}
27
28void *Thread2(void *p) {
29 Heap* heap = (Heap*)p;
30 pthread_barrier_wait(barrier: &heap->barrier_finalizer);
31 heap->data = 1;
32 __atomic_store_n(&heap->ready, 1, __ATOMIC_RELEASE);
33 return 0;
34}
35
36void *Thread3(void *p) {
37 Heap* heap = (Heap*)p;
38 pthread_barrier_wait(barrier: &heap->barrier_finalizer);
39 while (__atomic_load_n(&heap->ready, __ATOMIC_ACQUIRE) != 1)
40 pthread_yield();
41 while (__atomic_load_n(&heap->finalized, __ATOMIC_RELAXED) != 1)
42 pthread_yield();
43 __atomic_fetch_add(&heap->wg, 1, __ATOMIC_RELEASE);
44 return 0;
45}
46
47void *Ballast(void *p) {
48 Heap* heap = (Heap*)p;
49 pthread_barrier_wait(barrier: &heap->barrier_ballast);
50 return 0;
51}
52
53int main() {
54 Heap *heap = (Heap *)calloc(nmemb: 2, size: sizeof(Heap)) + 1;
55 __tsan_java_init(heap_begin: (jptr)heap, heap_size: sizeof(*heap));
56 __tsan_java_alloc(ptr: (jptr)heap, size: sizeof(*heap));
57 // Ballast threads merely make the bug a bit easier to trigger.
58 const int kBallastThreads = 100;
59 pthread_barrier_init(barrier: &heap->barrier_finalizer, attr: 0, count: 4);
60 pthread_barrier_init(barrier: &heap->barrier_ballast, attr: 0, count: kBallastThreads + 1);
61 pthread_t th[3];
62 pthread_create(newthread: &th[0], attr: 0, start_routine: Thread1, arg: heap);
63 pthread_create(newthread: &th[1], attr: 0, start_routine: Thread2, arg: heap);
64 pthread_t ballast[kBallastThreads];
65 for (int i = 0; i < kBallastThreads; i++)
66 pthread_create(newthread: &ballast[i], attr: 0, start_routine: Ballast, arg: heap);
67 pthread_create(newthread: &th[2], attr: 0, start_routine: Thread3, arg: heap);
68 pthread_barrier_wait(barrier: &heap->barrier_ballast);
69 for (int i = 0; i < kBallastThreads; i++)
70 pthread_join(th: ballast[i], thread_return: 0);
71 pthread_barrier_wait(barrier: &heap->barrier_finalizer);
72 while (__atomic_load_n(&heap->wg, __ATOMIC_ACQUIRE) != 2)
73 pthread_yield();
74 if (heap->data != 1)
75 exit(status: printf(format: "no data\n"));
76 for (int i = 0; i < 3; i++)
77 pthread_join(th: th[i], thread_return: 0);
78 pthread_barrier_destroy(barrier: &heap->barrier_ballast);
79 pthread_barrier_destroy(barrier: &heap->barrier_finalizer);
80 __tsan_java_free(ptr: (jptr)heap, size: sizeof(*heap));
81 fprintf(stderr, format: "DONE\n");
82 return __tsan_java_fini();
83}
84
85// CHECK-NOT: WARNING: ThreadSanitizer: data race
86// CHECK: DONE
87

source code of compiler-rt/test/tsan/java_finalizer2.cpp