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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
// 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,
bool overwrite,
net::URLRequestContextGetter* request_context,
const DownloadFinishedCallback& callback)
: callback_(callback),
fetcher_(URLFetcher::Create(url, URLFetcher::GET, this)),
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(kNumRetries);
fetcher_->SaveResponseToTemporaryFile(
BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE));
if (overwrite) {
fetcher_->Start();
} else {
base::PostTaskAndReplyWithResult(
BrowserThread::GetBlockingPool()
->GetTaskRunnerWithShutdownBehavior(
base::SequencedWorkerPool::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(false);
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(false);
return;
}
base::FilePath response_path;
bool success = source->GetResponseAsFilePath(false, &response_path);
if (!success) {
callback_.Run(false);
return;
}
base::PostTaskAndReplyWithResult(
BrowserThread::GetBlockingPool()
->GetTaskRunnerWithShutdownBehavior(
base::SequencedWorkerPool::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(true);
else
fetcher_->Start();
}
void FileDownloader::OnFileMoveDone(bool success) {
if (!success) {
DLOG(WARNING) << "Could not move file to "
<< local_path_.LossyDisplayName();
}
callback_.Run(success);
}
|
