TAO/tests/Smart_Proxies/Policy/server.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file     server.cpp
   5  *
   6  *   This is the server program that tests TAO's Smart Proxy extension.
   7  *
   8  *  @author  Kirthika Parameswaran <kirthika@cs.wustl.edu>
   9  */
  10 //=============================================================================
  11
  12
  13 #include "testS.h"
  14 #include "ace/Get_Opt.h"
  15 #include "ace/OS_NS_string.h"
  16
  17 // The servant
  18
  19 class Test_i : public POA_Test
  20 {
  21 public:
  22   Test_i (CORBA::ORB_ptr orb);
  23
  24   CORBA::Short method  (CORBA::Short boo);
  25
  26   //FUZZ: disable check_for_lack_ACE_OS
  27   ///FUZZ: enable check_for_lack_ACE_OS
  28   void shutdown  ();
  29
  30 private:
  31   CORBA::ORB_var orb_;
  32 };
  33
  34 Test_i::Test_i (CORBA::ORB_ptr orb)
  35   : orb_ (CORBA::ORB::_duplicate (orb))
  36 {
  37 }
  38
  39 CORBA::Short
  40 Test_i :: method (CORBA::Short boo)
  41 {
  42   ACE_DEBUG ((LM_DEBUG,
  43               ACE_TEXT ("Test_i::method () invoked\n")));
  44   if (boo == 5)
  45     throw Test::Oops ("Invalid boo\n");
  46
  47   return 0;
  48 }
  49
  50 void
  51 Test_i::shutdown ()
  52 {
  53   this->orb_->shutdown (false);
  54 }
  55
  56 static const ACE_TCHAR *ior_output_file = 0;
  57
  58 int
  59 parse_args (int argc, ACE_TCHAR *argv[])
  60 {
  61   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:"));
  62   int c;
  63
  64   while ((c = get_opts ()) != -1)
  65     switch (c)
  66       {
  67       case 'o':
  68         ior_output_file = get_opts.opt_arg ();
  69         break;
  70       case '?':
  71       default:
  72         ACE_ERROR_RETURN ((LM_ERROR,
  73                            "usage:  %s "
  74                            "-o <iorfile>"
  75                            "\n",
  76                            argv [0]),
  77                           -1);
  78       }
  79   // Indicates successful parsing of the command line
  80   return 0;
  81 }
  82
  83 int
  84 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  85 {
  86   try
  87     {
  88       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
  89
  90       if (parse_args (argc, argv) != 0)
  91         return 1;
  92
  93       Test_i servant (orb.in ());
  94       // Obtain RootPOA.
  95       CORBA::Object_var object =
  96         orb->resolve_initial_references ("RootPOA");
  97
  98       PortableServer::POA_var root_poa =
  99         PortableServer::POA::_narrow (object.in ());
 100
 101       // Get the POAManager of the RootPOA.
 102       PortableServer::POAManager_var poa_manager =
 103         root_poa->the_POAManager ();
 104
 105       PortableServer::ObjectId_var id =
 106         root_poa->activate_object (&servant);
 107
 108       CORBA::Object_var object_act = root_poa->id_to_reference (id.in ());
 109
 110       Test_var Test_object =
 111         Test::_narrow (object_act.in ());
 112
 113       CORBA::String_var ior =
 114         orb->object_to_string (Test_object.in ());
 115
 116       // If the ior_output_file exists, output the ior to it
 117       if (ior_output_file != 0)
 118         {
 119           FILE *output_file =
 120             ACE_OS::fopen (ior_output_file, "w");
 121
 122           if (output_file == 0)
 123             ACE_ERROR_RETURN ((LM_ERROR,
 124                                "Cannot open output file for writing IOR: %s",
 125                                ior_output_file),
 126                               1);
 127
 128           ACE_OS::fprintf (output_file,
 129                            "%s",
 130                            ior.in ());
 131           ACE_OS::fclose (output_file);
 132         }
 133
 134       poa_manager->activate ();
 135
 136       orb->run ();
 137
 138       ACE_DEBUG ((LM_DEBUG,
 139                   "event loop finished\n"));
 140
 141       root_poa->destroy (true, true);
 142
 143       orb->destroy ();
 144     }
 145   catch (const CORBA::Exception& ex)
 146     {
 147       ex._tao_print_exception ("Exception in setting up server");
 148       ACE_ASSERT (0);
 149     }
 150   return 0;
 151 }