able/windows-nt
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity
windows-nt/Source/XPSP1/NT/base/ntsetup/hwdb/utils/main/strings.c
CryptoAlgo Inc b3cac27891 Add source files
2020-09-26 16:20:57 +08:00

10938 lines
257 KiB
C
Raw Blame History

/*++
Copyright (c) 1996 Microsoft Corporation
Module Name:
strings.c
Abstract:
A number of string utilities useful for any project
Author:
Jim Schmidt (jimschm) 12-Sept-1996
Revisions:
ovidiut 12-Jan-2000 Added GetNodePatternMinMaxLevels,PatternIncludesPattern
ovidiut 14-Sep-1999 Updated for new coding conventions and Win64 compliance
marcw 2-Sep-1999 Moved over from Win9xUpg project.
jimschm 08-Jul-1999 IsPatternMatchEx
jimschm 07-Jan-1999 GetFileExtensionFromPath fixed again, added
GetDotExtensionFromPath
calinn 23-Sep-1998 GetFileExtensionFromPath bug fix
calinn 29-Jan-1998 Fixed a bug in EnumNextMultiSz.
calinn 11-Jan-1998 Added EnumFirstMultiSz and EnumNextMultiSz functions.
marcw 15-Dec-1997 Added ExpandEnvironmentTextEx functions.
marcw 14-Nov-1997 SlightJoinText revisions.
jimschm 21-May-1997 AppendWack revisions
marcw 24-Mar-1997 StringReplace functions.
jimschm 14-Mar-1997 New critical section stuff, enhanced message resource
routines, C runtime extensions, registry root utils
jimschm 26-Nov-1996 Added message resource tools.
mikeco 01-Jul-1997 Add FreeStringResourcePtr Fns
mikeco 29-Sep-1997 IsLeadByte wrapper for IsDBCSLeadByte
--*/
#include "pch.h"
//
// Includes
//
#include "utilsp.h"
#define DBG_STRINGS "Strings"
//
// Strings
//
// None
//
// Constants
//
// Error stack size (normally only one or two, so 32 is relatively huge)
#define MAX_STACK 32
#define WACK_REPLACE_CHAR 0x02
#define DWORD_MAX 0xFFFFFFFFu
//
// Macros
//
// None
//
// Types
//
typedef enum {
BEGIN_PATTERN,
BEGIN_COMPOUND_PATTERN,
BEGIN_PATTERN_EXPR,
SAVE_EXACT_MATCH,
SAVE_SEGMENT,
LOOK_FOR_NUMBER,
LOOK_FOR_INCLUDE,
LOOK_FOR_EXCLUDE,
ADVANCE_TO_END_OF_EXPR,
PARSE_CHAR_EXPR_OR_END,
SKIP_EXCLUDE_SET,
CONDENSE_SET,
PARSE_END_FOUND,
SKIP_INCLUDE_SET,
END_PATTERN_EXPR,
PATTERN_DONE,
PATTERN_ERROR
} PATTERNSTATE;
typedef struct {
UINT char1;
UINT char2;
UINT result;
} DHLIST, *PDHLIST;
//
// Globals
//
CHAR EscapedCharsA[] = "?*\020<>,^";
WCHAR EscapedCharsW[] = L"?*\020<>,^";
DWORD g_dwErrorStack[MAX_STACK];
DWORD g_dwStackPos = 0;
DHLIST g_DHList[] = {{0xB3, 0xDE, 0x8394},
{0xB6, 0xDE, 0x834B},
{0xB7, 0xDE, 0x834D},
{0xB8, 0xDE, 0x834F},
{0xB9, 0xDE, 0x8351},
{0xBA, 0xDE, 0x8353},
{0xBB, 0xDE, 0x8355},
{0xBC, 0xDE, 0x8357},
{0xBD, 0xDE, 0x8359},
{0xBE, 0xDE, 0x835B},
{0xBF, 0xDE, 0x835D},
{0xC0, 0xDE, 0x835F},
{0xC1, 0xDE, 0x8361},
{0xC2, 0xDE, 0x8364},
{0xC3, 0xDE, 0x8366},
{0xC4, 0xDE, 0x8368},
{0xCA, 0xDE, 0x836F},
{0xCB, 0xDE, 0x8372},
{0xCC, 0xDE, 0x8375},
{0xCD, 0xDE, 0x8378},
{0xCE, 0xDE, 0x837B},
{0xCA, 0xDF, 0x8370},
{0xCB, 0xDF, 0x8373},
{0xCC, 0xDF, 0x8376},
{0xCD, 0xDF, 0x8379},
{0xCE, 0xDF, 0x837C},
{0x00, 0x00, 0x0000}};
extern OUR_CRITICAL_SECTION g_MessageCs; // in main.c
extern PMHANDLE g_TextPool; // in main.c
PGROWBUFFER g_LastAllocTable;
//
// Macro expansion list
//
// None
//
// Private function prototypes
//
BOOL
pTestSetA (
IN MBCHAR ch,
IN PCSTR IncludeSet, OPTIONAL
IN PCSTR ExcludeSet OPTIONAL
);
BOOL
pTestSetW (
IN WCHAR ch,
IN PCWSTR IncludeSet, OPTIONAL
IN PCWSTR ExcludeSet OPTIONAL
);
//
// Macro expansion definition
//
// None
//
// Code
//
/*++
Routine Description:
StringCopy implements lstrcpyA and a UNICODE version. We don't use the Win32
api because of speed and because we want to compile lint-free.
Arguments:
Destination - Receivies the string copy
Source - Specifies the string to copy
Return Value:
None.
--*/
VOID
StringCopyA (
OUT PSTR Destination,
IN PCSTR Source
)
{
PCSTR current = Source;
PCSTR end;
while (*current) {
*Destination++ = *current++;
}
//
// Make sure DBCS string is properly terminated
//
end = current;
current--;
while (current >= Source) {
if (!IsLeadByte (*current)) {
//
// destEnd is correct
//
break;
}
current--;
}
if (!((end - current) & 1)) {
Destination--;
}
*Destination = 0;
}
VOID
StringCopyW (
OUT PWSTR Destination,
IN PCWSTR Source
)
{
while (*Source) {
*Destination++ = *Source++;
}
*Destination = 0;
}
/*++
Routine Description:
StringCopyByteCount implements lstrcpynA and a UNICODE version. We don't
use the Win32 api because of speed and because we want to compile lint-free.
Arguments:
Destination - Receivies the string copy
Source - Specifies the string to copy
Count - Specifies the maximum number of bytes to copy, including the
nul terminator. If Count is zero, then not even a nul
terminator is written.
Return Value:
None.
--*/
VOID
StringCopyByteCountA (
OUT PSTR Destination,
IN PCSTR Source,
IN UINT Count
)
{
PCSTR end;
PCSTR current;
PSTR destEnd;
if (Count >= sizeof (CHAR)) {
current = Source;
destEnd = Destination;
end = (PCSTR) ((PBYTE) Source + Count - sizeof (CHAR));
while (*current && current < end) {
*destEnd++ = *current++;
}
//
// If current has data left, we need to make sure a DBCS string
// is properly terminated.
//
if (*current) {
end = current;
current--;
while (current >= Source) {
if (!IsLeadByte (*current)) {
//
// destEnd is correct
//
break;
}
current--;
}
if (!((end - current) & 1)) {
destEnd--;
}
}
*destEnd = 0;
}
}
VOID
StringCopyByteCountW (
OUT PWSTR Destination,
IN PCWSTR Source,
IN UINT Count
)
{
PCWSTR end;
if (Count < sizeof (WCHAR)) {
DEBUGMSG_IF ((
Count != 0,
DBG_WHOOPS,
"Buffer passed to StringCopyByteCountW is a fraction of one character"
));
return;
}
end = (PCWSTR) ((PBYTE) Source + Count - sizeof (WCHAR));
while (Source < end) {
*Destination++ = *Source++;
}
*Destination = 0;
}
VOID
StringCopyByteCountABA (
OUT PSTR Destination,
IN PCSTR Start,
IN PCSTR End,
IN UINT MaxBytesToCopyIncNul
)
/*++
Routine Description:
StringCopyByteCountAB copies a string segment into a destination buffer,
and limits the copy to a maximum buffer size. The return string is always
nul-terminated, unless the buffer is too small to even hold a nul character.
Arguments:
Destination - Receives the string starting at Start and ending one
character before End.
Start - Specifies the start of the string.
End - Specifies the first character not to copy. If End
is equal or less than Start, then Destination is set
to an empty string (assuming the buffer can hold at
least one character)
MaxBytesToCopyIncNul - Specifies the size of Destination, in bytes, and
including the nul terminator.
Return Value:
None.
--*/
{
INT width;
#ifdef DEBUG
PCSTR check;
check = Start;
while (check < End) {
if (!(*check)) {
DEBUGMSG ((DBG_WHOOPS, "StringCopyByteCountABA: Nul found between start and end"));
break;
}
check++;
}
#endif
width = (INT) (End - Start) + sizeof (CHAR);
if (width > sizeof (CHAR)) {
StringCopyByteCountA (Destination, Start, min ((UINT) width, MaxBytesToCopyIncNul));
} else if (MaxBytesToCopyIncNul >= sizeof (CHAR)) {
*Destination = 0;
}
}
VOID
StringCopyByteCountABW (
OUT PWSTR Destination,
IN PCWSTR Start,
IN PCWSTR End,
IN UINT MaxBytesToCopyIncNul
)
{
INT width;
#ifdef DEBUG
PCWSTR check;
check = Start;
while (check < End) {
if (!(*check)) {
DEBUGMSG ((DBG_WHOOPS, "StringCopyByteCountABW: Nul found between start and end"));
break;
}
check++;
}
#endif
width = (INT) (End - Start) + sizeof (WCHAR);
if (width > sizeof (WCHAR)) {
StringCopyByteCountW (Destination, Start, min ((UINT) width, MaxBytesToCopyIncNul));
} else if (MaxBytesToCopyIncNul >= sizeof (WCHAR)) {
*Destination = 0;
}
}
/*++
Routine Description:
AllocTextEx allocates a block of memory from the specified pool, or g_TextPool
if no pool is specified, and is designated specifically for text processing.
The g_TextPool is initialized when migutil.lib loads up, and there is 64K of
guaranteed workspace, which will grow as necessary.
Arguments:
Pool - Specifies the pool to allocate memory from
CountOfChars - Specifies the number of characters (not bytes) to allocate. The
return pointer is a block of memory that can hold CountOfChars characters,
weather they are SBCS, DBCS or UNICODE.
Return Value:
A pointer to the allocated memory, or NULL if the pool could not be expanded
to hold the number of specified characters.
--*/
PSTR
RealAllocTextExA (
IN PMHANDLE Pool,
IN UINT CountOfChars
)
{
PSTR text;
if (CountOfChars == 0) {
return NULL;
}
if (!Pool) {
Pool = g_TextPool;
}
text = PmGetAlignedMemory (Pool, CountOfChars * sizeof (CHAR) * 2);
text [0] = 0;
return text;
}
PWSTR
RealAllocTextExW (
IN PMHANDLE Pool,
IN UINT CountOfChars
)
{
PWSTR text;
if (CountOfChars == 0) {
return NULL;
}
if (!Pool) {
Pool = g_TextPool;
}
text = PmGetAlignedMemory (Pool, CountOfChars * sizeof (WCHAR));
text [0] = 0;
return text;
}
/*++
Routine Description:
FreeText frees the memory allocated by AllocText. After all strings are freed,
the block will be emptied but not deallocated.
It is important NOT to leak memory, because a leak will cause the pool to
expand, and non-empty pools cause memory fragmentation.
Arguments:
Text - Specifies the text to free, as returned from AllocText, DuplicateText,
DuplicateTextEx, etc...
Return Value:
none
--*/
VOID
FreeTextExA (
IN PMHANDLE Pool, OPTIONAL
IN PCSTR Text OPTIONAL
)
{
if (Text) {
if (!Pool) {
Pool = g_TextPool;
}
PmReleaseMemory (Pool, (PVOID) Text);
}
}
VOID
FreeTextExW (
IN PMHANDLE Pool, OPTIONAL
IN PCWSTR Text OPTIONAL
)
{
if (Text) {
if (!Pool) {
Pool = g_TextPool;
}
PmReleaseMemory (Pool, (PVOID) Text);
}
}
/*++
Routine Description:
DuplicateTextEx duplicates a text string and allocates additional space a
caller needs to complete its processing. Optionally, the caller receives
a pointer to the nul of the duplicated string (to allow more efficient
appends).
Arguments:
Text - Specifies the text to duplicate
ExtraChars - Specifies the number of characters (not bytes) to allocate
space for. The characters can be from the SBCS, DBCS or
UNICODE character sets.
NulChar - Receives a pointer to the nul at the end of the duplicated
string. Use for fast appends.
Return Value:
A pointer to the duplicated and expanded string, or NULL if g_TextPool
could not be expanded to fit the duplicated string and extra characters.
--*/
PSTR
RealDuplicateTextExA (
IN PMHANDLE Pool, OPTIONAL
IN PCSTR Text,
IN UINT ExtraChars,
OUT PSTR *NulChar OPTIONAL
)
{
PSTR buf;
PSTR d;
PCSTR s;
buf = AllocTextExA (Pool, CharCountA (Text) + ExtraChars + 1);
if (buf) {
s = Text;
d = buf;
while (*s) {
if (IsLeadByte (*s)) {
*d++ = *s++;
}
*d++ = *s++;
}
*d = 0;
if (NulChar) {
*NulChar = d;
}
}
return buf;
}
PWSTR
RealDuplicateTextExW (
IN PMHANDLE Pool, OPTIONAL
IN PCWSTR Text,
IN UINT ExtraChars,
OUT PWSTR *NulChar OPTIONAL
)
{
PWSTR buf;
PWSTR d;
PCWSTR s;
buf = AllocTextExW (Pool, CharCountW (Text) + ExtraChars + 1);
if (buf) {
s = Text;
d = buf;
while (*s) {
*d++ = *s++;
}
*d = 0;
if (NulChar) {
*NulChar = d;
}
}
return buf;
}
/*++
Routine Description:
JoinText duplicates String1 and appends String2 to it delimited with the optional delimiterstring.
Arguments:
String1 - Specifies the text to duplciate
String2 - Specifies the text to append to String1
DelimiterString - Optionally specifies the string to place between string 1 and string 2.
ExtraChars - Specifies the number of characters (not bytes) to allocate
space for. The characters can be from the SBCS, DBCS or
UNICODE character sets.
NulChar - Receives a pointer to the nul at the end of the duplicated
string. Use for fast appends.
Return Value:
A pointer to the duplicated string and extra characters.
--*/
PSTR
RealJoinTextExA (
IN PMHANDLE Pool, OPTIONAL
IN PCSTR String1,
IN PCSTR String2,
IN PCSTR CenterString, OPTIONAL
IN UINT ExtraChars,
OUT PSTR *NulChar OPTIONAL
)
{
PSTR buf;
PSTR end;
PSTR d;
PCSTR s;
buf = DuplicateTextExA (
Pool,
String1,
CharCountA (String2) + ExtraChars + (CenterString ? CharCountA (CenterString) : 0),
&end
);
if (buf) {
d = end;
if (CenterString) {
s = CenterString;
while (*s) {
if (IsLeadByte (*s)) {
*d++ = *s++;
}
*d++ = *s++;
}
}
s = String2;
while (*s) {
if (IsLeadByte (*s)) {
*d++ = *s++;
}
*d++ = *s++;
}
*d = 0;
if (NulChar) {
*NulChar = d;
}
}
return buf;
}
PWSTR
RealJoinTextExW (
IN PMHANDLE Pool, OPTIONAL
IN PCWSTR String1,
IN PCWSTR String2,
IN PCWSTR CenterString, OPTIONAL
IN UINT ExtraChars,
OUT PWSTR *NulChar OPTIONAL
)
{
PWSTR buf;
PWSTR end;
PCWSTR s;
PWSTR d;
buf = DuplicateTextExW (
Pool,
String1,
CharCountW (String2) + ExtraChars + (CenterString ? CharCountW(CenterString) : 0),
&end
);
if (buf) {
d = end;
if (CenterString) {
s = CenterString;
while (*s) {
*d++ = *s++;
}
}
s = String2;
while (*s) {
*d++ = *s++;
}
*d = 0;
if (NulChar) {
*NulChar = d;
}
}
return buf;
}
/*++
Routine Description:
ExpandEnvironmentTextEx takes a block of text containing zero or more environment variables
(encoded in %'s) and returns the text with the environment variables expanded. The function
also allows the caller to specify additional environment variables in an array and will use
these variables before calling GetEnvironmentVariable.
The returned text is allocated out of the Text pool and should be freed using FreeText().
Arguments:
InString - The string containing environement variables to be processed.
ExtraVars - Optional var pointing to an array of environment variables to be used to supersede
or suppliment the system environment variables. Even entries in the list are the
names of environment variables, odd entries there values.
(e.g. {"name1","value1","name2","value2",...}
Return Value:
An expanded string.
--*/
PWSTR
RealExpandEnvironmentTextExW (
IN PCWSTR InString,
IN PCWSTR * ExtraVars OPTIONAL
)
{
PWSTR rString = NULL;
PWSTR newString = NULL;
PWSTR envName = NULL;
PWSTR envValue = NULL;
BOOL inSubstitution = FALSE;
BOOL ignoreNextPercent = FALSE;
BOOL errorOccurred = FALSE;
BOOL foundValue = FALSE;
BOOL freeValue = FALSE;
PCWSTR nextPercent = NULL;
PCWSTR source = NULL;
PCWSTR savedSource = NULL;
UINT maxSize = 0;
UINT curSize = 0;
UINT index = 0;
UINT size = 0;
if (!InString) {
return NULL;
}
if (*InString == 0) {
return DuplicateTextW (InString);
}
//
// Set source to the start of InString to begin with...
//
source = InString;
__try {
while (*source) {
//
// Reallocate the string if necessary. We assume that most strings
// are smaller than 1024 chars and that we will therefore only rarely
// reallocate a string.
//
if (curSize + 3 > maxSize) {
maxSize += 1024;
newString = AllocTextW (maxSize);
if (!newString) {
DEBUGMSG((DBG_ERROR,"ExpandEnvironmentTextEx: Memory Error!"));
errorOccurred = TRUE;
__leave;
}
if (rString) {
//lint -e(671)
CopyMemory (newString, rString, (SIZE_T) ((UINT)curSize * sizeof(WCHAR)));
FreeTextW(rString);
}
rString = newString;
}
//
// if we find a percent sign, and we are not currently expanding
// an environment variable (or copying an empty set of %'s),
// then we have probably found an environment variable. Attempt
// to expand it.
//
if (*source == L'%' && !inSubstitution) {
if (ignoreNextPercent) {
ignoreNextPercent = FALSE;
}
else {
ignoreNextPercent = FALSE;
nextPercent = wcschr(source + 1,L'%');
if (nextPercent == source + 1) {
//
// We found two consecutive %s in this string. We'll ignore them and simply copy them as
// normal text.
//
ignoreNextPercent = TRUE;
DEBUGMSGW((DBG_WARNING,"ExpandEnvironmentTextEx: Empty Environment variable in %s. Ignoring.",InString));
}
else if (nextPercent) {
//
// Create a variable to hold the envName.
//
envName = AllocTextW(nextPercent - source);
StringCopyByteCountABW (
envName,
source + 1,
nextPercent,
(UINT) ((UBINT)nextPercent - (UBINT)source)
);
//
// Try to find the variable.
//
foundValue = FALSE;
freeValue = FALSE;
if (ExtraVars) {
//
// Search through the list of extra vars passed in by the caller.
// Even entries of this list are env var names. Odd entries are env values.
// {envname1,envvalue1,envname2,envvalue2,...}
//
index = 0;
while (ExtraVars[index]) {
if (StringIMatchW(ExtraVars[index],envName) && ExtraVars[index + 1]) {
foundValue = TRUE;
envValue = (PWSTR) ExtraVars[index + 1];
break;
}
index +=2;
}
}
if (!foundValue) {
//
// Still haven't found the environment variable. Use GetEnvironmentString.
//
//
size = GetEnvironmentVariableW(envName,NULL,0);
if (!size) {
errorOccurred = TRUE;
DEBUGMSGW((DBG_WARNING,"ExpandEnvironmentTextEx: Environment variable %s not found!",envName));
} else {
//
// Create a buffer large enough to hold this value and copy it in.
//
envValue = AllocTextW(size);
if ((size - 1) != GetEnvironmentVariableW(envName,envValue,size)) {
errorOccurred = TRUE;
DEBUGMSGW((DBG_ERROR,"ExpandEnvironmentTextEx: Error from GetEnvironmentVariable."));
}
else {
foundValue = TRUE;
}
freeValue = TRUE;
}
}
if (foundValue) {
//
// Ok, we have a valid environment value. Need to copy this data over.
// To do this, we update and save the current source into old source, set source = to the envValue,
// and set the inSubstitution value so that we don't attempt to expand any percents within
// the value.
//
savedSource = nextPercent + 1;
source = envValue;
inSubstitution = TRUE;
}
else {
DEBUGMSGW ((DBG_WARNING, "ExpandEnvironmentTextEx: No Environment variable found for %s.", envName));
ignoreNextPercent = TRUE;
}
//
// We are done with the environment name at this time, so clean it up.
//
FreeTextW(envName);
envName = NULL;
}
ELSE_DEBUGMSGW((DBG_WARNING,"ExpandEnvironmentTextEx: No matching percent found in %s. Ignoring.",InString));
}
}
//
// Copy over the current character.
//
rString[curSize++] = *source++; //lint !e613
if (!*source) {
if (inSubstitution) {
//
// The source for the environment variable is fully copied.
// restore the old source.
//
inSubstitution = FALSE;
source = savedSource;
if (!*source) { //lint !e613
rString[curSize] = 0; //lint !e613
}
if (freeValue) {
FreeTextW(envValue);
freeValue = FALSE;
}
envValue = NULL;
}
else {
rString[curSize] = 0; //lint !e613
}
}
}
} //lint !e613
__finally {
DEBUGMSGW_IF (( errorOccurred, DBG_WARNING, "ExpandEnvironmentText: Some errors occurred while processing %s = %s.", InString, rString ? rString : L"NULL"));
if (envName) {
FreeTextW(envName);
}
if (envValue && freeValue) {
FreeTextW(envValue);
}
}
return rString;
}
PSTR
RealExpandEnvironmentTextExA (
IN PCSTR InString,
IN PCSTR * ExtraVars OPTIONAL
)
{
PSTR rString = NULL;
PSTR newString = NULL;
PSTR envName = NULL;
PSTR envValue = NULL;
BOOL inSubstitution = FALSE;
BOOL ignoreNextPercent = FALSE;
BOOL errorOccurred = FALSE;
BOOL foundValue = FALSE;
BOOL freeValue = FALSE;
PCSTR nextPercent = NULL;
PCSTR source = NULL;
PCSTR savedSource = NULL;
UINT maxSize = 0;
UINT curSize = 0;
UINT index = 0;
UINT size = 0;
if (!InString) {
return NULL;
}
if (*InString == 0) {
return DuplicateTextA (InString);
}
//
// Set source to the start of InString to begin with...
//
source = InString;
__try {
while (*source) {
//
// Reallocate the string if necessary. We assume that most strings
// are smaller than 1024 chars and that we will therefore only rarely
// reallocate a string.
//
if (curSize + 3 > maxSize) {
maxSize += 1024;
newString = AllocTextA (maxSize);
if (rString) {
CopyMemory (newString, rString, curSize * sizeof(CHAR)); //lint !e671
FreeTextA(rString);
}
rString = newString;
}
//
// if we find a percent sign, and we are not currently expanding
// an environment variable (or copying an empty set of %'s),
// then we have probably found an environment variable. Attempt
// to expand it.
//
if (!IsLeadByte(*source) && *source == '%' && !inSubstitution) {
if (ignoreNextPercent) {
ignoreNextPercent = FALSE;
}
else {
ignoreNextPercent = FALSE;
nextPercent = _mbschr(source + 1,'%');
if (nextPercent == source + 1) {
//
// We found two consecutive %s in this string. We'll ignore them and simply copy them as
// normal text.
//
ignoreNextPercent = TRUE;
DEBUGMSGA((DBG_WARNING,"ExpandEnvironmentTextEx: Empty Environment variable in %s. Ignoring.",InString));
}
else if (nextPercent) {
//
// Create a variable to hold the envName.
//
envName = AllocTextA(nextPercent - source);
StringCopyABA (envName, source+1, nextPercent);
//
// Try to find the variable.
//
foundValue = FALSE;
freeValue = FALSE;
if (ExtraVars) {
//
// Search through the list of extra vars passed in by the caller.
// Even entries of this list are env var names. Odd entries are env values.
// {envname1,envvalue1,envname2,envvalue2,...}
//
index = 0;
while (ExtraVars[index]) {
if (StringIMatch(ExtraVars[index],envName) && ExtraVars[index + 1]) {
foundValue = TRUE;
envValue = (PSTR) ExtraVars[index + 1];
break;
}
index +=2;
}
}
if (!foundValue) {
//
// Still haven't found the environment variable. Use GetEnvironmentString.
//
//
size = GetEnvironmentVariableA(envName,NULL,0);
if (!size) {
errorOccurred = TRUE;
DEBUGMSGA((DBG_WARNING,"ExpandEnvironmentTextEx: Environment variable %s not found!",envName));
}
else {
//
// Create a buffer large enough to hold this value and copy it in.
//
envValue = AllocTextA(size);
freeValue = TRUE;
if ((size - 1) != GetEnvironmentVariableA(envName,envValue,size)) {
errorOccurred = TRUE;
DEBUGMSGA((DBG_ERROR,"ExpandEnvironmentTextEx: Error from GetEnvironmentVariable."));
}
else {
foundValue = TRUE;
}
}
}
if (foundValue) {
//
// Ok, we have a valid environment value. Need to copy this data over.
// To do this, we update and save the current source into old source, set source = to the envValue,
// and set the inSubstitution value so that we don't attempt to expand any percents within
// the value.
//
savedSource = nextPercent + 1;
source = envValue;
inSubstitution = TRUE;
}
else {
DEBUGMSGA ((DBG_WARNING, "ExpandEnvironmentTextEx: No Environment variable found for %s.", envName));
ignoreNextPercent = TRUE;
}
//
// We are done with the environment name at this time, so clean it up.
//
FreeTextA(envName);
envName = NULL;
}
ELSE_DEBUGMSGA((DBG_WARNING,"ExpandEnvironmentTextEx: No matching percent found in %s. Ignoring.",InString));
}
}
//
// Copy over the current character.
//
if (IsLeadByte(*source)) { //lint !e613
rString[curSize++] = *source++; //lint !e613
}
rString[curSize++] = *source++; //lint !e613
if (!*source) {
if (inSubstitution) {
//
// The source for the environment variable is fully copied.
// restore the old source.
//
inSubstitution = FALSE;
source = savedSource;
if (!*source) { //lint !e613
rString[curSize] = 0; //lint !e613
}
if (freeValue) {
FreeTextA(envValue);
freeValue = FALSE;
}
envValue = NULL;
}
else {
rString[curSize] = 0; //lint !e613
}
}
}
} //lint !e613
__finally {
DEBUGMSGA_IF (( errorOccurred, DBG_WARNING, "ExpandEnvironmentText: Some errors occurred while processing %s = %s.", InString, rString ? rString : "NULL"));
if (envName) {
FreeTextA(envName);
}
if (envValue && freeValue) {
FreeTextA(envValue);
}
}
return rString;
}
/*++
Routine Description:
AppendWack adds a backslash to the end of any string, unless the string
already ends in a backslash.
AppendDosWack adds a backslash, but only if the path does not already
end in a backslash or colon. AppendWack supports DOS naming
conventions: it does not append a back-slash if the path is empty,
ends in a colon or if it ends in a back-slash already.
AppendUncWack supports UNC naming conventions: it does not append a
backslash if the path is empty or if it ends in a backslash already.
AppendPathWack supports both DOS and UNC naming conventions, and uses the
UNC naming convention if the string starts with double-wacks.
Arguments:
Str - A buffer that holds the path, plus additional space for another
backslash.
Return Value:
none
--*/
PSTR
AppendWackA (
IN PSTR Str
)
{
PCSTR last;
if (!Str)
return Str;
last = Str;
while (*Str) {
last = Str;
Str = _mbsinc (Str);
}
if (*last != '\\') {
*Str = '\\';
Str++;
*Str = 0;
}
return Str;
}
PWSTR
AppendWackW (
IN PWSTR Str
)
{
PCWSTR last;
if (!Str)
return Str;
if (*Str) {
Str = GetEndOfStringW (Str);
last = Str - 1;
} else {
last = Str;
}
if (*last != '\\') {
*Str = L'\\';
Str++;
*Str = 0;
}
return Str;
}
PSTR
AppendDosWackA (
IN PSTR Str
)
{
PCSTR last;
if (!Str || !(*Str))
return Str;
do {
last = Str;
Str = _mbsinc (Str);
} while (*Str);
if (*last != '\\' && *last != ':') {
*Str = '\\';
Str++;
*Str = 0;
}
return Str;
}
PWSTR
AppendDosWackW (
IN PWSTR Str
)
{
PWSTR last;
if (!Str || !(*Str))
return Str;
Str = GetEndOfStringW (Str);
last = Str - 1;
if (*last != L'\\' && *last != L':') {
*Str = L'\\';
Str++;
*Str = 0;
}
return Str;
}
PSTR
AppendUncWackA (
IN PSTR Str
)
{
PCSTR last;
if (!Str || !(*Str))
return Str;
do {
last = Str;
Str = _mbsinc (Str);
} while (*Str);
if (*last != '\\') {
*Str = '\\';
Str++;
*Str = 0;
}
return Str;
}
PWSTR
AppendUncWackW (
IN PWSTR Str
)
{
PWSTR last;
if (!Str || !(*Str))
return Str;
Str = GetEndOfStringW (Str);
last = Str - 1;
if (*last != L'\\') {
*Str = L'\\';
Str++;
*Str = 0;
}
return Str;
}
PSTR
AppendPathWackA (
IN PSTR Str
)
{
if (!Str) {
return Str;
}
if (Str[0] == '\\' && Str[1] == '\\') {
return AppendUncWackA (Str);
}
return AppendDosWackA (Str);
}
PWSTR
AppendPathWackW (
IN PWSTR Str
)
{
if (!Str) {
return Str;
}
if (Str[0] == L'\\' && Str[1] == L'\\') {
return AppendUncWackW (Str);
}
return AppendDosWackW (Str);
}
DWORD
pGetStringsTotalSizeA (
IN va_list args
)
{
DWORD size = 0;
PCSTR source;
for (source = va_arg(args, PCSTR); source != NULL; source = va_arg(args, PCSTR)) {
size += ByteCountA (source) + DWSIZEOF(CHAR);
}
return size;
}
DWORD
pGetStringsTotalSizeW (
IN va_list args
)
{
DWORD size = 0;
PCWSTR source;
for (source = va_arg(args, PCWSTR); source != NULL; source = va_arg(args, PCWSTR)) {
size += ByteCountW (source) + DWSIZEOF(WCHAR);
}
return size;
}
PSTR
pJoinPathsInBufferA (
OUT PSTR Buffer,
IN va_list args
)
{
PSTR end;
PSTR endMinusOne;
PCSTR source;
PCSTR p;
INT counter;
*Buffer = 0;
counter = 0;
p = end = Buffer;
for (source = va_arg(args, PCSTR); source != NULL; source = va_arg(args, PCSTR)) {
if (counter > 0) {
endMinusOne = _mbsdec2 (p, end);
if (endMinusOne) {
if (_mbsnextc (source) == '\\') {
if (_mbsnextc (endMinusOne) == '\\') {
source++;
}
} else {
if (_mbsnextc (endMinusOne) != '\\') {
*end = '\\';
end++;
*end = 0;
}
}
}
}
if (*source) {
p = end;
end = StringCatA (end, source);
}
counter++;
}
return end;
}
PWSTR
pJoinPathsInBufferW (
OUT PWSTR Buffer,
IN va_list args
)
{
PWSTR end;
PWSTR endMinusOne;
PCWSTR source;
PCWSTR p;
INT counter;
*Buffer = 0;
counter = 0;
p = end = Buffer;
for (source = va_arg(args, PCWSTR); source != NULL; source = va_arg(args, PCWSTR)) {
if (counter > 0) {
endMinusOne = end > p ? end - 1 : NULL;
if (endMinusOne) {
if (*source == L'\\') {
if (*endMinusOne == L'\\') {
source++;
}
} else {
if (*endMinusOne != L'\\') {
*end = L'\\';
end++;
*end = 0;
}
}
}
}
if (*source) {
p = end;
end = StringCatW (end, source);
}
counter++;
}
return end;
}
PSTR
_cdecl
RealJoinPathsInPoolExA (
IN PMHANDLE Pool, OPTIONAL
...
)
{
DWORD size;
PSTR dest;
va_list args;
if (!Pool) {
Pool = g_PathsPool;
}
va_start (args, Pool);
size = pGetStringsTotalSizeA (args);
va_end (args);
if (size == 0) {
return NULL;
}
dest = (PSTR) PmGetAlignedMemory (Pool, size);
MYASSERT (dest);
va_start (args, Pool);
pJoinPathsInBufferA (dest, args);
va_end (args);
return dest;
}
PWSTR
_cdecl
RealJoinPathsInPoolExW (
IN PMHANDLE Pool, OPTIONAL
...
)
{
DWORD size;
PWSTR dest;
va_list args;
if (!Pool) {
Pool = g_PathsPool;
}
va_start (args, Pool);
size = pGetStringsTotalSizeW (args);
va_end (args);
if (size == 0) {
return NULL;
}
dest = (PWSTR) PmGetAlignedMemory (Pool, size);
MYASSERT (dest);
va_start (args, Pool);
pJoinPathsInBufferW (dest, args);
va_end (args);
return dest;
}
BOOL
JoinPathsExA (
IN OUT PGROWBUFFER Gb,
...
)
{
PSTR end;
DWORD size;
va_list args;
MYASSERT (Gb);
if (!Gb) {
return FALSE;
}
va_start (args, Gb);
size = pGetStringsTotalSizeA (args);
va_end (args);
if (size == 0) {
return FALSE;
}
end = (PSTR) GbGrow (Gb, size);
if (!end) {
return FALSE;
}
va_start (args, Gb);
end = pJoinPathsInBufferA (end, args);
va_end (args);
//
// adjust Gb->End if resulting path is actually shorter than predicted
//
MYASSERT ((PBYTE)end >= Gb->Buf && (PBYTE)(end + 1) <= Gb->Buf + Gb->End);
Gb->End = (DWORD)((PBYTE)(end + 1) - Gb->Buf);
return TRUE;
}
BOOL
JoinPathsExW (
IN OUT PGROWBUFFER Gb,
...
)
{
PWSTR end;
DWORD size;
va_list args;
MYASSERT (Gb);
if (!Gb) {
return FALSE;
}
va_start (args, Gb);
size = pGetStringsTotalSizeW (args);
va_end (args);
if (size == 0) {
return FALSE;
}
end = (PWSTR) GbGrow (Gb, size);
if (!end) {
return FALSE;
}
va_start (args, Gb);
end = pJoinPathsInBufferW (end, args);
va_end (args);
//
// adjust Gb->End if resulting path is actually shorter than predicted
//
MYASSERT ((PBYTE)end >= Gb->Buf && (PBYTE)(end + 1) <= Gb->Buf + Gb->End);
Gb->End = (DWORD)((PBYTE)(end + 1) - Gb->Buf);
return TRUE;
}
PSTR
pBuildPathInBufferA (
OUT PSTR Buffer,
IN va_list args
)
{
PCSTR source;
INT counter;
*Buffer = 0;
counter = 0;
for (source = va_arg(args, PCSTR); source != NULL; source = va_arg(args, PCSTR)) {
if (counter > 0) {
*Buffer++ = '\\';
*Buffer = 0;
}
Buffer = StringCatA (Buffer, source);
counter++;
}
return Buffer;
}
PWSTR
pBuildPathInBufferW (
OUT PWSTR Buffer,
IN va_list args
)
{
PCWSTR source;
INT counter;
*Buffer = 0;
counter = 0;
for (source = va_arg(args, PCWSTR); source != NULL; source = va_arg(args, PCWSTR)) {
if (counter > 0) {
*Buffer++ = L'\\';
*Buffer = 0;
}
Buffer = StringCatW (Buffer, source);
counter++;
}
return Buffer;
}
DWORD
BuildPathA (
OUT PSTR Buffer, OPTIONAL
IN DWORD SizeInBytes, OPTIONAL
...
)
{
PSTR end;
DWORD size;
va_list args;
va_start (args, SizeInBytes);
size = pGetStringsTotalSizeA (args);
va_end (args);
if (!size) {
//
// no args
//
return 0;
}
if (!Buffer) {
return size;
}
if (SizeInBytes < size) {
//
// buffer too small
//
return 0;
}
va_start (args, SizeInBytes);
end = pBuildPathInBufferA (Buffer, args);
va_end (args);
MYASSERT (size == (DWORD)((PBYTE)(end + 1) - (PBYTE)Buffer));
return size;
}
DWORD
BuildPathW (
OUT PWSTR Buffer, OPTIONAL
IN DWORD SizeInBytes, OPTIONAL
...
)
{
PWSTR end;
DWORD size;
va_list args;
va_start (args, SizeInBytes);
size = pGetStringsTotalSizeW (args);
va_end (args);
if (!size) {
//
// no args
//
return 0;
}
if (!Buffer) {
return size;
}
if (SizeInBytes < size) {
//
// buffer too small
//
return 0;
}
va_start (args, SizeInBytes);
end = pBuildPathInBufferW (Buffer, args);
va_end (args);
MYASSERT (size == (DWORD)((PBYTE)(end + 1) - (PBYTE)Buffer));
return size;
}
BOOL
BuildPathExA (
IN OUT PGROWBUFFER Gb,
...
)
{
PSTR end;
DWORD size;
va_list args;
MYASSERT (Gb);
if (!Gb) {
return FALSE;
}
va_start (args, Gb);
size = pGetStringsTotalSizeA (args);
va_end (args);
if (!size) {
//
// no args
//
return FALSE;
}
end = (PSTR) GbGrow (Gb, size);
if (!end) {
return FALSE;
}
va_start (args, Gb);
end = pBuildPathInBufferA (end, args);
va_end (args);
MYASSERT ((PBYTE)(end + 1) == Gb->Buf + Gb->End);
return (size != 0);
}
BOOL
BuildPathExW (
IN OUT PGROWBUFFER Gb,
...
)
{
PWSTR end;
DWORD size;
va_list args;
MYASSERT (Gb);
if (!Gb) {
return FALSE;
}
va_start (args, Gb);
size = pGetStringsTotalSizeW (args);
va_end (args);
if (!size) {
//
// no args
//
return FALSE;
}
end = (PWSTR) GbGrow (Gb, size);
if (!end) {
return FALSE;
}
va_start (args, Gb);
end = pBuildPathInBufferW (end, args);
va_end (args);
MYASSERT ((PBYTE)(end + 1) == Gb->Buf + Gb->End);
return (size != 0);
}
PSTR
RealBuildPathInPoolA (
IN PMHANDLE Pool, OPTIONAL
...
)
{
PSTR dest;
DWORD size;
va_list args;
if (!Pool) {
Pool = g_PathsPool;
}
va_start (args, Pool);
size = pGetStringsTotalSizeA (args);
va_end (args);
if (!size) {
//
// no args
//
return NULL;
}
dest = (PSTR) PmGetAlignedMemory (Pool, size);
MYASSERT (dest);
va_start (args, Pool);
pBuildPathInBufferA (dest, args);
va_end (args);
return dest;
}
PWSTR
RealBuildPathInPoolW (
IN PMHANDLE Pool, OPTIONAL
...
)
{
PWSTR dest;
DWORD size;
va_list args;
if (!Pool) {
Pool = g_PathsPool;
}
va_start (args, Pool);
size = pGetStringsTotalSizeW (args);
va_end (args);
if (!size) {
//
// no args
//
return NULL;
}
dest = (PWSTR) PmGetAlignedMemory (Pool, size);
MYASSERT (dest);
va_start (args, Pool);
pBuildPathInBufferW (dest, args);
va_end (args);
return dest;
}
PSTR
RealAllocPathStringA (
IN DWORD Tchars
)
{
PSTR str;
if (Tchars == 0) {
Tchars = MAX_MBCHAR_PATH;
}
str = (PSTR) PmGetAlignedMemory (g_PathsPool, Tchars);
str [0] = 0;
return str;
}
PWSTR
RealAllocPathStringW (
IN DWORD Tchars
)
{
PWSTR str;
if (Tchars == 0) {
Tchars = MAX_WCHAR_PATH;
}
str = (PWSTR) PmGetAlignedMemory (g_PathsPool, Tchars * sizeof (WCHAR));
str [0] = 0;
return str;
}
VOID
RealSplitPathA (
IN PCSTR Path,
OUT PSTR *DrivePtr,
OUT PSTR *PathPtr,
OUT PSTR *FileNamePtr,
OUT PSTR *ExtPtr
)
{
CHAR drive[_MAX_DRIVE];
CHAR dir[_MAX_DIR];
CHAR fileName[_MAX_FNAME];
CHAR ext[_MAX_EXT];
_splitpath (Path, drive, dir, fileName, ext);
if (DrivePtr) {
*DrivePtr = PmDuplicateStringA (g_PathsPool, drive);
MYASSERT (*DrivePtr);
}
if (PathPtr) {
*PathPtr = PmDuplicateStringA (g_PathsPool, dir);
MYASSERT (*PathPtr);
}
if (FileNamePtr) {
*FileNamePtr = PmDuplicateStringA (g_PathsPool, fileName);
MYASSERT (*FileNamePtr);
}
if (ExtPtr) {
*ExtPtr = PmDuplicateStringA (g_PathsPool, ext);
MYASSERT (*ExtPtr);
}
}
VOID
RealSplitPathW (
IN PCWSTR Path,
OUT PWSTR *DrivePtr,
OUT PWSTR *PathPtr,
OUT PWSTR *FileNamePtr,
OUT PWSTR *ExtPtr
)
{
WCHAR drive[_MAX_DRIVE];
WCHAR dir[_MAX_DIR];
WCHAR fileName[_MAX_FNAME];
WCHAR ext[_MAX_EXT];
_wsplitpath (Path, drive, dir, fileName, ext);
if (DrivePtr) {
*DrivePtr = PmDuplicateStringW (g_PathsPool, drive);
MYASSERT (*DrivePtr);
}
if (PathPtr) {
*PathPtr = PmDuplicateStringW (g_PathsPool, dir);
MYASSERT (*PathPtr);
}
if (FileNamePtr) {
*FileNamePtr = PmDuplicateStringW (g_PathsPool, fileName);
MYASSERT (*FileNamePtr);
}
if (ExtPtr) {
*ExtPtr = PmDuplicateStringW (g_PathsPool, ext);
MYASSERT (*ExtPtr);
}
}
PSTR
RealDuplicatePathStringA (
IN PCSTR Path,
IN DWORD ExtraBytes
)
{
PSTR str;
str = PmGetAlignedMemory (
g_PathsPool,
SizeOfStringA (Path) + ExtraBytes
);
MYASSERT (str);
StringCopyA (str, Path);
return str;
}
PWSTR
RealDuplicatePathStringW (
IN PCWSTR Path,
IN DWORD ExtraBytes
)
{
PWSTR str;
str = PmGetAlignedMemory (
g_PathsPool,
SizeOfStringW (Path) + ExtraBytes
);
MYASSERT (str);
StringCopyW (str, Path);
return str;
}
BOOL
EnumFirstPathExA (
OUT PPATH_ENUMA PathEnum,
IN PCSTR AdditionalPath,
IN PCSTR WinDir,
IN PCSTR SysDir,
IN BOOL IncludeEnvPath
)
{
DWORD bufferSize;
DWORD pathSize;
PSTR currPathEnd;
if (PathEnum == NULL) {
return FALSE;
}
bufferSize = pathSize = GetEnvironmentVariable (TEXT("PATH"), NULL, 0);
bufferSize *= 2;
if (AdditionalPath != NULL) {
bufferSize += SizeOfStringA (AdditionalPath);
}
if (SysDir != NULL) {
bufferSize += SizeOfStringA (SysDir);
}
if (WinDir != NULL) {
bufferSize += SizeOfStringA (WinDir);
}
PathEnum->BufferPtr = HeapAlloc (g_hHeap, 0, bufferSize);
if (PathEnum->BufferPtr == NULL) {
return FALSE;
}
PathEnum->BufferPtr [0] = 0;
if (AdditionalPath != NULL) {
StringCopyA (PathEnum->BufferPtr, AdditionalPath);
StringCatA (PathEnum->BufferPtr, ";");
}
if (SysDir != NULL) {
StringCatA (PathEnum->BufferPtr, SysDir);
StringCatA (PathEnum->BufferPtr, ";");
}
if (WinDir != NULL) {
StringCatA (PathEnum->BufferPtr, WinDir);
StringCatA (PathEnum->BufferPtr, ";");
}
if (IncludeEnvPath) {
currPathEnd = GetEndOfStringA (PathEnum->BufferPtr);
GetEnvironmentVariable (TEXT("PATH"), currPathEnd, pathSize);
}
PathEnum->PtrNextPath = PathEnum-> BufferPtr;
return EnumNextPathA (PathEnum);
}
BOOL
EnumNextPathA (
IN OUT PPATH_ENUMA PathEnum
)
{
PSTR currPathEnd;
if (PathEnum->PtrNextPath == NULL) {
AbortPathEnumA (PathEnum);
return FALSE;
}
PathEnum->PtrCurrPath = PathEnum->PtrNextPath;
PathEnum->PtrNextPath = _mbschr (PathEnum->PtrNextPath, ';');
if (PathEnum->PtrNextPath == NULL) {
return TRUE;
}
currPathEnd = PathEnum->PtrNextPath;
PathEnum->PtrNextPath = _mbsinc (PathEnum->PtrNextPath);
*currPathEnd = 0;
if (*(PathEnum->PtrNextPath) == 0) {
PathEnum->PtrNextPath = NULL;
}
if (*(PathEnum->PtrCurrPath) == 0) {
//
// We found an empty path segment. Skip it.
//
return EnumNextPathA (PathEnum);
}
return TRUE;
}
BOOL
AbortPathEnumA (
IN OUT PPATH_ENUMA PathEnum
)
{
if (PathEnum->BufferPtr != NULL) {
HeapFree (g_hHeap, 0, PathEnum->BufferPtr);
PathEnum->BufferPtr = NULL;
}
return TRUE;
}
VOID
FreePathStringExA (
IN PMHANDLE Pool, OPTIONAL
IN PCSTR Path OPTIONAL
)
{
if (Path) {
if (!Pool) {
Pool = g_PathsPool;
}
PmReleaseMemory (Pool, (PSTR) Path);
}
}
VOID
FreePathStringExW (
IN PMHANDLE Pool, OPTIONAL
IN PCWSTR Path OPTIONAL
)
{
if (Path) {
if (!Pool) {
Pool = g_PathsPool;
}
PmReleaseMemory (Pool, (PWSTR) Path);
}
}
/*++
Routine Description:
PushError and PopError push the error code onto a stack or pull the
last pushed error code off the stack. PushError uses GetLastError
and PopError uses SetLastError to modify the last error value.
Arguments:
none
Return Value:
none
--*/
VOID
PushNewError (DWORD dwError)
{
if (g_dwStackPos == MAX_STACK)
return;
g_dwErrorStack[g_dwStackPos] = dwError;
g_dwStackPos++;
}
VOID
PushError (VOID)
{
if (g_dwStackPos == MAX_STACK)
return;
g_dwErrorStack[g_dwStackPos] = GetLastError ();
g_dwStackPos++;
}
DWORD
PopError (VOID)
{
if (!g_dwStackPos)
return GetLastError();
g_dwStackPos--;
SetLastError (g_dwErrorStack[g_dwStackPos]);
return g_dwErrorStack[g_dwStackPos];
}
/*++
Routine Description:
GetHexDigit is a simple base 16 ASCII to int convertor. The
convertor is case-insensitive.
Arguments:
c - Character to convert
Return Value:
Base 16 value corresponding to character supplied, or -1 if
the character is not 0-9, A-F or a-f.
--*/
int
GetHexDigit (IN int c)
{
if (c >= '0' && c <= '9')
return (c - '0');
c = towlower ((wint_t) c);
if (c >= 'a' && c <= 'f')
return (c - 'a' + 10);
return -1;
}
/*++
Routine Description:
_tcsnum is similar to strtoul, except is figures out which base
the number should be calculated from. It supports decimal and
hexadecimal numbers (using the 0x00 notation). The return
value is the decoded value, or 0 if a syntax error was found.
Arguments:
szNum - Pointer to the string holding the number. This number
can be either decimal (a series of 0-9 characters), or
hexadecimal (a series of 0-9, A-F or a-f characters,
prefixed with 0x or 0X).
Return Value:
The decoded unsigned long value, or zero if a syntax error was
found.
--*/
DWORD
_mbsnum (IN PCSTR szNum)
{
unsigned int d = 0;
int i;
if (szNum[0] == '0' && tolower (szNum[1]) == 'x') {
// Get hex value
szNum += 2;
while ((i = GetHexDigit ((int) *szNum)) != -1) {
d = d * 16 + (UINT)i;
szNum++;
}
}
else {
// Get decimal value
while (*szNum >= '0' && *szNum <= '9') {
d = d * 10 + (*szNum - '0');
szNum++;
}
}
return d;
}
DWORD
_wcsnum (
IN PCWSTR szNum
)
{
unsigned int d = 0;
int i;
if (szNum[0] == L'0' && towlower (szNum[1]) == L'x') {
// Get hex value
szNum += 2;
while ((i = GetHexDigit ((int) *szNum)) != -1) {
d = d * 16 + (UINT)i;
szNum++;
}
}
else {
// Get decimal value
while (*szNum >= L'0' && *szNum <= L'9') {
d = d * 10 + (*szNum - L'0');
szNum++;
}
}
return d;
}
/*++
Routine Description:
StringCat is a lstrcat-type routine. It returns the pointer to the end
of a string instead of the beginning, is faster, and has the proper types
to keep lint happy.
Arguments:
Destination - A pointer to a caller-allocated buffer that may point
anywhere within the string to append to
Source - A pointer to a string that is appended to Destination
Return Value:
A pointer to the NULL terminator within the Destination string.
--*/
PSTR
StringCatA (
OUT PSTR Destination,
IN PCSTR Source
)
{
PCSTR current = Source;
PCSTR end;
//
// Advance Destination to end of string
//
Destination = GetEndOfStringA (Destination);
while (*current) {
*Destination++ = *current++; //lint !e613
}
//
// Make sure DBCS string is properly terminated
//
end = current;
current--;
while (current >= Source) {
if (!IsLeadByte (*current)) {
//
// destEnd is correct
//
break;
}
current--;
}
if (!((end - current) & 1)) {
Destination--; //lint !e794
}
*Destination = 0; //lint !e794
return Destination;
}
PWSTR
StringCatW (
OUT PWSTR Destination,
IN PCWSTR Source
)
{
//
// Advance Destination to end of string
//
Destination = GetEndOfStringW (Destination);
//
// Copy string
//
while (*Source) {
*Destination++ = *Source++;
}
*Destination = 0;
return Destination;
}
/*++
Routine Description:
_tcsistr is a case-insensitive version of _tcsstr.
Arguments:
szStr - A pointer to the larger string, which may hold szSubStr
szSubStr - A pointer to a string that may be enclosed in szStr
Return Value:
A pointer to the first occurance of szSubStr in szStr, or NULL if
no match is found.
--*/
PCSTR
_mbsistr (PCSTR mbstrStr, PCSTR mbstrSubStr)
{
PCSTR mbstrStart, mbstrStrPos, mbstrSubStrPos;
PCSTR mbstrEnd;
mbstrEnd = (PSTR) ((LPBYTE) mbstrStr + ByteCountA (mbstrStr) - ByteCountA (mbstrSubStr));
for (mbstrStart = mbstrStr ; mbstrStart <= mbstrEnd ; mbstrStart = _mbsinc (mbstrStart)) {
mbstrStrPos = mbstrStart;
mbstrSubStrPos = mbstrSubStr;
while (*mbstrSubStrPos &&
_mbctolower ((MBCHAR) _mbsnextc (mbstrSubStrPos)) == _mbctolower ((MBCHAR) _mbsnextc (mbstrStrPos)))
{
mbstrStrPos = _mbsinc (mbstrStrPos);
mbstrSubStrPos = _mbsinc (mbstrSubStrPos);
}
if (!(*mbstrSubStrPos))
return mbstrStart;
}
return NULL;
}
PCWSTR
_wcsistr (PCWSTR wstrStr, PCWSTR wstrSubStr)
{
PCWSTR wstrStart, wstrStrPos, wstrSubStrPos;
PCWSTR wstrEnd;
wstrEnd = (PWSTR) ((LPBYTE) wstrStr + ByteCountW (wstrStr) - ByteCountW (wstrSubStr));
for (wstrStart = wstrStr ; wstrStart <= wstrEnd ; wstrStart++) {
wstrStrPos = wstrStart;
wstrSubStrPos = wstrSubStr;
while (*wstrSubStrPos &&
towlower (*wstrSubStrPos) == towlower (*wstrStrPos))
{
wstrStrPos++;
wstrSubStrPos++;
}
if (!(*wstrSubStrPos))
return wstrStart;
}
return NULL;
}
/*++
Routine Description:
StringCompareAB compares a string against a string between to string
pointers
Arguments:
String - Specifies the string to compare
Start - Specifies the start of the string to compare against
end - Specifies the end of the string to compare against. The character
pointed to by End is not included in the comparision.
Return Value:
Less than zero: String is numerically less than the string between Start and
End
Zero: String matches the string between Start and End identically
Greater than zero: String is numerically greater than the string between
Start and End
--*/
INT
StringCompareABA (
IN PCSTR String,
IN PCSTR Start,
IN PCSTR End
)
{
while (*String && Start < End) {
if (_mbsnextc (String) != _mbsnextc (Start)) {
break;
}
String = _mbsinc (String);
Start = _mbsinc (Start);
}
if (Start == End && *String == 0) {
return 0;
}
return (INT) (_mbsnextc (Start) - _mbsnextc (String));
}
INT
StringCompareABW (
IN PCWSTR String,
IN PCWSTR Start,
IN PCWSTR End
)
{
while (*String && Start < End) {
if (*String != *Start) {
break;
}
String++;
Start++;
}
if (Start == End && *String == 0) {
return 0;
}
return *Start - *String;
}
BOOL
StringMatchA (
IN PCSTR String1,
IN PCSTR String2
)
/*++
Routine Description:
StringMatchA is an optimized string compare. Usually a comparison is used to
see if two strings are identical, and the numeric releationships aren't
important. This routine exploits that fact and does a byte-by-byte compare.
Arguments:
String1 - Specifies the first string to compare
String2 - Specifies the second string to compare
Return Value:
TRUE if the strings match identically, FALSE otherwise.
--*/
{
while (*String1) {
if (*String1 != *String2) {
return FALSE;
}
String1++;
String2++;
}
if (*String2) {
return FALSE;
}
return TRUE;
}
BOOL
StringMatchABA (
IN PCSTR String,
IN PCSTR Start,
IN PCSTR End
)
/*++
Routine Description:
StringMatchABA is an optimized string compare. Usually a comparison is
used to see if two strings are identical, and the numeric releationships
aren't important. This routine exploits that fact and does a byte-by-byte
compare.
Arguments:
String - Specifies the first string to compare
Start - Specifies the beginning of the second string to compare
End - Specifies the end of the second string to compare (points to one
character beyond the last valid character of the second string)
Return Value:
TRUE if the strings match identically, FALSE otherwise. If End is equal
or less than Start, the return value is always TRUE.
--*/
{
while (*String && Start < End) {
if (*String != *Start) {
return FALSE;
}
String++;
Start++;
}
if (Start < End && *Start) {
return FALSE;
}
return TRUE;
}
INT
StringICompareABA (
IN PCSTR String,
IN PCSTR Start,
IN PCSTR End
)
{
while (*String && Start < End) {
if (tolower ((INT)(_mbsnextc (String))) != tolower ((INT)(_mbsnextc (Start)))) {
break;
}
String = _mbsinc (String);
Start = _mbsinc (Start);
}
if (Start == End && *String == 0) {
return 0;
}
return (tolower ((INT)(_mbsnextc (Start))) - tolower ((INT)(_mbsnextc (String))));
}
INT
StringICompareABW (
IN PCWSTR String,
IN PCWSTR Start,
IN PCWSTR End
)
{
while (*String && Start < End) {
if (towlower (*String) != towlower (*Start)) {
break;
}
String++;
Start++;
}
if (Start == End && *String == 0) {
return 0;
}
return towlower (*Start) - towlower (*String);
}
VOID
_setmbchar (
IN OUT PSTR Str,
IN MBCHAR c
)
/*++
Routine Description:
_setmbchar sets the character at the specified string position, shifting
bytes if necessary to keep the string in tact.
Arguments:
Str - String
c - Character to set
Return Value:
none
--*/
{
if (c < 256) {
if (IsLeadByte (*Str)) {
//
// Delete one byte from the string
//
MoveMemory (Str, Str+1, SizeOfStringA (Str+2) + 1);
}
*Str = (CHAR) c;
} else {
if (!IsLeadByte (*Str)) {
//
// Insert one byte in the string
//
MoveMemory (Str+1, Str, SizeOfStringA (Str));
}
*((WORD *) Str) = (WORD) c;
}
}
/*++
Routine Description:
GetNextRuleChar extracts the first character in the *p_szRule string,
and determines the character value, decoding the ~xx~ syntax (which
specifies any arbitrary value).
GetNextRuleChar returns a complete character for SBCS and UNICODE, but
it may return either a lead byte or non-lead byte for MBCS. To indicate
a MBCS character, two ~xx~ hex values are needed.
Arguments:
p_szRule - A pointer to a pointer; a caller-allocated buffer that
holds the rule string.
p_bFromHex - A pointer to a caller-allocated BOOL that receives TRUE
when the return value was decoded from the <xx> syntax.
Return Value:
The decoded character; *p_bFromHex identifies if the return value was
a literal or was a hex-encoded character.
--*/
MBCHAR
GetNextRuleCharA (
IN OUT PCSTR *PtrToRule,
OUT BOOL *FromHex
)
{
MBCHAR ch;
MBCHAR Value;
INT i;
PCSTR StartPtr;
StartPtr = *PtrToRule;
if (FromHex) {
*FromHex = FALSE;
}
if (_mbsnextc (StartPtr) == '~') {
*PtrToRule += 1;
Value = 0;
i = 0;
for (i = 0 ; **PtrToRule && i < 8 ; i++) {
ch = _mbsnextc (*PtrToRule);
*PtrToRule += 1;
if (ch == '~') {
if (FromHex) {
*FromHex = TRUE;
}
return Value;
}
Value *= 16;
if (ch >= '0' && ch <= '9') {
Value += ch - '0';
} else if (ch >= 'a' && ch <= 'f') {
Value += ch - 'a' + 10;
} else if (ch >= 'A' && ch <= 'F') {
Value += ch - 'A' + 10;
} else {
break;
}
}
DEBUGMSGA ((DBG_WHOOPS, "Bad formatting in encoded string %s", StartPtr));
}
*PtrToRule = _mbsinc (StartPtr);
return _mbsnextc (StartPtr);
}
WCHAR
GetNextRuleCharW (
IN OUT PCWSTR *PtrToRule,
OUT BOOL *FromHex
)
{
WCHAR ch;
WCHAR Value;
INT i;
PCWSTR StartPtr;
StartPtr = *PtrToRule;
if (FromHex) {
*FromHex = FALSE;
}
if (*StartPtr == L'~') {
*PtrToRule += 1;
Value = 0;
i = 0;
for (i = 0 ; **PtrToRule && i < 8 ; i++) {
ch = **PtrToRule;
*PtrToRule += 1;
if (ch == L'~') {
if (FromHex) {
*FromHex = TRUE;
}
return Value;
}
Value *= 16;
if (ch >= L'0' && ch <= L'9') {
Value += ch - L'0';
} else if (ch >= L'a' && ch <= L'f') {
Value += ch - L'a' + 10;
} else if (ch >= L'A' && ch <= L'F') {
Value += ch - L'A' + 10;
} else {
break;
}
}
DEBUGMSGW ((DBG_WHOOPS, "Bad formatting in encoded string %s", StartPtr));
}
*PtrToRule = StartPtr + 1;
return *StartPtr;
}
/*++
Routine Description:
DecodeRuleChars takes a complete rule string (szRule), possibly
encoded with hex-specified character values (~xx~). The output
string contains unencoded characters.
Arguments:
szRule - A caller-allocated buffer, big enough to hold an
unencoded rule. szRule can be equal to szEncRule.
szEncRule - The string holding a possibly encoded string.
Return Value:
Equal to szRule.
--*/
PSTR
DecodeRuleCharsA (PSTR mbstrRule, PCSTR mbstrEncRule)
{
MBCHAR c;
PSTR mbstrOrgRule;
mbstrOrgRule = mbstrRule;
//
// Copy string, converting ~xx~ to a single char
//
do {
c = GetNextRuleCharA (&mbstrEncRule, NULL);
*mbstrRule = (CHAR) c;
mbstrRule++; // MBCS->incomplete char will be finished in next loop iteration
} while (c);
return mbstrOrgRule;
}
PWSTR
DecodeRuleCharsW (PWSTR wstrRule, PCWSTR wstrEncRule)
{
WCHAR c;
PWSTR wstrOrgRule;
wstrOrgRule = wstrRule;
//
// Copy string, converting ~xx~ to a single char
//
do {
c = GetNextRuleCharW (&wstrEncRule, NULL);
*wstrRule = c;
wstrRule++;
} while (c);
return wstrOrgRule;
}
PSTR
DecodeRuleCharsABA (PSTR mbstrRule, PCSTR mbstrEncRule, PCSTR End)
{
MBCHAR c;
PSTR mbstrOrgRule;
mbstrOrgRule = mbstrRule;
//
// Copy string, converting ~xx~ to a single char
//
while (mbstrEncRule < End) {
c = GetNextRuleCharA (&mbstrEncRule, NULL);
*mbstrRule = (CHAR) c;
mbstrRule++; // MBCS->incomplete char will be finished in next loop iteration
}
*mbstrRule = 0;
return mbstrOrgRule;
}
PWSTR
DecodeRuleCharsABW (PWSTR wstrRule, PCWSTR wstrEncRule, PCWSTR End)
{
WCHAR c;
PWSTR wstrOrgRule;
wstrOrgRule = wstrRule;
//
// Copy string, converting ~xx~ to a single char
//
while (wstrEncRule < End) {
c = GetNextRuleCharW (&wstrEncRule, NULL);
*wstrRule = c;
wstrRule++;
}
*wstrRule = 0;
return wstrOrgRule;
}
/*++
Routine Description:
EncodeRuleChars takes an unencoded rule string (szRule), and
converts it to a string possibly encoded with hex-specified
character values (~xx~). The output string contains encoded
characters.
Arguments:
szEncRule - A caller-allocated buffer, big enough to hold an
encoded rule. szEncRule CAN NOT be equal to szRule.
One way to calculate a max buffer size for szEncRule
is to use the following code:
allocsize = SizeOfString (szRule) * 6;
In the worst case, each character in szRule will take
six single-byte characters in szEncRule. In the normal
case, szEncRule will only be a few bytes bigger than
szRule.
szRule - The string holding an unencoded string.
Return Value:
Equal to szEncRule.
--*/
PSTR
EncodeRuleCharsA (PSTR mbstrEncRule, PCSTR mbstrRule)
{
PSTR mbstrOrgRule;
static CHAR mbstrExclusions[] = "[]<>\'*$|:?\";,%";
MBCHAR c;
mbstrOrgRule = mbstrEncRule;
while (*mbstrRule) {
c = _mbsnextc (mbstrRule);
if (!_ismbcprint (c) || _mbschr (mbstrExclusions, c)) {
// Escape unprintable or excluded character
wsprintfA (mbstrEncRule, "~%X~", c);
mbstrEncRule = GetEndOfStringA (mbstrEncRule);
mbstrRule = _mbsinc (mbstrRule);
}
else {
// Copy multibyte character
if (IsLeadByte (*mbstrRule)) {
*mbstrEncRule = *mbstrRule;
mbstrEncRule++;
mbstrRule++;
}
*mbstrEncRule = *mbstrRule;
mbstrEncRule++;
mbstrRule++;
}
}
*mbstrEncRule = 0; //lint !e613
return mbstrOrgRule;
}
PWSTR
EncodeRuleCharsW (PWSTR wstrEncRule, PCWSTR wstrRule)
{
PWSTR wstrOrgRule;
static WCHAR wstrExclusions[] = L"[]<>\'*$|:?\";,%";
WCHAR c;
wstrOrgRule = wstrEncRule;
while (c = *wstrRule) { //lint !e720
if (!iswprint (c) || wcschr (wstrExclusions, c)) {
wsprintfW (wstrEncRule, L"~%X~", c);
wstrEncRule = GetEndOfStringW (wstrEncRule);
}
else {
*wstrEncRule = *wstrRule;
wstrEncRule++;
}
wstrRule++;
}
*wstrEncRule = 0;
return wstrOrgRule;
}
/*++
Routine Description:
_tcsisprint is a string version of _istprint.
Arguments:
szStr - A pointer to the string to examine
Return Value:
Non-zero if szStr is made up only of printable characters.
--*/
int
_mbsisprint (PCSTR mbstrStr)
{
while (*mbstrStr && _ismbcprint ((MBCHAR) _mbsnextc (mbstrStr))) {
mbstrStr = _mbsinc (mbstrStr);
}
return *mbstrStr == 0;
}
int
_wcsisprint (PCWSTR wstrStr)
{
while (*wstrStr && iswprint (*wstrStr)) {
wstrStr++;
}
return *wstrStr == 0;
}
/*++
Routine Description:
SkipSpace returns a pointer to the next position within a string
that does not have whitespace characters. It uses the C
runtime isspace to determine what a whitespace character is.
Arguments:
szStr - A pointer to the string to examine
Return Value:
A pointer to the first non-whitespace character in the string,
or NULL if the string is made up of all whitespace characters
or the string is empty.
--*/
PCSTR
SkipSpaceA (PCSTR mbstrStr)
{
while (_ismbcspace ((MBCHAR) _mbsnextc (mbstrStr)))
mbstrStr = _mbsinc (mbstrStr);
return mbstrStr;
}
PCWSTR
SkipSpaceW (PCWSTR wstrStr)
{
while (iswspace (*wstrStr))
wstrStr++;
return wstrStr;
}
/*++
Routine Description:
SkipSpaceR returns a pointer to the next position within a string
that does not have whitespace characters. It uses the C
runtime isspace to determine what a whitespace character is.
This function is identical to SkipSpace except it works from
right to left instead of left to right.
Arguments:
StrBase - A pointer to the first character in the string
Str - A pointer to the end of the string, or NULL if the
end is not known.
Return Value:
A pointer to the first non-whitespace character in the string,
as viewed from right to left, or NULL if the string is made up
of all whitespace characters or the string is empty.
--*/
PCSTR
SkipSpaceRA (
IN PCSTR StrBase,
IN PCSTR Str OPTIONAL
)
{
if (!Str) {
Str = GetEndOfStringA (StrBase);
}
if (*Str == 0) { //lint !e613
Str = _mbsdec2 (StrBase, Str);
if (!Str) {
return NULL;
}
}
do {
if (!_ismbcspace((MBCHAR) _mbsnextc(Str))) {
return Str;
}
} while (Str = _mbsdec2(StrBase, Str)); //lint !e720
return NULL;
}
PCWSTR
SkipSpaceRW (
IN PCWSTR StrBase,
IN PCWSTR Str OPTIONAL
)
{
if (!Str) {
Str = GetEndOfStringW (StrBase);
}
if (*Str == 0) {
Str--;
if (Str < StrBase) {
return NULL;
}
}
do {
if (!iswspace(*Str)) {
return Str;
}
} while (Str-- != StrBase);
return NULL;
}
/*++
Routine Description:
TruncateTrailingSpace trims the specified string after the
very last non-space character, or empties the string if it
contains only space characters. This routine uses isspace
to determine what a space is.
Arguments:
Str - Specifies string to process
Return Value:
none
--*/
VOID
TruncateTrailingSpaceA (
IN OUT PSTR Str
)
{
PSTR LastNonSpace;
PSTR OrgStr;
OrgStr = Str;
LastNonSpace = NULL;
while (*Str) {
if (!_ismbcspace ((MBCHAR) _mbsnextc (Str))) {
LastNonSpace = Str;
}
Str = _mbsinc (Str);
}
if (LastNonSpace) {
*_mbsinc (LastNonSpace) = 0;
} else {
*OrgStr = 0;
}
}
VOID
TruncateTrailingSpaceW (
IN OUT PWSTR Str
)
{
PWSTR LastNonSpace;
PWSTR OrgStr;
OrgStr = Str;
LastNonSpace = NULL;
while (*Str) {
if (!iswspace (*Str)) {
LastNonSpace = Str;
}
Str++;
}
if (LastNonSpace) {
*(LastNonSpace + 1) = 0;
} else {
*OrgStr = 0;
}
}
/*++
Routine Description:
IsPatternMatch compares a string against a pattern that may contain
standard * or ? wildcards.
Arguments:
wstrPattern - A pattern possibly containing wildcards
wstrStr - The string to compare against the pattern
Return Value:
TRUE when wstrStr and wstrPattern match when wildcards are expanded.
FALSE if wstrStr does not match wstrPattern.
--*/
BOOL
IsPatternMatchA (
IN PCSTR strPattern,
IN PCSTR strStr
)
{
MBCHAR chSrc, chPat;
while (*strStr) {
chSrc = _mbctolower ((MBCHAR) _mbsnextc (strStr));
chPat = _mbctolower ((MBCHAR) _mbsnextc (strPattern));
if (chPat == '*') {
// Skip all asterisks that are grouped together
while (_mbsnextc (_mbsinc (strStr)) == '*') {
strStr = _mbsinc (strStr);
}
// Check if asterisk is at the end. If so, we have a match already.
if (!_mbsnextc (_mbsinc (strPattern))) {
return TRUE;
}
// do recursive check for rest of pattern
if (IsPatternMatchA (_mbsinc (strPattern), strStr)) {
return TRUE;
}
// Allow any character and continue
strStr = _mbsinc (strStr);
continue;
}
if (chPat != '?') {
if (chSrc != chPat) {
return FALSE;
}
}
strStr = _mbsinc (strStr);
strPattern = _mbsinc (strPattern);
}
//
// Fail when there is more pattern and pattern does not end in an asterisk
//
while (_mbsnextc (strPattern) == '*') {
strPattern = _mbsinc (strPattern);
}
if (_mbsnextc (strPattern)) {
return FALSE;
}
return TRUE;
}
BOOL
IsPatternMatchW (
IN PCWSTR wstrPattern,
IN PCWSTR wstrStr
)
{
WCHAR chSrc, chPat;
while (*wstrStr) {
chSrc = towlower (*wstrStr);
chPat = towlower (*wstrPattern);
if (chPat == L'*') {
// Skip all asterisks that are grouped together
while (wstrPattern[1] == L'*')
wstrPattern++;
// Check if asterisk is at the end. If so, we have a match already.
chPat = towlower (wstrPattern[1]);
if (!chPat)
return TRUE;
// Otherwise check if next pattern char matches current char
if (chPat == chSrc || chPat == L'?') {
// do recursive check for rest of pattern
wstrPattern++;
if (IsPatternMatchW (wstrPattern, wstrStr))
return TRUE;
// no, that didn't work, stick with star
wstrPattern--;
}
//
// Allow any character and continue
//
wstrStr++;
continue;
}
if (chPat != L'?') {
//
// if next pattern character is not a question mark, src and pat
// must be identical.
//
if (chSrc != chPat)
return FALSE;
}
//
// Advance when pattern character matches string character
//
wstrPattern++;
wstrStr++;
}
//
// Fail when there is more pattern and pattern does not end in an asterisk
//
chPat = *wstrPattern;
if (chPat && (chPat != L'*' || wstrPattern[1]))
return FALSE;
return TRUE;
}
BOOL
IsPatternContainedA (
IN PCSTR Container,
IN PCSTR Contained
)
{
MBCHAR chSrc, chPat;
while (*Contained) {
chSrc = _mbctolower ((MBCHAR) _mbsnextc (Contained));
chPat = _mbctolower ((MBCHAR) _mbsnextc (Container));
if (chPat == '*') {
// Skip all asterisks that are grouped together
while (_mbsnextc (_mbsinc (Container)) == '*') {
Container = _mbsinc (Container);
}
// Check if asterisk is at the end. If so, we have a match already.
if (!_mbsnextc (_mbsinc (Container))) {
return TRUE;
}
// do recursive check for rest of pattern
if (IsPatternContainedA (_mbsinc (Container), Contained)) {
return TRUE;
}
// Allow any character and continue
Contained = _mbsinc (Contained);
continue;
} else if (chPat == '?') {
if (chSrc == '*') {
return FALSE;
}
} else {
if (chSrc != chPat) {
return FALSE;
}
}
Contained = _mbsinc (Contained);
Container = _mbsinc (Container);
}
//
// Fail when there is more pattern and pattern does not end in an asterisk
//
while (_mbsnextc (Container) == '*') {
Container = _mbsinc (Container);
}
if (_mbsnextc (Container)) {
return FALSE;
}
return TRUE;
}
BOOL
IsPatternContainedW (
IN PCWSTR Container,
IN PCWSTR Contained
)
{
while (*Contained) {
if (*Container == L'*') {
// Skip all asterisks that are grouped together
while (Container[1] == L'*') {
Container++;
}
// Check if asterisk is at the end. If so, we have a match already.
if (!Container[1]) {
return TRUE;
}
// do recursive check for rest of pattern
if (IsPatternContainedW (Container + 1, Contained)) {
return TRUE;
}
// Allow any character and continue
Contained++;
continue;
} else if (*Container == L'?') {
if (*Contained == L'*') {
return FALSE;
}
} else {
if (*Container != *Contained) {
return FALSE;
}
}
Contained++;
Container++;
}
//
// Fail when there is more pattern and pattern does not end in an asterisk
//
while (*Container == '*') {
Container++;
}
if (*Container) {
return FALSE;
}
return TRUE;
}
/*++
Routine Description:
IsPatternMatchAB compares a string against a pattern that may contain
standard * or ? wildcards. It only processes the string up to the
specified end.
Arguments:
Pattern - A pattern possibly containing wildcards
Start - The string to compare against the pattern
End - Specifies the end of Start
Return Value:
TRUE when the string between Start and End matches Pattern when wildcards are expanded.
FALSE if the pattern does not match.
--*/
BOOL
IsPatternMatchABA (
IN PCSTR Pattern,
IN PCSTR Start,
IN PCSTR End
)
{
MBCHAR chSrc, chPat;
while (*Start && Start < End) {
chSrc = _mbctolower ((MBCHAR) _mbsnextc (Start));
chPat = _mbctolower ((MBCHAR) _mbsnextc (Pattern));
if (chPat == '*') {
// Skip all asterisks that are grouped together
while (_mbsnextc (_mbsinc (Start)) == '*') {
Start = _mbsinc (Start);
}
// Check if asterisk is at the end. If so, we have a match already.
if (!_mbsnextc (_mbsinc (Pattern))) {
return TRUE;
}
// do recursive check for rest of pattern
if (IsPatternMatchABA (_mbsinc (Pattern), Start, End)) {
return TRUE;
}
// Allow any character and continue
Start = _mbsinc (Start);
continue;
}
if (chPat != '?') {
if (chSrc != chPat) {
return FALSE;
}
}
Start = _mbsinc (Start);
Pattern = _mbsinc (Pattern);
}
//
// Fail when there is more pattern and pattern does not end in an asterisk
//
while (_mbsnextc (Pattern) == '*') {
Pattern = _mbsinc (Pattern);
}
if (_mbsnextc (Pattern)) {
return FALSE;
}
return TRUE;
}
BOOL
IsPatternMatchABW (
IN PCWSTR Pattern,
IN PCWSTR Start,
IN PCWSTR End
)
{
WCHAR chSrc, chPat;
while (*Start && Start < End) {
chSrc = towlower (*Start);
chPat = towlower (*Pattern);
if (chPat == L'*') {
// Skip all asterisks that are grouped together
while (Pattern[1] == L'*') {
Pattern++;
}
// Check if asterisk is at the end. If so, we have a match already.
chPat = towlower (Pattern[1]);
if (!chPat) {
return TRUE;
}
// Otherwise check if next pattern char matches current char
if (chPat == chSrc || chPat == L'?') {
// do recursive check for rest of pattern
Pattern++;
if (IsPatternMatchABW (Pattern, Start, End)) {
return TRUE;
}
// no, that didn't work, stick with star
Pattern--;
}
//
// Allow any character and continue
//
Start++;
continue;
}
if (chPat != L'?') {
//
// if next pattern character is not a question mark, src and pat
// must be identical.
//
if (chSrc != chPat) {
return FALSE;
}
}
//
// Advance when pattern character matches string character
//
Pattern++;
Start++;
}
//
// Fail when there is more pattern and pattern does not end in an asterisk
//
chPat = *Pattern;
if (chPat && (chPat != L'*' || Pattern[1])) {
return FALSE;
}
return TRUE;
}
/*++
Routine Description:
IsPatternMatchEx compares a string against a pattern that may contain
any of the following expressions:
* - Specifies zero or more characters
? - Specifies any one character
*[set] - Specifies zero or more characters in set
?[set] - Specifies any one character in set
*[n:set] - Specifies zero to n characters in set
?[n:set] - Specifies exactly n characters in set
*[!(set)] - Specifies zero or more characters not in set
?[!(set)] - Specifies one character not in set
*[n:!(set)] - Specifies zero to n characters not in set
?[n:!(set)] - Specifies exactly n characters not in set
*[set1,!(set2)] - Specifies zero or more characters in set1 and
not in set2. It is assumed that set1 and set2
overlap.
?[set1,!(set2)] - Specifies one character in set1 and not in set2.
*[n:set1,!(set2)] - Specifies zero to n characters in set1 and not
in set 2.
?[n:set1,!(set2)] - Specifies exactly n characters in set1 and not
in set 2.
set, set1 and set2 are specified as follows:
a - Specifies a single character
a-b - Specifies a character range
a,b - Specifies two characters
a-b,c-d - Specifies two character ranges
a,b-c - Specifies a single character and a character range
etc...
Patterns can be joined by surrounding the entire expression in
greater than/less than braces.
Because of the syntax characters, the following characters must be
escaped by preceeding the character with a caret (^):
^? ^[ ^- ^< ^! ^^
^* ^] ^: ^> ^,
Here are some examples:
To specify any GUID:
{?[8:0-9,a-f]-?[4:0-9,a-f]-?[4:0-9,a-f]-?[4:0-9,a-f]-?[12:0-9,a-f]}
To specify a 32-bit hexadecimal number:
<0x*[8:0-9,a-f]><0*[7:0-9,a-f]h><?[1-9]*[7:0-9,a-f]h>
Arguments:
Pattern - A pattern possibly containing wildcards
Start - The string to compare against the pattern
End - Specifies the end of Start
Return Value:
TRUE when the string between Start and End matches Pattern when wildcards are expanded.
FALSE if the pattern does not match.
--*/
BOOL
IsPatternMatchExA (
IN PCSTR Pattern,
IN PCSTR String
)
{
PPARSEDPATTERNA Handle;
BOOL b;
Handle = CreateParsedPatternA (Pattern);
if (!Handle) {
return FALSE;
}
b = TestParsedPatternA (Handle, String);
DestroyParsedPatternA (Handle);
return b;
}
BOOL
IsPatternMatchExW (
IN PCWSTR Pattern,
IN PCWSTR String
)
{
PPARSEDPATTERNW Handle;
BOOL b;
Handle = CreateParsedPatternW (Pattern);
if (!Handle) {
return FALSE;
}
b = TestParsedPatternW (Handle, String);
DestroyParsedPatternW (Handle);
return b;
}
/*++
Routine Description:
IsPatternMatchExAB compares a string against a pattern that may contain
any of the following expressions:
* - Specifies zero or more characters
? - Specifies any one character
*[set] - Specifies zero or more characters in set
?[set] - Specifies any one character in set
*[n:set] - Specifies zero to n characters in set
?[n:set] - Specifies exactly n characters in set
*[!(set)] - Specifies zero or more characters not in set
?[!(set)] - Specifies one character not in set
*[n:!(set)] - Specifies zero to n characters not in set
?[n:!(set)] - Specifies exactly n characters not in set
*[set1,!(set2)] - Specifies zero or more characters in set1 and
not in set2. It is assumed that set1 and set2
overlap.
?[set1,!(set2)] - Specifies one character in set1 and not in set2.
*[n:set1,!(set2)] - Specifies zero to n characters in set1 and not
in set 2.
?[n:set1,!(set2)] - Specifies exactly n characters in set1 and not
in set 2.
set, set1 and set2 are specified as follows:
a - Specifies a single character
a-b - Specifies a character range
a,b - Specifies two characters
a-b,c-d - Specifies two character ranges
a,b-c - Specifies a single character and a character range
etc...
Patterns can be joined by surrounding the entire expression in
greater than/less than braces.
Because of the syntax characters, the following characters must be
escaped by preceeding the character with a caret (^):
^? ^[ ^- ^< ^! ^^
^* ^] ^: ^> ^,
Here are some examples:
To specify any GUID:
{?[8:0-9,a-f]-?[4:0-9,a-f]-?[4:0-9,a-f]-?[4:0-9,a-f]-?[12:0-9,a-f]}
To specify a 32-bit hexadecimal number:
<0x*[8:0-9,a-f]><0*[7:0-9,a-f]h><?[1-9]*[7:0-9,a-f]h>
Arguments:
Pattern - A pattern possibly containing wildcards
Start - The string to compare against the pattern
End - Specifies the end of Start
Return Value:
TRUE when the string between Start and End matches Pattern when wildcards are expanded.
FALSE if the pattern does not match.
--*/
BOOL
IsPatternMatchExABA (
IN PCSTR Pattern,
IN PCSTR Start,
IN PCSTR End
)
{
PPARSEDPATTERNA Handle;
BOOL b;
Handle = CreateParsedPatternA (Pattern);
if (!Handle) {
return FALSE;
}
b = TestParsedPatternABA (Handle, Start, End);
DestroyParsedPatternA (Handle);
return b;
}
BOOL
IsPatternMatchExABW (
IN PCWSTR Pattern,
IN PCWSTR Start,
IN PCWSTR End
)
{
PPARSEDPATTERNW Handle;
BOOL b;
Handle = CreateParsedPatternW (Pattern);
if (!Handle) {
return FALSE;
}
b = TestParsedPatternABW (Handle, Start, End);
DestroyParsedPatternW (Handle);
return b;
}
BOOL
pTestSetsA (
IN PCSTR Container,
IN PCSTR Contained,
IN BOOL ExcludeMode
)
{
MBCHAR ch;
if (ExcludeMode) {
if (!Contained) {
return TRUE;
}
if (!Container) {
return FALSE;
}
} else {
if (!Container) {
return TRUE;
}
if (!Contained) {
return FALSE;
}
}
while (*Contained) {
ch = _mbsnextc (Contained);
if (!pTestSetA (ch, Container, NULL)) {
return FALSE;
}
Contained = _mbsinc (Contained);
}
return TRUE;
}
BOOL
pTestSetsW (
IN PCWSTR Container,
IN PCWSTR Contained,
IN BOOL ExcludeMode
)
{
if (ExcludeMode) {
if (!Contained) {
return TRUE;
}
if (!Container) {
return FALSE;
}
} else {
if (!Container) {
return TRUE;
}
if (!Contained) {
return FALSE;
}
}
while (*Contained) {
if (!pTestSetW (*Contained, Container, NULL)) {
return FALSE;
}
Contained ++;
}
return TRUE;
}
BOOL
pMatchSegmentA (
IN PSEGMENTA Source,
IN PSEGMENTA Destination
)
{
switch (Source->Type) {
case SEGMENTTYPE_OPTIONAL:
switch (Destination->Type) {
case SEGMENTTYPE_OPTIONAL:
if (Source->Wildcard.MaxLen) {
if ((Destination->Wildcard.MaxLen == 0) ||
(Source->Wildcard.MaxLen < Destination->Wildcard.MaxLen)
) {
return FALSE;
}
}
if (!pTestSetsA (
Source->Wildcard.IncludeSet,
Destination->Wildcard.IncludeSet,
FALSE
)) {
return FALSE;
}
if (!pTestSetsA (
Destination->Wildcard.ExcludeSet,
Source->Wildcard.ExcludeSet,
TRUE
)) {
return FALSE;
}
return TRUE;
case SEGMENTTYPE_REQUIRED:
if (Source->Wildcard.MaxLen) {
if (Source->Wildcard.MaxLen < Destination->Wildcard.MaxLen) {
return FALSE;
}
}
if (!pTestSetsA (
Source->Wildcard.IncludeSet,
Destination->Wildcard.IncludeSet,
FALSE
)) {
return FALSE;
}
if (!pTestSetsA (
Destination->Wildcard.ExcludeSet,
Source->Wildcard.ExcludeSet,
TRUE
)) {
return FALSE;
}
return TRUE;
case SEGMENTTYPE_EXACTMATCH:
if (!pTestSetA (
_mbsnextc (Destination->Exact.LowerCasePhrase),
Source->Wildcard.IncludeSet,
Source->Wildcard.ExcludeSet
)) {
return FALSE;
}
return TRUE;
default:
return FALSE;
}
break;
case SEGMENTTYPE_REQUIRED:
switch (Destination->Type) {
case SEGMENTTYPE_OPTIONAL:
return FALSE;
case SEGMENTTYPE_REQUIRED:
if (!pTestSetsA (
Source->Wildcard.IncludeSet,
Destination->Wildcard.IncludeSet,
FALSE
)) {
return FALSE;
}
if (!pTestSetsA (
Destination->Wildcard.ExcludeSet,
Source->Wildcard.ExcludeSet,
TRUE
)) {
return FALSE;
}
return TRUE;
case SEGMENTTYPE_EXACTMATCH:
if (!pTestSetA (
_mbsnextc (Destination->Exact.LowerCasePhrase),
Source->Wildcard.IncludeSet,
Source->Wildcard.ExcludeSet
)) {
return FALSE;
}
return TRUE;
default:
return FALSE;
}
break;
case SEGMENTTYPE_EXACTMATCH:
switch (Destination->Type) {
case SEGMENTTYPE_OPTIONAL:
return FALSE;
case SEGMENTTYPE_REQUIRED:
return FALSE;
case SEGMENTTYPE_EXACTMATCH:
if (_mbsnextc (Destination->Exact.LowerCasePhrase) != _mbsnextc (Source->Exact.LowerCasePhrase)) {
return FALSE;
}
return TRUE;
default:
return FALSE;
}
break;
default:
return FALSE;
}
return FALSE;
}
BOOL
pMatchSegmentW (
IN PSEGMENTW Source,
IN PSEGMENTW Destination
)
{
switch (Source->Type) {
case SEGMENTTYPE_OPTIONAL:
switch (Destination->Type) {
case SEGMENTTYPE_OPTIONAL:
if (Source->Wildcard.MaxLen) {
if ((Destination->Wildcard.MaxLen == 0) ||
(Source->Wildcard.MaxLen < Destination->Wildcard.MaxLen)
) {
return FALSE;
}
}
if (!pTestSetsW (
Source->Wildcard.IncludeSet,
Destination->Wildcard.IncludeSet,
FALSE
)) {
return FALSE;
}
if (!pTestSetsW (
Destination->Wildcard.ExcludeSet,
Source->Wildcard.ExcludeSet,
TRUE
)) {
return FALSE;
}
return TRUE;
case SEGMENTTYPE_REQUIRED:
if (Source->Wildcard.MaxLen) {
if (Source->Wildcard.MaxLen < Destination->Wildcard.MaxLen) {
return FALSE;
}
}
if (!pTestSetsW (
Source->Wildcard.IncludeSet,
Destination->Wildcard.IncludeSet,
FALSE
)) {
return FALSE;
}
if (!pTestSetsW (
Destination->Wildcard.ExcludeSet,
Source->Wildcard.ExcludeSet,
TRUE
)) {
return FALSE;
}
return TRUE;
case SEGMENTTYPE_EXACTMATCH:
if (!pTestSetW (
*Destination->Exact.LowerCasePhrase,
Source->Wildcard.IncludeSet,
Source->Wildcard.ExcludeSet
)) {
return FALSE;
}
return TRUE;
default:
return FALSE;
}
break;
case SEGMENTTYPE_REQUIRED:
switch (Destination->Type) {
case SEGMENTTYPE_OPTIONAL:
return FALSE;
case SEGMENTTYPE_REQUIRED:
if (!pTestSetsW (
Source->Wildcard.IncludeSet,
Destination->Wildcard.IncludeSet,
FALSE
)) {
return FALSE;
}
if (!pTestSetsW (
Destination->Wildcard.ExcludeSet,
Source->Wildcard.ExcludeSet,
TRUE
)) {
return FALSE;
}
return TRUE;
case SEGMENTTYPE_EXACTMATCH:
if (!pTestSetW (
*Destination->Exact.LowerCasePhrase,
Source->Wildcard.IncludeSet,
Source->Wildcard.ExcludeSet
)) {
return FALSE;
}
return TRUE;
default:
return FALSE;
}
break;
case SEGMENTTYPE_EXACTMATCH:
switch (Destination->Type) {
case SEGMENTTYPE_OPTIONAL:
return FALSE;
case SEGMENTTYPE_REQUIRED:
return FALSE;
case SEGMENTTYPE_EXACTMATCH:
if (*Destination->Exact.LowerCasePhrase != *Source->Exact.LowerCasePhrase) {
return FALSE;
}
return TRUE;
default:
return FALSE;
}
break;
default:
return FALSE;
}
return FALSE;
}
BOOL
pIsOneParsedPatternContainedA (
IN PPATTERNPROPSA Container,
IN UINT StartContainer,
IN PPATTERNPROPSA Contained,
IN UINT StartContained
)
{
UINT indexContainer = StartContainer;
UINT indexContained = StartContained;
PSEGMENTA containerSeg, containedSeg;
if (StartContainer == Container->SegmentCount) {
return FALSE;
}
while (indexContained < Contained->SegmentCount) {
containerSeg = &Container->Segment [indexContainer];
containedSeg = &Contained->Segment [indexContained];
if (containerSeg->Type == SEGMENTTYPE_OPTIONAL) {
// see if we can match contained segment
if (!pMatchSegmentA (containerSeg, containedSeg)) {
indexContainer ++;
if (indexContainer == Container->SegmentCount) {
return FALSE;
}
continue;
}
if (pIsOneParsedPatternContainedA (
Container,
indexContainer + 1,
Contained,
indexContained
)) {
return TRUE;
}
indexContained ++;
continue;
} else if (containerSeg->Type == SEGMENTTYPE_REQUIRED) {
if (!pMatchSegmentA (containerSeg, containedSeg)) {
return FALSE;
}
} else {
if (!pMatchSegmentA (containerSeg, containedSeg)) {
return FALSE;
}
}
indexContainer ++;
indexContained ++;
}
//
// Fail when there is more pattern and pattern does not end in an asterisk
//
while (indexContainer < Container->SegmentCount) {
containerSeg = &Container->Segment [indexContainer];
if (containerSeg->Type != SEGMENTTYPE_OPTIONAL) {
return FALSE;
}
indexContainer ++;
}
return TRUE;
}
BOOL
pIsOneParsedPatternContainedW (
IN PPATTERNPROPSW Container,
IN UINT StartContainer,
IN PPATTERNPROPSW Contained,
IN UINT StartContained
)
{
UINT indexContainer = StartContainer;
UINT indexContained = StartContained;
PSEGMENTW containerSeg, containedSeg;
if (StartContainer == Container->SegmentCount) {
return FALSE;
}
while (indexContained < Contained->SegmentCount) {
containerSeg = &Container->Segment [indexContainer];
containedSeg = &Contained->Segment [indexContained];
if (containerSeg->Type == SEGMENTTYPE_OPTIONAL) {
// see if we can match contained segment
if (!pMatchSegmentW (containerSeg, containedSeg)) {
indexContainer ++;
if (indexContainer == Container->SegmentCount) {
return FALSE;
}
continue;
}
if (pIsOneParsedPatternContainedW (
Container,
indexContainer + 1,
Contained,
indexContained
)) {
return TRUE;
}
indexContained ++;
continue;
} else if (containerSeg->Type == SEGMENTTYPE_REQUIRED) {
if (!pMatchSegmentW (containerSeg, containedSeg)) {
return FALSE;
}
} else {
if (!pMatchSegmentW (containerSeg, containedSeg)) {
return FALSE;
}
}
indexContainer ++;
indexContained ++;
}
//
// Fail when there is more pattern and pattern does not end in an asterisk
//
while (indexContainer < Container->SegmentCount) {
containerSeg = &Container->Segment [indexContainer];
if (containerSeg->Type != SEGMENTTYPE_OPTIONAL) {
return FALSE;
}
indexContainer ++;
}
return TRUE;
}
BOOL
IsExplodedParsedPatternContainedExA (
IN PPARSEDPATTERNA Container,
IN PPARSEDPATTERNA Contained
)
{
UINT u1, u2;
BOOL b = FALSE;
for (u1 = 0 ; u1 < Contained->PatternCount ; u1++) {
b = FALSE;
for (u2 = 0 ; u2 < Container->PatternCount ; u2++) {
b = pIsOneParsedPatternContainedA (
&Container->Pattern[u2],
0,
&Contained->Pattern[u1],
0
);
if (b) break;
}
if (!b) {
return FALSE;
}
}
return TRUE;
}
BOOL
IsExplodedParsedPatternContainedExW (
IN PPARSEDPATTERNW Container,
IN PPARSEDPATTERNW Contained
)
{
UINT u1, u2;
BOOL b = FALSE;
for (u1 = 0 ; u1 < Contained->PatternCount ; u1++) {
b = FALSE;
for (u2 = 0 ; u2 < Container->PatternCount ; u2++) {
b = pIsOneParsedPatternContainedW (
&Container->Pattern[u2],
0,
&Contained->Pattern[u1],
0
);
if (b) break;
}
if (!b) {
return FALSE;
}
}
return TRUE;
}
PPARSEDPATTERNA
ExplodeParsedPatternA (
IN PPARSEDPATTERNA Pattern
)
{
PMHANDLE pool;
PPARSEDPATTERNA pattern;
PPATTERNPROPSA oldProps, newProps;
PSEGMENTA oldSeg, newSeg;
UINT i, j, k, newPropsSize, charCountTmp, oldSegIndex, byteIndex;
BOOL result = TRUE;
pool = PmCreateNamedPool ("Parsed Pattern");
__try {
pattern = (PPARSEDPATTERNA) PmGetAlignedMemory (pool, sizeof (PARSEDPATTERNA));
ZeroMemory (pattern, sizeof (PARSEDPATTERNA));
pattern->PatternCount = Pattern->PatternCount;
pattern->Pool = pool;
pattern->Pattern = (PPATTERNPROPSA) PmGetAlignedMemory (
pool,
pattern->PatternCount * sizeof (PATTERNPROPSA)
);
for (i=0; i<pattern->PatternCount; i++) {
oldProps = &Pattern->Pattern[i];
newProps = &pattern->Pattern[i];
ZeroMemory (newProps, sizeof (PATTERNPROPSA));
// now let's walk oldProps to see how many segments we are
// going to need
newPropsSize = 0;
for (j=0; j<oldProps->SegmentCount; j++) {
oldSeg = &oldProps->Segment[j];
switch (oldSeg->Type) {
case SEGMENTTYPE_EXACTMATCH:
charCountTmp = CharCountA (oldSeg->Exact.LowerCasePhrase);
newPropsSize += (charCountTmp?charCountTmp:1);
break;
case SEGMENTTYPE_REQUIRED:
newPropsSize += oldSeg->Wildcard.MaxLen;
break;
case SEGMENTTYPE_OPTIONAL:
newPropsSize ++;
break;
default:
result = FALSE;
__leave;
}
}
// now we allocate the required segments
newProps->SegmentCount = newPropsSize;
newProps->Segment = (PSEGMENTA) PmGetAlignedMemory (
pool,
newProps->SegmentCount * sizeof (SEGMENTA)
);
// now let's walk oldProps again and fill newProps segments.
k = 0;
newSeg = &newProps->Segment[k];
for (j=0; j<oldProps->SegmentCount; j++) {
oldSeg = &oldProps->Segment[j];
ZeroMemory (newSeg, sizeof (SEGMENTA));
switch (oldSeg->Type) {
case SEGMENTTYPE_EXACTMATCH:
oldSegIndex = CharCountA (oldSeg->Exact.LowerCasePhrase);
byteIndex = oldSeg->Exact.PhraseBytes;
if (!oldSegIndex) {
ZeroMemory (newSeg, sizeof (SEGMENTA));
newSeg->Type = oldSeg->Type;
newSeg->Exact.LowerCasePhrase = (PCSTR) PmGetAlignedMemory (
pool, sizeof(CHAR)
);
((PSTR)newSeg->Exact.LowerCasePhrase) [0] = 0;
newSeg->Exact.PhraseBytes = 0;
} else {
while (oldSegIndex) {
ZeroMemory (newSeg, sizeof (SEGMENTA));
newSeg->Type = oldSeg->Type;
newSeg->Exact.LowerCasePhrase = (PCSTR) PmGetAlignedMemory (
pool, 3 * sizeof(CHAR)
);
if (IsLeadByte (oldSeg->Exact.LowerCasePhrase [oldSeg->Exact.PhraseBytes - byteIndex])) {
((PSTR)newSeg->Exact.LowerCasePhrase)[0] = oldSeg->Exact.LowerCasePhrase [oldSeg->Exact.PhraseBytes - byteIndex];
byteIndex --;
((PSTR)newSeg->Exact.LowerCasePhrase)[1] = oldSeg->Exact.LowerCasePhrase [oldSeg->Exact.PhraseBytes - byteIndex];
byteIndex --;
((PSTR)newSeg->Exact.LowerCasePhrase)[2] = 0;
newSeg->Exact.PhraseBytes = 2;
} else {
((PSTR)newSeg->Exact.LowerCasePhrase)[0] = oldSeg->Exact.LowerCasePhrase [oldSeg->Exact.PhraseBytes - byteIndex];
byteIndex --;
((PSTR)newSeg->Exact.LowerCasePhrase)[1] = 0;
newSeg->Exact.PhraseBytes = 1;
}
oldSegIndex --;
k++;
newSeg = &newProps->Segment[k];
}
}
break;
case SEGMENTTYPE_REQUIRED:
oldSegIndex = oldSeg->Wildcard.MaxLen;
while (oldSegIndex) {
ZeroMemory (newSeg, sizeof (SEGMENTA));
newSeg->Type = oldSeg->Type;
newSeg->Wildcard.MaxLen = 1;
if (oldSeg->Wildcard.IncludeSet) {
newSeg->Wildcard.IncludeSet = PmDuplicateStringA (pool, oldSeg->Wildcard.IncludeSet);
}
if (oldSeg->Wildcard.ExcludeSet) {
newSeg->Wildcard.ExcludeSet = PmDuplicateStringA (pool, oldSeg->Wildcard.ExcludeSet);
}
oldSegIndex --;
k++;
newSeg = &newProps->Segment[k];
}
break;
case SEGMENTTYPE_OPTIONAL:
ZeroMemory (newSeg, sizeof (SEGMENTA));
newSeg->Type = oldSeg->Type;
newSeg->Wildcard.MaxLen = oldSeg->Wildcard.MaxLen;
if (oldSeg->Wildcard.IncludeSet) {
newSeg->Wildcard.IncludeSet = PmDuplicateStringA (pool, oldSeg->Wildcard.IncludeSet);
}
if (oldSeg->Wildcard.ExcludeSet) {
newSeg->Wildcard.ExcludeSet = PmDuplicateStringA (pool, oldSeg->Wildcard.ExcludeSet);
}
k++;
newSeg = &newProps->Segment[k];
break;
default:
result = FALSE;
__leave;
}
}
}
}
__finally {
if (!result) {
PmDestroyPool (pool);
pattern = NULL;
}
}
return pattern;
}
PPARSEDPATTERNW
ExplodeParsedPatternW (
IN PPARSEDPATTERNW Pattern
)
{
PMHANDLE pool;
PPARSEDPATTERNW pattern;
PPATTERNPROPSW oldProps, newProps;
PSEGMENTW oldSeg, newSeg;
UINT i, j, k, newPropsSize, charCountTmp, oldSegIndex;
BOOL result = TRUE;
pool = PmCreateNamedPool ("Parsed Pattern");
__try {
pattern = (PPARSEDPATTERNW) PmGetAlignedMemory (pool, sizeof (PARSEDPATTERNW));
ZeroMemory (pattern, sizeof (PARSEDPATTERNW));
pattern->PatternCount = Pattern->PatternCount;
pattern->Pool = pool;
pattern->Pattern = (PPATTERNPROPSW) PmGetAlignedMemory (
pool,
pattern->PatternCount * sizeof (PATTERNPROPSW)
);
for (i=0; i<pattern->PatternCount; i++) {
oldProps = &Pattern->Pattern[i];
newProps = &pattern->Pattern[i];
ZeroMemory (newProps, sizeof (PATTERNPROPSW));
// now let's walk oldProps to see how many segments we are
// going to need
newPropsSize = 0;
for (j=0; j<oldProps->SegmentCount; j++) {
oldSeg = &oldProps->Segment[j];
switch (oldSeg->Type) {
case SEGMENTTYPE_EXACTMATCH:
charCountTmp = CharCountW (oldSeg->Exact.LowerCasePhrase);
newPropsSize += (charCountTmp?charCountTmp:1);
break;
case SEGMENTTYPE_REQUIRED:
newPropsSize += oldSeg->Wildcard.MaxLen;
break;
case SEGMENTTYPE_OPTIONAL:
newPropsSize ++;
break;
default:
result = FALSE;
__leave;
}
}
// now we allocate the required segments
newProps->SegmentCount = newPropsSize;
newProps->Segment = (PSEGMENTW) PmGetAlignedMemory (
pool,
newProps->SegmentCount * sizeof (SEGMENTW)
);
// now let's walk oldProps again and fill newProps segments.
k = 0;
newSeg = &newProps->Segment[k];
for (j=0; j<oldProps->SegmentCount; j++) {
oldSeg = &oldProps->Segment[j];
ZeroMemory (newSeg, sizeof (SEGMENTW));
switch (oldSeg->Type) {
case SEGMENTTYPE_EXACTMATCH:
oldSegIndex = CharCountW (oldSeg->Exact.LowerCasePhrase);
if (!oldSegIndex) {
ZeroMemory (newSeg, sizeof (SEGMENTA));
newSeg->Type = oldSeg->Type;
newSeg->Exact.LowerCasePhrase = (PCWSTR) PmGetAlignedMemory (
pool, sizeof(WCHAR)
);
((PWSTR)newSeg->Exact.LowerCasePhrase) [0] = 0;
newSeg->Exact.PhraseBytes = 0;
} else {
while (oldSegIndex) {
ZeroMemory (newSeg, sizeof (SEGMENTW));
newSeg->Type = oldSeg->Type;
newSeg->Exact.LowerCasePhrase = (PCWSTR) PmGetAlignedMemory (
pool, 2 * sizeof(WCHAR)
);
((PWSTR)newSeg->Exact.LowerCasePhrase)[0] = oldSeg->Exact.LowerCasePhrase [(oldSeg->Exact.PhraseBytes / sizeof(WCHAR)) - oldSegIndex];
((PWSTR)newSeg->Exact.LowerCasePhrase)[1] = 0;
oldSegIndex --;
k++;
newSeg = &newProps->Segment[k];
}
}
break;
case SEGMENTTYPE_REQUIRED:
oldSegIndex = oldSeg->Wildcard.MaxLen;
while (oldSegIndex) {
ZeroMemory (newSeg, sizeof (SEGMENTW));
newSeg->Type = oldSeg->Type;
newSeg->Wildcard.MaxLen = 1;
if (oldSeg->Wildcard.IncludeSet) {
newSeg->Wildcard.IncludeSet = PmDuplicateStringW (pool, oldSeg->Wildcard.IncludeSet);
}
if (oldSeg->Wildcard.ExcludeSet) {
newSeg->Wildcard.ExcludeSet = PmDuplicateStringW (pool, oldSeg->Wildcard.ExcludeSet);
}
oldSegIndex --;
k++;
newSeg = &newProps->Segment[k];
}
break;
case SEGMENTTYPE_OPTIONAL:
ZeroMemory (newSeg, sizeof (SEGMENTW));
newSeg->Type = oldSeg->Type;
newSeg->Wildcard.MaxLen = oldSeg->Wildcard.MaxLen;
if (oldSeg->Wildcard.IncludeSet) {
newSeg->Wildcard.IncludeSet = PmDuplicateStringW (pool, oldSeg->Wildcard.IncludeSet);
}
if (oldSeg->Wildcard.ExcludeSet) {
newSeg->Wildcard.ExcludeSet = PmDuplicateStringW (pool, oldSeg->Wildcard.ExcludeSet);
}
k++;
newSeg = &newProps->Segment[k];
break;
default:
result = FALSE;
__leave;
}
}
}
}
__finally {
if (!result) {
PmDestroyPool (pool);
pattern = NULL;
}
}
return pattern;
}
/*++
Routine Description:
IsPatternContainedEx compares two patterns to see if one of them is
included in the other. Both patterns may contain any of the following
expressions:
* - Specifies zero or more characters
? - Specifies any one character
*[set] - Specifies zero or more characters in set
?[set] - Specifies any one character in set
*[n:set] - Specifies zero to n characters in set
?[n:set] - Specifies exactly n characters in set
*[!(set)] - Specifies zero or more characters not in set
?[!(set)] - Specifies one character not in set
*[n:!(set)] - Specifies zero to n characters not in set
?[n:!(set)] - Specifies exactly n characters not in set
*[set1,!(set2)] - Specifies zero or more characters in set1 and
not in set2. It is assumed that set1 and set2
overlap.
?[set1,!(set2)] - Specifies one character in set1 and not in set2.
*[n:set1,!(set2)] - Specifies zero to n characters in set1 and not
in set 2.
?[n:set1,!(set2)] - Specifies exactly n characters in set1 and not
in set 2.
set, set1 and set2 are specified as follows:
a - Specifies a single character
a-b - Specifies a character range
a,b - Specifies two characters
a-b,c-d - Specifies two character ranges
a,b-c - Specifies a single character and a character range
etc...
Patterns can be joined by surrounding the entire expression in
greater than/less than braces.
Because of the syntax characters, the following characters must be
escaped by preceeding the character with a caret (^):
^? ^[ ^- ^< ^! ^^
^* ^] ^: ^> ^,
Here are some examples:
To specify any GUID:
{?[8:0-9,a-f]-?[4:0-9,a-f]-?[4:0-9,a-f]-?[4:0-9,a-f]-?[12:0-9,a-f]}
To specify a 32-bit hexadecimal number:
<0x*[8:0-9,a-f]><0*[7:0-9,a-f]h><?[1-9]*[7:0-9,a-f]h>
Arguments:
Container - A container pattern possibly containing wildcards
Contained - A contained pattern possibly containing wildcards
Return Value:
TRUE when the second pattern is contained in the first one, FALSE if not.
--*/
BOOL
IsPatternContainedExA (
IN PCSTR Container,
IN PCSTR Contained
)
{
PPARSEDPATTERNA container = NULL, contained = NULL;
PPARSEDPATTERNA expContainer = NULL, expContained = NULL;
BOOL result = FALSE;
__try {
container = CreateParsedPatternA (Container);
if (!container) {
__leave;
}
expContainer = ExplodeParsedPatternA (container);
if (!expContainer) {
__leave;
}
contained = CreateParsedPatternA (Contained);
if (!contained) {
__leave;
}
expContained = ExplodeParsedPatternA (contained);
if (!expContained) {
__leave;
}
result = IsExplodedParsedPatternContainedExA (expContainer, expContained);
}
__finally {
if (expContained) {
DestroyParsedPatternA (expContained);
}
if (contained) {
DestroyParsedPatternA (contained);
}
if (expContainer) {
DestroyParsedPatternA (expContainer);
}
if (container) {
DestroyParsedPatternA (container);
}
}
return result;
}
BOOL
IsPatternContainedExW (
IN PCWSTR Container,
IN PCWSTR Contained
)
{
PPARSEDPATTERNW container = NULL, contained = NULL;
PPARSEDPATTERNW expContainer = NULL, expContained = NULL;
BOOL result = FALSE;
__try {
container = CreateParsedPatternW (Container);
if (!container) {
__leave;
}
expContainer = ExplodeParsedPatternW (container);
if (!expContainer) {
__leave;
}
contained = CreateParsedPatternW (Contained);
if (!contained) {
__leave;
}
expContained = ExplodeParsedPatternW (contained);
if (!expContained) {
__leave;
}
result = IsExplodedParsedPatternContainedExW (expContainer, expContained);
}
__finally {
if (expContained) {
DestroyParsedPatternW (expContained);
}
if (contained) {
DestroyParsedPatternW (contained);
}
if (expContainer) {
DestroyParsedPatternW (expContainer);
}
if (container) {
DestroyParsedPatternW (container);
}
}
return result;
}
BOOL
IsParsedPatternContainedExA (
IN PPARSEDPATTERNA Container,
IN PPARSEDPATTERNA Contained
)
{
PPARSEDPATTERNA expContainer = NULL, expContained = NULL;
BOOL result = FALSE;
__try {
expContainer = ExplodeParsedPatternA (Container);
if (!expContainer) {
__leave;
}
expContained = ExplodeParsedPatternA (Contained);
if (!expContained) {
__leave;
}
result = IsExplodedParsedPatternContainedExA (expContainer, expContained);
}
__finally {
if (expContained) {
DestroyParsedPatternA (expContained);
}
if (expContainer) {
DestroyParsedPatternA (expContainer);
}
}
return result;
}
BOOL
IsParsedPatternContainedExW (
IN PPARSEDPATTERNW Container,
IN PPARSEDPATTERNW Contained
)
{
PPARSEDPATTERNW expContainer = NULL, expContained = NULL;
BOOL result = FALSE;
__try {
expContainer = ExplodeParsedPatternW (Container);
if (!expContainer) {
__leave;
}
expContained = ExplodeParsedPatternW (Contained);
if (!expContained) {
__leave;
}
result = IsExplodedParsedPatternContainedExW (expContainer, expContained);
}
__finally {
if (expContained) {
DestroyParsedPatternW (expContained);
}
if (expContainer) {
DestroyParsedPatternW (expContainer);
}
}
return result;
}
/*++
Routine Description:
pAppendCharToGrowBuffer copies the first character in a caller specified
string into the specified grow buffer. This function is used to build up a
string inside a grow buffer, copying character by character.
Arguments:
buf - Specifies the grow buffer to add the character to, receives the
character in its buffer
PtrToChar - Specifies a pointer to the character to copy
Return Value:
None.
--*/
VOID
pAppendCharToGrowBufferA (
IN OUT PGROWBUFFER Buf,
IN PCSTR PtrToChar
)
{
PBYTE p;
UINT Len;
if (IsLeadByte (*PtrToChar) && PtrToChar[1]) {
Len = 2;
} else {
Len = 1;
}
p = GbGrow (Buf, Len);
CopyMemory (p, PtrToChar, (SIZE_T) Len);
}
VOID
pAppendCharToGrowBufferW (
IN OUT PGROWBUFFER Buf,
IN PCWSTR PtrToChar
)
{
PBYTE p;
p = GbGrow (Buf, sizeof(WCHAR));
CopyMemory (p, PtrToChar, sizeof(WCHAR));
}
/*++
Routine Description:
CreateParsedPattern parses the expanded pattern string into a set of
structures. Parsing is considered expensive relative to testing the
pattern, so callers should avoid calling this function inside loops. See
IsPatternMatchEx for a good description of the pattern string syntax.
Arguments:
Pattern - Specifies the pattern string, which can include the extended
wildcard syntax.
Return Value:
A pointer to a parsed pattern structure, which the caller will use like a
handle, or NULL if a syntax error occurred.
--*/
PPARSEDPATTERNA
CreateParsedPatternA (
IN PCSTR Pattern
)
{
PMHANDLE Pool;
PPARSEDPATTERNA Struct;
PATTERNSTATE State;
BOOL CompoundPattern = FALSE;
GROWBUFFER ExactMatchBuf = INIT_GROWBUFFER;
GROWBUFFER SegmentArray = INIT_GROWBUFFER;
GROWBUFFER PatternArray = INIT_GROWBUFFER;
GROWBUFFER SetBuf = INIT_GROWBUFFER;
PPATTERNPROPSA CurrentPattern;
MBCHAR ch = 0;
PCSTR LookAhead;
PCSTR SetBegin = NULL;
PATTERNSTATE ReturnState = 0;
SEGMENTA Segment;
PSEGMENTA SegmentElement;
UINT MaxLen;
Segment.Type = SEGMENTTYPE_UNKNOWN;
Pool = PmCreateNamedPool ("Parsed Pattern");
Struct = (PPARSEDPATTERNA) PmGetAlignedMemory (Pool, sizeof (PARSEDPATTERNA));
ZeroMemory (Struct, sizeof (PARSEDPATTERNA));
State = BEGIN_PATTERN;
for (;;) {
switch (State) {
case BEGIN_PATTERN:
//
// Here we test for either a compound pattern (one that
// is a brace-separated list), or a simple pattern (one
// that does not have a brace).
//
if (_mbsnextc (Pattern) == '<') {
CompoundPattern = TRUE;
State = BEGIN_COMPOUND_PATTERN;
} else if (*Pattern) {
State = BEGIN_PATTERN_EXPR;
} else {
State = PATTERN_DONE;
}
break;
case BEGIN_COMPOUND_PATTERN:
//
// We are looking for the start of a compound pattern.
// Space is allowed inbetween the patterns, but not
// at the start.
//
while (isspace ((INT)(_mbsnextc (Pattern)))) {
Pattern = _mbsinc (Pattern);
}
if (*Pattern == 0) {
State = PATTERN_DONE;
break;
}
if (_mbsnextc (Pattern) == '<') {
Pattern = _mbsinc (Pattern);
State = BEGIN_PATTERN_EXPR;
} else {
DEBUGMSGA ((DBG_ERROR, "Syntax error in pattern: %s", Pattern));
State = PATTERN_ERROR;
}
break;
case BEGIN_PATTERN_EXPR:
//
// We are now ready to condense the expression.
//
State = PARSE_CHAR_EXPR_OR_END;
ExactMatchBuf.End = 0;
SegmentArray.End = 0;
break;
case PARSE_END_FOUND:
State = END_PATTERN_EXPR;
if (ExactMatchBuf.End) {
ReturnState = State;
State = SAVE_EXACT_MATCH;
}
break;
case END_PATTERN_EXPR:
//
// Copy the segment array into the pool, reference the copy
// in the pattern array
//
if (SegmentArray.End) {
CurrentPattern = (PPATTERNPROPSA) GbGrow (&PatternArray, sizeof (PATTERNPROPSA));
CurrentPattern->Segment = (PSEGMENTA) PmGetAlignedMemory (Pool, SegmentArray.End);
CurrentPattern->SegmentCount = SegmentArray.End / sizeof (SEGMENTA);
CopyMemory (
CurrentPattern->Segment,
SegmentArray.Buf,
(SIZE_T) SegmentArray.End
);
}
if (CompoundPattern && *Pattern) {
State = BEGIN_COMPOUND_PATTERN;
} else {
State = PATTERN_DONE;
}
break;
case PARSE_CHAR_EXPR_OR_END:
//
// We now accept the following:
//
// 1. The end of the string or end of a compound pattern
// 2. An escaped character
// 3. The start of an expression
// 4. A non-syntax character
//
ch = _mbsnextc (Pattern);
if (ch == '>' && CompoundPattern) {
//
// Case 1, we found the end of a compound pattern
//
Pattern = _mbsinc (Pattern);
State = PARSE_END_FOUND;
break;
}
if (*Pattern == 0) {
//
// Case 1, we found the end of the pattern
//
if (CompoundPattern) {
State = PATTERN_ERROR;
} else {
State = PARSE_END_FOUND;
}
break;
}
if (ch == '^') {
//
// Case 2, we found an escaped character, so transfer
// it to the buffer.
//
MYASSERT (
Segment.Type == SEGMENTTYPE_UNKNOWN ||
Segment.Type == SEGMENTTYPE_EXACTMATCH
);
Segment.Type = SEGMENTTYPE_EXACTMATCH;
Pattern = _mbsinc (Pattern);
pAppendCharToGrowBufferA (&ExactMatchBuf, Pattern);
Pattern = _mbsinc (Pattern);
break;
}
if (ch == '*' || ch == '?') {
//
// Case 3, we found an expression. Save the wildcard type
// and parse the optional args.
//
if (ExactMatchBuf.End) {
State = SAVE_EXACT_MATCH;
ReturnState = PARSE_CHAR_EXPR_OR_END;
break;
}
ZeroMemory (&Segment, sizeof (Segment));
if (ch == '*') {
Segment.Type = SEGMENTTYPE_OPTIONAL;
} else {
Segment.Type = SEGMENTTYPE_REQUIRED;
Segment.Wildcard.MaxLen = 1;
}
Pattern = _mbsinc (Pattern);
if (_mbsnextc (Pattern) == '[') {
Pattern = _mbsinc (Pattern);
State = LOOK_FOR_NUMBER;
} else {
ReturnState = PARSE_CHAR_EXPR_OR_END;
State = SAVE_SEGMENT;
}
break;
}
//
// Case 4, we don't know about this character, so just copy it
// and continue parsing.
//
pAppendCharToGrowBufferA (&ExactMatchBuf, Pattern);
Pattern = _mbsinc (Pattern);
break;
case SAVE_EXACT_MATCH:
//
// Put the string in ExactMatchBuf into a segment struct
//
pAppendCharToGrowBufferA (&ExactMatchBuf, "");
Segment.Exact.LowerCasePhrase = PmDuplicateStringA (
Pool,
(PCSTR) ExactMatchBuf.Buf
);
Segment.Exact.PhraseBytes = ExactMatchBuf.End - sizeof (CHAR);
MYASSERT (Segment.Exact.LowerCasePhrase);
_mbslwr ((PSTR) Segment.Exact.LowerCasePhrase);
Segment.Type = SEGMENTTYPE_EXACTMATCH;
ExactMatchBuf.End = 0;
// FALL THROUGH!!
case SAVE_SEGMENT:
//
// Put the segment element into the segment array
//
SegmentElement = (PSEGMENTA) GbGrow (&SegmentArray, sizeof (SEGMENTA));
CopyMemory (SegmentElement, &Segment, sizeof (SEGMENTA));
State = ReturnState;
break;
case LOOK_FOR_NUMBER:
//
// Here we are inside a bracket, and there is an optional
// numeric arg, which must be followed by a colon. Test
// that here.
//
LookAhead = Pattern;
MaxLen = 0;
while (*LookAhead >= '0' && *LookAhead <= '9') {
MaxLen = MaxLen * 10 + (*LookAhead - '0');
LookAhead++;
}
if (LookAhead > Pattern && _mbsnextc (LookAhead) == ':') {
Pattern = _mbsinc (LookAhead);
//
// Check for special case syntax error: ?[0:]
//
if (Segment.Type == SEGMENTTYPE_EXACTMATCH && !MaxLen) {
State = PATTERN_ERROR;
break;
}
Segment.Wildcard.MaxLen = MaxLen;
}
SetBegin = Pattern;
State = LOOK_FOR_INCLUDE;
SetBuf.End = 0;
break;
case LOOK_FOR_INCLUDE:
//
// Here we are inside a bracket, past an optional numeric
// arg. Now we look for all the include sets, which are
// optional. We have the following possibilities:
//
// 1. End of set
// 2. An exclude set that needs to be skipped
// 3. A valid include set
// 4. Error
//
// We look at SetBegin, and not Pattern.
//
MYASSERT (SetBegin);
ch = _mbsnextc (SetBegin);
if (ch == ']') {
//
// Case 1: end of set
//
if (SetBuf.End) {
pAppendCharToGrowBufferA (&SetBuf, "");
Segment.Wildcard.IncludeSet = PmDuplicateStringA (
Pool,
(PCSTR) SetBuf.Buf
);
_mbslwr ((PSTR) Segment.Wildcard.IncludeSet);
} else {
Segment.Wildcard.IncludeSet = NULL;
}
SetBuf.End = 0;
State = LOOK_FOR_EXCLUDE;
SetBegin = Pattern;
break;
}
if (ch == '!') {
//
// Case 2: an exclude set
//
SetBegin = _mbsinc (SetBegin);
State = SKIP_EXCLUDE_SET;
ReturnState = LOOK_FOR_INCLUDE;
break;
}
if (*SetBegin == 0) { //lint !e613
State = PATTERN_ERROR;
break;
}
//
// Case 3: a valid include set.
//
State = CONDENSE_SET;
ReturnState = LOOK_FOR_INCLUDE;
break;
case LOOK_FOR_EXCLUDE:
//
// Here we are inside a bracket, past an optional numeric
// arg. All include sets are in the condensing buffer.
// Now we look for all the exclude sets, which are
// optional. We have the following possibilities:
//
// 1. End of set
// 2. A valid exclude set
// 3. An include set that needs to be skipped
// 4. Error
//
// We look at SetBegin, and not Pattern.
//
ch = _mbsnextc (SetBegin);
if (ch == ']') {
//
// Case 1: end of set; we're done with this expr
//
if (SetBuf.End) {
pAppendCharToGrowBufferA (&SetBuf, "");
Segment.Wildcard.ExcludeSet = PmDuplicateStringA (
Pool,
(PCSTR) SetBuf.Buf
);
_mbslwr ((PSTR) Segment.Wildcard.ExcludeSet);
} else {
Segment.Wildcard.ExcludeSet = NULL;
}
SetBuf.End = 0;
State = SAVE_SEGMENT;
ReturnState = PARSE_CHAR_EXPR_OR_END;
Pattern = _mbsinc (SetBegin);
break;
}
if (ch == '!') {
//
// Case 2: a valid exclude set; save it
//
SetBegin = _mbsinc (SetBegin);
if (_mbsnextc (SetBegin) != '(') {
State = PATTERN_ERROR;
break;
}
SetBegin = _mbsinc (SetBegin);
State = CONDENSE_SET;
ReturnState = LOOK_FOR_EXCLUDE;
break;
}
if (*SetBegin == 0) { //lint !e613
State = PATTERN_ERROR;
break;
}
//
// Case 3: an include set that needs to be skipped.
//
State = SKIP_INCLUDE_SET;
ReturnState = LOOK_FOR_EXCLUDE;
break;
case CONDENSE_SET:
//
// Here SetBegin points to a set range, and it is our
// job to copy the range into the set buffer, and
// return back to the previous state.
//
//
// Copy the character at SetBegin
//
if (_mbsnextc (SetBegin) == '^') {
SetBegin = _mbsinc (SetBegin);
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
}
pAppendCharToGrowBufferA (&SetBuf, SetBegin);
//
// Check if this is a range or not
//
LookAhead = _mbsinc (SetBegin);
if (_mbsnextc (LookAhead) == '-') {
//
// Range, copy the character after the dash
//
SetBegin = _mbsinc (LookAhead);
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
if (_mbsnextc (SetBegin) == '^') {
SetBegin = _mbsinc (SetBegin);
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
}
pAppendCharToGrowBufferA (&SetBuf, SetBegin);
} else {
//
// A single character, copy the character again
//
pAppendCharToGrowBufferA (&SetBuf, SetBegin);
}
SetBegin = _mbsinc (SetBegin);
ch = _mbsnextc (SetBegin);
//
// If this is an exclude set, we must have a closing paren
// or a comma
//
State = ReturnState;
if (ReturnState == LOOK_FOR_EXCLUDE) {
if (ch == ')') {
SetBegin = _mbsinc (SetBegin);
ch = _mbsnextc (SetBegin);
} else if (ch != ',') {
State = PATTERN_ERROR;
} else {
//
// Continue condensing the next part of this exclude set
//
State = CONDENSE_SET;
}
}
//
// We either need a comma or a close brace
//
if (ch == ',') {
SetBegin = _mbsinc (SetBegin);
} else if (ch != ']') {
State = PATTERN_ERROR;
}
break;
case SKIP_EXCLUDE_SET:
//
// Skip over the parenthesis group, assuming it is syntatically
// correct, and return to the previous state.
//
if (_mbsnextc (SetBegin) != '(') {
State = PATTERN_ERROR;
break;
}
SetBegin = _mbsinc (SetBegin);
while (*SetBegin) {
if (_mbsnextc (SetBegin) == '^') {
SetBegin = _mbsinc (SetBegin);
} else if (_mbsnextc (SetBegin) == ')') {
break;
}
if (IsLeadByte (SetBegin[0]) && SetBegin[1]) {
SetBegin += 2;
} else {
SetBegin += 1;
}
}
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
SetBegin = _mbsinc (SetBegin);
//
// Now we are either at a comma or a close brace
//
ch = _mbsnextc (SetBegin);
State = ReturnState;
if (ch == ',') {
SetBegin = _mbsinc (SetBegin);
} else if (ch != ']') {
State = PATTERN_ERROR;
}
break;
case SKIP_INCLUDE_SET:
//
// Skip to the next comma or closing brace. We know it is
// syntatically correct by now.
//
ch = 0;
while (*SetBegin) { //lint !e613
ch = _mbsnextc (SetBegin);
if (ch == '^') {
SetBegin = _mbsinc (SetBegin);
} else if (ch == ',' || ch == ']') {
break;
}
SetBegin = _mbsinc (SetBegin);
}
MYASSERT (*SetBegin); //lint !e794
if (ch == ',') {
SetBegin = _mbsinc (SetBegin);
}
State = ReturnState;
break;
} //lint !e787
if (State == PATTERN_DONE || State == PATTERN_ERROR) {
break;
}
}
GbFree (&ExactMatchBuf);
GbFree (&SetBuf);
GbFree (&SegmentArray);
if (State == PATTERN_ERROR) {
GbFree (&PatternArray);
PmDestroyPool (Pool);
return NULL;
}
if (PatternArray.End == 0) {
//build an empty parsed pattern
GbFree (&PatternArray);
Struct->PatternCount = 1;
Struct->Pool = Pool;
Struct->Pattern = (PPATTERNPROPSA) PmGetAlignedMemory (
Pool,
sizeof (PATTERNPROPSA)
);
Struct->Pattern[0].SegmentCount = 1;
Struct->Pattern[0].Segment = (PSEGMENTA) PmGetAlignedMemory (
Pool,
sizeof (PSEGMENTA)
);
Struct->Pattern[0].Segment[0].Type = SEGMENTTYPE_EXACTMATCH;
Struct->Pattern[0].Segment[0].Exact.LowerCasePhrase = PmDuplicateStringA (Pool, "");
Struct->Pattern[0].Segment[0].Exact.PhraseBytes = 0;
return Struct;
}
//
// Copy the fully parsed pattern array into the return struct
//
Struct->Pattern = (PPATTERNPROPSA) PmGetAlignedMemory (
Pool,
PatternArray.End
);
CopyMemory (Struct->Pattern, PatternArray.Buf, (SIZE_T) PatternArray.End);
Struct->PatternCount = PatternArray.End / sizeof (PATTERNPROPSA);
Struct->Pool = Pool;
GbFree (&PatternArray);
return Struct;
}
PPARSEDPATTERNW
CreateParsedPatternW (
IN PCWSTR Pattern
)
{
PMHANDLE Pool;
PPARSEDPATTERNW Struct;
PATTERNSTATE State;
BOOL CompoundPattern = FALSE;
GROWBUFFER ExactMatchBuf = INIT_GROWBUFFER;
GROWBUFFER SegmentArray = INIT_GROWBUFFER;
GROWBUFFER PatternArray = INIT_GROWBUFFER;
GROWBUFFER SetBuf = INIT_GROWBUFFER;
PPATTERNPROPSW CurrentPattern;
WCHAR ch = 0;
PCWSTR LookAhead;
PCWSTR SetBegin = NULL;
PATTERNSTATE ReturnState = 0;
SEGMENTW Segment;
PSEGMENTW SegmentElement;
UINT MaxLen;
Segment.Type = SEGMENTTYPE_UNKNOWN;
Pool = PmCreateNamedPool ("Parsed Pattern");
Struct = (PPARSEDPATTERNW) PmGetAlignedMemory (Pool, sizeof (PARSEDPATTERNW));
ZeroMemory (Struct, sizeof (PARSEDPATTERNW));
State = BEGIN_PATTERN;
for (;;) {
switch (State) {
case BEGIN_PATTERN:
//
// Here we test for either a compound pattern (one that
// is a brace-separated list), or a simple pattern (one
// that does not have a brace).
//
if (*Pattern == L'<') {
CompoundPattern = TRUE;
State = BEGIN_COMPOUND_PATTERN;
} else if (*Pattern) {
State = BEGIN_PATTERN_EXPR;
} else {
State = PATTERN_DONE;
}
break;
case BEGIN_COMPOUND_PATTERN:
//
// We are looking for the start of a compound pattern.
// Space is allowed inbetween the patterns, but not
// at the start.
//
while (iswspace (*Pattern)) {
Pattern++;
}
if (*Pattern == 0) {
State = PATTERN_DONE;
break;
}
if (*Pattern == L'<') {
Pattern++;
State = BEGIN_PATTERN_EXPR;
} else {
DEBUGMSGW ((DBG_ERROR, "Syntax error in pattern: %s", Pattern));
State = PATTERN_ERROR;
}
break;
case BEGIN_PATTERN_EXPR:
//
// We are now ready to condense the expression.
//
State = PARSE_CHAR_EXPR_OR_END;
ExactMatchBuf.End = 0;
SegmentArray.End = 0;
break;
case PARSE_END_FOUND:
State = END_PATTERN_EXPR;
if (ExactMatchBuf.End) {
ReturnState = State;
State = SAVE_EXACT_MATCH;
}
break;
case END_PATTERN_EXPR:
//
// Copy the segment array into the pool, reference the copy
// in the pattern array
//
if (SegmentArray.End) {
CurrentPattern = (PPATTERNPROPSW) GbGrow (&PatternArray, sizeof (PATTERNPROPSW));
CurrentPattern->Segment = (PSEGMENTW) PmGetAlignedMemory (Pool, SegmentArray.End);
CurrentPattern->SegmentCount = SegmentArray.End / sizeof (SEGMENTW);
CopyMemory (
CurrentPattern->Segment,
SegmentArray.Buf,
(SIZE_T) SegmentArray.End
);
}
if (CompoundPattern && *Pattern) {
State = BEGIN_COMPOUND_PATTERN;
} else {
State = PATTERN_DONE;
}
break;
case PARSE_CHAR_EXPR_OR_END:
//
// We now accept the following:
//
// 1. The end of the string or end of a compound pattern
// 2. An escaped character
// 3. The start of an expression
// 4. A non-syntax character
//
ch = *Pattern;
if (ch == L'>' && CompoundPattern) {
//
// Case 1, we found the end of a compound pattern
//
Pattern++;
State = PARSE_END_FOUND;
break;
}
if (*Pattern == 0) {
//
// Case 1, we found the end of the pattern
//
if (CompoundPattern) {
State = PATTERN_ERROR;
} else {
State = PARSE_END_FOUND;
}
break;
}
if (ch == L'^') {
//
// Case 2, we found an escaped character, so transfer
// it to the buffer.
//
MYASSERT (
Segment.Type == SEGMENTTYPE_UNKNOWN ||
Segment.Type == SEGMENTTYPE_EXACTMATCH
);
Segment.Type = SEGMENTTYPE_EXACTMATCH;
Pattern++;
pAppendCharToGrowBufferW (&ExactMatchBuf, Pattern);
Pattern++;
break;
}
if (ch == L'*' || ch == L'?') {
//
// Case 3, we found an expression. Save the wildcard type
// and parse the optional args.
//
if (ExactMatchBuf.End) {
State = SAVE_EXACT_MATCH;
ReturnState = PARSE_CHAR_EXPR_OR_END;
break;
}
ZeroMemory (&Segment, sizeof (Segment));
if (ch == L'*') {
Segment.Type = SEGMENTTYPE_OPTIONAL;
} else {
Segment.Type = SEGMENTTYPE_REQUIRED;
Segment.Wildcard.MaxLen = 1;
}
Pattern++;
if (*Pattern == L'[') {
Pattern++;
State = LOOK_FOR_NUMBER;
} else {
ReturnState = PARSE_CHAR_EXPR_OR_END;
State = SAVE_SEGMENT;
}
break;
}
//
// Case 4, we don't know about this character, so just copy it
// and continue parsing.
//
pAppendCharToGrowBufferW (&ExactMatchBuf, Pattern);
Pattern++;
break;
case SAVE_EXACT_MATCH:
//
// Put the string in ExactMatchBuf into a segment struct
//
pAppendCharToGrowBufferW (&ExactMatchBuf, L"");
Segment.Exact.LowerCasePhrase = PmDuplicateStringW (
Pool,
(PCWSTR) ExactMatchBuf.Buf
); //lint !e64
Segment.Exact.PhraseBytes = ExactMatchBuf.End - sizeof (WCHAR);
MYASSERT (Segment.Exact.LowerCasePhrase);
_wcslwr ((PWSTR) Segment.Exact.LowerCasePhrase);
Segment.Type = SEGMENTTYPE_EXACTMATCH;
ExactMatchBuf.End = 0;
// FALL THROUGH!!
case SAVE_SEGMENT:
//
// Put the segment element into the segment array
//
SegmentElement = (PSEGMENTW) GbGrow (&SegmentArray, sizeof (SEGMENTW));
CopyMemory (SegmentElement, &Segment, sizeof (SEGMENTW));
State = ReturnState;
break;
case LOOK_FOR_NUMBER:
//
// Here we are inside a bracket, and there is an optional
// numeric arg, which must be followed by a colon. Test
// that here.
//
LookAhead = Pattern;
MaxLen = 0;
while (*LookAhead >= L'0' && *LookAhead <= L'9') {
MaxLen = MaxLen * 10 + (*LookAhead - L'0');
LookAhead++;
}
if (LookAhead > Pattern && *LookAhead == L':') {
Pattern = LookAhead + 1;
//
// Check for special case syntax error: ?[0:]
//
if (Segment.Type == SEGMENTTYPE_EXACTMATCH && !MaxLen) {
State = PATTERN_ERROR;
break;
}
Segment.Wildcard.MaxLen = MaxLen;
}
SetBegin = Pattern;
State = LOOK_FOR_INCLUDE;
SetBuf.End = 0;
break;
case LOOK_FOR_INCLUDE:
//
// Here we are inside a bracket, past an optional numeric
// arg. Now we look for all the include sets, which are
// optional. We have the following possibilities:
//
// 1. End of set
// 2. An exclude set that needs to be skipped
// 3. A valid include set
// 4. Error
//
// We look at SetBegin, and not Pattern.
//
if (!SetBegin) {
State = PATTERN_ERROR;
break;
}
ch = *SetBegin;
if (ch == L']') {
//
// Case 1: end of set
//
if (SetBuf.End) {
pAppendCharToGrowBufferW (&SetBuf, L"");
Segment.Wildcard.IncludeSet = PmDuplicateStringW (
Pool,
(PCWSTR) SetBuf.Buf
); //lint !e64
_wcslwr ((PWSTR) Segment.Wildcard.IncludeSet);
} else {
Segment.Wildcard.IncludeSet = NULL;
}
SetBuf.End = 0;
State = LOOK_FOR_EXCLUDE;
SetBegin = Pattern;
break;
}
if (ch == L'!') {
//
// Case 2: an exclude set
//
SetBegin++;
State = SKIP_EXCLUDE_SET;
ReturnState = LOOK_FOR_INCLUDE;
break;
}
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
//
// Case 3: a valid include set.
//
State = CONDENSE_SET;
ReturnState = LOOK_FOR_INCLUDE;
break;
case LOOK_FOR_EXCLUDE:
//
// Here we are inside a bracket, past an optional numeric
// arg. All include sets are in the condensing buffer.
// Now we look for all the exclude sets, which are
// optional. We have the following possibilities:
//
// 1. End of set
// 2. A valid exclude set
// 3. An include set that needs to be skipped
// 4. Error
//
// We look at SetBegin, and not Pattern.
//
if (!SetBegin) {
State = PATTERN_ERROR;
break;
}
ch = *SetBegin;
if (ch == L']') {
//
// Case 1: end of set; we're done with this expr
//
if (SetBuf.End) {
pAppendCharToGrowBufferW (&SetBuf, L"");
Segment.Wildcard.ExcludeSet = PmDuplicateStringW (
Pool,
(PCWSTR) SetBuf.Buf
); //lint !e64
_wcslwr ((PWSTR) Segment.Wildcard.ExcludeSet);
} else {
Segment.Wildcard.ExcludeSet = NULL;
}
SetBuf.End = 0;
State = SAVE_SEGMENT;
ReturnState = PARSE_CHAR_EXPR_OR_END;
Pattern = SetBegin + 1;
break;
}
if (ch == L'!') {
//
// Case 2: a valid exclude set; save it
//
SetBegin++; //lint !e613
if (*SetBegin != L'(') {
State = PATTERN_ERROR;
break;
}
SetBegin++;
State = CONDENSE_SET;
ReturnState = LOOK_FOR_EXCLUDE;
break;
}
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
//
// Case 3: an include set that needs to be skipped.
//
State = SKIP_INCLUDE_SET;
ReturnState = LOOK_FOR_EXCLUDE;
break;
case CONDENSE_SET:
//
// Here SetBegin points to a set range, and it is our
// job to copy the range into the set buffer, and
// return back to the previous state.
//
//
// Copy the character at SetBegin
//
if (!SetBegin) {
State = PATTERN_ERROR;
break;
}
if (*SetBegin == L'^') {
SetBegin++;
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
}
pAppendCharToGrowBufferW (&SetBuf, SetBegin);
//
// Check if this is a range or not
//
LookAhead = SetBegin + 1;
if (*LookAhead == L'-') {
//
// Range, copy the character after the dash
//
SetBegin = LookAhead + 1;
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
if (*SetBegin == L'^') {
SetBegin++;
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
}
pAppendCharToGrowBufferW (&SetBuf, SetBegin);
} else {
//
// A single character, copy the character again
//
pAppendCharToGrowBufferW (&SetBuf, SetBegin);
}
SetBegin++;
ch = *SetBegin;
//
// If this is an exclude set, we must have a closing paren
// or a comma
//
State = ReturnState;
if (ReturnState == LOOK_FOR_EXCLUDE) {
if (ch == L')') {
SetBegin++;
ch = *SetBegin;
} else if (ch != L',') {
State = PATTERN_ERROR;
} else {
//
// Continue condensing the next part of this exclude set
//
State = CONDENSE_SET;
}
}
//
// We either need a comma or a close brace
//
if (ch == L',') {
SetBegin++;
} else if (ch != L']') {
State = PATTERN_ERROR;
}
break;
case SKIP_EXCLUDE_SET:
//
// Skip over the parenthesis group, assuming it is syntatically
// correct, and return to the previous state.
//
if (!SetBegin) {
State = PATTERN_ERROR;
break;
}
if (*SetBegin != L'(') {
State = PATTERN_ERROR;
break;
}
SetBegin++;
while (*SetBegin) {
if (*SetBegin == L'^') {
SetBegin++;
} else if (*SetBegin == L')') {
break;
}
SetBegin++;
}
if (*SetBegin == 0) {
State = PATTERN_ERROR;
break;
}
SetBegin++;
//
// Now we are either at a comma or a close brace
//
ch = *SetBegin;
State = ReturnState;
if (ch == L',') {
SetBegin++;
} else if (ch != L']') {
State = PATTERN_ERROR;
}
break;
case SKIP_INCLUDE_SET:
//
// Skip to the next comma or closing brace. We know it is
// syntatically correct by now.
//
if (!SetBegin) {
State = PATTERN_ERROR;
break;
}
ch = 0;
while (*SetBegin) {
ch = *SetBegin;
if (ch == L'^') {
SetBegin++; //lint !e613
} else if (ch == L',' || ch == L']') {
break;
}
SetBegin++;
}
MYASSERT (*SetBegin);
if (ch == L',') {
SetBegin++;
}
State = ReturnState;
break;
} //lint !e787
if (State == PATTERN_DONE || State == PATTERN_ERROR) {
break;
}
}
GbFree (&ExactMatchBuf);
GbFree (&SetBuf);
GbFree (&SegmentArray);
if (State == PATTERN_ERROR) {
GbFree (&PatternArray);
PmDestroyPool (Pool);
return NULL;
}
if (PatternArray.End == 0) {
//build an empty parsed pattern
GbFree (&PatternArray);
Struct->PatternCount = 1;
Struct->Pool = Pool;
Struct->Pattern = (PPATTERNPROPSW) PmGetAlignedMemory (
Pool,
sizeof (PATTERNPROPSW)
);
Struct->Pattern[0].SegmentCount = 1;
Struct->Pattern[0].Segment = (PSEGMENTW) PmGetAlignedMemory (
Pool,
sizeof (PSEGMENTW)
);
Struct->Pattern[0].Segment[0].Type = SEGMENTTYPE_EXACTMATCH;
Struct->Pattern[0].Segment[0].Exact.LowerCasePhrase = PmDuplicateStringW (Pool, L"");
Struct->Pattern[0].Segment[0].Exact.PhraseBytes = 0;
return Struct;
}
//
// Copy the fully parsed pattern array into the return struct
//
Struct->Pattern = (PPATTERNPROPSW) PmGetAlignedMemory (
Pool,
PatternArray.End
);
CopyMemory (Struct->Pattern, PatternArray.Buf, (SIZE_T) PatternArray.End);
Struct->PatternCount = PatternArray.End / sizeof (PATTERNPROPSW);
Struct->Pool = Pool;
GbFree (&PatternArray);
return Struct;
}
BOOL
WildCharsPatternA (
IN PPARSEDPATTERNA ParsedPattern
)
{
UINT i,j;
if (!ParsedPattern) {
return FALSE;
}
for (i=0; i<ParsedPattern->PatternCount; i++) {
if (ParsedPattern->Pattern[i].SegmentCount > 1) {
return TRUE;
}
for (j=0; j<ParsedPattern->Pattern[i].SegmentCount; j++) {
if ((ParsedPattern->Pattern[i].Segment[j].Type == SEGMENTTYPE_OPTIONAL) ||
(ParsedPattern->Pattern[i].Segment[j].Type == SEGMENTTYPE_REQUIRED)
) {
return TRUE;
}
}
}
return FALSE;
}
BOOL
WildCharsPatternW (
IN PPARSEDPATTERNW ParsedPattern
)
{
UINT i,j;
if (!ParsedPattern) {
return FALSE;
}
for (i=0; i<ParsedPattern->PatternCount; i++) {
if (ParsedPattern->Pattern[i].SegmentCount > 1) {
return TRUE;
}
for (j=0; j<ParsedPattern->Pattern[i].SegmentCount; j++) {
if ((ParsedPattern->Pattern[i].Segment[j].Type == SEGMENTTYPE_OPTIONAL) ||
(ParsedPattern->Pattern[i].Segment[j].Type == SEGMENTTYPE_REQUIRED)
) {
return TRUE;
}
}
}
return FALSE;
}
BOOL
ParsedPatternTrimLastCharA (
IN OUT PPARSEDPATTERNA ParsedPattern
)
{
if (!ParsedPatternHasRootA (ParsedPattern)) {
return FALSE;
}
ParsedPattern->Pattern->Segment[0].Exact.PhraseBytes -= DWSIZEOF (CHAR);
*(PSTR)((PBYTE)ParsedPattern->Pattern->Segment[0].Exact.LowerCasePhrase +
ParsedPattern->Pattern->Segment[0].Exact.PhraseBytes) = 0;
return TRUE;
}
BOOL
ParsedPatternTrimLastCharW (
IN OUT PPARSEDPATTERNW ParsedPattern
)
{
if (!ParsedPatternHasRootW (ParsedPattern)) {
return FALSE;
}
ParsedPattern->Pattern->Segment[0].Exact.PhraseBytes -= DWSIZEOF (WCHAR);
*(PWSTR)((PBYTE)ParsedPattern->Pattern->Segment[0].Exact.LowerCasePhrase +
ParsedPattern->Pattern->Segment[0].Exact.PhraseBytes) = 0;
return TRUE;
}
VOID
UBINTtoHexA (
IN UBINT Number,
OUT PSTR String
)
{
#ifdef IA64
sprintf (String, "0x%08X%08X", (DWORD)(Number >> 32), (DWORD)Number);
#else
sprintf (String, "0x00000000%08X", Number);
#endif
}
VOID
UBINTtoHexW (
IN UBINT Number,
OUT PWSTR String
)
{
#ifdef IA64
swprintf (String, L"0x%08X%08X", (DWORD)(Number >> 32), (DWORD)Number);
#else
swprintf (String, L"0x00000000%08X", Number);
#endif
}
VOID
UBINTtoDecA (
IN UBINT Number,
OUT PSTR String
)
{
#ifdef IA64
sprintf (String, "%I64u", Number);
#else
sprintf (String, "%lu", Number);
#endif
}
VOID
UBINTtoDecW (
IN UBINT Number,
OUT PWSTR String
)
{
#ifdef IA64
swprintf (String, L"%I64u", Number);
#else
swprintf (String, L"%lu", Number);
#endif
}
VOID
BINTtoDecA (
IN BINT Number,
OUT PSTR String
)
{
#ifdef IA64
sprintf (String, "%I64d", Number);
#else
sprintf (String, "%ld", Number);
#endif
}
VOID
BINTtoDecW (
IN BINT Number,
OUT PWSTR String
)
{
#ifdef IA64
swprintf (String, L"%I64d", Number);
#else
swprintf (String, L"%ld", Number);
#endif
}
VOID
PrintPattern (
IN PCSTR PatStr,
IN PPARSEDPATTERNA Struct
)
/*++
Routine Description:
PrintPattern is used for debugging the pattern parsing and testing
functions.
Arguments:
PatStr - Specifies the original pattern string (which is printed as a
heading)
Struct - Specifies the parsed pattern struct
Return Value:
None.
--*/
{
CHAR poolStr [sizeof (UBINT) * 2 + 2 + 1];
UINT u, v;
printf ("Pattern: %s\n\n", PatStr);
if (!Struct) {
printf ("Invalid Pattern\n\n");
return;
}
printf ("PatternCount: %u\n", Struct->PatternCount);
UBINTtoHexA ((UBINT)Struct->Pool, poolStr);
printf ("Pool: %s\n", poolStr);
for (u = 0 ; u < Struct->PatternCount ; u++) {
printf (" Segment Count: %u\n", Struct->Pattern[u].SegmentCount);
for (v = 0 ; v < Struct->Pattern->SegmentCount ; v++) {
printf (" Type: ");
switch (Struct->Pattern[u].Segment[v].Type) {
case SEGMENTTYPE_EXACTMATCH:
printf ("SEGMENTTYPE_EXACTMATCH\n");
printf (" String: %s\n", Struct->Pattern[u].Segment[v].Exact.LowerCasePhrase);
printf (" Bytes: %u\n", Struct->Pattern[u].Segment[v].Exact.PhraseBytes);
break;
case SEGMENTTYPE_OPTIONAL:
printf ("SEGMENTTYPE_OPTIONAL\n");
printf (" MaxLen: %u\n", Struct->Pattern[u].Segment[v].Wildcard.MaxLen);
printf (" IncludeSet: %s\n", Struct->Pattern[u].Segment[v].Wildcard.IncludeSet);
printf (" ExcludeSet: %s\n", Struct->Pattern[u].Segment[v].Wildcard.ExcludeSet);
break;
case SEGMENTTYPE_REQUIRED:
printf ("SEGMENTTYPE_REQUIRED\n");
printf (" MaxLen: %u\n", Struct->Pattern[u].Segment[v].Wildcard.MaxLen);
printf (" IncludeSet: %s\n", Struct->Pattern[u].Segment[v].Wildcard.IncludeSet);
printf (" ExcludeSet: %s\n", Struct->Pattern[u].Segment[v].Wildcard.ExcludeSet);
break;
} //lint !e744
}
}
printf ("\n");
}
/*++
Routine Description:
TestParsedPattern finds the end of the string to test and calls
TestParsedPatternAB.
Arguments:
ParsedPattern - Specifies the parsed pattern structure as returned by
CreateParsedPattern
StringToTest - Specifies the string to test against the pattern
Return Value:
TRUE if the string fits the pattern, FALSE if it does not
--*/
BOOL
TestParsedPatternA (
IN PPARSEDPATTERNA ParsedPattern,
IN PCSTR StringToTest
)
{
PCSTR EndPlusOne = GetEndOfStringA (StringToTest);
return TestParsedPatternABA (ParsedPattern, StringToTest, EndPlusOne);
}
BOOL
TestParsedPatternW (
IN PPARSEDPATTERNW ParsedPattern,
IN PCWSTR StringToTest
)
{
PCWSTR EndPlusOne = GetEndOfStringW (StringToTest);
return TestParsedPatternABW (ParsedPattern, StringToTest, EndPlusOne);
}
/*++
Routine Description:
pTestSet tests a character against an include and exclude set. The sets are
formatted in pairs of characters, where the first character in the pair is
the low range, and the second character in the pair is the high range. The
specified character will automatically be lower-cased, and all whitespace
characters are tested against the space character (ascii 32).
Arguments:
ch - Specifies the character to test. This character is converted
to lower case before the test.
IncludeSet - Specifies the set of characters that ch must be a member of.
If NULL is specified, then the include set is all characters.
ExcludeSet - Specifies the range of characters that ch cannot be a member
of. If NULL is specified, then no characters are excluded.
Return Value:
TRUE if ch is in the include set and not in the exclude set; FALSE
otherwise.
--*/
BOOL
pTestSetA (
IN MBCHAR ch,
IN PCSTR IncludeSet, OPTIONAL
IN PCSTR ExcludeSet OPTIONAL
)
{
MBCHAR LowChar, HighChar;
BOOL b = TRUE;
if (isspace ((INT)ch)) {
if (ch != ' ') {
if (pTestSetA (' ', IncludeSet, ExcludeSet)) {
return TRUE;
}
}
} else {
ch = _mbctolower (ch);
}
if (IncludeSet) {
b = FALSE;
while (*IncludeSet) {
LowChar = _mbsnextc (IncludeSet);
IncludeSet = _mbsinc (IncludeSet);
HighChar = _mbsnextc (IncludeSet);
IncludeSet = _mbsinc (IncludeSet);
if (ch >= LowChar && ch <= HighChar) {
b = TRUE;
break;
}
}
}
if (b && ExcludeSet) {
while (*ExcludeSet) {
LowChar = _mbsnextc (ExcludeSet);
ExcludeSet = _mbsinc (ExcludeSet);
HighChar = _mbsnextc (ExcludeSet);
ExcludeSet = _mbsinc (ExcludeSet);
if (ch >= LowChar && ch <= HighChar) {
b = FALSE;
break;
}
}
}
return b;
}
BOOL
pTestSetW (
IN WCHAR ch,
IN PCWSTR IncludeSet, OPTIONAL
IN PCWSTR ExcludeSet OPTIONAL
)
{
WCHAR LowChar, HighChar;
BOOL b = TRUE;
if (iswspace (ch)) {
if (ch != L' ') {
if (pTestSetW (L' ', IncludeSet, ExcludeSet)) {
return TRUE;
}
}
} else {
ch = towlower (ch);
}
if (IncludeSet) {
b = FALSE;
while (*IncludeSet) {
LowChar = *IncludeSet++;
HighChar = *IncludeSet++;
if (ch >= LowChar && ch <= HighChar) {
b = TRUE;
break;
}
}
}
if (b && ExcludeSet) {
while (*ExcludeSet) {
LowChar = *ExcludeSet++;
HighChar = *ExcludeSet++;
if (ch >= LowChar && ch <= HighChar) {
b = FALSE;
break;
}
}
}
return b;
}
/*++
Routine Description:
pTestOnePatternAB tests a string against a parsed pattern. It loops through
each segment in the pattern, and calls itself recursively in certain
circumstances.
Arguments:
Pattern - Specifies the parsed pattern, as returned from
CreateParsedPattern
StartSeg - Specifies the segment within Pattern to start testing. This
is used for recursion and outside callers should pass in 0.
StringToTest - Specifies the string to test against Pattern. In recursion,
this member will be a pointer to the start of the sub string
to test.
EndPlusOne - Specifies one character beyond the end of the string. This
typically points to the nul terminator.
Return Value:
TRUE if the string between StringToTest and EndPlusOne fits Pattern. FALSE
otherwise.
--*/
BOOL
pTestOnePatternABA (
IN PPATTERNPROPSA Pattern,
IN UINT StartSeg,
IN PCSTR StringToTest,
IN PCSTR EndPlusOne
)
{
UINT u;
PSEGMENTA Segment;
MBCHAR ch1, ch2;
PCSTR q;
PCSTR TempEnd;
UINT BytesLeft;
UINT Chars;
for (u = StartSeg ; u < Pattern->SegmentCount ; u++) {
Segment = &Pattern->Segment[u];
switch (Segment->Type) {
case SEGMENTTYPE_EXACTMATCH:
//
// Check if the exact match is long enough, or if
// the remaining string must match
//
BytesLeft = (UINT)((PBYTE) EndPlusOne - (PBYTE) StringToTest);
if (u + 1 == Pattern->SegmentCount) {
if (BytesLeft != Segment->Exact.PhraseBytes) {
return FALSE;
}
} else if (BytesLeft < Segment->Exact.PhraseBytes) {
return FALSE;
}
//
// Compare the strings
//
q = Segment->Exact.LowerCasePhrase;
TempEnd = (PCSTR) ((PBYTE) q + Segment->Exact.PhraseBytes);
ch1 = 0;
ch2 = 0;
while (q < TempEnd) {
ch1 = _mbsnextc (StringToTest);
ch2 = _mbsnextc (q);
ch1 = _mbctolower (ch1);
if (ch1 != ch2) {
if (ch2 == ' ') {
if (!isspace ((INT)ch1)) {
break;
}
} else {
break;
}
}
q = _mbsinc (q);
StringToTest = _mbsinc (StringToTest);
}
if (ch1 != ch2) {
return FALSE;
}
//
// Continue onto next segment
//
break;
case SEGMENTTYPE_REQUIRED:
MYASSERT (Segment->Wildcard.MaxLen > 0);
//
// Verify there are the correct number of characters
// in the specified char set
//
Chars = Segment->Wildcard.MaxLen;
if (Segment->Wildcard.IncludeSet || Segment->Wildcard.ExcludeSet) {
while (StringToTest < EndPlusOne && Chars > 0) {
if (!pTestSetA (
_mbsnextc (StringToTest),
Segment->Wildcard.IncludeSet,
Segment->Wildcard.ExcludeSet
)) {
return FALSE;
}
Chars--;
StringToTest = _mbsinc (StringToTest);
}
} else {
while (StringToTest < EndPlusOne && Chars > 0) {
Chars--;
StringToTest = _mbsinc (StringToTest);
}
}
if (Chars) {
return FALSE;
}
if (u + 1 == Pattern->SegmentCount) {
if (*StringToTest) {
return FALSE;
}
}
//
// Continue onto next segment
//
break;
case SEGMENTTYPE_OPTIONAL:
if (Segment->Wildcard.MaxLen == 0) {
//
// Last segment is "anything"
//
if (u + 1 == Pattern->SegmentCount &&
!Segment->Wildcard.IncludeSet &&
!Segment->Wildcard.ExcludeSet
) {
return TRUE;
}
}
//
// Find end of optional text
//
TempEnd = StringToTest;
Chars = Segment->Wildcard.MaxLen;
if (Segment->Wildcard.IncludeSet || Segment->Wildcard.ExcludeSet) {
if (Chars) {
while (TempEnd < EndPlusOne && Chars > 0) {
if (!pTestSetA (
_mbsnextc (TempEnd),
Segment->Wildcard.IncludeSet,
Segment->Wildcard.ExcludeSet
)) {
break;
}
TempEnd = _mbsinc (TempEnd);
Chars--;
}
} else {
while (TempEnd < EndPlusOne) {
if (!pTestSetA (
_mbsnextc (TempEnd),
Segment->Wildcard.IncludeSet,
Segment->Wildcard.ExcludeSet
)) {
break;
}
TempEnd = _mbsinc (TempEnd);
}
}
} else if (Chars) {
while (TempEnd < EndPlusOne && Chars > 0) {
TempEnd = _mbsinc (TempEnd);
Chars--;
}
} else {
TempEnd = EndPlusOne;
}
//
// If this is the last segment, then match only when
// the remaining text fits
//
if (u + 1 == Pattern->SegmentCount) {
return TempEnd >= EndPlusOne;
}
//
// Because other segments exist, we must check recursively
//
do {
if (pTestOnePatternABA (Pattern, u + 1, StringToTest, EndPlusOne)) {
return TRUE;
}
StringToTest = _mbsinc (StringToTest);
} while (StringToTest <= TempEnd);
//
// No match
//
return FALSE;
} //lint !e744
}
return TRUE;
}
BOOL
pTestOnePatternABW (
IN PPATTERNPROPSW Pattern,
IN UINT StartSeg,
IN PCWSTR StringToTest,
IN PCWSTR EndPlusOne
)
{
UINT u;
PSEGMENTW Segment;
WCHAR ch1, ch2;
PCWSTR q;
PCWSTR TempEnd;
UINT BytesLeft;
UINT Chars;
for (u = StartSeg ; u < Pattern->SegmentCount ; u++) {
Segment = &Pattern->Segment[u];
switch (Segment->Type) {
case SEGMENTTYPE_EXACTMATCH:
//
// Check if the exact match is long enough, or if
// the remaining string must match
//
BytesLeft = (UINT)((PBYTE) EndPlusOne - (PBYTE) StringToTest);
if (u + 1 == Pattern->SegmentCount) {
if (BytesLeft != Segment->Exact.PhraseBytes) {
return FALSE;
}
} else if (BytesLeft < Segment->Exact.PhraseBytes) {
return FALSE;
}
//
// Compare the strings
//
q = Segment->Exact.LowerCasePhrase; //lint !e64
TempEnd = (PCWSTR) ((PBYTE) q + Segment->Exact.PhraseBytes);
ch1 = 0;
ch2 = 0;
while (q < TempEnd) {
ch1 = towlower (*StringToTest);
ch2 = *q;
if (ch1 != ch2) {
if (ch2 == L' ') {
if (!iswspace (ch1)) {
break;
}
} else {
break;
}
}
q++;
StringToTest++;
}
if (ch1 != ch2) {
return FALSE;
}
//
// Continue onto next segment
//
break;
case SEGMENTTYPE_REQUIRED:
MYASSERT (Segment->Wildcard.MaxLen > 0);
//
// Verify there are the correct number of characters
// in the specified char set
//
Chars = Segment->Wildcard.MaxLen;
if (Segment->Wildcard.IncludeSet || Segment->Wildcard.ExcludeSet) {
while (StringToTest < EndPlusOne && Chars > 0) {
if (!pTestSetW (
*StringToTest,
Segment->Wildcard.IncludeSet, //lint !e64
Segment->Wildcard.ExcludeSet
)) { //lint !e64
return FALSE;
}
Chars--;
StringToTest++;
}
if (Chars) {
return FALSE;
}
} else {
StringToTest += Chars;
if (StringToTest > EndPlusOne) {
return FALSE;
}
}
if (u + 1 == Pattern->SegmentCount) {
if (*StringToTest) {
return FALSE;
}
}
//
// Continue onto next segment
//
break;
case SEGMENTTYPE_OPTIONAL:
if (Segment->Wildcard.MaxLen == 0) {
//
// Last segment is "anything"
//
if (u + 1 == Pattern->SegmentCount &&
!Segment->Wildcard.IncludeSet &&
!Segment->Wildcard.ExcludeSet
) {
return TRUE;
}
}
//
// Find end of optional text
//
TempEnd = StringToTest;
Chars = Segment->Wildcard.MaxLen;
if (Segment->Wildcard.IncludeSet || Segment->Wildcard.ExcludeSet) {
if (Chars) {
while (TempEnd < EndPlusOne && Chars > 0) {
if (!pTestSetW (
*TempEnd,
Segment->Wildcard.IncludeSet, //lint !e64
Segment->Wildcard.ExcludeSet
)) { //lint !e64
break;
}
TempEnd++;
Chars--;
}
} else {
while (TempEnd < EndPlusOne) {
if (!pTestSetW (
*TempEnd,
Segment->Wildcard.IncludeSet, //lint !e64
Segment->Wildcard.ExcludeSet
)) { //lint !e64
break;
}
TempEnd++;
}
}
} else if (Chars) {
TempEnd += Chars;
if (TempEnd > EndPlusOne) {
TempEnd = EndPlusOne;
}
} else {
TempEnd = EndPlusOne;
}
//
// If this is the last segment, then match only when
// the remaining text fits
//
if (u + 1 == Pattern->SegmentCount) {
return TempEnd >= EndPlusOne;
}
//
// Because other segments exist, we must check recursively
//
do {
if (pTestOnePatternABW (Pattern, u + 1, StringToTest, EndPlusOne)) {
return TRUE;
}
StringToTest++;
} while (StringToTest <= TempEnd);
//
// No match
//
return FALSE;
} //lint !e744
}
return TRUE;
}
/*++
Routine Description:
TestParsedPattternAB loops through all the patterns in ParsedPattern,
testing the specified string against each. The loop stops at the first
match.
Arguments:
ParsedPattern - Specifies the parsed pattern, as returned from
CreateParsedPattern
StringToTest - Specifies the start of the string to test.
EndPlusOne - Specifies a pointer to the first character after the end of
the string. This often points to the nul at the end of the
string. A nul must not exist in between StringToTest and
EndPlusOne; a nul can only be at *EndPlusOne. A nul is not
required.
Return Value:
TRUE if the string specified between StringToTest and EndPlusOne matches
Pattern. FALSE otherwise.
--*/
BOOL
TestParsedPatternABA (
IN PPARSEDPATTERNA ParsedPattern,
IN PCSTR StringToTest,
IN PCSTR EndPlusOne
)
{
UINT u;
BOOL b = FALSE;
for (u = 0 ; u < ParsedPattern->PatternCount ; u++) {
b = pTestOnePatternABA (
&ParsedPattern->Pattern[u],
0,
StringToTest,
EndPlusOne
);
if (b) {
break;
}
}
return b;
}
BOOL
TestParsedPatternABW (
IN PPARSEDPATTERNW ParsedPattern,
IN PCWSTR StringToTest,
IN PCWSTR EndPlusOne
)
{
UINT u;
BOOL b = FALSE;
for (u = 0 ; u < ParsedPattern->PatternCount ; u++) {
b = pTestOnePatternABW (
&ParsedPattern->Pattern[u],
0,
StringToTest,
EndPlusOne
);
if (b) {
break;
}
}
return b;
}
/*++
Routine Description:
DestroyParsedPattern cleans up a pattern allocated from CreateParsedPattern.
Arguments:
ParsedPattern - Specifies the value returned from CreateParsedPattern.
Return Value:
None.
--*/
VOID
DestroyParsedPatternA (
IN PPARSEDPATTERNA ParsedPattern
)
{
if (ParsedPattern) {
PmDestroyPool (ParsedPattern->Pool);
}
}
VOID
DestroyParsedPatternW (
IN PPARSEDPATTERNW ParsedPattern
)
{
if (ParsedPattern) {
PmDestroyPool (ParsedPattern->Pool);
}
}
/*++
Routine Description:
DecodeParsedPattern decodes all exact-matches sub-strings of the given pattern.
Arguments:
ParsedPattern - Specifies the parsed pattern.
Return Value:
None.
--*/
VOID
DecodeParsedPatternA (
IN PPARSEDPATTERNA ParsedPattern
)
{
UINT u;
UINT v;
PSTR phrase;
for (u = 0; u < ParsedPattern->PatternCount; u++) {
for (v = 0; v < ParsedPattern->Pattern[u].SegmentCount; v++) {
if (ParsedPattern->Pattern[u].Segment[v].Type == SEGMENTTYPE_EXACTMATCH) {
phrase = (PSTR)ParsedPattern->Pattern[u].Segment[v].Exact.LowerCasePhrase;
DecodeRuleCharsA (phrase, phrase);
ParsedPattern->Pattern[u].Segment[v].Exact.PhraseBytes = ByteCountA (phrase);
}
}
}
}
VOID
DecodeParsedPatternW (
IN PPARSEDPATTERNW ParsedPattern
)
{
UINT u;
UINT v;
PWSTR phrase;
for (u = 0; u < ParsedPattern->PatternCount; u++) {
for (v = 0; v < ParsedPattern->Pattern[u].SegmentCount; v++) {
if (ParsedPattern->Pattern[u].Segment[v].Type == SEGMENTTYPE_EXACTMATCH) {
phrase = (PWSTR)ParsedPattern->Pattern[u].Segment[v].Exact.LowerCasePhrase;
DecodeRuleCharsW (phrase, phrase);
ParsedPattern->Pattern[u].Segment[v].Exact.PhraseBytes = ByteCountW (phrase);
}
}
}
}
/*++
Routine Description:
GetParsedPatternMinMaxSize returns the minimum and the maximum size (in bytes)
of a string that would match the given parsed pattern.
Arguments:
ParsedPattern - Specifies the parsed pattern
MinSize - Receives the minimum size of a string that would match the pattern
MaxSize - Receives the maximum size of a string that would match the pattern
Return Value:
None.
--*/
VOID
GetParsedPatternMinMaxSizeA (
IN PPARSEDPATTERNA ParsedPattern,
OUT PDWORD MinSize,
OUT PDWORD MaxSize
)
{
UINT u;
UINT v;
DWORD pmin;
DWORD pmax;
DWORD smin;
DWORD smax;
*MinSize = *MaxSize = 0;
for (u = 0; u < ParsedPattern->PatternCount; u++) {
pmin = pmax = 0;
for (v = 0; v < ParsedPattern->Pattern[u].SegmentCount; v++) {
switch (ParsedPattern->Pattern[u].Segment[v].Type) {
case SEGMENTTYPE_EXACTMATCH:
smin = smax = ParsedPattern->Pattern[u].Segment[v].Exact.PhraseBytes;
break;
case SEGMENTTYPE_OPTIONAL:
smin = 0;
if (ParsedPattern->Pattern[u].Segment[v].Wildcard.MaxLen) {
smax = ParsedPattern->Pattern[u].Segment[v].Wildcard.MaxLen * DWSIZEOF (CHAR);
} else {
smax = DWORD_MAX;
}
break;
case SEGMENTTYPE_REQUIRED:
MYASSERT (ParsedPattern->Pattern[u].Segment[v].Wildcard.MaxLen > 0);
smin = smax = ParsedPattern->Pattern[u].Segment[v].Wildcard.MaxLen * DWSIZEOF (CHAR);
break;
default:
MYASSERT (FALSE); //lint !e506
smin = smax = 0;
}
pmin += smin;
if (pmax < DWORD_MAX) {
if (smax < DWORD_MAX) {
pmax += smax;
} else {
pmax = DWORD_MAX;
}
}
}
if (pmin < *MinSize) {
*MinSize = pmin;
}
if (pmax > *MaxSize) {
*MaxSize = pmax;
}
}
}
VOID
GetParsedPatternMinMaxSizeW (
IN PPARSEDPATTERNW ParsedPattern,
OUT PDWORD MinSize,
OUT PDWORD MaxSize
)
{
UINT u;
UINT v;
DWORD pmin;
DWORD pmax;
DWORD smin;
DWORD smax;
*MinSize = *MaxSize = 0;
for (u = 0; u < ParsedPattern->PatternCount; u++) {
pmin = pmax = 0;
for (v = 0; v < ParsedPattern->Pattern[u].SegmentCount; v++) {
switch (ParsedPattern->Pattern[u].Segment[v].Type) {
case SEGMENTTYPE_EXACTMATCH:
smin = smax = ParsedPattern->Pattern[u].Segment[v].Exact.PhraseBytes;
break;
case SEGMENTTYPE_OPTIONAL:
smin = 0;
if (ParsedPattern->Pattern[u].Segment[v].Wildcard.MaxLen) {
smax = ParsedPattern->Pattern[u].Segment[v].Wildcard.MaxLen * DWSIZEOF (WCHAR);
} else {
smax = DWORD_MAX;
}
break;
case SEGMENTTYPE_REQUIRED:
MYASSERT (ParsedPattern->Pattern[u].Segment[v].Wildcard.MaxLen > 0);
smin = smax = ParsedPattern->Pattern[u].Segment[v].Wildcard.MaxLen * DWSIZEOF (WCHAR);
break;
default:
MYASSERT (FALSE); //lint !e506
smin = smax = 0;
}
pmin += smin;
if (pmax < DWORD_MAX) {
if (smax < DWORD_MAX) {
pmax += smax;
} else {
pmax = DWORD_MAX;
}
}
}
if (pmin < *MinSize) {
*MinSize = pmin;
}
if (pmax > *MaxSize) {
*MaxSize = pmax;
}
}
}
/*++
Routine Description:
PatternIncludesPattern decides if a given pattern includes another pattern,
meaning that any string that would match the second will match the first.
Arguments:
IncludingPattern - Specifies the first parsed pattern
IncludedPattern - Specifies the second parsed pattern
Return Value:
TRUE if the first pattern includes the second
--*/
BOOL
PatternIncludesPatternA (
IN PPARSEDPATTERNA IncludingPattern,
IN PPARSEDPATTERNA IncludedPattern
)
{
PPATTERNPROPSA pp1;
PPATTERNPROPSA pp2;
PSEGMENTA ps1;
PSEGMENTA ps2;
DWORD min1;
DWORD max1;
DWORD min2;
DWORD max2;
//
// only deal with simple patterns for now (PatternCount == 1)
//
if (IncludingPattern->PatternCount > 1 || IncludedPattern->PatternCount > 1) {
DEBUGMSGA ((DBG_ERROR, "PatternIncludesPatternA: multiple patterns not supported yet"));
return FALSE;
}
//
// test the usual cases first, quickly
//
pp1 = IncludingPattern->Pattern;
MYASSERT (pp1);
if (pp1->SegmentCount == 1 && ParsedPatternSegmentIsPureOptionalA (pp1->Segment)) {
return TRUE;
}
pp2 = IncludedPattern->Pattern;
MYASSERT (pp2);
if (pp2->SegmentCount == 1 && ParsedPatternSegmentIsPureOptionalA (pp2->Segment)) {
return FALSE;
}
if (pp1->SegmentCount == 1) {
ps1 = pp1->Segment;
if (ps1->Type == SEGMENTTYPE_EXACTMATCH) {
if (pp2->SegmentCount == 1) {
ps2 = pp2->Segment;
if (ps2->Type == SEGMENTTYPE_EXACTMATCH) {
return ps1->Exact.PhraseBytes == ps2->Exact.PhraseBytes &&
StringMatchA (ps1->Exact.LowerCasePhrase, ps2->Exact.LowerCasePhrase);
}
}
}
} else if (pp1->SegmentCount == 2) {
ps1 = pp1->Segment;
if (ps1->Type == SEGMENTTYPE_EXACTMATCH) {
if (ParsedPatternSegmentIsPureOptionalA (pp1->Segment + 1)) {
if (pp2->SegmentCount == 1) {
ps2 = pp2->Segment;
if (ps2->Type == SEGMENTTYPE_EXACTMATCH) {
return ps1->Exact.PhraseBytes <= ps2->Exact.PhraseBytes &&
StringMatchByteCountA (
ps1->Exact.LowerCasePhrase,
ps2->Exact.LowerCasePhrase,
ps1->Exact.PhraseBytes
);
}
} else if (pp2->SegmentCount == 2) {
ps2 = pp2->Segment;
if (ps2->Type == SEGMENTTYPE_EXACTMATCH) {
if (ParsedPatternSegmentIsPureOptionalA (pp2->Segment + 1)) {
return ps1->Exact.PhraseBytes <= ps2->Exact.PhraseBytes &&
StringMatchByteCountA (
ps1->Exact.LowerCasePhrase,
ps2->Exact.LowerCasePhrase,
ps1->Exact.PhraseBytes
);
}
}
}
}
}
}
GetParsedPatternMinMaxSizeA (IncludingPattern, &min1, &max1);
GetParsedPatternMinMaxSizeA (IncludedPattern, &min2, &max2);
if (min2 < min1 || max2 > max1) {
return FALSE;
}
//
// BUGBUG - not implemented yet
//
return FALSE;
}
BOOL
PatternIncludesPatternW (
IN PPARSEDPATTERNW IncludingPattern,
IN PPARSEDPATTERNW IncludedPattern
)
{
PPATTERNPROPSW pp1;
PPATTERNPROPSW pp2;
PSEGMENTW ps1;
PSEGMENTW ps2;
DWORD min1;
DWORD max1;
DWORD min2;
DWORD max2;
//
// only deal with simple patterns for now (PatternCount == 1)
//
if (IncludingPattern->PatternCount > 1 || IncludedPattern->PatternCount > 1) {
DEBUGMSGW ((DBG_ERROR, "PatternIncludesPatternW: multiple patterns not supported yet"));
return FALSE;
}
//
// test the usual cases first, quickly
//
pp1 = IncludingPattern->Pattern;
MYASSERT (pp1);
if (pp1->SegmentCount == 1 && ParsedPatternSegmentIsPureOptionalW (pp1->Segment)) {
return TRUE;
}
pp2 = IncludedPattern->Pattern;
MYASSERT (pp2);
if (pp2->SegmentCount == 1 && ParsedPatternSegmentIsPureOptionalW (pp2->Segment)) {
return FALSE;
}
if (pp1->SegmentCount == 1) {
ps1 = pp1->Segment;
if (ps1->Type == SEGMENTTYPE_EXACTMATCH) {
if (pp2->SegmentCount == 1) {
ps2 = pp2->Segment;
if (ps2->Type == SEGMENTTYPE_EXACTMATCH) {
return ps1->Exact.PhraseBytes == ps2->Exact.PhraseBytes &&
StringMatchW (ps1->Exact.LowerCasePhrase, ps2->Exact.LowerCasePhrase); //lint !e64
}
}
}
} else if (pp1->SegmentCount == 2) {
ps1 = pp1->Segment;
if (ps1->Type == SEGMENTTYPE_EXACTMATCH) {
if (ParsedPatternSegmentIsPureOptionalW (pp1->Segment + 1)) {
if (pp2->SegmentCount == 1) {
ps2 = pp2->Segment;
if (ps2->Type == SEGMENTTYPE_EXACTMATCH) {
return ps1->Exact.PhraseBytes <= ps2->Exact.PhraseBytes &&
StringMatchByteCountW (
ps1->Exact.LowerCasePhrase,
ps2->Exact.LowerCasePhrase,
ps1->Exact.PhraseBytes
); //lint !e64
}
} else if (pp2->SegmentCount == 2) {
ps2 = pp2->Segment;
if (ps2->Type == SEGMENTTYPE_EXACTMATCH) {
if (ParsedPatternSegmentIsPureOptionalW (pp2->Segment + 1)) {
return ps1->Exact.PhraseBytes <= ps2->Exact.PhraseBytes &&
StringMatchByteCountW (
ps1->Exact.LowerCasePhrase,
ps2->Exact.LowerCasePhrase,
ps1->Exact.PhraseBytes
); //lint !e64
}
}
}
}
}
}
GetParsedPatternMinMaxSizeW (IncludingPattern, &min1, &max1);
GetParsedPatternMinMaxSizeW (IncludedPattern, &min2, &max2);
if (min2 < min1 || max2 > max1) {
return FALSE;
}
//
// BUGBUG - not implemented yet
//
return FALSE;
}
VOID
_copymbchar (
OUT PSTR sz1,
IN PCSTR sz2
)
/*++
Routine Description:
_copymbchar transfers the character at sz2 to sz1, which may be one or
two bytes long.
Arguments:
sz1 - The destination string
sz2 - The source string
Return Value:
none
--*/
{
if (IsLeadByte (*sz2)) {
sz1[1] = sz2[1];
}
*sz1 = *sz2;
}
/*++
Routine Description:
_tcsctrim removes character c from the end of str if it exists. It removes
only one character at the most.
Arguments:
str - A pointer to the string that may have character c at the end
c - The character that may be at the end of the string
Return Value:
TRUE if character c was at the end of the string, or FALSE if it was not.
--*/
BOOL
_mbsctrim (
OUT PSTR str,
IN MBCHAR c
)
{
PSTR end;
end = GetEndOfStringA (str);
end = _mbsdec2 (str, end);
if (end && _mbsnextc (end) == c) {
*end = 0;
return TRUE;
}
return FALSE;
}
BOOL
_wcsctrim (
OUT PWSTR str,
IN WCHAR c
)
{
PWSTR end;
end = GetEndOfStringW (str);
end == str ? end = NULL : end--;
if (end && *end == c) {
*end = 0;
return TRUE;
}
return FALSE;
}
/*++
Routine Description:
The FreeStringResourceEx functions are used to free a recently used
string that is not being passed back to the caller. In almost all
cases, this string is at the end of our array of pointers, so we can
efficiently search sequentially in reverse order. If the pointer is
not the last element of the array, it is first swapped with the real
last element of the array so the array size is reduced.
Arguments:
AllocTable - The GROWBUFFER table that holds the list of previously
allocated strings (return values of ParseMessageEx or
GetResourceStringEx).
String - A pointer to the string that is in AllocTable
Return Value:
none
--*/
VOID
FreeStringResourceExA (
IN PGROWBUFFER AllocTable,
IN PCSTR String
)
{
PCTSTR *ptr, *end, *start;
if (!String) {
return;
}
//
// Locate string (search sequentially in reverse order)
//
if (AllocTable->End < sizeof (PCSTR)) {
DEBUGMSG ((DBG_ERROR, "FreeStringResourceA: Attempt to free address %x (%s); address table empty", String, String));
return;
}
start = (PCSTR *) AllocTable->Buf;
end = (PCSTR *) (AllocTable->Buf + AllocTable->End - sizeof (PCSTR));
ptr = end;
while (ptr >= start) {
if (*ptr == String) {
break;
}
ptr--;
}
//
// String not found case
//
if (ptr < start) {
DEBUGMSG ((DBG_ERROR, "FreeStringResourceA: Attempt to free address %x (%s); address not found in table", String, String));
return;
}
//
// Free LocalAlloc'd memory
//
LocalFree ((HLOCAL) String);
//
// If this element is not the end, copy real end to the ptr
//
if (ptr < end) {
*ptr = *end;
}
//
// Shrink buffer size
//
AllocTable->End -= sizeof (PCSTR);
}
VOID
FreeStringResourcePtrExA (
IN PGROWBUFFER AllocTable,
IN OUT PCSTR * String
)
{
if (NULL != *String) {
FreeStringResourceExA(AllocTable, *String);
*String = NULL;
}
}
VOID
FreeStringResourceExW (
IN PGROWBUFFER AllocTable,
IN PCWSTR String
)
{
FreeStringResourceExA (AllocTable, (PCSTR) String);
}
VOID
FreeStringResourcePtrExW (
IN PGROWBUFFER AllocTable,
IN OUT PCWSTR * String
)
{
if (NULL != *String) {
FreeStringResourceExW(AllocTable, *String);
*String = NULL;
}
}
/*++
Routine Description:
The pAddStringResource function is used to track pointers allocated
by FormatMessage. They are added to an array (maintained in a GROWBUFFER
structure). This table of pointers is used by FreeStringResource or
StringResourceFree.
Arguments:
String - A pointer to a LocalAlloc'd string (the return value of
FormatMessage). This string is added to a table of allocated
strings.
Return Value:
none
--*/
VOID
pAddStringResource (
IN PGROWBUFFER GrowBuf,
IN PCSTR String
)
{
PCSTR *ptr;
ptr = (PCSTR *) GbGrow (GrowBuf, sizeof (PCSTR));
if (ptr) {
*ptr = String;
}
ELSE_DEBUGMSG ((DBG_ERROR, "pAddStringResource: GrowBuffer failure caused memory leak"));
}
/*++
Routine Description:
pFreeAllStringResourcesEx frees all strings currently listed in AllocTable.
This function allows the caller to wait until all processing is done
to clean up string resources that may have been allocated.
Arguments:
none
Return Value:
none
--*/
VOID
pFreeAllStringResourcesEx (
IN PGROWBUFFER AllocTable
)
{
PCSTR *ptr, *start, *end;
if (AllocTable->End) {
start = (PCSTR *) AllocTable->Buf;
end = (PCSTR *) (AllocTable->Buf + AllocTable->End);
for (ptr = start ; ptr < end ; ptr++) {
LocalFree ((HLOCAL) (*ptr));
}
}
GbFree (AllocTable);
}
/*++
Routine Description:
CreateAllocTable creates a GROWBUFFER structure that can be used with
ParseMessageEx, GetStringResourceEx, FreeStringResourceEx and
pFreeAllStringResourcesEx. Call this function to recieve a private
allocation table to pass to these functions. Call DestroyAllocTable
to clean up.
Arguments:
none
Return Value:
A pointer to a GROWBUFFER structure, or NULL if a memory allocation failed.
--*/
PGROWBUFFER
RealCreateAllocTable (
VOID
)
{
PGROWBUFFER allocTable;
GROWBUFFER tempForInit = INIT_GROWBUFFER;
allocTable = (PGROWBUFFER) MemAlloc (g_hHeap, 0, sizeof (GROWBUFFER));
CopyMemory (allocTable, &tempForInit, sizeof (GROWBUFFER));
return allocTable;
}
/*++
Routine Description:
DestroyAllocTable cleans up all memory associated with an AllocTable.
Arguments:
AllocTable - A pointer to a GROWBUFFER structure allocated by CreateAllocTable
Return Value:
none
--*/
VOID
DestroyAllocTable (
OUT PGROWBUFFER AllocTable
)
{
MYASSERT (AllocTable);
pFreeAllStringResourcesEx (AllocTable);
MemFree (g_hHeap, 0, AllocTable);
}
/*++
Routine Description:
BeginMessageProcessing enters a guarded section of code that plans to use the
ParseMessage and GetStringResource functions, but needs cleanup at the end
of processing.
EndMessageProcessing destroys all memory allocated within the message processing
block, and leaves the guarded section.
Arguments:
none
Return Value:
BeginMessageProcessing returns FALSE if an out-of-memory condition occurrs.
--*/
BOOL
BeginMessageProcessing (
VOID
)
{
if (!TryEnterOurCriticalSection (&g_MessageCs)) {
DEBUGMSG ((DBG_ERROR, "Thread attempting to enter BeginMessageProcessing while another"
"thread is processing messages as well."));
EnterOurCriticalSection (&g_MessageCs);
}
g_LastAllocTable = g_ShortTermAllocTable;
g_ShortTermAllocTable = CreateAllocTable();
MYASSERT (g_ShortTermAllocTable);
return TRUE;
}
VOID
EndMessageProcessing (
VOID
)
{
if (TryEnterOurCriticalSection (&g_MessageCs)) {
DEBUGMSG ((DBG_ERROR, "Thread attempting to end message processing when it hasn't been started"));
LeaveOurCriticalSection (&g_MessageCs);
return;
}
DestroyAllocTable (g_ShortTermAllocTable);
g_ShortTermAllocTable = g_LastAllocTable;
LeaveOurCriticalSection (&g_MessageCs);
}
/*++
Routine Description:
ParseMessage is used to obtain a string from the executable's message table
and parse it with FormatMessage. An array of arguments can be passed by
the caller. FormatMessage will replace %1 with the first element of the
array, %2 with the second element, and so on. The array does not need to
be terminated, and if a message string uses %n, element n must be non-NULL.
Arguments:
Template - A string indicating which message to extract, or a WORD value
cast as a string. (ParseMessageID does this cast via a macro.)
ArgArray - Optional array of string pointers, where the meaning depends on
the message string. A reference in the message string to %n
requires element n of ArgArray to be a valid string pointer.
Return Value:
Pointer to the string allocated. Call StringResourceFree to free all
allocated strings (a one-time cleanup for all strings). The pointer may
be NULL if the resource does not exist or is empty.
--*/
PCSTR
ParseMessageExA (
IN PGROWBUFFER AllocTable,
IN PCSTR Template,
IN PCSTR ArgArray[]
)
{
PSTR MsgBuf;
DWORD rc;
if (SHIFTRIGHT16 ((UBINT)Template)) {
//
// From string
//
rc = FormatMessageA (
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_ARGUMENT_ARRAY |
FORMAT_MESSAGE_FROM_STRING,
(LPVOID) Template,
0,
0,
(LPVOID) &MsgBuf,
0,
(va_list *) ArgArray
);
} else {
//
// From resource
//
rc = FormatMessageA (
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_ARGUMENT_ARRAY |
FORMAT_MESSAGE_FROM_HMODULE,
(LPVOID) g_hInst,
(DWORD)((UBINT)Template),
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPVOID) &MsgBuf,
0,
(va_list *) ArgArray
);
}
if (rc > 0) {
pAddStringResource (AllocTable, MsgBuf);
return MsgBuf;
}
return NULL;
}
PCWSTR
ParseMessageExW (
IN PGROWBUFFER AllocTable,
IN PCWSTR Template,
IN PCWSTR ArgArray[]
)
{
PWSTR MsgBuf;
DWORD rc;
if (SHIFTRIGHT16 ((UBINT)Template)) {
//
// From string
//
rc = FormatMessageW (
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_ARGUMENT_ARRAY |
FORMAT_MESSAGE_FROM_STRING,
(LPVOID) Template,
0,
0,
(LPVOID) &MsgBuf,
0,
(va_list *) ArgArray
);
} else {
//
// From resource
//
rc = FormatMessageW (
FORMAT_MESSAGE_ALLOCATE_BUFFER |
FORMAT_MESSAGE_ARGUMENT_ARRAY |
FORMAT_MESSAGE_FROM_HMODULE,
(LPVOID) g_hInst,
(DWORD)(UBINT)Template,
MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
(LPVOID) &MsgBuf,
0,
(va_list *) ArgArray
);
}
if (rc > 0) {
pAddStringResource (AllocTable, (PCSTR) MsgBuf);
return MsgBuf;
}
return NULL;
}
/*++
Routine Description:
GetStringResourceEx is an argument-less wrapper of ParseMessageEx. It allows
the caller to specify a message ID and recieve a pointer to the string if
it exists, and a table to track FormatMessage's allocations.
Arguments:
AllocTable - A pointer to a GROWBUFFER structure that is used to maintain
the handles of allocated strings
ID - The ID of the message resource to retrieve
Return Value:
Pointer to the string allocated. The return pointer may
be NULL if the resource does not exist or is empty.
Call FreeStringResource or DestroyAllocTable to clean up AllocTable.
--*/
PCSTR
GetStringResourceExA (
IN OUT PGROWBUFFER AllocTable,
IN UINT ID
)
{
return ParseMessageExA (AllocTable, (PSTR) (WORD) ID, NULL);
}
PCWSTR
GetStringResourceExW (
IN OUT PGROWBUFFER AllocTable,
IN UINT ID
)
{
return ParseMessageExW (AllocTable, (PWSTR) (WORD) ID, NULL);
}
/*++
Routine Description:
ParseMessageInWnd is used to exchange a string in a window with one from
the executable's message table. It is provided for dialog box initialization,
where a field in the dialog box requires dynamic data. The dialog box
resource should contain a control with its window text set to the message
string. Upon processing WM_INITDIALOG, the code should call ParseMessageInWnd,
supplying the necessary ArgArray, so the dialog box is initialized with
a dynamic message.
Arguments:
hwnd - The handle of a window whose title contains the message string ID
ArgArray - Optional array of string pointers, where the meaning depends on
the message string. A reference in the message string to %n
requires element n of ArgArray to be a valid string pointer.
Return Value:
none
--*/
VOID
ParseMessageInWndA (
IN HWND Hwnd,
IN PCSTR ArgArray[]
)
{
CHAR buffer[512];
PCSTR parsedMsg;
GetWindowTextA (Hwnd, buffer, 512);
parsedMsg = ParseMessageA (buffer, ArgArray);
if (parsedMsg) {
SetWindowTextA (Hwnd, parsedMsg);
FreeStringResourceA (parsedMsg);
}
}
VOID
ParseMessageInWndW (
IN HWND hwnd,
IN PCWSTR ArgArray[]
)
{
WCHAR buffer[512];
PCWSTR parsedMsg;
GetWindowTextW (hwnd, buffer, 512);
parsedMsg = ParseMessageW (buffer, ArgArray);
if (parsedMsg) {
SetWindowTextW (hwnd, parsedMsg);
FreeStringResourceW (parsedMsg);
}
}
/*++
Routine Description:
ResourceMessageBox is used to display a message based on a message resource
ID.
Arguments:
HwndOwner - The handle of the owner of the message box to be displayed
ID - The identifier of the message resource
Flags - MessageBox flags (MB_OK, etc.)
ArgArray - Optional array of string pointers, where the meaning depends on
the message string. A reference in the message string to %n
requires element n of ArgArray to be a valid string pointer.
Return Value:
The return value of MessageBox (MB_YES, etc.)
--*/
INT
ResourceMessageBoxA (
IN HWND HwndOwner,
IN UINT ID,
IN UINT Flags,
IN PCSTR ArgArray[]
)
{
PCSTR message;
PCSTR title;
int rc;
message = ParseMessageA ((PSTR)(UBINT)ID, ArgArray);
if (!message)
return -1;
title = GetStringResourceA (MSG_MESSAGEBOX_TITLE);
rc = MessageBoxA (HwndOwner, message, title, Flags);
FreeStringResourceA (message);
if (title) {
FreeStringResourceA (title);
}
return rc;
}
INT
ResourceMessageBoxW (
IN HWND HwndOwner,
IN UINT ID,
IN UINT Flags,
IN PCWSTR ArgArray[]
)
{
PCWSTR message;
PCWSTR title;
int rc;
message = ParseMessageW ((PWSTR)(UBINT)ID, ArgArray);
if (!message)
return -1;
title = GetStringResourceW (MSG_MESSAGEBOX_TITLE);
rc = MessageBoxW (HwndOwner, message, title, Flags);
FreeStringResourceW (message);
if (title) {
FreeStringResourceW (title);
}
return rc;
}
BOOL
StringReplaceA (
IN PSTR Buffer,
IN DWORD MaxSize,
IN PSTR ReplaceStartPos,
IN PSTR ReplaceEndPos,
IN PCSTR NewString
)
{
BOOL rf = FALSE;
DWORD oldSubStringLength;
DWORD newSubStringLength;
DWORD currentStringLength;
LONG offset;
PSTR movePosition;
//
// Check assumptions.
//
MYASSERT(Buffer);
MYASSERT(ReplaceStartPos && ReplaceStartPos >= Buffer);
MYASSERT(ReplaceEndPos && ReplaceEndPos >= ReplaceStartPos); //lint !e613
MYASSERT(NewString);
//
// Compute sizes.
//
oldSubStringLength = (DWORD)((UBINT)ReplaceEndPos - (UBINT)ReplaceStartPos);
newSubStringLength = ByteCountA(NewString);
currentStringLength = SizeOfStringA(Buffer) + 1;
offset = (LONG)newSubStringLength - (LONG)oldSubStringLength;
//
// Make sure there is enough room in the buffer to perform the replace
// operation.
//
if ((LONG)currentStringLength + offset > (LONG)MaxSize) {
DEBUGMSG((DBG_WARNING,"ERROR: Buffer to small to perform string replacement."));
rf = FALSE;
} else {
//
// Shift the rest of the buffer to adjust it to the size of the new string.
//
if (newSubStringLength > oldSubStringLength) {
//
// right shift.
//
for (movePosition = Buffer + currentStringLength;
(UBINT)movePosition >= (UBINT)ReplaceStartPos + oldSubStringLength;
movePosition--) {
*(movePosition + offset) = *movePosition;
}
} else {
//
// left or no shift.
//
for(movePosition = ReplaceStartPos + newSubStringLength; //lint !e613
movePosition < Buffer + currentStringLength;
movePosition++) {
*movePosition = *(movePosition - offset);
}
}
//
// Now, copy in the string.
//
CopyMemory (ReplaceStartPos, NewString, newSubStringLength); //lint !e668
//
// String replacement completed successfully.
//
rf = TRUE;
}
return rf;
}
BOOL
StringReplaceW (
IN PWSTR Buffer,
IN DWORD MaxSize,
IN PWSTR ReplaceStartPos,
IN PWSTR ReplaceEndPos,
IN PCWSTR NewString
)
{
BOOL rf = FALSE;
DWORD oldSubStringLength;
DWORD newSubStringLength;
DWORD currentStringLength;
LONG offset;
PWSTR movePosition;
//
// Check assumptions.
//
MYASSERT(Buffer);
MYASSERT(ReplaceStartPos && ReplaceStartPos >= Buffer);
MYASSERT(ReplaceEndPos && ReplaceEndPos >= ReplaceStartPos); //lint !e613
MYASSERT(NewString);
//
// Compute sizes.
//
oldSubStringLength = (DWORD)((UBINT)ReplaceEndPos - (UBINT)ReplaceStartPos);
newSubStringLength = CharCountW(NewString);
currentStringLength = CharCountW(Buffer) + 1;
offset = (LONG)newSubStringLength - (LONG)oldSubStringLength;
//
// Make sure there is enough room in the buffer to perform the replace
// operation.
//
if ((LONG)currentStringLength + offset > (LONG)MaxSize) {
DEBUGMSG((DBG_WARNING,"ERROR: Buffer to small to perform string replacement."));
rf = FALSE;
} else {
//
// Shift the rest of the buffer to adjust it to the size of the new string.
//
if (newSubStringLength > oldSubStringLength) {
//
// right shift.
//
for (movePosition = Buffer + currentStringLength;
(UBINT)movePosition >= (UBINT)ReplaceStartPos + oldSubStringLength;
movePosition--) {
*(movePosition + offset) = *movePosition;
}
} else {
//
// left or no shift.
//
for (movePosition = ReplaceStartPos + newSubStringLength; //lint !e613
movePosition < Buffer + currentStringLength;
movePosition++) {
*movePosition = *(movePosition - offset);
}
}
//
// Now, copy in the string.
//
wcsncpy(ReplaceStartPos,NewString,newSubStringLength);
//
// String replacement completed successfully.
//
rf = TRUE;
}
return rf;
}
#if 0
/*++
Routine Description:
AddInfSectionToHashTable enumerates the specified section and adds each
item to the string table. An optional callback allows data to be associated
with each item.
Arguments:
Table - Specifies the table that receives new entries
InfFile - Specifies an open INF handle of the file to read
Section - Specifies the INF section name to enumerate
Field - Specifies which field to extract text from. If the field
exists, it is added to the string table.
Callback - Specifies optional callback to be called before adding to
the string table. The callback supplies additional data.
CallbackParam - Data passed to the callback
Return Value:
TRUE if the INF file was processed successfullly, or FALSE if an error
occurred.
--*/
BOOL
AddInfSectionToHashTableA (
IN OUT HASHTABLE Table,
IN HINF InfFile,
IN PCSTR Section,
IN DWORD Field,
IN ADDINFSECTION_PROCA Callback,
IN PVOID CallbackData
)
{
INFCONTEXT ic;
LONG rc;
HASHTABLE ht;
DWORD reqSize;
DWORD currentSize = 0;
PSTR newBuffer, buffer = NULL;
PVOID data;
UINT dataSize;
BOOL b = FALSE;
//
// On NT, Setup API is compiled with UNICODE, so the string table
// functions are UNICODE only.
//
if (ISNT()) {
SetLastError (ERROR_CALL_NOT_IMPLEMENTED);
return FALSE;
}
if (SetupFindFirstLineA (InfFile, Section, NULL, &ic)) {
do {
if (!SetupGetStringFieldA (&ic, Field, NULL, 0, &reqSize)) {
continue;
}
if (reqSize > currentSize) {
reqSize = ((reqSize / 1024) + 1) * 1024;
if (buffer) {
newBuffer = (PSTR) MemReAlloc (g_hHeap, 0, buffer, reqSize);
} else {
newBuffer = (PSTR) MemAlloc (g_hHeap, 0, reqSize);
}
if (!newBuffer) {
goto cleanup;
}
buffer = newBuffer;
currentSize = reqSize;
}
if (!SetupGetStringFieldA (&ic, Field, buffer, currentSize, NULL)) {
DEBUGMSG ((DBG_ERROR, "AddInfSectionToHashTable: SetupGetStringField failed unexpectedly"));
continue;
}
data = NULL;
dataSize = 0;
if (Callback) {
rc = Callback (buffer, &data, &dataSize, CallbackData);
if (rc == CALLBACK_STOP) {
goto cleanup;
}
if (rc == CALLBACK_SKIP) {
continue;
}
}
ht = HtAddStringExA (
Table,
buffer,
data,
CASE_INSENSITIVE
);
if (!ht) {
goto cleanup;
}
} while (SetupFindNextLine (&ic, &ic));
}
b = TRUE;
cleanup:
if (buffer) {
PushError();
MemFree (g_hHeap, 0, buffer);
PopError();
}
return b;
}
BOOL
AddInfSectionToHashTableW (
IN OUT HASHTABLE Table,
IN HINF InfFile,
IN PCWSTR Section,
IN DWORD Field,
IN ADDINFSECTION_PROCW Callback,
IN PVOID CallbackData
)
{
INFCONTEXT ic;
HASHTABLE ht;
LONG rc;
DWORD reqSize;
DWORD currentSize = 0;
PWSTR newBuffer, buffer = NULL;
PVOID data;
UINT dataSize;
BOOL b = FALSE;
//
// On Win9x, Setup API is compiled with ANSI, so the string table
// functions are ANSI only.
//
if (ISWIN9X()) {
SetLastError (ERROR_CALL_NOT_IMPLEMENTED);
return FALSE;
}
if (SetupFindFirstLineW (InfFile, Section, NULL, &ic)) {
do {
if (!SetupGetStringFieldW (&ic, Field, NULL, 0, &reqSize)) {
continue;
}
if (reqSize > currentSize) {
reqSize = ((reqSize / 1024) + 1) * 1024;
if (buffer) {
newBuffer = (PWSTR) MemReAlloc (g_hHeap, 0, buffer, reqSize);
} else {
newBuffer = (PWSTR) MemAlloc (g_hHeap, 0, reqSize);
}
if (!newBuffer) {
goto cleanup;
}
buffer = newBuffer;
currentSize = reqSize;
}
if (!SetupGetStringFieldW (&ic, Field, buffer, currentSize, NULL)) {
DEBUGMSG ((DBG_ERROR, "AddInfSectionToHashTable: SetupGetStringField failed unexpectedly"));
continue;
}
data = NULL;
dataSize = 0;
if (Callback) {
rc = Callback (buffer, &data, &dataSize, CallbackData);
if (rc == CALLBACK_STOP) {
goto cleanup;
}
if (rc == CALLBACK_SKIP) {
continue;
}
}
ht = HtAddStringExW (
Table,
buffer,
data,
CASE_INSENSITIVE
);
if (!ht) {
goto cleanup;
}
} while (SetupFindNextLine (&ic, &ic));
}
b = TRUE;
cleanup:
if (buffer) {
PushError();
MemFree (g_hHeap, 0, buffer);
PopError();
}
return b;
}
#endif
/*++
Routine Description:
Finds the last wack in the path and returns a pointer to the next
character. If no wack is found, returns a pointer to the full
string.
Arguments:
PathSpec - Specifies the path that has a file at the end of it
Return Value:
A pointer to the file name in the path.
--*/
PCSTR
GetFileNameFromPathA (
IN PCSTR PathSpec
)
{
PCSTR p;
p = _mbsrchr (PathSpec, '\\');
if (p) {
p = _mbsinc (p);
} else {
p = PathSpec;
}
return p;
}
PCWSTR
GetFileNameFromPathW (
IN PCWSTR PathSpec
)
{
PCWSTR p;
p = wcsrchr (PathSpec, L'\\');
if (p) {
p++;
} else {
p = PathSpec;
}
return p;
}
/*++
Routine Description:
Finds the last wack in the path and then the last point from the remaining path
returning a pointer to the next character. If no point is found, returns a null pointer.
Arguments:
PathSpec - Specifies the path that has a file at the end of it
Return Value:
A pointer to the file extension, excluding the dot, or NULL if no extension exists.
--*/
PCSTR
GetFileExtensionFromPathA (
IN PCSTR PathSpec
)
{
PCSTR p;
PCSTR ReturnPtr = NULL;
p = PathSpec;
while (*p) {
if (*p == '.') {
ReturnPtr = p + 1;
} else if (*p == '\\') {
ReturnPtr = NULL;
}
p = _mbsinc (p);
}
return ReturnPtr;
}
PCWSTR
GetFileExtensionFromPathW (
IN PCWSTR PathSpec
)
{
PCWSTR p;
PCWSTR ReturnPtr = NULL;
p = PathSpec;
while (*p) {
if (*p == L'.') {
ReturnPtr = p + 1;
} else if (*p == L'\\') {
ReturnPtr = NULL;
}
p++;
}
return ReturnPtr;
}
/*++
Routine Description:
GetDotExtensionFromPath finds the last wack in the path and then the last dot from
the remaining path, returning a pointer to the dot. If no dot is found, returns the
end of the string.
Arguments:
PathSpec - Specifies the path that has a file at the end of it
Return Value:
A pointer to the file extension, including the dot, or the end of the string if
no extension exists.
--*/
PCSTR
GetDotExtensionFromPathA (
IN PCSTR PathSpec
)
{
PCSTR p;
PCSTR ReturnPtr = NULL;
p = PathSpec;
while (*p) {
if (*p == '.') {
ReturnPtr = p;
} else if (*p == '\\') {
ReturnPtr = NULL;
}
p = _mbsinc (p);
}
if (!ReturnPtr) {
return p;
}
return ReturnPtr;
}
PCWSTR
GetDotExtensionFromPathW (
IN PCWSTR PathSpec
)
{
PCWSTR p;
PCWSTR ReturnPtr = NULL;
p = PathSpec;
while (*p) {
if (*p == L'.') {
ReturnPtr = p;
} else if (*p == L'\\') {
ReturnPtr = NULL;
}
p++;
}
if (!ReturnPtr) {
return p;
}
return ReturnPtr;
}
/*++
Routine Description:
CountInstancesOfChar returns the number of occurances Char
is found in String.
Arguments:
String - Specifies the text that may or may not contain
search text
Char - Specifies the char to count
Return Value:
The number of times Char appears in String.
--*/
UINT
CountInstancesOfCharA (
IN PCSTR String,
IN MBCHAR Char
)
{
UINT count;
count = 0;
while (*String) {
if (_mbsnextc (String) == Char) {
count++;
}
String = _mbsinc (String);
}
return count;
}
UINT
CountInstancesOfCharW (
IN PCWSTR String,
IN WCHAR Char
)
{
UINT count;
count = 0;
while (*String) {
if (*String == Char) {
count++;
}
String++;
}
return count;
}
/*++
Routine Description:
CountInstancesOfCharI returns the number of occurances Char
is found in String. The comparison is case-insenetive.
Arguments:
String - Specifies the text that may or may not contain
search text
Char - Specifies the char to count
Return Value:
The number of times Char appears in String.
--*/
UINT
CountInstancesOfCharIA (
IN PCSTR String,
IN MBCHAR Char
)
{
UINT count;
Char = (MBCHAR)tolower ((INT)Char);
count = 0;
while (*String) {
if ((MBCHAR) tolower ((INT)_mbsnextc (String)) == Char) {
count++;
}
String = _mbsinc (String);
}
return count;
}
UINT
CountInstancesOfCharIW (
IN PCWSTR String,
IN WCHAR Char
)
{
UINT count;
Char = towlower (Char);
count = 0;
while (*String) {
if (towlower (*String) == Char) {
count++;
}
String++;
}
return count;
}
/*++
Routine Description:
Searches the string counting the number of occurances of
SearchString exist in SourceString.
Arguments:
SourceString - Specifies the text that may or may not contain
search text
SearchString - Specifies the text phrase to count
Return Value:
The number of times SearchString appears in SourceString.
--*/
UINT
CountInstancesOfSubStringA (
IN PCSTR SourceString,
IN PCSTR SearchString
)
{
PCSTR p;
UINT count;
UINT searchBytes;
count = 0;
p = SourceString;
searchBytes = ByteCountA (SearchString);
while (p = _mbsistr (p, SearchString)) { //lint !e720
count++;
p += searchBytes;
}
return count;
}
UINT
CountInstancesOfSubStringW (
IN PCWSTR SourceString,
IN PCWSTR SearchString
)
{
PCWSTR p;
UINT count;
UINT SearchChars;
count = 0;
p = SourceString;
SearchChars = CharCountW (SearchString);
while (p = _wcsistr (p, SearchString)) { //lint !e720
count++;
p += SearchChars;
}
return count;
}
/*++
Routine Description:
Searches and replaces all occurances of SearchString with
ReplaceString.
Arguments:
SourceString - String that contiains zero or more instances
of the search text
SearchString - String to search for. Cannot be zero-length or NULL.
ReplaceString - String to replace. Can be zero-length but cannot
be NULL.
Return Value:
A pointer to the pool-allocated string, or NULL if no instances
of SearchString were found in SourceString. Free the non-NULL
pointer with FreePathString.
--*/
PCSTR
StringSearchAndReplaceA (
IN PCSTR SourceString,
IN PCSTR SearchString,
IN PCSTR ReplaceString
)
{
PSTR newString;
PBYTE p, q;
PBYTE dest;
UINT count;
UINT size;
UINT searchBytes;
UINT replaceBytes;
UINT untouchedBytes;
//
// count occurances within the string
//
count = CountInstancesOfSubStringA (
SourceString,
SearchString
);
if (!count) {
return NULL;
}
searchBytes = ByteCountA (SearchString);
replaceBytes = ByteCountA (ReplaceString);
MYASSERT (searchBytes);
size = SizeOfStringA (SourceString) -
count * searchBytes +
count * replaceBytes;
newString = (PSTR) PmGetAlignedMemory (g_PathsPool, size);
if (!newString) {
return NULL;
}
p = (PBYTE) SourceString;
dest = (PBYTE) newString;
while (q = (PBYTE) _mbsistr ((PCSTR) p, SearchString)) { //lint !e720
untouchedBytes = (DWORD)(q - p);
if (untouchedBytes) {
CopyMemory (dest, p, (SIZE_T) untouchedBytes);
dest += untouchedBytes;
}
if (replaceBytes) {
CopyMemory (dest, (PBYTE) ReplaceString, (SIZE_T) replaceBytes);
dest += replaceBytes;
}
p = q + searchBytes;
}
StringCopyA ((PSTR) dest, (PSTR) p);
return newString;
}
PCWSTR
StringSearchAndReplaceW (
IN PCWSTR SourceString,
IN PCWSTR SearchString,
IN PCWSTR ReplaceString
)
{
PWSTR newString;
PBYTE p, q;
PBYTE dest;
UINT count;
UINT size;
UINT searchBytes;
UINT replaceBytes;
UINT untouchedBytes;
//
// count occurances within the string
//
count = CountInstancesOfSubStringW (
SourceString,
SearchString
);
if (!count) {
return NULL;
}
searchBytes = ByteCountW (SearchString);
replaceBytes = ByteCountW (ReplaceString);
MYASSERT (searchBytes);
size = SizeOfStringW (SourceString) -
count * searchBytes +
count * replaceBytes;
newString = (PWSTR) PmGetAlignedMemory (g_PathsPool, size);
if (!newString) {
return NULL;
}
p = (PBYTE) SourceString;
dest = (PBYTE) newString;
while (q = (PBYTE) _wcsistr ((PCWSTR) p, SearchString)) { //lint !e720
untouchedBytes = (DWORD)(q - p);
if (untouchedBytes) {
CopyMemory (dest, p, (SIZE_T) untouchedBytes);
dest += untouchedBytes;
}
if (replaceBytes) {
CopyMemory (dest, (PBYTE) ReplaceString, (SIZE_T) replaceBytes);
dest += replaceBytes;
}
p = q + searchBytes;
}
StringCopyW ((PWSTR) dest, (PWSTR) p);
return newString;
}
PSTR *
CommandLineToArgvA (
IN PCSTR CmdLine,
OUT PUINT NumArgs
)
/*++
Routine Description:
CommandLineToArgvA implements an ANSI version of the Win32 function
CommandLineToArgvW.
Arguments:
CmdLine - A pointer to the complete command line, including the
module name. This is the same string returned by
GetCommandLineA().
NumArgs - Receives the number of arguments allocated, identical to
main's argc parameter. That is, NumArgs is equal to
the number of command line arguments plus one for the
command itself.
Return Value:
A pointer to an array of string pointers, one per argument. The
command line arguments are placed in separate nul-terminated strings.
The caller must free the memory using a single call to GlobalFree or
LocalFree.
--*/
{
PCSTR start, end;
BOOL QuoteMode;
MBCHAR ch = 0;
UINT Pass;
UINT ArgStrSize;
UINT Args;
PSTR ArgStrEnd = NULL; // filled in on pass one, used on pass two
PSTR *ArgPtrArray = NULL; // filled in on pass one, used on pass two
//
// count args on first pass, then allocate memory and create arg string
//
ArgStrSize = 0;
Pass = 0;
do {
// Init loop
Pass++;
Args = 0;
start = CmdLine;
// Skip leading space
while (isspace (*start)) {
start++;
}
while (*start) {
// Look for quote mode
if (*start == '\"') {
QuoteMode = TRUE;
start++;
} else {
QuoteMode = FALSE;
}
// Find end of arg
end = start;
while (*end) {
ch = _mbsnextc (end);
if (QuoteMode) {
if (ch == '\"') {
break;
}
} else {
if (isspace ((INT)ch)) {
break;
}
}
end = _mbsinc (end);
}
// If Pass 1, add string size
if (Pass == 1) {
ArgStrSize += (UINT)((UBINT)end - (UBINT)start) + 1;
}
// If Pass 2, copy strings to buffer
else {
MYASSERT (ArgStrEnd);
MYASSERT (ArgPtrArray);
ArgPtrArray[Args] = ArgStrEnd; //lint !e613
StringCopyABA (ArgStrEnd, start, end);
ArgStrEnd = GetEndOfStringA (ArgStrEnd); //lint !e668
ArgStrEnd++; //lint !e613
}
// Set start to next arg
Args++;
if (QuoteMode && ch == '\"') {
end = _mbsinc (end);
}
start = end;
while (isspace (*start)) {
start++;
}
}
// If Pass 1, allocate strings
if (Pass == 1) {
if (Args) {
ArgPtrArray = (PSTR *) GlobalAlloc (
GPTR,
(UINT)(sizeof (PSTR) * Args + ArgStrSize)
);
if (!ArgPtrArray) {
return NULL;
}
ArgStrEnd = (PSTR) (&ArgPtrArray[Args]);
} else {
return NULL;
}
}
} while (Pass < 2);
*NumArgs = Args;
return ArgPtrArray;
}
BOOL
EnumNextMultiSzA (
IN OUT PMULTISZ_ENUMA MultiSzEnum
)
{
if (!MultiSzEnum->CurrentString || !(*MultiSzEnum->CurrentString)) {
return FALSE;
}
MultiSzEnum->CurrentString = GetEndOfStringA (MultiSzEnum->CurrentString) + 1; //lint !e613
return (MultiSzEnum->CurrentString [0] != 0);
}
BOOL
EnumFirstMultiSzA (
OUT PMULTISZ_ENUMA MultiSzEnum,
IN PCSTR MultiSzStr
)
{
if ((MultiSzStr == NULL) || (MultiSzStr [0] == 0)) {
return FALSE;
}
MultiSzEnum->Buffer = MultiSzStr;
MultiSzEnum->CurrentString = MultiSzStr;
return TRUE;
}
BOOL
EnumNextMultiSzW (
IN OUT PMULTISZ_ENUMW MultiSzEnum
)
{
if (!MultiSzEnum->CurrentString || !(*MultiSzEnum->CurrentString)) {
return FALSE;
}
MultiSzEnum->CurrentString = GetEndOfStringW (MultiSzEnum->CurrentString) + 1;
return (MultiSzEnum->CurrentString [0] != 0);
}
BOOL
EnumFirstMultiSzW (
OUT PMULTISZ_ENUMW MultiSzEnum,
IN PCWSTR MultiSzStr
)
{
if ((MultiSzStr == NULL) || (MultiSzStr [0] == 0)) {
return FALSE;
}
MultiSzEnum->Buffer = MultiSzStr;
MultiSzEnum->CurrentString = MultiSzStr;
return TRUE;
}
PSTR
GetPrevCharA (
IN PCSTR StartStr,
IN PCSTR CurrPtr,
IN MBCHAR SearchChar
)
{
PCSTR ptr = CurrPtr;
for (;;) {
ptr = _mbsdec2 (StartStr, ptr);
if (!ptr) {
return NULL;
}
if (_mbsnextc (ptr) == SearchChar) {
return (PSTR) ptr;
}
}
}
PWSTR
GetPrevCharW (
IN PCWSTR StartStr,
IN PCWSTR CurrPtr,
IN WCHAR SearchChar
)
{
PCWSTR ptr = CurrPtr;
while (ptr > StartStr) {
ptr--;
if (*ptr == SearchChar) {
return (PWSTR) ptr;
}
}
return NULL;
}
VOID
ToggleWacksA (
IN PSTR Line,
IN BOOL Operation
)
{
CHAR curChar;
CHAR newChar;
PSTR p = Line;
curChar = Operation ? WACK_REPLACE_CHAR : '\\';
newChar = Operation ? '\\' : WACK_REPLACE_CHAR;
do {
p = _mbschr (p, curChar);
if (p) {
*p = newChar;
p = _mbsinc (p);
}
} while (p);
}
VOID
ToggleWacksW (
IN PWSTR Line,
IN BOOL Operation
)
{
WCHAR curChar;
WCHAR newChar;
PWSTR p = Line;
curChar = Operation ? WACK_REPLACE_CHAR : L'\\';
newChar = Operation ? L'\\' : WACK_REPLACE_CHAR;
do {
p = wcschr (p, curChar);
if (p) {
*p = newChar;
p++;
}
} while (p);
}
PSTR
pGoBackA (
IN PSTR LastChar,
IN PSTR FirstChar,
IN UINT NumWacks
)
{
LastChar = _mbsdec2 (FirstChar, LastChar);
while (NumWacks && (LastChar >= FirstChar)) {
if (_mbsnextc (LastChar) == '\\') {
NumWacks --;
}
LastChar = _mbsdec2 (FirstChar, LastChar);
}
if (NumWacks) {
return NULL;
}
return LastChar + 2;
}
PWSTR
pGoBackW (
IN PWSTR LastChar,
IN PWSTR FirstChar,
IN UINT NumWacks
)
{
LastChar --;
while (NumWacks && (LastChar >= FirstChar)) {
if (*LastChar == L'\\') {
NumWacks --;
}
LastChar --;
}
if (NumWacks) {
return NULL;
}
return LastChar + 2;
}
UINT
pCountDotsA (
IN PCSTR PathSeg
)
{
UINT numDots = 0;
while (PathSeg && *PathSeg) {
if (_mbsnextc (PathSeg) != '.') {
return 0;
}
numDots ++;
PathSeg = _mbsinc (PathSeg);
}
return numDots;
}
UINT
pCountDotsW (
IN PCWSTR PathSeg
)
{
UINT numDots = 0;
while (PathSeg && *PathSeg) {
if (*PathSeg != L'.') {
return 0;
}
numDots ++;
PathSeg ++;
}
return numDots;
}
PCSTR
SanitizePathA (
IN PCSTR FileSpec
)
{
CHAR pathSeg [MAX_MBCHAR_PATH];
PCSTR wackPtr;
UINT dotNr;
PSTR newPath = DuplicatePathStringA (FileSpec, 0);
PSTR newPathPtr = newPath;
BOOL firstPass = TRUE;
do {
wackPtr = _mbschr (FileSpec, '\\');
if (wackPtr) {
if (firstPass && (wackPtr == FileSpec)) {
// this one starts with a wack, let's see if we have double wacks
wackPtr = _mbsinc (wackPtr);
if (!wackPtr) {
FreePathStringA (newPath);
return NULL;
}
if (_mbsnextc (wackPtr) == '\\') {
// this one starts with a double wack
wackPtr = _mbsinc (wackPtr);
if (!wackPtr) {
FreePathStringA (newPath);
return NULL;
}
wackPtr = _mbschr (wackPtr, '\\');
} else {
wackPtr = _mbschr (wackPtr, '\\');
}
}
firstPass = FALSE;
if (wackPtr) {
StringCopyByteCountABA (pathSeg, FileSpec, wackPtr, MAX_MBCHAR_PATH);
FileSpec = _mbsinc (wackPtr);
} else {
StringCopyByteCountABA (pathSeg, FileSpec, GetEndOfStringA (FileSpec), MAX_MBCHAR_PATH);
}
} else {
StringCopyByteCountABA (pathSeg, FileSpec, GetEndOfStringA (FileSpec), MAX_MBCHAR_PATH);
}
if (*pathSeg) {
dotNr = pCountDotsA (pathSeg);
if (dotNr>1) {
newPathPtr = pGoBackA (newPathPtr, newPath, dotNr);
if (newPathPtr == NULL) {
DEBUGMSGA ((DBG_WARNING, "Broken path detected:%s", FileSpec));
FreePathStringA (newPath);
return NULL;
}
} else {
StringCopyA (newPathPtr, pathSeg);
newPathPtr = GetEndOfStringA (newPathPtr);
if (wackPtr) {
*newPathPtr = '\\';
//we increment this because we know that \ is a single byte character.
newPathPtr ++;
}
}
}
} while (wackPtr);
*newPathPtr = 0;
return newPath;
}
PCWSTR
SanitizePathW (
IN PCWSTR FileSpec
)
{
WCHAR pathSeg [MEMDB_MAX];
PCWSTR wackPtr;
UINT dotNr;
PWSTR newPath = DuplicatePathStringW (FileSpec, 0);
PWSTR newPathPtr = newPath;
BOOL firstPass = TRUE;
do {
wackPtr = wcschr (FileSpec, L'\\');
if (wackPtr) {
if (firstPass && (wackPtr == FileSpec)) {
// this one starts with a wack, let's see if we have double wacks
wackPtr ++;
if (*wackPtr == 0) {
FreePathStringW (newPath);
return NULL;
}
if (*wackPtr == L'\\') {
// this one starts with a double wack
wackPtr ++;
if (!wackPtr) {
FreePathStringW (newPath);
return NULL;
}
wackPtr = wcschr (wackPtr, L'\\');
} else {
wackPtr = wcschr (wackPtr, L'\\');
}
}
firstPass = FALSE;
if (wackPtr) {
StringCopyByteCountABW (
pathSeg,
FileSpec,
wackPtr,
(UINT) sizeof (pathSeg)
);
FileSpec = wackPtr + 1;
} else {
StringCopyByteCountABW (pathSeg, FileSpec, GetEndOfStringW (FileSpec), MAX_WCHAR_PATH);
}
} else {
StringCopyByteCountABW (pathSeg, FileSpec, GetEndOfStringW (FileSpec), MAX_WCHAR_PATH);
}
if (*pathSeg) {
dotNr = pCountDotsW (pathSeg);
if (dotNr>1) {
newPathPtr = pGoBackW (newPathPtr, newPath, dotNr);
if (newPathPtr == NULL) {
DEBUGMSGW ((DBG_WARNING, "Broken path detected:%s", FileSpec));
FreePathStringW (newPath);
return NULL;
}
} else {
StringCopyW (newPathPtr, pathSeg);
newPathPtr = GetEndOfStringW (newPathPtr);
if (wackPtr) {
*newPathPtr = L'\\';
newPathPtr ++;
}
}
}
} while (wackPtr);
*newPathPtr = 0;
return newPath;
}
UINT
pBuildFromDHList (
IN UINT ch1,
IN UINT ch2
)
{
PDHLIST p;
UINT result = 0;
p = g_DHList;
while (p->char1) {
if ((p->char1 == ch1) && (p->char2 == ch2)) {
result = p->result;
break;
}
p++;
}
return result;
}
VOID
_mbssetchar (
OUT PSTR Dest,
IN UINT Char
)
{
if (Char >= 256) {
*(Dest+1) = *((PBYTE)(&Char));
*(Dest) = *((PBYTE)(&Char) + 1);
}
else {
*Dest = (CHAR) Char;
}
}
/*++
Routine Description:
FindLastWack finds the position of the last \ in the given string or NULL if none found
Arguments:
Str - Specifies the string
Return Value:
Pointer to the last occurence of a \ in the string or NULL
--*/
PCSTR
FindLastWackA (
IN PCSTR Str
)
{
PCSTR lastWack = NULL;
if (Str) {
while ((Str = _mbschr (Str, '\\')) != NULL) {
lastWack = Str;
Str++;
}
}
return lastWack;
}
PCWSTR
FindLastWackW (
IN PCWSTR Str
)
{
PCWSTR lastWack = NULL;
if (Str) {
while ((Str = wcschr (Str, L'\\')) != NULL) {
lastWack = Str;
Str++;
}
}
return lastWack;
}
/*++
Routine Description:
GetNodePatternMinMaxLevels treats the given string pattern as a path with \ as separator
and computes the min and max levels of the given node; the root has level 1; if a * is
followed by \ it is treated as a single level (e.g. *\ only enumerates roots)
Arguments:
NodePattern - Specifies the node as a string pattern
FormattedNodePattern - Receives the formatted string, eliminating duplicate * and the last \;
may be the same as NodePattern
MinLevel - Receives the minimum level of a node having this pattern
MaxLevel - Receives the maximum level of a node having this pattern; may be NODE_LEVEL_MAX
Return Value:
TRUE if NodePattern is a valid pattern and the function succeeded, FALSE otherwise
--*/
#define NODESTATE_BEGIN 0
#define NODESTATE_UNC 1
#define NODESTATE_BEGINSEG 2
#define NODESTATE_INSEG 3
#define NODESTATE_ESCAPED 4
#define NODESTATE_STAR 5
#define NODESTATE_STARONLY 6
BOOL
GetNodePatternMinMaxLevelsA (
IN PCSTR NodePattern,
OUT PSTR FormattedNode, OPTIONAL
OUT PDWORD MinLevel, OPTIONAL
OUT PDWORD MaxLevel OPTIONAL
)
{
PCSTR nodePattern = NodePattern;
MBCHAR currCh = 0;
DWORD minLevel = 0;
DWORD maxLevel = 0;
DWORD state = NODESTATE_BEGIN;
BOOL advance;
BOOL copyChar;
if (!NodePattern || *NodePattern == 0) {
return FALSE;
}
while (*nodePattern) {
advance = TRUE;
copyChar = TRUE;
currCh = _mbsnextc (nodePattern);
switch (state) {
case NODESTATE_BEGIN:
switch (currCh) {
case '\\':
state = NODESTATE_UNC;
break;
case '*':
minLevel ++;
maxLevel ++;
state = NODESTATE_INSEG;
advance = FALSE;
break;
case '?':
minLevel ++;
maxLevel ++;
state = NODESTATE_INSEG;
advance = FALSE;
break;
case '^':
minLevel ++;
maxLevel ++;
state = NODESTATE_ESCAPED;
break;
default:
minLevel ++;
maxLevel ++;
state = NODESTATE_INSEG;
break;
}
break;
case NODESTATE_UNC:
minLevel ++;
if (maxLevel != NODE_LEVEL_MAX) {
maxLevel ++;
}
switch (currCh) {
case '\\':
state = NODESTATE_BEGINSEG;
break;
case '*':
state = NODESTATE_BEGINSEG;
advance = FALSE;
break;
case '?':
state = NODESTATE_INSEG;
advance = FALSE;
break;
case '^':
state = NODESTATE_ESCAPED;
break;
default:
state = NODESTATE_INSEG;
break;
}
break;
case NODESTATE_BEGINSEG:
switch (currCh) {
case '\\':
DEBUGMSGA ((DBG_STRINGS, "GetNodeMinMaxLevelsA: two wacks in a row: %s", NodePattern));
return FALSE;
case '*':
minLevel --;
state = NODESTATE_STARONLY;
maxLevel = NODE_LEVEL_MAX;
break;
case '?':
state = NODESTATE_INSEG;
advance = FALSE;
break;
case '^':
state = NODESTATE_ESCAPED;
break;
default:
state = NODESTATE_INSEG;
break;
}
break;
case NODESTATE_STARONLY:
state = NODESTATE_INSEG;
if (currCh == '*') {
copyChar = FALSE;
} else {
minLevel ++;
if (maxLevel != NODE_LEVEL_MAX) {
maxLevel ++;
}
advance = FALSE;
}
break;
case NODESTATE_STAR:
state = NODESTATE_INSEG;
if (currCh == '*') {
copyChar = FALSE;
} else {
advance = FALSE;
}
break;
case NODESTATE_INSEG:
switch (currCh) {
case '\\':
minLevel ++;
if (maxLevel != NODE_LEVEL_MAX) {
maxLevel ++;
}
state = NODESTATE_BEGINSEG;
break;
case '*':
state = NODESTATE_STAR;
maxLevel = NODE_LEVEL_MAX;
break;
case '?':
state = NODESTATE_INSEG;
if (maxLevel != NODE_LEVEL_MAX) {
maxLevel ++;
}
break;
case '^':
state = NODESTATE_ESCAPED;
break;
default:
state = NODESTATE_INSEG;
break;
}
break;
case NODESTATE_ESCAPED:
if (!_mbschr (EscapedCharsA, currCh)) {
DEBUGMSGA ((DBG_STRINGS, "GetNodeMinMaxLevelsA: illegal escaped character: %s", NodePattern));
return FALSE;
}
state = NODESTATE_INSEG;
break;
default:
DEBUGMSGA ((DBG_STRINGS, "GetNodeMinMaxLevelsA: unknown state while processing: %s", NodePattern));
return FALSE;
}
if (advance) {
if (copyChar && FormattedNode) {
if (IsLeadByte (*nodePattern)) {
*FormattedNode = *nodePattern;
FormattedNode ++;
nodePattern ++;
}
*FormattedNode = *nodePattern;
FormattedNode ++;
nodePattern ++;
} else {
nodePattern = _mbsinc (nodePattern);
}
}
}
if (MinLevel) {
*MinLevel = minLevel;
}
if (MaxLevel) {
*MaxLevel = maxLevel;
}
if (FormattedNode) {
*FormattedNode = 0;
}
return TRUE;
}
BOOL
GetNodePatternMinMaxLevelsW (
IN PCWSTR NodePattern,
OUT PWSTR FormattedNode, OPTIONAL
OUT PDWORD MinLevel, OPTIONAL
OUT PDWORD MaxLevel OPTIONAL
)
{
PCWSTR nodePattern = NodePattern;
DWORD minLevel = 0;
DWORD maxLevel = 0;
DWORD state = NODESTATE_BEGIN;
BOOL advance;
BOOL copyChar;
if (!NodePattern || *NodePattern == 0) {
return FALSE;
}
while (*nodePattern) {
advance = TRUE;
copyChar = TRUE;
switch (state) {
case NODESTATE_BEGIN:
switch (*nodePattern) {
case L'\\':
state = NODESTATE_UNC;
break;
case L'*':
minLevel ++;
maxLevel ++;
state = NODESTATE_INSEG;
advance = FALSE;
break;
case L'?':
minLevel ++;
maxLevel ++;
state = NODESTATE_INSEG;
advance = FALSE;
break;
case L'^':
minLevel ++;
maxLevel ++;
state = NODESTATE_ESCAPED;
break;
default:
minLevel ++;
maxLevel ++;
state = NODESTATE_INSEG;
break;
}
break;
case NODESTATE_UNC:
minLevel ++;
if (maxLevel != NODE_LEVEL_MAX) {
maxLevel ++;
}
switch (*nodePattern) {
case L'\\':
state = NODESTATE_BEGINSEG;
break;
case L'*':
state = NODESTATE_BEGINSEG;
advance = FALSE;
break;
case L'?':
state = NODESTATE_INSEG;
advance = FALSE;
break;
case L'^':
state = NODESTATE_ESCAPED;
break;
default:
state = NODESTATE_INSEG;
break;
}
break;
case NODESTATE_BEGINSEG:
switch (*nodePattern) {
case L'\\':
DEBUGMSGW ((DBG_STRINGS, "GetNodeMinMaxLevelsA: two wacks in a row: %s", NodePattern));
return FALSE;
case L'*':
minLevel --;
state = NODESTATE_STARONLY;
maxLevel = NODE_LEVEL_MAX;
break;
case L'?':
state = NODESTATE_INSEG;
advance = FALSE;
break;
case L'^':
state = NODESTATE_ESCAPED;
break;
default:
state = NODESTATE_INSEG;
break;
}
break;
case NODESTATE_STARONLY:
state = NODESTATE_INSEG;
if (*nodePattern == L'*') {
copyChar = FALSE;
} else {
minLevel ++;
if (maxLevel != NODE_LEVEL_MAX) {
maxLevel ++;
}
advance = FALSE;
}
break;
case NODESTATE_STAR:
state = NODESTATE_INSEG;
if (*nodePattern == L'*') {
copyChar = FALSE;
} else {
advance = FALSE;
}
break;
case NODESTATE_INSEG:
switch (*nodePattern) {
case L'\\':
minLevel ++;
if (maxLevel != NODE_LEVEL_MAX) {
maxLevel ++;
}
state = NODESTATE_BEGINSEG;
break;
case L'*':
state = NODESTATE_STAR;
maxLevel = NODE_LEVEL_MAX;
break;
case L'?':
state = NODESTATE_INSEG;
if (maxLevel != NODE_LEVEL_MAX) {
maxLevel ++;
}
break;
case L'^':
state = NODESTATE_ESCAPED;
break;
default:
state = NODESTATE_INSEG;
break;
}
break;
case NODESTATE_ESCAPED:
if (!wcschr (EscapedCharsW, *nodePattern)) {
DEBUGMSGW ((DBG_STRINGS, "GetNodeMinMaxLevelsA: illegal escaped character: %s", NodePattern));
return FALSE;
}
state = NODESTATE_INSEG;
break;
default:
DEBUGMSGW ((DBG_STRINGS, "GetNodeMinMaxLevelsA: unknown state while processing: %s", NodePattern));
return FALSE;
}
if (advance) {
if (copyChar && FormattedNode) {
*FormattedNode = *nodePattern;
FormattedNode ++;
nodePattern ++;
} else {
nodePattern ++;
}
}
}
if (MinLevel) {
*MinLevel = minLevel;
}
if (MaxLevel) {
*MaxLevel = maxLevel;
}
if (FormattedNode) {
*FormattedNode = 0;
}
return TRUE;
}
#if 0
//
// PORTBUG Uses memdb max. #if 0'd out for now.
//
PCSTR
ConvertSBtoDB (
IN PCSTR RootPath,
IN PCSTR FullPath,
IN PCSTR Limit
)
{
CHAR result[MEMDB_MAX];
PCSTR p,p1,q;
PSTR s;
UINT ch;
UINT ch1;
BOOL dhCase = FALSE;
ZeroMemory (result, MAX_PATH);
p = FullPath;
q = RootPath;
s = result;
while (*p && (((DWORD)s - (DWORD)result) < MEMDB_MAX)) {
if (q && *q) {
_mbssetchar (s, _mbsnextc(p));
q = _mbsinc (q);
} else if (Limit && (p >= Limit)) {
_mbssetchar (s, _mbsnextc(p));
} else {
ch = _mbsnextc (p);
//
// It is very important not to make the conversion for characters below A1. Otherwise
// all english letters will be converted to large letters.
//
if (ch >= 0xA1 && ch <= 0xDF) {
// this is a candidate for conversion
// we need to see if there is a special Dakutenn/Handakuten conversion
dhCase = FALSE;
p1 = _mbsinc (p);
if (p1) {
ch1 = _mbsnextc (p1);
ch1 = pBuildFromDHList (ch, ch1);
if (ch1) {
p = _mbsinc (p);
_mbssetchar (s, ch1);
dhCase = TRUE;
}
}
if (!dhCase) {
_mbssetchar (s, _mbbtombc (ch));
}
} else {
_mbssetchar (s, ch);
}
}
p = _mbsinc (p);
s = _mbsinc (s);
}
result [MAX_PATH - 1] = 0;
return (DuplicatePathString (result, 0));
}
#endif
ULONGLONG
StringToUint64A (
IN PCSTR String,
OUT PCSTR *EndOfNumber OPTIONAL
)
{
ULONGLONG n;
n = 0;
while (*String >= '0' && *String <= '9') {
n = n * 10 + *String - '0';
String++;
}
if (EndOfNumber) {
*EndOfNumber = String;
}
return n;
}
ULONGLONG
StringToUint64W (
IN PCWSTR String,
OUT PCWSTR *EndOfNumber OPTIONAL
)
{
ULONGLONG n;
n = 0;
while (*String >= L'0' && *String <= L'9') {
n = n * 10 + *String - L'0';
String++;
}
if (EndOfNumber) {
*EndOfNumber = String;
}
return n;
}
LONGLONG
StringToInt64A (
IN PCSTR String,
OUT PCSTR *EndOfNumber OPTIONAL
)
{
LONGLONG n;
BOOL negate = FALSE;
if (*String == '-') {
negate = TRUE;
String++;
} else if (*String == '+') {
String++;
}
n = 0;
while (*String >= '0' && *String <= '9') {
n = n * 10 + *String - '0';
String++;
}
if (negate) {
n = -n;
}
if (EndOfNumber) {
*EndOfNumber = String;
}
return n;
}
LONGLONG
StringToInt64W (
IN PCWSTR String,
OUT PCWSTR *EndOfNumber OPTIONAL
)
{
LONGLONG n;
BOOL negate = FALSE;
if (*String == L'-') {
negate = TRUE;
String++;
} else if (*String == L'+') {
String++;
}
n = 0;
while (*String >= L'0' && *String <= L'9') {
n = n * 10 + *String - L'0';
String++;
}
if (negate) {
n = -n;
}
if (EndOfNumber) {
*EndOfNumber = String;
}
return n;
}
Reference in a new issue View git blame Copy permalink