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 } 108