My modified version of Chris Shoemaker's improved batch-file handling.
[rsync/rsync.git] / sender.c
index 9dbee0e..9fe5f62 100644 (file)
--- a/sender.c
+++ b/sender.c
@@ -25,6 +25,8 @@ extern struct stats stats;
 extern int io_error;
 extern int dry_run;
 extern int am_server;
+extern int am_daemon;
+extern int protocol_version;
 
 
 /**
@@ -34,12 +36,8 @@ extern int am_server;
  * and transmits them to the receiver.  The sender process runs on the
  * machine holding the source files.
  **/
-
-
 void read_sum_head(int f, struct sum_struct *sum)
 {
-       extern int protocol_version;
-
        sum->count = read_int(f);
        sum->blength = read_int(f);
        if (protocol_version < 27) {
@@ -71,10 +69,10 @@ static struct sum_struct *receive_sums(int f)
 
        s->sums = NULL;
 
-       if (verbose > 3)
-               rprintf(FINFO, "count=%ld n=%ld rem=%ld\n",
-                       (long) s->count, (long) s->blength,
-                       (long) s->remainder);
+       if (verbose > 3) {
+               rprintf(FINFO, "count=%ld n=%u rem=%u\n",
+                       (long)s->count, s->blength, s->remainder);
+       }
 
        if (s->count == 0)
                return(s);
@@ -112,7 +110,7 @@ void send_files(struct file_list *flist, int f_out, int f_in)
 {
        int fd = -1;
        struct sum_struct *s;
-       struct map_struct *buf = NULL;
+       struct map_struct *mbuf = NULL;
        STRUCT_STAT st;
        char fname[MAXPATHLEN];
        int i;
@@ -120,19 +118,13 @@ void send_files(struct file_list *flist, int f_out, int f_in)
        int phase = 0;
        extern struct stats stats;
        struct stats initial_stats;
-       extern int write_batch;
-       extern int read_batch;
-       int checksums_match;
-       int buff_len;
-       char buff[CHUNK_SIZE];
        int j;
-       int done;
 
        if (verbose > 2)
                rprintf(FINFO, "send_files starting\n");
 
        while (1) {
-               int offset = 0;
+               unsigned int offset;
 
                i = read_int(f_in);
                if (i == -1) {
@@ -155,28 +147,24 @@ void send_files(struct file_list *flist, int f_out, int f_in)
 
                file = flist->files[i];
 
+               stats.current_file_index = i;
                stats.num_transferred_files++;
                stats.total_transferred_size += file->length;
 
-               fname[0] = 0;
                if (file->basedir) {
-                       strlcpy(fname, file->basedir, MAXPATHLEN);
-                       if (strlen(fname) == MAXPATHLEN-1) {
-                               io_error |= IOERR_GENERAL;
-                               rprintf(FERROR, "send_files failed on long-named directory %s\n",
-                                       full_fname(fname));
-                               return;
-                       }
-                       strlcat(fname, "/", MAXPATHLEN);
-                       offset = strlen(file->basedir)+1;
-               }
-               f_name_to(file, fname + offset, MAXPATHLEN - offset);
+                       /* N.B. We're sure that this fits, so offset is OK. */
+                       offset = strlcpy(fname, file->basedir, sizeof fname);
+                       if (!offset || fname[offset-1] != '/')
+                               fname[offset++] = '/';
+               } else
+                       offset = 0;
+               f_name_to(file, fname + offset);
 
                if (verbose > 2)
                        rprintf(FINFO, "send_files(%d, %s)\n", i, fname);
 
                if (dry_run) {
-                       if (!am_server && verbose) {    /* log transfer */
+                       if (!am_server && verbose) {
                                rprintf(FINFO, "%s\n", fname+offset);
                        }
                        write_int(f_out, i);
@@ -192,114 +180,66 @@ void send_files(struct file_list *flist, int f_out, int f_in)
                        return;
                }
 
-               if (write_batch)
-                       write_batch_csum_info(&i, flist->count, s);
-
-               if (!read_batch) {
-                       fd = do_open(fname, O_RDONLY, 0);
-                       if (fd == -1) {
-                               if (errno == ENOENT) {
-                                       io_error |= IOERR_VANISHED;
-                                       rprintf(FINFO, "file has vanished: %s\n",
-                                               full_fname(fname));
-                               } else {
-                                       io_error |= IOERR_GENERAL;
-                                       rprintf(FERROR, "send_files failed to open %s: %s\n",
-                                               full_fname(fname), strerror(errno));
-                               }
-                               free_sums(s);
-                               continue;
-                       }
-
-                       /* map the local file */
-                       if (do_fstat(fd, &st) != 0) {
-                               io_error |= IOERR_GENERAL;
-                               rprintf(FERROR, "fstat failed: %s\n", strerror(errno));
-                               free_sums(s);
-                               close(fd);
-                               return;
-                       }
-
-                       if (st.st_size > 0) {
-                               buf = map_file(fd, st.st_size);
+               fd = do_open(fname, O_RDONLY, 0);
+               if (fd == -1) {
+                       if (errno == ENOENT) {
+                               enum logcode c = am_daemon
+                                   && protocol_version < 28 ? FERROR
+                                                            : FINFO;
+                               io_error |= IOERR_VANISHED;
+                               rprintf(c, "file has vanished: %s\n",
+                                       full_fname(fname));
                        } else {
-                               buf = NULL;
+                               io_error |= IOERR_GENERAL;
+                               rsyserr(FERROR, errno,
+                                       "send_files failed to open %s",
+                                       full_fname(fname));
                        }
+                       free_sums(s);
+                       continue;
+               }
 
-                       if (verbose > 2)
-                               rprintf(FINFO, "send_files mapped %s of size %.0f\n",
-                                       fname, (double)st.st_size);
-
-                       write_int(f_out, i);
+               /* map the local file */
+               if (do_fstat(fd, &st) != 0) {
+                       io_error |= IOERR_GENERAL;
+                       rsyserr(FERROR, errno, "fstat failed");
+                       free_sums(s);
+                       close(fd);
+                       return;
+               }
 
-                       if (write_batch)
-                               write_batch_delta_file((char *)&i, sizeof(i));
+               mbuf = st.st_size ? map_file(fd, st.st_size) : NULL;
 
-                       write_sum_head(f_out, s);
+               if (verbose > 2) {
+                       rprintf(FINFO, "send_files mapped %s of size %.0f\n",
+                               fname, (double)st.st_size);
                }
 
-               if (verbose > 2 && !read_batch)
+               write_int(f_out, i);
+               write_sum_head(f_out, s);
+
+               if (verbose > 2)
                        rprintf(FINFO, "calling match_sums %s\n", fname);
 
-               if (!am_server && verbose) {    /* log transfer */
+               if (!am_server && verbose) {
                        rprintf(FINFO, "%s\n", fname+offset);
                }
 
                set_compression(fname);
 
-               if (read_batch) {
-                       /* read checksums originally computed on sender side */
-                       read_batch_csum_info(i, s, &checksums_match);
-                       if (checksums_match) {
-                               read_batch_delta_file((char*)&j, sizeof (int));
-                               if (j != i) {    /* if flist index entries don't match*/
-                                       rprintf(FINFO, "index mismatch in send_files\n");
-                                       rprintf(FINFO, "read index = %d flist ndx = %d\n", j, i);
-                                       close_batch_delta_file();
-                                       close_batch_csums_file();
-                                       exit_cleanup(1);
-                               } else {
-                                       write_int(f_out, j);
-                                       write_sum_head(f_out, s);
-                                       done = 0;
-                                       while (!done) {
-                                               read_batch_delta_file((char*)&buff_len, sizeof (int));
-                                               write_int(f_out, buff_len);
-                                               if (buff_len == 0) {
-                                                       done = 1;
-                                               } else {
-                                                       if (buff_len > 0) {
-                                                               read_batch_delta_file(buff, buff_len);
-                                                               write_buf(f_out, buff, buff_len);
-                                                       }
-                                               }
-                                       }  /* end while  */
-                                       read_batch_delta_file( buff, MD4_SUM_LENGTH);
-                                       write_buf(f_out, buff, MD4_SUM_LENGTH);
-
-                               }  /* j=i */
-                       } else {  /* not checksum match */
-                               rprintf (FINFO, "readbatch & checksums don't match\n");
-                               rprintf (FINFO, "filename=%s is being skipped\n", fname);
-                               continue;
-                       }
-               } else  {
-                       match_sums(f_out, s, buf, st.st_size);
-                       log_send(file, &initial_stats);
-               }
+               match_sums(f_out, s, mbuf, st.st_size);
+               log_send(file, &initial_stats);
 
-               if (!read_batch) {
-                       if (buf) {
-                               j = unmap_file(buf);
-                               if (j) {
-                                       io_error |= IOERR_GENERAL;
-                                       rprintf(FERROR,
-                                           "read errors mapping %s: (%d) %s\n",
-                                           full_fname(fname), j, strerror(j));
-                               }
+               if (mbuf) {
+                       j = unmap_file(mbuf);
+                       if (j) {
+                               io_error |= IOERR_GENERAL;
+                               rsyserr(FERROR, j,
+                                       "read errors mapping %s",
+                                       full_fname(fname));
                        }
-                       close(fd);
                }
+               close(fd);
 
                free_sums(s);
 
@@ -313,14 +253,4 @@ void send_files(struct file_list *flist, int f_out, int f_in)
        match_report();
 
        write_int(f_out, -1);
-       if (write_batch || read_batch) {
-               close_batch_csums_file();
-               close_batch_delta_file();
-       }
-
 }
-
-
-
-
-