|
|
#include "stdafx.h"
#include "handlers.h"
//-------------------------------------------------------------------------//
// Declare registered message vars **here**
//-------------------------------------------------------------------------//
#define __NO_APPHACKS__
//-------------------------------------------------------------------------//
// Message Handlers
//-------------------------------------------------------------------------//
// Rules for message handlers [scotthan]:
//
// (1) Use DECL_ macros to declare message handler prototype and
// table entries for message handlers below.
// (2) A message handler implementation should NOT:
// 1. call DefWindowProc or CallWindowProc directly,
// but rather use DoMsgDefault().
// 2. delete the incoming CThemeWnd* object,
// (3) A message handler SHOULD:
// 1. Honor the codepage value in the message block when
// handling messages that carry string args.
// If the codepage member is CP_WINUNICODE, the widechar
// string processing should be assumed; otherwise, multibyte
// string processing should be assumed.
// 2. If a message should not be forwarded for default processing,
// mark the message as handled using MsgHandled().
// (4) Handlers should be listed in the BEGIN/ENDMSG() block
// below in decreasing order of expected frequency.
//---------------------//
// WndProc overrides
//---------------------//
// msg handler decls:
DECL_MSGHANDLER( OnOwpPostCreate );DECL_MSGHANDLER( OnOwpPreStyleChange );DECL_MSGHANDLER( OnOwpPreWindowPosChanging );DECL_MSGHANDLER( OnOwpPreWindowPosChanged );DECL_MSGHANDLER( OnOwpPostWindowPosChanged );DECL_MSGHANDLER( OnOwpPostSettingChange );DECL_MSGHANDLER( OnOwpPreMeasureItem );DECL_MSGHANDLER( OnOwpPreDrawItem );DECL_MSGHANDLER( OnOwpPreMenuChar );DECL_MSGHANDLER( OnOwpPostThemeChanged );DECL_MSGHANDLER( OnOwpPreNcPaint );DECL_MSGHANDLER( OnOwpPostNcPaint );
// handler table:
BEGIN_HANDLER_TABLE(_rgOwpHandlers) // frequent messages
DECL_MSGENTRY( WM_NCPAINT, OnOwpPreNcPaint, OnOwpPostNcPaint ) DECL_MSGENTRY( WM_WINDOWPOSCHANGING, OnOwpPreWindowPosChanging, NULL ) DECL_MSGENTRY( WM_WINDOWPOSCHANGED, OnOwpPreWindowPosChanged, OnOwpPostWindowPosChanged ) DECL_MSGENTRY( WM_SETTINGCHANGE, NULL, OnOwpPostSettingChange ) DECL_MSGENTRY( WM_MEASUREITEM, OnOwpPreMeasureItem, NULL ) DECL_MSGENTRY( WM_DRAWITEM, OnOwpPreDrawItem, NULL ) DECL_MSGENTRY( WM_MDISETMENU, NULL, NULL )
// rare messages:
DECL_MSGENTRY( WM_MENUCHAR, OnOwpPreMenuChar, NULL ) DECL_MSGENTRY( WM_STYLECHANGING, OnOwpPreStyleChange, NULL ) DECL_MSGENTRY( WM_STYLECHANGED, OnOwpPreStyleChange, NULL ) DECL_MSGENTRY( WM_NCCREATE, NULL, NULL ) DECL_MSGENTRY( WM_CREATE, NULL, OnOwpPostCreate ) DECL_MSGENTRY( WM_NCDESTROY, NULL, NULL ) DECL_MSGENTRY( WM_THEMECHANGED, NULL, OnOwpPostThemeChanged ) // we handle in line in ThemePreWndProc()
DECL_MSGENTRY( WM_THEMECHANGED_TRIGGER, NULL, NULL ) // we handle in line in ThemePreWndProc()
END_HANDLER_TABLE()
// Note: values of high owp message must be in sync w/ table.
#define WNDPROC_MSG_LAST WM_THEMECHANGED_TRIGGER // 0x031B (alias for WM_UAHINIT)
//------------------------//
// DefDlgProc overrides
//------------------------//
// msg handler decls:
DECL_MSGHANDLER( OnDdpPostCtlColor );DECL_MSGHANDLER( OnDdpCtlColor );DECL_MSGHANDLER( OnDdpPrint );DECL_MSGHANDLER( OnDdpPostInitDialog );
// handler table:
BEGIN_HANDLER_TABLE(_rgDdpHandlers) // frequent messages:
DECL_MSGENTRY( WM_CTLCOLORDLG, NULL, OnDdpPostCtlColor ) DECL_MSGENTRY( WM_CTLCOLORSTATIC, NULL, OnDdpCtlColor) DECL_MSGENTRY( WM_CTLCOLORBTN, NULL, OnDdpCtlColor) DECL_MSGENTRY( WM_CTLCOLORMSGBOX, NULL, OnDdpPostCtlColor ) DECL_MSGENTRY( WM_PRINTCLIENT, NULL, OnDdpPrint ) // rare messages:
DECL_MSGENTRY( WM_INITDIALOG, NULL, OnDdpPostInitDialog )END_HANDLER_TABLE()
// Note: values of high ddp message must be in sync w/ table.
#define DEFDLGPROC_MSG_LAST WM_PRINTCLIENT // 0x0318
//--------------------------//
// DefWindowProc override
//--------------------------//
// msg handler decls:
DECL_MSGHANDLER( OnDwpNcPaint );DECL_MSGHANDLER( OnDwpNcHitTest );DECL_MSGHANDLER( OnDwpNcActivate );DECL_MSGHANDLER( OnDwpNcLButtonDown );DECL_MSGHANDLER( OnDwpNcThemeDrawCaption );DECL_MSGHANDLER( OnDwpNcThemeDrawFrame );DECL_MSGHANDLER( OnDwpNcMouseMove );DECL_MSGHANDLER( OnDwpNcMouseLeave );DECL_MSGHANDLER( OnDwpWindowPosChanged );DECL_MSGHANDLER( OnDwpSysCommand );DECL_MSGHANDLER( OnDwpSetText );DECL_MSGHANDLER( OnDwpSetIcon );DECL_MSGHANDLER( OnDwpStyleChanged );DECL_MSGHANDLER( OnDwpPrint );DECL_MSGHANDLER( OnDwpPrintClient );DECL_MSGHANDLER( OnDwpContextMenu );
// handler table:
BEGIN_HANDLER_TABLE(_rgDwpHandlers) // frequent messages:
DECL_MSGENTRY( WM_NCHITTEST, OnDwpNcHitTest, NULL ) DECL_MSGENTRY( WM_NCPAINT, OnDwpNcPaint, NULL ) DECL_MSGENTRY( WM_NCACTIVATE, OnDwpNcActivate, NULL ) DECL_MSGENTRY( WM_NCMOUSEMOVE, OnDwpNcMouseMove, NULL ) DECL_MSGENTRY( WM_NCMOUSELEAVE, OnDwpNcMouseLeave, NULL ) DECL_MSGENTRY( WM_WINDOWPOSCHANGED, OnDwpWindowPosChanged, NULL ) DECL_MSGENTRY( WM_SYSCOMMAND, OnDwpSysCommand, NULL ) DECL_MSGENTRY( WM_NCLBUTTONDOWN, OnDwpNcLButtonDown, NULL ) DECL_MSGENTRY( WM_NCUAHDRAWCAPTION, OnDwpNcThemeDrawCaption, NULL ) DECL_MSGENTRY( WM_NCUAHDRAWFRAME, OnDwpNcThemeDrawFrame, NULL ) DECL_MSGENTRY( WM_PRINT, OnDwpPrint, NULL ) DECL_MSGENTRY( WM_PRINTCLIENT, OnDwpPrintClient, NULL ) DECL_MSGENTRY( WM_CTLCOLORMSGBOX, OnDdpPostCtlColor, NULL) // Strange: Sent to DefWindowProc, but is a Dialog message
DECL_MSGENTRY( WM_CTLCOLORSTATIC, OnDdpCtlColor, NULL) DECL_MSGENTRY( WM_CTLCOLORBTN, OnDdpCtlColor, NULL) // rare messages:
DECL_MSGENTRY( WM_SETTEXT, OnDwpSetText, NULL ) DECL_MSGENTRY( WM_SETICON, OnDwpSetIcon, NULL ) DECL_MSGENTRY( WM_STYLECHANGED, OnDwpStyleChanged, NULL ) DECL_MSGENTRY( WM_CONTEXTMENU, OnDwpContextMenu, NULL ) DECL_MSGENTRY( WM_THEMECHANGED_TRIGGER, NULL, NULL ) DECL_MSGENTRY( WM_NCDESTROY, NULL, NULL )END_HANDLER_TABLE()
// Note: values of high dwp message must be in sync w/ handler table.
#define DEFWNDPROC_MSG_LAST WM_THEMECHANGED_TRIGGER // 0x031B
//---------------------------------------------------------------------------
BOOL _FindMsgHandler( UINT, MSGENTRY [], int, IN HOOKEDMSGHANDLER*, IN HOOKEDMSGHANDLER* );BOOL _SetMsgHandler( UINT, MSGENTRY [], int, IN HOOKEDMSGHANDLER, BOOL );
//---------------------------------------------------------------------------
// Special case hook handling
//---------------------------------------------------------------------------
//---------------------------------------------------------------------------
BOOL _IsExcludedSystemProcess( LPCWSTR pszProcess ){ static const WCHAR* _rgszSystemProcessList[] = { L"lsass", // Local Security Authority sub-system
L"services", // Service Control Manager
L"svchost", // Service Host
L"mstask", // Microsoft Task Scheduler
L"dfssvc", // Distributed File System Service
L"winmgmt", // Windows Management Instrumentation
L"spoolsv", // Print Spool Service
L"msdtc", // Microsoft Distributed Transaction Co-ordinator
L"regsvc", // Remote Registry Service
L"webclnt", // Web Client
L"mspmspsv", // WMDM PMSP Service (what is this?)
L"ntvdm" // NT virtual DOS machine
};
return AsciiScanStringList( pszProcess, _rgszSystemProcessList, ARRAYSIZE(_rgszSystemProcessList), TRUE );}
//---------------------------------------------------------------------------
BOOL _IsProcessOnInteractiveWindowStation() // check if we're on winsta0.
{ BOOL fRet = FALSE; HWINSTA hWinSta = GetProcessWindowStation(); if( hWinSta != NULL ) { DWORD cbLength = 0; WCHAR wszName[32];
if (GetUserObjectInformationW(hWinSta, UOI_NAME, wszName, sizeof(wszName), &cbLength)) { fRet = (0 == AsciiStrCmpI(wszName, L"winsta0")); } else { // Problem! It's either:
// - our buffer was too small (in which case it can't be winsta0)
// - or some other failure we cannot overcome anyhow
} } return(fRet);}
//---------------------------------------------------------------------------
BOOL _IsWin16App() // check if this is a 16-bit process
{ GUITHREADINFO gti; gti.cbSize = sizeof(gti); gti.flags = GUI_16BITTASK;
return GetGUIThreadInfo( GetCurrentThreadId(), >i ) && TESTFLAG(gti.flags, GUI_16BITTASK );}
//---------------------------------------------------------------------------
BOOL ApiHandlerInit( const LPCTSTR pszProcess, USERAPIHOOK* puahTheme, const USERAPIHOOK* puahReal ){ // exclude known non-UI system processes
if( _IsExcludedSystemProcess( pszProcess ) ) return FALSE;
// exclude any process not hosted on winsta0.
if( !_IsProcessOnInteractiveWindowStation() ) return FALSE;
if( _IsWin16App() ) return FALSE; // SHIMSHIM [scotthan]:
#ifdef _DEBUG
//---- temp patch against msvcmon ----
if( 0 == AsciiStrCmpI(pszProcess, L"msvcmon") ) { return FALSE; }
//---- temp patch against msdev ----
if( 0 == AsciiStrCmpI(pszProcess, L"msdev") ) { return FALSE; }
//---- Aid in debugging classic/themed differences: ---
if( 0 == AsciiStrCmpI( pszProcess, L"mditest1" ) ) { return FALSE; }#endif
#ifndef __NO_APPHACKS__
static const WCHAR* _rgszExcludeAppList[] = {#ifdef THEME_CALCSIZE
// Invoking SetWindowPos from CThemeWnd::SetFrameTheme on our Post-WM_CREATE handler
// causes emacs to divide by zero after receiving meaningless rects from two
// successive calls to AdjustWindowRectEx from his WM_WINDOWPOSCHANGING handler.
// I don't believe it is related to the fact that AdjustWindowRectEx has yet
// to be implemented for themed windows (raid# 140989), but rather that the wndproc
// is not ready for a WM_WINDOWPOSCHANGING message on the abrubtly on the
// heels of a WM_CREATE handler.
L"emacs",
L"neoplanet", // 247283: We rush in to theme neoplanet's dialogs, which we almost
L"np", // immediately revoke, but not before sizing the dialog to theme-compatible
// client rect. When we withdraw, we leave it clipped. No good way to deal
// with this for beta2.
// HTML Editor++ v.8: 286676:
// This guy recomputes his nonclient area, and then AVs dereferencing a
// WM_WINDOWPOSCHANGING message under themes.
L"coffee", #endif THEME_CALCSIZE
L"refcntr", // 205059: Corel Reference Center; lower 10% of window is clipped.
L"KeyFramerPro", // 336456: Regardless of whether themes are enabled, Boris KeyFramer Pro v.5
// does two SetWindowRgn() calls for every WM_PAINT, the first with a region,
// the next with NULL, Is the app trying to clip his painting?
// If so, this is not what SetWindowRgn was intended for, and explains why this
// app is so clunky at window resizing. Rather, SelectClipRgn is the
// correct API.
// When themes are enabled, we keep revoking and re-attatching with each
// SetWindowRgn call, so we get substantial flicker.
// The ISV should be notified of this bug.
// Applications that do custom non-client painting and hence look broken when
// themed. Our only recourse at the moment it to exclude them from non-client
// themeing so that we don't stomp whatever they are trying to do.
L"RealJBox", // 273370: Real JukeBox
L"RealPlay", // 285368: Real AudioPlayer
L"TeamMgr", // 286654: Microsoft Team Manager97
L"TrpMaker", // 307107: Rand McNally TripMaker 2000
L"StrFindr", // 307535: Rand McNally StreetFinder 2000
L"Exceed", // 276244: Hummingbird Exceed 6.2/7.0
L"VP30", // 328676: Intel Video Phone
// 313407: Groove, build 760
// Calls DefWindowProc for NCPAINT, then paints his own caption over it.
// Note: this just might work correctly if we had a DrawFrameControl hook.
L"groove", // filever 1.1.0.760, 1/22/2001 tested.
// 303756: Exclude all Lotus SmartSuite apps to provide consistency among their
// apps. All of them draw into the caption bar.
L"WordPro", // 285065: Lotus WordPro, a particularly poorly implemented app.
L"SmartCtr", // It's WordPerfect compat menu is the elephant man of modern software.
L"123w", L"Approach", L"FastSite", L"F32Main", L"Org5",
// 358337: Best Technology - GCC Developer Lite. Custom caption bar fights with Luna.
L"GCCDevL", // install point: http://www.besttechnology.co.jp/download/GDL1_0_3_6.EXE
// 360422: J Zenrin The Real Digital Map Z3(T1):Max/Min/Close buttons are overlapped on classic buttons in title bar.
L"emZmain",
// 364337: Encarta World English Dictionary: Luna system buttons are overlaid on top of app's custom ones when mousing over
L"ewed.exe",
// 343171: Reaktor Realtime Instrument: pressing the close button while themed causes this app to
// spin in a tight loop running at realtime priority, effectively hanging the machine.
// The message loop for this app is extremely timing sensitive, the additional overhead
// introduced by theming alters the timing enough to break this app.
L"Reaktor", };
if( AsciiScanStringList( pszProcess, _rgszExcludeAppList, ARRAYSIZE(_rgszExcludeAppList), TRUE ) ) { return FALSE; }
#ifdef THEME_CALCSIZE
// Winstone 99 needs modified NC_CALCSIZE behavior for Netscape or it will hang.
if ( 0 == AsciiStrCmpI( pszProcess, L"Netscape" )) { if (FindWindowEx(NULL, NULL, L"ZDBench32Frame", NULL) != NULL) { _SetMsgHandler( WM_NCCALCSIZE, _rgDwpHandlers, ARRAYSIZE(_rgDwpHandlers), OnDwpNcCalcSize2, FALSE ); return TRUE; } }#endif THEME_CALCSIZE
//-------------------------
// This AppHack was once fixed, but got broke again with
// addition of logic for partial-screen maximized windows.
//
// Something in our answer to NCCALCSIZE causes quick time player
// to continously flood its 'control' frame window's winproc with
// WM_PAINTS by repeatedly calling InvalidateRgn + UpdateWindow. My
// suspicion is that he looks at what DefWindowProc returns from
// NCCALCSIZE to determine the area he needs to manage, and when
// this doesn't hash with other SYSMET values and/or AdjustWindowRect,
// he redundantly invalidates himself,
//
// This only repros if qtp is launched w/ .mov file, works fine if
// launched without a file and then a file is loaded.
#ifdef THEME_CALCSIZE
if( 0 == AsciiStrCmpI( pszProcess, L"QuickTimePlayer" )) { _SetMsgHandler( WM_NCCALCSIZE, _rgDwpHandlers, ARRAYSIZE(_rgDwpHandlers), OnDwpNcCalcSize2, FALSE ); return TRUE; }
// SEANHI DID NOT RECEIVE THE S/W FROM APPLIB AND SO WAS UNABLE TO VERIFY THIS
// NO LONGER REPROS W/ ELIMINATION OF THEMED SYSMETS
//-------------------------
// Paradox 9 appHack for nonclient button sizes:
//
// Paradox table schema view uses DrawFrameControl to render both
// a classic toolframe (small) caption and buttons, but uses the
// themed values of SM_CYSIZE instead of SM_CYSMSIZE to size the buttons.
// This apphack redirects requests in this process for SM_CX/YSIZE to SM_CX/YSMSIZE.
if( 0 == AsciiStrCmpI( pszProcess, L"pdxwin32" ) ) { _SetGsmHandler( SM_CXSIZE, OnGsmCxSmBtnSize ); _SetGsmHandler( SM_CYSIZE, OnGsmCySmBtnSize ); return TRUE; }#endif THEME_CALCSIZE
//-------------------------
#else
# pragma message("App hacks disabled")
#endif __NO_APPHACKS__
return TRUE;}
//---------------------------------------------------------------------------
// Handler table utility functions
//---------------------------------------------------------------------------
//---------------------------------------------------------------------------
void HandlerTableInit() {}
//---------------------------------------------------------------------------
BOOL _InitMsgMask( LPBYTE prgMsgMask, DWORD dwMaskBytes, MSGENTRY* prgEntries, int cEntries, IN OUT BOOL& fInit ){ if( !fInit ) { for( int i = 0; i < cEntries; i++ ) { if( -1 == prgEntries[i].nMsg ) { ASSERT(prgEntries[i].pnRegMsg); // Initialize registered message entry
prgEntries[i].nMsg = *prgEntries[i].pnRegMsg;
Log(LOG_TMHANDLE, L"InitMsgMsg corrected registered msg: 0x%x", prgEntries[i].nMsg); }
//---- ensure we set up limit on table correctly ----
ASSERT((prgEntries[i].nMsg)/8 < dwMaskBytes); SET_MSGMASK( prgMsgMask, prgEntries[i].nMsg ); } fInit = TRUE; }
return fInit;}
//---------------------------------------------------------------------------
// Scan of MSG table as linear array:
inline int _FindMsgHandler( UINT nMsg, MSGENTRY rgEntries[], int cEntries, OUT OPTIONAL HOOKEDMSGHANDLER* ppfnHandler, OUT OPTIONAL HOOKEDMSGHANDLER* ppfnHandler2 ){ ASSERT( nMsg ); ASSERT( nMsg != (UINT)-1 );
if( ppfnHandler ) *ppfnHandler = NULL; if( ppfnHandler2 ) *ppfnHandler2 = NULL;
for( int i = 0; i < cEntries; i++ ) { if( rgEntries[i].nMsg == nMsg ) { // If no handler requested, return success
if( NULL == ppfnHandler && NULL == ppfnHandler2 ) return i;
// Assign outbound handler values
if( ppfnHandler ) *ppfnHandler = rgEntries[i].pfnHandler; if( ppfnHandler2 ) *ppfnHandler2 = rgEntries[i].pfnHandler2;
// return TRUE iif caller got what he asked for.
return ((ppfnHandler && *ppfnHandler) || (ppfnHandler2 && *ppfnHandler2)) ? i : -1; } } return -1;}
//---------------------------------------------------------------------------
// Modify existing handler
inline BOOL _SetMsgHandler( UINT nMsg, MSGENTRY rgEntries[], int cEntries, IN HOOKEDMSGHANDLER pfnHandler, BOOL fHandler2 ){ int i = _FindMsgHandler( nMsg, rgEntries, cEntries, NULL, NULL ); if( i >= 0 ) { if( fHandler2 ) rgEntries[i].pfnHandler2 = pfnHandler; else rgEntries[i].pfnHandler = pfnHandler; return TRUE; } return FALSE;}
#define CBMSGMASK(msgHigh) (((msgHigh)+1)/8 + ((((msgHigh)+1) % 8) ? 1: 0))
//---------------------------------------------------------------------------
DWORD GetOwpMsgMask( LPBYTE* prgMsgMask ){ static BOOL _fOwpMask = FALSE; // initialized?
static BYTE _rgOwpMask[CBMSGMASK(WNDPROC_MSG_LAST)] = {0};
if( _InitMsgMask( _rgOwpMask, ARRAYSIZE(_rgOwpMask), _rgOwpHandlers, ARRAYSIZE(_rgOwpHandlers), _fOwpMask ) ) { *prgMsgMask = _rgOwpMask; return ARRAYSIZE(_rgOwpMask); } return 0;}
//---------------------------------------------------------------------------
DWORD GetDdpMsgMask( LPBYTE* prgMsgMask ){ static BOOL _fDdpMask = FALSE; // initialized?
static BYTE _rgDdpMask[CBMSGMASK(DEFDLGPROC_MSG_LAST)] = {0};
if( _InitMsgMask( _rgDdpMask, ARRAYSIZE(_rgDdpMask), _rgDdpHandlers, ARRAYSIZE(_rgDdpHandlers), _fDdpMask ) ) { *prgMsgMask = _rgDdpMask; return ARRAYSIZE(_rgDdpMask); } return 0;}
//---------------------------------------------------------------------------
DWORD GetDwpMsgMask( LPBYTE* prgMsgMask ){ static BOOL _fDwpMask = FALSE; // initialized?
static BYTE _rgDwpMask[CBMSGMASK(DEFWNDPROC_MSG_LAST)] = {0};
if( _InitMsgMask( _rgDwpMask, ARRAYSIZE(_rgDwpMask), _rgDwpHandlers, ARRAYSIZE(_rgDwpHandlers), _fDwpMask ) ) { *prgMsgMask = _rgDwpMask; return ARRAYSIZE(_rgDwpMask); } return 0;}
//---------------------------------------------------------------------------
BOOL FindOwpHandler( UINT nMsg, HOOKEDMSGHANDLER* ppfnPre, HOOKEDMSGHANDLER* ppfnPost ){ return _FindMsgHandler( nMsg, _rgOwpHandlers, ARRAYSIZE(_rgOwpHandlers), ppfnPre, ppfnPost ) >= 0;}
//---------------------------------------------------------------------------
BOOL FindDdpHandler( UINT nMsg, HOOKEDMSGHANDLER* ppfnPre, HOOKEDMSGHANDLER* ppfnPost ){ return _FindMsgHandler( nMsg, _rgDdpHandlers, ARRAYSIZE(_rgDdpHandlers), ppfnPre, ppfnPost ) >= 0;}
//---------------------------------------------------------------------------
BOOL FindDwpHandler( UINT nMsg, HOOKEDMSGHANDLER* ppfnPre ){ HOOKEDMSGHANDLER pfnPost; return _FindMsgHandler( nMsg, _rgDwpHandlers, ARRAYSIZE(_rgDwpHandlers), ppfnPre, &pfnPost ) >= 0;}
//---------------------------------------------------------------------------
// Performs default message processing.
LRESULT WINAPI DoMsgDefault( const THEME_MSG *ptm ){ ASSERT( ptm ); if( ptm->pfnDefProc ) { MsgHandled( ptm ); if( MSGTYPE_DEFWNDPROC == ptm->type ) return ptm->pfnDefProc( ptm->hwnd, ptm->uMsg, ptm->wParam, ptm->lParam ); else { ASSERT( NULL == ptm->pfnDefProc ); // bad initialization (_InitThemeMsg)
} } return 0L;}
|
