xref: /illumos-gate/usr/src/cmd/valtools/ckyorn.c (revision 7c478bd9)
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 
45 static char	*prog;
46 static char	*deflt, *prompt, *error, *help;
47 static int	kpid = BADPID;
48 static int	signo;
49 
50 static const char	husage[] = "Wh";
51 static const char	eusage[] = "We";
52 
53 static void
54 usage(void)
55 {
56 	switch (*prog) {
57 	default:
58 		(void) fprintf(stderr,
59 			gettext("usage: %s [options]\n"), prog);
60 		(void) fprintf(stderr, gettext(OPTMESG));
61 		(void) fprintf(stderr, gettext(STDOPTS));
62 		break;
63 
64 	case 'v':
65 		(void) fprintf(stderr,
66 			gettext("usage: %s input\n"), prog);
67 		break;
68 
69 	case 'h':
70 		(void) fprintf(stderr,
71 			gettext("usage: %s [options]\n"), prog);
72 		(void) fprintf(stderr, gettext(OPTMESG));
73 		(void) fprintf(stderr,
74 			gettext("\t-W width\n\t-h help\n"));
75 		break;
76 
77 	case 'e':
78 		(void) fprintf(stderr,
79 			gettext("usage: %s [options]\n"), prog);
80 		(void) fprintf(stderr, gettext(OPTMESG));
81 		(void) fprintf(stderr,
82 			gettext("\t-W width\n\t-e error\n"));
83 		break;
84 	}
85 	exit(1);
86 }
87 
88 /*
89  * Given argv[0], return a pointer to the basename of the program.
90  */
91 static char *
92 prog_name(char *arg0)
93 {
94 	char *str;
95 
96 	/* first strip trailing '/' characters (exec() allows these!) */
97 	str = arg0 + strlen(arg0);
98 	while (str > arg0 && *--str == '/')
99 		*str = '\0';
100 	if ((str = strrchr(arg0, '/')) != NULL)
101 		return (str + 1);
102 	return (arg0);
103 }
104 
105 int
106 main(int argc, char **argv)
107 {
108 	int	c, n;
109 	char	*ynval;
110 	size_t	len;
111 
112 	(void) setlocale(LC_ALL, "");
113 
114 #if	!defined(TEXT_DOMAIN)
115 #define	TEXT_DOMAIN	"SYS_TEST"
116 #endif
117 	(void) textdomain(TEXT_DOMAIN);
118 
119 	prog = prog_name(argv[0]);
120 
121 	while ((c = getopt(argc, argv, "d:p:e:h:k:s:QW:?")) != EOF) {
122 		/* check for invalid option */
123 		if (*prog == 'v')
124 			usage();
125 		if ((*prog == 'e') && !strchr(eusage, c))
126 			usage();
127 		if ((*prog == 'h') && !strchr(husage, c))
128 			usage();
129 
130 		switch (c) {
131 		case 'Q':
132 			ckquit = 0;
133 			break;
134 
135 		case 'W':
136 			ckwidth = atoi(optarg);
137 			if (ckwidth < 0) {
138 				(void) fprintf(stderr,
139 		gettext("%s: ERROR: negative display width specified\n"),
140 					prog);
141 				exit(1);
142 			}
143 			break;
144 
145 		case 'd':
146 			deflt = optarg;
147 			break;
148 
149 		case 'p':
150 			prompt = optarg;
151 			break;
152 
153 		case 'e':
154 			error = optarg;
155 			break;
156 
157 		case 'h':
158 			help = optarg;
159 			break;
160 
161 		case 'k':
162 			kpid = atoi(optarg);
163 			break;
164 
165 		case 's':
166 			signo = atoi(optarg);
167 			break;
168 
169 		default:
170 			usage();
171 		}
172 	}
173 
174 	if (signo) {
175 		if (kpid == BADPID)
176 			usage();
177 	} else
178 		signo = SIGTERM;
179 
180 	if (*prog == 'v') {
181 		if (argc != (optind + 1))
182 			usage();
183 		if (ckyorn_val(argv[optind]))
184 			exit(1);
185 		else
186 			exit(0);
187 	}
188 
189 	if (optind != argc)
190 		usage();
191 
192 	if (*prog == 'e') {
193 		ckindent = 0;
194 		ckyorn_err(error);
195 		exit(0);
196 	} else if (*prog == 'h') {
197 		ckindent = 0;
198 		ckyorn_hlp(help);
199 		exit(0);
200 	}
201 
202 	if (deflt) {
203 		len = strlen(deflt) + 1;
204 		if (len < MAX_INPUT)
205 			len = MAX_INPUT;
206 	} else {
207 		len = MAX_INPUT;
208 	}
209 	ynval = (char *)malloc(len);
210 	if (!ynval) {
211 		(void) fprintf(stderr,
212 			gettext("Not enough memory\n"));
213 		exit(1);
214 	}
215 	n = ckyorn(ynval, deflt, error, help, prompt);
216 	if (n == 3) {
217 		if (kpid > -2) {
218 			if (kill(kpid, signo)) {
219 				(void) fprintf(stderr,
220 			gettext("Failed to send signal %d to process %d\n"),
221 				signo, kpid);
222 			}
223 		}
224 		(void) puts("q");
225 	} else if (n == 0)
226 		(void) fputs(ynval, stdout);
227 	return (n);
228 }
229