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
|
// 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/loaded_module_verification.h"
#include "chrome/browser/install_verification/win/module_verification_test.h"
#include "testing/gtest/include/gtest/gtest.h"
class LoadedModuleVerificationTest : public ModuleVerificationTest { };
TEST_F(LoadedModuleVerificationTest, TestCase) {
std::set<ModuleInfo> loaded_modules;
ModuleIDs empty_modules_of_interest;
ModuleIDs non_matching_modules_of_interest;
ModuleIDs matching_modules_of_interest;
matching_modules_of_interest.insert(
std::make_pair(CalculateModuleNameDigest(L"fancy_pants.dll"), 999u));
matching_modules_of_interest.insert(
std::make_pair(CalculateModuleNameDigest(L"advapi32.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.
VerifyLoadedModules(loaded_modules,
empty_modules_of_interest,
&ModuleVerificationTest::ReportModule);
ASSERT_TRUE(reported_module_ids_.empty());
VerifyLoadedModules(loaded_modules,
non_matching_modules_of_interest,
&ModuleVerificationTest::ReportModule);
ASSERT_TRUE(reported_module_ids_.empty());
VerifyLoadedModules(loaded_modules,
matching_modules_of_interest,
&ModuleVerificationTest::ReportModule);
ASSERT_TRUE(reported_module_ids_.empty());
// With populated loaded_modules, only the 'matching' module data gives a
// match.
ASSERT_TRUE(GetLoadedModuleInfoSet(&loaded_modules));
VerifyLoadedModules(loaded_modules,
empty_modules_of_interest,
&ModuleVerificationTest::ReportModule);
ASSERT_TRUE(reported_module_ids_.empty());
VerifyLoadedModules(loaded_modules,
non_matching_modules_of_interest,
&ModuleVerificationTest::ReportModule);
ASSERT_TRUE(reported_module_ids_.empty());
VerifyLoadedModules(loaded_modules,
matching_modules_of_interest,
&ModuleVerificationTest::ReportModule);
ASSERT_EQ(3u, reported_module_ids_.size());
ASSERT_NE(reported_module_ids_.end(), reported_module_ids_.find(666u));
ASSERT_NE(reported_module_ids_.end(), reported_module_ids_.find(777u));
ASSERT_NE(reported_module_ids_.end(), reported_module_ids_.find(888u));
}
|
