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