RGBCube/uutils-coreutils
1
Fork 0
mirror of https://github.com/RGBCube/uutils-coreutils synced 2025-07-27 19:17:43 +00:00
Code Issues Activity

Merge pull request #1363 from rethab/reject-null-with-program

Browse source 
env: reject program with --null, error stderr
This commit is contained in:
Alex Lyon 2019-04-28 04:04:23 -07:00 • committed by GitHub
commit bcc821caa8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 24 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

24
src/env/env.rs vendored
View file

@ -47,7 +47,7 @@ pub fn uumain(args: Vec<String>) -> i32 {
.optflag(
"0",
"null",
"end each output line with a 0 byte rather than newline",
"end each output line with a 0 byte rather than newline (only valid when printing the environment)",
)
.optopt("u", "unset", "remove variable from the environment", "NAME");
@ -103,14 +103,14 @@ pub fn uumain(args: Vec<String>) -> i32 {
let var = iter.next();
match var {
None => println!("{}: this option requires an argument: {}", NAME, opt),
None => eprintln!("{}: this option requires an argument: {}", NAME, opt),
Some(s) => opts.unsets.push(s.to_owned()),
}
}
_ => {
println!("{}: invalid option \"{}\"", NAME, *opt);
println!("Type \"{} --help\" for detailed informations", NAME);
eprintln!("{}: invalid option \"{}\"", NAME, *opt);
eprintln!("Type \"{} --help\" for detailed information", NAME);
return 1;
}
}
@ -134,13 +134,13 @@ pub fn uumain(args: Vec<String>) -> i32 {
let var = iter.next();
match var {
None => println!("{}: this option requires an argument: {}", NAME, opt),
None => eprintln!("{}: this option requires an argument: {}", NAME, opt),
Some(s) => opts.unsets.push(s.to_owned()),
}
}
_ => {
println!("{}: illegal option -- {}", NAME, c);
println!("Type \"{} --help\" for detailed informations", NAME);
eprintln!("{}: illegal option -- {}", NAME, c);
eprintln!("Type \"{} --help\" for detailed information", NAME);
return 1;
}
}
@ -159,6 +159,11 @@ pub fn uumain(args: Vec<String>) -> i32 {
}
// no, its a program-like opt
_ => {
if opts.null {
eprintln!("{}: cannot specify --null (-0) with command", NAME);
eprintln!("Type \"{} --help\" for detailed information", NAME);
return 1;
}
opts.program.push(opt.clone());
break;
}
@ -170,6 +175,11 @@ pub fn uumain(args: Vec<String>) -> i32 {
// read program arguments
for opt in iter {
if opts.null {
eprintln!("{}: cannot specify --null (-0) with command", NAME);
eprintln!("Type \"{} --help\" for detailed information", NAME);
return 1;
}
opts.program.push(opt.clone())
}

2
tests/common/util.rs
View file

@ -606,7 +606,7 @@ impl UCommand {
}
/// Spawns the command, feeds the stdin if any, waits for the result,
/// asserts success, and returns a command result.
/// asserts failure, and returns a command result.
pub fn fails(&mut self) -> CmdResult {
let cmd_result = self.run();
cmd_result.failure();

6
tests/test_env.rs
View file

@ -81,3 +81,9 @@ fn test_unset_variable() {
assert_eq!(out.lines().any(|line| line.starts_with("HOME=")), false);
}
#[test]
fn test_fail_null_with_program() {
let out = new_ucmd!().arg("--null").arg("cd").fails().stderr;
assert!(out.contains("cannot specify --null (-0) with command"));
}