15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 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)#include "chrome_frame/bho_loader.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <atlbase.h> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <atlcomcli.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <exdisp.h> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome_frame/chrome_frame_helper_util.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome_frame/chrome_tab.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome_frame/event_hooker.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Describes the window class we look for. 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const wchar_t kStatusBarWindowClass[] = L"msctls_statusbar32"; 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// On IE9, the status bar is disabled by default, so we look for an 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// AsyncBoundaryLayer window instead. 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const wchar_t kAsyncBoundaryDnWindow[] = L"asynclayerboundarydn\0"; 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)BHOLoader::BHOLoader() : hooker_(new EventHooker()) { 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)BHOLoader::~BHOLoader() { 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (hooker_) { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delete hooker_; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hooker_ = NULL; 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void BHOLoader::OnHookEvent(DWORD event, HWND window) { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Step 1: Make sure that we are in a process named iexplore.exe. 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsNamedProcess(L"iexplore.exe")) { 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!IsWindowOfClass(window, kStatusBarWindowClass) && 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) !IsWindowOfClass(window, kAsyncBoundaryDnWindow)) { 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We have the right sort of window, check to make sure it was created 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // on the current thread. 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DWORD thread_id = GetWindowThreadProcessId(window, NULL); 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _ASSERTE(thread_id == GetCurrentThreadId()); 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Step 2: Check to see if the window is of the right class. 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HWND browser_hwnd = NULL; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsWindowOfClass(window, kStatusBarWindowClass)) { 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For IE8 and under, IE loads BHOs in the WM_CREATE handler of the tab 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // window approximately after it creates the status bar window. To be as 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // close to IE as possible in our simulation on BHO loading, we watch for 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the status bar to be created and do our simulated BHO loading at that 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // time. 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) browser_hwnd = GetParent(window); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else if (IsWindowOfClass(window, kAsyncBoundaryDnWindow)) { 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // For IE9, the status bar is disabled by default, so we look for an 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // AsyncBoundaryWindow to be created. When we find that, look for a 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // child window owned by the current thread named "tabwindowclass". 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // That will be our browser window. 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) browser_hwnd = RecurseFindWindow(NULL, L"tabwindowclass", NULL, 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetCurrentThreadId(), 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetCurrentProcessId()); 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _ASSERTE(NULL != browser_hwnd); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (browser_hwnd != NULL) { 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Step 3: 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Parent window of status bar window is the web browser window. Try to 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // get its IWebBrowser2 interface 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CComPtr<IWebBrowser2> browser; 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) UtilGetWebBrowserObjectFromWindow(browser_hwnd, __uuidof(browser), 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) reinterpret_cast<void**>(&browser)); 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (browser) { 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (IsSystemLevelChromeFrameInstalled()) { 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We're in the right place, but a system-level installation has 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // appeared. We should leave now. 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Figure out if we're already in the property map. 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) wchar_t bho_clsid_as_string[MAX_PATH] = {0}; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StringFromGUID2(CLSID_ChromeFrameBHO, bho_clsid_as_string, 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ARRAYSIZE(bho_clsid_as_string)); 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CComBSTR bho_clsid_as_string_bstr(bho_clsid_as_string); 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CComVariant existing_bho; 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT hr = browser->GetProperty(bho_clsid_as_string_bstr, 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &existing_bho); 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (V_VT(&existing_bho) != VT_DISPATCH && 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) V_VT(&existing_bho) != VT_UNKNOWN) { 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Step 4: 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // We have the IWebBrowser2 interface. Now create the BHO instance 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CComPtr<IObjectWithSite> bho_object; 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hr = bho_object.CoCreateInstance(CLSID_ChromeFrameBHO, 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CLSCTX_INPROC_SERVER); 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _ASSERTE(bho_object); 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (SUCCEEDED(hr) && bho_object) { 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Step 5: 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize the BHO by calling SetSite and passing it IWebBrowser2 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hr = bho_object->SetSite(browser); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _ASSERTE(bho_object); 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (SUCCEEDED(hr)) { 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Step 6: 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Now add the BHO to the collection of automation objects. This 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // will ensure that BHO will be accessible from the web pages as 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // any other BHO. Importantly, it will make sure that our BHO 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // will be cleaned up at the right time along with other BHOs. 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CComVariant object_variant(bho_object); 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) browser->PutProperty(bho_clsid_as_string_bstr, object_variant); 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool BHOLoader::StartHook() { 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return hooker_->StartHook(); 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void BHOLoader::StopHook() { 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) hooker_->StopHook(); 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)BHOLoader* BHOLoader::GetInstance() { 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static BHOLoader loader; 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return &loader; 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 133