Preparing for release of 3.0.0pre3
[rsync/rsync.git] / clientserver.c
... / ...
CommitLineData
1/*
2 * The socket based protocol for setting up a connection with rsyncd.
3 *
4 * Copyright (C) 1998-2001 Andrew Tridgell <tridge@samba.org>
5 * Copyright (C) 2001-2002 Martin Pool <mbp@samba.org>
6 * Copyright (C) 2002-2007 Wayne Davison
7 *
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 3 of the License, or
11 * (at your option) any later version.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, visit the http://fsf.org website.
20 */
21
22#include "rsync.h"
23#include "ifuncs.h"
24
25extern int verbose;
26extern int output_motd;
27extern int list_only;
28extern int am_sender;
29extern int am_server;
30extern int am_daemon;
31extern int am_root;
32extern int rsync_port;
33extern int ignore_errors;
34extern int kluge_around_eof;
35extern int daemon_over_rsh;
36extern int sanitize_paths;
37extern int filesfrom_fd;
38extern int remote_protocol;
39extern int protocol_version;
40extern int io_timeout;
41extern int no_detach;
42extern int default_af_hint;
43extern int logfile_format_has_i;
44extern int logfile_format_has_o_or_i;
45extern mode_t orig_umask;
46extern char *bind_address;
47extern char *sockopts;
48extern char *config_file;
49extern char *logfile_format;
50extern char *files_from;
51extern char *tmpdir;
52extern struct chmod_mode_struct *chmod_modes;
53extern struct filter_list_struct server_filter_list;
54extern char curr_dir[];
55
56char *auth_user;
57int read_only = 0;
58int module_id = -1;
59struct chmod_mode_struct *daemon_chmod_modes;
60
61/* module_dirlen is the length of the module_dir string when in daemon
62 * mode, not chrooted, and the path is not "/"; otherwise 0. */
63char *module_dir = NULL;
64unsigned int module_dirlen = 0;
65
66static int rl_nulls = 0;
67
68#ifdef HAVE_SIGACTION
69static struct sigaction sigact;
70#endif
71
72/**
73 * Run a client connected to an rsyncd. The alternative to this
74 * function for remote-shell connections is do_cmd().
75 *
76 * After negotiating which module to use and reading the server's
77 * motd, this hands over to client_run(). Telling the server the
78 * module will cause it to chroot/setuid/etc.
79 *
80 * Instead of doing a transfer, the client may at this stage instead
81 * get a listing of remote modules and exit.
82 *
83 * @return -1 for error in startup, or the result of client_run().
84 * Either way, it eventually gets passed to exit_cleanup().
85 **/
86int start_socket_client(char *host, int remote_argc, char *remote_argv[],
87 int argc, char *argv[])
88{
89 int fd, ret;
90 char *p, *user = NULL;
91
92 /* This is redundant with code in start_inband_exchange(), but this
93 * short-circuits a problem in the client before we open a socket,
94 * and the extra check won't hurt. */
95 if (**remote_argv == '/') {
96 rprintf(FERROR,
97 "ERROR: The remote path must start with a module name not a /\n");
98 return -1;
99 }
100
101 if ((p = strrchr(host, '@')) != NULL) {
102 user = host;
103 host = p+1;
104 *p = '\0';
105 }
106
107 fd = open_socket_out_wrapped(host, rsync_port, bind_address,
108 default_af_hint);
109 if (fd == -1)
110 exit_cleanup(RERR_SOCKETIO);
111
112 set_socket_options(fd, sockopts);
113
114 ret = start_inband_exchange(fd, fd, user, remote_argc, remote_argv);
115
116 return ret ? ret : client_run(fd, fd, -1, argc, argv);
117}
118
119static int exchange_protocols(int f_in, int f_out, char *buf, size_t bufsiz, int am_client)
120{
121 int remote_sub = -1;
122#if SUBPROTOCOL_VERSION != 0
123 int our_sub = protocol_version < PROTOCOL_VERSION ? 0 : SUBPROTOCOL_VERSION;
124#else
125 int our_sub = 0;
126#endif
127 char *motd;
128
129 io_printf(f_out, "@RSYNCD: %d.%d\n", protocol_version, our_sub);
130
131 if (!am_client) {
132 motd = lp_motd_file();
133 if (motd && *motd) {
134 FILE *f = fopen(motd,"r");
135 while (f && !feof(f)) {
136 int len = fread(buf, 1, bufsiz - 1, f);
137 if (len > 0)
138 write_buf(f_out, buf, len);
139 }
140 if (f)
141 fclose(f);
142 write_sbuf(f_out, "\n");
143 }
144 }
145
146 /* This strips the \n. */
147 if (!read_line_old(f_in, buf, bufsiz)) {
148 if (am_client)
149 rprintf(FERROR, "rsync: did not see server greeting\n");
150 return -1;
151 }
152
153 if (sscanf(buf, "@RSYNCD: %d.%d", &remote_protocol, &remote_sub) < 1) {
154 if (am_client)
155 rprintf(FERROR, "rsync: server sent \"%s\" rather than greeting\n", buf);
156 else
157 io_printf(f_out, "@ERROR: protocol startup error\n");
158 return -1;
159 }
160
161 if (remote_sub < 0) {
162 if (remote_protocol == 30) {
163 if (am_client)
164 rprintf(FERROR, "rsync: server is speaking an incompatible beta of protocol 30\n");
165 else
166 io_printf(f_out, "@ERROR: your client is speaking an incompatible beta of protocol 30\n");
167 return -1;
168 }
169 remote_sub = 0;
170 }
171
172 if (protocol_version > remote_protocol) {
173 protocol_version = remote_protocol;
174 if (remote_sub)
175 protocol_version--;
176 } else if (protocol_version == remote_protocol) {
177 if (remote_sub != our_sub)
178 protocol_version--;
179 }
180#if SUBPROTOCOL_VERSION != 0
181 else if (protocol_version < remote_protocol) {
182 if (our_sub)
183 protocol_version--;
184 }
185#endif
186
187 if (protocol_version >= 30)
188 rl_nulls = 1;
189
190 return 0;
191}
192
193int start_inband_exchange(int f_in, int f_out, const char *user, int argc, char *argv[])
194{
195 int i;
196 char line[BIGPATHBUFLEN];
197 char *sargs[MAX_ARGS];
198 int sargc = 0;
199 char *p, *path = *argv;
200
201 if (argc == 0 && !am_sender)
202 list_only |= 1;
203
204 if (*path == '/') {
205 rprintf(FERROR,
206 "ERROR: The remote path must start with a module name\n");
207 return -1;
208 }
209
210 if (!user)
211 user = getenv("USER");
212 if (!user)
213 user = getenv("LOGNAME");
214
215 if (exchange_protocols(f_in, f_out, line, sizeof line, 1) < 0)
216 return -1;
217
218 /* set daemon_over_rsh to false since we need to build the
219 * true set of args passed through the rsh/ssh connection;
220 * this is a no-op for direct-socket-connection mode */
221 daemon_over_rsh = 0;
222 server_options(sargs, &sargc);
223
224 if (sargc >= MAX_ARGS - 2)
225 goto arg_overflow;
226
227 sargs[sargc++] = ".";
228
229 while (argc > 0) {
230 if (sargc >= MAX_ARGS - 1) {
231 arg_overflow:
232 rprintf(FERROR, "internal: args[] overflowed in do_cmd()\n");
233 exit_cleanup(RERR_SYNTAX);
234 }
235 sargs[sargc++] = *argv++;
236 argc--;
237 }
238
239 sargs[sargc] = NULL;
240
241 if (verbose > 1)
242 print_child_argv("sending daemon args:", sargs);
243
244 p = strchr(path, '/');
245 if (p) *p = '\0';
246 io_printf(f_out, "%s\n", path);
247 if (p) *p = '/';
248
249 /* Old servers may just drop the connection here,
250 rather than sending a proper EXIT command. Yuck. */
251 kluge_around_eof = list_only && protocol_version < 25 ? 1 : 0;
252
253 while (1) {
254 if (!read_line_old(f_in, line, sizeof line)) {
255 rprintf(FERROR, "rsync: didn't get server startup line\n");
256 return -1;
257 }
258
259 if (strncmp(line,"@RSYNCD: AUTHREQD ",18) == 0) {
260 auth_client(f_out, user, line+18);
261 continue;
262 }
263
264 if (strcmp(line,"@RSYNCD: OK") == 0)
265 break;
266
267 if (strcmp(line,"@RSYNCD: EXIT") == 0) {
268 /* This is sent by recent versions of the
269 * server to terminate the listing of modules.
270 * We don't want to go on and transfer
271 * anything; just exit. */
272 exit(0);
273 }
274
275 if (strncmp(line, "@ERROR", 6) == 0) {
276 rprintf(FERROR, "%s\n", line);
277 /* This is always fatal; the server will now
278 * close the socket. */
279 return -1;
280 }
281
282 /* This might be a MOTD line or a module listing, but there is
283 * no way to differentiate it. The manpage mentions this. */
284 if (output_motd)
285 rprintf(FINFO, "%s\n", line);
286 }
287 kluge_around_eof = 0;
288
289 if (rl_nulls) {
290 for (i = 0; i < sargc; i++) {
291 write_sbuf(f_out, sargs[i]);
292 write_byte(f_out, 0);
293 }
294 write_byte(f_out, 0);
295 } else {
296 for (i = 0; i < sargc; i++)
297 io_printf(f_out, "%s\n", sargs[i]);
298 write_sbuf(f_out, "\n");
299 }
300
301 if (protocol_version < 23) {
302 if (protocol_version == 22 || !am_sender)
303 io_start_multiplex_in();
304 }
305
306 return 0;
307}
308
309static char *finish_pre_exec(pid_t pid, int fd, char *request,
310 int argc, char *argv[])
311{
312 int j, status = -1;
313
314 if (!request)
315 request = "(NONE)";
316
317 write_buf(fd, request, strlen(request)+1);
318 for (j = 0; j < argc; j++)
319 write_buf(fd, argv[j], strlen(argv[j])+1);
320 write_byte(fd, 0);
321
322 close(fd);
323
324 if (wait_process(pid, &status, 0) < 0
325 || !WIFEXITED(status) || WEXITSTATUS(status) != 0) {
326 char *e;
327 if (asprintf(&e, "pre-xfer exec returned failure (%d)%s%s\n",
328 status, status < 0 ? ": " : "",
329 status < 0 ? strerror(errno) : "") < 0)
330 out_of_memory("finish_pre_exec");
331 return e;
332 }
333 return NULL;
334}
335
336static int read_arg_from_pipe(int fd, char *buf, int limit)
337{
338 char *bp = buf, *eob = buf + limit - 1;
339
340 while (1) {
341 if (read(fd, bp, 1) != 1)
342 return -1;
343 if (*bp == '\0')
344 break;
345 if (bp < eob)
346 bp++;
347 }
348 *bp = '\0';
349
350 return bp - buf;
351}
352
353static int rsync_module(int f_in, int f_out, int i, char *addr, char *host)
354{
355 int argc, opt_cnt;
356 char **argv;
357 char line[BIGPATHBUFLEN];
358 uid_t uid = (uid_t)-2; /* canonically "nobody" */
359 gid_t gid = (gid_t)-2;
360 char *p, *err_msg = NULL;
361 char *name = lp_name(i);
362 int use_chroot = lp_use_chroot(i);
363 int ret, pre_exec_fd = -1;
364 pid_t pre_exec_pid = 0;
365 char *request = NULL;
366
367 if (!allow_access(addr, host, lp_hosts_allow(i), lp_hosts_deny(i))) {
368 rprintf(FLOG, "rsync denied on module %s from %s (%s)\n",
369 name, host, addr);
370 if (!lp_list(i))
371 io_printf(f_out, "@ERROR: Unknown module '%s'\n", name);
372 else {
373 io_printf(f_out,
374 "@ERROR: access denied to %s from %s (%s)\n",
375 name, host, addr);
376 }
377 return -1;
378 }
379
380 if (am_daemon && am_server) {
381 rprintf(FLOG, "rsync allowed access on module %s from %s (%s)\n",
382 name, host, addr);
383 }
384
385 if (!claim_connection(lp_lock_file(i), lp_max_connections(i))) {
386 if (errno) {
387 rsyserr(FLOG, errno, "failed to open lock file %s",
388 lp_lock_file(i));
389 io_printf(f_out, "@ERROR: failed to open lock file\n");
390 } else {
391 rprintf(FLOG, "max connections (%d) reached\n",
392 lp_max_connections(i));
393 io_printf(f_out, "@ERROR: max connections (%d) reached -- try again later\n",
394 lp_max_connections(i));
395 }
396 return -1;
397 }
398
399 auth_user = auth_server(f_in, f_out, i, host, addr, "@RSYNCD: AUTHREQD ");
400
401 if (!auth_user) {
402 io_printf(f_out, "@ERROR: auth failed on module %s\n", name);
403 return -1;
404 }
405
406 module_id = i;
407
408 if (lp_read_only(i))
409 read_only = 1;
410
411 if (lp_transfer_logging(i) && !logfile_format)
412 logfile_format = lp_log_format(i);
413 if (log_format_has(logfile_format, 'i'))
414 logfile_format_has_i = 1;
415 if (logfile_format_has_i || log_format_has(logfile_format, 'o'))
416 logfile_format_has_o_or_i = 1;
417
418 am_root = (MY_UID() == 0);
419
420 if (am_root) {
421 p = lp_uid(i);
422 if (!name_to_uid(p, &uid)) {
423 if (!isDigit(p)) {
424 rprintf(FLOG, "Invalid uid %s\n", p);
425 io_printf(f_out, "@ERROR: invalid uid %s\n", p);
426 return -1;
427 }
428 uid = atoi(p);
429 }
430
431 p = lp_gid(i);
432 if (!name_to_gid(p, &gid)) {
433 if (!isDigit(p)) {
434 rprintf(FLOG, "Invalid gid %s\n", p);
435 io_printf(f_out, "@ERROR: invalid gid %s\n", p);
436 return -1;
437 }
438 gid = atoi(p);
439 }
440 }
441
442 /* TODO: If we're not root, but the configuration requests
443 * that we change to some uid other than the current one, then
444 * log a warning. */
445
446 /* TODO: Perhaps take a list of gids, and make them into the
447 * supplementary groups. */
448
449 /* We do a push_dir() without actually calling chdir() in order
450 * to make sure that the module's path is absolute. After this
451 * check, module_dir will be set to an absolute path. */
452 module_dir = lp_path(i);
453 strlcpy(line, curr_dir, sizeof line);
454 if (!push_dir(module_dir, 1))
455 goto chdir_failed;
456 if (strcmp(curr_dir, module_dir) != 0)
457 module_dir = strdup(curr_dir);
458 push_dir(line, 1); /* Restore curr_dir. */
459
460 if (use_chroot || (module_dirlen = strlen(module_dir)) == 1) {
461 module_dirlen = 0;
462 set_filter_dir("/", 1);
463 } else
464 set_filter_dir(module_dir, module_dirlen);
465
466 p = lp_filter(i);
467 parse_rule(&server_filter_list, p, MATCHFLG_WORD_SPLIT,
468 XFLG_ABS_IF_SLASH);
469
470 p = lp_include_from(i);
471 parse_filter_file(&server_filter_list, p, MATCHFLG_INCLUDE,
472 XFLG_ABS_IF_SLASH | XFLG_OLD_PREFIXES | XFLG_FATAL_ERRORS);
473
474 p = lp_include(i);
475 parse_rule(&server_filter_list, p,
476 MATCHFLG_INCLUDE | MATCHFLG_WORD_SPLIT,
477 XFLG_ABS_IF_SLASH | XFLG_OLD_PREFIXES);
478
479 p = lp_exclude_from(i);
480 parse_filter_file(&server_filter_list, p, 0,
481 XFLG_ABS_IF_SLASH | XFLG_OLD_PREFIXES | XFLG_FATAL_ERRORS);
482
483 p = lp_exclude(i);
484 parse_rule(&server_filter_list, p, MATCHFLG_WORD_SPLIT,
485 XFLG_ABS_IF_SLASH | XFLG_OLD_PREFIXES);
486
487 log_init(1);
488
489#ifdef HAVE_PUTENV
490 if (*lp_prexfer_exec(i) || *lp_postxfer_exec(i)) {
491 char *modname, *modpath, *hostaddr, *hostname, *username;
492 int status;
493 if (asprintf(&modname, "RSYNC_MODULE_NAME=%s", name) < 0
494 || asprintf(&modpath, "RSYNC_MODULE_PATH=%s", module_dir) < 0
495 || asprintf(&hostaddr, "RSYNC_HOST_ADDR=%s", addr) < 0
496 || asprintf(&hostname, "RSYNC_HOST_NAME=%s", host) < 0
497 || asprintf(&username, "RSYNC_USER_NAME=%s", auth_user) < 0)
498 out_of_memory("rsync_module");
499 putenv(modname);
500 putenv(modpath);
501 putenv(hostaddr);
502 putenv(hostname);
503 putenv(username);
504 umask(orig_umask);
505 /* For post-xfer exec, fork a new process to run the rsync
506 * daemon while this process waits for the exit status and
507 * runs the indicated command at that point. */
508 if (*lp_postxfer_exec(i)) {
509 pid_t pid = fork();
510 if (pid < 0) {
511 rsyserr(FLOG, errno, "fork failed");
512 io_printf(f_out, "@ERROR: fork failed\n");
513 return -1;
514 }
515 if (pid) {
516 if (asprintf(&p, "RSYNC_PID=%ld", (long)pid) > 0)
517 putenv(p);
518 if (wait_process(pid, &status, 0) < 0)
519 status = -1;
520 if (asprintf(&p, "RSYNC_RAW_STATUS=%d", status) > 0)
521 putenv(p);
522 if (WIFEXITED(status))
523 status = WEXITSTATUS(status);
524 else
525 status = -1;
526 if (asprintf(&p, "RSYNC_EXIT_STATUS=%d", status) > 0)
527 putenv(p);
528 system(lp_postxfer_exec(i));
529 _exit(status);
530 }
531 }
532 /* For pre-xfer exec, fork a child process to run the indicated
533 * command, though it first waits for the parent process to
534 * send us the user's request via a pipe. */
535 if (*lp_prexfer_exec(i)) {
536 int fds[2];
537 if (asprintf(&p, "RSYNC_PID=%ld", (long)getpid()) > 0)
538 putenv(p);
539 if (pipe(fds) < 0 || (pre_exec_pid = fork()) < 0) {
540 rsyserr(FLOG, errno, "pre-xfer exec preparation failed");
541 io_printf(f_out, "@ERROR: pre-xfer exec preparation failed\n");
542 return -1;
543 }
544 if (pre_exec_pid == 0) {
545 char buf[BIGPATHBUFLEN];
546 int j, len;
547 close(fds[1]);
548 set_blocking(fds[0]);
549 len = read_arg_from_pipe(fds[0], buf, BIGPATHBUFLEN);
550 if (len <= 0)
551 _exit(1);
552 if (asprintf(&p, "RSYNC_REQUEST=%s", buf) > 0)
553 putenv(p);
554 for (j = 0; ; j++) {
555 len = read_arg_from_pipe(fds[0], buf,
556 BIGPATHBUFLEN);
557 if (len <= 0) {
558 if (!len)
559 break;
560 _exit(1);
561 }
562 if (asprintf(&p, "RSYNC_ARG%d=%s", j, buf) > 0)
563 putenv(p);
564 }
565 close(fds[0]);
566 close(STDIN_FILENO);
567 close(STDOUT_FILENO);
568 status = system(lp_prexfer_exec(i));
569 if (!WIFEXITED(status))
570 _exit(1);
571 _exit(WEXITSTATUS(status));
572 }
573 close(fds[0]);
574 set_blocking(fds[1]);
575 pre_exec_fd = fds[1];
576 }
577 umask(0);
578 }
579#endif
580
581 if (use_chroot) {
582 /*
583 * XXX: The 'use chroot' flag is a fairly reliable
584 * source of confusion, because it fails under two
585 * important circumstances: running as non-root,
586 * running on Win32 (or possibly others). On the
587 * other hand, if you are running as root, then it
588 * might be better to always use chroot.
589 *
590 * So, perhaps if we can't chroot we should just issue
591 * a warning, unless a "require chroot" flag is set,
592 * in which case we fail.
593 */
594 if (chroot(module_dir)) {
595 rsyserr(FLOG, errno, "chroot %s failed", module_dir);
596 io_printf(f_out, "@ERROR: chroot failed\n");
597 return -1;
598 }
599 if (!push_dir("/", 0))
600 goto chdir_failed;
601 } else {
602 if (!push_dir(module_dir, 0)) {
603 chdir_failed:
604 rsyserr(FLOG, errno, "chdir %s failed\n", module_dir);
605 io_printf(f_out, "@ERROR: chdir failed\n");
606 return -1;
607 }
608 sanitize_paths = 1;
609 }
610
611 if (am_root) {
612 /* XXXX: You could argue that if the daemon is started
613 * by a non-root user and they explicitly specify a
614 * gid, then we should try to change to that gid --
615 * this could be possible if it's already in their
616 * supplementary groups. */
617
618 /* TODO: Perhaps we need to document that if rsyncd is
619 * started by somebody other than root it will inherit
620 * all their supplementary groups. */
621
622 if (setgid(gid)) {
623 rsyserr(FLOG, errno, "setgid %d failed", (int)gid);
624 io_printf(f_out, "@ERROR: setgid failed\n");
625 return -1;
626 }
627#ifdef HAVE_SETGROUPS
628 /* Get rid of any supplementary groups this process
629 * might have inheristed. */
630 if (setgroups(1, &gid)) {
631 rsyserr(FLOG, errno, "setgroups failed");
632 io_printf(f_out, "@ERROR: setgroups failed\n");
633 return -1;
634 }
635#endif
636
637 if (setuid(uid)) {
638 rsyserr(FLOG, errno, "setuid %d failed", (int)uid);
639 io_printf(f_out, "@ERROR: setuid failed\n");
640 return -1;
641 }
642
643 am_root = (MY_UID() == 0);
644 }
645
646 if (lp_temp_dir(i) && *lp_temp_dir(i)) {
647 tmpdir = lp_temp_dir(i);
648 if (strlen(tmpdir) >= MAXPATHLEN - 10) {
649 rprintf(FLOG,
650 "the 'temp dir' value for %s is WAY too long -- ignoring.\n",
651 name);
652 tmpdir = NULL;
653 }
654 }
655
656 io_printf(f_out, "@RSYNCD: OK\n");
657
658 opt_cnt = read_args(f_in, name, line, sizeof line, rl_nulls, &argv, &argc, &request);
659
660 if (pre_exec_pid) {
661 err_msg = finish_pre_exec(pre_exec_pid, pre_exec_fd, request,
662 opt_cnt, argv);
663 }
664
665 verbose = 0; /* future verbosity is controlled by client options */
666 ret = parse_arguments(&argc, (const char ***) &argv, 0);
667 am_server = 1; /* Don't let someone try to be tricky. */
668 if (lp_ignore_errors(module_id))
669 ignore_errors = 1;
670
671 if (lp_fake_super(i))
672 am_root = -1;
673 else if (am_root < 0) /* Treat --fake-super from client as --super. */
674 am_root = 2;
675
676 if (filesfrom_fd == 0)
677 filesfrom_fd = f_in;
678
679 if (request) {
680 if (*auth_user) {
681 rprintf(FLOG, "rsync %s %s from %s@%s (%s)\n",
682 am_sender ? "on" : "to",
683 request, auth_user, host, addr);
684 } else {
685 rprintf(FLOG, "rsync %s %s from %s (%s)\n",
686 am_sender ? "on" : "to",
687 request, host, addr);
688 }
689 free(request);
690 }
691
692#ifndef DEBUG
693 /* don't allow the logs to be flooded too fast */
694 if (verbose > lp_max_verbosity(i))
695 verbose = lp_max_verbosity(i);
696#endif
697
698 if (protocol_version < 23
699 && (protocol_version == 22 || am_sender))
700 io_start_multiplex_out();
701 else if (!ret || err_msg) {
702 /* We have to get I/O multiplexing started so that we can
703 * get the error back to the client. This means getting
704 * the protocol setup finished first in later versions. */
705 setup_protocol(f_out, f_in);
706 if (!am_sender) {
707 /* Since we failed in our option parsing, we may not
708 * have finished parsing that the client sent us a
709 * --files-from option, so look for it manually.
710 * Without this, the socket would be in the wrong
711 * state for the upcoming error message. */
712 if (!files_from) {
713 int i;
714 for (i = 0; i < argc; i++) {
715 if (strncmp(argv[i], "--files-from", 12) == 0) {
716 files_from = "";
717 break;
718 }
719 }
720 }
721 if (files_from)
722 write_byte(f_out, 0);
723 }
724 io_start_multiplex_out();
725 }
726
727 if (!ret || err_msg) {
728 if (err_msg)
729 rwrite(FERROR, err_msg, strlen(err_msg), 0);
730 else
731 option_error();
732 msleep(400);
733 exit_cleanup(RERR_UNSUPPORTED);
734 }
735
736 if (lp_timeout(i) && lp_timeout(i) > io_timeout)
737 set_io_timeout(lp_timeout(i));
738
739 /* If we have some incoming/outgoing chmod changes, append them to
740 * any user-specified changes (making our changes have priority).
741 * We also get a pointer to just our changes so that a receiver
742 * process can use them separately if --perms wasn't specified. */
743 if (am_sender)
744 p = lp_outgoing_chmod(i);
745 else
746 p = lp_incoming_chmod(i);
747 if (*p && !(daemon_chmod_modes = parse_chmod(p, &chmod_modes))) {
748 rprintf(FLOG, "Invalid \"%sing chmod\" directive: %s\n",
749 am_sender ? "outgo" : "incom", p);
750 }
751
752 start_server(f_in, f_out, argc, argv);
753
754 return 0;
755}
756
757/* send a list of available modules to the client. Don't list those
758 with "list = False". */
759static void send_listing(int fd)
760{
761 int n = lp_numservices();
762 int i;
763
764 for (i = 0; i < n; i++) {
765 if (lp_list(i))
766 io_printf(fd, "%-15s\t%s\n", lp_name(i), lp_comment(i));
767 }
768
769 if (protocol_version >= 25)
770 io_printf(fd,"@RSYNCD: EXIT\n");
771}
772
773/* this is called when a connection is established to a client
774 and we want to start talking. The setup of the system is done from
775 here */
776int start_daemon(int f_in, int f_out)
777{
778 char line[1024];
779 char *addr, *host;
780 int i;
781
782 io_set_sock_fds(f_in, f_out);
783
784 /* We must load the config file before calling any function that
785 * might cause log-file output to occur. This ensures that the
786 * "log file" param gets honored for the 2 non-forked use-cases
787 * (when rsync is run by init and run by a remote shell). */
788 if (!lp_load(config_file, 0))
789 exit_cleanup(RERR_SYNTAX);
790
791 addr = client_addr(f_in);
792 host = client_name(f_in);
793 rprintf(FLOG, "connect from %s (%s)\n", host, addr);
794
795 if (!am_server) {
796 set_socket_options(f_in, "SO_KEEPALIVE");
797 if (sockopts)
798 set_socket_options(f_in, sockopts);
799 else
800 set_socket_options(f_in, lp_socket_options());
801 set_nonblocking(f_in);
802 }
803
804 if (exchange_protocols(f_in, f_out, line, sizeof line, 0) < 0)
805 return -1;
806
807 line[0] = 0;
808 if (!read_line_old(f_in, line, sizeof line))
809 return -1;
810
811 if (!*line || strcmp(line, "#list") == 0) {
812 rprintf(FLOG, "module-list request from %s (%s)\n",
813 host, addr);
814 send_listing(f_out);
815 return -1;
816 }
817
818 if (*line == '#') {
819 /* it's some sort of command that I don't understand */
820 io_printf(f_out, "@ERROR: Unknown command '%s'\n", line);
821 return -1;
822 }
823
824 if ((i = lp_number(line)) < 0) {
825 rprintf(FLOG, "unknown module '%s' tried from %s (%s)\n",
826 line, host, addr);
827 io_printf(f_out, "@ERROR: Unknown module '%s'\n", line);
828 return -1;
829 }
830
831#ifdef HAVE_SIGACTION
832 sigact.sa_flags = SA_NOCLDSTOP;
833#endif
834 SIGACTION(SIGCHLD, remember_children);
835
836 return rsync_module(f_in, f_out, i, addr, host);
837}
838
839static void create_pid_file(void)
840{
841 char *pid_file = lp_pid_file();
842 char pidbuf[16];
843 pid_t pid = getpid();
844 int fd;
845
846 if (!pid_file || !*pid_file)
847 return;
848
849 cleanup_set_pid(pid);
850 if ((fd = do_open(pid_file, O_WRONLY|O_CREAT|O_EXCL, 0666 & ~orig_umask)) == -1) {
851 cleanup_set_pid(0);
852 fprintf(stderr, "failed to create pid file %s: %s\n", pid_file, strerror(errno));
853 rsyserr(FLOG, errno, "failed to create pid file %s", pid_file);
854 exit_cleanup(RERR_FILEIO);
855 }
856 snprintf(pidbuf, sizeof pidbuf, "%ld\n", (long)pid);
857 write(fd, pidbuf, strlen(pidbuf));
858 close(fd);
859}
860
861/* Become a daemon, discarding the controlling terminal. */
862static void become_daemon(void)
863{
864 int i;
865 pid_t pid = fork();
866
867 if (pid) {
868 if (pid < 0) {
869 fprintf(stderr, "failed to fork: %s\n", strerror(errno));
870 exit_cleanup(RERR_FILEIO);
871 }
872 _exit(0);
873 }
874
875 create_pid_file();
876
877 /* detach from the terminal */
878#ifdef HAVE_SETSID
879 setsid();
880#elif defined TIOCNOTTY
881 i = open("/dev/tty", O_RDWR);
882 if (i >= 0) {
883 ioctl(i, (int)TIOCNOTTY, (char *)0);
884 close(i);
885 }
886#endif
887 /* make sure that stdin, stdout an stderr don't stuff things
888 * up (library functions, for example) */
889 for (i = 0; i < 3; i++) {
890 close(i);
891 open("/dev/null", O_RDWR);
892 }
893}
894
895int daemon_main(void)
896{
897 if (!config_file) {
898 if (am_server && am_root <= 0)
899 config_file = RSYNCD_USERCONF;
900 else
901 config_file = RSYNCD_SYSCONF;
902 }
903
904 if (is_a_socket(STDIN_FILENO)) {
905 int i;
906
907 /* we are running via inetd - close off stdout and
908 * stderr so that library functions (and getopt) don't
909 * try to use them. Redirect them to /dev/null */
910 for (i = 1; i < 3; i++) {
911 close(i);
912 open("/dev/null", O_RDWR);
913 }
914
915 return start_daemon(STDIN_FILENO, STDIN_FILENO);
916 }
917
918 if (!lp_load(config_file, 1)) {
919 fprintf(stderr, "Failed to parse config file: %s\n", config_file);
920 exit_cleanup(RERR_SYNTAX);
921 }
922
923 if (no_detach)
924 create_pid_file();
925 else
926 become_daemon();
927
928 if (rsync_port == 0 && (rsync_port = lp_rsync_port()) == 0)
929 rsync_port = RSYNC_PORT;
930 if (bind_address == NULL && *lp_bind_address())
931 bind_address = lp_bind_address();
932
933 log_init(0);
934
935 rprintf(FLOG, "rsyncd version %s starting, listening on port %d\n",
936 RSYNC_VERSION, rsync_port);
937 /* TODO: If listening on a particular address, then show that
938 * address too. In fact, why not just do inet_ntop on the
939 * local address??? */
940
941 start_accept_loop(rsync_port, start_daemon);
942 return -1;
943}