Visual C++项目开发案例精粹-光盘源码（11-20）

/****************************************************************** $Archive: /MfcExt/Source/CoolControlsManager.cpp $ $Workfile: CoolControlsManager.cpp $ $Author: Bogdan Ledwig $ $Date: 99-04-26 22:12 $ $Revision: 13 $ *******************************************************************/ #include "StdAfx.h" #include "CoolControlsManager.h" // If you don't want to see extra TRACE diagnostics, // modify the line below to: #define CCM_TRACE #define CCM_TRACE TRACE #define CCM_TIMER_VAL 100 // 100 ms timer period seems to be good enough... #define MAX_CLASSNAME 64 // Length of buffer for retrieving the class name //////////////////////////////////////////////////////////////////////// // CCMControl static members initialization HWND CCoolControlsManager::CCMControl::m_hWndOld = NULL; CMapPtrToPtr CCoolControlsManager::m_ctrlMap = 10; CMapPtrToPtr CCoolControlsManager::m_dlgMap = 10; BOOL CCoolControlsManager::m_bEnabled = TRUE; // Changed 02.03.1999 Mike Walter CMapWordToPtr CCoolControlsManager::m_threadMap = 10; /////////////////////////////////////////////////////////////////////// // Here is the one and only CCoolControlsManager object static CCoolControlsManager g_ctrlManager; CCoolControlsManager& GetCtrlManager() { return g_ctrlManager; } //////////////////////////////////////////////////////////////////////// // WH_CALLWNDPROC hook procedure LRESULT CALLBACK CCM_CallWndProc( int nCode, WPARAM wParam, LPARAM lParam ) { HOOKPROC hHookProc; if ( g_ctrlManager.m_threadMap.Lookup( (WORD)GetCurrentThreadId(), (void*&)hHookProc ) == FALSE ) { // TRACE( "CCoolControlsManager: No hook for this thread installed!\n" ); return 0; } if ( nCode == HC_ACTION ) { CWPSTRUCT* pwp = (CWPSTRUCT*)lParam; if ( g_ctrlManager.IsEnabled() ) { if ( g_ctrlManager.m_bDialogOnly == TRUE ) { if ( pwp->message == WM_INITDIALOG ) g_ctrlManager.Install( pwp->hwnd ); } else if ( pwp->message == WM_CREATE && g_ctrlManager.IsEnabled() ) { TCHAR szBuf[MAX_CLASSNAME]; if ( GetWindowLong( pwp->hwnd, GWL_STYLE ) & WS_CHILD ) { GetClassName( pwp->hwnd, szBuf, MAX_CLASSNAME ); if ( lstrcmp( szBuf, _T( "ScrollBar" ) ) ) // Don't add scrollbars g_ctrlManager.AddControl( pwp->hwnd ); } } } } // Changed 02.03.1999 Mike Walter return CallNextHookEx( (HHOOK)hHookProc, nCode, wParam, lParam ); } // Install a hook for the current thread only void CCoolControlsManager::InstallHook( DWORD dwThreadID, BOOL bDialogOnly ) { // ASSERT( m_hkWndProc == NULL ); m_bDialogOnly = bDialogOnly; // Changes 02.03.1999 Mike Walter HOOKPROC hNewHook; if ( m_threadMap.Lookup( (WORD)( dwThreadID == -1 ? GetCurrentThreadId() : dwThreadID ), (void*&)hNewHook ) == FALSE ) { hNewHook = (HOOKPROC)SetWindowsHookEx( WH_CALLWNDPROC, (HOOKPROC)CCM_CallWndProc, NULL, ( dwThreadID == -1 ? GetCurrentThreadId() : dwThreadID ) ); m_threadMap.SetAt( (WORD)( dwThreadID == -1 ? GetCurrentThreadId() : dwThreadID ), hNewHook ); CCM_TRACE( "CCoolControlsManager: WH_CALLWNDPROC hook installed for thread: %d\n", ( dwThreadID == -1 ? GetCurrentThreadId() : dwThreadID ) ); } else CCM_TRACE( "CCoolControlsManager: WH_CALLWNDPROC hook already installed for thread: %d!\n", ( dwThreadID == -1 ? GetCurrentThreadId() : dwThreadID ) ); } // Install a global hook for all windows in the system. // This function may be called only when is put in a DLL. void CCoolControlsManager::InstallGlobalHook( HINSTANCE hInstance, BOOL bDialogOnly ) { ASSERT( hInstance ); // hInstance must not be NULL! ASSERT( m_hkWndProc == NULL ); m_bDialogOnly = bDialogOnly; HOOKPROC hkProc = (HOOKPROC)GetProcAddress( hInstance, "CCM_CallWndProc" ); m_hkWndProc = (HOOKPROC)SetWindowsHookEx( WH_CALLWNDPROC, (HOOKPROC)hkProc, hInstance, 0 ); CCM_TRACE( _T( "CCoolControlsManager: WH_CALLWNDPROC global hook installed\n" ) ); } void CCoolControlsManager::UninstallHook( DWORD dwThreadID ) { // ASSERT( m_hkWndProc != NULL ); // Changes 02.03.1999 Mike Walter HOOKPROC hHookProc; if ( dwThreadID == -1 ) { if ( g_ctrlManager.m_threadMap.Lookup( (WORD)GetCurrentThreadId(), (void*&)hHookProc ) == FALSE ) { CCM_TRACE( "CCoolControlsManager: No hook installed for thread: %d!\n", GetCurrentThreadId() ); return; } UnhookWindowsHookEx( (HHOOK)hHookProc ); m_threadMap.RemoveKey( (WORD)GetCurrentThreadId() ); CCM_TRACE( "CCoolControlsManager: Hook uninstalled for thread: %d\n", GetCurrentThreadId() ); CCM_TRACE( "CCoolControlsManager: Thread map has %d items\n",g_ctrlManager.m_threadMap.GetCount() ); } else { if ( g_ctrlManager.m_threadMap.Lookup( (WORD)dwThreadID, (void*&)hHookProc) == FALSE ) { CCM_TRACE( "CCoolControlsManager: No hook installed for thread: %d!\n", dwThreadID ); return; } UnhookWindowsHookEx( (HHOOK)hHookProc ); m_threadMap.RemoveKey( (WORD)dwThreadID ); CCM_TRACE( "CCoolControlsManager: Hook uninstalled for thread: %d\n", dwThreadID ); CCM_TRACE( "CCoolControlsManager: Thread map has %d items\n", g_ctrlManager.m_threadMap.GetCount() ); } if ( m_uTimerID && g_ctrlManager.m_threadMap.IsEmpty() == TRUE ) KillTimer( NULL, m_uTimerID ); } ////////////////////////////////////////////////////////////////////// // Construction/Destruction ////////////////////////////////////////////////////////////////////// CCoolControlsManager::CCoolControlsManager() { m_hkWndProc = NULL; m_uTimerID = 0; CCM_TRACE( _T( "CCoolControlsManager::CCoolControlsManager()\n" ) ); } CCoolControlsManager::~CCoolControlsManager() { // Changed 02.03.1999 Mike Walter POSITION pos = m_threadMap.GetStartPosition(); while ( pos ) { HOOKPROC hHook; DWORD dwThreadID = 0; m_threadMap.GetNextAssoc( pos, (WORD&)dwThreadID, (void*&)hHook ); UninstallHook( dwThreadID ); } // If we have any elements in the map (normally impossible), unsubclass they and remove pos = m_ctrlMap.GetStartPosition(); while ( pos ) { HWND hWnd; CCMControl* pCtl; m_ctrlMap.GetNextAssoc( pos, (void*&)hWnd, (void*&)pCtl ); pCtl->Unsubclass(); m_ctrlMap.RemoveKey( hWnd ); delete pCtl; } // Now do the same things for dialog map pos = m_dlgMap.GetStartPosition(); while ( pos ) { HWND hWnd; CCMDialog* pCtl; m_dlgMap.GetNextAssoc( pos, (void*&)hWnd, (void*&)pCtl ); pCtl->Unsubclass(); m_dlgMap.RemoveKey( hWnd ); delete pCtl; } CCM_TRACE( "CCoolControlsManager::~CCoolControlsManager()\n" ); } void CCoolControlsManager::Install( HWND hWnd ) { CCMControl* pCtl; if ( m_dlgMap.Lookup( hWnd, (void*&)pCtl ) ) // Already in the dialog map return; // Iterate through all child windows HWND hCtrl = GetTopWindow( hWnd ); while ( hCtrl ) { if ( GetWindowLong( hCtrl, GWL_STYLE ) & WS_CHILD ) { TCHAR szBuf[MAX_CLASSNAME]; GetClassName( hCtrl, szBuf, MAX_CLASSNAME ); if ( lstrcmpi( szBuf, _T( "#32770" ) ) ) // Never add child dialogs! AddControl( hCtrl ); } hCtrl = GetNextWindow( hCtrl