[MD Settings] PROOF OF CONCEPT - DO NOT LAND - URL rewrite

components/sessions/content/content_serialized_navigation_driver.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 "components/sessions/content/content_serialized_navigation_driver.h"
 
 #include "base/memory/singleton.h"
 #include "build/build_config.h"
 #include "components/sessions/core/serialized_navigation_entry.h"
+#include "content/public/browser/browser_url_handler.h"
 #include "content/public/common/page_state.h"
 #include "content/public/common/referrer.h"
 #include "content/public/common/url_constants.h"
 
 namespace sessions {
 
 namespace {
 const int kObsoleteReferrerPolicyAlways = 0;
 const int kObsoleteReferrerPolicyDefault = 1;
 const int kObsoleteReferrerPolicyNever = 2;
@@ skipping 86 matching lines @@
   // 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) {
     navigation->referrer_url_ = GURL();
     navigation->referrer_policy_ = GetDefaultReferrerPolicy();
     navigation->encoded_page_state_ =
         StripReferrerFromPageState(navigation->encoded_page_state_);
   }
 
+  // Rewrite any WebUI links that do not have an appropriate handler any more.
+  if (navigation->virtual_url_.SchemeIs(content::kChromeUIScheme)) {
+    // I'd like to check for "is handled by WebUI" here, so we don't
+    // unnecessarily rewrite - but oh well. It's not accessible.
+    // content::WebUI::TypeID type_id =
+    //     content::WebUIControllerFactoryRegistry::GetInstance()->GetWebUIType(
+    //         nullptr, navigation->original_request_url_);
+    // if (type_id != content::WebUI::kNoWebUI) {
+    // in place of that, let's check if the original URL is a chrome:// URL as
+    // well.
+    if (navigation->original_request_url_.SchemeIs(content::kChromeUIScheme)) {
+      bool reverse_on_redirect;
+      navigation->original_request_url_ = navigation->virtual_url_;
+      content::BrowserURLHandler::GetInstance()->RewriteURLIfNecessary(
+          &navigation->original_request_url_, nullptr, &reverse_on_redirect);
+      navigation->encoded_page_state_ =
+          content::PageState::CreateFromURL(navigation->original_request_url_)
+              .ToEncodedData();
+    }
+  }
+
 #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() == "welcome" ||
        navigation->virtual_url_.host() == "newtab")) {
     navigation->virtual_url_ = GURL("chrome-native://newtab/");
     navigation->original_request_url_ = navigation->virtual_url_;
     navigation->encoded_page_state_ = content::PageState::CreateFromURL(
         navigation->virtual_url_).ToEncodedData();
   }
 #endif  // defined(OS_ANDROID)
 }
 
 std::string ContentSerializedNavigationDriver::StripReferrerFromPageState(
       const std::string& page_state) const {
   return content::PageState::CreateFromEncodedData(page_state)
       .RemoveReferrer()
       .ToEncodedData();
 }
 
 }  // namespace sessions