-
- while (1) {
- cleanup_fname = NULL;
- cleanup_got_literal = 0;
-
- i = read_int(f_in);
- if (i == -1) {
- if (phase==0 && remote_version >= 13) {
- phase++;
- csum_length = SUM_LENGTH;
- if (verbose > 2)
- rprintf(FINFO,"recv_files phase=%d\n",phase);
- write_int(f_gen,-1);
- continue;
- }
- break;
- }
-
- if (i < 0 || i >= flist->count) {
- rprintf(FERROR,"Invalid file index %d in recv_files (count=%d)\n",
- i, flist->count);
- exit_cleanup(1);
- }
-
- file = flist->files[i];
- fname = f_name(file);
-
- stats.num_transferred_files++;
- stats.total_transferred_size += file->length;
-
- if (local_name)
- fname = local_name;
-
- if (dry_run) {
- if (!am_server && verbose)
- rprintf(FINFO,"%s\n",fname);
- continue;
- }
-
- if (verbose > 2)
- rprintf(FINFO,"recv_files(%s)\n",fname);
-
- /* open the file */
- fd1 = open(fname,O_RDONLY);
-
- if (fd1 != -1 && do_fstat(fd1,&st) != 0) {
- rprintf(FERROR,"fstat %s : %s\n",fname,strerror(errno));
- receive_data(f_in,NULL,-1,NULL,file->length);
- close(fd1);
- continue;
- }
-
- if (fd1 != -1 && !S_ISREG(st.st_mode)) {
- rprintf(FERROR,"%s : not a regular file (recv_files)\n",fname);
- receive_data(f_in,NULL,-1,NULL,file->length);
- close(fd1);
- continue;
- }
-
- if (fd1 != -1 && st.st_size > 0) {
- buf = map_file(fd1,st.st_size);
- if (verbose > 2)
- rprintf(FINFO,"recv mapped %s of size %d\n",fname,(int)st.st_size);
- } else {
- buf = NULL;
- }
-
- if (!get_tmpname(fnametmp,fname)) {
- if (buf) unmap_file(buf);
- if (fd1 != -1) close(fd1);
- continue;
- }
-
- if (NULL == do_mktemp(fnametmp)) {
- rprintf(FERROR,"mktemp %s failed\n",fnametmp);
- receive_data(f_in,buf,-1,NULL,file->length);
- if (buf) unmap_file(buf);
- if (fd1 != -1) close(fd1);
- continue;
- }
-
- /* 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 */
- fd2 = do_open(fnametmp,O_WRONLY|O_CREAT|O_EXCL,
- file->mode & ACCESSPERMS);
-
- if (fd2 == -1 && relative_paths && errno == ENOENT &&
- create_directory_path(fnametmp) == 0) {
- fd2 = do_open(fnametmp,O_WRONLY|O_CREAT|O_EXCL,
- file->mode & ACCESSPERMS);
- }
- if (fd2 == -1) {
- rprintf(FERROR,"open %s : %s\n",fnametmp,strerror(errno));
- receive_data(f_in,buf,-1,NULL,file->length);
- if (buf) unmap_file(buf);
- if (fd1 != -1) close(fd1);
- continue;
- }
-
- cleanup_fname = fnametmp;
- cleanup_new_fname = fname;
- cleanup_file = file;
-
- if (!am_server && verbose)
- rprintf(FINFO,"%s\n",fname);
-
- /* recv file data */
- recv_ok = receive_data(f_in,buf,fd2,fname,file->length);
-
- if (buf) unmap_file(buf);
- if (fd1 != -1) {
- close(fd1);
- }
- close(fd2);
-
- if (verbose > 2)
- rprintf(FINFO,"renaming %s to %s\n",fnametmp,fname);
-
- finish_transfer(fname, fnametmp, file);
-
- cleanup_fname = NULL;
-
- if (!recv_ok) {
- if (csum_length == SUM_LENGTH) {
- rprintf(FERROR,"ERROR: file corruption in %s. File changed during transfer?\n",
- fname);
- } else {
- if (verbose > 1)
- rprintf(FINFO,"redoing %s(%d)\n",fname,i);
- write_int(f_gen,i);
- }
- }
- }
-
- if (preserve_hard_links)
- do_hard_links(flist);
-
- /* now we need to fix any directory permissions that were
- modified during the transfer */
- for (i = 0; i < flist->count; i++) {
- file = flist->files[i];
- if (!file->basename || !S_ISDIR(file->mode)) continue;
- recv_generator(f_name(file),flist,i,-1);
- }
-
- if (verbose > 2)
- rprintf(FINFO,"recv_files finished\n");
-
- return 0;
-}
-
-
-
-void send_files(struct file_list *flist,int f_out,int f_in)
-{
- int fd;
- struct sum_struct *s;
- struct map_struct *buf;
- STRUCT_STAT st;
- char fname[MAXPATHLEN];
- int i;
- struct file_struct *file;
- int phase = 0;
-
- if (verbose > 2)
- rprintf(FINFO,"send_files starting\n");
-
- setup_readbuffer(f_in);
-
- while (1) {
- int offset=0;
-
- i = read_int(f_in);
- if (i == -1) {
- if (phase==0 && remote_version >= 13) {
- phase++;
- csum_length = SUM_LENGTH;
- write_int(f_out,-1);
- if (verbose > 2)
- rprintf(FINFO,"send_files phase=%d\n",phase);
- continue;
- }
- break;
- }
-
- if (i < 0 || i >= flist->count) {
- rprintf(FERROR,"Invalid file index %d (count=%d)\n",
- i, flist->count);
- exit_cleanup(1);
- }
-
- file = flist->files[i];
-
- stats.num_transferred_files++;
- stats.total_transferred_size += file->length;
-
- fname[0] = 0;
- if (file->basedir) {
- strlcpy(fname,file->basedir,MAXPATHLEN-1);
- if (strlen(fname) == MAXPATHLEN-1) {
- io_error = 1;
- rprintf(FERROR, "send_files failed on long-named directory %s\n",
- fname);
- return;
- }
- strlcat(fname,"/",MAXPATHLEN-1);
- offset = strlen(file->basedir)+1;
- }
- strlcat(fname,f_name(file),MAXPATHLEN-strlen(fname));
-
- if (verbose > 2)
- rprintf(FINFO,"send_files(%d,%s)\n",i,fname);
-
- if (dry_run) {
- if (!am_server && verbose)
- rprintf(FINFO,"%s\n",fname);
- write_int(f_out,i);
- continue;
- }
-
- s = receive_sums(f_in);
- if (!s) {
- io_error = 1;
- rprintf(FERROR,"receive_sums failed\n");
- return;
- }
-
- fd = open(fname,O_RDONLY);
- if (fd == -1) {
- io_error = 1;
- rprintf(FERROR,"send_files failed to open %s: %s\n",
- fname,strerror(errno));
- free_sums(s);
- continue;
- }
-
- /* map the local file */
- if (do_fstat(fd,&st) != 0) {
- io_error = 1;
- 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);
- } else {
- buf = NULL;
- }
-
- if (verbose > 2)
- rprintf(FINFO,"send_files mapped %s of size %d\n",
- fname,(int)st.st_size);
-
- write_int(f_out,i);
-
- write_int(f_out,s->count);
- write_int(f_out,s->n);
- write_int(f_out,s->remainder);
-
- if (verbose > 2)
- rprintf(FINFO,"calling match_sums %s\n",fname);
-
- if (!am_server && verbose)
- rprintf(FINFO,"%s\n",fname+offset);
-
- match_sums(f_out,s,buf,st.st_size);
-
- if (buf) unmap_file(buf);
- close(fd);
-
- free_sums(s);
-
- if (verbose > 2)
- rprintf(FINFO,"sender finished %s\n",fname);
- }
-
- if (verbose > 2)
- rprintf(FINFO,"send files finished\n");
-
- match_report();
-
- write_int(f_out,-1);