X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/dc7a947853cf5f8edf5afb57babc634451b60e6b..f0fca04e4e136c4a487a922e8fb09acf46aeafa0:/main.c diff --git a/main.c b/main.c index 837d3097..0bcc7f54 100644 --- a/main.c +++ b/main.c @@ -21,11 +21,12 @@ int verbose = 0; int always_checksum = 0; -time_t starttime; -off_t total_size = 0; +time_t starttime = 0; +int64 total_size = 0; int block_size=BLOCK_SIZE; char *backup_suffix = BACKUP_SUFFIX; +char *tmpdir = NULL; static char *rsync_path = RSYNC_NAME; @@ -52,44 +53,55 @@ int do_compression=0; int am_root=0; int orig_umask=0; int relative_paths=0; +int numeric_ids = 0; +int force_delete = 0; +int io_timeout = 0; +int io_error = 0; +int read_only = 0; +static int module_id; + +static int port = RSYNC_PORT; + +static char *shell_cmd; extern int csum_length; int am_server = 0; -static int sender = 0; +int am_sender=0; int recurse = 0; +int am_daemon=0; -static void usage(FILE *f); +static void usage(int fd); static void report(int f) { - int in,out,tsize; + int64 in,out,tsize; time_t t = time(NULL); if (!verbose) return; - if (am_server && sender) { - write_int(f,read_total()); - write_int(f,write_total()); - write_int(f,total_size); + if (am_server && am_sender) { + write_longint(f,read_total()); + write_longint(f,write_total()); + write_longint(f,total_size); write_flush(f); return; } - if (sender) { + if (am_sender) { in = read_total(); out = write_total(); - tsize = (int)total_size; + tsize = total_size; } else { - in = read_int(f); - out = read_int(f); - tsize = read_int(f); + in = read_longint(f); + out = read_longint(f); + tsize = read_longint(f); } - printf("wrote %d bytes read %d bytes %g bytes/sec\n", - out,in,(in+out)/(0.5 + (t-starttime))); - printf("total size is %d speedup is %g\n", - tsize,(1.0*tsize)/(in+out)); + printf("wrote %.0f bytes read %.0f bytes %.2f bytes/sec\n", + (double)out,(double)in,(in+out)/(0.5 + (t-starttime))); + printf("total size is %.0f speedup is %.2f\n", + (double)tsize,(1.0*tsize)/(in+out)); } @@ -98,11 +110,12 @@ static void server_options(char **args,int *argc) int ac = *argc; static char argstr[50]; static char bsize[30]; + static char iotime[30]; int i, x; args[ac++] = "--server"; - if (!sender) + if (!am_sender) args[ac++] = "--sender"; x = 1; @@ -158,6 +171,11 @@ static void server_options(char **args,int *argc) args[ac++] = bsize; } + if (io_timeout) { + sprintf(iotime,"--timeout=%d",io_timeout); + args[ac++] = iotime; + } + if (strcmp(backup_suffix, BACKUP_SUFFIX)) { args[ac++] = "--suffix"; args[ac++] = backup_suffix; @@ -166,6 +184,17 @@ static void server_options(char **args,int *argc) if (delete_mode) args[ac++] = "--delete"; + if (force_delete) + args[ac++] = "--force"; + + if (numeric_ids) + args[ac++] = "--numeric-ids"; + + if (tmpdir) { + args[ac++] = "--temp-dir"; + args[ac++] = tmpdir; + } + *argc = ac; } @@ -173,78 +202,70 @@ static void server_options(char **args,int *argc) static int do_cmd(char *cmd,char *machine,char *user,char *path,int *f_in,int *f_out) { - char *args[100]; - int i,argc=0, ret; - char *tok,*p,*dir=NULL; - - if (!local_server) { - if (!cmd) - cmd = getenv(RSYNC_RSH_ENV); - if (!cmd) - cmd = RSYNC_RSH; - cmd = strdup(cmd); - if (!cmd) - goto oom; - - for (tok=strtok(cmd," ");tok;tok=strtok(NULL," ")) { - args[argc++] = tok; - } + char *args[100]; + int i,argc=0, ret; + char *tok,*dir=NULL; + + if (!local_server) { + if (!cmd) + cmd = getenv(RSYNC_RSH_ENV); + if (!cmd) + cmd = RSYNC_RSH; + cmd = strdup(cmd); + if (!cmd) + goto oom; + + for (tok=strtok(cmd," ");tok;tok=strtok(NULL," ")) { + args[argc++] = tok; + } #if HAVE_REMSH - /* remsh (on HPUX) takes the arguments the other way around */ - args[argc++] = machine; - if (user) { - args[argc++] = "-l"; - args[argc++] = user; - } + /* remsh (on HPUX) takes the arguments the other way around */ + args[argc++] = machine; + if (user) { + args[argc++] = "-l"; + args[argc++] = user; + } #else - if (user) { - args[argc++] = "-l"; - args[argc++] = user; - } - args[argc++] = machine; + if (user) { + args[argc++] = "-l"; + args[argc++] = user; + } + args[argc++] = machine; #endif - } - args[argc++] = rsync_path; - - server_options(args,&argc); - - if (path && *path) { - dir = strdup(path); - p = strrchr(dir,'/'); - if (p && !relative_paths) { - *p = 0; - if (!dir[0]) - args[argc++] = "/"; - else - args[argc++] = dir; - p++; - } else { - args[argc++] = "."; - p = dir; - } - if (p[0]) - args[argc++] = path; - } + args[argc++] = rsync_path; - args[argc] = NULL; + server_options(args,&argc); + } - if (verbose > 3) { - fprintf(FERROR,"cmd="); - for (i=0;i 3) { + rprintf(FINFO,"cmd="); + for (i=0;icount > 1) { - fprintf(FERROR,"ERROR: destination must be a directory when copying more than 1 file\n"); + rprintf(FERROR,"ERROR: destination must be a directory when copying more than 1 file\n"); exit_cleanup(1); } return name; @@ -275,15 +296,15 @@ static char *get_local_name(struct file_list *flist,char *name) if (!name) return NULL; - if (mkdir(name,0777 & ~orig_umask) != 0) { - fprintf(FERROR,"mkdir %s : %s (1)\n",name,strerror(errno)); + if (do_mkdir(name,0777 & ~orig_umask) != 0) { + rprintf(FERROR,"mkdir %s : %s (1)\n",name,strerror(errno)); exit_cleanup(1); } else { - fprintf(FINFO,"created directory %s\n",name); + rprintf(FINFO,"created directory %s\n",name); } if (chdir(name) != 0) { - fprintf(FERROR,"chdir %s : %s (2)\n",name,strerror(errno)); + rprintf(FERROR,"chdir %s : %s (2)\n",name,strerror(errno)); exit_cleanup(1); } @@ -293,40 +314,52 @@ static char *get_local_name(struct file_list *flist,char *name) -void do_server_sender(int argc,char *argv[]) +static void do_server_sender(int f_in, int f_out, int argc,char *argv[]) { int i; - char *dir = argv[0]; struct file_list *flist; + char *dir = argv[0]; if (verbose > 2) - fprintf(FERROR,"server_sender starting pid=%d\n",(int)getpid()); + rprintf(FINFO,"server_sender starting pid=%d\n",(int)getpid()); if (!relative_paths && chdir(dir) != 0) { - fprintf(FERROR,"chdir %s: %s (3)\n",dir,strerror(errno)); - exit_cleanup(1); + rprintf(FERROR,"chdir %s: %s (3)\n",dir,strerror(errno)); + exit_cleanup(1); } argc--; argv++; if (strcmp(dir,".")) { - int l = strlen(dir); - if (strcmp(dir,"/") == 0) - l = 0; - for (i=0;i 2) - fprintf(FERROR,"receiver read %d\n",read_total()); + rprintf(FINFO,"receiver read %ld\n",(long)read_total()); exit_cleanup(0); } @@ -361,101 +394,322 @@ static int do_recv(int f_in,int f_out,struct file_list *flist,char *local_name) } -void do_server_recv(int argc,char *argv[]) +static void do_server_recv(int f_in, int f_out, int argc,char *argv[]) { int status; - char *dir = NULL; struct file_list *flist; char *local_name=NULL; + char *dir = NULL; if (verbose > 2) - fprintf(FERROR,"server_recv(%d) starting pid=%d\n",argc,(int)getpid()); + rprintf(FINFO,"server_recv(%d) starting pid=%d\n",argc,(int)getpid()); + + if (am_daemon) { + char *name = lp_name(module_id); + int i, l = strlen(name); + for (i=0;i 0) { - dir = argv[0]; - argc--; - argv++; - if (chdir(dir) != 0) { - fprintf(FERROR,"chdir %s : %s (4)\n",dir,strerror(errno)); - exit_cleanup(1); - } + dir = argv[0]; + argc--; + argv++; + if (!am_daemon && chdir(dir) != 0) { + rprintf(FERROR,"chdir %s : %s (4)\n", + dir,strerror(errno)); + exit_cleanup(1); + } } if (delete_mode) - recv_exclude_list(STDIN_FILENO); + recv_exclude_list(f_in); - flist = recv_file_list(STDIN_FILENO); + flist = recv_file_list(f_in); if (!flist || flist->count == 0) { - fprintf(FERROR,"nothing to do\n"); + rprintf(FERROR,"nothing to do\n"); exit_cleanup(1); } if (argc > 0) { - if (strcmp(dir,".")) { - argv[0] += strlen(dir); - if (argv[0][0] == '/') argv[0]++; - } - local_name = get_local_name(flist,argv[0]); + if (strcmp(dir,".")) { + argv[0] += strlen(dir); + if (argv[0][0] == '/') argv[0]++; + } + local_name = get_local_name(flist,argv[0]); } - status = do_recv(STDIN_FILENO,STDOUT_FILENO,flist,local_name); + status = do_recv(f_in,f_out,flist,local_name); exit_cleanup(status); } -static void usage(FILE *f) +void start_server(int f_in, int f_out, int argc, char *argv[]) +{ + setup_protocol(f_out, f_in); + + if (am_sender) { + recv_exclude_list(f_in); + if (cvs_exclude) + add_cvs_excludes(); + do_server_sender(f_in, f_out, argc, argv); + } else { + do_server_recv(f_in, f_out, argc, argv); + } + exit_cleanup(0); +} + +static int client_run(int f_in, int f_out, int pid, int argc, char *argv[]) +{ + struct file_list *flist; + int status = 0, status2 = 0; + char *local_name = NULL; + + setup_protocol(f_out,f_in); + + if (am_sender) { + if (cvs_exclude) + add_cvs_excludes(); + if (delete_mode) + send_exclude_list(f_out); + flist = send_file_list(f_out,argc,argv); + if (verbose > 3) + rprintf(FINFO,"file list sent\n"); + send_files(flist,f_out,f_in); + if (pid != -1) { + if (verbose > 3) + rprintf(FINFO,"waiting on %d\n",pid); + waitpid(pid, &status, 0); + } + report(-1); + exit_cleanup(status); + } + + send_exclude_list(f_out); + + flist = recv_file_list(f_in); + if (!flist || flist->count == 0) { + rprintf(FINFO,"nothing to do\n"); + exit_cleanup(0); + } + + local_name = get_local_name(flist,argv[0]); + + status2 = do_recv(f_in,f_out,flist,local_name); + + report(f_in); + + if (pid != -1) { + waitpid(pid, &status, 0); + } + + return status | status2; +} + + +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; + int version; + + fd = open_socket_out(host, port); + if (fd == -1) { + exit_cleanup(1); + } + + server_options(sargs,&sargc); + + sargs[sargc++] = "."; + + if (path && *path) + sargs[sargc++] = path; + + sargs[sargc] = NULL; + + p = strchr(path,'/'); + if (p) *p = 0; + io_printf(fd,"%s\n",path); + if (p) *p = '/'; + + if (!read_line(fd, line, sizeof(line)-1)) { + return -1; + } + + if (sscanf(line,"RSYNCD %d", &version) != 1) { + return -1; + } + + while (1) { + if (!read_line(fd, line, sizeof(line)-1)) { + return -1; + } + if (strcmp(line,"RSYNCD: OK") == 0) break; + rprintf(FINFO,"%s\n", line); + } + + for (i=0;i 3) { + rprintf(FINFO,"cmd=%s machine=%s user=%s path=%s\n", + shell_cmd?shell_cmd:"", + shell_machine?shell_machine:"", + shell_user?shell_user:"", + shell_path?shell_path:""); + } + + if (!am_sender && argc != 1) { + usage(FERROR); + exit_cleanup(1); + } + + pid = do_cmd(shell_cmd,shell_machine,shell_user,shell_path,&f_in,&f_out); + +#if HAVE_SETLINEBUF + setlinebuf(stdout); + setlinebuf(stderr); +#endif + + return client_run(f_in, f_out, pid, argc, argv); +} + + +static void usage(int F) { - fprintf(f,"rsync version %s Copyright Andrew Tridgell and Paul Mackerras\n\n", + rprintf(F,"rsync version %s Copyright Andrew Tridgell and Paul Mackerras\n\n", VERSION); - fprintf(f,"Usage:\t%s [options] src user@host:dest\nOR",RSYNC_NAME); - fprintf(f,"\t%s [options] user@host:src dest\n\n",RSYNC_NAME); - fprintf(f,"Options:\n"); - fprintf(f,"-v, --verbose increase verbosity\n"); - fprintf(f,"-c, --checksum always checksum\n"); - fprintf(f,"-a, --archive archive mode (same as -rlptDog)\n"); - fprintf(f,"-r, --recursive recurse into directories\n"); - fprintf(f,"-R, --relative use relative path names\n"); - fprintf(f,"-b, --backup make backups (default ~ extension)\n"); - fprintf(f,"-u, --update update only (don't overwrite newer files)\n"); - fprintf(f,"-l, --links preserve soft links\n"); - fprintf(f,"-L, --copy-links treat soft links like regular files\n"); - fprintf(f,"-H, --hard-links preserve hard links\n"); - fprintf(f,"-p, --perms preserve permissions\n"); - fprintf(f,"-o, --owner preserve owner (root only)\n"); - fprintf(f,"-g, --group preserve group\n"); - fprintf(f,"-D, --devices preserve devices (root only)\n"); - fprintf(f,"-t, --times preserve times\n"); - fprintf(f,"-S, --sparse handle sparse files efficiently\n"); - fprintf(f,"-n, --dry-run show what would have been transferred\n"); - fprintf(f,"-W, --whole-file copy whole files, no incremental checks\n"); - fprintf(f,"-x, --one-file-system don't cross filesystem boundaries\n"); - fprintf(f,"-B, --block-size SIZE checksum blocking size\n"); - fprintf(f,"-e, --rsh COMMAND specify rsh replacement\n"); - fprintf(f," --rsync-path PATH specify path to rsync on the remote machine\n"); - fprintf(f,"-C, --cvs-exclude auto ignore files in the same way CVS does\n"); - fprintf(f," --delete delete files that don't exist on the sending side\n"); - fprintf(f,"-I, --ignore-times don't exclude files that match length and time\n"); - fprintf(f,"-z, --compress compress file data\n"); - fprintf(f," --exclude FILE exclude file FILE\n"); - fprintf(f," --exclude-from FILE exclude files listed in FILE\n"); - fprintf(f," --suffix SUFFIX override backup suffix\n"); - fprintf(f," --version print version number\n"); - - fprintf(f,"\n"); - fprintf(f,"the backup suffix defaults to %s\n",BACKUP_SUFFIX); - fprintf(f,"the block size defaults to %d\n",BLOCK_SIZE); + rprintf(F,"Usage:\t%s [options] src user@host:dest\nOR",RSYNC_NAME); + rprintf(F,"\t%s [options] user@host:src dest\n\n",RSYNC_NAME); + rprintf(F,"Options:\n"); + rprintf(F,"-v, --verbose increase verbosity\n"); + rprintf(F,"-c, --checksum always checksum\n"); + rprintf(F,"-a, --archive archive mode (same as -rlptDog)\n"); + rprintf(F,"-r, --recursive recurse into directories\n"); + rprintf(F,"-R, --relative use relative path names\n"); + rprintf(F,"-b, --backup make backups (default ~ extension)\n"); + rprintf(F,"-u, --update update only (don't overwrite newer files)\n"); + rprintf(F,"-l, --links preserve soft links\n"); + rprintf(F,"-L, --copy-links treat soft links like regular files\n"); + rprintf(F,"-H, --hard-links preserve hard links\n"); + rprintf(F,"-p, --perms preserve permissions\n"); + rprintf(F,"-o, --owner preserve owner (root only)\n"); + rprintf(F,"-g, --group preserve group\n"); + rprintf(F,"-D, --devices preserve devices (root only)\n"); + rprintf(F,"-t, --times preserve times\n"); + rprintf(F,"-S, --sparse handle sparse files efficiently\n"); + rprintf(F,"-n, --dry-run show what would have been transferred\n"); + rprintf(F,"-W, --whole-file copy whole files, no incremental checks\n"); + rprintf(F,"-x, --one-file-system don't cross filesystem boundaries\n"); + rprintf(F,"-B, --block-size SIZE checksum blocking size\n"); + rprintf(F,"-e, --rsh COMMAND specify rsh replacement\n"); + rprintf(F," --rsync-path PATH specify path to rsync on the remote machine\n"); + rprintf(F,"-C, --cvs-exclude auto ignore files in the same way CVS does\n"); + rprintf(F," --delete delete files that don't exist on the sending side\n"); + rprintf(F," --force force deletion of directories even if not empty\n"); + rprintf(F," --numeric-ids don't map uid/gid values by user/group name\n"); + rprintf(F," --timeout TIME set IO timeout in seconds\n"); + rprintf(F,"-I, --ignore-times don't exclude files that match length and time\n"); + rprintf(F,"-T --temp-dir DIR create temporary files in directory DIR\n"); + rprintf(F,"-z, --compress compress file data\n"); + rprintf(F," --exclude FILE exclude file FILE\n"); + rprintf(F," --exclude-from FILE exclude files listed in FILE\n"); + rprintf(F," --suffix SUFFIX override backup suffix\n"); + rprintf(F," --version print version number\n"); + + rprintf(F,"\n"); + rprintf(F,"the backup suffix defaults to %s\n",BACKUP_SUFFIX); + rprintf(F,"the block size defaults to %d\n",BLOCK_SIZE); } enum {OPT_VERSION,OPT_SUFFIX,OPT_SENDER,OPT_SERVER,OPT_EXCLUDE, - OPT_EXCLUDE_FROM,OPT_DELETE,OPT_RSYNC_PATH}; + OPT_EXCLUDE_FROM,OPT_DELETE,OPT_NUMERIC_IDS,OPT_RSYNC_PATH, + OPT_FORCE,OPT_TIMEOUT,OPT_DAEMON}; -static char *short_options = "oblLWHpguDCtcahvrRIxnSe:B:z"; +static char *short_options = "oblLWHpguDCtcahvrRIxnSe:B:T:z"; static struct option long_options[] = { {"version", 0, 0, OPT_VERSION}, {"server", 0, 0, OPT_SERVER}, {"sender", 0, 0, OPT_SENDER}, {"delete", 0, 0, OPT_DELETE}, + {"force", 0, 0, OPT_FORCE}, + {"numeric-ids", 0, 0, OPT_NUMERIC_IDS}, {"exclude", 1, 0, OPT_EXCLUDE}, {"exclude-from",1, 0, OPT_EXCLUDE_FROM}, {"rsync-path", 1, 0, OPT_RSYNC_PATH}, @@ -484,40 +738,21 @@ static struct option long_options[] = { {"rsh", 1, 0, 'e'}, {"suffix", 1, 0, OPT_SUFFIX}, {"block-size", 1, 0, 'B'}, + {"timeout", 1, 0, OPT_TIMEOUT}, + {"temp-dir", 1, 0, 'T'}, {"compress", 0, 0, 'z'}, + {"daemon", 0, 0, OPT_DAEMON}, {0,0,0,0}}; RETSIGTYPE sigusr1_handler(int val) { exit_cleanup(1); } -int main(int argc,char *argv[]) + +static void parse_arguments(int argc, char *argv[]) { - int pid, status = 0, status2 = 0; int opt; int option_index; - char *shell_cmd = NULL; - char *shell_machine = NULL; - char *shell_path = NULL; - char *shell_user = NULL; - char *p; - int f_in,f_out; - struct file_list *flist; - char *local_name = NULL; - -#ifdef SETPGRP_VOID - setpgrp(); -#else - setpgrp(0,0); -#endif - signal(SIGUSR1, sigusr1_handler); - - starttime = time(NULL); - am_root = (getuid() == 0); - - /* we set a 0 umask so that correct file permissions can be - carried across */ - orig_umask = umask(0); while ((opt = getopt_long(argc, argv, short_options, long_options, &option_index)) @@ -549,6 +784,14 @@ int main(int argc,char *argv[]) delete_mode = 1; break; + case OPT_FORCE: + force_delete = 1; + break; + + case OPT_NUMERIC_IDS: + numeric_ids = 1; + break; + case OPT_EXCLUDE: add_exclude(optarg); break; @@ -597,7 +840,7 @@ int main(int argc,char *argv[]) #if SUPPORT_HARD_LINKS preserve_hard_links=1; #else - fprintf(FERROR,"ERROR: hard links not supported on this platform\n"); + rprintf(FERROR,"ERROR: hard links not supported on this platform\n"); exit_cleanup(1); #endif break; @@ -653,7 +896,7 @@ int main(int argc,char *argv[]) usage(FERROR); exit_cleanup(1); } - sender = 1; + am_sender = 1; break; case 'r': @@ -672,150 +915,230 @@ int main(int argc,char *argv[]) block_size = atoi(optarg); break; + case OPT_TIMEOUT: + io_timeout = atoi(optarg); + break; + + case 'T': + tmpdir = optarg; + break; + case 'z': do_compression = 1; break; + case OPT_DAEMON: + am_daemon = 1; + break; + default: - /* fprintf(FERROR,"bad option -%c\n",opt); */ + /* rprintf(FERROR,"bad option -%c\n",opt); */ exit_cleanup(1); } } +} - while (optind--) { - argc--; - argv++; - } +static int rsync_module(int fd, int i) +{ + int argc=0; + char *argv[MAX_ARGS]; + char **argp; + char line[1024]; - signal(SIGCHLD,SIG_IGN); - signal(SIGINT,SIGNAL_CAST sig_int); - signal(SIGPIPE,SIGNAL_CAST sig_int); - signal(SIGHUP,SIGNAL_CAST sig_int); + module_id = i; - if (dry_run) - verbose = MAX(verbose,1); + if (lp_read_only(i)) + read_only = 1; -#ifndef SUPPORT_LINKS - if (!am_server && preserve_links) { - fprintf(FERROR,"ERROR: symbolic links not supported\n"); - exit_cleanup(1); - } -#endif + rprintf(FERROR,"rsyncd starting\n"); - if (am_server) { - setup_protocol(STDOUT_FILENO,STDIN_FILENO); + if (chroot(lp_path(i))) { + io_printf(fd,"ERROR: chroot failed\n"); + return -1; + } + + if (chdir("/")) { + io_printf(fd,"ERROR: chdir failed\n"); + return -1; + } + + if (setgid(lp_gid(i))) { + io_printf(fd,"ERROR: setgid failed\n"); + return -1; + } + + if (setuid(lp_uid(i))) { + io_printf(fd,"ERROR: setuid failed\n"); + return -1; + } + + io_printf(fd,"RSYNCD: OK\n"); + + argv[argc++] = "rsyncd"; + + while (1) { + if (!read_line(fd, line, sizeof(line)-1)) { + return -1; + } + + if (!*line) break; + + argv[argc] = strdup(line); + if (!argv[argc]) { + return -1; + } + + argc++; + if (argc == MAX_ARGS) { + return -1; + } + } + + parse_arguments(argc, argv); + + /* don't allow the logs to be flooded too fast */ + if (verbose > 1) verbose = 1; + + argc -= optind; + argp = argv + optind; + optind = 0; + + start_server(fd, fd, argc, argp); + + return 0; +} + +static void send_listing(int fd) +{ + int n = lp_numservices(); + int i; - if (sender) { - recv_exclude_list(STDIN_FILENO); - if (cvs_exclude) - add_cvs_excludes(); - do_server_sender(argc,argv); - } else { - do_server_recv(argc,argv); - } - exit_cleanup(0); - } + for (i=0;i 0) { + line[len] = 0; + io_printf(fd,"%s", line); + } + } + if (f) fclose(f); + io_printf(fd,"\n"); + } - p = strchr(argv[0],':'); - - if (p) { - sender = 0; - *p = 0; - shell_machine = argv[0]; - shell_path = p+1; - argc--; - argv++; - } else { - sender = 1; - - p = strchr(argv[argc-1],':'); - if (!p) { - local_server = 1; - } + /* read a single line indicating the resource that is wanted */ + while (1) { + int i; + + line[0] = 0; + if (!read_line(fd, line, sizeof(line)-1)) { + return -1; + } + + if (!*line || strcmp(line,"#list")==0) { + send_listing(fd); + 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; + } + + i = lp_number(line); + if (i == -1) { + io_printf(fd,"ERROR: Unknown module '%s'\n", line); + return -1; + } + + return rsync_module(fd, i); + } - if (local_server) { - shell_machine = NULL; - shell_path = argv[argc-1]; - } else { - *p = 0; - shell_machine = argv[argc-1]; - shell_path = p+1; - } - argc--; - } + return 0; +} - if (shell_machine) { - p = strchr(shell_machine,'@'); - if (p) { - *p = 0; - shell_user = shell_machine; - shell_machine = p+1; - } - } - if (verbose > 3) { - fprintf(FERROR,"cmd=%s machine=%s user=%s path=%s\n", - shell_cmd?shell_cmd:"", - shell_machine?shell_machine:"", - shell_user?shell_user:"", - shell_path?shell_path:""); - } - - if (!sender && argc != 1) { - usage(FERROR); - exit_cleanup(1); - } +static int daemon_main(void) +{ + if (!lp_load(RSYNCD_CONF)) { + exit_cleanup(1); + } - pid = do_cmd(shell_cmd,shell_machine,shell_user,shell_path,&f_in,&f_out); + if (is_a_socket(STDIN_FILENO)) { + /* we are running via inetd */ + return start_daemon(STDIN_FILENO); + } - setup_protocol(f_out,f_in); + become_daemon(); -#if HAVE_SETLINEBUF - setlinebuf(FINFO); - setlinebuf(FERROR); -#endif + return start_accept_loop(port, start_daemon); +} - if (verbose > 3) - fprintf(FERROR,"parent=%d child=%d sender=%d recurse=%d\n", - (int)getpid(),pid,sender,recurse); +int main(int argc,char *argv[]) +{ - if (sender) { - if (cvs_exclude) - add_cvs_excludes(); - if (delete_mode) - send_exclude_list(f_out); - flist = send_file_list(f_out,argc,argv); - if (verbose > 3) - fprintf(FERROR,"file list sent\n"); - send_files(flist,f_out,f_in); - if (verbose > 3) - fprintf(FERROR,"waiting on %d\n",pid); - waitpid(pid, &status, 0); - report(-1); - exit_cleanup(status); - } + signal(SIGUSR1, sigusr1_handler); - send_exclude_list(f_out); + starttime = time(NULL); + am_root = (getuid() == 0); - flist = recv_file_list(f_in); - if (!flist || flist->count == 0) { - fprintf(FERROR,"nothing to do\n"); - exit_cleanup(0); + /* we set a 0 umask so that correct file permissions can be + carried across */ + orig_umask = (int)umask(0); + + parse_arguments(argc, argv); + + argc -= optind; + argv += optind; + optind = 0; + + signal(SIGCHLD,SIG_IGN); + signal(SIGINT,SIGNAL_CAST sig_int); + signal(SIGPIPE,SIGNAL_CAST sig_int); + signal(SIGHUP,SIGNAL_CAST sig_int); + + if (am_daemon) { + return daemon_main(); } - local_name = get_local_name(flist,argv[0]); + if (dry_run) + verbose = MAX(verbose,1); - status2 = do_recv(f_in,f_out,flist,local_name); +#ifndef SUPPORT_LINKS + if (!am_server && preserve_links) { + rprintf(FERROR,"ERROR: symbolic links not supported\n"); + exit_cleanup(1); + } +#endif - report(f_in); + if (am_server) { + start_server(STDIN_FILENO, STDOUT_FILENO, argc, argv); + } - waitpid(pid, &status, 0); + if (argc < 2) { + usage(FERROR); + exit_cleanup(1); + } - return status | status2; + return start_client(argc, argv); }