1 //===- llvm/unittest/Support/ManagedStatic.cpp - ManagedStatic tests ------===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 #include "llvm/Support/ManagedStatic.h" 10 #include "llvm/Config/config.h" 11 #include "llvm/Support/Threading.h" 12 #ifdef HAVE_PTHREAD_H 13 #include <pthread.h> 14 #endif 15 16 #include "gtest/gtest.h" 17 18 using namespace llvm; 19 20 namespace { 21 22 #ifdef HAVE_PTHREAD_H 23 namespace test1 { 24 llvm::ManagedStatic<int> ms; 25 void *helper(void*) { 26 *ms; 27 return NULL; 28 } 29 30 // Valgrind's leak checker complains glibc's stack allocation. 31 // To appease valgrind, we provide our own stack for each thread. 32 void *allocate_stack(pthread_attr_t &a, size_t n = 65536) { 33 void *stack = malloc(n); 34 pthread_attr_init(&a); 35 #if defined(__linux__) 36 pthread_attr_setstack(&a, stack, n); 37 #endif 38 return stack; 39 } 40 } 41 42 TEST(Initialize, MultipleThreads) { 43 // Run this test under tsan: http://code.google.com/p/data-race-test/ 44 45 pthread_attr_t a1, a2; 46 void *p1 = test1::allocate_stack(a1); 47 void *p2 = test1::allocate_stack(a2); 48 49 llvm_start_multithreaded(); 50 pthread_t t1, t2; 51 pthread_create(&t1, &a1, test1::helper, NULL); 52 pthread_create(&t2, &a2, test1::helper, NULL); 53 pthread_join(t1, NULL); 54 pthread_join(t2, NULL); 55 free(p1); 56 free(p2); 57 llvm_stop_multithreaded(); 58 } 59 #endif 60 61 } // anonymous namespace 62