2008年（909）
分类：
2008-05-06 21:53:21
下载源代码
一直以来，编写非MFC下的窗口程序，都习惯把窗口过程及消息处理函数编写成全局函数。为了把窗口回调过程及窗口消息处理函数封装成C 窗口类的成员函数，于是我编写了抽象类CWndProc：
一、头文件
//wndpro.h
#ifndef __WNDPROC_H__
#define __WNDPROC_H__
class CWndProc
{
protected:
//保护的构造函数，必须由派生类来构造。
CWndProc();
virtual ~CWndProc();
protected:
//窗口回调过程，基类作为纯虚函数没有实现代码。
virtual LRESULT CALLBACK WndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) = 0;
private:
//Hook代码块。
char m_hook[40];
protected:
//m_pfnWndProc指针指向Hook代码块的始地址。
//注册窗口类（WNDCLASSEX），或者子类化控件窗口，或者DialogBox显示对话框
//等需要窗口回调过程参数时，使用m_pfnWndProc作为参数。
WNDPROC m_pfnWndProc;
};
#endif //__WNDPROC_H__
//end of file
二、实现代码文件
//wndproc.cpp
#include "stdafx.h"
#include "wndproc.h"
/*
全局的Hook代码，其C的伪代码为：
LRSULT CALLBACK WndProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
return (CWndProc派生类的this指针)->WndProc(hwnd, uMsg, wParam, lParam);
}
代码的功能就是直接转调用CWndProc派生类的WndProc。
*/
static unsigned char g_hook[] =
{
0x8B, 0x44, 0x24, 0x10, // mov eax,dword ptr [esp 10h] ; eax <- lParam
0x8B, 0x4C, 0x24, 0x0C, // mov ecx,dword ptr [esp 0Ch] ; ecx <- wParam
0x8B, 0x54, 0x24, 0x08, // mov edx,dword ptr [esp 8] ; edx <- uMsg
0x50, // push eax ; lParam 参数入栈
0x8B, 0x44, 0x24, 0x08, // mov eax,dword ptr [esp 8] ; eax <- hwnd
0x51, // push ecx ; wParam 参数入栈
0xB9, 0x00, 0x00, 0x00, 0x00, // mov ecx,0 ; ecx <- this指针，这里暂时用this(NULL)，
// ; 在类构造函数初始化时修改为实际类的this指针值
0x52, // push edx ; uMsg 参数入栈
0x50, // push eax ; hwnd 参数入栈
0x51, // push ecx ; this 参数入栈
0xE8, 0x00, 0x00, 0x00, 0x00, // call WndProc ; 调用派生类的WndProc，这暂时用0，
// ; 在类构造函数初始化时修改为实际类虚拟表WndProc指针偏移值
0xC2, 0x10, 0x00 // ret 10h ; return
};
CWndProc::CWndProc()
{
char *p;
LRESULT (CALLBACK CWndProc::*pfn)(HWND, UINT, WPARAM, LPARAM);
CopyMemory(m_hook, g_hook, sizeof(g_hook)); //把全局的Hook代码块，拷贝到类对象的Hook代码块
p = m_hook 19; //p指针指向 mov ecx, 0 处，以便修改this(NULL)指针为实际类对象的this指针
*((unsigned int *)p) = (unsigned int)this; //修改p所指向的位置为mov ecx, （指向实际的类对象的this指针）
pfn = WndProc; //pfn指向类虚拟表中WndProc函数的指针;
p = m_hook 27; //p指针指向 call WndProc处，以便修改WndProc虚表指针相对偏移值
//由于vc6.0无法修改pfn及强制其类型，所以下面使用几句汇编
__asm
{
mov eax, pfn ; eax <- pfn
sub eax, 4 ; eax <- eax-4
mov edi, p ; edi <- p指针
sub eax, edi ; eax <- eax-edi 计算WndProc虚表指针与当前 EIP 5 相对偏移值
mov [edi], eax ; eax <- [edi] 修改p所指向的位置为 call （WndProc虚表指针与当前 EIP 5 相对偏移值）
}
m_pfnWndProc = (WNDPROC)&m_hook[0]; //把Hook代码块始址赋给m_pfnWndProc
}
CWndProc::~CWndProc()
{
}
//enf of file.
三、例子代码