1 files changed, 135 insertions, 0 deletions
diff --git a/TAO/tests/TransportCurrent/lib/Client_Request_Interceptor.cpp b/TAO/tests/TransportCurrent/lib/Client_Request_Interceptor.cpp
new file mode 100644
index 00000000000..f62ed6e1df1
--- /dev/null
+++ b/TAO/tests/TransportCurrent/lib/Client_Request_Interceptor.cpp
@@ -0,0 +1,135 @@
+// -*- C++ -*-
+
+#include "ace/Log_Msg.h"
+
+ACE_RCSID (ForwardRequest,
+           Client_Request_Interceptor,
+           "$Id$")
+
+#include "Current_TestC.h"
+#include "Client_Request_Interceptor.h"
+#include "Client_ORBInitializer.h"
+
+namespace Test
+{
+
+  Client_Request_Interceptor::Client_Request_Interceptor (const char *orb_id,
+                                                          TEST test)
+    : orb_id_ (CORBA::string_dup (orb_id))
+    , request_count_ (0)
+    , requestID_ (1)
+    , test_ (test)
+  {
+  }
+
+  CORBA::Long
+  Client_Request_Interceptor::interceptions (void)
+  {
+    return this->request_count_;
+  }
+
+  char *
+  Client_Request_Interceptor::name (void)
+    throw (CORBA::SystemException)
+  {
+    return CORBA::string_dup ("Client_Request_Interceptor");
+  }
+
+  void
+  Client_Request_Interceptor::test_transport_current (const ACE_TCHAR* amethod)
+    throw (CORBA::SystemException, CORBA::UserException)
+  {
+    ACE_DEBUG ((LM_DEBUG,
+                ACE_TEXT ("CRI    (%P|%t) Test accessing Transport Current from %s\n"),
+                amethod));
+
+    ++this->request_count_;
+
+    int tmp = 0;
+    CORBA::ORB_var orb = CORBA::ORB_init (tmp,
+                                          0,
+                                          orb_id_.in ());
+
+    // Call the test function, which will throw an exception on
+    // failure
+    (*this->test_) (orb.in ());
+
+    ACE_DEBUG ((LM_DEBUG,
+                ACE_TEXT ("CRI    (%P|%t) Successfully tested Transport Current from %s\n"),
+                amethod));
+  }
+
+  void
+  Client_Request_Interceptor::destroy (void)
+    throw (CORBA::SystemException)
+  {
+  }
+
+  void
+  Client_Request_Interceptor::send_request (PortableInterceptor::ClientRequestInfo_ptr ri)
+    throw (CORBA::SystemException, PortableInterceptor::ForwardRequest)
+  {
+    //Test TC
+    test_transport_current ("send_request");
+
+    CORBA::Boolean response_expected =
+      ri->response_expected ();
+
+    // Oneway?
+    if (response_expected)
+      ACE_DEBUG ((LM_DEBUG, ACE_TEXT("CRI    (%P|%t) Sending a two-way\n")));
+    else
+      ACE_DEBUG ((LM_DEBUG, ACE_TEXT("CRI    (%P|%t) Sending a one-way\n")));
+
+    // Make the context to send the context to the target
+    IOP::ServiceContext sc;
+    sc.context_id = Test::Transport::CurrentTest::ContextTag;
+
+    // How long can a number really get?
+    char temp[32];
+    {
+      ACE_GUARD (TAO_SYNCH_MUTEX, monitor, this->lock_);
+
+      ACE_OS::sprintf (temp, "%d", this->requestID_);
+      this->requestID_++;
+    }
+
+    CORBA::ULong string_len = ACE_OS::strlen (temp) + 1;
+    CORBA::Octet *buf = CORBA::OctetSeq::allocbuf (string_len);
+    ACE_OS::strcpy (reinterpret_cast <char *> (buf), temp);
+
+    sc.context_data.replace (string_len, string_len, buf, 1);
+
+    // Add this context to the service context list.
+    ri->add_request_service_context (sc, 0);
+  }
+
+  void
+  Client_Request_Interceptor::send_poll (PortableInterceptor::ClientRequestInfo_ptr)
+    throw (CORBA::SystemException)
+  {
+    test_transport_current ("send_poll");
+  }
+
+  void
+  Client_Request_Interceptor::receive_reply (PortableInterceptor::ClientRequestInfo_ptr)
+    throw (CORBA::SystemException)
+  {
+    test_transport_current ("receive_reply");
+  }
+
+  void
+  Client_Request_Interceptor::receive_exception (PortableInterceptor::ClientRequestInfo_ptr)
+    throw (CORBA::SystemException, PortableInterceptor::ForwardRequest)
+  {
+    test_transport_current ("receive_exception");
+  }
+
+  void
+  Client_Request_Interceptor::receive_other (PortableInterceptor::ClientRequestInfo_ptr)
+    throw (CORBA::SystemException, PortableInterceptor::ForwardRequest)
+  {
+    test_transport_current ("receive_other");
+  }
+
+}