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