Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(20)

Side by Side Diff: base/process/process_unittest.cc

Issue 863253002: Update from https://crrev.com/312600 (Closed) Base URL: https://github.com/domokit/mojo.git@master
Patch Set: Created 5 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2014 The Chromium Authors. All rights reserved. 1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "base/process/process.h" 5 #include "base/process/process.h"
6 6
7 #include "base/files/file_path.h"
8 #include "base/files/file_util.h"
9 #include "base/files/scoped_file.h"
10 #include "base/posix/eintr_wrapper.h"
11 #include "base/process/kill.h" 7 #include "base/process/kill.h"
12 #include "base/test/multiprocess_test.h" 8 #include "base/test/multiprocess_test.h"
13 #include "base/test/test_timeouts.h" 9 #include "base/test/test_timeouts.h"
14 #include "base/third_party/dynamic_annotations/dynamic_annotations.h"
15 #include "base/threading/platform_thread.h" 10 #include "base/threading/platform_thread.h"
16 #include "build/build_config.h"
17 #include "testing/gtest/include/gtest/gtest.h" 11 #include "testing/gtest/include/gtest/gtest.h"
18 #include "testing/multiprocess_func_list.h" 12 #include "testing/multiprocess_func_list.h"
19 13
20 #if defined(OS_LINUX)
21 #include <errno.h>
22 #include <sched.h>
23 #include <sys/syscall.h>
24 #include <sys/types.h>
25 #include <sys/wait.h>
26 #include <unistd.h>
27 #endif
28
29 namespace { 14 namespace {
30 15
31 #if defined(OS_WIN) 16 #if defined(OS_WIN)
32 const int kExpectedStillRunningExitCode = 0x102; 17 const int kExpectedStillRunningExitCode = 0x102;
33 #else 18 #else
34 const int kExpectedStillRunningExitCode = 0; 19 const int kExpectedStillRunningExitCode = 0;
35 #endif 20 #endif
36 21
37 } // namespace 22 } // namespace
38 23
(...skipping 169 matching lines...) Expand 10 before | Expand all | Expand 10 after
208 EXPECT_TRUE(process.SetProcessBackgrounded(false)); 193 EXPECT_TRUE(process.SetProcessBackgrounded(false));
209 EXPECT_FALSE(process.IsProcessBackgrounded()); 194 EXPECT_FALSE(process.IsProcessBackgrounded());
210 #else 195 #else
211 process.SetProcessBackgrounded(true); 196 process.SetProcessBackgrounded(true);
212 process.SetProcessBackgrounded(false); 197 process.SetProcessBackgrounded(false);
213 #endif 198 #endif
214 int new_priority = process.GetPriority(); 199 int new_priority = process.GetPriority();
215 EXPECT_EQ(old_priority, new_priority); 200 EXPECT_EQ(old_priority, new_priority);
216 } 201 }
217 202
218 #if defined(OS_LINUX)
219 const int kSuccess = 0;
220
221 MULTIPROCESS_TEST_MAIN(CheckPidProcess) {
222 const pid_t kInitPid = 1;
223 const pid_t pid = syscall(__NR_getpid);
224 CHECK(pid == kInitPid);
225 CHECK(getpid() == pid);
226 return kSuccess;
227 }
228
229 TEST_F(ProcessTest, CloneFlags) {
230 if (RunningOnValgrind() || !PathExists(FilePath("/proc/self/ns/user")) ||
231 !PathExists(FilePath("/proc/self/ns/pid"))) {
232 // User or PID namespaces are not supported.
233 return;
234 }
235
236 LaunchOptions options;
237 options.clone_flags = CLONE_NEWUSER | CLONE_NEWPID;
238
239 Process process(SpawnChildWithOptions("CheckPidProcess", options));
240 ASSERT_TRUE(process.IsValid());
241
242 int exit_code = 42;
243 EXPECT_TRUE(process.WaitForExit(&exit_code));
244 EXPECT_EQ(kSuccess, exit_code);
245 }
246
247 TEST(ForkWithFlagsTest, UpdatesPidCache) {
248 // The libc clone function, which allows ForkWithFlags to keep the pid cache
249 // up to date, does not work on Valgrind.
250 if (RunningOnValgrind()) {
251 return;
252 }
253
254 // Warm up the libc pid cache, if there is one.
255 ASSERT_EQ(syscall(__NR_getpid), getpid());
256
257 pid_t ctid = 0;
258 const pid_t pid = ForkWithFlags(SIGCHLD | CLONE_CHILD_SETTID, nullptr, &ctid);
259 if (pid == 0) {
260 // In child. Check both the raw getpid syscall and the libc getpid wrapper
261 // (which may rely on a pid cache).
262 RAW_CHECK(syscall(__NR_getpid) == ctid);
263 RAW_CHECK(getpid() == ctid);
264 _exit(kSuccess);
265 }
266
267 ASSERT_NE(-1, pid);
268 int status = 42;
269 ASSERT_EQ(pid, HANDLE_EINTR(waitpid(pid, &status, 0)));
270 ASSERT_TRUE(WIFEXITED(status));
271 EXPECT_EQ(kSuccess, WEXITSTATUS(status));
272 }
273 #endif
274
275 #if defined(OS_POSIX) && !defined(OS_ANDROID)
276 const char kPipeValue = '\xcc';
277
278 class ReadFromPipeDelegate : public LaunchOptions::PreExecDelegate {
279 public:
280 explicit ReadFromPipeDelegate(int fd) : fd_(fd) {}
281 ~ReadFromPipeDelegate() override {}
282 void RunAsyncSafe() override {
283 char c;
284 RAW_CHECK(HANDLE_EINTR(read(fd_, &c, 1)) == 1);
285 RAW_CHECK(IGNORE_EINTR(close(fd_)) == 0);
286 RAW_CHECK(c == kPipeValue);
287 }
288
289 private:
290 int fd_;
291 DISALLOW_COPY_AND_ASSIGN(ReadFromPipeDelegate);
292 };
293
294 TEST_F(ProcessTest, PreExecHook) {
295 int pipe_fds[2];
296 ASSERT_EQ(0, pipe(pipe_fds));
297
298 ScopedFD read_fd(pipe_fds[0]);
299 ScopedFD write_fd(pipe_fds[1]);
300 base::FileHandleMappingVector fds_to_remap;
301 fds_to_remap.push_back(std::make_pair(read_fd.get(), read_fd.get()));
302
303 ReadFromPipeDelegate read_from_pipe_delegate(read_fd.get());
304 LaunchOptions options;
305 options.fds_to_remap = &fds_to_remap;
306 options.pre_exec_delegate = &read_from_pipe_delegate;
307 Process process(SpawnChildWithOptions("SimpleChildProcess", options));
308 ASSERT_TRUE(process.IsValid());
309
310 read_fd.reset();
311 ASSERT_EQ(1, HANDLE_EINTR(write(write_fd.get(), &kPipeValue, 1)));
312
313 int exit_code = 42;
314 EXPECT_TRUE(process.WaitForExit(&exit_code));
315 EXPECT_EQ(0, exit_code);
316 }
317 #endif // defined(OS_POSIX) && !defined(OS_ANDROID)
318
319 } // namespace base 203 } // namespace base
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698