开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > Internet/IE编程 > 查看源码
IEHlprObj.cpp
资源名称：IEHelper1.rar　[点击查看]
上传用户：lexian2
上传日期：2021-05-21
资源大小：3915k
文件大小：5k
源码类别：

Internet/IE编程

开发平台：

Visual C++

IEHlprObj.cpp：源码内容
  1. // IEHlprObj.cpp : Implementation of CIEHlprObj
  2. #include "stdafx.h"
  3. #include "IEHelper.h"
  4. #include "IEHlprObj.h"
  5. #include "ExDispID.h"
  6. #include <strstrea.h>
  7. #include <stdio.h>
  9. const char* const pszAppName = "IEHelper";
  11. /////////////////////////////////////////////////////////////////////////////
  12. // CIEHlprObj
  14. //
  15. // CIEHlprObj Methods
  16. //
  18. CIEHlprObj::~CIEHlprObj()
  19. {
  20. }
  22. BOOL CIEHlprObj::ManageConnection(enum ConnectType eConnectType)
  23. {
  24. if (!m_spWebBrowser2)
  25. return S_OK;
  27. HRESULT hr;
  28. //
  29. // If eConnectType is Advise then we are advising IE that we
  30. // want to handle events.  If eConnectType is Unadvise, we are
  31. // telling IE that we no longer want to handle events.
  32. //
  33. CComQIPtr<IConnectionPointContainer,
  34. &IID_IConnectionPointContainer> spCPContainer(m_spWebBrowser2);
  36. if (spCPContainer != NULL)
  37. {
  38. CComPtr<IConnectionPoint> spConnectionPoint;
  40. hr = spCPContainer->FindConnectionPoint(DIID_DWebBrowserEvents2, &spConnectionPoint);
  41. if (SUCCEEDED(hr))
  42. {
  43. if (eConnectType == Advise)
  44. {
  45. //
  46. // Advise the client site of our desire to be handle events
  47. //
  48. hr = spConnectionPoint->Advise((IDispatch*)this, &m_dwCookie);
  49. if (FAILED(hr))
  50. ATLTRACE("n%s: ManageConnection(): Failed to Advisenn", pszAppName);
  51. }
  52. else
  53. {
  54. // Remove us from the list of people interested...
  55. hr = spConnectionPoint->Unadvise(m_dwCookie);
  56. if (FAILED(hr))
  57. ATLTRACE("npszAppName: ManageConnection(): Failed to Unadvisenn", pszAppName);
  58. }
  59. }
  60. }
  62. return (SUCCEEDED(hr));
  63. }
  65. //
  66. // IOleObjectWithSite Methods
  67. //
  68. STDMETHODIMP CIEHlprObj::SetSite(IUnknown *pUnkSite)
  69. {
  70. USES_CONVERSION;
  72. if (!pUnkSite)
  73. ATLTRACE("nSetSite(): pUnkSite is NULLnn");
  74. else
  75. {
  76. // Query pUnkSite for the IWebBrowser2 interface.
  77. m_spWebBrowser2 = pUnkSite;
  79. if (m_spWebBrowser2)
  80. {
  81. // Connect to the browser in order to handle events.
  82. if (!ManageConnection(Advise))
  83. ::MessageBox(NULL, _T("Failure sinking events from IWebBrowser2"), pszAppName, MB_OK);
  84. }
  85. }
  87. return S_OK;
  88. }
  91. //
  92. // IDispatch Methods
  93. //
  94. STDMETHODIMP CIEHlprObj::Invoke(DISPID dispidMember, REFIID riid, LCID lcid, WORD wFlags,
  95.                                 DISPPARAMS* pDispParams, VARIANT* pvarResult,
  96.                                 EXCEPINFO*  pExcepInfo,  UINT* puArgErr)
  97. {
  98. USES_CONVERSION;
  100. if (!pDispParams)
  101. return E_INVALIDARG;
  103. //
  104. // Get the current URL
  105. //  
  106. LPOLESTR lpURL = NULL;
  107. m_spWebBrowser2->get_LocationURL(&lpURL);
  109. switch (dispidMember)
  110. {
  111. //
  112. // The parameters for this DISPID are as follows:
  113. // [0]: Cancel flag  - VT_BYREF|VT_BOOL
  114. // [1]: HTTP headers - VT_BYREF|VT_VARIANT
  115. // [2]: Address of HTTP POST data  - VT_BYREF|VT_VARIANT 
  116. // [3]: Target frame name - VT_BYREF|VT_VARIANT 
  117. // [4]: Option flags - VT_BYREF|VT_VARIANT
  118. // [5]: URL to navigate to - VT_BYREF|VT_VARIANT
  119. // [6]: An object that evaluates to the top-level or frame
  120. //      WebBrowser object corresponding to the event. 
  121. //
  122. case DISPID_BEFORENAVIGATE2:
  124. char *str;
  125. if (pDispParams->cArgs >= 5 && pDispParams->rgvarg[5].vt == (VT_BYREF|VT_VARIANT))
  126. {
  127.             CComVariant varURL(*pDispParams->rgvarg[5].pvarVal);
  128.             varURL.ChangeType(VT_BSTR);
  129. //转化要访问的网址为char *型
  130. str = OLE2A(varURL.bstrVal);
  131. }
  132. //如果正要访问的网址为要被拦截的，则stop
  133. if(strstr(str,"www.sina.com.cn")!=NULL)
  134. {
  135.             *pDispParams->rgvarg[0].pboolVal = TRUE;
  136.     MessageBox(NULL,"当前系统禁止浏览该页","警告",MB_ICONSTOP);
  137. return S_OK;
  138. }
  140. break;
  142. //
  143. // The parameters for this DISPID:
  144. // [0]: URL navigated to - VT_BYREF|VT_VARIANT
  145. // [1]: An object that evaluates to the top-level or frame
  146. //      WebBrowser object corresponding to the event. 
  147. //
  148. case DISPID_NAVIGATECOMPLETE2:
  150. break;
  152. case DISPID_DOCUMENTCOMPLETE:
  153. break;
  155. case DISPID_DOWNLOADBEGIN:
  156. break;
  158. case DISPID_DOWNLOADCOMPLETE:
  159. break;
  161. case DISPID_NEWWINDOW2:
  162. //设置是否允许弹出窗口
  163. /*READYSTATE m_ReadyState;
  164.         m_spWebBrowser2->get_ReadyState(&m_ReadyState);
  165.         if (m_ReadyState!=READYSTATE_COMPLETE)
  166.         {
  167.             *pDispParams->rgvarg[0].pboolVal = TRUE;
  168.             return S_OK;
  169.         }
  170. else
  171. {
  172.             *pDispParams->rgvarg[0].pboolVal = FALSE;
  173.             return S_OK;
  174. }*/
  175.     break;
  177. // The parameters for this DISPID:
  178. // [0]: Address of cancel flag - VT_BYREF|VT_BOOL
  179. //
  180. case DISPID_QUIT:
  182. break;
  184. default:
  186. break;
  187. }
  189. return S_OK;
  190. }