xref: /llvm-project/compiler-rt/test/tsan/java_symbolization.cpp (revision 6563bb53b5fd837a2a0368c58de1cf02ec921d27)
1 // RUN: %clangxx_tsan -O1 %s -o %t && %deflake %run %t | FileCheck %s
2 #include "java.h"
3 #include <memory.h>
4 
5 extern "C" void __tsan_symbolize_external_ex(
6     jptr pc, void (*add_frame)(void *, const char *, const char *, int, int),
7     void *ctx) {
8   if (pc == (1234 | kExternalPCBit)) {
9     add_frame(ctx, "MyInnerFunc", "MyInnerFile.java", 1234, 56);
10     add_frame(ctx, "MyOuterFunc", "MyOuterFile.java", 4321, 65);
11   }
12   if (pc == (2345 | kExternalPCBit)) {
13     add_frame(ctx, "Caller1", "CallerFile.java", 111, 22);
14     add_frame(ctx, "Caller2", "CallerFile.java", 333, 44);
15   }
16   if (pc == (3456 | kExternalPCBit)) {
17     add_frame(ctx, "Allocer1", "Alloc.java", 11, 222);
18     add_frame(ctx, "Allocer2", "Alloc.java", 33, 444);
19   }
20 }
21 
22 void *Thread(void *p) {
23   barrier_wait(&barrier);
24   __tsan_func_entry(2345 | kExternalPCBit);
25   __tsan_write1_pc((jptr)p + 16, 1234 | kExternalPCBit);
26   __tsan_func_exit();
27   return 0;
28 }
29 
30 jptr const kHeapSize = 64 * 1024;
31 jptr java_heap[kHeapSize];
32 
33 int main() {
34   barrier_init(&barrier, 2);
35   jptr jheap = (jptr)java_heap;
36   __tsan_java_init(jheap, kHeapSize);
37   const int kBlockSize = 32;
38   __tsan_func_entry(3456 | kExternalPCBit);
39   __tsan_java_alloc(jheap, kBlockSize);
40   __tsan_func_exit();
41   pthread_t th;
42   pthread_create(&th, 0, Thread, (void*)jheap);
43   __tsan_func_entry(2345 | kExternalPCBit);
44   __tsan_write1_pc(jheap + 16, 1234 | kExternalPCBit);
45   __tsan_func_exit();
46   barrier_wait(&barrier);
47   pthread_join(th, 0);
48   __tsan_java_free(jheap, kBlockSize);
49   fprintf(stderr, "DONE\n");
50   return __tsan_java_fini();
51 }
52 
53 // CHECK: WARNING: ThreadSanitizer: data race
54 // CHECK:   Write
55 // CHECK:     #0 MyInnerFunc MyInnerFile.java:1234:56
56 // CHECK:     #1 MyOuterFunc MyOuterFile.java:4321:65
57 // CHECK:     #2 Caller1 CallerFile.java:111:22
58 // CHECK:     #3 Caller2 CallerFile.java:333:44
59 // CHECK-NOT: #4
60 // CHECK:   Previous write
61 // CHECK:     #0 MyInnerFunc MyInnerFile.java:1234:56
62 // CHECK:     #1 MyOuterFunc MyOuterFile.java:4321:65
63 // CHECK:     #2 Caller1 CallerFile.java:111:22
64 // CHECK:     #3 Caller2 CallerFile.java:333:44
65 // On Linux/glibc #4 is __libc_start_main, but can be something else elsewhere.
66 // CHECK:     #4
67 // CHECK:   Location is heap block of size 32 at {{.*}} allocated by main thread:
68 // CHECK:     #0 Allocer1 Alloc.java:11:222
69 // CHECK:     #1 Allocer2 Alloc.java:33:444
70 // On Linux/glibc #2 is __libc_start_main, but can be something else elsewhere.
71 // CHECK:     #2
72 // CHECK: DONE
73