- Call map_file() with the new block_size arg (had to delay the
[rsync/rsync.git] / receiver.c
index 0aeda10..6f2e2f5 100644 (file)
@@ -48,6 +48,10 @@ extern int ignore_errors;
 extern int orig_umask;
 extern int keep_partial;
 extern int checksum_seed;
+extern int inplace;
+
+extern struct exclude_list_struct server_exclude_list;
+
 
 static void delete_one(char *fn, int is_dir)
 {
@@ -55,9 +59,8 @@ static void delete_one(char *fn, int is_dir)
                if (robust_unlink(fn) != 0) {
                        rsyserr(FERROR, errno, "delete_one: unlink %s failed",
                                full_fname(fn));
-               } else if (verbose) {
+               } else if (verbose)
                        rprintf(FINFO, "deleting %s\n", fn);
-               }
        } else {
                if (do_rmdir(fn) != 0) {
                        if (errno == ENOTDIR && keep_dirlinks) {
@@ -69,9 +72,8 @@ static void delete_one(char *fn, int is_dir)
                                        "delete_one: rmdir %s failed",
                                        full_fname(fn));
                        }
-               } else if (verbose) {
+               } else if (verbose)
                        rprintf(FINFO, "deleting directory %s\n", fn);
-               }
        }
 }
 
@@ -121,7 +123,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 {
@@ -241,7 +243,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;
@@ -257,16 +259,30 @@ 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) {
-                       rsyserr(FERROR, errno, "write failed on %s",
-                               full_fname(fname));
-                       exit_cleanup(RERR_FILEIO);
+               if (!inplace || offset != offset2) {
+                       if (fd != -1 && write_file(fd, map, len) != (int)len) {
+                               rsyserr(FERROR, errno, "write failed on %s",
+                                       full_fname(fname));
+                               exit_cleanup(RERR_FILEIO);
+                       }
+               } else {
+                       flush_write_file(fd);
+                       if (do_lseek(fd,(OFF_T)len,SEEK_CUR) != offset+len) {
+                               rprintf(FERROR, "lseek failed on %s: %s, %lli, %lli, %i\n",
+                                       full_fname(fname), strerror(errno), do_lseek(fd,0,SEEK_CUR), (offset+len), i);
+                               exit_cleanup(RERR_FILEIO);
+                       }
                }
                offset += len;
        }
 
        flush_write_file(fd);
 
+#ifdef HAVE_FTRUNCATE
+       if (inplace)
+               ftruncate(fd, offset);
+#endif
+
        if (do_progress)
                end_progress(total_size);
 
@@ -279,21 +295,25 @@ static int receive_data(int f_in,struct map_struct *mapbuf,int fd,char *fname,
        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;
 }
 
 
+static void discard_receive_data(int f_in, OFF_T length)
+{
+       receive_data(f_in, NULL, -1, NULL, length);
+}
+
+
 /**
  * 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;
@@ -308,9 +328,8 @@ int recv_files(int f_in,struct file_list *flist,char *local_name)
        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);
@@ -324,6 +343,7 @@ int recv_files(int f_in,struct file_list *flist,char *local_name)
                if (i == -1) {
                        if (phase)
                                break;
+
                        phase = 1;
                        csum_length = SUM_LENGTH;
                        if (verbose > 2)
@@ -353,9 +373,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;
                }
 
@@ -366,6 +385,18 @@ int recv_files(int f_in,struct file_list *flist,char *local_name)
 
                fnamecmp = fname;
 
+               if (server_exclude_list.head
+                   && check_exclude(&server_exclude_list, fname,
+                                    S_ISDIR(file->mode)) < 0) {
+                       if (verbose) {
+                               rprintf(FINFO,
+                                       "skipping server-excluded update for \"%s\"\n",
+                                       fname);
+                       }
+                       discard_receive_data(f_in, file->length);
+                       continue;
+               }
+
                /* open the file */
                fd1 = do_open(fnamecmp, O_RDONLY, 0);
 
@@ -380,7 +411,7 @@ int recv_files(int f_in,struct file_list *flist,char *local_name)
                if (fd1 != -1 && do_fstat(fd1,&st) != 0) {
                        rsyserr(FERROR, errno, "fstat %s failed",
                                full_fname(fnamecmp));
-                       receive_data(f_in,NULL,-1,NULL,file->length);
+                       discard_receive_data(f_in, file->length);
                        close(fd1);
                        continue;
                }
@@ -393,7 +424,7 @@ int recv_files(int f_in,struct file_list *flist,char *local_name)
                         */
                        rprintf(FERROR,"recv_files: %s is a directory\n",
                                full_fname(fnamecmp));
-                       receive_data(f_in, NULL, -1, NULL, file->length);
+                       discard_receive_data(f_in, file->length);
                        close(fd1);
                        continue;
                }
@@ -420,54 +451,73 @@ int recv_files(int f_in,struct file_list *flist,char *local_name)
                } else
                        mapbuf = NULL;
 
-               if (!get_tmpname(fnametmp,fname)) {
-                       if (mapbuf) unmap_file(mapbuf);
-                       if (fd1 != -1) close(fd1);
-                       continue;
-               }
+               /* We now check to see if we are writing file "inplace" */
+               if (inplace)  {
+                       fd2 = do_open(fnamecmp, O_WRONLY|O_CREAT, 0);
+                       if (fd2 == -1) {
+                               rsyserr(FERROR, errno, "open %s failed",
+                                       full_fname(fnamecmp));
+                               discard_receive_data(f_in, file->length);
+                               if (mapbuf)
+                                       unmap_file(mapbuf);
+                               if (fd1 != -1)
+                                       close(fd1);
+                               continue;
+                       }
+               } else {
+                       if (!get_tmpname(fnametmp,fname)) {
+                               discard_receive_data(f_in, file->length);
+                               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
-                * condition. They are then correctly updated after
-                * the lchown. Thanks to snabb@epipe.fi for pointing
-                * this out.  We also set it initially without group
-                * access because of a similar race condition. */
-               fd2 = do_mkstemp(fnametmp, file->mode & INITACCESSPERMS);
-
-               /* in most cases parent directories will already exist
-                * because their information should have been previously
-                * 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);
+                       /* we initially set the perms without the
+                        * setuid/setgid bits to ensure that there is no race
+                        * condition. They are then correctly updated after
+                        * the lchown. Thanks to snabb@epipe.fi for pointing
+                        * this out.  We also set it initially without group
+                        * access because of a similar race condition. */
                        fd2 = do_mkstemp(fnametmp, file->mode & INITACCESSPERMS);
-               }
-               if (fd2 == -1) {
-                       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);
-                       continue;
-               }
 
-               cleanup_set(fnametmp, fname, file, mapbuf, fd1, fd2);
+                       /* in most cases parent directories will already exist
+                        * because their information should have been previously
+                        * 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);
+                               fd2 = do_mkstemp(fnametmp, file->mode & INITACCESSPERMS);
+                       }
+                       if (fd2 == -1) {
+                               rsyserr(FERROR, errno, "mkstemp %s failed",
+                                       full_fname(fnametmp));
+                               discard_receive_data(f_in, file->length);
+                               if (mapbuf)
+                                       unmap_file(mapbuf);
+                               if (fd1 != -1)
+                                       close(fd1);
+                               continue;
+                       }
 
-               if (!am_server && verbose) {    /* log transfer */
-                       rprintf(FINFO, "%s\n", fname);
+                       cleanup_set(fnametmp, fname, file, mapbuf, fd1, fd2);
                }
 
+               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));