| // Copyright 2020 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. |
| |
| #ifndef CHROME_BROWSER_ASH_POLICY_DLP_DLP_CONTENT_TAB_HELPER_H_ |
| #define CHROME_BROWSER_ASH_POLICY_DLP_DLP_CONTENT_TAB_HELPER_H_ |
| |
| #include "base/containers/flat_map.h" |
| #include "chrome/browser/ash/policy/dlp/dlp_content_restriction_set.h" |
| #include "content/public/browser/visibility.h" |
| #include "content/public/browser/web_contents_observer.h" |
| #include "content/public/browser/web_contents_user_data.h" |
| |
| namespace content { |
| class NavigationHandle; |
| class RenderFrameHost; |
| class WebContents; |
| } // namespace content |
| |
| namespace policy { |
| |
| // DlpContentTabHelper attaches to relevant WebContents that are covered by |
| // DLP (Data Leak Prevention) feature and observes navigation in all sub-frames |
| // as well as visibility of the WebContents and reports it to system-wide |
| // DlpContentManager. |
| // WebContents is considered as confidential if either the main frame or any |
| // of sub-frames are confidential according to the current policy. |
| // In this case the applied restrictions for the WebContents will be the |
| // superset of all restrictions for sub-frames. |
| class DlpContentTabHelper |
| : public content::WebContentsUserData<DlpContentTabHelper>, |
| public content::WebContentsObserver { |
| public: |
| ~DlpContentTabHelper() override; |
| |
| // content::WebContentsObserver: |
| void RenderFrameCreated(content::RenderFrameHost* render_frame_host) override; |
| void RenderFrameDeleted(content::RenderFrameHost* render_frame_host) override; |
| void RenderFrameHostStateChanged( |
| content::RenderFrameHost* render_frame_host, |
| content::RenderFrameHost::LifecycleState old_state, |
| content::RenderFrameHost::LifecycleState new_state) override; |
| void DidFinishNavigation( |
| content::NavigationHandle* navigation_handle) override; |
| void WebContentsDestroyed() override; |
| void OnVisibilityChanged(content::Visibility visibility) override; |
| |
| private: |
| friend class content::WebContentsUserData<DlpContentTabHelper>; |
| |
| explicit DlpContentTabHelper(content::WebContents* web_contents); |
| DlpContentTabHelper(const DlpContentTabHelper&) = delete; |
| DlpContentTabHelper& operator=(const DlpContentTabHelper&) = delete; |
| |
| // Returns the superset of all restrictions for sub-frames. |
| DlpContentRestrictionSet GetRestrictionSet() const; |
| |
| // Adds or updates the |render_frame_host| and |restrictions| to/in the map |
| // and notifies DlpContentManager if needed. |
| void AddFrame(content::RenderFrameHost* render_frame_host, |
| DlpContentRestrictionSet restrictions); |
| |
| // Removes |render_frame_host| from the map and notifies DlpContentManager if |
| // needed. |
| void RemoveFrame(content::RenderFrameHost* render_frame_host); |
| |
| // Map from the currently known confidential frames to the corresponding |
| // restriction set. |
| base::flat_map<content::RenderFrameHost*, DlpContentRestrictionSet> |
| confidential_frames_; |
| |
| WEB_CONTENTS_USER_DATA_KEY_DECL(); |
| }; |
| |
| } // namespace policy |
| |
| #endif // CHROME_BROWSER_ASH_POLICY_DLP_DLP_CONTENT_TAB_HELPER_H_ |