让MFC DLL 中的窗口响应PreTranslateMessage

PreTranslateMessage是CWnd的虚函数，在这条函数里处理一些按键消息非常方便。但最近参与一个项目，这个项目由主程序和多个插件DLL组成，其中的一个插件DLL是带有界面的，并且以主程序的窗口为父窗口，在这个插件DLL窗口中怎么也响应不了PreTranslateMessage函数。

看了看MFC的PreTranslateMessage实现终于找到了解决方法。

当按下键盘时，首先主程序的CWinApp对象的PreTranslateMessage会被调用。在这条函数中最重要的函数是WalkPreTranslateTree,其实现如下

BOOL PASCAL CWnd::WalkPreTranslateTree(HWND hWndStop, MSG* pMsg){    ASSERT(hWndStop == NULL || ::IsWindow(hWndStop));    ASSERT(pMsg != NULL);    // walk from the target window up to the hWndStop window checking    //  if any window wants to translate this message    for (HWND hWnd = pMsg->hwnd; hWnd != NULL; hWnd = ::GetParent(hWnd))    {        CWnd* pWnd = CWnd::FromHandlePermanent(hWnd);        if (pWnd != NULL)        {            // target window is a C++ window            if (pWnd->PreTranslateMessage(pMsg))                return TRUE; // trapped by target window (eg: accelerators)        }        // got to hWndStop window without interest        if (hWnd == hWndStop)            break;    }    return FALSE;       // no special processing}

可以看到MFC会从当前窗口一直向最顶层窗口查找并调用CWnd的PreTranslateMessage，直到PreTranslateMessage返回True，或找到最顶层窗口。跟踪发现当在MFC DLL的窗口上按下键盘时，CWnd::FromHandlePermanent 的返回总是NULL，继续跟踪发现问题出在FromHandlePermanent这条函数里。MFC实现如下：

CWnd* PASCAL CWnd::FromHandlePermanent(HWND hWnd){    CHandleMap* pMap = afxMapHWND();    CWnd* pWnd = NULL;    if (pMap != NULL)    {        // only look in the permanent map - does no allocations        pWnd = (CWnd*)pMap->LookupPermanent(hWnd);        ASSERT(pWnd == NULL || pWnd->m_hWnd == hWnd);    }    return pWnd;}CHandleMap* PASCAL afxMapHWND(BOOL bCreate){    AFX_MODULE_THREAD_STATE* pState = AfxGetModuleThreadState();    if (pState->m_pmapHWND == NULL && bCreate)    {        BOOL bEnable = AfxEnableMemoryTracking(FALSE);#ifndef _AFX_PORTABLE        _PNH pnhOldHandler = AfxSetNewHandler(&AfxCriticalNewHandler);#endif        pState->m_pmapHWND = new CHandleMap(RUNTIME_CLASS(CWnd),            ConstructDestruct<CWnd>::Construct, ConstructDestruct<CWnd>::Destruct,             offsetof(CWnd, m_hWnd));#ifndef _AFX_PORTABLE        AfxSetNewHandler(pnhOldHandler);#endif        AfxEnableMemoryTracking(bEnable);    }    return pState->m_pmapHWND;}

原来CWnd::FromHandlePermanent 中调用了afxMapHWND,而该函数返回的是当前模块的模块线程状态中HandleMap,该Map记录了此模块中的HWND对应的CWnd对象的指针。而在调用CWnd的Create函数时会调用MFC会调用CWnd的Attach,该函数会把当前的CWnd指针记录到当前模块的模块线程状态中,代码如下：

BOOL CWnd::Attach(HWND hWndNew){    ASSERT(m_hWnd == NULL);     // only attach once, detach on destroy    ASSERT(FromHandlePermanent(hWndNew) == NULL);        // must not already be in permanent map    if (hWndNew == NULL)        return FALSE;    CHandleMap* pMap = afxMapHWND(TRUE); // create map if not exist    ASSERT(pMap != NULL);    pMap->SetPermanent(m_hWnd = hWndNew, this);#ifndef _AFX_NO_OCC_SUPPORT    AttachControlSite(pMap);#endif    return TRUE;}

因为DLL中窗口的创建是在一个导出函数中，并在调用CWnd::Create这前调用了AFX_MANAGE_STATE(AfxGetStaticModuleState())来切换模块线程状态，导致该窗口所在的模块线程状态和MFC调用CWinApp::PreTranslateMessage时的不同，所以DLL中的窗口就无法响应PreTranslateMessage函数了。

解决方案：

1、dll导出一条函数 DllPreTranslateMe

BOOL PASCAL DllPreTranslateMessage(MSG *pMsg){    AFX_MANAGE_STATE(AfxGetStaticModuleState());    return theApp.PreTranslateMessage(pMsg);}

2、在主程序的CWinApp的PreTranslateMessage中直接调用DLL的DllPreTranslateMessage函数。但记住要先调用DLL中的函数。

BOOL CMyApp::PreTranslateMessage(MSG* pMsg){    // TODO: Add your specialized code here and/or call the base class    if(DllPreTranslateMessage(pMsg))        return TRUE;    return CWinApp::PreTranslateMessage(pMsg);}

经过以上两步，DLL中的窗口就可以响应PreTranslateMessage了。