[compiler-rt] 43de9cb - [test][lsan] For thread args/result leak

[Vitaly Buka via llvm-commits]

Author: Vitaly Buka
Date: 2023-05-05T14:48:13-07:00
New Revision: 43de9cbec97aad44a3bba4cc878ab58caf6ea270

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

LOG: [test][lsan] For thread args/result leak

Added: 
    compiler-rt/test/lsan/TestCases/create_thread_leak.cpp

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/compiler-rt/test/lsan/TestCases/create_thread_leak.cpp b/compiler-rt/test/lsan/TestCases/create_thread_leak.cpp
new file mode 100644
index 0000000000000..7deab540df2a6
--- /dev/null
+++ b/compiler-rt/test/lsan/TestCases/create_thread_leak.cpp
@@ -0,0 +1,53 @@
+// Check that sanitizer does not leak args when passing to a child thread.
+
+// RUN: %clangxx_lsan -pthread %s -o %t -DLEAK_ARG && %run not %t 10 2>&1 | FileCheck %s --check-prefixes=LEAK,LEAK123
+// RUN: %clangxx_lsan -pthread %s -o %t -DLEAK_RES && %run not %t 10 2>&1 | FileCheck %s --check-prefixes=LEAK,LEAK234
+// RUN: %clangxx_lsan -pthread %s -o %t -DLEAK_DETACH && %run not %t 10 2>&1 | FileCheck %s --check-prefixes=LEAK,LEAK234
+
+// FIXME: Remove "not". There is no leak.
+// False LEAK123 is broken for HWASAN.
+// False LEAK234 is broken for ASAN, HWASAN, LSAN.
+// RUN: %clangxx_lsan -pthread %s -o %t && %run not %t 10
+
+#include <pthread.h>
+#include <stdlib.h>
+#include <vector>
+
+#include <sanitizer/lsan_interface.h>
+
+static void *thread_free(void *args) {
+#ifndef LEAK_ARG
+  free(args);
+#endif
+  return malloc(234);
+}
+
+int main(int argc, char **argv) {
+  int n = atoi(argv[1]);
+  for (int i = 0; i < n; ++i) {
+    std::vector<pthread_t> threads(10);
+
+    for (auto &thread : threads) {
+      pthread_create(&thread, 0, thread_free, malloc(123));
+      if (__lsan_do_recoverable_leak_check())
+        return 1;
+    }
+
+    for (auto &thread : threads) {
+#ifdef LEAK_DETACH
+      pthread_detach(thread);
+      continue;
+#endif
+      void *retval = 0;
+      pthread_join(thread, &retval);
+#ifndef LEAK_RES
+      free(retval);
+#endif
+    }
+  }
+  return 0;
+}
+
+// LEAK: LeakSanitizer: detected memory leaks
+// LEAK123: in main
+// LEAK234: in thread_free