From 2a1f89e5d941a77e8c93ad0a5fe78229a4207d61 Mon Sep 17 00:00:00 2001
From: Rob Landley <rob@landley.net>
Date: Tue, 16 Apr 2019 16:53:27 -0500
Subject: Add argument to xflush() so it can test for stdout err without
 flushing.

---
 lib/lib.h   |  2 +-
 lib/xwrap.c | 16 +++++++++-------
 2 files changed, 10 insertions(+), 8 deletions(-)

(limited to 'lib')

diff --git a/lib/lib.h b/lib/lib.h
index 0f15484e..6ec2c696 100644
--- a/lib/lib.h
+++ b/lib/lib.h
@@ -129,7 +129,7 @@ void xputsl(char *s, int len);
 void xputsn(char *s);
 void xputs(char *s);
 void xputc(char c);
-void xflush(void);
+void xflush(int flush);
 void xexec(char **argv);
 pid_t xpopen_both(char **argv, int *pipes);
 int xwaitpid(pid_t pid);
diff --git a/lib/xwrap.c b/lib/xwrap.c
index 12170ada..a242cd53 100644
--- a/lib/xwrap.c
+++ b/lib/xwrap.c
@@ -54,7 +54,7 @@ void xexit(void)
 
     free(al);
   }
-  if (fflush(0) || ferror(stdout)) if (!toys.exitval) perror_msg("write");
+  xflush(1);
   _xexit();
 }
 
@@ -139,9 +139,11 @@ char *xmprintf(char *format, ...)
   return ret;
 }
 
-void xflush(void)
+// if !flush just check for error on stdout without flushing 
+void xflush(int flush)
 {
-  if (fflush(stdout) || ferror(stdout)) perror_exit("write");
+  if ((flush && fflush(0)) || ferror(stdout))
+    if (!toys.exitval) perror_msg("write");
 }
 
 void xprintf(char *format, ...)
@@ -151,7 +153,7 @@ void xprintf(char *format, ...)
 
   vprintf(format, va);
   va_end(va);
-  xflush();
+  xflush(0);
 }
 
 // Put string with length (does not append newline)
@@ -164,7 +166,7 @@ void xputsl(char *s, int len)
     len -= out;
     s += out;
   }
-  xflush();
+  xflush(0);
 }
 
 // xputs with no newline
@@ -177,13 +179,13 @@ void xputsn(char *s)
 void xputs(char *s)
 {
   puts(s);
-  xflush();
+  xflush(0);
 }
 
 void xputc(char c)
 {
   if (EOF == fputc(c, stdout)) perror_exit("write");
-  xflush();
+  xflush(0);
 }
 
 // This is called through the XVFORK macro because parent/child of vfork
-- 
cgit v1.2.3