Annotation of src/usr.bin/rdist/rdist.c, Revision 1.23
1.23 ! guenther 1: /* $OpenBSD: rdist.c,v 1.22 2014/07/05 05:05:51 guenther Exp $ */
1.3 deraadt 2:
1.1 dm 3: /*
4: * Copyright (c) 1983 Regents of the University of California.
5: * All rights reserved.
6: *
7: * Redistribution and use in source and binary forms, with or without
8: * modification, are permitted provided that the following conditions
9: * are met:
10: * 1. Redistributions of source code must retain the above copyright
11: * notice, this list of conditions and the following disclaimer.
12: * 2. Redistributions in binary form must reproduce the above copyright
13: * notice, this list of conditions and the following disclaimer in the
14: * documentation and/or other materials provided with the distribution.
1.17 millert 15: * 3. Neither the name of the University nor the names of its contributors
1.1 dm 16: * may be used to endorse or promote products derived from this software
17: * without specific prior written permission.
18: *
19: * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20: * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21: * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22: * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23: * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24: * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25: * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26: * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27: * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28: * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29: * SUCH DAMAGE.
30: */
31:
1.16 millert 32: #include "defs.h"
33: #include "y.tab.h"
1.1 dm 34:
35: #include <netdb.h>
36: #include <sys/ioctl.h>
37:
38: /*
39: * Remote distribution program.
40: */
1.6 millert 41:
1.1 dm 42: char *distfile = NULL; /* Name of distfile to use */
43: int maxchildren = MAXCHILDREN; /* Max no of concurrent PIDs */
44: int nflag = 0; /* Say without doing */
1.21 krw 45: int64_t min_freespace = 0; /* Min filesys free space */
46: int64_t min_freefiles = 0; /* Min filesys free # files */
1.1 dm 47: FILE *fin = NULL; /* Input file pointer */
48: char localmsglist[] = "stdout=all:notify=all:syslog=nerror,ferror";
49: char *remotemsglist = NULL;
50: char optchars[] = "A:a:bcd:DFf:hil:L:M:m:NnOo:p:P:qRrst:Vvwxy";
51: char *path_rdistd = _PATH_RDISTD;
1.4 millert 52: char *path_remsh = NULL;
1.1 dm 53:
1.16 millert 54: static void addhostlist(char *, struct namelist **);
55: static void usage(void);
56: int main(int, char **, char **);
57:
1.1 dm 58: /*
59: * Add a hostname to the host list
60: */
1.16 millert 61: static void
62: addhostlist(char *name, struct namelist **hostlist)
1.1 dm 63: {
1.9 mpech 64: struct namelist *ptr, *new;
1.1 dm 65:
66: if (!name || !hostlist)
67: return;
68:
1.22 guenther 69: new = xmalloc(sizeof *new);
1.7 millert 70: new->n_name = xstrdup(name);
1.15 millert 71: new->n_regex = NULL;
1.1 dm 72: new->n_next = NULL;
73:
74: if (*hostlist) {
75: for (ptr = *hostlist; ptr && ptr->n_next; ptr = ptr->n_next)
76: ;
77: ptr->n_next = new;
78: } else
79: *hostlist = new;
80: }
81:
1.5 millert 82: int
1.16 millert 83: main(int argc, char **argv, char **envp)
1.1 dm 84: {
1.16 millert 85: extern char *__progname;
1.1 dm 86: struct namelist *hostlist = NULL;
1.9 mpech 87: int x;
88: char *cp;
1.1 dm 89: int cmdargs = 0;
90: int c;
1.20 krw 91: const char *errstr;
1.1 dm 92:
1.16 millert 93: progname = __progname;
1.1 dm 94:
1.16 millert 95: if ((cp = msgparseopts(localmsglist, TRUE)) != NULL) {
1.1 dm 96: error("Bad builtin log option (%s): %s.",
97: localmsglist, cp);
98: usage();
99: }
100:
1.16 millert 101: if ((cp = getenv("RDIST_OPTIONS")) != NULL)
102: if (parsedistopts(cp, &options, TRUE)) {
103: error("Bad dist option environment string \"%s\".",
104: cp);
105: exit(1);
106: }
107:
1.1 dm 108: if (init(argc, argv, envp) < 0)
109: exit(1);
110:
111: /*
112: * Be backwards compatible.
113: */
114: for (x = 1; x <= argc && argv[x]; x++) {
115: if (strcmp(argv[x], "-Server") != 0)
116: continue;
117: #if defined(_PATH_OLDRDIST)
118: message(MT_SYSLOG,
119: "Old rdist (-Server) requested; running %s",
120: _PATH_OLDRDIST);
121: (void) execl(_PATH_OLDRDIST, xbasename(_PATH_OLDRDIST),
1.8 deraadt 122: "-Server", (char *)NULL);
1.1 dm 123: fatalerr("Exec old rdist failed: %s: %s.",
124: _PATH_OLDRDIST, SYSERR);
125: #else /* !_PATH_OLDRDIST */
126: fatalerr("Old rdist not available.");
127: #endif /* _PATH_OLDRDIST */
128: exit(1);
129: }
130:
131: /*
132: * Perform check to make sure we are not incorrectly installed
133: * setuid to root or anybody else.
134: */
135: if (getuid() != geteuid())
136: fatalerr("This version of rdist should not be installed setuid.");
137:
138: while ((c = getopt(argc, argv, optchars)) != -1)
139: switch (c) {
140: case 'l':
1.16 millert 141: if ((cp = msgparseopts(optarg, TRUE)) != NULL) {
1.1 dm 142: error("Bad log option \"%s\": %s.", optarg,cp);
143: usage();
144: }
145: break;
146:
147: case 'L':
1.7 millert 148: remotemsglist = xstrdup(optarg);
1.1 dm 149: break;
150:
151: case 'A':
152: case 'a':
153: case 'M':
154: case 't':
1.16 millert 155: if (!isdigit((unsigned char)*optarg)) {
1.1 dm 156: error("\"%s\" is not a number.", optarg);
157: usage();
158: }
1.20 krw 159: if (c == 'a') {
160: min_freespace = (int64_t)strtonum(optarg,
161: 0, LLONG_MAX, &errstr);
162: if (errstr)
163: fatalerr("Minimum free space is %s: "
164: "'%s'", errstr, optarg);
165: }
166: else if (c == 'A') {
1.21 krw 167: min_freefiles = (int64_t)strtonum(optarg,
1.20 krw 168: 0, LLONG_MAX, &errstr);
169: if (errstr)
170: fatalerr("Minimum free files is %s: "
171: "'%s'", errstr, optarg);
172: }
1.1 dm 173: else if (c == 'M')
174: maxchildren = atoi(optarg);
175: else if (c == 't')
176: rtimeout = atoi(optarg);
177: break;
178:
179: case 'F':
180: do_fork = FALSE;
181: break;
182:
183: case 'f':
1.7 millert 184: distfile = xstrdup(optarg);
1.1 dm 185: if (distfile[0] == '-' && distfile[1] == CNULL)
186: fin = stdin;
187: break;
188:
189: case 'm':
190: addhostlist(optarg, &hostlist);
191: break;
192:
193: case 'd':
194: define(optarg);
195: break;
196:
197: case 'D':
198: debug = DM_ALL;
1.16 millert 199: if ((cp = msgparseopts("stdout=all,debug",
200: TRUE)) != NULL) {
1.1 dm 201: error("Enable debug messages failed: %s.", cp);
202: usage();
203: }
204: break;
205:
206: case 'c':
207: cmdargs++;
208: break;
209:
210: case 'n':
211: nflag++;
212: break;
213:
214: case 'V':
215: printf("%s\n", getversion());
216: exit(0);
217:
218: case 'o':
219: if (parsedistopts(optarg, &options, TRUE)) {
220: error("Bad dist option string \"%s\".",
221: optarg);
222: usage();
223: }
224: break;
225:
226: case 'p':
227: if (!optarg) {
228: error("No path specified to \"-p\".");
229: usage();
230: }
1.7 millert 231: path_rdistd = xstrdup(optarg);
1.1 dm 232: break;
233:
234: case 'P':
235: if (!optarg) {
236: error("No path specified to \"-P\".");
237: usage();
238: }
1.16 millert 239: if ((cp = searchpath(optarg)) != NULL)
1.7 millert 240: path_remsh = xstrdup(cp);
1.1 dm 241: else {
242: error("No component of path \"%s\" exists.",
243: optarg);
244: usage();
245: }
246: break;
247:
248: /*
249: * These options are obsoleted by -o. They are
250: * provided only for backwards compatibility
251: */
252: case 'v': FLAG_ON(options, DO_VERIFY); break;
253: case 'N': FLAG_ON(options, DO_CHKNFS); break;
254: case 'O': FLAG_ON(options, DO_CHKREADONLY); break;
255: case 'q': FLAG_ON(options, DO_QUIET); break;
256: case 'b': FLAG_ON(options, DO_COMPARE); break;
257: case 'r': FLAG_ON(options, DO_NODESCEND); break;
258: case 'R': FLAG_ON(options, DO_REMOVE); break;
259: case 's': FLAG_ON(options, DO_SAVETARGETS); break;
260: case 'w': FLAG_ON(options, DO_WHOLE); break;
261: case 'y': FLAG_ON(options, DO_YOUNGER); break;
262: case 'h': FLAG_ON(options, DO_FOLLOW); break;
263: case 'i': FLAG_ON(options, DO_IGNLNKS); break;
264: case 'x': FLAG_ON(options, DO_NOEXEC); break;
265:
266: case '?':
267: default:
268: usage();
269: }
270:
271: if (debug) {
272: printf("%s\n", getversion());
273: msgprconfig();
274: }
275:
276: if (nflag && IS_ON(options, DO_VERIFY))
277: fatalerr(
278: "The -n flag and \"verify\" mode may not both be used.");
1.4 millert 279:
1.13 millert 280: if (path_remsh == NULL) {
281: if ((cp = getenv("RSH")) != NULL && *cp != '\0')
282: path_remsh = cp;
283: else
1.16 millert 284: path_remsh = _PATH_REMSH;
1.13 millert 285: }
1.1 dm 286:
287: /*
288: * Don't fork children for nflag
289: */
290: if (nflag)
291: do_fork = 0;
292:
293: if (cmdargs)
294: docmdargs(realargc - optind, &realargv[optind]);
295: else {
296: if (fin == NULL)
297: fin = opendist(distfile);
298: (void) yyparse();
299: /*
300: * Need to keep stdin open for child processing later
301: */
302: if (fin != stdin)
303: (void) fclose(fin);
304: if (nerrs == 0)
305: docmds(hostlist, realargc-optind, &realargv[optind]);
306: }
307:
308: exit(nerrs != 0);
309: }
310:
311: /*
312: * Open a distfile
313: */
1.16 millert 314: FILE *
315: opendist(char *distfile)
1.1 dm 316: {
317: char *file = NULL;
318: FILE *fp;
319:
320: if (distfile == NULL) {
321: if (access("distfile", R_OK) == 0)
322: file = "distfile";
323: else if (access("Distfile", R_OK) == 0)
324: file = "Distfile";
325: } else {
326: /*
327: * Try to test to see if file is readable before running m4.
328: */
329: if (access(distfile, R_OK) != 0)
330: fatalerr("%s: Cannot access file: %s.",
331: distfile, SYSERR);
332: file = distfile;
333: }
334:
335: if (file == NULL)
336: fatalerr("No distfile found.");
337:
338: fp = fopen(file, "r");
339:
340: if (fp == NULL)
341: fatalerr("%s: open failed: %s.", file, SYSERR);
342:
343: return(fp);
344: }
345:
346: /*
347: * Print usage message and exit.
348: */
1.16 millert 349: static void
350: usage(void)
1.1 dm 351: {
1.18 jmc 352: extern char *__progname;
1.1 dm 353:
354: (void) fprintf(stderr,
1.18 jmc 355: "usage: %s [-DFnV] [-Server] [-A num] [-a num] "
356: "[-c mini_distfile]\n"
357: "\t[-d var=value] [-f distfile] [-L remote_logopts] "
358: "[-l local_logopts]\n"
359: "\t[-M maxproc] [-m host] [-o distopts] [-P rsh-path] "
360: "[-p rdistd-path]\n"
361: "\t[-t timeout] [name ...]\n", __progname);
1.1 dm 362:
363:
364: (void) fprintf(stderr, "\nThe values for <distopts> are:\n\t%s\n",
365: getdistoptlist());
366:
367: msgprusage();
368:
369: exit(1);
370: }
371:
372: /*
373: * rcp like interface for distributing files.
374: */
1.5 millert 375: void
1.16 millert 376: docmdargs(int nargs, char **args)
1.1 dm 377: {
1.9 mpech 378: struct namelist *nl, *prev;
379: char *cp;
1.1 dm 380: struct namelist *files, *hosts;
381: struct subcmd *cmds;
382: char *dest;
1.15 millert 383: static struct namelist tnl;
1.1 dm 384: int i;
385:
386: if (nargs < 2)
387: usage();
388:
389: prev = NULL;
390: files = NULL;
391: for (i = 0; i < nargs - 1; i++) {
392: nl = makenl(args[i]);
393: if (prev == NULL)
394: files = prev = nl;
395: else {
396: prev->n_next = nl;
397: prev = nl;
398: }
399: }
400:
401: cp = args[i];
402: if ((dest = strchr(cp, ':')) != NULL)
403: *dest++ = '\0';
404: tnl.n_name = cp;
1.15 millert 405: tnl.n_regex = NULL;
406: tnl.n_next = NULL;
1.1 dm 407: hosts = expand(&tnl, E_ALL);
408: if (nerrs)
409: exit(1);
410:
411: if (dest == NULL || *dest == '\0')
412: cmds = NULL;
413: else {
414: cmds = makesubcmd(INSTALL);
415: cmds->sc_options = options;
416: cmds->sc_name = dest;
417: }
418:
419: debugmsg(DM_MISC, "docmdargs()\nfiles = %s", getnlstr(files));
420: debugmsg(DM_MISC, "host = %s", getnlstr(hosts));
421:
1.5 millert 422: insert(NULL, files, hosts, cmds);
1.16 millert 423: docmds(NULL, 0, NULL);
1.1 dm 424: }
425:
426: /*
427: * Get a list of NAME blocks (mostly for debugging).
428: */
1.16 millert 429: char *
430: getnlstr(struct namelist *nl)
1.1 dm 431: {
432: static char buf[16384];
1.16 millert 433: size_t len = 0;
1.1 dm 434:
1.16 millert 435: (void) snprintf(buf, sizeof(buf), "(");
1.1 dm 436:
437: while (nl != NULL) {
438: if (nl->n_name == NULL)
439: continue;
440: len += strlen(nl->n_name) + 2;
441: if (len >= sizeof(buf)) {
1.14 deraadt 442: (void) strlcpy(buf,
1.16 millert 443: "getnlstr() Buffer not large enough",
444: sizeof(buf));
1.1 dm 445: return(buf);
446: }
1.16 millert 447: (void) strlcat(buf, " ", sizeof(buf));
448: (void) strlcat(buf, nl->n_name, sizeof(buf));
1.1 dm 449: nl = nl->n_next;
450: }
451:
1.16 millert 452: (void) strlcat(buf, " )", sizeof(buf));
1.1 dm 453:
454: return(buf);
455: }