xref: /netbsd-src/usr.sbin/timed/timedc/timedc.c (revision cda4f8f6ee55684e8d311b86c99ea59191e6b74f)
1 /*-
2  * Copyright (c) 1985, 1993
3  *	The Regents of the University of California.  All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by the University of
16  *	California, Berkeley and its contributors.
17  * 4. Neither the name of the University nor the names of its contributors
18  *    may be used to endorse or promote products derived from this software
19  *    without specific prior written permission.
20  *
21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31  * SUCH DAMAGE.
32  */
33 
34 #ifndef lint
35 static char copyright[] =
36 "@(#) Copyright (c) 1985, 1993\n\
37 	The Regents of the University of California.  All rights reserved.\n";
38 #endif /* not lint */
39 
40 #ifndef lint
41 static char sccsid[] = "@(#)timedc.c	8.1 (Berkeley) 6/6/93";
42 #endif /* not lint */
43 
44 #ifdef sgi
45 #ident "$Revision: 1.1.1.2 $"
46 #endif
47 
48 #include "timedc.h"
49 #include <strings.h>
50 #include <signal.h>
51 #include <ctype.h>
52 #include <setjmp.h>
53 #include <unistd.h>
54 #include <stdlib.h>
55 #include <syslog.h>
56 
57 int trace = 0;
58 FILE *fd = 0;
59 int	margc;
60 int	fromatty;
61 char	*margv[20];
62 char	cmdline[200];
63 jmp_buf	toplevel;
64 static struct cmd *getcmd __P((char *));
65 
66 int
67 main(argc, argv)
68 	int argc;
69 	char *argv[];
70 {
71 	register struct cmd *c;
72 
73 	openlog("timedc", LOG_ODELAY, LOG_AUTH);
74 
75 	/*
76 	 * security dictates!
77 	 */
78 	if (priv_resources() < 0) {
79 		fprintf(stderr, "Could not get privileged resources\n");
80 		exit(1);
81 	}
82 	(void) setuid(getuid());
83 
84 	if (--argc > 0) {
85 		c = getcmd(*++argv);
86 		if (c == (struct cmd *)-1) {
87 			printf("?Ambiguous command\n");
88 			exit(1);
89 		}
90 		if (c == 0) {
91 			printf("?Invalid command\n");
92 			exit(1);
93 		}
94 		if (c->c_priv && getuid()) {
95 			printf("?Privileged command\n");
96 			exit(1);
97 		}
98 		(*c->c_handler)(argc, argv);
99 		exit(0);
100 	}
101 
102 	fromatty = isatty(fileno(stdin));
103 	if (setjmp(toplevel))
104 		putchar('\n');
105 	(void) signal(SIGINT, intr);
106 	for (;;) {
107 		if (fromatty) {
108 			printf("timedc> ");
109 			(void) fflush(stdout);
110 		}
111 		if (fgets(cmdline, sizeof(cmdline), stdin) == 0)
112 			quit();
113 		if (cmdline[0] == 0)
114 			break;
115 		makeargv();
116 		if (margv[0] == 0)
117 			continue;
118 		c = getcmd(margv[0]);
119 		if (c == (struct cmd *)-1) {
120 			printf("?Ambiguous command\n");
121 			continue;
122 		}
123 		if (c == 0) {
124 			printf("?Invalid command\n");
125 			continue;
126 		}
127 		if (c->c_priv && getuid()) {
128 			printf("?Privileged command\n");
129 			continue;
130 		}
131 		(*c->c_handler)(margc, margv);
132 	}
133 	return 0;
134 }
135 
136 void
137 intr(signo)
138 	int signo;
139 {
140 	if (!fromatty)
141 		exit(0);
142 	longjmp(toplevel, 1);
143 }
144 
145 
146 static struct cmd *
147 getcmd(name)
148 	char *name;
149 {
150 	register char *p, *q;
151 	register struct cmd *c, *found;
152 	register int nmatches, longest;
153 	extern int NCMDS;
154 
155 	longest = 0;
156 	nmatches = 0;
157 	found = 0;
158 	for (c = cmdtab; c < &cmdtab[NCMDS]; c++) {
159 		p = c->c_name;
160 		for (q = name; *q == *p++; q++)
161 			if (*q == 0)		/* exact match? */
162 				return(c);
163 		if (!*q) {			/* the name was a prefix */
164 			if (q - name > longest) {
165 				longest = q - name;
166 				nmatches = 1;
167 				found = c;
168 			} else if (q - name == longest)
169 				nmatches++;
170 		}
171 	}
172 	if (nmatches > 1)
173 		return((struct cmd *)-1);
174 	return(found);
175 }
176 
177 /*
178  * Slice a string up into argc/argv.
179  */
180 void
181 makeargv()
182 {
183 	register char *cp;
184 	register char **argp = margv;
185 
186 	margc = 0;
187 	for (cp = cmdline; *cp;) {
188 		while (isspace(*cp))
189 			cp++;
190 		if (*cp == '\0')
191 			break;
192 		*argp++ = cp;
193 		margc += 1;
194 		while (*cp != '\0' && !isspace(*cp))
195 			cp++;
196 		if (*cp == '\0')
197 			break;
198 		*cp++ = '\0';
199 	}
200 	*argp++ = 0;
201 }
202 
203 #define HELPINDENT (sizeof ("directory"))
204 
205 /*
206  * Help command.
207  */
208 void
209 help(argc, argv)
210 	int argc;
211 	char *argv[];
212 {
213 	register struct cmd *c;
214 
215 	if (argc == 1) {
216 		register int i, j, w;
217 		int columns, width = 0, lines;
218 		extern int NCMDS;
219 
220 		printf("Commands may be abbreviated.  Commands are:\n\n");
221 		for (c = cmdtab; c < &cmdtab[NCMDS]; c++) {
222 			int len = strlen(c->c_name);
223 
224 			if (len > width)
225 				width = len;
226 		}
227 		width = (width + 8) &~ 7;
228 		columns = 80 / width;
229 		if (columns == 0)
230 			columns = 1;
231 		lines = (NCMDS + columns - 1) / columns;
232 		for (i = 0; i < lines; i++) {
233 			for (j = 0; j < columns; j++) {
234 				c = cmdtab + j * lines + i;
235 				printf("%s", c->c_name);
236 				if (c + lines >= &cmdtab[NCMDS]) {
237 					printf("\n");
238 					break;
239 				}
240 				w = strlen(c->c_name);
241 				while (w < width) {
242 					w = (w + 8) &~ 7;
243 					putchar('\t');
244 				}
245 			}
246 		}
247 		return;
248 	}
249 	while (--argc > 0) {
250 		register char *arg;
251 		arg = *++argv;
252 		c = getcmd(arg);
253 		if (c == (struct cmd *)-1)
254 			printf("?Ambiguous help command %s\n", arg);
255 		else if (c == (struct cmd *)0)
256 			printf("?Invalid help command %s\n", arg);
257 		else
258 			printf("%-*s\t%s\n", (int)HELPINDENT,
259 				c->c_name, c->c_help);
260 	}
261 }
262