xref: /netbsd-src/usr.bin/rdist/main.c (revision dc306354b0b29af51801a7632f1e95265a68cd81)
1 /*	$NetBSD: main.c,v 1.8 1997/10/19 14:25:30 mycroft Exp $	*/
2 
3 /*
4  * Copyright (c) 1983, 1993
5  *	The Regents of the University of California.  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.
15  * 3. All advertising materials mentioning features or use of this software
16  *    must display the following acknowledgement:
17  *	This product includes software developed by the University of
18  *	California, Berkeley and its contributors.
19  * 4. Neither the name of the University nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35 
36 #include <sys/cdefs.h>
37 #ifndef lint
38 __COPYRIGHT("@(#) Copyright (c) 1983, 1993\n\
39 	The Regents of the University of California.  All rights reserved.\n");
40 #endif /* not lint */
41 
42 #ifndef lint
43 #if 0
44 static char sccsid[] = "@(#)main.c	8.1 (Berkeley) 6/9/93";
45 #else
46 __RCSID("$NetBSD: main.c,v 1.8 1997/10/19 14:25:30 mycroft Exp $");
47 #endif
48 #endif /* not lint */
49 
50 #include <sys/types.h>
51 
52 #include <errno.h>
53 #include <pwd.h>
54 
55 #include "defs.h"
56 
57 #define NHOSTS 100
58 
59 /*
60  * Remote distribution program.
61  */
62 
63 char	*distfile = NULL;
64 #define _RDIST_TMP	"/rdistXXXXXX"
65 char	tempfile[sizeof _PATH_TMP + sizeof _RDIST_TMP + 1];
66 char	*tempname;
67 
68 int	debug;		/* debugging flag */
69 int	nflag;		/* NOP flag, just print commands without executing */
70 int	qflag;		/* Quiet. Don't print messages */
71 int	options;	/* global options */
72 int	iamremote;	/* act as remote server for transfering files */
73 
74 FILE	*fin = NULL;	/* input file pointer */
75 int	rem = -1;	/* file descriptor to remote source/sink process */
76 char	host[32];	/* host name */
77 int	nerrs;		/* number of errors while sending/receiving */
78 char	user[10];	/* user's name */
79 char	homedir[128];	/* user's home directory */
80 uid_t	userid;		/* user's user ID */
81 gid_t	groupid;	/* user's group ID */
82 
83 struct	passwd *pw;	/* pointer to static area used by getpwent */
84 struct	group *gr;	/* pointer to static area used by getgrent */
85 
86 int	main __P((int, char **));
87 static void usage __P((void));
88 static void docmdargs __P((int, char *[]));
89 
90 int
91 main(argc, argv)
92 	int argc;
93 	char *argv[];
94 {
95 	char *arg;
96 	int cmdargs = 0;
97 	char *dhosts[NHOSTS], **hp = dhosts;
98 
99 	pw = getpwuid(userid = getuid());
100 	if (pw == NULL) {
101 		fprintf(stderr, "%s: Who are you?\n", argv[0]);
102 		exit(1);
103 	}
104 	strcpy(user, pw->pw_name);
105 	strcpy(homedir, pw->pw_dir);
106 	groupid = pw->pw_gid;
107 	gethostname(host, sizeof(host));
108 	strcpy(tempfile, _PATH_TMP);
109 	strcat(tempfile, _RDIST_TMP);
110 	if ((tempname = strrchr(tempfile, '/')) != 0)
111 		tempname++;
112 	else
113 		tempname = tempfile;
114 
115 	while (--argc > 0) {
116 		if ((arg = *++argv)[0] != '-')
117 			break;
118 		if (!strcmp(arg, "-Server"))
119 			iamremote++;
120 		else while (*++arg)
121 			switch (*arg) {
122 			case 'f':
123 				if (--argc <= 0)
124 					usage();
125 				distfile = *++argv;
126 				if (distfile[0] == '-' && distfile[1] == '\0')
127 					fin = stdin;
128 				break;
129 
130 			case 'm':
131 				if (--argc <= 0)
132 					usage();
133 				if (hp >= &dhosts[NHOSTS-2]) {
134 					fprintf(stderr, "rdist: too many destination hosts\n");
135 					exit(1);
136 				}
137 				*hp++ = *++argv;
138 				break;
139 
140 			case 'd':
141 				if (--argc <= 0)
142 					usage();
143 				define(*++argv);
144 				break;
145 
146 			case 'D':
147 				debug++;
148 				break;
149 
150 			case 'c':
151 				cmdargs++;
152 				break;
153 
154 			case 'n':
155 				if (options & VERIFY) {
156 					printf("rdist: -n overrides -v\n");
157 					options &= ~VERIFY;
158 				}
159 				nflag++;
160 				break;
161 
162 			case 'q':
163 				qflag++;
164 				break;
165 
166 			case 'b':
167 				options |= COMPARE;
168 				break;
169 
170 			case 'R':
171 				options |= REMOVE;
172 				break;
173 
174 			case 'v':
175 				if (nflag) {
176 					printf("rdist: -n overrides -v\n");
177 					break;
178 				}
179 				options |= VERIFY;
180 				break;
181 
182 			case 'w':
183 				options |= WHOLE;
184 				break;
185 
186 			case 'y':
187 				options |= YOUNGER;
188 				break;
189 
190 			case 'h':
191 				options |= FOLLOW;
192 				break;
193 
194 			case 'i':
195 				options |= IGNLNKS;
196 				break;
197 
198 			default:
199 				usage();
200 			}
201 	}
202 	*hp = NULL;
203 
204 	seteuid(userid);
205 	mktemp(tempfile);
206 
207 	if (iamremote) {
208 		server();
209 		exit(nerrs != 0);
210 	}
211 
212 	if (cmdargs)
213 		docmdargs(argc, argv);
214 	else {
215 		if (fin == NULL) {
216 			if(distfile == NULL) {
217 				if((fin = fopen("distfile","r")) == NULL)
218 					fin = fopen("Distfile", "r");
219 			} else
220 				fin = fopen(distfile, "r");
221 			if(fin == NULL) {
222 				perror(distfile ? distfile : "distfile");
223 				exit(1);
224 			}
225 		}
226 		yyparse();
227 		if (nerrs == 0)
228 			docmds(dhosts, argc, argv);
229 	}
230 
231 	exit(nerrs != 0);
232 }
233 
234 static void
235 usage()
236 {
237 	printf("Usage: rdist [-nqbhirvwyD] [-f distfile] [-d var=value] [-m host] [file ...]\n");
238 	printf("or: rdist [-nqbhirvwyD] -c source [...] machine[:dest]\n");
239 	exit(1);
240 }
241 
242 /*
243  * rcp like interface for distributing files.
244  */
245 static void
246 docmdargs(nargs, args)
247 	int nargs;
248 	char *args[];
249 {
250 	struct namelist *nl, *prev;
251 	char *cp;
252 	struct namelist *files, *hosts;
253 	struct subcmd *cmds;
254 	char *dest;
255 	static struct namelist tnl = { NULL, NULL };
256 	int i;
257 
258 	if (nargs < 2)
259 		usage();
260 
261 	files = NULL;
262 	prev = NULL;
263 	for (i = 0; i < nargs - 1; i++) {
264 		nl = makenl(args[i]);
265 		if (prev == NULL)
266 			files = prev = nl;
267 		else {
268 			prev->n_next = nl;
269 			prev = nl;
270 		}
271 	}
272 
273 	cp = args[i];
274 	if ((dest = strchr(cp, ':')) != NULL)
275 		*dest++ = '\0';
276 	tnl.n_name = cp;
277 	hosts = expand(&tnl, E_ALL);
278 	if (nerrs)
279 		exit(1);
280 
281 	if (dest == NULL || *dest == '\0')
282 		cmds = NULL;
283 	else {
284 		cmds = makesubcmd(INSTALL);
285 		cmds->sc_options = options;
286 		cmds->sc_name = dest;
287 	}
288 
289 	if (debug) {
290 		printf("docmdargs()\nfiles = ");
291 		prnames(files);
292 		printf("hosts = ");
293 		prnames(hosts);
294 	}
295 	insert(NULL, files, hosts, cmds);
296 	docmds(NULL, 0, NULL);
297 }
298 
299 /*
300  * Print a list of NAME blocks (mostly for debugging).
301  */
302 void
303 prnames(nl)
304 	struct namelist *nl;
305 {
306 	printf("( ");
307 	while (nl != NULL) {
308 		printf("%s ", nl->n_name);
309 		nl = nl->n_next;
310 	}
311 	printf(")\n");
312 }
313