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