70 lines
2.7 KiB
C
70 lines
2.7 KiB
C
|
// --------------------------------------------------------------------------
|
||
|
|
// Module Name: PrivilegeEnable.h
|
||
|
|
//
|
||
|
|
// Copyright (c) 1999-2000, Microsoft Corporation
|
||
|
|
//
|
||
|
|
// Classes that handle state preservation, changing and restoration.
|
||
|
|
//
|
||
|
|
// History: 1999-08-18 vtan created
|
||
|
|
// 1999-11-16 vtan separate file
|
||
|
|
// 2000-02-01 vtan moved from Neptune to Whistler
|
||
|
|
// --------------------------------------------------------------------------
|
||
|
|
|
||
|
|
#ifndef _PrivilegeEnable_
|
||
|
|
#define _PrivilegeEnable_
|
||
|
|
|
||
|
|
// --------------------------------------------------------------------------
|
||
|
|
// CThreadToken
|
||
|
|
//
|
||
|
|
// Purpose: This class gets the current thread's token. If the thread is
|
||
|
|
// not impersonating it gets the current process' token.
|
||
|
|
//
|
||
|
|
// History: 1999-08-18 vtan created
|
||
|
|
// 2000-02-01 vtan moved from Neptune to Whistler
|
||
|
|
// --------------------------------------------------------------------------
|
||
|
|
|
||
|
|
class CThreadToken
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
CThreadToken (void);
|
||
|
|
CThreadToken (const CThreadToken& copyObject);
|
||
|
|
bool operator == (const CThreadToken& compareObject) const;
|
||
|
|
const CThreadToken& operator = (const CThreadToken& assignObject);
|
||
|
|
public:
|
||
|
|
CThreadToken (DWORD dwDesiredAccess);
|
||
|
|
~CThreadToken (void);
|
||
|
|
|
||
|
|
operator HANDLE (void) const;
|
||
|
|
private:
|
||
|
|
HANDLE _hToken;
|
||
|
|
};
|
||
|
|
|
||
|
|
// --------------------------------------------------------------------------
|
||
|
|
// CPrivilegeEnable
|
||
|
|
//
|
||
|
|
// Purpose: This class enables a privilege for the duration of its scope.
|
||
|
|
// The privilege is restored to its original state on
|
||
|
|
// destruction.
|
||
|
|
//
|
||
|
|
// History: 1999-08-18 vtan created
|
||
|
|
// 2000-02-01 vtan moved from Neptune to Whistler
|
||
|
|
// --------------------------------------------------------------------------
|
||
|
|
|
||
|
|
class CPrivilegeEnable
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
CPrivilegeEnable (void);
|
||
|
|
CPrivilegeEnable (const CPrivilegeEnable& copyObject);
|
||
|
|
const CPrivilegeEnable& operator = (const CPrivilegeEnable& assignObject);
|
||
|
|
public:
|
||
|
|
CPrivilegeEnable (const TCHAR *pszName);
|
||
|
|
~CPrivilegeEnable (void);
|
||
|
|
private:
|
||
|
|
bool _fSet;
|
||
|
|
CThreadToken _hToken;
|
||
|
|
TOKEN_PRIVILEGES _oldPrivilege;
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif /* _PrivilegeEnable_ */
|
||
|
|
|