summaryrefslogtreecommitdiff
path: root/ACE/TAO/tests/Bug_3683_Regression/Simple_util.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'ACE/TAO/tests/Bug_3683_Regression/Simple_util.cpp')
-rw-r--r--ACE/TAO/tests/Bug_3683_Regression/Simple_util.cpp340
1 files changed, 340 insertions, 0 deletions
diff --git a/ACE/TAO/tests/Bug_3683_Regression/Simple_util.cpp b/ACE/TAO/tests/Bug_3683_Regression/Simple_util.cpp
new file mode 100644
index 00000000000..392f20221cb
--- /dev/null
+++ b/ACE/TAO/tests/Bug_3683_Regression/Simple_util.cpp
@@ -0,0 +1,340 @@
+// $Id$
+
+#ifndef SIMPLE_UTIL_C
+#define SIMPLE_UTIL_C
+
+#include "Simple_util.h"
+#include "tao/IORTable/IORTable.h"
+#include "tao/debug.h"
+#include "ace/OS_NS_stdio.h"
+#include "ace/OS_NS_unistd.h"
+#include "ace/OS_NS_string.h"
+#include "ace/OS_NS_fcntl.h"
+
+// Constructor.
+template <class Servant>
+Server<Servant>::Server (void)
+ : servant_ (0)
+ , ior_output_file_ (0)
+ , ins_ (0)
+{
+ Servant *tmp = 0;
+ ACE_NEW_THROW_EX (tmp,
+ Servant (),
+ CORBA::NO_MEMORY ());
+ this->servant_ = tmp;
+}
+
+// Destructor.
+template <class Servant>
+Server<Servant>::~Server (void)
+{
+}
+
+// Parse the command-line arguments and set options.
+template <class Servant> int
+Server<Servant>::parse_args (void)
+{
+ ACE_Get_Opt get_opts (this->argc_, this->argv_,
+ ACE_TEXT ("do:i:"), 1, 0,
+ ACE_Get_Opt::REQUIRE_ORDER);
+ int c = 0;
+
+ while ((c = get_opts ()) != -1)
+ switch (c)
+ {
+ case 'd': // debug flag.
+ TAO_debug_level++;
+ break;
+ case 'o': // output the IOR to a file.
+ this->ior_output_file_ = get_opts.opt_arg ();
+ break;
+ case 'i': // For Testing the InterOperable Naming Service.
+ this->ins_ = get_opts.opt_arg ();
+ break;
+ }
+
+ // Indicates successful parsing of command line.
+ return 0;
+}
+
+// Add the ObjectID:IOR mapping to the IOR table of
+// the ORB. Ignore this method if you are not testing for
+// the InterOperable Naming Service.
+template <class Servant> int
+Server<Servant>::test_for_ins (const char *ior)
+{
+ CORBA::ORB_var orb = this->orb_manager_.orb ();
+
+ if (TAO_debug_level > 0)
+ ACE_DEBUG ((LM_DEBUG,
+ ACE_TEXT ("Adding (KEY:IOR) %s:%C\n"),
+ this->ins_,
+ ior));
+
+ try
+ {
+ CORBA::Object_var table_object =
+ orb->resolve_initial_references ("IORTable");
+
+ IORTable::Table_var adapter =
+ IORTable::Table::_narrow (table_object.in ());
+ if (CORBA::is_nil (adapter.in ()))
+ {
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("Nil IORTable\n")),
+ -1);
+ }
+
+ adapter->bind (ACE_TEXT_ALWAYS_CHAR (this->ins_), ior);
+ }
+ catch (const CORBA::Exception& ex)
+ {
+ ex._tao_print_exception ("ERROR: test_for_ins failed\n");
+ return -1;
+ }
+
+ return 0;
+}
+
+// Initialize the server.
+template <class Servant> int
+Server<Servant>::init (const char *servant_name,
+ int argc,
+ ACE_TCHAR *argv[])
+{
+ // Call the init of <TAO_ORB_Manager> to initialize the ORB and
+ // create a child POA under the root POA.
+ if (this->orb_manager_.init_child_poa (argc,
+ argv,
+ "child_poa") == -1)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("%p\n"),
+ ACE_TEXT ("init_child_poa")),
+ -1);
+
+ this->argc_ = argc;
+ this->argv_ = argv;
+
+ int retval = this->parse_args ();
+
+ if (retval != 0)
+ return retval;
+
+ CORBA::ORB_var orb = this->orb_manager_.orb ();
+
+ // Stash our ORB pointer for later reference.
+ this->servant_->orb (orb.in ());
+
+ // Activate the servant in its own child POA.
+
+ // Make sure that you check for failures here via the try?!
+ try
+ {
+ CORBA::String_var str =
+ this->orb_manager_.activate_under_child_poa (servant_name,
+ this->servant_.in ());
+
+ ACE_DEBUG ((LM_DEBUG,
+ ACE_TEXT ("The IOR is: <%C>\n"),
+ str.in ()));
+
+ if (this->ins_ && this->test_for_ins (str.in ()) != 0)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("test_for_ins (): failed\n")),
+ -1);
+
+ if (this->ior_output_file_)
+ {
+ FILE *fh = ACE_OS::fopen (this->ior_output_file_, "w");
+ if (fh == 0)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("Unable to open %s for writing (%p)\n"),
+ this->ior_output_file_,
+ ACE_TEXT ("fopen")),
+ -1);
+ ACE_OS::fprintf (fh, "%s", str.in ());
+ ACE_OS::fclose (fh);
+ }
+ }
+ catch (const CORBA::Exception& ex)
+ {
+ ex._tao_print_exception ("\tException in activation of POA");
+ return -1;
+ }
+
+ return 0;
+}
+
+template <class Servant>int
+Server<Servant>::run (void)
+{
+ // Run the main event loop for the ORB.
+ if (this->orb_manager_.run () == -1)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("Server_i::run")),
+ -1);
+
+ return 0;
+}
+
+// Constructor.
+template <class ServerInterface>
+Client<ServerInterface>::Client (void)
+ : ior_ ("")
+ , do_shutdown_ (0)
+{
+ //no-op
+}
+
+// Reads the Server ior from a file
+template <class ServerInterface> int
+Client<ServerInterface>::read_ior (ACE_TCHAR *filename)
+{
+ // Open the file for reading.
+ ACE_HANDLE f_handle = ACE_OS::open (filename, 0);
+
+ if (f_handle == ACE_INVALID_HANDLE)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("Unable to open %s for writing (%p)\n"),
+ filename,
+ ACE_TEXT ("open")),
+ -1);
+
+ ACE_Read_Buffer ior_buffer (f_handle);
+ char *data = ior_buffer.read ();
+
+ if (data == 0)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("Unable to read ior (%p)\n"),
+ ACE_TEXT ("read")),
+ -1);
+
+ this->ior_ = data;
+ ior_buffer.alloc ()->free (data);
+
+ ACE_OS::close (f_handle);
+
+ return 0;
+}
+
+// Parses the command line arguments and returns an error status.
+template <class ServerInterface> int
+Client<ServerInterface>::parse_args (void)
+{
+ ACE_Get_Opt get_opts (argc_, argv_,
+ ACE_TEXT ("df:nk:x"), 1, 0,
+ ACE_Get_Opt::REQUIRE_ORDER);
+ int c = 0;
+ int result = 0;
+
+ while ((c = get_opts ()) != -1)
+ switch (c)
+ {
+ case 'd': // debug flag
+ TAO_debug_level++;
+ break;
+ case 'k': // ior provide on command line
+ this->ior_ = ACE_TEXT_ALWAYS_CHAR(get_opts.opt_arg ());
+ break;
+ case 'f': // read the IOR from the file.
+ result = this->read_ior (get_opts.opt_arg ());
+ if (result < 0)
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("Unable to read ior from %s (%p)\n"),
+ get_opts.opt_arg (),
+ ACE_TEXT ("read_ior")),
+ -1);
+ break;
+ case 'x': // read the flag for shutting down
+ this->do_shutdown_ = 1;
+ break;
+ case 'h': // display help for use of the client.
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("usage: %s")
+ ACE_TEXT (" [-d (debug)]")
+ ACE_TEXT (" [-k] <ior>")
+ ACE_TEXT (" [-f] <ior_output_file>")
+ ACE_TEXT (" [-x (shutdown server)]")
+ ACE_TEXT (" [-h (help)]")
+ ACE_TEXT ("\n"),
+ this->argv_ [0]),
+ -1);
+ }
+
+ // Indicates successful parsing of command line.
+ return 0;
+}
+
+template <class ServerInterface>
+Client<ServerInterface>::~Client (void)
+{
+ this->orb_->destroy ();
+}
+
+template <class ServerInterface> int
+Client<ServerInterface>::init (const char *,
+ int argc,
+ ACE_TCHAR **argv)
+{
+ this->argc_ = argc;
+ this->argv_ = argv;
+
+ try
+ {
+ // Retrieve the ORB.
+ this->orb_ = CORBA::ORB_init (this->argc_, this->argv_);
+
+ // Parse command line and verify parameters.
+ if (this->parse_args () == -1)
+ return -1;
+
+ if (this->ior_.length () != 0)
+ {
+ CORBA::Object_var server_object =
+ this->orb_->string_to_object (this->ior_.c_str ());
+
+ if (CORBA::is_nil (server_object.in ()))
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("invalid ior <%C>\n"),
+ this->ior_.c_str ()),
+ -1);
+ this->server_ = ServerInterface::_narrow (server_object.in ());
+ if (CORBA::is_nil (this->server_.in ()))
+ {
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("Nil Server\n")),
+ -1);
+ }
+ }
+ else
+ ACE_ERROR_RETURN ((LM_ERROR,
+ ACE_TEXT ("no ior or naming options specified\n")),
+ -1);
+
+
+ }
+ catch (const CORBA::Exception& ex)
+ {
+ ex._tao_print_exception ("Client_i::init");
+ return -1;
+ }
+
+ return 0;
+}
+
+template <class ServerInterface> int
+Client<ServerInterface>::do_shutdown (void)
+{
+ // Returns the shutdwon flag
+ return this->do_shutdown_;
+}
+
+template <class ServerInterface> void
+Client<ServerInterface>::do_shutdown (int flag)
+{
+ // Fills the flag
+ this->do_shutdown_ = flag;
+}
+
+#endif /* SIMPLE_UTIL_C */
View Lorry Controller Status