Include referrer chain with inline install requests

chrome/browser/extensions/webstore_inline_installer.cc

 // Copyright (c) 2012 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/browser/extensions/webstore_inline_installer.h"
 
+#include <utility>
+
+#include "base/json/json_writer.h"
 #include "base/strings/stringprintf.h"
+#include "base/values.h"
+#include "chrome/browser/extensions/webstore_data_fetcher.h"
+#include "chrome/browser/history/history_service_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/exclusive_access/fullscreen_controller.h"
+#include "components/history/core/browser/history_service.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/web_contents.h"
 
 using content::WebContents;
 
 namespace extensions {
 
 const char kInvalidWebstoreResponseError[] =
     "Invalid Chrome Web Store response.";
 const char kNoVerifiedSitesError[] =
@@ skipping 67 matching lines @@
     }
   }
   if (!requestor_is_ok) {
     *error = kNotFromVerifiedSitesError;
     return false;
   }
   *error = "";
   return true;
 }
 
+void WebstoreInlineInstaller::StartWebstoreDataRequest(
+    WebstoreDataFetcher* fetcher, const history::RedirectList* redirect_list) {
+  if (redirect_list) {
+    base::DictionaryValue dictionary;
+    dictionary.SetString("id", id());
+    dictionary.SetString("referrer", requestor_url_.spec());
+    std::unique_ptr<base::ListValue> redirect_chain =
+        base::MakeUnique<base::ListValue>();
+    for (const GURL& url : *redirect_list) {
+      redirect_chain->AppendString(url.spec());
+    }
+    dictionary.Set("redirect_chain", std::move(redirect_chain));
+
+    std::unique_ptr<std::string> json = base::MakeUnique<std::string>();
+    base::JSONWriter::Write(dictionary, json.get());
+    fetcher->SetJsonPostData(std::move(json));
+  }
+  fetcher->Start();
+}
+
+void WebstoreInlineInstaller::OnBeforeWebstoreDataRequest(
+    WebstoreDataFetcher* fetcher) {
+  content::WebContents* web_contents = GetWebContents();
+  if (web_contents) {
+    history::HistoryService* history = HistoryServiceFactory::GetForProfile(
+        profile(), ServiceAccessType::EXPLICIT_ACCESS);
+    if (history) {
+      history->QueryRedirectsTo(
+          requestor_url_,
+          base::Bind(&WebstoreInlineInstaller::StartWebstoreDataRequest,
+                     this,
+                     base::Unretained(fetcher)),
+          &request_tracker_);
+      return;
+    }
+  }
+  StartWebstoreDataRequest(fetcher, nullptr);
+}
+
 bool WebstoreInlineInstaller::CheckRequestorAlive() const {
   // The frame or tab may have gone away - cancel installation in that case.
   return host_ != nullptr && web_contents() != nullptr;
 }
 
 const GURL& WebstoreInlineInstaller::GetRequestorURL() const {
   return requestor_url_;
 }
 
 std::unique_ptr<ExtensionInstallPrompt::Prompt>
@@ skipping 137 matching lines @@
     DLOG(WARNING) << "Could not parse " << verified_site_pattern_spec <<
         " as URL pattern " << parse_result;
     return false;
   }
   verified_site_pattern.SetScheme("*");
 
   return verified_site_pattern.MatchesURL(requestor_url);
 }
 
 }  // namespace extensions