Make idev, hlink and file_struct + strings use allocation
[rsync/rsync.git] / backup.c
index 54d5fae..248ae90 100644 (file)
--- a/backup.c
+++ b/backup.c
@@ -1,16 +1,16 @@
-/* 
+/*
    Copyright (C) Andrew Tridgell 1999
-   
+
    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
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
-   
+
    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 #include "rsync.h"
 
 extern int verbose;
+extern int backup_suffix_len;
+extern int backup_dir_len;
 extern char *backup_suffix;
+extern char *backup_dir;
+extern struct file_struct **hlink_list;
 
+extern int am_root;
+extern int preserve_devices;
+extern int preserve_links;
+extern int preserve_hard_links;
+extern int orig_umask;
 
-int make_backup(char *fname)
+/* simple backup creates a backup with a suffix in the same directory */
+static int make_simple_backup(char *fname)
 {
        char fnamebak[MAXPATHLEN];
-       if (strlen(fname) + strlen(backup_suffix) > (MAXPATHLEN-1)) {
-               rprintf(FERROR,"backup filename too long\n");
+
+       if (stringjoin(fnamebak, sizeof fnamebak, fname, backup_suffix, NULL)
+           >= sizeof fnamebak) {
+               rprintf(FERROR, "backup filename too long\n");
                return 0;
        }
 
-       slprintf(fnamebak,sizeof(fnamebak),"%s%s",fname,backup_suffix);
-       if (do_rename(fname,fnamebak) != 0) {
+       if (do_rename(fname, fnamebak) != 0) {
                /* cygwin (at least version b19) reports EINVAL */
                if (errno != ENOENT && errno != EINVAL) {
-                       rprintf(FERROR,"rename %s %s : %s\n",fname,fnamebak,strerror(errno));
+                       rsyserr(FERROR, errno, "rename %s to backup %s", fname, fnamebak);
                        return 0;
                }
        } else if (verbose > 1) {
-               rprintf(FINFO,"backed up %s to %s\n",fname,fnamebak);
+               rprintf(FINFO, "backed up %s to %s\n", fname, fnamebak);
+       }
+       return 1;
+}
+
+
+/* recursively make a directory path */
+static int make_dir(char *name, int mask)
+{
+       char newdir [MAXPATHLEN];
+       char *p, *d;
+
+       /* copy pathname over, look for last '/' */
+       for (p = d = newdir; *name; *d++ = *name++)
+               if (*name == '/')
+                       p = d;
+       if (p == newdir)
+               return 0;
+       *p = 0;
+
+       /* make the new directory, if that fails then make its parent */
+       while (do_mkdir(newdir, mask) != 0) {
+               if (errno != ENOENT || !make_dir(newdir, mask))
+                       return 0;
+       }
+
+       return 1;
+} /* make_dir */
+
+
+/****************************************************************************
+Create a directory given an absolute path, perms based upon another directory
+path
+****************************************************************************/
+static int make_bak_dir(char *fname, char *bak_path)
+{
+       STRUCT_STAT st;
+       STRUCT_STAT *st2;
+       char fullpath[MAXPATHLEN];
+       char *p;
+       char *q;
+
+       while(strncmp(bak_path, "./", 2) == 0) bak_path += 2;
+
+       if (pathjoin(fullpath, sizeof fullpath, bak_path, fname)
+           >= sizeof fullpath) {
+               rprintf(FERROR, "backup dirname too long\n");
+               return 0;
+       }
+       p = fullpath;
+       q = fullpath + strlen(bak_path);
+       if (*q == '/')
+               q++; /* Point past the middle '/' added by pathjoin(). */
+
+       /* Make the directories */
+       while ((p = strchr(p, '/')) != NULL) {
+               *p = 0;
+               if (do_lstat(fullpath, &st) != 0) {
+                       do_mkdir(fullpath, 0777 & ~orig_umask);
+                       if (p > q) {
+                               if (do_lstat(q, &st) != 0) {
+                                       rprintf(FERROR, "make_bak_dir stat %s failed: %s\n",
+                                               full_fname(fullpath), strerror(errno));
+                               } else {
+                                       st2 = &st;
+                                       set_modtime(fullpath, st2->st_mtime);
+                                       if (do_lchown(fullpath, st2->st_uid, st2->st_gid) != 0) {
+                                               rprintf(FERROR, "make_bak_dir chown %s failed: %s\n",
+                                                       full_fname(fullpath), strerror(errno));
+                                       }
+                                       if (do_chmod(fullpath, st2->st_mode) != 0) {
+                                               rprintf(FERROR, "make_bak_dir failed to set permissions on %s: %s\n",
+                                                       full_fname(fullpath), strerror(errno));
+                                       }
+                               }
+                       }
+               }
+               *p++ = '/';
+       }
+       return 0;
+}
+
+/* robustly move a file, creating new directory structures if necessary */
+static int robust_move(char *src, char *dst)
+{
+       int keep_trying = 4;
+       int keep_path_extfs = 0;
+       int failed;
+
+       while (keep_trying) {
+               if (keep_path_extfs) {
+                       failed = copy_file(src, dst, 0755);
+                       if (!failed)
+                               do_unlink(src);
+               } else
+                       failed = robust_rename(src, dst);
+
+               if (failed) {
+                       if (verbose > 2) {
+                               rprintf(FERROR, "robust_move failed: %s(%d)\n",
+                                       strerror(errno), errno);
+                       }
+                       switch (errno) {
+                       case EXDEV:     /* external filesystem */
+                               keep_path_extfs = 1;
+                               keep_trying--;
+                               break;
+                       case ENOENT:    /* no directory to write to */
+                               make_dir(dst, 0700);
+                               keep_trying--;
+                               break;
+                       default:
+                               keep_trying = 0;
+                               break;
+                       }
+               } else
+                       keep_trying = 0;
+       } /* while */
+       return !failed;
+} /* robust_move */
+
+
+/* if we have a backup_dir, then we get here from make_backup().
+   We will move the file to be deleted into a parallel directory tree */
+static int keep_backup(char *fname)
+{
+       static int initialised;
+       char keep_name[MAXPATHLEN];
+       STRUCT_STAT st;
+       struct file_struct *file;
+       int kept = 0;
+       int ret_code;
+
+       if (!initialised) {
+               if (backup_dir_len && backup_dir[backup_dir_len - 1] == '/')
+                       backup_dir[--backup_dir_len] = '\0';
+               if (verbose > 0)
+                       rprintf(FINFO, "backup_dir is %s\n", backup_dir);
+
+               initialised = 1;
        }
+
+       /* return if no file to keep */
+#if SUPPORT_LINKS
+       if (do_lstat(fname, &st)) return 1;
+#else
+       if (do_stat(fname, &st)) return 1;
+#endif
+
+       file = make_file(fname, NULL, NO_EXCLUDES);
+
+       /* the file could have disappeared */
+       if (!file) return 1;
+
+       /* make a complete pathname for backup file */
+       if (stringjoin(keep_name, sizeof keep_name,
+                      backup_dir, "/", fname, backup_suffix, NULL)
+           >= sizeof keep_name) {
+               rprintf(FERROR, "keep_backup filename too long\n");
+               return 0;
+       }
+
+#ifdef HAVE_MKNOD
+       /* Check to see if this is a device file, or link */
+       if (IS_DEVICE(file->mode)) {
+               if (am_root && preserve_devices) {
+                       make_bak_dir(fname, backup_dir);
+                       if (do_mknod(keep_name, file->mode, file->u.rdev) != 0) {
+                               rprintf(FERROR, "mknod %s failed: %s\n",
+                                       full_fname(keep_name), strerror(errno));
+                       } else if (verbose > 2) {
+                               rprintf(FINFO,
+                                       "make_backup: DEVICE %s successful.\n",
+                                       fname);
+                       }
+               }
+               kept = 1;
+               do_unlink(fname);
+       }
+#endif
+
+       if (!kept && S_ISDIR(file->mode)) {
+               /* make an empty directory */
+               make_bak_dir(fname, backup_dir);
+               do_mkdir(keep_name, file->mode);
+               ret_code = do_rmdir(fname);
+
+               if (verbose > 2) {
+                       rprintf(FINFO, "make_backup: RMDIR %s returns %i\n",
+                               full_fname(fname), ret_code);
+               }
+               kept = 1;
+       }
+
+#if SUPPORT_LINKS
+       if (!kept && preserve_links && S_ISLNK(file->mode)) {
+               extern int safe_symlinks;
+               if (safe_symlinks && unsafe_symlink(file->u.link, keep_name)) {
+                       if (verbose) {
+                               rprintf(FINFO, "ignoring unsafe symlink %s -> %s\n",
+                                       full_fname(keep_name), file->u.link);
+                       }
+                       kept = 1;
+               }
+               make_bak_dir(fname, backup_dir);
+               if (do_symlink(file->u.link, keep_name) != 0) {
+                       rprintf(FERROR, "link %s -> %s : %s\n",
+                               full_fname(keep_name), file->u.link, strerror(errno));
+               }
+               do_unlink(fname);
+               kept = 1;
+       }
+#endif
+       if (!kept && preserve_hard_links && file->link_u.links
+           && hlink_list[file->F_HLINDEX] != file) {
+               if (verbose > 1)
+                       rprintf(FINFO, "%s is a hard link\n", f_name(file));
+       }
+
+       if (!kept && !S_ISREG(file->mode)) {
+               rprintf(FINFO, "make_bak: skipping non-regular file %s\n",
+                       fname);
+       }
+
+       /* move to keep tree if a file */
+       if (!kept) {
+               if (!robust_move(fname, keep_name)) {
+                       rprintf(FERROR, "keep_backup failed: %s -> \"%s\": %s\n",
+                               full_fname(fname), keep_name, strerror(errno));
+               }
+       }
+       set_perms(keep_name, file, NULL, 0);
+       free(file);
+
+       if (verbose > 1)
+               rprintf(FINFO, "keep_backup %s -> %s\n", fname, keep_name);
        return 1;
+} /* keep_backup */
+
+
+/* main backup switch routine */
+int make_backup(char *fname)
+{
+       if (backup_dir)
+               return keep_backup(fname);
+       return make_simple_backup(fname);
 }