TAO/tests/TransportCurrent/IIOP/server.cpp

   1 // -*- C++ -*-
   2 #include "ace/Get_Opt.h"
   3 #include "ace/OS_NS_stdio.h"
   4 #include "ace/Task.h"
   5
   6 #include "tao/ORBInitializer_Registry.h"
   7
   8 #include "Current_Test_Impl.h"
   9 #include "Server_Request_Interceptor.h"
  10 #include "Server_ORBInitializer.h"
  11
  12 const ACE_TCHAR *ior_output_file = ACE_TEXT ("server.ior");
  13 int nthreads = 1;
  14 int use_collocated_call = 1;
  15
  16 int
  17 parse_args (int argc, ACE_TCHAR *argv[])
  18 {
  19   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("t:o:n:c:"));
  20   int c;
  21
  22   while ((c = get_opts ()) != -1)
  23     switch (c)
  24       {
  25       case 'c':
  26         use_collocated_call = ACE_OS::atoi (get_opts.opt_arg ());
  27         break;
  28       case 'o':
  29         ior_output_file = get_opts.opt_arg ();
  30         break;
  31       case 't':
  32         nthreads = ACE_OS::atoi (get_opts.opt_arg ());
  33         break;
  34       case 'n': // Does nothing. Added just for symetry with the client.
  35         break;
  36       default:
  37         ACE_ERROR_RETURN ((LM_ERROR,
  38                            "Usage: %s "
  39                            "-o <iorfile>"
  40                            "-t threads "
  41                            "-n n (ignored, if provided) "
  42                            "-c make_collocated_calls "
  43                            "\n",
  44                            argv[0]),
  45                           -1);
  46       }
  47
  48   // Indicates successful parsing of the command line
  49   return 0;
  50 }
  51
  52 /// A helper class to encapsulate a task
  53 class Worker : public ACE_Task_Base
  54 {
  55 public:
  56   Worker (CORBA::ORB_ptr);
  57   virtual int svc (void);
  58
  59 private:
  60   // The ORB
  61   CORBA::ORB_var orb_;
  62 };
  63
  64
  65 /// Ctor
  66 Worker::Worker (CORBA::ORB_ptr orb)
  67   : orb_ (CORBA::ORB::_duplicate (orb))
  68 {
  69 }
  70
  71 /// Test referencing the TC data *inside* the context of a client-side
  72 /// interceptor
  73 int
  74 Worker::svc (void)
  75 {
  76   try
  77     {
  78       this->orb_->run ();
  79     }
  80   catch (const CORBA::Exception& ex)
  81     {
  82       ex._tao_print_exception ("Server: exception raised");
  83     }
  84   return 0;
  85 }
  86
  87
  88
  89 /// Main driver
  90
  91 int
  92 server_main (int argc,
  93              ACE_TCHAR *argv[],
  94              Test::Server_Request_Interceptor *cri)
  95 {
  96
  97 #if TAO_HAS_TRANSPORT_CURRENT == 1
  98
  99   try
 100     {
 101       PortableInterceptor::ORBInitializer_ptr temp_initializer = 0;
 102       ACE_NEW_RETURN (temp_initializer,
 103                       Test::Server_ORBInitializer (cri),
 104                       -1);  // No exceptions yet!
 105
 106       PortableInterceptor::ORBInitializer_var orb_initializer (temp_initializer);
 107
 108       PortableInterceptor::register_orb_initializer (orb_initializer.in ());
 109
 110
 111       CORBA::ORB_var orb = CORBA::ORB_init (argc,
 112                                             argv,
 113                                             "test_orb");
 114
 115       CORBA::Object_var obj =
 116         orb->resolve_initial_references ("RootPOA");
 117
 118       PortableServer::POA_var root_poa =
 119         PortableServer::POA::_narrow (obj.in ());
 120
 121       if (CORBA::is_nil (root_poa.in ()))
 122         ACE_ERROR_RETURN ((LM_ERROR,
 123                            ACE_TEXT ("Server (%P|%t) Unable to obtain")
 124                            ACE_TEXT (" RootPOA reference.\n")),
 125                           -1);
 126
 127       PortableServer::POAManager_var poa_manager =
 128         root_poa->the_POAManager ();
 129
 130       poa_manager->activate ();
 131
 132       if (parse_args (argc, argv) != 0)
 133         return -1;
 134
 135       Current_Test_Impl server_impl (orb.in (),
 136                                      root_poa.in (),
 137                                      use_collocated_call);
 138
 139       obj = server_impl._this ();
 140
 141       Test::Transport::CurrentTest_var server =
 142         Test::Transport::CurrentTest::_narrow (obj.in ());
 143
 144       if (CORBA::is_nil (server.in ()))
 145         ACE_ERROR_RETURN ((LM_ERROR,
 146                            ACE_TEXT ("Server (%P|%t) Unable to obtain ")
 147                            ACE_TEXT ("reference to CurrentTest object.\n")),
 148                           -1);
 149
 150       CORBA::String_var ior =
 151         orb->object_to_string (server.in ());
 152
 153       // If the ior_output_file exists, output the IOR to it.
 154       if (ior_output_file != 0)
 155         {
 156           FILE *output_file = ACE_OS::fopen (ior_output_file, "w");
 157           if (output_file == 0)
 158             ACE_ERROR_RETURN ((LM_ERROR,
 159                                "Server (%P|%t) Cannot write file %s "
 160                                "IOR: %C - %m",
 161                                ior_output_file,
 162                                ior.in ()),
 163                               -1);
 164           ACE_OS::fprintf (output_file, "%s", ior.in ());
 165           ACE_OS::fclose (output_file);
 166         }
 167
 168       // Spawn a number of clients doing the same thing for a
 169       // predetermined number of times
 170       Worker worker (orb.in ());
 171
 172 #if defined (ACE_HAS_THREADS)
 173       if (worker.activate (THR_NEW_LWP | THR_JOINABLE,
 174                            nthreads) != 0)
 175         ACE_ERROR_RETURN ((LM_ERROR,
 176                            ACE_TEXT ("Server (%P|%t) Cannot activate %d threads\n"),
 177                            nthreads),
 178                           -1);
 179       worker.thr_mgr ()->wait ();
 180 #else
 181       if (nthreads > 1)
 182         ACE_ERROR ((LM_WARNING,
 183                     ACE_TEXT ("Server (%P|%t) Cannot use threads other than ")
 184                     ACE_TEXT ("the only one available.\n")));
 185       worker.svc ();
 186 #endif
 187
 188       if (TAO_debug_level >= 1)
 189         ACE_DEBUG ((LM_INFO, ACE_TEXT ("Server (%P|%t) Event loop finished.\n")));
 190
 191       if (!cri->self_test ())
 192         ACE_ERROR ((LM_ERROR,
 193                     ACE_TEXT ("Server (%P|%t) ERROR: Interceptor self_test failed\n")));
 194
 195       server->shutdown ();
 196
 197       orb->destroy ();
 198
 199     }
 200   catch (const CORBA::Exception& ex)
 201     {
 202       ex._tao_print_exception (ACE_TEXT ("Server (%P|%t) ERROR: "));
 203
 204       return -1;
 205     }
 206
 207   ACE_DEBUG ((LM_INFO, ACE_TEXT ("Server (%P|%t) Completed successfuly.\n")));
 208   return 0;
 209 #else /*  TAO_HAS_TRANSPORT_CURRENT == 1 */
 210   ACE_DEBUG ((LM_INFO, ACE_TEXT ("Server (%P|%t) Need TAO_HAS_TRANSPORT_CURRENT enabled to run.\n")));
 211   return 0;
 212 #endif /*  TAO_HAS_TRANSPORT_CURRENT == 1 */
 213 }