launch_posix.cc revision 9ab5563a3196760eb381d102cbb2bc0f7abc6a50
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
59ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/process/launch.h"
69ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <dirent.h>
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <errno.h>
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <fcntl.h>
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <signal.h>
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <stdlib.h>
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/resource.h>
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/time.h>
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/types.h>
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/wait.h>
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <unistd.h>
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <iterator>
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <limits>
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <set>
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/allocator/type_profiler_control.h"
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h"
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/compiler_specific.h"
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/debug/debugger.h"
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/debug/stack_trace.h"
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/file_util.h"
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/files/dir_reader_posix.h"
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/logging.h"
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h"
312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/posix/eintr_wrapper.h"
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/process_util.h"
33868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h"
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/synchronization/waitable_event.h"
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/third_party/dynamic_annotations/dynamic_annotations.h"
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/platform_thread.h"
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/threading/thread_restrictions.h"
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_CHROMEOS)
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/ioctl.h>
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_FREEBSD)
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/event.h>
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/ucontext.h>
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_MACOSX)
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <crt_externs.h>
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <sys/event.h>
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern char** environ;
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace base {
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace {
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Get the process's "environment" (i.e. the thing that setenv/getenv
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// work with).
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)char** GetEnvironment() {
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_MACOSX)
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return *_NSGetEnviron();
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return environ;
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Set the process's "environment" (i.e. the thing that setenv/getenv
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// work with).
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SetEnvironment(char** env) {
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_MACOSX)
735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  *_NSGetEnviron() = env;
745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else
755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  environ = env;
765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if !defined(OS_LINUX) || \
802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    (!defined(__i386__) && !defined(__x86_64__) && !defined(__arm__))
815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ResetChildSignalHandlersToDefaults() {
825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // The previous signal handlers are likely to be meaningless in the child's
835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // context so we reset them to the defaults for now. http://crbug.com/44953
845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // These signal handlers are set up at least in browser_main_posix.cc:
855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // BrowserMainPartsPosix::PreEarlyInitialization and stack_trace_posix.cc:
865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // EnableInProcessStackDumping.
875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGHUP, SIG_DFL);
885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGINT, SIG_DFL);
895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGILL, SIG_DFL);
905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGABRT, SIG_DFL);
915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGFPE, SIG_DFL);
925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGBUS, SIG_DFL);
935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGSEGV, SIG_DFL);
945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGSYS, SIG_DFL);
955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  signal(SIGTERM, SIG_DFL);
965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#else
992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// TODO(jln): remove the Linux special case once kernels are fixed.
1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Internally the kernel makes sigset_t an array of long large enough to have
1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// one bit per signal.
1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)typedef uint64_t kernel_sigset_t;
1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// This is what struct sigaction looks like to the kernel at least on X86 and
1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// ARM. MIPS, for instance, is very different.
1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)struct kernel_sigaction {
1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  void* k_sa_handler;  // For this usage it only needs to be a generic pointer.
1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  unsigned long k_sa_flags;
1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  void* k_sa_restorer;  // For this usage it only needs to be a generic pointer.
1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  kernel_sigset_t k_sa_mask;
1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)};
1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// glibc's sigaction() will prevent access to sa_restorer, so we need to roll
1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// our own.
1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)int sys_rt_sigaction(int sig, const struct kernel_sigaction* act,
1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                     struct kernel_sigaction* oact) {
1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  return syscall(SYS_rt_sigaction, sig, act, oact, sizeof(kernel_sigset_t));
1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// This function is intended to be used in between fork() and execve() and will
1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// reset all signal handlers to the default.
1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// The motivation for going through all of them is that sa_restorer can leak
1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// from parents and help defeat ASLR on buggy kernels.  We reset it to NULL.
1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// See crbug.com/177956.
1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ResetChildSignalHandlersToDefaults(void) {
1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  for (int signum = 1; ; ++signum) {
1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    struct kernel_sigaction act = {0};
1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    int sigaction_get_ret = sys_rt_sigaction(signum, NULL, &act);
1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (sigaction_get_ret && errno == EINVAL) {
1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if !defined(NDEBUG)
1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // Linux supports 32 real-time signals from 33 to 64.
1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // If the number of signals in the Linux kernel changes, someone should
1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      // look at this code.
1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      const int kNumberOfSignals = 64;
1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      RAW_CHECK(signum == kNumberOfSignals + 1);
1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif  // !defined(NDEBUG)
1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      break;
1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    }
1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // All other failures are fatal.
1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (sigaction_get_ret) {
1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      RAW_LOG(FATAL, "sigaction (get) failed.");
1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    }
1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // The kernel won't allow to re-set SIGKILL or SIGSTOP.
1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (signum != SIGSTOP && signum != SIGKILL) {
1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      act.k_sa_handler = reinterpret_cast<void*>(SIG_DFL);
1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      act.k_sa_restorer = NULL;
1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      if (sys_rt_sigaction(signum, &act, NULL)) {
1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        RAW_LOG(FATAL, "sigaction (set) failed.");
1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      }
1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    }
1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#if !defined(NDEBUG)
1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    // Now ask the kernel again and check that no restorer will leak.
1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    if (sys_rt_sigaction(signum, NULL, &act) || act.k_sa_restorer) {
1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)      RAW_LOG(FATAL, "Cound not fix sa_restorer.");
1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)    }
1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif  // !defined(NDEBUG)
1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  }
1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}
1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#endif  // !defined(OS_LINUX) ||
1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)        // (!defined(__i386__) && !defined(__x86_64__) && !defined(__arm__))
1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)
1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // anonymous namespace
1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A class to handle auto-closing of DIR*'s.
1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ScopedDIRClose {
1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public:
1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  inline void operator()(DIR* x) const {
1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (x) {
1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      closedir(x);
1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)};
1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)typedef scoped_ptr_malloc<DIR, ScopedDIRClose> ScopedDIR;
1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_LINUX)
1797dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochstatic const char kFDDir[] = "/proc/self/fd";
1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_MACOSX)
1817dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochstatic const char kFDDir[] = "/dev/fd";
1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_SOLARIS)
1837dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochstatic const char kFDDir[] = "/dev/fd";
1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_FREEBSD)
1857dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochstatic const char kFDDir[] = "/dev/fd";
1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_OPENBSD)
1877dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochstatic const char kFDDir[] = "/dev/fd";
1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#elif defined(OS_ANDROID)
1897dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochstatic const char kFDDir[] = "/proc/self/fd";
1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
192b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)void CloseSuperfluousFds(const base::InjectiveMultimap& saved_mapping) {
193b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)  // DANGER: no calls to malloc are allowed from now on:
194b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)  // http://crbug.com/36678
195b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)
196b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)  // Get the maximum number of FDs possible.
197b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)  size_t max_fds = GetMaxFds();
1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
199b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)  DirReaderPosix fd_dir(kFDDir);
2005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (!fd_dir.IsValid()) {
2015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Fallback case: Try every possible fd.
202b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles)    for (size_t i = 0; i < max_fds; ++i) {
2035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const int fd = static_cast<int>(i);
2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (fd == STDIN_FILENO || fd == STDOUT_FILENO || fd == STDERR_FILENO)
2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        continue;
2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      InjectiveMultimap::const_iterator j;
2075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      for (j = saved_mapping.begin(); j != saved_mapping.end(); j++) {
2085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (fd == j->dest)
2095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          break;
2105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
2115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (j != saved_mapping.end())
2125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        continue;
2135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Since we're just trying to close anything we can find,
2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // ignore any error return values of close().
2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      ignore_result(HANDLE_EINTR(close(fd)));
2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return;
2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  const int dir_fd = fd_dir.fd();
2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for ( ; fd_dir.Next(); ) {
2245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Skip . and .. entries.
2255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (fd_dir.name()[0] == '.')
2265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      continue;
2275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    char *endptr;
2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    errno = 0;
2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const long int fd = strtol(fd_dir.name(), &endptr, 10);
2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (fd_dir.name()[0] == 0 || *endptr || fd < 0 || errno)
2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      continue;
2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (fd == STDIN_FILENO || fd == STDOUT_FILENO || fd == STDERR_FILENO)
2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      continue;
2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    InjectiveMultimap::const_iterator i;
2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    for (i = saved_mapping.begin(); i != saved_mapping.end(); i++) {
2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (fd == i->dest)
2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        break;
2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (i != saved_mapping.end())
2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      continue;
2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (fd == dir_fd)
2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      continue;
2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // When running under Valgrind, Valgrind opens several FDs for its
2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // own use and will complain if we try to close them.  All of
2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // these FDs are >= |max_fds|, so we can check against that here
2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // before closing.  See https://bugs.kde.org/show_bug.cgi?id=191758
2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (fd < static_cast<int>(max_fds)) {
2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      int ret = HANDLE_EINTR(close(fd));
2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DPCHECK(ret == 0);
2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)char** AlterEnvironment(const EnvironmentVector& changes,
2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                        const char* const* const env) {
2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned count = 0;
2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned size = 0;
2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // First assume that all of the current environment will be included.
2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (unsigned i = 0; env[i]; i++) {
2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const char *const pair = env[i];
2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    count++;
2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    size += strlen(pair) + 1 /* terminating NUL */;
2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (EnvironmentVector::const_iterator j = changes.begin();
2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       j != changes.end();
2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       ++j) {
2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool found = false;
2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const char *pair;
2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    for (unsigned i = 0; env[i]; i++) {
2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      pair = env[i];
2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const char *const equals = strchr(pair, '=');
2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (!equals)
2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        continue;
2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const unsigned keylen = equals - pair;
2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (keylen == j->first.size() &&
2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          memcmp(pair, j->first.data(), keylen) == 0) {
2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        found = true;
2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        break;
2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // if found, we'll either be deleting or replacing this element.
2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (found) {
2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      count--;
2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      size -= strlen(pair) + 1;
2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (j->second.size())
2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        found = false;
2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // if !found, then we have a new element to add.
2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!found && !j->second.empty()) {
2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      count++;
2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      size += j->first.size() + 1 /* '=' */ + j->second.size() + 1 /* NUL */;
2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  count++;  // for the final NULL
3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  uint8_t *buffer = new uint8_t[sizeof(char*) * count + size];
3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  char **const ret = reinterpret_cast<char**>(buffer);
3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  unsigned k = 0;
3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  char *scratch = reinterpret_cast<char*>(buffer + sizeof(char*) * count);
3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (unsigned i = 0; env[i]; i++) {
3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const char *const pair = env[i];
3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const char *const equals = strchr(pair, '=');
3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!equals) {
3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const unsigned len = strlen(pair);
3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      ret[k++] = scratch;
3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      memcpy(scratch, pair, len + 1);
3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      scratch += len + 1;
3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      continue;
3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const unsigned keylen = equals - pair;
3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool handled = false;
3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    for (EnvironmentVector::const_iterator
3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)         j = changes.begin(); j != changes.end(); j++) {
3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (j->first.size() == keylen &&
3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          memcmp(j->first.data(), pair, keylen) == 0) {
3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (!j->second.empty()) {
3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          ret[k++] = scratch;
3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          memcpy(scratch, pair, keylen + 1);
3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          scratch += keylen + 1;
3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          memcpy(scratch, j->second.c_str(), j->second.size() + 1);
3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          scratch += j->second.size() + 1;
3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        }
3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        handled = true;
3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        break;
3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!handled) {
3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const unsigned len = strlen(pair);
3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      ret[k++] = scratch;
3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      memcpy(scratch, pair, len + 1);
3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      scratch += len + 1;
3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Now handle new elements
3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  for (EnvironmentVector::const_iterator
3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)       j = changes.begin(); j != changes.end(); j++) {
3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (j->second.empty())
3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      continue;
3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool found = false;
3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    for (unsigned i = 0; env[i]; i++) {
3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const char *const pair = env[i];
3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const char *const equals = strchr(pair, '=');
3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (!equals)
3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        continue;
3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      const unsigned keylen = equals - pair;
3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (keylen == j->first.size() &&
3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          memcmp(pair, j->first.data(), keylen) == 0) {
3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        found = true;
3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        break;
3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!found) {
3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      ret[k++] = scratch;
3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      memcpy(scratch, j->first.data(), j->first.size());
3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      scratch += j->first.size();
3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      *scratch++ = '=';
3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      memcpy(scratch, j->second.c_str(), j->second.size() + 1);
3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      scratch += j->second.size() + 1;
3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)     }
3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ret[k] = NULL;
3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return ret;
3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool LaunchProcess(const std::vector<std::string>& argv,
3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                   const LaunchOptions& options,
3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                   ProcessHandle* process_handle) {
3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  size_t fd_shuffle_size = 0;
3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (options.fds_to_remap) {
3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    fd_shuffle_size = options.fds_to_remap->size();
3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  InjectiveMultimap fd_shuffle1;
3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  InjectiveMultimap fd_shuffle2;
3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fd_shuffle1.reserve(fd_shuffle_size);
3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fd_shuffle2.reserve(fd_shuffle_size);
3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  scoped_ptr<char*[]> argv_cstr(new char*[argv.size() + 1]);
3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  scoped_ptr<char*[]> new_environ;
3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (options.environ)
3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    new_environ.reset(AlterEnvironment(*options.environ, GetEnvironment()));
3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  pid_t pid;
3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_LINUX)
3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (options.clone_flags) {
3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    pid = syscall(__NR_clone, options.clone_flags, 0, 0, 0);
4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else
4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  {
4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    pid = fork();
4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (pid < 0) {
4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    DPLOG(ERROR) << "fork";
4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return false;
4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else if (pid == 0) {
4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Child process
4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // DANGER: fork() rule: in the child, if you don't end up doing exec*(),
4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // you call _exit() instead of exit(). This is because _exit() does not
4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // call any previously-registered (in the parent) exit handlers, which
4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // might do things like block waiting for threads that don't even exist
4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // in the child.
4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // If a child process uses the readline library, the process block forever.
4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // In BSD like OSes including OS X it is safe to assign /dev/null as stdin.
4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // See http://crbug.com/56596.
4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int null_fd = HANDLE_EINTR(open("/dev/null", O_RDONLY));
4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (null_fd < 0) {
4235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      RAW_LOG(ERROR, "Failed to open /dev/null");
4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      _exit(127);
4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    file_util::ScopedFD null_fd_closer(&null_fd);
4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int new_fd = HANDLE_EINTR(dup2(null_fd, STDIN_FILENO));
4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (new_fd != STDIN_FILENO) {
4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      RAW_LOG(ERROR, "Failed to dup /dev/null for stdin");
4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      _exit(127);
4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (options.new_process_group) {
4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Instead of inheriting the process group ID of the parent, the child
4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // starts off a new process group with pgid equal to its process ID.
4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (setpgid(0, 0) < 0) {
4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        RAW_LOG(ERROR, "setpgid failed");
4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        _exit(127);
4405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Stop type-profiler.
4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // The profiler should be stopped between fork and exec since it inserts
4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // locks at new/delete expressions.  See http://crbug.com/36678.
4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    base::type_profiler::Controller::Stop();
4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (options.maximize_rlimits) {
4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Some resource limits need to be maximal in this child.
4505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      std::set<int>::const_iterator resource;
4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      for (resource = options.maximize_rlimits->begin();
4525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           resource != options.maximize_rlimits->end();
4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           ++resource) {
4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        struct rlimit limit;
4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (getrlimit(*resource, &limit) < 0) {
4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          RAW_LOG(WARNING, "getrlimit failed");
4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        } else if (limit.rlim_cur < limit.rlim_max) {
4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          limit.rlim_cur = limit.rlim_max;
4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          if (setrlimit(*resource, &limit) < 0) {
4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)            RAW_LOG(WARNING, "setrlimit failed");
4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          }
4625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        }
4635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
4645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_MACOSX)
4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    RestoreDefaultExceptionHandler();
4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif  // defined(OS_MACOSX)
4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    ResetChildSignalHandlersToDefaults();
4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if 0
4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // When debugging it can be helpful to check that we really aren't making
4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // any hidden calls to malloc.
4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    void *malloc_thunk =
4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        reinterpret_cast<void*>(reinterpret_cast<intptr_t>(malloc) & ~4095);
4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    mprotect(malloc_thunk, 4096, PROT_READ | PROT_WRITE | PROT_EXEC);
4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    memset(reinterpret_cast<void*>(malloc), 0xff, 8);
4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif  // 0
4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // DANGER: no calls to malloc are allowed from now on:
4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // http://crbug.com/36678
4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_CHROMEOS)
4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (options.ctrl_terminal_fd >= 0) {
4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // Set process' controlling terminal.
4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      if (HANDLE_EINTR(setsid()) != -1) {
4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (HANDLE_EINTR(
4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                ioctl(options.ctrl_terminal_fd, TIOCSCTTY, NULL)) == -1) {
4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          RAW_LOG(WARNING, "ioctl(TIOCSCTTY), ctrl terminal not set");
4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        }
4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      } else {
4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        RAW_LOG(WARNING, "setsid failed, ctrl terminal not set");
4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif  // defined(OS_CHROMEOS)
4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (options.fds_to_remap) {
4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      for (FileHandleMappingVector::const_iterator
5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)               it = options.fds_to_remap->begin();
5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)           it != options.fds_to_remap->end(); ++it) {
5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        fd_shuffle1.push_back(InjectionArc(it->first, it->second, false));
5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        fd_shuffle2.push_back(InjectionArc(it->first, it->second, false));
5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (options.environ)
5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      SetEnvironment(new_environ.get());
5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // fd_shuffle1 is mutated by this call because it cannot malloc.
5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (!ShuffleFileDescriptors(&fd_shuffle1))
5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      _exit(127);
5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    CloseSuperfluousFds(fd_shuffle2);
5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    for (size_t i = 0; i < argv.size(); i++)
5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      argv_cstr[i] = const_cast<char*>(argv[i].c_str());
5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    argv_cstr[argv.size()] = NULL;
5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    execvp(argv_cstr[0], argv_cstr.get());
5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    RAW_LOG(ERROR, "LaunchProcess: failed to execvp:");
5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    RAW_LOG(ERROR, argv_cstr[0]);
5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    _exit(127);
5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  } else {
5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // Parent process
5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (options.wait) {
5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // While this isn't strictly disk IO, waiting for another process to
5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      // finish is the sort of thing ThreadRestrictions is trying to prevent.
5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      base::ThreadRestrictions::AssertIOAllowed();
5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      pid_t ret = HANDLE_EINTR(waitpid(pid, 0, 0));
5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      DPCHECK(ret > 0);
5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (process_handle)
5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      *process_handle = pid;
5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return true;
5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool LaunchProcess(const CommandLine& cmdline,
5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                   const LaunchOptions& options,
5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                   ProcessHandle* process_handle) {
5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return LaunchProcess(cmdline.argv(), options, process_handle);
5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RaiseProcessToHighPriority() {
5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // On POSIX, we don't actually do anything here.  We could try to nice() or
5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // setpriority() or sched_getscheduler, but these all require extra rights.
5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Return value used by GetAppOutputInternal to encapsulate the various exit
5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// scenarios from the function.
5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)enum GetAppOutputInternalResult {
5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  EXECUTE_FAILURE,
5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  EXECUTE_SUCCESS,
5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  GOT_MAX_OUTPUT,
5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)};
5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Executes the application specified by |argv| and wait for it to exit. Stores
5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the output (stdout) in |output|. If |do_search_path| is set, it searches the
5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// path for the application; in that case, |envp| must be null, and it will use
5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the current environment. If |do_search_path| is false, |argv[0]| should fully
5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// specify the path of the application, and |envp| will be used as the
5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// environment. Redirects stderr to /dev/null.
5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// If we successfully start the application and get all requested output, we
5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// return GOT_MAX_OUTPUT, or if there is a problem starting or exiting
5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the application we return RUN_FAILURE. Otherwise we return EXECUTE_SUCCESS.
5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// The GOT_MAX_OUTPUT return value exists so a caller that asks for limited
5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// output can treat this as a success, despite having an exit code of SIG_PIPE
5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// due to us closing the output pipe.
5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// In the case of EXECUTE_SUCCESS, the application exit code will be returned
5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// in |*exit_code|, which should be checked to determine if the application
5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// ran successfully.
5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static GetAppOutputInternalResult GetAppOutputInternal(
5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    const std::vector<std::string>& argv,
5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    char* const envp[],
5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    std::string* output,
5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    size_t max_output,
5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    bool do_search_path,
5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    int* exit_code) {
5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Doing a blocking wait for another command to finish counts as IO.
5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  base::ThreadRestrictions::AssertIOAllowed();
5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // exit_code must be supplied so calling function can determine success.
5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(exit_code);
5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  *exit_code = EXIT_FAILURE;
5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int pipe_fd[2];
5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  pid_t pid;
5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  InjectiveMultimap fd_shuffle1, fd_shuffle2;
5922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)  scoped_ptr<char*[]> argv_cstr(new char*[argv.size() + 1]);
5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fd_shuffle1.reserve(3);
5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  fd_shuffle2.reserve(3);
5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Either |do_search_path| should be false or |envp| should be null, but not
5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // both.
5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  DCHECK(!do_search_path ^ !envp);
6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  if (pipe(pipe_fd) < 0)
6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return EXECUTE_FAILURE;
6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  switch (pid = fork()) {
6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case -1:  // error
6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      close(pipe_fd[0]);
6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      close(pipe_fd[1]);
6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      return EXECUTE_FAILURE;
6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    case 0:  // child
6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      {
6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#if defined(OS_MACOSX)
6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        RestoreDefaultExceptionHandler();
6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // DANGER: no calls to malloc are allowed from now on:
6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // http://crbug.com/36678
6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Obscure fork() rule: in the child, if you don't end up doing exec*(),
6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // you call _exit() instead of exit(). This is because _exit() does not
6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // call any previously-registered (in the parent) exit handlers, which
6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // might do things like block waiting for threads that don't even exist
6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // in the child.
6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        int dev_null = open("/dev/null", O_WRONLY);
6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (dev_null < 0)
6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          _exit(127);
6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Stop type-profiler.
6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // The profiler should be stopped between fork and exec since it inserts
6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // locks at new/delete expressions.  See http://crbug.com/36678.
6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        base::type_profiler::Controller::Stop();
6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        fd_shuffle1.push_back(InjectionArc(pipe_fd[1], STDOUT_FILENO, true));
6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        fd_shuffle1.push_back(InjectionArc(dev_null, STDERR_FILENO, true));
6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        fd_shuffle1.push_back(InjectionArc(dev_null, STDIN_FILENO, true));
6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Adding another element here? Remeber to increase the argument to
6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // reserve(), above.
6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        std::copy(fd_shuffle1.begin(), fd_shuffle1.end(),
6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                  std::back_inserter(fd_shuffle2));
6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (!ShuffleFileDescriptors(&fd_shuffle1))
6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          _exit(127);
6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        CloseSuperfluousFds(fd_shuffle2);
6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        for (size_t i = 0; i < argv.size(); i++)
6465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          argv_cstr[i] = const_cast<char*>(argv[i].c_str());
6475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        argv_cstr[argv.size()] = NULL;
6485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (do_search_path)
6495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          execvp(argv_cstr[0], argv_cstr.get());
6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        else
6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          execve(argv_cstr[0], argv_cstr.get(), envp);
6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        _exit(127);
6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    default:  // parent
6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      {
6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Close our writing end of pipe now. Otherwise later read would not
6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // be able to detect end of child's output (in theory we could still
6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // write to the pipe).
6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        close(pipe_fd[1]);
6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        output->clear();
6625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        char buffer[256];
6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        size_t output_buf_left = max_output;
6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        ssize_t bytes_read = 1;  // A lie to properly handle |max_output == 0|
6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                 // case in the logic below.
6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        while (output_buf_left > 0) {
6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          bytes_read = HANDLE_EINTR(read(pipe_fd[0], buffer,
6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                    std::min(output_buf_left, sizeof(buffer))));
6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          if (bytes_read <= 0)
6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)            break;
6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          output->append(buffer, bytes_read);
6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          output_buf_left -= static_cast<size_t>(bytes_read);
6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        }
6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        close(pipe_fd[0]);
6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // Always wait for exit code (even if we know we'll declare
6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // GOT_MAX_OUTPUT).
6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        bool success = WaitForExitCode(pid, exit_code);
6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // If we stopped because we read as much as we wanted, we return
6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        // GOT_MAX_OUTPUT (because the child may exit due to |SIGPIPE|).
6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        if (!output_buf_left && bytes_read > 0)
6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          return GOT_MAX_OUTPUT;
6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        else if (success)
6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)          return EXECUTE_SUCCESS;
6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        return EXECUTE_FAILURE;
6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      }
6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  }
6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GetAppOutput(const CommandLine& cl, std::string* output) {
6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return GetAppOutput(cl.argv(), output);
6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GetAppOutput(const std::vector<std::string>& argv, std::string* output) {
6975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Run |execve()| with the current environment and store "unlimited" data.
6985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int exit_code;
6995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  GetAppOutputInternalResult result = GetAppOutputInternal(
7005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      argv, NULL, output, std::numeric_limits<std::size_t>::max(), true,
7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      &exit_code);
7025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return result == EXECUTE_SUCCESS && exit_code == EXIT_SUCCESS;
7035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// TODO(viettrungluu): Conceivably, we should have a timeout as well, so we
7065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// don't hang if what we're calling hangs.
7075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GetAppOutputRestricted(const CommandLine& cl,
7085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                            std::string* output, size_t max_output) {
7095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Run |execve()| with the empty environment.
7105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  char* const empty_environ = NULL;
7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int exit_code;
7125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  GetAppOutputInternalResult result = GetAppOutputInternal(
7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      cl.argv(), &empty_environ, output, max_output, false, &exit_code);
7145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return result == GOT_MAX_OUTPUT || (result == EXECUTE_SUCCESS &&
7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                                      exit_code == EXIT_SUCCESS);
7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool GetAppOutputWithExitCode(const CommandLine& cl,
7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                              std::string* output,
7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)                              int* exit_code) {
7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Run |execve()| with the current environment and store "unlimited" data.
7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  GetAppOutputInternalResult result = GetAppOutputInternal(
7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      cl.argv(), NULL, output, std::numeric_limits<std::size_t>::max(), true,
7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)      exit_code);
7255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  return result == EXECUTE_SUCCESS;
7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
7275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
7285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace base
729