Corrected sizeof usage:
[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 * @sa http://lists.samba.org/pipermail/rsync/2000-June/002351.html
25 *
26 **/
27
28#include "rsync.h"
29
30extern struct stats stats;
31
32extern int verbose;
33extern int do_progress;
34extern int am_server;
35extern int always_checksum;
36extern int module_id;
37extern int ignore_errors;
38
39extern int cvs_exclude;
40
41extern int recurse;
42extern char curr_dir[MAXPATHLEN];
43extern char *files_from;
44extern int filesfrom_fd;
45
46extern int one_file_system;
47extern int make_backups;
48extern int preserve_links;
49extern int preserve_hard_links;
50extern int preserve_perms;
51extern int preserve_devices;
52extern int preserve_uid;
53extern int preserve_gid;
54extern int preserve_times;
55extern int relative_paths;
56extern int implied_dirs;
57extern int copy_links;
58extern int copy_unsafe_links;
59extern int protocol_version;
60extern int sanitize_paths;
61
62extern int read_batch;
63extern int write_batch;
64
65extern struct exclude_struct **exclude_list;
66extern struct exclude_struct **server_exclude_list;
67extern struct exclude_struct **local_exclude_list;
68
69int io_error;
70
71static char empty_sum[MD4_SUM_LENGTH];
72static unsigned int min_file_struct_len;
73
74static void clean_flist(struct file_list *flist, int strip_root, int no_dups);
75
76
77void init_flist(void)
78{
79 struct file_struct f;
80
81 /* Figure out how big the file_struct is without trailing padding */
82 min_file_struct_len = ((char*)&f.flags - (char*)&f) + sizeof f.flags;
83}
84
85
86static int show_filelist_p(void)
87{
88 return verbose && (recurse || files_from) && !am_server;
89}
90
91static void start_filelist_progress(char *kind)
92{
93 rprintf(FINFO, "%s ... ", kind);
94 if ((verbose > 1) || do_progress)
95 rprintf(FINFO, "\n");
96 rflush(FINFO);
97}
98
99
100static void emit_filelist_progress(const struct file_list *flist)
101{
102 rprintf(FINFO, " %d files...\r", flist->count);
103}
104
105
106static void maybe_emit_filelist_progress(const struct file_list *flist)
107{
108 if (do_progress && show_filelist_p() && ((flist->count % 100) == 0))
109 emit_filelist_progress(flist);
110}
111
112
113static void finish_filelist_progress(const struct file_list *flist)
114{
115 if (do_progress) {
116 /* This overwrites the progress line */
117 rprintf(FINFO, "%d file%sto consider\n",
118 flist->count, flist->count == 1 ? " " : "s ");
119 } else
120 rprintf(FINFO, "done\n");
121}
122
123void show_flist_stats(void)
124{
125 /* Nothing yet */
126}
127
128
129static void list_file_entry(struct file_struct *f)
130{
131 char perms[11];
132
133 if (!f->basename)
134 /* this can happen if duplicate names were removed */
135 return;
136
137 permstring(perms, f->mode);
138
139#if SUPPORT_LINKS
140 if (preserve_links && S_ISLNK(f->mode)) {
141 rprintf(FINFO, "%s %11.0f %s %s -> %s\n",
142 perms,
143 (double) f->length, timestring(f->modtime),
144 f_name(f), f->u.link);
145 } else
146#endif
147 rprintf(FINFO, "%s %11.0f %s %s\n",
148 perms,
149 (double) f->length, timestring(f->modtime),
150 f_name(f));
151}
152
153
154/**
155 * Stat either a symlink or its referent, depending on the settings of
156 * copy_links, copy_unsafe_links, etc.
157 *
158 * @retval -1 on error
159 *
160 * @retval 0 for success
161 *
162 * @post If @p path is a symlink, then @p linkbuf (of size @c
163 * MAXPATHLEN) contains the symlink target.
164 *
165 * @post @p buffer contains information about the link or the
166 * referrent as appropriate, if they exist.
167 **/
168int readlink_stat(const char *path, STRUCT_STAT *buffer, char *linkbuf)
169{
170#if SUPPORT_LINKS
171 if (copy_links)
172 return do_stat(path, buffer);
173 if (do_lstat(path, buffer) == -1)
174 return -1;
175 if (S_ISLNK(buffer->st_mode)) {
176 int l = readlink((char *) path, linkbuf, MAXPATHLEN - 1);
177 if (l == -1)
178 return -1;
179 linkbuf[l] = 0;
180 if (copy_unsafe_links && unsafe_symlink(linkbuf, path)) {
181 if (verbose > 1) {
182 rprintf(FINFO,"copying unsafe symlink \"%s\" -> \"%s\"\n",
183 path, linkbuf);
184 }
185 return do_stat(path, buffer);
186 }
187 }
188 return 0;
189#else
190 return do_stat(path, buffer);
191#endif
192}
193
194int link_stat(const char *path, STRUCT_STAT * buffer)
195{
196#if SUPPORT_LINKS
197 if (copy_links)
198 return do_stat(path, buffer);
199 return do_lstat(path, buffer);
200#else
201 return do_stat(path, buffer);
202#endif
203}
204
205/*
206 * This function is used to check if a file should be included/excluded
207 * from the list of files based on its name and type etc. The value of
208 * exclude_level is set to either SERVER_EXCLUDES or ALL_EXCLUDES.
209 */
210static int check_exclude_file(char *fname, int is_dir, int exclude_level)
211{
212#if 0 /* This currently never happens, so avoid a useless compare. */
213 if (exclude_level == NO_EXCLUDES)
214 return 0;
215#endif
216 if (fname) {
217 /* never exclude '.', even if somebody does --exclude '*' */
218 if (fname[0] == '.' && !fname[1])
219 return 0;
220 /* Handle the -R version of the '.' dir. */
221 if (fname[0] == '/') {
222 int len = strlen(fname);
223 if (fname[len-1] == '.' && fname[len-2] == '/')
224 return 0;
225 }
226 }
227 if (server_exclude_list
228 && check_exclude(server_exclude_list, fname, is_dir))
229 return 1;
230 if (exclude_level != ALL_EXCLUDES)
231 return 0;
232 if (exclude_list && check_exclude(exclude_list, fname, is_dir))
233 return 1;
234 if (local_exclude_list
235 && check_exclude(local_exclude_list, fname, is_dir))
236 return 1;
237 return 0;
238}
239
240/* used by the one_file_system code */
241static dev_t filesystem_dev;
242
243static void set_filesystem(char *fname)
244{
245 STRUCT_STAT st;
246 if (link_stat(fname, &st) != 0)
247 return;
248 filesystem_dev = st.st_dev;
249}
250
251
252static int to_wire_mode(mode_t mode)
253{
254#if SUPPORT_LINKS
255 if (S_ISLNK(mode) && (_S_IFLNK != 0120000))
256 return (mode & ~(_S_IFMT)) | 0120000;
257#endif
258 return (int) mode;
259}
260
261static mode_t from_wire_mode(int mode)
262{
263 if ((mode & (_S_IFMT)) == 0120000 && (_S_IFLNK != 0120000))
264 return (mode & ~(_S_IFMT)) | _S_IFLNK;
265 return (mode_t) mode;
266}
267
268
269static void send_directory(int f, struct file_list *flist, char *dir);
270
271static char *flist_dir;
272static int flist_dir_len;
273
274
275/**
276 * Make sure @p flist is big enough to hold at least @p flist->count
277 * entries.
278 **/
279static void flist_expand(struct file_list *flist)
280{
281 if (flist->count >= flist->malloced) {
282 void *new_ptr;
283
284 if (flist->malloced < 1000)
285 flist->malloced += 1000;
286 else
287 flist->malloced *= 2;
288
289 if (flist->files) {
290 new_ptr = realloc_array(flist->files,
291 struct file_struct *,
292 flist->malloced);
293 } else {
294 new_ptr = new_array(struct file_struct *,
295 flist->malloced);
296 }
297
298 if (verbose >= 2) {
299 rprintf(FINFO, "[%s] expand file_list to %.0f bytes, did%s move\n",
300 who_am_i(),
301 (double) sizeof flist->files[0] * flist->malloced,
302 (new_ptr == flist->files) ? " not" : "");
303 }
304
305 flist->files = (struct file_struct **) new_ptr;
306
307 if (!flist->files)
308 out_of_memory("flist_expand");
309 }
310}
311
312void send_file_entry(struct file_struct *file, int f, unsigned short base_flags)
313{
314 unsigned short flags;
315 static time_t modtime;
316 static mode_t mode;
317 static DEV64_T rdev, rdev_high;
318 static DEV64_T dev;
319 static uid_t uid;
320 static gid_t gid;
321 static char lastname[MAXPATHLEN];
322 char *fname, fbuf[MAXPATHLEN];
323 int l1, l2;
324
325 if (f == -1)
326 return;
327
328 if (!file) {
329 write_byte(f, 0);
330 modtime = 0, mode = 0;
331 rdev = 0, rdev_high = 0, dev = 0;
332 uid = 0, gid = 0;
333 *lastname = '\0';
334 return;
335 }
336
337 io_write_phase = "send_file_entry";
338
339 fname = f_name_to(file, fbuf);
340
341 flags = base_flags;
342
343 if (file->mode == mode)
344 flags |= XMIT_SAME_MODE;
345 else
346 mode = file->mode;
347 if (preserve_devices) {
348 if (protocol_version < 28) {
349 if (IS_DEVICE(mode)) {
350 if (file->u.rdev == rdev) {
351 /* Set both flags to simplify the test
352 * when writing the data. */
353 flags |= XMIT_SAME_RDEV_pre28
354 | XMIT_SAME_HIGH_RDEV;
355 } else
356 rdev = file->u.rdev;
357 } else
358 rdev = 0;
359 } else if (IS_DEVICE(mode)) {
360 if ((file->u.rdev & ~0xFF) == rdev_high)
361 flags |= XMIT_SAME_HIGH_RDEV;
362 else {
363 rdev = file->u.rdev;
364 rdev_high = rdev & ~0xFF;
365 }
366 }
367 }
368 if (file->uid == uid)
369 flags |= XMIT_SAME_UID;
370 else
371 uid = file->uid;
372 if (file->gid == gid)
373 flags |= XMIT_SAME_GID;
374 else
375 gid = file->gid;
376 if (file->modtime == modtime)
377 flags |= XMIT_SAME_TIME;
378 else
379 modtime = file->modtime;
380
381#if SUPPORT_HARD_LINKS
382 if (file->link_u.idev) {
383 if (file->F_DEV == dev) {
384 if (protocol_version >= 28)
385 flags |= XMIT_SAME_DEV;
386 } else
387 dev = file->F_DEV;
388 flags |= XMIT_HAS_IDEV_DATA;
389 }
390#endif
391
392 for (l1 = 0;
393 lastname[l1] && (fname[l1] == lastname[l1]) && (l1 < 255);
394 l1++) {}
395 l2 = strlen(fname+l1);
396
397 if (l1 > 0)
398 flags |= XMIT_SAME_NAME;
399 if (l2 > 255)
400 flags |= XMIT_LONG_NAME;
401
402 /* We must make sure we don't send a zero flag byte or the
403 * other end will terminate the flist transfer. Note that
404 * the use of XMIT_TOP_DIR on a non-dir has no meaning, so
405 * it's harmless way to add a bit to the first flag byte. */
406 if (protocol_version >= 28) {
407 if (!flags && !S_ISDIR(mode))
408 flags |= XMIT_TOP_DIR;
409 if ((flags & 0xFF00) || !flags) {
410 flags |= XMIT_EXTENDED_FLAGS;
411 write_byte(f, flags);
412 write_byte(f, flags >> 8);
413 } else
414 write_byte(f, flags);
415 } else {
416 if (!(flags & 0xFF) && !S_ISDIR(mode))
417 flags |= XMIT_TOP_DIR;
418 if (!(flags & 0xFF))
419 flags |= XMIT_LONG_NAME;
420 write_byte(f, flags);
421 }
422 if (flags & XMIT_SAME_NAME)
423 write_byte(f, l1);
424 if (flags & XMIT_LONG_NAME)
425 write_int(f, l2);
426 else
427 write_byte(f, l2);
428 write_buf(f, fname + l1, l2);
429
430 write_longint(f, file->length);
431 if (!(flags & XMIT_SAME_TIME))
432 write_int(f, modtime);
433 if (!(flags & XMIT_SAME_MODE))
434 write_int(f, to_wire_mode(mode));
435 if (preserve_uid && !(flags & XMIT_SAME_UID)) {
436 add_uid(uid);
437 write_int(f, uid);
438 }
439 if (preserve_gid && !(flags & XMIT_SAME_GID)) {
440 add_gid(gid);
441 write_int(f, gid);
442 }
443 if (preserve_devices && IS_DEVICE(mode)) {
444 /* If XMIT_SAME_HIGH_RDEV is off, XMIT_SAME_RDEV_pre28 is
445 * also off. */
446 if (!(flags & XMIT_SAME_HIGH_RDEV))
447 write_int(f, rdev);
448 else if (protocol_version >= 28)
449 write_byte(f, rdev);
450 }
451
452#if SUPPORT_LINKS
453 if (preserve_links && S_ISLNK(mode)) {
454 int len = strlen(file->u.link);
455 write_int(f, len);
456 write_buf(f, file->u.link, len);
457 }
458#endif
459
460#if SUPPORT_HARD_LINKS
461 if (flags & XMIT_HAS_IDEV_DATA) {
462 if (protocol_version < 26) {
463 /* 32-bit dev_t and ino_t */
464 write_int(f, dev);
465 write_int(f, file->F_INODE);
466 } else {
467 /* 64-bit dev_t and ino_t */
468 if (!(flags & XMIT_SAME_DEV))
469 write_longint(f, dev);
470 write_longint(f, file->F_INODE);
471 }
472 }
473#endif
474
475 if (always_checksum) {
476 char *sum;
477 if (S_ISREG(mode))
478 sum = file->u.sum;
479 else if (protocol_version < 28) {
480 /* Prior to 28, we sent a useless set of nulls. */
481 sum = empty_sum;
482 } else
483 sum = NULL;
484 if (sum) {
485 write_buf(f, sum, protocol_version < 21? 2
486 : MD4_SUM_LENGTH);
487 }
488 }
489
490 strlcpy(lastname, fname, MAXPATHLEN);
491
492 io_write_phase = "unknown";
493}
494
495
496
497void receive_file_entry(struct file_struct **fptr, unsigned short flags, int f)
498{
499 static time_t modtime;
500 static mode_t mode;
501 static DEV64_T rdev, rdev_high;
502 static DEV64_T dev;
503 static uid_t uid;
504 static gid_t gid;
505 static char lastname[MAXPATHLEN], *lastdir;
506 static int lastdir_len = -1;
507 char thisname[MAXPATHLEN];
508 unsigned int l1 = 0, l2 = 0;
509 int alloc_len, basename_len, dirname_len, linkname_len, sum_len;
510 int file_struct_len, idev_len;
511 OFF_T file_length;
512 char *basename, *dirname, *bp;
513 struct file_struct *file;
514
515 if (!fptr) {
516 modtime = 0, mode = 0;
517 rdev = 0, rdev_high = 0, dev = 0;
518 uid = 0, gid = 0;
519 *lastname = '\0';
520 return;
521 }
522
523 if (flags & XMIT_SAME_NAME)
524 l1 = read_byte(f);
525
526 if (flags & XMIT_LONG_NAME)
527 l2 = read_int(f);
528 else
529 l2 = read_byte(f);
530
531 if (l2 >= MAXPATHLEN - l1) {
532 rprintf(FERROR,
533 "overflow: flags=0x%x l1=%d l2=%d lastname=%s\n",
534 flags, l1, l2, lastname);
535 overflow("receive_file_entry");
536 }
537
538 strlcpy(thisname, lastname, l1 + 1);
539 read_sbuf(f, &thisname[l1], l2);
540 thisname[l1 + l2] = 0;
541
542 strlcpy(lastname, thisname, MAXPATHLEN);
543
544 clean_fname(thisname);
545
546 if (sanitize_paths)
547 sanitize_path(thisname, NULL);
548
549 if ((basename = strrchr(thisname, '/')) != NULL) {
550 dirname_len = ++basename - thisname; /* counts future '\0' */
551 if (lastdir_len == dirname_len - 1
552 && strncmp(thisname, lastdir, lastdir_len) == 0) {
553 dirname = lastdir;
554 dirname_len = 0; /* indicates no copy is needed */
555 } else
556 dirname = thisname;
557 } else {
558 basename = thisname;
559 dirname = NULL;
560 dirname_len = 0;
561 }
562 basename_len = strlen(basename) + 1; /* count the '\0' */
563
564 file_length = read_longint(f);
565 if (!(flags & XMIT_SAME_TIME))
566 modtime = (time_t)read_int(f);
567 if (!(flags & XMIT_SAME_MODE))
568 mode = from_wire_mode(read_int(f));
569
570 if (preserve_uid && !(flags & XMIT_SAME_UID))
571 uid = (uid_t)read_int(f);
572 if (preserve_gid && !(flags & XMIT_SAME_GID))
573 gid = (gid_t)read_int(f);
574
575 if (preserve_devices) {
576 if (protocol_version < 28) {
577 if (IS_DEVICE(mode)) {
578 if (!(flags & XMIT_SAME_RDEV_pre28))
579 rdev = (DEV64_T)read_int(f);
580 } else
581 rdev = 0;
582 } else if (IS_DEVICE(mode)) {
583 if (!(flags & XMIT_SAME_HIGH_RDEV)) {
584 rdev = (DEV64_T)read_int(f);
585 rdev_high = rdev & ~0xFF;
586 } else
587 rdev = rdev_high | (DEV64_T)read_byte(f);
588 }
589 }
590
591#if SUPPORT_LINKS
592 if (preserve_links && S_ISLNK(mode)) {
593 linkname_len = read_int(f) + 1; /* count the '\0' */
594 if (linkname_len <= 0 || linkname_len > MAXPATHLEN) {
595 rprintf(FERROR, "overflow: linkname_len=%d\n",
596 linkname_len - 1);
597 overflow("receive_file_entry");
598 }
599 }
600 else
601#endif
602 linkname_len = 0;
603
604#if SUPPORT_HARD_LINKS
605 if (preserve_hard_links && protocol_version < 28 && S_ISREG(mode))
606 flags |= XMIT_HAS_IDEV_DATA;
607 if (flags & XMIT_HAS_IDEV_DATA)
608 idev_len = sizeof (struct idev);
609 else
610#endif
611 idev_len = 0;
612
613 sum_len = always_checksum && S_ISREG(mode) ? MD4_SUM_LENGTH : 0;
614 file_struct_len = idev_len? sizeof file[0] : min_file_struct_len;
615
616 alloc_len = file_struct_len + dirname_len + basename_len
617 + linkname_len + sum_len + idev_len;
618 if (!(bp = new_array(char, alloc_len)))
619 out_of_memory("receive_file_entry");
620 file = *fptr = (struct file_struct *)bp;
621 memset(bp, 0, min_file_struct_len);
622 bp += file_struct_len;
623
624 file->flags = flags & XMIT_TOP_DIR ? FLAG_TOP_DIR : 0;
625 file->modtime = modtime;
626 file->length = file_length;
627 file->mode = mode;
628 file->uid = uid;
629 file->gid = gid;
630
631#if SUPPORT_HARD_LINKS
632 if (idev_len) {
633 file->link_u.idev = (struct idev *)bp;
634 bp += idev_len;
635 }
636#endif
637
638 if (dirname_len) {
639 file->dirname = lastdir = bp;
640 lastdir_len = dirname_len - 1;
641 memcpy(bp, dirname, dirname_len - 1);
642 bp += dirname_len;
643 bp[-1] = '\0';
644 } else if (dirname)
645 file->dirname = dirname;
646
647 file->basename = bp;
648 memcpy(bp, basename, basename_len);
649 bp += basename_len;
650
651 if (preserve_devices && IS_DEVICE(mode))
652 file->u.rdev = rdev;
653
654#if SUPPORT_LINKS
655 if (linkname_len) {
656 file->u.link = bp;
657 read_sbuf(f, bp, linkname_len - 1);
658 if (sanitize_paths)
659 sanitize_path(bp, lastdir);
660 bp += linkname_len;
661 }
662#endif
663
664#if SUPPORT_HARD_LINKS
665 if (idev_len) {
666 if (protocol_version < 26) {
667 dev = read_int(f);
668 file->F_INODE = read_int(f);
669 } else {
670 if (!(flags & XMIT_SAME_DEV))
671 dev = read_longint(f);
672 file->F_INODE = read_longint(f);
673 }
674 file->F_DEV = dev;
675 }
676#endif
677
678 if (always_checksum) {
679 char *sum;
680 if (sum_len) {
681 file->u.sum = sum = bp;
682 /*bp += sum_len;*/
683 } else if (protocol_version < 28) {
684 /* Prior to 28, we get a useless set of nulls. */
685 sum = empty_sum;
686 } else
687 sum = NULL;
688 if (sum) {
689 read_buf(f, sum, protocol_version < 21? 2
690 : MD4_SUM_LENGTH);
691 }
692 }
693
694 if (!preserve_perms) {
695 extern int orig_umask;
696 /* set an appropriate set of permissions based on original
697 * permissions and umask. This emulates what GNU cp does */
698 file->mode &= ~orig_umask;
699 }
700}
701
702
703/**
704 * Create a file_struct for a named file by reading its stat()
705 * information and performing extensive checks against global
706 * options.
707 *
708 * @return the new file, or NULL if there was an error or this file
709 * should be excluded.
710 *
711 * @todo There is a small optimization opportunity here to avoid
712 * stat()ing the file in some circumstances, which has a certain cost.
713 * We are called immediately after doing readdir(), and so we may
714 * already know the d_type of the file. We could for example avoid
715 * statting directories if we're not recursing, but this is not a very
716 * important case. Some systems may not have d_type.
717 **/
718struct file_struct *make_file(char *fname, int exclude_level)
719{
720 static char *lastdir;
721 static int lastdir_len = -1;
722 struct file_struct *file;
723 STRUCT_STAT st;
724 char sum[SUM_LENGTH];
725 char thisname[MAXPATHLEN];
726 char linkname[MAXPATHLEN];
727 int alloc_len, basename_len, dirname_len, linkname_len, sum_len;
728 int file_struct_len, idev_len;
729 char *basename, *dirname, *bp;
730 unsigned short flags = 0;
731
732 if (strlcpy(thisname, fname, sizeof thisname)
733 >= sizeof thisname - flist_dir_len) {
734 rprintf(FINFO, "skipping overly long name: %s\n", fname);
735 return NULL;
736 }
737 clean_fname(thisname);
738 if (sanitize_paths)
739 sanitize_path(thisname, NULL);
740
741 memset(sum, 0, SUM_LENGTH);
742
743 if (readlink_stat(thisname, &st, linkname) != 0) {
744 int save_errno = errno;
745 if (errno == ENOENT && exclude_level != NO_EXCLUDES) {
746 /* either symlink pointing nowhere or file that
747 * was removed during rsync run; see if excluded
748 * before reporting an error */
749 if (check_exclude_file(thisname, 0, exclude_level)) {
750 /* file is excluded anyway, ignore silently */
751 return NULL;
752 }
753 }
754 io_error |= IOERR_GENERAL;
755 rprintf(FERROR, "readlink %s failed: %s\n",
756 full_fname(thisname), strerror(save_errno));
757 return NULL;
758 }
759
760 /* backup.c calls us with exclude_level set to NO_EXCLUDES. */
761 if (exclude_level == NO_EXCLUDES)
762 goto skip_excludes;
763
764 if (S_ISDIR(st.st_mode) && !recurse && !files_from) {
765 rprintf(FINFO, "skipping directory %s\n", thisname);
766 return NULL;
767 }
768
769 if (one_file_system && st.st_dev != filesystem_dev) {
770 /* We allow a directory though to preserve the mount point.
771 * However, flag it so that we don't recurse. */
772 if (!S_ISDIR(st.st_mode))
773 return NULL;
774 flags |= FLAG_MOUNT_POINT;
775 }
776
777 if (check_exclude_file(thisname, S_ISDIR(st.st_mode) != 0, exclude_level))
778 return NULL;
779
780 if (lp_ignore_nonreadable(module_id) && access(thisname, R_OK) != 0)
781 return NULL;
782
783 skip_excludes:
784
785 if (verbose > 2) {
786 rprintf(FINFO, "[%s] make_file(%s,*,%d)\n",
787 who_am_i(), thisname, exclude_level);
788 }
789
790 if ((basename = strrchr(thisname, '/')) != NULL) {
791 dirname_len = ++basename - thisname; /* counts future '\0' */
792 if (lastdir_len == dirname_len - 1
793 && strncmp(thisname, lastdir, lastdir_len) == 0) {
794 dirname = lastdir;
795 dirname_len = 0; /* indicates no copy is needed */
796 } else
797 dirname = thisname;
798 } else {
799 basename = thisname;
800 dirname = NULL;
801 dirname_len = 0;
802 }
803 basename_len = strlen(basename) + 1; /* count the '\0' */
804
805#if SUPPORT_LINKS
806 linkname_len = S_ISLNK(st.st_mode) ? strlen(linkname) + 1 : 0;
807#else
808 linkname_len = 0;
809#endif
810
811 idev_len = 0;
812#if SUPPORT_HARD_LINKS
813 if (preserve_hard_links && st.st_nlink > 1) {
814 if (protocol_version < 28) {
815 if (S_ISREG(st.st_mode))
816 idev_len = sizeof (struct idev);
817 } else {
818 if (!S_ISDIR(st.st_mode))
819 idev_len = sizeof (struct idev);
820 }
821 }
822#endif
823
824 sum_len = always_checksum && S_ISREG(st.st_mode) ? MD4_SUM_LENGTH : 0;
825 file_struct_len = idev_len? sizeof file[0] : min_file_struct_len;
826
827 alloc_len = file_struct_len + dirname_len + basename_len
828 + linkname_len + sum_len + idev_len;
829 if (!(bp = new_array(char, alloc_len)))
830 out_of_memory("receive_file_entry");
831 file = (struct file_struct *)bp;
832 memset(bp, 0, min_file_struct_len);
833 bp += file_struct_len;
834
835 file->flags = flags;
836 file->modtime = st.st_mtime;
837 file->length = st.st_size;
838 file->mode = st.st_mode;
839 file->uid = st.st_uid;
840 file->gid = st.st_gid;
841
842#if SUPPORT_HARD_LINKS
843 if (idev_len) {
844 file->link_u.idev = (struct idev *)bp;
845 bp += idev_len;
846 file->F_DEV = st.st_dev;
847 file->F_INODE = st.st_ino;
848 }
849#endif
850
851 if (dirname_len) {
852 file->dirname = lastdir = bp;
853 lastdir_len = dirname_len - 1;
854 memcpy(bp, dirname, dirname_len - 1);
855 bp += dirname_len;
856 bp[-1] = '\0';
857 } else if (dirname)
858 file->dirname = dirname;
859
860 file->basename = bp;
861 memcpy(bp, basename, basename_len);
862 bp += basename_len;
863
864#ifdef HAVE_STRUCT_STAT_ST_RDEV
865 if (preserve_devices && IS_DEVICE(st.st_mode))
866 file->u.rdev = st.st_rdev;
867#endif
868
869#if SUPPORT_LINKS
870 if (linkname_len) {
871 file->u.link = bp;
872 memcpy(bp, linkname, linkname_len);
873 bp += linkname_len;
874 }
875#endif
876
877 if (sum_len) {
878 file->u.sum = bp;
879 file_checksum(thisname, bp, st.st_size);
880 /*bp += sum_len;*/
881 }
882
883 file->basedir = flist_dir;
884
885 if (!S_ISDIR(st.st_mode))
886 stats.total_size += st.st_size;
887
888 return file;
889}
890
891
892void send_file_name(int f, struct file_list *flist, char *fname,
893 int recursive, unsigned short base_flags)
894{
895 struct file_struct *file;
896 char fbuf[MAXPATHLEN];
897 extern int delete_excluded;
898
899 /* f is set to -1 when calculating deletion file list */
900 file = make_file(fname,
901 f == -1 && delete_excluded? SERVER_EXCLUDES
902 : ALL_EXCLUDES);
903
904 if (!file)
905 return;
906
907 maybe_emit_filelist_progress(flist);
908
909 flist_expand(flist);
910
911 if (write_batch)
912 file->flags |= FLAG_TOP_DIR;
913
914 if (file->basename[0]) {
915 flist->files[flist->count++] = file;
916 send_file_entry(file, f, base_flags);
917 }
918
919 if (recursive && S_ISDIR(file->mode)
920 && !(file->flags & FLAG_MOUNT_POINT)) {
921 struct exclude_struct **last_exclude_list = local_exclude_list;
922 send_directory(f, flist, f_name_to(file, fbuf));
923 local_exclude_list = last_exclude_list;
924 return;
925 }
926}
927
928
929static void send_directory(int f, struct file_list *flist, char *dir)
930{
931 DIR *d;
932 struct dirent *di;
933 char fname[MAXPATHLEN];
934 unsigned int offset;
935 char *p;
936
937 d = opendir(dir);
938 if (!d) {
939 io_error |= IOERR_GENERAL;
940 rprintf(FERROR, "opendir %s failed: %s\n",
941 full_fname(dir), strerror(errno));
942 return;
943 }
944
945 offset = strlcpy(fname, dir, MAXPATHLEN);
946 p = fname + offset;
947 if (offset >= MAXPATHLEN || p[-1] != '/') {
948 if (offset >= MAXPATHLEN - 1) {
949 io_error |= IOERR_GENERAL;
950 rprintf(FERROR, "skipping long-named directory: %s\n",
951 full_fname(fname));
952 closedir(d);
953 return;
954 }
955 *p++ = '/';
956 offset++;
957 }
958
959 local_exclude_list = NULL;
960
961 if (cvs_exclude) {
962 if (strlcpy(p, ".cvsignore", MAXPATHLEN - offset)
963 < MAXPATHLEN - offset)
964 add_exclude_file(&local_exclude_list,fname,MISSING_OK,ADD_EXCLUDE);
965 else {
966 io_error |= IOERR_GENERAL;
967 rprintf(FINFO,
968 "cannot cvs-exclude in long-named directory %s\n",
969 full_fname(fname));
970 }
971 }
972
973 for (errno = 0, di = readdir(d); di; errno = 0, di = readdir(d)) {
974 char *dname = d_name(di);
975 if (dname[0] == '.' && (dname[1] == '\0'
976 || (dname[1] == '.' && dname[2] == '\0')))
977 continue;
978 if (strlcpy(p, dname, MAXPATHLEN - offset) < MAXPATHLEN - offset)
979 send_file_name(f, flist, fname, recurse, 0);
980 else {
981 io_error |= IOERR_GENERAL;
982 rprintf(FINFO,
983 "cannot send long-named file %s\n",
984 full_fname(fname));
985 }
986 }
987 if (errno) {
988 io_error |= IOERR_GENERAL;
989 rprintf(FERROR, "readdir(%s): (%d) %s\n",
990 dir, errno, strerror(errno));
991 }
992
993 if (local_exclude_list)
994 free_exclude_list(&local_exclude_list); /* Zeros pointer too */
995
996 closedir(d);
997}
998
999
1000/**
1001 * The delete_files() function in receiver.c sets f to -1 so that we just
1002 * construct the file list in memory without sending it over the wire. It
1003 * also has the side-effect of ignoring user-excludes if delete_excluded
1004 * is set (so that the delete list includes user-excluded files).
1005 **/
1006struct file_list *send_file_list(int f, int argc, char *argv[])
1007{
1008 int l;
1009 STRUCT_STAT st;
1010 char *p, *dir, olddir[sizeof curr_dir];
1011 char lastpath[MAXPATHLEN] = "";
1012 struct file_list *flist;
1013 int64 start_write;
1014 int use_ff_fd = 0;
1015
1016 if (show_filelist_p() && f != -1)
1017 start_filelist_progress("building file list");
1018
1019 start_write = stats.total_written;
1020
1021 flist = flist_new();
1022
1023 if (f != -1) {
1024 io_start_buffering_out(f);
1025 if (filesfrom_fd >= 0) {
1026 if (argv[0] && !push_dir(argv[0])) {
1027 rprintf(FERROR, "push_dir %s failed: %s\n",
1028 full_fname(argv[0]), strerror(errno));
1029 exit_cleanup(RERR_FILESELECT);
1030 }
1031 use_ff_fd = 1;
1032 }
1033 }
1034
1035 while (1) {
1036 char fname2[MAXPATHLEN];
1037 char *fname = fname2;
1038
1039 if (use_ff_fd) {
1040 if (read_filesfrom_line(filesfrom_fd, fname) == 0)
1041 break;
1042 sanitize_path(fname, NULL);
1043 } else {
1044 if (argc-- == 0)
1045 break;
1046 strlcpy(fname, *argv++, MAXPATHLEN);
1047 if (sanitize_paths)
1048 sanitize_path(fname, NULL);
1049 }
1050
1051 l = strlen(fname);
1052 if (fname[l - 1] == '/') {
1053 if (l == 2 && fname[0] == '.') {
1054 /* Turn "./" into just "." rather than "./." */
1055 fname[1] = '\0';
1056 } else if (l < MAXPATHLEN) {
1057 fname[l++] = '.';
1058 fname[l] = '\0';
1059 }
1060 }
1061
1062 if (link_stat(fname, &st) != 0) {
1063 if (f != -1) {
1064 io_error |= IOERR_GENERAL;
1065 rprintf(FERROR, "link_stat %s failed: %s\n",
1066 full_fname(fname), strerror(errno));
1067 }
1068 continue;
1069 }
1070
1071 if (S_ISDIR(st.st_mode) && !recurse && !files_from) {
1072 rprintf(FINFO, "skipping directory %s\n", fname);
1073 continue;
1074 }
1075
1076 dir = NULL;
1077 olddir[0] = '\0';
1078
1079 if (!relative_paths) {
1080 p = strrchr(fname, '/');
1081 if (p) {
1082 *p = 0;
1083 if (p == fname)
1084 dir = "/";
1085 else
1086 dir = fname;
1087 fname = p + 1;
1088 }
1089 } else if (f != -1 && implied_dirs && (p=strrchr(fname,'/')) && p != fname) {
1090 /* this ensures we send the intermediate directories,
1091 thus getting their permissions right */
1092 char *lp = lastpath, *fn = fname, *slash = fname;
1093 *p = 0;
1094 /* Skip any initial directories in our path that we
1095 * have in common with lastpath. */
1096 while (*fn && *lp == *fn) {
1097 if (*fn == '/')
1098 slash = fn;
1099 lp++, fn++;
1100 }
1101 *p = '/';
1102 if (fn != p || (*lp && *lp != '/')) {
1103 int copy_links_saved = copy_links;
1104 int recurse_saved = recurse;
1105 copy_links = copy_unsafe_links;
1106 /* set recurse to 1 to prevent make_file
1107 * from ignoring directory, but still
1108 * turn off the recursive parameter to
1109 * send_file_name */
1110 recurse = 1;
1111 while ((slash = strchr(slash+1, '/')) != 0) {
1112 *slash = 0;
1113 send_file_name(f, flist, fname, 0, 0);
1114 *slash = '/';
1115 }
1116 copy_links = copy_links_saved;
1117 recurse = recurse_saved;
1118 *p = 0;
1119 strlcpy(lastpath, fname, sizeof lastpath);
1120 *p = '/';
1121 }
1122 }
1123
1124 if (!*fname)
1125 fname = ".";
1126
1127 if (dir && *dir) {
1128 static char *lastdir;
1129 static int lastdir_len;
1130
1131 strcpy(olddir, curr_dir); /* can't overflow */
1132
1133 if (!push_dir(dir)) {
1134 io_error |= IOERR_GENERAL;
1135 rprintf(FERROR, "push_dir %s failed: %s\n",
1136 full_fname(dir), strerror(errno));
1137 continue;
1138 }
1139
1140 if (lastdir && strcmp(lastdir, dir) == 0) {
1141 flist_dir = lastdir;
1142 flist_dir_len = lastdir_len;
1143 } else {
1144 flist_dir = lastdir = strdup(dir);
1145 flist_dir_len = lastdir_len = strlen(dir);
1146 }
1147 }
1148
1149 if (one_file_system)
1150 set_filesystem(fname);
1151
1152 send_file_name(f, flist, fname, recurse, XMIT_TOP_DIR);
1153
1154 if (olddir[0]) {
1155 flist_dir = NULL;
1156 flist_dir_len = 0;
1157 if (!pop_dir(olddir)) {
1158 rprintf(FERROR, "pop_dir %s failed: %s\n",
1159 full_fname(dir), strerror(errno));
1160 exit_cleanup(RERR_FILESELECT);
1161 }
1162 }
1163 }
1164
1165 if (f != -1) {
1166 send_file_entry(NULL, f, 0);
1167
1168 if (show_filelist_p())
1169 finish_filelist_progress(flist);
1170 }
1171
1172 clean_flist(flist, 0, 0);
1173
1174 if (f != -1) {
1175 /* Now send the uid/gid list. This was introduced in
1176 * protocol version 15 */
1177 send_uid_list(f);
1178
1179 /* send the io_error flag */
1180 write_int(f, lp_ignore_errors(module_id) ? 0 : io_error);
1181
1182 io_end_buffering();
1183 stats.flist_size = stats.total_written - start_write;
1184 stats.num_files = flist->count;
1185 if (write_batch)
1186 write_batch_flist_info(flist->count, flist->files);
1187 }
1188
1189 if (verbose > 2)
1190 rprintf(FINFO, "send_file_list done\n");
1191
1192 return flist;
1193}
1194
1195
1196struct file_list *recv_file_list(int f)
1197{
1198 struct file_list *flist;
1199 unsigned short flags;
1200 int64 start_read;
1201 extern int list_only;
1202
1203 if (show_filelist_p())
1204 start_filelist_progress("receiving file list");
1205
1206 start_read = stats.total_read;
1207
1208 flist = new(struct file_list);
1209 if (!flist)
1210 goto oom;
1211
1212 flist->count = 0;
1213 flist->malloced = 1000;
1214 flist->files = new_array(struct file_struct *, flist->malloced);
1215 if (!flist->files)
1216 goto oom;
1217
1218
1219 while ((flags = read_byte(f)) != 0) {
1220 int i = flist->count;
1221
1222 flist_expand(flist);
1223
1224 if (protocol_version >= 28 && (flags & XMIT_EXTENDED_FLAGS))
1225 flags |= read_byte(f) << 8;
1226 receive_file_entry(&flist->files[i], flags, f);
1227
1228 if (S_ISREG(flist->files[i]->mode))
1229 stats.total_size += flist->files[i]->length;
1230
1231 flist->count++;
1232
1233 maybe_emit_filelist_progress(flist);
1234
1235 if (verbose > 2) {
1236 rprintf(FINFO, "recv_file_name(%s)\n",
1237 f_name(flist->files[i]));
1238 }
1239 }
1240 receive_file_entry(NULL, 0, 0); /* Signal that we're done. */
1241
1242 if (verbose > 2)
1243 rprintf(FINFO, "received %d names\n", flist->count);
1244
1245 if (show_filelist_p())
1246 finish_filelist_progress(flist);
1247
1248 clean_flist(flist, relative_paths, 1);
1249
1250 if (f != -1) {
1251 /* Now send the uid/gid list. This was introduced in
1252 * protocol version 15 */
1253 recv_uid_list(f, flist);
1254
1255 if (!read_batch) {
1256 /* Recv the io_error flag */
1257 if (lp_ignore_errors(module_id) || ignore_errors)
1258 read_int(f);
1259 else
1260 io_error |= read_int(f);
1261 }
1262 }
1263
1264 if (list_only) {
1265 int i;
1266 for (i = 0; i < flist->count; i++)
1267 list_file_entry(flist->files[i]);
1268 }
1269
1270 if (verbose > 2)
1271 rprintf(FINFO, "recv_file_list done\n");
1272
1273 stats.flist_size = stats.total_read - start_read;
1274 stats.num_files = flist->count;
1275
1276 return flist;
1277
1278 oom:
1279 out_of_memory("recv_file_list");
1280 return NULL; /* not reached */
1281}
1282
1283
1284int file_compare(struct file_struct **file1, struct file_struct **file2)
1285{
1286 struct file_struct *f1 = *file1;
1287 struct file_struct *f2 = *file2;
1288
1289 if (!f1->basename && !f2->basename)
1290 return 0;
1291 if (!f1->basename)
1292 return -1;
1293 if (!f2->basename)
1294 return 1;
1295 if (f1->dirname == f2->dirname)
1296 return u_strcmp(f1->basename, f2->basename);
1297 return f_name_cmp(f1, f2);
1298}
1299
1300
1301int flist_find(struct file_list *flist, struct file_struct *f)
1302{
1303 int low = 0, high = flist->count - 1;
1304
1305 while (high >= 0 && !flist->files[high]->basename) high--;
1306
1307 if (high < 0)
1308 return -1;
1309
1310 while (low != high) {
1311 int mid = (low + high) / 2;
1312 int ret = file_compare(&flist->files[flist_up(flist, mid)],&f);
1313 if (ret == 0)
1314 return flist_up(flist, mid);
1315 if (ret > 0)
1316 high = mid;
1317 else
1318 low = mid + 1;
1319 }
1320
1321 if (file_compare(&flist->files[flist_up(flist, low)], &f) == 0)
1322 return flist_up(flist, low);
1323 return -1;
1324}
1325
1326
1327/*
1328 * Free up any resources a file_struct has allocated, and optionally free
1329 * it up as well.
1330 */
1331void free_file(struct file_struct *file, int free_the_struct)
1332{
1333 if (free_the_struct)
1334 free(file);
1335 else
1336 memset(file, 0, min_file_struct_len);
1337}
1338
1339
1340/*
1341 * allocate a new file list
1342 */
1343struct file_list *flist_new(void)
1344{
1345 struct file_list *flist;
1346
1347 flist = new(struct file_list);
1348 if (!flist)
1349 out_of_memory("send_file_list");
1350
1351 flist->count = 0;
1352 flist->malloced = 0;
1353 flist->files = NULL;
1354
1355 return flist;
1356}
1357
1358/*
1359 * free up all elements in a flist
1360 */
1361void flist_free(struct file_list *flist)
1362{
1363 int i;
1364 for (i = 1; i < flist->count; i++)
1365 free_file(flist->files[i], FREE_STRUCT);
1366 free(flist->files);
1367 free(flist);
1368}
1369
1370
1371/*
1372 * This routine ensures we don't have any duplicate names in our file list.
1373 * duplicate names can cause corruption because of the pipelining
1374 */
1375static void clean_flist(struct file_list *flist, int strip_root, int no_dups)
1376{
1377 int i, prev_i = 0;
1378
1379 if (!flist || flist->count == 0)
1380 return;
1381
1382 qsort(flist->files, flist->count,
1383 sizeof flist->files[0], (int (*)()) file_compare);
1384
1385 for (i = no_dups? 0 : flist->count; i < flist->count; i++) {
1386 if (flist->files[i]->basename) {
1387 prev_i = i;
1388 break;
1389 }
1390 }
1391 while (++i < flist->count) {
1392 if (!flist->files[i]->basename)
1393 continue;
1394 if (f_name_cmp(flist->files[i], flist->files[prev_i]) == 0) {
1395 if (verbose > 1 && !am_server) {
1396 rprintf(FINFO,
1397 "removing duplicate name %s from file list %d\n",
1398 f_name(flist->files[i]), i);
1399 }
1400 /* Make sure that if we unduplicate '.', that we don't
1401 * lose track of a user-specified starting point (or
1402 * else deletions will mysteriously fail with -R). */
1403 if (flist->files[i]->flags & FLAG_TOP_DIR)
1404 flist->files[prev_i]->flags |= FLAG_TOP_DIR;
1405 free_file(flist->files[i], CLEAR_STRUCT);
1406 } else
1407 prev_i = i;
1408 }
1409
1410 if (strip_root) {
1411 /* we need to strip off the root directory in the case
1412 of relative paths, but this must be done _after_
1413 the sorting phase */
1414 for (i = 0; i < flist->count; i++) {
1415 if (flist->files[i]->dirname &&
1416 flist->files[i]->dirname[0] == '/') {
1417 memmove(&flist->files[i]->dirname[0],
1418 &flist->files[i]->dirname[1],
1419 strlen(flist->files[i]->dirname));
1420 }
1421
1422 if (flist->files[i]->dirname &&
1423 !flist->files[i]->dirname[0]) {
1424 flist->files[i]->dirname = NULL;
1425 }
1426 }
1427 }
1428
1429 if (verbose <= 3)
1430 return;
1431
1432 for (i = 0; i < flist->count; i++) {
1433 rprintf(FINFO, "[%s] i=%d %s %s %s mode=0%o len=%.0f\n",
1434 who_am_i(), i,
1435 NS(flist->files[i]->basedir),
1436 NS(flist->files[i]->dirname),
1437 NS(flist->files[i]->basename),
1438 (int) flist->files[i]->mode,
1439 (double) flist->files[i]->length);
1440 }
1441}
1442
1443
1444enum fnc_state { fnc_DIR, fnc_SLASH, fnc_BASE };
1445
1446/* Compare the names of two file_struct entities, just like strcmp()
1447 * would do if it were operating on the joined strings. We assume
1448 * that there are no 0-length strings.
1449 */
1450int f_name_cmp(struct file_struct *f1, struct file_struct *f2)
1451{
1452 int dif;
1453 const uchar *c1, *c2;
1454 enum fnc_state state1, state2;
1455
1456 if (!f1 || !f1->basename) {
1457 if (!f2 || !f2->basename)
1458 return 0;
1459 return -1;
1460 }
1461 if (!f2 || !f2->basename)
1462 return 1;
1463
1464 if (!(c1 = (uchar*)f1->dirname)) {
1465 state1 = fnc_BASE;
1466 c1 = (uchar*)f1->basename;
1467 } else
1468 state1 = fnc_DIR;
1469 if (!(c2 = (uchar*)f2->dirname)) {
1470 state2 = fnc_BASE;
1471 c2 = (uchar*)f2->basename;
1472 } else
1473 state2 = fnc_DIR;
1474
1475 while (1) {
1476 if ((dif = (int)*c1 - (int)*c2) != 0)
1477 break;
1478 if (!*++c1) {
1479 switch (state1) {
1480 case fnc_DIR:
1481 state1 = fnc_SLASH;
1482 c1 = (uchar*)"/";
1483 break;
1484 case fnc_SLASH:
1485 state1 = fnc_BASE;
1486 c1 = (uchar*)f1->basename;
1487 break;
1488 case fnc_BASE:
1489 break;
1490 }
1491 }
1492 if (!*++c2) {
1493 switch (state2) {
1494 case fnc_DIR:
1495 state2 = fnc_SLASH;
1496 c2 = (uchar*)"/";
1497 break;
1498 case fnc_SLASH:
1499 state2 = fnc_BASE;
1500 c2 = (uchar*)f2->basename;
1501 break;
1502 case fnc_BASE:
1503 if (!*c1)
1504 return 0;
1505 break;
1506 }
1507 }
1508 }
1509
1510 return dif;
1511}
1512
1513
1514/* Return a copy of the full filename of a flist entry, using the indicated
1515 * buffer. No size-checking is done because we checked the size when creating
1516 * the file_struct entry.
1517 */
1518char *f_name_to(struct file_struct *f, char *fbuf)
1519{
1520 if (!f || !f->basename)
1521 return NULL;
1522
1523 if (f->dirname) {
1524 int len = strlen(f->dirname);
1525 memcpy(fbuf, f->dirname, len);
1526 fbuf[len] = '/';
1527 strcpy(fbuf + len + 1, f->basename);
1528 } else
1529 strcpy(fbuf, f->basename);
1530 return fbuf;
1531}
1532
1533
1534/* Like f_name_to(), but we rotate through 5 static buffers of our own.
1535 */
1536char *f_name(struct file_struct *f)
1537{
1538 static char names[5][MAXPATHLEN];
1539 static unsigned int n;
1540
1541 n = (n + 1) % (sizeof names / sizeof names[0]);
1542
1543 return f_name_to(f, names[n]);
1544}