src/third_party/libc++abi/trunk/test/test_exception_storage.cpp

   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
  22 typedef __cxxabiv1::__cxa_eh_globals globals_t ;
  23
  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_SINGLE_THREADED
  42 #define NUMTHREADS  10
  43 size_t      thread_globals [ NUMTHREADS ] = { 0 };
  44 pthread_t   threads        [ NUMTHREADS ];
  45 #endif
  46
  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
  54 int 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     }