// Copyright (c) 2011 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.

#include "chrome_frame/bho_loader.h"

#include <atlbase.h>
#include <atlcomcli.h>
#include <exdisp.h>

#include "chrome_frame/chrome_frame_helper_util.h"
#include "chrome_frame/chrome_tab.h"
#include "chrome_frame/event_hooker.h"


// Describes the window class we look for.
const wchar_t kStatusBarWindowClass[] = L"msctls_statusbar32";

// On IE9, the status bar is disabled by default, so we look for an
// AsyncBoundaryLayer window instead.
const wchar_t kAsyncBoundaryDnWindow[] = L"asynclayerboundarydn\0";

BHOLoader::BHOLoader() : hooker_(new EventHooker()) {
}

BHOLoader::~BHOLoader() {
  if (hooker_) {
    delete hooker_;
    hooker_ = NULL;
  }
}

void BHOLoader::OnHookEvent(DWORD event, HWND window) {
  // Step 1: Make sure that we are in a process named iexplore.exe.
  if (IsNamedProcess(L"iexplore.exe")) {
    if (!IsWindowOfClass(window, kStatusBarWindowClass) &&
        !IsWindowOfClass(window, kAsyncBoundaryDnWindow)) {
      return;
    } else {
      // We have the right sort of window, check to make sure it was created
      // on the current thread.
      DWORD thread_id = GetWindowThreadProcessId(window, NULL);
      _ASSERTE(thread_id == GetCurrentThreadId());
    }

    // Step 2: Check to see if the window is of the right class.
    HWND browser_hwnd = NULL;
    if (IsWindowOfClass(window, kStatusBarWindowClass)) {
      // For IE8 and under, IE loads BHOs in the WM_CREATE handler of the tab
      // window approximately after it creates the status bar window. To be as
      // close to IE as possible in our simulation on BHO loading, we watch for
      // the status bar to be created and do our simulated BHO loading at that
      // time.
      browser_hwnd = GetParent(window);
    } else if (IsWindowOfClass(window, kAsyncBoundaryDnWindow)) {
      // For IE9, the status bar is disabled by default, so we look for an
      // AsyncBoundaryWindow to be created. When we find that, look for a
      // child window owned by the current thread named "tabwindowclass".
      // That will be our browser window.
      browser_hwnd = RecurseFindWindow(NULL, L"tabwindowclass", NULL,
                                       GetCurrentThreadId(),
                                       GetCurrentProcessId());
      _ASSERTE(NULL != browser_hwnd);
    }

    if (browser_hwnd != NULL) {
      // Step 3:
      // Parent window of status bar window is the web browser window. Try to
      // get its IWebBrowser2 interface
      CComPtr<IWebBrowser2> browser;
      UtilGetWebBrowserObjectFromWindow(browser_hwnd, __uuidof(browser),
                                        reinterpret_cast<void**>(&browser));
      if (browser) {
        if (IsSystemLevelChromeFrameInstalled()) {
          // We're in the right place, but a system-level installation has
          // appeared. We should leave now.
          return;
        }

        // Figure out if we're already in the property map.
        wchar_t bho_clsid_as_string[MAX_PATH] = {0};
        StringFromGUID2(CLSID_ChromeFrameBHO, bho_clsid_as_string,
                        ARRAYSIZE(bho_clsid_as_string));
        CComBSTR bho_clsid_as_string_bstr(bho_clsid_as_string);

        CComVariant existing_bho;
        HRESULT hr = browser->GetProperty(bho_clsid_as_string_bstr,
                                          &existing_bho);

        if (V_VT(&existing_bho) != VT_DISPATCH &&
            V_VT(&existing_bho) != VT_UNKNOWN) {
          // Step 4:
          // We have the IWebBrowser2 interface. Now create the BHO instance
          CComPtr<IObjectWithSite> bho_object;
          hr =  bho_object.CoCreateInstance(CLSID_ChromeFrameBHO,
                                            NULL,
                                            CLSCTX_INPROC_SERVER);

          _ASSERTE(bho_object);
          if (SUCCEEDED(hr) && bho_object) {
            // Step 5:
            // Initialize the BHO by calling SetSite and passing it IWebBrowser2
            hr = bho_object->SetSite(browser);
            _ASSERTE(bho_object);
            if (SUCCEEDED(hr)) {
              // Step 6:
              // Now add the BHO to the collection of automation objects. This
              // will ensure that BHO will be accessible from the web pages as
              // any other BHO. Importantly, it will make sure that our BHO
              // will be cleaned up at the right time along with other BHOs.
              CComVariant object_variant(bho_object);
              browser->PutProperty(bho_clsid_as_string_bstr, object_variant);
            }
          }
        }
      }
    }
  }
}

bool BHOLoader::StartHook() {
  return hooker_->StartHook();
}

void BHOLoader::StopHook() {
  hooker_->StopHook();
}

BHOLoader* BHOLoader::GetInstance() {
  static BHOLoader loader;
  return &loader;
}