casting cleanups (rsync now compiles cleanly under IRIX 6.4)
[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 656 /* open tmp file */
13a1f792
AT
657 if (strlen(fname) > (MAXPATHLEN-8)) {
658 fprintf(FERROR,"filename too long\n");
659 continue;
660 }
c627d613
AT
661 sprintf(fnametmp,"%s.XXXXXX",fname);
662 if (NULL == mktemp(fnametmp)) {
dc5ddbcc 663 fprintf(FERROR,"mktemp %s failed\n",fnametmp);
7b8356d0
AT
664 receive_data(f_in,buf,-1,NULL);
665 if (buf) unmap_file(buf);
666 close(fd1);
667 continue;
c627d613 668 }
dc5ddbcc 669 fd2 = open(fnametmp,O_WRONLY|O_CREAT,file->mode);
6574b4f7
AT
670 if (relative_paths && errno == ENOENT &&
671 create_directory_path(fnametmp) == 0) {
672 fd2 = open(fnametmp,O_WRONLY|O_CREAT,file->mode);
673 }
c627d613 674 if (fd2 == -1) {
dc5ddbcc 675 fprintf(FERROR,"open %s : %s\n",fnametmp,strerror(errno));
7b8356d0
AT
676 receive_data(f_in,buf,-1,NULL);
677 if (buf) unmap_file(buf);
678 close(fd1);
679 continue;
c627d613
AT
680 }
681
682 cleanup_fname = fnametmp;
683
684 if (!am_server && verbose)
685 printf("%s\n",fname);
686
687 /* recv file data */
9e31c482 688 recv_ok = receive_data(f_in,buf,fd2,fname);
c627d613 689
981791bd 690 if (fd1 != -1) {
1cdc8b50 691 if (buf) unmap_file(buf);
981791bd
AT
692 close(fd1);
693 }
c627d613
AT
694 close(fd2);
695
696 if (verbose > 2)
dc5ddbcc 697 fprintf(FERROR,"renaming %s to %s\n",fnametmp,fname);
c627d613
AT
698
699 if (make_backups) {
700 char fnamebak[MAXPATHLEN];
13a1f792
AT
701 if (strlen(fname) + strlen(backup_suffix) > (MAXPATHLEN-1)) {
702 fprintf(FERROR,"backup filename too long\n");
703 continue;
704 }
c627d613 705 sprintf(fnamebak,"%s%s",fname,backup_suffix);
ac1eb754 706 if (rename(fname,fnamebak) != 0 && errno != ENOENT) {
dc5ddbcc 707 fprintf(FERROR,"rename %s %s : %s\n",fname,fnamebak,strerror(errno));
7b8356d0 708 continue;
c627d613
AT
709 }
710 }
711
712 /* move tmp file over real file */
713 if (rename(fnametmp,fname) != 0) {
dc5ddbcc 714 fprintf(FERROR,"rename %s -> %s : %s\n",
c627d613 715 fnametmp,fname,strerror(errno));
7b8356d0 716 unlink(fnametmp);
c627d613
AT
717 }
718
719 cleanup_fname = NULL;
720
dc5ddbcc 721 set_perms(fname,file,NULL,0);
9e31c482
AT
722
723 if (!recv_ok) {
724 if (verbose > 1)
725 fprintf(FERROR,"redoing %s(%d)\n",fname,i);
52296954
AT
726 if (csum_length == SUM_LENGTH)
727 fprintf(FERROR,"ERROR: file corruption in %s\n",fname);
9e31c482
AT
728 write_int(f_gen,i);
729 }
c627d613
AT
730 }
731
7b8356d0
AT
732 /* now we need to fix any directory permissions that were
733 modified during the transfer */
734 if (!am_root) {
735 for (i = 0; i < flist->count; i++) {
736 struct file_struct *file = &flist->files[i];
737 if (!file->name || !S_ISDIR(file->mode)) continue;
738 recv_generator(file->name,flist,i,-1);
739 }
740 }
741
742
c627d613 743 if (verbose > 2)
dc5ddbcc 744 fprintf(FERROR,"recv_files finished\n");
c627d613
AT
745
746 return 0;
747}
748
749
750
751off_t send_files(struct file_list *flist,int f_out,int f_in)
752{
753 int fd;
754 struct sum_struct *s;
c6e7fcb4 755 struct map_struct *buf;
c627d613
AT
756 struct stat st;
757 char fname[MAXPATHLEN];
758 off_t total=0;
759 int i;
dc5ddbcc 760 struct file_struct *file;
c6e7fcb4 761 int phase = 0;
c627d613
AT
762
763 if (verbose > 2)
dc5ddbcc 764 fprintf(FERROR,"send_files starting\n");
c627d613 765
720b47f2
AT
766 setup_nonblocking(f_in,f_out);
767
c627d613
AT
768 while (1)
769 {
770 i = read_int(f_in);
c6e7fcb4
AT
771 if (i == -1) {
772 if (phase==0 && remote_version >= 13) {
773 phase++;
9e31c482 774 csum_length = SUM_LENGTH;
c6e7fcb4
AT
775 write_int(f_out,-1);
776 write_flush(f_out);
9e31c482
AT
777 if (verbose > 2)
778 fprintf(FERROR,"send_files phase=%d\n",phase);
c6e7fcb4
AT
779 continue;
780 }
781 break;
782 }
c627d613 783
dc5ddbcc
AT
784 file = &flist->files[i];
785
c627d613 786 fname[0] = 0;
dc5ddbcc 787 if (file->dir) {
13a1f792
AT
788 strncpy(fname,file->dir,MAXPATHLEN-1);
789 fname[MAXPATHLEN-1] = 0;
c627d613
AT
790 strcat(fname,"/");
791 }
13a1f792 792 strncat(fname,file->name,MAXPATHLEN-strlen(fname));
c627d613
AT
793
794 if (verbose > 2)
dc5ddbcc 795 fprintf(FERROR,"send_files(%d,%s)\n",i,fname);
c627d613
AT
796
797 if (dry_run) {
798 if (!am_server && verbose)
799 printf("%s\n",fname);
800 write_int(f_out,i);
801 continue;
802 }
803
804 s = receive_sums(f_in);
805 if (!s) {
dc5ddbcc 806 fprintf(FERROR,"receive_sums failed\n");
c627d613
AT
807 return -1;
808 }
809
810 fd = open(fname,O_RDONLY);
811 if (fd == -1) {
dc5ddbcc 812 fprintf(FERROR,"send_files failed to open %s: %s\n",
c627d613
AT
813 fname,strerror(errno));
814 continue;
815 }
816
817 /* map the local file */
818 if (fstat(fd,&st) != 0) {
dc5ddbcc 819 fprintf(FERROR,"fstat failed : %s\n",strerror(errno));
c627d613
AT
820 return -1;
821 }
822
823 if (st.st_size > 0) {
824 buf = map_file(fd,st.st_size);
c627d613
AT
825 } else {
826 buf = NULL;
827 }
828
829 if (verbose > 2)
dc5ddbcc 830 fprintf(FERROR,"send_files mapped %s of size %d\n",
c627d613
AT
831 fname,(int)st.st_size);
832
833 write_int(f_out,i);
834
835 write_int(f_out,s->count);
836 write_int(f_out,s->n);
837 write_int(f_out,s->remainder);
838
839 if (verbose > 2)
dc5ddbcc 840 fprintf(FERROR,"calling match_sums %s\n",fname);
c627d613
AT
841
842 if (!am_server && verbose)
843 printf("%s\n",fname);
844
720b47f2 845 match_sums(f_out,s,buf,st.st_size);
c627d613
AT
846 write_flush(f_out);
847
1cdc8b50 848 if (buf) unmap_file(buf);
c627d613
AT
849 close(fd);
850
851 free_sums(s);
852
853 if (verbose > 2)
dc5ddbcc 854 fprintf(FERROR,"sender finished %s\n",fname);
c627d613
AT
855
856 total += st.st_size;
857 }
858
dc5ddbcc
AT
859 if (verbose > 2)
860 fprintf(FERROR,"send files finished\n");
861
c627d613
AT
862 match_report();
863
864 write_int(f_out,-1);
865 write_flush(f_out);
866
867 return total;
868}
869
870
871
c6e7fcb4 872void generate_files(int f,struct file_list *flist,char *local_name,int f_recv)
c627d613
AT
873{
874 int i;
9e31c482 875 int phase=0;
c627d613
AT
876
877 if (verbose > 2)
dc5ddbcc 878 fprintf(FERROR,"generator starting pid=%d count=%d\n",
c627d613
AT
879 (int)getpid(),flist->count);
880
881 for (i = 0; i < flist->count; i++) {
dc5ddbcc 882 struct file_struct *file = &flist->files[i];
7b8356d0 883 mode_t saved_mode = file->mode;
dc5ddbcc 884 if (!file->name) continue;
7b8356d0
AT
885
886 /* we need to ensure that any directories we create have writeable
887 permissions initially so that we can create the files within
888 them. This is then fixed after the files are transferred */
889 if (!am_root && S_ISDIR(file->mode)) {
890 file->mode |= S_IWUSR; /* user write */
c627d613 891 }
7b8356d0 892
dc5ddbcc 893 recv_generator(local_name?local_name:file->name,
c627d613 894 flist,i,f);
7b8356d0
AT
895
896 file->mode = saved_mode;
c627d613 897 }
c6e7fcb4 898
9e31c482
AT
899 phase++;
900 csum_length = SUM_LENGTH;
901 ignore_times=1;
902
903 if (verbose > 2)
904 fprintf(FERROR,"generate_files phase=%d\n",phase);
905
c627d613
AT
906 write_int(f,-1);
907 write_flush(f);
c6e7fcb4
AT
908
909 if (remote_version >= 13) {
7b8356d0
AT
910 /* in newer versions of the protocol the files can cycle through
911 the system more than once to catch initial checksum errors */
c6e7fcb4
AT
912 for (i=read_int(f_recv); i != -1; i=read_int(f_recv)) {
913 struct file_struct *file = &flist->files[i];
914 recv_generator(local_name?local_name:file->name,
915 flist,i,f);
916 }
917
9e31c482
AT
918 phase++;
919 if (verbose > 2)
920 fprintf(FERROR,"generate_files phase=%d\n",phase);
921
c6e7fcb4
AT
922 write_int(f,-1);
923 write_flush(f);
924 }
925
926
c627d613 927 if (verbose > 2)
dc5ddbcc 928 fprintf(FERROR,"generator wrote %d\n",write_total());
c627d613 929}
dc5ddbcc
AT
930
931