compiler-rt/test/tsan/libdispatch/io-barrier-race.c

   1 // RUN: %clang_tsan %s -o %t
   2 // RUN: %deflake %run %t 2>&1 | FileCheck %s
   3
   4 #include <dispatch/dispatch.h>
   5
   6 #import "../test.h"
   7
   8 dispatch_queue_t queue;
   9 dispatch_data_t data;
  10 dispatch_semaphore_t sem;
  11 const char *path;
  12
  13 long my_global = 0;
  14
  15 int main(int argc, const char *argv[]) {
  16   fprintf(stderr, "Hello world.\n");
  17   print_address("addr=", 1, &my_global);
  18   barrier_init(&barrier, 2);
  19
  20   queue = dispatch_queue_create("my.queue", DISPATCH_QUEUE_CONCURRENT);
  21   sem = dispatch_semaphore_create(0);
  22   path = tempnam(NULL, "libdispatch-io-barrier-race-");
  23   char buf[1000];
  24   data = dispatch_data_create(buf, sizeof(buf), NULL, DISPATCH_DATA_DESTRUCTOR_DEFAULT);
  25
  26   dispatch_io_t channel = dispatch_io_create_with_path(DISPATCH_IO_STREAM, path, O_CREAT | O_WRONLY, 0666, queue, ^(int error) { });
  27   if (! channel) abort();
  28   dispatch_io_set_high_water(channel, 1);
  29
  30   dispatch_io_write(channel, 0, data, queue, ^(bool done, dispatch_data_t remainingData, int error) {
  31     if (error) abort();
  32     my_global = 42;
  33     barrier_wait(&barrier);
  34   });
  35
  36   dispatch_io_barrier(channel, ^{
  37     barrier_wait(&barrier);
  38     my_global = 43;
  39
  40     dispatch_semaphore_signal(sem);
  41   });
  42
  43   dispatch_semaphore_wait(sem, DISPATCH_TIME_FOREVER);
  44   dispatch_io_close(channel, 0);
  45
  46   fprintf(stderr, "Done.\n");
  47   return 0;
  48 }
  49
  50 // CHECK: Hello world.
  51 // CHECK: addr=[[ADDR:0x[0-9,a-f]+]]
  52 // CHECK: WARNING: ThreadSanitizer: data race
  53 // CHECK: Location is global 'my_global' {{(of size 8 )?}}at [[ADDR]] (io-barrier-race.c.tmp+0x{{[0-9,a-f]+}})
  54 // CHECK: Done.