Source code of Windows XP (NT5)
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.
 
 
 
 
 
 

2151 lines
66 KiB

#include "shellprv.h"
#pragma hdrstop
#include "dspsprt.h"
#include "findfilter.h"
#include "cowsite.h"
#include "cobjsafe.h"
#include "cnctnpt.h"
#include "stdenum.h"
#include "exdisp.h"
#include "exdispid.h"
#include "shldisp.h"
#include "shdispid.h"
#include "dataprv.h"
#include "ids.h"
#include "views.h"
#include "findband.h"
#define WM_DF_SEARCHPROGRESS (WM_USER + 42)
#define WM_DF_ASYNCPROGRESS (WM_USER + 43)
#define WM_DF_SEARCHSTART (WM_USER + 44)
#define WM_DF_SEARCHCOMPLETE (WM_USER + 45)
#define WM_DF_FSNOTIFY (WM_USER + 46)
STDAPI CDocFindCommand_CreateInstance(IUnknown *punkOuter, REFIID riid, void **ppv);
typedef struct
{
BSTR bstrName;
VARIANT vValue;
} CMD_CONSTRAINT;
typedef struct
{
LPTSTR pszDotType;
LPTSTR pszDefaultValueMatch;
LPTSTR pszGuid; // If NULL, patch either pszDefaultValueMatch, or pszDotType whichever you find
} TYPE_FIX_ENTRY;
class CFindCmd : public ISearchCommandExt,
public CImpIDispatch,
public CObjectWithSite,
public CObjectSafety,
public IConnectionPointContainer,
public IProvideClassInfo2,
public CSimpleData,
public IRowsetWatchNotify,
public IFindControllerNotify
{
public:
// IUnknown
STDMETHODIMP QueryInterface(REFIID riid, void **ppv);
STDMETHODIMP_(ULONG) AddRef();
STDMETHODIMP_(ULONG) Release();
// IDispatch
STDMETHOD(GetTypeInfoCount)(UINT * pctinfo);
STDMETHOD(GetTypeInfo)(UINT itinfo, LCID lcid, ITypeInfo **pptinfo);
STDMETHOD(GetIDsOfNames)(REFIID riid, OLECHAR **rgszNames, UINT cNames, LCID lcid, DISPID * rgdispid);
STDMETHOD(Invoke)(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS * pdispparams, VARIANT * pvarResult, EXCEPINFO * pexcepinfo, UINT * puArgErr);
// IConnectionPointContainer
STDMETHOD(EnumConnectionPoints)(IEnumConnectionPoints **ppEnum);
STDMETHOD(FindConnectionPoint)(REFIID riid, IConnectionPoint **ppCP);
// IProvideClassInfo
STDMETHOD(GetClassInfo)(ITypeInfo **ppTI);
// IProvideClassInfo2
STDMETHOD(GetGUID)(DWORD dwGuidKind, GUID *pGUID);
// IObjectWithSite
STDMETHOD(SetSite)(IUnknown *punkSite);
// ISearchCommandExt
STDMETHOD(ClearResults)(void);
STDMETHOD(NavigateToSearchResults)(void);
STDMETHOD(get_ProgressText)(BSTR *pbs);
STDMETHOD(SaveSearch)(void);
STDMETHOD(RestoreSearch)(void);
STDMETHOD(GetErrorInfo)(BSTR *pbs, int *phr);
STDMETHOD(SearchFor)(int iFor);
STDMETHOD(GetScopeInfo)(BSTR bsScope, int *pdwScopeInfo);
STDMETHOD(RestoreSavedSearch)(VARIANT *pvarFile);
STDMETHOD(Execute)(VARIANT *RecordsAffected, VARIANT *Parameters, long Options);
STDMETHOD(AddConstraint)(BSTR Name, VARIANT Value);
STDMETHOD(GetNextConstraint)(VARIANT_BOOL fReset, DFConstraint **ppdfc);
// IRowsetWatchNotify
STDMETHODIMP OnChange(IRowset *prowset, DBWATCHNOTIFY eChangeReason);
// IFindControllerNotify
STDMETHODIMP DoSortOnColumn(UINT iCol, BOOL fSameCol);
STDMETHODIMP StopSearch(void);
STDMETHODIMP GetItemCount(UINT *pcItems);
STDMETHODIMP SetItemCount(UINT cItems);
STDMETHODIMP ViewDestroyed();
CFindCmd();
HRESULT Init(void);
private:
~CFindCmd();
HRESULT _GetSearchIDList(LPITEMIDLIST *ppidl);
HRESULT _SetEmptyText(UINT nID);
HRESULT _Clear();
void _SelectResults();
HWND _GetWindow();
struct THREAD_PARAMS {
CFindCmd *pThis;
IFindEnum *penum;
};
struct DEFER_UPDATE_DIR {
struct DEFER_UPDATE_DIR *pdudNext;
LPITEMIDLIST pidl;
BOOL fRecurse;
};
// Internal class to handle notifications from top level browser
class CWBEvents2: public DWebBrowserEvents, DWebBrowserEvents2
{
public:
STDMETHOD(QueryInterface) (REFIID riid, void **ppv);
STDMETHOD_(ULONG, AddRef)(void) { return _pcdfc->AddRef();}
STDMETHOD_(ULONG, Release)(void) { return _pcdfc->Release();}
// (DwebBrowserEvents)IDispatch
STDMETHOD(GetTypeInfoCount)(UINT * pctinfo) { return E_NOTIMPL;}
STDMETHOD(GetTypeInfo)(UINT itinfo, LCID lcid, ITypeInfo **pptinfo) { return E_NOTIMPL;}
STDMETHOD(GetIDsOfNames)(REFIID riid, OLECHAR **rgszNames, UINT cNames, LCID lcid, DISPID * rgdispid) { return E_NOTIMPL;}
STDMETHOD(Invoke)(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS * pdispparams, VARIANT * pvarResult, EXCEPINFO * pexcepinfo, UINT * puArgErr);
// Some helper functions...
void SetOwner(CFindCmd *pcdfc) { _pcdfc = pcdfc; } // Don't addref as part of larger object... }
void SetWaiting(BOOL fWait) {_fWaitingForNavigate = fWait;}
protected:
// Internal variables...
CFindCmd *_pcdfc; // pointer to top object... could cast, but...
BOOL _fWaitingForNavigate; // Are we waiting for the navigate to search resluts?
};
friend class CWBEvents2;
CWBEvents2 _cwbe;
IConnectionPoint *_pcpBrowser; // hold onto browsers connection point;
ULONG _dwCookie; // Cookie returned by Advise
HRESULT _UpdateFilter(IFindFilter *pfilter);
void _ClearConstraints();
static DWORD CALLBACK _ThreadProc(void *pv);
void _DoSearch(IFindEnum *penum);
HRESULT _Start(BOOL fNavigateIfFail, int iCol, LPCITEMIDLIST pidlUpdate);
HRESULT _Cancel();
HRESULT _Init(THREAD_PARAMS **ppParams, int iCol, LPCITEMIDLIST pidlUpdate);
static HRESULT _FreeThreadParams(THREAD_PARAMS *ptp);
HRESULT _ExecData_Init();
HRESULT _EnsureResultsViewIsCurrent(IUnknown *punk);
HRESULT _ExecData_Release();
BOOL _SetupBrowserCP();
void cdecl _NotifyProgressText(UINT ids,...);
static LRESULT CALLBACK _WndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam);
void _PTN_SearchProgress(void);
void _PTN_AsyncProgress(int nPercentComplete, DWORD cAsync);
void _PTN_AsyncToSync(void);
void _PTN_SearchComplete(HRESULT hr, BOOL fAbort);
void _OnChangeNotify(LONG code, LPITEMIDLIST *ppidl);
void _DeferHandleUpdateDir(LPCITEMIDLIST pidl, BOOL bRecurse);
void _ClearDeferUpdateDirList();
void _ClearItemDPA(HDPA hdpa);
HRESULT _SetLastError(HRESULT hr);
void _SearchResultsCLSID(CLSID *pclsid) { *pclsid = _clsidResults; };
IUnknown* _GetObjectToPersist();
HRESULT _ForcedUnadvise(void);
void _PostMessage(UINT uMsg, WPARAM wParam, LPARAM lParam);
HRESULT _GetShellView(REFIID riid, void **ppv);
BOOL _FixPersistHandler(LPCTSTR pszBase, LPCTSTR pszDefaultHandler);
void _ProcessTypes(const TYPE_FIX_ENTRY *ptfeTypes, UINT cTypes, TCHAR *pszClass);
void _FixBrokenTypes(void);
// These are the things that the second thread will use during it's processing...
struct {
CRITICAL_SECTION csSearch;
HWND hwndThreadNotify;
HDPA hdpa;
DWORD dwTimeLastNotify;
BOOL fFilesAdded : 1;
BOOL fDirChanged : 1;
BOOL fUpdatePosted : 1;
} _updateParams; // Pass callback params through this object to avoid alloc/free cycle
struct {
IShellFolder *psf;
IShellFolderView *psfv;
IFindFolder *pff;
TCHAR szProgressText[MAX_PATH];
} _execData;
private:
LONG _cRef;
HDSA _hdsaConstraints;
DWORD _cExecInProgress;
BOOL _fAsyncNotifyReceived;
BOOL _fDeferRestore;
BOOL _fDeferRestoreTried;
BOOL _fContinue;
BOOL _fNew;
CConnectionPoint _cpEvents;
OLEDBSimpleProviderListener *_pListener;
HDPA _hdpaItemsToAdd1;
HDPA _hdpaItemsToAdd2;
TCHAR _szProgressText[MAX_PATH+40]; // progress text leave room for chars...
LPITEMIDLIST _pidlUpdate; // Are we processing an updatedir?
LPITEMIDLIST _pidlRestore; // pidl to do restore from...
struct DEFER_UPDATE_DIR *_pdudFirst; // Do we have any defered update dirs?
HRESULT _hrLastError; // the last error reported.
UINT _uStatusMsgIndex; // Files or computers found...
CRITICAL_SECTION _csThread;
DFBSAVEINFO _dfbsi;
CLSID _clsidResults;
};
class CFindConstraint: public DFConstraint, public CImpIDispatch
{
public:
// IUnknown
STDMETHOD(QueryInterface) (REFIID riid, void **ppv);
STDMETHOD_(ULONG, AddRef)(void);
STDMETHOD_(ULONG, Release)(void);
// IDispatch
STDMETHOD(GetTypeInfoCount)(UINT * pctinfo);
STDMETHOD(GetTypeInfo)(UINT itinfo, LCID lcid, ITypeInfo **pptinfo);
STDMETHOD(GetIDsOfNames)(REFIID riid, OLECHAR **rgszNames, UINT cNames, LCID lcid, DISPID * rgdispid);
STDMETHOD(Invoke)(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS * pdispparams, VARIANT * pvarResult, EXCEPINFO * pexcepinfo, UINT * puArgErr);
// DFConstraint
STDMETHOD(get_Name)(BSTR *pbs);
STDMETHOD(get_Value)(VARIANT *pvar);
CFindConstraint(BSTR bstr, VARIANT var);
private:
~CFindConstraint();
LONG _cRef;
BSTR _bstr;
VARIANT _var;
};
CFindCmd::CFindCmd() : CImpIDispatch(LIBID_Shell32, 1, 0, IID_ISearchCommandExt), CSimpleData(&_pListener)
{
_cRef = 1;
_fAsyncNotifyReceived = 0;
_fContinue = TRUE;
ASSERT(NULL == _pidlRestore);
ASSERT(_cExecInProgress == 0);
InitializeCriticalSection(&_updateParams.csSearch);
InitializeCriticalSection(&_csThread);
_clsidResults = CLSID_DocFindFolder; // default
_cpEvents.SetOwner(SAFECAST(this, ISearchCommandExt *), &DIID_DSearchCommandEvents);
}
HRESULT CFindCmd::Init(void)
{
_hdsaConstraints = DSA_Create(sizeof(CMD_CONSTRAINT), 4);
if (!_hdsaConstraints)
return E_OUTOFMEMORY;
return S_OK;
}
CFindCmd::~CFindCmd()
{
if (_updateParams.hwndThreadNotify)
{
// make sure no outstanding fsnotifies registered.
SHChangeNotifyDeregisterWindow(_updateParams.hwndThreadNotify);
DestroyWindow(_updateParams.hwndThreadNotify);
}
_ClearConstraints();
DSA_Destroy(_hdsaConstraints);
_ExecData_Release();
DeleteCriticalSection(&_updateParams.csSearch);
DeleteCriticalSection(&_csThread);
// Make sure we have removed all outstanding update dirs...
_ClearDeferUpdateDirList();
if (_hdpaItemsToAdd1)
DPA_Destroy(_hdpaItemsToAdd1);
if (_hdpaItemsToAdd2)
DPA_Destroy(_hdpaItemsToAdd2);
ILFree(_pidlRestore);
}
STDMETHODIMP CFindCmd::QueryInterface(REFIID riid, void **ppv)
{
static const QITAB qit[] = {
QITABENT(CFindCmd, ISearchCommandExt),
QITABENTMULTI(CFindCmd, IDispatch, ISearchCommandExt),
QITABENT(CFindCmd, IProvideClassInfo2),
QITABENTMULTI(CFindCmd, IProvideClassInfo,IProvideClassInfo2),
QITABENT(CFindCmd, IObjectWithSite),
QITABENT(CFindCmd, IObjectSafety),
QITABENT(CFindCmd, IConnectionPointContainer),
QITABENT(CFindCmd, OLEDBSimpleProvider),
QITABENT(CFindCmd, IRowsetWatchNotify),
QITABENT(CFindCmd, IFindControllerNotify),
{ 0 },
};
return QISearch(this, qit, riid, ppv);
}
STDMETHODIMP_(ULONG) CFindCmd::AddRef()
{
return InterlockedIncrement(&_cRef);
}
STDMETHODIMP_(ULONG) CFindCmd::Release()
{
if (InterlockedDecrement(&_cRef))
return _cRef;
delete this;
return 0;
}
// IDispatch implementation
STDMETHODIMP CFindCmd::GetTypeInfoCount(UINT * pctinfo)
{
return CImpIDispatch::GetTypeInfoCount(pctinfo);
}
STDMETHODIMP CFindCmd::GetTypeInfo(UINT itinfo, LCID lcid, ITypeInfo **pptinfo)
{
return CImpIDispatch::GetTypeInfo(itinfo, lcid, pptinfo);
}
STDMETHODIMP CFindCmd::GetIDsOfNames(REFIID riid, OLECHAR **rgszNames, UINT cNames, LCID lcid, DISPID * rgdispid)
{
return CImpIDispatch::GetIDsOfNames(riid, rgszNames, cNames, lcid, rgdispid);
}
STDMETHODIMP CFindCmd::Invoke(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS * pdispparams, VARIANT * pvarResult, EXCEPINFO * pexcepinfo, UINT * puArgErr)
{
return CImpIDispatch::Invoke(dispidMember, riid, lcid, wFlags, pdispparams, pvarResult, pexcepinfo, puArgErr);
}
// ADOCommand implementation, dual interface method callable via script
STDMETHODIMP CFindCmd::AddConstraint(BSTR bstrName, VARIANT vValue)
{
HRESULT hr = E_OUTOFMEMORY;
CMD_CONSTRAINT dfcc = {0};
dfcc.bstrName = SysAllocString(bstrName);
if (dfcc.bstrName)
{
hr = VariantCopy(&dfcc.vValue, &vValue);
if (SUCCEEDED(hr))
{
if (DSA_ERR == DSA_InsertItem(_hdsaConstraints, DSA_APPEND, &dfcc))
{
SysFreeString(dfcc.bstrName);
VariantClear(&dfcc.vValue);
hr = E_OUTOFMEMORY;
}
}
else
{
SysFreeString(dfcc.bstrName);
}
}
return hr;
}
STDMETHODIMP CFindCmd::GetNextConstraint(VARIANT_BOOL fReset, DFConstraint **ppdfc)
{
*ppdfc = NULL;
IFindFilter *pfilter;
HRESULT hr = _execData.pff->GetFindFilter(&pfilter);
if (SUCCEEDED(hr))
{
BSTR bName;
VARIANT var;
VARIANT_BOOL fFound;
hr = pfilter->GetNextConstraint(fReset, &bName, &var, &fFound);
if (SUCCEEDED(hr))
{
if (!fFound)
{
// need a simple way to signal end list, how about an empty name string?
bName = SysAllocString(L"");
}
CFindConstraint *pdfc = new CFindConstraint(bName, var);
if (pdfc)
{
hr = pdfc->QueryInterface(IID_PPV_ARG(DFConstraint, ppdfc));
pdfc->Release();
}
else
{
// error release stuff we allocated.
hr = E_OUTOFMEMORY;
SysFreeString(bName);
VariantClear(&var);
}
}
pfilter->Release();
}
return hr;
}
HRESULT CFindCmd::_UpdateFilter(IFindFilter *pfilter)
{
HRESULT hr = S_OK;
pfilter->ResetFieldsToDefaults();
int cNumParams = DSA_GetItemCount(_hdsaConstraints);
for (int iItem = 0; iItem < cNumParams; iItem++)
{
CMD_CONSTRAINT *pdfcc = (CMD_CONSTRAINT *)DSA_GetItemPtr(_hdsaConstraints, iItem);
if (pdfcc)
{
hr = pfilter->UpdateField(pdfcc->bstrName, pdfcc->vValue);
}
}
// And clear out the constraint list...
_ClearConstraints();
return hr;
}
void CFindCmd::_ClearConstraints()
{
int cNumParams = DSA_GetItemCount(_hdsaConstraints);
for (int iItem = 0; iItem < cNumParams; iItem++)
{
CMD_CONSTRAINT *pdfcc = (CMD_CONSTRAINT *)DSA_GetItemPtr(_hdsaConstraints, iItem);
if (pdfcc)
{
SysFreeString(pdfcc->bstrName);
VariantClear(&pdfcc->vValue);
}
}
DSA_DeleteAllItems(_hdsaConstraints);
}
void cdecl CFindCmd::_NotifyProgressText(UINT ids,...)
{
va_list ArgList;
va_start(ArgList, ids);
LPTSTR psz = _ConstructMessageString(HINST_THISDLL, MAKEINTRESOURCE(ids), &ArgList);
va_end(ArgList);
if (psz)
{
LPTSTR pszDst = &_szProgressText[0];
StrCpyN(pszDst, psz, ARRAYSIZE(_szProgressText)-2);
LocalFree(psz);
}
else
{
_szProgressText[0] = 0;
}
_cpEvents.InvokeDispid(DISPID_SEARCHCOMMAND_PROGRESSTEXT);
}
STDAPI CDocFindCommand_CreateInstance(IUnknown *punkOuter, REFIID riid, void **ppv)
{
*ppv = NULL;
HRESULT hr;
CFindCmd *pfc = new CFindCmd();
if (pfc)
{
hr = pfc->Init();
if (SUCCEEDED(hr))
hr = pfc->QueryInterface(riid, ppv);
pfc->Release();
}
else
hr = E_OUTOFMEMORY;
return hr;
}
void CFindCmd::_PTN_SearchProgress(void)
{
HRESULT hr = S_OK;
HDPA hdpa = _updateParams.hdpa;
if (hdpa)
{
// Ok lets swap things out from under other thread so that we can process it and still
// let the other thread run...
EnterCriticalSection(&_updateParams.csSearch);
if (_updateParams.hdpa == _hdpaItemsToAdd2)
_updateParams.hdpa = _hdpaItemsToAdd1;
else
_updateParams.hdpa = _hdpaItemsToAdd2;
// say that we don't have any thing here such that other thread will reset up...
_updateParams.fFilesAdded = FALSE;
BOOL fDirChanged = _updateParams.fDirChanged;
_updateParams.fDirChanged = FALSE;
LeaveCriticalSection(&_updateParams.csSearch);
int cItemsToAdd = DPA_GetPtrCount(hdpa);
if (!_execData.pff)
return;
int iItem;
_execData.pff->GetItemCount(&iItem);
int iItemStart = iItem + 1; // needed for notifies 1 based.
if (cItemsToAdd)
{
if (_fContinue)
{
// Are we in an updatedir? If so then need to do merge, else...
if (_pidlUpdate)
{
// see if items in list, already if so we unmark the item
// for delete else if not there maybe add it...
int cItems = iItem;
for (int i = 0; i < cItemsToAdd; i++)
{
LPITEMIDLIST pidl = (LPITEMIDLIST)DPA_FastGetPtr(hdpa, i);
FIND_ITEM *pfi;
for (int j = cItems - 1; j >= 0; j--)
{
_execData.pff->GetItem(j, &pfi);
if (pfi && (_execData.pff->GetFolderIndex(pidl) == _execData.pff->GetFolderIndex(&pfi->idl)))
{
if (_execData.psf->CompareIDs(0, pidl, &pfi->idl) == 0)
break;
}
}
if (j == -1)
{
// Not already in the list so add it...
hr = _execData.pff->AddPidl(iItem, pidl, -1, NULL);
if (SUCCEEDED(hr))
iItem++;
}
else
{
// Item still there - remove possible delete flag...
if (pfi)
pfi->dwState &= ~CDFITEM_STATE_MAYBEDELETE;
}
ILFree(pidl); // The AddPidl does a clone of the pidl...
}
if (iItem && _execData.psfv)
{
hr = _execData.psfv->SetObjectCount(iItem, SFVSOC_NOSCROLL);
}
}
else
{
if (_pListener)
_pListener->aboutToInsertRows(iItemStart, cItemsToAdd);
for (int i = 0; i < cItemsToAdd; i++)
{
LPITEMIDLIST pidl = (LPITEMIDLIST)DPA_FastGetPtr(hdpa, i);
hr = _execData.pff->AddPidl(iItem, pidl, -1, NULL);
if (SUCCEEDED(hr))
iItem++;
ILFree(pidl); // AddPidl makes a copy
}
if (iItem >= iItemStart)
{
if (_execData.psfv)
hr = _execData.psfv->SetObjectCount(iItem, SFVSOC_NOSCROLL);
_execData.pff->SetItemsChangedSinceSort();
_cpEvents.InvokeDispid(DISPID_SEARCHCOMMAND_UPDATE);
}
if (_pListener)
{
_pListener->insertedRows(iItemStart, cItemsToAdd);
_pListener->rowsAvailable(iItemStart, cItemsToAdd);
}
}
}
else // _fContinue
{
for (int i = 0; i < cItemsToAdd; i++)
{
ILFree((LPITEMIDLIST)DPA_FastGetPtr(hdpa, i));
}
}
DPA_DeleteAllPtrs(hdpa);
}
if (fDirChanged)
{
_NotifyProgressText(IDS_SEARCHING, _execData.szProgressText);
}
}
_updateParams.dwTimeLastNotify = GetTickCount();
_updateParams.fUpdatePosted = FALSE;
}
void CFindCmd::_PTN_AsyncProgress(int nPercentComplete, DWORD cAsync)
{
if (!_execData.pff)
return;
// Async case try just setting the count...
_execData.pff->SetAsyncCount(cAsync);
if (_execData.psfv)
{
// -1 for the first item means verify visible items only
_execData.pff->ValidateItems(_execData.psfv, -1, -1, FALSE);
_execData.psfv->SetObjectCount(cAsync, SFVSOC_NOSCROLL);
}
_execData.pff->SetItemsChangedSinceSort();
_cpEvents.InvokeDispid(DISPID_SEARCHCOMMAND_UPDATE);
_NotifyProgressText(IDS_SEARCHINGASYNC, cAsync, nPercentComplete);
}
void CFindCmd::_PTN_AsyncToSync()
{
if (_execData.pff)
_execData.pff->CacheAllAsyncItems();
}
void CFindCmd::_ClearItemDPA(HDPA hdpa)
{
if (hdpa)
{
EnterCriticalSection(&_updateParams.csSearch);
int cItems = DPA_GetPtrCount(hdpa);
for (int i = 0; i < cItems; i++)
{
ILFree((LPITEMIDLIST)DPA_GetPtr(hdpa, i));
}
DPA_DeleteAllPtrs(hdpa);
LeaveCriticalSection(&_updateParams.csSearch);
}
}
void CFindCmd::_PTN_SearchComplete(HRESULT hr, BOOL fAbort)
{
int iItem;
// someone clicked on new button -- cannot set no files found text in listview
// because we'll overwrite enter search criteria to begin
if (!_fNew)
_SetEmptyText(IDS_FINDVIEWEMPTY);
_SetLastError(hr);
// _execData.pff is NULL when Searh is complete by navigating away from the search page
if (!_execData.pff)
{
// do clean up of hdpaToItemsToadd1 and 2
// make sure all items in buffer 1 and 2 are empty
_ClearItemDPA(_hdpaItemsToAdd1);
_ClearItemDPA(_hdpaItemsToAdd2);
}
else
{
// if we have a _pidlUpdate are completing an update
if (_pidlUpdate)
{
int i, cPidf;
UINT uItem;
_execData.pff->GetItemCount(&i);
for (; i-- > 0;)
{
// Pidl at start of structure...
FIND_ITEM *pfi;
_execData.pff->GetItem(i, &pfi);
if (pfi->dwState & CDFITEM_STATE_MAYBEDELETE)
{
_execData.psfv->RemoveObject(&pfi->idl, &uItem);
}
}
ILFree(_pidlUpdate);
_pidlUpdate = NULL;
// clear the update dir flags
_execData.pff->GetFolderListItemCount(&cPidf);
for (i = 0; i < cPidf; i++)
{
FIND_FOLDER_ITEM *pdffli;
if (SUCCEEDED(_execData.pff->GetFolderListItem(i, &pdffli)))
pdffli->fUpdateDir = FALSE;
}
}
// Release our reference count on the searching.
if (_cExecInProgress)
_cExecInProgress--;
// Tell everyone the final count and that we are done...
// But first check if there are any cached up Updatedirs to be processed...
if (_pdudFirst)
{
// first unlink the first one...
struct DEFER_UPDATE_DIR *pdud = _pdudFirst;
_pdudFirst = pdud->pdudNext;
if (_execData.pff->HandleUpdateDir(pdud->pidl, pdud->fRecurse))
{
// Need to spawn sub-search on this...
_Start(FALSE, -1, pdud->pidl);
}
ILFree(pdud->pidl);
LocalFree((HLOCAL)pdud);
}
else
{
if (_execData.psfv)
{
// validate all the items we pulled in already
_execData.pff->ValidateItems(_execData.psfv, 0, -1, TRUE);
}
_execData.pff->GetItemCount(&iItem);
_NotifyProgressText(_uStatusMsgIndex, iItem);
if (!fAbort)
_SelectResults();
}
}
// weird connection point corruption can happen here. somehow the number of sinks is 0 but
// some of the array entries are non null thus causing fault. this problem does not want to
// repro w/ manual testing or debug binaries, only sometimes after an automation run. when
// it happens it is too late to figure out what happened so just patch it here.
if (_cpEvents._HasSinks())
_cpEvents.InvokeDispid(fAbort ? DISPID_SEARCHCOMMAND_ABORT : DISPID_SEARCHCOMMAND_COMPLETE);
}
// see if we need to restart the search based on an update dir
BOOL ShouldRestartSearch(LPCITEMIDLIST pidl)
{
BOOL fRestart = TRUE; // assume we should, non file system pidls
WCHAR szPath[MAX_PATH];
if (SHGetPathFromIDList(pidl, szPath))
{
// Check if this is either a network drive or a remote drive:
if (PathIsRemote(szPath))
{
// If we can find the CI catalogs for the drive on the other machine, then we do
// not want to search.
WCHAR wszCatalog[MAX_PATH], wszMachine[32];
ULONG cchCatalog = ARRAYSIZE(wszCatalog), cchMachine = ARRAYSIZE(wszMachine);
fRestart = (S_OK != LocateCatalogsW(szPath, 0, wszMachine, &cchMachine, wszCatalog, &cchCatalog));
}
else if (-1 != PathGetDriveNumber(szPath))
{
// It is a local dirve...
// Is this machine running the content indexer (CI)?
BOOL fCiRunning, fCiIndexed, fCiPermission;
GetCIStatus(&fCiRunning, &fCiIndexed, &fCiPermission);
fRestart = !fCiRunning || !fCiIndexed; // restart if not running or not fully indexed
}
}
return fRestart;
}
void CFindCmd::_OnChangeNotify(LONG code, LPITEMIDLIST *ppidl)
{
LPITEMIDLIST pidlT;
UINT idsMsg;
UINT cItems;
if (!_execData.pff)
{
_ExecData_Init();
// If we are running async then for now ignore notifications...
// Unless we have cached all of the items...
if (!_execData.pff)
return; // we do not have anything to listen...
}
// see if we want to process the notificiation or not.
switch (code)
{
case SHCNE_RENAMEFOLDER: // With trashcan this is what we see...
case SHCNE_RENAMEITEM: // With trashcan this is what we see...
case SHCNE_DELETE:
case SHCNE_RMDIR:
case SHCNE_UPDATEITEM:
break;
case SHCNE_CREATE:
case SHCNE_MKDIR:
// Process this one out of place
_execData.pff->UpdateOrMaybeAddPidl(_execData.psfv, *ppidl, NULL);
break;
case SHCNE_UPDATEDIR:
TraceMsg(TF_DOCFIND, "DocFind got notify SHCNE_UPDATEDIR, pidl=0x%X",*ppidl);
if (ShouldRestartSearch(*ppidl))
{
BOOL bRecurse = (ppidl[1] != NULL);
if (_cExecInProgress)
{
_DeferHandleUpdateDir(*ppidl, bRecurse);
}
else
{
if (_execData.pff->HandleUpdateDir(*ppidl, bRecurse))
{
// Need to spawn sub-search on this...
_Start(FALSE, -1, *ppidl);
}
}
}
return;
default:
return; // we are not interested in this event
}
//
// Now we need to see if the item might be in our list
// First we need to extract off the last part of the id list
// and see if the contained id entry is in our list. If so we
// need to see if can get the defview find the item and update it.
//
_execData.pff->MapToSearchIDList(*ppidl, FALSE, &pidlT);
switch (code)
{
case SHCNE_RMDIR:
TraceMsg(TF_DOCFIND, "DocFind got notify SHCNE_RMDIR, pidl=0x%X",*ppidl);
_execData.pff->HandleRMDir(_execData.psfv, *ppidl);
if (pidlT)
{
_execData.psfv->RemoveObject(pidlT, &idsMsg);
}
break;
case SHCNE_DELETE:
TraceMsg(TF_DOCFIND, "DocFind got notify SHCNE_DELETE, pidl=0x%X",*ppidl);
if (pidlT)
{
_execData.psfv->RemoveObject(pidlT, &idsMsg);
}
break;
case SHCNE_RENAMEFOLDER:
case SHCNE_RENAMEITEM:
if (pidlT)
{
// If the two items dont have the same parent, we will go ahead
// and remove it...
LPITEMIDLIST pidl1;
if (SUCCEEDED(_execData.pff->GetParentsPIDL(pidlT, &pidl1)))
{
LPITEMIDLIST pidl2 = ILClone(ppidl[1]);
if (pidl2)
{
ILRemoveLastID(pidl2);
if (!ILIsEqual(pidl1, pidl2))
{
_execData.psfv->RemoveObject(pidlT, &idsMsg);
// And maybe add it back to the end... of the list
_execData.pff->UpdateOrMaybeAddPidl(_execData.psfv, ppidl[1], NULL);
}
else
{
// The object is in same folder so must be rename...
// And maybe add it back to the end... of the list
_execData.pff->UpdateOrMaybeAddPidl(_execData.psfv, ppidl[1], pidlT);
}
ILFree(pidl2);
}
ILFree(pidl1);
}
}
else
{
_execData.pff->UpdateOrMaybeAddPidl(_execData.psfv, ppidl[1], NULL);
}
break;
case SHCNE_UPDATEITEM:
TraceMsg(TF_DOCFIND, "DocFind got notify SHCNE_UPDATEITEM, pidl=0x%X",*ppidl);
if (pidlT)
_execData.pff->UpdateOrMaybeAddPidl(_execData.psfv, *ppidl, pidlT);
break;
}
// Update the count...
_execData.psfv->GetObjectCount(&cItems);
_NotifyProgressText(_uStatusMsgIndex, cItems);
ILFree(pidlT);
}
// Ok we need to add a defer
void CFindCmd::_DeferHandleUpdateDir(LPCITEMIDLIST pidl, BOOL bRecurse)
{
// See if we already have some items in the list which are lower down in the tree if so we
// can replace it. Or is there one that is higher up, in which case we can ignore it...
struct DEFER_UPDATE_DIR *pdudPrev = NULL;
struct DEFER_UPDATE_DIR *pdud = _pdudFirst;
while (pdud)
{
if (ILIsParent(pdud->pidl, pidl, FALSE))
return; // Already one in the list that will handle this one...
if (ILIsParent(pidl, pdud->pidl, FALSE))
break;
pdudPrev = pdud;
pdud = pdud->pdudNext;
}
// See if we found one that we can replace...
if (pdud)
{
LPITEMIDLIST pidlT = ILClone(pidl);
if (pidlT)
{
ILFree(pdud->pidl);
pdud->pidl = pidlT;
// See if there are others...
pdudPrev = pdud;
pdud = pdud->pdudNext;
while (pdud)
{
if (ILIsParent(pidl, pdud->pidl, FALSE))
{
// Yep lets trash this one.
ILFree(pdud->pidl);
pdudPrev->pdudNext = pdud->pdudNext;
pdud = pdudPrev; // Let it fall through to setup to look at next...
}
pdudPrev = pdud;
pdud = pdud->pdudNext;
}
}
}
else
{
// Nope simply add us in to the start of the list.
pdud = (struct DEFER_UPDATE_DIR*)LocalAlloc(LPTR, sizeof(struct DEFER_UPDATE_DIR));
if (!pdud)
return; // Ooop could not alloc...
pdud->pidl = ILClone(pidl);
if (!pdud->pidl)
{
LocalFree((HLOCAL)pdud);
return;
}
pdud->fRecurse = bRecurse;
pdud->pdudNext = _pdudFirst;
_pdudFirst = pdud;
}
}
void CFindCmd::_ClearDeferUpdateDirList()
{
// Cancel any Pending updatedirs also.
while (_pdudFirst)
{
struct DEFER_UPDATE_DIR *pdud = _pdudFirst;
_pdudFirst = pdud->pdudNext;
ILFree(pdud->pidl);
LocalFree((HLOCAL)pdud);
}
}
LRESULT CALLBACK CFindCmd::_WndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
CFindCmd* pThis = (CFindCmd*)GetWindowLongPtr(hwnd, 0);
LRESULT lRes = 0;
switch (uMsg)
{
case WM_DESTROY:
SetWindowLong(hwnd, 0, 0); // make sure we don't deref pThis
break;
case WM_DF_FSNOTIFY:
{
LPITEMIDLIST *ppidl;
LONG lEvent;
LPSHChangeNotificationLock pshcnl = SHChangeNotification_Lock((HANDLE)wParam, (DWORD)lParam, &ppidl, &lEvent);
if (pshcnl)
{
if (pThis)
pThis->_OnChangeNotify(lEvent, ppidl);
SHChangeNotification_Unlock(pshcnl);
}
}
break;
case WM_DF_SEARCHPROGRESS:
pThis->_PTN_SearchProgress();
pThis->Release();
break;
case WM_DF_ASYNCPROGRESS:
pThis->_PTN_AsyncProgress((int)wParam, (DWORD)lParam);
pThis->Release();
break;
case WM_DF_SEARCHSTART:
pThis->_cpEvents.InvokeDispid(DISPID_SEARCHCOMMAND_START);
pThis->_SetEmptyText(IDS_FINDVIEWEMPTYBUSY);
pThis->Release();
break;
case WM_DF_SEARCHCOMPLETE:
pThis->_PTN_SearchComplete((HRESULT)wParam, (BOOL)lParam);
pThis->Release();
break;
default:
lRes = ::DefWindowProc(hwnd, uMsg, wParam, lParam);
break;
}
return lRes;
}
// test to see if the view is in a mode where many items are displayed
BOOL LotsOfItemsInView(IUnknown *punkSite)
{
BOOL bLotsOfItemsInView = FALSE;
IFolderView * pfv;
HRESULT hr = IUnknown_QueryService(punkSite, SID_SFolderView, IID_PPV_ARG(IFolderView, &pfv));
if (SUCCEEDED(hr))
{
UINT uViewMode;
bLotsOfItemsInView = SUCCEEDED(pfv->GetCurrentViewMode(&uViewMode)) &&
((FVM_ICON == uViewMode) || (FVM_SMALLICON == uViewMode));
pfv->Release();
}
return bLotsOfItemsInView;
}
void CFindCmd::_DoSearch(IFindEnum *penum)
{
BOOL fAbort = FALSE;
BOOL bLotsOfItems = LotsOfItemsInView(_execData.psfv);
EnterCriticalSection(&_csThread);
// previous thread might have exited but we're still processing search complete message
if (_cExecInProgress > 1)
Sleep(1000); // give it a chance to finish
_updateParams.hdpa = NULL;
_updateParams.fFilesAdded = FALSE;
_updateParams.fDirChanged = FALSE;
_updateParams.fUpdatePosted = FALSE;
_PostMessage(WM_DF_SEARCHSTART, 0, 0);
// Now see if this is an Sync or an Async version of the search...
HRESULT hr = S_OK;
BOOL fQueryIsAsync = penum->FQueryIsAsync();
if (fQueryIsAsync)
{
DBCOUNTITEM dwTotalAsync;
BOOL fDone;
int nPercentComplete;
while (S_OK == (hr = penum->GetAsyncCount(&dwTotalAsync, &nPercentComplete, &fDone)))
{
if (!_fContinue)
{
fAbort = TRUE;
break;
}
_PostMessage(WM_DF_ASYNCPROGRESS, (WPARAM)nPercentComplete, (LPARAM)dwTotalAsync);
// If we are done we can simply let the ending callback tell of the new count...
if (fDone)
break;
// sleep .3 or 1.5 sec
Sleep(bLotsOfItems ? 1500 : 300); // wait between looking again...
}
}
if (!fQueryIsAsync || (fQueryIsAsync == DF_QUERYISMIXED))
{
int state, cItemsSearched = 0, cFoldersSearched = 0, cFoldersSearchedPrev = 0;
_updateParams.hdpa = _hdpaItemsToAdd1; // Assume first one now...
_updateParams.dwTimeLastNotify = GetTickCount();
LPITEMIDLIST pidl;
while (S_OK == (hr = penum->Next(&pidl, &cItemsSearched, &cFoldersSearched, &_fContinue, &state)))
{
if (state == GNF_DONE)
break; // no more
if (!_fContinue)
{
fAbort = TRUE;
break;
}
// See if we should abort
if (state == GNF_MATCH)
{
EnterCriticalSection(&_updateParams.csSearch);
DPA_AppendPtr(_updateParams.hdpa, pidl);
_updateParams.fFilesAdded = TRUE;
LeaveCriticalSection(&_updateParams.csSearch);
}
if (cFoldersSearchedPrev != cFoldersSearched)
{
_updateParams.fDirChanged = TRUE;
cFoldersSearchedPrev = cFoldersSearched;
}
if (!_updateParams.fUpdatePosted &&
(_updateParams.fDirChanged || _updateParams.fFilesAdded))
{
if ((GetTickCount() - _updateParams.dwTimeLastNotify) > 200)
{
_updateParams.fUpdatePosted = TRUE;
_PostMessage(WM_DF_SEARCHPROGRESS, 0, 0);
}
}
}
_PostMessage(WM_DF_SEARCHPROGRESS, 0, 0);
}
if (hr != S_OK)
{
fAbort = TRUE;
}
_PostMessage(WM_DF_SEARCHCOMPLETE, (WPARAM)hr, (LPARAM)fAbort);
LeaveCriticalSection(&_csThread);
}
DWORD CALLBACK CFindCmd::_ThreadProc(void *pv)
{
THREAD_PARAMS *pParams = (THREAD_PARAMS *)pv;
pParams->pThis->_DoSearch(pParams->penum);
_FreeThreadParams(pParams);
return 0;
}
HRESULT CFindCmd::_Cancel()
{
_ClearDeferUpdateDirList();
if (DSA_GetItemCount(_hdsaConstraints) == 0)
{
_fContinue = FALSE; // Cancel current query if we have a null paramter collection
return S_OK;
}
return E_FAIL;
}
HRESULT CFindCmd::_Init(THREAD_PARAMS **ppParams, int iCol, LPCITEMIDLIST pidlUpdate)
{
*ppParams = new THREAD_PARAMS;
if (NULL == *ppParams)
return E_OUTOFMEMORY;
// Clear any previous registrations...
SHChangeNotifyDeregisterWindow(_updateParams.hwndThreadNotify);
// Prepare to execute the query
IFindFilter *pfilter;
HRESULT hr = _execData.pff->GetFindFilter(&pfilter);
if (SUCCEEDED(hr))
{
// We do not need to update the filter if this is done as part of an FSNOTIFY or a Sort...
if ((iCol >= 0) || pidlUpdate || SUCCEEDED(hr = _UpdateFilter(pfilter)))
{
_execData.szProgressText[0] = 0;
pfilter->DeclareFSNotifyInterest(_updateParams.hwndThreadNotify, WM_DF_FSNOTIFY);
pfilter->GetStatusMessageIndex(0, &_uStatusMsgIndex);
DWORD dwFlags;
hr = pfilter->PrepareToEnumObjects(_GetWindow(), &dwFlags);
if (SUCCEEDED(hr))
{
hr = pfilter->EnumObjects(_execData.psf, pidlUpdate, dwFlags, iCol,
_execData.szProgressText, SAFECAST(this, IRowsetWatchNotify*), &(*ppParams)->penum);
}
}
pfilter->Release();
}
// Fill in the exec params
(*ppParams)->pThis = this;
AddRef(); // ExecParams_Free will release this interface addref...
if (FAILED(hr))
{
_FreeThreadParams(*ppParams);
*ppParams = NULL;
}
return hr;
}
HRESULT CFindCmd::_FreeThreadParams(THREAD_PARAMS *pParams)
{
if (!pParams)
return S_OK;
// Don't use atomic release as this a pointer to a class not an interface.
CFindCmd *pThis = pParams->pThis;
pParams->pThis = NULL;
pThis->Release();
ATOMICRELEASE(pParams->penum);
delete pParams;
return S_OK;
}
HRESULT CFindCmd::_ExecData_Release()
{
ATOMICRELEASE(_execData.psf);
ATOMICRELEASE(_execData.psfv);
if (_execData.pff)
_execData.pff->SetControllerNotifyObject(NULL); // release back pointer to us...
ATOMICRELEASE(_execData.pff);
_cExecInProgress = 0; // we must be in process of shutting down at least...
return S_OK;
}
HRESULT CFindCmd::_EnsureResultsViewIsCurrent(IUnknown *punk)
{
HRESULT hr = E_FAIL;
LPITEMIDLIST pidlFolder;
if (S_OK == SHGetIDListFromUnk(punk, &pidlFolder))
{
LPITEMIDLIST pidl;
if (SUCCEEDED(_GetSearchIDList(&pidl)))
{
if (ILIsEqual(pidlFolder, pidl))
hr = S_OK;
ILFree(pidl);
}
ILFree(pidlFolder);
}
return hr;
}
// the search results view callback proffeerd itself and we can use that
// to get a hold of defview and can program it
HRESULT CFindCmd::_GetShellView(REFIID riid, void **ppv)
{
return IUnknown_QueryService(_punkSite, SID_DocFindFolder, riid, ppv);
}
HRESULT CFindCmd::_ExecData_Init()
{
_ExecData_Release();
IFolderView *pfv;
HRESULT hr = _GetShellView(IID_PPV_ARG(IFolderView, &pfv));
if (SUCCEEDED(hr))
{
IShellFolder *psf;
hr = pfv->GetFolder(IID_PPV_ARG(IShellFolder, &psf));
if (SUCCEEDED(hr))
{
IFindFolder *pff;
hr = psf->QueryInterface(IID_PPV_ARG(IFindFolder, &pff));
if (SUCCEEDED(hr))
{
hr = _EnsureResultsViewIsCurrent(psf);
if (SUCCEEDED(hr))
{
IShellFolderView *psfv;
hr = pfv->QueryInterface(IID_PPV_ARG(IShellFolderView, &psfv));
if (SUCCEEDED(hr))
{
IUnknown_Set((IUnknown **)&_execData.pff, pff);
IUnknown_Set((IUnknown **)&_execData.psf, psf);
IUnknown_Set((IUnknown **)&_execData.psfv, psfv);
_execData.pff->SetControllerNotifyObject(SAFECAST(this, IFindControllerNotify*));
psfv->Release();
}
}
pff->Release();
}
psf->Release();
}
pfv->Release();
}
if (FAILED(hr))
_ExecData_Release();
else
SetShellFolder(_execData.psf);
return hr;
}
BOOL CFindCmd::_SetupBrowserCP()
{
if (!_dwCookie)
{
_cwbe.SetOwner(this); // make sure our owner is set...
// register ourself with the Defview to get any events that they may generate...
IServiceProvider *pspTLB;
HRESULT hr = IUnknown_QueryService(_punkSite, SID_STopLevelBrowser, IID_PPV_ARG(IServiceProvider, &pspTLB));
if (SUCCEEDED(hr))
{
IConnectionPointContainer *pcpc;
hr = pspTLB->QueryService(IID_IExpDispSupport, IID_PPV_ARG(IConnectionPointContainer, &pcpc));
if (SUCCEEDED(hr))
{
hr = ConnectToConnectionPoint(SAFECAST(&_cwbe, DWebBrowserEvents*), DIID_DWebBrowserEvents2,
TRUE, pcpc, &_dwCookie, &_pcpBrowser);
pcpc->Release();
}
pspTLB->Release();
}
}
if (_dwCookie)
_cwbe.SetWaiting(TRUE);
return _dwCookie ? TRUE : FALSE;
}
HRESULT CFindCmd::_Start(BOOL fNavigateIfFail, int iCol, LPCITEMIDLIST pidlUpdate)
{
if (_cExecInProgress)
return E_UNEXPECTED;
if (!_hdpaItemsToAdd1)
{
_hdpaItemsToAdd1 = DPA_CreateEx(64, GetProcessHeap());
if (!_hdpaItemsToAdd1)
return E_OUTOFMEMORY;
}
if (!_hdpaItemsToAdd2)
{
_hdpaItemsToAdd2 = DPA_CreateEx(64, GetProcessHeap());
if (!_hdpaItemsToAdd2)
return E_OUTOFMEMORY;
}
if (!_updateParams.hwndThreadNotify)
{
_updateParams.hwndThreadNotify = SHCreateWorkerWindow(_WndProc, NULL, 0, 0, 0, this);
if (!_updateParams.hwndThreadNotify)
return E_OUTOFMEMORY;
}
HRESULT hr = _ExecData_Init();
if (FAILED(hr))
{
if (fNavigateIfFail)
{
if (_SetupBrowserCP())
NavigateToSearchResults();
}
// Return S_False so that when we check if this succeeded in finddlg, we wee that it
// did, and therefore let the animation run. If we return a failure code here, we
// will stop the animation. This will only hapen when we are navigating to the search
// results as well as starting the search.
return S_FALSE;
}
THREAD_PARAMS *ptp;
hr = _Init(&ptp, iCol, pidlUpdate);
if (SUCCEEDED(hr))
{
// See if we should be saving away the selection...
if (iCol >= 0)
_execData.pff->RememberSelectedItems();
// If this is an update then we need to remember our IDList else clear list...
if (pidlUpdate)
{
_pidlUpdate = ILClone(pidlUpdate);
}
else
{
_Clear(); // tell defview to delete everything
}
_execData.pff->SetAsyncEnum(ptp->penum);
// Start the query
_cExecInProgress++;
_fContinue = TRUE;
_fNew = FALSE;
if (SHCreateThread(_ThreadProc, ptp, CTF_COINIT, NULL))
{
hr = S_OK;
}
else
{
_cExecInProgress--;
_FreeThreadParams(ptp);
_SetEmptyText(IDS_FINDVIEWEMPTY);
}
}
else
hr = _SetLastError(hr);
return hr;
}
HRESULT CFindCmd::_SetLastError(HRESULT hr)
{
if (HRESULT_FACILITY(hr) == FACILITY_SEARCHCOMMAND)
{
_hrLastError = hr;
hr = S_FALSE; // Don't error out script...
_cpEvents.InvokeDispid(DISPID_SEARCHCOMMAND_ERROR);
}
return hr;
}
STDMETHODIMP CFindCmd::Execute(VARIANT *RecordsAffected, VARIANT *Parameters, long Options)
{
if (Options == 0)
return _Cancel();
_FixBrokenTypes();
return _Start(TRUE, -1, NULL);
}
// IConnectionPointContainer
STDMETHODIMP CFindCmd::EnumConnectionPoints(IEnumConnectionPoints **ppEnum)
{
return CreateInstance_IEnumConnectionPoints(ppEnum, 1, _cpEvents.CastToIConnectionPoint());
}
STDMETHODIMP CFindCmd::FindConnectionPoint(REFIID iid, IConnectionPoint **ppCP)
{
if (IsEqualIID(iid, DIID_DSearchCommandEvents) ||
IsEqualIID(iid, IID_IDispatch))
{
*ppCP = _cpEvents.CastToIConnectionPoint();
}
else
{
*ppCP = NULL;
return E_NOINTERFACE;
}
(*ppCP)->AddRef();
return S_OK;
}
// IProvideClassInfo2 methods
STDMETHODIMP CFindCmd::GetClassInfo(ITypeInfo **ppTI)
{
return GetTypeInfoFromLibId(0, LIBID_Shell32, 1, 0, CLSID_DocFindCommand, ppTI);
}
STDMETHODIMP CFindCmd::GetGUID(DWORD dwGuidKind, GUID *pGUID)
{
if (dwGuidKind == GUIDKIND_DEFAULT_SOURCE_DISP_IID)
{
*pGUID = DIID_DSearchCommandEvents;
return S_OK;
}
*pGUID = GUID_NULL;
return E_FAIL;
}
STDMETHODIMP CFindCmd::SetSite(IUnknown *punkSite)
{
if (!punkSite)
{
if (!_cExecInProgress)
{
_ExecData_Release();
}
_fContinue = FALSE; // Cancel existing queries
// See if we have a connection point... If so unadvise now...
if (_dwCookie)
{
_pcpBrowser->Unadvise(_dwCookie);
ATOMICRELEASE(_pcpBrowser);
_dwCookie = 0;
}
// Bug #199671
// Trident won't call UnAdvise and they except ActiveX Controls
// to use IOleControl::Close() to do their own UnAdvise, and hope
// nobody will need events after that. I don't impl IOleControl so
// we need to do the same thing during IObjectWithSite::SetSite(NULL)
// and hope someone won't want to reparent us. This is awkward but
// saves Trident some perf so we will tolerate it.
EVAL(SUCCEEDED(_cpEvents.UnadviseAll()));
}
return CObjectWithSite::SetSite(punkSite);
}
void CFindCmd::_SelectResults()
{
if (_execData.psfv)
{
// If there are any items...
UINT cItems = 0;
if (SUCCEEDED(_execData.psfv->GetObjectCount(&cItems)) && cItems > 0)
{
IShellView* psv;
if (SUCCEEDED(_execData.psfv->QueryInterface(IID_PPV_ARG(IShellView, &psv))))
{
// If none are selected (don't want to rip the user's selection out of his hand)...
UINT cSelected = 0;
if (SUCCEEDED(_execData.psfv->GetSelectedCount(&cSelected)) && cSelected == 0)
{
// Retrieve the pidl for the first item in the list...
LPITEMIDLIST pidlFirst = NULL;
if (SUCCEEDED(_execData.psfv->GetObject(&pidlFirst, 0)))
{
// Give it the focus
psv->SelectItem(pidlFirst, SVSI_FOCUSED | SVSI_ENSUREVISIBLE);
}
}
// Activate the view.
psv->UIActivate(SVUIA_ACTIVATE_FOCUS);
psv->Release();
}
}
}
}
STDMETHODIMP CFindCmd::ClearResults(void)
{
HRESULT hr = _Clear();
if (SUCCEEDED(hr))
{
_fNew = TRUE;
_SetEmptyText(IDS_FINDVIEWEMPTYINIT);
}
return hr ;
}
HRESULT CFindCmd::_Clear()
{
// Tell defview to delete everything.
if (_execData.psfv)
{
UINT u;
_execData.psfv->RemoveObject(NULL, &u);
}
// And cleanup our folderList
if (_execData.pff)
{
_execData.pff->ClearItemList();
_execData.pff->ClearFolderList();
}
return S_OK;
}
HRESULT CFindCmd::_SetEmptyText(UINT nIDEmptyText)
{
IShellFolderViewCB *psfvcb;
HRESULT hr = IUnknown_QueryService(_execData.psfv, SID_ShellFolderViewCB, IID_PPV_ARG(IShellFolderViewCB, &psfvcb));
if (SUCCEEDED(hr))
{
TCHAR szEmptyText[128];
LoadString(HINST_THISDLL, nIDEmptyText, szEmptyText, ARRAYSIZE(szEmptyText));
hr = psfvcb->MessageSFVCB(SFVM_SETEMPTYTEXT, 0, (LPARAM)szEmptyText);
psfvcb->Release();
}
return hr;
}
HRESULT CFindCmd::_GetSearchIDList(LPITEMIDLIST *ppidl)
{
CLSID clsid;
_SearchResultsCLSID(&clsid);
return ILCreateFromCLSID(clsid, ppidl);
}
STDMETHODIMP CFindCmd::NavigateToSearchResults(void)
{
IShellBrowser *psb;
HRESULT hr = IUnknown_QueryService(_punkSite, SID_STopLevelBrowser, IID_PPV_ARG(IShellBrowser, &psb));
if (SUCCEEDED(hr))
{
LPITEMIDLIST pidl;
hr = _GetSearchIDList(&pidl);
if (SUCCEEDED(hr))
{
hr = psb->BrowseObject(pidl, SBSP_SAMEBROWSER | SBSP_ABSOLUTE | SBSP_WRITENOHISTORY);
ILFree(pidl);
}
psb->Release();
}
return hr;
}
IUnknown* CFindCmd::_GetObjectToPersist()
{
IOleObject *pole = NULL;
IShellView *psv;
HRESULT hr = _GetShellView(IID_PPV_ARG(IShellView, &psv));
if (SUCCEEDED(hr))
{
psv->GetItemObject(SVGIO_BACKGROUND, IID_PPV_ARG(IOleObject, &pole));
psv->Release();
}
return (IUnknown *)pole;
}
void CFindCmd::_PostMessage(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
AddRef(); // to be released after processing of the message bellow
if (!PostMessage(_updateParams.hwndThreadNotify, uMsg, wParam, lParam))
{
Release();
}
}
HWND CFindCmd::_GetWindow()
{
HWND hwnd;
return SUCCEEDED(IUnknown_GetWindow(_punkSite, &hwnd)) ? hwnd : NULL;
}
STDMETHODIMP CFindCmd::SaveSearch(void)
{
IFindFilter *pfilter;
HRESULT hr = _execData.pff->GetFindFilter(&pfilter);
if (SUCCEEDED(hr))
{
IShellView *psv;
hr = _GetShellView(IID_PPV_ARG(IShellView, &psv));
if (SUCCEEDED(hr))
{
IUnknown* punk = _GetObjectToPersist(); // NULL is OK
_execData.pff->Save(pfilter, _GetWindow(), &_dfbsi, psv, punk);
ATOMICRELEASE(punk);
psv->Release();
}
pfilter->Release();
}
return hr;
}
STDMETHODIMP CFindCmd::RestoreSearch(void)
{
// let script know that a restore happened...
_cpEvents.InvokeDispid(DISPID_SEARCHCOMMAND_RESTORE);
return S_OK;
}
STDMETHODIMP CFindCmd::StopSearch(void)
{
if (_cExecInProgress)
return _Cancel();
return S_OK;
}
STDMETHODIMP CFindCmd::GetItemCount(UINT *pcItems)
{
if (_execData.psfv)
{
return _execData.psfv->GetObjectCount(pcItems);
}
return E_FAIL;
}
STDMETHODIMP CFindCmd::SetItemCount(UINT cItems)
{
if (_execData.psfv)
{
return _execData.psfv->SetObjectCount(cItems, SFVSOC_NOSCROLL);
}
return E_FAIL;
}
STDMETHODIMP CFindCmd::ViewDestroyed()
{
_ExecData_Release();
return S_OK;
}
STDMETHODIMP CFindCmd::get_ProgressText(BSTR *pbs)
{
*pbs = SysAllocStringT(_szProgressText);
return *pbs ? S_OK : E_OUTOFMEMORY;
}
//------ error string mappings ------//
static const UINT error_strings[] =
{
SCEE_CONSTRAINT, IDS_DOCFIND_CONSTRAINT,
SCEE_PATHNOTFOUND, IDS_DOCFIND_PATHNOTFOUND,
SCEE_INDEXSEARCH, IDS_DOCFIND_SCOPEERROR,
SCEE_CASESENINDEX, IDS_DOCFIND_CI_NOT_CASE_SEN,
};
STDMETHODIMP CFindCmd::GetErrorInfo(BSTR *pbs, int *phr)
{
int nCode = HRESULT_CODE(_hrLastError);
UINT uSeverity = HRESULT_SEVERITY(_hrLastError);
if (phr)
*phr = nCode;
if (pbs)
{
UINT nIDString = 0;
*pbs = NULL;
for(int i = 0; i < ARRAYSIZE(error_strings); i += 2)
{
if (error_strings[i] == (UINT)nCode)
{
nIDString = error_strings[i+1];
break ;
}
}
if (nIDString)
{
WCHAR wszMsg[MAX_PATH];
EVAL(LoadStringW(HINST_THISDLL, nIDString, wszMsg, ARRAYSIZE(wszMsg)));
*pbs = SysAllocString(wszMsg);
}
else
*pbs = SysAllocString(L"");
}
return S_OK;
}
STDMETHODIMP CFindCmd::SearchFor(int iFor)
{
if (SCE_SEARCHFORFILES == iFor)
{
_clsidResults = CLSID_DocFindFolder;
}
else if (SCE_SEARCHFORCOMPUTERS == iFor)
{
_clsidResults = CLSID_ComputerFindFolder;
}
return S_OK;
}
STDMETHODIMP CFindCmd::GetScopeInfo(BSTR bsScope, int *pdwScopeInfo)
{
*pdwScopeInfo = 0;
return E_NOTIMPL;
}
STDMETHODIMP CFindCmd::RestoreSavedSearch(VARIANT *pvarFile)
{
if (pvarFile && pvarFile->vt != VT_EMPTY)
{
LPITEMIDLIST pidl = VariantToIDList(pvarFile);
if (pidl)
{
ILFree(_pidlRestore);
_pidlRestore = pidl ;
}
}
if (_pidlRestore)
{
IShellView *psv;
HRESULT hr = _GetShellView(IID_PPV_ARG(IShellView, &psv));
if (SUCCEEDED(hr))
{
psv->Release();
if (SUCCEEDED(_ExecData_Init()))
{
_execData.pff->RestoreSearchFromSaveFile(_pidlRestore, _execData.psfv);
_cpEvents.InvokeDispid(DISPID_SEARCHCOMMAND_RESTORE);
ILFree(_pidlRestore);
_pidlRestore = NULL;
}
}
else if (!_fDeferRestoreTried)
{
// appears to be race condition to load
TraceMsg(TF_WARNING, "CFindCmd::MaybeRestoreSearch - _GetShellView failed...");
_fDeferRestore = TRUE;
if (!_SetupBrowserCP())
_fDeferRestore = FALSE;
}
}
return S_OK;
}
STDMETHODIMP CFindCmd::OnChange(IRowset *prowset, DBWATCHNOTIFY eChangeReason)
{
_fAsyncNotifyReceived = TRUE;
return S_OK;
}
STDMETHODIMP CFindCmd::DoSortOnColumn(UINT iCol, BOOL fSameCol)
{
IFindEnum *pdfEnumAsync;
if (S_OK == _execData.pff->GetAsyncEnum(&pdfEnumAsync))
{
// If the search is still running we will restart with the other column else we
// will make sure all of the items have been cached and let the default processing happen
if (!fSameCol && _cExecInProgress)
{
// We should try to sort on the right column...
_Start(FALSE, iCol, NULL);
return S_FALSE; // tell system to not do default processing.
}
_execData.pff->CacheAllAsyncItems();
}
return S_OK; // let it do default processing.
}
// Implemention of our IDispatch to hookup to the top level browsers connnection point...
STDMETHODIMP CFindCmd::CWBEvents2::QueryInterface(REFIID riid, void **ppv)
{
static const QITAB qit[] = {
QITABENTMULTI(CFindCmd::CWBEvents2, IDispatch, DWebBrowserEvents2),
QITABENTMULTI2(CFindCmd::CWBEvents2, DIID_DWebBrowserEvents2, DWebBrowserEvents2),
QITABENTMULTI2(CFindCmd::CWBEvents2, DIID_DWebBrowserEvents, DWebBrowserEvents),
{ 0 },
};
return QISearch(this, qit, riid, ppv);
}
STDMETHODIMP CFindCmd::CWBEvents2::Invoke(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS * pdispparams, VARIANT * pvarResult, EXCEPINFO * pexcepinfo, UINT * puArgErr)
{
if (_fWaitingForNavigate)
{
if ((dispidMember == DISPID_NAVIGATECOMPLETE) ||
(dispidMember == DISPID_DOCUMENTCOMPLETE))
{
// Assume this is ours... Should maybe check parameters...
_fWaitingForNavigate = FALSE;
// Now see if it is a case where we are to restore the search...
if (_pcdfc->_fDeferRestore)
{
_pcdfc->_fDeferRestore = FALSE;
_pcdfc->_fDeferRestoreTried = TRUE;
_pcdfc->RestoreSavedSearch(NULL);
}
else
return _pcdfc->_Start(FALSE, -1, NULL);
}
}
return S_OK;
}
#define MAX_DEFAULT_VALUE 40 // Longest of all of the below pszDefaultValueMatch strings (plus slop)
#define MAX_KEY_PH_NAME 70 // "CLSID\{GUID}\PersistentHandler" (plus slop)
const TYPE_FIX_ENTRY g_tfeTextTypes[] =
{
{ TEXT(".rtf"), NULL, NULL },
};
const TYPE_FIX_ENTRY g_tfeNullTypes[] =
{
{ TEXT(".mdb"), TEXT("Access.Application.10"), TEXT("{73A4C9C1-D68D-11D0-98BF-00A0C90DC8D9}") },
{ TEXT(".msg"), TEXT("msgfile"), NULL },
{ TEXT(".sc2"), TEXT("SchedulePlus.Application.7"), TEXT("{0482E074-C5B7-101A-82E0-08002B36A333}") },
{ TEXT(".wll"), TEXT("Word.Addin.8"), NULL },
};
//
// rtf is listed twice, once above for TextTypes (to fix when office
// un-installed) and once here as an OfficeType (to fix when office
// is re-installed). Uninstalled = TextFilter, Reinstalled = OfficeFilter
//
const TYPE_FIX_ENTRY g_tfeOfficeTypes[] =
{
{ TEXT(".rtf"), TEXT("Word.RTF.8"), TEXT("{00020906-0000-0000-C000-000000000046}") },
{ TEXT(".doc"), TEXT("Word.Document.8"), TEXT("{00020906-0000-0000-C000-000000000046}") },
{ TEXT(".dot"), TEXT("Word.Template.8"), TEXT("{00020906-0000-0000-C000-000000000046}") },
{ TEXT(".pot"), TEXT("PowerPoint.Template.8"), TEXT("{64818D11-4F9B-11CF-86EA-00AA00B929E8}") },
{ TEXT(".pps"), TEXT("PowerPoint.SlideShow.8"), TEXT("{64818D10-4F9B-11CF-86EA-00AA00B929E8}") },
{ TEXT(".ppt"), TEXT("PowerPoint.Show.8"), TEXT("{64818D10-4F9B-11CF-86EA-00AA00B929E8}") },
{ TEXT(".rtf"), TEXT("Word.RTF.8"), TEXT("{00020906-0000-0000-C000-000000000046}") },
{ TEXT(".xlb"), TEXT("Excel.Sheet.8"), TEXT("{00020820-0000-0000-C000-000000000046}") },
{ TEXT(".xlc"), TEXT("Excel.Chart.8"), TEXT("{00020821-0000-0000-C000-000000000046}") },
{ TEXT(".xls"), TEXT("Excel.Sheet.8"), TEXT("{00020820-0000-0000-C000-000000000046}") },
{ TEXT(".xlt"), TEXT("Excel.Template"), TEXT("{00020820-0000-0000-C000-000000000046}") },
};
const TYPE_FIX_ENTRY g_tfeHtmlTypes[] =
{
{ TEXT(".asp"), TEXT("aspfile"), NULL },
{ TEXT(".htx"), TEXT("htxfile"), NULL },
};
BOOL CFindCmd::_FixPersistHandler(LPCTSTR pszBase, LPCTSTR pszDefaultHandler)
{
TCHAR szPHName[MAX_KEY_PH_NAME];
LONG lr;
HKEY hkeyPH;
HKEY hkeyBase;
wnsprintf(szPHName,ARRAYSIZE(szPHName), TEXT("%s\\PersistentHandler"), pszBase);
lr = RegOpenKey(HKEY_CLASSES_ROOT, szPHName, &hkeyPH);
if (lr == ERROR_SUCCESS)
{
// We found an existing PersistHandler key, leave it alone
RegCloseKey(hkeyPH);
return TRUE;
}
lr = RegOpenKey(HKEY_CLASSES_ROOT, pszBase, &hkeyBase);
if (lr != ERROR_SUCCESS)
{
// We didn't find the base key (normally "CLSID\\{GUID}"), get out
return FALSE;
}
RegCloseKey(hkeyBase);
lr = RegCreateKey(HKEY_CLASSES_ROOT, szPHName, &hkeyPH);
if (lr != ERROR_SUCCESS)
{
// We couldn't create the ...\PersistHandler key, get out
return FALSE;
}
// Able to create the ...\PersistHandler key, write out the default handler
lr = RegSetValue(hkeyPH, NULL, REG_SZ, pszDefaultHandler, lstrlen(pszDefaultHandler));
RegCloseKey(hkeyPH);
// Success if write succeeded
return (lr == ERROR_SUCCESS);
}
void CFindCmd::_ProcessTypes(
const TYPE_FIX_ENTRY *ptfeTypes,
UINT cTypes,
TCHAR *pszClass)
{
UINT iType;
LONG lr;
HKEY hkeyType;
for (iType = 0; iType < cTypes; iType++)
{
lr = RegOpenKey(HKEY_CLASSES_ROOT, ptfeTypes[iType].pszDotType, &hkeyType);
if (lr == ERROR_SUCCESS)
{
//
// If it has a default value to match, repair that (if it exists).
// If there is no default value to match, just repair the .foo type
//
if (ptfeTypes[iType].pszDefaultValueMatch)
{
TCHAR szDefaultValue[MAX_DEFAULT_VALUE];
LONG cb = sizeof(szDefaultValue);
lr = RegQueryValue(hkeyType, NULL, szDefaultValue, &cb);
if (lr == ERROR_SUCCESS)
{
if (lstrcmp(szDefaultValue,ptfeTypes[iType].pszDefaultValueMatch) == 0)
{
if (ptfeTypes[iType].pszGuid == NULL)
{
// Fix either the progid or the type, whichever we can
if (!_FixPersistHandler(ptfeTypes[iType].pszDefaultValueMatch,pszClass))
{
_FixPersistHandler(ptfeTypes[iType].pszDotType,pszClass);
}
}
else
{
// Fix the persist handler for the guid, since its specified
TCHAR szPHName[MAX_KEY_PH_NAME];
wnsprintf(szPHName, ARRAYSIZE(szPHName), TEXT("CLSID\\%s"), ptfeTypes[iType].pszGuid);
_FixPersistHandler(szPHName, pszClass);
}
}
}
}
else
{
_FixPersistHandler(ptfeTypes[iType].pszDotType, pszClass);
}
RegCloseKey(hkeyType);
}
else if (lr == ERROR_FILE_NOT_FOUND)
{
//
// .foo doesn't exist - this can happen because of bad un-install program
// Create .foo and .foo\PersistentHandler
//
lr = RegCreateKey(HKEY_CLASSES_ROOT, ptfeTypes[iType].pszDotType, &hkeyType);
if (lr == ERROR_SUCCESS)
{
_FixPersistHandler(ptfeTypes[iType].pszDotType, pszClass);
RegCloseKey(hkeyType);
}
}
}
}
void CFindCmd::_FixBrokenTypes(void)
{
_ProcessTypes(g_tfeNullTypes, ARRAYSIZE(g_tfeNullTypes), TEXT("{098f2470-bae0-11cd-b579-08002b30bfeb}"));
_ProcessTypes(g_tfeTextTypes, ARRAYSIZE(g_tfeTextTypes), TEXT("{5e941d80-bf96-11cd-b579-08002b30bfeb}"));
_ProcessTypes(g_tfeOfficeTypes, ARRAYSIZE(g_tfeOfficeTypes), TEXT("{98de59a0-d175-11cd-a7bd-00006b827d94}"));
_ProcessTypes(g_tfeHtmlTypes, ARRAYSIZE(g_tfeHtmlTypes), TEXT("{eec97550-47a9-11cf-b952-00aa0051fe20}"));
}
CFindConstraint::CFindConstraint(BSTR bstr, VARIANT var) : CImpIDispatch(LIBID_Shell32, 1, 0, IID_DFConstraint)
{
_cRef = 1;
_bstr = bstr;
_var = var;
}
CFindConstraint::~CFindConstraint()
{
SysFreeString(_bstr);
VariantClear(&_var);
}
STDMETHODIMP CFindConstraint::QueryInterface(REFIID riid, void **ppv)
{
static const QITAB qit[] = {
QITABENT(CFindConstraint, DFConstraint), // IID_DFConstraint
QITABENTMULTI(CFindConstraint, IDispatch, DFConstraint), // IID_IDispatch
{ 0 },
};
return QISearch(this, qit, riid, ppv);
}
STDMETHODIMP_(ULONG) CFindConstraint::AddRef()
{
return InterlockedIncrement(&_cRef);
}
STDMETHODIMP_(ULONG) CFindConstraint::Release()
{
if (InterlockedDecrement(&_cRef))
return _cRef;
delete this;
return 0;
}
STDMETHODIMP CFindConstraint::GetTypeInfoCount(UINT * pctinfo)
{
return CImpIDispatch::GetTypeInfoCount(pctinfo);
}
STDMETHODIMP CFindConstraint::GetTypeInfo(UINT itinfo, LCID lcid, ITypeInfo **pptinfo)
{
return CImpIDispatch::GetTypeInfo(itinfo, lcid, pptinfo);
}
STDMETHODIMP CFindConstraint::GetIDsOfNames(REFIID riid, OLECHAR **rgszNames, UINT cNames, LCID lcid, DISPID * rgdispid)
{
return CImpIDispatch::GetIDsOfNames(riid, rgszNames, cNames, lcid, rgdispid);
}
STDMETHODIMP CFindConstraint::Invoke(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags, DISPPARAMS * pdispparams, VARIANT * pvarResult, EXCEPINFO * pexcepinfo, UINT * puArgErr)
{
return CImpIDispatch::Invoke(dispidMember, riid, lcid, wFlags, pdispparams, pvarResult, pexcepinfo, puArgErr);
}
STDMETHODIMP CFindConstraint::get_Name(BSTR *pbs)
{
*pbs = SysAllocString(_bstr);
return *pbs? S_OK : E_OUTOFMEMORY;
}
STDMETHODIMP CFindConstraint::get_Value(VARIANT *pvar)
{
VariantInit(pvar);
return VariantCopy(pvar, &_var);
}