PVer4: V4Database: Return stores that have a hash prefix for a given full hash

components/safe_browsing_db/v4_database.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 <memory>
 
 #include "base/callback.h"
 #include "base/debug/leak_annotations.h"
 #include "base/files/file_util.h"
 #include "base/memory/ptr_util.h"
@@ skipping 150 matching lines @@
 
 std::unique_ptr<StoreStateMap> V4Database::GetStoreStateMap() {
   std::unique_ptr<StoreStateMap> store_state_map =
       base::MakeUnique<StoreStateMap>();
   for (const auto& store_map_iter : *store_map_) {
     (*store_state_map)[store_map_iter.first] = store_map_iter.second->state();
   }
   return store_state_map;
 }
 
+void V4Database::GetStoresMatchingFullHash(
+    const FullHash& full_hash,
+    const base::hash_set<UpdateListIdentifier>& stores_to_look,
+    MatchedHashPrefixMap* matched_hash_prefix_map) {
+  for (const UpdateListIdentifier& identifier : stores_to_look) {
+    const auto& store_pair = store_map_->find(identifier);
+    DCHECK(store_pair != store_map_->end());
+    const std::unique_ptr<V4Store>& store = store_pair->second;
+    HashPrefix hash_prefix = store->GetMatchingHashPrefix(full_hash);
+    if (!hash_prefix.empty()) {
+      (*matched_hash_prefix_map)[identifier] = hash_prefix;
+    }
+  }
+}
+
 }  // namespace safe_browsing