xref: /onnv-gate/usr/src/cmd/valtools/ckuid.c (revision 0:68f95e015346)
1*0Sstevel@tonic-gate /*
2*0Sstevel@tonic-gate  * CDDL HEADER START
3*0Sstevel@tonic-gate  *
4*0Sstevel@tonic-gate  * The contents of this file are subject to the terms of the
5*0Sstevel@tonic-gate  * Common Development and Distribution License, Version 1.0 only
6*0Sstevel@tonic-gate  * (the "License").  You may not use this file except in compliance
7*0Sstevel@tonic-gate  * with the License.
8*0Sstevel@tonic-gate  *
9*0Sstevel@tonic-gate  * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10*0Sstevel@tonic-gate  * or http://www.opensolaris.org/os/licensing.
11*0Sstevel@tonic-gate  * See the License for the specific language governing permissions
12*0Sstevel@tonic-gate  * and limitations under the License.
13*0Sstevel@tonic-gate  *
14*0Sstevel@tonic-gate  * When distributing Covered Code, include this CDDL HEADER in each
15*0Sstevel@tonic-gate  * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16*0Sstevel@tonic-gate  * If applicable, add the following below this CDDL HEADER, with the
17*0Sstevel@tonic-gate  * fields enclosed by brackets "[]" replaced with your own identifying
18*0Sstevel@tonic-gate  * information: Portions Copyright [yyyy] [name of copyright owner]
19*0Sstevel@tonic-gate  *
20*0Sstevel@tonic-gate  * CDDL HEADER END
21*0Sstevel@tonic-gate  */
22*0Sstevel@tonic-gate /*
23*0Sstevel@tonic-gate  * Copyright 2004 Sun Microsystems, Inc.  All rights reserved.
24*0Sstevel@tonic-gate  * Use is subject to license terms.
25*0Sstevel@tonic-gate  */
26*0Sstevel@tonic-gate 
27*0Sstevel@tonic-gate /*	Copyright (c) 1984, 1986, 1987, 1988, 1989 AT&T	*/
28*0Sstevel@tonic-gate /*	  All Rights Reserved  	*/
29*0Sstevel@tonic-gate 
30*0Sstevel@tonic-gate 
31*0Sstevel@tonic-gate #pragma ident	"%Z%%M%	%I%	%E% SMI"
32*0Sstevel@tonic-gate 
33*0Sstevel@tonic-gate #include <stdio.h>
34*0Sstevel@tonic-gate #include <string.h>
35*0Sstevel@tonic-gate #include <signal.h>
36*0Sstevel@tonic-gate #include <stdlib.h>
37*0Sstevel@tonic-gate #include <locale.h>
38*0Sstevel@tonic-gate #include <libintl.h>
39*0Sstevel@tonic-gate #include <limits.h>
40*0Sstevel@tonic-gate #include "usage.h"
41*0Sstevel@tonic-gate #include "libadm.h"
42*0Sstevel@tonic-gate 
43*0Sstevel@tonic-gate #define	BADPID	(-2)
44*0Sstevel@tonic-gate 
45*0Sstevel@tonic-gate static char	*prog;
46*0Sstevel@tonic-gate static char	*deflt = NULL, *prompt = NULL, *error = NULL, *help = NULL;
47*0Sstevel@tonic-gate static int	kpid = BADPID;
48*0Sstevel@tonic-gate static int	signo;
49*0Sstevel@tonic-gate 
50*0Sstevel@tonic-gate static const char	husage[] = "mWh";
51*0Sstevel@tonic-gate static const char	eusage[] = "mWe";
52*0Sstevel@tonic-gate 
53*0Sstevel@tonic-gate static void
usage(void)54*0Sstevel@tonic-gate usage(void)
55*0Sstevel@tonic-gate {
56*0Sstevel@tonic-gate 	switch (*prog) {
57*0Sstevel@tonic-gate 	default:
58*0Sstevel@tonic-gate 		(void) fprintf(stderr,
59*0Sstevel@tonic-gate 			gettext("usage: %s [options] [-m]\n"),
60*0Sstevel@tonic-gate 			prog);
61*0Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
62*0Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(STDOPTS));
63*0Sstevel@tonic-gate 		break;
64*0Sstevel@tonic-gate 
65*0Sstevel@tonic-gate 	case 'd':
66*0Sstevel@tonic-gate 		(void) fprintf(stderr,
67*0Sstevel@tonic-gate 			gettext("usage: %s\n"), prog);
68*0Sstevel@tonic-gate 		break;
69*0Sstevel@tonic-gate 
70*0Sstevel@tonic-gate 	case 'v':
71*0Sstevel@tonic-gate 		(void) fprintf(stderr,
72*0Sstevel@tonic-gate 			gettext("usage: %s input\n"), prog);
73*0Sstevel@tonic-gate 		break;
74*0Sstevel@tonic-gate 
75*0Sstevel@tonic-gate 	case 'h':
76*0Sstevel@tonic-gate 		(void) fprintf(stderr,
77*0Sstevel@tonic-gate 			gettext("usage: %s [options] [-m]\n"),
78*0Sstevel@tonic-gate 			prog);
79*0Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
80*0Sstevel@tonic-gate 		(void) fprintf(stderr,
81*0Sstevel@tonic-gate 			gettext("\t-W width\n\t-h help\n"));
82*0Sstevel@tonic-gate 		break;
83*0Sstevel@tonic-gate 
84*0Sstevel@tonic-gate 	case 'e':
85*0Sstevel@tonic-gate 		(void) fprintf(stderr,
86*0Sstevel@tonic-gate 			gettext("usage: %s [options] [-m]\n"),
87*0Sstevel@tonic-gate 			prog);
88*0Sstevel@tonic-gate 		(void) fprintf(stderr, gettext(OPTMESG));
89*0Sstevel@tonic-gate 		(void) fprintf(stderr,
90*0Sstevel@tonic-gate 			gettext("\t-W width\n\t-e error\n"));
91*0Sstevel@tonic-gate 		break;
92*0Sstevel@tonic-gate 	}
93*0Sstevel@tonic-gate 	exit(1);
94*0Sstevel@tonic-gate }
95*0Sstevel@tonic-gate 
96*0Sstevel@tonic-gate /*
97*0Sstevel@tonic-gate  * Given argv[0], return a pointer to the basename of the program.
98*0Sstevel@tonic-gate  */
99*0Sstevel@tonic-gate static char *
prog_name(char * arg0)100*0Sstevel@tonic-gate prog_name(char *arg0)
101*0Sstevel@tonic-gate {
102*0Sstevel@tonic-gate 	char *str;
103*0Sstevel@tonic-gate 
104*0Sstevel@tonic-gate 	/* first strip trailing '/' characters (exec() allows these!) */
105*0Sstevel@tonic-gate 	str = arg0 + strlen(arg0);
106*0Sstevel@tonic-gate 	while (str > arg0 && *--str == '/')
107*0Sstevel@tonic-gate 		*str = '\0';
108*0Sstevel@tonic-gate 	if ((str = strrchr(arg0, '/')) != NULL)
109*0Sstevel@tonic-gate 		return (str + 1);
110*0Sstevel@tonic-gate 	return (arg0);
111*0Sstevel@tonic-gate }
112*0Sstevel@tonic-gate 
113*0Sstevel@tonic-gate int
main(int argc,char ** argv)114*0Sstevel@tonic-gate main(int argc, char **argv)
115*0Sstevel@tonic-gate {
116*0Sstevel@tonic-gate 	int	c, n;
117*0Sstevel@tonic-gate 	char	*strval;
118*0Sstevel@tonic-gate 	size_t	len;
119*0Sstevel@tonic-gate 	int	disp = 0;	/* display set to no-display */
120*0Sstevel@tonic-gate 
121*0Sstevel@tonic-gate 	(void) setlocale(LC_ALL, "");
122*0Sstevel@tonic-gate 
123*0Sstevel@tonic-gate #if	!defined(TEXT_DOMAIN)
124*0Sstevel@tonic-gate #define	TEXT_DOMAIN	"SYS_TEST"
125*0Sstevel@tonic-gate #endif
126*0Sstevel@tonic-gate 	(void) textdomain(TEXT_DOMAIN);
127*0Sstevel@tonic-gate 
128*0Sstevel@tonic-gate 	prog = prog_name(argv[0]);
129*0Sstevel@tonic-gate 
130*0Sstevel@tonic-gate 	while ((c = getopt(argc, argv, "md:p:e:h:k:s:QW:?")) != EOF) {
131*0Sstevel@tonic-gate 		/* check for invalid option */
132*0Sstevel@tonic-gate 		if ((*prog == 'v') || (*prog == 'd'))
133*0Sstevel@tonic-gate 			usage();
134*0Sstevel@tonic-gate 		if ((*prog == 'e') && !strchr(eusage, c))
135*0Sstevel@tonic-gate 			usage();
136*0Sstevel@tonic-gate 		if ((*prog == 'h') && !strchr(husage, c))
137*0Sstevel@tonic-gate 			usage();
138*0Sstevel@tonic-gate 
139*0Sstevel@tonic-gate 		switch (c) {
140*0Sstevel@tonic-gate 		case 'Q':
141*0Sstevel@tonic-gate 			ckquit = 0;
142*0Sstevel@tonic-gate 			break;
143*0Sstevel@tonic-gate 
144*0Sstevel@tonic-gate 		case 'W':
145*0Sstevel@tonic-gate 			ckwidth = atoi(optarg);
146*0Sstevel@tonic-gate 			if (ckwidth < 0) {
147*0Sstevel@tonic-gate 				(void) fprintf(stderr,
148*0Sstevel@tonic-gate 		gettext("%s: ERROR: negative display width specified\n"),
149*0Sstevel@tonic-gate 					prog);
150*0Sstevel@tonic-gate 				exit(1);
151*0Sstevel@tonic-gate 			}
152*0Sstevel@tonic-gate 			break;
153*0Sstevel@tonic-gate 
154*0Sstevel@tonic-gate 		case 'm':
155*0Sstevel@tonic-gate 			disp = 1;
156*0Sstevel@tonic-gate 			break;
157*0Sstevel@tonic-gate 
158*0Sstevel@tonic-gate 		case 'd':
159*0Sstevel@tonic-gate 			deflt = optarg;
160*0Sstevel@tonic-gate 			break;
161*0Sstevel@tonic-gate 
162*0Sstevel@tonic-gate 		case 'p':
163*0Sstevel@tonic-gate 			prompt = optarg;
164*0Sstevel@tonic-gate 			break;
165*0Sstevel@tonic-gate 
166*0Sstevel@tonic-gate 		case 'e':
167*0Sstevel@tonic-gate 			error = optarg;
168*0Sstevel@tonic-gate 			break;
169*0Sstevel@tonic-gate 
170*0Sstevel@tonic-gate 		case 'h':
171*0Sstevel@tonic-gate 			help = optarg;
172*0Sstevel@tonic-gate 			break;
173*0Sstevel@tonic-gate 
174*0Sstevel@tonic-gate 		case 'k':
175*0Sstevel@tonic-gate 			kpid = atoi(optarg);
176*0Sstevel@tonic-gate 			break;
177*0Sstevel@tonic-gate 
178*0Sstevel@tonic-gate 		case 's':
179*0Sstevel@tonic-gate 			signo = atoi(optarg);
180*0Sstevel@tonic-gate 			break;
181*0Sstevel@tonic-gate 
182*0Sstevel@tonic-gate 		default:
183*0Sstevel@tonic-gate 			usage();
184*0Sstevel@tonic-gate 		}
185*0Sstevel@tonic-gate 	}
186*0Sstevel@tonic-gate 
187*0Sstevel@tonic-gate 	if (signo) {
188*0Sstevel@tonic-gate 		if (kpid == BADPID)
189*0Sstevel@tonic-gate 			usage();
190*0Sstevel@tonic-gate 	} else
191*0Sstevel@tonic-gate 		signo = SIGTERM;
192*0Sstevel@tonic-gate 
193*0Sstevel@tonic-gate 	if (*prog == 'v') {
194*0Sstevel@tonic-gate 		if (argc != (optind+1))
195*0Sstevel@tonic-gate 			usage();
196*0Sstevel@tonic-gate 		exit(ckuid_val(argv[optind]));
197*0Sstevel@tonic-gate 	}
198*0Sstevel@tonic-gate 
199*0Sstevel@tonic-gate 	if (optind != argc)
200*0Sstevel@tonic-gate 		usage();
201*0Sstevel@tonic-gate 
202*0Sstevel@tonic-gate 	if (*prog == 'e') {
203*0Sstevel@tonic-gate 		ckindent = 0;
204*0Sstevel@tonic-gate 		ckuid_err(disp, error);
205*0Sstevel@tonic-gate 		exit(0);
206*0Sstevel@tonic-gate 	} else if (*prog == 'h') {
207*0Sstevel@tonic-gate 		ckindent = 0;
208*0Sstevel@tonic-gate 		ckuid_hlp(disp, help);
209*0Sstevel@tonic-gate 		exit(0);
210*0Sstevel@tonic-gate 	} else if (*prog == 'd') {
211*0Sstevel@tonic-gate 		exit(ckuid_dsp());
212*0Sstevel@tonic-gate 	}
213*0Sstevel@tonic-gate 
214*0Sstevel@tonic-gate 	if (deflt) {
215*0Sstevel@tonic-gate 		len = strlen(deflt) + 1;
216*0Sstevel@tonic-gate 		if (len < MAX_INPUT)
217*0Sstevel@tonic-gate 			len = MAX_INPUT;
218*0Sstevel@tonic-gate 	} else {
219*0Sstevel@tonic-gate 		len = MAX_INPUT;
220*0Sstevel@tonic-gate 	}
221*0Sstevel@tonic-gate 	strval = (char *)malloc(len);
222*0Sstevel@tonic-gate 	if (!strval) {
223*0Sstevel@tonic-gate 		(void) fprintf(stderr,
224*0Sstevel@tonic-gate 			gettext("Not enough memory\n"));
225*0Sstevel@tonic-gate 		exit(1);
226*0Sstevel@tonic-gate 	}
227*0Sstevel@tonic-gate 	n = ckuid(strval, disp, deflt, error, help, prompt);
228*0Sstevel@tonic-gate 	if (n == 3) {
229*0Sstevel@tonic-gate 		if (kpid > -2)
230*0Sstevel@tonic-gate 			(void) kill(kpid, signo);
231*0Sstevel@tonic-gate 		(void) puts("q");
232*0Sstevel@tonic-gate 	} else if (n == 0)
233*0Sstevel@tonic-gate 		(void) fputs(strval, stdout);
234*0Sstevel@tonic-gate 	return (n);
235*0Sstevel@tonic-gate }
236