TAO/tao/DiffServPolicy/DiffServ_Protocols_Hooks.cpp

   1 // -*- C++ -*-
   2 #include "tao/DiffServPolicy/DiffServ_Protocols_Hooks.h"
   3 #include "tao/Stub.h"
   4 #include "tao/Profile.h"
   5 #include "tao/ORB_Core.h"
   6 #include "tao/Service_Context.h"
   7 #include "tao/CDR.h"
   8 #include "tao/SystemException.h"
   9 #include "tao/DiffServPolicy/Client_Network_Priority_Policy.h"
  10 #include "ace/OS_NS_string.h"
  11
  12 TAO_BEGIN_VERSIONED_NAMESPACE_DECL
  13
  14 TAO_DS_Network_Priority_Protocols_Hooks::
  15 TAO_DS_Network_Priority_Protocols_Hooks ()
  16   : orb_core_ (0)
  17 {
  18 }
  19
  20
  21 TAO_DS_Network_Priority_Protocols_Hooks::
  22 ~TAO_DS_Network_Priority_Protocols_Hooks ()
  23 {
  24 }
  25
  26 void
  27 TAO_DS_Network_Priority_Protocols_Hooks::init_hooks (TAO_ORB_Core *orb_core)
  28 {
  29   this->orb_core_ = orb_core;
  30 }
  31
  32 CORBA::Long
  33 TAO_DS_Network_Priority_Protocols_Hooks::get_dscp_codepoint (
  34   TAO_Service_Context &sc)
  35 {
  36   CORBA::Long dscp_codepoint = 0;
  37   const IOP::ServiceContext *context = 0;
  38
  39   if (sc.get_context (IOP::REP_NWPRIORITY, &context) == 1)
  40     {
  41       TAO_InputCDR cdr (reinterpret_cast
  42                         <const char*>
  43                         (context->context_data.get_buffer ()),
  44                         context->context_data.length ());
  45
  46       CORBA::Boolean byte_order;
  47       if ((cdr >> ACE_InputCDR::to_boolean (byte_order)) == 0)
  48         {
  49           throw CORBA::MARSHAL ();
  50         }
  51       cdr.reset_byte_order (static_cast<int> (byte_order));
  52
  53       if ((cdr >> dscp_codepoint) == 0)
  54         {
  55           throw CORBA::MARSHAL ();
  56         }
  57     }
  58
  59   return dscp_codepoint;
  60 }
  61
  62 CORBA::Long
  63 TAO_DS_Network_Priority_Protocols_Hooks::get_dscp_codepoint (
  64   TAO_Stub *stub, CORBA::Object *object)
  65 {
  66   CORBA::Long dscp = 0;
  67   TAO::DiffservCodepoint diffserv_codepoint;
  68   CORBA::Policy_var server_nw_priority_policy;
  69
  70   CORBA::Policy_var client_nw_priority_policy =
  71     stub->get_cached_policy (
  72         TAO_CACHED_POLICY_CLIENT_NETWORK_PRIORITY);
  73
  74   if (CORBA::is_nil (client_nw_priority_policy.in ()))
  75     {
  76       TAO_Stub *server_stub = object->_stubobj ();
  77       TAO_MProfile server_profiles = server_stub->base_profiles ();
  78       CORBA::PolicyList_var policy_list =
  79         server_profiles.policy_list ();
  80
  81       CORBA::ULong length = policy_list->length ();
  82
  83       for (CORBA::ULong i = 0; i < length; ++i)
  84         {
  85           if (policy_list[i]->policy_type () == TAO::NETWORK_PRIORITY_TYPE)
  86             {
  87               server_nw_priority_policy =
  88                 CORBA::Policy::_duplicate (policy_list[i]);
  89             }
  90         }
  91
  92       if (CORBA::is_nil (server_nw_priority_policy.in ()))
  93         {
  94           return dscp;
  95         }
  96       else
  97         {
  98           TAO::NetworkPriorityPolicy_var npp =
  99             TAO::NetworkPriorityPolicy::_narrow (
 100               server_nw_priority_policy.in ());
 101
 102           if (!CORBA::is_nil (npp.in ()))
 103             {
 104               TAO::NetworkPriorityModel network_priority_model =
 105                 npp->network_priority_model ();
 106
 107               if (network_priority_model ==
 108                   TAO::SERVER_DECLARED_NETWORK_PRIORITY)
 109                 {
 110                   diffserv_codepoint =
 111                     npp->request_diffserv_codepoint ();
 112                   dscp = diffserv_codepoint;
 113                   return dscp;
 114                 }
 115             }
 116         }
 117     }
 118   else
 119     {
 120       TAO::NetworkPriorityPolicy_var client_nw_priority =
 121         TAO::NetworkPriorityPolicy::_narrow (client_nw_priority_policy.in ());
 122
 123       if (!CORBA::is_nil (client_nw_priority.in ()))
 124         {
 125           diffserv_codepoint =
 126             client_nw_priority->request_diffserv_codepoint ();
 127           dscp = diffserv_codepoint;
 128           return dscp;
 129         }
 130     }
 131
 132   return dscp;
 133 }
 134
 135
 136 ACE_STATIC_SVC_DEFINE (TAO_DS_Network_Priority_Protocols_Hooks,
 137                        ACE_TEXT ("DS_Network_Priority_Protocols_Hooks"),
 138                        ACE_SVC_OBJ_T,
 139                        &ACE_SVC_NAME (TAO_DS_Network_Priority_Protocols_Hooks),
 140                        ACE_Service_Type::DELETE_THIS
 141                        | ACE_Service_Type::DELETE_OBJ,
 142                        0)
 143 ACE_FACTORY_DEFINE (TAO_DiffServPolicy,
 144                     TAO_DS_Network_Priority_Protocols_Hooks)
 145 TAO_END_VERSIONED_NAMESPACE_DECL