If read were to return with an error, bad things would happen. Fix it.

Also, make sure read errors are reflected in the applet exit code.

1 files changed, 6 insertions, 2 deletions

diff --git a/coreutils/tee.c b/coreutils/tee.c
index 6ec1d6dff..ba2e10f90 100644
--- a/coreutils/tee.c
+++ b/coreutils/tee.c
@@ -38,7 +38,7 @@ int tee_main(int argc, char **argv)
         int flags;
         int retval = EXIT_SUCCESS;
 #ifdef CONFIG_FEATURE_TEE_USE_BLOCK_IO
-        size_t c;
+        ssize_t c;
         RESERVE_CONFIG_BUFFER(buf, BUFSIZ);
 #else
         int c;
@@ -78,12 +78,16 @@ int tee_main(int argc, char **argv)
         *p = NULL;                              /* Store the sentinal value. */
 
 #ifdef CONFIG_FEATURE_TEE_USE_BLOCK_IO
-        while ((c = read(STDIN_FILENO, buf, BUFSIZ)) != 0) {
+        while ((c = safe_read(STDIN_FILENO, buf, BUFSIZ)) > 0) {
                 for (p=files ; *p ; p++) {
                         fwrite(buf, 1, c, *p);
                 }
         }
 
+        if (c < 0) {                    /* Make sure read errors are signaled. */
+                retval = EXIT_FAILURE;
+        }
+
 #ifdef CONFIG_FEATURE_CLEAN_UP
         RELEASE_CONFIG_BUFFER(buf);
 #endif