123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294 |
- /* vi: set sw=4 ts=4: */
- /*
- * Mini date implementation for busybox
- *
- * by Matthew Grant <grantma@anathoth.gen.nz>
- *
- * iso-format handling added by Robert Griebl <griebl@gmx.de>
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
- *
- */
- #include <stdlib.h>
- #include <errno.h>
- #include <sys/time.h>
- #include <unistd.h>
- #include <time.h>
- #include <stdio.h>
- #include <string.h>
- #include <getopt.h>
- #include "busybox.h"
- /* This 'date' command supports only 2 time setting formats,
- all the GNU strftime stuff (its in libc, lets use it),
- setting time using UTC and displaying int, as well as
- an RFC 822 complient date output for shell scripting
- mail commands */
- /* Input parsing code is always bulky - used heavy duty libc stuff as
- much as possible, missed out a lot of bounds checking */
- /* Default input handling to save surprising some people */
- static struct tm *date_conv_time(struct tm *tm_time, const char *t_string)
- {
- int nr;
- char *cp;
- nr = sscanf(t_string, "%2d%2d%2d%2d%d", &(tm_time->tm_mon),
- &(tm_time->tm_mday), &(tm_time->tm_hour), &(tm_time->tm_min),
- &(tm_time->tm_year));
- if (nr < 4 || nr > 5) {
- bb_error_msg_and_die(bb_msg_invalid_date, t_string);
- }
- cp = strchr(t_string, '.');
- if (cp) {
- nr = sscanf(cp + 1, "%2d", &(tm_time->tm_sec));
- if (nr != 1) {
- bb_error_msg_and_die(bb_msg_invalid_date, t_string);
- }
- }
- /* correct for century - minor Y2K problem here? */
- if (tm_time->tm_year >= 1900) {
- tm_time->tm_year -= 1900;
- }
- /* adjust date */
- tm_time->tm_mon -= 1;
- return (tm_time);
- }
- /* The new stuff for LRP */
- static struct tm *date_conv_ftime(struct tm *tm_time, const char *t_string)
- {
- struct tm t;
- /* Parse input and assign appropriately to tm_time */
- if (t =
- *tm_time, sscanf(t_string, "%d:%d:%d", &t.tm_hour, &t.tm_min,
- &t.tm_sec) == 3) {
- /* no adjustments needed */
- } else if (t =
- *tm_time, sscanf(t_string, "%d:%d", &t.tm_hour,
- &t.tm_min) == 2) {
- /* no adjustments needed */
- } else if (t =
- *tm_time, sscanf(t_string, "%d.%d-%d:%d:%d", &t.tm_mon,
- &t.tm_mday, &t.tm_hour, &t.tm_min,
- &t.tm_sec) == 5) {
- /* Adjust dates from 1-12 to 0-11 */
- t.tm_mon -= 1;
- } else if (t =
- *tm_time, sscanf(t_string, "%d.%d-%d:%d", &t.tm_mon,
- &t.tm_mday, &t.tm_hour, &t.tm_min) == 4) {
- /* Adjust dates from 1-12 to 0-11 */
- t.tm_mon -= 1;
- } else if (t =
- *tm_time, sscanf(t_string, "%d.%d.%d-%d:%d:%d", &t.tm_year,
- &t.tm_mon, &t.tm_mday, &t.tm_hour, &t.tm_min,
- &t.tm_sec) == 6) {
- t.tm_year -= 1900; /* Adjust years */
- t.tm_mon -= 1; /* Adjust dates from 1-12 to 0-11 */
- } else if (t =
- *tm_time, sscanf(t_string, "%d.%d.%d-%d:%d", &t.tm_year,
- &t.tm_mon, &t.tm_mday, &t.tm_hour,
- &t.tm_min) == 5) {
- t.tm_year -= 1900; /* Adjust years */
- t.tm_mon -= 1; /* Adjust dates from 1-12 to 0-11 */
- } else {
- bb_error_msg_and_die(bb_msg_invalid_date, t_string);
- }
- *tm_time = t;
- return (tm_time);
- }
- #define DATE_OPT_RFC2822 0x01
- #define DATE_OPT_SET 0x02
- #define DATE_OPT_UTC 0x04
- #define DATE_OPT_DATE 0x08
- #define DATE_OPT_REFERENCE 0x10
- #ifdef CONFIG_FEATURE_DATE_ISOFMT
- # define DATE_OPT_TIMESPEC 0x20
- #endif
- int date_main(int argc, char **argv)
- {
- char *date_str = NULL;
- char *date_fmt = NULL;
- int set_time;
- int utc;
- int use_arg = 0;
- time_t tm;
- unsigned long opt;
- struct tm tm_time;
- char *filename = NULL;
- #ifdef CONFIG_FEATURE_DATE_ISOFMT
- int ifmt = 0;
- char *isofmt_arg;
- # define GETOPT_ISOFMT "I::"
- #else
- # define GETOPT_ISOFMT
- #endif
- bb_opt_complementaly = "d~ds:s~ds";
- opt = bb_getopt_ulflags(argc, argv, "Rs:ud:r:" GETOPT_ISOFMT,
- &date_str, &date_str, &filename
- #ifdef CONFIG_FEATURE_DATE_ISOFMT
- , &isofmt_arg
- #endif
- );
- set_time = opt & DATE_OPT_SET;
- utc = opt & DATE_OPT_UTC;
- if ((utc) && (putenv("TZ=UTC0") != 0)) {
- bb_error_msg_and_die(bb_msg_memory_exhausted);
- }
- use_arg = opt & DATE_OPT_DATE;
- if(opt & BB_GETOPT_ERROR)
- bb_show_usage();
- #ifdef CONFIG_FEATURE_DATE_ISOFMT
- if(opt & DATE_OPT_TIMESPEC) {
- if (!isofmt_arg) {
- ifmt = 1;
- } else {
- int ifmt_len = bb_strlen(isofmt_arg);
- if ((ifmt_len <= 4)
- && (strncmp(isofmt_arg, "date", ifmt_len) == 0)) {
- ifmt = 1;
- } else if ((ifmt_len <= 5)
- && (strncmp(isofmt_arg, "hours", ifmt_len) == 0)) {
- ifmt = 2;
- } else if ((ifmt_len <= 7)
- && (strncmp(isofmt_arg, "minutes", ifmt_len) == 0)) {
- ifmt = 3;
- } else if ((ifmt_len <= 7)
- && (strncmp(isofmt_arg, "seconds", ifmt_len) == 0)) {
- ifmt = 4;
- }
- }
- if (!ifmt) {
- bb_show_usage();
- }
- }
- #endif
- if ((date_fmt == NULL) && (optind < argc) && (argv[optind][0] == '+')) {
- date_fmt = &argv[optind][1]; /* Skip over the '+' */
- } else if (date_str == NULL) {
- set_time = 1;
- date_str = argv[optind];
- }
- /* Now we have parsed all the information except the date format
- which depends on whether the clock is being set or read */
- if(filename) {
- struct stat statbuf;
- if(stat(filename,&statbuf))
- bb_perror_msg_and_die("File '%s' not found.\n",filename);
- tm=statbuf.st_mtime;
- } else time(&tm);
- memcpy(&tm_time, localtime(&tm), sizeof(tm_time));
- /* Zero out fields - take her back to midnight! */
- if (date_str != NULL) {
- tm_time.tm_sec = 0;
- tm_time.tm_min = 0;
- tm_time.tm_hour = 0;
- /* Process any date input to UNIX time since 1 Jan 1970 */
- if (strchr(date_str, ':') != NULL) {
- date_conv_ftime(&tm_time, date_str);
- } else {
- date_conv_time(&tm_time, date_str);
- }
- /* Correct any day of week and day of year etc. fields */
- tm_time.tm_isdst = -1; /* Be sure to recheck dst. */
- tm = mktime(&tm_time);
- if (tm < 0) {
- bb_error_msg_and_die(bb_msg_invalid_date, date_str);
- }
- if (utc && (putenv("TZ=UTC0") != 0)) {
- bb_error_msg_and_die(bb_msg_memory_exhausted);
- }
- /* if setting time, set it */
- if (set_time && (stime(&tm) < 0)) {
- bb_perror_msg("cannot set date");
- }
- }
- /* Display output */
- /* Deal with format string */
- if (date_fmt == NULL) {
- #ifdef CONFIG_FEATURE_DATE_ISOFMT
- switch (ifmt) {
- case 4:
- date_fmt = utc ? "%Y-%m-%dT%H:%M:%SZ" : "%Y-%m-%dT%H:%M:%S%z";
- break;
- case 3:
- date_fmt = utc ? "%Y-%m-%dT%H:%MZ" : "%Y-%m-%dT%H:%M%z";
- break;
- case 2:
- date_fmt = utc ? "%Y-%m-%dT%HZ" : "%Y-%m-%dT%H%z";
- break;
- case 1:
- date_fmt = "%Y-%m-%d";
- break;
- case 0:
- default:
- #endif
- date_fmt = (opt & DATE_OPT_RFC2822 ?
- (utc ? "%a, %d %b %Y %H:%M:%S GMT" :
- "%a, %d %b %Y %H:%M:%S %z") :
- "%a %b %e %H:%M:%S %Z %Y");
- #ifdef CONFIG_FEATURE_DATE_ISOFMT
- break;
- }
- #endif
- } else if (*date_fmt == '\0') {
- /* Imitate what GNU 'date' does with NO format string! */
- printf("\n");
- return EXIT_SUCCESS;
- }
- /* Handle special conversions */
- if (strncmp(date_fmt, "%f", 2) == 0) {
- date_fmt = "%Y.%m.%d-%H:%M:%S";
- }
- {
- /* Print OUTPUT (after ALL that!) */
- RESERVE_CONFIG_BUFFER(t_buff, 201);
- strftime(t_buff, 200, date_fmt, &tm_time);
- puts(t_buff);
- RELEASE_CONFIG_BUFFER(t_buff);
- }
- return EXIT_SUCCESS;
- }
|