TnbMfcDebugStringDialog.h

[詳解]

#pragma once
#include "TnbMfcTinyListDialog.h"
#include "TnbApiHooker.h"
 
 
 
//TNB Library
namespace TNB {
namespace MFC {
 
 
 
#ifndef _TnbDOXYGEN //Document作成用シンボル
namespace DSD {
    typedef VOID (WINAPI* P_OutputDebugStringA)(LPCSTR lpOutputString);
    typedef VOID (WINAPI* P_OutputDebugStringW)(LPCWSTR lpOutputString);
    _SELECTANY CApiHookerT<P_OutputDebugStringA>* s_pHookerA = NULL;
    _SELECTANY CApiHookerT<P_OutputDebugStringW>* s_pHookerW = NULL;
    _SELECTANY CTinyListDialog* s_pListDialog = NULL;
};
#endif
 
 
 
class CDebugStringDialog : public CTinyListDialog
{
    DEFSUPER(CTinyListDialog);
public:
 
    CDebugStringDialog(void) : _super(false)
    {
    }
 
protected:
 
    virtual void OnCreate(void)
    {
        if ( DSD::s_pListDialog == this || DSD::s_pListDialog != NULL )
        {
            return;
        }
        DSD::s_pHookerA = new CApiHookerT<DSD::P_OutputDebugStringA>;
        if ( DSD::s_pHookerA->Hook(NULL, "OutputDebugStringA", ms_MyOutputDebugStringA) )
        {
            DSD::s_pHookerW = new CApiHookerT<DSD::P_OutputDebugStringW>;
            if ( DSD::s_pHookerW->Hook(NULL, "OutputDebugStringW", ms_MyOutputDebugStringW) )
            {
                DSD::s_pListDialog = this;
                return;
            }
        }
        OnDestroy();
    }
 
    virtual void OnDestroy(void)
    {
        DSD::s_pListDialog = NULL;
        if ( DSD::s_pHookerA != NULL )
        {
            DSD::s_pHookerA->Unhook();
            delete DSD::s_pHookerA;
            DSD::s_pHookerA = NULL;
        }
        if ( DSD::s_pHookerW != NULL )
        {
            DSD::s_pHookerW->Unhook();
            delete DSD::s_pHookerW;
            DSD::s_pHookerW = NULL;
        }
    }
 
private:
 
    static void ms_PumpMessage(void)
    {
        HWND h = *DSD::s_pListDialog;
        MSG msg;
        while ( ::PeekMessage(&msg, h, NULL, NULL, PM_NOREMOVE) )
        {
            if ( ::GetMessage(&msg, h, NULL, NULL) )
            {
                ::TranslateMessage(&msg);
                ::DispatchMessage(&msg);
            }
        }
    }
 
    static VOID WINAPI ms_MyOutputDebugStringA(LPCSTR lpszOutputString)
    {
        if ( DSD::s_pListDialog != NULL )
        {
            DSD::s_pListDialog->PostString(CStr(lpszOutputString));
            ms_PumpMessage();
        }
        (*DSD::s_pHookerA)()(lpszOutputString);
    }
 
    static VOID WINAPI ms_MyOutputDebugStringW(LPCWSTR lpszOutputString)
    {
        if ( DSD::s_pListDialog != NULL )
        {
            DSD::s_pListDialog->PostString(CStr(lpszOutputString));
            ms_PumpMessage();
        }
        (*DSD::s_pHookerW)()(lpszOutputString);
    }
};
 
 
 
}; // MFC
}; // TNB