Annotation of src/usr.bin/rsync/main.c, Revision 1.54
1.54 ! claudio 1: /* $Id: main.c,v 1.53 2021/03/31 19:45:16 job Exp $ */
1.1 benno 2: /*
3: * Copyright (c) 2019 Kristaps Dzonsons <kristaps@bsd.lv>
4: *
5: * Permission to use, copy, modify, and distribute this software for any
6: * purpose with or without fee is hereby granted, provided that the above
7: * copyright notice and this permission notice appear in all copies.
8: *
9: * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10: * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11: * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
12: * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13: * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
14: * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
15: * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16: */
17: #include <sys/stat.h>
18: #include <sys/socket.h>
19: #include <sys/wait.h>
20:
21: #include <assert.h>
22: #include <err.h>
23: #include <getopt.h>
24: #include <stdint.h>
25: #include <stdio.h>
26: #include <stdlib.h>
27: #include <string.h>
28: #include <unistd.h>
29:
30: #include "extern.h"
31:
1.45 benno 32: int verbose;
1.50 claudio 33: int poll_timeout;
1.45 benno 34:
1.1 benno 35: /*
36: * A remote host is has a colon before the first path separator.
37: * This works for rsh remote hosts (host:/foo/bar), implicit rsync
38: * remote hosts (host::/foo/bar), and explicit (rsync://host/foo).
39: * Return zero if local, non-zero if remote.
40: */
41: static int
42: fargs_is_remote(const char *v)
43: {
44: size_t pos;
45:
46: pos = strcspn(v, ":/");
1.6 deraadt 47: return v[pos] == ':';
1.1 benno 48: }
49:
50: /*
51: * Test whether a remote host is specifically an rsync daemon.
52: * Return zero if not, non-zero if so.
53: */
54: static int
55: fargs_is_daemon(const char *v)
56: {
57: size_t pos;
58:
1.6 deraadt 59: if (strncasecmp(v, "rsync://", 8) == 0)
1.1 benno 60: return 1;
61:
62: pos = strcspn(v, ":/");
1.6 deraadt 63: return v[pos] == ':' && v[pos + 1] == ':';
1.1 benno 64: }
65:
66: /*
67: * Take the command-line filenames (e.g., rsync foo/ bar/ baz/) and
68: * determine our operating mode.
69: * For example, if the first argument is a remote file, this means that
70: * we're going to transfer from the remote to the local.
71: * We also make sure that the arguments are consistent, that is, if
72: * we're going to transfer from the local to the remote, that no
73: * filenames for the local transfer indicate remote hosts.
74: * Always returns the parsed and sanitised options.
75: */
76: static struct fargs *
1.26 deraadt 77: fargs_parse(size_t argc, char *argv[], struct opts *opts)
1.1 benno 78: {
79: struct fargs *f = NULL;
1.26 deraadt 80: char *cp, *ccp;
1.1 benno 81: size_t i, j, len = 0;
82:
83: /* Allocations. */
84:
1.6 deraadt 85: if ((f = calloc(1, sizeof(struct fargs))) == NULL)
1.54 ! claudio 86: err(ERR_NOMEM, NULL);
1.1 benno 87:
88: f->sourcesz = argc - 1;
1.6 deraadt 89: if ((f->sources = calloc(f->sourcesz, sizeof(char *))) == NULL)
1.54 ! claudio 90: err(ERR_NOMEM, NULL);
1.1 benno 91:
92: for (i = 0; i < argc - 1; i++)
1.6 deraadt 93: if ((f->sources[i] = strdup(argv[i])) == NULL)
1.54 ! claudio 94: err(ERR_NOMEM, NULL);
1.1 benno 95:
1.6 deraadt 96: if ((f->sink = strdup(argv[i])) == NULL)
1.54 ! claudio 97: err(ERR_NOMEM, NULL);
1.1 benno 98:
99: /*
100: * Test files for its locality.
101: * If the last is a remote host, then we're sending from the
102: * local to the remote host ("sender" mode).
103: * If the first, remote to local ("receiver" mode).
104: * If neither, a local transfer in sender style.
105: */
106:
107: f->mode = FARGS_SENDER;
108:
109: if (fargs_is_remote(f->sink)) {
110: f->mode = FARGS_SENDER;
1.6 deraadt 111: if ((f->host = strdup(f->sink)) == NULL)
1.54 ! claudio 112: err(ERR_NOMEM, NULL);
1.1 benno 113: }
114:
115: if (fargs_is_remote(f->sources[0])) {
1.6 deraadt 116: if (f->host != NULL)
1.54 ! claudio 117: errx(ERR_SYNTAX, "both source and destination "
! 118: "cannot be remote files");
1.1 benno 119: f->mode = FARGS_RECEIVER;
1.6 deraadt 120: if ((f->host = strdup(f->sources[0])) == NULL)
1.54 ! claudio 121: err(ERR_NOMEM, NULL);
1.1 benno 122: }
123:
1.6 deraadt 124: if (f->host != NULL) {
125: if (strncasecmp(f->host, "rsync://", 8) == 0) {
1.30 benno 126: /* rsync://host[:port]/module[/path] */
1.1 benno 127: f->remote = 1;
128: len = strlen(f->host) - 8 + 1;
129: memmove(f->host, f->host + 8, len);
1.6 deraadt 130: if ((cp = strchr(f->host, '/')) == NULL)
1.54 ! claudio 131: errx(ERR_SYNTAX,
! 132: "rsync protocol requires a module name");
1.1 benno 133: *cp++ = '\0';
134: f->module = cp;
1.6 deraadt 135: if ((cp = strchr(f->module, '/')) != NULL)
1.1 benno 136: *cp = '\0';
1.52 claudio 137: if ((cp = strchr(f->host, ':')) != NULL) {
1.26 deraadt 138: /* host:port --> extract port */
139: *cp++ = '\0';
140: opts->port = cp;
141: }
1.1 benno 142: } else {
143: /* host:[/path] */
144: cp = strchr(f->host, ':');
1.6 deraadt 145: assert(cp != NULL);
1.1 benno 146: *cp++ = '\0';
1.6 deraadt 147: if (*cp == ':') {
1.1 benno 148: /* host::module[/path] */
149: f->remote = 1;
150: f->module = ++cp;
151: cp = strchr(f->module, '/');
1.6 deraadt 152: if (cp != NULL)
1.1 benno 153: *cp = '\0';
154: }
155: }
1.6 deraadt 156: if ((len = strlen(f->host)) == 0)
1.54 ! claudio 157: errx(ERR_SYNTAX, "empty remote host");
1.6 deraadt 158: if (f->remote && strlen(f->module) == 0)
1.54 ! claudio 159: errx(ERR_SYNTAX, "empty remote module");
1.1 benno 160: }
161:
162: /* Make sure we have the same "hostspec" for all files. */
163:
1.4 deraadt 164: if (!f->remote) {
1.6 deraadt 165: if (f->mode == FARGS_SENDER)
1.1 benno 166: for (i = 0; i < f->sourcesz; i++) {
1.4 deraadt 167: if (!fargs_is_remote(f->sources[i]))
1.1 benno 168: continue;
1.54 ! claudio 169: errx(ERR_SYNTAX,
1.35 deraadt 170: "remote file in list of local sources: %s",
171: f->sources[i]);
1.1 benno 172: }
1.6 deraadt 173: if (f->mode == FARGS_RECEIVER)
1.1 benno 174: for (i = 0; i < f->sourcesz; i++) {
175: if (fargs_is_remote(f->sources[i]) &&
1.4 deraadt 176: !fargs_is_daemon(f->sources[i]))
1.1 benno 177: continue;
178: if (fargs_is_daemon(f->sources[i]))
1.54 ! claudio 179: errx(ERR_SYNTAX,
! 180: "remote daemon in list of remote "
! 181: "sources: %s", f->sources[i]);
! 182: errx(ERR_SYNTAX, "local file in list of "
! 183: "remote sources: %s", f->sources[i]);
1.1 benno 184: }
185: } else {
1.6 deraadt 186: if (f->mode != FARGS_RECEIVER)
1.54 ! claudio 187: errx(ERR_SYNTAX, "sender mode for remote "
1.1 benno 188: "daemon receivers not yet supported");
189: for (i = 0; i < f->sourcesz; i++) {
190: if (fargs_is_daemon(f->sources[i]))
191: continue;
1.54 ! claudio 192: errx(ERR_SYNTAX, "non-remote daemon file "
1.1 benno 193: "in list of remote daemon sources: "
194: "%s", f->sources[i]);
195: }
196: }
197:
198: /*
199: * If we're not remote and a sender, strip our hostname.
200: * Then exit if we're a sender or a local connection.
201: */
202:
1.4 deraadt 203: if (!f->remote) {
1.6 deraadt 204: if (f->host == NULL)
1.1 benno 205: return f;
1.6 deraadt 206: if (f->mode == FARGS_SENDER) {
207: assert(f->host != NULL);
1.1 benno 208: assert(len > 0);
209: j = strlen(f->sink);
210: memmove(f->sink, f->sink + len + 1, j - len);
211: return f;
1.6 deraadt 212: } else if (f->mode != FARGS_RECEIVER)
1.1 benno 213: return f;
214: }
215:
216: /*
217: * Now strip the hostnames from the remote host.
218: * rsync://host/module/path -> module/path
219: * host::module/path -> module/path
220: * host:path -> path
221: * Also make sure that the remote hosts are the same.
222: */
223:
1.6 deraadt 224: assert(f->host != NULL);
1.1 benno 225: assert(len > 0);
226:
227: for (i = 0; i < f->sourcesz; i++) {
228: cp = f->sources[i];
229: j = strlen(cp);
230: if (f->remote &&
1.6 deraadt 231: strncasecmp(cp, "rsync://", 8) == 0) {
1.1 benno 232: /* rsync://path */
233: cp += 8;
1.29 benno 234: if ((ccp = strchr(cp, ':'))) /* skip :port */
1.26 deraadt 235: *ccp = '\0';
1.1 benno 236: if (strncmp(cp, f->host, len) ||
1.11 benno 237: (cp[len] != '/' && cp[len] != '\0'))
1.54 ! claudio 238: errx(ERR_SYNTAX, "different remote host: %s",
1.35 deraadt 239: f->sources[i]);
1.1 benno 240: memmove(f->sources[i],
241: f->sources[i] + len + 8 + 1,
242: j - len - 8);
1.6 deraadt 243: } else if (f->remote && strncmp(cp, "::", 2) == 0) {
1.1 benno 244: /* ::path */
245: memmove(f->sources[i],
246: f->sources[i] + 2, j - 1);
247: } else if (f->remote) {
248: /* host::path */
249: if (strncmp(cp, f->host, len) ||
1.6 deraadt 250: (cp[len] != ':' && cp[len] != '\0'))
1.54 ! claudio 251: errx(ERR_SYNTAX, "different remote host: %s",
1.35 deraadt 252: f->sources[i]);
1.11 benno 253: memmove(f->sources[i], f->sources[i] + len + 2,
254: j - len - 1);
1.6 deraadt 255: } else if (cp[0] == ':') {
1.1 benno 256: /* :path */
257: memmove(f->sources[i], f->sources[i] + 1, j);
258: } else {
259: /* host:path */
260: if (strncmp(cp, f->host, len) ||
1.6 deraadt 261: (cp[len] != ':' && cp[len] != '\0'))
1.54 ! claudio 262: errx(ERR_SYNTAX, "different remote host: %s",
1.35 deraadt 263: f->sources[i]);
1.1 benno 264: memmove(f->sources[i],
265: f->sources[i] + len + 1, j - len);
266: }
267: }
268:
269: return f;
270: }
271:
272: int
273: main(int argc, char *argv[])
274: {
275: struct opts opts;
1.2 benno 276: pid_t child;
1.46 deraadt 277: int fds[2], sd = -1, rc, c, st, i;
1.32 deraadt 278: struct sess sess;
1.1 benno 279: struct fargs *fargs;
1.32 deraadt 280: char **args;
1.50 claudio 281: const char *errstr;
282: const struct option lopts[] = {
1.26 deraadt 283: { "port", required_argument, NULL, 3 },
1.21 deraadt 284: { "rsh", required_argument, NULL, 'e' },
1.8 deraadt 285: { "rsync-path", required_argument, NULL, 1 },
286: { "sender", no_argument, &opts.sender, 1 },
287: { "server", no_argument, &opts.server, 1 },
1.20 deraadt 288: { "dry-run", no_argument, &opts.dry_run, 1 },
289: { "version", no_argument, NULL, 2 },
1.23 deraadt 290: { "archive", no_argument, NULL, 'a' },
1.21 deraadt 291: { "help", no_argument, NULL, 'h' },
1.36 deraadt 292: { "compress", no_argument, NULL, 'z' },
1.37 naddy 293: { "del", no_argument, &opts.del, 1 },
1.20 deraadt 294: { "delete", no_argument, &opts.del, 1 },
295: { "devices", no_argument, &opts.devices, 1 },
296: { "no-devices", no_argument, &opts.devices, 0 },
297: { "group", no_argument, &opts.preserve_gids, 1 },
298: { "no-group", no_argument, &opts.preserve_gids, 0 },
1.8 deraadt 299: { "links", no_argument, &opts.preserve_links, 1 },
1.20 deraadt 300: { "no-links", no_argument, &opts.preserve_links, 0 },
301: { "owner", no_argument, &opts.preserve_uids, 1 },
302: { "no-owner", no_argument, &opts.preserve_uids, 0 },
1.8 deraadt 303: { "perms", no_argument, &opts.preserve_perms, 1 },
1.20 deraadt 304: { "no-perms", no_argument, &opts.preserve_perms, 0 },
1.31 benno 305: { "numeric-ids", no_argument, &opts.numeric_ids, 1 },
1.8 deraadt 306: { "recursive", no_argument, &opts.recursive, 1 },
1.20 deraadt 307: { "no-recursive", no_argument, &opts.recursive, 0 },
1.19 florian 308: { "specials", no_argument, &opts.specials, 1 },
1.21 deraadt 309: { "no-specials", no_argument, &opts.specials, 0 },
1.50 claudio 310: { "timeout", required_argument, NULL, 5 },
1.20 deraadt 311: { "times", no_argument, &opts.preserve_times, 1 },
312: { "no-times", no_argument, &opts.preserve_times, 0 },
1.45 benno 313: { "verbose", no_argument, &verbose, 1 },
314: { "no-verbose", no_argument, &verbose, 0 },
1.48 claudio 315: { "address", required_argument, NULL, 4 },
1.53 job 316: { "no-motd", no_argument, NULL, 6 },
1.8 deraadt 317: { NULL, 0, NULL, 0 }};
1.1 benno 318:
319: /* Global pledge. */
320:
1.19 florian 321: if (pledge("stdio unix rpath wpath cpath dpath inet fattr chown dns getpw proc exec unveil",
1.6 deraadt 322: NULL) == -1)
1.54 ! claudio 323: err(ERR_IPC, "pledge");
1.1 benno 324:
325: memset(&opts, 0, sizeof(struct opts));
326:
1.44 bket 327: while ((c = getopt_long(argc, argv, "Dae:ghlnoprtvxz", lopts, NULL))
1.30 benno 328: != -1) {
1.1 benno 329: switch (c) {
1.19 florian 330: case 'D':
331: opts.devices = 1;
332: opts.specials = 1;
333: break;
334: case 'a':
335: opts.recursive = 1;
336: opts.preserve_links = 1;
337: opts.preserve_perms = 1;
338: opts.preserve_times = 1;
339: opts.preserve_gids = 1;
340: opts.preserve_uids = 1;
341: opts.devices = 1;
342: opts.specials = 1;
343: break;
1.1 benno 344: case 'e':
1.9 deraadt 345: opts.ssh_prog = optarg;
1.1 benno 346: break;
1.10 benno 347: case 'g':
348: opts.preserve_gids = 1;
349: break;
1.1 benno 350: case 'l':
351: opts.preserve_links = 1;
352: break;
353: case 'n':
354: opts.dry_run = 1;
355: break;
1.13 florian 356: case 'o':
357: opts.preserve_uids = 1;
358: break;
1.1 benno 359: case 'p':
360: opts.preserve_perms = 1;
361: break;
362: case 'r':
363: opts.recursive = 1;
364: break;
365: case 't':
366: opts.preserve_times = 1;
367: break;
368: case 'v':
1.45 benno 369: verbose++;
1.1 benno 370: break;
1.44 bket 371: case 'x':
372: opts.one_file_system++;
373: break;
1.36 deraadt 374: case 'z':
375: fprintf(stderr, "%s: -z not supported yet\n", getprogname());
376: break;
1.1 benno 377: case 0:
378: /* Non-NULL flag values (e.g., --sender). */
379: break;
380: case 1:
381: opts.rsync_path = optarg;
382: break;
1.17 deraadt 383: case 2:
384: fprintf(stderr, "openrsync: protocol version %u\n",
385: RSYNC_PROTOCOL);
386: exit(0);
1.26 deraadt 387: case 3:
388: opts.port = optarg;
389: break;
1.48 claudio 390: case 4:
391: opts.address = optarg;
392: break;
1.50 claudio 393: case 5:
394: poll_timeout = strtonum(optarg, 0, 60*60, &errstr);
395: if (errstr != NULL)
1.54 ! claudio 396: errx(ERR_SYNTAX, "timeout is %s: %s",
! 397: errstr, optarg);
1.50 claudio 398: break;
1.53 job 399: case 6:
400: opts.no_motd = 1;
401: break;
1.21 deraadt 402: case 'h':
1.1 benno 403: default:
404: goto usage;
405: }
406: }
407:
408: argc -= optind;
409: argv += optind;
410:
411: /* FIXME: reference implementation rsync accepts this. */
412:
413: if (argc < 2)
414: goto usage;
415:
1.26 deraadt 416: if (opts.port == NULL)
1.29 benno 417: opts.port = "rsync";
1.26 deraadt 418:
1.51 claudio 419: /* by default and for --timeout=0 disable poll_timeout */
420: if (poll_timeout == 0)
421: poll_timeout = -1;
422: else
423: poll_timeout *= 1000;
424:
1.1 benno 425: /*
426: * This is what happens when we're started with the "hidden"
427: * --server option, which is invoked for the rsync on the remote
428: * host by the parent.
429: */
430:
1.32 deraadt 431: if (opts.server)
432: exit(rsync_server(&opts, (size_t)argc, argv));
1.50 claudio 433:
1.1 benno 434: /*
435: * Now we know that we're the client on the local machine
436: * invoking rsync(1).
437: * At this point, we need to start the client and server
438: * initiation logic.
439: * The client is what we continue running on this host; the
440: * server is what we'll use to connect to the remote and
441: * invoke rsync with the --server option.
442: */
443:
1.26 deraadt 444: fargs = fargs_parse(argc, argv, &opts);
1.6 deraadt 445: assert(fargs != NULL);
1.1 benno 446:
447: /*
448: * If we're contacting an rsync:// daemon, then we don't need to
449: * fork, because we won't start a server ourselves.
1.41 naddy 450: * Route directly into the socket code, unless a remote shell
451: * has explicitly been specified.
1.1 benno 452: */
453:
1.41 naddy 454: if (fargs->remote && opts.ssh_prog == NULL) {
1.6 deraadt 455: assert(fargs->mode == FARGS_RECEIVER);
1.47 naddy 456: if ((rc = rsync_connect(&opts, &sd, fargs)) == 0) {
1.41 naddy 457: rc = rsync_socket(&opts, sd, fargs);
1.47 naddy 458: close(sd);
459: }
1.41 naddy 460: exit(rc);
1.1 benno 461: }
462:
463: /* Drop the dns/inet possibility. */
464:
1.19 florian 465: if (pledge("stdio unix rpath wpath cpath dpath fattr chown getpw proc exec unveil",
1.6 deraadt 466: NULL) == -1)
1.54 ! claudio 467: err(ERR_IPC, "pledge");
1.1 benno 468:
469: /* Create a bidirectional socket and start our child. */
470:
1.6 deraadt 471: if (socketpair(AF_UNIX, SOCK_STREAM | SOCK_NONBLOCK, 0, fds) == -1)
1.54 ! claudio 472: err(ERR_IPC, "socketpair");
1.1 benno 473:
1.32 deraadt 474: switch ((child = fork())) {
475: case -1:
1.54 ! claudio 476: err(ERR_IPC, "fork");
1.32 deraadt 477: case 0:
1.1 benno 478: close(fds[0]);
1.32 deraadt 479: if (pledge("stdio exec", NULL) == -1)
1.54 ! claudio 480: err(ERR_IPC, "pledge");
1.32 deraadt 481:
482: memset(&sess, 0, sizeof(struct sess));
483: sess.opts = &opts;
1.1 benno 484:
1.54 ! claudio 485: args = fargs_cmdline(&sess, fargs, NULL);
1.1 benno 486:
1.32 deraadt 487: for (i = 0; args[i] != NULL; i++)
1.45 benno 488: LOG2("exec[%d] = %s", i, args[i]);
1.1 benno 489:
1.32 deraadt 490: /* Make sure the child's stdin is from the sender. */
1.54 ! claudio 491: if (dup2(fds[1], STDIN_FILENO) == -1)
! 492: err(ERR_IPC, "dup2");
! 493: if (dup2(fds[1], STDOUT_FILENO) == -1)
! 494: err(ERR_IPC, "dup2");
1.32 deraadt 495: execvp(args[0], args);
1.54 ! claudio 496: _exit(ERR_IPC);
1.1 benno 497: /* NOTREACHED */
1.32 deraadt 498: default:
499: close(fds[1]);
1.41 naddy 500: if (!fargs->remote)
501: rc = rsync_client(&opts, fds[0], fargs);
502: else
503: rc = rsync_socket(&opts, fds[0], fargs);
1.32 deraadt 504: break;
1.1 benno 505: }
506:
1.47 naddy 507: close(fds[0]);
1.1 benno 508:
1.6 deraadt 509: if (waitpid(child, &st, 0) == -1)
1.54 ! claudio 510: err(ERR_WAITPID, "waitpid");
1.34 deraadt 511:
512: /*
513: * If we don't already have an error (rc == 0), then inherit the
514: * error code of rsync_server() if it has exited.
515: * If it hasn't exited, it overrides our return value.
516: */
517:
1.54 ! claudio 518: if (rc == 0) {
! 519: if (WIFEXITED(st))
! 520: rc = WEXITSTATUS(st);
! 521: else if (WIFSIGNALED(st))
! 522: rc = ERR_TERMIMATED;
! 523: else
! 524: rc = ERR_WAITPID;
! 525: }
1.34 deraadt 526:
1.32 deraadt 527: exit(rc);
1.1 benno 528: usage:
1.42 schwarze 529: fprintf(stderr, "usage: %s"
1.49 deraadt 530: " [-aDglnoprtvx] [-e program] [--address=sourceaddr] [--del]\n"
1.53 job 531: "\t[--no-motd] [--numeric-ids] [--port=portnumber] "
532: "[--rsync-path=program]\n\t[--timeout=seconds] [--version] "
533: "source ... directory\n",
1.32 deraadt 534: getprogname());
1.54 ! claudio 535: exit(ERR_SYNTAX);
1.1 benno 536: }