added an 'ignore_zero' argument to Unify_unify()
[util-vserver.git] / util-vserver / lib_internal / unify-unify.c
1 // $Id$    --*- c -*--
2
3 // Copyright (C) 2004 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
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; version 2 of the License.
8 //  
9 // This program is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12 // GNU General Public License for more details.
13 //  
14 // You should have received a copy of the GNU General Public License
15 // along with this program; if not, write to the Free Software
16 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
17
18
19 #ifdef HAVE_CONFIG_H
20 #  include <config.h>
21 #endif
22
23 #include "unify.h"
24 #include "vserver.h"
25
26 #include <stdlib.h>
27 #include <string.h>
28 #include <stdio.h>
29 #include <unistd.h>
30 #include <errno.h>
31 #include <sys/stat.h>
32
33 bool
34 Unify_unify(char const *src, struct stat const UNUSED *src_stat,
35             char const *dst, bool ignore_zero)
36 {
37   size_t        l = strlen(dst);
38   char          tmpfile[l + sizeof(";XXXXXX")];
39   int           fd;
40   bool          res = false;
41   struct stat   st;
42   bool          lstat_succeeded;
43
44   // at first, set the ILI flags on 'src'
45   if (vc_set_iattr(src,
46                    0,
47                    VC_IATTR_IUNLINK|VC_IATTR_IMMUTABLE,
48                    VC_IATTR_IUNLINK|VC_IATTR_IMMUTABLE)==-1)
49     return false;
50
51   lstat_succeeded = lstat(dst, &st)==0;
52
53   // check if 'dst' already exists
54   // when ignore_zero is true, do not make backups of empty destinations
55   if (lstat_succeeded && (st.st_size>0 || !ignore_zero)) {
56       // now, create a temporary filename
57     memcpy(tmpfile,   dst, l);
58     memcpy(tmpfile+l, ";XXXXXX", 8);
59     fd = mkstemp(tmpfile);
60     close(fd);
61
62     if (fd==-1) {
63       perror("mkstemp()");
64       return false;
65     }
66
67       // and rename the old file to this name
68
69       // NOTE: this rename() is race-free; when an attacker makes 'tmpfile' a
70       // directory, the operation would fail; when making it a symlink to a file
71       // or directory, the symlink but not the file/directory would be overridden
72     if (rename(dst, tmpfile)==-1) {
73       perror("rename()");
74       goto err;
75     }
76   }
77   else {
78     if (lstat_succeeded) unlink(dst);   
79     tmpfile[0] = '\0';
80   }
81
82   // now, link the src-file to dst
83   if (link(src, dst)==-1) {
84     perror("link()");
85
86     unlink(dst);
87     if (tmpfile[0]!='\0' &&
88         rename(tmpfile, dst)==-1) {
89       perror("FATAL error in rename()");
90       _exit(1);
91     }
92     goto err;
93   }
94
95   res = true;
96
97   err:
98   if (tmpfile[0]!='\0') {
99     int old_errno = errno;
100     unlink(tmpfile);
101     errno = old_errno;
102   }
103
104   return res;
105 }