// -*- C++ -*- #include "tao/RTScheduling/Current.h" #include "tao/RTScheduling/Distributable_Thread.h" #include "tao/RTCORBA/Priority_Mapping_Manager.h" #include "tao/RTCORBA/RT_Current.h" #include "tao/ORB_Core.h" #include "tao/TSS_Resources.h" #include "ace/ACE.h" #include "ace/OS_NS_errno.h" #include "ace/OS_NS_string.h" TAO_BEGIN_VERSIONED_NAMESPACE_DECL std::atomic TAO_RTScheduler_Current::guid_counter; u_long TAO_DTId_Hash::operator () (const IdType &id) const { return ACE::hash_pjw ((const char *) id.get_buffer (), id.length ()); } TAO_RTScheduler_Current::~TAO_RTScheduler_Current () { } void TAO_RTScheduler_Current::init (TAO_ORB_Core* orb) { this->orb_ = orb; // Create the RT_Current. RTCORBA::Current_ptr current; ACE_NEW_THROW_EX (current, TAO_RT_Current (orb), CORBA::NO_MEMORY (CORBA::SystemException::_tao_minor_code ( TAO::VMCID, ENOMEM), CORBA::COMPLETED_NO)); this->rt_current_ = current; } void TAO_RTScheduler_Current::rt_current (RTCORBA::Current_ptr rt_current) { this->rt_current_ = RTCORBA::Current::_duplicate (rt_current); } TAO_ORB_Core* TAO_RTScheduler_Current::orb () { return this->orb_; } DT_Hash_Map* TAO_RTScheduler_Current::dt_hash () { return &this->dt_hash_; } void TAO_RTScheduler_Current::begin_scheduling_segment ( const char * name, CORBA::Policy_ptr sched_param, CORBA::Policy_ptr implicit_sched_param) { TAO_RTScheduler_Current_i *impl = this->implementation (); if (impl == 0) { ACE_NEW_THROW_EX (impl, TAO_RTScheduler_Current_i (this->orb_, &this->dt_hash_), CORBA::NO_MEMORY ( CORBA::SystemException::_tao_minor_code ( TAO::VMCID, ENOMEM), CORBA::COMPLETED_NO)); this->implementation (impl); } impl->begin_scheduling_segment (name, sched_param, implicit_sched_param); } void TAO_RTScheduler_Current::update_scheduling_segment (const char * name, CORBA::Policy_ptr sched_param, CORBA::Policy_ptr implicit_sched_param) { TAO_RTScheduler_Current_i *impl = this->implementation (); if (impl == 0) throw ::CORBA::BAD_INV_ORDER (); impl->update_scheduling_segment (name, sched_param, implicit_sched_param); } void TAO_RTScheduler_Current::end_scheduling_segment (const char * name) { TAO_RTScheduler_Current_i *impl = this->implementation (); if (impl == 0) { TAOLIB_ERROR ((LM_ERROR, "Missing scheduling context OR DT cancelled\n")); throw ::CORBA::BAD_INV_ORDER (); return; } impl->end_scheduling_segment (name); } RTScheduling::DistributableThread_ptr TAO_RTScheduler_Current::lookup(const RTScheduling::Current::IdType & id) { RTScheduling::DistributableThread_var DT; int result = this->dt_hash_.find (id, DT); if (result == 0) return DT._retn (); else return RTScheduling::DistributableThread::_nil (); } // returns a null reference if // the distributable thread is // not known to the local scheduler RTScheduling::DistributableThread_ptr TAO_RTScheduler_Current::spawn (RTScheduling::ThreadAction_ptr start, CORBA::VoidData data, const char* name, CORBA::Policy_ptr sched_param, CORBA::Policy_ptr implicit_sched_param, CORBA::ULong stack_size, RTCORBA::Priority base_priority) { TAO_RTScheduler_Current_i *impl = this->implementation (); if (impl == 0) throw ::CORBA::BAD_INV_ORDER (); return impl->spawn (start, data, name, sched_param, implicit_sched_param, stack_size, base_priority); } RTScheduling::Current::IdType * TAO_RTScheduler_Current::id (void) { TAO_RTScheduler_Current_i *impl = this->implementation (); if (impl == 0) throw ::CORBA::BAD_INV_ORDER (); return impl->id (); } CORBA::Policy_ptr TAO_RTScheduler_Current::scheduling_parameter (void) { TAO_RTScheduler_Current_i *impl = this->implementation (); if (impl == 0) throw ::CORBA::BAD_INV_ORDER (); return impl->scheduling_parameter (); } CORBA::Policy_ptr TAO_RTScheduler_Current::implicit_scheduling_parameter (void) { TAO_RTScheduler_Current_i *impl = this->implementation (); if (impl == 0) throw ::CORBA::BAD_INV_ORDER (); return impl->implicit_scheduling_parameter (); } RTScheduling::Current::NameList * TAO_RTScheduler_Current::current_scheduling_segment_names (void) { TAO_RTScheduler_Current_i *impl = this->implementation (); if (impl == 0) throw ::CORBA::BAD_INV_ORDER (); return impl->current_scheduling_segment_names (); } RTCORBA::Priority TAO_RTScheduler_Current::the_priority (void) { return this->rt_current_->the_priority (); } void TAO_RTScheduler_Current::the_priority (RTCORBA::Priority the_priority) { this->rt_current_->the_priority(the_priority); } TAO_RTScheduler_Current_i* TAO_RTScheduler_Current::implementation (TAO_RTScheduler_Current_i* new_current) { TAO_TSS_Resources *tss = TAO_TSS_Resources::instance (); TAO_RTScheduler_Current_i *old = static_cast (tss->rtscheduler_current_impl_); tss->rtscheduler_current_impl_ = new_current; return old; } TAO_RTScheduler_Current_i* TAO_RTScheduler_Current::implementation (void) { TAO_TSS_Resources *tss = TAO_TSS_Resources::instance (); TAO_RTScheduler_Current_i* impl = static_cast (tss->rtscheduler_current_impl_); return impl; } TAO_ORB_Core* TAO_RTScheduler_Current_i::orb () { return this->orb_; } DT_Hash_Map* TAO_RTScheduler_Current_i::dt_hash () { return this->dt_hash_; } RTScheduling::Scheduler_ptr TAO_RTScheduler_Current_i::scheduler () { return RTScheduling::Scheduler::_duplicate (this->scheduler_.in ()); } TAO_RTScheduler_Current_i::TAO_RTScheduler_Current_i (TAO_ORB_Core* orb, DT_Hash_Map* dt_hash) :orb_ (orb), dt_ (RTScheduling::DistributableThread::_nil ()), previous_current_ (0), dt_hash_ (dt_hash) { CORBA::Object_var scheduler_obj = this->orb_->object_ref_table ().resolve_initial_reference ( "RTScheduler"); this->scheduler_ = RTScheduling::Scheduler::_narrow (scheduler_obj.in ()); } TAO_RTScheduler_Current_i::TAO_RTScheduler_Current_i ( TAO_ORB_Core* orb, DT_Hash_Map* dt_hash, RTScheduling::Current::IdType guid, const char * name, CORBA::Policy_ptr sched_param, CORBA::Policy_ptr implicit_sched_param, RTScheduling::DistributableThread_ptr dt, TAO_RTScheduler_Current_i* prev_current) : orb_ (orb), guid_ (guid), name_ (CORBA::string_dup (name)), sched_param_ (CORBA::Policy::_duplicate (sched_param)), implicit_sched_param_ (CORBA::Policy::_duplicate (implicit_sched_param)), dt_ (RTScheduling::DistributableThread::_duplicate (dt)), previous_current_ (prev_current), dt_hash_ (dt_hash) { CORBA::Object_var scheduler_obj = orb->object_ref_table ().resolve_initial_reference ( "RTScheduler"); this->scheduler_ = RTScheduling::Scheduler::_narrow (scheduler_obj.in ()); } TAO_RTScheduler_Current_i::~TAO_RTScheduler_Current_i () { } void TAO_RTScheduler_Current_i::begin_scheduling_segment( const char * name, CORBA::Policy_ptr sched_param, CORBA::Policy_ptr implicit_sched_param) { // Check if it is a new Scheduling Segmnet if (this->guid_.length () == 0) { //Generate GUID size_t temp = ++TAO_RTScheduler_Current::guid_counter; this->guid_.length (sizeof(size_t)); ACE_OS::memcpy (this->guid_.get_buffer (), &temp, sizeof(size_t)); size_t guid; ACE_OS::memcpy (&guid, this->guid_.get_buffer (), this->guid_.length ()); // Inform the scheduler of the new scheduling segment. this->scheduler_->begin_new_scheduling_segment (this->guid_, name, sched_param, implicit_sched_param); if (CORBA::is_nil (this->dt_.in ())) //Create new DT. this->dt_ = TAO_DistributableThread_Factory::create_DT (); //Add new DT to map int result = this->dt_hash_->bind (this->guid_, this->dt_); // Error in binding to the map - cancel thread. if (result != 0) { this->cancel_thread (); } // Remember parameters for the scheduling segment. this->name_ = CORBA::string_dup (name); this->sched_param_ = CORBA::Policy::_duplicate (sched_param); this->implicit_sched_param_ = CORBA::Policy::_duplicate (implicit_sched_param); } else //Nested segment { // Check current DT state. if (this->dt_->state () == RTScheduling::DistributableThread::CANCELLED) { this->cancel_thread (); } // Inform scheduler of start of nested scheduling segment. this->scheduler_->begin_nested_scheduling_segment (this->guid_, name, sched_param, implicit_sched_param); TAO_TSS_Resources *tss = TAO_TSS_Resources::instance (); TAO_RTScheduler_Current_i* new_current = 0; ACE_NEW_THROW_EX (new_current, TAO_RTScheduler_Current_i (this->orb_, this->dt_hash_, this->guid_, name, sched_param, implicit_sched_param, this->dt_.in (), this), CORBA::NO_MEMORY ( CORBA::SystemException::_tao_minor_code ( TAO::VMCID, ENOMEM), CORBA::COMPLETED_NO)); tss->rtscheduler_current_impl_ = new_current; } } void TAO_RTScheduler_Current_i::update_scheduling_segment (const char * name, CORBA::Policy_ptr sched_param, CORBA::Policy_ptr implicit_sched_param) { // Check if DT has been cancelled if (this->dt_->state () == RTScheduling::DistributableThread::CANCELLED) { this->cancel_thread (); } // Let scheduler know of the updates. this->scheduler_->update_scheduling_segment (this->guid_, name, sched_param, implicit_sched_param); // Remember the new values. this->name_ = CORBA::string_dup (name); this->sched_param_ = CORBA::Policy::_duplicate (sched_param); this->implicit_sched_param_ = CORBA::Policy::_duplicate (implicit_sched_param); } void TAO_RTScheduler_Current_i::end_scheduling_segment (const char * name) { // Check if DT has been cancelled if (this->dt_->state () == RTScheduling::DistributableThread::CANCELLED) { this->cancel_thread (); } if (this->previous_current_ == 0) { // Let the scheduler know that the DT is // terminating. this->scheduler_->end_scheduling_segment(this->guid_, name); // Cleanup DT. this->cleanup_DT (); // Cleanup current. this->cleanup_current (); // A Nested segment. } else { // Inform scheduler of end of nested // scheduling segment. this->scheduler_->end_nested_scheduling_segment (this->guid_, name, this->previous_current_->sched_param_.in ()); // Cleanup current. this->cleanup_current (); } } // returns a null reference if // the distributable thread is // not known to the local scheduler RTScheduling::DistributableThread_ptr TAO_RTScheduler_Current_i::spawn (RTScheduling::ThreadAction_ptr start, CORBA::VoidData data, const char* name, CORBA::Policy_ptr sched_param, CORBA::Policy_ptr implicit_sched_param, CORBA::ULong stack_size, RTCORBA::Priority base_priority) { // Check if DT has been cancelled. if (this->dt_->state () == RTScheduling::DistributableThread::CANCELLED) { this->cancel_thread (); } // Create new task for new DT. DTTask *dttask = 0; // If no scheduling parameter is specified then use the current // implicit scheduling parameter as the scheduling parameter if (sched_param == 0) sched_param = this->implicit_sched_param_.in (); RTScheduling::DistributableThread_var dt = TAO_DistributableThread_Factory::create_DT (); TAO_RTScheduler_Current_i *new_current = 0; ACE_NEW_RETURN (new_current, TAO_RTScheduler_Current_i (this->orb_, this->dt_hash_), 0); new_current->DT (dt.in ()); ACE_NEW_RETURN (dttask, DTTask (//thread_manager_, this->orb_, this->dt_hash_, new_current, start, data, name, sched_param, implicit_sched_param), 0); if (dttask->activate_task (base_priority, stack_size) == -1) { TAOLIB_ERROR((LM_ERROR, "Unable to activate DistributableThread\n")); delete dttask; return RTScheduling::DistributableThread::_nil (); } return dt._retn (); } int DTTask::activate_task (RTCORBA::Priority base_priority, CORBA::ULong stack_size) { // Activate thread. long default_flags = THR_NEW_LWP | THR_JOINABLE; long flags = default_flags | this->orb_->orb_params ()->scope_policy () | this->orb_->orb_params ()->sched_policy (); CORBA::Object_var object = this->orb_->object_ref_table ().resolve_initial_reference ( TAO_OBJID_PRIORITYMAPPINGMANAGER); RTCORBA::PriorityMappingManager_var mapping_manager = RTCORBA::PriorityMappingManager::_narrow (object.in ()); RTCORBA::PriorityMapping *pm = mapping_manager->mapping (); RTCORBA::NativePriority native_priority; pm->to_native (base_priority, native_priority); size_t stack [1]; stack [0] = stack_size; if (this->activate (flags, 1, 0,//force_active native_priority,//priority -1,//grp_id 0,//ACE_Task_Base 0,//thread_handles 0,//stack stack//stack_size ) == -1) { if (ACE_OS::last_error () == EPERM) TAOLIB_ERROR_RETURN ((LM_ERROR, ACE_TEXT ("Insufficient privilege to run this test.\n")), -1); } return 0; } DTTask::DTTask (TAO_ORB_Core *orb, DT_Hash_Map *, TAO_RTScheduler_Current_i* new_current, RTScheduling::ThreadAction_ptr start, CORBA::VoidData data, const char *name, CORBA::Policy_ptr sched_param, CORBA::Policy_ptr implicit_sched_param) :orb_ (orb), // dt_hash_ (dt_hash), current_ (new_current), start_ (RTScheduling::ThreadAction::_duplicate (start)), data_ (data), name_ (CORBA::string_dup (name)), sched_param_ (CORBA::Policy::_duplicate (sched_param)), implicit_sched_param_ (CORBA::Policy::_duplicate (implicit_sched_param)) { } DTTask::~DTTask () { delete this->current_; } int DTTask::svc () { try { TAO_TSS_Resources *tss = TAO_TSS_Resources::instance (); tss->rtscheduler_current_impl_ = this->current_; this->current_->begin_scheduling_segment (this->name_.in (), this->sched_param_.in (), this->implicit_sched_param_.in ()); // Invoke entry point into new DT. this->start_->_cxx_do (this->data_); this->current_->end_scheduling_segment (this->name_.in ()); } catch (const ::CORBA::Exception& ex) { ex._tao_print_exception ("Caught exception:"); return -1; } return 0; } RTScheduling::Current::IdType * TAO_RTScheduler_Current_i::id () { RTScheduling::Current::IdType_var guid = this->guid_; return guid._retn (); } CORBA::Policy_ptr TAO_RTScheduler_Current_i::scheduling_parameter () { return CORBA::Policy::_duplicate (this->sched_param_.in ()); } CORBA::Policy_ptr TAO_RTScheduler_Current_i::implicit_scheduling_parameter () { return CORBA::Policy::_duplicate (this->implicit_sched_param_.in ()); } RTScheduling::Current::NameList * TAO_RTScheduler_Current_i::current_scheduling_segment_names (void) { RTScheduling::Current::NameList* name_list; ACE_NEW_RETURN (name_list, RTScheduling::Current::NameList, 0); TAO_RTScheduler_Current_i* current = this; for (int index = 0; current != 0; index++) { name_list->length (index+1); (*name_list) [index] = current->name (); current = current->previous_current_; } return name_list; } const char* TAO_RTScheduler_Current_i::name () { return this->name_.in (); } #if defined (THREAD_CANCELLED) #undef THREAD_CANCELLED #endif /* THREAD_CANCELLED */ void TAO_RTScheduler_Current_i::cancel_thread () { size_t guid; ACE_OS::memcpy (&guid, this->guid_.get_buffer (), this->guid_.length ()); TAOLIB_DEBUG ((LM_DEBUG, "Distributable Thread - %d is cancelled\n", guid)); // Let the scheduler know that the thread has // been cancelled. this->scheduler_->cancel (this->guid_); this->cleanup_DT (); // Remove all related nested currents. this->delete_all_currents (); // Throw exception. throw ::CORBA::THREAD_CANCELLED (); } void TAO_RTScheduler_Current_i::cleanup_DT () { // Remove DT from map. this->dt_hash_->unbind (this->guid_); } void TAO_RTScheduler_Current_i::cleanup_current () { TAO_TSS_Resources *tss = TAO_TSS_Resources::instance (); tss->rtscheduler_current_impl_ = this->previous_current_; // Delete this current. delete this; } void TAO_RTScheduler_Current_i::delete_all_currents () { TAO_RTScheduler_Current_i* current = this; while (current != 0) { TAO_RTScheduler_Current_i* prev_current = current->previous_current_; current->cleanup_current (); current = prev_current; } TAO_TSS_Resources *tss = TAO_TSS_Resources::instance (); tss->rtscheduler_current_impl_ = tss->rtscheduler_previous_current_impl_; } void TAO_RTScheduler_Current_i::id (RTScheduling::Current::IdType guid) { this->guid_ = guid; } void TAO_RTScheduler_Current_i::name (const char * name) { this->name_ = CORBA::string_dup (name); } RTScheduling::DistributableThread_ptr TAO_RTScheduler_Current_i::DT () { return this->dt_._retn (); } void TAO_RTScheduler_Current_i::DT (RTScheduling::DistributableThread_ptr dt) { this->dt_ = RTScheduling::DistributableThread::_duplicate (dt); } void TAO_RTScheduler_Current_i::scheduling_parameter (CORBA::Policy_ptr sched_param) { this->sched_param_ = CORBA::Policy::_duplicate (sched_param); } void TAO_RTScheduler_Current_i::implicit_scheduling_parameter (CORBA::Policy_ptr implicit_sched_param) { this->implicit_sched_param_ = CORBA::Policy::_duplicate (implicit_sched_param); } // ************************************************************* // ************************************************************* // Operations for class TAO_RTScheduler_Current_var // ************************************************************* TAO_RTScheduler_Current_var::TAO_RTScheduler_Current_var () // default constructor : ptr_ (TAO_RTScheduler_Current::_nil ()) {} ::TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current_var::ptr () const { return this->ptr_; } TAO_RTScheduler_Current_var::TAO_RTScheduler_Current_var (const ::TAO_RTScheduler_Current_var &p) : TAO_Base_var (), ptr_ (TAO_RTScheduler_Current::_duplicate (p.ptr ())) {} TAO_RTScheduler_Current_var::~TAO_RTScheduler_Current_var () // destructor { ::CORBA::release (this->ptr_); } TAO_RTScheduler_Current_var & TAO_RTScheduler_Current_var::operator= (TAO_RTScheduler_Current_ptr p) { ::CORBA::release (this->ptr_); this->ptr_ = p; return *this; } TAO_RTScheduler_Current_var & TAO_RTScheduler_Current_var::operator= (const ::TAO_RTScheduler_Current_var &p) { if (this != &p) { ::CORBA::release (this->ptr_); this->ptr_ = ::TAO_RTScheduler_Current::_duplicate (p.ptr ()); } return *this; } TAO_RTScheduler_Current_var::operator const ::TAO_RTScheduler_Current_ptr &() const { return this->ptr_; } TAO_RTScheduler_Current_var::operator ::TAO_RTScheduler_Current_ptr &() { return this->ptr_; } TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current_var::operator-> () const { return this->ptr_; } TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current_var::in () const { return this->ptr_; } TAO_RTScheduler_Current_ptr & TAO_RTScheduler_Current_var::inout () { return this->ptr_; } TAO_RTScheduler_Current_ptr & TAO_RTScheduler_Current_var::out () { ::CORBA::release (this->ptr_); this->ptr_ = ::TAO_RTScheduler_Current::_nil (); return this->ptr_; } TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current_var::_retn () { // yield ownership of managed obj reference ::TAO_RTScheduler_Current_ptr val = this->ptr_; this->ptr_ = ::TAO_RTScheduler_Current::_nil (); return val; } TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current_var::duplicate (TAO_RTScheduler_Current_ptr p) { return ::TAO_RTScheduler_Current::_duplicate (p); } void TAO_RTScheduler_Current_var::release (TAO_RTScheduler_Current_ptr p) { ::CORBA::release (p); } TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current_var::nil () { return ::TAO_RTScheduler_Current::_nil (); } TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current_var::narrow (CORBA::Object *p) { return ::TAO_RTScheduler_Current::_narrow (p); } CORBA::Object * TAO_RTScheduler_Current_var::upcast (void *src) { TAO_RTScheduler_Current **tmp = static_cast (src); return *tmp; } TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current::_narrow ( CORBA::Object_ptr obj) { return TAO_RTScheduler_Current::_duplicate ( dynamic_cast (obj)); } TAO_RTScheduler_Current_ptr TAO_RTScheduler_Current::_duplicate (TAO_RTScheduler_Current_ptr obj) { if (!CORBA::is_nil (obj)) obj->_add_ref (); return obj; } const char* TAO_RTScheduler_Current::_interface_repository_id () const { return "IDL:TAO_RTScheduler_Current:1.0"; } TAO_END_VERSIONED_NAMESPACE_DECL