Chromium Code Reviews| Index: content/browser/download/url_downloader.cc |
| diff --git a/content/browser/download/url_downloader.cc b/content/browser/download/url_downloader.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..ffaea7115086ee20f813405ac72dc133c92f7606 |
| --- /dev/null |
| +++ b/content/browser/download/url_downloader.cc |
| @@ -0,0 +1,210 @@ |
| +// Copyright 2015 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 "content/browser/download/url_downloader.h" |
| + |
| +#include "base/location.h" |
| +#include "base/thread_task_runner_handle.h" |
| +#include "content/browser/appcache/appcache_interceptor.h" |
| +#include "content/browser/download/download_manager_impl.h" |
| +#include "content/browser/service_worker/service_worker_request_handler.h" |
| +#include "content/browser/ssl/ssl_policy.h" |
| +#include "content/common/ssl_status_serialization.h" |
| +#include "content/public/browser/cert_store.h" |
| +#include "content/public/browser/download_save_info.h" |
| +#include "content/public/browser/signed_certificate_timestamp_store.h" |
| +#include "content/public/common/process_type.h" |
| +#include "content/public/common/security_style.h" |
| +#include "net/base/io_buffer.h" |
| +#include "net/base/load_flags.h" |
| +#include "net/base/net_errors.h" |
| +#include "net/http/http_response_headers.h" |
| +#include "net/http/http_status_code.h" |
| +#include "ui/base/page_transition_types.h" |
| + |
| +namespace content { |
| + |
| +// static |
| +scoped_ptr<UrlDownloader> UrlDownloader::BeginDownload( |
| + base::WeakPtr<DownloadManagerImpl> download_manager, |
| + scoped_ptr<net::URLRequest> request, |
| + const Referrer& referrer, |
| + bool is_content_initiated, |
| + bool prefer_cache, |
| + bool do_not_prompt_for_login, |
| + scoped_ptr<DownloadSaveInfo> save_info, |
| + uint32 download_id, |
| + const DownloadUrlParameters::OnStartedCallback& started_callback) { |
| + if (!referrer.url.is_valid()) |
| + request->SetReferrer(std::string()); |
| + else |
| + request->SetReferrer(referrer.url.spec()); |
|
jochen (gone - plz use gerrit)
2017/01/02 08:08:00
this is broken. compare with https://cs.chromium.o
|
| + |
| + int extra_load_flags = net::LOAD_NORMAL; |
| + if (prefer_cache) { |
| + // If there is upload data attached, only retrieve from cache because there |
| + // is no current mechanism to prompt the user for their consent for a |
| + // re-post. For GETs, try to retrieve data from the cache and skip |
| + // validating the entry if present. |
| + if (request->get_upload() != NULL) |
| + extra_load_flags |= net::LOAD_ONLY_FROM_CACHE; |
| + else |
| + extra_load_flags |= net::LOAD_PREFERRING_CACHE; |
| + } else { |
| + extra_load_flags |= net::LOAD_DISABLE_CACHE; |
| + } |
| + request->SetLoadFlags(request->load_flags() | extra_load_flags); |
| + |
| + if (request->url().SchemeIs(url::kBlobScheme)) |
| + return nullptr; |
| + |
| + scoped_ptr<DownloadRequestCore> handler( |
| + new DownloadRequestCore(download_id, request.get(), started_callback, |
| + save_info.Pass(), download_manager)); |
| + |
| + // From this point forward, the |UrlDownloader| is responsible for |
| + // |started_callback|. |
| + scoped_ptr<UrlDownloader> downloader( |
| + new UrlDownloader(request.Pass(), handler.Pass(), download_manager)); |
| + |
| + downloader->Start(); |
| + |
| + return downloader; |
| +} |
| + |
| +UrlDownloader::UrlDownloader(scoped_ptr<net::URLRequest> request, |
| + scoped_ptr<DownloadRequestCore> handler, |
| + base::WeakPtr<DownloadManagerImpl> manager) |
| + : request_(request.Pass()), |
| + handler_(handler.Pass()), |
| + manager_(manager), |
| + weak_ptr_factory_(this) { |
| + handler_->set_downloader(this); |
| +} |
| + |
| +UrlDownloader::~UrlDownloader() { |
| + handler_.reset(); |
| +} |
| + |
| +void UrlDownloader::Start() { |
| + DCHECK(!request_->is_pending()); |
| + |
| + if (!request_->status().is_success()) |
| + return; |
| + |
| + request_->set_delegate(this); |
| + request_->Start(); |
| +} |
| + |
| +void UrlDownloader::OnReceivedRedirect(net::URLRequest* request, |
| + const net::RedirectInfo& redirect_info, |
| + bool* defer_redirect) { |
| + DVLOG(1) << "OnReceivedRedirect: " << request_->url().spec(); |
| + request_->CancelWithError(net::ERR_ABORTED); |
| +} |
| + |
| +void UrlDownloader::OnResponseStarted(net::URLRequest* request) { |
| + DVLOG(1) << "OnResponseStarted: " << request_->url().spec(); |
| + |
| + if (!request_->status().is_success()) { |
| + ResponseCompleted(); |
| + return; |
| + } |
| + |
| + handler_->OnResponseStarted(); |
| + |
| + if (request_->status().is_success()) |
| + StartReading(false); // Read the first chunk. |
| + else |
| + ResponseCompleted(); |
| +} |
| + |
| +void UrlDownloader::StartReading(bool is_continuation) { |
| + int bytes_read; |
| + |
| + // Make sure we track the buffer in at least one place. This ensures it gets |
| + // deleted even in the case the request has already finished its job and |
| + // doesn't use the buffer. |
| + scoped_refptr<net::IOBuffer> buf; |
| + int buf_size; |
| + if (!handler_->OnWillRead(&buf, &buf_size, -1)) { |
| + request_->CancelWithError(net::ERR_ABORTED); |
| + base::ThreadTaskRunnerHandle::Get()->PostTask( |
| + FROM_HERE, base::Bind(&UrlDownloader::ResponseCompleted, |
| + weak_ptr_factory_.GetWeakPtr())); |
| + return; |
| + } |
| + |
| + DCHECK(buf.get()); |
| + DCHECK(buf_size > 0); |
| + |
| + request_->Read(buf.get(), buf_size, &bytes_read); |
| + |
| + // If IO is pending, wait for the URLRequest to call OnReadCompleted. |
| + if (request_->status().is_io_pending()) |
| + return; |
| + |
| + if (!is_continuation || bytes_read <= 0) { |
| + OnReadCompleted(request_.get(), bytes_read); |
| + } else { |
| + // Else, trigger OnReadCompleted asynchronously to avoid starving the IO |
| + // thread in case the URLRequest can provide data synchronously. |
| + base::ThreadTaskRunnerHandle::Get()->PostTask( |
| + FROM_HERE, |
| + base::Bind(&UrlDownloader::OnReadCompleted, |
| + weak_ptr_factory_.GetWeakPtr(), request_.get(), bytes_read)); |
| + } |
| +} |
| + |
| +void UrlDownloader::OnReadCompleted(net::URLRequest* request, int bytes_read) { |
| + DVLOG(1) << "OnReadCompleted: \"" << request_->url().spec() << "\"" |
| + << " bytes_read = " << bytes_read; |
| + |
| + // bytes_read == -1 always implies an error. |
| + if (bytes_read == -1 || !request_->status().is_success()) { |
| + ResponseCompleted(); |
| + return; |
| + } |
| + |
| + DCHECK(bytes_read >= 0); |
| + DCHECK(request_->status().is_success()); |
| + |
| + bool defer = false; |
| + if (!handler_->OnReadCompleted(bytes_read, &defer)) { |
| + request_->CancelWithError(net::ERR_ABORTED); |
| + return; |
| + } else if (defer) { |
| + return; |
| + } |
| + |
| + if (!request_->status().is_success()) |
| + return; |
| + |
| + if (bytes_read > 0) { |
| + StartReading(true); // Read the next chunk. |
| + } else { |
| + // URLRequest reported an EOF. Call ResponseCompleted. |
| + DCHECK_EQ(0, bytes_read); |
| + ResponseCompleted(); |
| + } |
| +} |
| + |
| +void UrlDownloader::ResponseCompleted() { |
| + DVLOG(1) << "ResponseCompleted: " << request_->url().spec(); |
| + |
| + handler_->OnResponseCompleted(request_->status()); |
| + BrowserThread::PostTask( |
| + BrowserThread::UI, FROM_HERE, |
| + base::Bind(&DownloadManagerImpl::RemoveUrlDownloader, manager_, this)); |
| +} |
| + |
| +void UrlDownloader::ResumeReading() { |
| + if (request_->status().is_success()) { |
| + StartReading(false); // Read the next chunk (OK to complete synchronously). |
| + } else { |
| + ResponseCompleted(); |
| + } |
| +} |
| + |
| +} // namespace content |