1// RUN: %clangxx_tsan -O1 %s -o %t && %run %t 66 2>&1 | FileCheck %s -check-prefix=CHECK-DIE
2// RUN: %clangxx_tsan -O1 %s -o %t && %env_tsan_opts=die_after_fork=0 %run %t 2>&1 | FileCheck %s -check-prefix=CHECK-NODIE
3#include "test.h"
4#include <errno.h>
5#include <sys/types.h>
6#include <sys/wait.h>
7
8static void *sleeper(void *p) {
9 barrier_wait(barrier: &barrier);
10 return 0;
11}
12
13static void *nop(void *p) {
14 return 0;
15}
16
17int main(int argc, const char **argv) {
18 barrier_init(barrier: &barrier, count: 3);
19 const int kSleeperThreads = 2;
20 barrier_init(barrier: &barrier, count: kSleeperThreads + 1);
21 pthread_t th0[kSleeperThreads];
22 for (int i = 0; i < kSleeperThreads; i++)
23 pthread_create(newthread: &th0[i], attr: 0, start_routine: sleeper, arg: 0);
24 const int kNopThreads = 5;
25 pthread_t th1[kNopThreads];
26 for (int i = 0; i < kNopThreads; i++)
27 pthread_create(newthread: &th1[i], attr: 0, start_routine: nop, arg: 0);
28 for (int i = 0; i < kNopThreads; i++)
29 pthread_join(th: th1[i], thread_return: 0);
30 int pid = fork();
31 if (pid < 0) {
32 fprintf(stderr, format: "failed to fork (%d)\n", errno);
33 exit(status: 1);
34 }
35 if (pid == 0) {
36 // child
37 const int kChildThreads = 4;
38 pthread_t th2[kChildThreads];
39 for (int i = 0; i < kChildThreads; i++)
40 pthread_create(newthread: &th2[i], attr: 0, start_routine: nop, arg: 0);
41 for (int i = 0; i < kChildThreads; i++)
42 pthread_join(th: th2[i], thread_return: 0);
43 exit(status: 0);
44 return 0;
45 }
46 // parent
47 int expect = argc > 1 ? atoi(nptr: argv[1]) : 0;
48 int status = 0;
49 while (waitpid(pid: pid, stat_loc: &status, options: 0) != pid) {
50 }
51 if (!WIFEXITED(status) || WEXITSTATUS(status) != expect) {
52 fprintf(stderr, format: "subprocess exited with %d, expected %d\n", status, expect);
53 exit(status: 1);
54 }
55 barrier_wait(barrier: &barrier);
56 for (int i = 0; i < kSleeperThreads; i++)
57 pthread_join(th: th0[i], thread_return: 0);
58 fprintf(stderr, format: "OK\n");
59 return 0;
60}
61
62// CHECK-DIE: ThreadSanitizer: starting new threads after multi-threaded fork is not supported
63
64// CHECK-NODIE-NOT: ThreadSanitizer:
65// CHECK-NODIE: OK
66// CHECK-NODIE-NOT: ThreadSanitizer:
67

source code of compiler-rt/test/tsan/fork_multithreaded.cpp