usr/src/cmd/valtools/ckstr.c

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