15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2010 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef CHROME_FRAME_BUGGY_BHO_HANDLING_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define CHROME_FRAME_BUGGY_BHO_HANDLING_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <atlbase.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <atlcom.h> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <exdisp.h> 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/thread_local.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_comptr.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace buggy_bho { 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Method prototype for IDispatch::Invoke. 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef HRESULT (__stdcall* InvokeFunc)(IDispatch* me, DISPID dispid, 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) REFIID riid, LCID lcid, WORD flags, 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISPPARAMS* params, VARIANT* result, 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXCEPINFO* ei, UINT* err); 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Construct a per thread instance of this class before firing web browser 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// events that can be sent to buggy BHOs. This class will intercept those 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// BHOs (see list in cc file) and ignore notifications to those components 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// as long as ChromeFrame is the active view. 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class BuggyBhoTls { 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This method traverses the list of DWebBrowserEvents and DWebBrowserEvents2 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // subscribers and checks if any of the sinks belong to a list of 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // known-to-be-buggy BHOs. 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For each of those, a patch will be applied that temporarily ignores certain 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // invokes. 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT PatchBuggyBHOs(IWebBrowser2* browser); 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Returns the instance of the BuggyBhoTls object for the current thread. 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static BuggyBhoTls* GetInstance(); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Destroys the BuggyBhoTls instance foe the current thread. 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void DestroyInstance(); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void set_web_browser(IWebBrowser2* web_browser2) { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_browser2_ = web_browser2; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BuggyBhoTls(); 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ~BuggyBhoTls(); 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // internal implementation: 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Called when a buggy instance is found to be subscribing to browser events. 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void AddBuggyObject(IDispatch* obj); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Called from our patch to check if calls for this object should be ignored. 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The reason we do this check is because there might be more than one browser 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // object running on the same thread (e.g. IE6) with one running CF and the 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // other MSHTML. We don't want to drop events being fired by MSHTML, only 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // events fired by CF since these BHOs should handle MSHTML correctly. 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool ShouldSkipInvoke(IDispatch* obj) const; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Static, protected member methods 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Patches a subscriber if it belongs to a buggy dll. 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool PatchIfBuggy(IUnknown* unk, const IID& diid); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Patches the IDispatch::Invoke method. 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static HRESULT PatchInvokeMethod(PROC* invoke); 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This is our substitute function that is called instead of the buggy DLLs. 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Here we call IsBuggyObject to check if we should ignore invokes or allow 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // them to go through. 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static STDMETHODIMP BuggyBhoInvoke(InvokeFunc original, IDispatch* me, 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISPID dispid, REFIID riid, LCID lcid, WORD flags, DISPPARAMS* params, 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) VARIANT* result, EXCEPINFO* ei, UINT* err); 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // List of buggy subscribers. 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<IDispatch*> bad_objects_; 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Where we store the current thread's instance. 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static base::ThreadLocalPointer<BuggyBhoTls> s_bad_object_tls_; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The IWebBrowser2 instance for this thread. 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedComPtr<IWebBrowser2> web_browser2_; 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set to true when we are done patching the event sinks of buggy bho's. 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool patched_; 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // end namespace buggy_bho 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // CHROME_FRAME_BUGGY_BHO_HANDLING_H_ 92