home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 35 Internet / 35-Internet.zip / rsync221.zip / options.c < prev    next >
C/C++ Source or Header  |  1999-03-06  |  16KB  |  602 lines
  1. /* 
  2.    Copyright (C) Andrew Tridgell 1998
  3.    
  4.    This program is free software; you can redistribute it and/or modify
  5.    it under the terms of the GNU General Public License as published by
  6.    the Free Software Foundation; either version 2 of the License, or
  7.    (at your option) any later version.
  8.    
  9.    This program is distributed in the hope that it will be useful,
  10.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.    GNU General Public License for more details.
  13.    
  14.    You should have received a copy of the GNU General Public License
  15.    along with this program; if not, write to the Free Software
  16.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17. */
  18.  
  19. /* options parsing code */
  20.  
  21. #include "rsync.h"
  22.  
  23.  
  24. int make_backups = 0;
  25. int whole_file = 0;
  26. int copy_links = 0;
  27. int preserve_links = 0;
  28. int preserve_hard_links = 0;
  29. int preserve_perms = 0;
  30. int preserve_devices = 0;
  31. int preserve_uid = 0;
  32. int preserve_gid = 0;
  33. int preserve_times = 0;
  34. int update_only = 0;
  35. int cvs_exclude = 0;
  36. int dry_run=0;
  37. int local_server=0;
  38. int ignore_times=0;
  39. int delete_mode=0;
  40. int one_file_system=0;
  41. int remote_version=0;
  42. int sparse_files=0;
  43. int do_compression=0;
  44. int am_root=0;
  45. int orig_umask=0;
  46. int relative_paths=0;
  47. int numeric_ids = 0;
  48. int force_delete = 0;
  49. int io_timeout = 0;
  50. int io_error = 0;
  51. int read_only = 0;
  52. int module_id = -1;
  53. int am_server = 0;
  54. int am_sender=0;
  55. int recurse = 0;
  56. int am_daemon=0;
  57. int am_client=0;
  58. int do_stats=0;
  59. int do_progress=0;
  60. int keep_partial=0;
  61. int safe_symlinks=0;
  62.  
  63. int block_size=BLOCK_SIZE;
  64.  
  65. char *backup_suffix = BACKUP_SUFFIX;
  66. char *tmpdir = NULL;
  67. char *compare_dest = NULL;
  68. char *config_file = RSYNCD_CONF;
  69. char *shell_cmd = NULL;
  70. char *log_format = NULL;
  71.  
  72. char *rsync_path = RSYNC_NAME;
  73. int rsync_port = RSYNC_PORT;
  74.  
  75. int verbose = 0;
  76. int always_checksum = 0;
  77. int list_only = 0;
  78.  
  79. void usage(int F)
  80. {
  81.   rprintf(F,"rsync version %s Copyright Andrew Tridgell and Paul Mackerras\n\n",
  82.       VERSION);
  83.  
  84.   rprintf(F,"rsync is a file transfer program capable of efficient remote update\nvia a fast differencing algorithm.\n\n");
  85.  
  86.   rprintf(F,"Usage: rsync [OPTION]... SRC [USER@]HOST:DEST\n");
  87.   rprintf(F,"  or   rsync [OPTION]... [USER@]HOST:SRC DEST\n");
  88.   rprintf(F,"  or   rsync [OPTION]... SRC DEST\n");
  89.   rprintf(F,"  or   rsync [OPTION]... [USER@]HOST::SRC [DEST]\n");
  90.   rprintf(F,"  or   rsync [OPTION]... SRC [USER@]HOST::DEST\n");
  91.   rprintf(F,"  or   rsync [OPTION]... rsync://[USER@]HOST[:PORT]/SRC [DEST]\n");
  92.   rprintf(F,"\nOptions\n");
  93.   rprintf(F," -v, --verbose               increase verbosity\n");
  94.   rprintf(F," -c, --checksum              always checksum\n");
  95.   rprintf(F," -a, --archive               archive mode\n");
  96.   rprintf(F," -r, --recursive             recurse into directories\n");
  97.   rprintf(F," -R, --relative              use relative path names\n");
  98.   rprintf(F," -b, --backup                make backups (default ~ extension)\n");
  99.   rprintf(F," -u, --update                update only (don't overwrite newer files)\n");
  100.   rprintf(F," -l, --links                 preserve soft links\n");
  101.   rprintf(F," -L, --copy-links            treat soft links like regular files\n");
  102.   rprintf(F,"     --safe-links            ignore links outside the destination tree\n");
  103.   rprintf(F," -H, --hard-links            preserve hard links\n");
  104.   rprintf(F," -p, --perms                 preserve permissions\n");
  105.   rprintf(F," -o, --owner                 preserve owner (root only)\n");
  106.   rprintf(F," -g, --group                 preserve group\n");
  107.   rprintf(F," -D, --devices               preserve devices (root only)\n");
  108.   rprintf(F," -t, --times                 preserve times\n");  
  109.   rprintf(F," -S, --sparse                handle sparse files efficiently\n");
  110.   rprintf(F," -n, --dry-run               show what would have been transferred\n");
  111.   rprintf(F," -W, --whole-file            copy whole files, no incremental checks\n");
  112.   rprintf(F," -x, --one-file-system       don't cross filesystem boundaries\n");
  113.   rprintf(F," -B, --block-size=SIZE       checksum blocking size\n");  
  114.   rprintf(F," -e, --rsh=COMMAND           specify rsh replacement\n");
  115.   rprintf(F,"     --rsync-path=PATH       specify path to rsync on the remote machine\n");
  116.   rprintf(F," -C, --cvs-exclude           auto ignore files in the same way CVS does\n");
  117.   rprintf(F,"     --delete                delete files that don't exist on the sending side\n");
  118.   rprintf(F,"     --partial               keep partially transferred files\n");
  119.   rprintf(F,"     --force                 force deletion of directories even if not empty\n");
  120.   rprintf(F,"     --numeric-ids           don't map uid/gid values by user/group name\n");
  121.   rprintf(F,"     --timeout=TIME          set IO timeout in seconds\n");
  122.   rprintf(F," -I, --ignore-times          don't exclude files that match length and time\n");
  123.   rprintf(F," -T  --temp-dir=DIR          create temporary files in directory DIR\n");
  124.   rprintf(F,"     --compare-dest=DIR      also compare destination files relative to DIR\n");
  125.   rprintf(F," -z, --compress              compress file data\n");
  126.   rprintf(F,"     --exclude=PATTERN       exclude files matching PATTERN\n");
  127.   rprintf(F,"     --exclude-from=FILE     exclude patterns listed in FILE\n");
  128.   rprintf(F,"     --include=PATTERN       don't exclude files matching PATTERN\n");
  129.   rprintf(F,"     --include-from=FILE     don't exclude patterns listed in FILE\n");
  130.   rprintf(F,"     --suffix=SUFFIX         override backup suffix\n");  
  131.   rprintf(F,"     --version               print version number\n");  
  132. #ifndef __EMX__
  133.   rprintf(F,"     --daemon                run as a rsync daemon\n");  
  134. #endif
  135.   rprintf(F,"     --config=FILE           specify alternate rsyncd.conf file\n");  
  136.   rprintf(F,"     --port=PORT             specify alternate rsyncd port number\n");
  137.   rprintf(F,"     --stats                 give some file transfer stats\n");  
  138.   rprintf(F,"     --progress              show progress during transfer\n");  
  139.   rprintf(F,"     --log-format=FORMAT     log file transfers using specified format\n");  
  140.   rprintf(F," -h, --help                  show this help screen\n");
  141.  
  142.   rprintf(F,"\n");
  143.   rprintf(F,"the backup suffix defaults to %s\n",BACKUP_SUFFIX);
  144.   rprintf(F,"the block size defaults to %d\n",BLOCK_SIZE);  
  145.  
  146.   rprintf(F,"\nPlease see the rsync(1) and rsyncd.conf(5) man pages for full documentation\n");
  147.   rprintf(F,"See http://rsync.samba.org/ for updates and bug reports\n");
  148. }
  149.  
  150. enum {OPT_VERSION,OPT_SUFFIX,OPT_SENDER,OPT_SERVER,OPT_EXCLUDE,
  151.       OPT_EXCLUDE_FROM,OPT_DELETE,OPT_NUMERIC_IDS,OPT_RSYNC_PATH,
  152.       OPT_FORCE,OPT_TIMEOUT,OPT_DAEMON,OPT_CONFIG,OPT_PORT,
  153.       OPT_INCLUDE, OPT_INCLUDE_FROM, OPT_STATS, OPT_PARTIAL, OPT_PROGRESS,
  154.       OPT_SAFE_LINKS, OPT_COMPARE_DEST, OPT_LOG_FORMAT};
  155.  
  156. static char *short_options = "oblLWHpguDCtcahvrRIxnSe:B:T:z";
  157.  
  158. static struct option long_options[] = {
  159.   {"version",     0,     0,    OPT_VERSION},
  160.   {"server",      0,     0,    OPT_SERVER},
  161.   {"sender",      0,     0,    OPT_SENDER},
  162.   {"delete",      0,     0,    OPT_DELETE},
  163.   {"force",       0,     0,    OPT_FORCE},
  164.   {"numeric-ids", 0,     0,    OPT_NUMERIC_IDS},
  165.   {"exclude",     1,     0,    OPT_EXCLUDE},
  166.   {"exclude-from",1,     0,    OPT_EXCLUDE_FROM},
  167.   {"include",     1,     0,    OPT_INCLUDE},
  168.   {"include-from",1,     0,    OPT_INCLUDE_FROM},
  169.   {"rsync-path",  1,     0,    OPT_RSYNC_PATH},
  170.   {"one-file-system",0,  0,    'x'},
  171.   {"ignore-times",0,     0,    'I'},
  172.   {"help",        0,     0,    'h'},
  173.   {"dry-run",     0,     0,    'n'},
  174.   {"sparse",      0,     0,    'S'},
  175.   {"cvs-exclude", 0,     0,    'C'},
  176.   {"archive",     0,     0,    'a'},
  177.   {"checksum",    0,     0,    'c'},
  178.   {"backup",      0,     0,    'b'},
  179.   {"update",      0,     0,    'u'},
  180.   {"verbose",     0,     0,    'v'},
  181.   {"recursive",   0,     0,    'r'},
  182.   {"relative",    0,     0,    'R'},
  183.   {"devices",     0,     0,    'D'},
  184.   {"perms",       0,     0,    'p'},
  185.   {"links",       0,     0,    'l'},
  186.   {"copy-links",  0,     0,    'L'},
  187.   {"safe-links",  0,     0,    OPT_SAFE_LINKS},
  188.   {"whole-file",  0,     0,    'W'},
  189.   {"hard-links",  0,     0,    'H'},
  190.   {"owner",       0,     0,    'o'},
  191.   {"group",       0,     0,    'g'},
  192.   {"times",       0,     0,    't'},
  193.   {"rsh",         1,     0,    'e'},
  194.   {"suffix",      1,     0,    OPT_SUFFIX},
  195.   {"block-size",  1,     0,    'B'},
  196.   {"timeout",     1,     0,    OPT_TIMEOUT},
  197.   {"temp-dir",    1,     0,    'T'},
  198.   {"compare-dest", 1,    0,    OPT_COMPARE_DEST},
  199.   {"compress",      0,     0,    'z'},
  200. #ifndef __EMX__
  201.   {"daemon",      0,     0,    OPT_DAEMON},
  202. #endif
  203.   {"stats",       0,     0,    OPT_STATS},
  204.   {"progress",    0,     0,    OPT_PROGRESS},
  205.   {"partial",     0,     0,    OPT_PARTIAL},
  206.   {"config",      1,     0,    OPT_CONFIG},
  207.   {"port",        1,     0,    OPT_PORT},
  208.   {"log-format",  1,     0,    OPT_LOG_FORMAT},
  209.   {0,0,0,0}};
  210.  
  211.  
  212. static char err_buf[100];
  213.  
  214. void option_error(void)
  215. {
  216.     if (err_buf[0]) {
  217.         rprintf(FLOG,"%s", err_buf);
  218.         rprintf(FERROR,"%s", err_buf);
  219.     } else {
  220.         rprintf(FLOG,"Error parsing options - unsupported option?\n");
  221.         rprintf(FERROR,"Error parsing options - unsupported option?\n");
  222.     }
  223.     exit_cleanup(RERR_UNSUPPORTED);
  224. }
  225.  
  226. /* check to see if we should refuse this option */
  227. static int check_refuse_options(char *ref, int opt)
  228. {
  229.     int i, len;
  230.     char *p;
  231.     const char *name;
  232.  
  233.     for (i=0; long_options[i].name; i++) {
  234.         if (long_options[i].val == opt) break;
  235.     }
  236.     
  237.     if (!long_options[i].name) return 0;
  238.  
  239.     name = long_options[i].name;
  240.     len = strlen(name);
  241.  
  242.     while ((p = strstr(ref,name))) {
  243.         if ((p==ref || p[-1]==' ') &&
  244.             (p[len] == ' ' || p[len] == 0)) {
  245.             slprintf(err_buf,sizeof(err_buf),
  246.                  "The '%s' option is not supported by this server\n", name);
  247.             return 1;
  248.         }
  249.         ref += len;
  250.     }
  251.     return 0;
  252. }
  253.  
  254.  
  255. int parse_arguments(int argc, char *argv[])
  256. {
  257.     int opt;
  258.     int option_index;
  259.     char *ref = lp_refuse_options(module_id);
  260.  
  261.     while ((opt = getopt_long(argc, argv, 
  262.                   short_options, long_options, &option_index)) 
  263.            != -1) {
  264.  
  265.         if (ref) {
  266.             if (check_refuse_options(ref, opt)) return 0;
  267.         }
  268.  
  269.         switch (opt) {
  270.         case OPT_VERSION:
  271.             rprintf(FINFO,"rsync version %s  protocol version %d\n\n",
  272.                 VERSION,PROTOCOL_VERSION);
  273.             rprintf(FINFO,"Written by Andrew Tridgell and Paul Mackerras\n");
  274.             exit_cleanup(0);
  275.             
  276.         case OPT_SUFFIX:
  277.             backup_suffix = optarg;
  278.             break;
  279.             
  280.         case OPT_RSYNC_PATH:
  281.             rsync_path = optarg;
  282.             break;
  283.             
  284.         case 'I':
  285.             ignore_times = 1;
  286.             break;
  287.  
  288.         case 'x':
  289.             one_file_system=1;
  290.             break;
  291.  
  292.         case OPT_DELETE:
  293.             delete_mode = 1;
  294.             break;
  295.  
  296.         case OPT_FORCE:
  297.             force_delete = 1;
  298.             break;
  299.  
  300.         case OPT_NUMERIC_IDS:
  301.             numeric_ids = 1;
  302.             break;
  303.  
  304.         case OPT_EXCLUDE:
  305.             add_exclude(optarg, 0);
  306.             break;
  307.  
  308.         case OPT_INCLUDE:
  309.             add_exclude(optarg, 1);
  310.             break;
  311.  
  312.         case OPT_EXCLUDE_FROM:
  313.             add_exclude_file(optarg,1, 0);
  314.             break;
  315.  
  316.         case OPT_INCLUDE_FROM:
  317.             add_exclude_file(optarg,1, 1);
  318.             break;
  319.  
  320.         case OPT_SAFE_LINKS:
  321.             safe_symlinks=1;
  322.             break;
  323.  
  324.         case 'h':
  325.             usage(FINFO);
  326.             exit_cleanup(0);
  327.  
  328.         case 'b':
  329.             make_backups=1;
  330.             break;
  331.  
  332.         case 'n':
  333.             dry_run=1;
  334.             break;
  335.  
  336.         case 'S':
  337.             sparse_files=1;
  338.             break;
  339.  
  340.         case 'C':
  341.             cvs_exclude=1;
  342.             break;
  343.  
  344.         case 'u':
  345.             update_only=1;
  346.             break;
  347.  
  348.         case 'l':
  349.             preserve_links=1;
  350.             break;
  351.  
  352.         case 'L':
  353.             copy_links=1;
  354.             break;
  355.  
  356.         case 'W':
  357.             whole_file=1;
  358.             break;
  359.  
  360.         case 'H':
  361. #if SUPPORT_HARD_LINKS
  362.             preserve_hard_links=1;
  363. #else 
  364.             slprintf(err_buf,sizeof(err_buf),"hard links are not supported on this server\n");
  365.             rprintf(FERROR,"ERROR: hard links not supported on this platform\n");
  366.             return 0;
  367. #endif
  368.             break;
  369.  
  370.         case 'p':
  371.             preserve_perms=1;
  372.             break;
  373.  
  374.         case 'o':
  375.             preserve_uid=1;
  376.             break;
  377.  
  378.         case 'g':
  379.             preserve_gid=1;
  380.             break;
  381.  
  382.         case 'D':
  383.             preserve_devices=1;
  384.             break;
  385.  
  386.         case 't':
  387.             preserve_times=1;
  388.             break;
  389.  
  390.         case 'c':
  391.             always_checksum=1;
  392.             break;
  393.  
  394.         case 'v':
  395.             verbose++;
  396.             break;
  397.  
  398.         case 'a':
  399.             recurse=1;
  400. #if SUPPORT_LINKS
  401.             preserve_links=1;
  402. #endif
  403.             preserve_perms=1;
  404.             preserve_times=1;
  405.             preserve_gid=1;
  406.             if (am_root) {
  407.                 preserve_devices=1;
  408.                 preserve_uid=1;
  409.             }
  410.             break;
  411.  
  412.         case OPT_SERVER:
  413.             am_server = 1;
  414.             break;
  415.  
  416.         case OPT_SENDER:
  417.             if (!am_server) {
  418.                 usage(FERROR);
  419.                 exit_cleanup(RERR_SYNTAX);
  420.             }
  421.             am_sender = 1;
  422.             break;
  423.  
  424.         case 'r':
  425.             recurse = 1;
  426.             break;
  427.  
  428.         case 'R':
  429.             relative_paths = 1;
  430.             break;
  431.  
  432.         case 'e':
  433.             shell_cmd = optarg;
  434.             break;
  435.  
  436.         case 'B':
  437.             block_size = atoi(optarg);
  438.             break;
  439.  
  440.         case OPT_TIMEOUT:
  441.             io_timeout = atoi(optarg);
  442.             break;
  443.  
  444.         case 'T':
  445.             tmpdir = optarg;
  446.             break;
  447.  
  448.         case OPT_COMPARE_DEST:
  449.             compare_dest = optarg;
  450.             break;
  451.  
  452.         case 'z':
  453.             do_compression = 1;
  454.             break;
  455. #ifndef __EMX__
  456.         case OPT_DAEMON:
  457.             am_daemon = 1;
  458.             break;
  459. #endif
  460.         case OPT_STATS:
  461.             do_stats = 1;
  462.             break;
  463.  
  464.         case OPT_PROGRESS:
  465.             do_progress = 1;
  466.             break;
  467.  
  468.         case OPT_PARTIAL:
  469.             keep_partial = 1;
  470.             break;
  471.  
  472.         case OPT_CONFIG:
  473.             config_file = optarg;
  474.             break;
  475.  
  476.         case OPT_PORT:
  477.             rsync_port = atoi(optarg);
  478.             break;
  479.  
  480.         case OPT_LOG_FORMAT:
  481.             log_format = optarg;
  482.             break;
  483.  
  484.         default:
  485.             slprintf(err_buf,sizeof(err_buf),"unrecognised option\n");
  486.             return 0;
  487.         }
  488.     }
  489.     return 1;
  490. }
  491.  
  492.  
  493. void server_options(char **args,int *argc)
  494. {
  495.     int ac = *argc;
  496.     static char argstr[50];
  497.     static char bsize[30];
  498.     static char iotime[30];
  499.     int i, x;
  500.  
  501.     args[ac++] = "--server";
  502.  
  503.     if (!am_sender)
  504.         args[ac++] = "--sender";
  505.  
  506.     x = 1;
  507.     argstr[0] = '-';
  508.     for (i=0;i<verbose;i++)
  509.         argstr[x++] = 'v';
  510.     if (make_backups)
  511.         argstr[x++] = 'b';
  512.     if (update_only)
  513.         argstr[x++] = 'u';
  514.     if (dry_run)
  515.         argstr[x++] = 'n';
  516.     if (preserve_links)
  517.         argstr[x++] = 'l';
  518.     if (copy_links)
  519.         argstr[x++] = 'L';
  520.     if (whole_file)
  521.         argstr[x++] = 'W';
  522.     if (preserve_hard_links)
  523.         argstr[x++] = 'H';
  524.     if (preserve_uid)
  525.         argstr[x++] = 'o';
  526.     if (preserve_gid)
  527.         argstr[x++] = 'g';
  528.     if (preserve_devices)
  529.         argstr[x++] = 'D';
  530.     if (preserve_times)
  531.         argstr[x++] = 't';
  532.     if (preserve_perms)
  533.         argstr[x++] = 'p';
  534.     if (recurse)
  535.         argstr[x++] = 'r';
  536.     if (always_checksum)
  537.         argstr[x++] = 'c';
  538.     if (cvs_exclude)
  539.         argstr[x++] = 'C';
  540.     if (ignore_times)
  541.         argstr[x++] = 'I';
  542.     if (relative_paths)
  543.         argstr[x++] = 'R';
  544.     if (one_file_system)
  545.         argstr[x++] = 'x';
  546.     if (sparse_files)
  547.         argstr[x++] = 'S';
  548.     if (do_compression)
  549.         argstr[x++] = 'z';
  550.     argstr[x] = 0;
  551.  
  552.     if (x != 1) args[ac++] = argstr;
  553.  
  554.     if (block_size != BLOCK_SIZE) {
  555.         sprintf(bsize,"-B%d",block_size);
  556.         args[ac++] = bsize;
  557.     }    
  558.  
  559.     if (io_timeout) {
  560.         sprintf(iotime,"--timeout=%d",io_timeout);
  561.         args[ac++] = iotime;
  562.     }    
  563.  
  564.     if (strcmp(backup_suffix, BACKUP_SUFFIX)) {
  565.         args[ac++] = "--suffix";
  566.         args[ac++] = backup_suffix;
  567.     }
  568.  
  569.     if (delete_mode)
  570.         args[ac++] = "--delete";
  571.  
  572.     if (keep_partial)
  573.         args[ac++] = "--partial";
  574.  
  575.     if (force_delete)
  576.         args[ac++] = "--force";
  577.  
  578.     if (safe_symlinks)
  579.         args[ac++] = "--safe-links";
  580.  
  581.     if (numeric_ids)
  582.         args[ac++] = "--numeric-ids";
  583.  
  584.     if (tmpdir) {
  585.         args[ac++] = "--temp-dir";
  586.         args[ac++] = tmpdir;
  587.     }
  588.  
  589.     if (compare_dest && am_sender) {
  590.         /* the server only needs this option if it is not the sender,
  591.          *   and it may be an older version that doesn't know this
  592.          *   option, so don't send it if client is the sender.
  593.          */
  594.         args[ac++] = "--compare-dest";
  595.         args[ac++] = compare_dest;
  596.     }
  597.  
  598.  
  599.     *argc = ac;
  600. }
  601.  
  602.