xref: /illumos-gate/usr/src/cmd/valtools/ckstr.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 <string.h>
35*7c478bd9Sstevel@tonic-gate #include <signal.h>
36*7c478bd9Sstevel@tonic-gate #include <stdlib.h>
37*7c478bd9Sstevel@tonic-gate #include <locale.h>
38*7c478bd9Sstevel@tonic-gate #include <libintl.h>
39*7c478bd9Sstevel@tonic-gate #include <limits.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 #define	MAXREGEXP	128
45*7c478bd9Sstevel@tonic-gate 
46*7c478bd9Sstevel@tonic-gate static char	*prog;
47*7c478bd9Sstevel@tonic-gate static char	*deflt = NULL, *prompt = NULL, *error = NULL, *help = NULL;
48*7c478bd9Sstevel@tonic-gate static int	kpid = BADPID;
49*7c478bd9Sstevel@tonic-gate static int	signo, length;
50*7c478bd9Sstevel@tonic-gate 
51*7c478bd9Sstevel@tonic-gate static const char	vusage[] = "rl";
52*7c478bd9Sstevel@tonic-gate static const char	husage[] = "rlWh";
53*7c478bd9Sstevel@tonic-gate static const char	eusage[] = "rlWe";
54*7c478bd9Sstevel@tonic-gate 
55*7c478bd9Sstevel@tonic-gate #define	USAGE	"[-l length] [[-r regexp] [...]]"
56*7c478bd9Sstevel@tonic-gate 
57*7c478bd9Sstevel@tonic-gate static void
usage(void)58*7c478bd9Sstevel@tonic-gate usage(void)
59*7c478bd9Sstevel@tonic-gate {
60*7c478bd9Sstevel@tonic-gate 	switch (*prog) {
61*7c478bd9Sstevel@tonic-gate 	default:
62*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
63*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] %s\n"),
64*7c478bd9Sstevel@tonic-gate 			prog, USAGE);
65*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
66*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(STDOPTS));
67*7c478bd9Sstevel@tonic-gate 		break;
68*7c478bd9Sstevel@tonic-gate 
69*7c478bd9Sstevel@tonic-gate 	case 'v':
70*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
71*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s %s input\n"), prog, USAGE);
72*7c478bd9Sstevel@tonic-gate 		break;
73*7c478bd9Sstevel@tonic-gate 
74*7c478bd9Sstevel@tonic-gate 	case 'h':
75*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
76*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] %s\n"),
77*7c478bd9Sstevel@tonic-gate 			prog, USAGE);
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-h help\n"));
81*7c478bd9Sstevel@tonic-gate 		break;
82*7c478bd9Sstevel@tonic-gate 
83*7c478bd9Sstevel@tonic-gate 	case 'e':
84*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
85*7c478bd9Sstevel@tonic-gate 			gettext("usage: %s [options] %s\n"),
86*7c478bd9Sstevel@tonic-gate 			prog, USAGE);
87*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
88*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
89*7c478bd9Sstevel@tonic-gate 			gettext("\t-W width\n\t-e error\n"));
90*7c478bd9Sstevel@tonic-gate 		break;
91*7c478bd9Sstevel@tonic-gate 	}
92*7c478bd9Sstevel@tonic-gate 	exit(1);
93*7c478bd9Sstevel@tonic-gate }
94*7c478bd9Sstevel@tonic-gate 
95*7c478bd9Sstevel@tonic-gate /*
96*7c478bd9Sstevel@tonic-gate  * Given argv[0], return a pointer to the basename of the program.
97*7c478bd9Sstevel@tonic-gate  */
98*7c478bd9Sstevel@tonic-gate static char *
prog_name(char * arg0)99*7c478bd9Sstevel@tonic-gate prog_name(char *arg0)
100*7c478bd9Sstevel@tonic-gate {
101*7c478bd9Sstevel@tonic-gate 	char *str;
102*7c478bd9Sstevel@tonic-gate 
103*7c478bd9Sstevel@tonic-gate 	/* first strip trailing '/' characters (exec() allows these!) */
104*7c478bd9Sstevel@tonic-gate 	str = arg0 + strlen(arg0);
105*7c478bd9Sstevel@tonic-gate 	while (str > arg0 && *--str == '/')
106*7c478bd9Sstevel@tonic-gate 		*str = '\0';
107*7c478bd9Sstevel@tonic-gate 	if ((str = strrchr(arg0, '/')) != NULL)
108*7c478bd9Sstevel@tonic-gate 		return (str + 1);
109*7c478bd9Sstevel@tonic-gate 	return (arg0);
110*7c478bd9Sstevel@tonic-gate }
111*7c478bd9Sstevel@tonic-gate 
112*7c478bd9Sstevel@tonic-gate int
main(int argc,char ** argv)113*7c478bd9Sstevel@tonic-gate main(int argc, char **argv)
114*7c478bd9Sstevel@tonic-gate {
115*7c478bd9Sstevel@tonic-gate 	int c, n;
116*7c478bd9Sstevel@tonic-gate 	char	*strval;
117*7c478bd9Sstevel@tonic-gate 	char	**regexp;
118*7c478bd9Sstevel@tonic-gate 	size_t	len;
119*7c478bd9Sstevel@tonic-gate 	size_t	maxregexp = MAXREGEXP;
120*7c478bd9Sstevel@tonic-gate 	size_t	nregexp = 0;
121*7c478bd9Sstevel@tonic-gate 
122*7c478bd9Sstevel@tonic-gate 	(void) setlocale(LC_ALL, "");
123*7c478bd9Sstevel@tonic-gate 
124*7c478bd9Sstevel@tonic-gate #if	!defined(TEXT_DOMAIN)
125*7c478bd9Sstevel@tonic-gate #define	TEXT_DOMAIN	"SYS_TEST"
126*7c478bd9Sstevel@tonic-gate #endif
127*7c478bd9Sstevel@tonic-gate 	(void) textdomain(TEXT_DOMAIN);
128*7c478bd9Sstevel@tonic-gate 
129*7c478bd9Sstevel@tonic-gate 	prog = prog_name(argv[0]);
130*7c478bd9Sstevel@tonic-gate 
131*7c478bd9Sstevel@tonic-gate 	regexp = (char **)calloc(maxregexp, sizeof (char *));
132*7c478bd9Sstevel@tonic-gate 	if (!regexp) {
133*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
134*7c478bd9Sstevel@tonic-gate 			gettext("Not enough memory\n"));
135*7c478bd9Sstevel@tonic-gate 		exit(1);
136*7c478bd9Sstevel@tonic-gate 	}
137*7c478bd9Sstevel@tonic-gate 	while ((c = getopt(argc, argv, "r:l:d:p:e:h:k:s:QW:?")) != EOF) {
138*7c478bd9Sstevel@tonic-gate 		/* check for invalid option */
139*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'v') && !strchr(vusage, c))
140*7c478bd9Sstevel@tonic-gate 			usage();
141*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'e') && !strchr(eusage, c))
142*7c478bd9Sstevel@tonic-gate 			usage();
143*7c478bd9Sstevel@tonic-gate 		if ((*prog == 'h') && !strchr(husage, c))
144*7c478bd9Sstevel@tonic-gate 			usage();
145*7c478bd9Sstevel@tonic-gate 
146*7c478bd9Sstevel@tonic-gate 		switch (c) {
147*7c478bd9Sstevel@tonic-gate 		case 'Q':
148*7c478bd9Sstevel@tonic-gate 			ckquit = 0;
149*7c478bd9Sstevel@tonic-gate 			break;
150*7c478bd9Sstevel@tonic-gate 
151*7c478bd9Sstevel@tonic-gate 		case 'W':
152*7c478bd9Sstevel@tonic-gate 			ckwidth = atoi(optarg);
153*7c478bd9Sstevel@tonic-gate 			if (ckwidth < 0) {
154*7c478bd9Sstevel@tonic-gate 				(void) fprintf(stderr,
155*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: negative display width specified\n"),
156*7c478bd9Sstevel@tonic-gate 					prog);
157*7c478bd9Sstevel@tonic-gate 				exit(1);
158*7c478bd9Sstevel@tonic-gate 			}
159*7c478bd9Sstevel@tonic-gate 			break;
160*7c478bd9Sstevel@tonic-gate 
161*7c478bd9Sstevel@tonic-gate 		case 'r':
162*7c478bd9Sstevel@tonic-gate 			regexp[nregexp++] = optarg;
163*7c478bd9Sstevel@tonic-gate 			if (nregexp == maxregexp) {
164*7c478bd9Sstevel@tonic-gate 				maxregexp += MAXREGEXP;
165*7c478bd9Sstevel@tonic-gate 				regexp = (char **)realloc(regexp,
166*7c478bd9Sstevel@tonic-gate 					maxregexp * sizeof (char *));
167*7c478bd9Sstevel@tonic-gate 				if (!regexp) {
168*7c478bd9Sstevel@tonic-gate 					(void) fprintf(stderr,
169*7c478bd9Sstevel@tonic-gate 						gettext("Not enough memory\n"));
170*7c478bd9Sstevel@tonic-gate 					exit(1);
171*7c478bd9Sstevel@tonic-gate 				}
172*7c478bd9Sstevel@tonic-gate 				(void) memset(regexp + nregexp, 0,
173*7c478bd9Sstevel@tonic-gate 					(maxregexp - nregexp) *
174*7c478bd9Sstevel@tonic-gate 					sizeof (char *));
175*7c478bd9Sstevel@tonic-gate 			}
176*7c478bd9Sstevel@tonic-gate 			break;
177*7c478bd9Sstevel@tonic-gate 
178*7c478bd9Sstevel@tonic-gate 		case 'l':
179*7c478bd9Sstevel@tonic-gate 			length = atoi(optarg);
180*7c478bd9Sstevel@tonic-gate 			if ((length <= 0) || (length > 128)) {
181*7c478bd9Sstevel@tonic-gate 				(void) fprintf(stderr,
182*7c478bd9Sstevel@tonic-gate 		gettext("%s: ERROR: length must be between 1 and 128\n"),
183*7c478bd9Sstevel@tonic-gate 					prog);
184*7c478bd9Sstevel@tonic-gate 				exit(1);
185*7c478bd9Sstevel@tonic-gate 			}
186*7c478bd9Sstevel@tonic-gate 			break;
187*7c478bd9Sstevel@tonic-gate 
188*7c478bd9Sstevel@tonic-gate 		case 'd':
189*7c478bd9Sstevel@tonic-gate 			deflt = optarg;
190*7c478bd9Sstevel@tonic-gate 			break;
191*7c478bd9Sstevel@tonic-gate 
192*7c478bd9Sstevel@tonic-gate 		case 'p':
193*7c478bd9Sstevel@tonic-gate 			prompt = optarg;
194*7c478bd9Sstevel@tonic-gate 			break;
195*7c478bd9Sstevel@tonic-gate 
196*7c478bd9Sstevel@tonic-gate 		case 'e':
197*7c478bd9Sstevel@tonic-gate 			error = optarg;
198*7c478bd9Sstevel@tonic-gate 			break;
199*7c478bd9Sstevel@tonic-gate 
200*7c478bd9Sstevel@tonic-gate 		case 'h':
201*7c478bd9Sstevel@tonic-gate 			help = optarg;
202*7c478bd9Sstevel@tonic-gate 			break;
203*7c478bd9Sstevel@tonic-gate 
204*7c478bd9Sstevel@tonic-gate 		case 'k':
205*7c478bd9Sstevel@tonic-gate 			kpid = atoi(optarg);
206*7c478bd9Sstevel@tonic-gate 			break;
207*7c478bd9Sstevel@tonic-gate 
208*7c478bd9Sstevel@tonic-gate 		case 's':
209*7c478bd9Sstevel@tonic-gate 			signo = atoi(optarg);
210*7c478bd9Sstevel@tonic-gate 			break;
211*7c478bd9Sstevel@tonic-gate 
212*7c478bd9Sstevel@tonic-gate 		default:
213*7c478bd9Sstevel@tonic-gate 			usage();
214*7c478bd9Sstevel@tonic-gate 		}
215*7c478bd9Sstevel@tonic-gate 	}
216*7c478bd9Sstevel@tonic-gate 
217*7c478bd9Sstevel@tonic-gate 	if (signo) {
218*7c478bd9Sstevel@tonic-gate 		if (kpid == BADPID)
219*7c478bd9Sstevel@tonic-gate 			usage();
220*7c478bd9Sstevel@tonic-gate 	} else
221*7c478bd9Sstevel@tonic-gate 		signo = SIGTERM;
222*7c478bd9Sstevel@tonic-gate 
223*7c478bd9Sstevel@tonic-gate 	if (*prog == 'v') {
224*7c478bd9Sstevel@tonic-gate 		if (argc != (optind+1))
225*7c478bd9Sstevel@tonic-gate 			usage();
226*7c478bd9Sstevel@tonic-gate 		if (ckstr_val(regexp, length, argv[optind]))
227*7c478bd9Sstevel@tonic-gate 			exit(1);
228*7c478bd9Sstevel@tonic-gate 		exit(0);
229*7c478bd9Sstevel@tonic-gate 	}
230*7c478bd9Sstevel@tonic-gate 
231*7c478bd9Sstevel@tonic-gate 	if (*prog == 'e') {
232*7c478bd9Sstevel@tonic-gate 		if (argc > (optind+1))
233*7c478bd9Sstevel@tonic-gate 			usage(); /* too many args */
234*7c478bd9Sstevel@tonic-gate 		ckindent = 0;
235*7c478bd9Sstevel@tonic-gate 		ckstr_err(regexp, length, error, argv[optind]);
236*7c478bd9Sstevel@tonic-gate 		exit(0);
237*7c478bd9Sstevel@tonic-gate 	}
238*7c478bd9Sstevel@tonic-gate 
239*7c478bd9Sstevel@tonic-gate 	if (optind != argc)
240*7c478bd9Sstevel@tonic-gate 		usage();
241*7c478bd9Sstevel@tonic-gate 
242*7c478bd9Sstevel@tonic-gate 	if (*prog == 'h') {
243*7c478bd9Sstevel@tonic-gate 		ckindent = 0;
244*7c478bd9Sstevel@tonic-gate 		ckstr_hlp(regexp, length, help);
245*7c478bd9Sstevel@tonic-gate 		exit(0);
246*7c478bd9Sstevel@tonic-gate 	}
247*7c478bd9Sstevel@tonic-gate 
248*7c478bd9Sstevel@tonic-gate 	regexp[nregexp] = NULL;
249*7c478bd9Sstevel@tonic-gate 
250*7c478bd9Sstevel@tonic-gate 	if (deflt) {
251*7c478bd9Sstevel@tonic-gate 		len = strlen(deflt) + 1;
252*7c478bd9Sstevel@tonic-gate 		if (len < MAX_INPUT)
253*7c478bd9Sstevel@tonic-gate 			len = MAX_INPUT;
254*7c478bd9Sstevel@tonic-gate 	} else {
255*7c478bd9Sstevel@tonic-gate 		len = MAX_INPUT;
256*7c478bd9Sstevel@tonic-gate 	}
257*7c478bd9Sstevel@tonic-gate 	strval = (char *)malloc(len);
258*7c478bd9Sstevel@tonic-gate 	if (!strval) {
259*7c478bd9Sstevel@tonic-gate 		(void) fprintf(stderr,
260*7c478bd9Sstevel@tonic-gate 			gettext("Not enough memory\n"));
261*7c478bd9Sstevel@tonic-gate 		exit(1);
262*7c478bd9Sstevel@tonic-gate 	}
263*7c478bd9Sstevel@tonic-gate 	n = ckstr(strval, regexp, length, deflt, error, help, prompt);
264*7c478bd9Sstevel@tonic-gate 	if (n == 3) {
265*7c478bd9Sstevel@tonic-gate 		if (kpid > -2)
266*7c478bd9Sstevel@tonic-gate 			(void) kill(kpid, signo);
267*7c478bd9Sstevel@tonic-gate 		(void) puts("q");
268*7c478bd9Sstevel@tonic-gate 	} else if (n == 0)
269*7c478bd9Sstevel@tonic-gate 		(void) fputs(strval, stdout);
270*7c478bd9Sstevel@tonic-gate 	return (n);
271*7c478bd9Sstevel@tonic-gate }
272