rwmutex.hpp

Go to the documentation of this file.

#ifndef LIBFILEZILLA_RWMUTEX_HEADER
#define LIBFILEZILLA_RWMUTEX_HEADER

#include "libfilezilla.hpp"
#include "time.hpp"
 
#ifdef FZ_WINDOWS
#include "glue/windows.hpp"
#else
#include <pthread.h>
#endif
 
namespace fz {

class FZ_PUBLIC_SYMBOL rwmutex final
{
public:
#ifdef FZ_WINDOWS
    explicit rwmutex() = default;
#else
    explicit rwmutex()
    {
        pthread_rwlock_init(&m_, nullptr);
    }
    ~rwmutex()
    {
        pthread_rwlock_destroy(&m_);
    }
#endif
 
    rwmutex(rwmutex const&) = delete;
    rwmutex& operator=(rwmutex const&) = delete;

    void lock_read()
    {
#ifdef FZ_WINDOWS
        AcquireSRWLockShared(&m_);
#else
        pthread_rwlock_rdlock(&m_);
#endif
    }
 
    void lock_write()
    {
#ifdef FZ_WINDOWS
        AcquireSRWLockExclusive(&m_);
#else
        pthread_rwlock_wrlock(&m_);
#endif
    }

    void unlock_read()
    {
#ifdef FZ_WINDOWS
        ReleaseSRWLockShared(&m_);
#else
        pthread_rwlock_unlock(&m_);
#endif
    }
 
    void unlock_write()
    {
#ifdef FZ_WINDOWS
        ReleaseSRWLockExclusive(&m_);
#else
        pthread_rwlock_unlock(&m_);
#endif
    }
 
private:
    friend class scoped_read_lock;
    friend class scoped_write_lock;
 
#ifdef FZ_WINDOWS
    SRWLOCK m_{};
#else
    pthread_rwlock_t m_;
#endif
};

class FZ_PUBLIC_SYMBOL scoped_read_lock final
{
public:
    explicit scoped_read_lock(rwmutex& m)
        : m_(&m.m_)
    {
#ifdef FZ_WINDOWS
        AcquireSRWLockShared(m_);
#else
        pthread_rwlock_rdlock(m_);
#endif
    }
 
    ~scoped_read_lock()
    {
        if (locked_) {
#ifdef FZ_WINDOWS
            ReleaseSRWLockShared(m_);
#else
            pthread_rwlock_unlock(m_);
#endif
        }
 
    }
 
    scoped_read_lock(scoped_read_lock const&) = delete;
    scoped_read_lock& operator=(scoped_read_lock const&) = delete;
 
    scoped_read_lock(scoped_read_lock&& op) noexcept
    {
        m_ = op.m_;
        op.m_ = 0;
        locked_ = op.locked_;
        op.locked_ = false;
    }
 
    scoped_read_lock& operator=(scoped_read_lock&& op) noexcept
    {
        if (this != &op) {
            m_ = op.m_;
            op.m_ = 0;
            locked_ = op.locked_;
            op.locked_ = false;
        }
        return *this;
    }

    void lock()
    {
        locked_ = true;
#ifdef FZ_WINDOWS
        AcquireSRWLockShared(m_);
#else
        pthread_rwlock_rdlock(m_);
#endif
    }

    void unlock()
    {
        locked_ = false;
#ifdef FZ_WINDOWS
        ReleaseSRWLockShared(m_);
#else
        pthread_rwlock_unlock(m_);
#endif
    }
 
private:
#ifdef FZ_WINDOWS
    SRWLOCK* m_;
#else
    pthread_rwlock_t* m_;
#endif
    bool locked_{ true };
};

class FZ_PUBLIC_SYMBOL scoped_write_lock final
{
public:
    explicit scoped_write_lock(rwmutex& m)
        : m_(&m.m_)
    {
#ifdef FZ_WINDOWS
        AcquireSRWLockExclusive(m_);
#else
        pthread_rwlock_wrlock(m_);
#endif
    }
 
    ~scoped_write_lock()
    {
        if (locked_) {
#ifdef FZ_WINDOWS
            ReleaseSRWLockExclusive(m_);
#else
            pthread_rwlock_unlock(m_);
#endif
        }
 
    }
 
    scoped_write_lock(scoped_write_lock const&) = delete;
    scoped_write_lock& operator=(scoped_write_lock const&) = delete;
 
    scoped_write_lock(scoped_write_lock&& op) noexcept
    {
        m_ = op.m_;
        op.m_ = 0;
        locked_ = op.locked_;
        op.locked_ = false;
    }
 
    scoped_write_lock& operator=(scoped_write_lock&& op) noexcept
    {
        if (this != &op) {
            m_ = op.m_;
            op.m_ = 0;
            locked_ = op.locked_;
            op.locked_ = false;
        }
        return *this;
    }

    void lock()
    {
        locked_ = true;
#ifdef FZ_WINDOWS
        AcquireSRWLockExclusive(m_);
#else
        pthread_rwlock_wrlock(m_);
#endif
    }

    void unlock()
    {
        locked_ = false;
#ifdef FZ_WINDOWS
        ReleaseSRWLockExclusive(m_);
#else
        pthread_rwlock_unlock(m_);
#endif
    }
 
private:
#ifdef FZ_WINDOWS
    SRWLOCK* m_;
#else
    pthread_rwlock_t* m_;
#endif
    bool locked_{ true };
};
 
}
 
#endif