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