date: make help text more understandable; small shrink
[oweals/busybox.git] / coreutils / date.c
index a922eb96c54c94bbc70e43bfe9ae13ce8b0bf167..064f758e25303916deebb38484781eaa540e3ea4 100644 (file)
@@ -5,22 +5,17 @@
  * by Matthew Grant <grantma@anathoth.gen.nz>
  *
  * iso-format handling added by Robert Griebl <griebl@gmx.de>
+ * bugfixes and cleanup by Bernhard Fischer
  *
  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
 */
 
-#include <stdlib.h>
-#include <errno.h>
-#include <unistd.h>
-#include <time.h>
-#include <stdio.h>
-#include <string.h>
-#include "busybox.h"
+#include "libbb.h"
 
 /* This 'date' command supports only 2 time setting formats,
    all the GNU strftime stuff (its in libc, lets use it),
-   setting time using UTC and displaying int, as well as
-   an RFC 822 complient date output for shell scripting
+   setting time using UTC and displaying it, as well as
+   an RFC 2822 compliant date output for shell scripting
    mail commands */
 
 /* Input parsing code is always bulky - used heavy duty libc stuff as
 
 /* Default input handling to save surprising some people */
 
-static struct tm *date_conv_time(struct tm *tm_time, const char *t_string)
-{
-       int nr;
-       char *cp;
-
-       nr = sscanf(t_string, "%2d%2d%2d%2d%d", &(tm_time->tm_mon),
-                               &(tm_time->tm_mday), &(tm_time->tm_hour), &(tm_time->tm_min),
-                               &(tm_time->tm_year));
-
-       if (nr < 4 || nr > 5) {
-               bb_error_msg_and_die(bb_msg_invalid_date, t_string);
-       }
-
-       cp = strchr(t_string, '.');
-       if (cp) {
-               nr = sscanf(cp + 1, "%2d", &(tm_time->tm_sec));
-               if (nr != 1) {
-                       bb_error_msg_and_die(bb_msg_invalid_date, t_string);
-               }
-       }
-
-       /* correct for century  - minor Y2K problem here? */
-       if (tm_time->tm_year >= 1900) {
-               tm_time->tm_year -= 1900;
-       }
-       /* adjust date */
-       tm_time->tm_mon -= 1;
-
-       return (tm_time);
-
-}
-
-
-/* The new stuff for LRP */
-
-static struct tm *date_conv_ftime(struct tm *tm_time, const char *t_string)
-{
-       struct tm t;
-
-       /* Parse input and assign appropriately to 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 */
-       } 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) {
-               /* Adjust dates from 1-12 to 0-11 */
-               t.tm_mon -= 1;
-       } 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) {
-               /* Adjust dates from 1-12 to 0-11 */
-               t.tm_mon -= 1;
-       } 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 {
-               bb_error_msg_and_die(bb_msg_invalid_date, t_string);
-       }
-       *tm_time = t;
-       return (tm_time);
-}
 
 #define DATE_OPT_RFC2822       0x01
 #define DATE_OPT_SET           0x02
@@ -111,76 +32,136 @@ static struct tm *date_conv_ftime(struct tm *tm_time, const char *t_string)
 #define DATE_OPT_TIMESPEC      0x20
 #define DATE_OPT_HINT          0x40
 
-int date_main(int argc, char **argv)
+static void maybe_set_utc(int opt)
 {
-       char *date_str = NULL;
-       char *date_fmt = NULL;
-       int set_time;
-       int utc;
-       time_t tm;
-       unsigned long opt;
-       struct tm tm_time;
-       char *filename = NULL;
-
-       int ifmt = 0;
-       char *isofmt_arg;
-       char *hintfmt_arg;
-
-       bb_opt_complementally = "?:d--s:s--d";
-       opt = bb_getopt_ulflags(argc, argv, "Rs:ud:r:"
-                                       USE_FEATURE_DATE_ISOFMT("I::D:"),
-                                       &date_str, &date_str, &filename
-                                       USE_FEATURE_DATE_ISOFMT(, &isofmt_arg, &hintfmt_arg));
-       set_time = opt & DATE_OPT_SET;
-       utc = opt & DATE_OPT_UTC;
-       if (utc && putenv("TZ=UTC0") != 0) {
-               bb_error_msg_and_die(bb_msg_memory_exhausted);
-       }
+       if (opt & DATE_OPT_UTC)
+               putenv((char*)"TZ=UTC0");
+}
 
-       if(ENABLE_FEATURE_DATE_ISOFMT && (opt & DATE_OPT_TIMESPEC)) {
-               if (!isofmt_arg) {
-                       ifmt = 1;
-               } else {
-                       char *isoformats[]={"date","hours","minutes","seconds"};
-                       for(ifmt = 4; ifmt;)
-                               if(!strcmp(isofmt_arg,isoformats[--ifmt]))
-                                       break;
-               }
-               if (!ifmt) {
-                       bb_show_usage();
+int date_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
+int date_main(int argc ATTRIBUTE_UNUSED, char **argv)
+{
+       struct tm tm_time;
+       time_t tm;
+       unsigned opt;
+       int ifmt = -1;
+       char *date_str;
+       char *fmt_dt2str;
+       char *fmt_str2dt;
+       char *filename;
+       char *isofmt_arg = NULL;
+
+       opt_complementary = "d--s:s--d"
+               USE_FEATURE_DATE_ISOFMT(":R--I:I--R");
+       opt = getopt32(argv, "Rs:ud:r:"
+                       USE_FEATURE_DATE_ISOFMT("I::D:"),
+                       &date_str, &date_str, &filename
+                       USE_FEATURE_DATE_ISOFMT(, &isofmt_arg, &fmt_str2dt));
+       argv += optind;
+       maybe_set_utc(opt);
+
+       if (ENABLE_FEATURE_DATE_ISOFMT && (opt & DATE_OPT_TIMESPEC)) {
+               ifmt = 0; /* default is date */
+               if (isofmt_arg) {
+                       static const char isoformats[] ALIGN1 =
+                               "date\0""hours\0""minutes\0""seconds\0";
+                       ifmt = index_in_strings(isoformats, isofmt_arg);
+                       if (ifmt < 0)
+                               bb_show_usage();
                }
        }
 
-       /* XXX, date_fmt == NULL from this always */
-       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];
+       fmt_dt2str = NULL;
+       if (argv[0] && argv[0][0] == '+') {
+               fmt_dt2str = &argv[0][1];       /* Skip over the '+' */
+               argv++;
+       }
+       if (!(opt & (DATE_OPT_SET | DATE_OPT_DATE))) {
+               opt |= DATE_OPT_SET;
+               date_str = argv[0]; /* can be NULL */
        }
 
        /* Now we have parsed all the information except the date format
           which depends on whether the clock is being set or read */
 
-       if(filename) {
+       if (opt & DATE_OPT_REFERENCE) {
                struct stat statbuf;
-               xstat(filename,&statbuf);
-               tm=statbuf.st_mtime;
-       } else time(&tm);
+               xstat(filename, &statbuf);
+               tm = statbuf.st_mtime;
+       } else
+               time(&tm);
        memcpy(&tm_time, localtime(&tm), sizeof(tm_time));
-       /* Zero out fields - take her back to midnight! */
+
+       /* If date string is given, update tm_time, and maybe set date */
        if (date_str != NULL) {
+               /* Zero out fields - take her back to midnight! */
                tm_time.tm_sec = 0;
                tm_time.tm_min = 0;
                tm_time.tm_hour = 0;
 
                /* Process any date input to UNIX time since 1 Jan 1970 */
                if (ENABLE_FEATURE_DATE_ISOFMT && (opt & DATE_OPT_HINT)) {
-                       strptime(date_str, hintfmt_arg, &tm_time);
+                       if (strptime(date_str, fmt_str2dt, &tm_time) == NULL)
+                               bb_error_msg_and_die(bb_msg_invalid_date, date_str);
                } else if (strchr(date_str, ':') != NULL) {
-                       date_conv_ftime(&tm_time, date_str);
+                       /* Parse input and assign appropriately to tm_time */
+
+                       if (sscanf(date_str, "%d:%d:%d", &tm_time.tm_hour, &tm_time.tm_min,
+                                                                &tm_time.tm_sec) == 3) {
+                               /* no adjustments needed */
+                       } else if (sscanf(date_str, "%d:%d", &tm_time.tm_hour,
+                                                                               &tm_time.tm_min) == 2) {
+                               /* no adjustments needed */
+                       } else if (sscanf(date_str, "%d.%d-%d:%d:%d", &tm_time.tm_mon,
+                                                               &tm_time.tm_mday, &tm_time.tm_hour,
+                                                               &tm_time.tm_min, &tm_time.tm_sec) == 5) {
+                               /* Adjust dates from 1-12 to 0-11 */
+                               tm_time.tm_mon -= 1;
+                       } else if (sscanf(date_str, "%d.%d-%d:%d", &tm_time.tm_mon,
+                                                               &tm_time.tm_mday,
+                                                               &tm_time.tm_hour, &tm_time.tm_min) == 4) {
+                               /* Adjust dates from 1-12 to 0-11 */
+                               tm_time.tm_mon -= 1;
+                       } else if (sscanf(date_str, "%d.%d.%d-%d:%d:%d", &tm_time.tm_year,
+                                                               &tm_time.tm_mon, &tm_time.tm_mday,
+                                                               &tm_time.tm_hour, &tm_time.tm_min,
+                                                                       &tm_time.tm_sec) == 6) {
+                               tm_time.tm_year -= 1900;        /* Adjust years */
+                               tm_time.tm_mon -= 1;    /* Adjust dates from 1-12 to 0-11 */
+                       } else if (sscanf(date_str, "%d.%d.%d-%d:%d", &tm_time.tm_year,
+                                                               &tm_time.tm_mon, &tm_time.tm_mday,
+                                                               &tm_time.tm_hour, &tm_time.tm_min) == 5) {
+                               tm_time.tm_year -= 1900;        /* Adjust years */
+                               tm_time.tm_mon -= 1;    /* Adjust dates from 1-12 to 0-11 */
+                       } else {
+                               bb_error_msg_and_die(bb_msg_invalid_date, date_str);
+                       }
                } else {
-                       date_conv_time(&tm_time, date_str);
+                       int nr;
+                       char *cp;
+
+                       nr = sscanf(date_str, "%2d%2d%2d%2d%d", &tm_time.tm_mon,
+                                               &tm_time.tm_mday, &tm_time.tm_hour, &tm_time.tm_min,
+                                               &tm_time.tm_year);
+
+                       if (nr < 4 || nr > 5) {
+                               bb_error_msg_and_die(bb_msg_invalid_date, date_str);
+                       }
+
+                       cp = strchr(date_str, '.');
+                       if (cp) {
+                               nr = sscanf(cp + 1, "%2d", &tm_time.tm_sec);
+                               if (nr != 1) {
+                                       bb_error_msg_and_die(bb_msg_invalid_date, date_str);
+                               }
+                       }
+
+                       /* correct for century  - minor Y2K problem here? */
+                       if (tm_time.tm_year >= 1900) {
+                               tm_time.tm_year -= 1900;
+                       }
+                       /* adjust date */
+                       tm_time.tm_mon -= 1;
                }
 
                /* Correct any day of week and day of year etc. fields */
@@ -189,12 +170,10 @@ int date_main(int argc, char **argv)
                if (tm < 0) {
                        bb_error_msg_and_die(bb_msg_invalid_date, date_str);
                }
-               if (utc && putenv("TZ=UTC0") != 0) {
-                       bb_error_msg_and_die(bb_msg_memory_exhausted);
-               }
+               maybe_set_utc(opt);
 
                /* if setting time, set it */
-               if (set_time && stime(&tm) < 0) {
+               if ((opt & DATE_OPT_SET) && stime(&tm) < 0) {
                        bb_perror_msg("cannot set date");
                }
        }
@@ -202,43 +181,55 @@ int date_main(int argc, char **argv)
        /* Display output */
 
        /* Deal with format string */
-       if (date_fmt == NULL) {
-               /* Start with the default case */
-               
-               date_fmt = (opt & DATE_OPT_RFC2822 ?
-                                       (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");
-
-               if (ENABLE_FEATURE_DATE_ISOFMT) {
-                       if (ifmt == 4)
-                               date_fmt = utc ? "%Y-%m-%dT%H:%M:%SZ" : "%Y-%m-%dT%H:%M:%S%z";
-                       else if (ifmt == 3)
-                               date_fmt = utc ? "%Y-%m-%dT%H:%MZ" : "%Y-%m-%dT%H:%M%z";
-                       else if (ifmt == 2) 
-                               date_fmt = utc ? "%Y-%m-%dT%HZ" : "%Y-%m-%dT%H%z";
-                       else if (ifmt == 1)
-                               date_fmt = "%Y-%m-%d";
-               }
+       if (fmt_dt2str == NULL) {
+               int i;
+               fmt_dt2str = xzalloc(32);
+               if (ENABLE_FEATURE_DATE_ISOFMT && ifmt >= 0) {
+                       strcpy(fmt_dt2str, "%Y-%m-%d");
+                       if (ifmt > 0) {
+                               i = 8;
+                               fmt_dt2str[i++] = 'T';
+                               fmt_dt2str[i++] = '%';
+                               fmt_dt2str[i++] = 'H';
+                               if (ifmt > 1) {
+                                       fmt_dt2str[i++] = ':';
+                                       fmt_dt2str[i++] = '%';
+                                       fmt_dt2str[i++] = 'M';
+                                       if (ifmt > 2) {
+                                               fmt_dt2str[i++] = ':';
+                                               fmt_dt2str[i++] = '%';
+                                               fmt_dt2str[i++] = 'S';
+                                       }
+                               }
+ format_utc:
+                               fmt_dt2str[i++] = '%';
+                               fmt_dt2str[i] = (opt & DATE_OPT_UTC) ? 'Z' : 'z';
+                       }
+               } else if (opt & DATE_OPT_RFC2822) {
+                       /* Undo busybox.c for date -R */
+                       if (ENABLE_LOCALE_SUPPORT)
+                               setlocale(LC_TIME, "C");
+                       strcpy(fmt_dt2str, "%a, %d %b %Y %H:%M:%S ");
+                       i = 22;
+                       goto format_utc;
+               } else /* default case */
+                       fmt_dt2str = (char*)"%a %b %e %H:%M:%S %Z %Y";
        }
-       
-       if (*date_fmt == '\0') {
 
+#define date_buf bb_common_bufsiz1
+       if (*fmt_dt2str == '\0') {
                /* With no format string, just print a blank line */
-               
-               *bb_common_bufsiz1=0;
+               date_buf[0] = '\0';
        } else {
-
                /* Handle special conversions */
-
-               if (strncmp(date_fmt, "%f", 2) == 0) {
-                       date_fmt = "%Y.%m.%d-%H:%M:%S";
+               if (strncmp(fmt_dt2str, "%f", 2) == 0) {
+                       fmt_dt2str = (char*)"%Y.%m.%d-%H:%M:%S";
                }
 
                /* Generate output string */
-               strftime(bb_common_bufsiz1, 200, date_fmt, &tm_time);
+               strftime(date_buf, sizeof(date_buf), fmt_dt2str, &tm_time);
        }
-       puts(bb_common_bufsiz1);
+       puts(date_buf);
 
        return EXIT_SUCCESS;
 }