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