Ignore GetHash results for lists that we don't support.

chrome/browser/safe_browsing/protocol_parser_unittest.cc

 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 //
 // Program to test the SafeBrowsing protocol parsing v2.1.
 
 #include "base/hash_tables.h"
 #include "base/logging.h"
 #include "base/string_util.h"
 #include "chrome/browser/safe_browsing/protocol_parser.h"
@@ skipping 473 matching lines @@
   EXPECT_TRUE(parser.ParseGetHash(reinterpret_cast<const char*>(get_hash),
                                   sizeof(get_hash),
                                   key,
                                   &re_key,
                                   &full_hashes));
   EXPECT_FALSE(re_key);
   EXPECT_EQ(full_hashes.size(), static_cast<size_t>(1));
   EXPECT_EQ(memcmp(hash_result, &full_hashes[0].hash, sizeof(SBFullHash)), 0);
 }
 
+TEST(SafeBrowsingProtocolParsingTest, TestGetHashWithUnknownList) {
+  std::string hash_response = "goog-phish-shavar:1:32\n"
+                              "12345678901234567890123456789012"
+                              "googpub-phish-shavar:19:32\n"
+                              "09876543210987654321098765432109";
+  bool re_key = false;
+  std::string key = "";
+  std::vector<SBFullHashResult> full_hashes;
+  SafeBrowsingProtocolParser parser;
+  EXPECT_TRUE(parser.ParseGetHash(hash_response.data(),
+                                  hash_response.size(),
+                                  key,
+                                  &re_key,
+                                  &full_hashes));
+
+  EXPECT_EQ(full_hashes.size(), 1);
+  EXPECT_EQ(memcmp("12345678901234567890123456789012", 
+                   &full_hashes[0].hash, sizeof(SBFullHash)), 0);
+  EXPECT_EQ(full_hashes[0].list_name, "goog-phish-shavar");
+  EXPECT_EQ(full_hashes[0].add_chunk_id, 1);
+
+  hash_response += "goog-malware-shavar:7:32\n"
+                   "abcdefghijklmnopqrstuvwxyz123457";
+  full_hashes.clear();
+  EXPECT_TRUE(parser.ParseGetHash(hash_response.data(),
+                                  hash_response.size(),
+                                  key,
+                                  &re_key,
+                                  &full_hashes));
+
+  EXPECT_EQ(full_hashes.size(), 2);
+  EXPECT_EQ(memcmp("12345678901234567890123456789012", 
+                   &full_hashes[0].hash, sizeof(SBFullHash)), 0);
+  EXPECT_EQ(full_hashes[0].list_name, "goog-phish-shavar");
+  EXPECT_EQ(full_hashes[0].add_chunk_id, 1);
+  EXPECT_EQ(memcmp("abcdefghijklmnopqrstuvwxyz123457", 
+                   &full_hashes[1].hash, sizeof(SBFullHash)), 0);
+  EXPECT_EQ(full_hashes[1].list_name, "goog-malware-shavar");
+  EXPECT_EQ(full_hashes[1].add_chunk_id, 7);
+}
+
 TEST(SafeBrowsingProtocolParsingTest, TestFormatHash) {
   SafeBrowsingProtocolParser parser;
   std::vector<SBPrefix> prefixes;
   std::string get_hash;
 
   prefixes.push_back(0x34333231);
   prefixes.push_back(0x64636261);
   prefixes.push_back(0x73727170);
 
   parser.FormatGetHash(prefixes, &get_hash);
@@ skipping 198 matching lines @@
   const std::string key("v_aDSz6jI92WeHCOoZ07QA==");
   const std::string mac("W9Xp2fUcQ9V66If6Cvsrstpa4Kk=");
 
   EXPECT_TRUE(parser.ParseChunk(reinterpret_cast<const char*>(chunk),
                                 sizeof(chunk), key, mac,
                                 &re_key, &chunks));
   EXPECT_FALSE(re_key);
 
   safe_browsing_util::FreeChunks(&chunks);
 }