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