f09592a5d1
This implements set_timeout() for std::io::Process which will affect wait() operations on the process. This follows the same pattern as the rest of the timeouts emerging in std::io::net. The implementation was super easy for everything except libnative on unix (backwards from usual!), which required a good bit of signal handling. There's a doc comment explaining the strategy in libnative. Internally, this also required refactoring the "helper thread" implementation used by libnative to allow for an extra helper thread (not just the timer). This is a breaking change in terms of the io::Process API. It is now possible for wait() to fail, and subsequently wait_with_output(). These two functions now return IoResult<T> due to the fact that they can time out. Additionally, the wait_with_output() function has moved from taking `&mut self` to taking `self`. If a timeout occurs while waiting with output, the semantics are undesirable in almost all cases if attempting to re-wait on the process. Equivalent functionality can still be achieved by dealing with the output handles manually. [breaking-change] cc #13523
136 lines
3.8 KiB
Rust
136 lines
3.8 KiB
Rust
// Copyright 2012-2013-2014 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.
|
|
|
|
// ignore-pretty
|
|
// compile-flags:--test
|
|
|
|
// NB: These tests kill child processes. Valgrind sees these children as leaking
|
|
// memory, which makes for some *confusing* logs. That's why these are here
|
|
// instead of in std.
|
|
|
|
#![feature(macro_rules)]
|
|
extern crate libc;
|
|
|
|
extern crate native;
|
|
extern crate green;
|
|
extern crate rustuv;
|
|
|
|
use std::io::Process;
|
|
|
|
macro_rules! succeed( ($e:expr) => (
|
|
match $e { Ok(..) => {}, Err(e) => fail!("failure: {}", e) }
|
|
) )
|
|
|
|
macro_rules! iotest (
|
|
{ fn $name:ident() $b:block $($a:attr)* } => (
|
|
mod $name {
|
|
#![allow(unused_imports)]
|
|
|
|
use std::io::timer;
|
|
use libc;
|
|
use std::str;
|
|
use std::io::process::{Process, ProcessOutput};
|
|
use native;
|
|
use super::*;
|
|
|
|
fn f() $b
|
|
|
|
$($a)* #[test] fn green() { f() }
|
|
$($a)* #[test] fn native() {
|
|
use native;
|
|
let (tx, rx) = channel();
|
|
native::task::spawn(proc() { tx.send(f()) });
|
|
rx.recv();
|
|
}
|
|
}
|
|
)
|
|
)
|
|
|
|
#[cfg(test)] #[start]
|
|
fn start(argc: int, argv: **u8) -> int {
|
|
green::start(argc, argv, rustuv::event_loop, __test::main)
|
|
}
|
|
|
|
iotest!(fn test_destroy_once() {
|
|
let mut p = sleeper();
|
|
match p.signal_exit() {
|
|
Ok(()) => {}
|
|
Err(e) => fail!("error: {}", e),
|
|
}
|
|
})
|
|
|
|
#[cfg(unix)]
|
|
pub fn sleeper() -> Process {
|
|
Process::new("sleep", ["1000".to_owned()]).unwrap()
|
|
}
|
|
#[cfg(windows)]
|
|
pub fn sleeper() -> Process {
|
|
// There's a `timeout` command on windows, but it doesn't like having
|
|
// its output piped, so instead just ping ourselves a few times with
|
|
// gaps inbetweeen so we're sure this process is alive for awhile
|
|
Process::new("ping", ["127.0.0.1".to_owned(), "-n".to_owned(), "1000".to_owned()]).unwrap()
|
|
}
|
|
|
|
iotest!(fn test_destroy_twice() {
|
|
let mut p = sleeper();
|
|
succeed!(p.signal_exit()); // this shouldnt crash...
|
|
let _ = p.signal_exit(); // ...and nor should this (and nor should the destructor)
|
|
})
|
|
|
|
pub fn test_destroy_actually_kills(force: bool) {
|
|
use std::io::process::{Process, ProcessOutput, ExitStatus, ExitSignal};
|
|
use std::io::timer;
|
|
use libc;
|
|
use std::str;
|
|
|
|
#[cfg(unix,not(target_os="android"))]
|
|
static BLOCK_COMMAND: &'static str = "cat";
|
|
|
|
#[cfg(unix,target_os="android")]
|
|
static BLOCK_COMMAND: &'static str = "/system/bin/cat";
|
|
|
|
#[cfg(windows)]
|
|
static BLOCK_COMMAND: &'static str = "cmd";
|
|
|
|
// this process will stay alive indefinitely trying to read from stdin
|
|
let mut p = Process::new(BLOCK_COMMAND, []).unwrap();
|
|
|
|
assert!(p.signal(0).is_ok());
|
|
|
|
if force {
|
|
p.signal_kill().unwrap();
|
|
} else {
|
|
p.signal_exit().unwrap();
|
|
}
|
|
|
|
// Don't let this test time out, this should be quick
|
|
let (tx, rx1) = channel();
|
|
let mut t = timer::Timer::new().unwrap();
|
|
let rx2 = t.oneshot(1000);
|
|
spawn(proc() {
|
|
select! {
|
|
() = rx2.recv() => unsafe { libc::exit(1) },
|
|
() = rx1.recv() => {}
|
|
}
|
|
});
|
|
match p.wait().unwrap() {
|
|
ExitStatus(..) => fail!("expected a signal"),
|
|
ExitSignal(..) => tx.send(()),
|
|
}
|
|
}
|
|
|
|
iotest!(fn test_unforced_destroy_actually_kills() {
|
|
test_destroy_actually_kills(false);
|
|
})
|
|
|
|
iotest!(fn test_forced_destroy_actually_kills() {
|
|
test_destroy_actually_kills(true);
|
|
})
|