154 lines
3.3 KiB
C++
154 lines
3.3 KiB
C++
/*++
|
|
|
|
Copyright (c) 2000 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
ShimMechanismVerificationTest3.cpp
|
|
|
|
Abstract:
|
|
|
|
This DLL serves as a test for the shim michanism.
|
|
|
|
Notes:
|
|
|
|
|
|
History:
|
|
|
|
11/13/2000 diaaf Created
|
|
11/27/2000 diaaf modified to support the new macros.
|
|
04/02/2001 diaaf modified to support the new macros.
|
|
|
|
--*/
|
|
|
|
#include "precomp.h"
|
|
|
|
IMPLEMENT_SHIM_BEGIN(ShimMechanismVerificationTest3)
|
|
#include "ShimHookMacroEx.h"
|
|
|
|
//
|
|
// Add APIs that you wish to hook to this macro construction.
|
|
//
|
|
APIHOOK_ENUM_BEGIN
|
|
APIHOOK_ENUM_ENTRY(GetCommandLineA)
|
|
APIHOOK_ENUM_ENTRY(GetCommandLineW)
|
|
APIHOOK_ENUM_END
|
|
|
|
//
|
|
// Global Variables
|
|
//
|
|
CHAR *gszCommandLineA3;
|
|
WCHAR *gwcsCommandLineW3;
|
|
|
|
/*++
|
|
|
|
This stub function intercepts all calls to GetCommandLineA
|
|
and appends "1" to the returned string.
|
|
|
|
--*/
|
|
|
|
LPSTR APIHOOK(GetCommandLineA)()
|
|
{
|
|
CHAR szAppendValue[] = "3";
|
|
LPSTR szReturnValue;
|
|
|
|
DPF("ShimMechanismVerificationTest3", eDbgLevelInfo,
|
|
"GetCommandLineA called.\n");
|
|
|
|
szReturnValue = ORIGINAL_API(GetCommandLineA)();
|
|
|
|
gszCommandLineA3 = (CHAR*)LocalAlloc(LPTR, strlen(szReturnValue)
|
|
+ strlen(szAppendValue) + sizeof(CHAR));
|
|
|
|
strcpy(gszCommandLineA3, szReturnValue);
|
|
strcat(gszCommandLineA3, szAppendValue);
|
|
|
|
LOG("ShimMechanismVerificationTest3", eDbgLevelWarning,
|
|
"GetCommandLineA is returning \"%S\".", gszCommandLineA3);
|
|
|
|
return gszCommandLineA3;
|
|
}
|
|
|
|
/*++
|
|
|
|
This stub function intercepts all calls to GetCommandLineW
|
|
and appends "1" to the returned string.
|
|
|
|
--*/
|
|
|
|
LPWSTR APIHOOK(GetCommandLineW)()
|
|
{
|
|
WCHAR wszAppendValue[] = L"3";
|
|
LPWSTR wszReturnValue;
|
|
|
|
DPF("ShimMechanismVerificationTest3", eDbgLevelInfo,
|
|
"GetCommandLineW called.\n");
|
|
|
|
wszReturnValue = ORIGINAL_API(GetCommandLineW)();
|
|
|
|
gwcsCommandLineW3 = (WCHAR*)LocalAlloc(LPTR, wcslen(wszReturnValue)
|
|
+ wcslen(wszAppendValue) + sizeof(WCHAR));
|
|
|
|
wcscpy(gwcsCommandLineW3, wszReturnValue);
|
|
wcscat(gwcsCommandLineW3, wszAppendValue);
|
|
|
|
LOG("ShimMechanismVerificationTest3", eDbgLevelWarning,
|
|
"GetCommandLineW is returning \"%S\".", gwcsCommandLineW3);
|
|
|
|
return gwcsCommandLineW3;
|
|
}
|
|
|
|
/*++
|
|
|
|
Handle DLL_PROCESS_ATTACH and DLL_PROCESS_DETACH in your notify function
|
|
to do initialization and uninitialization.
|
|
|
|
--*/
|
|
BOOL
|
|
NOTIFY_FUNCTION(
|
|
DWORD fdwReason
|
|
)
|
|
{
|
|
if (fdwReason == DLL_PROCESS_ATTACH) {
|
|
gszCommandLineA3 = NULL;
|
|
gwcsCommandLineW3 = NULL;
|
|
DPF("ShimMechanismVerificationTest3", eDbgLevelInfo,
|
|
"ShimMechanismVerificationTest3 initialized.");
|
|
} else {
|
|
if (gszCommandLineA3)
|
|
{
|
|
LocalFree(gszCommandLineA3);
|
|
gszCommandLineA3 = NULL;
|
|
}
|
|
|
|
if (gwcsCommandLineW3)
|
|
{
|
|
LocalFree(gwcsCommandLineW3);
|
|
gwcsCommandLineW3 = NULL;
|
|
}
|
|
DPF("ShimMechanismVerificationTest3", eDbgLevelInfo,
|
|
"ShimMechanismVerificationTest3 uninitialized.");
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
/*++
|
|
|
|
Register hooked functions
|
|
|
|
--*/
|
|
|
|
HOOK_BEGIN
|
|
|
|
CALL_NOTIFY_FUNCTION
|
|
|
|
APIHOOK_ENTRY(KERNEL32.DLL, GetCommandLineA)
|
|
APIHOOK_ENTRY(KERNEL32.DLL, GetCommandLineW)
|
|
|
|
HOOK_END
|
|
|
|
IMPLEMENT_SHIM_END
|
|
|
|
|
|
|