| 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 |
| (...skipping 687 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 698 public: | 698 public: |
| 699 SafeBrowsingServiceMetadataTest() {} | 699 SafeBrowsingServiceMetadataTest() {} |
| 700 | 700 |
| 701 void GenUrlFullhashResultWithMetadata(const GURL& url, | 701 void GenUrlFullhashResultWithMetadata(const GURL& url, |
| 702 SBFullHashResult* full_hash) { | 702 SBFullHashResult* full_hash) { |
| 703 GenUrlFullhashResult(url, MALWARE, full_hash); | 703 GenUrlFullhashResult(url, MALWARE, full_hash); |
| 704 | 704 |
| 705 MalwarePatternType proto; | 705 MalwarePatternType proto; |
| 706 switch (GetParam()) { | 706 switch (GetParam()) { |
| 707 case METADATA_NONE: | 707 case METADATA_NONE: |
| 708 full_hash->metadata = std::string(); | 708 full_hash->metadata.raw_metadata = std::string(); |
| 709 break; | 709 break; |
| 710 case METADATA_LANDING: | 710 case METADATA_LANDING: |
| 711 proto.set_pattern_type(MalwarePatternType::LANDING); | 711 proto.set_pattern_type(MalwarePatternType::LANDING); |
| 712 full_hash->metadata = proto.SerializeAsString(); | 712 full_hash->metadata.raw_metadata = proto.SerializeAsString(); |
| 713 break; | 713 break; |
| 714 case METADATA_DISTRIBUTION: | 714 case METADATA_DISTRIBUTION: |
| 715 proto.set_pattern_type(MalwarePatternType::DISTRIBUTION); | 715 proto.set_pattern_type(MalwarePatternType::DISTRIBUTION); |
| 716 full_hash->metadata = proto.SerializeAsString(); | 716 full_hash->metadata.raw_metadata = proto.SerializeAsString(); |
| 717 break; | 717 break; |
| 718 } | 718 } |
| 719 } | 719 } |
| 720 | 720 |
| 721 private: | 721 private: |
| 722 DISALLOW_COPY_AND_ASSIGN(SafeBrowsingServiceMetadataTest); | 722 DISALLOW_COPY_AND_ASSIGN(SafeBrowsingServiceMetadataTest); |
| 723 }; | 723 }; |
| 724 | 724 |
| 725 IN_PROC_BROWSER_TEST_P(SafeBrowsingServiceMetadataTest, MalwareMainFrame) { | 725 IN_PROC_BROWSER_TEST_P(SafeBrowsingServiceMetadataTest, MalwareMainFrame) { |
| 726 GURL url = embedded_test_server()->GetURL(kEmptyPage); | 726 GURL url = embedded_test_server()->GetURL(kEmptyPage); |
| (...skipping 485 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1212 void OnCheckDownloadUrlResult(const std::vector<GURL>& /* url_chain */, | 1212 void OnCheckDownloadUrlResult(const std::vector<GURL>& /* url_chain */, |
| 1213 SBThreatType threat_type) override { | 1213 SBThreatType threat_type) override { |
| 1214 threat_type_ = threat_type; | 1214 threat_type_ = threat_type; |
| 1215 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, | 1215 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
| 1216 base::Bind(&TestSBClient::CheckDone, this)); | 1216 base::Bind(&TestSBClient::CheckDone, this)); |
| 1217 } | 1217 } |
| 1218 | 1218 |
| 1219 // Called when the result of checking a browse URL is known. | 1219 // Called when the result of checking a browse URL is known. |
| 1220 void OnCheckBrowseUrlResult(const GURL& /* url */, | 1220 void OnCheckBrowseUrlResult(const GURL& /* url */, |
| 1221 SBThreatType threat_type, | 1221 SBThreatType threat_type, |
| 1222 const std::string& /* metadata */) override { | 1222 const ThreatMetadata& /* metadata */) override { |
| 1223 threat_type_ = threat_type; | 1223 threat_type_ = threat_type; |
| 1224 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, | 1224 BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, |
| 1225 base::Bind(&TestSBClient::CheckDone, this)); | 1225 base::Bind(&TestSBClient::CheckDone, this)); |
| 1226 } | 1226 } |
| 1227 | 1227 |
| 1228 // Called when the result of checking a resource URL is known. | 1228 // Called when the result of checking a resource URL is known. |
| 1229 void OnCheckResourceUrlResult(const GURL& /* url */, | 1229 void OnCheckResourceUrlResult(const GURL& /* url */, |
| 1230 SBThreatType threat_type, | 1230 SBThreatType threat_type, |
| 1231 const std::string& threat_hash) override { | 1231 const std::string& threat_hash) override { |
| 1232 threat_type_ = threat_type; | 1232 threat_type_ = threat_type; |
| (...skipping 516 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1749 chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, | 1749 chrome::NOTIFICATION_SAFE_BROWSING_UPDATE_COMPLETE, |
| 1750 content::Source<SafeBrowsingDatabaseManager>( | 1750 content::Source<SafeBrowsingDatabaseManager>( |
| 1751 sb_service_->database_manager().get())); | 1751 sb_service_->database_manager().get())); |
| 1752 BrowserThread::PostTask( | 1752 BrowserThread::PostTask( |
| 1753 BrowserThread::IO, FROM_HERE, | 1753 BrowserThread::IO, FROM_HERE, |
| 1754 base::Bind(&SafeBrowsingDatabaseManagerCookieTest::ForceUpdate, this)); | 1754 base::Bind(&SafeBrowsingDatabaseManagerCookieTest::ForceUpdate, this)); |
| 1755 observer.Wait(); | 1755 observer.Wait(); |
| 1756 } | 1756 } |
| 1757 | 1757 |
| 1758 } // namespace safe_browsing | 1758 } // namespace safe_browsing |
| OLD | NEW |