(C) Zeriph Enterprises, LLC
omni/sync/lock.hpp
001:
002:
003:
004:
005:
006:
007:
008:
009:
010:
011:
012:
013:
014:
015:
016:
017:
018:
019:
020:
021:
022:
023:
024:
025:
026:
027:
028:
029:
030:
031:
032:
033:
034:
035:
036:
037:
038:
039:
040:
041:
042:
043:
044:
045:
046:
047:
048:
049:
050:
051:
052:
053:
054:
055:
056:
057:
058:
059:
060:
061:
062:
063:
064:
065:
066:
067:
068:
069:
070:
071:
072:
073:
074:
075:
076:
077:
078:
079:
080:
081:
082:
083:
084:
085:
086:
087:
088:
089:
090:
091:
092:
093:
094:
095:
096:
097:
098:
099:
100:
101:
102:
103:
104:
105:
106:
107:
108:
109:
110:
111:
112:
113:
114:


















































































































/*
* Copyright (c), Zeriph Enterprises
* All rights reserved.
*
* Contributor(s):
* Zechariah Perez, omni (at) zeriph (dot) com
*
* THIS SOFTWARE IS PROVIDED BY ZERIPH AND CONTRIBUTORS "AS IS" AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL ZERIPH AND CONTRIBUTORS BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
 #if !defined(OMNI_SYNC_LOCK_HPP)
#define OMNI_SYNC_LOCK_HPP 1
#include <omni/types/lock_t.hpp>
#include <omni/chrono/tick.hpp>

namespace omni {
    namespace sync {
        inline void mutex_init(omni::sync::mutex_t& mtx)
        {
            #if defined(OMNI_OS_WIN)
                ::InitializeCriticalSection(&mtx);
            #else
                int lerr = 0;
                pthread_mutexattr_t mtxatr;
                if ((lerr = ::pthread_mutexattr_init(&mtxatr)) == 0) {
                    if ((lerr = ::pthread_mutexattr_settype(&mtxatr,
                        #if defined(__linux__) && !defined(PTHREAD_MUTEX_RECURSIVE)
                            PTHREAD_MUTEX_RECURSIVE_NP
                        #else
                            PTHREAD_MUTEX_RECURSIVE
                        #endif
                    )) == 0)
                    {
                        if ((lerr = ::pthread_mutex_init(&mtx, &mtxatr)) == 0) {
                            lerr = ::pthread_mutexattr_destroy(&mtxatr);
                        }
                    }
                }
                if (lerr != 0) {
                    OMNI_ERRV_FW("system create mutex failed", lerr, omni::exceptions::mutex_system_exception("system create mutex failed", lerr))
                }
            #endif
        }
        
        inline void mutex_destroy(omni::sync::mutex_t& mtx)
        {
            #if defined(OMNI_OS_WIN)
                ::DeleteCriticalSection(&mtx);
            #else
                int lerr = ::pthread_mutex_destroy(&mtx);
                if (lerr != 0) {
                    OMNI_ERRV_FW("mutex destroy failed", lerr, omni::exceptions::mutex_system_exception("mutex destroy failed", lerr))
                }
            #endif
        }
        
        inline bool mutex_trylock(omni::sync::mutex_t& mtx)
        {
            #if defined(OMNI_OS_WIN)
                return (::TryEnterCriticalSection(&mtx) == TRUE);
            #else
                return (::pthread_mutex_trylock(&mtx) == 0);
            #endif
        }
        
        inline void mutex_lock(omni::sync::mutex_t& mtx)
        {
            #if defined(OMNI_OS_WIN)
                ::EnterCriticalSection(&mtx);
            #else
                int lerr = ::pthread_mutex_lock(&mtx);
                if (lerr != 0) {
                    OMNI_ERRV_FW("mutex lock failed", lerr, omni::exceptions::mutex_system_exception("mutex lock failed", lerr))
                }
            #endif
        }
        
        inline bool mutex_lock(omni::sync::mutex_t& mtx, uint32_t timeout_ms)
        {
            if (timeout_ms == 0) { return omni::sync::mutex_trylock(mtx); }
            #if !defined(OMNI_CHRONO_AUTO_INIT_TICK)
                omni::chrono::monotonic::initialize();
            #endif
            omni::chrono::tick_t start = omni::chrono::monotonic_tick();
            while (omni::chrono::elapsed_ms(start) <= timeout_ms) {
                if (omni::sync::mutex_trylock(mtx)) { return true; }
            }
            return false;
        }
        
        inline void mutex_unlock(omni::sync::mutex_t& mtx)
        {
            #if defined(OMNI_OS_WIN)
                ::LeaveCriticalSection(&mtx);
            #else
                int lerr = ::pthread_mutex_unlock(&mtx);
                if (lerr != 0) {
                    OMNI_ERRV_FW("mutex unlock failed", lerr, omni::exceptions::mutex_system_exception("mutex unlock failed", lerr))
                }
            #endif
        }
    }
}

#endif // OMNI_SYNC_LOCK_HPP