[compiler-rt] fe4145b - [test][sanitizer] Add another stress test for pthread_create

[Vitaly Buka via llvm-commits]

Author: Vitaly Buka
Date: 2023-04-17T14:11:10-07:00
New Revision: fe4145b5798d2896bc04d7816b31cdb0130d2987

URL: https://github.com/llvm/llvm-project/commit/fe4145b5798d2896bc04d7816b31cdb0130d2987
DIFF: https://github.com/llvm/llvm-project/commit/fe4145b5798d2896bc04d7816b31cdb0130d2987.diff

LOG: [test][sanitizer] Add another stress test for pthread_create

Added: 
    compiler-rt/test/sanitizer_common/TestCases/Posix/create_thread_loop2.cpp

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/compiler-rt/test/sanitizer_common/TestCases/Posix/create_thread_loop2.cpp b/compiler-rt/test/sanitizer_common/TestCases/Posix/create_thread_loop2.cpp
new file mode 100644
index 0000000000000..afc4c769bead7
--- /dev/null
+++ b/compiler-rt/test/sanitizer_common/TestCases/Posix/create_thread_loop2.cpp
@@ -0,0 +1,36 @@
+// Simple stress test for of pthread_create. Increase arg to use as benchmark.
+
+// RUN: %clangxx -O3 -pthread %s -o %t && %run %t 10
+
+#include <cstdint>
+#include <pthread.h>
+#include <stdlib.h>
+
+extern "C" const char *__asan_default_options() {
+  // 32bit asan can allocate just a few FakeStacks.
+  return sizeof(void *) < 8 ? "detect_stack_use_after_return=0" : "";
+}
+
+static void *null_func(void *args) { return nullptr; }
+
+static void *loop(void *args) {
+  uintptr_t n = (uintptr_t)args;
+  for (int i = 0; i < n; ++i) {
+    pthread_t thread;
+    if (pthread_create(&thread, 0, null_func, nullptr) == 0)
+      pthread_detach(thread);
+  }
+  return nullptr;
+}
+
+int main(int argc, char **argv) {
+  uintptr_t n = atoi(argv[1]);
+  pthread_t threads[64];
+  for (auto &thread : threads)
+    while (pthread_create(&thread, 0, loop, (void *)n) != 0) {
+    }
+
+  for (auto &thread : threads)
+    pthread_join(thread, nullptr);
+  return 0;
+}