xref: /llvm-project/compiler-rt/test/tsan/java_lock_rec.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   __tsan_java_mutex_lock(lockaddr);
9*bcaeed49SFangrui Song   __tsan_java_mutex_lock(lockaddr);
10*bcaeed49SFangrui Song   *(int*)varaddr = 42;
11*bcaeed49SFangrui Song   int rec = __tsan_java_mutex_unlock_rec(lockaddr);
12*bcaeed49SFangrui Song   if (rec != 2) {
13*bcaeed49SFangrui Song     fprintf(stderr, "FAILED 0 rec=%d\n", rec);
14*bcaeed49SFangrui Song     exit(1);
15*bcaeed49SFangrui Song   }
16*bcaeed49SFangrui Song   barrier_wait(&barrier);
17*bcaeed49SFangrui Song   barrier_wait(&barrier);
18*bcaeed49SFangrui Song   __tsan_java_mutex_lock_rec(lockaddr, rec);
19*bcaeed49SFangrui Song   if (*(int*)varaddr != 43) {
20*bcaeed49SFangrui Song     fprintf(stderr, "FAILED 3 var=%d\n", *(int*)varaddr);
21*bcaeed49SFangrui Song     exit(1);
22*bcaeed49SFangrui Song   }
23*bcaeed49SFangrui Song   __tsan_java_mutex_unlock(lockaddr);
24*bcaeed49SFangrui Song   __tsan_java_mutex_unlock(lockaddr);
25*bcaeed49SFangrui Song   return 0;
26*bcaeed49SFangrui Song }
27*bcaeed49SFangrui Song 
main()28*bcaeed49SFangrui Song int main() {
29*bcaeed49SFangrui Song   barrier_init(&barrier, 2);
30*bcaeed49SFangrui Song   int const kHeapSize = 1024 * 1024;
31*bcaeed49SFangrui Song   jptr jheap = (jptr)malloc(kHeapSize + 8) + 8;
32*bcaeed49SFangrui Song   __tsan_java_init(jheap, kHeapSize);
33*bcaeed49SFangrui Song   const int kBlockSize = 16;
34*bcaeed49SFangrui Song   __tsan_java_alloc(jheap, kBlockSize);
35*bcaeed49SFangrui Song   varaddr = jheap;
36*bcaeed49SFangrui Song   *(int*)varaddr = 0;
37*bcaeed49SFangrui Song   lockaddr = jheap + 8;
38*bcaeed49SFangrui Song   pthread_t th;
39*bcaeed49SFangrui Song   pthread_create(&th, 0, Thread, 0);
40*bcaeed49SFangrui Song   barrier_wait(&barrier);
41*bcaeed49SFangrui Song   __tsan_java_mutex_lock(lockaddr);
42*bcaeed49SFangrui Song   if (*(int*)varaddr != 42) {
43*bcaeed49SFangrui Song     fprintf(stderr, "FAILED 1 var=%d\n", *(int*)varaddr);
44*bcaeed49SFangrui Song     exit(1);
45*bcaeed49SFangrui Song   }
46*bcaeed49SFangrui Song   *(int*)varaddr = 43;
47*bcaeed49SFangrui Song   __tsan_java_mutex_unlock(lockaddr);
48*bcaeed49SFangrui Song   barrier_wait(&barrier);
49*bcaeed49SFangrui Song   pthread_join(th, 0);
50*bcaeed49SFangrui Song   __tsan_java_free(jheap, kBlockSize);
51*bcaeed49SFangrui Song   fprintf(stderr, "DONE\n");
52*bcaeed49SFangrui Song   return __tsan_java_fini();
53*bcaeed49SFangrui Song }
54*bcaeed49SFangrui Song 
55*bcaeed49SFangrui Song // CHECK-NOT: WARNING: ThreadSanitizer: data race
56*bcaeed49SFangrui Song // CHECK-NOT: FAILED
57*bcaeed49SFangrui Song // CHECK: DONE
58