Commit | Line | Data |
---|---|---|
7a2fd68b | 1 | /* |
c627d613 AT |
2 | Copyright (C) Andrew Tridgell 1996 |
3 | Copyright (C) Paul Mackerras 1996 | |
7a2fd68b | 4 | |
c627d613 AT |
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. | |
7a2fd68b | 9 | |
c627d613 AT |
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. | |
7a2fd68b | 14 | |
c627d613 AT |
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 | ||
2f03f956 AT |
20 | /* this file contains code used by more than one part of the rsync |
21 | process */ | |
c627d613 | 22 | |
2f03f956 | 23 | #include "rsync.h" |
43a481dc | 24 | |
c627d613 | 25 | extern int verbose; |
c627d613 | 26 | extern int dry_run; |
2f03f956 | 27 | extern int preserve_times; |
7b8356d0 | 28 | extern int am_root; |
c3e5e585 WD |
29 | extern int am_sender; |
30 | extern int am_generator; | |
2f03f956 AT |
31 | extern int preserve_uid; |
32 | extern int preserve_gid; | |
ee1df1cc WD |
33 | extern int force_delete; |
34 | extern int recurse; | |
81b07870 | 35 | extern int keep_dirlinks; |
2f03f956 | 36 | extern int make_backups; |
ee1df1cc | 37 | extern char *backup_dir; |
a3221d2a | 38 | extern int inplace; |
fe055c71 AT |
39 | |
40 | ||
c627d613 AT |
41 | /* |
42 | free a sums struct | |
43 | */ | |
2f03f956 | 44 | void free_sums(struct sum_struct *s) |
c627d613 | 45 | { |
298c10d5 AT |
46 | if (s->sums) free(s->sums); |
47 | free(s); | |
c627d613 AT |
48 | } |
49 | ||
50 | ||
3cb6f5d6 | 51 | /* |
7a2fd68b WD |
52 | * delete a file or directory. If force_delete is set then delete |
53 | * recursively | |
3cb6f5d6 | 54 | */ |
2f03f956 | 55 | int delete_file(char *fname) |
3cb6f5d6 AT |
56 | { |
57 | DIR *d; | |
58 | struct dirent *di; | |
59 | char buf[MAXPATHLEN]; | |
bcacc18b | 60 | STRUCT_STAT st; |
3cb6f5d6 AT |
61 | int ret; |
62 | ||
3cb6f5d6 | 63 | #if SUPPORT_LINKS |
bcacc18b | 64 | ret = do_lstat(fname, &st); |
3cb6f5d6 | 65 | #else |
bcacc18b | 66 | ret = do_stat(fname, &st); |
3cb6f5d6 | 67 | #endif |
a7ed6ca6 | 68 | if (ret) |
3cb6f5d6 | 69 | return -1; |
3cb6f5d6 AT |
70 | |
71 | if (!S_ISDIR(st.st_mode)) { | |
a7ed6ca6 WD |
72 | if (robust_unlink(fname) == 0 || errno == ENOENT) |
73 | return 0; | |
d62bcc17 WD |
74 | rsyserr(FERROR, errno, "delete_file: unlink %s failed", |
75 | full_fname(fname)); | |
3cb6f5d6 AT |
76 | return -1; |
77 | } | |
78 | ||
a7ed6ca6 WD |
79 | if (do_rmdir(fname) == 0 || errno == ENOENT) |
80 | return 0; | |
81 | if (!force_delete || !recurse | |
82 | || (errno != ENOTEMPTY && errno != EEXIST)) { | |
d62bcc17 WD |
83 | rsyserr(FERROR, errno, "delete_file: rmdir %s failed", |
84 | full_fname(fname)); | |
3cb6f5d6 AT |
85 | return -1; |
86 | } | |
87 | ||
88 | /* now we do a recsursive delete on the directory ... */ | |
a7ed6ca6 | 89 | if (!(d = opendir(fname))) { |
d62bcc17 WD |
90 | rsyserr(FERROR, errno, "delete_file: opendir %s failed", |
91 | full_fname(fname)); | |
3cb6f5d6 AT |
92 | return -1; |
93 | } | |
94 | ||
6a7cc46c | 95 | for (errno = 0, di = readdir(d); di; errno = 0, di = readdir(d)) { |
d6e6ecbd | 96 | char *dname = d_name(di); |
a7ed6ca6 WD |
97 | if (dname[0] == '.' && (dname[1] == '\0' |
98 | || (dname[1] == '.' && dname[2] == '\0'))) | |
3cb6f5d6 | 99 | continue; |
a7725e6d | 100 | pathjoin(buf, sizeof buf, fname, dname); |
3cb6f5d6 | 101 | if (verbose > 0) |
ecc81fce | 102 | rprintf(FINFO, "deleting %s\n", safe_fname(buf)); |
3cb6f5d6 AT |
103 | if (delete_file(buf) != 0) { |
104 | closedir(d); | |
105 | return -1; | |
106 | } | |
7a2fd68b | 107 | } |
6a7cc46c | 108 | if (errno) { |
d62bcc17 WD |
109 | rsyserr(FERROR, errno, "delete_file: readdir %s failed", |
110 | full_fname(fname)); | |
6a7cc46c S |
111 | closedir(d); |
112 | return -1; | |
113 | } | |
3cb6f5d6 AT |
114 | |
115 | closedir(d); | |
7a2fd68b | 116 | |
3cb6f5d6 | 117 | if (do_rmdir(fname) != 0) { |
d62bcc17 WD |
118 | rsyserr(FERROR, errno, "delete_file: rmdir %s failed", |
119 | full_fname(fname)); | |
3cb6f5d6 AT |
120 | return -1; |
121 | } | |
122 | ||
123 | return 0; | |
124 | } | |
c627d613 | 125 | |
2f03f956 | 126 | int set_perms(char *fname,struct file_struct *file,STRUCT_STAT *st, |
4ecc9e6b | 127 | int flags) |
c627d613 | 128 | { |
667e72a1 AT |
129 | int updated = 0; |
130 | STRUCT_STAT st2; | |
460f6b99 | 131 | int change_uid, change_gid; |
c627d613 | 132 | |
ee1df1cc WD |
133 | if (dry_run) |
134 | return 0; | |
c627d613 | 135 | |
667e72a1 | 136 | if (!st) { |
373ef160 | 137 | if (link_stat(fname, &st2, 0) < 0) { |
d62bcc17 WD |
138 | rsyserr(FERROR, errno, "stat %s failed", |
139 | full_fname(fname)); | |
667e72a1 AT |
140 | return 0; |
141 | } | |
142 | st = &st2; | |
143 | } | |
c627d613 | 144 | |
ee1df1cc WD |
145 | if (!preserve_times || S_ISLNK(st->st_mode) |
146 | || (make_backups && !backup_dir && S_ISDIR(st->st_mode))) | |
38443188 WD |
147 | flags |= PERMS_SKIP_MTIME; |
148 | if (!(flags & PERMS_SKIP_MTIME) | |
4ecc9e6b | 149 | && cmp_modtime(st->st_mtime, file->modtime) != 0) { |
8d249b63 | 150 | /* don't complain about not setting times on directories |
a174e1ed | 151 | * because some filesystems can't do it */ |
8d249b63 AT |
152 | if (set_modtime(fname,file->modtime) != 0 && |
153 | !S_ISDIR(st->st_mode)) { | |
d62bcc17 WD |
154 | rsyserr(FERROR, errno, "failed to set times on %s", |
155 | full_fname(fname)); | |
667e72a1 AT |
156 | return 0; |
157 | } | |
a174e1ed | 158 | updated = 1; |
667e72a1 AT |
159 | } |
160 | ||
460f6b99 | 161 | change_uid = am_root && preserve_uid && st->st_uid != file->uid; |
a60e2dca | 162 | change_gid = preserve_gid && file->gid != GID_NONE |
7352b873 | 163 | && st->st_gid != file->gid; |
460f6b99 | 164 | if (change_uid || change_gid) { |
62c9e6b3 | 165 | if (verbose > 2) { |
ce37eb2d WD |
166 | if (change_uid) { |
167 | rprintf(FINFO, | |
168 | "set uid of %s from %ld to %ld\n", | |
169 | fname, (long)st->st_uid, (long)file->uid); | |
170 | } | |
171 | if (change_gid) { | |
172 | rprintf(FINFO, | |
173 | "set gid of %s from %ld to %ld\n", | |
174 | fname, (long)st->st_gid, (long)file->gid); | |
175 | } | |
176 | } | |
667e72a1 | 177 | if (do_lchown(fname, |
a174e1ed WD |
178 | change_uid ? file->uid : st->st_uid, |
179 | change_gid ? file->gid : st->st_gid) != 0) { | |
460f6b99 | 180 | /* shouldn't have attempted to change uid or gid |
a174e1ed | 181 | * unless have the privilege */ |
d62bcc17 | 182 | rsyserr(FERROR, errno, "%s %s failed", |
a174e1ed | 183 | change_uid ? "chown" : "chgrp", |
d62bcc17 | 184 | full_fname(fname)); |
460f6b99 | 185 | return 0; |
667e72a1 | 186 | } |
a5827a28 | 187 | /* a lchown had been done - we have to re-stat if the |
a174e1ed WD |
188 | * destination had the setuid or setgid bits set due |
189 | * to the side effect of the chown call */ | |
a5827a28 | 190 | if (st->st_mode & (S_ISUID | S_ISGID)) { |
81b07870 WD |
191 | link_stat(fname, st, |
192 | keep_dirlinks && S_ISDIR(st->st_mode)); | |
a5827a28 | 193 | } |
460f6b99 | 194 | updated = 1; |
667e72a1 | 195 | } |
c627d613 AT |
196 | |
197 | #ifdef HAVE_CHMOD | |
972a3619 | 198 | if (!S_ISLNK(st->st_mode)) { |
b0d791bb | 199 | if ((st->st_mode & CHMOD_BITS) != (file->mode & CHMOD_BITS)) { |
972a3619 | 200 | updated = 1; |
b0d791bb | 201 | if (do_chmod(fname,(file->mode & CHMOD_BITS)) != 0) { |
d62bcc17 WD |
202 | rsyserr(FERROR, errno, "failed to set permissions on %s", |
203 | full_fname(fname)); | |
972a3619 DD |
204 | return 0; |
205 | } | |
667e72a1 AT |
206 | } |
207 | } | |
c627d613 | 208 | #endif |
6a7cc46c | 209 | |
4ecc9e6b | 210 | if (verbose > 1 && flags & PERMS_REPORT) { |
667e72a1 AT |
211 | if (updated) |
212 | rprintf(FINFO,"%s\n",fname); | |
213 | else | |
214 | rprintf(FINFO,"%s is uptodate\n",fname); | |
215 | } | |
216 | return updated; | |
c627d613 AT |
217 | } |
218 | ||
219 | ||
34ccb63e AT |
220 | void sig_int(void) |
221 | { | |
d5a0b483 WD |
222 | /* KLUGE: if the user hits Ctrl-C while ssh is prompting |
223 | * for a password, then our cleanup's sending of a SIGUSR1 | |
224 | * signal to all our children may kill ssh before it has a | |
225 | * chance to restore the tty settings (i.e. turn echo back | |
226 | * on). By sleeping for a short time, ssh gets a bigger | |
227 | * chance to do the right thing. If child processes are | |
228 | * not ssh waiting for a password, then this tiny delay | |
229 | * shouldn't hurt anything. */ | |
230 | msleep(400); | |
65417579 | 231 | exit_cleanup(RERR_SIGNAL); |
c627d613 AT |
232 | } |
233 | ||
234 | ||
c95da96a AT |
235 | /* finish off a file transfer, renaming the file and setting the permissions |
236 | and ownership */ | |
4ecc9e6b WD |
237 | void finish_transfer(char *fname, char *fnametmp, struct file_struct *file, |
238 | int ok_to_set_time) | |
c95da96a | 239 | { |
62c9e6b3 WD |
240 | int ret; |
241 | ||
a3221d2a WD |
242 | if (inplace) { |
243 | if (verbose > 2) | |
244 | rprintf(FINFO, "finishing %s\n", fname); | |
077e59b7 | 245 | goto do_set_perms; |
a3221d2a WD |
246 | } |
247 | ||
e484f0cc WD |
248 | if (make_backups && !make_backup(fname)) |
249 | return; | |
250 | ||
ebeacb36 WD |
251 | /* Change permissions before putting the file into place. */ |
252 | set_perms(fnametmp, file, NULL, ok_to_set_time ? 0 : PERMS_SKIP_MTIME); | |
253 | ||
c95da96a | 254 | /* move tmp file over real file */ |
da38e779 WD |
255 | if (verbose > 2) |
256 | rprintf(FINFO, "renaming %s to %s\n", fnametmp, fname); | |
62c9e6b3 | 257 | ret = robust_rename(fnametmp, fname, file->mode & INITACCESSPERMS); |
3d061653 | 258 | if (ret < 0) { |
d62bcc17 | 259 | rsyserr(FERROR, errno, "%s %s -> \"%s\"", |
62c9e6b3 | 260 | ret == -2 ? "copy" : "rename", |
d62bcc17 | 261 | full_fname(fnametmp), fname); |
c7c11a0d | 262 | do_unlink(fnametmp); |
077e59b7 | 263 | return; |
c95da96a | 264 | } |
ebeacb36 WD |
265 | if (ret == 0) { |
266 | /* The file was moved into place (not copied), so it's done. */ | |
267 | return; | |
268 | } | |
077e59b7 WD |
269 | do_set_perms: |
270 | set_perms(fname, file, NULL, ok_to_set_time ? 0 : PERMS_SKIP_MTIME); | |
c95da96a | 271 | } |
c3e5e585 WD |
272 | |
273 | const char *who_am_i(void) | |
274 | { | |
ebeacb36 | 275 | return am_sender ? "sender" : am_generator ? "generator" : "receiver"; |
c3e5e585 | 276 | } |