diff options
Diffstat (limited to 'TAO/orbsvcs/tests/Bug_2800_Regression/server.cpp')
-rw-r--r-- | TAO/orbsvcs/tests/Bug_2800_Regression/server.cpp | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/TAO/orbsvcs/tests/Bug_2800_Regression/server.cpp b/TAO/orbsvcs/tests/Bug_2800_Regression/server.cpp new file mode 100644 index 00000000000..28d959ce4a1 --- /dev/null +++ b/TAO/orbsvcs/tests/Bug_2800_Regression/server.cpp @@ -0,0 +1,157 @@ +// $Id$ + +#include "Hello.h" +#include "orbsvcs/CosNamingC.h" +#include "ace/Get_Opt.h" +#include "ace/Task.h" + +const char *ior_output_file = "test.ior"; + +class TestTask : public ACE_Task_Base +{ +public: + TestTask(int argc, char **argv); + virtual int svc(); + + int parse_args (int argc, char **argv); + + void end(); +private: + CORBA::ORB_var orb_; + CORBA::Boolean shutdown_ns_; +}; + +TestTask::TestTask(int argc, char **argv) +{ + orb_ = CORBA::ORB_init(argc, argv, "ServerORB"); + shutdown_ns_ = false; + parse_args(argc, argv); +} + +void TestTask::end() +{ + orb_->shutdown(0); + wait(); +} + +int +TestTask::parse_args (int argc, char **argv) +{ + ACE_Get_Opt get_opts (argc, argv, "o:s"); + int c; + + while ((c = get_opts ()) != -1) + switch (c) + { + case 's': + shutdown_ns_ = true; + break; + case 'o': + ior_output_file = get_opts.opt_arg (); + break; + } + // Indicates sucessful parsing of the command line + return 0; +} + +int TestTask::svc() +{ + + try { + // Get reference to Root POA + CORBA::Object_var obj = orb_->resolve_initial_references("RootPOA"); + PortableServer::POA_var poa = PortableServer::POA::_narrow(obj.in()); + + // Activate POA Manager + PortableServer::POAManager_var mgr = poa->the_POAManager(); + mgr->activate(); + + // Find the Naming Service + obj = orb_->string_to_object ("corbaloc:iiop:1.2@localhost:9932/NameService"); + CosNaming::NamingContext_var root = + CosNaming::NamingContext::_narrow(obj.in()); + + if (CORBA::is_nil(root.in())) { + ACE_ERROR ((LM_ERROR, "Error, Nil Naming Context reference\n")); + return 1; + } + // Bind the example Naming Context, if necessary + CosNaming::NamingContext_var example_nc; + CosNaming::Name name; + name.length(1); + name[0].id = CORBA::string_dup("example"); + try + { + obj = root->resolve(name); + example_nc = + CosNaming::NamingContext::_narrow(obj.in()); + } + catch (const CosNaming::NamingContext::NotFound&) + { + example_nc = root->bind_new_context(name); + } + + // Bind the Test object + name.length(2); + name[1].id = CORBA::string_dup("Hello"); + + // Create an object + Hello servant(orb_.in ()); + PortableServer::ObjectId_var oid = poa->activate_object(&servant); + obj = poa->id_to_reference(oid.in()); + root->rebind(name, obj.in()); + + ACE_DEBUG ((LM_INFO, "Hello object bound in Naming Service B\n")); + + name.length(1); + obj = orb_->string_to_object ("corbaloc:iiop:1.2@localhost:9931/NameService"); + root = CosNaming::NamingContext::_narrow(obj.in()); + root->bind_context (name, example_nc.in ()); + + ACE_DEBUG ((LM_INFO, "'example' context of NS B bound in Naming Service A\n")); + + CORBA::String_var ior = + orb_->object_to_string (obj.in ()); + + // Output the IOR to the <ior_output_file> + FILE *output_file= ACE_OS::fopen (ior_output_file, "w"); + if (output_file == 0) + ACE_ERROR_RETURN ((LM_ERROR, + "Cannot open output file for writing IOR: %s\n", + ior_output_file), + 1); + ACE_OS::fprintf (output_file, "%s", ior.in ()); + ACE_OS::fclose (output_file); + + ACE_DEBUG ((LM_INFO, "Wrote IOR file\n")); + + // Normally we run the orb and the orb is shutdown by + // calling TestTask::end(). + // Accept requests + orb_->run(); + orb_->destroy(); + + return 0; + } + catch (CORBA::Exception& ex) + { + ex._tao_print_exception ("CORBA exception: "); + } + + return -1; +} + +int main(int argc, char* argv[]) +{ + // Start the Test task + TestTask test_(argc, argv); + if (test_.activate() == -1) + { + ACE_ERROR_RETURN ((LM_ERROR, "Unable to start test task.\n"), -1); + } + + // Wait the tasks to finish. + test_.thr_mgr ()->wait(); + + return 0; +} |