Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

138 lines
3.0 KiB

/*++
Copyright (c) 2000-2001 Microsoft Corporation
Module Name:
HandleDBCSUserName.cpp
Abstract:
ShFolder.Exe is failing installation when its app path has Hi Ascii characters (typically DBCS).
When app path is double quoted, the bug code is not processed and install will succeed.
This shim wrap the command line app path name with double quote at GetCommandLineA.
Example:
C:\DOCUME~1\DBCS\LOCALS~1\Temp\_ISTMP1.DIR\_ISTMP0.DIR\shfolder.exe /Q:a
"C:\DOCUME~1\DBCS\LOCALS~1\Temp\_ISTMP1.DIR\_ISTMP0.DIR\shfolder.exe" /Q:a
More info:
Self extractor ShFolder.Exe checks its app path name with space char 0x20 on signed char basis.
Hi Ascii character included in path treated below 0x20 and path is chopped at there.
When path has double quote at the top, the problem check is not proceeded.
History:
04/09/2001 hioh Created
--*/
#include "precomp.h"
IMPLEMENT_SHIM_BEGIN(HandleDBCSUserName)
#include "ShimHookMacro.h"
//
// Add APIs that you wish to hook to this macro construction.
//
APIHOOK_ENUM_BEGIN
APIHOOK_ENUM_ENTRY(GetCommandLineA)
APIHOOK_ENUM_END
char * g_lpszCommandLine = NULL;
CRITICAL_SECTION g_csCmdLine;
class CAutoLock
{
public:
CAutoLock()
{
EnterCriticalSection(&g_csCmdLine);
}
~CAutoLock()
{
LeaveCriticalSection(&g_csCmdLine);
}
};
/*++
Wrap the application path name with double quote.
--*/
LPSTR
APIHOOK(GetCommandLineA)(
void
)
{
CAutoLock lock;
if (g_lpszCommandLine)
{
// Been here, done that
return g_lpszCommandLine;
}
g_lpszCommandLine = ORIGINAL_API(GetCommandLineA)();
CSTRING_TRY
{
AppAndCommandLine appCmdLine(NULL, g_lpszCommandLine);
// The command line is often only the arguments: no application name
// If the "application name" doesn't exist, don't bother to put quotes
// around it.
DWORD dwAttr = GetFileAttributesW(appCmdLine.GetApplicationName());
if (dwAttr != INVALID_FILE_ATTRIBUTES)
{
CString csDQ = L'"';
CString csCL = csDQ;
CString csNA = appCmdLine.GetCommandlineNoAppName();
csCL += appCmdLine.GetApplicationName();
csCL += csDQ;
if (!csNA.IsEmpty())
{ // Add the rest
csCL += L" ";
csCL += csNA;
}
g_lpszCommandLine = csCL.ReleaseAnsi();
}
}
CSTRING_CATCH
{
// Do nothing, g_lpszCommandLine is initialized with good value
}
return g_lpszCommandLine;
}
BOOL
NOTIFY_FUNCTION(
DWORD fdwReason)
{
if (fdwReason == DLL_PROCESS_ATTACH)
{
return InitializeCriticalSectionAndSpinCount(&g_csCmdLine, 0x80000000);
}
return TRUE;
}
/*++
Register hooked functions
--*/
HOOK_BEGIN
CALL_NOTIFY_FUNCTION
APIHOOK_ENTRY(KERNEL32.DLL, GetCommandLineA)
HOOK_END
IMPLEMENT_SHIM_END