| // 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/test/scoped_task_environment.h" |
| #include "base/threading/sequenced_task_runner_handle.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() |
| : module_database_(base::MakeUnique<ModuleDatabase>( |
| base::SequencedTaskRunnerHandle::Get())) {} |
| |
| void CreateModuleSinkImpl() { |
| module_event_sink_impl_ = base::MakeUnique<ModuleEventSinkImpl>( |
| ::GetCurrentProcess(), content::PROCESS_TYPE_BROWSER, |
| module_database_.get()); |
| } |
| |
| 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_; } |
| |
| // Must be before |module_database_|. |
| base::test::ScopedTaskEnvironment scoped_task_environment_; |
| 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(::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()); |
| } |