1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 // MeasurePageLoadTimeBHO.h : Declaration of the CMeasurePageLoadTimeBHO 6 7 #include "resource.h" // main symbols 8 9 #include <shlguid.h> // IID_IWebBrowser2, DIID_DWebBrowserEvents2, et 10 #include <exdispid.h> // DISPID_DOCUMENTCOMPLETE, etc. 11 12 #include <string> 13 14 #include "MeasurePageLoadTime.h" 15 16 17 #if defined(_WIN32_WCE) && !defined(_CE_DCOM) && !defined(_CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA) 18 #error "Single-threaded COM objects are not properly supported on Windows CE platform, such as the Windows Mobile platforms that do not include full DCOM support. Define _CE_ALLOW_SINGLE_THREADED_OBJECTS_IN_MTA to force ATL to support creating single-thread COM objects and allow use of its single-threaded COM object implementations. The threading model in your rgs file was set to 'Free' as that is the only threading model supported in non DCOM Windows CE platforms." 19 #endif 20 21 22 23 // CMeasurePageLoadTimeBHO 24 25 class ATL_NO_VTABLE CMeasurePageLoadTimeBHO : 26 public CComObjectRootEx<CComSingleThreadModel>, 27 public CComCoClass<CMeasurePageLoadTimeBHO, &CLSID_MeasurePageLoadTimeBHO>, 28 public IObjectWithSiteImpl<CMeasurePageLoadTimeBHO>, 29 public IDispatchImpl<IMeasurePageLoadTimeBHO, &IID_IMeasurePageLoadTimeBHO, &LIBID_MeasurePageLoadTimeLib, /*wMajor =*/ 1, /*wMinor =*/ 0>, 30 public IDispEventImpl<1, CMeasurePageLoadTimeBHO, &DIID_DWebBrowserEvents2, &LIBID_SHDocVw, 1, 1> 31 { 32 public: 33 CMeasurePageLoadTimeBHO() 34 { 35 } 36 37 DECLARE_REGISTRY_RESOURCEID(IDR_MEASUREPAGELOADTIMEBHO) 38 39 DECLARE_NOT_AGGREGATABLE(CMeasurePageLoadTimeBHO) 40 41 BEGIN_COM_MAP(CMeasurePageLoadTimeBHO) 42 COM_INTERFACE_ENTRY(IMeasurePageLoadTimeBHO) 43 COM_INTERFACE_ENTRY(IDispatch) 44 COM_INTERFACE_ENTRY(IObjectWithSite) 45 END_COM_MAP() 46 47 BEGIN_SINK_MAP(CMeasurePageLoadTimeBHO) 48 SINK_ENTRY_EX(1, DIID_DWebBrowserEvents2, DISPID_DOCUMENTCOMPLETE, OnDocumentComplete) 49 END_SINK_MAP() 50 51 // DWebBrowserEvents2 52 void STDMETHODCALLTYPE OnDocumentComplete(IDispatch *pDisp, VARIANT *pvarURL); 53 STDMETHOD(SetSite)(IUnknown *pUnkSite); 54 55 DECLARE_PROTECT_FINAL_CONSTRUCT() 56 57 HRESULT FinalConstruct() 58 { 59 return S_OK; 60 } 61 62 void FinalRelease() 63 { 64 } 65 66 void ProcessPageTimeRequests(void); 67 void VisitNextURL(void); 68 void ErrorExit(void); 69 70 private: 71 CComPtr<IWebBrowser2> m_spWebBrowser; 72 BOOL m_fAdvised; 73 74 // Handle to global interface table 75 DWORD m_dwCookie; 76 77 // Handle to event to signal when navigation completes 78 HANDLE m_hEvent; 79 80 // Socket for accepting incoming connections 81 SOCKET m_sockListen; 82 83 // Socket for communicating with remote peers 84 SOCKET m_sockTransport; 85 }; 86 87 OBJECT_ENTRY_AUTO(__uuidof(MeasurePageLoadTimeBHO), CMeasurePageLoadTimeBHO) 88
