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
105
106
107
108
109
110
111
112
113
114
115
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/navigation_interception/intercept_navigation_throttle.h"
#include "base/functional/bind.h"
#include "base/task/single_thread_task_runner.h"
#include "content/public/browser/navigation_handle.h"
#include "url/gurl.h"
namespace navigation_interception {
// Note: this feature is a no-op on non-Android platforms.
BASE_FEATURE(kAsyncCheck,
"AsyncNavigationIntercept",
base::FEATURE_ENABLED_BY_DEFAULT);
InterceptNavigationThrottle::InterceptNavigationThrottle(
content::NavigationHandle* navigation_handle,
CheckCallback should_ignore_callback,
SynchronyMode async_mode)
: content::NavigationThrottle(navigation_handle),
should_ignore_callback_(should_ignore_callback),
ui_task_runner_(base::SingleThreadTaskRunner::GetCurrentDefault()),
mode_(async_mode) {}
InterceptNavigationThrottle::~InterceptNavigationThrottle() = default;
content::NavigationThrottle::ThrottleCheckResult
InterceptNavigationThrottle::WillStartRequest() {
DCHECK(!should_ignore_);
DCHECK(!navigation_handle()->WasServerRedirect());
return CheckIfShouldIgnoreNavigation();
}
content::NavigationThrottle::ThrottleCheckResult
InterceptNavigationThrottle::WillRedirectRequest() {
if (should_ignore_)
return content::NavigationThrottle::CANCEL_AND_IGNORE;
DCHECK(navigation_handle()->WasServerRedirect());
return CheckIfShouldIgnoreNavigation();
}
content::NavigationThrottle::ThrottleCheckResult
InterceptNavigationThrottle::WillProcessResponse() {
DCHECK(!deferring_);
if (should_ignore_)
return content::NavigationThrottle::CANCEL_AND_IGNORE;
if (pending_checks_ > 0) {
deferring_ = true;
return content::NavigationThrottle::DEFER;
}
return content::NavigationThrottle::PROCEED;
}
const char* InterceptNavigationThrottle::GetNameForLogging() {
return "InterceptNavigationThrottle";
}
content::NavigationThrottle::ThrottleCheckResult
InterceptNavigationThrottle::CheckIfShouldIgnoreNavigation() {
if (ShouldCheckAsynchronously()) {
pending_checks_++;
ui_task_runner_->PostTask(
FROM_HERE, base::BindOnce(&InterceptNavigationThrottle::RunCheckAsync,
weak_factory_.GetWeakPtr()));
return content::NavigationThrottle::PROCEED;
}
// No need to set |should_ignore_| since if it is true, we'll cancel the
// navigation immediately.
return should_ignore_callback_.Run(navigation_handle())
? content::NavigationThrottle::CANCEL_AND_IGNORE
: content::NavigationThrottle::PROCEED;
// Careful, |this| can be deleted at this point.
}
void InterceptNavigationThrottle::RunCheckAsync() {
DCHECK(base::FeatureList::IsEnabled(kAsyncCheck));
DCHECK_GT(pending_checks_, 0);
pending_checks_--;
bool final_deferred_check = deferring_ && pending_checks_ == 0;
auto weak_this = weak_factory_.GetWeakPtr();
bool should_ignore = should_ignore_callback_.Run(navigation_handle());
if (!weak_this)
return;
should_ignore_ |= should_ignore;
if (!final_deferred_check)
return;
if (should_ignore) {
CancelDeferredNavigation(content::NavigationThrottle::CANCEL_AND_IGNORE);
} else {
Resume();
}
}
bool InterceptNavigationThrottle::ShouldCheckAsynchronously() const {
// Do not apply the async optimization for:
// - Throttles in non-async mode.
// - POST navigations, to ensure we aren't violating idempotency.
// - Subframe navigations, which aren't observed on Android, and should be
// fast on other platforms.
// - non-http/s URLs, which are more likely to be intercepted.
return mode_ == SynchronyMode::kAsync &&
navigation_handle()->IsInMainFrame() &&
!navigation_handle()->IsPost() &&
navigation_handle()->GetURL().SchemeIsHTTPOrHTTPS() &&
base::FeatureList::IsEnabled(kAsyncCheck);
}
} // namespace navigation_interception
|
