357 lines
11 KiB
C++
357 lines
11 KiB
C++
|
/*++
|
||
|
|
|
||
|
|
Copyright (c) 2000, 2001 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
AppAndCommandLine.cpp
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
|
||
|
|
This class takes an application name and command line and
|
||
|
|
parses them *exactly* as as they would by CreateProcess.
|
||
|
|
If the Set routine returns TRUE, then the application name
|
||
|
|
will contain a value; however, it does not guarantee that the application
|
||
|
|
actually exists.
|
||
|
|
|
||
|
|
Example:
|
||
|
|
AppAndCommandLine.Set(NULL, "notepad.exe readme.txt");
|
||
|
|
GetApplicationName() == "notepad.exe"
|
||
|
|
GetCommandline() == "notepad.exe readme.txt"
|
||
|
|
GetCommandlineNoAppName() == "readme.txt"
|
||
|
|
|
||
|
|
|
||
|
|
Notes:
|
||
|
|
|
||
|
|
None
|
||
|
|
|
||
|
|
History:
|
||
|
|
|
||
|
|
07/21/2000 robkenny Created
|
||
|
|
12/18/2000 prashkud Modified GetAppAndCommandLine to fix the AV
|
||
|
|
caused by EmulateGetCommandLine when the
|
||
|
|
layer was applied to Oregon Trail 4th Edition.
|
||
|
|
03/04/2001 robkenny Converted to use CString
|
||
|
|
03/29/2001 prashkud Modified GetAppnameAndCommandline to set
|
||
|
|
the application name even when there is only
|
||
|
|
Commandline passed and the application name
|
||
|
|
passed is NULL.
|
||
|
|
08/14/2001 robkenny Moved code inside the ShimLib namespace.
|
||
|
|
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
#include "ShimLib.h"
|
||
|
|
#include "StrSafe.h"
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
namespace ShimLib
|
||
|
|
{
|
||
|
|
|
||
|
|
AppAndCommandLine::AppAndCommandLine(const char * applicationName, const char * commandLine)
|
||
|
|
{
|
||
|
|
CString csApp(applicationName);
|
||
|
|
CString csCl(commandLine);
|
||
|
|
|
||
|
|
GetAppnameAndCommandline(csApp.GetNIE(), csCl.GetNIE());
|
||
|
|
}
|
||
|
|
|
||
|
|
AppAndCommandLine::AppAndCommandLine(const WCHAR * applicationName, const WCHAR * commandLine)
|
||
|
|
{
|
||
|
|
GetAppnameAndCommandline(applicationName, commandLine);
|
||
|
|
}
|
||
|
|
|
||
|
|
// If the application name is the first entry on the command line,
|
||
|
|
// we convert the appName to its short name; thereby removing any spaces.
|
||
|
|
const CString & AppAndCommandLine::GetShortCommandLine()
|
||
|
|
{
|
||
|
|
// If lpCommandLineNoAppName is not the same as lpCommandLine,
|
||
|
|
// then the command line contains the application name.
|
||
|
|
if ( csShortCommandLine.IsEmpty() && // Haven't been here
|
||
|
|
!csApplicationName.IsEmpty() && // Set() has been called
|
||
|
|
!csCommandLine.IsEmpty() && // Set() has been called
|
||
|
|
csShortCommandLine.GetLength() != csCommandLine.GetLength()) // Command line actually contains app name
|
||
|
|
{
|
||
|
|
csShortCommandLine = csApplicationName;
|
||
|
|
csShortCommandLine.GetShortPathNameW();
|
||
|
|
csShortCommandLine += L" ";
|
||
|
|
csShortCommandLine += csCommandLineNoAppName;
|
||
|
|
}
|
||
|
|
|
||
|
|
// If we still don't have a short version of the command line,
|
||
|
|
// just duplicate the current command line
|
||
|
|
if (csShortCommandLine.IsEmpty())
|
||
|
|
{
|
||
|
|
csShortCommandLine = csCommandLine;
|
||
|
|
}
|
||
|
|
return csShortCommandLine;
|
||
|
|
}
|
||
|
|
|
||
|
|
BOOL AppAndCommandLine::GetAppnameAndCommandline(const WCHAR * lpcApp, const WCHAR * lpcCl)
|
||
|
|
{
|
||
|
|
BOOL SearchRetry = TRUE;
|
||
|
|
ULONG Length = 0;
|
||
|
|
WCHAR * NameBuffer = NULL;
|
||
|
|
BOOL success = FALSE;
|
||
|
|
DWORD dwAppNameLen = 0;
|
||
|
|
CString csTempAppName;
|
||
|
|
BOOL bFound = TRUE;
|
||
|
|
|
||
|
|
// It is really, really bad to remove the const from the Get,
|
||
|
|
// However we never change the length of the string, so it should be okay
|
||
|
|
WCHAR * lpApplicationName = (WCHAR *)lpcApp;
|
||
|
|
WCHAR * lpCommandLine = (WCHAR *)lpcCl;
|
||
|
|
|
||
|
|
// The following is done as there are lot of instances when the
|
||
|
|
// the pointer is not NULL but contains an EMPTY string.
|
||
|
|
if (lpApplicationName && !(*lpApplicationName))
|
||
|
|
{
|
||
|
|
lpApplicationName = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (lpCommandLine && !(*lpCommandLine))
|
||
|
|
{
|
||
|
|
lpCommandLine = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (lpApplicationName == NULL && lpCommandLine == NULL)
|
||
|
|
{
|
||
|
|
// Degenerate case
|
||
|
|
csApplicationName = L"";
|
||
|
|
csCommandLine = csApplicationName;
|
||
|
|
csCommandLineNoAppName = csApplicationName;
|
||
|
|
return FALSE; // Didn't find application name
|
||
|
|
}
|
||
|
|
|
||
|
|
csCommandLine = lpCommandLine;
|
||
|
|
|
||
|
|
DPF("Common",
|
||
|
|
eDbgLevelSpew,
|
||
|
|
"[AppAndCommandLineT::Set] BEFORE App(%S) CL(%S)\n",
|
||
|
|
lpApplicationName, lpCommandLine);
|
||
|
|
|
||
|
|
if (lpApplicationName == NULL)
|
||
|
|
{
|
||
|
|
DWORD fileattr;
|
||
|
|
WCHAR TempChar;
|
||
|
|
//
|
||
|
|
// Locate the image
|
||
|
|
//
|
||
|
|
|
||
|
|
NameBuffer = (WCHAR *) malloc(MAX_PATH * sizeof( WCHAR ));
|
||
|
|
if ( !NameBuffer )
|
||
|
|
{
|
||
|
|
goto errorExit;
|
||
|
|
}
|
||
|
|
lpApplicationName = lpCommandLine;
|
||
|
|
WCHAR * TempNull = lpApplicationName;
|
||
|
|
WCHAR * WhiteScan = lpApplicationName;
|
||
|
|
DWORD QuoteFound = 0;
|
||
|
|
|
||
|
|
//
|
||
|
|
// check for lead quote
|
||
|
|
//
|
||
|
|
if ( *WhiteScan == L'\"' ) {
|
||
|
|
SearchRetry = FALSE;
|
||
|
|
WhiteScan++;
|
||
|
|
lpApplicationName = WhiteScan;
|
||
|
|
while(*WhiteScan) {
|
||
|
|
if ( *WhiteScan == L'\"' ) {
|
||
|
|
TempNull = WhiteScan;
|
||
|
|
QuoteFound = 2;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
WhiteScan++;
|
||
|
|
TempNull = WhiteScan;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else {
|
||
|
|
retrywsscan:
|
||
|
|
lpApplicationName = lpCommandLine;
|
||
|
|
while(*WhiteScan) {
|
||
|
|
if ( *WhiteScan == L' ' ||
|
||
|
|
*WhiteScan == L'\t' ) {
|
||
|
|
TempNull = WhiteScan;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
WhiteScan++;
|
||
|
|
TempNull = WhiteScan;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
TempChar = *TempNull;
|
||
|
|
*TempNull = 0;
|
||
|
|
|
||
|
|
WCHAR * filePart;
|
||
|
|
Length = SearchPathW(
|
||
|
|
NULL,
|
||
|
|
lpApplicationName,
|
||
|
|
L".exe",
|
||
|
|
MAX_PATH,
|
||
|
|
NameBuffer,
|
||
|
|
&filePart
|
||
|
|
)*sizeof(WCHAR);
|
||
|
|
|
||
|
|
if (Length != 0 && Length < MAX_PATH * sizeof( WCHAR )) {
|
||
|
|
//
|
||
|
|
// SearchPathW worked, but file might be a directory
|
||
|
|
// if this happens, we need to keep trying
|
||
|
|
//
|
||
|
|
fileattr = GetFileAttributesW(NameBuffer);
|
||
|
|
if ( fileattr != 0xffffffff &&
|
||
|
|
(fileattr & FILE_ATTRIBUTE_DIRECTORY) ) {
|
||
|
|
Length = 0;
|
||
|
|
} else {
|
||
|
|
Length++;
|
||
|
|
Length++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ( !Length || Length >= MAX_PATH<<1 ) {
|
||
|
|
|
||
|
|
//
|
||
|
|
// restore the command line
|
||
|
|
//
|
||
|
|
|
||
|
|
*TempNull = TempChar;
|
||
|
|
lpApplicationName = lpCommandLine;
|
||
|
|
|
||
|
|
//
|
||
|
|
// If we still have command line left, then keep going
|
||
|
|
// the point is to march through the command line looking
|
||
|
|
// for whitespace so we can try to find an image name
|
||
|
|
// launches of things like:
|
||
|
|
// c:\word 95\winword.exe /embedding -automation
|
||
|
|
// require this. Our first iteration will stop at c:\word, our next
|
||
|
|
// will stop at c:\word 95\winword.exe
|
||
|
|
//
|
||
|
|
if (*WhiteScan && SearchRetry) {
|
||
|
|
WhiteScan++;
|
||
|
|
TempNull = WhiteScan;
|
||
|
|
goto retrywsscan;
|
||
|
|
}
|
||
|
|
|
||
|
|
// If we are here then the Application has not been found.
|
||
|
|
// We used to send back lpApplicationName as NULL earlier
|
||
|
|
// but now instead we fill the ApplicationName with the
|
||
|
|
// commandline till the first space or tab.This was added
|
||
|
|
// to support EmulateMissingExe SHIM which will fail if
|
||
|
|
// we return NULL as we used to earlier.
|
||
|
|
bFound = FALSE;
|
||
|
|
|
||
|
|
if (QuoteFound == 0)
|
||
|
|
{
|
||
|
|
// No quotes were found.
|
||
|
|
lpApplicationName = lpCommandLine;
|
||
|
|
|
||
|
|
// Since we just reset to the entire command line, we need to skip leading white space
|
||
|
|
SkipBlanksW(lpApplicationName);
|
||
|
|
|
||
|
|
TempNull = lpApplicationName;
|
||
|
|
|
||
|
|
while (*TempNull)
|
||
|
|
{
|
||
|
|
if ((*TempNull == L' ') ||
|
||
|
|
(*TempNull == L'\t') )
|
||
|
|
{
|
||
|
|
TempChar = *TempNull;
|
||
|
|
*TempNull = 0;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
TempNull++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Quotes were found.
|
||
|
|
lpApplicationName = lpCommandLine + 1; // Skip leading quote.
|
||
|
|
*TempNull = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
csTempAppName = lpApplicationName;
|
||
|
|
*TempNull = TempChar;
|
||
|
|
dwAppNameLen = (DWORD)(TempNull - lpCommandLine) + QuoteFound;
|
||
|
|
lpApplicationName = (WCHAR*)csTempAppName.Get();
|
||
|
|
|
||
|
|
goto successExit;
|
||
|
|
}
|
||
|
|
|
||
|
|
dwAppNameLen = (DWORD)(TempNull - lpApplicationName) + QuoteFound;
|
||
|
|
|
||
|
|
//
|
||
|
|
// restore the command line
|
||
|
|
//
|
||
|
|
|
||
|
|
*TempNull = TempChar;
|
||
|
|
lpApplicationName = NameBuffer;
|
||
|
|
}
|
||
|
|
else if (lpCommandLine == NULL || *lpCommandLine == 0 )
|
||
|
|
{
|
||
|
|
lpCommandLine = lpApplicationName;
|
||
|
|
|
||
|
|
dwAppNameLen = wcslen(lpApplicationName);
|
||
|
|
}
|
||
|
|
|
||
|
|
// If they provided both, check to see if the app name
|
||
|
|
// is the first entry on the command line.
|
||
|
|
else if (lpApplicationName != NULL && lpCommandLine != NULL )
|
||
|
|
{
|
||
|
|
int appNameLen = wcslen(lpApplicationName);
|
||
|
|
|
||
|
|
if (
|
||
|
|
_wcsnicmp(lpApplicationName, lpCommandLine, appNameLen) == 0 &&
|
||
|
|
(lpCommandLine[appNameLen] == 0 || iswspace(lpCommandLine[appNameLen]))
|
||
|
|
)
|
||
|
|
{
|
||
|
|
// lpApplicationName is the first entry on the command line
|
||
|
|
dwAppNameLen = appNameLen;
|
||
|
|
}
|
||
|
|
// check for quoted lpApplicationName
|
||
|
|
else if (
|
||
|
|
lpCommandLine[0] == L'"' &&
|
||
|
|
_wcsnicmp(lpApplicationName, lpCommandLine+1, appNameLen) == 0 &&
|
||
|
|
lpCommandLine[appNameLen+1] == L'"' &&
|
||
|
|
(lpCommandLine[appNameLen+2] == 0 || iswspace(lpCommandLine[appNameLen+2]))
|
||
|
|
)
|
||
|
|
{
|
||
|
|
// lpApplicationName is the first *quoted* entry on the command line
|
||
|
|
dwAppNameLen = appNameLen + 2;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
// Didn't find the application name at the beginning of the command line
|
||
|
|
dwAppNameLen = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
successExit:
|
||
|
|
if (bFound)
|
||
|
|
{
|
||
|
|
success = TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
csApplicationName = lpApplicationName;
|
||
|
|
csCommandLineNoAppName = lpCommandLine + dwAppNameLen;
|
||
|
|
csCommandLineNoAppName.TrimLeft();
|
||
|
|
|
||
|
|
errorExit:
|
||
|
|
|
||
|
|
free(NameBuffer);
|
||
|
|
|
||
|
|
DPF("Common",
|
||
|
|
eDbgLevelSpew,
|
||
|
|
"[AppAndCommandLineT::Set] AFTER App(%S) CL(%S)\n",
|
||
|
|
csApplicationName.Get(), csCommandLine.Get());
|
||
|
|
|
||
|
|
DPF("Common",
|
||
|
|
eDbgLevelSpew,
|
||
|
|
"[AppAndCommandLineT::Set] CL without App(%S)\n",
|
||
|
|
csCommandLineNoAppName.Get());
|
||
|
|
|
||
|
|
return success;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
}; // end of namespace ShimLib
|