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

[Vitaly Buka via llvm-commits]

Author: Vitaly Buka
Date: 2023-04-14T23:02:15-07:00
New Revision: 364884e088d45b162ecb47d093f955a2333eeee1

URL: https://github.com/llvm/llvm-project/commit/364884e088d45b162ecb47d093f955a2333eeee1
DIFF: https://github.com/llvm/llvm-project/commit/364884e088d45b162ecb47d093f955a2333eeee1.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..2776ad79ef89b
--- /dev/null
+++ b/compiler-rt/test/sanitizer_common/TestCases/Posix/create_thread_loop2.cpp
@@ -0,0 +1,35 @@
+// 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;
+    pthread_create(&thread, 0, null_func, nullptr);
+    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)
+    pthread_create(&thread, 0, loop, (void *)n);
+
+  for (auto &thread : threads)
+    pthread_join(thread, nullptr);
+  return 0;
+}