TAO/orbsvcs/tests/ImplRepo/Bug_4152_Regression/server.cpp

   1 // server.cpp
   2 // This version uses the Implementation Repository.
   3
   4 #include "Test_i.h"
   5
   6 #include "tao/IORTable/IORTable.h"
   7 #include "tao/PortableServer/Root_POA.h"
   8 #include "tao/ImR_Client/ImR_Client.h"
   9
  10 #include "ace/Get_Opt.h"
  11 #include "ace/Task.h"
  12 #include "ace/streams.h"
  13 #include "ace/OS_NS_unistd.h"
  14
  15 class ORB_Runner : public ACE_Task_Base
  16 {
  17 public:
  18   explicit ORB_Runner (CORBA::ORB_ptr orb) : orb_(CORBA::ORB::_duplicate(orb)) {}
  19   int svc ()
  20   {
  21     this->orb_->run ();
  22     return 0;
  23   }
  24
  25 private:
  26   CORBA::ORB_var orb_;
  27 };
  28
  29 PortableServer::POA_var root_poa;
  30 PortableServer::POA_var poa_a;
  31
  32 void
  33 createPOAs(ACE_CString &base)
  34 {
  35   PortableServer::LifespanPolicy_var life =
  36     root_poa->create_lifespan_policy(PortableServer::PERSISTENT);
  37
  38   PortableServer::IdAssignmentPolicy_var assign =
  39     root_poa->create_id_assignment_policy(PortableServer::USER_ID);
  40
  41   CORBA::PolicyList pols;
  42   pols.length(2);
  43   pols[0] = PortableServer::LifespanPolicy::_duplicate(life.in());
  44   pols[1] = PortableServer::IdAssignmentPolicy::_duplicate(assign.in());
  45
  46   PortableServer::POAManager_var mgr = PortableServer::POAManager::_nil();
  47   ACE_CString poa_name = base + ACE_CString ("_a");
  48   poa_a = root_poa->create_POA(poa_name.c_str(), mgr.in(), pols);
  49 }
  50
  51 void
  52 activatePOAs ()
  53 {
  54   PortableServer::POAManager_var mgr = root_poa->the_POAManager ();
  55   mgr->activate ();
  56   mgr = poa_a->the_POAManager ();
  57   mgr->activate ();
  58 }
  59
  60 int
  61 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  62 {
  63   CORBA::ORB_var orb = CORBA::ORB_init(argc, argv);
  64   ORB_Runner *runner = new ORB_Runner (orb.in ());
  65   int poa_delay = 10;
  66   int shutdown_delay = 0;
  67
  68   ACE_DEBUG ((LM_DEBUG, "(%P|%t) Start server main\n"));
  69
  70   try
  71     {
  72       ACE_Get_Opt get_opts (argc, argv, ACE_TEXT ("p:s:?"));
  73       int c;
  74
  75       while ((c = get_opts ()) != -1)
  76         switch (c)
  77           {
  78           case 'p':
  79             poa_delay = ACE_OS::atoi (get_opts.opt_arg ());
  80             break;
  81           case 's':
  82             shutdown_delay = ACE_OS::atoi (get_opts.opt_arg ());
  83             break;
  84           case '?':
  85             ACE_DEBUG ((LM_DEBUG,
  86                         "usage: %s "
  87                         "-d <seconds to delay before initializing POA> "
  88                         "-s <seconds to delay before exiting main after the ORB destroy>\n",
  89                         argv[0]));
  90             return 1;
  91             break;
  92           }
  93
  94       ACE_OS::sleep (poa_delay);
  95       CORBA::Object_var obj = orb->resolve_initial_references ("RootPOA");
  96       root_poa = PortableServer::POA::_narrow (obj.in ());
  97
  98       runner->activate ();
  99
 100       ACE_CString base = ACE_CString ("TestObject");
 101       createPOAs (base);
 102
 103       PortableServer::Servant_var<Test_i> test_servant = new Test_i (orb.in ());
 104
 105       PortableServer::ObjectId_var object_id =
 106         PortableServer::string_to_ObjectId (base.c_str());
 107
 108       poa_a->activate_object_with_id (object_id.in(), test_servant.in ());
 109       obj = poa_a->id_to_reference (object_id.in());
 110
 111       Test_var tva = Test::_narrow (obj.in());
 112
 113       ACE_DEBUG ((LM_DEBUG, "(%P|%t) Started Server pid <%P> poa delay <%d> shutdown delay <%d>\n", poa_delay, shutdown_delay));
 114
 115       {
 116         ACE_CString status_file = base + ACE_CString(".status");
 117         ofstream out(status_file.c_str (), ios_base::app);
 118         if (!out.good())
 119           {
 120             ACE_DEBUG ((LM_DEBUG, "(%P|%t) server did not get good bit from %s\n", status_file.c_str()));
 121           }
 122         out << ACE_OS::getpid () << endl;
 123         out.close ();
 124       }
 125       ACE_OS::sleep (poa_delay);
 126       activatePOAs ();
 127
 128       ACE_DEBUG ((LM_DEBUG, "(%P|%t) Activated POA pid <%P>\n"));
 129
 130       TAO_Root_POA* tpoa = dynamic_cast<TAO_Root_POA*> (poa_a.in ());
 131       ACE_ASSERT (tpoa != 0);
 132
 133       obj = tpoa->id_to_reference_i (object_id.in (), false);
 134       CORBA::String_var test_ior = orb->object_to_string (obj.in ());
 135       obj = orb->resolve_initial_references("IORTable");
 136       IORTable::Table_var table = IORTable::Table::_narrow (obj.in ());
 137       table->bind(base.c_str (), test_ior.in ());
 138
 139       test_ior = orb->object_to_string (tva.in());
 140       base += "_a";
 141       ACE_DEBUG ((LM_DEBUG, "(%P|%t) %C:\n%C\n", base.c_str(), test_ior.in()));
 142       table->bind (base.c_str (), test_ior.in ());
 143
 144       runner->wait ();
 145
 146       ACE_DEBUG ((LM_DEBUG, "(%P|%t) Destroying POA pid <%P>\n"));
 147
 148       root_poa->destroy(1,1);
 149       orb->destroy();
 150     }
 151   catch(const CORBA::Exception& ex)
 152     {
 153       ex._tao_print_exception ("Server main()");
 154       return 1;
 155     }
 156
 157   delete runner;
 158   orb = CORBA::ORB::_nil ();
 159
 160   ACE_OS::sleep (shutdown_delay);
 161
 162   ACE_DEBUG ((LM_DEBUG, "(%P|%t) Exiting Server pid <%P>\n"));
 163
 164   return 0;
 165 }