components/update_client/url_fetcher_downloader.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "components/update_client/url_fetcher_downloader.h"
   6
   7 #include <stdint.h>
   8
   9 #include "base/logging.h"
  10 #include "base/sequenced_task_runner.h"
  11 #include "components/update_client/utils.h"
  12 #include "net/base/load_flags.h"
  13 #include "net/url_request/url_fetcher.h"
  14 #include "url/gurl.h"
  15
  16 namespace update_client {
  17
  18 UrlFetcherDownloader::UrlFetcherDownloader(
  19     scoped_ptr<CrxDownloader> successor,
  20     net::URLRequestContextGetter* context_getter,
  21     scoped_refptr<base::SequencedTaskRunner> task_runner)
  22     : CrxDownloader(successor.Pass()),
  23       context_getter_(context_getter),
  24       task_runner_(task_runner),
  25       downloaded_bytes_(-1),
  26       total_bytes_(-1) {
  27 }
  28
  29 UrlFetcherDownloader::~UrlFetcherDownloader() {
  30   DCHECK(thread_checker_.CalledOnValidThread());
  31 }
  32
  33 void UrlFetcherDownloader::DoStartDownload(const GURL& url) {
  34   DCHECK(thread_checker_.CalledOnValidThread());
  35
  36   url_fetcher_.reset(
  37       net::URLFetcher::Create(0, url, net::URLFetcher::GET, this));
  38   url_fetcher_->SetRequestContext(context_getter_);
  39   url_fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES |
  40                              net::LOAD_DO_NOT_SAVE_COOKIES |
  41                              net::LOAD_DISABLE_CACHE);
  42   url_fetcher_->SetAutomaticallyRetryOn5xx(false);
  43   url_fetcher_->SaveResponseToTemporaryFile(task_runner_);
  44
  45   VLOG(1) << "Starting background download: " << url.spec();
  46   url_fetcher_->Start();
  47
  48   download_start_time_ = base::Time::Now();
  49
  50   downloaded_bytes_ = -1;
  51   total_bytes_ = -1;
  52 }
  53
  54 void UrlFetcherDownloader::OnURLFetchComplete(const net::URLFetcher* source) {
  55   DCHECK(thread_checker_.CalledOnValidThread());
  56
  57   const base::Time download_end_time(base::Time::Now());
  58   const base::TimeDelta download_time =
  59       download_end_time >= download_start_time_
  60           ? download_end_time - download_start_time_
  61           : base::TimeDelta();
  62
  63   // Consider a 5xx response from the server as an indication to terminate
  64   // the request and avoid overloading the server in this case.
  65   // is not accepting requests for the moment.
  66   const int fetch_error(GetFetchError(*url_fetcher_));
  67   const bool is_handled = fetch_error == 0 || IsHttpServerError(fetch_error);
  68
  69   Result result;
  70   result.error = fetch_error;
  71   if (!fetch_error) {
  72     source->GetResponseAsFilePath(true, &result.response);
  73   }
  74   result.downloaded_bytes = downloaded_bytes_;
  75   result.total_bytes = total_bytes_;
  76
  77   DownloadMetrics download_metrics;
  78   download_metrics.url = url();
  79   download_metrics.downloader = DownloadMetrics::kUrlFetcher;
  80   download_metrics.error = fetch_error;
  81   download_metrics.downloaded_bytes = downloaded_bytes_;
  82   download_metrics.total_bytes = total_bytes_;
  83   download_metrics.download_time_ms = download_time.InMilliseconds();
  84
  85   base::FilePath local_path_;
  86   source->GetResponseAsFilePath(false, &local_path_);
  87   VLOG(1) << "Downloaded " << downloaded_bytes_ << " bytes in "
  88           << download_time.InMilliseconds() << "ms from "
  89           << source->GetURL().spec() << " to " << local_path_.value();
  90   CrxDownloader::OnDownloadComplete(is_handled, result, download_metrics);
  91 }
  92
  93 void UrlFetcherDownloader::OnURLFetchDownloadProgress(
  94     const net::URLFetcher* source,
  95     int64_t current,
  96     int64_t total) {
  97   DCHECK(thread_checker_.CalledOnValidThread());
  98
  99   downloaded_bytes_ = current;
 100   total_bytes_ = total;
 101
 102   Result result;
 103   result.downloaded_bytes = downloaded_bytes_;
 104   result.total_bytes = total_bytes_;
 105
 106   OnDownloadProgress(result);
 107 }
 108
 109 }  // namespace update_client