Hook IE浏览器alert弹窗

需要拦截并获取到浏览器中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();
}

大功告成