123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175 |
- // Copyright (c) 2012 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 "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 "base/win/windows_version.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) {
- if (base::win::GetVersion() <= base::win::Version::WIN7) {
- // Disabled on Windows 7 due to flakiness. https://crbug.com/1336879
- GTEST_SKIP();
- }
- 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();
- }
|