- updated a few copyright/left file headers
[oweals/busybox.git] / coreutils / date.c
index b4c3e71535518997677834ec17326b9522c58d29..a9188a1ce1c72da2680c35ab568eda3a2b3d24b2 100644 (file)
@@ -3,6 +3,8 @@
  * Mini date implementation for busybox
  *
  * by Matthew Grant <grantma@anathoth.gen.nz>
+ * 
+ * iso-format handling added by Robert Griebl <griebl@gmx.de>
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
  *
 */
 
-#include "internal.h"
-#define BB_DECLARE_EXTERN
-#define bb_need_invalid_date
-#define bb_need_memory_exhausted
-#include "messages.c"
 #include <stdlib.h>
 #include <errno.h>
 #include <sys/time.h>
 #include <unistd.h>
 #include <time.h>
 #include <stdio.h>
+#include <string.h>
+#include <getopt.h>
+#include "busybox.h"
 
 
 /* This 'date' command supports only 2 time setting formats, 
    an RFC 822 complient date output for shell scripting
    mail commands */
 
-static const char date_usage[] = "date [OPTION]... [+FORMAT]\n"
-       "  or:  date [OPTION] [MMDDhhmm[[CC]YY][.ss]]\n\n"
-       "Display the current time in the given FORMAT, or set the system date.\n"
-       "\nOptions:\n\t-R\t\toutput RFC-822 compliant date string\n"
-       "\t-s\t\tset time described by STRING\n"
-
-       "\t-u\t\tprint or set Coordinated Universal Time\n";
-
-
 /* Input parsing code is always bulky - used heavy duty libc stuff as
    much as possible, missed out a lot of bounds checking */
 
 /* Default input handling to save suprising some people */
 
-struct tm *date_conv_time(struct tm *tm_time, const char *t_string)
+static struct tm *date_conv_time(struct tm *tm_time, const char *t_string)
 {
        int nr;
 
@@ -64,8 +55,7 @@ struct tm *date_conv_time(struct tm *tm_time, const char *t_string)
                                &(tm_time->tm_min), &(tm_time->tm_year));
 
        if (nr < 4 || nr > 5) {
-               fprintf(stderr, invalid_date, "date", t_string);
-               exit(FALSE);
+               error_msg_and_die(invalid_date, t_string); 
        }
 
        /* correct for century  - minor Y2K problem here? */
@@ -81,78 +71,59 @@ struct tm *date_conv_time(struct tm *tm_time, const char *t_string)
 
 /* The new stuff for LRP */
 
-struct tm *date_conv_ftime(struct tm *tm_time, const char *t_string)
+static struct tm *date_conv_ftime(struct tm *tm_time, const char *t_string)
 {
-       struct tm itm_time, jtm_time, ktm_time, ltm_time, mtm_time, ntm_time;
-
-       itm_time = *tm_time;
-       jtm_time = *tm_time;
-       ktm_time = *tm_time;
-       ltm_time = *tm_time;
-       mtm_time = *tm_time;
-       ntm_time = *tm_time;
+       struct tm t;
 
        /* Parse input and assign appropriately to tm_time */
 
-       if (sscanf(t_string, "%d:%d:%d",
-                          &itm_time.tm_hour, &itm_time.tm_min, &itm_time.tm_sec) == 3) {
-
-               *tm_time = itm_time;
-               return (tm_time);
-
-       } else if (sscanf(t_string, "%d:%d",
-                                         &jtm_time.tm_hour, &jtm_time.tm_min) == 2) {
-
-               *tm_time = jtm_time;
-               return (tm_time);
-
-       } else if (sscanf(t_string, "%d.%d-%d:%d:%d",
-                                         &ktm_time.tm_mon,
-                                         &ktm_time.tm_mday,
-                                         &ktm_time.tm_hour,
-                                         &ktm_time.tm_min, &ktm_time.tm_sec) == 5) {
-
-               ktm_time.tm_mon -= 1;   /* Adjust dates from 1-12 to 0-11 */
-               *tm_time = ktm_time;
-               return (tm_time);
-
-       } else if (sscanf(t_string, "%d.%d-%d:%d",
-                                         &ltm_time.tm_mon,
-                                         &ltm_time.tm_mday,
-                                         &ltm_time.tm_hour, &ltm_time.tm_min) == 4) {
-
-               ltm_time.tm_mon -= 1;   /* Adjust dates from 1-12 to 0-11 */
-               *tm_time = ltm_time;
-               return (tm_time);
-
-       } else if (sscanf(t_string, "%d.%d.%d-%d:%d:%d",
-                                         &mtm_time.tm_year,
-                                         &mtm_time.tm_mon,
-                                         &mtm_time.tm_mday,
-                                         &mtm_time.tm_hour,
-                                         &mtm_time.tm_min, &mtm_time.tm_sec) == 6) {
-
-               mtm_time.tm_year -= 1900;       /* Adjust years */
-               mtm_time.tm_mon -= 1;   /* Adjust dates from 1-12 to 0-11 */
-               *tm_time = mtm_time;
-               return (tm_time);
-
-       } else if (sscanf(t_string, "%d.%d.%d-%d:%d",
-                                         &ntm_time.tm_year,
-                                         &ntm_time.tm_mon,
-                                         &ntm_time.tm_mday,
-                                         &ntm_time.tm_hour, &ntm_time.tm_min) == 5) {
-               ntm_time.tm_year -= 1900;       /* Adjust years */
-               ntm_time.tm_mon -= 1;   /* Adjust dates from 1-12 to 0-11 */
-               *tm_time = ntm_time;
-               return (tm_time);
+       if (t=*tm_time,sscanf(t_string, "%d:%d:%d",
+                          &t.tm_hour, &t.tm_min, &t.tm_sec) == 3) {
+                                       /* no adjustments needed */
+
+       } else if (t=*tm_time,sscanf(t_string, "%d:%d",
+                                         &t.tm_hour, &t.tm_min) == 2) {
+                                       /* no adjustments needed */
 
-       }
 
-       fprintf(stderr, invalid_date, "date", t_string);
+       } else if (t=*tm_time,sscanf(t_string, "%d.%d-%d:%d:%d",
+                                         &t.tm_mon,
+                                         &t.tm_mday,
+                                         &t.tm_hour,
+                                         &t.tm_min, &t.tm_sec) == 5) {
 
-       exit(FALSE);
+               t.tm_mon -= 1;  /* Adjust dates from 1-12 to 0-11 */
 
+       } else if (t=*tm_time,sscanf(t_string, "%d.%d-%d:%d",
+                                         &t.tm_mon,
+                                         &t.tm_mday,
+                                         &t.tm_hour, &t.tm_min) == 4) {
+
+               t.tm_mon -= 1;  /* Adjust dates from 1-12 to 0-11 */
+
+       } else if (t=*tm_time,sscanf(t_string, "%d.%d.%d-%d:%d:%d",
+                                         &t.tm_year,
+                                         &t.tm_mon,
+                                         &t.tm_mday,
+                                         &t.tm_hour,
+                                         &t.tm_min, &t.tm_sec) == 6) {
+
+               t.tm_year -= 1900;      /* Adjust years */
+               t.tm_mon -= 1;  /* Adjust dates from 1-12 to 0-11 */
+
+       } else if (t=*tm_time,sscanf(t_string, "%d.%d.%d-%d:%d",
+                                         &t.tm_year,
+                                         &t.tm_mon,
+                                         &t.tm_mday,
+                                         &t.tm_hour, &t.tm_min) == 5) {
+               t.tm_year -= 1900;      /* Adjust years */
+               t.tm_mon -= 1;  /* Adjust dates from 1-12 to 0-11 */
+
+       } else {
+               error_msg_and_die(invalid_date, t_string); 
+       }
+       *tm_time = t;
+       return (tm_time);
 }
 
 
@@ -161,7 +132,7 @@ int date_main(int argc, char **argv)
        char *date_str = NULL;
        char *date_fmt = NULL;
        char *t_buff;
-       int i;
+       int c;
        int set_time = 0;
        int rfc822 = 0;
        int utc = 0;
@@ -169,53 +140,73 @@ int date_main(int argc, char **argv)
        time_t tm;
        struct tm tm_time;
 
+#ifdef CONFIG_FEATURE_DATE_ISOFMT
+       int ifmt = 0;
+#define GETOPT_ISOFMT  "I::"
+#else
+#define GETOPT_ISOFMT
+#endif
+
        /* Interpret command line args */
-       i = --argc;
-       argv++;
-       while (i > 0 && **argv) {
-               if (**argv == '-') {
-                       while (i > 0 && *++(*argv))
-                               switch (**argv) {
-                               case 'R':
-                                       rfc822 = 1;
-                                       break;
-                               case 's':
-                                       set_time = 1;
-                                       if (date_str != NULL)
-                                               usage(date_usage);
-                                       date_str = optarg;
-                                       break;
-                               case 'u':
-                                       utc = 1;
-                                       if (putenv("TZ=UTC0") != 0) {
-                                               fprintf(stderr, memory_exhausted, "date");
-                                               exit(FALSE);
-                                       }
-                                       /* Look ma, no break.  Don't fix it either. */
-                               case 'd':
-                                       use_arg = 1;
-                                       if (date_str != NULL)
-                                               usage(date_usage);
-                                       date_str = optarg;
-                                       break;
-                               case '-':
-                                       usage(date_usage);
-                               }
-               } else {
-                       if ((date_fmt == NULL) && (strcmp(*argv, "+") == 0))
-                               date_fmt = *argv;
-                       else if (date_str == NULL) {
+       while ((c = getopt(argc, argv, "Rs:ud:" GETOPT_ISOFMT )) != EOF) {
+               switch (c) {
+                       case 'R':
+                               rfc822 = 1;
+                               break;
+                       case 's':
                                set_time = 1;
-                               date_str = *argv;
-                       } else {
-                               usage(date_usage);
-                       }
+                               if ((date_str != NULL) || ((date_str = optarg) == NULL)) {
+                                       show_usage();
+                               }
+                               break;
+                       case 'u':
+                               utc = 1;
+                               if (putenv("TZ=UTC0") != 0)
+                                       error_msg_and_die(memory_exhausted);
+                               break;
+                       case 'd':
+                               use_arg = 1;
+                               if ((date_str != NULL) || ((date_str = optarg) == NULL))
+                                       show_usage();
+                               break;
+#ifdef CONFIG_FEATURE_DATE_ISOFMT
+                       case 'I': 
+                               if ( !optarg ) 
+                                       ifmt = 1;
+                               else {
+                                       int ifmt_len = xstrlen ( optarg );
+                       
+                                       if (( ifmt_len <= 4 ) && ( strncmp ( optarg, "date", ifmt_len ) == 0 ))
+                                               ifmt = 1;
+                                       else if (( ifmt_len <= 5 ) && ( strncmp ( optarg, "hours", ifmt_len ) == 0 ))
+                                               ifmt = 2;
+                                       else if (( ifmt_len <= 7 ) && ( strncmp ( optarg, "minutes", ifmt_len ) == 0 ))
+                                               ifmt = 3;
+                                       else if (( ifmt_len <= 7 ) && ( strncmp ( optarg, "seconds", ifmt_len ) == 0 ))
+                                               ifmt = 4;
+                               }                        
+                               if ( ifmt )
+                                       break; // else show_usage();
+#endif
+                       default:
+                               show_usage();
                }
-               i--;
-               argv++;
        }
 
 
+       if ((date_fmt == NULL) && (optind < argc) && (argv[optind][0] == '+'))
+               date_fmt = &argv[optind][1];   /* Skip over the '+' */
+       else if (date_str == NULL) {
+               set_time = 1;
+               date_str = argv[optind];
+       } 
+#if 0
+       else {
+               error_msg("date_str='%s'  date_fmt='%s'\n", date_str, date_fmt);
+               show_usage();
+       }
+#endif
+
        /* Now we have parsed all the information except the date format
           which depends on whether the clock is being set or read */
 
@@ -237,18 +228,19 @@ int date_main(int argc, char **argv)
                        date_conv_time(&tm_time, date_str);
                }
 
-               /* Correct any day of week and day of year etc fields */
+               /* Correct any day of week and day of year etc. fields */
                tm = mktime(&tm_time);
-               if (tm < 0) {
-                       fprintf(stderr, invalid_date, "date", date_str);
-                       exit(FALSE);
+               if (tm < 0)
+                       error_msg_and_die(invalid_date, date_str); 
+               if ( utc ) {
+                       if (putenv("TZ=UTC0") != 0)
+                               error_msg_and_die(memory_exhausted);
                }
 
                /* if setting time, set it */
                if (set_time) {
                        if (stime(&tm) < 0) {
-                               fprintf(stderr, "date: can't set date.\n");
-                               exit(FALSE);
+                               perror_msg("cannot set date");
                        }
                }
        }
@@ -257,16 +249,37 @@ int date_main(int argc, char **argv)
 
        /* Deal with format string */
        if (date_fmt == NULL) {
-               date_fmt = (rfc822
-                                       ? (utc
-                                          ? "%a, %_d %b %Y %H:%M:%S GMT"
-                                          : "%a, %_d %b %Y %H:%M:%S %z")
-                                       : "%a %b %e %H:%M:%S %Z %Y");
-
+#ifdef CONFIG_FEATURE_DATE_ISOFMT
+               switch ( ifmt ) {
+                       case  4: 
+                               date_fmt = utc ? "%Y-%m-%dT%H:%M:%SZ" : "%Y-%m-%dT%H:%M:%S%z";
+                               break;
+                       case  3: 
+                               date_fmt = utc ? "%Y-%m-%dT%H:%MZ" : "%Y-%m-%dT%H:%M%z";
+                               break;
+                       case  2: 
+                               date_fmt = utc ? "%Y-%m-%dT%HZ" : "%Y-%m-%dT%H%z";
+                               break;
+                       case  1: 
+                               date_fmt = "%Y-%m-%d";  
+                               break;
+                       case  0:
+                       default: 
+#endif
+                               date_fmt = (rfc822
+                                           ? (utc
+                                              ? "%a, %e %b %Y %H:%M:%S GMT"
+                                              : "%a, %e %b %Y %H:%M:%S %z")
+                                           : "%a %b %e %H:%M:%S %Z %Y");
+                                           
+#ifdef CONFIG_FEATURE_DATE_ISOFMT
+                               break;
+               }
+#endif
        } else if (*date_fmt == '\0') {
                /* Imitate what GNU 'date' does with NO format string! */
                printf("\n");
-               exit(TRUE);
+               return EXIT_SUCCESS;
        }
 
        /* Handle special conversions */
@@ -276,10 +289,9 @@ int date_main(int argc, char **argv)
        }
 
        /* Print OUTPUT (after ALL that!) */
-       t_buff = malloc(201);
+       t_buff = xmalloc(201);
        strftime(t_buff, 200, date_fmt, &tm_time);
-       printf("%s\n", t_buff);
-
-       exit(TRUE);
+       puts(t_buff);
 
+       return EXIT_SUCCESS;
 }