ACE/tests/Bug_2980_Regression_Test.cpp

   1 //FUZZ: disable check_for_lack_ACE_OS
   2
   3 #include "ace/config-lite.h"
   4
   5 #include <iostream>
   6 #include <assert.h>
   7 #include <stdio.h>
   8 #include <string.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11
  12 // This is a non-ACE driver program which loads an ACE-based DLL.  The
  13 // usual ACE-related defines will not apply and we must use
  14 // platform-specific ones.
  15 //
  16 // This test has not been made to work on Windows and vxWorks, yet ...
  17
  18 #if defined (ACE_HAS_THREADS)
  19 #  define CAN_USE_THREADS
  20 #else
  21 #  undef CAN_USE_THREADS
  22 #endif
  23
  24 #if !(defined (WIN32) || defined (ACE_VXWORKS) || defined (ACE_HAS_LYNXOS_178))\
  25     && !defined ACE_FACE_SAFETY_EXTENDED
  26 #  define CAN_RUN_TEST
  27
  28 #  include <dlfcn.h>
  29
  30 namespace {
  31 #  if defined (ACE_DLL_SUFFIX)
  32   const char *DllTestName = "./libBug_2980_Regression" ACE_DLL_SUFFIX;
  33 #else
  34   const char *DllTestName = "./libBug_2980_Regression.so";
  35 #endif /* ACE_DLL_SUFFIX */
  36 }
  37
  38 #  if defined CAN_USE_THREADS
  39 #    include <pthread.h>
  40 #  endif
  41
  42 #else
  43 #  undef CAN_RUN_TEST
  44 #endif
  45
  46 #if defined (__BORLANDC__)
  47 #  define PRINTF std::printf
  48 #else
  49 #  define PRINTF printf
  50 #endif
  51
  52
  53 typedef int (* voidfunction)(void);
  54
  55
  56 #if defined (CAN_RUN_TEST)
  57 static void * dllHandle;
  58 static voidfunction   capi_init = 0;
  59 static voidfunction   capi_fini = 0;
  60 static voidfunction   capi_dosomething = 0;
  61 #endif /* defined (CAN_RUN_TEST) */
  62
  63
  64 extern "C"
  65 void* loadDll(void*)
  66 {
  67 #if defined (CAN_RUN_TEST)
  68
  69   PRINTF ("loadDll - entered\n");
  70   const char *subdir_env = getenv ("ACE_EXE_SUB_DIR");
  71   if (subdir_env)
  72     {
  73       char *dllFile =
  74         (char *) malloc (2 + strlen (subdir_env) + strlen (DllTestName));
  75       strcpy (dllFile, subdir_env);
  76       strcat (dllFile, "/");
  77       strcat (dllFile, DllTestName);
  78       dllHandle = dlopen (dllFile, RTLD_NOW);
  79       free (dllFile);
  80     }
  81   else
  82     {
  83       dllHandle = dlopen (DllTestName, RTLD_NOW);
  84     }
  85
  86   if (dllHandle == 0)
  87   {
  88     PRINTF ("unable to load library: %s\n", dlerror());
  89     assert(dllHandle != 0);
  90   }
  91
  92   void* temp = dlsym (dllHandle, "capi_init");
  93   memcpy (&capi_init, &temp, sizeof (temp));
  94   if (capi_init == 0)
  95   {
  96     PRINTF ("unable to resolve symbol capi_init: %s\n", dlerror());
  97     assert(capi_init != 0);
  98   }
  99
 100   temp = dlsym (dllHandle, "capi_fini");
 101   memcpy (&capi_fini, &temp, sizeof (temp));
 102   if (capi_fini == 0)
 103   {
 104     PRINTF ("unable to resolve symbol capi_fini: %s\n", dlerror());
 105     assert(capi_fini != 0);
 106   }
 107
 108   temp = dlsym (dllHandle, "capi_dosomething");
 109   memcpy (&capi_dosomething, &temp, sizeof (temp));
 110   if (capi_dosomething == 0)
 111   {
 112     PRINTF ("unable to resolve symbol capi_dosomething: %s\n", dlerror());
 113     assert(capi_dosomething != 0);
 114   }
 115   capi_init();
 116   PRINTF ("loadDll - leaving\n");
 117 #endif /* defined (CAN_RUN_TEST) */
 118   return 0;
 119 }
 120
 121 extern "C"
 122 void* unloadDll(void*)
 123 {
 124 #if defined (CAN_RUN_TEST)
 125   PRINTF ("unloadDll - entered\n");
 126   capi_fini();
 127   dlclose(dllHandle);
 128   PRINTF ("unloadDll - leaving\n");
 129 #endif /* defined (CAN_RUN_TEST) */
 130   return 0;
 131 }
 132
 133 void * loadunloadDll(void *pp)
 134 {
 135   loadDll(pp);
 136
 137 #if defined (CAN_RUN_TEST)
 138   assert(capi_dosomething != 0);
 139   capi_dosomething();
 140 #endif /* defined (CAN_RUN_TEST) */
 141
 142   unloadDll(pp);
 143
 144   return 0;
 145 }
 146 // FUZZ: disable check_for_improper_main_declaration
 147 int main (int, char *[])
 148 {
 149 #if !defined (CAN_RUN_TEST)
 150 # ifndef ACE_FACE_SAFETY_EXTENDED
 151   PRINTF ("Terminating because this test has not been designed "
 152           "to run on WIN32 or VXWORKS.\n");
 153 # endif
 154 #else
 155   PRINTF ("main called\n");
 156   PRINTF ("main - calling loadDll\n");
 157
 158 #  if defined (CAN_USE_THREADS)
 159   int result = 0;
 160   pthread_t tid1;
 161   result = pthread_create(&tid1, 0, &loadDll, 0);
 162   if (result != 0)
 163   {
 164     PRINTF ("pthread_create() failed: %d\n", result);
 165     return result;
 166   }
 167
 168   pthread_join(tid1, 0);
 169   PRINTF ("loadDll thread finished and re-joined\n");
 170
 171 #  else
 172
 173   loadDll(0);
 174   PRINTF ("loadDll finished\n");
 175
 176 #  endif /* defined (CAN_USE_THREADS) */
 177
 178   PRINTF ("main - calling unloadDll\n");
 179
 180 #  if defined (CAN_USE_THREADS)
 181   pthread_t tid2;
 182   result = pthread_create(&tid2, 0, &unloadDll, 0);
 183   if (result != 0)
 184   {
 185     PRINTF ("pthread_create() failed: %d\n", result);
 186     return 1;
 187   }
 188   pthread_join(tid2, 0);
 189   PRINTF ("unloadDll thread finished and re-joined\n");
 190
 191 #  else
 192
 193   unloadDll(0);
 194   PRINTF ("unloadDll finished\n");
 195
 196 #  endif /* defined (CAN_USE_THREADS) */
 197
 198   PRINTF ("main finished\n");
 199 #endif /* defined (CAN_RUN_TEST) */
 200
 201   return 0;
 202
 203 }
 204   //FUZZ: enable check_for_lack_ACE_OS