added RSYNC_PROXY support from Stephen Rothwell. This allows access to
[rsync/rsync.git] / flist.c
... / ...
CommitLineData
1/*
2 Copyright (C) Andrew Tridgell 1996
3 Copyright (C) Paul Mackerras 1996
4
5 This program is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published by
7 the Free Software Foundation; either version 2 of the License, or
8 (at your option) any later version.
9
10 This program is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 GNU General Public License for more details.
14
15 You should have received a copy of the GNU General Public License
16 along with this program; if not, write to the Free Software
17 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18*/
19
20/* generate and receive file lists */
21
22#include "rsync.h"
23
24extern struct stats stats;
25
26extern int csum_length;
27
28extern int verbose;
29extern int am_server;
30extern int always_checksum;
31
32extern int cvs_exclude;
33
34extern int recurse;
35
36extern int one_file_system;
37extern int make_backups;
38extern int preserve_links;
39extern int preserve_hard_links;
40extern int preserve_perms;
41extern int preserve_devices;
42extern int preserve_uid;
43extern int preserve_gid;
44extern int preserve_times;
45extern int relative_paths;
46extern int copy_links;
47extern int copy_unsafe_links;
48extern int remote_version;
49extern int io_error;
50
51static char topsrcname[MAXPATHLEN];
52
53static struct exclude_struct **local_exclude_list;
54
55static void clean_flist(struct file_list *flist, int strip_root);
56
57
58static void list_file_entry(struct file_struct *f)
59{
60 char perms[11] = "----------";
61 char *perm_map = "rwxrwxrwx";
62 int i;
63
64 if (!f->basename)
65 /* this can happen if duplicate names were removed */
66 return;
67
68 for (i=0;i<9;i++) {
69 if (f->mode & (1<<i)) perms[9-i] = perm_map[8-i];
70 }
71 if (S_ISLNK(f->mode)) perms[0] = 'l';
72 if (S_ISDIR(f->mode)) perms[0] = 'd';
73 if (S_ISBLK(f->mode)) perms[0] = 'b';
74 if (S_ISCHR(f->mode)) perms[0] = 'c';
75 if (S_ISSOCK(f->mode)) perms[0] = 's';
76 if (S_ISFIFO(f->mode)) perms[0] = 'p';
77
78 if (preserve_links && S_ISLNK(f->mode)) {
79 rprintf(FINFO,"%s %11.0f %s %s -> %s\n",
80 perms,
81 (double)f->length, timestring(f->modtime),
82 f_name(f), f->link);
83 } else {
84 rprintf(FINFO,"%s %11.0f %s %s\n",
85 perms,
86 (double)f->length, timestring(f->modtime), f_name(f));
87 }
88}
89
90
91int readlink_stat(const char *Path, STRUCT_STAT *Buffer, char *Linkbuf)
92{
93#if SUPPORT_LINKS
94 if (copy_links) {
95 return do_stat(Path, Buffer);
96 }
97 if (do_lstat(Path, Buffer) == -1) {
98 return -1;
99 }
100 if (S_ISLNK(Buffer->st_mode)) {
101 int l;
102 if ((l = readlink(Path,Linkbuf,MAXPATHLEN-1)) == -1) {
103 return -1;
104 }
105 Linkbuf[l] = 0;
106 if (copy_unsafe_links && (topsrcname[0] != '\0') &&
107 unsafe_symlink(Linkbuf, topsrcname)) {
108 return do_stat(Path, Buffer);
109 }
110 }
111 return 0;
112#else
113 return do_stat(Path, Buffer);
114#endif
115}
116
117int link_stat(const char *Path, STRUCT_STAT *Buffer)
118{
119#if SUPPORT_LINKS
120 if (copy_links) {
121 return do_stat(Path, Buffer);
122 } else {
123 return do_lstat(Path, Buffer);
124 }
125#else
126 return do_stat(Path, Buffer);
127#endif
128}
129
130/*
131 This function is used to check if a file should be included/excluded
132 from the list of files based on its name and type etc
133 */
134static int match_file_name(char *fname,STRUCT_STAT *st)
135{
136 if (check_exclude(fname,local_exclude_list,st)) {
137 if (verbose > 2)
138 rprintf(FINFO,"excluding file %s\n",fname);
139 return 0;
140 }
141 return 1;
142}
143
144/* used by the one_file_system code */
145static dev_t filesystem_dev;
146
147static void set_filesystem(char *fname)
148{
149 STRUCT_STAT st;
150 if (link_stat(fname,&st) != 0) return;
151 filesystem_dev = st.st_dev;
152}
153
154
155static int to_wire_mode(mode_t mode)
156{
157 if (S_ISLNK(mode) && (S_IFLNK != 0120000)) {
158 return (mode & ~(_S_IFMT)) | 0120000;
159 }
160 return (int)mode;
161}
162
163static mode_t from_wire_mode(int mode)
164{
165 if ((mode & (_S_IFMT)) == 0120000 && (S_IFLNK != 0120000)) {
166 return (mode & ~(_S_IFMT)) | S_IFLNK;
167 }
168 return (mode_t)mode;
169}
170
171
172static void send_directory(int f,struct file_list *flist,char *dir);
173
174static char *flist_dir;
175
176
177static void send_file_entry(struct file_struct *file,int f,unsigned base_flags)
178{
179 unsigned char flags;
180 static time_t last_time;
181 static mode_t last_mode;
182 static dev_t last_rdev;
183 static uid_t last_uid;
184 static gid_t last_gid;
185 static char lastname[MAXPATHLEN];
186 char *fname;
187 int l1,l2;
188
189 if (f == -1) return;
190
191 if (!file) {
192 write_byte(f,0);
193 return;
194 }
195
196 fname = f_name(file);
197
198 flags = base_flags;
199
200 if (file->mode == last_mode) flags |= SAME_MODE;
201 if (file->rdev == last_rdev) flags |= SAME_RDEV;
202 if (file->uid == last_uid) flags |= SAME_UID;
203 if (file->gid == last_gid) flags |= SAME_GID;
204 if (file->modtime == last_time) flags |= SAME_TIME;
205
206 for (l1=0;lastname[l1] && (fname[l1] == lastname[l1]) && (l1 < 255);l1++) ;
207 l2 = strlen(fname) - l1;
208
209 if (l1 > 0) flags |= SAME_NAME;
210 if (l2 > 255) flags |= LONG_NAME;
211
212 /* we must make sure we don't send a zero flags byte or the other
213 end will terminate the flist transfer */
214 if (flags == 0 && !S_ISDIR(file->mode)) flags |= FLAG_DELETE;
215 if (flags == 0) flags |= LONG_NAME;
216
217 write_byte(f,flags);
218 if (flags & SAME_NAME)
219 write_byte(f,l1);
220 if (flags & LONG_NAME)
221 write_int(f,l2);
222 else
223 write_byte(f,l2);
224 write_buf(f,fname+l1,l2);
225
226 write_longint(f,file->length);
227 if (!(flags & SAME_TIME))
228 write_int(f,(int)file->modtime);
229 if (!(flags & SAME_MODE))
230 write_int(f,to_wire_mode(file->mode));
231 if (preserve_uid && !(flags & SAME_UID)) {
232 add_uid(file->uid);
233 write_int(f,(int)file->uid);
234 }
235 if (preserve_gid && !(flags & SAME_GID)) {
236 add_gid(file->gid);
237 write_int(f,(int)file->gid);
238 }
239 if (preserve_devices && IS_DEVICE(file->mode) && !(flags & SAME_RDEV))
240 write_int(f,(int)file->rdev);
241
242#if SUPPORT_LINKS
243 if (preserve_links && S_ISLNK(file->mode)) {
244 write_int(f,strlen(file->link));
245 write_buf(f,file->link,strlen(file->link));
246 }
247#endif
248
249#if SUPPORT_HARD_LINKS
250 if (preserve_hard_links && S_ISREG(file->mode)) {
251 write_int(f,(int)file->dev);
252 write_int(f,(int)file->inode);
253 }
254#endif
255
256 if (always_checksum) {
257 write_buf(f,file->sum,csum_length);
258 }
259
260 last_mode = file->mode;
261 last_rdev = file->rdev;
262 last_uid = file->uid;
263 last_gid = file->gid;
264 last_time = file->modtime;
265
266 strlcpy(lastname,fname,MAXPATHLEN);
267 lastname[MAXPATHLEN-1] = 0;
268}
269
270
271
272static void receive_file_entry(struct file_struct **fptr,
273 unsigned flags,int f)
274{
275 static time_t last_time;
276 static mode_t last_mode;
277 static dev_t last_rdev;
278 static uid_t last_uid;
279 static gid_t last_gid;
280 static char lastname[MAXPATHLEN];
281 char thisname[MAXPATHLEN];
282 int l1=0,l2=0;
283 char *p;
284 struct file_struct *file;
285
286 if (flags & SAME_NAME)
287 l1 = read_byte(f);
288
289 if (flags & LONG_NAME)
290 l2 = read_int(f);
291 else
292 l2 = read_byte(f);
293
294 file = (struct file_struct *)malloc(sizeof(*file));
295 if (!file) out_of_memory("receive_file_entry");
296 memset((char *)file, 0, sizeof(*file));
297 (*fptr) = file;
298
299 if (l2 >= MAXPATHLEN-l1) overflow("receive_file_entry");
300
301 strlcpy(thisname,lastname,l1+1);
302 read_sbuf(f,&thisname[l1],l2);
303 thisname[l1+l2] = 0;
304
305 strlcpy(lastname,thisname,MAXPATHLEN);
306 lastname[MAXPATHLEN-1] = 0;
307
308 clean_fname(thisname);
309
310 if ((p = strrchr(thisname,'/'))) {
311 static char *lastdir;
312 *p = 0;
313 if (lastdir && strcmp(thisname, lastdir)==0) {
314 file->dirname = lastdir;
315 } else {
316 file->dirname = strdup(thisname);
317 lastdir = file->dirname;
318 }
319 file->basename = strdup(p+1);
320 } else {
321 file->dirname = NULL;
322 file->basename = strdup(thisname);
323 }
324
325 if (!file->basename) out_of_memory("receive_file_entry 1");
326
327
328 file->flags = flags;
329 file->length = read_longint(f);
330 file->modtime = (flags & SAME_TIME) ? last_time : (time_t)read_int(f);
331 file->mode = (flags & SAME_MODE) ? last_mode : from_wire_mode(read_int(f));
332 if (preserve_uid)
333 file->uid = (flags & SAME_UID) ? last_uid : (uid_t)read_int(f);
334 if (preserve_gid)
335 file->gid = (flags & SAME_GID) ? last_gid : (gid_t)read_int(f);
336 if (preserve_devices && IS_DEVICE(file->mode))
337 file->rdev = (flags & SAME_RDEV) ? last_rdev : (dev_t)read_int(f);
338
339 if (preserve_links && S_ISLNK(file->mode)) {
340 int l = read_int(f);
341 file->link = (char *)malloc(l+1);
342 if (!file->link) out_of_memory("receive_file_entry 2");
343 read_sbuf(f,file->link,l);
344 }
345
346#if SUPPORT_HARD_LINKS
347 if (preserve_hard_links && S_ISREG(file->mode)) {
348 file->dev = read_int(f);
349 file->inode = read_int(f);
350 }
351#endif
352
353 if (always_checksum) {
354 file->sum = (char *)malloc(MD4_SUM_LENGTH);
355 if (!file->sum) out_of_memory("md4 sum");
356 read_buf(f,file->sum,csum_length);
357 }
358
359 last_mode = file->mode;
360 last_rdev = file->rdev;
361 last_uid = file->uid;
362 last_gid = file->gid;
363 last_time = file->modtime;
364
365 if (!preserve_perms) {
366 extern int orig_umask;
367 /* set an appropriate set of permissions based on original
368 permissions and umask. This emulates what GNU cp does */
369 file->mode &= ~orig_umask;
370 }
371}
372
373
374/* determine if a file in a different filesstem should be skipped
375 when one_file_system is set. We bascally only want to include
376 the mount points - but they can be hard to find! */
377static int skip_filesystem(char *fname, STRUCT_STAT *st)
378{
379 STRUCT_STAT st2;
380 char *p = strrchr(fname, '/');
381
382 /* skip all but directories */
383 if (!S_ISDIR(st->st_mode)) return 1;
384
385 /* if its not a subdirectory then allow */
386 if (!p) return 0;
387
388 *p = 0;
389 if (link_stat(fname, &st2)) {
390 *p = '/';
391 return 0;
392 }
393 *p = '/';
394
395 return (st2.st_dev != filesystem_dev);
396}
397
398static struct file_struct *make_file(int f, char *fname)
399{
400 struct file_struct *file;
401 STRUCT_STAT st;
402 char sum[SUM_LENGTH];
403 char *p;
404 char cleaned_name[MAXPATHLEN];
405 char linkbuf[MAXPATHLEN];
406 extern int delete_excluded;
407
408 strlcpy(cleaned_name, fname, MAXPATHLEN);
409 cleaned_name[MAXPATHLEN-1] = 0;
410 clean_fname(cleaned_name);
411 fname = cleaned_name;
412
413 memset(sum,0,SUM_LENGTH);
414
415 if (readlink_stat(fname,&st,linkbuf) != 0) {
416 io_error = 1;
417 rprintf(FERROR,"%s: %s\n",
418 fname,strerror(errno));
419 return NULL;
420 }
421
422 if (S_ISDIR(st.st_mode) && !recurse) {
423 rprintf(FINFO,"skipping directory %s\n",fname);
424 return NULL;
425 }
426
427 if (one_file_system && st.st_dev != filesystem_dev) {
428 if (skip_filesystem(fname, &st))
429 return NULL;
430 }
431
432 /* f is set to -1 when calculating deletion file list */
433 if (((f != -1) || !delete_excluded) && !match_file_name(fname,&st))
434 return NULL;
435
436 if (verbose > 2)
437 rprintf(FINFO,"make_file(%d,%s)\n",f,fname);
438
439 file = (struct file_struct *)malloc(sizeof(*file));
440 if (!file) out_of_memory("make_file");
441 memset((char *)file,0,sizeof(*file));
442
443 if ((p = strrchr(fname,'/'))) {
444 static char *lastdir;
445 *p = 0;
446 if (lastdir && strcmp(fname, lastdir)==0) {
447 file->dirname = lastdir;
448 } else {
449 file->dirname = strdup(fname);
450 lastdir = file->dirname;
451 }
452 file->basename = strdup(p+1);
453 *p = '/';
454 } else {
455 file->dirname = NULL;
456 file->basename = strdup(fname);
457 }
458
459 file->modtime = st.st_mtime;
460 file->length = st.st_size;
461 file->mode = st.st_mode;
462 file->uid = st.st_uid;
463 file->gid = st.st_gid;
464 file->dev = st.st_dev;
465 file->inode = st.st_ino;
466#ifdef HAVE_ST_RDEV
467 file->rdev = st.st_rdev;
468#endif
469
470#if SUPPORT_LINKS
471 if (S_ISLNK(st.st_mode)) {
472 file->link = strdup(linkbuf);
473 }
474#endif
475
476 if (always_checksum) {
477 file->sum = (char *)malloc(MD4_SUM_LENGTH);
478 if (!file->sum) out_of_memory("md4 sum");
479 /* drat. we have to provide a null checksum for non-regular
480 files in order to be compatible with earlier versions
481 of rsync */
482 if (S_ISREG(st.st_mode)) {
483 file_checksum(fname,file->sum,st.st_size);
484 } else {
485 memset(file->sum, 0, MD4_SUM_LENGTH);
486 }
487 }
488
489 if (flist_dir) {
490 static char *lastdir;
491 if (lastdir && strcmp(lastdir, flist_dir)==0) {
492 file->basedir = lastdir;
493 } else {
494 file->basedir = strdup(flist_dir);
495 lastdir = file->basedir;
496 }
497 } else {
498 file->basedir = NULL;
499 }
500
501 if (!S_ISDIR(st.st_mode))
502 stats.total_size += st.st_size;
503
504 return file;
505}
506
507
508
509void send_file_name(int f,struct file_list *flist,char *fname,
510 int recursive, unsigned base_flags)
511{
512 struct file_struct *file;
513
514 file = make_file(f,fname);
515
516 if (!file) return;
517
518 if (flist->count >= flist->malloced) {
519 if (flist->malloced < 1000)
520 flist->malloced += 1000;
521 else
522 flist->malloced *= 2;
523 flist->files = (struct file_struct **)realloc(flist->files,
524 sizeof(flist->files[0])*
525 flist->malloced);
526 if (!flist->files)
527 out_of_memory("send_file_name");
528 }
529
530 if (strcmp(file->basename,"")) {
531 flist->files[flist->count++] = file;
532 send_file_entry(file,f,base_flags);
533 }
534
535 if (S_ISDIR(file->mode) && recursive) {
536 struct exclude_struct **last_exclude_list = local_exclude_list;
537 send_directory(f,flist,f_name(file));
538 local_exclude_list = last_exclude_list;
539 return;
540 }
541}
542
543
544
545static void send_directory(int f,struct file_list *flist,char *dir)
546{
547 DIR *d;
548 struct dirent *di;
549 char fname[MAXPATHLEN];
550 int l;
551 char *p;
552
553 d = opendir(dir);
554 if (!d) {
555 io_error = 1;
556 rprintf(FERROR,"opendir(%s): %s\n",
557 dir,strerror(errno));
558 return;
559 }
560
561 strlcpy(fname,dir,MAXPATHLEN);
562 l = strlen(fname);
563 if (fname[l-1] != '/') {
564 if (l == MAXPATHLEN-1) {
565 io_error = 1;
566 rprintf(FERROR,"skipping long-named directory %s\n",fname);
567 closedir(d);
568 return;
569 }
570 strlcat(fname,"/", MAXPATHLEN);
571 l++;
572 }
573 p = fname + strlen(fname);
574
575 local_exclude_list = NULL;
576
577 if (cvs_exclude) {
578 if (strlen(fname) + strlen(".cvsignore") <= MAXPATHLEN-1) {
579 strcpy(p,".cvsignore");
580 local_exclude_list = make_exclude_list(fname,NULL,0,0);
581 } else {
582 io_error = 1;
583 rprintf(FINFO,"cannot cvs-exclude in long-named directory %s\n",fname);
584 }
585 }
586
587 for (di=readdir(d); di; di=readdir(d)) {
588 char *dname = d_name(di);
589 if (strcmp(dname,".")==0 ||
590 strcmp(dname,"..")==0)
591 continue;
592 strlcpy(p,dname,MAXPATHLEN-l);
593 send_file_name(f,flist,fname,recurse,0);
594 }
595
596 if (local_exclude_list) {
597 add_exclude_list("!", &local_exclude_list, 0);
598 }
599
600 closedir(d);
601}
602
603
604
605struct file_list *send_file_list(int f,int argc,char *argv[])
606{
607 int i,l;
608 STRUCT_STAT st;
609 char *p,*dir,*olddir;
610 char lastpath[MAXPATHLEN]="";
611 struct file_list *flist;
612 int64 start_write;
613
614 if (verbose && recurse && !am_server && f != -1) {
615 rprintf(FINFO,"building file list ... ");
616 rflush(FINFO);
617 }
618
619 start_write = stats.total_written;
620
621 flist = (struct file_list *)malloc(sizeof(flist[0]));
622 if (!flist) out_of_memory("send_file_list");
623
624 flist->count=0;
625 flist->malloced = 1000;
626 flist->files = (struct file_struct **)malloc(sizeof(flist->files[0])*
627 flist->malloced);
628 if (!flist->files) out_of_memory("send_file_list");
629
630 if (f != -1) {
631 io_start_buffering(f);
632 }
633
634 for (i=0;i<argc;i++) {
635 char *fname = topsrcname;
636
637 strlcpy(fname,argv[i],MAXPATHLEN);
638
639 l = strlen(fname);
640 if (l != 1 && fname[l-1] == '/') {
641 if ((l == 2) && (fname[0] == '.')) {
642 /* Turn ./ into just . rather than ./.
643 This was put in to avoid a problem with
644 rsync -aR --delete from ./
645 The send_file_name() below of ./ was
646 mysteriously preventing deletes */
647 fname[1] = 0;
648 } else {
649 strlcat(fname,".",MAXPATHLEN);
650 }
651 }
652
653 if (link_stat(fname,&st) != 0) {
654 io_error=1;
655 rprintf(FERROR,"%s : %s\n",fname,strerror(errno));
656 continue;
657 }
658
659 if (S_ISDIR(st.st_mode) && !recurse) {
660 rprintf(FINFO,"skipping directory %s\n",fname);
661 continue;
662 }
663
664 dir = NULL;
665 olddir = NULL;
666
667 if (!relative_paths) {
668 p = strrchr(fname,'/');
669 if (p) {
670 *p = 0;
671 if (p == fname)
672 dir = "/";
673 else
674 dir = fname;
675 fname = p+1;
676 }
677 } else if (f != -1 && (p=strrchr(fname,'/'))) {
678 /* this ensures we send the intermediate directories,
679 thus getting their permissions right */
680 *p = 0;
681 if (strcmp(lastpath,fname)) {
682 strlcpy(lastpath, fname, sizeof(lastpath));
683 *p = '/';
684 for (p=fname+1; (p=strchr(p,'/')); p++) {
685 int copy_links_saved = copy_links;
686 int recurse_saved = recurse;
687 *p = 0;
688 copy_links = copy_unsafe_links;
689 /* set recurse to 1 to prevent make_file
690 from ignoring directory, but still
691 turn off the recursive parameter to
692 send_file_name */
693 recurse = 1;
694 send_file_name(f, flist, fname, 0, 0);
695 copy_links = copy_links_saved;
696 recurse = recurse_saved;
697 *p = '/';
698 }
699 } else {
700 *p = '/';
701 }
702 }
703
704 if (!*fname)
705 fname = ".";
706
707 if (dir && *dir) {
708 olddir = push_dir(dir, 1);
709
710 if (!olddir) {
711 io_error=1;
712 rprintf(FERROR,"push_dir %s : %s\n",
713 dir,strerror(errno));
714 continue;
715 }
716
717 flist_dir = dir;
718 }
719
720 if (one_file_system)
721 set_filesystem(fname);
722
723 if (!recurse || !send_included_file_names(f,flist))
724 send_file_name(f,flist,fname,recurse,FLAG_DELETE);
725
726 if (olddir != NULL) {
727 flist_dir = NULL;
728 if (pop_dir(olddir) != 0) {
729 rprintf(FERROR,"pop_dir %s : %s\n",
730 dir,strerror(errno));
731 exit_cleanup(RERR_FILESELECT);
732 }
733 }
734 }
735
736 topsrcname[0] = '\0';
737
738 if (f != -1) {
739 send_file_entry(NULL,f,0);
740 }
741
742 if (verbose && recurse && !am_server && f != -1)
743 rprintf(FINFO,"done\n");
744
745 clean_flist(flist, 0);
746
747 /* now send the uid/gid list. This was introduced in protocol
748 version 15 */
749 if (f != -1 && remote_version >= 15) {
750 send_uid_list(f);
751 }
752
753 /* if protocol version is >= 17 then send the io_error flag */
754 if (f != -1 && remote_version >= 17) {
755 write_int(f, io_error);
756 }
757
758 if (f != -1) {
759 io_end_buffering(f);
760 stats.flist_size = stats.total_written - start_write;
761 stats.num_files = flist->count;
762 }
763
764 if (verbose > 2)
765 rprintf(FINFO,"send_file_list done\n");
766
767 return flist;
768}
769
770
771struct file_list *recv_file_list(int f)
772{
773 struct file_list *flist;
774 unsigned char flags;
775 int64 start_read;
776 extern int list_only;
777
778 if (verbose && recurse && !am_server) {
779 rprintf(FINFO,"receiving file list ... ");
780 rflush(FINFO);
781 }
782
783 start_read = stats.total_read;
784
785 flist = (struct file_list *)malloc(sizeof(flist[0]));
786 if (!flist)
787 goto oom;
788
789 flist->count=0;
790 flist->malloced=1000;
791 flist->files = (struct file_struct **)malloc(sizeof(flist->files[0])*
792 flist->malloced);
793 if (!flist->files)
794 goto oom;
795
796
797 for (flags=read_byte(f); flags; flags=read_byte(f)) {
798 int i = flist->count;
799
800 if (i >= flist->malloced) {
801 if (flist->malloced < 1000)
802 flist->malloced += 1000;
803 else
804 flist->malloced *= 2;
805 flist->files =(struct file_struct **)realloc(flist->files,
806 sizeof(flist->files[0])*
807 flist->malloced);
808 if (!flist->files)
809 goto oom;
810 }
811
812 receive_file_entry(&flist->files[i],flags,f);
813
814 if (S_ISREG(flist->files[i]->mode))
815 stats.total_size += flist->files[i]->length;
816
817 flist->count++;
818
819 if (verbose > 2)
820 rprintf(FINFO,"recv_file_name(%s)\n",f_name(flist->files[i]));
821 }
822
823
824 if (verbose > 2)
825 rprintf(FINFO,"received %d names\n",flist->count);
826
827 clean_flist(flist, relative_paths);
828
829 if (verbose && recurse && !am_server) {
830 rprintf(FINFO,"done\n");
831 }
832
833 /* now recv the uid/gid list. This was introduced in protocol version 15 */
834 if (f != -1 && remote_version >= 15) {
835 recv_uid_list(f, flist);
836 }
837
838 /* if protocol version is >= 17 then recv the io_error flag */
839 if (f != -1 && remote_version >= 17) {
840 io_error |= read_int(f);
841 }
842
843 if (list_only) {
844 int i;
845 for (i=0;i<flist->count;i++) {
846 list_file_entry(flist->files[i]);
847 }
848 }
849
850
851 if (verbose > 2)
852 rprintf(FINFO,"recv_file_list done\n");
853
854 stats.flist_size = stats.total_read - start_read;
855 stats.num_files = flist->count;
856
857 return flist;
858
859oom:
860 out_of_memory("recv_file_list");
861 return NULL; /* not reached */
862}
863
864
865int file_compare(struct file_struct **f1,struct file_struct **f2)
866{
867 if (!(*f1)->basename && !(*f2)->basename) return 0;
868 if (!(*f1)->basename) return -1;
869 if (!(*f2)->basename) return 1;
870 if ((*f1)->dirname == (*f2)->dirname)
871 return u_strcmp((*f1)->basename, (*f2)->basename);
872 return u_strcmp(f_name(*f1),f_name(*f2));
873}
874
875
876int flist_find(struct file_list *flist,struct file_struct *f)
877{
878 int low=0,high=flist->count-1;
879
880 if (flist->count <= 0) return -1;
881
882 while (low != high) {
883 int mid = (low+high)/2;
884 int ret = file_compare(&flist->files[flist_up(flist, mid)],&f);
885 if (ret == 0) return flist_up(flist, mid);
886 if (ret > 0) {
887 high=mid;
888 } else {
889 low=mid+1;
890 }
891 }
892
893 if (file_compare(&flist->files[flist_up(flist,low)],&f) == 0)
894 return flist_up(flist,low);
895 return -1;
896}
897
898
899/*
900 * free up one file
901 */
902static void free_file(struct file_struct *file)
903{
904 if (!file) return;
905 if (file->basename) free(file->basename);
906 if (file->link) free(file->link);
907 if (file->sum) free(file->sum);
908 memset((char *)file, 0, sizeof(*file));
909}
910
911
912/*
913 * free up all elements in a flist
914 */
915void flist_free(struct file_list *flist)
916{
917 int i;
918 for (i=1;i<flist->count;i++) {
919 free_file(flist->files[i]);
920 free(flist->files[i]);
921 }
922 memset((char *)flist->files, 0, sizeof(flist->files[0])*flist->count);
923 free(flist->files);
924 memset((char *)flist, 0, sizeof(*flist));
925 free(flist);
926}
927
928
929/*
930 * This routine ensures we don't have any duplicate names in our file list.
931 * duplicate names can cause corruption because of the pipelining
932 */
933static void clean_flist(struct file_list *flist, int strip_root)
934{
935 int i;
936
937 if (!flist || flist->count == 0)
938 return;
939
940 qsort(flist->files,flist->count,
941 sizeof(flist->files[0]),
942 (int (*)())file_compare);
943
944 for (i=1;i<flist->count;i++) {
945 if (flist->files[i]->basename &&
946 flist->files[i-1]->basename &&
947 strcmp(f_name(flist->files[i]),
948 f_name(flist->files[i-1])) == 0) {
949 if (verbose > 1 && !am_server)
950 rprintf(FINFO,"removing duplicate name %s from file list %d\n",
951 f_name(flist->files[i-1]),i-1);
952 free_file(flist->files[i]);
953 }
954 }
955
956 if (strip_root) {
957 /* we need to strip off the root directory in the case
958 of relative paths, but this must be done _after_
959 the sorting phase */
960 for (i=0;i<flist->count;i++) {
961 if (flist->files[i]->dirname &&
962 flist->files[i]->dirname[0] == '/') {
963 memmove(&flist->files[i]->dirname[0],
964 &flist->files[i]->dirname[1],
965 strlen(flist->files[i]->dirname));
966 }
967
968 if (flist->files[i]->dirname &&
969 !flist->files[i]->dirname[0]) {
970 flist->files[i]->dirname = NULL;
971 }
972 }
973 }
974
975
976 if (verbose <= 3) return;
977
978 for (i=0;i<flist->count;i++) {
979 rprintf(FINFO,"[%d] i=%d %s %s mode=0%o len=%d\n",
980 getpid(), i,
981 NS(flist->files[i]->dirname),
982 NS(flist->files[i]->basename),
983 flist->files[i]->mode,
984 (int)flist->files[i]->length);
985 }
986}
987
988
989/*
990 * return the full filename of a flist entry
991 */
992char *f_name(struct file_struct *f)
993{
994 static char names[10][MAXPATHLEN];
995 static int n;
996 char *p = names[n];
997
998 if (!f || !f->basename) return NULL;
999
1000 n = (n+1)%10;
1001
1002 if (f->dirname) {
1003 strlcpy(p, f->dirname, MAXPATHLEN);
1004 strlcat(p, "/", MAXPATHLEN);
1005 strlcat(p, f->basename, MAXPATHLEN);
1006 } else {
1007 strlcpy(p, f->basename, MAXPATHLEN);
1008 }
1009
1010 return p;
1011}
1012