Parse the new --human-readable (-m) option.
[rsync/rsync.git] / options.c
index 6a74117..b36f125 100644 (file)
--- a/options.c
+++ b/options.c
@@ -20,6 +20,7 @@
 
 #include "rsync.h"
 #include "popt.h"
+#include "zlib/zlib.h"
 
 extern int module_id;
 extern int sanitize_paths;
@@ -64,6 +65,7 @@ int one_file_system = 0;
 int protocol_version = PROTOCOL_VERSION;
 int sparse_files = 0;
 int do_compression = 0;
+int def_compress_level = Z_DEFAULT_COMPRESSION;
 int am_root = 0;
 int am_server = 0;
 int am_sender = 0;
@@ -80,6 +82,7 @@ char *files_from = NULL;
 int filesfrom_fd = -1;
 char *filesfrom_host = NULL;
 int eol_nulls = 0;
+int human_readable = 0;
 int recurse = 0;
 int xfer_dirs = -1;
 int am_daemon = 0;
@@ -142,6 +145,7 @@ char *log_format = NULL;
 char *password_file = NULL;
 char *rsync_path = RSYNC_PATH;
 char *backup_dir = NULL;
+char *chmod_mode = NULL;
 char backup_dir_buf[MAXPATHLEN];
 int rsync_port = 0;
 int compare_dest = 0;
@@ -161,11 +165,13 @@ int list_only = 0;
 #define MAX_BATCH_NAME_LEN 256 /* Must be less than MAXPATHLEN-13 */
 char *batch_name = NULL;
 
+struct chmod_mode_struct *chmod_modes = NULL;
+
 static int daemon_opt;   /* sets am_daemon after option error-reporting */
 static int F_option_cnt = 0;
 static int modify_window_set;
 static int itemize_changes = 0;
-static int refused_delete, refused_archive_part;
+static int refused_delete, refused_archive_part, refused_compress;
 static int refused_partial, refused_progress, refused_delete_before;
 static int refused_inplace;
 static char *max_size_arg, *min_size_arg;
@@ -289,6 +295,7 @@ void usage(enum logcode F)
   rprintf(F," -D, --devices               preserve devices (root only)\n");
   rprintf(F," -t, --times                 preserve times\n");
   rprintf(F," -O, --omit-dir-times        omit directories when preserving times\n");
+  rprintf(F,"     --chmod=CHMOD           change destination permissions\n");
   rprintf(F," -S, --sparse                handle sparse files efficiently\n");
   rprintf(F," -n, --dry-run               show what would have been transferred\n");
   rprintf(F," -W, --whole-file            copy files whole (without rsync algorithm)\n");
@@ -324,6 +331,7 @@ void usage(enum logcode F)
   rprintf(F,"     --copy-dest=DIR         ... and include copies of unchanged files\n");
   rprintf(F,"     --link-dest=DIR         hardlink to files in DIR when unchanged\n");
   rprintf(F," -z, --compress              compress file data during the transfer\n");
+  rprintf(F,"     --compress-level=NUM    explicitly set compression level\n");
   rprintf(F," -C, --cvs-exclude           auto-ignore files the same way CVS does\n");
   rprintf(F," -f, --filter=RULE           add a file-filtering RULE\n");
   rprintf(F," -F                          same as --filter='dir-merge /.rsync-filter'\n");
@@ -338,6 +346,8 @@ void usage(enum logcode F)
   rprintf(F,"     --port=PORT             specify double-colon alternate port number\n");
   rprintf(F,"     --blocking-io           use blocking I/O for the remote shell\n");
   rprintf(F,"     --stats                 give some file-transfer stats\n");
+  rprintf(F," -m, --human-readable        output numbers in a human-readable format\n");
+  rprintf(F,"     --si                    like human-readable, but use powers of 1000\n");
   rprintf(F,"     --progress              show progress during transfer\n");
   rprintf(F," -P                          same as --partial --progress\n");
   rprintf(F," -i, --itemize-changes       output a change-summary for all updates\n");
@@ -376,6 +386,8 @@ static struct poptOption long_options[] = {
   {"no-v",             0,  POPT_ARG_VAL,    &verbose, 0, 0, 0 },
   {"quiet",           'q', POPT_ARG_NONE,   0, 'q', 0, 0 },
   {"stats",            0,  POPT_ARG_NONE,   &do_stats, 0, 0, 0 },
+  {"human-readable",  'm', POPT_ARG_VAL,    &human_readable, 1, 0, 0},
+  {"si",               0,  POPT_ARG_VAL,    &human_readable, 2, 0, 0},
   {"dry-run",         'n', POPT_ARG_NONE,   &dry_run, 0, 0, 0 },
   {"archive",         'a', POPT_ARG_NONE,   0, 'a', 0, 0 },
   {"recursive",       'r', POPT_ARG_VAL,    &recurse, 2, 0, 0 },
@@ -413,6 +425,7 @@ static struct poptOption long_options[] = {
   {"no-relative",      0,  POPT_ARG_VAL,    &relative_paths, 0, 0, 0 },
   {"no-R",             0,  POPT_ARG_VAL,    &relative_paths, 0, 0, 0 },
   {"no-implied-dirs",  0,  POPT_ARG_VAL,    &implied_dirs, 0, 0, 0 },
+  {"chmod",            0,  POPT_ARG_STRING, &chmod_mode, 0, 0, 0 },
   {"ignore-times",    'I', POPT_ARG_NONE,   &ignore_times, 0, 0, 0 },
   {"size-only",        0,  POPT_ARG_NONE,   &size_only, 0, 0, 0 },
   {"one-file-system", 'x', POPT_ARG_NONE,   &one_file_system, 0, 0, 0 },
@@ -451,7 +464,8 @@ static struct poptOption long_options[] = {
   {"copy-dest",        0,  POPT_ARG_STRING, 0, OPT_COPY_DEST, 0, 0 },
   {"link-dest",        0,  POPT_ARG_STRING, 0, OPT_LINK_DEST, 0, 0 },
   {"fuzzy",           'y', POPT_ARG_NONE,   &fuzzy_basis, 0, 0, 0 },
-  {"compress",        'z', POPT_ARG_NONE,   &do_compression, 0, 0, 0 },
+  {"compress",        'z', POPT_ARG_NONE,   0, 'z', 0, 0 },
+  {"compress-level",   0,  POPT_ARG_INT,    &def_compress_level, 'z', 0, 0 },
   {0,                 'P', POPT_ARG_NONE,   0, 'P', 0, 0 },
   {"progress",         0,  POPT_ARG_VAL,    &do_progress, 1, 0, 0 },
   {"no-progress",      0,  POPT_ARG_VAL,    &do_progress, 0, 0, 0 },
@@ -598,6 +612,9 @@ static void set_refuse_options(char *bp)
                                case 't': case 'g': case 'o': case 'D':
                                        refused_archive_part = op->val;
                                        break;
+                               case 'z':
+                                       refused_compress = op->val;
+                                       break;
                                case '\0':
                                        if (wildmatch("delete", op->longName))
                                                refused_delete = op->val;
@@ -691,12 +708,10 @@ static OFF_T parse_size_arg(char **size_arg, char def_suf)
        if (size > 0 && make_compatible) {
                /* We convert this manually because we may need %lld precision,
                 * and that's not a portable sprintf() escape. */
-               char buf[128], *s = buf + sizeof buf;
+               char buf[128], *s = buf + sizeof buf - 1;
                OFF_T num = size;
-               *--s = '\0';
+               *s = '\0';
                while (num) {
-                       if (s == buf) /* impossible... */
-                               out_of_memory("parse_size_arg@buf");
                        *--s = (num % 10) + '0';
                        num /= 10;
                }
@@ -901,6 +916,21 @@ int parse_arguments(int *argc, const char ***argv, int frommain)
                        keep_partial = 1;
                        break;
 
+               case 'z':
+                       if (def_compress_level < Z_DEFAULT_COMPRESSION
+                        || def_compress_level > Z_BEST_COMPRESSION) {
+                               snprintf(err_buf, sizeof err_buf,
+                                       "--compress-level value is invalid: %d\n",
+                                       def_compress_level);
+                               return 0;
+                       }
+                       do_compression = def_compress_level != Z_NO_COMPRESSION;
+                       if (do_compression && refused_compress) {
+                               create_refuse_error(refused_compress);
+                               return 0;
+                       }
+                       break;
+
                case OPT_WRITE_BATCH:
                        /* batch_name is already set */
                        write_batch = 1;
@@ -1173,6 +1203,12 @@ int parse_arguments(int *argc, const char ***argv, int frommain)
        if (make_backups && !backup_dir)
                omit_dir_times = 1;
 
+       if (chmod_mode && !(chmod_modes = parse_chmod(chmod_mode))) {
+               snprintf(err_buf, sizeof err_buf,
+                   "Invalid argument passed to chmod\n");
+               return 0;
+       }
+
        if (log_format) {
                if (log_format_has(log_format, 'i'))
                        log_format_has_i = 1;
@@ -1431,6 +1467,12 @@ void server_options(char **args,int *argc)
        if (list_only > 1)
                args[ac++] = "--list-only";
 
+       if (do_compression && def_compress_level != Z_DEFAULT_COMPRESSION) {
+               if (asprintf(&arg, "--compress-level=%d", def_compress_level) < 0)
+                       goto oom;
+               args[ac++] = arg;
+       }
+
        /* The server side doesn't use our log-format, but in certain
         * circumstances they need to know a little about the option. */
        if (log_format && am_sender) {
@@ -1572,6 +1614,11 @@ void server_options(char **args,int *argc)
                }
        }
 
+       if (chmod_mode && !am_sender) {
+               args[ac++] = "--chmod";
+               args[ac++] = chmod_mode;
+       }
+
        if (files_from && (!am_sender || filesfrom_host)) {
                if (filesfrom_host) {
                        args[ac++] = "--files-from";