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