When Chrome hands off a URL to be opened by the external host by the ViewHost...

chrome_frame/bho.cc

 // Copyright (c) 2009 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.h"
 
 #include <shlguid.h>
 #include <shobjidl.h>
 
 #include "base/logging.h"
 #include "base/registry.h"
 #include "base/scoped_bstr_win.h"
 #include "base/scoped_comptr_win.h"
 #include "base/scoped_variant_win.h"
 #include "base/string_util.h"
 #include "chrome_tab.h" // NOLINT
 #include "chrome_frame/protocol_sink_wrap.h"
 #include "chrome_frame/utils.h"
 #include "chrome_frame/vtable_patch_manager.h"
+#include "net/http/http_util.h"
 
 const wchar_t kPatchProtocols[] = L"PatchProtocols";
 static const int kIBrowserServiceOnHttpEquivIndex = 30;
 
+base::LazyInstance<base::ThreadLocalPointer<Bho> >
+    Bho::bho_current_thread_instance_(base::LINKER_INITIALIZED);
+
 PatchHelper g_patch_helper;
 
 BEGIN_VTABLE_PATCHES(IBrowserService)
   VTABLE_PATCH_ENTRY(kIBrowserServiceOnHttpEquivIndex, Bho::OnHttpEquiv)
 END_VTABLE_PATCHES()
 
 _ATL_FUNC_INFO Bho::kBeforeNavigate2Info = {
   CC_STDCALL, VT_EMPTY, 7, {
     VT_DISPATCH,
     VT_VARIANT | VT_BYREF,
@@ skipping 22 matching lines @@
       ScopedComPtr<IBrowserService> browser_service;
       hr = DoQueryService(SID_SShellBrowser, site, browser_service.Receive());
       DCHECK(browser_service) << "DoQueryService - SID_SShellBrowser failed."
         << " Site: " << site << " Error: " << hr;
       if (browser_service) {
         g_patch_helper.PatchBrowserService(browser_service);
         DCHECK(SUCCEEDED(hr)) << "vtable_patch::PatchInterfaceMethods failed."
             << " Site: " << site << " Error: " << hr;
       }
     }
+    // Save away our BHO instance in TLS which enables it to be referenced by
+    // our active document/activex instances to query referrer and other
+    // information for a URL.
+    AddRef();
+    bho_current_thread_instance_.Pointer()->Set(this);
+  } else {
+    bho_current_thread_instance_.Pointer()->Set(NULL);
+    Release();
   }
 
   return IObjectWithSiteImpl<Bho>::SetSite(site);
 }
 
 STDMETHODIMP Bho::BeforeNavigate2(IDispatch* dispatch, VARIANT* url,
     VARIANT* flags, VARIANT* target_frame_name, VARIANT* post_data,
     VARIANT* headers, VARIANT_BOOL* cancel) {
   ScopedComPtr<IWebBrowser2> web_browser2;
   if (dispatch)
@@ skipping 29 matching lines @@
                                              target_frame_name, post_data,
                                              headers);
         DCHECK(SUCCEEDED(hr)) << "web_browser2->Navigate2 failed. Error: " << hr
             << std::endl << "Url: " << current_url
             << std::endl << "flags: " << flags
             << std::endl << "post data: " << post_data
             << std::endl << "headers: " << headers;
       }
     }
   }
+
+  referrer_.clear();
+
+  // Save away the referrer in case our active document needs it to initiate
+  // navigation in chrome.
+  if (headers && V_VT(headers) == VT_BSTR && headers->bstrVal != NULL) {
+    std::string raw_headers_utf8 = WideToUTF8(headers->bstrVal);
+    std::string http_headers =
+        net::HttpUtil::AssembleRawHeaders(raw_headers_utf8.c_str(),
+                                          raw_headers_utf8.length());
+    net::HttpUtil::HeadersIterator it(http_headers.begin(), http_headers.end(),
+                                      "\r\n");
+    while (it.GetNext()) {
+      if (LowerCaseEqualsASCII(it.name(), "referer")) {
+        referrer_ = it.values();
+        break;
+      }
+    }
+  }
   return S_OK;
 }
 
 HRESULT Bho::FinalConstruct() {
   return S_OK;
 }
 
 void Bho::FinalRelease() {
 }
 
@@ skipping 78 matching lines @@
     ScopedVariant url_var(url.c_str());
     hr = web_browser2->Navigate2(url_var.AsInput(), &flags, &empty, &empty,
                                  &empty);
     DCHECK(SUCCEEDED(hr)) << "web_browser2->Navigate2 failed. Error: " << hr
         << std::endl << "Url: " << url;
   }
 
   return S_OK;
 }
 
+Bho* Bho::GetCurrentThreadBhoInstance() {
+  DCHECK(bho_current_thread_instance_.Pointer()->Get() != NULL);
+  return bho_current_thread_instance_.Pointer()->Get();
+}
+
 void PatchHelper::InitializeAndPatchProtocolsIfNeeded() {
   if (state_ != UNKNOWN)
     return;
 
   bool patch_protocol = GetConfigBool(true, kPatchProtocols);
   if (patch_protocol) {
     ProtocolSinkWrap::PatchProtocolHandlers();
     state_ = PATCH_PROTOCOL;
   } else {
     state_ = PATCH_IBROWSER;
@@ skipping 10 matching lines @@
 void PatchHelper::UnpatchIfNeeded() {
   if (state_ == PATCH_PROTOCOL) {
     ProtocolSinkWrap::UnpatchProtocolHandlers();
   } else if (state_ == PATCH_IBROWSER_OK) {
     vtable_patch::UnpatchInterfaceMethods(IBrowserService_PatchInfo);
   }
 
   state_ = UNKNOWN;
 }