xref: /llvm-project/compiler-rt/test/tsan/java_volatile.cpp (revision bcaeed49cb063de9fe504aa29e1cadff8a7be710)
1*bcaeed49SFangrui Song // RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
2*bcaeed49SFangrui Song #include "java.h"
3*bcaeed49SFangrui Song 
4*bcaeed49SFangrui Song jptr varaddr;
5*bcaeed49SFangrui Song jptr lockaddr;
6*bcaeed49SFangrui Song 
Thread(void * p)7*bcaeed49SFangrui Song void *Thread(void *p) {
8*bcaeed49SFangrui Song   while (__atomic_load_n((int*)lockaddr, __ATOMIC_RELAXED) == 0)
9*bcaeed49SFangrui Song     usleep(1000);  // spin-wait
10*bcaeed49SFangrui Song   __tsan_java_acquire(lockaddr);
11*bcaeed49SFangrui Song   *(int*)varaddr = 42;
12*bcaeed49SFangrui Song   return 0;
13*bcaeed49SFangrui Song }
14*bcaeed49SFangrui Song 
main()15*bcaeed49SFangrui Song int main() {
16*bcaeed49SFangrui Song   barrier_init(&barrier, 2);
17*bcaeed49SFangrui Song   int const kHeapSize = 1024 * 1024;
18*bcaeed49SFangrui Song   jptr jheap = (jptr)malloc(kHeapSize + 8) + 8;
19*bcaeed49SFangrui Song   __tsan_java_init(jheap, kHeapSize);
20*bcaeed49SFangrui Song   const int kBlockSize = 16;
21*bcaeed49SFangrui Song   __tsan_java_alloc(jheap, kBlockSize);
22*bcaeed49SFangrui Song   varaddr = jheap;
23*bcaeed49SFangrui Song   lockaddr = jheap + 8;
24*bcaeed49SFangrui Song   pthread_t th;
25*bcaeed49SFangrui Song   pthread_create(&th, 0, Thread, 0);
26*bcaeed49SFangrui Song   *(int*)varaddr = 43;
27*bcaeed49SFangrui Song   __tsan_java_release(lockaddr);
28*bcaeed49SFangrui Song   __atomic_store_n((int*)lockaddr, 1, __ATOMIC_RELAXED);
29*bcaeed49SFangrui Song   pthread_join(th, 0);
30*bcaeed49SFangrui Song   *(int*)lockaddr = 0;
31*bcaeed49SFangrui Song   pthread_create(&th, 0, Thread, 0);
32*bcaeed49SFangrui Song   *(int*)varaddr = 43;
33*bcaeed49SFangrui Song   __tsan_java_release_store(lockaddr);
34*bcaeed49SFangrui Song   __atomic_store_n((int*)lockaddr, 1, __ATOMIC_RELAXED);
35*bcaeed49SFangrui Song   pthread_join(th, 0);
36*bcaeed49SFangrui Song   __tsan_java_free(jheap, kBlockSize);
37*bcaeed49SFangrui Song   fprintf(stderr, "DONE\n");
38*bcaeed49SFangrui Song   return __tsan_java_fini();
39*bcaeed49SFangrui Song }
40*bcaeed49SFangrui Song 
41*bcaeed49SFangrui Song // CHECK-NOT: WARNING: ThreadSanitizer: data race
42*bcaeed49SFangrui Song // CHECK: DONE
43