Preliminary work on resuming downloads whose connections have expired.

chrome/browser/renderer_host/download_resource_handler.cc

 // Copyright (c) 2006-2008 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 "chrome/browser/renderer_host/download_resource_handler.h"
 
+#include "base/file_util.h"
 #include "base/logging.h"
 #include "chrome/browser/chrome_thread.h"
 #include "chrome/browser/download/download_item.h"
 #include "chrome/browser/download/download_file_manager.h"
+#include "chrome/browser/download/download_util.h"
 #include "chrome/browser/history/download_create_info.h"
 #include "chrome/browser/renderer_host/global_request_id.h"
 #include "chrome/browser/renderer_host/resource_dispatcher_host.h"
+#include "chrome/browser/renderer_host/resource_dispatcher_host_request_info.h"
 #include "chrome/common/resource_response.h"
 #include "net/base/io_buffer.h"
 #include "net/http/http_response_headers.h"
 #include "net/url_request/url_request_context.h"
 
 DownloadResourceHandler::DownloadResourceHandler(
     ResourceDispatcherHost* rdh,
     int render_process_host_id,
     int render_view_id,
     int request_id,
@@ skipping 33 matching lines @@
 
 // Send the download creation information to the download thread.
 bool DownloadResourceHandler::OnResponseStarted(int request_id,
                                                 ResourceResponse* response) {
   std::string content_disposition;
   request_->GetResponseHeaderByName("content-disposition",
                                     &content_disposition);
   set_content_disposition(content_disposition);
   set_content_length(response->response_head.content_length);
 
-  download_id_ = download_file_manager_->GetNextId();
+  ResourceDispatcherHostRequestInfo* req_info =
+      ResourceDispatcherHost::InfoForRequest(request_);
+  download_id_ = -1;
+  if (req_info->was_interrupted()) {
+    FilePath orig_path = save_info_.file_path;
+    download_id_ = download_file_manager_->GetDownloadId(url_, orig_path);
+    if (download_id_ == -1) {
+      orig_path = download_util::RemoveCrDownloadPath(save_info_.file_path);
+      download_id_ = download_file_manager_->GetDownloadId(url_, orig_path);
+    }
+  }
+  if (download_id_ == -1)
+    download_id_ = download_file_manager_->GetNextId();
   // |download_file_manager_| consumes (deletes):
   DownloadCreateInfo* info = new DownloadCreateInfo;
   info->url = url_;
   info->referrer_url = GURL(request_->referrer());
   info->start_time = base::Time::Now();
-  info->received_bytes = 0;
+  info->received_bytes =
+      req_info->was_interrupted() ? req_info->interrupted_bytes() : 0;
   info->total_bytes = content_length_;
   info->state = DownloadItem::IN_PROGRESS;
   info->download_id = download_id_;
   info->child_id = global_id_.child_id;
   info->render_view_id = render_view_id_;
   info->request_id = global_id_.request_id;
   info->content_disposition = content_disposition_;
   info->mime_type = response->response_head.mime_type;
+  info->is_partial_download = req_info->was_interrupted();
+
+  // Handle server's response codes.
+  int status = request_->response_headers()->response_code();
+  switch (status) {
+    // Continue with download:
+    case 200:   // OK
+      // Downloading the full file, even if we asked for a range.
+      info->received_bytes = 0;
+      if (info->is_partial_download) {
+        // Delete the crdownload file explicitly.
+        FilePath crpath =
+            download_util::GetCrDownloadPath(save_info_.file_path);
+        file_util::Delete(crpath, false);
+      }
+      // Fall through.
+
+    case 206:   // Partial content.  Leave alone.
+      break;
+
+    case 412:   // Precondition failed.  Fails 'If-Unmodified-Since'.
+    case 204:   // No content.  File not present.
+    default:    // All other errors.
+      // TODO(ahendrickson) -- Cancel, and notify user.
+      break;
+
+    // Retry by downloading from the start automatically:
+    case 416:   // Range not available.  Fails 'Content-Range'.
+      // TODO(ahendrickson) -- Submit another request for the whole file.
+      break;
+  }
 
   std::string content_type_header;
   if (!response->response_head.headers ||
       !response->response_head.headers->GetMimeType(&content_type_header))
     content_type_header = "";
   info->original_mime_type = content_type_header;
 
   info->prompt_user_for_save_location =
       save_as_ && save_info_.file_path.empty();
   info->is_dangerous = false;
   info->referrer_charset = request_->context()->referrer_charset();
   info->save_info = save_info_;
   ChromeThread::PostTask(
-      ChromeThread::UI, FROM_HERE,
+      ChromeThread::UI,
+      FROM_HERE,
       NewRunnableMethod(
           download_file_manager_, &DownloadFileManager::StartDownload, info));
 
   // We can't start saving the data before we create the file on disk.
   // The request will be un-paused in DownloadFileManager::CreateDownloadFile.
   rdh_->PauseRequest(global_id_.child_id, global_id_.request_id, true);
 
   return true;
 }
 
@@ skipping 42 matching lines @@
   if (buffer_->contents.size() > kLoadsToWrite)
     StartPauseTimer();
 
   return true;
 }
 
 bool DownloadResourceHandler::OnResponseCompleted(
     int request_id,
     const URLRequestStatus& status,
     const std::string& security_info) {
+  int error_code =
+      (status.status() == URLRequestStatus::FAILED) ? status.os_error() : 0;
   ChromeThread::PostTask(
-      ChromeThread::FILE, FROM_HERE,
+      ChromeThread::FILE,
+      FROM_HERE,
       NewRunnableMethod(download_file_manager_,
                         &DownloadFileManager::OnResponseCompleted,
                         download_id_,
-                        buffer_));
+                        buffer_,
+                        error_code,
+                        security_info));
   read_buffer_ = NULL;
 
   // 'buffer_' is deleted by the DownloadFileManager.
   buffer_ = NULL;
   return true;
 }
 
 void DownloadResourceHandler::OnRequestClosed() {
 }
 
@@ skipping 35 matching lines @@
 }
 
 DownloadResourceHandler::~DownloadResourceHandler() {
 }
 
 void DownloadResourceHandler::StartPauseTimer() {
   if (!pause_timer_.IsRunning())
     pause_timer_.Start(base::TimeDelta::FromMilliseconds(kThrottleTimeMs), this,
                        &DownloadResourceHandler::CheckWriteProgress);
 }