chrome.experimental.downloads

content/browser/renderer_host/resource_dispatcher_host.cc

 // Copyright (c) 2011 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.
 
 // See http://dev.chromium.org/developers/design-documents/multi-process-resource-loading
 
 #include "content/browser/renderer_host/resource_dispatcher_host.h"
 
 #include <set>
+#include <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/metrics/histogram.h"
 #include "base/shared_memory.h"
 #include "base/stl_util-inl.h"
 #include "base/time.h"
+#include "base/values.h"
 #include "chrome/browser/download/download_file_manager.h"
 #include "chrome/browser/download/download_manager.h"
 #include "chrome/browser/download/download_request_limiter.h"
 #include "chrome/browser/download/download_util.h"
 #include "chrome/browser/download/save_file_manager.h"
 #include "chrome/browser/renderer_host/download_resource_handler.h"
 #include "chrome/browser/renderer_host/save_file_resource_handler.h"
 #include "content/browser/appcache/chrome_appcache_service.h"
 #include "content/browser/cert_store.h"
 #include "content/browser/child_process_security_policy.h"
@@ skipping 641 matching lines @@
 
 // We are explicitly forcing the download of 'url'.
 void ResourceDispatcherHost::BeginDownload(
     const GURL& url,
     const GURL& referrer,
     const DownloadSaveInfo& save_info,
     bool prompt_for_save_location,
     int child_id,
     int route_id,
     const content::ResourceContext& context) {
-  if (is_shutdown_)
+  net::URLRequest* request = new net::URLRequest(url, this);
+  request->set_method("GET");
+  request->set_referrer(MaybeStripReferrer(referrer).spec());
+  BeginDownload(
+      request,
+      save_info,
+      prompt_for_save_location,
+      DownloadResourceHandler::OnStartedCallback(),
+      child_id,
+      route_id,
+      context);
+}
+
+void ResourceDispatcherHost::BeginDownload(
+    net::URLRequest* request,
+    const DownloadSaveInfo& save_info,
+    bool prompt_for_save_location,
+    DownloadResourceHandler::OnStartedCallback started_cb,
+    int child_id,
+    int route_id,
+    const content::ResourceContext& context) {
+  CHECK(request);
+  scoped_ptr<net::URLRequest> delete_request(request);
+  if (is_shutdown_) {
     return;
+  }
+  const GURL& url = request->original_url();
+  request->set_referrer(MaybeStripReferrer(GURL(request->referrer())).spec());
 
   // Check if the renderer is permitted to request the requested URL.
   if (!ChildProcessSecurityPolicy::GetInstance()->
           CanRequestURL(child_id, url)) {
     VLOG(1) << "Denied unauthorized download request for "
             << url.possibly_invalid_spec();
     return;
   }
 
-  BrowserThread::PostTask(
-      BrowserThread::UI, FROM_HERE,
-      NewRunnableFunction(&download_util::NotifyDownloadInitiated,
-                          child_id, route_id));
-
-  net::URLRequest* request = new net::URLRequest(url, this);
+  BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, NewRunnableFunction(
+      &download_util::NotifyDownloadInitiated, child_id, route_id));
 
   request_id_--;
 
   scoped_refptr<ResourceHandler> handler(
       new DownloadResourceHandler(this,
                                   child_id,
                                   route_id,
                                   request_id_,
                                   url,
                                   download_file_manager_.get(),
-                                  request,
+                                  delete_request.release(),
                                   prompt_for_save_location,
+                                  started_cb,
                                   save_info));
 
-  if (delegate_)
+  if (delegate_) {
     handler = delegate_->DownloadStarting(handler, child_id, route_id);
+  }
 
   const net::URLRequestContext* request_context = context.request_context();
-
   if (!request_context->job_factory()->IsHandledURL(url)) {
     VLOG(1) << "Download request for unsupported protocol: "
             << url.possibly_invalid_spec();
     return;
   }
 
-  request->set_method("GET");
-  request->set_referrer(MaybeStripReferrer(referrer).spec());
   request->set_context(context.request_context());
   request->set_load_flags(request->load_flags() |
       net::LOAD_IS_DOWNLOAD);
 
   ResourceDispatcherHostRequestInfo* extra_info =
       CreateRequestInfoForBrowserRequest(
-          handler, child_id, route_id, true, context);
+          handler, child_id, route_id, true/*download*/, context);
   SetRequestInfo(request, extra_info);  // Request takes ownership.
 
   BeginRequestInternal(request);
 }
 
 // This function is only used for saving feature.
 void ResourceDispatcherHost::BeginSaveFile(
     const GURL& url,
     const GURL& referrer,
     int child_id,
@@ skipping 1261 matching lines @@
   return HTTP_AUTH_RESOURCE_BLOCKED_CROSS;
 }
 
 bool ResourceDispatcherHost::allow_cross_origin_auth_prompt() {
   return allow_cross_origin_auth_prompt_;
 }
 
 void ResourceDispatcherHost::set_allow_cross_origin_auth_prompt(bool value) {
   allow_cross_origin_auth_prompt_ = value;
 }