| 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 <map> | 7 #include <map> |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/base_paths.h" | 10 #include "base/base_paths.h" |
| 11 #include "base/bind.h" | 11 #include "base/bind.h" |
| 12 #include "base/callback.h" | 12 #include "base/callback.h" |
| 13 #include "base/file_util.h" | 13 #include "base/file_util.h" |
| 14 #include "base/files/file_path.h" | 14 #include "base/files/file_path.h" |
| 15 #include "base/files/scoped_temp_dir.h" | 15 #include "base/files/scoped_temp_dir.h" |
| 16 #include "base/memory/ref_counted.h" | 16 #include "base/memory/ref_counted.h" |
| 17 #include "base/memory/scoped_ptr.h" | 17 #include "base/memory/scoped_ptr.h" |
| 18 #include "base/message_loop/message_loop.h" | 18 #include "base/message_loop/message_loop.h" |
| 19 #include "base/path_service.h" | 19 #include "base/path_service.h" |
| 20 #include "base/run_loop.h" | 20 #include "base/run_loop.h" |
| 21 #include "base/strings/string_number_conversions.h" | 21 #include "base/strings/string_number_conversions.h" |
| 22 #include "base/threading/sequenced_worker_pool.h" | 22 #include "base/threading/sequenced_worker_pool.h" |
| 23 #include "chrome/browser/safe_browsing/database_manager.h" | 23 #include "chrome/browser/safe_browsing/database_manager.h" |
| 24 #include "chrome/browser/safe_browsing/download_feedback_service.h" | 24 #include "chrome/browser/safe_browsing/download_feedback_service.h" |
| 25 #include "chrome/browser/safe_browsing/safe_browsing_service.h" | 25 #include "chrome/browser/safe_browsing/safe_browsing_service.h" |
| 26 #include "chrome/browser/safe_browsing/signature_util.h" | 26 #include "chrome/browser/safe_browsing/signature_util.h" |
| 27 #include "chrome/common/safe_browsing/csd.pb.h" | 27 #include "chrome/common/safe_browsing/csd.pb.h" |
| 28 #include "content/public/browser/render_process_host.h" | |
| 29 #include "content/public/test/mock_download_item.h" | 28 #include "content/public/test/mock_download_item.h" |
| 30 #include "content/public/test/test_browser_thread_bundle.h" | 29 #include "content/public/test/test_browser_thread_bundle.h" |
| 30 #include "content/public/test/test_utils.h" |
| 31 #include "net/cert/x509_certificate.h" | 31 #include "net/cert/x509_certificate.h" |
| 32 #include "net/http/http_status_code.h" | 32 #include "net/http/http_status_code.h" |
| 33 #include "net/url_request/test_url_fetcher_factory.h" | 33 #include "net/url_request/test_url_fetcher_factory.h" |
| 34 #include "net/url_request/url_fetcher_delegate.h" | 34 #include "net/url_request/url_fetcher_delegate.h" |
| 35 #include "net/url_request/url_request_status.h" | 35 #include "net/url_request/url_request_status.h" |
| 36 #include "testing/gmock/include/gmock/gmock.h" | 36 #include "testing/gmock/include/gmock/gmock.h" |
| 37 #include "testing/gtest/include/gtest/gtest.h" | 37 #include "testing/gtest/include/gtest/gtest.h" |
| 38 #include "third_party/zlib/google/zip.h" | 38 #include "third_party/zlib/google/zip.h" |
| 39 #include "url/gurl.h" | 39 #include "url/gurl.h" |
| 40 | 40 |
| (...skipping 109 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 150 base::Owned(check))); | 150 base::Owned(check))); |
| 151 } | 151 } |
| 152 | 152 |
| 153 class DownloadProtectionServiceTest : public testing::Test { | 153 class DownloadProtectionServiceTest : public testing::Test { |
| 154 protected: | 154 protected: |
| 155 DownloadProtectionServiceTest() | 155 DownloadProtectionServiceTest() |
| 156 : test_browser_thread_bundle_( | 156 : test_browser_thread_bundle_( |
| 157 content::TestBrowserThreadBundle::IO_MAINLOOP) { | 157 content::TestBrowserThreadBundle::IO_MAINLOOP) { |
| 158 } | 158 } |
| 159 virtual void SetUp() { | 159 virtual void SetUp() { |
| 160 content::RenderProcessHost::SetRunRendererInProcess(true); | |
| 161 // Start real threads for the IO and File threads so that the DCHECKs | 160 // Start real threads for the IO and File threads so that the DCHECKs |
| 162 // to test that we're on the correct thread work. | 161 // to test that we're on the correct thread work. |
| 163 sb_service_ = new StrictMock<FakeSafeBrowsingService>(); | 162 sb_service_ = new StrictMock<FakeSafeBrowsingService>(); |
| 164 sb_service_->Initialize(); | 163 sb_service_->Initialize(); |
| 165 signature_util_ = new StrictMock<MockSignatureUtil>(); | 164 signature_util_ = new StrictMock<MockSignatureUtil>(); |
| 166 download_service_ = sb_service_->download_protection_service(); | 165 download_service_ = sb_service_->download_protection_service(); |
| 167 download_service_->signature_util_ = signature_util_; | 166 download_service_->signature_util_ = signature_util_; |
| 168 download_service_->SetEnabled(true); | 167 download_service_->SetEnabled(true); |
| 169 base::RunLoop().RunUntilIdle(); | 168 base::RunLoop().RunUntilIdle(); |
| 170 has_result_ = false; | 169 has_result_ = false; |
| 171 | 170 |
| 172 base::FilePath source_path; | 171 base::FilePath source_path; |
| 173 ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &source_path)); | 172 ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &source_path)); |
| 174 testdata_path_ = source_path | 173 testdata_path_ = source_path |
| 175 .AppendASCII("chrome") | 174 .AppendASCII("chrome") |
| 176 .AppendASCII("test") | 175 .AppendASCII("test") |
| 177 .AppendASCII("data") | 176 .AppendASCII("data") |
| 178 .AppendASCII("safe_browsing") | 177 .AppendASCII("safe_browsing") |
| 179 .AppendASCII("download_protection"); | 178 .AppendASCII("download_protection"); |
| 180 } | 179 } |
| 181 | 180 |
| 182 virtual void TearDown() { | 181 virtual void TearDown() { |
| 183 sb_service_->ShutDown(); | 182 sb_service_->ShutDown(); |
| 184 // Flush all of the thread message loops to ensure that there are no | 183 // Flush all of the thread message loops to ensure that there are no |
| 185 // tasks currently running. | 184 // tasks currently running. |
| 186 FlushThreadMessageLoops(); | 185 FlushThreadMessageLoops(); |
| 187 sb_service_ = NULL; | 186 sb_service_ = NULL; |
| 188 content::RenderProcessHost::SetRunRendererInProcess(false); | |
| 189 } | 187 } |
| 190 | 188 |
| 191 bool RequestContainsResource(const ClientDownloadRequest& request, | 189 bool RequestContainsResource(const ClientDownloadRequest& request, |
| 192 ClientDownloadRequest::ResourceType type, | 190 ClientDownloadRequest::ResourceType type, |
| 193 const std::string& url, | 191 const std::string& url, |
| 194 const std::string& referrer) { | 192 const std::string& referrer) { |
| 195 for (int i = 0; i < request.resources_size(); ++i) { | 193 for (int i = 0; i < request.resources_size(); ++i) { |
| 196 if (request.resources(i).url() == url && | 194 if (request.resources(i).url() == url && |
| 197 request.resources(i).type() == type && | 195 request.resources(i).type() == type && |
| 198 (referrer.empty() || request.resources(i).referrer() == referrer)) { | 196 (referrer.empty() || request.resources(i).referrer() == referrer)) { |
| (...skipping 112 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 311 ", got " << result_; | 309 ", got " << result_; |
| 312 } | 310 } |
| 313 | 311 |
| 314 protected: | 312 protected: |
| 315 scoped_refptr<FakeSafeBrowsingService> sb_service_; | 313 scoped_refptr<FakeSafeBrowsingService> sb_service_; |
| 316 scoped_refptr<MockSignatureUtil> signature_util_; | 314 scoped_refptr<MockSignatureUtil> signature_util_; |
| 317 DownloadProtectionService* download_service_; | 315 DownloadProtectionService* download_service_; |
| 318 DownloadProtectionService::DownloadCheckResult result_; | 316 DownloadProtectionService::DownloadCheckResult result_; |
| 319 bool has_result_; | 317 bool has_result_; |
| 320 content::TestBrowserThreadBundle test_browser_thread_bundle_; | 318 content::TestBrowserThreadBundle test_browser_thread_bundle_; |
| 319 content::InProcessUtilityThreadHelper in_process_utility_thread_helper_; |
| 321 base::FilePath testdata_path_; | 320 base::FilePath testdata_path_; |
| 322 }; | 321 }; |
| 323 | 322 |
| 324 TEST_F(DownloadProtectionServiceTest, CheckClientDownloadInvalidUrl) { | 323 TEST_F(DownloadProtectionServiceTest, CheckClientDownloadInvalidUrl) { |
| 325 base::FilePath a_tmp(FILE_PATH_LITERAL("a.tmp")); | 324 base::FilePath a_tmp(FILE_PATH_LITERAL("a.tmp")); |
| 326 base::FilePath a_exe(FILE_PATH_LITERAL("a.exe")); | 325 base::FilePath a_exe(FILE_PATH_LITERAL("a.exe")); |
| 327 std::vector<GURL> url_chain; | 326 std::vector<GURL> url_chain; |
| 328 GURL referrer("http://www.google.com/"); | 327 GURL referrer("http://www.google.com/"); |
| 329 | 328 |
| 330 content::MockDownloadItem item; | 329 content::MockDownloadItem item; |
| (...skipping 839 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1170 EXPECT_THAT(whitelist_strings, ElementsAre(cert_base + "/OU=unit")); | 1169 EXPECT_THAT(whitelist_strings, ElementsAre(cert_base + "/OU=unit")); |
| 1171 | 1170 |
| 1172 cert = ReadTestCertificate("test_c.pem"); | 1171 cert = ReadTestCertificate("test_c.pem"); |
| 1173 ASSERT_TRUE(cert.get()); | 1172 ASSERT_TRUE(cert.get()); |
| 1174 whitelist_strings.clear(); | 1173 whitelist_strings.clear(); |
| 1175 GetCertificateWhitelistStrings( | 1174 GetCertificateWhitelistStrings( |
| 1176 *cert.get(), *issuer_cert.get(), &whitelist_strings); | 1175 *cert.get(), *issuer_cert.get(), &whitelist_strings); |
| 1177 EXPECT_THAT(whitelist_strings, ElementsAre()); | 1176 EXPECT_THAT(whitelist_strings, ElementsAre()); |
| 1178 } | 1177 } |
| 1179 } // namespace safe_browsing | 1178 } // namespace safe_browsing |
| OLD | NEW |