2015-02-06 11:42:57 -06:00
|
|
|
// Copyright 2015 The Rust Project Developers. See the COPYRIGHT
|
|
|
|
// file at the top-level directory of this distribution and at
|
|
|
|
// http://rust-lang.org/COPYRIGHT.
|
|
|
|
//
|
|
|
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
|
|
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
|
|
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
|
|
|
// option. This file may not be copied, modified, or distributed
|
|
|
|
// except according to those terms.
|
|
|
|
|
|
|
|
//! Working with processes.
|
|
|
|
|
2015-02-27 18:23:21 -06:00
|
|
|
#![stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
#![allow(non_upper_case_globals)]
|
|
|
|
|
|
|
|
use prelude::v1::*;
|
|
|
|
use io::prelude::*;
|
|
|
|
|
|
|
|
use ffi::AsOsStr;
|
|
|
|
use fmt;
|
|
|
|
use io::{self, Error, ErrorKind};
|
|
|
|
use libc;
|
2015-03-18 11:14:54 -05:00
|
|
|
use path;
|
2015-02-06 11:42:57 -06:00
|
|
|
use sync::mpsc::{channel, Receiver};
|
|
|
|
use sys::pipe2::{self, AnonPipe};
|
|
|
|
use sys::process2::Process as ProcessImp;
|
|
|
|
use sys::process2::Command as CommandImp;
|
|
|
|
use sys::process2::ExitStatus as ExitStatusImp;
|
|
|
|
use sys_common::{AsInner, AsInnerMut};
|
2015-02-17 17:10:25 -06:00
|
|
|
use thread;
|
2015-02-06 11:42:57 -06:00
|
|
|
|
|
|
|
/// Representation of a running or exited child process.
|
|
|
|
///
|
|
|
|
/// This structure is used to represent and manage child processes. A child
|
|
|
|
/// process is created via the `Command` struct, which configures the spawning
|
|
|
|
/// process and can itself be constructed using a builder-style interface.
|
|
|
|
///
|
2015-03-11 20:11:40 -05:00
|
|
|
/// # Examples
|
2015-02-06 11:42:57 -06:00
|
|
|
///
|
|
|
|
/// ```should_fail
|
|
|
|
/// # #![feature(process)]
|
|
|
|
///
|
|
|
|
/// use std::process::Command;
|
|
|
|
///
|
|
|
|
/// let output = Command::new("/bin/cat").arg("file.txt").output().unwrap_or_else(|e| {
|
|
|
|
/// panic!("failed to execute child: {}", e)
|
|
|
|
/// });
|
|
|
|
/// let contents = output.stdout;
|
|
|
|
/// assert!(output.status.success());
|
|
|
|
/// ```
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub struct Child {
|
|
|
|
handle: ProcessImp,
|
|
|
|
|
|
|
|
/// None until wait() or wait_with_output() is called.
|
|
|
|
status: Option<ExitStatusImp>,
|
|
|
|
|
|
|
|
/// The handle for writing to the child's stdin, if it has been captured
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub stdin: Option<ChildStdin>,
|
|
|
|
|
|
|
|
/// The handle for reading from the child's stdout, if it has been captured
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub stdout: Option<ChildStdout>,
|
|
|
|
|
|
|
|
/// The handle for reading from the child's stderr, if it has been captured
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub stderr: Option<ChildStderr>,
|
|
|
|
}
|
|
|
|
|
|
|
|
/// A handle to a child procesess's stdin
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub struct ChildStdin {
|
|
|
|
inner: AnonPipe
|
|
|
|
}
|
|
|
|
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
impl Write for ChildStdin {
|
|
|
|
fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
|
|
|
|
self.inner.write(buf)
|
|
|
|
}
|
|
|
|
|
|
|
|
fn flush(&mut self) -> io::Result<()> {
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// A handle to a child procesess's stdout
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub struct ChildStdout {
|
|
|
|
inner: AnonPipe
|
|
|
|
}
|
|
|
|
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
impl Read for ChildStdout {
|
|
|
|
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
|
|
|
|
self.inner.read(buf)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// A handle to a child procesess's stderr
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub struct ChildStderr {
|
|
|
|
inner: AnonPipe
|
|
|
|
}
|
|
|
|
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
impl Read for ChildStderr {
|
|
|
|
fn read(&mut self, buf: &mut [u8]) -> io::Result<usize> {
|
|
|
|
self.inner.read(buf)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// The `Command` type acts as a process builder, providing fine-grained control
|
|
|
|
/// over how a new process should be spawned. A default configuration can be
|
|
|
|
/// generated using `Command::new(program)`, where `program` gives a path to the
|
|
|
|
/// program to be executed. Additional builder methods allow the configuration
|
|
|
|
/// to be changed (for example, by adding arguments) prior to spawning:
|
|
|
|
///
|
|
|
|
/// ```
|
|
|
|
/// use std::process::Command;
|
|
|
|
///
|
|
|
|
/// let output = Command::new("sh").arg("-c").arg("echo hello").output().unwrap_or_else(|e| {
|
|
|
|
/// panic!("failed to execute process: {}", e)
|
|
|
|
/// });
|
|
|
|
/// let hello = output.stdout;
|
|
|
|
/// ```
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub struct Command {
|
|
|
|
inner: CommandImp,
|
|
|
|
|
|
|
|
// Details explained in the builder methods
|
|
|
|
stdin: Option<StdioImp>,
|
|
|
|
stdout: Option<StdioImp>,
|
|
|
|
stderr: Option<StdioImp>,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Command {
|
|
|
|
/// Constructs a new `Command` for launching the program at
|
|
|
|
/// path `program`, with the following default configuration:
|
|
|
|
///
|
|
|
|
/// * No arguments to the program
|
|
|
|
/// * Inherit the current process's environment
|
|
|
|
/// * Inherit the current process's working directory
|
|
|
|
/// * Inherit stdin/stdout/stderr for `run` or `status`, but create pipes for `output`
|
|
|
|
///
|
|
|
|
/// Builder methods are provided to change these defaults and
|
|
|
|
/// otherwise configure the process.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-03-12 14:59:53 -05:00
|
|
|
pub fn new<S: AsOsStr>(program: S) -> Command {
|
2015-02-06 11:42:57 -06:00
|
|
|
Command {
|
|
|
|
inner: CommandImp::new(program.as_os_str()),
|
|
|
|
stdin: None,
|
|
|
|
stdout: None,
|
|
|
|
stderr: None,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Add an argument to pass to the program.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-03-12 14:59:53 -05:00
|
|
|
pub fn arg<S: AsOsStr>(&mut self, arg: S) -> &mut Command {
|
2015-02-06 11:42:57 -06:00
|
|
|
self.inner.arg(arg.as_os_str());
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Add multiple arguments to pass to the program.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn args<S: AsOsStr>(&mut self, args: &[S]) -> &mut Command {
|
|
|
|
self.inner.args(args.iter().map(AsOsStr::as_os_str));
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Inserts or updates an environment variable mapping.
|
|
|
|
///
|
|
|
|
/// Note that environment variable names are case-insensitive (but case-preserving) on Windows,
|
|
|
|
/// and case-sensitive on all other platforms.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-03-12 14:59:53 -05:00
|
|
|
pub fn env<K, V>(&mut self, key: K, val: V) -> &mut Command
|
2015-02-27 18:23:21 -06:00
|
|
|
where K: AsOsStr, V: AsOsStr
|
2015-02-06 11:42:57 -06:00
|
|
|
{
|
|
|
|
self.inner.env(key.as_os_str(), val.as_os_str());
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Removes an environment variable mapping.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-03-12 14:59:53 -05:00
|
|
|
pub fn env_remove<K: AsOsStr>(&mut self, key: K) -> &mut Command {
|
2015-02-06 11:42:57 -06:00
|
|
|
self.inner.env_remove(key.as_os_str());
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Clears the entire environment map for the child process.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn env_clear(&mut self) -> &mut Command {
|
|
|
|
self.inner.env_clear();
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Set the working directory for the child process.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-03-18 11:14:54 -05:00
|
|
|
pub fn current_dir<P: AsRef<path::Path>>(&mut self, dir: P) -> &mut Command {
|
|
|
|
self.inner.cwd(dir.as_ref().as_os_str());
|
2015-02-06 11:42:57 -06:00
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Configuration for the child process's stdin handle (file descriptor 0).
|
|
|
|
/// Defaults to `CreatePipe(true, false)` so the input can be written to.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn stdin(&mut self, cfg: Stdio) -> &mut Command {
|
|
|
|
self.stdin = Some(cfg.0);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Configuration for the child process's stdout handle (file descriptor 1).
|
|
|
|
/// Defaults to `CreatePipe(false, true)` so the output can be collected.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn stdout(&mut self, cfg: Stdio) -> &mut Command {
|
|
|
|
self.stdout = Some(cfg.0);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Configuration for the child process's stderr handle (file descriptor 2).
|
|
|
|
/// Defaults to `CreatePipe(false, true)` so the output can be collected.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn stderr(&mut self, cfg: Stdio) -> &mut Command {
|
|
|
|
self.stderr = Some(cfg.0);
|
|
|
|
self
|
|
|
|
}
|
|
|
|
|
|
|
|
fn spawn_inner(&self, default_io: StdioImp) -> io::Result<Child> {
|
|
|
|
let (their_stdin, our_stdin) = try!(
|
|
|
|
setup_io(self.stdin.as_ref().unwrap_or(&default_io), 0, true)
|
|
|
|
);
|
|
|
|
let (their_stdout, our_stdout) = try!(
|
|
|
|
setup_io(self.stdout.as_ref().unwrap_or(&default_io), 1, false)
|
|
|
|
);
|
|
|
|
let (their_stderr, our_stderr) = try!(
|
|
|
|
setup_io(self.stderr.as_ref().unwrap_or(&default_io), 2, false)
|
|
|
|
);
|
|
|
|
|
|
|
|
match ProcessImp::spawn(&self.inner, their_stdin, their_stdout, their_stderr) {
|
|
|
|
Err(e) => Err(e),
|
|
|
|
Ok(handle) => Ok(Child {
|
|
|
|
handle: handle,
|
|
|
|
status: None,
|
|
|
|
stdin: our_stdin.map(|fd| ChildStdin { inner: fd }),
|
|
|
|
stdout: our_stdout.map(|fd| ChildStdout { inner: fd }),
|
|
|
|
stderr: our_stderr.map(|fd| ChildStderr { inner: fd }),
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Executes the command as a child process, returning a handle to it.
|
|
|
|
///
|
|
|
|
/// By default, stdin, stdout and stderr are inherited by the parent.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn spawn(&mut self) -> io::Result<Child> {
|
|
|
|
self.spawn_inner(StdioImp::Inherit)
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Executes the command as a child process, waiting for it to finish and
|
|
|
|
/// collecting all of its output.
|
|
|
|
///
|
|
|
|
/// By default, stdin, stdout and stderr are captured (and used to
|
|
|
|
/// provide the resulting output).
|
|
|
|
///
|
2015-03-08 11:07:58 -05:00
|
|
|
/// # Examples
|
2015-02-06 11:42:57 -06:00
|
|
|
///
|
|
|
|
/// ```
|
|
|
|
/// # #![feature(process)]
|
|
|
|
/// use std::process::Command;
|
|
|
|
///
|
|
|
|
/// let output = Command::new("cat").arg("foot.txt").output().unwrap_or_else(|e| {
|
|
|
|
/// panic!("failed to execute process: {}", e)
|
|
|
|
/// });
|
|
|
|
///
|
|
|
|
/// println!("status: {}", output.status);
|
2015-03-08 11:07:58 -05:00
|
|
|
/// println!("stdout: {}", String::from_utf8_lossy(&output.stdout));
|
|
|
|
/// println!("stderr: {}", String::from_utf8_lossy(&output.stderr));
|
2015-02-06 11:42:57 -06:00
|
|
|
/// ```
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn output(&mut self) -> io::Result<Output> {
|
2015-02-27 18:23:21 -06:00
|
|
|
self.spawn_inner(StdioImp::Piped).and_then(|p| p.wait_with_output())
|
2015-02-06 11:42:57 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Executes a command as a child process, waiting for it to finish and
|
|
|
|
/// collecting its exit status.
|
|
|
|
///
|
|
|
|
/// By default, stdin, stdout and stderr are inherited by the parent.
|
|
|
|
///
|
2015-03-11 20:11:40 -05:00
|
|
|
/// # Examples
|
2015-02-06 11:42:57 -06:00
|
|
|
///
|
|
|
|
/// ```
|
|
|
|
/// # #![feature(process)]
|
|
|
|
/// use std::process::Command;
|
|
|
|
///
|
|
|
|
/// let status = Command::new("ls").status().unwrap_or_else(|e| {
|
|
|
|
/// panic!("failed to execute process: {}", e)
|
|
|
|
/// });
|
|
|
|
///
|
|
|
|
/// println!("process exited with: {}", status);
|
|
|
|
/// ```
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn status(&mut self) -> io::Result<ExitStatus> {
|
|
|
|
self.spawn().and_then(|mut p| p.wait())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[stable(feature = "rust1", since = "1.0.0")]
|
|
|
|
impl fmt::Debug for Command {
|
|
|
|
/// Format the program and arguments of a Command for display. Any
|
|
|
|
/// non-utf8 data is lossily converted using the utf8 replacement
|
|
|
|
/// character.
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
|
|
try!(write!(f, "{:?}", self.inner.program));
|
|
|
|
for arg in &self.inner.args {
|
|
|
|
try!(write!(f, " {:?}", arg));
|
|
|
|
}
|
|
|
|
Ok(())
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl AsInner<CommandImp> for Command {
|
|
|
|
fn as_inner(&self) -> &CommandImp { &self.inner }
|
|
|
|
}
|
|
|
|
|
|
|
|
impl AsInnerMut<CommandImp> for Command {
|
|
|
|
fn as_inner_mut(&mut self) -> &mut CommandImp { &mut self.inner }
|
|
|
|
}
|
|
|
|
|
|
|
|
fn setup_io(io: &StdioImp, fd: libc::c_int, readable: bool)
|
|
|
|
-> io::Result<(Option<AnonPipe>, Option<AnonPipe>)>
|
|
|
|
{
|
|
|
|
use self::StdioImp::*;
|
|
|
|
Ok(match *io {
|
|
|
|
Null => {
|
|
|
|
(None, None)
|
|
|
|
}
|
|
|
|
Inherit => {
|
|
|
|
(Some(AnonPipe::from_fd(fd)), None)
|
|
|
|
}
|
2015-02-27 18:23:21 -06:00
|
|
|
Piped => {
|
2015-02-06 11:42:57 -06:00
|
|
|
let (reader, writer) = try!(unsafe { pipe2::anon_pipe() });
|
|
|
|
if readable {
|
|
|
|
(Some(reader), Some(writer))
|
|
|
|
} else {
|
|
|
|
(Some(writer), Some(reader))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
/// The output of a finished process.
|
|
|
|
#[derive(PartialEq, Eq, Clone)]
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub struct Output {
|
|
|
|
/// The status (exit code) of the process.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub status: ExitStatus,
|
|
|
|
/// The data that the process wrote to stdout.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub stdout: Vec<u8>,
|
|
|
|
/// The data that the process wrote to stderr.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub stderr: Vec<u8>,
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Describes what to do with a standard io stream for a child process.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub struct Stdio(StdioImp);
|
|
|
|
|
|
|
|
// The internal enum for stdio setup; see below for descriptions.
|
|
|
|
#[derive(Clone)]
|
|
|
|
enum StdioImp {
|
2015-02-27 18:23:21 -06:00
|
|
|
Piped,
|
2015-02-06 11:42:57 -06:00
|
|
|
Inherit,
|
|
|
|
Null,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Stdio {
|
|
|
|
/// A new pipe should be arranged to connect the parent and child processes.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[unstable(feature = "process_capture")]
|
|
|
|
#[deprecated(since = "1.0.0", reason = "renamed to `Stdio::piped`")]
|
|
|
|
pub fn capture() -> Stdio { Stdio::piped() }
|
|
|
|
|
|
|
|
/// A new pipe should be arranged to connect the parent and child processes.
|
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
|
|
|
pub fn piped() -> Stdio { Stdio(StdioImp::Piped) }
|
2015-02-06 11:42:57 -06:00
|
|
|
|
|
|
|
/// The child inherits from the corresponding parent descriptor.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-23 23:41:27 -06:00
|
|
|
pub fn inherit() -> Stdio { Stdio(StdioImp::Inherit) }
|
2015-02-06 11:42:57 -06:00
|
|
|
|
|
|
|
/// This stream will be ignored. This is the equivalent of attaching the
|
|
|
|
/// stream to `/dev/null`
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-23 23:41:27 -06:00
|
|
|
pub fn null() -> Stdio { Stdio(StdioImp::Null) }
|
2015-02-06 11:42:57 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
/// Describes the result of a process after it has terminated.
|
|
|
|
#[derive(PartialEq, Eq, Clone, Copy, Debug)]
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub struct ExitStatus(ExitStatusImp);
|
|
|
|
|
|
|
|
impl ExitStatus {
|
|
|
|
/// Was termination successful? Signal termination not considered a success,
|
|
|
|
/// and success is defined as a zero exit status.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn success(&self) -> bool {
|
|
|
|
self.0.success()
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Return the exit code of the process, if any.
|
|
|
|
///
|
|
|
|
/// On Unix, this will return `None` if the process was terminated
|
|
|
|
/// by a signal; `std::os::unix` provides an extension trait for
|
|
|
|
/// extracting the signal and other details from the `ExitStatus`.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn code(&self) -> Option<i32> {
|
|
|
|
self.0.code()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl AsInner<ExitStatusImp> for ExitStatus {
|
|
|
|
fn as_inner(&self) -> &ExitStatusImp { &self.0 }
|
|
|
|
}
|
|
|
|
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
impl fmt::Display for ExitStatus {
|
|
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
|
|
self.0.fmt(f)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl Child {
|
|
|
|
/// Forces the child to exit. This is equivalent to sending a
|
|
|
|
/// SIGKILL on unix platforms.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn kill(&mut self) -> io::Result<()> {
|
|
|
|
#[cfg(unix)] fn collect_status(p: &mut Child) {
|
|
|
|
// On Linux (and possibly other unices), a process that has exited will
|
|
|
|
// continue to accept signals because it is "defunct". The delivery of
|
|
|
|
// signals will only fail once the child has been reaped. For this
|
|
|
|
// reason, if the process hasn't exited yet, then we attempt to collect
|
|
|
|
// their status with WNOHANG.
|
|
|
|
if p.status.is_none() {
|
|
|
|
match p.handle.try_wait() {
|
|
|
|
Some(status) => { p.status = Some(status); }
|
|
|
|
None => {}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#[cfg(windows)] fn collect_status(_p: &mut Child) {}
|
|
|
|
|
|
|
|
collect_status(self);
|
|
|
|
|
|
|
|
// if the process has finished, and therefore had waitpid called,
|
|
|
|
// and we kill it, then on unix we might ending up killing a
|
|
|
|
// newer process that happens to have the same (re-used) id
|
|
|
|
if self.status.is_some() {
|
|
|
|
return Err(Error::new(
|
|
|
|
ErrorKind::InvalidInput,
|
|
|
|
"invalid argument: can't kill an exited process",
|
|
|
|
None
|
|
|
|
))
|
|
|
|
}
|
|
|
|
|
|
|
|
unsafe { self.handle.kill() }
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Wait for the child to exit completely, returning the status that it
|
|
|
|
/// exited with. This function will continue to have the same return value
|
|
|
|
/// after it has been called at least once.
|
|
|
|
///
|
|
|
|
/// The stdin handle to the child process, if any, will be closed
|
|
|
|
/// before waiting. This helps avoid deadlock: it ensures that the
|
|
|
|
/// child does not block waiting for input from the parent, while
|
|
|
|
/// the parent waits for the child to exit.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn wait(&mut self) -> io::Result<ExitStatus> {
|
|
|
|
drop(self.stdin.take());
|
|
|
|
match self.status {
|
|
|
|
Some(code) => Ok(ExitStatus(code)),
|
|
|
|
None => {
|
|
|
|
let status = try!(self.handle.wait());
|
|
|
|
self.status = Some(status);
|
|
|
|
Ok(ExitStatus(status))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// Simultaneously wait for the child to exit and collect all remaining
|
|
|
|
/// output on the stdout/stderr handles, returning a `Output`
|
|
|
|
/// instance.
|
|
|
|
///
|
|
|
|
/// The stdin handle to the child process, if any, will be closed
|
|
|
|
/// before waiting. This helps avoid deadlock: it ensures that the
|
|
|
|
/// child does not block waiting for input from the parent, while
|
|
|
|
/// the parent waits for the child to exit.
|
2015-02-27 18:23:21 -06:00
|
|
|
#[stable(feature = "process", since = "1.0.0")]
|
2015-02-06 11:42:57 -06:00
|
|
|
pub fn wait_with_output(mut self) -> io::Result<Output> {
|
|
|
|
drop(self.stdin.take());
|
2015-02-17 06:48:32 -06:00
|
|
|
fn read<T: Read + Send + 'static>(stream: Option<T>) -> Receiver<io::Result<Vec<u8>>> {
|
2015-02-06 11:42:57 -06:00
|
|
|
let (tx, rx) = channel();
|
|
|
|
match stream {
|
|
|
|
Some(stream) => {
|
2015-02-17 17:10:25 -06:00
|
|
|
thread::spawn(move || {
|
2015-02-06 11:42:57 -06:00
|
|
|
let mut stream = stream;
|
|
|
|
let mut ret = Vec::new();
|
|
|
|
let res = stream.read_to_end(&mut ret);
|
|
|
|
tx.send(res.map(|_| ret)).unwrap();
|
|
|
|
});
|
|
|
|
}
|
|
|
|
None => tx.send(Ok(Vec::new())).unwrap()
|
|
|
|
}
|
|
|
|
rx
|
|
|
|
}
|
|
|
|
let stdout = read(self.stdout.take());
|
|
|
|
let stderr = read(self.stderr.take());
|
|
|
|
let status = try!(self.wait());
|
|
|
|
|
|
|
|
Ok(Output {
|
|
|
|
status: status,
|
|
|
|
stdout: stdout.recv().unwrap().unwrap_or(Vec::new()),
|
|
|
|
stderr: stderr.recv().unwrap().unwrap_or(Vec::new()),
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(test)]
|
|
|
|
mod tests {
|
|
|
|
use io::ErrorKind;
|
|
|
|
use io::prelude::*;
|
2015-02-19 11:57:25 -06:00
|
|
|
use prelude::v1::{Ok, Err, drop, Some, Vec};
|
2015-02-06 11:42:57 -06:00
|
|
|
use prelude::v1::{String, Clone};
|
2015-03-17 15:33:26 -05:00
|
|
|
use prelude::v1::{Str, AsSlice, ToString};
|
|
|
|
use old_path::{self, GenericPath};
|
2015-02-06 11:42:57 -06:00
|
|
|
use old_io::fs::PathExtensions;
|
|
|
|
use rt::running_on_valgrind;
|
|
|
|
use str;
|
2015-02-19 11:57:25 -06:00
|
|
|
use super::{Command, Output, Stdio};
|
2015-02-06 11:42:57 -06:00
|
|
|
|
|
|
|
// FIXME(#10380) these tests should not all be ignored on android.
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn smoke() {
|
|
|
|
let p = Command::new("true").spawn();
|
|
|
|
assert!(p.is_ok());
|
|
|
|
let mut p = p.unwrap();
|
|
|
|
assert!(p.wait().unwrap().success());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn smoke_failure() {
|
|
|
|
match Command::new("if-this-is-a-binary-then-the-world-has-ended").spawn() {
|
|
|
|
Ok(..) => panic!(),
|
|
|
|
Err(..) => {}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn exit_reported_right() {
|
|
|
|
let p = Command::new("false").spawn();
|
|
|
|
assert!(p.is_ok());
|
|
|
|
let mut p = p.unwrap();
|
|
|
|
assert!(p.wait().unwrap().code() == Some(1));
|
|
|
|
drop(p.wait().clone());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(all(unix, not(target_os="android")))]
|
|
|
|
#[test]
|
|
|
|
fn signal_reported_right() {
|
std: Stabilize portions of `std::os::$platform`
This commit starts to organize the `std::os::$platform` modules and in the
process stabilizes some of the functionality contained within. The organization
of these modules will reflect the organization of the standard library itself
with extension traits for primitives in the same corresponding module.
The OS-specific modules will grow more functionality over time including
concrete types that are not extending functionality of other structures, and
these will either go into the closest module in `std::os::$platform` or they
will grow a new module in the hierarchy.
The following items are now stable:
* `os::{unix, windows}`
* `unix::ffi`
* `unix::ffi::OsStrExt`
* `unix::ffi::OsStrExt::{from_bytes, as_bytes, to_cstring}`
* `unix::ffi::OsString`
* `unix::ffi::OsStringExt::{from_vec, into_vec}`
* `unix::process`
* `unix::process::CommandExt`
* `unix::process::CommandExt::{uid, gid}`
* `unix::process::ExitStatusExt`
* `unix::process::ExitStatusExt::signal`
* `unix::prelude`
* `windows::ffi`
* `windows::ffi::OsStringExt`
* `windows::ffi::OsStringExt::from_wide`
* `windows::ffi::OsStrExt`
* `windows::ffi::OsStrExt::encode_wide`
* `windows::prelude`
The following items remain unstable:
* `unix::io`
* `unix::io::{Fd, AsRawFd}`
* `unix::fs::{PermissionsExt, OpenOptionsExt}`
* `windows::io`
* `windows::io::{Handle, AsRawHandle}`
* `windows::io::{Socket, AsRawSocket}`
* `windows::fs`
* `windows::fs::OpenOptionsExt`
Due to the reorgnization of the platform extension modules, this commit is a
breaking change. Most imports can be fixed by adding the relevant libstd module
in the `use` path (such as `ffi` or `fs`).
[breaking-change]
2015-03-13 19:12:38 -05:00
|
|
|
use os::unix::process::ExitStatusExt;
|
2015-02-06 11:42:57 -06:00
|
|
|
|
2015-02-16 06:53:46 -06:00
|
|
|
let p = Command::new("/bin/sh").arg("-c").arg("kill -9 $$").spawn();
|
2015-02-06 11:42:57 -06:00
|
|
|
assert!(p.is_ok());
|
|
|
|
let mut p = p.unwrap();
|
|
|
|
match p.wait().unwrap().signal() {
|
2015-02-16 06:53:46 -06:00
|
|
|
Some(9) => {},
|
|
|
|
result => panic!("not terminated by signal 9 (instead, {:?})", result),
|
2015-02-06 11:42:57 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn run_output(mut cmd: Command) -> String {
|
|
|
|
let p = cmd.spawn();
|
|
|
|
assert!(p.is_ok());
|
|
|
|
let mut p = p.unwrap();
|
|
|
|
assert!(p.stdout.is_some());
|
|
|
|
let mut ret = String::new();
|
|
|
|
p.stdout.as_mut().unwrap().read_to_string(&mut ret).unwrap();
|
|
|
|
assert!(p.wait().unwrap().success());
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn stdout_works() {
|
|
|
|
let mut cmd = Command::new("echo");
|
2015-02-27 18:23:21 -06:00
|
|
|
cmd.arg("foobar").stdout(Stdio::piped());
|
2015-02-06 11:42:57 -06:00
|
|
|
assert_eq!(run_output(cmd), "foobar\n");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(all(unix, not(target_os="android")))]
|
|
|
|
#[test]
|
|
|
|
fn set_current_dir_works() {
|
|
|
|
let mut cmd = Command::new("/bin/sh");
|
|
|
|
cmd.arg("-c").arg("pwd")
|
|
|
|
.current_dir("/")
|
2015-02-27 18:23:21 -06:00
|
|
|
.stdout(Stdio::piped());
|
2015-02-06 11:42:57 -06:00
|
|
|
assert_eq!(run_output(cmd), "/\n");
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(all(unix, not(target_os="android")))]
|
|
|
|
#[test]
|
|
|
|
fn stdin_works() {
|
|
|
|
let mut p = Command::new("/bin/sh")
|
|
|
|
.arg("-c").arg("read line; echo $line")
|
2015-02-27 18:23:21 -06:00
|
|
|
.stdin(Stdio::piped())
|
|
|
|
.stdout(Stdio::piped())
|
2015-02-06 11:42:57 -06:00
|
|
|
.spawn().unwrap();
|
|
|
|
p.stdin.as_mut().unwrap().write("foobar".as_bytes()).unwrap();
|
|
|
|
drop(p.stdin.take());
|
|
|
|
let mut out = String::new();
|
|
|
|
p.stdout.as_mut().unwrap().read_to_string(&mut out).unwrap();
|
|
|
|
assert!(p.wait().unwrap().success());
|
|
|
|
assert_eq!(out, "foobar\n");
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
#[cfg(all(unix, not(target_os="android")))]
|
|
|
|
#[test]
|
|
|
|
fn uid_works() {
|
std: Stabilize portions of `std::os::$platform`
This commit starts to organize the `std::os::$platform` modules and in the
process stabilizes some of the functionality contained within. The organization
of these modules will reflect the organization of the standard library itself
with extension traits for primitives in the same corresponding module.
The OS-specific modules will grow more functionality over time including
concrete types that are not extending functionality of other structures, and
these will either go into the closest module in `std::os::$platform` or they
will grow a new module in the hierarchy.
The following items are now stable:
* `os::{unix, windows}`
* `unix::ffi`
* `unix::ffi::OsStrExt`
* `unix::ffi::OsStrExt::{from_bytes, as_bytes, to_cstring}`
* `unix::ffi::OsString`
* `unix::ffi::OsStringExt::{from_vec, into_vec}`
* `unix::process`
* `unix::process::CommandExt`
* `unix::process::CommandExt::{uid, gid}`
* `unix::process::ExitStatusExt`
* `unix::process::ExitStatusExt::signal`
* `unix::prelude`
* `windows::ffi`
* `windows::ffi::OsStringExt`
* `windows::ffi::OsStringExt::from_wide`
* `windows::ffi::OsStrExt`
* `windows::ffi::OsStrExt::encode_wide`
* `windows::prelude`
The following items remain unstable:
* `unix::io`
* `unix::io::{Fd, AsRawFd}`
* `unix::fs::{PermissionsExt, OpenOptionsExt}`
* `windows::io`
* `windows::io::{Handle, AsRawHandle}`
* `windows::io::{Socket, AsRawSocket}`
* `windows::fs`
* `windows::fs::OpenOptionsExt`
Due to the reorgnization of the platform extension modules, this commit is a
breaking change. Most imports can be fixed by adding the relevant libstd module
in the `use` path (such as `ffi` or `fs`).
[breaking-change]
2015-03-13 19:12:38 -05:00
|
|
|
use os::unix::prelude::*;
|
2015-02-06 11:42:57 -06:00
|
|
|
use libc;
|
|
|
|
let mut p = Command::new("/bin/sh")
|
|
|
|
.arg("-c").arg("true")
|
|
|
|
.uid(unsafe { libc::getuid() })
|
|
|
|
.gid(unsafe { libc::getgid() })
|
|
|
|
.spawn().unwrap();
|
|
|
|
assert!(p.wait().unwrap().success());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(all(unix, not(target_os="android")))]
|
|
|
|
#[test]
|
|
|
|
fn uid_to_root_fails() {
|
std: Stabilize portions of `std::os::$platform`
This commit starts to organize the `std::os::$platform` modules and in the
process stabilizes some of the functionality contained within. The organization
of these modules will reflect the organization of the standard library itself
with extension traits for primitives in the same corresponding module.
The OS-specific modules will grow more functionality over time including
concrete types that are not extending functionality of other structures, and
these will either go into the closest module in `std::os::$platform` or they
will grow a new module in the hierarchy.
The following items are now stable:
* `os::{unix, windows}`
* `unix::ffi`
* `unix::ffi::OsStrExt`
* `unix::ffi::OsStrExt::{from_bytes, as_bytes, to_cstring}`
* `unix::ffi::OsString`
* `unix::ffi::OsStringExt::{from_vec, into_vec}`
* `unix::process`
* `unix::process::CommandExt`
* `unix::process::CommandExt::{uid, gid}`
* `unix::process::ExitStatusExt`
* `unix::process::ExitStatusExt::signal`
* `unix::prelude`
* `windows::ffi`
* `windows::ffi::OsStringExt`
* `windows::ffi::OsStringExt::from_wide`
* `windows::ffi::OsStrExt`
* `windows::ffi::OsStrExt::encode_wide`
* `windows::prelude`
The following items remain unstable:
* `unix::io`
* `unix::io::{Fd, AsRawFd}`
* `unix::fs::{PermissionsExt, OpenOptionsExt}`
* `windows::io`
* `windows::io::{Handle, AsRawHandle}`
* `windows::io::{Socket, AsRawSocket}`
* `windows::fs`
* `windows::fs::OpenOptionsExt`
Due to the reorgnization of the platform extension modules, this commit is a
breaking change. Most imports can be fixed by adding the relevant libstd module
in the `use` path (such as `ffi` or `fs`).
[breaking-change]
2015-03-13 19:12:38 -05:00
|
|
|
use os::unix::prelude::*;
|
2015-02-06 11:42:57 -06:00
|
|
|
use libc;
|
|
|
|
|
|
|
|
// if we're already root, this isn't a valid test. Most of the bots run
|
|
|
|
// as non-root though (android is an exception).
|
|
|
|
if unsafe { libc::getuid() == 0 } { return }
|
|
|
|
assert!(Command::new("/bin/ls").uid(0).gid(0).spawn().is_err());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn test_process_status() {
|
|
|
|
let mut status = Command::new("false").status().unwrap();
|
|
|
|
assert!(status.code() == Some(1));
|
|
|
|
|
|
|
|
status = Command::new("true").status().unwrap();
|
|
|
|
assert!(status.success());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_process_output_fail_to_start() {
|
|
|
|
match Command::new("/no-binary-by-this-name-should-exist").output() {
|
2015-03-16 20:08:57 -05:00
|
|
|
Err(e) => assert_eq!(e.kind(), ErrorKind::NotFound),
|
2015-02-06 11:42:57 -06:00
|
|
|
Ok(..) => panic!()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn test_process_output_output() {
|
|
|
|
let Output {status, stdout, stderr}
|
|
|
|
= Command::new("echo").arg("hello").output().unwrap();
|
|
|
|
let output_str = str::from_utf8(stdout.as_slice()).unwrap();
|
|
|
|
|
|
|
|
assert!(status.success());
|
|
|
|
assert_eq!(output_str.trim().to_string(), "hello");
|
|
|
|
// FIXME #7224
|
|
|
|
if !running_on_valgrind() {
|
|
|
|
assert_eq!(stderr, Vec::new());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn test_process_output_error() {
|
|
|
|
let Output {status, stdout, stderr}
|
|
|
|
= Command::new("mkdir").arg(".").output().unwrap();
|
|
|
|
|
|
|
|
assert!(status.code() == Some(1));
|
|
|
|
assert_eq!(stdout, Vec::new());
|
|
|
|
assert!(!stderr.is_empty());
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn test_finish_once() {
|
|
|
|
let mut prog = Command::new("false").spawn().unwrap();
|
|
|
|
assert!(prog.wait().unwrap().code() == Some(1));
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn test_finish_twice() {
|
|
|
|
let mut prog = Command::new("false").spawn().unwrap();
|
|
|
|
assert!(prog.wait().unwrap().code() == Some(1));
|
|
|
|
assert!(prog.wait().unwrap().code() == Some(1));
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn test_wait_with_output_once() {
|
2015-02-27 18:23:21 -06:00
|
|
|
let prog = Command::new("echo").arg("hello").stdout(Stdio::piped())
|
2015-02-06 11:42:57 -06:00
|
|
|
.spawn().unwrap();
|
|
|
|
let Output {status, stdout, stderr} = prog.wait_with_output().unwrap();
|
|
|
|
let output_str = str::from_utf8(stdout.as_slice()).unwrap();
|
|
|
|
|
|
|
|
assert!(status.success());
|
|
|
|
assert_eq!(output_str.trim().to_string(), "hello");
|
|
|
|
// FIXME #7224
|
|
|
|
if !running_on_valgrind() {
|
|
|
|
assert_eq!(stderr, Vec::new());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(all(unix, not(target_os="android")))]
|
|
|
|
pub fn pwd_cmd() -> Command {
|
|
|
|
Command::new("pwd")
|
|
|
|
}
|
|
|
|
#[cfg(target_os="android")]
|
|
|
|
pub fn pwd_cmd() -> Command {
|
|
|
|
let mut cmd = Command::new("/system/bin/sh");
|
|
|
|
cmd.arg("-c").arg("pwd");
|
|
|
|
cmd
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
pub fn pwd_cmd() -> Command {
|
|
|
|
let mut cmd = Command::new("cmd");
|
|
|
|
cmd.arg("/c").arg("cd");
|
|
|
|
cmd
|
|
|
|
}
|
|
|
|
|
2015-03-25 04:17:33 -05:00
|
|
|
#[cfg(not(target_arch = "aarch64"))]
|
2015-02-06 11:42:57 -06:00
|
|
|
#[test]
|
|
|
|
fn test_keep_current_working_dir() {
|
|
|
|
use os;
|
|
|
|
let prog = pwd_cmd().spawn().unwrap();
|
|
|
|
|
|
|
|
let output = String::from_utf8(prog.wait_with_output().unwrap().stdout).unwrap();
|
|
|
|
let parent_dir = os::getcwd().unwrap();
|
|
|
|
let child_dir = old_path::Path::new(output.trim());
|
|
|
|
|
|
|
|
let parent_stat = parent_dir.stat().unwrap();
|
|
|
|
let child_stat = child_dir.stat().unwrap();
|
|
|
|
|
|
|
|
assert_eq!(parent_stat.unstable.device, child_stat.unstable.device);
|
|
|
|
assert_eq!(parent_stat.unstable.inode, child_stat.unstable.inode);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_change_working_directory() {
|
|
|
|
use os;
|
|
|
|
// test changing to the parent of os::getcwd() because we know
|
|
|
|
// the path exists (and os::getcwd() is not expected to be root)
|
|
|
|
let parent_dir = os::getcwd().unwrap().dir_path();
|
2015-03-23 17:54:39 -05:00
|
|
|
let result = pwd_cmd().current_dir(parent_dir.as_str().unwrap()).output().unwrap();
|
2015-02-06 11:42:57 -06:00
|
|
|
|
|
|
|
let output = String::from_utf8(result.stdout).unwrap();
|
|
|
|
let child_dir = old_path::Path::new(output.trim());
|
|
|
|
|
|
|
|
let parent_stat = parent_dir.stat().unwrap();
|
|
|
|
let child_stat = child_dir.stat().unwrap();
|
|
|
|
|
|
|
|
assert_eq!(parent_stat.unstable.device, child_stat.unstable.device);
|
|
|
|
assert_eq!(parent_stat.unstable.inode, child_stat.unstable.inode);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(all(unix, not(target_os="android")))]
|
|
|
|
pub fn env_cmd() -> Command {
|
|
|
|
Command::new("env")
|
|
|
|
}
|
|
|
|
#[cfg(target_os="android")]
|
|
|
|
pub fn env_cmd() -> Command {
|
|
|
|
let mut cmd = Command::new("/system/bin/sh");
|
|
|
|
cmd.arg("-c").arg("set");
|
|
|
|
cmd
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(windows)]
|
|
|
|
pub fn env_cmd() -> Command {
|
|
|
|
let mut cmd = Command::new("cmd");
|
|
|
|
cmd.arg("/c").arg("set");
|
|
|
|
cmd
|
|
|
|
}
|
|
|
|
|
|
|
|
#[cfg(not(target_os="android"))]
|
|
|
|
#[test]
|
|
|
|
fn test_inherit_env() {
|
2015-02-28 12:07:05 -06:00
|
|
|
use std::env;
|
2015-02-06 11:42:57 -06:00
|
|
|
if running_on_valgrind() { return; }
|
|
|
|
|
|
|
|
let result = env_cmd().output().unwrap();
|
|
|
|
let output = String::from_utf8(result.stdout).unwrap();
|
|
|
|
|
2015-02-28 12:07:05 -06:00
|
|
|
for (ref k, ref v) in env::vars() {
|
2015-02-06 11:42:57 -06:00
|
|
|
// don't check windows magical empty-named variables
|
|
|
|
assert!(k.is_empty() ||
|
2015-03-07 20:08:48 -06:00
|
|
|
output.contains(&format!("{}={}", *k, *v)),
|
2015-02-06 11:42:57 -06:00
|
|
|
"output doesn't contain `{}={}`\n{}",
|
|
|
|
k, v, output);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#[cfg(target_os="android")]
|
|
|
|
#[test]
|
|
|
|
fn test_inherit_env() {
|
|
|
|
use os;
|
|
|
|
if running_on_valgrind() { return; }
|
|
|
|
|
|
|
|
let mut result = env_cmd().output().unwrap();
|
|
|
|
let output = String::from_utf8(result.stdout).unwrap();
|
|
|
|
|
|
|
|
let r = os::env();
|
|
|
|
for &(ref k, ref v) in &r {
|
|
|
|
// don't check android RANDOM variables
|
|
|
|
if *k != "RANDOM".to_string() {
|
2015-03-07 20:08:48 -06:00
|
|
|
assert!(output.contains(&format!("{}={}",
|
|
|
|
*k,
|
|
|
|
*v)) ||
|
|
|
|
output.contains(&format!("{}=\'{}\'",
|
|
|
|
*k,
|
|
|
|
*v)));
|
2015-02-06 11:42:57 -06:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_override_env() {
|
|
|
|
use env;
|
|
|
|
|
|
|
|
// In some build environments (such as chrooted Nix builds), `env` can
|
|
|
|
// only be found in the explicitly-provided PATH env variable, not in
|
|
|
|
// default places such as /bin or /usr/bin. So we need to pass through
|
|
|
|
// PATH to our sub-process.
|
|
|
|
let mut cmd = env_cmd();
|
|
|
|
cmd.env_clear().env("RUN_TEST_NEW_ENV", "123");
|
|
|
|
if let Some(p) = env::var_os("PATH") {
|
|
|
|
cmd.env("PATH", &p);
|
|
|
|
}
|
|
|
|
let result = cmd.output().unwrap();
|
|
|
|
let output = String::from_utf8_lossy(result.stdout.as_slice()).to_string();
|
|
|
|
|
|
|
|
assert!(output.contains("RUN_TEST_NEW_ENV=123"),
|
|
|
|
"didn't find RUN_TEST_NEW_ENV inside of:\n\n{}", output);
|
|
|
|
}
|
|
|
|
|
|
|
|
#[test]
|
|
|
|
fn test_add_to_env() {
|
|
|
|
let result = env_cmd().env("RUN_TEST_NEW_ENV", "123").output().unwrap();
|
|
|
|
let output = String::from_utf8_lossy(result.stdout.as_slice()).to_string();
|
|
|
|
|
|
|
|
assert!(output.contains("RUN_TEST_NEW_ENV=123"),
|
|
|
|
"didn't find RUN_TEST_NEW_ENV inside of:\n\n{}", output);
|
|
|
|
}
|
|
|
|
}
|