Home | History | Annotate | Download | only in test
      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/containers/hash_tables.h"
     10 #include "base/logging.h"
     11 #include "base/posix/eintr_wrapper.h"
     12 #include "testing/multiprocess_func_list.h"
     13 
     14 namespace base {
     15 
     16 // A very basic implementation for android. On Android tests can run in an APK
     17 // and we don't have an executable to exec*. This implementation does the bare
     18 // minimum to execute the method specified by procname (in the child process).
     19 //  - |debug_on_start| is ignored.
     20 ProcessHandle MultiProcessTest::SpawnChildImpl(
     21     const std::string& procname,
     22     const FileHandleMappingVector& fds_to_remap,
     23     bool debug_on_start) {
     24   pid_t pid = fork();
     25 
     26   if (pid < 0) {
     27     PLOG(ERROR) << "fork";
     28     return kNullProcessHandle;
     29   }
     30   if (pid > 0) {
     31     // Parent process.
     32     return pid;
     33   }
     34   // Child process.
     35   std::hash_set<int> fds_to_keep_open;
     36   for (FileHandleMappingVector::const_iterator it = fds_to_remap.begin();
     37        it != fds_to_remap.end(); ++it) {
     38     fds_to_keep_open.insert(it->first);
     39   }
     40   // Keep stdin, stdout and stderr open since this is not meant to spawn a
     41   // daemon.
     42   const int kFdForAndroidLogging = 3;  // FD used by __android_log_write().
     43   for (int fd = kFdForAndroidLogging + 1; fd < getdtablesize(); ++fd) {
     44     if (fds_to_keep_open.find(fd) == fds_to_keep_open.end()) {
     45       HANDLE_EINTR(close(fd));
     46     }
     47   }
     48   for (FileHandleMappingVector::const_iterator it = fds_to_remap.begin();
     49        it != fds_to_remap.end(); ++it) {
     50     int old_fd = it->first;
     51     int new_fd = it->second;
     52     if (dup2(old_fd, new_fd) < 0) {
     53       PLOG(FATAL) << "dup2";
     54     }
     55     HANDLE_EINTR(close(old_fd));
     56   }
     57   _exit(multi_process_function_list::InvokeChildProcessTest(procname));
     58   return 0;
     59 }
     60 
     61 }  // namespace base
     62