compiler-rt/lib/tsan/go/test.c

   1 //===-- test.c ------------------------------------------------------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // Test for Go runtime.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include <sys/mman.h>
  14 #include <errno.h>
  15 #include <stdio.h>
  16 #include <stdlib.h>
  17
  18 void __tsan_init(void **thr, void **proc, void (*cb)(long, void*));
  19 void __tsan_fini();
  20 void __tsan_map_shadow(void *addr, unsigned long size);
  21 void __tsan_go_start(void *thr, void **chthr, void *pc);
  22 void __tsan_go_end(void *thr);
  23 void __tsan_proc_create(void **pproc);
  24 void __tsan_proc_destroy(void *proc);
  25 void __tsan_proc_wire(void *proc, void *thr);
  26 void __tsan_proc_unwire(void *proc, void *thr);
  27 void __tsan_read(void *thr, void *addr, void *pc);
  28 void __tsan_write(void *thr, void *addr, void *pc);
  29 void __tsan_func_enter(void *thr, void *pc);
  30 void __tsan_func_exit(void *thr);
  31 void __tsan_malloc(void *thr, void *pc, void *p, unsigned long sz);
  32 void __tsan_free(void *p, unsigned long sz);
  33 void __tsan_acquire(void *thr, void *addr);
  34 void __tsan_release(void *thr, void *addr);
  35 void __tsan_release_acquire(void *thr, void *addr);
  36 void __tsan_release_merge(void *thr, void *addr);
  37
  38 void *current_proc;
  39
  40 void symbolize_cb(long cmd, void *ctx) {
  41   switch (cmd) {
  42   case 0:
  43     if (current_proc == 0)
  44       abort();
  45     *(void**)ctx = current_proc;
  46   }
  47 }
  48
  49 /*
  50  * See lib/tsan/rtl/tsan_platform.h for details of what the memory layout
  51  * of Go programs looks like.  To prevent running over existing mappings,
  52  * we pick an address slightly inside the Go heap region.
  53  */
  54 void *go_heap = (void *)0xC011110000;
  55 char *buf0;
  56
  57 void foobar() {}
  58 void barfoo() {}
  59
  60 int main(void) {
  61   void *thr0 = 0;
  62   void *proc0 = 0;
  63   __tsan_init(&thr0, &proc0, symbolize_cb);
  64   current_proc = proc0;
  65
  66   // Allocate something resembling a heap in Go.
  67   buf0 = mmap(go_heap, 16384, PROT_READ | PROT_WRITE,
  68               MAP_PRIVATE | MAP_FIXED | MAP_ANON, -1, 0);
  69   if (buf0 == MAP_FAILED) {
  70     fprintf(stderr, "failed to allocate Go-like heap at %p; errno %d\n",
  71             go_heap, errno);
  72     return 1;
  73   }
  74   char *buf = (char*)((unsigned long)buf0 + (64<<10) - 1 & ~((64<<10) - 1));
  75   __tsan_map_shadow(buf, 4096);
  76   __tsan_malloc(thr0, (char*)&barfoo + 1, buf, 10);
  77   __tsan_free(buf, 10);
  78   __tsan_func_enter(thr0, (char*)&main + 1);
  79   __tsan_malloc(thr0, (char*)&barfoo + 1, buf, 10);
  80   __tsan_release(thr0, buf);
  81   __tsan_release_acquire(thr0, buf);
  82   __tsan_release_merge(thr0, buf);
  83   void *thr1 = 0;
  84   __tsan_go_start(thr0, &thr1, (char*)&barfoo + 1);
  85   void *thr2 = 0;
  86   __tsan_go_start(thr0, &thr2, (char*)&barfoo + 1);
  87   __tsan_func_exit(thr0);
  88   __tsan_func_enter(thr1, (char*)&foobar + 1);
  89   __tsan_func_enter(thr1, (char*)&foobar + 1);
  90   __tsan_write(thr1, buf, (char*)&barfoo + 1);
  91   __tsan_acquire(thr1, buf);
  92   __tsan_func_exit(thr1);
  93   __tsan_func_exit(thr1);
  94   __tsan_go_end(thr1);
  95   void *proc1 = 0;
  96   __tsan_proc_create(&proc1);
  97   current_proc = proc1;
  98   __tsan_func_enter(thr2, (char*)&foobar + 1);
  99   __tsan_read(thr2, buf, (char*)&barfoo + 1);
 100   __tsan_free(buf, 10);
 101   __tsan_func_exit(thr2);
 102   __tsan_go_end(thr2);
 103   __tsan_proc_destroy(proc1);
 104   current_proc = proc0;
 105   __tsan_fini();
 106   return 0;
 107 }