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
|
// Copyright (c) 2012 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/download/download_resource_throttle.h"
#include "base/bind.h"
#include "base/profiler/scoped_tracker.h"
#include "chrome/browser/download/download_stats.h"
#include "content/public/browser/resource_controller.h"
DownloadResourceThrottle::DownloadResourceThrottle(
DownloadRequestLimiter* limiter,
int render_process_id,
int render_view_id,
const GURL& url,
const std::string& request_method)
: querying_limiter_(true),
request_allowed_(false),
request_deferred_(false) {
limiter->CanDownloadOnIOThread(
render_process_id,
render_view_id,
url,
request_method,
base::Bind(&DownloadResourceThrottle::ContinueDownload,
AsWeakPtr()));
}
DownloadResourceThrottle::~DownloadResourceThrottle() {
}
void DownloadResourceThrottle::WillStartRequest(bool* defer) {
WillDownload(defer);
}
void DownloadResourceThrottle::WillRedirectRequest(const GURL& new_url,
bool* defer) {
WillDownload(defer);
}
void DownloadResourceThrottle::WillProcessResponse(bool* defer) {
// TODO(vadimt): Remove ScopedTracker below once crbug.com/422516 is fixed.
tracked_objects::ScopedTracker tracking_profile(
FROM_HERE_WITH_EXPLICIT_FUNCTION(
"422516 DownloadResourceThrottle::WillProcessResponse"));
WillDownload(defer);
}
const char* DownloadResourceThrottle::GetNameForLogging() const {
return "DownloadResourceThrottle";
}
void DownloadResourceThrottle::WillDownload(bool* defer) {
DCHECK(!request_deferred_);
// Defer the download until we have the DownloadRequestLimiter result.
if (querying_limiter_) {
request_deferred_ = true;
*defer = true;
return;
}
if (!request_allowed_)
controller()->Cancel();
}
void DownloadResourceThrottle::ContinueDownload(bool allow) {
querying_limiter_ = false;
request_allowed_ = allow;
if (allow) {
// Presumes all downloads initiated by navigation use this throttle and
// nothing else does.
RecordDownloadSource(DOWNLOAD_INITIATED_BY_NAVIGATION);
} else {
RecordDownloadCount(CHROME_DOWNLOAD_COUNT_BLOCKED_BY_THROTTLING);
}
if (request_deferred_) {
request_deferred_ = false;
if (allow) {
controller()->Resume();
} else {
controller()->Cancel();
}
}
}
|
