Source/kwsys/testDynamicLoader.cxx

   1 /*============================================================================
   2   KWSys - Kitware System Library
   3   Copyright 2000-2009 Kitware, Inc., Insight Software Consortium
   4
   5   Distributed under the OSI-approved BSD License (the "License");
   6   see accompanying file Copyright.txt for details.
   7
   8   This software is distributed WITHOUT ANY WARRANTY; without even the
   9   implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  10   See the License for more information.
  11 ============================================================================*/
  12 #include "kwsysPrivate.h"
  13
  14 #include KWSYS_HEADER(DynamicLoader.hxx)
  15 #include KWSYS_HEADER(ios/iostream)
  16 #include KWSYS_HEADER(stl/string)
  17
  18 #if defined(__BEOS__)
  19 #include <be/kernel/OS.h>  /* disable_debugger() API. */
  20 #endif
  21
  22 #if defined(__HAIKU__)
  23 #include <os/kernel/OS.h>  /* disable_debugger() API. */
  24 #endif
  25
  26 // Work-around CMake dependency scanning limitation.  This must
  27 // duplicate the above list of headers.
  28 #if 0
  29 # include "DynamicLoader.hxx.in"
  30 # include "kwsys_ios_iostream.h.in"
  31 # include "kwsys_stl_string.hxx.in"
  32 #endif
  33
  34 // Include with <> instead of "" to avoid getting any in-source copy
  35 // left on disk.
  36 #include <testSystemTools.h>
  37
  38 static kwsys_stl::string GetLibName(const char* lname)
  39 {
  40   // Construct proper name of lib
  41   kwsys_stl::string slname;
  42   slname = EXECUTABLE_OUTPUT_PATH;
  43 #ifdef CMAKE_INTDIR
  44   slname += "/";
  45   slname += CMAKE_INTDIR;
  46 #endif
  47   slname += "/";
  48   slname += kwsys::DynamicLoader::LibPrefix();
  49   slname += lname;
  50   slname += kwsys::DynamicLoader::LibExtension();
  51
  52   return slname;
  53 }
  54
  55 /* libname = Library name (proper prefix, proper extension)
  56  * System  = symbol to lookup in libname
  57  * r1: should OpenLibrary succeed ?
  58  * r2: should GetSymbolAddress succeed ?
  59  * r3: should CloseLibrary succeed ?
  60  */
  61 int TestDynamicLoader(const char* libname, const char* symbol, int r1, int r2, int r3)
  62 {
  63   kwsys_ios::cerr << "Testing: " << libname << kwsys_ios::endl;
  64   kwsys::DynamicLoader::LibraryHandle l
  65     = kwsys::DynamicLoader::OpenLibrary(libname);
  66   // If result is incompatible with expectation just fails (xor):
  67   if( (r1 && !l) || (!r1 && l) )
  68     {
  69     kwsys_ios::cerr
  70       << kwsys::DynamicLoader::LastError() << kwsys_ios::endl;
  71     return 1;
  72     }
  73   kwsys::DynamicLoader::SymbolPointer f
  74     = kwsys::DynamicLoader::GetSymbolAddress(l, symbol);
  75   if( (r2 && !f) || (!r2 && f) )
  76     {
  77     kwsys_ios::cerr
  78       << kwsys::DynamicLoader::LastError() << kwsys_ios::endl;
  79     return 1;
  80     }
  81 #ifndef __APPLE__
  82   int s = kwsys::DynamicLoader::CloseLibrary(l);
  83   if( (r3 && !s) || (!r3 && s) )
  84     {
  85     kwsys_ios::cerr
  86       << kwsys::DynamicLoader::LastError() << kwsys_ios::endl;
  87     return 1;
  88     }
  89 #else
  90   (void)r3;
  91 #endif
  92   return 0;
  93 }
  94
  95 int testDynamicLoader(int argc, char *argv[])
  96 {
  97 #if defined(_WIN32)
  98   SetErrorMode(SEM_FAILCRITICALERRORS | SEM_NOGPFAULTERRORBOX);
  99 #elif defined(__BEOS__) || defined(__HAIKU__)
 100   disable_debugger(1);
 101 #endif
 102   int res = 0;
 103   if( argc == 3 )
 104     {
 105     // User specify a libname and symbol to check.
 106     res = TestDynamicLoader(argv[1], argv[2],1,1,1);
 107     return res;
 108     }
 109
 110 // dlopen() on Syllable before 11/22/2007 doesn't return 0 on error
 111 #ifndef __SYLLABLE__
 112   // Make sure that inexistent lib is giving correct result
 113   res += TestDynamicLoader("azerty_", "foo_bar",0,0,0);
 114   // Make sure that random binary file cannot be assimilated as dylib
 115   res += TestDynamicLoader(TEST_SYSTEMTOOLS_BIN_FILE, "wp",0,0,0);
 116 #endif
 117
 118 #ifdef __linux__
 119   // This one is actually fun to test, since dlopen is by default loaded...wonder why :)
 120   res += TestDynamicLoader("foobar.lib", "dlopen",0,1,0);
 121   res += TestDynamicLoader("libdl.so", "dlopen",1,1,1);
 122   res += TestDynamicLoader("libdl.so", "TestDynamicLoader",1,0,1);
 123 #endif
 124   // Now try on the generated library
 125   kwsys_stl::string libname = GetLibName(KWSYS_NAMESPACE_STRING "TestDynload");
 126   res += TestDynamicLoader(libname.c_str(), "dummy",1,0,1);
 127   res += TestDynamicLoader(libname.c_str(), "TestDynamicLoaderSymbolPointer",1,1,1);
 128   res += TestDynamicLoader(libname.c_str(), "_TestDynamicLoaderSymbolPointer",1,0,1);
 129   res += TestDynamicLoader(libname.c_str(), "TestDynamicLoaderData",1,1,1);
 130   res += TestDynamicLoader(libname.c_str(), "_TestDynamicLoaderData",1,0,1);
 131
 132   return res;
 133 }