Move chrome-specific SerializedNavigation code to chrome/.

chrome/browser/sessions/session_common_utils.cc

 // Copyright 2014 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/sessions/session_common_utils.h"
 
 #include <algorithm>
 #include <string>
 
+#include "build/build_config.h"
 #include "chrome/common/url_constants.h"
+#include "components/sessions/core/serialized_navigation_driver.h"
 #include "components/sessions/core/session_types.h"
+#include "content/public/common/referrer.h"
 #include "url/gurl.h"
 
+#if defined(OS_ANDROID)
+#include "content/public/common/content_features.h"
+#include "content/public/common/page_state.h"
+#endif
+
+namespace {
+
+bool IsUberOrUberReplacementURL(const GURL& url) {
+  return url.SchemeIs(content::kChromeUIScheme) &&
+         (url.host_piece() == chrome::kChromeUIHistoryHost ||
+          url.host_piece() == chrome::kChromeUIUberHost);
+}
+
+}  // namespace
+
 bool ShouldTrackURLForRestore(const GURL& url) {
   return url.is_valid() &&
          !(url.SchemeIs(content::kChromeUIScheme) &&
            (url.host_piece() == chrome::kChromeUIQuitHost ||
             url.host_piece() == chrome::kChromeUIRestartHost));
 }
 
 int GetNavigationIndexToSelect(const sessions::SessionTab& tab) {
   DCHECK(!tab.navigations.empty());
   const int selected_index =
       std::max(0, std::min(tab.current_navigation_index,
                            static_cast<int>(tab.navigations.size() - 1)));
 
   // After user sign out, Chrome may navigate to the setting page from the
   // sign out page asynchronously. The browser may be closed before the
   // navigation callback finished.
   std::string setting_page_url = std::string(chrome::kChromeUISettingsURL);
   std::string sign_out_page_url =
       setting_page_url + std::string(chrome::kSignOutSubPage);
   if (selected_index > 0 &&
       tab.navigations[selected_index].virtual_url().spec() ==
           sign_out_page_url &&
       tab.navigations[selected_index - 1].virtual_url().spec() ==
           setting_page_url) {
     return selected_index - 1;
   }
 
   return selected_index;
 }
+
+void SanitizeNavigation(sessions::SerializedNavigationEntry* navigation) {
+  content::Referrer old_referrer(
+      navigation->referrer_url(),
+      static_cast<blink::WebReferrerPolicy>(navigation->referrer_policy()));
+  content::Referrer new_referrer = content::Referrer::SanitizeForRequest(
+      navigation->virtual_url(), old_referrer);
+
+  // Clear any Uber UI page state so that these pages are reloaded rather than
+  // restored from page state. This fixes session restore when WebUI URLs
+  // change.
+  if (IsUberOrUberReplacementURL(navigation->virtual_url()) &&
+      IsUberOrUberReplacementURL(navigation->original_request_url())) {
+    navigation->set_encoded_page_state(std::string());
+  }
+
+  // No need to compare the policy, as it doesn't change during
+  // sanitization. If there has been a change, the referrer needs to be
+  // stripped from the page state as well.
+  if (navigation->referrer_url() != new_referrer.url) {
+    auto* driver = sessions::SerializedNavigationDriver::Get();
+    navigation->set_referrer_url(GURL());
+    navigation->set_referrer_policy(driver->GetDefaultReferrerPolicy());
+    navigation->set_encoded_page_state(
+        driver->StripReferrerFromPageState(navigation->encoded_page_state()));
+  }
+
+#if defined(OS_ANDROID)
+  // Rewrite the old new tab and welcome page URLs to the new NTP URL.
+  if (navigation->virtual_url().SchemeIs(content::kChromeUIScheme) &&
+      (navigation->virtual_url().host_piece() == chrome::kChromeUIWelcomeHost ||
+       navigation->virtual_url().host_piece() == chrome::kChromeUINewTabHost)) {
+    navigation->set_virtual_url(GURL(chrome::kChromeUINativeNewTabURL));
+    navigation->set_original_request_url(navigation->virtual_url());
+    navigation->set_encoded_page_state(
+        content::PageState::CreateFromURL(navigation->virtual_url())
+            .ToEncodedData());
+  }
+
+  if (base::FeatureList::IsEnabled(features::kNativeAndroidHistoryManager) &&
+      navigation->virtual_url().SchemeIs(content::kChromeUIScheme) &&
+      (navigation->virtual_url().host_piece() == chrome::kChromeUIHistoryHost ||
+       navigation->virtual_url().host_piece() ==
+           chrome::kChromeUIHistoryFrameHost)) {
+    // Rewrite the old history Web UI to the new android native history.
+    navigation->set_virtual_url(GURL(chrome::kChromeUINativeHistoryURL));
+    navigation->set_original_request_url(navigation->virtual_url());
+    navigation->set_encoded_page_state(
+        content::PageState::CreateFromURL(navigation->virtual_url())
+            .ToEncodedData());
+  } else if (!base::FeatureList::IsEnabled(
+                 features::kNativeAndroidHistoryManager) &&
+             navigation->virtual_url().SchemeIs(
+                 chrome::kChromeUINativeScheme) &&
+             navigation->virtual_url().host_piece() ==
+                 chrome::kChromeUIHistoryHost) {
+    // If the android native history UI has been disabled, redirect
+    // chrome-native://history to the old web UI.
+    navigation->set_virtual_url(GURL(chrome::kChromeUIHistoryURL));
+    navigation->set_original_request_url(navigation->virtual_url());
+    navigation->set_encoded_page_state(std::string());
+  }
+#endif  // defined(OS_ANDROID)
+}