TAO/tao/DiffServPolicy/DiffServ_Service_Context_Handler.cpp

   1 #include "tao/DiffServPolicy/DiffServ_Service_Context_Handler.h"
   2
   3 #include "tao/DiffServPolicy/Client_Network_Priority_Policy.h"
   4 #include "tao/CDR.h"
   5 #include "tao/TAO_Server_Request.h"
   6 #include "tao/Transport.h"
   7 #include "tao/ORB_Core.h"
   8 #include "tao/GIOP_Message_Base.h"
   9 #include "tao/operation_details.h"
  10 #include "tao/Transport_Mux_Strategy.h"
  11 #include "tao/Stub.h"
  12
  13 TAO_BEGIN_VERSIONED_NAMESPACE_DECL
  14
  15 int
  16 TAO_DiffServ_Service_Context_Handler::process_service_context (
  17   TAO_Transport&,
  18   const IOP::ServiceContext&,
  19   TAO_ServerRequest *)
  20 {
  21   return 0;
  22 }
  23
  24 int
  25 TAO_DiffServ_Service_Context_Handler::generate_service_context (
  26   TAO_Stub *stub,
  27   TAO_Transport&,
  28   TAO_Operation_Details &opdetails,
  29   TAO_Target_Specification &,
  30   TAO_OutputCDR &)
  31 {
  32   if (stub)
  33     {
  34       CORBA::Policy_var cnpp =
  35         stub->get_cached_policy (TAO_CACHED_POLICY_CLIENT_NETWORK_PRIORITY);
  36
  37       TAO::NetworkPriorityPolicy_var cnp =
  38          TAO::NetworkPriorityPolicy::_narrow (cnpp.in ());
  39
  40       if (!CORBA::is_nil (cnp.in ()))
  41         {
  42           TAO::DiffservCodepoint const reply_diffserv_codepoint =
  43             cnp->reply_diffserv_codepoint ();
  44
  45           CORBA::Long const rep_dscp_codepoint = reply_diffserv_codepoint;
  46
  47           TAO_OutputCDR cdr;
  48           if (!(cdr << ACE_OutputCDR::from_boolean (TAO_ENCAP_BYTE_ORDER))
  49             || !(cdr << rep_dscp_codepoint))
  50             {
  51               throw CORBA::MARSHAL ();
  52             }
  53
  54           opdetails.request_service_context ().set_context (IOP::REP_NWPRIORITY, cdr);
  55         }
  56     }
  57
  58   return 0;
  59 }
  60
  61 TAO_END_VERSIONED_NAMESPACE_DECL