| // 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 "chrome/browser/net/file_downloader.h" |
| |
| #include "base/bind.h" |
| #include "base/files/file_path.h" |
| #include "base/files/file_util.h" |
| #include "base/logging.h" |
| #include "base/task_runner_util.h" |
| #include "base/task_scheduler/post_task.h" |
| #include "base/task_scheduler/task_traits.h" |
| #include "net/base/load_flags.h" |
| #include "net/http/http_status_code.h" |
| #include "net/url_request/url_fetcher.h" |
| #include "url/gurl.h" |
| |
| using net::URLFetcher; |
| |
| const int kNumFileDownloaderRetries = 1; |
| |
| FileDownloader::FileDownloader( |
| const GURL& url, |
| const base::FilePath& path, |
| bool overwrite, |
| net::URLRequestContextGetter* request_context, |
| const DownloadFinishedCallback& callback, |
| const net::NetworkTrafficAnnotationTag& traffic_annotation) |
| : callback_(callback), |
| fetcher_( |
| URLFetcher::Create(url, URLFetcher::GET, this, traffic_annotation)), |
| local_path_(path), |
| weak_ptr_factory_(this) { |
| fetcher_->SetRequestContext(request_context); |
| fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES | |
| net::LOAD_DO_NOT_SAVE_COOKIES); |
| fetcher_->SetAutomaticallyRetryOnNetworkChanges(kNumFileDownloaderRetries); |
| fetcher_->SaveResponseToTemporaryFile( |
| base::CreateSequencedTaskRunnerWithTraits( |
| {base::MayBlock(), base::TaskPriority::BACKGROUND, |
| base::TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN})); |
| |
| if (overwrite) { |
| fetcher_->Start(); |
| } else { |
| base::PostTaskAndReplyWithResult( |
| base::CreateTaskRunnerWithTraits( |
| {base::MayBlock(), base::TaskPriority::BACKGROUND, |
| base::TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN}) |
| .get(), |
| FROM_HERE, base::Bind(&base::PathExists, local_path_), |
| base::Bind(&FileDownloader::OnFileExistsCheckDone, |
| weak_ptr_factory_.GetWeakPtr())); |
| } |
| } |
| |
| FileDownloader::~FileDownloader() {} |
| |
| void FileDownloader::OnURLFetchComplete(const net::URLFetcher* source) { |
| DCHECK_EQ(fetcher_.get(), source); |
| |
| const net::URLRequestStatus& status = source->GetStatus(); |
| if (!status.is_success()) { |
| DLOG(WARNING) << "URLRequestStatus error " << status.error() |
| << " while trying to download " << source->GetURL().spec(); |
| callback_.Run(FAILED); |
| return; |
| } |
| |
| int response_code = source->GetResponseCode(); |
| if (response_code != net::HTTP_OK) { |
| DLOG(WARNING) << "HTTP error " << response_code |
| << " while trying to download " << source->GetURL().spec(); |
| callback_.Run(FAILED); |
| return; |
| } |
| |
| base::FilePath response_path; |
| bool success = source->GetResponseAsFilePath(false, &response_path); |
| if (!success) { |
| callback_.Run(FAILED); |
| return; |
| } |
| |
| base::PostTaskAndReplyWithResult( |
| base::CreateTaskRunnerWithTraits( |
| {base::MayBlock(), base::TaskPriority::BACKGROUND, |
| base::TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN}) |
| .get(), |
| FROM_HERE, base::Bind(&base::Move, response_path, local_path_), |
| base::Bind(&FileDownloader::OnFileMoveDone, |
| weak_ptr_factory_.GetWeakPtr())); |
| } |
| |
| void FileDownloader::OnFileExistsCheckDone(bool exists) { |
| if (exists) |
| callback_.Run(EXISTS); |
| else |
| fetcher_->Start(); |
| } |
| |
| void FileDownloader::OnFileMoveDone(bool success) { |
| if (!success) { |
| DLOG(WARNING) << "Could not move file to " |
| << local_path_.LossyDisplayName(); |
| } |
| |
| callback_.Run(success ? DOWNLOADED : FAILED); |
| } |