191 lines
4.9 KiB
C++
191 lines
4.9 KiB
C++
|
|
/*++
|
|
|
|
Copyright (c) 2000-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
EmulateGetCommandLine.cpp
|
|
|
|
Abstract:
|
|
|
|
This app uses GetCommandLine() to figure out what the drive letter of the
|
|
CD-ROM is. Unfortunately the behaviour of this API is different from Win9x
|
|
to NT:
|
|
|
|
Original command line:
|
|
E:\Final Doom\Doom95.exe -dm -cdrom
|
|
|
|
NT's GetCommandLine() returns:
|
|
Doom95.exe -dm -cdrom
|
|
|
|
Win9x's GetCommandLine() returns:
|
|
E:\FINALD~1\DOOM95.EXE -dm -cdrom
|
|
|
|
This app returns short pathnames for GetCommandLine and GetModuleFileName.
|
|
|
|
Notes:
|
|
|
|
This is a general purpose shim.
|
|
|
|
Created:
|
|
|
|
01/03/2000 markder Created
|
|
09/26/2000 mnikkel GetModuleFileName added
|
|
11/10/2000 robkenny Fixed PREFIX bugs, removed W routines.
|
|
11/21/2000 prashkud Fixed the GetCommandLineA hook bug when the CommandLine
|
|
had the executable name/path with spaces. Used
|
|
AppAndCommandLine functions.
|
|
02/27/2001 robkenny Converted to use CString
|
|
05/02/2001 pierreys If buffer is too small, GetModuleFileNameA puts \0 at end of it like 9X.
|
|
|
|
|
|
--*/
|
|
|
|
#include "precomp.h"
|
|
|
|
IMPLEMENT_SHIM_BEGIN(EmulateGetCommandLine)
|
|
#include "ShimHookMacro.h"
|
|
|
|
APIHOOK_ENUM_BEGIN
|
|
APIHOOK_ENUM_ENTRY(GetCommandLineA)
|
|
APIHOOK_ENUM_ENTRY(GetModuleFileNameA)
|
|
APIHOOK_ENUM_END
|
|
|
|
|
|
char * g_lpszCommandLine = NULL;
|
|
|
|
/*++
|
|
|
|
This stub function appends the commandline returned from GetCommandLine() to a
|
|
pre-determined path to emulate Win9x behavior.
|
|
|
|
--*/
|
|
|
|
LPSTR
|
|
APIHOOK(GetCommandLineA)(
|
|
void
|
|
)
|
|
{
|
|
// Been here, done that
|
|
if (g_lpszCommandLine)
|
|
{
|
|
return g_lpszCommandLine;
|
|
}
|
|
|
|
LPSTR lpszOrig = ORIGINAL_API(GetCommandLineA)();
|
|
|
|
// Seperate the app name and command line
|
|
AppAndCommandLine AppCmdLine(NULL, lpszOrig);
|
|
|
|
CSTRING_TRY
|
|
{
|
|
// retrieve the original command line
|
|
CString csAppName(AppCmdLine.GetApplicationName());
|
|
|
|
if (csAppName.Find(L' ') == -1)
|
|
{
|
|
// If no spaces in app name, return the original command line.
|
|
g_lpszCommandLine = lpszOrig;
|
|
}
|
|
else
|
|
{
|
|
// Spaces found so return short app path name
|
|
// and rest of original command line
|
|
csAppName.GetShortPathName();
|
|
csAppName += L" ";
|
|
csAppName += AppCmdLine.GetCommandlineNoAppName();
|
|
g_lpszCommandLine = csAppName.ReleaseAnsi();
|
|
|
|
LOGN( eDbgLevelError,
|
|
"[GetCommandLineA] Changed Command Line from <%s> to <%s>.",
|
|
lpszOrig, g_lpszCommandLine);
|
|
}
|
|
}
|
|
CSTRING_CATCH
|
|
{
|
|
g_lpszCommandLine = lpszOrig;
|
|
}
|
|
|
|
return g_lpszCommandLine;
|
|
}
|
|
|
|
|
|
DWORD
|
|
APIHOOK(GetModuleFileNameA)(
|
|
HMODULE hModule, // handle to module
|
|
LPSTR lpFilename, // file name of module
|
|
DWORD nSize // size of buffer
|
|
)
|
|
{
|
|
|
|
CSTRING_TRY
|
|
{
|
|
CString csExeFileName;
|
|
DWORD len;
|
|
|
|
len = csExeFileName.GetModuleFileNameW(hModule);
|
|
CString csLongFileName(csExeFileName);
|
|
|
|
if (csExeFileName.Find(L' ') > -1)
|
|
{
|
|
// Spaces found so return short app path name
|
|
// The return value
|
|
len = csExeFileName.GetShortPathNameW();
|
|
|
|
LOGN(
|
|
eDbgLevelError,
|
|
"[GetModuleFileNameA] Changed <%s> to <%s>.",
|
|
csLongFileName.GetAnsi(), csExeFileName.GetAnsi());
|
|
|
|
}
|
|
|
|
//
|
|
// From 9X's PELDR.C. If the buffer has no room for the '\0', 9X stuff the 0 at the
|
|
// last byte.
|
|
//
|
|
if (nSize) {
|
|
// len = pmte->iFileNameLen;
|
|
if (len >= nSize) {
|
|
len = nSize - 1;
|
|
LOGN(eDbgLevelError,
|
|
"[GetModuleFileNameA] Will shorten <%s> to %d characters.",
|
|
csLongFileName.GetAnsi(), len);
|
|
}
|
|
|
|
RtlCopyMemory(lpFilename, csExeFileName.GetAnsi() /* pmte->cfhid.lpFilename */, len);
|
|
lpFilename[len] = 0;
|
|
}
|
|
|
|
|
|
// Returned the double buffered name len.
|
|
return len;
|
|
}
|
|
CSTRING_CATCH
|
|
{
|
|
// If error return original api.
|
|
return ORIGINAL_API(GetModuleFileNameA)(
|
|
hModule,
|
|
lpFilename,
|
|
nSize);
|
|
|
|
}
|
|
}
|
|
|
|
/*++
|
|
|
|
Register hooked functions
|
|
|
|
--*/
|
|
|
|
HOOK_BEGIN
|
|
|
|
APIHOOK_ENTRY(KERNEL32.DLL, GetCommandLineA)
|
|
APIHOOK_ENTRY(KERNEL32.DLL, GetModuleFileNameA)
|
|
|
|
HOOK_END
|
|
|
|
|
|
IMPLEMENT_SHIM_END
|
|
|