adding DS library

2 files changed, 191 insertions, 0 deletions

diff --git a/TAO/tao/DiffServPolicy/DiffServ_Protocols_Hooks.cpp b/TAO/tao/DiffServPolicy/DiffServ_Protocols_Hooks.cpp
new file mode 100644
index 00000000000..2dc4a686676
--- /dev/null
+++ b/TAO/tao/DiffServPolicy/DiffServ_Protocols_Hooks.cpp
@@ -0,0 +1,118 @@
+#include "tao/DiffServPolicy/DiffServ_Protocols_Hooks.h"
+#include "ace/Log_Msg.h"
+#include "tao/Stub.h"
+#include "tao/ORB_Core.h"
+#include "tao/MProfile.h"
+#include "tao/Service_Context.h"
+#include "tao/Acceptor_Registry.h"
+#include "tao/Thread_Lane_Resources.h"
+#include "tao/Protocol_Factory.h"
+#include "tao/Transport_Acceptor.h"
+#include "tao/Transport_Connector.h"
+#include "tao/Policy_Set.h"
+#include "tao/debug.h"
+#include "tao/CDR.h"
+#include "tao/Client_Network_Priority_Policy.h"
+#include "ace/Dynamic_Service.h"
+#include "ace/OS_NS_string.h"
+
+ACE_RCSID (DiffServPolicy,
+           DiffServ_Protocols_Hooks,
+           "$Id$")
+
+TAO_BEGIN_VERSIONED_NAMESPACE_DECL
+
+TAO_DiffServ_Protocols_Hooks::TAO_RT_Protocols_Hooks (void)
+  : orb_core_ (0)
+  , dscp_codepoint_ ()
+{
+}
+
+
+TAO_DiffServ_Protocols_Hooks::~TAO_DiffServ_Protocols_Hooks (void)
+{
+}
+
+void
+TAO_DiffServ_Protocols_Hooks::init_hooks (TAO_ORB_Core *orb_core
+                                    ACE_ENV_ARG_DECL)
+{
+  this->orb_core_ = orb_core;
+}
+
+void
+TAO_DiffServ_Protocols_Hooks::np_service_context (
+    TAO_Stub *stub,
+    TAO_Service_Context &service_context,
+    CORBA::Boolean restart
+    ACE_ENV_ARG_DECL_NOT_USED)
+{
+  // If the restart flag is true, then this call for a
+  // reinvocation. We need not prepare the Service Context List once
+  // again. We can use the already existing one.
+  if (!restart)
+    {
+      CORBA::Policy_var cnpp =
+        stub->get_cached_policy (TAO_CACHED_POLICY_CLIENT_NETWORK_PRIORITY
+                        ACE_ENV_ARG_PARAMETER);
+      ACE_TRY_CHECK;
+
+      if (!CORBA::is_nil (cnpp.in ()))
+        {
+          TAO::NetworkPriorityPolicy_var cnp =
+            TAO::NetworkPriorityPolicy::_narrow (cnpp.in ()
+                            ACE_ENV_ARG_PARAMETER);
+          ACE_TRY_CHECK;
+
+          TAO::DiffservCodepoint reply_diffserv_codepoint;
+          reply_diffserv_codepoint = cnp->get_reply_diffserv_codepoint ();
+
+          CORBA::Long rep_dscp_codepoint = reply_diffserv_codepoint;
+
+          this->add_rep_np_service_context_hook (service_context,
+                                             rep_dscp_codepoint
+                                             ACE_ENV_ARG_PARAMETER);
+          ACE_CHECK;
+        }
+    }
+}
+
+void
+TAO_DiffServ_Protocols_Hooks::add_rep_np_service_context_hook (
+   TAO_Service_Context &service_context,
+   CORBA::Long &dscp_codepoint
+   ACE_ENV_ARG_DECL_NOT_USED)
+{
+  TAO_OutputCDR cdr;
+  if ((cdr << ACE_OutputCDR::from_boolean (TAO_ENCAP_BYTE_ORDER) == 0)
+      || (cdr << dscp_codepoint) == 0)
+    {
+      ACE_THROW (CORBA::MARSHAL ());
+    }
+
+  service_context.set_context (IOP::REP_NWPRIORITY, cdr);
+}
+
+void
+TAO_DiffServ_Protocols_Hooks::set_dscp_codepoint (CORBA::Long &dscp_codepoint
+                                   ACE_ENV_ARG_DECL_NOT_USED)
+{
+  this->dscp_codepoint_ = dscp_codepoint;
+}
+
+CORBA::Long
+TAO_DiffServ_Protocols_Hooks::get_dscp_codepoint (void)
+{
+  return this->dscp_codepoint_;
+}
+
+TAO_END_VERSIONED_NAMESPACE_DECL
+
+ACE_STATIC_SVC_DEFINE (TAO_DiffServ_Protocols_Hooks,
+                       ACE_TEXT ("DiffServ_Protocols_Hooks"),
+                       ACE_SVC_OBJ_T,
+                       &ACE_SVC_NAME (TAO_DiffServ_Protocols_Hooks),
+                       ACE_Service_Type::DELETE_THIS
+                       | ACE_Service_Type::DELETE_OBJ,
+                       0)
+ACE_FACTORY_DEFINE (TAO_DiffServPolicy, TAO_DiffServ_Protocols_Hooks)
diff --git a/TAO/tao/DiffServPolicy/DiffServ_Protocols_Hooks.h b/TAO/tao/DiffServPolicy/DiffServ_Protocols_Hooks.h
new file mode 100644
index 00000000000..b3e8c6401ba
--- /dev/null
+++ b/TAO/tao/DiffServPolicy/DiffServ_Protocols_Hooks.h
@@ -0,0 +1,73 @@
+// -*- C++ -*-
+
+// ===================================================================
+/**
+ *  @file   DiffServ_Protocols_Hooks.h
+ *
+ *  $Id$
+ *
+ *  @author Jaiganesh Balasubramanian <jai@dre.vanderbilt.edu>
+ *          Johnny Willemsen <jwillemsen@remedy.nl>
+ */
+// ===================================================================
+
+#ifndef TAO_DIFFSERV_PROTOCOLS_HOOKS_H
+#define TAO_DIFFSERVT_PROTOCOLS_HOOKS_H
+
+#include /**/ "ace/pre.h"
+
+#include "tao/orbconf.h"
+
+#include "tao/Protocols_Hooks.h"
+#include "ace/Service_Config.h"
+#include "tao/DiffServPolicy/DiffServPolicy_Export.h"
+
+#if !defined (ACE_LACKS_PRAGMA_ONCE)
+# pragma once
+#endif /* ACE_LACKS_PRAGMA_ONCE */
+
+TAO_BEGIN_VERSIONED_NAMESPACE_DECL
+
+class TAO_DiffServPolicy_Export TAO_DiffServ_Protocols_Hooks : public TAO_Protocols_Hooks
+{
+public:
+
+  /// Constructor
+  TAO_DiffServ_Protocols_Hooks (void);
+
+  /// Destructor
+  virtual ~TAO_DiffServ_Protocols_Hooks (void);
+
+  /// Initialize the protocols hooks instance.
+  void init_hooks (TAO_ORB_Core *orb_core
+                   ACE_ENV_ARG_DECL);
+
+  CORBA::Long get_dscp_codepoint (void);
+
+  void set_dscp_codepoint (CORBA::Long &dscp_codepoint
+                           ACE_ENV_ARG_DECL);
+
+  void np_service_context (TAO_Stub *stub,
+                           TAO_Service_Context &service_context,
+                           CORBA::Boolean restart
+                           ACE_ENV_ARG_DECL);
+
+  void add_rep_np_service_context_hook (
+    TAO_Service_Context &service_context,
+    CORBA::Long &dscp_codepoint
+    ACE_ENV_ARG_DECL);
+
+protected:
+
+  TAO_ORB_Core *orb_core_;
+
+  CORBA::Long dscp_codepoint_;
+};
+
+TAO_END_VERSIONED_NAMESPACE_DECL
+
+ACE_STATIC_SVC_DECLARE_EXPORT (TAO_DiffServPolicy, TAO_DiffServ_Protocols_Hooks)
+ACE_FACTORY_DECLARE (TAO_DiffServPolicy, TAO_DiffServ_Protocols_Hooks)
+
+#include /**/ "ace/post.h"
+#endif /* TAO_DIFFSERV_PROTOCOLS_HOOKS_H */