需要拦截并获取到浏览器中alert的信息。实际上就是Hook IHTMLWindow2.alert 接口 .
首先在引入Mshtml.h前,需要定义#define CINTERFACE,强制使用C类型的COM接口。
#define CINTERFACE
#include <MsHTML.h>
#include <exdispid.h>
#include <mshtmcid.h> // CMDSETID_Forms3 definition
#include <mshtmhst.h> // IDM_menu item definitions
我比较偷懒,直接使用detours来hook COM.
#include <detours.h>
#pragma comment(lib, "detours.lib")
对COM接口的hook, 在detours中和 API hook非常近似。首先定义函数指针原型。
typedef HRESULT (STDMETHODCALLTYPE *PFN_IHTMLWindow2_alert)(IHTMLWindow2 * This, BSTR message);
然后弄一个替代函数
// static
HRESULT STDMETHODCALLTYPE CWebBrowser::IHTMLWindow2_alert(IHTMLWindow2 * This, BSTR message)
{
return S_OK;
}
IHTMLWindow2 * pWnd2 = ...
// hook IHTMLWindow2.alert
if( pWnd2 )
{
PFN_IHTMLWindow2_alert pfnRealAlert = pWnd2->lpVtbl->alert;
DetourTransactionBegin();
DetourUpdateThread(GetCurrentThread());
DetourAttach(&(PVOID&)pfnRealAlert, IHTMLWindow2_alert);
DetourTransactionCommit();
}
大功告成