Make idev, hlink and file_struct + strings use allocation
[rsync/rsync.git] / backup.c
index 24f8523..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;
 
 /* 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;
 }
@@ -63,14 +72,67 @@ static int make_dir(char *name, int mask)
        *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))
+       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)
 {
@@ -79,33 +141,35 @@ static int robust_move(char *src, char *dst)
        int failed;
 
        while (keep_trying) {
-               if (keep_path_extfs)
-                       failed = copy_file (src, dst, 0755);
-               else
-                       failed = robust_rename (src, dst);
+               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);
+                       if (verbose > 2) {
+                               rprintf(FERROR, "robust_move failed: %s(%d)\n",
+                                       strerror(errno), errno);
+                       }
                        switch (errno) {
-                               /* external filesystem */
-                               case EXDEV:
-                                       keep_path_extfs = 1;
-                                       keep_trying--;
-                                       break;
-                               /* no directory to write to */
-                               case ENOENT:
-                                       make_dir (dst, 0755);
-                                       keep_trying--;
-                                       break;
-                               default:
-                                       keep_trying = 0;
-                       } /* switch */
+                       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);
+       return !failed;
 } /* robust_move */
 
 
@@ -114,56 +178,115 @@ static int robust_move(char *src, char *dst)
 static int keep_backup(char *fname)
 {
        static int initialised;
-
-       char keep_name [MAXPATHLEN];
+       char keep_name[MAXPATHLEN];
        STRUCT_STAT st;
        struct file_struct *file;
+       int kept = 0;
+       int ret_code;
 
        if (!initialised) {
-               if (backup_dir[strlen(backup_dir) - 1] == '/')
-                       backup_dir[strlen(backup_dir) - 1] = 0;
+               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);
+                       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;
+       if (do_lstat(fname, &st)) return 1;
 #else
-       if (do_stat (fname, &st)) return 1;
+       if (do_stat(fname, &st)) return 1;
 #endif
 
-       file = make_file (0, fname);
+       file = make_file(fname, NULL, NO_EXCLUDES);
+
+       /* the file could have disappeared */
+       if (!file) return 1;
 
        /* make a complete pathname for backup file */
-       if (strlen(backup_dir) + strlen(fname) > (MAXPATHLEN - 1)) {
-               rprintf (FERROR, "keep_backup filename too long\n");
+       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;
        }
-       slprintf(keep_name, sizeof (keep_name), "%s/%s", backup_dir, fname);
 
-       if (!S_ISDIR(file->mode)) {
-       /* move to keep tree if a file */
-               if (!robust_move (fname, keep_name))
-                       rprintf(FERROR, "keep_backup failed %s -> %s : %s\n",
-                               fname, keep_name, strerror(errno));
-       } else {
-       /* this bit only used to "keep" empty directories */
-               /* make the parent directories */
-               make_dir (keep_name, 0755);
-               /* now make the (empty) directory */
-               do_mkdir (keep_name, file->mode);
+#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, "keep_backup: made empty dir: %s\n",
-                               keep_name);
+                       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);
        }
 
-       set_perms (keep_name, file, NULL, 0);
-       free_file (file);
-       free (file);
+       /* 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);
+               rprintf(FINFO, "keep_backup %s -> %s\n", fname, keep_name);
        return 1;
 } /* keep_backup */
 
@@ -172,8 +295,6 @@ static int keep_backup(char *fname)
 int make_backup(char *fname)
 {
        if (backup_dir)
-               return (keep_backup(fname));
-       else
-               return (make_simple_backup(fname));
+               return keep_backup(fname);
+       return make_simple_backup(fname);
 }
-