1 files changed, 144 insertions, 0 deletions
diff --git a/TAO/orbsvcs/tests/Transport_Current/Framework/Client_Request_Interceptor.h b/TAO/orbsvcs/tests/Transport_Current/Framework/Client_Request_Interceptor.h
index bbda0f5f7a2..1cd7bdebacd 100644
--- a/TAO/orbsvcs/tests/Transport_Current/Framework/Client_Request_Interceptor.h
+++ b/TAO/orbsvcs/tests/Transport_Current/Framework/Client_Request_Interceptor.h
@@ -286,3 +286,147 @@ namespace Test
 #endif /* _MSC_VER */
 
 #endif  /* CLIENT_REQUEST_INTERCEPTOR_H */
+// -*- C++ -*-
+
+//=============================================================================
+/**
+ * @file Client_Request_Interceptor.h
+ *
+ * $Id$
+ *
+ * Implementation header for the client request interceptor for the
+ * Transport::Current test.
+ *
+ * @author Ossama Othman <ossama@uci.edu>
+ */
+//=============================================================================
+
+#ifndef CLIENT_REQUEST_INTERCEPTOR_H
+#define CLIENT_REQUEST_INTERCEPTOR_H
+
+#include "ace/config-all.h"
+
+#if !defined (ACE_LACKS_PRAGMA_ONCE)
+# pragma once
+#endif /* ACE_LACKS_PRAGMA_ONCE */
+
+#include "tao/PI/PI.h"
+#include "tao/PortableInterceptorC.h"
+#include "tao/LocalObject.h"
+#include "tao/ORB.h"
+#include "tao/CORBA_String.h"
+
+#if defined(_MSC_VER)
+#pragma warning(push)
+#pragma warning(disable:4250)
+#endif /* _MSC_VER */
+
+
+namespace Test
+{
+
+  /// A metod doing the actual testing, so we could change it for the
+  /// different traits implementations
+
+  typedef int (*TEST) (CORBA::ORB_ptr ACE_ENV_ARG_DECL_WITH_DEFAULTS);
+
+
+  /**
+   * @class Client_Request_Interceptor
+   *
+   * @brief Client request interceptor that exercises
+   * Transport::Current support.
+   *
+   * This client request interceptor forwards a request to a secondary
+   * object by throwing the PortableInterceptor::ForwardRequest
+   * exception from within the send_request() interception point.  It
+   * only forwards the initial request.  See the README file for
+   * details.
+   */
+  class Client_Request_Interceptor
+    : public virtual PortableInterceptor::ClientRequestInterceptor,
+      public virtual TAO_Local_RefCounted_Object
+  {
+ public:
+
+    /// Constructor.
+    Client_Request_Interceptor (const char *orb_id,
+                                TEST test);
+    /**
+     * @name Methods Required by the Client Request Interceptor
+     * Interface
+     *
+     * These are methods that must be implemented since they are pure
+     * virtual in the abstract base class.  They are the canonical
+     * methods required for all client request interceptors.
+     */
+    //@{
+    /// Return the name of this ClientRequestinterceptor.
+    virtual char * name (ACE_ENV_SINGLE_ARG_DECL_WITH_DEFAULTS)
+      ACE_THROW_SPEC ((CORBA::SystemException));
+
+    virtual void destroy (ACE_ENV_SINGLE_ARG_DECL_WITH_DEFAULTS)
+      ACE_THROW_SPEC ((CORBA::SystemException));
+
+    virtual void send_request (PortableInterceptor::ClientRequestInfo_ptr ri
+                               ACE_ENV_ARG_DECL_WITH_DEFAULTS)
+      ACE_THROW_SPEC ((CORBA::SystemException,
+                       PortableInterceptor::ForwardRequest));
+
+    virtual void send_poll (PortableInterceptor::ClientRequestInfo_ptr ri
+                            ACE_ENV_ARG_DECL_WITH_DEFAULTS)
+      ACE_THROW_SPEC ((CORBA::SystemException));
+
+    virtual void receive_reply (PortableInterceptor::ClientRequestInfo_ptr ri
+                                ACE_ENV_ARG_DECL_WITH_DEFAULTS)
+      ACE_THROW_SPEC ((CORBA::SystemException));
+
+    virtual void receive_exception (PortableInterceptor::ClientRequestInfo_ptr ri
+                                    ACE_ENV_ARG_DECL_WITH_DEFAULTS)
+      ACE_THROW_SPEC ((CORBA::SystemException,
+                       PortableInterceptor::ForwardRequest));
+
+    virtual void receive_other (PortableInterceptor::ClientRequestInfo_ptr ri
+                                ACE_ENV_ARG_DECL_WITH_DEFAULTS)
+      ACE_THROW_SPEC ((CORBA::SystemException,
+                       PortableInterceptor::ForwardRequest));
+    //@}
+
+    CORBA::Long interceptions (void);
+
+  private:
+    void test_transport_current (const ACE_TCHAR* amethod ACE_ENV_ARG_DECL)
+      ACE_THROW_SPEC ((CORBA::SystemException, CORBA::UserException));
+
+  private:
+    ///The ID of the ORB this interceptor is registered with.
+    CORBA::String_var orb_id_;
+
+    /// Pseudo-reference to the ORB this interceptor is registered
+    /// with.
+    CORBA::ORB_var orb_;
+
+    /// The number of requests intercepted by this interceptor.
+    CORBA::ULong request_count_;
+
+    /// Stringified reference to the object the initial request will be
+    /// forwarded to by this interceptor.
+    CORBA::String_var forward_str_;
+
+    /// Synchronize internal state...
+    TAO_SYNCH_MUTEX lock_;
+
+    /// Unique request ID
+    unsigned long requestID_;
+
+    /// Test function
+    TEST test_;
+  };
+
+}; // namespace Test
+
+#if defined(_MSC_VER)
+#pragma warning(pop)
+#endif /* _MSC_VER */
+
+#endif  /* CLIENT_REQUEST_INTERCEPTOR_H */