Enable Mojo to download from file URLs

mojo/shell/url_request_context_getter.cc

 // Copyright 2013 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 "mojo/shell/url_request_context_getter.h"
 
 #include "net/cookies/cookie_monster.h"
 #include "net/http/http_cache.h"
 #include "net/http/http_network_session.h"
 #include "net/http/http_server_properties_impl.h"
 #include "net/proxy/proxy_service.h"
 #include "net/ssl/ssl_config_service_defaults.h"
+#include "net/url_request/file_protocol_handler.h"
 #include "net/url_request/static_http_user_agent_settings.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_job_factory_impl.h"
 
 namespace mojo {
 namespace shell {
 
 URLRequestContextGetter::URLRequestContextGetter(
     base::FilePath base_path,
     base::SingleThreadTaskRunner* network_task_runner,
-    base::MessageLoopProxy* cache_task_runner)
+    base::SingleThreadTaskRunner* file_task_runner,
+    base::MessageLoopProxy* cache_task_runner,
+    scoped_ptr<net::NetworkDelegate> network_delegate)
     : base_path_(base_path),
+      file_task_runner_(file_task_runner),
       network_task_runner_(network_task_runner),
       cache_task_runner_(cache_task_runner),
+      network_delegate_(network_delegate.Pass()),
       net_log_(new net::NetLog()) {
 }
 
 URLRequestContextGetter::~URLRequestContextGetter() {
 }
 
 net::URLRequestContext* URLRequestContextGetter::GetURLRequestContext() {
   if (!url_request_context_) {
     url_request_context_.reset(new net::URLRequestContext());
     url_request_context_->set_net_log(net_log_.get());
+    url_request_context_->set_network_delegate(network_delegate_.get());
 
     storage_.reset(
         new net::URLRequestContextStorage(url_request_context_.get()));
 
     storage_->set_cookie_store(new net::CookieMonster(NULL, NULL));
     storage_->set_http_user_agent_settings(
         new net::StaticHttpUserAgentSettings("en-us,en", "Mojo/0.1"));
 
     storage_->set_proxy_service(net::ProxyService::CreateDirect());
     storage_->set_ssl_config_service(new net::SSLConfigServiceDefaults);
@@ skipping 24 matching lines @@
             cache_path,
             0,
             cache_task_runner_.get());
 
     net::HttpCache* main_cache = new net::HttpCache(
         network_session_params, main_backend);
     storage_->set_http_transaction_factory(main_cache);
 
     scoped_ptr<net::URLRequestJobFactoryImpl> job_factory(
         new net::URLRequestJobFactoryImpl());
-
+    job_factory->SetProtocolHandler(
+        "file",
+        new net::FileProtocolHandler(file_task_runner_));
     storage_->set_job_factory(job_factory.release());
   }
 
   return url_request_context_.get();
 }
 
 scoped_refptr<base::SingleThreadTaskRunner>
 URLRequestContextGetter::GetNetworkTaskRunner() const {
   return network_task_runner_;
 }
 
 }  // namespace shell
 }  // namespace mojo