Fixed the creation of symlinks and devices when both --relative
[rsync/rsync.git] / generator.c
CommitLineData
ef1aa910 1/* -*- c-file-style: "linux" -*-
91262d5d
MP
2
3 rsync -- fast file replication program
2cda2560
WD
4
5 Copyright (C) 1996-2000 by Andrew Tridgell
2f03f956 6 Copyright (C) Paul Mackerras 1996
91262d5d 7 Copyright (C) 2002 by Martin Pool <mbp@samba.org>
2cda2560 8
2f03f956
AT
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 2 of the License, or
12 (at your option) any later version.
2cda2560 13
2f03f956
AT
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
2cda2560 18
2f03f956
AT
19 You should have received a copy of the GNU General Public License
20 along with this program; if not, write to the Free Software
21 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22*/
23
24#include "rsync.h"
25
26extern int verbose;
27extern int dry_run;
beb51aa0 28extern int do_xfers;
6c3862fa
WD
29extern int log_format_has_i;
30extern int log_format_has_o_or_i;
31extern int daemon_log_format_has_i;
32extern int am_root;
33extern int am_server;
34extern int am_daemon;
083acd49 35extern int do_progress;
59faec8b 36extern int recurse;
2f03f956 37extern int relative_paths;
ccb16b46 38extern int implied_dirs;
716e73d4 39extern int keep_dirlinks;
2f03f956 40extern int preserve_links;
2f03f956 41extern int preserve_devices;
b5c6a6ae 42extern int preserve_specials;
2f03f956 43extern int preserve_hard_links;
6744b62d
WD
44extern int preserve_perms;
45extern int preserve_uid;
46extern int preserve_gid;
3ea9bbd6
WD
47extern int preserve_times;
48extern int omit_dir_times;
e35d9f2d 49extern int delete_mode;
59faec8b 50extern int delete_before;
fa13f396 51extern int delete_during;
59faec8b
WD
52extern int delete_after;
53extern int module_id;
54extern int ignore_errors;
fe960187 55extern int remove_sent_files;
70352269 56extern int delay_updates;
2f03f956 57extern int update_only;
e90aab49
WD
58extern int ignore_existing;
59extern int ignore_non_existing;
cd6aa5b5 60extern int inplace;
6cc11982 61extern int append_mode;
cd6aa5b5 62extern int make_backups;
2f03f956
AT
63extern int csum_length;
64extern int ignore_times;
f83f0548 65extern int size_only;
7d1bfaf7 66extern OFF_T max_size;
02b5cb23 67extern OFF_T min_size;
59faec8b 68extern int io_error;
ac40b747 69extern int allowed_lull;
ee1d11c4 70extern int sock_f_out;
9ac2395b 71extern int ignore_timeout;
d04e9c51 72extern int protocol_version;
8e85be0a 73extern int fuzzy_basis;
2f03f956 74extern int always_checksum;
cfe39780 75extern int checksum_len;
a7260c40 76extern char *partial_dir;
b7e8628c 77extern char *basis_dir[];
2be2fb3e 78extern int compare_dest;
967866d4 79extern int copy_dest;
59c95e42 80extern int link_dest;
5774786f 81extern int whole_file;
5774786f 82extern int list_only;
b9f592fb 83extern int read_batch;
5774786f
WD
84extern int orig_umask;
85extern int safe_symlinks;
a255c592 86extern long block_size; /* "long" because popt can't set an int32. */
59faec8b
WD
87extern int max_delete;
88extern int force_delete;
0e82af2d 89extern int one_file_system;
6c3862fa 90extern struct stats stats;
0e82af2d 91extern dev_t filesystem_dev;
59faec8b
WD
92extern char *backup_dir;
93extern char *backup_suffix;
94extern int backup_suffix_len;
ee1d11c4 95extern struct file_list *the_file_list;
7842418b 96extern struct filter_list_struct server_filter_list;
97f9dcae 97
59faec8b 98static int deletion_count = 0; /* used to implement --max-delete */
48224e4c
WD
99static int can_link_symlinks = 1; /* start out optimistic */
100static int can_link_devices = 1;
59faec8b 101
d71dad3b
WD
102/* For calling delete_file() */
103#define DEL_FORCE_RECURSE (1<<1) /* recurse even w/o --force */
104#define DEL_TERSE (1<<3)
105
59faec8b
WD
106
107static int is_backup_file(char *fn)
108{
109 int k = strlen(fn) - backup_suffix_len;
110 return k > 0 && strcmp(fn+k, backup_suffix) == 0;
111}
112
113
114/* Delete a file or directory. If DEL_FORCE_RECURSE is set in the flags, or if
d71dad3b 115 * force_delete is set, this will delete recursively.
0e5665d3
WD
116 *
117 * Note that fname must point to a MAXPATHLEN buffer if the mode indicates it's
118 * a directory! (The buffer is used for recursion, but returned unchanged.)
119 */
59faec8b
WD
120static int delete_item(char *fname, int mode, int flags)
121{
122 struct file_list *dirlist;
59faec8b 123 int j, dlen, zap_dir, ok;
0e5665d3 124 unsigned remainder;
59faec8b 125 void *save_filters;
0e5665d3 126 char *p;
59faec8b 127
59faec8b 128 if (!S_ISDIR(mode)) {
f75a53e7
WD
129 if (max_delete && ++deletion_count > max_delete)
130 return 0;
59faec8b
WD
131 if (make_backups && (backup_dir || !is_backup_file(fname)))
132 ok = make_backup(fname);
133 else
134 ok = robust_unlink(fname) == 0;
135 if (ok) {
136 if (!(flags & DEL_TERSE))
137 log_delete(fname, mode);
59faec8b
WD
138 return 0;
139 }
f75a53e7
WD
140 if (errno == ENOENT) {
141 deletion_count--;
59faec8b 142 return 0;
f75a53e7 143 }
59faec8b
WD
144 rsyserr(FERROR, errno, "delete_file: unlink %s failed",
145 full_fname(fname));
146 return -1;
147 }
148
4d167803 149 zap_dir = flags & DEL_FORCE_RECURSE || force_delete;
f75a53e7
WD
150 if ((max_delete && ++deletion_count > max_delete)
151 || (dry_run && zap_dir)) {
59faec8b
WD
152 ok = 0;
153 errno = ENOTEMPTY;
154 } else if (make_backups && !backup_dir && !is_backup_file(fname)
155 && !(flags & DEL_FORCE_RECURSE))
156 ok = make_backup(fname);
157 else
158 ok = do_rmdir(fname) == 0;
159 if (ok) {
160 if (!(flags & DEL_TERSE))
161 log_delete(fname, mode);
59faec8b
WD
162 return 0;
163 }
f75a53e7
WD
164 if (errno == ENOENT) {
165 deletion_count--;
59faec8b 166 return 0;
f75a53e7 167 }
26beb786 168 if (!zap_dir) {
59faec8b
WD
169 rsyserr(FERROR, errno, "delete_file: rmdir %s failed",
170 full_fname(fname));
171 return -1;
172 }
f62eaa24 173 flags |= DEL_FORCE_RECURSE; /* mark subdir dels as not "in the way" */
f75a53e7 174 deletion_count--;
59faec8b 175
0e5665d3
WD
176 dlen = strlen(fname);
177 save_filters = push_local_filters(fname, dlen);
178
179 dirlist = get_dirlist(fname, dlen, 0);
180
181 p = fname + dlen;
182 if (dlen != 1 || *fname != '/')
183 *p++ = '/';
184 remainder = MAXPATHLEN - (p - fname);
59faec8b 185
59faec8b
WD
186 for (j = dirlist->count; j--; ) {
187 struct file_struct *fp = dirlist->files[j];
188
189 if (fp->flags & FLAG_MOUNT_POINT)
190 continue;
191
0e5665d3
WD
192 strlcpy(p, fp->basename, remainder);
193 delete_item(fname, fp->mode, flags & ~DEL_TERSE);
59faec8b
WD
194 }
195 flist_free(dirlist);
196
0e5665d3
WD
197 fname[dlen] = '\0';
198
59faec8b
WD
199 pop_local_filters(save_filters);
200
f75a53e7
WD
201 if (max_delete && ++deletion_count > max_delete)
202 return 0;
59faec8b
WD
203
204 if (do_rmdir(fname) == 0) {
205 if (!(flags & DEL_TERSE))
206 log_delete(fname, mode);
4d474ad5 207 } else if (errno != ENOTEMPTY && errno != EEXIST && errno != ENOENT) {
59faec8b
WD
208 rsyserr(FERROR, errno, "delete_file: rmdir %s failed",
209 full_fname(fname));
210 return -1;
211 }
212
213 return 0;
214}
215
216
217/* This function is used to implement per-directory deletion, and is used by
218 * all the --delete-WHEN options. Note that the fbuf pointer must point to a
219 * MAXPATHLEN buffer with the name of the directory in it (the functions we
220 * call will append names onto the end, but the old dir value will be restored
221 * on exit). */
222static void delete_in_dir(struct file_list *flist, char *fbuf,
0e82af2d 223 struct file_struct *file, STRUCT_STAT *stp)
59faec8b
WD
224{
225 static int min_depth = MAXPATHLEN, cur_depth = -1;
226 static void *filt_array[MAXPATHLEN/2+1];
717b0430 227 static int already_warned = 0;
59faec8b
WD
228 struct file_list *dirlist;
229 char delbuf[MAXPATHLEN];
468d7668 230 int dlen, i;
59faec8b
WD
231
232 if (!flist) {
233 while (cur_depth >= min_depth)
234 pop_local_filters(filt_array[cur_depth--]);
235 min_depth = MAXPATHLEN;
236 cur_depth = -1;
237 return;
238 }
239
240 if (verbose > 2)
45c49b52 241 rprintf(FINFO, "delete_in_dir(%s)\n", fbuf);
59faec8b
WD
242
243 if (allowed_lull)
ee1d11c4 244 maybe_send_keepalive();
59faec8b
WD
245
246 if (file->dir.depth >= MAXPATHLEN/2+1)
247 return; /* Impossible... */
248
59faec8b 249 if (io_error && !(lp_ignore_errors(module_id) || ignore_errors)) {
717b0430 250 if (already_warned)
f75a53e7 251 return;
59faec8b
WD
252 rprintf(FINFO,
253 "IO error encountered -- skipping file deletion\n");
717b0430 254 already_warned = 1;
59faec8b
WD
255 return;
256 }
257
258 while (cur_depth >= file->dir.depth && cur_depth >= min_depth)
259 pop_local_filters(filt_array[cur_depth--]);
260 cur_depth = file->dir.depth;
261 if (min_depth > cur_depth)
262 min_depth = cur_depth;
263 dlen = strlen(fbuf);
264 filt_array[cur_depth] = push_local_filters(fbuf, dlen);
265
0e82af2d
WD
266 if (one_file_system) {
267 if (file->flags & FLAG_TOP_DIR)
268 filesystem_dev = stp->st_dev;
269 else if (filesystem_dev != stp->st_dev)
270 return;
271 }
59faec8b 272
59faec8b
WD
273 dirlist = get_dirlist(fbuf, dlen, 0);
274
275 /* If an item in dirlist is not found in flist, delete it
276 * from the filesystem. */
277 for (i = dirlist->count; i--; ) {
f3ab64d3 278 struct file_struct *fp = dirlist->files[i];
0e82af2d 279 if (!fp->basename || fp->flags & FLAG_MOUNT_POINT)
bb0d8edf 280 continue;
468d7668 281 if (flist_find(flist, fp) < 0) {
6cbde57d 282 f_name(fp, delbuf);
e912bd4d 283 delete_item(delbuf, fp->mode, DEL_FORCE_RECURSE);
468d7668 284 }
59faec8b
WD
285 }
286
287 flist_free(dirlist);
288}
289
290/* This deletes any files on the receiving side that are not present on the
291 * sending side. This is used by --delete-before and --delete-after. */
ee1d11c4 292static void do_delete_pass(struct file_list *flist)
59faec8b
WD
293{
294 char fbuf[MAXPATHLEN];
bb0d8edf 295 STRUCT_STAT st;
59faec8b
WD
296 int j;
297
b225b089
WD
298 if (dry_run > 1 /* destination doesn't exist yet */
299 || list_only)
f75a53e7
WD
300 return;
301
59faec8b
WD
302 for (j = 0; j < flist->count; j++) {
303 struct file_struct *file = flist->files[j];
304
305 if (!(file->flags & FLAG_DEL_HERE))
306 continue;
307
6cbde57d 308 f_name(file, fbuf);
59faec8b 309 if (verbose > 1 && file->flags & FLAG_TOP_DIR)
45c49b52 310 rprintf(FINFO, "deleting in %s\n", fbuf);
59faec8b 311
bb0d8edf
WD
312 if (link_stat(fbuf, &st, keep_dirlinks) < 0
313 || !S_ISDIR(st.st_mode))
314 continue;
315
0e82af2d 316 delete_in_dir(flist, fbuf, file, &st);
59faec8b 317 }
0e82af2d 318 delete_in_dir(NULL, NULL, NULL, NULL);
0e5665d3 319
a06e2b7c
WD
320 if (do_progress && !am_server)
321 rprintf(FINFO, " \r");
59faec8b
WD
322}
323
48224e4c 324int unchanged_attrs(struct file_struct *file, STRUCT_STAT *st)
2f03f956 325{
b7e8628c
WD
326 if (preserve_perms
327 && (st->st_mode & CHMOD_BITS) != (file->mode & CHMOD_BITS))
84acca07 328 return 0;
bb24028f 329
7b6fa00f 330 if (am_root && preserve_uid && st->st_uid != file->uid)
b7e8628c 331 return 0;
bb24028f 332
7b6fa00f 333 if (preserve_gid && file->gid != GID_NONE && st->st_gid != file->gid)
b7e8628c
WD
334 return 0;
335
336 return 1;
337}
338
ee1d11c4 339void itemize(struct file_struct *file, int ndx, int statret, STRUCT_STAT *st,
ef20efcb 340 int32 iflags, uchar fnamecmp_type, char *xname)
06a1dbad 341{
48224e4c
WD
342 if (statret >= 0) { /* A from-dest-dir statret can == 1! */
343 int keep_time = !preserve_times ? 0
344 : S_ISDIR(file->mode) ? !omit_dir_times
345 : !S_ISLNK(file->mode);
346
c557eb8c
WD
347 if (S_ISREG(file->mode) && file->length != st->st_size)
348 iflags |= ITEM_REPORT_SIZE;
48224e4c
WD
349 if ((iflags & (ITEM_TRANSFER|ITEM_LOCAL_CHANGE) && !keep_time
350 && (!(iflags & ITEM_XNAME_FOLLOWS) || *xname))
0f86c74e 351 || (keep_time && cmp_time(file->modtime, st->st_mtime) != 0))
48224e4c 352 iflags |= ITEM_REPORT_TIME;
b9887818 353 if ((file->mode & CHMOD_BITS) != (st->st_mode & CHMOD_BITS))
48224e4c 354 iflags |= ITEM_REPORT_PERMS;
7b6fa00f 355 if (preserve_uid && am_root && file->uid != st->st_uid)
48224e4c 356 iflags |= ITEM_REPORT_OWNER;
7b6fa00f
WD
357 if (preserve_gid && file->gid != GID_NONE
358 && st->st_gid != file->gid)
48224e4c 359 iflags |= ITEM_REPORT_GROUP;
ef20efcb 360 } else
ee1d11c4 361 iflags |= ITEM_IS_NEW;
c557eb8c 362
a1d23b53 363 iflags &= 0xffff;
ee1d11c4 364 if ((iflags & SIGNIFICANT_ITEM_FLAGS || verbose > 1
a9e47626 365 || log_format_has_i > 1 || (xname && *xname)) && !read_batch) {
6c3862fa
WD
366 if (protocol_version >= 29) {
367 if (ndx >= 0)
ee1d11c4
WD
368 write_int(sock_f_out, ndx);
369 write_shortint(sock_f_out, iflags);
ef20efcb
WD
370 if (iflags & ITEM_BASIS_TYPE_FOLLOWS)
371 write_byte(sock_f_out, fnamecmp_type);
372 if (iflags & ITEM_XNAME_FOLLOWS)
373 write_vstring(sock_f_out, xname, strlen(xname));
6c3862fa 374 } else if (ndx >= 0)
ef20efcb 375 log_item(file, &stats, iflags, xname);
06a1dbad
WD
376 }
377}
378
379
b7e8628c 380/* Perform our quick-check heuristic for determining if a file is unchanged. */
48224e4c 381int unchanged_file(char *fn, struct file_struct *file, STRUCT_STAT *st)
b7e8628c
WD
382{
383 if (st->st_size != file->length)
384 return 0;
59c95e42 385
2cda2560 386 /* if always checksum is set then we use the checksum instead
2f03f956
AT
387 of the file time to determine whether to sync */
388 if (always_checksum && S_ISREG(st->st_mode)) {
389 char sum[MD4_SUM_LENGTH];
b7e8628c 390 file_checksum(fn, sum, st->st_size);
cfe39780 391 return memcmp(sum, file->u.sum, checksum_len) == 0;
2f03f956
AT
392 }
393
cc1e997d 394 if (size_only)
84acca07 395 return 1;
2f03f956 396
cc1e997d 397 if (ignore_times)
84acca07 398 return 0;
cc1e997d 399
0f86c74e 400 return cmp_time(st->st_mtime, file->modtime) == 0;
2f03f956
AT
401}
402
403
ec8290c8 404/*
195bd906 405 * set (initialize) the size entries in the per-file sum_struct
ec8290c8 406 * calculating dynamic block and checksum sizes.
195bd906 407 *
ec8290c8 408 * This is only called from generate_and_send_sums() but is a separate
195bd906
S
409 * function to encapsulate the logic.
410 *
411 * The block size is a rounded square root of file length.
412 *
413 * The checksum size is determined according to:
ed7e7955 414 * blocksum_bits = BLOCKSUM_BIAS + 2*log2(file_len) - log2(block_len)
195bd906
S
415 * provided by Donovan Baarda which gives a probability of rsync
416 * algorithm corrupting data and falling back using the whole md4
417 * checksums.
418 *
419 * This might be made one of several selectable heuristics.
420 */
1490812a 421static void sum_sizes_sqroot(struct sum_struct *sum, int64 len)
195bd906 422{
a255c592 423 int32 blength;
da9d12f5 424 int s2length;
195bd906 425
a255c592 426 if (block_size)
195bd906 427 blength = block_size;
a255c592 428 else if (len <= BLOCK_SIZE * BLOCK_SIZE)
195bd906 429 blength = BLOCK_SIZE;
a255c592
WD
430 else {
431 int32 c;
1490812a 432 int64 l;
eae7165c
WD
433 int cnt;
434 for (c = 1, l = len, cnt = 0; l >>= 2; c <<= 1, cnt++) {}
435 if (cnt >= 31 || c >= MAX_BLOCK_SIZE)
436 blength = MAX_BLOCK_SIZE;
437 else {
438 blength = 0;
439 do {
440 blength |= c;
1490812a 441 if (len < (int64)blength * blength)
eae7165c
WD
442 blength &= ~c;
443 c >>= 1;
444 } while (c >= 8); /* round to multiple of 8 */
445 blength = MAX(blength, BLOCK_SIZE);
195bd906 446 }
195bd906
S
447 }
448
d04e9c51 449 if (protocol_version < 27) {
195bd906
S
450 s2length = csum_length;
451 } else if (csum_length == SUM_LENGTH) {
452 s2length = SUM_LENGTH;
453 } else {
a255c592 454 int32 c;
1490812a 455 int64 l;
da9d12f5 456 int b = BLOCKSUM_BIAS;
a255c592
WD
457 for (l = len; l >>= 1; b += 2) {}
458 for (c = blength; c >>= 1 && b; b--) {}
459 /* add a bit, subtract rollsum, round up. */
460 s2length = (b + 1 - 32 + 7) / 8; /* --optimize in compiler-- */
195bd906
S
461 s2length = MAX(s2length, csum_length);
462 s2length = MIN(s2length, SUM_LENGTH);
463 }
464
465 sum->flength = len;
466 sum->blength = blength;
467 sum->s2length = s2length;
a5b786d8
WD
468 sum->remainder = len % blength;
469 sum->count = len / blength + (sum->remainder != 0);
195bd906
S
470
471 if (sum->count && verbose > 2) {
a255c592
WD
472 rprintf(FINFO,
473 "count=%.0f rem=%ld blength=%ld s2length=%d flength=%.0f\n",
474 (double)sum->count, (long)sum->remainder, (long)sum->blength,
da9d12f5 475 sum->s2length, (double)sum->flength);
195bd906
S
476 }
477}
80605142 478
bceec82f 479
80605142
WD
480/*
481 * Generate and send a stream of signatures/checksums that describe a buffer
e66dfd18 482 *
80605142
WD
483 * Generate approximately one checksum every block_len bytes.
484 */
cd6aa5b5 485static void generate_and_send_sums(int fd, OFF_T len, int f_out, int f_copy)
2f03f956 486{
a1cbe76e 487 int32 i;
6e45e1dd 488 struct map_struct *mapbuf;
80605142 489 struct sum_struct sum;
2f03f956
AT
490 OFF_T offset = 0;
491
423dba8e 492 sum_sizes_sqroot(&sum, len);
6cc11982
WD
493 write_sum_head(f_out, &sum);
494
495 if (append_mode > 0 && f_copy < 0)
496 return;
e66dfd18 497
6e45e1dd 498 if (len > 0)
96d910c7 499 mapbuf = map_file(fd, len, MAX_MAP_SIZE, sum.blength);
6e45e1dd
WD
500 else
501 mapbuf = NULL;
502
80605142 503 for (i = 0; i < sum.count; i++) {
a255c592 504 int32 n1 = (int32)MIN(len, (OFF_T)sum.blength);
6e45e1dd 505 char *map = map_ptr(mapbuf, offset, n1);
80605142 506 char sum2[SUM_LENGTH];
6cc11982
WD
507 uint32 sum1;
508
509 len -= n1;
510 offset += n1;
2f03f956 511
6cc11982 512 if (f_copy >= 0) {
cd6aa5b5 513 full_write(f_copy, map, n1);
6cc11982
WD
514 if (append_mode > 0)
515 continue;
516 }
cd6aa5b5 517
6cc11982 518 sum1 = get_checksum1(map, n1);
80605142 519 get_checksum2(map, n1, sum2);
2f03f956 520
80605142 521 if (verbose > 3) {
e66dfd18 522 rprintf(FINFO,
a255c592 523 "chunk[%.0f] offset=%.0f len=%ld sum1=%08lx\n",
6cc11982 524 (double)i, (double)offset - n1, (long)n1,
0e36d9da 525 (unsigned long)sum1);
80605142
WD
526 }
527 write_int(f_out, sum1);
fc0257c9 528 write_buf(f_out, sum2, sum.s2length);
2f03f956 529 }
6e45e1dd
WD
530
531 if (mapbuf)
532 unmap_file(mapbuf);
2f03f956
AT
533}
534
06a1dbad 535
8e85be0a
WD
536/* Try to find a filename in the same dir as "fname" with a similar name. */
537static int find_fuzzy(struct file_struct *file, struct file_list *dirlist)
538{
539 int fname_len, fname_suf_len;
540 const char *fname_suf, *fname = file->basename;
f328e0f3 541 uint32 lowest_dist = 25 << 16; /* ignore a distance greater than 25 */
8e85be0a
WD
542 int j, lowest_j = -1;
543
544 fname_len = strlen(fname);
545 fname_suf = find_filename_suffix(fname, fname_len, &fname_suf_len);
546
547 for (j = 0; j < dirlist->count; j++) {
548 struct file_struct *fp = dirlist->files[j];
549 const char *suf, *name;
550 int len, suf_len;
551 uint32 dist;
552
553 if (!S_ISREG(fp->mode) || !fp->length
554 || fp->flags & FLAG_NO_FUZZY)
555 continue;
556
557 name = fp->basename;
558
559 if (fp->length == file->length
0f86c74e 560 && cmp_time(fp->modtime, file->modtime) == 0) {
8e85be0a
WD
561 if (verbose > 4) {
562 rprintf(FINFO,
563 "fuzzy size/modtime match for %s\n",
564 name);
565 }
566 return j;
567 }
568
569 len = strlen(name);
570 suf = find_filename_suffix(name, len, &suf_len);
571
572 dist = fuzzy_distance(name, len, fname, fname_len);
573 /* Add some extra weight to how well the suffixes match. */
574 dist += fuzzy_distance(suf, suf_len, fname_suf, fname_suf_len)
575 * 10;
576 if (verbose > 4) {
577 rprintf(FINFO, "fuzzy distance for %s = %d.%05d\n",
578 name, (int)(dist>>16), (int)(dist&0xFFFF));
579 }
580 if (dist <= lowest_dist) {
581 lowest_dist = dist;
582 lowest_j = j;
583 }
584 }
585
586 return lowest_j;
587}
588
ee1d11c4
WD
589void check_for_finished_hlinks(int itemizing, enum logcode code)
590{
591 struct file_struct *file;
592 int ndx;
593
594 while ((ndx = get_hlink_num()) != -1) {
595 if (ndx < 0 || ndx >= the_file_list->count)
596 continue;
597
598 file = the_file_list->files[ndx];
599 if (!file->link_u.links)
600 continue;
601
602 hard_link_cluster(file, ndx, itemizing, code);
603 }
604}
2f03f956 605
48224e4c
WD
606/* This is only called for regular files. We return -2 if we've finished
607 * handling the file, -1 if no dest-linking occurred, or a non-negative
608 * value if we found an alternate basis file. */
609static int try_dests_reg(struct file_struct *file, char *fname, int ndx,
610 char *cmpbuf, STRUCT_STAT *stp, int itemizing,
e912bd4d 611 int maybe_ATTRS_REPORT, enum logcode code)
48224e4c
WD
612{
613 int best_match = -1;
614 int match_level = 0;
615 int j = 0;
616
617 do {
618 pathjoin(cmpbuf, MAXPATHLEN, basis_dir[j], fname);
619 if (link_stat(cmpbuf, stp, 0) < 0 || !S_ISREG(stp->st_mode))
620 continue;
621 switch (match_level) {
622 case 0:
623 best_match = j;
624 match_level = 1;
625 /* FALL THROUGH */
626 case 1:
627 if (!unchanged_file(cmpbuf, file, stp))
628 continue;
629 best_match = j;
630 match_level = 2;
631 /* FALL THROUGH */
632 case 2:
633 if (!unchanged_attrs(file, stp))
634 continue;
635 if ((always_checksum || ignore_times)
0f86c74e 636 && cmp_time(stp->st_mtime, file->modtime))
48224e4c
WD
637 continue;
638 best_match = j;
639 match_level = 3;
640 break;
641 }
642 break;
643 } while (basis_dir[++j] != NULL);
644
645 if (!match_level)
646 return -1;
647
648 if (j != best_match) {
649 j = best_match;
650 pathjoin(cmpbuf, MAXPATHLEN, basis_dir[j], fname);
651 if (link_stat(cmpbuf, stp, 0) < 0)
652 match_level = 0;
653 }
654
655#ifdef HAVE_LINK
656 if (match_level == 3 && !copy_dest) {
657 if (link_dest) {
658 if (hard_link_one(file, ndx, fname, 0, stp,
659 cmpbuf, 1,
660 itemizing && verbose > 1,
661 code) < 0)
662 goto try_a_copy;
3447d610
WD
663 if (preserve_hard_links && file->link_u.links)
664 hard_link_cluster(file, ndx, itemizing, code);
48224e4c
WD
665 } else if (itemizing)
666 itemize(file, ndx, 0, stp, 0, 0, NULL);
e912bd4d 667 if (verbose > 1 && maybe_ATTRS_REPORT) {
48224e4c
WD
668 code = daemon_log_format_has_i || dry_run
669 ? FCLIENT : FINFO;
45c49b52 670 rprintf(code, "%s is uptodate\n", fname);
48224e4c
WD
671 }
672 return -2;
673 }
674#endif
675
676 if (match_level >= 2) {
677 try_a_copy: /* Copy the file locally. */
678 if (copy_file(cmpbuf, fname, file->mode) < 0) {
679 if (verbose) {
3447d610 680 rsyserr(FINFO, errno, "copy_file %s => %s",
45c49b52 681 full_fname(cmpbuf), fname);
48224e4c
WD
682 }
683 return -1;
684 }
3447d610
WD
685 if (itemizing)
686 itemize(file, ndx, 0, stp, ITEM_LOCAL_CHANGE, 0, NULL);
e912bd4d
WD
687 set_file_attrs(fname, file, NULL, 0);
688 if (maybe_ATTRS_REPORT
48224e4c
WD
689 && ((!itemizing && verbose && match_level == 2)
690 || (verbose > 1 && match_level == 3))) {
691 code = daemon_log_format_has_i || dry_run
692 ? FCLIENT : FINFO;
45c49b52 693 rprintf(code, "%s%s\n", fname,
48224e4c
WD
694 match_level == 3 ? " is uptodate" : "");
695 }
3447d610
WD
696 if (preserve_hard_links && file->link_u.links)
697 hard_link_cluster(file, ndx, itemizing, code);
48224e4c
WD
698 return -2;
699 }
700
701 return FNAMECMP_BASIS_DIR_LOW + j;
702}
703
704/* This is only called for non-regular files. We return -2 if we've finished
705 * handling the file, or -1 if no dest-linking occurred. */
706static int try_dests_non(struct file_struct *file, char *fname, int ndx,
707 int itemizing, int *possible_ptr,
e912bd4d 708 int maybe_ATTRS_REPORT, enum logcode code)
48224e4c
WD
709{
710 char fnamebuf[MAXPATHLEN], lnk[MAXPATHLEN];
711 STRUCT_STAT st;
712 int len, i = 0;
713
714 do {
715 pathjoin(fnamebuf, MAXPATHLEN, basis_dir[i], fname);
716 if (link_stat(fnamebuf, &st, 0) < 0 || S_ISDIR(st.st_mode)
717 || !unchanged_attrs(file, &st))
718 continue;
719 if (S_ISLNK(file->mode)) {
bb0d8edf 720#ifdef SUPPORT_LINKS
48224e4c
WD
721 if ((len = readlink(fnamebuf, lnk, MAXPATHLEN-1)) <= 0)
722 continue;
723 lnk[len] = '\0';
724 if (strcmp(lnk, file->u.link) != 0)
bb0d8edf 725#endif
48224e4c
WD
726 continue;
727 } else {
728 if (!IS_DEVICE(st.st_mode) || st.st_rdev != file->u.rdev)
729 continue;
730 }
731 if (link_dest) {
732 if (do_link(fnamebuf, fname) < 0) {
733 /* TODO improve this to be based on errno? */
734 *possible_ptr = 0;
735 break;
736 }
3447d610
WD
737 if (preserve_hard_links && file->link_u.links)
738 hard_link_cluster(file, ndx, itemizing, code);
48224e4c
WD
739 }
740 if (itemizing && log_format_has_i && verbose > 1) {
741 int changes = compare_dest ? 0 : ITEM_LOCAL_CHANGE
742 + (link_dest ? ITEM_XNAME_FOLLOWS : 0);
743 char *lp = link_dest ? "" : NULL;
744 itemize(file, ndx, 0, &st, changes, 0, lp);
745 }
e912bd4d 746 if (verbose > 1 && maybe_ATTRS_REPORT) {
3447d610
WD
747 code = daemon_log_format_has_i || dry_run
748 ? FCLIENT : FINFO;
45c49b52 749 rprintf(code, "%s is uptodate\n", fname);
48224e4c
WD
750 }
751 return -2;
752 } while (basis_dir[++i] != NULL);
753
754 return -1;
755}
756
ed7e7955
WD
757static int phase = 0;
758
ee1d11c4 759/* Acts on the_file_list->file's ndx'th item, whose name is fname. If a dir,
0492fdfb
WD
760 * make sure it exists, and has the right permissions/timestamp info. For
761 * all other non-regular files (symlinks, etc.) we create them here. For
762 * regular files that have changed, we try to find a basis file and then
763 * start sending checksums.
ef1aa910 764 *
0e5665d3
WD
765 * When fname is non-null, it must point to a MAXPATHLEN buffer!
766 *
0492fdfb
WD
767 * Note that f_out is set to -1 when doing final directory-permission and
768 * modification-time repair. */
ee1d11c4 769static void recv_generator(char *fname, struct file_struct *file, int ndx,
e912bd4d 770 int itemizing, int maybe_ATTRS_REPORT,
ef20efcb 771 enum logcode code, int f_out)
2cda2560 772{
8174bc35 773 static int missing_below = -1, excluded_below = -1;
a9cc1283 774 static char *parent_dirname = "";
8e85be0a
WD
775 static struct file_list *fuzzy_dirlist = NULL;
776 struct file_struct *fuzzy_file = NULL;
41cfde6b 777 int fd = -1, f_copy = -1;
1f1d368a 778 STRUCT_STAT st, real_st, partial_st;
41cfde6b 779 struct file_struct *back_file = NULL;
1f1d368a 780 int statret, real_ret, stat_errno;
41cfde6b 781 char *fnamecmp, *partialptr, *backupptr = NULL;
375a4556 782 char fnamecmpbuf[MAXPATHLEN];
41cfde6b 783 uchar fnamecmp_type;
e35d9f2d 784 int del_opts = DEL_TERSE | (delete_mode ? DEL_FORCE_RECURSE : 0);
f7632fc6 785
dfd5ba6a
WD
786 if (list_only)
787 return;
2f03f956 788
8e85be0a
WD
789 if (!fname) {
790 if (fuzzy_dirlist) {
791 flist_free(fuzzy_dirlist);
792 fuzzy_dirlist = NULL;
8e85be0a
WD
793 }
794 if (missing_below >= 0) {
795 dry_run--;
796 missing_below = -1;
797 }
8c9e06d7 798 parent_dirname = "";
8e85be0a
WD
799 return;
800 }
801
45c49b52
WD
802 if (verbose > 2)
803 rprintf(FINFO, "recv_generator(%s,%d)\n", fname, ndx);
2f03f956 804
8174bc35
WD
805 if (server_filter_list.head) {
806 if (excluded_below >= 0) {
807 if (file->dir.depth > excluded_below)
808 goto skipping;
809 excluded_below = -1;
810 }
811 if (check_filter(&server_filter_list, fname,
812 S_ISDIR(file->mode)) < 0) {
813 if (S_ISDIR(file->mode))
814 excluded_below = file->dir.depth;
b2e7c913 815 skipping:
8174bc35
WD
816 if (verbose) {
817 rprintf(FINFO,
818 "skipping server-excluded file \"%s\"\n",
45c49b52 819 fname);
8174bc35
WD
820 }
821 return;
3e35c34b 822 }
3e35c34b 823 }
97f9dcae 824
8e85be0a 825 if (missing_below >= 0 && file->dir.depth <= missing_below) {
df337831
WD
826 dry_run--;
827 missing_below = -1;
828 }
73f7af0e
WD
829 if (dry_run > 1) {
830 statret = -1;
831 stat_errno = ENOENT;
832 } else {
8c9e06d7
WD
833 char *dn = file->dirname ? file->dirname : ".";
834 if (parent_dirname != dn && strcmp(parent_dirname, dn) != 0) {
835 if (relative_paths && !implied_dirs && stat(dn, &st) < 0
836 && create_directory_path(fname, orig_umask) < 0) {
837 rsyserr(FERROR, errno,
838 "recv_generator: mkdir %s failed",
839 full_fname(dn));
8e85be0a 840 }
8c9e06d7
WD
841 if (fuzzy_dirlist) {
842 flist_free(fuzzy_dirlist);
843 fuzzy_dirlist = NULL;
844 }
845 if (fuzzy_basis)
846 fuzzy_dirlist = get_dirlist(dn, -1, 1);
8e85be0a 847 }
8c9e06d7 848 parent_dirname = dn;
8e85be0a 849
73f7af0e
WD
850 statret = link_stat(fname, &st,
851 keep_dirlinks && S_ISDIR(file->mode));
852 stat_errno = errno;
853 }
63787382 854
e90aab49 855 if (ignore_non_existing && statret == -1 && stat_errno == ENOENT) {
ecc81fce 856 if (verbose > 1) {
15164c0a
WD
857 rprintf(FINFO, "not creating new %s \"%s\"\n",
858 S_ISDIR(file->mode) ? "directory" : "file",
45c49b52 859 fname);
ecc81fce 860 }
1347d512
AT
861 return;
862 }
863
a9d6e6fc
WD
864 /* If we're not preserving permissions, change the file-list's
865 * mode based on the local permissions and some heuristics. */
866 if (!preserve_perms) {
867 int exists = statret == 0
868 && S_ISDIR(st.st_mode) == S_ISDIR(file->mode);
869 file->mode = dest_mode(file->mode, st.st_mode, exists);
4df9f368
AT
870 }
871
2f03f956 872 if (S_ISDIR(file->mode)) {
2cda2560
WD
873 /* The file to be received is a directory, so we need
874 * to prepare appropriately. If there is already a
875 * file of that name and it is *not* a directory, then
876 * we need to delete it. If it doesn't exist, then
027428eb 877 * (perhaps recursively) create it. */
2f03f956 878 if (statret == 0 && !S_ISDIR(st.st_mode)) {
90cf7d19 879 if (delete_item(fname, st.st_mode, del_opts) < 0)
1f1d368a 880 return;
2f03f956
AT
881 statret = -1;
882 }
df337831
WD
883 if (dry_run && statret != 0 && missing_below < 0) {
884 missing_below = file->dir.depth;
885 dry_run++;
886 }
ee1d11c4
WD
887 if (itemizing && f_out != -1) {
888 itemize(file, ndx, statret, &st,
ef20efcb 889 statret ? ITEM_LOCAL_CHANGE : 0, 0, NULL);
ee1d11c4 890 }
00b96184 891 if (statret != 0 && do_mkdir(fname,file->mode) < 0 && errno != EEXIST) {
027428eb
WD
892 if (!relative_paths || errno != ENOENT
893 || create_directory_path(fname, orig_umask) < 0
09290693 894 || (do_mkdir(fname, file->mode) < 0 && errno != EEXIST)) {
d62bcc17
WD
895 rsyserr(FERROR, errno,
896 "recv_generator: mkdir %s failed",
897 full_fname(fname));
2f03f956
AT
898 }
899 }
e912bd4d 900 if (set_file_attrs(fname, file, statret ? NULL : &st, 0)
6c3862fa 901 && verbose && code && f_out != -1)
45c49b52 902 rprintf(code, "%s/\n", fname);
f75a53e7 903 if (delete_during && f_out != -1 && !phase && dry_run < 2
31937d36 904 && (file->flags & FLAG_DEL_HERE))
0e82af2d 905 delete_in_dir(the_file_list, fname, file, &st);
2f03f956 906 return;
06a1dbad 907 }
6c3862fa 908
273a7ed5
WD
909 if (preserve_hard_links && file->link_u.links
910 && hard_link_check(file, ndx, fname, statret, &st,
911 itemizing, code, HL_CHECK_MASTER))
912 return;
913
2f03f956 914 if (preserve_links && S_ISLNK(file->mode)) {
4f5b0756 915#ifdef SUPPORT_LINKS
728d0922 916 if (safe_symlinks && unsafe_symlink(file->u.link, fname)) {
2f03f956 917 if (verbose) {
2a5d5a8c 918 if (the_file_list->count == 1)
6cbde57d 919 fname = f_name(file, NULL);
4875d6b6
WD
920 rprintf(FINFO,
921 "ignoring unsafe symlink %s -> \"%s\"\n",
45c49b52 922 full_fname(fname), file->u.link);
2f03f956
AT
923 }
924 return;
925 }
926 if (statret == 0) {
7e38410e
WD
927 char lnk[MAXPATHLEN];
928 int len;
929
8a8356b7 930 if (!S_ISDIR(st.st_mode)
7e38410e
WD
931 && (len = readlink(fname, lnk, MAXPATHLEN-1)) > 0) {
932 lnk[len] = 0;
85d4d142
MP
933 /* A link already pointing to the
934 * right place -- no further action
935 * required. */
7e38410e 936 if (strcmp(lnk, file->u.link) == 0) {
6c3862fa 937 if (itemizing) {
ee1d11c4 938 itemize(file, ndx, 0, &st, 0,
ef20efcb 939 0, NULL);
c557eb8c 940 }
e912bd4d
WD
941 set_file_attrs(fname, file, &st,
942 maybe_ATTRS_REPORT);
273a7ed5
WD
943 if (preserve_hard_links
944 && file->link_u.links) {
945 hard_link_cluster(file, ndx,
946 itemizing,
947 code);
948 }
2f03f956
AT
949 return;
950 }
2cda2560 951 }
7e38410e
WD
952 /* Not the right symlink (or not a symlink), so
953 * delete it. */
90cf7d19 954 if (delete_item(fname, st.st_mode, del_opts) < 0)
1f1d368a
WD
955 return;
956 if (!S_ISLNK(st.st_mode))
88b218fa 957 statret = -1;
48224e4c
WD
958 } else if (basis_dir[0] != NULL && can_link_symlinks) {
959 if (try_dests_non(file, fname, ndx, itemizing,
960 &can_link_symlinks,
e912bd4d 961 maybe_ATTRS_REPORT, code) == -2) {
48224e4c
WD
962 if (!copy_dest)
963 return;
964 itemizing = code = 0;
965 }
2f03f956 966 }
273a7ed5
WD
967 if (preserve_hard_links && file->link_u.links
968 && hard_link_check(file, ndx, fname, -1, &st,
969 itemizing, code, HL_SKIP))
970 return;
728d0922 971 if (do_symlink(file->u.link,fname) != 0) {
d62bcc17 972 rsyserr(FERROR, errno, "symlink %s -> \"%s\" failed",
45c49b52 973 full_fname(fname), file->u.link);
2f03f956 974 } else {
e912bd4d 975 set_file_attrs(fname, file, NULL, 0);
6c3862fa 976 if (itemizing) {
ee1d11c4 977 itemize(file, ndx, statret, &st,
ef20efcb 978 ITEM_LOCAL_CHANGE, 0, NULL);
6c3862fa
WD
979 }
980 if (code && verbose) {
45c49b52
WD
981 rprintf(code, "%s -> %s\n", fname,
982 file->u.link);
2f03f956 983 }
fe960187
WD
984 if (remove_sent_files && !dry_run) {
985 char numbuf[4];
986 SIVAL(numbuf, 0, ndx);
987 send_msg(MSG_SUCCESS, numbuf, 4);
988 }
273a7ed5
WD
989 if (preserve_hard_links && file->link_u.links)
990 hard_link_cluster(file, ndx, itemizing, code);
2f03f956
AT
991 }
992#endif
993 return;
994 }
995
b5c6a6ae
WD
996 if ((am_root && preserve_devices && IS_DEVICE(file->mode))
997 || (preserve_specials && IS_SPECIAL(file->mode))) {
48224e4c
WD
998 if (statret != 0
999 && (basis_dir[0] != NULL && can_link_devices)) {
1000 if (try_dests_non(file, fname, ndx, itemizing,
1001 &can_link_devices,
e912bd4d 1002 maybe_ATTRS_REPORT, code) == -2) {
48224e4c
WD
1003 if (!copy_dest)
1004 return;
1005 itemizing = code = 0;
1006 }
1007 }
15cf186b
WD
1008 if (statret != 0
1009 || (st.st_mode & ~CHMOD_BITS) != (file->mode & ~CHMOD_BITS)
1010 || st.st_rdev != file->u.rdev) {
0fdb1aa8 1011 if (statret == 0
90cf7d19 1012 && delete_item(fname, st.st_mode, del_opts) < 0)
1f1d368a 1013 return;
273a7ed5
WD
1014 if (preserve_hard_links && file->link_u.links
1015 && hard_link_check(file, ndx, fname, -1, &st,
1016 itemizing, code, HL_SKIP))
1017 return;
b5c6a6ae
WD
1018 if ((IS_DEVICE(file->mode) && !IS_DEVICE(st.st_mode))
1019 || (IS_SPECIAL(file->mode) && !IS_SPECIAL(st.st_mode)))
88b218fa 1020 statret = -1;
d62bcc17 1021 if (verbose > 2) {
2f03f956 1022 rprintf(FINFO,"mknod(%s,0%o,0x%x)\n",
45c49b52 1023 fname,
ecc81fce 1024 (int)file->mode, (int)file->u.rdev);
d62bcc17 1025 }
00b96184 1026 if (do_mknod(fname,file->mode,file->u.rdev) < 0) {
d62bcc17
WD
1027 rsyserr(FERROR, errno, "mknod %s failed",
1028 full_fname(fname));
2f03f956 1029 } else {
e912bd4d 1030 set_file_attrs(fname, file, NULL, 0);
6c3862fa 1031 if (itemizing) {
ee1d11c4 1032 itemize(file, ndx, statret, &st,
ef20efcb 1033 ITEM_LOCAL_CHANGE, 0, NULL);
6c3862fa 1034 }
45c49b52
WD
1035 if (code && verbose)
1036 rprintf(code, "%s\n", fname);
273a7ed5
WD
1037 if (preserve_hard_links && file->link_u.links) {
1038 hard_link_cluster(file, ndx,
1039 itemizing, code);
1040 }
2f03f956
AT
1041 }
1042 } else {
ee1d11c4 1043 if (itemizing)
ef20efcb 1044 itemize(file, ndx, statret, &st, 0, 0, NULL);
e912bd4d 1045 set_file_attrs(fname, file, &st, maybe_ATTRS_REPORT);
273a7ed5
WD
1046 if (preserve_hard_links && file->link_u.links)
1047 hard_link_cluster(file, ndx, itemizing, code);
2f03f956
AT
1048 }
1049 return;
1050 }
2f03f956 1051
2f03f956 1052 if (!S_ISREG(file->mode)) {
2a5d5a8c 1053 if (the_file_list->count == 1)
6cbde57d 1054 fname = f_name(file, NULL);
45c49b52 1055 rprintf(FINFO, "skipping non-regular file \"%s\"\n", fname);
2f03f956
AT
1056 return;
1057 }
1058
289a3216
WD
1059 if (max_size && file->length > max_size) {
1060 if (verbose > 1) {
1061 if (the_file_list->count == 1)
6cbde57d 1062 fname = f_name(file, NULL);
45c49b52 1063 rprintf(FINFO, "%s is over max-size\n", fname);
289a3216
WD
1064 }
1065 return;
1066 }
02b5cb23
WD
1067 if (min_size && file->length < min_size) {
1068 if (verbose > 1) {
1069 if (the_file_list->count == 1)
6cbde57d 1070 fname = f_name(file, NULL);
45c49b52 1071 rprintf(FINFO, "%s is under min-size\n", fname);
02b5cb23
WD
1072 }
1073 return;
1074 }
289a3216 1075
e90aab49 1076 if (ignore_existing && statret == 0) {
c3cbcfb8 1077 if (verbose > 1)
45c49b52 1078 rprintf(FINFO, "%s exists\n", fname);
c3cbcfb8
WD
1079 return;
1080 }
1081
1082 if (update_only && statret == 0
0f86c74e 1083 && cmp_time(st.st_mtime, file->modtime) > 0) {
c3cbcfb8 1084 if (verbose > 1)
45c49b52 1085 rprintf(FINFO, "%s is newer\n", fname);
c3cbcfb8
WD
1086 return;
1087 }
1088
375a4556 1089 fnamecmp = fname;
41cfde6b 1090 fnamecmp_type = FNAMECMP_FNAME;
375a4556 1091
1f1d368a 1092 if (statret == 0 && !S_ISREG(st.st_mode)) {
90cf7d19 1093 if (delete_item(fname, st.st_mode, del_opts) != 0)
1f1d368a
WD
1094 return;
1095 statret = -1;
1096 stat_errno = ENOENT;
1097 }
1098
06a1dbad 1099 if (statret != 0 && basis_dir[0] != NULL) {
48224e4c 1100 int j = try_dests_reg(file, fname, ndx, fnamecmpbuf, &st,
e912bd4d 1101 itemizing, maybe_ATTRS_REPORT, code);
48224e4c
WD
1102 if (j == -2)
1103 return;
1104 if (j != -1) {
1105 fnamecmp = fnamecmpbuf;
1106 fnamecmp_type = j;
9ba46343 1107 statret = 0;
e7d13fe5
WD
1108 }
1109 }
1110
1f1d368a
WD
1111 real_ret = statret;
1112 real_st = st;
375a4556 1113
9d954dca 1114 if (partial_dir && (partialptr = partial_dir_fname(fname)) != NULL
72c19bb3
WD
1115 && link_stat(partialptr, &partial_st, 0) == 0
1116 && S_ISREG(partial_st.st_mode)) {
06a1dbad 1117 if (statret != 0)
72c19bb3
WD
1118 goto prepare_to_open;
1119 } else
1120 partialptr = NULL;
89f7eff3 1121
ccb16b46 1122 if (statret != 0 && fuzzy_dirlist && dry_run <= 1) {
8e85be0a
WD
1123 int j = find_fuzzy(file, fuzzy_dirlist);
1124 if (j >= 0) {
1125 fuzzy_file = fuzzy_dirlist->files[j];
6cbde57d 1126 f_name(fuzzy_file, fnamecmpbuf);
8e85be0a
WD
1127 if (verbose > 2) {
1128 rprintf(FINFO, "fuzzy basis selected for %s: %s\n",
45c49b52 1129 fname, fnamecmpbuf);
8e85be0a 1130 }
8e85be0a 1131 st.st_size = fuzzy_file->length;
8e85be0a
WD
1132 statret = 0;
1133 fnamecmp = fnamecmpbuf;
1134 fnamecmp_type = FNAMECMP_FUZZY;
1135 }
1136 }
1137
06a1dbad 1138 if (statret != 0) {
273a7ed5 1139 if (preserve_hard_links && file->link_u.links
d8169e6f
WD
1140 && hard_link_check(file, ndx, fname, statret, &st,
1141 itemizing, code, HL_SKIP))
6dff5992 1142 return;
41cfde6b
WD
1143 if (stat_errno == ENOENT)
1144 goto notify_others;
991daf00
WD
1145 rsyserr(FERROR, stat_errno, "recv_generator: failed to stat %s",
1146 full_fname(fname));
2f03f956
AT
1147 return;
1148 }
1149
6cc11982
WD
1150 if (append_mode && st.st_size > file->length)
1151 return;
1152
48224e4c 1153 if (fnamecmp_type <= FNAMECMP_BASIS_DIR_HIGH)
b7e8628c 1154 ;
8e85be0a
WD
1155 else if (fnamecmp_type == FNAMECMP_FUZZY)
1156 ;
b7e8628c 1157 else if (unchanged_file(fnamecmp, file, &st)) {
d8b108c2
WD
1158 if (partialptr) {
1159 do_unlink(partialptr);
1160 handle_partial_dir(partialptr, PDIR_DELETE);
1161 }
48224e4c
WD
1162 if (itemizing) {
1163 itemize(file, ndx, real_ret, &real_st,
1164 0, 0, NULL);
a1d23b53 1165 }
e912bd4d 1166 set_file_attrs(fname, file, &st, maybe_ATTRS_REPORT);
48224e4c
WD
1167 if (preserve_hard_links && file->link_u.links)
1168 hard_link_cluster(file, ndx, itemizing, code);
967866d4 1169 return;
2f03f956
AT
1170 }
1171
b2e7c913 1172 prepare_to_open:
9d954dca
WD
1173 if (partialptr) {
1174 st = partial_st;
1175 fnamecmp = partialptr;
1176 fnamecmp_type = FNAMECMP_PARTIAL_DIR;
1177 statret = 0;
1178 }
1179
beb51aa0 1180 if (!do_xfers || read_batch || whole_file)
3841a04e 1181 goto notify_others;
2f03f956 1182
ccb16b46 1183 if (fuzzy_dirlist) {
8e85be0a
WD
1184 int j = flist_find(fuzzy_dirlist, file);
1185 if (j >= 0) /* don't use changing file as future fuzzy basis */
1186 fuzzy_dirlist->files[j]->flags |= FLAG_NO_FUZZY;
1187 }
1188
2cda2560 1189 /* open the file */
8c9fd200 1190 fd = do_open(fnamecmp, O_RDONLY, 0);
2f03f956
AT
1191
1192 if (fd == -1) {
d62bcc17
WD
1193 rsyserr(FERROR, errno, "failed to open %s, continuing",
1194 full_fname(fnamecmp));
b2e7c913 1195 pretend_missing:
60be6acf 1196 /* pretend the file didn't exist */
273a7ed5 1197 if (preserve_hard_links && file->link_u.links
d8169e6f
WD
1198 && hard_link_check(file, ndx, fname, statret, &st,
1199 itemizing, code, HL_SKIP))
6dff5992 1200 return;
1f1d368a 1201 statret = real_ret = -1;
41cfde6b 1202 goto notify_others;
2f03f956
AT
1203 }
1204
ef20efcb 1205 if (inplace && make_backups && fnamecmp_type == FNAMECMP_FNAME) {
cd6aa5b5
WD
1206 if (!(backupptr = get_backup_name(fname))) {
1207 close(fd);
1208 return;
1209 }
3de73827 1210 if (!(back_file = make_file(fname, NULL, NULL, 0, NO_FILTERS))) {
cd6aa5b5
WD
1211 close(fd);
1212 goto pretend_missing;
1213 }
1214 if (robust_unlink(backupptr) && errno != ENOENT) {
1215 rsyserr(FERROR, errno, "unlink %s",
1216 full_fname(backupptr));
1217 free(back_file);
1218 close(fd);
1219 return;
1220 }
1221 if ((f_copy = do_open(backupptr,
1222 O_WRONLY | O_CREAT | O_TRUNC | O_EXCL, 0600)) < 0) {
1223 rsyserr(FERROR, errno, "open %s",
1224 full_fname(backupptr));
1225 free(back_file);
1226 close(fd);
1227 return;
1228 }
41cfde6b 1229 fnamecmp_type = FNAMECMP_BACKUP;
cd6aa5b5
WD
1230 }
1231
dfd5ba6a 1232 if (verbose > 3) {
ecc81fce 1233 rprintf(FINFO, "gen mapped %s of size %.0f\n",
45c49b52 1234 fnamecmp, (double)st.st_size);
dfd5ba6a 1235 }
2f03f956 1236
2f03f956 1237 if (verbose > 2)
0492fdfb 1238 rprintf(FINFO, "generating and sending sums for %d\n", ndx);
2f03f956 1239
b2e7c913 1240 notify_others:
a7535022 1241 if (remove_sent_files && !delay_updates && !phase)
434573b2 1242 increment_active_files(ndx, itemizing, code);
0492fdfb 1243 write_int(f_out, ndx);
6c3862fa 1244 if (itemizing) {
ee1d11c4 1245 int iflags = ITEM_TRANSFER;
8237f930 1246 if (always_checksum)
a1d23b53 1247 iflags |= ITEM_REPORT_CHECKSUM;
352963dd 1248 if (fnamecmp_type != FNAMECMP_FNAME)
ef20efcb
WD
1249 iflags |= ITEM_BASIS_TYPE_FOLLOWS;
1250 if (fnamecmp_type == FNAMECMP_FUZZY)
1251 iflags |= ITEM_XNAME_FOLLOWS;
1f1d368a 1252 itemize(file, -1, real_ret, &real_st, iflags, fnamecmp_type,
ef20efcb 1253 fuzzy_file ? fuzzy_file->basename : NULL);
8e85be0a 1254 }
cd6aa5b5 1255
beb51aa0 1256 if (!do_xfers) {
ee1d11c4
WD
1257 if (preserve_hard_links && file->link_u.links)
1258 hard_link_cluster(file, ndx, itemizing, code);
1259 return;
1260 }
1261 if (read_batch)
41cfde6b
WD
1262 return;
1263
33ab4ad8 1264 if (statret != 0 || whole_file) {
e86ae6bc
WD
1265 write_sum_head(f_out, NULL);
1266 return;
1267 }
1268
1269 generate_and_send_sums(fd, st.st_size, f_out, f_copy);
1270
1271 if (f_copy >= 0) {
1272 close(f_copy);
e912bd4d 1273 set_file_attrs(backupptr, back_file, NULL, 0);
e86ae6bc
WD
1274 if (verbose > 1) {
1275 rprintf(FINFO, "backed up %s to %s\n",
45c49b52 1276 fname, backupptr);
41cfde6b 1277 }
e86ae6bc
WD
1278 free(back_file);
1279 }
41cfde6b 1280
e86ae6bc 1281 close(fd);
2f03f956
AT
1282}
1283
ef20efcb 1284void generate_files(int f_out, struct file_list *flist, char *local_name)
2f03f956 1285{
ac40b747 1286 int i;
968c8030 1287 char fbuf[MAXPATHLEN];
e912bd4d 1288 int itemizing, maybe_ATTRS_REPORT;
7433d73a 1289 enum logcode code;
ac40b747 1290 int lull_mod = allowed_lull * 5;
3ea9bbd6
WD
1291 int need_retouch_dir_times = preserve_times && !omit_dir_times;
1292 int need_retouch_dir_perms = 0;
e90aab49
WD
1293 int save_ignore_existing = ignore_existing;
1294 int save_ignore_non_existing = ignore_non_existing;
083acd49 1295 int save_do_progress = do_progress;
cd908ef4 1296 int save_make_backups = make_backups;
ee1d11c4 1297
7433d73a
WD
1298 if (protocol_version >= 29) {
1299 itemizing = 1;
e912bd4d 1300 maybe_ATTRS_REPORT = log_format_has_i ? 0 : ATTRS_REPORT;
7433d73a
WD
1301 code = daemon_log_format_has_i ? 0 : FLOG;
1302 } else if (am_daemon) {
beb51aa0 1303 itemizing = daemon_log_format_has_i && do_xfers;
e912bd4d 1304 maybe_ATTRS_REPORT = ATTRS_REPORT;
beb51aa0 1305 code = itemizing || !do_xfers ? FCLIENT : FINFO;
7433d73a
WD
1306 } else if (!am_server) {
1307 itemizing = log_format_has_i;
e912bd4d 1308 maybe_ATTRS_REPORT = log_format_has_i ? 0 : ATTRS_REPORT;
7433d73a
WD
1309 code = itemizing ? 0 : FINFO;
1310 } else {
1311 itemizing = 0;
e912bd4d 1312 maybe_ATTRS_REPORT = ATTRS_REPORT;
7433d73a
WD
1313 code = FINFO;
1314 }
2f03f956 1315
45e08edb
WD
1316 if (verbose > 2) {
1317 rprintf(FINFO, "generator starting pid=%ld count=%d\n",
1318 (long)getpid(), flist->count);
1319 }
2f03f956 1320
59faec8b 1321 if (delete_before && !local_name && flist->count > 0)
ee1d11c4 1322 do_delete_pass(flist);
083acd49 1323 do_progress = 0;
59faec8b 1324
6cc11982 1325 if (append_mode || whole_file < 0)
33ab4ad8 1326 whole_file = 0;
3e7053ac 1327 if (verbose >= 2) {
1b1fef20 1328 rprintf(FINFO, "delta-transmission %s\n",
33ab4ad8 1329 whole_file
1b1fef20
WD
1330 ? "disabled for local transfer or --whole-file"
1331 : "enabled");
3e7053ac 1332 }
2cda2560 1333
513fd04d
WD
1334 /* Since we often fill up the outgoing socket and then just sit around
1335 * waiting for the other 2 processes to do their thing, we don't want
1336 * to exit on a timeout. If the data stops flowing, the receiver will
1337 * notice that and let us know via the redo pipe (or its closing). */
1338 ignore_timeout = 1;
a57873b7 1339
2f03f956
AT
1340 for (i = 0; i < flist->count; i++) {
1341 struct file_struct *file = flist->files[i];
2f03f956 1342
dfd5ba6a
WD
1343 if (!file->basename)
1344 continue;
0492fdfb 1345
0e5665d3
WD
1346 if (local_name)
1347 strlcpy(fbuf, local_name, sizeof fbuf);
1348 else
6cbde57d 1349 f_name(file, fbuf);
e912bd4d 1350 recv_generator(fbuf, file, i, itemizing, maybe_ATTRS_REPORT,
0e5665d3 1351 code, f_out);
1f7e29b9 1352
0492fdfb
WD
1353 /* We need to ensure that any dirs we create have writeable
1354 * permissions during the time we are putting files within
1355 * them. This is then fixed after the transfer is done. */
00b96184 1356#ifdef HAVE_CHMOD
41b84ce0
WD
1357 if (!am_root && S_ISDIR(file->mode) && !(file->mode & S_IWUSR)
1358 && !list_only) {
e912bd4d 1359 mode_t mode = file->mode | S_IWUSR; /* user write */
1f7e29b9 1360 char *fname = local_name ? local_name : fbuf;
00b96184 1361 if (do_chmod(fname, mode) < 0) {
1f7e29b9
WD
1362 rsyserr(FERROR, errno,
1363 "failed to modify permissions on %s",
1364 full_fname(fname));
1365 }
3ea9bbd6 1366 need_retouch_dir_perms = 1;
2f03f956 1367 }
00b96184 1368#endif
2f03f956 1369
ee1d11c4
WD
1370 if (preserve_hard_links)
1371 check_for_finished_hlinks(itemizing, code);
9ac2395b 1372
18a11cfd 1373 if (allowed_lull && !(i % lull_mod))
ee1d11c4 1374 maybe_send_keepalive();
c2523a05 1375 else if (!(i % 200))
417099fa 1376 maybe_flush_socket();
2f03f956 1377 }
ef20efcb 1378 recv_generator(NULL, NULL, 0, 0, 0, code, -1);
c93fad5e 1379 if (delete_during)
0e82af2d 1380 delete_in_dir(NULL, NULL, NULL, NULL);
2f03f956
AT
1381
1382 phase++;
1383 csum_length = SUM_LENGTH;
02b5cb23 1384 max_size = min_size = ignore_existing = ignore_non_existing = 0;
0492fdfb 1385 update_only = always_checksum = size_only = 0;
e1f67417 1386 ignore_times = 1;
6cc11982
WD
1387 if (append_mode) /* resend w/o append mode */
1388 append_mode = -1; /* ... but only longer files */
e6bc6f42 1389 make_backups = 0; /* avoid a duplicate backup for inplace processing */
2f03f956
AT
1390
1391 if (verbose > 2)
1392 rprintf(FINFO,"generate_files phase=%d\n",phase);
1393
7daccb8e 1394 write_int(f_out, -1);
2f03f956 1395
bc63ae3f
S
1396 /* files can cycle through the system more than once
1397 * to catch initial checksum errors */
ee1d11c4 1398 while ((i = get_redo_num(itemizing, code)) != -1) {
bc63ae3f 1399 struct file_struct *file = flist->files[i];
0e5665d3
WD
1400 if (local_name)
1401 strlcpy(fbuf, local_name, sizeof fbuf);
1402 else
6cbde57d 1403 f_name(file, fbuf);
e912bd4d 1404 recv_generator(fbuf, file, i, itemizing, maybe_ATTRS_REPORT,
0e5665d3 1405 code, f_out);
bc63ae3f 1406 }
2f03f956 1407
bc63ae3f 1408 phase++;
e90aab49
WD
1409 ignore_non_existing = save_ignore_non_existing;
1410 ignore_existing = save_ignore_existing;
cd908ef4 1411 make_backups = save_make_backups;
0492fdfb 1412
bc63ae3f
S
1413 if (verbose > 2)
1414 rprintf(FINFO,"generate_files phase=%d\n",phase);
2f03f956 1415
7daccb8e 1416 write_int(f_out, -1);
70352269
WD
1417 /* Reduce round-trip lag-time for a useless delay-updates phase. */
1418 if (protocol_version >= 29 && !delay_updates)
1419 write_int(f_out, -1);
6dff5992 1420
0438f100 1421 /* Read MSG_DONE for the redo phase (and any prior messages). */
ee1d11c4 1422 get_redo_num(itemizing, code);
6dff5992 1423
0438f100
WD
1424 if (protocol_version >= 29) {
1425 phase++;
1426 if (verbose > 2)
1427 rprintf(FINFO, "generate_files phase=%d\n", phase);
70352269
WD
1428 if (delay_updates)
1429 write_int(f_out, -1);
1430 /* Read MSG_DONE for delay-updates phase & prior messages. */
0438f100
WD
1431 get_redo_num(itemizing, code);
1432 }
1433
083acd49 1434 do_progress = save_do_progress;
59faec8b 1435 if (delete_after && !local_name && flist->count > 0)
ee1d11c4 1436 do_delete_pass(flist);
59faec8b 1437
0492fdfb
WD
1438 if ((need_retouch_dir_perms || need_retouch_dir_times)
1439 && !list_only && !local_name && !dry_run) {
18a11cfd 1440 int j = 0;
3ea9bbd6
WD
1441 /* Now we need to fix any directory permissions that were
1442 * modified during the transfer and/or re-set any tweaked
1443 * modified-time values. */
1444 for (i = 0; i < flist->count; i++) {
1445 struct file_struct *file = flist->files[i];
b435d717 1446
3ea9bbd6
WD
1447 if (!file->basename || !S_ISDIR(file->mode))
1448 continue;
1449 if (!need_retouch_dir_times && file->mode & S_IWUSR)
1450 continue;
6cbde57d 1451 recv_generator(f_name(file, NULL), file, i, itemizing,
e912bd4d 1452 maybe_ATTRS_REPORT, code, -1);
417099fa 1453 if (allowed_lull && !(++j % lull_mod))
ee1d11c4 1454 maybe_send_keepalive();
c2523a05 1455 else if (!(j % 200))
417099fa 1456 maybe_flush_socket();
3ea9bbd6 1457 }
6dff5992 1458 }
ef20efcb 1459 recv_generator(NULL, NULL, 0, 0, 0, code, -1);
6dff5992 1460
f75a53e7
WD
1461 if (max_delete > 0 && deletion_count > max_delete) {
1462 rprintf(FINFO,
1463 "Deletions stopped due to --max-delete limit (%d skipped)\n",
1464 deletion_count - max_delete);
1465 io_error |= IOERR_DEL_LIMIT;
1466 }
1467
6dff5992
WD
1468 if (verbose > 2)
1469 rprintf(FINFO,"generate_files finished\n");
2f03f956 1470}