1// Stress test for https://reviews.llvm.org/D101881
2// RUN: %clangxx_hwasan -DREUSE=0 %s -pthread -O2 -o %t && %run %t 2>&1
3// RUN: %clangxx_hwasan -DREUSE=1 %s -pthread -O2 -o %t_reuse && %run %t_reuse 2>&1
4
5#include <thread>
6#include <vector>
7
8#include <sys/types.h>
9#include <sys/wait.h>
10#include <unistd.h>
11
12#include <stdio.h>
13
14constexpr int kTopThreads = 20;
15constexpr int kChildThreads = 30;
16constexpr int kChildIterations = REUSE ? 8 : 1;
17
18constexpr int kProcessIterations = 20;
19
20void Thread() {
21 for (int i = 0; i < kChildIterations; ++i) {
22 std::vector<std::thread> threads;
23 threads.reserve(n: kChildThreads);
24 for (int i = 0; i < kChildThreads; ++i)
25 threads.emplace_back(args: []() {});
26 for (auto &t : threads)
27 t.join();
28 }
29}
30
31void run() {
32 std::vector<std::thread> threads;
33 threads.reserve(n: kTopThreads);
34 for (int i = 0; i < kTopThreads; ++i)
35 threads.emplace_back(args&: Thread);
36 for (auto &t : threads)
37 t.join();
38}
39
40int main() {
41#if REUSE
42 // Test thread reuse with multiple iterations of thread create / join in a single process.
43 run();
44#else
45 // Test new, non-reused thread creation by running a single iteration of create / join in a freshly started process.
46 for (int i = 0; i < kProcessIterations; ++i) {
47 int pid = fork();
48 if (pid) {
49 int wstatus;
50 do {
51 waitpid(pid: pid, stat_loc: &wstatus, options: 0);
52 } while (!WIFEXITED(wstatus) && !WIFSIGNALED(wstatus));
53 if (!WIFEXITED(wstatus) || WEXITSTATUS(wstatus)) {
54 fprintf(stderr, format: "failed at iteration %d / %d\n", i, kProcessIterations);
55 return 1;
56 }
57 } else {
58 run();
59 return 0;
60 }
61 }
62#endif
63}
64

source code of compiler-rt/test/hwasan/TestCases/Linux/create-thread-stress.cpp