main.c revision 1.13 1 /* $NetBSD: main.c,v 1.13 2002/06/14 01:18:55 wiz 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.13 2002/06/14 01:18:55 wiz Exp $");
47 #endif
48 #endif /* not lint */
49
50 #include <sys/types.h>
51
52 #include <err.h>
53 #include <errno.h>
54 #include <pwd.h>
55
56 #include "defs.h"
57
58 #define NHOSTS 100
59
60 /*
61 * Remote distribution program.
62 */
63
64 char *distfile = NULL;
65 #define _RDIST_TMP "/rdistXXXXXX"
66 char tempfile[sizeof _PATH_TMP + sizeof _RDIST_TMP + 1];
67 char *tempname;
68
69 int debug; /* debugging flag */
70 int nflag; /* NOP flag, just print commands without executing */
71 int qflag; /* Quiet. Don't print messages */
72 int options; /* global options */
73 int iamremote; /* act as remote server for transfering files */
74
75 FILE *fin = NULL; /* input file pointer */
76 int rem = -1; /* file descriptor to remote source/sink process */
77 char host[MAXHOSTNAMELEN + 1]; /* host name */
78 int nerrs; /* number of errors while sending/receiving */
79 char user[34]; /* user's name */
80 char homedir[PATH_MAX]; /* user's home directory */
81 uid_t userid; /* user's user ID */
82 gid_t groupid; /* user's group ID */
83
84 struct passwd *pw; /* pointer to static area used by getpwent */
85 struct group *gr; /* pointer to static area used by getgrent */
86
87 int main(int, char **);
88 static void usage(void);
89 static void docmdargs(int, char *[]);
90
91 int
92 main(int argc, char **argv)
93 {
94 char *arg;
95 int cmdargs = 0;
96 char *dhosts[NHOSTS], **hp = dhosts;
97 int fd;
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 host[sizeof(host) - 1] = '\0';
109 strcpy(tempfile, _PATH_TMP);
110 strcat(tempfile, _RDIST_TMP);
111 if ((tempname = strrchr(tempfile, '/')) != 0)
112 tempname++;
113 else
114 tempname = tempfile;
115
116 while (--argc > 0) {
117 if ((arg = *++argv)[0] != '-')
118 break;
119 if (!strcmp(arg, "-Server"))
120 iamremote++;
121 else while (*++arg)
122 switch (*arg) {
123 case 'f':
124 if (--argc <= 0)
125 usage();
126 distfile = *++argv;
127 if (distfile[0] == '-' && distfile[1] == '\0')
128 fin = stdin;
129 break;
130
131 case 'm':
132 if (--argc <= 0)
133 usage();
134 if (hp >= &dhosts[NHOSTS-2]) {
135 fprintf(stderr, "rdist: too many destination hosts\n");
136 exit(1);
137 }
138 *hp++ = *++argv;
139 break;
140
141 case 'd':
142 if (--argc <= 0)
143 usage();
144 define(*++argv);
145 break;
146
147 case 'D':
148 debug++;
149 break;
150
151 case 'c':
152 cmdargs++;
153 break;
154
155 case 'n':
156 if (options & VERIFY) {
157 printf("rdist: -n overrides -v\n");
158 options &= ~VERIFY;
159 }
160 nflag++;
161 break;
162
163 case 'q':
164 qflag++;
165 break;
166
167 case 'b':
168 options |= COMPARE;
169 break;
170
171 case 'R':
172 options |= REMOVE;
173 break;
174
175 case 'v':
176 if (nflag) {
177 printf("rdist: -n overrides -v\n");
178 break;
179 }
180 options |= VERIFY;
181 break;
182
183 case 'w':
184 options |= WHOLE;
185 break;
186
187 case 'y':
188 options |= YOUNGER;
189 break;
190
191 case 'h':
192 options |= FOLLOW;
193 break;
194
195 case 'i':
196 options |= IGNLNKS;
197 break;
198
199 default:
200 usage();
201 }
202 }
203 *hp = NULL;
204
205 seteuid(userid);
206 fd = mkstemp(tempfile);
207 if (fd == -1)
208 err(1, "could not make a temporary file");
209 close (fd);
210
211 if (iamremote) {
212 server();
213 unlink(tempfile);
214 exit(nerrs != 0);
215 }
216
217 if (cmdargs)
218 docmdargs(argc, argv);
219 else {
220 if (fin == NULL) {
221 if (distfile == NULL) {
222 if ((fin = fopen("distfile","r")) == NULL)
223 fin = fopen("Distfile", "r");
224 } else
225 fin = fopen(distfile, "r");
226 if (fin == NULL) {
227 perror(distfile ? distfile : "distfile");
228 unlink(tempfile);
229 exit(1);
230 }
231 }
232 yyparse();
233 if (nerrs == 0)
234 docmds(dhosts, argc, argv);
235 }
236
237 unlink(tempfile);
238 exit(nerrs != 0);
239 }
240
241 static void
242 usage(void)
243 {
244 printf("Usage: rdist [-nqbRhivwyD] [-f distfile] [-d var=value] [-m host] [file ...]\n");
245 printf("or: rdist [-nqbRhivwyD] -c source [...] machine[:dest]\n");
246 exit(1);
247 }
248
249 /*
250 * rcp like interface for distributing files.
251 */
252 static void
253 docmdargs(int nargs, char **args)
254 {
255 struct namelist *nl, *prev;
256 char *cp;
257 struct namelist *files, *hosts;
258 struct subcmd *cmds;
259 char *dest;
260 static struct namelist tnl = { NULL, NULL };
261 int i;
262
263 if (nargs < 2)
264 usage();
265
266 files = NULL;
267 prev = NULL;
268 for (i = 0; i < nargs - 1; i++) {
269 nl = makenl(args[i]);
270 if (prev == NULL)
271 files = prev = nl;
272 else {
273 prev->n_next = nl;
274 prev = nl;
275 }
276 }
277
278 cp = args[i];
279 if ((dest = strchr(cp, ':')) != NULL)
280 *dest++ = '\0';
281 tnl.n_name = cp;
282 hosts = expand(&tnl, E_ALL);
283 if (nerrs)
284 return;
285
286 if (dest == NULL || *dest == '\0')
287 cmds = NULL;
288 else {
289 cmds = makesubcmd(INSTALL);
290 cmds->sc_options = options;
291 cmds->sc_name = dest;
292 }
293
294 if (debug) {
295 printf("docmdargs()\nfiles = ");
296 prnames(files);
297 printf("hosts = ");
298 prnames(hosts);
299 }
300 insert(NULL, files, hosts, cmds);
301 docmds(NULL, 0, NULL);
302 }
303
304 /*
305 * Print a list of NAME blocks (mostly for debugging).
306 */
307 void
308 prnames(struct namelist *nl)
309 {
310 printf("( ");
311 while (nl != NULL) {
312 printf("%s ", nl->n_name);
313 nl = nl->n_next;
314 }
315 printf(")\n");
316 }
317