| 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/safe_browsing/services_delegate_impl.h" | 5 #include "chrome/browser/safe_browsing/services_delegate_impl.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/command_line.h" | 9 #include "base/command_line.h" |
| 10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
| 11 #include "base/metrics/field_trial.h" |
| 11 #include "base/strings/string_util.h" | 12 #include "base/strings/string_util.h" |
| 12 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 13 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
| 13 #include "chrome/common/chrome_switches.h" | 14 #include "chrome/common/chrome_switches.h" |
| 14 #include "components/safe_browsing_db/v4_local_database_manager.h" | 15 #include "components/safe_browsing_db/v4_local_database_manager.h" |
| 15 #include "content/public/browser/browser_thread.h" | 16 #include "content/public/browser/browser_thread.h" |
| 16 | 17 |
| 17 namespace safe_browsing { | 18 namespace safe_browsing { |
| 18 | 19 |
| 20 #ifdef NDEBUG |
| 21 namespace { |
| 22 const base::Feature kSafeBrowsingV4LocalDatabaseManagerEnabled { |
| 23 "SafeBrowsingV4LocalDatabaseManagerEnabled", |
| 24 base::FEATURE_DISABLED_BY_DEFAULT |
| 25 }; |
| 26 } // namespace |
| 27 #endif |
| 28 |
| 19 // static | 29 // static |
| 20 std::unique_ptr<ServicesDelegate> ServicesDelegate::Create( | 30 std::unique_ptr<ServicesDelegate> ServicesDelegate::Create( |
| 21 SafeBrowsingService* safe_browsing_service) { | 31 SafeBrowsingService* safe_browsing_service) { |
| 22 return base::WrapUnique( | 32 return base::WrapUnique( |
| 23 new ServicesDelegateImpl(safe_browsing_service, nullptr)); | 33 new ServicesDelegateImpl(safe_browsing_service, nullptr)); |
| 24 } | 34 } |
| 25 | 35 |
| 26 // static | 36 // static |
| 27 std::unique_ptr<ServicesDelegate> ServicesDelegate::CreateForTest( | 37 std::unique_ptr<ServicesDelegate> ServicesDelegate::CreateForTest( |
| 28 SafeBrowsingService* safe_browsing_service, | 38 SafeBrowsingService* safe_browsing_service, |
| (...skipping 18 matching lines...) Expand all Loading... |
| 47 net::URLRequestContextGetter* context_getter) { | 57 net::URLRequestContextGetter* context_getter) { |
| 48 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); | 58 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 49 #if defined(SAFE_BROWSING_CSD) | 59 #if defined(SAFE_BROWSING_CSD) |
| 50 if (!base::CommandLine::ForCurrentProcess()->HasSwitch( | 60 if (!base::CommandLine::ForCurrentProcess()->HasSwitch( |
| 51 switches::kDisableClientSidePhishingDetection)) { | 61 switches::kDisableClientSidePhishingDetection)) { |
| 52 csd_service_.reset(ClientSideDetectionService::Create(context_getter)); | 62 csd_service_.reset(ClientSideDetectionService::Create(context_getter)); |
| 53 } | 63 } |
| 54 #endif // defined(SAFE_BROWSING_CSD) | 64 #endif // defined(SAFE_BROWSING_CSD) |
| 55 } | 65 } |
| 56 | 66 |
| 57 const scoped_refptr<V4LocalDatabaseManager>& | |
| 58 ServicesDelegateImpl::v4_local_database_manager() const { | |
| 59 return v4_local_database_manager_; | |
| 60 } | |
| 61 | |
| 62 void ServicesDelegateImpl::Initialize() { | 67 void ServicesDelegateImpl::Initialize() { |
| 63 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); | 68 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 64 download_service_.reset( | 69 download_service_.reset( |
| 65 (services_creator_ && | 70 (services_creator_ && |
| 66 services_creator_->CanCreateDownloadProtectionService()) | 71 services_creator_->CanCreateDownloadProtectionService()) |
| 67 ? services_creator_->CreateDownloadProtectionService() | 72 ? services_creator_->CreateDownloadProtectionService() |
| 68 : CreateDownloadProtectionService()); | 73 : CreateDownloadProtectionService()); |
| 69 incident_service_.reset( | 74 incident_service_.reset( |
| 70 (services_creator_ && | 75 (services_creator_ && |
| 71 services_creator_->CanCreateIncidentReportingService()) | 76 services_creator_->CanCreateIncidentReportingService()) |
| 72 ? services_creator_->CreateIncidentReportingService() | 77 ? services_creator_->CreateIncidentReportingService() |
| 73 : CreateIncidentReportingService()); | 78 : CreateIncidentReportingService()); |
| 74 resource_request_detector_.reset( | 79 resource_request_detector_.reset( |
| 75 (services_creator_ && | 80 (services_creator_ && |
| 76 services_creator_->CanCreateResourceRequestDetector()) | 81 services_creator_->CanCreateResourceRequestDetector()) |
| 77 ? services_creator_->CreateResourceRequestDetector() | 82 ? services_creator_->CreateResourceRequestDetector() |
| 78 : CreateResourceRequestDetector()); | 83 : CreateResourceRequestDetector()); |
| 79 | 84 |
| 80 v4_local_database_manager_ = | 85 if (IsV4LocalDatabaseManagerEnabled()) { |
| 81 V4LocalDatabaseManager::Create(SafeBrowsingService::GetBaseFilename()); | 86 v4_local_database_manager_ = CreateV4LocalDatabaseManager(); |
| 87 } |
| 82 } | 88 } |
| 83 | 89 |
| 84 void ServicesDelegateImpl::ShutdownServices() { | 90 void ServicesDelegateImpl::ShutdownServices() { |
| 85 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); | 91 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 86 // The IO thread is going away, so make sure the ClientSideDetectionService | 92 // The IO thread is going away, so make sure the ClientSideDetectionService |
| 87 // dtor executes now since it may call the dtor of URLFetcher which relies | 93 // dtor executes now since it may call the dtor of URLFetcher which relies |
| 88 // on it. | 94 // on it. |
| 89 csd_service_.reset(); | 95 csd_service_.reset(); |
| 90 | 96 |
| 91 resource_request_detector_.reset(); | 97 resource_request_detector_.reset(); |
| (...skipping 76 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 168 v4_config); | 174 v4_config); |
| 169 } | 175 } |
| 170 } | 176 } |
| 171 | 177 |
| 172 void ServicesDelegateImpl::StopOnIOThread(bool shutdown) { | 178 void ServicesDelegateImpl::StopOnIOThread(bool shutdown) { |
| 173 if (v4_local_database_manager_.get()) { | 179 if (v4_local_database_manager_.get()) { |
| 174 v4_local_database_manager_->StopOnIOThread(shutdown); | 180 v4_local_database_manager_->StopOnIOThread(shutdown); |
| 175 } | 181 } |
| 176 } | 182 } |
| 177 | 183 |
| 184 V4LocalDatabaseManager* ServicesDelegateImpl::CreateV4LocalDatabaseManager() { |
| 185 return new V4LocalDatabaseManager(SafeBrowsingService::GetBaseFilename()); |
| 186 } |
| 187 |
| 188 bool ServicesDelegateImpl::IsV4LocalDatabaseManagerEnabled() { |
| 189 #ifndef NDEBUG |
| 190 return true; |
| 191 #else |
| 192 return base::FeatureList::IsEnabled( |
| 193 kSafeBrowsingV4LocalDatabaseManagerEnabled); |
| 194 #endif |
| 195 } |
| 196 |
| 178 } // namespace safe_browsing | 197 } // namespace safe_browsing |
| OLD | NEW |