TAO/tests/Faults/client.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "ace/OS_NS_unistd.h"
   3
   4 #include "test_i.h"
   5
   6 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
   7 int niterations = 10;
   8 int do_shutdown = 0;
   9 int do_suicide = 0;
  10 int do_self_shutdown = 0;
  11 int do_abort = 0;
  12 int do_crash = 0;
  13
  14 int
  15 parse_args (int argc, ACE_TCHAR *argv[])
  16 {
  17   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("acszxk:i:"));
  18   int c;
  19
  20   while ((c = get_opts ()) != -1)
  21     switch (c)
  22       {
  23       case 'a':
  24         do_abort = 1;
  25         break;
  26
  27       case 'c':
  28         do_crash = 1;
  29         break;
  30
  31       case 'x':
  32         do_shutdown = 1;
  33         break;
  34
  35       case 's':
  36         do_suicide = 1;
  37         break;
  38
  39       case 'z':
  40         do_self_shutdown = 1;
  41         break;
  42
  43       case 'k':
  44         ior = get_opts.opt_arg ();
  45         break;
  46
  47       case 'i':
  48         niterations = ACE_OS::atoi (get_opts.opt_arg ());
  49         break;
  50
  51       case '?':
  52       default:
  53         ACE_ERROR_RETURN ((LM_ERROR,
  54                            "usage:  %s "
  55                            "[-a] [-c] [-x] [-s] [-z] "
  56                            "-k <ior> "
  57                            "-i <niterations> "
  58                            "\n",
  59                            argv [0]),
  60                           -1);
  61       }
  62   // Indicates successful parsing of the command line
  63   return 0;
  64 }
  65
  66 int
  67 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  68 {
  69   try
  70     {
  71       CORBA::ORB_var orb =
  72         CORBA::ORB_init (argc, argv);
  73
  74       CORBA::Object_var poa_object =
  75         orb->resolve_initial_references("RootPOA");
  76
  77       if (CORBA::is_nil (poa_object.in ()))
  78         ACE_ERROR_RETURN ((LM_ERROR,
  79                            " (%P|%t) Unable to initialize the POA.\n"),
  80                           1);
  81
  82       PortableServer::POA_var root_poa =
  83         PortableServer::POA::_narrow (poa_object.in ());
  84
  85       PortableServer::POAManager_var poa_manager =
  86         root_poa->the_POAManager ();
  87
  88       poa_manager->activate ();
  89
  90       if (parse_args (argc, argv) != 0)
  91         return 1;
  92
  93       CORBA::Object_var object =
  94         orb->string_to_object (ior);
  95
  96       Simple_Server_var server =
  97         Simple_Server::_narrow (object.in ());
  98
  99       if (CORBA::is_nil (server.in ()))
 100         {
 101           ACE_ERROR_RETURN ((LM_ERROR,
 102                              "Object reference <%s> is nil.\n",
 103                              ior),
 104                             1);
 105         }
 106
 107       Callback_i callback_impl (orb.in ());
 108
 109       PortableServer::ObjectId_var id =
 110         root_poa->activate_object (&callback_impl);
 111
 112       CORBA::Object_var object_act = root_poa->id_to_reference (id.in ());
 113
 114       Callback_var callback =
 115         Callback::_narrow (object_act.in ());
 116
 117       for (int i = 0; i != niterations; ++i)
 118         {
 119           CORBA::Long r =
 120             server->test_method (0, 0, callback.in ());
 121
 122           if (r != 0)
 123             {
 124               ACE_DEBUG ((LM_DEBUG,
 125                           "(%P|%t) unexpected result = %d for %d",
 126                           r, i));
 127             }
 128         }
 129
 130       if (do_abort)
 131         {
 132           try
 133             {
 134               server->shutdown_now (0);
 135             }
 136           catch (const CORBA::COMM_FAILURE&)
 137             {
 138               // Expected exception, continue....
 139             }
 140         }
 141       else if (do_crash)
 142         {
 143           try
 144             {
 145               server->shutdown_now (1);
 146             }
 147           catch (const CORBA::COMM_FAILURE&)
 148             {
 149               // Expected exception, continue....
 150             }
 151         }
 152       else if (do_suicide)
 153         {
 154           (void) server->test_method (1, 0, callback.in ());
 155           // The shutdown callback could arrive after this twoway invocation
 156           // returned. Wait for it shutdown callback and abort,
 157           // otherwise it will timeout (in run_test.pl).
 158           ACE_OS::sleep (120);
 159         }
 160       else if (do_self_shutdown)
 161         {
 162           (void) server->test_method (1, 1, callback.in ());
 163         }
 164
 165       if (do_shutdown)
 166         {
 167           server->shutdown ();
 168         }
 169
 170       root_poa->destroy (true, true);
 171     }
 172   catch (const CORBA::COMM_FAILURE& x)
 173     {
 174       // For other case this is expected.
 175       if (do_self_shutdown == 0)
 176         {
 177           x._tao_print_exception ("ERROR: Unexpected exception\n");
 178         }
 179     }
 180   catch (const CORBA::Exception& ex)
 181     {
 182       ex._tao_print_exception ("Exception caught in client:");
 183       return -1;
 184     }
 185
 186   return 0;
 187 }