1 | // RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s |
2 | #include "java.h" |
3 | |
4 | int const kHeapSize = 1024 * 1024; |
5 | |
6 | static void verify_find(jptr from, jptr to, jptr expected_addr, |
7 | jptr expected_size) { |
8 | jptr addr = from; |
9 | jptr size = __tsan_java_find(from_ptr: &addr, to); |
10 | if (expected_size) { |
11 | if (!size) { |
12 | fprintf(stderr, format: "FAILED: range: [%p..%p): found nothing\n" , (void *)from, |
13 | (void *)to); |
14 | return; |
15 | } else if (expected_size != size) { |
16 | fprintf(stderr, format: "FAILED: range: [%p..%p): wrong size, %lu instead of %lu\n" , |
17 | (void *)from, (void *)to, size, expected_size); |
18 | return; |
19 | } |
20 | } else if (size) { |
21 | fprintf(stderr, |
22 | format: "FAILED: range [%p..%p): did not expect to find anything here\n" , |
23 | (void *)from, (void *)to); |
24 | return; |
25 | } else { |
26 | return; |
27 | } |
28 | if (expected_addr != addr) { |
29 | fprintf( |
30 | stderr, |
31 | format: "FAILED: range [%p..%p): expected to find object at %p, found at %p\n" , |
32 | (void *)from, (void *)to, (void *)expected_addr, (void *)addr); |
33 | } |
34 | } |
35 | |
36 | int main() { |
37 | const jptr jheap = (jptr)malloc(size: kHeapSize + 8) + 8; |
38 | const jptr jheap_end = jheap + kHeapSize; |
39 | __tsan_java_init(heap_begin: jheap, heap_size: kHeapSize); |
40 | const jptr addr1 = jheap; |
41 | const int size1 = 16; |
42 | __tsan_java_alloc(ptr: jheap, size: size1); |
43 | |
44 | const jptr addr2 = addr1 + size1; |
45 | const int size2 = 32; |
46 | __tsan_java_alloc(ptr: jheap + size1, size: size2); |
47 | |
48 | const jptr addr3 = addr2 + size2; |
49 | const int size3 = 1024; |
50 | __tsan_java_alloc(ptr: jheap + size1 + size2, size: size3); |
51 | |
52 | const jptr addr4 = addr3 + size3; |
53 | |
54 | verify_find(from: jheap, to: jheap_end, expected_addr: addr1, expected_size: size1); |
55 | verify_find(from: jheap + 8, to: jheap_end, expected_addr: addr2, expected_size: size2); |
56 | verify_find(from: addr2 + 8, to: jheap_end, expected_addr: addr3, expected_size: size3); |
57 | verify_find(from: addr3 + 8, to: jheap_end, expected_addr: 0, expected_size: 0); |
58 | |
59 | __tsan_java_move(src: addr2, dst: addr4, size: size2); |
60 | verify_find(from: jheap + 8, to: jheap_end, expected_addr: addr3, expected_size: size3); |
61 | verify_find(from: addr3 + 8, to: jheap_end, expected_addr: addr4, expected_size: size2); |
62 | verify_find(from: addr4 + 8, to: jheap_end, expected_addr: 0, expected_size: 0); |
63 | |
64 | fprintf(stderr, format: "DONE\n" ); |
65 | return 0; |
66 | } |
67 | |
68 | // CHECK-NOT: FAILED |
69 | // CHECK: DONE |
70 | |