xref: /illumos-gate/usr/src/cmd/valtools/ckrange.c (revision 7c478bd9)
1*7c478bd9Sstevel@tonic-gate /*
2*7c478bd9Sstevel@tonic-gate  * CDDL HEADER START
3*7c478bd9Sstevel@tonic-gate  *
4*7c478bd9Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*7c478bd9Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*7c478bd9Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*7c478bd9Sstevel@tonic-gate  * with the License.
8*7c478bd9Sstevel@tonic-gate  *
9*7c478bd9Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*7c478bd9Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*7c478bd9Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*7c478bd9Sstevel@tonic-gate  * and limitations under the License.
13*7c478bd9Sstevel@tonic-gate  *
14*7c478bd9Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*7c478bd9Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*7c478bd9Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*7c478bd9Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*7c478bd9Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*7c478bd9Sstevel@tonic-gate  *
20*7c478bd9Sstevel@tonic-gate  * CDDL HEADER END
21*7c478bd9Sstevel@tonic-gate  */
22*7c478bd9Sstevel@tonic-gate /*
23*7c478bd9Sstevel@tonic-gate  * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
24*7c478bd9Sstevel@tonic-gate  * Use is subject to license terms.
25*7c478bd9Sstevel@tonic-gate  */
26*7c478bd9Sstevel@tonic-gate 
27*7c478bd9Sstevel@tonic-gate /*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
28*7c478bd9Sstevel@tonic-gate /*	  All Rights Reserved  	*/
29*7c478bd9Sstevel@tonic-gate 
30*7c478bd9Sstevel@tonic-gate 
31*7c478bd9Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
32*7c478bd9Sstevel@tonic-gate 
33*7c478bd9Sstevel@tonic-gate #include <stdio.h>
34*7c478bd9Sstevel@tonic-gate #include <limits.h>
35*7c478bd9Sstevel@tonic-gate #include <string.h>
36*7c478bd9Sstevel@tonic-gate #include <signal.h>
37*7c478bd9Sstevel@tonic-gate #include <stdlib.h>
38*7c478bd9Sstevel@tonic-gate #include <locale.h>
39*7c478bd9Sstevel@tonic-gate #include <libintl.h>
40*7c478bd9Sstevel@tonic-gate #include "usage.h"
41*7c478bd9Sstevel@tonic-gate #include "libadm.h"
42*7c478bd9Sstevel@tonic-gate 
43*7c478bd9Sstevel@tonic-gate #define	BADPID	(-2)
44*7c478bd9Sstevel@tonic-gate 
45*7c478bd9Sstevel@tonic-gate static char	*prog;
46*7c478bd9Sstevel@tonic-gate static char	*deflt = NULL, *prompt = NULL, *error = NULL, *help = NULL;
47*7c478bd9Sstevel@tonic-gate static int	kpid = BADPID;
48*7c478bd9Sstevel@tonic-gate static int	signo;
49*7c478bd9Sstevel@tonic-gate static int	base = 10;
50*7c478bd9Sstevel@tonic-gate static char	*upper;
51*7c478bd9Sstevel@tonic-gate static char	*lower;
52*7c478bd9Sstevel@tonic-gate 
53*7c478bd9Sstevel@tonic-gate static const char	vusage[] = "bul";
54*7c478bd9Sstevel@tonic-gate static const char	husage[] = "bulWh";
55*7c478bd9Sstevel@tonic-gate static const char	eusage[] = "bulWe";
56*7c478bd9Sstevel@tonic-gate 
57*7c478bd9Sstevel@tonic-gate #define	USAGE	"[-l lower] [-u upper] [-b base]"
58*7c478bd9Sstevel@tonic-gate 
59*7c478bd9Sstevel@tonic-gate static void
usage(void)60*7c478bd9Sstevel@tonic-gate usage(void)
61*7c478bd9Sstevel@tonic-gate {
62*7c478bd9Sstevel@tonic-gate 	switch (*prog) {
63*7c478bd9Sstevel@tonic-gate 	default:
64*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
65*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] %s\n"),
66*7c478bd9Sstevel@tonic-gate 			prog, USAGE);
67*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
68*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(STDOPTS));
69*7c478bd9Sstevel@tonic-gate 		break;
70*7c478bd9Sstevel@tonic-gate 
71*7c478bd9Sstevel@tonic-gate 	case 'v':
72*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
73*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s %s input\n"), prog, USAGE);
74*7c478bd9Sstevel@tonic-gate 		break;
75*7c478bd9Sstevel@tonic-gate 
76*7c478bd9Sstevel@tonic-gate 	case 'h':
77*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
78*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] %s\n"),
79*7c478bd9Sstevel@tonic-gate 			prog, USAGE);
80*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
81*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
82*7c478bd9Sstevel@tonic-gate 			gettext("\t-W width\n\t-h help\n"));
83*7c478bd9Sstevel@tonic-gate 		break;
84*7c478bd9Sstevel@tonic-gate 
85*7c478bd9Sstevel@tonic-gate 	case 'e':
86*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
87*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] %s\n"),
88*7c478bd9Sstevel@tonic-gate 			prog, USAGE);
89*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
90*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
91*7c478bd9Sstevel@tonic-gate 			gettext("\t-W width\n\t-e error\n"));
92*7c478bd9Sstevel@tonic-gate 		break;
93*7c478bd9Sstevel@tonic-gate 	}
94*7c478bd9Sstevel@tonic-gate 	exit(1);
95*7c478bd9Sstevel@tonic-gate }
96*7c478bd9Sstevel@tonic-gate 
97*7c478bd9Sstevel@tonic-gate /*
98*7c478bd9Sstevel@tonic-gate  * Given argv[0], return a pointer to the basename of the program.
99*7c478bd9Sstevel@tonic-gate  */
100*7c478bd9Sstevel@tonic-gate static char *
prog_name(char * arg0)101*7c478bd9Sstevel@tonic-gate prog_name(char *arg0)
102*7c478bd9Sstevel@tonic-gate {
103*7c478bd9Sstevel@tonic-gate 	char *str;
104*7c478bd9Sstevel@tonic-gate 
105*7c478bd9Sstevel@tonic-gate 	/* first strip trailing '/' characters (exec() allows these!) */
106*7c478bd9Sstevel@tonic-gate 	str = arg0 + strlen(arg0);
107*7c478bd9Sstevel@tonic-gate 	while (str > arg0 && *--str == '/')
108*7c478bd9Sstevel@tonic-gate 		*str = '\0';
109*7c478bd9Sstevel@tonic-gate 	if ((str = strrchr(arg0, '/')) != NULL)
110*7c478bd9Sstevel@tonic-gate 		return (str + 1);
111*7c478bd9Sstevel@tonic-gate 	return (arg0);
112*7c478bd9Sstevel@tonic-gate }
113*7c478bd9Sstevel@tonic-gate 
114*7c478bd9Sstevel@tonic-gate int
main(int argc,char ** argv)115*7c478bd9Sstevel@tonic-gate main(int argc, char **argv)
116*7c478bd9Sstevel@tonic-gate {
117*7c478bd9Sstevel@tonic-gate 	int	c, n;
118*7c478bd9Sstevel@tonic-gate 	long	lvalue, uvalue, intval;
119*7c478bd9Sstevel@tonic-gate 	char	*ptr = 0;
120*7c478bd9Sstevel@tonic-gate 
121*7c478bd9Sstevel@tonic-gate 	(void) setlocale(LC_ALL, "");
122*7c478bd9Sstevel@tonic-gate 
123*7c478bd9Sstevel@tonic-gate #if	!defined(TEXT_DOMAIN)
124*7c478bd9Sstevel@tonic-gate #define	TEXT_DOMAIN	"SYS_TEST"
125*7c478bd9Sstevel@tonic-gate #endif
126*7c478bd9Sstevel@tonic-gate 	(void) textdomain(TEXT_DOMAIN);
127*7c478bd9Sstevel@tonic-gate 
128*7c478bd9Sstevel@tonic-gate 	prog = prog_name(argv[0]);
129*7c478bd9Sstevel@tonic-gate 
130*7c478bd9Sstevel@tonic-gate 	while ((c = getopt(argc, argv, "l:u:b:d:p:e:h:k:s:QW:?")) != EOF) {
131*7c478bd9Sstevel@tonic-gate 		/* check for invalid option */
132*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'v') && !strchr(vusage, c))
133*7c478bd9Sstevel@tonic-gate 			usage();
134*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'e') && !strchr(eusage, c))
135*7c478bd9Sstevel@tonic-gate 			usage();
136*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'h') && !strchr(husage, c))
137*7c478bd9Sstevel@tonic-gate 			usage();
138*7c478bd9Sstevel@tonic-gate 
139*7c478bd9Sstevel@tonic-gate 		switch (c) {
140*7c478bd9Sstevel@tonic-gate 		case 'Q':
141*7c478bd9Sstevel@tonic-gate 			ckquit = 0;
142*7c478bd9Sstevel@tonic-gate 			break;
143*7c478bd9Sstevel@tonic-gate 
144*7c478bd9Sstevel@tonic-gate 		case 'W':
145*7c478bd9Sstevel@tonic-gate 			ckwidth = atoi(optarg);
146*7c478bd9Sstevel@tonic-gate 			if (ckwidth < 0) {
147*7c478bd9Sstevel@tonic-gate 				(void) fprintf(stderr,
148*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: negative display width specified\n"),
149*7c478bd9Sstevel@tonic-gate 					prog);
150*7c478bd9Sstevel@tonic-gate 				exit(1);
151*7c478bd9Sstevel@tonic-gate 			}
152*7c478bd9Sstevel@tonic-gate 			break;
153*7c478bd9Sstevel@tonic-gate 
154*7c478bd9Sstevel@tonic-gate 		case 'b':
155*7c478bd9Sstevel@tonic-gate 			base = atoi(optarg);
156*7c478bd9Sstevel@tonic-gate 			if ((base < 2) || (base > 36)) {
157*7c478bd9Sstevel@tonic-gate 				(void) fprintf(stderr,
158*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: base must be between 2 and 36\n"),
159*7c478bd9Sstevel@tonic-gate 					prog);
160*7c478bd9Sstevel@tonic-gate 				exit(1);
161*7c478bd9Sstevel@tonic-gate 			}
162*7c478bd9Sstevel@tonic-gate 			break;
163*7c478bd9Sstevel@tonic-gate 
164*7c478bd9Sstevel@tonic-gate 		case 'u':
165*7c478bd9Sstevel@tonic-gate 			upper = optarg;
166*7c478bd9Sstevel@tonic-gate 			break;
167*7c478bd9Sstevel@tonic-gate 
168*7c478bd9Sstevel@tonic-gate 		case 'l':
169*7c478bd9Sstevel@tonic-gate 			lower = optarg;
170*7c478bd9Sstevel@tonic-gate 			break;
171*7c478bd9Sstevel@tonic-gate 
172*7c478bd9Sstevel@tonic-gate 		case 'd':
173*7c478bd9Sstevel@tonic-gate 			deflt = optarg;
174*7c478bd9Sstevel@tonic-gate 			break;
175*7c478bd9Sstevel@tonic-gate 
176*7c478bd9Sstevel@tonic-gate 		case 'p':
177*7c478bd9Sstevel@tonic-gate 			prompt = optarg;
178*7c478bd9Sstevel@tonic-gate 			break;
179*7c478bd9Sstevel@tonic-gate 
180*7c478bd9Sstevel@tonic-gate 		case 'e':
181*7c478bd9Sstevel@tonic-gate 			error = optarg;
182*7c478bd9Sstevel@tonic-gate 			break;
183*7c478bd9Sstevel@tonic-gate 
184*7c478bd9Sstevel@tonic-gate 		case 'h':
185*7c478bd9Sstevel@tonic-gate 			help = optarg;
186*7c478bd9Sstevel@tonic-gate 			break;
187*7c478bd9Sstevel@tonic-gate 
188*7c478bd9Sstevel@tonic-gate 		case 'k':
189*7c478bd9Sstevel@tonic-gate 			kpid = atoi(optarg);
190*7c478bd9Sstevel@tonic-gate 			break;
191*7c478bd9Sstevel@tonic-gate 
192*7c478bd9Sstevel@tonic-gate 		case 's':
193*7c478bd9Sstevel@tonic-gate 			signo = atoi(optarg);
194*7c478bd9Sstevel@tonic-gate 			break;
195*7c478bd9Sstevel@tonic-gate 
196*7c478bd9Sstevel@tonic-gate 		default:
197*7c478bd9Sstevel@tonic-gate 			usage();
198*7c478bd9Sstevel@tonic-gate 		}
199*7c478bd9Sstevel@tonic-gate 	}
200*7c478bd9Sstevel@tonic-gate 
201*7c478bd9Sstevel@tonic-gate 	if (signo) {
202*7c478bd9Sstevel@tonic-gate 		if (kpid == BADPID)
203*7c478bd9Sstevel@tonic-gate 			usage();
204*7c478bd9Sstevel@tonic-gate 	} else
205*7c478bd9Sstevel@tonic-gate 		signo = SIGTERM;
206*7c478bd9Sstevel@tonic-gate 
207*7c478bd9Sstevel@tonic-gate 	if (upper) {
208*7c478bd9Sstevel@tonic-gate 		uvalue = strtol(upper, &ptr, base);
209*7c478bd9Sstevel@tonic-gate 		if (ptr == upper) {
210*7c478bd9Sstevel@tonic-gate 			(void) fprintf(stderr,
211*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: invalid upper value specification\n"),
212*7c478bd9Sstevel@tonic-gate 				prog);
213*7c478bd9Sstevel@tonic-gate 			exit(1);
214*7c478bd9Sstevel@tonic-gate 		}
215*7c478bd9Sstevel@tonic-gate 	} else
216*7c478bd9Sstevel@tonic-gate 		uvalue = LONG_MAX;
217*7c478bd9Sstevel@tonic-gate 	if (lower) {
218*7c478bd9Sstevel@tonic-gate 		lvalue =  strtol(lower, &ptr, base);
219*7c478bd9Sstevel@tonic-gate 		if (ptr == lower) {
220*7c478bd9Sstevel@tonic-gate 			(void) fprintf(stderr,
221*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: invalid lower value specification\n"),
222*7c478bd9Sstevel@tonic-gate 				prog);
223*7c478bd9Sstevel@tonic-gate 			exit(1);
224*7c478bd9Sstevel@tonic-gate 		}
225*7c478bd9Sstevel@tonic-gate 	} else
226*7c478bd9Sstevel@tonic-gate 		lvalue = LONG_MIN;
227*7c478bd9Sstevel@tonic-gate 
228*7c478bd9Sstevel@tonic-gate 	if (uvalue < lvalue) {
229*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
230*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: upper value is less than lower value\n"),
231*7c478bd9Sstevel@tonic-gate 			prog);
232*7c478bd9Sstevel@tonic-gate 		exit(1);
233*7c478bd9Sstevel@tonic-gate 	}
234*7c478bd9Sstevel@tonic-gate 
235*7c478bd9Sstevel@tonic-gate 	if (*prog == 'v') {
236*7c478bd9Sstevel@tonic-gate 		if (argc != (optind+1))
237*7c478bd9Sstevel@tonic-gate 			usage();
238*7c478bd9Sstevel@tonic-gate 		exit(ckrange_val(lvalue, uvalue, base, argv[optind]));
239*7c478bd9Sstevel@tonic-gate 	}
240*7c478bd9Sstevel@tonic-gate 
241*7c478bd9Sstevel@tonic-gate 	if (optind != argc)
242*7c478bd9Sstevel@tonic-gate 		usage();
243*7c478bd9Sstevel@tonic-gate 
244*7c478bd9Sstevel@tonic-gate 	if (*prog == 'e') {
245*7c478bd9Sstevel@tonic-gate 		ckindent = 0;
246*7c478bd9Sstevel@tonic-gate 		ckrange_err(lvalue, uvalue, base, error);
247*7c478bd9Sstevel@tonic-gate 		exit(0);
248*7c478bd9Sstevel@tonic-gate 	} else if (*prog == 'h') {
249*7c478bd9Sstevel@tonic-gate 		ckindent = 0;
250*7c478bd9Sstevel@tonic-gate 		ckrange_hlp(lvalue, uvalue, base, help);
251*7c478bd9Sstevel@tonic-gate 		exit(0);
252*7c478bd9Sstevel@tonic-gate 	}
253*7c478bd9Sstevel@tonic-gate 
254*7c478bd9Sstevel@tonic-gate 	n = ckrange(&intval, lvalue, uvalue, (short)base,
255*7c478bd9Sstevel@tonic-gate 		deflt, error, help, prompt);	/* libadm interface */
256*7c478bd9Sstevel@tonic-gate 	if (n == 3) {
257*7c478bd9Sstevel@tonic-gate 		if (kpid > -2)
258*7c478bd9Sstevel@tonic-gate 			(void) kill(kpid, signo);
259*7c478bd9Sstevel@tonic-gate 		(void) puts("q");
260*7c478bd9Sstevel@tonic-gate 	} else if (n == 0)
261*7c478bd9Sstevel@tonic-gate 		(void) printf("%ld", intval);
262*7c478bd9Sstevel@tonic-gate 	return (n);
263*7c478bd9Sstevel@tonic-gate }
264