X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/d9bea2ddd44fdcfbaf6f153f72d944603560036c..0f78b81511be65d8fe21af1e6ac674f9e80ac29d:/rsync.c diff --git a/rsync.c b/rsync.c index 8046b705..47268f4d 100644 --- a/rsync.c +++ b/rsync.c @@ -1,719 +1,307 @@ -/* - Copyright (C) Andrew Tridgell 1996 - Copyright (C) Paul Mackerras 1996 - - 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. -*/ +/* + * Routines common to more than one of the rsync processes. + * + * Copyright (C) 1996 Andrew Tridgell + * Copyright (C) 1996 Paul Mackerras + * Copyright (C) 2003, 2004, 2005, 2006 Wayne Davison + * + * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ #include "rsync.h" +#if defined HAVE_ICONV_OPEN && defined HAVE_ICONV_H +#include +#endif +#if defined HAVE_LIBCHARSET_H && defined HAVE_LOCALE_CHARSET +#include +#elif defined HAVE_LANGINFO_H && defined HAVE_NL_LANGINFO +#include +#endif extern int verbose; -extern int am_server; -extern int always_checksum; -extern time_t starttime; - -extern char *backup_suffix; - -extern int block_size; -extern int update_only; -extern int make_backups; -extern int preserve_links; +extern int dry_run; +extern int daemon_log_format_has_i; extern int preserve_perms; -extern int preserve_devices; +extern int preserve_executability; +extern int preserve_times; +extern int omit_dir_times; +extern int am_root; +extern int am_server; +extern int am_sender; +extern int am_generator; +extern int am_starting_up; +extern int allow_8bit_chars; extern int preserve_uid; extern int preserve_gid; -extern int preserve_times; -extern int dry_run; -extern int ignore_times; -extern int recurse; -extern int delete_mode; -extern int cvs_exclude; +extern int inplace; +extern int keep_dirlinks; +extern int make_backups; +extern mode_t orig_umask; +extern struct stats stats; +extern struct chmod_mode_struct *daemon_chmod_modes; -/* - free a sums struct - */ -static void free_sums(struct sum_struct *s) +#if defined HAVE_ICONV_OPEN && defined HAVE_ICONV_H +iconv_t ic_chck = (iconv_t)-1; + +static const char *default_charset(void) { - if (s->sums) free(s->sums); - free(s); +#if defined HAVE_LIBCHARSET_H && defined HAVE_LOCALE_CHARSET + return locale_charset(); +#elif defined HAVE_LANGINFO_H && defined HAVE_NL_LANGINFO + return nl_langinfo(CODESET); +#else + return ""; /* Works with (at the very least) gnu iconv... */ +#endif } - - -/* - send a sums struct down a fd - */ -static void send_sums(struct sum_struct *s,int f_out) +void setup_iconv() { - int i; - - /* tell the other guy how many we are going to be doing and how many - bytes there are in the last chunk */ - write_int(f_out,s?s->count:0); - write_int(f_out,s?s->n:block_size); - write_int(f_out,s?s->remainder:0); - if (s) - for (i=0;icount;i++) { - write_int(f_out,s->sums[i].sum1); - write_buf(f_out,s->sums[i].sum2,SUM_LENGTH); - } - write_flush(f_out); + if (!am_server && !allow_8bit_chars) { + const char *defset = default_charset(); + + /* It's OK if this fails... */ + ic_chck = iconv_open(defset, defset); + + if (verbose > 3) { + if (ic_chck == (iconv_t)-1) { + rprintf(FINFO, + "note: iconv_open(\"%s\", \"%s\") failed (%d)" + " -- using isprint() instead of iconv().\n", + defset, defset, errno); + } else { + rprintf(FINFO, + "note: iconv_open(\"%s\", \"%s\") succeeded.\n", + defset, defset); + } + } + } } - +#endif /* - generate a stream of signatures/checksums that describe a buffer - - generate approximately one checksum every n bytes + free a sums struct */ -static struct sum_struct *generate_sums(char *buf,off_t len,int n) +void free_sums(struct sum_struct *s) { - int i; - struct sum_struct *s; - int count; - int block_len = n; - int remainder = (len%block_len); - off_t offset = 0; - - count = (len+(block_len-1))/block_len; - - s = (struct sum_struct *)malloc(sizeof(*s)); - if (!s) out_of_memory("generate_sums"); - - s->count = count; - s->remainder = remainder; - s->n = n; - s->flength = len; - - if (count==0) { - s->sums = NULL; - return s; - } - - if (verbose > 3) - fprintf(stderr,"count=%d rem=%d n=%d flength=%d\n", - s->count,s->remainder,s->n,(int)s->flength); - - s->sums = (struct sum_buf *)malloc(sizeof(s->sums[0])*s->count); - if (!s->sums) out_of_memory("generate_sums"); - - for (i=0;isums[i].sum1 = get_checksum1(map,n1); - get_checksum2(map,n1,s->sums[i].sum2); - - s->sums[i].offset = offset; - s->sums[i].len = n1; - s->sums[i].i = i; - - if (verbose > 3) - fprintf(stderr,"chunk[%d] offset=%d len=%d sum1=%08x\n", - i,(int)s->sums[i].offset,s->sums[i].len,s->sums[i].sum1); - - len -= n1; - offset += n1; - } - - return s; + if (s->sums) free(s->sums); + free(s); } - -/* - receive the checksums for a buffer - */ -static struct sum_struct *receive_sums(int f) +/* This is only called when we aren't preserving permissions. Figure out what + * the permissions should be and return them merged back into the mode. */ +mode_t dest_mode(mode_t flist_mode, mode_t cur_mode, int exists) { - struct sum_struct *s; - int i; - off_t offset = 0; - int block_len; - - s = (struct sum_struct *)malloc(sizeof(*s)); - if (!s) out_of_memory("receive_sums"); - - s->count = read_int(f); - s->n = read_int(f); - s->remainder = read_int(f); - s->sums = NULL; - - if (verbose > 3) - fprintf(stderr,"count=%d n=%d rem=%d\n", - s->count,s->n,s->remainder); - - block_len = s->n; - - if (s->count == 0) - return(s); - - s->sums = (struct sum_buf *)malloc(sizeof(s->sums[0])*s->count); - if (!s->sums) out_of_memory("receive_sums"); - - for (i=0;icount;i++) { - s->sums[i].sum1 = read_int(f); - read_buf(f,s->sums[i].sum2,SUM_LENGTH); - - s->sums[i].offset = offset; - s->sums[i].i = i; - - if (i == s->count-1 && s->remainder != 0) { - s->sums[i].len = s->remainder; - } else { - s->sums[i].len = s->n; - } - offset += s->sums[i].len; - - if (verbose > 3) - fprintf(stderr,"chunk[%d] len=%d offset=%d sum1=%08x\n", - i,s->sums[i].len,(int)s->sums[i].offset,s->sums[i].sum1); - } - - s->flength = offset; - - return s; + /* If the file already exists, we'll return the local permissions, + * possibly tweaked by the --executability option. */ + if (exists) { + if (preserve_executability && S_ISREG(flist_mode)) { + /* If the source file is executable, grant execute + * rights to everyone who can read, but ONLY if the + * file isn't already executable. */ + if (!(flist_mode & 0111)) + cur_mode &= ~0111; + else if (!(cur_mode & 0111)) + cur_mode |= (cur_mode & 0444) >> 2; + } + } else + cur_mode = flist_mode & ACCESSPERMS & ~orig_umask; + if (daemon_chmod_modes && !S_ISLNK(flist_mode)) + cur_mode = tweak_mode(cur_mode, daemon_chmod_modes); + return (flist_mode & ~CHMOD_BITS) | (cur_mode & CHMOD_BITS); } - -static void set_perms(char *fname,struct file_struct *file,struct stat *st, - int report) +int set_file_attrs(char *fname, struct file_struct *file, STRUCT_STAT *st, + int flags) { - int updated = 0; - struct stat st2; - - if (dry_run) return; - - if (!st) { - if (stat(fname,&st2) != 0) { - fprintf(stderr,"stat %s : %s\n",fname,strerror(errno)); - return; - } - st = &st2; - } + int updated = 0; + STRUCT_STAT st2; + int change_uid, change_gid; + + if (!st) { + if (dry_run) + return 1; + if (link_stat(fname, &st2, 0) < 0) { + rsyserr(FERROR, errno, "stat %s failed", + full_fname(fname)); + return 0; + } + st = &st2; + if (!preserve_perms && S_ISDIR(file->mode) + && st->st_mode & S_ISGID) { + /* We just created this directory and its setgid + * bit is on, so make sure it stays on. */ + file->mode |= S_ISGID; + } + } - if (preserve_times && st->st_mtime != file->modtime) { - updated = 1; - if (set_modtime(fname,file->modtime) != 0) { - fprintf(stderr,"failed to set times on %s : %s\n", - fname,strerror(errno)); - return; - } - } + if (!preserve_times || (S_ISDIR(st->st_mode) && omit_dir_times)) + flags |= ATTRS_SKIP_MTIME; + if (!(flags & ATTRS_SKIP_MTIME) + && cmp_time(st->st_mtime, file->modtime) != 0) { + int ret = set_modtime(fname, file->modtime, st->st_mode); + if (ret < 0) { + rsyserr(FERROR, errno, "failed to set times on %s", + full_fname(fname)); + return 0; + } + if (ret == 0) /* ret == 1 if symlink could not be set */ + updated = 1; + } -#ifdef HAVE_CHMOD - if (preserve_perms && st->st_mode != file->mode) { - updated = 1; - if (chmod(fname,file->mode) != 0) { - fprintf(stderr,"failed to set permissions on %s : %s\n", - fname,strerror(errno)); - return; - } - } + change_uid = am_root && preserve_uid && st->st_uid != file->uid; + change_gid = preserve_gid && file->gid != GID_NONE + && st->st_gid != file->gid; +#if !defined HAVE_LCHOWN && !defined CHOWN_MODIFIES_SYMLINK + if (S_ISLNK(st->st_mode)) + ; + else #endif - - if ((preserve_uid && st->st_uid != file->uid) || - (preserve_gid && st->st_gid != file->gid)) { - updated = 1; - if (chown(fname, - preserve_uid?file->uid:-1, - preserve_gid?file->gid:-1) != 0) { - if (verbose>1 || preserve_uid) - fprintf(stderr,"chown %s : %s\n",fname,strerror(errno)); - return; - } - } - - if (verbose > 1 && report) { - if (updated) - fprintf(am_server?stderr:stdout,"%s\n",fname); - else - fprintf(am_server?stderr:stdout,"%s is uptodate\n",fname); - } -} - - -void recv_generator(char *fname,struct file_list *flist,int i,int f_out) -{ - int fd; - struct stat st; - char *buf; - struct sum_struct *s; - char sum[SUM_LENGTH]; - int statret; - - if (verbose > 2) - fprintf(stderr,"recv_generator(%s)\n",fname); - - statret = lstat(fname,&st); - -#if SUPPORT_LINKS - if (preserve_links && S_ISLNK(flist->files[i].mode)) { - char lnk[MAXPATHLEN]; - int l; - if (statret == 0) { - l = readlink(fname,lnk,MAXPATHLEN-1); - if (l > 0) { - lnk[l] = 0; - if (strcmp(lnk,flist->files[i].link) == 0) { - if (verbose > 1) - fprintf(am_server?stderr:stdout,"%s is uptodate\n",fname); - return; + if (change_uid || change_gid) { + if (verbose > 2) { + if (change_uid) { + rprintf(FINFO, + "set uid of %s from %ld to %ld\n", + fname, + (long)st->st_uid, (long)file->uid); + } + if (change_gid) { + rprintf(FINFO, + "set gid of %s from %ld to %ld\n", + fname, + (long)st->st_gid, (long)file->gid); + } + } + if (do_lchown(fname, + change_uid ? file->uid : st->st_uid, + change_gid ? file->gid : st->st_gid) != 0) { + /* shouldn't have attempted to change uid or gid + * unless have the privilege */ + rsyserr(FERROR, errno, "%s %s failed", + change_uid ? "chown" : "chgrp", + full_fname(fname)); + return 0; + } + /* a lchown had been done - we have to re-stat if the + * destination had the setuid or setgid bits set due + * to the side effect of the chown call */ + if (st->st_mode & (S_ISUID | S_ISGID)) { + link_stat(fname, st, + keep_dirlinks && S_ISDIR(st->st_mode)); + } + updated = 1; } - } - } - if (!dry_run) unlink(fname); - if (!dry_run && symlink(flist->files[i].link,fname) != 0) { - fprintf(stderr,"link %s -> %s : %s\n", - fname,flist->files[i].link,strerror(errno)); - } else { - if (verbose) - fprintf(am_server?stderr:stdout,"%s -> %s\n",fname,flist->files[i].link); - } - return; - } -#endif -#ifdef HAVE_MKNOD - if (preserve_devices && IS_DEVICE(flist->files[i].mode)) { - if (statret != 0 || - st.st_mode != flist->files[i].mode || - st.st_rdev != flist->files[i].dev) { - if (!dry_run) unlink(fname); - if (verbose > 2) - fprintf(stderr,"mknod(%s,0%o,0x%x)\n", - fname,(int)flist->files[i].mode,(int)flist->files[i].dev); - if (!dry_run && - mknod(fname,flist->files[i].mode,flist->files[i].dev) != 0) { - fprintf(stderr,"mknod %s : %s\n",fname,strerror(errno)); - } else { - set_perms(fname,&flist->files[i],NULL,0); - if (verbose) - fprintf(am_server?stderr:stdout,"%s\n",fname); - } - } else { - set_perms(fname,&flist->files[i],&st,1); - } - return; - } +#ifdef HAVE_CHMOD + if ((st->st_mode & CHMOD_BITS) != (file->mode & CHMOD_BITS)) { + int ret = do_chmod(fname, file->mode); + if (ret < 0) { + rsyserr(FERROR, errno, + "failed to set permissions on %s", + full_fname(fname)); + return 0; + } + if (ret == 0) /* ret == 1 if symlink could not be set */ + updated = 1; + } #endif - if (!S_ISREG(flist->files[i].mode)) { - fprintf(stderr,"skipping non-regular file %s\n",fname); - return; - } - - if (statret == -1) { - if (errno == ENOENT) { - write_int(f_out,i); - if (!dry_run) send_sums(NULL,f_out); - } else { - if (verbose > 1) - fprintf(stderr,"recv_generator failed to open %s\n",fname); - } - return; - } - - if (!S_ISREG(st.st_mode)) { - fprintf(stderr,"%s : not a regular file\n",fname); - return; - } - - if (update_only && st.st_mtime >= flist->files[i].modtime) { - if (verbose > 1) - fprintf(stderr,"%s is newer\n",fname); - return; - } - - if (always_checksum && S_ISREG(st.st_mode)) { - file_checksum(fname,sum,st.st_size); - } - - if (st.st_size == flist->files[i].length && - ((!ignore_times && st.st_mtime == flist->files[i].modtime) || - (always_checksum && S_ISREG(st.st_mode) && - memcmp(sum,flist->files[i].sum,SUM_LENGTH) == 0))) { - set_perms(fname,&flist->files[i],&st,1); - return; - } - - if (dry_run) { - write_int(f_out,i); - return; - } - - /* open the file */ - fd = open(fname,O_RDONLY); - - if (fd == -1) { - fprintf(stderr,"failed to open %s : %s\n",fname,strerror(errno)); - return; - } - - if (st.st_size > 0) { - buf = map_file(fd,st.st_size); - } else { - buf = NULL; - } - - if (verbose > 3) - fprintf(stderr,"mapped %s of size %d\n",fname,(int)st.st_size); - - s = generate_sums(buf,st.st_size,block_size); - - write_int(f_out,i); - send_sums(s,f_out); - write_flush(f_out); - - close(fd); - unmap_file(buf,st.st_size); - - free_sums(s); -} - - - -static void receive_data(int f_in,char *buf,int fd,char *fname) -{ - int i,n,remainder,len,count; - off_t offset = 0; - off_t offset2; - - count = read_int(f_in); - n = read_int(f_in); - remainder = read_int(f_in); - - for (i=read_int(f_in); i != 0; i=read_int(f_in)) { - if (i > 0) { - if (verbose > 3) - fprintf(stderr,"data recv %d at %d\n",i,(int)offset); - - if (read_write(f_in,fd,i) != i) { - fprintf(stderr,"write failed on %s : %s\n",fname,strerror(errno)); - exit(1); - } - offset += i; - } else { - i = -(i+1); - offset2 = i*n; - len = n; - if (i == count-1 && remainder != 0) - len = remainder; - - if (verbose > 3) - fprintf(stderr,"chunk[%d] of size %d at %d offset=%d\n", - i,len,(int)offset2,(int)offset); - - if (write(fd,map_ptr(buf,offset2,len),len) != len) { - fprintf(stderr,"write failed on %s : %s\n",fname,strerror(errno)); - exit(1); - } - offset += len; - } - } -} - - -static void delete_one(struct file_struct *f) -{ - if (!S_ISDIR(f->mode)) { - if (!dry_run && unlink(f->name) != 0) { - fprintf(stderr,"unlink %s : %s\n",f->name,strerror(errno)); - } else if (verbose) { - fprintf(stderr,"deleting %s\n",f->name); - } - } else { - if (!dry_run && rmdir(f->name) != 0) { - if (errno != ENOTEMPTY) - fprintf(stderr,"rmdir %s : %s\n",f->name,strerror(errno)); - } else if (verbose) { - fprintf(stderr,"deleting directory %s\n",f->name); - } - } + if (verbose > 1 && flags & ATTRS_REPORT) { + enum logcode code = daemon_log_format_has_i || dry_run + ? FCLIENT : FINFO; + if (updated) + rprintf(code, "%s\n", fname); + else + rprintf(code, "%s is uptodate\n", fname); + } + return updated; } - -static void delete_files(struct file_list *flist) +RETSIGTYPE sig_int(UNUSED(int val)) { - struct file_list *local_file_list; - char *dot="."; - int i; - - if (!(local_file_list = send_file_list(-1,recurse,1,&dot))) - return; - - for (i=local_file_list->count;i>=0;i--) { - if (!local_file_list->files[i].name) continue; - if (-1 == flist_find(flist,&local_file_list->files[i])) { - delete_one(&local_file_list->files[i]); - } - } + /* KLUGE: if the user hits Ctrl-C while ssh is prompting + * for a password, then our cleanup's sending of a SIGUSR1 + * signal to all our children may kill ssh before it has a + * chance to restore the tty settings (i.e. turn echo back + * on). By sleeping for a short time, ssh gets a bigger + * chance to do the right thing. If child processes are + * not ssh waiting for a password, then this tiny delay + * shouldn't hurt anything. */ + msleep(400); + exit_cleanup(RERR_SIGNAL); } -static char *cleanup_fname = NULL; - -void sig_int(void) +/* Finish off a file transfer: renaming the file and setting the file's + * attributes (e.g. permissions, ownership, etc.). If partialptr is not + * NULL and the robust_rename() call is forced to copy the temp file, we + * stage the file into the partial-dir and then rename it into place. */ +void finish_transfer(char *fname, char *fnametmp, char *partialptr, + struct file_struct *file, int ok_to_set_time, + int overwriting_basis) { - if (cleanup_fname) - unlink(cleanup_fname); - exit(1); -} - - -int recv_files(int f_in,struct file_list *flist,char *local_name) -{ - int fd1,fd2; - struct stat st; - char *fname; - char fnametmp[MAXPATHLEN]; - char *buf; - int i; - - if (verbose > 2) { - fprintf(stderr,"recv_files(%d) starting\n",flist->count); - } - - if (recurse && delete_mode && !local_name && flist->count>0) { - delete_files(flist); - } - - while (1) - { - i = read_int(f_in); - if (i == -1) break; - - fname = flist->files[i].name; - - if (local_name) - fname = local_name; - - if (dry_run) { - if (!am_server && verbose) - printf("%s\n",fname); - continue; - } - - if (verbose > 2) - fprintf(stderr,"recv_files(%s)\n",fname); - - /* open the file */ - fd1 = open(fname,O_RDONLY); - - if (fd1 != -1 && fstat(fd1,&st) != 0) { - fprintf(stderr,"fstat %s : %s\n",fname,strerror(errno)); - close(fd1); - return -1; - } - - if (fd1 != -1 && !S_ISREG(st.st_mode)) { - fprintf(stderr,"%s : not a regular file\n",fname); - close(fd1); - return -1; - } - - if (fd1 != -1 && st.st_size > 0) { - buf = map_file(fd1,st.st_size); - } else { - buf = NULL; - } + int ret; - if (verbose > 2) - fprintf(stderr,"mapped %s of size %d\n",fname,(int)st.st_size); - - /* open tmp file */ - sprintf(fnametmp,"%s.XXXXXX",fname); - if (NULL == mktemp(fnametmp)) { - fprintf(stderr,"mktemp %s failed\n",fnametmp); - return -1; - } - fd2 = open(fnametmp,O_WRONLY|O_CREAT,flist->files[i].mode); - if (fd2 == -1) { - fprintf(stderr,"open %s : %s\n",fnametmp,strerror(errno)); - return -1; - } - - cleanup_fname = fnametmp; - - if (!am_server && verbose) - printf("%s\n",fname); - - /* recv file data */ - receive_data(f_in,buf,fd2,fname); - - if (fd1 != -1) { - unmap_file(buf,st.st_size); - close(fd1); - } - close(fd2); - - if (verbose > 2) - fprintf(stderr,"renaming %s to %s\n",fnametmp,fname); - - if (make_backups) { - char fnamebak[MAXPATHLEN]; - sprintf(fnamebak,"%s%s",fname,backup_suffix); - if (rename(fname,fnamebak) != 0 && errno != ENOENT) { - fprintf(stderr,"rename %s %s : %s\n",fname,fnamebak,strerror(errno)); - exit(1); + if (inplace) { + if (verbose > 2) + rprintf(FINFO, "finishing %s\n", fname); + fnametmp = fname; + goto do_set_file_attrs; } - } - - /* move tmp file over real file */ - if (rename(fnametmp,fname) != 0) { - fprintf(stderr,"rename %s -> %s : %s\n", - fnametmp,fname,strerror(errno)); - } - - cleanup_fname = NULL; - - set_perms(fname,&flist->files[i],NULL,0); - } - - if (verbose > 2) - fprintf(stderr,"recv_files finished\n"); - - return 0; -} - - - -off_t send_files(struct file_list *flist,int f_out,int f_in) -{ - int fd; - struct sum_struct *s; - char *buf; - struct stat st; - char fname[MAXPATHLEN]; - off_t total=0; - int i; - - if (verbose > 2) - fprintf(stderr,"send_files starting\n"); - - setup_nonblocking(f_in,f_out); - - while (1) - { - i = read_int(f_in); - if (i == -1) break; - - fname[0] = 0; - if (flist->files[i].dir) { - strcpy(fname,flist->files[i].dir); - strcat(fname,"/"); - } - strcat(fname,flist->files[i].name); - - if (verbose > 2) - fprintf(stderr,"send_files(%d,%s)\n",i,fname); - - if (dry_run) { - if (!am_server && verbose) - printf("%s\n",fname); - write_int(f_out,i); - continue; - } - s = receive_sums(f_in); - if (!s) { - fprintf(stderr,"receive_sums failed\n"); - return -1; - } - - fd = open(fname,O_RDONLY); - if (fd == -1) { - fprintf(stderr,"send_files failed to open %s: %s\n", - fname,strerror(errno)); - continue; - } - - /* map the local file */ - if (fstat(fd,&st) != 0) { - fprintf(stderr,"fstat failed : %s\n",strerror(errno)); - return -1; - } - - if (st.st_size > 0) { - buf = map_file(fd,st.st_size); - } else { - buf = NULL; - } - - if (verbose > 2) - fprintf(stderr,"send_files mapped %s of size %d\n", - fname,(int)st.st_size); - - write_int(f_out,i); - - write_int(f_out,s->count); - write_int(f_out,s->n); - write_int(f_out,s->remainder); - - if (verbose > 2) - fprintf(stderr,"calling match_sums %s\n",fname); - - if (!am_server && verbose) - printf("%s\n",fname); - - match_sums(f_out,s,buf,st.st_size); - write_flush(f_out); - - unmap_file(buf,st.st_size); - close(fd); - - free_sums(s); - - if (verbose > 2) - fprintf(stderr,"sender finished %s\n",fname); - - total += st.st_size; - } - - match_report(); - - write_int(f_out,-1); - write_flush(f_out); - - return total; + if (make_backups && overwriting_basis && !make_backup(fname)) + return; + + /* Change permissions before putting the file into place. */ + set_file_attrs(fnametmp, file, NULL, + ok_to_set_time ? 0 : ATTRS_SKIP_MTIME); + + /* move tmp file over real file */ + if (verbose > 2) + rprintf(FINFO, "renaming %s to %s\n", fnametmp, fname); + ret = robust_rename(fnametmp, fname, partialptr, + file->mode & INITACCESSPERMS); + if (ret < 0) { + rsyserr(FERROR, errno, "%s %s -> \"%s\"", + ret == -2 ? "copy" : "rename", + full_fname(fnametmp), fname); + do_unlink(fnametmp); + return; + } + if (ret == 0) { + /* The file was moved into place (not copied), so it's done. */ + return; + } + /* The file was copied, so tweak the perms of the copied file. If it + * was copied to partialptr, move it into its final destination. */ + fnametmp = partialptr ? partialptr : fname; + + do_set_file_attrs: + set_file_attrs(fnametmp, file, NULL, + ok_to_set_time ? 0 : ATTRS_SKIP_MTIME); + + if (partialptr) { + if (do_rename(fnametmp, fname) < 0) { + rsyserr(FERROR, errno, "rename %s -> \"%s\"", + full_fname(fnametmp), fname); + } else + handle_partial_dir(partialptr, PDIR_DELETE); + } } - - -void generate_files(int f,struct file_list *flist,char *local_name) +const char *who_am_i(void) { - int i; - - if (verbose > 2) - fprintf(stderr,"generator starting pid=%d count=%d\n", - (int)getpid(),flist->count); - - for (i = 0; i < flist->count; i++) { - if (!flist->files[i].name) continue; - if (S_ISDIR(flist->files[i].mode)) { - if (dry_run) continue; - if (mkdir(flist->files[i].name,flist->files[i].mode) != 0 && - errno != EEXIST) { - fprintf(stderr,"mkdir %s : %s\n", - flist->files[i].name,strerror(errno)); - } - continue; - } - recv_generator(local_name?local_name:flist->files[i].name, - flist,i,f); - } - write_int(f,-1); - write_flush(f); - if (verbose > 2) - fprintf(stderr,"generator wrote %d\n",write_total()); + if (am_starting_up) + return am_server ? "server" : "client"; + return am_sender ? "sender" : am_generator ? "generator" : "receiver"; }