tee: correctly handle writing to read-only files

Author: BenWiederhake

src/uu/tee/src/tee.rs

@@ -5,7 +5,7 @@
 
 use clap::{builder::PossibleValue, crate_version, Arg, ArgAction, Command};
 use std::fs::OpenOptions;
-use std::io::{copy, sink, stdin, stdout, Error, ErrorKind, Read, Result, Write};
+use std::io::{copy, stdin, stdout, Error, ErrorKind, Read, Result, Write};
 use std::path::PathBuf;
 use uucore::display::Quotable;
 use uucore::error::UResult;
@@ -150,8 +150,9 @@ fn tee(options: &Options) -> Result<()> {
         .files
         .clone()
         .into_iter()
-        .map(|file| open(file, options.append, options.output_error.as_ref()))
+        .filter_map(|file| open(file, options.append, options.output_error.as_ref()))
         .collect::<Result<Vec<NamedWriter>>>()?;
+    let had_open_errors = writers.len() != options.files.len();
 
     writers.insert(
         0,
@@ -176,14 +177,21 @@ fn tee(options: &Options) -> Result<()> {
         _ => Ok(()),
     };
 
-    if res.is_err() || output.flush().is_err() || output.error_occurred() {
+    if had_open_errors || res.is_err() || output.flush().is_err() || output.error_occurred() {
         Err(Error::from(ErrorKind::Other))
     } else {
         Ok(())
     }
 }
 
-fn open(name: String, append: bool, output_error: Option<&OutputErrorMode>) -> Result<NamedWriter> {
+/// Tries to open the indicated file and return it. Reports an error if that's not possible.
+/// If that error should lead to program termination, this function returns Some(Err()),
+/// otherwise it returns None.
+fn open(
+    name: String,
+    append: bool,
+    output_error: Option<&OutputErrorMode>,
+) -> Option<Result<NamedWriter>> {
     let path = PathBuf::from(name.clone());
     let mut options = OpenOptions::new();
     let mode = if append {
@@ -192,18 +200,15 @@ fn open(name: String, append: bool, output_error: Option<&OutputErrorMode>) -> R
         options.truncate(true)
     };
     match mode.write(true).create(true).open(path.as_path()) {
-        Ok(file) => Ok(NamedWriter {
+        Ok(file) => Some(Ok(NamedWriter {
             inner: Box::new(file),
             name,
-        }),
+        })),
         Err(f) => {
             show_error!("{}: {}", name.maybe_quote(), f);
             match output_error {
-                Some(OutputErrorMode::Exit | OutputErrorMode::ExitNoPipe) => Err(f),
-                _ => Ok(NamedWriter {
-                    inner: Box::new(sink()),
-                    name,
-                }),
+                Some(OutputErrorMode::Exit | OutputErrorMode::ExitNoPipe) => Some(Err(f)),
+                _ => None,
             }
         }
     }

tests/by-util/test_tee.rs

@@ -92,6 +92,26 @@ fn test_tee_no_more_writeable_1() {
     assert_eq!(at.read(file_out), content);
 }
 
+#[test]
+fn test_readonly() {
+    let (at, mut ucmd) = at_and_ucmd!();
+    let content_tee = "hello";
+    let content_file = "world";
+    let file_out = "tee_file_out";
+    let writable_file = "tee_file_out2";
+    at.write(file_out, content_file);
+    at.set_readonly(file_out);
+    ucmd.arg(file_out)
+        .arg(writable_file)
+        .pipe_in(content_tee)
+        .ignore_stdin_write_error()
+        .fails()
+        .stdout_is(content_tee)
+        .stderr_contains("Permission denied");
+    assert_eq!(at.read(file_out), content_file);
+    assert_eq!(at.read(writable_file), content_tee);
+}
+
 #[test]
 #[cfg(target_os = "linux")]
 fn test_tee_no_more_writeable_2() {