File: supervised_user_navigation_observer.h

package info (click to toggle)
chromium 138.0.7204.157-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 6,071,864 kB
  • sloc: cpp: 34,936,859; ansic: 7,176,967; javascript: 4,110,704; python: 1,419,953; asm: 946,768; xml: 739,967; pascal: 187,324; sh: 89,623; perl: 88,663; objc: 79,944; sql: 50,304; cs: 41,786; fortran: 24,137; makefile: 21,806; php: 13,980; tcl: 13,166; yacc: 8,925; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (168 lines) | stat: -rw-r--r-- 6,832 bytes parent folder | download | duplicates (4)
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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_SUPERVISED_USER_SUPERVISED_USER_NAVIGATION_OBSERVER_H_
#define CHROME_BROWSER_SUPERVISED_USER_SUPERVISED_USER_NAVIGATION_OBSERVER_H_

#include <map>
#include <memory>
#include <set>
#include <vector>

#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/supervised_user/classify_url_navigation_throttle.h"
#include "chrome/common/supervised_user_commands.mojom.h"
#include "components/sessions/core/serialized_navigation_entry.h"
#include "components/supervised_user/core/browser/supervised_user_error_page.h"
#include "components/supervised_user/core/browser/supervised_user_service_observer.h"
#include "components/supervised_user/core/browser/supervised_user_url_filter.h"
#include "components/supervised_user/core/browser/supervised_user_utils.h"
#include "components/supervised_user/core/common/supervised_users.h"
#include "content/public/browser/render_frame_host_receiver_set.h"
#include "content/public/browser/web_contents_observer.h"
#include "content/public/browser/web_contents_user_data.h"

namespace supervised_user {
class SupervisedUserService;
class SupervisedUserInterstitial;
}  // namespace supervised_user

namespace content {
class NavigationHandle;
class RenderFrameHost;
class WebContents;
}  // namespace content

using OnInterstitialResultCallback = base::RepeatingCallback<
    void(supervised_user::InterstitialResultCallbackActions, bool, bool)>;

class SupervisedUserNavigationObserver
    : public content::WebContentsUserData<SupervisedUserNavigationObserver>,
      public content::WebContentsObserver,
      public SupervisedUserServiceObserver,
      public supervised_user::mojom::SupervisedUserCommands {
 public:
  SupervisedUserNavigationObserver(const SupervisedUserNavigationObserver&) =
      delete;
  SupervisedUserNavigationObserver& operator=(
      const SupervisedUserNavigationObserver&) = delete;

  ~SupervisedUserNavigationObserver() override;

  const std::vector<std::unique_ptr<const sessions::SerializedNavigationEntry>>&
  blocked_navigations() const {
    return blocked_navigations_;
  }

  static void BindSupervisedUserCommands(
      mojo::PendingAssociatedReceiver<
          supervised_user::mojom::SupervisedUserCommands> receiver,
      content::RenderFrameHost* rfh);

  // Called when a network request to |url| is blocked.
  static void OnRequestBlocked(content::WebContents* web_contents,
                               const GURL& url,
                               supervised_user::FilteringBehaviorReason reason,
                               int64_t navigation_id,
                               content::FrameTreeNodeId frame_id,
                               const OnInterstitialResultCallback& callback);

  // WebContentsObserver:
  void DidFinishNavigation(
      content::NavigationHandle* navigation_handle) override;
  void FrameDeleted(content::FrameTreeNodeId frame_tree_node_id) override;
  void DidFinishLoad(content::RenderFrameHost* render_frame_host,
                     const GURL& validated_url) override;

  // SupervisedUserServiceObserver:
  void OnURLFilterChanged() override;

  // Called when interstitial error page is no longer being shown in the main
  // frame.
  void OnInterstitialDone(content::FrameTreeNodeId frame_id);

  const std::map<content::FrameTreeNodeId,
                 std::unique_ptr<supervised_user::SupervisedUserInterstitial>>&
  interstitials_for_test() const {
    return supervised_user_interstitials_;
  }

  const std::set<std::string>& requested_hosts_for_test() const {
    return requested_hosts_;
  }

 private:
  friend class content::WebContentsUserData<SupervisedUserNavigationObserver>;

  explicit SupervisedUserNavigationObserver(content::WebContents* web_contents);

  void OnRequestBlockedInternal(const GURL& url,
                                supervised_user::FilteringBehaviorReason reason,
                                int64_t navigation_id,
                                content::FrameTreeNodeId frame_id,
                                const OnInterstitialResultCallback& callback);

  void URLFilterCheckCallback(
      int render_frame_process_id,
      int render_frame_routing_id,
      supervised_user::SupervisedUserURLFilter::Result result);

  void MaybeShowInterstitial(const GURL& url,
                             supervised_user::FilteringBehaviorReason reason,
                             bool initial_page_load,
                             int64_t navigation_id,
                             content::FrameTreeNodeId frame_id,
                             const OnInterstitialResultCallback& callback);

  // Filters the RenderFrameHost if render frame is live.
  void FilterRenderFrame(content::RenderFrameHost* render_frame_host);

  // supervised_user::mojom::SupervisedUserCommands implementation. Should not
  // be called when an interstitial is no longer showing. This should be
  // enforced by the mojo caller.
  void GoBack() override;
  void RequestUrlAccessRemote(RequestUrlAccessRemoteCallback callback) override;
  void RequestUrlAccessLocal(RequestUrlAccessLocalCallback callback) override;

  // When a remote URL approval request is successfully created, this method is
  // called asynchronously.
  void RequestCreated(RequestUrlAccessRemoteCallback callback,
                      const std::string& host,
                      bool successfully_created_request);

  // Called when the url filter changes i.e. allowlist or denylist change to
  // clear up  entries in |requested_hosts_| which have been allowed.
  void MaybeUpdateRequestedHosts();

  // Owned by SupervisedUserService.
  raw_ptr<supervised_user::SupervisedUserURLFilter> url_filter_;

  // Owned by SupervisedUserServiceFactory (lifetime of Profile).
  raw_ptr<supervised_user::SupervisedUserService> supervised_user_service_;

  // Keeps track of the blocked frames. It maps the frame's globally unique
  // id to its corresponding |SupervisedUserInterstitial| instance.
  std::map<content::FrameTreeNodeId,
           std::unique_ptr<supervised_user::SupervisedUserInterstitial>>
      supervised_user_interstitials_;

  std::set<std::string> requested_hosts_;

  std::vector<std::unique_ptr<const sessions::SerializedNavigationEntry>>
      blocked_navigations_;

  content::RenderFrameHostReceiverSet<
      supervised_user::mojom::SupervisedUserCommands>
      receivers_;

  base::WeakPtrFactory<SupervisedUserNavigationObserver> weak_ptr_factory_{
      this};

  void RecordPageLoadUKM(content::RenderFrameHost* render_frame_host);

  WEB_CONTENTS_USER_DATA_KEY_DECL();
};

#endif  // CHROME_BROWSER_SUPERVISED_USER_SUPERVISED_USER_NAVIGATION_OBSERVER_H_