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
18void __tsan_init(void **thr, void **proc, void (*cb)(long, void*));
19void __tsan_fini();
20void __tsan_map_shadow(void *addr, unsigned long size);
21void __tsan_go_start(void *thr, void **chthr, void *pc);
22void __tsan_go_end(void *thr);
23void __tsan_proc_create(void **pproc);
24void __tsan_proc_destroy(void *proc);
25void __tsan_proc_wire(void *proc, void *thr);
26void __tsan_proc_unwire(void *proc, void *thr);
27void __tsan_read(void *thr, void *addr, void *pc);
28void __tsan_write(void *thr, void *addr, void *pc);
29void __tsan_func_enter(void *thr, void *pc);
30void __tsan_func_exit(void *thr);
31void __tsan_malloc(void *thr, void *pc, void *p, unsigned long sz);
32void __tsan_free(void *p, unsigned long sz);
33void __tsan_acquire(void *thr, void *addr);
34void __tsan_release(void *thr, void *addr);
35void __tsan_release_acquire(void *thr, void *addr);
36void __tsan_release_merge(void *thr, void *addr);
37
38void *current_proc;
39
40void 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 */
54void *go_heap = (void *)0xC011110000;
55char *buf0;
56
57void foobar() {}
58void barfoo() {}
59
60int main(void) {
61 void *thr0 = 0;
62 void *proc0 = 0;
63 __tsan_init(thr: &thr0, proc: &proc0, cb: symbolize_cb);
64 current_proc = proc0;
65
66 // Allocate something resembling a heap in Go.
67 buf0 = mmap(addr: go_heap, len: 16384, PROT_READ | PROT_WRITE,
68 MAP_PRIVATE | MAP_FIXED | MAP_ANON, fd: -1, offset: 0);
69 if (buf0 == MAP_FAILED) {
70 fprintf(stderr, format: "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(addr: buf, size: 4096);
76 __tsan_malloc(thr: thr0, pc: (char*)&barfoo + 1, p: buf, sz: 10);
77 __tsan_free(p: buf, sz: 10);
78 __tsan_func_enter(thr: thr0, pc: (char*)&main + 1);
79 __tsan_malloc(thr: thr0, pc: (char*)&barfoo + 1, p: buf, sz: 10);
80 __tsan_release(thr: thr0, addr: buf);
81 __tsan_release_acquire(thr: thr0, addr: buf);
82 __tsan_release_merge(thr: thr0, addr: buf);
83 void *thr1 = 0;
84 __tsan_go_start(thr: thr0, chthr: &thr1, pc: (char*)&barfoo + 1);
85 void *thr2 = 0;
86 __tsan_go_start(thr: thr0, chthr: &thr2, pc: (char*)&barfoo + 1);
87 __tsan_func_exit(thr: thr0);
88 __tsan_func_enter(thr: thr1, pc: (char*)&foobar + 1);
89 __tsan_func_enter(thr: thr1, pc: (char*)&foobar + 1);
90 __tsan_write(thr: thr1, addr: buf, pc: (char*)&barfoo + 1);
91 __tsan_acquire(thr: thr1, addr: buf);
92 __tsan_func_exit(thr: thr1);
93 __tsan_func_exit(thr: thr1);
94 __tsan_go_end(thr: thr1);
95 void *proc1 = 0;
96 __tsan_proc_create(pproc: &proc1);
97 current_proc = proc1;
98 __tsan_func_enter(thr: thr2, pc: (char*)&foobar + 1);
99 __tsan_read(thr: thr2, addr: buf, pc: (char*)&barfoo + 1);
100 __tsan_free(p: buf, sz: 10);
101 __tsan_func_exit(thr: thr2);
102 __tsan_go_end(thr: thr2);
103 __tsan_proc_destroy(proc: proc1);
104 current_proc = proc0;
105 __tsan_fini();
106 return 0;
107}
108

source code of compiler-rt/lib/tsan/go/test.c