xref: /illumos-gate/usr/src/lib/libadm/common/ckdate.c (revision 9c88ac3ab4edaa5e8c0130ed1b4b376ea57c545a)
1 /*
2  * CDDL HEADER START
3  *
4  * The contents of this file are subject to the terms of the
5  * Common Development and Distribution License (the "License").
6  * You may not use this file except in compliance with the License.
7  *
8  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
9  * or http://www.opensolaris.org/os/licensing.
10  * See the License for the specific language governing permissions
11  * and limitations under the License.
12  *
13  * When distributing Covered Code, include this CDDL HEADER in each
14  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
15  * If applicable, add the following below this CDDL HEADER, with the
16  * fields enclosed by brackets "[]" replaced with your own identifying
17  * information: Portions Copyright [yyyy] [name of copyright owner]
18  *
19  * CDDL HEADER END
20  */
21 /*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
22 /*	  All Rights Reserved  	*/
23 
24 
25 /*
26  * Copyright (c) 1989, 2010, Oracle and/or its affiliates. All rights reserved.
27  */
28 
29 /*LINTLIBRARY*/
30 
31 #include <stdio.h>
32 #include <string.h>
33 #include <ctype.h>
34 #include <sys/types.h>
35 #include <stdlib.h>
36 #include <limits.h>
37 #include "libadm.h"
38 
39 static int	fmtcheck(char *);
40 
41 #define	MSGSIZ	64
42 #define	PROMPT	"Enter the date"
43 #define	MESG	"Please enter a date"
44 #define	DEFAULT	"%m/%d/%y"
45 
46 static char	*p_ndigit(char *, int *, int);
47 static char	*p_date(char *, int, int, int);
48 static char	*p_eday(char *, int, int);
49 static char	*p_dlm(char *, char);
50 
51 #define	MLIM 10
52 #define	STDIG 2
53 #define	LD2 10
54 #define	LD 01
55 #define	UD 31
56 #define	LM 01
57 #define	UM 12
58 /*
59  * All digits are valid for a YY year format
60  * 70-99 refer to the 20th Century
61  * 00-69 refer to the 21st Century
62  */
63 #define	LY 00
64 #define	UY 99
65 #define	LCY 1970
66 #define	UCY 9999
67 #define	CCYY 4
68 #define	DELIM1 '/'
69 #define	DELIM2 '-'
70 #define	BLANK ' '
71 #define	TAB '	'
72 
73 static void
74 setmsg(char *msg, char *fmt, size_t sz)
75 {
76 	if ((fmt == NULL) || strcmp(fmt, "%D") == 0)
77 		fmt = "%m/%d/%y";
78 	(void) snprintf(msg, sz, "%s. Format is <%s>.", MESG, fmt);
79 }
80 
81 static char *
82 p_ndigit(char *string, int *value, int n)
83 {
84 	char *ptr;
85 	int accum = 0;
86 
87 	if (!string)
88 		return (NULL);
89 	for (ptr = string; *ptr && n > 0; n--, ptr++) {
90 		if (! isdigit((unsigned char)*ptr))
91 			return (NULL);
92 		accum = (10 * accum) + (*ptr - '0');
93 	}
94 	if (n)
95 		return (NULL);
96 	*value = accum;
97 	return (ptr);
98 }
99 
100 static char *
101 p_date(char *string, int llim, int ulim, int ndig)
102 {
103 	char *ptr;
104 	int begin = -1;
105 
106 	if (!(ptr = p_ndigit(string, &begin, ndig)))
107 		return (NULL);
108 	if (begin >= llim && begin <= ulim)
109 		return (ptr);
110 	else
111 		return (NULL);
112 }
113 
114 static char *
115 p_eday(char *string, int llim, int ulim)
116 {
117 	char *ptr, *copy;
118 	int begin = -1;
119 	int iday = 0;
120 	int idaymax = 2;
121 
122 	if (*string == BLANK) {
123 		string++;
124 		idaymax--;
125 	}
126 	copy = string;
127 	while (isdigit((unsigned char)*copy) && (iday < idaymax)) {
128 		copy++;
129 		iday++;
130 	}
131 	if (iday == 1) {
132 		llim = 1;
133 		ulim = 9;
134 	} else if (iday == 2) {
135 		llim = 10;
136 		ulim = 31;
137 	}
138 	if (iday == 0)
139 		return (NULL);
140 
141 	if (!(ptr = p_ndigit(string, &begin, iday)))
142 		return (NULL);
143 
144 	if (begin >= llim && begin <= ulim)
145 		return (ptr);
146 	else
147 		return (NULL);
148 }
149 
150 /* p_month will parse the string for the month - abbr. form i.e. JAN - DEC */
151 
152 static char *
153 p_month(char *string, char mnabr)
154 {
155 	static char *fmonth[] = {
156 		    "JANUARY", "FEBRUARY", "MARCH", "APRIL",
157 		    "MAY", "JUNE", "JULY", "AUGUST",
158 		    "SEPTEMBER", "OCTOBER", "NOVEMBER", "DECEMBER"
159 	};
160 	static char *amonth[] = {
161 		    "JAN", "FEB", "MAR", "APR", "MAY", "JUN",
162 		    "JUL", "AUG", "SEP", "OCT", "NOV", "DEC"
163 	};
164 	int ichng, icnt;
165 	char *mnth[12];
166 	char *copy;
167 	char mletter[MLIM];
168 	int mlen;
169 	int imnth = 0;
170 	int legit = 0;
171 	int n = 0;
172 
173 	if (mnabr == 'a') {
174 		mlen = 3;
175 		for (icnt = 0; icnt < 12; icnt++)
176 			mnth[icnt] = amonth[icnt];
177 	} else {
178 		mlen = 9;
179 		for (icnt = 0; icnt < 12; icnt++)
180 			mnth[icnt] = fmonth[icnt];
181 	}
182 
183 	copy = string;
184 
185 	while (((islower((unsigned char)*copy)) ||
186 	    (isupper((unsigned char)*copy))) && (imnth < mlen)) {
187 		mletter[imnth] = toupper((unsigned char)*copy++);
188 		imnth++;
189 	}
190 	mletter[imnth] = '\0';
191 	while (!(legit) && (n < 12)) {
192 		if (strncmp(mletter, mnth[n],
193 		    (imnth = (int)strlen(mnth[n]))) == 0)
194 			legit = 1;	/* found legitimate string */
195 		n++;
196 	}
197 	if (legit) {
198 		for (ichng = 0; ichng < imnth; ichng++) {
199 			*string = toupper((unsigned char)*string);
200 			string++;
201 		}
202 
203 		return (string);
204 		/*
205 		 * I know this causes side effects, but it's less
206 		 * code  than adding in a copy for string and using that
207 		 */
208 	} else
209 		return (NULL);
210 }
211 
212 static char *
213 p_dlm(char *string, char dchoice)
214 {
215 	char dlm;
216 
217 
218 	if (! string)
219 		return (NULL);
220 	(void) sscanf(string, "%1c", &dlm);
221 	if (dchoice == '/')
222 		return (((dlm == DELIM1) || (dlm == DELIM2)) ? string+1 : NULL);
223 	else
224 		return ((dlm == dchoice) ? string + 1 : NULL);
225 }
226 
227 int
228 ckdate_err(char	*fmt, char *error)
229 {
230 	char	defmesg[MSGSIZ];
231 
232 	if ((fmt != NULL) && (fmtcheck(fmt) == 1))
233 		return (4);
234 	setmsg(defmesg, fmt, MSGSIZ);
235 	puterror(stdout, defmesg, error);
236 	return (0);
237 }
238 
239 int
240 ckdate_hlp(char *fmt, char *help)
241 {
242 	char	defmesg[MSGSIZ];
243 
244 	if ((fmt != NULL) && (fmtcheck(fmt) == 1))
245 		return (4);
246 	setmsg(defmesg, fmt, MSGSIZ);
247 	puthelp(stdout, defmesg, help);
248 	return (0);
249 }
250 
251 /*
252  *	A little state machine that checks out the format to
253  *	make sure it is acceptable.
254  *		return value 1: NG
255  *		return value 0: OK
256  */
257 static int
258 fmtcheck(char *fmt)
259 {
260 	int	percent = 0;
261 
262 	while (*fmt) {
263 		switch (*fmt++) {
264 			case '%': /* previous state must be start or letter */
265 				if (percent == 0)
266 					percent = 1;
267 				else
268 					return (1);
269 				break;
270 			case 'd': /* previous state must be "%" */
271 			case 'e':
272 			case 'm':
273 			case 'y':
274 			case 'Y':
275 			case 'D':
276 			case 'h':
277 			case 'b':
278 			case 'B':
279 				if (percent == 1)
280 					percent = 0;
281 				else
282 					return (1);
283 				break;
284 			case TAB: /* previous state must be start or letter */
285 			case BLANK:
286 			case DELIM1:
287 			case DELIM2:
288 				if (percent == 1)
289 					return (1);
290 				break;
291 			default:
292 				return (1);
293 		}
294 	}
295 	return (percent);
296 }
297 
298 int
299 ckdate_val(char *fmt, char *input)
300 {
301 	char ltrl, dfl;
302 	int valid = 1; 	/* time of day string is valid for format */
303 
304 	if ((fmt != NULL) && (fmtcheck(fmt) == 1))
305 		return (4);
306 
307 	if (fmt == NULL)
308 		fmt = DEFAULT;
309 	ltrl = '\0';
310 	while (*fmt && valid) {
311 		if ((*fmt) == '%') {
312 			fmt++;
313 			switch (*fmt) {
314 				case 'd':
315 					input = p_date(input, LD, UD, STDIG);
316 					if (!input)
317 						valid = 0;
318 					break;
319 
320 				case 'e':
321 					input = p_eday(input, LD2, UD);
322 					if (!input)
323 						valid = 0;
324 					break;
325 
326 				case 'm':
327 					input = p_date(input, LM, UM, STDIG);
328 					if (!input)
329 						valid = 0;
330 					break;
331 
332 				case 'y':
333 					input = p_date(input, LY, UY, STDIG);
334 					if (!input)
335 						valid = 0;
336 					break;
337 
338 				case 'Y':
339 					input = p_date(input, LCY, UCY, CCYY);
340 					if (!input)
341 						valid = 0;
342 					break;
343 
344 				case 'D':
345 					input = p_date(input, LM, UM, STDIG);
346 					if (!input) {
347 						valid = 0;
348 					break;
349 				}
350 				input = p_dlm(input, DELIM1);
351 				if (!input) {
352 					valid = 0;
353 					break;
354 				}
355 				input = p_date(input, LD, UD, STDIG);
356 				if (!input) {
357 					valid = 0;
358 					break;
359 				}
360 				input = p_dlm(input, DELIM1);
361 				if (!input) {
362 					valid = 0;
363 					break;
364 				}
365 				input = p_date(input, LY, UY, STDIG);
366 				if (!input)
367 					valid = 0;
368 				break;
369 
370 				case 'h':
371 				case 'b':
372 					input = p_month(input, 'a');
373 					if (!input)
374 						valid = 0;
375 					break;
376 
377 				case 'B':
378 					input = p_month(input, 'f');
379 					if (!input)
380 						valid = 0;
381 					break;
382 
383 				default:
384 					(void) sscanf(input, "%1c", &ltrl);
385 					input++;
386 			}
387 		} else {
388 			dfl = '\0';
389 			(void) sscanf(input, "%1c", &dfl);
390 			input++;
391 		}
392 		fmt++;
393 	}	 /* end of while fmt and valid */
394 
395 	if ((*fmt == NULL) && ((input != NULL) && *input != 0)) {
396 		if (*input != NULL)
397 			valid = 0;
398 	}
399 	return ((valid == 0));
400 }
401 
402 int
403 ckdate(char *date, char *fmt, char *defstr, char *error, char *help,
404     char *prompt)
405 {
406 	char	defmesg[MSGSIZ];
407 	char	input[MAX_INPUT];
408 	char	*ept, end[128];
409 
410 	ept = end;
411 	*ept = '\0';
412 
413 	if ((fmt != NULL) && (fmtcheck(fmt) == 1))
414 		return (4);
415 
416 	setmsg(defmesg, fmt, MSGSIZ);
417 	(void) sprintf(ept, "[?,q]");
418 
419 	if (!prompt)
420 		prompt = PROMPT;
421 
422 start:
423 	putprmpt(stderr, prompt, NULL, defstr);
424 	if (getinput(input))
425 		return (1);
426 
427 	if (!strlen(input)) {
428 		if (defstr) {
429 			(void) strcpy(date, defstr);
430 			return (0);
431 		}
432 		puterror(stderr, defmesg, error);
433 		goto start;
434 	} else if (strcmp(input, "?") == 0) {
435 		puthelp(stderr, defmesg, help);
436 		goto start;
437 	} else if (ckquit && strcmp(input, "q") == 0) {
438 		return (3);
439 	} else if (ckdate_val(fmt, input)) {
440 		puterror(stderr, defmesg, error);
441 		goto start;
442 	}
443 	(void) strcpy(date, input);
444 	return (0);
445 }
446