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 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_NAVIGATION_INTERCEPTION_INTERCEPT_NAVIGATION_THROTTLE_H_
#define COMPONENTS_NAVIGATION_INTERCEPTION_INTERCEPT_NAVIGATION_THROTTLE_H_
#include "base/feature_list.h"
#include "base/functional/callback.h"
#include "base/memory/ref_counted.h"
#include "base/memory/weak_ptr.h"
#include "base/task/single_thread_task_runner.h"
#include "content/public/browser/navigation_throttle.h"
namespace content {
class NavigationHandle;
}
namespace navigation_interception {
BASE_DECLARE_FEATURE(kAsyncCheck);
enum class SynchronyMode {
// Support async interception in some cases (See ShouldCheckAsynchronously).
kAsync,
// Only support synchronous interception.
kSync
};
// This class allows the provider of the Callback to selectively ignore top
// level navigations. This is a UI thread class.
class InterceptNavigationThrottle : public content::NavigationThrottle {
public:
typedef base::RepeatingCallback<bool(
content::NavigationHandle* /* navigation_handle */)>
CheckCallback;
InterceptNavigationThrottle(content::NavigationHandle* navigation_handle,
CheckCallback should_ignore_callback,
SynchronyMode async_mode);
InterceptNavigationThrottle(const InterceptNavigationThrottle&) = delete;
InterceptNavigationThrottle& operator=(const InterceptNavigationThrottle&) =
delete;
~InterceptNavigationThrottle() override;
// content::NavigationThrottle implementation:
ThrottleCheckResult WillStartRequest() override;
ThrottleCheckResult WillRedirectRequest() override;
ThrottleCheckResult WillProcessResponse() override;
const char* GetNameForLogging() override;
private:
ThrottleCheckResult CheckIfShouldIgnoreNavigation();
void RunCheckAsync();
bool ShouldCheckAsynchronously() const;
// This callback should be called at the start of navigation and every
// redirect, until |should_ignore_| is true.
// Note: the callback can delete |this|.
CheckCallback should_ignore_callback_;
// Note that the CheckCallback currently has thread affinity on the Java side.
scoped_refptr<base::SingleThreadTaskRunner> ui_task_runner_;
const SynchronyMode mode_ = SynchronyMode::kSync;
// The remaining members are only set for asynchronous checking.
//
// How many outbound pending checks are running. Normally this will be either
// 0 or 1, but making this a bool makes too many assumptions about the nature
// of Chrome's task queues (e.g. we could be scheduled after the task which
// redirects the navigation).
int pending_checks_ = 0;
// Whether the navigation should be ignored. Updated at every redirect.
bool should_ignore_ = false;
// Whether the navigation is currently deferred.
bool deferring_ = false;
base::WeakPtrFactory<InterceptNavigationThrottle> weak_factory_{this};
};
} // namespace navigation_interception
#endif // COMPONENTS_NAVIGATION_INTERCEPTION_INTERCEPT_NAVIGATION_THROTTLE_H_
|
