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
46 static char *prog;
47 static char *deflt = NULL, *prompt = NULL, *error = NULL, *help = NULL;
48 static int kpid = BADPID;
49 static int signo, length;
50
51 static const char vusage[] = "rl";
52 static const char husage[] = "rlWh";
53 static const char eusage[] = "rlWe";
54
55 #define USAGE "[-l length] [[-r regexp] [...]]"
56
57 static void
usage(void)58 usage(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 */
98 static char *
prog_name(char * arg0)99 prog_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
112 int
main(int argc,char ** argv)113 main(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