zobally/USBDK
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Utils: Generalize auto-locking mechanisms

Browse source 
This patch makes code ready for other lock types like RW-locks

Signed-off-by: Dmitry Fleytman <dfleytma@redhat.com>
This commit is contained in:
Dmitry Fleytman 2015-12-23 11:31:12 +02:00
parent fa27333c0e
commit 39e3a4709c
1 changed files with 15 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

26
UsbDk/UsbDkUtil.h
View file

@ -43,25 +43,29 @@ private:
CWdmSpinLock& operator= (const CWdmSpinLock&) = delete;
};
template <typename T>
class CLockedContext
template <typename T, void (T::*LockFunc)(), void (T::*UnlockFunc)()>
class CBaseLockedContext
{
public:
CLockedContext(T &LockObject)
CBaseLockedContext(T &LockObject)
: m_LockObject(LockObject)
{ m_LockObject.Lock(); }
{
(m_LockObject.*LockFunc)();
}
~CLockedContext()
{ m_LockObject.Unlock(); }
~CBaseLockedContext()
{
(m_LockObject.*UnlockFunc)();
}
private:
protected:
T &m_LockObject;
CLockedContext(const CLockedContext&) = delete;
CLockedContext& operator= (const CLockedContext&) = delete;
CBaseLockedContext(const CBaseLockedContext&) = delete;
CBaseLockedContext& operator= (const CBaseLockedContext&) = delete;
};
typedef CLockedContext<CWdmSpinLock> TSpinLocker;
template <typename T>
using CLockedContext = CBaseLockedContext <T, &T::Lock, &T::Unlock>;
class CWdmRefCounter
{