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