148 lines
3.1 KiB
C++
148 lines
3.1 KiB
C++
|
/*
|
||
|
|
Copyright (c) 2000 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
|
||
|
|
WinStone99.cpp
|
||
|
|
|
||
|
|
Bug: Whistler #185797
|
||
|
|
|
||
|
|
Problem:
|
||
|
|
|
||
|
|
Only for Winstone '99. Winstone uses scripts that hide the taskbar, and print stuff.
|
||
|
|
PrintUI displays a balloon tip informing the user that the printing job is done (this is a
|
||
|
|
new addition to Whistler).
|
||
|
|
|
||
|
|
The balloon tip utilizes user tracking code, and is hence left stationary on the machine,
|
||
|
|
till the user clicks on it, or there is 10 seconds of user activity on the machine.
|
||
|
|
Winstone runs these automated tests, hence there is no user activity on the machine, when
|
||
|
|
the balloon is up, so it stays up forever.
|
||
|
|
|
||
|
|
Later, when Winstone tries to enumerate the application windows, the presence of the
|
||
|
|
balloon tip throws it off track. Hence this apphack that disables the display of these
|
||
|
|
balloons when Winstone is running.
|
||
|
|
|
||
|
|
Winstone is a collection of Visual Test scripts, and zdbui32.exe is the only exe that runs
|
||
|
|
throughout when Winstone is running. So disable user tracking when Winstone is running.
|
||
|
|
|
||
|
|
Solution:
|
||
|
|
|
||
|
|
Disable display of balloon tips when Winstone is running and enable it when Winstone is
|
||
|
|
finished
|
||
|
|
|
||
|
|
Details:
|
||
|
|
|
||
|
|
Winstone sends a message to the tray that disables the balloon tip when it is running, and
|
||
|
|
re-sends the message to the tray when it is done, so that the tray can enable the balloon
|
||
|
|
tip
|
||
|
|
|
||
|
|
History:
|
||
|
|
|
||
|
|
09/20/2000 ramkumar Created
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "precomp.h"
|
||
|
|
#include <shlapip.h>
|
||
|
|
|
||
|
|
IMPLEMENT_SHIM_BEGIN(WinStone99)
|
||
|
|
#include "ShimHookMacro.h"
|
||
|
|
|
||
|
|
APIHOOK_ENUM_BEGIN
|
||
|
|
APIHOOK_ENUM_ENTRY(GetCommandLineA)
|
||
|
|
APIHOOK_ENUM_ENTRY(GetCommandLineW)
|
||
|
|
APIHOOK_ENUM_END
|
||
|
|
|
||
|
|
BOOL g_bInit = FALSE;
|
||
|
|
HWND g_hwndTray;
|
||
|
|
UINT g_uEnableBalloonMessage;
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Initialize
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
VOID
|
||
|
|
WinStone99_Initialize()
|
||
|
|
{
|
||
|
|
if (!g_bInit)
|
||
|
|
{
|
||
|
|
g_bInit = TRUE;
|
||
|
|
|
||
|
|
g_uEnableBalloonMessage = RegisterWindowMessage(ENABLE_BALLOONTIP_MESSAGE);
|
||
|
|
if (!g_uEnableBalloonMessage)
|
||
|
|
{
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
g_hwndTray = FindWindowA(WNDCLASS_TRAYNOTIFY, NULL);
|
||
|
|
if (g_hwndTray)
|
||
|
|
{
|
||
|
|
SendMessage(g_hwndTray, g_uEnableBalloonMessage, FALSE, 0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Initialize.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
LPSTR
|
||
|
|
APIHOOK(GetCommandLineA)()
|
||
|
|
{
|
||
|
|
WinStone99_Initialize();
|
||
|
|
return ORIGINAL_API(GetCommandLineA)();
|
||
|
|
}
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Initialize.
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
LPWSTR
|
||
|
|
APIHOOK(GetCommandLineW)()
|
||
|
|
{
|
||
|
|
WinStone99_Initialize();
|
||
|
|
return ORIGINAL_API(GetCommandLineW)();
|
||
|
|
}
|
||
|
|
|
||
|
|
/*++
|
||
|
|
|
||
|
|
Register hooked functions
|
||
|
|
|
||
|
|
--*/
|
||
|
|
|
||
|
|
BOOL
|
||
|
|
NOTIFY_FUNCTION(
|
||
|
|
DWORD fdwReason
|
||
|
|
)
|
||
|
|
{
|
||
|
|
if (fdwReason == DLL_PROCESS_DETACH)
|
||
|
|
{
|
||
|
|
if (g_bInit)
|
||
|
|
{
|
||
|
|
if (g_hwndTray)
|
||
|
|
{
|
||
|
|
SendMessage(g_hwndTray, g_uEnableBalloonMessage, TRUE, 0);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return TRUE;
|
||
|
|
}
|
||
|
|
|
||
|
|
HOOK_BEGIN
|
||
|
|
|
||
|
|
CALL_NOTIFY_FUNCTION
|
||
|
|
APIHOOK_ENTRY(KERNEL32.DLL, GetCommandLineA)
|
||
|
|
APIHOOK_ENTRY(KERNEL32.DLL, GetCommandLineW)
|
||
|
|
|
||
|
|
HOOK_END
|
||
|
|
|
||
|
|
|
||
|
|
IMPLEMENT_SHIM_END
|
||
|
|
|