| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 <vector> | 5 #include <vector> |
| 6 | 6 |
| 7 #include "base/base64.h" | 7 #include "base/base64.h" |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "base/strings/stringprintf.h" | 9 #include "base/strings/stringprintf.h" |
| 10 #include "base/time/time.h" | 10 #include "base/time/time.h" |
| 11 #include "components/safe_browsing_db/safebrowsing.pb.h" | 11 #include "components/safe_browsing_db/safebrowsing.pb.h" |
| 12 #include "components/safe_browsing_db/testing_util.h" |
| 12 #include "components/safe_browsing_db/util.h" | 13 #include "components/safe_browsing_db/util.h" |
| 13 #include "components/safe_browsing_db/v4_get_hash_protocol_manager.h" | 14 #include "components/safe_browsing_db/v4_get_hash_protocol_manager.h" |
| 14 #include "net/base/escape.h" | 15 #include "net/base/escape.h" |
| 15 #include "net/base/load_flags.h" | 16 #include "net/base/load_flags.h" |
| 16 #include "net/base/net_errors.h" | 17 #include "net/base/net_errors.h" |
| 17 #include "net/url_request/test_url_fetcher_factory.h" | 18 #include "net/url_request/test_url_fetcher_factory.h" |
| 18 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
| 19 | 20 |
| 20 using base::Time; | 21 using base::Time; |
| 21 using base::TimeDelta; | 22 using base::TimeDelta; |
| (...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 135 } | 136 } |
| 136 | 137 |
| 137 TEST_F(SafeBrowsingV4GetHashProtocolManagerTest, TestGetHashErrorHandlingOK) { | 138 TEST_F(SafeBrowsingV4GetHashProtocolManagerTest, TestGetHashErrorHandlingOK) { |
| 138 net::TestURLFetcherFactory factory; | 139 net::TestURLFetcherFactory factory; |
| 139 scoped_ptr<V4GetHashProtocolManager> pm(CreateProtocolManager()); | 140 scoped_ptr<V4GetHashProtocolManager> pm(CreateProtocolManager()); |
| 140 | 141 |
| 141 std::vector<SBPrefix> prefixes; | 142 std::vector<SBPrefix> prefixes; |
| 142 std::vector<SBFullHashResult> expected_full_hashes; | 143 std::vector<SBFullHashResult> expected_full_hashes; |
| 143 SBFullHashResult hash_result; | 144 SBFullHashResult hash_result; |
| 144 hash_result.hash = SBFullHashForString("Everything's shiny, Cap'n."); | 145 hash_result.hash = SBFullHashForString("Everything's shiny, Cap'n."); |
| 145 hash_result.metadata = "NOTIFICATIONS,"; | 146 hash_result.metadata.api_permissions.push_back("NOTIFICATIONS"); |
| 146 hash_result.cache_duration = base::TimeDelta::FromSeconds(300); | 147 hash_result.cache_duration = base::TimeDelta::FromSeconds(300); |
| 147 expected_full_hashes.push_back(hash_result); | 148 expected_full_hashes.push_back(hash_result); |
| 148 base::TimeDelta expected_cache_duration = base::TimeDelta::FromSeconds(600); | 149 base::TimeDelta expected_cache_duration = base::TimeDelta::FromSeconds(600); |
| 149 | 150 |
| 150 pm->GetFullHashesWithApis( | 151 pm->GetFullHashesWithApis( |
| 151 prefixes, base::Bind(&ValidateGetV4HashResults, expected_full_hashes, | 152 prefixes, base::Bind(&ValidateGetV4HashResults, expected_full_hashes, |
| 152 expected_cache_duration)); | 153 expected_cache_duration)); |
| 153 | 154 |
| 154 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); | 155 net::TestURLFetcher* fetcher = factory.GetFetcherByID(0); |
| 155 DCHECK(fetcher); | 156 DCHECK(fetcher); |
| (...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 222 | 223 |
| 223 Time now = Time::Now(); | 224 Time now = Time::Now(); |
| 224 std::vector<SBFullHashResult> full_hashes; | 225 std::vector<SBFullHashResult> full_hashes; |
| 225 base::TimeDelta cache_lifetime; | 226 base::TimeDelta cache_lifetime; |
| 226 EXPECT_TRUE(pm->ParseHashResponse(res_data, &full_hashes, &cache_lifetime)); | 227 EXPECT_TRUE(pm->ParseHashResponse(res_data, &full_hashes, &cache_lifetime)); |
| 227 | 228 |
| 228 EXPECT_EQ(base::TimeDelta::FromSeconds(600), cache_lifetime); | 229 EXPECT_EQ(base::TimeDelta::FromSeconds(600), cache_lifetime); |
| 229 EXPECT_EQ(1ul, full_hashes.size()); | 230 EXPECT_EQ(1ul, full_hashes.size()); |
| 230 EXPECT_TRUE(SBFullHashEqual(SBFullHashForString("Everything's shiny, Cap'n."), | 231 EXPECT_TRUE(SBFullHashEqual(SBFullHashForString("Everything's shiny, Cap'n."), |
| 231 full_hashes[0].hash)); | 232 full_hashes[0].hash)); |
| 232 EXPECT_EQ("NOTIFICATIONS,", full_hashes[0].metadata); | 233 EXPECT_EQ(1ul, full_hashes[0].metadata.api_permissions.size()); |
| 234 EXPECT_EQ("NOTIFICATIONS", full_hashes[0].metadata.api_permissions[0]); |
| 233 EXPECT_EQ(base::TimeDelta::FromSeconds(300), full_hashes[0].cache_duration); | 235 EXPECT_EQ(base::TimeDelta::FromSeconds(300), full_hashes[0].cache_duration); |
| 234 EXPECT_LE(now + base::TimeDelta::FromSeconds(400), pm->next_gethash_time_); | 236 EXPECT_LE(now + base::TimeDelta::FromSeconds(400), pm->next_gethash_time_); |
| 235 } | 237 } |
| 236 | 238 |
| 237 // Adds an entry with an ignored ThreatEntryType. | 239 // Adds an entry with an ignored ThreatEntryType. |
| 238 TEST_F(SafeBrowsingV4GetHashProtocolManagerTest, | 240 TEST_F(SafeBrowsingV4GetHashProtocolManagerTest, |
| 239 TestParseHashResponseWrongThreatEntryType) { | 241 TestParseHashResponseWrongThreatEntryType) { |
| 240 scoped_ptr<V4GetHashProtocolManager> pm(CreateProtocolManager()); | 242 scoped_ptr<V4GetHashProtocolManager> pm(CreateProtocolManager()); |
| 241 | 243 |
| 242 FindFullHashesResponse res; | 244 FindFullHashesResponse res; |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 311 std::vector<SBFullHashResult> full_hashes; | 313 std::vector<SBFullHashResult> full_hashes; |
| 312 base::TimeDelta cache_lifetime; | 314 base::TimeDelta cache_lifetime; |
| 313 EXPECT_TRUE(pm->ParseHashResponse(res_data, &full_hashes, &cache_lifetime)); | 315 EXPECT_TRUE(pm->ParseHashResponse(res_data, &full_hashes, &cache_lifetime)); |
| 314 | 316 |
| 315 EXPECT_EQ(base::TimeDelta::FromSeconds(600), cache_lifetime); | 317 EXPECT_EQ(base::TimeDelta::FromSeconds(600), cache_lifetime); |
| 316 EXPECT_EQ(1ul, full_hashes.size()); | 318 EXPECT_EQ(1ul, full_hashes.size()); |
| 317 | 319 |
| 318 EXPECT_TRUE(SBFullHashEqual(SBFullHashForString("Not to fret."), | 320 EXPECT_TRUE(SBFullHashEqual(SBFullHashForString("Not to fret."), |
| 319 full_hashes[0].hash)); | 321 full_hashes[0].hash)); |
| 320 // Metadata should be empty. | 322 // Metadata should be empty. |
| 321 EXPECT_EQ("", full_hashes[0].metadata); | 323 EXPECT_EQ(0ul, full_hashes[0].metadata.api_permissions.size()); |
| 322 EXPECT_EQ(base::TimeDelta::FromSeconds(0), full_hashes[0].cache_duration); | 324 EXPECT_EQ(base::TimeDelta::FromSeconds(0), full_hashes[0].cache_duration); |
| 323 } | 325 } |
| 324 | 326 |
| 325 TEST_F(SafeBrowsingV4GetHashProtocolManagerTest, | 327 TEST_F(SafeBrowsingV4GetHashProtocolManagerTest, |
| 326 TestParseHashResponseInconsistentThreatTypes) { | 328 TestParseHashResponseInconsistentThreatTypes) { |
| 327 scoped_ptr<V4GetHashProtocolManager> pm(CreateProtocolManager()); | 329 scoped_ptr<V4GetHashProtocolManager> pm(CreateProtocolManager()); |
| 328 | 330 |
| 329 FindFullHashesResponse res; | 331 FindFullHashesResponse res; |
| 330 ThreatMatch* m1 = res.add_matches(); | 332 ThreatMatch* m1 = res.add_matches(); |
| 331 m1->set_threat_type(API_ABUSE); | 333 m1->set_threat_type(API_ABUSE); |
| (...skipping 11 matching lines...) Expand all Loading... |
| 343 // Serialize. | 345 // Serialize. |
| 344 std::string res_data; | 346 std::string res_data; |
| 345 res.SerializeToString(&res_data); | 347 res.SerializeToString(&res_data); |
| 346 | 348 |
| 347 std::vector<SBFullHashResult> full_hashes; | 349 std::vector<SBFullHashResult> full_hashes; |
| 348 base::TimeDelta cache_lifetime; | 350 base::TimeDelta cache_lifetime; |
| 349 EXPECT_FALSE(pm->ParseHashResponse(res_data, &full_hashes, &cache_lifetime)); | 351 EXPECT_FALSE(pm->ParseHashResponse(res_data, &full_hashes, &cache_lifetime)); |
| 350 } | 352 } |
| 351 | 353 |
| 352 } // namespace safe_browsing | 354 } // namespace safe_browsing |
| OLD | NEW |