TAO/tests/Bug_2909_Regression/client.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    client.cpp
   5  *
   6  * A client which uses the AMI callback model and tests whether exceptions
   7  * are redirected to the reply handler when client and servant are
   8  *
   9  *  @author Johnny Willemsen  <jwillemsen@remedy.nl>
  10  */
  11 //=============================================================================
  12
  13
  14 #include "ace/Get_Opt.h"
  15 #include "ace/Task.h"
  16 #include "ace/Atomic_Op.h"
  17 #include "ace/Synch_Traits.h"
  18 #include "ami_test_i.h"
  19
  20 const ACE_TCHAR *ior = ACE_TEXT("file://test.ior");
  21 int nthreads = 1;
  22 int niterations = 2;
  23 int debug = 0;
  24 ACE_Atomic_Op<TAO_SYNCH_MUTEX, int> number_of_replies = 0;
  25
  26 int invalid_exception = 0;
  27
  28 int
  29 parse_args (int argc, ACE_TCHAR *argv[])
  30 {
  31   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("dk:n:i:"));
  32   int c;
  33
  34   while ((c = get_opts ()) != -1)
  35     switch (c)
  36       {
  37       case 'd':
  38         debug = 1;
  39         break;
  40       case 'k':
  41         ior = get_opts.opt_arg ();
  42         break;
  43       case 'n':
  44         nthreads = ACE_OS::atoi (get_opts.opt_arg ());
  45         break;
  46       case 'i':
  47         niterations = ACE_OS::atoi (get_opts.opt_arg ());
  48         break;
  49       case '?':
  50       default:
  51         ACE_ERROR_RETURN ((LM_ERROR,
  52                            "usage:  %s "
  53                            "-d "
  54                            "-k <ior> "
  55                            "-n <nthreads> "
  56                            "-i <niterations> "
  57                            "\n",
  58                            argv [0]),
  59                           -1);
  60       }
  61   // Indicates successful parsing of the command line
  62   return 0;
  63 }
  64
  65
  66 /**
  67  * @class Client
  68  *
  69  * @brief Run the client thread
  70  *
  71  * Use the ACE_Task_Base class to run the client threads.
  72  */
  73 class Client : public ACE_Task_Base
  74 {
  75 public:
  76   /// ctor
  77   Client (A::AMI_Test_ptr server, int niterations);
  78
  79   /// The thread entry point.
  80   virtual int svc ();
  81
  82   // private:
  83   /// Var for the AMI_Test object.
  84   A::AMI_Test_var ami_test_var_;
  85
  86   /// The number of iterations on each client thread.
  87   int niterations_;
  88
  89   /// Var for AMI_AMI_Test_ReplyHandler object.
  90   A::AMI_AMI_TestHandler_var the_handler_var_;
  91 };
  92
  93 class Handler : public POA_A::AMI_AMI_TestHandler
  94 {
  95 public:
  96   Handler () : request_ (0)
  97   {
  98   };
  99
 100   void foo ()
 101     {
 102       int const reply = --number_of_replies;
 103
 104       if (debug)
 105         {
 106           ACE_DEBUG ((LM_DEBUG,
 107                       "(%P | %t) : Callback method %d called\n",
 108                       (nthreads * niterations) - reply));
 109         }
 110       ACE_UNUSED_ARG (reply);
 111       ++request_;
 112     };
 113
 114    void foo_excep (::Messaging::ExceptionHolder * excep_holder)
 115     {
 116       int const reply = --number_of_replies;
 117       if (debug)
 118         {
 119           ACE_DEBUG ((LM_DEBUG,
 120                       "(%P | %t) : Callback excep method %d called\n",
 121                       (nthreads * niterations) - reply));
 122         }
 123       ACE_UNUSED_ARG (reply);
 124       try
 125         {
 126           excep_holder->raise_exception ();
 127         }
 128       catch (const A::DidTheRightThing& ex)
 129         {
 130           if (request_ != 0)
 131             {
 132               ++invalid_exception;
 133               ex._tao_print_exception ("Caught exception:");
 134             }
 135           else
 136             {
 137               ACE_DEBUG ((LM_DEBUG, "Caught correct exception\n"));
 138             }
 139         }
 140       catch (const CORBA::UNKNOWN& ex)
 141         {
 142           if (request_ != 1)
 143             {
 144               ++invalid_exception;
 145               ex._tao_print_exception ("Caught exception:");
 146             }
 147           else
 148             {
 149               ACE_DEBUG ((LM_DEBUG, "Caught correct exception\n"));
 150             }
 151         }
 152       ++request_;
 153     };
 154
 155   ~Handler ()
 156   {
 157   };
 158 private:
 159   int request_;
 160 };
 161
 162 // ReplyHandler.
 163 Handler handler;
 164
 165 int
 166 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 167 {
 168   try
 169     {
 170       CORBA::ORB_var orb =
 171         CORBA::ORB_init (argc, argv);
 172
 173       if (parse_args (argc, argv) != 0)
 174         return 1;
 175
 176       A::AMI_Test_var server;
 177
 178       AMI_Test_i * servant =
 179         new AMI_Test_i(orb.in());
 180       PortableServer::ServantBase_var safe (servant);
 181
 182       server = servant->_this();
 183
 184       if (CORBA::is_nil (server.in ()))
 185         {
 186           ACE_ERROR_RETURN ((LM_ERROR,
 187                              "Object reference <%s> is nil.\n",
 188                              ior),
 189                             1);
 190         }
 191
 192       // Activate POA to handle the call back.
 193
 194       CORBA::Object_var poa_object =
 195         orb->resolve_initial_references("RootPOA");
 196
 197       if (CORBA::is_nil (poa_object.in ()))
 198         ACE_ERROR_RETURN ((LM_ERROR,
 199                            " (%P|%t) Unable to initialize the POA.\n"),
 200                           1);
 201
 202       PortableServer::POA_var root_poa =
 203         PortableServer::POA::_narrow (poa_object.in ());
 204
 205       PortableServer::POAManager_var poa_manager =
 206         root_poa->the_POAManager ();
 207
 208       poa_manager->activate ();
 209
 210       // Main thread collects replies. It needs to collect
 211       // <nthreads*niterations> replies.
 212       number_of_replies = nthreads * niterations;
 213
 214       // Let the client perform the test in a separate thread
 215
 216       Client client (server.in (), niterations);
 217       if (client.activate (THR_NEW_LWP | THR_JOINABLE,
 218                            nthreads) != 0)
 219         ACE_ERROR_RETURN ((LM_ERROR,
 220                            "Cannot activate client threads\n"),
 221                           1);
 222
 223       if (debug)
 224         {
 225           ACE_DEBUG ((LM_DEBUG,
 226                       "(%P|%t) : Entering perform_work loop to receive <%d> replies\n",
 227                       number_of_replies.value ()));
 228         }
 229
 230       // ORB loop.
 231
 232       while (number_of_replies > 0)
 233         {
 234           CORBA::Boolean pending = orb->work_pending();
 235
 236           if (pending)
 237             {
 238               orb->perform_work();
 239             }
 240
 241           // On some systems this loop must yield or else the other threads
 242           // will not get a chance to run.
 243           ACE_OS::thr_yield();
 244         }
 245
 246       if (debug)
 247         {
 248           ACE_DEBUG ((LM_DEBUG,
 249                       "(%P|%t) : Exited perform_work loop Received <%d> replies\n",
 250                       (nthreads*niterations) - number_of_replies.value ()));
 251         }
 252
 253       client.thr_mgr ()->wait ();
 254
 255       ACE_DEBUG ((LM_DEBUG, "threads finished\n"));
 256
 257       root_poa->destroy (true,  // ethernalize objects
 258                          false); // wait for completion
 259
 260       orb->destroy ();
 261     }
 262   catch (const CORBA::Exception& ex)
 263     {
 264       ex._tao_print_exception ("Caught exception:");
 265       return 1;
 266     }
 267
 268   return invalid_exception;
 269 }
 270
 271 // ****************************************************************
 272
 273 Client::Client (A::AMI_Test_ptr server,
 274                 int niterations)
 275                 :  ami_test_var_ (A::AMI_Test::_duplicate (server)),
 276      niterations_ (niterations)
 277 {
 278   the_handler_var_ = handler._this (/* */);
 279 }
 280
 281 int
 282 Client::svc ()
 283 {
 284   try
 285     {
 286       for (int i = 0; i < this->niterations_; ++i)
 287         {
 288           ami_test_var_->sendc_foo (the_handler_var_.in (), i);
 289         }
 290       if (debug)
 291         {
 292           ACE_DEBUG ((LM_DEBUG,
 293                       "(%P | %t):<%d> Asynchronous methods issued\n",
 294                       niterations));
 295         }
 296     }
 297   catch (const CORBA::Exception& ex)
 298     {
 299       ex._tao_print_exception ("MT_Client: exception raised");
 300       invalid_exception = 1;
 301     }
 302   return 0;
 303 }