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