| Index: chrome/browser/safe_browsing/incident_reporting/blacklist_load_analyzer_win.cc
|
| diff --git a/chrome/browser/safe_browsing/incident_reporting/blacklist_load_analyzer_win.cc b/chrome/browser/safe_browsing/incident_reporting/blacklist_load_analyzer_win.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..1321ee2d6f127660f2f392cd5ad8fa89febacbee
|
| --- /dev/null
|
| +++ b/chrome/browser/safe_browsing/incident_reporting/blacklist_load_analyzer_win.cc
|
| @@ -0,0 +1,70 @@
|
| +// 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/incident_reporting/blacklist_load_analyzer.h"
|
| +
|
| +#include "base/files/file_path.h"
|
| +#include "base/logging.h"
|
| +#include "base/metrics/histogram.h"
|
| +#include "base/strings/string_number_conversions.h"
|
| +#include "base/strings/string_util.h"
|
| +#include "base/strings/utf_string_conversions.h"
|
| +#include "chrome/browser/install_verification/win/module_info.h"
|
| +#include "chrome/browser/install_verification/win/module_verification_common.h"
|
| +#include "chrome/browser/safe_browsing/binary_feature_extractor.h"
|
| +#include "chrome/browser/safe_browsing/incident_reporting/add_incident_callback.h"
|
| +#include "chrome/browser/safe_browsing/path_sanitizer.h"
|
| +#include "chrome/browser/safe_browsing/safe_browsing_service.h"
|
| +#include "chrome/common/safe_browsing/csd.pb.h"
|
| +#include "chrome_elf/blacklist/blacklist.h"
|
| +
|
| +namespace safe_browsing {
|
| +
|
| +// Retrieves the set of blacklisted modules that are loaded in the process.
|
| +// Returns true if successful, false otherwise.
|
| +bool GetLoadedBlacklistedModules(std::vector<base::string16>* module_names) {
|
| + DCHECK(module_names);
|
| +
|
| + std::set<ModuleInfo> module_info_set;
|
| + if (!GetLoadedModules(&module_info_set))
|
| + return false;
|
| +
|
| + std::set<ModuleInfo>::const_iterator module_iter(module_info_set.begin());
|
| + for (; module_iter != module_info_set.end(); ++module_iter) {
|
| + base::string16 module_file_name(base::StringToLowerASCII(
|
| + base::FilePath(module_iter->name).BaseName().value()));
|
| + if (blacklist::GetBlacklistIndex(module_file_name.c_str()) != -1) {
|
| + module_names->push_back(module_iter->name);
|
| + }
|
| + }
|
| +
|
| + return true;
|
| +}
|
| +
|
| +void VerifyBlacklistLoadState(const AddIncidentCallback& callback) {
|
| + std::vector<base::string16> module_names;
|
| + if (GetLoadedBlacklistedModules(&module_names)) {
|
| + PathSanitizer path_sanitizer;
|
| +
|
| + std::vector<base::string16>::const_iterator module_iter(
|
| + module_names.begin());
|
| + for (; module_iter != module_names.end(); ++module_iter) {
|
| + scoped_ptr<ClientIncidentReport_IncidentData> incident_data(
|
| + new ClientIncidentReport_IncidentData());
|
| + ClientIncidentReport_IncidentData_BlacklistLoadIncident* blacklist_load =
|
| + incident_data->mutable_blacklist_load();
|
| +
|
| + base::FilePath module_path(*module_iter);
|
| + path_sanitizer.StripHomeDirectory(&module_path);
|
| +
|
| + blacklist_load->set_path(base::WideToUTF8(module_path.value()));
|
| + // TODO(robertshield): Add computation of file digest and version here.
|
| +
|
| + // Send the report.
|
| + callback.Run(incident_data.Pass());
|
| + }
|
| + }
|
| +}
|
| +
|
| +} // namespace safe_browsing
|
|
|