xref: /aosp_15_r20/external/compiler-rt/test/tsan/java_alloc.cc (revision 7c3d14c8b49c529e04be81a3ce6f5cc23712e4c6)
1*7c3d14c8STreehugger Robot // RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
2*7c3d14c8STreehugger Robot #include "java.h"
3*7c3d14c8STreehugger Robot 
4*7c3d14c8STreehugger Robot int const kHeapSize = 1024 * 1024;
5*7c3d14c8STreehugger Robot 
stress(jptr addr)6*7c3d14c8STreehugger Robot void stress(jptr addr) {
7*7c3d14c8STreehugger Robot   for (jptr sz = 8; sz <= 32; sz <<= 1) {
8*7c3d14c8STreehugger Robot     for (jptr i = 0; i < kHeapSize / 4 / sz; i++) {
9*7c3d14c8STreehugger Robot       __tsan_java_alloc(addr + i * sz, sz);
10*7c3d14c8STreehugger Robot     }
11*7c3d14c8STreehugger Robot     __tsan_java_move(addr, addr + kHeapSize / 2, kHeapSize / 4);
12*7c3d14c8STreehugger Robot     __tsan_java_free(addr + kHeapSize / 2, kHeapSize / 4);
13*7c3d14c8STreehugger Robot   }
14*7c3d14c8STreehugger Robot }
15*7c3d14c8STreehugger Robot 
Thread(void * p)16*7c3d14c8STreehugger Robot void *Thread(void *p) {
17*7c3d14c8STreehugger Robot   stress((jptr)p);
18*7c3d14c8STreehugger Robot   return 0;
19*7c3d14c8STreehugger Robot }
20*7c3d14c8STreehugger Robot 
main()21*7c3d14c8STreehugger Robot int main() {
22*7c3d14c8STreehugger Robot   jptr jheap = (jptr)malloc(kHeapSize + 8) + 8;
23*7c3d14c8STreehugger Robot   __tsan_java_init(jheap, kHeapSize);
24*7c3d14c8STreehugger Robot   pthread_t th;
25*7c3d14c8STreehugger Robot   pthread_create(&th, 0, Thread, (void*)(jheap + kHeapSize / 4));
26*7c3d14c8STreehugger Robot   stress(jheap);
27*7c3d14c8STreehugger Robot   pthread_join(th, 0);
28*7c3d14c8STreehugger Robot   if (__tsan_java_fini() != 0) {
29*7c3d14c8STreehugger Robot     fprintf(stderr, "FAILED\n");
30*7c3d14c8STreehugger Robot     return 1;
31*7c3d14c8STreehugger Robot   }
32*7c3d14c8STreehugger Robot   fprintf(stderr, "DONE\n");
33*7c3d14c8STreehugger Robot   return 0;
34*7c3d14c8STreehugger Robot }
35*7c3d14c8STreehugger Robot 
36*7c3d14c8STreehugger Robot // CHECK-NOT: WARNING: ThreadSanitizer: data race
37*7c3d14c8STreehugger Robot // CHECK-NOT: FAILED
38*7c3d14c8STreehugger Robot // CHECK: DONE
39