xref: /llvm-project/compiler-rt/test/tsan/fiber_from_thread.cpp (revision bcaeed49cb063de9fe504aa29e1cadff8a7be710)
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 
7*bcaeed49SFangrui Song char stack[64 * 1024] __attribute__((aligned(16)));
8*bcaeed49SFangrui Song 
9*bcaeed49SFangrui Song ucontext_t uc, orig_uc1, orig_uc2;
10*bcaeed49SFangrui Song void *fiber, *orig_fiber1, *orig_fiber2;
11*bcaeed49SFangrui Song 
12*bcaeed49SFangrui Song int var;
13*bcaeed49SFangrui Song 
Thread(void * x)14*bcaeed49SFangrui Song void *Thread(void *x) {
15*bcaeed49SFangrui Song   orig_fiber2 = __tsan_get_current_fiber();
16*bcaeed49SFangrui Song   swapcontext(&orig_uc2, &orig_uc1);
17*bcaeed49SFangrui Song   return 0;
18*bcaeed49SFangrui Song }
19*bcaeed49SFangrui Song 
func()20*bcaeed49SFangrui Song void func() {
21*bcaeed49SFangrui Song   pthread_t t;
22*bcaeed49SFangrui Song   pthread_create(&t, 0, Thread, 0);
23*bcaeed49SFangrui Song   pthread_join(t, 0);
24*bcaeed49SFangrui Song   __tsan_switch_to_fiber(orig_fiber1, 0);
25*bcaeed49SFangrui Song   swapcontext(&uc, &orig_uc1);
26*bcaeed49SFangrui Song }
27*bcaeed49SFangrui Song 
main()28*bcaeed49SFangrui Song int main() {
29*bcaeed49SFangrui Song   orig_fiber1 = __tsan_get_current_fiber();
30*bcaeed49SFangrui Song   fiber = __tsan_create_fiber(0);
31*bcaeed49SFangrui Song   getcontext(&uc);
32*bcaeed49SFangrui Song   uc.uc_stack.ss_sp = stack;
33*bcaeed49SFangrui Song   uc.uc_stack.ss_size = sizeof(stack);
34*bcaeed49SFangrui Song   uc.uc_link = 0;
35*bcaeed49SFangrui Song   makecontext(&uc, func, 0);
36*bcaeed49SFangrui Song   var = 1;
37*bcaeed49SFangrui Song   __tsan_switch_to_fiber(fiber, 0);
38*bcaeed49SFangrui Song   swapcontext(&orig_uc1, &uc);
39*bcaeed49SFangrui Song   var = 2;
40*bcaeed49SFangrui Song   __tsan_switch_to_fiber(orig_fiber2, 0);
41*bcaeed49SFangrui Song   swapcontext(&orig_uc1, &orig_uc2);
42*bcaeed49SFangrui Song   var = 3;
43*bcaeed49SFangrui Song   __tsan_destroy_fiber(fiber);
44*bcaeed49SFangrui Song   fprintf(stderr, "PASS\n");
45*bcaeed49SFangrui Song   return 0;
46*bcaeed49SFangrui Song }
47*bcaeed49SFangrui Song 
48*bcaeed49SFangrui Song // CHECK-NOT: WARNING: ThreadSanitizer:
49*bcaeed49SFangrui Song // CHECK: PASS
50