1 files changed, 125 insertions, 0 deletions
diff --git a/TAO/tests/Portable_Interceptors/ForwardRequest/Client_Request_Interceptor.cpp b/TAO/tests/Portable_Interceptors/ForwardRequest/Client_Request_Interceptor.cpp
new file mode 100644
index 00000000000..7d20dc97510
--- /dev/null
+++ b/TAO/tests/Portable_Interceptors/ForwardRequest/Client_Request_Interceptor.cpp
@@ -0,0 +1,125 @@
+// -*- C++ -*-
+
+#include "Client_Request_Interceptor.h"
+#include "testC.h"
+
+#include "ace/Log_Msg.h"
+
+ACE_RCSID (ForwardRequest,
+           Client_Request_Interceptor,
+           "$Id$")
+
+Client_Request_Interceptor::Client_Request_Interceptor (
+  const char *orb_id,
+  const char *forward_str)
+  : orb_id_ (CORBA::string_dup (orb_id)),
+    orb_ (),
+    request_count_ (0),
+    forward_str_ (CORBA::string_dup (forward_str))
+{
+}
+
+char *
+Client_Request_Interceptor::name ()
+{
+  return CORBA::string_dup ("Client_Request_Interceptor");
+}
+
+void
+Client_Request_Interceptor::destroy (void)
+{
+}
+
+void
+Client_Request_Interceptor::send_request (
+      PortableInterceptor::ClientRequestInfo_ptr ri)
+{
+  ++this->request_count_;
+
+  CORBA::Boolean response_expected =
+    ri->response_expected ();
+
+  if (!response_expected)   // A one-way request.
+    return;
+
+  // Request 1 -- non-forwarded
+  // Request 2 -- forwarded by this interception point.
+
+  if (this->request_count_ == 2)
+    {
+      if (CORBA::is_nil (this->orb_.in ()))
+        {
+          int argc = 0;
+          this->orb_ = CORBA::ORB_init (argc,
+                                        0,
+                                        this->orb_id_.in ());
+        }
+
+      CORBA::Object_var forward =
+        this->orb_->string_to_object (this->forward_str_.in ());
+
+      CORBA::String_var forward_str =
+        this->orb_->object_to_string (forward.in ());
+
+      ACE_DEBUG ((LM_DEBUG,
+                  "CLIENT (%P|%t) Request %d will be forwarded "
+                  "to object 2\n"
+                  "CLIENT (%P|%t) via send_request().\n",
+                  this->request_count_));
+
+      // Notice that this is not a permanent forward.
+      throw PortableInterceptor::ForwardRequest (forward.in ());
+    }
+}
+
+void
+Client_Request_Interceptor::send_poll (
+    PortableInterceptor::ClientRequestInfo_ptr)
+{
+}
+
+void
+Client_Request_Interceptor::receive_reply (
+    PortableInterceptor::ClientRequestInfo_ptr)
+{
+}
+
+void
+Client_Request_Interceptor::receive_exception (
+    PortableInterceptor::ClientRequestInfo_ptr)
+{
+}
+
+void
+Client_Request_Interceptor::receive_other (
+    PortableInterceptor::ClientRequestInfo_ptr ri)
+{
+
+  CORBA::Boolean response_expected =
+    ri->response_expected ();
+
+  // Oneway
+  if (!response_expected)
+    return;
+
+  PortableInterceptor::ReplyStatus reply_status =
+    ri->reply_status ();
+
+  if (reply_status == PortableInterceptor::TRANSPORT_RETRY)
+    return;
+
+  // If we get this far then we should have received a
+  // LOCATION_FORWARD reply, either from another client request
+  // interceptor (not this one) or from the server.
+
+  // This will throw an exception if a location forward has not
+  // occured.  If an exception is thrown then something is wrong with
+  // the PortableInterceptor::ForwardRequest support.
+  CORBA::Object_var forward = ri->forward_reference ();
+
+  if (CORBA::is_nil (forward.in ()))
+    throw CORBA::INTERNAL ();
+
+  ACE_DEBUG ((LM_DEBUG,
+              "CLIENT (%P|%t) Received LOCATION_FORWARD reply.\n"));
+}