X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/ecc81fce17f2e32fbe33c0050693899ce4d6c3cf..37802f40dcbe8956ba8ee3bfbddf10c60f94c209:/sender.c diff --git a/sender.c b/sender.c index c5dffc1c..d88d245a 100644 --- a/sender.c +++ b/sender.c @@ -27,6 +27,9 @@ extern int dry_run; extern int am_server; extern int am_daemon; extern int protocol_version; +extern int updating_basis_file; +extern int make_backups; +extern int inplace; extern struct stats stats; @@ -37,22 +40,6 @@ extern struct stats stats; * 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) -{ - sum->count = read_int(f); - sum->blength = read_int(f); - if (protocol_version < 27) { - sum->s2length = csum_length; - } else { - sum->s2length = read_int(f); - if (sum->s2length > MD4_SUM_LENGTH) { - rprintf(FERROR, "Invalid checksum length %ld\n", - (long)sum->s2length); - exit_cleanup(RERR_PROTOCOL); - } - } - sum->remainder = read_int(f); -} /** * Receive the checksums for a buffer @@ -60,7 +47,7 @@ void read_sum_head(int f, struct sum_struct *sum) static struct sum_struct *receive_sums(int f) { struct sum_struct *s; - int i; + size_t i; OFF_T offset = 0; if (!(s = new(struct sum_struct))) @@ -71,8 +58,8 @@ static struct sum_struct *receive_sums(int f) s->sums = NULL; if (verbose > 3) { - rprintf(FINFO, "count=%ld n=%u rem=%u\n", - (long)s->count, s->blength, s->remainder); + rprintf(FINFO, "count=%.0f n=%ld rem=%ld\n", + (double)s->count, (long)s->blength, (long)s->remainder); } if (s->count == 0) @@ -81,14 +68,14 @@ static struct sum_struct *receive_sums(int f) 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 < 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].flags = 0; - if (i == (int)s->count-1 && s->remainder != 0) + if (i == s->count-1 && s->remainder != 0) s->sums[i].len = s->remainder; else s->sums[i].len = s->blength; @@ -120,6 +107,7 @@ void send_files(struct file_list *flist, int f_out, int f_in) struct file_struct *file; int phase = 0; struct stats initial_stats; + int save_make_backups = make_backups; int j; if (verbose > 2) @@ -136,6 +124,9 @@ void send_files(struct file_list *flist, int f_out, int f_in) write_int(f_out, -1); if (verbose > 2) rprintf(FINFO, "send_files phase=%d\n", phase); + /* For inplace: redo phase turns off the backup + * flag so that we do a regular inplace send. */ + make_backups = 0; continue; } break; @@ -147,15 +138,26 @@ void send_files(struct file_list *flist, int f_out, int f_in) exit_cleanup(RERR_PROTOCOL); } + if (inplace && protocol_version >= 29) { + uchar fnamecmp_type = read_byte(f_in); + updating_basis_file = fnamecmp_type == FNAMECMP_FNAME; + } else + updating_basis_file = inplace && !make_backups; + file = flist->files[i]; + if (S_ISDIR(file->mode)) { + rprintf(FERROR, "[%s] got index of directory: %d\n", + who_am_i(), i); + exit_cleanup(RERR_PROTOCOL); + } stats.current_file_index = i; stats.num_transferred_files++; stats.total_transferred_size += file->length; - if (file->basedir) { + if (file->dir.root) { /* N.B. We're sure that this fits, so offset is OK. */ - offset = strlcpy(fname, file->basedir, sizeof fname); + offset = strlcpy(fname, file->dir.root, sizeof fname); if (!offset || fname[offset-1] != '/') fname[offset++] = '/'; } else @@ -208,7 +210,11 @@ void send_files(struct file_list *flist, int f_out, int f_in) return; } - mbuf = st.st_size ? map_file(fd, st.st_size, s->blength) : NULL; + if (st.st_size) { + int32 read_size = MAX(s->blength * 3, MAX_MAP_SIZE); + mbuf = map_file(fd, st.st_size, read_size, s->blength); + } else + mbuf = NULL; if (verbose > 2) { rprintf(FINFO, "send_files mapped %s of size %.0f\n", @@ -249,6 +255,7 @@ void send_files(struct file_list *flist, int f_out, int f_in) safe_fname(fname)); } } + make_backups = save_make_backups; if (verbose > 2) rprintf(FINFO, "send files finished\n");