| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 // This test creates a safebrowsing service using test safebrowsing database | 5 // This test creates a safebrowsing service using test safebrowsing database |
| 6 // and a test protocol manager. It is used to test logics in safebrowsing | 6 // and a test protocol manager. It is used to test logics in safebrowsing |
| 7 // service. | 7 // service. |
| 8 | 8 |
| 9 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 9 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
| 10 | 10 |
| 11 #include <algorithm> | |
| 12 #include <utility> | 11 #include <utility> |
| 13 | 12 |
| 14 #include "base/bind.h" | 13 #include "base/bind.h" |
| 15 #include "base/command_line.h" | 14 #include "base/command_line.h" |
| 16 #include "base/files/file_path.h" | 15 #include "base/files/file_path.h" |
| 17 #include "base/files/scoped_temp_dir.h" | 16 #include "base/files/scoped_temp_dir.h" |
| 18 #include "base/macros.h" | 17 #include "base/macros.h" |
| 19 #include "base/memory/ref_counted.h" | 18 #include "base/memory/ref_counted.h" |
| 20 #include "base/metrics/field_trial.h" | 19 #include "base/metrics/field_trial.h" |
| 21 #include "base/path_service.h" | 20 #include "base/path_service.h" |
| 21 #include "base/stl_util.h" |
| 22 #include "base/strings/string_split.h" | 22 #include "base/strings/string_split.h" |
| 23 #include "base/strings/string_util.h" | 23 #include "base/strings/string_util.h" |
| 24 #include "base/strings/utf_string_conversions.h" | 24 #include "base/strings/utf_string_conversions.h" |
| 25 #include "base/test/thread_test_helper.h" | 25 #include "base/test/thread_test_helper.h" |
| 26 #include "base/time/time.h" | 26 #include "base/time/time.h" |
| 27 #include "build/build_config.h" | 27 #include "build/build_config.h" |
| 28 #include "chrome/browser/bookmarks/startup_task_runner_service_factory.h" | 28 #include "chrome/browser/bookmarks/startup_task_runner_service_factory.h" |
| 29 #include "chrome/browser/browser_process.h" | 29 #include "chrome/browser/browser_process.h" |
| 30 #include "chrome/browser/chrome_notification_types.h" | 30 #include "chrome/browser/chrome_notification_types.h" |
| 31 #include "chrome/browser/content_settings/host_content_settings_map_factory.h" | 31 #include "chrome/browser/content_settings/host_content_settings_map_factory.h" |
| (...skipping 280 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 312 std::vector<SBPrefix>* prefix_hits) { | 312 std::vector<SBPrefix>* prefix_hits) { |
| 313 bool hit = false; | 313 bool hit = false; |
| 314 for (const GURL& url : urls) { | 314 for (const GURL& url : urls) { |
| 315 base::hash_map<std::string, Hits>::const_iterator badurls_it = | 315 base::hash_map<std::string, Hits>::const_iterator badurls_it = |
| 316 badurls_.find(url.spec()); | 316 badurls_.find(url.spec()); |
| 317 | 317 |
| 318 if (badurls_it == badurls_.end()) | 318 if (badurls_it == badurls_.end()) |
| 319 continue; | 319 continue; |
| 320 | 320 |
| 321 std::vector<int> list_ids_for_url = badurls_it->second.list_ids; | 321 std::vector<int> list_ids_for_url = badurls_it->second.list_ids; |
| 322 if (std::find(list_ids_for_url.begin(), list_ids_for_url.end(), list_id0) | 322 if (ContainsValue(list_ids_for_url, list_id0) || |
| 323 != list_ids_for_url.end() || | 323 ContainsValue(list_ids_for_url, list_id1)) { |
| 324 std::find(list_ids_for_url.begin(), list_ids_for_url.end(), list_id1) | |
| 325 != list_ids_for_url.end()) { | |
| 326 prefix_hits->insert(prefix_hits->end(), | 324 prefix_hits->insert(prefix_hits->end(), |
| 327 badurls_it->second.prefix_hits.begin(), | 325 badurls_it->second.prefix_hits.begin(), |
| 328 badurls_it->second.prefix_hits.end()); | 326 badurls_it->second.prefix_hits.end()); |
| 329 hit = true; | 327 hit = true; |
| 330 } | 328 } |
| 331 } | 329 } |
| 332 return hit; | 330 return hit; |
| 333 } | 331 } |
| 334 | 332 |
| 335 std::vector<GURL> UrlsForHashes(const std::vector<SBFullHash>& full_hashes) { | 333 std::vector<GURL> UrlsForHashes(const std::vector<SBFullHash>& full_hashes) { |
| (...skipping 1335 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1671 chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, | 1669 chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, |
| 1672 content::Source<SafeBrowsingDatabaseManager>( | 1670 content::Source<SafeBrowsingDatabaseManager>( |
| 1673 sb_service_->database_manager().get())); | 1671 sb_service_->database_manager().get())); |
| 1674 BrowserThread::PostTask( | 1672 BrowserThread::PostTask( |
| 1675 BrowserThread::IO, FROM_HERE, | 1673 BrowserThread::IO, FROM_HERE, |
| 1676 base::Bind(&SafeBrowsingDatabaseManagerCookieTest::ForceUpdate, this)); | 1674 base::Bind(&SafeBrowsingDatabaseManagerCookieTest::ForceUpdate, this)); |
| 1677 observer.Wait(); | 1675 observer.Wait(); |
| 1678 } | 1676 } |
| 1679 | 1677 |
| 1680 } // namespace safe_browsing | 1678 } // namespace safe_browsing |
| OLD | NEW |