2 Copyright (C) Andrew Tridgell 1996
3 Copyright (C) Paul Mackerras 1996
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.
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.
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.
24 extern int always_checksum;
25 extern time_t starttime;
27 extern char *backup_suffix;
29 extern int block_size;
30 extern int update_only;
31 extern int make_backups;
32 extern int preserve_links;
33 extern int preserve_perms;
34 extern int preserve_devices;
35 extern int preserve_uid;
36 extern int preserve_gid;
37 extern int preserve_times;
39 extern int ignore_times;
41 extern int delete_mode;
42 extern int cvs_exclude;
47 static void free_sums(struct sum_struct *s)
49 if (s->sums) free(s->sums);
56 send a sums struct down a fd
58 static void send_sums(struct sum_struct *s,int f_out)
62 /* tell the other guy how many we are going to be doing and how many
63 bytes there are in the last chunk */
64 write_int(f_out,s?s->count:0);
65 write_int(f_out,s?s->n:block_size);
66 write_int(f_out,s?s->remainder:0);
68 for (i=0;i<s->count;i++) {
69 write_int(f_out,s->sums[i].sum1);
70 write_buf(f_out,s->sums[i].sum2,SUM_LENGTH);
77 generate a stream of signatures/checksums that describe a buffer
79 generate approximately one checksum every n bytes
81 static struct sum_struct *generate_sums(char *buf,off_t len,int n)
87 int remainder = (len%block_len);
90 count = (len+(block_len-1))/block_len;
92 s = (struct sum_struct *)malloc(sizeof(*s));
93 if (!s) out_of_memory("generate_sums");
96 s->remainder = remainder;
106 fprintf(stderr,"count=%d rem=%d n=%d flength=%d\n",
107 s->count,s->remainder,s->n,(int)s->flength);
109 s->sums = (struct sum_buf *)malloc(sizeof(s->sums[0])*s->count);
110 if (!s->sums) out_of_memory("generate_sums");
112 for (i=0;i<count;i++) {
114 char *map = map_ptr(buf,offset,n1);
116 s->sums[i].sum1 = get_checksum1(map,n1);
117 get_checksum2(map,n1,s->sums[i].sum2);
119 s->sums[i].offset = offset;
124 fprintf(stderr,"chunk[%d] offset=%d len=%d sum1=%08x\n",
125 i,(int)s->sums[i].offset,s->sums[i].len,s->sums[i].sum1);
136 receive the checksums for a buffer
138 static struct sum_struct *receive_sums(int f)
140 struct sum_struct *s;
145 s = (struct sum_struct *)malloc(sizeof(*s));
146 if (!s) out_of_memory("receive_sums");
148 s->count = read_int(f);
150 s->remainder = read_int(f);
154 fprintf(stderr,"count=%d n=%d rem=%d\n",
155 s->count,s->n,s->remainder);
162 s->sums = (struct sum_buf *)malloc(sizeof(s->sums[0])*s->count);
163 if (!s->sums) out_of_memory("receive_sums");
165 for (i=0;i<s->count;i++) {
166 s->sums[i].sum1 = read_int(f);
167 read_buf(f,s->sums[i].sum2,SUM_LENGTH);
169 s->sums[i].offset = offset;
172 if (i == s->count-1 && s->remainder != 0) {
173 s->sums[i].len = s->remainder;
175 s->sums[i].len = s->n;
177 offset += s->sums[i].len;
180 fprintf(stderr,"chunk[%d] len=%d offset=%d sum1=%08x\n",
181 i,s->sums[i].len,(int)s->sums[i].offset,s->sums[i].sum1);
190 static void set_perms(char *fname,struct file_struct *file,struct stat *st,
199 if (stat(fname,&st2) != 0) {
200 fprintf(stderr,"stat %s : %s\n",fname,strerror(errno));
206 if (preserve_times && st->st_mtime != file->modtime) {
208 if (set_modtime(fname,file->modtime) != 0) {
209 fprintf(stderr,"failed to set times on %s : %s\n",
210 fname,strerror(errno));
216 if (preserve_perms && st->st_mode != file->mode) {
218 if (chmod(fname,file->mode) != 0) {
219 fprintf(stderr,"failed to set permissions on %s : %s\n",
220 fname,strerror(errno));
226 if ((preserve_uid && st->st_uid != file->uid) ||
227 (preserve_gid && st->st_gid != file->gid)) {
230 preserve_uid?file->uid:-1,
231 preserve_gid?file->gid:-1) != 0) {
232 if (verbose>1 || preserve_uid)
233 fprintf(stderr,"chown %s : %s\n",fname,strerror(errno));
238 if (verbose > 1 && report) {
240 fprintf(am_server?stderr:stdout,"%s\n",fname);
242 fprintf(am_server?stderr:stdout,"%s is uptodate\n",fname);
247 void recv_generator(char *fname,struct file_list *flist,int i,int f_out)
252 struct sum_struct *s;
253 char sum[SUM_LENGTH];
257 fprintf(stderr,"recv_generator(%s)\n",fname);
259 statret = lstat(fname,&st);
262 if (preserve_links && S_ISLNK(flist->files[i].mode)) {
263 char lnk[MAXPATHLEN];
266 l = readlink(fname,lnk,MAXPATHLEN-1);
269 if (strcmp(lnk,flist->files[i].link) == 0) {
271 fprintf(am_server?stderr:stdout,"%s is uptodate\n",fname);
276 if (!dry_run) unlink(fname);
277 if (!dry_run && symlink(flist->files[i].link,fname) != 0) {
278 fprintf(stderr,"link %s -> %s : %s\n",
279 fname,flist->files[i].link,strerror(errno));
282 fprintf(am_server?stderr:stdout,"%s -> %s\n",fname,flist->files[i].link);
289 if (preserve_devices && IS_DEVICE(flist->files[i].mode)) {
291 st.st_mode != flist->files[i].mode ||
292 st.st_rdev != flist->files[i].dev) {
293 if (!dry_run) unlink(fname);
295 fprintf(stderr,"mknod(%s,0%o,0x%x)\n",
296 fname,(int)flist->files[i].mode,(int)flist->files[i].dev);
298 mknod(fname,flist->files[i].mode,flist->files[i].dev) != 0) {
299 fprintf(stderr,"mknod %s : %s\n",fname,strerror(errno));
301 set_perms(fname,&flist->files[i],NULL,0);
303 fprintf(am_server?stderr:stdout,"%s\n",fname);
306 set_perms(fname,&flist->files[i],&st,1);
312 if (!S_ISREG(flist->files[i].mode)) {
313 fprintf(stderr,"skipping non-regular file %s\n",fname);
318 if (errno == ENOENT) {
320 if (!dry_run) send_sums(NULL,f_out);
323 fprintf(stderr,"recv_generator failed to open %s\n",fname);
328 if (!S_ISREG(st.st_mode)) {
329 fprintf(stderr,"%s : not a regular file\n",fname);
333 if (update_only && st.st_mtime >= flist->files[i].modtime) {
335 fprintf(stderr,"%s is newer\n",fname);
339 if (always_checksum && S_ISREG(st.st_mode)) {
340 file_checksum(fname,sum,st.st_size);
343 if (st.st_size == flist->files[i].length &&
344 ((!ignore_times && st.st_mtime == flist->files[i].modtime) ||
345 (always_checksum && S_ISREG(st.st_mode) &&
346 memcmp(sum,flist->files[i].sum,SUM_LENGTH) == 0))) {
347 set_perms(fname,&flist->files[i],&st,1);
357 fd = open(fname,O_RDONLY);
360 fprintf(stderr,"failed to open %s : %s\n",fname,strerror(errno));
364 if (st.st_size > 0) {
365 buf = map_file(fd,st.st_size);
371 fprintf(stderr,"mapped %s of size %d\n",fname,(int)st.st_size);
373 s = generate_sums(buf,st.st_size,block_size);
380 unmap_file(buf,st.st_size);
387 static void receive_data(int f_in,char *buf,int fd,char *fname)
389 int i,n,remainder,len,count;
393 count = read_int(f_in);
395 remainder = read_int(f_in);
397 for (i=read_int(f_in); i != 0; i=read_int(f_in)) {
400 fprintf(stderr,"data recv %d at %d\n",i,(int)offset);
402 if (read_write(f_in,fd,i) != i) {
403 fprintf(stderr,"write failed on %s : %s\n",fname,strerror(errno));
411 if (i == count-1 && remainder != 0)
415 fprintf(stderr,"chunk[%d] of size %d at %d offset=%d\n",
416 i,len,(int)offset2,(int)offset);
418 if (write(fd,map_ptr(buf,offset2,len),len) != len) {
419 fprintf(stderr,"write failed on %s : %s\n",fname,strerror(errno));
428 static void delete_one(struct file_struct *f)
430 if (!S_ISDIR(f->mode)) {
431 if (!dry_run && unlink(f->name) != 0) {
432 fprintf(stderr,"unlink %s : %s\n",f->name,strerror(errno));
433 } else if (verbose) {
434 fprintf(stderr,"deleting %s\n",f->name);
437 if (!dry_run && rmdir(f->name) != 0) {
438 if (errno != ENOTEMPTY)
439 fprintf(stderr,"rmdir %s : %s\n",f->name,strerror(errno));
440 } else if (verbose) {
441 fprintf(stderr,"deleting directory %s\n",f->name);
447 static void delete_files(struct file_list *flist)
449 struct file_list *local_file_list;
453 if (!am_server && cvs_exclude)
455 if (!(local_file_list = send_file_list(-1,recurse,1,&dot)))
458 for (i=local_file_list->count;i>=0;i--) {
459 if (!local_file_list->files[i].name) continue;
460 if (-1 == flist_find(flist,&local_file_list->files[i])) {
461 delete_one(&local_file_list->files[i]);
466 static char *cleanup_fname = NULL;
471 unlink(cleanup_fname);
476 int recv_files(int f_in,struct file_list *flist,char *local_name)
481 char fnametmp[MAXPATHLEN];
486 fprintf(stderr,"recv_files(%d) starting\n",flist->count);
489 if (recurse && delete_mode && !local_name && flist->count>0) {
498 fname = flist->files[i].name;
504 if (!am_server && verbose)
505 printf("%s\n",fname);
510 fprintf(stderr,"recv_files(%s)\n",fname);
513 fd1 = open(fname,O_RDONLY);
515 if (fd1 != -1 && fstat(fd1,&st) != 0) {
516 fprintf(stderr,"fstat %s : %s\n",fname,strerror(errno));
521 if (fd1 != -1 && !S_ISREG(st.st_mode)) {
522 fprintf(stderr,"%s : not a regular file\n",fname);
527 if (fd1 != -1 && st.st_size > 0) {
528 buf = map_file(fd1,st.st_size);
534 fprintf(stderr,"mapped %s of size %d\n",fname,(int)st.st_size);
537 sprintf(fnametmp,"%s.XXXXXX",fname);
538 if (NULL == mktemp(fnametmp)) {
539 fprintf(stderr,"mktemp %s failed\n",fnametmp);
542 fd2 = open(fnametmp,O_WRONLY|O_CREAT,flist->files[i].mode);
544 fprintf(stderr,"open %s : %s\n",fnametmp,strerror(errno));
548 cleanup_fname = fnametmp;
550 if (!am_server && verbose)
551 printf("%s\n",fname);
554 receive_data(f_in,buf,fd2,fname);
557 unmap_file(buf,st.st_size);
563 fprintf(stderr,"renaming %s to %s\n",fnametmp,fname);
566 char fnamebak[MAXPATHLEN];
567 sprintf(fnamebak,"%s%s",fname,backup_suffix);
568 if (rename(fname,fnamebak) != 0 && errno != ENOENT) {
569 fprintf(stderr,"rename %s %s : %s\n",fname,fnamebak,strerror(errno));
574 /* move tmp file over real file */
575 if (rename(fnametmp,fname) != 0) {
576 fprintf(stderr,"rename %s -> %s : %s\n",
577 fnametmp,fname,strerror(errno));
580 cleanup_fname = NULL;
582 set_perms(fname,&flist->files[i],NULL,0);
586 fprintf(stderr,"recv_files finished\n");
593 off_t send_files(struct file_list *flist,int f_out,int f_in)
596 struct sum_struct *s;
599 char fname[MAXPATHLEN];
604 fprintf(stderr,"send_files starting\n");
606 setup_nonblocking(f_in,f_out);
614 if (flist->files[i].dir) {
615 strcpy(fname,flist->files[i].dir);
618 strcat(fname,flist->files[i].name);
621 fprintf(stderr,"send_files(%d,%s)\n",i,fname);
624 if (!am_server && verbose)
625 printf("%s\n",fname);
630 s = receive_sums(f_in);
632 fprintf(stderr,"receive_sums failed\n");
636 fd = open(fname,O_RDONLY);
638 fprintf(stderr,"send_files failed to open %s: %s\n",
639 fname,strerror(errno));
643 /* map the local file */
644 if (fstat(fd,&st) != 0) {
645 fprintf(stderr,"fstat failed : %s\n",strerror(errno));
649 if (st.st_size > 0) {
650 buf = map_file(fd,st.st_size);
656 fprintf(stderr,"send_files mapped %s of size %d\n",
657 fname,(int)st.st_size);
661 write_int(f_out,s->count);
662 write_int(f_out,s->n);
663 write_int(f_out,s->remainder);
666 fprintf(stderr,"calling match_sums %s\n",fname);
668 if (!am_server && verbose)
669 printf("%s\n",fname);
671 match_sums(f_out,s,buf,st.st_size);
674 unmap_file(buf,st.st_size);
680 fprintf(stderr,"sender finished %s\n",fname);
695 void generate_files(int f,struct file_list *flist,char *local_name)
700 fprintf(stderr,"generator starting pid=%d count=%d\n",
701 (int)getpid(),flist->count);
703 for (i = 0; i < flist->count; i++) {
704 if (!flist->files[i].name) continue;
705 if (S_ISDIR(flist->files[i].mode)) {
706 if (dry_run) continue;
707 if (mkdir(flist->files[i].name,flist->files[i].mode) != 0 &&
709 fprintf(stderr,"mkdir %s : %s\n",
710 flist->files[i].name,strerror(errno));
714 recv_generator(local_name?local_name:flist->files[i].name,
720 fprintf(stderr,"generator wrote %d\n",write_total());