X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/a53426441b67e3715099faa0c0d0bbca19fb6369..2c713fcdfa04eb7d58c67a4a51d4cbdc37f78536:/receiver.c diff --git a/receiver.c b/receiver.c index 71897a94..ac8f26bb 100644 --- a/receiver.c +++ b/receiver.c @@ -23,12 +23,16 @@ extern int verbose; extern int recurse; extern int delete_mode; +extern int delete_after; +extern int max_delete; extern int csum_length; extern struct stats stats; extern int dry_run; extern int am_server; extern int relative_paths; +extern int keep_dirlinks; extern int preserve_hard_links; +extern int preserve_perms; extern int cvs_exclude; extern int io_error; extern char *tmpdir; @@ -39,25 +43,33 @@ extern char *backup_dir; extern char *backup_suffix; extern int backup_suffix_len; extern int cleanup_got_literal; +extern int module_id; +extern int ignore_errors; +extern int orig_umask; +extern int keep_partial; +extern int checksum_seed; static void delete_one(char *fn, int is_dir) { if (!is_dir) { if (robust_unlink(fn) != 0) { - rprintf(FERROR, "delete_one: unlink %s failed: %s\n", - full_fname(fn), strerror(errno)); - } else if (verbose) { + rsyserr(FERROR, errno, "delete_one: unlink %s failed", + full_fname(fn)); + } else if (verbose) rprintf(FINFO, "deleting %s\n", fn); - } } else { if (do_rmdir(fn) != 0) { + if (errno == ENOTDIR && keep_dirlinks) { + delete_one(fn, 0); + return; + } if (errno != ENOTEMPTY && errno != EEXIST) { - rprintf(FERROR, "delete_one: rmdir %s failed: %s\n", - full_fname(fn), strerror(errno)); + rsyserr(FERROR, errno, + "delete_one: rmdir %s failed", + full_fname(fn)); } - } else if (verbose) { + } else if (verbose) rprintf(FINFO, "deleting directory %s\n", fn); - } } } @@ -69,17 +81,13 @@ static int is_backup_file(char *fn) } -/* this deletes any files on the receiving side that are not present - * on the sending side. For version 1.6.4 I have changed the behaviour - * to match more closely what most people seem to expect of this option */ +/* This deletes any files on the receiving side that are not present + * on the sending side. */ void delete_files(struct file_list *flist) { struct file_list *local_file_list; int i, j; char *argv[1], fbuf[MAXPATHLEN]; - extern int module_id; - extern int ignore_errors; - extern int max_delete; static int deletion_count; if (cvs_exclude) @@ -90,7 +98,7 @@ void delete_files(struct file_list *flist) return; } - for (j = 0;j < flist->count; j++) { + for (j = 0; j < flist->count; j++) { if (!(flist->files[j]->flags & FLAG_TOP_DIR) || !S_ISDIR(flist->files[j]->mode)) continue; @@ -101,15 +109,17 @@ void delete_files(struct file_list *flist) continue; if (verbose > 1) - rprintf(FINFO,"deleting in %s\n", fbuf); + rprintf(FINFO, "deleting in %s\n", fbuf); for (i = local_file_list->count-1; i >= 0; i--) { - if (max_delete && deletion_count > max_delete) break; - if (!local_file_list->files[i]->basename) continue; - if (-1 == flist_find(flist,local_file_list->files[i])) { + if (max_delete && deletion_count > max_delete) + break; + if (!local_file_list->files[i]->basename) + continue; + if (flist_find(flist,local_file_list->files[i]) < 0) { char *f = f_name(local_file_list->files[i]); if (make_backups && (backup_dir || !is_backup_file(f))) { - (void) make_backup(f); + make_backup(f); if (verbose) rprintf(FINFO, "deleting %s\n", f); } else { @@ -196,11 +206,11 @@ static int receive_data(int f_in,struct map_struct *mapbuf,int fd,char *fname, char *data; static char file_sum1[MD4_SUM_LENGTH]; static char file_sum2[MD4_SUM_LENGTH]; - char *map=NULL; + char *map = NULL; read_sum_head(f_in, &sum); - sum_init(); + sum_init(checksum_seed); while ((i = recv_token(f_in, &data)) != 0) { if (do_progress) @@ -218,8 +228,8 @@ static int receive_data(int f_in,struct map_struct *mapbuf,int fd,char *fname, sum_update(data,i); if (fd != -1 && write_file(fd,data,i) != i) { - rprintf(FERROR, "write failed on %s: %s\n", - full_fname(fname), strerror(errno)); + rsyserr(FERROR, errno, "write failed on %s", + full_fname(fname)); exit_cleanup(RERR_FILEIO); } offset += i; @@ -229,7 +239,7 @@ static int receive_data(int f_in,struct map_struct *mapbuf,int fd,char *fname, i = -(i+1); offset2 = i*(OFF_T)sum.blength; len = sum.blength; - if (i == (int) sum.count-1 && sum.remainder != 0) + if (i == (int)sum.count-1 && sum.remainder != 0) len = sum.remainder; stats.matched_data += len; @@ -245,9 +255,9 @@ static int receive_data(int f_in,struct map_struct *mapbuf,int fd,char *fname, sum_update(map,len); } - if (fd != -1 && write_file(fd,map,len) != (int) len) { - rprintf(FERROR, "write failed on %s: %s\n", - full_fname(fname), strerror(errno)); + if (fd != -1 && write_file(fd, map, len) != (int)len) { + rsyserr(FERROR, errno, "write failed on %s", + full_fname(fname)); exit_cleanup(RERR_FILEIO); } offset += len; @@ -259,20 +269,18 @@ static int receive_data(int f_in,struct map_struct *mapbuf,int fd,char *fname, end_progress(total_size); if (fd != -1 && offset > 0 && sparse_end(fd) != 0) { - rprintf(FERROR, "write failed on %s: %s\n", - full_fname(fname), strerror(errno)); + rsyserr(FERROR, errno, "write failed on %s", + full_fname(fname)); exit_cleanup(RERR_FILEIO); } sum_end(file_sum1); read_buf(f_in,file_sum2,MD4_SUM_LENGTH); - if (verbose > 2) { + if (verbose > 2) rprintf(FINFO,"got file_sum\n"); - } - if (fd != -1 && memcmp(file_sum1,file_sum2,MD4_SUM_LENGTH) != 0) { + if (fd != -1 && memcmp(file_sum1, file_sum2, MD4_SUM_LENGTH) != 0) return 0; - } return 1; } @@ -281,7 +289,7 @@ static int receive_data(int f_in,struct map_struct *mapbuf,int fd,char *fname, * main routine for receiver process. * * Receiver process runs on the same host as the generator process. */ -int recv_files(int f_in,struct file_list *flist,char *local_name) +int recv_files(int f_in, struct file_list *flist, char *local_name) { int fd1,fd2; STRUCT_STAT st; @@ -291,18 +299,17 @@ int recv_files(int f_in,struct file_list *flist,char *local_name) char *fnamecmp; char fnamecmpbuf[MAXPATHLEN]; struct map_struct *mapbuf; - int i; struct file_struct *file; - int phase=0; - int recv_ok; - extern struct stats stats; - extern int preserve_perms; - extern int delete_after; - extern int orig_umask; struct stats initial_stats; + int save_make_backups = make_backups; + int i, recv_ok, phase = 0; - if (verbose > 2) { + if (verbose > 2) rprintf(FINFO,"recv_files(%d) starting\n",flist->count); + + if (flist->hlink_pool) { + pool_destroy(flist->hlink_pool); + flist->hlink_pool = NULL; } while (1) { @@ -310,15 +317,16 @@ int recv_files(int f_in,struct file_list *flist,char *local_name) i = read_int(f_in); if (i == -1) { - if (phase == 0) { - phase++; - csum_length = SUM_LENGTH; - if (verbose > 2) - rprintf(FINFO,"recv_files phase=%d\n",phase); - send_msg(MSG_DONE, "", 0); - continue; - } - break; + if (phase) + break; + phase = 1; + csum_length = SUM_LENGTH; + if (verbose > 2) + rprintf(FINFO, "recv_files phase=%d\n", phase); + send_msg(MSG_DONE, "", 0); + if (keep_partial) + make_backups = 0; /* prevents double backup */ + continue; } if (i < 0 || i >= flist->count) { @@ -340,9 +348,8 @@ int recv_files(int f_in,struct file_list *flist,char *local_name) fname = f_name_to(file, fbuf); if (dry_run) { - if (!am_server && verbose) { /* log transfer */ + if (!am_server && verbose) rprintf(FINFO, "%s\n", fname); - } continue; } @@ -356,7 +363,7 @@ int recv_files(int f_in,struct file_list *flist,char *local_name) /* open the file */ fd1 = do_open(fnamecmp, O_RDONLY, 0); - if ((fd1 == -1) && (compare_dest != NULL)) { + if (fd1 == -1 && compare_dest != NULL) { /* try the file at compare_dest instead */ pathjoin(fnamecmpbuf, sizeof fnamecmpbuf, compare_dest, fname); @@ -365,8 +372,8 @@ int recv_files(int f_in,struct file_list *flist,char *local_name) } if (fd1 != -1 && do_fstat(fd1,&st) != 0) { - rprintf(FERROR, "fstat %s failed: %s\n", - full_fname(fnamecmp), strerror(errno)); + rsyserr(FERROR, errno, "fstat %s failed", + full_fname(fnamecmp)); receive_data(f_in,NULL,-1,NULL,file->length); close(fd1); continue; @@ -400,18 +407,22 @@ int recv_files(int f_in,struct file_list *flist,char *local_name) if (fd1 != -1 && st.st_size > 0) { mapbuf = map_file(fd1,st.st_size); - if (verbose > 2) - rprintf(FINFO,"recv mapped %s of size %.0f\n",fnamecmp,(double)st.st_size); + if (verbose > 2) { + rprintf(FINFO, "recv mapped %s of size %.0f\n", + fnamecmp, (double)st.st_size); + } } else mapbuf = NULL; if (!get_tmpname(fnametmp,fname)) { - if (mapbuf) unmap_file(mapbuf); - if (fd1 != -1) close(fd1); + if (mapbuf) + unmap_file(mapbuf); + if (fd1 != -1) + close(fd1); continue; } - strlcpy(template, fnametmp, sizeof(template)); + strlcpy(template, fnametmp, sizeof template); /* we initially set the perms without the * setuid/setgid bits to ensure that there is no race @@ -426,39 +437,44 @@ int recv_files(int f_in,struct file_list *flist,char *local_name) * transferred, but that may not be the case with -R */ if (fd2 == -1 && relative_paths && errno == ENOENT && create_directory_path(fnametmp, orig_umask) == 0) { - strlcpy(fnametmp, template, sizeof(fnametmp)); + strlcpy(fnametmp, template, sizeof fnametmp); fd2 = do_mkstemp(fnametmp, file->mode & INITACCESSPERMS); } if (fd2 == -1) { - rprintf(FERROR, "mkstemp %s failed: %s\n", - full_fname(fnametmp), strerror(errno)); + rsyserr(FERROR, errno, "mkstemp %s failed", + full_fname(fnametmp)); receive_data(f_in,mapbuf,-1,NULL,file->length); - if (mapbuf) unmap_file(mapbuf); - if (fd1 != -1) close(fd1); + if (mapbuf) + unmap_file(mapbuf); + if (fd1 != -1) + close(fd1); continue; } cleanup_set(fnametmp, fname, file, mapbuf, fd1, fd2); - if (!am_server && verbose) { /* log transfer */ + if (!am_server && verbose) rprintf(FINFO, "%s\n", fname); - } /* recv file data */ recv_ok = receive_data(f_in,mapbuf,fd2,fname,file->length); log_recv(file, &initial_stats); - if (mapbuf) unmap_file(mapbuf); - if (fd1 != -1) { + if (mapbuf) + unmap_file(mapbuf); + if (fd1 != -1) close(fd1); + if (close(fd2) < 0) { + rsyserr(FERROR, errno, "close failed on %s", + full_fname(fnametmp)); + exit_cleanup(RERR_FILEIO); } - close(fd2); - if (verbose > 2) - rprintf(FINFO,"renaming %s to %s\n",fnametmp,fname); - - finish_transfer(fname, fnametmp, file); + if (recv_ok || keep_partial) + finish_transfer(fname, fnametmp, file, recv_ok); + else + do_unlink(fnametmp); cleanup_disable(); @@ -475,6 +491,7 @@ int recv_files(int f_in,struct file_list *flist,char *local_name) } } } + make_backups = save_make_backups; if (delete_after && recurse && delete_mode && !local_name && flist->count > 0)