TnbWindowProcedureHooker.h

[詳解]

#pragma once
#include "TnbSimpleMap.h"
 
 
 
//TNB Library
namespace TNB
{
 
 
 
#ifndef _TnbDOXYGEN //Document作成用シンボル
 // HWNDとリスナーのMAP
 _SELECTANY CSimpleMapT<HWND, LPVOID>* s_pmapWphHwndProc = NULL; 
 _SELECTANY bool s_isWphInit = true; 
#endif
 
 
 
class CWindowProcedureHooker
{
public:
    CWindowProcedureHooker(void) : m_hWnd(NULL), m_pOrgWndProc(NULL)
    {
    }
 
    ~CWindowProcedureHooker(void)
    {
        Destroy();
    }
 
    bool IsAttached(void) const
    {
        return m_pOrgWndProc != NULL;
    }
 
    bool Attach(HWND hWnd)
    {
        Detach();
        if ( s_isWphInit && s_pmapWphHwndProc == NULL )
        {
            s_isWphInit = false;
            static CReject s_cRejeWPH;
            s_pmapWphHwndProc = new CSimpleMapT<HWND, LPVOID>;
        }
        if ( s_pmapWphHwndProc == NULL || s_pmapWphHwndProc->HasKey(hWnd) )
        {
            return false;
        }
        //
        PreAttach(hWnd);
        //
        WNDPROC oldWndProc = m_SetWndProc(hWnd, ms_WndProc);
        if ( oldWndProc == ms_WndProc || oldWndProc == NULL )
        {
            return false;
        }
        m_pOrgWndProc = oldWndProc;
        m_hWnd = hWnd;
        (*s_pmapWphHwndProc)[hWnd] = this;
        //
        OnAttached();
        return true;
    }
 
    void Detach(void)
    {
        if ( m_pOrgWndProc != NULL )
        {
            PreDetach();
            WNDPROC oldWndProc = m_SetWndProc(m_hWnd, m_pOrgWndProc);
            IgnoreUnusedValue(oldWndProc);
            if ( s_pmapWphHwndProc != NULL )
            {
                s_pmapWphHwndProc->RemoveKey(m_hWnd);
            }
            OnDetached();
            m_hWnd = NULL;
            m_pOrgWndProc = NULL;
        }
    }
 
    void Destroy(void)
    {
        if ( m_pOrgWndProc != NULL )
        {
            WNDPROC oldWndProc = m_SetWndProc(m_hWnd, m_pOrgWndProc);
            IgnoreUnusedValue(oldWndProc);
            if ( s_pmapWphHwndProc != NULL )
            {
//              s_pmapWphHwndProc->RemoveKey(m_hWnd);
            }
            m_hWnd = NULL;
            m_pOrgWndProc = NULL;
        }
    }
 
    HWND GetSafeHwnd(void) const
    {
        return m_hWnd;
    }
 
    LRESULT SendMessage(UINT message, WPARAM wParam, LPARAM lParam)
    {
        return ::SendMessage(m_hWnd, message, wParam, lParam);
    }
 
    BOOL PostMessage(UINT message, WPARAM wParam, LPARAM lParam)
    {
        return ::PostMessage(m_hWnd, message, wParam, lParam);
    }
 
    DWORD GetStyle(void) const
    {
        return ::GetWindowLong(m_hWnd, GWL_STYLE);
    }
 
    DWORD GetExStyle(void) const
    {
        return ::GetWindowLong(m_hWnd, GWL_EXSTYLE);
    }
 
    void ModifyStyle(DWORD dwRemove, DWORD dwAdd)
    {
        DWORD dwStyle = ::GetWindowLong(m_hWnd, GWL_STYLE);
        DWORD dwNewStyle = (dwStyle & ~dwRemove) | dwAdd;
        if ( dwStyle != dwNewStyle )
        {
            ::SetWindowLong(m_hWnd, GWL_STYLE, dwNewStyle);
        }
    }
 
    void ModifyStyleEx(DWORD dwRemove, DWORD dwAdd)
    {
        DWORD dwStyle = ::GetWindowLong(m_hWnd, GWL_EXSTYLE);
        DWORD dwNewStyle = (dwStyle & ~dwRemove) | dwAdd;
        if ( dwStyle != dwNewStyle )
        {
            ::SetWindowLong(m_hWnd, GWL_EXSTYLE, dwNewStyle);
        }
    }
 
protected:
 
    virtual void PreAttach(HWND hWnd) {}
    
    virtual void OnAttached(void) {}
    
    virtual void PreDetach(void) {}
 
    virtual void OnDetached(void) {}
 
    virtual LRESULT WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
    {
        LRESULT r = ::CallWindowProc(m_pOrgWndProc, m_hWnd, message, wParam, lParam);
        if ( message == WM_DESTROY )
        {
            Detach();
        }
        return r;
    }
 
    HWND                m_hWnd;                 
 
private:
    // WNDPROC設定
    WNDPROC m_SetWndProc(HWND hWnd, WNDPROC P)
    {
        LONG_PTR i = reinterpret_cast<LONG_PTR>(P);
        return reinterpret_cast<WNDPROC>(::SetWindowLongPtr(hWnd, GWLP_WNDPROC, i));
    }
 
    static LRESULT CALLBACK ms_WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
    {
        if ( s_pmapWphHwndProc != NULL )
        {
            CWindowProcedureHooker* P = static_cast<CWindowProcedureHooker*>((*s_pmapWphHwndProc)[hWnd]);
            if ( P != NULL )
            {
                return P->WindowProc(message, wParam, lParam);
            }
        }
        return 0;
    }
    //Unhook処理用サブクラス
    class CReject
    {
    public: 
        ~CReject(void)
        {
            if ( s_pmapWphHwndProc != NULL )
            {
//              ASSERT( s_pmapWphHwndProc->GetSize() == 0 );
                s_pmapWphHwndProc->RemoveAll();
                delete s_pmapWphHwndProc;
                s_pmapWphHwndProc = NULL;
            }
        }
    };
    WNDPROC             m_pOrgWndProc;          
};
 
 
 
}; //TNB