TAO/tests/POA/Persistent_ID/server.cpp

   1
   2 //=============================================================================
   3 /**
   4  *  @file    server.cpp
   5  *
   6  *  In this example of using Persistent IDs.
   7  *
   8  *  @author Irfan Pyarali
   9  */
  10 //=============================================================================
  11
  12
  13 #include "ace/Get_Opt.h"
  14 #include "ace/SString.h"
  15 #include "testS.h"
  16 #include "ace/OS_NS_stdio.h"
  17 #include "tao/PortableServer/PortableServer.h"
  18
  19 class test_i :
  20   public virtual POA_test
  21 {
  22 public:
  23   test_i (CORBA::ORB_ptr orb_ptr,
  24           PortableServer::POA_ptr poa);
  25
  26   void method ();
  27
  28   //FUZZ: disable check_for_lack_ACE_OS
  29   ///FUZZ: enable check_for_lack_ACE_OS
  30   void shutdown ();
  31
  32   test_ptr create_POA ();
  33
  34   void destroy_POA ();
  35
  36   PortableServer::POA_ptr _default_POA ();
  37
  38 protected:
  39   CORBA::ORB_var orb_;
  40   PortableServer::POA_var poa_;
  41   PortableServer::POA_var child_poa_;
  42   PortableServer::ObjectId_var oid_;
  43 };
  44
  45 test_i::test_i (CORBA::ORB_ptr orb,
  46                 PortableServer::POA_ptr poa)
  47   : orb_ (CORBA::ORB::_duplicate (orb)),
  48     poa_ (PortableServer::POA::_duplicate (poa))
  49 {
  50 }
  51
  52 void
  53 test_i::method ()
  54 {
  55   CORBA::Object_var obj =
  56     this->orb_->resolve_initial_references ("POACurrent");
  57
  58   PortableServer::Current_var current =
  59     PortableServer::Current::_narrow (obj.in ());
  60
  61   PortableServer::POA_var poa =
  62     current->get_POA ();
  63
  64   CORBA::String_var poa_name =
  65     poa->the_name ();
  66
  67   ACE_DEBUG ((LM_DEBUG,
  68               "Method invoked on servant in POA = %C\n",
  69               poa_name.in ()));
  70 }
  71
  72 void
  73 test_i::shutdown ()
  74 {
  75   this->orb_->shutdown (false);
  76 }
  77
  78 PortableServer::POA_ptr
  79 test_i::_default_POA ()
  80 {
  81   return PortableServer::POA::_duplicate (this->poa_.in ());
  82 }
  83
  84 test_ptr
  85 test_i::create_POA ()
  86 {
  87   CORBA::PolicyList policies (2);
  88   policies.length (2);
  89
  90   policies[0] =
  91     this->poa_->create_id_assignment_policy
  92     (this->oid_.ptr () == 0 ? PortableServer::SYSTEM_ID :
  93      PortableServer::USER_ID);
  94
  95   policies[1] =
  96     this->poa_->create_lifespan_policy (PortableServer::PERSISTENT);
  97
  98   PortableServer::POAManager_var poa_manager =
  99     this->poa_->the_POAManager ();
 100
 101   ACE_CString name = "childPOA";
 102   this->child_poa_ =
 103     this->poa_->create_POA (name.c_str (),
 104                             poa_manager.in (),
 105                             policies);
 106
 107   // Destroy the policies
 108   for (CORBA::ULong i = 0;
 109        i < policies.length ();
 110        ++i)
 111     {
 112       policies[i]->destroy ();
 113     }
 114
 115   test_i *servant =
 116     new test_i (this->orb_.in (),
 117                 this->child_poa_.in ());
 118
 119   PortableServer::ServantBase_var safe_servant (servant);
 120
 121   if (this->oid_.ptr () == 0)
 122     {
 123       this->oid_ =
 124         this->child_poa_->activate_object (servant);
 125     }
 126   else
 127     {
 128       this->child_poa_->activate_object_with_id (this->oid_.in (),
 129                                                  servant);
 130     }
 131
 132   CORBA::Object_var object = this->child_poa_->id_to_reference (this->oid_.in ());
 133
 134   test_var test = test::_narrow (object.in ());
 135
 136   return test._retn ();
 137 }
 138
 139 void
 140 test_i::destroy_POA ()
 141 {
 142   this->child_poa_->destroy (1, 0);
 143 }
 144
 145 const ACE_TCHAR *ior_file = ACE_TEXT ("ior");
 146
 147 static int
 148 parse_args (int argc, ACE_TCHAR **argv)
 149 {
 150   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("f:"));
 151   int c;
 152
 153   while ((c = get_opts ()) != -1)
 154     switch (c)
 155       {
 156       case 'f':
 157         ior_file = get_opts.opt_arg ();
 158         break;
 159
 160       case '?':
 161       default:
 162         ACE_ERROR_RETURN ((LM_ERROR,
 163                            "usage:  %s "
 164                            "-f ior_file"
 165                            "\n",
 166                            argv [0]),
 167                           -1);
 168       }
 169
 170   return 0;
 171 }
 172
 173 static int
 174 write_ior_to_file (const char *ior)
 175 {
 176   FILE *output_file =
 177     ACE_OS::fopen (ior_file, "w");
 178
 179   if (output_file == 0)
 180     ACE_ERROR_RETURN ((LM_ERROR, "Cannot open output files for writing IOR: %s\n",
 181                        ior_file),
 182                       -1);
 183
 184   u_int result = 0;
 185
 186   result = ACE_OS::fprintf (output_file,
 187                             "%s",
 188                             ior);
 189   if (result != ACE_OS::strlen (ior))
 190     ACE_ERROR_RETURN ((LM_ERROR,
 191                        "ACE_OS::fprintf failed while writing %C to %s\n",
 192                        ior,
 193                        ior_file),
 194                       -1);
 195
 196   ACE_OS::fclose (output_file);
 197
 198   return 0;
 199 }
 200
 201 int
 202 ACE_TMAIN (int argc, ACE_TCHAR *argv[])
 203 {
 204   try
 205     {
 206       CORBA::ORB_var orb =
 207         CORBA::ORB_init (argc, argv);
 208
 209       int result = parse_args (argc, argv);
 210       if (result != 0)
 211         return result;
 212
 213       CORBA::Object_var obj =
 214         orb->resolve_initial_references ("RootPOA");
 215
 216       PortableServer::POA_var root_poa =
 217         PortableServer::POA::_narrow (obj.in ());
 218
 219       PortableServer::POAManager_var poa_manager =
 220         root_poa->the_POAManager ();
 221
 222       test_i servant (orb.in (),
 223                       root_poa.in ());
 224
 225       PortableServer::ObjectId_var id =
 226         root_poa->activate_object (&servant);
 227
 228       CORBA::Object_var object = root_poa->id_to_reference (id.in ());
 229
 230       test_var test =
 231         test::_narrow (object.in ());
 232
 233       CORBA::String_var ior =
 234         orb->object_to_string (test.in ());
 235
 236       int write_result =
 237         write_ior_to_file (ior.in ());
 238       if (write_result != 0)
 239         return write_result;
 240
 241       poa_manager->activate ();
 242
 243       orb->run ();
 244
 245       orb->destroy ();
 246     }
 247   catch (const CORBA::Exception& ex)
 248     {
 249       ex._tao_print_exception ("Exception caught");
 250       return -1;
 251     }
 252
 253   return 0;
 254 }