TAO/orbsvcs/tests/InterfaceRepo/Application_Test/server.cpp

   1 // -*- C++ -*-
   2 #include "test_i.h"
   3 #include "ace/OS_NS_stdio.h"
   4 #include "orbsvcs/Shutdown_Utilities.h"
   5 #include "ace/Get_Opt.h"
   6
   7
   8
   9 static const ACE_TCHAR *ior_output_file = ACE_TEXT("iorfile");
  10
  11 int
  12 parse_args (int argc, ACE_TCHAR *argv[])
  13 {
  14   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:"));
  15   int c;
  16
  17   while ((c = get_opts ()) != -1)
  18     switch (c)
  19       {
  20       case 'o':
  21         ior_output_file = get_opts.opt_arg ();
  22         break;
  23
  24       case '?':
  25       default:
  26         ACE_ERROR_RETURN ((LM_ERROR,
  27                            "usage:  %s "
  28                            "-o <ior> "
  29                            "\n",
  30                            argv [0]),
  31                           -1);
  32       }
  33   // Indicates successful parsing of the command line
  34   return 0;
  35 }
  36
  37 class Service_Shutdown_Functor : public Shutdown_Functor
  38 {
  39 public:
  40   Service_Shutdown_Functor (CORBA::ORB_ptr orb)
  41     : orb_(CORBA::ORB::_duplicate (orb))
  42   {
  43   }
  44
  45   void operator() (int which_signal)
  46   {
  47     ACE_DEBUG ((LM_DEBUG,
  48                 "shutting down on signal %d\n", which_signal));
  49     (void) this->orb_->shutdown ();
  50   }
  51
  52 private:
  53   CORBA::ORB_var orb_;
  54 };
  55
  56 static int
  57 write_ior_to_file (const char *ior)
  58 {
  59   FILE *output_file = ACE_OS::fopen (ior_output_file, "w");
  60
  61   if (output_file == 0)
  62     {
  63       ACE_ERROR_RETURN ((LM_ERROR,
  64                          "Cannot open output files for writing IOR: %s\n",
  65                          ior_output_file),
  66                         -1);
  67     }
  68
  69   int result = ACE_OS::fprintf (output_file,
  70                                 "%s",
  71                                 ior);
  72   if (result < 0)
  73     {
  74       ACE_ERROR_RETURN ((LM_ERROR,
  75                          "ACE_OS::fprintf failed while writing %s to %s\n",
  76                          ior,
  77                          ior_output_file),
  78                         -1);
  79     }
  80
  81   ACE_OS::fclose (output_file);
  82
  83   return 0;
  84 }
  85
  86 int
  87 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
  88 {
  89   try
  90     {
  91       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
  92
  93       if (parse_args (argc, argv) != 0)
  94               return 1;
  95
  96       Service_Shutdown_Functor killer (orb.in ());
  97       Service_Shutdown kill_contractor (killer);
  98
  99       CORBA::Object_var poa_object =
 100         orb->resolve_initial_references ("RootPOA");
 101
 102       PortableServer::POA_var root_poa =
 103         PortableServer::POA::_narrow (poa_object.in ());
 104
 105       PortableServer::POAManager_var poa_manager =
 106         root_poa->the_POAManager ();
 107
 108       poa_manager->activate ();
 109
 110       inventory_i servant_impl;
 111
 112       warehouse::inventory_var servant =
 113         servant_impl._this ();
 114
 115       CORBA::String_var ior =
 116         orb->object_to_string (servant.in ());
 117
 118       int write_result = write_ior_to_file (ior.in ());
 119
 120       if (write_result != 0)
 121         {
 122           ACE_ERROR_RETURN ((LM_ERROR,
 123                              "%p\n",
 124                              "write_ior_to_file"),
 125                             -1);
 126         }
 127
 128       orb->run ();
 129
 130       root_poa->destroy (1,
 131                          1);
 132
 133       orb->destroy ();
 134     }
 135   catch (const CORBA::Exception& ex)
 136     {
 137       ex._tao_print_exception ("server exception:");
 138       return 1;
 139     }
 140
 141   return 0;
 142 }