TAO/tests/Bug_2936_Regression/DllORB.cpp

   1 #include "DllORB.h"
   2 #include "ace/Barrier.h"
   3 #include "ace/Arg_Shifter.h"
   4 #include "ace/Argv_Type_Converter.h"
   5 #include "ace/OS_NS_unistd.h"
   6 #include "tao/TAO_Singleton_Manager.h"
   7
   8
   9 DllORB::DllORB ()
  10  : failPrePostInit_ (3),
  11    mp_barrier_ (0),
  12    mv_orb_ (),
  13    mv_rootPOA_ ()
  14 {
  15 }
  16
  17
  18 DllORB::~DllORB ()
  19 {
  20   delete mp_barrier_;
  21 }
  22
  23
  24 int
  25 DllORB::init (int argc, ACE_TCHAR *argv[])
  26 {
  27   int threadCnt = 1;
  28
  29   try
  30     {
  31       ACE_Arg_Shifter as (argc, argv);
  32       const ACE_TCHAR *currentArg = 0;
  33       while (as.is_anything_left ())
  34         {
  35           if ((currentArg = as.get_the_parameter (ACE_TEXT ("-NumThreads"))))
  36             {
  37               int num = ACE_OS::atoi (currentArg);
  38               if (num >= 1)
  39                 threadCnt = num;
  40               as.consume_arg ();
  41             }
  42         else
  43           as.ignore_arg ();
  44         }
  45
  46       if (failPrePostInit_ < 3)
  47         {
  48           ACE_DEBUG ((LM_INFO,
  49                       ACE_TEXT ("Pre-ORB initialization ...\n")));
  50           // -----------------------------------------------------------------
  51           // Pre-ORB initialization steps necessary for proper DLL ORB
  52           // support.
  53           // -----------------------------------------------------------------
  54           // Make sure TAO's singleton manager is initialized, and set to not
  55           // register itself with the ACE_Object_Manager since it is under the
  56           // control of the Service Configurator.  If we register with the
  57           // ACE_Object_Manager, then the ACE_Object_Manager will still hold
  58           // (dangling) references to instances of objects created by this
  59           // module and destroyed by this object when it is dynamically
  60           // unloaded.
  61           int register_with_object_manager = 0;
  62           TAO_Singleton_Manager * p_tsm = TAO_Singleton_Manager::instance ();
  63           int result = p_tsm->init (register_with_object_manager);
  64
  65           if (result == -1)
  66             {
  67               if (failPrePostInit_ == 0)
  68                 {
  69                   ACE_DEBUG ((LM_ERROR,
  70                               ACE_TEXT ("Pre-ORB initialization failed.\n")));
  71                   return -1;
  72                 }
  73               else if (failPrePostInit_ < 2)
  74                 {
  75                   ACE_DEBUG ((LM_WARNING,
  76                               ACE_TEXT ("Pre-ORB initialization failed (ignored due to FailPrePostInit setting).\n")));
  77                 }
  78               else
  79                 {
  80                   ACE_DEBUG ((LM_INFO,
  81                               ACE_TEXT ("Pre-ORB initialization failed (ignored due to FailPrePostInit setting).\n")));
  82                 }
  83             }
  84           else
  85             {
  86               ACE_DEBUG ((LM_INFO,
  87                           ACE_TEXT ("Pre-ORB initialization done.\n")));
  88             }
  89         }
  90
  91       // Initialize the ORB
  92       mv_orb_ = CORBA::ORB_init (argc, argv);
  93       if (CORBA::is_nil (mv_orb_.in ()))
  94         {
  95           ACE_DEBUG ((LM_ERROR, ACE_TEXT ("nil ORB\n")));
  96           return -1;
  97         }
  98
  99       CORBA::Object_var v_poa =
 100         mv_orb_->resolve_initial_references ("RootPOA");
 101
 102       mv_rootPOA_ = PortableServer::POA::_narrow (v_poa.in ());
 103       if (CORBA::is_nil (mv_rootPOA_.in ()))
 104         {
 105           ACE_DEBUG ((LM_ERROR, ACE_TEXT ("nil RootPOA\n")));
 106           return -1;
 107         }
 108
 109       mv_poaManager_ = mv_rootPOA_->the_POAManager ();
 110       if (CORBA::is_nil (mv_poaManager_.in ()))
 111         {
 112           ACE_DEBUG ((LM_ERROR, ACE_TEXT ("nil POAManager\n")));
 113           return -1;
 114         }
 115
 116       mv_poaManager_->activate ();
 117     }
 118   catch (...)
 119   {
 120     ACE_DEBUG ((LM_ERROR, ACE_TEXT ("ERROR: exception\n")));
 121     return -1;
 122   }
 123
 124   mp_barrier_ = new ACE_Thread_Barrier (threadCnt + 1);
 125
 126   this->activate (THR_NEW_LWP|THR_JOINABLE|THR_INHERIT_SCHED, threadCnt);
 127   mp_barrier_->wait ();
 128
 129   return 0;
 130 }
 131
 132
 133 int
 134 DllORB::fini ()
 135 {
 136   try
 137     {
 138       mv_poaManager_->deactivate (1, 1);
 139       mv_poaManager_ = PortableServer::POAManager::_nil ();
 140
 141       // attempt to protect against sporadic BAD_INV_ORDER exceptions
 142       ACE_OS::sleep (ACE_Time_Value (0, 500));
 143
 144       mv_orb_->shutdown (true);
 145     }
 146   catch (...)
 147     {
 148       ACE_DEBUG ((LM_ERROR, ACE_TEXT ("ERROR: exception\n")));
 149       return -1;
 150     }
 151
 152   // wait for our threads to finish
 153   this->wait ();
 154
 155   delete mp_barrier_;
 156   mp_barrier_ = 0;
 157
 158   try
 159     {
 160       mv_orb_->destroy ();
 161       mv_orb_ = CORBA::ORB::_nil ();
 162     }
 163   catch (const CORBA::Exception& ex)
 164     {
 165       ex._tao_print_exception ("Exception caught:");
 166       return -1;
 167     }
 168
 169   if (failPrePostInit_ < 3)
 170     {
 171       // -----------------------------------------------------------------
 172       // Post-ORB finalization steps necessary for proper DLL ORB
 173       // support.
 174       // -----------------------------------------------------------------
 175       // Explicitly clean up singletons created by TAO before
 176       // unloading this module.
 177       TAO_Singleton_Manager * p_tsm = TAO_Singleton_Manager::instance ();
 178       int result = p_tsm->fini ();
 179       if (result == -1)
 180         {
 181           if (failPrePostInit_ == 0)
 182             {
 183               ACE_DEBUG ((LM_ERROR,
 184                           ACE_TEXT ("Post-ORB finalization failed.\n")));
 185               return -1;
 186             }
 187           else if (failPrePostInit_ < 2)
 188             {
 189               ACE_DEBUG ((LM_WARNING,
 190                           ACE_TEXT ("Post-ORB finalization failed (ignored due to FailPrePostInit setting).\n")));
 191             }
 192           else
 193             {
 194               ACE_DEBUG ((LM_INFO,
 195                           ACE_TEXT ("Post-ORB finalization failed (ignored due to FailPrePostInit setting).\n")));
 196             }
 197         }
 198       else
 199         {
 200           ACE_DEBUG ((LM_INFO,
 201                       ACE_TEXT ("Post-ORB finalization done.\n")));
 202         }
 203     }
 204
 205   return 0;
 206 }
 207
 208
 209 int DllORB::svc ()
 210 {
 211   mp_barrier_->wait ();
 212
 213   int result = 0;
 214
 215   try
 216     {
 217       mv_orb_->run ();
 218     }
 219   catch (...)
 220     {
 221       ACE_DEBUG ((LM_ERROR, ACE_TEXT ("ERROR: exception\n")));
 222       result = 1;
 223     }
 224
 225   return result;
 226 }
 227
 228
 229 ACE_FACTORY_DEFINE (bug2936, DllORB)