| Index: trunk/src/chrome/browser/safe_browsing/environment_data_collection_win_unittest.cc
|
| ===================================================================
|
| --- trunk/src/chrome/browser/safe_browsing/environment_data_collection_win_unittest.cc (revision 278720)
|
| +++ trunk/src/chrome/browser/safe_browsing/environment_data_collection_win_unittest.cc (working copy)
|
| @@ -1,113 +0,0 @@
|
| -// 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 <string>
|
| -
|
| -#include "base/files/file_path.h"
|
| -#include "base/scoped_native_library.h"
|
| -#include "chrome/common/safe_browsing/csd.pb.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -namespace {
|
| -
|
| -// Helper function that returns true if a dll with filename |dll_name| is
|
| -// found in |process_report|.
|
| -bool ProcessReportContainsDll(
|
| - const safe_browsing::ClientIncidentReport_EnvironmentData_Process&
|
| - process_report,
|
| - const base::FilePath& dll_name) {
|
| - for (int i = 0; i < process_report.dll_size(); ++i) {
|
| - base::FilePath current_dll =
|
| - base::FilePath::FromUTF8Unsafe(process_report.dll(i).path());
|
| -
|
| - if (current_dll.BaseName() == dll_name)
|
| - return true;
|
| - }
|
| -
|
| - return false;
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -TEST(SafeBrowsingEnvironmentDataCollectionWinTest, CollectDlls) {
|
| - // This test will check if the CollectDlls method works by loading
|
| - // a dll and then checking if we can find it within the process report.
|
| - // Pick msvidc32.dll as it is present from WinXP to Win8 and yet rarely used.
|
| - // msvidc32.dll exists in both 32 and 64 bit versions.
|
| - base::FilePath msvdc32_dll(L"msvidc32.dll");
|
| -
|
| - safe_browsing::ClientIncidentReport_EnvironmentData_Process process_report;
|
| - safe_browsing::CollectDlls(&process_report);
|
| -
|
| - ASSERT_FALSE(ProcessReportContainsDll(process_report, msvdc32_dll));
|
| -
|
| - // Redo the same verification after loading a new dll.
|
| - base::ScopedNativeLibrary library(msvdc32_dll);
|
| -
|
| - process_report.clear_dll();
|
| - safe_browsing::CollectDlls(&process_report);
|
| -
|
| - ASSERT_TRUE(ProcessReportContainsDll(process_report, msvdc32_dll));
|
| -}
|
| -
|
| -TEST(SafeBrowsingEnvironmentDataCollectionWinTest, RecordLspFeature) {
|
| - // Populate our incident report with loaded modules.
|
| - safe_browsing::ClientIncidentReport_EnvironmentData_Process process_report;
|
| - safe_browsing::CollectDlls(&process_report);
|
| -
|
| - // We'll test RecordLspFeatures against a real dll registered as a LSP.
|
| - std::string lsp_path = "C:\\Windows\\system32\\mswsock.dll";
|
| - int base_address = 0x77770000;
|
| - int length = 0x180000;
|
| -
|
| - // Look for dlls that are registered as a LSP.
|
| - safe_browsing::RecordLspFeature(&process_report);
|
| -
|
| - // Look through dll entries and check that none contains the LSP feature.
|
| - bool lsp_feature_found = false;
|
| - for (int i = 0; i < process_report.dll_size(); ++i) {
|
| - if (process_report.dll(i).path() == lsp_path) {
|
| - // Look for ClientIncidentReport_EnvironmentData_Process_DLL_Feature_LSP
|
| - // through the features of each dll.
|
| - for (int j = 0; j < process_report.dll(i).feature_size(); ++j) {
|
| - if (process_report.dll(i).feature(j) ==
|
| - safe_browsing::ClientIncidentReport_EnvironmentData_Process_Dll::
|
| - LSP)
|
| - lsp_feature_found = true;
|
| - }
|
| - }
|
| - }
|
| -
|
| - ASSERT_FALSE(lsp_feature_found);
|
| -
|
| - // Manually add an entry to the process report that will get marked as a LSP.
|
| - safe_browsing::ClientIncidentReport_EnvironmentData_Process_Dll* dll =
|
| - process_report.add_dll();
|
| - dll->set_path(lsp_path);
|
| - dll->set_base_address(base_address);
|
| - dll->set_length(length);
|
| -
|
| - // Look for dlls that are registered as a LSP.
|
| - safe_browsing::RecordLspFeature(&process_report);
|
| -
|
| - // Look through dll entries and check if the one we added contains the LSP
|
| - // feature.
|
| - lsp_feature_found = false;
|
| - for (int i = 0; i < process_report.dll_size(); ++i) {
|
| - if (process_report.dll(i).path() == lsp_path) {
|
| - // Look for ClientIncidentReport_EnvironmentData_Process_DLL_Feature_LSP
|
| - // through the features of each dll.
|
| - for (int j = 0; j < process_report.dll(i).feature_size(); ++j) {
|
| - if (process_report.dll(i).feature(j) ==
|
| - safe_browsing::ClientIncidentReport_EnvironmentData_Process_Dll::
|
| - LSP)
|
| - lsp_feature_found = true;
|
| - }
|
| - }
|
| - }
|
| -
|
| - ASSERT_TRUE(lsp_feature_found);
|
| -}
|
|
|