| 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 "base/test/multiprocess_test.h" | |
| 6 | |
| 7 #include <unistd.h> | |
| 8 | |
| 9 #include "base/base_switches.h" | |
| 10 #include "base/command_line.h" | |
| 11 #include "base/containers/hash_tables.h" | |
| 12 #include "base/logging.h" | |
| 13 #include "base/posix/global_descriptors.h" | |
| 14 #include "testing/multiprocess_func_list.h" | |
| 15 | |
| 16 namespace base { | |
| 17 | |
| 18 // A very basic implementation for Android. On Android tests can run in an APK | |
| 19 // and we don't have an executable to exec*. This implementation does the bare | |
| 20 // minimum to execute the method specified by procname (in the child process). | |
| 21 // - All options except |fds_to_remap| are ignored. | |
| 22 Process SpawnMultiProcessTestChild(const std::string& procname, | |
| 23 const CommandLine& base_command_line, | |
| 24 const LaunchOptions& options) { | |
| 25 // TODO(viettrungluu): The FD-remapping done below is wrong in the presence of | |
| 26 // cycles (e.g., fd1 -> fd2, fd2 -> fd1). crbug.com/326576 | |
| 27 FileHandleMappingVector empty; | |
| 28 const FileHandleMappingVector* fds_to_remap = | |
| 29 options.fds_to_remap ? options.fds_to_remap : ∅ | |
| 30 | |
| 31 pid_t pid = fork(); | |
| 32 | |
| 33 if (pid < 0) { | |
| 34 PLOG(ERROR) << "fork"; | |
| 35 return Process(); | |
| 36 } | |
| 37 if (pid > 0) { | |
| 38 // Parent process. | |
| 39 return Process(pid); | |
| 40 } | |
| 41 // Child process. | |
| 42 base::hash_set<int> fds_to_keep_open; | |
| 43 for (FileHandleMappingVector::const_iterator it = fds_to_remap->begin(); | |
| 44 it != fds_to_remap->end(); ++it) { | |
| 45 fds_to_keep_open.insert(it->first); | |
| 46 } | |
| 47 // Keep standard FDs (stdin, stdout, stderr, etc.) open since this | |
| 48 // is not meant to spawn a daemon. | |
| 49 int base = GlobalDescriptors::kBaseDescriptor; | |
| 50 for (int fd = base; fd < sysconf(_SC_OPEN_MAX); ++fd) { | |
| 51 if (fds_to_keep_open.find(fd) == fds_to_keep_open.end()) { | |
| 52 close(fd); | |
| 53 } | |
| 54 } | |
| 55 for (FileHandleMappingVector::const_iterator it = fds_to_remap->begin(); | |
| 56 it != fds_to_remap->end(); ++it) { | |
| 57 int old_fd = it->first; | |
| 58 int new_fd = it->second; | |
| 59 if (dup2(old_fd, new_fd) < 0) { | |
| 60 PLOG(FATAL) << "dup2"; | |
| 61 } | |
| 62 close(old_fd); | |
| 63 } | |
| 64 CommandLine::Reset(); | |
| 65 CommandLine::Init(0, nullptr); | |
| 66 CommandLine* command_line = CommandLine::ForCurrentProcess(); | |
| 67 command_line->InitFromArgv(base_command_line.argv()); | |
| 68 if (!command_line->HasSwitch(switches::kTestChildProcess)) | |
| 69 command_line->AppendSwitchASCII(switches::kTestChildProcess, procname); | |
| 70 | |
| 71 _exit(multi_process_function_list::InvokeChildProcessTest(procname)); | |
| 72 return Process(); | |
| 73 } | |
| 74 | |
| 75 } // namespace base | |
| OLD | NEW |