TAO/tests/Bug_1020_Regression/client.cpp

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