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  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17  * General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  22  *
  23 */
  24
  25 #include <stdlib.h>
  26 #include <errno.h>
  27 #include <sys/time.h>
  28 #include <unistd.h>
  29 #include <time.h>
  30 #include <stdio.h>
  31 #include <string.h>
  32 #include "busybox.h"
  33
  34
  35 /* This 'date' command supports only 2 time setting formats,
  36    all the GNU strftime stuff (its in libc, lets use it),
  37    setting time using UTC and displaying int, as well as
  38    an RFC 822 complient date output for shell scripting
  39    mail commands */
  40
  41 /* Input parsing code is always bulky - used heavy duty libc stuff as
  42    much as possible, missed out a lot of bounds checking */
  43
  44 /* Default input handling to save surprising some people */
  45
  46 static struct tm *date_conv_time(struct tm *tm_time, const char *t_string)
  47 {
  48         int nr;
  49         char *cp;
  50
  51         nr = sscanf(t_string, "%2d%2d%2d%2d%d", &(tm_time->tm_mon),
  52                                 &(tm_time->tm_mday), &(tm_time->tm_hour), &(tm_time->tm_min),
  53                                 &(tm_time->tm_year));
  54
  55         if (nr < 4 || nr > 5) {
  56                 bb_error_msg_and_die(bb_msg_invalid_date, t_string);
  57         }
  58
  59         cp = strchr(t_string, '.');
  60         if (cp) {
  61                 nr = sscanf(cp + 1, "%2d", &(tm_time->tm_sec));
  62                 if (nr != 1) {
  63                         bb_error_msg_and_die(bb_msg_invalid_date, t_string);
  64                 }
  65         }
  66
  67         /* correct for century  - minor Y2K problem here? */
  68         if (tm_time->tm_year >= 1900) {
  69                 tm_time->tm_year -= 1900;
  70         }
  71         /* adjust date */
  72         tm_time->tm_mon -= 1;
  73
  74         return (tm_time);
  75
  76 }
  77
  78
  79 /* The new stuff for LRP */
  80
  81 static struct tm *date_conv_ftime(struct tm *tm_time, const char *t_string)
  82 {
  83         struct tm t;
  84
  85         /* Parse input and assign appropriately to tm_time */
  86
  87         if (t =
  88                 *tm_time, sscanf(t_string, "%d:%d:%d", &t.tm_hour, &t.tm_min,
  89                                                  &t.tm_sec) == 3) {
  90                 /* no adjustments needed */
  91         } else if (t =
  92                            *tm_time, sscanf(t_string, "%d:%d", &t.tm_hour,
  93                                                                 &t.tm_min) == 2) {
  94                 /* no adjustments needed */
  95         } else if (t =
  96                            *tm_time, sscanf(t_string, "%d.%d-%d:%d:%d", &t.tm_mon,
  97                                                                 &t.tm_mday, &t.tm_hour, &t.tm_min,
  98                                                                 &t.tm_sec) == 5) {
  99                 /* Adjust dates from 1-12 to 0-11 */
 100                 t.tm_mon -= 1;
 101         } else if (t =
 102                            *tm_time, sscanf(t_string, "%d.%d-%d:%d", &t.tm_mon,
 103                                                                 &t.tm_mday, &t.tm_hour, &t.tm_min) == 4) {
 104                 /* Adjust dates from 1-12 to 0-11 */
 105                 t.tm_mon -= 1;
 106         } else if (t =
 107                            *tm_time, sscanf(t_string, "%d.%d.%d-%d:%d:%d", &t.tm_year,
 108                                                                 &t.tm_mon, &t.tm_mday, &t.tm_hour, &t.tm_min,
 109                                                                 &t.tm_sec) == 6) {
 110                 t.tm_year -= 1900;      /* Adjust years */
 111                 t.tm_mon -= 1;  /* Adjust dates from 1-12 to 0-11 */
 112         } else if (t =
 113                            *tm_time, sscanf(t_string, "%d.%d.%d-%d:%d", &t.tm_year,
 114                                                                 &t.tm_mon, &t.tm_mday, &t.tm_hour,
 115                                                                 &t.tm_min) == 5) {
 116                 t.tm_year -= 1900;      /* Adjust years */
 117                 t.tm_mon -= 1;  /* Adjust dates from 1-12 to 0-11 */
 118         } else {
 119                 bb_error_msg_and_die(bb_msg_invalid_date, t_string);
 120         }
 121         *tm_time = t;
 122         return (tm_time);
 123 }
 124
 125 #define DATE_OPT_RFC2822        0x01
 126 #define DATE_OPT_SET            0x02
 127 #define DATE_OPT_UTC            0x04
 128 #define DATE_OPT_DATE           0x08
 129 #define DATE_OPT_REFERENCE      0x10
 130 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 131 # define DATE_OPT_TIMESPEC      0x20
 132 #endif
 133
 134 int date_main(int argc, char **argv)
 135 {
 136         char *date_str = NULL;
 137         char *date_fmt = NULL;
 138         int set_time;
 139         int utc;
 140         int use_arg = 0;
 141         time_t tm;
 142         unsigned long opt;
 143         struct tm tm_time;
 144         char *filename = NULL;
 145
 146 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 147         int ifmt = 0;
 148         char *isofmt_arg;
 149
 150 # define GETOPT_ISOFMT  "I::"
 151 #else
 152 # define GETOPT_ISOFMT
 153 #endif
 154         bb_opt_complementally = "!d~ds:s~ds";
 155         opt = bb_getopt_ulflags(argc, argv, "Rs:ud:r:" GETOPT_ISOFMT,
 156                                         &date_str, &date_str, &filename
 157 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 158                                         , &isofmt_arg
 159 #endif
 160                                         );
 161         set_time = opt & DATE_OPT_SET;
 162         utc = opt & DATE_OPT_UTC;
 163         if ((utc) && (putenv("TZ=UTC0") != 0)) {
 164                 bb_error_msg_and_die(bb_msg_memory_exhausted);
 165         }
 166         use_arg = opt & DATE_OPT_DATE;
 167 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 168         if(opt & DATE_OPT_TIMESPEC) {
 169                 if (!isofmt_arg) {
 170                         ifmt = 1;
 171                 } else {
 172                         int ifmt_len = bb_strlen(isofmt_arg);
 173
 174                         if ((ifmt_len <= 4)
 175                                 && (strncmp(isofmt_arg, "date", ifmt_len) == 0)) {
 176                                 ifmt = 1;
 177                         } else if ((ifmt_len <= 5)
 178                                    && (strncmp(isofmt_arg, "hours", ifmt_len) == 0)) {
 179                                 ifmt = 2;
 180                         } else if ((ifmt_len <= 7)
 181                                    && (strncmp(isofmt_arg, "minutes", ifmt_len) == 0)) {
 182                                 ifmt = 3;
 183                         } else if ((ifmt_len <= 7)
 184                                    && (strncmp(isofmt_arg, "seconds", ifmt_len) == 0)) {
 185                                 ifmt = 4;
 186                         }
 187                 }
 188                 if (!ifmt) {
 189                         bb_show_usage();
 190                 }
 191         }
 192 #endif
 193
 194         if ((date_fmt == NULL) && (optind < argc) && (argv[optind][0] == '+')) {
 195                 date_fmt = &argv[optind][1];    /* Skip over the '+' */
 196         } else if (date_str == NULL) {
 197                 set_time = 1;
 198                 date_str = argv[optind];
 199         }
 200
 201         /* Now we have parsed all the information except the date format
 202            which depends on whether the clock is being set or read */
 203
 204         if(filename) {
 205                 struct stat statbuf;
 206                 if(stat(filename,&statbuf))
 207                         bb_perror_msg_and_die("File '%s' not found.\n",filename);
 208                 tm=statbuf.st_mtime;
 209         } else time(&tm);
 210         memcpy(&tm_time, localtime(&tm), sizeof(tm_time));
 211         /* Zero out fields - take her back to midnight! */
 212         if (date_str != NULL) {
 213                 tm_time.tm_sec = 0;
 214                 tm_time.tm_min = 0;
 215                 tm_time.tm_hour = 0;
 216
 217                 /* Process any date input to UNIX time since 1 Jan 1970 */
 218                 if (strchr(date_str, ':') != NULL) {
 219                         date_conv_ftime(&tm_time, date_str);
 220                 } else {
 221                         date_conv_time(&tm_time, date_str);
 222                 }
 223
 224                 /* Correct any day of week and day of year etc. fields */
 225                 tm_time.tm_isdst = -1;  /* Be sure to recheck dst. */
 226                 tm = mktime(&tm_time);
 227                 if (tm < 0) {
 228                         bb_error_msg_and_die(bb_msg_invalid_date, date_str);
 229                 }
 230                 if (utc && (putenv("TZ=UTC0") != 0)) {
 231                         bb_error_msg_and_die(bb_msg_memory_exhausted);
 232                 }
 233
 234                 /* if setting time, set it */
 235                 if (set_time && (stime(&tm) < 0)) {
 236                         bb_perror_msg("cannot set date");
 237                 }
 238         }
 239
 240         /* Display output */
 241
 242         /* Deal with format string */
 243         if (date_fmt == NULL) {
 244 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 245                 switch (ifmt) {
 246                 case 4:
 247                         date_fmt = utc ? "%Y-%m-%dT%H:%M:%SZ" : "%Y-%m-%dT%H:%M:%S%z";
 248                         break;
 249                 case 3:
 250                         date_fmt = utc ? "%Y-%m-%dT%H:%MZ" : "%Y-%m-%dT%H:%M%z";
 251                         break;
 252                 case 2:
 253                         date_fmt = utc ? "%Y-%m-%dT%HZ" : "%Y-%m-%dT%H%z";
 254                         break;
 255                 case 1:
 256                         date_fmt = "%Y-%m-%d";
 257                         break;
 258                 case 0:
 259                 default:
 260 #endif
 261                         date_fmt = (opt & DATE_OPT_RFC2822 ?
 262                                         (utc ? "%a, %d %b %Y %H:%M:%S GMT" :
 263                                         "%a, %d %b %Y %H:%M:%S %z") :
 264                                         "%a %b %e %H:%M:%S %Z %Y");
 265
 266 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 267                         break;
 268                 }
 269 #endif
 270         } else if (*date_fmt == '\0') {
 271                 /* Imitate what GNU 'date' does with NO format string! */
 272                 printf("\n");
 273                 return EXIT_SUCCESS;
 274         }
 275
 276         /* Handle special conversions */
 277
 278         if (strncmp(date_fmt, "%f", 2) == 0) {
 279                 date_fmt = "%Y.%m.%d-%H:%M:%S";
 280         }
 281
 282         {
 283                 /* Print OUTPUT (after ALL that!) */
 284                 RESERVE_CONFIG_BUFFER(t_buff, 201);
 285                 strftime(t_buff, 200, date_fmt, &tm_time);
 286                 puts(t_buff);
 287                 RELEASE_CONFIG_BUFFER(t_buff);
 288         }
 289
 290         return EXIT_SUCCESS;
 291 }