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 639 matching lines @@
                      params.closing_route_id,
                      &RenderViewHost::OnSwapOutACK);
 }
 
 // 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,
+    std::string method,
+    DictionaryValue* extra_headers,
+    std::string post_body,
+    DownloadResourceHandler::OnResponseStartedCallback started_cb,
+    DownloadResourceHandler::OnUnstartableCallback unstartable_cb,
     int child_id,
     int route_id,
     const content::ResourceContext& context) {
   if (is_shutdown_)
     return;
 
   // 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 "
@@ skipping 12 matching lines @@
 
   scoped_refptr<ResourceHandler> handler(
       new DownloadResourceHandler(this,
                                   child_id,
                                   route_id,
                                   request_id_,
                                   url,
                                   download_file_manager_.get(),
                                   request,
                                   prompt_for_save_location,
+                                  started_cb,
+                                  unstartable_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;
   }
+  if (method.empty()) {
+    method = "GET";
+  }
 
-  request->set_method("GET");
+  request->set_method(method);
   request->set_referrer(MaybeStripReferrer(referrer).spec());
   request->set_context(context.request_context());
   request->set_load_flags(request->load_flags() |
       net::LOAD_IS_DOWNLOAD);
 
+  if (extra_headers != NULL) {
+    DictionaryValue::key_iterator headers_end = extra_headers->end_keys();
+    for (DictionaryValue::key_iterator headers_iter =
+         extra_headers->begin_keys();
+         headers_iter != headers_end; ++headers_iter) {
+      std::string value;
+      if (extra_headers->GetStringWithoutPathExpansion(*headers_iter, &value)) {
+        request->SetExtraRequestHeaderByName(
+            *headers_iter, value, false/*overwrite*/);
+      }
+    }
+  }
+  if (!post_body.empty()) {
+    request->AppendBytesToUpload(post_body.data(), post_body.size());
+  }
+
   ResourceDispatcherHostRequestInfo* extra_info =
       CreateRequestInfoForBrowserRequest(
           handler, child_id, route_id, true, context);
   SetRequestInfo(request, extra_info);  // Request takes ownership.
 
   BeginRequestInternal(request);
 }
 
 // This function is only used for saving feature.
 void ResourceDispatcherHost::BeginSaveFile(
@@ skipping 1264 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;
 }