X-Git-Url: https://mattmccutchen.net/rsync/rsync.git/blobdiff_plain/217cc3b0459e0c4d3a01c2d6808750b94cba5a17..8a6f3fead8667434ef08b3c1de2e3c899f1d70ee:/flist.c diff --git a/flist.c b/flist.c index c513e5a2..8d4defb9 100644 --- a/flist.c +++ b/flist.c @@ -35,7 +35,6 @@ extern int am_root; extern int am_server; extern int am_daemon; extern int am_sender; -extern int delete_during; extern int always_checksum; extern int module_id; extern int ignore_errors; @@ -65,7 +64,6 @@ extern int copy_links; extern int copy_unsafe_links; extern int protocol_version; extern int sanitize_paths; -extern int delete_excluded; extern int max_delete; extern int orig_umask; extern int list_only; @@ -78,6 +76,8 @@ int io_error; 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); @@ -150,14 +150,14 @@ static void list_file_entry(struct file_struct *f) rprintf(FINFO, "%s %11.0f %s %s -> %s\n", perms, (double)f->length, timestring(f->modtime), - f_name(f), f->u.link); + 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)); + safe_fname(f_name(f))); } } @@ -191,7 +191,7 @@ static int readlink_stat(const char *path, STRUCT_STAT *buffer, char *linkbuf) 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); } @@ -251,18 +251,6 @@ static int is_excluded(char *fname, int is_dir, int filter_level) return 0; } -/* used by the one_file_system code */ -static dev_t filesystem_dev; - -static void set_filesystem(char *fname) -{ - STRUCT_STAT st; - if (do_stat(fname, &st) != 0) - return; - filesystem_dev = st.st_dev; -} - - static int to_wire_mode(mode_t mode) { #if SUPPORT_LINKS @@ -280,7 +268,8 @@ static mode_t from_wire_mode(int 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; @@ -419,11 +408,11 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) /* 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_DEL_START on a non-dir has no meaning, so + * 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 && !S_ISDIR(mode)) - flags |= XMIT_DEL_START; + flags |= XMIT_TOP_DIR; if ((flags & 0xFF00) || !flags) { flags |= XMIT_EXTENDED_FLAGS; write_byte(f, flags); @@ -432,7 +421,7 @@ void send_file_entry(struct file_struct *file, int f, unsigned short base_flags) write_byte(f, flags); } else { if (!(flags & 0xFF) && !S_ISDIR(mode)) - flags |= XMIT_DEL_START; + flags |= XMIT_TOP_DIR; if (!(flags & 0xFF)) flags |= XMIT_LONG_NAME; write_byte(f, flags); @@ -519,8 +508,8 @@ 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, - struct file_list *flist, 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; @@ -531,7 +520,7 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, static gid_t gid; static char lastname[MAXPATHLEN], *lastdir; static int lastdir_depth, lastdir_len = -1; - static unsigned int del_heir_name_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; @@ -540,13 +529,13 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, char *basename, *dirname, *bp; struct file_struct *file; - if (!fptr) { + if (!flist) { modtime = 0, mode = 0; dev = 0, rdev = makedev(0, 0); rdev_major = 0; uid = 0, gid = 0; *lastname = '\0'; - del_heir_name_len = lastdir_len = -1; + lastdir_len = -1; in_del_hier = 0; return; } @@ -562,7 +551,7 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, 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"); } @@ -641,7 +630,7 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, + linkname_len + sum_len; bp = pool_alloc(flist->file_pool, alloc_len, "receive_file_entry"); - file = *fptr = (struct file_struct *)bp; + file = flist->files[ndx] = (struct file_struct *)bp; memset(bp, 0, file_struct_len); bp += file_struct_len; @@ -652,30 +641,36 @@ void receive_file_entry(struct file_struct **fptr, unsigned short flags, file->uid = uid; file->gid = gid; - if (S_ISDIR(mode)) { - if (flags & XMIT_DEL_START) { - in_del_hier = 1; - del_heir_name_len = l1 + l2; - file->flags |= FLAG_DEL_START; - } else if (delete_during && in_del_hier) { - if (!relative_paths || (l1 >= del_heir_name_len - && thisname[del_heir_name_len] == '/')) - file->flags |= FLAG_DEL_START; - else - in_del_hier = 0; - } - } - if (dirname_len) { file->dirname = lastdir = bp; lastdir_len = dirname_len - 1; memcpy(bp, dirname, dirname_len - 1); bp += dirname_len; bp[-1] = '\0'; - if (sanitize_paths) - lastdir_depth = count_dir_elements(lastdir); - } else if (dirname) - file->dirname = dirname; + 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); @@ -774,7 +769,8 @@ struct file_struct *make_file(char *fname, struct file_list *flist, if (strlcpy(thisname, fname, sizeof thisname) >= sizeof thisname - flist_dir_len) { - rprintf(FINFO, "skipping overly long name: %s\n", fname); + rprintf(FINFO, "skipping overly long name: %s\n", + safe_fname(fname)); return NULL; } clean_fname(thisname, 0); @@ -819,7 +815,7 @@ struct file_struct *make_file(char *fname, struct file_list *flist, goto skip_filters; if (S_ISDIR(st.st_mode) && !xfer_dirs) { - rprintf(FINFO, "skipping directory %s\n", thisname); + rprintf(FINFO, "skipping directory %s\n", safe_fname(thisname)); return NULL; } @@ -845,7 +841,7 @@ skip_filters: if (verbose > 2) { rprintf(FINFO, "[%s] make_file(%s,*,%d)\n", - who_am_i(), thisname, filter_level); + who_am_i(), safe_fname(thisname), filter_level); } if ((basename = strrchr(thisname, '/')) != NULL) { @@ -925,7 +921,7 @@ skip_filters: memcpy(bp, basename, basename_len); bp += basename_len; -#ifdef HAVE_STRUCT_STAT_ST_RDEV +#if HAVE_STRUCT_STAT_ST_RDEV if (preserve_devices && IS_DEVICE(st.st_mode)) file->u.rdev = st.st_rdev; #endif @@ -944,7 +940,7 @@ skip_filters: /*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. */ @@ -979,11 +975,7 @@ void send_file_name(int f, struct file_list *flist, char *fname, struct file_struct *file; char fbuf[MAXPATHLEN]; - /* f is set to -1 when calculating deletion file list */ - file = make_file(fname, flist, - f == -1 && delete_excluded? SERVER_FILTERS : ALL_FILTERS); - - if (!file) + if (!(file = make_file(fname, flist, ALL_FILTERS))) return; maybe_emit_filelist_progress(flist); @@ -996,80 +988,72 @@ void send_file_name(int f, struct file_list *flist, char *fname, } if (recursive && S_ISDIR(file->mode) - && !(file->flags & FLAG_MOUNT_POINT)) { - send_directory(f, flist, f_name_to(file, fbuf)); + && !(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); } } /* Note that the "recurse" value either contains -1, for infinite recursion, - * or a number >= 0 indicating how many levels of recursion we will allow. */ -static void send_directory(int f, struct file_list *flist, char *dir) + * 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; - void *save_filters; char *p; + DIR *d; - d = opendir(dir); - if (!d) { + if (!(d = opendir(fbuf))) { io_error |= IOERR_GENERAL; - rsyserr(FERROR, errno, "opendir %s failed", full_fname(dir)); + 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++; - } - - save_filters = push_local_filters(fname, offset); + *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) { + if (strlcpy(p, dname, MAXPATHLEN - len) < MAXPATHLEN - len) { int do_subdirs = recurse >= 1 ? recurse-- : recurse; - send_file_name(f, flist, fname, do_subdirs, 0); + 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; - rsyserr(FERROR, errno, "readdir(%s)", dir); + *p = '\0'; + rsyserr(FERROR, errno, "readdir(%s)", full_fname(fbuf)); } - pop_local_filters(save_filters); - closedir(d); } -/* This function is normally called by the sender, but the receiving side - * also uses it to construct one or more file lists if one of the --delete - * options have been specified. The delete_in_dir() function 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) and it avoids some per-arg init code for limited recursion (since - * delete_in_dir() sets recurse before calling this function). */ struct file_list *send_file_list(int f, int argc, char *argv[]) { int l; @@ -1077,32 +1061,26 @@ 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; - BOOL need_first_push = True; + 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(f == -1 ? WITHOUT_HLINK : WITH_HLINK, - "send_file_list"); + flist = flist_new(WITH_HLINK, "send_file_list"); - if (f != -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; - if (curr_dir_len < MAXPATHLEN - 1) { - push_local_filters(curr_dir, curr_dir_len); - need_first_push = False; - } + 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) { @@ -1132,34 +1110,22 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) fname[l] = '\0'; } } - if (f == -1) - ; /* recurse is pre-set */ - else if (fname[l-1] == '.' && (l == 1 || fname[l-2] == '/')) { + 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 (need_first_push) { - if ((p = strrchr(fname, '/')) != NULL) { - if (*++p && strcmp(p, ".") != 0) - push_local_filters(fname, p - fname); - } else if (strcmp(fname, ".") != 0) - push_local_filters(fname, 0); - need_first_push = False; - } - if (link_stat(fname, &st, keep_dirlinks) != 0) { - if (f != -1) { - io_error |= IOERR_GENERAL; - rsyserr(FERROR, errno, "link_stat %s failed", - full_fname(fname)); - } + io_error |= IOERR_GENERAL; + rsyserr(FERROR, errno, "link_stat %s failed", + full_fname(fname)); continue; } if (S_ISDIR(st.st_mode) && !xfer_dirs) { - rprintf(FINFO, "skipping directory %s\n", fname); + rprintf(FINFO, "skipping directory %s\n", + safe_fname(fname)); continue; } @@ -1176,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; @@ -1233,10 +1199,10 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) } if (one_file_system) - set_filesystem(fname); + filesystem_dev = st.st_dev; do_subdirs = recurse >= 1 ? recurse-- : recurse; - send_file_name(f, flist, fname, do_subdirs, XMIT_DEL_START); + send_file_name(f, flist, fname, do_subdirs, XMIT_TOP_DIR); if (olddir[0]) { flist_dir = NULL; @@ -1249,12 +1215,21 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) } } - 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; - if (show_filelist_p()) - finish_filelist_progress(flist); - } + 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 (flist->hlink_pool) { pool_destroy(flist->hlink_pool); @@ -1263,21 +1238,19 @@ struct file_list *send_file_list(int f, int argc, char *argv[]) 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; - } + io_end_buffering(); + stats.flist_size = stats.total_written - start_write; + stats.num_files = flist->count; if (verbose > 3) - output_flist(flist, f < 0 ? "delete" : who_am_i()); + output_flist(flist, who_am_i()); if (verbose > 2) rprintf(FINFO, "send_file_list done\n"); @@ -1314,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, flist, f); + receive_file_entry(flist, i, flags, f); if (S_ISREG(flist->files[i]->mode)) stats.total_size += flist->files[i]->length; @@ -1325,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, NULL, 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); @@ -1373,45 +1346,31 @@ oom: } -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; } @@ -1487,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; @@ -1494,18 +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). */ - if (flist->files[i]->flags & FLAG_DEL_START) - flist->files[prev_i]->flags |= FLAG_DEL_START; + * lose track of a user-specified top directory. */ + if (flist->files[i]->flags & FLAG_TOP_DIR) + flist->files[prev_i]->flags |= FLAG_TOP_DIR; 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 @@ -1529,7 +1489,7 @@ 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) { - char uidbuf[16], gidbuf[16]; + char uidbuf[16], gidbuf[16], depthbuf[16]; struct file_struct *file; int i; @@ -1543,20 +1503,31 @@ static void output_flist(struct file_list *flist, const char *whose_list) sprintf(gidbuf, " gid=%ld", (long)file->gid); else *gidbuf = '\0'; - rprintf(FINFO, "[%s] i=%d %s %s %s mode=0%o len=%.0f%s%s\n", - whose_list, i, NS(file->basedir), NS(file->dirname), - NS(file->basename), (int)file->mode, - (double)file->length, uidbuf, gidbuf); + 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; @@ -1571,7 +1542,11 @@ 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) { @@ -1579,7 +1554,7 @@ int f_name_cmp(struct file_struct *f1, struct file_struct *f2) c1 = (uchar*)"/"; } else state1 = fnc_DIR; - if (!(c2 = (uchar*)f2->dirname)) { + if (!c2) { state2 = fnc_BASE; c2 = (uchar*)f2->basename; } else if (!*c2) { @@ -1602,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; } } @@ -1616,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; } } @@ -1647,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]; @@ -1659,63 +1643,98 @@ 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; } -void delete_in_dir(struct file_list *flist, char *fname) + +/* 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 deletion_count = 0; - struct file_list *del_flist; - int save_recurse = recurse; - int save_xfer_dirs = xfer_dirs; - int save_implied_dirs = implied_dirs; - int save_relative_paths = relative_paths; - char *argv[1]; - int i, j, mode; + 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"); + rprintf(FINFO, + "IO error encountered -- skipping file deletion\n"); max_delete = -1; /* avoid duplicating the above warning */ return; } - if (delete_during) { - recurse = 1; /* allow one level only */ - xfer_dirs = 1; - implied_dirs = 0; - relative_paths = 1; - } + 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; - argv[0] = fname; - del_flist = send_file_list(-1, 1, argv); + 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"); - relative_paths = save_relative_paths; - implied_dirs = save_implied_dirs; - xfer_dirs = save_xfer_dirs; - recurse = save_recurse; + delete_missing(flist, dir_list, fbuf); - if (!del_flist) + 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 > 1) - rprintf(FINFO, "deleting in %s\n", safe_fname(fname)); + if (verbose > 2) + rprintf(FINFO, "delete_missing(%s)\n", safe_fname(dirname)); - for (i = del_flist->count-1; i >= 0; i--) { - if (max_delete && deletion_count >= max_delete) - break; - if (!del_flist->files[i]->basename) + for (i = dir_list->count; i--; ) { + if (!dir_list->files[i]->basename) continue; - mode = del_flist->files[i]->mode; - if ((j = flist_find(flist, del_flist->files[i])) < 0 - || (delete_during && S_ISDIR(mode) - && !S_ISDIR(flist->files[j]->mode))) { - char *f = f_name(del_flist->files[i]); + 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); @@ -1723,14 +1742,13 @@ void delete_in_dir(struct file_list *flist, char *fname) rprintf(FINFO, "deleting %s\n", safe_fname(f)); } - } else { - int dflags = delete_during - ? DEL_DIR | DEL_FORCE_RECURSE - : DEL_DIR | DEL_NO_RECURSE; - delete_file(f, S_ISDIR(mode) ? dflags : 0); - } + } 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; } } - flist_free(del_flist); }