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 suprising some people */
  46
  47 static struct tm *date_conv_time(struct tm *tm_time, const char *t_string)
  48 {
  49         int nr;
  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         /* correct for century  - minor Y2K problem here? */
  60         if (tm_time->tm_year >= 1900) {
  61                 tm_time->tm_year -= 1900;
  62         }
  63         /* adjust date */
  64         tm_time->tm_mon -= 1;
  65
  66         return (tm_time);
  67
  68 }
  69
  70
  71 /* The new stuff for LRP */
  72
  73 static struct tm *date_conv_ftime(struct tm *tm_time, const char *t_string)
  74 {
  75         struct tm t;
  76
  77         /* Parse input and assign appropriately to tm_time */
  78
  79         if (t =
  80                 *tm_time, sscanf(t_string, "%d:%d:%d", &t.tm_hour, &t.tm_min,
  81                                                  &t.tm_sec) == 3) {
  82                 /* no adjustments needed */
  83         } else if (t =
  84                            *tm_time, sscanf(t_string, "%d:%d", &t.tm_hour,
  85                                                                 &t.tm_min) == 2) {
  86                 /* no adjustments needed */
  87         } else if (t =
  88                            *tm_time, sscanf(t_string, "%d.%d-%d:%d:%d", &t.tm_mon,
  89                                                                 &t.tm_mday, &t.tm_hour, &t.tm_min,
  90                                                                 &t.tm_sec) == 5) {
  91                 /* Adjust dates from 1-12 to 0-11 */
  92                 t.tm_mon -= 1;
  93         } else if (t =
  94                            *tm_time, sscanf(t_string, "%d.%d-%d:%d", &t.tm_mon,
  95                                                                 &t.tm_mday, &t.tm_hour, &t.tm_min) == 4) {
  96                 /* Adjust dates from 1-12 to 0-11 */
  97                 t.tm_mon -= 1;
  98         } else if (t =
  99                            *tm_time, sscanf(t_string, "%d.%d.%d-%d:%d:%d", &t.tm_year,
 100                                                                 &t.tm_mon, &t.tm_mday, &t.tm_hour, &t.tm_min,
 101                                                                 &t.tm_sec) == 6) {
 102                 t.tm_year -= 1900;      /* Adjust years */
 103                 t.tm_mon -= 1;  /* Adjust dates from 1-12 to 0-11 */
 104         } else if (t =
 105                            *tm_time, sscanf(t_string, "%d.%d.%d-%d:%d", &t.tm_year,
 106                                                                 &t.tm_mon, &t.tm_mday, &t.tm_hour,
 107                                                                 &t.tm_min) == 5) {
 108                 t.tm_year -= 1900;      /* Adjust years */
 109                 t.tm_mon -= 1;  /* Adjust dates from 1-12 to 0-11 */
 110         } else {
 111                 bb_error_msg_and_die(bb_msg_invalid_date, t_string);
 112         }
 113         *tm_time = t;
 114         return (tm_time);
 115 }
 116
 117
 118 int date_main(int argc, char **argv)
 119 {
 120         char *date_str = NULL;
 121         char *date_fmt = NULL;
 122         char *t_buff;
 123         int c;
 124         int set_time = 0;
 125         int rfc822 = 0;
 126         int utc = 0;
 127         int use_arg = 0;
 128         time_t tm;
 129         struct tm tm_time;
 130
 131 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 132         int ifmt = 0;
 133
 134 # define GETOPT_ISOFMT  "I::"
 135 #else
 136 # define GETOPT_ISOFMT
 137 #endif
 138
 139         /* Interpret command line args */
 140         while ((c = getopt(argc, argv, "Rs:ud:" GETOPT_ISOFMT)) != EOF) {
 141                 switch (c) {
 142                 case 'R':
 143                         rfc822 = 1;
 144                         break;
 145                 case 's':
 146                         set_time = 1;
 147                         if ((date_str != NULL) || ((date_str = optarg) == NULL)) {
 148                                 bb_show_usage();
 149                         }
 150                         break;
 151                 case 'u':
 152                         utc = 1;
 153                         if (putenv("TZ=UTC0") != 0)
 154                                 bb_error_msg_and_die(bb_msg_memory_exhausted);
 155                         break;
 156                 case 'd':
 157                         use_arg = 1;
 158                         if ((date_str != NULL) || ((date_str = optarg) == NULL))
 159                                 bb_show_usage();
 160                         break;
 161 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 162                 case 'I':
 163                         if (!optarg)
 164                                 ifmt = 1;
 165                         else {
 166                                 int ifmt_len = bb_strlen(optarg);
 167
 168                                 if ((ifmt_len <= 4)
 169                                         && (strncmp(optarg, "date", ifmt_len) == 0)) {
 170                                         ifmt = 1;
 171                                 } else if ((ifmt_len <= 5)
 172                                                    && (strncmp(optarg, "hours", ifmt_len) == 0)) {
 173                                         ifmt = 2;
 174                                 } else if ((ifmt_len <= 7)
 175                                                    && (strncmp(optarg, "minutes", ifmt_len) == 0)) {
 176                                         ifmt = 3;
 177                                 } else if ((ifmt_len <= 7)
 178                                                    && (strncmp(optarg, "seconds", ifmt_len) == 0)) {
 179                                         ifmt = 4;
 180                                 }
 181                         }
 182                         if (ifmt) {
 183                                 break;  /* else bb_show_usage(); */
 184                         }
 185 #endif
 186                 default:
 187                         bb_show_usage();
 188                 }
 189         }
 190
 191
 192         if ((date_fmt == NULL) && (optind < argc) && (argv[optind][0] == '+')) {
 193                 date_fmt = &argv[optind][1];    /* Skip over the '+' */
 194         } else if (date_str == NULL) {
 195                 set_time = 1;
 196                 date_str = argv[optind];
 197         }
 198
 199         /* Now we have parsed all the information except the date format
 200            which depends on whether the clock is being set or read */
 201
 202         time(&tm);
 203         memcpy(&tm_time, localtime(&tm), sizeof(tm_time));
 204         /* Zero out fields - take her back to midnight! */
 205         if (date_str != NULL) {
 206                 tm_time.tm_sec = 0;
 207                 tm_time.tm_min = 0;
 208                 tm_time.tm_hour = 0;
 209         }
 210
 211         /* Process any date input to UNIX time since 1 Jan 1970 */
 212         if (date_str != NULL) {
 213
 214                 if (strchr(date_str, ':') != NULL) {
 215                         date_conv_ftime(&tm_time, date_str);
 216                 } else {
 217                         date_conv_time(&tm_time, date_str);
 218                 }
 219
 220                 /* Correct any day of week and day of year etc. fields */
 221                 tm = mktime(&tm_time);
 222                 if (tm < 0) {
 223                         bb_error_msg_and_die(bb_msg_invalid_date, date_str);
 224                 }
 225                 if (utc && (putenv("TZ=UTC0") != 0)) {
 226                         bb_error_msg_and_die(bb_msg_memory_exhausted);
 227                 }
 228
 229                 /* if setting time, set it */
 230                 if (set_time && (stime(&tm) < 0)) {
 231                         bb_perror_msg("cannot set date");
 232                 }
 233         }
 234
 235         /* Display output */
 236
 237         /* Deal with format string */
 238         if (date_fmt == NULL) {
 239 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 240                 switch (ifmt) {
 241                 case 4:
 242                         date_fmt = utc ? "%Y-%m-%dT%H:%M:%SZ" : "%Y-%m-%dT%H:%M:%S%z";
 243                         break;
 244                 case 3:
 245                         date_fmt = utc ? "%Y-%m-%dT%H:%MZ" : "%Y-%m-%dT%H:%M%z";
 246                         break;
 247                 case 2:
 248                         date_fmt = utc ? "%Y-%m-%dT%HZ" : "%Y-%m-%dT%H%z";
 249                         break;
 250                 case 1:
 251                         date_fmt = "%Y-%m-%d";
 252                         break;
 253                 case 0:
 254                 default:
 255 #endif
 256                         date_fmt =
 257                                 (rfc822
 258                                  ? (utc ? "%a, %e %b %Y %H:%M:%S GMT" :
 259                                         "%a, %e %b %Y %H:%M:%S %z") : "%a %b %e %H:%M:%S %Z %Y");
 260
 261 #ifdef CONFIG_FEATURE_DATE_ISOFMT
 262                         break;
 263                 }
 264 #endif
 265         } else if (*date_fmt == '\0') {
 266                 /* Imitate what GNU 'date' does with NO format string! */
 267                 printf("\n");
 268                 return EXIT_SUCCESS;
 269         }
 270
 271         /* Handle special conversions */
 272
 273         if (strncmp(date_fmt, "%f", 2) == 0) {
 274                 date_fmt = "%Y.%m.%d-%H:%M:%S";
 275         }
 276
 277         /* Print OUTPUT (after ALL that!) */
 278         t_buff = xmalloc(201);
 279         strftime(t_buff, 200, date_fmt, &tm_time);
 280         puts(t_buff);
 281
 282         return EXIT_SUCCESS;
 283 }