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_SINGLE_THREADED
16#  include <pthread.h>
17#endif
18#include <unistd.h>
19
20#include "../src/cxa_exception.hpp"
21
22typedef __cxxabiv1::__cxa_eh_globals globals_t ;
23
24void *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_SINGLE_THREADED
42#define NUMTHREADS  10
43size_t      thread_globals [ NUMTHREADS ] = { 0 };
44pthread_t   threads        [ NUMTHREADS ];
45#endif
46
47void 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
54int main ( int argc, char *argv [] ) {
55    int retVal = 0;
56
57#if LIBCXXABI_SINGLE_THREADED
58    size_t thread_globals;
59    retVal = thread_code(&thread_globals) != 0;
60#else
61//  Make the threads, let them run, and wait for them to finish
62    for ( int i = 0; i < NUMTHREADS; ++i )
63        pthread_create( threads + i, NULL, thread_code, (void *) (thread_globals + i));
64    for ( int i = 0; i < NUMTHREADS; ++i )
65        pthread_join ( threads [ i ], NULL );
66
67    for ( int i = 0; i < NUMTHREADS; ++i )
68        if ( 0 == thread_globals [ i ] ) {
69            std::cerr << "Thread #" << i << " had a zero global" << std::endl;
70            retVal = 1;
71            }
72
73//  print_sizes ( thread_globals, thread_globals + NUMTHREADS );
74    std::sort ( thread_globals, thread_globals + NUMTHREADS );
75    for ( int i = 1; i < NUMTHREADS; ++i )
76        if ( thread_globals [ i - 1 ] == thread_globals [ i ] ) {
77            std::cerr << "Duplicate thread globals (" << i-1 << " and " << i << ")" << std::endl;
78            retVal = 2;
79            }
80//  print_sizes ( thread_globals, thread_globals + NUMTHREADS );
81
82#endif
83    return retVal;
84    }
85