Skip to content

Commit

Permalink
Merge pull request uutils#5402 from Luv-Ray/short-circuit
Browse files Browse the repository at this point in the history
`expr`: short-circuit evaluation for `&`
  • Loading branch information
cakebaker authored Oct 14, 2023
2 parents f979f14 + 40f05a3 commit 21c8538
Show file tree
Hide file tree
Showing 2 changed files with 34 additions and 9 deletions.
28 changes: 19 additions & 9 deletions src/uu/expr/src/syntax_tree.rs
Original file line number Diff line number Diff line change
Expand Up @@ -166,16 +166,26 @@ impl AstNode {
{
let mut out = Vec::with_capacity(operands.len());
let mut operands = operands.iter();
// check the first value before `|`, stop evaluate and return directly if it is true.
// push dummy to pass the check of `len() == 2`
if op_type == "|" {
if let Some(value) = operands.next() {
let value = value.evaluate()?;
out.push(value.clone());
if value_as_bool(&value) {
out.push(String::from("dummy"));
return Ok(out);

if let Some(value) = operands.next() {
let value = value.evaluate()?;
out.push(value.clone());
// short-circuit evaluation for `|` and `&`
// push dummy to pass `assert!(values.len() == 2);`
match op_type.as_ref() {
"|" => {
if value_as_bool(&value) {
out.push(String::from("dummy"));
return Ok(out);
}
}
"&" => {
if !value_as_bool(&value) {
out.push(String::from("dummy"));
return Ok(out);
}
}
_ => {}
}
}

Expand Down
15 changes: 15 additions & 0 deletions tests/by-util/test_expr.rs
Original file line number Diff line number Diff line change
Expand Up @@ -155,6 +155,21 @@ fn test_and() {
.args(&["-14", "&", "1"])
.run()
.stdout_is("-14\n");

new_ucmd!()
.args(&["0", "&", "a", "/", "5"])
.run()
.stdout_only("0\n");

new_ucmd!()
.args(&["", "&", "a", "/", "5"])
.run()
.stdout_only("0\n");

new_ucmd!()
.args(&["-1", "&", "10", "/", "5"])
.succeeds()
.stdout_only("-1\n");
}

#[test]
Expand Down

0 comments on commit 21c8538

Please sign in to comment.