Handle fifo read errors and ends of messages
[khatus.git] / x5 / khatus.c
index d95390a..93550a9 100644 (file)
 #include <X11/Xlib.h>
 
 #include "bsdtimespec.h"
+#include "khatus_lib_log.h"
+#include "khatus_lib_time.h"
 
-#define debug(args...) if (cfg->log_level >= Debug) {fprintf(stderr, "[debug] " args); fflush(stderr);}
-#define info( args...) if (cfg->log_level >= Info ) {fprintf(stderr, "[info] "  args); fflush(stderr);}
-#define error(args...) if (cfg->log_level >= Error) {fprintf(stderr, "[error] " args); fflush(stderr);}
-#define fatal(args...) {fprintf(stderr, "[fatal] " args); exit(EXIT_FAILURE);}
-#define usage(args...) {print_usage(); fatal("[usage] " args);}
-
+#define usage(...) {print_usage(); fprintf(stderr, "Error:\n    " __VA_ARGS__); exit(EXIT_FAILURE);}
 #define ERRMSG "ERROR"
 
 static const char errmsg[] = ERRMSG;
@@ -28,13 +25,6 @@ static const int  errlen   = sizeof(ERRMSG) - 1;
 
 char *argv0;
 
-typedef enum LogLevel {
-       Nothing,
-       Error,
-       Info,
-       Debug
-} LogLevel;
-
 /* TODO: Convert fifo list to fifo array. */
 typedef struct Fifo Fifo;
 struct Fifo {
@@ -43,58 +33,68 @@ struct Fifo {
        int     width;
        int     last_read;
        int     ttl;
-       int     pos;  /* Position on the output buffer. */
+       int     pos_init;  /* Initial position on the output buffer. */
+       int     pos_curr;  /* Current position on the output buffer. */
+       int     pos_final; /* Final   position on the output buffer. */
        Fifo   *next;
 };
 
 typedef struct Config Config;
 struct Config {
-       int    interval;
+       double interval;
        char * separator;
        Fifo * fifos;
        int    fifo_count;
        int    total_width;
        int    output_to_x_root_window;
-       LogLevel log_level;
 } defaults = {
-       .interval    = 1,
+       .interval    = 1.0,
        .separator   = "|",
        .fifos       = NULL,
        .fifo_count  = 0,
        .total_width = 0,
        .output_to_x_root_window = 0,
-       .log_level   = Info,
+};
+
+enum read_status {
+       FAIL_FINAL     = -2,
+       FAIL_TMP       = -1,
+       END_OF_FILE    =  0,
+       END_OF_MESSAGE =  1
 };
 
 void
-fifo_print_one(Fifo *f, Config *cfg)
+fifo_print_one(Fifo *f)
 {
-       info(
-               "Fifo "
-               "{"
-                       " name = %s,"
-                       " fd = %d,"
-                       " width = %d,"
-                       " last_read = %d,"
-                       " ttl = %d,"
-                       " pos = %d,"
-                       " next = %p,"
-               " }\n",
-               f->name,
-               f->fd,
-               f->width,
-               f->last_read,
-               f->ttl,
-               f->pos,
-               f->next
+       info("Fifo "
+           "{"
+           " name = %s,"
+           " fd = %d,"
+           " width = %d,"
+           " last_read = %d,"
+           " ttl = %d,"
+           " pos_init = %d,"
+           " pos_curr = %d,"
+           " pos_final = %d,"
+           " next = %p,"
+           " }\n",
+           f->name,
+           f->fd,
+           f->width,
+           f->last_read,
+           f->ttl,
+           f->pos_init,
+           f->pos_curr,
+           f->pos_final,
+           f->next
        );
 }
 
 void
-fifo_print_all(Fifo *head, Config *cfg)
+fifo_print_all(Fifo *head)
 {
        for (Fifo *f = head; f; f = f->next) {
-               fifo_print_one(f, cfg);
+               fifo_print_one(f);
        }
 }
 
@@ -102,22 +102,20 @@ void
 config_print(Config *cfg)
 {
        info(
-               "Config "
-               "{"
-                       " interval = %d,"
-                       " separator = %s,"
-                       " fifo_count = %d,"
-                       " total_width = %d,"
-                       " log_level = %d,"
-                       " fifos = ..."
-               " }\n",
-               cfg->interval,
-               cfg->separator,
-               cfg->fifo_count,
-               cfg->total_width,
-               cfg->log_level
+           "Config "
+           "{"
+           " interval = %f,"
+           " separator = %s,"
+           " fifo_count = %d,"
+           " total_width = %d,"
+           " fifos = ..."
+           " }\n",
+           cfg->interval,
+           cfg->separator,
+           cfg->fifo_count,
+           cfg->total_width
        );
-       fifo_print_all(cfg->fifos, cfg);
+       fifo_print_all(cfg->fifos);
 }
 
 int
@@ -129,36 +127,52 @@ is_pos_num(char *s)
        return 1;
 }
 
+int
+is_decimal(char *s)
+{
+       char c;
+       int seen = 0;
+
+       while ((c = *(s++)) != '\0')
+               if (!isdigit(c)) {
+                       if (c == '.' && !seen++)
+                               continue;
+                       else
+                               return 0;
+               }
+       return 1;
+}
+
 void
 print_usage()
 {
        assert(argv0);
        fprintf(
-               stderr,
-               "\n"
-               "Usage: %s [OPTION ...] SPEC [SPEC ...]\n"
-               "\n"
-               "  SPEC       = FILE_PATH DATA_WIDTH DATA_TTL\n"
-               "  FILE_PATH  = string\n"
-               "  DATA_WIDTH = int  (* (positive) number of characters *)\n"
-               "  DATA_TTL   = int  (* (positive) number of seconds *)\n"
-               "  OPTION     = -i INTERVAL\n"
-               "             | -s SEPARATOR\n"
-               "             | -x (* Output to X root window *)\n"
-               "             | -l LOG_LEVEL\n"
-               "  SEPARATOR  = string\n"
-               "  INTERVAL   = int  (* (positive) number of seconds *)\n"
-               "  LOG_LEVEL  = int  (* %d through %d *)\n"
-               "\n",
-               argv0,
-               Nothing,
-               Debug
+           stderr,
+           "\n"
+           "Usage: %s [OPTION ...] SPEC [SPEC ...]\n"
+           "\n"
+           "  SPEC       = FILE_PATH DATA_WIDTH DATA_TTL\n"
+           "  FILE_PATH  = string\n"
+           "  DATA_WIDTH = int  (* (positive) number of characters *)\n"
+           "  DATA_TTL   = int  (* (positive) number of seconds *)\n"
+           "  OPTION     = -i INTERVAL\n"
+           "             | -s SEPARATOR\n"
+           "             | -x (* Output to X root window *)\n"
+           "             | -l LOG_LEVEL\n"
+           "  SEPARATOR  = string\n"
+           "  INTERVAL   = int  (* (positive) number of seconds *)\n"
+           "  LOG_LEVEL  = int  (* %d through %d *)\n"
+           "\n",
+           argv0,
+           Nothing,
+           Debug
        );
        fprintf(
-               stderr,
-               "Example: %s -i 1 /dev/shm/khatus/khatus_sensor_x 4 10\n"
-               "\n",
-               argv0
+           stderr,
+           "Example: %s -i 1 /dev/shm/khatus/khatus_sensor_x 4 10\n"
+           "\n",
+           argv0
        );
 }
 
@@ -172,9 +186,9 @@ parse_opts_opt_i(Config *cfg, int argc, char *argv[], int i)
        if (i >= argc)
                usage("Option -i parameter is missing.\n");
        param = argv[i++];
-       if (!is_pos_num(param))
+       if (!is_decimal(param))
                usage("Option -i parameter is invalid: \"%s\"\n", param);
-       cfg->interval = atoi(param);
+       cfg->interval = atof(param);
        opts_parse_any(cfg, argc, argv, i);
 }
 
@@ -202,7 +216,7 @@ parse_opts_opt_l(Config *cfg, int argc, char *argv[], int i)
        log_level = atoi(param);
        if (log_level > Debug)
                usage("Option -l value (%d) exceeds maximum (%d)\n", log_level, Debug);
-       cfg->log_level = log_level;
+       _khatus_lib_log_level = log_level;
        opts_parse_any(cfg, argc, argv, i);
 }
 
@@ -252,7 +266,9 @@ parse_opts_spec(Config *cfg, int argc, char *argv[], int i)
                f->width     = atoi(w);
                f->ttl       = atoi(t);
                f->last_read = 0;
-               f->pos       = cfg->total_width;
+               f->pos_init  = cfg->total_width;
+               f->pos_curr  = f->pos_init;
+               f->pos_final = f->pos_init + f->width - 1;
                f->next      = cfg->fifos;
 
                cfg->fifos        = f;
@@ -299,7 +315,7 @@ fifo_read_error(Fifo *f, char *buf)
        char *b;
        int i;
 
-       b = buf + f->pos;
+       b = buf + f->pos_init;
        /* Copy as much of the error message as possible.
         * EXCLUDING the reminating \0. */
        for (i = 0; i < errlen && i < f->width; i++)
@@ -309,29 +325,48 @@ fifo_read_error(Fifo *f, char *buf)
                b[i] = '_';
 }
 
-void
-fifo_read_one(Fifo *f, char *buf, Config *cfg)
+enum read_status
+fifo_read_one(Fifo *f, char *buf)
 {
-       ssize_t current;
-       ssize_t total;
-       char *b;
-       char c;
+       /* Initialize all to an impossible value: */
+       ssize_t n = -5;  /* Number of bytes read. */
+       char    c = -1;  /* Character read. */
+       int     r = -1;  /* Remaining unused slots in buffer range. */
 
-       current = 0;
-       total = 0;
-       c = '\0';
-       b = buf + f->pos;
-       while ((current = read(f->fd, &c, 1)) && c != '\n' && c != '\0' && total++ < f->width)
-               *b++ = c;
-       if (current == -1) {
-               error("Failed to read: \"%s\". Error: %s\n", f->name, strerror(errno));
-               fifo_read_error(f, buf);
-       } else
-               while (total++ < f->width)
-                       *b++ = ' ';
+       for (;;) {
+               n = read(f->fd, &c, 1);
+               assert(n >= -1 && n <= 1);
+               switch (n) {
+               case -1:
+                       error("Failed to read: \"%s\". errno: %d, msg: %s\n",
+                           f->name, errno, strerror(errno));
+                       if (errno == 11)
+                               return FAIL_TMP;
+                       else
+                               return FAIL_FINAL;
+               case  0:
+                       debug("%s: End of FILE\n", f->name);
+                       f->pos_curr = f->pos_init;
+                       return END_OF_FILE;
+               case  1:
+                       /* TODO: Consider making msg term char a CLI option */
+                       if (c == '\n' || c == '\0') {
+                               r = f->pos_final - f->pos_curr;
+                               if (r > 0)
+                                       memset(buf + f->pos_curr, ' ', r);
+                               f->pos_curr = f->pos_init;
+                               return END_OF_MESSAGE;
+                       } else {
+                               if (f->pos_curr <= f->pos_final)
+                                       buf[f->pos_curr++] = c;
+                               /* Drop beyond available range. */
+                       }
+                       break;
+               default:
+                       assert(0);
+               }
+       }
        /* TODO Record timestamp read */
-       close(f->fd);
-       f->fd = -1;
 }
 
 void
@@ -355,15 +390,19 @@ fifo_read_all(Config *cfg, char *buf)
                        fifo_read_error(f, buf);
                        continue;
                }
-               debug("opening: %s\n", f->name);
-               if (f->fd < 0)
+               if (f->fd < 0) {
+                       debug("%s: closed. opening. fd: %d\n", f->name, f->fd);
                        f->fd = open(f->name, O_RDONLY | O_NONBLOCK);
+               } else {
+                       debug("%s: already openned. fd: %d\n", f->name, f->fd);
+               }
                if (f->fd == -1) {
                        /* TODO: Consider backing off retries for failed fifos. */
                        error("Failed to open \"%s\"\n", f->name);
                        fifo_read_error(f, buf);
                        continue;
                }
+               debug("%s: open. fd: %d\n", f->name, f->fd);
                if (f->fd > maxfd)
                        maxfd = f->fd;
                FD_SET(f->fd, &fds);
@@ -377,30 +416,18 @@ fifo_read_all(Config *cfg, char *buf)
        for (Fifo *f = cfg->fifos; f; f = f->next) {
                if (FD_ISSET(f->fd, &fds)) {
                        debug("reading: %s\n", f->name);
-                       fifo_read_one(f, buf, cfg);
-               }
-       }
-}
-
-void
-snooze(struct timespec *t, Config *cfg)
-{
-       struct timespec remainder;
-       int result;
-
-       result = nanosleep(t, &remainder);
-
-       if (result < 0) {
-               if (errno == EINTR) {
-                       info(
-                               "nanosleep interrupted. Remainder: "
-                               "{ tv_sec = %ld, tv_nsec = %ld }",
-                               remainder.tv_sec, remainder.tv_nsec);
-                       /* No big deal if we occasionally sleep less,
-                        * so not attempting to correct after an interruption.
-                        */
-               } else {
-                       fatal("nanosleep: %s\n", strerror(errno));
+                       switch (fifo_read_one(f, buf)) {
+                       case END_OF_FILE:
+                       case FAIL_FINAL:
+                               close(f->fd);
+                               f->fd = -1;
+                               break;
+                       case END_OF_MESSAGE:
+                       case FAIL_TMP:
+                               break;
+                       default:
+                               assert(0);
+                       }
                }
        }
 }
@@ -431,9 +458,7 @@ main(int argc, char *argv[])
        debug("argv0 = %s\n", argv0);
        config_print(cfg);
 
-       /* TODO: Support interval < 1. i.e. implement timespec_of_float */
-       ti.tv_sec  = cfg->interval;
-       ti.tv_nsec = 0;
+       ti = timespec_of_float(cfg->interval);
 
        if (cfg->fifos == NULL)
                usage("No fifo specs were given!\n");
@@ -459,7 +484,9 @@ main(int argc, char *argv[])
 
        /* 2nd pass to make space for separators */
        for (Fifo *f = cfg->fifos; f; f = f->next) {
-               f->pos += prefix;
+               f->pos_init  += prefix;
+               f->pos_final += prefix;
+               f->pos_curr = f->pos_init;
                prefix += seplen;
                nfifos++;
        }
@@ -471,9 +498,9 @@ main(int argc, char *argv[])
        buf[width] = '\0';
        /* 3rd pass to set the separators */
        for (Fifo *f = cfg->fifos; f; f = f->next) {
-               if (f->pos) {  /* Skip the first, left-most */
+               if (f->pos_init) {  /* Skip the first, left-most */
                        /* Copying only seplen ensures we omit the '\0' byte. */
-                       strncpy(buf + (f->pos - seplen), cfg->separator, seplen);
+                       strncpy(buf + (f->pos_init - seplen), cfg->separator, seplen);
                }
        }
 
@@ -506,9 +533,10 @@ main(int argc, char *argv[])
                         */
                        timespecsub(&ti, &td, &tc);
                        debug("snooze YES\n");
-                       snooze(&tc, cfg);
-               } else
+                       snooze(&tc);
+               } else {
                        debug("snooze NO\n");
+               }
        }
 
        return EXIT_SUCCESS;
This page took 0.050249 seconds and 4 git commands to generate.