Make the new WebUI's code handle all webui schemes, instead of just chrome://.

content/browser/loader/navigation_url_loader_network_service.cc

 // Copyright 2017 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/loader/navigation_url_loader_network_service.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/lazy_instance.h"
 #include "base/memory/ptr_util.h"
 #include "base/trace_event/trace_event.h"
 #include "content/browser/blob_storage/chrome_blob_storage_context.h"
 #include "content/browser/frame_host/frame_tree_node.h"
 #include "content/browser/frame_host/navigation_request_info.h"
 #include "content/browser/loader/navigation_resource_handler.h"
 #include "content/browser/loader/navigation_resource_throttle.h"
 #include "content/browser/loader/navigation_url_loader_delegate.h"
+#include "content/browser/webui/url_data_manager_backend.h"
 #include "content/browser/webui/web_ui_url_loader_factory.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/global_request_id.h"
 #include "content/public/browser/navigation_data.h"
 #include "content/public/browser/navigation_ui_data.h"
 #include "content/public/browser/ssl_status.h"
 #include "content/public/browser/stream_handle.h"
 #include "content/public/common/referrer.h"
 #include "content/public/common/service_manager_connection.h"
 #include "content/public/common/service_names.mojom.h"
@@ skipping 179 matching lines @@
 
   mojom::URLLoaderFactory* factory = nullptr;
   // This |factory_ptr| will be destroyed when it goes out of scope. Because
   // |url_loader_associated_ptr_| is associated with it, it will be disconnected
   // as well. That means NavigationURLLoaderNetworkService::FollowRedirect()
   // won't work as expected, the |url_loader_associated_ptr_| will silently drop
   // calls.
   // This is fine for now since the only user of this is WebUI which doesn't
   // need this, but we'll have to fix this when other consumers come up.
   mojom::URLLoaderFactoryPtr factory_ptr;
-  if (request->url.SchemeIs(kChromeUIScheme)) {
+  const auto& schemes = URLDataManagerBackend::GetWebUISchemes();
+  if (std::find(schemes.begin(), schemes.end(), request->url.scheme()) !=
+      schemes.end()) {
     FrameTreeNode* frame_tree_node =
         FrameTreeNode::GloballyFindByID(request_info_->frame_tree_node_id);
     factory_ptr = GetWebUIURLLoader(frame_tree_node);
     factory = factory_ptr.get();
   }
 
   if (!factory)
     factory = GetURLLoaderFactory();
 
   factory->CreateLoaderAndStart(mojo::MakeRequest(&url_loader_associated_ptr_),
                                 0 /* routing_id? */, 0 /* request_id? */,
                                 mojom::kURLLoadOptionSendSSLInfo, *request,
                                 std::move(url_loader_client_ptr_to_pass));
 }
 
 mojom::URLLoaderFactory*
 NavigationURLLoaderNetworkService::GetURLLoaderFactory() {
   // TODO(yzshen): We will need the ability to customize the factory per frame
   // e.g., for appcache or service worker.
   if (!g_url_loader_factory.Get()) {
     ServiceManagerConnection::GetForProcess()->GetConnector()->BindInterface(
         mojom::kNetworkServiceName, &g_url_loader_factory.Get());
   }
 
   return g_url_loader_factory.Get().get();
 }
 
 }  // namespace content