main.c

   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
  22 time_t starttime = 0;
  23 int64 total_size = 0;
  24
  25 extern int csum_length;
  26
  27 extern int verbose;
  28
  29 static void report(int f)
  30 {
  31         int64 in,out,tsize;
  32         time_t t = time(NULL);
  33         extern int am_server;
  34         extern int am_sender;
  35         extern int am_daemon;
  36
  37         if (am_daemon) {
  38                 syslog(LOG_INFO,"wrote %.0f bytes  read %.0f bytes  total size %.0f\n",
  39                        (double)write_total(),(double)read_total(),
  40                        (double)total_size);
  41                 if (f == -1 || !am_sender) return;
  42         }
  43
  44         if (!verbose) return;
  45
  46         if (am_server && !am_sender) return;
  47
  48         if (am_server && am_sender) {
  49                 write_longint(f,read_total());
  50                 write_longint(f,write_total());
  51                 write_longint(f,total_size);
  52                 return;
  53         }
  54
  55         if (am_sender) {
  56                 in = read_total();
  57                 out = write_total();
  58                 tsize = total_size;
  59         } else {
  60                 out = read_longint(f);
  61                 in = read_longint(f);
  62                 tsize = read_longint(f);
  63         }
  64
  65         printf("wrote %.0f bytes  read %.0f bytes  %.2f bytes/sec\n",
  66                (double)out,(double)in,(in+out)/(0.5 + (t-starttime)));
  67         printf("total size is %.0f  speedup is %.2f\n",
  68                (double)tsize,(1.0*tsize)/(in+out));
  69 }
  70
  71
  72 static int do_cmd(char *cmd,char *machine,char *user,char *path,int *f_in,int *f_out)
  73 {
  74         char *args[100];
  75         int i,argc=0, ret;
  76         char *tok,*dir=NULL;
  77         extern int local_server;
  78         extern char *rsync_path;
  79
  80         if (!local_server) {
  81                 if (!cmd)
  82                         cmd = getenv(RSYNC_RSH_ENV);
  83                 if (!cmd)
  84                         cmd = RSYNC_RSH;
  85                 cmd = strdup(cmd);
  86                 if (!cmd)
  87                         goto oom;
  88
  89                 for (tok=strtok(cmd," ");tok;tok=strtok(NULL," ")) {
  90                         args[argc++] = tok;
  91                 }
  92
  93 #if HAVE_REMSH
  94                 /* remsh (on HPUX) takes the arguments the other way around */
  95                 args[argc++] = machine;
  96                 if (user) {
  97                         args[argc++] = "-l";
  98                         args[argc++] = user;
  99                 }
 100 #else
 101                 if (user) {
 102                         args[argc++] = "-l";
 103                         args[argc++] = user;
 104                 }
 105                 args[argc++] = machine;
 106 #endif
 107
 108                 args[argc++] = rsync_path;
 109
 110                 server_options(args,&argc);
 111         }
 112
 113         args[argc++] = ".";
 114
 115         if (path && *path)
 116                 args[argc++] = path;
 117
 118         args[argc] = NULL;
 119
 120         if (verbose > 3) {
 121                 rprintf(FINFO,"cmd=");
 122                 for (i=0;i<argc;i++)
 123                         rprintf(FINFO,"%s ",args[i]);
 124                 rprintf(FINFO,"\n");
 125         }
 126
 127         if (local_server) {
 128                 ret = local_child(argc, args, f_in, f_out);
 129         } else {
 130                 ret = piped_child(args,f_in,f_out);
 131         }
 132
 133         if (dir) free(dir);
 134
 135         return ret;
 136
 137 oom:
 138         out_of_memory("do_cmd");
 139         return 0; /* not reached */
 140 }
 141
 142
 143
 144
 145 static char *get_local_name(struct file_list *flist,char *name)
 146 {
 147         STRUCT_STAT st;
 148         extern int orig_umask;
 149
 150         if (do_stat(name,&st) == 0) {
 151                 if (S_ISDIR(st.st_mode)) {
 152                         if (chdir(name) != 0) {
 153                                 rprintf(FERROR,"chdir %s : %s (1)\n",
 154                                         name,strerror(errno));
 155                                 exit_cleanup(1);
 156                         }
 157                         return NULL;
 158                 }
 159                 if (flist->count > 1) {
 160                         rprintf(FERROR,"ERROR: destination must be a directory when copying more than 1 file\n");
 161                         exit_cleanup(1);
 162                 }
 163                 return name;
 164         }
 165
 166         if (flist->count == 1)
 167                 return name;
 168
 169         if (!name)
 170                 return NULL;
 171
 172         if (do_mkdir(name,0777 & ~orig_umask) != 0) {
 173                 rprintf(FERROR,"mkdir %s : %s (1)\n",name,strerror(errno));
 174                 exit_cleanup(1);
 175         } else {
 176                 rprintf(FINFO,"created directory %s\n",name);
 177         }
 178
 179         if (chdir(name) != 0) {
 180                 rprintf(FERROR,"chdir %s : %s (2)\n",name,strerror(errno));
 181                 exit_cleanup(1);
 182         }
 183
 184         return NULL;
 185 }
 186
 187
 188
 189
 190 static void do_server_sender(int f_in, int f_out, int argc,char *argv[])
 191 {
 192         int i;
 193         struct file_list *flist;
 194         char *dir = argv[0];
 195         extern int relative_paths;
 196         extern int recurse;
 197
 198         if (verbose > 2)
 199                 rprintf(FINFO,"server_sender starting pid=%d\n",(int)getpid());
 200
 201         if (!relative_paths && chdir(dir) != 0) {
 202                 rprintf(FERROR,"chdir %s: %s (3)\n",dir,strerror(errno));
 203                 exit_cleanup(1);
 204         }
 205         argc--;
 206         argv++;
 207
 208         if (strcmp(dir,".")) {
 209                 int l = strlen(dir);
 210                 if (strcmp(dir,"/") == 0)
 211                         l = 0;
 212                 for (i=0;i<argc;i++)
 213                         argv[i] += l+1;
 214         }
 215
 216         if (argc == 0 && recurse) {
 217                 argc=1;
 218                 argv--;
 219                 argv[0] = ".";
 220         }
 221
 222         flist = send_file_list(f_out,argc,argv);
 223         if (!flist || flist->count == 0) {
 224                 exit_cleanup(0);
 225         }
 226
 227         send_files(flist,f_out,f_in);
 228         report(f_out);
 229         io_flush();
 230         exit_cleanup(0);
 231 }
 232
 233
 234 static int do_recv(int f_in,int f_out,struct file_list *flist,char *local_name)
 235 {
 236         int pid;
 237         int status=0;
 238         int recv_pipe[2];
 239         extern int preserve_hard_links;
 240
 241         if (preserve_hard_links)
 242                 init_hard_links(flist);
 243
 244         if (pipe(recv_pipe) < 0) {
 245                 rprintf(FERROR,"pipe failed in do_recv\n");
 246                 exit_cleanup(1);
 247         }
 248
 249         io_flush();
 250
 251         if ((pid=do_fork()) == 0) {
 252                 close(recv_pipe[0]);
 253                 if (f_in != f_out) close(f_out);
 254
 255                 recv_files(f_in,flist,local_name,recv_pipe[1]);
 256                 report(f_in);
 257
 258                 if (verbose > 3)
 259                         rprintf(FINFO,"do_recv waiting on %d\n",pid);
 260
 261                 io_flush();
 262                 _exit(0);
 263         }
 264
 265         close(recv_pipe[1]);
 266         io_close_input(f_in);
 267         if (f_in != f_out) close(f_in);
 268         generate_files(f_out,flist,local_name,recv_pipe[0]);
 269
 270         io_flush();
 271         waitpid(pid, &status, 0);
 272         return status;
 273 }
 274
 275
 276 static void do_server_recv(int f_in, int f_out, int argc,char *argv[])
 277 {
 278         int status;
 279         struct file_list *flist;
 280         char *local_name=NULL;
 281         char *dir = NULL;
 282         extern int delete_mode;
 283         extern int am_daemon;
 284
 285         if (verbose > 2)
 286                 rprintf(FINFO,"server_recv(%d) starting pid=%d\n",argc,(int)getpid());
 287
 288         if (argc > 0) {
 289                 dir = argv[0];
 290                 argc--;
 291                 argv++;
 292                 if (!am_daemon && chdir(dir) != 0) {
 293                         rprintf(FERROR,"chdir %s : %s (4)\n",
 294                                 dir,strerror(errno));
 295                         exit_cleanup(1);
 296                 }
 297         }
 298
 299         if (delete_mode)
 300                 recv_exclude_list(f_in);
 301
 302         flist = recv_file_list(f_in);
 303         if (!flist || flist->count == 0) {
 304                 rprintf(FERROR,"server_recv: nothing to do\n");
 305                 exit_cleanup(1);
 306         }
 307
 308         if (argc > 0) {
 309                 if (strcmp(dir,".")) {
 310                         argv[0] += strlen(dir);
 311                         if (argv[0][0] == '/') argv[0]++;
 312                 }
 313                 local_name = get_local_name(flist,argv[0]);
 314         }
 315
 316         status = do_recv(f_in,f_out,flist,local_name);
 317         exit_cleanup(status);
 318 }
 319
 320
 321 void start_server(int f_in, int f_out, int argc, char *argv[])
 322 {
 323         extern int cvs_exclude;
 324         extern int am_sender;
 325
 326         setup_protocol(f_out, f_in);
 327
 328         if (am_sender) {
 329                 recv_exclude_list(f_in);
 330                 if (cvs_exclude)
 331                         add_cvs_excludes();
 332                 do_server_sender(f_in, f_out, argc, argv);
 333         } else {
 334                 do_server_recv(f_in, f_out, argc, argv);
 335         }
 336         exit_cleanup(0);
 337 }
 338
 339 int client_run(int f_in, int f_out, int pid, int argc, char *argv[])
 340 {
 341         struct file_list *flist;
 342         int status = 0, status2 = 0;
 343         char *local_name = NULL;
 344         extern int am_sender;
 345
 346         setup_protocol(f_out,f_in);
 347
 348         if (am_sender) {
 349                 extern int cvs_exclude;
 350                 extern int delete_mode;
 351                 if (cvs_exclude)
 352                         add_cvs_excludes();
 353                 if (delete_mode)
 354                         send_exclude_list(f_out);
 355                 flist = send_file_list(f_out,argc,argv);
 356                 if (verbose > 3)
 357                         rprintf(FINFO,"file list sent\n");
 358                 send_files(flist,f_out,f_in);
 359                 if (pid != -1) {
 360                         if (verbose > 3)
 361                                 rprintf(FINFO,"client_run waiting on %d\n",pid);
 362                         io_flush();
 363                         waitpid(pid, &status, 0);
 364                 }
 365                 report(-1);
 366                 exit_cleanup(status);
 367         }
 368
 369         send_exclude_list(f_out);
 370
 371         flist = recv_file_list(f_in);
 372         if (!flist || flist->count == 0) {
 373                 rprintf(FINFO,"client: nothing to do\n");
 374                 exit_cleanup(0);
 375         }
 376
 377         local_name = get_local_name(flist,argv[0]);
 378
 379         status2 = do_recv(f_in,f_out,flist,local_name);
 380
 381         if (pid != -1) {
 382                 if (verbose > 3)
 383                         rprintf(FINFO,"client_run2 waiting on %d\n",pid);
 384                 io_flush();
 385                 waitpid(pid, &status, 0);
 386         }
 387
 388         return status | status2;
 389 }
 390
 391
 392 int start_client(int argc, char *argv[])
 393 {
 394         char *p;
 395         char *shell_machine = NULL;
 396         char *shell_path = NULL;
 397         char *shell_user = NULL;
 398         int pid;
 399         int f_in,f_out;
 400         extern int local_server;
 401         extern int am_sender;
 402         extern char *shell_cmd;
 403
 404         p = strchr(argv[0],':');
 405
 406         if (p) {
 407                 if (p[1] == ':') {
 408                         *p = 0;
 409                         return start_socket_client(argv[0], p+2, argc-1, argv+1);
 410                 }
 411
 412                 if (argc < 2) {
 413                         usage(FERROR);
 414                         exit_cleanup(1);
 415                 }
 416
 417                 am_sender = 0;
 418                 *p = 0;
 419                 shell_machine = argv[0];
 420                 shell_path = p+1;
 421                 argc--;
 422                 argv++;
 423         } else {
 424                 am_sender = 1;
 425
 426                 p = strchr(argv[argc-1],':');
 427                 if (!p) {
 428                         local_server = 1;
 429                 } else if (p[1] == ':') {
 430                         *p = 0;
 431                         return start_socket_client(argv[argc-1], p+2, argc-1, argv);
 432                 }
 433
 434                 if (argc < 2) {
 435                         usage(FERROR);
 436                         exit_cleanup(1);
 437                 }
 438
 439                 if (local_server) {
 440                         shell_machine = NULL;
 441                         shell_path = argv[argc-1];
 442                 } else {
 443                         *p = 0;
 444                         shell_machine = argv[argc-1];
 445                         shell_path = p+1;
 446                 }
 447                 argc--;
 448         }
 449
 450         if (shell_machine) {
 451                 p = strchr(shell_machine,'@');
 452                 if (p) {
 453                         *p = 0;
 454                         shell_user = shell_machine;
 455                         shell_machine = p+1;
 456                 }
 457         }
 458
 459         if (verbose > 3) {
 460                 rprintf(FINFO,"cmd=%s machine=%s user=%s path=%s\n",
 461                         shell_cmd?shell_cmd:"",
 462                         shell_machine?shell_machine:"",
 463                         shell_user?shell_user:"",
 464                         shell_path?shell_path:"");
 465         }
 466
 467         if (!am_sender && argc != 1) {
 468                 usage(FERROR);
 469                 exit_cleanup(1);
 470         }
 471
 472         pid = do_cmd(shell_cmd,shell_machine,shell_user,shell_path,&f_in,&f_out);
 473
 474 #if HAVE_SETLINEBUF
 475         setlinebuf(stdout);
 476         setlinebuf(stderr);
 477 #endif
 478
 479         return client_run(f_in, f_out, pid, argc, argv);
 480 }
 481
 482
 483 RETSIGTYPE sigusr1_handler(int val) {
 484         exit_cleanup(1);
 485 }
 486
 487 int main(int argc,char *argv[])
 488 {
 489         extern int am_root;
 490         extern int orig_umask;
 491         extern int dry_run;
 492         extern int am_daemon;
 493         extern int am_server;
 494
 495         signal(SIGUSR1, sigusr1_handler);
 496
 497         starttime = time(NULL);
 498         am_root = (getuid() == 0);
 499
 500         if (argc < 2) {
 501                 usage(FERROR);
 502                 exit_cleanup(1);
 503         }
 504
 505         /* we set a 0 umask so that correct file permissions can be
 506            carried across */
 507         orig_umask = (int)umask(0);
 508
 509         parse_arguments(argc, argv);
 510
 511         argc -= optind;
 512         argv += optind;
 513         optind = 0;
 514
 515         signal(SIGCHLD,SIG_IGN);
 516         signal(SIGINT,SIGNAL_CAST sig_int);
 517         signal(SIGPIPE,SIGNAL_CAST sig_int);
 518         signal(SIGHUP,SIGNAL_CAST sig_int);
 519
 520         if (am_daemon) {
 521                 return daemon_main();
 522         }
 523
 524         if (argc < 1) {
 525                 usage(FERROR);
 526                 exit_cleanup(1);
 527         }
 528
 529         if (dry_run)
 530                 verbose = MAX(verbose,1);
 531
 532 #ifndef SUPPORT_LINKS
 533         if (!am_server && preserve_links) {
 534                 rprintf(FERROR,"ERROR: symbolic links not supported\n");
 535                 exit_cleanup(1);
 536         }
 537 #endif
 538
 539         if (am_server) {
 540                 start_server(STDIN_FILENO, STDOUT_FILENO, argc, argv);
 541         }
 542
 543         return start_client(argc, argv);
 544 }
 545