1 files changed, 116 insertions, 0 deletions
diff --git a/TAO/tests/Client_Leaks/Client_Task.cpp b/TAO/tests/Client_Leaks/Client_Task.cpp
new file mode 100644
index 00000000000..2ae166c58be
--- /dev/null
+++ b/TAO/tests/Client_Leaks/Client_Task.cpp
@@ -0,0 +1,116 @@
+//
+// $Id$
+//
+
+#include "Client_Task.h"
+
+ACE_RCSID(Client_Leaks, Client_Task, "$Id$")
+
+Client_Task::Client_Task (Test::Process_Factory_ptr process_factory,
+                          int iterations)
+  : process_factory_ (Test::Process_Factory::_duplicate (process_factory))
+  , iterations_ (iterations)
+  , successful_calls_ (0)
+{
+}
+
+int
+Client_Task::successful_calls (void) const
+{
+  return this->successful_calls_;
+}
+
+int
+Client_Task::svc (void)
+{
+  ACE_DEBUG ((LM_DEBUG, "(%P|%t) Starting client task\n"));
+  ACE_DECLARE_NEW_CORBA_ENV;
+
+  int successful_calls = 0;
+
+  ACE_TRY
+    {
+      this->validate_connection (ACE_ENV_SINGLE_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      for (int i = 0; i != this->iterations_; ++i)
+        {
+          int retval = this->one_iteration (ACE_ENV_SINGLE_ARG_PARAMETER);
+          ACE_TRY_CHECK;
+
+          if (retval != 0)
+            successful_calls++;
+
+          if (i % 10 == 0)
+            {
+              ACE_DEBUG ((LM_DEBUG,
+                          "(%P|%t) - Client_Task::svc %d / %d iterations\n",
+                          i, this->iterations_));
+            }
+        }
+    }
+  ACE_CATCHANY
+    {
+      ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION,
+                           "Exception caught:");
+      return -1;
+    }
+  ACE_ENDTRY;
+  ACE_DEBUG ((LM_DEBUG, "(%P|%t) Client task finished\n"));
+
+  ACE_GUARD_RETURN (TAO_SYNCH_MUTEX, ace_mon, this->mutex_, -1);
+  this->successful_calls_ += successful_calls;
+
+  return 0;
+}
+
+void
+Client_Task::validate_connection (ACE_ENV_SINGLE_ARG_DECL)
+{
+  ACE_TRY
+    {
+      for (int i = 0; i != 100; ++i)
+        {
+          (void) this->process_factory_->noop (ACE_ENV_SINGLE_ARG_PARAMETER);
+          ACE_TRY_CHECK;
+        }
+    }
+  ACE_CATCH (CORBA::TRANSIENT, ex)
+    {
+      // Ignore transient exceptions
+    }
+  ACE_ENDTRY;
+  ACE_CHECK;
+}
+
+int
+Client_Task::one_iteration (ACE_ENV_SINGLE_ARG_DECL)
+{
+  ACE_TRY
+    {
+      Test::Process_var process =
+        this->process_factory_->create_new_process (ACE_ENV_SINGLE_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      (void) process->get_process_id (ACE_ENV_SINGLE_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      process->shutdown (ACE_ENV_SINGLE_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      return 1;
+    }
+  ACE_CATCH(Test::Spawn_Failed, ignored)
+    {
+      // Ignore this exception, it is usually caused by a transient
+      // condition
+    }
+  ACE_CATCHANY
+    {
+      ACE_PRINT_EXCEPTION (ACE_ANY_EXCEPTION,
+                           "Exception caught:");
+    }
+  ACE_ENDTRY;
+
+  return 0;
+}