TAO/tests/Portable_Interceptors/IORInterceptor/server.cpp

   1 // -*- C++ -*-
   2 #include "ace/Get_Opt.h"
   3
   4 #include "test_i.h"
   5 #include "FOO_IORInterceptor_ORBInitializer.h"
   6
   7 #include "tao/ORBInitializer_Registry.h"
   8 #include "ace/OS_NS_stdio.h"
   9
  10 const ACE_TCHAR *ior_output_file = ACE_TEXT("test.ior");
  11
  12 int
  13 parse_args (int argc, ACE_TCHAR *argv[])
  14 {
  15   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:"));
  16   int c;
  17
  18   while ((c = get_opts ()) != -1)
  19     switch (c)
  20       {
  21       case 'o':
  22         ior_output_file = get_opts.opt_arg ();
  23         break;
  24       default:
  25         ACE_ERROR_RETURN ((LM_ERROR,
  26                            "Usage: %s "
  27                            "-o <iorfile>"
  28                            "\n",
  29                            argv[0]),
  30                           -1);
  31       }
  32
  33   // Indicates successful parsing of the command line
  34   return 0;
  35 }
  36
  37 int
  38 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  39 {
  40   try
  41     {
  42 #if TAO_HAS_INTERCEPTORS == 1
  43       PortableInterceptor::ORBInitializer_ptr temp_initializer =
  44         PortableInterceptor::ORBInitializer::_nil ();
  45
  46       ACE_NEW_RETURN (temp_initializer,
  47                       FOO_IORInterceptor_ORBInitializer,
  48                       -1);  // No CORBA exceptions yet!
  49       PortableInterceptor::ORBInitializer_var orb_initializer =
  50         temp_initializer;
  51
  52       PortableInterceptor::register_orb_initializer (orb_initializer.in ());
  53
  54 #endif /* TAO_HAS_INTERCEPTORS == 1 */
  55
  56       CORBA::ORB_var orb = CORBA::ORB_init (argc,
  57                                             argv,
  58                                             "test_orb");
  59
  60       CORBA::Object_var obj =
  61         orb->resolve_initial_references ("RootPOA");
  62
  63       PortableServer::POA_var root_poa =
  64         PortableServer::POA::_narrow (obj.in ());
  65
  66       if (CORBA::is_nil (root_poa.in ()))
  67         ACE_ERROR_RETURN ((LM_ERROR,
  68                            "Unable to obtain RootPOA reference.\n"),
  69                           -1);
  70
  71       PortableServer::POAManager_var poa_manager =
  72         root_poa->the_POAManager ();
  73
  74       poa_manager->activate ();
  75
  76       if (parse_args (argc, argv) != 0)
  77         return -1;
  78
  79       test_i server_impl (orb.in ());
  80
  81       obj = server_impl._this ();
  82
  83       FOO::test_var server = FOO::test::_narrow (obj.in ());
  84
  85       if (CORBA::is_nil (server.in ()))
  86         ACE_ERROR_RETURN ((LM_ERROR,
  87                            "Unable to obtain reference to FOO::test "
  88                            "object.\n"),
  89                           -1);
  90
  91       CORBA::String_var ior =
  92         orb->object_to_string (server.in ());
  93
  94       ACE_DEBUG ((LM_INFO, "FOO::test: <%C>\n", ior.in ()));
  95
  96       // If the ior_output_file exists, output the IOR to it.
  97       if (ior_output_file != 0)
  98         {
  99           FILE *output_file = ACE_OS::fopen (ior_output_file, "w");
 100           if (output_file == 0)
 101             ACE_ERROR_RETURN ((LM_ERROR,
 102                                "Cannot open output file for writing "
 103                                "IOR: %s",
 104                                ior_output_file),
 105                               1);
 106           ACE_OS::fprintf (output_file, "%s", ior.in ());
 107           ACE_OS::fclose (output_file);
 108         }
 109
 110       orb->run ();
 111
 112       ACE_DEBUG ((LM_INFO, "Event loop finished.\n"));
 113
 114       // The interceptors will be destroyed when we call this
 115       orb->destroy ();
 116     }
 117   catch (const CORBA::Exception& ex)
 118     {
 119       ex._tao_print_exception ("IORInterceptor test (server-side):");
 120
 121       return -1;
 122     }
 123
 124   return 0;
 125 }