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 (void)
  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
  69 int port = 0;
  70 ACE_TCHAR const * target_host = ACE_TEXT("localhost");
  71
  72 int
  73 parse_args (int argc, ACE_TCHAR *argv[])
  74 {
  75   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("p:h:"));
  76   int c;
  77
  78   while ((c = get_opts ()) != -1)
  79     switch (c)
  80       {
  81       case 'p':
  82         port = ACE_OS::atoi (get_opts.opt_arg ());
  83         break;
  84       case 'h':
  85         target_host = get_opts.opt_arg ();
  86         break;
  87       case '?':
  88       default:
  89         ACE_ERROR_RETURN ((LM_ERROR,
  90                            "usage:  %s "
  91                            "-p <server_port> "
  92                            "\n",
  93                            argv [0]),
  94                           -1);
  95       }
  96   // Indicates successful parsing of the command line
  97   return 0;
  98 }
  99
 100 int
 101 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
 102 {
 103   try
 104     {
 105       CORBA::ORB_var orb =
 106         CORBA::ORB_init (argc, argv);
 107
 108       if (parse_args (argc, argv) != 0)
 109         return 1;
 110
 111       char ior[100];
 112       ACE_OS::sprintf (ior, "corbaloc::%s:%d/Racer",
 113                        ACE_TEXT_ALWAYS_CHAR (target_host), port);
 114
 115       Pinger pinger(orb, ior);
 116
 117       ACE_OS::sleep (1);
 118
 119       ACE_DEBUG ((LM_DEBUG,"(%P|%t) client - starting client threads\n"));
 120
 121       pinger.activate (THR_NEW_LWP | THR_JOINABLE, 5);
 122
 123       ACE_OS::sleep (1);
 124
 125       ACE_DEBUG ((LM_DEBUG,"(%P|%t) client - All running, time to shutdown server\n"));
 126       pinger.killit();
 127
 128       ACE_OS::sleep (2);
 129
 130       ACE_DEBUG ((LM_DEBUG,"(%P|%t) client - Stopping client threads\n"));
 131       pinger.stop ();
 132
 133       pinger.wait();
 134       ACE_DEBUG ((LM_DEBUG,"(%P|%t) client done\n"));
 135
 136     }
 137   catch (const CORBA::Exception &ex)
 138     {
 139       ACE_DEBUG ((LM_DEBUG,"Main caught %s\n",ex._name()));
 140       return 1;
 141     }
 142
 143   return 0;
 144 }