windows-nt/Source/XPSP1/NT/base/ntsetup/win95upg/common/fileenum/enumaw.c
2020-09-26 16:20:57 +08:00

778 lines
21 KiB
C

#define MAX_DRIVES 64
BOOL
EnumerateAllDrivesT (
IN FILEENUMPROCT fnEnumCallback,
IN FILEENUMFAILPROCT fnFailCallback,
IN DWORD EnumID,
IN LPVOID pParam,
IN PEXCLUDEINFT ExcludeInfStruct,
IN DWORD AttributeFilter
)
/*++
Routine Description:
EnumerateAllDrives first builds an exclusion list if an exclusion INF path
is provided, and then enumerates every file on every drive that is not
excluded. The callback function is called once per file. The pParam
parameter is passed to the callback.
Arguments:
fnEnumCallback - A pointer to your callback function
EnumID - A caller-defined value used to identify the exclusion list
pParam - LPVOID passed to callback function
ExcludeInfStruct - Struct containing INF file information for excluding dirs or files
AttributeFilter - FILTER_xxx constants
Return Value:
TRUE if function succeeds. Call GetLastError for error code if return
value is FALSE.
--*/
{
TCHAR LogicalDrives[MAX_DRIVES];
DWORD rc;
PCTSTR p;
UINT driveType;
rc = GetLogicalDriveStrings (
MAX_DRIVES,
LogicalDrives
);
if (!rc || rc > MAX_DRIVES) {
if (rc)
SetLastError (ERROR_OUTOFMEMORY);
return FALSE;
}
for (p = LogicalDrives ; *p ; p = GetEndOfString (p) + 1) {
driveType = GetDriveType(p);
if (driveType == DRIVE_REMOTE || driveType == DRIVE_CDROM) {
continue;
}
if (!EnumerateTreeT (p,
fnEnumCallback,
fnFailCallback,
EnumID,
pParam,
ENUM_ALL_LEVELS,
ExcludeInfStruct,
AttributeFilter
))
break;
}
return (*p == 0);
}
BOOL
EnumerateTreeT (
IN PCTSTR EnumRoot,
IN FILEENUMPROCT fnEnumCallback,
IN FILEENUMFAILPROCT fnFailCallback, OPTIONAL
IN DWORD EnumID,
IN LPVOID pParam,
IN DWORD Levels,
IN PEXCLUDEINFT ExcludeInfStruct, OPTIONAL
IN DWORD AttributeFilter
)
/*++
Routine Description:
EnumerateTree is similar to EnumerateAllDrives, except it allows you to
enumerate a specific drive, or a specific subdir on a drive. Supply the
drive letter and optional subdirectory in EnumRoot. Before enumerating,
EnumerateTree will first build an exclusion list if an exclusion INF path
is provided. Then every file below EnumRoot is enumerated, and the
callback is called once per file, passing pParam unchanged.
Arguments:
EnumRoot - Drive and optional path to enumerate
fnEnumCallback - A pointer to your callback function
fnFailCallback - A pointer to optional fn that logs enumeration errors
EnumID - A caller-defined value used to identify the exclusion list
pParam - LPVOID passed to callback function
ExcludeInfStruct - Struct containing INF file information for excluding dirs or files
AttributeFilter - FILTER_xxx constants
Return Value:
TRUE if function succeeds. Call GetLastError for error code if return
value is FALSE.
--*/
{
ENUMSTRUCTT es;
BOOL b;
if (ExcludeInfStruct)
if (!BuildExclusionsFromInfT (
EnumID,
ExcludeInfStruct
)) {
DEBUGMSG ((DBG_ERROR, "Error in exclusion file"));
return FALSE;
}
es.fnEnumCallback = fnEnumCallback;
es.fnFailCallback = fnFailCallback;
es.EnumID = EnumID;
es.pParam = pParam;
es.Levels = Levels;
es.CurrentLevel = 1;
es.AttributeFilter = AttributeFilter;
if (!IsPathLengthOk(EnumRoot))
{
if (NULL != fnFailCallback)
{
fnFailCallback(EnumRoot);
return TRUE;
}
}
if (IsPathExcludedT (EnumID, EnumRoot))
return TRUE;
b = EnumTreeEngineT (EnumRoot, &es);
return b;
}
BOOL
EnumTreeEngineT (
PCTSTR CurrentPath,
PENUMSTRUCTT pes
)
{
WIN32_FIND_DATA fd; // A find struct for this subdir
HANDLE hFind; // A find handle for this subdir
PTSTR FullFilePath; // Buffer used to build file path
static TCHAR FindPattern[MAX_TCHAR_PATH * 2]; // Temp buffer used to build pattern
BYTE byBitmask[MAX_PATH]; // Bitmask is used to speed exclusion lookup
static DWORD Attrib; // Temp attribute storage for filter processing
static INT rc; // Callback return value
DWORD PrevLevelCt; // Storage for parent's max depth setting
BOOL RecurseStatus;
DWORD CurrentDirData = 0;
//
// Do nothing when CurrentPath is at the size limit.
//
if (!IsPathLengthOk(CurrentPath))
{
if (NULL != pes->fnFailCallback)
{
pes->fnFailCallback(CurrentPath);
}
return TRUE;
}
PrevLevelCt = pes->Levels;
StringCopy (FindPattern, CurrentPath);
//
// Create a bitmask that tells us when subdirectories match partial
// file patterns.
//
ZeroMemory (byBitmask, sizeof (byBitmask));
CreateBitmaskT (pes->EnumID, FindPattern, byBitmask);
AppendPathWack (FindPattern);
StringCat (FindPattern, TEXT("*"));
hFind = FindFirstFile (FindPattern, &fd);
if (hFind != INVALID_HANDLE_VALUE) {
do {
FullFilePath = JoinPaths (CurrentPath, fd.cFileName);
__try {
//
// Ignore this path if FullFilePath is too long
// this way fd.cFileName will surely be within limits (since it's shorter)
//
if (!IsPathLengthOk (FullFilePath)) {
if (NULL != pes->fnFailCallback) {
pes->fnFailCallback(FullFilePath);
}
continue;
}
// Filter directories named ".", "..". Set Attrib symbol.
if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
if (!StringCompare (fd.cFileName, TEXT(".")) ||
!StringCompare (fd.cFileName, TEXT("..")))
continue;
Attrib = FILTER_DIRECTORIES;
} else {
Attrib = FILTER_FILES;
}
// Call the callback
if (Attrib & pes->AttributeFilter) {
rc = CALLBACK_CONTINUE;
switch (Attrib) {
case FILTER_DIRECTORIES:
// Ignore excluded paths
if (IsPathExcludedT (pes->EnumID, FullFilePath)) {
break;
}
// Callback for 'directory first'
if (!(pes->AttributeFilter & FILTER_DIRS_LAST)) {
rc = pes->fnEnumCallback (
FullFilePath,
NULL,
&fd,
pes->EnumID,
pes->pParam,
&CurrentDirData
);
}
if (rc >= CALLBACK_CONTINUE && pes->CurrentLevel != pes -> Levels) {
// Recurse on directory
pes->CurrentLevel++;
RecurseStatus = EnumTreeEngineT (FullFilePath, pes);
pes->CurrentLevel--;
if (!RecurseStatus) {
PushError();
FindClose(hFind);
PopError();
return FALSE;
}
}
// Callback for 'directory last'
if (pes->AttributeFilter & FILTER_DIRS_LAST) {
rc = pes->fnEnumCallback (
FullFilePath,
NULL,
&fd,
pes->EnumID,
pes->pParam,
&CurrentDirData
);
}
break;
case FILTER_FILES:
if (!IsFileExcludedT (pes->EnumID, FullFilePath, byBitmask)) {
rc = pes->fnEnumCallback (FullFilePath,
NULL,
&fd,
pes->EnumID,
pes->pParam,
&CurrentDirData
);
}
break;
}
if (rc == CALLBACK_FAILED) {
PushError();
FindClose (hFind);
PopError();
return FALSE;
}
else if (rc == CALLBACK_SUBDIR_DONE) {
break;
}
else if (rc > 0) {
pes->Levels = pes->CurrentLevel + rc;
}
}
else if (Attrib == FILTER_DIRECTORIES && !IsPathExcludedT (pes->EnumID, FullFilePath)) {
// Recurse on directory.
if (pes->CurrentLevel != pes -> Levels) {
pes->CurrentLevel++;
RecurseStatus = EnumTreeEngineT (FullFilePath, pes);
pes->CurrentLevel--;
if (!RecurseStatus) {
PushError();
FindClose(hFind);
PopError();
return FALSE;
}
}
}
}
__finally {
FreePathString (FullFilePath);
}
} while (FindNextFile (hFind, &fd));
FindClose (hFind);
//
// Test error code returned from FindNextFile
//
if (GetLastError() != ERROR_NO_MORE_FILES && GetLastError() != ERROR_SUCCESS)
{
//
// Caller to handle not-ready message
//
if (GetLastError() != ERROR_NOT_READY)
{
DEBUGMSG((DBG_ERROR,
"EnumTreeEngineT: Error from FindNextFile.\n"
" FindPattern: %s\n"
" Error: %u (%x)",
FindPattern,
GetLastError(),GetLastError()));
}
return FALSE;
}
SetLastError(ERROR_SUCCESS);
}
else {
//
// Test return codes from FindFirstFile
//
if (GetLastError () != ERROR_NO_MORE_FILES)
{
//
// Caller to handle not-ready message
//
if (GetLastError() != ERROR_NOT_READY)
{
DEBUGMSG((DBG_WARNING,
"EnumTreeEngineT: Warning from FindFirstFile.\n"
" FindPattern: %s\n",
FindPattern));
}
// return FALSE;
}
SetLastError (ERROR_SUCCESS);
}
// If a callback returned a positive, non-zero number, the depth
// of the subdirectory search was limited for this level. Now that
// this level is done, we must restore the depth value of our parent.
pes->Levels = PrevLevelCt;
return TRUE;
}
/*++
A bitmask is used in IsFileExcluded for faster relative directory searches.
Instead of looking in the MemDb for each part of the path, IsFileExcluded
skips segments that are known not to match. We create the bitmask here
by looking up each portion of FindPattern. Bit 1 is set if the last
subdirectory exists in the file exclusion list, Bit 2 is set if the last
two subdirectories exist in the file exclusion list, and so on.
For example, assume FindPattern is set to C:\DEV\FOO\BAR. CreateBitmask
first looks in the memory database for BAR\*, and if it is found bit 1 is set.
Then CreateBitmask looks in the memory database for FOO\BAR\*, and sets bit
2. Again the function looks up DEV\FOO\BAR\* for bit 3 and finally
C:\DEV\FOO\BAR\* for bit 4.
Bit 0 is always set (empty paths always match).
Once this bitmask is set up, IsFileExcluded can test only the patterns that
are known to exist.
--*/
void
CreateBitmaskT (
DWORD EnumID,
PCTSTR FindPattern,
BYTE byBitmask[]
)
{
TCHAR EnumPath[MAX_TCHAR_PATH * 2];
TCHAR ShortPath[MAX_TCHAR_PATH * 2];
PCTSTR p;
PTSTR End;
int nByte;
int nBitVal;
// Always set bit 0
byBitmask[0] |= 1;
// Build full file spec
wsprintf (
EnumPath,
TEXT("%s\\%X\\%s\\"),
MEMDB_CATEGORY_FILEENUM,
EnumID,
MEMDB_FIELD_FE_FILES
);
End = GetEndOfString (EnumPath);
StringCopy (End, FindPattern);
AppendPathWack (End);
StringCat (End, TEXT("*"));
// Start with last subdirectory, and build mask in reverse
p = _tcsrchr (EnumPath, TEXT('\\'));
nByte = 0;
nBitVal = 2;
do {
// Move back to previous backslash
for (p = _tcsdec (EnumPath, p) ;
p >= End && *p != TEXT('\\') ;
p = _tcsdec (EnumPath, p))
{
}
// Check if partial file is in the tree
wsprintf (
ShortPath,
TEXT("%s\\%X\\%s%s"),
MEMDB_CATEGORY_FILEENUM,
EnumID,
MEMDB_FIELD_FE_FILES,
p
);
if (MemDbGetPatternValueWithPattern (ShortPath, NULL))
byBitmask[nByte] |= nBitVal;
// Inc bit pos
nBitVal *= 2;
if (nBitVal == 256) {
nBitVal = 1;
nByte++;
}
} while (p > End);
}
BOOL
IsPathExcludedT (DWORD EnumID, PCTSTR Path)
{
TCHAR EnumPath[MAX_TCHAR_PATH * 2];
TCHAR ShortPath[MAX_TCHAR_PATH * 2];
PCTSTR p;
PTSTR End;
// Try full paths
wsprintf (
EnumPath,
TEXT("%s\\%X\\%s\\"),
MEMDB_CATEGORY_FILEENUM,
EnumID,
MEMDB_FIELD_FE_PATHS
);
End = GetEndOfString (EnumPath);
p = _tcsappend (End, Path);
if (MemDbGetPatternValue (EnumPath, NULL)) {
return TRUE;
}
// Try partial paths
do {
// Move back to previous backslash
for (p = _tcsdec (EnumPath, p) ;
p > End && (*p != TEXT('\\')) ;
p = _tcsdec (EnumPath, p))
{
}
// Check if partial path is in the tree
if (p > End && p[1]) {
wsprintf (
ShortPath,
TEXT("%s\\%X\\%s%s"),
MEMDB_CATEGORY_FILEENUM,
EnumID,
MEMDB_FIELD_FE_PATHS,
p
);
if (MemDbGetPatternValue (ShortPath, NULL)) {
return TRUE;
}
}
} while (p > End);
return FALSE;
}
BOOL
IsFileExcludedT (DWORD EnumID, PCTSTR File, BYTE byBitmask[])
{
TCHAR EnumPath[MAX_TCHAR_PATH * 2];
TCHAR ShortPath[MAX_TCHAR_PATH * 2];
PCTSTR p;
PTSTR End;
int nByte;
int nBit;
// Build full file spec
wsprintf (
EnumPath,
TEXT("%s\\%X\\%s\\"),
MEMDB_CATEGORY_FILEENUM,
EnumID,
MEMDB_FIELD_FE_FILES
);
End = GetEndOfString (EnumPath);
p = _tcsappend (End, File);
//
// Try partial file specs until full spec is reached
//
nByte = 0;
nBit = 1;
do {
//
// Move back to previous backslash in path
// (p starts at NULL of EnumPath, End is in the middle of EnumPath)
//
for (p = _tcsdec (EnumPath, p) ;
p >= End && (*p != TEXT('\\')) ;
p = _tcsdec (EnumPath, p))
{
}
// Bitmask is used to make sure slightly expensive query is necessary
if (byBitmask[nByte] & nBit) {
//
// Check if partial file is in the tree
//
wsprintf (
ShortPath,
TEXT("%s\\%X\\%s%s"),
MEMDB_CATEGORY_FILEENUM,
EnumID,
MEMDB_FIELD_FE_FILES,
p
);
if (MemDbGetPatternValue (ShortPath, NULL)) {
return TRUE;
}
}
nBit *= 2;
if (nBit == 256) {
nBit = 1;
nByte++;
}
} while (p > End);
return FALSE;
}
//
// ClearExclusions removes all enumaration exclusions. It is called
// automatically at the end of enumeration when an exclusion INF file is
// used. Use it when you need to programmatically build an exclusion list
// with ExcludeDrive, ExcludePath, and ExcludeFile.
//
// You can combine programmatic exclusions with an exclusion INF file, but
// beware that the programmatic exclusions will be cleared after
// EnumarteAllDrives or EnumerateTree completes.
//
// If you do not use an INF, the programmatic exclusions will not
// automatically be cleared.
//
// EnumID - Caller-defined value to identify enumeration exclusion list
//
VOID
ClearExclusionsT (
DWORD EnumID
)
{
TCHAR EnumPath[MAX_TCHAR_PATH * 2];
wsprintf (EnumPath, TEXT("%s\\%X"), MEMDB_CATEGORY_FILEENUM, EnumID);
MemDbDeleteTree (EnumPath);
}
/*++
Routine Description:
ExcludePath adds a path name to the exclusion list. There are two
cases:
1. A full path spec is supplied, including the drive letter or
UNC double-backslash.
2. The path does not start with a drive letter and is a portion of
a full path.
The dot and double-dot directories are not supported. Any part of
the path may contain wildcard characters, but a wildcard can not
be used in place of a backslash.
Arguments:
EnumID - A caller-defined value that identifies the exclusion list
Path - The path specification as described above
Return Value:
none
--*/
VOID
ExcludePathT (
IN DWORD EnumID,
IN PCTSTR Path
)
{
TCHAR EnumPath[MAX_TCHAR_PATH * 2];
wsprintf (
EnumPath,
TEXT("%s\\%X\\%s\\%s"),
MEMDB_CATEGORY_FILEENUM,
EnumID,
MEMDB_FIELD_FE_PATHS,
Path
);
MemDbSetValue (EnumPath, 0);
}
/*++
Routine Description:
ExcludeFile adds a file spec to the exclusion list. There are two
cases:
1. A full path spec is supplied, including the drive letter or
UNC double-backslash.
2. The path does not start with a drive letter and is a portion of
a full path.
The dot and double-dot directories are not supported. Any part of
the path may contain wildcard characters, but a wildcard can not
be used in place of a backslash.
Arguments:
EnumID - A caller-defined value that identifies the exclusion list
File - The file specification as described above
Return Value:
none
--*/
VOID
ExcludeFileT (
IN DWORD EnumID,
IN PCTSTR File
)
{
TCHAR EnumPath[MAX_TCHAR_PATH * 2];
wsprintf (
EnumPath,
TEXT("%s\\%X\\%s\\%s"),
MEMDB_CATEGORY_FILEENUM,
EnumID,
MEMDB_FIELD_FE_FILES,
File
);
MemDbSetValue (EnumPath, 0);
}
BOOL
BuildExclusionsFromInfT (DWORD EnumID,
PEXCLUDEINFT ExcludeInfStruct)
{
HINF hInf;
INFCONTEXT ic;
TCHAR Exclude[MAX_TCHAR_PATH * 2];
// Attempt to open
hInf = SetupOpenInfFile (ExcludeInfStruct->ExclusionInfPath, NULL, INF_STYLE_WIN4, NULL);
if (hInf == INVALID_HANDLE_VALUE)
return FALSE;
// Read in path exclusions
if (ExcludeInfStruct->PathSection) {
if (SetupFindFirstLine (hInf, ExcludeInfStruct->PathSection, NULL, &ic)) {
do {
if (SetupGetStringField (&ic, 1, Exclude, MAX_TCHAR_PATH, NULL)) {
ExcludePathT (EnumID, Exclude);
}
} while (SetupFindNextLine (&ic, &ic));
}
}
// Read in file exclusions
if (ExcludeInfStruct->FileSection) {
if (SetupFindFirstLine (hInf, ExcludeInfStruct->FileSection, NULL, &ic)) {
do {
if (SetupGetStringField (&ic, 1, Exclude, MAX_TCHAR_PATH, NULL)) {
ExcludeFileT (EnumID, Exclude);
}
} while (SetupFindNextLine (&ic, &ic));
}
}
// Clean up
SetupCloseInfFile (hInf);
return TRUE;
}