mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
2.0 KiB
79 lines
2.0 KiB
#pragma once
|
|
|
|
namespace CertSrv
|
|
{
|
|
|
|
typedef VOID (WINAPI* PFNAUTOCLEANUP)(VOID*);
|
|
|
|
|
|
template<class PTR, PFNAUTOCLEANUP pfn, UINT_PTR pInvalid = NULL>
|
|
class CAutoPtr {
|
|
|
|
public:
|
|
CAutoPtr() { m_ptr = pInvalid; }
|
|
CAutoPtr(PTR ptr) : m_ptr(ptr) { }
|
|
~CAutoPtr() { Cleanup(); }
|
|
|
|
bool IsValid() const { return(m_ptr != pInvalid); }
|
|
bool IsInvalid()const { return(!IsValid()); }
|
|
|
|
PTR operator=(PTR ptr) { return Attach(ptr); }
|
|
operator PTR() const { return m_ptr; }
|
|
PTR* operator &() { CSASSERT(NULL==m_ptr); return &m_ptr; }
|
|
PTR& operator *() const { CSASSERT(NULL!=m_ptr); return *m_ptr; }
|
|
PTR operator ->() const { return m_ptr; }
|
|
bool operator ==(const PTR ptr) const { return m_ptr==ptr; }
|
|
bool operator !=(const PTR ptr) const { return m_ptr!=ptr; }
|
|
bool operator !() const { return NULL == m_ptr; }
|
|
|
|
void Cleanup()
|
|
{
|
|
if (IsValid())
|
|
{
|
|
pfn(m_ptr);
|
|
m_ptr = pInvalid;
|
|
}
|
|
}
|
|
|
|
PTR Attach(PTR ptr)
|
|
{
|
|
Cleanup();
|
|
m_ptr = ptr;
|
|
return(*this);
|
|
}
|
|
|
|
PTR Detach()
|
|
{
|
|
PTR ptrTemp = m_ptr;
|
|
m_ptr = pInvalid;
|
|
return m_ptr;
|
|
}
|
|
|
|
private:
|
|
|
|
// disable default copy constructor and assignment operator for
|
|
// CAutoPtr objects
|
|
CAutoPtr(const CAutoPtr& src) { }
|
|
CAutoPtr operator=(CAutoPtr p) { }
|
|
|
|
PTR m_ptr; // The member representing the object
|
|
};
|
|
|
|
|
|
#define DefineAutoClass(className, tData, pfnCleanup) \
|
|
typedef CAutoPtr<tData, (PFNAUTOCLEANUP) pfnCleanup> className;
|
|
|
|
#define DefineAutoClassEx(className, tData, pfnCleanup, pInvalid) \
|
|
typedef CAutoPtr<tData, (PFNAUTOCLEANUP) pfnCleanup, \
|
|
pInvalid> className;
|
|
|
|
|
|
// Instances of the template C++ class for common data PTRs.
|
|
DefineAutoClass(CAutoLPWSTR, LPWSTR, LocalFree);
|
|
DefineAutoClass(CAutoLPSTR, LPSTR, LocalFree);
|
|
DefineAutoClass(CAutoPBYTE, PBYTE, LocalFree);
|
|
DefineAutoClass(CAutoHANDLE, HANDLE, CloseHandle);
|
|
DefineAutoClass(CAutoBSTR, BSTR, SysFreeString);
|
|
|
|
|
|
}; // namespace CertSrv
|