Fixed failing hunks.
[rsync/rsync-patches.git] / fuzzy.diff
index 82ee022..9ce20dc 100644 (file)
@@ -4,9 +4,9 @@ Lightly tested.
 
 Be sure to run "make proto" before "make".
 
---- orig/generator.c   2005-01-14 18:30:18
-+++ generator.c        2004-11-27 18:12:57
-@@ -44,6 +44,7 @@ extern int size_only;
+--- orig/generator.c   2005-02-03 02:04:20
++++ generator.c        2005-02-03 02:11:10
+@@ -47,6 +47,7 @@ extern int size_only;
  extern OFF_T max_size;
  extern int io_timeout;
  extern int protocol_version;
@@ -14,98 +14,96 @@ Be sure to run "make proto" before "make".
  extern int always_checksum;
  extern char *partial_dir;
  extern char *basis_dir[];
-@@ -239,6 +240,90 @@ static void generate_and_send_sums(int f
+@@ -227,6 +228,88 @@ static void generate_and_send_sums(int f
+               unmap_file(mapbuf);
  }
  
-+static void split_names(char *fname, char **dirname, char **basename)
-+{
-+      char *slash = strrchr(fname, '/');
-+      if (slash) {
-+              *dirname = fname;
-+              *slash = '\0';
-+              *basename = slash+1;
-+      } else {
-+              *basename = fname;
-+              *dirname = ".";
-+      }
-+}
-+
-+
-+static unsigned int measure_name(const char *name, const char *basename,
-+                               const char *ext)
-+{
-+      int namelen = strlen(name);
-+      int extlen = strlen(ext);
-+      unsigned int score = 0;
-+
-+      /* Extensions must match */
-+      if (namelen <= extlen || strcmp(name + namelen - extlen, ext) != 0)
-+              return 0;
-+
-+      /* Now score depends on similarity of prefix */
-+      for (; *name == *basename && *name; name++, basename++)
-+              score++;
-+      return score;
-+}
-+
-+
++/* Try to find a filename in the same dir as "fname" with a similar name.
++ *
++ * TODO:
++ *  - We should be using a cache of names for the current dir, not
++ *    re-reading the destination directory for every file.
++ *  - We must not return an rsync tempfile from the current transfer.
++ *  - If the highest-rated name is not a normal file, we should fall-
++ *    back to the next highest-rated file.
++ *  - We must not return a destination file that is being updated
++ *    during the current transfer, even if we already processed it
++ *    (since the receiver may not be done with it yet).
++ *  - We must weed out any names that a daemon's config has excluded.
++ */
 +static int find_fuzzy(const char *fname, char *buf, STRUCT_STAT *st_ptr)
 +{
 +      DIR *d;
 +      struct dirent *di;
-+      char *basename, *dirname;
-+      char mangled_name[MAXPATHLEN];
++      char *basename, *dirname, *slash;
 +      char bestname[MAXPATHLEN];
-+      unsigned int bestscore = 0;
-+      const char *ext;
++      int suf_len, basename_len;
++      uint32 lowest_dist = 0x7FFFFFFF;
++      const char *suf;
 +
-+      strlcpy(mangled_name, fname, sizeof mangled_name);
++      strlcpy(buf, fname, MAXPATHLEN);
++      if ((slash = strrchr(buf, '/')) != NULL) {
++              dirname = buf;
++              *slash = '\0';
++              basename = slash + 1;
++      } else {
++              basename = buf;
++              dirname = ".";
++      }
++      basename_len = strlen(basename);
 +
-+      split_names(mangled_name, &dirname, &basename);
 +      if (!(d = opendir(dirname))) {
 +              rsyserr(FERROR, errno, "recv_generator opendir(%s)", dirname);
 +              return -1;
 +      }
++      if (slash)
++              *slash = '/';
 +
-+      /* Get final extension, eg. .gz; never full basename though. */
-+      if (!(ext = strrchr(basename + 1, '.')))
-+              ext = basename + strlen(basename); /* ext = "" */
++      suf = find_filename_suffix(basename, basename_len, &suf_len);
 +
++      bestname[0] = '\0';
 +      while ((di = readdir(d)) != NULL) {
-+              const char *dname = d_name(di);
-+              unsigned int score;
++              const char *dname_suf, *dname = d_name(di);
++              uint32 dist;
++              int dname_len, dname_suf_len;
 +
 +              if (dname[0] == '.' && (dname[1] == '\0'
 +                  || (dname[1] == '.' && dname[2] == '\0')))
 +                      continue;
 +
-+              score = measure_name(dname, basename, ext);
++              dname_len = strlen(dname);
++              dname_suf = find_filename_suffix(dname, dname_len, &dname_suf_len);
++
++              dist = fuzzy_distance(dname, dname_len, basename, basename_len);
++              /* Add some extra weight to how well the suffixes match. */
++              dist += fuzzy_distance(dname_suf, dname_suf_len, suf, suf_len) * 10;
 +              if (verbose > 4) {
-+                      rprintf(FINFO, "fuzzy score for %s = %u\n",
-+                              dname, score);
++                      rprintf(FINFO, "fuzzy distance for %s = %d (%d)\n",
++                              dname, (int)(dist>>16), (int)(dist&0xFFFF));
 +              }
-+              if (score > bestscore) {
++              if (dist <= lowest_dist) {
 +                      strlcpy(bestname, dname, sizeof bestname);
-+                      bestscore = score;
++                      lowest_dist = dist;
 +              }
 +      }
 +      closedir(d);
 +
 +      /* Found a candidate. */
-+      if (bestscore != 0) {
-+              pathjoin(buf, MAXPATHLEN, dirname, bestname);
-+              if (verbose > 2)
-+                      rprintf(FINFO, "fuzzy match %s->%s\n", fname, buf);
++      if (bestname[0] != '\0') {
++              strlcpy(basename, bestname, MAXPATHLEN - (basename - buf));
++              if (verbose > 2) {
++                      rprintf(FINFO, "fuzzy match %s->%s\n",
++                              safe_fname(fname), buf);
++              }
 +              return link_stat(buf, st_ptr, 0);
 +      }
 +      return -1;
 +}
 +
  
- /*
-  * Acts on file number @p i from @p flist, whose name is @p fname.
-@@ -493,6 +578,15 @@ static void recv_generator(char *fname, 
+ /* Acts on flist->file's ndx'th item, whose name is fname.  If a directory,
+  * make sure it exists, and has the right permissions/timestamp info.  For
+@@ -477,6 +560,15 @@ static void recv_generator(char *fname, 
        } else
                partialptr = NULL;
  
@@ -121,19 +119,19 @@ Be sure to run "make proto" before "make".
        if (statret == -1) {
                if (preserve_hard_links && hard_link_check(file, HL_SKIP))
                        return;
-@@ -521,6 +615,8 @@ static void recv_generator(char *fname, 
+@@ -505,6 +597,8 @@ static void recv_generator(char *fname, 
  
-       if ((link_dest || copy_dest) && fnamecmp_type != FNAMECMP_FNAME)
+       if (!compare_dest && fnamecmp_type <= FNAMECMP_BASIS_DIR_HIGH)
                ;
 +      else if (fnamecmp_type == FNAMECMP_FUZZY)
 +              ;
        else if (unchanged_file(fnamecmp, file, &st)) {
                if (fnamecmp_type == FNAMECMP_FNAME)
                        set_perms(fname, file, &st, PERMS_REPORT);
-@@ -593,8 +689,24 @@ prepare_to_open:
- notify_others:
-       write_int(f_out, i);
+@@ -579,8 +673,24 @@ notify_others:
+       write_int(f_out, ndx);
+       if (protocol_version >= 29 && inplace && !read_batch)
+               write_byte(f_out, fnamecmp_type);
 -      if (f_out_name >= 0)
 +      if (f_out_name >= 0) {
                write_byte(f_out_name, fnamecmp_type);
@@ -156,7 +154,7 @@ Be sure to run "make proto" before "make".
  
        if (dry_run || read_batch)
                return;
---- orig/main.c        2005-01-14 18:30:18
+--- orig/main.c        2005-01-30 10:07:21
 +++ main.c     2005-01-14 18:33:15
 @@ -48,6 +48,7 @@ extern int keep_dirlinks;
  extern int preserve_hard_links;
@@ -166,7 +164,7 @@ Be sure to run "make proto" before "make".
  extern int relative_paths;
  extern int rsync_port;
  extern int whole_file;
-@@ -464,7 +465,8 @@ static int do_recv(int f_in,int f_out,st
+@@ -491,7 +492,8 @@ static int do_recv(int f_in,int f_out,st
        int pid;
        int status = 0;
        int error_pipe[2], name_pipe[2];
@@ -176,25 +174,25 @@ Be sure to run "make proto" before "make".
  
        /* The receiving side mustn't obey this, or an existing symlink that
         * points to an identical file won't be replaced by the referent. */
---- orig/options.c     2005-01-15 20:26:21
-+++ options.c  2004-11-29 01:36:48
-@@ -86,6 +86,7 @@ int copy_unsafe_links = 0;
+--- orig/options.c     2005-02-01 10:39:22
++++ options.c  2005-01-28 19:31:20
+@@ -90,6 +90,7 @@ int copy_unsafe_links = 0;
  int size_only = 0;
  int daemon_bwlimit = 0;
  int bwlimit = 0;
 +int fuzzy_basis = 0;
  size_t bwlimit_writemax = 0;
- int delete_after = 0;
  int only_existing = 0;
-@@ -288,6 +289,7 @@ void usage(enum logcode F)
+ int opt_ignore_existing = 0;
+@@ -303,6 +304,7 @@ void usage(enum logcode F)
    rprintf(F,"     --compare-dest=DIR      also compare destination files relative to DIR\n");
    rprintf(F,"     --copy-dest=DIR         ... and include copies of unchanged files\n");
    rprintf(F,"     --link-dest=DIR         hardlink to files in DIR when unchanged\n");
 +  rprintf(F,"     --fuzzy                 find similar file for basis when no dest file\n");
-   rprintf(F," -P                          equivalent to --partial --progress\n");
    rprintf(F," -z, --compress              compress file data\n");
-   rprintf(F," -C, --cvs-exclude           auto ignore files in the same way CVS does\n");
-@@ -384,6 +386,7 @@ static struct poptOption long_options[] 
+   rprintf(F," -C, --cvs-exclude           auto-ignore files the same way CVS does\n");
+   rprintf(F," -f, --filter=RULE           add a file-filtering RULE\n");
+@@ -408,6 +410,7 @@ static struct poptOption long_options[] 
    {"compare-dest",     0,  POPT_ARG_STRING, 0, OPT_COMPARE_DEST, 0, 0 },
    {"copy-dest",        0,  POPT_ARG_STRING, 0, OPT_COPY_DEST, 0, 0 },
    {"link-dest",        0,  POPT_ARG_STRING, 0, OPT_LINK_DEST, 0, 0 },
@@ -202,22 +200,9 @@ Be sure to run "make proto" before "make".
    /* TODO: Should this take an optional int giving the compression level? */
    {"compress",        'z', POPT_ARG_NONE,   &do_compression, 0, 0, 0 },
    {"stats",            0,  POPT_ARG_NONE,   &do_stats, 0, 0, 0 },
-@@ -952,10 +955,10 @@ int parse_arguments(int *argc, const cha
-                       return 0;
-               }
-               keep_partial = 0;
--              if (dest_option) {
-+              if (dest_option || fuzzy_basis) {
-                       snprintf(err_buf, sizeof err_buf,
-                                "--inplace does not yet work with %s\n",
--                               dest_option);
-+                               dest_option ? dest_option : "--fuzzy");
-                       return 0;
-               }
- #else
-@@ -1240,6 +1243,9 @@ void server_options(char **args,int *arg
-                       args[ac++] = "--no-relative";
-       }
+@@ -1321,6 +1324,9 @@ void server_options(char **args,int *arg
+       if (!implied_dirs && !am_sender)
+               args[ac++] = "--no-implied-dirs";
  
 +      if (fuzzy_basis && am_sender)
 +              args[ac++] = "--fuzzy";
@@ -225,9 +210,9 @@ Be sure to run "make proto" before "make".
        *argc = ac;
        return;
  
---- orig/receiver.c    2005-01-10 09:46:01
-+++ receiver.c 2004-11-27 18:15:01
-@@ -323,6 +323,27 @@ static int receive_data(int f_in, char *
+--- orig/receiver.c    2005-02-03 02:04:20
++++ receiver.c 2005-01-15 21:21:02
+@@ -256,6 +256,27 @@ static int receive_data(int f_in, char *
  }
  
  
@@ -255,7 +240,7 @@ Be sure to run "make proto" before "make".
  static void discard_receive_data(int f_in, OFF_T length)
  {
        receive_data(f_in, NULL, -1, 0, NULL, -1, length);
-@@ -453,6 +474,10 @@ int recv_files(int f_in, struct file_lis
+@@ -395,6 +416,10 @@ int recv_files(int f_in, struct file_lis
                        case FNAMECMP_BACKUP:
                                fnamecmp = get_backup_name(fname);
                                break;
@@ -263,26 +248,152 @@ Be sure to run "make proto" before "make".
 +                              read_gen_name(f_in_name, fnamecmpbuf);
 +                              fnamecmp = fnamecmpbuf;
 +                              break;
-                       case FNAMECMP_BASIS_DIR:
                        default:
-                               pathjoin(fnamecmpbuf, sizeof fnamecmpbuf,
---- orig/rsync.h       2005-01-10 00:21:12
-+++ rsync.h    2004-11-03 22:53:09
-@@ -125,6 +125,7 @@
+                               if (j >= basis_dir_cnt) {
+                                       rprintf(FERROR,
+--- orig/rsync.h       2005-02-03 02:04:20
++++ rsync.h    2005-01-19 18:36:47
+@@ -131,6 +131,7 @@
  #define FNAMECMP_FNAME                0x80
  #define FNAMECMP_PARTIAL_DIR  0x81
  #define FNAMECMP_BACKUP       0x82
 +#define FNAMECMP_FUZZY                0x83
  
- /* Log-message categories.  FLOG is only used on the daemon side to
---- orig/rsync.yo      2005-01-15 04:36:32
-+++ rsync.yo   2004-11-27 18:15:22
-@@ -358,6 +358,7 @@ verb(
+ /* For calling delete_file() */
+ #define DEL_DIR               (1<<0)
+--- orig/rsync.yo      2005-02-01 10:39:23
++++ rsync.yo   2005-01-28 19:31:36
+@@ -354,6 +354,7 @@ to the detailed description below for a 
       --compare-dest=DIR      also compare received files relative to DIR
       --copy-dest=DIR         ... and include copies of unchanged files
       --link-dest=DIR         hardlink to files in DIR when unchanged
 +     --fuzzy                 find similar file for basis when no dest
-  -P                          equivalent to --partial --progress
   -z, --compress              compress file data
-  -C, --cvs-exclude           auto ignore files in the same way CVS does
+  -C, --cvs-exclude           auto-ignore files in the same way CVS does
+  -f, --filter=RULE           add a file-filtering RULE
+@@ -937,6 +938,14 @@ bf(--link-dest) from working properly fo
+ (or implied by bf(-a)).  You can work-around this bug by avoiding the bf(-o) option
+ when sending to an old rsync.
++dit(bf(--fuzzy)) This option tells rsync that it should look around for a
++basis file for any destination file that is missing.  The current algorithm
++looks for a similarly-named file in the same directory as the destination
++file, and, if found, uses that to try to speed up the transfer.  Note that
++the use of the --delete option might get rid of any potential fuzzy-match
++files, so either use --delete-after or filename exclusions if you need to
++prevent this.
++
+ dit(bf(-z, --compress)) With this option, rsync compresses any data from
+ the files that it sends to the destination machine.  This
+ option is useful on slow connections.  The compression method used is the
+--- orig/util.c        2005-01-28 19:08:20
++++ util.c     2005-01-19 17:30:51
+@@ -1213,3 +1213,108 @@ void *_realloc_array(void *ptr, unsigned
+               return malloc(size * num);
+       return realloc(ptr, size * num);
+ }
++
++/* Take a filename and filename length and return the most significant
++ * filename suffix we can find.  This ignores suffixes such as "~",
++ * ".bak", ".orig", ".~1~", etc. */
++const char *find_filename_suffix(const char *fn, int fn_len, int *len_ptr)
++{
++      const char *suf, *s;
++      BOOL had_tilde;
++      int s_len;
++
++      /* One or more dots at the start aren't a suffix. */
++      while (fn_len && *fn == '.') fn++, fn_len--;
++
++      /* Ignore the ~ in a "foo~" filename. */
++      if (fn_len > 1 && fn[fn_len-1] == '~')
++              fn_len--, had_tilde = True;
++      else
++              had_tilde = False;
++
++      /* Assume we don't find an suffix. */
++      suf = "";
++      *len_ptr = 0;
++
++      /* Find the last significant suffix. */
++      for (s = fn + fn_len - 1; fn_len > 1; ) {
++              while (*s != '.' && s != fn) s--;
++              if (s == fn)
++                      break;
++              s_len = fn_len - (s - fn);
++              fn_len = s - fn;
++              if (s_len == 3) {
++                      if (strcmp(s+1, "bak") == 0
++                       || strcmp(s+1, "old") == 0)
++                              continue;
++              } else if (s_len == 4) {
++                      if (strcmp(s+1, "orig") == 0)
++                              continue;
++              } else if (s_len > 2 && had_tilde
++                  && s[1] == '~' && isdigit(s[2]))
++                      continue;
++              *len_ptr = s_len;
++              suf = s;
++              /* Determine if the suffix is all digits. */
++              for (s++, s_len--; s_len > 0; s++, s_len--) {
++                      if (!isdigit(*s))
++                              return suf;
++              }
++              /* An all-digit suffix may not be that signficant. */
++              continue;
++      }
++
++      return suf;
++}
++
++/* This is an implementation of the Levenshtein distance algorithm.  It
++ * was implemented to avoid needing a two-dimensional matrix (to save
++ * memory).  It was also tweaked to try to factor in the ASCII distance
++ * between changed characters as a minor distance quantity.  The normal
++ * Levenshtein units of distance (each signifying a single change between
++ * the two strings) are defined as a "UNIT". */
++
++#define UNIT (1 << 16)
++
++uint32 fuzzy_distance(const char *s1, int len1, const char *s2, int len2)
++{
++      uint32 a[MAXPATHLEN], diag, above, left, diag_inc, above_inc, left_inc;
++      int32 cost;
++      int i1, i2;
++
++      if (!len1 || !len2) {
++              if (!len1) {
++                      s1 = s2;
++                      len1 = len2;
++              }
++              for (i1 = 0, cost = 0; i1 < len1; i1++)
++                      cost += s1[i1];
++              return (int32)len1 * UNIT + cost;
++      }
++
++      for (i2 = 0; i2 < len2; i2++)
++              a[i2] = (i2+1) * UNIT;
++
++      for (i1 = 0; i1 < len1; i1++) {
++              diag = i1 * UNIT;
++              above = (i1+1) * UNIT;
++              for (i2 = 0; i2 < len2; i2++) {
++                      left = a[i2];
++                      if ((cost = *((uchar*)s1+i1) - *((uchar*)s2+i2)) != 0) {
++                              if (cost < 0)
++                                      cost = UNIT - cost;
++                              else
++                                      cost = UNIT + cost;
++                      }
++                      diag_inc = diag + cost;
++                      left_inc = left + UNIT + *((uchar*)s1+i1);
++                      above_inc = above + UNIT + *((uchar*)s2+i2);
++                      a[i2] = above = left < above
++                            ? (left_inc < diag_inc ? left_inc : diag_inc)
++                            : (above_inc < diag_inc ? above_inc : diag_inc);
++                      diag = left;
++              }
++      }
++
++      return a[len2-1];
++}