busybox/coreutils/date.c

/* 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>
 * bugfixes and cleanup by Bernhard Fischer
 *
 * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
*/

#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 it, as well as
   an RFC 2822 compliant 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 */


#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
#define DATE_OPT_TIMESPEC	0x20
#define DATE_OPT_HINT		0x40

static void xputenv(char *s)
{
	if (putenv(s) != 0)
		bb_error_msg_and_die(bb_msg_memory_exhausted);
}

static void maybe_set_utc(int opt)
{
	if (opt & DATE_OPT_UTC)
		xputenv("TZ=UTC0");
}

int date_main(int argc, char **argv)
{
	time_t tm;
	struct tm tm_time;
	unsigned opt;
	int ifmt = -1;
	char *date_str = NULL;
	char *date_fmt = NULL;
	char *filename = NULL;
	char *isofmt_arg;
	char *hintfmt_arg;

	opt_complementary = "?:d--s:s--d"
		USE_FEATURE_DATE_ISOFMT(":R--I:I--R");
	opt = getopt32(argc, argv, "Rs:ud:r:"
			USE_FEATURE_DATE_ISOFMT("I::D:"),
			&date_str, &date_str, &filename
			USE_FEATURE_DATE_ISOFMT(, &isofmt_arg, &hintfmt_arg));
	maybe_set_utc(opt);

	if (ENABLE_FEATURE_DATE_ISOFMT && (opt & DATE_OPT_TIMESPEC)) {
		if (!isofmt_arg) {
			ifmt = 0; /* default is date */
		} else {
			const char * const isoformats[] =
				{"date", "hours", "minutes", "seconds"};

			for (ifmt = 0; ifmt < 4; ifmt++)
				if (!strcmp(isofmt_arg, isoformats[ifmt])) {
					break;
				}
			if (ifmt == 4) /* parse error */
				bb_show_usage();
		}
	}

	/* XXX, date_fmt == NULL from this always */
	if ((date_fmt == NULL) && (optind < argc) && (argv[optind][0] == '+')) {
		date_fmt = &argv[optind][1];	/* Skip over the '+' */
	} else if (date_str == NULL) {
		opt |= DATE_OPT_SET;
		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;
		xstat(filename, &statbuf);
		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 (ENABLE_FEATURE_DATE_ISOFMT && (opt & DATE_OPT_HINT)) {
			strptime(date_str, hintfmt_arg, &tm_time);
		} else if (strchr(date_str, ':') != NULL) {
			/* Parse input and assign appropriately to tm_time */

			if (sscanf(date_str, "%d:%d:%d", &tm_time.tm_hour, &tm_time.tm_min,
								 &tm_time.tm_sec) == 3) {
				/* no adjustments needed */
			} else if (sscanf(date_str, "%d:%d", &tm_time.tm_hour,
										&tm_time.tm_min) == 2) {
				/* no adjustments needed */
			} else if (sscanf(date_str, "%d.%d-%d:%d:%d", &tm_time.tm_mon,
								&tm_time.tm_mday, &tm_time.tm_hour,
								&tm_time.tm_min, &tm_time.tm_sec) == 5) {
				/* Adjust dates from 1-12 to 0-11 */
				tm_time.tm_mon -= 1;
			} else if (sscanf(date_str, "%d.%d-%d:%d", &tm_time.tm_mon,
								&tm_time.tm_mday,
								&tm_time.tm_hour, &tm_time.tm_min) == 4) {
				/* Adjust dates from 1-12 to 0-11 */
				tm_time.tm_mon -= 1;
			} else if (sscanf(date_str, "%d.%d.%d-%d:%d:%d", &tm_time.tm_year,
								&tm_time.tm_mon, &tm_time.tm_mday,
								&tm_time.tm_hour, &tm_time.tm_min,
									&tm_time.tm_sec) == 6) {
				tm_time.tm_year -= 1900;	/* Adjust years */
				tm_time.tm_mon -= 1;	/* Adjust dates from 1-12 to 0-11 */
			} else if (sscanf(date_str, "%d.%d.%d-%d:%d", &tm_time.tm_year,
								&tm_time.tm_mon, &tm_time.tm_mday,
								&tm_time.tm_hour, &tm_time.tm_min) == 5) {
				tm_time.tm_year -= 1900;	/* Adjust years */
				tm_time.tm_mon -= 1;	/* Adjust dates from 1-12 to 0-11 */
			} else {
				bb_error_msg_and_die(bb_msg_invalid_date, date_str);
			}
		} else {
			int nr;
			char *cp;

			nr = sscanf(date_str, "%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, date_str);
			}

			cp = strchr(date_str, '.');
			if (cp) {
				nr = sscanf(cp + 1, "%2d", &tm_time.tm_sec);
				if (nr != 1) {
					bb_error_msg_and_die(bb_msg_invalid_date, date_str);
				}
			}

			/* 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;
		}

		/* 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);
		}
		maybe_set_utc(opt);

		/* if setting time, set it */
		if ((opt & DATE_OPT_SET) && stime(&tm) < 0) {
			bb_perror_msg("cannot set date");
		}
	}

	/* Display output */

	/* Deal with format string */

	if (date_fmt == NULL) {
		int i;
		date_fmt = xzalloc(32);
		if (ENABLE_FEATURE_DATE_ISOFMT && ifmt >= 0) {
			strcpy(date_fmt, "%Y-%m-%d");
			if (ifmt > 0) {
				i = 8;
				date_fmt[i++] = 'T';
				date_fmt[i++] = '%';
				date_fmt[i++] = 'H';
				if (ifmt > 1) {
					date_fmt[i++] = ':';
					date_fmt[i++] = '%';
					date_fmt[i++] = 'M';
				}
				if (ifmt > 2) {
					date_fmt[i++] = ':';
					date_fmt[i++] = '%';
					date_fmt[i++] = 'S';
				}
format_utc:
				date_fmt[i++] = '%';
				date_fmt[i] = (opt & DATE_OPT_UTC) ? 'Z' : 'z';
			}
		} else if (opt & DATE_OPT_RFC2822) {
			/* Undo busybox.c for date -R */
			if (ENABLE_LOCALE_SUPPORT)
				setlocale(LC_TIME, "C");
			strcpy(date_fmt, "%a, %d %b %Y %H:%M:%S ");
			i = 22;
			goto format_utc;
		} else /* default case */
			date_fmt = "%a %b %e %H:%M:%S %Z %Y";
	}

	if (*date_fmt == '\0') {
		/* With no format string, just print a blank line */
		*bb_common_bufsiz1 = 0;
	} else {
		/* Handle special conversions */

		if (strncmp(date_fmt, "%f", 2) == 0) {
			date_fmt = "%Y.%m.%d-%H:%M:%S";
		}

		/* Generate output string */
		strftime(bb_common_bufsiz1, 200, date_fmt, &tm_time);
	}
	puts(bb_common_bufsiz1);

	return EXIT_SUCCESS;
}
1	niro	532	/* 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			* bugfixes and cleanup by Bernhard Fischer
9			*
10			* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
11			*/
12
13			#include "busybox.h"
14
15			/* This 'date' command supports only 2 time setting formats,
16			all the GNU strftime stuff (its in libc, lets use it),
17			setting time using UTC and displaying it, as well as
18			an RFC 2822 compliant date output for shell scripting
19			mail commands */
20
21			/* Input parsing code is always bulky - used heavy duty libc stuff as
22			much as possible, missed out a lot of bounds checking */
23
24			/* Default input handling to save surprising some people */
25
26
27			#define DATE_OPT_RFC2822 0x01
28			#define DATE_OPT_SET 0x02
29			#define DATE_OPT_UTC 0x04
30			#define DATE_OPT_DATE 0x08
31			#define DATE_OPT_REFERENCE 0x10
32			#define DATE_OPT_TIMESPEC 0x20
33			#define DATE_OPT_HINT 0x40
34
35			static void xputenv(char *s)
36			{
37			if (putenv(s) != 0)
38			bb_error_msg_and_die(bb_msg_memory_exhausted);
39			}
40
41			static void maybe_set_utc(int opt)
42			{
43			if (opt & DATE_OPT_UTC)
44			xputenv("TZ=UTC0");
45			}
46
47			int date_main(int argc, char **argv)
48			{
49			time_t tm;
50			struct tm tm_time;
51			unsigned opt;
52			int ifmt = -1;
53			char *date_str = NULL;
54			char *date_fmt = NULL;
55			char *filename = NULL;
56			char *isofmt_arg;
57			char *hintfmt_arg;
58
59			opt_complementary = "?:d--s:s--d"
60			USE_FEATURE_DATE_ISOFMT(":R--I:I--R");
61			opt = getopt32(argc, argv, "Rs:ud:r:"
62			USE_FEATURE_DATE_ISOFMT("I::D:"),
63			&date_str, &date_str, &filename
64			USE_FEATURE_DATE_ISOFMT(, &isofmt_arg, &hintfmt_arg));
65			maybe_set_utc(opt);
66
67			if (ENABLE_FEATURE_DATE_ISOFMT && (opt & DATE_OPT_TIMESPEC)) {
68			if (!isofmt_arg) {
69			ifmt = 0; /* default is date */
70			} else {
71			const char * const isoformats[] =
72			{"date", "hours", "minutes", "seconds"};
73
74			for (ifmt = 0; ifmt < 4; ifmt++)
75			if (!strcmp(isofmt_arg, isoformats[ifmt])) {
76			break;
77			}
78			if (ifmt == 4) /* parse error */
79			bb_show_usage();
80			}
81			}
82
83			/* XXX, date_fmt == NULL from this always */
84			if ((date_fmt == NULL) && (optind < argc) && (argv[optind][0] == '+')) {
85			date_fmt = &argv[optind][1]; /* Skip over the '+' */
86			} else if (date_str == NULL) {
87			opt \|= DATE_OPT_SET;
88			date_str = argv[optind];
89			}
90
91			/* Now we have parsed all the information except the date format
92			which depends on whether the clock is being set or read */
93
94			if (filename) {
95			struct stat statbuf;
96			xstat(filename, &statbuf);
97			tm = statbuf.st_mtime;
98			} else
99			time(&tm);
100			memcpy(&tm_time, localtime(&tm), sizeof(tm_time));
101			/* Zero out fields - take her back to midnight! */
102			if (date_str != NULL) {
103			tm_time.tm_sec = 0;
104			tm_time.tm_min = 0;
105			tm_time.tm_hour = 0;
106
107			/* Process any date input to UNIX time since 1 Jan 1970 */
108			if (ENABLE_FEATURE_DATE_ISOFMT && (opt & DATE_OPT_HINT)) {
109			strptime(date_str, hintfmt_arg, &tm_time);
110			} else if (strchr(date_str, ':') != NULL) {
111			/* Parse input and assign appropriately to tm_time */
112
113			if (sscanf(date_str, "%d:%d:%d", &tm_time.tm_hour, &tm_time.tm_min,
114			&tm_time.tm_sec) == 3) {
115			/* no adjustments needed */
116			} else if (sscanf(date_str, "%d:%d", &tm_time.tm_hour,
117			&tm_time.tm_min) == 2) {
118			/* no adjustments needed */
119			} else if (sscanf(date_str, "%d.%d-%d:%d:%d", &tm_time.tm_mon,
120			&tm_time.tm_mday, &tm_time.tm_hour,
121			&tm_time.tm_min, &tm_time.tm_sec) == 5) {
122			/* Adjust dates from 1-12 to 0-11 */
123			tm_time.tm_mon -= 1;
124			} else if (sscanf(date_str, "%d.%d-%d:%d", &tm_time.tm_mon,
125			&tm_time.tm_mday,
126			&tm_time.tm_hour, &tm_time.tm_min) == 4) {
127			/* Adjust dates from 1-12 to 0-11 */
128			tm_time.tm_mon -= 1;
129			} else if (sscanf(date_str, "%d.%d.%d-%d:%d:%d", &tm_time.tm_year,
130			&tm_time.tm_mon, &tm_time.tm_mday,
131			&tm_time.tm_hour, &tm_time.tm_min,
132			&tm_time.tm_sec) == 6) {
133			tm_time.tm_year -= 1900; /* Adjust years */
134			tm_time.tm_mon -= 1; /* Adjust dates from 1-12 to 0-11 */
135			} else if (sscanf(date_str, "%d.%d.%d-%d:%d", &tm_time.tm_year,
136			&tm_time.tm_mon, &tm_time.tm_mday,
137			&tm_time.tm_hour, &tm_time.tm_min) == 5) {
138			tm_time.tm_year -= 1900; /* Adjust years */
139			tm_time.tm_mon -= 1; /* Adjust dates from 1-12 to 0-11 */
140			} else {
141			bb_error_msg_and_die(bb_msg_invalid_date, date_str);
142			}
143			} else {
144			int nr;
145			char *cp;
146
147			nr = sscanf(date_str, "%2d%2d%2d%2d%d", &tm_time.tm_mon,
148			&tm_time.tm_mday, &tm_time.tm_hour, &tm_time.tm_min,
149			&tm_time.tm_year);
150
151			if (nr < 4 \|\| nr > 5) {
152			bb_error_msg_and_die(bb_msg_invalid_date, date_str);
153			}
154
155			cp = strchr(date_str, '.');
156			if (cp) {
157			nr = sscanf(cp + 1, "%2d", &tm_time.tm_sec);
158			if (nr != 1) {
159			bb_error_msg_and_die(bb_msg_invalid_date, date_str);
160			}
161			}
162
163			/* correct for century - minor Y2K problem here? */
164			if (tm_time.tm_year >= 1900) {
165			tm_time.tm_year -= 1900;
166			}
167			/* adjust date */
168			tm_time.tm_mon -= 1;
169			}
170
171			/* Correct any day of week and day of year etc. fields */
172			tm_time.tm_isdst = -1; /* Be sure to recheck dst. */
173			tm = mktime(&tm_time);
174			if (tm < 0) {
175			bb_error_msg_and_die(bb_msg_invalid_date, date_str);
176			}
177			maybe_set_utc(opt);
178
179			/* if setting time, set it */
180			if ((opt & DATE_OPT_SET) && stime(&tm) < 0) {
181			bb_perror_msg("cannot set date");
182			}
183			}
184
185			/* Display output */
186
187			/* Deal with format string */
188
189			if (date_fmt == NULL) {
190			int i;
191			date_fmt = xzalloc(32);
192			if (ENABLE_FEATURE_DATE_ISOFMT && ifmt >= 0) {
193			strcpy(date_fmt, "%Y-%m-%d");
194			if (ifmt > 0) {
195			i = 8;
196			date_fmt[i++] = 'T';
197			date_fmt[i++] = '%';
198			date_fmt[i++] = 'H';
199			if (ifmt > 1) {
200			date_fmt[i++] = ':';
201			date_fmt[i++] = '%';
202			date_fmt[i++] = 'M';
203			}
204			if (ifmt > 2) {
205			date_fmt[i++] = ':';
206			date_fmt[i++] = '%';
207			date_fmt[i++] = 'S';
208			}
209			format_utc:
210			date_fmt[i++] = '%';
211			date_fmt[i] = (opt & DATE_OPT_UTC) ? 'Z' : 'z';
212			}
213			} else if (opt & DATE_OPT_RFC2822) {
214			/* Undo busybox.c for date -R */
215			if (ENABLE_LOCALE_SUPPORT)
216			setlocale(LC_TIME, "C");
217			strcpy(date_fmt, "%a, %d %b %Y %H:%M:%S ");
218			i = 22;
219			goto format_utc;
220			} else /* default case */
221			date_fmt = "%a %b %e %H:%M:%S %Z %Y";
222			}
223
224			if (*date_fmt == '\0') {
225			/* With no format string, just print a blank line */
226			*bb_common_bufsiz1 = 0;
227			} else {
228			/* Handle special conversions */
229
230			if (strncmp(date_fmt, "%f", 2) == 0) {
231			date_fmt = "%Y.%m.%d-%H:%M:%S";
232			}
233
234			/* Generate output string */
235			strftime(bb_common_bufsiz1, 200, date_fmt, &tm_time);
236			}
237			puts(bb_common_bufsiz1);
238
239			return EXIT_SUCCESS;
240			}