TAO/orbsvcs/tests/Bug_2800_Regression/server.cpp

   1 #include "Hello.h"
   2 #include "orbsvcs/CosNamingC.h"
   3 #include "ace/Get_Opt.h"
   4 #include "ace/Task.h"
   5
   6 const ACE_TCHAR *ior_output_file = ACE_TEXT ("test.ior");
   7
   8 class TestTask : public ACE_Task_Base
   9 {
  10 public:
  11   TestTask (int argc, ACE_TCHAR **argv);
  12   virtual int svc ();
  13
  14   int parse_args (int argc, ACE_TCHAR **argv);
  15
  16   void end();
  17 private:
  18   CORBA::ORB_var orb_;
  19   CORBA::Boolean shutdown_ns_;
  20 };
  21
  22 TestTask::TestTask(int argc, ACE_TCHAR **argv)
  23 {
  24   orb_ = CORBA::ORB_init (argc, argv, "ServerORB");
  25   shutdown_ns_ = false;
  26   parse_args (argc, argv);
  27 }
  28
  29 void TestTask::end()
  30 {
  31   orb_->shutdown(0);
  32   this->wait();
  33 }
  34
  35 int
  36 TestTask::parse_args (int argc, ACE_TCHAR **argv)
  37 {
  38   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT("o:s"));
  39   int c;
  40
  41   while ((c = get_opts ()) != -1)
  42     switch (c)
  43       {
  44       case 's':
  45         shutdown_ns_ = true;
  46         break;
  47       case 'o':
  48         ior_output_file = get_opts.opt_arg ();
  49         break;
  50       }
  51   // Indicates successful parsing of the command line
  52   return 0;
  53 }
  54
  55 int TestTask::svc()
  56 {
  57
  58   try {
  59     // Get reference to Root POA
  60     CORBA::Object_var obj = orb_->resolve_initial_references("RootPOA");
  61     PortableServer::POA_var poa = PortableServer::POA::_narrow(obj.in());
  62
  63     // Activate POA Manager
  64     PortableServer::POAManager_var mgr = poa->the_POAManager();
  65     mgr->activate();
  66
  67     // Find the Naming Service
  68     obj = orb_->string_to_object ("corbaloc:iiop:1.2@localhost:9932/NameService");
  69     CosNaming::NamingContext_var root =
  70       CosNaming::NamingContext::_narrow(obj.in());
  71
  72     if (CORBA::is_nil(root.in())) {
  73       ACE_ERROR ((LM_ERROR, "Error, Nil Naming Context reference\n"));
  74       return 1;
  75     }
  76     // Bind the example Naming Context, if necessary
  77     CosNaming::NamingContext_var example_nc;
  78     CosNaming::Name name;
  79     name.length(1);
  80     name[0].id = CORBA::string_dup("example");
  81     try
  82     {
  83       obj = root->resolve(name);
  84       example_nc =
  85         CosNaming::NamingContext::_narrow(obj.in());
  86     }
  87     catch (const CosNaming::NamingContext::NotFound&)
  88     {
  89       example_nc = root->bind_new_context(name);
  90     }
  91
  92     // Bind the Test object
  93     name.length(2);
  94     name[1].id = CORBA::string_dup("Hello");
  95
  96     // Create an object
  97     Hello servant(orb_.in ());
  98     PortableServer::ObjectId_var oid = poa->activate_object(&servant);
  99     obj = poa->id_to_reference(oid.in());
 100     root->rebind(name, obj.in());
 101
 102     ACE_DEBUG ((LM_INFO, "Hello object bound in Naming Service B\n"));
 103
 104     name.length(1);
 105     obj = orb_->string_to_object ("corbaloc:iiop:1.2@localhost:9931/NameService");
 106     root = CosNaming::NamingContext::_narrow(obj.in());
 107     root->bind_context (name, example_nc.in ());
 108
 109     ACE_DEBUG ((LM_INFO, "'example' context of NS B bound in Naming Service A\n"));
 110
 111     CORBA::String_var ior =
 112       orb_->object_to_string (obj.in ());
 113
 114     // Output the IOR to the <ior_output_file>
 115     FILE *output_file= ACE_OS::fopen (ior_output_file, "w");
 116     if (output_file == 0)
 117       ACE_ERROR_RETURN ((LM_ERROR,
 118                           "Cannot open output file %s for writing IOR: %C\n",
 119                           ior_output_file,
 120                           ior.in ()),
 121                           1);
 122     ACE_OS::fprintf (output_file, "%s", ior.in ());
 123     ACE_OS::fclose (output_file);
 124
 125     ACE_DEBUG ((LM_INFO, "Wrote IOR file\n"));
 126
 127     // Normally we run the orb and the orb is shutdown by
 128     // calling TestTask::end().
 129     // Accept requests
 130     orb_->run();
 131     orb_->destroy();
 132
 133     return 0;
 134   }
 135   catch (const CORBA::Exception& ex)
 136   {
 137     ex._tao_print_exception ("CORBA exception: ");
 138   }
 139
 140   return -1;
 141 }
 142
 143 int ACE_TMAIN (int argc, ACE_TCHAR *argv[])
 144 {
 145   // Start the Test task
 146   TestTask test_ (argc, argv);
 147   if (test_.activate() == -1)
 148     {
 149       ACE_ERROR_RETURN ((LM_ERROR, "Unable to start test task.\n"), -1);
 150     }
 151
 152   // Wait the tasks to finish.
 153   test_.thr_mgr ()->wait();
 154
 155   return 0;
 156 }