X-Git-Url: http://git.annexia.org/?a=blobdiff_plain;ds=sidebyside;f=fish%2Falloc.c;h=29d9845b4c74bd13618dbfdf5c4ca12d26aff39d;hb=42b62b8240a9b32f0a198105b03cfadaf97a7a7b;hp=f91c5bb4195e32ecce9884ce7893107069d62aee;hpb=4a9b979a3199dfef64a342f05d666da540203a7c;p=libguestfs.git diff --git a/fish/alloc.c b/fish/alloc.c index f91c5bb..29d9845 100644 --- a/fish/alloc.c +++ b/fish/alloc.c @@ -26,6 +26,8 @@ #include #include +#include "xstrtol.h" + #include "fish.h" int @@ -56,8 +58,6 @@ do_sparse (const char *cmd, int argc, char *argv[]) return 0; } -static int parse_size (const char *str, off_t *size_rtn); - /* This is the underlying allocation function. It's called from * a few other places in guestfish. */ @@ -142,33 +142,17 @@ alloc_disk (const char *filename, const char *size_str, int add, int sparse) return 0; } -static int +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"), + "parse_size", "xstrtoull", xerr); return -1; }