-
-
-/* this deletes any files on the receiving side that are not present
- on the sending side. For version 1.6.4 I have changed the behaviour
- to match more closely what most people seem to expect of this option */
-static void delete_files(struct file_list *flist)
-{
- struct file_list *local_file_list;
- int i, j;
-
- if (cvs_exclude)
- add_cvs_excludes();
-
- if (io_error) {
- fprintf(FINFO,"IO error encountered - skipping file deletion\n");
- return;
- }
-
- for (j=0;j<flist->count;j++) {
- char *name = f_name(flist->files[j]);
-
- if (!S_ISDIR(flist->files[j]->mode)) continue;
-
- if (delete_already_done(flist, j)) continue;
-
- if (!(local_file_list = send_file_list(-1,1,&name)))
- continue;
-
- if (verbose > 1)
- fprintf(FINFO,"deleting in %s\n", name);
-
- for (i=local_file_list->count-1;i>=0;i--) {
- if (!local_file_list->files[i]->basename) continue;
- if (-1 == flist_find(flist,local_file_list->files[i])) {
- delete_one(local_file_list->files[i]);
- }
- }
- flist_free(local_file_list);
- }
-}
-
-static char *cleanup_fname;
-
-void exit_cleanup(int code)
-{
- if (cleanup_fname)
- do_unlink(cleanup_fname);
- signal(SIGUSR1, SIG_IGN);
- if (code) {
- kill_all(SIGUSR1);
- }
- exit(code);
-}
-
-void sig_int(void)
-{
- exit_cleanup(1);
-}
-
-
-int recv_files(int f_in,struct file_list *flist,char *local_name,int f_gen)
-{
- int fd1,fd2;
- struct stat st;
- char *fname;
- char fnametmp[MAXPATHLEN];
- struct map_struct *buf;
- int i;
- struct file_struct *file;
- int phase=0;
- int recv_ok;
-
- if (verbose > 2) {
- fprintf(FINFO,"recv_files(%d) starting\n",flist->count);
- }
-
- if (recurse && delete_mode && !local_name && flist->count>0) {
- delete_files(flist);
- }
-
- while (1)
- {
- i = read_int(f_in);
- if (i == -1) {
- if (phase==0 && remote_version >= 13) {
- phase++;
- csum_length = SUM_LENGTH;
- if (verbose > 2)
- fprintf(FINFO,"recv_files phase=%d\n",phase);
- write_int(f_gen,-1);
- write_flush(f_gen);
- continue;
- }
- break;
- }
-
- file = flist->files[i];
- fname = f_name(file);
-
- if (local_name)
- fname = local_name;
-
- if (dry_run) {
- if (!am_server && verbose)
- printf("%s\n",fname);
- continue;
- }
-
- if (verbose > 2)
- fprintf(FINFO,"recv_files(%s)\n",fname);
-
- /* open the file */
- fd1 = open(fname,O_RDONLY);
-
- if (fd1 != -1 && fstat(fd1,&st) != 0) {
- fprintf(FERROR,"fstat %s : %s\n",fname,strerror(errno));
- receive_data(f_in,NULL,-1,NULL);
- close(fd1);
- continue;
- }
-
- if (fd1 != -1 && !S_ISREG(st.st_mode)) {
- fprintf(FERROR,"%s : not a regular file (recv_files)\n",fname);
- receive_data(f_in,NULL,-1,NULL);
- close(fd1);
- continue;
- }
-
- if (fd1 != -1 && st.st_size > 0) {
- buf = map_file(fd1,st.st_size);
- if (verbose > 2)
- fprintf(FINFO,"recv mapped %s of size %d\n",fname,(int)st.st_size);
- } else {
- buf = NULL;
- }
-
- /* open tmp file */
- if (strlen(fname) > (MAXPATHLEN-8)) {
- fprintf(FERROR,"filename too long\n");
- if (buf) unmap_file(buf);
- close(fd1);
- continue;
- }
- if (tmpdir) {
- char *f;
- f = strrchr(fname,'/');
- if (f == NULL)
- f = fname;
- else
- f++;
- sprintf(fnametmp,"%s/%s.XXXXXX",tmpdir,f);
- } else {
- sprintf(fnametmp,"%s.XXXXXX",fname);
- }
- if (NULL == mktemp(fnametmp)) {
- fprintf(FERROR,"mktemp %s failed\n",fnametmp);
- receive_data(f_in,buf,-1,NULL);
- if (buf) unmap_file(buf);
- close(fd1);
- continue;
- }
- fd2 = do_open(fnametmp,O_WRONLY|O_CREAT|O_EXCL,file->mode);
- if (fd2 == -1 && relative_paths && errno == ENOENT &&
- create_directory_path(fnametmp) == 0) {
- fd2 = do_open(fnametmp,O_WRONLY|O_CREAT|O_EXCL,file->mode);
- }
- if (fd2 == -1) {
- fprintf(FERROR,"open %s : %s\n",fnametmp,strerror(errno));
- receive_data(f_in,buf,-1,NULL);
- if (buf) unmap_file(buf);
- close(fd1);
- continue;
- }
-
- cleanup_fname = fnametmp;
-
- if (!am_server && verbose)
- printf("%s\n",fname);
-
- /* recv file data */
- recv_ok = receive_data(f_in,buf,fd2,fname);
-
- if (buf) unmap_file(buf);
- if (fd1 != -1) {
- close(fd1);
- }
- close(fd2);
-
- if (verbose > 2)
- fprintf(FINFO,"renaming %s to %s\n",fnametmp,fname);
-
- if (make_backups) {
- char fnamebak[MAXPATHLEN];
- if (strlen(fname) + strlen(backup_suffix) > (MAXPATHLEN-1)) {
- fprintf(FERROR,"backup filename too long\n");
- continue;
- }
- sprintf(fnamebak,"%s%s",fname,backup_suffix);
- if (rename(fname,fnamebak) != 0 && errno != ENOENT) {
- fprintf(FERROR,"rename %s %s : %s\n",fname,fnamebak,strerror(errno));
- continue;
- }
- }
-
- /* move tmp file over real file */
- if (rename(fnametmp,fname) != 0) {
- if (errno == EXDEV) {
- /* rename failed on cross-filesystem link.
- Copy the file instead. */
- if (copy_file(fnametmp,fname, file->mode)) {
- fprintf(FERROR,"copy %s -> %s : %s\n",
- fnametmp,fname,strerror(errno));
- } else {
- set_perms(fname,file,NULL,0);
- }
- do_unlink(fnametmp);
- } else {
- fprintf(FERROR,"rename %s -> %s : %s\n",
- fnametmp,fname,strerror(errno));
- do_unlink(fnametmp);
- }
- } else {
- set_perms(fname,file,NULL,0);
- }
-
- cleanup_fname = NULL;
-
-
- if (!recv_ok) {
- if (csum_length == SUM_LENGTH) {
- fprintf(FERROR,"ERROR: file corruption in %s. File changed during transfer?\n",
- fname);
- } else {
- if (verbose > 1)
- fprintf(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++) {
- struct file_struct *file = flist->files[i];
- if (!file->basename || !S_ISDIR(file->mode)) continue;
- recv_generator(f_name(file),flist,i,-1);
- }
-
- if (verbose > 2)
- fprintf(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;
- int offset=0;
-
- if (verbose > 2)
- fprintf(FINFO,"send_files starting\n");
-
- setup_nonblocking(f_in,f_out);
-
- while (1) {
- i = read_int(f_in);
- if (i == -1) {
- if (phase==0 && remote_version >= 13) {
- phase++;
- csum_length = SUM_LENGTH;
- write_int(f_out,-1);
- write_flush(f_out);
- if (verbose > 2)
- fprintf(FINFO,"send_files phase=%d\n",phase);
- continue;
- }
- break;
- }
-
- file = flist->files[i];
-
- fname[0] = 0;
- if (file->basedir) {
- strncpy(fname,file->basedir,MAXPATHLEN-1);
- fname[MAXPATHLEN-1] = 0;
- if (strlen(fname) == MAXPATHLEN-1) {
- io_error = 1;
- fprintf(FERROR, "send_files failed on long-named directory %s\n",
- fname);
- return;
- }
- strcat(fname,"/");
- offset = strlen(file->basedir)+1;
- }
- strncat(fname,f_name(file),MAXPATHLEN-strlen(fname));
-
- if (verbose > 2)
- fprintf(FINFO,"send_files(%d,%s)\n",i,fname);
-
- if (dry_run) {
- if (!am_server && verbose)
- printf("%s\n",fname);
- write_int(f_out,i);
- continue;
- }
-
- s = receive_sums(f_in);
- if (!s) {
- io_error = 1;
- fprintf(FERROR,"receive_sums failed\n");
- return;
- }
-
- fd = open(fname,O_RDONLY);
- if (fd == -1) {
- io_error = 1;
- fprintf(FERROR,"send_files failed to open %s: %s\n",
- fname,strerror(errno));
- free_sums(s);
- continue;
- }
-
- /* map the local file */
- if (fstat(fd,&st) != 0) {
- io_error = 1;
- fprintf(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)
- fprintf(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)
- fprintf(FINFO,"calling match_sums %s\n",fname);
-
- if (!am_server && verbose)
- printf("%s\n",fname+offset);
-
- match_sums(f_out,s,buf,st.st_size);
- write_flush(f_out);
-
- if (buf) unmap_file(buf);
- close(fd);
-
- free_sums(s);
-
- if (verbose > 2)
- fprintf(FINFO,"sender finished %s\n",fname);
- }
-
- if (verbose > 2)
- fprintf(FINFO,"send files finished\n");
-
- match_report();
-
- write_int(f_out,-1);
- write_flush(f_out);
-}
-
-
-
-void generate_files(int f,struct file_list *flist,char *local_name,int f_recv)
-{
- int i;
- int phase=0;
-
- if (verbose > 2)
- fprintf(FINFO,"generator starting pid=%d count=%d\n",
- (int)getpid(),flist->count);
-
- for (i = 0; i < flist->count; i++) {
- struct file_struct *file = flist->files[i];
- mode_t saved_mode = file->mode;
- if (!file->basename) continue;
-
- /* we need to ensure that any directories we create have writeable
- permissions initially so that we can create the files within
- them. This is then fixed after the files are transferred */
- if (!am_root && S_ISDIR(file->mode)) {
- file->mode |= S_IWUSR; /* user write */
- }
-
- recv_generator(local_name?local_name:f_name(file),
- flist,i,f);
-
- file->mode = saved_mode;
- }
-
- phase++;
- csum_length = SUM_LENGTH;
- ignore_times=1;
-
- if (verbose > 2)
- fprintf(FINFO,"generate_files phase=%d\n",phase);
-
- write_int(f,-1);
- write_flush(f);
-
- /* we expect to just sit around now, so don't exit on a timeout. If we
- really get a timeout then the other process should exit */
- io_timeout = 0;
-
- if (remote_version >= 13) {
- /* in newer versions of the protocol the files can cycle through
- the system more than once to catch initial checksum errors */
- for (i=read_int(f_recv); i != -1; i=read_int(f_recv)) {
- struct file_struct *file = flist->files[i];
- recv_generator(local_name?local_name:f_name(file),
- flist,i,f);
- }
-
- phase++;
- if (verbose > 2)
- fprintf(FINFO,"generate_files phase=%d\n",phase);
-
- write_int(f,-1);
- write_flush(f);
- }
-
-
- if (verbose > 2)
- fprintf(FINFO,"generator wrote %ld\n",(long)write_total());
-}
-
-