第一部分： C++开发BHO之HelloWorld

第一部分： C++开发BHO之HelloWorld

开发工具VS2010。

步骤如图：

生成的HelloWorldBHO.h的代码如下：

[cpp] view plain copy

1. // HelloWorldBHO.h : CHelloWorldBHO 的声明  
2.   
3. #pragma once  
4. #include "resource.h"       // 主符号  
5.   
6.   
7.   
8. #include "HelloWorld_i.h"  
9.   
10.   
11.   
12. #if defined(_WIN32_WCE) && !defined(_CE_DCOM) && !defined(_CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA)  
13. #error "Windows CE 平台(如不提供完全 DCOM 支持的 Windows Mobile 平台)上无法正确支持单线程 COM 对象。定义 _CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA 可强制 ATL 支持创建单线程 COM 对象实现并允许使用其单线程 COM 对象实现。rgs 文件中的线程模型已被设置为“Free”，原因是该模型是非 DCOM Windows CE 平台支持的唯一线程模型。"  
14. #endif  
15.   
16. using namespace ATL;  
17.   
18.   
19. // CHelloWorldBHO  
20.   
21. class ATL_NO_VTABLE CHelloWorldBHO :  
22.     public CComObjectRootEx<CComSingleThreadModel>,  
23.     public CComCoClass<CHelloWorldBHO, &CLSID_HelloWorldBHO>,  
24.     public IObjectWithSiteImpl<CHelloWorldBHO>,  
25.     public IDispatchImpl<IHelloWorldBHO, &IID_IHelloWorldBHO, &LIBID_HelloWorldLib, /*wMajor =*/ 1, /*wMinor =*/ 0>  
26. {  
27. public:  
28.     CHelloWorldBHO()  
29.     {  
30.     }  
31.   
32. DECLARE_REGISTRY_RESOURCEID(IDR_HELLOWORLDBHO)  
33.   
34. DECLARE_NOT_AGGREGATABLE(CHelloWorldBHO)  
35.   
36. BEGIN_COM_MAP(CHelloWorldBHO)  
37.     COM_INTERFACE_ENTRY(IHelloWorldBHO)  
38.     COM_INTERFACE_ENTRY(IDispatch)  
39.     COM_INTERFACE_ENTRY(IObjectWithSite)  
40. END_COM_MAP()  
41.   
42.   
43.   
44.     DECLARE_PROTECT_FINAL_CONSTRUCT()  
45.   
46.     HRESULT FinalConstruct()  
47.     {  
48.         return S_OK;  
49.     }  
50.   
51.     void FinalRelease()  
52.     {  
53.     }  
54.   
55. public:  
56.   
57.   
58.   
59. };  
60.   
61. OBJECT_ENTRY_AUTO(__uuidof(HelloWorldBHO), CHelloWorldBHO)  

HelloWorldBHO.cpp的代码如下：

[cpp] view plain copy

1. // HelloWorldBHO.cpp : CHelloWorldBHO 的实现  
2.   
3. #include "stdafx.h"  
4. #include "HelloWorldBHO.h"  
5.   
6.   
7. // CHelloWorldBHO  

HelloWorld.cpp的代码如下：

[cpp] view plain copy

1. // HelloWorld.cpp : DLL 导出的实现。  
2.   
3.   
4. #include "stdafx.h"  
5. #include "resource.h"  
6. #include "HelloWorld_i.h"  
7. #include "dllmain.h"  
8.   
9.   
10. // 用于确定 DLL 是否可由 OLE 卸载。  
11. STDAPI DllCanUnloadNow(void)  
12. {  
13.             return _AtlModule.DllCanUnloadNow();  
14.     }  
15.   
16. // 返回一个类工厂以创建所请求类型的对象。  
17. STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)  
18. {  
19.         return _AtlModule.DllGetClassObject(rclsid, riid, ppv);  
20. }  
21.   
22. // DllRegisterServer - 在系统注册表中添加项。  
23. STDAPI DllRegisterServer(void)  
24. {  
25.     // 注册对象、类型库和类型库中的所有接口  
26.     HRESULT hr = _AtlModule.DllRegisterServer();  
27.         return hr;  
28. }  
29.   
30. // DllUnregisterServer - 在系统注册表中移除项。  
31. STDAPI DllUnregisterServer(void)  
32. {  
33.     HRESULT hr = _AtlModule.DllUnregisterServer();  
34.         return hr;  
35. }  
36.   
37. // DllInstall - 按用户和计算机在系统注册表中逐一添加/移除项。  
38. STDAPI DllInstall(BOOL bInstall, LPCWSTR pszCmdLine)  
39. {  
40.     HRESULT hr = E_FAIL;  
41.     static const wchar_t szUserSwitch[] = L"user";  
42.   
43.     if (pszCmdLine != NULL)  
44.     {  
45.         if (_wcsnicmp(pszCmdLine, szUserSwitch, _countof(szUserSwitch)) == 0)  
46.         {  
47.             ATL::AtlSetPerUserRegistration(true);  
48.         }  
49.     }  
50.   
51.     if (bInstall)  
52.     {     
53.         hr = DllRegisterServer();  
54.         if (FAILED(hr))  
55.         {  
56.             DllUnregisterServer();  
57.         }  
58.     }  
59.     else  
60.     {  
61.         hr = DllUnregisterServer();  
62.     }  
63.   
64.     return hr;  
65. }  

dllmain.cpp的代码如下：

[cpp] view plain copy

1. // dllmain.cpp : DllMain 的实现。  
2.   
3. #include "stdafx.h"  
4. #include "resource.h"  
5. #include "HelloWorld_i.h"  
6. #include "dllmain.h"  
7.   
8. CHelloWorldModule _AtlModule;  
9.   
10. // DLL 入口点  
11. extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)  
12. {  
13.     hInstance;  
14.     return _AtlModule.DllMain(dwReason, lpReserved);   
15. }  

HelloWorld.rgs的内容如下：

[plain] view plain copy

1. HKCR  
2. {  
3. }  

HelloWorldBHO.rgs的内容如下：

[plain] view plain copy

1. HKCR  
2. {  
3.     NoRemove CLSID  
4.     {  
5.         ForceRemove {22024CC8-9043-49C7-AC80-A803A2912BF6} = s 'HelloWorldBHO Class'  
6.         {  
7.             ForceRemove Programmable  
8.             InprocServer32 = s '%MODULE%'  
9.             {  
10.                 val ThreadingModel = s 'Apartment'  
11.             }  
12.             TypeLib = s '{D05D8B0F-8126-4FE2-BC3D-D5A5248E7025}'  
13.             Version = s '1.0'  
14.         }  
15.     }  
16. }  

HelloWorld.idl的内容：

[plain] view plain copy

1. // HelloWorld.idl : HelloWorld 的 IDL 源  
2. //  
3.   
4. // 此文件将由 MIDL 工具处理以  
5. // 产生类型库(HelloWorld.tlb)和封送处理代码。  
6.   
7. import "oaidl.idl";  
8. import "ocidl.idl";  
9.   
10. [  
11.     object,  
12.     uuid(9C44C518-EEBE-4126-B368-5B901F840255),  
13.     dual,  
14.     nonextensible,  
15.     pointer_default(unique)  
16. ]  
17. interface IHelloWorldBHO : IDispatch{  
18. };  
19. [  
20.     uuid(D05D8B0F-8126-4FE2-BC3D-D5A5248E7025),  
21.     version(1.0),  
22. ]  
23. library HelloWorldLib  
24. {  
25.     importlib("stdole2.tlb");  
26.     [  
27.         uuid(22024CC8-9043-49C7-AC80-A803A2912BF6)        
28.     ]  
29.     coclass HelloWorldBHO  
30.     {  
31.         [default] interface IHelloWorldBHO;  
32.     };  
33. };  

HelloWorld.idl中的CLSID是22024CC8-9043-49C7-AC80-A803A2912BF6

其他的文件，暂时不重要。

修改HelloWorldBHO.h：

1、增加头文件

[cpp] view plain copy

1. #include <shlguid.h>  
2. #include <exdispid.h>  

2、增加一个公有的SetSite方法：

[cpp] view plain copy

1. public:  
2.   
3.     STDMETHOD(SetSite)(IUnknown *pUnkSite);  

3、增加私有的成员变量：

[cpp] view plain copy

1. private:  
2.   
3.     CComPtr<IWebBrowser2>  m_spWebBrowser;  
4.     BOOL m_fAdvised;  

4、多继承一个类：

[cpp] view plain copy

1. public IDispEventImpl<1, CHelloWorldBHO, &DIID_DWebBrowserEvents2, &LIBID_SHDocVw, 1, 1>  

5、增加公有的方法OnDocumentComplete：

[cpp] view plain copy

1. BEGIN_SINK_MAP(CHelloWorldBHO)  
2.     SINK_ENTRY_EX(1, DIID_DWebBrowserEvents2, DISPID_DOCUMENTCOMPLETE, OnDocumentComplete)  
3. END_SINK_MAP()  
4.   
5.     // DWebBrowserEvents2  
6.     void STDMETHODCALLTYPE OnDocumentComplete(IDispatch *pDisp, VARIANT *pvarURL);  

修改HelloWorldBHO.cpp，实现那两个方法：

[cpp] view plain copy

1. STDMETHODIMP CHelloWorldBHO::SetSite(IUnknown* pUnkSite)  
2. {  
3.     if (pUnkSite != NULL)  
4.     {  
5.         // Cache the pointer to IWebBrowser2.  
6.         HRESULT hr = pUnkSite->QueryInterface(IID_IWebBrowser2, (void **)&m_spWebBrowser);  
7.         if (SUCCEEDED(hr))  
8.         {  
9.             // Register to sink events from DWebBrowserEvents2.  
10.             hr = DispEventAdvise(m_spWebBrowser);  
11.             if (SUCCEEDED(hr))  
12.             {  
13.                 m_fAdvised = TRUE;  
14.             }  
15.         }  
16.     }  
17.     else  
18.     {  
19.         // Unregister event sink.  
20.         if (m_fAdvised)  
21.         {  
22.             DispEventUnadvise(m_spWebBrowser);  
23.             m_fAdvised = FALSE;  
24.         }  
25.   
26.         // Release cached pointers and other resources here.  
27.         m_spWebBrowser.Release();  
28.     }  
29.   
30.     // Call base class implementation.  
31.     return IObjectWithSiteImpl<CHelloWorldBHO>::SetSite(pUnkSite);  
32. }  
33.   
34. void STDMETHODCALLTYPE CHelloWorldBHO::OnDocumentComplete(IDispatch *pDisp, VARIANT *pvarURL)  
35. {  
36.     // Retrieve the top-level window from the site.  
37.     HWND hwnd;  
38.     HRESULT hr = m_spWebBrowser->get_HWND((LONG_PTR*)&hwnd);  
39.     if (SUCCEEDED(hr))  
40.     {  
41.         // Output a message box when page is loaded.  
42.         MessageBox(hwnd, L"Hello World!", L"BHO", MB_OK);  
43.     }  
44. }  

dllmain.cpp的DllMain修改成：

[cpp] view plain copy

1. extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)  
2. {  
3.     if (dwReason == DLL_PROCESS_ATTACH)  
4.     {  
5.         DisableThreadLibraryCalls(hInstance);  
6.     }  
7.     return _AtlModule.DllMain(dwReason, lpReserved);   
8. }  

修改HelloWorld.rgs：

修改成：

[plain] view plain copy

1. HKLM {  
2.   NoRemove SOFTWARE {  
3.     NoRemove Microsoft {     
4.       NoRemove Windows {  
5.         NoRemove CurrentVersion {  
6.           NoRemove Explorer {  
7.             NoRemove 'Browser Helper Objects' {  
8.               ForceRemove '{D2F7E1E3-C9DC-4349-B72C-D5A708D6DD77}' = s 'HelloWorldBHO' {  
9.                 val 'NoExplorer' = d '1'  
10.               }  
11.             }  
12.           }  
13.         }  
14.       }  
15.     }  
16.   }  
17. }  

其中的{D2F7E1E3-C9DC-4349-B72C-D5A708D6DD77}，需要改成HelloWorld.idl中的CLSID，当前文件的CLSID是

{22024CC8-9043-49C7-AC80-A803A2912BF6}

修改完后：

HelloWorldBHO.h：

[cpp] view plain copy

1. // HelloWorldBHO.h : CHelloWorldBHO 的声明  
2.   
3. #pragma once  
4. #include "resource.h"       // 主符号  
5.   
6. #include <shlguid.h>  
7. #include <exdispid.h>  
8.   
9. #include "HelloWorld_i.h"  
10.   
11.   
12.   
13. #if defined(_WIN32_WCE) && !defined(_CE_DCOM) && !defined(_CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA)  
14. #error "Windows CE 平台(如不提供完全 DCOM 支持的 Windows Mobile 平台)上无法正确支持单线程 COM 对象。定义 _CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA 可强制 ATL 支持创建单线程 COM 对象实现并允许使用其单线程 COM 对象实现。rgs 文件中的线程模型已被设置为“Free”，原因是该模型是非 DCOM Windows CE 平台支持的唯一线程模型。"  
15. #endif  
16.   
17. using namespace ATL;  
18.   
19.   
20. // CHelloWorldBHO  
21.   
22. class ATL_NO_VTABLE CHelloWorldBHO :  
23.     public CComObjectRootEx<CComSingleThreadModel>,  
24.     public CComCoClass<CHelloWorldBHO, &CLSID_HelloWorldBHO>,  
25.     public IObjectWithSiteImpl<CHelloWorldBHO>,  
26.     public IDispatchImpl<IHelloWorldBHO, &IID_IHelloWorldBHO, &LIBID_HelloWorldLib, /*wMajor =*/ 1, /*wMinor =*/ 0>,  
27.     public IDispEventImpl<1, CHelloWorldBHO, &DIID_DWebBrowserEvents2, &LIBID_SHDocVw, 1, 1>  
28. {  
29. public:  
30.     CHelloWorldBHO()  
31.     {  
32.     }  
33.   
34. DECLARE_REGISTRY_RESOURCEID(IDR_HELLOWORLDBHO)  
35.   
36. DECLARE_NOT_AGGREGATABLE(CHelloWorldBHO)  
37.   
38. BEGIN_COM_MAP(CHelloWorldBHO)  
39.     COM_INTERFACE_ENTRY(IHelloWorldBHO)  
40.     COM_INTERFACE_ENTRY(IDispatch)  
41.     COM_INTERFACE_ENTRY(IObjectWithSite)  
42. END_COM_MAP()  
43.   
44.   
45.   
46.     DECLARE_PROTECT_FINAL_CONSTRUCT()  
47.   
48.     HRESULT FinalConstruct()  
49.     {  
50.         return S_OK;  
51.     }  
52.   
53.     void FinalRelease()  
54.     {  
55.     }  
56.    
57. public:  
58.   
59.     STDMETHOD(SetSite)(IUnknown *pUnkSite);  
60.   
61. BEGIN_SINK_MAP(CHelloWorldBHO)  
62.     SINK_ENTRY_EX(1, DIID_DWebBrowserEvents2, DISPID_DOCUMENTCOMPLETE, OnDocumentComplete)  
63. END_SINK_MAP()  
64.   
65.     // DWebBrowserEvents2  
66.     void STDMETHODCALLTYPE OnDocumentComplete(IDispatch *pDisp, VARIANT *pvarURL);  
67.   
68. private:  
69.   
70.     CComPtr<IWebBrowser2>  m_spWebBrowser;  
71.     BOOL m_fAdvised;  
72. };  
73.   
74. OBJECT_ENTRY_AUTO(__uuidof(HelloWorldBHO), CHelloWorldBHO)  

HelloWorldBHO.cpp：

[cpp] view plain copy

1. // HelloWorldBHO.cpp : CHelloWorldBHO 的实现  
2.   
3. #include "stdafx.h"  
4. #include "HelloWorldBHO.h"  
5.   
6.   
7. // CHelloWorldBHO  
8.   
9. STDMETHODIMP CHelloWorldBHO::SetSite(IUnknown* pUnkSite)  
10. {  
11.     if (pUnkSite != NULL)  
12.     {  
13.         // Cache the pointer to IWebBrowser2.  
14.         HRESULT hr = pUnkSite->QueryInterface(IID_IWebBrowser2, (void **)&m_spWebBrowser);  
15.         if (SUCCEEDED(hr))  
16.         {  
17.             // Register to sink events from DWebBrowserEvents2.  
18.             hr = DispEventAdvise(m_spWebBrowser);  
19.             if (SUCCEEDED(hr))  
20.             {  
21.                 m_fAdvised = TRUE;  
22.             }  
23.         }  
24.     }  
25.     else  
26.     {  
27.         // Unregister event sink.  
28.         if (m_fAdvised)  
29.         {  
30.             DispEventUnadvise(m_spWebBrowser);  
31.             m_fAdvised = FALSE;  
32.         }  
33.   
34.         // Release cached pointers and other resources here.  
35.         m_spWebBrowser.Release();  
36.     }  
37.   
38.     // Call base class implementation.  
39.     return IObjectWithSiteImpl<CHelloWorldBHO>::SetSite(pUnkSite);  
40. }  
41.   
42. void STDMETHODCALLTYPE CHelloWorldBHO::OnDocumentComplete(IDispatch *pDisp, VARIANT *pvarURL)  
43. {  
44.     // Retrieve the top-level window from the site.  
45.     HWND hwnd;  
46.     HRESULT hr = m_spWebBrowser->get_HWND((LONG_PTR*)&hwnd);  
47.     if (SUCCEEDED(hr))  
48.     {  
49.         // Output a message box when page is loaded.  
50.         MessageBox(hwnd, L"Hello World!", L"BHO", MB_OK);  
51.     }  
52. }  

dllmain.cpp：

[cpp] view plain copy

1. // dllmain.cpp : DllMain 的实现。  
2.   
3. #include "stdafx.h"  
4. #include "resource.h"  
5. #include "HelloWorld_i.h"  
6. #include "dllmain.h"  
7.   
8. CHelloWorldModule _AtlModule;  
9.   
10. // DLL 入口点  
11. extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)  
12. {  
13.     if (dwReason == DLL_PROCESS_ATTACH)  
14.     {  
15.         DisableThreadLibraryCalls(hInstance);  
16.     }  
17.     return _AtlModule.DllMain(dwReason, lpReserved);   
18. }  

HelloWorld.rgs：

[plain] view plain copy

1. HKLM {  
2.   NoRemove SOFTWARE {  
3.     NoRemove Microsoft {     
4.       NoRemove Windows {  
5.         NoRemove CurrentVersion {  
6.           NoRemove Explorer {  
7.             NoRemove 'Browser Helper Objects' {  
8.               ForceRemove '{22024CC8-9043-49C7-AC80-A803A2912BF6}' = s 'HelloWorldBHO' {  
9.                 val 'NoExplorer' = d '1'  
10.               }  
11.             }  
12.           }  
13.         }  
14.       }  
15.     }  
16.   }  
17. }  

目前要修改的也就那么几个文件，然后关闭安全软件，进行build，生成的dll会自动注册到注册表里去，如图：

before：

after：

然后打开IE，如图：

也可以在IE的管理加载项中看到，工具-管理加载项：

可以启用或者禁用BHO。

这样，就完成了HelloWorld版本的BHO的开发。

但是，这样注册的BHO, 在IE管理加载项里面显示“未验证”，解决这也问题，

请看下一篇博文《BHO劫持数字证书验证与DLL后门通道（二）》