diff options
author | Richard Jones <rjones@redhat.com> | 2010-05-21 13:07:05 +0100 |
---|---|---|
committer | Richard Jones <rjones@redhat.com> | 2010-05-21 14:51:53 +0100 |
commit | 5e1aff7856f721bf5737815a5b65c0de23ab0b0c (patch) | |
tree | ddd3d988143b13ad341cd4fb246e34b84875742b /fish/alloc.c | |
parent | 48a216a06d6fdab3c8292c383a37cd990c0bf939 (diff) | |
download | libguestfs-5e1aff7856f721bf5737815a5b65c0de23ab0b0c.tar.gz libguestfs-5e1aff7856f721bf5737815a5b65c0de23ab0b0c.tar.xz libguestfs-5e1aff7856f721bf5737815a5b65c0de23ab0b0c.zip |
fish: Allow suffixes on number parameters (eg. 1M)
This small change uses the gnulib xstrtoll functionality to
enable suffixes on integer parameters in guestfish. For example:
truncate-size /file 1G
(previously you would have had to given the full number).
This also applies to the 'alloc' and 'sparse' commands (and
indirectly to the -N option). The specification for these commands
has changed slightly, in that 'alloc foo 1MB' would now use SI
units, allocating 1000000 bytes instead of a true megabyte. All
existing uses would use 'alloc foo 1M' which still allocates true
megabytes.
Diffstat (limited to 'fish/alloc.c')
-rw-r--r-- | fish/alloc.c | 34 |
1 files changed, 10 insertions, 24 deletions
diff --git a/fish/alloc.c b/fish/alloc.c index f91c5bb4..75337414 100644 --- a/fish/alloc.c +++ b/fish/alloc.c @@ -26,6 +26,8 @@ #include <inttypes.h> #include <errno.h> +#include "xstrtol.h" + #include "fish.h" int @@ -145,30 +147,14 @@ alloc_disk (const char *filename, const char *size_str, int add, int sparse) static int parse_size (const char *str, off_t *size_rtn) { - uint64_t size; - char type; - - /* Note that the parsing here is looser than what is specified in the - * help, but we may tighten it up in future so beware. - */ - if (sscanf (str, "%"SCNu64"%c", &size, &type) == 2) { - switch (type) { - case 'k': case 'K': size *= 1024ULL; break; - case 'm': case 'M': size *= 1024ULL * 1024ULL; break; - case 'g': case 'G': size *= 1024ULL * 1024ULL * 1024ULL; break; - case 't': case 'T': size *= 1024ULL * 1024ULL * 1024ULL * 1024ULL; break; - case 'p': case 'P': size *= 1024ULL * 1024ULL * 1024ULL * 1024ULL * 1024ULL; break; - case 'e': case 'E': size *= 1024ULL * 1024ULL * 1024ULL * 1024ULL * 1024ULL * 1024ULL; break; - case 's': size *= 512; break; - default: - fprintf (stderr, _("could not parse size specification '%s'\n"), str); - return -1; - } - } - else if (sscanf (str, "%"SCNu64, &size) == 1) - size *= 1024ULL; - else { - fprintf (stderr, _("could not parse size specification '%s'\n"), str); + unsigned long long size; + strtol_error xerr; + + xerr = xstrtoull (str, NULL, 0, &size, "0kKMGTPEZY"); + if (xerr != LONGINT_OK) { + fprintf (stderr, + _("%s: invalid integer parameter (%s returned %d)\n"), + "alloc_disk", "xstrtoull", xerr); return -1; } |