diff options
Diffstat (limited to '')
| -rw-r--r-- | mongotimer.c | 135 |
1 files changed, 126 insertions, 9 deletions
diff --git a/mongotimer.c b/mongotimer.c index 692d7a2..f65238f 100644 --- a/mongotimer.c +++ b/mongotimer.c @@ -1,6 +1,7 @@ /* See LICENSE file for copyright and license details. */ #include <sys/ioctl.h> #include <sys/timerfd.h> +#include <fcntl.h> #include <errno.h> #include <signal.h> #include <stdint.h> @@ -37,6 +38,7 @@ static const char **mongo_ds[] = { static volatile sig_atomic_t caught_sigterm = 0; static volatile sig_atomic_t caught_sigwinch = 1; +static volatile sig_atomic_t caught_sigio = 0; char *argv0; @@ -62,6 +64,13 @@ sigwinch(int signo) } static void +sigio(int signo) +{ + caught_sigio = 1; + (void) signo; +} + +static void print_time(const char ***str, size_t y, size_t x) { size_t r, c; @@ -88,9 +97,12 @@ display_stopwatch(int timerfd) const char **digits[23]; uint64_t h, m, s, h_div = 1, th; size_t h_digits = 0; - int small = 0; + int small = 0, paused = 0; + struct itimerspec old_time, zero_time; + ssize_t r; + char c; - /* TODO pause (yellow text) on <space> */ + memset(&zero_time, 0, sizeof(zero_time)); while (!caught_sigterm) { if (caught_sigwinch) { @@ -116,6 +128,31 @@ display_stopwatch(int timerfd) y /= 2; x /= 2; } + if (caught_sigio) { + caught_sigio = 0; + for (;;) { + r = read(STDIN_FILENO, &c, 1); + if (r <= 0) { + if (!r) + goto out; + if (errno == EAGAIN) + break; + if (errno == EINTR) + continue; + goto fail; + } + if (c == ' ' || c == '\n') { + paused ^= 1; + if (paused) { + if (timerfd_settime(timerfd, 0, &zero_time, &old_time)) + goto fail; + } else { + if (timerfd_settime(timerfd, 0, &old_time, NULL)) + goto fail; + } + } + } + } if (small) { fprintf(stdout, "\033[H\033[2J%s\n", "Screen is too small"); @@ -146,7 +183,15 @@ display_stopwatch(int timerfd) digits[i++] = mongo_ds[s % 10]; digits[i++] = NULL; - print_time(digits, y, x); + if (paused) { + fprintf(stdout, "\033[33m\n"); + print_time(digits, y, x); + fprintf(stdout, "\033[39m\n"); + pause(); + continue; + } else { + print_time(digits, y, x); + } if (read(timerfd, &overrun, sizeof(overrun)) < 0) { if (errno != EINTR) @@ -156,6 +201,7 @@ display_stopwatch(int timerfd) } } +out: return 0; fail: @@ -171,9 +217,12 @@ display_timer(int timerfd, int64_t time, int exit_on_zero) const char **digits[24]; uint64_t h, m, s, th, h_div; size_t h_digits, i; - int small = 0; + int small = 0, paused = 0; + struct itimerspec old_time, zero_time; + ssize_t r; + char c; - /* TODO pause (yellow text) on <space> */ + memset(&zero_time, 0, sizeof(zero_time)); while (!caught_sigterm) { if (caught_sigwinch) { @@ -184,6 +233,31 @@ display_timer(int timerfd, int64_t time, int exit_on_zero) } caught_sigwinch = 0; } + if (caught_sigio) { + caught_sigio = 0; + for (;;) { + r = read(STDIN_FILENO, &c, 1); + if (r <= 0) { + if (!r) + goto out; + if (errno == EAGAIN) + break; + if (errno == EINTR) + continue; + goto fail; + } + if (c == ' ' || c == '\n') { + paused ^= 1; + if (paused) { + if (timerfd_settime(timerfd, 0, &zero_time, &old_time)) + goto fail; + } else { + if (timerfd_settime(timerfd, 0, &old_time, NULL)) + goto fail; + } + } + } + } abstime = time < 0 ? (uint64_t)-time : (uint64_t)time; s = abstime % 60; @@ -228,7 +302,13 @@ display_timer(int timerfd, int64_t time, int exit_on_zero) digits[i++] = mongo_ds[s % 10]; digits[i++] = NULL; - if (time >= 0) { + if (paused) { + fprintf(stdout, "\033[33m\n"); + print_time(digits, y, x); + fprintf(stdout, "\033[39m\n"); + pause(); + continue; + } else if (time >= 0) { print_time(digits, y, x); } else { fprintf(stdout, "\033[31m\n"); @@ -246,6 +326,7 @@ display_timer(int timerfd, int64_t time, int exit_on_zero) } } +out: if (caught_sigterm && exit_on_zero) return -1; @@ -258,13 +339,14 @@ fail: int main(int argc, char *argv[]) { - int timerfd = -1; - int exit_on_zero = 0; + int timerfd = -1, old_flags = -1; + int exit_on_zero = 0, old_sig = 0; struct itimerspec itimerspec; struct sigaction sigact; - int64_t time = 0, t = 0; + int64_t time = 0, t = 0, owner_set = 0; size_t colons = 0; char *s; + struct f_owner_ex old_owner, new_owner; ARGBEGIN { case 'e': @@ -299,6 +381,26 @@ main(int argc, char *argv[]) sigact.sa_handler = sigwinch; sigaction(SIGWINCH, &sigact, NULL); + sigact.sa_handler = sigio; + sigaction(SIGIO, &sigact, NULL); + sigaction(SIGURG, &sigact, NULL); + + if (fcntl(STDIN_FILENO, F_GETOWN_EX, &old_owner)) + goto fail; + memset(&new_owner, 0, sizeof(new_owner)); + new_owner.type = F_OWNER_PID; + new_owner.pid = getpid(); + if (fcntl(STDIN_FILENO, F_SETOWN_EX, &new_owner)) + goto fail; + owner_set = 1; + old_flags = fcntl(STDIN_FILENO, F_GETFL); + fcntl(STDIN_FILENO, F_SETFL, old_flags | FASYNC | O_NONBLOCK); + fcntl(STDIN_FILENO, F_GETSIG, &old_sig); + if (old_sig) + fcntl(STDIN_FILENO, F_SETSIG, 0); + + /* TODO set ICANON and ECHO and reset on exit */ + if (argc) { for (s = argv[0]; *s; s++) { if ('0' <= *s && *s <= '9') { @@ -324,6 +426,9 @@ main(int argc, char *argv[]) fprintf(stdout, "\033[?25h\n\033[?1049l"); fflush(stdout); + fcntl(STDIN_FILENO, F_SETOWN_EX, &old_owner); + fcntl(STDIN_FILENO, F_SETFL, old_flags); + fcntl(STDIN_FILENO, F_SETSIG, old_sig); close(timerfd); return 0; @@ -331,6 +436,12 @@ fail: perror(argv0 ? argv0 : "mongotimer"); fprintf(stdout, "\033[?25h\n\033[?1049l"); fflush(stdout); + if (owner_set) + fcntl(STDIN_FILENO, F_SETOWN_EX, &old_owner); + if (old_flags != -1) + fcntl(STDIN_FILENO, F_SETFL, old_flags); + if (old_sig) + fcntl(STDIN_FILENO, F_SETSIG, old_sig); if (timerfd >= 0) close(timerfd); return 1; @@ -338,5 +449,11 @@ fail: fail_usage: fprintf(stdout, "\033[?25h\n\033[?1049l"); fflush(stdout); + if (owner_set) + fcntl(STDIN_FILENO, F_SETOWN_EX, &old_owner); + if (old_flags != -1) + fcntl(STDIN_FILENO, F_SETFL, old_flags); + if (old_sig) + fcntl(STDIN_FILENO, F_SETSIG, old_sig); usage(); } |