Parse the hash prefix map from the V4Store on disk.

components/safe_browsing_db/v4_store_unittest.cc

 // Copyright 2016 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.
 
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "base/test/test_simple_task_runner.h"
 #include "base/time/time.h"
@@ skipping 121 matching lines @@
   std::unique_ptr<ListUpdateResponse> list_update_response(
       new ListUpdateResponse);
   list_update_response->set_response_type(ListUpdateResponse::FULL_UPDATE);
   list_update_response->set_new_client_state("test_client_state");
   EXPECT_EQ(WRITE_SUCCESS, V4Store(task_runner_, store_path_)
                                .WriteToDisk(std::move(list_update_response)));
 
   std::unique_ptr<V4Store> read_store(new V4Store(task_runner_, store_path_));
   EXPECT_EQ(READ_SUCCESS, read_store->ReadFromDisk());
   EXPECT_EQ("test_client_state", read_store->state_);
+  EXPECT_TRUE(read_store->hash_prefix_map_.empty());
 }
 
 TEST_F(V4StoreTest, TestAddUnlumpedHashesWithInvalidAddition) {
   HashPrefixMap prefix_map;
   EXPECT_EQ(ADDITIONS_SIZE_UNEXPECTED_FAILURE,
             V4Store::AddUnlumpedHashes(5, "a", &prefix_map));
   EXPECT_TRUE(prefix_map.empty());
 }
 
 TEST_F(V4StoreTest, TestAddUnlumpedHashesWithEmptyString) {
@@ skipping 165 matching lines @@
   EXPECT_EQ(1u, prefix_map.size());
 
   PrefixSize prefix_size = 4;
   HashPrefixes hash_prefixes = prefix_map.at(prefix_size);
   EXPECT_EQ(3 * prefix_size, hash_prefixes.size());
   EXPECT_EQ("0000", hash_prefixes.substr(0 * prefix_size, prefix_size));
   EXPECT_EQ("1111", hash_prefixes.substr(1 * prefix_size, prefix_size));
   EXPECT_EQ("2222", hash_prefixes.substr(2 * prefix_size, prefix_size));
 }
 
-TEST_F(V4StoreTest, TestMergeUpdatesFailsForRepeatedhashPrefix) {
+TEST_F(V4StoreTest, TestMergeUpdatesFailsForRepeatedHashPrefix) {
   HashPrefixMap prefix_map_old;
   EXPECT_EQ(APPLY_UPDATE_SUCCESS,
             V4Store::AddUnlumpedHashes(4, "2222", &prefix_map_old));
   HashPrefixMap prefix_map_additions;
   EXPECT_EQ(APPLY_UPDATE_SUCCESS,
             V4Store::AddUnlumpedHashes(4, "2222", &prefix_map_additions));
 
   std::unique_ptr<V4Store> store(new V4Store(task_runner_, store_path_));
   EXPECT_EQ(ADDITIONS_HAS_EXISTING_PREFIX_FAILURE,
             store->MergeUpdate(prefix_map_old, prefix_map_additions));
 }
 
+TEST_F(V4StoreTest, TestReadFullResponseWithValidHashPrefixMap) {
+  std::unique_ptr<ListUpdateResponse> lur(new ListUpdateResponse);
+  lur->set_response_type(ListUpdateResponse::FULL_UPDATE);
+  lur->set_new_client_state("test_client_state");
+  lur->set_platform_type(WINDOWS_PLATFORM);
+  lur->set_threat_entry_type(URL);
+  lur->set_threat_type(MALWARE_THREAT);
+  ThreatEntrySet* additions = lur->add_additions();
+  additions->set_compression_type(RAW);
+  additions->mutable_raw_hashes()->set_prefix_size(5);
+  additions->mutable_raw_hashes()->set_raw_hashes("00000abcde");
+  additions = lur->add_additions();
+  additions->set_compression_type(RAW);
+  additions->mutable_raw_hashes()->set_prefix_size(4);
+  additions->mutable_raw_hashes()->set_raw_hashes("00000abc");
+  EXPECT_EQ(WRITE_SUCCESS,
+            V4Store(task_runner_, store_path_).WriteToDisk(std::move(lur)));
+
+  V4Store read_store (task_runner_, store_path_);
+  EXPECT_EQ(READ_SUCCESS, read_store.ReadFromDisk());
+  EXPECT_EQ("test_client_state", read_store.state_);
+  ASSERT_EQ(2u, read_store.hash_prefix_map_.size());
+  EXPECT_EQ("00000abc", read_store.hash_prefix_map_[4]);
+  EXPECT_EQ("00000abcde", read_store.hash_prefix_map_[5]);
+}
+
+// This tests fails to read the prefix map from the disk because the file on
+// disk is invalid. The hash prefixes string is 6 bytes long, but the prefix
+// size is 5 so the parser isn't able to split the hash prefixes list
+// completely.
+TEST_F(V4StoreTest, TestReadFullResponseWithInvalidHashPrefixMap) {
+  std::unique_ptr<ListUpdateResponse> lur(new ListUpdateResponse);
+  lur->set_response_type(ListUpdateResponse::FULL_UPDATE);
+  lur->set_new_client_state("test_client_state");
+  lur->set_platform_type(WINDOWS_PLATFORM);
+  lur->set_threat_entry_type(URL);
+  lur->set_threat_type(MALWARE_THREAT);
+  ThreatEntrySet* additions = lur->add_additions();
+  additions->set_compression_type(RAW);
+  additions->mutable_raw_hashes()->set_prefix_size(5);
+  additions->mutable_raw_hashes()->set_raw_hashes("abcdef");
+  EXPECT_EQ(WRITE_SUCCESS,
+            V4Store(task_runner_, store_path_).WriteToDisk(std::move(lur)));
+
+  V4Store read_store(task_runner_, store_path_);
+  EXPECT_EQ(HASH_PREFIX_MAP_GENERATION_FAILURE, read_store.ReadFromDisk());
+  EXPECT_TRUE(read_store.state_.empty());
+  EXPECT_TRUE(read_store.hash_prefix_map_.empty());
+}
+
 }  // namespace safe_browsing