Index: chrome/browser/subresource_filter/chrome_subresource_filter_client.cc |
diff --git a/chrome/browser/subresource_filter/chrome_subresource_filter_client.cc b/chrome/browser/subresource_filter/chrome_subresource_filter_client.cc |
index 60ff494e08cb30b2b9d75ca6ca5998cb58eb2b59..377807c8fcd280fc90a1f7e21715e44c3c38f996 100644 |
--- a/chrome/browser/subresource_filter/chrome_subresource_filter_client.cc |
+++ b/chrome/browser/subresource_filter/chrome_subresource_filter_client.cc |
@@ -5,19 +5,26 @@ |
#include "chrome/browser/subresource_filter/chrome_subresource_filter_client.h" |
#include <string> |
+#include <utility> |
#include "base/metrics/histogram_macros.h" |
+ |
engedy
2017/05/02 18:21:54
nit: Accidental blank line?
Charlie Harrison
2017/05/02 20:08:49
Yep, removed.
|
#include "chrome/browser/browser_process.h" |
#include "chrome/browser/content_settings/host_content_settings_map_factory.h" |
#include "chrome/browser/content_settings/tab_specific_content_settings.h" |
#include "chrome/browser/infobars/infobar_service.h" |
#include "chrome/browser/profiles/profile.h" |
+#include "chrome/browser/safe_browsing/safe_browsing_service.h" |
#include "chrome/browser/subresource_filter/subresource_filter_profile_context_factory.h" |
#include "chrome/browser/ui/android/content_settings/subresource_filter_infobar_delegate.h" |
#include "components/content_settings/core/browser/host_content_settings_map.h" |
#include "components/content_settings/core/common/content_settings_types.h" |
+#include "components/safe_browsing_db/database_manager.h" |
+#include "components/safe_browsing_db/v4_feature_list.h" |
#include "components/subresource_filter/content/browser/content_ruleset_service.h" |
#include "components/subresource_filter/content/browser/content_subresource_filter_driver_factory.h" |
+#include "components/subresource_filter/content/browser/subresource_filter_safe_browsing_activation_throttle.h" |
+#include "components/subresource_filter/core/browser/subresource_filter_features.h" |
#include "content/public/browser/navigation_handle.h" |
DEFINE_WEB_CONTENTS_USER_DATA_KEY(ChromeSubresourceFilterClient); |
@@ -35,6 +42,33 @@ ChromeSubresourceFilterClient::ChromeSubresourceFilterClient( |
ChromeSubresourceFilterClient::~ChromeSubresourceFilterClient() {} |
+void ChromeSubresourceFilterClient::MaybeAppendNavigationThrottles( |
+ content::NavigationHandle* navigation_handle, |
+ std::vector<std::unique_ptr<content::NavigationThrottle>>* throttles) { |
+ 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
|
+ safe_browsing::SafeBrowsingService* safe_browsing_service = |
+ g_browser_process->safe_browsing_service(); |
+ scoped_refptr<safe_browsing::SafeBrowsingDatabaseManager> database_manager; |
+ if (safe_browsing_service && |
+ safe_browsing_service->database_manager()->IsSupported() && |
+ safe_browsing::V4FeatureList::GetV4UsageStatus() == |
+ safe_browsing::V4FeatureList::V4UsageStatus::V4_ONLY) { |
+ database_manager = safe_browsing_service->database_manager(); |
+ } |
+ |
+ throttles->push_back( |
+ 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.
|
+ SubresourceFilterSafeBrowsingActivationThrottle>( |
+ navigation_handle, std::move(database_manager))); |
+ } |
+ |
+ auto* driver_factory = |
+ subresource_filter::ContentSubresourceFilterDriverFactory:: |
+ FromWebContents(navigation_handle->GetWebContents()); |
+ driver_factory->throttle_manager()->MaybeAppendNavigationThrottles( |
+ navigation_handle, throttles); |
+} |
+ |
void ChromeSubresourceFilterClient::ToggleNotificationVisibility( |
bool visibility) { |
if (did_show_ui_for_navigation_ && visibility) |