Add support in the network service for different contexts.

content/browser/webui/web_ui_url_loader_factory.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/webui/web_ui_url_loader_factory.h"
 
 #include <map>
 
 #include "base/bind.h"
 #include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/strings/string_piece.h"
 #include "content/browser/blob_storage/blob_internals_url_loader.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/render_frame_host_impl.h"
 #include "content/browser/histogram_internals_url_loader.h"
 #include "content/browser/resource_context_impl.h"
+#include "content/browser/storage_partition_impl.h"
 #include "content/browser/webui/network_error_url_loader.h"
 #include "content/browser/webui/url_data_manager_backend.h"
 #include "content/browser/webui/url_data_source_impl.h"
 #include "content/common/network_service.mojom.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
-#include "content/public/common/service_manager_connection.h"
-#include "content/public/common/service_names.mojom.h"
 #include "content/public/common/url_constants.h"
 #include "mojo/public/cpp/bindings/binding_set.h"
-#include "services/service_manager/public/cpp/connector.h"
 #include "third_party/zlib/google/compression_utils.h"
 #include "ui/base/template_expressions.h"
 
 namespace content {
 
 namespace {
 
 class WebUIURLLoaderFactory;
 base::LazyInstance<std::map<int, std::unique_ptr<WebUIURLLoaderFactory>>>::Leaky
     g_factories = LAZY_INSTANCE_INITIALIZER;
@@ skipping 176 matching lines @@
       FROM_HERE, base::BindOnce(&URLDataSource::StartDataRequest,
                                 base::Unretained(source->source()), path,
                                 wc_getter, data_available_callback));
 }
 
 class WebUIURLLoaderFactory : public mojom::URLLoaderFactory,
                               public FrameTreeNode::Observer {
  public:
   WebUIURLLoaderFactory(FrameTreeNode* ftn)
       : frame_tree_node_id_(ftn->frame_tree_node_id()),
-        browser_context_(
-            ftn->current_frame_host()->GetProcess()->GetBrowserContext()) {
+        storage_partition_(static_cast<StoragePartitionImpl*>(
+            ftn->current_frame_host()->GetProcess()->GetStoragePartition())) {
     ftn->AddObserver(this);
   }
 
   ~WebUIURLLoaderFactory() override {}
 
   mojom::URLLoaderFactoryPtr CreateBinding() {
     return loader_factory_bindings_.CreateInterfacePtrAndBind(this);
   }
 
   // mojom::URLLoaderFactory implementation:
   void CreateLoaderAndStart(mojom::URLLoaderAssociatedRequest loader,
                             int32_t routing_id,
                             int32_t request_id,
                             uint32_t options,
                             const ResourceRequest& request,
                             mojom::URLLoaderClientPtr client) override {
     DCHECK_CURRENTLY_ON(BrowserThread::UI);
     if (request.url.host_piece() == kChromeUINetworkViewCacheHost) {
-      mojom::NetworkServicePtr network_service;
-      ServiceManagerConnection::GetForProcess()->GetConnector()->BindInterface(
-          mojom::kNetworkServiceName, &network_service);
-      network_service->HandleViewCacheRequest(request, std::move(client));
+      storage_partition_->network_context()->HandleViewCacheRequest(
+          request.url, std::move(client));
       return;
     }
 
     if (request.url.host_piece() == kChromeUIBlobInternalsHost) {
       BrowserThread::PostTask(
           BrowserThread::IO, FROM_HERE,
-          base::BindOnce(
-              &StartBlobInternalsURLLoader, request, client.PassInterface(),
-              base::Unretained(
-                  ChromeBlobStorageContext::GetFor(browser_context_))));
+          base::BindOnce(&StartBlobInternalsURLLoader, request,
+                         client.PassInterface(),
+                         base::Unretained(ChromeBlobStorageContext::GetFor(
+                             storage_partition_->browser_context()))));
       return;
     }
 
     if (request.url.host_piece() == kChromeUINetworkErrorHost ||
         request.url.host_piece() == kChromeUIDinoHost) {
       StartNetworkErrorsURLLoader(request, std::move(client));
       return;
     }
 
     if (request.url.host_piece() == kChromeUIHistogramHost) {
       StartHistogramInternalsURLLoader(request, std::move(client));
       return;
     }
 
     BrowserThread::PostTask(
         BrowserThread::IO, FROM_HERE,
-        base::BindOnce(&StartURLLoader, request, frame_tree_node_id_,
-                       client.PassInterface(),
-                       browser_context_->GetResourceContext()));
+        base::BindOnce(
+            &StartURLLoader, request, frame_tree_node_id_,
+            client.PassInterface(),
+            storage_partition_->browser_context()->GetResourceContext()));
   }
 
   void SyncLoad(int32_t routing_id,
                 int32_t request_id,
                 const ResourceRequest& request,
                 SyncLoadCallback callback) override {
     NOTREACHED();
   }
 
   // FrameTreeNode::Observer implementation:
   void OnFrameTreeNodeDestroyed(FrameTreeNode* node) override {
     g_factories.Get().erase(frame_tree_node_id_);
   }
 
  private:
   int frame_tree_node_id_;
-  BrowserContext* browser_context_;
+  StoragePartitionImpl* storage_partition_;
   mojo::BindingSet<mojom::URLLoaderFactory> loader_factory_bindings_;
 
   DISALLOW_COPY_AND_ASSIGN(WebUIURLLoaderFactory);
 };
 
 }  // namespace
 
 mojom::URLLoaderFactoryPtr GetWebUIURLLoader(FrameTreeNode* node) {
   int ftn_id = node->frame_tree_node_id();
   if (g_factories.Get()[ftn_id].get() == nullptr)
     g_factories.Get()[ftn_id] = base::MakeUnique<WebUIURLLoaderFactory>(node);
   return g_factories.Get()[ftn_id]->CreateBinding();
 }
 
 }  // namespace content