wndproc.cpp

来自「使非MFC窗口程序的窗口回调过程成为C++类的成员函数。」· C++ 代码 · 共 51 行

#include "stdafx.h"
#include "wndproc.h"


static unsigned char g_hook[] =
{
	0x8B, 0x44, 0x24, 0x10,			//	mov         eax,dword ptr [esp+10h]
	0x8B, 0x4C, 0x24, 0x0C,			//	mov         ecx,dword ptr [esp+0Ch]
	0x8B, 0x54, 0x24, 0x08,			//	mov         edx,dword ptr [esp+8]
	0x50,							//	push        eax
	0x8B, 0x44, 0x24, 0x08,			//	mov         eax,dword ptr [esp+8]
	0x51,							//	push        ecx
	0xB9, 0x00, 0x00, 0x00, 0x00,	//	mov         ecx,0 （类的this指针）
	0x52,							//	push        edx
	0x50,							//	push        eax
	0x51,							//	push		ecx
	0xE8, 0x00, 0x00, 0x00, 0x00,	//	call        CWndProc::WndProc
	0xC2, 0x10, 0x00				//	ret         10h
};


CWndProc::CWndProc()
{
	char	*p;
	LRESULT (CALLBACK CWndProc::*pfn)(HWND, UINT, WPARAM, LPARAM);

	CopyMemory(m_hook, g_hook, sizeof(g_hook));
	p = m_hook + 19;
	*((unsigned int *)p) = (unsigned int)this;

	pfn = WndProc;
	p = m_hook + 27;
	__asm
	{
		mov eax, pfn
		sub eax, 4
		mov edi, p
		sub eax, edi
		mov [edi], eax
	}
	m_pfnWndProc = (WNDPROC)&m_hook[0];
}


CWndProc::~CWndProc()
{
}