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 #include "chrome/browser/subresource_filter/chrome_subresource_filter_client.h" | 5 #include "chrome/browser/subresource_filter/chrome_subresource_filter_client.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 #include <utility> | |
8 | 9 |
9 #include "base/metrics/histogram_macros.h" | 10 #include "base/metrics/histogram_macros.h" |
11 | |
engedy
2017/05/02 18:21:54
nit: Accidental blank line?
Charlie Harrison
2017/05/02 20:08:49
Yep, removed.
| |
10 #include "chrome/browser/browser_process.h" | 12 #include "chrome/browser/browser_process.h" |
11 #include "chrome/browser/content_settings/host_content_settings_map_factory.h" | 13 #include "chrome/browser/content_settings/host_content_settings_map_factory.h" |
12 #include "chrome/browser/content_settings/tab_specific_content_settings.h" | 14 #include "chrome/browser/content_settings/tab_specific_content_settings.h" |
13 #include "chrome/browser/infobars/infobar_service.h" | 15 #include "chrome/browser/infobars/infobar_service.h" |
14 #include "chrome/browser/profiles/profile.h" | 16 #include "chrome/browser/profiles/profile.h" |
17 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | |
15 #include "chrome/browser/subresource_filter/subresource_filter_profile_context_f actory.h" | 18 #include "chrome/browser/subresource_filter/subresource_filter_profile_context_f actory.h" |
16 #include "chrome/browser/ui/android/content_settings/subresource_filter_infobar_ delegate.h" | 19 #include "chrome/browser/ui/android/content_settings/subresource_filter_infobar_ delegate.h" |
17 #include "components/content_settings/core/browser/host_content_settings_map.h" | 20 #include "components/content_settings/core/browser/host_content_settings_map.h" |
18 #include "components/content_settings/core/common/content_settings_types.h" | 21 #include "components/content_settings/core/common/content_settings_types.h" |
22 #include "components/safe_browsing_db/database_manager.h" | |
23 #include "components/safe_browsing_db/v4_feature_list.h" | |
19 #include "components/subresource_filter/content/browser/content_ruleset_service. h" | 24 #include "components/subresource_filter/content/browser/content_ruleset_service. h" |
20 #include "components/subresource_filter/content/browser/content_subresource_filt er_driver_factory.h" | 25 #include "components/subresource_filter/content/browser/content_subresource_filt er_driver_factory.h" |
26 #include "components/subresource_filter/content/browser/subresource_filter_safe_ browsing_activation_throttle.h" | |
27 #include "components/subresource_filter/core/browser/subresource_filter_features .h" | |
21 #include "content/public/browser/navigation_handle.h" | 28 #include "content/public/browser/navigation_handle.h" |
22 | 29 |
23 DEFINE_WEB_CONTENTS_USER_DATA_KEY(ChromeSubresourceFilterClient); | 30 DEFINE_WEB_CONTENTS_USER_DATA_KEY(ChromeSubresourceFilterClient); |
24 | 31 |
25 ChromeSubresourceFilterClient::ChromeSubresourceFilterClient( | 32 ChromeSubresourceFilterClient::ChromeSubresourceFilterClient( |
26 content::WebContents* web_contents) | 33 content::WebContents* web_contents) |
27 : web_contents_(web_contents), did_show_ui_for_navigation_(false) { | 34 : web_contents_(web_contents), did_show_ui_for_navigation_(false) { |
28 DCHECK(web_contents); | 35 DCHECK(web_contents); |
29 SubresourceFilterProfileContextFactory::EnsureForProfile( | 36 SubresourceFilterProfileContextFactory::EnsureForProfile( |
30 Profile::FromBrowserContext(web_contents_->GetBrowserContext())); | 37 Profile::FromBrowserContext(web_contents_->GetBrowserContext())); |
31 | 38 |
32 subresource_filter::ContentSubresourceFilterDriverFactory:: | 39 subresource_filter::ContentSubresourceFilterDriverFactory:: |
33 CreateForWebContents(web_contents, this); | 40 CreateForWebContents(web_contents, this); |
34 } | 41 } |
35 | 42 |
36 ChromeSubresourceFilterClient::~ChromeSubresourceFilterClient() {} | 43 ChromeSubresourceFilterClient::~ChromeSubresourceFilterClient() {} |
37 | 44 |
45 void ChromeSubresourceFilterClient::MaybeAppendNavigationThrottles( | |
46 content::NavigationHandle* navigation_handle, | |
47 std::vector<std::unique_ptr<content::NavigationThrottle>>* throttles) { | |
48 if (navigation_handle->IsInMainFrame()) { | |
engedy
2017/05/02 18:21:54
Given that we are delaying the navigation, do we w
Charlie Harrison
2017/05/02 20:08:49
Done, using the activation_scope of the current ac
| |
49 safe_browsing::SafeBrowsingService* safe_browsing_service = | |
50 g_browser_process->safe_browsing_service(); | |
51 scoped_refptr<safe_browsing::SafeBrowsingDatabaseManager> database_manager; | |
52 if (safe_browsing_service && | |
53 safe_browsing_service->database_manager()->IsSupported() && | |
54 safe_browsing::V4FeatureList::GetV4UsageStatus() == | |
55 safe_browsing::V4FeatureList::V4UsageStatus::V4_ONLY) { | |
56 database_manager = safe_browsing_service->database_manager(); | |
57 } | |
58 | |
59 throttles->push_back( | |
60 base::MakeUnique<subresource_filter:: | |
engedy
2017/05/02 18:21:54
nit: #include "base/memory/ptr_util.h"
Charlie Harrison
2017/05/02 20:08:49
Done.
| |
61 SubresourceFilterSafeBrowsingActivationThrottle>( | |
62 navigation_handle, std::move(database_manager))); | |
63 } | |
64 | |
65 auto* driver_factory = | |
66 subresource_filter::ContentSubresourceFilterDriverFactory:: | |
67 FromWebContents(navigation_handle->GetWebContents()); | |
68 driver_factory->throttle_manager()->MaybeAppendNavigationThrottles( | |
69 navigation_handle, throttles); | |
70 } | |
71 | |
38 void ChromeSubresourceFilterClient::ToggleNotificationVisibility( | 72 void ChromeSubresourceFilterClient::ToggleNotificationVisibility( |
39 bool visibility) { | 73 bool visibility) { |
40 if (did_show_ui_for_navigation_ && visibility) | 74 if (did_show_ui_for_navigation_ && visibility) |
41 return; | 75 return; |
42 | 76 |
43 did_show_ui_for_navigation_ = visibility; | 77 did_show_ui_for_navigation_ = visibility; |
44 TabSpecificContentSettings* content_settings = | 78 TabSpecificContentSettings* content_settings = |
45 TabSpecificContentSettings::FromWebContents(web_contents_); | 79 TabSpecificContentSettings::FromWebContents(web_contents_); |
46 | 80 |
47 // |visibility| is false when a new navigation starts. | 81 // |visibility| is false when a new navigation starts. |
(...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
105 url, url, ContentSettingsType::CONTENT_SETTINGS_TYPE_SUBRESOURCE_FILTER, | 139 url, url, ContentSettingsType::CONTENT_SETTINGS_TYPE_SUBRESOURCE_FILTER, |
106 std::string()); | 140 std::string()); |
107 } | 141 } |
108 | 142 |
109 subresource_filter::VerifiedRulesetDealer::Handle* | 143 subresource_filter::VerifiedRulesetDealer::Handle* |
110 ChromeSubresourceFilterClient::GetRulesetDealer() { | 144 ChromeSubresourceFilterClient::GetRulesetDealer() { |
111 subresource_filter::ContentRulesetService* ruleset_service = | 145 subresource_filter::ContentRulesetService* ruleset_service = |
112 g_browser_process->subresource_filter_ruleset_service(); | 146 g_browser_process->subresource_filter_ruleset_service(); |
113 return ruleset_service ? ruleset_service->ruleset_dealer() : nullptr; | 147 return ruleset_service ? ruleset_service->ruleset_dealer() : nullptr; |
114 } | 148 } |
OLD | NEW |