1 //===-------------------- test_exception_storage.cpp ----------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include <algorithm>
10 #include <cstdio>
11 #include <cstdlib>
12 #include <__threading_support>
13 #include <unistd.h>
14 
15 #include "../src/cxa_exception.h"
16 
17 typedef __cxxabiv1::__cxa_eh_globals globals_t ;
18 
thread_code(void * parm)19 void *thread_code (void *parm) {
20     size_t *result = (size_t *) parm;
21     globals_t *glob1, *glob2;
22 
23     glob1 = __cxxabiv1::__cxa_get_globals ();
24     if ( NULL == glob1 )
25         std::printf("Got null result from __cxa_get_globals\n");
26 
27     glob2 = __cxxabiv1::__cxa_get_globals_fast ();
28     if ( glob1 != glob2 )
29         std::printf("Got different globals!\n");
30 
31     *result = (size_t) glob1;
32 #ifndef _LIBCXXABI_HAS_NO_THREADS
33     sleep ( 1 );
34 #endif
35     return parm;
36 }
37 
38 #ifndef _LIBCXXABI_HAS_NO_THREADS
39 #define NUMTHREADS  10
40 size_t                 thread_globals [ NUMTHREADS ] = { 0 };
41 std::__libcpp_thread_t   threads        [ NUMTHREADS ];
42 #endif
43 
main()44 int main () {
45     int retVal = 0;
46 
47 #ifndef _LIBCXXABI_HAS_NO_THREADS
48 //  Make the threads, let them run, and wait for them to finish
49     for ( int i = 0; i < NUMTHREADS; ++i )
50         std::__libcpp_thread_create ( threads + i, thread_code, (void *) (thread_globals + i));
51     for ( int i = 0; i < NUMTHREADS; ++i )
52         std::__libcpp_thread_join ( &threads [ i ] );
53 
54     for ( int i = 0; i < NUMTHREADS; ++i ) {
55         if ( 0 == thread_globals [ i ] ) {
56             std::printf("Thread #%d had a zero global\n", i);
57             retVal = 1;
58         }
59     }
60 
61     std::sort ( thread_globals, thread_globals + NUMTHREADS );
62     for ( int i = 1; i < NUMTHREADS; ++i ) {
63         if ( thread_globals [ i - 1 ] == thread_globals [ i ] ) {
64             std::printf("Duplicate thread globals (%d and %d)\n", i-1, i);
65             retVal = 2;
66         }
67     }
68 #else // _LIBCXXABI_HAS_NO_THREADS
69     size_t thread_globals;
70     // Check that __cxa_get_globals() is not NULL.
71     if (thread_code(&thread_globals) == 0) {
72         retVal = 1;
73     }
74 #endif // !_LIBCXXABI_HAS_NO_THREADS
75     return retVal;
76 }
77