SpinLock.h

Go to the documentation of this file.

#pragma once

#include "Common/TypesDefinitions.h"

#ifdef _M_AMD64
#include <mutex>

typedef std::mutex SpinLock;

#else


class SpinLock
{
private:
    volatile int32      m_locked;
protected:
public:
    SpinLock()
    {
        m_locked = 0;
    }
    ~SpinLock() = default;

    inline void lock();
    inline void unlock();
    inline bool try_lock();
};


inline void SpinLock::lock()
{
    _asm
    {
        mov         ecx,            [this]
        mov         EAX,            0xF
        TRY_LOCK_LABEL:
        xchg        EAX,            [ecx].m_locked
        and         EAX,            0xFF
        jnz         TRY_LOCK_LABEL
    }
}


inline void SpinLock::unlock()
{
    _asm mov        ecx,                [this]
    _asm mov        [ecx.m_locked],     0x0
    //_asm mov      EAX,        0x0
    //_asm xchg     EAX,        [ecx].m_locked
}


inline bool SpinLock::try_lock()
{
    bool success = false;
    _asm
    {
        mov         ecx,            [this]
        mov         EAX,            0xF
        xchg        EAX,            [ecx].m_locked
        cmp         EAX,            0x0
        jne         LOCK_FAILED_LABEL
    }
    return true;
    LOCK_FAILED_LABEL:
    return false;
}

#endif