hlink.c

   1 /*
   2    Copyright (C) Andrew Tridgell 1996
   3    Copyright (C) Paul Mackerras 1996
   4
   5    This program is free software; you can redistribute it and/or modify
   6    it under the terms of the GNU General Public License as published by
   7    the Free Software Foundation; either version 2 of the License, or
   8    (at your option) any later version.
   9
  10    This program is distributed in the hope that it will be useful,
  11    but WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13    GNU General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program; if not, write to the Free Software
  17    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18 */
  19
  20 #include "rsync.h"
  21
  22 extern int am_server;
  23 extern int dry_run;
  24
  25 #if SUPPORT_HARD_LINKS
  26 static int hlink_compare(struct file_struct *f1,struct file_struct *f2)
  27 {
  28   if (!S_ISREG(f1->mode) && !S_ISREG(f2->mode)) return 0;
  29   if (!S_ISREG(f1->mode)) return -1;
  30   if (!S_ISREG(f2->mode)) return 1;
  31
  32   if (f1->dev != f2->dev)
  33     return (f1->dev - f2->dev);
  34
  35   if (f1->inode != f2->inode)
  36     return (f1->inode - f2->inode);
  37
  38   return file_compare(f1,f2);
  39 }
  40
  41
  42 static struct file_struct *hlink_list = NULL;
  43 static int hlink_count=0;
  44 #endif
  45
  46 void init_hard_links(struct file_list *flist)
  47 {
  48 #if SUPPORT_HARD_LINKS
  49   if (flist->count < 2) return;
  50
  51   if (hlink_list) free(hlink_list);
  52
  53   if (!(hlink_list =
  54         (struct file_struct *)malloc(sizeof(hlink_list[0])*flist->count)))
  55     out_of_memory("init_hard_links");
  56
  57   bcopy((char *)flist->files,hlink_list,sizeof(hlink_list[0])*flist->count);
  58
  59   qsort(hlink_list,flist->count,
  60         sizeof(hlink_list[0]),
  61         (int (*)())hlink_compare);
  62
  63   hlink_count=flist->count;
  64 #endif
  65 }
  66
  67 /* check if a file should be skipped because it is the same as an
  68    earlier hard link */
  69 int check_hard_link(struct file_struct *file)
  70 {
  71 #if SUPPORT_HARD_LINKS
  72   int low=0,high=hlink_count;
  73   int mid=0,ret=0;
  74
  75   if (!hlink_list || !S_ISREG(file->mode)) return 0;
  76
  77   while (low != high) {
  78     mid = (low+high)/2;
  79     ret = hlink_compare(&hlink_list[mid],file);
  80     if (ret == 0) break;
  81     if (ret > 0)
  82       high=mid;
  83     else
  84       low=mid+1;
  85   }
  86
  87   if (hlink_compare(&hlink_list[mid],file) != 0) return 0;
  88
  89   if (mid > 0 &&
  90       S_ISREG(hlink_list[mid-1].mode) &&
  91       file->dev == hlink_list[mid-1].dev &&
  92       file->inode == hlink_list[mid-1].inode)
  93     return 1;
  94 #endif
  95
  96   return 0;
  97 }
  98
  99
 100 /* create any hard links in the flist */
 101 void do_hard_links(struct file_list *flist)
 102 {
 103 #if SUPPORT_HARD_LINKS
 104   int i;
 105
 106   if (!hlink_list) return;
 107
 108   for (i=1;i<hlink_count;i++) {
 109     if (S_ISREG(hlink_list[i].mode) &&
 110         S_ISREG(hlink_list[i-1].mode) &&
 111         hlink_list[i].name && hlink_list[i-1].name &&
 112         hlink_list[i].dev == hlink_list[i-1].dev &&
 113         hlink_list[i].inode == hlink_list[i-1].inode) {
 114       struct stat st1,st2;
 115
 116       if (stat(hlink_list[i-1].name,&st1) != 0) continue;
 117       if (stat(hlink_list[i].name,&st2) != 0) {
 118         if (!dry_run && link(hlink_list[i-1].name,hlink_list[i].name) != 0) {
 119           fprintf(FINFO,"link %s => %s : %s\n",
 120                   hlink_list[i].name,hlink_list[i-1].name,strerror(errno));
 121           continue;
 122         }
 123       } else {
 124         if (st2.st_dev == st1.st_dev && st2.st_ino == st1.st_ino) continue;
 125
 126         if (!dry_run && (unlink(hlink_list[i].name) != 0 ||
 127                          link(hlink_list[i-1].name,hlink_list[i].name) != 0)) {
 128           fprintf(FINFO,"link %s => %s : %s\n",
 129                   hlink_list[i].name,hlink_list[i-1].name,strerror(errno));
 130           continue;
 131         }
 132       }
 133       fprintf(FINFO,"%s => %s\n",
 134               hlink_list[i].name,hlink_list[i-1].name);
 135     }
 136   }
 137 #endif
 138 }