// -*- C++ -*- // $Id$ #include "tao/DiffServPolicy/DiffServPolicy_ORBInitializer.h" #include "tao/DiffServPolicy/DiffServPolicy.h" #include "tao/DiffServPolicy/DiffServPolicy_Factory.h" #include "tao/DiffServPolicy/DiffServ_Service_Context_Handler.h" #include "tao/DiffServPolicy/DiffServ_Protocols_Hooks.h" #include "tao/ORB_Core.h" #include "tao/PI/ORBInitInfo.h" TAO_BEGIN_VERSIONED_NAMESPACE_DECL void TAO_DiffServPolicy_ORBInitializer::pre_init ( PortableInterceptor::ORBInitInfo_ptr info) { // Narrow to a TAO_ORBInitInfo object to get access to the // orb_core() TAO extension. TAO_ORBInitInfo_var tao_info = TAO_ORBInitInfo::_narrow (info); if (CORBA::is_nil (tao_info.in ())) { if (TAO_debug_level > 0) TAOLIB_ERROR ((LM_ERROR, "(%P|%t) TAO_RT_ORBInitializer::pre_init:\n" "(%P|%t) Unable to narrow " "\"PortableInterceptor::ORBInitInfo_ptr\" to\n" "(%P|%t) \"TAO_ORBInitInfo *.\"\n")); throw ::CORBA::INTERNAL (); } // Bind the service context handler for Diffserv TAO_DiffServ_Service_Context_Handler* h = 0; ACE_NEW (h, TAO_DiffServ_Service_Context_Handler()); tao_info->orb_core ()->service_context_registry ().bind (IOP::REP_NWPRIORITY, h); } void TAO_DiffServPolicy_ORBInitializer::post_init ( PortableInterceptor::ORBInitInfo_ptr info) { this->register_policy_factories (info); } void TAO_DiffServPolicy_ORBInitializer::register_policy_factories ( PortableInterceptor::ORBInitInfo_ptr info) { if (CORBA::is_nil (this->policy_factory_.in ())) { PortableInterceptor::PolicyFactory_ptr policy_factory; ACE_NEW_THROW_EX (policy_factory, TAO_DiffServ_PolicyFactory, CORBA::NO_MEMORY ( CORBA::SystemException::_tao_minor_code ( TAO::VMCID, ENOMEM), CORBA::COMPLETED_NO)); this->policy_factory_ = policy_factory; } // Bind the same policy factory to all DiffServ related policy // types since a single policy factory is used to create each of // the different types of DiffServ policies. static CORBA::PolicyType const type[] = { TAO::CLIENT_NETWORK_PRIORITY_TYPE, TAO::NETWORK_PRIORITY_TYPE }; const CORBA::PolicyType *end = type + sizeof (type) / sizeof (type[0]); for (CORBA::PolicyType const * i = type; i != end; ++i) { try { info->register_policy_factory (*i, this->policy_factory_.in ()); } catch (const ::CORBA::BAD_INV_ORDER& ex) { if (ex.minor () == (CORBA::OMGVMCID | 16)) { // The factory is already there, it happens because the // magic initializer in PortableServer.cpp registers // with the ORB multiple times. This is an indication // that we should do no more work in this // ORBInitializer. return; } throw; } catch ( ::CORBA::Exception&) { // Rethrow any other exceptions... throw; } } } TAO_END_VERSIONED_NAMESPACE_DECL