X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/7b372642ec1179dece1b0b0162189112b85018bf..9352b0649b8d6aa21d9d567c1d819a0fa26c7b01:/clientserver.c diff --git a/clientserver.c b/clientserver.c index 765cba7e..468553c4 100644 --- a/clientserver.c +++ b/clientserver.c @@ -1,38 +1,84 @@ -/* - 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 orig_umask; +extern int no_detach; +extern int default_af_hint; +extern char *bind_address; +extern struct exclude_struct **server_exclude_list; +extern char *exclude_path_prefix; +extern char *config_file; + +char *auth_user; + +/** + * 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[MAXPATHLEN]; + int fd, ret; char *p, *user=NULL; - extern int remote_version; + + /* this is redundant with code in start_inband_exchange(), but + * this short-circuits a problem 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) { @@ -41,164 +87,311 @@ int start_socket_client(char *host, char *path, int argc, char *argv[]) *p = 0; } + if (verbose >= 2) { + /* FIXME: If we're going to use a socket program for + * testing, then this message is wrong. We need to + * say something like "(except really using %s)" */ + rprintf(FINFO, "opening tcp connection to %s port %d\n", + host, rsync_port); + } + fd = open_socket_out_wrapped(host, rsync_port, bind_address, + default_af_hint); + if (fd == -1) { + exit_cleanup(RERR_SOCKETIO); + } + + ret = start_inband_exchange(user, path, fd, fd, argc); + + return ret < 0? ret : client_run(fd, fd, -1, argc, argv); +} + +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; + + if (argc == 0 && !am_sender) + list_only = 1; + + if (*path == '/') { + rprintf(FERROR, "ERROR: The remote path must start with a module name\n"); + return -1; + } + if (!user) user = getenv("USER"); if (!user) user = getenv("LOGNAME"); - fd = open_socket_out(host, rsync_port); - if (fd == -1) { - exit_cleanup(1); - } - - server_options(sargs,&sargc); + /* 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) + if (path && *path) sargs[sargc++] = path; sargs[sargc] = NULL; - io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION); + io_printf(f_out, "@RSYNCD: %d\n", protocol_version); - if (!read_line(fd, line, sizeof(line)-1)) { + 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_version) != 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; 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: 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 1) verbose = 1; + if (verbose > lp_max_verbosity()) + verbose = lp_max_verbosity(); +#endif + + if (protocol_version < 23) { + if (protocol_version == 22 || am_sender) + io_start_multiplex_out(f_out); + } - argc -= optind; - argp = argv + optind; - optind = 0; + /* For later protocol versions, we don't start multiplexing + * until we've configured nonblocking in start_server. That + * means we're in a sticky situation now: there's no way to + * convey errors to the client. */ + + /* FIXME: Hold off on reporting option processing errors until + * we've set up nonblocking and multiplexed IO and can get the + * message back to them. */ + if (!ret) { + option_error(); + exit_cleanup(RERR_UNSUPPORTED); + } + + if (lp_timeout(i)) { + io_timeout = lp_timeout(i); + } - start_server(fd, fd, argc, argp); + start_server(f_in, f_out, argc, argp); return 0; } @@ -257,105 +494,140 @@ 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; if (!lp_load(config_file, 0)) { - exit_cleanup(1); + exit_cleanup(RERR_SYNTAX); } - set_socket_options(fd,"SO_KEEPALIVE"); - set_socket_options(fd,lp_socket_options()); - + log_init(); + + if (!am_server) { + set_socket_options(f_in, "SO_KEEPALIVE"); + set_socket_options(f_in, lp_socket_options()); + set_nonblocking(f_in); + } - io_printf(fd,"@RSYNCD: %d\n", PROTOCOL_VERSION); + 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); 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"); + 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)) { + if (!read_line(f_in, line, sizeof(line)-1)) { return -1; } if (!*line || strcmp(line,"#list")==0) { - send_listing(fd); + 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); + 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); + 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) { - extern char *config_file; + 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)) { - fprintf(stderr,"failed to load config file %s\n", config_file); - exit_cleanup(1); + exit_cleanup(RERR_SYNTAX); } - log_open(); - - rprintf(FINFO,"rsyncd version %s starting\n",VERSION); + log_init(); + + rprintf(FINFO, "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; } -