Tweaked some whitespace to match the latest version from autoconf.
[rsync/rsync.git] / syscall.c
1 /* 
2    Copyright (C) Andrew Tridgell 1998
3    Copyright (C) 2002 by Martin Pool
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 /**
21  * @file syscall.c
22  *
23  * Syscall wrappers to ensure that nothing gets done in dry_run mode
24  * and to handle system peculiarities.
25  **/
26
27 #include "rsync.h"
28
29 #if !defined MKNOD_CREATES_SOCKETS && defined HAVE_SYS_UN_H
30 #include <sys/un.h>
31 #endif
32
33 extern int dry_run;
34 extern int read_only;
35 extern int list_only;
36 extern int preserve_perms;
37
38 #define RETURN_ERROR_IF(x,e) \
39         do { \
40                 if (x) { \
41                         errno = (e); \
42                         return -1; \
43                 } \
44         } while (0)
45
46 #define RETURN_ERROR_IF_RO_OR_LO RETURN_ERROR_IF(read_only || list_only, EROFS)
47
48 int do_unlink(const char *fname)
49 {
50         if (dry_run) return 0;
51         RETURN_ERROR_IF_RO_OR_LO;
52         return unlink(fname);
53 }
54
55 int do_symlink(const char *fname1, const char *fname2)
56 {
57         if (dry_run) return 0;
58         RETURN_ERROR_IF_RO_OR_LO;
59         return symlink(fname1, fname2);
60 }
61
62 #ifdef HAVE_LINK
63 int do_link(const char *fname1, const char *fname2)
64 {
65         if (dry_run) return 0;
66         RETURN_ERROR_IF_RO_OR_LO;
67         return link(fname1, fname2);
68 }
69 #endif
70
71 int do_lchown(const char *path, uid_t owner, gid_t group)
72 {
73         if (dry_run) return 0;
74         RETURN_ERROR_IF_RO_OR_LO;
75 #ifndef HAVE_LCHOWN
76 #define lchown chown
77 #endif
78         return lchown(path, owner, group);
79 }
80
81 int do_mknod(char *pathname, mode_t mode, dev_t dev)
82 {
83         if (dry_run) return 0;
84         RETURN_ERROR_IF_RO_OR_LO;
85 #if !defined MKNOD_CREATES_FIFOS && defined HAVE_MKFIFO
86         if (S_ISFIFO(mode))
87                 return mkfifo(pathname, mode);
88 #endif
89 #if !defined MKNOD_CREATES_SOCKETS && defined HAVE_SYS_UN_H
90         if (S_ISSOCK(mode)) {
91                 int sock;
92                 struct sockaddr_un saddr;
93                 unsigned int len;
94
95                 saddr.sun_family = AF_UNIX;
96                 len = strlcpy(saddr.sun_path, pathname, sizeof saddr.sun_path);
97 #ifdef HAVE_SOCKADDR_UN_LEN
98                 saddr.sun_len = len >= sizeof saddr.sun_path
99                               ? sizeof saddr.sun_path : len + 1;
100 #endif
101
102                 if ((sock = socket(PF_UNIX, SOCK_STREAM, 0)) < 0
103                     || (unlink(pathname) < 0 && errno != ENOENT)
104                     || (bind(sock, (struct sockaddr*)&saddr, sizeof saddr)) < 0)
105                         return -1;
106                 close(sock);
107 #ifdef HAVE_CHMOD
108                 return do_chmod(pathname, mode);
109 #else
110                 return 0;
111 #endif
112         }
113 #endif
114 #ifdef HAVE_MKNOD
115         return mknod(pathname, mode, dev);
116 #else
117         return -1;
118 #endif
119 }
120
121 int do_rmdir(const char *pathname)
122 {
123         if (dry_run) return 0;
124         RETURN_ERROR_IF_RO_OR_LO;
125         return rmdir(pathname);
126 }
127
128 int do_open(const char *pathname, int flags, mode_t mode)
129 {
130         if (flags != O_RDONLY) {
131                 RETURN_ERROR_IF(dry_run, 0);
132                 RETURN_ERROR_IF_RO_OR_LO;
133         }
134
135         return open(pathname, flags | O_BINARY, mode);
136 }
137
138 #ifdef HAVE_CHMOD
139 int do_chmod(const char *path, mode_t mode)
140 {
141         int code;
142         if (dry_run) return 0;
143         RETURN_ERROR_IF_RO_OR_LO;
144         if (S_ISLNK(mode)) {
145 #ifdef HAVE_LCHMOD
146                 code = lchmod(path, mode & CHMOD_BITS);
147 #else
148                 code = 1;
149 #endif
150         } else
151                 code = chmod(path, mode & CHMOD_BITS);
152         if (code != 0 && preserve_perms)
153             return code;
154         return 0;
155 }
156 #endif
157
158 int do_rename(const char *fname1, const char *fname2)
159 {
160         if (dry_run) return 0;
161         RETURN_ERROR_IF_RO_OR_LO;
162         return rename(fname1, fname2);
163 }
164
165 void trim_trailing_slashes(char *name)
166 {
167         int l;
168         /* Some BSD systems cannot make a directory if the name
169          * contains a trailing slash.
170          * <http://www.opensource.apple.com/bugs/X/BSD%20Kernel/2734739.html> */
171         
172         /* Don't change empty string; and also we can't improve on
173          * "/" */
174         
175         l = strlen(name);
176         while (l > 1) {
177                 if (name[--l] != '/')
178                         break;
179                 name[l] = '\0';
180         }
181 }
182
183 int do_mkdir(char *fname, mode_t mode)
184 {
185         if (dry_run) return 0;
186         RETURN_ERROR_IF_RO_OR_LO;
187         trim_trailing_slashes(fname);   
188         return mkdir(fname, mode);
189 }
190
191 /* like mkstemp but forces permissions */
192 int do_mkstemp(char *template, mode_t perms)
193 {
194         RETURN_ERROR_IF(dry_run, 0);
195         RETURN_ERROR_IF(read_only, EROFS);
196
197 #if defined HAVE_SECURE_MKSTEMP && defined HAVE_FCHMOD && (!defined HAVE_OPEN64 || defined HAVE_MKSTEMP64)
198         {
199                 int fd = mkstemp(template);
200                 if (fd == -1)
201                         return -1;
202                 if (fchmod(fd, perms) != 0 && preserve_perms) {
203                         int errno_save = errno;
204                         close(fd);
205                         unlink(template);
206                         errno = errno_save;
207                         return -1;
208                 }
209 #if defined HAVE_SETMODE && O_BINARY
210                 setmode(fd, O_BINARY);
211 #endif
212                 return fd;
213         }
214 #else
215         if (!mktemp(template))
216                 return -1;
217         return do_open(template, O_RDWR|O_EXCL|O_CREAT, perms);
218 #endif
219 }
220
221 int do_stat(const char *fname, STRUCT_STAT *st)
222 {
223 #ifdef USE_STAT64_FUNCS
224         return stat64(fname, st);
225 #else
226         return stat(fname, st);
227 #endif
228 }
229
230 int do_lstat(const char *fname, STRUCT_STAT *st)
231 {
232 #ifdef SUPPORT_LINKS
233 # ifdef USE_STAT64_FUNCS
234         return lstat64(fname, st);
235 # else
236         return lstat(fname, st);
237 # endif
238 #else
239         return do_stat(fname, st);
240 #endif
241 }
242
243 int do_fstat(int fd, STRUCT_STAT *st)
244 {
245 #ifdef USE_STAT64_FUNCS
246         return fstat64(fd, st);
247 #else
248         return fstat(fd, st);
249 #endif
250 }
251
252 OFF_T do_lseek(int fd, OFF_T offset, int whence)
253 {
254 #ifdef HAVE_LSEEK64
255 #if !SIZEOF_OFF64_T
256         OFF_T lseek64();
257 #else
258         off64_t lseek64();
259 #endif
260         return lseek64(fd, offset, whence);
261 #else
262         return lseek(fd, offset, whence);
263 #endif
264 }
265
266 char *d_name(struct dirent *di)
267 {
268 #ifdef HAVE_BROKEN_READDIR
269         return (di->d_name - 2);
270 #else
271         return di->d_name;
272 #endif
273 }