对dui界面 组件 hook的通杀方案

做这件事难点在于，dui 聊天框用的是无窗口RichEdit，因此不能像普通的RichEdit那样通过FindWindow找到窗口然后发送WM_GETTEXT来获取文本，但是是不是就没有办法了呢，当然不是。
大家先用Dependency Walker看一下RICHED20.dll这个文件，位于Windows/System32目录下，可以看到该文件导出了几个函数：
Ordinal ^ Hint Function Entry Point
2 (0x0002) 1 (0x0001) IID_IRichEditOle 0x00014C60
3 (0x0003) 2 (0x0002) IID_IRichEditOleCallback 0x00014C50
4 (0x0004) 0 (0x0000) CreateTextServices 0x0000D882
5 (0x0005) 5 (0x0005) IID_ITextServices 0x00014C20
6 (0x0006) 3 (0x0003) IID_ITextHost 0x00014C30
7 (0x0007) 4 (0x0004) IID_ITextHost2 0x00014C40
8 (0x0008) 6 (0x0006) REExtendedRegisterClass 0x0004BA5C
9 (0x0009) 7 (0x0007) RichEdit10ANSIWndProc 0x0003DB01
10 (0x000A) 8 (0x0008) RichEditANSIWndProc 0x00015681

这里大家需要注意的是CreateTextServices函数，在MSDN上查询后得知，该函数的原型如下：

HRESULT CreateTextServices( IUnknown *punkOuter,  
  ITextHost *pITextHost,  
  IUnknown **ppUnk  
 );  

第一个参数不用管，第二个参数暂时也不用管，至于第三个参数，返回一个ppUnk，通过这个ppUnk，我们可以通过调用QueryInterface得到ITextServices接口指针，比如：

ITextServices* lpTs;  
((IUnknown*)(*ppUnk))->QueryInterface(IID_ITextServices, (void**)(&lpTs));  

这时候在MSDN上查询一下ITextServices接口，可以看到它的成员函数

HRESULT TxGetText( BSTR *pbstrText);  

通过这个函数，我们就可以得到创建的无窗口RichEdit中的文本了。

让我们来理一理思路，用qq为列，综上所述，我们需要做的就是Hook一下 CreateTextServices函数，然后得到ppUnk这个指针，关于Hook API的方法我这里就不多说了

接下来让我一步一步说明这个函数的功能
1. 首先获取System32目录，然后载入System32下真正的RICHED20.dll，
2. 用GetProcAddress获取真正的CreateTextServices函数
3. 调用CreateTextServices函数，得到ppUnk指针
4. 调用ppUnk的QueryInterface得到ITextServices指针
到这一步的时候，就有两个个问题，首先，我们如何保存得到的ITextServices指针；其次，如何获取聊天窗口的句柄，下面我们一个一个来讲。
因为QQ在创建聊天窗口时不只一次的调用了CreateTextServices，事实上每创建一个窗口，就会调用三次，经过我后面的验证，第一次调用时创建的是发送新消息框，第二次调用时创建的是消息显示框，所以这里我采用的方法，是定义一个multimap<HWND,ITextServices*>类型的map，然后把跟每个窗口对应的三个ITextServices指针插入到该map中，这样以后如果我需要得到某个窗口的聊天框内容，只需要根据该窗口的句柄，然后经过map的查找得到相应的迭代器，将迭代器++，就可以得到对应的聊天框ITextServices指针。但是问题是如何获取聊天窗口的窗口句柄呢？因为我查遍了ITextServices相关的接口，包括ITextHost等，没有任何返回HWND的函数，所以这里我用了一个小技巧，ITextHost接口中有一个TxSetCapture函数用以捕获鼠标，当外部调用我们的CreateTextServices时，会传进来一个ITextHost*类型的指针,我们调用它的TxSetCapture(TRUE)函数，然后紧接着调用一下GetCapture，这样得到的就是RichEdit的HostWindow的窗口句柄了，对应于QQ的聊天窗口，就是QQ聊天窗口的句柄了。
5. 在得到了窗口句柄以后，我们需要判断一下调用CreateTextServices的HostWindow是不是聊天窗口，通过SPY++，我们发现聊天窗口具有0x860F0000的风格以及0x300的扩展风格，这时候我们调用一下GetWindowInfo，获取到窗口的一些信息，然后将取到的窗口风格与这两个值做一个比较，如果相同，我们就认为它是一个聊天窗口，并将它插入到map中。
到此，主要的工作已经基本完成，接下来就是如何从map获取聊天窗口的ITextServices指针，并且调用它的TxGetText函数获取聊天窗口中的信息问题了，这里我就不多说了，大家可以根据自己的需要来进行改造。

#define NEW_DLL_NAME _T("//RichEd20.Dll")  
  
HRESULT WINAPI CreateTextServices(IUnknown *punkOuter, ITextHost *pITextHost, IUnknown **ppUnk)  
{  
TCHAR szLib[MAX_PATH]; //255 is enough  
DWORD dw = GetSystemDirectory(szLib, MAX_PATH);  
if(dw == 0) return 0;  
szLib[dw] = TCHAR('/0');  
::lstrcat(szLib, NEW_DLL_NAME);  
HMODULE hLib = LoadLibrary(szLib);  
  if(!hLib) return 0;  
  
  lpCreateTextServices _CreateTextServices = (HRESULT (__stdcall *)    
(IUnknown*, ITextHost*, IUnknown**))  
::GetProcAddress(hLib, "CreateTextServices");  
if(!_CreateTextServices) return 0;  
  
HRESULT hr = (_CreateTextServices)(punkOuter, pITextHost, ppUnk);  
  
ITextServices* lpTx;  
((IUnknown*)(*ppUnk))->QueryInterface(IID_ITextServices, (void**)(&lpTx));  
  
//这里是为了获取聊天窗口的句柄,因为RichEdit是Windowless，所以只能通过这种方式获取hostwindow  
pITextHost->TxSetCapture(TRUE);  
HWND hWnd = GetCapture();  
pITextHost->TxSetCapture(FALSE);  
TCHAR szText[200] = {0};  
WINDOWINFO wi = {sizeof(wi)};  
GetWindowInfo(hWnd,&wi);  
if (wi.dwStyle == 0x860F0000 && wi.dwExStyle == 0x300) //这个风格，表示是聊天窗口  
{  
g_mapTextServices.insert(std::make_pair(hWnd,lpTx));  
}  
  
return hr;  
}  

原文地址:https://blog.csdn.net/zhongguoren666/article/details/6972891