PlzNavigate: fix issue preventing navigations to WebUIs

content/browser/webui/web_ui_impl.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 "content/browser/webui/web_ui_impl.h"
 
 #include <stddef.h>
 
 #include "base/debug/dump_without_crashing.h"
 #include "base/json/json_writer.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "content/browser/child_process_security_policy_impl.h"
 #include "content/browser/renderer_host/dip_util.h"
 #include "content/browser/renderer_host/render_process_host_impl.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/browser/web_contents/web_contents_view.h"
 #include "content/browser/webui/web_ui_controller_factory_registry.h"
 #include "content/common/view_messages.h"
+#include "content/public/browser/browser_context.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/web_ui_controller.h"
 #include "content/public/browser/web_ui_message_handler.h"
 #include "content/public/common/bindings_policy.h"
 #include "content/public/common/content_client.h"
+#include "content/public/common/url_constants.h"
 
 namespace content {
 
 const WebUI::TypeID WebUI::kNoWebUI = NULL;
 
 // static
 base::string16 WebUI::GetJavascriptCall(
     const std::string& function_name,
     const std::vector<const base::Value*>& arg_list) {
   base::string16 parameters;
   std::string json;
   for (size_t i = 0; i < arg_list.size(); ++i) {
     if (i > 0)
       parameters += base::char16(',');
 
     base::JSONWriter::Write(*arg_list[i], &json);
     parameters += base::UTF8ToUTF16(json);
   }
   return base::ASCIIToUTF16(function_name) +
       base::char16('(') + parameters + base::char16(')') + base::char16(';');
 }
 
+// static
+bool WebUIImpl::RenderProcessAllowedForURL(RenderProcessHost* process,
+                                           const GURL& url) {
+  bool is_web_ui_scheme =
+      url.SchemeIs(kChromeDevToolsScheme) || url.SchemeIs(kChromeUIScheme);
+  std::vector<std::string> additional_schemes;
+  GetContentClient()->browser()->GetAdditionalWebUISchemes(&additional_schemes);
+  if (std::find(additional_schemes.begin(), additional_schemes.end(),
+                url.scheme()) != additional_schemes.end()) {
+    is_web_ui_scheme = true;
+  }
+
+  if (!is_web_ui_scheme)
+    return true;
+
+  if (!process)
+    return false;
+
+  StoragePartition* partition = BrowserContext::GetStoragePartitionForSite(
+      process->GetBrowserContext(), url);
+  return partition == process->GetStoragePartition();
+}
+
 WebUIImpl::WebUIImpl(WebContents* contents, const std::string& frame_name)
     : link_transition_type_(ui::PAGE_TRANSITION_LINK),
       bindings_(BINDINGS_POLICY_WEB_UI),
       web_contents_(contents),
       frame_name_(frame_name) {
   DCHECK(contents);
 }
 
 WebUIImpl::~WebUIImpl() {
   // Delete the controller first, since it may also be keeping a pointer to some
@@ skipping 210 matching lines @@
 }
 
 void WebUIImpl::AddToSetIfFrameNameMatches(
     std::set<RenderFrameHost*>* frame_set,
     RenderFrameHost* host) {
   if (host->GetFrameName() == frame_name_)
     frame_set->insert(host);
 }
 
 }  // namespace content