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 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 * 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 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