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 <ctype.h>
35 #include <string.h>
36 #include <signal.h>
37 #include <valtools.h>
38 #include <stdlib.h>
39 #include <locale.h>
40 #include <libintl.h>
41 #include <limits.h>
42 #include <wchar.h>
43 #include "usage.h"
44 #include "libadm.h"
45
46 #define BADPID (-2)
47 #define INVISMAXSIZE 36
48
49 static char *prog;
50 static char *deflt = NULL, *prompt = NULL, *error = NULL, *help = NULL;
51 static int kpid = BADPID;
52 static int signo;
53
54 static char *label, **invis;
55 static int ninvis = 0;
56 static int max = 1;
57 static int attr = CKALPHA;
58
59 #define MAXSIZE 128
60 #define LSIZE 1024
61 #define INTERR \
62 "%s: ERROR: internal error occurred while attempting menu setup\n"
63 #define MYOPTS \
64 "\t-f file #file containing choices\n" \
65 "\t-l label #menu label\n" \
66 "\t-i invis [, ...] #invisible menu choices\n" \
67 "\t-m max #maximum choices user may select\n" \
68 "\t-n #do not sort choices alphabetically\n" \
69 "\t-o #don't prompt if only one choice\n" \
70 "\t-u #unnumbered choices\n"
71
72 static const char husage[] = "Wh";
73 static const char eusage[] = "We";
74
75 static void
usage(void)76 usage(void)
77 {
78 switch (*prog) {
79 default:
80 (void) fprintf(stderr,
81 gettext("usage: %s [options] [choice [...]]\n"), prog);
82 (void) fprintf(stderr, gettext(OPTMESG));
83 (void) fprintf(stderr, gettext(MYOPTS));
84 (void) fprintf(stderr, gettext(STDOPTS));
85 break;
86
87 case 'h':
88 (void) fprintf(stderr,
89 gettext("usage: %s [options] [choice [...]]\n"), prog);
90 (void) fprintf(stderr, gettext(OPTMESG));
91 (void) fprintf(stderr,
92 gettext("\t-W width\n\t-h help\n"));
93 break;
94
95 case 'e':
96 (void) fprintf(stderr,
97 gettext("usage: %s [options] [choice [...]]\n"), prog);
98 (void) fprintf(stderr, gettext(OPTMESG));
99 (void) fprintf(stderr,
100 gettext("\t-W width\n\t-e error\n"));
101 break;
102 }
103 exit(1);
104 }
105
106 /*
107 * Given argv[0], return a pointer to the basename of the program.
108 */
109 static char *
prog_name(char * arg0)110 prog_name(char *arg0)
111 {
112 char *str;
113
114 /* first strip trailing '/' characters (exec() allows these!) */
115 str = arg0 + strlen(arg0);
116 while (str > arg0 && *--str == '/')
117 *str = '\0';
118 if ((str = strrchr(arg0, '/')) != NULL)
119 return (str + 1);
120 return (arg0);
121 }
122
123 int
main(int argc,char ** argv)124 main(int argc, char **argv)
125 {
126 CKMENU *mp;
127 FILE *fp = NULL;
128 int c, i;
129 char **item;
130 char temp[LSIZE * MB_LEN_MAX];
131 size_t mmax;
132 size_t invismaxsize = INVISMAXSIZE;
133 size_t n, r;
134 wchar_t wline[LSIZE], wtemp[LSIZE];
135
136 (void) setlocale(LC_ALL, "");
137
138 #if !defined(TEXT_DOMAIN)
139 #define TEXT_DOMAIN "SYS_TEST"
140 #endif
141 (void) textdomain(TEXT_DOMAIN);
142
143 prog = prog_name(argv[0]);
144
145 invis = (char **)calloc(invismaxsize, sizeof (char *));
146 if (!invis) {
147 (void) fprintf(stderr,
148 gettext("Not enough memory\n"));
149 exit(1);
150 }
151 while ((c = getopt(argc, argv, "m:oni:l:f:ud:p:e:h:k:s:QW:?")) != EOF) {
152 /* check for invalid option */
153 if ((*prog == 'e') && !strchr(eusage, c))
154 usage(); /* no valid options */
155 if ((*prog == 'h') && !strchr(husage, c))
156 usage();
157
158 switch (c) {
159 case 'Q':
160 ckquit = 0;
161 break;
162
163 case 'W':
164 ckwidth = atol(optarg);
165 if (ckwidth < 0) {
166 (void) fprintf(stderr,
167 gettext("%s: ERROR: negative display width specified\n"),
168 prog);
169 exit(1);
170 }
171 break;
172
173 case 'm':
174 max = atoi(optarg);
175 if (max > SHRT_MAX || max < SHRT_MIN) {
176 (void) fprintf(stderr,
177 gettext("%s: ERROR: too large or too small max value specified\n"),
178 prog);
179 exit(1);
180 }
181 break;
182
183 case 'o':
184 attr |= CKONEFLAG;
185 break;
186
187 case 'n':
188 attr &= ~CKALPHA;
189 break;
190
191 case 'i':
192 invis[ninvis++] = optarg;
193 if (ninvis == invismaxsize) {
194 invismaxsize += INVISMAXSIZE;
195 invis = (char **)realloc(invis,
196 invismaxsize * sizeof (char *));
197 if (!invis) {
198 (void) fprintf(stderr,
199 gettext("Not enough memory\n"));
200 exit(1);
201 }
202 (void) memset(invis + ninvis, 0,
203 (invismaxsize - ninvis) *
204 sizeof (char *));
205 }
206 break;
207
208 case 'l':
209 label = optarg;
210 break;
211
212 case 'f':
213 if ((fp = fopen(optarg, "r")) == NULL) {
214 (void) fprintf(stderr,
215 gettext("%s: ERROR: can't open %s\n"),
216 prog, optarg);
217 exit(1);
218 }
219 break;
220
221 case 'u':
222 attr |= CKUNNUM;
223 break;
224
225 case 'd':
226 deflt = optarg;
227 break;
228
229 case 'p':
230 prompt = optarg;
231 break;
232
233 case 'e':
234 error = optarg;
235 break;
236
237 case 'h':
238 help = optarg;
239 break;
240
241 case 'k':
242 kpid = atoi(optarg);
243 break;
244
245 case 's':
246 signo = atoi(optarg);
247 break;
248
249 default:
250 usage();
251 }
252 }
253
254 if (signo) {
255 if (kpid == BADPID)
256 usage();
257 } else
258 signo = SIGTERM;
259
260 mp = allocmenu(label, attr);
261 if (fp) {
262 *wtemp = L'\0';
263 while (fgetws(wline, LSIZE, fp)) {
264 /*
265 * Skip comment lines, those beginning with '#'.
266 * Note: AT&T forgot this & needs the next 2 lines.
267 */
268 if (*wline == L'#')
269 continue;
270 n = wcslen(wline);
271 if ((n != 0) && (wline[n - 1] == L'\n'))
272 wline[n - 1] = L'\0';
273 /*
274 * if the line begins with a space character,
275 * this is a continuous line to the previous line.
276 */
277 if (iswspace(*wline)) {
278 (void) wcscat(wtemp, L"\n");
279 (void) wcscat(wtemp, wline);
280 } else {
281 if (*wtemp) {
282 n = wcslen(wtemp);
283 r = wcstombs(temp, wtemp,
284 n * MB_LEN_MAX);
285 if (r == (size_t)-1) {
286 (void) fprintf(stderr,
287 gettext("Invalid character in the menu definition.\n"));
288 exit(1);
289 }
290 if (setitem(mp, temp)) {
291 (void) fprintf(stderr,
292 gettext(INTERR), prog);
293 exit(1);
294 }
295 }
296 (void) wcscpy(wtemp, wline);
297 }
298 }
299 if (*wtemp) {
300 n = wcslen(wtemp);
301 r = wcstombs(temp, wtemp, n * MB_LEN_MAX);
302 if (r == (size_t)-1) {
303 (void) fprintf(stderr,
304 gettext("Invalid character in the menu definition.\n"));
305 exit(1);
306 }
307 if (setitem(mp, temp)) {
308 (void) fprintf(stderr, gettext(INTERR), prog);
309 exit(1);
310 }
311 }
312 }
313
314 while (optind < argc) {
315 if (setitem(mp, argv[optind++])) {
316 (void) fprintf(stderr, gettext(INTERR), prog);
317 exit(1);
318 }
319 }
320
321 for (n = 0; n < ninvis; ) {
322 if (setinvis(mp, invis[n++])) {
323 (void) fprintf(stderr, gettext(INTERR), prog);
324 exit(1);
325 }
326 }
327
328 if (*prog == 'e') {
329 ckindent = 0;
330 ckitem_err(mp, error);
331 exit(0);
332 } else if (*prog == 'h') {
333 ckindent = 0;
334 ckitem_hlp(mp, help);
335 exit(0);
336 }
337
338 if (max < 1) {
339 mmax = mp->nchoices;
340 } else {
341 mmax = max;
342 }
343
344 /*
345 * if -o option is specified, mp->nchoices is 1, and if no invisible
346 * item is specified, ckitem() will consume two entries of item,
347 * even though 'max' is set to 1. So to take care of that problem, we
348 * allocate one extra element for item
349 */
350 item = (char **)calloc(mmax+1, sizeof (char *));
351 if (!item) {
352 (void) fprintf(stderr,
353 gettext("Not enough memory\n"));
354 exit(1);
355 }
356 n = ckitem(mp, item, max, deflt, error, help, prompt);
357 if (n == 3) {
358 if (kpid > -2)
359 (void) kill(kpid, signo);
360 (void) puts("q");
361 } else if (n == 0) {
362 i = 0;
363 while (item[i])
364 (void) puts(item[i++]);
365 }
366 return (n);
367 }
368