TAO/tests/Bug_2683_Regression/client.cpp

   1 #include "TestC.h"
   2 #include "ace/Get_Opt.h"
   3 #include "ace/Task.h"
   4 #include "ace/OS_NS_unistd.h"
   5
   6 class Pinger : public ACE_Task_Base
   7 {
   8 private:
   9   const char * ior_;
  10   CORBA::ORB_var orb_;
  11   bool do_shutdown_;
  12   bool stop_;
  13
  14 public:
  15   Pinger (CORBA::ORB_var &orb, const char *ior)
  16     : ior_ (ior),
  17       orb_(orb),
  18       do_shutdown_ (false),
  19       stop_ (false)
  20   {
  21   }
  22
  23   int svc ()
  24   {
  25     bool keep_going = true;
  26     while (keep_going && !this->stop_)
  27       {
  28         try
  29           {
  30             CORBA::Object_var tmp = this->orb_->string_to_object(this->ior_);
  31
  32             Test::IORTable_Shutdown_Race_var target =
  33               Test::IORTable_Shutdown_Race::_narrow(tmp.in ());
  34             if (CORBA::is_nil (target.in ()))
  35                 ACE_ERROR_RETURN ((LM_DEBUG,
  36                                    "(%P|%t) Nil target reference <%s>\n",
  37                                    this->ior_),
  38                                   1);
  39             target->ping();
  40             if (this->do_shutdown_)
  41               {
  42                 ACE_DEBUG ((LM_DEBUG,"(%P|%t) Calling shutdown\n"));
  43                 this->do_shutdown_ = false;
  44                 target->shutdown ();
  45               }
  46           }
  47         catch (const CORBA::Exception &ex)
  48           {
  49             ACE_DEBUG ((LM_DEBUG,
  50                         "(%P|%t) caught an exception - %C\n",ex._name()));
  51             keep_going = false;
  52           }
  53       }
  54     return 0;
  55   }
  56
  57   void killit()
  58   {
  59     do_shutdown_ = true;
  60   }
  61
  62   void stop ()
  63   {
  64     stop_ = true;
  65   }
  66 };
  67
  68 int port = 0;
  69 ACE_TCHAR const * target_host = ACE_TEXT("localhost");
  70
  71 int
  72 parse_args (int argc, ACE_TCHAR *argv[])
  73 {
  74   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("p:h:"));
  75   int c;
  76
  77   while ((c = get_opts ()) != -1)
  78     switch (c)
  79       {
  80       case 'p':
  81         port = ACE_OS::atoi (get_opts.opt_arg ());
  82         break;
  83       case 'h':
  84         target_host = get_opts.opt_arg ();
  85         break;
  86       case '?':
  87       default:
  88         ACE_ERROR_RETURN ((LM_ERROR,
  89                            "usage:  %s "
  90                            "-p <server_port> "
  91                            "\n",
  92                            argv [0]),
  93                           -1);
  94       }
  95   // Indicates successful parsing of the command line
  96   return 0;
  97 }
  98
  99 int
 100 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 101 {
 102   try
 103     {
 104       CORBA::ORB_var orb =
 105         CORBA::ORB_init (argc, argv);
 106
 107       if (parse_args (argc, argv) != 0)
 108         return 1;
 109
 110       char ior[100];
 111       ACE_OS::sprintf (ior, "corbaloc::%s:%d/Racer",
 112                        ACE_TEXT_ALWAYS_CHAR (target_host), port);
 113
 114       Pinger pinger(orb, ior);
 115
 116       ACE_OS::sleep (1);
 117
 118       ACE_DEBUG ((LM_DEBUG,"(%P|%t) client - starting client threads\n"));
 119
 120       pinger.activate (THR_NEW_LWP | THR_JOINABLE, 5);
 121
 122       ACE_OS::sleep (1);
 123
 124       ACE_DEBUG ((LM_DEBUG,"(%P|%t) client - All running, time to shutdown server\n"));
 125       pinger.killit();
 126
 127       ACE_OS::sleep (2);
 128
 129       ACE_DEBUG ((LM_DEBUG,"(%P|%t) client - Stopping client threads\n"));
 130       pinger.stop ();
 131
 132       pinger.wait();
 133       ACE_DEBUG ((LM_DEBUG,"(%P|%t) client done\n"));
 134     }
 135   catch (const CORBA::Exception &ex)
 136     {
 137       ACE_DEBUG ((LM_DEBUG,"Main caught %s\n",ex._name()));
 138       return 1;
 139     }
 140
 141   return 0;
 142 }