X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/bd6abc4939621c455f931905f0530115eb8c443c..efa95a184227e92bfde463483cc563e56aab5f09:/sender.c diff --git a/sender.c b/sender.c index e73ae62f..211aec25 100644 --- a/sender.c +++ b/sender.c @@ -27,6 +27,7 @@ extern int dry_run; extern int am_server; extern int am_daemon; extern int protocol_version; +extern struct stats stats; /** @@ -62,8 +63,8 @@ static struct sum_struct *receive_sums(int f) int i; OFF_T offset = 0; - s = new(struct sum_struct); - if (!s) out_of_memory("receive_sums"); + if (!(s = new(struct sum_struct))) + out_of_memory("receive_sums"); read_sum_head(f, s); @@ -77,26 +78,28 @@ static struct sum_struct *receive_sums(int f) if (s->count == 0) return(s); - s->sums = new_array(struct sum_buf, s->count); - if (!s->sums) out_of_memory("receive_sums"); + if (!(s->sums = new_array(struct sum_buf, s->count))) + out_of_memory("receive_sums"); - for (i = 0; i < (int) s->count; i++) { + for (i = 0; i < (int)s->count; i++) { s->sums[i].sum1 = read_int(f); read_buf(f, s->sums[i].sum2, s->s2length); s->sums[i].offset = offset; - s->sums[i].i = i; + s->sums[i].flags = 0; - if (i == (int) s->count-1 && s->remainder != 0) { + if (i == (int)s->count-1 && s->remainder != 0) s->sums[i].len = s->remainder; - } else { + else s->sums[i].len = s->blength; - } offset += s->sums[i].len; - if (verbose > 3) - rprintf(FINFO, "chunk[%d] len=%d offset=%.0f sum1=%08x\n", - i, s->sums[i].len, (double)s->sums[i].offset, s->sums[i].sum1); + if (verbose > 3) { + rprintf(FINFO, + "chunk[%d] len=%d offset=%.0f sum1=%08x\n", + i, s->sums[i].len, (double)s->sums[i].offset, + s->sums[i].sum1); + } } s->flength = offset; @@ -110,21 +113,14 @@ 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; struct file_struct *file; 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"); @@ -157,10 +153,11 @@ void send_files(struct file_list *flist, int f_out, int f_in) stats.num_transferred_files++; stats.total_transferred_size += file->length; - fname[0] = 0; if (file->basedir) { - offset = stringjoin(fname, sizeof fname, - file->basedir, "/", NULL); + /* 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); @@ -169,7 +166,7 @@ void send_files(struct file_list *flist, int f_out, int f_in) 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); @@ -178,124 +175,72 @@ void send_files(struct file_list *flist, int f_out, int f_in) initial_stats = stats; - s = receive_sums(f_in); - if (!s) { + if (!(s = receive_sums(f_in))) { io_error |= IOERR_GENERAL; rprintf(FERROR, "receive_sums failed\n"); return; } - if (write_batch) - write_batch_csum_info(&i, s); - - if (!read_batch) { - 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 { - 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, s->blength) : 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); @@ -309,14 +254,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(); - } - } - - - - -