TAO/tests/Bug_1361_Regression/client.cpp

   1 #include "Echo.h"
   2 #include "ORB_Task.h"
   3 #include "tao/ORB_Core.h"
   4 #include "ace/Get_Opt.h"
   5 #include "ace/Reactor.h"
   6 #include "ace/OS_NS_signal.h"
   7
   8 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
   9 int serverthreads = 4;
  10
  11 int
  12 parse_args (int argc, ACE_TCHAR *argv[]);
  13
  14 class Client_Timer : public ACE_Event_Handler
  15 {
  16 public:
  17   /// Constructor
  18   Client_Timer (ACE_Reactor * reactor)
  19     : ACE_Event_Handler (reactor)
  20   {
  21     this->reference_counting_policy ().value (
  22       ACE_Event_Handler::Reference_Counting_Policy::ENABLED);
  23   }
  24
  25   void activate ()
  26   {
  27     ACE_Time_Value tv (150, 0);
  28     this->reactor()->schedule_timer (this, 0, tv, tv);
  29   }
  30
  31   /// Thread entry point
  32   int handle_timeout (ACE_Time_Value const & , void const *)
  33   {
  34     this->reactor ()->cancel_timer (this);
  35     // kill the application
  36     ACE::terminate_process (ACE_OS::getpid ());
  37     return 0;
  38   }
  39 };
  40
  41 int
  42 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  43 {
  44   try
  45     {
  46       CORBA::ORB_var orb =
  47         CORBA::ORB_init (argc, argv);
  48
  49       CORBA::Object_var poa_object =
  50         orb->resolve_initial_references ("RootPOA");
  51
  52       PortableServer::POA_var root_poa =
  53         PortableServer::POA::_narrow (poa_object.in ());
  54
  55       if (CORBA::is_nil (root_poa.in ()))
  56         ACE_ERROR_RETURN ((LM_ERROR,
  57                            " (%P|%t) Panic: nil RootPOA\n"),
  58                           1);
  59
  60       PortableServer::POAManager_var poa_manager =
  61         root_poa->the_POAManager ();
  62
  63       if (parse_args (argc, argv) != 0)
  64         return 1;
  65
  66       PortableServer::ServantBase_var impl;
  67       {
  68         Echo * tmp = 0;
  69         // ACE_NEW_RETURN is the worst possible way to handle
  70         // exceptions (think: what if the constructor allocates memory
  71         // and fails?), but I'm not in the mood to fight for a more
  72         // reasonable way to handle allocation errors in ACE.
  73         ACE_NEW_RETURN (tmp,
  74                         Echo(orb.in(), 1000 / serverthreads),
  75                         1);
  76         impl = tmp;
  77       }
  78
  79       PortableServer::ObjectId_var id =
  80         root_poa->activate_object (impl.in ());
  81
  82       CORBA::Object_var object_act = root_poa->id_to_reference (id.in ());
  83
  84       Test::Echo_var echo =
  85         Test::Echo::_narrow (object_act.in ());
  86
  87       CORBA::Object_var tmp =
  88         orb->string_to_object(ior);
  89
  90       Test::Echo_Caller_var server =
  91         Test::Echo_Caller::_narrow(tmp.in ());
  92
  93       if (CORBA::is_nil (server.in ()))
  94         {
  95           ACE_ERROR_RETURN ((LM_DEBUG,
  96                              "Nil Test::Echo_Caller reference <%s>\n",
  97                              ior),
  98                             1);
  99         }
 100
 101       poa_manager->activate ();
 102
 103       ORB_Task worker (orb.in());
 104       worker.activate (THR_NEW_LWP | THR_JOINABLE,
 105                        serverthreads);
 106
 107       try
 108         {
 109           for(int i = serverthreads; i; --i)
 110             {
 111               server->start_task(echo.in());
 112             }
 113         }
 114       catch (...)
 115         {
 116         }
 117
 118       Client_Timer * task = new Client_Timer (orb->orb_core()->reactor());
 119       task->activate ();
 120       task->remove_reference ();
 121
 122       orb->run ();
 123
 124       worker.wait ();
 125
 126       ACE_DEBUG ((LM_DEBUG,
 127                   "(%P|%t) client - event loop finished\n"));
 128
 129       // Actually the code here should never be reached.
 130       root_poa->destroy (true, true);
 131
 132       orb->destroy ();
 133     }
 134   catch (const CORBA::Exception& ex)
 135     {
 136       ex._tao_print_exception ("Exception caught:");
 137       return 1;
 138     }
 139
 140   return 0;
 141 }
 142
 143 int
 144 parse_args (int argc, ACE_TCHAR *argv[])
 145 {
 146   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("k:t:"));
 147   int c;
 148
 149   while ((c = get_opts ()) != -1)
 150     switch (c)
 151       {
 152       case 'k':
 153         ior = get_opts.opt_arg ();
 154         break;
 155       case 't':
 156         serverthreads = ACE_OS::atoi(get_opts.opt_arg ());
 157         break;
 158
 159       case '?':
 160       default:
 161         ACE_ERROR_RETURN ((LM_ERROR,
 162                            "usage:  %s "
 163                            "-k <ior> "
 164                            "-t threads "
 165                            "\n",
 166                            argv [0]),
 167                           -1);
 168       }
 169   // Indicates successful parsing of the command line
 170   return 0;
 171 }