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