Update support/rrsync and some packaging/* scripts.
[rsync.git] / hlink.c
diff --git a/hlink.c b/hlink.c
index 962261c747cc1b79b6ceccf1346306738e3d09fe..cfa1065e207d6f11c16eb120d967a01f6c5ad699 100644 (file)
--- a/hlink.c
+++ b/hlink.c
@@ -4,7 +4,7 @@
  * Copyright (C) 1996 Andrew Tridgell
  * Copyright (C) 1996 Paul Mackerras
  * Copyright (C) 2002 Martin Pool <mbp@samba.org>
- * Copyright (C) 2004-2007 Wayne Davison
+ * Copyright (C) 2004-2009 Wayne Davison
  *
  * This program is free software; you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
 
 extern int verbose;
 extern int dry_run;
+extern int list_only;
 extern int am_sender;
 extern int inc_recurse;
 extern int do_xfers;
 extern int link_dest;
 extern int preserve_acls;
+extern int preserve_xattrs;
 extern int make_backups;
 extern int protocol_version;
 extern int remove_source_files;
@@ -55,7 +57,7 @@ static struct file_list *hlink_flist;
 void init_hard_links(void)
 {
        if (am_sender || protocol_version < 30)
-               dev_tbl = hashtable_create(16, SIZEOF_INT64 == 8);
+               dev_tbl = hashtable_create(16, 1);
        else if (inc_recurse)
                prior_hlinks = hashtable_create(1024, 0);
 }
@@ -65,11 +67,12 @@ struct ht_int64_node *idev_find(int64 dev, int64 ino)
        static struct ht_int64_node *dev_node = NULL;
        struct hashtable *tbl;
 
-       if (!dev_node || dev_node->key != dev) {
+       /* Note that some OSes have a dev == 0, so increment to avoid storing a 0. */
+       if (!dev_node || dev_node->key != dev+1) {
                /* We keep a separate hash table of inodes for every device. */
-               dev_node = hashtable_find(dev_tbl, dev, 1);
+               dev_node = hashtable_find(dev_tbl, dev+1, 1);
                if (!(tbl = dev_node->data))
-                       tbl = dev_node->data = hashtable_create(512, SIZEOF_INT64 == 8);
+                       tbl = dev_node->data = hashtable_create(512, 1);
        } else
                tbl = dev_node->data;
 
@@ -118,18 +121,23 @@ static void match_gnums(int32 *ndx_list, int ndx_count)
                if (inc_recurse) {
                        node = hashtable_find(prior_hlinks, gnum, 1);
                        if (!node->data) {
-                               node->data = new_array0(char, 5);
+                               if (!(node->data = new_array0(char, 5)))
+                                       out_of_memory("match_gnums");
                                assert(gnum >= hlink_flist->ndx_start);
                                file->flags |= FLAG_HLINK_FIRST;
                                prev = -1;
                        } else if (CVAL(node->data, 0) == 0) {
                                struct file_list *flist;
-                               struct file_struct *fp;
                                prev = IVAL(node->data, 1);
-                               flist = flist_for_ndx(prev);
-                               assert(flist != NULL);
-                               fp = flist->files[prev - flist->ndx_start];
-                               fp->flags &= ~FLAG_HLINK_LAST;
+                               flist = flist_for_ndx(prev, NULL);
+                               if (flist)
+                                       flist->files[prev - flist->ndx_start]->flags &= ~FLAG_HLINK_LAST;
+                               else {
+                                       /* We skipped all prior files in this
+                                        * group, so mark this as a "first". */
+                                       file->flags |= FLAG_HLINK_FIRST;
+                                       prev = -1;
+                               }
                        } else
                                prev = -1;
                } else {
@@ -142,7 +150,7 @@ static void match_gnums(int32 *ndx_list, int ndx_count)
                        if (gnum != gnum_next)
                                break;
                        F_HL_PREV(file) = prev;
-                       /* The linked list must use raw ndx values. */
+                       /* The linked list uses over-the-wire ndx values. */
                        if (unsort_ndx)
                                prev = F_NDX(file);
                        else
@@ -174,23 +182,25 @@ static void match_gnums(int32 *ndx_list, int ndx_count)
  * to first when we're done. */
 void match_hard_links(struct file_list *flist)
 {
-       int i, ndx_count = 0;
-       int32 *ndx_list;
+       if (!list_only && flist->used) {
+               int i, ndx_count = 0;
+               int32 *ndx_list;
 
-       if (!(ndx_list = new_array(int32, flist->used)))
-               out_of_memory("match_hard_links");
+               if (!(ndx_list = new_array(int32, flist->used)))
+                       out_of_memory("match_hard_links");
 
-       for (i = 0; i < flist->used; i++) {
-               if (F_IS_HLINKED(flist->sorted[i]))
-                       ndx_list[ndx_count++] = i;
-       }
+               for (i = 0; i < flist->used; i++) {
+                       if (F_IS_HLINKED(flist->sorted[i]))
+                               ndx_list[ndx_count++] = i;
+               }
 
-       hlink_flist = flist;
+               hlink_flist = flist;
 
-       if (ndx_count)
-               match_gnums(ndx_list, ndx_count);
+               if (ndx_count)
+                       match_gnums(ndx_list, ndx_count);
 
-       free(ndx_list);
+               free(ndx_list);
+       }
        if (protocol_version < 30)
                idev_destroy();
 }
@@ -217,7 +227,7 @@ static int maybe_hard_link(struct file_struct *file, int ndx,
                        if (!make_backup(fname))
                                return -1;
                } else if (robust_unlink(fname)) {
-                       rsyserr(FERROR, errno, "unlink %s failed",
+                       rsyserr(FERROR_XFER, errno, "unlink %s failed",
                                full_fname(fname));
                        return -1;
                }
@@ -237,21 +247,43 @@ static int maybe_hard_link(struct file_struct *file, int ndx,
 }
 
 /* Figure out if a prior entry is still there or if we just have a
- * cached name for it.  Never called with a FLAG_HLINK_FIRST entry. */
-static char *check_prior(int prev_ndx, int gnum, struct file_list **flist_p)
+ * cached name for it. */
+static char *check_prior(struct file_struct *file, int gnum,
+                        int *prev_ndx_p, struct file_list **flist_p)
 {
-       struct file_list *flist = flist_for_ndx(prev_ndx);
+       struct file_struct *fp;
        struct ht_int32_node *node;
+       int prev_ndx = F_HL_PREV(file);
 
-       if (flist) {
-               *flist_p = flist;
-               return NULL;
+       while (1) {
+               struct file_list *flist;
+               if (prev_ndx < 0
+                || (flist = flist_for_ndx(prev_ndx, NULL)) == NULL)
+                       break;
+               fp = flist->files[prev_ndx - flist->ndx_start];
+               if (!(fp->flags & FLAG_SKIP_HLINK)) {
+                       *prev_ndx_p = prev_ndx;
+                       *flist_p = flist;
+                       return NULL;
+               }
+               F_HL_PREV(file) = prev_ndx = F_HL_PREV(fp);
+       }
+
+       if (inc_recurse
+        && (node = hashtable_find(prior_hlinks, gnum, 0)) != NULL) {
+               assert(node->data != NULL);
+               if (CVAL(node->data, 0) != 0) {
+                       *prev_ndx_p = -1;
+                       *flist_p = NULL;
+                       return node->data;
+               }
+               /* The prior file must have been skipped. */
+               F_HL_PREV(file) = -1;
        }
 
-       node = hashtable_find(prior_hlinks, gnum, 0);
-       assert(node != NULL && node->data);
-       assert(CVAL(node->data, 0) != 0);
-       return node->data;
+       *prev_ndx_p = -1;
+       *flist_p = NULL;
+       return NULL;
 }
 
 /* Only called if FLAG_HLINKED is set and FLAG_HLINK_FIRST is not.  Returns:
@@ -265,20 +297,28 @@ int hard_link_check(struct file_struct *file, int ndx, const char *fname,
        char *realname, *prev_name;
        struct file_list *flist;
        int gnum = inc_recurse ? F_HL_GNUM(file) : -1;
-       int prev_ndx = F_HL_PREV(file);
+       int prev_ndx;
 
-       prev_name = realname = check_prior(prev_ndx, gnum, &flist);
+       prev_name = realname = check_prior(file, gnum, &prev_ndx, &flist);
 
        if (!prev_name) {
-               struct file_struct *prev_file = flist->files[prev_ndx - flist->ndx_start];
+               struct file_struct *prev_file;
 
-               /* Is the previous link is not complete yet? */
+               if (!flist) {
+                       /* The previous file was skipped, so this one is
+                        * treated as if it were the first in its group. */
+                       return 0;
+               }
+
+               prev_file = flist->files[prev_ndx - flist->ndx_start];
+
+               /* Is the previous link not complete yet? */
                if (!(prev_file->flags & FLAG_HLINK_DONE)) {
                        /* Is the previous link being transferred? */
                        if (prev_file->flags & FLAG_FILE_SENT) {
-                               /* Add ourselves to the list of files that will be
-                                * updated when the transfer completes, and mark
-                                * ourself as waiting for the transfer. */
+                               /* Add ourselves to the list of files that will
+                                * be updated when the transfer completes, and
+                                * mark ourself as waiting for the transfer. */
                                F_HL_PREV(file) = F_HL_PREV(prev_file);
                                F_HL_PREV(prev_file) = ndx;
                                file->flags |= FLAG_FILE_SENT;
@@ -291,8 +331,8 @@ int hard_link_check(struct file_struct *file, int ndx, const char *fname,
                /* There is a finished file to link with! */
                if (!(prev_file->flags & FLAG_HLINK_FIRST)) {
                        /* The previous previous is FIRST when prev is not. */
-                       prev_ndx = F_HL_PREV(prev_file);
-                       prev_name = realname = check_prior(prev_ndx, gnum, &flist);
+                       prev_name = realname = check_prior(prev_file, gnum, &prev_ndx, &flist);
+                       assert(prev_name != NULL || flist != NULL);
                        /* Update our previous pointer to point to the FIRST. */
                        F_HL_PREV(file) = prev_ndx;
                }
@@ -317,9 +357,13 @@ int hard_link_check(struct file_struct *file, int ndx, const char *fname,
        }
 
        if (link_stat(prev_name, &prev_st, 0) < 0) {
-               rsyserr(FERROR, errno, "stat %s failed",
-                       full_fname(prev_name));
-               return -1;
+               if (!dry_run || errno != ENOENT) {
+                       rsyserr(FERROR_XFER, errno, "stat %s failed", full_fname(prev_name));
+                       return -1;
+               }
+               /* A new hard-link will get a new dev & inode, so approximate
+                * those values in dry-run mode by zeroing them. */
+               memset(&prev_st, 0, sizeof prev_st);
        }
 
        if (statret < 0 && basis_dir[0] != NULL) {
@@ -329,6 +373,9 @@ int hard_link_check(struct file_struct *file, int ndx, const char *fname,
                int j = 0;
 #ifdef SUPPORT_ACLS
                alt_sx.acc_acl = alt_sx.def_acl = NULL;
+#endif
+#ifdef SUPPORT_XATTRS
+               alt_sx.xattr = NULL;
 #endif
                do {
                        pathjoin(cmpbuf, MAXPATHLEN, basis_dir[j], fname);
@@ -358,19 +405,37 @@ int hard_link_check(struct file_struct *file, int ndx, const char *fname,
                        sxp->st = alt_sx.st;
 #ifdef SUPPORT_ACLS
                        if (preserve_acls && !S_ISLNK(file->mode)) {
-                               if (!ACL_READY(*sxp))
+                               free_acl(sxp);
+                               if (!ACL_READY(alt_sx))
                                        get_acl(cmpbuf, sxp);
                                else {
                                        sxp->acc_acl = alt_sx.acc_acl;
                                        sxp->def_acl = alt_sx.def_acl;
+                                       alt_sx.acc_acl = alt_sx.def_acl = NULL;
                                }
                        }
 #endif
-               }
+#ifdef SUPPORT_XATTRS
+                       if (preserve_xattrs) {
+                               free_xattr(sxp);
+                               if (!XATTR_READY(alt_sx))
+                                       get_xattr(cmpbuf, sxp);
+                               else {
+                                       sxp->xattr = alt_sx.xattr;
+                                       alt_sx.xattr = NULL;
+                               }
+                       }
+#endif
+               } else {
 #ifdef SUPPORT_ACLS
-               else if (preserve_acls)
-                       free_acl(&alt_sx);
+                       if (preserve_acls)
+                               free_acl(&alt_sx);
 #endif
+#ifdef SUPPORT_XATTRS
+                       if (preserve_xattrs)
+                               free_xattr(&alt_sx);
+#endif
+               }
        }
 
        if (maybe_hard_link(file, ndx, fname, statret, sxp, prev_name, &prev_st,
@@ -390,10 +455,10 @@ int hard_link_one(struct file_struct *file, const char *fname,
                enum logcode code;
                if (terse) {
                        if (!verbose)
-                               return -1;
+                               return 0;
                        code = FINFO;
                } else
-                       code = FERROR;
+                       code = FERROR_XFER;
                rsyserr(code, errno, "link %s => %s failed",
                        full_fname(fname), oldname);
                return 0;
@@ -410,14 +475,14 @@ void finish_hard_link(struct file_struct *file, const char *fname, int fin_ndx,
 {
        stat_x prev_sx;
        STRUCT_STAT st;
-       char alt_name[MAXPATHLEN], *prev_name;
+       char prev_name[MAXPATHLEN], alt_name[MAXPATHLEN];
        const char *our_name;
        struct file_list *flist;
        int prev_statret, ndx, prev_ndx = F_HL_PREV(file);
 
        if (stp == NULL && prev_ndx >= 0) {
                if (link_stat(fname, &st, 0) < 0) {
-                       rsyserr(FERROR, errno, "stat %s failed",
+                       rsyserr(FERROR_XFER, errno, "stat %s failed",
                                full_fname(fname));
                        return;
                }
@@ -437,23 +502,28 @@ void finish_hard_link(struct file_struct *file, const char *fname, int fin_ndx,
 #ifdef SUPPORT_ACLS
        prev_sx.acc_acl = prev_sx.def_acl = NULL;
 #endif
+#ifdef SUPPORT_XATTRS
+       prev_sx.xattr = NULL;
+#endif
 
        while ((ndx = prev_ndx) >= 0) {
                int val;
-               flist = flist_for_ndx(ndx);
-               assert(flist != NULL);
+               flist = flist_for_ndx(ndx, "finish_hard_link");
                file = flist->files[ndx - flist->ndx_start];
                file->flags = (file->flags & ~FLAG_HLINK_FIRST) | FLAG_HLINK_DONE;
                prev_ndx = F_HL_PREV(file);
                F_HL_PREV(file) = fin_ndx;
-               prev_name = f_name(file, NULL);
-               prev_statret = link_stat(prev_name, &prev_sx.st, 0);
+               prev_statret = link_stat(f_name(file, prev_name), &prev_sx.st, 0);
                val = maybe_hard_link(file, ndx, prev_name, prev_statret, &prev_sx,
                                      our_name, stp, fname, itemizing, code);
                flist->in_progress--;
 #ifdef SUPPORT_ACLS
                if (preserve_acls)
                        free_acl(&prev_sx);
+#endif
+#ifdef SUPPORT_XATTRS
+               if (preserve_xattrs)
+                       free_xattr(&prev_sx);
 #endif
                if (val < 0)
                        continue;
@@ -464,11 +534,43 @@ void finish_hard_link(struct file_struct *file, const char *fname, int fin_ndx,
        if (inc_recurse) {
                int gnum = F_HL_GNUM(file);
                struct ht_int32_node *node = hashtable_find(prior_hlinks, gnum, 0);
-               assert(node != NULL && node->data != NULL);
-               assert(CVAL(node->data, 0) == 0);
+               if (node == NULL) {
+                       rprintf(FERROR, "Unable to find a hlink node for %d (%s)\n", gnum, f_name(file, prev_name));
+                       exit_cleanup(RERR_MESSAGEIO);
+               }
+               if (node->data == NULL) {
+                       rprintf(FERROR, "Hlink node data for %d is NULL (%s)\n", gnum, f_name(file, prev_name));
+                       exit_cleanup(RERR_MESSAGEIO);
+               }
+               if (CVAL(node->data, 0) != 0) {
+                       rprintf(FERROR, "Hlink node data for %d already has path=%s (%s)\n",
+                               gnum, (char*)node->data, f_name(file, prev_name));
+                       exit_cleanup(RERR_MESSAGEIO);
+               }
                free(node->data);
                if (!(node->data = strdup(our_name)))
                        out_of_memory("finish_hard_link");
        }
 }
+
+int skip_hard_link(struct file_struct *file, struct file_list **flist_p)
+{
+       struct file_list *flist;
+       int prev_ndx;
+
+       file->flags |= FLAG_SKIP_HLINK;
+       if (!(file->flags & FLAG_HLINK_LAST))
+               return -1;
+
+       check_prior(file, F_HL_GNUM(file), &prev_ndx, &flist);
+       if (prev_ndx >= 0) {
+               file = flist->files[prev_ndx - flist->ndx_start];
+               if (file->flags & (FLAG_HLINK_DONE|FLAG_FILE_SENT))
+                       return -1;
+               file->flags |= FLAG_HLINK_LAST;
+               *flist_p = flist;
+       }
+
+       return prev_ndx;
+}
 #endif