Chris@16: //////////////////////////////////////////////////////////////////////////////
Chris@16: //
Chris@16: // (C) Copyright Ion Gaztanaga 2005-2012. Distributed under the Boost
Chris@16: // Software License, Version 1.0. (See accompanying file
Chris@16: // LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
Chris@16: //
Chris@16: // See http://www.boost.org/libs/interprocess for documentation.
Chris@16: //
Chris@16: //////////////////////////////////////////////////////////////////////////////
Chris@16: //
Chris@16: // Parts of the pthread code come from Boost Threads code:
Chris@16: //
Chris@16: //////////////////////////////////////////////////////////////////////////////
Chris@16: //
Chris@16: // Copyright (C) 2001-2003
Chris@16: // William E. Kempf
Chris@16: //
Chris@16: // Permission to use, copy, modify, distribute and sell this software
Chris@16: // and its documentation for any purpose is hereby granted without fee,
Chris@16: // provided that the above copyright notice appear in all copies and
Chris@16: // that both that copyright notice and this permission notice appear
Chris@16: // in supporting documentation.  William E. Kempf makes no representations
Chris@16: // about the suitability of this software for any purpose.
Chris@16: // It is provided "as is" without express or implied warranty.
Chris@16: //////////////////////////////////////////////////////////////////////////////
Chris@16: 
Chris@16: #ifndef BOOST_INTERPROCESS_DETAIL_SPIN_RECURSIVE_MUTEX_HPP
Chris@16: #define BOOST_INTERPROCESS_DETAIL_SPIN_RECURSIVE_MUTEX_HPP
Chris@16: 
Chris@101: #ifndef BOOST_CONFIG_HPP
Chris@101: #  include <boost/config.hpp>
Chris@101: #endif
Chris@101: #
Chris@101: #if defined(BOOST_HAS_PRAGMA_ONCE)
Chris@16: #  pragma once
Chris@16: #endif
Chris@16: 
Chris@16: #include <boost/interprocess/detail/config_begin.hpp>
Chris@16: #include <boost/interprocess/detail/workaround.hpp>
Chris@16: 
Chris@16: #include <boost/interprocess/detail/posix_time_types_wrk.hpp>
Chris@16: #include <boost/interprocess/detail/os_thread_functions.hpp>
Chris@16: #include <boost/interprocess/exceptions.hpp>
Chris@16: #include <boost/interprocess/detail/atomic.hpp>
Chris@16: #include <boost/cstdint.hpp>
Chris@16: #include <boost/interprocess/detail/os_thread_functions.hpp>
Chris@16: #include <boost/interprocess/sync/spin/mutex.hpp>
Chris@16: #include <boost/assert.hpp>
Chris@16: 
Chris@16: namespace boost {
Chris@16: namespace interprocess {
Chris@16: namespace ipcdetail {
Chris@16: 
Chris@16: class spin_recursive_mutex
Chris@16: {
Chris@16:    spin_recursive_mutex(const spin_recursive_mutex &);
Chris@16:    spin_recursive_mutex &operator=(const spin_recursive_mutex &);
Chris@16:    public:
Chris@16: 
Chris@16:    spin_recursive_mutex();
Chris@16:    ~spin_recursive_mutex();
Chris@16: 
Chris@16:    void lock();
Chris@16:    bool try_lock();
Chris@16:    bool timed_lock(const boost::posix_time::ptime &abs_time);
Chris@16:    void unlock();
Chris@16:    void take_ownership();
Chris@16:    private:
Chris@16:    spin_mutex     m_mutex;
Chris@16:    unsigned int   m_nLockCount;
Chris@16:    volatile ipcdetail::OS_systemwide_thread_id_t   m_nOwner;
Chris@16:    volatile boost::uint32_t m_s;
Chris@16: };
Chris@16: 
Chris@16: inline spin_recursive_mutex::spin_recursive_mutex()
Chris@16:    : m_nLockCount(0), m_nOwner(ipcdetail::get_invalid_systemwide_thread_id()){}
Chris@16: 
Chris@16: inline spin_recursive_mutex::~spin_recursive_mutex(){}
Chris@16: 
Chris@16: inline void spin_recursive_mutex::lock()
Chris@16: {
Chris@16:    typedef ipcdetail::OS_systemwide_thread_id_t handle_t;
Chris@16:    const handle_t thr_id(ipcdetail::get_current_systemwide_thread_id());
Chris@16:    handle_t old_id;
Chris@16:    ipcdetail::systemwide_thread_id_copy(m_nOwner, old_id);
Chris@16:    if(ipcdetail::equal_systemwide_thread_id(thr_id , old_id)){
Chris@16:       if((unsigned int)(m_nLockCount+1) == 0){
Chris@16:          //Overflow, throw an exception
Chris@16:          throw interprocess_exception("boost::interprocess::spin_recursive_mutex recursive lock overflow");
Chris@16:       }
Chris@16:       ++m_nLockCount;
Chris@16:    }
Chris@16:    else{
Chris@16:       m_mutex.lock();
Chris@16:       ipcdetail::systemwide_thread_id_copy(thr_id, m_nOwner);
Chris@16:       m_nLockCount = 1;
Chris@16:    }
Chris@16: }
Chris@16: 
Chris@16: inline bool spin_recursive_mutex::try_lock()
Chris@16: {
Chris@16:    typedef ipcdetail::OS_systemwide_thread_id_t handle_t;
Chris@16:    handle_t thr_id(ipcdetail::get_current_systemwide_thread_id());
Chris@16:    handle_t old_id;
Chris@16:    ipcdetail::systemwide_thread_id_copy(m_nOwner, old_id);
Chris@16:    if(ipcdetail::equal_systemwide_thread_id(thr_id , old_id)) {  // we own it
Chris@16:       if((unsigned int)(m_nLockCount+1) == 0){
Chris@16:          //Overflow, throw an exception
Chris@16:          throw interprocess_exception("boost::interprocess::spin_recursive_mutex recursive lock overflow");
Chris@16:       }
Chris@16:       ++m_nLockCount;
Chris@16:       return true;
Chris@16:    }
Chris@16:    if(m_mutex.try_lock()){
Chris@16:       ipcdetail::systemwide_thread_id_copy(thr_id, m_nOwner);
Chris@16:       m_nLockCount = 1;
Chris@16:       return true;
Chris@16:    }
Chris@16:    return false;
Chris@16: }
Chris@16: 
Chris@16: inline bool spin_recursive_mutex::timed_lock(const boost::posix_time::ptime &abs_time)
Chris@16: {
Chris@16:    typedef ipcdetail::OS_systemwide_thread_id_t handle_t;
Chris@16:    const handle_t thr_id(ipcdetail::get_current_systemwide_thread_id());
Chris@16:    handle_t old_id;
Chris@16:    ipcdetail::systemwide_thread_id_copy(m_nOwner, old_id);
Chris@16:    if(ipcdetail::equal_systemwide_thread_id(thr_id , old_id)) {  // we own it
Chris@16:       if((unsigned int)(m_nLockCount+1) == 0){
Chris@16:          //Overflow, throw an exception
Chris@16:          throw interprocess_exception("boost::interprocess::spin_recursive_mutex recursive lock overflow");
Chris@16:       }
Chris@16:       ++m_nLockCount;
Chris@16:       return true;
Chris@16:    }
Chris@101:    //m_mutex supports abs_time so no need to check it
Chris@16:    if(m_mutex.timed_lock(abs_time)){
Chris@16:       ipcdetail::systemwide_thread_id_copy(thr_id, m_nOwner);
Chris@16:       m_nLockCount = 1;
Chris@16:       return true;
Chris@16:    }
Chris@16:    return false;
Chris@16: }
Chris@16: 
Chris@16: inline void spin_recursive_mutex::unlock()
Chris@16: {
Chris@16:    typedef ipcdetail::OS_systemwide_thread_id_t handle_t;
Chris@16:    handle_t old_id;
Chris@16:    ipcdetail::systemwide_thread_id_copy(m_nOwner, old_id);
Chris@16:    const handle_t thr_id(ipcdetail::get_current_systemwide_thread_id());
Chris@16:    (void)old_id;
Chris@16:    (void)thr_id;
Chris@16:    BOOST_ASSERT(ipcdetail::equal_systemwide_thread_id(thr_id, old_id));
Chris@16:    --m_nLockCount;
Chris@16:    if(!m_nLockCount){
Chris@16:       const handle_t new_id(ipcdetail::get_invalid_systemwide_thread_id());
Chris@16:       ipcdetail::systemwide_thread_id_copy(new_id, m_nOwner);
Chris@16:       m_mutex.unlock();
Chris@16:    }
Chris@16: }
Chris@16: 
Chris@16: inline void spin_recursive_mutex::take_ownership()
Chris@16: {
Chris@16:    typedef ipcdetail::OS_systemwide_thread_id_t handle_t;
Chris@16:    this->m_nLockCount = 1;
Chris@16:    const handle_t thr_id(ipcdetail::get_current_systemwide_thread_id());
Chris@16:    ipcdetail::systemwide_thread_id_copy(thr_id, m_nOwner);
Chris@16: }
Chris@16: 
Chris@16: }  //namespace ipcdetail {
Chris@16: }  //namespace interprocess {
Chris@16: }  //namespace boost {
Chris@16: 
Chris@16: #include <boost/interprocess/detail/config_end.hpp>
Chris@16: 
Chris@16: #endif   //BOOST_INTERPROCESS_DETAIL_SPIN_RECURSIVE_MUTEX_HPP