Correctly hash the input of MatchModuleWhitelistString() in pver4.

components/safe_browsing_db/v4_local_database_manager_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/scoped_temp_dir.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/run_loop.h"
 #include "base/test/test_simple_task_runner.h"
 #include "components/safe_browsing_db/v4_database.h"
 #include "components/safe_browsing_db/v4_local_database_manager.h"
 #include "components/safe_browsing_db/v4_test_util.h"
 #include "content/public/test/test_browser_thread_bundle.h"
+#include "crypto/sha2.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "testing/platform_test.h"
 
 namespace safe_browsing {
 
 class FakeV4Database : public V4Database {
  public:
   static void Create(
       const scoped_refptr<base::SequencedTaskRunner>& db_task_runner,
       std::unique_ptr<StoreMap> store_map,
@@ skipping 376 matching lines @@
   EXPECT_FALSE(v4_local_database_manager_->MatchMalwareIP("192.168.1.1"));
   EXPECT_FALSE(FakeV4LocalDatabaseManager::PerformFullHashCheckCalled(
       v4_local_database_manager_));
 
   // The fake database returns a matched hash prefix.
   EXPECT_TRUE(v4_local_database_manager_->MatchMalwareIP("192.168.1.2"));
   EXPECT_FALSE(FakeV4LocalDatabaseManager::PerformFullHashCheckCalled(
       v4_local_database_manager_));
 }
 
+TEST_F(V4LocalDatabaseManagerTest, TestMatchModuleWhitelist) {
+  StopLocalDatabaseManager();
+  v4_local_database_manager_ =
+      make_scoped_refptr(new FakeV4LocalDatabaseManager(base_dir_.GetPath()));
+  SetTaskRunnerForTest();
+  StartLocalDatabaseManager();
+  WaitForTasksOnTaskRunner();
+
+  StoreAndHashPrefixes store_and_hash_prefixes;
+  store_and_hash_prefixes.emplace_back(GetChromeFilenameClientIncidentId(),
+                                       crypto::SHA256HashString("chrome.dll"));
+
+  ReplaceV4Database(store_and_hash_prefixes);
+
+  // No match -- i.e. not whitelisted
+  EXPECT_FALSE(
+      v4_local_database_manager_->MatchModuleWhitelistString("badstuff.dll"));
+
+  // Whitelisted.
+  EXPECT_TRUE(
+      v4_local_database_manager_->MatchModuleWhitelistString("chrome.dll"));
+}
+
 }  // namespace safe_browsing