1*bcaeed49SFangrui Song // RUN: %clang_tsan -O1 %s -o %t && %run %t 2>&1 | FileCheck %s
2*bcaeed49SFangrui Song // UNSUPPORTED: tvos, watchos
3*bcaeed49SFangrui Song // XFAIL: ios && !iossim
4*bcaeed49SFangrui Song #include "sanitizer_common/sanitizer_ucontext.h"
5*bcaeed49SFangrui Song #include "test.h"
6*bcaeed49SFangrui Song #include <setjmp.h>
7*bcaeed49SFangrui Song
8*bcaeed49SFangrui Song char stack[64 * 1024] __attribute__((aligned(16)));
9*bcaeed49SFangrui Song
10*bcaeed49SFangrui Song sigjmp_buf jmpbuf, orig_jmpbuf[2];
11*bcaeed49SFangrui Song void *fiber, *orig_fiber[2];
12*bcaeed49SFangrui Song
13*bcaeed49SFangrui Song const unsigned N = 1000;
14*bcaeed49SFangrui Song
15*bcaeed49SFangrui Song __attribute__((noinline))
switch0()16*bcaeed49SFangrui Song void switch0() {
17*bcaeed49SFangrui Song if (!sigsetjmp(jmpbuf, 0)) {
18*bcaeed49SFangrui Song __tsan_switch_to_fiber(orig_fiber[0], 0);
19*bcaeed49SFangrui Song siglongjmp(orig_jmpbuf[0], 1);
20*bcaeed49SFangrui Song }
21*bcaeed49SFangrui Song }
22*bcaeed49SFangrui Song
func()23*bcaeed49SFangrui Song void func() {
24*bcaeed49SFangrui Song if (!sigsetjmp(jmpbuf, 0)) {
25*bcaeed49SFangrui Song __tsan_switch_to_fiber(orig_fiber[0], 0);
26*bcaeed49SFangrui Song siglongjmp(orig_jmpbuf[0], 1);
27*bcaeed49SFangrui Song }
28*bcaeed49SFangrui Song for (;;) {
29*bcaeed49SFangrui Song switch0();
30*bcaeed49SFangrui Song if (!sigsetjmp(jmpbuf, 0)) {
31*bcaeed49SFangrui Song __tsan_switch_to_fiber(orig_fiber[1], 0);
32*bcaeed49SFangrui Song siglongjmp(orig_jmpbuf[1], 1);
33*bcaeed49SFangrui Song }
34*bcaeed49SFangrui Song }
35*bcaeed49SFangrui Song }
36*bcaeed49SFangrui Song
Thread(void * x)37*bcaeed49SFangrui Song void *Thread(void *x) {
38*bcaeed49SFangrui Song orig_fiber[1] = __tsan_get_current_fiber();
39*bcaeed49SFangrui Song for (unsigned i = 0; i < N; i++) {
40*bcaeed49SFangrui Song barrier_wait(&barrier);
41*bcaeed49SFangrui Song if (!sigsetjmp(orig_jmpbuf[1], 0)) {
42*bcaeed49SFangrui Song __tsan_switch_to_fiber(fiber, 0);
43*bcaeed49SFangrui Song siglongjmp(jmpbuf, 1);
44*bcaeed49SFangrui Song }
45*bcaeed49SFangrui Song barrier_wait(&barrier);
46*bcaeed49SFangrui Song }
47*bcaeed49SFangrui Song return 0;
48*bcaeed49SFangrui Song }
49*bcaeed49SFangrui Song
main()50*bcaeed49SFangrui Song int main() {
51*bcaeed49SFangrui Song fiber = __tsan_create_fiber(0);
52*bcaeed49SFangrui Song barrier_init(&barrier, 2);
53*bcaeed49SFangrui Song pthread_t t;
54*bcaeed49SFangrui Song pthread_create(&t, 0, Thread, 0);
55*bcaeed49SFangrui Song orig_fiber[0] = __tsan_get_current_fiber();
56*bcaeed49SFangrui Song ucontext_t uc, orig_uc;
57*bcaeed49SFangrui Song getcontext(&uc);
58*bcaeed49SFangrui Song uc.uc_stack.ss_sp = stack;
59*bcaeed49SFangrui Song uc.uc_stack.ss_size = sizeof(stack);
60*bcaeed49SFangrui Song uc.uc_link = 0;
61*bcaeed49SFangrui Song makecontext(&uc, func, 0);
62*bcaeed49SFangrui Song if (!sigsetjmp(orig_jmpbuf[0], 0)) {
63*bcaeed49SFangrui Song __tsan_switch_to_fiber(fiber, 0);
64*bcaeed49SFangrui Song swapcontext(&orig_uc, &uc);
65*bcaeed49SFangrui Song }
66*bcaeed49SFangrui Song for (unsigned i = 0; i < N; i++) {
67*bcaeed49SFangrui Song if (!sigsetjmp(orig_jmpbuf[0], 0)) {
68*bcaeed49SFangrui Song __tsan_switch_to_fiber(fiber, 0);
69*bcaeed49SFangrui Song siglongjmp(jmpbuf, 1);
70*bcaeed49SFangrui Song }
71*bcaeed49SFangrui Song barrier_wait(&barrier);
72*bcaeed49SFangrui Song barrier_wait(&barrier);
73*bcaeed49SFangrui Song }
74*bcaeed49SFangrui Song pthread_join(t, 0);
75*bcaeed49SFangrui Song __tsan_destroy_fiber(fiber);
76*bcaeed49SFangrui Song fprintf(stderr, "PASS\n");
77*bcaeed49SFangrui Song return 0;
78*bcaeed49SFangrui Song }
79*bcaeed49SFangrui Song
80*bcaeed49SFangrui Song // CHECK-NOT: WARNING: ThreadSanitizer:
81*bcaeed49SFangrui Song // CHECK: PASS
82