| 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 #include "chrome/browser/safe_browsing/download_protection_service.h" | 5 #include "chrome/browser/safe_browsing/download_protection_service.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 #include <stdint.h> | 8 #include <stdint.h> |
| 9 | 9 |
| 10 #include <map> | 10 #include <map> |
| (...skipping 15 matching lines...) Expand all Loading... |
| 26 #include "base/sha1.h" | 26 #include "base/sha1.h" |
| 27 #include "base/single_thread_task_runner.h" | 27 #include "base/single_thread_task_runner.h" |
| 28 #include "base/strings/string_number_conversions.h" | 28 #include "base/strings/string_number_conversions.h" |
| 29 #include "base/threading/sequenced_worker_pool.h" | 29 #include "base/threading/sequenced_worker_pool.h" |
| 30 #include "base/threading/thread_task_runner_handle.h" | 30 #include "base/threading/thread_task_runner_handle.h" |
| 31 #include "chrome/browser/history/history_service_factory.h" | 31 #include "chrome/browser/history/history_service_factory.h" |
| 32 #include "chrome/browser/safe_browsing/download_feedback_service.h" | 32 #include "chrome/browser/safe_browsing/download_feedback_service.h" |
| 33 #include "chrome/browser/safe_browsing/incident_reporting/incident_reporting_ser
vice.h" | 33 #include "chrome/browser/safe_browsing/incident_reporting/incident_reporting_ser
vice.h" |
| 34 #include "chrome/browser/safe_browsing/local_database_manager.h" | 34 #include "chrome/browser/safe_browsing/local_database_manager.h" |
| 35 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 35 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
| 36 #include "chrome/common/chrome_switches.h" | |
| 37 #include "chrome/common/safe_browsing/binary_feature_extractor.h" | 36 #include "chrome/common/safe_browsing/binary_feature_extractor.h" |
| 38 #include "chrome/common/safe_browsing/csd.pb.h" | 37 #include "chrome/common/safe_browsing/csd.pb.h" |
| 39 #include "chrome/common/safe_browsing/file_type_policies_test_util.h" | 38 #include "chrome/common/safe_browsing/file_type_policies_test_util.h" |
| 40 #include "chrome/test/base/testing_profile.h" | 39 #include "chrome/test/base/testing_profile.h" |
| 41 #include "components/history/core/browser/history_service.h" | 40 #include "components/history/core/browser/history_service.h" |
| 42 #include "components/prefs/pref_service.h" | 41 #include "components/prefs/pref_service.h" |
| 42 #include "components/safe_browsing/common/safebrowsing_switches.h" |
| 43 #include "components/safe_browsing_db/database_manager.h" | 43 #include "components/safe_browsing_db/database_manager.h" |
| 44 #include "components/safe_browsing_db/safe_browsing_prefs.h" | 44 #include "components/safe_browsing_db/safe_browsing_prefs.h" |
| 45 #include "components/safe_browsing_db/test_database_manager.h" | 45 #include "components/safe_browsing_db/test_database_manager.h" |
| 46 #include "content/public/browser/download_danger_type.h" | 46 #include "content/public/browser/download_danger_type.h" |
| 47 #include "content/public/browser/page_navigator.h" | 47 #include "content/public/browser/page_navigator.h" |
| 48 #include "content/public/test/mock_download_item.h" | 48 #include "content/public/test/mock_download_item.h" |
| 49 #include "content/public/test/test_browser_thread_bundle.h" | 49 #include "content/public/test/test_browser_thread_bundle.h" |
| 50 #include "content/public/test/test_utils.h" | 50 #include "content/public/test/test_utils.h" |
| 51 #include "net/base/url_util.h" | 51 #include "net/base/url_util.h" |
| 52 #include "net/cert/x509_certificate.h" | 52 #include "net/cert/x509_certificate.h" |
| (...skipping 2192 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2245 // Matches unsigned.exe within zipfile_one_unsigned_binary.zip | 2245 // Matches unsigned.exe within zipfile_one_unsigned_binary.zip |
| 2246 : blacklisted_hash_hex_("1e954d9ce0389e2ba7447216f21761f98d1e6540c2abecdbe
cff570e36c493db") {} | 2246 : blacklisted_hash_hex_("1e954d9ce0389e2ba7447216f21761f98d1e6540c2abecdbe
cff570e36c493db") {} |
| 2247 | 2247 |
| 2248 void SetUp() override { | 2248 void SetUp() override { |
| 2249 std::vector<uint8_t> bytes; | 2249 std::vector<uint8_t> bytes; |
| 2250 ASSERT_TRUE(base::HexStringToBytes(blacklisted_hash_hex_, &bytes) && | 2250 ASSERT_TRUE(base::HexStringToBytes(blacklisted_hash_hex_, &bytes) && |
| 2251 bytes.size() == 32); | 2251 bytes.size() == 32); |
| 2252 blacklisted_hash_ = std::string(bytes.begin(), bytes.end()); | 2252 blacklisted_hash_ = std::string(bytes.begin(), bytes.end()); |
| 2253 | 2253 |
| 2254 base::CommandLine::ForCurrentProcess()->AppendSwitchASCII( | 2254 base::CommandLine::ForCurrentProcess()->AppendSwitchASCII( |
| 2255 switches::kSbManualDownloadBlacklist, | 2255 safe_browsing::switches::kSbManualDownloadBlacklist, |
| 2256 blacklisted_hash_hex_); | 2256 blacklisted_hash_hex_); |
| 2257 | 2257 |
| 2258 DownloadProtectionServiceTest::SetUp(); | 2258 DownloadProtectionServiceTest::SetUp(); |
| 2259 } | 2259 } |
| 2260 | 2260 |
| 2261 // Hex 64 chars | 2261 // Hex 64 chars |
| 2262 const std::string blacklisted_hash_hex_; | 2262 const std::string blacklisted_hash_hex_; |
| 2263 // Binary 32 bytes | 2263 // Binary 32 bytes |
| 2264 std::string blacklisted_hash_; | 2264 std::string blacklisted_hash_; |
| 2265 }; | 2265 }; |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2314 &item, base::Bind(&DownloadProtectionServiceTest::CheckDoneCallback, | 2314 &item, base::Bind(&DownloadProtectionServiceTest::CheckDoneCallback, |
| 2315 base::Unretained(this), run_loop.QuitClosure())); | 2315 base::Unretained(this), run_loop.QuitClosure())); |
| 2316 run_loop.Run(); | 2316 run_loop.Run(); |
| 2317 | 2317 |
| 2318 EXPECT_FALSE(HasClientDownloadRequest()); | 2318 EXPECT_FALSE(HasClientDownloadRequest()); |
| 2319 // Overriden by flag: | 2319 // Overriden by flag: |
| 2320 EXPECT_TRUE(IsResult(DownloadProtectionService::DANGEROUS)); | 2320 EXPECT_TRUE(IsResult(DownloadProtectionService::DANGEROUS)); |
| 2321 } | 2321 } |
| 2322 | 2322 |
| 2323 } // namespace safe_browsing | 2323 } // namespace safe_browsing |
| OLD | NEW |