Changed --stats implementation to work without -v in only these two
[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(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
407 strlcpy(cleaned_name, fname, MAXPATHLEN);
408 cleaned_name[MAXPATHLEN-1] = 0;
409 clean_fname(cleaned_name);
410 fname = cleaned_name;
411
412 memset(sum,0,SUM_LENGTH);
413
414 if (readlink_stat(fname,&st,linkbuf) != 0) {
415 io_error = 1;
416 rprintf(FERROR,"%s: %s\n",
417 fname,strerror(errno));
418 return NULL;
419 }
420
421 if (S_ISDIR(st.st_mode) && !recurse) {
422 rprintf(FINFO,"skipping directory %s\n",fname);
423 return NULL;
424 }
425
426 if (one_file_system && st.st_dev != filesystem_dev) {
427 if (skip_filesystem(fname, &st))
428 return NULL;
429 }
430
431 if (!match_file_name(fname,&st))
432 return NULL;
433
434 if (verbose > 2)
435 rprintf(FINFO,"make_file(%s)\n",fname);
436
437 file = (struct file_struct *)malloc(sizeof(*file));
438 if (!file) out_of_memory("make_file");
439 memset((char *)file,0,sizeof(*file));
440
441 if ((p = strrchr(fname,'/'))) {
442 static char *lastdir;
443 *p = 0;
444 if (lastdir && strcmp(fname, lastdir)==0) {
445 file->dirname = lastdir;
446 } else {
447 file->dirname = strdup(fname);
448 lastdir = file->dirname;
449 }
450 file->basename = strdup(p+1);
451 *p = '/';
452 } else {
453 file->dirname = NULL;
454 file->basename = strdup(fname);
455 }
456
457 file->modtime = st.st_mtime;
458 file->length = st.st_size;
459 file->mode = st.st_mode;
460 file->uid = st.st_uid;
461 file->gid = st.st_gid;
462 file->dev = st.st_dev;
463 file->inode = st.st_ino;
464#ifdef HAVE_ST_RDEV
465 file->rdev = st.st_rdev;
466#endif
467
468#if SUPPORT_LINKS
469 if (S_ISLNK(st.st_mode)) {
470 file->link = strdup(linkbuf);
471 }
472#endif
473
474 if (always_checksum) {
475 file->sum = (char *)malloc(MD4_SUM_LENGTH);
476 if (!file->sum) out_of_memory("md4 sum");
477 /* drat. we have to provide a null checksum for non-regular
478 files in order to be compatible with earlier versions
479 of rsync */
480 if (S_ISREG(st.st_mode)) {
481 file_checksum(fname,file->sum,st.st_size);
482 } else {
483 memset(file->sum, 0, MD4_SUM_LENGTH);
484 }
485 }
486
487 if (flist_dir) {
488 static char *lastdir;
489 if (lastdir && strcmp(lastdir, flist_dir)==0) {
490 file->basedir = lastdir;
491 } else {
492 file->basedir = strdup(flist_dir);
493 lastdir = file->basedir;
494 }
495 } else {
496 file->basedir = NULL;
497 }
498
499 if (!S_ISDIR(st.st_mode))
500 stats.total_size += st.st_size;
501
502 return file;
503}
504
505
506
507void send_file_name(int f,struct file_list *flist,char *fname,
508 int recursive, unsigned base_flags)
509{
510 struct file_struct *file;
511
512 file = make_file(fname);
513
514 if (!file) return;
515
516 if (flist->count >= flist->malloced) {
517 if (flist->malloced < 1000)
518 flist->malloced += 1000;
519 else
520 flist->malloced *= 2;
521 flist->files = (struct file_struct **)realloc(flist->files,
522 sizeof(flist->files[0])*
523 flist->malloced);
524 if (!flist->files)
525 out_of_memory("send_file_name");
526 }
527
528 if (strcmp(file->basename,"")) {
529 flist->files[flist->count++] = file;
530 send_file_entry(file,f,base_flags);
531 }
532
533 if (S_ISDIR(file->mode) && recursive) {
534 struct exclude_struct **last_exclude_list = local_exclude_list;
535 send_directory(f,flist,f_name(file));
536 local_exclude_list = last_exclude_list;
537 return;
538 }
539}
540
541
542
543static void send_directory(int f,struct file_list *flist,char *dir)
544{
545 DIR *d;
546 struct dirent *di;
547 char fname[MAXPATHLEN];
548 int l;
549 char *p;
550
551 d = opendir(dir);
552 if (!d) {
553 io_error = 1;
554 rprintf(FERROR,"opendir(%s): %s\n",
555 dir,strerror(errno));
556 return;
557 }
558
559 strlcpy(fname,dir,MAXPATHLEN);
560 l = strlen(fname);
561 if (fname[l-1] != '/') {
562 if (l == MAXPATHLEN-1) {
563 io_error = 1;
564 rprintf(FERROR,"skipping long-named directory %s\n",fname);
565 closedir(d);
566 return;
567 }
568 strlcat(fname,"/", MAXPATHLEN);
569 l++;
570 }
571 p = fname + strlen(fname);
572
573 local_exclude_list = NULL;
574
575 if (cvs_exclude) {
576 if (strlen(fname) + strlen(".cvsignore") <= MAXPATHLEN-1) {
577 strcpy(p,".cvsignore");
578 local_exclude_list = make_exclude_list(fname,NULL,0,0);
579 } else {
580 io_error = 1;
581 rprintf(FINFO,"cannot cvs-exclude in long-named directory %s\n",fname);
582 }
583 }
584
585 for (di=readdir(d); di; di=readdir(d)) {
586 char *dname = d_name(di);
587 if (strcmp(dname,".")==0 ||
588 strcmp(dname,"..")==0)
589 continue;
590 strlcpy(p,dname,MAXPATHLEN-l);
591 send_file_name(f,flist,fname,recurse,0);
592 }
593
594 if (local_exclude_list) {
595 add_exclude_list("!", &local_exclude_list, 0);
596 }
597
598 closedir(d);
599}
600
601
602
603struct file_list *send_file_list(int f,int argc,char *argv[])
604{
605 int i,l;
606 STRUCT_STAT st;
607 char *p,*dir,*olddir;
608 char lastpath[MAXPATHLEN]="";
609 struct file_list *flist;
610 int64 start_write;
611
612 if (verbose && recurse && !am_server && f != -1) {
613 rprintf(FINFO,"building file list ... ");
614 rflush(FINFO);
615 }
616
617 start_write = stats.total_written;
618
619 flist = (struct file_list *)malloc(sizeof(flist[0]));
620 if (!flist) out_of_memory("send_file_list");
621
622 flist->count=0;
623 flist->malloced = 1000;
624 flist->files = (struct file_struct **)malloc(sizeof(flist->files[0])*
625 flist->malloced);
626 if (!flist->files) out_of_memory("send_file_list");
627
628 if (f != -1) {
629 io_start_buffering(f);
630 }
631
632 for (i=0;i<argc;i++) {
633 char *fname = topsrcname;
634
635 strlcpy(fname,argv[i],MAXPATHLEN);
636
637 l = strlen(fname);
638 if (l != 1 && fname[l-1] == '/') {
639 strlcat(fname,".",MAXPATHLEN);
640 }
641
642 if (link_stat(fname,&st) != 0) {
643 io_error=1;
644 rprintf(FERROR,"%s : %s\n",fname,strerror(errno));
645 continue;
646 }
647
648 if (S_ISDIR(st.st_mode) && !recurse) {
649 rprintf(FINFO,"skipping directory %s\n",fname);
650 continue;
651 }
652
653 dir = NULL;
654 olddir = NULL;
655
656 if (!relative_paths) {
657 p = strrchr(fname,'/');
658 if (p) {
659 *p = 0;
660 if (p == fname)
661 dir = "/";
662 else
663 dir = fname;
664 fname = p+1;
665 }
666 } else if (f != -1 && (p=strrchr(fname,'/'))) {
667 /* this ensures we send the intermediate directories,
668 thus getting their permissions right */
669 *p = 0;
670 if (strcmp(lastpath,fname)) {
671 strlcpy(lastpath, fname, sizeof(lastpath));
672 *p = '/';
673 for (p=fname+1; (p=strchr(p,'/')); p++) {
674 int copy_links_saved = copy_links;
675 *p = 0;
676 copy_links = copy_unsafe_links;
677 send_file_name(f, flist, fname, 0, 0);
678 copy_links = copy_links_saved;
679 *p = '/';
680 }
681 } else {
682 *p = '/';
683 }
684 }
685
686 if (!*fname)
687 fname = ".";
688
689 if (dir && *dir) {
690 olddir = push_dir(dir, 1);
691
692 if (!olddir) {
693 io_error=1;
694 rprintf(FERROR,"push_dir %s : %s\n",
695 dir,strerror(errno));
696 continue;
697 }
698
699 flist_dir = dir;
700 }
701
702 if (one_file_system)
703 set_filesystem(fname);
704
705 if (!recurse || !send_included_file_names(f,flist))
706 send_file_name(f,flist,fname,recurse,FLAG_DELETE);
707
708 if (olddir != NULL) {
709 flist_dir = NULL;
710 if (pop_dir(olddir) != 0) {
711 rprintf(FERROR,"pop_dir %s : %s\n",
712 dir,strerror(errno));
713 exit_cleanup(RERR_FILESELECT);
714 }
715 }
716 }
717
718 topsrcname[0] = '\0';
719
720 if (f != -1) {
721 send_file_entry(NULL,f,0);
722 }
723
724 if (verbose && recurse && !am_server && f != -1)
725 rprintf(FINFO,"done\n");
726
727 clean_flist(flist, 0);
728
729 /* now send the uid/gid list. This was introduced in protocol
730 version 15 */
731 if (f != -1 && remote_version >= 15) {
732 send_uid_list(f);
733 }
734
735 /* if protocol version is >= 17 then send the io_error flag */
736 if (f != -1 && remote_version >= 17) {
737 write_int(f, io_error);
738 }
739
740 if (f != -1) {
741 io_end_buffering(f);
742 stats.flist_size = stats.total_written - start_write;
743 stats.num_files = flist->count;
744 }
745
746 if (verbose > 2)
747 rprintf(FINFO,"send_file_list done\n");
748
749 return flist;
750}
751
752
753struct file_list *recv_file_list(int f)
754{
755 struct file_list *flist;
756 unsigned char flags;
757 int64 start_read;
758 extern int list_only;
759
760 if (verbose && recurse && !am_server) {
761 rprintf(FINFO,"receiving file list ... ");
762 rflush(FINFO);
763 }
764
765 start_read = stats.total_read;
766
767 flist = (struct file_list *)malloc(sizeof(flist[0]));
768 if (!flist)
769 goto oom;
770
771 flist->count=0;
772 flist->malloced=1000;
773 flist->files = (struct file_struct **)malloc(sizeof(flist->files[0])*
774 flist->malloced);
775 if (!flist->files)
776 goto oom;
777
778
779 for (flags=read_byte(f); flags; flags=read_byte(f)) {
780 int i = flist->count;
781
782 if (i >= flist->malloced) {
783 if (flist->malloced < 1000)
784 flist->malloced += 1000;
785 else
786 flist->malloced *= 2;
787 flist->files =(struct file_struct **)realloc(flist->files,
788 sizeof(flist->files[0])*
789 flist->malloced);
790 if (!flist->files)
791 goto oom;
792 }
793
794 receive_file_entry(&flist->files[i],flags,f);
795
796 if (S_ISREG(flist->files[i]->mode))
797 stats.total_size += flist->files[i]->length;
798
799 flist->count++;
800
801 if (verbose > 2)
802 rprintf(FINFO,"recv_file_name(%s)\n",f_name(flist->files[i]));
803 }
804
805
806 if (verbose > 2)
807 rprintf(FINFO,"received %d names\n",flist->count);
808
809 clean_flist(flist, relative_paths);
810
811 if (verbose && recurse && !am_server) {
812 rprintf(FINFO,"done\n");
813 }
814
815 /* now recv the uid/gid list. This was introduced in protocol version 15 */
816 if (f != -1 && remote_version >= 15) {
817 recv_uid_list(f, flist);
818 }
819
820 /* if protocol version is >= 17 then recv the io_error flag */
821 if (f != -1 && remote_version >= 17) {
822 io_error |= read_int(f);
823 }
824
825 if (list_only) {
826 int i;
827 for (i=0;i<flist->count;i++) {
828 list_file_entry(flist->files[i]);
829 }
830 }
831
832
833 if (verbose > 2)
834 rprintf(FINFO,"recv_file_list done\n");
835
836 stats.flist_size = stats.total_read - start_read;
837 stats.num_files = flist->count;
838
839 return flist;
840
841oom:
842 out_of_memory("recv_file_list");
843 return NULL; /* not reached */
844}
845
846
847int file_compare(struct file_struct **f1,struct file_struct **f2)
848{
849 if (!(*f1)->basename && !(*f2)->basename) return 0;
850 if (!(*f1)->basename) return -1;
851 if (!(*f2)->basename) return 1;
852 if ((*f1)->dirname == (*f2)->dirname)
853 return u_strcmp((*f1)->basename, (*f2)->basename);
854 return u_strcmp(f_name(*f1),f_name(*f2));
855}
856
857
858int flist_find(struct file_list *flist,struct file_struct *f)
859{
860 int low=0,high=flist->count-1;
861
862 if (flist->count <= 0) return -1;
863
864 while (low != high) {
865 int mid = (low+high)/2;
866 int ret = file_compare(&flist->files[flist_up(flist, mid)],&f);
867 if (ret == 0) return flist_up(flist, mid);
868 if (ret > 0) {
869 high=mid;
870 } else {
871 low=mid+1;
872 }
873 }
874
875 if (file_compare(&flist->files[flist_up(flist,low)],&f) == 0)
876 return flist_up(flist,low);
877 return -1;
878}
879
880
881/*
882 * free up one file
883 */
884static void free_file(struct file_struct *file)
885{
886 if (!file) return;
887 if (file->basename) free(file->basename);
888 if (file->link) free(file->link);
889 if (file->sum) free(file->sum);
890 memset((char *)file, 0, sizeof(*file));
891}
892
893
894/*
895 * free up all elements in a flist
896 */
897void flist_free(struct file_list *flist)
898{
899 int i;
900 for (i=1;i<flist->count;i++) {
901 free_file(flist->files[i]);
902 free(flist->files[i]);
903 }
904 memset((char *)flist->files, 0, sizeof(flist->files[0])*flist->count);
905 free(flist->files);
906 memset((char *)flist, 0, sizeof(*flist));
907 free(flist);
908}
909
910
911/*
912 * This routine ensures we don't have any duplicate names in our file list.
913 * duplicate names can cause corruption because of the pipelining
914 */
915static void clean_flist(struct file_list *flist, int strip_root)
916{
917 int i;
918
919 if (!flist || flist->count == 0)
920 return;
921
922 qsort(flist->files,flist->count,
923 sizeof(flist->files[0]),
924 (int (*)())file_compare);
925
926 for (i=1;i<flist->count;i++) {
927 if (flist->files[i]->basename &&
928 flist->files[i-1]->basename &&
929 strcmp(f_name(flist->files[i]),
930 f_name(flist->files[i-1])) == 0) {
931 if (verbose > 1 && !am_server)
932 rprintf(FINFO,"removing duplicate name %s from file list %d\n",
933 f_name(flist->files[i-1]),i-1);
934 free_file(flist->files[i]);
935 }
936 }
937
938 if (strip_root) {
939 /* we need to strip off the root directory in the case
940 of relative paths, but this must be done _after_
941 the sorting phase */
942 for (i=0;i<flist->count;i++) {
943 if (flist->files[i]->dirname &&
944 flist->files[i]->dirname[0] == '/') {
945 memmove(&flist->files[i]->dirname[0],
946 &flist->files[i]->dirname[1],
947 strlen(flist->files[i]->dirname));
948 }
949
950 if (flist->files[i]->dirname &&
951 !flist->files[i]->dirname[0]) {
952 flist->files[i]->dirname = NULL;
953 }
954 }
955 }
956
957
958 if (verbose <= 3) return;
959
960 for (i=0;i<flist->count;i++) {
961 rprintf(FINFO,"[%d] i=%d %s %s mode=0%o len=%d\n",
962 getpid(), i,
963 NS(flist->files[i]->dirname),
964 NS(flist->files[i]->basename),
965 flist->files[i]->mode,
966 (int)flist->files[i]->length);
967 }
968}
969
970
971/*
972 * return the full filename of a flist entry
973 */
974char *f_name(struct file_struct *f)
975{
976 static char names[10][MAXPATHLEN];
977 static int n;
978 char *p = names[n];
979
980 if (!f || !f->basename) return NULL;
981
982 n = (n+1)%10;
983
984 if (f->dirname) {
985 strlcpy(p, f->dirname, MAXPATHLEN);
986 strlcat(p, "/", MAXPATHLEN);
987 strlcat(p, f->basename, MAXPATHLEN);
988 } else {
989 strlcpy(p, f->basename, MAXPATHLEN);
990 }
991
992 return p;
993}
994