windows-nt/Source/XPSP1/NT/admin/activec/nodemgr/mmcmt.h
2020-09-26 16:20:57 +08:00

123 lines
2.1 KiB
C++

//+-------------------------------------------------------------------------
//
// Microsoft Windows
//
// Copyright (C) Microsoft Corporation, 1999 - 1999
//
// File: mmcmt.h
//
//--------------------------------------------------------------------------
#ifndef __MMCMT_H_
#define __MMCMT_H_
/*
mmcmt.h
Simple thread synchronization classes
*/
/////////////////////////////////////////////////////////////////////////////
// CSyncObject
class CSyncObject
{
friend class CMutex;
public:
CSyncObject();
virtual ~CSyncObject();
operator HANDLE() const { return m_hObject; }
virtual BOOL Lock(DWORD dwTimeout = INFINITE);
virtual BOOL Unlock() = 0;
virtual BOOL Unlock(LONG /* lCount */, LPLONG /* lpPrevCount=NULL */) { return TRUE; }
protected:
HANDLE m_hObject;
};
/////////////////////////////////////////////////////////////////////////////
// CMutex
class CMutex : public CSyncObject
{
public:
CMutex(BOOL bInitiallyOwn = FALSE);
BOOL Unlock();
};
/////////////////////////////////////////////////////////////////////////////
// CCriticalSection
class CCriticalSection : public CSyncObject
{
public:
CCriticalSection();
virtual ~CCriticalSection();
operator CRITICAL_SECTION*() { return &m_sect; }
BOOL Lock();
BOOL Unlock();
protected:
CRITICAL_SECTION m_sect;
};
inline CCriticalSection::CCriticalSection() :
CSyncObject()
{
::InitializeCriticalSection( &m_sect );
}
inline CCriticalSection::~CCriticalSection()
{
::DeleteCriticalSection( &m_sect );
}
inline BOOL CCriticalSection::Lock()
{
::EnterCriticalSection( &m_sect );
return TRUE;
}
inline BOOL CCriticalSection::Unlock()
{
::LeaveCriticalSection( &m_sect );
return TRUE;
}
/////////////////////////////////////////////////////////////////////////////
// CSingleLock
class CSingleLock
{
public:
CSingleLock(CSyncObject* pObject, BOOL bInitialLock = FALSE);
~CSingleLock() { Unlock(); }
BOOL Lock(DWORD dwTimeOut = INFINITE);
BOOL Unlock();
BOOL IsLocked() { return m_bAcquired; }
protected:
CSyncObject* m_pObject;
HANDLE m_hObject;
BOOL m_bAcquired;
};
#endif // __MMCMT_H__