The FCLIENT log code aviods putting the message into the daemon's
[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;
88b218fa 28extern int itemize_changes;
2f03f956 29extern int relative_paths;
716e73d4 30extern int keep_dirlinks;
2f03f956
AT
31extern int preserve_links;
32extern int am_root;
33extern int preserve_devices;
34extern int preserve_hard_links;
6744b62d
WD
35extern int preserve_perms;
36extern int preserve_uid;
37extern int preserve_gid;
3ea9bbd6
WD
38extern int preserve_times;
39extern int omit_dir_times;
fa13f396 40extern int delete_during;
2f03f956 41extern int update_only;
3d6feada 42extern int opt_ignore_existing;
cd6aa5b5
WD
43extern int inplace;
44extern int make_backups;
2f03f956
AT
45extern int csum_length;
46extern int ignore_times;
f83f0548 47extern int size_only;
7d1bfaf7 48extern OFF_T max_size;
2f03f956 49extern int io_timeout;
d04e9c51 50extern int protocol_version;
8e85be0a 51extern int fuzzy_basis;
2f03f956 52extern int always_checksum;
a7260c40 53extern char *partial_dir;
b7e8628c 54extern char *basis_dir[];
2be2fb3e 55extern int compare_dest;
59c95e42 56extern int link_dest;
5774786f
WD
57extern int whole_file;
58extern int local_server;
5774786f 59extern int list_only;
b9f592fb 60extern int read_batch;
5774786f
WD
61extern int only_existing;
62extern int orig_umask;
63extern int safe_symlinks;
a255c592 64extern long block_size; /* "long" because popt can't set an int32. */
2f03f956 65
7842418b 66extern struct filter_list_struct server_filter_list;
97f9dcae 67
b7e8628c 68static int unchanged_attrs(struct file_struct *file, STRUCT_STAT *st)
2f03f956 69{
b7e8628c
WD
70 if (preserve_perms
71 && (st->st_mode & CHMOD_BITS) != (file->mode & CHMOD_BITS))
84acca07 72 return 0;
bb24028f 73
b7e8628c
WD
74 if (am_root && preserve_uid && st->st_uid != file->uid)
75 return 0;
bb24028f 76
b7e8628c
WD
77 if (preserve_gid && file->gid != GID_NONE && st->st_gid != file->gid)
78 return 0;
79
80 return 1;
81}
82
06a1dbad 83
c557eb8c
WD
84#define SID_UPDATING ITEM_UPDATING
85#define SID_REPORT_CHECKSUM ITEM_REPORT_CHECKSUM
86#define SID_NO_DEST_AND_NO_UPDATE (1<<16)
06a1dbad 87
c557eb8c
WD
88static void itemize(struct file_struct *file, int statret, STRUCT_STAT *st,
89 int32 sflags, int f_out, int ndx)
06a1dbad 90{
c557eb8c 91 int iflags = sflags & (SID_UPDATING | SID_REPORT_CHECKSUM);
06a1dbad 92
c557eb8c
WD
93 if (statret >= 0) {
94 if (S_ISREG(file->mode) && file->length != st->st_size)
95 iflags |= ITEM_REPORT_SIZE;
88b218fa
WD
96 if (!(sflags & SID_NO_DEST_AND_NO_UPDATE)) {
97 int keep_time = !preserve_times ? 0
98 : S_ISDIR(file->mode) ? !omit_dir_times
99 : !S_ISLNK(file->mode);
100
101 if ((iflags & ITEM_UPDATING && !keep_time)
102 || (keep_time && file->modtime != st->st_mtime))
103 iflags |= ITEM_REPORT_TIME;
104 if (preserve_perms && file->mode != st->st_mode)
105 iflags |= ITEM_REPORT_PERMS;
106 if (preserve_uid && am_root && file->uid != st->st_uid)
107 iflags |= ITEM_REPORT_OWNER;
108 if (preserve_gid && file->gid != GID_NONE
109 && st->st_gid != file->gid)
110 iflags |= ITEM_REPORT_GROUP;
111 }
c557eb8c 112 } else
88b218fa 113 iflags |= ITEM_IS_NEW | ITEM_UPDATING;
c557eb8c 114
8a8356b7 115 if ((iflags || verbose > 1) && !read_batch) {
c557eb8c
WD
116 if (ndx >= 0)
117 write_int(f_out, ndx);
8a8356b7 118 write_shortint(f_out, iflags);
06a1dbad
WD
119 }
120}
121
122
b7e8628c
WD
123/* Perform our quick-check heuristic for determining if a file is unchanged. */
124static int unchanged_file(char *fn, struct file_struct *file, STRUCT_STAT *st)
125{
126 if (st->st_size != file->length)
127 return 0;
59c95e42 128
2cda2560 129 /* if always checksum is set then we use the checksum instead
2f03f956
AT
130 of the file time to determine whether to sync */
131 if (always_checksum && S_ISREG(st->st_mode)) {
132 char sum[MD4_SUM_LENGTH];
b7e8628c 133 file_checksum(fn, sum, st->st_size);
728d0922 134 return memcmp(sum, file->u.sum, protocol_version < 21 ? 2
84acca07 135 : MD4_SUM_LENGTH) == 0;
2f03f956
AT
136 }
137
cc1e997d 138 if (size_only)
84acca07 139 return 1;
2f03f956 140
cc1e997d 141 if (ignore_times)
84acca07 142 return 0;
cc1e997d 143
84acca07 144 return cmp_modtime(st->st_mtime, file->modtime) == 0;
2f03f956
AT
145}
146
147
ec8290c8 148/*
195bd906 149 * set (initialize) the size entries in the per-file sum_struct
ec8290c8 150 * calculating dynamic block and checksum sizes.
195bd906 151 *
ec8290c8 152 * This is only called from generate_and_send_sums() but is a separate
195bd906
S
153 * function to encapsulate the logic.
154 *
155 * The block size is a rounded square root of file length.
156 *
157 * The checksum size is determined according to:
158 * blocksum_bits = BLOCKSUM_EXP + 2*log2(file_len) - log2(block_len)
159 * provided by Donovan Baarda which gives a probability of rsync
160 * algorithm corrupting data and falling back using the whole md4
161 * checksums.
162 *
163 * This might be made one of several selectable heuristics.
164 */
1490812a 165static void sum_sizes_sqroot(struct sum_struct *sum, int64 len)
195bd906 166{
a255c592 167 int32 blength;
da9d12f5 168 int s2length;
195bd906 169
a255c592 170 if (block_size)
195bd906 171 blength = block_size;
a255c592 172 else if (len <= BLOCK_SIZE * BLOCK_SIZE)
195bd906 173 blength = BLOCK_SIZE;
a255c592
WD
174 else {
175 int32 c;
1490812a 176 int64 l;
eae7165c
WD
177 int cnt;
178 for (c = 1, l = len, cnt = 0; l >>= 2; c <<= 1, cnt++) {}
179 if (cnt >= 31 || c >= MAX_BLOCK_SIZE)
180 blength = MAX_BLOCK_SIZE;
181 else {
182 blength = 0;
183 do {
184 blength |= c;
1490812a 185 if (len < (int64)blength * blength)
eae7165c
WD
186 blength &= ~c;
187 c >>= 1;
188 } while (c >= 8); /* round to multiple of 8 */
189 blength = MAX(blength, BLOCK_SIZE);
195bd906 190 }
195bd906
S
191 }
192
d04e9c51 193 if (protocol_version < 27) {
195bd906
S
194 s2length = csum_length;
195 } else if (csum_length == SUM_LENGTH) {
196 s2length = SUM_LENGTH;
197 } else {
a255c592 198 int32 c;
1490812a 199 int64 l;
da9d12f5 200 int b = BLOCKSUM_BIAS;
a255c592
WD
201 for (l = len; l >>= 1; b += 2) {}
202 for (c = blength; c >>= 1 && b; b--) {}
203 /* add a bit, subtract rollsum, round up. */
204 s2length = (b + 1 - 32 + 7) / 8; /* --optimize in compiler-- */
195bd906
S
205 s2length = MAX(s2length, csum_length);
206 s2length = MIN(s2length, SUM_LENGTH);
207 }
208
209 sum->flength = len;
210 sum->blength = blength;
211 sum->s2length = s2length;
212 sum->count = (len + (blength - 1)) / blength;
213 sum->remainder = (len % blength);
214
215 if (sum->count && verbose > 2) {
a255c592
WD
216 rprintf(FINFO,
217 "count=%.0f rem=%ld blength=%ld s2length=%d flength=%.0f\n",
218 (double)sum->count, (long)sum->remainder, (long)sum->blength,
da9d12f5 219 sum->s2length, (double)sum->flength);
195bd906
S
220 }
221}
80605142 222
bceec82f 223
80605142
WD
224/*
225 * Generate and send a stream of signatures/checksums that describe a buffer
e66dfd18 226 *
80605142
WD
227 * Generate approximately one checksum every block_len bytes.
228 */
cd6aa5b5 229static void generate_and_send_sums(int fd, OFF_T len, int f_out, int f_copy)
2f03f956 230{
a1cbe76e 231 int32 i;
6e45e1dd 232 struct map_struct *mapbuf;
80605142 233 struct sum_struct sum;
2f03f956
AT
234 OFF_T offset = 0;
235
423dba8e 236 sum_sizes_sqroot(&sum, len);
e66dfd18 237
6e45e1dd 238 if (len > 0)
96d910c7 239 mapbuf = map_file(fd, len, MAX_MAP_SIZE, sum.blength);
6e45e1dd
WD
240 else
241 mapbuf = NULL;
242
fc0257c9 243 write_sum_head(f_out, &sum);
2f03f956 244
80605142 245 for (i = 0; i < sum.count; i++) {
a255c592 246 int32 n1 = (int32)MIN(len, (OFF_T)sum.blength);
6e45e1dd 247 char *map = map_ptr(mapbuf, offset, n1);
80605142
WD
248 uint32 sum1 = get_checksum1(map, n1);
249 char sum2[SUM_LENGTH];
2f03f956 250
cd6aa5b5
WD
251 if (f_copy >= 0)
252 full_write(f_copy, map, n1);
253
80605142 254 get_checksum2(map, n1, sum2);
2f03f956 255
80605142 256 if (verbose > 3) {
e66dfd18 257 rprintf(FINFO,
a255c592
WD
258 "chunk[%.0f] offset=%.0f len=%ld sum1=%08lx\n",
259 (double)i, (double)offset, (long)n1,
0e36d9da 260 (unsigned long)sum1);
80605142
WD
261 }
262 write_int(f_out, sum1);
fc0257c9 263 write_buf(f_out, sum2, sum.s2length);
2f03f956
AT
264 len -= n1;
265 offset += n1;
266 }
6e45e1dd
WD
267
268 if (mapbuf)
269 unmap_file(mapbuf);
2f03f956
AT
270}
271
06a1dbad 272
8e85be0a
WD
273/* Try to find a filename in the same dir as "fname" with a similar name. */
274static int find_fuzzy(struct file_struct *file, struct file_list *dirlist)
275{
276 int fname_len, fname_suf_len;
277 const char *fname_suf, *fname = file->basename;
278 uint32 lowest_dist = 0x7FFFFFFF;
279 int j, lowest_j = -1;
280
281 fname_len = strlen(fname);
282 fname_suf = find_filename_suffix(fname, fname_len, &fname_suf_len);
283
284 for (j = 0; j < dirlist->count; j++) {
285 struct file_struct *fp = dirlist->files[j];
286 const char *suf, *name;
287 int len, suf_len;
288 uint32 dist;
289
290 if (!S_ISREG(fp->mode) || !fp->length
291 || fp->flags & FLAG_NO_FUZZY)
292 continue;
293
294 name = fp->basename;
295
296 if (fp->length == file->length
297 && fp->modtime == file->modtime) {
298 if (verbose > 4) {
299 rprintf(FINFO,
300 "fuzzy size/modtime match for %s\n",
301 name);
302 }
303 return j;
304 }
305
306 len = strlen(name);
307 suf = find_filename_suffix(name, len, &suf_len);
308
309 dist = fuzzy_distance(name, len, fname, fname_len);
310 /* Add some extra weight to how well the suffixes match. */
311 dist += fuzzy_distance(suf, suf_len, fname_suf, fname_suf_len)
312 * 10;
313 if (verbose > 4) {
314 rprintf(FINFO, "fuzzy distance for %s = %d.%05d\n",
315 name, (int)(dist>>16), (int)(dist&0xFFFF));
316 }
317 if (dist <= lowest_dist) {
318 lowest_dist = dist;
319 lowest_j = j;
320 }
321 }
322
323 return lowest_j;
324}
325
2f03f956 326
0492fdfb
WD
327/* Acts on flist->file's ndx'th item, whose name is fname. If a directory,
328 * make sure it exists, and has the right permissions/timestamp info. For
329 * all other non-regular files (symlinks, etc.) we create them here. For
330 * regular files that have changed, we try to find a basis file and then
331 * start sending checksums.
ef1aa910 332 *
0492fdfb
WD
333 * Note that f_out is set to -1 when doing final directory-permission and
334 * modification-time repair. */
fa13f396 335static void recv_generator(char *fname, struct file_list *flist,
0492fdfb 336 struct file_struct *file, int ndx,
41cfde6b 337 int f_out, int f_out_name)
2cda2560 338{
df337831 339 static int missing_below = -1;
8e85be0a
WD
340 static char *fuzzy_dirname = NULL;
341 static struct file_list *fuzzy_dirlist = NULL;
342 struct file_struct *fuzzy_file = NULL;
41cfde6b 343 int fd = -1, f_copy = -1;
89f7eff3 344 STRUCT_STAT st, partial_st;
41cfde6b 345 struct file_struct *back_file = NULL;
e7d13fe5 346 int statret, stat_errno;
41cfde6b 347 char *fnamecmp, *partialptr, *backupptr = NULL;
375a4556 348 char fnamecmpbuf[MAXPATHLEN];
41cfde6b 349 uchar fnamecmp_type;
88b218fa 350 int maybe_DEL_TERSE = itemize_changes ? 0 : DEL_TERSE;
8a8356b7 351 int maybe_PERMS_REPORT = itemize_changes ? 0 : PERMS_REPORT;
f7632fc6 352
dfd5ba6a
WD
353 if (list_only)
354 return;
2f03f956 355
8e85be0a
WD
356 if (!fname) {
357 if (fuzzy_dirlist) {
358 flist_free(fuzzy_dirlist);
359 fuzzy_dirlist = NULL;
360 fuzzy_dirname = NULL;
361 }
362 if (missing_below >= 0) {
363 dry_run--;
364 missing_below = -1;
365 }
366 return;
367 }
368
4875d6b6
WD
369 if (verbose > 2) {
370 rprintf(FINFO, "recv_generator(%s,%d)\n",
371 safe_fname(fname), ndx);
372 }
2f03f956 373
7842418b
WD
374 if (server_filter_list.head
375 && check_filter(&server_filter_list, fname,
376 S_ISDIR(file->mode)) < 0) {
3e35c34b
WD
377 if (verbose) {
378 rprintf(FINFO, "skipping server-excluded file \"%s\"\n",
ecc81fce 379 safe_fname(fname));
3e35c34b 380 }
97f9dcae 381 return;
3e35c34b 382 }
97f9dcae 383
8e85be0a 384 if (missing_below >= 0 && file->dir.depth <= missing_below) {
df337831
WD
385 dry_run--;
386 missing_below = -1;
387 }
73f7af0e
WD
388 if (dry_run > 1) {
389 statret = -1;
390 stat_errno = ENOENT;
391 } else {
8e85be0a
WD
392 if (fuzzy_basis && S_ISREG(file->mode)) {
393 char *dn = file->dirname ? file->dirname : ".";
394 /* Yes, identical dirnames are guaranteed to have
395 * identical pointers at this point. */
396 if (fuzzy_dirname != dn) {
397 if (fuzzy_dirlist)
398 flist_free(fuzzy_dirlist);
399 fuzzy_dirname = dn;
400 fuzzy_dirlist = get_dirlist(fuzzy_dirname, 1);
401 }
402 }
403
73f7af0e
WD
404 statret = link_stat(fname, &st,
405 keep_dirlinks && S_ISDIR(file->mode));
406 stat_errno = errno;
407 }
63787382 408
a7260c40 409 if (only_existing && statret == -1 && stat_errno == ENOENT) {
1347d512 410 /* we only want to update existing files */
ecc81fce
WD
411 if (verbose > 1) {
412 rprintf(FINFO, "not creating new file \"%s\"\n",
413 safe_fname(fname));
414 }
1347d512
AT
415 return;
416 }
417
d9b4d267
WD
418 if (statret == 0 && !preserve_perms
419 && S_ISDIR(st.st_mode) == S_ISDIR(file->mode)) {
4df9f368 420 /* if the file exists already and we aren't perserving
2cda2560
WD
421 * permissions then act as though the remote end sent
422 * us the file permissions we already have */
67e78a82
WD
423 file->mode = (file->mode & ~CHMOD_BITS)
424 | (st.st_mode & CHMOD_BITS);
4df9f368
AT
425 }
426
2f03f956 427 if (S_ISDIR(file->mode)) {
2cda2560
WD
428 /* The file to be received is a directory, so we need
429 * to prepare appropriately. If there is already a
430 * file of that name and it is *not* a directory, then
431 * we need to delete it. If it doesn't exist, then
027428eb 432 * (perhaps recursively) create it. */
2f03f956 433 if (statret == 0 && !S_ISDIR(st.st_mode)) {
8a8356b7 434 delete_file(fname, st.st_mode, maybe_DEL_TERSE);
2f03f956
AT
435 statret = -1;
436 }
df337831
WD
437 if (dry_run && statret != 0 && missing_below < 0) {
438 missing_below = file->dir.depth;
439 dry_run++;
440 }
e3bcd893 441 if (protocol_version >= 29 && f_out != -1)
c557eb8c 442 itemize(file, statret, &st, 0, f_out, ndx);
2f03f956 443 if (statret != 0 && do_mkdir(fname,file->mode) != 0 && errno != EEXIST) {
027428eb
WD
444 if (!relative_paths || errno != ENOENT
445 || create_directory_path(fname, orig_umask) < 0
446 || do_mkdir(fname, file->mode) < 0) {
d62bcc17
WD
447 rsyserr(FERROR, errno,
448 "recv_generator: mkdir %s failed",
449 full_fname(fname));
2f03f956
AT
450 }
451 }
716e73d4 452 if (set_perms(fname, file, statret ? NULL : &st, 0)
e3bcd893 453 && verbose && protocol_version < 29 && f_out != -1)
ecc81fce 454 rprintf(FINFO, "%s/\n", safe_fname(fname));
c93fad5e 455 if (delete_during && f_out != -1 && csum_length != SUM_LENGTH
31937d36
WD
456 && (file->flags & FLAG_DEL_HERE))
457 delete_in_dir(flist, fname, file);
2f03f956 458 return;
06a1dbad
WD
459 }
460
461 if (max_size && file->length > max_size) {
4875d6b6
WD
462 if (verbose > 1) {
463 rprintf(FINFO, "%s is over max-size\n",
464 safe_fname(fname));
465 }
7d1bfaf7 466 return;
2f03f956
AT
467 }
468
469 if (preserve_links && S_ISLNK(file->mode)) {
4f5b0756 470#ifdef SUPPORT_LINKS
728d0922 471 if (safe_symlinks && unsafe_symlink(file->u.link, fname)) {
2f03f956 472 if (verbose) {
4875d6b6
WD
473 rprintf(FINFO,
474 "ignoring unsafe symlink %s -> \"%s\"\n",
475 full_fname(fname),
476 safe_fname(file->u.link));
2f03f956
AT
477 }
478 return;
479 }
480 if (statret == 0) {
7e38410e
WD
481 char lnk[MAXPATHLEN];
482 int len;
483
8a8356b7 484 if (!S_ISDIR(st.st_mode)
7e38410e
WD
485 && (len = readlink(fname, lnk, MAXPATHLEN-1)) > 0) {
486 lnk[len] = 0;
85d4d142
MP
487 /* A link already pointing to the
488 * right place -- no further action
489 * required. */
7e38410e 490 if (strcmp(lnk, file->u.link) == 0) {
e3bcd893 491 if (protocol_version >= 29) {
c557eb8c
WD
492 itemize(file, 0, &st, 0,
493 f_out, ndx);
494 }
c41b52c4 495 set_perms(fname, file, &st,
8a8356b7 496 maybe_PERMS_REPORT);
2f03f956
AT
497 return;
498 }
2cda2560 499 }
7e38410e
WD
500 /* Not the right symlink (or not a symlink), so
501 * delete it. */
88b218fa 502 if (S_ISLNK(st.st_mode))
8a8356b7 503 delete_file(fname, st.st_mode, DEL_TERSE);
88b218fa 504 else {
8a8356b7 505 delete_file(fname, st.st_mode, maybe_DEL_TERSE);
88b218fa
WD
506 statret = -1;
507 }
2f03f956 508 }
728d0922 509 if (do_symlink(file->u.link,fname) != 0) {
d62bcc17 510 rsyserr(FERROR, errno, "symlink %s -> \"%s\" failed",
ecc81fce 511 full_fname(fname), safe_fname(file->u.link));
2f03f956
AT
512 } else {
513 set_perms(fname,file,NULL,0);
e3bcd893 514 if (protocol_version >= 29) {
c557eb8c
WD
515 itemize(file, statret, &st, SID_UPDATING,
516 f_out, ndx);
517 } else if (verbose) {
518 rprintf(FINFO, "%s -> %s\n", safe_fname(fname),
519 safe_fname(file->u.link));
2f03f956
AT
520 }
521 }
522#endif
523 return;
524 }
525
2f03f956 526 if (am_root && preserve_devices && IS_DEVICE(file->mode)) {
2cda2560 527 if (statret != 0 ||
2f03f956 528 st.st_mode != file->mode ||
3915fd75 529 st.st_rdev != file->u.rdev) {
88b218fa 530 if (IS_DEVICE(st.st_mode))
8a8356b7 531 delete_file(fname, st.st_mode, DEL_TERSE);
88b218fa 532 else {
8a8356b7 533 delete_file(fname, st.st_mode, maybe_DEL_TERSE);
88b218fa 534 statret = -1;
c557eb8c 535 }
d62bcc17 536 if (verbose > 2) {
2f03f956 537 rprintf(FINFO,"mknod(%s,0%o,0x%x)\n",
ecc81fce
WD
538 safe_fname(fname),
539 (int)file->mode, (int)file->u.rdev);
d62bcc17 540 }
728d0922 541 if (do_mknod(fname,file->mode,file->u.rdev) != 0) {
d62bcc17
WD
542 rsyserr(FERROR, errno, "mknod %s failed",
543 full_fname(fname));
2f03f956
AT
544 } else {
545 set_perms(fname,file,NULL,0);
88b218fa
WD
546 if (protocol_version >= 29) {
547 itemize(file, statret, &st, SID_UPDATING,
548 f_out, ndx);
549 } else if (verbose) {
ecc81fce
WD
550 rprintf(FINFO, "%s\n",
551 safe_fname(fname));
552 }
2f03f956
AT
553 }
554 } else {
e3bcd893 555 if (protocol_version >= 29) {
c557eb8c
WD
556 itemize(file, statret, &st, 0,
557 f_out, ndx);
558 }
8a8356b7 559 set_perms(fname, file, &st, maybe_PERMS_REPORT);
2f03f956
AT
560 }
561 return;
562 }
2f03f956 563
6dff5992 564 if (preserve_hard_links && hard_link_check(file, HL_CHECK_MASTER))
2f03f956 565 return;
2f03f956
AT
566
567 if (!S_ISREG(file->mode)) {
ecc81fce
WD
568 rprintf(FINFO, "skipping non-regular file \"%s\"\n",
569 safe_fname(fname));
2f03f956
AT
570 return;
571 }
572
375a4556 573 fnamecmp = fname;
41cfde6b 574 fnamecmp_type = FNAMECMP_FNAME;
375a4556 575
06a1dbad 576 if (statret != 0 && basis_dir[0] != NULL) {
b7e8628c
WD
577 int fallback_match = -1;
578 int match_level = 0;
579 int i = 0;
580 do {
581 pathjoin(fnamecmpbuf, sizeof fnamecmpbuf,
582 basis_dir[i], fname);
583 if (link_stat(fnamecmpbuf, &st, 0) == 0
584 && S_ISREG(st.st_mode)) {
585 statret = 0;
586 if (link_dest) {
587 if (!match_level) {
588 fallback_match = i;
589 match_level = 1;
590 } else if (match_level == 2
591 && !unchanged_attrs(file, &st))
592 continue;
593 if (!unchanged_file(fnamecmpbuf, file, &st))
594 continue;
595 fallback_match = i;
596 match_level = 2;
597 if (!unchanged_attrs(file, &st))
598 continue;
599 }
600 match_level = 3;
601 break;
602 }
603 } while (basis_dir[++i] != NULL);
604 if (statret == 0) {
605 if (match_level < 3) {
606 i = fallback_match;
607 pathjoin(fnamecmpbuf, sizeof fnamecmpbuf,
608 basis_dir[i], fname);
609 }
4f5b0756 610#ifdef HAVE_LINK
b7e8628c 611 if (link_dest && match_level == 3 && !dry_run) {
e7d13fe5
WD
612 if (do_link(fnamecmpbuf, fname) < 0) {
613 if (verbose) {
614 rsyserr(FINFO, errno,
615 "link %s => %s",
4875d6b6 616 full_fname(fnamecmpbuf),
e7d13fe5
WD
617 safe_fname(fname));
618 }
619 fnamecmp = fnamecmpbuf;
2be2fb3e 620 fnamecmp_type = i;
e7bc9b64 621 }
e7d13fe5 622 } else
59c95e42 623#endif
41cfde6b 624 {
e7d13fe5 625 fnamecmp = fnamecmpbuf;
2be2fb3e 626 fnamecmp_type = i;
41cfde6b 627 }
e7d13fe5
WD
628 }
629 }
630
631 if (statret == 0 && !S_ISREG(st.st_mode)) {
8a8356b7 632 if (delete_file(fname, st.st_mode, maybe_DEL_TERSE) != 0)
e7d13fe5
WD
633 return;
634 statret = -1;
635 stat_errno = ENOENT;
375a4556
DD
636 }
637
9d954dca 638 if (partial_dir && (partialptr = partial_dir_fname(fname)) != NULL
72c19bb3
WD
639 && link_stat(partialptr, &partial_st, 0) == 0
640 && S_ISREG(partial_st.st_mode)) {
06a1dbad 641 if (statret != 0)
72c19bb3
WD
642 goto prepare_to_open;
643 } else
644 partialptr = NULL;
89f7eff3 645
06a1dbad 646 if (statret != 0 && fuzzy_basis && dry_run <= 1) {
8e85be0a
WD
647 int j = find_fuzzy(file, fuzzy_dirlist);
648 if (j >= 0) {
649 fuzzy_file = fuzzy_dirlist->files[j];
650 f_name_to(fuzzy_file, fnamecmpbuf);
651 if (verbose > 2) {
652 rprintf(FINFO, "fuzzy basis selected for %s: %s\n",
653 safe_fname(fname), safe_fname(fnamecmpbuf));
654 }
655 st.st_mode = fuzzy_file->mode;
656 st.st_size = fuzzy_file->length;
657 st.st_mtime = fuzzy_file->modtime;
658 statret = 0;
659 fnamecmp = fnamecmpbuf;
660 fnamecmp_type = FNAMECMP_FUZZY;
661 }
662 }
663
06a1dbad 664 if (statret != 0) {
6dff5992
WD
665 if (preserve_hard_links && hard_link_check(file, HL_SKIP))
666 return;
41cfde6b
WD
667 if (stat_errno == ENOENT)
668 goto notify_others;
669 if (verbose > 1) {
e7d13fe5
WD
670 rsyserr(FERROR, stat_errno,
671 "recv_generator: failed to stat %s",
d62bcc17 672 full_fname(fname));
2f03f956
AT
673 }
674 return;
675 }
676
41cfde6b 677 if (opt_ignore_existing && fnamecmp_type == FNAMECMP_FNAME) {
3d6feada 678 if (verbose > 1)
ecc81fce 679 rprintf(FINFO, "%s exists\n", safe_fname(fname));
3d6feada 680 return;
2cda2560 681 }
3d6feada 682
41cfde6b 683 if (update_only && fnamecmp_type == FNAMECMP_FNAME
d3a4375f 684 && cmp_modtime(st.st_mtime, file->modtime) > 0) {
2f03f956 685 if (verbose > 1)
ecc81fce 686 rprintf(FINFO, "%s is newer\n", safe_fname(fname));
2f03f956
AT
687 return;
688 }
689
2be2fb3e 690 if (!compare_dest && fnamecmp_type <= FNAMECMP_BASIS_DIR_HIGH)
b7e8628c 691 ;
8e85be0a
WD
692 else if (fnamecmp_type == FNAMECMP_FUZZY)
693 ;
b7e8628c 694 else if (unchanged_file(fnamecmp, file, &st)) {
e3bcd893 695 if (protocol_version >= 29) {
c557eb8c
WD
696 itemize(file, statret, &st,
697 fnamecmp_type == FNAMECMP_FNAME
698 ? 0 : SID_NO_DEST_AND_NO_UPDATE,
699 f_out, ndx);
06a1dbad 700 }
41cfde6b 701 if (fnamecmp_type == FNAMECMP_FNAME)
8a8356b7 702 set_perms(fname, file, &st, maybe_PERMS_REPORT);
2f03f956
AT
703 return;
704 }
705
89f7eff3 706prepare_to_open:
9d954dca
WD
707 if (partialptr) {
708 st = partial_st;
709 fnamecmp = partialptr;
710 fnamecmp_type = FNAMECMP_PARTIAL_DIR;
711 statret = 0;
712 }
713
3841a04e
WD
714 if (dry_run || read_batch)
715 goto notify_others;
716 if (whole_file > 0) {
06a1dbad
WD
717 if (statret == 0)
718 statret = 1;
41cfde6b 719 goto notify_others;
2f03f956
AT
720 }
721
8e85be0a
WD
722 if (fuzzy_basis) {
723 int j = flist_find(fuzzy_dirlist, file);
724 if (j >= 0) /* don't use changing file as future fuzzy basis */
725 fuzzy_dirlist->files[j]->flags |= FLAG_NO_FUZZY;
726 }
727
2cda2560 728 /* open the file */
8c9fd200 729 fd = do_open(fnamecmp, O_RDONLY, 0);
2f03f956
AT
730
731 if (fd == -1) {
d62bcc17
WD
732 rsyserr(FERROR, errno, "failed to open %s, continuing",
733 full_fname(fnamecmp));
cd6aa5b5 734 pretend_missing:
60be6acf 735 /* pretend the file didn't exist */
6dff5992
WD
736 if (preserve_hard_links && hard_link_check(file, HL_SKIP))
737 return;
41cfde6b
WD
738 statret = -1;
739 goto notify_others;
2f03f956
AT
740 }
741
cd6aa5b5
WD
742 if (inplace && make_backups) {
743 if (!(backupptr = get_backup_name(fname))) {
744 close(fd);
745 return;
746 }
7842418b 747 if (!(back_file = make_file(fname, NULL, NO_FILTERS))) {
cd6aa5b5
WD
748 close(fd);
749 goto pretend_missing;
750 }
751 if (robust_unlink(backupptr) && errno != ENOENT) {
752 rsyserr(FERROR, errno, "unlink %s",
753 full_fname(backupptr));
754 free(back_file);
755 close(fd);
756 return;
757 }
758 if ((f_copy = do_open(backupptr,
759 O_WRONLY | O_CREAT | O_TRUNC | O_EXCL, 0600)) < 0) {
760 rsyserr(FERROR, errno, "open %s",
761 full_fname(backupptr));
762 free(back_file);
763 close(fd);
764 return;
765 }
41cfde6b 766 fnamecmp_type = FNAMECMP_BACKUP;
cd6aa5b5
WD
767 }
768
dfd5ba6a 769 if (verbose > 3) {
ecc81fce
WD
770 rprintf(FINFO, "gen mapped %s of size %.0f\n",
771 safe_fname(fnamecmp), (double)st.st_size);
dfd5ba6a 772 }
2f03f956 773
2f03f956 774 if (verbose > 2)
0492fdfb 775 rprintf(FINFO, "generating and sending sums for %d\n", ndx);
2f03f956 776
41cfde6b 777notify_others:
0492fdfb 778 write_int(f_out, ndx);
e3bcd893
WD
779 if (protocol_version >= 29) {
780 itemize(file, statret, &st, SID_UPDATING
781 | (always_checksum ? SID_REPORT_CHECKSUM : 0),
782 f_out, -1);
783 if (inplace && !read_batch)
784 write_byte(f_out, fnamecmp_type);
785 }
8e85be0a 786 if (f_out_name >= 0) {
41cfde6b 787 write_byte(f_out_name, fnamecmp_type);
8e85be0a
WD
788 if (fnamecmp_type == FNAMECMP_FUZZY) {
789 uchar lenbuf[3], *lb = lenbuf;
790 int len = strlen(fuzzy_file->basename);
791 if (len > 0x7F) {
792#if MAXPATHLEN > 0x7FFF
793 *lb++ = len / 0x10000 + 0x80;
794 *lb++ = len / 0x100;
795#else
796 *lb++ = len / 0x100 + 0x80;
797#endif
798 }
799 *lb = len;
800 write_buf(f_out_name, lenbuf, lb - lenbuf + 1);
801 write_buf(f_out_name, fuzzy_file->basename, len);
802 }
803 }
cd6aa5b5 804
41cfde6b
WD
805 if (dry_run || read_batch)
806 return;
807
808 if (statret == 0) {
809 generate_and_send_sums(fd, st.st_size, f_out, f_copy);
2f03f956 810
41cfde6b
WD
811 if (f_copy >= 0) {
812 close(f_copy);
813 set_perms(backupptr, back_file, NULL, 0);
814 if (verbose > 1) {
815 rprintf(FINFO, "backed up %s to %s\n",
4875d6b6 816 safe_fname(fname), safe_fname(backupptr));
41cfde6b
WD
817 }
818 free(back_file);
819 }
820
821 close(fd);
822 } else
823 write_sum_head(f_out, NULL);
2f03f956
AT
824}
825
826
41cfde6b
WD
827void generate_files(int f_out, struct file_list *flist, char *local_name,
828 int f_out_name)
2f03f956
AT
829{
830 int i;
e1f67417 831 int phase = 0;
968c8030 832 char fbuf[MAXPATHLEN];
3ea9bbd6
WD
833 int need_retouch_dir_times = preserve_times && !omit_dir_times;
834 int need_retouch_dir_perms = 0;
0492fdfb
WD
835 int save_only_existing = only_existing;
836 int save_opt_ignore_existing = opt_ignore_existing;
2f03f956 837
45e08edb
WD
838 if (verbose > 2) {
839 rprintf(FINFO, "generator starting pid=%ld count=%d\n",
840 (long)getpid(), flist->count);
841 }
2f03f956 842
3e7053ac
MP
843 if (verbose >= 2) {
844 rprintf(FINFO,
f38bd4a0 845 whole_file > 0
3e7053ac
MP
846 ? "delta-transmission disabled for local transfer or --whole-file\n"
847 : "delta transmission enabled\n");
848 }
2cda2560 849
0492fdfb
WD
850 /* We expect to just sit around now, so don't exit on a timeout.
851 * If we really get a timeout then the other process should exit. */
a57873b7
AT
852 io_timeout = 0;
853
2f03f956
AT
854 for (i = 0; i < flist->count; i++) {
855 struct file_struct *file = flist->files[i];
dfd5ba6a 856 struct file_struct copy;
2f03f956 857
dfd5ba6a
WD
858 if (!file->basename)
859 continue;
0492fdfb
WD
860
861 /* We need to ensure that any dirs we create have writeable
862 * permissions during the time we are putting files within
863 * them. This is then fixed after the transfer is done. */
dfd5ba6a
WD
864 if (!am_root && S_ISDIR(file->mode) && !(file->mode & S_IWUSR)) {
865 copy = *file;
dfd5ba6a
WD
866 copy.mode |= S_IWUSR; /* user write */
867 file = &copy;
3ea9bbd6 868 need_retouch_dir_perms = 1;
2f03f956
AT
869 }
870
3fef5364 871 recv_generator(local_name ? local_name : f_name_to(file, fbuf),
fa13f396 872 flist, file, i, f_out, f_out_name);
2f03f956 873 }
8e85be0a 874 recv_generator(NULL, NULL, NULL, 0, -1, -1);
c93fad5e 875 if (delete_during)
31937d36 876 delete_in_dir(NULL, NULL, NULL);
2f03f956
AT
877
878 phase++;
879 csum_length = SUM_LENGTH;
0492fdfb
WD
880 only_existing = max_size = opt_ignore_existing = 0;
881 update_only = always_checksum = size_only = 0;
e1f67417 882 ignore_times = 1;
2f03f956
AT
883
884 if (verbose > 2)
885 rprintf(FINFO,"generate_files phase=%d\n",phase);
886
7daccb8e 887 write_int(f_out, -1);
2f03f956 888
bc63ae3f
S
889 /* files can cycle through the system more than once
890 * to catch initial checksum errors */
b9b15fb1 891 while ((i = get_redo_num()) != -1) {
bc63ae3f 892 struct file_struct *file = flist->files[i];
3fef5364 893 recv_generator(local_name ? local_name : f_name_to(file, fbuf),
fa13f396 894 flist, file, i, f_out, f_out_name);
bc63ae3f 895 }
2f03f956 896
bc63ae3f 897 phase++;
0492fdfb
WD
898 only_existing = save_only_existing;
899 opt_ignore_existing = save_opt_ignore_existing;
900
bc63ae3f
S
901 if (verbose > 2)
902 rprintf(FINFO,"generate_files phase=%d\n",phase);
2f03f956 903
7daccb8e 904 write_int(f_out, -1);
6dff5992 905
b0da4b23
WD
906 /* Read post-redo-phase MSG_DONE and any prior messages. */
907 get_redo_num();
908
6dff5992
WD
909 if (preserve_hard_links)
910 do_hard_links();
911
0492fdfb
WD
912 if ((need_retouch_dir_perms || need_retouch_dir_times)
913 && !list_only && !local_name && !dry_run) {
3ea9bbd6
WD
914 /* Now we need to fix any directory permissions that were
915 * modified during the transfer and/or re-set any tweaked
916 * modified-time values. */
917 for (i = 0; i < flist->count; i++) {
918 struct file_struct *file = flist->files[i];
919 if (!file->basename || !S_ISDIR(file->mode))
920 continue;
921 if (!need_retouch_dir_times && file->mode & S_IWUSR)
922 continue;
923 recv_generator(local_name ? local_name : f_name(file),
924 flist, file, i, -1, -1);
925 }
6dff5992 926 }
8e85be0a 927 recv_generator(NULL, NULL, NULL, 0, -1, -1);
6dff5992
WD
928
929 if (verbose > 2)
930 rprintf(FINFO,"generate_files finished\n");
2f03f956 931}