| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FILTER
_DRIVER_FACTORY_H_ | 5 #ifndef COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FILTER
_DRIVER_FACTORY_H_ |
| 6 #define COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FILTER
_DRIVER_FACTORY_H_ | 6 #define COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FILTER
_DRIVER_FACTORY_H_ |
| 7 | 7 |
| 8 #include <map> |
| 8 #include <memory> | 9 #include <memory> |
| 10 #include <set> |
| 11 #include <string> |
| 12 #include <utility> |
| 13 #include <vector> |
| 9 | 14 |
| 10 #include "base/macros.h" | 15 #include "base/macros.h" |
| 16 #include "base/time/time.h" |
| 11 #include "components/safe_browsing_db/util.h" | 17 #include "components/safe_browsing_db/util.h" |
| 12 #include "components/subresource_filter/content/browser/content_subresource_filt
er_throttle_manager.h" | 18 #include "components/subresource_filter/content/browser/content_subresource_filt
er_throttle_manager.h" |
| 13 #include "components/subresource_filter/core/browser/subresource_filter_features
.h" | 19 #include "components/subresource_filter/core/browser/subresource_filter_features
.h" |
| 14 #include "components/subresource_filter/core/common/activation_decision.h" | 20 #include "components/subresource_filter/core/common/activation_decision.h" |
| 15 #include "content/public/browser/web_contents_observer.h" | 21 #include "content/public/browser/web_contents_observer.h" |
| 16 #include "content/public/browser/web_contents_user_data.h" | 22 #include "content/public/browser/web_contents_user_data.h" |
| 17 #include "ui/base/page_transition_types.h" | 23 #include "ui/base/page_transition_types.h" |
| 18 | 24 #include "url/gurl.h" |
| 19 namespace base { | |
| 20 class GURL; | |
| 21 } // namespace base | |
| 22 | 25 |
| 23 namespace content { | 26 namespace content { |
| 24 class WebContents; | 27 class WebContents; |
| 25 } // namespace content | 28 } // namespace content |
| 26 | 29 |
| 27 namespace safe_browsing { | 30 namespace safe_browsing { |
| 28 class SafeBrowsingServiceTest; | 31 class SafeBrowsingServiceTest; |
| 29 }; | 32 }; |
| 30 | 33 |
| 31 namespace subresource_filter { | 34 namespace subresource_filter { |
| 32 | 35 |
| 33 class SubresourceFilterClient; | 36 class SubresourceFilterClient; |
| 34 enum class ActivationLevel; | 37 enum class ActivationLevel; |
| 35 enum class ActivationList; | 38 enum class ActivationList; |
| 36 | 39 |
| 40 using URLToActivationListsMap = |
| 41 std::unordered_map<std::string, std::set<ActivationList>>; |
| 42 |
| 37 // Controls the activation of subresource filtering for each page load in a | 43 // Controls the activation of subresource filtering for each page load in a |
| 38 // WebContents and is responsible for sending the activation signal to all the | 44 // WebContents and is responsible for sending the activation signal to all the |
| 39 // per-frame SubresourceFilterAgents on the renderer side. | 45 // per-frame SubresourceFilterAgents on the renderer side. |
| 40 class ContentSubresourceFilterDriverFactory | 46 class ContentSubresourceFilterDriverFactory |
| 41 : public content::WebContentsUserData< | 47 : public content::WebContentsUserData< |
| 42 ContentSubresourceFilterDriverFactory>, | 48 ContentSubresourceFilterDriverFactory>, |
| 43 public content::WebContentsObserver, | 49 public content::WebContentsObserver, |
| 44 public ContentSubresourceFilterThrottleManager::Delegate { | 50 public ContentSubresourceFilterThrottleManager::Delegate { |
| 45 public: | 51 public: |
| 46 static void CreateForWebContents(content::WebContents* web_contents, | 52 static void CreateForWebContents(content::WebContents* web_contents, |
| 47 SubresourceFilterClient* client); | 53 SubresourceFilterClient* client); |
| 48 | 54 |
| 49 // Whether the |url|, |referrer|, and |transition| are considered to be | 55 // Whether the |url|, |referrer|, and |transition| are considered to be |
| 50 // associated with a page reload. | 56 // associated with a page reload. |
| 51 static bool NavigationIsPageReload(const GURL& url, | 57 static bool NavigationIsPageReload(const GURL& url, |
| 52 const content::Referrer& referrer, | 58 const content::Referrer& referrer, |
| 53 ui::PageTransition transition); | 59 ui::PageTransition transition); |
| 54 | 60 |
| 55 explicit ContentSubresourceFilterDriverFactory( | 61 explicit ContentSubresourceFilterDriverFactory( |
| 56 content::WebContents* web_contents, | 62 content::WebContents* web_contents, |
| 57 SubresourceFilterClient* client); | 63 SubresourceFilterClient* client); |
| 58 ~ContentSubresourceFilterDriverFactory() override; | 64 ~ContentSubresourceFilterDriverFactory() override; |
| 59 | 65 |
| 60 // Called when the Safe Browsing throttle has computed the threat type and | 66 // Called when Safe Browsing detects that the |url| corresponding to the load |
| 61 // metadata for this navigation. Will be called once per navigation before | 67 // of the main frame belongs to the blacklist with |threat_type|. If the |
| 62 // the ReadyToCommit stage. | 68 // blacklist is the Safe Browsing Social Engineering ads landing, then |url| |
| 63 void OnSafeBrowsingMatchComputed( | 69 // and |redirects| are saved. |
| 64 content::NavigationHandle* navigation_handle, | 70 void OnMainResourceMatchedSafeBrowsingBlacklist( |
| 71 const GURL& url, |
| 65 safe_browsing::SBThreatType threat_type, | 72 safe_browsing::SBThreatType threat_type, |
| 66 safe_browsing::ThreatPatternType threat_type_metadata); | 73 safe_browsing::ThreatPatternType threat_type_metadata); |
| 67 | 74 |
| 68 // Reloads the page and inserts the host of its URL to the whitelist. | 75 // Reloads the page and inserts the host of its URL to the whitelist. |
| 69 void OnReloadRequested(); | 76 void OnReloadRequested(); |
| 70 | 77 |
| 71 // Returns the |ActivationDecision| for the current main frame | 78 // Returns the |ActivationDecision| for the current main frame |
| 72 // document. | 79 // document. |
| 73 ActivationDecision GetActivationDecisionForLastCommittedPageLoad() const { | 80 ActivationDecision GetActivationDecisionForLastCommittedPageLoad() const { |
| 74 return activation_decision_; | 81 return activation_decision_; |
| 75 } | 82 } |
| 76 | 83 |
| 77 // ContentSubresourceFilterThrottleManager::Delegate: | 84 // ContentSubresourceFilterThrottleManager::Delegate: |
| 78 void OnFirstSubresourceLoadDisallowed() override; | 85 void OnFirstSubresourceLoadDisallowed() override; |
| 86 void WillProcessResponse( |
| 87 content::NavigationHandle* navigation_handle) override; |
| 79 | 88 |
| 80 ContentSubresourceFilterThrottleManager* throttle_manager() { | 89 ContentSubresourceFilterThrottleManager* throttle_manager() { |
| 81 return throttle_manager_.get(); | 90 return throttle_manager_.get(); |
| 82 } | 91 } |
| 83 | 92 |
| 84 SubresourceFilterClient* client() { return client_; } | 93 SubresourceFilterClient* client() { return client_; } |
| 85 | 94 |
| 86 private: | 95 private: |
| 87 friend class ContentSubresourceFilterDriverFactoryTest; | 96 friend class ContentSubresourceFilterDriverFactoryTest; |
| 88 friend class safe_browsing::SafeBrowsingServiceTest; | 97 friend class safe_browsing::SafeBrowsingServiceTest; |
| 89 | 98 |
| 90 // content::WebContentsObserver: | 99 // content::WebContentsObserver: |
| 91 void DidStartNavigation( | 100 void DidStartNavigation( |
| 92 content::NavigationHandle* navigation_handle) override; | 101 content::NavigationHandle* navigation_handle) override; |
| 93 void DidFinishNavigation( | 102 void DidFinishNavigation( |
| 94 content::NavigationHandle* navigation_handle) override; | 103 content::NavigationHandle* navigation_handle) override; |
| 95 | 104 |
| 96 // Checks base on the value of |url| and current activation scope if | 105 // Checks base on the value of |url| and current activation scope if |
| 97 // activation signal should be sent. | 106 // activation signal should be sent. |
| 98 void ComputeActivationForMainFrameNavigation( | 107 void ComputeActivationForMainFrameNavigation( |
| 99 content::NavigationHandle* navigation_handle, | 108 content::NavigationHandle* navigation_handle); |
| 100 ActivationList matched_list); | |
| 101 | 109 |
| 102 // Returns whether a main-frame navigation to the given |url| satisfies the | 110 // Returns whether a main-frame navigation to the given |url| satisfies the |
| 103 // activation |conditions| of a given configuration, except for |priority|. | 111 // activation |conditions| of a given configuration, except for |priority|. |
| 104 bool DoesMainFrameURLSatisfyActivationConditions( | 112 bool DoesMainFrameURLSatisfyActivationConditions( |
| 105 const GURL& url, | 113 const GURL& url, |
| 106 const Configuration::ActivationConditions& conditions, | 114 const Configuration::ActivationConditions& conditions) const; |
| 107 ActivationList matched_list) const; | 115 |
| 116 bool DidURLMatchActivationList(const GURL& url, |
| 117 ActivationList activation_list) const; |
| 118 |
| 119 void AddActivationListMatch(const GURL& url, ActivationList match_type); |
| 108 | 120 |
| 109 // Must outlive this class. | 121 // Must outlive this class. |
| 110 SubresourceFilterClient* client_; | 122 SubresourceFilterClient* client_; |
| 111 | 123 |
| 112 std::unique_ptr<ContentSubresourceFilterThrottleManager> throttle_manager_; | 124 std::unique_ptr<ContentSubresourceFilterThrottleManager> throttle_manager_; |
| 113 | 125 |
| 114 // The activation decision corresponding to the most recently _started_ | 126 // The activation decision corresponding to the most recently _started_ |
| 115 // non-same-document navigation in the main frame. | 127 // non-same-document navigation in the main frame. |
| 116 // | 128 // |
| 117 // The value is reset to ActivationDecision::UNKNOWN at the start of each such | 129 // The value is reset to ActivationDecision::UNKNOWN at the start of each such |
| 118 // navigation, and will not be assigned until the navigation successfully | 130 // navigation, and will not be assigned until the navigation successfully |
| 119 // reaches the WillProcessResponse stage (or successfully finishes if | 131 // reaches the WillProcessResponse stage (or successfully finishes if |
| 120 // throttles are not invoked). This means that after a cancelled or otherwise | 132 // throttles are not invoked). This means that after a cancelled or otherwise |
| 121 // unsuccessful navigation, the value will be left at UNKNOWN indefinitely. | 133 // unsuccessful navigation, the value will be left at UNKNOWN indefinitely. |
| 122 ActivationDecision activation_decision_ = | 134 ActivationDecision activation_decision_ = |
| 123 ActivationDecision::ACTIVATION_DISABLED; | 135 ActivationDecision::ACTIVATION_DISABLED; |
| 124 | 136 |
| 125 // The activation options corresponding to the most recently _committed_ | 137 // The activation options corresponding to the most recently _committed_ |
| 126 // non-same-document navigation in the main frame. | 138 // non-same-document navigation in the main frame. |
| 127 // | 139 // |
| 128 // The value corresponding to the previous such navigation will be retained, | 140 // The value corresponding to the previous such navigation will be retained, |
| 129 // and the new value not assigned until a subsequent navigation successfully | 141 // and the new value not assigned until a subsequent navigation successfully |
| 130 // reaches the WillProcessResponse stage (or successfully finishes if | 142 // reaches the WillProcessResponse stage (or successfully finishes if |
| 131 // throttles are not invoked). | 143 // throttles are not invoked). |
| 132 Configuration::ActivationOptions activation_options_; | 144 Configuration::ActivationOptions activation_options_; |
| 133 | 145 |
| 146 URLToActivationListsMap activation_list_matches_; |
| 147 |
| 134 DISALLOW_COPY_AND_ASSIGN(ContentSubresourceFilterDriverFactory); | 148 DISALLOW_COPY_AND_ASSIGN(ContentSubresourceFilterDriverFactory); |
| 135 }; | 149 }; |
| 136 | 150 |
| 137 } // namespace subresource_filter | 151 } // namespace subresource_filter |
| 138 | 152 |
| 139 #endif // COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FIL
TER_DRIVER_FACTORY_H_ | 153 #endif // COMPONENTS_SUBRESOURCE_FILTER_CONTENT_BROWSER_CONTENT_SUBRESOURCE_FIL
TER_DRIVER_FACTORY_H_ |
| OLD | NEW |