| Index: chrome/browser/renderer_host/download_resource_handler.cc
|
| ===================================================================
|
| --- chrome/browser/renderer_host/download_resource_handler.cc (revision 96829)
|
| +++ chrome/browser/renderer_host/download_resource_handler.cc (working copy)
|
| @@ -1,265 +0,0 @@
|
| -// 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.
|
| -
|
| -#include "chrome/browser/renderer_host/download_resource_handler.h"
|
| -
|
| -#include <string>
|
| -
|
| -#include "base/logging.h"
|
| -#include "base/metrics/histogram.h"
|
| -#include "base/metrics/stats_counters.h"
|
| -#include "base/stringprintf.h"
|
| -#include "chrome/browser/download/download_util.h"
|
| -#include "content/browser/browser_thread.h"
|
| -#include "content/browser/download/download_create_info.h"
|
| -#include "content/browser/download/download_file_manager.h"
|
| -#include "content/browser/download/download_item.h"
|
| -#include "content/browser/download/download_request_handle.h"
|
| -#include "content/browser/renderer_host/global_request_id.h"
|
| -#include "content/browser/renderer_host/resource_dispatcher_host.h"
|
| -#include "content/browser/renderer_host/resource_dispatcher_host_request_info.h"
|
| -#include "content/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,
|
| - const GURL& url,
|
| - DownloadFileManager* download_file_manager,
|
| - net::URLRequest* request,
|
| - bool save_as,
|
| - const DownloadSaveInfo& save_info)
|
| - : download_id_(-1),
|
| - global_id_(render_process_host_id, request_id),
|
| - render_view_id_(render_view_id),
|
| - content_length_(0),
|
| - download_file_manager_(download_file_manager),
|
| - request_(request),
|
| - save_as_(save_as),
|
| - save_info_(save_info),
|
| - buffer_(new DownloadBuffer),
|
| - rdh_(rdh),
|
| - is_paused_(false) {
|
| - download_util::RecordDownloadCount(download_util::UNTHROTTLED_COUNT);
|
| -}
|
| -
|
| -bool DownloadResourceHandler::OnUploadProgress(int request_id,
|
| - uint64 position,
|
| - uint64 size) {
|
| - return true;
|
| -}
|
| -
|
| -// Not needed, as this event handler ought to be the final resource.
|
| -bool DownloadResourceHandler::OnRequestRedirected(int request_id,
|
| - const GURL& url,
|
| - ResourceResponse* response,
|
| - bool* defer) {
|
| - return true;
|
| -}
|
| -
|
| -// Send the download creation information to the download thread.
|
| -bool DownloadResourceHandler::OnResponseStarted(int request_id,
|
| - ResourceResponse* response) {
|
| - VLOG(20) << __FUNCTION__ << "()" << DebugString()
|
| - << " request_id = " << request_id;
|
| - download_start_time_ = base::TimeTicks::Now();
|
| - std::string content_disposition;
|
| - request_->GetResponseHeaderByName("content-disposition",
|
| - &content_disposition);
|
| - set_content_disposition(content_disposition);
|
| - set_content_length(response->response_head.content_length);
|
| -
|
| - const ResourceDispatcherHostRequestInfo* request_info =
|
| - ResourceDispatcherHost::InfoForRequest(request_);
|
| -
|
| - download_id_ = download_file_manager_->GetNextId();
|
| -
|
| - // Deleted in DownloadManager.
|
| - DownloadCreateInfo* info = new DownloadCreateInfo;
|
| - info->url_chain = request_->url_chain();
|
| - info->referrer_url = GURL(request_->referrer());
|
| - info->start_time = base::Time::Now();
|
| - info->received_bytes = 0;
|
| - info->total_bytes = content_length_;
|
| - info->state = DownloadItem::IN_PROGRESS;
|
| - info->download_id = download_id_;
|
| - info->has_user_gesture = request_info->has_user_gesture();
|
| - info->request_handle = DownloadRequestHandle(rdh_,
|
| - global_id_.child_id,
|
| - render_view_id_,
|
| - global_id_.request_id);
|
| - info->content_disposition = content_disposition_;
|
| - info->mime_type = response->response_head.mime_type;
|
| - download_util::RecordDownloadMimeType(info->mime_type);
|
| - // TODO(ahendrickson) -- Get the last modified time and etag, so we can
|
| - // resume downloading.
|
| -
|
| - 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->referrer_charset = request_->context()->referrer_charset();
|
| - info->save_info = save_info_;
|
| - BrowserThread::PostTask(
|
| - BrowserThread::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;
|
| -}
|
| -
|
| -bool DownloadResourceHandler::OnWillStart(int request_id,
|
| - const GURL& url,
|
| - bool* defer) {
|
| - return true;
|
| -}
|
| -
|
| -// Create a new buffer, which will be handed to the download thread for file
|
| -// writing and deletion.
|
| -bool DownloadResourceHandler::OnWillRead(int request_id, net::IOBuffer** buf,
|
| - int* buf_size, int min_size) {
|
| - DCHECK(buf && buf_size);
|
| - if (!read_buffer_) {
|
| - *buf_size = min_size < 0 ? kReadBufSize : min_size;
|
| - read_buffer_ = new net::IOBuffer(*buf_size);
|
| - }
|
| - *buf = read_buffer_.get();
|
| - return true;
|
| -}
|
| -
|
| -// Pass the buffer to the download file writer.
|
| -bool DownloadResourceHandler::OnReadCompleted(int request_id, int* bytes_read) {
|
| - if (!*bytes_read)
|
| - return true;
|
| - DCHECK(read_buffer_);
|
| - base::AutoLock auto_lock(buffer_->lock);
|
| - bool need_update = buffer_->contents.empty();
|
| -
|
| - // We are passing ownership of this buffer to the download file manager.
|
| - net::IOBuffer* buffer = NULL;
|
| - read_buffer_.swap(&buffer);
|
| - buffer_->contents.push_back(std::make_pair(buffer, *bytes_read));
|
| - if (need_update) {
|
| - BrowserThread::PostTask(
|
| - BrowserThread::FILE, FROM_HERE,
|
| - NewRunnableMethod(download_file_manager_,
|
| - &DownloadFileManager::UpdateDownload,
|
| - download_id_,
|
| - buffer_.get()));
|
| - }
|
| -
|
| - // We schedule a pause outside of the read loop if there is too much file
|
| - // writing work to do.
|
| - if (buffer_->contents.size() > kLoadsToWrite)
|
| - StartPauseTimer();
|
| -
|
| - return true;
|
| -}
|
| -
|
| -bool DownloadResourceHandler::OnResponseCompleted(
|
| - int request_id,
|
| - const net::URLRequestStatus& status,
|
| - const std::string& security_info) {
|
| - VLOG(20) << __FUNCTION__ << "()" << DebugString()
|
| - << " request_id = " << request_id
|
| - << " status.status() = " << status.status()
|
| - << " status.os_error() = " << status.os_error();
|
| - int error_code = (status.status() == net::URLRequestStatus::FAILED) ?
|
| - status.os_error() : 0;
|
| - // We transfer ownership to |DownloadFileManager| to delete |buffer_|,
|
| - // so that any functions queued up on the FILE thread are executed
|
| - // before deletion.
|
| - BrowserThread::PostTask(
|
| - BrowserThread::FILE, FROM_HERE,
|
| - NewRunnableMethod(download_file_manager_,
|
| - &DownloadFileManager::OnResponseCompleted,
|
| - download_id_,
|
| - buffer_.release(),
|
| - error_code,
|
| - security_info));
|
| - read_buffer_ = NULL;
|
| - return true;
|
| -}
|
| -
|
| -void DownloadResourceHandler::OnRequestClosed() {
|
| - UMA_HISTOGRAM_TIMES("SB2.DownloadDuration",
|
| - base::TimeTicks::Now() - download_start_time_);
|
| -}
|
| -
|
| -// If the content-length header is not present (or contains something other
|
| -// than numbers), the incoming content_length is -1 (unknown size).
|
| -// Set the content length to 0 to indicate unknown size to DownloadManager.
|
| -void DownloadResourceHandler::set_content_length(const int64& content_length) {
|
| - content_length_ = 0;
|
| - if (content_length > 0)
|
| - content_length_ = content_length;
|
| -}
|
| -
|
| -void DownloadResourceHandler::set_content_disposition(
|
| - const std::string& content_disposition) {
|
| - content_disposition_ = content_disposition;
|
| -}
|
| -
|
| -void DownloadResourceHandler::CheckWriteProgress() {
|
| - if (!buffer_.get())
|
| - return; // The download completed while we were waiting to run.
|
| -
|
| - size_t contents_size;
|
| - {
|
| - base::AutoLock lock(buffer_->lock);
|
| - contents_size = buffer_->contents.size();
|
| - }
|
| -
|
| - bool should_pause = contents_size > kLoadsToWrite;
|
| -
|
| - // We'll come back later and see if it's okay to unpause the request.
|
| - if (should_pause)
|
| - StartPauseTimer();
|
| -
|
| - if (is_paused_ != should_pause) {
|
| - rdh_->PauseRequest(global_id_.child_id, global_id_.request_id,
|
| - should_pause);
|
| - is_paused_ = should_pause;
|
| - }
|
| -}
|
| -
|
| -DownloadResourceHandler::~DownloadResourceHandler() {
|
| -}
|
| -
|
| -void DownloadResourceHandler::StartPauseTimer() {
|
| - if (!pause_timer_.IsRunning())
|
| - pause_timer_.Start(base::TimeDelta::FromMilliseconds(kThrottleTimeMs), this,
|
| - &DownloadResourceHandler::CheckWriteProgress);
|
| -}
|
| -
|
| -std::string DownloadResourceHandler::DebugString() const {
|
| - return base::StringPrintf("{"
|
| - " url_ = " "\"%s\""
|
| - " download_id_ = " "%d"
|
| - " global_id_ = {"
|
| - " child_id = " "%d"
|
| - " request_id = " "%d"
|
| - " }"
|
| - " render_view_id_ = " "%d"
|
| - " save_info_.file_path = \"%" PRFilePath "\""
|
| - " }",
|
| - request_->url().spec().c_str(),
|
| - download_id_,
|
| - global_id_.child_id,
|
| - global_id_.request_id,
|
| - render_view_id_,
|
| - save_info_.file_path.value().c_str());
|
| -}
|
|
|