TAO/tests/NestedUpcall/MT_Client_Test/local_server.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    local_server.cpp
   5  *
   6  *  This server will run the ORB briefly and then make
   7  *  several calls on the distant MT Object.
   8  *
   9  *  @author Michael Kircher
  10  */
  11 //=============================================================================
  12
  13
  14 #include "local_server.h"
  15 #include "tao/debug.h"
  16 #include "ace/Read_Buffer.h"
  17 #include "ace/OS_NS_unistd.h"
  18 #include "ace/OS_NS_fcntl.h"
  19 #include "tao/Utils/ORB_Manager.h"
  20
  21 MT_Server_Task::MT_Server_Task (ACE_Thread_Manager* thr_mgr_ptr,
  22                                 int argc,
  23                                 ACE_TCHAR **argv,
  24                                 TAO_ORB_Manager* orb_manager_ptr)
  25    :ACE_Task<ACE_SYNCH> (thr_mgr_ptr),
  26     argc_ (argc),
  27     argv_ (argv),
  28     orb_manager_ptr_ (orb_manager_ptr)
  29 {
  30 }
  31
  32 int
  33 MT_Server_Task::svc ()
  34 {
  35   if (this->mT_Server_.init (this->argc_,
  36                              this->argv_,
  37                              this->orb_manager_ptr_) == -1)
  38     return 1;
  39   else
  40     return this->mT_Server_.run_ORB_briefly ();
  41 }
  42
  43
  44 MT_Server::MT_Server ()
  45   : object_key_ (0),
  46     ior_output_file_ (0),
  47     orb_manager_ptr_ (0),
  48     iterations_ (1)
  49 {
  50 }
  51
  52 // Reads the MT Object IOR from a file
  53 int
  54 MT_Server::read_ior (ACE_TCHAR *filename)
  55 {
  56   // Open the file for reading.
  57   ACE_HANDLE f_handle = ACE_OS::open (filename,0);
  58
  59   if (f_handle == ACE_INVALID_HANDLE)
  60     ACE_ERROR_RETURN ((LM_ERROR,
  61                        "Unable to open %s for reading\n",
  62                        filename),
  63                       -1);
  64
  65   ACE_Read_Buffer ior_buffer (f_handle);
  66
  67   this->object_key_ = ior_buffer.read ();
  68   if (this->object_key_ == 0)
  69     ACE_ERROR_RETURN ((LM_ERROR,
  70                        "Unable to allocate memory to read ior: %p\n"),
  71                        -1);
  72
  73   ACE_OS::close (f_handle);
  74   return 0;
  75 }
  76
  77
  78 int
  79 MT_Server::parse_args ()
  80 {
  81   ACE_Get_Opt get_opts (argc_, argv_, ACE_TEXT("d:f:g:h:i:n:s:"));
  82   int c;
  83
  84   while ((c = get_opts ()) != -1)
  85     switch (c)
  86       {
  87       case 'd':  // debug flag.
  88         TAO_debug_level++;
  89         break;
  90       case 'h': // read the IOR from the file.
  91         int result;
  92         result = this->read_ior (get_opts.opt_arg ());
  93         // read IOR for MT Object
  94         if (result < 0)
  95           ACE_ERROR_RETURN ((LM_ERROR,
  96                              "Unable to read ior from %s : %p\n",
  97                              get_opts.opt_arg ()),
  98                             -1);
  99         break;
 100       case 'f':
 101       case 'g':
 102       case 'i':
 103       case 'n':
 104         break;
 105       case 's': this->iterations_ = ACE_OS::atoi (get_opts.opt_arg ());
 106         break;
 107       case '?':
 108       default:
 109         ACE_ERROR_RETURN ((LM_ERROR,
 110                            "usage:  %s"
 111                            " [-d]\n"
 112                            " [-f] first server ior file\n"
 113                            " [-g] second server ior file\n"
 114                            " [-h] third server ior file\n"
 115                            " [-i] client iterations\n"
 116                            " [-n] number of client threads\n"
 117                            " [-s] number of server iterations\n"
 118                            "\n",
 119                            argv_ [0]),
 120                           -1);
 121       }
 122
 123   // Indicates successful parsing of command line.
 124   return 0;
 125 }
 126
 127 int
 128 MT_Server::init (int argc,
 129                  ACE_TCHAR** argv,
 130                  TAO_ORB_Manager* orb_manager_ptr)
 131 {
 132   this->argc_ = argc;
 133   this->argv_ = argv;
 134   if ((this->orb_manager_ptr_ = orb_manager_ptr) == 0)
 135     ACE_ERROR_RETURN ((LM_ERROR,
 136                        "MT_Server::init: ORB_Manager is nil!\n"),
 137                        -1);
 138
 139   try
 140     {
 141       // Call the init of TAO_ORB_Manager to create a child POA
 142       // under the root POA.
 143       this->orb_manager_ptr_->init_child_poa (argc,
 144                                               argv,
 145                                               "child_poa");
 146
 147
 148       this->parse_args ();
 149       // ~~ check for the return value here
 150
 151       this->str_  =
 152         this->orb_manager_ptr_->activate_under_child_poa ("MT",
 153                                                           &this->mT_Object_i_);
 154
 155 #if 0
 156       ACE_DEBUG ((LM_DEBUG,
 157                   "The IOR is: <%s>\n",
 158                   this->str_.in ()));
 159 #endif /*if 0*/
 160
 161       if (this->ior_output_file_)
 162         {
 163           ACE_OS::fprintf (this->ior_output_file_,
 164                            "%s",
 165                            this->str_.in ());
 166           ACE_OS::fclose (this->ior_output_file_);
 167         }
 168
 169       // retrieve the object reference to the distant mt object
 170       if (this->object_key_ == 0)
 171         ACE_ERROR_RETURN ((LM_ERROR,
 172                            "The IOR is nil, not able to get the object.\n"),
 173                           -1);
 174
 175       CORBA::ORB_var orb_var = this->orb_manager_ptr_->orb ();
 176
 177       CORBA::Object_var object_var =
 178         orb_var->string_to_object (this->object_key_);
 179
 180       if (CORBA::is_nil (object_var.in()))
 181         ACE_ERROR_RETURN ((LM_ERROR,
 182                            "No proper object has been returned.\n"),
 183                           -1);
 184
 185       this->mT_Object_var_ = MT_Object::_narrow (object_var.in());
 186
 187       if (CORBA::is_nil (this->mT_Object_var_.in()))
 188         {
 189           ACE_ERROR_RETURN ((LM_ERROR,
 190                              "We have no proper reference to the Object.\n"),
 191                             -1);
 192         }
 193
 194       if (TAO_debug_level > 0)
 195         ACE_DEBUG ((LM_DEBUG, "We have a proper reference to the Object.\n"));
 196     }
 197   catch (const CORBA::Exception& ex)
 198     {
 199       ex._tao_print_exception ("MT_Client::init");
 200       return -1;
 201     }
 202
 203   return 0;
 204 }
 205
 206 int
 207 MT_Server::run ()
 208 {
 209   try
 210     {
 211       int r = this->orb_manager_ptr_->run ();
 212
 213       if (r == -1)
 214         ACE_ERROR_RETURN ((LM_ERROR,
 215                            "MT_Server::run"),
 216                           -1);
 217     }
 218   catch (const CORBA::Exception& ex)
 219     {
 220       ex._tao_print_exception ("MT_Server::run");
 221       return -1;
 222     }
 223   return 0;
 224 }
 225
 226 MT_Server::~MT_Server ()
 227 {
 228   if (this->object_key_ != 0)
 229     ACE_OS::free (this->object_key_);
 230
 231   try
 232     {
 233       if (this->orb_manager_ptr_)
 234         this->orb_manager_ptr_->deactivate_under_child_poa (this->str_.in ());
 235     }
 236   catch (const CORBA::Exception& ex)
 237     {
 238       ex._tao_print_exception ("MT_Client::~MT_Client");
 239     }
 240 }
 241
 242
 243 int
 244 MT_Server::run_ORB_briefly ()
 245 {
 246   if (this->iterations_ > 0)
 247     {
 248       try
 249         {
 250           ACE_DEBUG ((LM_DEBUG,
 251                       "(%P|%t) MT_Server::run: "
 252                       "going to call distant MT Object\n"));
 253
 254           PortableServer::POA_var root_poa =
 255             this->orb_manager_ptr_->root_poa ();
 256
 257           for (unsigned int i = 0; i < this->iterations_; i++)
 258             {
 259               PortableServer::ObjectId_var id =
 260                 root_poa->activate_object (&mT_Object_i_);
 261
 262               CORBA::Object_var object_act = root_poa->id_to_reference (id.in ());
 263
 264               MT_Object_var tmp =
 265                 MT_Object::_narrow (object_act.in ());
 266
 267               this->mT_Object_var_->yadda (0,
 268                                            tmp.in ());
 269             }
 270         }
 271       catch (const CORBA::Exception& ex)
 272         {
 273           ex._tao_print_exception ("MT_Server::run_ORB_briefly");
 274           return -1;
 275         }
 276     }
 277   return 0;
 278 }