1 //===-------------------- test_exception_storage.cpp ----------------------===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is dual licensed under the MIT and the University of Illinois Open
6 // Source Licenses. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "../src/config.h"
11
12 #include <cstdlib>
13 #include <algorithm>
14 #include <iostream>
15 #if !LIBCXXABI_HAS_NO_THREADS
16 # include <pthread.h>
17 #endif
18 #include <unistd.h>
19
20 #include "../src/cxa_exception.hpp"
21
22 typedef __cxxabiv1::__cxa_eh_globals globals_t ;
23
thread_code(void * parm)24 void *thread_code (void *parm) {
25 size_t *result = (size_t *) parm;
26 globals_t *glob1, *glob2;
27
28 glob1 = __cxxabiv1::__cxa_get_globals ();
29 if ( NULL == glob1 )
30 std::cerr << "Got null result from __cxa_get_globals" << std::endl;
31
32 glob2 = __cxxabiv1::__cxa_get_globals_fast ();
33 if ( glob1 != glob2 )
34 std::cerr << "Got different globals!" << std::endl;
35
36 *result = (size_t) glob1;
37 sleep ( 1 );
38 return parm;
39 }
40
41 #if !LIBCXXABI_HAS_NO_THREADS
42 #define NUMTHREADS 10
43 size_t thread_globals [ NUMTHREADS ] = { 0 };
44 pthread_t threads [ NUMTHREADS ];
45 #endif
46
print_sizes(size_t * first,size_t * last)47 void print_sizes ( size_t *first, size_t *last ) {
48 std::cout << "{ " << std::hex;
49 for ( size_t *iter = first; iter != last; ++iter )
50 std::cout << *iter << " ";
51 std::cout << "}" << std::dec << std::endl;
52 }
53
main(int argc,char * argv[])54 int main ( int argc, char *argv [] ) {
55 int retVal = 0;
56
57 #if LIBCXXABI_HAS_NO_THREADS
58 size_t thread_globals;
59 // Check that __cxa_get_globals() is not NULL.
60 if (thread_code(&thread_globals) == 0) {
61 retVal = 1;
62 }
63 #else
64 // Make the threads, let them run, and wait for them to finish
65 for ( int i = 0; i < NUMTHREADS; ++i )
66 pthread_create( threads + i, NULL, thread_code, (void *) (thread_globals + i));
67 for ( int i = 0; i < NUMTHREADS; ++i )
68 pthread_join ( threads [ i ], NULL );
69
70 for ( int i = 0; i < NUMTHREADS; ++i )
71 if ( 0 == thread_globals [ i ] ) {
72 std::cerr << "Thread #" << i << " had a zero global" << std::endl;
73 retVal = 1;
74 }
75
76 // print_sizes ( thread_globals, thread_globals + NUMTHREADS );
77 std::sort ( thread_globals, thread_globals + NUMTHREADS );
78 for ( int i = 1; i < NUMTHREADS; ++i )
79 if ( thread_globals [ i - 1 ] == thread_globals [ i ] ) {
80 std::cerr << "Duplicate thread globals (" << i-1 << " and " << i << ")" << std::endl;
81 retVal = 2;
82 }
83 // print_sizes ( thread_globals, thread_globals + NUMTHREADS );
84
85 #endif
86 return retVal;
87 }
88