Fix typo

[khatus.git] / x5 / khatus_sensor_time.c

diff --git a/x5/khatus_sensor_time.c b/x5/khatus_sensor_time.c
index f156f45..869134c 100644 (file)
--- a/x5/khatus_sensor_time.c
+++ b/x5/khatus_sensor_time.c
@@ -1,5 +1,8 @@
+#include <assert.h>
+#include <ctype.h>
 #include <errno.h>
-#include <stdarg.h>
+#include <fcntl.h>
+#include <signal.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
@@ -9,35 +12,35 @@
 #include "khatus_lib_log.h"
 #include "khatus_lib_time.h"
 
-char *argv0;
+#define usage(...) {print_usage(); fprintf(stderr, "Error:\n    " __VA_ARGS__); exit(EXIT_FAILURE);}
+
+#define MAX_LEN 20
+#define END_OF_MESSAGE '\n'
+
+char *argv0 = NULL;
+
+double opt_interval = 1.0;
+char *opt_fmt = "%a %b %d %H:%M:%S";
+char *fifo_name = NULL;
 
 void
-usage()
+print_usage()
 {
        printf(
-           "%s: [OPT ...]\n"
+           "%s: [OPT ...] FIFO\n"
            "\n"
+           "FIFO = string    # path to fifo file\n"
            "OPT = -i int     # interval\n"
            "    | -f string  # format string\n"
            "    | -h         # help message (i.e. what you're reading now :) )\n",
            argv0);
-       fatal("usage\n");
 }
 
-int
-main(int argc, char **argv)
+void
+opt_parse(int argc, char **argv)
 {
-       argv0 = argv[0];
-
-       double opt_interval = 1.0;
-       char *opt_fmt = "%a %b %d %H:%M:%S";
-
-       time_t t;
-       struct timespec ti;
-       char buf[128];
        char c;
 
-       memset(buf, '\0', 128);
        while ((c = getopt(argc, argv, "f:i:h")) != -1)
                switch (c) {
                case 'f':
@@ -48,17 +51,86 @@ main(int argc, char **argv)
                        opt_interval = atof(optarg);
                        break;
                case 'h':
-                       usage();
-                       break;
+                       print_usage();
+                       exit(EXIT_SUCCESS);
+               case '?':
+                       if (optopt == 'f' || optopt == 'i')
+                               fprintf(stderr,
+                                       "Option -%c requires an argument.\n",
+                                       optopt);
+                       else if (isprint(optopt))
+                               fprintf (stderr,
+                                       "Unknown option `-%c'.\n",
+                                       optopt);
+                       else
+                               fprintf(stderr,
+                                       "Unknown option character `\\x%x'.\n",
+                                       optopt);
+                       exit(EXIT_FAILURE);
                default:
-                       usage();
+                       assert(0);
                }
+       fifo_name = argv[optind];
+       debug("fifo_name: %s\n", fifo_name);
+       if (!fifo_name)
+               usage("No filename was provided\n");
+}
+
+int
+main(int argc, char **argv)
+{
+       argv0 = argv[0];
+
+       time_t t;
+       struct timespec ti;
+       char buf[MAX_LEN];
+
+       int fd = -1;
+
+       int n = 0;  /* written */
+       int r = 0;  /* remaining */
+       int i = 0;  /* buffer position */
+
+       opt_parse(argc, argv);
+
+       signal(SIGPIPE, SIG_IGN);  /* Handling manually */
+
+       memset(buf, '\0', MAX_LEN);
        ti = timespec_of_float(opt_interval);
        for (;;) {
+               debug("openning \"%s\"\n", fifo_name);
+               fd = open(fifo_name, O_WRONLY);
+               if (fd < 0)
+                       fatal("Failed to open FIFO file: \"%s\". Error: %s\n",
+                           fifo_name,
+                           strerror(errno));
+               debug("openned. fd: %d\n", fd);
                t = time(NULL);
-               strftime(buf, sizeof(buf), opt_fmt, localtime(&t));
-               puts(buf);
-               fflush(stdout);
+               strftime(buf, MAX_LEN, opt_fmt, localtime(&t));
+               r = strlen(buf);
+               buf[r] = END_OF_MESSAGE;
+               for (i = 0; (n = write(fd, buf + i++, 1)) && r; r--)
+                       ;
+               if (n < 0)
+                       switch (errno) {
+                       case EPIPE:
+                               error("Broken pipe. Msg buf: %s\n", buf);
+                               break;
+                       default:
+                               fatal(
+                                   "Failed to write to %s. "
+                                   "Err num: %d, Err msg: %s\n",
+                                   fifo_name,
+                                   errno,
+                                   strerror(errno));
+                       }
+               if (close(fd) < 0)
+                       fatal("Failed to close %s. Err num: %d, Err msg: %s\n",
+                           fifo_name,
+                           errno,
+                           strerror(errno));
+               fd = -1;
+               debug("closed. fd: %d\n", fd);
                snooze(&ti);
        }
        return EXIT_SUCCESS;