ceph/src/boost/libs/test/tools/console_test_runner/src/console_test_runner.cpp

   1 //  (C) Copyright Gennadiy Rozental 2005-2014.
   2 //  Distributed under the Boost Software License, Version 1.0.
   3 //  (See accompanying file LICENSE_1_0.txt or copy at
   4 //  http://www.boost.org/LICENSE_1_0.txt)
   5
   6 //  See http://www.boost.org/libs/test for the library home page.
   7
   8 // Boost.Test
   9 #ifndef BOOST_TEST_DYN_LINK
  10 #define BOOST_TEST_DYN_LINK
  11 #endif
  12 #include <boost/test/unit_test.hpp>
  13
  14 // Boost.Runtime.Param
  15 #include <boost/test/utils/runtime/cla/named_parameter.hpp>
  16 #include <boost/test/utils/runtime/cla/parser.hpp>
  17
  18 namespace rt  = boost::runtime;
  19 namespace cla = boost::runtime::cla;
  20
  21 // STL
  22 #include <iostream>
  23
  24 //_________________________________________________________________//
  25
  26 // System API
  27
  28 namespace dyn_lib {
  29
  30 #if defined(BOOST_WINDOWS) && !defined(BOOST_DISABLE_WIN32) // WIN32 API
  31
  32 #include <windows.h>
  33
  34 typedef HINSTANCE handle;
  35
  36 inline handle
  37 open( std::string const& file_name )
  38 {
  39     return LoadLibrary( file_name.c_str() );
  40 }
  41
  42 //_________________________________________________________________//
  43
  44 template<typename TargType>
  45 inline TargType
  46 locate_symbol( handle h, std::string const& symbol )
  47 {
  48     return reinterpret_cast<TargType>( GetProcAddress( h, symbol.c_str() ) );
  49 }
  50
  51 //_________________________________________________________________//
  52
  53 inline void
  54 close( handle h )
  55 {
  56     if( h )
  57         FreeLibrary( h );
  58 }
  59
  60 //_________________________________________________________________//
  61
  62 inline std::string
  63 error()
  64 {
  65     LPTSTR msg = NULL;
  66
  67     FormatMessage( FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM,
  68                    NULL,
  69                    GetLastError(),
  70                    MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
  71                    (LPTSTR)&msg,
  72                    0, NULL );
  73
  74     std::string res( msg );
  75
  76     if( msg )
  77         LocalFree( msg );
  78
  79     return res;
  80 }
  81
  82 //_________________________________________________________________//
  83
  84 #elif defined(BOOST_HAS_UNISTD_H) // POSIX API
  85
  86 #include <dlfcn.h>
  87
  88 #include <sys/types.h>
  89 #include <sys/stat.h>
  90 #include <fcntl.h>
  91
  92
  93 typedef void* handle;
  94
  95 inline handle
  96 open( std::string const& file_name )
  97 {
  98     return dlopen( file_name.c_str(), RTLD_LOCAL | RTLD_LAZY );
  99 }
 100
 101 //_________________________________________________________________//
 102
 103 template<typename TargType>
 104 inline TargType
 105 locate_symbol( handle h, std::string const& symbol )
 106 {
 107     return reinterpret_cast<TargType>( dlsym( h, symbol.c_str() ) );
 108 }
 109
 110 //_________________________________________________________________//
 111
 112 inline void
 113 close( handle h )
 114 {
 115     if( h )
 116         dlclose( h );
 117 }
 118
 119 //_________________________________________________________________//
 120
 121 inline std::string
 122 error()
 123 {
 124     return dlerror();
 125 }
 126
 127 //_________________________________________________________________//
 128
 129 #else
 130
 131 #error "Dynamic library API is unknown"
 132
 133 #endif
 134
 135 } // namespace dyn_lib
 136
 137 //____________________________________________________________________________//
 138
 139 static std::string test_lib_name;
 140 static std::string init_func_name( "init_unit_test" );
 141
 142 dyn_lib::handle test_lib_handle;
 143
 144 bool load_test_lib()
 145 {
 146     typedef bool (*init_func_ptr)();
 147     init_func_ptr init_func;
 148
 149     test_lib_handle = dyn_lib::open( test_lib_name );
 150     if( !test_lib_handle )
 151         throw std::logic_error( std::string("Fail to load test library: ")
 152                                     .append( dyn_lib::error() ) );
 153
 154     init_func =  dyn_lib::locate_symbol<init_func_ptr>( test_lib_handle, init_func_name );
 155
 156     if( !init_func )
 157         throw std::logic_error( std::string("Can't locate test initilization function ")
 158                                     .append( init_func_name )
 159                                     .append( ": " )
 160                                     .append( dyn_lib::error() ) );
 161
 162     return (*init_func)();
 163 }
 164
 165 //____________________________________________________________________________//
 166
 167 int main( int argc, char* argv[] )
 168 {
 169     try {
 170         cla::parser P;
 171
 172         P - cla::ignore_mismatch
 173             << cla::named_parameter<rt::cstring>( "test" ) - (cla::prefix = "--")
 174             << cla::named_parameter<rt::cstring>( "init" ) - (cla::prefix = "--",cla::optional);
 175
 176         P.parse( argc, argv );
 177
 178         assign_op( test_lib_name, P.get( "test" ), 0 );
 179         if( P["init"] )
 180             assign_op( init_func_name, P.get( "init" ), 0 );
 181
 182         int res = ::boost::unit_test::unit_test_main( &load_test_lib, argc, argv );
 183
 184         ::boost::unit_test::framework::clear();
 185         dyn_lib::close( test_lib_handle );
 186
 187         return res;
 188     }
 189     catch( rt::logic_error const& ex ) {
 190         std::cout << "Fail to parse command line arguments: " << ex.msg() << std::endl;
 191         return -1;
 192     }
 193 }
 194
 195 //____________________________________________________________________________//
 196
 197 // EOF