1 // Copyright 2012 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/win/scoped_process_information.h"
6
7 #include <windows.h>
8
9 #include <string>
10
11 #include "base/command_line.h"
12 #include "base/process/kill.h"
13 #include "base/process/process.h"
14 #include "base/strings/string_util.h"
15 #include "base/test/multiprocess_test.h"
16 #include "testing/multiprocess_func_list.h"
17
18 namespace {
19
20 const DWORD kProcessId = 4321;
21 const DWORD kThreadId = 1234;
22 const HANDLE kProcessHandle = reinterpret_cast<HANDLE>(7651);
23 const HANDLE kThreadHandle = reinterpret_cast<HANDLE>(1567);
24
MockCreateProcess(base::win::ScopedProcessInformation * process_info)25 void MockCreateProcess(base::win::ScopedProcessInformation* process_info) {
26 PROCESS_INFORMATION process_information = {};
27 process_information.dwProcessId = kProcessId;
28 process_information.dwThreadId = kThreadId;
29 process_information.hProcess = kProcessHandle;
30 process_information.hThread = kThreadHandle;
31 process_info->Set(process_information);
32 }
33
34 } // namespace
35
36 class ScopedProcessInformationTest : public base::MultiProcessTest {
37 protected:
38 void DoCreateProcess(const std::string& main_id,
39 PROCESS_INFORMATION* process_handle);
40 };
41
MULTIPROCESS_TEST_MAIN(ReturnSeven)42 MULTIPROCESS_TEST_MAIN(ReturnSeven) {
43 return 7;
44 }
45
MULTIPROCESS_TEST_MAIN(ReturnNine)46 MULTIPROCESS_TEST_MAIN(ReturnNine) {
47 return 9;
48 }
49
DoCreateProcess(const std::string & main_id,PROCESS_INFORMATION * process_handle)50 void ScopedProcessInformationTest::DoCreateProcess(
51 const std::string& main_id,
52 PROCESS_INFORMATION* process_handle) {
53 base::CommandLine::StringType cmd_line =
54 MakeCmdLine(main_id).GetCommandLineString();
55 STARTUPINFO startup_info = {};
56 startup_info.cb = sizeof(startup_info);
57
58 EXPECT_TRUE(::CreateProcess(nullptr, std::data(cmd_line), nullptr, nullptr,
59 false, 0, nullptr, nullptr, &startup_info,
60 process_handle));
61 }
62
TEST_F(ScopedProcessInformationTest,InitiallyInvalid)63 TEST_F(ScopedProcessInformationTest, InitiallyInvalid) {
64 base::win::ScopedProcessInformation process_info;
65 ASSERT_FALSE(process_info.IsValid());
66 }
67
TEST_F(ScopedProcessInformationTest,Receive)68 TEST_F(ScopedProcessInformationTest, Receive) {
69 base::win::ScopedProcessInformation process_info;
70 MockCreateProcess(&process_info);
71
72 EXPECT_TRUE(process_info.IsValid());
73 EXPECT_EQ(kProcessId, process_info.process_id());
74 EXPECT_EQ(kThreadId, process_info.thread_id());
75 EXPECT_EQ(kProcessHandle, process_info.process_handle());
76 EXPECT_EQ(kThreadHandle, process_info.thread_handle());
77 process_info.Take();
78 }
79
TEST_F(ScopedProcessInformationTest,TakeProcess)80 TEST_F(ScopedProcessInformationTest, TakeProcess) {
81 base::win::ScopedProcessInformation process_info;
82 MockCreateProcess(&process_info);
83
84 HANDLE process = process_info.TakeProcessHandle();
85 EXPECT_EQ(kProcessHandle, process);
86 EXPECT_EQ(nullptr, process_info.process_handle());
87 EXPECT_EQ(0u, process_info.process_id());
88 EXPECT_TRUE(process_info.IsValid());
89 process_info.Take();
90 }
91
TEST_F(ScopedProcessInformationTest,TakeThread)92 TEST_F(ScopedProcessInformationTest, TakeThread) {
93 base::win::ScopedProcessInformation process_info;
94 MockCreateProcess(&process_info);
95
96 HANDLE thread = process_info.TakeThreadHandle();
97 EXPECT_EQ(kThreadHandle, thread);
98 EXPECT_EQ(nullptr, process_info.thread_handle());
99 EXPECT_EQ(0u, process_info.thread_id());
100 EXPECT_TRUE(process_info.IsValid());
101 process_info.Take();
102 }
103
TEST_F(ScopedProcessInformationTest,TakeBoth)104 TEST_F(ScopedProcessInformationTest, TakeBoth) {
105 base::win::ScopedProcessInformation process_info;
106 MockCreateProcess(&process_info);
107
108 process_info.TakeProcessHandle();
109 process_info.TakeThreadHandle();
110 EXPECT_FALSE(process_info.IsValid());
111 process_info.Take();
112 }
113
TEST_F(ScopedProcessInformationTest,TakeWholeStruct)114 TEST_F(ScopedProcessInformationTest, TakeWholeStruct) {
115 base::win::ScopedProcessInformation process_info;
116 MockCreateProcess(&process_info);
117
118 PROCESS_INFORMATION to_discard = process_info.Take();
119 EXPECT_EQ(kProcessId, to_discard.dwProcessId);
120 EXPECT_EQ(kThreadId, to_discard.dwThreadId);
121 EXPECT_EQ(kProcessHandle, to_discard.hProcess);
122 EXPECT_EQ(kThreadHandle, to_discard.hThread);
123 EXPECT_FALSE(process_info.IsValid());
124 }
125
TEST_F(ScopedProcessInformationTest,Duplicate)126 TEST_F(ScopedProcessInformationTest, Duplicate) {
127 PROCESS_INFORMATION temp_process_information;
128 DoCreateProcess("ReturnSeven", &temp_process_information);
129 base::win::ScopedProcessInformation process_info;
130 process_info.Set(temp_process_information);
131
132 base::win::ScopedProcessInformation duplicate;
133 duplicate.DuplicateFrom(process_info);
134
135 ASSERT_TRUE(process_info.IsValid());
136 ASSERT_NE(0u, process_info.process_id());
137 ASSERT_EQ(duplicate.process_id(), process_info.process_id());
138 ASSERT_NE(0u, process_info.thread_id());
139 ASSERT_EQ(duplicate.thread_id(), process_info.thread_id());
140
141 // Validate that we have separate handles that are good.
142 int exit_code = 0;
143 base::Process process(process_info.TakeProcessHandle());
144 ASSERT_TRUE(process.WaitForExit(&exit_code));
145 ASSERT_EQ(7, exit_code);
146
147 exit_code = 0;
148 base::Process dup_process(duplicate.TakeProcessHandle());
149 ASSERT_TRUE(dup_process.WaitForExit(&exit_code));
150 ASSERT_EQ(7, exit_code);
151
152 ASSERT_TRUE(::CloseHandle(process_info.TakeThreadHandle()));
153 ASSERT_TRUE(::CloseHandle(duplicate.TakeThreadHandle()));
154 }
155
TEST_F(ScopedProcessInformationTest,Set)156 TEST_F(ScopedProcessInformationTest, Set) {
157 base::win::ScopedProcessInformation base_process_info;
158 MockCreateProcess(&base_process_info);
159
160 PROCESS_INFORMATION base_struct = base_process_info.Take();
161
162 base::win::ScopedProcessInformation process_info;
163 process_info.Set(base_struct);
164
165 EXPECT_EQ(kProcessId, process_info.process_id());
166 EXPECT_EQ(kThreadId, process_info.thread_id());
167 EXPECT_EQ(kProcessHandle, process_info.process_handle());
168 EXPECT_EQ(kThreadHandle, process_info.thread_handle());
169 base_struct = process_info.Take();
170 }
171