From 3c271304f5496b75e6f8f287934bf2d47a3c071f Mon Sep 17 00:00:00 2001
From: Michael Debertol <michael.debertol@gmail.com>
Date: Wed, 16 Jun 2021 16:54:28 +0200
Subject: [PATCH 1/3] tty: correct exit code for wrong args

---
 src/uu/tty/src/tty.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/uu/tty/src/tty.rs b/src/uu/tty/src/tty.rs
index edcdf091e..331f5e254 100644
--- a/src/uu/tty/src/tty.rs
+++ b/src/uu/tty/src/tty.rs
@@ -44,7 +44,15 @@ pub fn uumain(args: impl uucore::Args) -> i32 {
                 .help("print nothing, only return an exit status")
                 .required(false),
         )
-        .get_matches_from(args);
+        .get_matches_from_safe(args);
+
+    let matches = match matches {
+        Ok(m) => m,
+        Err(e) => {
+            eprint!("{}", e);
+            return 2;
+        }
+    };
 
     let silent = matches.is_present(options::SILENT);
 

From aeaf2cebfb3b931a0538ad0f2ec9929180cb2ea8 Mon Sep 17 00:00:00 2001
From: Michael Debertol <michael.debertol@gmail.com>
Date: Wed, 16 Jun 2021 17:16:28 +0200
Subject: [PATCH 2/3] tests/tty: fix test inputs

calling `pipe_in("</dev/null")` does not pipe in the `/dev/null` file,
but the raw string "</dev/null".

Closes https://github.com/uutils/coreutils/issues/2301
---
 tests/by-util/test_tty.rs | 46 +++++++++++++++++++++++----------------
 1 file changed, 27 insertions(+), 19 deletions(-)

diff --git a/tests/by-util/test_tty.rs b/tests/by-util/test_tty.rs
index 6bca54e03..a59be61b4 100644
--- a/tests/by-util/test_tty.rs
+++ b/tests/by-util/test_tty.rs
@@ -1,11 +1,14 @@
+use std::fs::File;
+
 use crate::common::util::*;
 
 #[test]
 #[cfg(not(windows))]
 fn test_dev_null() {
     new_ucmd!()
-        .pipe_in("</dev/null")
+        .set_stdin(File::open("/dev/null").unwrap())
         .fails()
+        .code_is(1)
         .stdout_is("not a tty\n");
 }
 
@@ -14,44 +17,49 @@ fn test_dev_null() {
 fn test_dev_null_silent() {
     new_ucmd!()
         .args(&["-s"])
-        .pipe_in("</dev/null")
+        .set_stdin(File::open("/dev/null").unwrap())
         .fails()
+        .code_is(1)
         .stdout_is("");
 }
 
 #[test]
 fn test_close_stdin() {
-    new_ucmd!().pipe_in("<&-").fails().stdout_is("not a tty\n");
+    let mut child = new_ucmd!().run_no_wait();
+    drop(child.stdin.take());
+    let output = child.wait_with_output().unwrap();
+    assert_eq!(output.status.code(), Some(1));
+    assert_eq!(std::str::from_utf8(&output.stdout), Ok("not a tty\n"));
 }
 
 #[test]
 fn test_close_stdin_silent() {
-    new_ucmd!()
-        .args(&["-s"])
-        .pipe_in("<&-")
-        .fails()
-        .stdout_is("");
+    let mut child = new_ucmd!().arg("-s").run_no_wait();
+    drop(child.stdin.take());
+    let output = child.wait_with_output().unwrap();
+    assert_eq!(output.status.code(), Some(1));
+    assert!(output.stdout.is_empty());
 }
 
 #[test]
 fn test_close_stdin_silent_long() {
-    new_ucmd!()
-        .args(&["--silent"])
-        .pipe_in("<&-")
-        .fails()
-        .stdout_is("");
+    let mut child = new_ucmd!().arg("--silent").run_no_wait();
+    drop(child.stdin.take());
+    let output = child.wait_with_output().unwrap();
+    assert_eq!(output.status.code(), Some(1));
+    assert!(output.stdout.is_empty());
 }
 
 #[test]
 fn test_close_stdin_silent_alias() {
-    new_ucmd!()
-        .args(&["--quiet"])
-        .pipe_in("<&-")
-        .fails()
-        .stdout_is("");
+    let mut child = new_ucmd!().arg("--quiet").run_no_wait();
+    drop(child.stdin.take());
+    let output = child.wait_with_output().unwrap();
+    assert_eq!(output.status.code(), Some(1));
+    assert!(output.stdout.is_empty());
 }
 
 #[test]
 fn test_wrong_argument() {
-    new_ucmd!().args(&["a"]).fails();
+    new_ucmd!().args(&["a"]).fails().code_is(2);
 }

From 4c5ee1dbd7373fab1f28540bbbd005d4c5898f47 Mon Sep 17 00:00:00 2001
From: Michael Debertol <michael.debertol@gmail.com>
Date: Wed, 16 Jun 2021 17:38:07 +0200
Subject: [PATCH 3/3] tty: correct exit code for write errrors

---
 src/uu/tty/src/tty.rs     | 14 +++++++++++---
 tests/by-util/test_tty.rs |  9 +++++++++
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/src/uu/tty/src/tty.rs b/src/uu/tty/src/tty.rs
index 331f5e254..cc5052dea 100644
--- a/src/uu/tty/src/tty.rs
+++ b/src/uu/tty/src/tty.rs
@@ -14,6 +14,7 @@ extern crate uucore;
 
 use clap::{crate_version, App, Arg};
 use std::ffi::CStr;
+use std::io::Write;
 use uucore::InvalidEncodingHandling;
 
 static ABOUT: &str = "Print the file name of the terminal connected to standard input.";
@@ -66,11 +67,18 @@ pub fn uumain(args: impl uucore::Args) -> i32 {
         }
     };
 
+    let mut stdout = std::io::stdout();
+
     if !silent {
-        if !tty.chars().all(|c| c.is_whitespace()) {
-            println!("{}", tty);
+        let write_result = if !tty.chars().all(|c| c.is_whitespace()) {
+            writeln!(stdout, "{}", tty)
         } else {
-            println!("not a tty");
+            writeln!(stdout, "not a tty")
+        };
+        if write_result.is_err() || stdout.flush().is_err() {
+            // Don't return to prevent a panic later when another flush is attempted
+            // because the `uucore_procs::main` macro inserts a flush after execution for every utility.
+            std::process::exit(3);
         }
     }
 
diff --git a/tests/by-util/test_tty.rs b/tests/by-util/test_tty.rs
index a59be61b4..6ba8cd029 100644
--- a/tests/by-util/test_tty.rs
+++ b/tests/by-util/test_tty.rs
@@ -63,3 +63,12 @@ fn test_close_stdin_silent_alias() {
 fn test_wrong_argument() {
     new_ucmd!().args(&["a"]).fails().code_is(2);
 }
+
+#[test]
+#[cfg(not(windows))]
+fn test_stdout_fail() {
+    let mut child = new_ucmd!().run_no_wait();
+    drop(child.stdout.take());
+    let status = child.wait().unwrap();
+    assert_eq!(status.code(), Some(3));
+}