eversinc33-Banshee/Banshee/AutoLock.hpp

#pragma once

// From Pavel Yosifovich: Windows Kernel Programming

class FastMutex
{
private:
    FAST_MUTEX _mutex;

public:
    void Init()
    {
        ExInitializeFastMutex(&_mutex);
    }

    void Lock()
    {
        ExAcquireFastMutex(&_mutex);
    }

    void Unlock()
    {
        ExReleaseFastMutex(&_mutex);
    }
};

template<typename TLock>
struct AutoLock {
    AutoLock(TLock& lock) 
        : _lock(lock) 
    {
        _lock.Lock();
    }

    ~AutoLock() 
    {
        _lock.Unlock();
    }

private:
    TLock& _lock;
};