Moved GID_NONE define from rsync.h to uidlist.c.
[rsync/rsync.git] / receiver.c
1 /*
2  * Routines only used by the receiving process.
3  *
4  * Copyright (C) 1996-2000 Andrew Tridgell
5  * Copyright (C) 1996 Paul Mackerras
6  * Copyright (C) 2003-2007 Wayne Davison
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License version 2 as
10  * published by the Free Software Foundation.
11  *
12  * This program is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  * GNU General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License along
18  * with this program; if not, write to the Free Software Foundation, Inc.,
19  * 51 Franklin Street - Fifth Floor, Boston, MA 02110-1301, USA.
20  */
21
22 #include "rsync.h"
23
24 extern int verbose;
25 extern int dry_run;
26 extern int do_xfers;
27 extern int am_server;
28 extern int do_progress;
29 extern int inc_recurse;
30 extern int log_before_transfer;
31 extern int stdout_format_has_i;
32 extern int logfile_format_has_i;
33 extern int csum_length;
34 extern int read_batch;
35 extern int write_batch;
36 extern int batch_gen_fd;
37 extern int protocol_version;
38 extern int relative_paths;
39 extern int preserve_hard_links;
40 extern int preserve_perms;
41 extern int preserve_xattrs;
42 extern int basis_dir_cnt;
43 extern int make_backups;
44 extern int cleanup_got_literal;
45 extern int remove_source_files;
46 extern int append_mode;
47 extern int sparse_files;
48 extern int keep_partial;
49 extern int checksum_seed;
50 extern int inplace;
51 extern int delay_updates;
52 extern mode_t orig_umask;
53 extern struct stats stats;
54 extern char *tmpdir;
55 extern char *partial_dir;
56 extern char *basis_dir[];
57 extern struct file_list *cur_flist, *first_flist;
58 extern struct filter_list_struct server_filter_list;
59
60 static struct bitbag *delayed_bits = NULL;
61 static int phase = 0, redoing = 0;
62 /* We're either updating the basis file or an identical copy: */
63 static int updating_basis;
64
65 /*
66  * get_tmpname() - create a tmp filename for a given filename
67  *
68  *   If a tmpdir is defined, use that as the directory to
69  *   put it in.  Otherwise, the tmp filename is in the same
70  *   directory as the given name.  Note that there may be no
71  *   directory at all in the given name!
72  *
73  *   The tmp filename is basically the given filename with a
74  *   dot prepended, and .XXXXXX appended (for mkstemp() to
75  *   put its unique gunk in).  Take care to not exceed
76  *   either the MAXPATHLEN or NAME_MAX, esp. the last, as
77  *   the basename basically becomes 8 chars longer. In that
78  *   case, the original name is shortened sufficiently to
79  *   make it all fit.
80  *
81  *   Of course, there's no real reason for the tmp name to
82  *   look like the original, except to satisfy us humans.
83  *   As long as it's unique, rsync will work.
84  */
85
86 int get_tmpname(char *fnametmp, char *fname)
87 {
88         int maxname, added, length = 0;
89         char *f;
90
91         if (tmpdir) {
92                 /* Note: this can't overflow, so the return value is safe */
93                 length = strlcpy(fnametmp, tmpdir, MAXPATHLEN - 2);
94                 fnametmp[length++] = '/';
95         }
96
97         if ((f = strrchr(fname, '/')) != NULL) {
98                 ++f;
99                 if (!tmpdir) {
100                         length = f - fname;
101                         /* copy up to and including the slash */
102                         strlcpy(fnametmp, fname, length + 1);
103                 }
104         } else
105                 f = fname;
106         fnametmp[length++] = '.';
107
108         /* The maxname value is bufsize, and includes space for the '\0'.
109          * (Note that NAME_MAX get -8 for the leading '.' above.) */
110         maxname = MIN(MAXPATHLEN - 7 - length, NAME_MAX - 8);
111
112         if (maxname < 1) {
113                 rprintf(FERROR, "temporary filename too long: %s\n", fname);
114                 fnametmp[0] = '\0';
115                 return 0;
116         }
117
118         added = strlcpy(fnametmp + length, f, maxname);
119         if (added >= maxname)
120                 added = maxname - 1;
121         memcpy(fnametmp + length + added, ".XXXXXX", 8);
122
123         return 1;
124 }
125
126
127 static int receive_data(int f_in, char *fname_r, int fd_r, OFF_T size_r,
128                         const char *fname, int fd, OFF_T total_size)
129 {
130         static char file_sum1[MAX_DIGEST_LEN];
131         static char file_sum2[MAX_DIGEST_LEN];
132         struct map_struct *mapbuf;
133         struct sum_struct sum;
134         int32 len, sum_len;
135         OFF_T offset = 0;
136         OFF_T offset2;
137         char *data;
138         int32 i;
139         char *map = NULL;
140
141         read_sum_head(f_in, &sum);
142
143         if (fd_r >= 0 && size_r > 0) {
144                 int32 read_size = MAX(sum.blength * 2, 16*1024);
145                 mapbuf = map_file(fd_r, size_r, read_size, sum.blength);
146                 if (verbose > 2) {
147                         rprintf(FINFO, "recv mapped %s of size %.0f\n",
148                                 fname_r, (double)size_r);
149                 }
150         } else
151                 mapbuf = NULL;
152
153         sum_init(checksum_seed);
154
155         if (append_mode > 0) {
156                 OFF_T j;
157                 sum.flength = (OFF_T)sum.count * sum.blength;
158                 if (sum.remainder)
159                         sum.flength -= sum.blength - sum.remainder;
160                 for (j = CHUNK_SIZE; j < sum.flength; j += CHUNK_SIZE) {
161                         if (do_progress)
162                                 show_progress(offset, total_size);
163                         sum_update(map_ptr(mapbuf, offset, CHUNK_SIZE),
164                                    CHUNK_SIZE);
165                         offset = j;
166                 }
167                 if (offset < sum.flength) {
168                         int32 len = (int32)(sum.flength - offset);
169                         if (do_progress)
170                                 show_progress(offset, total_size);
171                         sum_update(map_ptr(mapbuf, offset, len), len);
172                         offset = sum.flength;
173                 }
174                 if (fd != -1 && (j = do_lseek(fd, offset, SEEK_SET)) != offset) {
175                         rsyserr(FERROR, errno, "lseek of %s returned %.0f, not %.0f",
176                                 full_fname(fname), (double)j, (double)offset);
177                         exit_cleanup(RERR_FILEIO);
178                 }
179         }
180
181         while ((i = recv_token(f_in, &data)) != 0) {
182                 if (do_progress)
183                         show_progress(offset, total_size);
184
185                 if (i > 0) {
186                         if (verbose > 3) {
187                                 rprintf(FINFO,"data recv %d at %.0f\n",
188                                         i,(double)offset);
189                         }
190
191                         stats.literal_data += i;
192                         cleanup_got_literal = 1;
193
194                         sum_update(data, i);
195
196                         if (fd != -1 && write_file(fd,data,i) != i)
197                                 goto report_write_error;
198                         offset += i;
199                         continue;
200                 }
201
202                 i = -(i+1);
203                 offset2 = i * (OFF_T)sum.blength;
204                 len = sum.blength;
205                 if (i == (int)sum.count-1 && sum.remainder != 0)
206                         len = sum.remainder;
207
208                 stats.matched_data += len;
209
210                 if (verbose > 3) {
211                         rprintf(FINFO,
212                                 "chunk[%d] of size %ld at %.0f offset=%.0f\n",
213                                 i, (long)len, (double)offset2, (double)offset);
214                 }
215
216                 if (mapbuf) {
217                         map = map_ptr(mapbuf,offset2,len);
218
219                         see_token(map, len);
220                         sum_update(map, len);
221                 }
222
223                 if (updating_basis) {
224                         if (offset == offset2 && fd != -1) {
225                                 OFF_T pos;
226                                 if (flush_write_file(fd) < 0)
227                                         goto report_write_error;
228                                 offset += len;
229                                 if ((pos = do_lseek(fd, len, SEEK_CUR)) != offset) {
230                                         rsyserr(FERROR, errno,
231                                                 "lseek of %s returned %.0f, not %.0f",
232                                                 full_fname(fname),
233                                                 (double)pos, (double)offset);
234                                         exit_cleanup(RERR_FILEIO);
235                                 }
236                                 continue;
237                         }
238                 }
239                 if (fd != -1 && map && write_file(fd, map, len) != (int)len)
240                         goto report_write_error;
241                 offset += len;
242         }
243
244         if (flush_write_file(fd) < 0)
245                 goto report_write_error;
246
247 #ifdef HAVE_FTRUNCATE
248         if (inplace && fd != -1)
249                 ftruncate(fd, offset);
250 #endif
251
252         if (do_progress)
253                 end_progress(total_size);
254
255         if (fd != -1 && offset > 0 && sparse_end(fd) != 0) {
256             report_write_error:
257                 rsyserr(FERROR, errno, "write failed on %s",
258                         full_fname(fname));
259                 exit_cleanup(RERR_FILEIO);
260         }
261
262         sum_len = sum_end(file_sum1);
263
264         if (mapbuf)
265                 unmap_file(mapbuf);
266
267         read_buf(f_in, file_sum2, sum_len);
268         if (verbose > 2)
269                 rprintf(FINFO,"got file_sum\n");
270         if (fd != -1 && memcmp(file_sum1, file_sum2, sum_len) != 0)
271                 return 0;
272         return 1;
273 }
274
275
276 static void discard_receive_data(int f_in, OFF_T length)
277 {
278         receive_data(f_in, NULL, -1, 0, NULL, -1, length);
279 }
280
281 static void handle_delayed_updates(char *local_name)
282 {
283         char *fname, *partialptr;
284         int ndx;
285
286         for (ndx = -1; (ndx = bitbag_next_bit(delayed_bits, ndx)) >= 0; ) {
287                 struct file_struct *file = cur_flist->files[ndx];
288                 fname = local_name ? local_name : f_name(file, NULL);
289                 if ((partialptr = partial_dir_fname(fname)) != NULL) {
290                         if (make_backups > 0 && !make_backup(fname))
291                                 continue;
292                         if (verbose > 2) {
293                                 rprintf(FINFO, "renaming %s to %s\n",
294                                         partialptr, fname);
295                         }
296                         /* We don't use robust_rename() here because the
297                          * partial-dir must be on the same drive. */
298                         if (do_rename(partialptr, fname) < 0) {
299                                 rsyserr(FERROR, errno,
300                                         "rename failed for %s (from %s)",
301                                         full_fname(fname), partialptr);
302                         } else {
303                                 if (remove_source_files
304                                  || (preserve_hard_links && F_IS_HLINKED(file)))
305                                         send_msg_int(MSG_SUCCESS, ndx);
306                                 handle_partial_dir(partialptr, PDIR_DELETE);
307                         }
308                 }
309         }
310 }
311
312 static int get_next_gen_ndx(int fd, int next_gen_ndx, int desired_ndx)
313 {
314         while (next_gen_ndx < desired_ndx) {
315                 if (next_gen_ndx >= 0) {
316                         rprintf(FINFO,
317                                 "(No batched update for%s \"%s\")\n",
318                                 redoing ? " resend of" : "",
319                                 f_name(cur_flist->files[next_gen_ndx], NULL));
320                 }
321                 next_gen_ndx = read_int(fd);
322                 if (next_gen_ndx == -1) {
323                         if (inc_recurse)
324                                 next_gen_ndx = first_flist->prev->count + first_flist->prev->ndx_start;
325                         else
326                                 next_gen_ndx = cur_flist->count;
327                 }
328         }
329         return next_gen_ndx;
330 }
331
332 /**
333  * main routine for receiver process.
334  *
335  * Receiver process runs on the same host as the generator process. */
336 int recv_files(int f_in, char *local_name)
337 {
338         int next_gen_ndx = -1;
339         int fd1,fd2;
340         STRUCT_STAT st;
341         int iflags, xlen;
342         char *fname, fbuf[MAXPATHLEN];
343         char xname[MAXPATHLEN];
344         char fnametmp[MAXPATHLEN];
345         char *fnamecmp, *partialptr;
346         char fnamecmpbuf[MAXPATHLEN];
347         uchar fnamecmp_type;
348         struct file_struct *file;
349         struct stats initial_stats;
350         int itemizing = am_server ? logfile_format_has_i : stdout_format_has_i;
351         enum logcode log_code = log_before_transfer ? FLOG : FINFO;
352         int max_phase = protocol_version >= 29 ? 2 : 1;
353         int dflt_perms = (ACCESSPERMS & ~orig_umask);
354 #ifdef SUPPORT_ACLS
355         const char *parent_dirname = "";
356 #endif
357         int ndx, recv_ok;
358
359         if (verbose > 2)
360                 rprintf(FINFO, "recv_files(%d) starting\n", cur_flist->count);
361
362         if (delay_updates)
363                 delayed_bits = bitbag_create(cur_flist->count + 1);
364
365         updating_basis = inplace;
366
367         while (1) {
368                 cleanup_disable();
369
370                 /* This call also sets cur_flist. */
371                 ndx = read_ndx_and_attrs(f_in, &iflags, &fnamecmp_type,
372                                          xname, &xlen);
373                 if (ndx == NDX_DONE) {
374                         if (inc_recurse && first_flist) {
375                                 flist_free(first_flist);
376                                 if (first_flist)
377                                         continue;
378                         }
379                         if (read_batch && cur_flist) {
380                                 int high = inc_recurse
381                                     ? first_flist->prev->count + first_flist->prev->ndx_start
382                                     : cur_flist->count;
383                                 get_next_gen_ndx(batch_gen_fd, next_gen_ndx, high);
384                                 next_gen_ndx = -1;
385                         }
386                         if (++phase > max_phase)
387                                 break;
388                         if (verbose > 2)
389                                 rprintf(FINFO, "recv_files phase=%d\n", phase);
390                         if (phase == 2 && delay_updates)
391                                 handle_delayed_updates(local_name);
392                         send_msg(MSG_DONE, "", 0, 0);
393                         continue;
394                 }
395
396                 file = cur_flist->files[ndx - cur_flist->ndx_start];
397                 fname = local_name ? local_name : f_name(file, fbuf);
398
399                 if (verbose > 2)
400                         rprintf(FINFO, "recv_files(%s)\n", fname);
401
402 #ifdef SUPPORT_XATTRS
403                 if (iflags & ITEM_REPORT_XATTR && !dry_run)
404                         recv_xattr_request(file, f_in);
405 #endif
406
407                 if (!(iflags & ITEM_TRANSFER)) {
408                         maybe_log_item(file, iflags, itemizing, xname);
409 #ifdef SUPPORT_XATTRS
410                         if (preserve_xattrs && iflags & ITEM_REPORT_XATTR && !dry_run)
411                                 set_file_attrs(fname, file, NULL, fname, 0);
412 #endif
413                         continue;
414                 }
415                 if (phase == 2) {
416                         rprintf(FERROR,
417                                 "got transfer request in phase 2 [%s]\n",
418                                 who_am_i());
419                         exit_cleanup(RERR_PROTOCOL);
420                 }
421
422                 if (file->flags & FLAG_FILE_SENT) {
423                         if (csum_length == SHORT_SUM_LENGTH) {
424                                 if (keep_partial && !partial_dir)
425                                         make_backups = -make_backups; /* prevents double backup */
426                                 append_mode = -append_mode;
427                                 sparse_files = -sparse_files;
428                                 csum_length = SUM_LENGTH;
429                                 redoing = 1;
430                         }
431                 } else {
432                         if (csum_length != SHORT_SUM_LENGTH) {
433                                 if (keep_partial && !partial_dir)
434                                         make_backups = -make_backups;
435                                 append_mode = -append_mode;
436                                 sparse_files = -sparse_files;
437                                 csum_length = SHORT_SUM_LENGTH;
438                                 redoing = 0;
439                         }
440                 }
441
442                 stats.current_file_index = ndx;
443                 stats.num_transferred_files++;
444                 stats.total_transferred_size += F_LENGTH(file);
445                 cleanup_got_literal = 0;
446
447                 if (server_filter_list.head
448                     && check_filter(&server_filter_list, fname, 0) < 0) {
449                         rprintf(FERROR, "attempt to hack rsync failed.\n");
450                         exit_cleanup(RERR_PROTOCOL);
451                 }
452
453                 if (!do_xfers) { /* log the transfer */
454                         log_item(FCLIENT, file, &stats, iflags, NULL);
455                         if (read_batch)
456                                 discard_receive_data(f_in, F_LENGTH(file));
457                         continue;
458                 }
459                 if (write_batch < 0) {
460                         log_item(FINFO, file, &stats, iflags, NULL);
461                         if (!am_server)
462                                 discard_receive_data(f_in, F_LENGTH(file));
463                         continue;
464                 }
465
466                 if (read_batch) {
467                         next_gen_ndx = get_next_gen_ndx(batch_gen_fd, next_gen_ndx, ndx);
468                         if (ndx < next_gen_ndx) {
469                                 rprintf(FINFO,
470                                         "(Skipping batched update for \"%s\")\n",
471                                         fname);
472                                 discard_receive_data(f_in, F_LENGTH(file));
473                                 if (inc_recurse)
474                                         send_msg_int(MSG_NO_SEND, ndx);
475                                 continue;
476                         }
477                         next_gen_ndx = -1;
478                 }
479
480                 partialptr = partial_dir ? partial_dir_fname(fname) : fname;
481
482                 if (protocol_version >= 29) {
483                         switch (fnamecmp_type) {
484                         case FNAMECMP_FNAME:
485                                 fnamecmp = fname;
486                                 break;
487                         case FNAMECMP_PARTIAL_DIR:
488                                 fnamecmp = partialptr;
489                                 break;
490                         case FNAMECMP_BACKUP:
491                                 fnamecmp = get_backup_name(fname);
492                                 break;
493                         case FNAMECMP_FUZZY:
494                                 updating_basis = 0;
495                                 if (file->dirname) {
496                                         pathjoin(fnamecmpbuf, MAXPATHLEN,
497                                                  file->dirname, xname);
498                                         fnamecmp = fnamecmpbuf;
499                                 } else
500                                         fnamecmp = xname;
501                                 break;
502                         default:
503                                 updating_basis = 0;
504                                 if (fnamecmp_type >= basis_dir_cnt) {
505                                         rprintf(FERROR,
506                                                 "invalid basis_dir index: %d.\n",
507                                                 fnamecmp_type);
508                                         exit_cleanup(RERR_PROTOCOL);
509                                 }
510                                 pathjoin(fnamecmpbuf, sizeof fnamecmpbuf,
511                                          basis_dir[fnamecmp_type], fname);
512                                 fnamecmp = fnamecmpbuf;
513                                 break;
514                         }
515                         if (!fnamecmp || (server_filter_list.head
516                           && check_filter(&server_filter_list, fname, 0) < 0))
517                                 fnamecmp = fname;
518                 } else {
519                         /* Reminder: --inplace && --partial-dir are never
520                          * enabled at the same time. */
521                         if (inplace && make_backups > 0) {
522                                 if (!(fnamecmp = get_backup_name(fname)))
523                                         fnamecmp = fname;
524                         } else if (partial_dir && partialptr)
525                                 fnamecmp = partialptr;
526                         else
527                                 fnamecmp = fname;
528                 }
529
530                 initial_stats = stats;
531
532                 /* open the file */
533                 fd1 = do_open(fnamecmp, O_RDONLY, 0);
534
535                 if (fd1 == -1 && protocol_version < 29) {
536                         if (fnamecmp != fname) {
537                                 fnamecmp = fname;
538                                 fd1 = do_open(fnamecmp, O_RDONLY, 0);
539                         }
540
541                         if (fd1 == -1 && basis_dir[0]) {
542                                 /* pre-29 allowed only one alternate basis */
543                                 pathjoin(fnamecmpbuf, sizeof fnamecmpbuf,
544                                          basis_dir[0], fname);
545                                 fnamecmp = fnamecmpbuf;
546                                 fd1 = do_open(fnamecmp, O_RDONLY, 0);
547                         }
548                 }
549
550                 if (fd1 == -1) {
551                         st.st_mode = 0;
552                         st.st_size = 0;
553                 } else if (do_fstat(fd1,&st) != 0) {
554                         rsyserr(FERROR, errno, "fstat %s failed",
555                                 full_fname(fnamecmp));
556                         discard_receive_data(f_in, F_LENGTH(file));
557                         close(fd1);
558                         if (inc_recurse)
559                                 send_msg_int(MSG_NO_SEND, ndx);
560                         continue;
561                 }
562
563                 if (fd1 != -1 && S_ISDIR(st.st_mode) && fnamecmp == fname) {
564                         /* this special handling for directories
565                          * wouldn't be necessary if robust_rename()
566                          * and the underlying robust_unlink could cope
567                          * with directories
568                          */
569                         rprintf(FERROR,"recv_files: %s is a directory\n",
570                                 full_fname(fnamecmp));
571                         discard_receive_data(f_in, F_LENGTH(file));
572                         close(fd1);
573                         if (inc_recurse)
574                                 send_msg_int(MSG_NO_SEND, ndx);
575                         continue;
576                 }
577
578                 if (fd1 != -1 && !S_ISREG(st.st_mode)) {
579                         close(fd1);
580                         fd1 = -1;
581                 }
582
583                 /* If we're not preserving permissions, change the file-list's
584                  * mode based on the local permissions and some heuristics. */
585                 if (!preserve_perms) {
586                         int exists = fd1 != -1;
587 #ifdef SUPPORT_ACLS
588                         const char *dn = file->dirname ? file->dirname : ".";
589                         if (parent_dirname != dn
590                          && strcmp(parent_dirname, dn) != 0) {
591                                 dflt_perms = default_perms_for_dir(dn);
592                                 parent_dirname = dn;
593                         }
594 #endif
595                         file->mode = dest_mode(file->mode, st.st_mode,
596                                                dflt_perms, exists);
597                 }
598
599                 /* We now check to see if we are writing the file "inplace" */
600                 if (inplace)  {
601                         fd2 = do_open(fname, O_WRONLY|O_CREAT, 0600);
602                         if (fd2 == -1) {
603                                 rsyserr(FERROR, errno, "open %s failed",
604                                         full_fname(fname));
605                                 discard_receive_data(f_in, F_LENGTH(file));
606                                 if (fd1 != -1)
607                                         close(fd1);
608                                 if (inc_recurse)
609                                         send_msg_int(MSG_NO_SEND, ndx);
610                                 continue;
611                         }
612                 } else {
613                         if (!get_tmpname(fnametmp,fname)) {
614                                 discard_receive_data(f_in, F_LENGTH(file));
615                                 if (fd1 != -1)
616                                         close(fd1);
617                                 if (inc_recurse)
618                                         send_msg_int(MSG_NO_SEND, ndx);
619                                 continue;
620                         }
621
622                         /* we initially set the perms without the
623                          * setuid/setgid bits to ensure that there is no race
624                          * condition. They are then correctly updated after
625                          * the lchown. Thanks to snabb@epipe.fi for pointing
626                          * this out.  We also set it initially without group
627                          * access because of a similar race condition. */
628                         fd2 = do_mkstemp(fnametmp, file->mode & INITACCESSPERMS);
629
630                         /* in most cases parent directories will already exist
631                          * because their information should have been previously
632                          * transferred, but that may not be the case with -R */
633                         if (fd2 == -1 && relative_paths && errno == ENOENT
634                             && create_directory_path(fnametmp) == 0) {
635                                 /* Get back to name with XXXXXX in it. */
636                                 get_tmpname(fnametmp, fname);
637                                 fd2 = do_mkstemp(fnametmp, file->mode & INITACCESSPERMS);
638                         }
639                         if (fd2 == -1) {
640                                 rsyserr(FERROR, errno, "mkstemp %s failed",
641                                         full_fname(fnametmp));
642                                 discard_receive_data(f_in, F_LENGTH(file));
643                                 if (fd1 != -1)
644                                         close(fd1);
645                                 if (inc_recurse)
646                                         send_msg_int(MSG_NO_SEND, ndx);
647                                 continue;
648                         }
649
650                         cleanup_set(fnametmp, partialptr, file, fd1, fd2);
651                 }
652
653                 /* log the transfer */
654                 if (log_before_transfer)
655                         log_item(FCLIENT, file, &initial_stats, iflags, NULL);
656                 else if (!am_server && verbose && do_progress)
657                         rprintf(FINFO, "%s\n", fname);
658
659                 /* recv file data */
660                 recv_ok = receive_data(f_in, fnamecmp, fd1, st.st_size,
661                                        fname, fd2, F_LENGTH(file));
662
663                 log_item(log_code, file, &initial_stats, iflags, NULL);
664
665                 if (fd1 != -1)
666                         close(fd1);
667                 if (close(fd2) < 0) {
668                         rsyserr(FERROR, errno, "close failed on %s",
669                                 full_fname(fnametmp));
670                         exit_cleanup(RERR_FILEIO);
671                 }
672
673                 if ((recv_ok && (!delay_updates || !partialptr)) || inplace) {
674                         char *temp_copy_name;
675                         if (partialptr == fname)
676                                 partialptr = temp_copy_name = NULL;
677                         else if (*partial_dir == '/')
678                                 temp_copy_name = NULL;
679                         else
680                                 temp_copy_name = partialptr;
681                         finish_transfer(fname, fnametmp, fnamecmp,
682                                         temp_copy_name, file, recv_ok, 1);
683                         if (fnamecmp == partialptr) {
684                                 do_unlink(partialptr);
685                                 handle_partial_dir(partialptr, PDIR_DELETE);
686                         }
687                 } else if (keep_partial && partialptr
688                     && handle_partial_dir(partialptr, PDIR_CREATE)) {
689                         finish_transfer(partialptr, fnametmp, fnamecmp, NULL,
690                                         file, recv_ok, !partial_dir);
691                         if (delay_updates && recv_ok) {
692                                 bitbag_set_bit(delayed_bits, ndx);
693                                 recv_ok = -1;
694                         }
695                 } else {
696                         partialptr = NULL;
697                         do_unlink(fnametmp);
698                 }
699
700                 cleanup_disable();
701
702                 if (recv_ok > 0) {
703                         if (remove_source_files || inc_recurse
704                          || (preserve_hard_links && F_IS_HLINKED(file)))
705                                 send_msg_int(MSG_SUCCESS, ndx);
706                 } else if (!recv_ok) {
707                         enum logcode msgtype = redoing || read_batch ? FERROR : FINFO;
708                         if (msgtype == FERROR || verbose) {
709                                 char *errstr, *redostr, *keptstr;
710                                 if (!(keep_partial && partialptr) && !inplace)
711                                         keptstr = "discarded";
712                                 else if (partial_dir)
713                                         keptstr = "put into partial-dir";
714                                 else
715                                         keptstr = "retained";
716                                 if (msgtype == FERROR) {
717                                         errstr = "ERROR";
718                                         redostr = "";
719                                 } else {
720                                         errstr = "WARNING";
721                                         redostr = " (will try again)";
722                                 }
723                                 rprintf(msgtype,
724                                         "%s: %s failed verification -- update %s%s.\n",
725                                         errstr, fname, keptstr, redostr);
726                         }
727                         if (!redoing) {
728                                 send_msg_int(MSG_REDO, ndx);
729                                 file->flags |= FLAG_FILE_SENT;
730                         } else if (inc_recurse)
731                                 send_msg_int(MSG_NO_SEND, ndx);
732                 }
733         }
734         if (make_backups < 0)
735                 make_backups = -make_backups;
736
737         if (phase == 2 && delay_updates) /* for protocol_version < 29 */
738                 handle_delayed_updates(local_name);
739
740         if (verbose > 2)
741                 rprintf(FINFO,"recv_files finished\n");
742
743         return 0;
744 }