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

chrome_frame/chrome_active_document.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.
 
 // Implementation of ChromeActiveDocument
 #include "chrome_frame/chrome_active_document.h"
 
 #include <hlink.h>
 #include <htiface.h>
 #include <mshtmcid.h>
@@ skipping 15 matching lines @@
 #include "base/string_util.h"
 #include "base/thread.h"
 #include "base/thread_local.h"
 
 #include "grit/generated_resources.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/navigation_types.h"
 #include "chrome/test/automation/browser_proxy.h"
 #include "chrome/test/automation/tab_proxy.h"
+#include "chrome_frame/bho.h"
 #include "chrome_frame/utils.h"
 
 const wchar_t kChromeAttachExternalTabPrefix[] = L"attach_external_tab";
 
 static const wchar_t kUseChromeNetworking[] = L"UseChromeNetworking";
 static const wchar_t kHandleTopLevelRequests[] =
     L"HandleTopLevelRequests";
 
 base::ThreadLocalPointer<ChromeActiveDocument> g_active_doc_cache;
 
@@ skipping 468 matching lines @@
     }
 
     find_dialog_.ShowWindow(SW_SHOW);
   }
 }
 
 void ChromeActiveDocument::OnViewSource() {
   DCHECK(navigation_info_.url.is_valid());
   std::string url_to_open = "view-source:";
   url_to_open += navigation_info_.url.spec();
-  OnOpenURL(0, GURL(url_to_open), NEW_WINDOW);
+  OnOpenURL(0, GURL(url_to_open), GURL(), NEW_WINDOW);
 }
 
 void ChromeActiveDocument::OnDetermineSecurityZone(const GUID* cmd_group_guid,
                                                    DWORD command_id,
                                                    DWORD cmd_exec_opt,
                                                    VARIANT* in_args,
                                                    VARIANT* out_args) {
   if (out_args != NULL) {
     out_args->vt = VT_UI4;
     out_args->ulVal = URLZONE_INTERNET;
   }
 }
 
-void ChromeActiveDocument::OnOpenURL(int tab_handle, const GURL& url_to_open,
+void ChromeActiveDocument::OnOpenURL(int tab_handle,
+                                     const GURL& url_to_open,
+                                     const GURL& referrer,
                                      int open_disposition) {
   // If the disposition indicates that we should be opening the URL in the
   // current tab, then we can reuse the ChromeFrameAutomationClient instance
   // maintained by the current ChromeActiveDocument instance. We cache this
   // instance so that it can be used by the new ChromeActiveDocument instance
   // which may be instantiated for handling the new URL.
   if (open_disposition == CURRENT_TAB) {
     // Grab a reference to ensure that the document remains valid.
     AddRef();
     g_active_doc_cache.Set(this);
   }
 
-  Base::OnOpenURL(tab_handle, url_to_open, open_disposition);
+  Base::OnOpenURL(tab_handle, url_to_open, referrer, open_disposition);
 }
 
 void ChromeActiveDocument::OnLoad(int tab_handle, const GURL& url) {
   if (ready_state_ < READYSTATE_COMPLETE) {
     ready_state_ = READYSTATE_COMPLETE;
     FireOnChanged(DISPID_READYSTATE);
   }
 }
 
 bool ChromeActiveDocument::PreProcessContextMenu(HMENU menu) {
@@ skipping 154 matching lines @@
 
     automation_client_->AttachExternalTab(external_tab_cookie);
   } else {
     // Initiate navigation before launching chrome so that the url will be
     // cached and sent with launch settings.
     if (is_new_navigation) {
       url_.Reset(::SysAllocString(url.c_str()));
       if (url_.Length()) {
         std::string utf8_url;
         WideToUTF8(url_, url_.Length(), &utf8_url);
-        if (!automation_client_->InitiateNavigation(utf8_url, is_privileged_)) {
+
+        std::string referrer;
+        Bho* chrome_frame_bho = Bho::GetCurrentThreadBhoInstance();
+        DCHECK(chrome_frame_bho != NULL);
+        if (chrome_frame_bho) {
+          referrer = chrome_frame_bho->referrer();
+        }
+        if (!automation_client_->InitiateNavigation(utf8_url,
+                                                    referrer,
+                                                    is_privileged_)) {
           DLOG(ERROR) << "Invalid URL: " << url;
           Error(L"Invalid URL");
           url_.Reset();
           return false;
         }
 
         DLOG(INFO) << "Url is " << url_;
       }
     }
   }
@@ skipping 59 matching lines @@
   ScopedComPtr<IBrowserService> browser_service;
   DoQueryService(SID_SShellBrowser, m_spClientSite, browser_service.Receive());
   if (browser_service) {
     ScopedComPtr<ITravelLog> travel_log;
     browser_service->GetTravelLog(travel_log.Receive());
     if (travel_log) {
       travel_log->Travel(browser_service, offset);
     }
   }
 }