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