| 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 215 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 226 LocalSafeBrowsingDatabaseManager::SafeBrowsingCheck* check) { | 226 LocalSafeBrowsingDatabaseManager::SafeBrowsingCheck* check) { |
| 227 check->OnSafeBrowsingResult(); | 227 check->OnSafeBrowsingResult(); |
| 228 } | 228 } |
| 229 | 229 |
| 230 ACTION_P(CheckDownloadUrlDone, threat_type) { | 230 ACTION_P(CheckDownloadUrlDone, threat_type) { |
| 231 // TODO(nparker): Remove use of SafeBrowsingCheck and instead call | 231 // TODO(nparker): Remove use of SafeBrowsingCheck and instead call |
| 232 // client->OnCheckDownloadUrlResult(..) directly. | 232 // client->OnCheckDownloadUrlResult(..) directly. |
| 233 LocalSafeBrowsingDatabaseManager::SafeBrowsingCheck* check = | 233 LocalSafeBrowsingDatabaseManager::SafeBrowsingCheck* check = |
| 234 new LocalSafeBrowsingDatabaseManager::SafeBrowsingCheck( | 234 new LocalSafeBrowsingDatabaseManager::SafeBrowsingCheck( |
| 235 arg0, std::vector<SBFullHash>(), arg1, BINURL, | 235 arg0, std::vector<SBFullHash>(), arg1, BINURL, |
| 236 std::vector<SBThreatType>(1, SB_THREAT_TYPE_BINARY_MALWARE_URL)); | 236 std::vector<SBThreatType>(1, SB_THREAT_TYPE_URL_BINARY_MALWARE)); |
| 237 for (size_t i = 0; i < check->url_results.size(); ++i) | 237 for (size_t i = 0; i < check->url_results.size(); ++i) |
| 238 check->url_results[i] = threat_type; | 238 check->url_results[i] = threat_type; |
| 239 BrowserThread::PostTask( | 239 BrowserThread::PostTask( |
| 240 BrowserThread::IO, FROM_HERE, | 240 BrowserThread::IO, FROM_HERE, |
| 241 base::BindOnce(&OnSafeBrowsingResult, base::Owned(check))); | 241 base::BindOnce(&OnSafeBrowsingResult, base::Owned(check))); |
| 242 } | 242 } |
| 243 | 243 |
| 244 class DownloadProtectionServiceTest : public testing::Test { | 244 class DownloadProtectionServiceTest : public testing::Test { |
| 245 protected: | 245 protected: |
| 246 DownloadProtectionServiceTest() | 246 DownloadProtectionServiceTest() |
| (...skipping 1543 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1790 download_service_->CheckDownloadUrl( | 1790 download_service_->CheckDownloadUrl( |
| 1791 &item, base::Bind(&DownloadProtectionServiceTest::CheckDoneCallback, | 1791 &item, base::Bind(&DownloadProtectionServiceTest::CheckDoneCallback, |
| 1792 base::Unretained(this), run_loop.QuitClosure())); | 1792 base::Unretained(this), run_loop.QuitClosure())); |
| 1793 run_loop.Run(); | 1793 run_loop.Run(); |
| 1794 EXPECT_TRUE(IsResult(DownloadProtectionService::SAFE)); | 1794 EXPECT_TRUE(IsResult(DownloadProtectionService::SAFE)); |
| 1795 Mock::VerifyAndClearExpectations(sb_service_.get()); | 1795 Mock::VerifyAndClearExpectations(sb_service_.get()); |
| 1796 } | 1796 } |
| 1797 { | 1797 { |
| 1798 EXPECT_CALL(*sb_service_->mock_database_manager(), | 1798 EXPECT_CALL(*sb_service_->mock_database_manager(), |
| 1799 CheckDownloadUrl(ContainerEq(url_chain), NotNull())) | 1799 CheckDownloadUrl(ContainerEq(url_chain), NotNull())) |
| 1800 .WillOnce(DoAll(CheckDownloadUrlDone(SB_THREAT_TYPE_BINARY_MALWARE_URL), | 1800 .WillOnce(DoAll(CheckDownloadUrlDone(SB_THREAT_TYPE_URL_BINARY_MALWARE), |
| 1801 Return(false))); | 1801 Return(false))); |
| 1802 RunLoop run_loop; | 1802 RunLoop run_loop; |
| 1803 download_service_->CheckDownloadUrl( | 1803 download_service_->CheckDownloadUrl( |
| 1804 &item, base::Bind(&DownloadProtectionServiceTest::CheckDoneCallback, | 1804 &item, base::Bind(&DownloadProtectionServiceTest::CheckDoneCallback, |
| 1805 base::Unretained(this), run_loop.QuitClosure())); | 1805 base::Unretained(this), run_loop.QuitClosure())); |
| 1806 run_loop.Run(); | 1806 run_loop.Run(); |
| 1807 EXPECT_TRUE(IsResult(DownloadProtectionService::DANGEROUS)); | 1807 EXPECT_TRUE(IsResult(DownloadProtectionService::DANGEROUS)); |
| 1808 } | 1808 } |
| 1809 } | 1809 } |
| 1810 | 1810 |
| (...skipping 534 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 2345 &item, base::Bind(&DownloadProtectionServiceTest::CheckDoneCallback, | 2345 &item, base::Bind(&DownloadProtectionServiceTest::CheckDoneCallback, |
| 2346 base::Unretained(this), run_loop.QuitClosure())); | 2346 base::Unretained(this), run_loop.QuitClosure())); |
| 2347 run_loop.Run(); | 2347 run_loop.Run(); |
| 2348 | 2348 |
| 2349 EXPECT_FALSE(HasClientDownloadRequest()); | 2349 EXPECT_FALSE(HasClientDownloadRequest()); |
| 2350 // Overriden by flag: | 2350 // Overriden by flag: |
| 2351 EXPECT_TRUE(IsResult(DownloadProtectionService::DANGEROUS)); | 2351 EXPECT_TRUE(IsResult(DownloadProtectionService::DANGEROUS)); |
| 2352 } | 2352 } |
| 2353 | 2353 |
| 2354 } // namespace safe_browsing | 2354 } // namespace safe_browsing |
| OLD | NEW |