compiler-rt/test/tsan/java_finalizer2.cpp

   1 // RUN: %clangxx_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
   2 // Regression test for https://github.com/golang/go/issues/39186
   3
   4 // pthread barriers are not available on OS X
   5 // UNSUPPORTED: darwin
   6
   7 #include "java.h"
   8 #include <string.h>
   9
  10 struct Heap {
  11   uint64_t data;
  12   uint64_t ready;
  13   uint64_t finalized;
  14   uint64_t wg;
  15   pthread_barrier_t barrier_finalizer;
  16   pthread_barrier_t barrier_ballast;
  17 };
  18
  19 void *Thread1(void *p) {
  20   Heap* heap = (Heap*)p;
  21   pthread_barrier_wait(&heap->barrier_finalizer);
  22   __tsan_java_finalize();
  23   __atomic_fetch_add(&heap->wg, 1, __ATOMIC_RELEASE);
  24   __atomic_store_n(&heap->finalized, 1, __ATOMIC_RELAXED);
  25   return 0;
  26 }
  27
  28 void *Thread2(void *p) {
  29   Heap* heap = (Heap*)p;
  30   pthread_barrier_wait(&heap->barrier_finalizer);
  31   heap->data = 1;
  32   __atomic_store_n(&heap->ready, 1, __ATOMIC_RELEASE);
  33   return 0;
  34 }
  35
  36 void *Thread3(void *p) {
  37   Heap* heap = (Heap*)p;
  38   pthread_barrier_wait(&heap->barrier_finalizer);
  39   while (__atomic_load_n(&heap->ready, __ATOMIC_ACQUIRE) != 1)
  40     pthread_yield();
  41   while (__atomic_load_n(&heap->finalized, __ATOMIC_RELAXED) != 1)
  42     pthread_yield();
  43   __atomic_fetch_add(&heap->wg, 1, __ATOMIC_RELEASE);
  44   return 0;
  45 }
  46
  47 void *Ballast(void *p) {
  48   Heap* heap = (Heap*)p;
  49   pthread_barrier_wait(&heap->barrier_ballast);
  50   return 0;
  51 }
  52
  53 int main() {
  54   Heap* heap = (Heap*)calloc(sizeof(Heap), 2) + 1;
  55   __tsan_java_init((jptr)heap, sizeof(*heap));
  56   __tsan_java_alloc((jptr)heap, sizeof(*heap));
  57   // Ballast threads merely make the bug a bit easier to trigger.
  58   const int kBallastThreads = 100;
  59   pthread_barrier_init(&heap->barrier_finalizer, 0, 4);
  60   pthread_barrier_init(&heap->barrier_ballast, 0, kBallastThreads + 1);
  61   pthread_t th[3];
  62   pthread_create(&th[0], 0, Thread1, heap);
  63   pthread_create(&th[1], 0, Thread2, heap);
  64   pthread_t ballast[kBallastThreads];
  65   for (int i = 0; i < kBallastThreads; i++)
  66     pthread_create(&ballast[i], 0, Ballast, heap);
  67   pthread_create(&th[2], 0, Thread3, heap);
  68   pthread_barrier_wait(&heap->barrier_ballast);
  69   for (int i = 0; i < kBallastThreads; i++)
  70     pthread_join(ballast[i], 0);
  71   pthread_barrier_wait(&heap->barrier_finalizer);
  72   while (__atomic_load_n(&heap->wg, __ATOMIC_ACQUIRE) != 2)
  73     pthread_yield();
  74   if (heap->data != 1)
  75     exit(printf("no data\n"));
  76   for (int i = 0; i < 3; i++)
  77     pthread_join(th[i], 0);
  78   pthread_barrier_destroy(&heap->barrier_ballast);
  79   pthread_barrier_destroy(&heap->barrier_finalizer);
  80   __tsan_java_free((jptr)heap, sizeof(*heap));
  81   fprintf(stderr, "DONE\n");
  82   return __tsan_java_fini();
  83 }
  84
  85 // CHECK-NOT: WARNING: ThreadSanitizer: data race
  86 // CHECK: DONE