324 lines
10 KiB
C++
324 lines
10 KiB
C++
//+----------------------------------------------------------------------------
|
|
//
|
|
// DS Administration MMC snapin.
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 1992 - 1999
|
|
//
|
|
// File: query.h
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
|
|
#ifndef __QUERY_
|
|
#define __QUERY_
|
|
|
|
#include "dscookie.h"
|
|
#include "dsfilter.h" // CDSAdminPersistQueryFilterImpl
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
class CFavoritesNode : public CGenericUINode
|
|
{
|
|
public:
|
|
CFavoritesNode();
|
|
|
|
void RemoveQueryResults();
|
|
void FindCookiesInQueries(LPCWSTR lpszCookieDN, CUINodeList* pNodeList);
|
|
|
|
//
|
|
// These set the state of the standard context menu items
|
|
//
|
|
virtual BOOL IsDeleteAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsRenameAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsRefreshAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsCutAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsCopyAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsPasteAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL ArePropertiesAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL HasPropertyPages(LPDATAOBJECT) { return TRUE; }
|
|
|
|
virtual CContextMenuVerbs* GetContextMenuVerbsObject(CDSComponentData* pComponentData);
|
|
virtual HRESULT OnCommand(long lCommandID, CDSComponentData* pComponentData);
|
|
|
|
virtual BOOL IsUniqueName(PCWSTR pszName, CUINode** ppDuplicateNode);
|
|
virtual HRESULT Rename(LPCWSTR lpszNewName, CDSComponentData* pComponentData);
|
|
virtual HRESULT Delete(CDSComponentData* pComponentData);
|
|
virtual HRESULT DeleteMultiselect(CDSComponentData* pComponentData, CInternalFormatCracker* pObjCracker);
|
|
|
|
virtual void Paste(IDataObject* pPasteData, CDSComponentData* pComponentData, LPDATAOBJECT* ppCutDataObj);
|
|
virtual HRESULT QueryPaste(IDataObject* pPasteData, CDSComponentData* pComponentData);
|
|
|
|
CFavoritesNode* GetFavoritesNode() { return dynamic_cast<CFavoritesNode*>(GetParent()); }
|
|
virtual BOOL IsFavoritesRoot() { return m_bFavoritesRoot; }
|
|
virtual void SetRoot(BOOL bRoot) { m_bFavoritesRoot = bRoot; }
|
|
|
|
virtual LPCWSTR GetDisplayString(int nCol, CDSColumnSet*)
|
|
{
|
|
if (IsFavoritesRoot())
|
|
{
|
|
if (nCol == 0)
|
|
return GetName();
|
|
else if (nCol == 2)
|
|
return GetDesc();
|
|
}
|
|
else
|
|
{
|
|
if (nCol == 0)
|
|
return GetName();
|
|
else if (nCol == 1)
|
|
return GetDesc();
|
|
}
|
|
return L"";
|
|
}
|
|
|
|
//
|
|
// Context menu command handlers
|
|
//
|
|
void OnImportQuery(CDSComponentData* pComponentData);
|
|
void OnNewFavoritesFolder(CDSComponentData* pComponentData);
|
|
void OnNewSavedQuery(CDSComponentData* pComponentData);
|
|
|
|
BOOL DeepCopyChildren(CUINode* pUINodeToCopy);
|
|
|
|
virtual HRESULT XMLSave(IXMLDOMDocument* pXMLDoc,
|
|
IXMLDOMNode** ppXMLDOMNode);
|
|
|
|
//
|
|
// Used to load the embedded favorites root node
|
|
// All other favorites folders are loaded through the static method XMLLoad
|
|
//
|
|
HRESULT CFavoritesNode::Load(IXMLDOMNode* pXDN, CDSComponentData* pComponentData);
|
|
|
|
//
|
|
// static members for XML serialization
|
|
//
|
|
static HRESULT XMLLoad(CDSComponentData* pComponentData,
|
|
IXMLDOMNode* pXDN,
|
|
CFavoritesNode** ppNode);
|
|
static LPCWSTR g_szObjXMLTag;
|
|
|
|
virtual HRESULT CreatePropertyPages(LPPROPERTYSHEETCALLBACK pCall,
|
|
LONG_PTR lNotifyHandle,
|
|
LPDATAOBJECT pDataObject,
|
|
CDSComponentData* pComponentData);
|
|
|
|
virtual void InvalidateSavedQueriesContainingObjects(CDSComponentData* pComponentData,
|
|
const CStringList& refDNList);
|
|
|
|
private:
|
|
bool IsSameSnapin(IDataObject* pPasteData, CDSComponentData* pComponentData);
|
|
|
|
|
|
|
|
BOOL m_bFavoritesRoot;
|
|
|
|
};
|
|
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
HRESULT FilterDisabledAccounts(CUINodeList* pUINodeList);
|
|
HRESULT FilterNonExpiringPwds(CUINodeList* pUINodeList);
|
|
|
|
class CSavedQueryNode : public CGenericUINode
|
|
{
|
|
public:
|
|
CSavedQueryNode(MyBasePathsInfo* pBasePathsInfo,
|
|
SnapinType snapinType = SNAPINTYPE_DS);
|
|
CSavedQueryNode(const CSavedQueryNode& copyNode);
|
|
virtual ~CSavedQueryNode();
|
|
|
|
void SetRootPath(LPCWSTR lpszRootPath);
|
|
void SetQueryString(LPCWSTR lpszQueryString) { m_szQueryString = lpszQueryString;}
|
|
|
|
LPCWSTR GetRootPath();
|
|
LPCWSTR GetQueryString();
|
|
void AppendLastLogonQuery(CString& szQuery, DWORD dwDays);
|
|
|
|
CFavoritesNode* GetFavoritesNode() { return dynamic_cast<CFavoritesNode*>(GetParent()); }
|
|
|
|
void SetOneLevel(BOOL bOneLevel) { m_bOneLevel = bOneLevel;}
|
|
BOOL IsOneLevel() { return m_bOneLevel;}
|
|
|
|
void SetValid(BOOL bValid) { m_bQueryValid = bValid; }
|
|
BOOL IsValid() { return m_bQueryValid; }
|
|
|
|
void SetQueryPersist(CComObject<CDSAdminPersistQueryFilterImpl>* pPersistQueryImpl);
|
|
CComObject<CDSAdminPersistQueryFilterImpl>* GetQueryPersist() { return m_pPersistQueryImpl; }
|
|
|
|
void FindCookieByDN(LPCWSTR lpszCookieDN, CUINodeList* pNodeList);
|
|
|
|
void SetColumnID(CDSComponentData* pComponentData, PCWSTR pszColumnID);
|
|
CDSColumnSet* GetColumnSet(CDSComponentData* pComponentData);
|
|
|
|
//
|
|
// These set the state of the standard context menu items
|
|
//
|
|
virtual BOOL IsDeleteAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsRenameAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsRefreshAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsPasteAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsCutAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
virtual BOOL IsCopyAllowed(CDSComponentData* pComponentData, BOOL* pbHide);
|
|
|
|
virtual CContextMenuVerbs* GetContextMenuVerbsObject(CDSComponentData* pComponentData);
|
|
virtual HRESULT OnCommand(long lCommandID, CDSComponentData* pComponentData);
|
|
|
|
virtual HRESULT Delete(CDSComponentData* pComponentData);
|
|
virtual HRESULT Rename(LPCWSTR lpszNewName, CDSComponentData* pComponentData);
|
|
|
|
void OnEditQuery(CDSComponentData* pComponentData);
|
|
void OnExportQuery(CDSComponentData* pComponentData);
|
|
|
|
virtual HRESULT XMLSave(IXMLDOMDocument* pXMLDoc,
|
|
IXMLDOMNode** ppXMLDOMNode);
|
|
|
|
//
|
|
// static members for XML serialization
|
|
//
|
|
static HRESULT XMLLoad(CDSComponentData* pComponentData,
|
|
IXMLDOMNode* pXDN,
|
|
CSavedQueryNode** ppQuery);
|
|
static LPCWSTR g_szObjXMLTag;
|
|
static LPCWSTR g_szOneLevelXMLTag;
|
|
static LPCWSTR g_szQueryStringXMLTag;
|
|
static LPCWSTR g_szDSQueryPersistTag;
|
|
static LPCWSTR g_szDisabledAccountsFilterTag;
|
|
static LPCWSTR g_szNonExpiringPwdFilterTag;
|
|
static LPCWSTR g_szLastLogonFilterTag;
|
|
static LPCWSTR g_szColumnIDTag;
|
|
|
|
BOOL IsFilterDisabledAccounts();
|
|
BOOL IsFilterNonExpiringPwds();
|
|
BOOL IsFilterLastLogon();
|
|
|
|
void SetLastLogonQuery(DWORD dwLastLogonDays)
|
|
{
|
|
m_bLastLogonFilter = dwLastLogonDays != (DWORD)-1;
|
|
m_dwLastLogonDays = dwLastLogonDays;
|
|
}
|
|
|
|
DWORD GetLastLogonDays() { return m_dwLastLogonDays; }
|
|
|
|
virtual void InvalidateSavedQueriesContainingObjects(CDSComponentData* pComponentData,
|
|
const CStringList& refDNList);
|
|
|
|
private:
|
|
CString m_szRelativeRootPath;
|
|
CString m_szCurrentFullPath;
|
|
|
|
CString m_szColumnID;
|
|
CString m_szQueryString;
|
|
BOOL m_bOneLevel;
|
|
|
|
BOOL m_bQueryValid;
|
|
|
|
BOOL m_bLastLogonFilter;
|
|
DWORD m_dwLastLogonDays;
|
|
|
|
MyBasePathsInfo* m_pBasePathsInfo;
|
|
|
|
//
|
|
// for presisting DSQuery dialog info
|
|
//
|
|
CComObject<CDSAdminPersistQueryFilterImpl>* m_pPersistQueryImpl;
|
|
};
|
|
|
|
////////////////////////////////////////////////////////////////////////////
|
|
|
|
class CFavoritesNodesHolder
|
|
{
|
|
public:
|
|
CFavoritesNodesHolder()
|
|
{
|
|
AFX_MANAGE_STATE(AfxGetStaticModuleState());
|
|
|
|
// fill in the root
|
|
m_favoritesRoot.SetRoot(TRUE);
|
|
|
|
CString szName;
|
|
szName.LoadString(IDS_SAVED_QUERIES);
|
|
m_favoritesRoot.SetName(szName);
|
|
|
|
CString szDescription;
|
|
szDescription.LoadString(IDS_SAVED_QUERIES_DESC);
|
|
m_favoritesRoot.SetDesc(szDescription);
|
|
}
|
|
|
|
CFavoritesNode* GetFavoritesRoot() { return &m_favoritesRoot;}
|
|
|
|
/* For test purposes ONLY
|
|
void BuildTestTree(LPCWSTR lpszXMLFileName, SnapinType snapinType);
|
|
BOOL BuildTestTreefromXML(LPCWSTR lpszXMLFileName, SnapinType snapinType);
|
|
void BuildTestTreeHardWired(SnapinType snapinType);
|
|
*/
|
|
|
|
HRESULT Save(IStream* pStm);
|
|
HRESULT Load(IStream* pStm, CDSComponentData* pComponentData);
|
|
|
|
virtual void InvalidateSavedQueriesContainingObjects(CDSComponentData* pComponentData,
|
|
const CStringList& refDNList);
|
|
|
|
private:
|
|
CFavoritesNode m_favoritesRoot;
|
|
};
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
|
|
class CDummyStream :
|
|
public IStream,
|
|
public CComObjectRoot
|
|
{
|
|
BEGIN_COM_MAP(CDummyStream)
|
|
COM_INTERFACE_ENTRY(IStream)
|
|
END_COM_MAP()
|
|
|
|
public:
|
|
CDummyStream()
|
|
{
|
|
m_pByteArray = NULL;
|
|
m_nReadIndex = 0;
|
|
m_nByteCount = 0;
|
|
}
|
|
~CDummyStream()
|
|
{
|
|
if (m_pByteArray != NULL)
|
|
{
|
|
delete[] m_pByteArray;
|
|
m_pByteArray = NULL;
|
|
}
|
|
}
|
|
|
|
//
|
|
// ISequentialStream Interfaces
|
|
//
|
|
HRESULT STDMETHODCALLTYPE Read(void *pv, ULONG cb, ULONG *pcbRead);
|
|
HRESULT STDMETHODCALLTYPE Write(void const *pv, ULONG cb, ULONG *pcbWritten);
|
|
|
|
//
|
|
// IStream Interfaces (none of these are implemented)
|
|
//
|
|
HRESULT STDMETHODCALLTYPE Seek(LARGE_INTEGER, DWORD, ULARGE_INTEGER*) { return E_NOTIMPL; }
|
|
HRESULT STDMETHODCALLTYPE SetSize(ULARGE_INTEGER) { return E_NOTIMPL; }
|
|
HRESULT STDMETHODCALLTYPE CopyTo(IStream*, ULARGE_INTEGER, ULARGE_INTEGER*, ULARGE_INTEGER*) { return E_NOTIMPL; }
|
|
HRESULT STDMETHODCALLTYPE Commit(DWORD) { return E_NOTIMPL; }
|
|
HRESULT STDMETHODCALLTYPE Revert(void) { return E_NOTIMPL; }
|
|
HRESULT STDMETHODCALLTYPE LockRegion(ULARGE_INTEGER, ULARGE_INTEGER, DWORD) { return E_NOTIMPL; }
|
|
HRESULT STDMETHODCALLTYPE UnlockRegion(ULARGE_INTEGER, ULARGE_INTEGER, DWORD) { return E_NOTIMPL; }
|
|
HRESULT STDMETHODCALLTYPE Stat(STATSTG*, DWORD) { return E_NOTIMPL; }
|
|
HRESULT STDMETHODCALLTYPE Clone(IStream**) { return E_NOTIMPL; }
|
|
|
|
ULONG GetStreamData(BYTE** ppByteArray);
|
|
HRESULT SetStreamData(BYTE* pByteArray, ULONG nByteCount);
|
|
|
|
private:
|
|
BYTE* m_pByteArray;
|
|
ULONG m_nByteCount;
|
|
|
|
ULONG m_nReadIndex;
|
|
};
|
|
|
|
|
|
#endif // __QUERY_
|