RGBCube/uutils-coreutils
1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-08-02 14:07:46 +00:00
Code Issues Activity

Update timeout.rs to new io

Browse source
This commit is contained in:
Remi Rampin 2015-06-30 15:23:55 -04:00
parent 0ed57b3896
commit 66a40d555d
1 changed files with 22 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

51
src/timeout/timeout.rs
View file

@ -1,5 +1,4 @@
#![crate_name = "timeout"]
#![feature(collections, core, io, old_io, rustc_private)]
/*
* This file is part of the uutils coreutils package.
@ -13,8 +12,9 @@
extern crate getopts;
extern crate libc;
use std::old_io::{PathDoesntExist, FileNotFound};
use std::old_io::process::{Command, ExitStatus, ExitSignal, InheritFd};
use std::io::{ErrorKind, Write};
use std::process::{Command, Stdio};
use std::os::unix::process::ExitStatusExt;
#[path = "../common/util.rs"]
#[macro_use]
@ -38,15 +38,14 @@ static ERR_EXIT_STATUS: i32 = 125;
pub fn uumain(args: Vec<String>) -> i32 {
let program = args[0].clone();
let opts = [
getopts::optflag("", "preserve-status", "exit with the same status as COMMAND, even when the command times out"),
getopts::optflag("", "foreground", "when not running timeout directly from a shell prompt, allow COMMAND to read from the TTY and get TTY signals; in this mode, children of COMMAND will not be timed out"),
getopts::optopt("k", "kill-after", "also send a KILL signal if COMMAND is still running this long after the initial signal was sent", "DURATION"),
getopts::optflag("s", "signal", "specify the signal to be sent on timeout; SIGNAL may be a name like 'HUP' or a number; see 'kill -l' for a list of signals"),
getopts::optflag("h", "help", "display this help and exit"),
getopts::optflag("V", "version", "output version information and exit")
];
let matches = match getopts::getopts(args.tail(), &opts) {
let mut opts = getopts::Options::new();
opts.optflag("", "preserve-status", "exit with the same status as COMMAND, even when the command times out");
opts.optflag("", "foreground", "when not running timeout directly from a shell prompt, allow COMMAND to read from the TTY and get TTY signals; in this mode, children of COMMAND will not be timed out");
opts.optopt("k", "kill-after", "also send a KILL signal if COMMAND is still running this long after the initial signal was sent", "DURATION");
opts.optflag("s", "signal", "specify the signal to be sent on timeout; SIGNAL may be a name like 'HUP' or a number; see 'kill -l' for a list of signals");
opts.optflag("h", "help", "display this help and exit");
opts.optflag("V", "version", "output version information and exit");
let matches = match opts.parse(&args[1..]) {
Ok(m) => m,
Err(f) => {
crash!(ERR_EXIT_STATUS, "{}", f)
@ -58,7 +57,7 @@ pub fn uumain(args: Vec<String>) -> i32 {
Usage:
{} [OPTION] DURATION COMMAND [ARG]...
{}", NAME, VERSION, program, getopts::usage("Start COMMAND, and kill it if still running after DURATION.", &opts));
{}", NAME, VERSION, program, &opts.usage("Start COMMAND, and kill it if still running after DURATION."));
} else if matches.opt_present("version") {
println!("{} {}", NAME, VERSION);
} else if matches.free.len() < 2 {
@ -69,7 +68,7 @@ Usage:
let status = matches.opt_present("preserve-status");
let foreground = matches.opt_present("foreground");
let kill_after = match matches.opt_str("kill-after") {
Some(tstr) => match time::from_str(tstr.as_slice()) {
Some(tstr) => match time::from_str(&tstr) {
Ok(time) => time,
Err(f) => {
show_error!("{}", f);
@ -79,7 +78,7 @@ Usage:
None => 0f64
};
let signal = match matches.opt_str("signal") {
Some(sigstr) => match signals::signal_by_name_or_value(sigstr.as_slice()) {
Some(sigstr) => match signals::signal_by_name_or_value(&sigstr) {
Some(sig) => sig,
None => {
show_error!("invalid signal '{}'", sigstr);
@ -88,14 +87,14 @@ Usage:
},
None => signals::signal_by_name_or_value("TERM").unwrap()
};
let duration = match time::from_str(matches.free[0].as_slice()) {
let duration = match time::from_str(&matches.free[0]) {
Ok(time) => time,
Err(f) => {
show_error!("{}", f);
return ERR_EXIT_STATUS;
}
};
return timeout(matches.free[1].as_slice(), &matches.free[2..], duration, signal, kill_after, foreground, status);
return timeout(&matches.free[1], &matches.free[2..], duration, signal, kill_after, foreground, status);
}
0
@ -106,14 +105,14 @@ fn timeout(cmdname: &str, args: &[String], duration: f64, signal: usize, kill_af
unsafe { setpgid(0, 0) };
}
let mut process = match Command::new(cmdname).args(args)
.stdin(InheritFd(0))
.stdout(InheritFd(1))
.stderr(InheritFd(2))
.stdin(Stdio::inherit())
.stdout(Stdio::inherit())
.stderr(Stdio::inherit())
.spawn() {
Ok(p) => p,
Err(err) => {
show_error!("failed to execute process: {}", err);
if err.kind == FileNotFound || err.kind == PathDoesntExist {
if err.kind() == ErrorKind::NotFound {
// XXX: not sure which to use
return 127;
} else {
@ -124,20 +123,14 @@ fn timeout(cmdname: &str, args: &[String], duration: f64, signal: usize, kill_af
};
process.set_timeout(Some((duration * 1000f64) as u64)); // FIXME: this ignores the f64...
match process.wait() {
Ok(status) => match status {
ExitStatus(stat) => stat as i32,
ExitSignal(stat) => stat as i32
},
Ok(status) => status.code().unwrap_or_else(|| status.signal().unwrap()),
Err(_) => {
return_if_err!(ERR_EXIT_STATUS, process.signal(signal as isize));
process.set_timeout(Some((kill_after * 1000f64) as u64));
match process.wait() {
Ok(status) => {
if preserve_status {
match status {
ExitStatus(stat) => stat as i32,
ExitSignal(stat) => stat as i32
}
status.code().unwrap_or_else(|| status.signal().unwrap())
} else {
124
}