Added a check for ftruncate.
[rsync/rsync.git] / receiver.c
index 108c856..ac8f26b 100644 (file)
@@ -30,6 +30,7 @@ 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;
@@ -45,25 +46,30 @@ 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);
-               }
        }
 }
 
@@ -113,7 +119,7 @@ void delete_files(struct file_list *flist)
                        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 {
@@ -200,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)
@@ -222,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;
@@ -233,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;
@@ -249,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;
@@ -263,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;
 }
 
@@ -285,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;
@@ -295,15 +299,13 @@ 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;
        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);
@@ -315,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) {
@@ -345,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;
                }
 
@@ -370,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;
@@ -405,14 +407,18 @@ 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;
                }
 
@@ -435,35 +441,40 @@ int recv_files(int f_in,struct file_list *flist,char *local_name)
                        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();
 
@@ -480,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)