X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/5805327b2852e2e65bd409f1d322353128690be5..25ff04417e4b4fa0aeb44e0b1576a01021aa1650:/clientserver.c diff --git a/clientserver.c b/clientserver.c index 244bbc68..4825b00a 100644 --- a/clientserver.c +++ b/clientserver.c @@ -1,241 +1,508 @@ -/* - Copyright (C) Andrew Tridgell 1998 - - This program is free software; you can redistribute it and/or modify - it under the terms of the GNU General Public License as published by - the Free Software Foundation; either version 2 of the License, or - (at your option) any later version. - - This program is distributed in the hope that it will be useful, - but WITHOUT ANY WARRANTY; without even the implied warranty of - MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - GNU General Public License for more details. - - You should have received a copy of the GNU General Public License - along with this program; if not, write to the Free Software - Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. -*/ - -/* the socket based protocol for setting up a connection wit rsyncd */ +/* -*- c-file-style: "linux"; -*- + * + * Copyright (C) 1998-2001 by Andrew Tridgell + * Copyright (C) 2001-2002 by Martin Pool + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + */ + +/** + * @file + * + * The socket based protocol for setting up a connection with + * rsyncd. + **/ #include "rsync.h" +extern int am_sender; +extern int am_server; +extern int am_daemon; +extern int am_root; extern int module_id; -extern int read_only; extern int verbose; extern int rsync_port; - +extern int kludge_around_eof; +extern int daemon_over_rsh; +extern int list_only; +extern int sanitize_paths; +extern int filesfrom_fd; +extern int remote_protocol; +extern int protocol_version; +extern int io_timeout; +extern int select_timeout; +extern int orig_umask; +extern int no_detach; +extern int default_af_hint; +extern char *bind_address; +extern struct filter_list_struct server_filter_list; +extern char *config_file; +extern char *files_from; + +char *auth_user; +int read_only = 0; + +/* Length of lp_path() string when in daemon mode & not chrooted, else 0. */ +unsigned int module_dirlen = 0; + +/** + * Run a client connected to an rsyncd. The alternative to this + * function for remote-shell connections is do_cmd(). + * + * After negotiating which module to use and reading the server's + * motd, this hands over to client_run(). Telling the server the + * module will cause it to chroot/setuid/etc. + * + * Instead of doing a transfer, the client may at this stage instead + * get a listing of remote modules and exit. + * + * @return -1 for error in startup, or the result of client_run(). + * Either way, it eventually gets passed to exit_cleanup(). + **/ int start_socket_client(char *host, char *path, int argc, char *argv[]) { - int fd, i; - char *sargs[MAX_ARGS]; - int sargc=0; - char line[1024]; - char *p, *user=NULL; - extern int remote_version; + int fd, ret; + char *p, *user = NULL; + + /* This is redundant with code in start_inband_exchange(), but this + * short-circuits a problem in the client before we open a socket, + * and the extra check won't hurt. */ + if (*path == '/') { + rprintf(FERROR, + "ERROR: The remote path must start with a module name not a /\n"); + return -1; + } - p = strchr(host, '@'); - if (p) { + if ((p = strchr(host, '@')) != NULL) { user = host; host = p+1; - *p = 0; + *p = '\0'; } - if (!user) user = getenv("USER"); - if (!user) user = getenv("LOGNAME"); + if (rsync_port == 0) + rsync_port = RSYNC_PORT; - fd = open_socket_out(host, rsync_port); - if (fd == -1) { - exit_cleanup(1); - } - - server_options(sargs,&sargc); + fd = open_socket_out_wrapped(host, rsync_port, bind_address, + default_af_hint); + if (fd == -1) + exit_cleanup(RERR_SOCKETIO); - sargs[sargc++] = "."; + ret = start_inband_exchange(user, path, fd, fd, argc); - if (path && *path) - sargs[sargc++] = path; + return ret < 0? ret : client_run(fd, fd, -1, argc, argv); +} - sargs[sargc] = NULL; +int start_inband_exchange(char *user, char *path, int f_in, int f_out, + int argc) +{ + int i; + char *sargs[MAX_ARGS]; + int sargc = 0; + char line[MAXPATHLEN]; + char *p; - io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION); + if (argc == 0 && !am_sender) + list_only |= 1; - if (!read_line(fd, line, sizeof(line)-1)) { + if (*path == '/') { + rprintf(FERROR, + "ERROR: The remote path must start with a module name\n"); return -1; } - if (sscanf(line,"@RSYNCD: %d", &remote_version) != 1) { + if (!user) + user = getenv("USER"); + if (!user) + user = getenv("LOGNAME"); + + io_printf(f_out, "@RSYNCD: %d\n", protocol_version); + + if (!read_line(f_in, line, sizeof line - 1)) { + rprintf(FERROR, "rsync: did not see server greeting\n"); return -1; } + if (sscanf(line,"@RSYNCD: %d", &remote_protocol) != 1) { + /* note that read_line strips of \n or \r */ + rprintf(FERROR, "rsync: server sent \"%s\" rather than greeting\n", + line); + return -1; + } + if (protocol_version > remote_protocol) + protocol_version = remote_protocol; + + if (list_only && protocol_version >= 29) + list_only |= 2; + + /* set daemon_over_rsh to false since we need to build the + * true set of args passed through the rsh/ssh connection; + * this is a no-op for direct-socket-connection mode */ + daemon_over_rsh = 0; + server_options(sargs, &sargc); + + sargs[sargc++] = "."; + + if (path && *path) + sargs[sargc++] = path; + + sargs[sargc] = NULL; + + if (verbose > 1) + print_child_argv(sargs); + p = strchr(path,'/'); if (p) *p = 0; - io_printf(fd,"%s\n",path); + io_printf(f_out, "%s\n", path); if (p) *p = '/'; + /* Old servers may just drop the connection here, + rather than sending a proper EXIT command. Yuck. */ + kludge_around_eof = list_only && (protocol_version < 25); + while (1) { - if (!read_line(fd, line, sizeof(line)-1)) { + if (!read_line(f_in, line, sizeof line - 1)) { + rprintf(FERROR, "rsync: didn't get server startup line\n"); return -1; } if (strncmp(line,"@RSYNCD: AUTHREQD ",18) == 0) { - auth_client(fd, user, line+18); + auth_client(f_out, user, line+18); continue; } - if (strcmp(line,"@RSYNCD: OK") == 0) break; - rprintf(FINFO,"%s\n", line); + if (strcmp(line,"@RSYNCD: OK") == 0) + break; + + if (strcmp(line,"@RSYNCD: EXIT") == 0) { + /* This is sent by recent versions of the + * server to terminate the listing of modules. + * We don't want to go on and transfer + * anything; just exit. */ + exit(0); + } + + if (strncmp(line, "@ERROR", 6) == 0) { + rprintf(FERROR, "%s\n", line); + /* This is always fatal; the server will now + * close the socket. */ + return RERR_STARTCLIENT; + } else { + rprintf(FINFO,"%s\n", line); + } + } + kludge_around_eof = False; + + for (i = 0; i < sargc; i++) { + io_printf(f_out, "%s\n", sargs[i]); } + io_printf(f_out, "\n"); - for (i=0;i 1) verbose = 1; + if (verbose > lp_max_verbosity()) + verbose = lp_max_verbosity(); +#endif + + if (protocol_version < 23 + && (protocol_version == 22 || am_sender)) + io_start_multiplex_out(); + else if (!ret) { + /* We have to get I/O multiplexing started so that we can + * get the error back to the client. This means getting + * the protocol setup finished first in later versions. */ + setup_protocol(f_out, f_in); + if (files_from && !am_sender && strcmp(files_from, "-") != 0) + write_byte(f_out, 0); + io_start_multiplex_out(); + } + + if (!ret) { + option_error(); + msleep(400); + exit_cleanup(RERR_UNSUPPORTED); + } - argc -= optind; - argp = argv + optind; - optind = 0; + if (lp_timeout(i)) { + io_timeout = lp_timeout(i); + if (io_timeout < select_timeout) + select_timeout = io_timeout; + } - start_server(fd, fd, argc, argp); + start_server(f_in, f_out, argc, argp); return 0; } @@ -246,92 +513,146 @@ static void send_listing(int fd) { int n = lp_numservices(); int i; - - for (i=0;i= 25) + io_printf(fd,"@RSYNCD: EXIT\n"); } -/* this is called when a socket connection is established to a client +/* this is called when a connection is established to a client and we want to start talking. The setup of the system is done from here */ -static int start_daemon(int fd) +int start_daemon(int f_in, int f_out) { char line[200]; char *motd; - int i = -1; - extern char *config_file; - extern int remote_version; + int i; - if (!lp_load(config_file)) { - exit_cleanup(1); - } + io_set_sock_fds(f_in, f_out); + + if (!lp_load(config_file, 0)) + exit_cleanup(RERR_SYNTAX); - set_socket_options(fd,"SO_KEEPALIVE"); + log_init(); - io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION); + if (!am_server) { + set_socket_options(f_in, "SO_KEEPALIVE"); + set_socket_options(f_in, lp_socket_options()); + set_nonblocking(f_in); + } + + io_printf(f_out, "@RSYNCD: %d\n", protocol_version); motd = lp_motd_file(); - if (*motd) { + if (motd && *motd) { FILE *f = fopen(motd,"r"); while (f && !feof(f)) { - int len = fread(line, 1, sizeof(line)-1, f); + int len = fread(line, 1, sizeof line - 1, f); if (len > 0) { line[len] = 0; - io_printf(fd,"%s", line); + io_printf(f_out, "%s", line); } } - if (f) fclose(f); - io_printf(fd,"\n"); + if (f) + fclose(f); + io_printf(f_out, "\n"); } - if (!read_line(fd, line, sizeof(line)-1)) { + if (!read_line(f_in, line, sizeof line - 1)) return -1; - } - if (sscanf(line,"@RSYNCD: %d", &remote_version) != 1) { - io_printf(fd,"@ERROR: protocol startup error\n"); + if (sscanf(line,"@RSYNCD: %d", &remote_protocol) != 1) { + io_printf(f_out, "@ERROR: protocol startup error\n"); return -1; - } + } + if (protocol_version > remote_protocol) + protocol_version = remote_protocol; - while (i == -1) { - line[0] = 0; - if (!read_line(fd, line, sizeof(line)-1)) { - return -1; - } + line[0] = 0; + if (!read_line(f_in, line, sizeof line - 1)) + return -1; - if (!*line || strcmp(line,"#list")==0) { - send_listing(fd); - return -1; - } + if (!*line || strcmp(line, "#list") == 0) { + send_listing(f_out); + return -1; + } - if (*line == '#') { - /* it's some sort of command that I don't understand */ - io_printf(fd,"@ERROR: Unknown command '%s'\n", line); - return -1; - } + if (*line == '#') { + /* it's some sort of command that I don't understand */ + io_printf(f_out, "@ERROR: Unknown command '%s'\n", line); + return -1; + } - i = lp_number(line); - if (i == -1) { - io_printf(fd,"@ERROR: Unknown module '%s'\n", line); - return -1; - } + if ((i = lp_number(line)) < 0) { + char *addr = client_addr(f_in); + char *host = client_name(f_in); + rprintf(FLOG, "unknown module '%s' tried from %s (%s)\n", + line, host, addr); + io_printf(f_out, "@ERROR: Unknown module '%s'\n", line); + return -1; } - return rsync_module(fd, i); + return rsync_module(f_in, f_out, i); } int daemon_main(void) { + char *pid_file; + if (is_a_socket(STDIN_FILENO)) { - /* we are running via inetd */ - return start_daemon(STDIN_FILENO); + int i; + + /* we are running via inetd - close off stdout and + * stderr so that library functions (and getopt) don't + * try to use them. Redirect them to /dev/null */ + for (i = 1; i < 3; i++) { + close(i); + open("/dev/null", O_RDWR); + } + + return start_daemon(STDIN_FILENO, STDIN_FILENO); } - become_daemon(); + if (!no_detach) + become_daemon(); + + if (!lp_load(config_file, 1)) + exit_cleanup(RERR_SYNTAX); + + if (rsync_port == 0 && (rsync_port = lp_rsync_port()) == 0) + rsync_port = RSYNC_PORT; + if (bind_address == NULL && *lp_bind_address()) + bind_address = lp_bind_address(); + + log_init(); + + rprintf(FLOG, "rsyncd version %s starting, listening on port %d\n", + RSYNC_VERSION, rsync_port); + /* TODO: If listening on a particular address, then show that + * address too. In fact, why not just do inet_ntop on the + * local address??? */ + + if (((pid_file = lp_pid_file()) != NULL) && (*pid_file != '\0')) { + char pidbuf[16]; + int fd; + pid_t pid = getpid(); + cleanup_set_pid(pid); + if ((fd = do_open(lp_pid_file(), O_WRONLY|O_CREAT|O_TRUNC, + 0666 & ~orig_umask)) == -1) { + cleanup_set_pid(0); + rsyserr(FLOG, errno, "failed to create pid file %s", pid_file); + exit_cleanup(RERR_FILEIO); + } + snprintf(pidbuf, sizeof pidbuf, "%ld\n", (long)pid); + write(fd, pidbuf, strlen(pidbuf)); + close(fd); + } start_accept_loop(rsync_port, start_daemon); return -1; } -