| // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef CHROME_FRAME_BUGGY_BHO_HANDLING_H_ |
| #define CHROME_FRAME_BUGGY_BHO_HANDLING_H_ |
| |
| #include <atlbase.h> |
| #include <atlcom.h> |
| #include <exdisp.h> |
| |
| #include <vector> |
| |
| #include "base/threading/thread_local.h" |
| #include "base/win/scoped_comptr.h" |
| |
| namespace buggy_bho { |
| |
| // Method prototype for IDispatch::Invoke. |
| typedef HRESULT (__stdcall* InvokeFunc)(IDispatch* me, DISPID dispid, |
| REFIID riid, LCID lcid, WORD flags, |
| DISPPARAMS* params, VARIANT* result, |
| EXCEPINFO* ei, UINT* err); |
| |
| // Construct a per thread instance of this class before firing web browser |
| // events that can be sent to buggy BHOs. This class will intercept those |
| // BHOs (see list in cc file) and ignore notifications to those components |
| // as long as ChromeFrame is the active view. |
| class BuggyBhoTls { |
| public: |
| // This method traverses the list of DWebBrowserEvents and DWebBrowserEvents2 |
| // subscribers and checks if any of the sinks belong to a list of |
| // known-to-be-buggy BHOs. |
| // For each of those, a patch will be applied that temporarily ignores certain |
| // invokes. |
| HRESULT PatchBuggyBHOs(IWebBrowser2* browser); |
| |
| // Returns the instance of the BuggyBhoTls object for the current thread. |
| static BuggyBhoTls* GetInstance(); |
| |
| // Destroys the BuggyBhoTls instance foe the current thread. |
| static void DestroyInstance(); |
| |
| void set_web_browser(IWebBrowser2* web_browser2) { |
| web_browser2_ = web_browser2; |
| } |
| |
| protected: |
| BuggyBhoTls(); |
| ~BuggyBhoTls(); |
| // internal implementation: |
| |
| // Called when a buggy instance is found to be subscribing to browser events. |
| void AddBuggyObject(IDispatch* obj); |
| |
| // Called from our patch to check if calls for this object should be ignored. |
| // The reason we do this check is because there might be more than one browser |
| // object running on the same thread (e.g. IE6) with one running CF and the |
| // other MSHTML. We don't want to drop events being fired by MSHTML, only |
| // events fired by CF since these BHOs should handle MSHTML correctly. |
| bool ShouldSkipInvoke(IDispatch* obj) const; |
| |
| // Static, protected member methods |
| |
| // Patches a subscriber if it belongs to a buggy dll. |
| bool PatchIfBuggy(IUnknown* unk, const IID& diid); |
| |
| // Patches the IDispatch::Invoke method. |
| static HRESULT PatchInvokeMethod(PROC* invoke); |
| |
| // This is our substitute function that is called instead of the buggy DLLs. |
| // Here we call IsBuggyObject to check if we should ignore invokes or allow |
| // them to go through. |
| static STDMETHODIMP BuggyBhoInvoke(InvokeFunc original, IDispatch* me, |
| DISPID dispid, REFIID riid, LCID lcid, WORD flags, DISPPARAMS* params, |
| VARIANT* result, EXCEPINFO* ei, UINT* err); |
| |
| protected: |
| // List of buggy subscribers. |
| std::vector<IDispatch*> bad_objects_; |
| // Where we store the current thread's instance. |
| static base::ThreadLocalPointer<BuggyBhoTls> s_bad_object_tls_; |
| // The IWebBrowser2 instance for this thread. |
| base::win::ScopedComPtr<IWebBrowser2> web_browser2_; |
| // Set to true when we are done patching the event sinks of buggy bho's. |
| bool patched_; |
| }; |
| |
| } // end namespace buggy_bho |
| |
| #endif // CHROME_FRAME_BUGGY_BHO_HANDLING_H_ |