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 |