minor optimizations
[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 <sys/stat.h>
31
32 bool
33 Unify_unify(char const *src, struct stat const UNUSED *src_stat,
34             char const *dst)
35 {
36   size_t        l = strlen(dst);
37   char          tmpfile[l + sizeof(";XXXXXX")];
38   int           fd;
39   bool          res = false;
40   struct stat   st;
41
42   // at first, set the ILI flags on 'src'
43   if (vc_set_iattr(src,
44                    0,
45                    VC_IATTR_IUNLINK|VC_IATTR_IMMUTABLE,
46                    VC_IATTR_IUNLINK|VC_IATTR_IMMUTABLE)==-1)
47     return false;
48
49   // check if 'dst' already exists
50   if (lstat(dst, &st)==0) {
51       // now, create a temporary filename
52     memcpy(tmpfile,   dst, l);
53     memcpy(tmpfile+l, ";XXXXXX", 8);
54     fd = mkstemp(tmpfile);
55     close(fd);
56
57     if (fd==-1) {
58       perror("mkstemp()");
59       return false;
60     }
61
62       // and rename the old file to this name
63
64       // NOTE: this rename() is race-free; when an attacker makes 'tmpfile' a
65       // directory, the operation would fail; when making it a symlink to a file
66       // or directory, the symlink but not the file/directory would be overridden
67     if (rename(dst, tmpfile)==-1) {
68       perror("rename()");
69       goto err;
70     }
71   }
72   else
73     tmpfile[0] = '\0';
74
75   // now, link the src-file to dst
76   if (link(src, dst)==-1) {
77     perror("link()");
78
79     unlink(dst);
80     if (tmpfile[0]!='\0' &&
81         rename(tmpfile, dst)==-1) {
82       perror("FATAL error in rename()");
83       _exit(1);
84     }
85     goto err;
86   }
87
88   res = true;
89
90   err:
91   if (tmpfile[0]!='\0')
92     unlink(tmpfile);
93
94   return res;
95 }