chrome/browser/net/file_downloader.cc

   1 // Copyright 2015 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 "chrome/browser/net/file_downloader.h"
   6
   7 #include "base/bind.h"
   8 #include "base/files/file_path.h"
   9 #include "base/files/file_util.h"
  10 #include "base/logging.h"
  11 #include "content/public/browser/browser_thread.h"
  12 #include "net/base/load_flags.h"
  13 #include "net/http/http_status_code.h"
  14 #include "net/url_request/url_fetcher.h"
  15 #include "url/gurl.h"
  16
  17 using content::BrowserThread;
  18 using net::URLFetcher;
  19
  20 const int kNumRetries = 1;
  21
  22 FileDownloader::FileDownloader(const GURL& url,
  23                                const base::FilePath& path,
  24                                net::URLRequestContextGetter* request_context,
  25                                const DownloadFinishedCallback& callback)
  26     : callback_(callback),
  27       fetcher_(URLFetcher::Create(url, URLFetcher::GET, this)),
  28       weak_ptr_factory_(this) {
  29   fetcher_->SetRequestContext(request_context);
  30   fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SEND_COOKIES |
  31                          net::LOAD_DO_NOT_SAVE_COOKIES);
  32   fetcher_->SetAutomaticallyRetryOnNetworkChanges(kNumRetries);
  33   fetcher_->SaveResponseToFileAtPath(
  34       path,
  35       BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE));
  36
  37   base::PostTaskAndReplyWithResult(
  38       BrowserThread::GetBlockingPool()->GetTaskRunnerWithShutdownBehavior(
  39           base::SequencedWorkerPool::CONTINUE_ON_SHUTDOWN).get(),
  40       FROM_HERE,
  41       base::Bind(&base::PathExists, path),
  42       base::Bind(&FileDownloader::OnFileExistsCheckDone,
  43                  weak_ptr_factory_.GetWeakPtr()));
  44 }
  45
  46 FileDownloader::~FileDownloader() {}
  47
  48 void FileDownloader::OnURLFetchComplete(const net::URLFetcher* source) {
  49   DCHECK_EQ(fetcher_.get(), source);
  50   // Delete |fetcher_| when we leave this method. This is necessary so the
  51   // download file will be deleted if the download failed.
  52   scoped_ptr<net::URLFetcher> fetcher(fetcher_.Pass());
  53
  54   const net::URLRequestStatus& status = source->GetStatus();
  55   if (!status.is_success()) {
  56     DLOG(WARNING) << "URLRequestStatus error " << status.error();
  57     callback_.Run(false);
  58     return;
  59   }
  60
  61   int response_code = source->GetResponseCode();
  62   if (response_code != net::HTTP_OK) {
  63     DLOG(WARNING) << "HTTP error " << response_code;
  64     callback_.Run(false);
  65     return;
  66   }
  67
  68   // Take ownership of the new file.
  69   base::FilePath response_path;
  70   bool success = source->GetResponseAsFilePath(true, &response_path);
  71   callback_.Run(success);
  72 }
  73
  74 void FileDownloader::OnFileExistsCheckDone(bool exists) {
  75   if (exists)
  76     callback_.Run(true);
  77   else
  78     fetcher_->Start();
  79 }