| 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 |