File: imported_module_verification_unittest.cc

package info (click to toggle)
chromium-browser 41.0.2272.118-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie-kfreebsd
  • size: 2,189,132 kB
  • sloc: cpp: 9,691,462; ansic: 3,341,451; python: 712,689; asm: 518,779; xml: 208,926; java: 169,820; sh: 119,353; perl: 68,907; makefile: 28,311; yacc: 13,305; objc: 11,385; tcl: 3,186; cs: 2,225; sql: 2,217; lex: 2,215; lisp: 1,349; pascal: 1,256; awk: 407; ruby: 155; sed: 53; php: 14; exp: 11
file content (70 lines) | stat: -rw-r--r-- 3,137 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// Copyright 2013 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/install_verification/win/imported_module_verification.h"

#include <set>
#include <utility>
#include <vector>
#include "chrome/browser/install_verification/win/module_ids.h"
#include "chrome/browser/install_verification/win/module_info.h"
#include "chrome/browser/install_verification/win/module_verification_test.h"
#include "testing/gtest/include/gtest/gtest.h"

class ImportedModuleVerificationTest : public ModuleVerificationTest {};

TEST_F(ImportedModuleVerificationTest, TestCase) {
// Note that module verification is temporarily disabled for 64-bit builds.
// See crbug.com/316274.
#if !defined(_WIN64)
  std::set<ModuleInfo> loaded_modules;
  ModuleIDs empty_modules_of_interest;
  ModuleIDs non_matching_modules_of_interest;
  ModuleIDs matching_modules_of_interest;

  // This set of IDs will match because user32.dll imports kernel32.dll.
  matching_modules_of_interest.insert(
      std::make_pair(CalculateModuleNameDigest(L"fancy_pants.dll"), 999u));
  matching_modules_of_interest.insert(
      std::make_pair(CalculateModuleNameDigest(L"kernel32.dll"), 666u));
  matching_modules_of_interest.insert(
      std::make_pair(CalculateModuleNameDigest(L"unit_tests.exe"), 777u));
  matching_modules_of_interest.insert(
      std::make_pair(CalculateModuleNameDigest(L"user32.dll"), 888u));

  non_matching_modules_of_interest.insert(
      std::make_pair(CalculateModuleNameDigest(L"fancy_pants.dll"), 999u));

  // With empty loaded_modules, nothing matches.
  VerifyImportedModules(loaded_modules,
                        empty_modules_of_interest,
                        &ImportedModuleVerificationTest::ReportModule);
  ASSERT_TRUE(reported_module_ids_.empty());
  VerifyImportedModules(loaded_modules,
                        non_matching_modules_of_interest,
                        &ImportedModuleVerificationTest::ReportModule);
  ASSERT_TRUE(reported_module_ids_.empty());
  VerifyImportedModules(loaded_modules,
                        matching_modules_of_interest,
                        &ImportedModuleVerificationTest::ReportModule);
  ASSERT_TRUE(reported_module_ids_.empty());

  // With populated loaded_modules, only the 'matching' module data gives a
  // match.
  ASSERT_TRUE(GetLoadedModuleInfoSet(&loaded_modules));
  VerifyImportedModules(loaded_modules,
                        empty_modules_of_interest,
                        &ImportedModuleVerificationTest::ReportModule);
  ASSERT_TRUE(reported_module_ids_.empty());
  VerifyImportedModules(loaded_modules,
                        non_matching_modules_of_interest,
                        &ImportedModuleVerificationTest::ReportModule);
  ASSERT_TRUE(reported_module_ids_.empty());
  VerifyImportedModules(loaded_modules,
                        matching_modules_of_interest,
                        &ImportedModuleVerificationTest::ReportModule);
  ASSERT_EQ(1u, reported_module_ids_.size());
  ASSERT_EQ(666u, *reported_module_ids_.begin());
#endif
}