Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1509)

Unified Diff: chrome/browser/renderer_host/download_resource_handler.cc

Issue 7664005: Move DownloadResourceHandler to content. (Closed) Base URL: svn://chrome-svn/chrome/trunk/src/
Patch Set: '' Created 9 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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());
-}

Powered by Google App Engine
This is Rietveld 408576698