| // Copyright 2012 The Chromium Authors |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "base/win/scoped_process_information.h" |
| |
| #include <windows.h> |
| |
| #include <string> |
| |
| #include "base/command_line.h" |
| #include "base/process/kill.h" |
| #include "base/process/process.h" |
| #include "base/strings/string_util.h" |
| #include "base/test/multiprocess_test.h" |
| #include "testing/multiprocess_func_list.h" |
| |
| namespace { |
| |
| const DWORD kProcessId = 4321; |
| const DWORD kThreadId = 1234; |
| const HANDLE kProcessHandle = reinterpret_cast<HANDLE>(7651); |
| const HANDLE kThreadHandle = reinterpret_cast<HANDLE>(1567); |
| |
| void MockCreateProcess(base::win::ScopedProcessInformation* process_info) { |
| PROCESS_INFORMATION process_information = {}; |
| process_information.dwProcessId = kProcessId; |
| process_information.dwThreadId = kThreadId; |
| process_information.hProcess = kProcessHandle; |
| process_information.hThread = kThreadHandle; |
| process_info->Set(process_information); |
| } |
| |
| } // namespace |
| |
| class ScopedProcessInformationTest : public base::MultiProcessTest { |
| protected: |
| void DoCreateProcess(const std::string& main_id, |
| PROCESS_INFORMATION* process_handle); |
| }; |
| |
| MULTIPROCESS_TEST_MAIN(ReturnSeven) { |
| return 7; |
| } |
| |
| MULTIPROCESS_TEST_MAIN(ReturnNine) { |
| return 9; |
| } |
| |
| void ScopedProcessInformationTest::DoCreateProcess( |
| const std::string& main_id, |
| PROCESS_INFORMATION* process_handle) { |
| base::CommandLine::StringType cmd_line = |
| MakeCmdLine(main_id).GetCommandLineString(); |
| STARTUPINFO startup_info = {}; |
| startup_info.cb = sizeof(startup_info); |
| |
| EXPECT_TRUE(::CreateProcess(nullptr, std::data(cmd_line), nullptr, nullptr, |
| false, 0, nullptr, nullptr, &startup_info, |
| process_handle)); |
| } |
| |
| TEST_F(ScopedProcessInformationTest, InitiallyInvalid) { |
| base::win::ScopedProcessInformation process_info; |
| ASSERT_FALSE(process_info.IsValid()); |
| } |
| |
| TEST_F(ScopedProcessInformationTest, Receive) { |
| base::win::ScopedProcessInformation process_info; |
| MockCreateProcess(&process_info); |
| |
| EXPECT_TRUE(process_info.IsValid()); |
| EXPECT_EQ(kProcessId, process_info.process_id()); |
| EXPECT_EQ(kThreadId, process_info.thread_id()); |
| EXPECT_EQ(kProcessHandle, process_info.process_handle()); |
| EXPECT_EQ(kThreadHandle, process_info.thread_handle()); |
| process_info.Take(); |
| } |
| |
| TEST_F(ScopedProcessInformationTest, TakeProcess) { |
| base::win::ScopedProcessInformation process_info; |
| MockCreateProcess(&process_info); |
| |
| HANDLE process = process_info.TakeProcessHandle(); |
| EXPECT_EQ(kProcessHandle, process); |
| EXPECT_EQ(nullptr, process_info.process_handle()); |
| EXPECT_EQ(0u, process_info.process_id()); |
| EXPECT_TRUE(process_info.IsValid()); |
| process_info.Take(); |
| } |
| |
| TEST_F(ScopedProcessInformationTest, TakeThread) { |
| base::win::ScopedProcessInformation process_info; |
| MockCreateProcess(&process_info); |
| |
| HANDLE thread = process_info.TakeThreadHandle(); |
| EXPECT_EQ(kThreadHandle, thread); |
| EXPECT_EQ(nullptr, process_info.thread_handle()); |
| EXPECT_EQ(0u, process_info.thread_id()); |
| EXPECT_TRUE(process_info.IsValid()); |
| process_info.Take(); |
| } |
| |
| TEST_F(ScopedProcessInformationTest, TakeBoth) { |
| base::win::ScopedProcessInformation process_info; |
| MockCreateProcess(&process_info); |
| |
| process_info.TakeProcessHandle(); |
| process_info.TakeThreadHandle(); |
| EXPECT_FALSE(process_info.IsValid()); |
| process_info.Take(); |
| } |
| |
| TEST_F(ScopedProcessInformationTest, TakeWholeStruct) { |
| base::win::ScopedProcessInformation process_info; |
| MockCreateProcess(&process_info); |
| |
| PROCESS_INFORMATION to_discard = process_info.Take(); |
| EXPECT_EQ(kProcessId, to_discard.dwProcessId); |
| EXPECT_EQ(kThreadId, to_discard.dwThreadId); |
| EXPECT_EQ(kProcessHandle, to_discard.hProcess); |
| EXPECT_EQ(kThreadHandle, to_discard.hThread); |
| EXPECT_FALSE(process_info.IsValid()); |
| } |
| |
| TEST_F(ScopedProcessInformationTest, Duplicate) { |
| PROCESS_INFORMATION temp_process_information; |
| DoCreateProcess("ReturnSeven", &temp_process_information); |
| base::win::ScopedProcessInformation process_info; |
| process_info.Set(temp_process_information); |
| |
| base::win::ScopedProcessInformation duplicate; |
| duplicate.DuplicateFrom(process_info); |
| |
| ASSERT_TRUE(process_info.IsValid()); |
| ASSERT_NE(0u, process_info.process_id()); |
| ASSERT_EQ(duplicate.process_id(), process_info.process_id()); |
| ASSERT_NE(0u, process_info.thread_id()); |
| ASSERT_EQ(duplicate.thread_id(), process_info.thread_id()); |
| |
| // Validate that we have separate handles that are good. |
| int exit_code = 0; |
| base::Process process(process_info.TakeProcessHandle()); |
| ASSERT_TRUE(process.WaitForExit(&exit_code)); |
| ASSERT_EQ(7, exit_code); |
| |
| exit_code = 0; |
| base::Process dup_process(duplicate.TakeProcessHandle()); |
| ASSERT_TRUE(dup_process.WaitForExit(&exit_code)); |
| ASSERT_EQ(7, exit_code); |
| |
| ASSERT_TRUE(::CloseHandle(process_info.TakeThreadHandle())); |
| ASSERT_TRUE(::CloseHandle(duplicate.TakeThreadHandle())); |
| } |
| |
| TEST_F(ScopedProcessInformationTest, Set) { |
| base::win::ScopedProcessInformation base_process_info; |
| MockCreateProcess(&base_process_info); |
| |
| PROCESS_INFORMATION base_struct = base_process_info.Take(); |
| |
| base::win::ScopedProcessInformation process_info; |
| process_info.Set(base_struct); |
| |
| EXPECT_EQ(kProcessId, process_info.process_id()); |
| EXPECT_EQ(kThreadId, process_info.thread_id()); |
| EXPECT_EQ(kProcessHandle, process_info.process_handle()); |
| EXPECT_EQ(kThreadHandle, process_info.thread_handle()); |
| base_struct = process_info.Take(); |
| } |