Support the new --files-from and --from0 options.
[rsync/rsync.git] / flist.c
... / ...
CommitLineData
1/*
2 Copyright (C) Andrew Tridgell 1996
3 Copyright (C) Paul Mackerras 1996
4 Copyright (C) 2001, 2002 by Martin Pool <mbp@samba.org>
5
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
10
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
15
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19*/
20
21/** @file flist.c
22 * Generate and receive file lists
23 *
24 * @todo Get rid of the string_area optimization. Efficiently
25 * allocating blocks is the responsibility of the system's malloc
26 * library, not of rsync.
27 *
28 * @sa http://lists.samba.org/pipermail/rsync/2000-June/002351.html
29 *
30 **/
31
32#include "rsync.h"
33
34extern struct stats stats;
35
36extern int verbose;
37extern int do_progress;
38extern int am_server;
39extern int always_checksum;
40
41extern int cvs_exclude;
42
43extern int recurse;
44
45extern int one_file_system;
46extern int make_backups;
47extern int preserve_links;
48extern int preserve_hard_links;
49extern int preserve_perms;
50extern int preserve_devices;
51extern int preserve_uid;
52extern int preserve_gid;
53extern int preserve_times;
54extern int relative_paths;
55extern int copy_links;
56extern int copy_unsafe_links;
57extern int remote_version;
58extern int io_error;
59extern int sanitize_paths;
60
61extern int read_batch;
62extern int write_batch;
63
64static struct exclude_struct **local_exclude_list;
65
66static struct file_struct null_file;
67
68static void clean_flist(struct file_list *flist, int strip_root, int no_dups);
69
70
71static int show_filelist_p(void)
72{
73 return verbose && recurse && !am_server;
74}
75
76static void start_filelist_progress(char *kind)
77{
78 rprintf(FINFO, "%s ... ", kind);
79 if ((verbose > 1) || do_progress)
80 rprintf(FINFO, "\n");
81 rflush(FINFO);
82}
83
84
85static void emit_filelist_progress(const struct file_list *flist)
86{
87 rprintf(FINFO, " %d files...\r", flist->count);
88}
89
90
91static void maybe_emit_filelist_progress(const struct file_list *flist)
92{
93 if (do_progress && show_filelist_p() && ((flist->count % 100) == 0))
94 emit_filelist_progress(flist);
95}
96
97
98static void finish_filelist_progress(const struct file_list *flist)
99{
100 if (do_progress) {
101 /* This overwrites the progress line */
102 rprintf(FINFO, "%d file%sto consider\n",
103 flist->count, flist->count == 1 ? " " : "s ");
104 } else {
105 rprintf(FINFO, "done\n");
106 }
107}
108
109void show_flist_stats(void)
110{
111 /* Nothing yet */
112}
113
114
115static struct string_area *string_area_new(int size)
116{
117 struct string_area *a;
118
119 if (size <= 0)
120 size = ARENA_SIZE;
121 a = malloc(sizeof(*a));
122 if (!a)
123 out_of_memory("string_area_new");
124 a->current = a->base = malloc(size);
125 if (!a->current)
126 out_of_memory("string_area_new buffer");
127 a->end = a->base + size;
128 a->next = NULL;
129
130 return a;
131}
132
133static void string_area_free(struct string_area *a)
134{
135 struct string_area *next;
136
137 for (; a; a = next) {
138 next = a->next;
139 free(a->base);
140 }
141}
142
143static char *string_area_malloc(struct string_area **ap, int size)
144{
145 char *p;
146 struct string_area *a;
147
148 /* does the request fit into the current space? */
149 a = *ap;
150 if (a->current + size >= a->end) {
151 /* no; get space, move new string_area to front of the list */
152 a = string_area_new(size > ARENA_SIZE ? size : ARENA_SIZE);
153 a->next = *ap;
154 *ap = a;
155 }
156
157 /* have space; do the "allocation." */
158 p = a->current;
159 a->current += size;
160 return p;
161}
162
163static char *string_area_strdup(struct string_area **ap, const char *src)
164{
165 char *dest = string_area_malloc(ap, strlen(src) + 1);
166 return strcpy(dest, src);
167}
168
169static void list_file_entry(struct file_struct *f)
170{
171 char perms[11];
172
173 if (!f->basename)
174 /* this can happen if duplicate names were removed */
175 return;
176
177 permstring(perms, f->mode);
178
179 if (preserve_links && S_ISLNK(f->mode)) {
180 rprintf(FINFO, "%s %11.0f %s %s -> %s\n",
181 perms,
182 (double) f->length, timestring(f->modtime),
183 f_name(f), f->link);
184 } else {
185 rprintf(FINFO, "%s %11.0f %s %s\n",
186 perms,
187 (double) f->length, timestring(f->modtime),
188 f_name(f));
189 }
190}
191
192
193/**
194 * Stat either a symlink or its referent, depending on the settings of
195 * copy_links, copy_unsafe_links, etc.
196 *
197 * @retval -1 on error
198 *
199 * @retval 0 for success
200 *
201 * @post If @p path is a symlink, then @p linkbuf (of size @c
202 * MAXPATHLEN) contains the symlink target.
203 *
204 * @post @p buffer contains information about the link or the
205 * referrent as appropriate, if they exist.
206 **/
207int readlink_stat(const char *path, STRUCT_STAT * buffer, char *linkbuf)
208{
209#if SUPPORT_LINKS
210 if (copy_links) {
211 return do_stat(path, buffer);
212 }
213 if (do_lstat(path, buffer) == -1) {
214 return -1;
215 }
216 if (S_ISLNK(buffer->st_mode)) {
217 int l;
218 l = readlink((char *) path, linkbuf, MAXPATHLEN - 1);
219 if (l == -1)
220 return -1;
221 linkbuf[l] = 0;
222 if (copy_unsafe_links && unsafe_symlink(linkbuf, path)) {
223 if (verbose > 1) {
224 rprintf(FINFO,"copying unsafe symlink \"%s\" -> \"%s\"\n",
225 path, linkbuf);
226 }
227 return do_stat(path, buffer);
228 }
229 }
230 return 0;
231#else
232 return do_stat(path, buffer);
233#endif
234}
235
236int link_stat(const char *path, STRUCT_STAT * buffer)
237{
238#if SUPPORT_LINKS
239 if (copy_links) {
240 return do_stat(path, buffer);
241 } else {
242 return do_lstat(path, buffer);
243 }
244#else
245 return do_stat(path, buffer);
246#endif
247}
248
249/*
250 This function is used to check if a file should be included/excluded
251 from the list of files based on its name and type etc
252 */
253static int check_exclude_file(int f, char *fname, STRUCT_STAT * st)
254{
255 extern int delete_excluded;
256
257 /* f is set to -1 when calculating deletion file list */
258 if ((f == -1) && delete_excluded) {
259 return 0;
260 }
261 if (check_exclude(fname, local_exclude_list, st)) {
262 return 1;
263 }
264 return 0;
265}
266
267/* used by the one_file_system code */
268static dev_t filesystem_dev;
269
270static void set_filesystem(char *fname)
271{
272 STRUCT_STAT st;
273 if (link_stat(fname, &st) != 0)
274 return;
275 filesystem_dev = st.st_dev;
276}
277
278
279static int to_wire_mode(mode_t mode)
280{
281 if (S_ISLNK(mode) && (_S_IFLNK != 0120000)) {
282 return (mode & ~(_S_IFMT)) | 0120000;
283 }
284 return (int) mode;
285}
286
287static mode_t from_wire_mode(int mode)
288{
289 if ((mode & (_S_IFMT)) == 0120000 && (_S_IFLNK != 0120000)) {
290 return (mode & ~(_S_IFMT)) | _S_IFLNK;
291 }
292 return (mode_t) mode;
293}
294
295
296static void send_directory(int f, struct file_list *flist, char *dir);
297
298static char *flist_dir;
299
300
301/**
302 * Make sure @p flist is big enough to hold at least @p flist->count
303 * entries.
304 **/
305static void flist_expand(struct file_list *flist)
306{
307 if (flist->count >= flist->malloced) {
308 size_t new_bytes;
309 void *new_ptr;
310
311 if (flist->malloced < 1000)
312 flist->malloced += 1000;
313 else
314 flist->malloced *= 2;
315
316 new_bytes = sizeof(flist->files[0]) * flist->malloced;
317
318 if (flist->files)
319 new_ptr = realloc(flist->files, new_bytes);
320 else
321 new_ptr = malloc(new_bytes);
322
323 if (verbose >= 2) {
324 rprintf(FINFO, "expand file_list to %.0f bytes, did%s move\n",
325 (double) new_bytes,
326 (new_ptr == flist->files) ? " not" : "");
327 }
328
329 flist->files = (struct file_struct **) new_ptr;
330
331 if (!flist->files)
332 out_of_memory("flist_expand");
333 }
334}
335
336
337static void send_file_entry(struct file_struct *file, int f,
338 unsigned base_flags)
339{
340 unsigned char flags;
341 static time_t last_time;
342 static mode_t last_mode;
343 static DEV64_T last_rdev;
344 static uid_t last_uid;
345 static gid_t last_gid;
346 static char lastname[MAXPATHLEN];
347 char *fname;
348 int l1, l2;
349
350 if (f == -1)
351 return;
352
353 if (!file) {
354 write_byte(f, 0);
355 return;
356 }
357
358 io_write_phase = "send_file_entry";
359
360 fname = f_name(file);
361
362 flags = base_flags;
363
364 if (file->mode == last_mode)
365 flags |= SAME_MODE;
366 if (file->rdev == last_rdev)
367 flags |= SAME_RDEV;
368 if (file->uid == last_uid)
369 flags |= SAME_UID;
370 if (file->gid == last_gid)
371 flags |= SAME_GID;
372 if (file->modtime == last_time)
373 flags |= SAME_TIME;
374
375 for (l1 = 0;
376 lastname[l1] && (fname[l1] == lastname[l1]) && (l1 < 255);
377 l1++) {}
378 l2 = strlen(fname) - l1;
379
380 if (l1 > 0)
381 flags |= SAME_NAME;
382 if (l2 > 255)
383 flags |= LONG_NAME;
384
385 /* we must make sure we don't send a zero flags byte or the other
386 end will terminate the flist transfer */
387 if (flags == 0 && !S_ISDIR(file->mode))
388 flags |= FLAG_DELETE;
389 if (flags == 0)
390 flags |= LONG_NAME;
391
392 write_byte(f, flags);
393 if (flags & SAME_NAME)
394 write_byte(f, l1);
395 if (flags & LONG_NAME)
396 write_int(f, l2);
397 else
398 write_byte(f, l2);
399 write_buf(f, fname + l1, l2);
400
401 write_longint(f, file->length);
402 if (!(flags & SAME_TIME))
403 write_int(f, (int) file->modtime);
404 if (!(flags & SAME_MODE))
405 write_int(f, to_wire_mode(file->mode));
406 if (preserve_uid && !(flags & SAME_UID)) {
407 add_uid(file->uid);
408 write_int(f, (int) file->uid);
409 }
410 if (preserve_gid && !(flags & SAME_GID)) {
411 add_gid(file->gid);
412 write_int(f, (int) file->gid);
413 }
414 if (preserve_devices && IS_DEVICE(file->mode)
415 && !(flags & SAME_RDEV))
416 write_int(f, (int) file->rdev);
417
418#if SUPPORT_LINKS
419 if (preserve_links && S_ISLNK(file->mode)) {
420 write_int(f, strlen(file->link));
421 write_buf(f, file->link, strlen(file->link));
422 }
423#endif
424
425#if SUPPORT_HARD_LINKS
426 if (preserve_hard_links && S_ISREG(file->mode)) {
427 if (remote_version < 26) {
428 /* 32-bit dev_t and ino_t */
429 write_int(f, (int) file->dev);
430 write_int(f, (int) file->inode);
431 } else {
432 /* 64-bit dev_t and ino_t */
433 write_longint(f, file->dev);
434 write_longint(f, file->inode);
435 }
436 }
437#endif
438
439 if (always_checksum) {
440 if (remote_version < 21) {
441 write_buf(f, file->sum, 2);
442 } else {
443 write_buf(f, file->sum, MD4_SUM_LENGTH);
444 }
445 }
446
447 last_mode = file->mode;
448 last_rdev = file->rdev;
449 last_uid = file->uid;
450 last_gid = file->gid;
451 last_time = file->modtime;
452
453 strlcpy(lastname, fname, MAXPATHLEN);
454 lastname[MAXPATHLEN - 1] = 0;
455
456 io_write_phase = "unknown";
457}
458
459
460
461static void receive_file_entry(struct file_struct **fptr,
462 unsigned flags, int f)
463{
464 static time_t last_time;
465 static mode_t last_mode;
466 static DEV64_T last_rdev;
467 static uid_t last_uid;
468 static gid_t last_gid;
469 static char lastname[MAXPATHLEN];
470 char thisname[MAXPATHLEN];
471 unsigned int l1 = 0, l2 = 0;
472 char *p;
473 struct file_struct *file;
474
475 if (flags & SAME_NAME)
476 l1 = read_byte(f);
477
478 if (flags & LONG_NAME)
479 l2 = read_int(f);
480 else
481 l2 = read_byte(f);
482
483 file = (struct file_struct *) malloc(sizeof(*file));
484 if (!file)
485 out_of_memory("receive_file_entry");
486 memset((char *) file, 0, sizeof(*file));
487 (*fptr) = file;
488
489 if (l2 >= MAXPATHLEN - l1) {
490 rprintf(FERROR,
491 "overflow: flags=0x%x l1=%d l2=%d lastname=%s\n",
492 flags, l1, l2, lastname);
493 overflow("receive_file_entry");
494 }
495
496 strlcpy(thisname, lastname, l1 + 1);
497 read_sbuf(f, &thisname[l1], l2);
498 thisname[l1 + l2] = 0;
499
500 strlcpy(lastname, thisname, MAXPATHLEN);
501 lastname[MAXPATHLEN - 1] = 0;
502
503 clean_fname(thisname);
504
505 if (sanitize_paths) {
506 sanitize_path(thisname, NULL);
507 }
508
509 if ((p = strrchr(thisname, '/'))) {
510 static char *lastdir;
511 *p = 0;
512 if (lastdir && strcmp(thisname, lastdir) == 0) {
513 file->dirname = lastdir;
514 } else {
515 file->dirname = strdup(thisname);
516 lastdir = file->dirname;
517 }
518 file->basename = strdup(p + 1);
519 } else {
520 file->dirname = NULL;
521 file->basename = strdup(thisname);
522 }
523
524 if (!file->basename)
525 out_of_memory("receive_file_entry 1");
526
527
528 file->flags = flags;
529 file->length = read_longint(f);
530 file->modtime =
531 (flags & SAME_TIME) ? last_time : (time_t) read_int(f);
532 file->mode =
533 (flags & SAME_MODE) ? last_mode : from_wire_mode(read_int(f));
534 if (preserve_uid)
535 file->uid =
536 (flags & SAME_UID) ? last_uid : (uid_t) read_int(f);
537 if (preserve_gid)
538 file->gid =
539 (flags & SAME_GID) ? last_gid : (gid_t) read_int(f);
540 if (preserve_devices && IS_DEVICE(file->mode))
541 file->rdev =
542 (flags & SAME_RDEV) ? last_rdev : (DEV64_T) read_int(f);
543
544 if (preserve_links && S_ISLNK(file->mode)) {
545 int l = read_int(f);
546 if (l < 0) {
547 rprintf(FERROR, "overflow: l=%d\n", l);
548 overflow("receive_file_entry");
549 }
550 file->link = (char *) malloc(l + 1);
551 if (!file->link)
552 out_of_memory("receive_file_entry 2");
553 read_sbuf(f, file->link, l);
554 if (sanitize_paths) {
555 sanitize_path(file->link, file->dirname);
556 }
557 }
558#if SUPPORT_HARD_LINKS
559 if (preserve_hard_links && S_ISREG(file->mode)) {
560 if (remote_version < 26) {
561 file->dev = read_int(f);
562 file->inode = read_int(f);
563 } else {
564 file->dev = read_longint(f);
565 file->inode = read_longint(f);
566 }
567 }
568#endif
569
570 if (always_checksum) {
571 file->sum = (char *) malloc(MD4_SUM_LENGTH);
572 if (!file->sum)
573 out_of_memory("md4 sum");
574 if (remote_version < 21) {
575 read_buf(f, file->sum, 2);
576 } else {
577 read_buf(f, file->sum, MD4_SUM_LENGTH);
578 }
579 }
580
581 last_mode = file->mode;
582 last_rdev = file->rdev;
583 last_uid = file->uid;
584 last_gid = file->gid;
585 last_time = file->modtime;
586
587 if (!preserve_perms) {
588 extern int orig_umask;
589 /* set an appropriate set of permissions based on original
590 permissions and umask. This emulates what GNU cp does */
591 file->mode &= ~orig_umask;
592 }
593}
594
595
596/* determine if a file in a different filesstem should be skipped
597 when one_file_system is set. We bascally only want to include
598 the mount points - but they can be hard to find! */
599static int skip_filesystem(char *fname, STRUCT_STAT * st)
600{
601 STRUCT_STAT st2;
602 char *p = strrchr(fname, '/');
603
604 /* skip all but directories */
605 if (!S_ISDIR(st->st_mode))
606 return 1;
607
608 /* if its not a subdirectory then allow */
609 if (!p)
610 return 0;
611
612 *p = 0;
613 if (link_stat(fname, &st2)) {
614 *p = '/';
615 return 0;
616 }
617 *p = '/';
618
619 return (st2.st_dev != filesystem_dev);
620}
621
622#define STRDUP(ap, p) (ap ? string_area_strdup(ap, p) : strdup(p))
623/* IRIX cc cares that the operands to the ternary have the same type. */
624#define MALLOC(ap, i) (ap ? (void*) string_area_malloc(ap, i) : malloc(i))
625
626/**
627 * Create a file_struct for a named file by reading its stat()
628 * information and performing extensive checks against global
629 * options.
630 *
631 * @return the new file, or NULL if there was an error or this file
632 * should be excluded.
633 *
634 * @todo There is a small optimization opportunity here to avoid
635 * stat()ing the file in some circumstances, which has a certain cost.
636 * We are called immediately after doing readdir(), and so we may
637 * already know the d_type of the file. We could for example avoid
638 * statting directories if we're not recursing, but this is not a very
639 * important case. Some systems may not have d_type.
640 **/
641struct file_struct *make_file(int f, char *fname, struct string_area **ap,
642 int noexcludes)
643{
644 struct file_struct *file;
645 STRUCT_STAT st;
646 char sum[SUM_LENGTH];
647 char *p;
648 char cleaned_name[MAXPATHLEN];
649 char linkbuf[MAXPATHLEN];
650 extern int module_id;
651
652 strlcpy(cleaned_name, fname, MAXPATHLEN);
653 cleaned_name[MAXPATHLEN - 1] = 0;
654 clean_fname(cleaned_name);
655 if (sanitize_paths) {
656 sanitize_path(cleaned_name, NULL);
657 }
658 fname = cleaned_name;
659
660 memset(sum, 0, SUM_LENGTH);
661
662 if (readlink_stat(fname, &st, linkbuf) != 0) {
663 int save_errno = errno;
664 if ((errno == ENOENT) && !noexcludes) {
665 /* either symlink pointing nowhere or file that
666 * was removed during rsync run; see if excluded
667 * before reporting an error */
668 memset((char *) &st, 0, sizeof(st));
669 if (check_exclude_file(f, fname, &st)) {
670 /* file is excluded anyway, ignore silently */
671 return NULL;
672 }
673 }
674 io_error = 1;
675 rprintf(FERROR, "readlink %s: %s\n",
676 fname, strerror(save_errno));
677 return NULL;
678 }
679
680 /* we use noexcludes from backup.c */
681 if (noexcludes)
682 goto skip_excludes;
683
684 if (S_ISDIR(st.st_mode) && !recurse) {
685 rprintf(FINFO, "skipping directory %s\n", fname);
686 return NULL;
687 }
688
689 if (one_file_system && st.st_dev != filesystem_dev) {
690 if (skip_filesystem(fname, &st))
691 return NULL;
692 }
693
694 if (check_exclude_file(f, fname, &st))
695 return NULL;
696
697
698 if (lp_ignore_nonreadable(module_id) && access(fname, R_OK) != 0)
699 return NULL;
700
701 skip_excludes:
702
703 if (verbose > 2)
704 rprintf(FINFO, "make_file(%d,%s)\n", f, fname);
705
706 file = (struct file_struct *) malloc(sizeof(*file));
707 if (!file)
708 out_of_memory("make_file");
709 memset((char *) file, 0, sizeof(*file));
710
711 if ((p = strrchr(fname, '/'))) {
712 static char *lastdir;
713 *p = 0;
714 if (lastdir && strcmp(fname, lastdir) == 0) {
715 file->dirname = lastdir;
716 } else {
717 file->dirname = strdup(fname);
718 lastdir = file->dirname;
719 }
720 file->basename = STRDUP(ap, p + 1);
721 *p = '/';
722 } else {
723 file->dirname = NULL;
724 file->basename = STRDUP(ap, fname);
725 }
726
727 file->modtime = st.st_mtime;
728 file->length = st.st_size;
729 file->mode = st.st_mode;
730 file->uid = st.st_uid;
731 file->gid = st.st_gid;
732 file->dev = st.st_dev;
733 file->inode = st.st_ino;
734#ifdef HAVE_STRUCT_STAT_ST_RDEV
735 file->rdev = st.st_rdev;
736#endif
737
738#if SUPPORT_LINKS
739 if (S_ISLNK(st.st_mode)) {
740 file->link = STRDUP(ap, linkbuf);
741 }
742#endif
743
744 if (always_checksum) {
745 file->sum = (char *) MALLOC(ap, MD4_SUM_LENGTH);
746 if (!file->sum)
747 out_of_memory("md4 sum");
748 /* drat. we have to provide a null checksum for non-regular
749 files in order to be compatible with earlier versions
750 of rsync */
751 if (S_ISREG(st.st_mode)) {
752 file_checksum(fname, file->sum, st.st_size);
753 } else {
754 memset(file->sum, 0, MD4_SUM_LENGTH);
755 }
756 }
757
758 if (flist_dir) {
759 static char *lastdir;
760 if (lastdir && strcmp(lastdir, flist_dir) == 0) {
761 file->basedir = lastdir;
762 } else {
763 file->basedir = strdup(flist_dir);
764 lastdir = file->basedir;
765 }
766 } else {
767 file->basedir = NULL;
768 }
769
770 if (!S_ISDIR(st.st_mode))
771 stats.total_size += st.st_size;
772
773 return file;
774}
775
776
777
778void send_file_name(int f, struct file_list *flist, char *fname,
779 int recursive, unsigned base_flags)
780{
781 struct file_struct *file;
782
783 file = make_file(f, fname, &flist->string_area, 0);
784
785 if (!file)
786 return;
787
788 maybe_emit_filelist_progress(flist);
789
790 flist_expand(flist);
791
792 if (write_batch) /* dw */
793 file->flags = FLAG_DELETE;
794
795 if (strcmp(file->basename, "")) {
796 flist->files[flist->count++] = file;
797 send_file_entry(file, f, base_flags);
798 }
799
800 if (S_ISDIR(file->mode) && recursive) {
801 struct exclude_struct **last_exclude_list =
802 local_exclude_list;
803 send_directory(f, flist, f_name(file));
804 local_exclude_list = last_exclude_list;
805 return;
806 }
807}
808
809
810
811static void send_directory(int f, struct file_list *flist, char *dir)
812{
813 DIR *d;
814 struct dirent *di;
815 char fname[MAXPATHLEN];
816 int l;
817 char *p;
818
819 d = opendir(dir);
820 if (!d) {
821 io_error = 1;
822 rprintf(FERROR, "opendir(%s): %s\n", dir, strerror(errno));
823 return;
824 }
825
826 strlcpy(fname, dir, MAXPATHLEN);
827 l = strlen(fname);
828 if (fname[l - 1] != '/') {
829 if (l == MAXPATHLEN - 1) {
830 io_error = 1;
831 rprintf(FERROR,
832 "skipping long-named directory %s\n",
833 fname);
834 closedir(d);
835 return;
836 }
837 strlcat(fname, "/", MAXPATHLEN);
838 l++;
839 }
840 p = fname + strlen(fname);
841
842 local_exclude_list = NULL;
843
844 if (cvs_exclude) {
845 if (strlen(fname) + strlen(".cvsignore") <= MAXPATHLEN - 1) {
846 strcpy(p, ".cvsignore");
847 local_exclude_list =
848 make_exclude_list(fname, NULL, 0, 0);
849 } else {
850 io_error = 1;
851 rprintf(FINFO,
852 "cannot cvs-exclude in long-named directory %s\n",
853 fname);
854 }
855 }
856
857 for (di = readdir(d); di; di = readdir(d)) {
858 char *dname = d_name(di);
859 if (strcmp(dname, ".") == 0 || strcmp(dname, "..") == 0)
860 continue;
861 strlcpy(p, dname, MAXPATHLEN - l);
862 send_file_name(f, flist, fname, recurse, 0);
863 }
864
865 if (local_exclude_list) {
866 add_exclude_list("!", &local_exclude_list, 0);
867 }
868
869 closedir(d);
870}
871
872
873/**
874 *
875 * I <b>think</b> f==-1 means that the list should just be built in
876 * memory and not transmitted. But who can tell? -- mbp
877 **/
878struct file_list *send_file_list(int f, int argc, char *argv[])
879{
880 int i, l;
881 STRUCT_STAT st;
882 char *p, *dir, *olddir;
883 char lastpath[MAXPATHLEN] = "";
884 struct file_list *flist;
885 int64 start_write;
886
887 if (show_filelist_p() && f != -1)
888 start_filelist_progress("building file list");
889
890 start_write = stats.total_written;
891
892 flist = flist_new();
893
894 if (f != -1) {
895 io_start_buffering(f);
896 }
897
898 for (i = 0; i < argc; i++) {
899 char fname2[MAXPATHLEN];
900 char *fname = fname2;
901
902 strlcpy(fname, argv[i], MAXPATHLEN);
903
904 l = strlen(fname);
905 if (l != 1 && fname[l - 1] == '/') {
906 if ((l == 2) && (fname[0] == '.')) {
907 /* Turn ./ into just . rather than ./.
908 This was put in to avoid a problem with
909 rsync -aR --delete from ./
910 The send_file_name() below of ./ was
911 mysteriously preventing deletes */
912 fname[1] = 0;
913 } else {
914 strlcat(fname, ".", MAXPATHLEN);
915 }
916 }
917
918 if (link_stat(fname, &st) != 0) {
919 if (f != -1) {
920 io_error = 1;
921 rprintf(FERROR, "link_stat %s : %s\n",
922 fname, strerror(errno));
923 }
924 continue;
925 }
926
927 if (S_ISDIR(st.st_mode) && !recurse) {
928 rprintf(FINFO, "skipping directory %s\n", fname);
929 continue;
930 }
931
932 dir = NULL;
933 olddir = NULL;
934
935 if (!relative_paths) {
936 p = strrchr(fname, '/');
937 if (p) {
938 *p = 0;
939 if (p == fname)
940 dir = "/";
941 else
942 dir = fname;
943 fname = p + 1;
944 }
945 } else if (f != -1 && (p=strrchr(fname,'/')) && p != fname) {
946 /* this ensures we send the intermediate directories,
947 thus getting their permissions right */
948 char *lp = lastpath, *fn = fname, *slash = fname;
949 *p = 0;
950 /* Skip any initial directories in our path that we
951 * have in common with lastpath. */
952 while (*fn && *lp == *fn) {
953 if (*fn == '/')
954 slash = fn;
955 lp++, fn++;
956 }
957 *p = '/';
958 if (fn != p || (*lp && *lp != '/')) {
959 int copy_links_saved = copy_links;
960 int recurse_saved = recurse;
961 copy_links = copy_unsafe_links;
962 /* set recurse to 1 to prevent make_file
963 * from ignoring directory, but still
964 * turn off the recursive parameter to
965 * send_file_name */
966 recurse = 1;
967 while ((slash = strchr(slash+1, '/')) != 0) {
968 *slash = 0;
969 send_file_name(f, flist, fname, 0, 0);
970 *slash = '/';
971 }
972 copy_links = copy_links_saved;
973 recurse = recurse_saved;
974 *p = 0;
975 strlcpy(lastpath, fname, sizeof lastpath);
976 *p = '/';
977 }
978 }
979
980 if (!*fname)
981 fname = ".";
982
983 if (dir && *dir) {
984 olddir = push_dir(dir, 1);
985
986 if (!olddir) {
987 io_error = 1;
988 rprintf(FERROR, "push_dir %s : %s\n",
989 dir, strerror(errno));
990 continue;
991 }
992
993 flist_dir = dir;
994 }
995
996 if (one_file_system)
997 set_filesystem(fname);
998
999 send_file_name(f, flist, fname, recurse, FLAG_DELETE);
1000
1001 if (olddir != NULL) {
1002 flist_dir = NULL;
1003 if (pop_dir(olddir) != 0) {
1004 rprintf(FERROR, "pop_dir %s : %s\n",
1005 dir, strerror(errno));
1006 exit_cleanup(RERR_FILESELECT);
1007 }
1008 }
1009 }
1010
1011 if (f != -1) {
1012 send_file_entry(NULL, f, 0);
1013 }
1014
1015 if (show_filelist_p() && f != -1) {
1016 finish_filelist_progress(flist);
1017 }
1018
1019 clean_flist(flist, 0, 0);
1020
1021 /* now send the uid/gid list. This was introduced in protocol
1022 version 15 */
1023 if (f != -1 && remote_version >= 15) {
1024 send_uid_list(f);
1025 }
1026
1027 /* if protocol version is >= 17 then send the io_error flag */
1028 if (f != -1 && remote_version >= 17) {
1029 extern int module_id;
1030 write_int(f, lp_ignore_errors(module_id) ? 0 : io_error);
1031 }
1032
1033 if (f != -1) {
1034 io_end_buffering();
1035 stats.flist_size = stats.total_written - start_write;
1036 stats.num_files = flist->count;
1037 if (write_batch) /* dw */
1038 write_batch_flist_info(flist->count, flist->files);
1039 }
1040
1041 if (verbose > 2)
1042 rprintf(FINFO, "send_file_list done\n");
1043
1044 return flist;
1045}
1046
1047
1048struct file_list *recv_file_list(int f)
1049{
1050 struct file_list *flist;
1051 unsigned char flags;
1052 int64 start_read;
1053 extern int list_only;
1054
1055 if (show_filelist_p())
1056 start_filelist_progress("receiving file list");
1057
1058 start_read = stats.total_read;
1059
1060 flist = (struct file_list *) malloc(sizeof(flist[0]));
1061 if (!flist)
1062 goto oom;
1063
1064 flist->count = 0;
1065 flist->malloced = 1000;
1066 flist->files =
1067 (struct file_struct **) malloc(sizeof(flist->files[0]) *
1068 flist->malloced);
1069 if (!flist->files)
1070 goto oom;
1071
1072
1073 for (flags = read_byte(f); flags; flags = read_byte(f)) {
1074 int i = flist->count;
1075
1076 flist_expand(flist);
1077
1078 receive_file_entry(&flist->files[i], flags, f);
1079
1080 if (S_ISREG(flist->files[i]->mode))
1081 stats.total_size += flist->files[i]->length;
1082
1083 flist->count++;
1084
1085 maybe_emit_filelist_progress(flist);
1086
1087 if (verbose > 2)
1088 rprintf(FINFO, "recv_file_name(%s)\n",
1089 f_name(flist->files[i]));
1090 }
1091
1092
1093 if (verbose > 2)
1094 rprintf(FINFO, "received %d names\n", flist->count);
1095
1096 clean_flist(flist, relative_paths, 1);
1097
1098 if (show_filelist_p()) {
1099 finish_filelist_progress(flist);
1100 }
1101
1102 /* now recv the uid/gid list. This was introduced in protocol version 15 */
1103 if (f != -1 && remote_version >= 15) {
1104 recv_uid_list(f, flist);
1105 }
1106
1107 /* if protocol version is >= 17 then recv the io_error flag */
1108 if (f != -1 && remote_version >= 17 && !read_batch) { /* dw-added readbatch */
1109 extern int module_id;
1110 extern int ignore_errors;
1111 if (lp_ignore_errors(module_id) || ignore_errors) {
1112 read_int(f);
1113 } else {
1114 io_error |= read_int(f);
1115 }
1116 }
1117
1118 if (list_only) {
1119 int i;
1120 for (i = 0; i < flist->count; i++) {
1121 list_file_entry(flist->files[i]);
1122 }
1123 }
1124
1125
1126 if (verbose > 2)
1127 rprintf(FINFO, "recv_file_list done\n");
1128
1129 stats.flist_size = stats.total_read - start_read;
1130 stats.num_files = flist->count;
1131
1132 return flist;
1133
1134 oom:
1135 out_of_memory("recv_file_list");
1136 return NULL; /* not reached */
1137}
1138
1139
1140/*
1141 * XXX: This is currently the hottest function while building the file
1142 * list, because building f_name()s every time is expensive.
1143 **/
1144int file_compare(struct file_struct **f1, struct file_struct **f2)
1145{
1146 if (!(*f1)->basename && !(*f2)->basename)
1147 return 0;
1148 if (!(*f1)->basename)
1149 return -1;
1150 if (!(*f2)->basename)
1151 return 1;
1152 if ((*f1)->dirname == (*f2)->dirname)
1153 return u_strcmp((*f1)->basename, (*f2)->basename);
1154 return u_strcmp(f_name(*f1), f_name(*f2));
1155}
1156
1157
1158int flist_find(struct file_list *flist, struct file_struct *f)
1159{
1160 int low = 0, high = flist->count - 1;
1161
1162 while (high >= 0 && !flist->files[high]->basename) high--;
1163
1164 if (high < 0)
1165 return -1;
1166
1167 while (low != high) {
1168 int mid = (low + high) / 2;
1169 int ret =
1170 file_compare(&flist->files[flist_up(flist, mid)], &f);
1171 if (ret == 0)
1172 return flist_up(flist, mid);
1173 if (ret > 0) {
1174 high = mid;
1175 } else {
1176 low = mid + 1;
1177 }
1178 }
1179
1180 if (file_compare(&flist->files[flist_up(flist, low)], &f) == 0)
1181 return flist_up(flist, low);
1182 return -1;
1183}
1184
1185
1186/*
1187 * free up one file
1188 */
1189void free_file(struct file_struct *file)
1190{
1191 if (!file)
1192 return;
1193 if (file->basename)
1194 free(file->basename);
1195 if (file->link)
1196 free(file->link);
1197 if (file->sum)
1198 free(file->sum);
1199 *file = null_file;
1200}
1201
1202
1203/*
1204 * allocate a new file list
1205 */
1206struct file_list *flist_new(void)
1207{
1208 struct file_list *flist;
1209
1210 flist = (struct file_list *) malloc(sizeof(flist[0]));
1211 if (!flist)
1212 out_of_memory("send_file_list");
1213
1214 flist->count = 0;
1215 flist->malloced = 0;
1216 flist->files = NULL;
1217
1218#if ARENA_SIZE > 0
1219 flist->string_area = string_area_new(0);
1220#else
1221 flist->string_area = NULL;
1222#endif
1223 return flist;
1224}
1225
1226/*
1227 * free up all elements in a flist
1228 */
1229void flist_free(struct file_list *flist)
1230{
1231 int i;
1232 for (i = 1; i < flist->count; i++) {
1233 if (!flist->string_area)
1234 free_file(flist->files[i]);
1235 free(flist->files[i]);
1236 }
1237 /* FIXME: I don't think we generally need to blank the flist
1238 * since it's about to be freed. This will just cause more
1239 * memory traffic. If you want a freed-memory debugger, you
1240 * know where to get it. */
1241 memset((char *) flist->files, 0,
1242 sizeof(flist->files[0]) * flist->count);
1243 free(flist->files);
1244 if (flist->string_area)
1245 string_area_free(flist->string_area);
1246 memset((char *) flist, 0, sizeof(*flist));
1247 free(flist);
1248}
1249
1250
1251/*
1252 * This routine ensures we don't have any duplicate names in our file list.
1253 * duplicate names can cause corruption because of the pipelining
1254 */
1255static void clean_flist(struct file_list *flist, int strip_root, int no_dups)
1256{
1257 int i;
1258 char *name, *prev_name = NULL;
1259
1260 if (!flist || flist->count == 0)
1261 return;
1262
1263 qsort(flist->files, flist->count,
1264 sizeof(flist->files[0]), (int (*)()) file_compare);
1265
1266 for (i = no_dups? 0 : flist->count; i < flist->count; i++) {
1267 if (flist->files[i]->basename) {
1268 prev_name = f_name(flist->files[i]);
1269 break;
1270 }
1271 }
1272 while (++i < flist->count) {
1273 if (!flist->files[i]->basename)
1274 continue;
1275 name = f_name(flist->files[i]);
1276 if (strcmp(name, prev_name) == 0) {
1277 if (verbose > 1 && !am_server) {
1278 rprintf(FINFO,
1279 "removing duplicate name %s from file list %d\n",
1280 name, i);
1281 }
1282 /* it's not great that the flist knows the semantics of
1283 * the file memory usage, but i'd rather not add a flag
1284 * byte to that struct.
1285 * XXX can i use a bit in the flags field? */
1286 if (flist->string_area)
1287 flist->files[i][0] = null_file;
1288 else
1289 free_file(flist->files[i]);
1290 }
1291 prev_name = name;
1292 }
1293
1294 if (strip_root) {
1295 /* we need to strip off the root directory in the case
1296 of relative paths, but this must be done _after_
1297 the sorting phase */
1298 for (i = 0; i < flist->count; i++) {
1299 if (flist->files[i]->dirname &&
1300 flist->files[i]->dirname[0] == '/') {
1301 memmove(&flist->files[i]->dirname[0],
1302 &flist->files[i]->dirname[1],
1303 strlen(flist->files[i]->dirname));
1304 }
1305
1306 if (flist->files[i]->dirname &&
1307 !flist->files[i]->dirname[0]) {
1308 flist->files[i]->dirname = NULL;
1309 }
1310 }
1311 }
1312
1313 if (verbose <= 3)
1314 return;
1315
1316 for (i = 0; i < flist->count; i++) {
1317 rprintf(FINFO, "[%d] i=%d %s %s mode=0%o len=%.0f\n",
1318 (int) getpid(), i,
1319 NS(flist->files[i]->dirname),
1320 NS(flist->files[i]->basename),
1321 (int) flist->files[i]->mode,
1322 (double) flist->files[i]->length);
1323 }
1324}
1325
1326
1327/*
1328 * return the full filename of a flist entry
1329 *
1330 * This function is too expensive at the moment, because it copies
1331 * strings when often we only want to compare them. In any case,
1332 * using strlcat is silly because it will walk the string repeatedly.
1333 */
1334char *f_name(struct file_struct *f)
1335{
1336 static char names[10][MAXPATHLEN];
1337 static int n;
1338 char *p = names[n];
1339
1340 if (!f || !f->basename)
1341 return NULL;
1342
1343 n = (n + 1) % 10;
1344
1345 if (f->dirname) {
1346 int off;
1347
1348 off = strlcpy(p, f->dirname, MAXPATHLEN);
1349 off += strlcpy(p + off, "/", MAXPATHLEN - off);
1350 off += strlcpy(p + off, f->basename, MAXPATHLEN - off);
1351 } else {
1352 strlcpy(p, f->basename, MAXPATHLEN);
1353 }
1354
1355 return p;
1356}