coreutils/date.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini date implementation for busybox
   4  *
   5  * by Matthew Grant <grantma@anathoth.gen.nz>
   6  *
   7  * iso-format handling added by Robert Griebl <griebl@gmx.de>
   8  * bugfixes and cleanup by Bernhard Reutner-Fischer
   9  *
  10  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  11 */
  12
  13 /* This 'date' command supports only 2 time setting formats,
  14    all the GNU strftime stuff (its in libc, lets use it),
  15    setting time using UTC and displaying it, as well as
  16    an RFC 2822 compliant date output for shell scripting
  17    mail commands */
  18
  19 /* Input parsing code is always bulky - used heavy duty libc stuff as
  20    much as possible, missed out a lot of bounds checking */
  21
  22 /* Default input handling to save surprising some people */
  23
  24 /* GNU coreutils 6.9 man page:
  25  * date [OPTION]... [+FORMAT]
  26  * date [-u|--utc|--universal] [MMDDhhmm[[CC]YY][.ss]]
  27  * -d, --date=STRING
  28  *      display time described by STRING, not `now'
  29  * -f, --file=DATEFILE
  30  *      like --date once for each line of DATEFILE
  31  * -r, --reference=FILE
  32  *      display the last modification time of FILE
  33  * -R, --rfc-2822
  34  *      output date and time in RFC 2822 format.
  35  *      Example: Mon, 07 Aug 2006 12:34:56 -0600
  36  * --rfc-3339=TIMESPEC
  37  *      output date and time in RFC 3339 format.
  38  *      TIMESPEC='date', 'seconds', or 'ns'
  39  *      Date and time components are separated by a single space:
  40  *      2006-08-07 12:34:56-06:00
  41  * -s, --set=STRING
  42  *      set time described by STRING
  43  * -u, --utc, --universal
  44  *      print or set Coordinated Universal Time
  45  *
  46  * Busybox:
  47  * long options are not supported
  48  * -f is not supported
  49  * -I seems to roughly match --rfc-3339, but -I has _optional_ param
  50  *    (thus "-I seconds" doesn't work, only "-Iseconds"),
  51  *    and does not support -Ins
  52  * -D FMT is a bbox extension for _input_ conversion of -d DATE
  53  */
  54 #include "libbb.h"
  55
  56 enum {
  57         OPT_RFC2822   = (1 << 0), /* R */
  58         OPT_SET       = (1 << 1), /* s */
  59         OPT_UTC       = (1 << 2), /* u */
  60         OPT_DATE      = (1 << 3), /* d */
  61         OPT_REFERENCE = (1 << 4), /* r */
  62         OPT_TIMESPEC  = (1 << 5) * ENABLE_FEATURE_DATE_ISOFMT, /* I */
  63         OPT_HINT      = (1 << 6) * ENABLE_FEATURE_DATE_ISOFMT, /* D */
  64 };
  65
  66 static void maybe_set_utc(int opt)
  67 {
  68         if (opt & OPT_UTC)
  69                 putenv((char*)"TZ=UTC0");
  70 }
  71
  72 int date_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
  73 int date_main(int argc UNUSED_PARAM, char **argv)
  74 {
  75         struct tm tm_time;
  76         time_t tm;
  77         unsigned opt;
  78         int ifmt = -1;
  79         char *date_str;
  80         char *fmt_dt2str;
  81         char *fmt_str2dt;
  82         char *filename;
  83         char *isofmt_arg = NULL;
  84
  85         opt_complementary = "d--s:s--d"
  86                 IF_FEATURE_DATE_ISOFMT(":R--I:I--R");
  87         opt = getopt32(argv, "Rs:ud:r:"
  88                         IF_FEATURE_DATE_ISOFMT("I::D:"),
  89                         &date_str, &date_str, &filename
  90                         IF_FEATURE_DATE_ISOFMT(, &isofmt_arg, &fmt_str2dt));
  91         argv += optind;
  92         maybe_set_utc(opt);
  93
  94         if (ENABLE_FEATURE_DATE_ISOFMT && (opt & OPT_TIMESPEC)) {
  95                 ifmt = 0; /* default is date */
  96                 if (isofmt_arg) {
  97                         static const char isoformats[] ALIGN1 =
  98                                 "date\0""hours\0""minutes\0""seconds\0";
  99                         ifmt = index_in_strings(isoformats, isofmt_arg);
 100                         if (ifmt < 0)
 101                                 bb_show_usage();
 102                 }
 103         }
 104
 105         fmt_dt2str = NULL;
 106         if (argv[0] && argv[0][0] == '+') {
 107                 fmt_dt2str = &argv[0][1];       /* Skip over the '+' */
 108                 argv++;
 109         }
 110         if (!(opt & (OPT_SET | OPT_DATE))) {
 111                 opt |= OPT_SET;
 112                 date_str = argv[0]; /* can be NULL */
 113                 if (date_str)
 114                         argv++;
 115         }
 116         if (*argv)
 117                 bb_show_usage();
 118
 119         /* Now we have parsed all the information except the date format
 120            which depends on whether the clock is being set or read */
 121
 122         if (opt & OPT_REFERENCE) {
 123                 struct stat statbuf;
 124                 xstat(filename, &statbuf);
 125                 tm = statbuf.st_mtime;
 126         } else {
 127                 time(&tm);
 128         }
 129         localtime_r(&tm, &tm_time);
 130
 131         /* If date string is given, update tm_time, and maybe set date */
 132         if (date_str != NULL) {
 133                 /* Zero out fields - take her back to midnight! */
 134                 tm_time.tm_sec = 0;
 135                 tm_time.tm_min = 0;
 136                 tm_time.tm_hour = 0;
 137
 138                 /* Process any date input to UNIX time since 1 Jan 1970 */
 139                 if (ENABLE_FEATURE_DATE_ISOFMT && (opt & OPT_HINT)) {
 140                         if (strptime(date_str, fmt_str2dt, &tm_time) == NULL)
 141                                 bb_error_msg_and_die(bb_msg_invalid_date, date_str);
 142                 } else {
 143                         parse_datestr(date_str, &tm_time);
 144                 }
 145
 146                 /* Correct any day of week and day of year etc. fields */
 147                 tm_time.tm_isdst = -1;  /* Be sure to recheck dst */
 148                 tm = validate_tm_time(date_str, &tm_time);
 149
 150                 maybe_set_utc(opt);
 151
 152                 /* if setting time, set it */
 153                 if ((opt & OPT_SET) && stime(&tm) < 0) {
 154                         bb_perror_msg("cannot set date");
 155                 }
 156         }
 157
 158         /* Display output */
 159
 160         /* Deal with format string */
 161         if (fmt_dt2str == NULL) {
 162                 int i;
 163                 fmt_dt2str = xzalloc(32);
 164                 if (ENABLE_FEATURE_DATE_ISOFMT && ifmt >= 0) {
 165                         strcpy(fmt_dt2str, "%Y-%m-%d");
 166                         if (ifmt > 0) {
 167                                 i = 8;
 168                                 fmt_dt2str[i++] = 'T';
 169                                 fmt_dt2str[i++] = '%';
 170                                 fmt_dt2str[i++] = 'H';
 171                                 if (ifmt > 1) {
 172                                         fmt_dt2str[i++] = ':';
 173                                         fmt_dt2str[i++] = '%';
 174                                         fmt_dt2str[i++] = 'M';
 175                                         if (ifmt > 2) {
 176                                                 fmt_dt2str[i++] = ':';
 177                                                 fmt_dt2str[i++] = '%';
 178                                                 fmt_dt2str[i++] = 'S';
 179                                         }
 180                                 }
 181  format_utc:
 182                                 fmt_dt2str[i++] = '%';
 183                                 fmt_dt2str[i] = (opt & OPT_UTC) ? 'Z' : 'z';
 184                         }
 185                 } else if (opt & OPT_RFC2822) {
 186                         /* Undo busybox.c for date -R */
 187                         if (ENABLE_LOCALE_SUPPORT)
 188                                 setlocale(LC_TIME, "C");
 189                         strcpy(fmt_dt2str, "%a, %d %b %Y %H:%M:%S ");
 190                         i = 22;
 191                         goto format_utc;
 192                 } else /* default case */
 193                         fmt_dt2str = (char*)"%a %b %e %H:%M:%S %Z %Y";
 194         }
 195
 196 #define date_buf bb_common_bufsiz1
 197         if (*fmt_dt2str == '\0') {
 198                 /* With no format string, just print a blank line */
 199                 date_buf[0] = '\0';
 200         } else {
 201                 /* Handle special conversions */
 202                 if (strncmp(fmt_dt2str, "%f", 2) == 0) {
 203                         fmt_dt2str = (char*)"%Y.%m.%d-%H:%M:%S";
 204                 }
 205
 206                 /* Generate output string */
 207                 strftime(date_buf, sizeof(date_buf), fmt_dt2str, &tm_time);
 208         }
 209         puts(date_buf);
 210
 211         return EXIT_SUCCESS;
 212 }