When a hard-linked device is duplicating the data from the prior
[rsync/rsync.git] / flist.c
... / ...
CommitLineData
1/*
2 * Generate and receive file lists.
3 *
4 * Copyright (C) 1996 Andrew Tridgell
5 * Copyright (C) 1996 Paul Mackerras
6 * Copyright (C) 2001, 2002 Martin Pool <mbp@samba.org>
7 * Copyright (C) 2002-2007 Wayne Davison
8 *
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License version 2 as
11 * published by the Free Software Foundation.
12 *
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
17 *
18 * You should have received a copy of the GNU General Public License along
19 * with this program; if not, write to the Free Software Foundation, Inc.,
20 * 51 Franklin Street - Fifth Floor, Boston, MA 02110-1301, USA.
21 */
22
23#include "rsync.h"
24#include "rounding.h"
25#include "io.h"
26
27extern int verbose;
28extern int list_only;
29extern int am_root;
30extern int am_server;
31extern int am_daemon;
32extern int am_sender;
33extern int inc_recurse;
34extern int do_progress;
35extern int always_checksum;
36extern int module_id;
37extern int ignore_errors;
38extern int numeric_ids;
39extern int recurse;
40extern int xfer_dirs;
41extern int filesfrom_fd;
42extern int one_file_system;
43extern int copy_dirlinks;
44extern int keep_dirlinks;
45extern int preserve_acls;
46extern int preserve_links;
47extern int preserve_hard_links;
48extern int preserve_devices;
49extern int preserve_specials;
50extern int preserve_uid;
51extern int preserve_gid;
52extern int relative_paths;
53extern int implied_dirs;
54extern int file_extra_cnt;
55extern int ignore_perishable;
56extern int non_perishable_cnt;
57extern int prune_empty_dirs;
58extern int copy_links;
59extern int copy_unsafe_links;
60extern int protocol_version;
61extern int sanitize_paths;
62extern struct stats stats;
63
64extern char curr_dir[MAXPATHLEN];
65
66extern struct chmod_mode_struct *chmod_modes;
67
68extern struct filter_list_struct filter_list;
69extern struct filter_list_struct server_filter_list;
70
71int io_error;
72int checksum_len;
73dev_t filesystem_dev; /* used to implement -x */
74
75struct file_list *cur_flist, *first_flist, *dir_flist;
76int send_dir_ndx = -1, send_dir_depth = 0;
77int flist_cnt = 0; /* how many (non-tmp) file list objects exist */
78int file_total = 0; /* total of all active items over all file-lists */
79int flist_eof = 0; /* all the file-lists are now known */
80
81/* The tmp_* vars are used as a cache area by make_file() to store data
82 * that the sender doesn't need to remember in its file list. The data
83 * will survive just long enough to be used by send_file_entry(). */
84static dev_t tmp_rdev;
85#ifdef SUPPORT_HARD_LINKS
86static int64 tmp_dev, tmp_ino;
87#endif
88static char tmp_sum[MD4_SUM_LENGTH];
89
90static char empty_sum[MD4_SUM_LENGTH];
91static int flist_count_offset; /* for --delete --progress */
92
93static void clean_flist(struct file_list *flist, int strip_root, int no_dups);
94static void output_flist(struct file_list *flist);
95
96void init_flist(void)
97{
98 if (verbose > 4) {
99 rprintf(FINFO, "FILE_STRUCT_LEN=%d, EXTRA_LEN=%d\n",
100 (int)FILE_STRUCT_LEN, (int)EXTRA_LEN);
101 }
102 checksum_len = protocol_version < 21 ? 2 : MD4_SUM_LENGTH;
103}
104
105static int show_filelist_p(void)
106{
107 return verbose && xfer_dirs && !am_server && !inc_recurse;
108}
109
110static void start_filelist_progress(char *kind)
111{
112 rprintf(FCLIENT, "%s ... ", kind);
113 if (verbose > 1 || do_progress)
114 rprintf(FCLIENT, "\n");
115 rflush(FINFO);
116}
117
118static void emit_filelist_progress(int count)
119{
120 rprintf(FCLIENT, " %d files...\r", count);
121}
122
123static void maybe_emit_filelist_progress(int count)
124{
125 if (do_progress && show_filelist_p() && (count % 100) == 0)
126 emit_filelist_progress(count);
127}
128
129static void finish_filelist_progress(const struct file_list *flist)
130{
131 if (do_progress) {
132 /* This overwrites the progress line */
133 rprintf(FINFO, "%d file%sto consider\n",
134 flist->count, flist->count == 1 ? " " : "s ");
135 } else
136 rprintf(FINFO, "done\n");
137}
138
139void show_flist_stats(void)
140{
141 /* Nothing yet */
142}
143
144static void list_file_entry(struct file_struct *f)
145{
146 char permbuf[PERMSTRING_SIZE];
147 double len;
148
149 if (!F_IS_ACTIVE(f)) {
150 /* this can happen if duplicate names were removed */
151 return;
152 }
153
154 permstring(permbuf, f->mode);
155 len = F_LENGTH(f);
156
157 /* TODO: indicate '+' if the entry has an ACL. */
158
159#ifdef SUPPORT_LINKS
160 if (preserve_links && S_ISLNK(f->mode)) {
161 rprintf(FINFO, "%s %11.0f %s %s -> %s\n",
162 permbuf, len, timestring(f->modtime),
163 f_name(f, NULL), F_SYMLINK(f));
164 } else
165#endif
166 {
167 rprintf(FINFO, "%s %11.0f %s %s\n",
168 permbuf, len, timestring(f->modtime),
169 f_name(f, NULL));
170 }
171}
172
173/* Stat either a symlink or its referent, depending on the settings of
174 * copy_links, copy_unsafe_links, etc. Returns -1 on error, 0 on success.
175 *
176 * If path is the name of a symlink, then the linkbuf buffer (which must hold
177 * MAXPATHLEN chars) will be set to the symlink's target string.
178 *
179 * The stat structure pointed to by stp will contain information about the
180 * link or the referent as appropriate, if they exist. */
181static int readlink_stat(const char *path, STRUCT_STAT *stp, char *linkbuf)
182{
183#ifdef SUPPORT_LINKS
184 if (link_stat(path, stp, copy_dirlinks) < 0)
185 return -1;
186 if (S_ISLNK(stp->st_mode)) {
187 int llen = readlink(path, linkbuf, MAXPATHLEN - 1);
188 if (llen < 0)
189 return -1;
190 linkbuf[llen] = '\0';
191 if (copy_unsafe_links && unsafe_symlink(linkbuf, path)) {
192 if (verbose > 1) {
193 rprintf(FINFO,"copying unsafe symlink \"%s\" -> \"%s\"\n",
194 path, linkbuf);
195 }
196 return do_stat(path, stp);
197 }
198 }
199 return 0;
200#else
201 return do_stat(path, stp);
202#endif
203}
204
205int link_stat(const char *path, STRUCT_STAT *stp, int follow_dirlinks)
206{
207#ifdef SUPPORT_LINKS
208 if (copy_links)
209 return do_stat(path, stp);
210 if (do_lstat(path, stp) < 0)
211 return -1;
212 if (follow_dirlinks && S_ISLNK(stp->st_mode)) {
213 STRUCT_STAT st;
214 if (do_stat(path, &st) == 0 && S_ISDIR(st.st_mode))
215 *stp = st;
216 }
217 return 0;
218#else
219 return do_stat(path, stp);
220#endif
221}
222
223/* This function is used to check if a file should be included/excluded
224 * from the list of files based on its name and type etc. The value of
225 * filter_level is set to either SERVER_FILTERS or ALL_FILTERS. */
226static int is_excluded(char *fname, int is_dir, int filter_level)
227{
228#if 0 /* This currently never happens, so avoid a useless compare. */
229 if (filter_level == NO_FILTERS)
230 return 0;
231#endif
232 if (fname) {
233 /* never exclude '.', even if somebody does --exclude '*' */
234 if (fname[0] == '.' && !fname[1])
235 return 0;
236 /* Handle the -R version of the '.' dir. */
237 if (fname[0] == '/') {
238 int len = strlen(fname);
239 if (fname[len-1] == '.' && fname[len-2] == '/')
240 return 0;
241 }
242 }
243 if (server_filter_list.head
244 && check_filter(&server_filter_list, fname, is_dir) < 0)
245 return 1;
246 if (filter_level != ALL_FILTERS)
247 return 0;
248 if (filter_list.head
249 && check_filter(&filter_list, fname, is_dir) < 0)
250 return 1;
251 return 0;
252}
253
254static int to_wire_mode(mode_t mode)
255{
256#ifdef SUPPORT_LINKS
257#if _S_IFLNK != 0120000
258 if (S_ISLNK(mode))
259 return (mode & ~(_S_IFMT)) | 0120000;
260#endif
261#endif
262 return mode;
263}
264
265static mode_t from_wire_mode(int mode)
266{
267#if _S_IFLNK != 0120000
268 if ((mode & (_S_IFMT)) == 0120000)
269 return (mode & ~(_S_IFMT)) | _S_IFLNK;
270#endif
271 return mode;
272}
273
274static void send_directory(int f, struct file_list *flist, int ndx,
275 char *fbuf, int len, int flags);
276
277static const char *flist_dir, *orig_dir;
278static int flist_dir_len;
279
280
281/**
282 * Make sure @p flist is big enough to hold at least @p flist->count
283 * entries.
284 **/
285void flist_expand(struct file_list *flist)
286{
287 struct file_struct **new_ptr;
288
289 if (flist->count < flist->malloced)
290 return;
291
292 if (flist->malloced < FLIST_START)
293 flist->malloced = FLIST_START;
294 else if (flist->malloced >= FLIST_LINEAR)
295 flist->malloced += FLIST_LINEAR;
296 else
297 flist->malloced *= 2;
298
299 /*
300 * In case count jumped or we are starting the list
301 * with a known size just set it.
302 */
303 if (flist->malloced < flist->count)
304 flist->malloced = flist->count;
305
306 new_ptr = realloc_array(flist->files, struct file_struct *,
307 flist->malloced);
308
309 if (verbose >= 2 && flist->malloced != FLIST_START) {
310 rprintf(FCLIENT, "[%s] expand file_list to %.0f bytes, did%s move\n",
311 who_am_i(),
312 (double)sizeof flist->files[0] * flist->malloced,
313 (new_ptr == flist->files) ? " not" : "");
314 }
315
316 flist->files = new_ptr;
317
318 if (!flist->files)
319 out_of_memory("flist_expand");
320}
321
322int push_flist_dir(const char *dir, int len)
323{
324 if (dir == flist_dir)
325 return 1;
326
327 if (!orig_dir)
328 orig_dir = strdup(curr_dir);
329
330 if (flist_dir && !pop_dir(orig_dir)) {
331 rsyserr(FERROR, errno, "pop_dir %s failed",
332 full_fname(orig_dir));
333 exit_cleanup(RERR_FILESELECT);
334 }
335
336 if (dir && !push_dir(dir, 0)) {
337 io_error |= IOERR_GENERAL;
338 rsyserr(FERROR, errno, "push_dir %s failed",
339 full_fname(dir));
340 return 0;
341 }
342
343 flist_dir = dir;
344 flist_dir_len = len >= 0 ? len : dir ? (int)strlen(dir) : 0;
345
346 return 1;
347}
348
349static void send_file_entry(int f, struct file_struct *file, int ndx)
350{
351 static time_t modtime;
352 static mode_t mode;
353 static int64 dev;
354 static dev_t rdev;
355 static uint32 rdev_major;
356 static uid_t uid;
357 static gid_t gid;
358 static char *user_name, *group_name;
359 static char lastname[MAXPATHLEN];
360 char fname[MAXPATHLEN];
361 int first_hlink_ndx = -1;
362 int l1, l2;
363 int flags;
364
365 f_name(file, fname);
366
367 flags = file->flags & FLAG_TOP_DIR; /* FLAG_TOP_DIR == XMIT_TOP_DIR */
368
369 if (file->mode == mode)
370 flags |= XMIT_SAME_MODE;
371 else
372 mode = file->mode;
373 if ((preserve_devices && IS_DEVICE(mode))
374 || (preserve_specials && IS_SPECIAL(mode))) {
375 if (protocol_version < 28) {
376 if (tmp_rdev == rdev)
377 flags |= XMIT_SAME_RDEV_pre28;
378 else
379 rdev = tmp_rdev;
380 } else {
381 rdev = tmp_rdev;
382 if ((uint32)major(rdev) == rdev_major)
383 flags |= XMIT_SAME_RDEV_MAJOR;
384 else
385 rdev_major = major(rdev);
386 if ((uint32)minor(rdev) <= 0xFFu)
387 flags |= XMIT_RDEV_MINOR_IS_SMALL;
388 }
389 } else if (protocol_version < 28)
390 rdev = MAKEDEV(0, 0);
391 if (preserve_uid) {
392 if (F_UID(file) == uid && *lastname)
393 flags |= XMIT_SAME_UID;
394 else {
395 uid = F_UID(file);
396 if (preserve_uid && !numeric_ids) {
397 user_name = add_uid(uid);
398 if (inc_recurse && user_name)
399 flags |= XMIT_USER_NAME_FOLLOWS;
400 }
401 }
402 }
403 if (preserve_gid) {
404 if (F_GID(file) == gid && *lastname)
405 flags |= XMIT_SAME_GID;
406 else {
407 gid = F_GID(file);
408 if (preserve_gid && !numeric_ids) {
409 group_name = add_gid(gid);
410 if (inc_recurse && group_name)
411 flags |= XMIT_GROUP_NAME_FOLLOWS;
412 }
413 }
414 }
415 if (file->modtime == modtime)
416 flags |= XMIT_SAME_TIME;
417 else
418 modtime = file->modtime;
419
420#ifdef SUPPORT_HARD_LINKS
421 if (tmp_dev != 0) {
422 if (protocol_version >= 30) {
423 struct idev_node *np = idev_node(tmp_dev, tmp_ino);
424 first_hlink_ndx = (int32)(long)np->data - 1;
425 if (first_hlink_ndx < 0) {
426 np->data = (void*)(long)(ndx + 1);
427 flags |= XMIT_HLINK_FIRST;
428 }
429 flags |= XMIT_HLINKED;
430 } else {
431 if (tmp_dev == dev) {
432 if (protocol_version >= 28)
433 flags |= XMIT_SAME_DEV_pre30;
434 } else
435 dev = tmp_dev;
436 flags |= XMIT_HLINKED;
437 }
438 }
439#endif
440
441 for (l1 = 0;
442 lastname[l1] && (fname[l1] == lastname[l1]) && (l1 < 255);
443 l1++) {}
444 l2 = strlen(fname+l1);
445
446 if (l1 > 0)
447 flags |= XMIT_SAME_NAME;
448 if (l2 > 255)
449 flags |= XMIT_LONG_NAME;
450
451 /* We must make sure we don't send a zero flag byte or the
452 * other end will terminate the flist transfer. Note that
453 * the use of XMIT_TOP_DIR on a non-dir has no meaning, so
454 * it's harmless way to add a bit to the first flag byte. */
455 if (protocol_version >= 28) {
456 if (!flags && !S_ISDIR(mode))
457 flags |= XMIT_TOP_DIR;
458 if ((flags & 0xFF00) || !flags) {
459 flags |= XMIT_EXTENDED_FLAGS;
460 write_shortint(f, flags);
461 } else
462 write_byte(f, flags);
463 } else {
464 if (!(flags & 0xFF))
465 flags |= S_ISDIR(mode) ? XMIT_LONG_NAME : XMIT_TOP_DIR;
466 write_byte(f, flags);
467 }
468 if (flags & XMIT_SAME_NAME)
469 write_byte(f, l1);
470 if (flags & XMIT_LONG_NAME)
471 write_abbrevint30(f, l2);
472 else
473 write_byte(f, l2);
474 write_buf(f, fname + l1, l2);
475
476 if (first_hlink_ndx >= 0) {
477 write_abbrevint30(f, first_hlink_ndx);
478 goto the_end;
479 }
480
481 write_longint(f, F_LENGTH(file));
482 if (!(flags & XMIT_SAME_TIME))
483 write_int(f, modtime);
484 if (!(flags & XMIT_SAME_MODE))
485 write_int(f, to_wire_mode(mode));
486 if (preserve_uid && !(flags & XMIT_SAME_UID)) {
487 write_abbrevint30(f, uid);
488 if (flags & XMIT_USER_NAME_FOLLOWS) {
489 int len = strlen(user_name);
490 write_byte(f, len);
491 write_buf(f, user_name, len);
492 }
493 }
494 if (preserve_gid && !(flags & XMIT_SAME_GID)) {
495 write_abbrevint30(f, gid);
496 if (flags & XMIT_GROUP_NAME_FOLLOWS) {
497 int len = strlen(group_name);
498 write_byte(f, len);
499 write_buf(f, group_name, len);
500 }
501 }
502 if ((preserve_devices && IS_DEVICE(mode))
503 || (preserve_specials && IS_SPECIAL(mode))) {
504 if (protocol_version < 28) {
505 if (!(flags & XMIT_SAME_RDEV_pre28))
506 write_int(f, (int)rdev);
507 } else {
508 if (!(flags & XMIT_SAME_RDEV_MAJOR))
509 write_int(f, major(rdev));
510 if (flags & XMIT_RDEV_MINOR_IS_SMALL)
511 write_byte(f, minor(rdev));
512 else
513 write_int(f, minor(rdev));
514 }
515 }
516
517#ifdef SUPPORT_LINKS
518 if (preserve_links && S_ISLNK(mode)) {
519 const char *sl = F_SYMLINK(file);
520 int len = strlen(sl);
521 write_abbrevint30(f, len);
522 write_buf(f, sl, len);
523 }
524#endif
525
526#ifdef SUPPORT_HARD_LINKS
527 if (tmp_dev != 0 && protocol_version < 30) {
528 if (protocol_version < 26) {
529 /* 32-bit dev_t and ino_t */
530 write_int(f, (int32)dev);
531 write_int(f, (int32)tmp_ino);
532 } else {
533 /* 64-bit dev_t and ino_t */
534 if (!(flags & XMIT_SAME_DEV_pre30))
535 write_longint(f, dev);
536 write_longint(f, tmp_ino);
537 }
538 }
539#endif
540
541 if (always_checksum && (S_ISREG(mode) || protocol_version < 28)) {
542 const char *sum;
543 if (S_ISREG(mode))
544 sum = tmp_sum;
545 else {
546 /* Prior to 28, we sent a useless set of nulls. */
547 sum = empty_sum;
548 }
549 write_buf(f, sum, checksum_len);
550 }
551
552 the_end:
553 strlcpy(lastname, fname, MAXPATHLEN);
554
555 if (S_ISREG(mode) || S_ISLNK(mode))
556 stats.total_size += F_LENGTH(file);
557}
558
559static struct file_struct *recv_file_entry(struct file_list *flist,
560 int flags, int f)
561{
562 static time_t modtime;
563 static mode_t mode;
564 static int64 dev;
565 static dev_t rdev;
566 static uint32 rdev_major;
567 static uid_t uid;
568 static gid_t gid;
569 static char lastname[MAXPATHLEN], *lastdir;
570 static int lastdir_depth, lastdir_len = -1;
571 static unsigned int del_hier_name_len = 0;
572 static int in_del_hier = 0;
573 char thisname[MAXPATHLEN];
574 unsigned int l1 = 0, l2 = 0;
575 int alloc_len, basename_len, linkname_len;
576 int extra_len = file_extra_cnt * EXTRA_LEN;
577 int first_hlink_ndx = -1;
578 OFF_T file_length;
579 const char *basename;
580 char *bp;
581 struct file_struct *file;
582
583 if (flags & XMIT_SAME_NAME)
584 l1 = read_byte(f);
585
586 if (flags & XMIT_LONG_NAME)
587 l2 = read_abbrevint30(f);
588 else
589 l2 = read_byte(f);
590
591 if (l2 >= MAXPATHLEN - l1) {
592 rprintf(FERROR,
593 "overflow: flags=0x%x l1=%d l2=%d lastname=%s [%s]\n",
594 flags, l1, l2, lastname, who_am_i());
595 overflow_exit("recv_file_entry");
596 }
597
598 strlcpy(thisname, lastname, l1 + 1);
599 read_sbuf(f, &thisname[l1], l2);
600 thisname[l1 + l2] = 0;
601
602 strlcpy(lastname, thisname, MAXPATHLEN);
603
604 clean_fname(thisname, 0);
605
606 if (sanitize_paths)
607 sanitize_path(thisname, thisname, "", 0, NULL);
608
609 if ((basename = strrchr(thisname, '/')) != NULL) {
610 int len = basename++ - thisname;
611 if (len != lastdir_len || memcmp(thisname, lastdir, len) != 0) {
612 lastdir = new_array(char, len + 1);
613 memcpy(lastdir, thisname, len);
614 lastdir[len] = '\0';
615 lastdir_len = len;
616 lastdir_depth = count_dir_elements(lastdir);
617 }
618 } else
619 basename = thisname;
620 basename_len = strlen(basename) + 1; /* count the '\0' */
621
622#ifdef SUPPORT_HARD_LINKS
623 if (protocol_version >= 30
624 && BITS_SETnUNSET(flags, XMIT_HLINKED, XMIT_HLINK_FIRST)) {
625 struct file_struct *first;
626 first_hlink_ndx = read_abbrevint30(f);
627 if (first_hlink_ndx < 0 || first_hlink_ndx >= flist->count) {
628 rprintf(FERROR,
629 "hard-link reference out of range: %d (%d)\n",
630 first_hlink_ndx, flist->count);
631 exit_cleanup(RERR_PROTOCOL);
632 }
633 first = flist->files[first_hlink_ndx];
634 file_length = F_LENGTH(first);
635 modtime = first->modtime;
636 mode = first->mode;
637 if (preserve_uid)
638 uid = F_UID(first);
639 if (preserve_gid)
640 gid = F_GID(first);
641 if ((preserve_devices && IS_DEVICE(mode))
642 || (preserve_specials && IS_SPECIAL(mode))) {
643 uint32 *devp = F_RDEV_P(first);
644 rdev = MAKEDEV(DEV_MAJOR(devp), DEV_MINOR(devp));
645 extra_len += 2 * EXTRA_LEN;
646 }
647 if (preserve_links && S_ISLNK(mode))
648 linkname_len = strlen(F_SYMLINK(first)) + 1;
649 else
650 linkname_len = 0;
651 goto create_object;
652 }
653#endif
654
655 file_length = read_longint(f);
656 if (!(flags & XMIT_SAME_TIME))
657 modtime = (time_t)read_int(f);
658 if (!(flags & XMIT_SAME_MODE))
659 mode = from_wire_mode(read_int(f));
660
661 if (chmod_modes && !S_ISLNK(mode))
662 mode = tweak_mode(mode, chmod_modes);
663
664 if (preserve_uid && !(flags & XMIT_SAME_UID)) {
665 uid = (uid_t)read_abbrevint30(f);
666 if (flags & XMIT_USER_NAME_FOLLOWS)
667 uid = recv_user_name(f, uid);
668 else if (inc_recurse && am_root && !numeric_ids)
669 uid = match_uid(uid);
670 }
671 if (preserve_gid && !(flags & XMIT_SAME_GID)) {
672 gid = (gid_t)read_abbrevint30(f);
673 if (flags & XMIT_GROUP_NAME_FOLLOWS)
674 gid = recv_group_name(f, gid);
675 else if (inc_recurse && (!am_root || !numeric_ids))
676 gid = match_gid(gid);
677 }
678
679 if ((preserve_devices && IS_DEVICE(mode))
680 || (preserve_specials && IS_SPECIAL(mode))) {
681 if (protocol_version < 28) {
682 if (!(flags & XMIT_SAME_RDEV_pre28))
683 rdev = (dev_t)read_int(f);
684 } else {
685 uint32 rdev_minor;
686 if (!(flags & XMIT_SAME_RDEV_MAJOR))
687 rdev_major = read_int(f);
688 if (flags & XMIT_RDEV_MINOR_IS_SMALL)
689 rdev_minor = read_byte(f);
690 else
691 rdev_minor = read_int(f);
692 rdev = MAKEDEV(rdev_major, rdev_minor);
693 }
694 extra_len += 2 * EXTRA_LEN;
695 file_length = 0;
696 } else if (protocol_version < 28)
697 rdev = MAKEDEV(0, 0);
698
699#ifdef SUPPORT_LINKS
700 if (preserve_links && S_ISLNK(mode)) {
701 linkname_len = read_abbrevint30(f) + 1; /* count the '\0' */
702 if (linkname_len <= 0 || linkname_len > MAXPATHLEN) {
703 rprintf(FERROR, "overflow: linkname_len=%d\n",
704 linkname_len - 1);
705 overflow_exit("recv_file_entry");
706 }
707 }
708 else
709#endif
710 linkname_len = 0;
711
712#ifdef SUPPORT_HARD_LINKS
713 create_object:
714 if (preserve_hard_links) {
715 if (protocol_version < 28 && S_ISREG(mode))
716 flags |= XMIT_HLINKED;
717 if (flags & XMIT_HLINKED)
718 extra_len += EXTRA_LEN;
719 }
720#endif
721
722#ifdef SUPPORT_ACLS
723 /* We need one or two index int32s when we're preserving ACLs. */
724 if (preserve_acls)
725 extra_len += (S_ISDIR(mode) ? 2 : 1) * EXTRA_LEN;
726#endif
727
728 if (always_checksum && S_ISREG(mode))
729 extra_len += SUM_EXTRA_CNT * EXTRA_LEN;
730
731 if (file_length > 0xFFFFFFFFu && S_ISREG(mode))
732 extra_len += EXTRA_LEN;
733
734#if EXTRA_ROUNDING > 0
735 if (extra_len & (EXTRA_ROUNDING * EXTRA_LEN))
736 extra_len = (extra_len | (EXTRA_ROUNDING * EXTRA_LEN)) + EXTRA_LEN;
737#endif
738
739 if (inc_recurse && S_ISDIR(mode)) {
740 if (one_file_system) {
741 /* Room to save the dir's device for -x */
742 extra_len += 2 * EXTRA_LEN;
743 }
744 flist = dir_flist;
745 }
746
747 alloc_len = FILE_STRUCT_LEN + extra_len + basename_len
748 + linkname_len;
749 bp = pool_alloc(flist->file_pool, alloc_len, "recv_file_entry");
750
751 memset(bp, 0, extra_len + FILE_STRUCT_LEN);
752 bp += extra_len;
753 file = (struct file_struct *)bp;
754 bp += FILE_STRUCT_LEN;
755
756 memcpy(bp, basename, basename_len);
757 bp += basename_len + linkname_len; /* skip space for symlink too */
758
759#ifdef SUPPORT_HARD_LINKS
760 if (flags & XMIT_HLINKED)
761 file->flags |= FLAG_HLINKED;
762#endif
763 file->modtime = modtime;
764 file->len32 = (uint32)file_length;
765 if (file_length > 0xFFFFFFFFu && S_ISREG(mode)) {
766 file->flags |= FLAG_LENGTH64;
767 OPT_EXTRA(file, 0)->unum = (uint32)(file_length >> 32);
768 }
769 file->mode = mode;
770 if (preserve_uid)
771 F_OWNER(file) = uid;
772 if (preserve_gid)
773 F_GROUP(file) = gid;
774
775 if (basename != thisname) {
776 file->dirname = lastdir;
777 F_DEPTH(file) = lastdir_depth + 1;
778 } else
779 F_DEPTH(file) = 1;
780
781 if (S_ISDIR(mode)) {
782 if (basename_len == 1+1 && *basename == '.') /* +1 for '\0' */
783 F_DEPTH(file)--;
784 if (flags & XMIT_TOP_DIR) {
785 in_del_hier = recurse;
786 del_hier_name_len = F_DEPTH(file) == 0 ? 0 : l1 + l2;
787 if (relative_paths && del_hier_name_len > 2
788 && lastname[del_hier_name_len-1] == '.'
789 && lastname[del_hier_name_len-2] == '/')
790 del_hier_name_len -= 2;
791 file->flags |= FLAG_TOP_DIR | FLAG_XFER_DIR;
792 } else if (in_del_hier) {
793 if (!relative_paths || !del_hier_name_len
794 || (l1 >= del_hier_name_len
795 && lastname[del_hier_name_len] == '/'))
796 file->flags |= FLAG_XFER_DIR;
797 else
798 in_del_hier = 0;
799 }
800 }
801
802 if ((preserve_devices && IS_DEVICE(mode))
803 || (preserve_specials && IS_SPECIAL(mode))) {
804 uint32 *devp = F_RDEV_P(file);
805 DEV_MAJOR(devp) = major(rdev);
806 DEV_MINOR(devp) = minor(rdev);
807 }
808
809#ifdef SUPPORT_LINKS
810 if (linkname_len) {
811 bp = (char*)file->basename + basename_len;
812 if (first_hlink_ndx >= 0) {
813 struct file_struct *first = flist->files[first_hlink_ndx];
814 memcpy(bp, F_SYMLINK(first), linkname_len);
815 } else
816 read_sbuf(f, bp, linkname_len - 1);
817 if (sanitize_paths)
818 sanitize_path(bp, bp, "", lastdir_depth, NULL);
819 }
820#endif
821
822#ifdef SUPPORT_HARD_LINKS
823 if (preserve_hard_links && flags & XMIT_HLINKED) {
824 if (protocol_version >= 30) {
825 F_HL_GNUM(file) = flags & XMIT_HLINK_FIRST
826 ? flist->count : first_hlink_ndx;
827 } else {
828 static int32 cnt = 0;
829 struct idev_node *np;
830 int64 ino;
831 int32 ndx;
832 if (protocol_version < 26) {
833 dev = read_int(f);
834 ino = read_int(f);
835 } else {
836 if (!(flags & XMIT_SAME_DEV_pre30))
837 dev = read_longint(f);
838 ino = read_longint(f);
839 }
840 np = idev_node(dev, ino);
841 ndx = (int32)(long)np->data - 1;
842 if (ndx < 0) {
843 ndx = cnt++;
844 np->data = (void*)(long)cnt;
845 }
846 F_HL_GNUM(file) = ndx;
847 }
848 }
849#endif
850
851 if (always_checksum && (S_ISREG(mode) || protocol_version < 28)) {
852 if (S_ISREG(mode))
853 bp = (char*)F_SUM(file);
854 else {
855 /* Prior to 28, we get a useless set of nulls. */
856 bp = tmp_sum;
857 }
858 if (first_hlink_ndx >= 0) {
859 struct file_struct *first = flist->files[first_hlink_ndx];
860 memcpy(bp, F_SUM(first), checksum_len);
861 } else
862 read_buf(f, bp, checksum_len);
863 }
864
865#ifdef SUPPORT_ACLS
866 if (preserve_acls && !S_ISLNK(mode))
867 receive_acl(file, f);
868#endif
869
870 if (S_ISREG(mode) || S_ISLNK(mode))
871 stats.total_size += file_length;
872
873 return file;
874}
875
876/**
877 * Create a file_struct for a named file by reading its stat()
878 * information and performing extensive checks against global
879 * options.
880 *
881 * @return the new file, or NULL if there was an error or this file
882 * should be excluded.
883 *
884 * @todo There is a small optimization opportunity here to avoid
885 * stat()ing the file in some circumstances, which has a certain cost.
886 * We are called immediately after doing readdir(), and so we may
887 * already know the d_type of the file. We could for example avoid
888 * statting directories if we're not recursing, but this is not a very
889 * important case. Some systems may not have d_type.
890 **/
891struct file_struct *make_file(const char *fname, struct file_list *flist,
892 STRUCT_STAT *stp, int flags, int filter_level)
893{
894 static char *lastdir;
895 static int lastdir_len = -1;
896 struct file_struct *file;
897 STRUCT_STAT st;
898 char thisname[MAXPATHLEN];
899 char linkname[MAXPATHLEN];
900 int alloc_len, basename_len, linkname_len;
901 int extra_len = file_extra_cnt * EXTRA_LEN;
902 const char *basename;
903 char *bp;
904
905 if (strlcpy(thisname, fname, sizeof thisname)
906 >= sizeof thisname - flist_dir_len) {
907 rprintf(FINFO, "skipping overly long name: %s\n", fname);
908 return NULL;
909 }
910 clean_fname(thisname, 0);
911 if (sanitize_paths)
912 sanitize_path(thisname, thisname, "", 0, NULL);
913
914 if (stp && S_ISDIR(stp->st_mode)) {
915 st = *stp; /* Needed for "symlink/." with --relative. */
916 *linkname = '\0'; /* make IBM code checker happy */
917 } else if (readlink_stat(thisname, &st, linkname) != 0) {
918 int save_errno = errno;
919 /* See if file is excluded before reporting an error. */
920 if (filter_level != NO_FILTERS
921 && (is_excluded(thisname, 0, filter_level)
922 || is_excluded(thisname, 1, filter_level))) {
923 if (ignore_perishable && save_errno != ENOENT)
924 non_perishable_cnt++;
925 return NULL;
926 }
927 if (save_errno == ENOENT) {
928#ifdef SUPPORT_LINKS
929 /* Avoid "vanished" error if symlink points nowhere. */
930 if (copy_links && do_lstat(thisname, &st) == 0
931 && S_ISLNK(st.st_mode)) {
932 io_error |= IOERR_GENERAL;
933 rprintf(FERROR, "symlink has no referent: %s\n",
934 full_fname(thisname));
935 } else
936#endif
937 {
938 enum logcode c = am_daemon && protocol_version < 28
939 ? FERROR : FINFO;
940 io_error |= IOERR_VANISHED;
941 rprintf(c, "file has vanished: %s\n",
942 full_fname(thisname));
943 }
944 } else {
945 io_error |= IOERR_GENERAL;
946 rsyserr(FERROR, save_errno, "readlink %s failed",
947 full_fname(thisname));
948 }
949 return NULL;
950 }
951
952 /* backup.c calls us with filter_level set to NO_FILTERS. */
953 if (filter_level == NO_FILTERS)
954 goto skip_filters;
955
956 if (S_ISDIR(st.st_mode) && !xfer_dirs) {
957 rprintf(FINFO, "skipping directory %s\n", thisname);
958 return NULL;
959 }
960
961 /* -x only affects directories because we need to avoid recursing
962 * into a mount-point directory, not to avoid copying a symlinked
963 * file if -L (or similar) was specified. */
964 if (one_file_system && st.st_dev != filesystem_dev
965 && S_ISDIR(st.st_mode)) {
966 if (one_file_system > 1) {
967 if (verbose > 2) {
968 rprintf(FINFO, "skipping mount-point dir %s\n",
969 thisname);
970 }
971 return NULL;
972 }
973 flags |= FLAG_MOUNT_DIR;
974 }
975
976 if (is_excluded(thisname, S_ISDIR(st.st_mode) != 0, filter_level)) {
977 if (ignore_perishable)
978 non_perishable_cnt++;
979 return NULL;
980 }
981
982 if (lp_ignore_nonreadable(module_id)) {
983#ifdef SUPPORT_LINKS
984 if (!S_ISLNK(st.st_mode))
985#endif
986 if (access(thisname, R_OK) != 0)
987 return NULL;
988 }
989
990 skip_filters:
991
992 /* Only divert a directory in the main transfer. */
993 if (flist && flist->prev && S_ISDIR(st.st_mode)
994 && flags & FLAG_DIVERT_DIRS) {
995 flist = dir_flist;
996 /* Room for parent/sibling/next-child info. */
997 extra_len += 3 * EXTRA_LEN;
998 }
999
1000 if (verbose > 2) {
1001 rprintf(FINFO, "[%s] make_file(%s,*,%d)\n",
1002 who_am_i(), thisname, filter_level);
1003 }
1004
1005 if ((basename = strrchr(thisname, '/')) != NULL) {
1006 int len = basename++ - thisname;
1007 if (len != lastdir_len || memcmp(thisname, lastdir, len) != 0) {
1008 lastdir = new_array(char, len + 1);
1009 memcpy(lastdir, thisname, len);
1010 lastdir[len] = '\0';
1011 lastdir_len = len;
1012 }
1013 } else
1014 basename = thisname;
1015 basename_len = strlen(basename) + 1; /* count the '\0' */
1016
1017#ifdef SUPPORT_LINKS
1018 linkname_len = S_ISLNK(st.st_mode) ? strlen(linkname) + 1 : 0;
1019#else
1020 linkname_len = 0;
1021#endif
1022
1023 if (st.st_size > 0xFFFFFFFFu && S_ISREG(st.st_mode))
1024 extra_len += EXTRA_LEN;
1025
1026#if EXTRA_ROUNDING > 0
1027 if (extra_len & (EXTRA_ROUNDING * EXTRA_LEN))
1028 extra_len = (extra_len | (EXTRA_ROUNDING * EXTRA_LEN)) + EXTRA_LEN;
1029#endif
1030
1031 alloc_len = FILE_STRUCT_LEN + extra_len + basename_len
1032 + linkname_len;
1033 if (flist)
1034 bp = pool_alloc(flist->file_pool, alloc_len, "make_file");
1035 else {
1036 if (!(bp = new_array(char, alloc_len)))
1037 out_of_memory("make_file");
1038 }
1039
1040 memset(bp, 0, extra_len + FILE_STRUCT_LEN);
1041 bp += extra_len;
1042 file = (struct file_struct *)bp;
1043 bp += FILE_STRUCT_LEN;
1044
1045 memcpy(bp, basename, basename_len);
1046 bp += basename_len + linkname_len; /* skip space for symlink too */
1047
1048#ifdef SUPPORT_HARD_LINKS
1049 if (preserve_hard_links && flist && flist->prev) {
1050 if (protocol_version >= 28
1051 ? (!S_ISDIR(st.st_mode) && st.st_nlink > 1)
1052 : S_ISREG(st.st_mode)) {
1053 tmp_dev = st.st_dev;
1054 tmp_ino = st.st_ino;
1055 } else
1056 tmp_dev = 0;
1057 }
1058#endif
1059
1060#ifdef HAVE_STRUCT_STAT_ST_RDEV
1061 if (IS_DEVICE(st.st_mode) || IS_SPECIAL(st.st_mode)) {
1062 tmp_rdev = st.st_rdev;
1063 st.st_size = 0;
1064 }
1065#endif
1066
1067 file->flags = flags;
1068 file->modtime = st.st_mtime;
1069 file->len32 = (uint32)st.st_size;
1070 if (st.st_size > 0xFFFFFFFFu && S_ISREG(st.st_mode)) {
1071 file->flags |= FLAG_LENGTH64;
1072 OPT_EXTRA(file, 0)->unum = (uint32)(st.st_size >> 32);
1073 }
1074 file->mode = st.st_mode;
1075 if (preserve_uid)
1076 F_OWNER(file) = st.st_uid;
1077 if (preserve_gid)
1078 F_GROUP(file) = st.st_gid;
1079
1080 if (basename != thisname)
1081 file->dirname = lastdir;
1082
1083#ifdef SUPPORT_LINKS
1084 if (linkname_len) {
1085 bp = (char*)file->basename + basename_len;
1086 memcpy(bp, linkname, linkname_len);
1087 }
1088#endif
1089
1090 if (always_checksum && am_sender && S_ISREG(st.st_mode))
1091 file_checksum(thisname, tmp_sum, st.st_size);
1092
1093 F_ROOTDIR(file) = flist_dir;
1094
1095 /* This code is only used by the receiver when it is building
1096 * a list of files for a delete pass. */
1097 if (keep_dirlinks && linkname_len && flist) {
1098 STRUCT_STAT st2;
1099 int save_mode = file->mode;
1100 file->mode = S_IFDIR; /* Find a directory with our name. */
1101 if (flist_find(dir_flist, file) >= 0
1102 && do_stat(thisname, &st2) == 0 && S_ISDIR(st2.st_mode)) {
1103 file->modtime = st2.st_mtime;
1104 file->len32 = 0;
1105 file->mode = st2.st_mode;
1106 if (preserve_uid)
1107 F_OWNER(file) = st2.st_uid;
1108 if (preserve_gid)
1109 F_GROUP(file) = st2.st_gid;
1110 } else
1111 file->mode = save_mode;
1112 }
1113
1114 if (basename_len == 0+1)
1115 return NULL;
1116
1117 if (inc_recurse && flist == dir_flist) {
1118 flist_expand(flist);
1119 flist->files[flist->count++] = file;
1120 }
1121
1122 return file;
1123}
1124
1125/* Only called for temporary file_struct entries created by make_file(). */
1126void unmake_file(struct file_struct *file)
1127{
1128 int extra_cnt = file_extra_cnt + LEN64_BUMP(file);
1129#if EXTRA_ROUNDING > 0
1130 if (extra_cnt & EXTRA_ROUNDING)
1131 extra_cnt = (extra_cnt | EXTRA_ROUNDING) + 1;
1132#endif
1133 free(REQ_EXTRA(file, extra_cnt));
1134}
1135
1136static struct file_struct *send_file_name(int f, struct file_list *flist,
1137 char *fname, STRUCT_STAT *stp,
1138 int flags, int filter_flags)
1139{
1140 struct file_struct *file;
1141#ifdef SUPPORT_ACLS
1142 statx sx;
1143#endif
1144
1145 file = make_file(fname, flist, stp, flags, filter_flags);
1146 if (!file)
1147 return NULL;
1148
1149 if (chmod_modes && !S_ISLNK(file->mode))
1150 file->mode = tweak_mode(file->mode, chmod_modes);
1151
1152#ifdef SUPPORT_ACLS
1153 if (preserve_acls && !S_ISLNK(file->mode) && f >= 0) {
1154 sx.st.st_mode = file->mode;
1155 sx.acc_acl = sx.def_acl = NULL;
1156 if (get_acl(fname, &sx) < 0)
1157 return NULL;
1158 }
1159#endif
1160
1161 maybe_emit_filelist_progress(flist->count + flist_count_offset);
1162
1163 flist_expand(flist);
1164 flist->files[flist->count++] = file;
1165 if (f >= 0) {
1166 send_file_entry(f, file, flist->count - 1);
1167#ifdef SUPPORT_ACLS
1168 if (preserve_acls && !S_ISLNK(file->mode)) {
1169 send_acl(&sx, f);
1170 free_acl(&sx);
1171 }
1172#endif
1173 }
1174 return file;
1175}
1176
1177static void send_if_directory(int f, struct file_list *flist,
1178 struct file_struct *file,
1179 char *fbuf, unsigned int ol,
1180 int flags)
1181{
1182 char is_dot_dir = fbuf[ol-1] == '.' && (ol == 1 || fbuf[ol-2] == '/');
1183
1184 if (S_ISDIR(file->mode)
1185 && !(file->flags & FLAG_MOUNT_DIR) && f_name(file, fbuf)) {
1186 void *save_filters;
1187 unsigned int len = strlen(fbuf);
1188 if (len > 1 && fbuf[len-1] == '/')
1189 fbuf[--len] = '\0';
1190 if (len >= MAXPATHLEN - 1) {
1191 io_error |= IOERR_GENERAL;
1192 rprintf(FERROR, "skipping long-named directory: %s\n",
1193 full_fname(fbuf));
1194 return;
1195 }
1196 save_filters = push_local_filters(fbuf, len);
1197 send_directory(f, flist, -1, fbuf, len, flags);
1198 pop_local_filters(save_filters);
1199 fbuf[ol] = '\0';
1200 if (is_dot_dir)
1201 fbuf[ol-1] = '.';
1202 }
1203}
1204
1205static int file_compare(struct file_struct **file1, struct file_struct **file2)
1206{
1207 return f_name_cmp(*file1, *file2);
1208}
1209
1210/* We take an entire set of sibling dirs from dir_flist (start <= ndx <= end),
1211 * sort them by name, and link them into the tree, setting the appropriate
1212 * parent/child/sibling pointers. */
1213static void add_dirs_to_tree(int parent_ndx, int start, int end)
1214{
1215 int i;
1216 int32 *dp = NULL;
1217 int32 *parent_dp = parent_ndx < 0 ? NULL
1218 : F_DIRNODE_P(dir_flist->files[parent_ndx]);
1219
1220 qsort(dir_flist->files + start, end - start + 1,
1221 sizeof dir_flist->files[0], (int (*)())file_compare);
1222
1223 for (i = start; i <= end; i++) {
1224 struct file_struct *file = dir_flist->files[i];
1225 if (!(file->flags & FLAG_XFER_DIR)
1226 || file->flags & FLAG_MOUNT_DIR)
1227 continue;
1228 if (dp)
1229 DIR_NEXT_SIBLING(dp) = i;
1230 else if (parent_dp)
1231 DIR_FIRST_CHILD(parent_dp) = i;
1232 else
1233 send_dir_ndx = i;
1234 dp = F_DIRNODE_P(file);
1235 DIR_PARENT(dp) = parent_ndx;
1236 DIR_FIRST_CHILD(dp) = -1;
1237 }
1238 if (dp)
1239 DIR_NEXT_SIBLING(dp) = -1;
1240}
1241
1242/* This function is normally called by the sender, but the receiving side also
1243 * calls it from get_dirlist() with f set to -1 so that we just construct the
1244 * file list in memory without sending it over the wire. Also, get_dirlist()
1245 * might call this with f set to -2, which also indicates that local filter
1246 * rules should be ignored. */
1247static void send_directory(int f, struct file_list *flist, int parent_ndx,
1248 char *fbuf, int len, int flags)
1249{
1250 struct dirent *di;
1251 unsigned remainder;
1252 char *p;
1253 DIR *d;
1254 int divert_dirs = (flags & FLAG_DIVERT_DIRS) != 0;
1255 int start = divert_dirs ? dir_flist->count : flist->count;
1256 int filter_flags = f == -2 ? SERVER_FILTERS : ALL_FILTERS;
1257
1258 assert(flist != NULL);
1259
1260 if (!(d = opendir(fbuf))) {
1261 io_error |= IOERR_GENERAL;
1262 rsyserr(FERROR, errno, "opendir %s failed", full_fname(fbuf));
1263 return;
1264 }
1265
1266 p = fbuf + len;
1267 if (len != 1 || *fbuf != '/')
1268 *p++ = '/';
1269 *p = '\0';
1270 remainder = MAXPATHLEN - (p - fbuf);
1271
1272 for (errno = 0, di = readdir(d); di; errno = 0, di = readdir(d)) {
1273 char *dname = d_name(di);
1274 if (dname[0] == '.' && (dname[1] == '\0'
1275 || (dname[1] == '.' && dname[2] == '\0')))
1276 continue;
1277 if (strlcpy(p, dname, remainder) >= remainder) {
1278 io_error |= IOERR_GENERAL;
1279 rprintf(FINFO,
1280 "cannot send long-named file %s\n",
1281 full_fname(fbuf));
1282 continue;
1283 }
1284
1285 send_file_name(f, flist, fbuf, NULL, flags, filter_flags);
1286 }
1287
1288 fbuf[len] = '\0';
1289
1290 if (errno) {
1291 io_error |= IOERR_GENERAL;
1292 rsyserr(FERROR, errno, "readdir(%s)", full_fname(fbuf));
1293 }
1294
1295 closedir(d);
1296
1297 if (f < 0)
1298 return;
1299
1300 if (divert_dirs)
1301 add_dirs_to_tree(parent_ndx, start, dir_flist->count - 1);
1302 else if (recurse) {
1303 int i, end = flist->count - 1;
1304 /* send_if_directory() bumps flist->count, so use "end". */
1305 for (i = start; i <= end; i++)
1306 send_if_directory(f, flist, flist->files[i], fbuf, len, flags);
1307 }
1308}
1309
1310void send_extra_file_list(int f, int at_least)
1311{
1312 char fbuf[MAXPATHLEN];
1313 struct file_list *flist;
1314 int64 start_write;
1315 int future_cnt, save_io_error = io_error;
1316
1317 if (send_dir_ndx < 0)
1318 return;
1319
1320 /* Keep sending data until we have the requested number of
1321 * files in the upcoming file-lists. */
1322 if (cur_flist->next) {
1323 flist = first_flist->prev; /* the newest flist */
1324 future_cnt = flist->count
1325 + flist->ndx_start - cur_flist->next->ndx_start;
1326 } else
1327 future_cnt = 0;
1328 while (future_cnt < at_least) {
1329 struct file_struct *file = dir_flist->files[send_dir_ndx];
1330 int32 *dp;
1331 int dlen;
1332
1333 f_name(file, fbuf);
1334 dlen = strlen(fbuf);
1335
1336 if (F_ROOTDIR(file) != flist_dir) {
1337 if (!push_flist_dir(F_ROOTDIR(file), -1))
1338 exit_cleanup(RERR_FILESELECT);
1339 }
1340
1341 flist = flist_new(0, "send_extra_file_list");
1342 start_write = stats.total_written;
1343
1344 write_ndx(f, NDX_FLIST_OFFSET - send_dir_ndx);
1345 change_local_filter_dir(fbuf, dlen, send_dir_depth);
1346 send_directory(f, flist, send_dir_ndx, fbuf, dlen, FLAG_DIVERT_DIRS | FLAG_XFER_DIR);
1347 write_byte(f, 0);
1348
1349 clean_flist(flist, 0, 0);
1350 file_total += flist->count;
1351 future_cnt += flist->count;
1352 stats.flist_size += stats.total_written - start_write;
1353 stats.num_files += flist->count;
1354 if (verbose > 3)
1355 output_flist(flist);
1356
1357 dp = F_DIRNODE_P(file);
1358 if (DIR_FIRST_CHILD(dp) >= 0) {
1359 send_dir_ndx = DIR_FIRST_CHILD(dp);
1360 send_dir_depth++;
1361 } else {
1362 while (DIR_NEXT_SIBLING(dp) < 0) {
1363 if ((send_dir_ndx = DIR_PARENT(dp)) < 0) {
1364 write_ndx(f, NDX_FLIST_EOF);
1365 flist_eof = 1;
1366 change_local_filter_dir(NULL, 0, 0);
1367 goto finish;
1368 }
1369 send_dir_depth--;
1370 file = dir_flist->files[send_dir_ndx];
1371 dp = F_DIRNODE_P(file);
1372 }
1373 send_dir_ndx = DIR_NEXT_SIBLING(dp);
1374 }
1375 }
1376
1377 finish:
1378 if (io_error != save_io_error && !ignore_errors)
1379 send_msg_int(MSG_IO_ERROR, io_error);
1380}
1381
1382struct file_list *send_file_list(int f, int argc, char *argv[])
1383{
1384 int len;
1385 STRUCT_STAT st;
1386 char *p, *dir;
1387 char lastpath[MAXPATHLEN] = "";
1388 struct file_list *flist;
1389 struct timeval start_tv, end_tv;
1390 int64 start_write;
1391 int use_ff_fd = 0;
1392 int flags, disable_buffering;
1393
1394 rprintf(FLOG, "building file list\n");
1395 if (show_filelist_p())
1396 start_filelist_progress("building file list");
1397 else if (inc_recurse && verbose && !am_server)
1398 rprintf(FCLIENT, "sending incremental file list\n");
1399
1400 start_write = stats.total_written;
1401 gettimeofday(&start_tv, NULL);
1402
1403#ifdef SUPPORT_HARD_LINKS
1404 if (preserve_hard_links && protocol_version >= 30 && !cur_flist)
1405 init_hard_links();
1406#endif
1407
1408 flist = cur_flist = flist_new(0, "send_file_list");
1409 if (inc_recurse) {
1410 dir_flist = flist_new(FLIST_TEMP, "send_file_list");
1411 flags = FLAG_DIVERT_DIRS;
1412 } else {
1413 dir_flist = cur_flist;
1414 flags = 0;
1415 }
1416
1417 disable_buffering = io_start_buffering_out(f);
1418 if (filesfrom_fd >= 0) {
1419 if (argv[0] && !push_dir(argv[0], 0)) {
1420 rsyserr(FERROR, errno, "push_dir %s failed",
1421 full_fname(argv[0]));
1422 exit_cleanup(RERR_FILESELECT);
1423 }
1424 use_ff_fd = 1;
1425 }
1426
1427 while (1) {
1428 char fbuf[MAXPATHLEN];
1429 char *fn;
1430 int is_dot_dir;
1431
1432 if (use_ff_fd) {
1433 if (read_filesfrom_line(filesfrom_fd, fbuf) == 0)
1434 break;
1435 sanitize_path(fbuf, fbuf, "", 0, NULL);
1436 } else {
1437 if (argc-- == 0)
1438 break;
1439 strlcpy(fbuf, *argv++, MAXPATHLEN);
1440 if (sanitize_paths)
1441 sanitize_path(fbuf, fbuf, "", 0, NULL);
1442 }
1443
1444 len = strlen(fbuf);
1445 if (relative_paths) {
1446 /* We clean up fbuf below. */
1447 is_dot_dir = 0;
1448 } else if (!len || fbuf[len - 1] == '/') {
1449 if (len == 2 && fbuf[0] == '.') {
1450 /* Turn "./" into just "." rather than "./." */
1451 fbuf[1] = '\0';
1452 } else {
1453 if (len + 1 >= MAXPATHLEN)
1454 overflow_exit("send_file_list");
1455 fbuf[len++] = '.';
1456 fbuf[len] = '\0';
1457 }
1458 is_dot_dir = 1;
1459 } else if (len > 1 && fbuf[len-1] == '.' && fbuf[len-2] == '.'
1460 && (len == 2 || fbuf[len-3] == '/')) {
1461 if (len + 2 >= MAXPATHLEN)
1462 overflow_exit("send_file_list");
1463 fbuf[len++] = '/';
1464 fbuf[len++] = '.';
1465 fbuf[len] = '\0';
1466 is_dot_dir = 1;
1467 } else {
1468 is_dot_dir = fbuf[len-1] == '.'
1469 && (len == 1 || fbuf[len-2] == '/');
1470 }
1471
1472 if (link_stat(fbuf, &st, copy_dirlinks) != 0) {
1473 io_error |= IOERR_GENERAL;
1474 rsyserr(FERROR, errno, "link_stat %s failed",
1475 full_fname(fbuf));
1476 continue;
1477 }
1478
1479 if (S_ISDIR(st.st_mode) && !xfer_dirs) {
1480 rprintf(FINFO, "skipping directory %s\n", fbuf);
1481 continue;
1482 }
1483
1484 dir = NULL;
1485
1486 if (!relative_paths) {
1487 p = strrchr(fbuf, '/');
1488 if (p) {
1489 *p = '\0';
1490 if (p == fbuf)
1491 dir = "/";
1492 else
1493 dir = fbuf;
1494 len -= p - fbuf + 1;
1495 fn = p + 1;
1496 } else
1497 fn = fbuf;
1498 } else {
1499 if ((p = strstr(fbuf, "/./")) != NULL) {
1500 *p = '\0';
1501 if (p == fbuf)
1502 dir = "/";
1503 else
1504 dir = fbuf;
1505 len -= p - fbuf + 3;
1506 fn = p + 3;
1507 } else
1508 fn = fbuf;
1509 /* Get rid of trailing "/" and "/.". */
1510 while (len) {
1511 if (fn[len - 1] == '/') {
1512 is_dot_dir = 1;
1513 if (!--len && !dir) {
1514 len++;
1515 break;
1516 }
1517 }
1518 else if (len >= 2 && fn[len - 1] == '.'
1519 && fn[len - 2] == '/') {
1520 is_dot_dir = 1;
1521 if (!(len -= 2) && !dir) {
1522 len++;
1523 break;
1524 }
1525 } else
1526 break;
1527 }
1528 if (len == 1 && fn[0] == '/')
1529 fn[len++] = '.';
1530 fn[len] = '\0';
1531 /* Reject a ".." dir in the active part of the path. */
1532 for (p = fn; (p = strstr(p, "..")) != NULL; p += 2) {
1533 if ((p[2] == '/' || p[2] == '\0')
1534 && (p == fn || p[-1] == '/')) {
1535 rprintf(FERROR,
1536 "found \"..\" dir in relative path: %s\n",
1537 fbuf);
1538 exit_cleanup(RERR_SYNTAX);
1539 }
1540 }
1541 }
1542
1543 if (!*fn) {
1544 len = 1;
1545 fn = ".";
1546 }
1547
1548 if (dir && *dir) {
1549 static const char *lastdir;
1550 static int lastdir_len = -1;
1551 int len = strlen(dir);
1552
1553 if (len != lastdir_len || memcmp(lastdir, dir, len) != 0) {
1554 if (!push_flist_dir(strdup(dir), len))
1555 goto push_error;
1556 lastdir = flist_dir;
1557 lastdir_len = flist_dir_len;
1558 } else if (!push_flist_dir(lastdir, lastdir_len)) {
1559 push_error:
1560 io_error |= IOERR_GENERAL;
1561 rsyserr(FERROR, errno, "push_dir %s failed",
1562 full_fname(dir));
1563 continue;
1564 }
1565 }
1566
1567 if (fn != fbuf)
1568 memmove(fbuf, fn, len + 1);
1569
1570 if (implied_dirs && (p=strrchr(fbuf,'/')) && p != fbuf) {
1571 /* Send the implied directories at the start of the
1572 * source spec, so we get their permissions right. */
1573 char *lp = lastpath, *slash = fbuf;
1574 *p = '\0';
1575 /* Skip any initial directories in our path that we
1576 * have in common with lastpath. */
1577 for (fn = fbuf; *fn && *lp == *fn; lp++, fn++) {
1578 if (*fn == '/')
1579 slash = fn;
1580 }
1581 *p = '/';
1582 if (fn != p || (*lp && *lp != '/')) {
1583 int save_copy_links = copy_links;
1584 int save_xfer_dirs = xfer_dirs;
1585 int dir_flags = inc_recurse ? FLAG_DIVERT_DIRS : 0;
1586 copy_links |= copy_unsafe_links;
1587 xfer_dirs = 1;
1588 while ((slash = strchr(slash+1, '/')) != 0) {
1589 *slash = '\0';
1590 send_file_name(f, flist, fbuf, NULL,
1591 dir_flags, ALL_FILTERS);
1592 *slash = '/';
1593 }
1594 copy_links = save_copy_links;
1595 xfer_dirs = save_xfer_dirs;
1596 *p = '\0';
1597 strlcpy(lastpath, fbuf, sizeof lastpath);
1598 *p = '/';
1599 }
1600 }
1601
1602 if (one_file_system)
1603 filesystem_dev = st.st_dev;
1604
1605 if (recurse || (xfer_dirs && is_dot_dir)) {
1606 struct file_struct *file;
1607 int top_flags = FLAG_TOP_DIR | FLAG_XFER_DIR
1608 | (is_dot_dir ? 0 : flags)
1609 | (inc_recurse ? FLAG_DIVERT_DIRS : 0);
1610 file = send_file_name(f, flist, fbuf, &st,
1611 top_flags, ALL_FILTERS);
1612 if (file && !inc_recurse)
1613 send_if_directory(f, flist, file, fbuf, len, flags);
1614 } else
1615 send_file_name(f, flist, fbuf, &st, flags, ALL_FILTERS);
1616 }
1617
1618 gettimeofday(&end_tv, NULL);
1619 stats.flist_buildtime = (int64)(end_tv.tv_sec - start_tv.tv_sec) * 1000
1620 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
1621 if (stats.flist_buildtime == 0)
1622 stats.flist_buildtime = 1;
1623 start_tv = end_tv;
1624
1625 write_byte(f, 0); /* Indicate end of file list */
1626
1627#ifdef SUPPORT_HARD_LINKS
1628 if (preserve_hard_links && protocol_version >= 30 && !inc_recurse)
1629 idev_destroy();
1630#endif
1631
1632 if (show_filelist_p())
1633 finish_filelist_progress(flist);
1634
1635 gettimeofday(&end_tv, NULL);
1636 stats.flist_xfertime = (int64)(end_tv.tv_sec - start_tv.tv_sec) * 1000
1637 + (end_tv.tv_usec - start_tv.tv_usec) / 1000;
1638
1639 /* Sort the list without removing any duplicates in non-incremental
1640 * mode. This allows the receiving side to ask for whatever name it
1641 * kept. For incremental mode, the sender also removes duplicates
1642 * in this initial file-list so that it avoids re-sending duplicated
1643 * directories. */
1644 clean_flist(flist, 0, inc_recurse);
1645 file_total += flist->count;
1646
1647 if (!numeric_ids && !inc_recurse)
1648 send_uid_list(f);
1649
1650 /* send the io_error flag */
1651 if (protocol_version < 30)
1652 write_int(f, ignore_errors ? 0 : io_error);
1653 else if (io_error && !ignore_errors)
1654 send_msg_int(MSG_IO_ERROR, io_error);
1655
1656 if (disable_buffering)
1657 io_end_buffering_out();
1658
1659 stats.flist_size = stats.total_written - start_write;
1660 stats.num_files = flist->count;
1661
1662 if (verbose > 3)
1663 output_flist(flist);
1664
1665 if (verbose > 2)
1666 rprintf(FINFO, "send_file_list done\n");
1667
1668 if (inc_recurse) {
1669 add_dirs_to_tree(-1, 0, dir_flist->count - 1);
1670 if (file_total == 1) {
1671 /* If we're creating incremental file-lists and there
1672 * was just 1 item in the first file-list, send 1 more
1673 * file-list to check if this is a 1-file xfer. */
1674 if (send_dir_ndx < 0)
1675 write_ndx(f, NDX_DONE);
1676 else
1677 send_extra_file_list(f, 1);
1678 }
1679 }
1680
1681 return flist;
1682}
1683
1684struct file_list *recv_file_list(int f)
1685{
1686 struct file_list *flist;
1687 int dstart, flags;
1688 int64 start_read;
1689
1690 if (!first_flist)
1691 rprintf(FLOG, "receiving file list\n");
1692 if (show_filelist_p())
1693 start_filelist_progress("receiving file list");
1694 else if (inc_recurse && verbose && !am_server && !first_flist)
1695 rprintf(FCLIENT, "receiving incremental file list\n");
1696
1697 start_read = stats.total_read;
1698
1699 flist = flist_new(0, "recv_file_list");
1700
1701#ifdef SUPPORT_HARD_LINKS
1702 if (preserve_hard_links && protocol_version < 30)
1703 init_hard_links();
1704#endif
1705
1706 if (inc_recurse) {
1707 if (flist->ndx_start == 0)
1708 dir_flist = flist_new(FLIST_TEMP, "recv_file_list");
1709 dstart = dir_flist->count;
1710 } else {
1711 dir_flist = flist;
1712 dstart = 0;
1713 }
1714
1715 while ((flags = read_byte(f)) != 0) {
1716 struct file_struct *file;
1717
1718 flist_expand(flist);
1719
1720 if (protocol_version >= 28 && (flags & XMIT_EXTENDED_FLAGS))
1721 flags |= read_byte(f) << 8;
1722 file = recv_file_entry(flist, flags, f);
1723
1724 if (inc_recurse && S_ISDIR(file->mode)) {
1725 flist_expand(dir_flist);
1726 dir_flist->files[dir_flist->count++] = file;
1727 }
1728
1729 flist->files[flist->count++] = file;
1730
1731 maybe_emit_filelist_progress(flist->count);
1732
1733 if (verbose > 2) {
1734 rprintf(FINFO, "recv_file_name(%s)\n",
1735 f_name(file, NULL));
1736 }
1737 }
1738 file_total += flist->count;
1739
1740 if (verbose > 2)
1741 rprintf(FINFO, "received %d names\n", flist->count);
1742
1743 if (show_filelist_p())
1744 finish_filelist_progress(flist);
1745
1746 clean_flist(flist, relative_paths, 1);
1747
1748 if (inc_recurse) {
1749 qsort(dir_flist->files + dstart, dir_flist->count - dstart,
1750 sizeof dir_flist->files[0], (int (*)())file_compare);
1751 } else if (f >= 0)
1752 recv_uid_list(f, flist);
1753
1754 if (protocol_version < 30) {
1755 /* Recv the io_error flag */
1756 if (ignore_errors)
1757 read_int(f);
1758 else
1759 io_error |= read_int(f);
1760 }
1761
1762 if (verbose > 3)
1763 output_flist(flist);
1764
1765 if (list_only) {
1766 int i;
1767 for (i = 0; i < flist->count; i++)
1768 list_file_entry(flist->files[i]);
1769 }
1770
1771 if (verbose > 2)
1772 rprintf(FINFO, "recv_file_list done\n");
1773
1774 stats.flist_size += stats.total_read - start_read;
1775 stats.num_files += flist->count;
1776
1777 return flist;
1778}
1779
1780/* This is only used once by the receiver if the very first file-list
1781 * has exactly one item in it. */
1782void recv_additional_file_list(int f)
1783{
1784 struct file_list *flist;
1785 int ndx = read_ndx(f);
1786 if (ndx == NDX_DONE) {
1787 flist_eof = 1;
1788 change_local_filter_dir(NULL, 0, 0);
1789 } else {
1790 ndx = NDX_FLIST_OFFSET - ndx;
1791 if (ndx < 0 || ndx >= dir_flist->count) {
1792 ndx = NDX_FLIST_OFFSET - ndx;
1793 rprintf(FERROR,
1794 "Invalid dir index: %d (%d - %d)\n",
1795 ndx, NDX_FLIST_OFFSET,
1796 NDX_FLIST_OFFSET - dir_flist->count);
1797 exit_cleanup(RERR_PROTOCOL);
1798 }
1799 flist = recv_file_list(f);
1800 flist->parent_ndx = ndx;
1801 }
1802}
1803
1804/* Search for an identically-named item in the file list. Note that the
1805 * items must agree in their directory-ness, or no match is returned. */
1806int flist_find(struct file_list *flist, struct file_struct *f)
1807{
1808 int low = flist->low, high = flist->high;
1809 int diff, mid, mid_up;
1810
1811 while (low <= high) {
1812 mid = (low + high) / 2;
1813 if (F_IS_ACTIVE(flist->files[mid]))
1814 mid_up = mid;
1815 else {
1816 /* Scan for the next non-empty entry using the cached
1817 * distance values. If the value isn't fully up-to-
1818 * date, update it. */
1819 mid_up = mid + F_DEPTH(flist->files[mid]);
1820 if (!F_IS_ACTIVE(flist->files[mid_up])) {
1821 do {
1822 mid_up += F_DEPTH(flist->files[mid_up]);
1823 } while (!F_IS_ACTIVE(flist->files[mid_up]));
1824 F_DEPTH(flist->files[mid]) = mid_up - mid;
1825 }
1826 if (mid_up > high) {
1827 /* If there's nothing left above us, set high to
1828 * a non-empty entry below us and continue. */
1829 high = mid - (int)flist->files[mid]->len32;
1830 if (!F_IS_ACTIVE(flist->files[high])) {
1831 do {
1832 high -= (int)flist->files[high]->len32;
1833 } while (!F_IS_ACTIVE(flist->files[high]));
1834 flist->files[mid]->len32 = mid - high;
1835 }
1836 continue;
1837 }
1838 }
1839 diff = f_name_cmp(flist->files[mid_up], f);
1840 if (diff == 0) {
1841 if (protocol_version < 29
1842 && S_ISDIR(flist->files[mid_up]->mode)
1843 != S_ISDIR(f->mode))
1844 return -1;
1845 return mid_up;
1846 }
1847 if (diff < 0)
1848 low = mid_up + 1;
1849 else
1850 high = mid - 1;
1851 }
1852 return -1;
1853}
1854
1855/*
1856 * Free up any resources a file_struct has allocated
1857 * and clear the file.
1858 */
1859void clear_file(struct file_struct *file)
1860{
1861 /* The +1 zeros out the first char of the basename. */
1862 memset(file, 0, FILE_STRUCT_LEN + 1);
1863 /* In an empty entry, F_DEPTH() is an offset to the next non-empty
1864 * entry. Likewise for len32 in the opposite direction. We assume
1865 * that we're alone for now since flist_find() will adjust the counts
1866 * it runs into that aren't up-to-date. */
1867 file->len32 = F_DEPTH(file) = 1;
1868}
1869
1870/* Allocate a new file list. */
1871struct file_list *flist_new(int flags, char *msg)
1872{
1873 struct file_list *flist;
1874
1875 flist = new(struct file_list);
1876 if (!flist)
1877 out_of_memory(msg);
1878
1879 memset(flist, 0, sizeof flist[0]);
1880
1881 if (!(flags & FLIST_TEMP)) {
1882 if (first_flist) {
1883 flist->ndx_start = first_flist->prev->ndx_start
1884 + first_flist->prev->count;
1885 }
1886 /* This is a doubly linked list with prev looping back to
1887 * the end of the list, but the last next pointer is NULL. */
1888 if (!first_flist)
1889 first_flist = cur_flist = flist->prev = flist;
1890 else {
1891 flist->prev = first_flist->prev;
1892 flist->prev->next = first_flist->prev = flist;
1893 }
1894 flist_cnt++;
1895 }
1896
1897 if (!(flist->file_pool = pool_create(FILE_EXTENT, 0, out_of_memory, POOL_INTERN)))
1898 out_of_memory(msg);
1899
1900 return flist;
1901}
1902
1903/* Free up all elements in a flist. */
1904void flist_free(struct file_list *flist)
1905{
1906 if (!flist->prev)
1907 ; /* Was FLIST_TEMP dir-list. */
1908 else if (flist == flist->prev) {
1909 first_flist = cur_flist = NULL;
1910 file_total = 0;
1911 flist_cnt = 0;
1912 } else {
1913 if (flist == cur_flist)
1914 cur_flist = flist->next;
1915 if (flist == first_flist)
1916 first_flist = first_flist->next;
1917 else {
1918 flist->prev->next = flist->next;
1919 if (!flist->next)
1920 flist->next = first_flist;
1921 }
1922 flist->next->prev = flist->prev;
1923 file_total -= flist->count;
1924 flist_cnt--;
1925 }
1926
1927 pool_destroy(flist->file_pool);
1928 free(flist->files);
1929 free(flist);
1930}
1931
1932/*
1933 * This routine ensures we don't have any duplicate names in our file list.
1934 * duplicate names can cause corruption because of the pipelining
1935 */
1936static void clean_flist(struct file_list *flist, int strip_root, int no_dups)
1937{
1938 char fbuf[MAXPATHLEN];
1939 int i, prev_i = 0;
1940
1941 if (!flist)
1942 return;
1943 if (flist->count == 0) {
1944 flist->high = -1;
1945 return;
1946 }
1947
1948 qsort(flist->files, flist->count,
1949 sizeof flist->files[0], (int (*)())file_compare);
1950
1951 for (i = no_dups? 0 : flist->count; i < flist->count; i++) {
1952 if (F_IS_ACTIVE(flist->files[i])) {
1953 prev_i = i;
1954 break;
1955 }
1956 }
1957 flist->low = prev_i;
1958 while (++i < flist->count) {
1959 int j;
1960 struct file_struct *file = flist->files[i];
1961
1962 if (!F_IS_ACTIVE(file))
1963 continue;
1964 if (f_name_cmp(file, flist->files[prev_i]) == 0)
1965 j = prev_i;
1966 else if (protocol_version >= 29 && S_ISDIR(file->mode)) {
1967 int save_mode = file->mode;
1968 /* Make sure that this directory doesn't duplicate a
1969 * non-directory earlier in the list. */
1970 flist->high = prev_i;
1971 file->mode = S_IFREG;
1972 j = flist_find(flist, file);
1973 file->mode = save_mode;
1974 } else
1975 j = -1;
1976 if (j >= 0) {
1977 struct file_struct *fp = flist->files[j];
1978 int keep, drop;
1979 /* If one is a dir and the other is not, we want to
1980 * keep the dir because it might have contents in the
1981 * list. */
1982 if (S_ISDIR(file->mode) != S_ISDIR(fp->mode)) {
1983 if (S_ISDIR(file->mode))
1984 keep = i, drop = j;
1985 else
1986 keep = j, drop = i;
1987 } else if (protocol_version < 27)
1988 keep = j, drop = i;
1989 else
1990 keep = i, drop = j;
1991 if (verbose > 1 && !am_server) {
1992 rprintf(FINFO,
1993 "removing duplicate name %s from file list (%d)\n",
1994 f_name(file, fbuf), drop);
1995 }
1996 /* Make sure we don't lose track of a user-specified
1997 * top directory. */
1998 flist->files[keep]->flags |= flist->files[drop]->flags
1999 & (FLAG_TOP_DIR|FLAG_XFER_DIR);
2000
2001 clear_file(flist->files[drop]);
2002
2003 if (keep == i) {
2004 if (flist->low == drop) {
2005 for (j = drop + 1;
2006 j < i && !F_IS_ACTIVE(flist->files[j]);
2007 j++) {}
2008 flist->low = j;
2009 }
2010 prev_i = i;
2011 }
2012 } else
2013 prev_i = i;
2014 }
2015 flist->high = no_dups ? prev_i : flist->count - 1;
2016
2017 if (strip_root) {
2018 /* We need to strip off the leading slashes for relative
2019 * paths, but this must be done _after_ the sorting phase. */
2020 for (i = flist->low; i <= flist->high; i++) {
2021 struct file_struct *file = flist->files[i];
2022
2023 if (!file->dirname)
2024 continue;
2025 while (*file->dirname == '/')
2026 file->dirname++;
2027 if (!*file->dirname)
2028 file->dirname = NULL;
2029 }
2030 }
2031
2032 if (prune_empty_dirs && no_dups) {
2033 int j, prev_depth = 0;
2034
2035 prev_i = 0; /* It's OK that this isn't really true. */
2036
2037 for (i = flist->low; i <= flist->high; i++) {
2038 struct file_struct *fp, *file = flist->files[i];
2039
2040 /* This temporarily abuses the F_DEPTH() value for a
2041 * directory that is in a chain that might get pruned.
2042 * We restore the old value if it gets a reprieve. */
2043 if (S_ISDIR(file->mode) && F_DEPTH(file)) {
2044 /* Dump empty dirs when coming back down. */
2045 for (j = prev_depth; j >= F_DEPTH(file); j--) {
2046 fp = flist->files[prev_i];
2047 if (F_DEPTH(fp) >= 0)
2048 break;
2049 prev_i = -F_DEPTH(fp)-1;
2050 clear_file(fp);
2051 }
2052 prev_depth = F_DEPTH(file);
2053 if (is_excluded(f_name(file, fbuf), 1,
2054 ALL_FILTERS)) {
2055 /* Keep dirs through this dir. */
2056 for (j = prev_depth-1; ; j--) {
2057 fp = flist->files[prev_i];
2058 if (F_DEPTH(fp) >= 0)
2059 break;
2060 prev_i = -F_DEPTH(fp)-1;
2061 F_DEPTH(fp) = j;
2062 }
2063 } else
2064 F_DEPTH(file) = -prev_i-1;
2065 prev_i = i;
2066 } else {
2067 /* Keep dirs through this non-dir. */
2068 for (j = prev_depth; ; j--) {
2069 fp = flist->files[prev_i];
2070 if (F_DEPTH(fp) >= 0)
2071 break;
2072 prev_i = -F_DEPTH(fp)-1;
2073 F_DEPTH(fp) = j;
2074 }
2075 }
2076 }
2077 /* Dump all remaining empty dirs. */
2078 while (1) {
2079 struct file_struct *fp = flist->files[prev_i];
2080 if (F_DEPTH(fp) >= 0)
2081 break;
2082 prev_i = -F_DEPTH(fp)-1;
2083 clear_file(fp);
2084 }
2085
2086 for (i = flist->low; i <= flist->high; i++) {
2087 if (F_IS_ACTIVE(flist->files[i]))
2088 break;
2089 }
2090 flist->low = i;
2091 for (i = flist->high; i >= flist->low; i--) {
2092 if (F_IS_ACTIVE(flist->files[i]))
2093 break;
2094 }
2095 flist->high = i;
2096 }
2097}
2098
2099static void output_flist(struct file_list *flist)
2100{
2101 char uidbuf[16], gidbuf[16], depthbuf[16];
2102 struct file_struct *file;
2103 const char *root, *dir, *slash, *name, *trail;
2104 const char *who = who_am_i();
2105 int i;
2106
2107 rprintf(FINFO, "[%s] flist start=%d, count=%d, low=%d, high=%d\n",
2108 who, flist->ndx_start, flist->count, flist->low, flist->high);
2109 for (i = 0; i < flist->count; i++) {
2110 file = flist->files[i];
2111 if ((am_root || am_sender) && preserve_uid) {
2112 snprintf(uidbuf, sizeof uidbuf, " uid=%ld",
2113 (long)F_UID(file));
2114 } else
2115 *uidbuf = '\0';
2116 if (preserve_gid && F_GID(file) != GID_NONE) {
2117 snprintf(gidbuf, sizeof gidbuf, " gid=%ld",
2118 (long)F_GID(file));
2119 } else
2120 *gidbuf = '\0';
2121 if (!am_sender)
2122 snprintf(depthbuf, sizeof depthbuf, "%d", F_DEPTH(file));
2123 if (F_IS_ACTIVE(file)) {
2124 root = am_sender ? NS(F_ROOTDIR(file)) : depthbuf;
2125 if ((dir = file->dirname) == NULL)
2126 dir = slash = "";
2127 else
2128 slash = "/";
2129 name = file->basename;
2130 trail = S_ISDIR(file->mode) ? "/" : "";
2131 } else
2132 root = dir = slash = name = trail = "";
2133 rprintf(FINFO, "[%s] i=%d %s %s%s%s%s mode=0%o len=%.0f%s%s flags=%x\n",
2134 who, i, root, dir, slash, name, trail, (int)file->mode,
2135 (double)F_LENGTH(file), uidbuf, gidbuf, file->flags);
2136 }
2137}
2138
2139enum fnc_state { s_DIR, s_SLASH, s_BASE, s_TRAILING };
2140enum fnc_type { t_PATH, t_ITEM };
2141
2142/* Compare the names of two file_struct entities, similar to how strcmp()
2143 * would do if it were operating on the joined strings.
2144 *
2145 * Some differences beginning with protocol_version 29: (1) directory names
2146 * are compared with an assumed trailing slash so that they compare in a
2147 * way that would cause them to sort immediately prior to any content they
2148 * may have; (2) a directory of any name compares after a non-directory of
2149 * any name at the same depth; (3) a directory with name "." compares prior
2150 * to anything else. These changes mean that a directory and a non-dir
2151 * with the same name will not compare as equal (protocol_version >= 29).
2152 *
2153 * The dirname component can be an empty string, but the basename component
2154 * cannot (and never is in the current codebase). The basename component
2155 * may be NULL (for a removed item), in which case it is considered to be
2156 * after any existing item. */
2157int f_name_cmp(struct file_struct *f1, struct file_struct *f2)
2158{
2159 int dif;
2160 const uchar *c1, *c2;
2161 enum fnc_state state1, state2;
2162 enum fnc_type type1, type2;
2163 enum fnc_type t_path = protocol_version >= 29 ? t_PATH : t_ITEM;
2164
2165 if (!f1 || !F_IS_ACTIVE(f1)) {
2166 if (!f2 || !F_IS_ACTIVE(f2))
2167 return 0;
2168 return -1;
2169 }
2170 if (!f2 || !F_IS_ACTIVE(f2))
2171 return 1;
2172
2173 c1 = (uchar*)f1->dirname;
2174 c2 = (uchar*)f2->dirname;
2175 if (c1 == c2)
2176 c1 = c2 = NULL;
2177 if (!c1) {
2178 type1 = S_ISDIR(f1->mode) ? t_path : t_ITEM;
2179 c1 = (const uchar*)f1->basename;
2180 if (type1 == t_PATH && *c1 == '.' && !c1[1]) {
2181 type1 = t_ITEM;
2182 state1 = s_TRAILING;
2183 c1 = (uchar*)"";
2184 } else
2185 state1 = s_BASE;
2186 } else {
2187 type1 = t_path;
2188 state1 = s_DIR;
2189 }
2190 if (!c2) {
2191 type2 = S_ISDIR(f2->mode) ? t_path : t_ITEM;
2192 c2 = (const uchar*)f2->basename;
2193 if (type2 == t_PATH && *c2 == '.' && !c2[1]) {
2194 type2 = t_ITEM;
2195 state2 = s_TRAILING;
2196 c2 = (uchar*)"";
2197 } else
2198 state2 = s_BASE;
2199 } else {
2200 type2 = t_path;
2201 state2 = s_DIR;
2202 }
2203
2204 if (type1 != type2)
2205 return type1 == t_PATH ? 1 : -1;
2206
2207 do {
2208 if (!*c1) {
2209 switch (state1) {
2210 case s_DIR:
2211 state1 = s_SLASH;
2212 c1 = (uchar*)"/";
2213 break;
2214 case s_SLASH:
2215 type1 = S_ISDIR(f1->mode) ? t_path : t_ITEM;
2216 c1 = (const uchar*)f1->basename;
2217 if (type1 == t_PATH && *c1 == '.' && !c1[1]) {
2218 type1 = t_ITEM;
2219 state1 = s_TRAILING;
2220 c1 = (uchar*)"";
2221 } else
2222 state1 = s_BASE;
2223 break;
2224 case s_BASE:
2225 state1 = s_TRAILING;
2226 if (type1 == t_PATH) {
2227 c1 = (uchar*)"/";
2228 break;
2229 }
2230 /* FALL THROUGH */
2231 case s_TRAILING:
2232 type1 = t_ITEM;
2233 break;
2234 }
2235 if (*c2 && type1 != type2)
2236 return type1 == t_PATH ? 1 : -1;
2237 }
2238 if (!*c2) {
2239 switch (state2) {
2240 case s_DIR:
2241 state2 = s_SLASH;
2242 c2 = (uchar*)"/";
2243 break;
2244 case s_SLASH:
2245 type2 = S_ISDIR(f2->mode) ? t_path : t_ITEM;
2246 c2 = (const uchar*)f2->basename;
2247 if (type2 == t_PATH && *c2 == '.' && !c2[1]) {
2248 type2 = t_ITEM;
2249 state2 = s_TRAILING;
2250 c2 = (uchar*)"";
2251 } else
2252 state2 = s_BASE;
2253 break;
2254 case s_BASE:
2255 state2 = s_TRAILING;
2256 if (type2 == t_PATH) {
2257 c2 = (uchar*)"/";
2258 break;
2259 }
2260 /* FALL THROUGH */
2261 case s_TRAILING:
2262 if (!*c1)
2263 return 0;
2264 type2 = t_ITEM;
2265 break;
2266 }
2267 if (type1 != type2)
2268 return type1 == t_PATH ? 1 : -1;
2269 }
2270 } while ((dif = (int)*c1++ - (int)*c2++) == 0);
2271
2272 return dif;
2273}
2274
2275char *f_name_buf(void)
2276{
2277 static char names[5][MAXPATHLEN];
2278 static unsigned int n;
2279
2280 n = (n + 1) % (sizeof names / sizeof names[0]);
2281
2282 return names[n];
2283}
2284
2285/* Return a copy of the full filename of a flist entry, using the indicated
2286 * buffer or one of 5 static buffers if fbuf is NULL. No size-checking is
2287 * done because we checked the size when creating the file_struct entry.
2288 */
2289char *f_name(struct file_struct *f, char *fbuf)
2290{
2291 if (!f || !F_IS_ACTIVE(f))
2292 return NULL;
2293
2294 if (!fbuf)
2295 fbuf = f_name_buf();
2296
2297 if (f->dirname) {
2298 int len = strlen(f->dirname);
2299 memcpy(fbuf, f->dirname, len);
2300 fbuf[len] = '/';
2301 strlcpy(fbuf + len + 1, f->basename, MAXPATHLEN - (len + 1));
2302 } else
2303 strlcpy(fbuf, f->basename, MAXPATHLEN);
2304
2305 return fbuf;
2306}
2307
2308/* Do a non-recursive scan of the named directory, possibly ignoring all
2309 * exclude rules except for the daemon's. If "dlen" is >=0, it is the length
2310 * of the dirname string, and also indicates that "dirname" is a MAXPATHLEN
2311 * buffer (the functions we call will append names onto the end, but the old
2312 * dir value will be restored on exit). */
2313struct file_list *get_dirlist(char *dirname, int dlen, int ignore_filter_rules)
2314{
2315 struct file_list *dirlist;
2316 char dirbuf[MAXPATHLEN];
2317 int save_recurse = recurse;
2318 int save_xfer_dirs = xfer_dirs;
2319
2320 if (dlen < 0) {
2321 dlen = strlcpy(dirbuf, dirname, MAXPATHLEN);
2322 if (dlen >= MAXPATHLEN)
2323 return NULL;
2324 dirname = dirbuf;
2325 }
2326
2327 dirlist = flist_new(FLIST_TEMP, "get_dirlist");
2328
2329 recurse = 0;
2330 xfer_dirs = 1;
2331 send_directory(ignore_filter_rules ? -2 : -1, dirlist, -1, dirname, dlen, 0);
2332 xfer_dirs = save_xfer_dirs;
2333 recurse = save_recurse;
2334 if (do_progress)
2335 flist_count_offset += dirlist->count;
2336
2337 clean_flist(dirlist, 0, 0);
2338
2339 if (verbose > 3)
2340 output_flist(dirlist);
2341
2342 return dirlist;
2343}