aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--colorize.c36
1 files changed, 34 insertions, 2 deletions
diff --git a/colorize.c b/colorize.c
index eb8f7c0..da03804 100644
--- a/colorize.c
+++ b/colorize.c
@@ -26,6 +26,7 @@
#include <assert.h>
#include <ctype.h>
#include <errno.h>
+#include <fcntl.h>
#include <getopt.h>
#include <pwd.h>
#include <stdarg.h>
@@ -403,6 +404,12 @@ main (int argc, char **argv)
program_name = argv[0];
atexit (cleanup);
+ if (fcntl (STDIN_FILENO, F_SETFL, fcntl (STDIN_FILENO, F_GETFL) | O_NONBLOCK) == -1)
+ {
+ perror ("fcntl");
+ exit (EXIT_FAILURE);
+ }
+
setvbuf (stdout, NULL, _IOLBF, 0);
#if DEBUG
@@ -1348,9 +1355,16 @@ read_print_stream (const char *attr, const struct color **colors, const char *fi
size_t bytes_read;
char *eol;
const char *line;
+ bool sleep = false;
+ errno = 0;
bytes_read = fread (buf, 1, BUF_SIZE, stream);
- if (bytes_read != BUF_SIZE && ferror (stream))
- vfprintf_fail (formats[FMT_ERROR], BUF_SIZE, "read");
+ if (bytes_read != BUF_SIZE)
+ {
+ if (errno == EAGAIN || errno == EWOULDBLOCK)
+ sleep = true;
+ else if (fileno (stream) != STDIN_FILENO && ferror (stream))
+ vfprintf_fail (formats[FMT_ERROR], BUF_SIZE, "read");
+ }
buf[bytes_read] = '\0';
line = buf;
while ((eol = strpbrk (line, "\n\r")))
@@ -1389,6 +1403,24 @@ read_print_stream (const char *attr, const struct color **colors, const char *fi
else
print_line (attr, colors, line, 0, true);
}
+ if (sleep)
+ {
+ /* This entire construct is a bit of a hack...
+ I don't see /currently/ how it could be properly converted
+ to using open() and friends with {poll,select}().
+
+ It has been tested on older hardware to assert that the
+ performance loss should only be minimal...
+
+ -stsc / 2024-11-03
+ */
+ const struct timespec ts = { 0, 50000000L }; /* pause 50ms */
+ if (nanosleep (&ts, NULL) == -1)
+ {
+ perror ("nanosleep");
+ exit (EXIT_FAILURE);
+ }
+ }
}
}