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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
// Copyright 2017 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/conflicts/module_event_sink_impl_win.h"
#include <memory>
#include "base/memory/ptr_util.h"
#include "base/message_loop/message_loop.h"
#include "chrome/browser/conflicts/module_database_win.h"
#include "chrome/common/conflicts/module_watcher_win.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
// The address of this module in memory. The linker will take care of defining
// this symbol.
extern "C" IMAGE_DOS_HEADER __ImageBase;
// An invalid load address.
const uint64_t kInvalidLoadAddress = 0xDEADBEEF;
} // namespace
class ModuleEventSinkImplTest : public testing::Test {
protected:
ModuleEventSinkImplTest()
: message_loop_(base::MakeUnique<base::MessageLoop>()),
module_database_(
base::MakeUnique<ModuleDatabase>(message_loop_->task_runner())) {}
void CreateModuleSinkImpl() {
module_event_sink_impl_ = base::MakeUnique<ModuleEventSinkImpl>(
::GetCurrentProcess(), content::PROCESS_TYPE_BROWSER,
module_database_.get());
}
ModuleDatabase* module_database() {
return module_event_sink_impl_->module_database_;
}
const ModuleDatabase::ModuleMap& modules() {
return module_database_->modules_;
}
const ModuleDatabase::ProcessMap& processes() {
return module_database_->processes_;
}
uint32_t process_id() { return module_event_sink_impl_->process_id_; }
std::unique_ptr<base::MessageLoop> message_loop_;
std::unique_ptr<ModuleDatabase> module_database_;
std::unique_ptr<ModuleEventSinkImpl> module_event_sink_impl_;
private:
DISALLOW_COPY_AND_ASSIGN(ModuleEventSinkImplTest);
};
TEST_F(ModuleEventSinkImplTest, CallsForwardedAsExpected) {
const uintptr_t kValidLoadAddress = reinterpret_cast<uintptr_t>(&__ImageBase);
EXPECT_EQ(0u, modules().size());
EXPECT_EQ(0u, processes().size());
// Construction should immediately fire off a call to OnProcessStarted and
// create a process entry in the module database.
CreateModuleSinkImpl();
EXPECT_EQ(module_database_.get(), module_database());
EXPECT_EQ(::GetCurrentProcessId(), process_id());
EXPECT_EQ(0u, modules().size());
EXPECT_EQ(1u, processes().size());
// An invalid load event should not cause a module entry.
module_event_sink_impl_->OnModuleEvent(
mojom::ModuleEventType::MODULE_ALREADY_LOADED, kInvalidLoadAddress);
EXPECT_EQ(0u, modules().size());
EXPECT_EQ(1u, processes().size());
// A valid load event should cause a module entry.
module_event_sink_impl_->OnModuleEvent(mojom::ModuleEventType::MODULE_LOADED,
kValidLoadAddress);
EXPECT_EQ(1u, modules().size());
EXPECT_EQ(1u, processes().size());
}
|
