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