Adding blacklisted dlls to safe browsing incident reports.

chrome/browser/safe_browsing/environment_data_collection_win.cc

 // Copyright 2014 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 "chrome/browser/safe_browsing/environment_data_collection_win.h"
 
 #include <windows.h>
 #include <set>
 
 #include "base/i18n/case_conversion.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/win/registry.h"
 #include "chrome/browser/install_verification/win/module_info.h"
 #include "chrome/browser/install_verification/win/module_verification_common.h"
 #include "chrome/browser/net/service_providers_win.h"
 #include "chrome/browser/safe_browsing/path_sanitizer.h"
 #include "chrome/common/safe_browsing/csd.pb.h"
+#include "chrome_elf/chrome_elf_constants.h"
 
 namespace safe_browsing {
 
 namespace {
 
 // Helper function for expanding all environment variables in |path|.
 std::wstring ExpandEnvironmentVariables(const std::wstring& path) {
   static const DWORD kMaxBuffer = 32 * 1024;  // Max according to MSDN.
   std::wstring path_expanded;
   DWORD path_len = MAX_PATH;
@@ skipping 53 matching lines @@
 
   // Look for a match between LSPs and loaded dlls.
   for (int i = 0; i < process->dll_size(); ++i) {
     if (lsp_paths.count(base::UTF8ToWide(process->dll(i).path()))) {
       process->mutable_dll(i)
           ->add_feature(ClientIncidentReport_EnvironmentData_Process_Dll::LSP);
     }
   }
 }
 
+void CollectDllBlacklistData(
+    ClientIncidentReport_EnvironmentData_Process* process) {
+  PathSanitizer path_sanitizer;
+  base::win::RegistryValueIterator iter(HKEY_CURRENT_USER,
+                                        blacklist::kRegistryFinchListPath);
+  for (; iter.Valid(); ++iter) {
+    base::FilePath dll_name(iter.Value());
+    path_sanitizer.StripHomeDirectory(&dll_name);
+    process->add_blacklisted_dll(dll_name.AsUTF8Unsafe());
+  }
+}
+
 void CollectPlatformProcessData(
     ClientIncidentReport_EnvironmentData_Process* process) {
   CollectDlls(process);
   RecordLspFeature(process);
+  CollectDllBlacklistData(process);
 }
 
 }  // namespace safe_browsing