576 lines
20 KiB
C++
576 lines
20 KiB
C++
|
|
|
|
// Copyright (c) Microsoft Corporation
|
|
|
|
// File: TreeWalk.cpp
|
|
|
|
// This file contains the implementation of CShellTreeWalker, a COM object
|
|
// that inherits IShellTreeWalker, and it will recursively enumerate all the
|
|
// files (or directories or both) starting from a root directory that match a
|
|
// certain spec.
|
|
// 1. The tree walker is reparse point aware, it does not traverse into reparse
|
|
// point folders by default, but will if specified
|
|
// 2. It keeps track of the number of files, directories, depth, and total
|
|
// size of all files encountered.
|
|
// 3. It will stop the traversal right away if any error message is returned
|
|
// from the callback functions except for E_NOTIMPL
|
|
// 4. It will jump out of the current working directory if S_FALSE is returned from callback
|
|
// functions.
|
|
|
|
// History:
|
|
// 12-5-97 by dli
|
|
|
|
#include "shellprv.h"
|
|
#include "validate.h"
|
|
|
|
#define MAX_TREE_DEPTH 0xFFFFFFFF// maximum file system tree depth we walk into
|
|
|
|
#define IS_FILE_DIRECTORY(pwfd) ((pwfd)->dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
|
|
#define IS_FILE_REPARSE_POINT(pwfd) ((pwfd)->dwFileAttributes & FILE_ATTRIBUTE_REPARSE_POINT)
|
|
|
|
// Call back flags for _CallCallBack
|
|
#define STWCB_FILE 1
|
|
#define STWCB_ERROR 2
|
|
#define STWCB_ENTERDIR 3
|
|
#define STWCB_LEAVEDIR 4
|
|
|
|
#define TF_TREEWALKER 0
|
|
|
|
extern "C" {
|
|
DWORD PathGetClusterSize(LPCTSTR pszPath);
|
|
}
|
|
|
|
class CShellTreeWalker : public IShellTreeWalker
|
|
{
|
|
public:
|
|
CShellTreeWalker();
|
|
|
|
// ** IUnknown Methods
|
|
virtual STDMETHODIMP QueryInterface(REFIID riid, LPVOID * ppvObj);
|
|
virtual STDMETHODIMP_(ULONG) AddRef(void) ;
|
|
virtual STDMETHODIMP_(ULONG) Release(void);
|
|
|
|
// ** IShellTreeWalker Methods
|
|
virtual STDMETHODIMP WalkTree(DWORD dwFlags, LPCWSTR pwszWalkRoot, LPCWSTR pwszWalkSpec, int iMaxPath, IShellTreeWalkerCallBack * pstwcb);
|
|
|
|
protected:
|
|
|
|
UINT _cRef;
|
|
|
|
DWORD _dwFlags; // Flags indicating the search status
|
|
UINT _nMaxDepth; // Maximum depth we walk into
|
|
UINT _nDepth; // Current depth
|
|
UINT _nFiles; // Number of files we have seen so far
|
|
UINT _nDirs; // Number of directories we have seen
|
|
|
|
BOOL _bFolderFirst; // Do the folders first
|
|
DWORD _dwClusterSize; // the size of a cluster
|
|
ULONGLONG _ulTotalSize; // total size of all files we have seen
|
|
ULONGLONG _ulActualSize; // total size on disk, taking into account compression, sparse files, and cluster slop
|
|
|
|
TCHAR _szWalkBuf[MAX_PATH]; // The path buffer used in the walk
|
|
LPCTSTR _pszWalkSpec; // The spec we use in FindFirstFile and FindNextFile
|
|
|
|
IShellTreeWalkerCallBack * _pstwcb; // The call back interface pointer
|
|
|
|
WIN32_FIND_DATA _wfd; // The temp storage of WIN32_FIND_DATA
|
|
|
|
WIN32_FIND_DATA _fdTopLevelFolder; // The top level folder info
|
|
|
|
HRESULT _CallCallBacks(DWORD dwCallReason, WIN32_FIND_DATA * pwfd);
|
|
HRESULT _ProcessAndRecurse(WIN32_FIND_DATA * pwfd);
|
|
HRESULT _TreeWalkerHelper();
|
|
};
|
|
|
|
|
|
|
|
// default constructor
|
|
|
|
CShellTreeWalker::CShellTreeWalker() : _cRef(1)
|
|
{
|
|
ASSERT(_dwFlags == 0);
|
|
ASSERT(_bFolderFirst == FALSE);
|
|
ASSERT(_nMaxDepth == 0);
|
|
ASSERT(_nDepth == 0);
|
|
ASSERT(_nDirs == 0);
|
|
ASSERT(_ulTotalSize == 0);
|
|
ASSERT(_ulActualSize == 0);
|
|
ASSERT(_pszWalkSpec == NULL);
|
|
ASSERT(_pstwcb == NULL);
|
|
ASSERT(_szWalkBuf[0] == 0);
|
|
}
|
|
|
|
|
|
// _CallCallBack: convert the TCHARs to WCHARs and call the callback functions
|
|
HRESULT CShellTreeWalker::_CallCallBacks(DWORD dwReason, WIN32_FIND_DATA * pwfd)
|
|
{
|
|
HRESULT hres;
|
|
WCHAR wszDir[MAX_PATH];
|
|
WCHAR wszFileName[MAX_PATH];
|
|
#ifndef UNICODE
|
|
CHAR szTemp[MAX_PATH];
|
|
#endif
|
|
WIN32_FIND_DATAW wfdw = {0};
|
|
WIN32_FIND_DATAW* pwfdw = NULL;
|
|
TREEWALKERSTATS tws = {0};
|
|
|
|
tws.nFiles = _nFiles;
|
|
tws.nFolders = _nDirs;
|
|
tws.nDepth = _nDepth;
|
|
tws.ulTotalSize = _ulTotalSize;
|
|
tws.ulActualSize = _ulActualSize;
|
|
tws.dwClusterSize = _dwClusterSize;
|
|
|
|
// _szWalkBuf to wszDir
|
|
#ifdef UNICODE
|
|
lstrcpy(wszDir, _szWalkBuf);
|
|
lstrcpy(wszFileName, wszDir);
|
|
PathCombine(wszFileName, wszFileName, pwfd->cFileName);
|
|
#else
|
|
SHAnsiToUnicode(_szWalkBuf, wszDir, ARRAYSIZE(wszDir));
|
|
lstrcpy(szTemp, _szWalkBuf);
|
|
PathCombine(szTemp, szTemp, pwfd->cFileName);
|
|
SHAnsiToUnicode(szTemp, wszFileName, ARRAYSIZE(wszFileName));
|
|
#endif
|
|
|
|
if (pwfd && ((dwReason == STWCB_FILE) || (dwReason == STWCB_ENTERDIR)))
|
|
{
|
|
// WIN32_FIND_DATAA to WIN32_FIND_DATAW
|
|
hmemcpy(&wfdw, pwfd, SIZEOF(WIN32_FIND_DATA));
|
|
#ifndef UNICODE
|
|
SHAnsiToUnicode(pwfd->cFileName, wfdw.cFileName, ARRAYSIZE(wfdw.cFileName));
|
|
SHAnsiToUnicode(pwfd->cAlternateFileName, wfdw.cAlternateFileName, ARRAYSIZE(wfdw.cAlternateFileName));
|
|
#endif
|
|
pwfdw = &wfdw;
|
|
}
|
|
|
|
switch (dwReason) {
|
|
case STWCB_FILE:
|
|
hres = _pstwcb->FoundFile(wszFileName, &tws, pwfdw);
|
|
TraceMsg(TF_TREEWALKER, "TreeWalker Callback FoundFile: %s\\%s dwReason: %x nFiles: %d nDepth: %d nDirs: %d",
|
|
_szWalkBuf, pwfd->cFileName, dwReason, _nFiles, _nDepth, _nDirs);
|
|
|
|
break;
|
|
case STWCB_ENTERDIR:
|
|
hres = _pstwcb->EnterFolder(wszDir, &tws, pwfdw);
|
|
TraceMsg(TF_TREEWALKER, "TreeWalker Callback EnterFolder: %s dwReason: %x nFiles: %d nDepth: %d nDirs: %d",
|
|
_szWalkBuf, dwReason, _nFiles, _nDepth, _nDirs);
|
|
break;
|
|
case STWCB_LEAVEDIR:
|
|
hres = _pstwcb->LeaveFolder(wszDir, &tws);
|
|
break;
|
|
// case STWCB_ERROR:
|
|
// hres = _pstwcb->HandleError(S_OK, wszDir, &tws);
|
|
// break;
|
|
default:
|
|
hres = S_OK;
|
|
break;
|
|
}
|
|
|
|
// Error messages are significant to us, all E_ messages are interpreted as "Stop right now!!"
|
|
if (hres == E_NOTIMPL)
|
|
hres = S_OK;
|
|
|
|
return hres;
|
|
}
|
|
|
|
|
|
// Call call back funtions on directories and files, recurse on directories if there is no objection
|
|
// from the callback object
|
|
HRESULT CShellTreeWalker::_ProcessAndRecurse(WIN32_FIND_DATA * pwfd)
|
|
{
|
|
HRESULT hres = S_OK;
|
|
|
|
// Don't recurse on reparse points by default
|
|
if (IS_FILE_DIRECTORY(pwfd) && (!IS_FILE_REPARSE_POINT(pwfd) || (_dwFlags & WT_GOINTOREPARSEPOINT)))
|
|
{
|
|
// BUGBUG: If we are in a symbolic link, we need to detect cycles,
|
|
// the common prefix method in BeenThereDoneThat will work as long as we
|
|
// keep track of all junction point targets we ran into.
|
|
|
|
// use _szWalkBuf since we dont want any stack variables, because we are a recursive function
|
|
if (PathCombine(_szWalkBuf, _szWalkBuf, pwfd->cFileName))
|
|
{
|
|
// We remember the total number of sub directories we have seen
|
|
// doesn't matter if the client approves or not(call back returns S_OK or S_FALSE or E_FAIL)
|
|
_nDirs++;
|
|
|
|
// Let the CallBack object know that we are about to enter a directory
|
|
if (_dwFlags & WT_NOTIFYFOLDERENTER)
|
|
hres = _CallCallBacks(STWCB_ENTERDIR, pwfd);
|
|
|
|
if ((hres == S_OK) && (_nDepth < _nMaxDepth))
|
|
{
|
|
_nDepth++;
|
|
hres = _TreeWalkerHelper();
|
|
_nDepth--;
|
|
}
|
|
else if (hres == S_FALSE)
|
|
hres = S_OK;
|
|
|
|
// Let the CallBack object know that we are about to leave a directory
|
|
if (_dwFlags & WT_NOTIFYFOLDERLEAVE)
|
|
_CallCallBacks(STWCB_LEAVEDIR, NULL);
|
|
|
|
// Peel off the subdirectory we tagged on in the above PathCombine Ex:"c:\bin\fun --> c:\bin"
|
|
PathRemoveFileSpec(_szWalkBuf);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Count the number of files and compute the total size before callling the
|
|
// call back object.
|
|
ULARGE_INTEGER ulTemp;
|
|
_nFiles++;
|
|
|
|
ulTemp.LowPart = pwfd->nFileSizeLow;
|
|
ulTemp.HighPart = pwfd->nFileSizeHigh;
|
|
_ulTotalSize += ulTemp.QuadPart;
|
|
|
|
#ifdef WINNT
|
|
// when calculating the total size, we need to find out if the file is compressed or sparse (NTFS only case)
|
|
if (pwfd->dwFileAttributes & (FILE_ATTRIBUTE_COMPRESSED | FILE_ATTRIBUTE_SPARSE_FILE))
|
|
{
|
|
// use _szWalkBuf since we dont want any stack variables, because we are a recursive function
|
|
PathCombine(_szWalkBuf, _szWalkBuf, pwfd->cFileName);
|
|
|
|
// eithe the file is compressed or sparse, we need to call GetCompressedFileSize to get the real
|
|
// size on disk for this file (NOTE: GetCompressedFileSize takes into account cluster slop, except
|
|
// for files < 1 cluster, and we take care of that below)
|
|
ulTemp.LowPart = SHGetCompressedFileSize(_szWalkBuf, &ulTemp.HighPart);
|
|
|
|
_ulActualSize += ulTemp.QuadPart;
|
|
|
|
// Peel off the filename we tagged on above
|
|
PathRemoveFileSpec(_szWalkBuf);
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
// BUGBUG (reinerf) the cluster size could change if we started on one volume and have now
|
|
// walked onto another mounted volume
|
|
|
|
// if its not compressed, we just round up to the drive's cluster size. ulTemp was setup
|
|
// already for us above, so just round it to the cluster and add it in
|
|
_ulActualSize += ROUND_TO_CLUSTER(ulTemp.QuadPart, _dwClusterSize);
|
|
}
|
|
|
|
hres = _CallCallBacks(STWCB_FILE, pwfd);
|
|
}
|
|
|
|
return hres;
|
|
}
|
|
|
|
#define DELAY_ARRAY_GROW 32
|
|
|
|
// Recursive function that does the real work on the traversal,
|
|
HRESULT CShellTreeWalker::_TreeWalkerHelper()
|
|
{
|
|
HRESULT hres = S_OK;
|
|
TraceMsg(TF_TREEWALKER, "TreeWalkerHelper started on: %s flags: %x nFiles: %d nDepth: %d nDirs: %d",
|
|
_szWalkBuf, _dwFlags, _nFiles, _nDepth, _nDirs);
|
|
|
|
// Let the CallBack object know that we are about to start the walk
|
|
// provided he cares about the root
|
|
if (_nDepth == 0 && !(_dwFlags & WT_EXCLUDEWALKROOT) &&
|
|
(_dwFlags & WT_NOTIFYFOLDERENTER))
|
|
{
|
|
HANDLE hTopLevelFolder;
|
|
|
|
// Get the info for the TopLevelFolder
|
|
hTopLevelFolder = FindFirstFile(_szWalkBuf, &_fdTopLevelFolder);
|
|
|
|
if (hTopLevelFolder == INVALID_HANDLE_VALUE)
|
|
{
|
|
LPTSTR szFileName;
|
|
DWORD dwAttribs = -1; // assume failure
|
|
|
|
// We could have failed if we tried to do a FindFirstFile on the root (c:\)
|
|
// or if something is really wrong, to test for this we do a GetFileAttributes (GetFileAttributesEx on NT)
|
|
#ifdef WINNT
|
|
// on NT we can use GetFileAttributesEx to get both the attribs and part of the win32fd
|
|
if (GetFileAttributesEx(_szWalkBuf, GetFileExInfoStandard, (LPVOID)&_fdTopLevelFolder))
|
|
{
|
|
// success!
|
|
dwAttribs = _fdTopLevelFolder.dwFileAttributes;
|
|
szFileName = PathFindFileName(_szWalkBuf);
|
|
lstrcpyn(_fdTopLevelFolder.cFileName, szFileName, ARRAYSIZE(_fdTopLevelFolder.cFileName));
|
|
lstrcpyn(_fdTopLevelFolder.cAlternateFileName, szFileName, ARRAYSIZE(_fdTopLevelFolder.cAlternateFileName));
|
|
}
|
|
#else
|
|
// on win9x we call GetFileAttrbutes, since we dont have GetFileAttributesEx
|
|
dwAttribs = GetFileAttributes(_szWalkBuf);
|
|
|
|
if (dwAttribs != -1)
|
|
{
|
|
// success!
|
|
|
|
// On win95 we steal a bunch of the find data from our first child, and fake the rest. Its lame,
|
|
// but its the best we can do.
|
|
|
|
memcpy(&_fdTopLevelFolder, &_wfd, SIZEOF(_fdTopLevelFolder));
|
|
|
|
_fdTopLevelFolder.dwFileAttributes = dwAttribs;
|
|
|
|
szFileName = PathFindFileName(_szWalkBuf);
|
|
lstrcpyn(_fdTopLevelFolder.cFileName, szFileName, ARRAYSIZE(_fdTopLevelFolder.cFileName));
|
|
lstrcpyn(_fdTopLevelFolder.cAlternateFileName, szFileName, ARRAYSIZE(_fdTopLevelFolder.cAlternateFileName));
|
|
}
|
|
#endif
|
|
if (dwAttribs == -1)
|
|
{
|
|
// this is very bad, so we bail
|
|
TraceMsg(TF_ERROR, "Tree Walker: GetFileAttributes/Ex(%s) failed. Stopping the walk.", _szWalkBuf);
|
|
return E_FAIL;
|
|
}
|
|
|
|
}
|
|
else
|
|
{
|
|
// We sucessfully got the find data, good.
|
|
FindClose(hTopLevelFolder);
|
|
}
|
|
|
|
// call the callback for the first enterdir
|
|
hres = _CallCallBacks(STWCB_ENTERDIR, &_fdTopLevelFolder);
|
|
}
|
|
|
|
// Do the real tree walk here
|
|
if (hres == S_OK)
|
|
{
|
|
// always use *.* to search when we are not at our maximum level because
|
|
// we need the sub directories
|
|
// BUGBUG: this can be changed on NT by using FindFirstFileEx if WT_FOLDERONLY
|
|
LPCTSTR pszSpec = (_pszWalkSpec && (_nDepth == _nMaxDepth)) ? _pszWalkSpec : c_szStarDotStar;
|
|
if (PathCombine(_szWalkBuf, _szWalkBuf, pszSpec))
|
|
{
|
|
HDSA hdsaDelayed = NULL; // array of found items that will be delayed and processed later
|
|
HANDLE hFind;
|
|
|
|
// Start finding the sub folders and files
|
|
hFind = FindFirstFile(_szWalkBuf, &_wfd);
|
|
|
|
// Peel off the find spec Ex:"c:\bin\*.* --> c:\bin"
|
|
PathRemoveFileSpec(_szWalkBuf);
|
|
|
|
if (hFind != INVALID_HANDLE_VALUE)
|
|
{
|
|
BOOL bDir = FALSE;
|
|
|
|
do
|
|
{
|
|
// Skip over the . and .. entries.
|
|
if (PathIsDotOrDotDot(_wfd.cFileName))
|
|
continue;
|
|
|
|
bDir = BOOLIFY(IS_FILE_DIRECTORY(&_wfd));
|
|
|
|
// If this is a file, and we are not interested in files or this file spec does not match the one we were
|
|
// looking for.
|
|
if ((!bDir) && ((_dwFlags & WT_FOLDERONLY) ||
|
|
(_pszWalkSpec && (_nDepth < _nMaxDepth) && !PathMatchSpec(_wfd.cFileName, _pszWalkSpec))))
|
|
continue;
|
|
|
|
// The following EQUAL determines whether we want to process
|
|
// the data found or save it in the HDSA array and process them later.
|
|
|
|
// Enumerate the folder or file now? (determined by the WT_FOLDERFIRST flag)
|
|
if (bDir == BOOLIFY(_bFolderFirst))
|
|
{
|
|
// Yes
|
|
hres = _ProcessAndRecurse(&_wfd);
|
|
|
|
// if hres is failure code someone said "stop right now"
|
|
// if hres is S_FALSE some one said "quit this directory and start with next one"
|
|
if (hres != S_OK)
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
// No; enumerate it once we're finished with the opposite.
|
|
if (!hdsaDelayed)
|
|
hdsaDelayed = DSA_Create(SIZEOF(WIN32_FIND_DATA), DELAY_ARRAY_GROW);
|
|
if (!hdsaDelayed)
|
|
{
|
|
hres = E_OUTOFMEMORY;
|
|
break;
|
|
}
|
|
DSA_AppendItem(hdsaDelayed, &_wfd);
|
|
|
|
}
|
|
} while (FindNextFile(hFind, &_wfd));
|
|
|
|
FindClose(hFind);
|
|
}
|
|
else
|
|
{
|
|
// find first file failed, this is a good place to report error
|
|
DWORD dwErr = GetLastError();
|
|
TraceMsg(TF_TREEWALKER, "**WARNING**: FindFirstFile faied on %s%s with error = %d", _szWalkBuf, _pszWalkSpec, dwErr);
|
|
}
|
|
|
|
// Process the delayed items, these are either directories or files
|
|
if (hdsaDelayed)
|
|
{
|
|
// we should have finished everything in the above do while loop in folderonly case
|
|
ASSERT(!(_dwFlags & WT_FOLDERONLY));
|
|
|
|
// if hres is failure code someone said "stop right now"
|
|
// if hres is S_FALSE some one said "quit this directory and start with next one"
|
|
if (hres == S_OK)
|
|
{
|
|
int ihdsa;
|
|
for (ihdsa = 0; ihdsa < DSA_GetItemCount(hdsaDelayed); ihdsa++)
|
|
{
|
|
WIN32_FIND_DATA * pwfd = (WIN32_FIND_DATA *)DSA_GetItemPtr(hdsaDelayed, ihdsa);
|
|
hres = _ProcessAndRecurse(pwfd);
|
|
if (hres != S_OK)
|
|
break;
|
|
}
|
|
}
|
|
DSA_Destroy(hdsaDelayed);
|
|
}
|
|
|
|
// Let the CallBack object know that we are finishing the walk
|
|
if (_nDepth == 0 && !(_dwFlags & WT_EXCLUDEWALKROOT) &&
|
|
(_dwFlags & WT_NOTIFYFOLDERLEAVE) && (S_OK == hres))
|
|
hres = _CallCallBacks(STWCB_LEAVEDIR, &_fdTopLevelFolder);
|
|
|
|
// hres was S_FALSE because someone wanted us to jump out of this current directory
|
|
// but don't pass it back to our parent directory
|
|
if (hres == S_FALSE)
|
|
hres = S_OK;
|
|
}
|
|
else
|
|
TraceMsg(TF_TREEWALKER, "**WARNING**: PathCombine failed!!!!");
|
|
}
|
|
|
|
return hres;
|
|
}
|
|
|
|
// IShellTreeWalker::WalkTree is the main function for the IShellTreeWalker interface
|
|
HRESULT CShellTreeWalker::WalkTree(DWORD dwFlags, LPCWSTR pwszWalkRoot, LPCWSTR pwszWalkSpec, int iMaxDepth, IShellTreeWalkerCallBack * pstwcb)
|
|
{
|
|
HRESULT hres = E_FAIL;
|
|
TCHAR szWalkSpec[64];
|
|
|
|
// must have a call back object to talk to
|
|
ASSERT(IS_VALID_CODE_PTR(pstwcb, IShellTreeWalkerCackBack));
|
|
if (pstwcb == NULL)
|
|
return E_INVALIDARG;
|
|
|
|
// make sure we have a valid directory to start with
|
|
ASSERT(IS_VALID_STRING_PTRW(pwszWalkRoot, -1));
|
|
if ((pwszWalkRoot != NULL) && (pwszWalkRoot[0] != L'\0'))
|
|
{
|
|
#ifdef UNICODE
|
|
lstrcpy(_szWalkBuf, pwszWalkRoot);
|
|
#else
|
|
WideCharToMultiByte(CP_ACP, 0,
|
|
pwszWalkRoot, -1,
|
|
_szWalkBuf, ARRAYSIZE(_szWalkBuf), NULL, NULL);
|
|
#endif
|
|
// call back
|
|
_pstwcb = pstwcb;
|
|
|
|
// copy the search flags and fix it up
|
|
_dwFlags = dwFlags & WT_ALL;
|
|
|
|
// this will save us from using the hdsa array to hold the directories
|
|
if (_dwFlags & WT_FOLDERONLY)
|
|
{
|
|
_dwFlags |= WT_FOLDERFIRST;
|
|
|
|
// It will be pretty meanless if the below flags are not set, because
|
|
// we don't call FoundFile in the FolderOnly case.
|
|
ASSERT(_dwFlags & (WT_NOTIFYFOLDERENTER | WT_NOTIFYFOLDERLEAVE));
|
|
}
|
|
|
|
if (_dwFlags & WT_FOLDERFIRST)
|
|
_bFolderFirst = TRUE;
|
|
|
|
if ((pwszWalkSpec != NULL) && (pwszWalkSpec[0] != L'\0'))
|
|
{
|
|
#ifdef UNICODE
|
|
lstrcpyn(szWalkSpec, pwszWalkSpec, ARRAYSIZE(szWalkSpec));
|
|
#else
|
|
WideCharToMultiByte(CP_ACP, 0,
|
|
pwszWalkSpec, -1,
|
|
szWalkSpec, ARRAYSIZE(szWalkSpec), NULL, NULL);
|
|
#endif
|
|
_pszWalkSpec = szWalkSpec;
|
|
}
|
|
|
|
_nMaxDepth = (dwFlags & WT_MAXDEPTH) ? iMaxDepth : MAX_TREE_DEPTH;
|
|
_dwClusterSize = PathGetClusterSize(_szWalkBuf);
|
|
hres = _TreeWalkerHelper();
|
|
}
|
|
else
|
|
TraceMsg(TF_WARNING, "CShellTreeWalker::WalkTree Failed! due to bad _szWalkBuf");
|
|
|
|
return hres;
|
|
}
|
|
|
|
// IShellTreeWalker::QueryInterface
|
|
HRESULT CShellTreeWalker::QueryInterface(REFIID riid, LPVOID * ppvObj)
|
|
{
|
|
ASSERT(ppvObj != NULL);
|
|
|
|
if (IsEqualIID(riid, IID_IUnknown))
|
|
{
|
|
*ppvObj = SAFECAST(this, IUnknown *);
|
|
TraceMsg(TF_TREEWALKER, "CShellTreeWalker::QI IUnknown succeeded");
|
|
}
|
|
else if (IsEqualIID(riid, IID_IShellTreeWalker))
|
|
{
|
|
*ppvObj = SAFECAST(this, IShellTreeWalker*);
|
|
TraceMsg(TF_TREEWALKER, "CShellTreeWalker::QI IShellTreeWalker succeeded");
|
|
}
|
|
else
|
|
{
|
|
*ppvObj = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
AddRef();
|
|
return S_OK;
|
|
}
|
|
|
|
// IShellTreeWalker::AddRef
|
|
ULONG CShellTreeWalker::AddRef()
|
|
{
|
|
_cRef++;
|
|
TraceMsg(TF_TREEWALKER, "CShellTreeWalker()::AddRef called, new _cRef=%lX", _cRef);
|
|
return _cRef;
|
|
}
|
|
|
|
// IShellTreeWalker::Release
|
|
ULONG CShellTreeWalker::Release()
|
|
{
|
|
_cRef--;
|
|
TraceMsg(TF_TREEWALKER, "CShellTreeWalker()::Release called, new _cRef=%lX", _cRef);
|
|
if (_cRef > 0)
|
|
return _cRef;
|
|
|
|
delete this;
|
|
return 0;
|
|
}
|
|
|
|
STDAPI CShellTreeWalker_CreateInstance(IUnknown* pUnkOuter, REFIID riid, OUT LPVOID * ppvOut)
|
|
{
|
|
HRESULT hr = E_FAIL;
|
|
|
|
TraceMsg(TF_TREEWALKER, "CShellTreeWalker_CreateInstance()");
|
|
*ppvOut = NULL;
|
|
|
|
CShellTreeWalker *pstw = new CShellTreeWalker;
|
|
if (!pstw)
|
|
return E_OUTOFMEMORY;
|
|
|
|
hr = pstw->QueryInterface(riid, (LPVOID *)ppvOut);
|
|
|
|
pstw->Release();
|
|
|
|
return hr;
|
|
}
|