| // Copyright 2014 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/chromeos/customization/customization_wallpaper_downloader.h" |
| |
| #include <math.h> |
| #include <algorithm> |
| #include <utility> |
| |
| #include "base/files/file_util.h" |
| #include "base/task_scheduler/post_task.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "net/base/load_flags.h" |
| #include "net/http/http_status_code.h" |
| #include "net/url_request/url_fetcher.h" |
| #include "net/url_request/url_request_context_getter.h" |
| #include "url/gurl.h" |
| |
| namespace chromeos { |
| namespace { |
| // This is temporary file suffix (for downloading or resizing). |
| const char kTemporarySuffix[] = ".tmp"; |
| |
| // Sleep between wallpaper retries (used multiplied by squared retry number). |
| const unsigned kRetrySleepSeconds = 10; |
| |
| // Retry is infinite with increasing intervals. When calculated delay becomes |
| // longer than maximum (kMaxRetrySleepSeconds) it is set to the maximum. |
| const double kMaxRetrySleepSeconds = 6 * 3600; // 6 hours |
| |
| void CreateWallpaperDirectory(const base::FilePath& wallpaper_dir, |
| bool* success) { |
| DCHECK(success); |
| |
| *success = CreateDirectoryAndGetError(wallpaper_dir, NULL); |
| if (!*success) { |
| NOTREACHED() << "Failed to create directory '" << wallpaper_dir.value() |
| << "'"; |
| } |
| } |
| |
| void RenameTemporaryFile(const base::FilePath& from, |
| const base::FilePath& to, |
| bool* success) { |
| DCHECK(success); |
| |
| base::File::Error error; |
| if (base::ReplaceFile(from, to, &error)) { |
| *success = true; |
| } else { |
| LOG(WARNING) |
| << "Failed to rename temporary file of Customized Wallpaper. error=" |
| << error; |
| *success = false; |
| } |
| } |
| |
| } // namespace |
| |
| CustomizationWallpaperDownloader::CustomizationWallpaperDownloader( |
| net::URLRequestContextGetter* url_context_getter, |
| const GURL& wallpaper_url, |
| const base::FilePath& wallpaper_dir, |
| const base::FilePath& wallpaper_downloaded_file, |
| base::Callback<void(bool success, const GURL&)> |
| on_wallpaper_fetch_completed) |
| : url_context_getter_(url_context_getter), |
| wallpaper_url_(wallpaper_url), |
| wallpaper_dir_(wallpaper_dir), |
| wallpaper_downloaded_file_(wallpaper_downloaded_file), |
| wallpaper_temporary_file_(wallpaper_downloaded_file.value() + |
| kTemporarySuffix), |
| retries_(0), |
| retry_delay_(base::TimeDelta::FromSeconds(kRetrySleepSeconds)), |
| on_wallpaper_fetch_completed_(on_wallpaper_fetch_completed), |
| weak_factory_(this) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| } |
| |
| CustomizationWallpaperDownloader::~CustomizationWallpaperDownloader() { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| } |
| |
| void CustomizationWallpaperDownloader::StartRequest() { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| DCHECK(wallpaper_url_.is_valid()); |
| |
| url_fetcher_ = |
| net::URLFetcher::Create(wallpaper_url_, net::URLFetcher::GET, this); |
| url_fetcher_->SetRequestContext(url_context_getter_.get()); |
| url_fetcher_->SetLoadFlags(net::LOAD_BYPASS_CACHE | |
| net::LOAD_DISABLE_CACHE | |
| net::LOAD_DO_NOT_SAVE_COOKIES | |
| net::LOAD_DO_NOT_SEND_COOKIES | |
| net::LOAD_DO_NOT_SEND_AUTH_DATA); |
| url_fetcher_->SaveResponseToFileAtPath( |
| wallpaper_temporary_file_, |
| base::CreateSequencedTaskRunnerWithTraits( |
| {base::MayBlock(), base::TaskPriority::BACKGROUND})); |
| url_fetcher_->Start(); |
| } |
| |
| void CustomizationWallpaperDownloader::Retry() { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| ++retries_; |
| |
| const double delay_seconds = std::min( |
| kMaxRetrySleepSeconds, |
| static_cast<double>(retries_) * retries_ * retry_delay_.InSecondsF()); |
| const base::TimeDelta delay = base::TimeDelta::FromSecondsD(delay_seconds); |
| |
| VLOG(1) << "Schedule Customized Wallpaper download in " << delay.InSecondsF() |
| << " seconds (retry = " << retries_ << ")."; |
| retry_current_delay_ = delay; |
| request_scheduled_.Start( |
| FROM_HERE, delay, this, &CustomizationWallpaperDownloader::StartRequest); |
| } |
| |
| void CustomizationWallpaperDownloader::Start() { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| std::unique_ptr<bool> success(new bool(false)); |
| |
| base::OnceClosure mkdir_closure = |
| base::BindOnce(&CreateWallpaperDirectory, wallpaper_dir_, |
| base::Unretained(success.get())); |
| base::OnceClosure on_created_closure = base::BindOnce( |
| &CustomizationWallpaperDownloader::OnWallpaperDirectoryCreated, |
| weak_factory_.GetWeakPtr(), base::Passed(std::move(success))); |
| base::PostTaskWithTraitsAndReply( |
| FROM_HERE, {base::MayBlock(), base::TaskPriority::BACKGROUND}, |
| std::move(mkdir_closure), std::move(on_created_closure)); |
| } |
| |
| void CustomizationWallpaperDownloader::OnWallpaperDirectoryCreated( |
| std::unique_ptr<bool> success) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| if (*success) |
| StartRequest(); |
| } |
| |
| void CustomizationWallpaperDownloader::OnURLFetchComplete( |
| const net::URLFetcher* source) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| DCHECK_EQ(url_fetcher_.get(), source); |
| |
| const net::URLRequestStatus status = source->GetStatus(); |
| const int response_code = source->GetResponseCode(); |
| |
| const bool server_error = |
| !status.is_success() || |
| (response_code >= net::HTTP_INTERNAL_SERVER_ERROR && |
| response_code < (net::HTTP_INTERNAL_SERVER_ERROR + 100)); |
| |
| VLOG(1) << "CustomizationWallpaperDownloader::OnURLFetchComplete(): status=" |
| << status.status(); |
| |
| if (server_error) { |
| url_fetcher_.reset(); |
| Retry(); |
| return; |
| } |
| |
| base::FilePath response_path; |
| url_fetcher_->GetResponseAsFilePath(true, &response_path); |
| url_fetcher_.reset(); |
| |
| std::unique_ptr<bool> success(new bool(false)); |
| |
| base::OnceClosure rename_closure = base::BindOnce( |
| &RenameTemporaryFile, response_path, wallpaper_downloaded_file_, |
| base::Unretained(success.get())); |
| base::OnceClosure on_rename_closure = base::BindOnce( |
| &CustomizationWallpaperDownloader::OnTemporaryFileRenamed, |
| weak_factory_.GetWeakPtr(), base::Passed(std::move(success))); |
| base::PostTaskWithTraitsAndReply( |
| FROM_HERE, {base::MayBlock(), base::TaskPriority::BACKGROUND}, |
| std::move(rename_closure), std::move(on_rename_closure)); |
| } |
| |
| void CustomizationWallpaperDownloader::OnTemporaryFileRenamed( |
| std::unique_ptr<bool> success) { |
| DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| on_wallpaper_fetch_completed_.Run(*success, wallpaper_url_); |
| } |
| |
| } // namespace chromeos |