1 files changed, 373 insertions, 0 deletions
diff --git a/TAO/tests/RTCORBA/Persistent_IOR/server.cpp b/TAO/tests/RTCORBA/Persistent_IOR/server.cpp
new file mode 100644
index 00000000000..3802b89a475
--- /dev/null
+++ b/TAO/tests/RTCORBA/Persistent_IOR/server.cpp
@@ -0,0 +1,373 @@
+// $Id$
+
+#include "ace/Get_Opt.h"
+#include "testS.h"
+#include "tao/RTPortableServer/RTPortableServer.h"
+#include "ace/OS_NS_stdio.h"
+#include "ace/OS_NS_string.h"
+
+class test_i :
+  public POA_test
+{
+public:
+  test_i (CORBA::ORB_ptr orb,
+          PortableServer::POA_ptr poa);
+
+  void method (void);
+
+  void shutdown (void);
+
+  PortableServer::POA_ptr _default_POA (void);
+
+private:
+
+  CORBA::ORB_var orb_;
+  PortableServer::POA_var poa_;
+};
+
+test_i::test_i (CORBA::ORB_ptr orb,
+                PortableServer::POA_ptr poa)
+  : orb_ (CORBA::ORB::_duplicate (orb)),
+    poa_ (PortableServer::POA::_duplicate (poa))
+{
+}
+
+void
+test_i::method (void)
+{
+  ACE_DEBUG ((LM_DEBUG,
+              "test_i::method\n"));
+}
+
+void
+test_i::shutdown (void)
+{
+  ACE_DEBUG ((LM_DEBUG,
+              "test_i::shutdown\n"));
+
+  this->orb_->shutdown (0);
+}
+
+PortableServer::POA_ptr
+test_i::_default_POA (void)
+{
+  return PortableServer::POA::_duplicate (this->poa_.in ());
+}
+
+static const char *persistent_ior = "persistent_ior";
+static const char *transient_ior = "transient_ior";
+static const char *tp_persistent_ior = "tp_persistent_ior";
+static int debug = 0;
+
+static CORBA::ULong stacksize = 0;
+static CORBA::ULong static_threads = 2;
+static CORBA::ULong dynamic_threads = 0;
+static RTCORBA::Priority default_thread_priority;
+static CORBA::Boolean allow_request_buffering = 0;
+static CORBA::ULong max_buffered_requests = 0;
+static CORBA::ULong max_request_buffer_size = 0;
+
+int
+parse_args (int argc, char *argv[])
+{
+  ACE_Get_Opt get_opts (argc, argv, "a:p:t:d:");
+  int c;
+
+  while ((c = get_opts ()) != -1)
+    switch (c)
+      {
+      case 'a':
+        tp_persistent_ior = get_opts.opt_arg ();
+        break;
+
+      case 'p':
+        persistent_ior = get_opts.opt_arg ();
+        break;
+
+      case 't':
+        transient_ior = get_opts.opt_arg ();
+        break;
+
+      case 'd':
+        debug = ACE_OS::atoi (get_opts.opt_arg ());
+        break;
+
+      case '?':
+      default:
+        ACE_ERROR_RETURN ((LM_ERROR,
+                           "usage:  %s "
+                           "-a <tp persistent ior file> "
+                           "-p <persistent ior file> "
+                           "-t <transient ior file> "
+                           "-d <debug> "
+                           "\n",
+                           argv [0]),
+                          -1);
+      }
+
+  // Indicates sucessful parsing of the command line
+  return 0;
+}
+
+static void
+write_iors_to_file (CORBA::Object_ptr object,
+                    CORBA::ORB_ptr orb,
+                    const char *filename)
+{
+  FILE *file =
+    ACE_OS::fopen (filename, "w");
+  ACE_ASSERT (file != 0);
+
+  CORBA::String_var ior =
+    orb->object_to_string (object);
+
+  if (debug)
+    ACE_DEBUG ((LM_DEBUG,
+                "%s: %s\n",
+                filename,
+                ior.in ()));
+
+  u_int result = 0;
+  result =
+    ACE_OS::fprintf (file,
+                     "%s",
+                     ior.in ());
+
+  ACE_ASSERT (result == ACE_OS::strlen (ior.in ()));
+  ACE_UNUSED_ARG (result);
+
+  ACE_OS::fclose (file);
+}
+
+void
+create_poa_and_servant_with_persistent_policy (CORBA::ORB_ptr orb,
+                                               PortableServer::POA_ptr root_poa,
+                                               PortableServer::POAManager_ptr poa_manager)
+{
+  CORBA::PolicyList policies;
+
+  CORBA::Policy_var implicit_activation_policy =
+    root_poa->create_implicit_activation_policy (PortableServer::IMPLICIT_ACTIVATION);
+
+  CORBA::Policy_var lifespan_policy =
+    root_poa->create_lifespan_policy (PortableServer::PERSISTENT);
+
+  policies.length (policies.length () + 1);
+  policies[policies.length () - 1] =
+    implicit_activation_policy;
+
+  policies.length (policies.length () + 1);
+  policies[policies.length () - 1] =
+    lifespan_policy;
+
+  PortableServer::POA_var poa =
+    root_poa->create_POA ("persistent_child",
+                          poa_manager,
+                          policies);
+
+  test_i *servant = 0;
+  ACE_NEW_THROW_EX (servant,
+                    test_i (orb,
+                            poa.in ()),
+                    CORBA::NO_MEMORY ());
+
+  PortableServer::ServantBase_var safe_servant (servant);
+
+  test_var test =
+    servant->_this ();
+
+  write_iors_to_file (test.in (),
+                      orb,
+                      persistent_ior);
+}
+
+void
+create_poa_and_servant_with_tp_policy (CORBA::ORB_ptr orb,
+                                       RTCORBA::RTORB_ptr rt_orb,
+                                       PortableServer::POA_ptr root_poa,
+                                       PortableServer::POAManager_ptr poa_manager)
+{
+  RTCORBA::ThreadpoolId threadpool_id =
+    rt_orb->create_threadpool (stacksize,
+                               static_threads,
+                               dynamic_threads,
+                               default_thread_priority,
+                               allow_request_buffering,
+                               max_buffered_requests,
+                               max_request_buffer_size);
+
+  CORBA::Policy_var threadpool_policy =
+    rt_orb->create_threadpool_policy (threadpool_id);
+
+  CORBA::Policy_var implicit_activation_policy =
+    root_poa->create_implicit_activation_policy (PortableServer::IMPLICIT_ACTIVATION);
+
+  CORBA::PolicyList policies;
+
+  policies.length (policies.length () + 1);
+  policies[policies.length () - 1] =
+    implicit_activation_policy;
+
+  policies.length (policies.length () + 1);
+  policies[policies.length () - 1] =
+    threadpool_policy;
+
+  PortableServer::POA_var poa =
+    root_poa->create_POA ("tp_child",
+                          poa_manager,
+                          policies);
+
+  test_i *servant = 0;
+  ACE_NEW_THROW_EX (servant,
+                    test_i (orb,
+                            poa.in ()),
+                    CORBA::NO_MEMORY ());
+
+  PortableServer::ServantBase_var safe_servant (servant);
+
+  test_var test =
+    servant->_this ();
+
+  write_iors_to_file (test.in (),
+                      orb,
+                      transient_ior);
+}
+
+void
+create_poa_with_tp_and_persistent_policy (CORBA::ORB_ptr orb,
+                                          RTCORBA::RTORB_ptr rt_orb,
+                                          PortableServer::POA_ptr root_poa,
+                                          PortableServer::POAManager_ptr poa_manager)
+{
+  RTCORBA::ThreadpoolId threadpool_id =
+    rt_orb->create_threadpool (stacksize,
+                               static_threads,
+                               dynamic_threads,
+                               default_thread_priority,
+                               allow_request_buffering,
+                               max_buffered_requests,
+                               max_request_buffer_size);
+
+  CORBA::Policy_var threadpool_policy =
+    rt_orb->create_threadpool_policy (threadpool_id);
+
+  CORBA::Policy_var implicit_activation_policy =
+    root_poa->create_implicit_activation_policy (PortableServer::IMPLICIT_ACTIVATION);
+
+  CORBA::Policy_var lifespan_policy =
+    root_poa->create_lifespan_policy (PortableServer::PERSISTENT);
+
+  CORBA::PolicyList policies;
+
+  policies.length (policies.length () + 1);
+  policies[policies.length () - 1] =
+    implicit_activation_policy;
+
+  policies.length (policies.length () + 1);
+  policies[policies.length () - 1] =
+    threadpool_policy;
+
+  policies.length (policies.length () + 1);
+  policies[policies.length () - 1] =
+    lifespan_policy;
+
+  PortableServer::POA_var poa =
+    root_poa->create_POA ("tp_persistent_child",
+                          poa_manager,
+                          policies);
+
+  test_i *servant = 0;
+  ACE_NEW_THROW_EX (servant,
+                    test_i (orb,
+                            poa.in ()),
+                    CORBA::NO_MEMORY ());
+
+  PortableServer::ServantBase_var safe_servant (servant);
+
+  test_var test =
+    servant->_this ();
+
+  write_iors_to_file (test.in (),
+                      orb,
+                      tp_persistent_ior);
+}
+
+int
+main (int argc, char **argv)
+{
+  try
+    {
+      CORBA::ORB_var orb =
+        CORBA::ORB_init (argc,
+                         argv,
+                         0);
+
+      int result =
+        parse_args (argc, argv);
+      if (result != 0)
+        return result;
+
+      CORBA::Object_var object =
+        orb->resolve_initial_references ("RTORB");
+
+      RTCORBA::RTORB_var rt_orb =
+        RTCORBA::RTORB::_narrow (object.in ());
+
+      object =
+        orb->resolve_initial_references ("RTCurrent");
+
+      RTCORBA::Current_var current =
+        RTCORBA::Current::_narrow (object.in ());
+
+      default_thread_priority =
+        current->the_priority ();
+
+      object =
+        orb->resolve_initial_references ("RootPOA");
+
+      PortableServer::POA_var root_poa =
+        PortableServer::POA::_narrow (object.in ());
+
+      PortableServer::POAManager_var poa_manager =
+        root_poa->the_POAManager ();
+
+      create_poa_and_servant_with_tp_policy (orb.in (),
+                                             rt_orb.in (),
+                                             root_poa.in  (),
+                                             poa_manager.in ());
+
+      create_poa_and_servant_with_persistent_policy (orb.in (),
+                                                     root_poa.in  (),
+                                                     poa_manager.in ());
+
+      create_poa_with_tp_and_persistent_policy (orb.in (),
+                                                rt_orb.in (),
+                                                root_poa.in  (),
+                                                poa_manager.in ());
+
+      poa_manager->activate ();
+
+      orb->run ();
+
+      orb->destroy ();
+    }
+  catch (const CORBA::INTERNAL& ex)
+    {
+      if ((ex.minor() & 0x1F) == 9) {
+        ACE_DEBUG ((LM_DEBUG, "You must be superuser to run "
+                              "this test on this platform.\n"));
+        return 0;
+      }
+      else {
+        ex._tao_print_exception ("Exception caught");
+        return -1;
+      }
+    }
+  catch (const CORBA::Exception& ex)
+    {
+      ex._tao_print_exception ("Exception caught");
+      return -1;
+    }
+
+  return 0;
+}