[收藏]千方百计得到IHTMLDocument2的接口指针

天云上人

  根据IE浏览器的运行方式，有多种不同的方式可以获取文档指针。

9 h! g( e8 \+ k- p$ h8 @$ x' X  <1> 如果你在程序中使用MFC的 CHtmlView 视来浏览网页。
! K* d3 B, K3 }0 Z6 A& q: n5 z        取得文档的方法最简单，调用 CHtmlView::GetHtmlDocument() 函数。
: ?9 W6 r5 d' {: U5 k* c  <2> 如果你的程序中使用了“Web 浏览器” 的ActiveX 控件。
       取得文档的方法也比较简单，调用 CWebBrowser2::GetDocument() 函数。
  <3> 如果你的程序是用 ATL 写的 ActiveX 控件。
3 E3 B/ e6 S/ o# v- A& u        那么需要调用 IOleClientSite::GetContainer 得到 IOleContainer 接口，然后就可以通过 QueryInterface() 查询得到 IHTMLDocument2 的接口。主要代码如下：

CComPtr < IOleContainer > spContainer;
% H  Z8 K7 A5 X5 @m_spClientSite->GetContainer( &spContainer );
CComQIPtr < IHTMLDocument2 > spDoc = spContainer;
if ( spDoc )
{
- s% [1 ]+ R. b- P6 D1 }     // 已经得到了 IHTMLDocument2 的接口指针
; h. R: D, q+ `}

  <4> 如果你的程序是用 MFC 写的 ActiveX 控件。
       那么需要调用 COleControl::GetClientSite() 得到 IOleContainer 接口，然后的操作和<3>是一致的了。
  <5> IE 浏览器作为独立的进程正在运行。
0 z" M- g* ?" _* f6 P        每个运行的浏览器（IE 和 资源浏览器）都会在 ShellWindows 中进行登记，因此我们要通过 IShellWindows 取得实例（示例程序中使用的就是这个方法）。主要代码如下：

) x1 c' g. t) x8 i4 ]' O#include < atlbase.h >
' w  r. t1 V- |+ K#include < mshtml.h >
' }+ \% U  e: e1 L$ v/ \
! W7 n+ B6 o* z0 o$ P5 z/ {8 `void FindFromShell()
4 q. ~. s6 Y5 ]2 l7 p5 L{
9 {# H' m4 i4 A: F. t  l   CComPtr< IShellWindows > spShellWin;
! |5 b* R0 w. L   HRESULT hr = spShellWin.CoCreateInstance( CLSID_ShellWindows );
  if ( FAILED( hr ) )    return;
' e+ ?+ N( \/ R1 g7 N0 P1 `
$ b( E+ n: f: F' W8 u   long nCount=0;
  spShellWin->get_Count(&nCount);   // 取得浏览器实例个数

  ~8 g  d1 ?4 }1 T% Y   for(long i=0; i<nCount; i++)
, w9 W1 B* A+ U8 X4 k2 ^       {
) M6 p' h2 i9 x# k" S                CComPtr< IDispatch > spDisp;
     hr=spShellWin->Item(CComVariant( i ), &spDisp );
     if ( FAILED( hr ) )   continue;

     CComQIPtr< IWebBrowser2 > spBrowser = spDisp;
7 ?# Z/ u$ A5 R- P8 b      if ( !spBrowser )     continue;

     spDisp.Release();
% C# }* K$ ~3 w* p: E      hr = spBrowser->get_Document( &spDisp );
     if ( FAILED ( hr ) )  continue;
& {, b1 ?1 T( M' f6 c
     CComQIPtr< IHTMLDocument2 > spDoc = spDisp;
* \: e+ v8 \3 y% f/ q1 J      if ( !spDoc )         continue;

     // 程序运行到此，已经找到了 IHTMLDocument2 的接口指针
  }
}
' G3 L) }1 x* _4 b/ `. m
  <6> IE 浏览器控件被一个进程包装在一个子窗口中。那么你首先要得到那个进程的顶层窗口句柄（使用 FindWindow() 函数，或其它任何可行的方法），然后枚举所有子窗口，通过判断窗口类名是否是“Internet Explorer_Server”，从而得到浏览器的窗口句柄，再向窗口发消息取得文档的接口指针。主要代码如下：
5 X& M! @0 y: P4 ]# @0 x! J% V+ R
- x# n8 ]% Q7 `9 N7 T
  J* @2 N% S, k' J/ u#include < atlbase.h >
5 ?% x) u/ o" i/ K# ~* a5 [#include < mshtml.h >
- }4 l3 X' L" r/ t- h# _#include < oleacc.h >
#pragma comment ( lib, "oleacc" )

  t4 _+ Z; F( Y& B0 j% ZBOOL CALLBACK EnumChildProc(HWND hwnd,LPARAM lParam)
{
/ l: {+ h/ t! C$ V. F   TCHAR szClassName[100];

2 k  F8 q/ h" d% v; ~5 _7 h   ::GetClassName( hwnd,  &szClassName,  sizeof(szClassName) );
  if ( _tcscmp( szClassName,  _T("Internet Explorer_Server") ) == 0 )
5 x: N$ G% a$ G( v2 |% R7 `2 t   {
     *(HWND*)lParam = hwnd;
1 h  J3 N3 H! ?% }' \      return FALSE;      // 找到第一个 IE 控件的子窗口就停止
  }
  else   return TRUE;      // 继续枚举子窗口
};

void FindFromHwnd(HWND hWnd)
{
  HWND hWndChild=NULL;
+ d6 L$ p$ Q. W$ A( }6 m   ::EnumChildWindows( hWnd, EnumChildProc, (LPARAM)&hWndChild );
  if(NULL == hWndChild)   return;
4 U0 v9 [. l0 a7 V+ V4 j
  UINT nMsg = ::RegisterWindowMessage( _T("WM_HTML_GETOBJECT") );
. I+ n2 K& K- p' y+ L   LRESULT lRes;
' K8 K  P, B& V   ::SendMessageTimeout( hWndChild, nMsg, 0L, 0L, SMTO_ABORTIFHUNG, 1000, (DWORD*) &lRes );

  CComPtr < IHTMLDocument2 > spDoc;
  HRESULT hr = ::ObjectFromLresult ( lRes, IID_IHTMLDocument2, 0 , (LPVOID *) &spDoc );
2 U/ z5 n% V$ L1 W2 A   if ( FAILED ( hr ) )   return;

  // 程序运行到此，已经找到了 IHTMLDocument2 的接口指针
% q2 V+ I5 ], @' u* l' m) }( g}