TAO/tests/Bug_1495_Regression/Threaded_Server.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "ace/OS_NS_stdio.h"
   3 #include "test_i.h"
   4 #include "server_interceptor.h"
   5 #include "Server_ORBInitializer.h"
   6 #include "tao/ORBInitializer_Registry.h"
   7
   8 const ACE_TCHAR *ior_output_file = ACE_TEXT("test.ior");
   9 const ACE_TCHAR *ior_input_file = ACE_TEXT("file://thr_server.ior");
  10
  11 int
  12 parse_args (int argc, ACE_TCHAR *argv[])
  13 {
  14   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("i:o:"));
  15   int c;
  16
  17
  18   while ((c = get_opts ()) != -1)
  19     switch (c)
  20       {
  21         case 'i':
  22           ior_input_file = get_opts.opt_arg ();
  23           break;
  24         case 'o':
  25           ior_output_file = get_opts.opt_arg ();
  26           break;
  27         case '?':
  28           default:
  29           ACE_ERROR_RETURN ((LM_ERROR,
  30                            "usage:  %s "
  31                            "-i <iorfile>"
  32                            "-o <iorfile>"
  33                            "\n",
  34                            argv [0]),
  35                           -1);
  36       }
  37   // Indicates successful parsing of the command line
  38   return 0;
  39 }
  40
  41 int
  42 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  43 {
  44   try
  45     {
  46       // Create the ORB initializer.
  47       Server_ORBInitializer *temp_initializer = 0;
  48       ACE_NEW_RETURN (temp_initializer,
  49                       Server_ORBInitializer,
  50                       -1);
  51
  52       PortableInterceptor::ORBInitializer_var initializer =
  53         temp_initializer;
  54
  55       PortableInterceptor::register_orb_initializer (initializer.in ());
  56
  57       // Now initialize the ORB.
  58       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv,
  59                                             "Remote_Server_ORB");
  60
  61       CORBA::Object_var poa_object =
  62         orb->resolve_initial_references ("RootPOA");
  63
  64       if (CORBA::is_nil (poa_object.in ()))
  65         {
  66           ACE_ERROR_RETURN ((LM_ERROR,
  67                              " (%P|%t) Unable to initialize the POA.\n" ),
  68                              1);
  69         }
  70
  71       PortableServer::POA_var root_poa =
  72         PortableServer::POA::_narrow (poa_object.in ());
  73
  74       PortableServer::POAManager_var poa_manager =
  75         root_poa->the_POAManager ();
  76
  77       poa_manager->activate ();
  78
  79       if (parse_args (argc, argv) != 0)
  80         {
  81           return 1;
  82         }
  83
  84       // Create the interceptor.
  85       Echo_Server_Request_Interceptor * server_interceptor =
  86         temp_initializer->server_interceptor();
  87
  88       if (server_interceptor == 0)
  89         {
  90           ACE_ERROR_RETURN ((LM_ERROR,
  91             "(%P|%t) Could not obtain reference to "
  92             "server request interceptor.\n"),
  93             -1);
  94         }
  95
  96       // Pull in the ior from the remote server to use as the forward location.
  97       CORBA::Object_var forward_location = orb->string_to_object (ior_input_file);
  98
  99       if (CORBA::is_nil (forward_location.in ()))
 100         {
 101           ACE_ERROR_RETURN ((LM_ERROR,
 102                              "Object reference <%s> is nil\n",
 103                              ior_input_file),
 104                              1);
 105         }
 106
 107       server_interceptor->forward_reference (forward_location.in ());
 108
 109       Bug1495_i server_impl (orb.in ());
 110
 111       PortableServer::ObjectId_var id =
 112         root_poa->activate_object (&server_impl);
 113
 114       CORBA::Object_var test_obj =
 115         root_poa->id_to_reference (id.in ());
 116
 117       Bug1495_Regression::Bug1495_var server =
 118         Bug1495_Regression::Bug1495::_narrow (test_obj.in ());
 119
 120       CORBA::String_var ior =
 121         orb->object_to_string (server.in ());
 122
 123       // Output the server IOR to a file
 124       if (ior_output_file != 0)
 125         {
 126           FILE *output_file = ACE_OS::fopen (ior_output_file, "w");
 127
 128           if (output_file == 0)
 129             {
 130               ACE_ERROR_RETURN ((LM_ERROR,
 131                                  "Cannot open output file for writing the "
 132                                  "server IOR: %s", ior_output_file),
 133                                  1);
 134             }
 135
 136           ACE_OS::fprintf (output_file, "%s", ior.in ());
 137           ACE_OS::fclose (output_file);
 138         }
 139
 140       ACE_Time_Value tv (15, 0);
 141
 142       orb->run (tv);
 143
 144       if (server_interceptor->forward_location_done() == false)
 145         {
 146           ACE_ERROR ((LM_ERROR, "ERROR: Forward location has not occurred!\n"));
 147         }
 148
 149       ACE_DEBUG ((LM_DEBUG, "Threaded Server event loop finished\n"));
 150       root_poa->destroy (1, 1);
 151
 152       orb->destroy ();
 153     }
 154   catch (const CORBA::Exception& ex)
 155     {
 156       ex._tao_print_exception ("Caught an exception in server:");
 157       return 1;
 158     }
 159
 160   ACE_DEBUG ((LM_DEBUG, "Threaded Server ready\n"));
 161
 162   return 0;
 163 }
 164
 165