Chromium Code Reviews| 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) |