Cosmetic changes to file_compare() for readability and to
[rsync/rsync.git] / flist.c
diff --git a/flist.c b/flist.c
index d50a947..a14ce5f 100644 (file)
--- a/flist.c
+++ b/flist.c
@@ -107,9 +107,8 @@ static void finish_filelist_progress(const struct file_list *flist)
                /* This overwrites the progress line */
                rprintf(FINFO, "%d file%sto consider\n",
                        flist->count, flist->count == 1 ? " " : "s ");
-       } else {
+       } else
                rprintf(FINFO, "done\n");
-       }
 }
 
 void show_flist_stats(void)
@@ -210,18 +209,15 @@ 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)
+int readlink_stat(const char *path, STRUCT_STAT *buffer, char *linkbuf)
 {
 #if SUPPORT_LINKS
-       if (copy_links) {
+       if (copy_links)
                return do_stat(path, buffer);
-       }
-       if (do_lstat(path, buffer) == -1) {
+       if (do_lstat(path, buffer) == -1)
                return -1;
-       }
        if (S_ISLNK(buffer->st_mode)) {
-               int l;
-               l = readlink((char *) path, linkbuf, MAXPATHLEN - 1);
+               int l = readlink((char *) path, linkbuf, MAXPATHLEN - 1);
                if (l == -1)
                        return -1;
                linkbuf[l] = 0;
@@ -242,11 +238,9 @@ int readlink_stat(const char *path, STRUCT_STAT * buffer, char *linkbuf)
 int link_stat(const char *path, STRUCT_STAT * buffer)
 {
 #if SUPPORT_LINKS
-       if (copy_links) {
+       if (copy_links)
                return do_stat(path, buffer);
-       } else {
-               return do_lstat(path, buffer);
-       }
+       return do_lstat(path, buffer);
 #else
        return do_stat(path, buffer);
 #endif
@@ -301,17 +295,15 @@ static void set_filesystem(char *fname)
 
 static int to_wire_mode(mode_t mode)
 {
-       if (S_ISLNK(mode) && (_S_IFLNK != 0120000)) {
+       if (S_ISLNK(mode) && (_S_IFLNK != 0120000))
                return (mode & ~(_S_IFMT)) | 0120000;
-       }
        return (int) mode;
 }
 
 static mode_t from_wire_mode(int mode)
 {
-       if ((mode & (_S_IFMT)) == 0120000 && (_S_IFLNK != 0120000)) {
+       if ((mode & (_S_IFMT)) == 0120000 && (_S_IFLNK != 0120000))
                return (mode & ~(_S_IFMT)) | _S_IFLNK;
-       }
        return (mode_t) mode;
 }
 
@@ -369,7 +361,7 @@ static void send_file_entry(struct file_struct *file, int f,
        static uid_t last_uid;
        static gid_t last_gid;
        static char lastname[MAXPATHLEN];
-       char *fname;
+       char *fname, fbuf[MAXPATHLEN];
        int l1, l2;
 
        if (f == -1)
@@ -382,7 +374,7 @@ static void send_file_entry(struct file_struct *file, int f,
 
        io_write_phase = "send_file_entry";
 
-       fname = f_name(file);
+       fname = f_name_to(file, fbuf, sizeof fbuf);
 
        flags = base_flags;
 
@@ -462,11 +454,10 @@ static void send_file_entry(struct file_struct *file, int f,
 #endif
 
        if (always_checksum) {
-               if (protocol_version < 21) {
+               if (protocol_version < 21)
                        write_buf(f, file->sum, 2);
-               } else {
+               else
                        write_buf(f, file->sum, MD4_SUM_LENGTH);
-               }
        }
 
        last_mode = file->mode;
@@ -534,9 +525,9 @@ static void receive_file_entry(struct file_struct **fptr,
        if ((p = strrchr(thisname, '/'))) {
                static char *lastdir;
                *p = 0;
-               if (lastdir && strcmp(thisname, lastdir) == 0) {
+               if (lastdir && strcmp(thisname, lastdir) == 0)
                        file->dirname = lastdir;
-               else {
+               else {
                        file->dirname = strdup(thisname);
                        lastdir = file->dirname;
                }
@@ -549,22 +540,19 @@ static void receive_file_entry(struct file_struct **fptr,
        if (!file->basename)
                out_of_memory("receive_file_entry 1");
 
-
        file->flags = flags;
        file->length = read_longint(f);
-       file->modtime =
-           (flags & SAME_TIME) ? last_time : (time_t) read_int(f);
-       file->mode =
-           (flags & SAME_MODE) ? last_mode : from_wire_mode(read_int(f));
+       file->modtime = (flags & SAME_TIME) ? last_time : (time_t)read_int(f);
+       file->mode = (flags & SAME_MODE) ? last_mode
+                                        : from_wire_mode(read_int(f));
        if (preserve_uid)
-               file->uid =
-                   (flags & SAME_UID) ? last_uid : (uid_t) read_int(f);
+               file->uid = (flags & SAME_UID) ? last_uid : (uid_t)read_int(f);
        if (preserve_gid)
-               file->gid =
-                   (flags & SAME_GID) ? last_gid : (gid_t) read_int(f);
-       if (preserve_devices && IS_DEVICE(file->mode))
-               file->rdev =
-                   (flags & SAME_RDEV) ? last_rdev : (DEV64_T) read_int(f);
+               file->gid = (flags & SAME_GID) ? last_gid : (gid_t)read_int(f);
+       if (preserve_devices && IS_DEVICE(file->mode)) {
+               file->rdev = (flags & SAME_RDEV) ? last_rdev
+                                                : (DEV64_T)read_int(f);
+       }
 
        if (preserve_links && S_ISLNK(file->mode)) {
                int l = read_int(f);
@@ -576,9 +564,8 @@ static void receive_file_entry(struct file_struct **fptr,
                if (!file->link)
                        out_of_memory("receive_file_entry 2");
                read_sbuf(f, file->link, l);
-               if (sanitize_paths) {
+               if (sanitize_paths)
                        sanitize_path(file->link, file->dirname);
-               }
        }
 #if SUPPORT_HARD_LINKS
        if (preserve_hard_links && S_ISREG(file->mode)) {
@@ -596,11 +583,10 @@ static void receive_file_entry(struct file_struct **fptr,
                file->sum = new_array(char, MD4_SUM_LENGTH);
                if (!file->sum)
                        out_of_memory("md4 sum");
-               if (protocol_version < 21) {
+               if (protocol_version < 21)
                        read_buf(f, file->sum, 2);
-               } else {
+               else
                        read_buf(f, file->sum, MD4_SUM_LENGTH);
-               }
        }
 
        last_mode = file->mode;
@@ -677,9 +663,8 @@ struct file_struct *make_file(char *fname, struct string_area **ap,
        strlcpy(cleaned_name, fname, MAXPATHLEN);
        cleaned_name[MAXPATHLEN - 1] = 0;
        clean_fname(cleaned_name);
-       if (sanitize_paths) {
+       if (sanitize_paths)
                sanitize_path(cleaned_name, NULL);
-       }
        fname = cleaned_name;
 
        memset(sum, 0, SUM_LENGTH);
@@ -734,9 +719,9 @@ struct file_struct *make_file(char *fname, struct string_area **ap,
        if ((p = strrchr(fname, '/'))) {
                static char *lastdir;
                *p = 0;
-               if (lastdir && strcmp(fname, lastdir) == 0) {
+               if (lastdir && strcmp(fname, lastdir) == 0)
                        file->dirname = lastdir;
-               else {
+               else {
                        file->dirname = strdup(fname);
                        lastdir = file->dirname;
                }
@@ -759,9 +744,8 @@ struct file_struct *make_file(char *fname, struct string_area **ap,
 #endif
 
 #if SUPPORT_LINKS
-       if (S_ISLNK(st.st_mode)) {
+       if (S_ISLNK(st.st_mode))
                file->link = STRDUP(ap, linkbuf);
-       }
 #endif
 
        if (always_checksum) {
@@ -780,15 +764,14 @@ struct file_struct *make_file(char *fname, struct string_area **ap,
 
        if (flist_dir) {
                static char *lastdir;
-               if (lastdir && strcmp(lastdir, flist_dir) == 0) {
+               if (lastdir && strcmp(lastdir, flist_dir) == 0)
                        file->basedir = lastdir;
-               else {
+               else {
                        file->basedir = strdup(flist_dir);
                        lastdir = file->basedir;
                }
-       } else {
+       } else
                file->basedir = NULL;
-       }
 
        if (!S_ISDIR(st.st_mode))
                stats.total_size += st.st_size;
@@ -797,11 +780,11 @@ struct file_struct *make_file(char *fname, struct string_area **ap,
 }
 
 
-
 void send_file_name(int f, struct file_list *flist, char *fname,
                    int recursive, unsigned base_flags)
 {
        struct file_struct *file;
+       char fbuf[MAXPATHLEN];
        extern int delete_excluded;
 
        /* f is set to -1 when calculating deletion file list */
@@ -816,7 +799,7 @@ void send_file_name(int f, struct file_list *flist, char *fname,
 
        flist_expand(flist);
 
-       if (write_batch)        /*  dw  */
+       if (write_batch)
                file->flags = FLAG_DELETE;
 
        if (file->basename[0]) {
@@ -827,14 +810,13 @@ void send_file_name(int f, struct file_list *flist, char *fname,
        if (S_ISDIR(file->mode) && recursive) {
                struct exclude_struct **last_exclude_list =
                    local_exclude_list;
-               send_directory(f, flist, f_name(file));
+               send_directory(f, flist, f_name_to(file, fbuf, sizeof fbuf));
                local_exclude_list = last_exclude_list;
                return;
        }
 }
 
 
-
 static void send_directory(int f, struct file_list *flist, char *dir)
 {
        DIR *d;
@@ -925,7 +907,7 @@ struct file_list *send_file_list(int f, int argc, char *argv[])
        flist = flist_new();
 
        if (f != -1) {
-               io_start_buffering(f);
+               io_start_buffering_out(f);
                if (filesfrom_fd >= 0) {
                        if (argv[0] && !push_dir(argv[0], 0)) {
                                rprintf(FERROR, "push_dir %s failed: %s\n",
@@ -1055,21 +1037,18 @@ struct file_list *send_file_list(int f, int argc, char *argv[])
                }
        }
 
-       if (f != -1) {
+       if (f != -1)
                send_file_entry(NULL, f, 0);
-       }
 
-       if (show_filelist_p() && f != -1) {
+       if (show_filelist_p() && f != -1)
                finish_filelist_progress(flist);
-       }
 
        clean_flist(flist, 0, 0);
 
        /* now send the uid/gid list. This was introduced in protocol
           version 15 */
-       if (f != -1) {
+       if (f != -1)
                send_uid_list(f);
-       }
 
        /* send the io_error flag */
        if (f != -1) {
@@ -1081,7 +1060,7 @@ struct file_list *send_file_list(int f, int argc, char *argv[])
                io_end_buffering();
                stats.flist_size = stats.total_written - start_write;
                stats.num_files = flist->count;
-               if (write_batch)        /*  dw  */
+               if (write_batch)
                        write_batch_flist_info(flist->count, flist->files);
        }
 
@@ -1129,9 +1108,10 @@ struct file_list *recv_file_list(int f)
 
                maybe_emit_filelist_progress(flist);
 
-               if (verbose > 2)
+               if (verbose > 2) {
                        rprintf(FINFO, "recv_file_name(%s)\n",
                                f_name(flist->files[i]));
+               }
        }
 
 
@@ -1140,31 +1120,27 @@ struct file_list *recv_file_list(int f)
 
        clean_flist(flist, relative_paths, 1);
 
-       if (show_filelist_p()) {
+       if (show_filelist_p())
                finish_filelist_progress(flist);
-       }
 
        /* now recv the uid/gid list. This was introduced in protocol version 15 */
-       if (f != -1) {
+       if (f != -1)
                recv_uid_list(f, flist);
-       }
 
        /* recv the io_error flag */
        if (f != -1 && !read_batch) {   /* dw-added readbatch */
                extern int module_id;
                extern int ignore_errors;
-               if (lp_ignore_errors(module_id) || ignore_errors) {
+               if (lp_ignore_errors(module_id) || ignore_errors)
                        read_int(f);
-               } else {
+               else
                        io_error |= read_int(f);
-               }
        }
 
        if (list_only) {
                int i;
-               for (i = 0; i < flist->count; i++) {
+               for (i = 0; i < flist->count; i++)
                        list_file_entry(flist->files[i]);
-               }
        }
 
 
@@ -1182,21 +1158,20 @@ struct file_list *recv_file_list(int f)
 }
 
 
-/*
- * XXX: This is currently the hottest function while building the file
- * list, because building f_name()s every time is expensive.
- **/
-int file_compare(struct file_struct **f1, struct file_struct **f2)
+int file_compare(struct file_struct **file1, struct file_struct **file2)
 {
-       if (!(*f1)->basename && !(*f2)->basename)
+       struct file_struct *f1 = *file1;
+       struct file_struct *f2 = *file2;
+
+       if (!f1->basename && !f2->basename)
                return 0;
-       if (!(*f1)->basename)
+       if (!f1->basename)
                return -1;
-       if (!(*f2)->basename)
+       if (!f2->basename)
                return 1;
-       if ((*f1)->dirname == (*f2)->dirname)
-               return u_strcmp((*f1)->basename, (*f2)->basename);
-       return u_strcmp(f_name(*f1), f_name(*f2));
+       if (f1->dirname == f2->dirname)
+               return u_strcmp(f1->basename, f2->basename);
+       return f_name_cmp(f1, f2);
 }
 
 
@@ -1211,15 +1186,13 @@ int flist_find(struct file_list *flist, struct file_struct *f)
 
        while (low != high) {
                int mid = (low + high) / 2;
-               int ret =
-                   file_compare(&flist->files[flist_up(flist, mid)], &f);
+               int ret = file_compare(&flist->files[flist_up(flist, mid)],&f);
                if (ret == 0)
                        return flist_up(flist, mid);
-               if (ret > 0) {
+               if (ret > 0)
                        high = mid;
-               } else {
+               else
                        low = mid + 1;
-               }
        }
 
        if (file_compare(&flist->files[flist_up(flist, low)], &f) == 0)
@@ -1300,7 +1273,6 @@ void flist_free(struct file_list *flist)
 static void clean_flist(struct file_list *flist, int strip_root, int no_dups)
 {
        int i, prev_i = 0;
-       char *name, *prev_name = NULL;
 
        if (!flist || flist->count == 0)
                return;
@@ -1311,19 +1283,17 @@ static void clean_flist(struct file_list *flist, int strip_root, int no_dups)
        for (i = no_dups? 0 : flist->count; i < flist->count; i++) {
                if (flist->files[i]->basename) {
                        prev_i = i;
-                       prev_name = f_name(flist->files[i]);
                        break;
                }
        }
        while (++i < flist->count) {
                if (!flist->files[i]->basename)
                        continue;
-               name = f_name(flist->files[i]);
-               if (strcmp(name, prev_name) == 0) {
+               if (f_name_cmp(flist->files[i], flist->files[prev_i]) == 0) {
                        if (verbose > 1 && !am_server) {
                                rprintf(FINFO,
                                        "removing duplicate name %s from file list %d\n",
-                                       name, i);
+                                       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
@@ -1341,9 +1311,6 @@ static void clean_flist(struct file_list *flist, int strip_root, int no_dups)
                }
                else
                        prev_i = i;
-               /* We set prev_name every iteration to avoid it becoming
-                * invalid when names[][] in f_name() wraps around. */
-               prev_name = name;
        }
 
        if (strip_root) {
@@ -1379,33 +1346,100 @@ static void clean_flist(struct file_list *flist, int strip_root, int no_dups)
 }
 
 
-/*
- * return the full filename of a flist entry
- *
- * This function is too expensive at the moment, because it copies
- * strings when often we only want to compare them.  In any case,
- * using strlcat is silly because it will walk the string repeatedly.
+enum fnc_state { fnc_DIR, fnc_SLASH, fnc_BASE };
+
+/* 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.
  */
-char *f_name(struct file_struct *f)
+int f_name_cmp(struct file_struct *f1, struct file_struct *f2)
 {
-       static char names[10][MAXPATHLEN];
-       static int n;
-       char *p = names[n];
+       int dif;
+       const uchar *c1, *c2;
+       enum fnc_state state1 = fnc_DIR, state2 = fnc_DIR;
 
+       if (!f1 || !f1->basename) {
+               if (!f2 || !f2->basename)
+                       return 0;
+               return -1;
+       }
+       if (!f2 || !f2->basename)
+               return 1;
+
+       if (!(c1 = (uchar*)f1->dirname)) {
+               state1 = fnc_BASE;
+               c1 = (uchar*)f1->basename;
+       }
+       if (!(c2 = (uchar*)f2->dirname)) {
+               state2 = fnc_BASE;
+               c2 = (uchar*)f2->basename;
+       }
+
+       while (1) {
+               if ((dif = (int)*c1 - (int)*c2) != 0)
+                       break;
+               if (!*++c1) {
+                       switch (state1) {
+                       case fnc_DIR:
+                               state1 = fnc_SLASH;
+                               c1 = (uchar*)"/";
+                               break;
+                       case fnc_SLASH:
+                               state1 = fnc_BASE;
+                               c1 = (uchar*)f1->basename;
+                               break;
+                       case fnc_BASE:
+                               break;
+                       }
+               }
+               if (!*++c2) {
+                       switch (state2) {
+                       case fnc_DIR:
+                               state2 = fnc_SLASH;
+                               c2 = (uchar*)"/";
+                               break;
+                       case fnc_SLASH:
+                               state2 = fnc_BASE;
+                               c2 = (uchar*)f2->basename;
+                               break;
+                       case fnc_BASE:
+                               if (!*c1)
+                                       return 0;
+                               break;
+                       }
+               }
+       }
+
+       return dif;
+}
+
+
+/* Return a copy of the full filename of a flist entry, using the indicated
+ * buffer.
+ */
+char *f_name_to(struct file_struct *f, char *fbuf, int bsize)
+{
        if (!f || !f->basename)
                return NULL;
 
-       n = (n + 1) % 10;
-
        if (f->dirname) {
-               int off;
+               int off = strlcpy(fbuf, f->dirname, bsize);
+               off += strlcpy(fbuf + off, "/", bsize - off);
+               strlcpy(fbuf + off, f->basename, bsize - off);
+       } else
+               strlcpy(fbuf, f->basename, bsize);
+       return fbuf;
+}
 
-               off = strlcpy(p, f->dirname, MAXPATHLEN);
-               off += strlcpy(p + off, "/", MAXPATHLEN - off);
-               off += strlcpy(p + off, f->basename, MAXPATHLEN - off);
-       } else {
-               strlcpy(p, f->basename, MAXPATHLEN);
-       }
 
-       return p;
+/* 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];
+       static unsigned int n;
+
+       n = (n + 1) % (sizeof names / sizeof names[0]);
+
+       return f_name_to(f, names[n], sizeof names[0]);
 }