X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/ea847c623231ebbf9cceabe855ae0b40780f9040..8a6f3fead8667434ef08b3c1de2e3c899f1d70ee:/flist.c diff --git a/flist.c b/flist.c index ee18d4d9..8d4defb9 100644 --- a/flist.c +++ b/flist.c @@ -21,10 +21,6 @@ /** @file flist.c * Generate and receive file lists * - * @todo Get rid of the string_area optimization. Efficiently - * allocating blocks is the responsibility of the system's malloc - * library, not of rsync. - * * @sa http://lists.samba.org/pipermail/rsync/2000-June/002351.html * **/ @@ -35,58 +31,75 @@ extern struct stats stats; extern int verbose; extern int do_progress; +extern int am_root; extern int am_server; +extern int am_daemon; +extern int am_sender; extern int always_checksum; extern int module_id; extern int ignore_errors; - -extern int cvs_exclude; +extern int numeric_ids; extern int recurse; +extern int xfer_dirs; extern char curr_dir[MAXPATHLEN]; -extern char *files_from; +extern unsigned int curr_dir_len; +extern char *backup_dir; +extern char *backup_suffix; extern int filesfrom_fd; extern int one_file_system; -extern int make_backups; +extern int keep_dirlinks; extern int preserve_links; extern int preserve_hard_links; extern int preserve_perms; extern int preserve_devices; extern int preserve_uid; extern int preserve_gid; -extern int preserve_times; extern int relative_paths; extern int implied_dirs; +extern int make_backups; +extern int backup_suffix_len; extern int copy_links; extern int copy_unsafe_links; extern int protocol_version; extern int sanitize_paths; +extern int max_delete; +extern int orig_umask; +extern int list_only; -extern int read_batch; -extern int write_batch; - -extern struct exclude_struct **exclude_list; -extern struct exclude_struct **server_exclude_list; -extern struct exclude_struct **local_exclude_list; +extern struct filter_list_struct filter_list; +extern struct filter_list_struct server_filter_list; int io_error; -static struct file_struct null_file; static char empty_sum[MD4_SUM_LENGTH]; +static unsigned int file_struct_len; +static struct file_list *received_flist; +static dev_t filesystem_dev; /* used to implement -x */ +static int deletion_count = 0; /* used to implement --max-delete */ static void clean_flist(struct file_list *flist, int strip_root, int no_dups); +static void output_flist(struct file_list *flist, const char *whose_list); + +void init_flist(void) +{ + struct file_struct f; + + /* Figure out how big the file_struct is without trailing padding */ + file_struct_len = offsetof(struct file_struct, flags) + sizeof f.flags; +} static int show_filelist_p(void) { - return verbose && (recurse || files_from) && !am_server; + return verbose && xfer_dirs && !am_server; } static void start_filelist_progress(char *kind) { rprintf(FINFO, "%s ... ", kind); - if ((verbose > 1) || do_progress) + if (verbose > 1 || do_progress) rprintf(FINFO, "\n"); rflush(FINFO); } @@ -100,7 +113,7 @@ static void emit_filelist_progress(const struct file_list *flist) static void maybe_emit_filelist_progress(const struct file_list *flist) { - if (do_progress && show_filelist_p() && ((flist->count % 100) == 0)) + if (do_progress && show_filelist_p() && (flist->count % 100) == 0) emit_filelist_progress(flist); } @@ -121,80 +134,30 @@ void show_flist_stats(void) } -static struct string_area *string_area_new(int size) -{ - struct string_area *a; - - if (size <= 0) - size = ARENA_SIZE; - a = new(struct string_area); - if (!a) - out_of_memory("string_area_new"); - a->current = a->base = new_array(char, size); - if (!a->current) - out_of_memory("string_area_new buffer"); - a->end = a->base + size; - a->next = NULL; - - return a; -} - -static void string_area_free(struct string_area *a) -{ - struct string_area *next; - - for (; a; a = next) { - next = a->next; - free(a->base); - } -} - -static char *string_area_malloc(struct string_area **ap, int size) -{ - char *p; - struct string_area *a; - - /* does the request fit into the current space? */ - a = *ap; - if (a->current + size >= a->end) { - /* no; get space, move new string_area to front of the list */ - a = string_area_new(size > ARENA_SIZE ? size : ARENA_SIZE); - a->next = *ap; - *ap = a; - } - - /* have space; do the "allocation." */ - p = a->current; - a->current += size; - return p; -} - -static char *string_area_strdup(struct string_area **ap, const char *src) -{ - char *dest = string_area_malloc(ap, strlen(src) + 1); - return strcpy(dest, src); -} - static void list_file_entry(struct file_struct *f) { char perms[11]; - if (!f->basename) + if (!f->basename) { /* this can happen if duplicate names were removed */ return; + } permstring(perms, f->mode); +#if SUPPORT_LINKS if (preserve_links && S_ISLNK(f->mode)) { rprintf(FINFO, "%s %11.0f %s %s -> %s\n", perms, - (double) f->length, timestring(f->modtime), - f_name(f), f->u.link); - } else { + (double)f->length, timestring(f->modtime), + safe_fname(f_name(f)), safe_fname(f->u.link)); + } else +#endif + { rprintf(FINFO, "%s %11.0f %s %s\n", perms, - (double) f->length, timestring(f->modtime), - f_name(f)); + (double)f->length, timestring(f->modtime), + safe_fname(f_name(f))); } } @@ -213,22 +176,22 @@ static void list_file_entry(struct file_struct *f) * @post @p buffer contains information about the link or the * referrent as appropriate, if they exist. **/ -int readlink_stat(const char *path, STRUCT_STAT *buffer, char *linkbuf) +static int readlink_stat(const char *path, STRUCT_STAT *buffer, char *linkbuf) { #if SUPPORT_LINKS if (copy_links) return do_stat(path, buffer); - if (do_lstat(path, buffer) == -1) + if (link_stat(path, buffer, 0) < 0) return -1; if (S_ISLNK(buffer->st_mode)) { - int l = readlink((char *) path, linkbuf, MAXPATHLEN - 1); + int l = readlink((char *)path, linkbuf, MAXPATHLEN - 1); if (l == -1) return -1; linkbuf[l] = 0; if (copy_unsafe_links && unsafe_symlink(linkbuf, path)) { if (verbose > 1) { rprintf(FINFO,"copying unsafe symlink \"%s\" -> \"%s\"\n", - path, linkbuf); + safe_fname(path), safe_fname(linkbuf)); } return do_stat(path, buffer); } @@ -239,26 +202,31 @@ int readlink_stat(const char *path, STRUCT_STAT *buffer, char *linkbuf) #endif } -int link_stat(const char *path, STRUCT_STAT * buffer) +int link_stat(const char *path, STRUCT_STAT *buffer, int follow_dirlinks) { #if SUPPORT_LINKS if (copy_links) return do_stat(path, buffer); - return do_lstat(path, buffer); + if (do_lstat(path, buffer) < 0) + return -1; + if (follow_dirlinks && S_ISLNK(buffer->st_mode)) { + STRUCT_STAT st; + if (do_stat(path, &st) == 0 && S_ISDIR(st.st_mode)) + *buffer = st; + } + return 0; #else return do_stat(path, buffer); #endif } -/* - * This function is used to check if a file should be included/excluded +/* This function is used to check if a file should be included/excluded * from the list of files based on its name and type etc. The value of - * exclude_level is set to either SERVER_EXCLUDES or ALL_EXCLUDES. - */ -static int check_exclude_file(char *fname, int is_dir, int exclude_level) + * filter_level is set to either SERVER_FILTERS or ALL_FILTERS. */ +static int is_excluded(char *fname, int is_dir, int filter_level) { #if 0 /* This currently never happens, so avoid a useless compare. */ - if (exclude_level == NO_EXCLUDES) + if (filter_level == NO_FILTERS) return 0; #endif if (fname) { @@ -272,47 +240,36 @@ static int check_exclude_file(char *fname, int is_dir, int exclude_level) return 0; } } - if (server_exclude_list - && check_exclude(server_exclude_list, fname, is_dir)) + if (server_filter_list.head + && check_filter(&server_filter_list, fname, is_dir) < 0) return 1; - if (exclude_level != ALL_EXCLUDES) + if (filter_level != ALL_FILTERS) return 0; - if (exclude_list && check_exclude(exclude_list, fname, is_dir)) - return 1; - if (local_exclude_list - && check_exclude(local_exclude_list, fname, is_dir)) + if (filter_list.head + && check_filter(&filter_list, fname, is_dir) < 0) return 1; return 0; } -/* used by the one_file_system code */ -static dev_t filesystem_dev; - -static void set_filesystem(char *fname) -{ - STRUCT_STAT st; - if (link_stat(fname, &st) != 0) - return; - filesystem_dev = st.st_dev; -} - - static int to_wire_mode(mode_t mode) { +#if SUPPORT_LINKS if (S_ISLNK(mode) && (_S_IFLNK != 0120000)) return (mode & ~(_S_IFMT)) | 0120000; - return (int) mode; +#endif + return (int)mode; } static mode_t from_wire_mode(int mode) { if ((mode & (_S_IFMT)) == 0120000 && (_S_IFLNK != 0120000)) return (mode & ~(_S_IFMT)) | _S_IFLNK; - return (mode_t) mode; + return (mode_t)mode; } -static void send_directory(int f, struct file_list *flist, char *dir); +static void send_directory(int f, struct file_list *flist, + char *fbuf, unsigned int offset); static char *flist_dir; static int flist_dir_len; @@ -322,38 +279,41 @@ static int flist_dir_len; * Make sure @p flist is big enough to hold at least @p flist->count * entries. **/ -static void flist_expand(struct file_list *flist) +void flist_expand(struct file_list *flist) { - if (flist->count >= flist->malloced) { - void *new_ptr; - - if (flist->malloced < 1000) - flist->malloced += 1000; - else - flist->malloced *= 2; + struct file_struct **new_ptr; - if (flist->files) { - new_ptr = realloc_array(flist->files, - struct file_struct *, - flist->malloced); - } else { - new_ptr = new_array(struct file_struct *, - flist->malloced); - } + if (flist->count < flist->malloced) + return; - if (verbose >= 2) { - rprintf(FINFO, "[%s] expand file_list to %.0f bytes, did%s move\n", - who_am_i(), - (double)sizeof(flist->files[0]) - * flist->malloced, - (new_ptr == flist->files) ? " not" : ""); - } + if (flist->malloced < FLIST_START) + flist->malloced = FLIST_START; + else if (flist->malloced >= FLIST_LINEAR) + flist->malloced += FLIST_LINEAR; + else + flist->malloced *= 2; + + /* + * In case count jumped or we are starting the list + * with a known size just set it. + */ + if (flist->malloced < flist->count) + flist->malloced = flist->count; + + new_ptr = realloc_array(flist->files, struct file_struct *, + flist->malloced); + + if (verbose >= 2 && flist->malloced != FLIST_START) { + rprintf(FINFO, "[%s] expand file_list to %.0f bytes, did%s move\n", + who_am_i(), + (double)sizeof flist->files[0] * flist->malloced, + (new_ptr == flist->files) ? " not" : ""); + } - flist->files = (struct file_struct **) new_ptr; + flist->files = new_ptr; - if (!flist->files) - out_of_memory("flist_expand"); - } + if (!flist->files) + out_of_memory("flist_expand"); } void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) @@ -361,12 +321,13 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) unsigned short flags; static time_t modtime; static mode_t mode; - static DEV64_T rdev; /* just high bytes in p28 onward */ - static DEV64_T dev; + static int64 dev; + static dev_t rdev; + static uint32 rdev_major; static uid_t uid; static gid_t gid; static char lastname[MAXPATHLEN]; - char *fname, fbuf[MAXPATHLEN]; + char fname[MAXPATHLEN]; int l1, l2; if (f == -1) @@ -375,7 +336,8 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) if (!file) { write_byte(f, 0); modtime = 0, mode = 0; - rdev = 0, dev = 0; + dev = 0, rdev = makedev(0, 0); + rdev_major = 0; uid = 0, gid = 0; *lastname = '\0'; return; @@ -383,7 +345,7 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) io_write_phase = "send_file_entry"; - fname = f_name_to(file, fbuf); + f_name_to(file, fname); flags = base_flags; @@ -394,20 +356,20 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) if (preserve_devices) { if (protocol_version < 28) { if (IS_DEVICE(mode)) { - if (file->u.rdev == rdev) { - /* Set both flags to simplify the test - * when writing the data. */ - flags |= XMIT_SAME_RDEV_pre28 - | XMIT_SAME_HIGH_RDEV; - } else + if (file->u.rdev == rdev) + flags |= XMIT_SAME_RDEV_pre28; + else rdev = file->u.rdev; } else - rdev = 0; + rdev = makedev(0, 0); } else if (IS_DEVICE(mode)) { - if ((file->u.rdev & ~0xFF) == rdev) - flags |= XMIT_SAME_HIGH_RDEV; + rdev = file->u.rdev; + if ((uint32)major(rdev) == rdev_major) + flags |= XMIT_SAME_RDEV_MAJOR; else - rdev = file->u.rdev & ~0xFF; + rdev_major = major(rdev); + if ((uint32)minor(rdev) <= 0xFFu) + flags |= XMIT_RDEV_MINOR_IS_SMALL; } } if (file->uid == uid) @@ -422,6 +384,8 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) flags |= XMIT_SAME_TIME; else modtime = file->modtime; + +#if SUPPORT_HARD_LINKS if (file->link_u.idev) { if (file->F_DEV == dev) { if (protocol_version >= 28) @@ -430,10 +394,11 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) dev = file->F_DEV; flags |= XMIT_HAS_IDEV_DATA; } +#endif for (l1 = 0; - lastname[l1] && (fname[l1] == lastname[l1]) && (l1 < 255); - l1++) {} + lastname[l1] && (fname[l1] == lastname[l1]) && (l1 < 255); + l1++) {} l2 = strlen(fname+l1); if (l1 > 0) @@ -441,19 +406,23 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) if (l2 > 255) flags |= XMIT_LONG_NAME; - /* We must make sure we don't send a zero flags byte or - * the other end will terminate the flist transfer. */ - if (flags == 0 && !S_ISDIR(mode)) - flags |= XMIT_TOP_DIR; /* NOTE: no meaning for non-dir */ + /* We must make sure we don't send a zero flag byte or the + * other end will terminate the flist transfer. Note that + * the use of XMIT_TOP_DIR on a non-dir has no meaning, so + * it's harmless way to add a bit to the first flag byte. */ if (protocol_version >= 28) { - if ((flags & 0xFF00) || flags == 0) { + if (!flags && !S_ISDIR(mode)) + flags |= XMIT_TOP_DIR; + if ((flags & 0xFF00) || !flags) { flags |= XMIT_EXTENDED_FLAGS; write_byte(f, flags); write_byte(f, flags >> 8); } else write_byte(f, flags); } else { - if (flags == 0) + if (!(flags & 0xFF) && !S_ISDIR(mode)) + flags |= XMIT_TOP_DIR; + if (!(flags & 0xFF)) flags |= XMIT_LONG_NAME; write_byte(f, flags); } @@ -471,27 +440,34 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) if (!(flags & XMIT_SAME_MODE)) write_int(f, to_wire_mode(mode)); if (preserve_uid && !(flags & XMIT_SAME_UID)) { - add_uid(uid); + if (!numeric_ids) + add_uid(uid); write_int(f, uid); } if (preserve_gid && !(flags & XMIT_SAME_GID)) { - add_gid(gid); + if (!numeric_ids) + add_gid(gid); write_int(f, gid); } if (preserve_devices && IS_DEVICE(mode)) { - /* If XMIT_SAME_HIGH_RDEV is off, XMIT_SAME_RDEV_pre28 is - * also off. Also, avoid using "rdev" because it may be - * incomplete. */ - if (!(flags & XMIT_SAME_HIGH_RDEV)) - write_int(f, file->u.rdev); - else if (protocol_version >= 28) - write_byte(f, file->u.rdev); + if (protocol_version < 28) { + if (!(flags & XMIT_SAME_RDEV_pre28)) + write_int(f, (int)rdev); + } else { + if (!(flags & XMIT_SAME_RDEV_MAJOR)) + write_int(f, major(rdev)); + if (flags & XMIT_RDEV_MINOR_IS_SMALL) + write_byte(f, minor(rdev)); + else + write_int(f, minor(rdev)); + } } #if SUPPORT_LINKS if (preserve_links && S_ISLNK(mode)) { - write_int(f, strlen(file->u.link)); - write_buf(f, file->u.link, strlen(file->u.link)); + int len = strlen(file->u.link); + write_int(f, len); + write_buf(f, file->u.link, len); } #endif @@ -520,8 +496,8 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) } else sum = NULL; if (sum) { - write_buf(f, sum, protocol_version < 21? 2 - : MD4_SUM_LENGTH); + write_buf(f, sum, + protocol_version < 21 ? 2 : MD4_SUM_LENGTH); } } @@ -532,25 +508,35 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) -void receive_file_entry(struct file_struct **fptr, unsigned short flags, int f) +static void receive_file_entry(struct file_list *flist, int ndx, + unsigned short flags, int f) { static time_t modtime; static mode_t mode; - static DEV64_T rdev; /* just high bytes in p28 onward */ - static DEV64_T dev; + static int64 dev; + static dev_t rdev; + static uint32 rdev_major; static uid_t uid; static gid_t gid; - static char lastname[MAXPATHLEN]; + static char lastname[MAXPATHLEN], *lastdir; + static int lastdir_depth, lastdir_len = -1; + static unsigned int del_hier_name_len = 0; + static int in_del_hier = 0; char thisname[MAXPATHLEN]; unsigned int l1 = 0, l2 = 0; - char *p; + int alloc_len, basename_len, dirname_len, linkname_len, sum_len; + OFF_T file_length; + char *basename, *dirname, *bp; struct file_struct *file; - if (!fptr) { + if (!flist) { modtime = 0, mode = 0; - rdev = 0, dev = 0; + dev = 0, rdev = makedev(0, 0); + rdev_major = 0; uid = 0, gid = 0; *lastname = '\0'; + lastdir_len = -1; + in_del_hier = 0; return; } @@ -562,16 +548,10 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, int f) else l2 = read_byte(f); - file = new(struct file_struct); - if (!file) - out_of_memory("receive_file_entry"); - memset((char *) file, 0, sizeof(*file)); - (*fptr) = file; - if (l2 >= MAXPATHLEN - l1) { rprintf(FERROR, "overflow: flags=0x%x l1=%d l2=%d lastname=%s\n", - flags, l1, l2, lastname); + flags, l1, l2, safe_fname(lastname)); overflow("receive_file_entry"); } @@ -580,117 +560,174 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, int f) thisname[l1 + l2] = 0; strlcpy(lastname, thisname, MAXPATHLEN); - lastname[MAXPATHLEN - 1] = 0; - - clean_fname(thisname); - if (sanitize_paths) { - sanitize_path(thisname, NULL); - } + clean_fname(thisname, 0); - if ((p = strrchr(thisname, '/'))) { - static char *lastdir; - *p = 0; - if (lastdir && strcmp(thisname, lastdir) == 0) - file->dirname = lastdir; - else { - file->dirname = strdup(thisname); - lastdir = file->dirname; - } - file->basename = strdup(p + 1); + if (sanitize_paths) + sanitize_path(thisname, thisname, "", 0); + + if ((basename = strrchr(thisname, '/')) != NULL) { + dirname_len = ++basename - thisname; /* counts future '\0' */ + if (lastdir_len == dirname_len - 1 + && strncmp(thisname, lastdir, lastdir_len) == 0) { + dirname = lastdir; + dirname_len = 0; /* indicates no copy is needed */ + } else + dirname = thisname; } else { - file->dirname = NULL; - file->basename = strdup(thisname); + basename = thisname; + dirname = NULL; + dirname_len = 0; } + basename_len = strlen(basename) + 1; /* count the '\0' */ - if (!file->basename) - out_of_memory("receive_file_entry 1"); - - file->flags = flags & XMIT_TOP_DIR ? FLAG_TOP_DIR : 0; - file->length = read_longint(f); + file_length = read_longint(f); if (!(flags & XMIT_SAME_TIME)) modtime = (time_t)read_int(f); - file->modtime = modtime; if (!(flags & XMIT_SAME_MODE)) mode = from_wire_mode(read_int(f)); - file->mode = mode; - if (preserve_uid) { - if (!(flags & XMIT_SAME_UID)) - uid = (uid_t)read_int(f); - file->uid = uid; - } - if (preserve_gid) { - if (!(flags & XMIT_SAME_GID)) - gid = (gid_t)read_int(f); - file->gid = gid; - } + if (preserve_uid && !(flags & XMIT_SAME_UID)) + uid = (uid_t)read_int(f); + if (preserve_gid && !(flags & XMIT_SAME_GID)) + gid = (gid_t)read_int(f); + if (preserve_devices) { if (protocol_version < 28) { if (IS_DEVICE(mode)) { if (!(flags & XMIT_SAME_RDEV_pre28)) - rdev = (DEV64_T)read_int(f); - file->u.rdev = rdev; + rdev = (dev_t)read_int(f); } else - rdev = 0; + rdev = makedev(0, 0); } else if (IS_DEVICE(mode)) { - if (!(flags & XMIT_SAME_HIGH_RDEV)) { - file->u.rdev = (DEV64_T)read_int(f); - rdev = file->u.rdev & ~0xFF; - } else - file->u.rdev = rdev | (DEV64_T)read_byte(f); + uint32 rdev_minor; + if (!(flags & XMIT_SAME_RDEV_MAJOR)) + rdev_major = read_int(f); + if (flags & XMIT_RDEV_MINOR_IS_SMALL) + rdev_minor = read_byte(f); + else + rdev_minor = read_int(f); + rdev = makedev(rdev_major, rdev_minor); } } +#if SUPPORT_LINKS if (preserve_links && S_ISLNK(mode)) { - int l = read_int(f); - if (l < 0) { - rprintf(FERROR, "overflow: l=%d\n", l); + linkname_len = read_int(f) + 1; /* count the '\0' */ + if (linkname_len <= 0 || linkname_len > MAXPATHLEN) { + rprintf(FERROR, "overflow: linkname_len=%d\n", + linkname_len - 1); overflow("receive_file_entry"); } - if (!(file->u.link = new_array(char, l + 1))) - out_of_memory("receive_file_entry 2"); - read_sbuf(f, file->u.link, l); + } + else +#endif + linkname_len = 0; + + sum_len = always_checksum && S_ISREG(mode) ? MD4_SUM_LENGTH : 0; + + alloc_len = file_struct_len + dirname_len + basename_len + + linkname_len + sum_len; + bp = pool_alloc(flist->file_pool, alloc_len, "receive_file_entry"); + + file = flist->files[ndx] = (struct file_struct *)bp; + memset(bp, 0, file_struct_len); + bp += file_struct_len; + + file->flags = 0; + file->modtime = modtime; + file->length = file_length; + file->mode = mode; + file->uid = uid; + file->gid = gid; + + if (dirname_len) { + file->dirname = lastdir = bp; + lastdir_len = dirname_len - 1; + memcpy(bp, dirname, dirname_len - 1); + bp += dirname_len; + bp[-1] = '\0'; + lastdir_depth = count_dir_elements(lastdir); + file->dir.depth = lastdir_depth + 1; + } else if (dirname) { + file->dirname = dirname; /* we're reusing lastname */ + file->dir.depth = lastdir_depth + 1; + } else + file->dir.depth = 1; + + if (S_ISDIR(mode)) { + if (basename_len == 1+1 && *basename == '.') /* +1 for '\0' */ + file->dir.depth--; + if (flags & XMIT_TOP_DIR) { + in_del_hier = 1; + del_hier_name_len = file->dir.depth == 0 ? 0 : l1 + l2; + file->flags |= FLAG_TOP_DIR | FLAG_DEL_HERE; + } else if (in_del_hier) { + if (!relative_paths || !del_hier_name_len + || (l1 >= del_hier_name_len + && thisname[del_hier_name_len] == '/')) + file->flags |= FLAG_DEL_HERE; + else + in_del_hier = 0; + } + } + + file->basename = bp; + memcpy(bp, basename, basename_len); + bp += basename_len; + + if (preserve_devices && IS_DEVICE(mode)) + file->u.rdev = rdev; + +#if SUPPORT_LINKS + if (linkname_len) { + file->u.link = bp; + read_sbuf(f, bp, linkname_len - 1); if (sanitize_paths) - sanitize_path(file->u.link, file->dirname); + sanitize_path(bp, bp, "", lastdir_depth); + bp += linkname_len; } +#endif + #if SUPPORT_HARD_LINKS if (preserve_hard_links && protocol_version < 28 && S_ISREG(mode)) flags |= XMIT_HAS_IDEV_DATA; if (flags & XMIT_HAS_IDEV_DATA) { - if (!(file->link_u.idev = new(struct idev))) - out_of_memory("file inode data"); + int64 inode; if (protocol_version < 26) { dev = read_int(f); - file->F_INODE = read_int(f); + inode = read_int(f); } else { if (!(flags & XMIT_SAME_DEV)) dev = read_longint(f); - file->F_INODE = read_longint(f); + inode = read_longint(f); + } + if (flist->hlink_pool) { + file->link_u.idev = pool_talloc(flist->hlink_pool, + struct idev, 1, "inode_table"); + file->F_INODE = inode; + file->F_DEV = dev; } - file->F_DEV = dev; } #endif if (always_checksum) { char *sum; - if (S_ISREG(mode)) { - sum = file->u.sum = new_array(char, MD4_SUM_LENGTH); - if (!sum) - out_of_memory("md4 sum"); + if (sum_len) { + file->u.sum = sum = bp; + /*bp += sum_len;*/ } else if (protocol_version < 28) { /* Prior to 28, we get a useless set of nulls. */ sum = empty_sum; } else sum = NULL; if (sum) { - read_buf(f, sum, protocol_version < 21? 2 - : MD4_SUM_LENGTH); + read_buf(f, sum, + protocol_version < 21 ? 2 : MD4_SUM_LENGTH); } } if (!preserve_perms) { - extern int orig_umask; /* set an appropriate set of permissions based on original * permissions and umask. This emulates what GNU cp does */ file->mode &= ~orig_umask; @@ -698,10 +735,6 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, int f) } -#define STRDUP(ap, p) (ap ? string_area_strdup(ap, p) : strdup(p)) -/* IRIX cc cares that the operands to the ternary have the same type. */ -#define MALLOC(ap, i) (ap ? (void*) string_area_malloc(ap, i) : malloc(i)) - /** * Create a file_struct for a named file by reading its stat() * information and performing extensive checks against global @@ -717,129 +750,217 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, int f) * statting directories if we're not recursing, but this is not a very * important case. Some systems may not have d_type. **/ -struct file_struct *make_file(char *fname, struct string_area **ap, - int exclude_level) +struct file_struct *make_file(char *fname, struct file_list *flist, + int filter_level) { + static char *lastdir; + static int lastdir_len = -1; struct file_struct *file; STRUCT_STAT st; char sum[SUM_LENGTH]; - char *p; - char cleaned_name[MAXPATHLEN]; - char linkbuf[MAXPATHLEN]; + char thisname[MAXPATHLEN]; + char linkname[MAXPATHLEN]; + int alloc_len, basename_len, dirname_len, linkname_len, sum_len; + char *basename, *dirname, *bp; unsigned short flags = 0; - if (strlcpy(cleaned_name, fname, sizeof cleaned_name) - >= sizeof cleaned_name - flist_dir_len) { - rprintf(FINFO, "skipping overly long name: %s\n", fname); + if (!flist || !flist->count) /* Ignore lastdir when invalid. */ + lastdir_len = -1; + + if (strlcpy(thisname, fname, sizeof thisname) + >= sizeof thisname - flist_dir_len) { + rprintf(FINFO, "skipping overly long name: %s\n", + safe_fname(fname)); return NULL; } - clean_fname(cleaned_name); + clean_fname(thisname, 0); if (sanitize_paths) - sanitize_path(cleaned_name, NULL); - fname = cleaned_name; + sanitize_path(thisname, thisname, "", 0); memset(sum, 0, SUM_LENGTH); - if (readlink_stat(fname, &st, linkbuf) != 0) { + if (readlink_stat(thisname, &st, linkname) != 0) { int save_errno = errno; - if (errno == ENOENT && exclude_level != NO_EXCLUDES) { - /* either symlink pointing nowhere or file that - * was removed during rsync run; see if excluded - * before reporting an error */ - if (check_exclude_file(fname, 0, exclude_level)) { - /* file is excluded anyway, ignore silently */ - return NULL; + /* See if file is excluded before reporting an error. */ + if (filter_level != NO_FILTERS + && is_excluded(thisname, 0, filter_level)) + return NULL; + if (save_errno == ENOENT) { +#if SUPPORT_LINKS + /* Avoid "vanished" error if symlink points nowhere. */ + if (copy_links && do_lstat(thisname, &st) == 0 + && S_ISLNK(st.st_mode)) { + io_error |= IOERR_GENERAL; + rprintf(FERROR, "symlink has no referent: %s\n", + full_fname(thisname)); + } else +#endif + { + enum logcode c = am_daemon && protocol_version < 28 + ? FERROR : FINFO; + io_error |= IOERR_VANISHED; + rprintf(c, "file has vanished: %s\n", + full_fname(thisname)); } + } else { + io_error |= IOERR_GENERAL; + rsyserr(FERROR, save_errno, "readlink %s failed", + full_fname(thisname)); } - io_error |= IOERR_GENERAL; - rprintf(FERROR, "readlink %s failed: %s\n", - full_fname(fname), strerror(save_errno)); return NULL; } - /* backup.c calls us with exclude_level set to NO_EXCLUDES. */ - if (exclude_level == NO_EXCLUDES) - goto skip_excludes; + /* backup.c calls us with filter_level set to NO_FILTERS. */ + if (filter_level == NO_FILTERS) + goto skip_filters; - if (S_ISDIR(st.st_mode) && !recurse && !files_from) { - rprintf(FINFO, "skipping directory %s\n", fname); + if (S_ISDIR(st.st_mode) && !xfer_dirs) { + rprintf(FINFO, "skipping directory %s\n", safe_fname(thisname)); return NULL; } - if (one_file_system && st.st_dev != filesystem_dev) { - /* We allow a directory though to preserve the mount point. - * However, flag it so that we don't recurse. */ - if (!S_ISDIR(st.st_mode)) - return NULL; + /* We only care about directories because we need to avoid recursing + * into a mount-point directory, not to avoid copying a symlinked + * file if -L (or similar) was specified. */ + if (one_file_system && st.st_dev != filesystem_dev + && S_ISDIR(st.st_mode)) flags |= FLAG_MOUNT_POINT; - } - if (check_exclude_file(fname, S_ISDIR(st.st_mode) != 0, exclude_level)) + if (is_excluded(thisname, S_ISDIR(st.st_mode) != 0, filter_level)) return NULL; - if (lp_ignore_nonreadable(module_id) && access(fname, R_OK) != 0) - return NULL; + if (lp_ignore_nonreadable(module_id)) { +#if SUPPORT_LINKS + if (!S_ISLNK(st.st_mode)) +#endif + if (access(thisname, R_OK) != 0) + return NULL; + } - skip_excludes: +skip_filters: if (verbose > 2) { rprintf(FINFO, "[%s] make_file(%s,*,%d)\n", - who_am_i(), fname, exclude_level); + who_am_i(), safe_fname(thisname), filter_level); } - file = new(struct file_struct); - if (!file) - out_of_memory("make_file"); - memset((char *) file, 0, sizeof(*file)); - file->flags = flags; + if ((basename = strrchr(thisname, '/')) != NULL) { + dirname_len = ++basename - thisname; /* counts future '\0' */ + if (lastdir_len == dirname_len - 1 + && strncmp(thisname, lastdir, lastdir_len) == 0) { + dirname = lastdir; + dirname_len = 0; /* indicates no copy is needed */ + } else + dirname = thisname; + } else { + basename = thisname; + dirname = NULL; + dirname_len = 0; + } + basename_len = strlen(basename) + 1; /* count the '\0' */ - if ((p = strrchr(fname, '/'))) { - static char *lastdir; - *p = 0; - if (lastdir && strcmp(fname, lastdir) == 0) - file->dirname = lastdir; - else { - file->dirname = strdup(fname); - lastdir = file->dirname; - } - file->basename = STRDUP(ap, p + 1); - *p = '/'; +#if SUPPORT_LINKS + linkname_len = S_ISLNK(st.st_mode) ? strlen(linkname) + 1 : 0; +#else + linkname_len = 0; +#endif + + sum_len = always_checksum && S_ISREG(st.st_mode) ? MD4_SUM_LENGTH : 0; + + alloc_len = file_struct_len + dirname_len + basename_len + + linkname_len + sum_len; + if (flist) { + bp = pool_alloc(flist->file_pool, alloc_len, + "receive_file_entry"); } else { - file->dirname = NULL; - file->basename = STRDUP(ap, fname); + if (!(bp = new_array(char, alloc_len))) + out_of_memory("receive_file_entry"); } + file = (struct file_struct *)bp; + memset(bp, 0, file_struct_len); + bp += file_struct_len; + + file->flags = flags; file->modtime = st.st_mtime; file->length = st.st_size; file->mode = st.st_mode; file->uid = st.st_uid; file->gid = st.st_gid; - if (preserve_hard_links) { - if (protocol_version < 28 ? S_ISREG(st.st_mode) - : !S_ISDIR(st.st_mode) && st.st_nlink > 1) { - if (!(file->link_u.idev = new(struct idev))) - out_of_memory("file inode data"); - file->F_DEV = st.st_dev; - file->F_INODE = st.st_ino; + +#if SUPPORT_HARD_LINKS + if (flist && flist->hlink_pool) { + if (protocol_version < 28) { + if (S_ISREG(st.st_mode)) + file->link_u.idev = pool_talloc( + flist->hlink_pool, struct idev, 1, + "inode_table"); + } else { + if (!S_ISDIR(st.st_mode) && st.st_nlink > 1) + file->link_u.idev = pool_talloc( + flist->hlink_pool, struct idev, 1, + "inode_table"); } } -#ifdef HAVE_STRUCT_STAT_ST_RDEV - if (IS_DEVICE(st.st_mode)) + if (file->link_u.idev) { + file->F_DEV = st.st_dev; + file->F_INODE = st.st_ino; + } +#endif + + if (dirname_len) { + file->dirname = lastdir = bp; + lastdir_len = dirname_len - 1; + memcpy(bp, dirname, dirname_len - 1); + bp += dirname_len; + bp[-1] = '\0'; + } else if (dirname) + file->dirname = dirname; + + file->basename = bp; + memcpy(bp, basename, basename_len); + bp += basename_len; + +#if HAVE_STRUCT_STAT_ST_RDEV + if (preserve_devices && IS_DEVICE(st.st_mode)) file->u.rdev = st.st_rdev; #endif #if SUPPORT_LINKS - if (S_ISLNK(st.st_mode)) - file->u.link = STRDUP(ap, linkbuf); + if (linkname_len) { + file->u.link = bp; + memcpy(bp, linkname, linkname_len); + bp += linkname_len; + } #endif - if (always_checksum && S_ISREG(st.st_mode)) { - if (!(file->u.sum = (char*)MALLOC(ap, MD4_SUM_LENGTH))) - out_of_memory("md4 sum"); - file_checksum(fname, file->u.sum, st.st_size); + if (sum_len) { + file->u.sum = bp; + file_checksum(thisname, bp, st.st_size); + /*bp += sum_len;*/ } - file->basedir = flist_dir; + file->dir.root = flist_dir; + + /* This code is only used by the receiver when it is building + * a list of files for a delete pass. */ + if (keep_dirlinks && linkname_len && flist) { + STRUCT_STAT st2; + int i = flist_find(received_flist, file); + if (i >= 0 && S_ISDIR(received_flist->files[i]->mode) + && do_stat(thisname, &st2) == 0 && S_ISDIR(st2.st_mode)) { + file->modtime = st2.st_mtime; + file->length = st2.st_size; + file->mode = st2.st_mode; + file->uid = st2.st_uid; + file->gid = st2.st_gid; + file->u.link = NULL; + if (file->link_u.idev) { + pool_free(flist->hlink_pool, 0, file->link_u.idev); + file->link_u.idev = NULL; + } + } + } if (!S_ISDIR(st.st_mode)) stats.total_size += st.st_size; @@ -853,115 +974,86 @@ void send_file_name(int f, struct file_list *flist, char *fname, { struct file_struct *file; char fbuf[MAXPATHLEN]; - extern int delete_excluded; - - /* f is set to -1 when calculating deletion file list */ - file = make_file(fname, &flist->string_area, - f == -1 && delete_excluded? SERVER_EXCLUDES - : ALL_EXCLUDES); - if (!file) + if (!(file = make_file(fname, flist, ALL_FILTERS))) return; maybe_emit_filelist_progress(flist); flist_expand(flist); - if (write_batch) - file->flags |= FLAG_TOP_DIR; - if (file->basename[0]) { flist->files[flist->count++] = file; send_file_entry(file, f, base_flags); } if (recursive && S_ISDIR(file->mode) - && !(file->flags & FLAG_MOUNT_POINT)) { - struct exclude_struct **last_exclude_list = local_exclude_list; - send_directory(f, flist, f_name_to(file, fbuf)); - local_exclude_list = last_exclude_list; - return; + && !(file->flags & FLAG_MOUNT_POINT) && f_name_to(file, fbuf)) { + void *save_filters; + unsigned int len = strlen(fbuf); + if (len > 1 && fbuf[len-1] == '/') + fbuf[--len] = '\0'; + if (len >= MAXPATHLEN - 1) { + io_error |= IOERR_GENERAL; + rprintf(FERROR, "skipping long-named directory: %s\n", + full_fname(fbuf)); + return; + } + save_filters = push_local_filters(fbuf, len); + send_directory(f, flist, fbuf, len); + pop_local_filters(save_filters); } } -static void send_directory(int f, struct file_list *flist, char *dir) +/* Note that the "recurse" value either contains -1, for infinite recursion, + * or a number >= 0 indicating how many levels of recursion we will allow. + * This function is normally called by the sender, but the receiving side + * also calls it from delete_in_dir() with f set to -1 so that we just + * construct the file list in memory without sending it over the wire. */ +static void send_directory(int f, struct file_list *flist, + char *fbuf, unsigned int len) { - DIR *d; struct dirent *di; - char fname[MAXPATHLEN]; - unsigned int offset; char *p; + DIR *d; - d = opendir(dir); - if (!d) { + if (!(d = opendir(fbuf))) { io_error |= IOERR_GENERAL; - rprintf(FERROR, "opendir %s failed: %s\n", - full_fname(dir), strerror(errno)); + rsyserr(FERROR, errno, "opendir %s failed", full_fname(fbuf)); return; } - offset = strlcpy(fname, dir, MAXPATHLEN); - p = fname + offset; - if (offset >= MAXPATHLEN || p[-1] != '/') { - if (offset >= MAXPATHLEN - 1) { - io_error |= IOERR_GENERAL; - rprintf(FERROR, "skipping long-named directory: %s\n", - full_fname(fname)); - closedir(d); - return; - } + p = fbuf + len; + if (len != 1 || *fbuf != '/') *p++ = '/'; - offset++; - } - - local_exclude_list = NULL; - - if (cvs_exclude) { - if (strlcpy(p, ".cvsignore", MAXPATHLEN - offset) - < MAXPATHLEN - offset) - add_exclude_file(&local_exclude_list,fname,MISSING_OK,ADD_EXCLUDE); - else { - io_error |= IOERR_GENERAL; - rprintf(FINFO, - "cannot cvs-exclude in long-named directory %s\n", - full_fname(fname)); - } - } + *p = '\0'; for (errno = 0, di = readdir(d); di; errno = 0, di = readdir(d)) { char *dname = d_name(di); if (dname[0] == '.' && (dname[1] == '\0' || (dname[1] == '.' && dname[2] == '\0'))) continue; - if (strlcpy(p, dname, MAXPATHLEN - offset) < MAXPATHLEN - offset) - send_file_name(f, flist, fname, recurse, 0); - else { + if (strlcpy(p, dname, MAXPATHLEN - len) < MAXPATHLEN - len) { + int do_subdirs = recurse >= 1 ? recurse-- : recurse; + send_file_name(f, flist, fbuf, do_subdirs, 0); + } else { io_error |= IOERR_GENERAL; rprintf(FINFO, "cannot send long-named file %s\n", - full_fname(fname)); + full_fname(fbuf)); } } if (errno) { io_error |= IOERR_GENERAL; - rprintf(FERROR, "readdir(%s): (%d) %s\n", - dir, errno, strerror(errno)); + *p = '\0'; + rsyserr(FERROR, errno, "readdir(%s)", full_fname(fbuf)); } - if (local_exclude_list) - free_exclude_list(&local_exclude_list); /* Zeros pointer too */ - closedir(d); } -/** - * The delete_files() function in receiver.c sets f to -1 so that we just - * construct the file list in memory without sending it over the wire. It - * also has the side-effect of ignoring user-excludes if delete_excluded - * is set (so that the delete list includes user-excluded files). - **/ struct file_list *send_file_list(int f, int argc, char *argv[]) { int l; @@ -969,46 +1061,47 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) char *p, *dir, olddir[sizeof curr_dir]; char lastpath[MAXPATHLEN] = ""; struct file_list *flist; + struct timeval start_tv, end_tv; int64 start_write; int use_ff_fd = 0; - if (show_filelist_p() && f != -1) + if (show_filelist_p()) start_filelist_progress("building file list"); start_write = stats.total_written; + gettimeofday(&start_tv, NULL); - flist = flist_new(); + flist = flist_new(WITH_HLINK, "send_file_list"); - if (f != -1) { - io_start_buffering_out(f); - if (filesfrom_fd >= 0) { - if (argv[0] && !push_dir(argv[0])) { - rprintf(FERROR, "push_dir %s failed: %s\n", - full_fname(argv[0]), strerror(errno)); - exit_cleanup(RERR_FILESELECT); - } - use_ff_fd = 1; + io_start_buffering_out(); + if (filesfrom_fd >= 0) { + if (argv[0] && !push_dir(argv[0])) { + rsyserr(FERROR, errno, "push_dir %s failed", + full_fname(argv[0])); + exit_cleanup(RERR_FILESELECT); } + use_ff_fd = 1; } while (1) { char fname2[MAXPATHLEN]; char *fname = fname2; + int do_subdirs; if (use_ff_fd) { if (read_filesfrom_line(filesfrom_fd, fname) == 0) break; - sanitize_path(fname, NULL); + sanitize_path(fname, fname, "", 0); } else { if (argc-- == 0) break; strlcpy(fname, *argv++, MAXPATHLEN); if (sanitize_paths) - sanitize_path(fname, NULL); + sanitize_path(fname, fname, "", 0); } l = strlen(fname); - if (fname[l - 1] == '/') { + if (!l || fname[l - 1] == '/') { if (l == 2 && fname[0] == '.') { /* Turn "./" into just "." rather than "./." */ fname[1] = '\0'; @@ -1017,18 +1110,22 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) fname[l] = '\0'; } } + if (fname[l-1] == '.' && (l == 1 || fname[l-2] == '/')) { + if (!recurse && xfer_dirs) + recurse = 1; /* allow one level */ + } else if (recurse > 0) + recurse = 0; - if (link_stat(fname, &st) != 0) { - if (f != -1) { - io_error |= IOERR_GENERAL; - rprintf(FERROR, "link_stat %s failed: %s\n", - full_fname(fname), strerror(errno)); - } + if (link_stat(fname, &st, keep_dirlinks) != 0) { + io_error |= IOERR_GENERAL; + rsyserr(FERROR, errno, "link_stat %s failed", + full_fname(fname)); continue; } - if (S_ISDIR(st.st_mode) && !recurse && !files_from) { - rprintf(FINFO, "skipping directory %s\n", fname); + if (S_ISDIR(st.st_mode) && !xfer_dirs) { + rprintf(FINFO, "skipping directory %s\n", + safe_fname(fname)); continue; } @@ -1045,7 +1142,7 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) dir = fname; fname = p + 1; } - } else if (f != -1 && implied_dirs && (p=strrchr(fname,'/')) && p != fname) { + } else if (implied_dirs && (p=strrchr(fname,'/')) && p != fname) { /* this ensures we send the intermediate directories, thus getting their permissions right */ char *lp = lastpath, *fn = fname, *slash = fname; @@ -1059,21 +1156,17 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) } *p = '/'; if (fn != p || (*lp && *lp != '/')) { - int copy_links_saved = copy_links; - int recurse_saved = recurse; + int save_copy_links = copy_links; + int save_xfer_dirs = xfer_dirs; copy_links = copy_unsafe_links; - /* set recurse to 1 to prevent make_file - * from ignoring directory, but still - * turn off the recursive parameter to - * send_file_name */ - recurse = 1; + xfer_dirs = 1; while ((slash = strchr(slash+1, '/')) != 0) { *slash = 0; send_file_name(f, flist, fname, 0, 0); *slash = '/'; } - copy_links = copy_links_saved; - recurse = recurse_saved; + copy_links = save_copy_links; + xfer_dirs = save_xfer_dirs; *p = 0; strlcpy(lastpath, fname, sizeof lastpath); *p = '/'; @@ -1091,8 +1184,8 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) if (!push_dir(dir)) { io_error |= IOERR_GENERAL; - rprintf(FERROR, "push_dir %s failed: %s\n", - full_fname(dir), strerror(errno)); + rsyserr(FERROR, errno, "push_dir %s failed", + full_fname(dir)); continue; } @@ -1106,44 +1199,58 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) } if (one_file_system) - set_filesystem(fname); + filesystem_dev = st.st_dev; - send_file_name(f, flist, fname, recurse, XMIT_TOP_DIR); + do_subdirs = recurse >= 1 ? recurse-- : recurse; + send_file_name(f, flist, fname, do_subdirs, XMIT_TOP_DIR); if (olddir[0]) { flist_dir = NULL; flist_dir_len = 0; if (!pop_dir(olddir)) { - rprintf(FERROR, "pop_dir %s failed: %s\n", - full_fname(dir), strerror(errno)); + rsyserr(FERROR, errno, "pop_dir %s failed", + full_fname(dir)); exit_cleanup(RERR_FILESELECT); } } } - if (f != -1) { - send_file_entry(NULL, f, 0); + gettimeofday(&end_tv, NULL); + stats.flist_buildtime = (int64)(end_tv.tv_sec - start_tv.tv_sec) * 1000 + + (end_tv.tv_usec - start_tv.tv_usec) / 1000; + if (stats.flist_buildtime == 0) + stats.flist_buildtime = 1; + start_tv = end_tv; + + send_file_entry(NULL, f, 0); + + if (show_filelist_p()) + finish_filelist_progress(flist); + + gettimeofday(&end_tv, NULL); + stats.flist_xfertime = (int64)(end_tv.tv_sec - start_tv.tv_sec) * 1000 + + (end_tv.tv_usec - start_tv.tv_usec) / 1000; - if (show_filelist_p()) - finish_filelist_progress(flist); + if (flist->hlink_pool) { + pool_destroy(flist->hlink_pool); + flist->hlink_pool = NULL; } clean_flist(flist, 0, 0); - if (f != -1) { - /* Now send the uid/gid list. This was introduced in - * protocol version 15 */ - send_uid_list(f); + /* Now send the uid/gid list. This was introduced in + * protocol version 15 */ + send_uid_list(f); - /* send the io_error flag */ - write_int(f, lp_ignore_errors(module_id) ? 0 : io_error); + /* send the io_error flag */ + write_int(f, lp_ignore_errors(module_id) ? 0 : io_error); - io_end_buffering(); - stats.flist_size = stats.total_written - start_write; - stats.num_files = flist->count; - if (write_batch) - write_batch_flist_info(flist->count, flist->files); - } + io_end_buffering(); + stats.flist_size = stats.total_written - start_write; + stats.num_files = flist->count; + + if (verbose > 3) + output_flist(flist, who_am_i()); if (verbose > 2) rprintf(FINFO, "send_file_list done\n"); @@ -1157,16 +1264,14 @@ struct file_list *recv_file_list(int f) struct file_list *flist; unsigned short flags; int64 start_read; - extern int list_only; if (show_filelist_p()) start_filelist_progress("receiving file list"); start_read = stats.total_read; - flist = new(struct file_list); - if (!flist) - goto oom; + flist = flist_new(WITH_HLINK, "recv_file_list"); + received_flist = flist; flist->count = 0; flist->malloced = 1000; @@ -1182,7 +1287,7 @@ struct file_list *recv_file_list(int f) if (protocol_version >= 28 && (flags & XMIT_EXTENDED_FLAGS)) flags |= read_byte(f) << 8; - receive_file_entry(&flist->files[i], flags, f); + receive_file_entry(flist, i, flags, f); if (S_ISREG(flist->files[i]->mode)) stats.total_size += flist->files[i]->length; @@ -1193,10 +1298,10 @@ struct file_list *recv_file_list(int f) if (verbose > 2) { rprintf(FINFO, "recv_file_name(%s)\n", - f_name(flist->files[i])); + safe_fname(f_name(flist->files[i]))); } } - receive_file_entry(NULL, 0, 0); /* Signal that we're done. */ + receive_file_entry(NULL, 0, 0, 0); /* Signal that we're done. */ if (verbose > 2) rprintf(FINFO, "received %d names\n", flist->count); @@ -1211,15 +1316,16 @@ struct file_list *recv_file_list(int f) * protocol version 15 */ recv_uid_list(f, flist); - if (!read_batch) { - /* Recv the io_error flag */ - if (lp_ignore_errors(module_id) || ignore_errors) - read_int(f); - else - io_error |= read_int(f); - } + /* Recv the io_error flag */ + if (lp_ignore_errors(module_id) || ignore_errors) + read_int(f); + else + io_error |= read_int(f); } + if (verbose > 3) + output_flist(flist, who_am_i()); + if (list_only) { int i; for (i = 0; i < flist->count; i++) @@ -1234,92 +1340,77 @@ struct file_list *recv_file_list(int f) return flist; - oom: +oom: out_of_memory("recv_file_list"); return NULL; /* not reached */ } -int file_compare(struct file_struct **file1, struct file_struct **file2) +static int file_compare(struct file_struct **file1, struct file_struct **file2) { - struct file_struct *f1 = *file1; - struct file_struct *f2 = *file2; - - if (!f1->basename && !f2->basename) - return 0; - if (!f1->basename) - return -1; - if (!f2->basename) - return 1; - if (f1->dirname == f2->dirname) - return u_strcmp(f1->basename, f2->basename); - return f_name_cmp(f1, f2); + return f_name_cmp(*file1, *file2); } int flist_find(struct file_list *flist, struct file_struct *f) { - int low = 0, high = flist->count - 1; - - while (high >= 0 && !flist->files[high]->basename) high--; - - if (high < 0) - return -1; - - while (low != high) { - int mid = (low + high) / 2; - int ret = file_compare(&flist->files[flist_up(flist, mid)],&f); + int low = flist->low, high = flist->high; + int ret, mid, mid_up; + + while (low <= high) { + mid = (low + high) / 2; + for (mid_up = mid; !flist->files[mid_up]->basename; mid_up++) {} + if (mid_up <= high) + ret = f_name_cmp(flist->files[mid_up], f); + else + ret = 1; if (ret == 0) - return flist_up(flist, mid); + return mid_up; if (ret > 0) - high = mid; + high = mid - 1; else - low = mid + 1; + low = mid_up + 1; } - - if (file_compare(&flist->files[flist_up(flist, low)], &f) == 0) - return flist_up(flist, low); return -1; } - /* - * free up one file + * Free up any resources a file_struct has allocated + * and clear the file. */ -void free_file(struct file_struct *file) +void clear_file(int i, struct file_list *flist) { - if (!file) - return; - if (file->basename) - free(file->basename); - if (!IS_DEVICE(file->mode) && file->u.link) - free(file->u.link); /* Handles u.sum too. */ - if (file->link_u.idev) - free((char*)file->link_u.idev); /* Handles link_u.links too. */ - *file = null_file; + if (flist->hlink_pool && flist->files[i]->link_u.idev) + pool_free(flist->hlink_pool, 0, flist->files[i]->link_u.idev); + memset(flist->files[i], 0, file_struct_len); } /* * allocate a new file list */ -struct file_list *flist_new(void) +struct file_list *flist_new(int with_hlink, char *msg) { struct file_list *flist; flist = new(struct file_list); if (!flist) - out_of_memory("send_file_list"); + out_of_memory(msg); - flist->count = 0; - flist->malloced = 0; - flist->files = NULL; + memset(flist, 0, sizeof (struct file_list)); -#if ARENA_SIZE > 0 - flist->string_area = string_area_new(0); -#else - flist->string_area = NULL; + if (!(flist->file_pool = pool_create(FILE_EXTENT, 0, + out_of_memory, POOL_INTERN))) + out_of_memory(msg); + +#if SUPPORT_HARD_LINKS + if (with_hlink && preserve_hard_links) { + if (!(flist->hlink_pool = pool_create(HLINK_EXTENT, + sizeof (struct idev), out_of_memory, POOL_INTERN))) + out_of_memory(msg); + } #endif + return flist; } @@ -1328,22 +1419,9 @@ struct file_list *flist_new(void) */ void flist_free(struct file_list *flist) { - int i; - for (i = 1; i < flist->count; i++) { - if (!flist->string_area) - free_file(flist->files[i]); - free(flist->files[i]); - } - /* FIXME: I don't think we generally need to blank the flist - * since it's about to be freed. This will just cause more - * memory traffic. If you want a freed-memory debugger, you - * know where to get it. */ - memset((char *) flist->files, 0, - sizeof(flist->files[0]) * flist->count); + pool_destroy(flist->file_pool); + pool_destroy(flist->hlink_pool); free(flist->files); - if (flist->string_area) - string_area_free(flist->string_area); - memset((char *) flist, 0, sizeof(*flist)); free(flist); } @@ -1360,7 +1438,7 @@ static void clean_flist(struct file_list *flist, int strip_root, int no_dups) return; qsort(flist->files, flist->count, - sizeof(flist->files[0]), (int (*)()) file_compare); + sizeof flist->files[0], (int (*)())file_compare); for (i = no_dups? 0 : flist->count; i < flist->count; i++) { if (flist->files[i]->basename) { @@ -1368,6 +1446,7 @@ static void clean_flist(struct file_list *flist, int strip_root, int no_dups) break; } } + flist->low = prev_i; while (++i < flist->count) { if (!flist->files[i]->basename) continue; @@ -1375,24 +1454,18 @@ static void clean_flist(struct file_list *flist, int strip_root, int no_dups) if (verbose > 1 && !am_server) { rprintf(FINFO, "removing duplicate name %s from file list %d\n", - f_name(flist->files[i]), i); + safe_fname(f_name(flist->files[i])), i); } /* Make sure that if we unduplicate '.', that we don't - * lose track of a user-specified starting point (or - * else deletions will mysteriously fail with -R). */ + * lose track of a user-specified top directory. */ if (flist->files[i]->flags & FLAG_TOP_DIR) flist->files[prev_i]->flags |= FLAG_TOP_DIR; - /* it's not great that the flist knows the semantics of - * the file memory usage, but i'd rather not add a flag - * byte to that struct. - * XXX can i use a bit in the flags field? */ - if (flist->string_area) - flist->files[i][0] = null_file; - else - free_file(flist->files[i]); + + clear_file(i, flist); } else prev_i = i; } + flist->high = prev_i; if (strip_root) { /* we need to strip off the root directory in the case @@ -1412,28 +1485,49 @@ static void clean_flist(struct file_list *flist, int strip_root, int no_dups) } } } +} - if (verbose <= 3) - return; +static void output_flist(struct file_list *flist, const char *whose_list) +{ + char uidbuf[16], gidbuf[16], depthbuf[16]; + struct file_struct *file; + int i; for (i = 0; i < flist->count; i++) { - rprintf(FINFO, "[%s] i=%d %s %s %s mode=0%o len=%.0f\n", - who_am_i(), i, - NS(flist->files[i]->basedir), - NS(flist->files[i]->dirname), - NS(flist->files[i]->basename), - (int) flist->files[i]->mode, - (double) flist->files[i]->length); + file = flist->files[i]; + if ((am_root || am_sender) && preserve_uid) + sprintf(uidbuf, " uid=%ld", (long)file->uid); + else + *uidbuf = '\0'; + if (preserve_gid && file->gid != GID_NONE) + sprintf(gidbuf, " gid=%ld", (long)file->gid); + else + *gidbuf = '\0'; + if (!am_sender) + sprintf(depthbuf, "%d", file->dir.depth); + rprintf(FINFO, "[%s] i=%d %s %s%s%s%s mode=0%o len=%.0f%s%s flags=%x\n", + whose_list, i, am_sender ? NS(file->dir.root) : depthbuf, + file->dirname ? safe_fname(file->dirname) : "", + file->dirname ? "/" : "", NS(file->basename), + S_ISDIR(file->mode) ? "/" : "", (int)file->mode, + (double)file->length, uidbuf, gidbuf, file->flags); } } -enum fnc_state { fnc_DIR, fnc_SLASH, fnc_BASE }; +enum fnc_state { fnc_DIR, fnc_SLASH, fnc_BASE, fnc_TRAILING }; -/* Compare the names of two file_struct entities, just like strcmp() - * would do if it were operating on the joined strings. We assume - * that there are no 0-length strings. - */ +/* Compare the names of two file_struct entities, similar to how strcmp() + * would do if it were operating on the joined strings. The only difference + * is that, beginning with protocol_version 29, a directory name will always + * sort immediately prior to its contents (previously "foo." would sort in + * between directory "foo" and "foo/bar"). We do this by assuming that a dir + * has a trailing slash for comparison purposes, but only if we aren't about + * to match a file of the same name (because we need all identically named + * items to match each other). The dirname component can be an empty string, + * but the basename component cannot (and never is in the current codebase). + * The basename component may be NULL, in which case it is sorted to the end + * of the list (as a removed item). */ int f_name_cmp(struct file_struct *f1, struct file_struct *f2) { int dif; @@ -1448,14 +1542,24 @@ int f_name_cmp(struct file_struct *f1, struct file_struct *f2) if (!f2 || !f2->basename) return 1; - if (!(c1 = (uchar*)f1->dirname)) { + c1 = (uchar*)f1->dirname; + c2 = (uchar*)f2->dirname; + if (c1 == c2) + c1 = c2 = NULL; + if (!c1) { state1 = fnc_BASE; c1 = (uchar*)f1->basename; + } else if (!*c1) { + state1 = fnc_SLASH; + c1 = (uchar*)"/"; } else state1 = fnc_DIR; - if (!(c2 = (uchar*)f2->dirname)) { + if (!c2) { state2 = fnc_BASE; c2 = (uchar*)f2->basename; + } else if (!*c2) { + state2 = fnc_SLASH; + c2 = (uchar*)"/"; } else state2 = fnc_DIR; @@ -1473,6 +1577,11 @@ int f_name_cmp(struct file_struct *f1, struct file_struct *f2) c1 = (uchar*)f1->basename; break; case fnc_BASE: + state1 = fnc_TRAILING; + if (protocol_version >= 29 && S_ISDIR(f1->mode)) + c1 = (uchar*)"/"; + break; + case fnc_TRAILING: break; } } @@ -1487,8 +1596,13 @@ int f_name_cmp(struct file_struct *f1, struct file_struct *f2) c2 = (uchar*)f2->basename; break; case fnc_BASE: - if (!*c1) + if (state1 == fnc_TRAILING) return 0; + state2 = fnc_TRAILING; + if (protocol_version >= 29 && S_ISDIR(f2->mode)) + c2 = (uchar*)"/"; + break; + case fnc_TRAILING: break; } } @@ -1518,8 +1632,7 @@ char *f_name_to(struct file_struct *f, char *fbuf) } -/* Like f_name_to(), but we rotate through 5 static buffers of our own. - */ +/* Like f_name_to(), but we rotate through 5 static buffers of our own. */ char *f_name(struct file_struct *f) { static char names[5][MAXPATHLEN]; @@ -1529,3 +1642,113 @@ char *f_name(struct file_struct *f) return f_name_to(f, names[n]); } + + +static int is_backup_file(char *fn) +{ + int k = strlen(fn) - backup_suffix_len; + return k > 0 && strcmp(fn+k, backup_suffix) == 0; +} + + +/* This function is used to implement per-directory deletion, and + * is used by all the --delete-WHEN options. Note that the fbuf + * pointer must point to a MAXPATHLEN buffer with the name of the + * directory in it (the functions we call will append names onto + * the end, but the old dir value will be restored on exit). */ +void delete_in_dir(struct file_list *flist, char *fbuf, + struct file_struct *file) +{ + static int min_depth = MAXPATHLEN, cur_depth = -1; + static void *filt_array[MAXPATHLEN/2+1]; + struct file_list *dir_list; + STRUCT_STAT st; + int dlen; + + if (!flist) { + while (cur_depth >= min_depth) + pop_local_filters(filt_array[cur_depth--]); + min_depth = MAXPATHLEN; + cur_depth = -1; + return; + } + if (file->dir.depth >= MAXPATHLEN/2+1) + return; /* Impossible... */ + + if (max_delete && deletion_count >= max_delete) + return; + + if (io_error && !(lp_ignore_errors(module_id) || ignore_errors)) { + rprintf(FINFO, + "IO error encountered -- skipping file deletion\n"); + max_delete = -1; /* avoid duplicating the above warning */ + return; + } + + while (cur_depth >= file->dir.depth && cur_depth >= min_depth) + pop_local_filters(filt_array[cur_depth--]); + cur_depth = file->dir.depth; + if (min_depth > cur_depth) + min_depth = cur_depth; + dlen = strlen(fbuf); + filt_array[cur_depth] = push_local_filters(fbuf, dlen); + + if (link_stat(fbuf, &st, keep_dirlinks) < 0) + return; + + if (one_file_system && file->flags & FLAG_TOP_DIR) + filesystem_dev = st.st_dev; + + dir_list = flist_new(WITHOUT_HLINK, "delete_in_dir"); + + recurse = 0; + send_directory(-1, dir_list, fbuf, dlen); + recurse = -1; + fbuf[dlen] = '\0'; + + if (verbose > 3) + output_flist(dir_list, "delete"); + + delete_missing(flist, dir_list, fbuf); + + flist_free(dir_list); +} + + +/* If an item in dir_list is not found in full_list, delete it from the + * filesystem. */ +void delete_missing(struct file_list *full_list, struct file_list *dir_list, + const char *dirname) +{ + int i, j, mode; + + if (max_delete && deletion_count >= max_delete) + return; + + if (verbose > 2) + rprintf(FINFO, "delete_missing(%s)\n", safe_fname(dirname)); + + for (i = dir_list->count; i--; ) { + if (!dir_list->files[i]->basename) + continue; + mode = dir_list->files[i]->mode; + if ((j = flist_find(full_list, dir_list->files[i])) < 0 + || (S_ISDIR(mode) && !S_ISDIR(full_list->files[j]->mode))) { + char *f = f_name(dir_list->files[i]); + if (make_backups && (backup_dir || !is_backup_file(f)) + && !S_ISDIR(mode)) { + make_backup(f); + if (verbose) { + rprintf(FINFO, "deleting %s\n", + safe_fname(f)); + } + } else if (S_ISDIR(mode)) + delete_file(f, DEL_DIR | DEL_FORCE_RECURSE); + else + delete_file(f, 0); + deletion_count++; + if (max_delete && deletion_count >= max_delete) + break; + } + } +}