373 lines
9.9 KiB
C
373 lines
9.9 KiB
C
|
#ifndef _INC_DSKQUOTA_CONTROL_H
|
||
|
|
#define _INC_DSKQUOTA_CONTROL_H
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
/* File: control.h
|
||
|
|
|
||
|
|
Description: Contains declaration for class DiskQuotaControl.
|
||
|
|
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
Date Description Programmer
|
||
|
|
-------- --------------------------------------------------- ----------
|
||
|
|
05/22/96 Initial creation. BrianAu
|
||
|
|
*/
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
#ifndef _INC_DSKQUOTA_H
|
||
|
|
# include "dskquota.h"
|
||
|
|
#endif
|
||
|
|
#ifndef _INC_DSKQUOTA_FSOBJECT_H
|
||
|
|
# include "fsobject.h" // File System object types.
|
||
|
|
#endif
|
||
|
|
#ifndef _INC_DSKQUOTA_SIDNAME_H
|
||
|
|
# include "sidname.h" // Sid Name Resolver.
|
||
|
|
#endif
|
||
|
|
#ifndef _INC_DSKQUOTA_NTDS_H
|
||
|
|
# include "ntds.h" // For DS versions of LookupAccountSid/Name
|
||
|
|
#endif
|
||
|
|
#ifndef _INC_DSKQUOTA_DISPATCH_H
|
||
|
|
# include "dispatch.h" // MIDL-generated header.
|
||
|
|
#endif
|
||
|
|
#ifndef _INC_DSKQUOTA_OADISP_H
|
||
|
|
# include "oadisp.h" // OleAutoDispatch class.
|
||
|
|
#endif
|
||
|
|
|
||
|
|
|
||
|
|
class DiskQuotaControl : public IDiskQuotaControl
|
||
|
|
{
|
||
|
|
private:
|
||
|
|
LONG m_cRef; // Object ref count.
|
||
|
|
BOOL m_bInitialized; // Controller initialized?
|
||
|
|
LONGLONG m_llDefaultQuotaThreshold; // "New User" default threshold.
|
||
|
|
LONGLONG m_llDefaultQuotaLimit; // "New User" default limit.
|
||
|
|
FSObject *m_pFSObject; // Volume or directory.
|
||
|
|
DWORD m_dwFlags; // State of quota system.
|
||
|
|
PSID_NAME_RESOLVER m_pSidNameResolver; // For getting SID account names.
|
||
|
|
CMutex m_mutex; // Ensures safe shutdown.
|
||
|
|
|
||
|
|
//
|
||
|
|
// Support for IConnectionPointContainer.
|
||
|
|
// 1. Static array of supported interface IDs.
|
||
|
|
// 2. Array of connection pt interface pointers.
|
||
|
|
// Dynamically grows as clients connect for events.
|
||
|
|
//
|
||
|
|
static const IID * const m_rgpIConnPtsSupported[];
|
||
|
|
PCONNECTIONPOINT *m_rgConnPts; // Array of conn pt object ptrs.
|
||
|
|
UINT m_cConnPts; // Count of conn pts supported.
|
||
|
|
|
||
|
|
//
|
||
|
|
// Create connection point objects for the supported connection
|
||
|
|
// point types.
|
||
|
|
//
|
||
|
|
HRESULT
|
||
|
|
InitConnectionPoints(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Read quota information from disk to member variables.
|
||
|
|
//
|
||
|
|
HRESULT
|
||
|
|
QueryQuotaInformation(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Write quota information from member variables to disk.
|
||
|
|
//
|
||
|
|
HRESULT
|
||
|
|
SetQuotaInformation(
|
||
|
|
DWORD dwChangeFlags);
|
||
|
|
|
||
|
|
HRESULT
|
||
|
|
GetDefaultQuotaItem(
|
||
|
|
PLONGLONG pllItem,
|
||
|
|
PLONGLONG pllValueOut);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Prevent copy construction.
|
||
|
|
//
|
||
|
|
DiskQuotaControl(const DiskQuotaControl& control);
|
||
|
|
void operator = (const DiskQuotaControl& control);
|
||
|
|
|
||
|
|
public:
|
||
|
|
NTDS m_NTDS;
|
||
|
|
|
||
|
|
//
|
||
|
|
// If you add a new connection point type, add a corresponding enumeration
|
||
|
|
// member that identifies the location of the conn pt IID in
|
||
|
|
// m_rgpIConnPtsSupported[].
|
||
|
|
//
|
||
|
|
enum { ConnPt_iQuotaEvents = 0,
|
||
|
|
ConnPt_iQuotaEventsDisp = 1, };
|
||
|
|
|
||
|
|
DiskQuotaControl(VOID);
|
||
|
|
~DiskQuotaControl(VOID);
|
||
|
|
|
||
|
|
HRESULT NotifyUserNameChanged(PDISKQUOTA_USER pUser);
|
||
|
|
|
||
|
|
FSObject *GetFSObjectPtr(VOID)
|
||
|
|
{ return m_pFSObject; }
|
||
|
|
|
||
|
|
//
|
||
|
|
// IUnknown methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
QueryInterface(
|
||
|
|
REFIID riid,
|
||
|
|
LPVOID *ppvOut);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
AddRef(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
Release(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IConnectionPointContainer methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
FindConnectionPoint(
|
||
|
|
REFIID,
|
||
|
|
PCONNECTIONPOINT *pCP);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
EnumConnectionPoints(
|
||
|
|
PENUMCONNECTIONPOINTS *pEnumCP);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IDiskQuotaControl methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
Initialize(
|
||
|
|
LPCWSTR pszFSObjectName,
|
||
|
|
BOOL bReadWrite);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetQuotaState(
|
||
|
|
LPDWORD pdwState);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
SetQuotaState(
|
||
|
|
DWORD dwState);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
SetQuotaLogFlags(
|
||
|
|
DWORD dwFlags);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetQuotaLogFlags(
|
||
|
|
LPDWORD pdwFlags);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
SetDefaultQuotaThreshold(
|
||
|
|
LONGLONG llThreshold);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
SetDefaultQuotaLimit(
|
||
|
|
LONGLONG llLimit);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetDefaultQuotaThreshold(
|
||
|
|
PLONGLONG pllThreshold);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetDefaultQuotaThresholdText(
|
||
|
|
LPWSTR pszText,
|
||
|
|
DWORD cchText);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetDefaultQuotaLimit(
|
||
|
|
PLONGLONG pllLimit);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetDefaultQuotaLimitText(
|
||
|
|
LPWSTR pszText,
|
||
|
|
DWORD cchText);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
AddUserSid(
|
||
|
|
PSID pSid,
|
||
|
|
DWORD fNameResolution,
|
||
|
|
PDISKQUOTA_USER *ppUser);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
AddUserName(
|
||
|
|
LPCWSTR pszLogonName,
|
||
|
|
DWORD fNameResolution,
|
||
|
|
PDISKQUOTA_USER *ppUser);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
DeleteUser(
|
||
|
|
PDISKQUOTA_USER pUser);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
FindUserSid(
|
||
|
|
PSID pSid,
|
||
|
|
DWORD fNameResolution,
|
||
|
|
PDISKQUOTA_USER *ppUser);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
FindUserName(
|
||
|
|
LPCWSTR pszLogonName,
|
||
|
|
PDISKQUOTA_USER *ppUser);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
CreateEnumUsers(
|
||
|
|
PSID *rgpSids,
|
||
|
|
DWORD cpSids,
|
||
|
|
DWORD fNameResolution,
|
||
|
|
PENUM_DISKQUOTA_USERS *ppEnum);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
CreateUserBatch(
|
||
|
|
PDISKQUOTA_USER_BATCH *ppUserBatch);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
InvalidateSidNameCache(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GiveUserNameResolutionPriority(
|
||
|
|
PDISKQUOTA_USER pUser);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
ShutdownNameResolution(
|
||
|
|
VOID);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
class DiskQuotaControlDisp : public DIDiskQuotaControl
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
DiskQuotaControlDisp(PDISKQUOTA_CONTROL pQC);
|
||
|
|
|
||
|
|
~DiskQuotaControlDisp(VOID);
|
||
|
|
|
||
|
|
//
|
||
|
|
// IUnknown methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
QueryInterface(
|
||
|
|
REFIID riid,
|
||
|
|
LPVOID *ppvOut);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
AddRef(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
STDMETHODIMP_(ULONG)
|
||
|
|
Release(
|
||
|
|
VOID);
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// IDispatch methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP
|
||
|
|
GetIDsOfNames(
|
||
|
|
REFIID riid,
|
||
|
|
OLECHAR ** rgszNames,
|
||
|
|
UINT cNames,
|
||
|
|
LCID lcid,
|
||
|
|
DISPID *rgDispId);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetTypeInfo(
|
||
|
|
UINT iTInfo,
|
||
|
|
LCID lcid,
|
||
|
|
ITypeInfo **ppTInfo);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
GetTypeInfoCount(
|
||
|
|
UINT *pctinfo);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
Invoke(
|
||
|
|
DISPID dispIdMember,
|
||
|
|
REFIID riid,
|
||
|
|
LCID lcid,
|
||
|
|
WORD wFlags,
|
||
|
|
DISPPARAMS *pDispParams,
|
||
|
|
VARIANT *pVarResult,
|
||
|
|
EXCEPINFO *pExcepInfo,
|
||
|
|
UINT *puArgErr);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Automation Properties.
|
||
|
|
//
|
||
|
|
STDMETHODIMP put_QuotaState(QuotaStateConstants State);
|
||
|
|
STDMETHODIMP get_QuotaState(QuotaStateConstants *pState);
|
||
|
|
|
||
|
|
STDMETHODIMP get_QuotaFileIncomplete(VARIANT_BOOL *pbIncomplete);
|
||
|
|
|
||
|
|
STDMETHODIMP get_QuotaFileRebuilding(VARIANT_BOOL *pbRebuilding);
|
||
|
|
|
||
|
|
STDMETHODIMP put_LogQuotaThreshold(VARIANT_BOOL bLogThreshold);
|
||
|
|
STDMETHODIMP get_LogQuotaThreshold(VARIANT_BOOL *pbLogThreshold);
|
||
|
|
|
||
|
|
STDMETHODIMP put_LogQuotaLimit(VARIANT_BOOL bLogLimit);
|
||
|
|
STDMETHODIMP get_LogQuotaLimit(VARIANT_BOOL *pbLogLimit);
|
||
|
|
|
||
|
|
STDMETHODIMP put_DefaultQuotaThreshold(double Threshold);
|
||
|
|
STDMETHODIMP get_DefaultQuotaThreshold(double *pThreshold);
|
||
|
|
STDMETHODIMP get_DefaultQuotaThresholdText(BSTR *pThresholdText);
|
||
|
|
|
||
|
|
STDMETHODIMP put_DefaultQuotaLimit(double Limit);
|
||
|
|
STDMETHODIMP get_DefaultQuotaLimit(double *pLimit);
|
||
|
|
STDMETHODIMP get_DefaultQuotaLimitText(BSTR *pLimitText);
|
||
|
|
|
||
|
|
STDMETHODIMP put_UserNameResolution(UserNameResolutionConstants ResolutionType);
|
||
|
|
STDMETHODIMP get_UserNameResolution(UserNameResolutionConstants *pResolutionType);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Automation Methods.
|
||
|
|
//
|
||
|
|
STDMETHODIMP Initialize(
|
||
|
|
BSTR path,
|
||
|
|
VARIANT_BOOL bReadOnly);
|
||
|
|
|
||
|
|
STDMETHODIMP AddUser(
|
||
|
|
BSTR LogonName,
|
||
|
|
DIDiskQuotaUser **ppUser);
|
||
|
|
|
||
|
|
STDMETHODIMP DeleteUser(
|
||
|
|
DIDiskQuotaUser *pUser);
|
||
|
|
|
||
|
|
STDMETHODIMP FindUser(
|
||
|
|
BSTR LogonName,
|
||
|
|
DIDiskQuotaUser **ppUser);
|
||
|
|
|
||
|
|
STDMETHODIMP TranslateLogonNameToSID(
|
||
|
|
BSTR LogonName,
|
||
|
|
BSTR *psid);
|
||
|
|
|
||
|
|
STDMETHODIMP
|
||
|
|
_NewEnum(
|
||
|
|
IDispatch **ppEnum);
|
||
|
|
|
||
|
|
STDMETHODIMP InvalidateSidNameCache(void);
|
||
|
|
|
||
|
|
STDMETHODIMP GiveUserNameResolutionPriority(
|
||
|
|
DIDiskQuotaUser *pUser);
|
||
|
|
|
||
|
|
STDMETHODIMP ShutdownNameResolution(void);
|
||
|
|
|
||
|
|
private:
|
||
|
|
LONG m_cRef;
|
||
|
|
PDISKQUOTA_CONTROL m_pQC; // For delegation
|
||
|
|
OleAutoDispatch m_Dispatch;
|
||
|
|
DWORD m_fOleAutoNameResolution;
|
||
|
|
PENUM_DISKQUOTA_USERS m_pUserEnum;
|
||
|
|
|
||
|
|
//
|
||
|
|
// Prevent copy.
|
||
|
|
//
|
||
|
|
DiskQuotaControlDisp(const DiskQuotaControlDisp& rhs);
|
||
|
|
DiskQuotaControlDisp& operator = (const DiskQuotaControlDisp& rhs);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
#endif // __DISK_QUOTA_CONTROL_H
|