VC版本的MakeObjectInstance把WNDPROC映射到类的成员函数

这段时间用VC封装Windows类库，没有MakeObjectInstance处理窗口消息确实不爽，又不想使用MFC的消息映射，这玩意的效率和美观只能呵呵。

至于MakeObjectInstance是什么，Delphi转过来的同学必然很了解这个方便的功能,就是动态构造一个函数把普通函数转到一个类的成员函数。

VC X86实现起来没问题，但是X64实现起来的麻烦在于不能内嵌汇编了，X64必须结合ASM文件编译的obj（这一点还是感激Delphi的编译器，X86和X64都可以内联汇编）。

我的实现方案是通过构造一段ShellCode来达到目的。

SIZE_T PageSize = 4096;
template < typename T> //产生一个代理函数
WNDPROC  MakeObjectInstance( LPVOID AObject, T AMethod)
{
union
{
T        MethodAddr; //成员函数指针
LPVOID   NomralAddr; //正常指针
}ut; //因为VC不允许成员函数指针转换到普通指针。只能变通的通过union来实现
const unsigned char BlockCode[] = {
#ifdef _WIN64
0x55, //{ push rbp }
0x48, 0x83, 0xEC, 0x40, //{ sub rsp,0x40 }
0x48, 0x8B, 0xEC, //{ mov rbp,rsp }
0x48, 0x89, 0x4D, 0x50, //{ mov[rbp + 0x50],rcx }
0x89, 0x55, 0x58, //{ mov[rbp + 0x58],edx }
0x4C, 0x89, 0x45, 0x60, //{ mov[rbp + 0x60],r8 }
0x4C, 0x89, 0x4D, 0x68, //{ mov[rbp + 0x68],r9 }
0x48, 0xB9, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, //{ mov rcx,AObject }
0x48, 0x8B, 0x55, 0x50, //{ mov rdx,[rbp + 0x50] }
0x44, 0x8B, 0x45, 0x58, //{ mov r8,[rbp + 0x58] }
0x4C, 0x8B, 0x4D, 0x60, //{ mov r9,[rbp + 0x60] }
0x48, 0x8B, 0x45, 0x68, //{ mov rax,[rbp + 0x68] }
0x48, 0x89, 0x44, 0x24, 0x20, //{ mov[rsp + 0x20],rax }
0x49, 0xBB, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, //{ mov r11, AMethod }
0x49, 0xFF, 0xD3, //{ call r11 }
0x48, 0x8D, 0x65, 0x40, //{ lea rsp,[rbp + 0x40] }
0x5D, //{ pop rbp }
0xC3 //{ ret }

#else
0x58, //{ pop eax }
0x68, 0x00, 0x00, 0x00, 0x00, //{ push AObject }
0x50, //{ push eax }
0xB8, 0x00, 0x00, 0x00, 0x00, //{ mov eax, AMethod }
0xFF, 0xE0 //{ jmp eax }

#endif // endif
};

size_t CodeBytes = sizeof (BlockCode);
LPVOID  Block = VirtualAlloc( nullptr , PageSize, MEM_COMMIT, PAGE_EXECUTE_READWRITE);
memcpy (Block, BlockCode, CodeBytes);
unsigned char * bBlock = (unsigned char *)Block;
ut.MethodAddr = AMethod;
#ifdef _WIN64
* PLONG64 (&bBlock[25])= LONG64 (AObject);
* PLONG64 (&bBlock[0x38]) = LONG64 (ut.NomralAddr);
#else
* PLONG32 (&bBlock[2]) = LONG32 (AObject);
* PLONG32 (&bBlock[8]) = LONG32 (ut.NomralAddr);
#endif
return (WNDPROC)Block;
}

//释放代理函数
void FreeObjectInstance(WNDPROC wndProc)
{
VirtualFree(wndProc, PageSize, MEM_RELEASE);
}

用法类似的如下

 

class MyClass{

LRESULT CALLBACK WndProc( HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)；

}

MyClass  c;

WNDCLASSEXW wcex;

wcex.cbSize = sizeof (WNDCLASSEX);

wcex.style          = CS_HREDRAW | CS_VREDRAW;
wcex.lpfnWndProc    = MakeObjectInstance(&c, &MyClass::WndProc); //使用MyClass::WndProc作为创消息处理函数
wcex.cbClsExtra     = 0;
wcex.cbWndExtra     = 0;
wcex.hInstance      = hInstance;
wcex.hIcon          = LoadIcon(hInstance,            MAKEINTRESOURCE(IDI_WNDPROCTEST));
wcex.hCursor        = LoadCursor( nullptr , IDC_ARROW);
wcex.hbrBackground  = ( HBRUSH )(COLOR_WINDOW+1);
wcex.lpszMenuName   = MAKEINTRESOURCEW(IDC_WNDPROCTEST);
wcex.lpszClassName  = szWindowClass;
wcex.hIconSm        = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_SMALL));

RegisterClassExW(&wcex);

再例如

class MyClass{

LRESULT CALLBACK WndProc( HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)；

}

MyClass  c;

SetWindowLongPtr(hWnd, GWL_WNDPROC,  ( LONG_PTR )MakeObjectInstance(&c, &MyClass::WndProc));

 

 

http://www.raysoftware.cn/?p=552