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