OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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/extensions/fake_safe_browsing_database_manager.h" | 5 #include "chrome/browser/extensions/fake_safe_browsing_database_manager.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <iterator> | 8 #include <iterator> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/bind_helpers.h" | 11 #include "base/bind_helpers.h" |
12 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
13 #include "base/run_loop.h" | 13 #include "base/run_loop.h" |
14 #include "base/thread_task_runner_handle.h" | 14 #include "base/thread_task_runner_handle.h" |
15 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 15 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
16 #include "chrome/browser/safe_browsing/safe_browsing_util.h" | 16 #include "chrome/browser/safe_browsing/safe_browsing_util.h" |
17 | 17 |
18 namespace extensions { | 18 namespace extensions { |
19 | 19 |
20 FakeSafeBrowsingDatabaseManager::FakeSafeBrowsingDatabaseManager(bool enabled) | 20 FakeSafeBrowsingDatabaseManager::FakeSafeBrowsingDatabaseManager(bool enabled) |
21 : SafeBrowsingDatabaseManager( | 21 : LocalSafeBrowsingDatabaseManager( |
22 make_scoped_refptr(SafeBrowsingService::CreateSafeBrowsingService())), | 22 make_scoped_refptr(SafeBrowsingService::CreateSafeBrowsingService())), |
23 enabled_(enabled) { | 23 enabled_(enabled) { |
24 } | 24 } |
25 | 25 |
26 FakeSafeBrowsingDatabaseManager::~FakeSafeBrowsingDatabaseManager() { | 26 FakeSafeBrowsingDatabaseManager::~FakeSafeBrowsingDatabaseManager() { |
27 } | 27 } |
28 | 28 |
29 FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::Enable() { | 29 FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::Enable() { |
30 enabled_ = true; | 30 enabled_ = true; |
31 return *this; | 31 return *this; |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
77 return *this; | 77 return *this; |
78 } | 78 } |
79 | 79 |
80 FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::RemoveUnsafe( | 80 FakeSafeBrowsingDatabaseManager& FakeSafeBrowsingDatabaseManager::RemoveUnsafe( |
81 const std::string& a) { | 81 const std::string& a) { |
82 unsafe_ids_.erase(a); | 82 unsafe_ids_.erase(a); |
83 return *this; | 83 return *this; |
84 } | 84 } |
85 | 85 |
86 void FakeSafeBrowsingDatabaseManager::NotifyUpdate() { | 86 void FakeSafeBrowsingDatabaseManager::NotifyUpdate() { |
87 SafeBrowsingDatabaseManager::NotifyDatabaseUpdateFinished(true); | 87 LocalSafeBrowsingDatabaseManager::NotifyDatabaseUpdateFinished(true); |
88 } | 88 } |
89 | 89 |
90 bool FakeSafeBrowsingDatabaseManager::CheckExtensionIDs( | 90 bool FakeSafeBrowsingDatabaseManager::CheckExtensionIDs( |
91 const std::set<std::string>& extension_ids, | 91 const std::set<std::string>& extension_ids, |
92 Client* client) { | 92 Client* client) { |
93 if (!enabled_) | 93 if (!enabled_) |
94 return true; | 94 return true; |
95 | 95 |
96 // Need to construct the full SafeBrowsingCheck rather than calling | 96 // Need to construct the full SafeBrowsingCheck rather than calling |
97 // OnCheckExtensionsResult directly because it's protected. Grr! | 97 // OnCheckExtensionsResult directly because it's protected. Grr! |
(...skipping 13 matching lines...) Expand all Loading... |
111 std::vector<SBThreatType>(1, SB_THREAT_TYPE_EXTENSION))); | 111 std::vector<SBThreatType>(1, SB_THREAT_TYPE_EXTENSION))); |
112 | 112 |
113 for (size_t i = 0; i < extension_ids_vector.size(); ++i) { | 113 for (size_t i = 0; i < extension_ids_vector.size(); ++i) { |
114 const std::string& extension_id = extension_ids_vector[i]; | 114 const std::string& extension_id = extension_ids_vector[i]; |
115 if (unsafe_ids_.count(extension_id)) | 115 if (unsafe_ids_.count(extension_id)) |
116 safe_browsing_check->full_hash_results[i] = SB_THREAT_TYPE_EXTENSION; | 116 safe_browsing_check->full_hash_results[i] = SB_THREAT_TYPE_EXTENSION; |
117 } | 117 } |
118 | 118 |
119 base::ThreadTaskRunnerHandle::Get()->PostTask( | 119 base::ThreadTaskRunnerHandle::Get()->PostTask( |
120 FROM_HERE, | 120 FROM_HERE, |
121 base::Bind(&FakeSafeBrowsingDatabaseManager::OnSafeBrowsingResult, | 121 base::Bind(&FakeSafeBrowsingDatabaseManager::OnSafeBrowsingResult, this, |
122 this, | 122 base::Passed(&safe_browsing_check))); |
123 base::Passed(&safe_browsing_check), | |
124 client)); | |
125 return false; | 123 return false; |
126 } | 124 } |
127 | 125 |
128 void FakeSafeBrowsingDatabaseManager::OnSafeBrowsingResult( | 126 void FakeSafeBrowsingDatabaseManager::OnSafeBrowsingResult( |
129 scoped_ptr<SafeBrowsingCheck> result, | 127 scoped_ptr<SafeBrowsingCheck> result) { |
130 Client* client) { | 128 result->OnSafeBrowsingResult(); |
131 client->OnSafeBrowsingResult(*result); | |
132 } | 129 } |
133 | 130 |
134 } // namespace extensions | 131 } // namespace extensions |
OLD | NEW |