several changes:
[rsync/rsync.git] / rsync.c
CommitLineData
c627d613
AT
1/*
2 Copyright (C) Andrew Tridgell 1996
3 Copyright (C) Paul Mackerras 1996
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#include "rsync.h"
21
43a481dc
AT
22extern int csum_length;
23
c627d613
AT
24extern int verbose;
25extern int am_server;
26extern int always_checksum;
27extern time_t starttime;
28
c6e7fcb4
AT
29extern int remote_version;
30
c627d613
AT
31extern char *backup_suffix;
32
33extern int block_size;
34extern int update_only;
35extern int make_backups;
36extern int preserve_links;
dc5ddbcc 37extern int preserve_hard_links;
c627d613
AT
38extern int preserve_perms;
39extern int preserve_devices;
40extern int preserve_uid;
41extern int preserve_gid;
42extern int preserve_times;
43extern int dry_run;
44extern int ignore_times;
45extern int recurse;
46extern int delete_mode;
47extern int cvs_exclude;
7b8356d0 48extern int am_root;
6574b4f7 49extern int relative_paths;
c627d613
AT
50
51/*
52 free a sums struct
53 */
54static void free_sums(struct sum_struct *s)
55{
56 if (s->sums) free(s->sums);
57 free(s);
58}
59
60
61
62/*
63 send a sums struct down a fd
64 */
65static void send_sums(struct sum_struct *s,int f_out)
66{
67 int i;
68
69 /* tell the other guy how many we are going to be doing and how many
70 bytes there are in the last chunk */
71 write_int(f_out,s?s->count:0);
72 write_int(f_out,s?s->n:block_size);
73 write_int(f_out,s?s->remainder:0);
74 if (s)
75 for (i=0;i<s->count;i++) {
76 write_int(f_out,s->sums[i].sum1);
43a481dc 77 write_buf(f_out,s->sums[i].sum2,csum_length);
c627d613
AT
78 }
79 write_flush(f_out);
80}
81
82
83/*
84 generate a stream of signatures/checksums that describe a buffer
85
86 generate approximately one checksum every n bytes
87 */
c6e7fcb4 88static struct sum_struct *generate_sums(struct map_struct *buf,off_t len,int n)
c627d613
AT
89{
90 int i;
91 struct sum_struct *s;
92 int count;
93 int block_len = n;
94 int remainder = (len%block_len);
95 off_t offset = 0;
96
97 count = (len+(block_len-1))/block_len;
98
99 s = (struct sum_struct *)malloc(sizeof(*s));
100 if (!s) out_of_memory("generate_sums");
101
102 s->count = count;
103 s->remainder = remainder;
104 s->n = n;
105 s->flength = len;
106
107 if (count==0) {
108 s->sums = NULL;
109 return s;
110 }
111
112 if (verbose > 3)
dc5ddbcc 113 fprintf(FERROR,"count=%d rem=%d n=%d flength=%d\n",
c627d613
AT
114 s->count,s->remainder,s->n,(int)s->flength);
115
116 s->sums = (struct sum_buf *)malloc(sizeof(s->sums[0])*s->count);
117 if (!s->sums) out_of_memory("generate_sums");
118
119 for (i=0;i<count;i++) {
120 int n1 = MIN(len,n);
d9bea2dd 121 char *map = map_ptr(buf,offset,n1);
c627d613 122
d9bea2dd
AT
123 s->sums[i].sum1 = get_checksum1(map,n1);
124 get_checksum2(map,n1,s->sums[i].sum2);
c627d613
AT
125
126 s->sums[i].offset = offset;
127 s->sums[i].len = n1;
128 s->sums[i].i = i;
129
130 if (verbose > 3)
dc5ddbcc 131 fprintf(FERROR,"chunk[%d] offset=%d len=%d sum1=%08x\n",
c627d613
AT
132 i,(int)s->sums[i].offset,s->sums[i].len,s->sums[i].sum1);
133
134 len -= n1;
c627d613
AT
135 offset += n1;
136 }
137
138 return s;
139}
140
141
142/*
143 receive the checksums for a buffer
144 */
145static struct sum_struct *receive_sums(int f)
146{
147 struct sum_struct *s;
148 int i;
149 off_t offset = 0;
c627d613
AT
150
151 s = (struct sum_struct *)malloc(sizeof(*s));
152 if (!s) out_of_memory("receive_sums");
153
154 s->count = read_int(f);
155 s->n = read_int(f);
156 s->remainder = read_int(f);
157 s->sums = NULL;
158
159 if (verbose > 3)
dc5ddbcc 160 fprintf(FERROR,"count=%d n=%d rem=%d\n",
c627d613
AT
161 s->count,s->n,s->remainder);
162
c627d613
AT
163 if (s->count == 0)
164 return(s);
165
166 s->sums = (struct sum_buf *)malloc(sizeof(s->sums[0])*s->count);
167 if (!s->sums) out_of_memory("receive_sums");
168
169 for (i=0;i<s->count;i++) {
170 s->sums[i].sum1 = read_int(f);
43a481dc 171 read_buf(f,s->sums[i].sum2,csum_length);
c627d613
AT
172
173 s->sums[i].offset = offset;
174 s->sums[i].i = i;
175
176 if (i == s->count-1 && s->remainder != 0) {
177 s->sums[i].len = s->remainder;
178 } else {
179 s->sums[i].len = s->n;
180 }
181 offset += s->sums[i].len;
182
183 if (verbose > 3)
dc5ddbcc 184 fprintf(FERROR,"chunk[%d] len=%d offset=%d sum1=%08x\n",
c627d613
AT
185 i,s->sums[i].len,(int)s->sums[i].offset,s->sums[i].sum1);
186 }
187
188 s->flength = offset;
189
190 return s;
191}
192
193
7b8356d0
AT
194static int set_perms(char *fname,struct file_struct *file,struct stat *st,
195 int report)
c627d613
AT
196{
197 int updated = 0;
198 struct stat st2;
199
7b8356d0 200 if (dry_run) return 0;
c627d613
AT
201
202 if (!st) {
7b8356d0 203 if (lstat(fname,&st2) != 0) {
dc5ddbcc 204 fprintf(FERROR,"stat %s : %s\n",fname,strerror(errno));
7b8356d0 205 return 0;
c627d613
AT
206 }
207 st = &st2;
208 }
209
7bec6a5c
AT
210 if (preserve_times && !S_ISLNK(st->st_mode) &&
211 st->st_mtime != file->modtime) {
c627d613
AT
212 updated = 1;
213 if (set_modtime(fname,file->modtime) != 0) {
dc5ddbcc 214 fprintf(FERROR,"failed to set times on %s : %s\n",
c627d613 215 fname,strerror(errno));
7b8356d0 216 return 0;
c627d613
AT
217 }
218 }
219
220#ifdef HAVE_CHMOD
7bec6a5c
AT
221 if (preserve_perms && !S_ISLNK(st->st_mode) &&
222 st->st_mode != file->mode) {
c627d613
AT
223 updated = 1;
224 if (chmod(fname,file->mode) != 0) {
dc5ddbcc 225 fprintf(FERROR,"failed to set permissions on %s : %s\n",
c627d613 226 fname,strerror(errno));
7b8356d0 227 return 0;
c627d613
AT
228 }
229 }
230#endif
231
7b8356d0 232 if ((am_root && preserve_uid && st->st_uid != file->uid) ||
c627d613
AT
233 (preserve_gid && st->st_gid != file->gid)) {
234 updated = 1;
8bf73749
AT
235 if (lchown(fname,
236 (am_root&&preserve_uid)?file->uid:-1,
237 preserve_gid?file->gid:-1) != 0) {
c627d613 238 if (verbose>1 || preserve_uid)
dc5ddbcc 239 fprintf(FERROR,"chown %s : %s\n",fname,strerror(errno));
7b8356d0 240 return updated;
c627d613
AT
241 }
242 }
243
244 if (verbose > 1 && report) {
245 if (updated)
dc5ddbcc 246 fprintf(FINFO,"%s\n",fname);
c627d613 247 else
dc5ddbcc 248 fprintf(FINFO,"%s is uptodate\n",fname);
c627d613 249 }
7b8356d0 250 return updated;
c627d613
AT
251}
252
253
254void recv_generator(char *fname,struct file_list *flist,int i,int f_out)
255{
256 int fd;
257 struct stat st;
c6e7fcb4 258 struct map_struct *buf;
c627d613 259 struct sum_struct *s;
ebb0a6f6 260 char sum[MD4_SUM_LENGTH];
c627d613 261 int statret;
dc5ddbcc 262 struct file_struct *file = &flist->files[i];
c627d613
AT
263
264 if (verbose > 2)
9e31c482 265 fprintf(FERROR,"recv_generator(%s,%d)\n",fname,i);
c627d613
AT
266
267 statret = lstat(fname,&st);
268
7b8356d0
AT
269 if (S_ISDIR(file->mode)) {
270 if (dry_run) return;
271 if (statret == 0 && !S_ISDIR(st.st_mode)) {
272 if (unlink(fname) != 0) {
273 fprintf(FERROR,"unlink %s : %s\n",fname,strerror(errno));
274 return;
275 }
276 statret = -1;
277 }
6574b4f7
AT
278 if (statret != 0 && mkdir(fname,file->mode) != 0 && errno != EEXIST) {
279 if (!(relative_paths && errno==ENOENT &&
280 create_directory_path(fname)==0 &&
281 mkdir(fname,file->mode)==0)) {
282 fprintf(FERROR,"mkdir %s : %s (2)\n",
283 fname,strerror(errno));
284 }
285 }
7b8356d0
AT
286 if (set_perms(fname,file,NULL,0) && verbose)
287 fprintf(FINFO,"%s/\n",fname);
288 return;
289 }
290
c627d613 291#if SUPPORT_LINKS
dc5ddbcc 292 if (preserve_links && S_ISLNK(file->mode)) {
c627d613
AT
293 char lnk[MAXPATHLEN];
294 int l;
295 if (statret == 0) {
296 l = readlink(fname,lnk,MAXPATHLEN-1);
297 if (l > 0) {
298 lnk[l] = 0;
dc5ddbcc
AT
299 if (strcmp(lnk,file->link) == 0) {
300 set_perms(fname,file,&st,1);
c627d613
AT
301 return;
302 }
303 }
304 }
305 if (!dry_run) unlink(fname);
dc5ddbcc
AT
306 if (!dry_run && symlink(file->link,fname) != 0) {
307 fprintf(FERROR,"link %s -> %s : %s\n",
308 fname,file->link,strerror(errno));
c627d613 309 } else {
dc5ddbcc 310 set_perms(fname,file,NULL,0);
c627d613 311 if (verbose)
dc5ddbcc
AT
312 fprintf(FINFO,"%s -> %s\n",
313 fname,file->link);
c627d613
AT
314 }
315 return;
316 }
317#endif
318
319#ifdef HAVE_MKNOD
7b8356d0 320 if (am_root && preserve_devices && IS_DEVICE(file->mode)) {
c627d613 321 if (statret != 0 ||
dc5ddbcc
AT
322 st.st_mode != file->mode ||
323 st.st_rdev != file->rdev) {
c627d613
AT
324 if (!dry_run) unlink(fname);
325 if (verbose > 2)
dc5ddbcc
AT
326 fprintf(FERROR,"mknod(%s,0%o,0x%x)\n",
327 fname,(int)file->mode,(int)file->rdev);
c627d613 328 if (!dry_run &&
dc5ddbcc
AT
329 mknod(fname,file->mode,file->rdev) != 0) {
330 fprintf(FERROR,"mknod %s : %s\n",fname,strerror(errno));
c627d613 331 } else {
dc5ddbcc 332 set_perms(fname,file,NULL,0);
c627d613 333 if (verbose)
dc5ddbcc 334 fprintf(FINFO,"%s\n",fname);
c627d613
AT
335 }
336 } else {
dc5ddbcc 337 set_perms(fname,file,&st,1);
c627d613
AT
338 }
339 return;
340 }
341#endif
342
dc5ddbcc
AT
343 if (preserve_hard_links && check_hard_link(file)) {
344 if (verbose > 1)
345 fprintf(FINFO,"%s is a hard link\n",file->name);
346 return;
347 }
348
349 if (!S_ISREG(file->mode)) {
350 fprintf(FERROR,"skipping non-regular file %s\n",fname);
c627d613
AT
351 return;
352 }
353
354 if (statret == -1) {
355 if (errno == ENOENT) {
356 write_int(f_out,i);
357 if (!dry_run) send_sums(NULL,f_out);
358 } else {
359 if (verbose > 1)
dc5ddbcc 360 fprintf(FERROR,"recv_generator failed to open %s\n",fname);
c627d613
AT
361 }
362 return;
363 }
364
365 if (!S_ISREG(st.st_mode)) {
7b8356d0
AT
366 /* its not a regular file on the receiving end, but it is on the
367 sending end. If its a directory then skip it (too dangerous to
368 do a recursive deletion??) otherwise try to unlink it */
369 if (S_ISDIR(st.st_mode)) {
370 fprintf(FERROR,"ERROR: %s is a directory\n",fname);
371 return;
372 }
373 if (unlink(fname) != 0) {
374 fprintf(FERROR,"%s : not a regular file (generator)\n",fname);
375 return;
376 }
377
378 /* now pretend the file didn't exist */
379 write_int(f_out,i);
380 if (!dry_run) send_sums(NULL,f_out);
c627d613
AT
381 return;
382 }
383
dc5ddbcc 384 if (update_only && st.st_mtime >= file->modtime) {
c627d613 385 if (verbose > 1)
dc5ddbcc 386 fprintf(FERROR,"%s is newer\n",fname);
c627d613
AT
387 return;
388 }
389
390 if (always_checksum && S_ISREG(st.st_mode)) {
391 file_checksum(fname,sum,st.st_size);
392 }
393
dc5ddbcc
AT
394 if (st.st_size == file->length &&
395 ((!ignore_times && st.st_mtime == file->modtime) ||
c627d613 396 (always_checksum && S_ISREG(st.st_mode) &&
dc5ddbcc
AT
397 memcmp(sum,file->sum,csum_length) == 0))) {
398 set_perms(fname,file,&st,1);
c627d613
AT
399 return;
400 }
401
402 if (dry_run) {
403 write_int(f_out,i);
404 return;
405 }
406
407 /* open the file */
408 fd = open(fname,O_RDONLY);
409
410 if (fd == -1) {
dc5ddbcc 411 fprintf(FERROR,"failed to open %s : %s\n",fname,strerror(errno));
c627d613
AT
412 return;
413 }
414
415 if (st.st_size > 0) {
416 buf = map_file(fd,st.st_size);
c627d613
AT
417 } else {
418 buf = NULL;
419 }
420
421 if (verbose > 3)
9e31c482 422 fprintf(FERROR,"gen mapped %s of size %d\n",fname,(int)st.st_size);
c627d613
AT
423
424 s = generate_sums(buf,st.st_size,block_size);
425
9e31c482
AT
426 if (verbose > 2)
427 fprintf(FERROR,"sending sums for %d\n",i);
428
c627d613
AT
429 write_int(f_out,i);
430 send_sums(s,f_out);
431 write_flush(f_out);
432
433 close(fd);
1cdc8b50 434 if (buf) unmap_file(buf);
c627d613
AT
435
436 free_sums(s);
437}
438
439
440
9e31c482 441static int receive_data(int f_in,struct map_struct *buf,int fd,char *fname)
c627d613
AT
442{
443 int i,n,remainder,len,count;
444 off_t offset = 0;
445 off_t offset2;
70d794dc 446 char *data;
ebb0a6f6
AT
447 static char file_sum1[MD4_SUM_LENGTH];
448 static char file_sum2[MD4_SUM_LENGTH];
9e31c482 449 char *map=NULL;
c627d613
AT
450
451 count = read_int(f_in);
452 n = read_int(f_in);
453 remainder = read_int(f_in);
454
9e31c482
AT
455 sum_init();
456
70d794dc 457 for (i=recv_token(f_in,&data); i != 0; i=recv_token(f_in,&data)) {
c627d613
AT
458 if (i > 0) {
459 if (verbose > 3)
dc5ddbcc 460 fprintf(FERROR,"data recv %d at %d\n",i,(int)offset);
c627d613 461
9e31c482
AT
462 sum_update(data,i);
463
7b8356d0 464 if (fd != -1 && write_sparse(fd,data,i) != i) {
dc5ddbcc 465 fprintf(FERROR,"write failed on %s : %s\n",fname,strerror(errno));
34ccb63e 466 exit_cleanup(1);
c627d613
AT
467 }
468 offset += i;
469 } else {
470 i = -(i+1);
471 offset2 = i*n;
472 len = n;
473 if (i == count-1 && remainder != 0)
474 len = remainder;
475
476 if (verbose > 3)
dc5ddbcc 477 fprintf(FERROR,"chunk[%d] of size %d at %d offset=%d\n",
c627d613
AT
478 i,len,(int)offset2,(int)offset);
479
9e31c482
AT
480 map = map_ptr(buf,offset2,len);
481
861c20b4 482 see_token(map, len);
9e31c482
AT
483 sum_update(map,len);
484
7b8356d0 485 if (fd != -1 && write_sparse(fd,map,len) != len) {
dc5ddbcc 486 fprintf(FERROR,"write failed on %s : %s\n",fname,strerror(errno));
34ccb63e 487 exit_cleanup(1);
c627d613
AT
488 }
489 offset += len;
490 }
491 }
7bec6a5c 492
7b8356d0 493 if (fd != -1 && offset > 0 && sparse_end(fd) != 0) {
dc5ddbcc 494 fprintf(FERROR,"write failed on %s : %s\n",fname,strerror(errno));
34ccb63e 495 exit_cleanup(1);
7bec6a5c 496 }
9e31c482
AT
497
498 sum_end(file_sum1);
499
500 if (remote_version >= 14) {
ebb0a6f6 501 read_buf(f_in,file_sum2,MD4_SUM_LENGTH);
9e31c482
AT
502 if (verbose > 2)
503 fprintf(FERROR,"got file_sum\n");
7b8356d0 504 if (fd != -1 && memcmp(file_sum1,file_sum2,MD4_SUM_LENGTH) != 0)
9e31c482
AT
505 return 0;
506 }
507 return 1;
c627d613
AT
508}
509
510
511static void delete_one(struct file_struct *f)
512{
513 if (!S_ISDIR(f->mode)) {
514 if (!dry_run && unlink(f->name) != 0) {
dc5ddbcc 515 fprintf(FERROR,"unlink %s : %s\n",f->name,strerror(errno));
c627d613 516 } else if (verbose) {
dc5ddbcc 517 fprintf(FERROR,"deleting %s\n",f->name);
c627d613
AT
518 }
519 } else {
520 if (!dry_run && rmdir(f->name) != 0) {
521 if (errno != ENOTEMPTY)
dc5ddbcc 522 fprintf(FERROR,"rmdir %s : %s\n",f->name,strerror(errno));
c627d613 523 } else if (verbose) {
dc5ddbcc 524 fprintf(FERROR,"deleting directory %s\n",f->name);
c627d613
AT
525 }
526 }
527}
528
529
e92338c8
AT
530/* this deletes any files on the receiving side that are not present
531 on the sending side. For version 1.6.4 I have changed the behaviour
532 to match more closely what most people seem to expect of this option */
c627d613
AT
533static void delete_files(struct file_list *flist)
534{
535 struct file_list *local_file_list;
536 char *dot=".";
e92338c8
AT
537 int i, j;
538 char *last_name=NULL;
c627d613 539
4fe159a8 540 if (cvs_exclude)
79fbb6f5 541 add_cvs_excludes();
4fe159a8 542
e92338c8
AT
543 for (j=0;j<flist->count;j++) {
544 if (!S_ISDIR(flist->files[j].mode)) continue;
545 if (strcmp(flist->files[j].name,".")==0) continue;
546 if (last_name &&
547 flist->files[j].name[strlen(last_name)] == '/' &&
548 strncmp(flist->files[j].name,last_name, strlen(last_name))==0)
549 continue;
550 last_name = flist->files[j].name;
551 if (verbose > 1)
552 fprintf(FINFO,"deleting in %s\n", last_name);
553 if (!(local_file_list = send_file_list(-1,1,&last_name)))
554 return;
555
556 for (i=local_file_list->count-1;i>=0;i--) {
557 if (!local_file_list->files[i].name) continue;
558 if (-1 == flist_find(flist,&local_file_list->files[i])) {
559 delete_one(&local_file_list->files[i]);
560 }
561 }
c627d613
AT
562 }
563}
564
565static char *cleanup_fname = NULL;
566
34ccb63e 567void exit_cleanup(int code)
c627d613
AT
568{
569 if (cleanup_fname)
570 unlink(cleanup_fname);
34ccb63e
AT
571 exit(code);
572}
573
574void sig_int(void)
575{
576 exit_cleanup(1);
c627d613
AT
577}
578
579
c6e7fcb4 580int recv_files(int f_in,struct file_list *flist,char *local_name,int f_gen)
c627d613
AT
581{
582 int fd1,fd2;
583 struct stat st;
584 char *fname;
585 char fnametmp[MAXPATHLEN];
c6e7fcb4 586 struct map_struct *buf;
c627d613 587 int i;
dc5ddbcc 588 struct file_struct *file;
c6e7fcb4 589 int phase=0;
9e31c482 590 int recv_ok;
c627d613 591
981791bd 592 if (verbose > 2) {
dc5ddbcc 593 fprintf(FERROR,"recv_files(%d) starting\n",flist->count);
981791bd 594 }
c627d613
AT
595
596 if (recurse && delete_mode && !local_name && flist->count>0) {
597 delete_files(flist);
598 }
599
600 while (1)
9e31c482 601 {
c627d613 602 i = read_int(f_in);
c6e7fcb4
AT
603 if (i == -1) {
604 if (phase==0 && remote_version >= 13) {
605 phase++;
9e31c482
AT
606 csum_length = SUM_LENGTH;
607 if (verbose > 2)
608 fprintf(FERROR,"recv_files phase=%d\n",phase);
c6e7fcb4
AT
609 write_int(f_gen,-1);
610 write_flush(f_gen);
611 continue;
612 }
613 break;
614 }
c627d613 615
dc5ddbcc
AT
616 file = &flist->files[i];
617 fname = file->name;
c627d613
AT
618
619 if (local_name)
620 fname = local_name;
621
622 if (dry_run) {
623 if (!am_server && verbose)
624 printf("%s\n",fname);
625 continue;
626 }
627
628 if (verbose > 2)
dc5ddbcc 629 fprintf(FERROR,"recv_files(%s)\n",fname);
c627d613
AT
630
631 /* open the file */
ac1eb754 632 fd1 = open(fname,O_RDONLY);
c627d613 633
ac1eb754 634 if (fd1 != -1 && fstat(fd1,&st) != 0) {
dc5ddbcc 635 fprintf(FERROR,"fstat %s : %s\n",fname,strerror(errno));
7b8356d0 636 receive_data(f_in,NULL,-1,NULL);
c627d613 637 close(fd1);
7b8356d0 638 continue;
c627d613
AT
639 }
640
ac1eb754 641 if (fd1 != -1 && !S_ISREG(st.st_mode)) {
7b8356d0
AT
642 fprintf(FERROR,"%s : not a regular file (recv_files)\n",fname);
643 receive_data(f_in,NULL,-1,NULL);
c627d613 644 close(fd1);
7b8356d0 645 continue;
c627d613
AT
646 }
647
ac1eb754 648 if (fd1 != -1 && st.st_size > 0) {
c627d613 649 buf = map_file(fd1,st.st_size);
9e31c482
AT
650 if (verbose > 2)
651 fprintf(FERROR,"recv mapped %s of size %d\n",fname,(int)st.st_size);
c627d613
AT
652 } else {
653 buf = NULL;
654 }
655
c627d613
AT
656 /* open tmp file */
657 sprintf(fnametmp,"%s.XXXXXX",fname);
658 if (NULL == mktemp(fnametmp)) {
dc5ddbcc 659 fprintf(FERROR,"mktemp %s failed\n",fnametmp);
7b8356d0
AT
660 receive_data(f_in,buf,-1,NULL);
661 if (buf) unmap_file(buf);
662 close(fd1);
663 continue;
c627d613 664 }
dc5ddbcc 665 fd2 = open(fnametmp,O_WRONLY|O_CREAT,file->mode);
6574b4f7
AT
666 if (relative_paths && errno == ENOENT &&
667 create_directory_path(fnametmp) == 0) {
668 fd2 = open(fnametmp,O_WRONLY|O_CREAT,file->mode);
669 }
c627d613 670 if (fd2 == -1) {
dc5ddbcc 671 fprintf(FERROR,"open %s : %s\n",fnametmp,strerror(errno));
7b8356d0
AT
672 receive_data(f_in,buf,-1,NULL);
673 if (buf) unmap_file(buf);
674 close(fd1);
675 continue;
c627d613
AT
676 }
677
678 cleanup_fname = fnametmp;
679
680 if (!am_server && verbose)
681 printf("%s\n",fname);
682
683 /* recv file data */
9e31c482 684 recv_ok = receive_data(f_in,buf,fd2,fname);
c627d613 685
981791bd 686 if (fd1 != -1) {
1cdc8b50 687 if (buf) unmap_file(buf);
981791bd
AT
688 close(fd1);
689 }
c627d613
AT
690 close(fd2);
691
692 if (verbose > 2)
dc5ddbcc 693 fprintf(FERROR,"renaming %s to %s\n",fnametmp,fname);
c627d613
AT
694
695 if (make_backups) {
696 char fnamebak[MAXPATHLEN];
697 sprintf(fnamebak,"%s%s",fname,backup_suffix);
ac1eb754 698 if (rename(fname,fnamebak) != 0 && errno != ENOENT) {
dc5ddbcc 699 fprintf(FERROR,"rename %s %s : %s\n",fname,fnamebak,strerror(errno));
7b8356d0 700 continue;
c627d613
AT
701 }
702 }
703
704 /* move tmp file over real file */
705 if (rename(fnametmp,fname) != 0) {
dc5ddbcc 706 fprintf(FERROR,"rename %s -> %s : %s\n",
c627d613 707 fnametmp,fname,strerror(errno));
7b8356d0 708 unlink(fnametmp);
c627d613
AT
709 }
710
711 cleanup_fname = NULL;
712
dc5ddbcc 713 set_perms(fname,file,NULL,0);
9e31c482
AT
714
715 if (!recv_ok) {
716 if (verbose > 1)
717 fprintf(FERROR,"redoing %s(%d)\n",fname,i);
52296954
AT
718 if (csum_length == SUM_LENGTH)
719 fprintf(FERROR,"ERROR: file corruption in %s\n",fname);
9e31c482
AT
720 write_int(f_gen,i);
721 }
c627d613
AT
722 }
723
7b8356d0
AT
724 /* now we need to fix any directory permissions that were
725 modified during the transfer */
726 if (!am_root) {
727 for (i = 0; i < flist->count; i++) {
728 struct file_struct *file = &flist->files[i];
729 if (!file->name || !S_ISDIR(file->mode)) continue;
730 recv_generator(file->name,flist,i,-1);
731 }
732 }
733
734
c627d613 735 if (verbose > 2)
dc5ddbcc 736 fprintf(FERROR,"recv_files finished\n");
c627d613
AT
737
738 return 0;
739}
740
741
742
743off_t send_files(struct file_list *flist,int f_out,int f_in)
744{
745 int fd;
746 struct sum_struct *s;
c6e7fcb4 747 struct map_struct *buf;
c627d613
AT
748 struct stat st;
749 char fname[MAXPATHLEN];
750 off_t total=0;
751 int i;
dc5ddbcc 752 struct file_struct *file;
c6e7fcb4 753 int phase = 0;
c627d613
AT
754
755 if (verbose > 2)
dc5ddbcc 756 fprintf(FERROR,"send_files starting\n");
c627d613 757
720b47f2
AT
758 setup_nonblocking(f_in,f_out);
759
c627d613
AT
760 while (1)
761 {
762 i = read_int(f_in);
c6e7fcb4
AT
763 if (i == -1) {
764 if (phase==0 && remote_version >= 13) {
765 phase++;
9e31c482 766 csum_length = SUM_LENGTH;
c6e7fcb4
AT
767 write_int(f_out,-1);
768 write_flush(f_out);
9e31c482
AT
769 if (verbose > 2)
770 fprintf(FERROR,"send_files phase=%d\n",phase);
c6e7fcb4
AT
771 continue;
772 }
773 break;
774 }
c627d613 775
dc5ddbcc
AT
776 file = &flist->files[i];
777
c627d613 778 fname[0] = 0;
dc5ddbcc
AT
779 if (file->dir) {
780 strcpy(fname,file->dir);
c627d613
AT
781 strcat(fname,"/");
782 }
dc5ddbcc 783 strcat(fname,file->name);
c627d613
AT
784
785 if (verbose > 2)
dc5ddbcc 786 fprintf(FERROR,"send_files(%d,%s)\n",i,fname);
c627d613
AT
787
788 if (dry_run) {
789 if (!am_server && verbose)
790 printf("%s\n",fname);
791 write_int(f_out,i);
792 continue;
793 }
794
795 s = receive_sums(f_in);
796 if (!s) {
dc5ddbcc 797 fprintf(FERROR,"receive_sums failed\n");
c627d613
AT
798 return -1;
799 }
800
801 fd = open(fname,O_RDONLY);
802 if (fd == -1) {
dc5ddbcc 803 fprintf(FERROR,"send_files failed to open %s: %s\n",
c627d613
AT
804 fname,strerror(errno));
805 continue;
806 }
807
808 /* map the local file */
809 if (fstat(fd,&st) != 0) {
dc5ddbcc 810 fprintf(FERROR,"fstat failed : %s\n",strerror(errno));
c627d613
AT
811 return -1;
812 }
813
814 if (st.st_size > 0) {
815 buf = map_file(fd,st.st_size);
c627d613
AT
816 } else {
817 buf = NULL;
818 }
819
820 if (verbose > 2)
dc5ddbcc 821 fprintf(FERROR,"send_files mapped %s of size %d\n",
c627d613
AT
822 fname,(int)st.st_size);
823
824 write_int(f_out,i);
825
826 write_int(f_out,s->count);
827 write_int(f_out,s->n);
828 write_int(f_out,s->remainder);
829
830 if (verbose > 2)
dc5ddbcc 831 fprintf(FERROR,"calling match_sums %s\n",fname);
c627d613
AT
832
833 if (!am_server && verbose)
834 printf("%s\n",fname);
835
720b47f2 836 match_sums(f_out,s,buf,st.st_size);
c627d613
AT
837 write_flush(f_out);
838
1cdc8b50 839 if (buf) unmap_file(buf);
c627d613
AT
840 close(fd);
841
842 free_sums(s);
843
844 if (verbose > 2)
dc5ddbcc 845 fprintf(FERROR,"sender finished %s\n",fname);
c627d613
AT
846
847 total += st.st_size;
848 }
849
dc5ddbcc
AT
850 if (verbose > 2)
851 fprintf(FERROR,"send files finished\n");
852
c627d613
AT
853 match_report();
854
855 write_int(f_out,-1);
856 write_flush(f_out);
857
858 return total;
859}
860
861
862
c6e7fcb4 863void generate_files(int f,struct file_list *flist,char *local_name,int f_recv)
c627d613
AT
864{
865 int i;
9e31c482 866 int phase=0;
c627d613
AT
867
868 if (verbose > 2)
dc5ddbcc 869 fprintf(FERROR,"generator starting pid=%d count=%d\n",
c627d613
AT
870 (int)getpid(),flist->count);
871
872 for (i = 0; i < flist->count; i++) {
dc5ddbcc 873 struct file_struct *file = &flist->files[i];
7b8356d0 874 mode_t saved_mode = file->mode;
dc5ddbcc 875 if (!file->name) continue;
7b8356d0
AT
876
877 /* we need to ensure that any directories we create have writeable
878 permissions initially so that we can create the files within
879 them. This is then fixed after the files are transferred */
880 if (!am_root && S_ISDIR(file->mode)) {
881 file->mode |= S_IWUSR; /* user write */
c627d613 882 }
7b8356d0 883
dc5ddbcc 884 recv_generator(local_name?local_name:file->name,
c627d613 885 flist,i,f);
7b8356d0
AT
886
887 file->mode = saved_mode;
c627d613 888 }
c6e7fcb4 889
9e31c482
AT
890 phase++;
891 csum_length = SUM_LENGTH;
892 ignore_times=1;
893
894 if (verbose > 2)
895 fprintf(FERROR,"generate_files phase=%d\n",phase);
896
c627d613
AT
897 write_int(f,-1);
898 write_flush(f);
c6e7fcb4
AT
899
900 if (remote_version >= 13) {
7b8356d0
AT
901 /* in newer versions of the protocol the files can cycle through
902 the system more than once to catch initial checksum errors */
c6e7fcb4
AT
903 for (i=read_int(f_recv); i != -1; i=read_int(f_recv)) {
904 struct file_struct *file = &flist->files[i];
905 recv_generator(local_name?local_name:file->name,
906 flist,i,f);
907 }
908
9e31c482
AT
909 phase++;
910 if (verbose > 2)
911 fprintf(FERROR,"generate_files phase=%d\n",phase);
912
c6e7fcb4
AT
913 write_int(f,-1);
914 write_flush(f);
915 }
916
917
c627d613 918 if (verbose > 2)
dc5ddbcc 919 fprintf(FERROR,"generator wrote %d\n",write_total());
c627d613 920}
dc5ddbcc
AT
921
922