xref: /illumos-gate/usr/src/cmd/valtools/ckint.c (revision 2a8bcb4e)
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 #include <stdio.h>
31*7c478bd9Sstevel@tonic-gate #include <string.h>
32*7c478bd9Sstevel@tonic-gate #include <signal.h>
33*7c478bd9Sstevel@tonic-gate #include <stdlib.h>
34*7c478bd9Sstevel@tonic-gate #include <locale.h>
35*7c478bd9Sstevel@tonic-gate #include <libintl.h>
36*7c478bd9Sstevel@tonic-gate #include "usage.h"
37*7c478bd9Sstevel@tonic-gate #include "libadm.h"
38*7c478bd9Sstevel@tonic-gate 
39*7c478bd9Sstevel@tonic-gate #define	BADPID (-2)
40*7c478bd9Sstevel@tonic-gate 
41*7c478bd9Sstevel@tonic-gate static char	*prog;
42*7c478bd9Sstevel@tonic-gate static char	*deflt, *prompt, *error, *help;
43*7c478bd9Sstevel@tonic-gate static int	kpid = BADPID;
44*7c478bd9Sstevel@tonic-gate static int	signo;
45*7c478bd9Sstevel@tonic-gate static short	base;
46*7c478bd9Sstevel@tonic-gate 
47*7c478bd9Sstevel@tonic-gate static const char	vusage[] = "b";
48*7c478bd9Sstevel@tonic-gate static const char	husage[] = "bWh";
49*7c478bd9Sstevel@tonic-gate static const char	eusage[] = "bWe";
50*7c478bd9Sstevel@tonic-gate 
51*7c478bd9Sstevel@tonic-gate static void
usage(void)52*7c478bd9Sstevel@tonic-gate usage(void)
53*7c478bd9Sstevel@tonic-gate {
54*7c478bd9Sstevel@tonic-gate 	switch (*prog) {
55*7c478bd9Sstevel@tonic-gate 	default:
56*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
57*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] [-b base]\n"), prog);
58*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
59*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(STDOPTS));
60*7c478bd9Sstevel@tonic-gate 		break;
61*7c478bd9Sstevel@tonic-gate 
62*7c478bd9Sstevel@tonic-gate 	case 'v':
63*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
64*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [-b base] input\n"), prog);
65*7c478bd9Sstevel@tonic-gate 		break;
66*7c478bd9Sstevel@tonic-gate 
67*7c478bd9Sstevel@tonic-gate 	case 'h':
68*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
69*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] [-b base]\n"), prog);
70*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
71*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
72*7c478bd9Sstevel@tonic-gate 			gettext("\t-W width\n\t-h help\n"));
73*7c478bd9Sstevel@tonic-gate 		break;
74*7c478bd9Sstevel@tonic-gate 
75*7c478bd9Sstevel@tonic-gate 	case 'e':
76*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
77*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] [-b base]\n"), prog);
78*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
79*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
80*7c478bd9Sstevel@tonic-gate 			gettext("\t-W width\n\t-e error\n"));
81*7c478bd9Sstevel@tonic-gate 		break;
82*7c478bd9Sstevel@tonic-gate 	}
83*7c478bd9Sstevel@tonic-gate 	exit(1);
84*7c478bd9Sstevel@tonic-gate }
85*7c478bd9Sstevel@tonic-gate 
86*7c478bd9Sstevel@tonic-gate /*
87*7c478bd9Sstevel@tonic-gate  * Given argv[0], return a pointer to the basename of the program.
88*7c478bd9Sstevel@tonic-gate  */
89*7c478bd9Sstevel@tonic-gate static char *
prog_name(char * arg0)90*7c478bd9Sstevel@tonic-gate prog_name(char *arg0)
91*7c478bd9Sstevel@tonic-gate {
92*7c478bd9Sstevel@tonic-gate 	char *str;
93*7c478bd9Sstevel@tonic-gate 
94*7c478bd9Sstevel@tonic-gate 	/* first strip trailing '/' characters (exec() allows these!) */
95*7c478bd9Sstevel@tonic-gate 	str = arg0 + strlen(arg0);
96*7c478bd9Sstevel@tonic-gate 	while (str > arg0 && *--str == '/')
97*7c478bd9Sstevel@tonic-gate 		*str = '\0';
98*7c478bd9Sstevel@tonic-gate 	if ((str = strrchr(arg0, '/')) != NULL)
99*7c478bd9Sstevel@tonic-gate 		return (str + 1);
100*7c478bd9Sstevel@tonic-gate 	return (arg0);
101*7c478bd9Sstevel@tonic-gate }
102*7c478bd9Sstevel@tonic-gate 
103*7c478bd9Sstevel@tonic-gate int
main(int argc,char ** argv)104*7c478bd9Sstevel@tonic-gate main(int argc, char **argv)
105*7c478bd9Sstevel@tonic-gate {
106*7c478bd9Sstevel@tonic-gate 	int	c, n;
107*7c478bd9Sstevel@tonic-gate 	long	intval;
108*7c478bd9Sstevel@tonic-gate 
109*7c478bd9Sstevel@tonic-gate 	(void) setlocale(LC_ALL, "");
110*7c478bd9Sstevel@tonic-gate 
111*7c478bd9Sstevel@tonic-gate #if	!defined(TEXT_DOMAIN)
112*7c478bd9Sstevel@tonic-gate #define	TEXT_DOMAIN	"SYS_TEST"
113*7c478bd9Sstevel@tonic-gate #endif
114*7c478bd9Sstevel@tonic-gate 	(void) textdomain(TEXT_DOMAIN);
115*7c478bd9Sstevel@tonic-gate 
116*7c478bd9Sstevel@tonic-gate 	prog = prog_name(argv[0]);
117*7c478bd9Sstevel@tonic-gate 
118*7c478bd9Sstevel@tonic-gate 	while ((c = getopt(argc, argv, "b:d:p:e:h:k:s:QW:?")) != EOF) {
119*7c478bd9Sstevel@tonic-gate 		/* check for invalid option */
120*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'v') && !strchr(vusage, c))
121*7c478bd9Sstevel@tonic-gate 			usage(); /* no valid options */
122*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'e') && !strchr(eusage, c))
123*7c478bd9Sstevel@tonic-gate 			usage();
124*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'h') && !strchr(husage, c))
125*7c478bd9Sstevel@tonic-gate 			usage();
126*7c478bd9Sstevel@tonic-gate 
127*7c478bd9Sstevel@tonic-gate 		switch (c) {
128*7c478bd9Sstevel@tonic-gate 		case 'Q':
129*7c478bd9Sstevel@tonic-gate 			ckquit = 0;
130*7c478bd9Sstevel@tonic-gate 			break;
131*7c478bd9Sstevel@tonic-gate 
132*7c478bd9Sstevel@tonic-gate 		case 'W':
133*7c478bd9Sstevel@tonic-gate 			ckwidth = atoi(optarg);
134*7c478bd9Sstevel@tonic-gate 			if (ckwidth < 0) {
135*7c478bd9Sstevel@tonic-gate 				(void) fprintf(stderr,
136*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: negative display width specified\n"),
137*7c478bd9Sstevel@tonic-gate 					prog);
138*7c478bd9Sstevel@tonic-gate 				exit(1);
139*7c478bd9Sstevel@tonic-gate 			}
140*7c478bd9Sstevel@tonic-gate 			break;
141*7c478bd9Sstevel@tonic-gate 
142*7c478bd9Sstevel@tonic-gate 		case 'b':
143*7c478bd9Sstevel@tonic-gate 			base = atoi(optarg);
144*7c478bd9Sstevel@tonic-gate 			if ((base < 2) || (base > 36)) {
145*7c478bd9Sstevel@tonic-gate 				(void) fprintf(stderr,
146*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: base must be between 2 and 36\n"),
147*7c478bd9Sstevel@tonic-gate 					prog);
148*7c478bd9Sstevel@tonic-gate 				exit(1);
149*7c478bd9Sstevel@tonic-gate 			}
150*7c478bd9Sstevel@tonic-gate 			break;
151*7c478bd9Sstevel@tonic-gate 
152*7c478bd9Sstevel@tonic-gate 		case 'd':
153*7c478bd9Sstevel@tonic-gate 			deflt = optarg;
154*7c478bd9Sstevel@tonic-gate 			break;
155*7c478bd9Sstevel@tonic-gate 
156*7c478bd9Sstevel@tonic-gate 		case 'p':
157*7c478bd9Sstevel@tonic-gate 			prompt = optarg;
158*7c478bd9Sstevel@tonic-gate 			break;
159*7c478bd9Sstevel@tonic-gate 
160*7c478bd9Sstevel@tonic-gate 		case 'e':
161*7c478bd9Sstevel@tonic-gate 			error = optarg;
162*7c478bd9Sstevel@tonic-gate 			break;
163*7c478bd9Sstevel@tonic-gate 
164*7c478bd9Sstevel@tonic-gate 		case 'h':
165*7c478bd9Sstevel@tonic-gate 			help = optarg;
166*7c478bd9Sstevel@tonic-gate 			break;
167*7c478bd9Sstevel@tonic-gate 
168*7c478bd9Sstevel@tonic-gate 		case 'k':
169*7c478bd9Sstevel@tonic-gate 			kpid = atoi(optarg);
170*7c478bd9Sstevel@tonic-gate 			break;
171*7c478bd9Sstevel@tonic-gate 
172*7c478bd9Sstevel@tonic-gate 		case 's':
173*7c478bd9Sstevel@tonic-gate 			signo = atoi(optarg);
174*7c478bd9Sstevel@tonic-gate 			break;
175*7c478bd9Sstevel@tonic-gate 
176*7c478bd9Sstevel@tonic-gate 		default:
177*7c478bd9Sstevel@tonic-gate 			usage();
178*7c478bd9Sstevel@tonic-gate 		}
179*7c478bd9Sstevel@tonic-gate 	}
180*7c478bd9Sstevel@tonic-gate 
181*7c478bd9Sstevel@tonic-gate 	if (signo) {
182*7c478bd9Sstevel@tonic-gate 		if (kpid == BADPID)
183*7c478bd9Sstevel@tonic-gate 			usage();
184*7c478bd9Sstevel@tonic-gate 	} else
185*7c478bd9Sstevel@tonic-gate 		signo = SIGTERM;
186*7c478bd9Sstevel@tonic-gate 
187*7c478bd9Sstevel@tonic-gate 	if (*prog == 'v') {
188*7c478bd9Sstevel@tonic-gate 		if (argc != (optind + 1))
189*7c478bd9Sstevel@tonic-gate 			usage();
190*7c478bd9Sstevel@tonic-gate 		exit(ckint_val(argv[optind], base));
191*7c478bd9Sstevel@tonic-gate 	}
192*7c478bd9Sstevel@tonic-gate 
193*7c478bd9Sstevel@tonic-gate 	if (optind != argc)
194*7c478bd9Sstevel@tonic-gate 		usage();
195*7c478bd9Sstevel@tonic-gate 
196*7c478bd9Sstevel@tonic-gate 	if (*prog == 'e') {
197*7c478bd9Sstevel@tonic-gate 		ckindent = 0;
198*7c478bd9Sstevel@tonic-gate 		ckint_err(base, error);
199*7c478bd9Sstevel@tonic-gate 		exit(0);
200*7c478bd9Sstevel@tonic-gate 	} else if (*prog == 'h') {
201*7c478bd9Sstevel@tonic-gate 		ckindent = 0;
202*7c478bd9Sstevel@tonic-gate 		ckint_hlp(base, help);
203*7c478bd9Sstevel@tonic-gate 		exit(0);
204*7c478bd9Sstevel@tonic-gate 	}
205*7c478bd9Sstevel@tonic-gate 
206*7c478bd9Sstevel@tonic-gate 	n = ckint(&intval, base, deflt, error, help, prompt);
207*7c478bd9Sstevel@tonic-gate 	if (n == 3) {
208*7c478bd9Sstevel@tonic-gate 		if (kpid > -2)
209*7c478bd9Sstevel@tonic-gate 			(void) kill(kpid, signo);
210*7c478bd9Sstevel@tonic-gate 		(void) puts("q");
211*7c478bd9Sstevel@tonic-gate 	} else if (n == 0)
212*7c478bd9Sstevel@tonic-gate 		(void) printf("%ld", intval);
213*7c478bd9Sstevel@tonic-gate 	return (n);
214*7c478bd9Sstevel@tonic-gate }
215