|
|
/*++
Copyright (c) 2000 Microsoft Corporation
Module Name:
DisableW2KOwnerDrawButtonStates.cpp
Abstract:
Hooks all application-defined window procedures and filters out new owner-draw buttons states (introduced in Win2000).
Notes:
This shim can be reused for other shims that require WindowProc hooking. Copy all APIHook_* functions and simply replace the code in WindowProcHook and DialogProcHook.
History:
11/01/1999 markder Created 02/15/1999 markder Reworked WndProc hooking mechanism so that it generically hooks all WndProcs for the process. 11/29/2000 andyseti Converted into GeneralPurpose shim.
--*/
#include "precomp.h"
IMPLEMENT_SHIM_BEGIN(DisableW2KOwnerDrawButtonStates)#include "ShimHookMacro.h"
APIHOOK_ENUM_BEGIN APIHOOK_ENUM_ENTRY(RegisterClassA) APIHOOK_ENUM_ENTRY(RegisterClassW) APIHOOK_ENUM_ENTRY(RegisterClassExA) APIHOOK_ENUM_ENTRY(RegisterClassExW) APIHOOK_ENUM_ENTRY(CreateDialogParamA) APIHOOK_ENUM_ENTRY(CreateDialogParamW) APIHOOK_ENUM_ENTRY(CreateDialogIndirectParamA) APIHOOK_ENUM_ENTRY(CreateDialogIndirectParamW) APIHOOK_ENUM_ENTRY(CreateDialogIndirectParamAorW) APIHOOK_ENUM_ENTRY(SetWindowLongA) APIHOOK_ENUM_ENTRY(SetWindowLongW)APIHOOK_ENUM_END
/*++
Change WM_DRAWITEM behaviour
--*/
LRESULT CALLBACK WindowProcHook( WNDPROC pfnOld, // address of old WindowProc
HWND hwnd, // handle to window
UINT uMsg, // message identifier
WPARAM wParam, // first message parameter
LPARAM lParam // second message parameter
){ // Check for message we're interested in
if (uMsg == WM_DRAWITEM) { if (((LPDRAWITEMSTRUCT) lParam)->itemState & ~(ODS_SELECTED | ODS_GRAYED | ODS_DISABLED | ODS_CHECKED | ODS_FOCUS | ODS_DEFAULT | ODS_COMBOBOXEDIT | ODS_HOTLIGHT | ODS_INACTIVE)) { LOGN(eDbgLevelError, "Removed Win2K-specific Owner-draw button flags.");
// Remove all Win9x-incompatible owner draw button states.
((LPDRAWITEMSTRUCT) lParam)->itemState &= (ODS_SELECTED | ODS_GRAYED | ODS_DISABLED | ODS_CHECKED | ODS_FOCUS | ODS_DEFAULT | ODS_COMBOBOXEDIT | ODS_HOTLIGHT | ODS_INACTIVE); } }
return (*pfnOld)(hwnd, uMsg, wParam, lParam); }
INT_PTR CALLBACK DialogProcHook( DLGPROC pfnOld, // address of old DialogProc
HWND hwndDlg, // handle to dialog box
UINT uMsg, // message
WPARAM wParam, // first message parameter
LPARAM lParam // second message parameter
){ // Check for message we're interested in
if (uMsg == WM_DRAWITEM) { if (((LPDRAWITEMSTRUCT) lParam)->itemState & ~(ODS_SELECTED | ODS_GRAYED | ODS_DISABLED | ODS_CHECKED | ODS_FOCUS | ODS_DEFAULT | ODS_COMBOBOXEDIT | ODS_HOTLIGHT | ODS_INACTIVE)) { LOGN(eDbgLevelError, "Removed Win2K-specific Owner-draw button flags."); // Remove all Win9x-incompatible owner draw button states.
((LPDRAWITEMSTRUCT) lParam)->itemState &= (ODS_SELECTED | ODS_GRAYED | ODS_DISABLED | ODS_CHECKED | ODS_FOCUS | ODS_DEFAULT | ODS_COMBOBOXEDIT | ODS_HOTLIGHT | ODS_INACTIVE); } }
return (*pfnOld)(hwndDlg, uMsg, wParam, lParam); }
/*++
Hook all possible calls that can initialize or change a window's WindowProc (or DialogProc)
--*/
ATOMAPIHOOK(RegisterClassA)( CONST WNDCLASSA *lpWndClass // class data
){ WNDCLASSA wcNewWndClass = *lpWndClass;
wcNewWndClass.lpfnWndProc = (WNDPROC) HookCallback(lpWndClass->lpfnWndProc, WindowProcHook); if( NULL == wcNewWndClass.lpfnWndProc ) { DPFN(eDbgLevelInfo, "Failed to hook window proc via RegisterClassA."); return ORIGINAL_API(RegisterClassA)(lpWndClass); }
DPFN(eDbgLevelInfo, "Hooked window proc via RegisterClassA.");
return ORIGINAL_API(RegisterClassA)(&wcNewWndClass);}
ATOMAPIHOOK(RegisterClassW)( CONST WNDCLASSW *lpWndClass // class data
){ WNDCLASSW wcNewWndClass = *lpWndClass;
wcNewWndClass.lpfnWndProc = (WNDPROC) HookCallback(lpWndClass->lpfnWndProc, WindowProcHook); if( NULL == wcNewWndClass.lpfnWndProc ) { DPFN(eDbgLevelInfo, "Failed to hook window proc via RegisterClassW."); return ORIGINAL_API(RegisterClassW)(lpWndClass); }
DPFN( eDbgLevelInfo, "Hooked window proc via RegisterClassW.");
return ORIGINAL_API(RegisterClassW)(&wcNewWndClass);}
ATOMAPIHOOK(RegisterClassExA)( CONST WNDCLASSEXA *lpwcx // class data
){ WNDCLASSEXA wcNewWndClass = *lpwcx;
wcNewWndClass.lpfnWndProc = (WNDPROC) HookCallback(lpwcx->lpfnWndProc, WindowProcHook); if( NULL == wcNewWndClass.lpfnWndProc ) { DPFN(eDbgLevelInfo, "Failed to hook window proc via RegisterClassExA."); return ORIGINAL_API(RegisterClassExA)(lpwcx); }
DPFN( eDbgLevelInfo, "Hooked window proc via RegisterClassExA.");
return ORIGINAL_API(RegisterClassExA)(&wcNewWndClass);}
ATOMAPIHOOK(RegisterClassExW)( CONST WNDCLASSEXW *lpwcx // class data
){ WNDCLASSEXW wcNewWndClass = *lpwcx;
wcNewWndClass.lpfnWndProc = (WNDPROC) HookCallback(lpwcx->lpfnWndProc, WindowProcHook); if( NULL == wcNewWndClass.lpfnWndProc ) { DPFN(eDbgLevelInfo, "Failed to hook window proc via RegisterClassExW."); return ORIGINAL_API(RegisterClassExW)(lpwcx); }
DPFN( eDbgLevelInfo, "Hooked window proc via RegisterClassExW.");
return ORIGINAL_API(RegisterClassExW)(&wcNewWndClass);}
HWNDAPIHOOK(CreateDialogParamA)( HINSTANCE hInstance, // handle to module
LPCSTR lpTemplateName, // dialog box template
HWND hWndParent, // handle to owner window
DLGPROC lpDialogFunc, // dialog box procedure
LPARAM dwInitParam // initialization value
){ DLGPROC lpNewDialogFunc = (DLGPROC) HookCallback(lpDialogFunc, DialogProcHook); if( NULL == lpNewDialogFunc ) { DPFN( eDbgLevelInfo, "Failed to hook window proc via CreateDialogParamA.");
return ORIGINAL_API(CreateDialogParamA)( hInstance, lpTemplateName, hWndParent, lpDialogFunc, dwInitParam ); }
DPFN( eDbgLevelInfo, "Hooked window proc via CreateDialogParamA.");
return ORIGINAL_API(CreateDialogParamA)( hInstance, lpTemplateName, hWndParent, lpNewDialogFunc, dwInitParam );}
HWNDAPIHOOK(CreateDialogParamW)( HINSTANCE hInstance, // handle to module
LPCWSTR lpTemplateName, // dialog box template
HWND hWndParent, // handle to owner window
DLGPROC lpDialogFunc, // dialog box procedure
LPARAM dwInitParam // initialization value
){ DLGPROC lpNewDialogFunc = (DLGPROC) HookCallback(lpDialogFunc, DialogProcHook); if( NULL == lpNewDialogFunc ) { DPFN( eDbgLevelInfo, "Failed to hook window proc via CreateDialogParamW.");
return ORIGINAL_API(CreateDialogParamW)( hInstance, lpTemplateName, hWndParent, lpDialogFunc, dwInitParam ); }
DPFN( eDbgLevelInfo, "Hooked window proc via CreateDialogParamW.");
return ORIGINAL_API(CreateDialogParamW)( hInstance, lpTemplateName, hWndParent, lpNewDialogFunc, dwInitParam );}
HWNDAPIHOOK(CreateDialogIndirectParamA)( HINSTANCE hInstance, // handle to module
LPCDLGTEMPLATE lpTemplate, // dialog box template
HWND hWndParent, // handle to owner window
DLGPROC lpDialogFunc, // dialog box procedure
LPARAM lParamInit // initialization value
){ DLGPROC lpNewDialogFunc = (DLGPROC) HookCallback(lpDialogFunc, DialogProcHook); if( NULL == lpNewDialogFunc ) { DPFN( eDbgLevelInfo, "Failed to hook window proc via CreateDialogIndirectParamA.");
return ORIGINAL_API(CreateDialogIndirectParamA)( hInstance, lpTemplate, hWndParent, lpDialogFunc, lParamInit ); }
DPFN( eDbgLevelInfo, "Hooked window proc via CreateDialogIndirectParamA.");
return ORIGINAL_API(CreateDialogIndirectParamA)( hInstance, lpTemplate, hWndParent, lpNewDialogFunc, lParamInit );}
HWNDAPIHOOK(CreateDialogIndirectParamW)( HINSTANCE hInstance, // handle to module
LPCDLGTEMPLATE lpTemplate, // dialog box template
HWND hWndParent, // handle to owner window
DLGPROC lpDialogFunc, // dialog box procedure
LPARAM lParamInit // initialization value
){ DLGPROC lpNewDialogFunc = (DLGPROC) HookCallback(lpDialogFunc, DialogProcHook); if( NULL == lpNewDialogFunc ) { DPFN( eDbgLevelInfo, "Failed to hook window proc via CreateDialogIndirectParamW.");
return ORIGINAL_API(CreateDialogIndirectParamW)( hInstance, lpTemplate, hWndParent, lpDialogFunc, lParamInit ); }
DPFN( eDbgLevelInfo, "Hooked window proc via CreateDialogIndirectParamW.");
return ORIGINAL_API(CreateDialogIndirectParamW)( hInstance, lpTemplate, hWndParent, lpNewDialogFunc, lParamInit );}
HWNDAPIHOOK(CreateDialogIndirectParamAorW)( HINSTANCE hInstance, // handle to module
LPCDLGTEMPLATE lpTemplate, // dialog box template
HWND hWndParent, // handle to owner window
DLGPROC lpDialogFunc, // dialog box procedure
LPARAM lParamInit // initialization value
){ DLGPROC lpNewDialogFunc = (DLGPROC) HookCallback(lpDialogFunc, DialogProcHook); if( NULL == lpNewDialogFunc ) { DPFN( eDbgLevelInfo, "Failed to hook window proc via CreateDialogIndirectParamAorW.");
return ORIGINAL_API(CreateDialogIndirectParamAorW)( hInstance, lpTemplate, hWndParent, lpDialogFunc, lParamInit ); }
DPFN( eDbgLevelInfo, "Hooked window proc via CreateDialogIndirectParamAorW.");
return ORIGINAL_API(CreateDialogIndirectParamAorW)( hInstance, lpTemplate, hWndParent, lpNewDialogFunc, lParamInit );}
LONG APIHOOK(SetWindowLongA)( HWND hWnd, int nIndex, LONG dwNewLong ){ if (nIndex == GWL_WNDPROC) {
LOGN( eDbgLevelError, "Hooked window proc via SetWindowLongA. Pre-hook: 0x%X. ", dwNewLong);
LONG tmp = (LONG) HookCallback((PVOID)dwNewLong, WindowProcHook); if( NULL != tmp) { dwNewLong = tmp; }
DPFN( eDbgLevelInfo, "Post-hook: 0x%X.", dwNewLong);
} else if (nIndex == DWL_DLGPROC) {
LOGN( eDbgLevelError, "Hooked dialog proc via SetWindowLongA. Pre-hook: 0x%X. ", dwNewLong);
LONG tmp = (LONG) HookCallback((PVOID)dwNewLong, DialogProcHook); if( NULL != tmp) { dwNewLong = tmp; }
DPFN( eDbgLevelInfo, "Post-hook: 0x%X.", dwNewLong); }
return ORIGINAL_API(SetWindowLongA)( hWnd, nIndex, dwNewLong );}
LONG APIHOOK(SetWindowLongW)( HWND hWnd, int nIndex, LONG dwNewLong ){ if (nIndex == GWL_WNDPROC) { LOGN( eDbgLevelError, "Hooked window proc via SetWindowLongW. Pre-hook: 0x%X. ", dwNewLong);
LONG tmp = (LONG) HookCallback((PVOID)dwNewLong, WindowProcHook); if( NULL != tmp) { dwNewLong = tmp; }
DPFN( eDbgLevelInfo, "Post-hook: 0x%X.", dwNewLong); } else if (nIndex == DWL_DLGPROC) { LOGN( eDbgLevelError, "Hooked dialog proc via SetWindowLongW. Pre-hook: 0x%X. ", dwNewLong);
LONG tmp = (LONG) HookCallback((PVOID)dwNewLong, DialogProcHook); if( NULL != tmp) { dwNewLong = tmp; }
DPFN( eDbgLevelInfo, "Post-hook: 0x%X.", dwNewLong); }
return ORIGINAL_API(SetWindowLongW)( hWnd, nIndex, dwNewLong );}
/*++
Register hooked functions
--*/
HOOK_BEGIN APIHOOK_ENTRY(USER32.DLL, RegisterClassA) APIHOOK_ENTRY(USER32.DLL, RegisterClassW); APIHOOK_ENTRY(USER32.DLL, RegisterClassExA); APIHOOK_ENTRY(USER32.DLL, RegisterClassExW); APIHOOK_ENTRY(USER32.DLL, CreateDialogParamA); APIHOOK_ENTRY(USER32.DLL, CreateDialogParamW); APIHOOK_ENTRY(USER32.DLL, CreateDialogIndirectParamA); APIHOOK_ENTRY(USER32.DLL, CreateDialogIndirectParamW); APIHOOK_ENTRY(USER32.DLL, CreateDialogIndirectParamAorW); APIHOOK_ENTRY(USER32.DLL, SetWindowLongA); APIHOOK_ENTRY(USER32.DLL, SetWindowLongW);
HOOK_END
IMPLEMENT_SHIM_END
|
