214 lines
5.1 KiB
C++
214 lines
5.1 KiB
C++
//+-----------------------------------------------------------------------
|
|
//
|
|
// Copyright (C) Microsoft Corporation, 1993-1999
|
|
// File: resource.hxx
|
|
//
|
|
// Contents: CResource, CReadLock, CWriteLock,
|
|
// CSafeReadLock, CSafeWriteLock.
|
|
//
|
|
//
|
|
// History: 8 Mar 93 WadeR Created.
|
|
// 08-Jul-93 WadeR Moved from security to common
|
|
// 18-Nov-93 WadeR Made inline, added Safe locks
|
|
//
|
|
//------------------------------------------------------------------------
|
|
|
|
#ifndef _INC_RESOURCE_HXX_
|
|
#define _INC_RESOURCE_HXX_
|
|
|
|
#if _MSC_VER > 1000
|
|
#pragma once
|
|
#endif
|
|
|
|
#include <nt.h>
|
|
#include <ntrtl.h>
|
|
#include <nturtl.h>
|
|
#include <except.hxx>
|
|
|
|
//+-----------------------------------------------------------------
|
|
//
|
|
// Class: CResource
|
|
//
|
|
// Purpose: Allows multiple readers or single writer at a resource.
|
|
//
|
|
// Interface: [CResource] --
|
|
// [~CResource] --
|
|
// [GetRead] --
|
|
// [GetWrite] --
|
|
// [ReadToWrite] --
|
|
// [WriteToRead] --
|
|
// [Release] --
|
|
//
|
|
// History: 24-Feb-93 WadeR Created.
|
|
//
|
|
// Notes: This wrapper will allow starving writers, since it just
|
|
// passes everything through to the RTL Resource code, which
|
|
// allows starving writers.
|
|
//
|
|
//------------------------------------------------------------------
|
|
class CResource
|
|
{
|
|
RTL_RESOURCE _rResource;
|
|
public:
|
|
CResource();
|
|
~CResource();
|
|
void Release();
|
|
void GetRead();
|
|
void GetWrite();
|
|
void ReadToWrite();
|
|
void WriteToRead();
|
|
};
|
|
|
|
|
|
inline
|
|
CResource::CResource()
|
|
{
|
|
RtlInitializeResource( &_rResource );
|
|
}
|
|
|
|
inline
|
|
CResource::~CResource()
|
|
{
|
|
RtlDeleteResource( &_rResource );
|
|
}
|
|
|
|
inline
|
|
void CResource::Release()
|
|
{
|
|
RtlReleaseResource( &_rResource );
|
|
}
|
|
|
|
inline
|
|
void CResource::GetRead()
|
|
{
|
|
RtlAcquireResourceShared( &_rResource, TRUE );
|
|
}
|
|
|
|
inline
|
|
void CResource::GetWrite()
|
|
{
|
|
RtlAcquireResourceExclusive( &_rResource, TRUE );
|
|
}
|
|
|
|
inline
|
|
void CResource::ReadToWrite()
|
|
{
|
|
RtlConvertSharedToExclusive( &_rResource );
|
|
}
|
|
|
|
inline
|
|
void CResource::WriteToRead()
|
|
{
|
|
RtlConvertExclusiveToShared( &_rResource );
|
|
}
|
|
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Class: CReadLock
|
|
//
|
|
// Purpose: Lock using a Resource monitor
|
|
//
|
|
// History: 24-Feb-93 WadeR Created.
|
|
//
|
|
// Notes: Simple lock object to be created on the stack.
|
|
// The constructor acquires the resource, the destructor
|
|
// (called when lock is going out of scope) releases it.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
class CReadLock
|
|
{
|
|
public:
|
|
CReadLock ( CResource& r ) : _r(r)
|
|
{ _r.GetRead(); };
|
|
~CReadLock ()
|
|
{ _r.Release(); };
|
|
private:
|
|
CResource& _r;
|
|
};
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Class: CWriteLock
|
|
//
|
|
// Purpose: Lock using a Resource monitor
|
|
//
|
|
// History: 24-Feb-93 WadeR Created.
|
|
//
|
|
// Notes: Simple lock object to be created on the stack.
|
|
// The constructor acquires the resource, the destructor
|
|
// (called when lock is going out of scope) releases it.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
class CWriteLock
|
|
{
|
|
public:
|
|
CWriteLock ( CResource& r ) : _r(r)
|
|
{ _r.GetWrite(); };
|
|
~CWriteLock ()
|
|
{ _r.Release(); };
|
|
private:
|
|
CResource& _r;
|
|
};
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Class: CSafeReadLock
|
|
//
|
|
// Purpose: Exception safe lock using a Resource monitor
|
|
//
|
|
// History: 24-Feb-93 WadeR Created.
|
|
//
|
|
// Notes: Simple lock object to be created on the stack.
|
|
// The constructor acquires the resource, the destructor
|
|
// (called when lock is going out of scope) releases it.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
class CSafeReadLock : INHERIT_UNWIND
|
|
{
|
|
INLINE_UNWIND(CSafeReadLock);
|
|
|
|
public:
|
|
CSafeReadLock ( CResource& r ) : _r(r)
|
|
{ _r.GetRead(); END_CONSTRUCTION (CSafeReadLock); };
|
|
~CSafeReadLock ()
|
|
{ _r.Release(); };
|
|
private:
|
|
CResource& _r;
|
|
};
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Class: CSafeWriteLock
|
|
//
|
|
// Purpose: Exception safe lock using a Resource monitor
|
|
//
|
|
// History: 24-Feb-93 WadeR Created.
|
|
//
|
|
// Notes: Simple lock object to be created on the stack.
|
|
// The constructor acquires the resource, the destructor
|
|
// (called when lock is going out of scope) releases it.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
class CSafeWriteLock : INHERIT_UNWIND
|
|
{
|
|
INLINE_UNWIND(CSafeWriteLock);
|
|
|
|
public:
|
|
CSafeWriteLock ( CResource& r ) : _r(r)
|
|
{ _r.GetWrite(); END_CONSTRUCTION (CSafeWriteLock); };
|
|
~CSafeWriteLock ()
|
|
{ _r.Release(); };
|
|
private:
|
|
CResource& _r;
|
|
};
|
|
#endif // _INC_RESOURCE_HXX_
|